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