xref: /freebsd/sbin/fsdb/fsdb.c (revision ab40f58ccfe6c07ebefddc72f4661a52fe746353)
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();
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();
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();
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();
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();
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();
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();
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();
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();
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();
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();
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();
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();
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();
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();
1214     printactive(0);
1215     return 0;
1216 }
1217