NetBSD/sys/arch/i386/stand/netboot/main.c

/*	$NetBSD: main.c,v 1.1.1.1 1997/03/14 02:40:31 perry Exp $	*/

/*
 * Copyright (c) 1996
 * 	Matthias Drochner.  All rights reserved.
 * Copyright (c) 1996
 * 	Perry E. Metzger.  All rights reserved.
 *
 * 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 acknowledgements:
 *	This product includes software developed for the NetBSD Project
 *	by Matthias Drochner.
 *	This product includes software developed for the NetBSD Project
 *	by Perry E. Metzger.
 * 4. The names of the authors may not be used to endorse or promote products
 *    derived from this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``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 AUTHOR 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 <lib/libkern/libkern.h>

#include <lib/libsa/stand.h>

#include <libi386.h>

extern char *strerror __P((int)); /* XXX missing in stand.h */

int errno;
static char *consdev;

extern char version[];
extern char etherdev[];

#ifdef SUPPORT_NFS /* XXX */
int debug = 0;
#endif

#define TIMEOUT 5
#define POLL_FREQ 10

#define MAXBOOTFILE 20

#ifdef COMPAT_OLDBOOT
int
parsebootfile(fname, fsname, devname, unit, partition, file)
const char *fname;
char **fsname; /* out */
char **devname; /* out */
unsigned int *unit, *partition; /* out */
const char **file; /* out */
{
    return(EINVAL);
}

int biosdisk_gettype(f)
struct open_file *f;
{
    return(0);
}
#endif

int bootit(filename, howto)
const char *filename;
int howto;
{
  if(exec_netbsd(filename, 0, howto, etherdev, "pc") < 0)
    printf("boot: %s\n", strerror(errno));
  else
    printf("boot returned\n");
  return(-1);
}

static void helpme()
{
  printf("commands are:\n"
	 "boot [filename] [-adrs]\n"
	 "     (ex. \"netbsd.old -s\"\n"
	 "help|?\n"
	 "quit\n");
}

/*
 * chops the head from the arguments and returns the arguments if any,
 * or possibly an empty string.
 */
static char *
gettrailer(arg)
char *arg;
{
  char *options;

  if ((options = strchr(arg, ' ')) == NULL)
    options = "";
  else
    *options++ = '\0';
  /* trim leading blanks */
  while (*options && *options == ' ')
      options++;

  return(options);
}

static int
parseopts(opts, howto)
char *opts;
int *howto;
{
  int tmpopt = 0;

  opts++; /* skip - */
  while (*opts && *opts != ' ') {
    tmpopt |= netbsd_opt(*opts);
    if(tmpopt == -1) {
      printf("-%c: unknown flag\n", *opts);
      helpme();
      return(0);
    }
    opts++;
  }
  *howto = tmpopt;
  return(1);
}

static int
parseboot(arg, filename, howto)
char *arg;
char **filename;
int *howto;
{
  char *opts = NULL;

  *filename = 0;
  *howto = 0;

  /* if there were no arguments */
  if (!*arg)
    return(1);

  /* format is... */
  /*[[xxNx:]filename] [-adrs]*/

  /* check for just args */
  if (arg[0] == '-'){
    opts = arg;
  } else { /* at least a file name */
    *filename = arg;

    opts = gettrailer(arg);
    if (!*opts)
      opts = NULL;
    else if (*opts != '-') {
      printf("invalid arguments\n");
      helpme();
      return(0);
    }
  }
  /* at this point, we have dealt with filenames. */

  /* now, deal with options */
  if (opts) {
    if (!parseopts(opts, howto))
      return(0);
  }

  return(1);
}

static void
docommand(arg)
char *arg;
{
  char *options;

  options = gettrailer(arg);

  if ((strcmp("help", arg) == 0) ||
      (strcmp("?", arg) == 0)) {
    helpme();
    return;
  }
  if (strcmp("quit", arg) == 0){
    printf("Exiting... goodbye...\n");
    exit(0);
  }
  if (strcmp("boot", arg) == 0){
      char *filename;
      int howto;
      if(parseboot(options, &filename, &howto))
	  bootit(filename, howto);
      return;
  }
  printf("unknown command\n");
  helpme();
}

void bootmenu()
{
  printf("\ntype \"?\" or \"help\" for help.\n");
  for(;;) {
    char input[80];

    input[0] = '\0';
    printf("> ");
    gets(input);

    docommand(input);
  }
}

static int
awaitkey(void)
{
  int i;

  i = TIMEOUT * POLL_FREQ;

  while (i--) {
    if(iskey()) {
      /* flush input buffer */
      while(iskey())
	getchar(); 

      return(1);
    }
    delay(1000000 / POLL_FREQ);
    if (!(i % POLL_FREQ))
      printf("%d\b", i/POLL_FREQ);
  }
  printf("0\n");
  return(0);
}

static void
print_banner(void)
{
  printf("\n"
	 ">> NetBSD BOOT: %d/%d k [%s]\n",
	 getbasemem(),
	 getextmem(),
	 version);
}

int
main()
{
    consdev = initio(CONSDEV_AUTO);
    gateA20();

    print_banner();

    printf("press any key for boot menu\n"
	   "starting in %d\b", TIMEOUT);

    if(awaitkey())
	bootmenu(); /* does not return */

    bootit("netbsd", 0);

    /* if that fails, let BIOS look for boot device */
    return(1);
}