[apple/file_cmds.git] / mtree / spec.c

/*-
 * Copyright (c) 1989, 1993
 *	The Regents of the University of California.  All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 * 3. Neither the name of the University nor the names of its contributors
 *    may be used to endorse or promote products derived from this software
 *    without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
 * SUCH DAMAGE.
 */

#if 0
#ifndef lint
static char sccsid[] = "@(#)spec.c	8.1 (Berkeley) 6/6/93";
#endif /* not lint */
#endif
#include <sys/cdefs.h>
__FBSDID("$FreeBSD: src/usr.sbin/mtree/spec.c,v 1.22 2005/03/29 11:44:17 tobez Exp $");

#include <sys/types.h>
#include <sys/stat.h>
#include <ctype.h>
#include <err.h>
#include <errno.h>
#include <fts.h>
#include <grp.h>
#include <pwd.h>
#include <stdio.h>
#include <stdint.h>
#include <unistd.h>
#include <vis.h>
#include "mtree.h"
#include "extern.h"

int lineno;				/* Current spec line number. */

static void	 set(char *, NODE *);
static void	 unset(char *, NODE *);

NODE *
mtree_readspec(FILE *fi)
{
	NODE *centry, *last;
	char *p;
	NODE ginfo, *root;
	int c_cur, c_next;
	char buf[2048];

	centry = last = root = NULL;
	bzero(&ginfo, sizeof(ginfo));
	c_cur = c_next = 0;
	for (lineno = 1; fgets(buf, sizeof(buf), fi);
	    ++lineno, c_cur = c_next, c_next = 0) {
		/* Skip empty lines. */
		if (buf[0] == '\n')
			continue;

		/* Find end of line. */
		if ((p = index(buf, '\n')) == NULL)
			errx(1, "line %d too long", lineno);

		/* See if next line is continuation line. */
		if (p[-1] == '\\') {
			--p;
			c_next = 1;
		}

		/* Null-terminate the line. */
		*p = '\0';

		/* Skip leading whitespace. */
		for (p = buf; *p && isspace(*p); ++p);

		/* If nothing but whitespace or comment char, continue. */
		if (!*p || *p == '#')
			continue;

#ifdef DEBUG
		(void)fprintf(stderr, "line %d: {%s}\n", lineno, p);
#endif
		if (c_cur) {
			set(p, centry);
			continue;
		}

		/* Grab file name, "$", "set", or "unset". */
		if ((p = strtok(p, "\n\t ")) == NULL)
			errx(1, "line %d: missing field", lineno);

		if (p[0] == '/')
			switch(p[1]) {
			case 's':
				if (strcmp(p + 1, "set"))
					break;
				set(NULL, &ginfo);
				continue;
			case 'u':
				if (strcmp(p + 1, "unset"))
					break;
				unset(NULL, &ginfo);
				continue;
			}

		if (index(p, '/'))
			errx(1, "line %d: slash character in file name",
			lineno);

		if (!strcmp(p, "..")) {
			/* Don't go up, if haven't gone down. */
			if (!root)
				goto noparent;
			if (last->type != F_DIR || last->flags & F_DONE) {
				if (last == root)
					goto noparent;
				last = last->parent;
			}
			last->flags |= F_DONE;
			continue;

noparent:		errx(1, "line %d: no parent node", lineno);
		}

		if ((centry = calloc(1, sizeof(NODE) + strlen(p))) == NULL)
			errx(1, "calloc");
		*centry = ginfo;
#define	MAGIC	"?*["
		if (strpbrk(p, MAGIC))
			centry->flags |= F_MAGIC;
		if (strunvis(centry->name, p) == -1)
			errx(1, "filename %s is ill-encoded", p);
		set(NULL, centry);

		if (!root) {
			last = root = centry;
			root->parent = root;
		} else if (last->type == F_DIR && !(last->flags & F_DONE)) {
			centry->parent = last;
			last = last->child = centry;
		} else {
			centry->parent = last->parent;
			centry->prev = last;
			last = last->next = centry;
		}
	}
	return (root);
}

static void
set(char *t, NODE *ip)
{
	int type;
	char *kw, *val = NULL;
	struct group *gr;
	struct passwd *pw;
	mode_t *m;
	int value;
	char *ep;

	for (; (kw = strtok(t, "= \t\n")); t = NULL) {
		ip->flags |= type = parsekey(kw, &value);
		if (value && (val = strtok(NULL, " \t\n")) == NULL)
			errx(1, "line %d: missing value", lineno);
		switch(type) {
		case F_CKSUM:
			ip->cksum = strtoul(val, &ep, 10);
			if (*ep)
				errx(1, "line %d: invalid checksum %s",
				lineno, val);
			break;
		case F_MD5:
			ip->md5digest = strdup(val);
			if(!ip->md5digest)
				errx(1, "strdup");
			break;
		case F_SHA1:
			ip->sha1digest = strdup(val);
			if(!ip->sha1digest)
				errx(1, "strdup");
			break;
		case F_SHA256:
			ip->sha256digest = strdup(val);
			if(!ip->sha256digest)
				errx(1, "strdup");
			break;
		case F_RMD160:
			ip->rmd160digest = strdup(val);
			if(!ip->rmd160digest)
				errx(1, "strdup");
			break;
		case F_FLAGS:
			if (strcmp("none", val) == 0)
				ip->st_flags = 0;
			else if (strtofflags(&val, &ip->st_flags, NULL) != 0)
				errx(1, "line %d: invalid flag %s",lineno, val);
 			break;
		case F_GID:
			ip->st_gid = (gid_t)strtoul(val, &ep, 10);
			if (*ep)
				errx(1, "line %d: invalid gid %s", lineno, val);
			break;
		case F_GNAME:
			if ((gr = getgrnam(val)) == NULL)
			    errx(1, "line %d: unknown group %s", lineno, val);
			ip->st_gid = gr->gr_gid;
			break;
		case F_IGN:
			/* just set flag bit */
			break;
		case F_MODE:
			if ((m = setmode(val)) == NULL)
				errx(1, "line %d: invalid file mode %s",
				lineno, val);
			ip->st_mode = getmode(m, 0);
			free(m);
			break;
		case F_NLINK:
			ip->st_nlink = strtoul(val, &ep, 10);
			if (*ep)
				errx(1, "line %d: invalid link count %s",
				lineno,  val);
			break;
		case F_SIZE:
			ip->st_size = strtoq(val, &ep, 10);
			if (*ep)
				errx(1, "line %d: invalid size %s",
				lineno, val);
			break;
		case F_SLINK:
			ip->slink = malloc(strlen(val) + 1);
			if (ip->slink == NULL)
				errx(1, "malloc");
			if (strunvis(ip->slink, val) == -1)
				errx(1, "symlink %s is ill-encoded", val);
			break;
		case F_TIME:
			ip->st_mtimespec.tv_sec = strtoul(val, &ep, 10);
			if (*ep != '.')
				errx(1, "line %d: invalid time %s",
				lineno, val);
			val = ep + 1;
			ip->st_mtimespec.tv_nsec = strtoul(val, &ep, 10);
			if (*ep)
				errx(1, "line %d: invalid time %s",
				lineno, val);
			break;
		case F_TYPE:
			switch(*val) {
			case 'b':
				if (!strcmp(val, "block"))
					ip->type = F_BLOCK;
				break;
			case 'c':
				if (!strcmp(val, "char"))
					ip->type = F_CHAR;
				break;
			case 'd':
				if (!strcmp(val, "dir"))
					ip->type = F_DIR;
				break;
			case 'f':
				if (!strcmp(val, "file"))
					ip->type = F_FILE;
				if (!strcmp(val, "fifo"))
					ip->type = F_FIFO;
				break;
			case 'l':
				if (!strcmp(val, "link"))
					ip->type = F_LINK;
				break;
			case 's':
				if (!strcmp(val, "socket"))
					ip->type = F_SOCK;
				break;
			default:
				errx(1, "line %d: unknown file type %s",
				lineno, val);
			}
			break;
		case F_UID:
			ip->st_uid = (uid_t)strtoul(val, &ep, 10);
			if (*ep)
				errx(1, "line %d: invalid uid %s", lineno, val);
			break;
		case F_UNAME:
			if ((pw = getpwnam(val)) == NULL)
			    errx(1, "line %d: unknown user %s", lineno, val);
			ip->st_uid = pw->pw_uid;
			break;
		}
	}
}

static void
unset(char *t, NODE *ip)
{
	char *p;

	while ((p = strtok(t, "\n\t ")))
		ip->flags &= ~parsekey(p, NULL);
}
Commit	Line	Data
44a7a5ab A	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.
864a4b6e	13	* 3. Neither the name of the University nor the names of its contributors
44a7a5ab A	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
44a7a5ab	30	#if 0
864a4b6e	31	#ifndef lint
440bd198	32	static char sccsid[] = "@(#)spec.c 8.1 (Berkeley) 6/6/93";
44a7a5ab	33	#endif /* not lint */
864a4b6e A	34	#endif
	35	#include <sys/cdefs.h>
	36	__FBSDID("$FreeBSD: src/usr.sbin/mtree/spec.c,v 1.22 2005/03/29 11:44:17 tobez Exp $");
44a7a5ab A	37
	38	#include <sys/types.h>
	39	#include <sys/stat.h>
	40	#include <ctype.h>
440bd198	41	#include <err.h>
44a7a5ab A	42	#include <errno.h>
	43	#include <fts.h>
	44	#include <grp.h>
	45	#include <pwd.h>
	46	#include <stdio.h>
864a4b6e	47	#include <stdint.h>
44a7a5ab	48	#include <unistd.h>
440bd198	49	#include <vis.h>
44a7a5ab A	50	#include "mtree.h"
	51	#include "extern.h"
	52
	53	int lineno; /* Current spec line number. */
	54
864a4b6e A	55	static void set(char , NODE );
864a4b6e A	56	static void unset(char , NODE );
44a7a5ab A	57
44a7a5ab A	58	NODE *
864a4b6e	59	mtree_readspec(FILE *fi)
44a7a5ab	60	{
864a4b6e A	61	NODE centry, last;
864a4b6e A	62	char *p;
44a7a5ab A	63	NODE ginfo, *root;
	64	int c_cur, c_next;
	65	char buf[2048];
	66
440bd198 A	67	centry = last = root = NULL;
440bd198 A	68	bzero(&ginfo, sizeof(ginfo));
44a7a5ab	69	c_cur = c_next = 0;
864a4b6e	70	for (lineno = 1; fgets(buf, sizeof(buf), fi);
44a7a5ab A	71	++lineno, c_cur = c_next, c_next = 0) {
	72	/* Skip empty lines. */
	73	if (buf[0] == '\n')
	74	continue;
	75
	76	/* Find end of line. */
440bd198 A	77	if ((p = index(buf, '\n')) == NULL)
440bd198 A	78	errx(1, "line %d too long", lineno);
44a7a5ab A	79
	80	/* See if next line is continuation line. */
	81	if (p[-1] == '\\') {
	82	--p;
	83	c_next = 1;
	84	}
	85
	86	/* Null-terminate the line. */
	87	*p = '\0';
	88
	89	/* Skip leading whitespace. */
	90	for (p = buf; p && isspace(p); ++p);
	91
	92	/* If nothing but whitespace or comment char, continue. */
	93	if (!p \|\| p == '#')
	94	continue;
	95
	96	#ifdef DEBUG
	97	(void)fprintf(stderr, "line %d: {%s}\n", lineno, p);
	98	#endif
	99	if (c_cur) {
	100	set(p, centry);
	101	continue;
	102	}
440bd198	103
44a7a5ab A	104	/* Grab file name, "$", "set", or "unset". */
44a7a5ab A	105	if ((p = strtok(p, "\n\t ")) == NULL)
440bd198	106	errx(1, "line %d: missing field", lineno);
44a7a5ab A	107
	108	if (p[0] == '/')
	109	switch(p[1]) {
	110	case 's':
	111	if (strcmp(p + 1, "set"))
	112	break;
	113	set(NULL, &ginfo);
	114	continue;
	115	case 'u':
	116	if (strcmp(p + 1, "unset"))
	117	break;
	118	unset(NULL, &ginfo);
	119	continue;
	120	}
	121
440bd198 A	122	if (index(p, '/'))
	123	errx(1, "line %d: slash character in file name",
	124	lineno);
44a7a5ab A	125
	126	if (!strcmp(p, "..")) {
	127	/* Don't go up, if haven't gone down. */
	128	if (!root)
	129	goto noparent;
	130	if (last->type != F_DIR \|\| last->flags & F_DONE) {
	131	if (last == root)
	132	goto noparent;
	133	last = last->parent;
	134	}
	135	last->flags \|= F_DONE;
	136	continue;
	137
440bd198	138	noparent: errx(1, "line %d: no parent node", lineno);
44a7a5ab A	139	}
	140
	141	if ((centry = calloc(1, sizeof(NODE) + strlen(p))) == NULL)
440bd198	142	errx(1, "calloc");
44a7a5ab	143	*centry = ginfo;
44a7a5ab A	144	#define MAGIC "?*["
	145	if (strpbrk(p, MAGIC))
	146	centry->flags \|= F_MAGIC;
864a4b6e A	147	if (strunvis(centry->name, p) == -1)
864a4b6e A	148	errx(1, "filename %s is ill-encoded", p);
44a7a5ab A	149	set(NULL, centry);
	150
	151	if (!root) {
	152	last = root = centry;
	153	root->parent = root;
	154	} else if (last->type == F_DIR && !(last->flags & F_DONE)) {
	155	centry->parent = last;
	156	last = last->child = centry;
	157	} else {
	158	centry->parent = last->parent;
	159	centry->prev = last;
	160	last = last->next = centry;
	161	}
	162	}
	163	return (root);
	164	}
	165
	166	static void
864a4b6e	167	set(char t, NODE ip)
44a7a5ab	168	{
864a4b6e	169	int type;
440bd198	170	char kw, val = NULL;
44a7a5ab A	171	struct group *gr;
	172	struct passwd *pw;
	173	mode_t *m;
	174	int value;
	175	char *ep;
	176
440bd198	177	for (; (kw = strtok(t, "= \t\n")); t = NULL) {
44a7a5ab A	178	ip->flags \|= type = parsekey(kw, &value);
44a7a5ab A	179	if (value && (val = strtok(NULL, " \t\n")) == NULL)
440bd198	180	errx(1, "line %d: missing value", lineno);
44a7a5ab A	181	switch(type) {
	182	case F_CKSUM:
	183	ip->cksum = strtoul(val, &ep, 10);
	184	if (*ep)
440bd198 A	185	errx(1, "line %d: invalid checksum %s",
	186	lineno, val);
	187	break;
	188	case F_MD5:
	189	ip->md5digest = strdup(val);
864a4b6e	190	if(!ip->md5digest)
440bd198	191	errx(1, "strdup");
44a7a5ab	192	break;
440bd198 A	193	case F_SHA1:
440bd198 A	194	ip->sha1digest = strdup(val);
864a4b6e A	195	if(!ip->sha1digest)
	196	errx(1, "strdup");
	197	break;
	198	case F_SHA256:
	199	ip->sha256digest = strdup(val);
	200	if(!ip->sha256digest)
440bd198	201	errx(1, "strdup");
440bd198 A	202	break;
	203	case F_RMD160:
	204	ip->rmd160digest = strdup(val);
864a4b6e	205	if(!ip->rmd160digest)
440bd198	206	errx(1, "strdup");
440bd198 A	207	break;
	208	case F_FLAGS:
	209	if (strcmp("none", val) == 0)
	210	ip->st_flags = 0;
	211	else if (strtofflags(&val, &ip->st_flags, NULL) != 0)
	212	errx(1, "line %d: invalid flag %s",lineno, val);
	213	break;
44a7a5ab	214	case F_GID:
4d0bb651	215	ip->st_gid = (gid_t)strtoul(val, &ep, 10);
44a7a5ab	216	if (*ep)
440bd198	217	errx(1, "line %d: invalid gid %s", lineno, val);
44a7a5ab A	218	break;
	219	case F_GNAME:
	220	if ((gr = getgrnam(val)) == NULL)
440bd198	221	errx(1, "line %d: unknown group %s", lineno, val);
44a7a5ab A	222	ip->st_gid = gr->gr_gid;
	223	break;
	224	case F_IGN:
	225	/* just set flag bit */
	226	break;
	227	case F_MODE:
	228	if ((m = setmode(val)) == NULL)
440bd198 A	229	errx(1, "line %d: invalid file mode %s",
440bd198 A	230	lineno, val);
44a7a5ab A	231	ip->st_mode = getmode(m, 0);
	232	free(m);
	233	break;
	234	case F_NLINK:
440bd198	235	ip->st_nlink = strtoul(val, &ep, 10);
44a7a5ab	236	if (*ep)
440bd198 A	237	errx(1, "line %d: invalid link count %s",
440bd198 A	238	lineno, val);
44a7a5ab A	239	break;
44a7a5ab A	240	case F_SIZE:
440bd198	241	ip->st_size = strtoq(val, &ep, 10);
44a7a5ab	242	if (*ep)
440bd198 A	243	errx(1, "line %d: invalid size %s",
440bd198 A	244	lineno, val);
44a7a5ab A	245	break;
44a7a5ab A	246	case F_SLINK:
864a4b6e A	247	ip->slink = malloc(strlen(val) + 1);
	248	if (ip->slink == NULL)
	249	errx(1, "malloc");
	250	if (strunvis(ip->slink, val) == -1)
	251	errx(1, "symlink %s is ill-encoded", val);
44a7a5ab A	252	break;
44a7a5ab A	253	case F_TIME:
440bd198	254	ip->st_mtimespec.tv_sec = strtoul(val, &ep, 10);
44a7a5ab	255	if (*ep != '.')
440bd198 A	256	errx(1, "line %d: invalid time %s",
440bd198 A	257	lineno, val);
44a7a5ab	258	val = ep + 1;
440bd198	259	ip->st_mtimespec.tv_nsec = strtoul(val, &ep, 10);
44a7a5ab	260	if (*ep)
440bd198 A	261	errx(1, "line %d: invalid time %s",
440bd198 A	262	lineno, val);
44a7a5ab A	263	break;
	264	case F_TYPE:
	265	switch(*val) {
	266	case 'b':
	267	if (!strcmp(val, "block"))
	268	ip->type = F_BLOCK;
	269	break;
	270	case 'c':
	271	if (!strcmp(val, "char"))
	272	ip->type = F_CHAR;
	273	break;
	274	case 'd':
	275	if (!strcmp(val, "dir"))
	276	ip->type = F_DIR;
	277	break;
	278	case 'f':
	279	if (!strcmp(val, "file"))
	280	ip->type = F_FILE;
	281	if (!strcmp(val, "fifo"))
	282	ip->type = F_FIFO;
	283	break;
	284	case 'l':
	285	if (!strcmp(val, "link"))
	286	ip->type = F_LINK;
	287	break;
	288	case 's':
	289	if (!strcmp(val, "socket"))
	290	ip->type = F_SOCK;
	291	break;
	292	default:
440bd198 A	293	errx(1, "line %d: unknown file type %s",
440bd198 A	294	lineno, val);
44a7a5ab A	295	}
	296	break;
	297	case F_UID:
4d0bb651	298	ip->st_uid = (uid_t)strtoul(val, &ep, 10);
44a7a5ab	299	if (*ep)
440bd198	300	errx(1, "line %d: invalid uid %s", lineno, val);
44a7a5ab A	301	break;
	302	case F_UNAME:
	303	if ((pw = getpwnam(val)) == NULL)
440bd198	304	errx(1, "line %d: unknown user %s", lineno, val);
44a7a5ab A	305	ip->st_uid = pw->pw_uid;
	306	break;
	307	}
	308	}
	309	}
	310
	311	static void
864a4b6e	312	unset(char t, NODE ip)
44a7a5ab	313	{
864a4b6e	314	char *p;
44a7a5ab	315
440bd198	316	while ((p = strtok(t, "\n\t ")))
44a7a5ab A	317	ip->flags &= ~parsekey(p, NULL);
44a7a5ab A	318	}