1 /* 2 * Copyright (c) 2011-2012 The DragonFly Project. All rights reserved. 3 * 4 * This code is derived from software contributed to The DragonFly Project 5 * by Matthew Dillon <dillon@dragonflybsd.org> 6 * by Venkatesh Srinivas <vsrinivas@dragonflybsd.org> 7 * 8 * Redistribution and use in source and binary forms, with or without 9 * modification, are permitted provided that the following conditions 10 * are met: 11 * 12 * 1. Redistributions of source code must retain the above copyright 13 * notice, this list of conditions and the following disclaimer. 14 * 2. Redistributions in binary form must reproduce the above copyright 15 * notice, this list of conditions and the following disclaimer in 16 * the documentation and/or other materials provided with the 17 * distribution. 18 * 3. Neither the name of The DragonFly Project nor the names of its 19 * contributors may be used to endorse or promote products derived 20 * from this software without specific, prior written permission. 21 * 22 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 23 * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 24 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS 25 * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE 26 * COPYRIGHT HOLDERS OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, 27 * INCIDENTAL, SPECIAL, EXEMPLARY OR CONSEQUENTIAL DAMAGES (INCLUDING, 28 * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; 29 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED 30 * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, 31 * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT 32 * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 33 * SUCH DAMAGE. 34 */ 35 36 #include "hammer2.h" 37 38 #define SHOW_TAB 2 39 40 static void shell_rcvmsg(dmsg_msg_t *msg); 41 static void shell_ttymsg(dmsg_iocom_t *iocom); 42 43 /************************************************************************ 44 * SHELL * 45 ************************************************************************/ 46 47 int 48 cmd_shell(const char *hostname) 49 { 50 struct dmsg_iocom iocom; 51 dmsg_msg_t *msg; 52 int fd; 53 54 /* 55 * Connect to the target 56 */ 57 fd = dmsg_connect(hostname); 58 if (fd < 0) 59 return 1; 60 61 /* 62 * Run the session. The remote end transmits our prompt. 63 */ 64 dmsg_iocom_init(&iocom, fd, 0, 65 NULL, 66 shell_rcvmsg, 67 hammer2_shell_parse, 68 shell_ttymsg); 69 fcntl(0, F_SETFL, O_NONBLOCK); 70 printf("debug: connected\n"); 71 72 msg = dmsg_msg_alloc(&iocom.circuit0, 0, DMSG_DBG_SHELL, NULL, NULL); 73 dmsg_msg_write(msg); 74 dmsg_iocom_core(&iocom); 75 fprintf(stderr, "debug: disconnected\n"); 76 close(fd); 77 return 0; 78 } 79 80 /* 81 * Callback from dmsg_iocom_core() when messages might be present 82 * on the socket. 83 */ 84 static 85 void 86 shell_rcvmsg(dmsg_msg_t *msg) 87 { 88 switch(msg->any.head.cmd & DMSGF_TRANSMASK) { 89 case DMSG_LNK_ERROR: 90 case DMSG_LNK_ERROR | DMSGF_REPLY: 91 /* 92 * One-way non-transactional LNK_ERROR messages typically 93 * indicate a connection failure. Error code 0 is used by 94 * the debug shell to indicate no more results from last cmd. 95 */ 96 if (msg->any.head.error) { 97 fprintf(stderr, "Stream failure: %s\n", 98 dmsg_msg_str(msg)); 99 } else { 100 write(1, "debug> ", 7); 101 } 102 break; 103 case DMSG_LNK_ERROR | DMSGF_DELETE: 104 /* ignore termination of LNK_CONN */ 105 break; 106 case DMSG_DBG_SHELL: 107 /* 108 * We send the commands, not accept them. 109 * (one-way message, not transactional) 110 */ 111 dmsg_msg_reply(msg, DMSG_ERR_NOSUPP); 112 break; 113 case DMSG_DBG_SHELL | DMSGF_REPLY: 114 /* 115 * A reply from the remote is data we copy to stdout. 116 * (one-way message, not transactional) 117 */ 118 if (msg->aux_size) { 119 msg->aux_data[msg->aux_size - 1] = 0; 120 write(1, msg->aux_data, strlen(msg->aux_data)); 121 } 122 break; 123 case DMSG_LNK_CONN | DMSGF_CREATE: 124 fprintf(stderr, "Debug Shell is ignoring received LNK_CONN\n"); 125 dmsg_msg_reply(msg, DMSG_ERR_NOSUPP); 126 break; 127 case DMSG_LNK_CONN | DMSGF_DELETE: 128 break; 129 default: 130 /* 131 * Ignore any unknown messages, Terminate any unknown 132 * transactions with an error. 133 */ 134 fprintf(stderr, "Unknown message: %s\n", dmsg_msg_str(msg)); 135 if (msg->any.head.cmd & DMSGF_CREATE) 136 dmsg_msg_reply(msg, DMSG_ERR_NOSUPP); 137 if (msg->any.head.cmd & DMSGF_DELETE) 138 dmsg_msg_reply(msg, DMSG_ERR_NOSUPP); 139 break; 140 } 141 } 142 143 static 144 void 145 shell_ttymsg(dmsg_iocom_t *iocom) 146 { 147 dmsg_msg_t *msg; 148 char buf[256]; 149 size_t len; 150 151 if (fgets(buf, sizeof(buf), stdin) != NULL) { 152 len = strlen(buf); 153 if (len && buf[len - 1] == '\n') 154 buf[--len] = 0; 155 ++len; 156 msg = dmsg_msg_alloc(&iocom->circuit0, len, DMSG_DBG_SHELL, 157 NULL, NULL); 158 bcopy(buf, msg->aux_data, len); 159 dmsg_msg_write(msg); 160 } else if (feof(stdin)) { 161 /* 162 * Set EOF flag without setting any error code for normal 163 * EOF. 164 */ 165 iocom->flags |= DMSG_IOCOMF_EOF; 166 } else { 167 clearerr(stdin); 168 } 169 } 170 171 static void shell_span(dmsg_circuit_t *circuit, char *cmdbuf); 172 static void shell_circ(dmsg_circuit_t *circuit, char *cmdbuf); 173 174 void 175 hammer2_shell_parse(dmsg_msg_t *msg) 176 { 177 dmsg_circuit_t *circuit = msg->circuit; 178 char *cmdbuf = msg->aux_data; 179 char *cmd = strsep(&cmdbuf, " \t"); 180 181 if (cmd == NULL || *cmd == 0) { 182 ; 183 } else if (strcmp(cmd, "span") == 0) { 184 shell_span(circuit, cmdbuf); 185 } else if (strcmp(cmd, "circ") == 0) { 186 shell_circ(circuit, cmdbuf); 187 } else if (strcmp(cmd, "tree") == 0) { 188 dmsg_shell_tree(circuit, cmdbuf); /* dump spanning tree */ 189 } else if (strcmp(cmd, "help") == 0 || strcmp(cmd, "?") == 0) { 190 dmsg_circuit_printf(circuit, "help Command help\n"); 191 dmsg_circuit_printf(circuit, "span <host> Span to target host\n"); 192 dmsg_circuit_printf(circuit, "circ <msgid> Create VC to msgid of rx SPAN\n"); 193 dmsg_circuit_printf(circuit, "tree Dump spanning tree\n"); 194 } else { 195 dmsg_circuit_printf(circuit, "Unrecognized command: %s\n", cmd); 196 } 197 } 198 199 static void 200 shell_span(dmsg_circuit_t *circuit, char *cmdbuf) 201 { 202 dmsg_master_service_info_t *info; 203 const char *hostname = strsep(&cmdbuf, " \t"); 204 pthread_t thread; 205 int fd; 206 207 /* 208 * Connect to the target 209 */ 210 if (hostname == NULL) { 211 fd = -1; 212 } else { 213 fd = dmsg_connect(hostname); 214 } 215 216 /* 217 * Start master service 218 */ 219 if (fd < 0) { 220 dmsg_circuit_printf(circuit, 221 "Connection to %s failed\n", 222 hostname); 223 } else { 224 dmsg_circuit_printf(circuit, "Connected to %s\n", hostname); 225 226 info = malloc(sizeof(*info)); 227 bzero(info, sizeof(*info)); 228 info->fd = fd; 229 info->detachme = 1; 230 info->dbgmsg_callback = hammer2_shell_parse; 231 info->label = strdup("client"); 232 233 pthread_create(&thread, NULL, dmsg_master_service, info); 234 /*pthread_join(thread, &res);*/ 235 } 236 } 237 238 static void shell_circ_reply(dmsg_msg_t *msg); 239 240 static void 241 shell_circ(dmsg_circuit_t *circuit, char *cmdbuf) 242 { 243 uint64_t msgid = strtoull(cmdbuf, NULL, 16); 244 dmsg_state_t *state; 245 dmsg_msg_t *msg; 246 247 if (dmsg_debug_findspan(msgid, &state) == 0) { 248 dmsg_circuit_printf(circuit, "Found state %p\n", state); 249 250 dmsg_circuit_printf(circuit, "Establishing CIRC\n"); 251 msg = dmsg_msg_alloc(&state->iocom->circuit0, 0, 252 DMSG_LNK_CIRC | DMSGF_CREATE, 253 shell_circ_reply, circuit); 254 msg->any.lnk_circ.target = state->msgid; 255 dmsg_msg_write(msg); 256 } else { 257 dmsg_circuit_printf(circuit, 258 "Unable to locate %016jx\n", 259 (intmax_t)msgid); 260 } 261 } 262 263 static void 264 shell_circ_reply(dmsg_msg_t *msg) 265 { 266 dmsg_circuit_t *circ = msg->state->any.circ; 267 268 if (msg->any.head.cmd & DMSGF_DELETE) { 269 dmsg_circuit_printf(circ, "rxmsg DELETE error %d\n", 270 msg->any.head.error); 271 msg->state->any.circ = NULL; 272 } else { 273 dmsg_circuit_printf(circ, "rxmsg result error %d\n", 274 msg->any.head.error); 275 } 276 } 277 278 /************************************************************************ 279 * DEBUGSPAN * 280 ************************************************************************ 281 * 282 * Connect to the target manually (not via the cluster list embedded in 283 * a hammer2 filesystem) and initiate the SPAN protocol. 284 */ 285 int 286 cmd_debugspan(const char *hostname) 287 { 288 pthread_t thread; 289 int fd; 290 void *res; 291 292 /* 293 * Connect to the target 294 */ 295 fd = dmsg_connect(hostname); 296 if (fd < 0) 297 return 1; 298 299 printf("debugspan: connected to %s, starting CONN/SPAN\n", hostname); 300 pthread_create(&thread, NULL, 301 dmsg_master_service, (void *)(intptr_t)fd); 302 pthread_join(thread, &res); 303 return(0); 304 } 305 306 /************************************************************************ 307 * SHOW * 308 ************************************************************************/ 309 310 static void show_bref(int fd, int tab, int bi, hammer2_blockref_t *bref, 311 int dofreemap); 312 static void tabprintf(int tab, const char *ctl, ...); 313 314 int 315 cmd_show(const char *devpath, int dofreemap) 316 { 317 hammer2_blockref_t broot; 318 hammer2_blockref_t best; 319 hammer2_media_data_t media; 320 int fd; 321 int i; 322 int best_i; 323 324 fd = open(devpath, O_RDONLY); 325 if (fd < 0) { 326 perror("open"); 327 return 1; 328 } 329 330 /* 331 * Show the tree using the best volume header. 332 * -vvv will show the tree for all four volume headers. 333 */ 334 best_i = -1; 335 bzero(&best, sizeof(best)); 336 for (i = 0; i < 4; ++i) { 337 bzero(&broot, sizeof(broot)); 338 broot.type = HAMMER2_BREF_TYPE_VOLUME; 339 broot.data_off = (i * HAMMER2_ZONE_BYTES64) | 340 HAMMER2_PBUFRADIX; 341 lseek(fd, broot.data_off & ~HAMMER2_OFF_MASK_RADIX, 0); 342 if (read(fd, &media, HAMMER2_PBUFSIZE) == 343 (ssize_t)HAMMER2_PBUFSIZE) { 344 broot.mirror_tid = media.voldata.mirror_tid; 345 if (best_i < 0 || best.mirror_tid < broot.mirror_tid) { 346 best_i = i; 347 best = broot; 348 } 349 if (VerboseOpt >= 3) 350 show_bref(fd, 0, i, &broot, dofreemap); 351 } 352 } 353 if (VerboseOpt < 3) 354 show_bref(fd, 0, best_i, &best, dofreemap); 355 close(fd); 356 357 return 0; 358 } 359 360 static void 361 show_bref(int fd, int tab, int bi, hammer2_blockref_t *bref, int dofreemap) 362 { 363 hammer2_media_data_t media; 364 hammer2_blockref_t *bscan; 365 int bcount; 366 int i; 367 int didnl; 368 int namelen; 369 int obrace = 1; 370 size_t bytes; 371 const char *type_str; 372 char *str = NULL; 373 374 switch(bref->type) { 375 case HAMMER2_BREF_TYPE_EMPTY: 376 type_str = "empty"; 377 break; 378 case HAMMER2_BREF_TYPE_INODE: 379 type_str = "inode"; 380 break; 381 case HAMMER2_BREF_TYPE_INDIRECT: 382 type_str = "indblk"; 383 break; 384 case HAMMER2_BREF_TYPE_DATA: 385 type_str = "data"; 386 break; 387 case HAMMER2_BREF_TYPE_VOLUME: 388 type_str = "volume"; 389 break; 390 case HAMMER2_BREF_TYPE_FREEMAP: 391 type_str = "freemap"; 392 break; 393 case HAMMER2_BREF_TYPE_FREEMAP_NODE: 394 type_str = "fmapnode"; 395 break; 396 case HAMMER2_BREF_TYPE_FREEMAP_LEAF: 397 type_str = "fbitmap"; 398 break; 399 default: 400 type_str = "unknown"; 401 break; 402 } 403 404 tabprintf(tab, "%s.%-3d %016jx %016jx/%-2d mir=%016jx mod=%016jx ", 405 type_str, bi, (intmax_t)bref->data_off, 406 (intmax_t)bref->key, (intmax_t)bref->keybits, 407 (intmax_t)bref->mirror_tid, (intmax_t)bref->modify_tid); 408 tab += SHOW_TAB; 409 410 bytes = (size_t)1 << (bref->data_off & HAMMER2_OFF_MASK_RADIX); 411 412 { 413 hammer2_off_t io_off; 414 hammer2_off_t io_base; 415 size_t io_bytes; 416 size_t boff; 417 418 io_off = bref->data_off & ~HAMMER2_OFF_MASK_RADIX; 419 io_base = io_off & ~(hammer2_off_t)(HAMMER2_MINIOSIZE - 1); 420 io_bytes = bytes; 421 boff = io_off - io_base; 422 423 io_bytes = HAMMER2_MINIOSIZE; 424 while (io_bytes + boff < bytes) 425 io_bytes <<= 1; 426 427 if (io_bytes > sizeof(media)) { 428 printf("(bad block size %zd)\n", bytes); 429 return; 430 } 431 if (bref->type != HAMMER2_BREF_TYPE_DATA || VerboseOpt >= 1) { 432 lseek(fd, io_base, 0); 433 if (read(fd, &media, io_bytes) != (ssize_t)io_bytes) { 434 printf("(media read failed)\n"); 435 return; 436 } 437 if (boff) 438 bcopy((char *)&media + boff, &media, bytes); 439 } 440 } 441 442 bscan = NULL; 443 bcount = 0; 444 didnl = 1; 445 446 switch(bref->type) { 447 case HAMMER2_BREF_TYPE_EMPTY: 448 obrace = 0; 449 break; 450 case HAMMER2_BREF_TYPE_INODE: 451 printf("{\n"); 452 if (media.ipdata.op_flags & HAMMER2_OPFLAG_DIRECTDATA) { 453 /* no blockrefs */ 454 } else { 455 bscan = &media.ipdata.u.blockset.blockref[0]; 456 bcount = HAMMER2_SET_COUNT; 457 } 458 namelen = media.ipdata.name_len; 459 if (namelen > HAMMER2_INODE_MAXNAME) 460 namelen = 0; 461 tabprintf(tab, "filename \"%*.*s\"\n", 462 namelen, namelen, media.ipdata.filename); 463 tabprintf(tab, "version %d\n", media.ipdata.version); 464 tabprintf(tab, "uflags 0x%08x\n", 465 media.ipdata.uflags); 466 if (media.ipdata.rmajor || media.ipdata.rminor) { 467 tabprintf(tab, "rmajor %d\n", 468 media.ipdata.rmajor); 469 tabprintf(tab, "rminor %d\n", 470 media.ipdata.rminor); 471 } 472 tabprintf(tab, "ctime %s\n", 473 hammer2_time64_to_str(media.ipdata.ctime, &str)); 474 tabprintf(tab, "mtime %s\n", 475 hammer2_time64_to_str(media.ipdata.mtime, &str)); 476 tabprintf(tab, "atime %s\n", 477 hammer2_time64_to_str(media.ipdata.atime, &str)); 478 tabprintf(tab, "btime %s\n", 479 hammer2_time64_to_str(media.ipdata.btime, &str)); 480 tabprintf(tab, "uid %s\n", 481 hammer2_uuid_to_str(&media.ipdata.uid, &str)); 482 tabprintf(tab, "gid %s\n", 483 hammer2_uuid_to_str(&media.ipdata.gid, &str)); 484 tabprintf(tab, "type %s\n", 485 hammer2_iptype_to_str(media.ipdata.type)); 486 tabprintf(tab, "opflgs 0x%02x\n", 487 media.ipdata.op_flags); 488 tabprintf(tab, "capflgs 0x%04x\n", 489 media.ipdata.cap_flags); 490 tabprintf(tab, "mode %-7o\n", 491 media.ipdata.mode); 492 tabprintf(tab, "inum 0x%016jx\n", 493 media.ipdata.inum); 494 tabprintf(tab, "size %ju\n", 495 (uintmax_t)media.ipdata.size); 496 tabprintf(tab, "nlinks %ju\n", 497 (uintmax_t)media.ipdata.nlinks); 498 tabprintf(tab, "iparent 0x%016jx\n", 499 (uintmax_t)media.ipdata.iparent); 500 tabprintf(tab, "name_key 0x%016jx\n", 501 (uintmax_t)media.ipdata.name_key); 502 tabprintf(tab, "name_len %u\n", 503 media.ipdata.name_len); 504 tabprintf(tab, "ncopies %u\n", 505 media.ipdata.ncopies); 506 tabprintf(tab, "compalg %u\n", 507 media.ipdata.comp_algo); 508 if (media.ipdata.op_flags & HAMMER2_OPFLAG_PFSROOT) { 509 tabprintf(tab, "pfs_type %u (%s)\n", 510 media.ipdata.pfs_type, 511 hammer2_pfstype_to_str(media.ipdata.pfs_type)); 512 tabprintf(tab, "pfs_inum 0x%016jx\n", 513 (uintmax_t)media.ipdata.pfs_inum); 514 tabprintf(tab, "pfs_clid %s\n", 515 hammer2_uuid_to_str(&media.ipdata.pfs_clid, 516 &str)); 517 tabprintf(tab, "pfs_fsid %s\n", 518 hammer2_uuid_to_str(&media.ipdata.pfs_fsid, 519 &str)); 520 } 521 tabprintf(tab, "data_quota %ju\n", 522 (uintmax_t)media.ipdata.data_quota); 523 tabprintf(tab, "data_count %ju\n", 524 (uintmax_t)media.ipdata.data_count); 525 tabprintf(tab, "inode_quota %ju\n", 526 (uintmax_t)media.ipdata.inode_quota); 527 tabprintf(tab, "inode_count %ju\n", 528 (uintmax_t)media.ipdata.inode_count); 529 tabprintf(tab, "attr_tid 0x%016jx\n", 530 (uintmax_t)media.ipdata.attr_tid); 531 if (media.ipdata.type == HAMMER2_OBJTYPE_DIRECTORY) { 532 tabprintf(tab, "dirent_tid %016jx\n", 533 (uintmax_t)media.ipdata.dirent_tid); 534 } 535 break; 536 case HAMMER2_BREF_TYPE_INDIRECT: 537 bscan = &media.npdata[0]; 538 bcount = bytes / sizeof(hammer2_blockref_t); 539 didnl = 1; 540 printf("{\n"); 541 break; 542 case HAMMER2_BREF_TYPE_DATA: 543 if (VerboseOpt >= 2) { 544 printf("{\n"); 545 } else { 546 printf("\n"); 547 obrace = 0; 548 } 549 break; 550 case HAMMER2_BREF_TYPE_VOLUME: 551 if (dofreemap) { 552 bscan = &media.voldata.freemap_blockset.blockref[0]; 553 bcount = HAMMER2_SET_COUNT; 554 } else { 555 bscan = &media.voldata.sroot_blockset.blockref[0]; 556 bcount = HAMMER2_SET_COUNT; 557 } 558 printf("{\n"); 559 break; 560 case HAMMER2_BREF_TYPE_FREEMAP_LEAF: 561 printf("{\n"); 562 for (i = 0; i < HAMMER2_FREEMAP_COUNT; ++i) { 563 if (media.bmdata[i].class == 0 && 564 media.bmdata[i].avail == 0) { 565 continue; 566 } 567 tabprintf(tab + 4, "%04d.%04x (avail=%5d) " 568 "%08x %08x %08x %08x %08x %08x %08x %08x\n", 569 i, media.bmdata[i].class, 570 media.bmdata[i].avail, 571 media.bmdata[i].bitmap[0], 572 media.bmdata[i].bitmap[1], 573 media.bmdata[i].bitmap[2], 574 media.bmdata[i].bitmap[3], 575 media.bmdata[i].bitmap[4], 576 media.bmdata[i].bitmap[5], 577 media.bmdata[i].bitmap[6], 578 media.bmdata[i].bitmap[7]); 579 } 580 tabprintf(tab, "}\n"); 581 break; 582 case HAMMER2_BREF_TYPE_FREEMAP_NODE: 583 printf("{\n"); 584 bscan = &media.npdata[0]; 585 bcount = bytes / sizeof(hammer2_blockref_t); 586 break; 587 default: 588 printf("\n"); 589 obrace = 0; 590 break; 591 } 592 if (str) 593 free(str); 594 for (i = 0; i < bcount; ++i) { 595 if (bscan[i].type != HAMMER2_BREF_TYPE_EMPTY) { 596 if (didnl == 0) { 597 printf("\n"); 598 didnl = 1; 599 } 600 show_bref(fd, tab, i, &bscan[i], dofreemap); 601 } 602 } 603 tab -= SHOW_TAB; 604 if (obrace) { 605 if (bref->type == HAMMER2_BREF_TYPE_INODE) 606 tabprintf(tab, "} (%s.%d, \"%s\")\n", 607 type_str, bi, media.ipdata.filename); 608 else 609 tabprintf(tab, "} (%s.%d)\n", type_str,bi); 610 } 611 } 612 613 int 614 cmd_hash(int ac, const char **av) 615 { 616 int i; 617 618 for (i = 0; i < ac; ++i) { 619 printf("%016jx %s\n", dirhash(av[i], strlen(av[i])), av[i]); 620 } 621 return(0); 622 } 623 624 625 static 626 void 627 tabprintf(int tab, const char *ctl, ...) 628 { 629 va_list va; 630 631 printf("%*.*s", tab, tab, ""); 632 va_start(va, ctl); 633 vprintf(ctl, va); 634 va_end(va); 635 } 636