Add some methods for WaitObj
[c11tester.git] / history.cc
index bf40312c770df768ac3562f5b7fc8c668691b973..93abfbc0091806410df585607958c954c68ab069 100644 (file)
@@ -5,6 +5,7 @@
 #include "funcinst.h"
 #include "common.h"
 #include "concretepredicate.h"
+#include "waitobj.h"
 
 #include "model.h"
 #include "execution.h"
@@ -19,21 +20,23 @@ ModelHistory::ModelHistory() :
 {
        /* The following are snapshot data structures */
        write_history = new HashTable<void *, value_set_t *, uintptr_t, 4>();
-       loc_func_nodes_map = new HashTable<void *, SnapList<FuncNode *> *, uintptr_t, 0>();
-       loc_wr_func_nodes_map = new HashTable<void *, SnapList<FuncNode *> *, uintptr_t, 0>();
-       thrd_last_entered_func = new SnapVector<uint32_t>();
+       loc_rd_func_nodes_map = new HashTable<void *, SnapVector<FuncNode *> *, uintptr_t, 0>();
+       loc_wr_func_nodes_map = new HashTable<void *, SnapVector<FuncNode *> *, uintptr_t, 0>();
        loc_waiting_writes_map = new HashTable<void *, SnapVector<ConcretePredicate *> *, uintptr_t, 0>();
        thrd_waiting_write = new SnapVector<ConcretePredicate *>();
-       func_inst_act_maps = new HashTable<uint32_t, SnapVector<inst_act_map_t *> *, int, 0>();
+       thrd_wait_obj = new SnapVector<WaitObj *>();
+       func_inst_act_maps = new HashTable<uint32_t, SnapVector<inst_act_map_t *> *, int, 0>(128);
 }
 
 void ModelHistory::enter_function(const uint32_t func_id, thread_id_t tid)
 {
        //model_print("thread %d entering func %d\n", tid, func_id);
+       ModelExecution * execution = model->get_execution();
        uint id = id_to_int(tid);
-       SnapVector<func_id_list_t> * thrd_func_list = model->get_execution()->get_thrd_func_list();
+       SnapVector<func_id_list_t> * thrd_func_list = execution->get_thrd_func_list();
        SnapVector< SnapList<action_list_t *> *> *
-               thrd_func_act_lists = model->get_execution()->get_thrd_func_act_lists();
+               thrd_func_act_lists = execution->get_thrd_func_act_lists();
+       SnapVector<uint32_t> * thrd_last_entered_func = execution->get_thrd_last_entered_func();
 
        if ( thrd_func_list->size() <= id ) {
                uint oldsize = thrd_func_list->size();
@@ -41,18 +44,15 @@ void ModelHistory::enter_function(const uint32_t func_id, thread_id_t tid)
                thrd_func_act_lists->resize( id + 1 );
 
                for (uint i = oldsize; i < id + 1; i++) {
-                       new (&(*thrd_func_list)[i]) func_id_list_t();
                        // push 0 as a dummy function id to a void seg fault
+                       new (&(*thrd_func_list)[i]) func_id_list_t();
                        (*thrd_func_list)[i].push_back(0);
 
                        (*thrd_func_act_lists)[i] = new SnapList<action_list_t *>();
+                       thrd_last_entered_func->push_back(0);
                }
        }
 
-       while ( thrd_last_entered_func->size() <= id ) {
-               thrd_last_entered_func->push_back(0);   // 0 is a dummy function id
-       }
-
        SnapList<action_list_t *> * func_act_lists = (*thrd_func_act_lists)[id];
        func_act_lists->push_back( new action_list_t() );
 
@@ -77,10 +77,11 @@ void ModelHistory::enter_function(const uint32_t func_id, thread_id_t tid)
 /* @param func_id a non-zero value */
 void ModelHistory::exit_function(const uint32_t func_id, thread_id_t tid)
 {
+       ModelExecution * execution = model->get_execution();
        uint32_t id = id_to_int(tid);
-       SnapVector<func_id_list_t> * thrd_func_list = model->get_execution()->get_thrd_func_list();
+       SnapVector<func_id_list_t> * thrd_func_list = execution->get_thrd_func_list();
        SnapVector< SnapList<action_list_t *> *> *
-               thrd_func_act_lists = model->get_execution()->get_thrd_func_act_lists();
+               thrd_func_act_lists = execution->get_thrd_func_act_lists();
 
        SnapList<action_list_t *> * func_act_lists = (*thrd_func_act_lists)[id];
        uint32_t last_func_id = (*thrd_func_list)[id].back();
@@ -135,17 +136,18 @@ void ModelHistory::resize_func_nodes(uint32_t new_size)
 
 void ModelHistory::process_action(ModelAction *act, thread_id_t tid)
 {
-       /* return if thread i has not entered any function or has exited
+       ModelExecution * execution = model->get_execution();
+       /* Return if thread i has not entered any function or has exited
           from all functions */
-       SnapVector<func_id_list_t> * thrd_func_list = model->get_execution()->get_thrd_func_list();
+       SnapVector<func_id_list_t> * thrd_func_list = execution->get_thrd_func_list();
        SnapVector< SnapList<action_list_t *> *> *
-               thrd_func_act_lists = model->get_execution()->get_thrd_func_act_lists();
+               thrd_func_act_lists = execution->get_thrd_func_act_lists();
 
        uint32_t id = id_to_int(tid);
        if ( thrd_func_list->size() <= id )
                return;
 
-       /* get the function id that thread i is currently in */
+       /* Get the function id that thread i is currently in */
        uint32_t func_id = (*thrd_func_list)[id].back();
        SnapList<action_list_t *> * func_act_lists = (*thrd_func_act_lists)[id];
 
@@ -155,42 +157,28 @@ void ModelHistory::process_action(ModelAction *act, thread_id_t tid)
                update_write_history(location, value);
 
                /* Update FuncNodes that may read from this location */
-               SnapList<FuncNode *> * func_nodes = loc_func_nodes_map->get(location);
-               if (func_nodes != NULL) {
-                       sllnode<FuncNode *> * it = func_nodes->begin();
-                       for (; it != NULL; it = it->getNext()) {
-                               FuncNode * func_node = it->getVal();
-                               func_node->add_to_val_loc_map(value, location);
-                       }
+               SnapVector<FuncNode *> * func_node_list = getRdFuncNodes(location);
+               for (uint i = 0; i < func_node_list->size(); i++) {
+                       FuncNode * func_node = (*func_node_list)[i];
+                       func_node->add_to_val_loc_map(value, location);
                }
 
                check_waiting_write(act);
        }
 
-       /* the following does not care about actions without a position */
+       /* The following does not care about actions without a position */
        if (func_id == 0 || act->get_position() == NULL)
                return;
 
-       bool second_part_of_rmw = act->is_rmwc() || act->is_rmw();
-
        action_list_t * curr_act_list = func_act_lists->back();
        ASSERT(curr_act_list != NULL);
 
-       modelclock_t curr_seq_number = act->get_seq_number();
-       /* Skip actions that are second part of a read modify write or actions with the same sequence number */
-       if (curr_act_list->size() != 0) {
-               ModelAction * last_act = curr_act_list->back();
-               if (second_part_of_rmw || last_act->get_seq_number() == curr_seq_number)
-                       return;
-       }
-
-       /* skip actions that are paused by fuzzer (sequence number is 0) */
-       if (curr_seq_number == 0)
+       if (skip_action(act, curr_act_list))
                return;
 
        FuncNode * func_node = func_nodes[func_id];
 
-       /* add to curr_inst_list */
+       /* Add to curr_inst_list */
        curr_act_list->push_back(act);
        func_node->add_inst(act);
 
@@ -198,7 +186,7 @@ void ModelHistory::process_action(ModelAction *act, thread_id_t tid)
                func_node->update_inst_act_map(tid, act);
 
                // Update predicate tree position
-               Fuzzer * fuzzer = model->get_execution()->getFuzzer();
+               Fuzzer * fuzzer = execution->getFuzzer();
                Predicate * selected_branch = fuzzer->get_selected_child_branch(tid);
                func_node->set_predicate_tree_position(tid, selected_branch);
        }
@@ -243,26 +231,38 @@ void ModelHistory::update_write_history(void * location, uint64_t write_val)
        write_set->add(write_val);
 }
 
-void ModelHistory::update_loc_func_nodes_map(void * location, FuncNode * node)
+void ModelHistory::update_loc_rd_func_nodes_map(void * location, FuncNode * node)
+{
+       SnapVector<FuncNode *> * func_node_list = getRdFuncNodes(location);
+       func_node_list->push_back(node);
+}
+
+void ModelHistory::update_loc_wr_func_nodes_map(void * location, FuncNode * node)
+{
+       SnapVector<FuncNode *> * func_node_list = getWrFuncNodes(location);
+       func_node_list->push_back(node);
+}
+
+SnapVector<FuncNode *> * ModelHistory::getRdFuncNodes(void * location)
 {
-       SnapList<FuncNode *> * func_node_list = loc_func_nodes_map->get(location);
+       SnapVector<FuncNode *> * func_node_list = loc_rd_func_nodes_map->get(location);
        if (func_node_list == NULL) {
-               func_node_list = new SnapList<FuncNode *>();
-               loc_func_nodes_map->put(location, func_node_list);
+               func_node_list = new SnapVector<FuncNode *>();
+               loc_rd_func_nodes_map->put(location, func_node_list);
        }
 
-       func_node_list->push_back(node);
+       return func_node_list;
 }
 
-void ModelHistory::update_loc_wr_func_nodes_map(void * location, FuncNode * node)
+SnapVector<FuncNode *> * ModelHistory::getWrFuncNodes(void * location)
 {
-       SnapList<FuncNode *> * func_node_list = loc_wr_func_nodes_map->get(location);
+       SnapVector<FuncNode *> * func_node_list = loc_wr_func_nodes_map->get(location);
        if (func_node_list == NULL) {
-               func_node_list = new SnapList<FuncNode *>();
-               loc_func_nodes_map->put(location, func_node_list);
+               func_node_list = new SnapVector<FuncNode *>();
+               loc_wr_func_nodes_map->put(location, func_node_list);
        }
 
-       func_node_list->push_back(node);
+       return func_node_list;
 }
 
 /* When a thread is paused by Fuzzer, keep track of the condition it is waiting for */
@@ -292,6 +292,8 @@ void ModelHistory::remove_waiting_write(thread_id_t tid)
        void * location = concrete->get_location();
        SnapVector<ConcretePredicate *> * concrete_preds = loc_waiting_writes_map->get(location);
 
+       /* Linear search should be fine because presumably not many ConcretePredicates
+        * are at the same memory location */
        for (uint i = 0; i < concrete_preds->size(); i++) {
                ConcretePredicate * current = (*concrete_preds)[i];
                if (concrete == current) {
@@ -362,6 +364,42 @@ void ModelHistory::check_waiting_write(ModelAction * write_act)
        }
 }
 
+WaitObj * ModelHistory::getWaitObj(thread_id_t tid)
+{
+       int thread_id = id_to_int(tid);
+       int old_size = thrd_wait_obj->size();
+       if (old_size <= thread_id) {
+               thrd_wait_obj->resize(thread_id + 1);
+               for (int i = old_size; i < thread_id + 1; i++) {
+                       (*thrd_wait_obj)[i] = new WaitObj( int_to_id(i) );
+               }
+       }
+
+       return (*thrd_wait_obj)[thread_id];
+}
+
+void ModelHistory::add_waiting_thread(thread_id_t self_id,
+               thread_id_t waiting_for_id, int dist)
+{
+       WaitObj * self_wait_obj = getWaitObj(self_id);
+       self_wait_obj->add_waiting_for(waiting_for_id, dist);
+
+       /* Update waited-by relation */
+       WaitObj * other_wait_obj = getWaitObj(waiting_for_id);
+       other_wait_obj->add_waited_by(self_id);
+}
+
+void ModelHistory::remove_waiting_thread(thread_id_t self_id, thread_id_t waiting_for_id)
+{
+       WaitObj * self_wait_obj = getWaitObj(self_id);
+       self_wait_obj->remove_waiting_for(waiting_for_id);
+
+       /* Update waited-by relation */
+       WaitObj * other_wait_obj = getWaitObj(waiting_for_id);
+       other_wait_obj->remove_waited_by(self_id);
+}
+
+
 SnapVector<inst_act_map_t *> * ModelHistory::getThrdInstActMap(uint32_t func_id)
 {
        ASSERT(func_id != 0);
@@ -375,6 +413,29 @@ SnapVector<inst_act_map_t *> * ModelHistory::getThrdInstActMap(uint32_t func_id)
        return maps;
 }
 
+bool ModelHistory::skip_action(ModelAction * act, SnapList<ModelAction *> * curr_act_list)
+{
+       bool second_part_of_rmw = act->is_rmwc() || act->is_rmw();
+       modelclock_t curr_seq_number = act->get_seq_number();
+
+       /* Skip actions that are second part of a read modify write */
+       if (second_part_of_rmw)
+               return true;
+
+       /* Skip actions with the same sequence number */
+       if (curr_act_list->size() != 0) {
+               ModelAction * last_act = curr_act_list->back();
+               if (last_act->get_seq_number() == curr_seq_number)
+                       return true;
+       }
+
+       /* Skip actions that are paused by fuzzer (sequence number is 0) */
+       if (curr_seq_number == 0)
+               return true;
+
+       return false;
+}
+
 /* Reallocate some snapshotted memories when new executions start */
 void ModelHistory::set_new_exec_flag()
 {
@@ -417,3 +478,19 @@ void ModelHistory::print_func_node()
                }
        }
 }
+
+void ModelHistory::print_waiting_threads()
+{
+       ModelExecution * execution = model->get_execution();
+       for (unsigned int i = 0; i < execution->get_num_threads();i++) {
+               thread_id_t tid = int_to_id(i);
+               WaitObj * wait_obj = getWaitObj(tid);
+               wait_obj->print_waiting_for();
+       }
+
+       for (unsigned int i = 0; i < execution->get_num_threads();i++) {
+               thread_id_t tid = int_to_id(i);
+               WaitObj * wait_obj = getWaitObj(tid);
+               wait_obj->print_waited_by();
+       }
+}