setup main wrapper and then call into norris code

[model-checker.git] / main.cc

diff --git a/main.cc b/main.cc
index e456ac6b326469a85383af6a97a422e9a2c65f92..e629f6092992ce8fb59a80a7e14388b7b7724143 100644 (file)
--- a/main.cc
+++ b/main.cc
@@ -4,6 +4,7 @@
 
 /* global "model" object */
 #include "model.h"
+#include "snapshot.h"
 
 /*
  * Return 1 if found next thread, 0 otherwise
@@ -12,63 +13,70 @@ int num;
 int num1;
 int num2;
 
-static int thread_system_next(void)
-{
-       Thread *curr, *next;
-
-       curr = thread_current();
-       if (curr) {
-               if (curr->get_state() == THREAD_READY) {
-                       model->check_current_action();
-                       model->scheduler->add_thread(curr);
-               } else if (curr->get_state() == THREAD_RUNNING)
-                       /* Stopped while running; i.e., completed */
-                       curr->complete();
-               else
-                       ASSERT(false);
-       }
-       next = model->scheduler->next_thread();
-       if (next)
-               next->set_state(THREAD_RUNNING);
-       DEBUG("(%d, %d)\n", curr ? curr->get_id() : -1, next ? next->get_id() : -1);
-       if (!next)
-               return 1;
-       return Thread::swap(model->get_system_context(), next);
+static int thread_system_next(void) {
+  Thread *curr, *next;
+  
+  curr = thread_current();
+  if (curr) {
+    if (curr->get_state() == THREAD_READY) {
+      model->check_current_action();
+      model->scheduler->add_thread(curr);
+    } else if (curr->get_state() == THREAD_RUNNING)
+      /* Stopped while running; i.e., completed */
+      curr->complete();
+    else
+      ASSERT(false);
+  }
+  next = model->scheduler->next_thread();
+  if (next)
+    next->set_state(THREAD_RUNNING);
+  DEBUG("(%d, %d)\n", curr ? curr->get_id() : -1, next ? next->get_id() : -1);
+  if (!next)
+    return 1;
+  return Thread::swap(model->get_system_context(), next);
 }
 
-static void thread_wait_finish(void)
-{
+static void thread_wait_finish(void) {
+  DBG();
+
+  while (!thread_system_next());
+}
 
-       DBG();
+void real_main() {
+  thrd_t user_thread;
+  ucontext_t main_context;
+  
+  model = new ModelChecker();
+  
+  if (getcontext(&main_context))
+    return;
+  
+  model->set_system_context(&main_context);
 
-       while (!thread_system_next());
+  do {
+    /* Start user program */
+    model->add_thread(new Thread(&user_thread, &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
  */
-int main()
-{
-       thrd_t user_thread;
-       ucontext_t main_context;
-
-       model = new ModelChecker();
-
-       if (getcontext(&main_context))
-               return 1;
-
-       model->set_system_context(&main_context);
-
-       do {
-               /* Start user program */
-               model->add_thread(new Thread(&user_thread, &user_main, NULL));
-
-               /* Wait for all threads to complete */
-               thread_wait_finish();
-       } while (model->next_execution());
-
-       delete model;
+int main(int numargs, char ** args) {
+  /* Stash this stuff in case someone wants it eventually */
+  main_numargs=numargs;
+  main_args=args;
 
-       DEBUG("Exiting\n");
-       return 0;
+  /* Let's jump in quickly and start running stuff */
+  initSnapShotLibrary(10000 /*int numbackingpages*/, 1024 /*unsigned int numsnapshots*/, 1024 /*unsigned int nummemoryregions*/ , &real_main /*MyFuncPtr entryPoint*/);
 }