306 lines
7.6 KiB
C
306 lines
7.6 KiB
C
/*
|
|
* GRUB -- GRand Unified Bootloader
|
|
* Copyright (C) 2006,2007,2008,2013 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/term.h>
|
|
#include <grub/misc.h>
|
|
#include <grub/types.h>
|
|
#include <grub/err.h>
|
|
|
|
#include <grub/emu/console.h>
|
|
|
|
#include <windows.h>
|
|
|
|
static HANDLE hStdin, hStdout;
|
|
static DWORD orig_mode;
|
|
static int saved_orig;
|
|
|
|
|
|
static void
|
|
grub_console_putchar (struct grub_term_output *term __attribute__ ((unused)),
|
|
const struct grub_unicode_glyph *c)
|
|
{
|
|
TCHAR str[2 + c->ncomb];
|
|
unsigned i, j;
|
|
DWORD written;
|
|
|
|
/* For now, do not try to use a surrogate pair. */
|
|
if (c->base > 0xffff)
|
|
str[0] = '?';
|
|
else
|
|
str[0] = (c->base & 0xffff);
|
|
j = 1;
|
|
for (i = 0; i < c->ncomb; i++)
|
|
if (c->base < 0xffff)
|
|
str[j++] = grub_unicode_get_comb (c)[i].code;
|
|
str[j] = 0;
|
|
|
|
WriteConsole (hStdout, str, j, &written, NULL);
|
|
}
|
|
|
|
const unsigned windows_codes[] =
|
|
{
|
|
/* 0x21 */ [VK_PRIOR] = GRUB_TERM_KEY_PPAGE,
|
|
/* 0x22 */ [VK_NEXT] = GRUB_TERM_KEY_NPAGE,
|
|
/* 0x23 */ [VK_END] = GRUB_TERM_KEY_END,
|
|
/* 0x24 */ [VK_HOME] = GRUB_TERM_KEY_HOME,
|
|
/* 0x25 */ [VK_LEFT] = GRUB_TERM_KEY_LEFT,
|
|
/* 0x26 */ [VK_UP] = GRUB_TERM_KEY_UP,
|
|
/* 0x27 */ [VK_RIGHT] = GRUB_TERM_KEY_RIGHT,
|
|
/* 0x28 */ [VK_DOWN] = GRUB_TERM_KEY_DOWN,
|
|
/* 0x2e */ [VK_DELETE] = GRUB_TERM_KEY_DC,
|
|
/* 0x70 */ [VK_F1] = GRUB_TERM_KEY_F1,
|
|
/* 0x71 */ [VK_F2] = GRUB_TERM_KEY_F2,
|
|
/* 0x72 */ [VK_F3] = GRUB_TERM_KEY_F3,
|
|
/* 0x73 */ [VK_F4] = GRUB_TERM_KEY_F4,
|
|
/* 0x74 */ [VK_F5] = GRUB_TERM_KEY_F5,
|
|
/* 0x75 */ [VK_F6] = GRUB_TERM_KEY_F6,
|
|
/* 0x76 */ [VK_F7] = GRUB_TERM_KEY_F7,
|
|
/* 0x77 */ [VK_F8] = GRUB_TERM_KEY_F8,
|
|
/* 0x78 */ [VK_F9] = GRUB_TERM_KEY_F9,
|
|
/* 0x79 */ [VK_F10] = GRUB_TERM_KEY_F10,
|
|
/* 0x7a */ [VK_F11] = GRUB_TERM_KEY_F11,
|
|
/* 0x7b */ [VK_F12] = GRUB_TERM_KEY_F12,
|
|
};
|
|
|
|
|
|
static int
|
|
grub_console_getkey (struct grub_term_input *term __attribute__ ((unused)))
|
|
{
|
|
while (1)
|
|
{
|
|
DWORD nev;
|
|
INPUT_RECORD ir;
|
|
int ret;
|
|
|
|
if (!GetNumberOfConsoleInputEvents (hStdin, &nev))
|
|
return GRUB_TERM_NO_KEY;
|
|
|
|
if (nev == 0)
|
|
return GRUB_TERM_NO_KEY;
|
|
|
|
if (!ReadConsoleInput (hStdin, &ir, 1,
|
|
&nev))
|
|
return GRUB_TERM_NO_KEY;
|
|
|
|
if (ir.EventType != KEY_EVENT)
|
|
continue;
|
|
|
|
if (!ir.Event.KeyEvent.bKeyDown)
|
|
continue;
|
|
ret = ir.Event.KeyEvent.uChar.UnicodeChar;
|
|
if (ret == 0)
|
|
{
|
|
if (ir.Event.KeyEvent.wVirtualKeyCode >= 0
|
|
&& ir.Event.KeyEvent.wVirtualKeyCode
|
|
< ARRAY_SIZE (windows_codes)
|
|
&& windows_codes[(int) ir.Event.KeyEvent.wVirtualKeyCode])
|
|
ret = windows_codes[(int) ir.Event.KeyEvent.wVirtualKeyCode];
|
|
else
|
|
continue;
|
|
if (ir.Event.KeyEvent.dwControlKeyState & SHIFT_PRESSED)
|
|
ret |= GRUB_TERM_SHIFT;
|
|
}
|
|
/* Workaround for AltGr bug. */
|
|
if (ir.Event.KeyEvent.dwControlKeyState & RIGHT_ALT_PRESSED)
|
|
return ret;
|
|
if (ir.Event.KeyEvent.dwControlKeyState & (LEFT_ALT_PRESSED | RIGHT_ALT_PRESSED))
|
|
ret |= GRUB_TERM_ALT;
|
|
if (ir.Event.KeyEvent.dwControlKeyState & (LEFT_CTRL_PRESSED | RIGHT_CTRL_PRESSED))
|
|
ret |= GRUB_TERM_CTRL;
|
|
return ret;
|
|
}
|
|
}
|
|
|
|
static grub_uint16_t
|
|
grub_console_getwh (struct grub_term_output *term __attribute__ ((unused)))
|
|
{
|
|
CONSOLE_SCREEN_BUFFER_INFO csbi;
|
|
|
|
csbi.dwSize.X = 80;
|
|
csbi.dwSize.Y = 25;
|
|
|
|
GetConsoleScreenBufferInfo (hStdout, &csbi);
|
|
|
|
return ((csbi.dwSize.X << 8) | csbi.dwSize.Y);
|
|
}
|
|
|
|
static grub_uint16_t
|
|
grub_console_getxy (struct grub_term_output *term __attribute__ ((unused)))
|
|
{
|
|
CONSOLE_SCREEN_BUFFER_INFO csbi;
|
|
|
|
GetConsoleScreenBufferInfo (hStdout, &csbi);
|
|
|
|
return ((csbi.dwCursorPosition.X << 8) | csbi.dwCursorPosition.Y);
|
|
}
|
|
|
|
static void
|
|
grub_console_gotoxy (struct grub_term_output *term __attribute__ ((unused)),
|
|
grub_uint8_t x, grub_uint8_t y)
|
|
{
|
|
COORD coord = { x, y };
|
|
|
|
SetConsoleCursorPosition (hStdout, coord);
|
|
}
|
|
|
|
static void
|
|
grub_console_cls (struct grub_term_output *term)
|
|
{
|
|
int tsz;
|
|
CONSOLE_SCREEN_BUFFER_INFO csbi;
|
|
|
|
struct grub_unicode_glyph c =
|
|
{
|
|
.base = ' ',
|
|
.variant = 0,
|
|
.attributes = 0,
|
|
.ncomb = 0,
|
|
.estimated_width = 1
|
|
};
|
|
|
|
GetConsoleScreenBufferInfo (hStdout, &csbi);
|
|
|
|
SetConsoleTextAttribute (hStdout, 0);
|
|
grub_console_gotoxy (term, 0, 0);
|
|
tsz = csbi.dwSize.X * csbi.dwSize.Y;
|
|
|
|
while (tsz--)
|
|
grub_console_putchar (term, &c);
|
|
|
|
grub_console_gotoxy (term, 0, 0);
|
|
SetConsoleTextAttribute (hStdout, csbi.wAttributes);
|
|
}
|
|
|
|
static void
|
|
grub_console_setcolorstate (struct grub_term_output *term
|
|
__attribute__ ((unused)),
|
|
grub_term_color_state state)
|
|
{
|
|
|
|
|
|
switch (state) {
|
|
case GRUB_TERM_COLOR_STANDARD:
|
|
SetConsoleTextAttribute (hStdout, GRUB_TERM_DEFAULT_STANDARD_COLOR
|
|
& 0x7f);
|
|
break;
|
|
case GRUB_TERM_COLOR_NORMAL:
|
|
SetConsoleTextAttribute (hStdout, grub_term_normal_color & 0x7f);
|
|
break;
|
|
case GRUB_TERM_COLOR_HIGHLIGHT:
|
|
SetConsoleTextAttribute (hStdout, grub_term_highlight_color & 0x7f);
|
|
break;
|
|
default:
|
|
break;
|
|
}
|
|
}
|
|
|
|
static void
|
|
grub_console_setcursor (struct grub_term_output *term __attribute__ ((unused)),
|
|
int on)
|
|
{
|
|
CONSOLE_CURSOR_INFO ci;
|
|
ci.dwSize = 5;
|
|
ci.bVisible = on;
|
|
SetConsoleCursorInfo (hStdout, &ci);
|
|
}
|
|
|
|
static grub_err_t
|
|
grub_efi_console_init (struct grub_term_output *term)
|
|
{
|
|
grub_console_setcursor (term, 1);
|
|
return 0;
|
|
}
|
|
|
|
static grub_err_t
|
|
grub_efi_console_fini (struct grub_term_output *term)
|
|
{
|
|
grub_console_setcursor (term, 1);
|
|
return 0;
|
|
}
|
|
|
|
|
|
static grub_err_t
|
|
grub_console_init_input (struct grub_term_input *term)
|
|
{
|
|
if (!saved_orig)
|
|
{
|
|
GetConsoleMode (hStdin, &orig_mode);
|
|
}
|
|
|
|
saved_orig = 1;
|
|
|
|
SetConsoleMode (hStdin, orig_mode & ~ENABLE_ECHO_INPUT
|
|
& ~ENABLE_LINE_INPUT & ~ENABLE_PROCESSED_INPUT);
|
|
|
|
return GRUB_ERR_NONE;
|
|
}
|
|
|
|
static grub_err_t
|
|
grub_console_fini_input (struct grub_term_input *term
|
|
__attribute__ ((unused)))
|
|
{
|
|
SetConsoleMode (hStdin, orig_mode);
|
|
saved_orig = 0;
|
|
return GRUB_ERR_NONE;
|
|
}
|
|
|
|
|
|
static struct grub_term_input grub_console_term_input =
|
|
{
|
|
.name = "console",
|
|
.getkey = grub_console_getkey,
|
|
.init = grub_console_init_input,
|
|
.fini = grub_console_fini_input,
|
|
};
|
|
|
|
static struct grub_term_output grub_console_term_output =
|
|
{
|
|
.name = "console",
|
|
.init = grub_efi_console_init,
|
|
.fini = grub_efi_console_fini,
|
|
.putchar = grub_console_putchar,
|
|
.getwh = grub_console_getwh,
|
|
.getxy = grub_console_getxy,
|
|
.gotoxy = grub_console_gotoxy,
|
|
.cls = grub_console_cls,
|
|
.setcolorstate = grub_console_setcolorstate,
|
|
.setcursor = grub_console_setcursor,
|
|
.flags = GRUB_TERM_CODE_TYPE_VISUAL_GLYPHS
|
|
};
|
|
|
|
void
|
|
grub_console_init (void)
|
|
{
|
|
hStdin = GetStdHandle (STD_INPUT_HANDLE);
|
|
hStdout = GetStdHandle (STD_OUTPUT_HANDLE);
|
|
|
|
grub_term_register_input ("console", &grub_console_term_input);
|
|
grub_term_register_output ("console", &grub_console_term_output);
|
|
}
|
|
|
|
void
|
|
grub_console_fini (void)
|
|
{
|
|
if (saved_orig)
|
|
{
|
|
SetConsoleMode (hStdin, orig_mode);
|
|
saved_orig = 0;
|
|
}
|
|
grub_term_unregister_input (&grub_console_term_input);
|
|
grub_term_unregister_output (&grub_console_term_output);
|
|
}
|