Logo AND Algorithmique Numérique Distribuée

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