xref: /freebsd/sbin/restore/dirs.c (revision 2be1a816b9ff69588e55be0a84cbe2a31efc0f2f)
1 /*
2  * Copyright (c) 1983, 1993
3  *	The Regents of the University of California.  All rights reserved.
4  * (c) UNIX System Laboratories, Inc.
5  * All or some portions of this file are derived from material licensed
6  * to the University of California by American Telephone and Telegraph
7  * Co. or Unix System Laboratories, Inc. and are reproduced herein with
8  * the permission of UNIX System Laboratories, Inc.
9  *
10  * Redistribution and use in source and binary forms, with or without
11  * modification, are permitted provided that the following conditions
12  * are met:
13  * 1. Redistributions of source code must retain the above copyright
14  *    notice, this list of conditions and the following disclaimer.
15  * 2. Redistributions in binary form must reproduce the above copyright
16  *    notice, this list of conditions and the following disclaimer in the
17  *    documentation and/or other materials provided with the distribution.
18  * 4. Neither the name of the University nor the names of its contributors
19  *    may be used to endorse or promote products derived from this software
20  *    without specific prior written permission.
21  *
22  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
23  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
24  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
25  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
26  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
27  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
28  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
29  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
30  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
31  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
32  * SUCH DAMAGE.
33  */
34 
35 #ifndef lint
36 #if 0
37 static char sccsid[] = "@(#)dirs.c	8.7 (Berkeley) 5/1/95";
38 #endif
39 static const char rcsid[] =
40   "$FreeBSD$";
41 #endif /* not lint */
42 
43 #include <sys/param.h>
44 #include <sys/file.h>
45 #include <sys/stat.h>
46 #include <sys/time.h>
47 
48 #include <ufs/ufs/dinode.h>
49 #include <ufs/ufs/dir.h>
50 #include <protocols/dumprestore.h>
51 
52 #include <err.h>
53 #include <errno.h>
54 #include <limits.h>
55 #include <paths.h>
56 #include <stdio.h>
57 #include <stdlib.h>
58 #include <string.h>
59 #include <unistd.h>
60 
61 #include "restore.h"
62 #include "extern.h"
63 
64 /*
65  * Symbol table of directories read from tape.
66  */
67 #define HASHSIZE	1000
68 #define INOHASH(val) (val % HASHSIZE)
69 struct inotab {
70 	struct	inotab *t_next;
71 	ino_t	t_ino;
72 	int32_t	t_seekpt;
73 	int32_t	t_size;
74 };
75 static struct inotab *inotab[HASHSIZE];
76 
77 /*
78  * Information retained about directories.
79  */
80 struct modeinfo {
81 	ino_t ino;
82 	struct timeval ctimep[2];
83 	struct timeval mtimep[2];
84 	mode_t mode;
85 	uid_t uid;
86 	gid_t gid;
87 	int flags;
88 	int extsize;
89 };
90 
91 /*
92  * Definitions for library routines operating on directories.
93  */
94 #undef DIRBLKSIZ
95 #define DIRBLKSIZ 1024
96 struct rstdirdesc {
97 	int	dd_fd;
98 	int32_t	dd_loc;
99 	int32_t	dd_size;
100 	char	dd_buf[DIRBLKSIZ];
101 };
102 
103 /*
104  * Global variables for this file.
105  */
106 static long	seekpt;
107 static FILE	*df, *mf;
108 static RST_DIR	*dirp;
109 static char	dirfile[MAXPATHLEN] = "#";	/* No file */
110 static char	modefile[MAXPATHLEN] = "#";	/* No file */
111 static char	dot[2] = ".";			/* So it can be modified */
112 
113 static struct inotab	*allocinotab(struct context *, long);
114 static void		 flushent(void);
115 static struct inotab	*inotablookup(ino_t);
116 static RST_DIR		*opendirfile(const char *);
117 static void		 putdir(char *, long);
118 static void		 putdirattrs(char *, long);
119 static void		 putent(struct direct *);
120 static void		 rst_seekdir(RST_DIR *, long, long);
121 static long		 rst_telldir(RST_DIR *);
122 static struct direct	*searchdir(ino_t, char *);
123 static void		 fail_dirtmp(char *);
124 
125 /*
126  *	Extract directory contents, building up a directory structure
127  *	on disk for extraction by name.
128  *	If genmode is requested, save mode, owner, and times for all
129  *	directories on the tape.
130  */
131 void
132 extractdirs(int genmode)
133 {
134 	struct inotab *itp;
135 	struct direct nulldir;
136 	int i, fd;
137 	const char *tmpdir;
138 
139 	vprintf(stdout, "Extract directories from tape\n");
140 	if ((tmpdir = getenv("TMPDIR")) == NULL || tmpdir[0] == '\0')
141 		tmpdir = _PATH_TMP;
142 	(void) sprintf(dirfile, "%s/rstdir%d", tmpdir, dumpdate);
143 	if (command != 'r' && command != 'R') {
144 		(void *) strcat(dirfile, "-XXXXXX");
145 		fd = mkstemp(dirfile);
146 	} else
147 		fd = open(dirfile, O_RDWR|O_CREAT|O_EXCL, 0666);
148 	if (fd == -1 || (df = fdopen(fd, "w")) == NULL) {
149 		if (fd != -1)
150 			close(fd);
151 		warn("%s: cannot create directory database", dirfile);
152 		done(1);
153 	}
154 	if (genmode != 0) {
155 		(void) sprintf(modefile, "%s/rstmode%d", tmpdir, dumpdate);
156 		if (command != 'r' && command != 'R') {
157 			(void *) strcat(modefile, "-XXXXXX");
158 			fd = mkstemp(modefile);
159 		} else
160 			fd = open(modefile, O_RDWR|O_CREAT|O_EXCL, 0666);
161 		if (fd == -1 || (mf = fdopen(fd, "w")) == NULL) {
162 			if (fd != -1)
163 				close(fd);
164 			warn("%s: cannot create modefile", modefile);
165 			done(1);
166 		}
167 	}
168 	nulldir.d_ino = 0;
169 	nulldir.d_type = DT_DIR;
170 	nulldir.d_namlen = 1;
171 	(void) strcpy(nulldir.d_name, "/");
172 	nulldir.d_reclen = DIRSIZ(0, &nulldir);
173 	for (;;) {
174 		curfile.name = "<directory file - name unknown>";
175 		curfile.action = USING;
176 		if (curfile.mode == 0 || (curfile.mode & IFMT) != IFDIR)
177 			break;
178 		itp = allocinotab(&curfile, seekpt);
179 		getfile(putdir, putdirattrs, xtrnull);
180 		putent(&nulldir);
181 		flushent();
182 		itp->t_size = seekpt - itp->t_seekpt;
183 	}
184 	if (fclose(df) != 0)
185 		fail_dirtmp(dirfile);
186 	dirp = opendirfile(dirfile);
187 	if (dirp == NULL)
188 		fprintf(stderr, "opendirfile: %s\n", strerror(errno));
189 	if (mf != NULL && fclose(mf) != 0)
190 		fail_dirtmp(modefile);
191 	i = dirlookup(dot);
192 	if (i == 0)
193 		panic("Root directory is not on tape\n");
194 }
195 
196 /*
197  * skip over all the directories on the tape
198  */
199 void
200 skipdirs(void)
201 {
202 
203 	while (curfile.ino && (curfile.mode & IFMT) == IFDIR) {
204 		skipfile();
205 	}
206 }
207 
208 /*
209  *	Recursively find names and inumbers of all files in subtree
210  *	pname and pass them off to be processed.
211  */
212 void
213 treescan(char *pname, ino_t ino, long (*todo)(char *, ino_t, int))
214 {
215 	struct inotab *itp;
216 	struct direct *dp;
217 	int namelen;
218 	long bpt;
219 	char locname[MAXPATHLEN];
220 
221 	itp = inotablookup(ino);
222 	if (itp == NULL) {
223 		/*
224 		 * Pname is name of a simple file or an unchanged directory.
225 		 */
226 		(void) (*todo)(pname, ino, LEAF);
227 		return;
228 	}
229 	/*
230 	 * Pname is a dumped directory name.
231 	 */
232 	if ((*todo)(pname, ino, NODE) == FAIL)
233 		return;
234 	/*
235 	 * begin search through the directory
236 	 * skipping over "." and ".."
237 	 */
238 	(void) strlcpy(locname, pname, sizeof(locname));
239 	(void) strlcat(locname, "/", sizeof(locname));
240 	namelen = strlen(locname);
241 	rst_seekdir(dirp, itp->t_seekpt, itp->t_seekpt);
242 	dp = rst_readdir(dirp); /* "." */
243 	if (dp != NULL && strcmp(dp->d_name, ".") == 0)
244 		dp = rst_readdir(dirp); /* ".." */
245 	else
246 		fprintf(stderr, "Warning: `.' missing from directory %s\n",
247 			pname);
248 	if (dp != NULL && strcmp(dp->d_name, "..") == 0)
249 		dp = rst_readdir(dirp); /* first real entry */
250 	else
251 		fprintf(stderr, "Warning: `..' missing from directory %s\n",
252 			pname);
253 	bpt = rst_telldir(dirp);
254 	/*
255 	 * a zero inode signals end of directory
256 	 */
257 	while (dp != NULL) {
258 		locname[namelen] = '\0';
259 		if (namelen + dp->d_namlen >= sizeof(locname)) {
260 			fprintf(stderr, "%s%s: name exceeds %d char\n",
261 				locname, dp->d_name, sizeof(locname) - 1);
262 		} else {
263 			(void)strlcat(locname, dp->d_name, sizeof(locname));
264 			treescan(locname, dp->d_ino, todo);
265 			rst_seekdir(dirp, bpt, itp->t_seekpt);
266 		}
267 		dp = rst_readdir(dirp);
268 		bpt = rst_telldir(dirp);
269 	}
270 }
271 
272 /*
273  * Lookup a pathname which is always assumed to start from the ROOTINO.
274  */
275 struct direct *
276 pathsearch(const char *pathname)
277 {
278 	ino_t ino;
279 	struct direct *dp;
280 	char *path, *name, buffer[MAXPATHLEN];
281 
282 	strcpy(buffer, pathname);
283 	path = buffer;
284 	ino = ROOTINO;
285 	while (*path == '/')
286 		path++;
287 	dp = NULL;
288 	while ((name = strsep(&path, "/")) != NULL && *name != '\0') {
289 		if ((dp = searchdir(ino, name)) == NULL)
290 			return (NULL);
291 		ino = dp->d_ino;
292 	}
293 	return (dp);
294 }
295 
296 /*
297  * Lookup the requested name in directory inum.
298  * Return its inode number if found, zero if it does not exist.
299  */
300 static struct direct *
301 searchdir(ino_t	inum, char *name)
302 {
303 	struct direct *dp;
304 	struct inotab *itp;
305 	int len;
306 
307 	itp = inotablookup(inum);
308 	if (itp == NULL)
309 		return (NULL);
310 	rst_seekdir(dirp, itp->t_seekpt, itp->t_seekpt);
311 	len = strlen(name);
312 	do {
313 		dp = rst_readdir(dirp);
314 		if (dp == NULL)
315 			return (NULL);
316 	} while (dp->d_namlen != len || strncmp(dp->d_name, name, len) != 0);
317 	return (dp);
318 }
319 
320 /*
321  * Put the directory entries in the directory file
322  */
323 static void
324 putdir(char *buf, long size)
325 {
326 	struct direct *dp;
327 	long loc, i;
328 
329 	for (loc = 0; loc < size; ) {
330 		dp = (struct direct *)(buf + loc);
331 		if (Bcvt)
332 			swabst((u_char *)"ls", (u_char *) dp);
333 		if (oldinofmt && dp->d_ino != 0) {
334 #if BYTE_ORDER == BIG_ENDIAN
335 			if (Bcvt)
336 				dp->d_namlen = dp->d_type;
337 #else
338 			if (!Bcvt && dp->d_namlen == 0)
339 				dp->d_namlen = dp->d_type;
340 #endif
341 			dp->d_type = DT_UNKNOWN;
342 		}
343 		i = DIRBLKSIZ - (loc & (DIRBLKSIZ - 1));
344 		if ((dp->d_reclen & 0x3) != 0 ||
345 		    dp->d_reclen > i ||
346 		    dp->d_reclen < DIRSIZ(0, dp)
347 #if NAME_MAX < 255
348 		    || dp->d_namlen > NAME_MAX
349 #endif
350 		    ) {
351 			vprintf(stdout, "Mangled directory: ");
352 			if ((dp->d_reclen & 0x3) != 0)
353 				vprintf(stdout,
354 				   "reclen not multiple of 4 ");
355 			if (dp->d_reclen < DIRSIZ(0, dp))
356 				vprintf(stdout,
357 				   "reclen less than DIRSIZ (%d < %d) ",
358 				   dp->d_reclen, DIRSIZ(0, dp));
359 #if NAME_MAX < 255
360 			if (dp->d_namlen > NAME_MAX)
361 				vprintf(stdout,
362 				   "reclen name too big (%d > %d) ",
363 				   dp->d_namlen, NAME_MAX);
364 #endif
365 			vprintf(stdout, "\n");
366 			loc += i;
367 			continue;
368 		}
369 		loc += dp->d_reclen;
370 		if (dp->d_ino != 0) {
371 			putent(dp);
372 		}
373 	}
374 }
375 
376 /*
377  * These variables are "local" to the following two functions.
378  */
379 char dirbuf[DIRBLKSIZ];
380 long dirloc = 0;
381 long prev = 0;
382 
383 /*
384  * add a new directory entry to a file.
385  */
386 static void
387 putent(struct direct *dp)
388 {
389 	dp->d_reclen = DIRSIZ(0, dp);
390 	if (dirloc + dp->d_reclen > DIRBLKSIZ) {
391 		((struct direct *)(dirbuf + prev))->d_reclen =
392 		    DIRBLKSIZ - prev;
393 		if (fwrite(dirbuf, DIRBLKSIZ, 1, df) != 1)
394 			fail_dirtmp(dirfile);
395 		dirloc = 0;
396 	}
397 	memmove(dirbuf + dirloc, dp, (long)dp->d_reclen);
398 	prev = dirloc;
399 	dirloc += dp->d_reclen;
400 }
401 
402 /*
403  * flush out a directory that is finished.
404  */
405 static void
406 flushent(void)
407 {
408 	((struct direct *)(dirbuf + prev))->d_reclen = DIRBLKSIZ - prev;
409 	if (fwrite(dirbuf, (int)dirloc, 1, df) != 1)
410 		fail_dirtmp(dirfile);
411 	seekpt = ftell(df);
412 	dirloc = 0;
413 }
414 
415 /*
416  * Save extended attributes for a directory entry to a file.
417  */
418 static void
419 putdirattrs(char *buf, long size)
420 {
421 
422 	if (mf != NULL && fwrite(buf, size, 1, mf) != 1)
423 		fail_dirtmp(modefile);
424 }
425 
426 /*
427  * Seek to an entry in a directory.
428  * Only values returned by rst_telldir should be passed to rst_seekdir.
429  * This routine handles many directories in a single file.
430  * It takes the base of the directory in the file, plus
431  * the desired seek offset into it.
432  */
433 static void
434 rst_seekdir(RST_DIR *dirp, long loc, long base)
435 {
436 
437 	if (loc == rst_telldir(dirp))
438 		return;
439 	loc -= base;
440 	if (loc < 0)
441 		fprintf(stderr, "bad seek pointer to rst_seekdir %ld\n", loc);
442 	(void) lseek(dirp->dd_fd, base + (loc & ~(DIRBLKSIZ - 1)), SEEK_SET);
443 	dirp->dd_loc = loc & (DIRBLKSIZ - 1);
444 	if (dirp->dd_loc != 0)
445 		dirp->dd_size = read(dirp->dd_fd, dirp->dd_buf, DIRBLKSIZ);
446 }
447 
448 /*
449  * get next entry in a directory.
450  */
451 struct direct *
452 rst_readdir(RST_DIR *dirp)
453 {
454 	struct direct *dp;
455 
456 	for (;;) {
457 		if (dirp->dd_loc == 0) {
458 			dirp->dd_size = read(dirp->dd_fd, dirp->dd_buf,
459 			    DIRBLKSIZ);
460 			if (dirp->dd_size <= 0) {
461 				dprintf(stderr, "error reading directory\n");
462 				return (NULL);
463 			}
464 		}
465 		if (dirp->dd_loc >= dirp->dd_size) {
466 			dirp->dd_loc = 0;
467 			continue;
468 		}
469 		dp = (struct direct *)(dirp->dd_buf + dirp->dd_loc);
470 		if (dp->d_reclen == 0 ||
471 		    dp->d_reclen > DIRBLKSIZ + 1 - dirp->dd_loc) {
472 			dprintf(stderr, "corrupted directory: bad reclen %d\n",
473 				dp->d_reclen);
474 			return (NULL);
475 		}
476 		dirp->dd_loc += dp->d_reclen;
477 		if (dp->d_ino == 0 && strcmp(dp->d_name, "/") == 0)
478 			return (NULL);
479 		if (dp->d_ino >= maxino) {
480 			dprintf(stderr, "corrupted directory: bad inum %d\n",
481 				dp->d_ino);
482 			continue;
483 		}
484 		return (dp);
485 	}
486 }
487 
488 /*
489  * Simulate the opening of a directory
490  */
491 void *
492 rst_opendir(const char *name)
493 {
494 	struct inotab *itp;
495 	RST_DIR *dirp;
496 	ino_t ino;
497 
498 	if ((ino = dirlookup(name)) > 0 &&
499 	    (itp = inotablookup(ino)) != NULL) {
500 		dirp = opendirfile(dirfile);
501 		rst_seekdir(dirp, itp->t_seekpt, itp->t_seekpt);
502 		return (dirp);
503 	}
504 	return (NULL);
505 }
506 
507 /*
508  * In our case, there is nothing to do when closing a directory.
509  */
510 void
511 rst_closedir(void *arg)
512 {
513 	RST_DIR *dirp;
514 
515 	dirp = arg;
516 	(void)close(dirp->dd_fd);
517 	free(dirp);
518 	return;
519 }
520 
521 /*
522  * Simulate finding the current offset in the directory.
523  */
524 static long
525 rst_telldir(RST_DIR *dirp)
526 {
527 	return ((long)lseek(dirp->dd_fd,
528 	    (off_t)0, SEEK_CUR) - dirp->dd_size + dirp->dd_loc);
529 }
530 
531 /*
532  * Open a directory file.
533  */
534 static RST_DIR *
535 opendirfile(const char *name)
536 {
537 	RST_DIR *dirp;
538 	int fd;
539 
540 	if ((fd = open(name, O_RDONLY)) == -1)
541 		return (NULL);
542 	if ((dirp = malloc(sizeof(RST_DIR))) == NULL) {
543 		(void)close(fd);
544 		return (NULL);
545 	}
546 	dirp->dd_fd = fd;
547 	dirp->dd_loc = 0;
548 	return (dirp);
549 }
550 
551 /*
552  * Set the mode, owner, and times for all new or changed directories
553  */
554 void
555 setdirmodes(int flags)
556 {
557 	FILE *mf;
558 	struct modeinfo node;
559 	struct entry *ep;
560 	char *cp, *buf;
561 	const char *tmpdir;
562 	int bufsize;
563 	uid_t myuid;
564 
565 	vprintf(stdout, "Set directory mode, owner, and times.\n");
566 	if ((tmpdir = getenv("TMPDIR")) == NULL || tmpdir[0] == '\0')
567 		tmpdir = _PATH_TMP;
568 	if (command == 'r' || command == 'R')
569 		(void) sprintf(modefile, "%s/rstmode%d", tmpdir, dumpdate);
570 	if (modefile[0] == '#') {
571 		panic("modefile not defined\n");
572 		fprintf(stderr, "directory mode, owner, and times not set\n");
573 		return;
574 	}
575 	mf = fopen(modefile, "r");
576 	if (mf == NULL) {
577 		fprintf(stderr, "fopen: %s\n", strerror(errno));
578 		fprintf(stderr, "cannot open mode file %s\n", modefile);
579 		fprintf(stderr, "directory mode, owner, and times not set\n");
580 		return;
581 	}
582 	clearerr(mf);
583 	bufsize = 0;
584 	myuid = getuid();
585 	for (;;) {
586 		(void) fread((char *)&node, 1, sizeof(struct modeinfo), mf);
587 		if (ferror(mf)) {
588 			warn("%s: cannot read modefile.", modefile);
589 			fprintf(stderr, "Mode, owner, and times not set.\n");
590 			break;
591 		}
592 		if (feof(mf))
593 			break;
594 		if (node.extsize > 0) {
595 			if (bufsize < node.extsize) {
596 				if (bufsize > 0)
597 					free(buf);
598 				if ((buf = malloc(node.extsize)) != 0) {
599 					bufsize = node.extsize;
600 				} else {
601 					bufsize = 0;
602 				}
603 			}
604 			if (bufsize >= node.extsize) {
605 				(void) fread(buf, 1, node.extsize, mf);
606 				if (ferror(mf)) {
607 					warn("%s: cannot read modefile.",
608 					    modefile);
609 					fprintf(stderr, "Not all external ");
610 					fprintf(stderr, "attributes set.\n");
611 					break;
612 				}
613 			} else {
614 				(void) fseek(mf, node.extsize, SEEK_CUR);
615 				if (ferror(mf)) {
616 					warn("%s: cannot seek in modefile.",
617 					    modefile);
618 					fprintf(stderr, "Not all directory ");
619 					fprintf(stderr, "attributes set.\n");
620 					break;
621 				}
622 			}
623 		}
624 		ep = lookupino(node.ino);
625 		if (command == 'i' || command == 'x') {
626 			if (ep == NULL)
627 				continue;
628 			if ((flags & FORCE) == 0 && ep->e_flags & EXISTED) {
629 				ep->e_flags &= ~NEW;
630 				continue;
631 			}
632 			if (node.ino == ROOTINO &&
633 		   	    reply("set owner/mode for '.'") == FAIL)
634 				continue;
635 		}
636 		if (ep == NULL) {
637 			panic("cannot find directory inode %d\n", node.ino);
638 			continue;
639 		}
640 		cp = myname(ep);
641 		if (!Nflag) {
642 			if (node.extsize > 0) {
643 				if (bufsize >= node.extsize) {
644 					set_extattr_file(cp, buf, node.extsize);
645 				} else {
646 					fprintf(stderr, "Cannot restore %s%s\n",
647 					    "extended attributes for ", cp);
648 				}
649 			}
650 			if (myuid != 0)
651 				(void) chown(cp, myuid, node.gid);
652 			else
653 				(void) chown(cp, node.uid, node.gid);
654 			(void) chmod(cp, node.mode);
655 			utimes(cp, node.ctimep);
656 			utimes(cp, node.mtimep);
657 			(void) chflags(cp, node.flags);
658 		}
659 		ep->e_flags &= ~NEW;
660 	}
661 	if (bufsize > 0)
662 		free(buf);
663 	(void) fclose(mf);
664 }
665 
666 /*
667  * Generate a literal copy of a directory.
668  */
669 int
670 genliteraldir(char *name, ino_t ino)
671 {
672 	struct inotab *itp;
673 	int ofile, dp, i, size;
674 	char buf[BUFSIZ];
675 
676 	itp = inotablookup(ino);
677 	if (itp == NULL)
678 		panic("Cannot find directory inode %d named %s\n", ino, name);
679 	if ((ofile = open(name, O_WRONLY | O_CREAT | O_TRUNC, 0666)) < 0) {
680 		fprintf(stderr, "%s: ", name);
681 		(void) fflush(stderr);
682 		fprintf(stderr, "cannot create file: %s\n", strerror(errno));
683 		return (FAIL);
684 	}
685 	rst_seekdir(dirp, itp->t_seekpt, itp->t_seekpt);
686 	dp = dup(dirp->dd_fd);
687 	for (i = itp->t_size; i > 0; i -= BUFSIZ) {
688 		size = i < BUFSIZ ? i : BUFSIZ;
689 		if (read(dp, buf, (int) size) == -1) {
690 			fprintf(stderr,
691 				"write error extracting inode %d, name %s\n",
692 				curfile.ino, curfile.name);
693 			fprintf(stderr, "read: %s\n", strerror(errno));
694 			done(1);
695 		}
696 		if (!Nflag && write(ofile, buf, (int) size) == -1) {
697 			fprintf(stderr,
698 				"write error extracting inode %d, name %s\n",
699 				curfile.ino, curfile.name);
700 			fprintf(stderr, "write: %s\n", strerror(errno));
701 			done(1);
702 		}
703 	}
704 	(void) close(dp);
705 	(void) close(ofile);
706 	return (GOOD);
707 }
708 
709 /*
710  * Determine the type of an inode
711  */
712 int
713 inodetype(ino_t ino)
714 {
715 	struct inotab *itp;
716 
717 	itp = inotablookup(ino);
718 	if (itp == NULL)
719 		return (LEAF);
720 	return (NODE);
721 }
722 
723 /*
724  * Allocate and initialize a directory inode entry.
725  * If requested, save its pertinent mode, owner, and time info.
726  */
727 static struct inotab *
728 allocinotab(struct context *ctxp, long seekpt)
729 {
730 	struct inotab	*itp;
731 	struct modeinfo node;
732 
733 	itp = calloc(1, sizeof(struct inotab));
734 	if (itp == NULL)
735 		panic("no memory for directory table\n");
736 	itp->t_next = inotab[INOHASH(ctxp->ino)];
737 	inotab[INOHASH(ctxp->ino)] = itp;
738 	itp->t_ino = ctxp->ino;
739 	itp->t_seekpt = seekpt;
740 	if (mf == NULL)
741 		return (itp);
742 	node.ino = ctxp->ino;
743 	node.mtimep[0].tv_sec = ctxp->atime_sec;
744 	node.mtimep[0].tv_usec = ctxp->atime_nsec / 1000;
745 	node.mtimep[1].tv_sec = ctxp->mtime_sec;
746 	node.mtimep[1].tv_usec = ctxp->mtime_nsec / 1000;
747 	node.ctimep[0].tv_sec = ctxp->atime_sec;
748 	node.ctimep[0].tv_usec = ctxp->atime_nsec / 1000;
749 	node.ctimep[1].tv_sec = ctxp->birthtime_sec;
750 	node.ctimep[1].tv_usec = ctxp->birthtime_nsec / 1000;
751 	node.extsize = ctxp->extsize;
752 	node.mode = ctxp->mode;
753 	node.flags = ctxp->file_flags;
754 	node.uid = ctxp->uid;
755 	node.gid = ctxp->gid;
756 	if (fwrite((char *)&node, sizeof(struct modeinfo), 1, mf) != 1)
757 		fail_dirtmp(modefile);
758 	return (itp);
759 }
760 
761 /*
762  * Look up an inode in the table of directories
763  */
764 static struct inotab *
765 inotablookup(ino_t ino)
766 {
767 	struct inotab *itp;
768 
769 	for (itp = inotab[INOHASH(ino)]; itp != NULL; itp = itp->t_next)
770 		if (itp->t_ino == ino)
771 			return (itp);
772 	return (NULL);
773 }
774 
775 /*
776  * Clean up and exit
777  */
778 void
779 done(int exitcode)
780 {
781 
782 	closemt();
783 	if (modefile[0] != '#') {
784 		(void) truncate(modefile, 0);
785 		(void) unlink(modefile);
786 	}
787 	if (dirfile[0] != '#') {
788 		(void) truncate(dirfile, 0);
789 		(void) unlink(dirfile);
790 	}
791 	exit(exitcode);
792 }
793 
794 /*
795  * Print out information about the failure to save directory,
796  * extended attribute, and mode information.
797  */
798 static void
799 fail_dirtmp(char *filename)
800 {
801 	const char *tmpdir;
802 
803 	warn("%s: cannot write directory database", filename);
804 	if (errno == ENOSPC) {
805 		if ((tmpdir = getenv("TMPDIR")) == NULL || tmpdir[0] == '\0')
806 			tmpdir = _PATH_TMP;
807 		fprintf(stderr, "Try making space in %s, %s\n%s\n", tmpdir,
808 		    "or set environment variable TMPDIR",
809 		    "to an alternate location with more disk space.");
810 	}
811 	done(1);
812 }
813