usr.sbin/makefs/makefs.c

   1 /*      $NetBSD: makefs.c,v 1.20 2004/06/20 22:20:18 jmc Exp $  */
   2
   3 /*
   4  * Copyright (c) 2001-2003 Wasabi Systems, Inc.
   5  * All rights reserved.
   6  *
   7  * Written by Luke Mewburn for Wasabi Systems, Inc.
   8  *
   9  * Redistribution and use in source and binary forms, with or without
  10  * modification, are permitted provided that the following conditions
  11  * are met:
  12  * 1. Redistributions of source code must retain the above copyright
  13  *    notice, this list of conditions and the following disclaimer.
  14  * 2. Redistributions in binary form must reproduce the above copyright
  15  *    notice, this list of conditions and the following disclaimer in the
  16  *    documentation and/or other materials provided with the distribution.
  17  * 3. All advertising materials mentioning features or use of this software
  18  *    must display the following acknowledgement:
  19  *      This product includes software developed for the NetBSD Project by
  20  *      Wasabi Systems, Inc.
  21  * 4. The name of Wasabi Systems, Inc. may not be used to endorse
  22  *    or promote products derived from this software without specific prior
  23  *    written permission.
  24  *
  25  * THIS SOFTWARE IS PROVIDED BY WASABI SYSTEMS, INC. ``AS IS'' AND
  26  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
  27  * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
  28  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL WASABI SYSTEMS, INC
  29  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
  30  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
  31  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
  32  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
  33  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
  34  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
  35  * POSSIBILITY OF SUCH DAMAGE.
  36  */
  37
  38 #include <sys/cdefs.h>
  39 __FBSDID("$FreeBSD$");
  40
  41 #include <assert.h>
  42 #include <ctype.h>
  43 #include <errno.h>
  44 #include <limits.h>
  45 #include <stdio.h>
  46 #include <stdlib.h>
  47 #include <string.h>
  48 #include <unistd.h>
  49
  50 #include "makefs.h"
  51 #include "mtree.h"
  52
  53 /*
  54  * list of supported file systems and dispatch functions
  55  */
  56 typedef struct {
  57         const char      *type;
  58         int             (*parse_options)(const char *, fsinfo_t *);
  59         void            (*make_fs)(const char *, const char *, fsnode *,
  60                                 fsinfo_t *);
  61 } fstype_t;
  62
  63 static fstype_t fstypes[] = {
  64         { "ffs",        ffs_parse_opts,         ffs_makefs },
  65         { NULL  },
  66 };
  67
  68 u_int           debug;
  69 struct timespec start_time;
  70
  71 static  fstype_t *get_fstype(const char *);
  72 static  void    usage(void);
  73 int             main(int, char *[]);
  74
  75 int
  76 main(int argc, char *argv[])
  77 {
  78         struct timeval   start;
  79         fstype_t        *fstype;
  80         fsinfo_t         fsoptions;
  81         fsnode          *root;
  82         int              ch, len;
  83         char            *specfile;
  84
  85         setprogname(argv[0]);
  86
  87         debug = 0;
  88         if ((fstype = get_fstype(DEFAULT_FSTYPE)) == NULL)
  89                 errx(1, "Unknown default fs type `%s'.", DEFAULT_FSTYPE);
  90
  91                 /* set default fsoptions */
  92         (void)memset(&fsoptions, 0, sizeof(fsoptions));
  93         fsoptions.fd = -1;
  94         fsoptions.sectorsize = -1;
  95         fsoptions.bsize= -1;
  96         fsoptions.fsize= -1;
  97         fsoptions.cpg= -1;
  98         fsoptions.density= -1;
  99         fsoptions.minfree= -1;
 100         fsoptions.optimization= -1;
 101         fsoptions.maxcontig= -1;
 102         fsoptions.maxbpg= -1;
 103         fsoptions.avgfilesize= -1;
 104         fsoptions.avgfpdir= -1;
 105         fsoptions.version = 1;
 106
 107         specfile = NULL;
 108         if (gettimeofday(&start, NULL) == -1)
 109                 err(1, "Unable to get system time");
 110
 111         start_time.tv_sec = start.tv_sec;
 112         start_time.tv_nsec = start.tv_usec * 1000;
 113
 114         while ((ch = getopt(argc, argv, "B:b:d:f:F:M:m:N:o:s:S:t:x")) != -1) {
 115                 switch (ch) {
 116
 117                 case 'B':
 118                         if (strcmp(optarg, "be") == 0 ||
 119                             strcmp(optarg, "4321") == 0 ||
 120                             strcmp(optarg, "big") == 0) {
 121 #if BYTE_ORDER == LITTLE_ENDIAN
 122                                 fsoptions.needswap = 1;
 123 #endif
 124                         } else if (strcmp(optarg, "le") == 0 ||
 125                             strcmp(optarg, "1234") == 0 ||
 126                             strcmp(optarg, "little") == 0) {
 127 #if BYTE_ORDER == BIG_ENDIAN
 128                                 fsoptions.needswap = 1;
 129 #endif
 130                         } else {
 131                                 warnx("Invalid endian `%s'.", optarg);
 132                                 usage();
 133                         }
 134                         break;
 135
 136                 case 'b':
 137                         len = strlen(optarg) - 1;
 138                         if (optarg[len] == '%') {
 139                                 optarg[len] = '\0';
 140                                 fsoptions.freeblockpc =
 141                                     strsuftoll("free block percentage",
 142                                         optarg, 0, 99);
 143                         } else {
 144                                 fsoptions.freeblocks =
 145                                     strsuftoll("free blocks",
 146                                         optarg, 0, LLONG_MAX);
 147                         }
 148                         break;
 149
 150                 case 'd':
 151                         debug =
 152                             (int)strsuftoll("debug mask", optarg, 0, UINT_MAX);
 153                         break;
 154
 155                 case 'f':
 156                         len = strlen(optarg) - 1;
 157                         if (optarg[len] == '%') {
 158                                 optarg[len] = '\0';
 159                                 fsoptions.freefilepc =
 160                                     strsuftoll("free file percentage",
 161                                         optarg, 0, 99);
 162                         } else {
 163                                 fsoptions.freefiles =
 164                                     strsuftoll("free files",
 165                                         optarg, 0, LLONG_MAX);
 166                         }
 167                         break;
 168
 169                 case 'F':
 170                         specfile = optarg;
 171                         break;
 172
 173                 case 'M':
 174                         fsoptions.minsize =
 175                             strsuftoll("minimum size", optarg, 1LL, LLONG_MAX);
 176                         break;
 177
 178                 case 'N':
 179                         if (! setup_getid(optarg))
 180                                 errx(1,
 181                             "Unable to use user and group databases in `%s'",
 182                                     optarg);
 183                         break;
 184
 185                 case 'm':
 186                         fsoptions.maxsize =
 187                             strsuftoll("maximum size", optarg, 1LL, LLONG_MAX);
 188                         break;
 189
 190                 case 'o':
 191                 {
 192                         char *p;
 193
 194                         while ((p = strsep(&optarg, ",")) != NULL) {
 195                                 if (*p == '\0')
 196                                         errx(1, "Empty option");
 197                                 if (! fstype->parse_options(p, &fsoptions))
 198                                         usage();
 199                         }
 200                         break;
 201                 }
 202
 203                 case 's':
 204                         fsoptions.minsize = fsoptions.maxsize =
 205                             strsuftoll("size", optarg, 1LL, LLONG_MAX);
 206                         break;
 207
 208                 case 'S':
 209                         fsoptions.sectorsize =
 210                             (int)strsuftoll("sector size", optarg,
 211                                 1LL, INT_MAX);
 212                         break;
 213
 214                 case 't':
 215                         if ((fstype = get_fstype(optarg)) == NULL)
 216                                 errx(1, "Unknown fs type `%s'.", optarg);
 217                         break;
 218
 219                 case 'x':
 220                         fsoptions.onlyspec = 1;
 221                         break;
 222
 223                 case '?':
 224                 default:
 225                         usage();
 226                         /* NOTREACHED */
 227
 228                 }
 229         }
 230         if (debug) {
 231                 printf("debug mask: 0x%08x\n", debug);
 232                 printf("start time: %ld.%ld, %s",
 233                     (long)start_time.tv_sec, (long)start_time.tv_nsec,
 234                     ctime(&start_time.tv_sec));
 235         }
 236         argc -= optind;
 237         argv += optind;
 238
 239         if (argc != 2)
 240                 usage();
 241
 242         /* -x must be accompanied by -F */
 243         if (fsoptions.onlyspec != 0 && specfile == NULL)
 244                 errx(1, "-x requires -F mtree-specfile.");
 245
 246                                 /* walk the tree */
 247         TIMER_START(start);
 248         root = walk_dir(argv[1], NULL);
 249         TIMER_RESULTS(start, "walk_dir");
 250
 251         if (specfile) {         /* apply a specfile */
 252                 TIMER_START(start);
 253                 apply_specfile(specfile, argv[1], root);
 254                 TIMER_RESULTS(start, "apply_specfile");
 255         }
 256
 257         if (debug & DEBUG_DUMP_FSNODES) {
 258                 printf("\nparent: %s\n", argv[1]);
 259                 dump_fsnodes(".", root);
 260                 putchar('\n');
 261         }
 262
 263                                 /* build the file system */
 264         TIMER_START(start);
 265         fstype->make_fs(argv[0], argv[1], root, &fsoptions);
 266         TIMER_RESULTS(start, "make_fs");
 267
 268         exit(0);
 269         /* NOTREACHED */
 270 }
 271
 272
 273 int
 274 set_option(option_t *options, const char *var, const char *val)
 275 {
 276         int     i;
 277
 278         for (i = 0; options[i].name != NULL; i++) {
 279                 if (strcmp(options[i].name, var) != 0)
 280                         continue;
 281                 *options[i].value = (int)strsuftoll(options[i].desc, val,
 282                     options[i].minimum, options[i].maximum);
 283                 return (1);
 284         }
 285         warnx("Unknown option `%s'", var);
 286         return (0);
 287 }
 288
 289
 290 static fstype_t *
 291 get_fstype(const char *type)
 292 {
 293         int i;
 294
 295         for (i = 0; fstypes[i].type != NULL; i++)
 296                 if (strcmp(fstypes[i].type, type) == 0)
 297                         return (&fstypes[i]);
 298         return (NULL);
 299 }
 300
 301 static void
 302 usage(void)
 303 {
 304         const char *prog;
 305
 306         prog = getprogname();
 307         fprintf(stderr,
 308 "usage: %s [-t fs-type] [-o fs-options] [-d debug-mask] [-B endian]\n"
 309 "\t[-S sector-size] [-M minimum-size] [-m maximum-size] [-s image-size]\n"
 310 "\t[-b free-blocks] [-f free-files] [-F mtree-specfile] [-x]\n"
 311 "\t[-N userdb-dir] image-file directory\n",
 312             prog);
 313         exit(1);
 314 }