model.h

   1 #ifndef __MODEL_H__
   2 #define __MODEL_H__
   3
   4 #include <list>
   5 #include <map>
   6 #include <cstddef>
   7 #include <ucontext.h>
   8
   9 #include "schedule.h"
  10 #include "libthreads.h"
  11 #include "libatomic.h"
  12 #include "threads.h"
  13 #include "action.h"
  14
  15 /* Forward declaration */
  16 class TreeNode;
  17 class Backtrack;
  18
  19 class ModelChecker {
  20 public:
  21         ModelChecker();
  22         ~ModelChecker();
  23         class Scheduler *scheduler;
  24
  25         void set_system_context(ucontext_t *ctxt) { system_context = ctxt; }
  26         ucontext_t * get_system_context(void) { return system_context; }
  27
  28         void set_current_action(ModelAction *act) { current_action = act; }
  29         void check_current_action(void);
  30         void print_summary(void);
  31         Thread * schedule_next_thread();
  32
  33         int add_thread(Thread *t);
  34         void remove_thread(Thread *t);
  35         Thread * get_thread(thread_id_t tid) { return thread_map[id_to_int(tid)]; }
  36
  37         thread_id_t get_next_id();
  38         int get_next_seq_num();
  39
  40         int switch_to_master(ModelAction *act);
  41
  42         bool next_execution();
  43 private:
  44         int next_thread_id;
  45         int used_sequence_numbers;
  46         int num_executions;
  47
  48         ModelAction * get_last_conflict(ModelAction *act);
  49         void set_backtracking(ModelAction *act);
  50         thread_id_t get_next_replay_thread();
  51         Backtrack * get_next_backtrack();
  52         void reset_to_initial_state();
  53
  54         void print_list(action_list_t *list);
  55
  56         class ModelAction *current_action;
  57         Backtrack *exploring;
  58         thread_id_t nextThread;
  59
  60         ucontext_t *system_context;
  61         action_list_t *action_trace;
  62         std::map<int, class Thread *> thread_map;
  63         class TreeNode *rootNode, *currentNode;
  64         std::list<class Backtrack *> backtrack_list;
  65 };
  66
  67 extern ModelChecker *model;
  68
  69 #endif /* __MODEL_H__ */