spec.c revision 1.58 1 /* $NetBSD: spec.c,v 1.58 2004/12/01 10:07:56 lukem 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 * 3. All advertising materials mentioning features or use of this software
48 * must display the following acknowledgement:
49 * This product includes software developed by the NetBSD
50 * Foundation, Inc. and its contributors.
51 * 4. Neither the name of The NetBSD Foundation nor the names of its
52 * contributors may be used to endorse or promote products derived
53 * from this software without specific prior written permission.
54 *
55 * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
56 * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
57 * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
58 * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
59 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
60 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
61 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
62 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
63 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
64 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
65 * POSSIBILITY OF SUCH DAMAGE.
66 */
67
68 #if HAVE_NBTOOL_CONFIG_H
69 #include "nbtool_config.h"
70 #endif
71
72 #include <sys/cdefs.h>
73 #if defined(__RCSID) && !defined(lint)
74 #if 0
75 static char sccsid[] = "@(#)spec.c 8.2 (Berkeley) 4/28/95";
76 #else
77 __RCSID("$NetBSD: spec.c,v 1.58 2004/12/01 10:07:56 lukem Exp $");
78 #endif
79 #endif /* not lint */
80
81 #include <sys/param.h>
82 #include <sys/stat.h>
83
84 #include <ctype.h>
85 #include <errno.h>
86 #include <grp.h>
87 #include <pwd.h>
88 #include <stdio.h>
89 #include <stdlib.h>
90 #include <string.h>
91 #include <unistd.h>
92 #include <vis.h>
93 #include <util.h>
94
95 #include "extern.h"
96 #include "pack_dev.h"
97
98 size_t mtree_lineno; /* Current spec line number */
99 int mtree_Mflag; /* Merge duplicate entries */
100 int mtree_Wflag; /* Don't "whack" permissions */
101
102 static dev_t parsedev(char *);
103 static void replacenode(NODE *, NODE *);
104 static void set(char *, NODE *);
105 static void unset(char *, NODE *);
106
107 #define REPLACEPTR(x,v) do { if ((x)) free((x)); (x) = (v); } while (0)
108
109 NODE *
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 | FPARSELN_UNESCCONT | FPARSELN_UNESCESC));
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 strerror(ENOENT));
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 if (strcmp(centry->name, ".") != 0 ||
222 centry->type != F_DIR)
223 mtree_err(
224 "root node must be the directory `.'");
225 last = root = centry;
226 root->parent = root;
227 } else if (pathparent != NULL) {
228 /*
229 * full path entry
230 */
231 centry->parent = pathparent;
232 cur = pathparent->child;
233 if (cur == NULL) {
234 pathparent->child = centry;
235 last = centry;
236 } else {
237 for (; cur != NULL; cur = cur->next) {
238 if (strcmp(cur->name, centry->name)
239 == 0) {
240 /* existing entry; replace */
241 replacenode(cur, centry);
242 break;
243 }
244 if (cur->next == NULL) {
245 /* last entry; add new */
246 cur->next = centry;
247 centry->prev = cur;
248 break;
249 }
250 }
251 last = cur;
252 while (last->next != NULL)
253 last = last->next;
254 }
255 } else if (strcmp(centry->name, ".") == 0) {
256 /*
257 * duplicate "." entry; always replace
258 */
259 replacenode(root, centry);
260 } else if (last->type == F_DIR && !(last->flags & F_DONE)) {
261 /*
262 * new relative child
263 * (no duplicate check)
264 */
265 centry->parent = last;
266 last = last->child = centry;
267 } else {
268 /*
269 * relative entry, up one directory
270 * (no duplicate check)
271 */
272 centry->parent = last->parent;
273 centry->prev = last;
274 last = last->next = centry;
275 }
276 }
277 return (root);
278 }
279
280 void
281 free_nodes(NODE *root)
282 {
283 NODE *cur, *next;
284
285 if (root == NULL)
286 return;
287
288 next = NULL;
289 for (cur = root; cur != NULL; cur = next) {
290 next = cur->next;
291 free_nodes(cur->child);
292 REPLACEPTR(cur->slink, NULL);
293 REPLACEPTR(cur->md5digest, NULL);
294 REPLACEPTR(cur->rmd160digest, NULL);
295 REPLACEPTR(cur->sha1digest, NULL);
296 REPLACEPTR(cur->tags, NULL);
297 REPLACEPTR(cur, NULL);
298 }
299 }
300
301 /*
302 * dump_nodes --
303 * dump the NODEs from `cur', based in the directory `dir'.
304 * if pathlast is none zero, print the path last, otherwise print
305 * it first.
306 */
307 void
308 dump_nodes(const char *dir, NODE *root, int pathlast)
309 {
310 NODE *cur;
311 char path[MAXPATHLEN];
312 const char *name;
313
314 for (cur = root; cur != NULL; cur = cur->next) {
315 if (cur->type != F_DIR && !matchtags(cur))
316 continue;
317
318 if (snprintf(path, sizeof(path), "%s%s%s",
319 dir, *dir ? "/" : "", cur->name)
320 >= sizeof(path))
321 mtree_err("Pathname too long.");
322
323 if (!pathlast)
324 printf("%s ", vispath(path));
325
326 #define MATCHFLAG(f) ((keys & (f)) && (cur->flags & (f)))
327 if (MATCHFLAG(F_TYPE))
328 printf("type=%s ", nodetype(cur->type));
329 if (MATCHFLAG(F_UID | F_UNAME)) {
330 if (keys & F_UNAME &&
331 (name = user_from_uid(cur->st_uid, 1)) != NULL)
332 printf("uname=%s ", name);
333 else
334 printf("uid=%u ", cur->st_uid);
335 }
336 if (MATCHFLAG(F_GID | F_GNAME)) {
337 if (keys & F_GNAME &&
338 (name = group_from_gid(cur->st_gid, 1)) != NULL)
339 printf("gname=%s ", name);
340 else
341 printf("gid=%u ", cur->st_gid);
342 }
343 if (MATCHFLAG(F_MODE))
344 printf("mode=%#o ", cur->st_mode);
345 if (MATCHFLAG(F_DEV) &&
346 (cur->type == F_BLOCK || cur->type == F_CHAR))
347 printf("device=%#x ", cur->st_rdev);
348 if (MATCHFLAG(F_NLINK))
349 printf("nlink=%d ", cur->st_nlink);
350 if (MATCHFLAG(F_SLINK))
351 printf("link=%s ", vispath(cur->slink));
352 if (MATCHFLAG(F_SIZE))
353 printf("size=%lld ", (long long)cur->st_size);
354 if (MATCHFLAG(F_TIME))
355 printf("time=%ld.%ld ", (long)cur->st_mtimespec.tv_sec,
356 cur->st_mtimespec.tv_nsec);
357 if (MATCHFLAG(F_CKSUM))
358 printf("cksum=%lu ", cur->cksum);
359 if (MATCHFLAG(F_MD5))
360 printf("md5=%s ", cur->md5digest);
361 if (MATCHFLAG(F_RMD160))
362 printf("rmd160=%s ", cur->rmd160digest);
363 if (MATCHFLAG(F_SHA1))
364 printf("sha1=%s ", cur->sha1digest);
365 if (MATCHFLAG(F_FLAGS))
366 printf("flags=%s ",
367 flags_to_string(cur->st_flags, "none"));
368 if (MATCHFLAG(F_IGN))
369 printf("ignore ");
370 if (MATCHFLAG(F_OPT))
371 printf("optional ");
372 if (MATCHFLAG(F_TAGS))
373 printf("tags=%s ", cur->tags);
374 puts(pathlast ? vispath(path) : "");
375
376 if (cur->child)
377 dump_nodes(path, cur->child, pathlast);
378 }
379 }
380
381 /*
382 * vispath --
383 * strsvis(3) encodes path, which must not be longer than MAXPATHLEN
384 * characters long, and returns a pointer to a static buffer containing
385 * the result.
386 */
387 char *
388 vispath(const char *path)
389 {
390 const char extra[] = { ' ', '\t', '\n', '\\', '#', '\0' };
391 static char pathbuf[4*MAXPATHLEN + 1];
392
393 strsvis(pathbuf, path, VIS_CSTYLE, extra);
394 return(pathbuf);
395 }
396
397
398 static dev_t
399 parsedev(char *arg)
400 {
401 #define MAX_PACK_ARGS 3
402 u_long numbers[MAX_PACK_ARGS];
403 char *p, *ep, *dev;
404 int argc;
405 pack_t *pack;
406 dev_t result;
407 const char *error = NULL;
408
409 if ((dev = strchr(arg, ',')) != NULL) {
410 *dev++='\0';
411 if ((pack = pack_find(arg)) == NULL)
412 mtree_err("unknown format `%s'", arg);
413 argc = 0;
414 while ((p = strsep(&dev, ",")) != NULL) {
415 if (*p == '\0')
416 mtree_err("missing number");
417 numbers[argc++] = strtoul(p, &ep, 0);
418 if (*ep != '\0')
419 mtree_err("invalid number `%s'",
420 p);
421 if (argc > MAX_PACK_ARGS)
422 mtree_err("too many arguments");
423 }
424 if (argc < 2)
425 mtree_err("not enough arguments");
426 result = (*pack)(argc, numbers, &error);
427 if (error != NULL)
428 mtree_err(error);
429 } else {
430 result = (dev_t)strtoul(arg, &ep, 0);
431 if (*ep != '\0')
432 mtree_err("invalid device `%s'", arg);
433 }
434 return (result);
435 }
436
437 static void
438 replacenode(NODE *cur, NODE *new)
439 {
440
441 #define REPLACE(x) cur->x = new->x
442 #define REPLACESTR(x) REPLACEPTR(cur->x,new->x)
443
444 if (cur->type != new->type) {
445 if (mtree_Mflag) {
446 /*
447 * merge entries with different types; we
448 * don't want children retained in this case.
449 */
450 REPLACE(type);
451 free_nodes(cur->child);
452 cur->child = NULL;
453 } else {
454 mtree_err(
455 "existing entry for `%s', type `%s'"
456 " does not match type `%s'",
457 cur->name, nodetype(cur->type),
458 nodetype(new->type));
459 }
460 }
461
462 REPLACE(st_size);
463 REPLACE(st_mtimespec);
464 REPLACESTR(slink);
465 if (strunvis(cur->slink, new->slink) == -1)
466 mtree_err("strunvis failed on `%s'", new->slink);
467 REPLACE(st_uid);
468 REPLACE(st_gid);
469 REPLACE(st_mode);
470 REPLACE(st_rdev);
471 REPLACE(st_flags);
472 REPLACE(st_nlink);
473 REPLACE(cksum);
474 REPLACESTR(md5digest);
475 REPLACESTR(rmd160digest);
476 REPLACESTR(sha1digest);
477 REPLACESTR(tags);
478 REPLACE(lineno);
479 REPLACE(flags);
480 free(new);
481 }
482
483 static void
484 set(char *t, NODE *ip)
485 {
486 int type, value, len;
487 gid_t gid;
488 uid_t uid;
489 char *kw, *val, *md, *ep;
490 void *m;
491
492 val = NULL;
493 while ((kw = strsep(&t, "= \t")) != NULL) {
494 if (*kw == '\0')
495 continue;
496 if (strcmp(kw, "all") == 0)
497 mtree_err("invalid keyword `all'");
498 ip->flags |= type = parsekey(kw, &value);
499 if (value) {
500 while ((val = strsep(&t, " \t")) != NULL &&
501 *val == '\0')
502 continue;
503 if (val == NULL)
504 mtree_err("missing value");
505 }
506 switch(type) {
507 case F_CKSUM:
508 ip->cksum = strtoul(val, &ep, 10);
509 if (*ep)
510 mtree_err("invalid checksum `%s'", val);
511 break;
512 case F_DEV:
513 ip->st_rdev = parsedev(val);
514 break;
515 case F_FLAGS:
516 if (strcmp("none", val) == 0)
517 ip->st_flags = 0;
518 else if (string_to_flags(&val, &ip->st_flags, NULL)
519 != 0)
520 mtree_err("invalid flag `%s'", val);
521 break;
522 case F_GID:
523 ip->st_gid = (gid_t)strtoul(val, &ep, 10);
524 if (*ep)
525 mtree_err("invalid gid `%s'", val);
526 break;
527 case F_GNAME:
528 if (mtree_Wflag) /* don't parse if whacking */
529 break;
530 if (gid_from_group(val, &gid) == -1)
531 mtree_err("unknown group `%s'", val);
532 ip->st_gid = gid;
533 break;
534 case F_IGN:
535 /* just set flag bit */
536 break;
537 case F_MD5:
538 if (val[0]=='0' && val[1]=='x')
539 md=&val[2];
540 else
541 md=val;
542 if ((ip->md5digest = strdup(md)) == NULL)
543 mtree_err("memory allocation error");
544 break;
545 case F_MODE:
546 if ((m = setmode(val)) == NULL)
547 mtree_err("invalid file mode `%s'", val);
548 ip->st_mode = getmode(m, 0);
549 free(m);
550 break;
551 case F_NLINK:
552 ip->st_nlink = (nlink_t)strtoul(val, &ep, 10);
553 if (*ep)
554 mtree_err("invalid link count `%s'", val);
555 break;
556 case F_OPT:
557 /* just set flag bit */
558 break;
559 case F_RMD160:
560 if (val[0]=='0' && val[1]=='x')
561 md=&val[2];
562 else
563 md=val;
564 if ((ip->rmd160digest = strdup(md)) == NULL)
565 mtree_err("memory allocation error");
566 break;
567 case F_SHA1:
568 if (val[0]=='0' && val[1]=='x')
569 md=&val[2];
570 else
571 md=val;
572 if ((ip->sha1digest = strdup(md)) == NULL)
573 mtree_err("memory allocation error");
574 break;
575 case F_SIZE:
576 ip->st_size = (off_t)strtoll(val, &ep, 10);
577 if (*ep)
578 mtree_err("invalid size `%s'", val);
579 break;
580 case F_SLINK:
581 if ((ip->slink = strdup(val)) == NULL)
582 mtree_err("memory allocation error");
583 if (strunvis(ip->slink, val) == -1)
584 mtree_err("strunvis failed on `%s'", val);
585 break;
586 case F_TAGS:
587 len = strlen(val) + 3; /* "," + str + ",\0" */
588 if ((ip->tags = malloc(len)) == NULL)
589 mtree_err("memory allocation error");
590 snprintf(ip->tags, len, ",%s,", val);
591 break;
592 case F_TIME:
593 ip->st_mtimespec.tv_sec =
594 (time_t)strtoul(val, &ep, 10);
595 if (*ep != '.')
596 mtree_err("invalid time `%s'", val);
597 val = ep + 1;
598 ip->st_mtimespec.tv_nsec = strtoul(val, &ep, 10);
599 if (*ep)
600 mtree_err("invalid time `%s'", val);
601 break;
602 case F_TYPE:
603 ip->type = parsetype(val);
604 break;
605 case F_UID:
606 ip->st_uid = (uid_t)strtoul(val, &ep, 10);
607 if (*ep)
608 mtree_err("invalid uid `%s'", val);
609 break;
610 case F_UNAME:
611 if (mtree_Wflag) /* don't parse if whacking */
612 break;
613 if (uid_from_user(val, &uid) == -1)
614 mtree_err("unknown user `%s'", val);
615 ip->st_uid = uid;
616 break;
617 default:
618 mtree_err(
619 "set(): unsupported key type 0x%x (INTERNAL ERROR)",
620 type);
621 /* NOTREACHED */
622 }
623 }
624 }
625
626 static void
627 unset(char *t, NODE *ip)
628 {
629 char *p;
630
631 while ((p = strsep(&t, " \t")) != NULL) {
632 if (*p == '\0')
633 continue;
634 ip->flags &= ~parsekey(p, NULL);
635 }
636 }
637