xref: /freebsd/sbin/restore/dirs.c (revision aed3576d26f2c2e34691fea10f894511703157aa)
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>
568fae3551SRodney W. Grimes #include <stdio.h>
578fae3551SRodney W. Grimes #include <stdlib.h>
588fae3551SRodney W. Grimes #include <string.h>
598fae3551SRodney W. Grimes #include <unistd.h>
60a7ab72edSPaul Traina 
618fae3551SRodney W. Grimes #include "restore.h"
628fae3551SRodney W. Grimes #include "extern.h"
638fae3551SRodney W. Grimes 
648fae3551SRodney W. Grimes /*
658fae3551SRodney W. Grimes  * Symbol table of directories read from tape.
668fae3551SRodney W. Grimes  */
678fae3551SRodney W. Grimes #define HASHSIZE	1000
688fae3551SRodney W. Grimes #define INOHASH(val) (val % HASHSIZE)
698fae3551SRodney W. Grimes struct inotab {
708fae3551SRodney W. Grimes 	struct	inotab *t_next;
718fae3551SRodney W. Grimes 	ino_t	t_ino;
7297b465b1SDima Ruban 	int32_t	t_seekpt;
7397b465b1SDima Ruban 	int32_t	t_size;
748fae3551SRodney W. Grimes };
758fae3551SRodney W. Grimes static struct inotab *inotab[HASHSIZE];
768fae3551SRodney W. Grimes 
778fae3551SRodney W. Grimes /*
788fae3551SRodney W. Grimes  * Information retained about directories.
798fae3551SRodney W. Grimes  */
808fae3551SRodney W. Grimes struct modeinfo {
818fae3551SRodney W. Grimes 	ino_t ino;
82fb36a3d8SKirk McKusick 	struct timeval ctimep[2];
83fb36a3d8SKirk McKusick 	struct timeval mtimep[2];
84d87d79aeSPeter Wemm 	mode_t mode;
85d87d79aeSPeter Wemm 	uid_t uid;
86d87d79aeSPeter Wemm 	gid_t gid;
87d87d79aeSPeter Wemm 	int flags;
88772ad651SKirk McKusick 	int extsize;
898fae3551SRodney W. Grimes };
908fae3551SRodney W. Grimes 
918fae3551SRodney W. Grimes /*
928fae3551SRodney W. Grimes  * Definitions for library routines operating on directories.
938fae3551SRodney W. Grimes  */
948fae3551SRodney W. Grimes #undef DIRBLKSIZ
958fae3551SRodney W. Grimes #define DIRBLKSIZ 1024
968fae3551SRodney W. Grimes struct rstdirdesc {
978fae3551SRodney W. Grimes 	int	dd_fd;
9897b465b1SDima Ruban 	int32_t	dd_loc;
9997b465b1SDima Ruban 	int32_t	dd_size;
1008fae3551SRodney W. Grimes 	char	dd_buf[DIRBLKSIZ];
1018fae3551SRodney W. Grimes };
1028fae3551SRodney W. Grimes 
1038fae3551SRodney W. Grimes /*
1048fae3551SRodney W. Grimes  * Global variables for this file.
1058fae3551SRodney W. Grimes  */
1068fae3551SRodney W. Grimes static long	seekpt;
1078fae3551SRodney W. Grimes static FILE	*df, *mf;
1088fae3551SRodney W. Grimes static RST_DIR	*dirp;
109335524b9SWarner Losh static char	dirfile[MAXPATHLEN] = "#";	/* No file */
110335524b9SWarner Losh static char	modefile[MAXPATHLEN] = "#";	/* No file */
1118fae3551SRodney W. Grimes static char	dot[2] = ".";			/* So it can be modified */
1128fae3551SRodney W. Grimes 
1131c85e6a3SKirk McKusick static struct inotab	*allocinotab(struct context *, long);
1142db673abSWarner Losh static void		 flushent(void);
1152db673abSWarner Losh static struct inotab	*inotablookup(ino_t);
1162db673abSWarner Losh static RST_DIR		*opendirfile(const char *);
1172db673abSWarner Losh static void		 putdir(char *, long);
118772ad651SKirk McKusick static void		 putdirattrs(char *, long);
1192db673abSWarner Losh static void		 putent(struct direct *);
1202db673abSWarner Losh static void		 rst_seekdir(RST_DIR *, long, long);
1212db673abSWarner Losh static long		 rst_telldir(RST_DIR *);
1222db673abSWarner Losh static struct direct	*searchdir(ino_t, char *);
123aed3576dSKirk McKusick static void		 fail_dirtmp(char *);
1248fae3551SRodney W. Grimes 
1258fae3551SRodney W. Grimes /*
1268fae3551SRodney W. Grimes  *	Extract directory contents, building up a directory structure
1278fae3551SRodney W. Grimes  *	on disk for extraction by name.
1288fae3551SRodney W. Grimes  *	If genmode is requested, save mode, owner, and times for all
1298fae3551SRodney W. Grimes  *	directories on the tape.
1308fae3551SRodney W. Grimes  */
1318fae3551SRodney W. Grimes void
1322db673abSWarner Losh extractdirs(int genmode)
1338fae3551SRodney W. Grimes {
1348fae3551SRodney W. Grimes 	struct inotab *itp;
1358fae3551SRodney W. Grimes 	struct direct nulldir;
1361c85e6a3SKirk McKusick 	int i, fd;
1374afde855SMaxim Sobolev 	const char *tmpdir;
1388fae3551SRodney W. Grimes 
1398fae3551SRodney W. Grimes 	vprintf(stdout, "Extract directories from tape\n");
1404afde855SMaxim Sobolev 	if ((tmpdir = getenv("TMPDIR")) == NULL || tmpdir[0] == '\0')
1414afde855SMaxim Sobolev 		tmpdir = _PATH_TMP;
1424afde855SMaxim Sobolev 	(void) sprintf(dirfile, "%s/rstdir%d", tmpdir, dumpdate);
143335524b9SWarner Losh 	if (command != 'r' && command != 'R') {
144335524b9SWarner Losh 		(void *) strcat(dirfile, "-XXXXXX");
145335524b9SWarner Losh 		fd = mkstemp(dirfile);
146335524b9SWarner Losh 	} else
147335524b9SWarner Losh 		fd = open(dirfile, O_RDWR|O_CREAT|O_EXCL, 0666);
148335524b9SWarner Losh 	if (fd == -1 || (df = fdopen(fd, "w")) == NULL) {
149335524b9SWarner Losh 		if (fd != -1)
150335524b9SWarner Losh 			close(fd);
151aed3576dSKirk McKusick 		warn("%s: cannot create directory database", dirfile);
1528fae3551SRodney W. Grimes 		done(1);
1538fae3551SRodney W. Grimes 	}
1548fae3551SRodney W. Grimes 	if (genmode != 0) {
1554afde855SMaxim Sobolev 		(void) sprintf(modefile, "%s/rstmode%d", tmpdir, dumpdate);
156335524b9SWarner Losh 		if (command != 'r' && command != 'R') {
157335524b9SWarner Losh 			(void *) strcat(modefile, "-XXXXXX");
158335524b9SWarner Losh 			fd = mkstemp(modefile);
159335524b9SWarner Losh 		} else
160335524b9SWarner Losh 			fd = open(modefile, O_RDWR|O_CREAT|O_EXCL, 0666);
161335524b9SWarner Losh 		if (fd == -1 || (mf = fdopen(fd, "w")) == NULL) {
162335524b9SWarner Losh 			if (fd != -1)
163335524b9SWarner Losh 				close(fd);
164aed3576dSKirk McKusick 			warn("%s: cannot create modefile", modefile);
1658fae3551SRodney W. Grimes 			done(1);
1668fae3551SRodney W. Grimes 		}
1678fae3551SRodney W. Grimes 	}
1688fae3551SRodney W. Grimes 	nulldir.d_ino = 0;
1698fae3551SRodney W. Grimes 	nulldir.d_type = DT_DIR;
1708fae3551SRodney W. Grimes 	nulldir.d_namlen = 1;
1718fae3551SRodney W. Grimes 	(void) strcpy(nulldir.d_name, "/");
1728fae3551SRodney W. Grimes 	nulldir.d_reclen = DIRSIZ(0, &nulldir);
1738fae3551SRodney W. Grimes 	for (;;) {
1748fae3551SRodney W. Grimes 		curfile.name = "<directory file - name unknown>";
1758fae3551SRodney W. Grimes 		curfile.action = USING;
176aed3576dSKirk McKusick 		if (curfile.mode == 0 || (curfile.mode & IFMT) != IFDIR)
177aed3576dSKirk McKusick 			break;
1781c85e6a3SKirk McKusick 		itp = allocinotab(&curfile, seekpt);
179772ad651SKirk McKusick 		getfile(putdir, putdirattrs, xtrnull);
1808fae3551SRodney W. Grimes 		putent(&nulldir);
1818fae3551SRodney W. Grimes 		flushent();
1828fae3551SRodney W. Grimes 		itp->t_size = seekpt - itp->t_seekpt;
1838fae3551SRodney W. Grimes 	}
184aed3576dSKirk McKusick 	if (fclose(df) != 0)
185aed3576dSKirk McKusick 		fail_dirtmp(dirfile);
186aed3576dSKirk McKusick 	dirp = opendirfile(dirfile);
187aed3576dSKirk McKusick 	if (dirp == NULL)
188aed3576dSKirk McKusick 		fprintf(stderr, "opendirfile: %s\n", strerror(errno));
189aed3576dSKirk McKusick 	if (mf != NULL && fclose(mf) != 0)
190aed3576dSKirk McKusick 		fail_dirtmp(modefile);
191aed3576dSKirk McKusick 	i = dirlookup(dot);
192aed3576dSKirk McKusick 	if (i == 0)
193aed3576dSKirk McKusick 		panic("Root directory is not on tape\n");
1948fae3551SRodney W. Grimes }
1958fae3551SRodney W. Grimes 
1968fae3551SRodney W. Grimes /*
1978fae3551SRodney W. Grimes  * skip over all the directories on the tape
1988fae3551SRodney W. Grimes  */
1998fae3551SRodney W. Grimes void
2002db673abSWarner Losh skipdirs(void)
2018fae3551SRodney W. Grimes {
2028fae3551SRodney W. Grimes 
2031c85e6a3SKirk McKusick 	while (curfile.ino && (curfile.mode & IFMT) == IFDIR) {
2048fae3551SRodney W. Grimes 		skipfile();
2058fae3551SRodney W. Grimes 	}
2068fae3551SRodney W. Grimes }
2078fae3551SRodney W. Grimes 
2088fae3551SRodney W. Grimes /*
2098fae3551SRodney W. Grimes  *	Recursively find names and inumbers of all files in subtree
2108fae3551SRodney W. Grimes  *	pname and pass them off to be processed.
2118fae3551SRodney W. Grimes  */
2128fae3551SRodney W. Grimes void
2132db673abSWarner Losh treescan(char *pname, ino_t ino, long (*todo)(char *, ino_t, int))
2148fae3551SRodney W. Grimes {
2153d438ad6SDavid E. O'Brien 	struct inotab *itp;
2163d438ad6SDavid E. O'Brien 	struct direct *dp;
2178fae3551SRodney W. Grimes 	int namelen;
2188fae3551SRodney W. Grimes 	long bpt;
21952b370feSWarner Losh 	char locname[MAXPATHLEN];
2208fae3551SRodney W. Grimes 
2218fae3551SRodney W. Grimes 	itp = inotablookup(ino);
2228fae3551SRodney W. Grimes 	if (itp == NULL) {
2238fae3551SRodney W. Grimes 		/*
2248fae3551SRodney W. Grimes 		 * Pname is name of a simple file or an unchanged directory.
2258fae3551SRodney W. Grimes 		 */
2268fae3551SRodney W. Grimes 		(void) (*todo)(pname, ino, LEAF);
2278fae3551SRodney W. Grimes 		return;
2288fae3551SRodney W. Grimes 	}
2298fae3551SRodney W. Grimes 	/*
2308fae3551SRodney W. Grimes 	 * Pname is a dumped directory name.
2318fae3551SRodney W. Grimes 	 */
2328fae3551SRodney W. Grimes 	if ((*todo)(pname, ino, NODE) == FAIL)
2338fae3551SRodney W. Grimes 		return;
2348fae3551SRodney W. Grimes 	/*
2358fae3551SRodney W. Grimes 	 * begin search through the directory
2368fae3551SRodney W. Grimes 	 * skipping over "." and ".."
2378fae3551SRodney W. Grimes 	 */
23852b370feSWarner Losh 	(void) strlcpy(locname, pname, sizeof(locname));
23952b370feSWarner Losh 	(void) strlcat(locname, "/", sizeof(locname));
2408fae3551SRodney W. Grimes 	namelen = strlen(locname);
2418fae3551SRodney W. Grimes 	rst_seekdir(dirp, itp->t_seekpt, itp->t_seekpt);
2428fae3551SRodney W. Grimes 	dp = rst_readdir(dirp); /* "." */
2438fae3551SRodney W. Grimes 	if (dp != NULL && strcmp(dp->d_name, ".") == 0)
2448fae3551SRodney W. Grimes 		dp = rst_readdir(dirp); /* ".." */
2458fae3551SRodney W. Grimes 	else
2468fae3551SRodney W. Grimes 		fprintf(stderr, "Warning: `.' missing from directory %s\n",
2478fae3551SRodney W. Grimes 			pname);
2488fae3551SRodney W. Grimes 	if (dp != NULL && strcmp(dp->d_name, "..") == 0)
2498fae3551SRodney W. Grimes 		dp = rst_readdir(dirp); /* first real entry */
2508fae3551SRodney W. Grimes 	else
2518fae3551SRodney W. Grimes 		fprintf(stderr, "Warning: `..' missing from directory %s\n",
2528fae3551SRodney W. Grimes 			pname);
2538fae3551SRodney W. Grimes 	bpt = rst_telldir(dirp);
2548fae3551SRodney W. Grimes 	/*
2558fae3551SRodney W. Grimes 	 * a zero inode signals end of directory
2568fae3551SRodney W. Grimes 	 */
257d87d79aeSPeter Wemm 	while (dp != NULL) {
2588fae3551SRodney W. Grimes 		locname[namelen] = '\0';
259335524b9SWarner Losh 		if (namelen + dp->d_namlen >= sizeof(locname)) {
2608fae3551SRodney W. Grimes 			fprintf(stderr, "%s%s: name exceeds %d char\n",
261335524b9SWarner Losh 				locname, dp->d_name, sizeof(locname) - 1);
2628fae3551SRodney W. Grimes 		} else {
26352b370feSWarner Losh 			(void)strlcat(locname, dp->d_name, sizeof(locname));
2648fae3551SRodney W. Grimes 			treescan(locname, dp->d_ino, todo);
2658fae3551SRodney W. Grimes 			rst_seekdir(dirp, bpt, itp->t_seekpt);
2668fae3551SRodney W. Grimes 		}
2678fae3551SRodney W. Grimes 		dp = rst_readdir(dirp);
2688fae3551SRodney W. Grimes 		bpt = rst_telldir(dirp);
2698fae3551SRodney W. Grimes 	}
2708fae3551SRodney W. Grimes }
2718fae3551SRodney W. Grimes 
2728fae3551SRodney W. Grimes /*
2738fae3551SRodney W. Grimes  * Lookup a pathname which is always assumed to start from the ROOTINO.
2748fae3551SRodney W. Grimes  */
2758fae3551SRodney W. Grimes struct direct *
2762db673abSWarner Losh pathsearch(const char *pathname)
2778fae3551SRodney W. Grimes {
2788fae3551SRodney W. Grimes 	ino_t ino;
2798fae3551SRodney W. Grimes 	struct direct *dp;
2808fae3551SRodney W. Grimes 	char *path, *name, buffer[MAXPATHLEN];
2818fae3551SRodney W. Grimes 
2828fae3551SRodney W. Grimes 	strcpy(buffer, pathname);
2838fae3551SRodney W. Grimes 	path = buffer;
2848fae3551SRodney W. Grimes 	ino = ROOTINO;
2858fae3551SRodney W. Grimes 	while (*path == '/')
2868fae3551SRodney W. Grimes 		path++;
2878fae3551SRodney W. Grimes 	dp = NULL;
288d030d2d2SPoul-Henning Kamp 	while ((name = strsep(&path, "/")) != NULL && *name != '\0') {
2898fae3551SRodney W. Grimes 		if ((dp = searchdir(ino, name)) == NULL)
2908fae3551SRodney W. Grimes 			return (NULL);
2918fae3551SRodney W. Grimes 		ino = dp->d_ino;
2928fae3551SRodney W. Grimes 	}
2938fae3551SRodney W. Grimes 	return (dp);
2948fae3551SRodney W. Grimes }
2958fae3551SRodney W. Grimes 
2968fae3551SRodney W. Grimes /*
2978fae3551SRodney W. Grimes  * Lookup the requested name in directory inum.
2988fae3551SRodney W. Grimes  * Return its inode number if found, zero if it does not exist.
2998fae3551SRodney W. Grimes  */
3008fae3551SRodney W. Grimes static struct direct *
3012db673abSWarner Losh searchdir(ino_t	inum, char *name)
3028fae3551SRodney W. Grimes {
3033d438ad6SDavid E. O'Brien 	struct direct *dp;
3043d438ad6SDavid E. O'Brien 	struct inotab *itp;
3058fae3551SRodney W. Grimes 	int len;
3068fae3551SRodney W. Grimes 
3078fae3551SRodney W. Grimes 	itp = inotablookup(inum);
3088fae3551SRodney W. Grimes 	if (itp == NULL)
3098fae3551SRodney W. Grimes 		return (NULL);
3108fae3551SRodney W. Grimes 	rst_seekdir(dirp, itp->t_seekpt, itp->t_seekpt);
3118fae3551SRodney W. Grimes 	len = strlen(name);
3128fae3551SRodney W. Grimes 	do {
3138fae3551SRodney W. Grimes 		dp = rst_readdir(dirp);
314d87d79aeSPeter Wemm 		if (dp == NULL)
3158fae3551SRodney W. Grimes 			return (NULL);
3168fae3551SRodney W. Grimes 	} while (dp->d_namlen != len || strncmp(dp->d_name, name, len) != 0);
3178fae3551SRodney W. Grimes 	return (dp);
3188fae3551SRodney W. Grimes }
3198fae3551SRodney W. Grimes 
3208fae3551SRodney W. Grimes /*
3218fae3551SRodney W. Grimes  * Put the directory entries in the directory file
3228fae3551SRodney W. Grimes  */
3238fae3551SRodney W. Grimes static void
3242db673abSWarner Losh putdir(char *buf, long size)
3258fae3551SRodney W. Grimes {
3263d438ad6SDavid E. O'Brien 	struct direct *dp;
3278fae3551SRodney W. Grimes 	long loc, i;
3288fae3551SRodney W. Grimes 
3298fae3551SRodney W. Grimes 	for (loc = 0; loc < size; ) {
3308fae3551SRodney W. Grimes 		dp = (struct direct *)(buf + loc);
331a7ab72edSPaul Traina 		if (Bcvt)
3328fae3551SRodney W. Grimes 			swabst((u_char *)"ls", (u_char *) dp);
33366b42171SWarner Losh 		if (oldinofmt && dp->d_ino != 0) {
33466b42171SWarner Losh #if BYTE_ORDER == BIG_ENDIAN
33566b42171SWarner Losh 			if (Bcvt)
33666b42171SWarner Losh 				dp->d_namlen = dp->d_type;
33766b42171SWarner Losh #else
33866b42171SWarner Losh 			if (!Bcvt && dp->d_namlen == 0)
33966b42171SWarner Losh 				dp->d_namlen = dp->d_type;
34066b42171SWarner Losh #endif
34166b42171SWarner Losh 			dp->d_type = DT_UNKNOWN;
34266b42171SWarner Losh 		}
3438fae3551SRodney W. Grimes 		i = DIRBLKSIZ - (loc & (DIRBLKSIZ - 1));
3448fae3551SRodney W. Grimes 		if ((dp->d_reclen & 0x3) != 0 ||
3458fae3551SRodney W. Grimes 		    dp->d_reclen > i ||
34630d38f7bSPeter Wemm 		    dp->d_reclen < DIRSIZ(0, dp)
34730d38f7bSPeter Wemm #if NAME_MAX < 255
34830d38f7bSPeter Wemm 		    || dp->d_namlen > NAME_MAX
34930d38f7bSPeter Wemm #endif
35030d38f7bSPeter Wemm 		    ) {
3518fae3551SRodney W. Grimes 			vprintf(stdout, "Mangled directory: ");
3528fae3551SRodney W. Grimes 			if ((dp->d_reclen & 0x3) != 0)
3538fae3551SRodney W. Grimes 				vprintf(stdout,
3548fae3551SRodney W. Grimes 				   "reclen not multiple of 4 ");
3558fae3551SRodney W. Grimes 			if (dp->d_reclen < DIRSIZ(0, dp))
3568fae3551SRodney W. Grimes 				vprintf(stdout,
3578fae3551SRodney W. Grimes 				   "reclen less than DIRSIZ (%d < %d) ",
3588fae3551SRodney W. Grimes 				   dp->d_reclen, DIRSIZ(0, dp));
35930d38f7bSPeter Wemm #if NAME_MAX < 255
3608fae3551SRodney W. Grimes 			if (dp->d_namlen > NAME_MAX)
3618fae3551SRodney W. Grimes 				vprintf(stdout,
3628fae3551SRodney W. Grimes 				   "reclen name too big (%d > %d) ",
3638fae3551SRodney W. Grimes 				   dp->d_namlen, NAME_MAX);
36430d38f7bSPeter Wemm #endif
3658fae3551SRodney W. Grimes 			vprintf(stdout, "\n");
3668fae3551SRodney W. Grimes 			loc += i;
3678fae3551SRodney W. Grimes 			continue;
3688fae3551SRodney W. Grimes 		}
3698fae3551SRodney W. Grimes 		loc += dp->d_reclen;
3708fae3551SRodney W. Grimes 		if (dp->d_ino != 0) {
3718fae3551SRodney W. Grimes 			putent(dp);
3728fae3551SRodney W. Grimes 		}
3738fae3551SRodney W. Grimes 	}
3748fae3551SRodney W. Grimes }
3758fae3551SRodney W. Grimes 
3768fae3551SRodney W. Grimes /*
3778fae3551SRodney W. Grimes  * These variables are "local" to the following two functions.
3788fae3551SRodney W. Grimes  */
3798fae3551SRodney W. Grimes char dirbuf[DIRBLKSIZ];
3808fae3551SRodney W. Grimes long dirloc = 0;
3818fae3551SRodney W. Grimes long prev = 0;
3828fae3551SRodney W. Grimes 
3838fae3551SRodney W. Grimes /*
3848fae3551SRodney W. Grimes  * add a new directory entry to a file.
3858fae3551SRodney W. Grimes  */
3868fae3551SRodney W. Grimes static void
3872db673abSWarner Losh putent(struct direct *dp)
3888fae3551SRodney W. Grimes {
3898fae3551SRodney W. Grimes 	dp->d_reclen = DIRSIZ(0, dp);
3908fae3551SRodney W. Grimes 	if (dirloc + dp->d_reclen > DIRBLKSIZ) {
3918fae3551SRodney W. Grimes 		((struct direct *)(dirbuf + prev))->d_reclen =
3928fae3551SRodney W. Grimes 		    DIRBLKSIZ - prev;
393aed3576dSKirk McKusick 		if (fwrite(dirbuf, DIRBLKSIZ, 1, df) != 1)
394aed3576dSKirk McKusick 			fail_dirtmp(dirfile);
3958fae3551SRodney W. Grimes 		dirloc = 0;
3968fae3551SRodney W. Grimes 	}
397d87d79aeSPeter Wemm 	memmove(dirbuf + dirloc, dp, (long)dp->d_reclen);
3988fae3551SRodney W. Grimes 	prev = dirloc;
3998fae3551SRodney W. Grimes 	dirloc += dp->d_reclen;
4008fae3551SRodney W. Grimes }
4018fae3551SRodney W. Grimes 
4028fae3551SRodney W. Grimes /*
4038fae3551SRodney W. Grimes  * flush out a directory that is finished.
4048fae3551SRodney W. Grimes  */
4058fae3551SRodney W. Grimes static void
4062db673abSWarner Losh flushent(void)
4078fae3551SRodney W. Grimes {
4088fae3551SRodney W. Grimes 	((struct direct *)(dirbuf + prev))->d_reclen = DIRBLKSIZ - prev;
409aed3576dSKirk McKusick 	if (fwrite(dirbuf, (int)dirloc, 1, df) != 1)
410aed3576dSKirk McKusick 		fail_dirtmp(dirfile);
4118fae3551SRodney W. Grimes 	seekpt = ftell(df);
4128fae3551SRodney W. Grimes 	dirloc = 0;
4138fae3551SRodney W. Grimes }
4148fae3551SRodney W. Grimes 
4158fae3551SRodney W. Grimes /*
416772ad651SKirk McKusick  * Save extended attributes for a directory entry to a file.
417772ad651SKirk McKusick  */
418772ad651SKirk McKusick static void
419772ad651SKirk McKusick putdirattrs(char *buf, long size)
420772ad651SKirk McKusick {
421772ad651SKirk McKusick 
422aed3576dSKirk McKusick 	if (mf != NULL && fwrite(buf, size, 1, mf) != 1)
423aed3576dSKirk McKusick 		fail_dirtmp(modefile);
424772ad651SKirk McKusick }
425772ad651SKirk McKusick 
426772ad651SKirk McKusick /*
4278fae3551SRodney W. Grimes  * Seek to an entry in a directory.
4288fae3551SRodney W. Grimes  * Only values returned by rst_telldir should be passed to rst_seekdir.
4298fae3551SRodney W. Grimes  * This routine handles many directories in a single file.
4308fae3551SRodney W. Grimes  * It takes the base of the directory in the file, plus
4318fae3551SRodney W. Grimes  * the desired seek offset into it.
4328fae3551SRodney W. Grimes  */
4338fae3551SRodney W. Grimes static void
4342db673abSWarner Losh rst_seekdir(RST_DIR *dirp, long loc, long base)
4358fae3551SRodney W. Grimes {
4368fae3551SRodney W. Grimes 
4378fae3551SRodney W. Grimes 	if (loc == rst_telldir(dirp))
4388fae3551SRodney W. Grimes 		return;
4398fae3551SRodney W. Grimes 	loc -= base;
4408fae3551SRodney W. Grimes 	if (loc < 0)
4417fafc6f6SBruce Evans 		fprintf(stderr, "bad seek pointer to rst_seekdir %ld\n", loc);
4428fae3551SRodney W. Grimes 	(void) lseek(dirp->dd_fd, base + (loc & ~(DIRBLKSIZ - 1)), SEEK_SET);
4438fae3551SRodney W. Grimes 	dirp->dd_loc = loc & (DIRBLKSIZ - 1);
4448fae3551SRodney W. Grimes 	if (dirp->dd_loc != 0)
4458fae3551SRodney W. Grimes 		dirp->dd_size = read(dirp->dd_fd, dirp->dd_buf, DIRBLKSIZ);
4468fae3551SRodney W. Grimes }
4478fae3551SRodney W. Grimes 
4488fae3551SRodney W. Grimes /*
4498fae3551SRodney W. Grimes  * get next entry in a directory.
4508fae3551SRodney W. Grimes  */
4518fae3551SRodney W. Grimes struct direct *
4522db673abSWarner Losh rst_readdir(RST_DIR *dirp)
4538fae3551SRodney W. Grimes {
4543d438ad6SDavid E. O'Brien 	struct direct *dp;
4558fae3551SRodney W. Grimes 
4568fae3551SRodney W. Grimes 	for (;;) {
4578fae3551SRodney W. Grimes 		if (dirp->dd_loc == 0) {
4588fae3551SRodney W. Grimes 			dirp->dd_size = read(dirp->dd_fd, dirp->dd_buf,
4598fae3551SRodney W. Grimes 			    DIRBLKSIZ);
4608fae3551SRodney W. Grimes 			if (dirp->dd_size <= 0) {
4618fae3551SRodney W. Grimes 				dprintf(stderr, "error reading directory\n");
4628fae3551SRodney W. Grimes 				return (NULL);
4638fae3551SRodney W. Grimes 			}
4648fae3551SRodney W. Grimes 		}
4658fae3551SRodney W. Grimes 		if (dirp->dd_loc >= dirp->dd_size) {
4668fae3551SRodney W. Grimes 			dirp->dd_loc = 0;
4678fae3551SRodney W. Grimes 			continue;
4688fae3551SRodney W. Grimes 		}
4698fae3551SRodney W. Grimes 		dp = (struct direct *)(dirp->dd_buf + dirp->dd_loc);
4708fae3551SRodney W. Grimes 		if (dp->d_reclen == 0 ||
4718fae3551SRodney W. Grimes 		    dp->d_reclen > DIRBLKSIZ + 1 - dirp->dd_loc) {
4728fae3551SRodney W. Grimes 			dprintf(stderr, "corrupted directory: bad reclen %d\n",
4738fae3551SRodney W. Grimes 				dp->d_reclen);
4748fae3551SRodney W. Grimes 			return (NULL);
4758fae3551SRodney W. Grimes 		}
4768fae3551SRodney W. Grimes 		dirp->dd_loc += dp->d_reclen;
477d87d79aeSPeter Wemm 		if (dp->d_ino == 0 && strcmp(dp->d_name, "/") == 0)
478d87d79aeSPeter Wemm 			return (NULL);
4798fae3551SRodney W. Grimes 		if (dp->d_ino >= maxino) {
4808fae3551SRodney W. Grimes 			dprintf(stderr, "corrupted directory: bad inum %d\n",
4818fae3551SRodney W. Grimes 				dp->d_ino);
4828fae3551SRodney W. Grimes 			continue;
4838fae3551SRodney W. Grimes 		}
4848fae3551SRodney W. Grimes 		return (dp);
4858fae3551SRodney W. Grimes 	}
4868fae3551SRodney W. Grimes }
4878fae3551SRodney W. Grimes 
4888fae3551SRodney W. Grimes /*
4898fae3551SRodney W. Grimes  * Simulate the opening of a directory
4908fae3551SRodney W. Grimes  */
4911916772eSStefan Farfeleder void *
4922db673abSWarner Losh rst_opendir(const char *name)
4938fae3551SRodney W. Grimes {
4948fae3551SRodney W. Grimes 	struct inotab *itp;
4958fae3551SRodney W. Grimes 	RST_DIR *dirp;
4968fae3551SRodney W. Grimes 	ino_t ino;
4978fae3551SRodney W. Grimes 
4988fae3551SRodney W. Grimes 	if ((ino = dirlookup(name)) > 0 &&
4998fae3551SRodney W. Grimes 	    (itp = inotablookup(ino)) != NULL) {
5008fae3551SRodney W. Grimes 		dirp = opendirfile(dirfile);
5018fae3551SRodney W. Grimes 		rst_seekdir(dirp, itp->t_seekpt, itp->t_seekpt);
5028fae3551SRodney W. Grimes 		return (dirp);
5038fae3551SRodney W. Grimes 	}
5048fae3551SRodney W. Grimes 	return (NULL);
5058fae3551SRodney W. Grimes }
5068fae3551SRodney W. Grimes 
5078fae3551SRodney W. Grimes /*
5088fae3551SRodney W. Grimes  * In our case, there is nothing to do when closing a directory.
5098fae3551SRodney W. Grimes  */
5108fae3551SRodney W. Grimes void
5111916772eSStefan Farfeleder rst_closedir(void *arg)
5128fae3551SRodney W. Grimes {
5131916772eSStefan Farfeleder 	RST_DIR *dirp;
5148fae3551SRodney W. Grimes 
5151916772eSStefan Farfeleder 	dirp = arg;
5168fae3551SRodney W. Grimes 	(void)close(dirp->dd_fd);
5178fae3551SRodney W. Grimes 	free(dirp);
5188fae3551SRodney W. Grimes 	return;
5198fae3551SRodney W. Grimes }
5208fae3551SRodney W. Grimes 
5218fae3551SRodney W. Grimes /*
5228fae3551SRodney W. Grimes  * Simulate finding the current offset in the directory.
5238fae3551SRodney W. Grimes  */
5248fae3551SRodney W. Grimes static long
5252db673abSWarner Losh rst_telldir(RST_DIR *dirp)
5268fae3551SRodney W. Grimes {
5278fae3551SRodney W. Grimes 	return ((long)lseek(dirp->dd_fd,
5288fae3551SRodney W. Grimes 	    (off_t)0, SEEK_CUR) - dirp->dd_size + dirp->dd_loc);
5298fae3551SRodney W. Grimes }
5308fae3551SRodney W. Grimes 
5318fae3551SRodney W. Grimes /*
5328fae3551SRodney W. Grimes  * Open a directory file.
5338fae3551SRodney W. Grimes  */
5348fae3551SRodney W. Grimes static RST_DIR *
5352db673abSWarner Losh opendirfile(const char *name)
5368fae3551SRodney W. Grimes {
5373d438ad6SDavid E. O'Brien 	RST_DIR *dirp;
5383d438ad6SDavid E. O'Brien 	int fd;
5398fae3551SRodney W. Grimes 
5408fae3551SRodney W. Grimes 	if ((fd = open(name, O_RDONLY)) == -1)
5418fae3551SRodney W. Grimes 		return (NULL);
5428fae3551SRodney W. Grimes 	if ((dirp = malloc(sizeof(RST_DIR))) == NULL) {
5438fae3551SRodney W. Grimes 		(void)close(fd);
5448fae3551SRodney W. Grimes 		return (NULL);
5458fae3551SRodney W. Grimes 	}
5468fae3551SRodney W. Grimes 	dirp->dd_fd = fd;
5478fae3551SRodney W. Grimes 	dirp->dd_loc = 0;
5488fae3551SRodney W. Grimes 	return (dirp);
5498fae3551SRodney W. Grimes }
5508fae3551SRodney W. Grimes 
5518fae3551SRodney W. Grimes /*
5528fae3551SRodney W. Grimes  * Set the mode, owner, and times for all new or changed directories
5538fae3551SRodney W. Grimes  */
5548fae3551SRodney W. Grimes void
5552db673abSWarner Losh setdirmodes(int flags)
5568fae3551SRodney W. Grimes {
5578fae3551SRodney W. Grimes 	FILE *mf;
5588fae3551SRodney W. Grimes 	struct modeinfo node;
5598fae3551SRodney W. Grimes 	struct entry *ep;
560772ad651SKirk McKusick 	char *cp, *buf;
5614afde855SMaxim Sobolev 	const char *tmpdir;
562772ad651SKirk McKusick 	int bufsize;
563c028393dSKirk McKusick 	uid_t myuid;
5648fae3551SRodney W. Grimes 
5658fae3551SRodney W. Grimes 	vprintf(stdout, "Set directory mode, owner, and times.\n");
5664afde855SMaxim Sobolev 	if ((tmpdir = getenv("TMPDIR")) == NULL || tmpdir[0] == '\0')
5674afde855SMaxim Sobolev 		tmpdir = _PATH_TMP;
568335524b9SWarner Losh 	if (command == 'r' || command == 'R')
5694afde855SMaxim Sobolev 		(void) sprintf(modefile, "%s/rstmode%d", tmpdir, dumpdate);
570335524b9SWarner Losh 	if (modefile[0] == '#') {
571335524b9SWarner Losh 		panic("modefile not defined\n");
572335524b9SWarner Losh 		fprintf(stderr, "directory mode, owner, and times not set\n");
573335524b9SWarner Losh 		return;
574335524b9SWarner Losh 	}
5758fae3551SRodney W. Grimes 	mf = fopen(modefile, "r");
5768fae3551SRodney W. Grimes 	if (mf == NULL) {
5778fae3551SRodney W. Grimes 		fprintf(stderr, "fopen: %s\n", strerror(errno));
5788fae3551SRodney W. Grimes 		fprintf(stderr, "cannot open mode file %s\n", modefile);
5798fae3551SRodney W. Grimes 		fprintf(stderr, "directory mode, owner, and times not set\n");
5808fae3551SRodney W. Grimes 		return;
5818fae3551SRodney W. Grimes 	}
5828fae3551SRodney W. Grimes 	clearerr(mf);
583772ad651SKirk McKusick 	bufsize = 0;
584c028393dSKirk McKusick 	myuid = getuid();
5858fae3551SRodney W. Grimes 	for (;;) {
5868fae3551SRodney W. Grimes 		(void) fread((char *)&node, 1, sizeof(struct modeinfo), mf);
587aed3576dSKirk McKusick 		if (ferror(mf)) {
588aed3576dSKirk McKusick 			warn("%s: cannot read modefile.", modefile);
589aed3576dSKirk McKusick 			fprintf(stderr, "Mode, owner, and times not set.\n");
590aed3576dSKirk McKusick 			break;
591aed3576dSKirk McKusick 		}
5928fae3551SRodney W. Grimes 		if (feof(mf))
5938fae3551SRodney W. Grimes 			break;
594772ad651SKirk McKusick 		if (node.extsize > 0) {
595772ad651SKirk McKusick 			if (bufsize < node.extsize) {
596772ad651SKirk McKusick 				if (bufsize > 0)
597772ad651SKirk McKusick 					free(buf);
598772ad651SKirk McKusick 				if ((buf = malloc(node.extsize)) != 0) {
599772ad651SKirk McKusick 					bufsize = node.extsize;
600772ad651SKirk McKusick 				} else {
601772ad651SKirk McKusick 					bufsize = 0;
602772ad651SKirk McKusick 				}
603772ad651SKirk McKusick 			}
604772ad651SKirk McKusick 			if (bufsize >= node.extsize) {
605772ad651SKirk McKusick 				(void) fread(buf, 1, node.extsize, mf);
606aed3576dSKirk McKusick 				if (ferror(mf)) {
607aed3576dSKirk McKusick 					warn("%s: cannot read modefile.",
608aed3576dSKirk McKusick 					    modefile);
609aed3576dSKirk McKusick 					fprintf(stderr, "Not all external ");
610aed3576dSKirk McKusick 					fprintf(stderr, "attributes set.\n");
611aed3576dSKirk McKusick 					break;
612aed3576dSKirk McKusick 				}
613772ad651SKirk McKusick 			} else {
614772ad651SKirk McKusick 				(void) fseek(mf, node.extsize, SEEK_CUR);
615aed3576dSKirk McKusick 				if (ferror(mf)) {
616aed3576dSKirk McKusick 					warn("%s: cannot seek in modefile.",
617aed3576dSKirk McKusick 					    modefile);
618aed3576dSKirk McKusick 					fprintf(stderr, "Not all directory ");
619aed3576dSKirk McKusick 					fprintf(stderr, "attributes set.\n");
620aed3576dSKirk McKusick 					break;
621aed3576dSKirk McKusick 				}
622772ad651SKirk McKusick 			}
623772ad651SKirk McKusick 		}
6248fae3551SRodney W. Grimes 		ep = lookupino(node.ino);
6258fae3551SRodney W. Grimes 		if (command == 'i' || command == 'x') {
6268fae3551SRodney W. Grimes 			if (ep == NULL)
6278fae3551SRodney W. Grimes 				continue;
6288fae3551SRodney W. Grimes 			if ((flags & FORCE) == 0 && ep->e_flags & EXISTED) {
6298fae3551SRodney W. Grimes 				ep->e_flags &= ~NEW;
6308fae3551SRodney W. Grimes 				continue;
6318fae3551SRodney W. Grimes 			}
6328fae3551SRodney W. Grimes 			if (node.ino == ROOTINO &&
6338fae3551SRodney W. Grimes 		   	    reply("set owner/mode for '.'") == FAIL)
6348fae3551SRodney W. Grimes 				continue;
6358fae3551SRodney W. Grimes 		}
6368fae3551SRodney W. Grimes 		if (ep == NULL) {
6378fae3551SRodney W. Grimes 			panic("cannot find directory inode %d\n", node.ino);
638772ad651SKirk McKusick 			continue;
639772ad651SKirk McKusick 		}
6408fae3551SRodney W. Grimes 		cp = myname(ep);
641b9595aa5SDima Dorfman 		if (!Nflag) {
642772ad651SKirk McKusick 			if (node.extsize > 0) {
643772ad651SKirk McKusick 				if (bufsize >= node.extsize) {
644772ad651SKirk McKusick 					set_extattr_file(cp, buf, node.extsize);
645772ad651SKirk McKusick 				} else {
646772ad651SKirk McKusick 					fprintf(stderr, "Cannot restore %s%s\n",
647772ad651SKirk McKusick 					    "extended attributes for ", cp);
648772ad651SKirk McKusick 				}
649772ad651SKirk McKusick 			}
650c028393dSKirk McKusick 			if (myuid != 0)
651c028393dSKirk McKusick 				(void) chown(cp, myuid, node.gid);
652c028393dSKirk McKusick 			else
6538fae3551SRodney W. Grimes 				(void) chown(cp, node.uid, node.gid);
6548fae3551SRodney W. Grimes 			(void) chmod(cp, node.mode);
655fb36a3d8SKirk McKusick 			utimes(cp, node.ctimep);
656fb36a3d8SKirk McKusick 			utimes(cp, node.mtimep);
657fddde8b0SDavid Malone 			(void) chflags(cp, node.flags);
658b9595aa5SDima Dorfman 		}
6598fae3551SRodney W. Grimes 		ep->e_flags &= ~NEW;
6608fae3551SRodney W. Grimes 	}
661772ad651SKirk McKusick 	if (bufsize > 0)
662772ad651SKirk McKusick 		free(buf);
6638fae3551SRodney W. Grimes 	(void) fclose(mf);
6648fae3551SRodney W. Grimes }
6658fae3551SRodney W. Grimes 
6668fae3551SRodney W. Grimes /*
6678fae3551SRodney W. Grimes  * Generate a literal copy of a directory.
6688fae3551SRodney W. Grimes  */
6698fae3551SRodney W. Grimes int
6702db673abSWarner Losh genliteraldir(char *name, ino_t ino)
6718fae3551SRodney W. Grimes {
6723d438ad6SDavid E. O'Brien 	struct inotab *itp;
6738fae3551SRodney W. Grimes 	int ofile, dp, i, size;
6748fae3551SRodney W. Grimes 	char buf[BUFSIZ];
6758fae3551SRodney W. Grimes 
6768fae3551SRodney W. Grimes 	itp = inotablookup(ino);
6778fae3551SRodney W. Grimes 	if (itp == NULL)
6788fae3551SRodney W. Grimes 		panic("Cannot find directory inode %d named %s\n", ino, name);
679335524b9SWarner Losh 	if ((ofile = open(name, O_WRONLY | O_CREAT | O_TRUNC, 0666)) < 0) {
6808fae3551SRodney W. Grimes 		fprintf(stderr, "%s: ", name);
6818fae3551SRodney W. Grimes 		(void) fflush(stderr);
6828fae3551SRodney W. Grimes 		fprintf(stderr, "cannot create file: %s\n", strerror(errno));
6838fae3551SRodney W. Grimes 		return (FAIL);
6848fae3551SRodney W. Grimes 	}
6858fae3551SRodney W. Grimes 	rst_seekdir(dirp, itp->t_seekpt, itp->t_seekpt);
6868fae3551SRodney W. Grimes 	dp = dup(dirp->dd_fd);
6878fae3551SRodney W. Grimes 	for (i = itp->t_size; i > 0; i -= BUFSIZ) {
6888fae3551SRodney W. Grimes 		size = i < BUFSIZ ? i : BUFSIZ;
6898fae3551SRodney W. Grimes 		if (read(dp, buf, (int) size) == -1) {
6908fae3551SRodney W. Grimes 			fprintf(stderr,
6918fae3551SRodney W. Grimes 				"write error extracting inode %d, name %s\n",
6928fae3551SRodney W. Grimes 				curfile.ino, curfile.name);
6938fae3551SRodney W. Grimes 			fprintf(stderr, "read: %s\n", strerror(errno));
6948fae3551SRodney W. Grimes 			done(1);
6958fae3551SRodney W. Grimes 		}
6968fae3551SRodney W. Grimes 		if (!Nflag && write(ofile, buf, (int) size) == -1) {
6978fae3551SRodney W. Grimes 			fprintf(stderr,
6988fae3551SRodney W. Grimes 				"write error extracting inode %d, name %s\n",
6998fae3551SRodney W. Grimes 				curfile.ino, curfile.name);
7008fae3551SRodney W. Grimes 			fprintf(stderr, "write: %s\n", strerror(errno));
7018fae3551SRodney W. Grimes 			done(1);
7028fae3551SRodney W. Grimes 		}
7038fae3551SRodney W. Grimes 	}
7048fae3551SRodney W. Grimes 	(void) close(dp);
7058fae3551SRodney W. Grimes 	(void) close(ofile);
7068fae3551SRodney W. Grimes 	return (GOOD);
7078fae3551SRodney W. Grimes }
7088fae3551SRodney W. Grimes 
7098fae3551SRodney W. Grimes /*
7108fae3551SRodney W. Grimes  * Determine the type of an inode
7118fae3551SRodney W. Grimes  */
7128fae3551SRodney W. Grimes int
7132db673abSWarner Losh inodetype(ino_t ino)
7148fae3551SRodney W. Grimes {
7158fae3551SRodney W. Grimes 	struct inotab *itp;
7168fae3551SRodney W. Grimes 
7178fae3551SRodney W. Grimes 	itp = inotablookup(ino);
7188fae3551SRodney W. Grimes 	if (itp == NULL)
7198fae3551SRodney W. Grimes 		return (LEAF);
7208fae3551SRodney W. Grimes 	return (NODE);
7218fae3551SRodney W. Grimes }
7228fae3551SRodney W. Grimes 
7238fae3551SRodney W. Grimes /*
7248fae3551SRodney W. Grimes  * Allocate and initialize a directory inode entry.
7258fae3551SRodney W. Grimes  * If requested, save its pertinent mode, owner, and time info.
7268fae3551SRodney W. Grimes  */
7278fae3551SRodney W. Grimes static struct inotab *
7281c85e6a3SKirk McKusick allocinotab(struct context *ctxp, long seekpt)
7298fae3551SRodney W. Grimes {
7303d438ad6SDavid E. O'Brien 	struct inotab	*itp;
7318fae3551SRodney W. Grimes 	struct modeinfo node;
7328fae3551SRodney W. Grimes 
7338fae3551SRodney W. Grimes 	itp = calloc(1, sizeof(struct inotab));
7348fae3551SRodney W. Grimes 	if (itp == NULL)
735772ad651SKirk McKusick 		panic("no memory for directory table\n");
7361c85e6a3SKirk McKusick 	itp->t_next = inotab[INOHASH(ctxp->ino)];
7371c85e6a3SKirk McKusick 	inotab[INOHASH(ctxp->ino)] = itp;
7381c85e6a3SKirk McKusick 	itp->t_ino = ctxp->ino;
7398fae3551SRodney W. Grimes 	itp->t_seekpt = seekpt;
7408fae3551SRodney W. Grimes 	if (mf == NULL)
7418fae3551SRodney W. Grimes 		return (itp);
7421c85e6a3SKirk McKusick 	node.ino = ctxp->ino;
743fb36a3d8SKirk McKusick 	node.mtimep[0].tv_sec = ctxp->atime_sec;
744fb36a3d8SKirk McKusick 	node.mtimep[0].tv_usec = ctxp->atime_nsec / 1000;
745fb36a3d8SKirk McKusick 	node.mtimep[1].tv_sec = ctxp->mtime_sec;
746fb36a3d8SKirk McKusick 	node.mtimep[1].tv_usec = ctxp->mtime_nsec / 1000;
747fb36a3d8SKirk McKusick 	node.ctimep[0].tv_sec = ctxp->atime_sec;
748fb36a3d8SKirk McKusick 	node.ctimep[0].tv_usec = ctxp->atime_nsec / 1000;
749fb36a3d8SKirk McKusick 	node.ctimep[1].tv_sec = ctxp->birthtime_sec;
750fb36a3d8SKirk McKusick 	node.ctimep[1].tv_usec = ctxp->birthtime_nsec / 1000;
751772ad651SKirk McKusick 	node.extsize = ctxp->extsize;
7521c85e6a3SKirk McKusick 	node.mode = ctxp->mode;
7531c85e6a3SKirk McKusick 	node.flags = ctxp->file_flags;
7541c85e6a3SKirk McKusick 	node.uid = ctxp->uid;
7551c85e6a3SKirk McKusick 	node.gid = ctxp->gid;
756aed3576dSKirk McKusick 	if (fwrite((char *)&node, sizeof(struct modeinfo), 1, mf) != 1)
757aed3576dSKirk McKusick 		fail_dirtmp(modefile);
7588fae3551SRodney W. Grimes 	return (itp);
7598fae3551SRodney W. Grimes }
7608fae3551SRodney W. Grimes 
7618fae3551SRodney W. Grimes /*
7628fae3551SRodney W. Grimes  * Look up an inode in the table of directories
7638fae3551SRodney W. Grimes  */
7648fae3551SRodney W. Grimes static struct inotab *
7652db673abSWarner Losh inotablookup(ino_t ino)
7668fae3551SRodney W. Grimes {
7673d438ad6SDavid E. O'Brien 	struct inotab *itp;
7688fae3551SRodney W. Grimes 
7698fae3551SRodney W. Grimes 	for (itp = inotab[INOHASH(ino)]; itp != NULL; itp = itp->t_next)
7708fae3551SRodney W. Grimes 		if (itp->t_ino == ino)
7718fae3551SRodney W. Grimes 			return (itp);
7728fae3551SRodney W. Grimes 	return (NULL);
7738fae3551SRodney W. Grimes }
7748fae3551SRodney W. Grimes 
7758fae3551SRodney W. Grimes /*
7768fae3551SRodney W. Grimes  * Clean up and exit
7778fae3551SRodney W. Grimes  */
778eaa86f9dSBruce Evans void
7792db673abSWarner Losh done(int exitcode)
7808fae3551SRodney W. Grimes {
7818fae3551SRodney W. Grimes 
7828fae3551SRodney W. Grimes 	closemt();
783aed3576dSKirk McKusick 	if (modefile[0] != '#') {
784aed3576dSKirk McKusick 		(void) truncate(modefile, 0);
7858fae3551SRodney W. Grimes 		(void) unlink(modefile);
786aed3576dSKirk McKusick 	}
787aed3576dSKirk McKusick 	if (dirfile[0] != '#') {
788aed3576dSKirk McKusick 		(void) truncate(dirfile, 0);
7898fae3551SRodney W. Grimes 		(void) unlink(dirfile);
790aed3576dSKirk McKusick 	}
7918fae3551SRodney W. Grimes 	exit(exitcode);
7928fae3551SRodney W. Grimes }
793aed3576dSKirk McKusick 
794aed3576dSKirk McKusick /*
795aed3576dSKirk McKusick  * Print out information about the failure to save directory,
796aed3576dSKirk McKusick  * extended attribute, and mode information.
797aed3576dSKirk McKusick  */
798aed3576dSKirk McKusick static void
799aed3576dSKirk McKusick fail_dirtmp(char *filename)
800aed3576dSKirk McKusick {
801aed3576dSKirk McKusick 	const char *tmpdir;
802aed3576dSKirk McKusick 
803aed3576dSKirk McKusick 	warn("%s: cannot write directory database", filename);
804aed3576dSKirk McKusick 	if (errno == ENOSPC) {
805aed3576dSKirk McKusick 		if ((tmpdir = getenv("TMPDIR")) == NULL || tmpdir[0] == '\0')
806aed3576dSKirk McKusick 			tmpdir = _PATH_TMP;
807aed3576dSKirk McKusick 		fprintf(stderr, "Try making space in %s, %s\n%s\n", tmpdir,
808aed3576dSKirk McKusick 		    "or set environment variable TMPDIR",
809aed3576dSKirk McKusick 		    "to an alternate location with more disk space.");
810aed3576dSKirk McKusick 	}
811aed3576dSKirk McKusick 	done(1);
812aed3576dSKirk McKusick }
813