model: flesh out check_current_action()

[c11tester.git] / model.h

diff --git a/model.h b/model.h
index e075ea254640e7f32c0392d027af60a308b329c2..b6a0004d5a3d119e413d01e5184741930ac5e505 100644 (file)
--- a/model.h
+++ b/model.h
@@ -1,12 +1,76 @@
 #ifndef __MODEL_H__
 #define __MODEL_H__
 
-struct model_checker {
-       struct scheduler *scheduler;
+#include <list>
+#include <map>
+
+#include "schedule.h"
+#include "libthreads.h"
+#include "libatomic.h"
+#include "threads.h"
+#include "tree.h"
+
+#define VALUE_NONE -1
+
+typedef enum action_type {
+       THREAD_CREATE,
+       THREAD_YIELD,
+       THREAD_JOIN,
+       ATOMIC_READ,
+       ATOMIC_WRITE
+} action_type_t;
+
+class ModelAction {
+public:
+       ModelAction(action_type_t type, memory_order order, void *loc, int value);
+       void print(void);
+
+       thread_id_t get_tid() { return tid; }
+       action_type get_type() { return type; }
+       memory_order get_mo() { return order; }
+       void *get_location() { return location; }
+
+       TreeNode *get_node() { return node; }
+       void set_node(TreeNode *n) { node = n; }
+private:
+       action_type type;
+       memory_order order;
+       void *location;
+       thread_id_t tid;
+       int value;
+       TreeNode *node;
+};
+
+class ModelChecker {
+public:
+       ModelChecker();
+       ~ModelChecker();
+       class Scheduler *scheduler;
+       Thread *system_thread;
+
+       void add_system_thread(Thread *t);
+
+       void set_current_action(ModelAction *act) { current_action = act; }
+       void check_current_action(void);
+       void set_backtracking(ModelAction *act);
+       void print_trace(void);
+
+       int add_thread(Thread *t);
+       Thread *get_thread(thread_id_t tid) { return thread_map[tid]; }
+
+       void assign_id(Thread *t);
+
+       int switch_to_master(ModelAction *act);
+private:
+       int used_thread_id;
+       class ModelAction *current_action;
+       std::list<class ModelAction *> action_trace;
+       std::map<thread_id_t, class Thread *> thread_map;
+       class TreeNode *rootNode, *currentNode;
 };
 
-extern struct model_checker *model;
+extern ModelChecker *model;
 
-void model_checker_init(void);
+int thread_switch_to_master(ModelAction *act);
 
 #endif /* __MODEL_H__ */