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