src/mc/Session.cpp

   1 /* Copyright (c) 2015-2016. The SimGrid Team.
   2  * All rights reserved.                                                     */
   3
   4 /* This program is free software; you can redistribute it and/or modify it
   5  * under the terms of the license (GNU LGPL) which comes with this package. */
   6
   7 #include <fcntl.h>
   8 #include <signal.h>
   9
  10 #include <functional>
  11
  12 #include "xbt/log.h"
  13 #include "xbt/system_error.hpp"
  14 #include <mc/mc.h>
  15 #include <simgrid/modelchecker.h>
  16 #include <simgrid/sg_config.h>
  17
  18 #include "src/mc/Session.hpp"
  19 #include "src/mc/mc_state.h"
  20 #include "src/mc/mc_private.h"
  21 #include "src/mc/checker/Checker.hpp"
  22
  23 #include "src/smpi/private.hpp"
  24
  25 XBT_LOG_NEW_DEFAULT_SUBCATEGORY(mc_Session, mc, "Model-checker session");
  26
  27 namespace simgrid {
  28 namespace mc {
  29
  30 static void setup_child_environment(int socket)
  31 {
  32 #ifdef __linux__
  33   // Make sure we do not outlive our parent:
  34   sigset_t mask;
  35   sigemptyset (&mask);
  36   if (sigprocmask(SIG_SETMASK, &mask, nullptr) < 0)
  37     throw simgrid::xbt::errno_error("Could not unblock signals");
  38   if (prctl(PR_SET_PDEATHSIG, SIGHUP) != 0)
  39     throw simgrid::xbt::errno_error("Could not PR_SET_PDEATHSIG");
  40 #endif
  41
  42   int res;
  43
  44   // Remove CLOEXEC in order to pass the socket to the exec-ed program:
  45   int fdflags = fcntl(socket, F_GETFD, 0);
  46   if (fdflags == -1 || fcntl(socket, F_SETFD, fdflags & ~FD_CLOEXEC) == -1)
  47     throw simgrid::xbt::errno_error("Could not remove CLOEXEC for socket");
  48
  49   // Set environment:
  50   setenv(MC_ENV_VARIABLE, "1", 1);
  51
  52   // Disable lazy relocation in the model-checked process.
  53   // We don't want the model-checked process to modify its .got.plt during
  54   // snapshot.
  55   setenv("LC_BIND_NOW", "1", 1);
  56
  57   char buffer[64];
  58   res = std::snprintf(buffer, sizeof(buffer), "%i", socket);
  59   if ((size_t) res >= sizeof(buffer) || res == -1)
  60     std::abort();
  61   setenv(MC_ENV_SOCKET_FD, buffer, 1);
  62 }
  63
  64 /** Execute some code in a forked process */
  65 template<class F>
  66 static inline
  67 pid_t do_fork(F code)
  68 {
  69   pid_t pid = fork();
  70   if (pid < 0)
  71     throw simgrid::xbt::errno_error("Could not fork model-checked process");
  72   if (pid != 0)
  73     return pid;
  74
  75   // Child-process:
  76   try {
  77     code();
  78     _exit(EXIT_SUCCESS);
  79   }
  80   catch(...) {
  81     // The callback should catch exceptions:
  82     std::terminate();
  83   }
  84 }
  85
  86 Session::Session(pid_t pid, int socket)
  87 {
  88   std::unique_ptr<simgrid::mc::RemoteClient> process(new simgrid::mc::RemoteClient(pid, socket));
  89   // TODO, automatic detection of the config from the process
  90   process->privatized(smpi_privatize_global_variables != SMPI_PRIVATIZE_NONE);
  91   modelChecker_ = std::unique_ptr<ModelChecker>(
  92     new simgrid::mc::ModelChecker(std::move(process)));
  93   xbt_assert(mc_model_checker == nullptr);
  94   mc_model_checker = modelChecker_.get();
  95   mc_model_checker->start();
  96 }
  97
  98 Session::~Session()
  99 {
 100   this->close();
 101 }
 102
 103 void Session::initialize()
 104 {
 105   xbt_assert(initialSnapshot_ == nullptr);
 106   mc_model_checker->wait_for_requests();
 107   initialSnapshot_ = simgrid::mc::take_snapshot(0);
 108 }
 109
 110 void Session::execute(Transition const& transition)
 111 {
 112   modelChecker_->handle_simcall(transition);
 113   modelChecker_->wait_for_requests();
 114 }
 115
 116 void Session::restoreInitialState()
 117 {
 118   simgrid::mc::restore_snapshot(this->initialSnapshot_);
 119 }
 120
 121 void Session::logState()
 122 {
 123   mc_model_checker->getChecker()->logState();
 124
 125   if ((_sg_mc_dot_output_file != nullptr) && (_sg_mc_dot_output_file[0] != '\0')) {
 126     fprintf(dot_output, "}\n");
 127     fclose(dot_output);
 128   }
 129   if (getenv("SIMGRID_MC_SYSTEM_STATISTICS")){
 130     int ret=system("free");
 131     if(ret!=0)XBT_WARN("system call did not return 0, but %d",ret);
 132   }
 133 }
 134
 135 // static
 136 Session* Session::fork(std::function<void()> code)
 137 {
 138   // Create a AF_LOCAL socketpair used for exchanging messages
 139   // bewteen the model-checker process (ourselves) and the model-checked
 140   // process:
 141   int res;
 142   int sockets[2];
 143   res = socketpair(AF_LOCAL, SOCK_DGRAM | SOCK_CLOEXEC, 0, sockets);
 144   if (res == -1)
 145     throw simgrid::xbt::errno_error("Could not create socketpair");
 146
 147   pid_t pid = do_fork([sockets, &code] {
 148     ::close(sockets[1]);
 149     setup_child_environment(sockets[0]);
 150     code();
 151     xbt_die("The model-checked process failed to exec()");
 152   });
 153
 154   // Parent (model-checker):
 155   ::close(sockets[0]);
 156
 157   return new Session(pid, sockets[1]);
 158 }
 159
 160 // static
 161 Session* Session::spawnv(const char *path, char *const argv[])
 162 {
 163   return Session::fork([path, argv] {
 164     execv(path, argv);
 165   });
 166 }
 167
 168 // static
 169 Session* Session::spawnvp(const char *file, char *const argv[])
 170 {
 171   return Session::fork([file, argv] {
 172     execvp(file, argv);
 173   });
 174 }
 175
 176 void Session::close()
 177 {
 178   initialSnapshot_ = nullptr;
 179   if (modelChecker_) {
 180     modelChecker_->shutdown();
 181     modelChecker_ = nullptr;
 182     mc_model_checker = nullptr;
 183   }
 184 }
 185
 186 simgrid::mc::Session* session;
 187
 188 }
 189 }