Home | History | Annotate | Line # | Download | only in mtree
mtree.c revision 1.47
      1 /*	$NetBSD: mtree.c,v 1.47 2013/02/03 19:15:17 christos Exp $	*/
      2 
      3 /*-
      4  * Copyright (c) 1989, 1990, 1993
      5  *	The Regents of the University of California.  All rights reserved.
      6  *
      7  * Redistribution and use in source and binary forms, with or without
      8  * modification, are permitted provided that the following conditions
      9  * are met:
     10  * 1. Redistributions of source code must retain the above copyright
     11  *    notice, this list of conditions and the following disclaimer.
     12  * 2. Redistributions in binary form must reproduce the above copyright
     13  *    notice, this list of conditions and the following disclaimer in the
     14  *    documentation and/or other materials provided with the distribution.
     15  * 3. Neither the name of the University nor the names of its contributors
     16  *    may be used to endorse or promote products derived from this software
     17  *    without specific prior written permission.
     18  *
     19  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
     20  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
     21  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
     22  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
     23  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
     24  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
     25  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
     26  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
     27  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
     28  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
     29  * SUCH DAMAGE.
     30  */
     31 
     32 #if HAVE_NBTOOL_CONFIG_H
     33 #include "nbtool_config.h"
     34 #endif
     35 
     36 #include <sys/cdefs.h>
     37 #if defined(__COPYRIGHT) && !defined(lint)
     38 __COPYRIGHT("@(#) Copyright (c) 1989, 1990, 1993\
     39  The Regents of the University of California.  All rights reserved.");
     40 #endif /* not lint */
     41 
     42 #if defined(__RCSID) && !defined(lint)
     43 #if 0
     44 static char sccsid[] = "@(#)mtree.c	8.1 (Berkeley) 6/6/93";
     45 #else
     46 __RCSID("$NetBSD: mtree.c,v 1.47 2013/02/03 19:15:17 christos Exp $");
     47 #endif
     48 #endif /* not lint */
     49 
     50 #include <sys/param.h>
     51 #include <sys/stat.h>
     52 
     53 #include <errno.h>
     54 #include <stdio.h>
     55 #include <stdlib.h>
     56 #include <string.h>
     57 #include <unistd.h>
     58 
     59 #include "extern.h"
     60 
     61 int	ftsoptions = FTS_PHYSICAL;
     62 int	bflag, cflag, Cflag, dflag, Dflag, eflag, iflag, jflag, lflag, mflag,
     63     	nflag, qflag, rflag, sflag, tflag, uflag, Uflag, wflag;
     64 char	fullpath[MAXPATHLEN];
     65 
     66 static struct {
     67 	enum flavor flavor;
     68 	const char name[9];
     69 } flavors[] = {
     70 	{F_MTREE, "mtree"},
     71 	{F_FREEBSD9, "freebsd9"},
     72 	{F_NETBSD6, "netbsd6"},
     73 };
     74 
     75 __dead static	void	usage(void);
     76 
     77 int
     78 main(int argc, char **argv)
     79 {
     80 	int	ch, status;
     81 	unsigned int	i;
     82 	char	*dir, *p;
     83 	FILE	*spec1, *spec2;
     84 
     85 	setprogname(argv[0]);
     86 
     87 	dir = NULL;
     88 	init_excludes();
     89 	spec1 = stdin;
     90 	spec2 = NULL;
     91 
     92 	while ((ch = getopt(argc, argv,
     93 	    "bcCdDeE:f:F:I:ijk:K:lLmMnN:O:p:PqrR:s:StuUwWxX:"))
     94 	    != -1) {
     95 		switch((char)ch) {
     96 		case 'b':
     97 			bflag = 1;
     98 			break;
     99 		case 'c':
    100 			cflag = 1;
    101 			break;
    102 		case 'C':
    103 			Cflag = 1;
    104 			break;
    105 		case 'd':
    106 			dflag = 1;
    107 			break;
    108 		case 'D':
    109 			Dflag = 1;
    110 			break;
    111 		case 'E':
    112 			parsetags(&excludetags, optarg);
    113 			break;
    114 		case 'e':
    115 			eflag = 1;
    116 			break;
    117 		case 'f':
    118 			if (spec1 == stdin) {
    119 				spec1 = fopen(optarg, "r");
    120 				if (spec1 == NULL)
    121 					mtree_err("%s: %s", optarg,
    122 					    strerror(errno));
    123 			} else if (spec2 == NULL) {
    124 				spec2 = fopen(optarg, "r");
    125 				if (spec2 == NULL)
    126 					mtree_err("%s: %s", optarg,
    127 					    strerror(errno));
    128 			} else
    129 				usage();
    130 			break;
    131 		case 'F':
    132 			for (i = 0; i < __arraycount(flavors); i++)
    133 				if (strcmp(optarg, flavors[i].name) == 0) {
    134 					flavor = flavors[i].flavor;
    135 					break;
    136 				}
    137 			if (i == __arraycount(flavors))
    138 				usage();
    139 			break;
    140 		case 'i':
    141 			iflag = 1;
    142 			break;
    143 		case 'I':
    144 			parsetags(&includetags, optarg);
    145 			break;
    146 		case 'j':
    147 			jflag = 1;
    148 			break;
    149 		case 'k':
    150 			keys = F_TYPE;
    151 			while ((p = strsep(&optarg, " \t,")) != NULL)
    152 				if (*p != '\0')
    153 					keys |= parsekey(p, NULL);
    154 			break;
    155 		case 'K':
    156 			while ((p = strsep(&optarg, " \t,")) != NULL)
    157 				if (*p != '\0')
    158 					keys |= parsekey(p, NULL);
    159 			break;
    160 		case 'l':
    161 			lflag = 1;
    162 			break;
    163 		case 'L':
    164 			ftsoptions &= ~FTS_PHYSICAL;
    165 			ftsoptions |= FTS_LOGICAL;
    166 			break;
    167 		case 'm':
    168 			mflag = 1;
    169 			break;
    170 		case 'M':
    171 			mtree_Mflag = 1;
    172 			break;
    173 		case 'n':
    174 			nflag = 1;
    175 			break;
    176 		case 'N':
    177 			if (! setup_getid(optarg))
    178 				mtree_err(
    179 			    "Unable to use user and group databases in `%s'",
    180 				    optarg);
    181 			break;
    182 		case 'O':
    183 			load_only(optarg);
    184 			break;
    185 		case 'p':
    186 			dir = optarg;
    187 			break;
    188 		case 'P':
    189 			ftsoptions &= ~FTS_LOGICAL;
    190 			ftsoptions |= FTS_PHYSICAL;
    191 			break;
    192 		case 'q':
    193 			qflag = 1;
    194 			break;
    195 		case 'r':
    196 			rflag = 1;
    197 			break;
    198 		case 'R':
    199 			while ((p = strsep(&optarg, " \t,")) != NULL)
    200 				if (*p != '\0')
    201 					keys &= ~parsekey(p, NULL);
    202 			break;
    203 		case 's':
    204 			sflag = 1;
    205 			crc_total = ~strtol(optarg, &p, 0);
    206 			if (*p)
    207 				mtree_err("illegal seed value -- %s", optarg);
    208 			break;
    209 		case 'S':
    210 			mtree_Sflag = 1;
    211 			break;
    212 		case 't':
    213 			tflag = 1;
    214 			break;
    215 		case 'u':
    216 			uflag = 1;
    217 			break;
    218 		case 'U':
    219 			Uflag = uflag = 1;
    220 			break;
    221 		case 'w':
    222 			wflag = 1;
    223 			break;
    224 		case 'W':
    225 			mtree_Wflag = 1;
    226 			break;
    227 		case 'x':
    228 			ftsoptions |= FTS_XDEV;
    229 			break;
    230 		case 'X':
    231 			read_excludes_file(optarg);
    232 			break;
    233 		case '?':
    234 		default:
    235 			usage();
    236 		}
    237 	}
    238 	argc -= optind;
    239 	argv += optind;
    240 
    241 	if (argc)
    242 		usage();
    243 
    244 	switch (flavor) {
    245 	case F_FREEBSD9:
    246 		if (cflag && iflag) {
    247 			warnx("-c and -i passed, replacing -i with -j for "
    248 			    "FreeBSD compatibility");
    249 			iflag = 0;
    250 			jflag = 1;
    251 		}
    252 		if (dflag && !bflag) {
    253 			warnx("Adding -b to -d for FreeBSD compatibility");
    254 			bflag = 1;
    255 		}
    256 		if (uflag && !iflag) {
    257 			warnx("Adding -i to -%c for FreeBSD compatibility",
    258 			    Uflag ? 'U' : 'u');
    259 			iflag = 1;
    260 		}
    261 		if (uflag && !tflag) {
    262 			warnx("Adding -t to -%c for FreeBSD compatibility",
    263 			    Uflag ? 'U' : 'u');
    264 			tflag = 1;
    265 		}
    266 		if (wflag)
    267 			warnx("The -w flag is a no-op");
    268 		break;
    269 	default:
    270 		if (wflag)
    271 			usage();
    272 	}
    273 
    274 	if (spec2 && (cflag || Cflag || Dflag))
    275 		mtree_err("Double -f, -c, -C and -D flags are mutually "
    276 		    "exclusive");
    277 
    278 	if (dir && spec2)
    279 		mtree_err("Double -f and -p flags are mutually exclusive");
    280 
    281 	if (dir && chdir(dir))
    282 		mtree_err("%s: %s", dir, strerror(errno));
    283 
    284 	if ((cflag || sflag) && !getcwd(fullpath, sizeof(fullpath)))
    285 		mtree_err("%s", strerror(errno));
    286 
    287 	if ((cflag && Cflag) || (cflag && Dflag) || (Cflag && Dflag))
    288 		mtree_err("-c, -C and -D flags are mutually exclusive");
    289 
    290 	if (iflag && mflag)
    291 		mtree_err("-i and -m flags are mutually exclusive");
    292 
    293 	if (lflag && uflag)
    294 		mtree_err("-l and -u flags are mutually exclusive");
    295 
    296 	if (cflag) {
    297 		cwalk();
    298 		exit(0);
    299 	}
    300 	if (Cflag || Dflag) {
    301 		dump_nodes("", spec(spec1), Dflag);
    302 		exit(0);
    303 	}
    304 	if (spec2 != NULL)
    305 		status = mtree_specspec(spec1, spec2);
    306 	else
    307 		status = verify(spec1);
    308 	if (Uflag && (status == MISMATCHEXIT))
    309 		status = 0;
    310 	exit(status);
    311 }
    312 
    313 static void
    314 usage(void)
    315 {
    316 	unsigned int i;
    317 
    318 	fprintf(stderr,
    319 	    "usage: %s [-bCcDdejLlMnPqrStUuWx] [-i|-m] [-E tags]\n"
    320 	    "\t\t[-f spec] [-f spec]\n"
    321 	    "\t\t[-I tags] [-K keywords] [-k keywords] [-N dbdir] [-p path]\n"
    322 	    "\t\t[-R keywords] [-s seed] [-X exclude-file]\n"
    323 	    "\t\t[-F flavor]\n",
    324 	    getprogname());
    325 	fprintf(stderr, "\nflavors:");
    326 	for (i = 0; i < __arraycount(flavors); i++)
    327 		fprintf(stderr, " %s", flavors[i].name);
    328 	fprintf(stderr, "\n");
    329 	exit(1);
    330 }
    331