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

Comparing Coro/Coro/State.xs (file contents):
Revision 1.166 by root, Wed Sep 26 21:11:28 2007 UTC vs.
Revision 1.190 by root, Fri Oct 5 20:39:39 2007 UTC

…		…
75	/* 5.8.7 */	75	/* 5.8.7 */
76	#ifndef SvRV_set	76	#ifndef SvRV_set
77	# define SvRV_set(s,v) SvRV(s) = (v)	77	# define SvRV_set(s,v) SvRV(s) = (v)
78	#endif	78	#endif
79		79
		80	/* 5.8.8 */
		81	#ifndef GV_NOTQUAL
		82	# define GV_NOTQUAL 0
		83	#endif
		84
80	#if !__i386 && !__x86_64 && !__powerpc && !__m68k && !__alpha && !__mips && !__sparc64	85	#if !__i386 && !__x86_64 && !__powerpc && !__m68k && !__alpha && !__mips && !__sparc64
81	# undef CORO_STACKGUARD	86	# undef CORO_STACKGUARD
82	#endif	87	#endif
83		88
84	#ifndef CORO_STACKGUARD	89	#ifndef CORO_STACKGUARD
…		…
132	static AV main_mainstack; / used to differentiate between $main and others */	137	static AV main_mainstack; / used to differentiate between $main and others */
133	static JMPENV *main_top_env;	138	static JMPENV *main_top_env;
134	static HV coro_state_stash, coro_stash;	139	static HV coro_state_stash, coro_stash;
135	static SV coro_mortal; / will be freed after next transfer */	140	static SV coro_mortal; / will be freed after next transfer */
136		141
		142	static GV irsgv; / $/ */
		143	static GV stdoutgv; / STDOUT /
		144
137	/* async_pool helper stuff */	145	/* async_pool helper stuff */
138	static SV *sv_pool_rss;	146	static SV *sv_pool_rss;
139	static SV *sv_pool_size;	147	static SV *sv_pool_size;
140	static AV *av_async_pool;	148	static AV *av_async_pool;
141		149
142	static struct coro_cctx *cctx_first;	150	static struct coro_cctx *cctx_first;
143	static int cctx_count, cctx_idle;	151	static int cctx_count, cctx_idle;
144		152
145	enum {	153	enum {
146	CC_MAPPED = 0x01,	154	CC_MAPPED = 0x01,
147	CC_TRACE = 0x02,
148	CC_NOREUSE = 0x04, /* throw this away after tracing */	155	CC_NOREUSE = 0x02, /* throw this away after tracing */
		156	CC_TRACE = 0x04,
		157	CC_TRACE_SUB = 0x08, /* trace sub calls */
		158	CC_TRACE_LINE = 0x10, /* trace each statement */
		159	CC_TRACE_ALL = CC_TRACE_SUB \| CC_TRACE_LINE,
149	};	160	};
150		161
151	/* this is a structure representing a c-level coroutine */	162	/* this is a structure representing a c-level coroutine */
152	typedef struct coro_cctx {	163	typedef struct coro_cctx {
153	struct coro_cctx *next;	164	struct coro_cctx *next;
…		…
181	coro_cctx *cctx;	192	coro_cctx *cctx;
182		193
183	/* data associated with this coroutine (initial args) */	194	/* data associated with this coroutine (initial args) */
184	AV *args;	195	AV *args;
185	int refcnt;	196	int refcnt;
186	int save; /* CORO_SAVE flags */
187	int flags; /* CF_ flags */	197	int flags; /* CF_ flags */
188		198
189	/* optionally saved, might be zero */	199	/* optionally saved, might be zero */
190	AV defav; / @_ */	200	AV defav; / @_ */
191	SV defsv; / $_ */	201	SV defsv; / $_ */
192	SV errsv; / $@ */	202	SV errsv; / $@ */
193	GV deffh; / default filehandle */	203	SV deffh; / default filehandle */
194	SV irssv; / $/ */	204	SV irssv; / $/ */
195	SV irssv_sv; / real $/ cache */	205	SV irssv_sv; / real $/ cache */
196		206
197	#define VAR(name,type) type name;	207	#define VAR(name,type) type name;
198	# include "state.h"	208	# include "state.h"
199	#undef VAR	209	#undef VAR
200		210
		211	/* statistics */
		212	int usecount; /* number of transfers to this coro */
		213
201	/* coro process data */	214	/* coro process data */
202	int prio;	215	int prio;
		216	//SV *throw;
		217
		218	/* async_pool */
		219	SV *saved_deffh;
203		220
204	/* linked list */	221	/* linked list */
205	struct coro next, prev;	222	struct coro next, prev;
206	HV hv; / the perl hash associated with this coro, if any */	223	HV hv; / the perl hash associated with this coro, if any */
207	};	224	};
…		…
295	SvMAGIC (cv) \	312	SvMAGIC (cv) \
296	? SvMAGIC (cv)->mg_type == PERL_MAGIC_coro \	313	? SvMAGIC (cv)->mg_type == PERL_MAGIC_coro \
297	? SvMAGIC (cv) \	314	? SvMAGIC (cv) \
298	: mg_find ((SV *)cv, PERL_MAGIC_coro) \	315	: mg_find ((SV *)cv, PERL_MAGIC_coro) \
299	: 0	316	: 0
		317
		318	static struct coro *
		319	SvSTATE_ (pTHX_ SV *coro)
		320	{
		321	HV *stash;
		322	MAGIC *mg;
		323
		324	if (SvROK (coro))
		325	coro = SvRV (coro);
		326
		327	if (SvTYPE (coro) != SVt_PVHV)
		328	croak ("Coro::State object required");
		329
		330	stash = SvSTASH (coro);
		331	if (stash != coro_stash && stash != coro_state_stash)
		332	{
		333	/* very slow, but rare, check */
		334	if (!sv_derived_from (sv_2mortal (newRV_inc (coro)), "Coro::State"))
		335	croak ("Coro::State object required");
		336	}
		337
		338	mg = CORO_MAGIC (coro);
		339	return (struct coro *)mg->mg_ptr;
		340	}
		341
		342	#define SvSTATE(sv) SvSTATE_ (aTHX_ (sv))
300		343
301	/* the next two functions merely cache the padlists */	344	/* the next two functions merely cache the padlists */
302	static void	345	static void
303	get_padlist (pTHX_ CV *cv)	346	get_padlist (pTHX_ CV *cv)
304	{	347	{
…		…
343	AvARRAY (av)[++AvFILLp (av)] = (SV *)CvPADLIST (cv);	386	AvARRAY (av)[++AvFILLp (av)] = (SV *)CvPADLIST (cv);
344	}	387	}
345		388
346	/ load & save, init *****************************************************/	389	/ load & save, init *****************************************************/
347		390
348	#define SB do {
349	#define SE } while (0)
350
351	#define REPLACE_SV(sv,val) SB SvREFCNT_dec (sv); (sv) = (val); (val) = 0; SE
352
353	static void	391	static void
354	load_perl (pTHX_ Coro__State c)	392	load_perl (pTHX_ Coro__State c)
355	{	393	{
356	#define VAR(name,type) PL_ ## name = c->name;	394	#define VAR(name,type) PL_ ## name = c->name;
357	# include "state.h"	395	# include "state.h"
358	#undef VAR	396	#undef VAR
359		397
360	if (c->defav) REPLACE_SV (GvAV (PL_defgv), c->defav);	398	GvSV (PL_defgv) = c->defsv;
361	if (c->defsv) REPLACE_SV (DEFSV , c->defsv);	399	GvAV (PL_defgv) = c->defav;
362	if (c->errsv) REPLACE_SV (ERRSV , c->errsv);	400	GvSV (PL_errgv) = c->errsv;
363	if (c->deffh) REPLACE_SV (PL_defoutgv , c->deffh);	401	GvSV (irsgv) = c->irssv_sv;
364
365	if (c->irssv)
366	{
367	if (c->irssv == PL_rs \|\| sv_eq (PL_rs, c->irssv))
368	{
369	SvREFCNT_dec (c->irssv);
370	c->irssv = 0;
371	}
372	else
373	{
374	REPLACE_SV (PL_rs, c->irssv);
375	if (!c->irssv_sv) c->irssv_sv = get_sv ("/", 0);
376	sv_setsv (c->irssv_sv, PL_rs);
377	}
378	}
379		402
380	{	403	{
381	dSP;	404	dSP;
382	CV *cv;	405	CV *cv;
383		406
…		…
441	}	464	}
442		465
443	PUTBACK;	466	PUTBACK;
444	}	467	}
445		468
446	c->defav = c->save & CORO_SAVE_DEFAV ? (AV *)SvREFCNT_inc (GvAV (PL_defgv)) : 0;	469	c->defav = GvAV (PL_defgv);
447	c->defsv = c->save & CORO_SAVE_DEFSV ? SvREFCNT_inc (DEFSV) : 0;	470	c->defsv = DEFSV;
448	c->errsv = c->save & CORO_SAVE_ERRSV ? SvREFCNT_inc (ERRSV) : 0;	471	c->errsv = ERRSV;
449	c->deffh = c->save & CORO_SAVE_DEFFH ? (GV *)SvREFCNT_inc (PL_defoutgv) : 0;	472	c->irssv_sv = GvSV (irsgv);
450	c->irssv = c->save & CORO_SAVE_IRSSV ? SvREFCNT_inc (PL_rs) : 0;
451		473
452	#define VAR(name,type)c->name = PL_ ## name;	474	#define VAR(name,type)c->name = PL_ ## name;
453	# include "state.h"	475	# include "state.h"
454	#undef VAR	476	#undef VAR
455	}	477	}
…		…
508	* destroy the stacks, the callchain etc...	530	* destroy the stacks, the callchain etc...
509	*/	531	*/
510	static void	532	static void
511	coro_destroy_stacks (pTHX)	533	coro_destroy_stacks (pTHX)
512	{	534	{
513	if (!IN_DESTRUCT)
514	{
515	/* restore all saved variables and stuff */
516	LEAVE_SCOPE (0);
517	assert (PL_tmps_floor == -1);
518
519	/* free all temporaries */
520	FREETMPS;
521	assert (PL_tmps_ix == -1);
522
523	/* unwind all extra stacks */
524	POPSTACK_TO (PL_mainstack);
525
526	/* unwind main stack */
527	dounwind (-1);
528	}
529
530	while (PL_curstackinfo->si_next)	535	while (PL_curstackinfo->si_next)
531	PL_curstackinfo = PL_curstackinfo->si_next;	536	PL_curstackinfo = PL_curstackinfo->si_next;
532		537
533	while (PL_curstackinfo)	538	while (PL_curstackinfo)
534	{	539	{
…		…
550	Safefree (PL_retstack);	555	Safefree (PL_retstack);
551	#endif	556	#endif
552	}	557	}
553		558
554	static size_t	559	static size_t
555	coro_rss (struct coro *coro)	560	coro_rss (pTHX_ struct coro *coro)
556	{	561	{
557	size_t rss = sizeof (coro);	562	size_t rss = sizeof (*coro);
558		563
559	if (coro->mainstack)	564	if (coro->mainstack)
560	{	565	{
561	if (coro->flags & CF_RUNNING)	566	if (coro->flags & CF_RUNNING)
562	{	567	{
…		…
564	# include "state.h"	569	# include "state.h"
565	#undef VAR	570	#undef VAR
566	}	571	}
567		572
568	rss += sizeof (coro->curstackinfo);	573	rss += sizeof (coro->curstackinfo);
569	rss += sizeof (struct xpvav) + (1 + AvFILL (coro->curstackinfo->si_stack)) * sizeof (SV *);	574	rss += sizeof (SV) + sizeof (struct xpvav) + (1 + AvFILL (coro->curstackinfo->si_stack)) * sizeof (SV *);
570	rss += (coro->curstackinfo->si_cxmax + 1) * sizeof (PERL_CONTEXT);	575	rss += (coro->curstackinfo->si_cxmax + 1) * sizeof (PERL_CONTEXT);
571	rss += sizeof (struct xpvav) + (1 + AvFILL (coro->curstack)) * sizeof (SV *);	576	rss += sizeof (SV) + sizeof (struct xpvav) + (1 + AvFILL (coro->curstack)) * sizeof (SV *);
572	rss += coro->tmps_max * sizeof (SV *);	577	rss += coro->tmps_max * sizeof (SV *);
573	rss += (coro->markstack_max - coro->markstack_ptr) * sizeof (I32);	578	rss += (coro->markstack_max - coro->markstack_ptr) * sizeof (I32);
574	rss += coro->scopestack_max * sizeof (I32);	579	rss += coro->scopestack_max * sizeof (I32);
575	rss += coro->savestack_max * sizeof (ANY);	580	rss += coro->savestack_max * sizeof (ANY);
576		581
…		…
583	}	588	}
584		589
585	/ coroutine stack handling **********************************************/	590	/ coroutine stack handling **********************************************/
586		591
587	static void	592	static void
588	setup_coro (pTHX_ struct coro *coro)	593	coro_setup (pTHX_ struct coro *coro)
589	{	594	{
590	/*	595	/*
591	* emulate part of the perl startup here.	596	* emulate part of the perl startup here.
592	*/	597	*/
593	coro_init_stacks (aTHX);	598	coro_init_stacks (aTHX);
594		599
		600	PL_runops = RUNOPS_DEFAULT;
595	PL_curcop = &PL_compiling;	601	PL_curcop = &PL_compiling;
596	PL_in_eval = EVAL_NULL;	602	PL_in_eval = EVAL_NULL;
597	PL_comppad = 0;	603	PL_comppad = 0;
598	PL_curpm = 0;	604	PL_curpm = 0;
599	PL_localizing = 0;	605	PL_localizing = 0;
600	PL_dirty = 0;	606	PL_dirty = 0;
601	PL_restartop = 0;	607	PL_restartop = 0;
		608
		609	GvSV (PL_defgv) = NEWSV (0, 0);
		610	GvAV (PL_defgv) = coro->args; coro->args = 0;
		611	GvSV (PL_errgv) = NEWSV (0, 0);
		612	GvSV (irsgv) = newSVpvn ("\n", 1); sv_magic (GvSV (irsgv), (SV *)irsgv, PERL_MAGIC_sv, "/", 0);
		613	PL_rs = newSVsv (GvSV (irsgv));
		614	PL_defoutgv = SvREFCNT_inc (stdoutgv);
602		615
603	{	616	{
604	dSP;	617	dSP;
605	LOGOP myop;	618	LOGOP myop;
606
607	SvREFCNT_dec (GvAV (PL_defgv));
608	GvAV (PL_defgv) = coro->args; coro->args = 0;
609		619
610	Zero (&myop, 1, LOGOP);	620	Zero (&myop, 1, LOGOP);
611	myop.op_next = Nullop;	621	myop.op_next = Nullop;
612	myop.op_flags = OPf_WANT_VOID;	622	myop.op_flags = OPf_WANT_VOID;
613		623
…		…
621		631
622	ENTER; /* necessary e.g. for dounwind */	632	ENTER; /* necessary e.g. for dounwind */
623	}	633	}
624		634
625	static void	635	static void
		636	coro_destroy (pTHX_ struct coro *coro)
		637	{
		638	if (!IN_DESTRUCT)
		639	{
		640	/* restore all saved variables and stuff */
		641	LEAVE_SCOPE (0);
		642	assert (PL_tmps_floor == -1);
		643
		644	/* free all temporaries */
		645	FREETMPS;
		646	assert (PL_tmps_ix == -1);
		647
		648	/* unwind all extra stacks */
		649	POPSTACK_TO (PL_mainstack);
		650
		651	/* unwind main stack */
		652	dounwind (-1);
		653	}
		654
		655	SvREFCNT_dec (GvSV (PL_defgv));
		656	SvREFCNT_dec (GvAV (PL_defgv));
		657	SvREFCNT_dec (GvSV (PL_errgv));
		658	SvREFCNT_dec (PL_defoutgv);
		659	SvREFCNT_dec (PL_rs);
		660	SvREFCNT_dec (GvSV (irsgv));
		661
		662	SvREFCNT_dec (coro->saved_deffh);
		663	//SvREFCNT_dec (coro->throw);
		664
		665	coro_destroy_stacks (aTHX);
		666	}
		667
		668	static void
626	free_coro_mortal (pTHX)	669	free_coro_mortal (pTHX)
627	{	670	{
628	if (coro_mortal)	671	if (coro_mortal)
629	{	672	{
630	SvREFCNT_dec (coro_mortal);	673	SvREFCNT_dec (coro_mortal);
631	coro_mortal = 0;	674	coro_mortal = 0;
632	}	675	}
633	}	676	}
634		677
635	static void
636	do_trace (pTHX)
637	{
638	if (PL_curcop != &PL_compiling)
639	{
640	runops_proc_t old_runops = PL_runops;
641	dSP;
642	ENTER;
643	SAVETMPS;
644	PUSHMARK (SP);
645	PUTBACK;
646	PL_runops = RUNOPS_DEFAULT;
647	call_pv ("Coro::_do_trace", G_KEEPERR \| G_EVAL \| G_VOID \| G_DISCARD \| G_NOARGS);
648	PL_runops = old_runops;
649	SPAGAIN;
650	FREETMPS;
651	LEAVE;
652	PUTBACK;
653	}
654	}
655
656	static int	678	static int
657	runops_coro (pTHX)	679	runops_trace (pTHX)
658	{	680	{
659	COP *oldcop = 0;	681	COP *oldcop = 0;
		682	int oldcxix = -2;
		683	struct coro coro = SvSTATE (coro_current); / trace cctx is tied to specific coro */
		684	coro_cctx *cctx = coro->cctx;
660		685
661	while ((PL_op = CALL_FPTR (PL_op->op_ppaddr) (aTHX)))	686	while ((PL_op = CALL_FPTR (PL_op->op_ppaddr) (aTHX)))
662	{	687	{
663	PERL_ASYNC_CHECK ();	688	PERL_ASYNC_CHECK ();
664		689
665	if (oldcop != PL_curcop)	690	if (cctx->flags & CC_TRACE_ALL)
666	{	691	{
		692	if (PL_op->op_type == OP_LEAVESUB && cctx->flags & CC_TRACE_SUB)
		693	{
		694	PERL_CONTEXT *cx = &cxstack[cxstack_ix];
		695	SV bot, top;
		696	AV av = newAV (); / return values */
		697	SV **cb;
		698	dSP;
		699
		700	GV *gv = CvGV (cx->blk_sub.cv);
		701	SV *fullname = sv_2mortal (newSV (0));
		702	if (isGV (gv))
		703	gv_efullname3 (fullname, gv, 0);
		704
		705	bot = PL_stack_base + cx->blk_oldsp + 1;
		706	top = cx->blk_gimme == G_ARRAY ? SP + 1
		707	: cx->blk_gimme == G_SCALAR ? bot + 1
		708	: bot;
		709
		710	while (bot < top)
		711	av_push (av, SvREFCNT_inc (*bot++));
		712
		713	PL_runops = RUNOPS_DEFAULT;
		714	ENTER;
		715	SAVETMPS;
		716	EXTEND (SP, 3);
		717	PUSHMARK (SP);
		718	PUSHs (&PL_sv_no);
		719	PUSHs (fullname);
		720	PUSHs (sv_2mortal (newRV_noinc ((SV *)av)));
		721	PUTBACK;
		722	cb = hv_fetch ((HV *)SvRV (coro_current), "_trace_sub_cb", sizeof ("_trace_sub_cb") - 1, 0);
		723	if (cb) call_sv (*cb, G_KEEPERR \| G_EVAL \| G_VOID \| G_DISCARD);
		724	SPAGAIN;
		725	FREETMPS;
		726	LEAVE;
		727	PL_runops = runops_trace;
		728	}
		729
		730	if (oldcop != PL_curcop)
		731	{
667	oldcop = PL_curcop;	732	oldcop = PL_curcop;
668	do_trace (aTHX);	733
		734	if (PL_curcop != &PL_compiling)
		735	{
		736	SV **cb;
		737
		738	if (oldcxix != cxstack_ix && cctx->flags & CC_TRACE_SUB)
		739	{
		740	PERL_CONTEXT *cx = &cxstack[cxstack_ix];
		741
		742	if (CxTYPE (cx) == CXt_SUB && oldcxix < cxstack_ix)
		743	{
		744	runops_proc_t old_runops = PL_runops;
		745	dSP;
		746	GV *gv = CvGV (cx->blk_sub.cv);
		747	SV *fullname = sv_2mortal (newSV (0));
		748
		749	if (isGV (gv))
		750	gv_efullname3 (fullname, gv, 0);
		751
		752	PL_runops = RUNOPS_DEFAULT;
		753	ENTER;
		754	SAVETMPS;
		755	EXTEND (SP, 3);
		756	PUSHMARK (SP);
		757	PUSHs (&PL_sv_yes);
		758	PUSHs (fullname);
		759	PUSHs (cx->blk_sub.hasargs ? sv_2mortal (newRV_inc ((SV *)cx->blk_sub.argarray)) : &PL_sv_undef);
		760	PUTBACK;
		761	cb = hv_fetch ((HV *)SvRV (coro_current), "_trace_sub_cb", sizeof ("_trace_sub_cb") - 1, 0);
		762	if (cb) call_sv (*cb, G_KEEPERR \| G_EVAL \| G_VOID \| G_DISCARD);
		763	SPAGAIN;
		764	FREETMPS;
		765	LEAVE;
		766	PL_runops = runops_trace;
		767	}
		768
		769	oldcxix = cxstack_ix;
		770	}
		771
		772	if (cctx->flags & CC_TRACE_LINE)
		773	{
		774	dSP;
		775
		776	PL_runops = RUNOPS_DEFAULT;
		777	ENTER;
		778	SAVETMPS;
		779	EXTEND (SP, 3);
		780	PL_runops = RUNOPS_DEFAULT;
		781	PUSHMARK (SP);
		782	PUSHs (sv_2mortal (newSVpv (OutCopFILE (oldcop), 0)));
		783	PUSHs (sv_2mortal (newSViv (CopLINE (oldcop))));
		784	PUTBACK;
		785	cb = hv_fetch ((HV *)SvRV (coro_current), "_trace_line_cb", sizeof ("_trace_line_cb") - 1, 0);
		786	if (cb) call_sv (*cb, G_KEEPERR \| G_EVAL \| G_VOID \| G_DISCARD);
		787	SPAGAIN;
		788	FREETMPS;
		789	LEAVE;
		790	PL_runops = runops_trace;
		791	}
		792	}
		793	}
669	}	794	}
670	}	795	}
671		796
672	TAINT_NOT;	797	TAINT_NOT;
673	return 0;	798	return 0;
…		…
683	LOGOP myop;	808	LOGOP myop;
684		809
685	PL_top_env = &PL_start_env;	810	PL_top_env = &PL_start_env;
686		811
687	if (cctx->flags & CC_TRACE)	812	if (cctx->flags & CC_TRACE)
688	PL_runops = runops_coro;	813	PL_runops = runops_trace;
689		814
690	Zero (&myop, 1, LOGOP);	815	Zero (&myop, 1, LOGOP);
691	myop.op_next = PL_op;	816	myop.op_next = PL_op;
692	myop.op_flags = OPf_WANT_VOID \| OPf_STACKED;	817	myop.op_flags = OPf_WANT_VOID \| OPf_STACKED;
693		818
…		…
839	cctx_first = cctx;	964	cctx_first = cctx;
840	}	965	}
841		966
842	/ coroutine switching ***************************************************/	967	/ coroutine switching ***************************************************/
843		968
844	/* never call directly, always through the coro_state_transfer global variable */	969	static void NOINLINE
		970	transfer_check (pTHX_ struct coro prev, struct coro next)
		971	{
		972	if (prev != next)
		973	{
		974	if (!(prev->flags & (CF_RUNNING \| CF_NEW)))
		975	croak ("Coro::State::transfer called with non-running/new prev Coro::State, but can only transfer from running or new states");
		976
		977	if (next->flags & CF_RUNNING)
		978	croak ("Coro::State::transfer called with running next Coro::State, but can only transfer to inactive states");
		979
		980	if (next->flags & CF_DESTROYED)
		981	croak ("Coro::State::transfer called with destroyed next Coro::State, but can only transfer to inactive states");
		982
		983	if (PL_lex_state != LEX_NOTPARSING)
		984	croak ("Coro::State::transfer called while parsing, but this is not supported");
		985	}
		986	}
		987
		988	/* always use the TRANSFER macro */
845	static void NOINLINE	989	static void NOINLINE
846	transfer (pTHX_ struct coro prev, struct coro next)	990	transfer (pTHX_ struct coro prev, struct coro next)
847	{	991	{
848	dSTACKLEVEL;	992	dSTACKLEVEL;
849		993
…		…
863	Newz (0, prev->cctx, 1, coro_cctx);	1007	Newz (0, prev->cctx, 1, coro_cctx);
864	prev->flags &= ~CF_NEW;	1008	prev->flags &= ~CF_NEW;
865	prev->flags \|= CF_RUNNING;	1009	prev->flags \|= CF_RUNNING;
866	}	1010	}
867		1011
868	/TODO: must not croak here /
869	if (!prev->flags & CF_RUNNING)
870	croak ("Coro::State::transfer called with non-running prev Coro::State, but can only transfer from running states");
871
872	if (next->flags & CF_RUNNING)
873	croak ("Coro::State::transfer called with running next Coro::State, but can only transfer to inactive states");
874
875	if (next->flags & CF_DESTROYED)
876	croak ("Coro::State::transfer called with destroyed next Coro::State, but can only transfer to inactive states");
877
878	prev->flags &= ~CF_RUNNING;	1012	prev->flags &= ~CF_RUNNING;
879	next->flags \|= CF_RUNNING;	1013	next->flags \|= CF_RUNNING;
880		1014
881	LOCK;	1015	LOCK;
882		1016
…		…
885	/* need to start coroutine */	1019	/* need to start coroutine */
886	next->flags &= ~CF_NEW;	1020	next->flags &= ~CF_NEW;
887	/* first get rid of the old state */	1021	/* first get rid of the old state */
888	save_perl (aTHX_ prev);	1022	save_perl (aTHX_ prev);
889	/* setup coroutine call */	1023	/* setup coroutine call */
890	setup_coro (aTHX_ next);	1024	coro_setup (aTHX_ next);
891	}	1025	}
892	else	1026	else
893	{	1027	{
894	/* coroutine already started */	1028	/* coroutine already started */
895	save_perl (aTHX_ prev);	1029	save_perl (aTHX_ prev);
…		…
897	}	1031	}
898		1032
899	prev__cctx = prev->cctx;	1033	prev__cctx = prev->cctx;
900		1034
901	/* possibly "free" the cctx */	1035	/* possibly "free" the cctx */
902	if (prev__cctx->idle_sp == STACKLEVEL)	1036	if (prev__cctx->idle_sp == STACKLEVEL && !(prev__cctx->flags & CC_TRACE))
903	{	1037	{
904	/* I assume that STACKLEVEL is a stronger indicator than PL_top_env changes */	1038	/* I assume that STACKLEVEL is a stronger indicator than PL_top_env changes */
905	assert (("ERROR: current top_env must equal previous top_env", PL_top_env == prev__cctx->idle_te));	1039	assert (("ERROR: current top_env must equal previous top_env", PL_top_env == prev__cctx->idle_te));
906		1040
907	prev->cctx = 0;	1041	prev->cctx = 0;
908		1042
909	cctx_put (prev__cctx);	1043	cctx_put (prev__cctx);
910	}	1044	}
		1045
		1046	++next->usecount;
911		1047
912	if (!next->cctx)	1048	if (!next->cctx)
913	next->cctx = cctx_get (aTHX);	1049	next->cctx = cctx_get (aTHX);
914		1050
915	if (prev__cctx != next->cctx)	1051	if (prev__cctx != next->cctx)
…		…
928	{	1064	{
929	struct coro prev, next;	1065	struct coro prev, next;
930	};	1066	};
931		1067
932	#define TRANSFER(ta) transfer (aTHX_ (ta).prev, (ta).next)	1068	#define TRANSFER(ta) transfer (aTHX_ (ta).prev, (ta).next)
		1069	#define TRANSFER_CHECK(ta) transfer_check (aTHX_ (ta).prev, (ta).next)
933		1070
934	/ high level stuff ******************************************************/	1071	/ high level stuff ******************************************************/
935		1072
936	static int	1073	static int
937	coro_state_destroy (pTHX_ struct coro *coro)	1074	coro_state_destroy (pTHX_ struct coro *coro)
…		…
954		1091
955	if (coro->mainstack && coro->mainstack != main_mainstack)	1092	if (coro->mainstack && coro->mainstack != main_mainstack)
956	{	1093	{
957	struct coro temp;	1094	struct coro temp;
958		1095
959	assert (!(coro->flags & CF_RUNNING));
960
961	Zero (&temp, 1, struct coro);
962	temp.save = CORO_SAVE_ALL;
963
964	if (coro->flags & CF_RUNNING)	1096	if (coro->flags & CF_RUNNING)
965	croak ("FATAL: tried to destroy currently running coroutine");	1097	croak ("FATAL: tried to destroy currently running coroutine");
966		1098
967	save_perl (aTHX_ &temp);	1099	save_perl (aTHX_ &temp);
968	load_perl (aTHX_ coro);	1100	load_perl (aTHX_ coro);
969		1101
970	coro_destroy_stacks (aTHX);	1102	coro_destroy (aTHX_ coro);
971		1103
972	load_perl (aTHX_ &temp); /* this will get rid of defsv etc.. */	1104	load_perl (aTHX_ &temp); /* this will get rid of defsv etc.. */
973		1105
974	coro->mainstack = 0;	1106	coro->mainstack = 0;
975	}	1107	}
…		…
1020	#else	1152	#else
1021	# define MGf_DUP 0	1153	# define MGf_DUP 0
1022	#endif	1154	#endif
1023	};	1155	};
1024		1156
1025	static struct coro *
1026	SvSTATE_ (pTHX_ SV *coro)
1027	{
1028	HV *stash;
1029	MAGIC *mg;
1030
1031	if (SvROK (coro))
1032	coro = SvRV (coro);
1033
1034	stash = SvSTASH (coro);
1035	if (stash != coro_stash && stash != coro_state_stash)
1036	{
1037	/* very slow, but rare, check */
1038	if (!sv_derived_from (sv_2mortal (newRV_inc (coro)), "Coro::State"))
1039	croak ("Coro::State object required");
1040	}
1041
1042	mg = CORO_MAGIC (coro);
1043	return (struct coro *)mg->mg_ptr;
1044	}
1045
1046	#define SvSTATE(sv) SvSTATE_ (aTHX_ (sv))
1047
1048	static void	1157	static void
1049	prepare_transfer (pTHX_ struct transfer_args ta, SV prev_sv, SV *next_sv)	1158	prepare_transfer (pTHX_ struct transfer_args ta, SV prev_sv, SV *next_sv)
1050	{	1159	{
1051	ta->prev = SvSTATE (prev_sv);	1160	ta->prev = SvSTATE (prev_sv);
1052	ta->next = SvSTATE (next_sv);	1161	ta->next = SvSTATE (next_sv);
		1162	TRANSFER_CHECK (*ta);
1053	}	1163	}
1054		1164
1055	static void	1165	static void
1056	api_transfer (SV prev_sv, SV next_sv)	1166	api_transfer (SV prev_sv, SV next_sv)
1057	{	1167	{
1058	dTHX;	1168	dTHX;
1059	struct transfer_args ta;	1169	struct transfer_args ta;
1060		1170
1061	prepare_transfer (aTHX_ &ta, prev_sv, next_sv);	1171	prepare_transfer (aTHX_ &ta, prev_sv, next_sv);
1062	TRANSFER (ta);	1172	TRANSFER (ta);
1063	}
1064
1065	static int
1066	api_save (SV *coro_sv, int new_save)
1067	{
1068	dTHX;
1069	struct coro *coro = SvSTATE (coro_sv);
1070	int old_save = coro->save;
1071
1072	if (new_save >= 0)
1073	coro->save = new_save;
1074
1075	return old_save;
1076	}	1173	}
1077		1174
1078	/ Coro ******************************************************************/	1175	/ Coro ******************************************************************/
1079		1176
1080	static void	1177	static void
…		…
1174	break;	1271	break;
1175	}	1272	}
1176		1273
1177	/* free this only after the transfer */	1274	/* free this only after the transfer */
1178	prev_sv = SvRV (coro_current);	1275	prev_sv = SvRV (coro_current);
1179	SvRV_set (coro_current, next_sv);
1180	ta->prev = SvSTATE (prev_sv);	1276	ta->prev = SvSTATE (prev_sv);
1181		1277	TRANSFER_CHECK (*ta);
1182	assert (ta->next->flags & CF_READY);	1278	assert (ta->next->flags & CF_READY);
1183	ta->next->flags &= ~CF_READY;	1279	ta->next->flags &= ~CF_READY;
		1280	SvRV_set (coro_current, next_sv);
1184		1281
1185	LOCK;	1282	LOCK;
1186	free_coro_mortal (aTHX);	1283	free_coro_mortal (aTHX);
1187	coro_mortal = prev_sv;	1284	coro_mortal = prev_sv;
1188	UNLOCK;	1285	UNLOCK;
…		…
1249	}	1346	}
1250	else	1347	else
1251	return 0;	1348	return 0;
1252	}	1349	}
1253		1350
		1351	static void
		1352	api_trace (SV *coro_sv, int flags)
		1353	{
		1354	dTHX;
		1355	struct coro *coro = SvSTATE (coro_sv);
		1356
		1357	if (flags & CC_TRACE)
		1358	{
		1359	if (!coro->cctx)
		1360	coro->cctx = cctx_new ();
		1361	else if (!(coro->cctx->flags & CC_TRACE))
		1362	croak ("cannot enable tracing on coroutine with custom stack");
		1363
		1364	coro->cctx->flags \|= CC_NOREUSE \| (flags & (CC_TRACE \| CC_TRACE_ALL));
		1365	}
		1366	else if (coro->cctx && coro->cctx->flags & CC_TRACE)
		1367	{
		1368	coro->cctx->flags &= ~(CC_TRACE \| CC_TRACE_ALL);
		1369
		1370	if (coro->flags & CF_RUNNING)
		1371	PL_runops = RUNOPS_DEFAULT;
		1372	else
		1373	coro->runops = RUNOPS_DEFAULT;
		1374	}
		1375	}
		1376
1254	MODULE = Coro::State PACKAGE = Coro::State	1377	MODULE = Coro::State PACKAGE = Coro::State PREFIX = api_
1255		1378
1256	PROTOTYPES: DISABLE	1379	PROTOTYPES: DISABLE
1257		1380
1258	BOOT:	1381	BOOT:
1259	{	1382	{
1260	#ifdef USE_ITHREADS	1383	#ifdef USE_ITHREADS
1261	MUTEX_INIT (&coro_mutex);	1384	MUTEX_INIT (&coro_mutex);
1262	#endif	1385	#endif
1263	BOOT_PAGESIZE;	1386	BOOT_PAGESIZE;
1264		1387
		1388	irsgv = gv_fetchpv ("/" , GV_ADD\|GV_NOTQUAL, SVt_PV);
		1389	stdoutgv = gv_fetchpv ("STDOUT", GV_ADD\|GV_NOTQUAL, SVt_PVIO);
		1390
1265	coro_state_stash = gv_stashpv ("Coro::State", TRUE);	1391	coro_state_stash = gv_stashpv ("Coro::State", TRUE);
1266		1392
1267	newCONSTSUB (coro_state_stash, "SAVE_DEFAV", newSViv (CORO_SAVE_DEFAV));	1393	newCONSTSUB (coro_state_stash, "CC_TRACE" , newSViv (CC_TRACE));
1268	newCONSTSUB (coro_state_stash, "SAVE_DEFSV", newSViv (CORO_SAVE_DEFSV));	1394	newCONSTSUB (coro_state_stash, "CC_TRACE_SUB" , newSViv (CC_TRACE_SUB));
1269	newCONSTSUB (coro_state_stash, "SAVE_ERRSV", newSViv (CORO_SAVE_ERRSV));
1270	newCONSTSUB (coro_state_stash, "SAVE_IRSSV", newSViv (CORO_SAVE_IRSSV));
1271	newCONSTSUB (coro_state_stash, "SAVE_DEFFH", newSViv (CORO_SAVE_DEFFH));
1272	newCONSTSUB (coro_state_stash, "SAVE_DEF", newSViv (CORO_SAVE_DEF));	1395	newCONSTSUB (coro_state_stash, "CC_TRACE_LINE", newSViv (CC_TRACE_LINE));
1273	newCONSTSUB (coro_state_stash, "SAVE_ALL", newSViv (CORO_SAVE_ALL));	1396	newCONSTSUB (coro_state_stash, "CC_TRACE_ALL" , newSViv (CC_TRACE_ALL));
1274		1397
1275	main_mainstack = PL_mainstack;	1398	main_mainstack = PL_mainstack;
1276	main_top_env = PL_top_env;	1399	main_top_env = PL_top_env;
1277		1400
1278	while (main_top_env->je_prev)	1401	while (main_top_env->je_prev)
…		…
1291	struct coro *coro;	1414	struct coro *coro;
1292	HV *hv;	1415	HV *hv;
1293	int i;	1416	int i;
1294		1417
1295	Newz (0, coro, 1, struct coro);	1418	Newz (0, coro, 1, struct coro);
1296	coro->args = newAV ();	1419	coro->args = newAV ();
1297	coro->save = CORO_SAVE_DEF;
1298	coro->flags = CF_NEW;	1420	coro->flags = CF_NEW;
1299		1421
1300	if (coro_first) coro_first->prev = coro;	1422	if (coro_first) coro_first->prev = coro;
1301	coro->next = coro_first;	1423	coro->next = coro_first;
1302	coro_first = coro;	1424	coro_first = coro;
…		…
1309	av_push (coro->args, newSVsv (ST (i)));	1431	av_push (coro->args, newSVsv (ST (i)));
1310	}	1432	}
1311	OUTPUT:	1433	OUTPUT:
1312	RETVAL	1434	RETVAL
1313		1435
1314	int	1436	# these not obviously related functions are all rolled into the same xs
1315	save (SV *coro, int new_save = -1)	1437	# function to increase chances that they all will call transfer with the same
1316	CODE:	1438	# stack offset
1317	RETVAL = api_save (coro, new_save);
1318	OUTPUT:
1319	RETVAL
1320
1321	int
1322	save_also (SV *coro_sv, int save_also)
1323	CODE:
1324	{
1325	struct coro *coro = SvSTATE (coro_sv);
1326	RETVAL = coro->save;
1327	coro->save \|= save_also;
1328	}
1329	OUTPUT:
1330	RETVAL
1331
1332	void	1439	void
1333	_set_stacklevel (...)	1440	_set_stacklevel (...)
1334	ALIAS:	1441	ALIAS:
1335	Coro::State::transfer = 1	1442	Coro::State::transfer = 1
1336	Coro::schedule = 2	1443	Coro::schedule = 2
…		…
1431	{	1538	{
1432	if (coro->mainstack)	1539	if (coro->mainstack)
1433	{	1540	{
1434	struct coro temp;	1541	struct coro temp;
1435	Zero (&temp, 1, struct coro);	1542	Zero (&temp, 1, struct coro);
1436	temp.save = CORO_SAVE_ALL;
1437		1543
1438	if (!(coro->flags & CF_RUNNING))	1544	if (!(coro->flags & CF_RUNNING))
1439	{	1545	{
1440	save_perl (aTHX_ &temp);	1546	save_perl (aTHX_ &temp);
1441	load_perl (aTHX_ coro);	1547	load_perl (aTHX_ coro);
…		…
1462	save_perl (aTHX_ coro);	1568	save_perl (aTHX_ coro);
1463	load_perl (aTHX_ &temp);	1569	load_perl (aTHX_ &temp);
1464	}	1570	}
1465	}	1571	}
1466	}	1572	}
1467		1573
1468	SV *	1574	SV *
1469	is_ready (Coro::State coro)	1575	is_ready (Coro::State coro)
1470	PROTOTYPE: $	1576	PROTOTYPE: $
1471	ALIAS:	1577	ALIAS:
1472	is_ready = CF_READY	1578	is_ready = CF_READY
…		…
1477	RETVAL = boolSV (coro->flags & ix);	1583	RETVAL = boolSV (coro->flags & ix);
1478	OUTPUT:	1584	OUTPUT:
1479	RETVAL	1585	RETVAL
1480		1586
1481	void	1587	void
1482	trace (Coro::State coro, int enable = 0)	1588	api_trace (SV *coro, int flags = CC_TRACE \| CC_TRACE_SUB)
1483	CODE:
1484	if (enable)
1485	{
1486	if (coro->cctx && coro->cctx->flags & CC_TRACE)
1487	XSRETURN_EMPTY;
1488
1489	if (coro->flags & CF_RUNNING)
1490	croak ("cannot enable tracing on running coroutine");
1491
1492	if (coro->cctx)
1493	croak ("cannot enable tracing on coroutine with custom stack");
1494
1495	coro->cctx = cctx_new ();
1496	coro->cctx->flags \|= CC_TRACE;
1497	}
1498	else
1499	if (coro->cctx && coro->cctx->flags & CC_TRACE)
1500	{
1501	coro->cctx->flags &= ~CC_TRACE;
1502	coro->cctx->flags \|= CC_NOREUSE;
1503	}
1504		1589
1505	SV *	1590	SV *
1506	has_stack (Coro::State coro)	1591	has_stack (Coro::State coro)
1507	PROTOTYPE: $	1592	PROTOTYPE: $
1508	CODE:	1593	CODE:
1509	RETVAL = boolSV (!!coro->cctx);	1594	RETVAL = boolSV (!!coro->cctx);
1510	OUTPUT:	1595	OUTPUT:
1511	RETVAL	1596	RETVAL
1512		1597
		1598	int
		1599	is_traced (Coro::State coro)
		1600	PROTOTYPE: $
		1601	CODE:
		1602	RETVAL = (coro->cctx ? coro->cctx->flags : 0) & CC_TRACE_ALL;
		1603	OUTPUT:
		1604	RETVAL
		1605
1513	IV	1606	IV
1514	rss (Coro::State coro)	1607	rss (Coro::State coro)
1515	PROTOTYPE: $	1608	PROTOTYPE: $
		1609	ALIAS:
		1610	usecount = 1
1516	CODE:	1611	CODE:
1517	RETVAL = coro_rss (coro);	1612	switch (ix)
		1613	{
		1614	case 0: RETVAL = coro_rss (aTHX_ coro); break;
		1615	case 1: RETVAL = coro->usecount; break;
		1616	}
1518	OUTPUT:	1617	OUTPUT:
1519	RETVAL	1618	RETVAL
1520		1619
1521		1620
1522	MODULE = Coro::State PACKAGE = Coro	1621	MODULE = Coro::State PACKAGE = Coro
…		…
1527		1626
1528	sv_pool_rss = get_sv ("Coro::POOL_RSS" , TRUE);	1627	sv_pool_rss = get_sv ("Coro::POOL_RSS" , TRUE);
1529	sv_pool_size = get_sv ("Coro::POOL_SIZE" , TRUE);	1628	sv_pool_size = get_sv ("Coro::POOL_SIZE" , TRUE);
1530	av_async_pool = get_av ("Coro::async_pool", TRUE);	1629	av_async_pool = get_av ("Coro::async_pool", TRUE);
1531		1630
1532	coro_current = get_sv ("Coro::current", FALSE);	1631	coro_current = get_sv ("Coro::current", FALSE);
1533	SvREADONLY_on (coro_current);	1632	SvREADONLY_on (coro_current);
1534		1633
1535	coro_stash = gv_stashpv ("Coro", TRUE);	1634	coro_stash = gv_stashpv ("Coro", TRUE);
1536		1635
1537	newCONSTSUB (coro_stash, "PRIO_MAX", newSViv (PRIO_MAX));	1636	newCONSTSUB (coro_stash, "PRIO_MAX", newSViv (PRIO_MAX));
…		…
1546		1645
1547	{	1646	{
1548	SV *sv = perl_get_sv("Coro::API", 1);	1647	SV *sv = perl_get_sv("Coro::API", 1);
1549		1648
1550	coroapi.schedule = api_schedule;	1649	coroapi.schedule = api_schedule;
1551	coroapi.save = api_save;
1552	coroapi.cede = api_cede;	1650	coroapi.cede = api_cede;
1553	coroapi.cede_notself = api_cede_notself;	1651	coroapi.cede_notself = api_cede_notself;
1554	coroapi.ready = api_ready;	1652	coroapi.ready = api_ready;
1555	coroapi.is_ready = api_is_ready;	1653	coroapi.is_ready = api_is_ready;
1556	coroapi.nready = &coro_nready;	1654	coroapi.nready = &coro_nready;
…		…
1610	# for async_pool speedup	1708	# for async_pool speedup
1611	void	1709	void
1612	_pool_1 (SV *cb)	1710	_pool_1 (SV *cb)
1613	CODE:	1711	CODE:
1614	{	1712	{
1615	int i, len;	1713	struct coro *coro = SvSTATE (coro_current);
1616	HV hv = (HV )SvRV (coro_current);	1714	HV hv = (HV )SvRV (coro_current);
1617	AV *defav = GvAV (PL_defgv);	1715	AV *defav = GvAV (PL_defgv);
1618	SV *invoke = hv_delete (hv, "_invoke", sizeof ("_invoke") - 1, 0);	1716	SV *invoke = hv_delete (hv, "_invoke", sizeof ("_invoke") - 1, 0);
1619	AV *invoke_av;	1717	AV *invoke_av;
		1718	int i, len;
1620		1719
1621	if (!invoke)	1720	if (!invoke)
1622	croak ("\3terminate\2\n");	1721	croak ("\3terminate\2\n");
		1722
		1723	SvREFCNT_dec (coro->saved_deffh);
		1724	coro->saved_deffh = SvREFCNT_inc ((SV *)PL_defoutgv);
1623		1725
1624	hv_store (hv, "desc", sizeof ("desc") - 1,	1726	hv_store (hv, "desc", sizeof ("desc") - 1,
1625	newSVpvn ("[async_pool]", sizeof ("[async_pool]") - 1), 0);	1727	newSVpvn ("[async_pool]", sizeof ("[async_pool]") - 1), 0);
1626		1728
1627	invoke_av = (AV *)SvRV (invoke);	1729	invoke_av = (AV *)SvRV (invoke);
…		…
1645	{	1747	{
1646	struct coro *coro = SvSTATE (coro_current);	1748	struct coro *coro = SvSTATE (coro_current);
1647		1749
1648	sv_setsv (cb, &PL_sv_undef);	1750	sv_setsv (cb, &PL_sv_undef);
1649		1751
		1752	SvREFCNT_dec ((SV )PL_defoutgv); PL_defoutgv = (GV )coro->saved_deffh;
		1753	coro->saved_deffh = 0;
		1754
1650	if (coro_rss (coro) > SvIV (sv_pool_rss)	1755	if (coro_rss (aTHX_ coro) > SvIV (sv_pool_rss)
1651	\|\| av_len (av_async_pool) + 1 >= SvIV (sv_pool_size))	1756	\|\| av_len (av_async_pool) + 1 >= SvIV (sv_pool_size))
1652	croak ("\3terminate\2\n");	1757	croak ("\3terminate\2\n");
1653		1758
1654	av_clear (GvAV (PL_defgv));	1759	av_clear (GvAV (PL_defgv));
1655	hv_store ((HV *)SvRV (coro_current), "desc", sizeof ("desc") - 1,	1760	hv_store ((HV *)SvRV (coro_current), "desc", sizeof ("desc") - 1,
1656	newSVpvn ("[async_pool idle]", sizeof ("[async_pool idle]") - 1), 0);	1761	newSVpvn ("[async_pool idle]", sizeof ("[async_pool idle]") - 1), 0);
1657	coro->save = CORO_SAVE_DEF;	1762
1658	coro->prio = 0;	1763	coro->prio = 0;
		1764
		1765	if (coro->cctx && (coro->cctx->flags & CC_TRACE))
		1766	api_trace (coro_current, 0);
		1767
1659	av_push (av_async_pool, newSVsv (coro_current));	1768	av_push (av_async_pool, newSVsv (coro_current));
1660	}	1769	}
1661		1770
1662		1771
1663	MODULE = Coro::State PACKAGE = Coro::AIO	1772	MODULE = Coro::State PACKAGE = Coro::AIO

Diff Legend

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

Comparing Coro/Coro/State.xs (file contents): Revision 1.166 by root, Wed Sep 26 21:11:28 2007 UTC vs. Revision 1.190 by root, Fri Oct 5 20:39:39 2007 UTC

Diff Legend

Comparing Coro/Coro/State.xs (file contents):
Revision 1.166 by root, Wed Sep 26 21:11:28 2007 UTC vs.
Revision 1.190 by root, Fri Oct 5 20:39:39 2007 UTC