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