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

Comparing Coro/Coro/State.xs (file contents):
Revision 1.169 by root, Thu Sep 27 16:25:10 2007 UTC vs.
Revision 1.188 by root, Fri Oct 5 20:36:31 2007 UTC

…		…
132	static AV main_mainstack; / used to differentiate between $main and others */	132	static AV main_mainstack; / used to differentiate between $main and others */
133	static JMPENV *main_top_env;	133	static JMPENV *main_top_env;
134	static HV coro_state_stash, coro_stash;	134	static HV coro_state_stash, coro_stash;
135	static SV coro_mortal; / will be freed after next transfer */	135	static SV coro_mortal; / will be freed after next transfer */
136		136
		137	static GV irsgv; / $/ */
		138	static GV stdoutgv; / STDOUT */
		139
137	/* async_pool helper stuff */	140	/* async_pool helper stuff */
138	static SV *sv_pool_rss;	141	static SV *sv_pool_rss;
139	static SV *sv_pool_size;	142	static SV *sv_pool_size;
140	static AV *av_async_pool;	143	static AV *av_async_pool;
141		144
…		…
184	coro_cctx *cctx;	187	coro_cctx *cctx;
185		188
186	/* data associated with this coroutine (initial args) */	189	/* data associated with this coroutine (initial args) */
187	AV *args;	190	AV *args;
188	int refcnt;	191	int refcnt;
189	int save; /* CORO_SAVE flags */
190	int flags; /* CF_ flags */	192	int flags; /* CF_ flags */
191		193
192	/* optionally saved, might be zero */	194	/* optionally saved, might be zero */
193	AV defav; / @_ */	195	AV defav; / @_ */
194	SV defsv; / $_ */	196	SV defsv; / $_ */
195	SV errsv; / $@ */	197	SV errsv; / $@ */
196	GV deffh; / default filehandle */	198	SV deffh; / default filehandle */
197	SV irssv; / $/ */	199	SV irssv; / $/ */
198	SV irssv_sv; / real $/ cache */	200	SV irssv_sv; / real $/ cache */
199		201
200	#define VAR(name,type) type name;	202	#define VAR(name,type) type name;
201	# include "state.h"	203	# include "state.h"
202	#undef VAR	204	#undef VAR
203		205
		206	/* statistics */
		207	int usecount; /* number of transfers to this coro */
		208
204	/* coro process data */	209	/* coro process data */
205	int prio;	210	int prio;
		211	//SV *throw;
		212
		213	/* async_pool */
		214	SV *saved_deffh;
206		215
207	/* linked list */	216	/* linked list */
208	struct coro next, prev;	217	struct coro next, prev;
209	HV hv; / the perl hash associated with this coro, if any */	218	HV hv; / the perl hash associated with this coro, if any */
210	};	219	};
…		…
307	HV *stash;	316	HV *stash;
308	MAGIC *mg;	317	MAGIC *mg;
309		318
310	if (SvROK (coro))	319	if (SvROK (coro))
311	coro = SvRV (coro);	320	coro = SvRV (coro);
		321
		322	if (SvTYPE (coro) != SVt_PVHV)
		323	croak ("Coro::State object required");
312		324
313	stash = SvSTASH (coro);	325	stash = SvSTASH (coro);
314	if (stash != coro_stash && stash != coro_state_stash)	326	if (stash != coro_stash && stash != coro_state_stash)
315	{	327	{
316	/* very slow, but rare, check */	328	/* very slow, but rare, check */
…		…
369	AvARRAY (av)[++AvFILLp (av)] = (SV *)CvPADLIST (cv);	381	AvARRAY (av)[++AvFILLp (av)] = (SV *)CvPADLIST (cv);
370	}	382	}
371		383
372	/ load & save, init *****************************************************/	384	/ load & save, init *****************************************************/
373		385
374	#define SB do {
375	#define SE } while (0)
376
377	#define REPLACE_SV(sv,val) SB SvREFCNT_dec (sv); (sv) = (val); (val) = 0; SE
378
379	static void	386	static void
380	load_perl (pTHX_ Coro__State c)	387	load_perl (pTHX_ Coro__State c)
381	{	388	{
382	#define VAR(name,type) PL_ ## name = c->name;	389	#define VAR(name,type) PL_ ## name = c->name;
383	# include "state.h"	390	# include "state.h"
384	#undef VAR	391	#undef VAR
385		392
386	if (c->defav) REPLACE_SV (GvAV (PL_defgv), c->defav);	393	GvSV (PL_defgv) = c->defsv;
387	if (c->defsv) REPLACE_SV (DEFSV , c->defsv);	394	GvAV (PL_defgv) = c->defav;
388	if (c->errsv) REPLACE_SV (ERRSV , c->errsv);	395	GvSV (PL_errgv) = c->errsv;
389	if (c->deffh) REPLACE_SV (PL_defoutgv , c->deffh);	396	GvSV (irsgv) = c->irssv_sv;
390
391	if (c->irssv)
392	{
393	if (c->irssv == PL_rs \|\| sv_eq (PL_rs, c->irssv))
394	{
395	SvREFCNT_dec (c->irssv);
396	c->irssv = 0;
397	}
398	else
399	{
400	REPLACE_SV (PL_rs, c->irssv);
401	if (!c->irssv_sv) c->irssv_sv = get_sv ("/", 0);
402	sv_setsv (c->irssv_sv, PL_rs);
403	}
404	}
405		397
406	{	398	{
407	dSP;	399	dSP;
408	CV *cv;	400	CV *cv;
409		401
…		…
467	}	459	}
468		460
469	PUTBACK;	461	PUTBACK;
470	}	462	}
471		463
472	c->defav = c->save & CORO_SAVE_DEFAV ? (AV *)SvREFCNT_inc (GvAV (PL_defgv)) : 0;	464	c->defav = GvAV (PL_defgv);
473	c->defsv = c->save & CORO_SAVE_DEFSV ? SvREFCNT_inc (DEFSV) : 0;	465	c->defsv = DEFSV;
474	c->errsv = c->save & CORO_SAVE_ERRSV ? SvREFCNT_inc (ERRSV) : 0;	466	c->errsv = ERRSV;
475	c->deffh = c->save & CORO_SAVE_DEFFH ? (GV *)SvREFCNT_inc (PL_defoutgv) : 0;	467	c->irssv_sv = GvSV (irsgv);
476	c->irssv = c->save & CORO_SAVE_IRSSV ? SvREFCNT_inc (PL_rs) : 0;
477		468
478	#define VAR(name,type)c->name = PL_ ## name;	469	#define VAR(name,type)c->name = PL_ ## name;
479	# include "state.h"	470	# include "state.h"
480	#undef VAR	471	#undef VAR
481	}	472	}
…		…
534	* destroy the stacks, the callchain etc...	525	* destroy the stacks, the callchain etc...
535	*/	526	*/
536	static void	527	static void
537	coro_destroy_stacks (pTHX)	528	coro_destroy_stacks (pTHX)
538	{	529	{
539	if (!IN_DESTRUCT)
540	{
541	/* restore all saved variables and stuff */
542	LEAVE_SCOPE (0);
543	assert (PL_tmps_floor == -1);
544
545	/* free all temporaries */
546	FREETMPS;
547	assert (PL_tmps_ix == -1);
548
549	/* unwind all extra stacks */
550	POPSTACK_TO (PL_mainstack);
551
552	/* unwind main stack */
553	dounwind (-1);
554	}
555
556	while (PL_curstackinfo->si_next)	530	while (PL_curstackinfo->si_next)
557	PL_curstackinfo = PL_curstackinfo->si_next;	531	PL_curstackinfo = PL_curstackinfo->si_next;
558		532
559	while (PL_curstackinfo)	533	while (PL_curstackinfo)
560	{	534	{
…		…
576	Safefree (PL_retstack);	550	Safefree (PL_retstack);
577	#endif	551	#endif
578	}	552	}
579		553
580	static size_t	554	static size_t
581	coro_rss (struct coro *coro)	555	coro_rss (pTHX_ struct coro *coro)
582	{	556	{
583	size_t rss = sizeof (coro);	557	size_t rss = sizeof (*coro);
584		558
585	if (coro->mainstack)	559	if (coro->mainstack)
586	{	560	{
587	if (coro->flags & CF_RUNNING)	561	if (coro->flags & CF_RUNNING)
588	{	562	{
…		…
590	# include "state.h"	564	# include "state.h"
591	#undef VAR	565	#undef VAR
592	}	566	}
593		567
594	rss += sizeof (coro->curstackinfo);	568	rss += sizeof (coro->curstackinfo);
595	rss += sizeof (struct xpvav) + (1 + AvFILL (coro->curstackinfo->si_stack)) * sizeof (SV *);	569	rss += sizeof (SV) + sizeof (struct xpvav) + (1 + AvFILL (coro->curstackinfo->si_stack)) * sizeof (SV *);
596	rss += (coro->curstackinfo->si_cxmax + 1) * sizeof (PERL_CONTEXT);	570	rss += (coro->curstackinfo->si_cxmax + 1) * sizeof (PERL_CONTEXT);
597	rss += sizeof (struct xpvav) + (1 + AvFILL (coro->curstack)) * sizeof (SV *);	571	rss += sizeof (SV) + sizeof (struct xpvav) + (1 + AvFILL (coro->curstack)) * sizeof (SV *);
598	rss += coro->tmps_max * sizeof (SV *);	572	rss += coro->tmps_max * sizeof (SV *);
599	rss += (coro->markstack_max - coro->markstack_ptr) * sizeof (I32);	573	rss += (coro->markstack_max - coro->markstack_ptr) * sizeof (I32);
600	rss += coro->scopestack_max * sizeof (I32);	574	rss += coro->scopestack_max * sizeof (I32);
601	rss += coro->savestack_max * sizeof (ANY);	575	rss += coro->savestack_max * sizeof (ANY);
602		576
…		…
609	}	583	}
610		584
611	/ coroutine stack handling **********************************************/	585	/ coroutine stack handling **********************************************/
612		586
613	static void	587	static void
614	setup_coro (pTHX_ struct coro *coro)	588	coro_setup (pTHX_ struct coro *coro)
615	{	589	{
616	/*	590	/*
617	* emulate part of the perl startup here.	591	* emulate part of the perl startup here.
618	*/	592	*/
619	coro_init_stacks (aTHX);	593	coro_init_stacks (aTHX);
…		…
624	PL_comppad = 0;	598	PL_comppad = 0;
625	PL_curpm = 0;	599	PL_curpm = 0;
626	PL_localizing = 0;	600	PL_localizing = 0;
627	PL_dirty = 0;	601	PL_dirty = 0;
628	PL_restartop = 0;	602	PL_restartop = 0;
		603
		604	GvSV (PL_defgv) = NEWSV (0, 0);
		605	GvAV (PL_defgv) = coro->args; coro->args = 0;
		606	GvSV (PL_errgv) = NEWSV (0, 0);
		607	GvSV (irsgv) = newSVpvn ("\n", 1); sv_magic (GvSV (irsgv), (SV *)irsgv, PERL_MAGIC_sv, "/", 0);
		608	PL_rs = newSVsv (GvSV (irsgv));
		609	PL_defoutgv = SvREFCNT_inc (stdoutgv);
629		610
630	{	611	{
631	dSP;	612	dSP;
632	LOGOP myop;	613	LOGOP myop;
633
634	SvREFCNT_dec (GvAV (PL_defgv));
635	GvAV (PL_defgv) = coro->args; coro->args = 0;
636		614
637	Zero (&myop, 1, LOGOP);	615	Zero (&myop, 1, LOGOP);
638	myop.op_next = Nullop;	616	myop.op_next = Nullop;
639	myop.op_flags = OPf_WANT_VOID;	617	myop.op_flags = OPf_WANT_VOID;
640		618
…		…
648		626
649	ENTER; /* necessary e.g. for dounwind */	627	ENTER; /* necessary e.g. for dounwind */
650	}	628	}
651		629
652	static void	630	static void
		631	coro_destroy (pTHX_ struct coro *coro)
		632	{
		633	if (!IN_DESTRUCT)
		634	{
		635	/* restore all saved variables and stuff */
		636	LEAVE_SCOPE (0);
		637	assert (PL_tmps_floor == -1);
		638
		639	/* free all temporaries */
		640	FREETMPS;
		641	assert (PL_tmps_ix == -1);
		642
		643	/* unwind all extra stacks */
		644	POPSTACK_TO (PL_mainstack);
		645
		646	/* unwind main stack */
		647	dounwind (-1);
		648	}
		649
		650	SvREFCNT_dec (GvSV (PL_defgv));
		651	SvREFCNT_dec (GvAV (PL_defgv));
		652	SvREFCNT_dec (GvSV (PL_errgv));
		653	SvREFCNT_dec (PL_defoutgv);
		654	SvREFCNT_dec (PL_rs);
		655	SvREFCNT_dec (GvSV (irsgv));
		656
		657	SvREFCNT_dec (coro->saved_deffh);
		658	//SvREFCNT_dec (coro->throw);
		659
		660	coro_destroy_stacks (aTHX);
		661	}
		662
		663	static void
653	free_coro_mortal (pTHX)	664	free_coro_mortal (pTHX)
654	{	665	{
655	if (coro_mortal)	666	if (coro_mortal)
656	{	667	{
657	SvREFCNT_dec (coro_mortal);	668	SvREFCNT_dec (coro_mortal);
…		…
669		680
670	while ((PL_op = CALL_FPTR (PL_op->op_ppaddr) (aTHX)))	681	while ((PL_op = CALL_FPTR (PL_op->op_ppaddr) (aTHX)))
671	{	682	{
672	PERL_ASYNC_CHECK ();	683	PERL_ASYNC_CHECK ();
673		684
674	if (PL_op->op_type == OP_LEAVESUB && cctx->flags & CC_TRACE_SUB)	685	if (cctx->flags & CC_TRACE_ALL)
675	{	686	{
676	PERL_CONTEXT *cx = &cxstack[cxstack_ix];	687	if (PL_op->op_type == OP_LEAVESUB && cctx->flags & CC_TRACE_SUB)
677	SV bot, top;
678	AV av = newAV (); / return values */
679	SV **cb;
680	runops_proc_t old_runops = PL_runops;
681	dSP;
682	ENTER;
683	SAVETMPS;
684	EXTEND (SP, 3);
685	PL_runops = RUNOPS_DEFAULT;
686
687	GV *gv = CvGV (cx->blk_sub.cv);
688	SV *fullname = sv_2mortal (newSV (0));
689	if (isGV (gv))
690	gv_efullname3 (fullname, gv, 0);
691
692	bot = PL_stack_base + cx->blk_oldsp + 1;
693	top = cx->blk_gimme == G_ARRAY ? SP + 1
694	: cx->blk_gimme == G_SCALAR ? bot + 1
695	: bot;
696
697	while (bot < top)
698	av_push (av, SvREFCNT_inc (*bot++));
699
700	PUSHMARK (SP);
701	PUSHs (&PL_sv_no);
702	PUSHs (fullname);
703	PUSHs (sv_2mortal (newRV_noinc ((SV *)av)));
704	PUTBACK;
705	cb = hv_fetch ((HV *)SvRV (coro_current), "_trace_sub_cb", sizeof ("_trace_sub_cb") - 1, 0);
706	if (cb) call_sv (*cb, G_KEEPERR \| G_EVAL \| G_VOID \| G_DISCARD);
707	SPAGAIN;
708
709	FREETMPS;
710	LEAVE;
711	PL_runops = old_runops;
712	}
713
714	if (oldcop != PL_curcop)
715	{
716	oldcop = PL_curcop;
717
718	if (PL_curcop != &PL_compiling)
719	{	688	{
		689	PERL_CONTEXT *cx = &cxstack[cxstack_ix];
		690	SV bot, top;
		691	AV av = newAV (); / return values */
720	SV **cb;	692	SV **cb;
721	runops_proc_t old_runops = PL_runops;
722	dSP;	693	dSP;
		694
		695	GV *gv = CvGV (cx->blk_sub.cv);
		696	SV *fullname = sv_2mortal (newSV (0));
		697	if (isGV (gv))
		698	gv_efullname3 (fullname, gv, 0);
		699
		700	bot = PL_stack_base + cx->blk_oldsp + 1;
		701	top = cx->blk_gimme == G_ARRAY ? SP + 1
		702	: cx->blk_gimme == G_SCALAR ? bot + 1
		703	: bot;
		704
		705	while (bot < top)
		706	av_push (av, SvREFCNT_inc (*bot++));
		707
		708	PL_runops = RUNOPS_DEFAULT;
723	ENTER;	709	ENTER;
724	SAVETMPS;	710	SAVETMPS;
725	EXTEND (SP, 3);	711	EXTEND (SP, 3);
726	PL_runops = RUNOPS_DEFAULT;	712	PUSHMARK (SP);
		713	PUSHs (&PL_sv_no);
		714	PUSHs (fullname);
		715	PUSHs (sv_2mortal (newRV_noinc ((SV *)av)));
		716	PUTBACK;
		717	cb = hv_fetch ((HV *)SvRV (coro_current), "_trace_sub_cb", sizeof ("_trace_sub_cb") - 1, 0);
		718	if (cb) call_sv (*cb, G_KEEPERR \| G_EVAL \| G_VOID \| G_DISCARD);
		719	SPAGAIN;
		720	FREETMPS;
		721	LEAVE;
		722	PL_runops = runops_trace;
		723	}
727		724
728	if (oldcxix != cxstack_ix && cctx->flags & CC_TRACE_SUB)	725	if (oldcop != PL_curcop)
		726	{
		727	oldcop = PL_curcop;
		728
		729	if (PL_curcop != &PL_compiling)
729	{	730	{
730	PERL_CONTEXT *cx = &cxstack[cxstack_ix];	731	SV **cb;
731		732
732	if (CxTYPE (cx) == CXt_SUB && oldcxix < cxstack_ix)	733	if (oldcxix != cxstack_ix && cctx->flags & CC_TRACE_SUB)
733	{	734	{
		735	PERL_CONTEXT *cx = &cxstack[cxstack_ix];
		736
		737	if (CxTYPE (cx) == CXt_SUB && oldcxix < cxstack_ix)
		738	{
		739	runops_proc_t old_runops = PL_runops;
		740	dSP;
734	GV *gv = CvGV (cx->blk_sub.cv);	741	GV *gv = CvGV (cx->blk_sub.cv);
735	SV *fullname = sv_2mortal (newSV (0));	742	SV *fullname = sv_2mortal (newSV (0));
		743
736	if (isGV (gv))	744	if (isGV (gv))
737	gv_efullname3 (fullname, gv, 0);	745	gv_efullname3 (fullname, gv, 0);
738		746
		747	PL_runops = RUNOPS_DEFAULT;
		748	ENTER;
		749	SAVETMPS;
		750	EXTEND (SP, 3);
		751	PUSHMARK (SP);
		752	PUSHs (&PL_sv_yes);
		753	PUSHs (fullname);
		754	PUSHs (cx->blk_sub.hasargs ? sv_2mortal (newRV_inc ((SV *)cx->blk_sub.argarray)) : &PL_sv_undef);
		755	PUTBACK;
		756	cb = hv_fetch ((HV *)SvRV (coro_current), "_trace_sub_cb", sizeof ("_trace_sub_cb") - 1, 0);
		757	if (cb) call_sv (*cb, G_KEEPERR \| G_EVAL \| G_VOID \| G_DISCARD);
		758	SPAGAIN;
		759	FREETMPS;
		760	LEAVE;
		761	PL_runops = runops_trace;
		762	}
		763
		764	oldcxix = cxstack_ix;
		765	}
		766
		767	if (cctx->flags & CC_TRACE_LINE)
		768	{
		769	dSP;
		770
		771	PL_runops = RUNOPS_DEFAULT;
		772	ENTER;
		773	SAVETMPS;
		774	EXTEND (SP, 3);
		775	PL_runops = RUNOPS_DEFAULT;
739	PUSHMARK (SP);	776	PUSHMARK (SP);
740	PUSHs (&PL_sv_yes);	777	PUSHs (sv_2mortal (newSVpv (OutCopFILE (oldcop), 0)));
741	PUSHs (fullname);	778	PUSHs (sv_2mortal (newSViv (CopLINE (oldcop))));
742	PUSHs (cx->blk_sub.hasargs ? sv_2mortal (newRV_inc ((SV *)cx->blk_sub.argarray)) : &PL_sv_undef);
743	PUTBACK;	779	PUTBACK;
744	cb = hv_fetch ((HV *)SvRV (coro_current), "_trace_sub_cb", sizeof ("_trace_sub_cb") - 1, 0);	780	cb = hv_fetch ((HV *)SvRV (coro_current), "_trace_line_cb", sizeof ("_trace_line_cb") - 1, 0);
745	if (cb) call_sv (*cb, G_KEEPERR \| G_EVAL \| G_VOID \| G_DISCARD);	781	if (cb) call_sv (*cb, G_KEEPERR \| G_EVAL \| G_VOID \| G_DISCARD);
746	SPAGAIN;	782	SPAGAIN;
		783	FREETMPS;
		784	LEAVE;
		785	PL_runops = runops_trace;
747	}	786	}
748
749	oldcxix = cxstack_ix;
750	}	787	}
751
752	if (cctx->flags & CC_TRACE_LINE)
753	{
754	PUSHMARK (SP);
755	PUSHs (sv_2mortal (newSVpv (OutCopFILE (oldcop), 0)));
756	PUSHs (sv_2mortal (newSViv (CopLINE (oldcop))));
757	PUTBACK;
758	cb = hv_fetch ((HV *)SvRV (coro_current), "_trace_line_cb", sizeof ("_trace_line_cb") - 1, 0);
759	if (cb) call_sv (*cb, G_KEEPERR \| G_EVAL \| G_VOID \| G_DISCARD);
760	SPAGAIN;
761	}
762
763	FREETMPS;
764	LEAVE;
765	PL_runops = old_runops;
766	}	788	}
767	}	789	}
768	}	790	}
769		791
770	TAINT_NOT;	792	TAINT_NOT;
…		…
937	cctx_first = cctx;	959	cctx_first = cctx;
938	}	960	}
939		961
940	/ coroutine switching ***************************************************/	962	/ coroutine switching ***************************************************/
941		963
942	/* never call directly, always through the coro_state_transfer global variable */	964	static void NOINLINE
		965	transfer_check (pTHX_ struct coro prev, struct coro next)
		966	{
		967	if (prev != next)
		968	{
		969	if (!(prev->flags & (CF_RUNNING \| CF_NEW)))
		970	croak ("Coro::State::transfer called with non-running/new prev Coro::State, but can only transfer from running or new states");
		971
		972	if (next->flags & CF_RUNNING)
		973	croak ("Coro::State::transfer called with running next Coro::State, but can only transfer to inactive states");
		974
		975	if (next->flags & CF_DESTROYED)
		976	croak ("Coro::State::transfer called with destroyed next Coro::State, but can only transfer to inactive states");
		977
		978	if (PL_lex_state != LEX_NOTPARSING)
		979	croak ("Coro::State::transfer called while parsing, but this is not supported");
		980	}
		981	}
		982
		983	/* always use the TRANSFER macro */
943	static void NOINLINE	984	static void NOINLINE
944	transfer (pTHX_ struct coro prev, struct coro next)	985	transfer (pTHX_ struct coro prev, struct coro next)
945	{	986	{
946	dSTACKLEVEL;	987	dSTACKLEVEL;
947		988
…		…
961	Newz (0, prev->cctx, 1, coro_cctx);	1002	Newz (0, prev->cctx, 1, coro_cctx);
962	prev->flags &= ~CF_NEW;	1003	prev->flags &= ~CF_NEW;
963	prev->flags \|= CF_RUNNING;	1004	prev->flags \|= CF_RUNNING;
964	}	1005	}
965		1006
966	/TODO: must not croak here /
967	if (!prev->flags & CF_RUNNING)
968	croak ("Coro::State::transfer called with non-running prev Coro::State, but can only transfer from running states");
969
970	if (next->flags & CF_RUNNING)
971	croak ("Coro::State::transfer called with running next Coro::State, but can only transfer to inactive states");
972
973	if (next->flags & CF_DESTROYED)
974	croak ("Coro::State::transfer called with destroyed next Coro::State, but can only transfer to inactive states");
975
976	prev->flags &= ~CF_RUNNING;	1007	prev->flags &= ~CF_RUNNING;
977	next->flags \|= CF_RUNNING;	1008	next->flags \|= CF_RUNNING;
978		1009
979	LOCK;	1010	LOCK;
980		1011
…		…
983	/* need to start coroutine */	1014	/* need to start coroutine */
984	next->flags &= ~CF_NEW;	1015	next->flags &= ~CF_NEW;
985	/* first get rid of the old state */	1016	/* first get rid of the old state */
986	save_perl (aTHX_ prev);	1017	save_perl (aTHX_ prev);
987	/* setup coroutine call */	1018	/* setup coroutine call */
988	setup_coro (aTHX_ next);	1019	coro_setup (aTHX_ next);
989	}	1020	}
990	else	1021	else
991	{	1022	{
992	/* coroutine already started */	1023	/* coroutine already started */
993	save_perl (aTHX_ prev);	1024	save_perl (aTHX_ prev);
…		…
1005	prev->cctx = 0;	1036	prev->cctx = 0;
1006		1037
1007	cctx_put (prev__cctx);	1038	cctx_put (prev__cctx);
1008	}	1039	}
1009		1040
		1041	++next->usecount;
		1042
1010	if (!next->cctx)	1043	if (!next->cctx)
1011	next->cctx = cctx_get (aTHX);	1044	next->cctx = cctx_get (aTHX);
1012		1045
1013	if (prev__cctx != next->cctx)	1046	if (prev__cctx != next->cctx)
1014	{	1047	{
…		…
1026	{	1059	{
1027	struct coro prev, next;	1060	struct coro prev, next;
1028	};	1061	};
1029		1062
1030	#define TRANSFER(ta) transfer (aTHX_ (ta).prev, (ta).next)	1063	#define TRANSFER(ta) transfer (aTHX_ (ta).prev, (ta).next)
		1064	#define TRANSFER_CHECK(ta) transfer_check (aTHX_ (ta).prev, (ta).next)
1031		1065
1032	/ high level stuff ******************************************************/	1066	/ high level stuff ******************************************************/
1033		1067
1034	static int	1068	static int
1035	coro_state_destroy (pTHX_ struct coro *coro)	1069	coro_state_destroy (pTHX_ struct coro *coro)
…		…
1052		1086
1053	if (coro->mainstack && coro->mainstack != main_mainstack)	1087	if (coro->mainstack && coro->mainstack != main_mainstack)
1054	{	1088	{
1055	struct coro temp;	1089	struct coro temp;
1056		1090
1057	assert (!(coro->flags & CF_RUNNING));
1058
1059	Zero (&temp, 1, struct coro);
1060	temp.save = CORO_SAVE_ALL;
1061
1062	if (coro->flags & CF_RUNNING)	1091	if (coro->flags & CF_RUNNING)
1063	croak ("FATAL: tried to destroy currently running coroutine");	1092	croak ("FATAL: tried to destroy currently running coroutine");
1064		1093
1065	save_perl (aTHX_ &temp);	1094	save_perl (aTHX_ &temp);
1066	load_perl (aTHX_ coro);	1095	load_perl (aTHX_ coro);
1067		1096
1068	coro_destroy_stacks (aTHX);	1097	coro_destroy (aTHX_ coro);
1069		1098
1070	load_perl (aTHX_ &temp); /* this will get rid of defsv etc.. */	1099	load_perl (aTHX_ &temp); /* this will get rid of defsv etc.. */
1071		1100
1072	coro->mainstack = 0;	1101	coro->mainstack = 0;
1073	}	1102	}
…		…
1123	static void	1152	static void
1124	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)
1125	{	1154	{
1126	ta->prev = SvSTATE (prev_sv);	1155	ta->prev = SvSTATE (prev_sv);
1127	ta->next = SvSTATE (next_sv);	1156	ta->next = SvSTATE (next_sv);
		1157	TRANSFER_CHECK (*ta);
1128	}	1158	}
1129		1159
1130	static void	1160	static void
1131	api_transfer (SV prev_sv, SV next_sv)	1161	api_transfer (SV prev_sv, SV next_sv)
1132	{	1162	{
1133	dTHX;	1163	dTHX;
1134	struct transfer_args ta;	1164	struct transfer_args ta;
1135		1165
1136	prepare_transfer (aTHX_ &ta, prev_sv, next_sv);	1166	prepare_transfer (aTHX_ &ta, prev_sv, next_sv);
1137	TRANSFER (ta);	1167	TRANSFER (ta);
1138	}
1139
1140	static int
1141	api_save (SV *coro_sv, int new_save)
1142	{
1143	dTHX;
1144	struct coro *coro = SvSTATE (coro_sv);
1145	int old_save = coro->save;
1146
1147	if (new_save >= 0)
1148	coro->save = new_save;
1149
1150	return old_save;
1151	}	1168	}
1152		1169
1153	/ Coro ******************************************************************/	1170	/ Coro ******************************************************************/
1154		1171
1155	static void	1172	static void
…		…
1249	break;	1266	break;
1250	}	1267	}
1251		1268
1252	/* free this only after the transfer */	1269	/* free this only after the transfer */
1253	prev_sv = SvRV (coro_current);	1270	prev_sv = SvRV (coro_current);
1254	SvRV_set (coro_current, next_sv);
1255	ta->prev = SvSTATE (prev_sv);	1271	ta->prev = SvSTATE (prev_sv);
1256		1272	TRANSFER_CHECK (*ta);
1257	assert (ta->next->flags & CF_READY);	1273	assert (ta->next->flags & CF_READY);
1258	ta->next->flags &= ~CF_READY;	1274	ta->next->flags &= ~CF_READY;
		1275	SvRV_set (coro_current, next_sv);
1259		1276
1260	LOCK;	1277	LOCK;
1261	free_coro_mortal (aTHX);	1278	free_coro_mortal (aTHX);
1262	coro_mortal = prev_sv;	1279	coro_mortal = prev_sv;
1263	UNLOCK;	1280	UNLOCK;
…		…
1324	}	1341	}
1325	else	1342	else
1326	return 0;	1343	return 0;
1327	}	1344	}
1328		1345
		1346	static void
		1347	api_trace (SV *coro_sv, int flags)
		1348	{
		1349	dTHX;
		1350	struct coro *coro = SvSTATE (coro_sv);
		1351
		1352	if (flags & CC_TRACE)
		1353	{
		1354	if (!coro->cctx)
		1355	coro->cctx = cctx_new ();
		1356	else if (!(coro->cctx->flags & CC_TRACE))
		1357	croak ("cannot enable tracing on coroutine with custom stack");
		1358
		1359	coro->cctx->flags \|= CC_NOREUSE \| (flags & (CC_TRACE \| CC_TRACE_ALL));
		1360	}
		1361	else if (coro->cctx && coro->cctx->flags & CC_TRACE)
		1362	{
		1363	coro->cctx->flags &= ~(CC_TRACE \| CC_TRACE_ALL);
		1364
		1365	if (coro->flags & CF_RUNNING)
		1366	PL_runops = RUNOPS_DEFAULT;
		1367	else
		1368	coro->runops = RUNOPS_DEFAULT;
		1369	}
		1370	}
		1371
1329	MODULE = Coro::State PACKAGE = Coro::State	1372	MODULE = Coro::State PACKAGE = Coro::State PREFIX = api_
1330		1373
1331	PROTOTYPES: DISABLE	1374	PROTOTYPES: DISABLE
1332		1375
1333	BOOT:	1376	BOOT:
1334	{	1377	{
1335	#ifdef USE_ITHREADS	1378	#ifdef USE_ITHREADS
1336	MUTEX_INIT (&coro_mutex);	1379	MUTEX_INIT (&coro_mutex);
1337	#endif	1380	#endif
1338	BOOT_PAGESIZE;	1381	BOOT_PAGESIZE;
		1382
		1383	irsgv = gv_fetchpv ("/" , GV_ADD\|GV_NOTQUAL, SVt_PV);
		1384	stdoutgv = gv_fetchpv ("STDOUT", GV_ADD\|GV_NOTQUAL, SVt_PVIO);
1339		1385
1340	coro_state_stash = gv_stashpv ("Coro::State", TRUE);	1386	coro_state_stash = gv_stashpv ("Coro::State", TRUE);
1341		1387
1342	newCONSTSUB (coro_state_stash, "CC_TRACE" , newSViv (CC_TRACE));	1388	newCONSTSUB (coro_state_stash, "CC_TRACE" , newSViv (CC_TRACE));
1343	newCONSTSUB (coro_state_stash, "CC_TRACE_SUB" , newSViv (CC_TRACE_SUB));	1389	newCONSTSUB (coro_state_stash, "CC_TRACE_SUB" , newSViv (CC_TRACE_SUB));
1344	newCONSTSUB (coro_state_stash, "CC_TRACE_LINE", newSViv (CC_TRACE_LINE));	1390	newCONSTSUB (coro_state_stash, "CC_TRACE_LINE", newSViv (CC_TRACE_LINE));
1345	newCONSTSUB (coro_state_stash, "CC_TRACE_ALL" , newSViv (CC_TRACE_ALL));	1391	newCONSTSUB (coro_state_stash, "CC_TRACE_ALL" , newSViv (CC_TRACE_ALL));
1346		1392
1347	newCONSTSUB (coro_state_stash, "SAVE_DEFAV", newSViv (CORO_SAVE_DEFAV));
1348	newCONSTSUB (coro_state_stash, "SAVE_DEFSV", newSViv (CORO_SAVE_DEFSV));
1349	newCONSTSUB (coro_state_stash, "SAVE_ERRSV", newSViv (CORO_SAVE_ERRSV));
1350	newCONSTSUB (coro_state_stash, "SAVE_IRSSV", newSViv (CORO_SAVE_IRSSV));
1351	newCONSTSUB (coro_state_stash, "SAVE_DEFFH", newSViv (CORO_SAVE_DEFFH));
1352	newCONSTSUB (coro_state_stash, "SAVE_DEF", newSViv (CORO_SAVE_DEF));
1353	newCONSTSUB (coro_state_stash, "SAVE_ALL", newSViv (CORO_SAVE_ALL));
1354
1355	main_mainstack = PL_mainstack;	1393	main_mainstack = PL_mainstack;
1356	main_top_env = PL_top_env;	1394	main_top_env = PL_top_env;
1357		1395
1358	while (main_top_env->je_prev)	1396	while (main_top_env->je_prev)
1359	main_top_env = main_top_env->je_prev;	1397	main_top_env = main_top_env->je_prev;
…		…
1371	struct coro *coro;	1409	struct coro *coro;
1372	HV *hv;	1410	HV *hv;
1373	int i;	1411	int i;
1374		1412
1375	Newz (0, coro, 1, struct coro);	1413	Newz (0, coro, 1, struct coro);
1376	coro->args = newAV ();	1414	coro->args = newAV ();
1377	coro->save = CORO_SAVE_DEF;
1378	coro->flags = CF_NEW;	1415	coro->flags = CF_NEW;
1379		1416
1380	if (coro_first) coro_first->prev = coro;	1417	if (coro_first) coro_first->prev = coro;
1381	coro->next = coro_first;	1418	coro->next = coro_first;
1382	coro_first = coro;	1419	coro_first = coro;
…		…
1389	av_push (coro->args, newSVsv (ST (i)));	1426	av_push (coro->args, newSVsv (ST (i)));
1390	}	1427	}
1391	OUTPUT:	1428	OUTPUT:
1392	RETVAL	1429	RETVAL
1393		1430
1394	int	1431	# these not obviously related functions are all rolled into the same xs
1395	save (SV *coro, int new_save = -1)	1432	# function to increase chances that they all will call transfer with the same
1396	CODE:	1433	# stack offset
1397	RETVAL = api_save (coro, new_save);
1398	OUTPUT:
1399	RETVAL
1400
1401	int
1402	save_also (SV *coro_sv, int save_also)
1403	CODE:
1404	{
1405	struct coro *coro = SvSTATE (coro_sv);
1406	RETVAL = coro->save;
1407	coro->save \|= save_also;
1408	}
1409	OUTPUT:
1410	RETVAL
1411
1412	void	1434	void
1413	_set_stacklevel (...)	1435	_set_stacklevel (...)
1414	ALIAS:	1436	ALIAS:
1415	Coro::State::transfer = 1	1437	Coro::State::transfer = 1
1416	Coro::schedule = 2	1438	Coro::schedule = 2
…		…
1511	{	1533	{
1512	if (coro->mainstack)	1534	if (coro->mainstack)
1513	{	1535	{
1514	struct coro temp;	1536	struct coro temp;
1515	Zero (&temp, 1, struct coro);	1537	Zero (&temp, 1, struct coro);
1516	temp.save = CORO_SAVE_ALL;
1517		1538
1518	if (!(coro->flags & CF_RUNNING))	1539	if (!(coro->flags & CF_RUNNING))
1519	{	1540	{
1520	save_perl (aTHX_ &temp);	1541	save_perl (aTHX_ &temp);
1521	load_perl (aTHX_ coro);	1542	load_perl (aTHX_ coro);
…		…
1542	save_perl (aTHX_ coro);	1563	save_perl (aTHX_ coro);
1543	load_perl (aTHX_ &temp);	1564	load_perl (aTHX_ &temp);
1544	}	1565	}
1545	}	1566	}
1546	}	1567	}
1547		1568
1548	SV *	1569	SV *
1549	is_ready (Coro::State coro)	1570	is_ready (Coro::State coro)
1550	PROTOTYPE: $	1571	PROTOTYPE: $
1551	ALIAS:	1572	ALIAS:
1552	is_ready = CF_READY	1573	is_ready = CF_READY
…		…
1557	RETVAL = boolSV (coro->flags & ix);	1578	RETVAL = boolSV (coro->flags & ix);
1558	OUTPUT:	1579	OUTPUT:
1559	RETVAL	1580	RETVAL
1560		1581
1561	void	1582	void
1562	trace (Coro::State coro, int flags = CC_TRACE \| CC_TRACE_SUB)	1583	api_trace (SV *coro, int flags = CC_TRACE \| CC_TRACE_SUB)
1563	CODE:
1564	if (flags & CC_TRACE)
1565	{
1566	if (!coro->cctx)
1567	coro->cctx = cctx_new ();
1568	else if (!(coro->cctx->flags & CC_TRACE))
1569	croak ("cannot enable tracing on coroutine with custom stack");
1570
1571	coro->cctx->flags \|= flags & (CC_TRACE \| CC_TRACE_ALL);
1572	}
1573	else
1574	if (coro->cctx && coro->cctx->flags & CC_TRACE)
1575	{
1576	coro->cctx->flags &= ~(CC_TRACE \| CC_TRACE_ALL);
1577	coro->cctx->flags \|= CC_NOREUSE;
1578
1579	if (coro->flags & CF_RUNNING)
1580	PL_runops = RUNOPS_DEFAULT;
1581	else
1582	coro->runops = RUNOPS_DEFAULT;
1583	}
1584		1584
1585	SV *	1585	SV *
1586	has_stack (Coro::State coro)	1586	has_stack (Coro::State coro)
1587	PROTOTYPE: $	1587	PROTOTYPE: $
1588	CODE:	1588	CODE:
…		…
1599	RETVAL	1599	RETVAL
1600		1600
1601	IV	1601	IV
1602	rss (Coro::State coro)	1602	rss (Coro::State coro)
1603	PROTOTYPE: $	1603	PROTOTYPE: $
		1604	ALIAS:
		1605	usecount = 1
1604	CODE:	1606	CODE:
1605	RETVAL = coro_rss (coro);	1607	switch (ix)
		1608	{
		1609	case 0: RETVAL = coro_rss (aTHX_ coro); break;
		1610	case 1: RETVAL = coro->usecount; break;
		1611	}
1606	OUTPUT:	1612	OUTPUT:
1607	RETVAL	1613	RETVAL
1608		1614
1609		1615
1610	MODULE = Coro::State PACKAGE = Coro	1616	MODULE = Coro::State PACKAGE = Coro
…		…
1615		1621
1616	sv_pool_rss = get_sv ("Coro::POOL_RSS" , TRUE);	1622	sv_pool_rss = get_sv ("Coro::POOL_RSS" , TRUE);
1617	sv_pool_size = get_sv ("Coro::POOL_SIZE" , TRUE);	1623	sv_pool_size = get_sv ("Coro::POOL_SIZE" , TRUE);
1618	av_async_pool = get_av ("Coro::async_pool", TRUE);	1624	av_async_pool = get_av ("Coro::async_pool", TRUE);
1619		1625
1620	coro_current = get_sv ("Coro::current", FALSE);	1626	coro_current = get_sv ("Coro::current", FALSE);
1621	SvREADONLY_on (coro_current);	1627	SvREADONLY_on (coro_current);
1622		1628
1623	coro_stash = gv_stashpv ("Coro", TRUE);	1629	coro_stash = gv_stashpv ("Coro", TRUE);
1624		1630
1625	newCONSTSUB (coro_stash, "PRIO_MAX", newSViv (PRIO_MAX));	1631	newCONSTSUB (coro_stash, "PRIO_MAX", newSViv (PRIO_MAX));
…		…
1634		1640
1635	{	1641	{
1636	SV *sv = perl_get_sv("Coro::API", 1);	1642	SV *sv = perl_get_sv("Coro::API", 1);
1637		1643
1638	coroapi.schedule = api_schedule;	1644	coroapi.schedule = api_schedule;
1639	coroapi.save = api_save;
1640	coroapi.cede = api_cede;	1645	coroapi.cede = api_cede;
1641	coroapi.cede_notself = api_cede_notself;	1646	coroapi.cede_notself = api_cede_notself;
1642	coroapi.ready = api_ready;	1647	coroapi.ready = api_ready;
1643	coroapi.is_ready = api_is_ready;	1648	coroapi.is_ready = api_is_ready;
1644	coroapi.nready = &coro_nready;	1649	coroapi.nready = &coro_nready;
…		…
1698	# for async_pool speedup	1703	# for async_pool speedup
1699	void	1704	void
1700	_pool_1 (SV *cb)	1705	_pool_1 (SV *cb)
1701	CODE:	1706	CODE:
1702	{	1707	{
1703	int i, len;	1708	struct coro *coro = SvSTATE (coro_current);
1704	HV hv = (HV )SvRV (coro_current);	1709	HV hv = (HV )SvRV (coro_current);
1705	AV *defav = GvAV (PL_defgv);	1710	AV *defav = GvAV (PL_defgv);
1706	SV *invoke = hv_delete (hv, "_invoke", sizeof ("_invoke") - 1, 0);	1711	SV *invoke = hv_delete (hv, "_invoke", sizeof ("_invoke") - 1, 0);
1707	AV *invoke_av;	1712	AV *invoke_av;
		1713	int i, len;
1708		1714
1709	if (!invoke)	1715	if (!invoke)
1710	croak ("\3terminate\2\n");	1716	croak ("\3terminate\2\n");
		1717
		1718	SvREFCNT_dec (coro->saved_deffh);
		1719	coro->saved_deffh = SvREFCNT_inc ((SV *)PL_defoutgv);
1711		1720
1712	hv_store (hv, "desc", sizeof ("desc") - 1,	1721	hv_store (hv, "desc", sizeof ("desc") - 1,
1713	newSVpvn ("[async_pool]", sizeof ("[async_pool]") - 1), 0);	1722	newSVpvn ("[async_pool]", sizeof ("[async_pool]") - 1), 0);
1714		1723
1715	invoke_av = (AV *)SvRV (invoke);	1724	invoke_av = (AV *)SvRV (invoke);
…		…
1733	{	1742	{
1734	struct coro *coro = SvSTATE (coro_current);	1743	struct coro *coro = SvSTATE (coro_current);
1735		1744
1736	sv_setsv (cb, &PL_sv_undef);	1745	sv_setsv (cb, &PL_sv_undef);
1737		1746
		1747	SvREFCNT_dec ((SV )PL_defoutgv); PL_defoutgv = (GV )coro->saved_deffh;
		1748	coro->saved_deffh = 0;
		1749
1738	if (coro_rss (coro) > SvIV (sv_pool_rss)	1750	if (coro_rss (aTHX_ coro) > SvIV (sv_pool_rss)
1739	\|\| av_len (av_async_pool) + 1 >= SvIV (sv_pool_size))	1751	\|\| av_len (av_async_pool) + 1 >= SvIV (sv_pool_size))
1740	croak ("\3terminate\2\n");	1752	croak ("\3terminate\2\n");
1741		1753
1742	av_clear (GvAV (PL_defgv));	1754	av_clear (GvAV (PL_defgv));
1743	hv_store ((HV *)SvRV (coro_current), "desc", sizeof ("desc") - 1,	1755	hv_store ((HV *)SvRV (coro_current), "desc", sizeof ("desc") - 1,
1744	newSVpvn ("[async_pool idle]", sizeof ("[async_pool idle]") - 1), 0);	1756	newSVpvn ("[async_pool idle]", sizeof ("[async_pool idle]") - 1), 0);
1745	coro->save = CORO_SAVE_DEF;	1757
1746	coro->prio = 0;	1758	coro->prio = 0;
		1759
		1760	if (coro->cctx && (coro->cctx->flags & CC_TRACE))
		1761	api_trace (coro_current, 0);
		1762
1747	av_push (av_async_pool, newSVsv (coro_current));	1763	av_push (av_async_pool, newSVsv (coro_current));
1748	}	1764	}
1749		1765
1750		1766
1751	MODULE = Coro::State PACKAGE = Coro::AIO	1767	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.169 by root, Thu Sep 27 16:25:10 2007 UTC vs. Revision 1.188 by root, Fri Oct 5 20:36:31 2007 UTC

Diff Legend

Comparing Coro/Coro/State.xs (file contents):
Revision 1.169 by root, Thu Sep 27 16:25:10 2007 UTC vs.
Revision 1.188 by root, Fri Oct 5 20:36:31 2007 UTC