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

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

Diff Legend

Comparing Coro/Coro/State.xs (file contents):
Revision 1.165 by root, Wed Sep 26 19:26:48 2007 UTC vs.
Revision 1.191 by root, Fri Oct 5 20:47:49 2007 UTC