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