[ViewVC] Diff of: cvs/Coro/Coro/State.xs

Comparing Coro/Coro/State.xs (file contents):
Revision 1.152 by root, Thu Sep 20 12:02:25 2007 UTC vs.
Revision 1.175 by root, Sun Sep 30 13:43:45 2007 UTC

…		…
88	/* prefer perl internal functions over our own? */	88	/* prefer perl internal functions over our own? */
89	#ifndef CORO_PREFER_PERL_FUNCTIONS	89	#ifndef CORO_PREFER_PERL_FUNCTIONS
90	# define CORO_PREFER_PERL_FUNCTIONS 0	90	# define CORO_PREFER_PERL_FUNCTIONS 0
91	#endif	91	#endif
92		92
93	/* The next macro should declare a variable stacklevel that contains and approximation	93	/* The next macros try to return the current stack pointer, in an as
94	* to the current C stack pointer. Its property is that it changes with each call	94	* portable way as possible. */
95	* and should be unique. */
96	#define dSTACKLEVEL int stacklevel	95	#define dSTACKLEVEL volatile char stacklevel
97	#define STACKLEVEL ((void *)&stacklevel)	96	#define STACKLEVEL ((void *)&stacklevel)
98		97
99	#define IN_DESTRUCT (PL_main_cv == Nullcv)	98	#define IN_DESTRUCT (PL_main_cv == Nullcv)
100		99
101	#if __GNUC__ >= 3	100	#if __GNUC__ >= 3
…		…
133	static AV main_mainstack; / used to differentiate between $main and others */	132	static AV main_mainstack; / used to differentiate between $main and others */
134	static JMPENV *main_top_env;	133	static JMPENV *main_top_env;
135	static HV coro_state_stash, coro_stash;	134	static HV coro_state_stash, coro_stash;
136	static SV coro_mortal; / will be freed after next transfer */	135	static SV coro_mortal; / will be freed after next transfer */
137		136
		137	/* async_pool helper stuff */
		138	static SV *sv_pool_rss;
		139	static SV *sv_pool_size;
		140	static AV *av_async_pool;
		141
138	static struct coro_cctx *cctx_first;	142	static struct coro_cctx *cctx_first;
139	static int cctx_count, cctx_idle;	143	static int cctx_count, cctx_idle;
		144
		145	enum {
		146	CC_MAPPED = 0x01,
		147	CC_NOREUSE = 0x02, /* throw this away after tracing */
		148	CC_TRACE = 0x04,
		149	CC_TRACE_SUB = 0x08, /* trace sub calls */
		150	CC_TRACE_LINE = 0x10, /* trace each statement */
		151	CC_TRACE_ALL = CC_TRACE_SUB \| CC_TRACE_LINE,
		152	};
140		153
141	/* this is a structure representing a c-level coroutine */	154	/* this is a structure representing a c-level coroutine */
142	typedef struct coro_cctx {	155	typedef struct coro_cctx {
143	struct coro_cctx *next;	156	struct coro_cctx *next;
144		157
…		…
153	coro_context cctx;	166	coro_context cctx;
154		167
155	#if CORO_USE_VALGRIND	168	#if CORO_USE_VALGRIND
156	int valgrind_id;	169	int valgrind_id;
157	#endif	170	#endif
158	char inuse, mapped;	171	unsigned char flags;
159	} coro_cctx;	172	} coro_cctx;
160		173
161	enum {	174	enum {
162	CF_RUNNING = 0x0001, /* coroutine is running */	175	CF_RUNNING = 0x0001, /* coroutine is running */
163	CF_READY = 0x0002, /* coroutine is ready */	176	CF_READY = 0x0002, /* coroutine is ready */
…		…
186		199
187	#define VAR(name,type) type name;	200	#define VAR(name,type) type name;
188	# include "state.h"	201	# include "state.h"
189	#undef VAR	202	#undef VAR
190		203
		204	/* statistics */
		205	int usecount; /* number of switches to this coro */
		206
191	/* coro process data */	207	/* coro process data */
192	int prio;	208	int prio;
193		209
194	/* linked list */	210	/* linked list */
195	struct coro next, prev;	211	struct coro next, prev;
…		…
210		226
211	/* for Coro.pm */	227	/* for Coro.pm */
212	static SV *coro_current;	228	static SV *coro_current;
213	static AV *coro_ready [PRIO_MAX-PRIO_MIN+1];	229	static AV *coro_ready [PRIO_MAX-PRIO_MIN+1];
214	static int coro_nready;	230	static int coro_nready;
215	static struct coro *first;	231	static struct coro *coro_first;
216		232
217	/ lowlevel stuff ********************************************************/	233	/ lowlevel stuff ********************************************************/
218		234
219	static AV *	235	static AV *
220	coro_clone_padlist (pTHX_ CV *cv)	236	coro_clone_padlist (pTHX_ CV *cv)
…		…
285	SvMAGIC (cv) \	301	SvMAGIC (cv) \
286	? SvMAGIC (cv)->mg_type == PERL_MAGIC_coro \	302	? SvMAGIC (cv)->mg_type == PERL_MAGIC_coro \
287	? SvMAGIC (cv) \	303	? SvMAGIC (cv) \
288	: mg_find ((SV *)cv, PERL_MAGIC_coro) \	304	: mg_find ((SV *)cv, PERL_MAGIC_coro) \
289	: 0	305	: 0
		306
		307	static struct coro *
		308	SvSTATE_ (pTHX_ SV *coro)
		309	{
		310	HV *stash;
		311	MAGIC *mg;
		312
		313	if (SvROK (coro))
		314	coro = SvRV (coro);
		315
		316	if (SvTYPE (coro) != SVt_PVHV)
		317	croak ("Coro::State object required");
		318
		319	stash = SvSTASH (coro);
		320	if (stash != coro_stash && stash != coro_state_stash)
		321	{
		322	/* very slow, but rare, check */
		323	if (!sv_derived_from (sv_2mortal (newRV_inc (coro)), "Coro::State"))
		324	croak ("Coro::State object required");
		325	}
		326
		327	mg = CORO_MAGIC (coro);
		328	return (struct coro *)mg->mg_ptr;
		329	}
		330
		331	#define SvSTATE(sv) SvSTATE_ (aTHX_ (sv))
290		332
291	/* the next two functions merely cache the padlists */	333	/* the next two functions merely cache the padlists */
292	static void	334	static void
293	get_padlist (pTHX_ CV *cv)	335	get_padlist (pTHX_ CV *cv)
294	{	336	{
…		…
395	/*	437	/*
396	* the worst thing you can imagine happens first - we have to save	438	* the worst thing you can imagine happens first - we have to save
397	* (and reinitialize) all cv's in the whole callchain :(	439	* (and reinitialize) all cv's in the whole callchain :(
398	*/	440	*/
399		441
400	EXTEND (SP, 3 + 1);
401	PUSHs (Nullsv);	442	XPUSHs (Nullsv);
402	/* this loop was inspired by pp_caller */	443	/* this loop was inspired by pp_caller */
403	for (;;)	444	for (;;)
404	{	445	{
405	while (cxix >= 0)	446	while (cxix >= 0)
406	{	447	{
…		…
425		466
426	if (top_si->si_type == PERLSI_MAIN)	467	if (top_si->si_type == PERLSI_MAIN)
427	break;	468	break;
428		469
429	top_si = top_si->si_prev;	470	top_si = top_si->si_prev;
430	ccstk = top_si->si_cxstack;	471	ccstk = top_si->si_cxstack;
431	cxix = top_si->si_cxix;	472	cxix = top_si->si_cxix;
432	}	473	}
433		474
434	PUTBACK;	475	PUTBACK;
435	}	476	}
436		477
…		…
455	# define coro_init_stacks init_stacks	496	# define coro_init_stacks init_stacks
456	#else	497	#else
457	static void	498	static void
458	coro_init_stacks (pTHX)	499	coro_init_stacks (pTHX)
459	{	500	{
460	PL_curstackinfo = new_stackinfo(128, 1024/sizeof(PERL_CONTEXT));	501	PL_curstackinfo = new_stackinfo(64, 6);
461	PL_curstackinfo->si_type = PERLSI_MAIN;	502	PL_curstackinfo->si_type = PERLSI_MAIN;
462	PL_curstack = PL_curstackinfo->si_stack;	503	PL_curstack = PL_curstackinfo->si_stack;
463	PL_mainstack = PL_curstack; /* remember in case we switch stacks */	504	PL_mainstack = PL_curstack; /* remember in case we switch stacks */
464		505
465	PL_stack_base = AvARRAY(PL_curstack);	506	PL_stack_base = AvARRAY(PL_curstack);
466	PL_stack_sp = PL_stack_base;	507	PL_stack_sp = PL_stack_base;
467	PL_stack_max = PL_stack_base + AvMAX(PL_curstack);	508	PL_stack_max = PL_stack_base + AvMAX(PL_curstack);
468		509
469	New(50,PL_tmps_stack,128,SV*);	510	New(50,PL_tmps_stack,64,SV*);
470	PL_tmps_floor = -1;	511	PL_tmps_floor = -1;
471	PL_tmps_ix = -1;	512	PL_tmps_ix = -1;
472	PL_tmps_max = 128;	513	PL_tmps_max = 64;
473		514
474	New(54,PL_markstack,32,I32);	515	New(54,PL_markstack,16,I32);
475	PL_markstack_ptr = PL_markstack;	516	PL_markstack_ptr = PL_markstack;
476	PL_markstack_max = PL_markstack + 32;	517	PL_markstack_max = PL_markstack + 16;
477		518
478	#ifdef SET_MARK_OFFSET	519	#ifdef SET_MARK_OFFSET
479	SET_MARK_OFFSET;	520	SET_MARK_OFFSET;
480	#endif	521	#endif
481		522
482	New(54,PL_scopestack,32,I32);	523	New(54,PL_scopestack,16,I32);
483	PL_scopestack_ix = 0;	524	PL_scopestack_ix = 0;
484	PL_scopestack_max = 32;	525	PL_scopestack_max = 16;
485		526
486	New(54,PL_savestack,64,ANY);	527	New(54,PL_savestack,64,ANY);
487	PL_savestack_ix = 0;	528	PL_savestack_ix = 0;
488	PL_savestack_max = 64;	529	PL_savestack_max = 64;
489		530
490	#if !PERL_VERSION_ATLEAST (5,9,0)	531	#if !PERL_VERSION_ATLEAST (5,9,0)
491	New(54,PL_retstack,16,OP*);	532	New(54,PL_retstack,4,OP*);
492	PL_retstack_ix = 0;	533	PL_retstack_ix = 0;
493	PL_retstack_max = 16;	534	PL_retstack_max = 4;
494	#endif	535	#endif
495	}	536	}
496	#endif	537	#endif
497		538
498	/*	539	/*
…		…
541	Safefree (PL_retstack);	582	Safefree (PL_retstack);
542	#endif	583	#endif
543	}	584	}
544		585
545	static size_t	586	static size_t
546	coro_rss (struct coro *coro)	587	coro_rss (pTHX_ struct coro *coro)
547	{	588	{
548	size_t rss = sizeof (coro);	589	size_t rss = sizeof (coro);
549		590
550	if (coro->mainstack)	591	if (coro->mainstack)
551	{	592	{
		593	if (coro->flags & CF_RUNNING)
		594	{
		595	#define VAR(name,type)coro->name = PL_ ## name;
		596	# include "state.h"
		597	#undef VAR
		598	}
		599
552	rss += sizeof (coro->curstackinfo);	600	rss += sizeof (coro->curstackinfo);
553	rss += sizeof (struct xpvav) + (1 + AvFILL (coro->curstackinfo->si_stack)) * sizeof (SV *);	601	rss += sizeof (struct xpvav) + (1 + AvFILL (coro->curstackinfo->si_stack)) * sizeof (SV *);
554	rss += (coro->curstackinfo->si_cxmax + 1) * sizeof (PERL_CONTEXT);	602	rss += (coro->curstackinfo->si_cxmax + 1) * sizeof (PERL_CONTEXT);
555
556	rss += sizeof (struct xpvav) + (1 + AvFILL (coro->curstack)) * sizeof (SV *);	603	rss += sizeof (struct xpvav) + (1 + AvFILL (coro->curstack)) * sizeof (SV *);
557
558	rss += coro->tmps_max * sizeof (SV *);	604	rss += coro->tmps_max * sizeof (SV *);
559
560	rss += (coro->markstack_max - coro->markstack_ptr) * sizeof (I32);	605	rss += (coro->markstack_max - coro->markstack_ptr) * sizeof (I32);
561
562	rss += coro->scopestack_max * sizeof (I32);	606	rss += coro->scopestack_max * sizeof (I32);
563
564	rss += coro->savestack_max * sizeof (ANY);	607	rss += coro->savestack_max * sizeof (ANY);
565		608
566	#if !PERL_VERSION_ATLEAST (5,9,0)	609	#if !PERL_VERSION_ATLEAST (5,9,0)
567	rss += coro->retstack_max * sizeof (OP *);	610	rss += coro->retstack_max * sizeof (OP *);
568	#endif	611	#endif
…		…
579	/*	622	/*
580	* emulate part of the perl startup here.	623	* emulate part of the perl startup here.
581	*/	624	*/
582	coro_init_stacks (aTHX);	625	coro_init_stacks (aTHX);
583		626
		627	PL_runops = RUNOPS_DEFAULT;
584	PL_curcop = &PL_compiling;	628	PL_curcop = &PL_compiling;
585	PL_in_eval = EVAL_NULL;	629	PL_in_eval = EVAL_NULL;
586	PL_comppad = 0;	630	PL_comppad = 0;
587	PL_curpm = 0;	631	PL_curpm = 0;
588	PL_localizing = 0;	632	PL_localizing = 0;
…		…
619	SvREFCNT_dec (coro_mortal);	663	SvREFCNT_dec (coro_mortal);
620	coro_mortal = 0;	664	coro_mortal = 0;
621	}	665	}
622	}	666	}
623		667
		668	static int
		669	runops_trace (pTHX)
		670	{
		671	COP *oldcop = 0;
		672	int oldcxix = -2;
		673	struct coro coro = SvSTATE (coro_current); / trace cctx is tied to specific coro */
		674	coro_cctx *cctx = coro->cctx;
		675
		676	while ((PL_op = CALL_FPTR (PL_op->op_ppaddr) (aTHX)))
		677	{
		678	PERL_ASYNC_CHECK ();
		679
		680	if (cctx->flags & CC_TRACE_ALL)
		681	{
		682	if (PL_op->op_type == OP_LEAVESUB && cctx->flags & CC_TRACE_SUB)
		683	{
		684	PERL_CONTEXT *cx = &cxstack[cxstack_ix];
		685	SV bot, top;
		686	AV av = newAV (); / return values */
		687	SV **cb;
		688	dSP;
		689
		690	GV *gv = CvGV (cx->blk_sub.cv);
		691	SV *fullname = sv_2mortal (newSV (0));
		692	if (isGV (gv))
		693	gv_efullname3 (fullname, gv, 0);
		694
		695	bot = PL_stack_base + cx->blk_oldsp + 1;
		696	top = cx->blk_gimme == G_ARRAY ? SP + 1
		697	: cx->blk_gimme == G_SCALAR ? bot + 1
		698	: bot;
		699
		700	while (bot < top)
		701	av_push (av, SvREFCNT_inc (*bot++));
		702
		703	PL_runops = RUNOPS_DEFAULT;
		704	ENTER;
		705	SAVETMPS;
		706	EXTEND (SP, 3);
		707	PUSHMARK (SP);
		708	PUSHs (&PL_sv_no);
		709	PUSHs (fullname);
		710	PUSHs (sv_2mortal (newRV_noinc ((SV *)av)));
		711	PUTBACK;
		712	cb = hv_fetch ((HV *)SvRV (coro_current), "_trace_sub_cb", sizeof ("_trace_sub_cb") - 1, 0);
		713	if (cb) call_sv (*cb, G_KEEPERR \| G_EVAL \| G_VOID \| G_DISCARD);
		714	SPAGAIN;
		715	FREETMPS;
		716	LEAVE;
		717	PL_runops = runops_trace;
		718	}
		719
		720	if (oldcop != PL_curcop)
		721	{
		722	oldcop = PL_curcop;
		723
		724	if (PL_curcop != &PL_compiling)
		725	{
		726	SV **cb;
		727
		728	if (oldcxix != cxstack_ix && cctx->flags & CC_TRACE_SUB)
		729	{
		730	PERL_CONTEXT *cx = &cxstack[cxstack_ix];
		731
		732	if (CxTYPE (cx) == CXt_SUB && oldcxix < cxstack_ix)
		733	{
		734	runops_proc_t old_runops = PL_runops;
		735	dSP;
		736	GV *gv = CvGV (cx->blk_sub.cv);
		737	SV *fullname = sv_2mortal (newSV (0));
		738
		739	if (isGV (gv))
		740	gv_efullname3 (fullname, gv, 0);
		741
		742	PL_runops = RUNOPS_DEFAULT;
		743	ENTER;
		744	SAVETMPS;
		745	EXTEND (SP, 3);
		746	PUSHMARK (SP);
		747	PUSHs (&PL_sv_yes);
		748	PUSHs (fullname);
		749	PUSHs (cx->blk_sub.hasargs ? sv_2mortal (newRV_inc ((SV *)cx->blk_sub.argarray)) : &PL_sv_undef);
		750	PUTBACK;
		751	cb = hv_fetch ((HV *)SvRV (coro_current), "_trace_sub_cb", sizeof ("_trace_sub_cb") - 1, 0);
		752	if (cb) call_sv (*cb, G_KEEPERR \| G_EVAL \| G_VOID \| G_DISCARD);
		753	SPAGAIN;
		754	FREETMPS;
		755	LEAVE;
		756	PL_runops = runops_trace;
		757	}
		758
		759	oldcxix = cxstack_ix;
		760	}
		761
		762	if (cctx->flags & CC_TRACE_LINE)
		763	{
		764	dSP;
		765
		766	PL_runops = RUNOPS_DEFAULT;
		767	ENTER;
		768	SAVETMPS;
		769	EXTEND (SP, 3);
		770	PL_runops = RUNOPS_DEFAULT;
		771	PUSHMARK (SP);
		772	PUSHs (sv_2mortal (newSVpv (OutCopFILE (oldcop), 0)));
		773	PUSHs (sv_2mortal (newSViv (CopLINE (oldcop))));
		774	PUTBACK;
		775	cb = hv_fetch ((HV *)SvRV (coro_current), "_trace_line_cb", sizeof ("_trace_line_cb") - 1, 0);
		776	if (cb) call_sv (*cb, G_KEEPERR \| G_EVAL \| G_VOID \| G_DISCARD);
		777	SPAGAIN;
		778	FREETMPS;
		779	LEAVE;
		780	PL_runops = runops_trace;
		781	}
		782	}
		783	}
		784	}
		785	}
		786
		787	TAINT_NOT;
		788	return 0;
		789	}
		790
624	/* inject a fake call to Coro::State::_cctx_init into the execution */	791	/* inject a fake call to Coro::State::_cctx_init into the execution */
625	/* _cctx_init should be careful, as it could be called at almost any time */	792	/* _cctx_init should be careful, as it could be called at almost any time */
626	/* during execution of a perl program */	793	/* during execution of a perl program */
627	static void NOINLINE	794	static void NOINLINE
628	prepare_cctx (pTHX_ coro_cctx *cctx)	795	prepare_cctx (pTHX_ coro_cctx *cctx)
629	{	796	{
630	dSP;	797	dSP;
631	LOGOP myop;	798	LOGOP myop;
		799
		800	PL_top_env = &PL_start_env;
		801
		802	if (cctx->flags & CC_TRACE)
		803	PL_runops = runops_trace;
632		804
633	Zero (&myop, 1, LOGOP);	805	Zero (&myop, 1, LOGOP);
634	myop.op_next = PL_op;	806	myop.op_next = PL_op;
635	myop.op_flags = OPf_WANT_VOID \| OPf_STACKED;	807	myop.op_flags = OPf_WANT_VOID \| OPf_STACKED;
636		808
…		…
653	dTHX;	825	dTHX;
654		826
655	/* coro_run is the alternative tail of transfer(), so unlock here. */	827	/* coro_run is the alternative tail of transfer(), so unlock here. */
656	UNLOCK;	828	UNLOCK;
657		829
658	PL_top_env = &PL_start_env;	830	/* we now skip the entersub that lead to transfer() */
		831	PL_op = PL_op->op_next;
659		832
660	/* inject a fake subroutine call to cctx_init */	833	/* inject a fake subroutine call to cctx_init */
661	prepare_cctx (aTHX_ (coro_cctx *)arg);	834	prepare_cctx (aTHX_ (coro_cctx *)arg);
662		835
663	/* somebody or something will hit me for both perl_run and PL_restartop */	836	/* somebody or something will hit me for both perl_run and PL_restartop */
…		…
697	# if CORO_STACKGUARD	870	# if CORO_STACKGUARD
698	mprotect (cctx->sptr, CORO_STACKGUARD * PAGESIZE, PROT_NONE);	871	mprotect (cctx->sptr, CORO_STACKGUARD * PAGESIZE, PROT_NONE);
699	# endif	872	# endif
700	stack_start = CORO_STACKGUARD * PAGESIZE + (char *)cctx->sptr;	873	stack_start = CORO_STACKGUARD * PAGESIZE + (char *)cctx->sptr;
701	stack_size = cctx->ssize - CORO_STACKGUARD * PAGESIZE;	874	stack_size = cctx->ssize - CORO_STACKGUARD * PAGESIZE;
702	cctx->mapped = 1;	875	cctx->flags \|= CC_MAPPED;
703	}	876	}
704	else	877	else
705	#endif	878	#endif
706	{	879	{
707	cctx->ssize = coro_stacksize * (long)sizeof (long);	880	cctx->ssize = coro_stacksize * (long)sizeof (long);
…		…
734	#if CORO_USE_VALGRIND	907	#if CORO_USE_VALGRIND
735	VALGRIND_STACK_DEREGISTER (cctx->valgrind_id);	908	VALGRIND_STACK_DEREGISTER (cctx->valgrind_id);
736	#endif	909	#endif
737		910
738	#if HAVE_MMAP	911	#if HAVE_MMAP
739	if (cctx->mapped)	912	if (cctx->flags & CC_MAPPED)
740	munmap (cctx->sptr, cctx->ssize);	913	munmap (cctx->sptr, cctx->ssize);
741	else	914	else
742	#endif	915	#endif
743	Safefree (cctx->sptr);	916	Safefree (cctx->sptr);
744		917
…		…
752	{	925	{
753	coro_cctx *cctx = cctx_first;	926	coro_cctx *cctx = cctx_first;
754	cctx_first = cctx->next;	927	cctx_first = cctx->next;
755	--cctx_idle;	928	--cctx_idle;
756		929
757	if (cctx->ssize >= coro_stacksize)	930	if (cctx->ssize >= coro_stacksize && !(cctx->flags & CC_NOREUSE))
758	return cctx;	931	return cctx;
759		932
760	cctx_destroy (cctx);	933	cctx_destroy (cctx);
761	}	934	}
762		935
763	PL_op = PL_op->op_next;
764	return cctx_new ();	936	return cctx_new ();
765	}	937	}
766		938
767	static void	939	static void
768	cctx_put (coro_cctx *cctx)	940	cctx_put (coro_cctx *cctx)
…		…
772	{	944	{
773	coro_cctx *first = cctx_first;	945	coro_cctx *first = cctx_first;
774	cctx_first = first->next;	946	cctx_first = first->next;
775	--cctx_idle;	947	--cctx_idle;
776		948
777	assert (!first->inuse);
778	cctx_destroy (first);	949	cctx_destroy (first);
779	}	950	}
780		951
781	++cctx_idle;	952	++cctx_idle;
782	cctx->next = cctx_first;	953	cctx->next = cctx_first;
783	cctx_first = cctx;	954	cctx_first = cctx;
784	}	955	}
785		956
786	/ coroutine switching ***************************************************/	957	/ coroutine switching ***************************************************/
787		958
788	/* never call directly, always through the coro_state_transfer global variable */	959	static void NOINLINE
		960	transfer_check (pTHX_ struct coro prev, struct coro next)
		961	{
		962	if (prev != next)
		963	{
		964	if (!(prev->flags & (CF_RUNNING \| CF_NEW)))
		965	croak ("Coro::State::transfer called with non-running/new prev Coro::State, but can only transfer from running or new states");
		966
		967	if (next->flags & CF_RUNNING)
		968	croak ("Coro::State::transfer called with running next Coro::State, but can only transfer to inactive states");
		969
		970	if (next->flags & CF_DESTROYED)
		971	croak ("Coro::State::transfer called with destroyed next Coro::State, but can only transfer to inactive states");
		972
		973	if (PL_lex_state != LEX_NOTPARSING)
		974	croak ("Coro::State::transfer called while parsing, but this is not supported");
		975	}
		976	}
		977
		978	/* always use the TRANSFER macro */
789	static void NOINLINE	979	static void NOINLINE
790	transfer (pTHX_ struct coro prev, struct coro next)	980	transfer (pTHX_ struct coro prev, struct coro next)
791	{	981	{
792	dSTACKLEVEL;	982	dSTACKLEVEL;
793		983
…		…
803		993
804	if (prev->flags & CF_NEW)	994	if (prev->flags & CF_NEW)
805	{	995	{
806	/* create a new empty context */	996	/* create a new empty context */
807	Newz (0, prev->cctx, 1, coro_cctx);	997	Newz (0, prev->cctx, 1, coro_cctx);
808	prev->cctx->inuse = 1;
809	prev->flags &= ~CF_NEW;	998	prev->flags &= ~CF_NEW;
810	prev->flags \|= CF_RUNNING;	999	prev->flags \|= CF_RUNNING;
811	}	1000	}
812
813	/TODO: must not croak here /
814	if (!prev->flags & CF_RUNNING)
815	croak ("Coro::State::transfer called with non-running prev Coro::State, but can only transfer from running states");
816
817	if (next->flags & CF_RUNNING)
818	croak ("Coro::State::transfer called with running next Coro::State, but can only transfer to inactive states");
819
820	if (next->flags & CF_DESTROYED)
821	croak ("Coro::State::transfer called with destroyed next Coro::State, but can only transfer to inactive states");
822		1001
823	prev->flags &= ~CF_RUNNING;	1002	prev->flags &= ~CF_RUNNING;
824	next->flags \|= CF_RUNNING;	1003	next->flags \|= CF_RUNNING;
825		1004
826	LOCK;	1005	LOCK;
…		…
831	next->flags &= ~CF_NEW;	1010	next->flags &= ~CF_NEW;
832	/* first get rid of the old state */	1011	/* first get rid of the old state */
833	save_perl (aTHX_ prev);	1012	save_perl (aTHX_ prev);
834	/* setup coroutine call */	1013	/* setup coroutine call */
835	setup_coro (aTHX_ next);	1014	setup_coro (aTHX_ next);
836	/* need a new stack */
837	assert (!next->cctx);
838	}	1015	}
839	else	1016	else
840	{	1017	{
841	/* coroutine already started */	1018	/* coroutine already started */
842	save_perl (aTHX_ prev);	1019	save_perl (aTHX_ prev);
…		…
844	}	1021	}
845		1022
846	prev__cctx = prev->cctx;	1023	prev__cctx = prev->cctx;
847		1024
848	/* possibly "free" the cctx */	1025	/* possibly "free" the cctx */
849	if (prev__cctx->idle_sp == STACKLEVEL)	1026	if (prev__cctx->idle_sp == STACKLEVEL && !(prev__cctx->flags & CC_TRACE))
850	{	1027	{
851	/* I assume that STACKLEVEL is a stronger indicator than PL_top_env changes */	1028	/* I assume that STACKLEVEL is a stronger indicator than PL_top_env changes */
852	assert (("ERROR: current top_env must equal previous top_env", PL_top_env == prev__cctx->idle_te));	1029	assert (("ERROR: current top_env must equal previous top_env", PL_top_env == prev__cctx->idle_te));
853		1030
854	prev->cctx = 0;	1031	prev->cctx = 0;
855		1032
856	cctx_put (prev__cctx);	1033	cctx_put (prev__cctx);
857	prev__cctx->inuse = 0;
858	}	1034	}
859		1035
		1036	++next->usecount;
		1037
860	if (!next->cctx)	1038	if (!next->cctx)
861	{
862	next->cctx = cctx_get (aTHX);	1039	next->cctx = cctx_get (aTHX);
863	assert (!next->cctx->inuse);
864	next->cctx->inuse = 1;
865	}
866		1040
867	if (prev__cctx != next->cctx)	1041	if (prev__cctx != next->cctx)
868	{	1042	{
869	prev__cctx->top_env = PL_top_env;	1043	prev__cctx->top_env = PL_top_env;
870	PL_top_env = next->cctx->top_env;	1044	PL_top_env = next->cctx->top_env;
…		…
880	{	1054	{
881	struct coro prev, next;	1055	struct coro prev, next;
882	};	1056	};
883		1057
884	#define TRANSFER(ta) transfer (aTHX_ (ta).prev, (ta).next)	1058	#define TRANSFER(ta) transfer (aTHX_ (ta).prev, (ta).next)
		1059	#define TRANSFER_CHECK(ta) transfer_check (aTHX_ (ta).prev, (ta).next)
885		1060
886	/ high level stuff ******************************************************/	1061	/ high level stuff ******************************************************/
887		1062
888	static int	1063	static int
889	coro_state_destroy (pTHX_ struct coro *coro)	1064	coro_state_destroy (pTHX_ struct coro *coro)
…		…
929	cctx_destroy (coro->cctx);	1104	cctx_destroy (coro->cctx);
930	SvREFCNT_dec (coro->args);	1105	SvREFCNT_dec (coro->args);
931		1106
932	if (coro->next) coro->next->prev = coro->prev;	1107	if (coro->next) coro->next->prev = coro->prev;
933	if (coro->prev) coro->prev->next = coro->next;	1108	if (coro->prev) coro->prev->next = coro->next;
934	if (coro == first) first = coro->next;	1109	if (coro == coro_first) coro_first = coro->next;
935		1110
936	return 1;	1111	return 1;
937	}	1112	}
938		1113
939	static int	1114	static int
…		…
972	#else	1147	#else
973	# define MGf_DUP 0	1148	# define MGf_DUP 0
974	#endif	1149	#endif
975	};	1150	};
976		1151
977	static struct coro *
978	SvSTATE_ (pTHX_ SV *coro)
979	{
980	HV *stash;
981	MAGIC *mg;
982
983	if (SvROK (coro))
984	coro = SvRV (coro);
985
986	stash = SvSTASH (coro);
987	if (stash != coro_stash && stash != coro_state_stash)
988	{
989	/* very slow, but rare, check */
990	if (!sv_derived_from (sv_2mortal (newRV_inc (coro)), "Coro::State"))
991	croak ("Coro::State object required");
992	}
993
994	mg = CORO_MAGIC (coro);
995	return (struct coro *)mg->mg_ptr;
996	}
997
998	#define SvSTATE(sv) SvSTATE_ (aTHX_ (sv))
999
1000	static void	1152	static void
1001	prepare_transfer (pTHX_ struct transfer_args ta, SV prev_sv, SV *next_sv)	1153	prepare_transfer (pTHX_ struct transfer_args ta, SV prev_sv, SV *next_sv)
1002	{	1154	{
1003	ta->prev = SvSTATE (prev_sv);	1155	ta->prev = SvSTATE (prev_sv);
1004	ta->next = SvSTATE (next_sv);	1156	ta->next = SvSTATE (next_sv);
		1157	TRANSFER_CHECK (*ta);
1005	}	1158	}
1006		1159
1007	static void	1160	static void
1008	api_transfer (SV prev_sv, SV next_sv)	1161	api_transfer (SV prev_sv, SV next_sv)
1009	{	1162	{
…		…
1126	break;	1279	break;
1127	}	1280	}
1128		1281
1129	/* free this only after the transfer */	1282	/* free this only after the transfer */
1130	prev_sv = SvRV (coro_current);	1283	prev_sv = SvRV (coro_current);
1131	SvRV_set (coro_current, next_sv);
1132	ta->prev = SvSTATE (prev_sv);	1284	ta->prev = SvSTATE (prev_sv);
1133		1285	TRANSFER_CHECK (*ta);
1134	assert (ta->next->flags & CF_READY);	1286	assert (ta->next->flags & CF_READY);
1135	ta->next->flags &= ~CF_READY;	1287	ta->next->flags &= ~CF_READY;
		1288	SvRV_set (coro_current, next_sv);
1136		1289
1137	LOCK;	1290	LOCK;
1138	free_coro_mortal (aTHX);	1291	free_coro_mortal (aTHX);
1139	coro_mortal = prev_sv;	1292	coro_mortal = prev_sv;
1140	UNLOCK;	1293	UNLOCK;
…		…
1201	}	1354	}
1202	else	1355	else
1203	return 0;	1356	return 0;
1204	}	1357	}
1205		1358
		1359	static void
		1360	api_trace (SV *coro_sv, int flags)
		1361	{
		1362	dTHX;
		1363	struct coro *coro = SvSTATE (coro_sv);
		1364
		1365	if (flags & CC_TRACE)
		1366	{
		1367	if (!coro->cctx)
		1368	coro->cctx = cctx_new ();
		1369	else if (!(coro->cctx->flags & CC_TRACE))
		1370	croak ("cannot enable tracing on coroutine with custom stack");
		1371
		1372	coro->cctx->flags \|= CC_NOREUSE \| (flags & (CC_TRACE \| CC_TRACE_ALL));
		1373	}
		1374	else if (coro->cctx && coro->cctx->flags & CC_TRACE)
		1375	{
		1376	coro->cctx->flags &= ~(CC_TRACE \| CC_TRACE_ALL);
		1377
		1378	if (coro->flags & CF_RUNNING)
		1379	PL_runops = RUNOPS_DEFAULT;
		1380	else
		1381	coro->runops = RUNOPS_DEFAULT;
		1382	}
		1383	}
		1384
1206	MODULE = Coro::State PACKAGE = Coro::State	1385	MODULE = Coro::State PACKAGE = Coro::State PREFIX = api_
1207		1386
1208	PROTOTYPES: DISABLE	1387	PROTOTYPES: DISABLE
1209		1388
1210	BOOT:	1389	BOOT:
1211	{	1390	{
…		…
1213	MUTEX_INIT (&coro_mutex);	1392	MUTEX_INIT (&coro_mutex);
1214	#endif	1393	#endif
1215	BOOT_PAGESIZE;	1394	BOOT_PAGESIZE;
1216		1395
1217	coro_state_stash = gv_stashpv ("Coro::State", TRUE);	1396	coro_state_stash = gv_stashpv ("Coro::State", TRUE);
		1397
		1398	newCONSTSUB (coro_state_stash, "CC_TRACE" , newSViv (CC_TRACE));
		1399	newCONSTSUB (coro_state_stash, "CC_TRACE_SUB" , newSViv (CC_TRACE_SUB));
		1400	newCONSTSUB (coro_state_stash, "CC_TRACE_LINE", newSViv (CC_TRACE_LINE));
		1401	newCONSTSUB (coro_state_stash, "CC_TRACE_ALL" , newSViv (CC_TRACE_ALL));
1218		1402
1219	newCONSTSUB (coro_state_stash, "SAVE_DEFAV", newSViv (CORO_SAVE_DEFAV));	1403	newCONSTSUB (coro_state_stash, "SAVE_DEFAV", newSViv (CORO_SAVE_DEFAV));
1220	newCONSTSUB (coro_state_stash, "SAVE_DEFSV", newSViv (CORO_SAVE_DEFSV));	1404	newCONSTSUB (coro_state_stash, "SAVE_DEFSV", newSViv (CORO_SAVE_DEFSV));
1221	newCONSTSUB (coro_state_stash, "SAVE_ERRSV", newSViv (CORO_SAVE_ERRSV));	1405	newCONSTSUB (coro_state_stash, "SAVE_ERRSV", newSViv (CORO_SAVE_ERRSV));
1222	newCONSTSUB (coro_state_stash, "SAVE_IRSSV", newSViv (CORO_SAVE_IRSSV));	1406	newCONSTSUB (coro_state_stash, "SAVE_IRSSV", newSViv (CORO_SAVE_IRSSV));
…		…
1247	Newz (0, coro, 1, struct coro);	1431	Newz (0, coro, 1, struct coro);
1248	coro->args = newAV ();	1432	coro->args = newAV ();
1249	coro->save = CORO_SAVE_DEF;	1433	coro->save = CORO_SAVE_DEF;
1250	coro->flags = CF_NEW;	1434	coro->flags = CF_NEW;
1251		1435
1252	if (first) first->prev = coro;	1436	if (coro_first) coro_first->prev = coro;
1253	coro->next = first;	1437	coro->next = coro_first;
1254	first = coro;	1438	coro_first = coro;
1255		1439
1256	coro->hv = hv = newHV ();	1440	coro->hv = hv = newHV ();
1257	sv_magicext ((SV )hv, 0, PERL_MAGIC_ext, &coro_state_vtbl, (char )coro, 0)->mg_flags \|= MGf_DUP;	1441	sv_magicext ((SV )hv, 0, PERL_MAGIC_ext, &coro_state_vtbl, (char )coro, 0)->mg_flags \|= MGf_DUP;
1258	RETVAL = sv_bless (newRV_noinc ((SV *)hv), gv_stashpv (klass, 1));	1442	RETVAL = sv_bless (newRV_noinc ((SV *)hv), gv_stashpv (klass, 1));
1259		1443
…		…
1368	void	1552	void
1369	list ()	1553	list ()
1370	PPCODE:	1554	PPCODE:
1371	{	1555	{
1372	struct coro *coro;	1556	struct coro *coro;
1373	for (coro = first; coro; coro = coro->next)	1557	for (coro = coro_first; coro; coro = coro->next)
1374	if (coro->hv)	1558	if (coro->hv)
1375	XPUSHs (sv_2mortal (newRV_inc ((SV *)coro->hv)));	1559	XPUSHs (sv_2mortal (newRV_inc ((SV *)coro->hv)));
1376	}	1560	}
1377		1561
1378	void	1562	void
1379	_eval (Coro::State coro, SV *coderef)	1563	call (Coro::State coro, SV *coderef)
		1564	ALIAS:
		1565	eval = 1
1380	CODE:	1566	CODE:
1381	{	1567	{
1382	if (coro->mainstack)	1568	if (coro->mainstack)
1383	{	1569	{
1384	struct coro temp;	1570	struct coro temp;
…		…
1395	dSP;	1581	dSP;
1396	ENTER;	1582	ENTER;
1397	SAVETMPS;	1583	SAVETMPS;
1398	PUSHMARK (SP);	1584	PUSHMARK (SP);
1399	PUTBACK;	1585	PUTBACK;
		1586	if (ix)
		1587	eval_sv (coderef, 0);
		1588	else
1400	call_sv (coderef, G_KEEPERR \| G_EVAL \| G_VOID \| G_DISCARD);	1589	call_sv (coderef, G_KEEPERR \| G_EVAL \| G_VOID \| G_DISCARD);
1401	SPAGAIN;	1590	SPAGAIN;
1402	FREETMPS;	1591	FREETMPS;
1403	LEAVE;	1592	LEAVE;
1404	PUTBACK;	1593	PUTBACK;
1405	}	1594	}
…		…
1409	save_perl (aTHX_ coro);	1598	save_perl (aTHX_ coro);
1410	load_perl (aTHX_ &temp);	1599	load_perl (aTHX_ &temp);
1411	}	1600	}
1412	}	1601	}
1413	}	1602	}
1414		1603
1415	SV *	1604	SV *
1416	is_ready (Coro::State coro)	1605	is_ready (Coro::State coro)
1417	PROTOTYPE: $	1606	PROTOTYPE: $
1418	ALIAS:	1607	ALIAS:
1419	is_ready = CF_READY	1608	is_ready = CF_READY
…		…
1423	CODE:	1612	CODE:
1424	RETVAL = boolSV (coro->flags & ix);	1613	RETVAL = boolSV (coro->flags & ix);
1425	OUTPUT:	1614	OUTPUT:
1426	RETVAL	1615	RETVAL
1427		1616
		1617	void
		1618	api_trace (SV *coro, int flags = CC_TRACE \| CC_TRACE_SUB)
		1619
		1620	SV *
		1621	has_stack (Coro::State coro)
		1622	PROTOTYPE: $
		1623	CODE:
		1624	RETVAL = boolSV (!!coro->cctx);
		1625	OUTPUT:
		1626	RETVAL
		1627
		1628	int
		1629	is_traced (Coro::State coro)
		1630	PROTOTYPE: $
		1631	CODE:
		1632	RETVAL = (coro->cctx ? coro->cctx->flags : 0) & CC_TRACE_ALL;
		1633	OUTPUT:
		1634	RETVAL
		1635
1428	IV	1636	IV
1429	rss (Coro::State coro)	1637	rss (Coro::State coro)
1430	PROTOTYPE: $	1638	PROTOTYPE: $
		1639	ALIAS:
		1640	usecount = 1
1431	CODE:	1641	CODE:
1432	RETVAL = coro_rss (coro);	1642	switch (ix)
		1643	{
		1644	case 0: RETVAL = coro_rss (aTHX_ coro); break;
		1645	case 1: RETVAL = coro->usecount; break;
		1646	}
1433	OUTPUT:	1647	OUTPUT:
1434	RETVAL	1648	RETVAL
1435		1649
1436		1650
1437	MODULE = Coro::State PACKAGE = Coro	1651	MODULE = Coro::State PACKAGE = Coro
1438		1652
1439	BOOT:	1653	BOOT:
1440	{	1654	{
1441	int i;	1655	int i;
		1656
		1657	sv_pool_rss = get_sv ("Coro::POOL_RSS" , TRUE);
		1658	sv_pool_size = get_sv ("Coro::POOL_SIZE" , TRUE);
		1659	av_async_pool = get_av ("Coro::async_pool", TRUE);
		1660
		1661	coro_current = get_sv ("Coro::current", FALSE);
		1662	SvREADONLY_on (coro_current);
1442		1663
1443	coro_stash = gv_stashpv ("Coro", TRUE);	1664	coro_stash = gv_stashpv ("Coro", TRUE);
1444		1665
1445	newCONSTSUB (coro_stash, "PRIO_MAX", newSViv (PRIO_MAX));	1666	newCONSTSUB (coro_stash, "PRIO_MAX", newSViv (PRIO_MAX));
1446	newCONSTSUB (coro_stash, "PRIO_HIGH", newSViv (PRIO_HIGH));	1667	newCONSTSUB (coro_stash, "PRIO_HIGH", newSViv (PRIO_HIGH));
1447	newCONSTSUB (coro_stash, "PRIO_NORMAL", newSViv (PRIO_NORMAL));	1668	newCONSTSUB (coro_stash, "PRIO_NORMAL", newSViv (PRIO_NORMAL));
1448	newCONSTSUB (coro_stash, "PRIO_LOW", newSViv (PRIO_LOW));	1669	newCONSTSUB (coro_stash, "PRIO_LOW", newSViv (PRIO_LOW));
1449	newCONSTSUB (coro_stash, "PRIO_IDLE", newSViv (PRIO_IDLE));	1670	newCONSTSUB (coro_stash, "PRIO_IDLE", newSViv (PRIO_IDLE));
1450	newCONSTSUB (coro_stash, "PRIO_MIN", newSViv (PRIO_MIN));	1671	newCONSTSUB (coro_stash, "PRIO_MIN", newSViv (PRIO_MIN));
1451
1452	coro_current = get_sv ("Coro::current", FALSE);
1453	SvREADONLY_on (coro_current);
1454		1672
1455	for (i = PRIO_MAX - PRIO_MIN + 1; i--; )	1673	for (i = PRIO_MAX - PRIO_MIN + 1; i--; )
1456	coro_ready[i] = newAV ();	1674	coro_ready[i] = newAV ();
1457		1675
1458	{	1676	{
…		…
1516	CODE:	1734	CODE:
1517	RETVAL = coro_nready;	1735	RETVAL = coro_nready;
1518	OUTPUT:	1736	OUTPUT:
1519	RETVAL	1737	RETVAL
1520		1738
		1739	# for async_pool speedup
		1740	void
		1741	_pool_1 (SV *cb)
		1742	CODE:
		1743	{
		1744	int i, len;
		1745	HV hv = (HV )SvRV (coro_current);
		1746	AV *defav = GvAV (PL_defgv);
		1747	SV *invoke = hv_delete (hv, "_invoke", sizeof ("_invoke") - 1, 0);
		1748	AV *invoke_av;
		1749
		1750	if (!invoke)
		1751	croak ("\3terminate\2\n");
		1752
		1753	hv_store (hv, "desc", sizeof ("desc") - 1,
		1754	newSVpvn ("[async_pool]", sizeof ("[async_pool]") - 1), 0);
		1755
		1756	invoke_av = (AV *)SvRV (invoke);
		1757	len = av_len (invoke_av);
		1758
		1759	sv_setsv (cb, AvARRAY (invoke_av)[0]);
		1760
		1761	if (len > 0)
		1762	{
		1763	av_fill (defav, len - 1);
		1764	for (i = 0; i < len; ++i)
		1765	av_store (defav, i, SvREFCNT_inc (AvARRAY (invoke_av)[i + 1]));
		1766	}
		1767
		1768	SvREFCNT_dec (invoke);
		1769	}
		1770
		1771	void
		1772	_pool_2 (SV *cb)
		1773	CODE:
		1774	{
		1775	struct coro *coro = SvSTATE (coro_current);
		1776
		1777	sv_setsv (cb, &PL_sv_undef);
		1778
		1779	if (coro_rss (aTHX_ coro) > SvIV (sv_pool_rss)
		1780	\|\| av_len (av_async_pool) + 1 >= SvIV (sv_pool_size))
		1781	croak ("\3terminate\2\n");
		1782
		1783	av_clear (GvAV (PL_defgv));
		1784	hv_store ((HV *)SvRV (coro_current), "desc", sizeof ("desc") - 1,
		1785	newSVpvn ("[async_pool idle]", sizeof ("[async_pool idle]") - 1), 0);
		1786
		1787	coro->save = CORO_SAVE_DEF;
		1788	coro->prio = 0;
		1789
		1790	if (coro->cctx && (coro->cctx->flags & CC_TRACE))
		1791	api_trace (coro_current, 0);
		1792
		1793	av_push (av_async_pool, newSVsv (coro_current));
		1794	}
		1795
		1796
1521	MODULE = Coro::State PACKAGE = Coro::AIO	1797	MODULE = Coro::State PACKAGE = Coro::AIO
1522		1798
1523	SV *	1799	SV *
1524	_get_state ()	1800	_get_state ()
1525	CODE:	1801	CODE:

Diff Legend

-–
+Removed lines
-+
+Added lines
-<
+Changed lines
->
+Changed lines

Comparing Coro/Coro/State.xs (file contents): Revision 1.152 by root, Thu Sep 20 12:02:25 2007 UTC vs. Revision 1.175 by root, Sun Sep 30 13:43:45 2007 UTC

Diff Legend

Comparing Coro/Coro/State.xs (file contents):
Revision 1.152 by root, Thu Sep 20 12:02:25 2007 UTC vs.
Revision 1.175 by root, Sun Sep 30 13:43:45 2007 UTC