3 files changed, 16 insertions, 0 deletions
diff --git a/UefiMonitorTest/UefiMonitorTest.c b/UefiMonitorTest/UefiMonitorTest.c
index 92e5799..679eed9 100644
--- a/UefiMonitorTest/UefiMonitorTest.c
+++ b/UefiMonitorTest/UefiMonitorTest.c
@@ -193,6 +193,17 @@ PrepareGraphicsInfo (
   Graphics->PixelWidth  = PixelWidth;
   Graphics->Pitch       = Gop->Mode->Info->PixelsPerScanLine;
 
+  // Find TextInEx in System Table ConsoleInHandle
+  Status = gBS->HandleProtocol (
+                  gST->ConsoleInHandle,
+                  &gEfiSimpleTextInputExProtocolGuid,
+                  (VOID **)&Graphics->TextInputEx
+                  );
+  if (EFI_ERROR (Status)) {
+    FreePool (Graphics->BackBuffer);
+    return Status;
+  }
+
   // Find mouse in System Table ConsoleInHandle
   Status = gBS->HandleProtocol (
                   gST->ConsoleInHandle,
diff --git a/UefiMonitorTest/UefiMonitorTest.h b/UefiMonitorTest/UefiMonitorTest.h
index 3f82246..7b07c31 100644
--- a/UefiMonitorTest/UefiMonitorTest.h
+++ b/UefiMonitorTest/UefiMonitorTest.h
@@ -57,6 +57,10 @@ typedef struct {
   ///
   UINT32                                  Pitch;
   //
+  // The EFI_SIMPLE_TEXT_INPUT_EX_PROTOCOL instance.
+  //
+  EFI_SIMPLE_TEXT_INPUT_EX_PROTOCOL       *TextInputEx;
+  //
   // A mouse support.
   //
   BOOLEAN                                 MouseSupported;
diff --git a/UefiMonitorTest/UefiMonitorTest.inf b/UefiMonitorTest/UefiMonitorTest.inf
index 7eb27cc..e0e7c13 100644
--- a/UefiMonitorTest/UefiMonitorTest.inf
+++ b/UefiMonitorTest/UefiMonitorTest.inf
@@ -26,6 +26,7 @@
 
 [Protocols]
   gEfiSimplePointerProtocolGuid
+  gEfiSimpleTextInputExProtocolGuid
 
 [FeaturePcd]