xref: /dragonfly/usr.sbin/mtree/spec.c (revision 1de703da)
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. All advertising materials mentioning features or use of this software
14  *    must display the following acknowledgement:
15  *	This product includes software developed by the University of
16  *	California, Berkeley and its contributors.
17  * 4. Neither the name of the University nor the names of its contributors
18  *    may be used to endorse or promote products derived from this software
19  *    without specific prior written permission.
20  *
21  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
22  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
23  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
24  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
25  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
26  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
27  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
28  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
29  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
30  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
31  * SUCH DAMAGE.
32  *
33  * @(#)spec.c	8.1 (Berkeley) 6/6/93
34  * $FreeBSD: src/usr.sbin/mtree/spec.c,v 1.13.2.1 2000/06/28 02:33:17 joe Exp $
35  * $DragonFly: src/usr.sbin/mtree/spec.c,v 1.2 2003/06/17 04:29:57 dillon Exp $
36  */
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 
52 int lineno;				/* Current spec line number. */
53 
54 static void	 set __P((char *, NODE *));
55 static void	 unset __P((char *, NODE *));
56 
57 NODE *
58 spec()
59 {
60 	register NODE *centry, *last;
61 	register 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), stdin);
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 
137 noparent:		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 			warnx("filename %s is ill-encoded and literally used",
148 			    p);
149 			strcpy(centry->name, p);
150 		}
151 		set(NULL, centry);
152 
153 		if (!root) {
154 			last = root = centry;
155 			root->parent = root;
156 		} else if (last->type == F_DIR && !(last->flags & F_DONE)) {
157 			centry->parent = last;
158 			last = last->child = centry;
159 		} else {
160 			centry->parent = last->parent;
161 			centry->prev = last;
162 			last = last->next = centry;
163 		}
164 	}
165 	return (root);
166 }
167 
168 static void
169 set(t, ip)
170 	char *t;
171 	NODE *ip;
172 {
173 	register int type;
174 	char *kw, *val = NULL;
175 	struct group *gr;
176 	struct passwd *pw;
177 	mode_t *m;
178 	int value;
179 	char *ep;
180 
181 	for (; (kw = strtok(t, "= \t\n")); t = NULL) {
182 		ip->flags |= type = parsekey(kw, &value);
183 		if (value && (val = strtok(NULL, " \t\n")) == NULL)
184 			errx(1, "line %d: missing value", lineno);
185 		switch(type) {
186 		case F_CKSUM:
187 			ip->cksum = strtoul(val, &ep, 10);
188 			if (*ep)
189 				errx(1, "line %d: invalid checksum %s",
190 				lineno, val);
191 			break;
192 		case F_MD5:
193 			ip->md5digest = strdup(val);
194 			if(!ip->md5digest) {
195 				errx(1, "strdup");
196 			}
197 			break;
198 		case F_SHA1:
199 			ip->sha1digest = strdup(val);
200 			if(!ip->sha1digest) {
201 				errx(1, "strdup");
202 			}
203 			break;
204 		case F_RMD160:
205 			ip->rmd160digest = strdup(val);
206 			if(!ip->rmd160digest) {
207 				errx(1, "strdup");
208 			}
209 			break;
210 		case F_FLAGS:
211 			if (strcmp("none", val) == 0)
212 				ip->st_flags = 0;
213 			else if (strtofflags(&val, &ip->st_flags, NULL) != 0)
214 				errx(1, "line %d: invalid flag %s",lineno, val);
215  			break;
216 		case F_GID:
217 			ip->st_gid = strtoul(val, &ep, 10);
218 			if (*ep)
219 				errx(1, "line %d: invalid gid %s", lineno, val);
220 			break;
221 		case F_GNAME:
222 			if ((gr = getgrnam(val)) == NULL)
223 			    errx(1, "line %d: unknown group %s", lineno, val);
224 			ip->st_gid = gr->gr_gid;
225 			break;
226 		case F_IGN:
227 			/* just set flag bit */
228 			break;
229 		case F_MODE:
230 			if ((m = setmode(val)) == NULL)
231 				errx(1, "line %d: invalid file mode %s",
232 				lineno, val);
233 			ip->st_mode = getmode(m, 0);
234 			free(m);
235 			break;
236 		case F_NLINK:
237 			ip->st_nlink = strtoul(val, &ep, 10);
238 			if (*ep)
239 				errx(1, "line %d: invalid link count %s",
240 				lineno,  val);
241 			break;
242 		case F_SIZE:
243 			ip->st_size = strtoq(val, &ep, 10);
244 			if (*ep)
245 				errx(1, "line %d: invalid size %s",
246 				lineno, val);
247 			break;
248 		case F_SLINK:
249 			if ((ip->slink = strdup(val)) == NULL)
250 				errx(1, "strdup");
251 			break;
252 		case F_TIME:
253 			ip->st_mtimespec.tv_sec = strtoul(val, &ep, 10);
254 			if (*ep != '.')
255 				errx(1, "line %d: invalid time %s",
256 				lineno, val);
257 			val = ep + 1;
258 			ip->st_mtimespec.tv_nsec = strtoul(val, &ep, 10);
259 			if (*ep)
260 				errx(1, "line %d: invalid time %s",
261 				lineno, val);
262 			break;
263 		case F_TYPE:
264 			switch(*val) {
265 			case 'b':
266 				if (!strcmp(val, "block"))
267 					ip->type = F_BLOCK;
268 				break;
269 			case 'c':
270 				if (!strcmp(val, "char"))
271 					ip->type = F_CHAR;
272 				break;
273 			case 'd':
274 				if (!strcmp(val, "dir"))
275 					ip->type = F_DIR;
276 				break;
277 			case 'f':
278 				if (!strcmp(val, "file"))
279 					ip->type = F_FILE;
280 				if (!strcmp(val, "fifo"))
281 					ip->type = F_FIFO;
282 				break;
283 			case 'l':
284 				if (!strcmp(val, "link"))
285 					ip->type = F_LINK;
286 				break;
287 			case 's':
288 				if (!strcmp(val, "socket"))
289 					ip->type = F_SOCK;
290 				break;
291 			default:
292 				errx(1, "line %d: unknown file type %s",
293 				lineno, val);
294 			}
295 			break;
296 		case F_UID:
297 			ip->st_uid = strtoul(val, &ep, 10);
298 			if (*ep)
299 				errx(1, "line %d: invalid uid %s", lineno, val);
300 			break;
301 		case F_UNAME:
302 			if ((pw = getpwnam(val)) == NULL)
303 			    errx(1, "line %d: unknown user %s", lineno, val);
304 			ip->st_uid = pw->pw_uid;
305 			break;
306 		}
307 	}
308 }
309 
310 static void
311 unset(t, ip)
312 	char *t;
313 	register NODE *ip;
314 {
315 	register char *p;
316 
317 	while ((p = strtok(t, "\n\t ")))
318 		ip->flags &= ~parsekey(p, NULL);
319 }
320