demote 'system_thread' to just 'system_context'
[model-checker.git] / model.h
diff --git a/model.h b/model.h
index 54144f38fc322df9a79edf2e8d0c542f25d027d8..ad8760a28576aafecc17d05d9cb01b92e7df259d 100644 (file)
--- a/model.h
+++ b/model.h
@@ -4,6 +4,7 @@
 #include <list>
 #include <map>
 #include <cstddef>
+#include <ucontext.h>
 
 #include "schedule.h"
 #include "libthreads.h"
@@ -31,9 +32,9 @@ public:
        thread_id_t get_tid() { return tid; }
        action_type get_type() { return type; }
        memory_order get_mo() { return order; }
-       void *get_location() { return location; }
+       void * get_location() { return location; }
 
-       TreeNode *get_node() { return node; }
+       TreeNode * get_node() { return node; }
        void set_node(TreeNode *n) { node = n; }
 private:
        action_type type;
@@ -49,15 +50,19 @@ public:
        Backtrack(ModelAction *d, action_list_t *t) {
                diverge = d;
                actionTrace = t;
-               //currentIterator = actionTrace->getFirst();
+               iter = actionTrace->begin();
+       }
+       ModelAction * get_diverge() { return diverge; }
+       action_list_t * get_trace() { return actionTrace; }
+       void advance_state() { iter++; }
+       ModelAction * get_state() {
+               return iter == actionTrace->end() ? NULL : *iter;
        }
-       ModelAction *get_diverge() { return diverge; }
-       action_list_t *get_trace() { return actionTrace; }
 private:
        ModelAction *diverge;
-       /* unused for now; will be used when re-exploring this path? */
-       //MyListElement *currentIterator;
        action_list_t *actionTrace;
+       /* points to position in actionTrace as we replay */
+       action_list_t::iterator iter;
 };
 
 class ModelChecker {
@@ -65,25 +70,40 @@ public:
        ModelChecker();
        ~ModelChecker();
        class Scheduler *scheduler;
-       Thread *system_thread;
 
-       void add_system_thread(Thread *t);
+       void set_system_context(ucontext_t *ctxt) { system_context = ctxt; }
+       ucontext_t * get_system_context(void) { return system_context; }
 
        void set_current_action(ModelAction *act) { current_action = act; }
-       ModelAction *get_last_conflict(ModelAction *act);
        void check_current_action(void);
-       void set_backtracking(ModelAction *act);
-       void print_trace(void);
+       void print_summary(void);
+       Thread * schedule_next_thread();
 
        int add_thread(Thread *t);
-       Thread *get_thread(thread_id_t tid) { return thread_map[tid]; }
+       void remove_thread(Thread *t);
+       Thread * get_thread(thread_id_t tid) { return thread_map[tid]; }
 
-       void assign_id(Thread *t);
+       int get_next_id();
 
        int switch_to_master(ModelAction *act);
+
+       bool next_execution();
 private:
        int used_thread_id;
+       int num_executions;
+
+       ModelAction * get_last_conflict(ModelAction *act);
+       void set_backtracking(ModelAction *act);
+       thread_id_t advance_backtracking_state();
+       thread_id_t get_next_replay_thread();
+       Backtrack * get_next_backtrack();
+       void reset_to_initial_state();
+
        class ModelAction *current_action;
+       Backtrack *exploring;
+       thread_id_t nextThread;
+
+       ucontext_t *system_context;
        action_list_t *action_trace;
        std::map<thread_id_t, class Thread *> thread_map;
        class TreeNode *rootNode, *currentNode;
@@ -92,6 +112,4 @@ private:
 
 extern ModelChecker *model;
 
-int thread_switch_to_master(ModelAction *act);
-
 #endif /* __MODEL_H__ */