[p5sagit/p5-mst-13.2.git] / ext / Thread / Thread.xs

#include "EXTERN.h"
#include "perl.h"
#include "XSUB.h"

/* Magic signature for Thread's mg_private is "Th" */ 
#define Thread_MAGIC_SIGNATURE 0x5468

static U32 threadnum = 0;
static int sig_pipe[2];

static void
remove_thread(t)
Thread t;
{
    DEBUG_L(WITH_THR(PerlIO_printf(PerlIO_stderr(),
				   "%p: remove_thread %p\n", thr, t)));
    MUTEX_LOCK(&threads_mutex);
    MUTEX_DESTROY(&t->mutex);
    nthreads--;
    t->prev->next = t->next;
    t->next->prev = t->prev;
    COND_BROADCAST(&nthreads_cond);
    MUTEX_UNLOCK(&threads_mutex);
}

static THREAD_RET_TYPE
threadstart(arg)
void *arg;
{
#ifdef FAKE_THREADS
    Thread savethread = thr;
    LOGOP myop;
    dSP;
    I32 oldscope = scopestack_ix;
    I32 retval;
    AV *returnav;
    int i;

    DEBUG_L(PerlIO_printf(PerlIO_stderr(), "new thread %p starting at %s\n",
			  thr, SvPEEK(TOPs)));
    thr = (Thread) arg;
    savemark = TOPMARK;
    thr->prev = thr->prev_run = savethread;
    thr->next = savethread->next;
    thr->next_run = savethread->next_run;
    savethread->next = savethread->next_run = thr;
    thr->wait_queue = 0;
    thr->private = 0;

    /* Now duplicate most of perl_call_sv but with a few twists */
    op = (OP*)&myop;
    Zero(op, 1, LOGOP);
    myop.op_flags = OPf_STACKED;
    myop.op_next = Nullop;
    myop.op_flags |= OPf_KNOW;
    myop.op_flags |= OPf_WANT_LIST;
    op = pp_entersub(ARGS);
    DEBUG_L(if (!op)
	    PerlIO_printf(PerlIO_stderr(), "thread starts at Nullop\n"));
    /*
     * When this thread is next scheduled, we start in the right
     * place. When the thread runs off the end of the sub, perl.c
     * handles things, using savemark to figure out how much of the
     * stack is the return value for any join.
     */
    thr = savethread;		/* back to the old thread */
    return 0;
#else
    Thread thr = (Thread) arg;
    LOGOP myop;
    dSP;
    I32 oldmark = TOPMARK;
    I32 oldscope = scopestack_ix;
    I32 retval;
    AV *returnav;
    int i, ret;
    dJMPENV;

    /* Don't call *anything* requiring dTHR until after pthread_setspecific */
    /*
     * Wait until our creator releases us. If we didn't do this, then
     * it would be potentially possible for out thread to carry on and
     * do stuff before our creator fills in our "self" field. For example,
     * if we went and created another thread which tried to JOIN with us,
     * then we'd be in a mess.
     */
    MUTEX_LOCK(&thr->mutex);
    MUTEX_UNLOCK(&thr->mutex);

    /*
     * It's safe to wait until now to set the thread-specific pointer
     * from our pthread_t structure to our struct thread, since we're
     * the only thread who can get at it anyway.
     */
    SET_THR(thr);

    /* Only now can we use SvPEEK (which calls sv_newmortal which does dTHR) */
    DEBUG_L(PerlIO_printf(PerlIO_stderr(), "new thread %p starting at %s\n",
			  thr, SvPEEK(TOPs)));

    JMPENV_PUSH(ret);
    switch (ret) {
    case 3:
        PerlIO_printf(PerlIO_stderr(), "panic: threadstart\n");
	/* fall through */
    case 1:
	STATUS_ALL_FAILURE;
	/* fall through */
    case 2:
	/* my_exit() was called */
	while (scopestack_ix > oldscope)
	    LEAVE;
	JMPENV_POP;
	av_store(returnav, 0, newSViv(statusvalue));
	goto finishoff;
    }

    /* Now duplicate most of perl_call_sv but with a few twists */
    op = (OP*)&myop;
    Zero(op, 1, LOGOP);
    myop.op_flags = OPf_STACKED;
    myop.op_next = Nullop;
    myop.op_flags |= OPf_KNOW;
    myop.op_flags |= OPf_WANT_LIST;
    op = pp_entersub(ARGS);
    if (op)
	runops();
    SPAGAIN;
    retval = sp - (stack_base + oldmark);
    sp = stack_base + oldmark + 1;
    DEBUG_L(for (i = 1; i <= retval; i++)
		PerlIO_printf(PerlIO_stderr(),
			      "%p returnav[%d] = %s\n",
			      thr, i, SvPEEK(sp[i - 1]));)
    returnav = newAV();
    av_store(returnav, 0, newSVpv("", 0));
    for (i = 1; i <= retval; i++, sp++)
	sv_setsv(*av_fetch(returnav, i, TRUE), SvREFCNT_inc(*sp));
    
  finishoff:
#if 0    
    /* removed for debug */
    SvREFCNT_dec(curstack);
#endif
    SvREFCNT_dec(cvcache);
    SvREFCNT_dec(thr->magicals);
    SvREFCNT_dec(thr->specific);
    Safefree(markstack);
    Safefree(scopestack);
    Safefree(savestack);
    Safefree(retstack);
    Safefree(cxstack);
    Safefree(tmps_stack);

    MUTEX_LOCK(&thr->mutex);
    DEBUG_L(PerlIO_printf(PerlIO_stderr(),
			  "%p: threadstart finishing: state is %u\n",
			  thr, ThrSTATE(thr)));
    switch (ThrSTATE(thr)) {
    case THRf_R_JOINABLE:
	ThrSETSTATE(thr, THRf_ZOMBIE);
	MUTEX_UNLOCK(&thr->mutex);
	DEBUG_L(PerlIO_printf(PerlIO_stderr(),
			      "%p: R_JOINABLE thread finished\n", thr));
	break;
    case THRf_R_JOINED:
	ThrSETSTATE(thr, THRf_DEAD);
	MUTEX_UNLOCK(&thr->mutex);
	remove_thread(thr);
	DEBUG_L(PerlIO_printf(PerlIO_stderr(),
			      "%p: R_JOINED thread finished\n", thr));
	break;
    case THRf_R_DETACHED:
	ThrSETSTATE(thr, THRf_DEAD);
	MUTEX_UNLOCK(&thr->mutex);
	SvREFCNT_dec(returnav);
	DEBUG_L(PerlIO_printf(PerlIO_stderr(),
			      "%p: DETACHED thread finished\n", thr));
	remove_thread(thr);	/* This might trigger main thread to finish */
	break;
    default:
	MUTEX_UNLOCK(&thr->mutex);
	croak("panic: illegal state %u at end of threadstart", ThrSTATE(thr));
	/* NOTREACHED */
    }
    return THREAD_RET_CAST(returnav);	/* Available for anyone to join with */
					/* us unless we're detached, in which */
					/* case noone sees the value anyway. */
#endif    
}

static SV *
newthread(startsv, initargs, class)
SV *startsv;
AV *initargs;
char *class;
{
    dTHR;
    dSP;
    Thread savethread;
    int i;
    SV *sv;
    int err;
#ifndef THREAD_CREATE
    sigset_t fullmask, oldmask;
#endif
    
    savethread = thr;
    thr = new_struct_thread(thr);
    init_stacks(ARGS);
    SPAGAIN;
    DEBUG_L(PerlIO_printf(PerlIO_stderr(),
			  "%p: newthread, tid is %u, preparing stack\n",
			  savethread, thr->tid));
    /* The following pushes the arg list and startsv onto the *new* stack */
    PUSHMARK(sp);
    /* Could easily speed up the following greatly */
    for (i = 0; i <= AvFILL(initargs); i++)
	XPUSHs(SvREFCNT_inc(*av_fetch(initargs, i, FALSE)));
    XPUSHs(SvREFCNT_inc(startsv));
    PUTBACK;

#ifdef THREAD_CREATE
    THREAD_CREATE(thr, threadstart);
#else    
    /* On your marks... */
    MUTEX_LOCK(&thr->mutex);
    /* Get set...  */
    sigfillset(&fullmask);
    if (sigprocmask(SIG_SETMASK, &fullmask, &oldmask) == -1)
	croak("panic: sigprocmask");
    err = pthread_create(&thr->self, pthread_attr_default,
			 threadstart, (void*) thr);
    /* Go */
    MUTEX_UNLOCK(&thr->mutex);
#endif
    if (err) {
	/* Thread creation failed--clean up */
	SvREFCNT_dec(cvcache);
	remove_thread(thr);
	MUTEX_DESTROY(&thr->mutex);
	for (i = 0; i <= AvFILL(initargs); i++)
	    SvREFCNT_dec(*av_fetch(initargs, i, FALSE));
	SvREFCNT_dec(startsv);
	return NULL;
    }
#ifdef THREAD_POST_CREATE
    THREAD_POST_CREATE(thr);
#else
    if (sigprocmask(SIG_SETMASK, &oldmask, 0))
	croak("panic: sigprocmask");
#endif
    sv = newSViv(thr->tid);
    sv_magic(sv, oursv, '~', 0, 0);
    SvMAGIC(sv)->mg_private = Thread_MAGIC_SIGNATURE;
    return sv_bless(newRV_noinc(sv), gv_stashpv(class, TRUE));
}

static Signal_t
handle_thread_signal(sig)
int sig;
{
    char c = (char) sig;
    write(sig_pipe[0], &c, 1);
}

MODULE = Thread		PACKAGE = Thread

void
new(class, startsv, ...)
	char *		class
	SV *		startsv
	AV *		av = av_make(items - 2, &ST(2));
    PPCODE:
	XPUSHs(sv_2mortal(newthread(startsv, av, class)));

void
join(t)
	Thread	t
	AV *	av = NO_INIT
	int	i = NO_INIT
    PPCODE:
	DEBUG_L(PerlIO_printf(PerlIO_stderr(), "%p: joining %p (state %u)\n",
			      thr, t, ThrSTATE(t)););
    	MUTEX_LOCK(&t->mutex);
	switch (ThrSTATE(t)) {
	case THRf_R_JOINABLE:
	case THRf_R_JOINED:
	    ThrSETSTATE(t, THRf_R_JOINED);
	    MUTEX_UNLOCK(&t->mutex);
	    break;
	case THRf_ZOMBIE:
	    ThrSETSTATE(t, THRf_DEAD);
	    MUTEX_UNLOCK(&t->mutex);
	    remove_thread(t);
	    break;
	default:
	    MUTEX_UNLOCK(&t->mutex);
	    croak("can't join with thread");
	    /* NOTREACHED */
	}
	JOIN(t, &av);

	/* Could easily speed up the following if necessary */
	for (i = 0; i <= AvFILL(av); i++)
	    XPUSHs(sv_2mortal(*av_fetch(av, i, FALSE)));

void
detach(t)
	Thread	t
    CODE:
	DEBUG_L(PerlIO_printf(PerlIO_stderr(), "%p: detaching %p (state %u)\n",
			      thr, t, ThrSTATE(t)););
    	MUTEX_LOCK(&t->mutex);
	switch (ThrSTATE(t)) {
	case THRf_R_JOINABLE:
	    ThrSETSTATE(t, THRf_R_DETACHED);
	    /* fall through */
	case THRf_R_DETACHED:
	    DETACH(t);
	    MUTEX_UNLOCK(&t->mutex);
	    break;
	case THRf_ZOMBIE:
	    ThrSETSTATE(t, THRf_DEAD);
	    DETACH(t);
	    MUTEX_UNLOCK(&t->mutex);
	    remove_thread(t);
	    break;
	default:
	    MUTEX_UNLOCK(&t->mutex);
	    croak("can't detach thread");
	    /* NOTREACHED */
	}

void
equal(t1, t2)
	Thread	t1
	Thread	t2
    PPCODE:
	PUSHs((t1 == t2) ? &sv_yes : &sv_no);

void
flags(t)
	Thread	t
    PPCODE:
	PUSHs(sv_2mortal(newSViv(t->flags)));

void
self(class)
	char *	class
    PREINIT:
	SV *sv;
    PPCODE:
	sv = newSViv(thr->tid);
	sv_magic(sv, oursv, '~', 0, 0);
	SvMAGIC(sv)->mg_private = Thread_MAGIC_SIGNATURE;
	PUSHs(sv_2mortal(sv_bless(newRV_noinc(sv), gv_stashpv(class, TRUE))));

U32
tid(t)
	Thread	t
    CODE:
    	MUTEX_LOCK(&t->mutex);
	RETVAL = t->tid;
    	MUTEX_UNLOCK(&t->mutex);
    OUTPUT:
	RETVAL

void
DESTROY(t)
	SV *	t
    PPCODE:
	PUSHs(&sv_yes);

void
yield()
    CODE:
	YIELD;

void
cond_wait(sv)
	SV *	sv
	MAGIC *	mg = NO_INIT
CODE:
	if (SvROK(sv))
	    sv = SvRV(sv);

	mg = condpair_magic(sv);
	DEBUG_L(PerlIO_printf(PerlIO_stderr(), "%p: cond_wait %p\n", thr, sv));
	MUTEX_LOCK(MgMUTEXP(mg));
	if (MgOWNER(mg) != thr) {
	    MUTEX_UNLOCK(MgMUTEXP(mg));
	    croak("cond_wait for lock that we don't own\n");
	}
	MgOWNER(mg) = 0;
	COND_WAIT(MgCONDP(mg), MgMUTEXP(mg));
	while (MgOWNER(mg))
	    COND_WAIT(MgOWNERCONDP(mg), MgMUTEXP(mg));
	MgOWNER(mg) = thr;
	MUTEX_UNLOCK(MgMUTEXP(mg));
	
void
cond_signal(sv)
	SV *	sv
	MAGIC *	mg = NO_INIT
CODE:
	if (SvROK(sv))
	    sv = SvRV(sv);

	mg = condpair_magic(sv);
	DEBUG_L(PerlIO_printf(PerlIO_stderr(), "%p: cond_signal %p\n",thr,sv));
	MUTEX_LOCK(MgMUTEXP(mg));
	if (MgOWNER(mg) != thr) {
	    MUTEX_UNLOCK(MgMUTEXP(mg));
	    croak("cond_signal for lock that we don't own\n");
	}
	COND_SIGNAL(MgCONDP(mg));
	MUTEX_UNLOCK(MgMUTEXP(mg));

void
cond_broadcast(sv)
	SV *	sv
	MAGIC *	mg = NO_INIT
CODE:
	if (SvROK(sv))
	    sv = SvRV(sv);

	mg = condpair_magic(sv);
	DEBUG_L(PerlIO_printf(PerlIO_stderr(), "%p: cond_broadcast %p\n",
			      thr, sv));
	MUTEX_LOCK(MgMUTEXP(mg));
	if (MgOWNER(mg) != thr) {
	    MUTEX_UNLOCK(MgMUTEXP(mg));
	    croak("cond_broadcast for lock that we don't own\n");
	}
	COND_BROADCAST(MgCONDP(mg));
	MUTEX_UNLOCK(MgMUTEXP(mg));

void
list(class)
	char *	class
    PREINIT:
	Thread	t;
	AV *	av;
	SV **	svp;
	int	n = 0;
    PPCODE:
	av = newAV();
	/*
	 * Iterate until we have enough dynamic storage for all threads.
	 * We mustn't do any allocation while holding threads_mutex though.
	 */
	MUTEX_LOCK(&threads_mutex);
	do {
	    n = nthreads;
	    MUTEX_UNLOCK(&threads_mutex);
	    if (AvFILL(av) < n - 1) {
		int i = AvFILL(av);
		for (i = AvFILL(av); i < n - 1; i++) {
		    SV *sv = newSViv(0);	/* fill in tid later */
		    sv_magic(sv, 0, '~', 0, 0);	/* fill in other magic later */
		    av_push(av, sv_bless(newRV_noinc(sv),
					 gv_stashpv(class, TRUE)));
	
		}
	    }
	    MUTEX_LOCK(&threads_mutex);
	} while (n < nthreads);
	n = nthreads;	/* Get the final correct value */

	/*
	 * At this point, there's enough room to fill in av.
	 * Note that we are holding threads_mutex so the list
	 * won't change out from under us but all the remaining
	 * processing is "fast" (no blocking, malloc etc.)
	 */
	t = thr;
	svp = AvARRAY(av);
	do {
	    SV *sv = (SV*)SvRV(*svp);
	    sv_setiv(sv, t->tid);
	    SvMAGIC(sv)->mg_obj = SvREFCNT_inc(t->Toursv);
	    SvMAGIC(sv)->mg_flags |= MGf_REFCOUNTED;
	    SvMAGIC(sv)->mg_private = Thread_MAGIC_SIGNATURE;
	    t = t->next;
	    svp++;
	} while (t != thr);
	/*  */
	MUTEX_UNLOCK(&threads_mutex);
	/* Truncate any unneeded slots in av */
	av_fill(av, n - 1);
	/* Finally, push all the new objects onto the stack and drop av */
	EXTEND(sp, n);
	for (svp = AvARRAY(av); n > 0; n--, svp++)
	    PUSHs(*svp);
	(void)sv_2mortal((SV*)av);


MODULE = Thread		PACKAGE = Thread::Signal

void
kill_sighandler_thread()
    PPCODE:
	write(sig_pipe[0], "\0", 1);
	PUSHs(&sv_yes);

void
init_thread_signals()
    PPCODE:
	sighandlerp = handle_thread_signal;
	if (pipe(sig_pipe) == -1)
	    XSRETURN_UNDEF;
	PUSHs(&sv_yes);

SV *
await_signal()
    PREINIT:
	char c;
	SSize_t ret;
    CODE:
	do {
	    ret = read(sig_pipe[1], &c, 1);
	} while (ret == -1 && errno == EINTR);
	if (ret == -1)
	    croak("panic: await_signal");
	if (ret == 0)
	    XSRETURN_UNDEF;
	RETVAL = c ? psig_ptr[c] : &sv_no;
    OUTPUT:
	RETVAL
Commit	Line	Data
d9bb3666	1	#include "EXTERN.h"
	2	#include "perl.h"
	3	#include "XSUB.h"
	4
7d901afa	5	/* Magic signature for Thread's mg_private is "Th" */
	6	#define Thread_MAGIC_SIGNATURE 0x5468
	7
	8	static U32 threadnum = 0;
85ced67f	9	static int sig_pipe[2];
683929b4	10
7d901afa	11	static void
	12	remove_thread(t)
	13	Thread t;
	14	{
	15	DEBUG_L(WITH_THR(PerlIO_printf(PerlIO_stderr(),
	16	"%p: remove_thread %p\n", thr, t)));
	17	MUTEX_LOCK(&threads_mutex);
0a00ffdb	18	MUTEX_DESTROY(&t->mutex);
7d901afa	19	nthreads--;
	20	t->prev->next = t->next;
	21	t->next->prev = t->prev;
	22	COND_BROADCAST(&nthreads_cond);
	23	MUTEX_UNLOCK(&threads_mutex);
	24	}
	25
ea0efc06	26	static THREAD_RET_TYPE
d9bb3666	27	threadstart(arg)
	28	void *arg;
	29	{
783070da	30	#ifdef FAKE_THREADS
	31	Thread savethread = thr;
	32	LOGOP myop;
	33	dSP;
	34	I32 oldscope = scopestack_ix;
	35	I32 retval;
50112d62	36	AV *returnav;
783070da	37	int i;
783070da	38
683929b4	39	DEBUG_L(PerlIO_printf(PerlIO_stderr(), "new thread %p starting at %s\n",
683929b4	40	thr, SvPEEK(TOPs)));
783070da	41	thr = (Thread) arg;
	42	savemark = TOPMARK;
	43	thr->prev = thr->prev_run = savethread;
	44	thr->next = savethread->next;
	45	thr->next_run = savethread->next_run;
	46	savethread->next = savethread->next_run = thr;
	47	thr->wait_queue = 0;
	48	thr->private = 0;
	49
	50	/* Now duplicate most of perl_call_sv but with a few twists */
	51	op = (OP*)&myop;
	52	Zero(op, 1, LOGOP);
	53	myop.op_flags = OPf_STACKED;
	54	myop.op_next = Nullop;
	55	myop.op_flags \|= OPf_KNOW;
	56	myop.op_flags \|= OPf_WANT_LIST;
	57	op = pp_entersub(ARGS);
	58	DEBUG_L(if (!op)
	59	PerlIO_printf(PerlIO_stderr(), "thread starts at Nullop\n"));
	60	/*
	61	* When this thread is next scheduled, we start in the right
	62	* place. When the thread runs off the end of the sub, perl.c
	63	* handles things, using savemark to figure out how much of the
	64	* stack is the return value for any join.
	65	*/
	66	thr = savethread; /* back to the old thread */
	67	return 0;
	68	#else
d9bb3666	69	Thread thr = (Thread) arg;
	70	LOGOP myop;
	71	dSP;
	72	I32 oldmark = TOPMARK;
	73	I32 oldscope = scopestack_ix;
	74	I32 retval;
50112d62	75	AV *returnav;
14fcddff	76	int i, ret;
783070da	77	dJMPENV;
783070da	78
783070da	79	/* Don't call anything requiring dTHR until after pthread_setspecific */
d9bb3666	80	/*
	81	* Wait until our creator releases us. If we didn't do this, then
	82	* it would be potentially possible for out thread to carry on and
	83	* do stuff before our creator fills in our "self" field. For example,
ea0efc06	84	* if we went and created another thread which tried to JOIN with us,
ea0efc06	85	* then we'd be in a mess.
d9bb3666	86	*/
50112d62	87	MUTEX_LOCK(&thr->mutex);
50112d62	88	MUTEX_UNLOCK(&thr->mutex);
d9bb3666	89
d9bb3666	90	/*
	91	* It's safe to wait until now to set the thread-specific pointer
	92	* from our pthread_t structure to our struct thread, since we're
	93	* the only thread who can get at it anyway.
	94	*/
ea0efc06	95	SET_THR(thr);
d9bb3666	96
783070da	97	/* Only now can we use SvPEEK (which calls sv_newmortal which does dTHR) */
683929b4	98	DEBUG_L(PerlIO_printf(PerlIO_stderr(), "new thread %p starting at %s\n",
683929b4	99	thr, SvPEEK(TOPs)));
783070da	100
	101	JMPENV_PUSH(ret);
	102	switch (ret) {
	103	case 3:
	104	PerlIO_printf(PerlIO_stderr(), "panic: threadstart\n");
d9bb3666	105	/* fall through */
783070da	106	case 1:
783070da	107	STATUS_ALL_FAILURE;
d9bb3666	108	/* fall through */
783070da	109	case 2:
	110	/* my_exit() was called */
	111	while (scopestack_ix > oldscope)
	112	LEAVE;
	113	JMPENV_POP;
d9bb3666	114	av_store(returnav, 0, newSViv(statusvalue));
	115	goto finishoff;
	116	}
	117
	118	/* Now duplicate most of perl_call_sv but with a few twists */
	119	op = (OP*)&myop;
	120	Zero(op, 1, LOGOP);
	121	myop.op_flags = OPf_STACKED;
	122	myop.op_next = Nullop;
	123	myop.op_flags \|= OPf_KNOW;
783070da	124	myop.op_flags \|= OPf_WANT_LIST;
d9bb3666	125	op = pp_entersub(ARGS);
	126	if (op)
	127	runops();
734689b1	128	SPAGAIN;
	129	retval = sp - (stack_base + oldmark);
	130	sp = stack_base + oldmark + 1;
783070da	131	DEBUG_L(for (i = 1; i <= retval; i++)
	132	PerlIO_printf(PerlIO_stderr(),
	133	"%p returnav[%d] = %s\n",
	134	thr, i, SvPEEK(sp[i - 1]));)
50112d62	135	returnav = newAV();
d9bb3666	136	av_store(returnav, 0, newSVpv("", 0));
734689b1	137	for (i = 1; i <= retval; i++, sp++)
	138	sv_setsv(av_fetch(returnav, i, TRUE), SvREFCNT_inc(sp));
	139
d9bb3666	140	finishoff:
783070da	141	#if 0
	142	/* removed for debug */
	143	SvREFCNT_dec(curstack);
	144	#endif
d9bb3666	145	SvREFCNT_dec(cvcache);
554b3eca	146	SvREFCNT_dec(thr->magicals);
554b3eca	147	SvREFCNT_dec(thr->specific);
d9bb3666	148	Safefree(markstack);
	149	Safefree(scopestack);
	150	Safefree(savestack);
	151	Safefree(retstack);
	152	Safefree(cxstack);
	153	Safefree(tmps_stack);
	154
14fcddff	155	MUTEX_LOCK(&thr->mutex);
50112d62	156	DEBUG_L(PerlIO_printf(PerlIO_stderr(),
	157	"%p: threadstart finishing: state is %u\n",
	158	thr, ThrSTATE(thr)));
14fcddff	159	switch (ThrSTATE(thr)) {
	160	case THRf_R_JOINABLE:
	161	ThrSETSTATE(thr, THRf_ZOMBIE);
	162	MUTEX_UNLOCK(&thr->mutex);
783070da	163	DEBUG_L(PerlIO_printf(PerlIO_stderr(),
14fcddff	164	"%p: R_JOINABLE thread finished\n", thr));
	165	break;
	166	case THRf_R_JOINED:
	167	ThrSETSTATE(thr, THRf_DEAD);
	168	MUTEX_UNLOCK(&thr->mutex);
50112d62	169	remove_thread(thr);
14fcddff	170	DEBUG_L(PerlIO_printf(PerlIO_stderr(),
	171	"%p: R_JOINED thread finished\n", thr));
	172	break;
50112d62	173	case THRf_R_DETACHED:
683929b4	174	ThrSETSTATE(thr, THRf_DEAD);
14fcddff	175	MUTEX_UNLOCK(&thr->mutex);
14fcddff	176	SvREFCNT_dec(returnav);
	177	DEBUG_L(PerlIO_printf(PerlIO_stderr(),
	178	"%p: DETACHED thread finished\n", thr));
50112d62	179	remove_thread(thr); /* This might trigger main thread to finish */
14fcddff	180	break;
	181	default:
	182	MUTEX_UNLOCK(&thr->mutex);
	183	croak("panic: illegal state %u at end of threadstart", ThrSTATE(thr));
	184	/* NOTREACHED */
734689b1	185	}
ea0efc06	186	return THREAD_RET_CAST(returnav); /* Available for anyone to join with */
	187	/* us unless we're detached, in which */
	188	/* case noone sees the value anyway. */
783070da	189	#endif
d9bb3666	190	}
d9bb3666	191
683929b4	192	static SV *
683929b4	193	newthread(startsv, initargs, class)
d9bb3666	194	SV *startsv;
d9bb3666	195	AV *initargs;
683929b4	196	char *class;
d9bb3666	197	{
	198	dTHR;
	199	dSP;
	200	Thread savethread;
	201	int i;
683929b4	202	SV *sv;
ea0efc06	203	int err;
ea0efc06	204	#ifndef THREAD_CREATE
f152979c	205	sigset_t fullmask, oldmask;
ea0efc06	206	#endif
d9bb3666	207
d9bb3666	208	savethread = thr;
a863c7d1	209	thr = new_struct_thread(thr);
d9bb3666	210	init_stacks(ARGS);
d9bb3666	211	SPAGAIN;
50112d62	212	DEBUG_L(PerlIO_printf(PerlIO_stderr(),
	213	"%p: newthread, tid is %u, preparing stack\n",
	214	savethread, thr->tid));
d9bb3666	215	/* The following pushes the arg list and startsv onto the new stack */
	216	PUSHMARK(sp);
	217	/* Could easily speed up the following greatly */
734689b1	218	for (i = 0; i <= AvFILL(initargs); i++)
d9bb3666	219	XPUSHs(SvREFCNT_inc(*av_fetch(initargs, i, FALSE)));
	220	XPUSHs(SvREFCNT_inc(startsv));
	221	PUTBACK;
	222
ea0efc06	223	#ifdef THREAD_CREATE
ea0efc06	224	THREAD_CREATE(thr, threadstart);
783070da	225	#else
d9bb3666	226	/* On your marks... */
14fcddff	227	MUTEX_LOCK(&thr->mutex);
ea0efc06	228	/* Get set... */
f152979c	229	sigfillset(&fullmask);
	230	if (sigprocmask(SIG_SETMASK, &fullmask, &oldmask) == -1)
	231	croak("panic: sigprocmask");
46930d8f	232	err = pthread_create(&thr->self, pthread_attr_default,
46930d8f	233	threadstart, (void*) thr);
d9bb3666	234	/* Go */
14fcddff	235	MUTEX_UNLOCK(&thr->mutex);
ea0efc06	236	#endif
	237	if (err) {
	238	/* Thread creation failed--clean up */
	239	SvREFCNT_dec(cvcache);
	240	remove_thread(thr);
	241	MUTEX_DESTROY(&thr->mutex);
	242	for (i = 0; i <= AvFILL(initargs); i++)
	243	SvREFCNT_dec(*av_fetch(initargs, i, FALSE));
	244	SvREFCNT_dec(startsv);
	245	return NULL;
	246	}
	247	#ifdef THREAD_POST_CREATE
	248	THREAD_POST_CREATE(thr);
	249	#else
f152979c	250	if (sigprocmask(SIG_SETMASK, &oldmask, 0))
f152979c	251	croak("panic: sigprocmask");
783070da	252	#endif
7d901afa	253	sv = newSViv(thr->tid);
683929b4	254	sv_magic(sv, oursv, '~', 0, 0);
7d901afa	255	SvMAGIC(sv)->mg_private = Thread_MAGIC_SIGNATURE;
7d901afa	256	return sv_bless(newRV_noinc(sv), gv_stashpv(class, TRUE));
d9bb3666	257	}
d9bb3666	258
f152979c	259	static Signal_t
	260	handle_thread_signal(sig)
	261	int sig;
	262	{
	263	char c = (char) sig;
	264	write(sig_pipe[0], &c, 1);
	265	}
	266
d9bb3666	267	MODULE = Thread PACKAGE = Thread
d9bb3666	268
683929b4	269	void
d9bb3666	270	new(class, startsv, ...)
683929b4	271	char * class
d9bb3666	272	SV * startsv
734689b1	273	AV * av = av_make(items - 2, &ST(2));
683929b4	274	PPCODE:
683929b4	275	XPUSHs(sv_2mortal(newthread(startsv, av, class)));
d9bb3666	276
d9bb3666	277	void
d9bb3666	278	join(t)
	279	Thread t
	280	AV * av = NO_INIT
	281	int i = NO_INIT
	282	PPCODE:
7d901afa	283	DEBUG_L(PerlIO_printf(PerlIO_stderr(), "%p: joining %p (state %u)\n",
7d901afa	284	thr, t, ThrSTATE(t)););
50112d62	285	MUTEX_LOCK(&t->mutex);
50112d62	286	switch (ThrSTATE(t)) {
14fcddff	287	case THRf_R_JOINABLE:
14fcddff	288	case THRf_R_JOINED:
50112d62	289	ThrSETSTATE(t, THRf_R_JOINED);
50112d62	290	MUTEX_UNLOCK(&t->mutex);
14fcddff	291	break;
14fcddff	292	case THRf_ZOMBIE:
50112d62	293	ThrSETSTATE(t, THRf_DEAD);
	294	MUTEX_UNLOCK(&t->mutex);
	295	remove_thread(t);
14fcddff	296	break;
14fcddff	297	default:
50112d62	298	MUTEX_UNLOCK(&t->mutex);
14fcddff	299	croak("can't join with thread");
	300	/* NOTREACHED */
	301	}
ea0efc06	302	JOIN(t, &av);
7d901afa	303
d9bb3666	304	/* Could easily speed up the following if necessary */
	305	for (i = 0; i <= AvFILL(av); i++)
	306	XPUSHs(sv_2mortal(*av_fetch(av, i, FALSE)));
	307
	308	void
734689b1	309	detach(t)
d9bb3666	310	Thread t
d9bb3666	311	CODE:
7d901afa	312	DEBUG_L(PerlIO_printf(PerlIO_stderr(), "%p: detaching %p (state %u)\n",
7d901afa	313	thr, t, ThrSTATE(t)););
50112d62	314	MUTEX_LOCK(&t->mutex);
50112d62	315	switch (ThrSTATE(t)) {
14fcddff	316	case THRf_R_JOINABLE:
50112d62	317	ThrSETSTATE(t, THRf_R_DETACHED);
14fcddff	318	/* fall through */
50112d62	319	case THRf_R_DETACHED:
14fcddff	320	DETACH(t);
50112d62	321	MUTEX_UNLOCK(&t->mutex);
14fcddff	322	break;
14fcddff	323	case THRf_ZOMBIE:
50112d62	324	ThrSETSTATE(t, THRf_DEAD);
7d901afa	325	DETACH(t);
50112d62	326	MUTEX_UNLOCK(&t->mutex);
50112d62	327	remove_thread(t);
14fcddff	328	break;
14fcddff	329	default:
50112d62	330	MUTEX_UNLOCK(&t->mutex);
14fcddff	331	croak("can't detach thread");
14fcddff	332	/* NOTREACHED */
734689b1	333	}
d9bb3666	334
d9bb3666	335	void
7d901afa	336	equal(t1, t2)
	337	Thread t1
	338	Thread t2
	339	PPCODE:
	340	PUSHs((t1 == t2) ? &sv_yes : &sv_no);
	341
	342	void
	343	flags(t)
	344	Thread t
	345	PPCODE:
	346	PUSHs(sv_2mortal(newSViv(t->flags)));
	347
	348	void
	349	self(class)
	350	char * class
	351	PREINIT:
	352	SV *sv;
	353	PPCODE:
	354	sv = newSViv(thr->tid);
	355	sv_magic(sv, oursv, '~', 0, 0);
	356	SvMAGIC(sv)->mg_private = Thread_MAGIC_SIGNATURE;
	357	PUSHs(sv_2mortal(sv_bless(newRV_noinc(sv), gv_stashpv(class, TRUE))));
	358
50112d62	359	U32
	360	tid(t)
	361	Thread t
	362	CODE:
	363	MUTEX_LOCK(&t->mutex);
	364	RETVAL = t->tid;
	365	MUTEX_UNLOCK(&t->mutex);
	366	OUTPUT:
	367	RETVAL
	368
	369	void
	370	DESTROY(t)
	371	SV * t
	372	PPCODE:
	373	PUSHs(&sv_yes);
	374
7d901afa	375	void
734689b1	376	yield()
d9bb3666	377	CODE:
ea0efc06	378	YIELD;
d9bb3666	379
d9bb3666	380	void
734689b1	381	cond_wait(sv)
	382	SV * sv
	383	MAGIC * mg = NO_INIT
	384	CODE:
2c127b02	385	if (SvROK(sv))
734689b1	386	sv = SvRV(sv);
2c127b02	387
734689b1	388	mg = condpair_magic(sv);
683929b4	389	DEBUG_L(PerlIO_printf(PerlIO_stderr(), "%p: cond_wait %p\n", thr, sv));
734689b1	390	MUTEX_LOCK(MgMUTEXP(mg));
	391	if (MgOWNER(mg) != thr) {
	392	MUTEX_UNLOCK(MgMUTEXP(mg));
	393	croak("cond_wait for lock that we don't own\n");
	394	}
	395	MgOWNER(mg) = 0;
	396	COND_WAIT(MgCONDP(mg), MgMUTEXP(mg));
50112d62	397	while (MgOWNER(mg))
50112d62	398	COND_WAIT(MgOWNERCONDP(mg), MgMUTEXP(mg));
734689b1	399	MgOWNER(mg) = thr;
	400	MUTEX_UNLOCK(MgMUTEXP(mg));
	401
	402	void
	403	cond_signal(sv)
	404	SV * sv
	405	MAGIC * mg = NO_INIT
	406	CODE:
50112d62	407	if (SvROK(sv))
734689b1	408	sv = SvRV(sv);
50112d62	409
734689b1	410	mg = condpair_magic(sv);
683929b4	411	DEBUG_L(PerlIO_printf(PerlIO_stderr(), "%p: cond_signal %p\n",thr,sv));
734689b1	412	MUTEX_LOCK(MgMUTEXP(mg));
	413	if (MgOWNER(mg) != thr) {
	414	MUTEX_UNLOCK(MgMUTEXP(mg));
	415	croak("cond_signal for lock that we don't own\n");
	416	}
	417	COND_SIGNAL(MgCONDP(mg));
	418	MUTEX_UNLOCK(MgMUTEXP(mg));
d9bb3666	419
734689b1	420	void
	421	cond_broadcast(sv)
	422	SV * sv
	423	MAGIC * mg = NO_INIT
	424	CODE:
783070da	425	if (SvROK(sv))
734689b1	426	sv = SvRV(sv);
783070da	427
734689b1	428	mg = condpair_magic(sv);
683929b4	429	DEBUG_L(PerlIO_printf(PerlIO_stderr(), "%p: cond_broadcast %p\n",
683929b4	430	thr, sv));
734689b1	431	MUTEX_LOCK(MgMUTEXP(mg));
	432	if (MgOWNER(mg) != thr) {
	433	MUTEX_UNLOCK(MgMUTEXP(mg));
	434	croak("cond_broadcast for lock that we don't own\n");
	435	}
	436	COND_BROADCAST(MgCONDP(mg));
	437	MUTEX_UNLOCK(MgMUTEXP(mg));
f152979c	438
7d901afa	439	void
	440	list(class)
	441	char * class
	442	PREINIT:
	443	Thread t;
	444	AV * av;
	445	SV ** svp;
	446	int n = 0;
	447	PPCODE:
	448	av = newAV();
	449	/*
	450	* Iterate until we have enough dynamic storage for all threads.
	451	* We mustn't do any allocation while holding threads_mutex though.
	452	*/
	453	MUTEX_LOCK(&threads_mutex);
	454	do {
	455	n = nthreads;
	456	MUTEX_UNLOCK(&threads_mutex);
	457	if (AvFILL(av) < n - 1) {
	458	int i = AvFILL(av);
	459	for (i = AvFILL(av); i < n - 1; i++) {
	460	SV sv = newSViv(0); / fill in tid later */
	461	sv_magic(sv, 0, '~', 0, 0); /* fill in other magic later */
	462	av_push(av, sv_bless(newRV_noinc(sv),
	463	gv_stashpv(class, TRUE)));
50112d62	464
7d901afa	465	}
	466	}
	467	MUTEX_LOCK(&threads_mutex);
	468	} while (n < nthreads);
50112d62	469	n = nthreads; /* Get the final correct value */
7d901afa	470
	471	/*
	472	* At this point, there's enough room to fill in av.
	473	* Note that we are holding threads_mutex so the list
	474	* won't change out from under us but all the remaining
	475	* processing is "fast" (no blocking, malloc etc.)
	476	*/
	477	t = thr;
	478	svp = AvARRAY(av);
	479	do {
0a00ffdb	480	SV sv = (SV)SvRV(*svp);
7d901afa	481	sv_setiv(sv, t->tid);
	482	SvMAGIC(sv)->mg_obj = SvREFCNT_inc(t->Toursv);
	483	SvMAGIC(sv)->mg_flags \|= MGf_REFCOUNTED;
	484	SvMAGIC(sv)->mg_private = Thread_MAGIC_SIGNATURE;
	485	t = t->next;
0a00ffdb	486	svp++;
7d901afa	487	} while (t != thr);
50112d62	488	/* */
7d901afa	489	MUTEX_UNLOCK(&threads_mutex);
7d901afa	490	/* Truncate any unneeded slots in av */
50112d62	491	av_fill(av, n - 1);
7d901afa	492	/* Finally, push all the new objects onto the stack and drop av */
	493	EXTEND(sp, n);
	494	for (svp = AvARRAY(av); n > 0; n--, svp++)
	495	PUSHs(*svp);
	496	(void)sv_2mortal((SV*)av);
	497
	498
f152979c	499	MODULE = Thread PACKAGE = Thread::Signal
	500
	501	void
	502	kill_sighandler_thread()
	503	PPCODE:
	504	write(sig_pipe[0], "\0", 1);
	505	PUSHs(&sv_yes);
	506
	507	void
	508	init_thread_signals()
	509	PPCODE:
	510	sighandlerp = handle_thread_signal;
	511	if (pipe(sig_pipe) == -1)
	512	XSRETURN_UNDEF;
	513	PUSHs(&sv_yes);
	514
	515	SV *
	516	await_signal()
	517	PREINIT:
	518	char c;
ea0efc06	519	SSize_t ret;
f152979c	520	CODE:
	521	do {
	522	ret = read(sig_pipe[1], &c, 1);
	523	} while (ret == -1 && errno == EINTR);
	524	if (ret == -1)
	525	croak("panic: await_signal");
	526	if (ret == 0)
	527	XSRETURN_UNDEF;
	528	RETVAL = c ? psig_ptr[c] : &sv_no;
	529	OUTPUT:
	530	RETVAL