Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
Forgot that ifort needs argv to stay alive.
[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 <dlfcn.h>
19 #include <fcntl.h>
20 #include <fstream>
21
22 #if not defined(__APPLE__)
23 #include <link.h>
24 #endif
25
26 #if defined(__APPLE__)
27 # include <AvailabilityMacros.h>
28 # ifndef MAC_OS_X_VERSION_10_12
29 #   define MAC_OS_X_VERSION_10_12 101200
30 # endif
31 # define HAVE_WORKING_MMAP (MAC_OS_X_VERSION_MIN_REQUIRED >= MAC_OS_X_VERSION_10_12)
32 #elif defined(__FreeBSD__)
33 # define HAVE_WORKING_MMAP 0
34 #else
35 # define HAVE_WORKING_MMAP 1
36 #endif
37
38 #if HAVE_SENDFILE
39 #include <sys/sendfile.h>
40 #endif
41
42 XBT_LOG_NEW_DEFAULT_SUBCATEGORY(smpi_kernel, smpi, "Logging specific to SMPI (kernel)");
43 #include <boost/tokenizer.hpp>
44 #include <boost/algorithm/string.hpp> /* trim_right / trim_left */
45
46 #if SMPI_IFORT
47   extern "C" void for_rtl_init_ (int *, char **);
48   extern "C" void for_rtl_finish_ ();
49 #elif SMPI_FLANG
50   extern "C" void __io_set_argc(int);
51   extern "C" void __io_set_argv(char **);
52 #elif SMPI_GFORTRAN
53   extern "C" void _gfortran_set_args(int, char **);
54 #endif
55
56 #ifndef RTLD_DEEPBIND
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 #define RTLD_DEEPBIND 0
62 #endif
63
64 #if HAVE_PAPI
65 #include "papi.h"
66 std::string papi_default_config_name = "default";
67 std::map</* computation unit name */ std::string, papi_process_data> units2papi_setup;
68 #endif
69
70 using simgrid::s4u::Actor;
71 using simgrid::s4u::ActorPtr;
72 std::unordered_map<std::string, double> location2speedup;
73
74 static std::map</*process_id*/ ActorPtr, simgrid::smpi::ActorExt*> process_data;
75 int process_count = 0;
76 static int smpi_exit_status = 0;
77 int smpi_universe_size = 0;
78 extern double smpi_total_benched_time;
79 xbt_os_timer_t global_timer;
80 static std::vector<std::string> privatize_libs_paths;
81 /**
82  * Setting MPI_COMM_WORLD to MPI_COMM_UNINITIALIZED (it's a variable)
83  * is important because the implementation of MPI_Comm checks
84  * "this == MPI_COMM_UNINITIALIZED"? If yes, it uses smpi_process()->comm_world()
85  * instead of "this".
86  * This is basically how we only have one global variable but all processes have
87  * different communicators (the one their SMPI instance uses).
88  *
89  * See smpi_comm.cpp and the functions therein for details.
90  */
91 MPI_Comm MPI_COMM_WORLD = MPI_COMM_UNINITIALIZED;
92 MPI_Errhandler *MPI_ERRORS_RETURN = nullptr;
93 MPI_Errhandler *MPI_ERRORS_ARE_FATAL = nullptr;
94 MPI_Errhandler *MPI_ERRHANDLER_NULL = nullptr;
95 // No instance gets manually created; check also the smpirun.in script as
96 // this default name is used there as well (when the <actor> tag is generated).
97 static const std::string smpi_default_instance_name("smpirun");
98 static simgrid::config::Flag<double> smpi_wtime_sleep(
99   "smpi/wtime", "Minimum time to inject inside a call to MPI_Wtime", 0.0);
100 static simgrid::config::Flag<double> smpi_init_sleep(
101   "smpi/init", "Time to inject inside a call to MPI_Init", 0.0);
102
103 void (*smpi_comm_copy_data_callback) (smx_activity_t, void*, size_t) = &smpi_comm_copy_buffer_callback;
104
105 int smpi_process_count()
106 {
107   return process_count;
108 }
109
110 simgrid::smpi::ActorExt* smpi_process()
111 {
112   ActorPtr me = Actor::self();
113
114   if (me == nullptr) // This happens sometimes (eg, when linking against NS3 because it pulls openMPI...)
115     return nullptr;
116
117   return process_data.at(me);
118 }
119
120 simgrid::smpi::ActorExt* smpi_process_remote(ActorPtr actor)
121 {
122   return process_data.at(actor);
123 }
124
125 MPI_Comm smpi_process_comm_self(){
126   return smpi_process()->comm_self();
127 }
128
129 void smpi_process_init(int *argc, char ***argv){
130   simgrid::smpi::ActorExt::init(argc, argv);
131 }
132
133 int smpi_process_index(){
134   return simgrid::s4u::this_actor::get_pid();
135 }
136
137 void * smpi_process_get_user_data(){
138   return Actor::self()->get_impl()->get_user_data();
139 }
140
141 void smpi_process_set_user_data(void *data){
142   Actor::self()->get_impl()->set_user_data(data);
143 }
144
145
146 int smpi_global_size()
147 {
148   char *value = getenv("SMPI_GLOBAL_SIZE");
149   xbt_assert(value,"Please set env var SMPI_GLOBAL_SIZE to the expected number of processes.");
150
151   return xbt_str_parse_int(value, "SMPI_GLOBAL_SIZE contains a non-numerical value: %s");
152 }
153
154 void smpi_comm_set_copy_data_callback(void (*callback) (smx_activity_t, void*, size_t))
155 {
156   smpi_comm_copy_data_callback = callback;
157 }
158
159 static void memcpy_private(void* dest, const void* src, std::vector<std::pair<size_t, size_t>>& private_blocks)
160 {
161   for (auto const& block : private_blocks)
162     memcpy((uint8_t*)dest+block.first, (uint8_t*)src+block.first, block.second-block.first);
163 }
164
165 static void check_blocks(std::vector<std::pair<size_t, size_t>> &private_blocks, size_t buff_size) {
166   for (auto const& block : private_blocks)
167     xbt_assert(block.first <= block.second && block.second <= buff_size, "Oops, bug in shared malloc.");
168 }
169
170 void smpi_comm_copy_buffer_callback(smx_activity_t synchro, void *buff, size_t buff_size)
171 {
172   simgrid::kernel::activity::CommImplPtr comm =
173       boost::dynamic_pointer_cast<simgrid::kernel::activity::CommImpl>(synchro);
174   int src_shared                        = 0;
175   int dst_shared                        = 0;
176   size_t src_offset                     = 0;
177   size_t dst_offset                     = 0;
178   std::vector<std::pair<size_t, size_t>> src_private_blocks;
179   std::vector<std::pair<size_t, size_t>> dst_private_blocks;
180   XBT_DEBUG("Copy the data over");
181   if((src_shared=smpi_is_shared(buff, src_private_blocks, &src_offset))) {
182     XBT_DEBUG("Sender %p is shared. Let's ignore it.", buff);
183     src_private_blocks = shift_and_frame_private_blocks(src_private_blocks, src_offset, buff_size);
184   }
185   else {
186     src_private_blocks.clear();
187     src_private_blocks.push_back(std::make_pair(0, buff_size));
188   }
189   if((dst_shared=smpi_is_shared((char*)comm->dst_buff, dst_private_blocks, &dst_offset))) {
190     XBT_DEBUG("Receiver %p is shared. Let's ignore it.", (char*)comm->dst_buff);
191     dst_private_blocks = shift_and_frame_private_blocks(dst_private_blocks, dst_offset, buff_size);
192   }
193   else {
194     dst_private_blocks.clear();
195     dst_private_blocks.push_back(std::make_pair(0, buff_size));
196   }
197   check_blocks(src_private_blocks, buff_size);
198   check_blocks(dst_private_blocks, buff_size);
199   auto private_blocks = merge_private_blocks(src_private_blocks, dst_private_blocks);
200   check_blocks(private_blocks, buff_size);
201   void* tmpbuff=buff;
202   if ((smpi_privatize_global_variables == SmpiPrivStrategies::MMAP) &&
203       (static_cast<char*>(buff) >= smpi_data_exe_start) &&
204       (static_cast<char*>(buff) < smpi_data_exe_start + smpi_data_exe_size)) {
205     XBT_DEBUG("Privatization : We are copying from a zone inside global memory... Saving data to temp buffer !");
206     smpi_switch_data_segment(comm->src_proc->iface());
207     tmpbuff = static_cast<void*>(xbt_malloc(buff_size));
208     memcpy_private(tmpbuff, buff, private_blocks);
209   }
210
211   if ((smpi_privatize_global_variables == SmpiPrivStrategies::MMAP) && ((char*)comm->dst_buff >= smpi_data_exe_start) &&
212       ((char*)comm->dst_buff < smpi_data_exe_start + smpi_data_exe_size)) {
213     XBT_DEBUG("Privatization : We are copying to a zone inside global memory - Switch data segment");
214     smpi_switch_data_segment(comm->dst_proc->iface());
215   }
216   XBT_DEBUG("Copying %zu bytes from %p to %p", buff_size, tmpbuff,comm->dst_buff);
217   memcpy_private(comm->dst_buff, tmpbuff, private_blocks);
218
219   if (comm->detached) {
220     // if this is a detached send, the source buffer was duplicated by SMPI
221     // sender to make the original buffer available to the application ASAP
222     xbt_free(buff);
223     //It seems that the request is used after the call there this should be free somewhere else but where???
224     //xbt_free(comm->comm.src_data);// inside SMPI the request is kept inside the user data and should be free
225     comm->src_buff = nullptr;
226   }
227   if (tmpbuff != buff)
228     xbt_free(tmpbuff);
229 }
230
231 void smpi_comm_null_copy_buffer_callback(smx_activity_t comm, void *buff, size_t buff_size)
232 {
233   /* nothing done in this version */
234 }
235
236 static void smpi_check_options()
237 {
238   //check correctness of MPI parameters
239
240   xbt_assert(simgrid::config::get_value<int>("smpi/async-small-thresh") <=
241              simgrid::config::get_value<int>("smpi/send-is-detached-thresh"));
242
243   if (simgrid::config::is_default("smpi/host-speed")) {
244     XBT_INFO("You did not set the power of the host running the simulation.  "
245              "The timings will certainly not be accurate.  "
246              "Use the option \"--cfg=smpi/host-speed:<flops>\" to set its value."
247              "Check http://simgrid.org/simgrid/latest/doc/options.html#options_smpi_bench for more information.");
248   }
249
250   xbt_assert(simgrid::config::get_value<double>("smpi/cpu-threshold") >= 0,
251              "The 'smpi/cpu-threshold' option cannot have negative values [anymore]. If you want to discard "
252              "the simulation of any computation, please use 'smpi/simulate-computation:no' instead.");
253 }
254
255 int smpi_enabled() {
256   return not process_data.empty();
257 }
258
259 void smpi_global_init()
260 {
261   if (not MC_is_active()) {
262     global_timer = xbt_os_timer_new();
263     xbt_os_walltimer_start(global_timer);
264   }
265
266   std::string filename = simgrid::config::get_value<std::string>("smpi/comp-adjustment-file");
267   if (not filename.empty()) {
268     std::ifstream fstream(filename);
269     if (not fstream.is_open()) {
270       xbt_die("Could not open file %s. Does it exist?", filename.c_str());
271     }
272
273     std::string line;
274     typedef boost::tokenizer< boost::escaped_list_separator<char>> Tokenizer;
275     std::getline(fstream, line); // Skip the header line
276     while (std::getline(fstream, line)) {
277       Tokenizer tok(line);
278       Tokenizer::iterator it  = tok.begin();
279       Tokenizer::iterator end = std::next(tok.begin());
280
281       std::string location = *it;
282       boost::trim(location);
283       location2speedup.insert(std::pair<std::string, double>(location, std::stod(*end)));
284     }
285   }
286
287 #if HAVE_PAPI
288   // This map holds for each computation unit (such as "default" or "process1" etc.)
289   // the configuration as given by the user (counter data as a pair of (counter_name, counter_counter))
290   // and the (computed) event_set.
291
292   if (not simgrid::config::get_value<std::string>("smpi/papi-events").empty()) {
293     if (PAPI_library_init(PAPI_VER_CURRENT) != PAPI_VER_CURRENT)
294       XBT_ERROR("Could not initialize PAPI library; is it correctly installed and linked?"
295                 " Expected version is %u", PAPI_VER_CURRENT);
296
297     typedef boost::tokenizer<boost::char_separator<char>> Tokenizer;
298     boost::char_separator<char> separator_units(";");
299     std::string str = simgrid::config::get_value<std::string>("smpi/papi-events");
300     Tokenizer tokens(str, separator_units);
301
302     // Iterate over all the computational units. This could be processes, hosts, threads, ranks... You name it.
303     // I'm not exactly sure what we will support eventually, so I'll leave it at the general term "units".
304     for (auto const& unit_it : tokens) {
305       boost::char_separator<char> separator_events(":");
306       Tokenizer event_tokens(unit_it, separator_events);
307
308       int event_set = PAPI_NULL;
309       if (PAPI_create_eventset(&event_set) != PAPI_OK) {
310         // TODO: Should this let the whole simulation die?
311         XBT_CRITICAL("Could not create PAPI event set during init.");
312       }
313
314       // NOTE: We cannot use a map here, as we must obey the order of the counters
315       // This is important for PAPI: We need to map the values of counters back
316       // to the event_names (so, when PAPI_read() has finished)!
317       papi_counter_t counters2values;
318
319       // Iterate over all counters that were specified for this specific
320       // unit.
321       // Note that we need to remove the name of the unit
322       // (that could also be the "default" value), which always comes first.
323       // Hence, we start at ++(events.begin())!
324       for (Tokenizer::iterator events_it = ++(event_tokens.begin()); events_it != event_tokens.end(); ++events_it) {
325
326         int event_code   = PAPI_NULL;
327         char* event_name = const_cast<char*>((*events_it).c_str());
328         if (PAPI_event_name_to_code(event_name, &event_code) == PAPI_OK) {
329           if (PAPI_add_event(event_set, event_code) != PAPI_OK) {
330             XBT_ERROR("Could not add PAPI event '%s'. Skipping.", event_name);
331             continue;
332           } else {
333             XBT_DEBUG("Successfully added PAPI event '%s' to the event set.", event_name);
334           }
335         } else {
336           XBT_CRITICAL("Could not find PAPI event '%s'. Skipping.", event_name);
337           continue;
338         }
339
340         counters2values.push_back(
341             // We cannot just pass *events_it, as this is of type const basic_string
342             std::make_pair<std::string, long long>(std::string(*events_it), 0));
343       }
344
345       std::string unit_name    = *(event_tokens.begin());
346       papi_process_data config = {.counter_data = std::move(counters2values), .event_set = event_set};
347
348       units2papi_setup.insert(std::make_pair(unit_name, std::move(config)));
349     }
350   }
351 #endif
352 }
353
354 void smpi_global_destroy()
355 {
356   smpi_bench_destroy();
357   smpi_shared_destroy();
358   smpi_deployment_cleanup_instances();
359
360   if (simgrid::smpi::Colls::smpi_coll_cleanup_callback != nullptr)
361     simgrid::smpi::Colls::smpi_coll_cleanup_callback();
362
363   MPI_COMM_WORLD = MPI_COMM_NULL;
364
365   if (not MC_is_active()) {
366     xbt_os_timer_free(global_timer);
367   }
368
369   if (smpi_privatize_global_variables == SmpiPrivStrategies::MMAP)
370     smpi_destroy_global_memory_segments();
371   smpi_free_static();
372   if(simgrid::smpi::F2C::lookup() != nullptr)
373     simgrid::smpi::F2C::delete_lookup();
374 }
375
376 static void smpi_init_options(){
377   // return if already called
378   if (smpi_cpu_threshold > -1)
379     return;
380   simgrid::smpi::Colls::set_collectives();
381   simgrid::smpi::Colls::smpi_coll_cleanup_callback = nullptr;
382   smpi_cpu_threshold                               = simgrid::config::get_value<double>("smpi/cpu-threshold");
383   smpi_host_speed                                  = simgrid::config::get_value<double>("smpi/host-speed");
384   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);
385   std::string smpi_privatize_option = simgrid::config::get_value<std::string>("smpi/privatization");
386   if (smpi_privatize_option == "no" || smpi_privatize_option == "0")
387     smpi_privatize_global_variables = SmpiPrivStrategies::NONE;
388   else if (smpi_privatize_option == "yes" || smpi_privatize_option == "1")
389     smpi_privatize_global_variables = SmpiPrivStrategies::DEFAULT;
390   else if (smpi_privatize_option == "mmap")
391     smpi_privatize_global_variables = SmpiPrivStrategies::MMAP;
392   else if (smpi_privatize_option == "dlopen")
393     smpi_privatize_global_variables = SmpiPrivStrategies::DLOPEN;
394   else
395     xbt_die("Invalid value for smpi/privatization: '%s'", smpi_privatize_option.c_str());
396
397   if (not SMPI_switch_data_segment) {
398     XBT_DEBUG("Running without smpi_main(); disable smpi/privatization.");
399     smpi_privatize_global_variables = SmpiPrivStrategies::NONE;
400   }
401 #if !HAVE_WORKING_MMAP
402   if (smpi_privatize_global_variables == SmpiPrivStrategies::MMAP) {
403     XBT_INFO("mmap privatization is broken on this platform, switching to dlopen privatization instead.");
404     smpi_privatize_global_variables = SmpiPrivStrategies::DLOPEN;
405   }
406 #endif
407
408   if (smpi_cpu_threshold < 0)
409     smpi_cpu_threshold = DBL_MAX;
410
411   std::string val = simgrid::config::get_value<std::string>("smpi/shared-malloc");
412   if ((val == "yes") || (val == "1") || (val == "on") || (val == "global")) {
413     smpi_cfg_shared_malloc = SharedMallocType::GLOBAL;
414   } else if (val == "local") {
415     smpi_cfg_shared_malloc = SharedMallocType::LOCAL;
416   } else if ((val == "no") || (val == "0") || (val == "off")) {
417     smpi_cfg_shared_malloc = SharedMallocType::NONE;
418   } else {
419     xbt_die("Invalid value '%s' for option smpi/shared-malloc. Possible values: 'on' or 'global', 'local', 'off'",
420             val.c_str());
421   }
422 }
423
424 typedef std::function<int(int argc, char *argv[])> smpi_entry_point_type;
425 typedef int (* smpi_c_entry_point_type)(int argc, char **argv);
426 typedef void (*smpi_fortran_entry_point_type)();
427
428 static int smpi_run_entry_point(smpi_entry_point_type entry_point, std::vector<std::string> args)
429 {
430   // copy C strings, we need them writable
431   std::vector<char*>* args4argv = new std::vector<char*>(args.size());
432   std::transform(begin(args), end(args), begin(*args4argv), [](const std::string& s) { return xbt_strdup(s.c_str()); });
433
434   // take a copy of args4argv to keep reference of the allocated strings
435   const std::vector<char*> args2str(*args4argv);
436   int argc = args4argv->size();
437   args4argv->push_back(nullptr);
438   char** argv = args4argv->data();
439
440   simgrid::smpi::ActorExt::init(&argc, &argv);
441 #if SMPI_IFORT
442   for_rtl_init_ (&argc, argv);
443 #elif SMPI_FLANG
444   __io_set_argc(argc);
445   __io_set_argv(argv);
446 #elif SMPI_GFORTRAN
447   _gfortran_set_args(argc, argv);
448 #endif 
449   int res = entry_point(argc, argv);
450
451 #if SMPI_IFORT
452   for_rtl_finish_ ();
453 #else
454   for (char* s : args2str)
455     xbt_free(s);
456   delete args4argv;
457 #endif
458
459   if (res != 0){
460     XBT_WARN("SMPI process did not return 0. Return value : %d", res);
461     if (smpi_exit_status == 0)
462       smpi_exit_status = res;
463   }
464   return 0;
465 }
466
467
468 // TODO, remove the number of functions involved here
469 static smpi_entry_point_type smpi_resolve_function(void* handle)
470 {
471   smpi_fortran_entry_point_type entry_point_fortran = (smpi_fortran_entry_point_type)dlsym(handle, "user_main_");
472   if (entry_point_fortran != nullptr) {
473     return [entry_point_fortran](int argc, char** argv) {
474       entry_point_fortran();
475       return 0;
476     };
477   }
478
479   smpi_c_entry_point_type entry_point = (smpi_c_entry_point_type)dlsym(handle, "main");
480   if (entry_point != nullptr) {
481     return entry_point;
482   }
483
484   return smpi_entry_point_type();
485 }
486
487 static void smpi_copy_file(std::string src, std::string target, off_t fdin_size)
488 {
489   int fdin = open(src.c_str(), O_RDONLY);
490   xbt_assert(fdin >= 0, "Cannot read from %s. Please make sure that the file exists and is executable.", src.c_str());
491   int fdout = open(target.c_str(), O_CREAT | O_RDWR, S_IRWXU);
492   xbt_assert(fdout >= 0, "Cannot write into %s", target.c_str());
493
494   XBT_DEBUG("Copy %ld bytes into %s", static_cast<long>(fdin_size), target.c_str());
495 #if HAVE_SENDFILE
496   ssize_t sent_size = sendfile(fdout, fdin, NULL, fdin_size);
497   xbt_assert(sent_size == fdin_size, "Error while copying %s: only %zd bytes copied instead of %ld (errno: %d -- %s)",
498              target.c_str(), sent_size, fdin_size, errno, strerror(errno));
499 #else
500   const int bufsize = 1024 * 1024 * 4;
501   char buf[bufsize];
502   while (int got = read(fdin, buf, bufsize)) {
503     if (got == -1) {
504       xbt_assert(errno == EINTR, "Cannot read from %s", src.c_str());
505     } else {
506       char* p  = buf;
507       int todo = got;
508       while (int done = write(fdout, p, todo)) {
509         if (done == -1) {
510           xbt_assert(errno == EINTR, "Cannot write into %s", target.c_str());
511         } else {
512           p += done;
513           todo -= done;
514         }
515       }
516     }
517   }
518 #endif
519   close(fdin);
520   close(fdout);
521 }
522
523 #if not defined(__APPLE__)
524 static int visit_libs(struct dl_phdr_info* info, size_t, void* data)
525 {
526   char* libname = (char*)(data);
527   const char *path = info->dlpi_name;
528   if(strstr(path, libname)){
529     strncpy(libname, path, 512);
530     return 1;
531   }
532   
533   return 0;
534 }
535 #endif
536
537 static void smpi_init_privatization_dlopen(std::string executable)
538 {
539   // Prepare the copy of the binary (get its size)
540   struct stat fdin_stat;
541   stat(executable.c_str(), &fdin_stat);
542   off_t fdin_size         = fdin_stat.st_size;
543   static std::size_t rank = 0;
544
545   std::string libnames = simgrid::config::get_value<std::string>("smpi/privatize-libs");
546   if (not libnames.empty()) {
547     // split option
548     std::vector<std::string> privatize_libs;
549     boost::split(privatize_libs, libnames, boost::is_any_of(";"));
550
551     for (auto const& libname : privatize_libs) {
552       // load the library once to add it to the local libs, to get the absolute path
553       void* libhandle = dlopen(libname.c_str(), RTLD_LAZY);
554       // get library name from path
555       char fullpath[512] = {'\0'};
556       strcpy(fullpath, libname.c_str());
557 #if not defined(__APPLE__)
558       int ret = dl_iterate_phdr(visit_libs, fullpath);
559       if (ret == 0)
560         xbt_die("Can't find a linked %s - check the setting you gave to smpi/privatize-libs", fullpath);
561       else
562         XBT_DEBUG("Extra lib to privatize found : %s", fullpath);
563 #else
564       xbt_die("smpi/privatize-libs is not (yet) compatible with OSX");
565 #endif
566       privatize_libs_paths.push_back(fullpath);
567       dlclose(libhandle);
568     }
569   }
570
571   simix_global->default_function = [executable, fdin_size](std::vector<std::string> args) {
572     return std::function<void()>([executable, fdin_size, args] {
573
574       // Copy the dynamic library:
575       std::string target_executable =
576           executable + "_" + std::to_string(getpid()) + "_" + std::to_string(rank) + ".so";
577
578       smpi_copy_file(executable, target_executable, fdin_size);
579       // if smpi/privatize-libs is set, duplicate pointed lib and link each executable copy to a different one.
580       std::string target_lib;
581       for (auto const& libpath : privatize_libs_paths) {
582         // if we were given a full path, strip it
583         size_t index = libpath.find_last_of("/\\");
584         std::string libname;
585         if (index != std::string::npos)
586           libname = libpath.substr(index + 1);
587
588         if (not libname.empty()) {
589           // load the library to add it to the local libs, to get the absolute path
590           struct stat fdin_stat2;
591           stat(libpath.c_str(), &fdin_stat2);
592           off_t fdin_size2 = fdin_stat2.st_size;
593
594           // Copy the dynamic library, the new name must be the same length as the old one
595           // just replace the name with 7 digits for the rank and the rest of the name.
596           unsigned int pad = 7;
597           if (libname.length() < pad)
598             pad = libname.length();
599           target_lib =
600               std::string(pad - std::to_string(rank).length(), '0') + std::to_string(rank) + libname.substr(pad);
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 | 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         if (not target_lib.empty())
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 }
746
747 double smpi_mpi_wtime(){
748   double time;
749   if (smpi_process()->initialized() != 0 && smpi_process()->finalized() == 0 && smpi_process()->sampling() == 0) {
750     smpi_bench_end();
751     time = SIMIX_get_clock();
752     // to avoid deadlocks if used as a break condition, such as
753     //     while (MPI_Wtime(...) < time_limit) {
754     //       ....
755     //     }
756     // because the time will not normally advance when only calls to MPI_Wtime
757     // are made -> deadlock (MPI_Wtime never reaches the time limit)
758     if(smpi_wtime_sleep > 0)
759       simcall_process_sleep(smpi_wtime_sleep);
760     smpi_bench_begin();
761   } else {
762     time = SIMIX_get_clock();
763   }
764   return time;
765 }
766