spec.c revision 122135
1/*-
2 * Copyright (c) 1989, 1993
3 *	The Regents of the University of California.  All rights reserved.
4 *
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
7 * are met:
8 * 1. Redistributions of source code must retain the above copyright
9 *    notice, this list of conditions and the following disclaimer.
10 * 2. Redistributions in binary form must reproduce the above copyright
11 *    notice, this list of conditions and the following disclaimer in the
12 *    documentation and/or other materials provided with the distribution.
13 * 3. Neither the name of the University nor the names of its contributors
14 *    may be used to endorse or promote products derived from this software
15 *    without specific prior written permission.
16 *
17 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
18 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
19 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
20 * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
21 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
22 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
23 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
24 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
25 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
26 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
27 * SUCH DAMAGE.
28 */
29
30#if 0
31#ifndef lint
32static char sccsid[] = "@(#)spec.c	8.1 (Berkeley) 6/6/93";
33#endif /* not lint */
34#endif
35#include <sys/cdefs.h>
36__FBSDID("$FreeBSD: head/usr.sbin/mtree/spec.c 122135 2003-11-05 20:07:40Z phk $");
37
38#include <sys/types.h>
39#include <sys/stat.h>
40#include <ctype.h>
41#include <err.h>
42#include <errno.h>
43#include <fts.h>
44#include <grp.h>
45#include <pwd.h>
46#include <stdio.h>
47#include <unistd.h>
48#include <vis.h>
49#include "mtree.h"
50#include "extern.h"
51
52int lineno;				/* Current spec line number. */
53
54static void	 set(char *, NODE *);
55static void	 unset(char *, NODE *);
56
57NODE *
58mtree_readspec(FILE *fi)
59{
60	NODE *centry, *last;
61	char *p;
62	NODE ginfo, *root;
63	int c_cur, c_next;
64	char buf[2048];
65
66	centry = last = root = NULL;
67	bzero(&ginfo, sizeof(ginfo));
68	c_cur = c_next = 0;
69	for (lineno = 1; fgets(buf, sizeof(buf), fi);
70	    ++lineno, c_cur = c_next, c_next = 0) {
71		/* Skip empty lines. */
72		if (buf[0] == '\n')
73			continue;
74
75		/* Find end of line. */
76		if ((p = index(buf, '\n')) == NULL)
77			errx(1, "line %d too long", lineno);
78
79		/* See if next line is continuation line. */
80		if (p[-1] == '\\') {
81			--p;
82			c_next = 1;
83		}
84
85		/* Null-terminate the line. */
86		*p = '\0';
87
88		/* Skip leading whitespace. */
89		for (p = buf; *p && isspace(*p); ++p);
90
91		/* If nothing but whitespace or comment char, continue. */
92		if (!*p || *p == '#')
93			continue;
94
95#ifdef DEBUG
96		(void)fprintf(stderr, "line %d: {%s}\n", lineno, p);
97#endif
98		if (c_cur) {
99			set(p, centry);
100			continue;
101		}
102
103		/* Grab file name, "$", "set", or "unset". */
104		if ((p = strtok(p, "\n\t ")) == NULL)
105			errx(1, "line %d: missing field", lineno);
106
107		if (p[0] == '/')
108			switch(p[1]) {
109			case 's':
110				if (strcmp(p + 1, "set"))
111					break;
112				set(NULL, &ginfo);
113				continue;
114			case 'u':
115				if (strcmp(p + 1, "unset"))
116					break;
117				unset(NULL, &ginfo);
118				continue;
119			}
120
121		if (index(p, '/'))
122			errx(1, "line %d: slash character in file name",
123			lineno);
124
125		if (!strcmp(p, "..")) {
126			/* Don't go up, if haven't gone down. */
127			if (!root)
128				goto noparent;
129			if (last->type != F_DIR || last->flags & F_DONE) {
130				if (last == root)
131					goto noparent;
132				last = last->parent;
133			}
134			last->flags |= F_DONE;
135			continue;
136
137noparent:		errx(1, "line %d: no parent node", lineno);
138		}
139
140		if ((centry = calloc(1, sizeof(NODE) + strlen(p))) == NULL)
141			errx(1, "calloc");
142		*centry = ginfo;
143#define	MAGIC	"?*["
144		if (strpbrk(p, MAGIC))
145			centry->flags |= F_MAGIC;
146		if (strunvis(centry->name, p) == -1)
147			errx(1, "filename %s is ill-encoded", p);
148		set(NULL, centry);
149
150		if (!root) {
151			last = root = centry;
152			root->parent = root;
153		} else if (last->type == F_DIR && !(last->flags & F_DONE)) {
154			centry->parent = last;
155			last = last->child = centry;
156		} else {
157			centry->parent = last->parent;
158			centry->prev = last;
159			last = last->next = centry;
160		}
161	}
162	return (root);
163}
164
165static void
166set(char *t, NODE *ip)
167{
168	int type;
169	char *kw, *val = NULL;
170	struct group *gr;
171	struct passwd *pw;
172	mode_t *m;
173	int value;
174	char *ep;
175
176	for (; (kw = strtok(t, "= \t\n")); t = NULL) {
177		ip->flags |= type = parsekey(kw, &value);
178		if (value && (val = strtok(NULL, " \t\n")) == NULL)
179			errx(1, "line %d: missing value", lineno);
180		switch(type) {
181		case F_CKSUM:
182			ip->cksum = strtoul(val, &ep, 10);
183			if (*ep)
184				errx(1, "line %d: invalid checksum %s",
185				lineno, val);
186			break;
187		case F_MD5:
188			ip->md5digest = strdup(val);
189			if(!ip->md5digest)
190				errx(1, "strdup");
191			break;
192		case F_SHA1:
193			ip->sha1digest = strdup(val);
194			if(!ip->sha1digest)
195				errx(1, "strdup");
196			break;
197		case F_RMD160:
198			ip->rmd160digest = strdup(val);
199			if(!ip->rmd160digest)
200				errx(1, "strdup");
201			break;
202		case F_FLAGS:
203			if (strcmp("none", val) == 0)
204				ip->st_flags = 0;
205			else if (strtofflags(&val, &ip->st_flags, NULL) != 0)
206				errx(1, "line %d: invalid flag %s",lineno, val);
207 			break;
208		case F_GID:
209			ip->st_gid = strtoul(val, &ep, 10);
210			if (*ep)
211				errx(1, "line %d: invalid gid %s", lineno, val);
212			break;
213		case F_GNAME:
214			if ((gr = getgrnam(val)) == NULL)
215			    errx(1, "line %d: unknown group %s", lineno, val);
216			ip->st_gid = gr->gr_gid;
217			break;
218		case F_IGN:
219			/* just set flag bit */
220			break;
221		case F_MODE:
222			if ((m = setmode(val)) == NULL)
223				errx(1, "line %d: invalid file mode %s",
224				lineno, val);
225			ip->st_mode = getmode(m, 0);
226			free(m);
227			break;
228		case F_NLINK:
229			ip->st_nlink = strtoul(val, &ep, 10);
230			if (*ep)
231				errx(1, "line %d: invalid link count %s",
232				lineno,  val);
233			break;
234		case F_SIZE:
235			ip->st_size = strtoq(val, &ep, 10);
236			if (*ep)
237				errx(1, "line %d: invalid size %s",
238				lineno, val);
239			break;
240		case F_SLINK:
241			ip->slink = malloc(strlen(val));
242			if (ip->slink == NULL)
243				errx(1, "malloc");
244			if (strunvis(ip->slink, val) == -1)
245				errx(1, "symlink %s is ill-encoded", val);
246			break;
247		case F_TIME:
248			ip->st_mtimespec.tv_sec = strtoul(val, &ep, 10);
249			if (*ep != '.')
250				errx(1, "line %d: invalid time %s",
251				lineno, val);
252			val = ep + 1;
253			ip->st_mtimespec.tv_nsec = strtoul(val, &ep, 10);
254			if (*ep)
255				errx(1, "line %d: invalid time %s",
256				lineno, val);
257			break;
258		case F_TYPE:
259			switch(*val) {
260			case 'b':
261				if (!strcmp(val, "block"))
262					ip->type = F_BLOCK;
263				break;
264			case 'c':
265				if (!strcmp(val, "char"))
266					ip->type = F_CHAR;
267				break;
268			case 'd':
269				if (!strcmp(val, "dir"))
270					ip->type = F_DIR;
271				break;
272			case 'f':
273				if (!strcmp(val, "file"))
274					ip->type = F_FILE;
275				if (!strcmp(val, "fifo"))
276					ip->type = F_FIFO;
277				break;
278			case 'l':
279				if (!strcmp(val, "link"))
280					ip->type = F_LINK;
281				break;
282			case 's':
283				if (!strcmp(val, "socket"))
284					ip->type = F_SOCK;
285				break;
286			default:
287				errx(1, "line %d: unknown file type %s",
288				lineno, val);
289			}
290			break;
291		case F_UID:
292			ip->st_uid = strtoul(val, &ep, 10);
293			if (*ep)
294				errx(1, "line %d: invalid uid %s", lineno, val);
295			break;
296		case F_UNAME:
297			if ((pw = getpwnam(val)) == NULL)
298			    errx(1, "line %d: unknown user %s", lineno, val);
299			ip->st_uid = pw->pw_uid;
300			break;
301		}
302	}
303}
304
305static void
306unset(char *t, NODE *ip)
307{
308	char *p;
309
310	while ((p = strtok(t, "\n\t ")))
311		ip->flags &= ~parsekey(p, NULL);
312}
313