Implement more wrappers and refactoring

sdk/xtdk/bltypes.h

@@ -39,6 +39,8 @@ typedef VOID (*PBL_CONSOLE_DISABLE_CURSOR)();
 typedef VOID (*PBL_CONSOLE_ENABLE_CURSOR)();
 typedef VOID (*PBL_CONSOLE_PRINT)(IN PUINT16 Format, IN ...);
 typedef VOID (*PBL_CONSOLE_QUERY_MODE)(OUT PUINT_PTR ResX, OUT PUINT_PTR ResY);
+typedef VOID (*PBL_CONSOLE_READ_KEY_STROKE)(OUT PEFI_INPUT_KEY Key);
+typedef VOID (*PBL_CONSOLE_RESET_INPUT_BUFFER)();
 typedef VOID (*PBL_CONSOLE_SET_ATTRIBUTES)(IN ULONGLONG Attributes);
 typedef VOID (*PBL_CONSOLE_SET_CURSOR_POSITION)(IN ULONGLONG PosX, IN ULONGLONG PosY);
 typedef VOID (*PBL_CONSOLE_WRITE)(IN PUSHORT String);
@@ -51,6 +53,7 @@ typedef EFI_STATUS (*PBL_OPEN_VOLUME)(IN PEFI_DEVICE_PATH_PROTOCOL DevicePath, O
 typedef EFI_STATUS (*PBL_OPEN_XT_PROTOCOL)(OUT PVOID *ProtocolHandler, IN PEFI_GUID ProtocolGuid);
 typedef EFI_STATUS (*PBL_READ_FILE)(IN PEFI_FILE_HANDLE DirHandle, IN CONST PWCHAR FileName, OUT PVOID *FileData, OUT PSIZE_T FileSize);
 typedef VOID (*PBL_SLEEP_EXECUTION)(IN ULONG_PTR Milliseconds);
+typedef EFI_STATUS (*PBL_WAIT_FOR_EFI_EVENT)(IN UINT_PTR NumberOfEvents, IN PEFI_EVENT Event, OUT PUINT_PTR Index);
 
 /* XTLDR Configuration data */
 typedef struct _XTBL_CONFIG_ENTRY
@@ -86,6 +89,8 @@ typedef struct _XTBL_LOADER_PROTOCOL
         PBL_CONSOLE_ENABLE_CURSOR EnableCursor;
         PBL_CONSOLE_PRINT Print;
         PBL_CONSOLE_QUERY_MODE QueryMode;
+        PBL_CONSOLE_READ_KEY_STROKE ReadKeyStroke;
+        PBL_CONSOLE_RESET_INPUT_BUFFER ResetInputBuffer;
         PBL_CONSOLE_SET_ATTRIBUTES SetAttributes;
         PBL_CONSOLE_SET_CURSOR_POSITION SetCursorPosition;
         PBL_CONSOLE_WRITE Write;
@@ -116,6 +121,7 @@ typedef struct _XTBL_LOADER_PROTOCOL
         PBL_EXIT_BOOT_SERVICES ExitBootServices;
         PBL_GET_SECURE_BOOT_STATUS GetSecureBootStatus;
         PBL_SLEEP_EXECUTION SleepExecution;
+        PBL_WAIT_FOR_EFI_EVENT WaitForEfiEvent;
     } Util;
 } XTBL_LOADER_PROTOCOL, *PXTBL_LOADER_PROTOCOL;