trivial changes
[c11tester.git] / model.h
diff --git a/model.h b/model.h
index 73c3d0c0f91ad0e5d246a9bb788adda0f8d76a91..fbf1b3f0909ff3fd67fbb10e905d1c0b7ff9003b 100644 (file)
--- a/model.h
+++ b/model.h
@@ -1,8 +1,100 @@
+/** @file model.h
+ *  @brief Core model checker. 
+ */
+
 #ifndef __MODEL_H__
 #define __MODEL_H__
 
-struct model_checker {
-       struct scheduler *scheduler;
+#include <list>
+#include <map>
+#include <vector>
+#include <cstddef>
+#include <ucontext.h>
+
+#include "schedule.h"
+#include "mymemory.h"
+#include <utility>
+#include "libthreads.h"
+#include "libatomic.h"
+#include "threads.h"
+#include "action.h"
+#include "clockvector.h"
+
+/* Forward declaration */
+class NodeStack;
+
+/** @brief The central structure for model-checking */
+class ModelChecker {
+public:
+       ModelChecker();
+       ~ModelChecker();
+
+       /** The scheduler to use: tracks the running/ready Threads */
+       Scheduler *scheduler;
+
+       /** Stores the context for the main model-checking system thread (call
+        * once)
+        * @param ctxt The system context structure
+        */
+       void set_system_context(ucontext_t *ctxt) { system_context = ctxt; }
+
+       /** @returns the context for the main model-checking system thread */
+       ucontext_t * get_system_context(void) { return system_context; }
+
+       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_num_threads();
+       modelclock_t get_next_seq_num();
+
+       int switch_to_master(ModelAction *act);
+
+       bool next_execution();
+
+       MEMALLOC
+private:
+       int next_thread_id;
+       modelclock_t used_sequence_numbers;
+       int num_executions;
+
+       /**
+        * Stores the ModelAction for the current thread action.  Call this
+        * immediately before switching from user- to system-context to pass
+        * data between them.
+        * @param act The ModelAction created by the user-thread action
+        */
+       void set_current_action(ModelAction *act) { current_action = act; }
+
+       ModelAction * get_last_conflict(ModelAction *act);
+       void set_backtracking(ModelAction *act);
+       thread_id_t get_next_replay_thread();
+       ModelAction * get_next_backtrack();
+       void reset_to_initial_state();
+
+       void add_action_to_lists(ModelAction *act);
+       ModelAction * get_last_action(thread_id_t tid);
+       ModelAction * get_parent_action(thread_id_t tid);
+       void build_reads_from_past(ModelAction *curr);
+
+       ModelAction *current_action;
+       ModelAction *diverge;
+       thread_id_t nextThread;
+
+       ucontext_t *system_context;
+       action_list_t *action_trace;
+       std::map<int, Thread *> *thread_map;
+       std::map<void *, std::vector<action_list_t> > *obj_thrd_map;
+       std::vector<ModelAction *> *thrd_last_action;
+       NodeStack *node_stack;
+       ModelAction *next_backtrack;
 };
 
+extern ModelChecker *model;
+
 #endif /* __MODEL_H__ */