funcnode.h

   1 #ifndef __FUNCNODE_H__
   2 #define __FUNCNODE_H__
   3
   4 #include "action.h"
   5 #include "funcinst.h"
   6 #include "hashtable.h"
   7 #include "hashset.h"
   8 #include "predicate.h"
   9 #include "history.h"
  10
  11 typedef ModelList<FuncInst *> func_inst_list_mt;
  12 typedef HashTable<void *, uint64_t, uintptr_t, 4> read_map_t;
  13
  14 class FuncNode {
  15 public:
  16         FuncNode(ModelHistory * history);
  17         ~FuncNode();
  18
  19         uint32_t get_func_id() { return func_id; }
  20         const char * get_func_name() { return func_name; }
  21         void set_func_id(uint32_t id) { func_id = id; }
  22         void set_func_name(const char * name) { func_name = name; }
  23         void set_new_exec_flag();
  24
  25         void add_inst(ModelAction *act);
  26         FuncInst * get_inst(ModelAction *act);
  27
  28         HashTable<const char *, FuncInst *, uintptr_t, 4, model_malloc, model_calloc, model_free> * getFuncInstMap() { return &func_inst_map; }
  29         func_inst_list_mt * get_inst_list() { return &inst_list; }
  30         func_inst_list_mt * get_entry_insts() { return &entry_insts; }
  31         void add_entry_inst(FuncInst * inst);
  32
  33         void update_tree(action_list_t * act_list);
  34         void update_inst_tree(func_inst_list_t * inst_list);
  35
  36         void store_read(ModelAction * act, uint32_t tid);
  37         uint64_t query_last_read(void * location, uint32_t tid);
  38         void clear_read_map(uint32_t tid);
  39
  40         /* TODO: generate EQUALITY or NULLITY predicate based on write_history in history.cc */
  41         void update_predicate_tree(action_list_t * act_list);
  42         void deep_update(Predicate * pred);
  43         bool follow_branch(Predicate ** curr_pred, FuncInst * next_inst, ModelAction * next_act, HashTable<FuncInst *, ModelAction *, uintptr_t, 0>* inst_act_map, SnapVector<Predicate *> * unset_predicates);
  44
  45         void incr_exit_count() { exit_count++; }
  46         uint32_t get_exit_count() { return exit_count; }
  47
  48         ModelList<action_list_t *> * get_action_list_buffer() { return &action_list_buffer; }
  49
  50         void print_predicate_tree();
  51         void print_last_read(uint32_t tid);
  52
  53         MEMALLOC
  54 private:
  55         uint32_t func_id;
  56         const char * func_name;
  57         ModelHistory * history;
  58         bool predicate_tree_initialized;
  59         Predicate * predicate_tree_entry;       // a dummy node whose children are the real entries
  60
  61         uint32_t exit_count;
  62
  63         /* Use source line number as the key of hashtable, to check if
  64          * atomic operation with this line number has been added or not
  65          */
  66         HashTable<const char *, FuncInst *, uintptr_t, 4, model_malloc, model_calloc, model_free> func_inst_map;
  67
  68         /* list of all atomic actions in this function */
  69         func_inst_list_mt inst_list;
  70
  71         /* possible entry atomic actions in this function */
  72         func_inst_list_mt entry_insts;
  73
  74         /* Store the values read by atomic read actions per memory location for each thread */
  75         ModelVector<read_map_t *> thrd_read_map;
  76
  77         ModelList<action_list_t *> action_list_buffer;
  78 };
  79
  80 #endif /* __FUNCNODE_H__ */