ViewVC Help

View File | Revision Log | Show Annotations | Download File

/cvs/Coro/Coro/State.xs

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

Diff Legend

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