sample_substr_validator: bug fix for --reject_if_present

MrBrain295 · Feb 1, 2015 · 57e8f8c · 57e8f8c
1 parent 5c5f996
commit 57e8f8c
Showing 1 changed file with 8 additions and 5 deletions.
diff --git a/sched/sample_substr_validator.cpp b/sched/sample_substr_validator.cpp
@@ -31,19 +31,22 @@ char* stderr_string;
 bool reject_if_present = false;
 
 void parse_cmdline() {
+    bool found = false;
     for (int i=1; i<g_argc; i++) {
         if (!strcmp(g_argv[i], "--stderr_string")) {
             stderr_string = g_argv[++i];
-            return;
+            found = true;
         }
         if (!strcmp(g_argv[i], "--reject_if_present")) {
             reject_if_present = true;
         }
     }
-    log_messages.printf(MSG_CRITICAL,
-        "--stderr_string missing from command line\n"
-    );
-    exit(1);
+    if (!found) {
+        log_messages.printf(MSG_CRITICAL,
+            "--stderr_string missing from command line\n"
+        );
+        exit(1);
+    }
 }
 
 int init_result(RESULT& r, void*&) {