usr.sbin/mtree/mtree.c

   1 /*      $NetBSD: mtree.c,v 1.50 2015/01/23 02:27:01 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.50 2015/01/23 02:27:01 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, dflag, eflag, iflag, jflag, lflag, mflag, nflag, qflag, rflag,
  63         sflag, tflag, uflag;
  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         int     cflag, Cflag, Dflag, Uflag, wflag;
  83         char    *dir, *p;
  84         FILE    *spec1, *spec2;
  85
  86         setprogname(argv[0]);
  87
  88         cflag = Cflag = Dflag = Uflag = wflag = 0;
  89         dir = NULL;
  90         init_excludes();
  91         spec1 = stdin;
  92         spec2 = NULL;
  93
  94         while ((ch = getopt(argc, argv,
  95             "bcCdDeE:f:F:I:ijk:K:lLmMnN:O:p:PqrR:s:StuUwWxX:"))
  96             != -1) {
  97                 switch((char)ch) {
  98                 case 'b':
  99                         bflag = 1;
 100                         break;
 101                 case 'c':
 102                         cflag = 1;
 103                         break;
 104                 case 'C':
 105                         Cflag = 1;
 106                         break;
 107                 case 'd':
 108                         dflag = 1;
 109                         break;
 110                 case 'D':
 111                         Dflag = 1;
 112                         break;
 113                 case 'E':
 114                         parsetags(&excludetags, optarg);
 115                         break;
 116                 case 'e':
 117                         eflag = 1;
 118                         break;
 119                 case 'f':
 120                         if (spec1 == stdin) {
 121                                 spec1 = fopen(optarg, "r");
 122                                 if (spec1 == NULL)
 123                                         mtree_err("%s: %s", optarg,
 124                                             strerror(errno));
 125                         } else if (spec2 == NULL) {
 126                                 spec2 = fopen(optarg, "r");
 127                                 if (spec2 == NULL)
 128                                         mtree_err("%s: %s", optarg,
 129                                             strerror(errno));
 130                         } else
 131                                 usage();
 132                         break;
 133                 case 'F':
 134                         for (i = 0; i < __arraycount(flavors); i++)
 135                                 if (strcmp(optarg, flavors[i].name) == 0) {
 136                                         flavor = flavors[i].flavor;
 137                                         break;
 138                                 }
 139                         if (i == __arraycount(flavors))
 140                                 usage();
 141                         break;
 142                 case 'i':
 143                         iflag = 1;
 144                         break;
 145                 case 'I':
 146                         parsetags(&includetags, optarg);
 147                         break;
 148                 case 'j':
 149                         jflag = 1;
 150                         break;
 151                 case 'k':
 152                         keys = F_TYPE;
 153                         while ((p = strsep(&optarg, " \t,")) != NULL)
 154                                 if (*p != '\0')
 155                                         keys |= parsekey(p, NULL);
 156                         break;
 157                 case 'K':
 158                         while ((p = strsep(&optarg, " \t,")) != NULL)
 159                                 if (*p != '\0')
 160                                         keys |= parsekey(p, NULL);
 161                         break;
 162                 case 'l':
 163                         lflag = 1;
 164                         break;
 165                 case 'L':
 166                         ftsoptions &= ~FTS_PHYSICAL;
 167                         ftsoptions |= FTS_LOGICAL;
 168                         break;
 169                 case 'm':
 170                         mflag = 1;
 171                         break;
 172                 case 'M':
 173                         mtree_Mflag = 1;
 174                         break;
 175                 case 'n':
 176                         nflag = 1;
 177                         break;
 178                 case 'N':
 179                         if (! setup_getid(optarg))
 180                                 mtree_err(
 181                             "Unable to use user and group databases in `%s'",
 182                                     optarg);
 183                         break;
 184                 case 'O':
 185                         load_only(optarg);
 186                         break;
 187                 case 'p':
 188                         dir = optarg;
 189                         break;
 190                 case 'P':
 191                         ftsoptions &= ~FTS_LOGICAL;
 192                         ftsoptions |= FTS_PHYSICAL;
 193                         break;
 194                 case 'q':
 195                         qflag = 1;
 196                         break;
 197                 case 'r':
 198                         rflag++;
 199                         break;
 200                 case 'R':
 201                         while ((p = strsep(&optarg, " \t,")) != NULL)
 202                                 if (*p != '\0')
 203                                         keys &= ~parsekey(p, NULL);
 204                         break;
 205                 case 's':
 206                         sflag = 1;
 207                         crc_total = ~strtol(optarg, &p, 0);
 208                         if (*p)
 209                                 mtree_err("illegal seed value -- %s", optarg);
 210                         break;
 211                 case 'S':
 212                         mtree_Sflag = 1;
 213                         break;
 214                 case 't':
 215                         tflag = 1;
 216                         break;
 217                 case 'u':
 218                         uflag = 1;
 219                         break;
 220                 case 'U':
 221                         Uflag = uflag = 1;
 222                         break;
 223                 case 'w':
 224                         wflag = 1;
 225                         break;
 226                 case 'W':
 227                         mtree_Wflag = 1;
 228                         break;
 229                 case 'x':
 230                         ftsoptions |= FTS_XDEV;
 231                         break;
 232                 case 'X':
 233                         read_excludes_file(optarg);
 234                         break;
 235                 case '?':
 236                 default:
 237                         usage();
 238                 }
 239         }
 240         argc -= optind;
 241         argv += optind;
 242
 243         if (argc)
 244                 usage();
 245
 246         switch (flavor) {
 247         case F_FREEBSD9:
 248                 if (cflag && iflag) {
 249                         warnx("-c and -i passed, replacing -i with -j for "
 250                             "FreeBSD compatibility");
 251                         iflag = 0;
 252                         jflag = 1;
 253                 }
 254                 if (dflag && !bflag) {
 255                         warnx("Adding -b to -d for FreeBSD compatibility");
 256                         bflag = 1;
 257                 }
 258                 if (uflag && !iflag) {
 259                         warnx("Adding -i to -%c for FreeBSD compatibility",
 260                             Uflag ? 'U' : 'u');
 261                         iflag = 1;
 262                 }
 263                 if (uflag && !tflag) {
 264                         warnx("Adding -t to -%c for FreeBSD compatibility",
 265                             Uflag ? 'U' : 'u');
 266                         tflag = 1;
 267                 }
 268                 if (wflag)
 269                         warnx("The -w flag is a no-op");
 270                 break;
 271         default:
 272                 if (wflag)
 273                         usage();
 274         }
 275
 276         if (spec2 && (cflag || Cflag || Dflag))
 277                 mtree_err("Double -f, -c, -C and -D flags are mutually "
 278                     "exclusive");
 279
 280         if (dir && spec2)
 281                 mtree_err("Double -f and -p flags are mutually exclusive");
 282
 283         if (dir && chdir(dir))
 284                 mtree_err("%s: %s", dir, strerror(errno));
 285
 286         if ((cflag || sflag) && !getcwd(fullpath, sizeof(fullpath)))
 287                 mtree_err("%s", strerror(errno));
 288
 289         if ((cflag && Cflag) || (cflag && Dflag) || (Cflag && Dflag))
 290                 mtree_err("-c, -C and -D flags are mutually exclusive");
 291
 292         if (iflag && mflag)
 293                 mtree_err("-i and -m flags are mutually exclusive");
 294
 295         if (lflag && uflag)
 296                 mtree_err("-l and -u flags are mutually exclusive");
 297
 298         if (cflag) {
 299                 cwalk(stdout);
 300                 exit(0);
 301         }
 302         if (Cflag || Dflag) {
 303                 dump_nodes(stdout, "", spec(spec1), Dflag);
 304                 exit(0);
 305         }
 306         if (spec2 != NULL)
 307                 status = mtree_specspec(spec1, spec2);
 308         else
 309                 status = verify(spec1);
 310         if (Uflag && (status == MISMATCHEXIT))
 311                 status = 0;
 312         exit(status);
 313 }
 314
 315 static void
 316 usage(void)
 317 {
 318         unsigned int i;
 319
 320         fprintf(stderr,
 321             "usage: %s [-bCcDdejLlMnPqrStUuWx] [-i|-m] [-E tags]\n"
 322             "\t\t[-f spec] [-f spec]\n"
 323             "\t\t[-I tags] [-K keywords] [-k keywords] [-N dbdir] [-p path]\n"
 324             "\t\t[-R keywords] [-s seed] [-X exclude-file]\n"
 325             "\t\t[-F flavor]\n",
 326             getprogname());
 327         fprintf(stderr, "\nflavors:");
 328         for (i = 0; i < __arraycount(flavors); i++)
 329                 fprintf(stderr, " %s", flavors[i].name);
 330         fprintf(stderr, "\n");
 331         exit(1);
 332 }