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