+/** @file model.h
+ * @brief Core model checker.
+ */
+
#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 NodeStack;
+
+/** @brief The central structure for model-checking */
+class ModelChecker {
+public:
+ ModelChecker();
+ ~ModelChecker();
+
+ /** The scheduler to use: tracks the running/ready Threads */
+ class 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; }
+
+ /**
+ * 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; }
+ 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)]; }
-extern struct model_checker *model;
+ thread_id_t get_next_id();
+ int get_num_threads();
+ int get_next_seq_num();
+
+ 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 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, class Thread *> *thread_map;
+ std::map<void *, std::vector<action_list_t> > *obj_thrd_map;
+ std::vector<ModelAction *> *thrd_last_action;
+ class NodeStack *node_stack;
+ ModelAction *next_backtrack;
+};
-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__ */