ViewVC Help

View File | Revision Log | Show Annotations | Download File

/cvs/Coro/Coro/State.xs

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

Diff Legend

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