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