sbin/fsdb/fsdb.c

   1 /*      $NetBSD: fsdb.c,v 1.2 1995/10/08 23:18:10 thorpej Exp $ */
   2
   3 /*-
   4  * SPDX-License-Identifier: BSD-3-Clause
   5  *
   6  *  Copyright (c) 1995 John T. Kohl
   7  *  All rights reserved.
   8  *
   9  *  Redistribution and use in source and binary forms, with or without
  10  *  modification, are permitted provided that the following conditions
  11  *  are met:
  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 the
  16  *     documentation and/or other materials provided with the distribution.
  17  *  3. The name of the author may not be used to endorse or promote products
  18  *     derived from this software without specific prior written permission.
  19  *
  20  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR `AS IS'' AND ANY EXPRESS OR
  21  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
  22  * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
  23  * DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT,
  24  * INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
  25  * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
  26  * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  27  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
  28  * STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
  29  * ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
  30  * POSSIBILITY OF SUCH DAMAGE.
  31  */
  32
  33 #ifndef lint
  34 static const char rcsid[] =
  35   "$FreeBSD$";
  36 #endif /* not lint */
  37
  38 #include <sys/param.h>
  39 #include <ctype.h>
  40 #include <err.h>
  41 #include <grp.h>
  42 #include <histedit.h>
  43 #include <pwd.h>
  44 #include <stdint.h>
  45 #include <string.h>
  46 #include <time.h>
  47 #include <timeconv.h>
  48
  49 #include <ufs/ufs/dinode.h>
  50 #include <ufs/ufs/dir.h>
  51 #include <ufs/ffs/fs.h>
  52
  53 #include "fsdb.h"
  54 #include "fsck.h"
  55
  56 static void usage(void) __dead2;
  57 int cmdloop(void);
  58 static int compare_blk32(uint32_t *wantedblk, uint32_t curblk);
  59 static int compare_blk64(uint64_t *wantedblk, uint64_t curblk);
  60 static int founddatablk(uint64_t blk);
  61 static int find_blks32(uint32_t *buf, int size, uint32_t *blknum);
  62 static int find_blks64(uint64_t *buf, int size, uint64_t *blknum);
  63 static int find_indirblks32(uint32_t blk, int ind_level, uint32_t *blknum);
  64 static int find_indirblks64(uint64_t blk, int ind_level, uint64_t *blknum);
  65
  66 struct inode curip;
  67 union dinode *curinode;
  68 ino_t curinum, ocurrent;
  69
  70 static void
  71 usage(void)
  72 {
  73         fprintf(stderr, "usage: fsdb [-d] [-f] [-r] fsname\n");
  74         exit(1);
  75 }
  76
  77 /*
  78  * We suck in lots of fsck code, and just pick & choose the stuff we want.
  79  *
  80  * fsreadfd is set up to read from the file system, fswritefd to write to
  81  * the file system.
  82  */
  83 int
  84 main(int argc, char *argv[])
  85 {
  86         int ch, rval;
  87         char *fsys = NULL;
  88
  89         while (-1 != (ch = getopt(argc, argv, "fdr"))) {
  90                 switch (ch) {
  91                 case 'f':
  92                         /* The -f option is left for historical
  93                          * reasons and has no meaning.
  94                          */
  95                         break;
  96                 case 'd':
  97                         debug++;
  98                         break;
  99                 case 'r':
 100                         nflag++; /* "no" in fsck, readonly for us */
 101                         break;
 102                 default:
 103                         usage();
 104                 }
 105         }
 106         argc -= optind;
 107         argv += optind;
 108         if (argc != 1)
 109                 usage();
 110         else
 111                 fsys = argv[0];
 112
 113         sblock_init();
 114         if (!setup(fsys))
 115                 errx(1, "cannot set up file system `%s'", fsys);
 116         if (fswritefd < 0)
 117                 nflag++;
 118         printf("%s file system `%s'\nLast Mounted on %s\n",
 119                nflag? "Examining": "Editing", fsys, sblock.fs_fsmnt);
 120         rval = cmdloop();
 121         if (!nflag) {
 122                 sblock.fs_clean = 0;    /* mark it dirty */
 123                 sbdirty();
 124                 ckfini(0);
 125                 printf("*** FILE SYSTEM MARKED DIRTY\n");
 126                 printf("*** BE SURE TO RUN FSCK TO CLEAN UP ANY DAMAGE\n");
 127                 printf("*** IF IT IS MOUNTED, RE-MOUNT WITH -u -o reload\n");
 128         }
 129         exit(rval);
 130 }
 131
 132 #define CMDFUNC(func) int func(int argc, char *argv[])
 133 #define CMDFUNCSTART(func) int func(int argc, char *argv[])
 134
 135 CMDFUNC(helpfn);
 136 CMDFUNC(focus);                         /* focus on inode */
 137 CMDFUNC(active);                        /* print active inode */
 138 CMDFUNC(blocks);                        /* print blocks for active inode */
 139 CMDFUNC(focusname);                     /* focus by name */
 140 CMDFUNC(zapi);                          /* clear inode */
 141 CMDFUNC(uplink);                        /* incr link */
 142 CMDFUNC(downlink);                      /* decr link */
 143 CMDFUNC(linkcount);                     /* set link count */
 144 CMDFUNC(quit);                          /* quit */
 145 CMDFUNC(findblk);                       /* find block */
 146 CMDFUNC(ls);                            /* list directory */
 147 CMDFUNC(rm);                            /* remove name */
 148 CMDFUNC(ln);                            /* add name */
 149 CMDFUNC(newtype);                       /* change type */
 150 CMDFUNC(chmode);                        /* change mode */
 151 CMDFUNC(chlen);                         /* change length */
 152 CMDFUNC(chaflags);                      /* change flags */
 153 CMDFUNC(chgen);                         /* change generation */
 154 CMDFUNC(chowner);                       /* change owner */
 155 CMDFUNC(chgroup);                       /* Change group */
 156 CMDFUNC(back);                          /* pop back to last ino */
 157 CMDFUNC(chbtime);                       /* Change btime */
 158 CMDFUNC(chmtime);                       /* Change mtime */
 159 CMDFUNC(chctime);                       /* Change ctime */
 160 CMDFUNC(chatime);                       /* Change atime */
 161 CMDFUNC(chinum);                        /* Change inode # of dirent */
 162 CMDFUNC(chname);                        /* Change dirname of dirent */
 163 CMDFUNC(chsize);                        /* Change size */
 164
 165 struct cmdtable cmds[] = {
 166         { "help", "Print out help", 1, 1, FL_RO, helpfn },
 167         { "?", "Print out help", 1, 1, FL_RO, helpfn },
 168         { "inode", "Set active inode to INUM", 2, 2, FL_RO, focus },
 169         { "clri", "Clear inode INUM", 2, 2, FL_WR, zapi },
 170         { "lookup", "Set active inode by looking up NAME", 2, 2, FL_RO | FL_ST, focusname },
 171         { "cd", "Set active inode by looking up NAME", 2, 2, FL_RO | FL_ST, focusname },
 172         { "back", "Go to previous active inode", 1, 1, FL_RO, back },
 173         { "active", "Print active inode", 1, 1, FL_RO, active },
 174         { "print", "Print active inode", 1, 1, FL_RO, active },
 175         { "blocks", "Print block numbers of active inode", 1, 1, FL_RO, blocks },
 176         { "uplink", "Increment link count", 1, 1, FL_WR, uplink },
 177         { "downlink", "Decrement link count", 1, 1, FL_WR, downlink },
 178         { "linkcount", "Set link count to COUNT", 2, 2, FL_WR, linkcount },
 179         { "findblk", "Find inode owning disk block(s)", 2, 33, FL_RO, findblk},
 180         { "ls", "List current inode as directory", 1, 1, FL_RO, ls },
 181         { "rm", "Remove NAME from current inode directory", 2, 2, FL_WR | FL_ST, rm },
 182         { "del", "Remove NAME from current inode directory", 2, 2, FL_WR | FL_ST, rm },
 183         { "ln", "Hardlink INO into current inode directory as NAME", 3, 3, FL_WR | FL_ST, ln },
 184         { "chinum", "Change dir entry number INDEX to INUM", 3, 3, FL_WR, chinum },
 185         { "chname", "Change dir entry number INDEX to NAME", 3, 3, FL_WR | FL_ST, chname },
 186         { "chtype", "Change type of current inode to TYPE", 2, 2, FL_WR, newtype },
 187         { "chmod", "Change mode of current inode to MODE", 2, 2, FL_WR, chmode },
 188         { "chlen", "Change length of current inode to LENGTH", 2, 2, FL_WR, chlen },
 189         { "chown", "Change owner of current inode to OWNER", 2, 2, FL_WR, chowner },
 190         { "chgrp", "Change group of current inode to GROUP", 2, 2, FL_WR, chgroup },
 191         { "chflags", "Change flags of current inode to FLAGS", 2, 2, FL_WR, chaflags },
 192         { "chgen", "Change generation number of current inode to GEN", 2, 2, FL_WR, chgen },
 193         { "chsize", "Change size of current inode to SIZE", 2, 2, FL_WR, chsize },
 194         { "btime", "Change btime of current inode to BTIME", 2, 2, FL_WR, chbtime },
 195         { "mtime", "Change mtime of current inode to MTIME", 2, 2, FL_WR, chmtime },
 196         { "ctime", "Change ctime of current inode to CTIME", 2, 2, FL_WR, chctime },
 197         { "atime", "Change atime of current inode to ATIME", 2, 2, FL_WR, chatime },
 198         { "quit", "Exit", 1, 1, FL_RO, quit },
 199         { "q", "Exit", 1, 1, FL_RO, quit },
 200         { "exit", "Exit", 1, 1, FL_RO, quit },
 201         { NULL, 0, 0, 0, 0, NULL },
 202 };
 203
 204 int
 205 helpfn(int argc, char *argv[])
 206 {
 207     struct cmdtable *cmdtp;
 208
 209     printf("Commands are:\n%-10s %5s %5s   %s\n",
 210            "command", "min args", "max args", "what");
 211
 212     for (cmdtp = cmds; cmdtp->cmd; cmdtp++)
 213         printf("%-10s %5u %5u   %s\n",
 214                 cmdtp->cmd, cmdtp->minargc-1, cmdtp->maxargc-1, cmdtp->helptxt);
 215     return 0;
 216 }
 217
 218 char *
 219 prompt(EditLine *el)
 220 {
 221     static char pstring[64];
 222     snprintf(pstring, sizeof(pstring), "fsdb (inum: %ju)> ",
 223         (uintmax_t)curinum);
 224     return pstring;
 225 }
 226
 227 static void
 228 setcurinode(ino_t inum)
 229 {
 230
 231         if (curip.i_number != 0)
 232                 irelse(&curip);
 233         ginode(inum, &curip);
 234         curinode = curip.i_dp;
 235         curinum = inum;
 236 }
 237
 238 int
 239 cmdloop(void)
 240 {
 241     char *line;
 242     const char *elline;
 243     int cmd_argc, rval = 0, known;
 244 #define scratch known
 245     char **cmd_argv;
 246     struct cmdtable *cmdp;
 247     History *hist;
 248     EditLine *elptr;
 249     HistEvent he;
 250
 251     setcurinode(UFS_ROOTINO);
 252     printactive(0);
 253
 254     hist = history_init();
 255     history(hist, &he, H_SETSIZE, 100); /* 100 elt history buffer */
 256
 257     elptr = el_init("fsdb", stdin, stdout, stderr);
 258     el_set(elptr, EL_EDITOR, "emacs");
 259     el_set(elptr, EL_PROMPT, prompt);
 260     el_set(elptr, EL_HIST, history, hist);
 261     el_source(elptr, NULL);
 262
 263     while ((elline = el_gets(elptr, &scratch)) != NULL && scratch != 0) {
 264         if (debug)
 265             printf("command `%s'\n", elline);
 266
 267         history(hist, &he, H_ENTER, elline);
 268
 269         line = strdup(elline);
 270         cmd_argv = crack(line, &cmd_argc);
 271         /*
 272          * el_parse returns -1 to signal that it's not been handled
 273          * internally.
 274          */
 275         if (el_parse(elptr, cmd_argc, (const char **)cmd_argv) != -1)
 276             continue;
 277         if (cmd_argc) {
 278             known = 0;
 279             for (cmdp = cmds; cmdp->cmd; cmdp++) {
 280                 if (!strcmp(cmdp->cmd, cmd_argv[0])) {
 281                     if ((cmdp->flags & FL_WR) == FL_WR && nflag)
 282                         warnx("`%s' requires write access", cmd_argv[0]),
 283                             rval = 1;
 284                     else if (cmd_argc >= cmdp->minargc &&
 285                         cmd_argc <= cmdp->maxargc)
 286                         rval = (*cmdp->handler)(cmd_argc, cmd_argv);
 287                     else if (cmd_argc >= cmdp->minargc &&
 288                         (cmdp->flags & FL_ST) == FL_ST) {
 289                         strcpy(line, elline);
 290                         cmd_argv = recrack(line, &cmd_argc, cmdp->maxargc);
 291                         rval = (*cmdp->handler)(cmd_argc, cmd_argv);
 292                     } else
 293                         rval = argcount(cmdp, cmd_argc, cmd_argv);
 294                     known = 1;
 295                     break;
 296                 }
 297             }
 298             if (!known)
 299                 warnx("unknown command `%s'", cmd_argv[0]), rval = 1;
 300         } else
 301             rval = 0;
 302         free(line);
 303         if (rval < 0) {
 304             /* user typed "quit" */
 305             irelse(&curip);
 306             return 0;
 307         }
 308         if (rval)
 309             warnx("rval was %d", rval);
 310     }
 311     el_end(elptr);
 312     history_end(hist);
 313     irelse(&curip);
 314     return rval;
 315 }
 316
 317 #define GETINUM(ac,inum)    inum = strtoul(argv[ac], &cp, 0); \
 318 if (inum < UFS_ROOTINO || inum > maxino || cp == argv[ac] || *cp != '\0' ) { \
 319         printf("inode %ju out of range; range is [%ju,%ju]\n",          \
 320             (uintmax_t)inum, (uintmax_t)UFS_ROOTINO, (uintmax_t)maxino);\
 321         return 1; \
 322 }
 323
 324 /*
 325  * Focus on given inode number
 326  */
 327 CMDFUNCSTART(focus)
 328 {
 329     ino_t inum;
 330     char *cp;
 331
 332     GETINUM(1,inum);
 333     ocurrent = curinum;
 334     setcurinode(inum);
 335     printactive(0);
 336     return 0;
 337 }
 338
 339 CMDFUNCSTART(back)
 340 {
 341     setcurinode(ocurrent);
 342     printactive(0);
 343     return 0;
 344 }
 345
 346 CMDFUNCSTART(zapi)
 347 {
 348     struct inode ip;
 349     ino_t inum;
 350     char *cp;
 351
 352     GETINUM(1,inum);
 353     ginode(inum, &ip);
 354     clearinode(ip.i_dp);
 355     inodirty(&ip);
 356     irelse(&ip);
 357     return 0;
 358 }
 359
 360 CMDFUNCSTART(active)
 361 {
 362     printactive(0);
 363     return 0;
 364 }
 365
 366 CMDFUNCSTART(blocks)
 367 {
 368     printactive(1);
 369     return 0;
 370 }
 371
 372 CMDFUNCSTART(quit)
 373 {
 374     return -1;
 375 }
 376
 377 CMDFUNCSTART(uplink)
 378 {
 379     if (!checkactive())
 380         return 1;
 381     DIP_SET(curinode, di_nlink, DIP(curinode, di_nlink) + 1);
 382     printf("inode %ju link count now %d\n",
 383         (uintmax_t)curinum, DIP(curinode, di_nlink));
 384     inodirty(&curip);
 385     return 0;
 386 }
 387
 388 CMDFUNCSTART(downlink)
 389 {
 390     if (!checkactive())
 391         return 1;
 392     DIP_SET(curinode, di_nlink, DIP(curinode, di_nlink) - 1);
 393     printf("inode %ju link count now %d\n",
 394         (uintmax_t)curinum, DIP(curinode, di_nlink));
 395     inodirty(&curip);
 396     return 0;
 397 }
 398
 399 const char *typename[] = {
 400     "unknown",
 401     "fifo",
 402     "char special",
 403     "unregistered #3",
 404     "directory",
 405     "unregistered #5",
 406     "blk special",
 407     "unregistered #7",
 408     "regular",
 409     "unregistered #9",
 410     "symlink",
 411     "unregistered #11",
 412     "socket",
 413     "unregistered #13",
 414     "whiteout",
 415 };
 416
 417 int diroff;
 418 int slot;
 419
 420 int
 421 scannames(struct inodesc *idesc)
 422 {
 423         struct direct *dirp = idesc->id_dirp;
 424
 425         printf("slot %d off %d ino %d reclen %d: %s, `%.*s'\n",
 426                slot++, diroff, dirp->d_ino, dirp->d_reclen,
 427                typename[dirp->d_type], dirp->d_namlen, dirp->d_name);
 428         diroff += dirp->d_reclen;
 429         return (KEEPON);
 430 }
 431
 432 CMDFUNCSTART(ls)
 433 {
 434     struct inodesc idesc;
 435     checkactivedir();                   /* let it go on anyway */
 436
 437     slot = 0;
 438     diroff = 0;
 439     idesc.id_number = curinum;
 440     idesc.id_func = scannames;
 441     idesc.id_type = DATA;
 442     idesc.id_fix = IGNORE;
 443     ckinode(curinode, &idesc);
 444
 445     return 0;
 446 }
 447
 448 static int findblk_numtofind;
 449 static int wantedblksize;
 450
 451 CMDFUNCSTART(findblk)
 452 {
 453     ino_t inum, inosused;
 454     uint32_t *wantedblk32;
 455     uint64_t *wantedblk64;
 456     struct bufarea *cgbp;
 457     struct cg *cgp;
 458     int c, i, is_ufs2;
 459
 460     wantedblksize = (argc - 1);
 461     is_ufs2 = sblock.fs_magic == FS_UFS2_MAGIC;
 462     ocurrent = curinum;
 463
 464     if (is_ufs2) {
 465         wantedblk64 = calloc(wantedblksize, sizeof(uint64_t));
 466         if (wantedblk64 == NULL)
 467             err(1, "malloc");
 468         for (i = 1; i < argc; i++)
 469             wantedblk64[i - 1] = dbtofsb(&sblock, strtoull(argv[i], NULL, 0));
 470     } else {
 471         wantedblk32 = calloc(wantedblksize, sizeof(uint32_t));
 472         if (wantedblk32 == NULL)
 473             err(1, "malloc");
 474         for (i = 1; i < argc; i++)
 475             wantedblk32[i - 1] = dbtofsb(&sblock, strtoull(argv[i], NULL, 0));
 476     }
 477     findblk_numtofind = wantedblksize;
 478     /*
 479      * sblock.fs_ncg holds a number of cylinder groups.
 480      * Iterate over all cylinder groups.
 481      */
 482     for (c = 0; c < sblock.fs_ncg; c++) {
 483         /*
 484          * sblock.fs_ipg holds a number of inodes per cylinder group.
 485          * Calculate a highest inode number for a given cylinder group.
 486          */
 487         inum = c * sblock.fs_ipg;
 488         /* Read cylinder group. */
 489         cgbp = cglookup(c);
 490         cgp = cgbp->b_un.b_cg;
 491         /*
 492          * Get a highest used inode number for a given cylinder group.
 493          * For UFS1 all inodes initialized at the newfs stage.
 494          */
 495         if (is_ufs2)
 496             inosused = cgp->cg_initediblk;
 497         else
 498             inosused = sblock.fs_ipg;
 499
 500         for (; inosused > 0; inum++, inosused--) {
 501             /* Skip magic inodes: 0, UFS_WINO, UFS_ROOTINO. */
 502             if (inum < UFS_ROOTINO)
 503                 continue;
 504             /*
 505              * Check if the block we are looking for is just an inode block.
 506              *
 507              * ino_to_fsba() - get block containing inode from its number.
 508              * INOPB() - get a number of inodes in one disk block.
 509              */
 510             if (is_ufs2 ?
 511                 compare_blk64(wantedblk64, ino_to_fsba(&sblock, inum)) :
 512                 compare_blk32(wantedblk32, ino_to_fsba(&sblock, inum))) {
 513                 printf("block %llu: inode block (%ju-%ju)\n",
 514                     (unsigned long long)fsbtodb(&sblock,
 515                         ino_to_fsba(&sblock, inum)),
 516                     (uintmax_t)(inum / INOPB(&sblock)) * INOPB(&sblock),
 517                     (uintmax_t)(inum / INOPB(&sblock) + 1) * INOPB(&sblock));
 518                 findblk_numtofind--;
 519                 if (findblk_numtofind == 0)
 520                     goto end;
 521             }
 522             /* Get on-disk inode aka dinode. */
 523             setcurinode(inum);
 524             /* Find IFLNK dinode with allocated data blocks. */
 525             switch (DIP(curinode, di_mode) & IFMT) {
 526             case IFDIR:
 527             case IFREG:
 528                 if (DIP(curinode, di_blocks) == 0)
 529                     continue;
 530                 break;
 531             case IFLNK:
 532                 {
 533                     uint64_t size = DIP(curinode, di_size);
 534                     if (size > 0 && size < sblock.fs_maxsymlinklen &&
 535                         DIP(curinode, di_blocks) == 0)
 536                         continue;
 537                     else
 538                         break;
 539                 }
 540             default:
 541                 continue;
 542             }
 543             /* Look through direct data blocks. */
 544             if (is_ufs2 ?
 545                 find_blks64(curinode->dp2.di_db, UFS_NDADDR, wantedblk64) :
 546                 find_blks32(curinode->dp1.di_db, UFS_NDADDR, wantedblk32))
 547                 goto end;
 548             for (i = 0; i < UFS_NIADDR; i++) {
 549                 /*
 550                  * Does the block we are looking for belongs to the
 551                  * indirect blocks?
 552                  */
 553                 if (is_ufs2 ?
 554                     compare_blk64(wantedblk64, curinode->dp2.di_ib[i]) :
 555                     compare_blk32(wantedblk32, curinode->dp1.di_ib[i]))
 556                     if (founddatablk(is_ufs2 ? curinode->dp2.di_ib[i] :
 557                         curinode->dp1.di_ib[i]))
 558                         goto end;
 559                 /*
 560                  * Search through indirect, double and triple indirect
 561                  * data blocks.
 562                  */
 563                 if (is_ufs2 ? (curinode->dp2.di_ib[i] != 0) :
 564                     (curinode->dp1.di_ib[i] != 0))
 565                     if (is_ufs2 ?
 566                         find_indirblks64(curinode->dp2.di_ib[i], i,
 567                             wantedblk64) :
 568                         find_indirblks32(curinode->dp1.di_ib[i], i,
 569                             wantedblk32))
 570                         goto end;
 571             }
 572         }
 573     }
 574 end:
 575     setcurinode(ocurrent);
 576     if (is_ufs2)
 577         free(wantedblk64);
 578     else
 579         free(wantedblk32);
 580     return 0;
 581 }
 582
 583 static int
 584 compare_blk32(uint32_t *wantedblk, uint32_t curblk)
 585 {
 586     int i;
 587
 588     for (i = 0; i < wantedblksize; i++) {
 589         if (wantedblk[i] != 0 && wantedblk[i] == curblk) {
 590             wantedblk[i] = 0;
 591             return 1;
 592         }
 593     }
 594     return 0;
 595 }
 596
 597 static int
 598 compare_blk64(uint64_t *wantedblk, uint64_t curblk)
 599 {
 600     int i;
 601
 602     for (i = 0; i < wantedblksize; i++) {
 603         if (wantedblk[i] != 0 && wantedblk[i] == curblk) {
 604             wantedblk[i] = 0;
 605             return 1;
 606         }
 607     }
 608     return 0;
 609 }
 610
 611 static int
 612 founddatablk(uint64_t blk)
 613 {
 614
 615     printf("%llu: data block of inode %ju\n",
 616         (unsigned long long)fsbtodb(&sblock, blk), (uintmax_t)curinum);
 617     findblk_numtofind--;
 618     if (findblk_numtofind == 0)
 619         return 1;
 620     return 0;
 621 }
 622
 623 static int
 624 find_blks32(uint32_t *buf, int size, uint32_t *wantedblk)
 625 {
 626     int blk;
 627     for (blk = 0; blk < size; blk++) {
 628         if (buf[blk] == 0)
 629             continue;
 630         if (compare_blk32(wantedblk, buf[blk])) {
 631             if (founddatablk(buf[blk]))
 632                 return 1;
 633         }
 634     }
 635     return 0;
 636 }
 637
 638 static int
 639 find_indirblks32(uint32_t blk, int ind_level, uint32_t *wantedblk)
 640 {
 641 #define MAXNINDIR      (MAXBSIZE / sizeof(uint32_t))
 642     uint32_t idblk[MAXNINDIR];
 643     int i;
 644
 645     blread(fsreadfd, (char *)idblk, fsbtodb(&sblock, blk), (int)sblock.fs_bsize);
 646     if (ind_level <= 0) {
 647         if (find_blks32(idblk, sblock.fs_bsize / sizeof(uint32_t), wantedblk))
 648             return 1;
 649     } else {
 650         ind_level--;
 651         for (i = 0; i < sblock.fs_bsize / sizeof(uint32_t); i++) {
 652             if (compare_blk32(wantedblk, idblk[i])) {
 653                 if (founddatablk(idblk[i]))
 654                     return 1;
 655             }
 656             if (idblk[i] != 0)
 657                 if (find_indirblks32(idblk[i], ind_level, wantedblk))
 658                     return 1;
 659         }
 660     }
 661 #undef MAXNINDIR
 662     return 0;
 663 }
 664
 665 static int
 666 find_blks64(uint64_t *buf, int size, uint64_t *wantedblk)
 667 {
 668     int blk;
 669     for (blk = 0; blk < size; blk++) {
 670         if (buf[blk] == 0)
 671             continue;
 672         if (compare_blk64(wantedblk, buf[blk])) {
 673             if (founddatablk(buf[blk]))
 674                 return 1;
 675         }
 676     }
 677     return 0;
 678 }
 679
 680 static int
 681 find_indirblks64(uint64_t blk, int ind_level, uint64_t *wantedblk)
 682 {
 683 #define MAXNINDIR      (MAXBSIZE / sizeof(uint64_t))
 684     uint64_t idblk[MAXNINDIR];
 685     int i;
 686
 687     blread(fsreadfd, (char *)idblk, fsbtodb(&sblock, blk), (int)sblock.fs_bsize);
 688     if (ind_level <= 0) {
 689         if (find_blks64(idblk, sblock.fs_bsize / sizeof(uint64_t), wantedblk))
 690             return 1;
 691     } else {
 692         ind_level--;
 693         for (i = 0; i < sblock.fs_bsize / sizeof(uint64_t); i++) {
 694             if (compare_blk64(wantedblk, idblk[i])) {
 695                 if (founddatablk(idblk[i]))
 696                     return 1;
 697             }
 698             if (idblk[i] != 0)
 699                 if (find_indirblks64(idblk[i], ind_level, wantedblk))
 700                     return 1;
 701         }
 702     }
 703 #undef MAXNINDIR
 704     return 0;
 705 }
 706
 707 int findino(struct inodesc *idesc); /* from fsck */
 708 static int dolookup(char *name);
 709
 710 static int
 711 dolookup(char *name)
 712 {
 713     struct inodesc idesc;
 714
 715     if (!checkactivedir())
 716             return 0;
 717     idesc.id_number = curinum;
 718     idesc.id_func = findino;
 719     idesc.id_name = name;
 720     idesc.id_type = DATA;
 721     idesc.id_fix = IGNORE;
 722     if (ckinode(curinode, &idesc) & FOUND) {
 723         setcurinode(idesc.id_parent);
 724         printactive(0);
 725         return 1;
 726     } else {
 727         warnx("name `%s' not found in current inode directory", name);
 728         return 0;
 729     }
 730 }
 731
 732 CMDFUNCSTART(focusname)
 733 {
 734     char *p, *val;
 735
 736     if (!checkactive())
 737         return 1;
 738
 739     ocurrent = curinum;
 740
 741     if (argv[1][0] == '/') {
 742         setcurinode(UFS_ROOTINO);
 743     } else {
 744         if (!checkactivedir())
 745             return 1;
 746     }
 747     for (p = argv[1]; p != NULL;) {
 748         while ((val = strsep(&p, "/")) != NULL && *val == '\0');
 749         if (val) {
 750             printf("component `%s': ", val);
 751             fflush(stdout);
 752             if (!dolookup(val)) {
 753                 return(1);
 754             }
 755         }
 756     }
 757     return 0;
 758 }
 759
 760 CMDFUNCSTART(ln)
 761 {
 762     ino_t inum;
 763     int rval;
 764     char *cp;
 765
 766     GETINUM(1,inum);
 767
 768     if (!checkactivedir())
 769         return 1;
 770     rval = makeentry(curinum, inum, argv[2]);
 771     if (rval)
 772             printf("Ino %ju entered as `%s'\n", (uintmax_t)inum, argv[2]);
 773     else
 774         printf("could not enter name? weird.\n");
 775     return rval;
 776 }
 777
 778 CMDFUNCSTART(rm)
 779 {
 780     int rval;
 781
 782     if (!checkactivedir())
 783         return 1;
 784     rval = changeino(curinum, argv[1], 0);
 785     if (rval & ALTERED) {
 786         printf("Name `%s' removed\n", argv[1]);
 787         return 0;
 788     } else {
 789         printf("could not remove name ('%s')? weird.\n", argv[1]);
 790         return 1;
 791     }
 792 }
 793
 794 long slotcount, desired;
 795
 796 int
 797 chinumfunc(struct inodesc *idesc)
 798 {
 799         struct direct *dirp = idesc->id_dirp;
 800
 801         if (slotcount++ == desired) {
 802             dirp->d_ino = idesc->id_parent;
 803             return STOP|ALTERED|FOUND;
 804         }
 805         return KEEPON;
 806 }
 807
 808 CMDFUNCSTART(chinum)
 809 {
 810     char *cp;
 811     ino_t inum;
 812     struct inodesc idesc;
 813
 814     slotcount = 0;
 815     if (!checkactivedir())
 816         return 1;
 817     GETINUM(2,inum);
 818
 819     desired = strtol(argv[1], &cp, 0);
 820     if (cp == argv[1] || *cp != '\0' || desired < 0) {
 821         printf("invalid slot number `%s'\n", argv[1]);
 822         return 1;
 823     }
 824
 825     idesc.id_number = curinum;
 826     idesc.id_func = chinumfunc;
 827     idesc.id_fix = IGNORE;
 828     idesc.id_type = DATA;
 829     idesc.id_parent = inum;             /* XXX convenient hiding place */
 830
 831     if (ckinode(curinode, &idesc) & FOUND)
 832         return 0;
 833     else {
 834         warnx("no %sth slot in current directory", argv[1]);
 835         return 1;
 836     }
 837 }
 838
 839 int
 840 chnamefunc(struct inodesc *idesc)
 841 {
 842         struct direct *dirp = idesc->id_dirp;
 843         struct direct testdir;
 844
 845         if (slotcount++ == desired) {
 846             /* will name fit? */
 847             testdir.d_namlen = strlen(idesc->id_name);
 848             if (DIRSIZ(NEWDIRFMT, &testdir) <= dirp->d_reclen) {
 849                 dirp->d_namlen = testdir.d_namlen;
 850                 strcpy(dirp->d_name, idesc->id_name);
 851                 return STOP|ALTERED|FOUND;
 852             } else
 853                 return STOP|FOUND;      /* won't fit, so give up */
 854         }
 855         return KEEPON;
 856 }
 857
 858 CMDFUNCSTART(chname)
 859 {
 860     int rval;
 861     char *cp;
 862     struct inodesc idesc;
 863
 864     slotcount = 0;
 865     if (!checkactivedir())
 866         return 1;
 867
 868     desired = strtoul(argv[1], &cp, 0);
 869     if (cp == argv[1] || *cp != '\0') {
 870         printf("invalid slot number `%s'\n", argv[1]);
 871         return 1;
 872     }
 873
 874     idesc.id_number = curinum;
 875     idesc.id_func = chnamefunc;
 876     idesc.id_fix = IGNORE;
 877     idesc.id_type = DATA;
 878     idesc.id_name = argv[2];
 879
 880     rval = ckinode(curinode, &idesc);
 881     if ((rval & (FOUND|ALTERED)) == (FOUND|ALTERED))
 882         return 0;
 883     else if (rval & FOUND) {
 884         warnx("new name `%s' does not fit in slot %s\n", argv[2], argv[1]);
 885         return 1;
 886     } else {
 887         warnx("no %sth slot in current directory", argv[1]);
 888         return 1;
 889     }
 890 }
 891
 892 struct typemap {
 893     const char *typename;
 894     int typebits;
 895 } typenamemap[]  = {
 896     {"file", IFREG},
 897     {"dir", IFDIR},
 898     {"socket", IFSOCK},
 899     {"fifo", IFIFO},
 900 };
 901
 902 CMDFUNCSTART(newtype)
 903 {
 904     int type;
 905     struct typemap *tp;
 906
 907     if (!checkactive())
 908         return 1;
 909     type = DIP(curinode, di_mode) & IFMT;
 910     for (tp = typenamemap;
 911          tp < &typenamemap[nitems(typenamemap)];
 912          tp++) {
 913         if (!strcmp(argv[1], tp->typename)) {
 914             printf("setting type to %s\n", tp->typename);
 915             type = tp->typebits;
 916             break;
 917         }
 918     }
 919     if (tp == &typenamemap[nitems(typenamemap)]) {
 920         warnx("type `%s' not known", argv[1]);
 921         warnx("try one of `file', `dir', `socket', `fifo'");
 922         return 1;
 923     }
 924     DIP_SET(curinode, di_mode, DIP(curinode, di_mode) & ~IFMT);
 925     DIP_SET(curinode, di_mode, DIP(curinode, di_mode) | type);
 926     inodirty(&curip);
 927     printactive(0);
 928     return 0;
 929 }
 930
 931 CMDFUNCSTART(chlen)
 932 {
 933     int rval = 1;
 934     long len;
 935     char *cp;
 936
 937     if (!checkactive())
 938         return 1;
 939
 940     len = strtol(argv[1], &cp, 0);
 941     if (cp == argv[1] || *cp != '\0' || len < 0) {
 942         warnx("bad length `%s'", argv[1]);
 943         return 1;
 944     }
 945
 946     DIP_SET(curinode, di_size, len);
 947     inodirty(&curip);
 948     printactive(0);
 949     return rval;
 950 }
 951
 952 CMDFUNCSTART(chmode)
 953 {
 954     int rval = 1;
 955     long modebits;
 956     char *cp;
 957
 958     if (!checkactive())
 959         return 1;
 960
 961     modebits = strtol(argv[1], &cp, 8);
 962     if (cp == argv[1] || *cp != '\0' || (modebits & ~07777)) {
 963         warnx("bad modebits `%s'", argv[1]);
 964         return 1;
 965     }
 966
 967     DIP_SET(curinode, di_mode, DIP(curinode, di_mode) & ~07777);
 968     DIP_SET(curinode, di_mode, DIP(curinode, di_mode) | modebits);
 969     inodirty(&curip);
 970     printactive(0);
 971     return rval;
 972 }
 973
 974 CMDFUNCSTART(chaflags)
 975 {
 976     int rval = 1;
 977     u_long flags;
 978     char *cp;
 979
 980     if (!checkactive())
 981         return 1;
 982
 983     flags = strtoul(argv[1], &cp, 0);
 984     if (cp == argv[1] || *cp != '\0' ) {
 985         warnx("bad flags `%s'", argv[1]);
 986         return 1;
 987     }
 988
 989     if (flags > UINT_MAX) {
 990         warnx("flags set beyond 32-bit range of field (%lx)\n", flags);
 991         return(1);
 992     }
 993     DIP_SET(curinode, di_flags, flags);
 994     inodirty(&curip);
 995     printactive(0);
 996     return rval;
 997 }
 998
 999 CMDFUNCSTART(chgen)
1000 {
1001     int rval = 1;
1002     long gen;
1003     char *cp;
1004
1005     if (!checkactive())
1006         return 1;
1007
1008     gen = strtol(argv[1], &cp, 0);
1009     if (cp == argv[1] || *cp != '\0' ) {
1010         warnx("bad gen `%s'", argv[1]);
1011         return 1;
1012     }
1013
1014     if (gen > INT_MAX || gen < INT_MIN) {
1015         warnx("gen set beyond 32-bit range of field (%lx)\n", gen);
1016         return(1);
1017     }
1018     DIP_SET(curinode, di_gen, gen);
1019     inodirty(&curip);
1020     printactive(0);
1021     return rval;
1022 }
1023
1024 CMDFUNCSTART(chsize)
1025 {
1026     int rval = 1;
1027     off_t size;
1028     char *cp;
1029
1030     if (!checkactive())
1031         return 1;
1032
1033     size = strtoll(argv[1], &cp, 0);
1034     if (cp == argv[1] || *cp != '\0') {
1035         warnx("bad size `%s'", argv[1]);
1036         return 1;
1037     }
1038
1039     if (size < 0) {
1040         warnx("size set to negative (%jd)\n", (intmax_t)size);
1041         return(1);
1042     }
1043     DIP_SET(curinode, di_size, size);
1044     inodirty(&curip);
1045     printactive(0);
1046     return rval;
1047 }
1048
1049 CMDFUNCSTART(linkcount)
1050 {
1051     int rval = 1;
1052     int lcnt;
1053     char *cp;
1054
1055     if (!checkactive())
1056         return 1;
1057
1058     lcnt = strtol(argv[1], &cp, 0);
1059     if (cp == argv[1] || *cp != '\0' ) {
1060         warnx("bad link count `%s'", argv[1]);
1061         return 1;
1062     }
1063     if (lcnt > USHRT_MAX || lcnt < 0) {
1064         warnx("max link count is %d\n", USHRT_MAX);
1065         return 1;
1066     }
1067
1068     DIP_SET(curinode, di_nlink, lcnt);
1069     inodirty(&curip);
1070     printactive(0);
1071     return rval;
1072 }
1073
1074 CMDFUNCSTART(chowner)
1075 {
1076     int rval = 1;
1077     unsigned long uid;
1078     char *cp;
1079     struct passwd *pwd;
1080
1081     if (!checkactive())
1082         return 1;
1083
1084     uid = strtoul(argv[1], &cp, 0);
1085     if (cp == argv[1] || *cp != '\0' ) {
1086         /* try looking up name */
1087         if ((pwd = getpwnam(argv[1]))) {
1088             uid = pwd->pw_uid;
1089         } else {
1090             warnx("bad uid `%s'", argv[1]);
1091             return 1;
1092         }
1093     }
1094
1095     DIP_SET(curinode, di_uid, uid);
1096     inodirty(&curip);
1097     printactive(0);
1098     return rval;
1099 }
1100
1101 CMDFUNCSTART(chgroup)
1102 {
1103     int rval = 1;
1104     unsigned long gid;
1105     char *cp;
1106     struct group *grp;
1107
1108     if (!checkactive())
1109         return 1;
1110
1111     gid = strtoul(argv[1], &cp, 0);
1112     if (cp == argv[1] || *cp != '\0' ) {
1113         if ((grp = getgrnam(argv[1]))) {
1114             gid = grp->gr_gid;
1115         } else {
1116             warnx("bad gid `%s'", argv[1]);
1117             return 1;
1118         }
1119     }
1120
1121     DIP_SET(curinode, di_gid, gid);
1122     inodirty(&curip);
1123     printactive(0);
1124     return rval;
1125 }
1126
1127 int
1128 dotime(char *name, time_t *secp, int32_t *nsecp)
1129 {
1130     char *p, *val;
1131     struct tm t;
1132     int32_t nsec;
1133     p = strchr(name, '.');
1134     if (p) {
1135         *p = '\0';
1136         nsec = strtoul(++p, &val, 0);
1137         if (val == p || *val != '\0' || nsec >= 1000000000 || nsec < 0) {
1138                 warnx("invalid nanoseconds");
1139                 goto badformat;
1140         }
1141     } else
1142         nsec = 0;
1143     if (strlen(name) != 14) {
1144 badformat:
1145         warnx("date format: YYYYMMDDHHMMSS[.nsec]");
1146         return 1;
1147     }
1148     *nsecp = nsec;
1149
1150     for (p = name; *p; p++)
1151         if (*p < '0' || *p > '9')
1152             goto badformat;
1153
1154     p = name;
1155 #define VAL() ((*p++) - '0')
1156     t.tm_year = VAL();
1157     t.tm_year = VAL() + t.tm_year * 10;
1158     t.tm_year = VAL() + t.tm_year * 10;
1159     t.tm_year = VAL() + t.tm_year * 10 - 1900;
1160     t.tm_mon = VAL();
1161     t.tm_mon = VAL() + t.tm_mon * 10 - 1;
1162     t.tm_mday = VAL();
1163     t.tm_mday = VAL() + t.tm_mday * 10;
1164     t.tm_hour = VAL();
1165     t.tm_hour = VAL() + t.tm_hour * 10;
1166     t.tm_min = VAL();
1167     t.tm_min = VAL() + t.tm_min * 10;
1168     t.tm_sec = VAL();
1169     t.tm_sec = VAL() + t.tm_sec * 10;
1170     t.tm_isdst = -1;
1171
1172     *secp = mktime(&t);
1173     if (*secp == -1) {
1174         warnx("date/time out of range");
1175         return 1;
1176     }
1177     return 0;
1178 }
1179
1180 CMDFUNCSTART(chbtime)
1181 {
1182     time_t secs;
1183     int32_t nsecs;
1184
1185     if (dotime(argv[1], &secs, &nsecs))
1186         return 1;
1187     if (sblock.fs_magic == FS_UFS1_MAGIC)
1188         return 1;
1189     curinode->dp2.di_birthtime = _time_to_time64(secs);
1190     curinode->dp2.di_birthnsec = nsecs;
1191     inodirty(&curip);
1192     printactive(0);
1193     return 0;
1194 }
1195
1196 CMDFUNCSTART(chmtime)
1197 {
1198     time_t secs;
1199     int32_t nsecs;
1200
1201     if (dotime(argv[1], &secs, &nsecs))
1202         return 1;
1203     if (sblock.fs_magic == FS_UFS1_MAGIC)
1204         curinode->dp1.di_mtime = _time_to_time32(secs);
1205     else
1206         curinode->dp2.di_mtime = _time_to_time64(secs);
1207     DIP_SET(curinode, di_mtimensec, nsecs);
1208     inodirty(&curip);
1209     printactive(0);
1210     return 0;
1211 }
1212
1213 CMDFUNCSTART(chatime)
1214 {
1215     time_t secs;
1216     int32_t nsecs;
1217
1218     if (dotime(argv[1], &secs, &nsecs))
1219         return 1;
1220     if (sblock.fs_magic == FS_UFS1_MAGIC)
1221         curinode->dp1.di_atime = _time_to_time32(secs);
1222     else
1223         curinode->dp2.di_atime = _time_to_time64(secs);
1224     DIP_SET(curinode, di_atimensec, nsecs);
1225     inodirty(&curip);
1226     printactive(0);
1227     return 0;
1228 }
1229
1230 CMDFUNCSTART(chctime)
1231 {
1232     time_t secs;
1233     int32_t nsecs;
1234
1235     if (dotime(argv[1], &secs, &nsecs))
1236         return 1;
1237     if (sblock.fs_magic == FS_UFS1_MAGIC)
1238         curinode->dp1.di_ctime = _time_to_time32(secs);
1239     else
1240         curinode->dp2.di_ctime = _time_to_time64(secs);
1241     DIP_SET(curinode, di_ctimensec, nsecs);
1242     inodirty(&curip);
1243     printactive(0);
1244     return 0;
1245 }