/*      $NetBSD: main.c,v 1.23 2013/07/02 11:59:46 joerg Exp $  */

/*-
 * Copyright (c) 1996 The NetBSD Foundation, Inc.
 * All rights reserved.
 *
 * This code is derived from software contributed to The NetBSD Foundation
 * by Jason R. Thorpe.
 *
 * 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>
#ifndef lint
__COPYRIGHT("@(#) Copyright (c) 1996\
 The NetBSD Foundation, Inc.  All rights reserved.");
__RCSID("$NetBSD: main.c,v 1.23 2013/07/02 11:59:46 joerg Exp $");
#endif

#include <sys/param.h>
#include <err.h>
#include <string.h>
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>

#if defined(USE_EEPROM) || defined(USE_OPENPROM)
#include <machine/eeprom.h>
#endif

#include "defs.h"
#include "pathnames.h"

#ifdef USE_OPENPROM
# ifndef USE_EEPROM
#  define ee_action(a,b)
#  define ee_dump()
#  define ee_updatechecksums() (void)0
#  define check_for_openprom() 1
# endif
#endif

static  void action(char *);
static  void dump_prom(void);
static  void usage(void) __dead;

const char *path_eeprom = _PATH_EEPROM;
const char *path_openprom = _PATH_OPENPROM;
const char *path_openfirm = _PATH_OPENFIRM;
const char *path_prepnvram = _PATH_PREPNVRAM;
int     fix_checksum = 0;
int     ignore_checksum = 0;
int     update_checksums = 0;
int     cksumfail = 0;
u_short writecount;
int     eval = 0;
#ifdef USE_OPENPROM
int     verbose = 0;
int     use_openprom;
#endif
#if defined(USE_OPENFIRM) || defined (USE_PREPNVRAM)
int     verbose=0;
#endif

int
main(int argc, char *argv[])
{
        int ch, do_stdin = 0;
        char *cp, line[BUFSIZE];
#if defined(USE_OPENPROM) || defined(USE_OPENFIRM) || defined(USE_PREPNVRAM)
        const char *optstring = "-cf:iv";
#else
        const char *optstring = "-cf:i";
#endif /* USE_OPENPROM */

        while ((ch = getopt(argc, argv, optstring)) != -1)
                switch (ch) {
                case '-':
                        do_stdin = 1;
                        break;

                case 'c':
                        fix_checksum = 1;
                        break;

                case 'f':
                        path_eeprom = path_openprom = optarg;
                        break;

                case 'i':
                        ignore_checksum = 1;
                        break;

#if defined(USE_OPENPROM) || defined(USE_OPENFIRM) || defined(USE_PREPNVRAM)
                case 'v':
                        verbose = 1;
                        break;
#endif /* USE_OPENPROM */

                case '?':
                default:
                        usage();
                }
        argc -= optind;
        argv += optind;

#ifdef USE_OPENPROM
        use_openprom = check_for_openprom();

        if (use_openprom == 0) {
#endif /* USE_OPENPROM */
#if !defined(USE_OPENFIRM) && !defined(USE_PREPNVRAM)
                ee_verifychecksums();
                if (fix_checksum || cksumfail)
                        exit(cksumfail);
#endif
#ifdef USE_OPENPROM
        }
#endif /* USE_OPENPROM */

        if (do_stdin) {
                while (fgets(line, BUFSIZE, stdin) != NULL) {
                        if (line[0] == '\n')
                                continue;
                        if ((cp = strrchr(line, '\n')) != NULL)
                                *cp = '\0';
                        action(line);
                }
                if (ferror(stdin))
                        err(++eval, "stdin");
        } else {
                if (argc == 0) {
                        dump_prom();
                        exit(eval + cksumfail);
                }

                while (argc) {
                        action(*argv);
                        ++argv;
                        --argc;
                }
        }

#ifdef USE_OPENPROM
        if (use_openprom == 0)
#endif /* USE_OPENPROM */
#if !defined(USE_OPENFIRM) && !defined(USE_PREPNVRAM)
                if (update_checksums) {
                        ++writecount;
                        ee_updatechecksums();
                }

        exit(eval + cksumfail);
#endif
        return 0;
}

/*
 * Separate the keyword from the argument (if any), find the keyword in
 * the table, and call the corresponding handler function.
 */
static void
action(char *line)
{
        char *keyword, *arg;

        keyword = strdup(line);
        if ((arg = strrchr(keyword, '=')) != NULL)
                *arg++ = '\0';

#ifdef USE_PREPNVRAM
        prep_action(keyword, arg);
#else
#ifdef USE_OPENFIRM
        of_action(keyword, arg);
#else
#ifdef USE_OPENPROM
        if (use_openprom)
                op_action(keyword, arg);
        else {
#endif /* USE_OPENPROM */
                ee_action(keyword, arg);
#ifdef USE_OPENPROM
        }
#endif /* USE_OPENPROM */
#endif /* USE_OPENFIRM */
#endif /* USE_PREPNVRAM */
}

/*
 * Dump the contents of the prom corresponding to all known keywords.
 */
static void
dump_prom(void)
{

#ifdef USE_PREPNVRAM
        prep_dump();
#else
#ifdef USE_OPENFIRM
        of_dump();
#else
#ifdef USE_OPENPROM
        if (use_openprom)
                /*
                 * We have a special dump routine for this.
                 */
                op_dump();
        else {
#endif /* USE_OPENPROM */
                ee_dump();
#ifdef USE_OPENPROM
        }
#endif /* USE_OPENPROM */
#endif /* USE_OPENFIRM */
#endif /* USE_PREPNVRAM */
}

__dead static void
usage(void)
{

#if defined(USE_OPENPROM) || defined(USE_OPENFIRM) || defined(USE_PREPNVRAM)
        fprintf(stderr, "usage: %s %s\n", getprogname(),
            "[-] [-c] [-f device] [-i] [-v] [field[=value] ...]");
#else
        fprintf(stderr, "usage: %s %s\n", getprogname(),
            "[-] [-c] [-f device] [-i] [field[=value] ...]");
#endif /* __us */
        exit(1);
}