| /* |
| * 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 (! auth && 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 (! auth && 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++; |
| |
| /* 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 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); |
| } |
| } |
| } |