walk.c revision 1.37 1 1.37 christos /* $NetBSD: walk.c,v 1.37 2024/04/24 14:02:39 christos Exp $ */
2 1.1 lukem
3 1.1 lukem /*
4 1.5 lukem * Copyright (c) 2001 Wasabi Systems, Inc.
5 1.1 lukem * All rights reserved.
6 1.1 lukem *
7 1.1 lukem * Written by Luke Mewburn for Wasabi Systems, Inc.
8 1.1 lukem *
9 1.1 lukem * Redistribution and use in source and binary forms, with or without
10 1.1 lukem * modification, are permitted provided that the following conditions
11 1.1 lukem * are met:
12 1.1 lukem * 1. Redistributions of source code must retain the above copyright
13 1.1 lukem * notice, this list of conditions and the following disclaimer.
14 1.1 lukem * 2. Redistributions in binary form must reproduce the above copyright
15 1.1 lukem * notice, this list of conditions and the following disclaimer in the
16 1.1 lukem * documentation and/or other materials provided with the distribution.
17 1.1 lukem * 3. All advertising materials mentioning features or use of this software
18 1.1 lukem * must display the following acknowledgement:
19 1.1 lukem * This product includes software developed for the NetBSD Project by
20 1.1 lukem * Wasabi Systems, Inc.
21 1.1 lukem * 4. The name of Wasabi Systems, Inc. may not be used to endorse
22 1.1 lukem * or promote products derived from this software without specific prior
23 1.1 lukem * written permission.
24 1.1 lukem *
25 1.1 lukem * THIS SOFTWARE IS PROVIDED BY WASABI SYSTEMS, INC. ``AS IS'' AND
26 1.1 lukem * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
27 1.1 lukem * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
28 1.1 lukem * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL WASABI SYSTEMS, INC
29 1.1 lukem * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
30 1.1 lukem * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
31 1.1 lukem * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
32 1.1 lukem * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
33 1.1 lukem * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
34 1.1 lukem * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
35 1.1 lukem * POSSIBILITY OF SUCH DAMAGE.
36 1.1 lukem */
37 1.1 lukem
38 1.17 jmc #if HAVE_NBTOOL_CONFIG_H
39 1.17 jmc #include "nbtool_config.h"
40 1.17 jmc #endif
41 1.17 jmc
42 1.2 lukem #include <sys/cdefs.h>
43 1.8 tv #if defined(__RCSID) && !defined(__lint)
44 1.37 christos __RCSID("$NetBSD: walk.c,v 1.37 2024/04/24 14:02:39 christos Exp $");
45 1.2 lukem #endif /* !__lint */
46 1.1 lukem
47 1.1 lukem #include <sys/param.h>
48 1.27 christos #include <sys/stat.h>
49 1.1 lukem
50 1.1 lukem #include <assert.h>
51 1.1 lukem #include <errno.h>
52 1.1 lukem #include <fcntl.h>
53 1.1 lukem #include <stdio.h>
54 1.1 lukem #include <dirent.h>
55 1.1 lukem #include <stdlib.h>
56 1.1 lukem #include <string.h>
57 1.1 lukem #include <unistd.h>
58 1.27 christos #include <util.h>
59 1.1 lukem
60 1.1 lukem #include "makefs.h"
61 1.1 lukem #include "mtree.h"
62 1.1 lukem
63 1.23 dbj static void apply_specdir(const char *, NODE *, fsnode *, int);
64 1.1 lukem static void apply_specentry(const char *, NODE *, fsnode *);
65 1.25 christos static fsnode *create_fsnode(const char *, const char *, const char *,
66 1.25 christos struct stat *);
67 1.6 lukem static fsinode *link_check(fsinode *);
68 1.1 lukem
69 1.34 christos /*
70 1.34 christos * fsnode_cmp --
71 1.34 christos * This function is used by `qsort` so sort one directory's
72 1.34 christos * entries. `.` is always first, sollowed by anything else
73 1.34 christos * as compared by `strcmp()`.
74 1.34 christos */
75 1.34 christos static int
76 1.36 christos fsnode_cmp(const void *vleft, const void *vright)
77 1.34 christos {
78 1.36 christos const fsnode * const *left = vleft;
79 1.36 christos const fsnode * const *right = vright;
80 1.36 christos const char *lname = (*left)->name, *rname = (*right)->name;
81 1.34 christos
82 1.36 christos if (strcmp(lname, ".") == 0)
83 1.34 christos return -1;
84 1.36 christos if (strcmp(rname, ".") == 0)
85 1.34 christos return 1;
86 1.36 christos return strcmp(lname, rname);
87 1.34 christos }
88 1.1 lukem
89 1.37 christos static fsnode *
90 1.37 christos fsnode_sort(fsnode *first, const char *root, const char *dir)
91 1.37 christos {
92 1.37 christos fsnode **list, **listptr;
93 1.37 christos size_t num = 0;
94 1.37 christos
95 1.37 christos for (fsnode *tmp = first; tmp; tmp = tmp->next, num++) {
96 1.37 christos num++;
97 1.37 christos if (debug & DEBUG_DUMP_FSNODES_VERBOSE)
98 1.37 christos printf ("pre sort: %s %s %s\n", root, dir, tmp->name);
99 1.37 christos }
100 1.37 christos
101 1.37 christos list = listptr = ecalloc(num, sizeof(*list));
102 1.37 christos for (fsnode *tmp = first; tmp; tmp = tmp->next)
103 1.37 christos *listptr++ = tmp;
104 1.37 christos
105 1.37 christos qsort (list, num, sizeof(*list), fsnode_cmp);
106 1.37 christos
107 1.37 christos for (size_t i = 0; i < num - 1; ++i)
108 1.37 christos list[i]->next = list[i + 1];
109 1.37 christos list[num - 1]->next = NULL;
110 1.37 christos first = list[0];
111 1.37 christos assert(strcmp(first->name, ".") == 0);
112 1.37 christos free(list);
113 1.37 christos if (debug & DEBUG_DUMP_FSNODES_VERBOSE)
114 1.37 christos for (fsnode *tmp = first; tmp; tmp = tmp->next)
115 1.37 christos printf("post sort: %s %s %s\n", root, dir, tmp->name);
116 1.37 christos
117 1.37 christos return first;
118 1.37 christos }
119 1.37 christos
120 1.1 lukem /*
121 1.1 lukem * walk_dir --
122 1.25 christos * build a tree of fsnodes from `root' and `dir', with a parent
123 1.25 christos * fsnode of `parent' (which may be NULL for the root of the tree).
124 1.25 christos * append the tree to a fsnode of `join' if it is not NULL.
125 1.1 lukem * each "level" is a directory, with the "." entry guaranteed to be
126 1.1 lukem * at the start of the list, and without ".." entries.
127 1.1 lukem */
128 1.1 lukem fsnode *
129 1.28 christos walk_dir(const char *root, const char *dir, fsnode *parent, fsnode *join,
130 1.30 simonb int replace, int follow)
131 1.1 lukem {
132 1.25 christos fsnode *first, *cur, *prev, *last;
133 1.1 lukem DIR *dirp;
134 1.1 lukem struct dirent *dent;
135 1.1 lukem char path[MAXPATHLEN + 1];
136 1.1 lukem struct stat stbuf;
137 1.25 christos char *name, *rp;
138 1.25 christos int dot, len;
139 1.1 lukem
140 1.25 christos assert(root != NULL);
141 1.1 lukem assert(dir != NULL);
142 1.1 lukem
143 1.25 christos len = snprintf(path, sizeof(path), "%s/%s", root, dir);
144 1.37 christos if ((size_t)len >= sizeof(path))
145 1.33 tsutsui errx(EXIT_FAILURE, "Pathname too long.");
146 1.1 lukem if (debug & DEBUG_WALK_DIR)
147 1.25 christos printf("walk_dir: %s %p\n", path, parent);
148 1.25 christos if ((dirp = opendir(path)) == NULL)
149 1.33 tsutsui err(EXIT_FAILURE, "Can't opendir `%s'", path);
150 1.25 christos rp = path + strlen(root) + 1;
151 1.25 christos if (join != NULL) {
152 1.25 christos first = cur = join;
153 1.25 christos while (cur->next != NULL)
154 1.25 christos cur = cur->next;
155 1.25 christos prev = last = cur;
156 1.25 christos } else
157 1.25 christos last = first = prev = NULL;
158 1.1 lukem while ((dent = readdir(dirp)) != NULL) {
159 1.25 christos name = dent->d_name;
160 1.25 christos dot = 0;
161 1.25 christos if (name[0] == '.')
162 1.25 christos switch (name[1]) {
163 1.25 christos case '\0': /* "." */
164 1.25 christos if (join != NULL)
165 1.25 christos continue;
166 1.25 christos dot = 1;
167 1.25 christos break;
168 1.25 christos case '.': /* ".." */
169 1.25 christos if (name[2] == '\0')
170 1.25 christos continue;
171 1.25 christos /* FALLTHROUGH */
172 1.25 christos default:
173 1.25 christos dot = 0;
174 1.25 christos }
175 1.1 lukem if (debug & DEBUG_WALK_DIR_NODE)
176 1.25 christos printf("scanning %s/%s/%s\n", root, dir, name);
177 1.25 christos if (snprintf(path + len, sizeof(path) - len, "/%s", name) >=
178 1.25 christos (int)sizeof(path) - len)
179 1.33 tsutsui errx(EXIT_FAILURE, "Pathname too long.");
180 1.30 simonb if (follow) {
181 1.30 simonb if (stat(path, &stbuf) == -1)
182 1.33 tsutsui err(EXIT_FAILURE, "Can't stat `%s'", path);
183 1.30 simonb } else {
184 1.30 simonb if (lstat(path, &stbuf) == -1)
185 1.33 tsutsui err(EXIT_FAILURE, "Can't lstat `%s'", path);
186 1.37 christos /*
187 1.37 christos * Symlink permission bits vary between filesystems/OSs
188 1.37 christos * (ie. 0755 on FFS/NetBSD, 0777 for ext[234]/Linux),
189 1.37 christos * force them to 0755.
190 1.37 christos */
191 1.35 christos if (S_ISLNK(stbuf.st_mode)) {
192 1.35 christos stbuf.st_mode &= ~(S_IRWXU | S_IRWXG | S_IRWXO);
193 1.35 christos stbuf.st_mode |= S_IRWXU
194 1.37 christos | S_IRGRP | S_IXGRP
195 1.37 christos | S_IROTH | S_IXOTH;
196 1.35 christos }
197 1.30 simonb }
198 1.17 jmc #ifdef S_ISSOCK
199 1.1 lukem if (S_ISSOCK(stbuf.st_mode & S_IFMT)) {
200 1.1 lukem if (debug & DEBUG_WALK_DIR_NODE)
201 1.1 lukem printf(" skipping socket %s\n", path);
202 1.1 lukem continue;
203 1.1 lukem }
204 1.17 jmc #endif
205 1.1 lukem
206 1.25 christos if (join != NULL) {
207 1.25 christos cur = join->next;
208 1.25 christos for (;;) {
209 1.25 christos if (cur == NULL || strcmp(cur->name, name) == 0)
210 1.25 christos break;
211 1.25 christos if (cur == last) {
212 1.25 christos cur = NULL;
213 1.25 christos break;
214 1.25 christos }
215 1.25 christos cur = cur->next;
216 1.25 christos }
217 1.25 christos if (cur != NULL) {
218 1.25 christos if (S_ISDIR(cur->type) &&
219 1.25 christos S_ISDIR(stbuf.st_mode)) {
220 1.25 christos if (debug & DEBUG_WALK_DIR_NODE)
221 1.25 christos printf("merging %s with %p\n",
222 1.25 christos path, cur->child);
223 1.25 christos cur->child = walk_dir(root, rp, cur,
224 1.30 simonb cur->child, replace, follow);
225 1.25 christos continue;
226 1.25 christos }
227 1.28 christos if (!replace)
228 1.33 tsutsui errx(EXIT_FAILURE,
229 1.33 tsutsui "Can't merge %s `%s' with "
230 1.28 christos "existing %s",
231 1.28 christos inode_type(stbuf.st_mode), path,
232 1.28 christos inode_type(cur->type));
233 1.28 christos else {
234 1.28 christos if (debug & DEBUG_WALK_DIR_NODE)
235 1.28 christos printf("replacing %s %s\n",
236 1.28 christos inode_type(stbuf.st_mode),
237 1.28 christos path);
238 1.28 christos if (cur == join->next)
239 1.28 christos join->next = cur->next;
240 1.28 christos else {
241 1.28 christos fsnode *p;
242 1.28 christos for (p = join->next;
243 1.28 christos p->next != cur; p = p->next)
244 1.28 christos continue;
245 1.28 christos p->next = cur->next;
246 1.28 christos }
247 1.28 christos free(cur);
248 1.28 christos }
249 1.25 christos }
250 1.25 christos }
251 1.25 christos
252 1.25 christos cur = create_fsnode(root, dir, name, &stbuf);
253 1.1 lukem cur->parent = parent;
254 1.25 christos if (dot) {
255 1.1 lukem /* ensure "." is at the start of the list */
256 1.1 lukem cur->next = first;
257 1.1 lukem first = cur;
258 1.1 lukem if (! prev)
259 1.1 lukem prev = cur;
260 1.25 christos cur->first = first;
261 1.1 lukem } else { /* not "." */
262 1.1 lukem if (prev)
263 1.1 lukem prev->next = cur;
264 1.1 lukem prev = cur;
265 1.1 lukem if (!first)
266 1.1 lukem first = cur;
267 1.25 christos cur->first = first;
268 1.1 lukem if (S_ISDIR(cur->type)) {
269 1.28 christos cur->child = walk_dir(root, rp, cur, NULL,
270 1.30 simonb replace, follow);
271 1.1 lukem continue;
272 1.1 lukem }
273 1.1 lukem }
274 1.6 lukem if (stbuf.st_nlink > 1) {
275 1.6 lukem fsinode *curino;
276 1.6 lukem
277 1.6 lukem curino = link_check(cur->inode);
278 1.6 lukem if (curino != NULL) {
279 1.6 lukem free(cur->inode);
280 1.6 lukem cur->inode = curino;
281 1.6 lukem cur->inode->nlink++;
282 1.20 dbj if (debug & DEBUG_WALK_DIR_LINKCHECK)
283 1.36 christos printf("link_check: found [%ju, %ju]\n",
284 1.36 christos (uintmax_t)curino->st.st_dev,
285 1.36 christos (uintmax_t)curino->st.st_ino);
286 1.6 lukem }
287 1.1 lukem }
288 1.1 lukem if (S_ISLNK(cur->type)) {
289 1.1 lukem char slink[PATH_MAX+1];
290 1.36 christos ssize_t llen;
291 1.1 lukem
292 1.13 itojun llen = readlink(path, slink, sizeof(slink) - 1);
293 1.1 lukem if (llen == -1)
294 1.33 tsutsui err(EXIT_FAILURE, "Readlink `%s'", path);
295 1.1 lukem slink[llen] = '\0';
296 1.27 christos cur->symlink = estrdup(slink);
297 1.1 lukem }
298 1.1 lukem }
299 1.25 christos assert(first != NULL);
300 1.25 christos if (join == NULL)
301 1.25 christos for (cur = first->next; cur != NULL; cur = cur->next)
302 1.25 christos cur->first = first;
303 1.1 lukem if (closedir(dirp) == -1)
304 1.33 tsutsui err(EXIT_FAILURE, "Can't closedir `%s/%s'", root, dir);
305 1.34 christos
306 1.37 christos return fsnode_sort(first, root, dir);
307 1.1 lukem }
308 1.1 lukem
309 1.1 lukem static fsnode *
310 1.25 christos create_fsnode(const char *root, const char *path, const char *name,
311 1.25 christos struct stat *stbuf)
312 1.1 lukem {
313 1.1 lukem fsnode *cur;
314 1.1 lukem
315 1.27 christos cur = ecalloc(1, sizeof(*cur));
316 1.27 christos cur->path = estrdup(path);
317 1.27 christos cur->name = estrdup(name);
318 1.27 christos cur->inode = ecalloc(1, sizeof(*cur->inode));
319 1.25 christos cur->root = root;
320 1.6 lukem cur->type = stbuf->st_mode & S_IFMT;
321 1.7 lukem cur->inode->nlink = 1;
322 1.7 lukem cur->inode->st = *stbuf;
323 1.29 christos if (stampst.st_ino) {
324 1.29 christos cur->inode->st.st_atime = stampst.st_atime;
325 1.29 christos cur->inode->st.st_mtime = stampst.st_mtime;
326 1.29 christos cur->inode->st.st_ctime = stampst.st_ctime;
327 1.29 christos #if HAVE_STRUCT_STAT_ST_MTIMENSEC
328 1.29 christos cur->inode->st.st_atimensec = stampst.st_atimensec;
329 1.29 christos cur->inode->st.st_mtimensec = stampst.st_mtimensec;
330 1.29 christos cur->inode->st.st_ctimensec = stampst.st_ctimensec;
331 1.29 christos #endif
332 1.32 riastrad #if HAVE_STRUCT_STAT_BIRTHTIME
333 1.29 christos cur->inode->st.st_birthtime = stampst.st_birthtime;
334 1.29 christos cur->inode->st.st_birthtimensec = stampst.st_birthtimensec;
335 1.29 christos #endif
336 1.29 christos }
337 1.1 lukem return (cur);
338 1.1 lukem }
339 1.1 lukem
340 1.1 lukem /*
341 1.22 dbj * free_fsnodes --
342 1.22 dbj * Removes node from tree and frees it and all of
343 1.31 wiz * its descendents.
344 1.22 dbj */
345 1.22 dbj void
346 1.22 dbj free_fsnodes(fsnode *node)
347 1.22 dbj {
348 1.22 dbj fsnode *cur, *next;
349 1.22 dbj
350 1.22 dbj assert(node != NULL);
351 1.22 dbj
352 1.22 dbj /* for ".", start with actual parent node */
353 1.22 dbj if (node->first == node) {
354 1.22 dbj assert(node->name[0] == '.' && node->name[1] == '\0');
355 1.22 dbj if (node->parent) {
356 1.22 dbj assert(node->parent->child == node);
357 1.22 dbj node = node->parent;
358 1.22 dbj }
359 1.22 dbj }
360 1.22 dbj
361 1.22 dbj /* Find ourselves in our sibling list and unlink */
362 1.22 dbj if (node->first != node) {
363 1.22 dbj for (cur = node->first; cur->next; cur = cur->next) {
364 1.22 dbj if (cur->next == node) {
365 1.22 dbj cur->next = node->next;
366 1.22 dbj node->next = NULL;
367 1.22 dbj break;
368 1.22 dbj }
369 1.22 dbj }
370 1.22 dbj }
371 1.22 dbj
372 1.22 dbj for (cur = node; cur != NULL; cur = next) {
373 1.22 dbj next = cur->next;
374 1.22 dbj if (cur->child) {
375 1.22 dbj cur->child->parent = NULL;
376 1.22 dbj free_fsnodes(cur->child);
377 1.22 dbj }
378 1.22 dbj if (cur->inode->nlink-- == 1)
379 1.22 dbj free(cur->inode);
380 1.22 dbj if (cur->symlink)
381 1.22 dbj free(cur->symlink);
382 1.25 christos free(cur->path);
383 1.22 dbj free(cur->name);
384 1.22 dbj free(cur);
385 1.22 dbj }
386 1.22 dbj }
387 1.22 dbj
388 1.22 dbj /*
389 1.1 lukem * apply_specfile --
390 1.1 lukem * read in the mtree(8) specfile, and apply it to the tree
391 1.1 lukem * at dir,parent. parameters in parent on equivalent types
392 1.1 lukem * will be changed to those found in specfile, and missing
393 1.1 lukem * entries will be added.
394 1.1 lukem */
395 1.1 lukem void
396 1.23 dbj apply_specfile(const char *specfile, const char *dir, fsnode *parent, int speconly)
397 1.1 lukem {
398 1.1 lukem struct timeval start;
399 1.1 lukem FILE *fp;
400 1.1 lukem NODE *root;
401 1.1 lukem
402 1.1 lukem assert(specfile != NULL);
403 1.1 lukem assert(parent != NULL);
404 1.1 lukem
405 1.1 lukem if (debug & DEBUG_APPLY_SPECFILE)
406 1.1 lukem printf("apply_specfile: %s, %s %p\n", specfile, dir, parent);
407 1.1 lukem
408 1.1 lukem /* read in the specfile */
409 1.1 lukem if ((fp = fopen(specfile, "r")) == NULL)
410 1.33 tsutsui err(EXIT_FAILURE, "Can't open `%s'", specfile);
411 1.1 lukem TIMER_START(start);
412 1.1 lukem root = spec(fp);
413 1.1 lukem TIMER_RESULTS(start, "spec");
414 1.1 lukem if (fclose(fp) == EOF)
415 1.33 tsutsui err(EXIT_FAILURE, "Can't close `%s'", specfile);
416 1.1 lukem
417 1.1 lukem /* perform some sanity checks */
418 1.1 lukem if (root == NULL)
419 1.33 tsutsui errx(EXIT_FAILURE,
420 1.33 tsutsui "Specfile `%s' did not contain a tree", specfile);
421 1.1 lukem assert(strcmp(root->name, ".") == 0);
422 1.1 lukem assert(root->type == F_DIR);
423 1.1 lukem
424 1.1 lukem /* merge in the changes */
425 1.23 dbj apply_specdir(dir, root, parent, speconly);
426 1.21 dbj
427 1.21 dbj free_nodes(root);
428 1.1 lukem }
429 1.1 lukem
430 1.1 lukem static void
431 1.23 dbj apply_specdir(const char *dir, NODE *specnode, fsnode *dirnode, int speconly)
432 1.1 lukem {
433 1.1 lukem char path[MAXPATHLEN + 1];
434 1.1 lukem NODE *curnode;
435 1.1 lukem fsnode *curfsnode;
436 1.1 lukem
437 1.1 lukem assert(specnode != NULL);
438 1.1 lukem assert(dirnode != NULL);
439 1.1 lukem
440 1.1 lukem if (debug & DEBUG_APPLY_SPECFILE)
441 1.1 lukem printf("apply_specdir: %s %p %p\n", dir, specnode, dirnode);
442 1.1 lukem
443 1.1 lukem if (specnode->type != F_DIR)
444 1.33 tsutsui errx(EXIT_FAILURE, "Specfile node `%s/%s' is not a directory",
445 1.1 lukem dir, specnode->name);
446 1.1 lukem if (dirnode->type != S_IFDIR)
447 1.33 tsutsui errx(EXIT_FAILURE, "Directory node `%s/%s' is not a directory",
448 1.1 lukem dir, dirnode->name);
449 1.1 lukem
450 1.1 lukem apply_specentry(dir, specnode, dirnode);
451 1.1 lukem
452 1.23 dbj /* Remove any filesystem nodes not found in specfile */
453 1.23 dbj /* XXX inefficient. This is O^2 in each dir and it would
454 1.23 dbj * have been better never to have walked this part of the tree
455 1.23 dbj * to begin with
456 1.23 dbj */
457 1.23 dbj if (speconly) {
458 1.23 dbj fsnode *next;
459 1.23 dbj assert(dirnode->name[0] == '.' && dirnode->name[1] == '\0');
460 1.23 dbj for (curfsnode = dirnode->next; curfsnode != NULL; curfsnode = next) {
461 1.23 dbj next = curfsnode->next;
462 1.23 dbj for (curnode = specnode->child; curnode != NULL;
463 1.23 dbj curnode = curnode->next) {
464 1.23 dbj if (strcmp(curnode->name, curfsnode->name) == 0)
465 1.23 dbj break;
466 1.23 dbj }
467 1.23 dbj if (curnode == NULL) {
468 1.23 dbj if (debug & DEBUG_APPLY_SPECONLY) {
469 1.23 dbj printf("apply_specdir: trimming %s/%s %p\n", dir, curfsnode->name, curfsnode);
470 1.23 dbj }
471 1.23 dbj free_fsnodes(curfsnode);
472 1.23 dbj }
473 1.23 dbj }
474 1.23 dbj }
475 1.23 dbj
476 1.1 lukem /* now walk specnode->child matching up with dirnode */
477 1.1 lukem for (curnode = specnode->child; curnode != NULL;
478 1.1 lukem curnode = curnode->next) {
479 1.1 lukem if (debug & DEBUG_APPLY_SPECENTRY)
480 1.1 lukem printf("apply_specdir: spec %s\n",
481 1.1 lukem curnode->name);
482 1.1 lukem for (curfsnode = dirnode->next; curfsnode != NULL;
483 1.1 lukem curfsnode = curfsnode->next) {
484 1.3 lukem #if 0 /* too verbose for now */
485 1.1 lukem if (debug & DEBUG_APPLY_SPECENTRY)
486 1.1 lukem printf("apply_specdir: dirent %s\n",
487 1.1 lukem curfsnode->name);
488 1.3 lukem #endif
489 1.1 lukem if (strcmp(curnode->name, curfsnode->name) == 0)
490 1.1 lukem break;
491 1.1 lukem }
492 1.26 christos if ((size_t)snprintf(path, sizeof(path), "%s/%s",
493 1.9 lukem dir, curnode->name) >= sizeof(path))
494 1.33 tsutsui errx(EXIT_FAILURE, "Pathname too long.");
495 1.1 lukem if (curfsnode == NULL) { /* need new entry */
496 1.1 lukem struct stat stbuf;
497 1.1 lukem
498 1.9 lukem /*
499 1.9 lukem * don't add optional spec entries
500 1.9 lukem * that lack an existing fs entry
501 1.9 lukem */
502 1.9 lukem if ((curnode->flags & F_OPT) &&
503 1.9 lukem lstat(path, &stbuf) == -1)
504 1.9 lukem continue;
505 1.9 lukem
506 1.1 lukem /* check that enough info is provided */
507 1.1 lukem #define NODETEST(t, m) \
508 1.1 lukem if (!(t)) \
509 1.33 tsutsui errx(EXIT_FAILURE, \
510 1.33 tsutsui "`%s': %s not provided", path, m)
511 1.1 lukem NODETEST(curnode->flags & F_TYPE, "type");
512 1.1 lukem NODETEST(curnode->flags & F_MODE, "mode");
513 1.1 lukem /* XXX: require F_TIME ? */
514 1.1 lukem NODETEST(curnode->flags & F_GID ||
515 1.1 lukem curnode->flags & F_GNAME, "group");
516 1.1 lukem NODETEST(curnode->flags & F_UID ||
517 1.1 lukem curnode->flags & F_UNAME, "user");
518 1.1 lukem if (curnode->type == F_BLOCK || curnode->type == F_CHAR)
519 1.1 lukem NODETEST(curnode->flags & F_DEV,
520 1.1 lukem "device number");
521 1.1 lukem #undef NODETEST
522 1.1 lukem
523 1.1 lukem if (debug & DEBUG_APPLY_SPECFILE)
524 1.1 lukem printf("apply_specdir: adding %s\n",
525 1.1 lukem curnode->name);
526 1.1 lukem /* build minimal fsnode */
527 1.1 lukem memset(&stbuf, 0, sizeof(stbuf));
528 1.1 lukem stbuf.st_mode = nodetoino(curnode->type);
529 1.6 lukem stbuf.st_nlink = 1;
530 1.1 lukem stbuf.st_mtime = stbuf.st_atime =
531 1.1 lukem stbuf.st_ctime = start_time.tv_sec;
532 1.8 tv #if HAVE_STRUCT_STAT_ST_MTIMENSEC
533 1.1 lukem stbuf.st_mtimensec = stbuf.st_atimensec =
534 1.1 lukem stbuf.st_ctimensec = start_time.tv_nsec;
535 1.8 tv #endif
536 1.25 christos curfsnode = create_fsnode(".", ".", curnode->name,
537 1.25 christos &stbuf);
538 1.1 lukem curfsnode->parent = dirnode->parent;
539 1.1 lukem curfsnode->first = dirnode;
540 1.1 lukem curfsnode->next = dirnode->next;
541 1.1 lukem dirnode->next = curfsnode;
542 1.1 lukem if (curfsnode->type == S_IFDIR) {
543 1.1 lukem /* for dirs, make "." entry as well */
544 1.25 christos curfsnode->child = create_fsnode(".", ".", ".",
545 1.25 christos &stbuf);
546 1.1 lukem curfsnode->child->parent = curfsnode;
547 1.1 lukem curfsnode->child->first = curfsnode->child;
548 1.1 lukem }
549 1.1 lukem if (curfsnode->type == S_IFLNK) {
550 1.3 lukem assert(curnode->slink != NULL);
551 1.1 lukem /* for symlinks, copy the target */
552 1.27 christos curfsnode->symlink = estrdup(curnode->slink);
553 1.1 lukem }
554 1.1 lukem }
555 1.1 lukem apply_specentry(dir, curnode, curfsnode);
556 1.1 lukem if (curnode->type == F_DIR) {
557 1.1 lukem if (curfsnode->type != S_IFDIR)
558 1.33 tsutsui errx(EXIT_FAILURE,
559 1.33 tsutsui "`%s' is not a directory", path);
560 1.1 lukem assert (curfsnode->child != NULL);
561 1.23 dbj apply_specdir(path, curnode, curfsnode->child, speconly);
562 1.1 lukem }
563 1.1 lukem }
564 1.1 lukem }
565 1.1 lukem
566 1.1 lukem static void
567 1.1 lukem apply_specentry(const char *dir, NODE *specnode, fsnode *dirnode)
568 1.1 lukem {
569 1.1 lukem
570 1.1 lukem assert(specnode != NULL);
571 1.1 lukem assert(dirnode != NULL);
572 1.1 lukem
573 1.1 lukem if (nodetoino(specnode->type) != dirnode->type)
574 1.33 tsutsui errx(EXIT_FAILURE,
575 1.33 tsutsui "`%s/%s' type mismatch: specfile %s, tree %s",
576 1.1 lukem dir, specnode->name, inode_type(nodetoino(specnode->type)),
577 1.1 lukem inode_type(dirnode->type));
578 1.1 lukem
579 1.1 lukem if (debug & DEBUG_APPLY_SPECENTRY)
580 1.1 lukem printf("apply_specentry: %s/%s\n", dir, dirnode->name);
581 1.1 lukem
582 1.1 lukem #define ASEPRINT(t, b, o, n) \
583 1.1 lukem if (debug & DEBUG_APPLY_SPECENTRY) \
584 1.1 lukem printf("\t\t\tchanging %s from " b " to " b "\n", \
585 1.1 lukem t, o, n)
586 1.1 lukem
587 1.1 lukem if (specnode->flags & (F_GID | F_GNAME)) {
588 1.1 lukem ASEPRINT("gid", "%d",
589 1.6 lukem dirnode->inode->st.st_gid, specnode->st_gid);
590 1.6 lukem dirnode->inode->st.st_gid = specnode->st_gid;
591 1.1 lukem }
592 1.1 lukem if (specnode->flags & F_MODE) {
593 1.1 lukem ASEPRINT("mode", "%#o",
594 1.6 lukem dirnode->inode->st.st_mode & ALLPERMS, specnode->st_mode);
595 1.6 lukem dirnode->inode->st.st_mode &= ~ALLPERMS;
596 1.6 lukem dirnode->inode->st.st_mode |= (specnode->st_mode & ALLPERMS);
597 1.1 lukem }
598 1.1 lukem /* XXX: ignoring F_NLINK for now */
599 1.1 lukem if (specnode->flags & F_SIZE) {
600 1.36 christos ASEPRINT("size", "%jd",
601 1.36 christos (intmax_t)dirnode->inode->st.st_size,
602 1.36 christos (intmax_t)specnode->st_size);
603 1.6 lukem dirnode->inode->st.st_size = specnode->st_size;
604 1.1 lukem }
605 1.1 lukem if (specnode->flags & F_SLINK) {
606 1.1 lukem assert(dirnode->symlink != NULL);
607 1.1 lukem assert(specnode->slink != NULL);
608 1.1 lukem ASEPRINT("symlink", "%s", dirnode->symlink, specnode->slink);
609 1.1 lukem free(dirnode->symlink);
610 1.27 christos dirnode->symlink = estrdup(specnode->slink);
611 1.1 lukem }
612 1.1 lukem if (specnode->flags & F_TIME) {
613 1.1 lukem ASEPRINT("time", "%ld",
614 1.6 lukem (long)dirnode->inode->st.st_mtime,
615 1.8 tv (long)specnode->st_mtimespec.tv_sec);
616 1.8 tv dirnode->inode->st.st_mtime = specnode->st_mtimespec.tv_sec;
617 1.8 tv dirnode->inode->st.st_atime = specnode->st_mtimespec.tv_sec;
618 1.8 tv dirnode->inode->st.st_ctime = start_time.tv_sec;
619 1.8 tv #if HAVE_STRUCT_STAT_ST_MTIMENSEC
620 1.17 jmc dirnode->inode->st.st_mtimensec = specnode->st_mtimespec.tv_nsec;
621 1.17 jmc dirnode->inode->st.st_atimensec = specnode->st_mtimespec.tv_nsec;
622 1.6 lukem dirnode->inode->st.st_ctimensec = start_time.tv_nsec;
623 1.8 tv #endif
624 1.1 lukem }
625 1.1 lukem if (specnode->flags & (F_UID | F_UNAME)) {
626 1.1 lukem ASEPRINT("uid", "%d",
627 1.6 lukem dirnode->inode->st.st_uid, specnode->st_uid);
628 1.6 lukem dirnode->inode->st.st_uid = specnode->st_uid;
629 1.1 lukem }
630 1.8 tv #if HAVE_STRUCT_STAT_ST_FLAGS
631 1.1 lukem if (specnode->flags & F_FLAGS) {
632 1.1 lukem ASEPRINT("flags", "%#lX",
633 1.11 uwe (unsigned long)dirnode->inode->st.st_flags,
634 1.11 uwe (unsigned long)specnode->st_flags);
635 1.36 christos dirnode->inode->st.st_flags = (unsigned int)specnode->st_flags;
636 1.1 lukem }
637 1.8 tv #endif
638 1.1 lukem if (specnode->flags & F_DEV) {
639 1.36 christos ASEPRINT("rdev", "%#jx",
640 1.36 christos (uintmax_t)dirnode->inode->st.st_rdev,
641 1.36 christos (uintmax_t)specnode->st_rdev);
642 1.6 lukem dirnode->inode->st.st_rdev = specnode->st_rdev;
643 1.1 lukem }
644 1.1 lukem #undef ASEPRINT
645 1.12 thorpej
646 1.12 thorpej dirnode->flags |= FSNODE_F_HASSPEC;
647 1.1 lukem }
648 1.1 lukem
649 1.1 lukem
650 1.1 lukem /*
651 1.1 lukem * dump_fsnodes --
652 1.25 christos * dump the fsnodes from `cur'
653 1.1 lukem */
654 1.1 lukem void
655 1.25 christos dump_fsnodes(fsnode *root)
656 1.1 lukem {
657 1.1 lukem fsnode *cur;
658 1.1 lukem char path[MAXPATHLEN + 1];
659 1.1 lukem
660 1.25 christos printf("dump_fsnodes: %s %p\n", root->path, root);
661 1.1 lukem for (cur = root; cur != NULL; cur = cur->next) {
662 1.25 christos if (snprintf(path, sizeof(path), "%s/%s", cur->path,
663 1.25 christos cur->name) >= (int)sizeof(path))
664 1.33 tsutsui errx(EXIT_FAILURE, "Pathname too long.");
665 1.1 lukem
666 1.1 lukem if (debug & DEBUG_DUMP_FSNODES_VERBOSE)
667 1.1 lukem printf("cur=%8p parent=%8p first=%8p ",
668 1.1 lukem cur, cur->parent, cur->first);
669 1.1 lukem printf("%7s: %s", inode_type(cur->type), path);
670 1.1 lukem if (S_ISLNK(cur->type)) {
671 1.1 lukem assert(cur->symlink != NULL);
672 1.1 lukem printf(" -> %s", cur->symlink);
673 1.1 lukem } else {
674 1.1 lukem assert (cur->symlink == NULL);
675 1.1 lukem }
676 1.6 lukem if (cur->inode->nlink > 1)
677 1.6 lukem printf(", nlinks=%d", cur->inode->nlink);
678 1.1 lukem putchar('\n');
679 1.1 lukem
680 1.1 lukem if (cur->child) {
681 1.1 lukem assert (cur->type == S_IFDIR);
682 1.25 christos dump_fsnodes(cur->child);
683 1.1 lukem }
684 1.1 lukem }
685 1.25 christos printf("dump_fsnodes: finished %s/%s\n", root->path, root->name);
686 1.1 lukem }
687 1.1 lukem
688 1.1 lukem
689 1.1 lukem /*
690 1.1 lukem * inode_type --
691 1.1 lukem * for a given inode type `mode', return a descriptive string.
692 1.1 lukem * for most cases, uses inotype() from mtree/misc.c
693 1.1 lukem */
694 1.1 lukem const char *
695 1.1 lukem inode_type(mode_t mode)
696 1.1 lukem {
697 1.1 lukem
698 1.3 lukem if (S_ISLNK(mode))
699 1.1 lukem return ("symlink"); /* inotype() returns "link"... */
700 1.1 lukem return (inotype(mode));
701 1.1 lukem }
702 1.1 lukem
703 1.1 lukem
704 1.1 lukem /*
705 1.1 lukem * link_check --
706 1.20 dbj * return pointer to fsinode matching `entry's st_ino & st_dev if it exists,
707 1.1 lukem * otherwise add `entry' to table and return NULL
708 1.1 lukem */
709 1.32 riastrad /* This was borrowed from du.c and tweaked to keep an fsnode
710 1.20 dbj * pointer instead. -- dbj (at) netbsd.org
711 1.20 dbj */
712 1.6 lukem static fsinode *
713 1.6 lukem link_check(fsinode *entry)
714 1.1 lukem {
715 1.20 dbj static struct entry {
716 1.20 dbj fsinode *data;
717 1.20 dbj } *htable;
718 1.36 christos static size_t htshift; /* log(allocated size) */
719 1.36 christos static size_t htmask; /* allocated size - 1 */
720 1.36 christos static size_t htused; /* 2*number of insertions */
721 1.36 christos size_t h, h2;
722 1.20 dbj uint64_t tmp;
723 1.20 dbj /* this constant is (1<<64)/((1+sqrt(5))/2)
724 1.20 dbj * aka (word size)/(golden ratio)
725 1.20 dbj */
726 1.20 dbj const uint64_t HTCONST = 11400714819323198485ULL;
727 1.36 christos const size_t HTBITS = 64;
728 1.32 riastrad
729 1.20 dbj /* Never store zero in hashtable */
730 1.20 dbj assert(entry);
731 1.20 dbj
732 1.20 dbj /* Extend hash table if necessary, keep load under 0.5 */
733 1.20 dbj if (htused<<1 >= htmask) {
734 1.20 dbj struct entry *ohtable;
735 1.20 dbj
736 1.20 dbj if (!htable)
737 1.20 dbj htshift = 10; /* starting hashtable size */
738 1.20 dbj else
739 1.20 dbj htshift++; /* exponential hashtable growth */
740 1.20 dbj
741 1.20 dbj htmask = (1 << htshift) - 1;
742 1.20 dbj htused = 0;
743 1.20 dbj
744 1.20 dbj ohtable = htable;
745 1.27 christos htable = ecalloc(htmask+1, sizeof(*htable));
746 1.20 dbj /* populate newly allocated hashtable */
747 1.20 dbj if (ohtable) {
748 1.36 christos for (size_t i = 0; i <= htmask>>1; i++)
749 1.20 dbj if (ohtable[i].data)
750 1.20 dbj link_check(ohtable[i].data);
751 1.20 dbj free(ohtable);
752 1.1 lukem }
753 1.1 lukem }
754 1.1 lukem
755 1.20 dbj /* multiplicative hashing */
756 1.20 dbj tmp = entry->st.st_dev;
757 1.20 dbj tmp <<= HTBITS>>1;
758 1.20 dbj tmp |= entry->st.st_ino;
759 1.20 dbj tmp *= HTCONST;
760 1.20 dbj h = tmp >> (HTBITS - htshift);
761 1.20 dbj h2 = 1 | ( tmp >> (HTBITS - (htshift<<1) - 1)); /* must be odd */
762 1.32 riastrad
763 1.20 dbj /* open address hashtable search with double hash probing */
764 1.20 dbj while (htable[h].data) {
765 1.20 dbj if ((htable[h].data->st.st_ino == entry->st.st_ino) &&
766 1.20 dbj (htable[h].data->st.st_dev == entry->st.st_dev)) {
767 1.20 dbj return htable[h].data;
768 1.20 dbj }
769 1.20 dbj h = (h + h2) & htmask;
770 1.1 lukem }
771 1.1 lukem
772 1.20 dbj /* Insert the current entry into hashtable */
773 1.20 dbj htable[h].data = entry;
774 1.20 dbj htused++;
775 1.20 dbj return NULL;
776 1.1 lukem }
777