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