fix bug
[c11tester.git] / main.cc
diff --git a/main.cc b/main.cc
index 741ef7775a8bffbb69a5065b5ec2d8613eebd0e6..a400fe3c2defff0eafc93e812b005f6b2b6fcf69 100644 (file)
--- a/main.cc
+++ b/main.cc
@@ -1,18 +1,25 @@
+/** @file main.cc
+ *  @brief Entry point for the model checker.
+ */
+
 #include "libthreads.h"
 #include "common.h"
 #include "threads.h"
 
+#include "datarace.h"
+
 /* global "model" object */
 #include "model.h"
-#include "snapshot.h"
 #include "snapshot-interface.h"
 
-/*
- * Return 1 if found next thread, 0 otherwise
+/** The thread_system_next function takes the next step in the
+ *  execution.  @return Returns the 1 if there is another step and 0
+ *  otherwise.
  */
+
 static int thread_system_next(void) {
        Thread *curr, *next;
-  
+
        curr = thread_current();
        if (curr) {
                if (curr->get_state() == THREAD_READY) {
@@ -33,44 +40,56 @@ static int thread_system_next(void) {
        return Thread::swap(model->get_system_context(), next);
 }
 
+/** The thread_wait_finish method runs the current execution until we
+ *  have no more steps to take.
+ */
+
 static void thread_wait_finish(void) {
        DBG();
 
        while (!thread_system_next());
 }
 
+
+/** The real_main function contains the main model checking loop. */
+
 void real_main() {
        thrd_t user_thread;
        ucontext_t main_context;
+  //Initialize race detector
+  initRaceDetector();
+
+       //Create the singleton SnapshotStack object
+       snapshotObject = new SnapshotStack();
 
-       //Create the singleton snapshotStack object
-       snapshotObject = new snapshotStack();
-  
        model = new ModelChecker();
-  
+
        if (getcontext(&main_context))
                return;
-  
+
        model->set_system_context(&main_context);
 
+       snapshotObject->snapshotStep(0);
+
        do {
                /* Start user program */
                model->add_thread(new Thread(&user_thread, (void (*)(void *)) &user_main, NULL));
-    
+
                /* Wait for all threads to complete */
                thread_wait_finish();
        } while (model->next_execution());
-  
+
        delete model;
-  
+
        DEBUG("Exiting\n");
 }
 
 int main_numargs;
 char ** main_args;
 
-/*
- * Main system function
+/**
+ * Main function.  Just initializes snapshotting library and the
+ * snapshotting library calls the real_main function.
  */
 int main(int numargs, char ** args) {
        /* Stash this stuff in case someone wants it eventually */