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
   9 typedef ModelList<FuncInst *> func_inst_list_mt;
  10 typedef HashTable<void *, uint64_t, uintptr_t, 4, model_malloc, model_calloc, model_free> read_map_t;
  11 typedef HashSet<uint64_t, uint64_t, 0, model_malloc, model_calloc, model_free> write_set_t;
  12
  13 class FuncNode {
  14 public:
  15         FuncNode();
  16         ~FuncNode();
  17
  18         uint32_t get_func_id() { return func_id; }
  19         const char * get_func_name() { return func_name; }
  20         void set_func_id(uint32_t id) { func_id = id; }
  21         void set_func_name(const char * name) { func_name = name; }
  22
  23         FuncInst * get_or_add_action(ModelAction *act);
  24
  25         HashTable<const char *, FuncInst *, uintptr_t, 4, model_malloc, model_calloc, model_free> * getFuncInstMap() { return &func_inst_map; }
  26         func_inst_list_mt * get_inst_list() { return &inst_list; }
  27         func_inst_list_mt * get_entry_insts() { return &entry_insts; }
  28         void add_entry_inst(FuncInst * inst);
  29         void link_insts(func_inst_list_t * inst_list);
  30
  31         void store_read(ModelAction * act, uint32_t tid);
  32         uint64_t query_last_read(void * location, uint32_t tid);
  33         void clear_read_map(uint32_t tid);
  34
  35         void add_to_write_history(void * location, uint64_t write_val);
  36
  37         /* TODO: generate EQUALITY or NULLITY predicate based on write_history in history.cc */
  38         void generate_predicate(FuncInst * func_inst);
  39
  40         void print_last_read(uint32_t tid);
  41         void print_write();
  42
  43         MEMALLOC
  44 private:
  45         uint32_t func_id;
  46         const char * func_name;
  47
  48         /* Use source line number as the key of hashtable, to check if
  49          * atomic operation with this line number has been added or not
  50          */
  51         HashTable<const char *, FuncInst *, uintptr_t, 4, model_malloc, model_calloc, model_free> func_inst_map;
  52
  53         /* list of all atomic actions in this function */
  54         func_inst_list_mt inst_list;
  55
  56         /* possible entry atomic actions in this function */
  57         func_inst_list_mt entry_insts;
  58
  59         /* Store the values read by atomic read actions per memory location for each thread */
  60         ModelVector<read_map_t *> thrd_read_map;
  61
  62         HashTable<void *, write_set_t *, uintptr_t, 4, model_malloc, model_calloc, model_free> write_history;
  63         HashSet<void *, uintptr_t, 4, model_malloc, model_calloc, model_free> write_locations;
  64 };
  65
  66 #endif /* __FUNCNODE_H__ */