Bug fix

[c11tester.git] / model.h

diff --git a/model.h b/model.h
index b974419fe851e3042088ee94a322d99201e1cd46..c5a5aa78647deb2bfdb3ae7cf269b972ae5f89f0 100644 (file)
--- a/model.h
+++ b/model.h
@@ -16,6 +16,7 @@
 #include "context.h"
 #include "params.h"
 #include "classlist.h"
+#include "snapshot-interface.h"
 
 typedef SnapList<ModelAction *> action_list_t;
 
@@ -56,14 +57,13 @@ public:
        void switch_from_master(Thread *thread);
        uint64_t switch_to_master(ModelAction *act);
 
-       bool assert_bug(const char *msg, ...);
+       void assert_bug(const char *msg, ...);
 
        void assert_user_bug(const char *msg);
 
        model_params params;
        void add_trace_analysis(TraceAnalysis *a) {     trace_analyses.push_back(a); }
        void set_inspect_plugin(TraceAnalysis *a) {     inspect_plugin=a;       }
-       void startMainThread();
        void startChecker();
        Thread * getInitThread() {return init_thread;}
        Scheduler * getScheduler() {return scheduler;}
@@ -72,6 +72,9 @@ private:
        /** Flag indicates whether to restart the model checker. */
        bool restart_flag;
 
+       /** Snapshot id we return to restart. */
+       snapshot_id snapshot;
+
        /** The scheduler to use: tracks the running/ready Threads */
        Scheduler * const scheduler;
        ModelHistory * history;
@@ -103,9 +106,10 @@ private:
        void print_bugs() const;
        void print_execution(bool printbugs) const;
        void print_stats() const;
-
-       friend void user_main_wrapper();
 };
 
 extern ModelChecker *model;
+void parse_options(struct model_params *params);
+void install_trace_analyses(ModelExecution *execution);
+
 #endif /* __MODEL_H__ */