fun issues...
[c11tester.git] / main.cc
diff --git a/main.cc b/main.cc
index 75efafed43870f8be079fc8fcb1cb49bd8412706..215a2197d35e91fa29cf65b869a292165be4a9ad 100644 (file)
--- a/main.cc
+++ b/main.cc
@@ -2,6 +2,8 @@
  *  @brief Entry point for the model checker.
  */
 
+#include <unistd.h>
+
 #include "libthreads.h"
 #include "common.h"
 #include "threads.h"
 #include "model.h"
 #include "snapshot-interface.h"
 
-static void parse_options(struct model_params *params, int argc, char **argv) {
+static void param_defaults(struct model_params * params) {
+       params->maxreads = 0;
+}
+
+static void print_usage() {
+       printf(
+"Usage: <program name> [MC_OPTIONS] -- [PROGRAM ARGUMENTS]\n"
+"\n"
+"Options:\n"
+"-h                    Display this help message and exit\n"
+"-m                    Maximum times a thread can read from the same write while other writes exist\n"
+"--                    Program arguments follow.\n\n");
+       exit(EXIT_SUCCESS);
+}
+
+static void parse_options(struct model_params *params, int *argc, char ***argv) {
+       const char *shortopts = "hm:";
+       int opt;
+       bool error = false;
+       while (!error && (opt = getopt(*argc, *argv, shortopts)) != -1) {
+               switch (opt) {
+               case 'h':
+                       print_usage();
+                       break;
+               case 'm':
+                       params->maxreads = atoi(optarg);
+                       break;
+               default: /* '?' */
+                       error = true;
+                       break;
+               }
+       }
+       (*argc) -= optind;
+       (*argv) += optind;
+
+       if (error)
+               print_usage();
 }
 
 int main_argc;
@@ -23,7 +61,9 @@ static void real_main() {
        thrd_t user_thread;
        struct model_params params;
 
-       parse_options(&params, main_argc, main_argv);
+       param_defaults(&params);
+
+       parse_options(&params, &main_argc, &main_argv);
 
        //Initialize race detector
        initRaceDetector();