X-Git-Url: http://bilbo.iut-bm.univ-fcomte.fr/pub/gitweb/simgrid.git/blobdiff_plain/9b989509fa80930cd2b4cfa9b8253bc1d34dc655..3c7c64745aa5e60415bb85af482c7b0d0fca2b2b:/src/sthread/sthread.c diff --git a/src/sthread/sthread.c b/src/sthread/sthread.c index a444e23f68..2005f297ad 100644 --- a/src/sthread/sthread.c +++ b/src/sthread/sthread.c @@ -1,3 +1,8 @@ +/* Copyright (c) 2002-2023. The SimGrid Team. All rights reserved. */ + +/* This program is free software; you can redistribute it and/or modify it + * under the terms of the license (GNU LGPL) which comes with this package. */ + /* SimGrid's pthread interposer. Redefinition of the pthread symbols (see the comment in sthread.h) */ #define _GNU_SOURCE @@ -14,42 +19,60 @@ #include #endif -/* We don't want to intercept pthread within simgrid. Instead we should provide the real implem to simgrid */ +/* We don't want to intercept pthread within SimGrid. Instead we should provide the real implem to SimGrid */ static int (*raw_pthread_create)(pthread_t*, const pthread_attr_t*, void* (*)(void*), void*); static int (*raw_pthread_join)(pthread_t, void**); -static int (*raw_mutex_init)(pthread_mutex_t*, const pthread_mutexattr_t*) = NULL; -static int (*raw_mutex_lock)(pthread_mutex_t*) = NULL; -static int (*raw_mutex_trylock)(pthread_mutex_t*) = NULL; -static int (*raw_mutex_unlock)(pthread_mutex_t*) = NULL; -static int (*raw_mutex_destroy)(pthread_mutex_t*) = NULL; - -static unsigned int (*raw_sleep)(unsigned int) = NULL; -static int (*raw_usleep)(useconds_t) = NULL; -static int (*raw_gettimeofday)(struct timeval*, void*) = NULL; - -static sem_t* (*raw_sem_open)(const char*, int) = NULL; -static int (*raw_sem_init)(sem_t*, int, unsigned int) = NULL; -static int (*raw_sem_wait)(sem_t*) = NULL; -static int (*raw_sem_post)(sem_t*) = NULL; +static int (*raw_mutex_init)(pthread_mutex_t*, const pthread_mutexattr_t*); +static int (*raw_mutex_lock)(pthread_mutex_t*); +static int (*raw_mutex_trylock)(pthread_mutex_t*); +static int (*raw_mutex_unlock)(pthread_mutex_t*); +static int (*raw_mutex_destroy)(pthread_mutex_t*); + +static int (*raw_pthread_mutexattr_init)(pthread_mutexattr_t*); +static int (*raw_pthread_mutexattr_settype)(pthread_mutexattr_t*, int); +static int (*raw_pthread_mutexattr_gettype)(const pthread_mutexattr_t* restrict, int* restrict); +static int (*raw_pthread_mutexattr_getrobust)(const pthread_mutexattr_t*, int*); +static int (*raw_pthread_mutexattr_setrobust)(pthread_mutexattr_t*, int); + +static unsigned int (*raw_sleep)(unsigned int); +static int (*raw_usleep)(useconds_t); +static int (*raw_gettimeofday)(struct timeval*, void*); + +static sem_t* (*raw_sem_open)(const char*, int); +static int (*raw_sem_init)(sem_t*, int, unsigned int); +static int (*raw_sem_wait)(sem_t*); +static int (*raw_sem_post)(sem_t*); +static int (*raw_sem_destroy)(sem_t*); +static int (*raw_sem_trywait)(sem_t*); +static int (*raw_sem_timedwait)(sem_t*, const struct timespec*); static void intercepter_init() { - raw_pthread_create = (typeof(raw_pthread_create))dlsym(RTLD_NEXT, "pthread_create"); - raw_pthread_join = (typeof(raw_pthread_join))dlsym(RTLD_NEXT, "pthread_join"); - raw_mutex_init = (int (*)(pthread_mutex_t*, const pthread_mutexattr_t*))dlsym(RTLD_NEXT, "pthread_mutex_init"); - raw_mutex_lock = (int (*)(pthread_mutex_t*))dlsym(RTLD_NEXT, "pthread_mutex_lock"); - raw_mutex_trylock = (int (*)(pthread_mutex_t*))dlsym(RTLD_NEXT, "pthread_mutex_trylock"); - raw_mutex_unlock = (int (*)(pthread_mutex_t*))dlsym(RTLD_NEXT, "pthread_mutex_unlock"); - raw_mutex_destroy = (int (*)(pthread_mutex_t*))dlsym(RTLD_NEXT, "pthread_mutex_destroy"); - - raw_sleep = (unsigned int (*)(unsigned int))dlsym(RTLD_NEXT, "sleep"); - raw_usleep = (int (*)(useconds_t usec))dlsym(RTLD_NEXT, "usleep"); - raw_gettimeofday = (int (*)(struct timeval*, void*))dlsym(RTLD_NEXT, "gettimeofday"); - - raw_sem_open = (sem_t * (*)(const char*, int)) dlsym(RTLD_NEXT, "sem_open"); - raw_sem_init = (int (*)(sem_t*, int, unsigned int))dlsym(RTLD_NEXT, "sem_init"); - raw_sem_wait = (int (*)(sem_t*))dlsym(RTLD_NEXT, "sem_wait"); - raw_sem_post = (int (*)(sem_t*))dlsym(RTLD_NEXT, "sem_post"); + raw_pthread_create = dlsym(RTLD_NEXT, "pthread_create"); + raw_pthread_join = dlsym(RTLD_NEXT, "pthread_join"); + raw_mutex_init = dlsym(RTLD_NEXT, "pthread_mutex_init"); + raw_mutex_lock = dlsym(RTLD_NEXT, "pthread_mutex_lock"); + raw_mutex_trylock = dlsym(RTLD_NEXT, "pthread_mutex_trylock"); + raw_mutex_unlock = dlsym(RTLD_NEXT, "pthread_mutex_unlock"); + raw_mutex_destroy = dlsym(RTLD_NEXT, "pthread_mutex_destroy"); + + raw_pthread_mutexattr_init = dlsym(RTLD_NEXT, "pthread_mutexattr_init"); + raw_pthread_mutexattr_settype = dlsym(RTLD_NEXT, "pthread_mutexattr_settype"); + raw_pthread_mutexattr_gettype = dlsym(RTLD_NEXT, "pthread_mutexattr_gettype"); + raw_pthread_mutexattr_getrobust = dlsym(RTLD_NEXT, "pthread_mutexattr_getrobust"); + raw_pthread_mutexattr_setrobust = dlsym(RTLD_NEXT, "pthread_mutexattr_setrobust"); + + raw_sleep = dlsym(RTLD_NEXT, "sleep"); + raw_usleep = dlsym(RTLD_NEXT, "usleep"); + raw_gettimeofday = dlsym(RTLD_NEXT, "gettimeofday"); + + raw_sem_open = dlsym(RTLD_NEXT, "sem_open"); + raw_sem_init = dlsym(RTLD_NEXT, "sem_init"); + raw_sem_wait = dlsym(RTLD_NEXT, "sem_wait"); + raw_sem_post = dlsym(RTLD_NEXT, "sem_post"); + raw_sem_destroy = dlsym(RTLD_NEXT, "sem_destroy"); + raw_sem_trywait = dlsym(RTLD_NEXT, "sem_trywait"); + raw_sem_timedwait = dlsym(RTLD_NEXT, "sem_timedwait"); } static int sthread_inside_simgrid = 1; @@ -69,22 +92,47 @@ int pthread_create(pthread_t* thread, const pthread_attr_t* attr, void* (*start_ if (sthread_inside_simgrid) return raw_pthread_create(thread, attr, start_routine, arg); - sthread_inside_simgrid = 1; - int res = sthread_create((sthread_t*)thread, attr, start_routine, arg); - sthread_inside_simgrid = 0; + sthread_disable(); + int res = sthread_create((sthread_t*)thread, attr, start_routine, arg); + sthread_enable(); return res; } + +#define _STHREAD_CONCAT(a, b) a##b +#define intercepted_call(name, raw_params, call_params, sim_params) \ + int _STHREAD_CONCAT(pthread_, name) raw_params \ + { \ + if (_STHREAD_CONCAT(raw_pthread_, name) == NULL) \ + intercepter_init(); \ + if (sthread_inside_simgrid) \ + return _STHREAD_CONCAT(raw_pthread_, name) call_params; \ + \ + sthread_disable(); \ + int res = _STHREAD_CONCAT(sthread_, name) sim_params; \ + sthread_enable(); \ + return res; \ + } + +intercepted_call(mutexattr_init, (pthread_mutexattr_t * attr), (attr), ((sthread_mutexattr_t*)attr)); +intercepted_call(mutexattr_settype, (pthread_mutexattr_t * attr, int type), (attr, type), + ((sthread_mutexattr_t*)attr, type)); +intercepted_call(mutexattr_gettype, (const pthread_mutexattr_t* restrict attr, int* type), (attr, type), + ((sthread_mutexattr_t*)attr, type)); +intercepted_call(mutexattr_setrobust, (pthread_mutexattr_t* restrict attr, int robustness), (attr, robustness), + ((sthread_mutexattr_t*)attr, robustness)); +intercepted_call(mutexattr_getrobust, (const pthread_mutexattr_t* restrict attr, int* restrict robustness), + (attr, robustness), ((sthread_mutexattr_t*)attr, robustness)); + int pthread_join(pthread_t thread, void** retval) { if (raw_pthread_join == NULL) intercepter_init(); - if (sthread_inside_simgrid) return raw_pthread_join(thread, retval); - sthread_inside_simgrid = 1; - int res = sthread_join((sthread_t)thread, retval); - sthread_inside_simgrid = 0; + sthread_disable(); + int res = sthread_join((sthread_t)thread, retval); + sthread_enable(); return res; } @@ -96,9 +144,9 @@ int pthread_mutex_init(pthread_mutex_t* mutex, const pthread_mutexattr_t* attr) if (sthread_inside_simgrid) return raw_mutex_init(mutex, attr); - sthread_inside_simgrid = 1; - int res = sthread_mutex_init((sthread_mutex_t*)mutex, attr); - sthread_inside_simgrid = 0; + sthread_disable(); + int res = sthread_mutex_init((sthread_mutex_t*)mutex, (sthread_mutexattr_t*)attr); + sthread_enable(); return res; } @@ -110,9 +158,9 @@ int pthread_mutex_lock(pthread_mutex_t* mutex) if (sthread_inside_simgrid) return raw_mutex_lock(mutex); - sthread_inside_simgrid = 1; - int res = sthread_mutex_lock((sthread_mutex_t*)mutex); - sthread_inside_simgrid = 0; + sthread_disable(); + int res = sthread_mutex_lock((sthread_mutex_t*)mutex); + sthread_enable(); return res; } @@ -124,9 +172,9 @@ int pthread_mutex_trylock(pthread_mutex_t* mutex) if (sthread_inside_simgrid) return raw_mutex_trylock(mutex); - sthread_inside_simgrid = 1; - int res = sthread_mutex_trylock((sthread_mutex_t*)mutex); - sthread_inside_simgrid = 0; + sthread_disable(); + int res = sthread_mutex_trylock((sthread_mutex_t*)mutex); + sthread_enable(); return res; } @@ -138,9 +186,9 @@ int pthread_mutex_unlock(pthread_mutex_t* mutex) if (sthread_inside_simgrid) return raw_mutex_unlock(mutex); - sthread_inside_simgrid = 1; - int res = sthread_mutex_unlock((sthread_mutex_t*)mutex); - sthread_inside_simgrid = 0; + sthread_disable(); + int res = sthread_mutex_unlock((sthread_mutex_t*)mutex); + sthread_enable(); return res; } int pthread_mutex_destroy(pthread_mutex_t* mutex) @@ -151,9 +199,87 @@ int pthread_mutex_destroy(pthread_mutex_t* mutex) if (sthread_inside_simgrid) return raw_mutex_destroy(mutex); - sthread_inside_simgrid = 1; - int res = sthread_mutex_destroy((sthread_mutex_t*)mutex); - sthread_inside_simgrid = 0; + sthread_disable(); + int res = sthread_mutex_destroy((sthread_mutex_t*)mutex); + sthread_enable(); + return res; +} +int sem_init(sem_t* sem, int pshared, unsigned int value) +{ + if (raw_sem_init == NULL) + intercepter_init(); + + if (sthread_inside_simgrid) + return raw_sem_init(sem, pshared, value); + + sthread_disable(); + int res = sthread_sem_init((sthread_sem_t*)sem, pshared, value); + sthread_enable(); + return res; +} +int sem_destroy(sem_t* sem) +{ + if (raw_sem_destroy == NULL) + intercepter_init(); + + if (sthread_inside_simgrid) + return raw_sem_destroy(sem); + + sthread_disable(); + int res = sthread_sem_destroy((sthread_sem_t*)sem); + sthread_enable(); + return res; +} +int sem_post(sem_t* sem) +{ + if (raw_sem_post == NULL) + intercepter_init(); + + if (sthread_inside_simgrid) + return raw_sem_post(sem); + + sthread_disable(); + int res = sthread_sem_post((sthread_sem_t*)sem); + sthread_enable(); + return res; +} +int sem_wait(sem_t* sem) +{ + if (raw_sem_wait == NULL) + intercepter_init(); + + if (sthread_inside_simgrid) + return raw_sem_wait(sem); + + sthread_disable(); + int res = sthread_sem_wait((sthread_sem_t*)sem); + sthread_enable(); + return res; +} +int sem_trywait(sem_t* sem) +{ + if (raw_sem_trywait == NULL) + intercepter_init(); + + if (sthread_inside_simgrid) + return raw_sem_trywait(sem); + + sthread_disable(); + int res = sthread_sem_trywait((sthread_sem_t*)sem); + sthread_enable(); + return res; +} +int sem_timedwait(sem_t* sem, const struct timespec* abs_timeout) +{ + if (raw_sem_timedwait == NULL) + intercepter_init(); + + if (sthread_inside_simgrid) + return raw_sem_timedwait(sem, abs_timeout); + + sthread_disable(); + int res = sthread_sem_timedwait((sthread_sem_t*)sem, abs_timeout); + sthread_enable(); return res; } @@ -176,9 +302,9 @@ int gettimeofday(struct timeval* tv, XBT_ATTRIB_UNUSED TIMEZONE_TYPE* tz) if (sthread_inside_simgrid) return raw_gettimeofday(tv, tz); - sthread_inside_simgrid = 1; - int res = sthread_gettimeofday(tv); - sthread_inside_simgrid = 0; + sthread_disable(); + int res = sthread_gettimeofday(tv); + sthread_enable(); return res; } @@ -190,9 +316,9 @@ unsigned int sleep(unsigned int seconds) if (sthread_inside_simgrid) return raw_sleep(seconds); - sthread_inside_simgrid = 1; + sthread_disable(); sthread_sleep(seconds); - sthread_inside_simgrid = 0; + sthread_enable(); return 0; } @@ -204,36 +330,13 @@ int usleep(useconds_t usec) if (sthread_inside_simgrid) return raw_usleep(usec); - sthread_inside_simgrid = 1; + sthread_disable(); sthread_sleep(((double)usec) / 1000000.); - sthread_inside_simgrid = 0; + sthread_enable(); return 0; } #if 0 -int sem_init(sem_t *sem, int pshared, unsigned int value) { - int res; - - res=raw_sem_init(sem,pshared,value); - return res; -} - -int sem_wait(sem_t *sem) { - int res; - - res = raw_sem_wait(sem); - return res; -} - -int sem_post(sem_t *sem) { - return raw_sem_post(sem); -} - -int pthread_join(pthread_t thread, void **retval) { - sg_actor_join(thread, -1); - return 0; -} - int pthread_cond_init(pthread_cond_t *cond, pthread_condattr_t *cond_attr) { *cond = sg_cond_init(); return 0;