| /* |
| * GRUB -- GRand Unified Bootloader |
| * Copyright (C) 2000, 2001 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; |
| |
| #if defined(PRESET_MENU_STRING) || defined(SUPPORT_DISKLESS) |
| |
| # if defined(PRESET_MENU_STRING) |
| static const char *preset_menu = PRESET_MENU_STRING; |
| # elif defined(SUPPORT_DISKLESS) |
| /* Execute the command "bootp" automatically. */ |
| static const char *preset_menu = "bootp\n"; |
| # endif /* SUPPORT_DISKLESS */ |
| |
| static int preset_menu_offset; |
| |
| static int |
| open_preset_menu (void) |
| { |
| #ifdef GRUB_UTIL |
| /* Unless the user explicitly requests to use the preset menu, |
| always opening the preset menu fails in the grub shell. */ |
| if (! use_preset_menu) |
| return 0; |
| #endif /* GRUB_UTIL */ |
| |
| preset_menu_offset = 0; |
| return preset_menu != 0; |
| } |
| |
| static int |
| read_from_preset_menu (char *buf, int maxlen) |
| { |
| int len = grub_strlen (preset_menu + preset_menu_offset); |
| |
| if (len > maxlen) |
| len = maxlen; |
| |
| grub_memmove (buf, preset_menu + preset_menu_offset, len); |
| preset_menu_offset += len; |
| |
| return len; |
| } |
| |
| static void |
| close_preset_menu (void) |
| { |
| /* Disable the preset menu. */ |
| preset_menu = 0; |
| } |
| |
| #else /* ! PRESET_MENU_STRING && ! SUPPORT_DISKLESS */ |
| |
| #define open_preset_menu() 0 |
| #define read_from_preset_menu(buf, maxlen) 0 |
| #define close_preset_menu() |
| |
| #endif /* ! PRESET_MENU_STRING && ! SUPPORT_DISKLESS */ |
| |
| |
| 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; |
| int disp_up = DISP_UP; |
| int disp_down = DISP_DOWN; |
| |
| #ifdef SUPPORT_SERIAL |
| if (terminal & TERMINAL_SERIAL) |
| { |
| disp_up = ACS_UARROW; |
| disp_down = ACS_DARROW; |
| } |
| #endif /* SUPPORT_SERIAL */ |
| |
| gotoxy (77, y + 1); |
| |
| if (first) |
| grub_putchar (disp_up); |
| else |
| grub_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) |
| { |
| grub_putchar (*menu_entries); |
| j++; |
| } |
| |
| menu_entries++; |
| } |
| |
| if (*(menu_entries - 1)) |
| menu_entries++; |
| |
| for (; j < 71; j++) |
| grub_putchar (' '); |
| } |
| |
| gotoxy (77, y + size); |
| |
| if (*menu_entries) |
| grub_putchar (disp_down); |
| else |
| grub_putchar (' '); |
| } |
| |
| |
| static void |
| print_entries_raw (int size, int first, char *menu_entries) |
| { |
| int i; |
| |
| #define LINE_LENGTH 67 |
| |
| for (i = 0; i < LINE_LENGTH; i++) |
| grub_putchar ('-'); |
| grub_putchar ('\n'); |
| |
| for (i = first; i < size; i++) |
| { |
| /* grub's printf can't %02d so ... */ |
| if (i < 10) |
| grub_putchar (' '); |
| grub_printf ("%d: %s\n", i, get_entry (menu_entries, i, 0)); |
| } |
| |
| for (i = 0; i < LINE_LENGTH; i++) |
| grub_putchar ('-'); |
| grub_putchar ('\n'); |
| |
| #undef LINE_LENGTH |
| } |
| |
| |
| static void |
| print_border (int y, int size) |
| { |
| int i; |
| int disp_ul = DISP_UL; |
| int disp_ur = DISP_UR; |
| int disp_ll = DISP_LL; |
| int disp_lr = DISP_LR; |
| int disp_horiz = DISP_HORIZ; |
| int disp_vert = DISP_VERT; |
| |
| #ifdef SUPPORT_SERIAL |
| if (terminal & TERMINAL_SERIAL) |
| { |
| disp_ul = ACS_ULCORNER; |
| disp_ur = ACS_URCORNER; |
| disp_ll = ACS_LLCORNER; |
| disp_lr = ACS_LRCORNER; |
| disp_horiz = ACS_HLINE; |
| disp_vert = ACS_VLINE; |
| } |
| #endif /* SUPPORT_SERIAL */ |
| |
| #ifndef GRUB_UTIL |
| /* Color the menu. The menu is 75 * 14 characters. */ |
| # ifdef SUPPORT_SERIAL |
| if ((terminal & TERMINAL_CONSOLE) |
| # ifdef SUPPORT_HERCULES |
| || (terminal & TERMINAL_HERCULES) |
| # endif |
| ) |
| # endif |
| { |
| 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); |
| |
| grub_putchar (disp_ul); |
| for (i = 0; i < 73; i++) |
| grub_putchar (disp_horiz); |
| grub_putchar (disp_ur); |
| |
| i = 1; |
| |
| while (1) |
| { |
| gotoxy (1, y + i); |
| |
| if (i > size) |
| break; |
| |
| grub_putchar (disp_vert); |
| gotoxy (75, y + i); |
| grub_putchar (disp_vert); |
| |
| i++; |
| } |
| |
| grub_putchar (disp_ll); |
| for (i = 0; i < 73; i++) |
| grub_putchar (disp_horiz); |
| grub_putchar (disp_lr); |
| } |
| |
| static void |
| set_line (int y, int entryno, int attr, char *menu_entries) |
| { |
| int x; |
| |
| #ifdef SUPPORT_SERIAL |
| if (terminal & TERMINAL_SERIAL) |
| { |
| menu_entries = get_entry (menu_entries, entryno, 0); |
| gotoxy (2, y); |
| grub_putchar (' '); |
| for (x = 3; x < 75; x++) |
| { |
| if (*menu_entries && x < 71) |
| grub_putchar (*menu_entries++); |
| else |
| grub_putchar (' '); |
| } |
| } |
| else |
| #endif /* SUPPORT_SERIAL */ |
| { |
| for (x = 2; x < 75; x++) |
| { |
| gotoxy (x, y); |
| set_attrib (attr); |
| } |
| } |
| |
| gotoxy (74, y); |
| } |
| |
| /* Set the attribute of the line Y to normal state. */ |
| static void |
| set_line_normal (int y, int entryno, char *menu_entries) |
| { |
| #ifdef GRUB_UTIL |
| set_line (y, entryno, A_NORMAL, menu_entries); |
| #else |
| set_line (y, entryno, normal_color, menu_entries); |
| #endif |
| } |
| |
| /* Set the attribute of the line Y to highlight state. */ |
| static void |
| set_line_highlight (int y, int entryno, char *menu_entries) |
| { |
| #ifdef SUPPORT_SERIAL |
| if (terminal & TERMINAL_SERIAL) |
| grub_printf ("\e[7m"); |
| #endif /* SUPPORT_SERIAL */ |
| |
| #ifdef GRUB_UTIL |
| set_line (y, entryno, A_REVERSE, menu_entries); |
| #else |
| set_line (y, entryno, highlight_color, menu_entries); |
| #endif |
| |
| #ifdef SUPPORT_SERIAL |
| if (terminal & TERMINAL_SERIAL) |
| grub_printf ("\e[0m"); |
| #endif /* SUPPORT_SERIAL */ |
| } |
| |
| 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 = 0; |
| int disp_up = DISP_UP; |
| int disp_down = DISP_DOWN; |
| |
| /* |
| * Main loop for menu UI. |
| */ |
| |
| restart: |
| /* Dumb terminal always use all entries for display |
| invariant for TERMINAL_DUMB: first_entry == 0 */ |
| if (! (terminal & TERMINAL_DUMB)) |
| { |
| while (entryno > 11) |
| { |
| first_entry++; |
| entryno--; |
| } |
| } |
| |
| /* If the timeout was expired or wasn't set, force to show the menu |
| interface. */ |
| if (grub_timeout < 0) |
| show_menu = 1; |
| |
| /* If SHOW_MENU is false, don't display the menu until ESC is pressed. */ |
| if (! show_menu) |
| { |
| /* Get current time. */ |
| while ((time1 = getrtsecs ()) == 0xFF) |
| ; |
| |
| while (1) |
| { |
| /* Check if ESC is pressed. */ |
| if (checkkey () != -1 && ASCII_CHAR (getkey ()) == '\e') |
| { |
| grub_timeout = -1; |
| show_menu = 1; |
| break; |
| } |
| |
| /* If GRUB_TIMEOUT is expired, boot the default entry. */ |
| if (grub_timeout >=0 |
| && (time1 = getrtsecs ()) != time2 |
| && time1 != 0xFF) |
| { |
| if (grub_timeout <= 0) |
| { |
| grub_timeout = -1; |
| goto boot_entry; |
| } |
| |
| time2 = time1; |
| grub_timeout--; |
| |
| /* Print a message. */ |
| grub_printf ("\rPress `ESC' to enter the menu... %d ", |
| grub_timeout); |
| } |
| } |
| } |
| |
| /* Only display the menu if the user wants to see it. */ |
| if (show_menu) |
| { |
| /* Disable the auto fill mode. */ |
| auto_fill = 0; |
| |
| init_page (); |
| #ifndef GRUB_UTIL |
| # ifdef SUPPORT_SERIAL |
| if (terminal & TERMINAL_CONSOLE) |
| # endif /* SUPPORT_SERIAL */ |
| nocursor (); |
| #endif /* ! GRUB_UTIL */ |
| |
| if (! (terminal & TERMINAL_DUMB)) |
| print_border (3, 12); |
| |
| #ifdef GRUB_UTIL |
| /* In the grub shell, always use ACS_*. */ |
| disp_up = ACS_UARROW; |
| disp_down = ACS_DARROW; |
| #else /* ! GRUB_UTIL */ |
| # ifdef SUPPORT_SERIAL |
| if ((terminal & TERMINAL_CONSOLE) |
| # ifdef SUPPORT_HERCULES |
| || (terminal & TERMINAL_HERCULES) |
| # endif /* SUPPORT_HERCULES */ |
| ) |
| { |
| disp_up = DISP_UP; |
| disp_down = DISP_DOWN; |
| } |
| else |
| { |
| disp_up = ACS_UARROW; |
| disp_down = ACS_DARROW; |
| } |
| # endif /* SUPPORT_SERIAL */ |
| #endif /* ! GRUB_UTIL */ |
| |
| if (terminal & TERMINAL_DUMB) |
| print_entries_raw (num_entries, first_entry, menu_entries); |
| |
| grub_printf ("\n\ |
| Use the %c and %c keys to select which entry is highlighted.\n", |
| disp_up, disp_down); |
| |
| if (! auth && password) |
| { |
| printf ("\ |
| Press enter to boot the selected OS or \'p\' to enter a\n\ |
| password to unlock the next set of features."); |
| } |
| else |
| { |
| if (config_entries) |
| printf ("\ |
| Press enter to boot the selected OS, \'e\' to edit the\n\ |
| commands before booting, or \'c\' for a command-line."); |
| else |
| printf ("\ |
| Press \'b\' to boot, \'e\' to edit the selected command in the\n\ |
| boot sequence, \'c\' for a command-line, \'o\' to open a new line\n\ |
| after (\'O\' for before) the selected line, \'d\' to remove the\n\ |
| selected line, or escape to go back to the main menu."); |
| } |
| |
| if (terminal & TERMINAL_DUMB) |
| grub_printf ("\n\nThe selected entry is %d ", entryno); |
| else |
| { |
| print_entries (3, 12, first_entry, menu_entries); |
| |
| /* highlight initial line */ |
| set_line_highlight (4 + entryno, first_entry + entryno, |
| menu_entries); |
| } |
| } |
| |
| /* XX using RT clock now, need to initialize value */ |
| while ((time1 = getrtsecs()) == 0xFF); |
| |
| while (1) |
| { |
| /* Initialize 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; |
| |
| if (terminal & TERMINAL_DUMB) |
| grub_printf ("\r Entry %d will be booted automatically in %d seconds. ", |
| entryno, grub_timeout); |
| else |
| { |
| gotoxy (3, 22); |
| printf ("The highlighted entry will be booted automatically in %d seconds. ", grub_timeout); |
| gotoxy (74, 4 + entryno); |
| } |
| |
| grub_timeout--; |
| } |
| |
| /* Check for a keypress, however if TIMEOUT has been expired |
| (GRUB_TIMEOUT == -1) relax in GETKEY even if no key has been |
| pressed. |
| This avoids polling (relevant in the grub-shell and later on |
| in grub if interrupt driven I/O is done). */ |
| if ((checkkey () != -1) || (grub_timeout == -1)) |
| { |
| /* Key was pressed, show which entry is selected before GETKEY, |
| since we're comming in here also on GRUB_TIMEOUT == -1 and |
| hang in GETKEY */ |
| if (terminal & TERMINAL_DUMB) |
| grub_printf ("\r Highlighted entry is %d: ", entryno); |
| |
| c = translate_keycode (getkey ()); |
| |
| if (grub_timeout >= 0) |
| { |
| if (terminal & TERMINAL_DUMB) |
| grub_putchar ('\r'); |
| else |
| gotoxy (3, 22); |
| printf (" "); |
| grub_timeout = -1; |
| fallback_entry = -1; |
| if (! (terminal & TERMINAL_DUMB)) |
| gotoxy (74, 4 + entryno); |
| } |
| |
| /* We told them above (at least in SUPPORT_SERIAL) to use |
| '^' or 'v' so accept these keys. */ |
| if (c == 16 || c == '^') |
| { |
| if (terminal & TERMINAL_DUMB) |
| { |
| if (entryno > 0) |
| entryno--; |
| } |
| else |
| { |
| if (entryno > 0) |
| { |
| set_line_normal (4 + entryno, first_entry + entryno, |
| menu_entries); |
| entryno--; |
| set_line_highlight (4 + entryno, first_entry + entryno, |
| menu_entries); |
| } |
| else if (first_entry > 0) |
| { |
| first_entry--; |
| print_entries (3, 12, first_entry, menu_entries); |
| set_line_highlight (4, first_entry + entryno, |
| menu_entries); |
| } |
| } |
| } |
| if ((c == 14 || c == 'v') && first_entry + entryno + 1 < num_entries) |
| { |
| if (terminal & TERMINAL_DUMB) |
| entryno++; |
| else |
| if (entryno < 11) |
| { |
| set_line_normal (4 + entryno, first_entry + entryno, |
| menu_entries); |
| entryno++; |
| set_line_highlight (4 + entryno, first_entry + entryno, |
| menu_entries); |
| } |
| else if (num_entries > 12 + first_entry) |
| { |
| first_entry++; |
| print_entries (3, 12, first_entry, menu_entries); |
| set_line_highlight (15, first_entry + entryno, menu_entries); |
| } |
| } |
| |
| if (config_entries) |
| { |
| if ((c == '\n') || (c == '\r')) |
| break; |
| } |
| else |
| { |
| if ((c == 'd') || (c == 'o') || (c == 'O')) |
| { |
| if (! (terminal & TERMINAL_DUMB)) |
| set_line_normal (4 + entryno, first_entry + entryno, |
| menu_entries); |
| |
| /* 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 || (terminal & TERMINAL_DUMB)) |
| 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--; |
| } |
| |
| if (terminal & TERMINAL_DUMB) |
| { |
| grub_printf ("\n\n"); |
| print_entries_raw (num_entries, first_entry, |
| menu_entries); |
| grub_printf ("\n"); |
| } |
| else |
| { |
| print_entries (3, 12, first_entry, menu_entries); |
| set_line_highlight (4 + entryno, first_entry + entryno, |
| menu_entries); |
| } |
| } |
| |
| cur_entry = menu_entries; |
| if (c == 27) |
| return; |
| if (c == 'b') |
| break; |
| } |
| |
| if (! auth && password) |
| { |
| if (c == 'p') |
| { |
| /* Do password check here! */ |
| char entered[32]; |
| char *pptr = password; |
| |
| if (terminal & TERMINAL_DUMB) |
| grub_printf ("\r "); |
| else |
| 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 (! check_password (entered, password, password_type)) |
| { |
| char *new_file = config_file; |
| while (isspace (*pptr)) |
| pptr++; |
| |
| /* If *PPTR is NUL, then allow the user to use |
| privileged instructions, otherwise, load |
| another configuration file. */ |
| if (*pptr != 0) |
| { |
| while ((*(new_file++) = *(pptr++)) != 0) |
| ; |
| |
| /* Make sure that the user will not have |
| authority in the next configuration. */ |
| auth = 0; |
| return; |
| } |
| else |
| { |
| /* Now the user is superhuman. */ |
| auth = 1; |
| goto restart; |
| } |
| } |
| else |
| { |
| printf ("Failed!\n Press any key to continue..."); |
| getkey (); |
| goto restart; |
| } |
| } |
| } |
| else |
| { |
| if (c == 'e') |
| { |
| int new_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); |
| new_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, new_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. */ |
| |
| boot_entry: |
| /* Enable the auto fill mode. */ |
| auto_fill = 1; |
| |
| cls (); |
| |
| while (1) |
| { |
| 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); |
| |
| /* Set CURRENT_ENTRYNO for the command "savedefault". */ |
| current_entryno = first_entry + entryno; |
| |
| 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; |
| } |
| |
| show_menu = 1; |
| goto restart; |
| } |
| |
| |
| static int |
| get_line_from_config (char *cmdline, int maxlen, int read_from_file) |
| { |
| int pos = 0, literal = 0, comment = 0; |
| char c; /* since we're loading it a byte at a time! */ |
| |
| while (1) |
| { |
| if (read_from_file) |
| { |
| if (! grub_read (&c, 1)) |
| break; |
| } |
| else |
| { |
| if (! read_from_preset_menu (&c, 1)) |
| break; |
| } |
| |
| /* 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_buf = (char *) KILL_BUF; |
| |
| auto void reset (void); |
| void reset (void) |
| { |
| auto_fill = 1; |
| config_len = 0; |
| menu_len = 0; |
| num_entries = 0; |
| config_entries = (char *) mbi.drives_addr + mbi.drives_length; |
| menu_entries = (char *) MENU_BUF; |
| init_config (); |
| } |
| |
| /* Initialize the environment for restarting Stage 2. */ |
| grub_setjmp (restart_env); |
| |
| /* Initialize the kill buffer. */ |
| *kill_buf = 0; |
| |
| /* Never return. */ |
| for (;;) |
| { |
| int is_opened, is_preset; |
| |
| reset (); |
| |
| /* Here load the configuration file. */ |
| |
| #ifdef GRUB_UTIL |
| if (use_config_file) |
| #endif /* GRUB_UTIL */ |
| { |
| do |
| { |
| /* 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; |
| |
| /* Try the preset menu first. This will succeed at most once, |
| because close_preset_menu disables the preset menu. */ |
| is_opened = is_preset = open_preset_menu (); |
| if (! is_opened) |
| { |
| is_opened = grub_open (config_file); |
| errnum = ERR_NONE; |
| } |
| |
| if (! is_opened) |
| break; |
| |
| /* This is necessary, because the menu must be overrided. */ |
| reset (); |
| |
| cmdline = (char *) CMDLINE_BUF; |
| while (get_line_from_config (cmdline, NEW_HEAPSIZE, |
| ! is_preset)) |
| { |
| 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; |
| |
| /* Check if the default entry is present. Otherwise reset |
| it to fallback if fallback is valid, or to DEFAULT_ENTRY |
| if not. */ |
| if (default_entry >= num_entries) |
| { |
| if (fallback_entry < 0 || fallback_entry >= num_entries) |
| default_entry = 0; |
| else |
| default_entry = fallback_entry; |
| } |
| |
| if (is_preset) |
| close_preset_menu (); |
| else |
| grub_close (); |
| } |
| while (is_preset); |
| } |
| |
| 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); |
| } |
| } |
| } |