1 /* Copyright (c) 2002-2023. The SimGrid Team. All rights reserved. */
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. */
6 /* SimGrid's pthread interposer. Actual implementation of the symbols (see the comment in sthread.h) */
9 #include "xbt/string.hpp"
10 #include <simgrid/actor.h>
11 #include <simgrid/s4u/Actor.hpp>
12 #include <simgrid/s4u/Engine.hpp>
13 #include <simgrid/s4u/Mutex.hpp>
14 #include <simgrid/s4u/NetZone.hpp>
15 #include <simgrid/s4u/Semaphore.hpp>
17 #include <xbt/sysdep.h>
19 #include "src/internal_config.h"
20 #include "src/sthread/sthread.h"
25 #include <semaphore.h>
29 #include <string_view>
32 XBT_LOG_NEW_DEFAULT_CATEGORY(sthread, "pthread intercepter");
33 namespace sg4 = simgrid::s4u;
35 static sg4::Host* lilibeth = nullptr;
37 int sthread_main(int argc, char** argv, char** envp, int (*raw_main)(int, char**, char**))
39 /* Do not intercept the main when run from SMPI: it will initialize the simulation properly */
40 for (int i = 0; envp[i] != nullptr; i++)
41 if (std::string_view(envp[i]).rfind("SMPI_GLOBAL_SIZE", 0) == 0)
42 return raw_main(argc, argv, envp);
44 /* If not in SMPI, the old main becomes an actor in a newly created simulation */
45 std::ostringstream id;
46 id << std::this_thread::get_id();
48 XBT_DEBUG("sthread main() is starting in thread %s", id.str().c_str());
50 sg4::Engine e(&argc, argv);
51 auto* zone = sg4::create_full_zone("world");
52 lilibeth = zone->create_host("Lilibeth", 1e15);
55 /* Launch the user's main() on an actor */
57 sg4::ActorPtr main_actor = sg4::Actor::create("main thread", lilibeth, raw_main, argc, argv, envp);
59 XBT_INFO("Starting the simulation.");
60 sg4::Engine::get_instance()->run();
62 XBT_INFO("All threads exited. Terminating the simulation.");
67 struct sthread_mutex {
71 int sthread_create(unsigned long int* thread, const void* /*pthread_attr_t* attr*/, void* (*start_routine)(void*),
76 XBT_VERB("Create thread %d", TID);
77 std::string name = std::string("thread ") + std::to_string(TID);
79 if (SMPI_is_inited()) {
81 MPI_Comm_rank(MPI_COMM_WORLD, &rank);
82 name = simgrid::xbt::string_printf("%d:%d", rank, TID);
85 sg4::ActorPtr actor = sg4::Actor::create(
87 [](auto* user_function, auto* param) {
98 intrusive_ptr_add_ref(actor.get());
99 *thread = reinterpret_cast<unsigned long>(actor.get());
102 int sthread_join(sthread_t thread, void** /*retval*/)
104 sg4::ActorPtr actor(reinterpret_cast<sg4::Actor*>(thread));
106 intrusive_ptr_release(actor.get());
111 int sthread_mutex_init(sthread_mutex_t* mutex, const void* /*pthread_mutexattr_t* attr*/)
113 auto m = sg4::Mutex::create();
114 intrusive_ptr_add_ref(m.get());
116 mutex->mutex = m.get();
120 int sthread_mutex_lock(sthread_mutex_t* mutex)
122 /* At least in glibc, PTHREAD_STATIC_INITIALIZER sets every fields to 0 */
123 if (mutex->mutex == nullptr)
124 sthread_mutex_init(mutex, nullptr);
126 static_cast<sg4::Mutex*>(mutex->mutex)->lock();
130 int sthread_mutex_trylock(sthread_mutex_t* mutex)
132 /* At least in glibc, PTHREAD_STATIC_INITIALIZER sets every fields to 0 */
133 if (mutex->mutex == nullptr)
134 sthread_mutex_init(mutex, nullptr);
136 return static_cast<sg4::Mutex*>(mutex->mutex)->try_lock();
139 int sthread_mutex_unlock(sthread_mutex_t* mutex)
141 /* At least in glibc, PTHREAD_STATIC_INITIALIZER sets every fields to 0 */
142 if (mutex->mutex == nullptr)
143 sthread_mutex_init(mutex, nullptr);
145 static_cast<sg4::Mutex*>(mutex->mutex)->unlock();
148 int sthread_mutex_destroy(sthread_mutex_t* mutex)
150 /* At least in glibc, PTHREAD_STATIC_INITIALIZER sets every fields to 0 */
151 if (mutex->mutex == nullptr)
152 sthread_mutex_init(mutex, nullptr);
154 intrusive_ptr_release(static_cast<sg4::Mutex*>(mutex->mutex));
157 int sthread_sem_init(sthread_sem_t* sem, int /*pshared*/, unsigned int value)
159 auto s = sg4::Semaphore::create(value);
160 intrusive_ptr_add_ref(s.get());
165 int sthread_sem_destroy(sthread_sem_t* sem)
167 intrusive_ptr_release(static_cast<sg4::Semaphore*>(sem->sem));
170 int sthread_sem_post(sthread_sem_t* sem)
172 static_cast<sg4::Semaphore*>(sem->sem)->release();
175 int sthread_sem_wait(sthread_sem_t* sem)
177 static_cast<sg4::Semaphore*>(sem->sem)->acquire();
180 int sthread_sem_trywait(sthread_sem_t* sem)
182 auto* s = static_cast<sg4::Semaphore*>(sem->sem);
183 if (s->would_block()) {
190 int sthread_sem_timedwait(sthread_sem_t* sem, const struct timespec* abs_timeout)
192 if (static_cast<sg4::Semaphore*>(sem->sem)->acquire_timeout(static_cast<double>(abs_timeout->tv_sec) +
193 static_cast<double>(abs_timeout->tv_nsec) / 1E9)) {
200 int sthread_gettimeofday(struct timeval* tv)
203 double now = simgrid::s4u::Engine::get_clock();
204 double secs = trunc(now);
205 double usecs = (now - secs) * 1e6;
206 tv->tv_sec = static_cast<time_t>(secs);
207 tv->tv_usec = static_cast<decltype(tv->tv_usec)>(usecs); // suseconds_t
212 void sthread_sleep(double seconds)
214 simgrid::s4u::this_actor::sleep_for(seconds);
218 int pthread_cond_init(pthread_cond_t *cond, pthread_condattr_t *cond_attr) {
219 *cond = sg_cond_init();
223 int pthread_cond_signal(pthread_cond_t *cond) {
224 sg_cond_notify_one(*cond);
228 int pthread_cond_broadcast(pthread_cond_t *cond) {
229 sg_cond_notify_all(*cond);
233 int pthread_cond_wait(pthread_cond_t *cond, pthread_mutex_t *mutex) {
234 sg_cond_wait(*cond, *mutex);
238 int pthread_cond_destroy(pthread_cond_t *cond) {
239 sg_cond_destroy(*cond);