radare2/libr/util/thread.c
2015-07-14 13:55:58 +02:00

129 lines
2.3 KiB
C

/* radare - LGPL - Copyright 2009-2012 - pancake */
#include <r_th.h>
static void *_r_th_launcher(void *_th) {
int ret;
struct r_th_t *th = _th;
th->ready = R_TRUE;
#if __UNIX__
if (th->delay>0) sleep(th->delay);
else if (th->delay<0) r_th_lock_wait(th->lock);
#else
if (th->delay<0) r_th_lock_wait(th->lock);
#endif
do {
r_th_lock_leave(th->lock);
th->running = R_TRUE;
ret = th->fun(th);
th->running = R_FALSE;
r_th_lock_enter(th->lock);
} while(ret);
#if HAVE_PTHREAD
pthread_exit(&ret);
#endif
return 0;
}
R_API int r_th_push_task(struct r_th_t *th, void *user) {
int ret = R_TRUE;
th->user = user;
r_th_lock_leave(th->lock);
return ret;
}
R_API struct r_th_t *r_th_new(R_TH_FUNCTION(fun), void *user, int delay) {
RThread *th = R_NEW0 (RThread);
if (th) {
th->lock = r_th_lock_new();
th->running = R_FALSE;
th->fun = fun;
th->user = user;
th->delay = delay;
th->breaked = R_FALSE;
th->ready = R_FALSE;
#if HAVE_PTHREAD
pthread_create(&th->tid, NULL, _r_th_launcher, th);
#elif __WIN32__ || __WINDOWS__ && !defined(__CYGWIN__)
th->tid = CreateThread(NULL, 0, _r_th_launcher, th, 0, &th->tid);
#endif
}
return th;
}
R_API void r_th_break(struct r_th_t *th) {
th->breaked = R_TRUE;
}
R_API int r_th_kill(struct r_th_t *th, int force) {
th->breaked = R_TRUE;
r_th_break(th);
r_th_wait(th);
#if HAVE_PTHREAD
#ifdef __ANDROID__
pthread_kill (th->tid, 9);
#else
pthread_cancel (th->tid);
#endif
#endif
return 0;
}
R_API int r_th_start(struct r_th_t *th, int enable) {
int ret = R_TRUE;
if (enable) {
if (!th->running) {
// start thread
while (!th->ready);
r_th_lock_leave (th->lock);
}
} else {
if (th->running) {
// stop thread
r_th_kill (th, 0);
r_th_lock_enter (th->lock); // deadlock?
}
}
th->running = enable;
return ret;
}
R_API int r_th_wait(struct r_th_t *th) {
int ret = R_FALSE;
#if HAVE_PTHREAD
void *thret;
if (th) {
ret = pthread_join (th->tid, &thret);
th->running = R_FALSE;
}
#endif
return ret;
}
R_API int r_th_wait_async(struct r_th_t *th) {
return th->running;
}
R_API void *r_th_free(struct r_th_t *th) {
r_th_kill (th, R_TRUE);
r_th_lock_free (th->lock);
free (th);
return NULL;
}
#if 0
// Thread Pipes
typedef struct r_th_pipe_t {
RList *msglist;
RThread *th;
//RThreadLock *lock;
} RThreadPipe;
r_th_pipe_new();
#endif