#include "schedule.h"
#include "clockvector.h"
-#ifdef TLS
#include <dlfcn.h>
+
+#ifdef TLS
uintptr_t get_tls_addr() {
uintptr_t addr;
asm ("mov %%fs:0, %0" : "=r" (addr));
return model->get_current_thread();
}
+/**
+ * @brief Get the current Thread id
+ *
+ * Must be called from a user context
+ *
+ * @return The id of the currently executing thread
+ */
+thread_id_t thread_current_id(void)
+{
+ ASSERT(model);
+ return model->get_current_thread_id();
+}
+
void modelexit() {
model->switch_thread(new ModelAction(THREAD_FINISH, std::memory_order_seq_cst, thread_current()));
}
#endif
}
+
+static int (*real_epoll_wait_p)(int epfd, struct epoll_event *events, int maxevents, int timeout) = NULL;
+
+int real_epoll_wait(int epfd, struct epoll_event *events, int maxevents, int timeout) {
+ return real_epoll_wait_p(epfd, events, maxevents, timeout);
+}
+
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) {
void real_init_all() {
char * error;
+ if (!real_epoll_wait_p) {
+ real_epoll_wait_p = (int (*)(int epfd, struct epoll_event *events, int maxevents, int timeout))dlsym(RTLD_NEXT, "epoll_wait");
+ if ((error = dlerror()) != NULL) {
+ fputs(error, stderr);
+ exit(EXIT_FAILURE);
+ }
+ }
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");
if ((error = dlerror()) != NULL) {
int Thread::swap(Thread *t, Thread *t2)
{
- t->set_state(THREAD_READY);
t2->set_state(THREAD_RUNNING);
if (t == t2)
return 0;
stack_free(stack);
#ifdef TLS
if (this != model->getInitThread()) {
- ASSERT(thread_current()==NULL);
real_pthread_mutex_unlock(&mutex2);
real_pthread_join(thread, NULL);
stack_free(helper_stack);
}
#endif
+ state = THREAD_FREED;
}
/**
acq_fence_cv(new ClockVector()),
creation(NULL),
pending(NULL),
+ wakeup_state(false),
start_routine(NULL),
arg(NULL),
stack(NULL),
last_action_val(0),
model_thread(true)
{
+ // real_memset is not defined when
+ // the model thread is constructed
memset(&context, 0, sizeof(context));
}
acq_fence_cv(new ClockVector()),
creation(NULL),
pending(NULL),
+ wakeup_state(false),
start_routine(func),
pstart_routine(NULL),
arg(a),
acq_fence_cv(new ClockVector()),
creation(NULL),
pending(NULL),
+ wakeup_state(false),
start_routine(NULL),
pstart_routine(func),
arg(a),
if (!pending)
return NULL;
- if (pending->get_type() == THREAD_JOIN)
- return pending->get_thread_operand();
- else if (pending->get_type() == PTHREAD_JOIN)
- return pending->get_thread_operand();
- else if (pending->is_lock())
- return (Thread *)pending->get_mutex()->get_state()->locked;
- return NULL;
+ switch (pending->get_type()) {
+ case THREAD_JOIN:
+ case PTHREAD_JOIN:
+ return pending->get_thread_operand();
+ case ATOMIC_LOCK:
+ return (Thread *)pending->get_mutex()->get_state()->locked;
+ default:
+ return NULL;
+ }
}
/**