history.h

   1 #ifndef __HISTORY_H__
   2 #define __HISTORY_H__
   3
   4 #include "stl-model.h"
   5 #include "common.h"
   6 #include "classlist.h"
   7 #include "hashtable.h"
   8 #include "threads-model.h"
   9
  10 class ModelHistory {
  11 public:
  12         ModelHistory();
  13         ~ModelHistory();
  14
  15         void enter_function(const uint32_t func_id, thread_id_t tid);
  16         void exit_function(const uint32_t func_id, thread_id_t tid);
  17
  18         uint32_t get_func_counter() { return func_counter; }
  19         void incr_func_counter() { func_counter++; }
  20
  21         void resize_func_nodes(uint32_t max_func_id);
  22         void process_action(ModelAction *act, thread_id_t tid);
  23
  24         HashTable<const char *, uint32_t, uintptr_t, 4, model_malloc, model_calloc, model_free> * getFuncMap() { return &func_map; }
  25         ModelVector<const char *> * getFuncMapRev() { return &func_map_rev; }
  26
  27         ModelVector<FuncNode *> * getFuncNodes() { return &func_nodes; }
  28         FuncNode * get_func_node(uint32_t func_id);
  29         FuncNode * get_curr_func_node(thread_id_t tid);
  30
  31         void update_write_history(void * location, uint64_t write_val);
  32         HashTable<void *, value_set_t *, uintptr_t, 4> * getWriteHistory() { return write_history; }
  33         void update_loc_rd_func_nodes_map(void * location, FuncNode * node);
  34         void update_loc_wr_func_nodes_map(void * location, FuncNode * node);
  35         SnapVector<FuncNode *> * getRdFuncNodes(void * location);
  36         SnapVector<FuncNode *> * getWrFuncNodes(void * location);
  37
  38         void add_waiting_write(ConcretePredicate * concrete);
  39         void remove_waiting_write(thread_id_t tid);
  40         void check_waiting_write(ModelAction * write_act);
  41         SnapVector<ConcretePredicate *> * getThrdWaitingWrite() { return thrd_waiting_write; }
  42
  43         SnapVector<inst_act_map_t *> * getThrdInstActMap(uint32_t func_id);
  44
  45         void set_new_exec_flag();
  46         void dump_func_node_graph();
  47         void print_func_node();
  48
  49         MEMALLOC
  50 private:
  51         uint32_t func_counter;
  52
  53         /* Map function names to integer ids */
  54         HashTable<const char *, uint32_t, uintptr_t, 4, model_malloc, model_calloc, model_free> func_map;
  55
  56         /* Map integer ids to function names */
  57         ModelVector<const char *> func_map_rev;
  58
  59         ModelVector<FuncNode *> func_nodes;
  60
  61         /* Map a location to a set of values that have been written to it */
  62         HashTable<void *, value_set_t *, uintptr_t, 4> * write_history;
  63
  64         /* Map a location to FuncNodes that may read from it */
  65         HashTable<void *, SnapVector<FuncNode *> *, uintptr_t, 0> * loc_rd_func_nodes_map;
  66
  67         /* Map a location to FuncNodes that may write to it */
  68         HashTable<void *, SnapVector<FuncNode *> *, uintptr_t, 0> * loc_wr_func_nodes_map;
  69
  70         HashTable<void *, SnapVector<ConcretePredicate *> *, uintptr_t, 0> * loc_waiting_writes_map;
  71         SnapVector<ConcretePredicate *> * thrd_waiting_write;
  72
  73         /* A run-time map from FuncInst to ModelAction per each FuncNode, per each thread.
  74          * Manipulated by FuncNode, and needed by NewFuzzer */
  75         HashTable<uint32_t, SnapVector<inst_act_map_t *> *, int, 0> * func_inst_act_maps;
  76
  77         bool skip_action(ModelAction * act, SnapList<ModelAction *> * curr_act_list);
  78 };
  79
  80 #endif  /* __HISTORY_H__ */