Albert ARIBAUD
12 years ago
commit
1812201997
@ -0,0 +1,517 @@ |
||||
/*
|
||||
* (C) Copyright 2011-2013 Pali Rohár <pali.rohar@gmail.com> |
||||
* |
||||
* See file CREDITS for list of people who contributed to this |
||||
* project. |
||||
* |
||||
* 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., 59 Temple Place, Suite 330, Boston, |
||||
* MA 02111-1307 USA |
||||
*/ |
||||
|
||||
#include <common.h> |
||||
#include <command.h> |
||||
#include <ansi.h> |
||||
#include <menu.h> |
||||
#include <hush.h> |
||||
#include <watchdog.h> |
||||
#include <malloc.h> |
||||
#include <linux/string.h> |
||||
|
||||
/* maximum bootmenu entries */ |
||||
#define MAX_COUNT 99 |
||||
|
||||
/* maximal size of bootmenu env
|
||||
* 9 = strlen("bootmenu_") |
||||
* 2 = strlen(MAX_COUNT) |
||||
* 1 = NULL term |
||||
*/ |
||||
#define MAX_ENV_SIZE (9 + 2 + 1) |
||||
|
||||
struct bootmenu_entry { |
||||
unsigned short int num; /* unique number 0 .. MAX_COUNT */ |
||||
char key[3]; /* key identifier of number */ |
||||
char *title; /* title of entry */ |
||||
char *command; /* hush command of entry */ |
||||
struct bootmenu_data *menu; /* this bootmenu */ |
||||
struct bootmenu_entry *next; /* next menu entry (num+1) */ |
||||
}; |
||||
|
||||
struct bootmenu_data { |
||||
int delay; /* delay for autoboot */ |
||||
int active; /* active menu entry */ |
||||
int count; /* total count of menu entries */ |
||||
struct bootmenu_entry *first; /* first menu entry */ |
||||
}; |
||||
|
||||
enum bootmenu_key { |
||||
KEY_NONE = 0, |
||||
KEY_UP, |
||||
KEY_DOWN, |
||||
KEY_SELECT, |
||||
}; |
||||
|
||||
static char *bootmenu_getoption(unsigned short int n) |
||||
{ |
||||
char name[MAX_ENV_SIZE] = "bootmenu_"; |
||||
|
||||
if (n > MAX_COUNT) |
||||
return NULL; |
||||
|
||||
sprintf(name + 9, "%d", n); |
||||
return getenv(name); |
||||
} |
||||
|
||||
static void bootmenu_print_entry(void *data) |
||||
{ |
||||
struct bootmenu_entry *entry = data; |
||||
int reverse = (entry->menu->active == entry->num); |
||||
|
||||
/*
|
||||
* Move cursor to line where the entry will be drown (entry->num) |
||||
* First 3 lines contain bootmenu header + 1 empty line |
||||
*/ |
||||
printf(ANSI_CURSOR_POSITION, entry->num + 4, 1); |
||||
|
||||
puts(" "); |
||||
|
||||
if (reverse) |
||||
puts(ANSI_COLOR_REVERSE); |
||||
|
||||
puts(entry->title); |
||||
|
||||
if (reverse) |
||||
puts(ANSI_COLOR_RESET); |
||||
} |
||||
|
||||
static void bootmenu_autoboot_loop(struct bootmenu_data *menu, |
||||
enum bootmenu_key *key, int *esc) |
||||
{ |
||||
int i, c; |
||||
|
||||
if (menu->delay > 0) { |
||||
printf(ANSI_CURSOR_POSITION, menu->count + 5, 1); |
||||
printf(" Hit any key to stop autoboot: %2d ", menu->delay); |
||||
} |
||||
|
||||
while (menu->delay > 0) { |
||||
for (i = 0; i < 100; ++i) { |
||||
if (!tstc()) { |
||||
WATCHDOG_RESET(); |
||||
mdelay(10); |
||||
continue; |
||||
} |
||||
|
||||
menu->delay = -1; |
||||
c = getc(); |
||||
|
||||
switch (c) { |
||||
case '\e': |
||||
*esc = 1; |
||||
*key = KEY_NONE; |
||||
break; |
||||
case '\r': |
||||
*key = KEY_SELECT; |
||||
break; |
||||
default: |
||||
*key = KEY_NONE; |
||||
break; |
||||
} |
||||
|
||||
break; |
||||
} |
||||
|
||||
if (menu->delay < 0) |
||||
break; |
||||
|
||||
--menu->delay; |
||||
printf("\b\b\b%2d ", menu->delay); |
||||
} |
||||
|
||||
printf(ANSI_CURSOR_POSITION, menu->count + 5, 1); |
||||
puts(ANSI_CLEAR_LINE); |
||||
|
||||
if (menu->delay == 0) |
||||
*key = KEY_SELECT; |
||||
} |
||||
|
||||
static void bootmenu_loop(struct bootmenu_data *menu, |
||||
enum bootmenu_key *key, int *esc) |
||||
{ |
||||
int c; |
||||
|
||||
while (!tstc()) { |
||||
WATCHDOG_RESET(); |
||||
mdelay(10); |
||||
} |
||||
|
||||
c = getc(); |
||||
|
||||
switch (*esc) { |
||||
case 0: |
||||
/* First char of ANSI escape sequence '\e' */ |
||||
if (c == '\e') { |
||||
*esc = 1; |
||||
*key = KEY_NONE; |
||||
} |
||||
break; |
||||
case 1: |
||||
/* Second char of ANSI '[' */ |
||||
if (c == '[') { |
||||
*esc = 2; |
||||
*key = KEY_NONE; |
||||
} else { |
||||
*esc = 0; |
||||
} |
||||
break; |
||||
case 2: |
||||
case 3: |
||||
/* Third char of ANSI (number '1') - optional */ |
||||
if (*esc == 2 && c == '1') { |
||||
*esc = 3; |
||||
*key = KEY_NONE; |
||||
break; |
||||
} |
||||
|
||||
*esc = 0; |
||||
|
||||
/* ANSI 'A' - key up was pressed */ |
||||
if (c == 'A') |
||||
*key = KEY_UP; |
||||
/* ANSI 'B' - key down was pressed */ |
||||
else if (c == 'B') |
||||
*key = KEY_DOWN; |
||||
/* other key was pressed */ |
||||
else |
||||
*key = KEY_NONE; |
||||
|
||||
break; |
||||
} |
||||
|
||||
/* enter key was pressed */ |
||||
if (c == '\r') |
||||
*key = KEY_SELECT; |
||||
} |
||||
|
||||
static char *bootmenu_choice_entry(void *data) |
||||
{ |
||||
struct bootmenu_data *menu = data; |
||||
struct bootmenu_entry *iter; |
||||
enum bootmenu_key key = KEY_NONE; |
||||
int esc = 0; |
||||
int i; |
||||
|
||||
while (1) { |
||||
if (menu->delay >= 0) { |
||||
/* Autoboot was not stopped */ |
||||
bootmenu_autoboot_loop(menu, &key, &esc); |
||||
} else { |
||||
/* Some key was pressed, so autoboot was stopped */ |
||||
bootmenu_loop(menu, &key, &esc); |
||||
} |
||||
|
||||
switch (key) { |
||||
case KEY_UP: |
||||
if (menu->active > 0) |
||||
--menu->active; |
||||
/* no menu key selected, regenerate menu */ |
||||
return NULL; |
||||
case KEY_DOWN: |
||||
if (menu->active < menu->count - 1) |
||||
++menu->active; |
||||
/* no menu key selected, regenerate menu */ |
||||
return NULL; |
||||
case KEY_SELECT: |
||||
iter = menu->first; |
||||
for (i = 0; i < menu->active; ++i) |
||||
iter = iter->next; |
||||
return iter->key; |
||||
default: |
||||
break; |
||||
} |
||||
} |
||||
|
||||
/* never happens */ |
||||
debug("bootmenu: this should not happen"); |
||||
return NULL; |
||||
} |
||||
|
||||
static void bootmenu_destroy(struct bootmenu_data *menu) |
||||
{ |
||||
struct bootmenu_entry *iter = menu->first; |
||||
struct bootmenu_entry *next; |
||||
|
||||
while (iter) { |
||||
next = iter->next; |
||||
free(iter->title); |
||||
free(iter->command); |
||||
free(iter); |
||||
iter = next; |
||||
} |
||||
free(menu); |
||||
} |
||||
|
||||
static struct bootmenu_data *bootmenu_create(int delay) |
||||
{ |
||||
unsigned short int i = 0; |
||||
const char *option; |
||||
struct bootmenu_data *menu; |
||||
struct bootmenu_entry *iter = NULL; |
||||
|
||||
int len; |
||||
char *sep; |
||||
struct bootmenu_entry *entry; |
||||
|
||||
menu = malloc(sizeof(struct bootmenu_data)); |
||||
if (!menu) |
||||
return NULL; |
||||
|
||||
menu->delay = delay; |
||||
menu->active = 0; |
||||
menu->first = NULL; |
||||
|
||||
while ((option = bootmenu_getoption(i))) { |
||||
sep = strchr(option, '='); |
||||
if (!sep) { |
||||
printf("Invalid bootmenu entry: %s\n", option); |
||||
break; |
||||
} |
||||
|
||||
entry = malloc(sizeof(struct bootmenu_entry)); |
||||
if (!entry) |
||||
goto cleanup; |
||||
|
||||
len = sep-option; |
||||
entry->title = malloc(len + 1); |
||||
if (!entry->title) { |
||||
free(entry); |
||||
goto cleanup; |
||||
} |
||||
memcpy(entry->title, option, len); |
||||
entry->title[len] = 0; |
||||
|
||||
len = strlen(sep + 1); |
||||
entry->command = malloc(len + 1); |
||||
if (!entry->command) { |
||||
free(entry->title); |
||||
free(entry); |
||||
goto cleanup; |
||||
} |
||||
memcpy(entry->command, sep + 1, len); |
||||
entry->command[len] = 0; |
||||
|
||||
sprintf(entry->key, "%d", i); |
||||
|
||||
entry->num = i; |
||||
entry->menu = menu; |
||||
entry->next = NULL; |
||||
|
||||
if (!iter) |
||||
menu->first = entry; |
||||
else |
||||
iter->next = entry; |
||||
|
||||
iter = entry; |
||||
++i; |
||||
|
||||
if (i == MAX_COUNT - 1) |
||||
break; |
||||
} |
||||
|
||||
/* Add U-Boot console entry at the end */ |
||||
if (i <= MAX_COUNT - 1) { |
||||
entry = malloc(sizeof(struct bootmenu_entry)); |
||||
if (!entry) |
||||
goto cleanup; |
||||
|
||||
entry->title = strdup("U-Boot console"); |
||||
if (!entry->title) { |
||||
free(entry); |
||||
goto cleanup; |
||||
} |
||||
|
||||
entry->command = strdup(""); |
||||
if (!entry->command) { |
||||
free(entry->title); |
||||
free(entry); |
||||
goto cleanup; |
||||
} |
||||
|
||||
sprintf(entry->key, "%d", i); |
||||
|
||||
entry->num = i; |
||||
entry->menu = menu; |
||||
entry->next = NULL; |
||||
|
||||
if (!iter) |
||||
menu->first = entry; |
||||
else |
||||
iter->next = entry; |
||||
|
||||
iter = entry; |
||||
++i; |
||||
} |
||||
|
||||
menu->count = i; |
||||
return menu; |
||||
|
||||
cleanup: |
||||
bootmenu_destroy(menu); |
||||
return NULL; |
||||
} |
||||
|
||||
static void bootmenu_show(int delay) |
||||
{ |
||||
int init = 0; |
||||
void *choice = NULL; |
||||
char *title = NULL; |
||||
char *command = NULL; |
||||
struct menu *menu; |
||||
struct bootmenu_data *bootmenu; |
||||
struct bootmenu_entry *iter; |
||||
char *option, *sep; |
||||
|
||||
/* If delay is 0 do not create menu, just run first entry */ |
||||
if (delay == 0) { |
||||
option = bootmenu_getoption(0); |
||||
if (!option) { |
||||
puts("bootmenu option 0 was not found\n"); |
||||
return; |
||||
} |
||||
sep = strchr(option, '='); |
||||
if (!sep) { |
||||
puts("bootmenu option 0 is invalid\n"); |
||||
return; |
||||
} |
||||
run_command(sep+1, 0); |
||||
return; |
||||
} |
||||
|
||||
bootmenu = bootmenu_create(delay); |
||||
if (!bootmenu) |
||||
return; |
||||
|
||||
menu = menu_create(NULL, bootmenu->delay, 1, bootmenu_print_entry, |
||||
bootmenu_choice_entry, bootmenu); |
||||
if (!menu) { |
||||
bootmenu_destroy(bootmenu); |
||||
return; |
||||
} |
||||
|
||||
for (iter = bootmenu->first; iter; iter = iter->next) { |
||||
if (!menu_item_add(menu, iter->key, iter)) |
||||
goto cleanup; |
||||
} |
||||
|
||||
/* Default menu entry is always first */ |
||||
menu_default_set(menu, "0"); |
||||
|
||||
puts(ANSI_CURSOR_HIDE); |
||||
puts(ANSI_CLEAR_CONSOLE); |
||||
printf(ANSI_CURSOR_POSITION, 1, 1); |
||||
|
||||
init = 1; |
||||
|
||||
if (menu_get_choice(menu, &choice)) { |
||||
iter = choice; |
||||
title = strdup(iter->title); |
||||
command = strdup(iter->command); |
||||
} |
||||
|
||||
cleanup: |
||||
menu_destroy(menu); |
||||
bootmenu_destroy(bootmenu); |
||||
|
||||
if (init) { |
||||
puts(ANSI_CURSOR_SHOW); |
||||
puts(ANSI_CLEAR_CONSOLE); |
||||
printf(ANSI_CURSOR_POSITION, 1, 1); |
||||
} |
||||
|
||||
if (title && command) { |
||||
debug("Starting entry '%s'\n", title); |
||||
free(title); |
||||
run_command(command, 0); |
||||
free(command); |
||||
} |
||||
|
||||
#ifdef CONFIG_POSTBOOTMENU |
||||
run_command(CONFIG_POSTBOOTMENU, 0); |
||||
#endif |
||||
} |
||||
|
||||
void menu_display_statusline(struct menu *m) |
||||
{ |
||||
struct bootmenu_entry *entry; |
||||
struct bootmenu_data *menu; |
||||
|
||||
if (menu_default_choice(m, (void *)&entry) < 0) |
||||
return; |
||||
|
||||
menu = entry->menu; |
||||
|
||||
printf(ANSI_CURSOR_POSITION, 1, 1); |
||||
puts(ANSI_CLEAR_LINE); |
||||
printf(ANSI_CURSOR_POSITION, 2, 1); |
||||
puts(" *** U-Boot Boot Menu ***"); |
||||
puts(ANSI_CLEAR_LINE_TO_END); |
||||
printf(ANSI_CURSOR_POSITION, 3, 1); |
||||
puts(ANSI_CLEAR_LINE); |
||||
|
||||
/* First 3 lines are bootmenu header + 2 empty lines between entries */ |
||||
printf(ANSI_CURSOR_POSITION, menu->count + 5, 1); |
||||
puts(ANSI_CLEAR_LINE); |
||||
printf(ANSI_CURSOR_POSITION, menu->count + 6, 1); |
||||
puts(" Press UP/DOWN to move, ENTER to select"); |
||||
puts(ANSI_CLEAR_LINE_TO_END); |
||||
printf(ANSI_CURSOR_POSITION, menu->count + 7, 1); |
||||
puts(ANSI_CLEAR_LINE); |
||||
} |
||||
|
||||
#ifdef CONFIG_MENU_SHOW |
||||
int menu_show(int bootdelay) |
||||
{ |
||||
bootmenu_show(bootdelay); |
||||
return -1; /* -1 - abort boot and run monitor code */ |
||||
} |
||||
#endif |
||||
|
||||
int do_bootmenu(cmd_tbl_t *cmdtp, int flag, int argc, char *const argv[]) |
||||
{ |
||||
char *delay_str = NULL; |
||||
int delay = 10; |
||||
|
||||
#if defined(CONFIG_BOOTDELAY) && (CONFIG_BOOTDELAY >= 0) |
||||
delay = CONFIG_BOOTDELAY; |
||||
#endif |
||||
|
||||
if (argc >= 2) |
||||
delay_str = argv[1]; |
||||
|
||||
if (!delay_str) |
||||
delay_str = getenv("bootmenu_delay"); |
||||
|
||||
if (delay_str) |
||||
delay = (int)simple_strtol(delay_str, NULL, 10); |
||||
|
||||
bootmenu_show(delay); |
||||
return 0; |
||||
} |
||||
|
||||
U_BOOT_CMD( |
||||
bootmenu, 2, 1, do_bootmenu, |
||||
"ANSI terminal bootmenu", |
||||
"[delay]\n" |
||||
" - show ANSI terminal bootmenu with autoboot delay" |
||||
); |
||||
Some files were not shown because too many files have changed in this diff Show More
Loading…
Reference in new issue