Defining the interfaces to add various regions to snapshot
[model-checker.git] / model.h
diff --git a/model.h b/model.h
index 625d2726fe775921f981e5d666426eb85cb32148..d3b42053fcee011ed32e47df7fef9822aadd144d 100644 (file)
--- a/model.h
+++ b/model.h
@@ -1,18 +1,74 @@
 #ifndef __MODEL_H__
 #define __MODEL_H__
 
-struct model_checker {
-       struct scheduler *scheduler;
-       struct thread *system_thread;
+#include <list>
+#include <map>
+#include <vector>
+#include <cstddef>
+#include <ucontext.h>
 
-       /* "Private" fields */
-       int used_thread_id;
-};
+#include "schedule.h"
+#include "mymemory.h"
+#include <utility>
+#include "libthreads.h"
+#include "libatomic.h"
+#include "threads.h"
+#include "action.h"
+
+/* Forward declaration */
+class TreeNode;
+class Backtrack;
+
+class ModelChecker {
+public:
+       ModelChecker();
+       ~ModelChecker();
+       class Scheduler *scheduler;
+
+       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; }
+       void check_current_action(void);
+       void print_summary(void);
+       Thread * schedule_next_thread();
+
+       int add_thread(Thread *t);
+       void remove_thread(Thread *t);
+       Thread * get_thread(thread_id_t tid) { return thread_map[id_to_int(tid)]; }
+
+       thread_id_t get_next_id();
+       int get_next_seq_num();
 
-extern struct model_checker *model;
+       int switch_to_master(ModelAction *act);
+
+       bool next_execution();
+  MEMALLOC
+private:
+       int next_thread_id;
+       int used_sequence_numbers;
+       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();
+
+       void print_list(action_list_t *list);
+
+       class ModelAction *current_action;
+       Backtrack *exploring;
+       thread_id_t nextThread;
+
+       ucontext_t *system_context;
+       action_list_t *action_trace;
+       std::map<int, class Thread *, std::less< int >, MyAlloc< std::pair< const int, class Thread * > > > thread_map;
+       class TreeNode *rootNode, *currentNode;
+       std::list<class Backtrack *, MyAlloc< class Backtrack * > > backtrack_list;
+};
 
-void model_checker_init(void);
-void model_checker_add_system_thread(struct thread *t);
-void model_checker_assign_id(struct thread *t);
+extern ModelChecker *model;
 
 #endif /* __MODEL_H__ */