mirror of
https://github.com/ashinn/chibi-scheme.git
synced 2025-05-18 21:29:19 +02:00
696 lines
27 KiB
C
696 lines
27 KiB
C
/* threads.c -- SRFI-18 thread primitives */
|
|
/* Copyright (c) 2010-2012 Alex Shinn. All rights reserved. */
|
|
/* BSD-style license: http://synthcode.com/license.txt */
|
|
|
|
#include <chibi/eval.h>
|
|
#include <time.h>
|
|
#include <sys/time.h>
|
|
#include <unistd.h>
|
|
#include <poll.h>
|
|
|
|
#define sexp_mutexp(ctx, x) (sexp_check_tag(x, sexp_unbox_fixnum(sexp_global(ctx, SEXP_G_THREADS_MUTEX_ID))))
|
|
#define sexp_mutex_name(x) sexp_slot_ref(x, 0)
|
|
#define sexp_mutex_specific(x) sexp_slot_ref(x, 1)
|
|
#define sexp_mutex_thread(x) sexp_slot_ref(x, 2)
|
|
#define sexp_mutex_lockp(x) sexp_slot_ref(x, 3)
|
|
|
|
#define sexp_condvar_name(x) sexp_slot_ref(x, 0)
|
|
#define sexp_condvar_specific(x) sexp_slot_ref(x, 1)
|
|
#define sexp_condvar_threads(x) sexp_slot_ref(x, 2)
|
|
|
|
struct sexp_pollfds_t {
|
|
struct pollfd *fds;
|
|
nfds_t nfds, mfds;
|
|
};
|
|
|
|
#define SEXP_INIT_POLLFDS_MAX_FDS 16
|
|
|
|
#define sexp_pollfdsp(ctx, x) (sexp_check_tag(x, sexp_unbox_fixnum(sexp_global(ctx, SEXP_G_THREADS_POLLFDS_ID))))
|
|
#define sexp_pollfds_fds(x) (((struct sexp_pollfds_t*)(&(x)->value))->fds)
|
|
#define sexp_pollfds_num_fds(x) (((struct sexp_pollfds_t*)(&(x)->value))->nfds)
|
|
#define sexp_pollfds_max_fds(x) (((struct sexp_pollfds_t*)(&(x)->value))->mfds)
|
|
|
|
#define sexp_sizeof_pollfds (sexp_sizeof_header + sizeof(struct sexp_pollfds_t))
|
|
|
|
#define timeval_le(a, b) (((a).tv_sec < (b).tv_sec) || (((a).tv_sec == (b).tv_sec) && ((a).tv_usec < (b).tv_usec)))
|
|
#define sexp_context_before(c, t) (((sexp_context_timeval(c).tv_sec != 0) || (sexp_context_timeval(c).tv_usec != 0)) && timeval_le(sexp_context_timeval(c), t))
|
|
|
|
/**************************** threads *************************************/
|
|
|
|
#if SEXP_USE_GREEN_THREADS
|
|
|
|
static void sexp_define_type_predicate_by_tag (sexp ctx, sexp env, const char *cname, sexp_uint_t type) {
|
|
sexp_gc_var2(name, op);
|
|
sexp_gc_preserve2(ctx, name, op);
|
|
name = sexp_c_string(ctx, cname, -1);
|
|
op = sexp_make_type_predicate(ctx, name, sexp_make_fixnum(type));
|
|
sexp_env_define(ctx, env, name=sexp_intern(ctx, cname, -1), op);
|
|
sexp_gc_release2(ctx);
|
|
}
|
|
|
|
sexp sexp_thread_timeoutp (sexp ctx, sexp self, sexp_sint_t n) {
|
|
return sexp_make_boolean(sexp_context_timeoutp(ctx));
|
|
}
|
|
|
|
sexp sexp_thread_name (sexp ctx, sexp self, sexp_sint_t n, sexp thread) {
|
|
sexp_assert_type(ctx, sexp_contextp, SEXP_CONTEXT, thread);
|
|
return sexp_context_name(thread);
|
|
}
|
|
|
|
sexp sexp_thread_specific (sexp ctx, sexp self, sexp_sint_t n, sexp thread) {
|
|
sexp_assert_type(ctx, sexp_contextp, SEXP_CONTEXT, thread);
|
|
return sexp_context_specific(thread);
|
|
}
|
|
|
|
sexp sexp_thread_specific_set (sexp ctx, sexp self, sexp_sint_t n, sexp thread, sexp val) {
|
|
sexp_assert_type(ctx, sexp_contextp, SEXP_CONTEXT, thread);
|
|
sexp_context_specific(thread) = val;
|
|
return SEXP_VOID;
|
|
}
|
|
|
|
sexp sexp_thread_end_result (sexp ctx, sexp self, sexp_sint_t n, sexp thread) {
|
|
sexp_assert_type(ctx, sexp_contextp, SEXP_CONTEXT, thread);
|
|
return sexp_context_result(thread) ? sexp_context_result(thread) : SEXP_VOID;
|
|
}
|
|
|
|
sexp sexp_thread_exceptionp (sexp ctx, sexp self, sexp_sint_t n, sexp thread) {
|
|
sexp_assert_type(ctx, sexp_contextp, SEXP_CONTEXT, thread);
|
|
return sexp_make_boolean(sexp_context_errorp(thread));
|
|
}
|
|
|
|
sexp sexp_current_thread (sexp ctx, sexp self, sexp_sint_t n) {
|
|
return ctx;
|
|
}
|
|
|
|
sexp sexp_make_thread (sexp ctx, sexp self, sexp_sint_t n, sexp thunk, sexp name) {
|
|
sexp *stack;
|
|
sexp_gc_var1(res);
|
|
sexp_assert_type(ctx, sexp_procedurep, SEXP_PROCEDURE, thunk);
|
|
sexp_gc_preserve1(ctx, res);
|
|
res = sexp_make_eval_context(ctx, SEXP_FALSE, sexp_context_env(ctx), 0, 0);
|
|
sexp_context_name(res) = name;
|
|
sexp_context_proc(res) = thunk;
|
|
sexp_context_ip(res) = sexp_bytecode_data(sexp_procedure_code(thunk));
|
|
stack = sexp_stack_data(sexp_context_stack(res));
|
|
stack[0] = stack[1] = stack[3] = SEXP_ZERO;
|
|
stack[2] = sexp_global(ctx, SEXP_G_FINAL_RESUMER);
|
|
sexp_context_top(res) = 4;
|
|
sexp_context_last_fp(res) = 0;
|
|
sexp_context_dk(res) = sexp_make_vector(res, SEXP_FOUR, SEXP_FALSE);
|
|
sexp_vector_set(sexp_context_dk(res), SEXP_ZERO, SEXP_ZERO);
|
|
/* reset parameters */
|
|
sexp_context_params(res) = SEXP_NULL;
|
|
/* alternately reset only the current exception handler */
|
|
/* for (ls1=sexp_context_params(ctx), ls2=SEXP_NULL; sexp_pairp(ls1); ls1=sexp_cdr(ls1)) */
|
|
/* if (sexp_caar(ls1) != sexp_global(ctx, SEXP_G_ERR_HANDLER)) */
|
|
/* ls2 = sexp_cons(ctx, sexp_car(ls1), ls2); */
|
|
/* sexp_context_params(res) = ls2; */
|
|
sexp_gc_release1(ctx);
|
|
return res;
|
|
}
|
|
|
|
sexp sexp_thread_start (sexp ctx, sexp self, sexp_sint_t n, sexp thread) {
|
|
sexp cell;
|
|
sexp_assert_type(ctx, sexp_contextp, SEXP_CONTEXT, thread);
|
|
sexp_context_errorp(thread) = 0;
|
|
cell = sexp_cons(ctx, thread, SEXP_NULL);
|
|
if (sexp_pairp(sexp_global(ctx, SEXP_G_THREADS_BACK))) {
|
|
sexp_cdr(sexp_global(ctx, SEXP_G_THREADS_BACK)) = cell;
|
|
sexp_global(ctx, SEXP_G_THREADS_BACK) = cell;
|
|
} else { /* init queue */
|
|
sexp_global(ctx, SEXP_G_THREADS_BACK) = sexp_global(ctx, SEXP_G_THREADS_FRONT) = cell;
|
|
}
|
|
return thread;
|
|
}
|
|
|
|
static int sexp_delete_list (sexp ctx, int global, sexp x) {
|
|
sexp ls1=NULL, ls2=sexp_global(ctx, global);
|
|
for ( ; sexp_pairp(ls2) && sexp_car(ls2) != x; ls1=ls2, ls2=sexp_cdr(ls2))
|
|
;
|
|
if (sexp_pairp(ls2)) {
|
|
if (ls1) sexp_cdr(ls1) = sexp_cdr(ls2);
|
|
else sexp_global(ctx, global) = sexp_cdr(ls2);
|
|
return 1;
|
|
} else {
|
|
return 0;
|
|
}
|
|
}
|
|
|
|
sexp sexp_thread_terminate (sexp ctx, sexp self, sexp_sint_t n, sexp thread) {
|
|
sexp res = sexp_make_boolean(ctx == thread);
|
|
sexp_assert_type(ctx, sexp_contextp, SEXP_CONTEXT, thread);
|
|
/* terminate the thread and all children */
|
|
for ( ; thread && sexp_contextp(thread); thread=sexp_context_child(thread)) {
|
|
/* if not already terminated set an exception status */
|
|
if (sexp_context_refuel(ctx) > 0) {
|
|
sexp_context_errorp(thread) = 1;
|
|
sexp_context_result(thread) =
|
|
sexp_global(ctx, SEXP_G_THREAD_TERMINATE_ERROR);
|
|
/* zero the refuel - this tells the scheduler the thread is terminated */
|
|
sexp_context_refuel(thread) = 0;
|
|
}
|
|
/* unblock the thread if needed so it can be scheduled and terminated */
|
|
if (sexp_delete_list(ctx, SEXP_G_THREADS_PAUSED, thread))
|
|
sexp_thread_start(ctx, self, 1, thread);
|
|
}
|
|
/* return true if terminating self, then we can yield */
|
|
return res;
|
|
}
|
|
|
|
static void sexp_insert_timed (sexp ctx, sexp thread, sexp timeout) {
|
|
#if SEXP_USE_FLONUMS
|
|
double d;
|
|
#endif
|
|
sexp ls1=SEXP_NULL, ls2;
|
|
sexp_delete_list(ctx, SEXP_G_THREADS_PAUSED, thread);
|
|
ls2 = sexp_global(ctx, SEXP_G_THREADS_PAUSED);
|
|
if (sexp_realp(timeout))
|
|
gettimeofday(&sexp_context_timeval(thread), NULL);
|
|
if (sexp_fixnump(timeout)) {
|
|
sexp_context_timeval(thread).tv_sec += sexp_unbox_fixnum(timeout);
|
|
#if SEXP_USE_FLONUMS
|
|
} else if (sexp_flonump(timeout)) {
|
|
d = sexp_flonum_value(timeout);
|
|
sexp_context_timeval(thread).tv_sec += trunc(d);
|
|
sexp_context_timeval(thread).tv_usec += (d-trunc(d))*1000000;
|
|
if (sexp_context_timeval(thread).tv_usec > 1000000) {
|
|
sexp_context_timeval(thread).tv_sec += 1;
|
|
sexp_context_timeval(thread).tv_usec -= 1000000;
|
|
}
|
|
#endif
|
|
#if SEXP_USE_RATIOS
|
|
} else if (sexp_ratiop(timeout)) {
|
|
d = sexp_ratio_to_double(timeout);
|
|
sexp_context_timeval(thread).tv_sec += trunc(d);
|
|
sexp_context_timeval(thread).tv_usec += (d-trunc(d))*1000000;
|
|
if (sexp_context_timeval(thread).tv_usec > 1000000) {
|
|
sexp_context_timeval(thread).tv_sec += 1;
|
|
sexp_context_timeval(thread).tv_usec -= 1000000;
|
|
}
|
|
#endif
|
|
} else if (sexp_contextp(timeout)) {
|
|
sexp_context_timeval(thread).tv_sec = sexp_context_timeval(timeout).tv_sec;
|
|
sexp_context_timeval(thread).tv_usec = sexp_context_timeval(timeout).tv_usec;
|
|
} else {
|
|
sexp_context_timeval(thread).tv_sec = 0;
|
|
sexp_context_timeval(thread).tv_usec = 0;
|
|
}
|
|
if (sexp_realp(timeout) || sexp_contextp(timeout))
|
|
while (sexp_pairp(ls2)
|
|
&& sexp_context_before(sexp_car(ls2), sexp_context_timeval(thread)))
|
|
ls1=ls2, ls2=sexp_cdr(ls2);
|
|
else
|
|
while (sexp_pairp(ls2) && sexp_context_timeval(sexp_car(ls2)).tv_sec)
|
|
ls1=ls2, ls2=sexp_cdr(ls2);
|
|
if (ls1 == SEXP_NULL)
|
|
sexp_global(ctx, SEXP_G_THREADS_PAUSED) = sexp_cons(ctx, thread, ls2);
|
|
else
|
|
sexp_cdr(ls1) = sexp_cons(ctx, thread, ls2);
|
|
}
|
|
|
|
sexp sexp_thread_join (sexp ctx, sexp self, sexp_sint_t n, sexp thread, sexp timeout) {
|
|
sexp_assert_type(ctx, sexp_contextp, SEXP_CONTEXT, thread);
|
|
if (sexp_context_refuel(thread) <= 0) /* return true if already terminated */ {
|
|
return SEXP_TRUE;
|
|
}
|
|
sexp_context_timeoutp(ctx) = 0;
|
|
sexp_context_waitp(ctx) = 1;
|
|
sexp_context_event(ctx) = thread;
|
|
sexp_insert_timed(ctx, ctx, timeout);
|
|
return SEXP_FALSE;
|
|
}
|
|
|
|
sexp sexp_thread_sleep (sexp ctx, sexp self, sexp_sint_t n, sexp timeout) {
|
|
sexp_context_waitp(ctx) = 1;
|
|
if (timeout != SEXP_TRUE) {
|
|
sexp_assert_type(ctx, sexp_realp, SEXP_NUMBER, timeout);
|
|
sexp_context_event(ctx) = SEXP_FALSE;
|
|
sexp_insert_timed(ctx, ctx, timeout);
|
|
}
|
|
return SEXP_FALSE;
|
|
}
|
|
|
|
/**************************** mutexes *************************************/
|
|
|
|
sexp sexp_mutex_state (sexp ctx, sexp self, sexp_sint_t n, sexp mutex) {
|
|
if (!sexp_mutexp(ctx, mutex))
|
|
return sexp_type_exception(ctx, self, sexp_unbox_fixnum(sexp_global(ctx, SEXP_G_THREADS_POLLFDS_ID)), mutex);
|
|
if (sexp_truep(sexp_mutex_lockp(mutex))) {
|
|
if (sexp_contextp(sexp_mutex_thread(mutex)))
|
|
return sexp_mutex_thread(mutex);
|
|
else
|
|
return sexp_intern(ctx, "not-owned", -1);
|
|
} else {
|
|
return sexp_intern(ctx, (sexp_mutex_thread(mutex) ? "not-abandoned" : "abandoned"), -1);
|
|
}
|
|
}
|
|
|
|
sexp sexp_mutex_lock (sexp ctx, sexp self, sexp_sint_t n, sexp mutex, sexp timeout, sexp thread) {
|
|
if (thread == SEXP_TRUE)
|
|
thread = ctx;
|
|
if (sexp_not(sexp_mutex_lockp(mutex))) {
|
|
sexp_mutex_lockp(mutex) = SEXP_TRUE;
|
|
sexp_mutex_thread(mutex) = thread;
|
|
return SEXP_TRUE;
|
|
} else {
|
|
sexp_context_waitp(ctx) = 1;
|
|
sexp_context_event(ctx) = mutex;
|
|
sexp_insert_timed(ctx, ctx, timeout);
|
|
return SEXP_FALSE;
|
|
}
|
|
}
|
|
|
|
sexp sexp_mutex_unlock (sexp ctx, sexp self, sexp_sint_t n, sexp mutex, sexp condvar, sexp timeout) {
|
|
sexp ls1, ls2;
|
|
/* first unlock and unblock threads */
|
|
if (sexp_truep(sexp_mutex_lockp(mutex))) {
|
|
sexp_mutex_lockp(mutex) = SEXP_FALSE;
|
|
sexp_mutex_thread(mutex) = ctx;
|
|
/* search for threads blocked on this mutex */
|
|
for (ls1=SEXP_NULL, ls2=sexp_global(ctx, SEXP_G_THREADS_PAUSED);
|
|
sexp_pairp(ls2); ls1=ls2, ls2=sexp_cdr(ls2))
|
|
if (sexp_context_event(sexp_car(ls2)) == mutex) {
|
|
if (ls1==SEXP_NULL)
|
|
sexp_global(ctx, SEXP_G_THREADS_PAUSED) = sexp_cdr(ls2);
|
|
else
|
|
sexp_cdr(ls1) = sexp_cdr(ls2);
|
|
sexp_cdr(ls2) = sexp_global(ctx, SEXP_G_THREADS_FRONT);
|
|
sexp_global(ctx, SEXP_G_THREADS_FRONT) = ls2;
|
|
if (! sexp_pairp(sexp_cdr(ls2)))
|
|
sexp_global(ctx, SEXP_G_THREADS_BACK) = ls2;
|
|
sexp_context_waitp(sexp_car(ls2))
|
|
= sexp_context_timeoutp(sexp_car(ls2)) = 0;
|
|
break;
|
|
}
|
|
}
|
|
if (sexp_truep(condvar)) {
|
|
/* wait on condition var if specified */
|
|
sexp_context_waitp(ctx) = 1;
|
|
sexp_context_event(ctx) = condvar;
|
|
sexp_insert_timed(ctx, ctx, timeout);
|
|
return SEXP_FALSE;
|
|
}
|
|
return SEXP_TRUE;
|
|
}
|
|
|
|
/**************************** condition variables *************************/
|
|
|
|
sexp sexp_condition_variable_signal (sexp ctx, sexp self, sexp_sint_t n, sexp condvar) {
|
|
sexp ls1=SEXP_NULL, ls2=sexp_global(ctx, SEXP_G_THREADS_PAUSED);
|
|
for ( ; sexp_pairp(ls2); ls1=ls2, ls2=sexp_cdr(ls2))
|
|
if (sexp_context_event(sexp_car(ls2)) == condvar) {
|
|
if (ls1==SEXP_NULL)
|
|
sexp_global(ctx, SEXP_G_THREADS_PAUSED) = sexp_cdr(ls2);
|
|
else
|
|
sexp_cdr(ls1) = sexp_cdr(ls2);
|
|
sexp_cdr(ls2) = sexp_global(ctx, SEXP_G_THREADS_FRONT);
|
|
sexp_global(ctx, SEXP_G_THREADS_FRONT) = ls2;
|
|
if (! sexp_pairp(sexp_cdr(ls2)))
|
|
sexp_global(ctx, SEXP_G_THREADS_BACK) = ls2;
|
|
sexp_context_waitp(sexp_car(ls2)) = sexp_context_timeoutp(sexp_car(ls2)) = 0;
|
|
return SEXP_TRUE;
|
|
}
|
|
return SEXP_FALSE;
|
|
}
|
|
|
|
sexp sexp_condition_variable_broadcast (sexp ctx, sexp self, sexp_sint_t n, sexp condvar) {
|
|
sexp res = SEXP_FALSE;
|
|
while (sexp_truep(sexp_condition_variable_signal(ctx, self, n, condvar)))
|
|
res = SEXP_TRUE;
|
|
return res;
|
|
}
|
|
|
|
/**************************** the scheduler *******************************/
|
|
|
|
static const sexp_uint_t sexp_log2_lookup[32] = {
|
|
0, 1, 28, 2, 29, 14, 24, 3, 30, 22, 20, 15, 25, 17, 4, 8,
|
|
31, 27, 13, 23, 21, 19, 16, 7, 26, 12, 18, 6, 11, 5, 10, 9
|
|
};
|
|
|
|
/* only works on powers of two */
|
|
static sexp_uint_t sexp_log2_of_pow2 (sexp_uint_t n) {
|
|
return sexp_log2_lookup[((unsigned)n * 0x077CB531U) >> 27];
|
|
}
|
|
|
|
sexp sexp_pop_signal (sexp ctx, sexp self, sexp_sint_t n) {
|
|
int allsigs, restsigs, signum;
|
|
if (sexp_global(ctx, SEXP_G_THREADS_SIGNALS) == SEXP_ZERO) {
|
|
return SEXP_FALSE;
|
|
} else {
|
|
allsigs = sexp_unbox_fixnum(sexp_global(ctx, SEXP_G_THREADS_SIGNALS));
|
|
restsigs = allsigs & (allsigs-1);
|
|
sexp_global(ctx, SEXP_G_THREADS_SIGNALS) = sexp_make_fixnum(restsigs);
|
|
signum = sexp_log2_of_pow2(allsigs-restsigs);
|
|
return sexp_make_fixnum(signum);
|
|
}
|
|
}
|
|
|
|
sexp sexp_get_signal_handler (sexp ctx, sexp self, sexp_sint_t n, sexp signum) {
|
|
sexp_assert_type(ctx, sexp_fixnump, SEXP_FIXNUM, signum);
|
|
return sexp_vector_ref(sexp_global(ctx, SEXP_G_SIGNAL_HANDLERS), signum);
|
|
}
|
|
|
|
static sexp sexp_make_pollfds (sexp ctx) {
|
|
sexp res = sexp_alloc_tagged(ctx, sexp_sizeof_pollfds, sexp_unbox_fixnum(sexp_global(ctx, SEXP_G_THREADS_POLLFDS_ID)));
|
|
sexp_pollfds_fds(res) = (struct pollfd*)malloc(SEXP_INIT_POLLFDS_MAX_FDS * sizeof(struct pollfd));
|
|
sexp_pollfds_num_fds(res) = 0;
|
|
sexp_pollfds_max_fds(res) = SEXP_INIT_POLLFDS_MAX_FDS;
|
|
return res;
|
|
}
|
|
|
|
sexp sexp_free_pollfds (sexp ctx, sexp self, sexp_sint_t n, sexp pollfds) {
|
|
if (sexp_pollfds_fds(pollfds)) {
|
|
free(sexp_pollfds_fds(pollfds));
|
|
sexp_pollfds_fds(pollfds) = NULL;
|
|
sexp_pollfds_num_fds(pollfds) = 0;
|
|
sexp_pollfds_max_fds(pollfds) = 0;
|
|
}
|
|
return SEXP_VOID;
|
|
}
|
|
|
|
/* return true if this fd was already being polled */
|
|
static sexp sexp_insert_pollfd (sexp ctx, int fd, int events) {
|
|
int i;
|
|
struct pollfd *pfd;
|
|
sexp pollfds = sexp_global(ctx, SEXP_G_THREADS_POLL_FDS);
|
|
if (! (pollfds && sexp_pollfdsp(ctx, pollfds))) {
|
|
sexp_global(ctx, SEXP_G_THREADS_POLL_FDS) = pollfds = sexp_make_pollfds(ctx);
|
|
}
|
|
for (i=0; i<sexp_pollfds_num_fds(pollfds); ++i) {
|
|
if (sexp_pollfds_fds(pollfds)[i].fd == fd) {
|
|
sexp_pollfds_fds(pollfds)[i].events |= events;
|
|
return SEXP_TRUE;
|
|
}
|
|
}
|
|
if (sexp_pollfds_num_fds(pollfds) == sexp_pollfds_max_fds(pollfds)) {
|
|
sexp_pollfds_max_fds(pollfds) = i*2;
|
|
pfd = sexp_pollfds_fds(pollfds);
|
|
sexp_pollfds_fds(pollfds) = (struct pollfd*)malloc(i*2*sizeof(struct pollfd));
|
|
if (sexp_pollfds_fds(pollfds))
|
|
memcpy(sexp_pollfds_fds(pollfds), pfd, i*2*sizeof(struct pollfd));
|
|
free(pfd);
|
|
}
|
|
pfd = &(sexp_pollfds_fds(pollfds)[sexp_pollfds_num_fds(pollfds)++]);
|
|
pfd->fd = fd;
|
|
pfd->events = events;
|
|
return SEXP_FALSE;
|
|
}
|
|
|
|
/* block the current thread on the specified port */
|
|
sexp sexp_blocker (sexp ctx, sexp self, sexp_sint_t n, sexp portorfd, sexp timeout) {
|
|
int fd;
|
|
/* register the fd */
|
|
if (sexp_portp(portorfd))
|
|
fd = sexp_port_fileno(portorfd);
|
|
else if (sexp_filenop(portorfd))
|
|
fd = sexp_fileno_fd(portorfd);
|
|
else if (sexp_fixnump(portorfd))
|
|
fd = sexp_unbox_fixnum(portorfd);
|
|
else
|
|
return sexp_type_exception(ctx, self, SEXP_IPORT, portorfd);
|
|
if (fd >= 0)
|
|
sexp_insert_pollfd(ctx, fd, sexp_oportp(portorfd) ? POLLOUT : POLLIN);
|
|
/* pause the current thread */
|
|
sexp_context_waitp(ctx) = 1;
|
|
sexp_context_event(ctx) = portorfd;
|
|
sexp_insert_timed(ctx, ctx, timeout);
|
|
return SEXP_VOID;
|
|
}
|
|
|
|
sexp sexp_scheduler (sexp ctx, sexp self, sexp_sint_t n, sexp root_thread) {
|
|
int i, k;
|
|
struct timeval tval;
|
|
struct pollfd *pfds;
|
|
useconds_t usecs = 0;
|
|
sexp res, ls1, ls2, evt, runner, paused, front, pollfds;
|
|
sexp_gc_var1(tmp);
|
|
sexp_gc_preserve1(ctx, tmp);
|
|
|
|
front = sexp_global(ctx, SEXP_G_THREADS_FRONT);
|
|
paused = sexp_global(ctx, SEXP_G_THREADS_PAUSED);
|
|
|
|
/* check signals */
|
|
if (sexp_global(ctx, SEXP_G_THREADS_SIGNALS) != SEXP_ZERO) {
|
|
runner = sexp_global(ctx, SEXP_G_THREADS_SIGNAL_RUNNER);
|
|
if (! sexp_contextp(runner)) { /* ensure the runner exists */
|
|
if (sexp_envp(runner)) {
|
|
tmp = sexp_env_cell(ctx, runner, (tmp=sexp_intern(ctx, "signal-runner", -1)), 0);
|
|
if (sexp_pairp(tmp) && sexp_procedurep(sexp_cdr(tmp))) {
|
|
runner = sexp_make_thread(ctx, self, 2, sexp_cdr(tmp), SEXP_FALSE);
|
|
sexp_global(ctx, SEXP_G_THREADS_SIGNAL_RUNNER) = runner;
|
|
sexp_thread_start(ctx, self, 1, runner);
|
|
if (!sexp_pairp(front))
|
|
front = sexp_global(ctx, SEXP_G_THREADS_FRONT);
|
|
}
|
|
}
|
|
} else if (sexp_context_waitp(runner)) { /* wake it if it's sleeping */
|
|
sexp_context_waitp(runner) = 0;
|
|
sexp_thread_start(ctx, self, 1, runner);
|
|
}
|
|
}
|
|
|
|
/* check blocked fds */
|
|
pollfds = sexp_global(ctx, SEXP_G_THREADS_POLL_FDS);
|
|
if (sexp_pollfdsp(ctx, pollfds) && sexp_pollfds_num_fds(pollfds) > 0) {
|
|
pfds = sexp_pollfds_fds(pollfds);
|
|
k = poll(sexp_pollfds_fds(pollfds), sexp_pollfds_num_fds(pollfds), 0);
|
|
for (i=sexp_pollfds_num_fds(pollfds)-1; i>=0 && k>0; --i) {
|
|
if (pfds[i].revents > 0) { /* free all threads blocked on this fd */
|
|
k--;
|
|
/* maybe unblock the current thread */
|
|
evt = sexp_context_event(ctx);
|
|
if ((sexp_portp(evt) && (sexp_port_fileno(evt) == pfds[i].fd))
|
|
|| (sexp_fixnump(evt) && (sexp_unbox_fixnum(evt) == pfds[i].fd))) {
|
|
sexp_context_waitp(ctx) = 0;
|
|
sexp_context_timeoutp(ctx) = 0;
|
|
sexp_context_event(ctx) = SEXP_FALSE;
|
|
}
|
|
/* maybe unblock paused threads */
|
|
for (ls1=SEXP_NULL, ls2=paused; sexp_pairp(ls2); ) {
|
|
/* TODO: distinguish input and output on the same fd? */
|
|
evt = sexp_context_event(sexp_car(ls2));
|
|
if ((sexp_portp(evt) && sexp_port_fileno(evt) == pfds[i].fd)
|
|
|| (sexp_fixnump(evt) && sexp_unbox_fixnum(evt) == pfds[i].fd)) {
|
|
sexp_context_waitp(sexp_car(ls2)) = 0;
|
|
sexp_context_timeoutp(sexp_car(ls2)) = 0;
|
|
sexp_context_event(sexp_car(ls2)) = SEXP_FALSE;
|
|
if (ls1==SEXP_NULL)
|
|
sexp_global(ctx, SEXP_G_THREADS_PAUSED) = paused = sexp_cdr(ls2);
|
|
else
|
|
sexp_cdr(ls1) = sexp_cdr(ls2);
|
|
tmp = sexp_cdr(ls2);
|
|
sexp_cdr(ls2) = SEXP_NULL;
|
|
if (sexp_car(ls2) != ctx) {
|
|
if (! sexp_pairp(sexp_global(ctx, SEXP_G_THREADS_BACK))) {
|
|
sexp_global(ctx, SEXP_G_THREADS_FRONT) = front = ls2;
|
|
} else {
|
|
sexp_cdr(sexp_global(ctx, SEXP_G_THREADS_BACK)) = ls2;
|
|
}
|
|
sexp_global(ctx, SEXP_G_THREADS_BACK) = ls2;
|
|
}
|
|
ls2 = tmp;
|
|
} else {
|
|
ls1 = ls2;
|
|
ls2 = sexp_cdr(ls2);
|
|
}
|
|
}
|
|
if (i < (sexp_pollfds_num_fds(pollfds) - 1)) {
|
|
pfds[i] = pfds[sexp_pollfds_num_fds(pollfds) - 1];
|
|
}
|
|
sexp_pollfds_num_fds(pollfds) -= 1;
|
|
}
|
|
}
|
|
}
|
|
|
|
/* if we've terminated, check threads joining us */
|
|
if (sexp_context_refuel(ctx) <= 0) {
|
|
for (ls1=SEXP_NULL, ls2=paused; sexp_pairp(ls2); ) {
|
|
if (sexp_context_event(sexp_car(ls2)) == ctx) {
|
|
sexp_context_waitp(sexp_car(ls2)) = 0;
|
|
sexp_context_timeoutp(sexp_car(ls2)) = 0;
|
|
if (ls1==SEXP_NULL)
|
|
sexp_global(ctx, SEXP_G_THREADS_PAUSED) = paused = sexp_cdr(ls2);
|
|
else
|
|
sexp_cdr(ls1) = sexp_cdr(ls2);
|
|
tmp = sexp_cdr(ls2);
|
|
sexp_cdr(ls2) = SEXP_NULL;
|
|
if (! sexp_pairp(sexp_global(ctx, SEXP_G_THREADS_BACK))) {
|
|
sexp_global(ctx, SEXP_G_THREADS_FRONT) = front = ls2;
|
|
} else {
|
|
sexp_cdr(sexp_global(ctx, SEXP_G_THREADS_BACK)) = ls2;
|
|
}
|
|
sexp_global(ctx, SEXP_G_THREADS_BACK) = ls2;
|
|
ls2 = tmp;
|
|
} else {
|
|
ls1 = ls2;
|
|
ls2 = sexp_cdr(ls2);
|
|
}
|
|
}
|
|
}
|
|
|
|
/* check timeouts */
|
|
if (sexp_pairp(paused)) {
|
|
if (gettimeofday(&tval, NULL) == 0) {
|
|
ls1 = SEXP_NULL;
|
|
ls2 = paused;
|
|
while (sexp_pairp(ls2) && sexp_context_before(sexp_car(ls2), tval)) {
|
|
sexp_context_timeoutp(sexp_car(ls2)) = 1;
|
|
sexp_context_waitp(sexp_car(ls2)) = 0;
|
|
ls1 = ls2;
|
|
ls2 = sexp_cdr(ls2);
|
|
}
|
|
if (sexp_pairp(ls1)) {
|
|
sexp_cdr(ls1) = SEXP_NULL;
|
|
if (! sexp_pairp(sexp_global(ctx, SEXP_G_THREADS_BACK))) {
|
|
sexp_global(ctx, SEXP_G_THREADS_FRONT) = front = paused;
|
|
} else {
|
|
sexp_cdr(sexp_global(ctx, SEXP_G_THREADS_BACK)) = paused;
|
|
}
|
|
sexp_global(ctx, SEXP_G_THREADS_BACK) = ls1;
|
|
sexp_global(ctx, SEXP_G_THREADS_PAUSED) = paused = ls2;
|
|
}
|
|
}
|
|
}
|
|
|
|
/* dequeue next thread */
|
|
if (sexp_pairp(front)) {
|
|
res = sexp_car(front);
|
|
if ((sexp_context_refuel(ctx) <= 0) || sexp_context_waitp(ctx)) {
|
|
/* orig ctx is either terminated or paused */
|
|
sexp_global(ctx, SEXP_G_THREADS_FRONT) = sexp_cdr(front);
|
|
if (! sexp_pairp(sexp_cdr(front)))
|
|
sexp_global(ctx, SEXP_G_THREADS_BACK) = SEXP_NULL;
|
|
if (sexp_context_refuel(ctx) > 0 && sexp_not(sexp_memq(ctx, ctx, paused)))
|
|
sexp_insert_timed(ctx, ctx, SEXP_FALSE);
|
|
paused = sexp_global(ctx, SEXP_G_THREADS_PAUSED);
|
|
} else {
|
|
/* swap with front of queue */
|
|
sexp_car(sexp_global(ctx, SEXP_G_THREADS_FRONT)) = ctx;
|
|
/* rotate front of queue to back */
|
|
sexp_cdr(sexp_global(ctx, SEXP_G_THREADS_BACK))
|
|
= sexp_global(ctx, SEXP_G_THREADS_FRONT);
|
|
sexp_global(ctx, SEXP_G_THREADS_FRONT)
|
|
= sexp_cdr(sexp_global(ctx, SEXP_G_THREADS_FRONT));
|
|
sexp_global(ctx, SEXP_G_THREADS_BACK)
|
|
= sexp_cdr(sexp_global(ctx, SEXP_G_THREADS_BACK));
|
|
sexp_cdr(sexp_global(ctx, SEXP_G_THREADS_BACK)) = SEXP_NULL;
|
|
}
|
|
} else {
|
|
/* no threads to dequeue */
|
|
res = ctx;
|
|
/* prefer a thread we can wait on instead of spinning */
|
|
if (sexp_context_refuel(ctx) <= 0) {
|
|
for (ls1=paused; sexp_pairp(ls1); ls1=sexp_cdr(ls1)) {
|
|
evt = sexp_context_event(sexp_car(ls1));
|
|
if (sexp_fixnump(evt) || sexp_portp(evt)) {
|
|
res = sexp_car(ls1);
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
if (sexp_context_waitp(res)) {
|
|
/* the only thread available was waiting */
|
|
if (sexp_pairp(paused)
|
|
&& sexp_context_before(sexp_car(paused), sexp_context_timeval(res))) {
|
|
tmp = res;
|
|
res = sexp_car(paused);
|
|
paused = sexp_global(ctx, SEXP_G_THREADS_PAUSED) = sexp_cdr(paused);
|
|
if (sexp_not(sexp_memq(ctx, tmp, paused)))
|
|
sexp_insert_timed(ctx, tmp, tmp);
|
|
} else {
|
|
sexp_delete_list(ctx, SEXP_G_THREADS_PAUSED, res);
|
|
}
|
|
paused = sexp_global(ctx, SEXP_G_THREADS_PAUSED);
|
|
usecs = 0;
|
|
if ((sexp_context_timeval(res).tv_sec == 0)
|
|
&& (sexp_context_timeval(res).tv_usec == 0)) {
|
|
/* no timeout, wait for default 10ms */
|
|
usecs = 10*1000;
|
|
} else {
|
|
/* wait until the next timeout */
|
|
gettimeofday(&tval, NULL);
|
|
if (tval.tv_sec <= sexp_context_timeval(res).tv_sec) {
|
|
usecs = (sexp_context_timeval(res).tv_sec - tval.tv_sec) * 1000000;
|
|
if (tval.tv_usec < sexp_context_timeval(res).tv_usec || usecs > 0)
|
|
usecs += sexp_context_timeval(res).tv_usec - tval.tv_usec;
|
|
}
|
|
}
|
|
/* take a nap to avoid busy looping */
|
|
usleep(usecs);
|
|
sexp_context_waitp(res) = 0;
|
|
sexp_context_timeoutp(res) = 1;
|
|
}
|
|
|
|
sexp_gc_release1(ctx);
|
|
return res;
|
|
}
|
|
|
|
/**************************************************************************/
|
|
|
|
sexp sexp_lookup_named_type (sexp ctx, sexp env, const char *name) {
|
|
sexp t = sexp_env_ref(ctx, env, sexp_intern(ctx, name, -1), SEXP_FALSE);
|
|
return sexp_make_fixnum((sexp_typep(t)) ? sexp_type_tag(t) : -1);
|
|
}
|
|
|
|
#endif /* SEXP_USE_GREEN_THREADS */
|
|
|
|
sexp sexp_init_library (sexp ctx, sexp self, sexp_sint_t n, sexp env, const char* version, const sexp_abi_identifier_t abi) {
|
|
sexp t;
|
|
sexp_gc_var1(name);
|
|
if (!(sexp_version_compatible(ctx, version, sexp_version)
|
|
&& sexp_abi_compatible(ctx, abi, SEXP_ABI_IDENTIFIER)))
|
|
return SEXP_ABI_ERROR;
|
|
|
|
#if SEXP_USE_GREEN_THREADS
|
|
|
|
sexp_gc_preserve1(ctx, name);
|
|
|
|
sexp_global(ctx, SEXP_G_THREADS_MUTEX_ID) = sexp_lookup_named_type(ctx, env, "Mutex");
|
|
name = sexp_c_string(ctx, "pollfds", -1);
|
|
t = sexp_register_type(ctx, name, SEXP_FALSE, SEXP_FALSE,
|
|
SEXP_ZERO, SEXP_ZERO, SEXP_ZERO, SEXP_ZERO,
|
|
SEXP_ZERO, sexp_make_fixnum(sexp_sizeof_pollfds),
|
|
SEXP_ZERO, SEXP_ZERO, SEXP_ZERO, SEXP_ZERO, SEXP_ZERO,
|
|
SEXP_ZERO, SEXP_ZERO, NULL, "sexp_free_pollfds",
|
|
(sexp_proc2)sexp_free_pollfds);
|
|
if (sexp_typep(t)) {
|
|
sexp_global(ctx, SEXP_G_THREADS_POLLFDS_ID) = sexp_make_fixnum(sexp_type_tag(t));
|
|
}
|
|
|
|
sexp_define_type_predicate_by_tag(ctx, env, "thread?", SEXP_CONTEXT);
|
|
sexp_define_foreign(ctx, env, "thread-timeout?", 0, sexp_thread_timeoutp);
|
|
sexp_define_foreign(ctx, env, "current-thread", 0, sexp_current_thread);
|
|
sexp_define_foreign_opt(ctx, env, "make-thread", 2, sexp_make_thread, SEXP_FALSE);
|
|
sexp_define_foreign(ctx, env, "thread-start!", 1, sexp_thread_start);
|
|
sexp_define_foreign(ctx, env, "%thread-terminate!", 1, sexp_thread_terminate);
|
|
sexp_define_foreign(ctx, env, "%thread-join!", 2, sexp_thread_join);
|
|
sexp_define_foreign(ctx, env, "%thread-sleep!", 1, sexp_thread_sleep);
|
|
sexp_define_foreign(ctx, env, "thread-name", 1, sexp_thread_name);
|
|
sexp_define_foreign(ctx, env, "thread-specific", 1, sexp_thread_specific);
|
|
sexp_define_foreign(ctx, env, "thread-specific-set!", 2, sexp_thread_specific_set);
|
|
sexp_define_foreign(ctx, env, "%thread-end-result", 1, sexp_thread_end_result);
|
|
sexp_define_foreign(ctx, env, "%thread-exception?", 1, sexp_thread_exceptionp);
|
|
sexp_define_foreign(ctx, env, "mutex-state", 1, sexp_mutex_state);
|
|
sexp_define_foreign(ctx, env, "%mutex-lock!", 3, sexp_mutex_lock);
|
|
sexp_define_foreign(ctx, env, "%mutex-unlock!", 3, sexp_mutex_unlock);
|
|
sexp_define_foreign(ctx, env, "condition-variable-signal!", 1, sexp_condition_variable_signal);
|
|
sexp_define_foreign(ctx, env, "condition-variable-broadcast!", 1, sexp_condition_variable_broadcast);
|
|
sexp_define_foreign(ctx, env, "pop-signal!", 0, sexp_pop_signal);
|
|
sexp_define_foreign(ctx, env, "get-signal-handler", 1, sexp_get_signal_handler);
|
|
|
|
sexp_global(ctx, SEXP_G_THREADS_SCHEDULER)
|
|
= sexp_make_foreign(ctx, "scheduler", 1, 0, "sexp_scheduler", (sexp_proc1)sexp_scheduler, SEXP_FALSE);
|
|
sexp_global(ctx, SEXP_G_THREADS_BLOCKER)
|
|
= sexp_make_foreign(ctx, "blocker", 2, 0, "sexp_blocker", (sexp_proc1)sexp_blocker, SEXP_FALSE);
|
|
|
|
/* remember the env to lookup the runner later */
|
|
sexp_global(ctx, SEXP_G_THREADS_SIGNAL_RUNNER) = env;
|
|
|
|
sexp_gc_release1(ctx);
|
|
|
|
#endif /* SEXP_USE_GREEN_THREADS */
|
|
|
|
return SEXP_VOID;
|
|
}
|