361 lines
		
	
	
		
			7.6 KiB
		
	
	
	
		
			C
		
	
	
	
			
		
		
	
	
			361 lines
		
	
	
		
			7.6 KiB
		
	
	
	
		
			C
		
	
	
	
| /*
 | |
|  *  EFI application console interface
 | |
|  *
 | |
|  *  Copyright (c) 2016 Alexander Graf
 | |
|  *
 | |
|  *  SPDX-License-Identifier:     GPL-2.0+
 | |
|  */
 | |
| 
 | |
| #include <common.h>
 | |
| #include <efi_loader.h>
 | |
| 
 | |
| /* If we can't determine the console size, default to 80x24 */
 | |
| static int console_columns = 80;
 | |
| static int console_rows = 24;
 | |
| static bool console_size_queried;
 | |
| 
 | |
| const efi_guid_t efi_guid_console_control = CONSOLE_CONTROL_GUID;
 | |
| 
 | |
| #define cESC '\x1b'
 | |
| #define ESC "\x1b"
 | |
| 
 | |
| static efi_status_t EFIAPI efi_cin_get_mode(
 | |
| 			struct efi_console_control_protocol *this,
 | |
| 			int *mode, char *uga_exists, char *std_in_locked)
 | |
| {
 | |
| 	EFI_ENTRY("%p, %p, %p, %p", this, mode, uga_exists, std_in_locked);
 | |
| 
 | |
| 	if (mode)
 | |
| 		*mode = EFI_CONSOLE_MODE_TEXT;
 | |
| 	if (uga_exists)
 | |
| 		*uga_exists = 0;
 | |
| 	if (std_in_locked)
 | |
| 		*std_in_locked = 0;
 | |
| 
 | |
| 	return EFI_EXIT(EFI_SUCCESS);
 | |
| }
 | |
| 
 | |
| static efi_status_t EFIAPI efi_cin_set_mode(
 | |
| 			struct efi_console_control_protocol *this, int mode)
 | |
| {
 | |
| 	EFI_ENTRY("%p, %d", this, mode);
 | |
| 	return EFI_EXIT(EFI_UNSUPPORTED);
 | |
| }
 | |
| 
 | |
| static efi_status_t EFIAPI efi_cin_lock_std_in(
 | |
| 			struct efi_console_control_protocol *this,
 | |
| 			uint16_t *password)
 | |
| {
 | |
| 	EFI_ENTRY("%p, %p", this, password);
 | |
| 	return EFI_EXIT(EFI_UNSUPPORTED);
 | |
| }
 | |
| 
 | |
| const struct efi_console_control_protocol efi_console_control = {
 | |
| 	.get_mode = efi_cin_get_mode,
 | |
| 	.set_mode = efi_cin_set_mode,
 | |
| 	.lock_std_in = efi_cin_lock_std_in,
 | |
| };
 | |
| 
 | |
| static struct simple_text_output_mode efi_con_mode = {
 | |
| 	.max_mode = 0,
 | |
| 	.mode = 0,
 | |
| 	.attribute = 0,
 | |
| 	.cursor_column = 0,
 | |
| 	.cursor_row = 0,
 | |
| 	.cursor_visible = 1,
 | |
| };
 | |
| 
 | |
| static int term_read_reply(int *n, int maxnum, char end_char)
 | |
| {
 | |
| 	char c;
 | |
| 	int i = 0;
 | |
| 
 | |
| 	c = getc();
 | |
| 	if (c != cESC)
 | |
| 		return -1;
 | |
| 	c = getc();
 | |
| 	if (c != '[')
 | |
| 		return -1;
 | |
| 
 | |
| 	n[0] = 0;
 | |
| 	while (1) {
 | |
| 		c = getc();
 | |
| 		if (c == ';') {
 | |
| 			i++;
 | |
| 			if (i >= maxnum)
 | |
| 				return -1;
 | |
| 			n[i] = 0;
 | |
| 			continue;
 | |
| 		} else if (c == end_char) {
 | |
| 			break;
 | |
| 		} else if (c > '9' || c < '0') {
 | |
| 			return -1;
 | |
| 		}
 | |
| 
 | |
| 		/* Read one more decimal position */
 | |
| 		n[i] *= 10;
 | |
| 		n[i] += c - '0';
 | |
| 	}
 | |
| 
 | |
| 	return 0;
 | |
| }
 | |
| 
 | |
| static efi_status_t EFIAPI efi_cout_reset(
 | |
| 			struct efi_simple_text_output_protocol *this,
 | |
| 			char extended_verification)
 | |
| {
 | |
| 	EFI_ENTRY("%p, %d", this, extended_verification);
 | |
| 	return EFI_EXIT(EFI_UNSUPPORTED);
 | |
| }
 | |
| 
 | |
| static void print_unicode_in_utf8(u16 c)
 | |
| {
 | |
| 	char utf8[4] = { 0 };
 | |
| 	char *b = utf8;
 | |
| 
 | |
| 	if (c < 0x80) {
 | |
| 		*(b++) = c;
 | |
| 	} else if (c < 0x800) {
 | |
| 		*(b++) = 192 + c / 64;
 | |
| 		*(b++) = 128 + c % 64;
 | |
| 	} else {
 | |
| 		*(b++) = 224 + c / 4096;
 | |
| 		*(b++) = 128 + c / 64 % 64;
 | |
| 		*(b++) = 128 + c % 64;
 | |
| 	}
 | |
| 
 | |
| 	puts(utf8);
 | |
| }
 | |
| 
 | |
| static efi_status_t EFIAPI efi_cout_output_string(
 | |
| 			struct efi_simple_text_output_protocol *this,
 | |
| 			const unsigned short *string)
 | |
| {
 | |
| 	u16 ch;
 | |
| 
 | |
| 	EFI_ENTRY("%p, %p", this, string);
 | |
| 	for (;(ch = *string); string++) {
 | |
| 		print_unicode_in_utf8(ch);
 | |
| 		efi_con_mode.cursor_column++;
 | |
| 		if (ch == '\n') {
 | |
| 			efi_con_mode.cursor_column = 1;
 | |
| 			efi_con_mode.cursor_row++;
 | |
| 		} else if (efi_con_mode.cursor_column > console_columns) {
 | |
| 			efi_con_mode.cursor_column = 1;
 | |
| 			efi_con_mode.cursor_row++;
 | |
| 		}
 | |
| 		if (efi_con_mode.cursor_row > console_rows) {
 | |
| 			efi_con_mode.cursor_row = console_rows;
 | |
| 		}
 | |
| 	}
 | |
| 
 | |
| 	return EFI_EXIT(EFI_SUCCESS);
 | |
| }
 | |
| 
 | |
| static efi_status_t EFIAPI efi_cout_test_string(
 | |
| 			struct efi_simple_text_output_protocol *this,
 | |
| 			const unsigned short *string)
 | |
| {
 | |
| 	EFI_ENTRY("%p, %p", this, string);
 | |
| 	return EFI_EXIT(EFI_SUCCESS);
 | |
| }
 | |
| 
 | |
| static efi_status_t EFIAPI efi_cout_query_mode(
 | |
| 			struct efi_simple_text_output_protocol *this,
 | |
| 			unsigned long mode_number, unsigned long *columns,
 | |
| 			unsigned long *rows)
 | |
| {
 | |
| 	EFI_ENTRY("%p, %ld, %p, %p", this, mode_number, columns, rows);
 | |
| 
 | |
| 	if (!console_size_queried) {
 | |
| 		/* Ask the terminal about its size */
 | |
| 		int n[3];
 | |
| 		u64 timeout;
 | |
| 
 | |
| 		console_size_queried = true;
 | |
| 
 | |
| 		/* Empty input buffer */
 | |
| 		while (tstc())
 | |
| 			getc();
 | |
| 
 | |
| 		printf(ESC"[18t");
 | |
| 
 | |
| 		/* Check if we have a terminal that understands */
 | |
| 		timeout = timer_get_us() + 1000000;
 | |
| 		while (!tstc())
 | |
| 			if (timer_get_us() > timeout)
 | |
| 				goto out;
 | |
| 
 | |
| 		/* Read {depth,rows,cols} */
 | |
| 		if (term_read_reply(n, 3, 't')) {
 | |
| 			goto out;
 | |
| 		}
 | |
| 
 | |
| 		console_columns = n[2];
 | |
| 		console_rows = n[1];
 | |
| 	}
 | |
| 
 | |
| out:
 | |
| 	if (columns)
 | |
| 		*columns = console_columns;
 | |
| 	if (rows)
 | |
| 		*rows = console_rows;
 | |
| 
 | |
| 	return EFI_EXIT(EFI_SUCCESS);
 | |
| }
 | |
| 
 | |
| static efi_status_t EFIAPI efi_cout_set_mode(
 | |
| 			struct efi_simple_text_output_protocol *this,
 | |
| 			unsigned long mode_number)
 | |
| {
 | |
| 	EFI_ENTRY("%p, %ld", this, mode_number);
 | |
| 
 | |
| 	/* We only support text output for now */
 | |
| 	if (mode_number == EFI_CONSOLE_MODE_TEXT)
 | |
| 		return EFI_EXIT(EFI_SUCCESS);
 | |
| 
 | |
| 	return EFI_EXIT(EFI_UNSUPPORTED);
 | |
| }
 | |
| 
 | |
| static efi_status_t EFIAPI efi_cout_set_attribute(
 | |
| 			struct efi_simple_text_output_protocol *this,
 | |
| 			unsigned long attribute)
 | |
| {
 | |
| 	EFI_ENTRY("%p, %lx", this, attribute);
 | |
| 
 | |
| 	/* Just ignore attributes (colors) for now */
 | |
| 	return EFI_EXIT(EFI_UNSUPPORTED);
 | |
| }
 | |
| 
 | |
| static efi_status_t EFIAPI efi_cout_clear_screen(
 | |
| 			struct efi_simple_text_output_protocol *this)
 | |
| {
 | |
| 	EFI_ENTRY("%p", this);
 | |
| 
 | |
| 	printf(ESC"[2J");
 | |
| 
 | |
| 	return EFI_EXIT(EFI_SUCCESS);
 | |
| }
 | |
| 
 | |
| static efi_status_t EFIAPI efi_cout_set_cursor_position(
 | |
| 			struct efi_simple_text_output_protocol *this,
 | |
| 			unsigned long column, unsigned long row)
 | |
| {
 | |
| 	EFI_ENTRY("%p, %ld, %ld", this, column, row);
 | |
| 
 | |
| 	printf(ESC"[%d;%df", (int)row, (int)column);
 | |
| 	efi_con_mode.cursor_column = column;
 | |
| 	efi_con_mode.cursor_row = row;
 | |
| 
 | |
| 	return EFI_EXIT(EFI_SUCCESS);
 | |
| }
 | |
| 
 | |
| static efi_status_t EFIAPI efi_cout_enable_cursor(
 | |
| 			struct efi_simple_text_output_protocol *this,
 | |
| 			bool enable)
 | |
| {
 | |
| 	EFI_ENTRY("%p, %d", this, enable);
 | |
| 
 | |
| 	printf(ESC"[?25%c", enable ? 'h' : 'l');
 | |
| 
 | |
| 	return EFI_EXIT(EFI_SUCCESS);
 | |
| }
 | |
| 
 | |
| const struct efi_simple_text_output_protocol efi_con_out = {
 | |
| 	.reset = efi_cout_reset,
 | |
| 	.output_string = efi_cout_output_string,
 | |
| 	.test_string = efi_cout_test_string,
 | |
| 	.query_mode = efi_cout_query_mode,
 | |
| 	.set_mode = efi_cout_set_mode,
 | |
| 	.set_attribute = efi_cout_set_attribute,
 | |
| 	.clear_screen = efi_cout_clear_screen,
 | |
| 	.set_cursor_position = efi_cout_set_cursor_position,
 | |
| 	.enable_cursor = efi_cout_enable_cursor,
 | |
| 	.mode = (void*)&efi_con_mode,
 | |
| };
 | |
| 
 | |
| static efi_status_t EFIAPI efi_cin_reset(
 | |
| 			struct efi_simple_input_interface *this,
 | |
| 			bool extended_verification)
 | |
| {
 | |
| 	EFI_ENTRY("%p, %d", this, extended_verification);
 | |
| 	return EFI_EXIT(EFI_UNSUPPORTED);
 | |
| }
 | |
| 
 | |
| static efi_status_t EFIAPI efi_cin_read_key_stroke(
 | |
| 			struct efi_simple_input_interface *this,
 | |
| 			struct efi_input_key *key)
 | |
| {
 | |
| 	struct efi_input_key pressed_key = {
 | |
| 		.scan_code = 0,
 | |
| 		.unicode_char = 0,
 | |
| 	};
 | |
| 	char ch;
 | |
| 
 | |
| 	EFI_ENTRY("%p, %p", this, key);
 | |
| 
 | |
| 	/* We don't do interrupts, so check for timers cooperatively */
 | |
| 	efi_timer_check();
 | |
| 
 | |
| 	if (!tstc()) {
 | |
| 		/* No key pressed */
 | |
| 		return EFI_EXIT(EFI_NOT_READY);
 | |
| 	}
 | |
| 
 | |
| 	ch = getc();
 | |
| 	if (ch == cESC) {
 | |
| 		/* Escape Sequence */
 | |
| 		ch = getc();
 | |
| 		switch (ch) {
 | |
| 		case cESC: /* ESC */
 | |
| 			pressed_key.scan_code = 23;
 | |
| 			break;
 | |
| 		case 'O': /* F1 - F4 */
 | |
| 			pressed_key.scan_code = getc() - 'P' + 11;
 | |
| 			break;
 | |
| 		case 'a'...'z':
 | |
| 			ch = ch - 'a';
 | |
| 			break;
 | |
| 		case '[':
 | |
| 			ch = getc();
 | |
| 			switch (ch) {
 | |
| 			case 'A'...'D': /* up, down right, left */
 | |
| 				pressed_key.scan_code = ch - 'A' + 1;
 | |
| 				break;
 | |
| 			case 'F': /* End */
 | |
| 				pressed_key.scan_code = 6;
 | |
| 				break;
 | |
| 			case 'H': /* Home */
 | |
| 				pressed_key.scan_code = 5;
 | |
| 				break;
 | |
| 			case '1': /* F5 - F8 */
 | |
| 				pressed_key.scan_code = getc() - '0' + 11;
 | |
| 				getc();
 | |
| 				break;
 | |
| 			case '2': /* F9 - F12 */
 | |
| 				pressed_key.scan_code = getc() - '0' + 19;
 | |
| 				getc();
 | |
| 				break;
 | |
| 			case '3': /* DEL */
 | |
| 				pressed_key.scan_code = 8;
 | |
| 				getc();
 | |
| 				break;
 | |
| 			}
 | |
| 			break;
 | |
| 		}
 | |
| 	} else if (ch == 0x7f) {
 | |
| 		/* Backspace */
 | |
| 		ch = 0x08;
 | |
| 	}
 | |
| 	pressed_key.unicode_char = ch;
 | |
| 	*key = pressed_key;
 | |
| 
 | |
| 	return EFI_EXIT(EFI_SUCCESS);
 | |
| }
 | |
| 
 | |
| const struct efi_simple_input_interface efi_con_in = {
 | |
| 	.reset = efi_cin_reset,
 | |
| 	.read_key_stroke = efi_cin_read_key_stroke,
 | |
| 	.wait_for_key = NULL,
 | |
| };
 |