/*      $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;
}