cbf97a87d9
(grub_menu_viewer_should_return): Likewise. * normal/main.c (GRUB_MOD_INIT (normal)): Call grub_menu_viewer_init. * normal/menu_text.c (run_menu): Enable menu switching. * normal/menu_viewer.c (should_return): New variable. (menu_viewer_changed): Likewise. (grub_menu_viewer_show_menu): Handle menu viewer changes. (grub_menu_viewer_should_return): New function. (menuviewer_write_hook): Likewise. (grub_menu_viewer_init): Likewise.
112 lines
2.6 KiB
C
112 lines
2.6 KiB
C
/*
|
|
* GRUB -- GRand Unified Bootloader
|
|
* Copyright (C) 2009 Free Software Foundation, Inc.
|
|
*
|
|
* GRUB 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 3 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* GRUB 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 GRUB. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
#include <grub/mm.h>
|
|
#include <grub/misc.h>
|
|
#include <grub/env.h>
|
|
#include <grub/menu_viewer.h>
|
|
#include <grub/menu.h>
|
|
#include <grub/auth.h>
|
|
|
|
/* The list of menu viewers. */
|
|
static grub_menu_viewer_t menu_viewer_list;
|
|
|
|
static int should_return;
|
|
static int menu_viewer_changed;
|
|
|
|
void
|
|
grub_menu_viewer_register (grub_menu_viewer_t viewer)
|
|
{
|
|
viewer->next = menu_viewer_list;
|
|
menu_viewer_list = viewer;
|
|
}
|
|
|
|
static grub_menu_viewer_t get_current_menu_viewer (void)
|
|
{
|
|
const char *selected_name = grub_env_get ("menuviewer");
|
|
|
|
/* If none selected, pick the last registered one. */
|
|
if (selected_name == 0)
|
|
return menu_viewer_list;
|
|
|
|
grub_menu_viewer_t cur;
|
|
for (cur = menu_viewer_list; cur; cur = cur->next)
|
|
{
|
|
if (grub_strcmp (cur->name, selected_name) == 0)
|
|
return cur;
|
|
}
|
|
|
|
/* Fall back to the first entry (or null). */
|
|
return menu_viewer_list;
|
|
}
|
|
|
|
grub_err_t
|
|
grub_menu_viewer_show_menu (grub_menu_t menu, int nested)
|
|
{
|
|
grub_err_t err1, err2;
|
|
|
|
while (1)
|
|
{
|
|
grub_menu_viewer_t cur = get_current_menu_viewer ();
|
|
if (!cur)
|
|
return grub_error (GRUB_ERR_BAD_ARGUMENT, "No menu viewer available.");
|
|
|
|
menu_viewer_changed = 0;
|
|
should_return = 0;
|
|
|
|
err1 = cur->show_menu (menu, nested);
|
|
grub_print_error ();
|
|
|
|
if (menu_viewer_changed)
|
|
continue;
|
|
|
|
err2 = grub_auth_check_authentication (NULL);
|
|
if (err2)
|
|
{
|
|
grub_print_error ();
|
|
grub_errno = GRUB_ERR_NONE;
|
|
continue;
|
|
}
|
|
|
|
break;
|
|
}
|
|
|
|
return err1;
|
|
}
|
|
|
|
int
|
|
grub_menu_viewer_should_return (void)
|
|
{
|
|
return should_return;
|
|
}
|
|
|
|
static char *
|
|
menuviewer_write_hook (struct grub_env_var *var __attribute__ ((unused)),
|
|
const char *val)
|
|
{
|
|
menu_viewer_changed = 1;
|
|
should_return = 1;
|
|
return grub_strdup (val);
|
|
}
|
|
|
|
void
|
|
grub_menu_viewer_init (void)
|
|
{
|
|
grub_register_variable_hook ("menuviewer", 0, menuviewer_write_hook);
|
|
}
|
|
|