stage2/stage2.c - grub4dos-solaris - Rivoreo Source Code Repositories

 /*
  *  GRUB  --  GRand Unified Bootloader
  *  Copyright (C) 1996  Erich Boleyn  <erich@uruk.org>
  *  Copyright (C) 2000  Free Software Foundation, Inc.
  *
  *  This program is free software; you can redistribute it and/or modify
  *  it under the terms of the GNU General Public License as published by
  *  the Free Software Foundation; either version 2 of the License, or
  *  (at your option) any later version.
  *
  *  This program is distributed in the hope that it will be useful,
  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  *  GNU General Public License for more details.
  *
  *  You should have received a copy of the GNU General Public License
  *  along with this program; if not, write to the Free Software
  *  Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  */

 #include "shared.h"

 grub_jmp_buf restart_env;

 static char *
 get_entry (char *list, int num, int nested)
 {
   int i;

   for (i = 0; i < num; i++)
     {
       do
 	{
 	  while (*(list++));
 	}
       while (nested && *(list++));
     }

   return list;
 }


 static void
 print_entries (int y, int size, int first, char *menu_entries)
 {
   int i;

   gotoxy (77, y + 1);

   if (first)
     putchar (DISP_UP);
   else
     putchar (' ');

   menu_entries = get_entry (menu_entries, first, 0);

   for (i = 1; i <= size; i++)
     {
       int j = 0;

       gotoxy (3, y + i);

       while (*menu_entries)
 	{
 	  if (j < 71)
 	    {
 	      putchar (*menu_entries);
 	      j++;
 	    }

 	  menu_entries++;
 	}

       if (*(menu_entries - 1))
 	menu_entries++;

       for (; j < 71; j++)
 	putchar (' ');
     }

   gotoxy (77, y + size);

   if (*menu_entries)
     putchar (DISP_DOWN);
   else
     putchar (' ');
 }


 static void
 print_border (int y, int size)
 {
   int i;

 #ifndef GRUB_UTIL
   /* Color the menu. The menu is 75 * 14 characters.  */
   for (i = 0; i < 14; i++)
     {
       int j;
       for (j = 0; j < 75; j++)
 	{
 	  gotoxy (j + 1, i + y);
 	  set_attrib (normal_color);
 	}
     }
 #endif

   gotoxy (1, y);

   putchar (DISP_UL);
   for (i = 0; i < 73; i++)
     putchar (DISP_HORIZ);
   putchar (DISP_UR);

   i = 1;

   while (1)
     {
       gotoxy (1, y + i);

       if (i > size)
 	break;

       putchar (DISP_VERT);
       gotoxy (75, y + i);
       putchar (DISP_VERT);

       i++;
     }

   putchar (DISP_LL);
   for (i = 0; i < 73; i++)
     putchar (DISP_HORIZ);
   putchar (DISP_LR);
 }

 static void
 set_line (int y, int attr)
 {
   int x;

   for (x = 2; x < 75; x++)
     {
       gotoxy (x, y);
       set_attrib (attr);
     }
 }

 /* Set the attribute of the line Y to normal state.  */
 static void
 set_line_normal (int y)
 {
 #ifdef GRUB_UTIL
   set_line (y, A_NORMAL);
 #else
   set_line (y, normal_color);
 #endif
 }

 /* Set the attribute of the line Y to highlight state.  */
 static void
 set_line_highlight (int y)
 {
 #ifdef GRUB_UTIL
   set_line (y, A_REVERSE);
 #else
   set_line (y, highlight_color);
 #endif
 }

 static void
 run_menu (char *menu_entries, char *config_entries, int num_entries,
 	  char *heap, int entryno)
 {
   int c, time1, time2 = -1, first_entry = 0;
   char *cur_entry;

   /*
    *  Main loop for menu UI.
    */

 restart:
   while (entryno > 11)
     {
       first_entry++;
       entryno--;
     }

   init_page ();
 #ifndef GRUB_UTIL
   nocursor ();
 #endif

   print_border (3, 12);

 #ifdef GRUB_UTIL
   grub_printf ("\n
       Use the up and down arrows for selecting which entry is highlighted.\n");
 #else
   grub_printf ("\n
       Use the %c and %c keys for selecting which entry is highlighted.\n",
 	  DISP_UP, DISP_DOWN);
 #endif

   if (password)
     {
       printf ("       Press enter to boot the selected OS or \'p\' to enter a
         password to unlock the next set of features.");
     }
   else
     {
       if (config_entries)
 	printf ("       Press enter to boot the selected OS, \'e\' to edit the
         commands before booting, or \'c\' for a command-line.");
       else
 	printf (
 "      Press \'b\' to boot, \'e\' to edit the selected command in the
       boot sequence, \'c\' for a command-line, \'o\' to open a new line
       after (\'O\' for before) the selected line, \'d\' to remove the
       selected line, or escape to go back to the main menu.");
     }

   print_entries (3, 12, first_entry, menu_entries);

   /* highlight initial line */
   set_line_highlight (4 + entryno);

   /* XX using RT clock now, need to initialize value */
   while ((time1 = getrtsecs()) == 0xFF);

   while (1)
     {
       /* initilize to NULL just in case... */
       cur_entry = NULL;

       if (grub_timeout >= 0 && (time1 = getrtsecs()) != time2 && time1 != 0xFF)
 	{
 	  if (grub_timeout <= 0)
 	    {
 	      grub_timeout = -1;
 	      break;
 	    }

 	  /* else not booting yet! */
 	  time2  = time1;
 	  gotoxy (3, 22);
 	  printf ("The highlighted entry will be booted automatically in %d seconds.    ", grub_timeout);
 	  gotoxy (74, 4 + entryno);
 	  grub_timeout--;
 	}

       if (checkkey () != -1)
 	{
 	  c = getkey ();

 	  if (grub_timeout >= 0)
 	    {
 	      gotoxy (3, 22);
 	      printf ("                                                                    ");
 	      grub_timeout = -1;
 	      fallback_entry = -1;
 	      gotoxy (74, 4 + entryno);
 	    }

 	  if ((c == KEY_UP) || (ASCII_CHAR (c) == 16))
 	    {
 	      if (entryno > 0)
 		{
 		  set_line_normal (4 + entryno);
 		  entryno--;
 		  set_line_highlight (4 + entryno);
 		}
 	      else if (first_entry > 0)
 		{
 		  first_entry--;
 		  print_entries (3, 12, first_entry, menu_entries);
 		  set_line_highlight (4);
 		}
 	    }
 	  if (((c == KEY_DOWN) || (ASCII_CHAR (c) == 14))
 	      && (first_entry + entryno + 1) < num_entries)
 	    {
 	      if (entryno < 11)
 		{
 		  set_line_normal (4 + entryno);
 		  entryno++;
 		  set_line_highlight (4 + entryno);
 		}
 	      else if (num_entries > 12 + first_entry)
 		{
 		  first_entry++;
 		  print_entries (3, 12, first_entry, menu_entries);
 		  set_line_highlight (15);
 		}
 	    }

 	  c = ASCII_CHAR (c);

 	  if (config_entries)
 	    {
 	      if ((c == '\n') || (c == '\r'))
 		break;
 	    }
 	  else
 	    {
 	      if ((c == 'd') || (c == 'o') || (c == 'O'))
 		{
 		  set_line_normal (4 + entryno);

 		  /* insert after is almost exactly like insert before */
 		  if (c == 'o')
 		    {
 		      /* But `o' differs from `O', since it may causes
 			 the menu screen to scroll up.  */
 		      if (entryno < 11)
 			entryno++;
 		      else
 			first_entry++;

 		      c = 'O';
 		    }

 		  cur_entry = get_entry (menu_entries,
 					 first_entry + entryno,
 					 0);

 		  if (c == 'O')
 		    {
 		      memmove (cur_entry + 2, cur_entry,
 			       ((int) heap) - ((int) cur_entry));

 		      cur_entry[0] = ' ';
 		      cur_entry[1] = 0;

 		      heap += 2;

 		      num_entries++;
 		    }
 		  else if (num_entries > 0)
 		    {
 		      char *ptr = get_entry(menu_entries,
 					    first_entry + entryno + 1,
 					    0);

 		      memmove (cur_entry, ptr, ((int) heap) - ((int) ptr));
 		      heap -= (((int) ptr) - ((int) cur_entry));

 		      num_entries--;

 		      if (entryno >= num_entries)
 			entryno--;
 		      if (first_entry && num_entries < 12 + first_entry)
 			first_entry--;
 		    }

 		  print_entries (3, 12, first_entry, menu_entries);
 		  set_line_highlight (4 + entryno);
 		}

 	      cur_entry = menu_entries;
 	      if (c == 27)
 		return;
 	      if (c == 'b')
 		break;
 	    }

 	  if (password)
 	    {
 	      if (c == 'p')
 		{
 		  /* Do password check here! */
 		  char entered[32];
 		  char *pptr = password;

 		  gotoxy (1, 21);

 		  /* Wipe out the previously entered password */
 		  memset (entered, 0, sizeof (entered));
 		  get_cmdline (" Password: ", entered, 31, '*', 0);

 		  while (! isspace (*pptr) && *pptr)
 		    pptr++;

 		  /* Make sure that PASSWORD is NUL-terminated.  */
 		  *pptr++ = 0;

 		  if (! strcmp (password, entered))
 		    {
 		      char *new_file = config_file;
 		      while (isspace (*pptr))
 			pptr++;
 		      while ((*(new_file++) = *(pptr++)) != 0)
 			;
 		      return;
 		    }
 		  else
 		    {
 		      printf ("Failed!\n      Press any key to continue...");
 		      getkey ();
 		      goto restart;
 		    }
 		}
 	    }
 	  else
 	    {
 	      if (c == 'e')
 		{
 		  int num_entries = 0, i = 0;
 		  char *new_heap;

 		  if (config_entries)
 		    {
 		      new_heap = heap;
 		      cur_entry = get_entry (config_entries,
 					     first_entry + entryno,
 					     1);
 		    }
 		  else
 		    {
 		      /* safe area! */
 		      new_heap = heap + NEW_HEAPSIZE + 1;
 		      cur_entry = get_entry (menu_entries,
 					     first_entry + entryno,
 					     0);
 		    }

 		  do
 		    {
 		      while ((*(new_heap++) = cur_entry[i++]) != 0);
 		      num_entries++;
 		    }
 		  while (config_entries && cur_entry[i]);

 		  /* this only needs to be done if config_entries is non-NULL,
 		     but it doesn't hurt to do it always */
 		  *(new_heap++) = 0;

 		  if (config_entries)
 		    run_menu (heap, NULL, num_entries, new_heap, 0);
 		  else
 		    {
 		      cls ();
 		      print_cmdline_message (0);

 		      new_heap = heap + NEW_HEAPSIZE + 1;

 		      saved_drive = boot_drive;
 		      saved_partition = install_partition;
 		      current_drive = 0xFF;

 		      if (! get_cmdline (PACKAGE " edit> ", new_heap,
 					 NEW_HEAPSIZE + 1, 0, 1))
 			{
 			  int j = 0;

 			  /* get length of new command */
 			  while (new_heap[j++])
 			    ;

 			  if (j < 2)
 			    {
 			      j = 2;
 			      new_heap[0] = ' ';
 			      new_heap[1] = 0;
 			    }

 			  /* align rest of commands properly */
 			  memmove (cur_entry + j, cur_entry + i,
 				   ((int) heap) - (((int) cur_entry) + i));

 			  /* copy command to correct area */
 			  memmove (cur_entry, new_heap, j);

 			  heap += (j - i);
 			}
 		    }

 		  goto restart;
 		}
 	      if (c == 'c')
 		{
 		  enter_cmdline (heap, 0);
 		  goto restart;
 		}
 #ifdef GRUB_UTIL
 	      if (c == 'q')
 		{
 		  /* The same as ``quit''.  */
 		  stop ();
 		}
 #endif
 	    }
 	}
     }

   /*
    *  Attempt to boot an entry.
    */

   while (1)
     {
       cls ();

       if (config_entries)
 	printf ("  Booting \'%s\'\n\n",
 		get_entry (menu_entries, first_entry + entryno, 0));
       else
 	printf ("  Booting command-list\n\n");

       if (! cur_entry)
 	cur_entry = get_entry (config_entries, first_entry + entryno, 1);

       if (run_script (cur_entry, heap))
 	{
 	  if (fallback_entry < 0)
 	    break;
 	  else
 	    {
 	      cur_entry = NULL;
 	      first_entry = 0;
 	      entryno = fallback_entry;
 	      fallback_entry = -1;
 	    }
 	}
       else
 	break;
     }

   goto restart;
 }


 static int
 get_line_from_config (char *cmdline, int maxlen)
 {
   int pos = 0, literal = 0, comment = 0;
   char c;  /* since we're loading it a byte at a time! */

   while (grub_read (&c, 1))
     {
       /* translate characters first! */
       if (c == '\\' && ! literal)
 	{
 	  literal = 1;
 	  continue;
 	}
       if (c == '\r')
 	continue;
       if ((c == '\t') || (literal && (c == '\n')))
 	c = ' ';

       literal = 0;

       if (comment)
 	{
 	  if (c == '\n')
 	    comment = 0;
 	}
       else if (! pos)
 	{
 	  if (c == '#')
 	    comment = 1;
 	  else if ((c != ' ') && (c != '\n'))
 	    cmdline[pos++] = c;
 	}
       else
 	{
 	  if (c == '\n')
 	    break;

 	  if (pos < maxlen)
 	    cmdline[pos++] = c;
 	}
     }

   cmdline[pos] = 0;

   return pos;
 }


 /* This is the starting function in C.  */
 void
 cmain (void)
 {
   int config_len, menu_len, num_entries;
   char *config_entries, *menu_entries;
   char *kill = (char *) KILL_BUF;

   /* Initialize the environment for restarting Stage 2.  */
   grub_setjmp (restart_env);

   /* Initialize the kill buffer.  */
   *kill = 0;

   /* Never return.  */
   for (;;)
     {
       config_len = 0;
       menu_len = 0;
       num_entries = 0;
       config_entries = (char *) (mbi.mmap_addr + mbi.mmap_length);
       menu_entries = (char *) MENU_BUF;
       init_config ();

       /* Here load the configuration file.  */

 #ifdef GRUB_UTIL
       if (use_config_file && grub_open (config_file))
 #else
       if (grub_open (config_file))
 #endif
 	{
 	  /* STATE 0:  Before any title command.
 	     STATE 1:  In a title command.
 	     STATE >1: In a entry after a title command.  */
 	  int state = 0, prev_config_len = 0, prev_menu_len = 0;
 	  char *cmdline;

 	  cmdline = (char *) CMDLINE_BUF;
 	  while (get_line_from_config (cmdline, NEW_HEAPSIZE))
 	    {
 	      struct builtin *builtin;

 	      /* Get the pointer to the builtin structure.  */
 	      builtin = find_command (cmdline);
 	      errnum = 0;
 	      if (! builtin)
 		/* Unknown command. Just skip now.  */
 		continue;

 	      if (builtin->flags & BUILTIN_TITLE)
 		{
 		  char *ptr;

 		  /* the command "title" is specially treated.  */
 		  if (state > 1)
 		    {
 		      /* The next title is found.  */
 		      num_entries++;
 		      config_entries[config_len++] = 0;
 		      prev_menu_len = menu_len;
 		      prev_config_len = config_len;
 		    }
 		  else
 		    {
 		      /* The first title is found.  */
 		      menu_len = prev_menu_len;
 		      config_len = prev_config_len;
 		    }

 		  /* Reset the state.  */
 		  state = 1;

 		  /* Copy title into menu area.  */
 		  ptr = skip_to (1, cmdline);
 		  while ((menu_entries[menu_len++] = *(ptr++)) != 0)
 		    ;
 		}
 	      else if (! state)
 		{
 		  /* Run a command found is possible.  */
 		  if (builtin->flags & BUILTIN_MENU)
 		    {
 		      char *arg = skip_to (1, cmdline);
 		      (builtin->func) (arg, BUILTIN_MENU);
 		      errnum = 0;
 		    }
 		  else
 		    /* Ignored.  */
 		    continue;
 		}
 	      else
 		{
 		  char *ptr = cmdline;

 		  state++;
 		  /* Copy config file data to config area.  */
 		  while ((config_entries[config_len++] = *ptr++) != 0)
 		    ;
 		}
 	    }

 	  if (state > 1)
 	    {
 	      /* Finish the last entry.  */
 	      num_entries++;
 	      config_entries[config_len++] = 0;
 	    }
 	  else
 	    {
 	      menu_len = prev_menu_len;
 	      config_len = prev_config_len;
 	    }

 	  menu_entries[menu_len++] = 0;
 	  config_entries[config_len++] = 0;
 	  grub_memmove (config_entries + config_len, menu_entries, menu_len);
 	  menu_entries = config_entries + config_len;

 	  grub_close ();
 	}

       if (! num_entries)
 	{
 	  /* If no acceptable config file, goto command-line, starting
 	     heap from where the config entries would have been stored
 	     if there were any.  */
 	  enter_cmdline (config_entries, 1);
 	}
       else
 	{
 	  /* Run menu interface.  */
 	  run_menu (menu_entries, config_entries, num_entries,
 		    menu_entries + menu_len, default_entry);
 	}
     }
 }
	/*
	* GRUB -- GRand Unified Bootloader
	* Copyright (C) 1996 Erich Boleyn <erich@uruk.org>
	* Copyright (C) 2000 Free Software Foundation, Inc.
	*
	* This program is free software; you can redistribute it and/or modify
	* it under the terms of the GNU General Public License as published by
	* the Free Software Foundation; either version 2 of the License, or
	* (at your option) any later version.
	*
	* This program is distributed in the hope that it will be useful,
	* but WITHOUT ANY WARRANTY; without even the implied warranty of
	* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
	* GNU General Public License for more details.
	*
	* You should have received a copy of the GNU General Public License
	* along with this program; if not, write to the Free Software
	* Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
	*/

	#include "shared.h"

	grub_jmp_buf restart_env;

	static char *
	get_entry (char *list, int num, int nested)
	{
	int i;

	for (i = 0; i < num; i++)
	{
	do
	{
	while (*(list++));
	}
	while (nested && *(list++));
	}

	return list;
	}


	static void
	print_entries (int y, int size, int first, char *menu_entries)
	{
	int i;

	gotoxy (77, y + 1);

	if (first)
	putchar (DISP_UP);
	else
	putchar (' ');

	menu_entries = get_entry (menu_entries, first, 0);

	for (i = 1; i <= size; i++)
	{
	int j = 0;

	gotoxy (3, y + i);

	while (*menu_entries)
	{
	if (j < 71)
	{
	putchar (*menu_entries);
	j++;
	}

	menu_entries++;
	}

	if (*(menu_entries - 1))
	menu_entries++;

	for (; j < 71; j++)
	putchar (' ');
	}

	gotoxy (77, y + size);

	if (*menu_entries)
	putchar (DISP_DOWN);
	else
	putchar (' ');
	}


	static void
	print_border (int y, int size)
	{
	int i;

	#ifndef GRUB_UTIL
	/* Color the menu. The menu is 75 * 14 characters. */
	for (i = 0; i < 14; i++)
	{
	int j;
	for (j = 0; j < 75; j++)
	{
	gotoxy (j + 1, i + y);
	set_attrib (normal_color);
	}
	}
	#endif

	gotoxy (1, y);

	putchar (DISP_UL);
	for (i = 0; i < 73; i++)
	putchar (DISP_HORIZ);
	putchar (DISP_UR);

	i = 1;

	while (1)
	{
	gotoxy (1, y + i);

	if (i > size)
	break;

	putchar (DISP_VERT);
	gotoxy (75, y + i);
	putchar (DISP_VERT);

	i++;
	}

	putchar (DISP_LL);
	for (i = 0; i < 73; i++)
	putchar (DISP_HORIZ);
	putchar (DISP_LR);
	}

	static void
	set_line (int y, int attr)
	{
	int x;

	for (x = 2; x < 75; x++)
	{
	gotoxy (x, y);
	set_attrib (attr);
	}
	}

	/* Set the attribute of the line Y to normal state. */
	static void
	set_line_normal (int y)
	{
	#ifdef GRUB_UTIL
	set_line (y, A_NORMAL);
	#else
	set_line (y, normal_color);
	#endif
	}

	/* Set the attribute of the line Y to highlight state. */
	static void
	set_line_highlight (int y)
	{
	#ifdef GRUB_UTIL
	set_line (y, A_REVERSE);
	#else
	set_line (y, highlight_color);
	#endif
	}

	static void
	run_menu (char menu_entries, char config_entries, int num_entries,
	char *heap, int entryno)
	{
	int c, time1, time2 = -1, first_entry = 0;
	char *cur_entry;

	/*
	* Main loop for menu UI.
	*/

	restart:
	while (entryno > 11)
	{
	first_entry++;
	entryno--;
	}

	init_page ();
	#ifndef GRUB_UTIL
	nocursor ();
	#endif

	print_border (3, 12);

	#ifdef GRUB_UTIL
	grub_printf ("\n
	Use the up and down arrows for selecting which entry is highlighted.\n");
	#else
	grub_printf ("\n
	Use the %c and %c keys for selecting which entry is highlighted.\n",
	DISP_UP, DISP_DOWN);
	#endif

	if (password)
	{
	printf (" Press enter to boot the selected OS or \'p\' to enter a
	password to unlock the next set of features.");
	}
	else
	{
	if (config_entries)
	printf (" Press enter to boot the selected OS, \'e\' to edit the
	commands before booting, or \'c\' for a command-line.");
	else
	printf (
	" Press \'b\' to boot, \'e\' to edit the selected command in the
	boot sequence, \'c\' for a command-line, \'o\' to open a new line
	after (\'O\' for before) the selected line, \'d\' to remove the
	selected line, or escape to go back to the main menu.");
	}

	print_entries (3, 12, first_entry, menu_entries);

	/* highlight initial line */
	set_line_highlight (4 + entryno);

	/* XX using RT clock now, need to initialize value */
	while ((time1 = getrtsecs()) == 0xFF);

	while (1)
	{
	/* initilize to NULL just in case... */
	cur_entry = NULL;

	if (grub_timeout >= 0 && (time1 = getrtsecs()) != time2 && time1 != 0xFF)
	{
	if (grub_timeout <= 0)
	{
	grub_timeout = -1;
	break;
	}

	/* else not booting yet! */
	time2 = time1;
	gotoxy (3, 22);
	printf ("The highlighted entry will be booted automatically in %d seconds. ", grub_timeout);
	gotoxy (74, 4 + entryno);
	grub_timeout--;
	}

	if (checkkey () != -1)
	{
	c = getkey ();

	if (grub_timeout >= 0)
	{
	gotoxy (3, 22);
	printf (" ");
	grub_timeout = -1;
	fallback_entry = -1;
	gotoxy (74, 4 + entryno);
	}

	if ((c == KEY_UP) \|\| (ASCII_CHAR (c) == 16))
	{
	if (entryno > 0)
	{
	set_line_normal (4 + entryno);
	entryno--;
	set_line_highlight (4 + entryno);
	}
	else if (first_entry > 0)
	{
	first_entry--;
	print_entries (3, 12, first_entry, menu_entries);
	set_line_highlight (4);
	}
	}
	if (((c == KEY_DOWN) \|\| (ASCII_CHAR (c) == 14))
	&& (first_entry + entryno + 1) < num_entries)
	{
	if (entryno < 11)
	{
	set_line_normal (4 + entryno);
	entryno++;
	set_line_highlight (4 + entryno);
	}
	else if (num_entries > 12 + first_entry)
	{
	first_entry++;
	print_entries (3, 12, first_entry, menu_entries);
	set_line_highlight (15);
	}
	}

	c = ASCII_CHAR (c);

	if (config_entries)
	{
	if ((c == '\n') \|\| (c == '\r'))
	break;
	}
	else
	{
	if ((c == 'd') \|\| (c == 'o') \|\| (c == 'O'))
	{
	set_line_normal (4 + entryno);

	/* insert after is almost exactly like insert before */
	if (c == 'o')
	{
	/* But `o' differs from `O', since it may causes
	the menu screen to scroll up. */
	if (entryno < 11)
	entryno++;
	else
	first_entry++;

	c = 'O';
	}

	cur_entry = get_entry (menu_entries,
	first_entry + entryno,
	0);

	if (c == 'O')
	{
	memmove (cur_entry + 2, cur_entry,
	((int) heap) - ((int) cur_entry));

	cur_entry[0] = ' ';
	cur_entry[1] = 0;

	heap += 2;

	num_entries++;
	}
	else if (num_entries > 0)
	{
	char *ptr = get_entry(menu_entries,
	first_entry + entryno + 1,
	0);

	memmove (cur_entry, ptr, ((int) heap) - ((int) ptr));
	heap -= (((int) ptr) - ((int) cur_entry));

	num_entries--;

	if (entryno >= num_entries)
	entryno--;
	if (first_entry && num_entries < 12 + first_entry)
	first_entry--;
	}

	print_entries (3, 12, first_entry, menu_entries);
	set_line_highlight (4 + entryno);
	}

	cur_entry = menu_entries;
	if (c == 27)
	return;
	if (c == 'b')
	break;
	}

	if (password)
	{
	if (c == 'p')
	{
	/* Do password check here! */
	char entered[32];
	char *pptr = password;

	gotoxy (1, 21);

	/* Wipe out the previously entered password */
	memset (entered, 0, sizeof (entered));
	get_cmdline (" Password: ", entered, 31, '*', 0);

	while (! isspace (pptr) && pptr)
	pptr++;

	/* Make sure that PASSWORD is NUL-terminated. */
	*pptr++ = 0;

	if (! strcmp (password, entered))
	{
	char *new_file = config_file;
	while (isspace (*pptr))
	pptr++;
	while (((new_file++) = (pptr++)) != 0)
	;
	return;
	}
	else
	{
	printf ("Failed!\n Press any key to continue...");
	getkey ();
	goto restart;
	}
	}
	}
	else
	{
	if (c == 'e')
	{
	int num_entries = 0, i = 0;
	char *new_heap;

	if (config_entries)
	{
	new_heap = heap;
	cur_entry = get_entry (config_entries,
	first_entry + entryno,
	1);
	}
	else
	{
	/* safe area! */
	new_heap = heap + NEW_HEAPSIZE + 1;
	cur_entry = get_entry (menu_entries,
	first_entry + entryno,
	0);
	}

	do
	{
	while ((*(new_heap++) = cur_entry[i++]) != 0);
	num_entries++;
	}
	while (config_entries && cur_entry[i]);

	/* this only needs to be done if config_entries is non-NULL,
	but it doesn't hurt to do it always */
	*(new_heap++) = 0;

	if (config_entries)
	run_menu (heap, NULL, num_entries, new_heap, 0);
	else
	{
	cls ();
	print_cmdline_message (0);

	new_heap = heap + NEW_HEAPSIZE + 1;

	saved_drive = boot_drive;
	saved_partition = install_partition;
	current_drive = 0xFF;

	if (! get_cmdline (PACKAGE " edit> ", new_heap,
	NEW_HEAPSIZE + 1, 0, 1))
	{
	int j = 0;

	/* get length of new command */
	while (new_heap[j++])
	;

	if (j < 2)
	{
	j = 2;
	new_heap[0] = ' ';
	new_heap[1] = 0;
	}

	/* align rest of commands properly */
	memmove (cur_entry + j, cur_entry + i,
	((int) heap) - (((int) cur_entry) + i));

	/* copy command to correct area */
	memmove (cur_entry, new_heap, j);

	heap += (j - i);
	}
	}

	goto restart;
	}
	if (c == 'c')
	{
	enter_cmdline (heap, 0);
	goto restart;
	}
	#ifdef GRUB_UTIL
	if (c == 'q')
	{
	/* The same as ``quit''. */
	stop ();
	}
	#endif
	}
	}
	}

	/*
	* Attempt to boot an entry.
	*/

	while (1)
	{
	cls ();

	if (config_entries)
	printf (" Booting \'%s\'\n\n",
	get_entry (menu_entries, first_entry + entryno, 0));
	else
	printf (" Booting command-list\n\n");

	if (! cur_entry)
	cur_entry = get_entry (config_entries, first_entry + entryno, 1);

	if (run_script (cur_entry, heap))
	{
	if (fallback_entry < 0)
	break;
	else
	{
	cur_entry = NULL;
	first_entry = 0;
	entryno = fallback_entry;
	fallback_entry = -1;
	}
	}
	else
	break;
	}

	goto restart;
	}


	static int
	get_line_from_config (char *cmdline, int maxlen)
	{
	int pos = 0, literal = 0, comment = 0;
	char c; /* since we're loading it a byte at a time! */

	while (grub_read (&c, 1))
	{
	/* translate characters first! */
	if (c == '\\' && ! literal)
	{
	literal = 1;
	continue;
	}
	if (c == '\r')
	continue;
	if ((c == '\t') \|\| (literal && (c == '\n')))
	c = ' ';

	literal = 0;

	if (comment)
	{
	if (c == '\n')
	comment = 0;
	}
	else if (! pos)
	{
	if (c == '#')
	comment = 1;
	else if ((c != ' ') && (c != '\n'))
	cmdline[pos++] = c;
	}
	else
	{
	if (c == '\n')
	break;

	if (pos < maxlen)
	cmdline[pos++] = c;
	}
	}

	cmdline[pos] = 0;

	return pos;
	}


	/* This is the starting function in C. */
	void
	cmain (void)
	{
	int config_len, menu_len, num_entries;
	char config_entries, menu_entries;
	char kill = (char ) KILL_BUF;

	/* Initialize the environment for restarting Stage 2. */
	grub_setjmp (restart_env);

	/* Initialize the kill buffer. */
	*kill = 0;

	/* Never return. */
	for (;;)
	{
	config_len = 0;
	menu_len = 0;
	num_entries = 0;
	config_entries = (char *) (mbi.mmap_addr + mbi.mmap_length);
	menu_entries = (char *) MENU_BUF;
	init_config ();

	/* Here load the configuration file. */

	#ifdef GRUB_UTIL
	if (use_config_file && grub_open (config_file))
	#else
	if (grub_open (config_file))
	#endif
	{
	/* STATE 0: Before any title command.
	STATE 1: In a title command.
	STATE >1: In a entry after a title command. */
	int state = 0, prev_config_len = 0, prev_menu_len = 0;
	char *cmdline;

	cmdline = (char *) CMDLINE_BUF;
	while (get_line_from_config (cmdline, NEW_HEAPSIZE))
	{
	struct builtin *builtin;

	/* Get the pointer to the builtin structure. */
	builtin = find_command (cmdline);
	errnum = 0;
	if (! builtin)
	/* Unknown command. Just skip now. */
	continue;

	if (builtin->flags & BUILTIN_TITLE)
	{
	char *ptr;

	/* the command "title" is specially treated. */
	if (state > 1)
	{
	/* The next title is found. */
	num_entries++;
	config_entries[config_len++] = 0;
	prev_menu_len = menu_len;
	prev_config_len = config_len;
	}
	else
	{
	/* The first title is found. */
	menu_len = prev_menu_len;
	config_len = prev_config_len;
	}

	/* Reset the state. */
	state = 1;

	/* Copy title into menu area. */
	ptr = skip_to (1, cmdline);
	while ((menu_entries[menu_len++] = *(ptr++)) != 0)
	;
	}
	else if (! state)
	{
	/* Run a command found is possible. */
	if (builtin->flags & BUILTIN_MENU)
	{
	char *arg = skip_to (1, cmdline);
	(builtin->func) (arg, BUILTIN_MENU);
	errnum = 0;
	}
	else
	/* Ignored. */
	continue;
	}
	else
	{
	char *ptr = cmdline;

	state++;
	/* Copy config file data to config area. */
	while ((config_entries[config_len++] = *ptr++) != 0)
	;
	}
	}

	if (state > 1)
	{
	/* Finish the last entry. */
	num_entries++;
	config_entries[config_len++] = 0;
	}
	else
	{
	menu_len = prev_menu_len;
	config_len = prev_config_len;
	}

	menu_entries[menu_len++] = 0;
	config_entries[config_len++] = 0;
	grub_memmove (config_entries + config_len, menu_entries, menu_len);
	menu_entries = config_entries + config_len;

	grub_close ();
	}

	if (! num_entries)
	{
	/* If no acceptable config file, goto command-line, starting
	heap from where the config entries would have been stored
	if there were any. */
	enter_cmdline (config_entries, 1);
	}
	else
	{
	/* Run menu interface. */
	run_menu (menu_entries, config_entries, num_entries,
	menu_entries + menu_len, default_entry);
	}
	}
	}