Add some methods for WaitObj
[c11tester.git] / history.h
index b42c34b1ca99884e82a836bd016e0ceb7421c570..803077693e6ef8edab7ddd0f2552a1e341502708 100644 (file)
--- a/history.h
+++ b/history.h
@@ -1,10 +1,11 @@
-#include "stl-model.h"
+#ifndef __HISTORY_H__
+#define __HISTORY_H__
+
 #include "common.h"
+#include "classlist.h"
 #include "hashtable.h"
 #include "threads-model.h"
 
-typedef SnapList<uint32_t> func_id_list_t;
-
 class ModelHistory {
 public:
        ModelHistory();
@@ -16,27 +17,70 @@ public:
        uint32_t get_func_counter() { return func_counter; }
        void incr_func_counter() { func_counter++; }
 
-       void add_func_atomic(ModelAction *act, thread_id_t tid);
+       void resize_func_nodes(uint32_t max_func_id);
+       void process_action(ModelAction *act, thread_id_t tid);
+
+       HashTable<const char *, uint32_t, uintptr_t, 4, model_malloc, model_calloc, model_free> * getFuncMap() { return &func_map; }
+       ModelVector<const char *> * getFuncMapRev() { return &func_map_rev; }
+
+       ModelVector<FuncNode *> * getFuncNodes() { return &func_nodes; }
+       FuncNode * get_func_node(uint32_t func_id);
+       FuncNode * get_curr_func_node(thread_id_t tid);
+
+       void update_write_history(void * location, uint64_t write_val);
+       HashTable<void *, value_set_t *, uintptr_t, 4> * getWriteHistory() { return write_history; }
+       void update_loc_rd_func_nodes_map(void * location, FuncNode * node);
+       void update_loc_wr_func_nodes_map(void * location, FuncNode * node);
+       SnapVector<FuncNode *> * getRdFuncNodes(void * location);
+       SnapVector<FuncNode *> * getWrFuncNodes(void * location);
+
+       void add_waiting_write(ConcretePredicate * concrete);
+       void remove_waiting_write(thread_id_t tid);
+       void check_waiting_write(ModelAction * write_act);
+       SnapVector<ConcretePredicate *> * getThrdWaitingWrite() { return thrd_waiting_write; }
+
+       WaitObj * getWaitObj(thread_id_t tid);
+       void add_waiting_thread(thread_id_t self_id, thread_id_t waiting_for_id, int dist);
+       void remove_waiting_thread(thread_id_t self_id, thread_id_t waiting_for_id);
 
-       HashTable<const char *, uint32_t, uintptr_t, 4> * getFuncMap() { return &func_map; }
-       ModelVector<FuncNode *> * getFuncAtomics() { return &func_atomics; }
+       SnapVector<inst_act_map_t *> * getThrdInstActMap(uint32_t func_id);
 
-       void print();
+       void set_new_exec_flag();
+       void dump_func_node_graph();
+       void print_func_node();
+       void print_waiting_threads();
 
        MEMALLOC
 private:
        uint32_t func_counter;
 
-       /* map function names to integer ids */ 
-       HashTable<const char *, uint32_t, uintptr_t, 4> func_map;
+       /* Map function names to integer ids */
+       HashTable<const char *, uint32_t, uintptr_t, 4, model_malloc, model_calloc, model_free> func_map;
 
-       ModelVector<FuncNode *> func_atomics;
+       /* Map integer ids to function names */
+       ModelVector<const char *> func_map_rev;
 
-       /* Work_list stores a list of function ids for each thread. 
-        * Each element in work_list is intended to be used as a stack storing
-        * the functions that thread i has entered and yet to exit from 
-        */
+       ModelVector<FuncNode *> func_nodes;
 
-       /* todo: move work_list to execution.cc to avoid seg fault */
-       SnapVector< func_id_list_t * > work_list;
+       /* Map a location to a set of values that have been written to it */
+       HashTable<void *, value_set_t *, uintptr_t, 4> * write_history;
+
+       /* Map a location to FuncNodes that may read from it */
+       HashTable<void *, SnapVector<FuncNode *> *, uintptr_t, 0> * loc_rd_func_nodes_map;
+
+       /* Map a location to FuncNodes that may write to it */
+       HashTable<void *, SnapVector<FuncNode *> *, uintptr_t, 0> * loc_wr_func_nodes_map;
+
+       HashTable<void *, SnapVector<ConcretePredicate *> *, uintptr_t, 0> * loc_waiting_writes_map;
+       /* The write values each paused thread is waiting for */
+       SnapVector<ConcretePredicate *> * thrd_waiting_write;
+       SnapVector<WaitObj *> * thrd_wait_obj;
+
+       /* A run-time map from FuncInst to ModelAction per each thread, per each FuncNode.
+        * Manipulated by FuncNode, and needed by NewFuzzer */
+       HashTable<uint32_t, SnapVector<inst_act_map_t *> *, int, 0> * func_inst_act_maps;
+
+       bool skip_action(ModelAction * act, SnapList<ModelAction *> * curr_act_list);
 };
+
+#endif /* __HISTORY_H__ */