From b6d7ea98fc46c20b59860edcffac13fee89d69b0 Mon Sep 17 00:00:00 2001
From: kavsrf <kavsrf@gmail.com>
Date: Sat, 20 Aug 2016 10:35:10 +0300
Subject: fix double keystrokes

---
 Library/CommonLib/EfiConsole.c        | 15 +++++++++++----
 Library/PasswordLib/PicturePassword.c | 11 +++++++++--
 2 files changed, 20 insertions(+), 6 deletions(-)

(limited to 'Library')

diff --git a/Library/CommonLib/EfiConsole.c b/Library/CommonLib/EfiConsole.c
index 8a87ec8..f03f3f8 100644
--- a/Library/CommonLib/EfiConsole.c
+++ b/Library/CommonLib/EfiConsole.c
@@ -72,7 +72,9 @@ KeyWait(
       OUT_PRINT(Prompt, mDelay);
       gBS->WaitForEvent(2, InputEvents, &EventIndex);
       if (EventIndex == 0) {
-         gST->ConIn->ReadKeyStroke(gST->ConIn, &key);
+			if (EFI_ERROR(gST->ConIn->ReadKeyStroke(gST->ConIn, &key))) {
+				continue;
+			}
          break;
       }
       else {
@@ -89,9 +91,12 @@ GetKey(void)
 {
    EFI_INPUT_KEY key;
    UINTN EventIndex;
-
-   gBS->WaitForEvent(1, &gST->ConIn->WaitForKey, &EventIndex);
-   gST->ConIn->ReadKeyStroke(gST->ConIn, &key);
+	EFI_STATUS res1;
+	EFI_STATUS res2;
+	do {
+		res1 = gBS->WaitForEvent(1, &gST->ConIn->WaitForKey, &EventIndex);
+		res2 = gST->ConIn->ReadKeyStroke(gST->ConIn, &key);
+	} while (EFI_ERROR(res1) || EFI_ERROR(res2));
    return key;
 }
 
@@ -135,6 +140,8 @@ GetLine (
 
    do {
       key = GetKey();
+		// Remove dirty chars 0.1s
+		FlushInputDelay(100000);
 
       if ((count >= line_max &&
          key.UnicodeChar != CHAR_BACKSPACE) ||
diff --git a/Library/PasswordLib/PicturePassword.c b/Library/PasswordLib/PicturePassword.c
index f917899..67ffcbe 100644
--- a/Library/PasswordLib/PicturePassword.c
+++ b/Library/PasswordLib/PicturePassword.c
@@ -15,6 +15,7 @@ https://opensource.org/licenses/LGPL-3.0
 #include <Uefi.h>
 #include <Library/UefiBootServicesTableLib.h>
 #include <Library/DebugLib.h>
+#include <Library/BaseMemoryLib.h>
 
 #include <Library/CommonLib.h>
 #include <Library/GraphLib.h>
@@ -379,7 +380,14 @@ AskPictPwdInt(
 	{
 		curPrevX = curX;
 		curPrevY = curY;
+		ZeroMem(&key, sizeof(key));
 		res = gBS->WaitForEvent(eventsCount, InputEvents, &EventIndex);
+		if (EventIndex == 0) {
+			res = gST->ConIn->ReadKeyStroke(gST->ConIn, &key);
+			if (EFI_ERROR(res)) {
+				continue;
+			}
+		}
 		// OUT_PRINT(L" \r%05d %05d", (UINT32)curX, (UINT32)curY, );
 		// recharge timeout event and stop beep
 		if (gBeepControlEnabled && gBeepEnabled && beepOn) {
@@ -406,9 +414,8 @@ AskPictPwdInt(
 		}
 
 		if (EventIndex == 0) {
-			gST->ConIn->ReadKeyStroke(gST->ConIn, &key);
+//			gST->ConIn->ReadKeyStroke(gST->ConIn, &key);
 			// OUT_PRINT(L" %04x, %04x\r", key.ScanCode, key.UnicodeChar);
-
 			// Remove dirty chars 0.1s
 			FlushInputDelay(100000);
 			switch (key.ScanCode)
-- 
cgit v1.2.3