summaryrefslogtreecommitdiff
path: root/support.cc
diff options
context:
space:
mode:
Diffstat (limited to 'support.cc')
-rw-r--r--support.cc11
1 files changed, 9 insertions, 2 deletions
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