xref: /freebsd/sbin/restore/dirs.c (revision 59966d51c60f6f98fad5feae4e90d98fdf9b2017)
18fae3551SRodney W. Grimes /*
28fae3551SRodney W. Grimes  * Copyright (c) 1983, 1993
38fae3551SRodney W. Grimes  *	The Regents of the University of California.  All rights reserved.
48fae3551SRodney W. Grimes  * (c) UNIX System Laboratories, Inc.
58fae3551SRodney W. Grimes  * All or some portions of this file are derived from material licensed
68fae3551SRodney W. Grimes  * to the University of California by American Telephone and Telegraph
78fae3551SRodney W. Grimes  * Co. or Unix System Laboratories, Inc. and are reproduced herein with
88fae3551SRodney W. Grimes  * the permission of UNIX System Laboratories, Inc.
98fae3551SRodney W. Grimes  *
108fae3551SRodney W. Grimes  * Redistribution and use in source and binary forms, with or without
118fae3551SRodney W. Grimes  * modification, are permitted provided that the following conditions
128fae3551SRodney W. Grimes  * are met:
138fae3551SRodney W. Grimes  * 1. Redistributions of source code must retain the above copyright
148fae3551SRodney W. Grimes  *    notice, this list of conditions and the following disclaimer.
158fae3551SRodney W. Grimes  * 2. Redistributions in binary form must reproduce the above copyright
168fae3551SRodney W. Grimes  *    notice, this list of conditions and the following disclaimer in the
178fae3551SRodney W. Grimes  *    documentation and/or other materials provided with the distribution.
188fae3551SRodney W. Grimes  * 4. Neither the name of the University nor the names of its contributors
198fae3551SRodney W. Grimes  *    may be used to endorse or promote products derived from this software
208fae3551SRodney W. Grimes  *    without specific prior written permission.
218fae3551SRodney W. Grimes  *
228fae3551SRodney W. Grimes  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
238fae3551SRodney W. Grimes  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
248fae3551SRodney W. Grimes  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
258fae3551SRodney W. Grimes  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
268fae3551SRodney W. Grimes  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
278fae3551SRodney W. Grimes  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
288fae3551SRodney W. Grimes  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
298fae3551SRodney W. Grimes  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
308fae3551SRodney W. Grimes  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
318fae3551SRodney W. Grimes  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
328fae3551SRodney W. Grimes  * SUCH DAMAGE.
338fae3551SRodney W. Grimes  */
348fae3551SRodney W. Grimes 
358fae3551SRodney W. Grimes #ifndef lint
3691ac32e4SPhilippe Charnier #if 0
37d87d79aeSPeter Wemm static char sccsid[] = "@(#)dirs.c	8.7 (Berkeley) 5/1/95";
3891ac32e4SPhilippe Charnier #endif
3991ac32e4SPhilippe Charnier static const char rcsid[] =
407f3dea24SPeter Wemm   "$FreeBSD$";
418fae3551SRodney W. Grimes #endif /* not lint */
428fae3551SRodney W. Grimes 
438fae3551SRodney W. Grimes #include <sys/param.h>
448fae3551SRodney W. Grimes #include <sys/file.h>
458fae3551SRodney W. Grimes #include <sys/stat.h>
46eb2fc780SGarrett Wollman #include <sys/time.h>
478fae3551SRodney W. Grimes 
488fae3551SRodney W. Grimes #include <ufs/ufs/dinode.h>
498fae3551SRodney W. Grimes #include <ufs/ufs/dir.h>
508fae3551SRodney W. Grimes #include <protocols/dumprestore.h>
518fae3551SRodney W. Grimes 
5291ac32e4SPhilippe Charnier #include <err.h>
538fae3551SRodney W. Grimes #include <errno.h>
5489fdc4e1SMike Barcroft #include <limits.h>
558454c72cSDavid E. O'Brien #include <paths.h>
56cca1b3aaSJaakko Heinonen #include <stdint.h>
578fae3551SRodney W. Grimes #include <stdio.h>
588fae3551SRodney W. Grimes #include <stdlib.h>
598fae3551SRodney W. Grimes #include <string.h>
608fae3551SRodney W. Grimes #include <unistd.h>
61a7ab72edSPaul Traina 
628fae3551SRodney W. Grimes #include "restore.h"
638fae3551SRodney W. Grimes #include "extern.h"
648fae3551SRodney W. Grimes 
658fae3551SRodney W. Grimes /*
668fae3551SRodney W. Grimes  * Symbol table of directories read from tape.
678fae3551SRodney W. Grimes  */
688fae3551SRodney W. Grimes #define HASHSIZE	1000
698fae3551SRodney W. Grimes #define INOHASH(val) (val % HASHSIZE)
708fae3551SRodney W. Grimes struct inotab {
718fae3551SRodney W. Grimes 	struct	inotab *t_next;
728fae3551SRodney W. Grimes 	ino_t	t_ino;
7397b465b1SDima Ruban 	int32_t	t_seekpt;
7497b465b1SDima Ruban 	int32_t	t_size;
758fae3551SRodney W. Grimes };
768fae3551SRodney W. Grimes static struct inotab *inotab[HASHSIZE];
778fae3551SRodney W. Grimes 
788fae3551SRodney W. Grimes /*
798fae3551SRodney W. Grimes  * Information retained about directories.
808fae3551SRodney W. Grimes  */
818fae3551SRodney W. Grimes struct modeinfo {
828fae3551SRodney W. Grimes 	ino_t ino;
83*59966d51SJilles Tjoelker 	struct timespec ctimep[2];
84*59966d51SJilles Tjoelker 	struct timespec mtimep[2];
85d87d79aeSPeter Wemm 	mode_t mode;
86d87d79aeSPeter Wemm 	uid_t uid;
87d87d79aeSPeter Wemm 	gid_t gid;
88d87d79aeSPeter Wemm 	int flags;
89772ad651SKirk McKusick 	int extsize;
908fae3551SRodney W. Grimes };
918fae3551SRodney W. Grimes 
928fae3551SRodney W. Grimes /*
938fae3551SRodney W. Grimes  * Definitions for library routines operating on directories.
948fae3551SRodney W. Grimes  */
958fae3551SRodney W. Grimes #undef DIRBLKSIZ
968fae3551SRodney W. Grimes #define DIRBLKSIZ 1024
978fae3551SRodney W. Grimes struct rstdirdesc {
988fae3551SRodney W. Grimes 	int	dd_fd;
9997b465b1SDima Ruban 	int32_t	dd_loc;
10097b465b1SDima Ruban 	int32_t	dd_size;
1018fae3551SRodney W. Grimes 	char	dd_buf[DIRBLKSIZ];
1028fae3551SRodney W. Grimes };
1038fae3551SRodney W. Grimes 
1048fae3551SRodney W. Grimes /*
1058fae3551SRodney W. Grimes  * Global variables for this file.
1068fae3551SRodney W. Grimes  */
1078fae3551SRodney W. Grimes static long	seekpt;
1088fae3551SRodney W. Grimes static FILE	*df, *mf;
1098fae3551SRodney W. Grimes static RST_DIR	*dirp;
110335524b9SWarner Losh static char	dirfile[MAXPATHLEN] = "#";	/* No file */
111335524b9SWarner Losh static char	modefile[MAXPATHLEN] = "#";	/* No file */
1128fae3551SRodney W. Grimes static char	dot[2] = ".";			/* So it can be modified */
1138fae3551SRodney W. Grimes 
1141c85e6a3SKirk McKusick static struct inotab	*allocinotab(struct context *, long);
1152db673abSWarner Losh static void		 flushent(void);
1162db673abSWarner Losh static struct inotab	*inotablookup(ino_t);
1172db673abSWarner Losh static RST_DIR		*opendirfile(const char *);
1182db673abSWarner Losh static void		 putdir(char *, long);
119772ad651SKirk McKusick static void		 putdirattrs(char *, long);
1202db673abSWarner Losh static void		 putent(struct direct *);
1212db673abSWarner Losh static void		 rst_seekdir(RST_DIR *, long, long);
1222db673abSWarner Losh static long		 rst_telldir(RST_DIR *);
1232db673abSWarner Losh static struct direct	*searchdir(ino_t, char *);
124aed3576dSKirk McKusick static void		 fail_dirtmp(char *);
1258fae3551SRodney W. Grimes 
1268fae3551SRodney W. Grimes /*
1278fae3551SRodney W. Grimes  *	Extract directory contents, building up a directory structure
1288fae3551SRodney W. Grimes  *	on disk for extraction by name.
1298fae3551SRodney W. Grimes  *	If genmode is requested, save mode, owner, and times for all
1308fae3551SRodney W. Grimes  *	directories on the tape.
1318fae3551SRodney W. Grimes  */
1328fae3551SRodney W. Grimes void
1332db673abSWarner Losh extractdirs(int genmode)
1348fae3551SRodney W. Grimes {
1358fae3551SRodney W. Grimes 	struct inotab *itp;
1368fae3551SRodney W. Grimes 	struct direct nulldir;
1371c85e6a3SKirk McKusick 	int i, fd;
1384afde855SMaxim Sobolev 	const char *tmpdir;
1398fae3551SRodney W. Grimes 
1408fae3551SRodney W. Grimes 	vprintf(stdout, "Extract directories from tape\n");
1414afde855SMaxim Sobolev 	if ((tmpdir = getenv("TMPDIR")) == NULL || tmpdir[0] == '\0')
1424afde855SMaxim Sobolev 		tmpdir = _PATH_TMP;
143cca1b3aaSJaakko Heinonen 	(void) sprintf(dirfile, "%s/rstdir%jd", tmpdir, (intmax_t)dumpdate);
144335524b9SWarner Losh 	if (command != 'r' && command != 'R') {
145cca1b3aaSJaakko Heinonen 		(void) strcat(dirfile, "-XXXXXX");
146335524b9SWarner Losh 		fd = mkstemp(dirfile);
147335524b9SWarner Losh 	} else
148335524b9SWarner Losh 		fd = open(dirfile, O_RDWR|O_CREAT|O_EXCL, 0666);
149335524b9SWarner Losh 	if (fd == -1 || (df = fdopen(fd, "w")) == NULL) {
150335524b9SWarner Losh 		if (fd != -1)
151335524b9SWarner Losh 			close(fd);
152aed3576dSKirk McKusick 		warn("%s: cannot create directory database", dirfile);
1538fae3551SRodney W. Grimes 		done(1);
1548fae3551SRodney W. Grimes 	}
1558fae3551SRodney W. Grimes 	if (genmode != 0) {
156cca1b3aaSJaakko Heinonen 		(void) sprintf(modefile, "%s/rstmode%jd", tmpdir,
157cca1b3aaSJaakko Heinonen 		    (intmax_t)dumpdate);
158335524b9SWarner Losh 		if (command != 'r' && command != 'R') {
159cca1b3aaSJaakko Heinonen 			(void) strcat(modefile, "-XXXXXX");
160335524b9SWarner Losh 			fd = mkstemp(modefile);
161335524b9SWarner Losh 		} else
162335524b9SWarner Losh 			fd = open(modefile, O_RDWR|O_CREAT|O_EXCL, 0666);
163335524b9SWarner Losh 		if (fd == -1 || (mf = fdopen(fd, "w")) == NULL) {
164335524b9SWarner Losh 			if (fd != -1)
165335524b9SWarner Losh 				close(fd);
166aed3576dSKirk McKusick 			warn("%s: cannot create modefile", modefile);
1678fae3551SRodney W. Grimes 			done(1);
1688fae3551SRodney W. Grimes 		}
1698fae3551SRodney W. Grimes 	}
1708fae3551SRodney W. Grimes 	nulldir.d_ino = 0;
1718fae3551SRodney W. Grimes 	nulldir.d_type = DT_DIR;
1728fae3551SRodney W. Grimes 	nulldir.d_namlen = 1;
1738fae3551SRodney W. Grimes 	(void) strcpy(nulldir.d_name, "/");
1748fae3551SRodney W. Grimes 	nulldir.d_reclen = DIRSIZ(0, &nulldir);
1758fae3551SRodney W. Grimes 	for (;;) {
1768fae3551SRodney W. Grimes 		curfile.name = "<directory file - name unknown>";
1778fae3551SRodney W. Grimes 		curfile.action = USING;
178aed3576dSKirk McKusick 		if (curfile.mode == 0 || (curfile.mode & IFMT) != IFDIR)
179aed3576dSKirk McKusick 			break;
1801c85e6a3SKirk McKusick 		itp = allocinotab(&curfile, seekpt);
181772ad651SKirk McKusick 		getfile(putdir, putdirattrs, xtrnull);
1828fae3551SRodney W. Grimes 		putent(&nulldir);
1838fae3551SRodney W. Grimes 		flushent();
1848fae3551SRodney W. Grimes 		itp->t_size = seekpt - itp->t_seekpt;
1858fae3551SRodney W. Grimes 	}
186aed3576dSKirk McKusick 	if (fclose(df) != 0)
187aed3576dSKirk McKusick 		fail_dirtmp(dirfile);
188aed3576dSKirk McKusick 	dirp = opendirfile(dirfile);
189aed3576dSKirk McKusick 	if (dirp == NULL)
190aed3576dSKirk McKusick 		fprintf(stderr, "opendirfile: %s\n", strerror(errno));
191aed3576dSKirk McKusick 	if (mf != NULL && fclose(mf) != 0)
192aed3576dSKirk McKusick 		fail_dirtmp(modefile);
193aed3576dSKirk McKusick 	i = dirlookup(dot);
194aed3576dSKirk McKusick 	if (i == 0)
195aed3576dSKirk McKusick 		panic("Root directory is not on tape\n");
1968fae3551SRodney W. Grimes }
1978fae3551SRodney W. Grimes 
1988fae3551SRodney W. Grimes /*
1998fae3551SRodney W. Grimes  * skip over all the directories on the tape
2008fae3551SRodney W. Grimes  */
2018fae3551SRodney W. Grimes void
2022db673abSWarner Losh skipdirs(void)
2038fae3551SRodney W. Grimes {
2048fae3551SRodney W. Grimes 
2051c85e6a3SKirk McKusick 	while (curfile.ino && (curfile.mode & IFMT) == IFDIR) {
2068fae3551SRodney W. Grimes 		skipfile();
2078fae3551SRodney W. Grimes 	}
2088fae3551SRodney W. Grimes }
2098fae3551SRodney W. Grimes 
2108fae3551SRodney W. Grimes /*
2118fae3551SRodney W. Grimes  *	Recursively find names and inumbers of all files in subtree
2128fae3551SRodney W. Grimes  *	pname and pass them off to be processed.
2138fae3551SRodney W. Grimes  */
2148fae3551SRodney W. Grimes void
2152db673abSWarner Losh treescan(char *pname, ino_t ino, long (*todo)(char *, ino_t, int))
2168fae3551SRodney W. Grimes {
2173d438ad6SDavid E. O'Brien 	struct inotab *itp;
2183d438ad6SDavid E. O'Brien 	struct direct *dp;
2198fae3551SRodney W. Grimes 	int namelen;
2208fae3551SRodney W. Grimes 	long bpt;
22152b370feSWarner Losh 	char locname[MAXPATHLEN];
2228fae3551SRodney W. Grimes 
2238fae3551SRodney W. Grimes 	itp = inotablookup(ino);
2248fae3551SRodney W. Grimes 	if (itp == NULL) {
2258fae3551SRodney W. Grimes 		/*
2268fae3551SRodney W. Grimes 		 * Pname is name of a simple file or an unchanged directory.
2278fae3551SRodney W. Grimes 		 */
2288fae3551SRodney W. Grimes 		(void) (*todo)(pname, ino, LEAF);
2298fae3551SRodney W. Grimes 		return;
2308fae3551SRodney W. Grimes 	}
2318fae3551SRodney W. Grimes 	/*
2328fae3551SRodney W. Grimes 	 * Pname is a dumped directory name.
2338fae3551SRodney W. Grimes 	 */
2348fae3551SRodney W. Grimes 	if ((*todo)(pname, ino, NODE) == FAIL)
2358fae3551SRodney W. Grimes 		return;
2368fae3551SRodney W. Grimes 	/*
2378fae3551SRodney W. Grimes 	 * begin search through the directory
2388fae3551SRodney W. Grimes 	 * skipping over "." and ".."
2398fae3551SRodney W. Grimes 	 */
24052b370feSWarner Losh 	(void) strlcpy(locname, pname, sizeof(locname));
24152b370feSWarner Losh 	(void) strlcat(locname, "/", sizeof(locname));
2428fae3551SRodney W. Grimes 	namelen = strlen(locname);
2438fae3551SRodney W. Grimes 	rst_seekdir(dirp, itp->t_seekpt, itp->t_seekpt);
2448fae3551SRodney W. Grimes 	dp = rst_readdir(dirp); /* "." */
2458fae3551SRodney W. Grimes 	if (dp != NULL && strcmp(dp->d_name, ".") == 0)
2468fae3551SRodney W. Grimes 		dp = rst_readdir(dirp); /* ".." */
2478fae3551SRodney W. Grimes 	else
2488fae3551SRodney W. Grimes 		fprintf(stderr, "Warning: `.' missing from directory %s\n",
2498fae3551SRodney W. Grimes 			pname);
2508fae3551SRodney W. Grimes 	if (dp != NULL && strcmp(dp->d_name, "..") == 0)
2518fae3551SRodney W. Grimes 		dp = rst_readdir(dirp); /* first real entry */
2528fae3551SRodney W. Grimes 	else
2538fae3551SRodney W. Grimes 		fprintf(stderr, "Warning: `..' missing from directory %s\n",
2548fae3551SRodney W. Grimes 			pname);
2558fae3551SRodney W. Grimes 	bpt = rst_telldir(dirp);
2568fae3551SRodney W. Grimes 	/*
2578fae3551SRodney W. Grimes 	 * a zero inode signals end of directory
2588fae3551SRodney W. Grimes 	 */
259d87d79aeSPeter Wemm 	while (dp != NULL) {
2608fae3551SRodney W. Grimes 		locname[namelen] = '\0';
261335524b9SWarner Losh 		if (namelen + dp->d_namlen >= sizeof(locname)) {
262cca1b3aaSJaakko Heinonen 			fprintf(stderr, "%s%s: name exceeds %zu char\n",
263335524b9SWarner Losh 			    locname, dp->d_name, sizeof(locname) - 1);
2648fae3551SRodney W. Grimes 		} else {
26552b370feSWarner Losh 			(void)strlcat(locname, dp->d_name, sizeof(locname));
2668fae3551SRodney W. Grimes 			treescan(locname, dp->d_ino, todo);
2678fae3551SRodney W. Grimes 			rst_seekdir(dirp, bpt, itp->t_seekpt);
2688fae3551SRodney W. Grimes 		}
2698fae3551SRodney W. Grimes 		dp = rst_readdir(dirp);
2708fae3551SRodney W. Grimes 		bpt = rst_telldir(dirp);
2718fae3551SRodney W. Grimes 	}
2728fae3551SRodney W. Grimes }
2738fae3551SRodney W. Grimes 
2748fae3551SRodney W. Grimes /*
2758fae3551SRodney W. Grimes  * Lookup a pathname which is always assumed to start from the ROOTINO.
2768fae3551SRodney W. Grimes  */
2778fae3551SRodney W. Grimes struct direct *
2782db673abSWarner Losh pathsearch(const char *pathname)
2798fae3551SRodney W. Grimes {
2808fae3551SRodney W. Grimes 	ino_t ino;
2818fae3551SRodney W. Grimes 	struct direct *dp;
2828fae3551SRodney W. Grimes 	char *path, *name, buffer[MAXPATHLEN];
2838fae3551SRodney W. Grimes 
2848fae3551SRodney W. Grimes 	strcpy(buffer, pathname);
2858fae3551SRodney W. Grimes 	path = buffer;
2868fae3551SRodney W. Grimes 	ino = ROOTINO;
2878fae3551SRodney W. Grimes 	while (*path == '/')
2888fae3551SRodney W. Grimes 		path++;
2898fae3551SRodney W. Grimes 	dp = NULL;
290d030d2d2SPoul-Henning Kamp 	while ((name = strsep(&path, "/")) != NULL && *name != '\0') {
2918fae3551SRodney W. Grimes 		if ((dp = searchdir(ino, name)) == NULL)
2928fae3551SRodney W. Grimes 			return (NULL);
2938fae3551SRodney W. Grimes 		ino = dp->d_ino;
2948fae3551SRodney W. Grimes 	}
2958fae3551SRodney W. Grimes 	return (dp);
2968fae3551SRodney W. Grimes }
2978fae3551SRodney W. Grimes 
2988fae3551SRodney W. Grimes /*
2998fae3551SRodney W. Grimes  * Lookup the requested name in directory inum.
3008fae3551SRodney W. Grimes  * Return its inode number if found, zero if it does not exist.
3018fae3551SRodney W. Grimes  */
3028fae3551SRodney W. Grimes static struct direct *
3032db673abSWarner Losh searchdir(ino_t	inum, char *name)
3048fae3551SRodney W. Grimes {
3053d438ad6SDavid E. O'Brien 	struct direct *dp;
3063d438ad6SDavid E. O'Brien 	struct inotab *itp;
3078fae3551SRodney W. Grimes 	int len;
3088fae3551SRodney W. Grimes 
3098fae3551SRodney W. Grimes 	itp = inotablookup(inum);
3108fae3551SRodney W. Grimes 	if (itp == NULL)
3118fae3551SRodney W. Grimes 		return (NULL);
3128fae3551SRodney W. Grimes 	rst_seekdir(dirp, itp->t_seekpt, itp->t_seekpt);
3138fae3551SRodney W. Grimes 	len = strlen(name);
3148fae3551SRodney W. Grimes 	do {
3158fae3551SRodney W. Grimes 		dp = rst_readdir(dirp);
316d87d79aeSPeter Wemm 		if (dp == NULL)
3178fae3551SRodney W. Grimes 			return (NULL);
3188fae3551SRodney W. Grimes 	} while (dp->d_namlen != len || strncmp(dp->d_name, name, len) != 0);
3198fae3551SRodney W. Grimes 	return (dp);
3208fae3551SRodney W. Grimes }
3218fae3551SRodney W. Grimes 
3228fae3551SRodney W. Grimes /*
3238fae3551SRodney W. Grimes  * Put the directory entries in the directory file
3248fae3551SRodney W. Grimes  */
3258fae3551SRodney W. Grimes static void
3262db673abSWarner Losh putdir(char *buf, long size)
3278fae3551SRodney W. Grimes {
3283d438ad6SDavid E. O'Brien 	struct direct *dp;
3298fae3551SRodney W. Grimes 	long loc, i;
3308fae3551SRodney W. Grimes 
3318fae3551SRodney W. Grimes 	for (loc = 0; loc < size; ) {
3328fae3551SRodney W. Grimes 		dp = (struct direct *)(buf + loc);
333a7ab72edSPaul Traina 		if (Bcvt)
3348fae3551SRodney W. Grimes 			swabst((u_char *)"ls", (u_char *) dp);
33566b42171SWarner Losh 		if (oldinofmt && dp->d_ino != 0) {
33666b42171SWarner Losh #if BYTE_ORDER == BIG_ENDIAN
33766b42171SWarner Losh 			if (Bcvt)
33866b42171SWarner Losh 				dp->d_namlen = dp->d_type;
33966b42171SWarner Losh #else
34066b42171SWarner Losh 			if (!Bcvt && dp->d_namlen == 0)
34166b42171SWarner Losh 				dp->d_namlen = dp->d_type;
34266b42171SWarner Losh #endif
34366b42171SWarner Losh 			dp->d_type = DT_UNKNOWN;
34466b42171SWarner Losh 		}
3458fae3551SRodney W. Grimes 		i = DIRBLKSIZ - (loc & (DIRBLKSIZ - 1));
3468fae3551SRodney W. Grimes 		if ((dp->d_reclen & 0x3) != 0 ||
3478fae3551SRodney W. Grimes 		    dp->d_reclen > i ||
34830d38f7bSPeter Wemm 		    dp->d_reclen < DIRSIZ(0, dp)
34930d38f7bSPeter Wemm #if NAME_MAX < 255
35030d38f7bSPeter Wemm 		    || dp->d_namlen > NAME_MAX
35130d38f7bSPeter Wemm #endif
35230d38f7bSPeter Wemm 		    ) {
3538fae3551SRodney W. Grimes 			vprintf(stdout, "Mangled directory: ");
3548fae3551SRodney W. Grimes 			if ((dp->d_reclen & 0x3) != 0)
3558fae3551SRodney W. Grimes 				vprintf(stdout,
3568fae3551SRodney W. Grimes 				   "reclen not multiple of 4 ");
3578fae3551SRodney W. Grimes 			if (dp->d_reclen < DIRSIZ(0, dp))
3588fae3551SRodney W. Grimes 				vprintf(stdout,
359cca1b3aaSJaakko Heinonen 				   "reclen less than DIRSIZ (%d < %zu) ",
3608fae3551SRodney W. Grimes 				   dp->d_reclen, DIRSIZ(0, dp));
36130d38f7bSPeter Wemm #if NAME_MAX < 255
3628fae3551SRodney W. Grimes 			if (dp->d_namlen > NAME_MAX)
3638fae3551SRodney W. Grimes 				vprintf(stdout,
3648fae3551SRodney W. Grimes 				   "reclen name too big (%d > %d) ",
3658fae3551SRodney W. Grimes 				   dp->d_namlen, NAME_MAX);
36630d38f7bSPeter Wemm #endif
3678fae3551SRodney W. Grimes 			vprintf(stdout, "\n");
3688fae3551SRodney W. Grimes 			loc += i;
3698fae3551SRodney W. Grimes 			continue;
3708fae3551SRodney W. Grimes 		}
3718fae3551SRodney W. Grimes 		loc += dp->d_reclen;
3728fae3551SRodney W. Grimes 		if (dp->d_ino != 0) {
3738fae3551SRodney W. Grimes 			putent(dp);
3748fae3551SRodney W. Grimes 		}
3758fae3551SRodney W. Grimes 	}
3768fae3551SRodney W. Grimes }
3778fae3551SRodney W. Grimes 
3788fae3551SRodney W. Grimes /*
3798fae3551SRodney W. Grimes  * These variables are "local" to the following two functions.
3808fae3551SRodney W. Grimes  */
3818fae3551SRodney W. Grimes char dirbuf[DIRBLKSIZ];
3828fae3551SRodney W. Grimes long dirloc = 0;
3838fae3551SRodney W. Grimes long prev = 0;
3848fae3551SRodney W. Grimes 
3858fae3551SRodney W. Grimes /*
3868fae3551SRodney W. Grimes  * add a new directory entry to a file.
3878fae3551SRodney W. Grimes  */
3888fae3551SRodney W. Grimes static void
3892db673abSWarner Losh putent(struct direct *dp)
3908fae3551SRodney W. Grimes {
3918fae3551SRodney W. Grimes 	dp->d_reclen = DIRSIZ(0, dp);
3928fae3551SRodney W. Grimes 	if (dirloc + dp->d_reclen > DIRBLKSIZ) {
3938fae3551SRodney W. Grimes 		((struct direct *)(dirbuf + prev))->d_reclen =
3948fae3551SRodney W. Grimes 		    DIRBLKSIZ - prev;
395aed3576dSKirk McKusick 		if (fwrite(dirbuf, DIRBLKSIZ, 1, df) != 1)
396aed3576dSKirk McKusick 			fail_dirtmp(dirfile);
3978fae3551SRodney W. Grimes 		dirloc = 0;
3988fae3551SRodney W. Grimes 	}
399d87d79aeSPeter Wemm 	memmove(dirbuf + dirloc, dp, (long)dp->d_reclen);
4008fae3551SRodney W. Grimes 	prev = dirloc;
4018fae3551SRodney W. Grimes 	dirloc += dp->d_reclen;
4028fae3551SRodney W. Grimes }
4038fae3551SRodney W. Grimes 
4048fae3551SRodney W. Grimes /*
4058fae3551SRodney W. Grimes  * flush out a directory that is finished.
4068fae3551SRodney W. Grimes  */
4078fae3551SRodney W. Grimes static void
4082db673abSWarner Losh flushent(void)
4098fae3551SRodney W. Grimes {
4108fae3551SRodney W. Grimes 	((struct direct *)(dirbuf + prev))->d_reclen = DIRBLKSIZ - prev;
411aed3576dSKirk McKusick 	if (fwrite(dirbuf, (int)dirloc, 1, df) != 1)
412aed3576dSKirk McKusick 		fail_dirtmp(dirfile);
4138fae3551SRodney W. Grimes 	seekpt = ftell(df);
4148fae3551SRodney W. Grimes 	dirloc = 0;
4158fae3551SRodney W. Grimes }
4168fae3551SRodney W. Grimes 
4178fae3551SRodney W. Grimes /*
418772ad651SKirk McKusick  * Save extended attributes for a directory entry to a file.
419772ad651SKirk McKusick  */
420772ad651SKirk McKusick static void
421772ad651SKirk McKusick putdirattrs(char *buf, long size)
422772ad651SKirk McKusick {
423772ad651SKirk McKusick 
424aed3576dSKirk McKusick 	if (mf != NULL && fwrite(buf, size, 1, mf) != 1)
425aed3576dSKirk McKusick 		fail_dirtmp(modefile);
426772ad651SKirk McKusick }
427772ad651SKirk McKusick 
428772ad651SKirk McKusick /*
4298fae3551SRodney W. Grimes  * Seek to an entry in a directory.
4308fae3551SRodney W. Grimes  * Only values returned by rst_telldir should be passed to rst_seekdir.
4318fae3551SRodney W. Grimes  * This routine handles many directories in a single file.
4328fae3551SRodney W. Grimes  * It takes the base of the directory in the file, plus
4338fae3551SRodney W. Grimes  * the desired seek offset into it.
4348fae3551SRodney W. Grimes  */
4358fae3551SRodney W. Grimes static void
4362db673abSWarner Losh rst_seekdir(RST_DIR *dirp, long loc, long base)
4378fae3551SRodney W. Grimes {
4388fae3551SRodney W. Grimes 
4398fae3551SRodney W. Grimes 	if (loc == rst_telldir(dirp))
4408fae3551SRodney W. Grimes 		return;
4418fae3551SRodney W. Grimes 	loc -= base;
4428fae3551SRodney W. Grimes 	if (loc < 0)
4437fafc6f6SBruce Evans 		fprintf(stderr, "bad seek pointer to rst_seekdir %ld\n", loc);
4448fae3551SRodney W. Grimes 	(void) lseek(dirp->dd_fd, base + (loc & ~(DIRBLKSIZ - 1)), SEEK_SET);
4458fae3551SRodney W. Grimes 	dirp->dd_loc = loc & (DIRBLKSIZ - 1);
4468fae3551SRodney W. Grimes 	if (dirp->dd_loc != 0)
4478fae3551SRodney W. Grimes 		dirp->dd_size = read(dirp->dd_fd, dirp->dd_buf, DIRBLKSIZ);
4488fae3551SRodney W. Grimes }
4498fae3551SRodney W. Grimes 
4508fae3551SRodney W. Grimes /*
4518fae3551SRodney W. Grimes  * get next entry in a directory.
4528fae3551SRodney W. Grimes  */
4538fae3551SRodney W. Grimes struct direct *
4542db673abSWarner Losh rst_readdir(RST_DIR *dirp)
4558fae3551SRodney W. Grimes {
4563d438ad6SDavid E. O'Brien 	struct direct *dp;
4578fae3551SRodney W. Grimes 
4588fae3551SRodney W. Grimes 	for (;;) {
4598fae3551SRodney W. Grimes 		if (dirp->dd_loc == 0) {
4608fae3551SRodney W. Grimes 			dirp->dd_size = read(dirp->dd_fd, dirp->dd_buf,
4618fae3551SRodney W. Grimes 			    DIRBLKSIZ);
4628fae3551SRodney W. Grimes 			if (dirp->dd_size <= 0) {
4638fae3551SRodney W. Grimes 				dprintf(stderr, "error reading directory\n");
4648fae3551SRodney W. Grimes 				return (NULL);
4658fae3551SRodney W. Grimes 			}
4668fae3551SRodney W. Grimes 		}
4678fae3551SRodney W. Grimes 		if (dirp->dd_loc >= dirp->dd_size) {
4688fae3551SRodney W. Grimes 			dirp->dd_loc = 0;
4698fae3551SRodney W. Grimes 			continue;
4708fae3551SRodney W. Grimes 		}
4718fae3551SRodney W. Grimes 		dp = (struct direct *)(dirp->dd_buf + dirp->dd_loc);
4728fae3551SRodney W. Grimes 		if (dp->d_reclen == 0 ||
4738fae3551SRodney W. Grimes 		    dp->d_reclen > DIRBLKSIZ + 1 - dirp->dd_loc) {
4748fae3551SRodney W. Grimes 			dprintf(stderr, "corrupted directory: bad reclen %d\n",
4758fae3551SRodney W. Grimes 				dp->d_reclen);
4768fae3551SRodney W. Grimes 			return (NULL);
4778fae3551SRodney W. Grimes 		}
4788fae3551SRodney W. Grimes 		dirp->dd_loc += dp->d_reclen;
479d87d79aeSPeter Wemm 		if (dp->d_ino == 0 && strcmp(dp->d_name, "/") == 0)
480d87d79aeSPeter Wemm 			return (NULL);
4818fae3551SRodney W. Grimes 		if (dp->d_ino >= maxino) {
4828fae3551SRodney W. Grimes 			dprintf(stderr, "corrupted directory: bad inum %d\n",
4838fae3551SRodney W. Grimes 				dp->d_ino);
4848fae3551SRodney W. Grimes 			continue;
4858fae3551SRodney W. Grimes 		}
4868fae3551SRodney W. Grimes 		return (dp);
4878fae3551SRodney W. Grimes 	}
4888fae3551SRodney W. Grimes }
4898fae3551SRodney W. Grimes 
4908fae3551SRodney W. Grimes /*
4918fae3551SRodney W. Grimes  * Simulate the opening of a directory
4928fae3551SRodney W. Grimes  */
4931916772eSStefan Farfeleder void *
4942db673abSWarner Losh rst_opendir(const char *name)
4958fae3551SRodney W. Grimes {
4968fae3551SRodney W. Grimes 	struct inotab *itp;
4978fae3551SRodney W. Grimes 	RST_DIR *dirp;
4988fae3551SRodney W. Grimes 	ino_t ino;
4998fae3551SRodney W. Grimes 
5008fae3551SRodney W. Grimes 	if ((ino = dirlookup(name)) > 0 &&
5018fae3551SRodney W. Grimes 	    (itp = inotablookup(ino)) != NULL) {
5028fae3551SRodney W. Grimes 		dirp = opendirfile(dirfile);
5038fae3551SRodney W. Grimes 		rst_seekdir(dirp, itp->t_seekpt, itp->t_seekpt);
5048fae3551SRodney W. Grimes 		return (dirp);
5058fae3551SRodney W. Grimes 	}
5068fae3551SRodney W. Grimes 	return (NULL);
5078fae3551SRodney W. Grimes }
5088fae3551SRodney W. Grimes 
5098fae3551SRodney W. Grimes /*
5108fae3551SRodney W. Grimes  * In our case, there is nothing to do when closing a directory.
5118fae3551SRodney W. Grimes  */
5128fae3551SRodney W. Grimes void
5131916772eSStefan Farfeleder rst_closedir(void *arg)
5148fae3551SRodney W. Grimes {
5151916772eSStefan Farfeleder 	RST_DIR *dirp;
5168fae3551SRodney W. Grimes 
5171916772eSStefan Farfeleder 	dirp = arg;
5188fae3551SRodney W. Grimes 	(void)close(dirp->dd_fd);
5198fae3551SRodney W. Grimes 	free(dirp);
5208fae3551SRodney W. Grimes 	return;
5218fae3551SRodney W. Grimes }
5228fae3551SRodney W. Grimes 
5238fae3551SRodney W. Grimes /*
5248fae3551SRodney W. Grimes  * Simulate finding the current offset in the directory.
5258fae3551SRodney W. Grimes  */
5268fae3551SRodney W. Grimes static long
5272db673abSWarner Losh rst_telldir(RST_DIR *dirp)
5288fae3551SRodney W. Grimes {
5298fae3551SRodney W. Grimes 	return ((long)lseek(dirp->dd_fd,
5308fae3551SRodney W. Grimes 	    (off_t)0, SEEK_CUR) - dirp->dd_size + dirp->dd_loc);
5318fae3551SRodney W. Grimes }
5328fae3551SRodney W. Grimes 
5338fae3551SRodney W. Grimes /*
5348fae3551SRodney W. Grimes  * Open a directory file.
5358fae3551SRodney W. Grimes  */
5368fae3551SRodney W. Grimes static RST_DIR *
5372db673abSWarner Losh opendirfile(const char *name)
5388fae3551SRodney W. Grimes {
5393d438ad6SDavid E. O'Brien 	RST_DIR *dirp;
5403d438ad6SDavid E. O'Brien 	int fd;
5418fae3551SRodney W. Grimes 
5428fae3551SRodney W. Grimes 	if ((fd = open(name, O_RDONLY)) == -1)
5438fae3551SRodney W. Grimes 		return (NULL);
5448fae3551SRodney W. Grimes 	if ((dirp = malloc(sizeof(RST_DIR))) == NULL) {
5458fae3551SRodney W. Grimes 		(void)close(fd);
5468fae3551SRodney W. Grimes 		return (NULL);
5478fae3551SRodney W. Grimes 	}
5488fae3551SRodney W. Grimes 	dirp->dd_fd = fd;
5498fae3551SRodney W. Grimes 	dirp->dd_loc = 0;
5508fae3551SRodney W. Grimes 	return (dirp);
5518fae3551SRodney W. Grimes }
5528fae3551SRodney W. Grimes 
5538fae3551SRodney W. Grimes /*
5548fae3551SRodney W. Grimes  * Set the mode, owner, and times for all new or changed directories
5558fae3551SRodney W. Grimes  */
5568fae3551SRodney W. Grimes void
5572db673abSWarner Losh setdirmodes(int flags)
5588fae3551SRodney W. Grimes {
5598fae3551SRodney W. Grimes 	FILE *mf;
5608fae3551SRodney W. Grimes 	struct modeinfo node;
5618fae3551SRodney W. Grimes 	struct entry *ep;
562772ad651SKirk McKusick 	char *cp, *buf;
5634afde855SMaxim Sobolev 	const char *tmpdir;
564772ad651SKirk McKusick 	int bufsize;
565c028393dSKirk McKusick 	uid_t myuid;
5668fae3551SRodney W. Grimes 
5678fae3551SRodney W. Grimes 	vprintf(stdout, "Set directory mode, owner, and times.\n");
5684afde855SMaxim Sobolev 	if ((tmpdir = getenv("TMPDIR")) == NULL || tmpdir[0] == '\0')
5694afde855SMaxim Sobolev 		tmpdir = _PATH_TMP;
570335524b9SWarner Losh 	if (command == 'r' || command == 'R')
571cca1b3aaSJaakko Heinonen 		(void) sprintf(modefile, "%s/rstmode%jd", tmpdir,
572cca1b3aaSJaakko Heinonen 		    (intmax_t)dumpdate);
573335524b9SWarner Losh 	if (modefile[0] == '#') {
574335524b9SWarner Losh 		panic("modefile not defined\n");
575335524b9SWarner Losh 		fprintf(stderr, "directory mode, owner, and times not set\n");
576335524b9SWarner Losh 		return;
577335524b9SWarner Losh 	}
5788fae3551SRodney W. Grimes 	mf = fopen(modefile, "r");
5798fae3551SRodney W. Grimes 	if (mf == NULL) {
5808fae3551SRodney W. Grimes 		fprintf(stderr, "fopen: %s\n", strerror(errno));
5818fae3551SRodney W. Grimes 		fprintf(stderr, "cannot open mode file %s\n", modefile);
5828fae3551SRodney W. Grimes 		fprintf(stderr, "directory mode, owner, and times not set\n");
5838fae3551SRodney W. Grimes 		return;
5848fae3551SRodney W. Grimes 	}
5858fae3551SRodney W. Grimes 	clearerr(mf);
586772ad651SKirk McKusick 	bufsize = 0;
587c028393dSKirk McKusick 	myuid = getuid();
5888fae3551SRodney W. Grimes 	for (;;) {
5898fae3551SRodney W. Grimes 		(void) fread((char *)&node, 1, sizeof(struct modeinfo), mf);
590aed3576dSKirk McKusick 		if (ferror(mf)) {
591aed3576dSKirk McKusick 			warn("%s: cannot read modefile.", modefile);
592aed3576dSKirk McKusick 			fprintf(stderr, "Mode, owner, and times not set.\n");
593aed3576dSKirk McKusick 			break;
594aed3576dSKirk McKusick 		}
5958fae3551SRodney W. Grimes 		if (feof(mf))
5968fae3551SRodney W. Grimes 			break;
597772ad651SKirk McKusick 		if (node.extsize > 0) {
598772ad651SKirk McKusick 			if (bufsize < node.extsize) {
599772ad651SKirk McKusick 				if (bufsize > 0)
600772ad651SKirk McKusick 					free(buf);
601772ad651SKirk McKusick 				if ((buf = malloc(node.extsize)) != 0) {
602772ad651SKirk McKusick 					bufsize = node.extsize;
603772ad651SKirk McKusick 				} else {
604772ad651SKirk McKusick 					bufsize = 0;
605772ad651SKirk McKusick 				}
606772ad651SKirk McKusick 			}
607772ad651SKirk McKusick 			if (bufsize >= node.extsize) {
608772ad651SKirk McKusick 				(void) fread(buf, 1, node.extsize, mf);
609aed3576dSKirk McKusick 				if (ferror(mf)) {
610aed3576dSKirk McKusick 					warn("%s: cannot read modefile.",
611aed3576dSKirk McKusick 					    modefile);
612aed3576dSKirk McKusick 					fprintf(stderr, "Not all external ");
613aed3576dSKirk McKusick 					fprintf(stderr, "attributes set.\n");
614aed3576dSKirk McKusick 					break;
615aed3576dSKirk McKusick 				}
616772ad651SKirk McKusick 			} else {
617772ad651SKirk McKusick 				(void) fseek(mf, node.extsize, SEEK_CUR);
618aed3576dSKirk McKusick 				if (ferror(mf)) {
619aed3576dSKirk McKusick 					warn("%s: cannot seek in modefile.",
620aed3576dSKirk McKusick 					    modefile);
621aed3576dSKirk McKusick 					fprintf(stderr, "Not all directory ");
622aed3576dSKirk McKusick 					fprintf(stderr, "attributes set.\n");
623aed3576dSKirk McKusick 					break;
624aed3576dSKirk McKusick 				}
625772ad651SKirk McKusick 			}
626772ad651SKirk McKusick 		}
6278fae3551SRodney W. Grimes 		ep = lookupino(node.ino);
6288fae3551SRodney W. Grimes 		if (command == 'i' || command == 'x') {
6298fae3551SRodney W. Grimes 			if (ep == NULL)
6308fae3551SRodney W. Grimes 				continue;
6318fae3551SRodney W. Grimes 			if ((flags & FORCE) == 0 && ep->e_flags & EXISTED) {
6328fae3551SRodney W. Grimes 				ep->e_flags &= ~NEW;
6338fae3551SRodney W. Grimes 				continue;
6348fae3551SRodney W. Grimes 			}
6358fae3551SRodney W. Grimes 			if (node.ino == ROOTINO &&
6368fae3551SRodney W. Grimes 		   	    reply("set owner/mode for '.'") == FAIL)
6378fae3551SRodney W. Grimes 				continue;
6388fae3551SRodney W. Grimes 		}
6398fae3551SRodney W. Grimes 		if (ep == NULL) {
640e25a029eSMatthew D Fleming 			panic("cannot find directory inode %ju\n",
641e25a029eSMatthew D Fleming 			    (uintmax_t)node.ino);
642772ad651SKirk McKusick 			continue;
643772ad651SKirk McKusick 		}
6448fae3551SRodney W. Grimes 		cp = myname(ep);
645b9595aa5SDima Dorfman 		if (!Nflag) {
646772ad651SKirk McKusick 			if (node.extsize > 0) {
647772ad651SKirk McKusick 				if (bufsize >= node.extsize) {
648772ad651SKirk McKusick 					set_extattr_file(cp, buf, node.extsize);
649772ad651SKirk McKusick 				} else {
650772ad651SKirk McKusick 					fprintf(stderr, "Cannot restore %s%s\n",
651772ad651SKirk McKusick 					    "extended attributes for ", cp);
652772ad651SKirk McKusick 				}
653772ad651SKirk McKusick 			}
654c028393dSKirk McKusick 			if (myuid != 0)
655c028393dSKirk McKusick 				(void) chown(cp, myuid, node.gid);
656c028393dSKirk McKusick 			else
6578fae3551SRodney W. Grimes 				(void) chown(cp, node.uid, node.gid);
6588fae3551SRodney W. Grimes 			(void) chmod(cp, node.mode);
659*59966d51SJilles Tjoelker 			utimensat(AT_FDCWD, cp, node.ctimep, 0);
660*59966d51SJilles Tjoelker 			utimensat(AT_FDCWD, cp, node.mtimep, 0);
661fddde8b0SDavid Malone 			(void) chflags(cp, node.flags);
662b9595aa5SDima Dorfman 		}
6638fae3551SRodney W. Grimes 		ep->e_flags &= ~NEW;
6648fae3551SRodney W. Grimes 	}
665772ad651SKirk McKusick 	if (bufsize > 0)
666772ad651SKirk McKusick 		free(buf);
6678fae3551SRodney W. Grimes 	(void) fclose(mf);
6688fae3551SRodney W. Grimes }
6698fae3551SRodney W. Grimes 
6708fae3551SRodney W. Grimes /*
6718fae3551SRodney W. Grimes  * Generate a literal copy of a directory.
6728fae3551SRodney W. Grimes  */
6738fae3551SRodney W. Grimes int
6742db673abSWarner Losh genliteraldir(char *name, ino_t ino)
6758fae3551SRodney W. Grimes {
6763d438ad6SDavid E. O'Brien 	struct inotab *itp;
6778fae3551SRodney W. Grimes 	int ofile, dp, i, size;
6788fae3551SRodney W. Grimes 	char buf[BUFSIZ];
6798fae3551SRodney W. Grimes 
6808fae3551SRodney W. Grimes 	itp = inotablookup(ino);
6818fae3551SRodney W. Grimes 	if (itp == NULL)
682e25a029eSMatthew D Fleming 		panic("Cannot find directory inode %ju named %s\n",
683e25a029eSMatthew D Fleming 		    (uintmax_t)ino, name);
684335524b9SWarner Losh 	if ((ofile = open(name, O_WRONLY | O_CREAT | O_TRUNC, 0666)) < 0) {
6858fae3551SRodney W. Grimes 		fprintf(stderr, "%s: ", name);
6868fae3551SRodney W. Grimes 		(void) fflush(stderr);
6878fae3551SRodney W. Grimes 		fprintf(stderr, "cannot create file: %s\n", strerror(errno));
6888fae3551SRodney W. Grimes 		return (FAIL);
6898fae3551SRodney W. Grimes 	}
6908fae3551SRodney W. Grimes 	rst_seekdir(dirp, itp->t_seekpt, itp->t_seekpt);
6918fae3551SRodney W. Grimes 	dp = dup(dirp->dd_fd);
6928fae3551SRodney W. Grimes 	for (i = itp->t_size; i > 0; i -= BUFSIZ) {
6938fae3551SRodney W. Grimes 		size = i < BUFSIZ ? i : BUFSIZ;
6948fae3551SRodney W. Grimes 		if (read(dp, buf, (int) size) == -1) {
6958fae3551SRodney W. Grimes 			fprintf(stderr,
696e25a029eSMatthew D Fleming 			    "write error extracting inode %ju, name %s\n",
697e25a029eSMatthew D Fleming 			    (uintmax_t)curfile.ino, curfile.name);
6988fae3551SRodney W. Grimes 			fprintf(stderr, "read: %s\n", strerror(errno));
6998fae3551SRodney W. Grimes 			done(1);
7008fae3551SRodney W. Grimes 		}
7018fae3551SRodney W. Grimes 		if (!Nflag && write(ofile, buf, (int) size) == -1) {
7028fae3551SRodney W. Grimes 			fprintf(stderr,
703e25a029eSMatthew D Fleming 			    "write error extracting inode %ju, name %s\n",
704e25a029eSMatthew D Fleming 			    (uintmax_t)curfile.ino, curfile.name);
7058fae3551SRodney W. Grimes 			fprintf(stderr, "write: %s\n", strerror(errno));
7068fae3551SRodney W. Grimes 			done(1);
7078fae3551SRodney W. Grimes 		}
7088fae3551SRodney W. Grimes 	}
7098fae3551SRodney W. Grimes 	(void) close(dp);
7108fae3551SRodney W. Grimes 	(void) close(ofile);
7118fae3551SRodney W. Grimes 	return (GOOD);
7128fae3551SRodney W. Grimes }
7138fae3551SRodney W. Grimes 
7148fae3551SRodney W. Grimes /*
7158fae3551SRodney W. Grimes  * Determine the type of an inode
7168fae3551SRodney W. Grimes  */
7178fae3551SRodney W. Grimes int
7182db673abSWarner Losh inodetype(ino_t ino)
7198fae3551SRodney W. Grimes {
7208fae3551SRodney W. Grimes 	struct inotab *itp;
7218fae3551SRodney W. Grimes 
7228fae3551SRodney W. Grimes 	itp = inotablookup(ino);
7238fae3551SRodney W. Grimes 	if (itp == NULL)
7248fae3551SRodney W. Grimes 		return (LEAF);
7258fae3551SRodney W. Grimes 	return (NODE);
7268fae3551SRodney W. Grimes }
7278fae3551SRodney W. Grimes 
7288fae3551SRodney W. Grimes /*
7298fae3551SRodney W. Grimes  * Allocate and initialize a directory inode entry.
7308fae3551SRodney W. Grimes  * If requested, save its pertinent mode, owner, and time info.
7318fae3551SRodney W. Grimes  */
7328fae3551SRodney W. Grimes static struct inotab *
7331c85e6a3SKirk McKusick allocinotab(struct context *ctxp, long seekpt)
7348fae3551SRodney W. Grimes {
7353d438ad6SDavid E. O'Brien 	struct inotab	*itp;
7368fae3551SRodney W. Grimes 	struct modeinfo node;
7378fae3551SRodney W. Grimes 
7388fae3551SRodney W. Grimes 	itp = calloc(1, sizeof(struct inotab));
7398fae3551SRodney W. Grimes 	if (itp == NULL)
740772ad651SKirk McKusick 		panic("no memory for directory table\n");
7411c85e6a3SKirk McKusick 	itp->t_next = inotab[INOHASH(ctxp->ino)];
7421c85e6a3SKirk McKusick 	inotab[INOHASH(ctxp->ino)] = itp;
7431c85e6a3SKirk McKusick 	itp->t_ino = ctxp->ino;
7448fae3551SRodney W. Grimes 	itp->t_seekpt = seekpt;
7458fae3551SRodney W. Grimes 	if (mf == NULL)
7468fae3551SRodney W. Grimes 		return (itp);
7471c85e6a3SKirk McKusick 	node.ino = ctxp->ino;
748fb36a3d8SKirk McKusick 	node.mtimep[0].tv_sec = ctxp->atime_sec;
749*59966d51SJilles Tjoelker 	node.mtimep[0].tv_nsec = ctxp->atime_nsec;
750fb36a3d8SKirk McKusick 	node.mtimep[1].tv_sec = ctxp->mtime_sec;
751*59966d51SJilles Tjoelker 	node.mtimep[1].tv_nsec = ctxp->mtime_nsec;
752fb36a3d8SKirk McKusick 	node.ctimep[0].tv_sec = ctxp->atime_sec;
753*59966d51SJilles Tjoelker 	node.ctimep[0].tv_nsec = ctxp->atime_nsec;
754fb36a3d8SKirk McKusick 	node.ctimep[1].tv_sec = ctxp->birthtime_sec;
755*59966d51SJilles Tjoelker 	node.ctimep[1].tv_nsec = ctxp->birthtime_nsec;
756772ad651SKirk McKusick 	node.extsize = ctxp->extsize;
7571c85e6a3SKirk McKusick 	node.mode = ctxp->mode;
7581c85e6a3SKirk McKusick 	node.flags = ctxp->file_flags;
7591c85e6a3SKirk McKusick 	node.uid = ctxp->uid;
7601c85e6a3SKirk McKusick 	node.gid = ctxp->gid;
761aed3576dSKirk McKusick 	if (fwrite((char *)&node, sizeof(struct modeinfo), 1, mf) != 1)
762aed3576dSKirk McKusick 		fail_dirtmp(modefile);
7638fae3551SRodney W. Grimes 	return (itp);
7648fae3551SRodney W. Grimes }
7658fae3551SRodney W. Grimes 
7668fae3551SRodney W. Grimes /*
7678fae3551SRodney W. Grimes  * Look up an inode in the table of directories
7688fae3551SRodney W. Grimes  */
7698fae3551SRodney W. Grimes static struct inotab *
7702db673abSWarner Losh inotablookup(ino_t ino)
7718fae3551SRodney W. Grimes {
7723d438ad6SDavid E. O'Brien 	struct inotab *itp;
7738fae3551SRodney W. Grimes 
7748fae3551SRodney W. Grimes 	for (itp = inotab[INOHASH(ino)]; itp != NULL; itp = itp->t_next)
7758fae3551SRodney W. Grimes 		if (itp->t_ino == ino)
7768fae3551SRodney W. Grimes 			return (itp);
7778fae3551SRodney W. Grimes 	return (NULL);
7788fae3551SRodney W. Grimes }
7798fae3551SRodney W. Grimes 
7808fae3551SRodney W. Grimes /*
7818fae3551SRodney W. Grimes  * Clean up and exit
7828fae3551SRodney W. Grimes  */
783eaa86f9dSBruce Evans void
7842db673abSWarner Losh done(int exitcode)
7858fae3551SRodney W. Grimes {
7868fae3551SRodney W. Grimes 
7878fae3551SRodney W. Grimes 	closemt();
788aed3576dSKirk McKusick 	if (modefile[0] != '#') {
789aed3576dSKirk McKusick 		(void) truncate(modefile, 0);
7908fae3551SRodney W. Grimes 		(void) unlink(modefile);
791aed3576dSKirk McKusick 	}
792aed3576dSKirk McKusick 	if (dirfile[0] != '#') {
793aed3576dSKirk McKusick 		(void) truncate(dirfile, 0);
7948fae3551SRodney W. Grimes 		(void) unlink(dirfile);
795aed3576dSKirk McKusick 	}
7968fae3551SRodney W. Grimes 	exit(exitcode);
7978fae3551SRodney W. Grimes }
798aed3576dSKirk McKusick 
799aed3576dSKirk McKusick /*
800aed3576dSKirk McKusick  * Print out information about the failure to save directory,
801aed3576dSKirk McKusick  * extended attribute, and mode information.
802aed3576dSKirk McKusick  */
803aed3576dSKirk McKusick static void
804aed3576dSKirk McKusick fail_dirtmp(char *filename)
805aed3576dSKirk McKusick {
806aed3576dSKirk McKusick 	const char *tmpdir;
807aed3576dSKirk McKusick 
808aed3576dSKirk McKusick 	warn("%s: cannot write directory database", filename);
809aed3576dSKirk McKusick 	if (errno == ENOSPC) {
810aed3576dSKirk McKusick 		if ((tmpdir = getenv("TMPDIR")) == NULL || tmpdir[0] == '\0')
811aed3576dSKirk McKusick 			tmpdir = _PATH_TMP;
812aed3576dSKirk McKusick 		fprintf(stderr, "Try making space in %s, %s\n%s\n", tmpdir,
813aed3576dSKirk McKusick 		    "or set environment variable TMPDIR",
814aed3576dSKirk McKusick 		    "to an alternate location with more disk space.");
815aed3576dSKirk McKusick 	}
816aed3576dSKirk McKusick 	done(1);
817aed3576dSKirk McKusick }
818