ViewVC Help

View File | Revision Log | Show Annotations | Download File

/cvs/Coro/Coro/State.xs

Comparing Coro/Coro/State.xs (file contents):
Revision 1.250 by root, Wed Oct 22 16:34:07 2008 UTC vs.
Revision 1.280 by root, Sun Nov 16 09:43:18 2008 UTC

…		…
46	# define BOOT_PAGESIZE (void)0	46	# define BOOT_PAGESIZE (void)0
47	#endif	47	#endif
48		48
49	#if CORO_USE_VALGRIND	49	#if CORO_USE_VALGRIND
50	# include <valgrind/valgrind.h>	50	# include <valgrind/valgrind.h>
51	# define REGISTER_STACK(cctx,start,end) (cctx)->valgrind_id = VALGRIND_STACK_REGISTER ((start), (end))
52	#else
53	# define REGISTER_STACK(cctx,start,end)
54	#endif	51	#endif
55		52
56	/* the maximum number of idle cctx that will be pooled */	53	/* the maximum number of idle cctx that will be pooled */
57	#define MAX_IDLE_CCTX 8	54	static int cctx_max_idle = 4;
58		55
59	#define PERL_VERSION_ATLEAST(a,b,c) \	56	#define PERL_VERSION_ATLEAST(a,b,c) \
60	(PERL_REVISION > (a) \	57	(PERL_REVISION > (a) \
61	|| (PERL_REVISION == (a) \	58	|| (PERL_REVISION == (a) \
62	&& (PERL_VERSION > (b) \	59	&& (PERL_VERSION > (b) \
…		…
81	# ifndef IS_PADCONST	78	# ifndef IS_PADCONST
82	# define IS_PADCONST(v) 0	79	# define IS_PADCONST(v) 0
83	# endif	80	# endif
84	#endif	81	#endif
85		82
		83	/* 5.11 */
		84	#ifndef CxHASARGS
		85	# define CxHASARGS(cx) (cx)->blk_sub.hasargs
		86	#endif
		87
		88	/* 5.10.0 */
		89	#ifndef SvREFCNT_inc_NN
		90	# define SvREFCNT_inc_NN(sv) SvREFCNT_inc (sv)
		91	#endif
		92
86	/* 5.8.8 */	93	/* 5.8.8 */
87	#ifndef GV_NOTQUAL	94	#ifndef GV_NOTQUAL
88	# define GV_NOTQUAL 0	95	# define GV_NOTQUAL 0
89	#endif	96	#endif
90	#ifndef newSV	97	#ifndef newSV
91	# define newSV(l) NEWSV(0,l)	98	# define newSV(l) NEWSV(0,l)
92	#endif	99	#endif
93	#ifndef SvREFCNT_inc_NN
94	# define SvREFCNT_inc_NN(sv) SvREFCNT_inc (sv)
95	#endif
96
97	/* 5.11 */
98	#ifndef CxHASARGS
99	# define CxHASARGS(cx) (cx)->blk_sub.hasargs
100	#endif
101		100
102	/* 5.8.7 */	101	/* 5.8.7 */
103	#ifndef SvRV_set	102	#ifndef SvRV_set
104	# define SvRV_set(s,v) SvRV(s) = (v)	103	# define SvRV_set(s,v) SvRV(s) = (v)
105	#endif	104	#endif
…		…
117	# define CORO_PREFER_PERL_FUNCTIONS 0	116	# define CORO_PREFER_PERL_FUNCTIONS 0
118	#endif	117	#endif
119		118
120	/* The next macros try to return the current stack pointer, in an as	119	/* The next macros try to return the current stack pointer, in an as
121	* portable way as possible. */	120	* portable way as possible. */
122	#define dSTACKLEVEL volatile char stacklevel	121	#if __GNUC__ >= 4
123	#define STACKLEVEL ((void *)&stacklevel)	122	# define dSTACKLEVEL void *stacklevel = __builtin_frame_address (0)
		123	#else
		124	# define dSTACKLEVEL volatile void *stacklevel = (volatile void *)&stacklevel
		125	#endif
124		126
125	#define IN_DESTRUCT (PL_main_cv == Nullcv)	127	#define IN_DESTRUCT (PL_main_cv == Nullcv)
126		128
127	#if __GNUC__ >= 3	129	#if __GNUC__ >= 3
128	# define attribute(x) __attribute__(x)	130	# define attribute(x) __attribute__(x)
129	# define BARRIER __asm__ __volatile__ ("" : : : "memory")
130	# define expect(expr,value) __builtin_expect ((expr),(value))	131	# define expect(expr,value) __builtin_expect ((expr),(value))
		132	# define INLINE static inline
131	#else	133	#else
132	# define attribute(x)	134	# define attribute(x)
133	# define BARRIER
134	# define expect(expr,value) (expr)	135	# define expect(expr,value) (expr)
		136	# define INLINE static
135	#endif	137	#endif
136		138
137	#define expect_false(expr) expect ((expr) != 0, 0)	139	#define expect_false(expr) expect ((expr) != 0, 0)
138	#define expect_true(expr) expect ((expr) != 0, 1)	140	#define expect_true(expr) expect ((expr) != 0, 1)
139		141
140	#define NOINLINE attribute ((noinline))	142	#define NOINLINE attribute ((noinline))
141		143
142	#include "CoroAPI.h"	144	#include "CoroAPI.h"
143		145
144	#ifdef USE_ITHREADS	146	#ifdef USE_ITHREADS
145	static perl_mutex coro_mutex;	147	# if CORO_PTHREAD
146	# define LOCK do { MUTEX_LOCK (&coro_mutex); } while (0)	148	static void *coro_thx;
147	# define UNLOCK do { MUTEX_UNLOCK (&coro_mutex); } while (0)	149	# endif
148	#else
149	# define LOCK (void)0
150	# define UNLOCK (void)0
151	#endif	150	#endif
152		151
153	/* helper storage struct for Coro::AIO */	152	/* helper storage struct for Coro::AIO */
154	struct io_state	153	struct io_state
155	{	154	{
156	AV *res;	155	AV *res;
157	int errorno;	156	int errorno;
158	I32 laststype;	157	I32 laststype; /* U16 in 5.10.0 */
159	int laststatval;	158	int laststatval;
160	Stat_t statcache;	159	Stat_t statcache;
161	};	160	};
162		161
163	static double (*nvtime)(); /* so why doesn't it take void? */	162	static double (*nvtime)(); /* so why doesn't it take void? */
164		163
		164	static U32 cctx_gen;
165	static size_t coro_stacksize = CORO_STACKSIZE;	165	static size_t cctx_stacksize = CORO_STACKSIZE;
166	static struct CoroAPI coroapi;	166	static struct CoroAPI coroapi;
167	static AV *main_mainstack; /* used to differentiate between $main and others */	167	static AV *main_mainstack; /* used to differentiate between $main and others */
168	static JMPENV *main_top_env;	168	static JMPENV *main_top_env;
169	static HV *coro_state_stash, *coro_stash;	169	static HV *coro_state_stash, *coro_stash;
170	static volatile SV *coro_mortal; /* will be freed after next transfer */	170	static volatile SV *coro_mortal; /* will be freed/thrown after next transfer */
		171	static volatile struct coro *transfer_next;
171		172
172	static GV *irsgv; /* $/ */	173	static GV *irsgv; /* $/ */
173	static GV *stdoutgv; /* *STDOUT */	174	static GV *stdoutgv; /* *STDOUT */
174	static SV *rv_diehook;	175	static SV *rv_diehook;
175	static SV *rv_warnhook;	176	static SV *rv_warnhook;
…		…
194	CC_TRACE_LINE = 0x10, /* trace each statement */	195	CC_TRACE_LINE = 0x10, /* trace each statement */
195	CC_TRACE_ALL = CC_TRACE_SUB | CC_TRACE_LINE,	196	CC_TRACE_ALL = CC_TRACE_SUB | CC_TRACE_LINE,
196	};	197	};
197		198
198	/* this is a structure representing a c-level coroutine */	199	/* this is a structure representing a c-level coroutine */
199	typedef struct coro_cctx {	200	typedef struct coro_cctx
		201	{
200	struct coro_cctx *next;	202	struct coro_cctx *next;
201		203
202	/* the stack */	204	/* the stack */
203	void *sptr;	205	void *sptr;
204	size_t ssize;	206	size_t ssize;
…		…
207	void *idle_sp; /* sp of top-level transfer/schedule/cede call */	209	void *idle_sp; /* sp of top-level transfer/schedule/cede call */
208	JMPENV *idle_te; /* same as idle_sp, but for top_env, TODO: remove once stable */	210	JMPENV *idle_te; /* same as idle_sp, but for top_env, TODO: remove once stable */
209	JMPENV *top_env;	211	JMPENV *top_env;
210	coro_context cctx;	212	coro_context cctx;
211		213
		214	U32 gen;
212	#if CORO_USE_VALGRIND	215	#if CORO_USE_VALGRIND
213	int valgrind_id;	216	int valgrind_id;
214	#endif	217	#endif
215	unsigned char flags;	218	unsigned char flags;
216	} coro_cctx;	219	} coro_cctx;
…		…
221	CF_NEW = 0x0004, /* has never been switched to */	224	CF_NEW = 0x0004, /* has never been switched to */
222	CF_DESTROYED = 0x0008, /* coroutine data has been freed */	225	CF_DESTROYED = 0x0008, /* coroutine data has been freed */
223	};	226	};
224		227
225	/* the structure where most of the perl state is stored, overlaid on the cxstack */	228	/* the structure where most of the perl state is stored, overlaid on the cxstack */
226	typedef struct {	229	typedef struct
		230	{
227	SV *defsv;	231	SV *defsv;
228	AV *defav;	232	AV *defav;
229	SV *errsv;	233	SV *errsv;
230	SV *irsgv;	234	SV *irsgv;
231	#define VAR(name,type) type name;	235	#define VAR(name,type) type name;
…		…
235		239
236	#define SLOT_COUNT ((sizeof (perl_slots) + sizeof (PERL_CONTEXT) - 1) / sizeof (PERL_CONTEXT))	240	#define SLOT_COUNT ((sizeof (perl_slots) + sizeof (PERL_CONTEXT) - 1) / sizeof (PERL_CONTEXT))
237		241
238	/* this is a structure representing a perl-level coroutine */	242	/* this is a structure representing a perl-level coroutine */
239	struct coro {	243	struct coro {
240	/* the c coroutine allocated to this perl coroutine, if any */	244	/* the C coroutine allocated to this perl coroutine, if any */
241	coro_cctx *cctx;	245	coro_cctx *cctx;
242		246
243	/* process data */	247	/* process data */
		248	struct CoroSLF slf_frame; /* saved slf frame */
244	AV *mainstack;	249	AV *mainstack;
245	perl_slots *slot; /* basically the saved sp */	250	perl_slots *slot; /* basically the saved sp */
246		251
247	AV *args; /* data associated with this coroutine (initial args) */	252	AV *args; /* data associated with this coroutine (initial args) */
248	int refcnt; /* coroutines are refcounted, yes */	253	int refcnt; /* coroutines are refcounted, yes */
249	int flags; /* CF_ flags */	254	int flags; /* CF_ flags */
250	HV *hv; /* the perl hash associated with this coro, if any */	255	HV *hv; /* the perl hash associated with this coro, if any */
		256	void (*on_destroy)(pTHX_ struct coro *coro);
251		257
252	/* statistics */	258	/* statistics */
253	int usecount; /* number of transfers to this coro */	259	int usecount; /* number of transfers to this coro */
254		260
255	/* coro process data */	261	/* coro process data */
…		…
263	struct coro *next, *prev;	269	struct coro *next, *prev;
264	};	270	};
265		271
266	typedef struct coro *Coro__State;	272	typedef struct coro *Coro__State;
267	typedef struct coro *Coro__State_or_hashref;	273	typedef struct coro *Coro__State_or_hashref;
		274
		275	static struct CoroSLF slf_frame; /* the current slf frame */
268		276
269	/** Coro ********************************************************************/	277	/** Coro ********************************************************************/
270		278
271	#define PRIO_MAX 3	279	#define PRIO_MAX 3
272	#define PRIO_HIGH 1	280	#define PRIO_HIGH 1
…		…
276	#define PRIO_MIN -4	284	#define PRIO_MIN -4
277		285
278	/* for Coro.pm */	286	/* for Coro.pm */
279	static SV *coro_current;	287	static SV *coro_current;
280	static SV *coro_readyhook;	288	static SV *coro_readyhook;
281	static AV *coro_ready [PRIO_MAX-PRIO_MIN+1];	289	static AV *coro_ready [PRIO_MAX - PRIO_MIN + 1];
282	static int coro_nready;
283	static struct coro *coro_first;	290	static struct coro *coro_first;
		291	#define coro_nready coroapi.nready
284		292
285	/** lowlevel stuff **********************************************************/	293	/** lowlevel stuff **********************************************************/
286		294
287	static SV *	295	static SV *
288	coro_get_sv (pTHX_ const char *name, int create)	296	coro_get_sv (pTHX_ const char *name, int create)
…		…
381	static MGVTBL coro_cv_vtbl = {	389	static MGVTBL coro_cv_vtbl = {
382	0, 0, 0, 0,	390	0, 0, 0, 0,
383	coro_cv_free	391	coro_cv_free
384	};	392	};
385		393
386	#define CORO_MAGIC(sv,type) \	394	#define CORO_MAGIC(sv, type) \
387	SvMAGIC (sv) \	395	SvMAGIC (sv) \
388	? SvMAGIC (sv)->mg_type == type \	396	? SvMAGIC (sv)->mg_type == type \
389	? SvMAGIC (sv) \	397	? SvMAGIC (sv) \
390	: mg_find (sv, type) \	398	: mg_find (sv, type) \
391	: 0	399	: 0
392		400
393	#define CORO_MAGIC_cv(cv) CORO_MAGIC (((SV *)(cv)), CORO_MAGIC_type_cv)	401	#define CORO_MAGIC_cv(cv) CORO_MAGIC (((SV *)(cv)), CORO_MAGIC_type_cv)
394	#define CORO_MAGIC_state(sv) CORO_MAGIC (((SV *)(sv)), CORO_MAGIC_type_state)	402	#define CORO_MAGIC_state(sv) CORO_MAGIC (((SV *)(sv)), CORO_MAGIC_type_state)
395		403
396	static struct coro *	404	INLINE struct coro *
397	SvSTATE_ (pTHX_ SV *coro)	405	SvSTATE_ (pTHX_ SV *coro)
398	{	406	{
399	HV *stash;	407	HV *stash;
400	MAGIC *mg;	408	MAGIC *mg;
401		409
…		…
492	CvPADLIST (cv) = (AV *)POPs;	500	CvPADLIST (cv) = (AV *)POPs;
493	}	501	}
494		502
495	PUTBACK;	503	PUTBACK;
496	}	504	}
		505
		506	slf_frame = c->slf_frame;
497	}	507	}
498		508
499	static void	509	static void
500	save_perl (pTHX_ Coro__State c)	510	save_perl (pTHX_ Coro__State c)
501	{	511	{
		512	c->slf_frame = slf_frame;
		513
502	{	514	{
503	dSP;	515	dSP;
504	I32 cxix = cxstack_ix;	516	I32 cxix = cxstack_ix;
505	PERL_CONTEXT *ccstk = cxstack;	517	PERL_CONTEXT *ccstk = cxstack;
506	PERL_SI *top_si = PL_curstackinfo;	518	PERL_SI *top_si = PL_curstackinfo;
…		…
573	#undef VAR	585	#undef VAR
574	}	586	}
575	}	587	}
576		588
577	/*	589	/*
578	* allocate various perl stacks. This is an exact copy	590	* allocate various perl stacks. This is almost an exact copy
579	* of perl.c:init_stacks, except that it uses less memory	591	* of perl.c:init_stacks, except that it uses less memory
580	* on the (sometimes correct) assumption that coroutines do	592	* on the (sometimes correct) assumption that coroutines do
581	* not usually need a lot of stackspace.	593	* not usually need a lot of stackspace.
582	*/	594	*/
583	#if CORO_PREFER_PERL_FUNCTIONS	595	#if CORO_PREFER_PERL_FUNCTIONS
…		…
626		638
627	/*	639	/*
628	* destroy the stacks, the callchain etc...	640	* destroy the stacks, the callchain etc...
629	*/	641	*/
630	static void	642	static void
631	coro_destroy_stacks (pTHX)	643	coro_destruct_stacks (pTHX)
632	{	644	{
633	while (PL_curstackinfo->si_next)	645	while (PL_curstackinfo->si_next)
634	PL_curstackinfo = PL_curstackinfo->si_next;	646	PL_curstackinfo = PL_curstackinfo->si_next;
635		647
636	while (PL_curstackinfo)	648	while (PL_curstackinfo)
…		…
785		797
786	return orig_sigelem_set ? orig_sigelem_set (aTHX_ sv, mg) : 0;	798	return orig_sigelem_set ? orig_sigelem_set (aTHX_ sv, mg) : 0;
787	}	799	}
788		800
789	static void	801	static void
		802	prepare_nop (pTHX_ struct coro_transfer_args *ta)
		803	{
		804	/* kind of mega-hacky, but works */
		805	ta->next = ta->prev = (struct coro *)ta;
		806	}
		807
		808	static int
		809	slf_check_nop (pTHX_ struct CoroSLF *frame)
		810	{
		811	return 0;
		812	}
		813
		814	static void
790	coro_setup (pTHX_ struct coro *coro)	815	coro_setup (pTHX_ struct coro *coro)
791	{	816	{
792	/*	817	/*
793	* emulate part of the perl startup here.	818	* emulate part of the perl startup here.
794	*/	819	*/
…		…
818	PL_rs = newSVsv (GvSV (irsgv));	843	PL_rs = newSVsv (GvSV (irsgv));
819	PL_defoutgv = (GV *)SvREFCNT_inc_NN (stdoutgv);	844	PL_defoutgv = (GV *)SvREFCNT_inc_NN (stdoutgv);
820		845
821	{	846	{
822	dSP;	847	dSP;
823	LOGOP myop;	848	UNOP myop;
824		849
825	Zero (&myop, 1, LOGOP);	850	Zero (&myop, 1, UNOP);
826	myop.op_next = Nullop;	851	myop.op_next = Nullop;
827	myop.op_flags = OPf_WANT_VOID;	852	myop.op_flags = OPf_WANT_VOID;
828		853
829	PUSHMARK (SP);	854	PUSHMARK (SP);
830	XPUSHs (sv_2mortal (av_shift (GvAV (PL_defgv))));	855	XPUSHs (sv_2mortal (av_shift (GvAV (PL_defgv))));
…		…
833	PL_op = PL_ppaddr[OP_ENTERSUB](aTHX);	858	PL_op = PL_ppaddr[OP_ENTERSUB](aTHX);
834	SPAGAIN;	859	SPAGAIN;
835	}	860	}
836		861
837	/* this newly created coroutine might be run on an existing cctx which most	862	/* this newly created coroutine might be run on an existing cctx which most
838	* likely was suspended in set_stacklevel, called from entersub.	863	* likely was suspended in pp_slf, so we have to emulate entering pp_slf here.
839	* set_stacklevl doesn't do anything on return, but entersub does LEAVE,
840	* so we ENTER here for symmetry
841	*/	864	*/
842	ENTER;	865	slf_frame.prepare = prepare_nop; /* provide a nop function for an eventual pp_slf */
		866	slf_frame.check = slf_check_nop; /* signal pp_slf to not repeat */
843	}	867	}
844		868
845	static void	869	static void
846	coro_destroy (pTHX_ struct coro *coro)	870	coro_destruct (pTHX_ struct coro *coro)
847	{	871	{
848	if (!IN_DESTRUCT)	872	if (!IN_DESTRUCT)
849	{	873	{
850	/* restore all saved variables and stuff */	874	/* restore all saved variables and stuff */
851	LEAVE_SCOPE (0);	875	LEAVE_SCOPE (0);
…		…
873	SvREFCNT_dec (PL_warnhook);	897	SvREFCNT_dec (PL_warnhook);
874		898
875	SvREFCNT_dec (coro->saved_deffh);	899	SvREFCNT_dec (coro->saved_deffh);
876	SvREFCNT_dec (coro->throw);	900	SvREFCNT_dec (coro->throw);
877		901
878	coro_destroy_stacks (aTHX);	902	coro_destruct_stacks (aTHX);
879	}	903	}
880		904
881	static void	905	INLINE void
882	free_coro_mortal (pTHX)	906	free_coro_mortal (pTHX)
883	{	907	{
884	if (expect_true (coro_mortal))	908	if (expect_true (coro_mortal))
885	{	909	{
886	SvREFCNT_dec (coro_mortal);	910	SvREFCNT_dec (coro_mortal);
…		…
1010		1034
1011	TAINT_NOT;	1035	TAINT_NOT;
1012	return 0;	1036	return 0;
1013	}	1037	}
1014		1038
		1039	static void
		1040	prepare_set_stacklevel (struct coro_transfer_args *ta, struct coro_cctx *cctx)
		1041	{
		1042	ta->prev = (struct coro *)cctx;
		1043	ta->next = 0;
		1044	}
		1045
1015	/* inject a fake call to Coro::State::_cctx_init into the execution */	1046	/* inject a fake call to Coro::State::_cctx_init into the execution */
1016	/* _cctx_init should be careful, as it could be called at almost any time */	1047	/* _cctx_init should be careful, as it could be called at almost any time */
1017	/* during execution of a perl program */	1048	/* during execution of a perl program */
		1049	/* also initialises PL_top_env */
1018	static void NOINLINE	1050	static void NOINLINE
1019	cctx_prepare (pTHX_ coro_cctx *cctx)	1051	cctx_prepare (pTHX_ coro_cctx *cctx)
1020	{	1052	{
1021	dSP;	1053	dSP;
1022	LOGOP myop;	1054	UNOP myop;
1023		1055
1024	PL_top_env = &PL_start_env;	1056	PL_top_env = &PL_start_env;
1025		1057
1026	if (cctx->flags & CC_TRACE)	1058	if (cctx->flags & CC_TRACE)
1027	PL_runops = runops_trace;	1059	PL_runops = runops_trace;
1028		1060
1029	Zero (&myop, 1, LOGOP);	1061	Zero (&myop, 1, UNOP);
1030	myop.op_next = PL_op;	1062	myop.op_next = PL_op;
1031	myop.op_flags = OPf_WANT_VOID | OPf_STACKED;	1063	myop.op_flags = OPf_WANT_VOID | OPf_STACKED;
1032		1064
1033	PUSHMARK (SP);	1065	PUSHMARK (SP);
1034	EXTEND (SP, 2);	1066	EXTEND (SP, 2);
1035	PUSHs (sv_2mortal (newSViv (PTR2IV (cctx))));	1067	PUSHs (sv_2mortal (newSViv ((IV)cctx)));
1036	PUSHs ((SV *)get_cv ("Coro::State::_cctx_init", FALSE));	1068	PUSHs ((SV *)get_cv ("Coro::State::_cctx_init", FALSE));
1037	PUTBACK;	1069	PUTBACK;
1038	PL_op = (OP *)&myop;	1070	PL_op = (OP *)&myop;
1039	PL_op = PL_ppaddr[OP_ENTERSUB](aTHX);	1071	PL_op = PL_ppaddr[OP_ENTERSUB](aTHX);
1040	SPAGAIN;	1072	SPAGAIN;
1041	}	1073	}
1042		1074
		1075	/* the tail of transfer: execute stuff we can only do after a transfer */
		1076	INLINE void
		1077	transfer_tail (pTHX)
		1078	{
		1079	struct coro *next = (struct coro *)transfer_next;
		1080	assert (!(transfer_next = 0)); /* just used for the side effect when asserts are enabled */
		1081	assert (("FATAL: next coroutine was zero in transfer_tail (please report)", next));
		1082
		1083	free_coro_mortal (aTHX);
		1084
		1085	if (expect_false (next->throw))
		1086	{
		1087	SV *exception = sv_2mortal (next->throw);
		1088
		1089	next->throw = 0;
		1090	sv_setsv (ERRSV, exception);
		1091	croak (0);
		1092	}
		1093	}
		1094
1043	/*	1095	/*
1044	* this is a _very_ stripped down perl interpreter ;)	1096	* this is a _very_ stripped down perl interpreter ;)
1045	*/	1097	*/
1046	static void	1098	static void
1047	cctx_run (void *arg)	1099	cctx_run (void *arg)
1048	{	1100	{
		1101	#ifdef USE_ITHREADS
		1102	# if CORO_PTHREAD
		1103	PERL_SET_CONTEXT (coro_thx);
		1104	# endif
		1105	#endif
		1106	{
1049	dTHX;	1107	dTHX;
1050		1108
1051	/* cctx_run is the alternative tail of transfer(), so unlock here. */	1109	/* normally we would need to skip the entersub here */
1052	UNLOCK;	1110	/* not doing so will re-execute it, which is exactly what we want */
1053
1054	/* we now skip the entersub that lead to transfer() */
1055	PL_op = PL_op->op_next;	1111	/* PL_nop = PL_nop->op_next */
1056		1112
1057	/* inject a fake subroutine call to cctx_init */	1113	/* inject a fake subroutine call to cctx_init */
1058	cctx_prepare (aTHX_ (coro_cctx *)arg);	1114	cctx_prepare (aTHX_ (coro_cctx *)arg);
1059		1115
		1116	/* cctx_run is the alternative tail of transfer() */
		1117	/* TODO: throwing an exception here might be deadly, VERIFY */
		1118	transfer_tail (aTHX);
		1119
1060	/* somebody or something will hit me for both perl_run and PL_restartop */	1120	/* somebody or something will hit me for both perl_run and PL_restartop */
1061	PL_restartop = PL_op;	1121	PL_restartop = PL_op;
1062	perl_run (PL_curinterp);	1122	perl_run (PL_curinterp);
1063		1123
1064	/*	1124	/*
1065	* If perl-run returns we assume exit() was being called or the coro	1125	* If perl-run returns we assume exit() was being called or the coro
1066	* fell off the end, which seems to be the only valid (non-bug)	1126	* fell off the end, which seems to be the only valid (non-bug)
1067	* reason for perl_run to return. We try to exit by jumping to the	1127	* reason for perl_run to return. We try to exit by jumping to the
1068	* bootstrap-time "top" top_env, as we cannot restore the "main"	1128	* bootstrap-time "top" top_env, as we cannot restore the "main"
1069	* coroutine as Coro has no such concept	1129	* coroutine as Coro has no such concept
1070	*/	1130	*/
1071	PL_top_env = main_top_env;	1131	PL_top_env = main_top_env;
1072	JMPENV_JUMP (2); /* I do not feel well about the hardcoded 2 at all */	1132	JMPENV_JUMP (2); /* I do not feel well about the hardcoded 2 at all */
		1133	}
1073	}	1134	}
1074		1135
1075	static coro_cctx *	1136	static coro_cctx *
1076	cctx_new ()	1137	cctx_new ()
1077	{	1138	{
1078	coro_cctx *cctx;	1139	coro_cctx *cctx;
		1140
		1141	++cctx_count;
		1142	New (0, cctx, 1, coro_cctx);
		1143
		1144	cctx->gen = cctx_gen;
		1145	cctx->flags = 0;
		1146	cctx->idle_sp = 0; /* can be accessed by transfer between cctx_run and set_stacklevel, on throw */
		1147
		1148	return cctx;
		1149	}
		1150
		1151	/* create a new cctx only suitable as source */
		1152	static coro_cctx *
		1153	cctx_new_empty ()
		1154	{
		1155	coro_cctx *cctx = cctx_new ();
		1156
		1157	cctx->sptr = 0;
		1158	coro_create (&cctx->cctx, 0, 0, 0, 0);
		1159
		1160	return cctx;
		1161	}
		1162
		1163	/* create a new cctx suitable as destination/running a perl interpreter */
		1164	static coro_cctx *
		1165	cctx_new_run ()
		1166	{
		1167	coro_cctx *cctx = cctx_new ();
1079	void *stack_start;	1168	void *stack_start;
1080	size_t stack_size;	1169	size_t stack_size;
1081		1170
1082	++cctx_count;
1083
1084	Newz (0, cctx, 1, coro_cctx);
1085
1086	#if HAVE_MMAP	1171	#if HAVE_MMAP
1087	cctx->ssize = ((coro_stacksize * sizeof (long) + PAGESIZE - 1) / PAGESIZE + CORO_STACKGUARD) * PAGESIZE;	1172	cctx->ssize = ((cctx_stacksize * sizeof (long) + PAGESIZE - 1) / PAGESIZE + CORO_STACKGUARD) * PAGESIZE;
1088	/* mmap supposedly does allocate-on-write for us */	1173	/* mmap supposedly does allocate-on-write for us */
1089	cctx->sptr = mmap (0, cctx->ssize, PROT_EXEC|PROT_READ|PROT_WRITE, MAP_PRIVATE|MAP_ANONYMOUS, 0, 0);	1174	cctx->sptr = mmap (0, cctx->ssize, PROT_EXEC|PROT_READ|PROT_WRITE, MAP_PRIVATE|MAP_ANONYMOUS, 0, 0);
1090		1175
1091	if (cctx->sptr != (void *)-1)	1176	if (cctx->sptr != (void *)-1)
1092	{	1177	{
1093	# if CORO_STACKGUARD	1178	#if CORO_STACKGUARD
1094	mprotect (cctx->sptr, CORO_STACKGUARD * PAGESIZE, PROT_NONE);	1179	mprotect (cctx->sptr, CORO_STACKGUARD * PAGESIZE, PROT_NONE);
1095	# endif	1180	#endif
1096	stack_start = CORO_STACKGUARD * PAGESIZE + (char *)cctx->sptr;	1181	stack_start = (char *)cctx->sptr + CORO_STACKGUARD * PAGESIZE;
1097	stack_size = cctx->ssize - CORO_STACKGUARD * PAGESIZE;	1182	stack_size = cctx->ssize - CORO_STACKGUARD * PAGESIZE;
1098	cctx->flags |= CC_MAPPED;	1183	cctx->flags |= CC_MAPPED;
1099	}	1184	}
1100	else	1185	else
1101	#endif	1186	#endif
1102	{	1187	{
1103	cctx->ssize = coro_stacksize * (long)sizeof (long);	1188	cctx->ssize = cctx_stacksize * (long)sizeof (long);
1104	New (0, cctx->sptr, coro_stacksize, long);	1189	New (0, cctx->sptr, cctx_stacksize, long);
1105		1190
1106	if (!cctx->sptr)	1191	if (!cctx->sptr)
1107	{	1192	{
1108	perror ("FATAL: unable to allocate stack for coroutine");	1193	perror ("FATAL: unable to allocate stack for coroutine, exiting.");
1109	_exit (EXIT_FAILURE);	1194	_exit (EXIT_FAILURE);
1110	}	1195	}
1111		1196
1112	stack_start = cctx->sptr;	1197	stack_start = cctx->sptr;
1113	stack_size = cctx->ssize;	1198	stack_size = cctx->ssize;
1114	}	1199	}
1115		1200
1116	REGISTER_STACK (cctx, (char *)stack_start, (char *)stack_start + stack_size);	1201	#if CORO_USE_VALGRIND
		1202	cctx->valgrind_id = VALGRIND_STACK_REGISTER ((char *)stack_start, (char *)stack_start + stack_size);
		1203	#endif
		1204
1117	coro_create (&cctx->cctx, cctx_run, (void *)cctx, stack_start, stack_size);	1205	coro_create (&cctx->cctx, cctx_run, (void *)cctx, stack_start, stack_size);
1118		1206
1119	return cctx;	1207	return cctx;
1120	}	1208	}
1121		1209
…		…
1124	{	1212	{
1125	if (!cctx)	1213	if (!cctx)
1126	return;	1214	return;
1127		1215
1128	--cctx_count;	1216	--cctx_count;
		1217	coro_destroy (&cctx->cctx);
1129		1218
		1219	/* coro_transfer creates new, empty cctx's */
		1220	if (cctx->sptr)
		1221	{
1130	#if CORO_USE_VALGRIND	1222	#if CORO_USE_VALGRIND
1131	VALGRIND_STACK_DEREGISTER (cctx->valgrind_id);	1223	VALGRIND_STACK_DEREGISTER (cctx->valgrind_id);
1132	#endif	1224	#endif
1133		1225
1134	#if HAVE_MMAP	1226	#if HAVE_MMAP
1135	if (cctx->flags & CC_MAPPED)	1227	if (cctx->flags & CC_MAPPED)
1136	munmap (cctx->sptr, cctx->ssize);	1228	munmap (cctx->sptr, cctx->ssize);
1137	else	1229	else
1138	#endif	1230	#endif
1139	Safefree (cctx->sptr);	1231	Safefree (cctx->sptr);
		1232	}
1140		1233
1141	Safefree (cctx);	1234	Safefree (cctx);
1142	}	1235	}
1143		1236
1144	/* wether this cctx should be destructed */	1237	/* wether this cctx should be destructed */
1145	#define CCTX_EXPIRED(cctx) ((cctx)->ssize < coro_stacksize || ((cctx)->flags & CC_NOREUSE))	1238	#define CCTX_EXPIRED(cctx) ((cctx)->gen != cctx_gen || ((cctx)->flags & CC_NOREUSE))
1146		1239
1147	static coro_cctx *	1240	static coro_cctx *
1148	cctx_get (pTHX)	1241	cctx_get (pTHX)
1149	{	1242	{
1150	while (expect_true (cctx_first))	1243	while (expect_true (cctx_first))
…		…
1157	return cctx;	1250	return cctx;
1158		1251
1159	cctx_destroy (cctx);	1252	cctx_destroy (cctx);
1160	}	1253	}
1161		1254
1162	return cctx_new ();	1255	return cctx_new_run ();
1163	}	1256	}
1164		1257
1165	static void	1258	static void
1166	cctx_put (coro_cctx *cctx)	1259	cctx_put (coro_cctx *cctx)
1167	{	1260	{
		1261	assert (("FATAL: cctx_put called on non-initialised cctx in Coro (please report)", cctx->sptr));
		1262
1168	/* free another cctx if overlimit */	1263	/* free another cctx if overlimit */
1169	if (expect_false (cctx_idle >= MAX_IDLE_CCTX))	1264	if (expect_false (cctx_idle >= cctx_max_idle))
1170	{	1265	{
1171	coro_cctx *first = cctx_first;	1266	coro_cctx *first = cctx_first;
1172	cctx_first = first->next;	1267	cctx_first = first->next;
1173	--cctx_idle;	1268	--cctx_idle;
1174		1269
…		…
1186	transfer_check (pTHX_ struct coro *prev, struct coro *next)	1281	transfer_check (pTHX_ struct coro *prev, struct coro *next)
1187	{	1282	{
1188	if (expect_true (prev != next))	1283	if (expect_true (prev != next))
1189	{	1284	{
1190	if (expect_false (!(prev->flags & (CF_RUNNING | CF_NEW))))	1285	if (expect_false (!(prev->flags & (CF_RUNNING | CF_NEW))))
1191	croak ("Coro::State::transfer called with non-running/new prev Coro::State, but can only transfer from running or new states");	1286	croak ("Coro::State::transfer called with non-running/new prev Coro::State, but can only transfer from running or new states,");
1192		1287
1193	if (expect_false (next->flags & CF_RUNNING))	1288	if (expect_false (next->flags & CF_RUNNING))
1194	croak ("Coro::State::transfer called with running next Coro::State, but can only transfer to inactive states");	1289	croak ("Coro::State::transfer called with running next Coro::State, but can only transfer to inactive states,");
1195		1290
1196	if (expect_false (next->flags & CF_DESTROYED))	1291	if (expect_false (next->flags & CF_DESTROYED))
1197	croak ("Coro::State::transfer called with destroyed next Coro::State, but can only transfer to inactive states");	1292	croak ("Coro::State::transfer called with destroyed next Coro::State, but can only transfer to inactive states,");
1198		1293
1199	#if !PERL_VERSION_ATLEAST (5,10,0)	1294	#if !PERL_VERSION_ATLEAST (5,10,0)
1200	if (expect_false (PL_lex_state != LEX_NOTPARSING))	1295	if (expect_false (PL_lex_state != LEX_NOTPARSING))
1201	croak ("Coro::State::transfer called while parsing, but this is not supported in your perl version");	1296	croak ("Coro::State::transfer called while parsing, but this is not supported in your perl version,");
1202	#endif	1297	#endif
1203	}	1298	}
1204	}	1299	}
1205		1300
1206	/* always use the TRANSFER macro */	1301	/* always use the TRANSFER macro */
1207	static void NOINLINE	1302	static void NOINLINE
1208	transfer (pTHX_ struct coro *prev, struct coro *next, int force_cctx)	1303	transfer (pTHX_ struct coro *prev, struct coro *next, int force_cctx)
1209	{	1304	{
1210	dSTACKLEVEL;	1305	dSTACKLEVEL;
1211	static volatile int has_throw;
1212		1306
1213	/* sometimes transfer is only called to set idle_sp */	1307	/* sometimes transfer is only called to set idle_sp */
1214	if (expect_false (!next))	1308	if (expect_false (!next))
1215	{	1309	{
1216	((coro_cctx *)prev)->idle_sp = STACKLEVEL;	1310	((coro_cctx *)prev)->idle_sp = (void *)stacklevel;
1217	assert (((coro_cctx *)prev)->idle_te = PL_top_env); /* just for the side-effect when asserts are enabled */	1311	assert (((coro_cctx *)prev)->idle_te = PL_top_env); /* just for the side-effect when asserts are enabled */
1218	}	1312	}
1219	else if (expect_true (prev != next))	1313	else if (expect_true (prev != next))
1220	{	1314	{
1221	coro_cctx *prev__cctx;	1315	coro_cctx *prev__cctx;
1222		1316
1223	if (expect_false (prev->flags & CF_NEW))	1317	if (expect_false (prev->flags & CF_NEW))
1224	{	1318	{
1225	/* create a new empty context */	1319	/* create a new empty/source context */
1226	Newz (0, prev->cctx, 1, coro_cctx);	1320	prev->cctx = cctx_new_empty ();
1227	prev->flags &= ~CF_NEW;	1321	prev->flags &= ~CF_NEW;
1228	prev->flags |= CF_RUNNING;	1322	prev->flags |= CF_RUNNING;
1229	}	1323	}
1230		1324
1231	prev->flags &= ~CF_RUNNING;	1325	prev->flags &= ~CF_RUNNING;
1232	next->flags |= CF_RUNNING;	1326	next->flags |= CF_RUNNING;
1233
1234	LOCK;
1235		1327
1236	/* first get rid of the old state */	1328	/* first get rid of the old state */
1237	save_perl (aTHX_ prev);	1329	save_perl (aTHX_ prev);
1238		1330
1239	if (expect_false (next->flags & CF_NEW))	1331	if (expect_false (next->flags & CF_NEW))
…		…
1246	else	1338	else
1247	load_perl (aTHX_ next);	1339	load_perl (aTHX_ next);
1248		1340
1249	prev__cctx = prev->cctx;	1341	prev__cctx = prev->cctx;
1250		1342
1251	/* possibly "free" the cctx */	1343	/* possibly untie and reuse the cctx */
1252	if (expect_true (	1344	if (expect_true (
1253	prev__cctx->idle_sp == STACKLEVEL	1345	prev__cctx->idle_sp == (void *)stacklevel
1254	&& !(prev__cctx->flags & CC_TRACE)	1346	&& !(prev__cctx->flags & CC_TRACE)
1255	&& !force_cctx	1347	&& !force_cctx
1256	))	1348	))
1257	{	1349	{
1258	/* I assume that STACKLEVEL is a stronger indicator than PL_top_env changes */	1350	/* I assume that stacklevel is a stronger indicator than PL_top_env changes */
1259	assert (("ERROR: current top_env must equal previous top_env", PL_top_env == prev__cctx->idle_te));	1351	assert (("FATAL: current top_env must equal previous top_env in Coro (please report)", PL_top_env == prev__cctx->idle_te));
1260		1352
1261	prev->cctx = 0;	1353	prev->cctx = 0;
1262		1354
1263	/* if the cctx is about to be destroyed we need to make sure we won't see it in cctx_get */	1355	/* if the cctx is about to be destroyed we need to make sure we won't see it in cctx_get */
1264	/* without this the next cctx_get might destroy the prev__cctx while still in use */	1356	/* without this the next cctx_get might destroy the prev__cctx while still in use */
…		…
1272	++next->usecount;	1364	++next->usecount;
1273		1365
1274	if (expect_true (!next->cctx))	1366	if (expect_true (!next->cctx))
1275	next->cctx = cctx_get (aTHX);	1367	next->cctx = cctx_get (aTHX);
1276		1368
1277	has_throw = !!next->throw;	1369	assert (("FATAL: transfer_next already nonzero in Coro (please report)", !transfer_next));
		1370	transfer_next = next;
1278		1371
1279	if (expect_false (prev__cctx != next->cctx))	1372	if (expect_false (prev__cctx != next->cctx))
1280	{	1373	{
1281	prev__cctx->top_env = PL_top_env;	1374	prev__cctx->top_env = PL_top_env;
1282	PL_top_env = next->cctx->top_env;	1375	PL_top_env = next->cctx->top_env;
1283	coro_transfer (&prev__cctx->cctx, &next->cctx->cctx);	1376	coro_transfer (&prev__cctx->cctx, &next->cctx->cctx);
1284	}	1377	}
1285		1378
1286	free_coro_mortal (aTHX);	1379	transfer_tail (aTHX);
1287	UNLOCK;
1288
1289	if (expect_false (has_throw))
1290	{
1291	struct coro *coro = SvSTATE (coro_current);
1292
1293	if (coro->throw)
1294	{
1295	SV *exception = coro->throw;
1296	coro->throw = 0;
1297	sv_setsv (ERRSV, exception);
1298	croak (0);
1299	}
1300	}
1301	}	1380	}
1302	}	1381	}
1303
1304	struct transfer_args
1305	{
1306	struct coro *prev, *next;
1307	};
1308		1382
1309	#define TRANSFER(ta, force_cctx) transfer (aTHX_ (ta).prev, (ta).next, (force_cctx))	1383	#define TRANSFER(ta, force_cctx) transfer (aTHX_ (ta).prev, (ta).next, (force_cctx))
1310	#define TRANSFER_CHECK(ta) transfer_check (aTHX_ (ta).prev, (ta).next)	1384	#define TRANSFER_CHECK(ta) transfer_check (aTHX_ (ta).prev, (ta).next)
1311		1385
1312	/** high level stuff ********************************************************/	1386	/** high level stuff ********************************************************/
…		…
1314	static int	1388	static int
1315	coro_state_destroy (pTHX_ struct coro *coro)	1389	coro_state_destroy (pTHX_ struct coro *coro)
1316	{	1390	{
1317	if (coro->flags & CF_DESTROYED)	1391	if (coro->flags & CF_DESTROYED)
1318	return 0;	1392	return 0;
		1393
		1394	if (coro->on_destroy)
		1395	coro->on_destroy (aTHX_ coro);
1319		1396
1320	coro->flags |= CF_DESTROYED;	1397	coro->flags |= CF_DESTROYED;
1321		1398
1322	if (coro->flags & CF_READY)	1399	if (coro->flags & CF_READY)
1323	{	1400	{
1324	/* reduce nready, as destroying a ready coro effectively unreadies it */	1401	/* reduce nready, as destroying a ready coro effectively unreadies it */
1325	/* alternative: look through all ready queues and remove the coro */	1402	/* alternative: look through all ready queues and remove the coro */
1326	LOCK;
1327	--coro_nready;	1403	--coro_nready;
1328	UNLOCK;
1329	}	1404	}
1330	else	1405	else
1331	coro->flags |= CF_READY; /* make sure it is NOT put into the readyqueue */	1406	coro->flags |= CF_READY; /* make sure it is NOT put into the readyqueue */
1332		1407
1333	if (coro->mainstack && coro->mainstack != main_mainstack)	1408	if (coro->mainstack && coro->mainstack != main_mainstack)
1334	{	1409	{
1335	struct coro temp;	1410	struct coro temp;
1336		1411
1337	if (coro->flags & CF_RUNNING)	1412	assert (("FATAL: tried to destroy currently running coroutine (please report)", !(coro->flags & CF_RUNNING)));
1338	croak ("FATAL: tried to destroy currently running coroutine");
1339		1413
1340	save_perl (aTHX_ &temp);	1414	save_perl (aTHX_ &temp);
1341	load_perl (aTHX_ coro);	1415	load_perl (aTHX_ coro);
1342		1416
1343	coro_destroy (aTHX_ coro);	1417	coro_destruct (aTHX_ coro);
1344		1418
1345	load_perl (aTHX_ &temp);	1419	load_perl (aTHX_ &temp);
1346		1420
1347	coro->slot = 0;	1421	coro->slot = 0;
1348	}	1422	}
…		…
1394	# define MGf_DUP 0	1468	# define MGf_DUP 0
1395	#endif	1469	#endif
1396	};	1470	};
1397		1471
1398	static void	1472	static void
1399	prepare_transfer (pTHX_ struct transfer_args *ta, SV *prev_sv, SV *next_sv)	1473	prepare_transfer (pTHX_ struct coro_transfer_args *ta, SV *prev_sv, SV *next_sv)
1400	{	1474	{
1401	ta->prev = SvSTATE (prev_sv);	1475	ta->prev = SvSTATE (prev_sv);
1402	ta->next = SvSTATE (next_sv);	1476	ta->next = SvSTATE (next_sv);
1403	TRANSFER_CHECK (*ta);	1477	TRANSFER_CHECK (*ta);
1404	}	1478	}
1405		1479
1406	static void	1480	static void
1407	api_transfer (SV *prev_sv, SV *next_sv)	1481	api_transfer (pTHX_ SV *prev_sv, SV *next_sv)
1408	{	1482	{
1409	dTHX;
1410	struct transfer_args ta;	1483	struct coro_transfer_args ta;
1411		1484
1412	prepare_transfer (aTHX_ &ta, prev_sv, next_sv);	1485	prepare_transfer (aTHX_ &ta, prev_sv, next_sv);
1413	TRANSFER (ta, 1);	1486	TRANSFER (ta, 1);
1414	}	1487	}
1415		1488
…		…
1432		1505
1433	return 0;	1506	return 0;
1434	}	1507	}
1435		1508
1436	static int	1509	static int
1437	api_ready (SV *coro_sv)	1510	api_ready (pTHX_ SV *coro_sv)
1438	{	1511	{
1439	dTHX;
1440	struct coro *coro;	1512	struct coro *coro;
1441	SV *sv_hook;	1513	SV *sv_hook;
1442	void (*xs_hook)(void);	1514	void (*xs_hook)(void);
1443		1515
1444	if (SvROK (coro_sv))	1516	if (SvROK (coro_sv))
…		…
1449	if (coro->flags & CF_READY)	1521	if (coro->flags & CF_READY)
1450	return 0;	1522	return 0;
1451		1523
1452	coro->flags |= CF_READY;	1524	coro->flags |= CF_READY;
1453		1525
1454	LOCK;
1455
1456	sv_hook = coro_nready ? 0 : coro_readyhook;	1526	sv_hook = coro_nready ? 0 : coro_readyhook;
1457	xs_hook = coro_nready ? 0 : coroapi.readyhook;	1527	xs_hook = coro_nready ? 0 : coroapi.readyhook;
1458		1528
1459	coro_enq (aTHX_ SvREFCNT_inc_NN (coro_sv));	1529	coro_enq (aTHX_ SvREFCNT_inc_NN (coro_sv));
1460	++coro_nready;	1530	++coro_nready;
1461		1531
1462	UNLOCK;
1463
1464	if (sv_hook)	1532	if (sv_hook)
1465	{	1533	{
1466	dSP;	1534	dSP;
1467		1535
1468	ENTER;	1536	ENTER;
…		…
1482		1550
1483	return 1;	1551	return 1;
1484	}	1552	}
1485		1553
1486	static int	1554	static int
1487	api_is_ready (SV *coro_sv)	1555	api_is_ready (pTHX_ SV *coro_sv)
1488	{	1556	{
1489	dTHX;
1490	return !!(SvSTATE (coro_sv)->flags & CF_READY);	1557	return !!(SvSTATE (coro_sv)->flags & CF_READY);
1491	}	1558	}
1492		1559
1493	static void	1560	INLINE void
1494	prepare_schedule (pTHX_ struct transfer_args *ta)	1561	prepare_schedule (pTHX_ struct coro_transfer_args *ta)
1495	{	1562	{
1496	SV *prev_sv, *next_sv;	1563	SV *prev_sv, *next_sv;
1497		1564
1498	for (;;)	1565	for (;;)
1499	{	1566	{
1500	LOCK;
1501	next_sv = coro_deq (aTHX);	1567	next_sv = coro_deq (aTHX);
1502		1568
1503	/* nothing to schedule: call the idle handler */	1569	/* nothing to schedule: call the idle handler */
1504	if (expect_false (!next_sv))	1570	if (expect_false (!next_sv))
1505	{	1571	{
1506	dSP;	1572	dSP;
1507	UNLOCK;
1508		1573
1509	ENTER;	1574	ENTER;
1510	SAVETMPS;	1575	SAVETMPS;
1511		1576
1512	PUSHMARK (SP);	1577	PUSHMARK (SP);
…		…
1522	ta->next = SvSTATE (next_sv);	1587	ta->next = SvSTATE (next_sv);
1523		1588
1524	/* cannot transfer to destroyed coros, skip and look for next */	1589	/* cannot transfer to destroyed coros, skip and look for next */
1525	if (expect_false (ta->next->flags & CF_DESTROYED))	1590	if (expect_false (ta->next->flags & CF_DESTROYED))
1526	{	1591	{
1527	UNLOCK;
1528	SvREFCNT_dec (next_sv);	1592	SvREFCNT_dec (next_sv);
1529	/* coro_nready is already taken care of by destroy */	1593	/* coro_nready has already been taken care of by destroy */
1530	continue;	1594	continue;
1531	}	1595	}
1532		1596
1533	--coro_nready;	1597	--coro_nready;
1534	UNLOCK;
1535	break;	1598	break;
1536	}	1599	}
1537		1600
1538	/* free this only after the transfer */	1601	/* free this only after the transfer */
1539	prev_sv = SvRV (coro_current);	1602	prev_sv = SvRV (coro_current);
1540	ta->prev = SvSTATE (prev_sv);	1603	ta->prev = SvSTATE (prev_sv);
1541	TRANSFER_CHECK (*ta);	1604	TRANSFER_CHECK (*ta);
1542	assert (ta->next->flags & CF_READY);	1605	assert (("FATAL: next coroutine isn't marked as ready in Coro (please report)", ta->next->flags & CF_READY));
1543	ta->next->flags &= ~CF_READY;	1606	ta->next->flags &= ~CF_READY;
1544	SvRV_set (coro_current, next_sv);	1607	SvRV_set (coro_current, next_sv);
1545		1608
1546	LOCK;
1547	free_coro_mortal (aTHX);	1609	free_coro_mortal (aTHX);
1548	coro_mortal = prev_sv;	1610	coro_mortal = prev_sv;
1549	UNLOCK;
1550	}	1611	}
1551		1612
1552	static void	1613	INLINE void
1553	prepare_cede (pTHX_ struct transfer_args *ta)	1614	prepare_cede (pTHX_ struct coro_transfer_args *ta)
1554	{	1615	{
1555	api_ready (coro_current);	1616	api_ready (aTHX_ coro_current);
1556	prepare_schedule (aTHX_ ta);	1617	prepare_schedule (aTHX_ ta);
1557	}	1618	}
1558		1619
		1620	INLINE void
		1621	prepare_cede_notself (pTHX_ struct coro_transfer_args *ta)
		1622	{
		1623	SV *prev = SvRV (coro_current);
		1624
		1625	if (coro_nready)
		1626	{
		1627	prepare_schedule (aTHX_ ta);
		1628	api_ready (aTHX_ prev);
		1629	}
		1630	else
		1631	prepare_nop (aTHX_ ta);
		1632	}
		1633
		1634	static void
		1635	api_schedule (pTHX)
		1636	{
		1637	struct coro_transfer_args ta;
		1638
		1639	prepare_schedule (aTHX_ &ta);
		1640	TRANSFER (ta, 1);
		1641	}
		1642
1559	static int	1643	static int
1560	prepare_cede_notself (pTHX_ struct transfer_args *ta)	1644	api_cede (pTHX)
1561	{	1645	{
1562	if (coro_nready)	1646	struct coro_transfer_args ta;
1563	{	1647
1564	SV *prev = SvRV (coro_current);
1565	prepare_schedule (aTHX_ ta);	1648	prepare_cede (aTHX_ &ta);
1566	api_ready (prev);	1649
		1650	if (expect_true (ta.prev != ta.next))
		1651	{
		1652	TRANSFER (ta, 1);
1567	return 1;	1653	return 1;
1568	}	1654	}
1569	else	1655	else
1570	return 0;	1656	return 0;
1571	}	1657	}
1572		1658
1573	static void
1574	api_schedule (void)
1575	{
1576	dTHX;
1577	struct transfer_args ta;
1578
1579	prepare_schedule (aTHX_ &ta);
1580	TRANSFER (ta, 1);
1581	}
1582
1583	static int	1659	static int
1584	api_cede (void)	1660	api_cede_notself (pTHX)
1585	{	1661	{
1586	dTHX;	1662	if (coro_nready)
		1663	{
1587	struct transfer_args ta;	1664	struct coro_transfer_args ta;
1588		1665
1589	prepare_cede (aTHX_ &ta);	1666	prepare_cede_notself (aTHX_ &ta);
1590
1591	if (expect_true (ta.prev != ta.next))
1592	{
1593	TRANSFER (ta, 1);	1667	TRANSFER (ta, 1);
1594	return 1;	1668	return 1;
1595	}	1669	}
1596	else	1670	else
1597	return 0;	1671	return 0;
1598	}	1672	}
1599		1673
1600	static int	1674	static void
1601	api_cede_notself (void)
1602	{
1603	dTHX;
1604	struct transfer_args ta;
1605
1606	if (prepare_cede_notself (aTHX_ &ta))
1607	{
1608	TRANSFER (ta, 1);
1609	return 1;
1610	}
1611	else
1612	return 0;
1613	}
1614
1615	static void
1616	api_trace (SV *coro_sv, int flags)	1675	api_trace (pTHX_ SV *coro_sv, int flags)
1617	{	1676	{
1618	dTHX;
1619	struct coro *coro = SvSTATE (coro_sv);	1677	struct coro *coro = SvSTATE (coro_sv);
1620		1678
1621	if (flags & CC_TRACE)	1679	if (flags & CC_TRACE)
1622	{	1680	{
1623	if (!coro->cctx)	1681	if (!coro->cctx)
1624	coro->cctx = cctx_new ();	1682	coro->cctx = cctx_new_run ();
1625	else if (!(coro->cctx->flags & CC_TRACE))	1683	else if (!(coro->cctx->flags & CC_TRACE))
1626	croak ("cannot enable tracing on coroutine with custom stack");	1684	croak ("cannot enable tracing on coroutine with custom stack,");
1627		1685
1628	coro->cctx->flags |= CC_NOREUSE | (flags & (CC_TRACE | CC_TRACE_ALL));	1686	coro->cctx->flags |= CC_NOREUSE | (flags & (CC_TRACE | CC_TRACE_ALL));
1629	}	1687	}
1630	else if (coro->cctx && coro->cctx->flags & CC_TRACE)	1688	else if (coro->cctx && coro->cctx->flags & CC_TRACE)
1631	{	1689	{
…		…
1636	else	1694	else
1637	coro->slot->runops = RUNOPS_DEFAULT;	1695	coro->slot->runops = RUNOPS_DEFAULT;
1638	}	1696	}
1639	}	1697	}
1640		1698
1641	static int
1642	coro_gensub_free (pTHX_ SV *sv, MAGIC *mg)
1643	{
1644	AV *padlist;
1645	AV *av = (AV *)mg->mg_obj;
1646
1647	abort ();
1648
1649	return 0;
1650	}
1651
1652	static MGVTBL coro_gensub_vtbl = {
1653	0, 0, 0, 0,
1654	coro_gensub_free
1655	};
1656
1657	/*****************************************************************************/	1699	/*****************************************************************************/
1658	/* PerlIO::cede */	1700	/* PerlIO::cede */
1659		1701
1660	typedef struct	1702	typedef struct
1661	{	1703	{
…		…
1688	PerlIOCede *self = PerlIOSelf (f, PerlIOCede);	1730	PerlIOCede *self = PerlIOSelf (f, PerlIOCede);
1689	double now = nvtime ();	1731	double now = nvtime ();
1690		1732
1691	if (now >= self->next)	1733	if (now >= self->next)
1692	{	1734	{
1693	api_cede ();	1735	api_cede (aTHX);
1694	self->next = now + self->every;	1736	self->next = now + self->every;
1695	}	1737	}
1696		1738
1697	return PerlIOBuf_flush (aTHX_ f);	1739	return PerlIOBuf_flush (aTHX_ f);
1698	}	1740	}
…		…
1727	PerlIOBuf_get_ptr,	1769	PerlIOBuf_get_ptr,
1728	PerlIOBuf_get_cnt,	1770	PerlIOBuf_get_cnt,
1729	PerlIOBuf_set_ptrcnt,	1771	PerlIOBuf_set_ptrcnt,
1730	};	1772	};
1731		1773
		1774	/*****************************************************************************/
		1775
		1776	static const CV *slf_cv; /* for quick consistency check */
		1777
		1778	static UNOP slf_restore; /* restore stack as entersub did, for first-re-run */
		1779	static SV *slf_arg0;
		1780	static SV *slf_arg1;
		1781	static SV *slf_arg2;
		1782
		1783	/* this restores the stack in the case we patched the entersub, to */
		1784	/* recreate the stack frame as perl will on following calls */
		1785	/* since entersub cleared the stack */
		1786	static OP *
		1787	pp_restore (pTHX)
		1788	{
		1789	dSP;
		1790
		1791	PUSHMARK (SP);
		1792
		1793	EXTEND (SP, 3);
		1794	if (slf_arg0) PUSHs (sv_2mortal (slf_arg0));
		1795	if (slf_arg1) PUSHs (sv_2mortal (slf_arg1));
		1796	if (slf_arg2) PUSHs (sv_2mortal (slf_arg2));
		1797	PUSHs ((SV *)CvGV (slf_cv));
		1798
		1799	RETURNOP (slf_restore.op_first);
		1800	}
		1801
		1802	static void
		1803	slf_prepare_set_stacklevel (pTHX_ struct coro_transfer_args *ta)
		1804	{
		1805	prepare_set_stacklevel (ta, (struct coro_cctx *)slf_frame.data);
		1806	}
		1807
		1808	static void
		1809	slf_init_set_stacklevel (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		1810	{
		1811	assert (("FATAL: set_stacklevel needs the coro cctx as sole argument", items == 1));
		1812
		1813	frame->prepare = slf_prepare_set_stacklevel;
		1814	frame->check = slf_check_nop;
		1815	frame->data = (void *)SvIV (arg [0]);
		1816	}
		1817
		1818	static void
		1819	slf_prepare_transfer (pTHX_ struct coro_transfer_args *ta)
		1820	{
		1821	SV **arg = (SV **)slf_frame.data;
		1822
		1823	prepare_transfer (aTHX_ ta, arg [0], arg [1]);
		1824	}
		1825
		1826	static void
		1827	slf_init_transfer (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		1828	{
		1829	if (items != 2)
		1830	croak ("Coro::State::transfer (prev, next) expects two arguments, not %d,", items);
		1831
		1832	frame->prepare = slf_prepare_transfer;
		1833	frame->check = slf_check_nop;
		1834	frame->data = (void *)arg; /* let's hope it will stay valid */
		1835	}
		1836
		1837	static void
		1838	slf_init_schedule (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		1839	{
		1840	frame->prepare = prepare_schedule;
		1841	frame->check = slf_check_nop;
		1842	}
		1843
		1844	static void
		1845	slf_init_cede (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		1846	{
		1847	frame->prepare = prepare_cede;
		1848	frame->check = slf_check_nop;
		1849	}
		1850
		1851	static void
		1852	slf_init_cede_notself (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		1853	{
		1854	frame->prepare = prepare_cede_notself;
		1855	frame->check = slf_check_nop;
		1856	}
		1857
		1858	/* we hijack an hopefully unused CV flag for our purposes */
		1859	#define CVf_SLF 0x4000
		1860
		1861	/*
		1862	* these not obviously related functions are all rolled into one
		1863	* function to increase chances that they all will call transfer with the same
		1864	* stack offset
		1865	* SLF stands for "schedule-like-function".
		1866	*/
		1867	static OP *
		1868	pp_slf (pTHX)
		1869	{
		1870	I32 checkmark; /* mark SP to see how many elements check has pushed */
		1871
		1872	/* set up the slf frame, unless it has already been set-up */
		1873	/* the latter happens when a new coro has been started */
		1874	/* or when a new cctx was attached to an existing coroutine */
		1875	if (expect_true (!slf_frame.prepare))
		1876	{
		1877	/* first iteration */
		1878	dSP;
		1879	SV **arg = PL_stack_base + TOPMARK + 1;
		1880	int items = SP - arg; /* args without function object */
		1881	SV *gv = *sp;
		1882
		1883	/* do a quick consistency check on the "function" object, and if it isn't */
		1884	/* for us, divert to the real entersub */
		1885	if (SvTYPE (gv) != SVt_PVGV || !(CvFLAGS (GvCV (gv)) & CVf_SLF))
		1886	return PL_ppaddr[OP_ENTERSUB](aTHX);
		1887
		1888	/* pop args */
		1889	SP = PL_stack_base + POPMARK;
		1890
		1891	if (!(PL_op->op_flags & OPf_STACKED))
		1892	{
		1893	/* ampersand-form of call, use @_ instead of stack */
		1894	AV *av = GvAV (PL_defgv);
		1895	arg = AvARRAY (av);
		1896	items = AvFILLp (av) + 1;
		1897	}
		1898
		1899	PUTBACK;
		1900
		1901	/* now call the init function, which needs to set up slf_frame */
		1902	((coro_slf_cb)CvXSUBANY (GvCV (gv)).any_ptr)
		1903	(aTHX_ &slf_frame, GvCV (gv), arg, items);
		1904	}
		1905
		1906	/* now that we have a slf_frame, interpret it! */
		1907	/* we use a callback system not to make the code needlessly */
		1908	/* complicated, but so we can run multiple perl coros from one cctx */
		1909
		1910	do
		1911	{
		1912	struct coro_transfer_args ta;
		1913
		1914	slf_frame.prepare (aTHX_ &ta);
		1915	TRANSFER (ta, 0);
		1916
		1917	checkmark = PL_stack_sp - PL_stack_base;
		1918	}
		1919	while (slf_frame.check (aTHX_ &slf_frame));
		1920
		1921	{
		1922	dSP;
		1923	SV **bot = PL_stack_base + checkmark;
		1924	int gimme = GIMME_V;
		1925
		1926	slf_frame.prepare = 0; /* invalidate the frame, so it gets initialised again next time */
		1927
		1928	/* make sure we put something on the stack in scalar context */
		1929	if (gimme == G_SCALAR)
		1930	{
		1931	if (sp == bot)
		1932	XPUSHs (&PL_sv_undef);
		1933
		1934	SP = bot + 1;
		1935	}
		1936
		1937	PUTBACK;
		1938	}
		1939
		1940	return NORMAL;
		1941	}
		1942
		1943	static void
		1944	api_execute_slf (pTHX_ CV *cv, coro_slf_cb init_cb, SV **arg, int items)
		1945	{
		1946	assert (("FATAL: SLF call with illegal CV value", !CvANON (cv)));
		1947
		1948	if (PL_op->op_ppaddr != PL_ppaddr [OP_ENTERSUB]
		1949	&& PL_op->op_ppaddr != pp_slf)
		1950	croak ("FATAL: Coro SLF calls can only be made normally, not via goto or any other means, caught");
		1951
		1952	if (items > 3)
		1953	croak ("Coro only supports up to three arguments to SLF functions currently (not %d), caught", items);
		1954
		1955	CvFLAGS (cv) |= CVf_SLF;
		1956	CvXSUBANY (cv).any_ptr = (void *)init_cb;
		1957	slf_cv = cv;
		1958
		1959	/* we patch the op, and then re-run the whole call */
		1960	/* we have to put the same argument on the stack for this to work */
		1961	/* and this will be done by pp_restore */
		1962	slf_restore.op_next = (OP *)&slf_restore;
		1963	slf_restore.op_type = OP_CUSTOM;
		1964	slf_restore.op_ppaddr = pp_restore;
		1965	slf_restore.op_first = PL_op;
		1966
		1967	slf_arg0 = items > 0 ? SvREFCNT_inc (arg [0]) : 0;
		1968	slf_arg1 = items > 1 ? SvREFCNT_inc (arg [1]) : 0;
		1969	slf_arg2 = items > 2 ? SvREFCNT_inc (arg [2]) : 0;
		1970
		1971	PL_op->op_ppaddr = pp_slf;
		1972
		1973	PL_op = (OP *)&slf_restore;
		1974	}
		1975
		1976	/*****************************************************************************/
		1977
		1978	static void
		1979	coro_semaphore_adjust (AV *av, int adjust)
		1980	{
		1981	SV *count_sv = AvARRAY (av)[0];
		1982	IV count = SvIVX (count_sv);
		1983
		1984	count += adjust;
		1985	SvIVX (count_sv) = count;
		1986
		1987	/* now wake up as many waiters as possible */
		1988	while (count > 0 && AvFILLp (av) >= count)
		1989	{
		1990	SV *cb;
		1991
		1992	/* swap first two elements so we can shift a waiter */
		1993	AvARRAY (av)[0] = AvARRAY (av)[1];
		1994	AvARRAY (av)[1] = count_sv;
		1995	cb = av_shift (av);
		1996
		1997	if (SvOBJECT (cb))
		1998	api_ready (aTHX_ cb);
		1999	else
		2000	croak ("callbacks not yet supported");
		2001
		2002	SvREFCNT_dec (cb);
		2003
		2004	--count;
		2005	}
		2006	}
		2007
		2008	static void
		2009	coro_semaphore_on_destroy (pTHX_ struct coro *coro)
		2010	{
		2011	/* call $sem->adjust (0) to possibly wake up some waiters */
		2012	coro_semaphore_adjust ((AV *)coro->slf_frame.data, 0);
		2013	}
		2014
		2015	static int
		2016	slf_check_semaphore_down (pTHX_ struct CoroSLF *frame)
		2017	{
		2018	AV *av = (AV *)frame->data;
		2019	SV *count_sv = AvARRAY (av)[0];
		2020
		2021	if (SvIVX (count_sv) > 0)
		2022	{
		2023	SvSTATE (coro_current)->on_destroy = 0;
		2024	SvIVX (count_sv) = SvIVX (count_sv) - 1;
		2025	return 0;
		2026	}
		2027	else
		2028	{
		2029	int i;
		2030	/* if we were woken up but can't down, we look through the whole */
		2031	/* waiters list and only add us if we aren't in there already */
		2032	/* this avoids some degenerate memory usage cases */
		2033
		2034	for (i = 1; i <= AvFILLp (av); ++i)
		2035	if (AvARRAY (av)[i] == SvRV (coro_current))
		2036	return 1;
		2037
		2038	av_push (av, SvREFCNT_inc (SvRV (coro_current)));
		2039	return 1;
		2040	}
		2041	}
		2042
		2043	static void
		2044	slf_init_semaphore_down (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		2045	{
		2046	AV *av = (AV *)SvRV (arg [0]);
		2047
		2048	if (SvIVX (AvARRAY (av)[0]) > 0)
		2049	{
		2050	frame->data = (void *)av;
		2051	frame->prepare = prepare_nop;
		2052	}
		2053	else
		2054	{
		2055	av_push (av, SvREFCNT_inc (SvRV (coro_current)));
		2056
		2057	frame->data = (void *)sv_2mortal (SvREFCNT_inc ((SV *)av));
		2058	frame->prepare = prepare_schedule;
		2059
		2060	/* to avoid race conditions when a woken-up coro gets terminated */
		2061	/* we arrange for a temporary on_destroy that calls adjust (0) */
		2062	SvSTATE (coro_current)->on_destroy = coro_semaphore_on_destroy;
		2063	}
		2064
		2065	frame->check = slf_check_semaphore_down;
		2066
		2067	}
		2068
		2069	/*****************************************************************************/
		2070
		2071	#define GENSUB_ARG CvXSUBANY (cv).any_ptr
		2072
		2073	/* create a closure from XS, returns a code reference */
		2074	/* the arg can be accessed via GENSUB_ARG from the callback */
		2075	/* the callback must use dXSARGS/XSRETURN */
		2076	static SV *
		2077	gensub (pTHX_ void (*xsub)(pTHX_ CV *), void *arg)
		2078	{
		2079	CV *cv = (CV *)NEWSV (0, 0);
		2080
		2081	sv_upgrade ((SV *)cv, SVt_PVCV);
		2082
		2083	CvANON_on (cv);
		2084	CvISXSUB_on (cv);
		2085	CvXSUB (cv) = xsub;
		2086	GENSUB_ARG = arg;
		2087
		2088	return newRV_noinc ((SV *)cv);
		2089	}
		2090
		2091	/*****************************************************************************/
1732		2092
1733	MODULE = Coro::State PACKAGE = Coro::State PREFIX = api_	2093	MODULE = Coro::State PACKAGE = Coro::State PREFIX = api_
1734		2094
1735	PROTOTYPES: DISABLE	2095	PROTOTYPES: DISABLE
1736		2096
1737	BOOT:	2097	BOOT:
1738	{	2098	{
1739	#ifdef USE_ITHREADS	2099	#ifdef USE_ITHREADS
1740	MUTEX_INIT (&coro_mutex);	2100	MUTEX_INIT (&coro_lock);
		2101	# if CORO_PTHREAD
		2102	coro_thx = PERL_GET_CONTEXT;
		2103	# endif
1741	#endif	2104	#endif
1742	BOOT_PAGESIZE;	2105	BOOT_PAGESIZE;
1743		2106
1744	irsgv = gv_fetchpv ("/" , GV_ADD|GV_NOTQUAL, SVt_PV);	2107	irsgv = gv_fetchpv ("/" , GV_ADD|GV_NOTQUAL, SVt_PV);
1745	stdoutgv = gv_fetchpv ("STDOUT", GV_ADD|GV_NOTQUAL, SVt_PVIO);	2108	stdoutgv = gv_fetchpv ("STDOUT", GV_ADD|GV_NOTQUAL, SVt_PVIO);
…		…
1763	main_top_env = PL_top_env;	2126	main_top_env = PL_top_env;
1764		2127
1765	while (main_top_env->je_prev)	2128	while (main_top_env->je_prev)
1766	main_top_env = main_top_env->je_prev;	2129	main_top_env = main_top_env->je_prev;
1767		2130
		2131	{
		2132	SV *slf = sv_2mortal (newSViv (PTR2IV (pp_slf)));
		2133
		2134	if (!PL_custom_op_names) PL_custom_op_names = newHV ();
		2135	hv_store_ent (PL_custom_op_names, slf,
		2136	newSVpv ("coro_slf", 0), 0);
		2137
		2138	if (!PL_custom_op_descs) PL_custom_op_descs = newHV ();
		2139	hv_store_ent (PL_custom_op_descs, slf,
		2140	newSVpv ("coro schedule like function", 0), 0);
		2141	}
		2142
1768	coroapi.ver = CORO_API_VERSION;	2143	coroapi.ver = CORO_API_VERSION;
1769	coroapi.rev = CORO_API_REVISION;	2144	coroapi.rev = CORO_API_REVISION;
		2145
1770	coroapi.transfer = api_transfer;	2146	coroapi.transfer = api_transfer;
		2147
		2148	coroapi.sv_state = SvSTATE_;
		2149	coroapi.execute_slf = api_execute_slf;
		2150	coroapi.prepare_nop = prepare_nop;
		2151	coroapi.prepare_schedule = prepare_schedule;
		2152	coroapi.prepare_cede = prepare_cede;
		2153	coroapi.prepare_cede_notself = prepare_cede_notself;
1771		2154
1772	{	2155	{
1773	SV **svp = hv_fetch (PL_modglobal, "Time::NVtime", 12, 0);	2156	SV **svp = hv_fetch (PL_modglobal, "Time::NVtime", 12, 0);
1774		2157
1775	if (!svp) croak ("Time::HiRes is required");	2158	if (!svp) croak ("Time::HiRes is required");
…		…
1808	av_push (coro->args, newSVsv (ST (i)));	2191	av_push (coro->args, newSVsv (ST (i)));
1809	}	2192	}
1810	OUTPUT:	2193	OUTPUT:
1811	RETVAL	2194	RETVAL
1812		2195
1813	# these not obviously related functions are all rolled into the same xs
1814	# function to increase chances that they all will call transfer with the same
1815	# stack offset
1816	void	2196	void
1817	_set_stacklevel (...)	2197	_set_stacklevel (...)
1818	ALIAS:	2198	CODE:
1819	Coro::State::transfer = 1	2199	api_execute_slf (aTHX_ cv, slf_init_set_stacklevel, &ST (0), items);
1820	Coro::schedule = 2
1821	Coro::cede = 3
1822	Coro::cede_notself = 4
1823	CODE:
1824	{
1825	struct transfer_args ta;
1826		2200
1827	PUTBACK;	2201	void
1828	switch (ix)	2202	transfer (...)
1829	{	2203	PROTOTYPE: $$
1830	case 0:	2204	CODE:
1831	ta.prev = (struct coro *)INT2PTR (coro_cctx *, SvIV (ST (0)));	2205	api_execute_slf (aTHX_ cv, slf_init_transfer, &ST (0), items);
1832	ta.next = 0;
1833	break;
1834
1835	case 1:
1836	if (items != 2)
1837	croak ("Coro::State::transfer (prev,next) expects two arguments, not %d", items);
1838
1839	prepare_transfer (aTHX_ &ta, ST (0), ST (1));
1840	break;
1841
1842	case 2:
1843	prepare_schedule (aTHX_ &ta);
1844	break;
1845
1846	case 3:
1847	prepare_cede (aTHX_ &ta);
1848	break;
1849
1850	case 4:
1851	if (!prepare_cede_notself (aTHX_ &ta))
1852	XSRETURN_EMPTY;
1853
1854	break;
1855	}
1856	SPAGAIN;
1857
1858	BARRIER;
1859	PUTBACK;
1860	TRANSFER (ta, 0);
1861	SPAGAIN; /* might be the sp of a different coroutine now */
1862	/* be extra careful not to ever do anything after TRANSFER */
1863	}
1864		2206
1865	bool	2207	bool
1866	_destroy (SV *coro_sv)	2208	_destroy (SV *coro_sv)
1867	CODE:	2209	CODE:
1868	RETVAL = coro_state_destroy (aTHX_ SvSTATE (coro_sv));	2210	RETVAL = coro_state_destroy (aTHX_ SvSTATE (coro_sv));
…		…
1875	CODE:	2217	CODE:
1876	_exit (code);	2218	_exit (code);
1877		2219
1878	int	2220	int
1879	cctx_stacksize (int new_stacksize = 0)	2221	cctx_stacksize (int new_stacksize = 0)
		2222	PROTOTYPE: ;$
1880	CODE:	2223	CODE:
1881	RETVAL = coro_stacksize;	2224	RETVAL = cctx_stacksize;
1882	if (new_stacksize)	2225	if (new_stacksize)
		2226	{
1883	coro_stacksize = new_stacksize;	2227	cctx_stacksize = new_stacksize;
		2228	++cctx_gen;
		2229	}
1884	OUTPUT:	2230	OUTPUT:
1885	RETVAL	2231	RETVAL
1886		2232
1887	int	2233	int
		2234	cctx_max_idle (int max_idle = 0)
		2235	PROTOTYPE: ;$
		2236	CODE:
		2237	RETVAL = cctx_max_idle;
		2238	if (max_idle > 1)
		2239	cctx_max_idle = max_idle;
		2240	OUTPUT:
		2241	RETVAL
		2242
		2243	int
1888	cctx_count ()	2244	cctx_count ()
		2245	PROTOTYPE:
1889	CODE:	2246	CODE:
1890	RETVAL = cctx_count;	2247	RETVAL = cctx_count;
1891	OUTPUT:	2248	OUTPUT:
1892	RETVAL	2249	RETVAL
1893		2250
1894	int	2251	int
1895	cctx_idle ()	2252	cctx_idle ()
		2253	PROTOTYPE:
1896	CODE:	2254	CODE:
1897	RETVAL = cctx_idle;	2255	RETVAL = cctx_idle;
1898	OUTPUT:	2256	OUTPUT:
1899	RETVAL	2257	RETVAL
1900		2258
1901	void	2259	void
1902	list ()	2260	list ()
		2261	PROTOTYPE:
1903	PPCODE:	2262	PPCODE:
1904	{	2263	{
1905	struct coro *coro;	2264	struct coro *coro;
1906	for (coro = coro_first; coro; coro = coro->next)	2265	for (coro = coro_first; coro; coro = coro->next)
1907	if (coro->hv)	2266	if (coro->hv)
…		…
1966	RETVAL = boolSV (coro->flags & ix);	2325	RETVAL = boolSV (coro->flags & ix);
1967	OUTPUT:	2326	OUTPUT:
1968	RETVAL	2327	RETVAL
1969		2328
1970	void	2329	void
		2330	throw (Coro::State self, SV *throw = &PL_sv_undef)
		2331	PROTOTYPE: $;$
		2332	CODE:
		2333	SvREFCNT_dec (self->throw);
		2334	self->throw = SvOK (throw) ? newSVsv (throw) : 0;
		2335
		2336	void
1971	api_trace (SV *coro, int flags = CC_TRACE | CC_TRACE_SUB)	2337	api_trace (SV *coro, int flags = CC_TRACE | CC_TRACE_SUB)
		2338	PROTOTYPE: $;$
		2339	C_ARGS: aTHX_ coro, flags
1972		2340
1973	SV *	2341	SV *
1974	has_cctx (Coro::State coro)	2342	has_cctx (Coro::State coro)
1975	PROTOTYPE: $	2343	PROTOTYPE: $
1976	CODE:	2344	CODE:
…		…
1984	CODE:	2352	CODE:
1985	RETVAL = (coro->cctx ? coro->cctx->flags : 0) & CC_TRACE_ALL;	2353	RETVAL = (coro->cctx ? coro->cctx->flags : 0) & CC_TRACE_ALL;
1986	OUTPUT:	2354	OUTPUT:
1987	RETVAL	2355	RETVAL
1988		2356
1989	IV	2357	UV
1990	rss (Coro::State coro)	2358	rss (Coro::State coro)
1991	PROTOTYPE: $	2359	PROTOTYPE: $
1992	ALIAS:	2360	ALIAS:
1993	usecount = 1	2361	usecount = 1
1994	CODE:	2362	CODE:
…		…
2000	OUTPUT:	2368	OUTPUT:
2001	RETVAL	2369	RETVAL
2002		2370
2003	void	2371	void
2004	force_cctx ()	2372	force_cctx ()
		2373	PROTOTYPE:
2005	CODE:	2374	CODE:
2006	struct coro *coro = SvSTATE (coro_current);	2375	struct coro *coro = SvSTATE (coro_current);
2007	coro->cctx->idle_sp = 0;	2376	coro->cctx->idle_sp = 0;
2008
2009	void
2010	throw (Coro::State self, SV *throw = &PL_sv_undef)
2011	PROTOTYPE: $;$
2012	CODE:
2013	SvREFCNT_dec (self->throw);
2014	self->throw = SvOK (throw) ? newSVsv (throw) : 0;
2015		2377
2016	void	2378	void
2017	swap_defsv (Coro::State self)	2379	swap_defsv (Coro::State self)
2018	PROTOTYPE: $	2380	PROTOTYPE: $
2019	ALIAS:	2381	ALIAS:
2020	swap_defav = 1	2382	swap_defav = 1
2021	CODE:	2383	CODE:
2022	if (!self->slot)	2384	if (!self->slot)
2023	croak ("cannot swap state with coroutine that has no saved state");	2385	croak ("cannot swap state with coroutine that has no saved state,");
2024	else	2386	else
2025	{	2387	{
2026	SV **src = ix ? (SV **)&GvAV (PL_defgv) : &GvSV (PL_defgv);	2388	SV **src = ix ? (SV **)&GvAV (PL_defgv) : &GvSV (PL_defgv);
2027	SV **dst = ix ? (SV **)&self->slot->defav : (SV **)&self->slot->defsv;	2389	SV **dst = ix ? (SV **)&self->slot->defav : (SV **)&self->slot->defsv;
2028		2390
…		…
2053		2415
2054	for (i = PRIO_MAX - PRIO_MIN + 1; i--; )	2416	for (i = PRIO_MAX - PRIO_MIN + 1; i--; )
2055	coro_ready[i] = newAV ();	2417	coro_ready[i] = newAV ();
2056		2418
2057	{	2419	{
2058	SV *sv = perl_get_sv ("Coro::API", TRUE);	2420	SV *sv = coro_get_sv (aTHX_ "Coro::API", TRUE);
2059	perl_get_sv ("Coro::API", TRUE); /* silence 5.10 warning */
2060		2421
2061	coroapi.schedule = api_schedule;	2422	coroapi.schedule = api_schedule;
2062	coroapi.cede = api_cede;	2423	coroapi.cede = api_cede;
2063	coroapi.cede_notself = api_cede_notself;	2424	coroapi.cede_notself = api_cede_notself;
2064	coroapi.ready = api_ready;	2425	coroapi.ready = api_ready;
2065	coroapi.is_ready = api_is_ready;	2426	coroapi.is_ready = api_is_ready;
2066	coroapi.nready = &coro_nready;	2427	coroapi.nready = coro_nready;
2067	coroapi.current = coro_current;	2428	coroapi.current = coro_current;
2068		2429
2069	GCoroAPI = &coroapi;	2430	GCoroAPI = &coroapi;
2070	sv_setiv (sv, (IV)&coroapi);	2431	sv_setiv (sv, (IV)&coroapi);
2071	SvREADONLY_on (sv);	2432	SvREADONLY_on (sv);
2072	}	2433	}
2073	}	2434	}
		2435
		2436	void
		2437	schedule (...)
		2438	CODE:
		2439	api_execute_slf (aTHX_ cv, slf_init_schedule, &ST (0), 0);
		2440
		2441	void
		2442	cede (...)
		2443	CODE:
		2444	api_execute_slf (aTHX_ cv, slf_init_cede, &ST (0), 0);
		2445
		2446	void
		2447	cede_notself (...)
		2448	CODE:
		2449	api_execute_slf (aTHX_ cv, slf_init_cede_notself, &ST (0), 0);
2074		2450
2075	void	2451	void
2076	_set_current (SV *current)	2452	_set_current (SV *current)
2077	PROTOTYPE: $	2453	PROTOTYPE: $
2078	CODE:	2454	CODE:
…		…
2081		2457
2082	void	2458	void
2083	_set_readyhook (SV *hook)	2459	_set_readyhook (SV *hook)
2084	PROTOTYPE: $	2460	PROTOTYPE: $
2085	CODE:	2461	CODE:
2086	LOCK;
2087	SvREFCNT_dec (coro_readyhook);	2462	SvREFCNT_dec (coro_readyhook);
2088	coro_readyhook = SvOK (hook) ? newSVsv (hook) : 0;	2463	coro_readyhook = SvOK (hook) ? newSVsv (hook) : 0;
2089	UNLOCK;
2090		2464
2091	int	2465	int
2092	prio (Coro::State coro, int newprio = 0)	2466	prio (Coro::State coro, int newprio = 0)
		2467	PROTOTYPE: $;$
2093	ALIAS:	2468	ALIAS:
2094	nice = 1	2469	nice = 1
2095	CODE:	2470	CODE:
2096	{	2471	{
2097	RETVAL = coro->prio;	2472	RETVAL = coro->prio;
…		…
2112		2487
2113	SV *	2488	SV *
2114	ready (SV *self)	2489	ready (SV *self)
2115	PROTOTYPE: $	2490	PROTOTYPE: $
2116	CODE:	2491	CODE:
2117	RETVAL = boolSV (api_ready (self));	2492	RETVAL = boolSV (api_ready (aTHX_ self));
2118	OUTPUT:	2493	OUTPUT:
2119	RETVAL	2494	RETVAL
2120		2495
2121	int	2496	int
2122	nready (...)	2497	nready (...)
…		…
2161	{	2536	{
2162	av_fill (defav, len - 1);	2537	av_fill (defav, len - 1);
2163	for (i = 0; i < len; ++i)	2538	for (i = 0; i < len; ++i)
2164	av_store (defav, i, SvREFCNT_inc_NN (AvARRAY (invoke_av)[i + 1]));	2539	av_store (defav, i, SvREFCNT_inc_NN (AvARRAY (invoke_av)[i + 1]));
2165	}	2540	}
2166
2167	SvREFCNT_dec (invoke);
2168	}	2541	}
2169		2542
2170	void	2543	void
2171	_pool_2 (SV *cb)	2544	_pool_2 (SV *cb)
2172	CODE:	2545	CODE:
…		…
2176	sv_setsv (cb, &PL_sv_undef);	2549	sv_setsv (cb, &PL_sv_undef);
2177		2550
2178	SvREFCNT_dec ((SV *)PL_defoutgv); PL_defoutgv = (GV *)coro->saved_deffh;	2551	SvREFCNT_dec ((SV *)PL_defoutgv); PL_defoutgv = (GV *)coro->saved_deffh;
2179	coro->saved_deffh = 0;	2552	coro->saved_deffh = 0;
2180		2553
2181	if (coro_rss (aTHX_ coro) > SvIV (sv_pool_rss)	2554	if (coro_rss (aTHX_ coro) > SvUV (sv_pool_rss)
2182	|| av_len (av_async_pool) + 1 >= SvIV (sv_pool_size))	2555	|| av_len (av_async_pool) + 1 >= SvIV (sv_pool_size))
2183	{	2556	{
2184	SV *old = PL_diehook;	2557	SV *old = PL_diehook;
2185	PL_diehook = 0;	2558	PL_diehook = 0;
2186	SvREFCNT_dec (old);	2559	SvREFCNT_dec (old);
…		…
2192	newSVpvn ("[async_pool idle]", sizeof ("[async_pool idle]") - 1), 0);	2565	newSVpvn ("[async_pool idle]", sizeof ("[async_pool idle]") - 1), 0);
2193		2566
2194	coro->prio = 0;	2567	coro->prio = 0;
2195		2568
2196	if (coro->cctx && (coro->cctx->flags & CC_TRACE))	2569	if (coro->cctx && (coro->cctx->flags & CC_TRACE))
2197	api_trace (coro_current, 0);	2570	api_trace (aTHX_ coro_current, 0);
2198		2571
2199	av_push (av_async_pool, newSVsv (coro_current));	2572	av_push (av_async_pool, newSVsv (coro_current));
2200	}	2573	}
2201
2202	#if 0
2203
2204	void
2205	_generator_call (...)
2206	PROTOTYPE: @
2207	PPCODE:
2208	fprintf (stderr, "call %p\n", CvXSUBANY(cv).any_ptr);
2209	xxxx
2210	abort ();
2211
2212	SV *
2213	gensub (SV *sub, ...)
2214	PROTOTYPE: &;@
2215	CODE:
2216	{
2217	struct coro *coro;
2218	MAGIC *mg;
2219	CV *xcv;
2220	CV *ncv = (CV *)newSV_type (SVt_PVCV);
2221	int i;
2222
2223	CvGV (ncv) = CvGV (cv);
2224	CvFILE (ncv) = CvFILE (cv);
2225
2226	Newz (0, coro, 1, struct coro);
2227	coro->args = newAV ();
2228	coro->flags = CF_NEW;
2229
2230	av_extend (coro->args, items - 1);
2231	for (i = 1; i < items; i++)
2232	av_push (coro->args, newSVsv (ST (i)));
2233
2234	CvISXSUB_on (ncv);
2235	CvXSUBANY (ncv).any_ptr = (void *)coro;
2236
2237	xcv = GvCV (gv_fetchpv ("Coro::_generator_call", 0, SVt_PVCV));
2238
2239	CvXSUB (ncv) = CvXSUB (xcv);
2240	CvANON_on (ncv);
2241
2242	mg = sv_magicext ((SV *)ncv, 0, CORO_MAGIC_type_state, &coro_gensub_vtbl, (char *)coro, 0);
2243	RETVAL = newRV_noinc ((SV *)ncv);
2244	}
2245	OUTPUT:
2246	RETVAL
2247
2248	#endif
2249		2574
2250		2575
2251	MODULE = Coro::State PACKAGE = Coro::AIO	2576	MODULE = Coro::State PACKAGE = Coro::AIO
2252		2577
2253	void	2578	void
2254	_get_state (SV *self)	2579	_get_state (SV *self)
		2580	PROTOTYPE: $
2255	PPCODE:	2581	PPCODE:
2256	{	2582	{
2257	AV *defav = GvAV (PL_defgv);	2583	AV *defav = GvAV (PL_defgv);
2258	AV *av = newAV ();	2584	AV *av = newAV ();
2259	int i;	2585	int i;
…		…
2274		2600
2275	av_push (av, data_sv);	2601	av_push (av, data_sv);
2276		2602
2277	XPUSHs (sv_2mortal (newRV_noinc ((SV *)av)));	2603	XPUSHs (sv_2mortal (newRV_noinc ((SV *)av)));
2278		2604
2279	api_ready (self);	2605	api_ready (aTHX_ self);
2280	}	2606	}
2281		2607
2282	void	2608	void
2283	_set_state (SV *state)	2609	_set_state (SV *state)
2284	PROTOTYPE: $	2610	PROTOTYPE: $
…		…
2302	MODULE = Coro::State PACKAGE = Coro::AnyEvent	2628	MODULE = Coro::State PACKAGE = Coro::AnyEvent
2303		2629
2304	BOOT:	2630	BOOT:
2305	sv_activity = coro_get_sv (aTHX_ "Coro::AnyEvent::ACTIVITY", TRUE);	2631	sv_activity = coro_get_sv (aTHX_ "Coro::AnyEvent::ACTIVITY", TRUE);
2306		2632
2307	SV *	2633	void
2308	_schedule (...)	2634	_schedule (...)
2309	PROTOTYPE: @
2310	CODE:	2635	CODE:
2311	{	2636	{
2312	static int incede;	2637	static int incede;
2313		2638
2314	api_cede_notself ();	2639	api_cede_notself (aTHX);
2315		2640
2316	++incede;	2641	++incede;
2317	while (coro_nready >= incede && api_cede ())	2642	while (coro_nready >= incede && api_cede (aTHX))
2318	;	2643	;
2319		2644
2320	sv_setsv (sv_activity, &PL_sv_undef);	2645	sv_setsv (sv_activity, &PL_sv_undef);
2321	if (coro_nready >= incede)	2646	if (coro_nready >= incede)
2322	{	2647	{
…		…
2332		2657
2333	MODULE = Coro::State PACKAGE = PerlIO::cede	2658	MODULE = Coro::State PACKAGE = PerlIO::cede
2334		2659
2335	BOOT:	2660	BOOT:
2336	PerlIO_define_layer (aTHX_ &PerlIO_cede);	2661	PerlIO_define_layer (aTHX_ &PerlIO_cede);
		2662
		2663	MODULE = Coro::State PACKAGE = Coro::Semaphore
		2664
		2665	SV *
		2666	new (SV *klass, SV *count_ = 0)
		2667	CODE:
		2668	{
		2669	/* a semaphore contains a counter IV in $sem->[0] and any waiters after that */
		2670	AV *av = newAV ();
		2671	av_push (av, newSViv (count_ && SvOK (count_) ? SvIV (count_) : 1));
		2672	RETVAL = sv_bless (newRV_noinc ((SV *)av), GvSTASH (CvGV (cv)));
		2673	}
		2674	OUTPUT:
		2675	RETVAL
		2676
		2677	SV *
		2678	count (SV *self)
		2679	CODE:
		2680	RETVAL = newSVsv (AvARRAY ((AV *)SvRV (self))[0]);
		2681	OUTPUT:
		2682	RETVAL
		2683
		2684	void
		2685	up (SV *self, int adjust = 1)
		2686	ALIAS:
		2687	adjust = 1
		2688	CODE:
		2689	coro_semaphore_adjust ((AV *)SvRV (self), ix ? adjust : 1);
		2690
		2691	void
		2692	down (SV *self)
		2693	CODE:
		2694	api_execute_slf (aTHX_ cv, slf_init_semaphore_down, &ST (0), 1);
		2695
		2696	void
		2697	try (SV *self)
		2698	PPCODE:
		2699	{
		2700	AV *av = (AV *)SvRV (self);
		2701	SV *count_sv = AvARRAY (av)[0];
		2702	IV count = SvIVX (count_sv);
		2703
		2704	if (count > 0)
		2705	{
		2706	--count;
		2707	SvIVX (count_sv) = count;
		2708	XSRETURN_YES;
		2709	}
		2710	else
		2711	XSRETURN_NO;
		2712	}
		2713
		2714	void
		2715	waiters (SV *self)
		2716	CODE:
		2717	{
		2718	AV *av = (AV *)SvRV (self);
		2719
		2720	if (GIMME_V == G_SCALAR)
		2721	XPUSHs (sv_2mortal (newSVsv (AvARRAY (av)[0])));
		2722	else
		2723	{
		2724	int i;
		2725	EXTEND (SP, AvFILLp (av) + 1 - 1);
		2726	for (i = 1; i <= AvFILLp (av); ++i)
		2727	PUSHs (newSVsv (AvARRAY (av)[i]));
		2728	}
		2729	}
		2730

Diff Legend

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