/*      $NetBSD: main.c,v 1.3 2014/01/22 06:17:25 riastradh Exp $       */

/*-
 * Copyright (c) 2013 The NetBSD Foundation, Inc.
 * All rights reserved.
 *
 * This code is derived from software contributed to The NetBSD Foundation
 * by Taylor R. Campbell.
 *
 * 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.
 *
 * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. 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 FOUNDATION 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/cdefs.h>
__RCSID("$NetBSD: main.c,v 1.3 2014/01/22 06:17:25 riastradh Exp $");

#include <assert.h>
#include <err.h>
#include <getopt.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>

#include "common.h"

static int (*operation)(int, char **, const struct options *) = &vndcompress;

int
main(int argc, char **argv)
{
        static const struct options zero_options;
        struct options options = zero_options, *O = &options;
        int ch;

        setprogname(argv[0]);

        if (strcmp(getprogname(), "vndcompress") == 0)
                operation = &vndcompress;
        else if (strcmp(getprogname(), "vnduncompress") == 0)
                operation = &vnduncompress;
        else
                warnx("unknown program name, defaulting to vndcompress: %s",
                    getprogname());

        while ((ch = getopt(argc, argv, "b:cdk:l:p:rRw:")) != -1) {
                switch (ch) {
                case 'b':
                        if (ISSET(O->flags, FLAG_b)) {
                                warnx("-b may be supplied only once");
                                usage();
                        }
                        O->flags |= FLAG_b;
                        __CTASSERT(MIN_BLOCKSIZE <= MAX_BLOCKSIZE);
                        __CTASSERT(MAX_BLOCKSIZE <= LLONG_MAX);
                        O->blocksize = strsuftoll("block size", optarg,
                            MIN_BLOCKSIZE, MAX_BLOCKSIZE);
                        break;

                case 'c':
                        if (ISSET(O->flags, FLAG_d)) {
                                warnx("-c and -d are mutually exclusive");
                                usage();
                        }
                        O->flags |= FLAG_c;
                        operation = &vndcompress;
                        break;

                case 'd':
                        if (ISSET(O->flags, FLAG_c)) {
                                warnx("-c and -d are mutually exclusive");
                                usage();
                        }
                        O->flags |= FLAG_d;
                        operation = &vnduncompress;
                        break;

                case 'k':
                        if (ISSET(O->flags, FLAG_k)) {
                                warnx("-k may be supplied only once");
                                usage();
                        }
                        O->flags |= FLAG_k;
                        O->checkpoint_blocks = strsuftoll("checkpoint blocks",
                            optarg,
                            0, MIN(UINT32_MAX, (OFF_MAX / MIN_BLOCKSIZE)));
                        break;

                case 'l':
                        if (ISSET(O->flags, FLAG_l)) {
                                warnx("-l may be supplied only once");
                                usage();
                        }
                        O->flags |= FLAG_l;
                        O->length = strsuftoll("length", optarg,
                            0, MIN(OFF_MAX, UINT64_MAX));
                        break;

                case 'p':
                        O->flags |= FLAG_p;
                        __CTASSERT(OFF_MAX <= LLONG_MAX);
                        O->end_block = strsuftoll("end block", optarg,
                            0, MIN(UINT32_MAX, (OFF_MAX / MIN_BLOCKSIZE)));
                        break;

                case 'r':
                        O->flags |= FLAG_r;
                        break;

                case 'R':
                        O->flags |= FLAG_R;
                        break;

                case 'w':
                        if (ISSET(O->flags, FLAG_w)) {
                                warnx("-w may be supplied only once");
                                usage();
                        }
                        O->flags |= FLAG_w;
                        O->window_size = strsuftoll("window size", optarg,
                            0, MAX_WINDOW_SIZE);
                        break;

                case '?':
                default:
                        usage();
                }
        }

        argc -= optind;
        argv += optind;

        if (operation == &vnduncompress) {
                if (ISSET(O->flags, ~(FLAG_d | FLAG_w)))
                        usage();
        } else {
                assert(operation == &vndcompress);
                if (ISSET(O->flags, ~(FLAG_b | FLAG_c | FLAG_k | FLAG_l |
                            FLAG_p | FLAG_r | FLAG_R | FLAG_w)))
                        usage();
                if (ISSET(O->flags, FLAG_R) && !ISSET(O->flags, FLAG_r)) {
                        warnx("-R makes no sense without -r");
                        usage();
                }
        }

        return (*operation)(argc, argv, O);
}

void __dead
usage(void)
{

        (void)fprintf(stderr,
            "Usage: %s -c [-rR] [-b <blocksize>] [-k <checkpoint-blocks>]\n"
            "          [-l <length>] [-p <partial-offset>] [-w <winsize>]\n"
            "          <image> <compressed-image> [<blocksize>]\n"
            "       %s -d <compressed-image> <image>\n",
            getprogname(), getprogname());
        exit(1);
}