Get rid of main
[c11tester.git] / funcnode.h
1 #ifndef __FUNCNODE_H__
2 #define __FUNCNODE_H__
3
4 #include "hashset.h"
5 #include "hashfunction.h"
6 #include "classlist.h"
7 #include "threads-model.h"
8
9 #define MAX_DIST 10
10
11 typedef ModelList<FuncInst *> func_inst_list_mt;
12 typedef enum edge_type {
13         IN_EDGE, OUT_EDGE, BI_EDGE
14 } edge_type_t;
15
16 class FuncNode {
17 public:
18         FuncNode(ModelHistory * history);
19         ~FuncNode();
20
21         uint32_t get_func_id() { return func_id; }
22         const char * get_func_name() { return func_name; }
23         void set_func_id(uint32_t id) { func_id = id; }
24         void set_func_name(const char * name) { func_name = name; }
25         void set_new_exec_flag();
26
27         void add_inst(ModelAction *act);
28         FuncInst * create_new_inst(ModelAction *act);
29         FuncInst * get_inst(ModelAction *act);
30
31         HashTable<const char *, FuncInst *, uintptr_t, 4, model_malloc, model_calloc, model_free> * getFuncInstMap() { return &func_inst_map; }
32         func_inst_list_mt * get_inst_list() { return &inst_list; }
33         func_inst_list_mt * get_entry_insts() { return &entry_insts; }
34         void add_entry_inst(FuncInst * inst);
35
36         void update_tree(action_list_t * act_list);
37         void update_inst_tree(func_inst_list_t * inst_list);
38         void update_predicate_tree(action_list_t * act_list);
39         bool follow_branch(Predicate ** curr_pred, FuncInst * next_inst, ModelAction * next_act, SnapVector<Predicate *> * unset_predicates);
40
41         void incr_exit_count() { exit_count++; }
42         uint32_t get_exit_count() { return exit_count; }
43
44         SnapList<action_list_t *> * get_action_list_buffer() { return action_list_buffer; }
45
46         void add_to_val_loc_map(uint64_t val, void * loc);
47         void add_to_val_loc_map(value_set_t * values, void * loc);
48         void update_loc_may_equal_map(void * new_loc, loc_set_t * old_locations);
49
50         void init_predicate_tree_position(thread_id_t tid);
51         void set_predicate_tree_position(thread_id_t tid, Predicate * pred);
52         Predicate * get_predicate_tree_position(thread_id_t tid);
53
54         void init_inst_act_map(thread_id_t tid);
55         void reset_inst_act_map(thread_id_t tid);
56         void update_inst_act_map(thread_id_t tid, ModelAction * read_act);
57         inst_act_map_t * get_inst_act_map(thread_id_t tid);
58
59         void add_out_edge(FuncNode * other);
60         ModelList<FuncNode *> * get_out_edges() { return &out_edges; }
61         int compute_distance(FuncNode * target, int max_step = MAX_DIST);
62
63         void print_predicate_tree();
64         void print_val_loc_map();
65         void print_last_read(thread_id_t tid);
66
67         MEMALLOC
68 private:
69         uint32_t func_id;
70         const char * func_name;
71         ModelHistory * history;
72         Predicate * predicate_tree_entry;       // A dummy node whose children are the real entries
73         Predicate * predicate_tree_exit;        // A dummy node
74
75
76         uint32_t exit_count;
77         uint32_t marker;
78
79         void incr_marker() { marker++; }
80
81         /* Use source line number as the key of hashtable, to check if
82          * atomic operation with this line number has been added or not
83          */
84         HashTable<const char *, FuncInst *, uintptr_t, 4, model_malloc, model_calloc, model_free> func_inst_map;
85
86         /* List of all atomic actions in this function */
87         func_inst_list_mt inst_list;
88
89         /* Possible entry atomic actions in this function */
90         func_inst_list_mt entry_insts;
91
92         void infer_predicates(FuncInst * next_inst, ModelAction * next_act, HashTable<void *, ModelAction *, uintptr_t, 0> * loc_act_map, SnapVector<struct half_pred_expr *> * half_pred_expressions);
93         void generate_predicates(Predicate ** curr_pred, FuncInst * next_inst, SnapVector<struct half_pred_expr *> * half_pred_expressions);
94         bool amend_predicate_expr(Predicate ** curr_pred, FuncInst * next_inst, ModelAction * next_act);
95
96         /* Store action_lists when calls to update_tree are deferred */
97         SnapList<action_list_t *> * action_list_buffer;
98
99         /* Set of locations read by this FuncNode */
100         loc_set_t * read_locations;
101
102         /* Set of locations written to by this FuncNode */
103         loc_set_t * write_locations;
104
105         /* Keeps track of locations that have the same values written to */
106         HashTable<uint64_t, loc_set_t *, uint64_t, 0, snapshot_malloc, snapshot_calloc, snapshot_free, int64_hash> * val_loc_map;
107
108         /* Keeps track of locations that may share the same value as key, deduced from val_loc_map */
109         HashTable<void *, loc_set_t *, uintptr_t, 0> * loc_may_equal_map;
110
111         // value_set_t * values_may_read_from;
112
113         /* Run-time position in the predicate tree for each thread */
114         ModelVector<Predicate *> predicate_tree_position;
115
116         /* Store the relation between this FuncNode and other FuncNodes */
117         HashTable<FuncNode *, edge_type_t, uintptr_t, 0, model_malloc, model_calloc, model_free> edge_table;
118
119         /* FuncNodes that follow this node */
120         ModelList<FuncNode *> out_edges;
121 };
122
123 #endif  /* __FUNCNODE_H__ */