/*      $NetBSD: makefs.c,v 1.59 2024/10/27 18:35:52 christos Exp $     */

/*
* Copyright (c) 2001-2003 Wasabi Systems, Inc.
* All rights reserved.
*
* Written by Luke Mewburn for Wasabi Systems, Inc.
*
* 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 for the NetBSD Project by
*      Wasabi Systems, Inc.
* 4. The name of Wasabi Systems, Inc. may not be used to endorse
*    or promote products derived from this software without specific prior
*    written permission.
*
* THIS SOFTWARE IS PROVIDED BY WASABI SYSTEMS, INC. ``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 WASABI SYSTEMS, INC
* 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 HAVE_NBTOOL_CONFIG_H
#include "nbtool_config.h"
#endif

#include <sys/cdefs.h>
#if defined(__RCSID) && !defined(__lint)
__RCSID("$NetBSD: makefs.c,v 1.59 2024/10/27 18:35:52 christos Exp $");
#endif  /* !__lint */

#include <assert.h>
#include <ctype.h>
#include <errno.h>
#include <limits.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <stdbool.h>
#include <util.h>

#include "makefs.h"
#include "mtree.h"
#include "cd9660.h"

/*
* list of supported file systems and dispatch functions
*/
typedef struct {
       const char      *type;
       void            (*prepare_options)(fsinfo_t *);
       int             (*parse_options)(const char *, fsinfo_t *);
       void            (*cleanup_options)(fsinfo_t *);
       void            (*make_fs)(const char *, const char *, fsnode *,
                               fsinfo_t *);
} fstype_t;

static fstype_t fstypes[] = {
#define ENTRY(name) { \
       # name, name ## _prep_opts, name ## _parse_opts, \
       name ## _cleanup_opts, name ## _makefs  \
}
       ENTRY(ffs),
       ENTRY(cd9660),
       ENTRY(chfs),
       ENTRY(v7fs),
       ENTRY(msdos),
       ENTRY(udf),
       { .type = NULL  },
};

u_int           debug;
struct timespec start_time;
struct stat stampst;

static fstype_t *get_fstype(const char *);
static int get_tstamp(const char *, struct stat *);
static void usage(fstype_t *, fsinfo_t *) __dead;
static u_int parse_debug(char *);

int
main(int argc, char *argv[])
{
       struct timeval   start;
       fstype_t        *fstype;
       fsinfo_t         fsoptions;
       fsnode          *root;
       int              ch, i;
       size_t           len;
       char            *specfile;

       setprogname(argv[0]);

       debug = 0;
       if ((fstype = get_fstype(DEFAULT_FSTYPE)) == NULL)
               errx(EXIT_FAILURE,
                   "Unknown default fs type `%s'.", DEFAULT_FSTYPE);

               /* set default fsoptions */
       (void)memset(&fsoptions, 0, sizeof(fsoptions));
       fsoptions.fd = -1;
       fsoptions.sectorsize = -1;

       if (fstype->prepare_options)
               fstype->prepare_options(&fsoptions);

       specfile = NULL;
#ifdef CLOCK_REALTIME
       ch = clock_gettime(CLOCK_REALTIME, &start_time);
#else
       ch = gettimeofday(&start, NULL);
       start_time.tv_sec = start.tv_sec;
       start_time.tv_nsec = start.tv_usec * 1000;
#endif
       if (ch == -1)
               err(EXIT_FAILURE, "Unable to get system time");


       while ((ch = getopt(argc, argv, "B:b:d:f:F:LM:m:N:O:o:rs:S:t:T:xZ")) != -1) {
               switch (ch) {

               case 'B':
                       if (strcmp(optarg, "be") == 0 ||
                           strcmp(optarg, "4321") == 0 ||
                           strcmp(optarg, "big") == 0) {
#if BYTE_ORDER == LITTLE_ENDIAN
                               fsoptions.needswap = 1;
#endif
                       } else if (strcmp(optarg, "le") == 0 ||
                           strcmp(optarg, "1234") == 0 ||
                           strcmp(optarg, "little") == 0) {
#if BYTE_ORDER == BIG_ENDIAN
                               fsoptions.needswap = 1;
#endif
                       } else {
                               warnx("Invalid endian `%s'.", optarg);
                               usage(fstype, &fsoptions);
                       }
                       break;

               case 'b':
                       len = strlen(optarg) - 1;
                       if (optarg[len] == '%') {
                               optarg[len] = '\0';
                               fsoptions.freeblockpc = (int)
                                   strsuftoll("free block percentage",
                                       optarg, 0, 99);
                       } else {
                               fsoptions.freeblocks =
                                   strsuftoll("free blocks",
                                       optarg, 0, LLONG_MAX);
                       }
                       break;

               case 'd':
                       debug = parse_debug(optarg);
                       break;

               case 'f':
                       len = strlen(optarg) - 1;
                       if (optarg[len] == '%') {
                               optarg[len] = '\0';
                               fsoptions.freefilepc = (int)
                                   strsuftoll("free file percentage",
                                       optarg, 0, 99);
                       } else {
                               fsoptions.freefiles =
                                   strsuftoll("free files",
                                       optarg, 0, LLONG_MAX);
                       }
                       break;

               case 'F':
                       specfile = optarg;
                       break;

               case 'L':
                       fsoptions.follow = true;
                       break;

               case 'M':
                       fsoptions.minsize =
                           strsuftoll("minimum size", optarg, 1LL, LLONG_MAX);
                       break;

               case 'N':
                       if (! setup_getid(optarg))
                               errx(EXIT_FAILURE,
                           "Unable to use user and group databases in `%s'",
                                   optarg);
                       break;

               case 'm':
                       fsoptions.maxsize =
                           strsuftoll("maximum size", optarg, 1LL, LLONG_MAX);
                       break;

               case 'O':
                       fsoptions.offset =
                           strsuftoll("offset", optarg, 0LL, LLONG_MAX);
                       break;

               case 'o':
               {
                       char *p;

                       while ((p = strsep(&optarg, ",")) != NULL) {
                               if (*p == '\0')
                                       errx(EXIT_FAILURE, "Empty option");
                               if (! fstype->parse_options(p, &fsoptions))
                                       usage(fstype, &fsoptions);
                       }
                       break;
               }

               case 'r':
                       fsoptions.replace = 1;
                       break;

               case 's':
                       fsoptions.minsize = fsoptions.maxsize =
                           strsuftoll("size", optarg, 1LL, LLONG_MAX);
                       break;

               case 'S':
                       fsoptions.sectorsize =
                           (int)strsuftoll("sector size", optarg,
                               1LL, INT_MAX);
                       break;

               case 't':
                       /* Check current one and cleanup if necessary. */
                       if (fstype->cleanup_options)
                               fstype->cleanup_options(&fsoptions);
                       fsoptions.fs_specific = NULL;
                       if ((fstype = get_fstype(optarg)) == NULL)
                               errx(EXIT_FAILURE,
                                   "Unknown fs type `%s'.", optarg);
                       fstype->prepare_options(&fsoptions);
                       break;

               case 'T':
                       if (get_tstamp(optarg, &stampst) == -1)
                               errx(EXIT_FAILURE,
                                   "Cannot get timestamp from `%s'", optarg);
                       break;

               case 'x':
                       fsoptions.onlyspec++;
                       break;

               case 'Z':
                       fsoptions.sparse = 1;
                       break;

               case '?':
               default:
                       usage(fstype, &fsoptions);
                       /* NOTREACHED */

               }
       }
       if (debug) {
               printf("debug mask: 0x%08x\n", debug);
               printf("start time: %ld.%ld, %s",
                   (long)start_time.tv_sec, (long)start_time.tv_nsec,
                   ctime(&start_time.tv_sec));
       }
       argc -= optind;
       argv += optind;

       if (argc < 2)
               usage(fstype, &fsoptions);

       /* -x must be accompanied by -F */
       if (fsoptions.onlyspec != 0 && specfile == NULL)
               errx(EXIT_FAILURE, "-x requires -F mtree-specfile.");

                               /* walk the tree */
       TIMER_START(start);
       root = walk_dir(argv[1], ".", NULL, NULL, fsoptions.replace,
           fsoptions.follow);
       TIMER_RESULTS(start, "walk_dir");

       /* append extra directory */
       for (i = 2; i < argc; i++) {
               struct stat sb;
               if (stat(argv[i], &sb) == -1)
                       err(EXIT_FAILURE, "Can't stat `%s'", argv[i]);
               if (!S_ISDIR(sb.st_mode))
                       errx(EXIT_FAILURE, "%s: not a directory", argv[i]);
               TIMER_START(start);
               root = walk_dir(argv[i], ".", NULL, root, fsoptions.replace,
                   fsoptions.follow);
               TIMER_RESULTS(start, "walk_dir2");
       }

       if (specfile) {         /* apply a specfile */
               TIMER_START(start);
               apply_specfile(specfile, argv[1], root, fsoptions.onlyspec);
               TIMER_RESULTS(start, "apply_specfile");
       }

       if (debug & DEBUG_DUMP_FSNODES) {
               printf("\nparent: %s\n", argv[1]);
               dump_fsnodes(root);
               putchar('\n');
       }

                               /* build the file system */
       TIMER_START(start);
       fstype->make_fs(argv[0], argv[1], root, &fsoptions);
       TIMER_RESULTS(start, "make_fs");

       free_fsnodes(root);

       exit(EXIT_SUCCESS);
       /* NOTREACHED */
}

int
set_option(const option_t *options, const char *option, char *buf, size_t len)
{
       char *var, *val;
       int retval;

       assert(option != NULL);

       var = estrdup(option);
       for (val = var; *val; val++)
               if (*val == '=') {
                       *val++ = '\0';
                       break;
               }
       retval = set_option_var(options, var, val, buf, len);
       free(var);
       return retval;
}

void
print_options(FILE *fp, const option_t *options)
{
       for (size_t i = 0; options[i].name != NULL; i++) {
               fprintf(fp, "%s=", options[i].name);
               switch (options[i].type) {
               case OPT_BOOL:
                       fputs(*(bool *)options[i].value ? "true\n" : "false\n",
                           fp);
                       break;
               case OPT_STRARRAY:
               case OPT_STRPTR:
               case OPT_STRBUF:
                       fprintf(fp, "%s\n", *(const char **)options[i].value);
                       break;
               case OPT_INT64:
                       fprintf(fp, "%" PRIu64 "\n",
                           *(uint64_t *)options[i].value);
                       break;
               case OPT_INT32:
                       fprintf(fp, "%" PRIu32 "\n",
                           *(uint32_t *)options[i].value);
                       break;
               case OPT_INT16:
                       fprintf(fp, "%" PRIu16 "\n",
                           *(uint16_t *)options[i].value);
                       break;
               case OPT_INT8:
                       fprintf(fp, "%" PRIu8 "\n",
                           *(uint8_t *)options[i].value);
                       break;
               default:
                       warnx("Unknown type %d in option %s", options[i].type,
                           options[i].name);
                       return;
               }
       }
}

int
set_option_var(const option_t *options, const char *var, const char *val,
   char *buf, size_t len)
{
       char *s;
       size_t i;

#define NUM(type) \
       if (!*val) { \
               *(type *)options[i].value = 1; \
               break; \
       } \
       *(type *)options[i].value = (type)strsuftoll(options[i].desc, val, \
           options[i].minimum, options[i].maximum); break

       for (i = 0; options[i].name != NULL; i++) {
               if (var[1] == '\0') {
                       if (options[i].letter != var[0])
                               continue;
               } else if (strcmp(options[i].name, var) != 0)
                       continue;
               switch (options[i].type) {
               case OPT_BOOL:
                       *(bool *)options[i].value = 1;
                       break;
               case OPT_STRARRAY:
                       strlcpy((void *)options[i].value, val, (size_t)
                           options[i].maximum);
                       break;
               case OPT_STRPTR:
                       s = estrdup(val);
                       *(char **)options[i].value = s;
                       break;
               case OPT_STRBUF:
                       if (buf == NULL)
                               abort();
                       strlcpy(buf, val, len);
                       break;
               case OPT_INT64:
                       NUM(uint64_t);
               case OPT_INT32:
                       NUM(uint32_t);
               case OPT_INT16:
                       NUM(uint16_t);
               case OPT_INT8:
                       NUM(uint8_t);
               default:
                       warnx("Unknown type %d in option %s", options[i].type,
                           val);
                       return 0;
               }
               return (int)i;
       }
       warnx("Unknown option `%s'", var);
       return -1;
}


static fstype_t *
get_fstype(const char *type)
{
       int i;

       for (i = 0; fstypes[i].type != NULL; i++)
               if (strcmp(fstypes[i].type, type) == 0)
                       return (&fstypes[i]);
       return (NULL);
}

option_t *
copy_opts(const option_t *o)
{
       size_t i;
       for (i = 0; o[i].name; i++)
               continue;
       i++;
       return memcpy(ecalloc(i, sizeof(*o)), o, i * sizeof(*o));
}

static int
get_tstamp(const char *b, struct stat *st)
{
       time_t when;
       char *eb;
       long long l;

       if (stat(b, st) != -1)
               return 0;

#ifndef HAVE_NBTOOL_CONFIG_H
       errno = 0;
       if ((when = parsedate(b, NULL, NULL)) == -1 && errno != 0)
#endif
       {
               errno = 0;
               l = strtoll(b, &eb, 0);
               if (b == eb || *eb || errno)
                       return -1;
               when = (time_t)l;
       }

       st->st_ino = 1;
#if HAVE_STRUCT_STAT_BIRTHTIME
       st->st_birthtime =
#endif
       st->st_mtime = st->st_ctime = st->st_atime = when;
       return 0;
}

static struct {
       const char *n;
       u_int v;
} nv[] = {
       DEBUG_STRINGS
};

static void
usage(fstype_t *fstype, fsinfo_t *fsoptions)
{
       const char *prog;

       prog = getprogname();
       fprintf(stderr,
"Usage: %s [-rxZ] [-B endian] [-b free-blocks] [-d debug-mask|comma-separated-option]\n"
"\t[-F mtree-specfile] [-f free-files] [-M minimum-size] [-m maximum-size]\n"
"\t[-N userdb-dir] [-O offset] [-o fs-options] [-S sector-size]\n"
"\t[-s image-size] [-T <timestamp/file>] [-t fs-type]"
" image-file directory [extra-directory ...]\n",
           prog);

       fprintf(stderr, "\nDebugging options:\n");
       for (size_t i = 0; i < __arraycount(nv); i++)
               fprintf(stderr, "\t0x%8.8x\t%s\n", nv[i].v, nv[i].n);

       if (fstype) {
               size_t i;
               option_t *o = fsoptions->fs_options;

               fprintf(stderr, "\n%s specific options:\n", fstype->type);
               for (i = 0; o[i].name != NULL; i++)
                       fprintf(stderr, "\t%c%c%20.20s\t%s\n",
                           o[i].letter ? o[i].letter : ' ',
                           o[i].letter ? ',' : ' ',
                           o[i].name, o[i].desc);
       }
       exit(EXIT_FAILURE);
}


static u_int
parse_debug(char *str)
{
       char *ep;
       u_int d;
       size_t i;

       errno = 0;
       d = (u_int)strtoul(str, &ep, 0);
       if (str != ep && !*ep && errno == 0)
               return d;
       d = 0;
       for (char *a = strtok(str, ","); a != NULL; a = strtok(NULL, ",")) {
               for (i = 0; i < __arraycount(nv); i++)
                       if (strcmp(nv[i].n, a) == 0) {
                               d |= nv[i].v;
                               break;
                       }
               if (i == __arraycount(nv))
                       errx(EXIT_FAILURE, "Unknown debug option `%s'", a);
       }
       return d;
}

You are viewing proxied material from ftp.icm.edu.pl. The copyright of proxied material belongs to its original authors. Any comments or complaints in relation to proxied material should be directed to the original authors of the content concerned. Please see the disclaimer for more details.