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