From 080ff55bc2d1b4d928b30159812033c31ae27ae5 Mon Sep 17 00:00:00 2001
From: "Roderick W. Smith" <rodsmith@rodsbooks.com>
Date: Sun, 18 Oct 2015 13:43:03 -0400
Subject: Fixed bug that caused input glitches (trailing line feeds and
 difficulty entering type codes) on EFI version.

---
 support.cc | 11 +++++++++--
 1 file changed, 9 insertions(+), 2 deletions(-)

(limited to 'support.cc')

diff --git a/support.cc b/support.cc
index 7a6354d..0ff3485 100644
--- a/support.cc
+++ b/support.cc
@@ -39,9 +39,16 @@ extern int __sscanf( const char * str , const char * format , ... ) ;
 string ReadString(void) {
    string inString;
    char efiString[256];
+   int stringLength;
 
-   fgets(efiString, 255, stdin);
-   inString = efiString;
+   if (fgets(efiString, 255, stdin) != NULL) {
+      stringLength = strlen(efiString);
+      if ((stringLength > 0) && (efiString[stringLength - 1] == '\n'))
+          efiString[stringLength - 1] = '\0';
+      inString = efiString;
+   } else {
+      inString = "";
+   }
    return inString;
 } // ReadString()
 #else
-- 
cgit v1.2.1