add loop

hyrise · Mar 11, 2024 · 444f5bf · 444f5bf
1 parent 26d0045
commit 444f5bf
Showing 1 changed file with 13 additions and 11 deletions.
diff --git a/bm_utils.c b/bm_utils.c
@@ -130,18 +130,20 @@ yes_no(char *prompt)
 #pragma warning(default:4127)
 #endif 
         printf("%s [Y/N]: ", prompt);
-        // HYRISE: Replace gets() with fgets().
-        fgets(reply, sizeof(reply), stdin);
-        switch (*reply)
+        // HYRISE: Replace gets() with fgets(). Add loop.
+        while (fgets(reply, sizeof(reply), stdin) != NULL)
             {
-            case 'y':
-            case 'Y':
-                return (1);
-            case 'n':
-            case 'N':
-                return (0);
-            default:
-                printf("Please answer 'yes' or 'no'.\n");
+            switch (*reply)
+                {
+                case 'y':
+                case 'Y':
+                    return (1);
+                case 'n':
+                case 'N':
+                    return (0);
+                default:
+                    printf("Please answer 'yes' or 'no'.\n");
+                }
             }
         }
 }