stand/dosboot/main.c

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

/*
 * Copyright (c) 1996, 1997
 * 	Matthias Drochner.  All rights reserved.
 * Copyright (c) 1996, 1997
 * 	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 <sys/reboot.h>

#include <lib/libkern/libkern.h>
#include <lib/libsa/stand.h>

#include <libi386.h>

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

extern void ls __P((char*));
extern int getopt __P((int, char**, const char*));

int errno;
static char *consdev;

extern char version[];

#define MAXDEVNAME 16

static char *current_fsmode;
static char *default_devname;
static int default_unit, default_partition;
static char *default_filename;

int
parsebootfile(fname, fsmode, devname, unit, partition, file)
const char *fname;
char **fsmode;
char **devname; /* out */
unsigned int *unit, *partition; /* out */
const char **file; /* out */
{
    const char *col, *help;

    *fsmode = current_fsmode;
    *devname = default_devname;
    *unit = default_unit;
    *partition = default_partition;
    *file = default_filename;

    if(!fname) return(0);

    if((col = strchr(fname, ':'))) { /* device given */
	static char savedevname[MAXDEVNAME+1];
	int devlen;
	unsigned int u = 0, p = 0;
	int i = 0;

	devlen = col - fname;
	if(devlen > MAXDEVNAME)
	    return(EINVAL);

#define isvalidname(c) ((c) >= 'a' && (c) <= 'z')
	if(!isvalidname(fname[i]))
	    return(EINVAL);
	do {
	    savedevname[i] = fname[i];
	    i++;
	} while(isvalidname(fname[i]));
	savedevname[i] = '\0';

#define isnum(c) ((c) >= '0' && (c) <= '9')
	if(i < devlen) {
	    if(!isnum(fname[i]))
		return(EUNIT);
	    do {
		u *= 10;
		u += fname[i++] - '0';
	    } while(isnum(fname[i]));
	}

#define isvalidpart(c) ((c) >= 'a' && (c) <= 'z')
	if(i < devlen) {
	    if(!isvalidpart(fname[i]))
		return(EPART);
	    p = fname[i++] - 'a';
	}

	if(i != devlen)
	    return(ENXIO);

	*devname = savedevname;
	*unit = u;
	*partition = p;
	help = col + 1;
    } else
	help = fname;

    if(*help) *file = help;

    return(0);
}

static void
print_bootsel(filename)
char *filename;
{
    char *fsname;
    char *devname;
    int unit, partition;
    const char *file;

    if(!parsebootfile(filename, &fsname, &devname, &unit, &partition, &file)) {
	if(!strcmp(fsname, "dos"))
		printf("booting %s\n", file);
	else if(!strcmp(fsname, "ufs"))
		printf("booting %s%d%c:%s\n", devname, unit,
		       'a' + partition, file);
    }
}

static void
bootit(filename, howto, tell)
const char *filename;
int howto, tell;
{
    if(tell)
	print_bootsel(filename);

    if(exec_netbsd(filename, 0, howto, 0, consdev) < 0)
	printf("boot: %s\n", strerror(errno));
    else
	printf("boot returned\n");
}

static void helpme()
{
  printf("commands are:\n"
	 "boot [xdNx:][filename] [-adrs]\n"
	 "     (ex. \"sd0a:netbsd.old -s\"\n"
	 "ls [path]\n"
	 "mode ufs|dos\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
parsemode(arg, mode)
char *arg;
char **mode;
{
    if(!strcmp("dos", arg))
	*mode = "dos";
    else if(!strcmp("ufs", arg))
	*mode = "ufs";
    else
	printf("invalid mode\n");
}

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

  options = gettrailer(arg);

  if ((strcmp("help", arg) == 0) ||
      (strcmp("?", arg) == 0)) {
    helpme();
    return;
  }
  if (strcmp("ls", arg) == 0){
      char *help = default_filename;
      if(strcmp(current_fsmode, "ufs")) {
	  printf("UFS only\n");
	  return;
      }
      default_filename = "/";
      ls(options);
      default_filename = help;
      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, 1);
      return;
  }
  if (strcmp("mode", arg) == 0){
      parsemode(options, &current_fsmode);
      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 void
print_banner(void)
{
  printf("\n"
	 ">> NetBSD BOOT: %d/%d k [%s]\n",
	 getbasemem(),
	 getextmem(),
	 version);
}

void usage()
{
  printf("dosboot [-u] [-c <commands>] [-i] [filename [-bootopts]]\n");
}

int main(argc, argv)
int argc;
char **argv;
{
  int ch;
  int interactive = 0;
  int howto;
  extern char *optarg;
  extern int optind;

  consdev = initio(CONSDEV_PC);
  gateA20();

  print_banner();

  current_fsmode = "dos";
  default_devname = "hd";
  default_unit = 0;
  default_partition = 0;
  default_filename = "netbsd";

  while((ch = getopt(argc, argv, "c:iu")) != -1) {
    switch(ch) {
      case 'c':
	docommand(optarg);
	return(1);
	break;
      case 'i':
	interactive = 1;
	break;
    case 'u':
	current_fsmode = "ufs";
	break;
      default:
	usage();
	return(1);
    }
  }

  if(interactive)
      bootmenu();

  argc -= optind;
  argv += optind;

  if(argc > 2) {
      usage();
      return(1);
  }

  howto = 0;
  if(argc > 1 && !parseopts(argv[1], &howto))
      return(1);

  bootit((argc > 0 ? argv[0] : "netbsd"), howto, 1);
  return(1);
}