Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
Fix format string.
[simgrid.git] / src / smpi / internals / smpi_global.cpp
1 /* Copyright (c) 2007-2018. The SimGrid Team. All rights reserved.          */
2
3 /* This program is free software; you can redistribute it and/or modify it
4  * under the terms of the license (GNU LGPL) which comes with this package. */
5
6 #include "mc/mc.h"
7 #include "simgrid/s4u/Engine.hpp"
8 #include "smpi_coll.hpp"
9 #include "smpi_f2c.hpp"
10 #include "smpi_host.hpp"
11 #include "src/kernel/activity/CommImpl.hpp"
12 #include "src/simix/smx_private.hpp"
13 #include "src/smpi/include/smpi_actor.hpp"
14 #include "xbt/config.hpp"
15
16 #include <algorithm>
17 #include <cfloat> /* DBL_MAX */
18 #include <cstdint> /* intmax_t */
19 #include <dlfcn.h>
20 #include <fcntl.h>
21 #include <fstream>
22
23 #if not defined(__APPLE__)
24 #include <link.h>
25 #endif
26
27 #if defined(__APPLE__)
28 # include <AvailabilityMacros.h>
29 # ifndef MAC_OS_X_VERSION_10_12
30 #   define MAC_OS_X_VERSION_10_12 101200
31 # endif
32 # define HAVE_WORKING_MMAP (MAC_OS_X_VERSION_MIN_REQUIRED >= MAC_OS_X_VERSION_10_12)
33 #elif defined(__FreeBSD__)
34 # define HAVE_WORKING_MMAP 0
35 #else
36 # define HAVE_WORKING_MMAP 1
37 #endif
38
39 #if HAVE_SENDFILE
40 #include <sys/sendfile.h>
41 #endif
42
43 XBT_LOG_NEW_DEFAULT_SUBCATEGORY(smpi_kernel, smpi, "Logging specific to SMPI (kernel)");
44 #include <boost/tokenizer.hpp>
45 #include <boost/algorithm/string.hpp> /* trim_right / trim_left */
46
47 #if SMPI_IFORT
48   extern "C" void for_rtl_init_ (int *, char **);
49   extern "C" void for_rtl_finish_ ();
50 #elif SMPI_FLANG
51   extern "C" void __io_set_argc(int);
52   extern "C" void __io_set_argv(char **);
53 #elif SMPI_GFORTRAN
54   extern "C" void _gfortran_set_args(int, char **);
55 #endif
56
57 /* RTLD_DEEPBIND is a bad idea of GNU ld that obviously does not exist on other platforms
58  * See https://www.akkadia.org/drepper/dsohowto.pdf
59  * and https://lists.freebsd.org/pipermail/freebsd-current/2016-March/060284.html
60 */
61 #if !defined(RTLD_DEEPBIND) || HAVE_SANITIZER_ADDRESS || HAVE_SANITIZER_THREAD
62 #define WANT_RTLD_DEEPBIND 0
63 #else
64 #define WANT_RTLD_DEEPBIND RTLD_DEEPBIND
65 #endif
66
67 #if HAVE_PAPI
68 #include "papi.h"
69 std::string papi_default_config_name = "default";
70 std::map</* computation unit name */ std::string, papi_process_data> units2papi_setup;
71 #endif
72
73 using simgrid::s4u::Actor;
74 using simgrid::s4u::ActorPtr;
75 std::unordered_map<std::string, double> location2speedup;
76
77 static std::map</*process_id*/ ActorPtr, simgrid::smpi::ActorExt*> process_data;
78 int process_count = 0;
79 static int smpi_exit_status = 0;
80 int smpi_universe_size = 0;
81 extern double smpi_total_benched_time;
82 xbt_os_timer_t global_timer;
83 static std::vector<std::string> privatize_libs_paths;
84 /**
85  * Setting MPI_COMM_WORLD to MPI_COMM_UNINITIALIZED (it's a variable)
86  * is important because the implementation of MPI_Comm checks
87  * "this == MPI_COMM_UNINITIALIZED"? If yes, it uses smpi_process()->comm_world()
88  * instead of "this".
89  * This is basically how we only have one global variable but all processes have
90  * different communicators (the one their SMPI instance uses).
91  *
92  * See smpi_comm.cpp and the functions therein for details.
93  */
94 MPI_Comm MPI_COMM_WORLD = MPI_COMM_UNINITIALIZED;
95 MPI_Errhandler *MPI_ERRORS_RETURN = nullptr;
96 MPI_Errhandler *MPI_ERRORS_ARE_FATAL = nullptr;
97 MPI_Errhandler *MPI_ERRHANDLER_NULL = nullptr;
98 // No instance gets manually created; check also the smpirun.in script as
99 // this default name is used there as well (when the <actor> tag is generated).
100 static const std::string smpi_default_instance_name("smpirun");
101 static simgrid::config::Flag<double> smpi_init_sleep(
102   "smpi/init", "Time to inject inside a call to MPI_Init", 0.0);
103
104 void (*smpi_comm_copy_data_callback) (smx_activity_t, void*, size_t) = &smpi_comm_copy_buffer_callback;
105
106 int smpi_process_count()
107 {
108   return process_count;
109 }
110
111 simgrid::smpi::ActorExt* smpi_process()
112 {
113   ActorPtr me = Actor::self();
114
115   if (me == nullptr) // This happens sometimes (eg, when linking against NS3 because it pulls openMPI...)
116     return nullptr;
117
118   return process_data.at(me);
119 }
120
121 simgrid::smpi::ActorExt* smpi_process_remote(ActorPtr actor)
122 {
123   return process_data.at(actor);
124 }
125
126 MPI_Comm smpi_process_comm_self(){
127   return smpi_process()->comm_self();
128 }
129
130 void smpi_process_init(int *argc, char ***argv){
131   simgrid::smpi::ActorExt::init(argc, argv);
132 }
133
134 void * smpi_process_get_user_data(){
135   return Actor::self()->get_impl()->get_user_data();
136 }
137
138 void smpi_process_set_user_data(void *data){
139   Actor::self()->get_impl()->set_user_data(data);
140 }
141
142
143 int smpi_global_size()
144 {
145   char *value = getenv("SMPI_GLOBAL_SIZE");
146   xbt_assert(value,"Please set env var SMPI_GLOBAL_SIZE to the expected number of processes.");
147
148   return xbt_str_parse_int(value, "SMPI_GLOBAL_SIZE contains a non-numerical value: %s");
149 }
150
151 void smpi_comm_set_copy_data_callback(void (*callback) (smx_activity_t, void*, size_t))
152 {
153   smpi_comm_copy_data_callback = callback;
154 }
155
156 static void memcpy_private(void* dest, const void* src, std::vector<std::pair<size_t, size_t>>& private_blocks)
157 {
158   for (auto const& block : private_blocks)
159     memcpy((uint8_t*)dest+block.first, (uint8_t*)src+block.first, block.second-block.first);
160 }
161
162 static void check_blocks(std::vector<std::pair<size_t, size_t>> &private_blocks, size_t buff_size) {
163   for (auto const& block : private_blocks)
164     xbt_assert(block.first <= block.second && block.second <= buff_size, "Oops, bug in shared malloc.");
165 }
166
167 void smpi_comm_copy_buffer_callback(smx_activity_t synchro, void *buff, size_t buff_size)
168 {
169   simgrid::kernel::activity::CommImplPtr comm =
170       boost::dynamic_pointer_cast<simgrid::kernel::activity::CommImpl>(synchro);
171   int src_shared                        = 0;
172   int dst_shared                        = 0;
173   size_t src_offset                     = 0;
174   size_t dst_offset                     = 0;
175   std::vector<std::pair<size_t, size_t>> src_private_blocks;
176   std::vector<std::pair<size_t, size_t>> dst_private_blocks;
177   XBT_DEBUG("Copy the data over");
178   if((src_shared=smpi_is_shared(buff, src_private_blocks, &src_offset))) {
179     XBT_DEBUG("Sender %p is shared. Let's ignore it.", buff);
180     src_private_blocks = shift_and_frame_private_blocks(src_private_blocks, src_offset, buff_size);
181   }
182   else {
183     src_private_blocks.clear();
184     src_private_blocks.push_back(std::make_pair(0, buff_size));
185   }
186   if((dst_shared=smpi_is_shared((char*)comm->dst_buff, dst_private_blocks, &dst_offset))) {
187     XBT_DEBUG("Receiver %p is shared. Let's ignore it.", (char*)comm->dst_buff);
188     dst_private_blocks = shift_and_frame_private_blocks(dst_private_blocks, dst_offset, buff_size);
189   }
190   else {
191     dst_private_blocks.clear();
192     dst_private_blocks.push_back(std::make_pair(0, buff_size));
193   }
194   check_blocks(src_private_blocks, buff_size);
195   check_blocks(dst_private_blocks, buff_size);
196   auto private_blocks = merge_private_blocks(src_private_blocks, dst_private_blocks);
197   check_blocks(private_blocks, buff_size);
198   void* tmpbuff=buff;
199   if ((smpi_privatize_global_variables == SmpiPrivStrategies::MMAP) &&
200       (static_cast<char*>(buff) >= smpi_data_exe_start) &&
201       (static_cast<char*>(buff) < smpi_data_exe_start + smpi_data_exe_size)) {
202     XBT_DEBUG("Privatization : We are copying from a zone inside global memory... Saving data to temp buffer !");
203     smpi_switch_data_segment(comm->src_proc->iface());
204     tmpbuff = static_cast<void*>(xbt_malloc(buff_size));
205     memcpy_private(tmpbuff, buff, private_blocks);
206   }
207
208   if ((smpi_privatize_global_variables == SmpiPrivStrategies::MMAP) && ((char*)comm->dst_buff >= smpi_data_exe_start) &&
209       ((char*)comm->dst_buff < smpi_data_exe_start + smpi_data_exe_size)) {
210     XBT_DEBUG("Privatization : We are copying to a zone inside global memory - Switch data segment");
211     smpi_switch_data_segment(comm->dst_proc->iface());
212   }
213   XBT_DEBUG("Copying %zu bytes from %p to %p", buff_size, tmpbuff,comm->dst_buff);
214   memcpy_private(comm->dst_buff, tmpbuff, private_blocks);
215
216   if (comm->detached) {
217     // if this is a detached send, the source buffer was duplicated by SMPI
218     // sender to make the original buffer available to the application ASAP
219     xbt_free(buff);
220     //It seems that the request is used after the call there this should be free somewhere else but where???
221     //xbt_free(comm->comm.src_data);// inside SMPI the request is kept inside the user data and should be free
222     comm->src_buff = nullptr;
223   }
224   if (tmpbuff != buff)
225     xbt_free(tmpbuff);
226 }
227
228 void smpi_comm_null_copy_buffer_callback(smx_activity_t comm, void *buff, size_t buff_size)
229 {
230   /* nothing done in this version */
231 }
232
233 static void smpi_check_options()
234 {
235   //check correctness of MPI parameters
236
237   xbt_assert(simgrid::config::get_value<int>("smpi/async-small-thresh") <=
238              simgrid::config::get_value<int>("smpi/send-is-detached-thresh"));
239
240   if (simgrid::config::is_default("smpi/host-speed")) {
241     XBT_INFO("You did not set the power of the host running the simulation.  "
242              "The timings will certainly not be accurate.  "
243              "Use the option \"--cfg=smpi/host-speed:<flops>\" to set its value."
244              "Check http://simgrid.org/simgrid/latest/doc/options.html#options_smpi_bench for more information.");
245   }
246
247   xbt_assert(simgrid::config::get_value<double>("smpi/cpu-threshold") >= 0,
248              "The 'smpi/cpu-threshold' option cannot have negative values [anymore]. If you want to discard "
249              "the simulation of any computation, please use 'smpi/simulate-computation:no' instead.");
250 }
251
252 int smpi_enabled() {
253   return not process_data.empty();
254 }
255
256 void smpi_global_init()
257 {
258   if (not MC_is_active()) {
259     global_timer = xbt_os_timer_new();
260     xbt_os_walltimer_start(global_timer);
261   }
262
263   std::string filename = simgrid::config::get_value<std::string>("smpi/comp-adjustment-file");
264   if (not filename.empty()) {
265     std::ifstream fstream(filename);
266     if (not fstream.is_open()) {
267       xbt_die("Could not open file %s. Does it exist?", filename.c_str());
268     }
269
270     std::string line;
271     typedef boost::tokenizer< boost::escaped_list_separator<char>> Tokenizer;
272     std::getline(fstream, line); // Skip the header line
273     while (std::getline(fstream, line)) {
274       Tokenizer tok(line);
275       Tokenizer::iterator it  = tok.begin();
276       Tokenizer::iterator end = std::next(tok.begin());
277
278       std::string location = *it;
279       boost::trim(location);
280       location2speedup.insert(std::pair<std::string, double>(location, std::stod(*end)));
281     }
282   }
283
284 #if HAVE_PAPI
285   // This map holds for each computation unit (such as "default" or "process1" etc.)
286   // the configuration as given by the user (counter data as a pair of (counter_name, counter_counter))
287   // and the (computed) event_set.
288
289   if (not simgrid::config::get_value<std::string>("smpi/papi-events").empty()) {
290     if (PAPI_library_init(PAPI_VER_CURRENT) != PAPI_VER_CURRENT)
291       XBT_ERROR("Could not initialize PAPI library; is it correctly installed and linked?"
292                 " Expected version is %u", PAPI_VER_CURRENT);
293
294     typedef boost::tokenizer<boost::char_separator<char>> Tokenizer;
295     boost::char_separator<char> separator_units(";");
296     std::string str = simgrid::config::get_value<std::string>("smpi/papi-events");
297     Tokenizer tokens(str, separator_units);
298
299     // Iterate over all the computational units. This could be processes, hosts, threads, ranks... You name it.
300     // I'm not exactly sure what we will support eventually, so I'll leave it at the general term "units".
301     for (auto const& unit_it : tokens) {
302       boost::char_separator<char> separator_events(":");
303       Tokenizer event_tokens(unit_it, separator_events);
304
305       int event_set = PAPI_NULL;
306       if (PAPI_create_eventset(&event_set) != PAPI_OK) {
307         // TODO: Should this let the whole simulation die?
308         XBT_CRITICAL("Could not create PAPI event set during init.");
309       }
310
311       // NOTE: We cannot use a map here, as we must obey the order of the counters
312       // This is important for PAPI: We need to map the values of counters back
313       // to the event_names (so, when PAPI_read() has finished)!
314       papi_counter_t counters2values;
315
316       // Iterate over all counters that were specified for this specific
317       // unit.
318       // Note that we need to remove the name of the unit
319       // (that could also be the "default" value), which always comes first.
320       // Hence, we start at ++(events.begin())!
321       for (Tokenizer::iterator events_it = ++(event_tokens.begin()); events_it != event_tokens.end(); ++events_it) {
322
323         int event_code   = PAPI_NULL;
324         char* event_name = const_cast<char*>((*events_it).c_str());
325         if (PAPI_event_name_to_code(event_name, &event_code) == PAPI_OK) {
326           if (PAPI_add_event(event_set, event_code) != PAPI_OK) {
327             XBT_ERROR("Could not add PAPI event '%s'. Skipping.", event_name);
328             continue;
329           } else {
330             XBT_DEBUG("Successfully added PAPI event '%s' to the event set.", event_name);
331           }
332         } else {
333           XBT_CRITICAL("Could not find PAPI event '%s'. Skipping.", event_name);
334           continue;
335         }
336
337         counters2values.push_back(
338             // We cannot just pass *events_it, as this is of type const basic_string
339             std::make_pair<std::string, long long>(std::string(*events_it), 0));
340       }
341
342       std::string unit_name    = *(event_tokens.begin());
343       papi_process_data config = {.counter_data = std::move(counters2values), .event_set = event_set};
344
345       units2papi_setup.insert(std::make_pair(unit_name, std::move(config)));
346     }
347   }
348 #endif
349 }
350
351 void smpi_global_destroy()
352 {
353   smpi_bench_destroy();
354   smpi_shared_destroy();
355   smpi_deployment_cleanup_instances();
356
357   if (simgrid::smpi::Colls::smpi_coll_cleanup_callback != nullptr)
358     simgrid::smpi::Colls::smpi_coll_cleanup_callback();
359
360   MPI_COMM_WORLD = MPI_COMM_NULL;
361
362   if (not MC_is_active()) {
363     xbt_os_timer_free(global_timer);
364   }
365
366   if (smpi_privatize_global_variables == SmpiPrivStrategies::MMAP)
367     smpi_destroy_global_memory_segments();
368   smpi_free_static();
369   if(simgrid::smpi::F2C::lookup() != nullptr)
370     simgrid::smpi::F2C::delete_lookup();
371 }
372
373 static void smpi_init_options(){
374   // return if already called
375   if (smpi_cpu_threshold > -1)
376     return;
377   simgrid::smpi::Colls::set_collectives();
378   simgrid::smpi::Colls::smpi_coll_cleanup_callback = nullptr;
379   smpi_cpu_threshold                               = simgrid::config::get_value<double>("smpi/cpu-threshold");
380   smpi_host_speed                                  = simgrid::config::get_value<double>("smpi/host-speed");
381   xbt_assert(smpi_host_speed > 0.0, "You're trying to set the host_speed to a non-positive value (given: %f)", smpi_host_speed);
382   std::string smpi_privatize_option = simgrid::config::get_value<std::string>("smpi/privatization");
383   if (smpi_privatize_option == "no" || smpi_privatize_option == "0")
384     smpi_privatize_global_variables = SmpiPrivStrategies::NONE;
385   else if (smpi_privatize_option == "yes" || smpi_privatize_option == "1")
386     smpi_privatize_global_variables = SmpiPrivStrategies::DEFAULT;
387   else if (smpi_privatize_option == "mmap")
388     smpi_privatize_global_variables = SmpiPrivStrategies::MMAP;
389   else if (smpi_privatize_option == "dlopen")
390     smpi_privatize_global_variables = SmpiPrivStrategies::DLOPEN;
391   else
392     xbt_die("Invalid value for smpi/privatization: '%s'", smpi_privatize_option.c_str());
393
394   if (not SMPI_switch_data_segment) {
395     XBT_DEBUG("Running without smpi_main(); disable smpi/privatization.");
396     smpi_privatize_global_variables = SmpiPrivStrategies::NONE;
397   }
398 #if !HAVE_WORKING_MMAP
399   if (smpi_privatize_global_variables == SmpiPrivStrategies::MMAP) {
400     XBT_INFO("mmap privatization is broken on this platform, switching to dlopen privatization instead.");
401     smpi_privatize_global_variables = SmpiPrivStrategies::DLOPEN;
402   }
403 #endif
404
405   if (smpi_cpu_threshold < 0)
406     smpi_cpu_threshold = DBL_MAX;
407
408   std::string val = simgrid::config::get_value<std::string>("smpi/shared-malloc");
409   if ((val == "yes") || (val == "1") || (val == "on") || (val == "global")) {
410     smpi_cfg_shared_malloc = SharedMallocType::GLOBAL;
411   } else if (val == "local") {
412     smpi_cfg_shared_malloc = SharedMallocType::LOCAL;
413   } else if ((val == "no") || (val == "0") || (val == "off")) {
414     smpi_cfg_shared_malloc = SharedMallocType::NONE;
415   } else {
416     xbt_die("Invalid value '%s' for option smpi/shared-malloc. Possible values: 'on' or 'global', 'local', 'off'",
417             val.c_str());
418   }
419 }
420
421 typedef std::function<int(int argc, char *argv[])> smpi_entry_point_type;
422 typedef int (* smpi_c_entry_point_type)(int argc, char **argv);
423 typedef void (*smpi_fortran_entry_point_type)();
424
425 static int smpi_run_entry_point(smpi_entry_point_type entry_point, std::vector<std::string> args)
426 {
427   // copy C strings, we need them writable
428   std::vector<char*>* args4argv = new std::vector<char*>(args.size());
429   std::transform(begin(args), end(args), begin(*args4argv), [](const std::string& s) { return xbt_strdup(s.c_str()); });
430
431 #if !SMPI_IFORT
432   // take a copy of args4argv to keep reference of the allocated strings
433   const std::vector<char*> args2str(*args4argv);
434 #endif
435   int argc = args4argv->size();
436   args4argv->push_back(nullptr);
437   char** argv = args4argv->data();
438
439   simgrid::smpi::ActorExt::init(&argc, &argv);
440 #if SMPI_IFORT
441   for_rtl_init_ (&argc, argv);
442 #elif SMPI_FLANG
443   __io_set_argc(argc);
444   __io_set_argv(argv);
445 #elif SMPI_GFORTRAN
446   _gfortran_set_args(argc, argv);
447 #endif 
448   int res = entry_point(argc, argv);
449
450 #if SMPI_IFORT
451   for_rtl_finish_ ();
452 #else
453   for (char* s : args2str)
454     xbt_free(s);
455   delete args4argv;
456 #endif
457
458   if (res != 0){
459     XBT_WARN("SMPI process did not return 0. Return value : %d", res);
460     if (smpi_exit_status == 0)
461       smpi_exit_status = res;
462   }
463   return 0;
464 }
465
466
467 // TODO, remove the number of functions involved here
468 static smpi_entry_point_type smpi_resolve_function(void* handle)
469 {
470   smpi_fortran_entry_point_type entry_point_fortran = (smpi_fortran_entry_point_type)dlsym(handle, "user_main_");
471   if (entry_point_fortran != nullptr) {
472     return [entry_point_fortran](int argc, char** argv) {
473       entry_point_fortran();
474       return 0;
475     };
476   }
477
478   smpi_c_entry_point_type entry_point = (smpi_c_entry_point_type)dlsym(handle, "main");
479   if (entry_point != nullptr) {
480     return entry_point;
481   }
482
483   return smpi_entry_point_type();
484 }
485
486 static void smpi_copy_file(std::string src, std::string target, off_t fdin_size)
487 {
488   int fdin = open(src.c_str(), O_RDONLY);
489   xbt_assert(fdin >= 0, "Cannot read from %s. Please make sure that the file exists and is executable.", src.c_str());
490   int fdout = open(target.c_str(), O_CREAT | O_RDWR, S_IRWXU);
491   xbt_assert(fdout >= 0, "Cannot write into %s", target.c_str());
492
493   XBT_DEBUG("Copy %jd bytes into %s", static_cast<intmax_t>(fdin_size), target.c_str());
494 #if HAVE_SENDFILE
495   ssize_t sent_size = sendfile(fdout, fdin, NULL, fdin_size);
496   xbt_assert(sent_size == fdin_size, "Error while copying %s: only %zd bytes copied instead of %jd (errno: %d -- %s)",
497              target.c_str(), sent_size, static_cast<intmax_t>(fdin_size), errno, strerror(errno));
498 #else
499   const int bufsize = 1024 * 1024 * 4;
500   char buf[bufsize];
501   while (int got = read(fdin, buf, bufsize)) {
502     if (got == -1) {
503       xbt_assert(errno == EINTR, "Cannot read from %s", src.c_str());
504     } else {
505       char* p  = buf;
506       int todo = got;
507       while (int done = write(fdout, p, todo)) {
508         if (done == -1) {
509           xbt_assert(errno == EINTR, "Cannot write into %s", target.c_str());
510         } else {
511           p += done;
512           todo -= done;
513         }
514       }
515     }
516   }
517 #endif
518   close(fdin);
519   close(fdout);
520 }
521
522 #if not defined(__APPLE__)
523 static int visit_libs(struct dl_phdr_info* info, size_t, void* data)
524 {
525   char* libname = (char*)(data);
526   const char *path = info->dlpi_name;
527   if(strstr(path, libname)){
528     strncpy(libname, path, 512);
529     return 1;
530   }
531   
532   return 0;
533 }
534 #endif
535
536 static void smpi_init_privatization_dlopen(std::string executable)
537 {
538   // Prepare the copy of the binary (get its size)
539   struct stat fdin_stat;
540   stat(executable.c_str(), &fdin_stat);
541   off_t fdin_size         = fdin_stat.st_size;
542   static std::size_t rank = 0;
543
544   std::string libnames = simgrid::config::get_value<std::string>("smpi/privatize-libs");
545   if (not libnames.empty()) {
546     // split option
547     std::vector<std::string> privatize_libs;
548     boost::split(privatize_libs, libnames, boost::is_any_of(";"));
549
550     for (auto const& libname : privatize_libs) {
551       // load the library once to add it to the local libs, to get the absolute path
552       void* libhandle = dlopen(libname.c_str(), RTLD_LAZY);
553       // get library name from path
554       char fullpath[512] = {'\0'};
555       strcpy(fullpath, libname.c_str());
556 #if not defined(__APPLE__)
557       int ret = dl_iterate_phdr(visit_libs, fullpath);
558       if (ret == 0)
559         xbt_die("Can't find a linked %s - check the setting you gave to smpi/privatize-libs", fullpath);
560       else
561         XBT_DEBUG("Extra lib to privatize found : %s", fullpath);
562 #else
563       xbt_die("smpi/privatize-libs is not (yet) compatible with OSX");
564 #endif
565       privatize_libs_paths.push_back(fullpath);
566       dlclose(libhandle);
567     }
568   }
569
570   simix_global->default_function = [executable, fdin_size](std::vector<std::string> args) {
571     return std::function<void()>([executable, fdin_size, args] {
572
573       // Copy the dynamic library:
574       std::string target_executable =
575           executable + "_" + std::to_string(getpid()) + "_" + std::to_string(rank) + ".so";
576
577       smpi_copy_file(executable, target_executable, fdin_size);
578       // if smpi/privatize-libs is set, duplicate pointed lib and link each executable copy to a different one.
579       std::vector<std::string> target_libs;
580       for (auto const& libpath : privatize_libs_paths) {
581         // if we were given a full path, strip it
582         size_t index = libpath.find_last_of("/\\");
583         std::string libname;
584         if (index != std::string::npos)
585           libname = libpath.substr(index + 1);
586
587         if (not libname.empty()) {
588           // load the library to add it to the local libs, to get the absolute path
589           struct stat fdin_stat2;
590           stat(libpath.c_str(), &fdin_stat2);
591           off_t fdin_size2 = fdin_stat2.st_size;
592
593           // Copy the dynamic library, the new name must be the same length as the old one
594           // just replace the name with 7 digits for the rank and the rest of the name.
595           unsigned int pad = 7;
596           if (libname.length() < pad)
597             pad = libname.length();
598           std::string target_lib =
599               std::string(pad - std::to_string(rank).length(), '0') + std::to_string(rank) + libname.substr(pad);
600           target_libs.push_back(target_lib);
601           XBT_DEBUG("copy lib %s to %s, with size %lld", libpath.c_str(), target_lib.c_str(), (long long)fdin_size2);
602           smpi_copy_file(libpath, target_lib, fdin_size2);
603
604           std::string sedcommand = "sed -i -e 's/" + libname + "/" + target_lib + "/g' " + target_executable;
605           int ret                = system(sedcommand.c_str());
606           if (ret != 0)
607             xbt_die("error while applying sed command %s \n", sedcommand.c_str());
608         }
609       }
610
611       rank++;
612       // Load the copy and resolve the entry point:
613       void* handle    = dlopen(target_executable.c_str(), RTLD_LAZY | RTLD_LOCAL | WANT_RTLD_DEEPBIND);
614       int saved_errno = errno;
615       if (simgrid::config::get_value<bool>("smpi/keep-temps") == false) {
616         unlink(target_executable.c_str());
617         for (const std::string& target_lib : target_libs)
618           unlink(target_lib.c_str());
619       }
620       if (handle == nullptr)
621         xbt_die("dlopen failed: %s (errno: %d -- %s)", dlerror(), saved_errno, strerror(saved_errno));
622       smpi_entry_point_type entry_point = smpi_resolve_function(handle);
623       if (not entry_point)
624         xbt_die("Could not resolve entry point");
625       smpi_run_entry_point(entry_point, args);
626     });
627   };
628 }
629
630 static void smpi_init_privatization_no_dlopen(std::string executable)
631 {
632   if (smpi_privatize_global_variables == SmpiPrivStrategies::MMAP)
633     smpi_prepare_global_memory_segment();
634   // Load the dynamic library and resolve the entry point:
635   void* handle = dlopen(executable.c_str(), RTLD_LAZY | RTLD_LOCAL);
636   if (handle == nullptr)
637     xbt_die("dlopen failed for %s: %s (errno: %d -- %s)", executable.c_str(), dlerror(), errno, strerror(errno));
638   smpi_entry_point_type entry_point = smpi_resolve_function(handle);
639   if (not entry_point)
640     xbt_die("main not found in %s", executable.c_str());
641   if (smpi_privatize_global_variables == SmpiPrivStrategies::MMAP)
642     smpi_backup_global_memory_segment();
643
644   // Execute the same entry point for each simulated process:
645   simix_global->default_function = [entry_point](std::vector<std::string> args) {
646     return std::function<void()>([entry_point, args] { smpi_run_entry_point(entry_point, args); });
647   };
648 }
649
650 int smpi_main(const char* executable, int argc, char* argv[])
651 {
652   srand(SMPI_RAND_SEED);
653
654   if (getenv("SMPI_PRETEND_CC") != nullptr) {
655     /* Hack to ensure that smpicc can pretend to be a simple compiler. Particularly handy to pass it to the
656      * configuration tools */
657     return 0;
658   }
659
660   TRACE_global_init();
661   SIMIX_global_init(&argc, argv);
662
663   SMPI_switch_data_segment = &smpi_switch_data_segment;
664
665   // TODO This will not be executed in the case where smpi_main is not called,
666   // e.g., not for smpi_msg_masterslave. This should be moved to another location
667   // that is always called -- maybe close to Actor::on_creation?
668   simgrid::s4u::Host::on_creation.connect(
669       [](simgrid::s4u::Host& host) { host.extension_set(new simgrid::smpi::Host(&host)); });
670
671   // parse the platform file: get the host list
672   simgrid::s4u::Engine::get_instance()->load_platform(argv[1]);
673   SIMIX_comm_set_copy_data_callback(smpi_comm_copy_buffer_callback);
674
675   smpi_init_options();
676   if (smpi_privatize_global_variables == SmpiPrivStrategies::DLOPEN)
677     smpi_init_privatization_dlopen(executable);
678   else
679     smpi_init_privatization_no_dlopen(executable);
680
681   SMPI_init();
682   simgrid::s4u::Engine::get_instance()->load_deployment(argv[2]);
683   SMPI_app_instance_register(smpi_default_instance_name.c_str(), nullptr,
684                              process_data.size()); // This call has a side effect on process_count...
685   MPI_COMM_WORLD = *smpi_deployment_comm_world(smpi_default_instance_name);
686   smpi_universe_size = process_count;
687
688
689   /* Clean IO before the run */
690   fflush(stdout);
691   fflush(stderr);
692
693   if (MC_is_active()) {
694     MC_run();
695   } else {
696
697     SIMIX_run();
698
699     xbt_os_walltimer_stop(global_timer);
700     if (simgrid::config::get_value<bool>("smpi/display-timing")) {
701       double global_time = xbt_os_timer_elapsed(global_timer);
702       XBT_INFO("Simulated time: %g seconds. \n\n"
703           "The simulation took %g seconds (after parsing and platform setup)\n"
704           "%g seconds were actual computation of the application",
705           SIMIX_get_clock(), global_time , smpi_total_benched_time);
706
707       if (smpi_total_benched_time/global_time>=0.75)
708       XBT_INFO("More than 75%% of the time was spent inside the application code.\n"
709       "You may want to use sampling functions or trace replay to reduce this.");
710     }
711   }
712   smpi_global_destroy();
713
714   return smpi_exit_status;
715 }
716
717 // Called either directly from the user code, or from the code called by smpirun
718 void SMPI_init(){
719   simgrid::s4u::Actor::on_creation.connect([](simgrid::s4u::ActorPtr actor) {
720     if (not actor->is_daemon()) {
721       process_data.insert({actor, new simgrid::smpi::ActorExt(actor, nullptr)});
722     }
723   });
724   simgrid::s4u::Actor::on_destruction.connect([](simgrid::s4u::ActorPtr actor) {
725     auto it = process_data.find(actor);
726     if (it != process_data.end()) {
727       delete it->second;
728       process_data.erase(it);
729     }
730   });
731
732   smpi_init_options();
733   smpi_global_init();
734   smpi_check_options();
735 }
736
737 void SMPI_finalize(){
738   smpi_global_destroy();
739 }
740
741 void smpi_mpi_init() {
742   smpi_init_fortran_types();
743   if(smpi_init_sleep > 0)
744     simcall_process_sleep(smpi_init_sleep);
745 }