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