/*
 * Copyright (c) 2011-2015 The DragonFly Project.  All rights reserved.
 *
 * This code is derived from software contributed to The DragonFly Project
 * by Matthew Dillon <dillon@dragonflybsd.org>
 *
 * 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. Neither the name of The DragonFly Project nor the names of its
 *    contributors may be used to endorse or promote products derived
 *    from this software without specific, prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
 * ``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 THE
 * COPYRIGHT HOLDERS OR CONTRIBUTORS 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.
 */

#include <sys/types.h>

#include <stdio.h>
#include <stdlib.h>
#include <stddef.h>
#include <unistd.h>
#include <string.h>
#include <assert.h>
#include <err.h>

#include "mkfs_hammer2.h"

static void parse_fs_size(hammer2_mkfs_options_t *, const char *);
static void usage(void);

int
main(int ac, char **av)
{
        hammer2_mkfs_options_t opt;
        int ch;
        int label_specified = 0;

        /*
         * Initialize option structure.
         */
        hammer2_mkfs_init(&opt);

        /*
         * Parse arguments.
         */
        while ((ch = getopt(ac, av, "L:b:r:V:s:d")) != -1) {
                int i;

                switch(ch) {
                case 'b':
                        opt.BootAreaSize = getsize(optarg,
                                         HAMMER2_NEWFS_ALIGN,
                                         HAMMER2_BOOT_MAX_BYTES, 2);
                        break;
                case 'r':
                        opt.AuxAreaSize = getsize(optarg,
                                         HAMMER2_NEWFS_ALIGN,
                                         HAMMER2_AUX_MAX_BYTES, 2);
                        break;
                case 'V':
                        opt.Hammer2Version = strtol(optarg, NULL, 0);
                        if (opt.Hammer2Version < HAMMER2_VOL_VERSION_MIN ||
                            opt.Hammer2Version >= HAMMER2_VOL_VERSION_WIP) {
                                errx(1, "I don't understand how to format "
                                     "HAMMER2 version %d",
                                     opt.Hammer2Version);
                        }
                        break;
                case 'L':
                        label_specified = 1;
                        if (strcasecmp(optarg, "none") == 0) {
                                break;
                        }
                        for (i = 0; i < opt.NLabels; i++) {
                                if (strcasecmp(opt.Label[i], optarg) == 0) {
                                        errx(1, "duplicate label \"%s\"; "
                                            "note \"LOCAL\" is auto-created",
                                            optarg);
                                }
                        }
                        if (opt.NLabels >= MAXLABELS) {
                                errx(1, "Limit of %d local labels",
                                     MAXLABELS - 1);
                        }
                        if (strlen(optarg) == 0) {
                                errx(1, "Volume label '%s' cannot be 0-length",
                                        optarg);
                        }
                        if (strlen(optarg) >= HAMMER2_INODE_MAXNAME) {
                                errx(1, "Volume label '%s' is too long "
                                        "(%d chars max)",
                                        optarg,
                                        HAMMER2_INODE_MAXNAME - 1);
                        }
                        opt.Label[opt.NLabels++] = strdup(optarg);
                        break;
                case 's':
                        parse_fs_size(&opt, optarg);
                        break;
                case 'd':
                        opt.DebugOpt = 1;
                        break;
                default:
                        usage();
                        break;
                }
        }

        ac -= optind;
        av += optind;

        if (ac == 0)
                errx(1, "You must specify at least one disk device");
        if (ac > HAMMER2_MAX_VOLUMES)
                errx(1, "The maximum number of volumes is %d",
                     HAMMER2_MAX_VOLUMES);

        /*
         * Check default label type.
         */
        if (!label_specified) {
                char c = av[0][strlen(av[0]) - 1];
                if (c == 'a')
                        opt.DefaultLabelType = HAMMER2_LABEL_BOOT;
                else if (c == 'd')
                        opt.DefaultLabelType = HAMMER2_LABEL_ROOT;
                else
                        opt.DefaultLabelType = HAMMER2_LABEL_DATA;
        }

        /*
         * Create Hammer2 filesystem.
         */
        assert(opt.CompType == HAMMER2_COMP_DEFAULT);
        assert(opt.CheckType == HAMMER2_CHECK_DEFAULT);
        hammer2_mkfs(ac, av, &opt);

        /*
         * Cleanup option structure.
         */
        hammer2_mkfs_cleanup(&opt);

        return(0);
}

static
void
parse_fs_size(hammer2_mkfs_options_t *opt, const char *arg)
{
        char *o, *p, *s;

        opt->NFileSystemSizes = 0;
        o = p = strdup(arg);

        while ((s = p) != NULL) {
                if ((p = strchr(p, ':')) != NULL)
                        *p++ = 0;
                /* XXX 0x7fffffffffffffff isn't limitation of HAMMER2 */
                opt->FileSystemSize[opt->NFileSystemSizes++] = getsize(s,
                                 HAMMER2_FREEMAP_LEVEL1_SIZE,
                                 0x7fffffffffffffff, 2);
                if (opt->NFileSystemSizes >= HAMMER2_MAX_VOLUMES)
                        break;
        }
        free(o);
}

static
void
usage(void)
{
        fprintf(stderr,
                "usage: newfs_hammer2 [-b bootsize] [-r auxsize] "
                "[-V version] [-L label ...] [-s size] special ...\n"
        );
        exit(1);
}