xref: /freebsd/contrib/mtree/spec.c (revision 1c324569)
1 /*	$NetBSD: spec.c,v 1.89 2014/04/24 17:22:41 christos Exp $	*/
2 
3 /*-
4  * Copyright (c) 1989, 1993
5  *	The Regents of the University of California.  All rights reserved.
6  *
7  * Redistribution and use in source and binary forms, with or without
8  * modification, are permitted provided that the following conditions
9  * are met:
10  * 1. Redistributions of source code must retain the above copyright
11  *    notice, this list of conditions and the following disclaimer.
12  * 2. Redistributions in binary form must reproduce the above copyright
13  *    notice, this list of conditions and the following disclaimer in the
14  *    documentation and/or other materials provided with the distribution.
15  * 3. Neither the name of the University nor the names of its contributors
16  *    may be used to endorse or promote products derived from this software
17  *    without specific prior written permission.
18  *
19  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
20  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
21  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
22  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
23  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
24  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
25  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
26  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
27  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
28  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
29  * SUCH DAMAGE.
30  */
31 
32 /*-
33  * Copyright (c) 2001-2004 The NetBSD Foundation, Inc.
34  * All rights reserved.
35  *
36  * This code is derived from software contributed to The NetBSD Foundation
37  * by Luke Mewburn of Wasabi Systems.
38  *
39  * Redistribution and use in source and binary forms, with or without
40  * modification, are permitted provided that the following conditions
41  * are met:
42  * 1. Redistributions of source code must retain the above copyright
43  *    notice, this list of conditions and the following disclaimer.
44  * 2. Redistributions in binary form must reproduce the above copyright
45  *    notice, this list of conditions and the following disclaimer in the
46  *    documentation and/or other materials provided with the distribution.
47  *
48  * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
49  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
50  * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
51  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
52  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
53  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
54  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
55  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
56  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
57  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
58  * POSSIBILITY OF SUCH DAMAGE.
59  */
60 
61 #if HAVE_NBTOOL_CONFIG_H
62 #include "nbtool_config.h"
63 #endif
64 
65 #include <sys/cdefs.h>
66 #if defined(__RCSID) && !defined(lint)
67 #if 0
68 static char sccsid[] = "@(#)spec.c	8.2 (Berkeley) 4/28/95";
69 #else
70 __RCSID("$NetBSD: spec.c,v 1.89 2014/04/24 17:22:41 christos Exp $");
71 #endif
72 #endif /* not lint */
73 
74 #include <sys/param.h>
75 #include <sys/stat.h>
76 
77 #include <assert.h>
78 #include <ctype.h>
79 #include <errno.h>
80 #include <grp.h>
81 #include <pwd.h>
82 #include <stdarg.h>
83 #include <stdio.h>
84 #include <stdint.h>
85 #include <stdlib.h>
86 #include <string.h>
87 #include <unistd.h>
88 #include <vis.h>
89 #include <util.h>
90 
91 #include "extern.h"
92 #include "pack_dev.h"
93 
94 size_t	mtree_lineno;			/* Current spec line number */
95 int	mtree_Mflag;			/* Merge duplicate entries */
96 int	mtree_Wflag;			/* Don't "whack" permissions */
97 int	mtree_Sflag;			/* Sort entries */
98 
99 static	dev_t	parsedev(char *);
100 static	void	replacenode(NODE *, NODE *);
101 static	void	set(char *, NODE *);
102 static	void	unset(char *, NODE *);
103 static	void	addchild(NODE *, NODE *);
104 static	int	nodecmp(const NODE *, const NODE *);
105 static	int	appendfield(FILE *, int, const char *, ...) __printflike(3, 4);
106 
107 #define REPLACEPTR(x,v)	do { if ((x)) free((x)); (x) = (v); } while (0)
108 
109 NODE *
spec(FILE * fp)110 spec(FILE *fp)
111 {
112 	NODE *centry, *last, *pathparent, *cur;
113 	char *p, *e, *next;
114 	NODE ginfo, *root;
115 	char *buf, *tname, *ntname;
116 	size_t tnamelen, plen;
117 
118 	root = NULL;
119 	centry = last = NULL;
120 	tname = NULL;
121 	tnamelen = 0;
122 	memset(&ginfo, 0, sizeof(ginfo));
123 	for (mtree_lineno = 0;
124 	    (buf = fparseln(fp, NULL, &mtree_lineno, NULL,
125 		FPARSELN_UNESCCOMM));
126 	    free(buf)) {
127 		/* Skip leading whitespace. */
128 		for (p = buf; *p && isspace((unsigned char)*p); ++p)
129 			continue;
130 
131 		/* If nothing but whitespace, continue. */
132 		if (!*p)
133 			continue;
134 
135 #ifdef DEBUG
136 		fprintf(stderr, "line %lu: {%s}\n",
137 		    (u_long)mtree_lineno, p);
138 #endif
139 		/* Grab file name, "$", "set", or "unset". */
140 		next = buf;
141 		while ((p = strsep(&next, " \t")) != NULL && *p == '\0')
142 			continue;
143 		if (p == NULL)
144 			mtree_err("missing field");
145 
146 		if (p[0] == '/') {
147 			if (strcmp(p + 1, "set") == 0)
148 				set(next, &ginfo);
149 			else if (strcmp(p + 1, "unset") == 0)
150 				unset(next, &ginfo);
151 			else
152 				mtree_err("invalid specification `%s'", p);
153 			continue;
154 		}
155 
156 		if (strcmp(p, "..") == 0) {
157 			/* Don't go up, if haven't gone down. */
158 			if (root == NULL)
159 				goto noparent;
160 			if (last->type != F_DIR || last->flags & F_DONE) {
161 				if (last == root)
162 					goto noparent;
163 				last = last->parent;
164 			}
165 			last->flags |= F_DONE;
166 			continue;
167 
168 noparent:		mtree_err("no parent node");
169 		}
170 
171 		plen = strlen(p) + 1;
172 		if (plen > tnamelen) {
173 			if ((ntname = realloc(tname, plen)) == NULL)
174 				mtree_err("realloc: %s", strerror(errno));
175 			tname = ntname;
176 			tnamelen = plen;
177 		}
178 		if (strunvis(tname, p) == -1)
179 			mtree_err("strunvis failed on `%s'", p);
180 		p = tname;
181 
182 		pathparent = NULL;
183 		if (strchr(p, '/') != NULL) {
184 			cur = root;
185 			for (; (e = strchr(p, '/')) != NULL; p = e+1) {
186 				if (p == e)
187 					continue;	/* handle // */
188 				*e = '\0';
189 				if (strcmp(p, ".") != 0) {
190 					while (cur &&
191 					    strcmp(cur->name, p) != 0) {
192 						cur = cur->next;
193 					}
194 				}
195 				if (cur == NULL || cur->type != F_DIR) {
196 					mtree_err("%s: %s", tname,
197 					"missing directory in specification");
198 				}
199 				*e = '/';
200 				pathparent = cur;
201 				cur = cur->child;
202 			}
203 			if (*p == '\0')
204 				mtree_err("%s: empty leaf element", tname);
205 		}
206 
207 		if ((centry = calloc(1, sizeof(NODE) + strlen(p))) == NULL)
208 			mtree_err("%s", strerror(errno));
209 		*centry = ginfo;
210 		centry->lineno = mtree_lineno;
211 		strcpy(centry->name, p);
212 #define	MAGIC	"?*["
213 		if (strpbrk(p, MAGIC))
214 			centry->flags |= F_MAGIC;
215 		set(next, centry);
216 
217 		if (root == NULL) {
218 				/*
219 				 * empty tree
220 				 */
221 			/*
222 			 * Allow a bare "." root node by forcing it to
223 			 * type=dir for compatibility with FreeBSD.
224 			 */
225 			if (strcmp(centry->name, ".") == 0 && centry->type == 0)
226 				centry->type = F_DIR;
227 			if (strcmp(centry->name, ".") != 0 ||
228 			    centry->type != F_DIR)
229 				mtree_err(
230 				    "root node must be the directory `.'");
231 			last = root = centry;
232 			root->parent = root;
233 		} else if (pathparent != NULL) {
234 				/*
235 				 * full path entry; add or replace
236 				 */
237 			centry->parent = pathparent;
238 			addchild(pathparent, centry);
239 			last = centry;
240 		} else if (strcmp(centry->name, ".") == 0) {
241 				/*
242 				 * duplicate "." entry; always replace
243 				 */
244 			replacenode(root, centry);
245 		} else if (last->type == F_DIR && !(last->flags & F_DONE)) {
246 				/*
247 				 * new relative child in current dir;
248 				 * add or replace
249 				 */
250 			centry->parent = last;
251 			addchild(last, centry);
252 			last = centry;
253 		} else {
254 				/*
255 				 * new relative child in parent dir
256 				 * (after encountering ".." entry);
257 				 * add or replace
258 				 */
259 			centry->parent = last->parent;
260 			addchild(last->parent, centry);
261 			last = centry;
262 		}
263 	}
264 	return (root);
265 }
266 
267 void
free_nodes(NODE * root)268 free_nodes(NODE *root)
269 {
270 	NODE	*cur, *next;
271 
272 	if (root == NULL)
273 		return;
274 
275 	next = NULL;
276 	for (cur = root; cur != NULL; cur = next) {
277 		next = cur->next;
278 		free_nodes(cur->child);
279 		REPLACEPTR(cur->slink, NULL);
280 		REPLACEPTR(cur->md5digest, NULL);
281 		REPLACEPTR(cur->rmd160digest, NULL);
282 		REPLACEPTR(cur->sha1digest, NULL);
283 		REPLACEPTR(cur->sha256digest, NULL);
284 		REPLACEPTR(cur->sha384digest, NULL);
285 		REPLACEPTR(cur->sha512digest, NULL);
286 		REPLACEPTR(cur->tags, NULL);
287 		REPLACEPTR(cur, NULL);
288 	}
289 }
290 
291 /*
292  * appendfield --
293  *	Like fprintf(), but output a space either before or after
294  *	the regular output, according to the pathlast flag.
295  */
296 static int
appendfield(FILE * fp,int pathlast,const char * fmt,...)297 appendfield(FILE *fp, int pathlast, const char *fmt, ...)
298 {
299 	va_list ap;
300 	int result;
301 
302 	va_start(ap, fmt);
303 	if (!pathlast)
304 		fprintf(fp, " ");
305 	result = vprintf(fmt, ap);
306 	if (pathlast)
307 		fprintf(fp, " ");
308 	va_end(ap);
309 	return result;
310 }
311 
312 /*
313  * dump_nodes --
314  *	dump the NODEs from `cur', based in the directory `dir'.
315  *	if pathlast is none zero, print the path last, otherwise print
316  *	it first.
317  */
318 void
dump_nodes(FILE * fp,const char * dir,NODE * root,int pathlast)319 dump_nodes(FILE *fp, const char *dir, NODE *root, int pathlast)
320 {
321 	NODE	*cur;
322 	char	path[MAXPATHLEN];
323 	const char *name;
324 	char	*str;
325 	char	*p, *q;
326 
327 	for (cur = root; cur != NULL; cur = cur->next) {
328 		if (cur->type != F_DIR && !matchtags(cur))
329 			continue;
330 
331 		if (snprintf(path, sizeof(path), "%s%s%s",
332 		    dir, *dir ? "/" : "", cur->name)
333 		    >= (int)sizeof(path))
334 			mtree_err("Pathname too long.");
335 
336 		if (!pathlast)
337 			fprintf(fp, "%s", vispath(path));
338 
339 #define MATCHFLAG(f)	((keys & (f)) && (cur->flags & (f)))
340 		if (MATCHFLAG(F_TYPE))
341 			appendfield(fp, pathlast, "type=%s",
342 			    nodetype(cur->type));
343 		if (MATCHFLAG(F_UID | F_UNAME)) {
344 			if (keys & F_UNAME &&
345 			    (name = user_from_uid(cur->st_uid, 1)) != NULL)
346 				appendfield(fp, pathlast, "uname=%s", name);
347 			else
348 				appendfield(fp, pathlast, "uid=%u",
349 				    cur->st_uid);
350 		}
351 		if (MATCHFLAG(F_GID | F_GNAME)) {
352 			if (keys & F_GNAME &&
353 			    (name = group_from_gid(cur->st_gid, 1)) != NULL)
354 				appendfield(fp, pathlast, "gname=%s", name);
355 			else
356 				appendfield(fp, pathlast, "gid=%u",
357 				    cur->st_gid);
358 		}
359 		if (MATCHFLAG(F_MODE))
360 			appendfield(fp, pathlast, "mode=%#o", cur->st_mode);
361 		if (MATCHFLAG(F_DEV) &&
362 		    (cur->type == F_BLOCK || cur->type == F_CHAR))
363 			appendfield(fp, pathlast, "device=%#jx",
364 			    (uintmax_t)cur->st_rdev);
365 		if (MATCHFLAG(F_NLINK))
366 			appendfield(fp, pathlast, "nlink=%ju",
367 			    (uintmax_t)cur->st_nlink);
368 		if (MATCHFLAG(F_SLINK))
369 			appendfield(fp, pathlast, "link=%s",
370 			    vispath(cur->slink));
371 		if (MATCHFLAG(F_SIZE))
372 			appendfield(fp, pathlast, "size=%ju",
373 			    (uintmax_t)cur->st_size);
374 		if (MATCHFLAG(F_TIME))
375 			appendfield(fp, pathlast, "time=%jd.%09ld",
376 			    (intmax_t)cur->st_mtimespec.tv_sec,
377 			    cur->st_mtimespec.tv_nsec);
378 		if (MATCHFLAG(F_CKSUM))
379 			appendfield(fp, pathlast, "cksum=%lu", cur->cksum);
380 		if (MATCHFLAG(F_MD5))
381 			appendfield(fp, pathlast, "%s=%s", MD5KEY,
382 			    cur->md5digest);
383 		if (MATCHFLAG(F_RMD160))
384 			appendfield(fp, pathlast, "%s=%s", RMD160KEY,
385 			    cur->rmd160digest);
386 		if (MATCHFLAG(F_SHA1))
387 			appendfield(fp, pathlast, "%s=%s", SHA1KEY,
388 			    cur->sha1digest);
389 		if (MATCHFLAG(F_SHA256))
390 			appendfield(fp, pathlast, "%s=%s", SHA256KEY,
391 			    cur->sha256digest);
392 		if (MATCHFLAG(F_SHA384))
393 			appendfield(fp, pathlast, "%s=%s", SHA384KEY,
394 			    cur->sha384digest);
395 		if (MATCHFLAG(F_SHA512))
396 			appendfield(fp, pathlast, "%s=%s", SHA512KEY,
397 			    cur->sha512digest);
398 		if (MATCHFLAG(F_FLAGS)) {
399 			str = flags_to_string(cur->st_flags, "none");
400 			appendfield(fp, pathlast, "flags=%s", str);
401 			free(str);
402 		}
403 		if (MATCHFLAG(F_IGN))
404 			appendfield(fp, pathlast, "ignore");
405 		if (MATCHFLAG(F_OPT))
406 			appendfield(fp, pathlast, "optional");
407 		if (MATCHFLAG(F_TAGS)) {
408 			/* don't output leading or trailing commas */
409 			p = cur->tags;
410 			while (*p == ',')
411 				p++;
412 			q = p + strlen(p);
413 			while(q > p && q[-1] == ',')
414 				q--;
415 			appendfield(fp, pathlast, "tags=%.*s", (int)(q - p), p);
416 		}
417 		puts(pathlast ? vispath(path) : "");
418 
419 		if (cur->child)
420 			dump_nodes(fp, path, cur->child, pathlast);
421 	}
422 }
423 
424 /*
425  * vispath --
426  *	strsvis(3) encodes path, which must not be longer than MAXPATHLEN
427  *	characters long, and returns a pointer to a static buffer containing
428  *	the result.
429  */
430 char *
vispath(const char * path)431 vispath(const char *path)
432 {
433 	static const char extra[] = { ' ', '\t', '\n', '\\', '#', '\0' };
434 	static const char extra_glob[] = { ' ', '\t', '\n', '\\', '#', '*',
435 	    '?', '[', '\0' };
436 	static char pathbuf[4*MAXPATHLEN + 1];
437 
438 	if (flavor == F_NETBSD6)
439 		strsvis(pathbuf, path, VIS_CSTYLE, extra);
440 	else
441 		strsvis(pathbuf, path, VIS_OCTAL, extra_glob);
442 	return pathbuf;
443 }
444 
445 
446 static dev_t
parsedev(char * arg)447 parsedev(char *arg)
448 {
449 #define MAX_PACK_ARGS	3
450 	u_long	numbers[MAX_PACK_ARGS];
451 	char	*p, *ep, *dev;
452 	int	argc;
453 	pack_t	*pack;
454 	dev_t	result;
455 	const char *error = NULL;
456 
457 	if ((dev = strchr(arg, ',')) != NULL) {
458 		*dev++='\0';
459 		if ((pack = pack_find(arg)) == NULL)
460 			mtree_err("unknown format `%s'", arg);
461 		argc = 0;
462 		while ((p = strsep(&dev, ",")) != NULL) {
463 			if (*p == '\0')
464 				mtree_err("missing number");
465 			numbers[argc++] = strtoul(p, &ep, 0);
466 			if (*ep != '\0')
467 				mtree_err("invalid number `%s'",
468 				    p);
469 			if (argc > MAX_PACK_ARGS)
470 				mtree_err("too many arguments");
471 		}
472 		if (argc < 2)
473 			mtree_err("not enough arguments");
474 		result = (*pack)(argc, numbers, &error);
475 		if (error != NULL)
476 			mtree_err("%s", error);
477 	} else {
478 		result = (dev_t)strtoul(arg, &ep, 0);
479 		if (*ep != '\0')
480 			mtree_err("invalid device `%s'", arg);
481 	}
482 	return (result);
483 }
484 
485 static void
replacenode(NODE * cur,NODE * new)486 replacenode(NODE *cur, NODE *new)
487 {
488 
489 #define REPLACE(x)	cur->x = new->x
490 #define REPLACESTR(x)	REPLACEPTR(cur->x,new->x)
491 
492 	if (cur->type != new->type) {
493 		if (mtree_Mflag) {
494 				/*
495 				 * merge entries with different types; we
496 				 * don't want children retained in this case.
497 				 */
498 			REPLACE(type);
499 			free_nodes(cur->child);
500 			cur->child = NULL;
501 		} else {
502 			mtree_err(
503 			    "existing entry for `%s', type `%s'"
504 			    " does not match type `%s'",
505 			    cur->name, nodetype(cur->type),
506 			    nodetype(new->type));
507 		}
508 	}
509 
510 	REPLACE(st_size);
511 	REPLACE(st_mtimespec);
512 	REPLACESTR(slink);
513 	if (cur->slink != NULL) {
514 		if ((cur->slink = strdup(new->slink)) == NULL)
515 			mtree_err("memory allocation error");
516 		if (strunvis(cur->slink, new->slink) == -1)
517 			mtree_err("strunvis failed on `%s'", new->slink);
518 		free(new->slink);
519 	}
520 	REPLACE(st_uid);
521 	REPLACE(st_gid);
522 	REPLACE(st_mode);
523 	REPLACE(st_rdev);
524 	REPLACE(st_flags);
525 	REPLACE(st_nlink);
526 	REPLACE(cksum);
527 	REPLACESTR(md5digest);
528 	REPLACESTR(rmd160digest);
529 	REPLACESTR(sha1digest);
530 	REPLACESTR(sha256digest);
531 	REPLACESTR(sha384digest);
532 	REPLACESTR(sha512digest);
533 	REPLACESTR(tags);
534 	REPLACE(lineno);
535 	REPLACE(flags);
536 	free(new);
537 }
538 
539 static void
set(char * t,NODE * ip)540 set(char *t, NODE *ip)
541 {
542 	int	type, value, len;
543 	gid_t	gid;
544 	uid_t	uid;
545 	char	*kw, *val, *md, *ep;
546 	void	*m;
547 
548 	while ((kw = strsep(&t, "= \t")) != NULL) {
549 		if (*kw == '\0')
550 			continue;
551 		if (strcmp(kw, "all") == 0)
552 			mtree_err("invalid keyword `all'");
553 		ip->flags |= type = parsekey(kw, &value);
554 		if (!value)
555 			/* Just set flag bit (F_IGN and F_OPT) */
556 			continue;
557 		while ((val = strsep(&t, " \t")) != NULL && *val == '\0')
558 			continue;
559 		if (val == NULL)
560 			mtree_err("missing value");
561 		switch (type) {
562 		case F_CKSUM:
563 			ip->cksum = strtoul(val, &ep, 10);
564 			if (*ep)
565 				mtree_err("invalid checksum `%s'", val);
566 			break;
567 		case F_DEV:
568 			ip->st_rdev = parsedev(val);
569 			break;
570 		case F_FLAGS:
571 			if (strcmp("none", val) == 0)
572 				ip->st_flags = 0;
573 			else if (string_to_flags(&val, &ip->st_flags, NULL)
574 			    != 0)
575 				mtree_err("invalid flag `%s'", val);
576 			break;
577 		case F_GID:
578 			ip->st_gid = (gid_t)strtoul(val, &ep, 10);
579 			if (*ep)
580 				mtree_err("invalid gid `%s'", val);
581 			break;
582 		case F_GNAME:
583 			if (mtree_Wflag)	/* don't parse if whacking */
584 				break;
585 			if (gid_from_group(val, &gid) == -1)
586 				mtree_err("unknown group `%s'", val);
587 			ip->st_gid = gid;
588 			break;
589 		case F_MD5:
590 			if (val[0]=='0' && val[1]=='x')
591 				md=&val[2];
592 			else
593 				md=val;
594 			if ((ip->md5digest = strdup(md)) == NULL)
595 				mtree_err("memory allocation error");
596 			break;
597 		case F_MODE:
598 			if ((m = setmode(val)) == NULL)
599 				mtree_err("cannot set file mode `%s' (%s)",
600 				    val, strerror(errno));
601 			ip->st_mode = getmode(m, 0);
602 			free(m);
603 			break;
604 		case F_NLINK:
605 			ip->st_nlink = (nlink_t)strtoul(val, &ep, 10);
606 			if (*ep)
607 				mtree_err("invalid link count `%s'", val);
608 			break;
609 		case F_RMD160:
610 			if (val[0]=='0' && val[1]=='x')
611 				md=&val[2];
612 			else
613 				md=val;
614 			if ((ip->rmd160digest = strdup(md)) == NULL)
615 				mtree_err("memory allocation error");
616 			break;
617 		case F_SHA1:
618 			if (val[0]=='0' && val[1]=='x')
619 				md=&val[2];
620 			else
621 				md=val;
622 			if ((ip->sha1digest = strdup(md)) == NULL)
623 				mtree_err("memory allocation error");
624 			break;
625 		case F_SIZE:
626 			ip->st_size = (off_t)strtoll(val, &ep, 10);
627 			if (*ep)
628 				mtree_err("invalid size `%s'", val);
629 			break;
630 		case F_SLINK:
631 			if ((ip->slink = strdup(val)) == NULL)
632 				mtree_err("memory allocation error");
633 			if (strunvis(ip->slink, val) == -1)
634 				mtree_err("strunvis failed on `%s'", val);
635 			break;
636 		case F_TAGS:
637 			len = strlen(val) + 3;	/* "," + str + ",\0" */
638 			if ((ip->tags = malloc(len)) == NULL)
639 				mtree_err("memory allocation error");
640 			snprintf(ip->tags, len, ",%s,", val);
641 			break;
642 		case F_TIME:
643 			ip->st_mtimespec.tv_sec =
644 			    (time_t)strtoll(val, &ep, 10);
645 			if (*ep != '.')
646 				mtree_err("invalid time `%s'", val);
647 			val = ep + 1;
648 			ip->st_mtimespec.tv_nsec = strtol(val, &ep, 10);
649 			if (*ep)
650 				mtree_err("invalid time `%s'", val);
651 			break;
652 		case F_TYPE:
653 			ip->type = parsetype(val);
654 			break;
655 		case F_UID:
656 			ip->st_uid = (uid_t)strtoul(val, &ep, 10);
657 			if (*ep)
658 				mtree_err("invalid uid `%s'", val);
659 			break;
660 		case F_UNAME:
661 			if (mtree_Wflag)	/* don't parse if whacking */
662 				break;
663 			if (uid_from_user(val, &uid) == -1)
664 				mtree_err("unknown user `%s'", val);
665 			ip->st_uid = uid;
666 			break;
667 		case F_SHA256:
668 			if (val[0]=='0' && val[1]=='x')
669 				md=&val[2];
670 			else
671 				md=val;
672 			if ((ip->sha256digest = strdup(md)) == NULL)
673 				mtree_err("memory allocation error");
674 			break;
675 		case F_SHA384:
676 			if (val[0]=='0' && val[1]=='x')
677 				md=&val[2];
678 			else
679 				md=val;
680 			if ((ip->sha384digest = strdup(md)) == NULL)
681 				mtree_err("memory allocation error");
682 			break;
683 		case F_SHA512:
684 			if (val[0]=='0' && val[1]=='x')
685 				md=&val[2];
686 			else
687 				md=val;
688 			if ((ip->sha512digest = strdup(md)) == NULL)
689 				mtree_err("memory allocation error");
690 			break;
691 		default:
692 			mtree_err(
693 			    "set(): unsupported key type 0x%x (INTERNAL ERROR)",
694 			    type);
695 			/* NOTREACHED */
696 		}
697 	}
698 }
699 
700 static void
unset(char * t,NODE * ip)701 unset(char *t, NODE *ip)
702 {
703 	char *p;
704 
705 	while ((p = strsep(&t, " \t")) != NULL) {
706 		if (*p == '\0')
707 			continue;
708 		ip->flags &= ~parsekey(p, NULL);
709 	}
710 }
711 
712 /*
713  * addchild --
714  *	Add the centry node as a child of the pathparent node.	If
715  *	centry is a duplicate, call replacenode().  If centry is not
716  *	a duplicate, insert it into the linked list referenced by
717  *	pathparent->child.  Keep the list sorted if Sflag is set.
718  */
719 static void
addchild(NODE * pathparent,NODE * centry)720 addchild(NODE *pathparent, NODE *centry)
721 {
722 	NODE *samename;      /* node with the same name as centry */
723 	NODE *replacepos;    /* if non-NULL, centry should replace this node */
724 	NODE *insertpos;     /* if non-NULL, centry should be inserted
725 			      * after this node */
726 	NODE *cur;           /* for stepping through the list */
727 	NODE *last;          /* the last node in the list */
728 	int cmp;
729 
730 	samename = NULL;
731 	replacepos = NULL;
732 	insertpos = NULL;
733 	last = NULL;
734 	cur = pathparent->child;
735 	if (cur == NULL) {
736 		/* centry is pathparent's first and only child node so far */
737 		pathparent->child = centry;
738 		return;
739 	}
740 
741 	/*
742 	 * pathparent already has at least one other child, so add the
743 	 * centry node to the list.
744 	 *
745 	 * We first scan through the list looking for an existing node
746 	 * with the same name (setting samename), and also looking
747 	 * for the correct position to replace or insert the new node
748 	 * (setting replacepos and/or insertpos).
749 	 */
750 	for (; cur != NULL; last = cur, cur = cur->next) {
751 		if (strcmp(centry->name, cur->name) == 0) {
752 			samename = cur;
753 		}
754 		if (mtree_Sflag) {
755 			cmp = nodecmp(centry, cur);
756 			if (cmp == 0) {
757 				replacepos = cur;
758 			} else if (cmp > 0) {
759 				insertpos = cur;
760 			}
761 		}
762 	}
763 	if (! mtree_Sflag) {
764 		if (samename != NULL) {
765 			/* replace node with same name */
766 			replacepos = samename;
767 		} else {
768 			/* add new node at end of list */
769 			insertpos = last;
770 		}
771 	}
772 
773 	if (samename != NULL) {
774 		/*
775 		 * We found a node with the same name above.  Call
776 		 * replacenode(), which will either exit with an error,
777 		 * or replace the information in the samename node and
778 		 * free the information in the centry node.
779 		 */
780 		replacenode(samename, centry);
781 		if (samename == replacepos) {
782 			/* The just-replaced node was in the correct position */
783 			return;
784 		}
785 		if (samename == insertpos || samename->prev == insertpos) {
786 			/*
787 			 * We thought the new node should be just before
788 			 * or just after the replaced node, but that would
789 			 * be equivalent to just retaining the replaced node.
790 			 */
791 			return;
792 		}
793 
794 		/*
795 		 * The just-replaced node is in the wrong position in
796 		 * the list.  This can happen if sort order depends on
797 		 * criteria other than the node name.
798 		 *
799 		 * Make centry point to the just-replaced node.	 Unlink
800 		 * the just-replaced node from the list, and allow it to
801 		 * be insterted in the correct position later.
802 		 */
803 		centry = samename;
804 		if (centry->prev)
805 			centry->prev->next = centry->next;
806 		else {
807 			/* centry->next is the new head of the list */
808 			pathparent->child = centry->next;
809 			assert(centry->next != NULL);
810 		}
811 		if (centry->next)
812 			centry->next->prev = centry->prev;
813 		centry->prev = NULL;
814 		centry->next = NULL;
815 	}
816 
817 	if (insertpos == NULL) {
818 		/* insert centry at the beginning of the list */
819 		pathparent->child->prev = centry;
820 		centry->next = pathparent->child;
821 		centry->prev = NULL;
822 		pathparent->child = centry;
823 	} else {
824 		/* insert centry into the list just after insertpos */
825 		centry->next = insertpos->next;
826 		insertpos->next = centry;
827 		centry->prev = insertpos;
828 		if (centry->next)
829 			centry->next->prev = centry;
830 	}
831 	return;
832 }
833 
834 /*
835  * nodecmp --
836  *	used as a comparison function by addchild() to control the order
837  *	in which entries appear within a list of sibling nodes.	 We make
838  *	directories sort after non-directories, but otherwise sort in
839  *	strcmp() order.
840  *
841  * Keep this in sync with dcmp() in create.c.
842  */
843 static int
nodecmp(const NODE * a,const NODE * b)844 nodecmp(const NODE *a, const NODE *b)
845 {
846 
847 	if ((a->type & F_DIR) != 0) {
848 		if ((b->type & F_DIR) == 0)
849 			return 1;
850 	} else if ((b->type & F_DIR) != 0)
851 		return -1;
852 	return strcmp(a->name, b->name);
853 }
854