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