xref: /titanic_44/usr/src/cmd/backup/restore/dirs.c (revision fe0e7ec4d916b05b52d8c7cc8a3e6a1b28e77b6f)
1 /*
2  * Copyright 2005 Sun Microsystems, Inc.  All rights reserved.
3  * Use is subject to license terms.
4  */
5 
6 /*	Copyright (c) 1984, 1986, 1987, 1988, 1989 AT&T	*/
7 /*	  All Rights Reserved	*/
8 
9 /*
10  * Copyright (c) 1983 Regents of the University of California.
11  * All rights reserved.  The Berkeley software License Agreement
12  * specifies the terms and conditions for redistribution.
13  */
14 
15 #pragma ident	"%Z%%M%	%I%	%E% SMI"
16 
17 #include "restore.h"
18 #include <byteorder.h>
19 #include <stdlib.h>
20 #include <unistd.h>
21 #include <utime.h>
22 
23 /*
24  * Symbol table of directories read from tape.
25  */
26 #define	HASHSIZE	1000
27 #define	INOHASH(val) (val % HASHSIZE)
28 struct inotab {
29 	struct inotab *t_next;
30 	ino_t	t_ino;
31 	offset_t t_seekpt;
32 	offset_t t_size;
33 	struct inotab *t_xattr;
34 };
35 static struct inotab *inotab[HASHSIZE];
36 static struct inotab *xattrlist = NULL;
37 
38 /*
39  * Information retained about directories.
40  */
41 static struct modeinfo {
42 	ino_t	ino;
43 	time_t	timep[2];
44 	mode_t	mode;
45 	uid_t	uid;
46 	gid_t	gid;
47 	size_t	metasize;
48 } node;
49 
50 /*
51  * Global variables for this file.
52  */
53 static off64_t	g_seekpt;		/* some people have a local seekpt */
54 static FILE	*df, *mf;
55 static char	dirfile[MAXPATHLEN] = "#";	/* No file */
56 static char	modefile[MAXPATHLEN] = "#";	/* No file */
57 
58 static RST_DIR	*dirp;
59 
60 #define	INIT_TEMPFILE(name, type) \
61 	if (name[0] == '#') { \
62 		if (tmpdir == (char *)NULL) /* can't happen; be paranoid */ \
63 			tmpdir = "/tmp"; \
64 		(void) snprintf(name, sizeof (name), \
65 		    "%s/rst" type "%ld.XXXXXX", tmpdir, dumpdate); \
66 		(void) mktemp(name); \
67 	}
68 
69 #define	INIT_DIRFILE()	INIT_TEMPFILE(dirfile, "dir")
70 #define	INIT_MODEFILE()	INIT_TEMPFILE(modefile, "mode")
71 
72 /*
73  * Format of old style directories.
74  */
75 #define	ODIRSIZ 14
76 struct odirect {
77 	ushort_t d_ino;
78 	char	d_name[ODIRSIZ];
79 };
80 
81 #ifdef __STDC__
82 static ino_t search(ino_t, char	*);
83 static void putdir(char *, size_t);
84 static void putent(struct direct *);
85 static void skipmetadata(FILE *, size_t);
86 static void flushent(void);
87 static void dcvt(struct odirect *, struct direct *);
88 static RST_DIR *rst_initdirfile(char *);
89 static offset_t rst_telldir(RST_DIR *);
90 static void rst_seekdir(RST_DIR *, offset_t, offset_t);
91 static struct inotab *allocinotab(ino_t, struct dinode *, off64_t);
92 static void nodeflush(void);
93 static struct inotab *inotablookup(ino_t);
94 #else
95 static ino_t search();
96 static void putdir();
97 static void putent();
98 static void skipmetadata();
99 static void flushent();
100 static void dcvt();
101 static RST_DIR *rst_initdirfile();
102 static offset_t rst_telldir();
103 static void rst_seekdir();
104 static struct inotab *allocinotab();
105 static void nodeflush();
106 static struct inotab *inotablookup();
107 #endif
108 
109 /*
110  *	Extract directory contents, building up a directory structure
111  *	on disk for extraction by name.
112  *	If genmode is requested, save mode, owner, and times for all
113  *	directories on the tape.
114  */
115 void
extractdirs(int genmode)116 extractdirs(int genmode)
117 {
118 	int ts;
119 	struct dinode *ip;
120 	int saverr;
121 	struct inotab *itp;
122 	struct direct nulldir;
123 	static char dotname[] = "."; /* dirlookup/psearch writes to its arg */
124 
125 	vprintf(stdout, gettext("Extract directories from tape\n"));
126 	INIT_DIRFILE();
127 	if ((df = safe_fopen(dirfile, "w", 0600)) == (FILE *)NULL) {
128 		saverr = errno;
129 		(void) fprintf(stderr,
130 		    gettext("%s: %s - cannot create directory temporary\n"),
131 			progname, dirfile);
132 		errno = saverr;
133 		perror("fopen");
134 		done(1);
135 	}
136 	if (genmode != 0) {
137 		INIT_MODEFILE();
138 		if ((mf = safe_fopen(modefile, "w", 0600)) == (FILE *)NULL) {
139 			saverr = errno;
140 			(void) fprintf(stderr,
141 			    gettext("%s: %s - cannot create modefile \n"),
142 				progname, modefile);
143 			errno = saverr;
144 			perror("fopen");
145 			done(1);
146 		}
147 	}
148 	nulldir.d_ino = 0;
149 	nulldir.d_namlen = 1;
150 	(void) strcpy(nulldir.d_name, "/");
151 	/* LINTED DIRSIZ will always fit into a ushort_t */
152 	nulldir.d_reclen = (ushort_t)DIRSIZ(&nulldir);
153 	/* LINTED sign extension ok in assert */
154 	assert(DIRSIZ(&nulldir) == (ulong_t)nulldir.d_reclen);
155 	for (;;) {
156 		curfile.name = gettext("<directory file - name unknown>");
157 		curfile.action = USING;
158 		ip = curfile.dip;
159 		ts = curfile.ts;
160 		if (ts != TS_END && ts != TS_INODE) {
161 			getfile(null, null);
162 			continue;
163 		}
164 		if (ts == TS_INODE && ip == NULL) {
165 			(void) fprintf(stderr, gettext(
166 "%s: extractdirs: Failed internal consistency check, curfile.dip is NULL\n"),
167 			    progname);
168 			done(1);
169 		}
170 		if ((ts == TS_INODE && (ip->di_mode & IFMT) != IFDIR &&
171 		    (ip->di_mode & IFMT) != IFATTRDIR) ||
172 		    (ts == TS_END)) {
173 			(void) fflush(df);
174 			/* XXX Legitimate error, bad complaint string */
175 			if (ferror(df))
176 				panic("%s: %s\n", dirfile, strerror(errno));
177 			(void) fclose(df);
178 			rst_closedir(dirp);
179 			dirp = rst_initdirfile(dirfile);
180 			if (dirp == NULL)
181 				perror("initdirfile");
182 			if (mf != NULL) {
183 				(void) fflush(mf);
184 				/* XXX Legitimate error, bad complaint string */
185 				if (ferror(mf))
186 					panic("%s: %s\n",
187 					    modefile, strerror(errno));
188 				(void) fclose(mf);
189 			}
190 			if (dirlookup(dotname) == 0) {
191 				(void) fprintf(stderr, gettext(
192 				    "Root directory is not on tape\n"));
193 				done(1);
194 			}
195 			return;
196 		}
197 		itp = allocinotab(curfile.ino, ip, g_seekpt);
198 		getfile(putdir, null);
199 		if (mf != NULL)
200 			nodeflush();
201 
202 		putent(&nulldir);
203 		flushent();
204 		itp->t_size = g_seekpt - itp->t_seekpt;
205 	}
206 }
207 
208 /*
209  * skip over all the directories on the tape
210  */
211 void
skipdirs()212 skipdirs()
213 {
214 	while (curfile.dip != NULL &&
215 		((curfile.dip->di_mode & IFMT) == IFDIR ||
216 		(curfile.dip->di_mode & IFMT) == IFATTRDIR)) {
217 		skipfile();
218 	}
219 }
220 
221 /*
222  *	Recursively find names and inumbers of all files in subtree
223  *	pname and pass them off to be processed.
224  */
225 void
treescan(char * pname,ino_t ino,long (* todo)())226 treescan(char *pname, ino_t ino, long (*todo)())
227 {
228 	struct inotab *itp;
229 	struct direct *dp;
230 	uint_t loclen;
231 	offset_t bpt;
232 	char locname[MAXCOMPLEXLEN];
233 
234 	itp = inotablookup(ino);
235 	if (itp == NULL) {
236 		/*
237 		 * Pname is name of a simple file or an unchanged directory.
238 		 */
239 		(void) (*todo)(pname, ino, LEAF);
240 		return;
241 	}
242 	/*
243 	 * Pname is a dumped directory name.
244 	 */
245 	if ((*todo)(pname, ino, NODE) == FAIL)
246 		return;
247 	/*
248 	 * begin search through the directory
249 	 * skipping over "." and ".."
250 	 */
251 	loclen = complexcpy(locname, pname, MAXCOMPLEXLEN);
252 	locname[loclen-1] = '/';
253 	rst_seekdir(dirp, itp->t_seekpt, itp->t_seekpt);
254 	dp = rst_readdir(dirp); /* "." */
255 
256 	if (dp != NULL && strcmp(dp->d_name, ".") == 0)
257 		dp = rst_readdir(dirp); /* ".." */
258 	else
259 		(void) fprintf(stderr,
260 		    gettext("Warning: `.' missing from directory %s\n"),
261 			pname);
262 	if (dp != NULL && strcmp(dp->d_name, "..") == 0)
263 		dp = rst_readdir(dirp); /* first real entry */
264 	else
265 		(void) fprintf(stderr,
266 		    gettext("Warning: `..' missing from directory %s\n"),
267 			pname);
268 	bpt = rst_telldir(dirp);
269 	/*
270 	 * a zero inode signals end of directory
271 	 */
272 	while (dp != NULL && dp->d_ino != 0) {
273 		locname[loclen] = '\0';
274 		if ((loclen + dp->d_namlen) >= (sizeof (locname) - 2)) {
275 			(void) fprintf(stderr,
276 			    gettext(
277 				"%s%s: ignoring name that exceeds %d char\n"),
278 			    locname, dp->d_name, MAXCOMPLEXLEN);
279 		} else {
280 			/* Always fits by if() condition */
281 			(void) strcpy(locname + loclen, dp->d_name);
282 			/* put a double null on string for lookupname() */
283 			locname[loclen+dp->d_namlen+1] = '\0';
284 			treescan(locname, dp->d_ino, todo);
285 			rst_seekdir(dirp, bpt, itp->t_seekpt);
286 		}
287 		dp = rst_readdir(dirp);
288 		bpt = rst_telldir(dirp);
289 	}
290 	if (dp == NULL)
291 		(void) fprintf(stderr,
292 			gettext("corrupted directory: %s.\n"), locname);
293 }
294 
295 /*
296  * Scan the directory table looking for extended attribute trees.
297  * Recursively find names and inumbers in each tree and pass them
298  * off to be processed.  If the always parameter is not set, only
299  * process the attribute tree if the attribute tree parent is to
300  * be extracted.
301  */
302 void
attrscan(int always,long (* todo)())303 attrscan(int always, long (*todo)())
304 {
305 	struct inotab *itp;
306 	struct entry *ep, *parent;
307 	struct direct *dp;
308 	char name[MAXCOMPLEXLEN];
309 	int len;
310 
311 	for (itp = xattrlist; itp != NULL; itp = itp->t_xattr) {
312 		rst_seekdir(dirp, itp->t_seekpt, itp->t_seekpt);
313 		if ((dp = rst_readdir(dirp)) != NULL &&	/* "." */
314 		    (dp = rst_readdir(dirp)) != NULL &&	/* ".." */
315 		    strcmp(dp->d_name, "..") == 0) {
316 			if ((parent = lookupino(dp->d_ino)) != NULL) {
317 				if (!always &&
318 				    (parent->e_flags & (NEW|EXTRACT)) == 0)
319 					continue;
320 				len = complexcpy(name, myname(parent),
321 							MAXCOMPLEXLEN - 3);
322 				name[len] = '.';
323 				name[len+1] = '\0';
324 				name[len+2] = '\0';
325 				inattrspace = 1;
326 				if ((ep = lookupino(itp->t_ino)) == NULL) {
327 					ep = addentry(name, itp->t_ino,
328 								NODE|ROOT);
329 				}
330 				ep->e_flags |= XATTRROOT;
331 				treescan(name, itp->t_ino, todo);
332 				inattrspace = 0;
333 			} else {
334 				(void) fprintf(stderr,
335 			gettext("Warning: orphaned attribute directory\n"));
336 			}
337 		} else {
338 			(void) fprintf(stderr,
339 	    gettext("Warning: `..' missing from attribute directory\n"));
340 		}
341 	}
342 }
343 
344 /*
345  * Search the directory tree rooted at inode ROOTINO
346  * for the path pointed at by n.  Note that n must be
347  * modifiable, although it is returned in the same
348  * condition it was given to us in.
349  */
350 ino_t
psearch(char * n)351 psearch(char *n)
352 {
353 	char *cp, *cp1;
354 	ino_t ino;
355 	char c;
356 
357 	ino = ROOTINO;
358 	if (*(cp = n) == '/')
359 		cp++;
360 next:
361 	cp1 = cp + 1;
362 	while (*cp1 != '/' && *cp1)
363 		cp1++;
364 	c = *cp1;
365 	*cp1 = 0;
366 	ino = search(ino, cp);
367 	if (ino == 0) {
368 		*cp1 = c;
369 		return (0);
370 	}
371 	*cp1 = c;
372 	if (c == '/') {
373 		cp = cp1+1;
374 		goto next;
375 	}
376 	return (ino);
377 }
378 
379 /*
380  * search the directory inode ino
381  * looking for entry cp
382  */
383 static ino_t
search(ino_t inum,char * cp)384 search(ino_t inum, char *cp)
385 {
386 	struct direct *dp;
387 	struct inotab *itp;
388 	uint_t len;
389 
390 	itp = inotablookup(inum);
391 	if (itp == NULL)
392 		return (0);
393 	rst_seekdir(dirp, itp->t_seekpt, itp->t_seekpt);
394 	len = strlen(cp);
395 	do {
396 		dp = rst_readdir(dirp);
397 		if (dp == NULL || dp->d_ino == 0)
398 			return (0);
399 	} while (dp->d_namlen != len || strncmp(dp->d_name, cp, len) != 0);
400 	return (dp->d_ino);
401 }
402 
403 /*
404  * Put the directory entries in the directory file
405  */
406 static void
putdir(char * buf,size_t size)407 putdir(char *buf, size_t size)
408 {
409 	struct direct cvtbuf;
410 	struct odirect *odp;
411 	struct odirect *eodp;
412 	struct direct *dp;
413 	size_t loc, i;
414 
415 	if (cvtflag) {
416 		/*LINTED [buf is char[] in getfile, size % fs_fsize == 0]*/
417 		eodp = (struct odirect *)&buf[size];
418 		/*LINTED [buf is char[] in getfile]*/
419 		for (odp = (struct odirect *)buf; odp < eodp; odp++)
420 			if (odp->d_ino != 0) {
421 				dcvt(odp, &cvtbuf);
422 				putent(&cvtbuf);
423 			}
424 	} else {
425 		loc = 0;
426 		while (loc < size) {
427 			/*LINTED [buf is char[] in getfile, loc % 4 == 0]*/
428 			dp = (struct direct *)(buf + loc);
429 			normdirect(byteorder, dp);
430 			i = DIRBLKSIZ - (loc & (DIRBLKSIZ - 1));
431 			if (dp->d_reclen == 0 || (long)dp->d_reclen > i) {
432 				loc += i;
433 				continue;
434 			}
435 			loc += dp->d_reclen;
436 			if (dp->d_ino != 0) {
437 				putent(dp);
438 			}
439 		}
440 	}
441 }
442 
443 /*
444  * These variables are "local" to the following two functions.
445  */
446 static char dirbuf[DIRBLKSIZ];
447 static int32_t dirloc = 0;
448 static int32_t prev = 0;
449 
450 /*
451  * add a new directory entry to a file.
452  */
453 static void
putent(struct direct * dp)454 putent(struct direct *dp)
455 {
456 	/* LINTED DIRSIZ will always fit in a ushort_t */
457 	dp->d_reclen = (ushort_t)DIRSIZ(dp);
458 	/* LINTED sign extension ok in assert */
459 	assert(DIRSIZ(dp) == (ulong_t)dp->d_reclen);
460 	if (dirloc + (long)dp->d_reclen > DIRBLKSIZ) {
461 		/*LINTED [prev += dp->d_reclen, prev % 4 == 0]*/
462 		((struct direct *)(dirbuf + prev))->d_reclen =
463 		    DIRBLKSIZ - prev;
464 		(void) fwrite(dirbuf, 1, DIRBLKSIZ, df);
465 		if (ferror(df))
466 			panic("%s: %s\n", dirfile, strerror(errno));
467 		dirloc = 0;
468 	}
469 	bcopy((char *)dp, dirbuf + dirloc, (size_t)dp->d_reclen);
470 	prev = dirloc;
471 	dirloc += dp->d_reclen;
472 }
473 
474 /*
475  * flush out a directory that is finished.
476  */
477 static void
478 #ifdef __STDC__
flushent(void)479 flushent(void)
480 #else
481 flushent()
482 #endif
483 {
484 
485 	/* LINTED prev += dp->d_reclen, prev % 4 == 0 */
486 	((struct direct *)(dirbuf + prev))->d_reclen = DIRBLKSIZ - prev;
487 	(void) fwrite(dirbuf, (size_t)dirloc, 1, df);
488 	if (ferror(df))
489 		panic("%s: %s\n", dirfile, strerror(errno));
490 	g_seekpt = ftello64(df);
491 	dirloc = 0;
492 }
493 
494 static void
dcvt(struct odirect * odp,struct direct * ndp)495 dcvt(struct odirect *odp, struct direct *ndp)
496 {
497 
498 	(void) bzero((char *)ndp, sizeof (*ndp));
499 	ndp->d_ino =  odp->d_ino;
500 	/* Note that odp->d_name may not be null-terminated */
501 	/* LINTED assertion always true */
502 	assert(sizeof (ndp->d_name) > sizeof (odp->d_name));
503 	(void) strncpy(ndp->d_name, odp->d_name, sizeof (odp->d_name));
504 	ndp->d_name[sizeof (odp->d_name)] = '\0';
505 	/* LINTED: strlen will fit into d_namlen */
506 	ndp->d_namlen = strlen(ndp->d_name);
507 
508 	/* LINTED sign extension ok in assert */
509 	assert(DIRSIZ(ndp) == (ulong_t)ndp->d_reclen);
510 	/* LINTED DIRSIZ always fits in ushort_t */
511 	ndp->d_reclen = (ushort_t)DIRSIZ(ndp);
512 }
513 
514 /*
515  * Initialize the directory file
516  */
517 static RST_DIR *
rst_initdirfile(char * name)518 rst_initdirfile(char *name)
519 {
520 	RST_DIR *dp;
521 	int fd;
522 
523 	if ((fd = open(name, O_RDONLY | O_LARGEFILE)) == -1)
524 		return ((RST_DIR *)0);
525 	if ((dp = (RST_DIR *)malloc(sizeof (*dp))) == NULL) {
526 		(void) close(fd);
527 		return ((RST_DIR *)0);
528 	}
529 	dp->dd_fd = fd;
530 	dp->dd_loc = 0;
531 	dp->dd_refcnt = 1;
532 	return (dp);
533 }
534 
535 /*
536  * Simulate the opening of a directory
537  */
538 RST_DIR *
rst_opendir(char * name)539 rst_opendir(char *name)
540 {
541 	struct inotab *itp;
542 	ino_t ino;
543 
544 	if ((ino = dirlookup(name)) > 0 &&
545 	    (itp = inotablookup(ino)) != NULL) {
546 		rst_seekdir(dirp, itp->t_seekpt, itp->t_seekpt);
547 		dirp->dd_refcnt++;
548 		return (dirp);
549 	}
550 	return ((RST_DIR *)0);
551 }
552 
553 /*
554  * Releases the hidden state created by rst_opendir().
555  * Specifically, the dirp it provided to the caller is malloc'd.
556  */
557 void
rst_closedir(RST_DIR * cdirp)558 rst_closedir(RST_DIR *cdirp)
559 {
560 	if ((cdirp != NULL) && (--(cdirp->dd_refcnt) < 1))
561 		free(cdirp);
562 }
563 
564 /*
565  * return a pointer into a directory
566  */
567 static offset_t
rst_telldir(RST_DIR * tdirp)568 rst_telldir(RST_DIR *tdirp)
569 {
570 	offset_t pos = llseek(tdirp->dd_fd, (offset_t)0, SEEK_CUR);
571 
572 	if (pos == (offset_t)-1) {
573 		perror("Could not determine position in directory file");
574 		done(1);
575 	}
576 
577 	return ((pos - tdirp->dd_size) + tdirp->dd_loc);
578 }
579 
580 /*
581  * Seek to an entry in a directory.
582  * Only values returned by ``rst_telldir'' should be passed to rst_seekdir.
583  * This routine handles many directories in a single file.
584  * It takes the base of the directory in the file, plus
585  * the desired seek offset into it.
586  */
587 static void
rst_seekdir(RST_DIR * sdirp,offset_t loc,offset_t base)588 rst_seekdir(RST_DIR *sdirp, offset_t loc, offset_t base)
589 {
590 
591 	if (loc == rst_telldir(sdirp))
592 		return;
593 	loc -= base;
594 	if (loc < 0)
595 		(void) fprintf(stderr,
596 			gettext("bad seek pointer to rst_seekdir %d\n"), loc);
597 	(void) llseek(sdirp->dd_fd, base + (loc & ~(DIRBLKSIZ - 1)), 0);
598 	sdirp->dd_loc = loc & (DIRBLKSIZ - 1);
599 	if (sdirp->dd_loc != 0)
600 		sdirp->dd_size = read(sdirp->dd_fd, sdirp->dd_buf, DIRBLKSIZ);
601 }
602 
603 /*
604  * get next entry in a directory.
605  */
606 struct direct *
rst_readdir(RST_DIR * rdirp)607 rst_readdir(RST_DIR *rdirp)
608 {
609 	struct direct *dp;
610 
611 	for (;;) {
612 		if (rdirp->dd_loc == 0) {
613 			rdirp->dd_size = read(rdirp->dd_fd, rdirp->dd_buf,
614 			    DIRBLKSIZ);
615 			if (rdirp->dd_size <= 0) {
616 				dprintf(stderr,
617 					gettext("error reading directory\n"));
618 				return ((struct direct *)0);
619 			}
620 		}
621 		if (rdirp->dd_loc >= rdirp->dd_size) {
622 			rdirp->dd_loc = 0;
623 			continue;
624 		}
625 		/*LINTED [rvalue will be aligned on int boundary]*/
626 		dp = (struct direct *)(rdirp->dd_buf + rdirp->dd_loc);
627 		if (dp->d_reclen == 0 ||
628 		    (long)dp->d_reclen > (DIRBLKSIZ + 1 - rdirp->dd_loc)) {
629 			dprintf(stderr,
630 			    gettext("corrupted directory: bad reclen %d\n"),
631 				dp->d_reclen);
632 			return ((struct direct *)0);
633 		}
634 		rdirp->dd_loc += dp->d_reclen;
635 		if (dp->d_ino == 0 && strcmp(dp->d_name, "/") != 0)
636 			continue;
637 		if ((ino_t)(dp->d_ino) >= maxino) {
638 			dprintf(stderr,
639 				gettext("corrupted directory: bad inum %lu\n"),
640 				dp->d_ino);
641 			continue;
642 		}
643 		return (dp);
644 	}
645 }
646 
647 /*
648  * Set the mode, owner, and times for all new or changed directories
649  */
650 void
651 #ifdef __STDC__
setdirmodes(void)652 setdirmodes(void)
653 #else
654 setdirmodes()
655 #endif
656 {
657 	FILE *smf;
658 	struct entry *ep;
659 	char *cp, *metadata = NULL;
660 	size_t metasize = 0;
661 	int override = -1;
662 	int saverr;
663 	static int complained_chown = 0;
664 	static int complained_chmod = 0;
665 	int dfd;
666 
667 	vprintf(stdout, gettext("Set directory mode, owner, and times.\n"));
668 	/* XXX if modefile[0] == '#', shouldn't we just bail here? */
669 	/* XXX why isn't it set already? */
670 	INIT_MODEFILE();
671 	smf = fopen64(modefile, "r");
672 	if (smf == NULL) {
673 		perror("fopen");
674 		(void) fprintf(stderr,
675 			gettext("cannot open mode file %s\n"), modefile);
676 		(void) fprintf(stderr,
677 			gettext("directory mode, owner, and times not set\n"));
678 		return;
679 	}
680 	clearerr(smf);
681 	for (;;) {
682 		(void) fread((char *)&node, 1, sizeof (node), smf);
683 		if (feof(smf))
684 			break;
685 		ep = lookupino(node.ino);
686 		if (command == 'i' || command == 'x') {
687 			if (ep == NIL) {
688 				skipmetadata(smf, node.metasize);
689 				continue;
690 			}
691 			if (ep->e_flags & EXISTED) {
692 				if (override < 0) {
693 					if (reply(gettext(
694 				"Directories already exist, set modes anyway"))
695 					    == FAIL)
696 						override = 0;
697 					else
698 						override = 1;
699 				}
700 				if (override == 0) {
701 					/* LINTED: result fits into short */
702 					ep->e_flags &= ~NEW;
703 					skipmetadata(smf, node.metasize);
704 					continue;
705 				}
706 			}
707 			if (node.ino == ROOTINO &&
708 			    reply(gettext("set owner/mode for '.'")) == FAIL) {
709 				skipmetadata(smf, node.metasize);
710 				continue;
711 			}
712 		}
713 		if (ep == NIL) {
714 			panic(gettext("cannot find directory inode %d\n"),
715 				node.ino);
716 			skipmetadata(smf, node.metasize);
717 			continue;
718 		}
719 		cp = myname(ep);
720 		resolve(myname(ep), &dfd, &cp);
721 		if (dfd != AT_FDCWD) {
722 			if (fchdir(dfd) < 0) {
723 				saverr = errno;
724 				(void) fprintf(stderr,
725 			    gettext("Can not set attribute context: %s\n"),
726 					strerror(saverr));
727 				(void) close(dfd);
728 				continue;
729 			}
730 		}
731 		if (chmod(cp, node.mode) < 0 && !complained_chmod) {
732 			saverr = errno;
733 			(void) fprintf(stderr,
734 			gettext("Can not set directory permissions: %s\n"),
735 				strerror(saverr));
736 			complained_chmod = 1;
737 		}
738 		if (node.metasize != 0) {
739 			if (node.metasize > metasize)
740 				metadata = realloc(metadata,
741 				    metasize = node.metasize);
742 			if (metadata == NULL) {
743 				(void) fprintf(stderr,
744 					gettext("Cannot malloc metadata\n"));
745 				done(1);
746 			}
747 			(void) fread(metadata, 1, node.metasize, smf);
748 			metaproc(cp, metadata, node.metasize);
749 		}
750 
751 		/*
752 		 * BUG 4302943
753 		 * Since the ACLs must be set before fixing the ownership,
754 		 * chown should be called only after metaproc
755 		 */
756 		if (chown(cp, node.uid, node.gid) < 0 && !complained_chown) {
757 			saverr = errno;
758 			(void) fprintf(stderr,
759 			    gettext("Can not set directory ownership: %s\n"),
760 			    strerror(saverr));
761 			complained_chown = 1;
762 		}
763 		utime(cp, (struct utimbuf *)node.timep);
764 		/* LINTED: result fits into short */
765 		ep->e_flags &= ~NEW;
766 		if (dfd != AT_FDCWD) {
767 			fchdir(savepwd);
768 			(void) close(dfd);
769 		}
770 	}
771 	if (ferror(smf))
772 		panic(gettext("error setting directory modes\n"));
773 	if (metadata != NULL)
774 		(void) free(metadata);
775 	(void) fclose(smf);
776 }
777 
778 void
skipmetadata(FILE * f,size_t size)779 skipmetadata(FILE *f, size_t size)
780 {
781 	/* XXX should we bail if this doesn't work? */
782 	/* LINTED unsigned -> signed conversion ok here */
783 	(void) fseeko(f, (off_t)size, SEEK_CUR);
784 }
785 
786 /*
787  * Generate a literal copy of a directory.
788  */
789 int
genliteraldir(char * name,ino_t ino)790 genliteraldir(char *name, ino_t ino)
791 {
792 	struct inotab *itp;
793 	int ofile, dp;
794 	off64_t i;
795 	size_t size;
796 	char buf[BUFSIZ];
797 
798 	itp = inotablookup(ino);
799 	if (itp == NULL) {
800 		(void) fprintf(stderr,
801 		    gettext("Cannot find directory inode %d named %s\n"),
802 		    ino, name);
803 		return (FAIL);
804 	}
805 	if ((ofile = creat(name, 0666)) < 0) {
806 		(void) fprintf(stderr, "%s: ", name);
807 		(void) fflush(stderr);
808 		perror(gettext("cannot create file"));
809 		return (FAIL);
810 	}
811 	rst_seekdir(dirp, itp->t_seekpt, itp->t_seekpt);
812 	dp = dup(dirp->dd_fd);
813 	if (dp < 0) {
814 		perror(gettext("dup(2) failed"));
815 		(void) close(ofile);
816 		(void) unlink(name);
817 		return (FAIL);
818 	}
819 	for (i = itp->t_size; i != 0; i -= size) {
820 		/* LINTED cast is safe due to comparison */
821 		size = i < BUFSIZ ? (size_t)i : BUFSIZ;
822 		/* XXX instead of done(), clean up and return FAIL? */
823 		if (read(dp, buf, size) == -1) {
824 			(void) fprintf(stderr, gettext(
825 				"read error extracting inode %d, name %s\n"),
826 				curfile.ino, curfile.name);
827 			perror("read");
828 			done(1);
829 		}
830 		if (write(ofile, buf, size) == -1) {
831 			(void) fprintf(stderr, gettext(
832 				"write error extracting inode %d, name %s\n"),
833 				curfile.ino, curfile.name);
834 			perror("write");
835 			done(1);
836 		}
837 	}
838 	(void) close(dp);
839 	(void) close(ofile);
840 	return (GOOD);
841 }
842 
843 /*
844  * Determine the type of an inode
845  */
846 int
inodetype(ino_t ino)847 inodetype(ino_t ino)
848 {
849 	struct inotab *itp;
850 
851 	itp = inotablookup(ino);
852 	if (itp == NULL)
853 		return (LEAF);
854 	return (NODE);
855 }
856 
857 /*
858  * Allocate and initialize a directory inode entry.
859  * If requested, save its pertinent mode, owner, and time info.
860  */
861 static struct inotab *
allocinotab(ino_t ino,struct dinode * dip,off64_t seekpt)862 allocinotab(ino_t ino, struct dinode *dip, off64_t seekpt)
863 {
864 	struct inotab	*itp;
865 
866 	itp = (struct inotab *)calloc(1, sizeof (*itp));
867 	if (itp == 0) {
868 		(void) fprintf(stderr,
869 		    gettext("no memory for directory table\n"));
870 		done(1);
871 	}
872 	itp->t_next = inotab[INOHASH(ino)];
873 	inotab[INOHASH(ino)] = itp;
874 	itp->t_ino = ino;
875 	itp->t_seekpt = seekpt;
876 	if ((dip->di_mode & IFMT) == IFATTRDIR) {
877 		itp->t_xattr = xattrlist;
878 		xattrlist = itp;
879 	}
880 	if (mf == NULL)
881 		return (itp);
882 	node.ino = ino;
883 	node.timep[0] = dip->di_atime;
884 	node.timep[1] = dip->di_mtime;
885 	node.mode = dip->di_mode;
886 	node.uid =
887 		dip->di_suid == UID_LONG ? dip->di_uid : (uid_t)dip->di_suid;
888 	node.gid =
889 		dip->di_sgid == GID_LONG ? dip->di_gid : (gid_t)dip->di_sgid;
890 	return (itp);
891 }
892 
893 void
nodeflush()894 nodeflush()
895 {
896 	char *metadata;
897 
898 	if (mf == NULL) {
899 		(void) fprintf(stderr, gettext(
900 		    "Inconsistency detected: modefile pointer is NULL\n"));
901 		done(1);
902 	}
903 	metaget(&metadata, &(node.metasize));
904 	(void) fwrite((char *)&node, 1, sizeof (node), mf);
905 	if (node.metasize != 0)
906 		(void) fwrite(metadata, 1, node.metasize, mf);
907 	if (ferror(mf))
908 		panic("%s: %s\n", modefile, strerror(errno));
909 }
910 
911 /*
912  * Look up an inode in the table of directories
913  */
914 static struct inotab *
inotablookup(ino_t ino)915 inotablookup(ino_t ino)
916 {
917 	struct inotab *itp;
918 
919 	for (itp = inotab[INOHASH(ino)]; itp != NULL; itp = itp->t_next)
920 		if (itp->t_ino == ino)
921 			return (itp);
922 	return ((struct inotab *)0);
923 }
924 
925 /*
926  * Clean up and exit
927  */
928 void
done(int exitcode)929 done(int exitcode)
930 {
931 	closemt(ALLOW_OFFLINE);		/* don't force offline on exit */
932 	if (modefile[0] != '#')
933 		(void) unlink(modefile);
934 	if (dirfile[0] != '#')
935 		(void) unlink(dirfile);
936 	exit(exitcode);
937 }
938