dcavalca / rpms / grub2

Forked from rpms/grub2 3 years ago
Clone

Blame SOURCES/0129-Make-CTRL-and-ALT-keys-work-as-expected-on-EFI-syste.patch

4fe85b
From 0000000000000000000000000000000000000000 Mon Sep 17 00:00:00 2001
4fe85b
From: Peter Jones <pjones@redhat.com>
4fe85b
Date: Mon, 3 Feb 2014 15:21:46 -0500
4fe85b
Subject: [PATCH] Make CTRL and ALT keys work as expected on EFI systems
4fe85b
 (version 5).
4fe85b
4fe85b
This is version 4.
4fe85b
4fe85b
Changes from version 1:
4fe85b
- handles SHIFT as a modifier
4fe85b
- handles F11 and F12 keys
4fe85b
- uses the handle provided by the system table to find our _EX protocol.
4fe85b
4fe85b
Changes from version 2:
4fe85b
- eliminate duplicate keycode translation.
4fe85b
4fe85b
Changes from version 3:
4fe85b
- Do not add the shift modifier for any ascii character between space
4fe85b
  (0x20) and DEL (0x7f); the combination of the modifier and many of the
4fe85b
  keys causes it not to be recognized at all.  Specifically, if we
4fe85b
  include the modifier on any querty punctuation character, i.e.
4fe85b
  anything the string "~!@#$%^&*()_+{}|:\"<>?" represents in C, it stops
4fe85b
  being recognized whatsoever.
4fe85b
4fe85b
Changes from version 4:
4fe85b
- Always initialize term->data from locate protocol (i.e. make it
4fe85b
  unconditional.)
4fe85b
4fe85b
Signed-off-by: Peter Jones <pjones@redhat.com>
4fe85b
---
4fe85b
 grub-core/term/efi/console.c | 118 +++++++++++++++++++++++++++++++++++--------
4fe85b
 include/grub/efi/api.h       |  65 +++++++++++++++++++++++-
4fe85b
 2 files changed, 161 insertions(+), 22 deletions(-)
4fe85b
4fe85b
diff --git a/grub-core/term/efi/console.c b/grub-core/term/efi/console.c
4fe85b
index a37eb841c72..677eab5820f 100644
4fe85b
--- a/grub-core/term/efi/console.c
4fe85b
+++ b/grub-core/term/efi/console.c
4fe85b
@@ -104,26 +104,12 @@ const unsigned efi_codes[] =
4fe85b
     GRUB_TERM_KEY_DC, GRUB_TERM_KEY_PPAGE, GRUB_TERM_KEY_NPAGE, GRUB_TERM_KEY_F1,
4fe85b
     GRUB_TERM_KEY_F2, GRUB_TERM_KEY_F3, GRUB_TERM_KEY_F4, GRUB_TERM_KEY_F5,
4fe85b
     GRUB_TERM_KEY_F6, GRUB_TERM_KEY_F7, GRUB_TERM_KEY_F8, GRUB_TERM_KEY_F9,
4fe85b
-    GRUB_TERM_KEY_F10, 0, 0, '\e'
4fe85b
+    GRUB_TERM_KEY_F10, GRUB_TERM_KEY_F10, GRUB_TERM_KEY_F11, '\e'
4fe85b
   };
4fe85b
 
4fe85b
-
4fe85b
 static int
4fe85b
-grub_console_getkey (struct grub_term_input *term __attribute__ ((unused)))
4fe85b
+grub_efi_translate_key (grub_efi_input_key_t key)
4fe85b
 {
4fe85b
-  grub_efi_simple_input_interface_t *i;
4fe85b
-  grub_efi_input_key_t key;
4fe85b
-  grub_efi_status_t status;
4fe85b
-
4fe85b
-  if (grub_efi_is_finished)
4fe85b
-    return 0;
4fe85b
-
4fe85b
-  i = grub_efi_system_table->con_in;
4fe85b
-  status = efi_call_2 (i->read_key_stroke, i, &key);
4fe85b
-
4fe85b
-  if (status != GRUB_EFI_SUCCESS)
4fe85b
-    return GRUB_TERM_NO_KEY;
4fe85b
-
4fe85b
   if (key.scan_code == 0)
4fe85b
     {
4fe85b
       /* Some firmware implementations use VT100-style codes against the spec.
4fe85b
@@ -139,9 +125,98 @@ grub_console_getkey (struct grub_term_input *term __attribute__ ((unused)))
4fe85b
   else if (key.scan_code < ARRAY_SIZE (efi_codes))
4fe85b
     return efi_codes[key.scan_code];
4fe85b
 
4fe85b
+  if (key.unicode_char >= 0x20 && key.unicode_char <= 0x7f)
4fe85b
+    return key.unicode_char;
4fe85b
+
4fe85b
   return GRUB_TERM_NO_KEY;
4fe85b
 }
4fe85b
 
4fe85b
+static int
4fe85b
+grub_console_getkey_con (struct grub_term_input *term __attribute__ ((unused)))
4fe85b
+{
4fe85b
+  grub_efi_simple_input_interface_t *i;
4fe85b
+  grub_efi_input_key_t key;
4fe85b
+  grub_efi_status_t status;
4fe85b
+
4fe85b
+  i = grub_efi_system_table->con_in;
4fe85b
+  status = efi_call_2 (i->read_key_stroke, i, &key);
4fe85b
+
4fe85b
+  if (status != GRUB_EFI_SUCCESS)
4fe85b
+    return GRUB_TERM_NO_KEY;
4fe85b
+
4fe85b
+  return grub_efi_translate_key(key);
4fe85b
+}
4fe85b
+
4fe85b
+static int
4fe85b
+grub_console_getkey_ex(struct grub_term_input *term)
4fe85b
+{
4fe85b
+  grub_efi_key_data_t key_data;
4fe85b
+  grub_efi_status_t status;
4fe85b
+  grub_efi_uint32_t kss;
4fe85b
+  int key = -1;
4fe85b
+
4fe85b
+  grub_efi_simple_text_input_ex_interface_t *text_input = term->data;
4fe85b
+
4fe85b
+  status = efi_call_2 (text_input->read_key_stroke, text_input, &key_data);
4fe85b
+
4fe85b
+  if (status != GRUB_EFI_SUCCESS)
4fe85b
+    return GRUB_TERM_NO_KEY;
4fe85b
+
4fe85b
+  kss = key_data.key_state.key_shift_state;
4fe85b
+  key = grub_efi_translate_key(key_data.key);
4fe85b
+
4fe85b
+  if (key == GRUB_TERM_NO_KEY)
4fe85b
+    return GRUB_TERM_NO_KEY;
4fe85b
+
4fe85b
+  if (kss & GRUB_EFI_SHIFT_STATE_VALID)
4fe85b
+    {
4fe85b
+      if ((kss & GRUB_EFI_LEFT_SHIFT_PRESSED
4fe85b
+	   || kss & GRUB_EFI_RIGHT_SHIFT_PRESSED)
4fe85b
+	  && !(key >= 0x20 && key <= 0x7f))
4fe85b
+	key |= GRUB_TERM_SHIFT;
4fe85b
+      if (kss & GRUB_EFI_LEFT_ALT_PRESSED || kss & GRUB_EFI_RIGHT_ALT_PRESSED)
4fe85b
+	key |= GRUB_TERM_ALT;
4fe85b
+      if (kss & GRUB_EFI_LEFT_CONTROL_PRESSED
4fe85b
+	  || kss & GRUB_EFI_RIGHT_CONTROL_PRESSED)
4fe85b
+	key |= GRUB_TERM_CTRL;
4fe85b
+    }
4fe85b
+
4fe85b
+  return key;
4fe85b
+}
4fe85b
+
4fe85b
+static grub_err_t
4fe85b
+grub_efi_console_input_init (struct grub_term_input *term)
4fe85b
+{
4fe85b
+  grub_efi_guid_t text_input_ex_guid =
4fe85b
+    GRUB_EFI_SIMPLE_TEXT_INPUT_EX_PROTOCOL_GUID;
4fe85b
+
4fe85b
+  if (grub_efi_is_finished)
4fe85b
+    return 0;
4fe85b
+
4fe85b
+  grub_efi_simple_text_input_ex_interface_t *text_input = term->data;
4fe85b
+  if (text_input)
4fe85b
+    return 0;
4fe85b
+
4fe85b
+  text_input = grub_efi_open_protocol(grub_efi_system_table->console_in_handler,
4fe85b
+				      &text_input_ex_guid,
4fe85b
+				      GRUB_EFI_OPEN_PROTOCOL_GET_PROTOCOL);
4fe85b
+  term->data = (void *)text_input;
4fe85b
+
4fe85b
+  return 0;
4fe85b
+}
4fe85b
+
4fe85b
+static int
4fe85b
+grub_console_getkey (struct grub_term_input *term)
4fe85b
+{
4fe85b
+  if (grub_efi_is_finished)
4fe85b
+    return 0;
4fe85b
+
4fe85b
+  if (term->data)
4fe85b
+    return grub_console_getkey_ex(term);
4fe85b
+  else
4fe85b
+    return grub_console_getkey_con(term);
4fe85b
+}
4fe85b
+
4fe85b
 static struct grub_term_coordinate
4fe85b
 grub_console_getwh (struct grub_term_output *term __attribute__ ((unused)))
4fe85b
 {
4fe85b
@@ -243,7 +318,7 @@ grub_console_setcursor (struct grub_term_output *term __attribute__ ((unused)),
4fe85b
 }
4fe85b
 
4fe85b
 static grub_err_t
4fe85b
-grub_efi_console_init (struct grub_term_output *term)
4fe85b
+grub_efi_console_output_init (struct grub_term_output *term)
4fe85b
 {
4fe85b
   grub_efi_set_text_mode (1);
4fe85b
   grub_console_setcursor (term, 1);
4fe85b
@@ -251,7 +326,7 @@ grub_efi_console_init (struct grub_term_output *term)
4fe85b
 }
4fe85b
 
4fe85b
 static grub_err_t
4fe85b
-grub_efi_console_fini (struct grub_term_output *term)
4fe85b
+grub_efi_console_output_fini (struct grub_term_output *term)
4fe85b
 {
4fe85b
   grub_console_setcursor (term, 0);
4fe85b
   grub_efi_set_text_mode (0);
4fe85b
@@ -262,13 +337,14 @@ static struct grub_term_input grub_console_term_input =
4fe85b
   {
4fe85b
     .name = "console",
4fe85b
     .getkey = grub_console_getkey,
4fe85b
+    .init = grub_efi_console_input_init,
4fe85b
   };
4fe85b
 
4fe85b
 static struct grub_term_output grub_console_term_output =
4fe85b
   {
4fe85b
     .name = "console",
4fe85b
-    .init = grub_efi_console_init,
4fe85b
-    .fini = grub_efi_console_fini,
4fe85b
+    .init = grub_efi_console_output_init,
4fe85b
+    .fini = grub_efi_console_output_fini,
4fe85b
     .putchar = grub_console_putchar,
4fe85b
     .getwh = grub_console_getwh,
4fe85b
     .getxy = grub_console_getxy,
4fe85b
@@ -291,8 +367,8 @@ grub_console_init (void)
4fe85b
       return;
4fe85b
     }
4fe85b
 
4fe85b
-  grub_term_register_input ("console", &grub_console_term_input);
4fe85b
   grub_term_register_output ("console", &grub_console_term_output);
4fe85b
+  grub_term_register_input ("console", &grub_console_term_input);
4fe85b
 }
4fe85b
 
4fe85b
 void
4fe85b
diff --git a/include/grub/efi/api.h b/include/grub/efi/api.h
4fe85b
index e5dd543a801..142340372e1 100644
4fe85b
--- a/include/grub/efi/api.h
4fe85b
+++ b/include/grub/efi/api.h
4fe85b
@@ -111,7 +111,7 @@
4fe85b
     { 0x8e, 0x39, 0x00, 0xa0, 0xc9, 0x69, 0x72, 0x3b } \
4fe85b
   }
4fe85b
 
4fe85b
-#define EFI_SIMPLE_TEXT_INPUT_EX_PROTOCOL_GUID \
4fe85b
+#define GRUB_EFI_SIMPLE_TEXT_INPUT_EX_PROTOCOL_GUID \
4fe85b
   { 0xdd9e7534, 0x7762, 0x4698, \
4fe85b
     { 0x8c, 0x14, 0xf5, 0x85, 0x17, 0xa6, 0x25, 0xaa } \
4fe85b
   }
4fe85b
@@ -952,6 +952,32 @@ struct grub_efi_input_key
4fe85b
 };
4fe85b
 typedef struct grub_efi_input_key grub_efi_input_key_t;
4fe85b
 
4fe85b
+typedef grub_efi_uint8_t grub_efi_key_toggle_state_t;
4fe85b
+struct grub_efi_key_state
4fe85b
+{
4fe85b
+	grub_efi_uint32_t key_shift_state;
4fe85b
+	grub_efi_key_toggle_state_t key_toggle_state;
4fe85b
+};
4fe85b
+typedef struct grub_efi_key_state grub_efi_key_state_t;
4fe85b
+
4fe85b
+#define GRUB_EFI_SHIFT_STATE_VALID     0x80000000
4fe85b
+#define GRUB_EFI_RIGHT_SHIFT_PRESSED   0x00000001
4fe85b
+#define GRUB_EFI_LEFT_SHIFT_PRESSED    0x00000002
4fe85b
+#define GRUB_EFI_RIGHT_CONTROL_PRESSED 0x00000004
4fe85b
+#define GRUB_EFI_LEFT_CONTROL_PRESSED  0x00000008
4fe85b
+#define GRUB_EFI_RIGHT_ALT_PRESSED     0x00000010
4fe85b
+#define GRUB_EFI_LEFT_ALT_PRESSED      0x00000020
4fe85b
+#define GRUB_EFI_RIGHT_LOGO_PRESSED    0x00000040
4fe85b
+#define GRUB_EFI_LEFT_LOGO_PRESSED     0x00000080
4fe85b
+#define GRUB_EFI_MENU_KEY_PRESSED      0x00000100
4fe85b
+#define GRUB_EFI_SYS_REQ_PRESSED       0x00000200
4fe85b
+
4fe85b
+#define GRUB_EFI_TOGGLE_STATE_VALID 0x80
4fe85b
+#define GRUB_EFI_KEY_STATE_EXPOSED  0x40
4fe85b
+#define GRUB_EFI_SCROLL_LOCK_ACTIVE 0x01
4fe85b
+#define GRUB_EFI_NUM_LOCK_ACTIVE    0x02
4fe85b
+#define GRUB_EFI_CAPS_LOCK_ACTIVE   0x04
4fe85b
+
4fe85b
 struct grub_efi_simple_text_output_mode
4fe85b
 {
4fe85b
   grub_efi_int32_t max_mode;
4fe85b
@@ -1294,6 +1320,43 @@ struct grub_efi_simple_input_interface
4fe85b
 };
4fe85b
 typedef struct grub_efi_simple_input_interface grub_efi_simple_input_interface_t;
4fe85b
 
4fe85b
+struct grub_efi_key_data {
4fe85b
+	grub_efi_input_key_t key;
4fe85b
+	grub_efi_key_state_t key_state;
4fe85b
+};
4fe85b
+typedef struct grub_efi_key_data grub_efi_key_data_t;
4fe85b
+
4fe85b
+typedef grub_efi_status_t (*grub_efi_key_notify_function_t) (
4fe85b
+	grub_efi_key_data_t *key_data
4fe85b
+	);
4fe85b
+
4fe85b
+struct grub_efi_simple_text_input_ex_interface
4fe85b
+{
4fe85b
+	grub_efi_status_t
4fe85b
+	(*reset) (struct grub_efi_simple_text_input_ex_interface *this,
4fe85b
+		  grub_efi_boolean_t extended_verification);
4fe85b
+
4fe85b
+	grub_efi_status_t
4fe85b
+	(*read_key_stroke) (struct grub_efi_simple_text_input_ex_interface *this,
4fe85b
+			    grub_efi_key_data_t *key_data);
4fe85b
+
4fe85b
+	grub_efi_event_t wait_for_key;
4fe85b
+
4fe85b
+	grub_efi_status_t
4fe85b
+	(*set_state) (struct grub_efi_simple_text_input_ex_interface *this,
4fe85b
+		      grub_efi_key_toggle_state_t *key_toggle_state);
4fe85b
+
4fe85b
+	grub_efi_status_t
4fe85b
+	(*register_key_notify) (struct grub_efi_simple_text_input_ex_interface *this,
4fe85b
+				grub_efi_key_data_t *key_data,
4fe85b
+				grub_efi_key_notify_function_t key_notification_function);
4fe85b
+
4fe85b
+	grub_efi_status_t
4fe85b
+	(*unregister_key_notify) (struct grub_efi_simple_text_input_ex_interface *this,
4fe85b
+				  void *notification_handle);
4fe85b
+};
4fe85b
+typedef struct grub_efi_simple_text_input_ex_interface grub_efi_simple_text_input_ex_interface_t;
4fe85b
+
4fe85b
 struct grub_efi_simple_text_output_interface
4fe85b
 {
4fe85b
   grub_efi_status_t