X-Git-Url: http://info.iut-bm.univ-fcomte.fr/pub/gitweb/simgrid.git/blobdiff_plain/12cc56f2c5f8993698a07b19c37f845b175b420a..9692d43fa911bdc2d6d0263a3cb3e22d5e3167fe:/src/mc/SafetyChecker.hpp

diff --git a/src/mc/SafetyChecker.hpp b/src/mc/SafetyChecker.hpp
index d7fe099f5d..b2dd2696ae 100644
--- a/src/mc/SafetyChecker.hpp
+++ b/src/mc/SafetyChecker.hpp
@@ -7,21 +7,39 @@
 #ifndef SIMGRID_MC_SAFETY_CHECKER_HPP
 #define SIMGRID_MC_SAFETY_CHECKER_HPP
 
+#include <list>
+#include <memory>
+#include <string>
+#include <vector>
+
 #include "src/mc/mc_forward.hpp"
 #include "src/mc/Checker.hpp"
+#include "src/mc/VisitedState.hpp"
 
 namespace simgrid {
 namespace mc {
 
-class SafetyChecker : public Checker {
+class XBT_PRIVATE SafetyChecker : public Checker {
   simgrid::mc::ReductionMode reductionMode_ = simgrid::mc::ReductionMode::unset;
 public:
   SafetyChecker(Session& session);
   ~SafetyChecker();
   int run() override;
+  RecordTrace getRecordTrace() override;
+  std::vector<std::string> getTextualTrace() override;
+  void logState() override;
 private:
   // Temp
   void init();
+  bool checkNonTermination(simgrid::mc::State* current_state);
+  int backtrack();
+  void restoreState();
+private:
+  /** Stack representing the position in the exploration graph */
+  std::list<std::unique_ptr<simgrid::mc::State>> stack_;
+  simgrid::mc::VisitedStates visitedStates_;
+  std::unique_ptr<simgrid::mc::VisitedState> visitedState_;
+  unsigned long expandedStatesCount_ = 0;
 };
 
 }