/* global "model" object */
#include "model.h"
#include "execution.h"
+#include "schedule.h"
#ifdef TLS
#include <dlfcn.h>
return model->get_current_thread();
}
-void main_thread_startup() {
-#ifdef TLS
+void modelexit() {
+ model->switch_to_master(new ModelAction(THREAD_FINISH, std::memory_order_seq_cst, thread_current()));
+}
+
+void initMainThread() {
+ atexit(modelexit);
Thread * curr_thread = thread_current();
- /* Add dummy "start" action, just to create a first clock vector */
model->switch_to_master(new ModelAction(THREAD_START, std::memory_order_seq_cst, curr_thread));
-#endif
- thread_startup();
}
/**
void *retval = curr_thread->pstart_routine(curr_thread->arg);
curr_thread->set_pthread_return(retval);
}
+#ifndef TLS
/* Finish thread properly */
model->switch_to_master(new ModelAction(THREAD_FINISH, std::memory_order_seq_cst, curr_thread));
+#endif
}
-#ifdef TLS
+static int (*pthread_mutex_init_p)(pthread_mutex_t *__mutex, const pthread_mutexattr_t *__mutexattr) = NULL;
+
int real_pthread_mutex_init(pthread_mutex_t *__mutex, const pthread_mutexattr_t *__mutexattr) {
- static int (*pthread_mutex_init_p) (pthread_mutex_t *__mutex, const pthread_mutexattr_t *__mutexattr) = NULL;
+ return pthread_mutex_init_p(__mutex, __mutexattr);
+}
+
+static int (*pthread_mutex_lock_p) (pthread_mutex_t *__mutex) = NULL;
+
+int real_pthread_mutex_lock (pthread_mutex_t *__mutex) {
+ return pthread_mutex_lock_p(__mutex);
+}
+
+static int (*pthread_mutex_unlock_p) (pthread_mutex_t *__mutex) = NULL;
+
+int real_pthread_mutex_unlock (pthread_mutex_t *__mutex) {
+ return pthread_mutex_unlock_p(__mutex);
+}
+
+static int (*pthread_create_p) (pthread_t *__restrict, const pthread_attr_t *__restrict, void *(*)(void *), void * __restrict) = NULL;
+
+int real_pthread_create (pthread_t *__restrict __newthread, const pthread_attr_t *__restrict __attr, void *(*__start_routine)(void *), void *__restrict __arg) {
+ return pthread_create_p(__newthread, __attr, __start_routine, __arg);
+}
+
+static int (*pthread_join_p) (pthread_t __th, void ** __thread_return) = NULL;
+
+int real_pthread_join (pthread_t __th, void ** __thread_return) {
+ return pthread_join_p(__th, __thread_return);
+}
+
+static void (*pthread_exit_p)(void *) __attribute__((noreturn))= NULL;
+
+void real_pthread_exit (void * value_ptr) {
+ pthread_exit_p(value_ptr);
+}
+
+void real_init_all() {
char * error;
if (!pthread_mutex_init_p) {
pthread_mutex_init_p = (int (*)(pthread_mutex_t *__mutex, const pthread_mutexattr_t *__mutexattr))dlsym(RTLD_NEXT, "pthread_mutex_init");
exit(EXIT_FAILURE);
}
}
- return pthread_mutex_init_p(__mutex, __mutexattr);
-}
-
-int real_pthread_mutex_lock (pthread_mutex_t *__mutex) {
- static int (*pthread_mutex_lock_p) (pthread_mutex_t *__mutex) = NULL;
- char * error;
if (!pthread_mutex_lock_p) {
pthread_mutex_lock_p = (int (*)(pthread_mutex_t *__mutex))dlsym(RTLD_NEXT, "pthread_mutex_lock");
if ((error = dlerror()) != NULL) {
exit(EXIT_FAILURE);
}
}
- return pthread_mutex_lock_p(__mutex);
-}
-
-int real_pthread_mutex_unlock (pthread_mutex_t *__mutex) {
- static int (*pthread_mutex_unlock_p) (pthread_mutex_t *__mutex) = NULL;
- char * error;
if (!pthread_mutex_unlock_p) {
pthread_mutex_unlock_p = (int (*)(pthread_mutex_t *__mutex))dlsym(RTLD_NEXT, "pthread_mutex_unlock");
if ((error = dlerror()) != NULL) {
exit(EXIT_FAILURE);
}
}
- return pthread_mutex_unlock_p(__mutex);
-}
-
-int real_pthread_create (pthread_t *__restrict __newthread, const pthread_attr_t *__restrict __attr, void *(*__start_routine)(void *), void *__restrict __arg) {
- static int (*pthread_create_p) (pthread_t *__restrict, const pthread_attr_t *__restrict, void *(*)(void *), void * __restrict) = NULL;
- char * error;
if (!pthread_create_p) {
pthread_create_p = (int (*)(pthread_t *__restrict, const pthread_attr_t *__restrict, void *(*)(void *), void *__restrict))dlsym(RTLD_NEXT, "pthread_create");
if ((error = dlerror()) != NULL) {
exit(EXIT_FAILURE);
}
}
- return pthread_create_p(__newthread, __attr, __start_routine, __arg);
-}
-
-int real_pthread_join (pthread_t __th, void ** __thread_return) {
- static int (*pthread_join_p) (pthread_t __th, void ** __thread_return) = NULL;
- char * error;
if (!pthread_join_p) {
pthread_join_p = (int (*)(pthread_t __th, void ** __thread_return))dlsym(RTLD_NEXT, "pthread_join");
if ((error = dlerror()) != NULL) {
exit(EXIT_FAILURE);
}
}
- return pthread_join_p(__th, __thread_return);
+
+ if (!pthread_exit_p) {
+ *((void (**)(void *)) &pthread_exit_p) = (void (*)(void *))dlsym(RTLD_NEXT, "pthread_exit");
+ if ((error = dlerror()) != NULL) {
+ fputs(error, stderr);
+ exit(EXIT_FAILURE);
+ }
+ }
}
+#ifdef TLS
void finalize_helper_thread() {
Thread * curr_thread = thread_current();
- model_print("finalize_helper_thread\n");
real_pthread_mutex_lock(&curr_thread->mutex);
curr_thread->tls = (char *) get_tls_addr();
real_pthread_mutex_unlock(&curr_thread->mutex);
void * helper_thread(void * ptr) {
Thread * curr_thread = thread_current();
- model_print("helper_thread\n");
//build a context for this real thread so we can take it's context
int ret = getcontext(&curr_thread->helpercontext);
ASSERT(!ret);
+ //Setup destructor
+ if (pthread_setspecific(model->get_execution()->getPthreadKey(), (const void *)4)) {
+ printf("Destructor setup failed\n");
+ exit(-1);
+ }
+
+
/* Initialize new managed context */
- void *helperstack = stack_allocate(STACK_SIZE);
- curr_thread->helpercontext.uc_stack.ss_sp = helperstack;
+ curr_thread->helper_stack = stack_allocate(STACK_SIZE);
+ curr_thread->helpercontext.uc_stack.ss_sp = curr_thread->helper_stack;
curr_thread->helpercontext.uc_stack.ss_size = STACK_SIZE;
curr_thread->helpercontext.uc_stack.ss_flags = 0;
curr_thread->helpercontext.uc_link = model->get_system_context();
model_swapcontext(&curr_thread->context, &curr_thread->helpercontext);
+
//start the real thread
thread_startup();
- //now the real thread has control again
- stack_free(helperstack);
-
return NULL;
}
+#ifdef TLS
+void tlsdestructor(void *v) {
+ uintptr_t count = (uintptr_t) v;
+ if (count > 1) {
+ if (pthread_setspecific(model->get_execution()->getPthreadKey(), (const void *)(count - 1))) {
+ printf("Destructor setup failed\n");
+ exit(-1);
+ }
+ return;
+ }
+ /* Finish thread properly */
+ model->switch_to_master(new ModelAction(THREAD_FINISH, std::memory_order_seq_cst, thread_current()));
+}
+#endif
+
void setup_context() {
Thread * curr_thread = thread_current();
- model_print("setup_context\n");
/* Add dummy "start" action, just to create a first clock vector */
model->switch_to_master(new ModelAction(THREAD_START, std::memory_order_seq_cst, curr_thread));
+ real_init_all();
+
/* Initialize our lock */
real_pthread_mutex_init(&curr_thread->mutex, NULL);
real_pthread_mutex_init(&curr_thread->mutex2, NULL);
/* Create the real thread */
real_pthread_create(&curr_thread->thread, NULL, helper_thread, NULL);
- model_print("thread_created\n");
bool notdone = true;
while(notdone) {
real_pthread_mutex_lock(&curr_thread->mutex);
}
set_tls_addr((uintptr_t)curr_thread->tls);
- model_print("tls taken\n");
setcontext(&curr_thread->context);
}
#endif
context.uc_stack.ss_flags = 0;
context.uc_link = model->get_system_context();
#ifdef TLS
- if (model != NULL)
- makecontext(&context, setup_context, 0);
- else
- makecontext(&context, thread_startup, 0);
+ makecontext(&context, setup_context, 0);
#else
makecontext(&context, thread_startup, 0);
#endif
stack_free(stack);
#ifdef TLS
if (this != model->getInitThread()) {
- modellock = 1;
+ ASSERT(thread_current()==NULL);
real_pthread_mutex_unlock(&mutex2);
real_pthread_join(thread, NULL);
- modellock = 0;
+ stack_free(helper_stack);
}
#endif
}