vaseboot/VasEBoot-core/kern/rescue_reader.c

112 lines
2.7 KiB
C

/* rescue_reader.c - rescue mode reader */
/*
* VAS_EBOOT -- GRand Unified Bootloader
* Copyright (C) 2009,2010 Free Software Foundation, Inc.
*
* VAS_EBOOT 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.
*
* VAS_EBOOT 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 VAS_EBOOT. If not, see <http://www.gnu.org/licenses/>.
*/
#include <VasEBoot/types.h>
#include <VasEBoot/reader.h>
#include <VasEBoot/parser.h>
#include <VasEBoot/misc.h>
#include <VasEBoot/term.h>
#include <VasEBoot/mm.h>
#define VAS_EBOOT_RESCUE_BUF_SIZE 256
static char linebuf[VAS_EBOOT_RESCUE_BUF_SIZE];
/* Prompt to input a command and read the line. */
static VasEBoot_err_t
VasEBoot_rescue_read_line (char **line, int cont,
void *data __attribute__ ((unused)))
{
int c;
int pos = 0;
char str[4];
VasEBoot_printf ((cont) ? "> " : "VasEBoot rescue> ");
VasEBoot_memset (linebuf, 0, VAS_EBOOT_RESCUE_BUF_SIZE);
while ((c = VasEBoot_getkey ()) != '\n' && c != '\r')
{
if (VasEBoot_isprint (c))
{
if (pos < VAS_EBOOT_RESCUE_BUF_SIZE - 1)
{
str[0] = c;
str[1] = 0;
linebuf[pos++] = c;
VasEBoot_xputs (str);
}
}
else if (c == '\b')
{
if (pos > 0)
{
str[0] = c;
str[1] = ' ';
str[2] = c;
str[3] = 0;
linebuf[--pos] = 0;
VasEBoot_xputs (str);
}
}
VasEBoot_refresh ();
}
VasEBoot_xputs ("\n");
VasEBoot_refresh ();
*line = VasEBoot_strdup (linebuf);
return 0;
}
void __attribute__ ((noreturn))
VasEBoot_rescue_run (void)
{
/* Stall if the CLI has been disabled */
if (VasEBoot_is_cli_disabled () || VasEBoot_is_cli_need_auth ())
{
VasEBoot_printf ("Rescue mode has been disabled...\n");
do
{
/* Do not optimize out the loop. */
asm volatile ("");
}
while (1);
}
VasEBoot_printf ("Entering rescue mode...\n");
while (1)
{
char *line;
/* Print an error, if any. */
VasEBoot_print_error ();
VasEBoot_errno = VAS_EBOOT_ERR_NONE;
VasEBoot_rescue_read_line (&line, 0, NULL);
if (! line || line[0] == '\0')
continue;
VasEBoot_rescue_parse_line (line, VasEBoot_rescue_read_line, NULL);
VasEBoot_free (line);
}
}