NetBSD/usr.sbin/mtree/spec.c

/*-
 * Copyright (c) 1989 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. All advertising materials mentioning features or use of this software
 *    must display the following acknowledgement:
 *	This product includes software developed by the University of
 *	California, Berkeley and its contributors.
 * 4. 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.
 */

#ifndef lint
/*static char sccsid[] = "from: @(#)spec.c	5.14 (Berkeley) 3/2/91";*/
static char rcsid[] = "$Id: spec.c,v 1.2 1993/08/01 17:58:24 mycroft Exp $";
#endif /* not lint */

#include <sys/types.h>
#include <pwd.h>
#include <grp.h>
#include <stdio.h>
#include <errno.h>
#include <ctype.h>
#include "mtree.h"

extern NODE *root;			/* root of the tree */

static int lineno;			/* current spec line number */

spec()
{
	register NODE *centry, *last;
	register char *p;
	NODE ginfo, *emalloc();
	char buf[2048];

	bzero((void *)&ginfo, sizeof(ginfo));
	for (lineno = 1; fgets(buf, sizeof(buf), stdin); ++lineno) {
		if (!(p = index(buf, '\n'))) {
			(void)fprintf(stderr,
			    "mtree: line %d too long.\n", lineno);
			exit(1);
		}
		*p = '\0';
		for (p = buf; *p && isspace(*p); ++p);
		if (!*p || *p == '#')
			continue;

		/* grab file name, "$", "set", or "unset" */
		if (!(p = strtok(p, "\n\t ")))
			specerr();

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

		if (index(p, '/')) {
			(void)fprintf(stderr,
			    "mtree: file names may not contain slashes.\n");
			specerr();
		}

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

		centry = emalloc(sizeof(NODE) + strlen(p));
		*centry = ginfo;
		(void)strcpy(centry->name, p);
#define	MAGIC	"?*["
		if (strpbrk(p, MAGIC))
			centry->flags |= F_MAGIC;
		set(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;
		}
	}
}

set(ip)
	register NODE *ip;
{
	register int type;
	register char *kw, *val;
	gid_t getgroup();
	uid_t getowner();
	long atol(), strtol();

	while (kw = strtok((char *)NULL, "= \t\n")) {
		ip->flags |= type = key(kw);
		val = strtok((char *)NULL, " \t\n");
		if (!val)
			specerr();
		switch(type) {
		case F_CKSUM:
			ip->cksum = atol(val);
			break;
		case F_GROUP:
			ip->st_gid = getgroup(val);
			break;
		case F_IGN:
			/* just set flag bit */
			break;
		case F_MODE: {
			mode_t *m, *setmode();

			if (!(m = setmode(val))) {
				(void)fprintf(stderr,
				    "mtree: invalid file mode %s.\n", val);
				specerr();
			}
			ip->st_mode = getmode(m, 0);
			break;
		}
		case F_NLINK:
			ip->st_nlink = atoi(val);
			break;
		case F_OWNER:
			ip->st_uid = getowner(val);
			break;
		case F_SIZE:
			ip->st_size = atol(val);
			break;
		case F_SLINK:
			if (!(ip->slink = strdup(val)))
				nomem();
			break;
		case F_TIME:
			ip->st_mtime = atol(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:
				(void)fprintf(stderr,
				    "mtree: unknown file type %s.\n", val);
				specerr();
			}
			break;
		}
	}
}

unset(ip)
	register NODE *ip;
{
	register char *p;

	while (p = strtok((char *)NULL, "\n\t "))
		ip->flags &= ~key(p);
}

key(p)
	char *p;
{
	switch(*p) {
	case 'c':
		if (!strcmp(p, "cksum"))
			return(F_CKSUM);
		break;
	case 'g':
		if (!strcmp(p, "group"))
			return(F_GROUP);
		break;
	case 'i':
		if (!strcmp(p, "ignore"))
			return(F_IGN);
		break;
	case 'l':
		if (!strcmp(p, "link"))
			return(F_SLINK);
		break;
	case 'm':
		if (!strcmp(p, "mode"))
			return(F_MODE);
		break;
	case 'n':
		if (!strcmp(p, "nlink"))
			return(F_NLINK);
		break;
	case 'o':
		if (!strcmp(p, "owner"))
			return(F_OWNER);
		break;
	case 's':
		if (!strcmp(p, "size"))
			return(F_SIZE);
		break;
	case 't':
		if (!strcmp(p, "type"))
			return(F_TYPE);
		if (!strcmp(p, "time"))
			return(F_TIME);
		break;
	}
	(void)fprintf(stderr, "mtree: unknown keyword %s.\n", p);
	specerr();
	/* NOTREACHED */
}


uid_t
getowner(p)
	register char *p;
{
	struct passwd *pw;
	int val;

	if (isdigit(*p)) {
		if ((val = atoi(p)) >= 0)
			return((uid_t)val);
		(void)fprintf(stderr, "mtree: illegal uid value %s.\n", p);
	} else if (pw = getpwnam(p))
		return(pw->pw_uid);
	else
		(void)fprintf(stderr, "mtree: unknown user %s.\n", p);
	specerr();
	/* NOTREACHED */
}

gid_t
getgroup(p)
	register char *p;
{
	struct group *gr;
	int val;

	if (isdigit(*p)) {
		if ((val = atoi(p)) >= 0)
			return((gid_t)val);
		(void)fprintf(stderr, "mtree: illegal gid value %s.\n", p);
	} else if (gr = getgrnam(p))
		return(gr->gr_gid);
	else
		(void)fprintf(stderr, "mtree: unknown group %s.\n", p);
	specerr();
	/* NOTREACHED */
}

noparent()
{
	(void)fprintf(stderr, "mtree: no parent node.\n");
	specerr();
}

specerr()
{
	(void)fprintf(stderr,
	    "mtree: line %d of the specification is incorrect.\n", lineno);
	exit(1);
}

NODE *
emalloc(size)
	int size;
{
	void *p;

	/* NOSTRICT */
	if (!(p = malloc((u_int)size)))
		nomem();
	bzero(p, size);
	return((NODE *)p);
}

nomem()
{
	(void)fprintf(stderr, "mtree: %s.\n", strerror(ENOMEM));
	exit(1);
}