ViewVC Help

View File | Revision Log | Show Annotations | Download File

/cvs/Coro/Coro/State.xs

Comparing Coro/Coro/State.xs (file contents):
Revision 1.276 by root, Sat Nov 15 07:49:48 2008 UTC vs.
Revision 1.305 by root, Wed Nov 19 10:44:41 2008 UTC

…		…
16		16
17	#ifdef WIN32	17	#ifdef WIN32
18	# undef setjmp	18	# undef setjmp
19	# undef longjmp	19	# undef longjmp
20	# undef _exit	20	# undef _exit
21	# define setjmp _setjmp // deep magic, don't ask	21	# define setjmp _setjmp /* deep magic */
22	#else	22	#else
23	# include <inttypes.h> /* most portable stdint.h */	23	# include <inttypes.h> /* most portable stdint.h */
24	#endif	24	#endif
25		25
26	#ifdef HAVE_MMAP	26	#ifdef HAVE_MMAP
…		…
95	# define GV_NOTQUAL 0	95	# define GV_NOTQUAL 0
96	#endif	96	#endif
97	#ifndef newSV	97	#ifndef newSV
98	# define newSV(l) NEWSV(0,l)	98	# define newSV(l) NEWSV(0,l)
99	#endif	99	#endif
		100	#ifndef CvISXSUB_on
		101	# define CvISXSUB_on(cv) (void)cv
		102	#endif
100		103
101	/* 5.8.7 */	104	/* 5.8.7 */
102	#ifndef SvRV_set	105	#ifndef SvRV_set
103	# define SvRV_set(s,v) SvRV(s) = (v)	106	# define SvRV_set(s,v) SvRV(s) = (v)
104	#endif	107	#endif
…		…
117	#endif	120	#endif
118		121
119	/* The next macros try to return the current stack pointer, in an as	122	/* The next macros try to return the current stack pointer, in an as
120	* portable way as possible. */	123	* portable way as possible. */
121	#if __GNUC__ >= 4	124	#if __GNUC__ >= 4
		125	# define dSTACKLEVEL int stacklevel_dummy
122	# define dSTACKLEVEL void *stacklevel = __builtin_frame_address (0)	126	# define STACKLEVEL __builtin_frame_address (0)
123	#else	127	#else
124	# define dSTACKLEVEL volatile void *stacklevel = (volatile void *)&stacklevel	128	# define dSTACKLEVEL volatile void *stacklevel
		129	# define STACKLEVEL ((void *)&stacklevel)
125	#endif	130	#endif
126		131
127	#define IN_DESTRUCT (PL_main_cv == Nullcv)	132	#define IN_DESTRUCT (PL_main_cv == Nullcv)
128		133
129	#if __GNUC__ >= 3	134	#if __GNUC__ >= 3
…		…
140	#define expect_true(expr) expect ((expr) != 0, 1)	145	#define expect_true(expr) expect ((expr) != 0, 1)
141		146
142	#define NOINLINE attribute ((noinline))	147	#define NOINLINE attribute ((noinline))
143		148
144	#include "CoroAPI.h"	149	#include "CoroAPI.h"
		150	#define GCoroAPI (&coroapi) /* very sneaky */
145		151
146	#ifdef USE_ITHREADS	152	#ifdef USE_ITHREADS
147
148	static perl_mutex coro_lock;
149	# define LOCK do { MUTEX_LOCK (&coro_lock); } while (0)
150	# define UNLOCK do { MUTEX_UNLOCK (&coro_lock); } while (0)
151	# if CORO_PTHREAD	153	# if CORO_PTHREAD
152	static void *coro_thx;	154	static void *coro_thx;
153	# endif	155	# endif
154
155	#else
156
157	# define LOCK (void)0
158	# define UNLOCK (void)0
159
160	#endif	156	#endif
161
162	# undef LOCK
163	# define LOCK (void)0
164	# undef UNLOCK
165	# define UNLOCK (void)0
166
167	/* helper storage struct for Coro::AIO */
168	struct io_state
169	{
170	AV *res;
171	int errorno;
172	I32 laststype; /* U16 in 5.10.0 */
173	int laststatval;
174	Stat_t statcache;
175	};
176		157
177	static double (*nvtime)(); /* so why doesn't it take void? */	158	static double (*nvtime)(); /* so why doesn't it take void? */
		159
		160	/* we hijack an hopefully unused CV flag for our purposes */
		161	#define CVf_SLF 0x4000
		162	static OP *pp_slf (pTHX);
178		163
179	static U32 cctx_gen;	164	static U32 cctx_gen;
180	static size_t cctx_stacksize = CORO_STACKSIZE;	165	static size_t cctx_stacksize = CORO_STACKSIZE;
181	static struct CoroAPI coroapi;	166	static struct CoroAPI coroapi;
182	static AV *main_mainstack; /* used to differentiate between $main and others */	167	static AV *main_mainstack; /* used to differentiate between $main and others */
183	static JMPENV *main_top_env;	168	static JMPENV *main_top_env;
184	static HV *coro_state_stash, *coro_stash;	169	static HV *coro_state_stash, *coro_stash;
185	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 */
186	static volatile struct coro *transfer_next;
187		171
188	static GV *irsgv; /* $/ */	172	static GV *irsgv; /* $/ */
189	static GV *stdoutgv; /* *STDOUT */	173	static GV *stdoutgv; /* *STDOUT */
190	static SV *rv_diehook;	174	static SV *rv_diehook;
191	static SV *rv_warnhook;	175	static SV *rv_warnhook;
…		…
257	/* this is a structure representing a perl-level coroutine */	241	/* this is a structure representing a perl-level coroutine */
258	struct coro {	242	struct coro {
259	/* the C coroutine allocated to this perl coroutine, if any */	243	/* the C coroutine allocated to this perl coroutine, if any */
260	coro_cctx *cctx;	244	coro_cctx *cctx;
261		245
262	/* process data */	246	/* state data */
263	struct CoroSLF slf_frame; /* saved slf frame */	247	struct CoroSLF slf_frame; /* saved slf frame */
264	AV *mainstack;	248	AV *mainstack;
265	perl_slots *slot; /* basically the saved sp */	249	perl_slots *slot; /* basically the saved sp */
266		250
267	AV *args; /* data associated with this coroutine (initial args) */	251	AV *args; /* data associated with this coroutine (initial args) */
268	int refcnt; /* coroutines are refcounted, yes */	252	int refcnt; /* coroutines are refcounted, yes */
269	int flags; /* CF_ flags */	253	int flags; /* CF_ flags */
270	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);
271		256
272	/* statistics */	257	/* statistics */
273	int usecount; /* number of transfers to this coro */	258	int usecount; /* number of transfers to this coro */
274		259
275	/* coro process data */	260	/* coro process data */
276	int prio;	261	int prio;
277	SV *throw; /* exception to be thrown */	262	SV *except; /* exception to be thrown */
		263	SV *rouse_cb;
278		264
279	/* async_pool */	265	/* async_pool */
280	SV *saved_deffh;	266	SV *saved_deffh;
281		267
282	/* linked list */	268	/* linked list */
…		…
284	};	270	};
285		271
286	typedef struct coro *Coro__State;	272	typedef struct coro *Coro__State;
287	typedef struct coro *Coro__State_or_hashref;	273	typedef struct coro *Coro__State_or_hashref;
288		274
		275	/* the following variables are effectively part of the perl context */
		276	/* and get copied between struct coro and these variables */
		277	/* the mainr easonw e don't support windows process emulation */
289	static struct CoroSLF slf_frame; /* the current slf frame */	278	static struct CoroSLF slf_frame; /* the current slf frame */
290		279
291	/** Coro ********************************************************************/	280	/** Coro ********************************************************************/
292		281
293	#define PRIO_MAX 3	282	#define PRIO_MAX 3
…		…
395	SvREFCNT_dec (av); /* sv_magicext increased the refcount */	384	SvREFCNT_dec (av); /* sv_magicext increased the refcount */
396		385
397	return 0;	386	return 0;
398	}	387	}
399		388
400	#define CORO_MAGIC_type_cv PERL_MAGIC_ext	389	#define CORO_MAGIC_type_cv 26
401	#define CORO_MAGIC_type_state PERL_MAGIC_ext	390	#define CORO_MAGIC_type_state PERL_MAGIC_ext
402		391
403	static MGVTBL coro_cv_vtbl = {	392	static MGVTBL coro_cv_vtbl = {
404	0, 0, 0, 0,	393	0, 0, 0, 0,
405	coro_cv_free	394	coro_cv_free
406	};	395	};
407		396
		397	#define CORO_MAGIC_NN(sv, type) \
		398	(expect_true (SvMAGIC (sv)->mg_type == type) \
		399	? SvMAGIC (sv) \
		400	: mg_find (sv, type))
		401
408	#define CORO_MAGIC(sv, type) \	402	#define CORO_MAGIC(sv, type) \
409	SvMAGIC (sv) \	403	(expect_true (SvMAGIC (sv)) \
410	? SvMAGIC (sv)->mg_type == type \	404	? CORO_MAGIC_NN (sv, type) \
411	? SvMAGIC (sv) \
412	: mg_find (sv, type) \
413	: 0	405	: 0)
414		406
415	#define CORO_MAGIC_cv(cv) CORO_MAGIC (((SV *)(cv)), CORO_MAGIC_type_cv)	407	#define CORO_MAGIC_cv(cv) CORO_MAGIC (((SV *)(cv)), CORO_MAGIC_type_cv)
416	#define CORO_MAGIC_state(sv) CORO_MAGIC (((SV *)(sv)), CORO_MAGIC_type_state)	408	#define CORO_MAGIC_state(sv) CORO_MAGIC_NN (((SV *)(sv)), CORO_MAGIC_type_state)
417		409
418	INLINE struct coro *	410	INLINE struct coro *
419	SvSTATE_ (pTHX_ SV *coro)	411	SvSTATE_ (pTHX_ SV *coro)
420	{	412	{
421	HV *stash;	413	HV *stash;
…		…
438	mg = CORO_MAGIC_state (coro);	430	mg = CORO_MAGIC_state (coro);
439	return (struct coro *)mg->mg_ptr;	431	return (struct coro *)mg->mg_ptr;
440	}	432	}
441		433
442	#define SvSTATE(sv) SvSTATE_ (aTHX_ (sv))	434	#define SvSTATE(sv) SvSTATE_ (aTHX_ (sv))
		435
		436	/* faster than SvSTATE, but expects a coroutine hv */
		437	#define SvSTATE_hv(hv) ((struct coro *)CORO_MAGIC_NN ((SV *)hv, CORO_MAGIC_type_state)->mg_ptr)
		438	#define SvSTATE_current SvSTATE_hv (SvRV (coro_current))
443		439
444	/* the next two functions merely cache the padlists */	440	/* the next two functions merely cache the padlists */
445	static void	441	static void
446	get_padlist (pTHX_ CV *cv)	442	get_padlist (pTHX_ CV *cv)
447	{	443	{
…		…
453	else	449	else
454	{	450	{
455	#if CORO_PREFER_PERL_FUNCTIONS	451	#if CORO_PREFER_PERL_FUNCTIONS
456	/* this is probably cleaner? but also slower! */	452	/* this is probably cleaner? but also slower! */
457	/* in practise, it seems to be less stable */	453	/* in practise, it seems to be less stable */
458	CV *cp = Perl_cv_clone (cv);	454	CV *cp = Perl_cv_clone (aTHX_ cv);
459	CvPADLIST (cv) = CvPADLIST (cp);	455	CvPADLIST (cv) = CvPADLIST (cp);
460	CvPADLIST (cp) = 0;	456	CvPADLIST (cp) = 0;
461	SvREFCNT_dec (cp);	457	SvREFCNT_dec (cp);
462	#else	458	#else
463	CvPADLIST (cv) = coro_clone_padlist (aTHX_ cv);	459	CvPADLIST (cv) = coro_clone_padlist (aTHX_ cv);
…		…
515	}	511	}
516		512
517	PUTBACK;	513	PUTBACK;
518	}	514	}
519		515
520	slf_frame = c->slf_frame;	516	slf_frame = c->slf_frame;
		517	CORO_THROW = c->except;
521	}	518	}
522		519
523	static void	520	static void
524	save_perl (pTHX_ Coro__State c)	521	save_perl (pTHX_ Coro__State c)
525	{	522	{
		523	c->except = CORO_THROW;
526	c->slf_frame = slf_frame;	524	c->slf_frame = slf_frame;
527		525
528	{	526	{
529	dSP;	527	dSP;
530	I32 cxix = cxstack_ix;	528	I32 cxix = cxstack_ix;
…		…
605	* of perl.c:init_stacks, except that it uses less memory	603	* of perl.c:init_stacks, except that it uses less memory
606	* on the (sometimes correct) assumption that coroutines do	604	* on the (sometimes correct) assumption that coroutines do
607	* not usually need a lot of stackspace.	605	* not usually need a lot of stackspace.
608	*/	606	*/
609	#if CORO_PREFER_PERL_FUNCTIONS	607	#if CORO_PREFER_PERL_FUNCTIONS
610	# define coro_init_stacks init_stacks	608	# define coro_init_stacks(thx) init_stacks ()
611	#else	609	#else
612	static void	610	static void
613	coro_init_stacks (pTHX)	611	coro_init_stacks (pTHX)
614	{	612	{
615	PL_curstackinfo = new_stackinfo(32, 8);	613	PL_curstackinfo = new_stackinfo(32, 8);
…		…
811		809
812	return orig_sigelem_set ? orig_sigelem_set (aTHX_ sv, mg) : 0;	810	return orig_sigelem_set ? orig_sigelem_set (aTHX_ sv, mg) : 0;
813	}	811	}
814		812
815	static void	813	static void
816	prepare_nop (aTHX_ struct coro_transfer_args *ta)	814	prepare_nop (pTHX_ struct coro_transfer_args *ta)
817	{	815	{
818	/* kind of mega-hacky, but works */	816	/* kind of mega-hacky, but works */
819	ta->next = ta->prev = (struct coro *)ta;	817	ta->next = ta->prev = (struct coro *)ta;
820	}	818	}
821		819
822	static int	820	static int
823	slf_check_nop (aTHX)	821	slf_check_nop (pTHX_ struct CoroSLF *frame)
824	{	822	{
825	return 0;	823	return 0;
826	}	824	}
827		825
828	static void	826	static UNOP coro_setup_op;
		827
		828	static void NOINLINE /* noinline to keep it out of the transfer fast path */
829	coro_setup (pTHX_ struct coro *coro)	829	coro_setup (pTHX_ struct coro *coro)
830	{	830	{
831	/*	831	/*
832	* emulate part of the perl startup here.	832	* emulate part of the perl startup here.
833	*/	833	*/
…		…
860	{	860	{
861	dSP;	861	dSP;
862	UNOP myop;	862	UNOP myop;
863		863
864	Zero (&myop, 1, UNOP);	864	Zero (&myop, 1, UNOP);
865	myop.op_next = Nullop;	865	myop.op_next = Nullop;
866	myop.op_flags = OPf_WANT_VOID;	866	myop.op_flags = OPf_WANT_VOID;
867		867
868	PUSHMARK (SP);	868	PUSHMARK (SP);
869	XPUSHs (sv_2mortal (av_shift (GvAV (PL_defgv))));	869	XPUSHs (sv_2mortal (av_shift (GvAV (PL_defgv))));
870	PUTBACK;	870	PUTBACK;
…		…
876	/* this newly created coroutine might be run on an existing cctx which most	876	/* this newly created coroutine might be run on an existing cctx which most
877	* likely was suspended in pp_slf, so we have to emulate entering pp_slf here.	877	* likely was suspended in pp_slf, so we have to emulate entering pp_slf here.
878	*/	878	*/
879	slf_frame.prepare = prepare_nop; /* provide a nop function for an eventual pp_slf */	879	slf_frame.prepare = prepare_nop; /* provide a nop function for an eventual pp_slf */
880	slf_frame.check = slf_check_nop; /* signal pp_slf to not repeat */	880	slf_frame.check = slf_check_nop; /* signal pp_slf to not repeat */
		881
		882	/* and we have to provide the pp_slf op in any case, so pp_slf can skip it */
		883	coro_setup_op.op_next = PL_op;
		884	coro_setup_op.op_type = OP_CUSTOM;
		885	coro_setup_op.op_ppaddr = pp_slf;
		886	/* no flags required, as an init function won't be called */
		887
		888	PL_op = (OP *)&coro_setup_op;
		889
		890	/* copy throw, in case it was set before coro_setup */
		891	CORO_THROW = coro->except;
881	}	892	}
882		893
883	static void	894	static void
884	coro_destruct (pTHX_ struct coro *coro)	895	coro_destruct (pTHX_ struct coro *coro)
885	{	896	{
…		…
908	SvREFCNT_dec (GvSV (irsgv));	919	SvREFCNT_dec (GvSV (irsgv));
909		920
910	SvREFCNT_dec (PL_diehook);	921	SvREFCNT_dec (PL_diehook);
911	SvREFCNT_dec (PL_warnhook);	922	SvREFCNT_dec (PL_warnhook);
912		923
		924	SvREFCNT_dec (CORO_THROW);
913	SvREFCNT_dec (coro->saved_deffh);	925	SvREFCNT_dec (coro->saved_deffh);
914	SvREFCNT_dec (coro->throw);	926	SvREFCNT_dec (coro->rouse_cb);
915		927
916	coro_destruct_stacks (aTHX);	928	coro_destruct_stacks (aTHX);
917	}	929	}
918		930
919	INLINE void	931	INLINE void
…		…
929	static int	941	static int
930	runops_trace (pTHX)	942	runops_trace (pTHX)
931	{	943	{
932	COP *oldcop = 0;	944	COP *oldcop = 0;
933	int oldcxix = -2;	945	int oldcxix = -2;
934	struct coro *coro = SvSTATE (coro_current); /* trace cctx is tied to specific coro */	946	struct coro *coro = SvSTATE_current; /* trace cctx is tied to specific coro */
935	coro_cctx *cctx = coro->cctx;	947	coro_cctx *cctx = coro->cctx;
936		948
937	while ((PL_op = CALL_FPTR (PL_op->op_ppaddr) (aTHX)))	949	while ((PL_op = CALL_FPTR (PL_op->op_ppaddr) (aTHX)))
938	{	950	{
939	PERL_ASYNC_CHECK ();	951	PERL_ASYNC_CHECK ();
…		…
1048		1060
1049	TAINT_NOT;	1061	TAINT_NOT;
1050	return 0;	1062	return 0;
1051	}	1063	}
1052		1064
		1065	static struct coro_cctx *cctx_ssl_cctx;
		1066	static struct CoroSLF cctx_ssl_frame;
		1067
1053	static void	1068	static void
1054	prepare_set_stacklevel (struct coro_transfer_args *ta, struct coro_cctx *cctx)	1069	slf_prepare_set_stacklevel (pTHX_ struct coro_transfer_args *ta)
1055	{	1070	{
1056	ta->prev = (struct coro *)cctx;	1071	ta->prev = (struct coro *)cctx_ssl_cctx;
1057	ta->next = 0;	1072	ta->next = 0;
1058	}	1073	}
1059		1074
1060	/* inject a fake call to Coro::State::_cctx_init into the execution */	1075	static int
1061	/* _cctx_init should be careful, as it could be called at almost any time */	1076	slf_check_set_stacklevel (pTHX_ struct CoroSLF *frame)
1062	/* during execution of a perl program */	1077	{
1063	/* also initialises PL_top_env */	1078	*frame = cctx_ssl_frame;
		1079
		1080	return frame->check (aTHX_ frame); /* execute the restored frame - there must be one */
		1081	}
		1082
		1083	/* initialises PL_top_env and injects a pseudo-slf-call to set the stacklevel */
1064	static void NOINLINE	1084	static void NOINLINE
1065	cctx_prepare (pTHX_ coro_cctx *cctx)	1085	cctx_prepare (pTHX_ coro_cctx *cctx)
1066	{	1086	{
1067	dSP;
1068	UNOP myop;
1069
1070	PL_top_env = &PL_start_env;	1087	PL_top_env = &PL_start_env;
1071		1088
1072	if (cctx->flags & CC_TRACE)	1089	if (cctx->flags & CC_TRACE)
1073	PL_runops = runops_trace;	1090	PL_runops = runops_trace;
1074		1091
1075	Zero (&myop, 1, UNOP);	1092	/* we already must be executing an SLF op, there is no other valid way
1076	myop.op_next = PL_op;	1093	* that can lead to creation of a new cctx */
1077	myop.op_flags = OPf_WANT_VOID | OPf_STACKED;	1094	assert (("FATAL: can't prepare slf-less cctx in Coro module (please report)",
		1095	slf_frame.prepare && PL_op->op_ppaddr == pp_slf));
1078		1096
1079	PUSHMARK (SP);	1097	/* we must emulate leaving pp_slf, which is done inside slf_check_set_stacklevel */
1080	EXTEND (SP, 2);	1098	cctx_ssl_cctx = cctx;
1081	PUSHs (sv_2mortal (newSViv ((IV)cctx)));	1099	cctx_ssl_frame = slf_frame;
1082	PUSHs ((SV *)get_cv ("Coro::State::_cctx_init", FALSE));	1100
1083	PUTBACK;	1101	slf_frame.prepare = slf_prepare_set_stacklevel;
1084	PL_op = (OP *)&myop;	1102	slf_frame.check = slf_check_set_stacklevel;
1085	PL_op = PL_ppaddr[OP_ENTERSUB](aTHX);
1086	SPAGAIN;
1087	}	1103	}
1088		1104
1089	/* the tail of transfer: execute stuff we can only do after a transfer */	1105	/* the tail of transfer: execute stuff we can only do after a transfer */
1090	INLINE void	1106	INLINE void
1091	transfer_tail (pTHX)	1107	transfer_tail (pTHX)
1092	{	1108	{
1093	struct coro *next = (struct coro *)transfer_next;
1094	assert (!(transfer_next = 0)); /* just used for the side effect when asserts are enabled */
1095	assert (("FATAL: next coroutine was zero in transfer_tail (please report)", next));
1096
1097	free_coro_mortal (aTHX);	1109	free_coro_mortal (aTHX);
1098	UNLOCK;
1099
1100	if (expect_false (next->throw))
1101	{
1102	SV *exception = sv_2mortal (next->throw);
1103
1104	next->throw = 0;
1105	sv_setsv (ERRSV, exception);
1106	croak (0);
1107	}
1108	}	1110	}
1109		1111
1110	/*	1112	/*
1111	* this is a _very_ stripped down perl interpreter ;)	1113	* this is a _very_ stripped down perl interpreter ;)
1112	*/	1114	*/
…		…
1127		1129
1128	/* inject a fake subroutine call to cctx_init */	1130	/* inject a fake subroutine call to cctx_init */
1129	cctx_prepare (aTHX_ (coro_cctx *)arg);	1131	cctx_prepare (aTHX_ (coro_cctx *)arg);
1130		1132
1131	/* cctx_run is the alternative tail of transfer() */	1133	/* cctx_run is the alternative tail of transfer() */
1132	/* TODO: throwing an exception here might be deadly, VERIFY */
1133	transfer_tail (aTHX);	1134	transfer_tail (aTHX);
1134		1135
1135	/* somebody or something will hit me for both perl_run and PL_restartop */	1136	/* somebody or something will hit me for both perl_run and PL_restartop */
1136	PL_restartop = PL_op;	1137	PL_restartop = PL_op;
1137	perl_run (PL_curinterp);	1138	perl_run (PL_curinterp);
…		…
1293	/** coroutine switching *****************************************************/	1294	/** coroutine switching *****************************************************/
1294		1295
1295	static void	1296	static void
1296	transfer_check (pTHX_ struct coro *prev, struct coro *next)	1297	transfer_check (pTHX_ struct coro *prev, struct coro *next)
1297	{	1298	{
		1299	/* TODO: throwing up here is considered harmful */
		1300
1298	if (expect_true (prev != next))	1301	if (expect_true (prev != next))
1299	{	1302	{
1300	if (expect_false (!(prev->flags & (CF_RUNNING | CF_NEW))))	1303	if (expect_false (!(prev->flags & (CF_RUNNING | CF_NEW))))
1301	croak ("Coro::State::transfer called with non-running/new prev Coro::State, but can only transfer from running or new states,");	1304	croak ("Coro::State::transfer called with non-running/new prev Coro::State, but can only transfer from running or new states,");
1302		1305
…		…
1312	#endif	1315	#endif
1313	}	1316	}
1314	}	1317	}
1315		1318
1316	/* always use the TRANSFER macro */	1319	/* always use the TRANSFER macro */
1317	static void NOINLINE	1320	static void NOINLINE /* noinline so we have a fixed stackframe */
1318	transfer (pTHX_ struct coro *prev, struct coro *next, int force_cctx)	1321	transfer (pTHX_ struct coro *prev, struct coro *next, int force_cctx)
1319	{	1322	{
1320	dSTACKLEVEL;	1323	dSTACKLEVEL;
1321		1324
1322	/* sometimes transfer is only called to set idle_sp */	1325	/* sometimes transfer is only called to set idle_sp */
1323	if (expect_false (!next))	1326	if (expect_false (!next))
1324	{	1327	{
1325	((coro_cctx *)prev)->idle_sp = stacklevel;	1328	((coro_cctx *)prev)->idle_sp = STACKLEVEL;
1326	assert (((coro_cctx *)prev)->idle_te = PL_top_env); /* just for the side-effect when asserts are enabled */	1329	assert (((coro_cctx *)prev)->idle_te = PL_top_env); /* just for the side-effect when asserts are enabled */
1327	}	1330	}
1328	else if (expect_true (prev != next))	1331	else if (expect_true (prev != next))
1329	{	1332	{
1330	coro_cctx *prev__cctx;	1333	coro_cctx *prev__cctx;
…		…
1337	prev->flags |= CF_RUNNING;	1340	prev->flags |= CF_RUNNING;
1338	}	1341	}
1339		1342
1340	prev->flags &= ~CF_RUNNING;	1343	prev->flags &= ~CF_RUNNING;
1341	next->flags |= CF_RUNNING;	1344	next->flags |= CF_RUNNING;
1342
1343	LOCK;
1344		1345
1345	/* first get rid of the old state */	1346	/* first get rid of the old state */
1346	save_perl (aTHX_ prev);	1347	save_perl (aTHX_ prev);
1347		1348
1348	if (expect_false (next->flags & CF_NEW))	1349	if (expect_false (next->flags & CF_NEW))
…		…
1357		1358
1358	prev__cctx = prev->cctx;	1359	prev__cctx = prev->cctx;
1359		1360
1360	/* possibly untie and reuse the cctx */	1361	/* possibly untie and reuse the cctx */
1361	if (expect_true (	1362	if (expect_true (
1362	prev__cctx->idle_sp == stacklevel	1363	prev__cctx->idle_sp == STACKLEVEL
1363	&& !(prev__cctx->flags & CC_TRACE)	1364	&& !(prev__cctx->flags & CC_TRACE)
1364	&& !force_cctx	1365	&& !force_cctx
1365	))	1366	))
1366	{	1367	{
1367	/* I assume that stacklevel is a stronger indicator than PL_top_env changes */	1368	/* I assume that stacklevel is a stronger indicator than PL_top_env changes */
…		…
1381	++next->usecount;	1382	++next->usecount;
1382		1383
1383	if (expect_true (!next->cctx))	1384	if (expect_true (!next->cctx))
1384	next->cctx = cctx_get (aTHX);	1385	next->cctx = cctx_get (aTHX);
1385		1386
1386	assert (("FATAL: transfer_next already nonzero in Coro (please report)", !transfer_next));
1387	transfer_next = next;
1388
1389	if (expect_false (prev__cctx != next->cctx))	1387	if (expect_false (prev__cctx != next->cctx))
1390	{	1388	{
1391	prev__cctx->top_env = PL_top_env;	1389	prev__cctx->top_env = PL_top_env;
1392	PL_top_env = next->cctx->top_env;	1390	PL_top_env = next->cctx->top_env;
1393	coro_transfer (&prev__cctx->cctx, &next->cctx->cctx);	1391	coro_transfer (&prev__cctx->cctx, &next->cctx->cctx);
…		…
1406	coro_state_destroy (pTHX_ struct coro *coro)	1404	coro_state_destroy (pTHX_ struct coro *coro)
1407	{	1405	{
1408	if (coro->flags & CF_DESTROYED)	1406	if (coro->flags & CF_DESTROYED)
1409	return 0;	1407	return 0;
1410		1408
		1409	if (coro->on_destroy)
		1410	coro->on_destroy (aTHX_ coro);
		1411
1411	coro->flags |= CF_DESTROYED;	1412	coro->flags |= CF_DESTROYED;
1412		1413
1413	if (coro->flags & CF_READY)	1414	if (coro->flags & CF_READY)
1414	{	1415	{
1415	/* reduce nready, as destroying a ready coro effectively unreadies it */	1416	/* reduce nready, as destroying a ready coro effectively unreadies it */
1416	/* alternative: look through all ready queues and remove the coro */	1417	/* alternative: look through all ready queues and remove the coro */
1417	LOCK;
1418	--coro_nready;	1418	--coro_nready;
1419	UNLOCK;
1420	}	1419	}
1421	else	1420	else
1422	coro->flags |= CF_READY; /* make sure it is NOT put into the readyqueue */	1421	coro->flags |= CF_READY; /* make sure it is NOT put into the readyqueue */
1423		1422
1424	if (coro->mainstack && coro->mainstack != main_mainstack)	1423	if (coro->mainstack && coro->mainstack != main_mainstack)
…		…
1500		1499
1501	prepare_transfer (aTHX_ &ta, prev_sv, next_sv);	1500	prepare_transfer (aTHX_ &ta, prev_sv, next_sv);
1502	TRANSFER (ta, 1);	1501	TRANSFER (ta, 1);
1503	}	1502	}
1504		1503
		1504	/*****************************************************************************/
		1505	/* gensub: simple closure generation utility */
		1506
		1507	#define GENSUB_ARG CvXSUBANY (cv).any_ptr
		1508
		1509	/* create a closure from XS, returns a code reference */
		1510	/* the arg can be accessed via GENSUB_ARG from the callback */
		1511	/* the callback must use dXSARGS/XSRETURN */
		1512	static SV *
		1513	gensub (pTHX_ void (*xsub)(pTHX_ CV *), void *arg)
		1514	{
		1515	CV *cv = (CV *)newSV (0);
		1516
		1517	sv_upgrade ((SV *)cv, SVt_PVCV);
		1518
		1519	CvANON_on (cv);
		1520	CvISXSUB_on (cv);
		1521	CvXSUB (cv) = xsub;
		1522	GENSUB_ARG = arg;
		1523
		1524	return newRV_noinc ((SV *)cv);
		1525	}
		1526
1505	/** Coro ********************************************************************/	1527	/** Coro ********************************************************************/
1506		1528
1507	static void	1529	INLINE void
1508	coro_enq (pTHX_ SV *coro_sv)	1530	coro_enq (pTHX_ struct coro *coro)
1509	{	1531	{
1510	av_push (coro_ready [SvSTATE (coro_sv)->prio - PRIO_MIN], coro_sv);	1532	av_push (coro_ready [coro->prio - PRIO_MIN], SvREFCNT_inc_NN (coro->hv));
1511	}	1533	}
1512		1534
1513	static SV *	1535	INLINE SV *
1514	coro_deq (pTHX)	1536	coro_deq (pTHX)
1515	{	1537	{
1516	int prio;	1538	int prio;
1517		1539
1518	for (prio = PRIO_MAX - PRIO_MIN + 1; --prio >= 0; )	1540	for (prio = PRIO_MAX - PRIO_MIN + 1; --prio >= 0; )
…		…
1537	if (coro->flags & CF_READY)	1559	if (coro->flags & CF_READY)
1538	return 0;	1560	return 0;
1539		1561
1540	coro->flags |= CF_READY;	1562	coro->flags |= CF_READY;
1541		1563
1542	LOCK;
1543
1544	sv_hook = coro_nready ? 0 : coro_readyhook;	1564	sv_hook = coro_nready ? 0 : coro_readyhook;
1545	xs_hook = coro_nready ? 0 : coroapi.readyhook;	1565	xs_hook = coro_nready ? 0 : coroapi.readyhook;
1546		1566
1547	coro_enq (aTHX_ SvREFCNT_inc_NN (coro_sv));	1567	coro_enq (aTHX_ coro);
1548	++coro_nready;	1568	++coro_nready;
1549		1569
1550	UNLOCK;
1551
1552	if (sv_hook)	1570	if (sv_hook)
1553	{	1571	{
1554	dSP;	1572	dSP;
1555		1573
1556	ENTER;	1574	ENTER;
1557	SAVETMPS;	1575	SAVETMPS;
1558		1576
1559	PUSHMARK (SP);	1577	PUSHMARK (SP);
1560	PUTBACK;	1578	PUTBACK;
1561	call_sv (sv_hook, G_DISCARD);	1579	call_sv (sv_hook, G_VOID | G_DISCARD);
1562	SPAGAIN;
1563		1580
1564	FREETMPS;	1581	FREETMPS;
1565	LEAVE;	1582	LEAVE;
1566	}	1583	}
1567		1584
…		…
1582	{	1599	{
1583	SV *prev_sv, *next_sv;	1600	SV *prev_sv, *next_sv;
1584		1601
1585	for (;;)	1602	for (;;)
1586	{	1603	{
1587	LOCK;
1588	next_sv = coro_deq (aTHX);	1604	next_sv = coro_deq (aTHX);
1589		1605
1590	/* nothing to schedule: call the idle handler */	1606	/* nothing to schedule: call the idle handler */
1591	if (expect_false (!next_sv))	1607	if (expect_false (!next_sv))
1592	{	1608	{
1593	dSP;	1609	dSP;
1594	UNLOCK;
1595		1610
1596	ENTER;	1611	ENTER;
1597	SAVETMPS;	1612	SAVETMPS;
1598		1613
1599	PUSHMARK (SP);	1614	PUSHMARK (SP);
1600	PUTBACK;	1615	PUTBACK;
1601	call_sv (get_sv ("Coro::idle", FALSE), G_DISCARD);	1616	call_sv (get_sv ("Coro::idle", FALSE), G_VOID | G_DISCARD);
1602	SPAGAIN;
1603		1617
1604	FREETMPS;	1618	FREETMPS;
1605	LEAVE;	1619	LEAVE;
1606	continue;	1620	continue;
1607	}	1621	}
1608		1622
1609	ta->next = SvSTATE (next_sv);	1623	ta->next = SvSTATE_hv (next_sv);
1610		1624
1611	/* cannot transfer to destroyed coros, skip and look for next */	1625	/* cannot transfer to destroyed coros, skip and look for next */
1612	if (expect_false (ta->next->flags & CF_DESTROYED))	1626	if (expect_false (ta->next->flags & CF_DESTROYED))
1613	{	1627	{
1614	UNLOCK;
1615	SvREFCNT_dec (next_sv);	1628	SvREFCNT_dec (next_sv);
1616	/* coro_nready has already been taken care of by destroy */	1629	/* coro_nready has already been taken care of by destroy */
1617	continue;	1630	continue;
1618	}	1631	}
1619		1632
1620	--coro_nready;	1633	--coro_nready;
1621	UNLOCK;
1622	break;	1634	break;
1623	}	1635	}
1624		1636
1625	/* free this only after the transfer */	1637	/* free this only after the transfer */
1626	prev_sv = SvRV (coro_current);	1638	prev_sv = SvRV (coro_current);
1627	ta->prev = SvSTATE (prev_sv);	1639	ta->prev = SvSTATE_hv (prev_sv);
1628	TRANSFER_CHECK (*ta);	1640	TRANSFER_CHECK (*ta);
1629	assert (("FATAL: next coroutine isn't marked as ready in Coro (please report)", ta->next->flags & CF_READY));	1641	assert (("FATAL: next coroutine isn't marked as ready in Coro (please report)", ta->next->flags & CF_READY));
1630	ta->next->flags &= ~CF_READY;	1642	ta->next->flags &= ~CF_READY;
1631	SvRV_set (coro_current, next_sv);	1643	SvRV_set (coro_current, next_sv);
1632		1644
1633	LOCK;
1634	free_coro_mortal (aTHX);	1645	free_coro_mortal (aTHX);
1635	coro_mortal = prev_sv;	1646	coro_mortal = prev_sv;
1636	UNLOCK;
1637	}	1647	}
1638		1648
1639	INLINE void	1649	INLINE void
1640	prepare_cede (pTHX_ struct coro_transfer_args *ta)	1650	prepare_cede (pTHX_ struct coro_transfer_args *ta)
1641	{	1651	{
…		…
1720	else	1730	else
1721	coro->slot->runops = RUNOPS_DEFAULT;	1731	coro->slot->runops = RUNOPS_DEFAULT;
1722	}	1732	}
1723	}	1733	}
1724		1734
1725	#if 0	1735	/*****************************************************************************/
		1736	/* rouse callback */
		1737
		1738	#define CORO_MAGIC_type_rouse PERL_MAGIC_ext
		1739
		1740	static void
		1741	coro_rouse_callback (pTHX_ CV *cv)
		1742	{
		1743	dXSARGS;
		1744	SV *data = (SV *)GENSUB_ARG;
		1745
		1746	if (SvTYPE (SvRV (data)) != SVt_PVAV)
		1747	{
		1748	/* first call, set args */
		1749	int i;
		1750	AV *av = newAV ();
		1751	SV *coro = SvRV (data);
		1752
		1753	SvRV_set (data, (SV *)av);
		1754	api_ready (aTHX_ coro);
		1755	SvREFCNT_dec (coro);
		1756
		1757	/* better take a full copy of the arguments */
		1758	while (items--)
		1759	av_store (av, items, newSVsv (ST (items)));
		1760	}
		1761
		1762	XSRETURN_EMPTY;
		1763	}
		1764
1726	static int	1765	static int
1727	coro_gensub_free (pTHX_ SV *sv, MAGIC *mg)	1766	slf_check_rouse_wait (pTHX_ struct CoroSLF *frame)
1728	{	1767	{
1729	AV *padlist;	1768	SV *data = (SV *)frame->data;
1730	AV *av = (AV *)mg->mg_obj;	1769
		1770	if (CORO_THROW)
		1771	return 0;
1731		1772
1732	abort ();	1773	if (SvTYPE (SvRV (data)) != SVt_PVAV)
		1774	return 1;
		1775
		1776	/* now push all results on the stack */
		1777	{
		1778	dSP;
		1779	AV *av = (AV *)SvRV (data);
		1780	int i;
		1781
		1782	EXTEND (SP, AvFILLp (av) + 1);
		1783	for (i = 0; i <= AvFILLp (av); ++i)
		1784	PUSHs (sv_2mortal (AvARRAY (av)[i]));
		1785
		1786	/* we have stolen the elements, so ste length to zero and free */
		1787	AvFILLp (av) = -1;
		1788	av_undef (av);
		1789
		1790	PUTBACK;
		1791	}
1733		1792
1734	return 0;	1793	return 0;
1735	}	1794	}
1736		1795
1737	static MGVTBL coro_gensub_vtbl = {	1796	static void
1738	0, 0, 0, 0,	1797	slf_init_rouse_wait (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
1739	coro_gensub_free	1798	{
1740	};	1799	SV *cb;
1741	#endif	1800
		1801	if (items)
		1802	cb = arg [0];
		1803	else
		1804	{
		1805	struct coro *coro = SvSTATE_current;
		1806
		1807	if (!coro->rouse_cb)
		1808	croak ("Coro::rouse_wait called without rouse callback, and no default rouse callback found either,");
		1809
		1810	cb = sv_2mortal (coro->rouse_cb);
		1811	coro->rouse_cb = 0;
		1812	}
		1813
		1814	if (!SvROK (cb)
		1815	|| SvTYPE (SvRV (cb)) != SVt_PVCV
		1816	|| CvXSUB ((CV *)SvRV (cb)) != coro_rouse_callback)
		1817	croak ("Coro::rouse_wait called with illegal callback argument,");
		1818
		1819	{
		1820	CV *cv = (CV *)SvRV (cb); /* for GENSUB_ARG */
		1821	SV *data = (SV *)GENSUB_ARG;
		1822
		1823	frame->data = (void *)data;
		1824	frame->prepare = SvTYPE (SvRV (data)) == SVt_PVAV ? prepare_nop : prepare_schedule;
		1825	frame->check = slf_check_rouse_wait;
		1826	}
		1827	}
		1828
		1829	static SV *
		1830	coro_new_rouse_cb (pTHX)
		1831	{
		1832	HV *hv = (HV *)SvRV (coro_current);
		1833	struct coro *coro = SvSTATE_hv (hv);
		1834	SV *data = newRV_inc ((SV *)hv);
		1835	SV *cb = gensub (aTHX_ coro_rouse_callback, (void *)data);
		1836
		1837	sv_magicext (SvRV (cb), data, CORO_MAGIC_type_rouse, 0, 0, 0);
		1838	SvREFCNT_dec (data); /* magicext increases the refcount */
		1839
		1840	SvREFCNT_dec (coro->rouse_cb);
		1841	coro->rouse_cb = SvREFCNT_inc_NN (cb);
		1842
		1843	return cb;
		1844	}
		1845
		1846	/*****************************************************************************/
		1847	/* schedule-like-function opcode (SLF) */
		1848
		1849	static UNOP slf_restore; /* restore stack as entersub did, for first-re-run */
		1850	static const CV *slf_cv;
		1851	static SV **slf_argv;
		1852	static int slf_argc, slf_arga; /* count, allocated */
		1853	static I32 slf_ax; /* top of stack, for restore */
		1854
		1855	/* this restores the stack in the case we patched the entersub, to */
		1856	/* recreate the stack frame as perl will on following calls */
		1857	/* since entersub cleared the stack */
		1858	static OP *
		1859	pp_restore (pTHX)
		1860	{
		1861	int i;
		1862	SV **SP = PL_stack_base + slf_ax;
		1863
		1864	PUSHMARK (SP);
		1865
		1866	EXTEND (SP, slf_argc + 1);
		1867
		1868	for (i = 0; i < slf_argc; ++i)
		1869	PUSHs (sv_2mortal (slf_argv [i]));
		1870
		1871	PUSHs ((SV *)CvGV (slf_cv));
		1872
		1873	RETURNOP (slf_restore.op_first);
		1874	}
		1875
		1876	static void
		1877	slf_prepare_transfer (pTHX_ struct coro_transfer_args *ta)
		1878	{
		1879	SV **arg = (SV **)slf_frame.data;
		1880
		1881	prepare_transfer (aTHX_ ta, arg [0], arg [1]);
		1882	}
		1883
		1884	static void
		1885	slf_init_transfer (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		1886	{
		1887	if (items != 2)
		1888	croak ("Coro::State::transfer (prev, next) expects two arguments, not %d,", items);
		1889
		1890	frame->prepare = slf_prepare_transfer;
		1891	frame->check = slf_check_nop;
		1892	frame->data = (void *)arg; /* let's hope it will stay valid */
		1893	}
		1894
		1895	static void
		1896	slf_init_schedule (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		1897	{
		1898	frame->prepare = prepare_schedule;
		1899	frame->check = slf_check_nop;
		1900	}
		1901
		1902	static void
		1903	slf_init_cede (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		1904	{
		1905	frame->prepare = prepare_cede;
		1906	frame->check = slf_check_nop;
		1907	}
		1908
		1909	static void
		1910	slf_init_cede_notself (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		1911	{
		1912	frame->prepare = prepare_cede_notself;
		1913	frame->check = slf_check_nop;
		1914	}
		1915
		1916	/*
		1917	* these not obviously related functions are all rolled into one
		1918	* function to increase chances that they all will call transfer with the same
		1919	* stack offset
		1920	* SLF stands for "schedule-like-function".
		1921	*/
		1922	static OP *
		1923	pp_slf (pTHX)
		1924	{
		1925	I32 checkmark; /* mark SP to see how many elements check has pushed */
		1926
		1927	/* set up the slf frame, unless it has already been set-up */
		1928	/* the latter happens when a new coro has been started */
		1929	/* or when a new cctx was attached to an existing coroutine */
		1930	if (expect_true (!slf_frame.prepare))
		1931	{
		1932	/* first iteration */
		1933	dSP;
		1934	SV **arg = PL_stack_base + TOPMARK + 1;
		1935	int items = SP - arg; /* args without function object */
		1936	SV *gv = *sp;
		1937
		1938	/* do a quick consistency check on the "function" object, and if it isn't */
		1939	/* for us, divert to the real entersub */
		1940	if (SvTYPE (gv) != SVt_PVGV
		1941	|| !GvCV (gv)
		1942	|| !(CvFLAGS (GvCV (gv)) & CVf_SLF))
		1943	return PL_ppaddr[OP_ENTERSUB](aTHX);
		1944
		1945	if (!(PL_op->op_flags & OPf_STACKED))
		1946	{
		1947	/* ampersand-form of call, use @_ instead of stack */
		1948	AV *av = GvAV (PL_defgv);
		1949	arg = AvARRAY (av);
		1950	items = AvFILLp (av) + 1;
		1951	}
		1952
		1953	/* now call the init function, which needs to set up slf_frame */
		1954	((coro_slf_cb)CvXSUBANY (GvCV (gv)).any_ptr)
		1955	(aTHX_ &slf_frame, GvCV (gv), arg, items);
		1956
		1957	/* pop args */
		1958	SP = PL_stack_base + POPMARK;
		1959
		1960	PUTBACK;
		1961	}
		1962
		1963	/* now that we have a slf_frame, interpret it! */
		1964	/* we use a callback system not to make the code needlessly */
		1965	/* complicated, but so we can run multiple perl coros from one cctx */
		1966
		1967	do
		1968	{
		1969	struct coro_transfer_args ta;
		1970
		1971	slf_frame.prepare (aTHX_ &ta);
		1972	TRANSFER (ta, 0);
		1973
		1974	checkmark = PL_stack_sp - PL_stack_base;
		1975	}
		1976	while (slf_frame.check (aTHX_ &slf_frame));
		1977
		1978	slf_frame.prepare = 0; /* invalidate the frame, we are done processing it */
		1979
		1980	/* exception handling */
		1981	if (expect_false (CORO_THROW))
		1982	{
		1983	SV *exception = sv_2mortal (CORO_THROW);
		1984
		1985	CORO_THROW = 0;
		1986	sv_setsv (ERRSV, exception);
		1987	croak (0);
		1988	}
		1989
		1990	/* return value handling - mostly like entersub */
		1991	/* make sure we put something on the stack in scalar context */
		1992	if (GIMME_V == G_SCALAR)
		1993	{
		1994	dSP;
		1995	SV **bot = PL_stack_base + checkmark;
		1996
		1997	if (sp == bot) /* too few, push undef */
		1998	bot [1] = &PL_sv_undef;
		1999	else if (sp != bot + 1) /* too many, take last one */
		2000	bot [1] = *sp;
		2001
		2002	SP = bot + 1;
		2003
		2004	PUTBACK;
		2005	}
		2006
		2007	return NORMAL;
		2008	}
		2009
		2010	static void
		2011	api_execute_slf (pTHX_ CV *cv, coro_slf_cb init_cb, I32 ax)
		2012	{
		2013	int i;
		2014	SV **arg = PL_stack_base + ax;
		2015	int items = PL_stack_sp - arg + 1;
		2016
		2017	assert (("FATAL: SLF call with illegal CV value", !CvANON (cv)));
		2018
		2019	if (PL_op->op_ppaddr != PL_ppaddr [OP_ENTERSUB]
		2020	&& PL_op->op_ppaddr != pp_slf)
		2021	croak ("FATAL: Coro SLF calls can only be made normally, not via goto or any other means, caught");
		2022
		2023	CvFLAGS (cv) |= CVf_SLF;
		2024	CvXSUBANY (cv).any_ptr = (void *)init_cb;
		2025	slf_cv = cv;
		2026
		2027	/* we patch the op, and then re-run the whole call */
		2028	/* we have to put the same argument on the stack for this to work */
		2029	/* and this will be done by pp_restore */
		2030	slf_restore.op_next = (OP *)&slf_restore;
		2031	slf_restore.op_type = OP_CUSTOM;
		2032	slf_restore.op_ppaddr = pp_restore;
		2033	slf_restore.op_first = PL_op;
		2034
		2035	slf_ax = ax - 1; /* undo the ax++ inside dAXMARK */
		2036
		2037	if (PL_op->op_flags & OPf_STACKED)
		2038	{
		2039	if (items > slf_arga)
		2040	{
		2041	slf_arga = items;
		2042	free (slf_argv);
		2043	slf_argv = malloc (slf_arga * sizeof (SV *));
		2044	}
		2045
		2046	slf_argc = items;
		2047
		2048	for (i = 0; i < items; ++i)
		2049	slf_argv [i] = SvREFCNT_inc (arg [i]);
		2050	}
		2051	else
		2052	slf_argc = 0;
		2053
		2054	PL_op->op_ppaddr = pp_slf;
		2055	PL_op->op_type = OP_CUSTOM; /* maybe we should leave it at entersub? */
		2056
		2057	PL_op = (OP *)&slf_restore;
		2058	}
1742		2059
1743	/*****************************************************************************/	2060	/*****************************************************************************/
1744	/* PerlIO::cede */	2061	/* PerlIO::cede */
1745		2062
1746	typedef struct	2063	typedef struct
…		…
1814	PerlIOBuf_get_cnt,	2131	PerlIOBuf_get_cnt,
1815	PerlIOBuf_set_ptrcnt,	2132	PerlIOBuf_set_ptrcnt,
1816	};	2133	};
1817		2134
1818	/*****************************************************************************/	2135	/*****************************************************************************/
		2136	/* Coro::Semaphore & Coro::Signal */
1819		2137
1820	static const CV *slf_cv; /* for quick consistency check */
1821
1822	static UNOP slf_restore; /* restore stack as entersub did, for first-re-run */
1823	static SV *slf_arg0;
1824	static SV *slf_arg1;
1825	static SV *slf_arg2;
1826
1827	/* this restores the stack in the case we patched the entersub, to */
1828	/* recreate the stack frame as perl will on following calls */
1829	/* since entersub cleared the stack */
1830	static OP *	2138	static SV *
1831	pp_restore (pTHX)	2139	coro_waitarray_new (pTHX_ int count)
1832	{	2140	{
1833	dSP;	2141	/* a semaphore contains a counter IV in $sem->[0] and any waiters after that */
		2142	AV *av = newAV ();
		2143	SV **ary;
1834		2144
1835	PUSHMARK (SP);	2145	/* unfortunately, building manually saves memory */
		2146	Newx (ary, 2, SV *);
		2147	AvALLOC (av) = ary;
		2148	/*AvARRAY (av) = ary;*/
		2149	SvPVX ((SV *)av) = (char *)ary; /* 5.8.8 needs this syntax instead of AvARRAY = ary */
		2150	AvMAX (av) = 1;
		2151	AvFILLp (av) = 0;
		2152	ary [0] = newSViv (count);
1836		2153
1837	EXTEND (SP, 3);	2154	return newRV_noinc ((SV *)av);
1838	if (slf_arg0) PUSHs (sv_2mortal (slf_arg0));
1839	if (slf_arg1) PUSHs (sv_2mortal (slf_arg1));
1840	if (slf_arg2) PUSHs (sv_2mortal (slf_arg2));
1841	PUSHs ((SV *)CvGV (slf_cv));
1842
1843	RETURNOP (slf_restore.op_first);
1844	}	2155	}
1845		2156
1846	static void	2157	/* semaphore */
1847	slf_prepare_set_stacklevel (pTHX_ struct coro_transfer_args *ta)
1848	{
1849	prepare_set_stacklevel (ta, (struct coro_cctx *)slf_frame.data);
1850	}
1851		2158
1852	static void	2159	static void
1853	slf_init_set_stacklevel (pTHX_ struct CoroSLF *frame, SV **arg, int items)	2160	coro_semaphore_adjust (pTHX_ AV *av, IV adjust)
1854	{	2161	{
1855	assert (("FATAL: set_stacklevel needs the coro cctx as sole argument", items == 1));	2162	SV *count_sv = AvARRAY (av)[0];
		2163	IV count = SvIVX (count_sv);
1856		2164
1857	frame->prepare = slf_prepare_set_stacklevel;	2165	count += adjust;
1858	frame->check = slf_check_nop;	2166	SvIVX (count_sv) = count;
1859	frame->data = (void *)SvIV (arg [0]);
1860	}
1861		2167
1862	static void	2168	/* now wake up as many waiters as are expected to lock */
1863	slf_prepare_transfer (pTHX_ struct coro_transfer_args *ta)	2169	while (count > 0 && AvFILLp (av) > 0)
1864	{
1865	SV **arg = (SV **)slf_frame.data;
1866
1867	prepare_transfer (ta, arg [0], arg [1]);
1868	}
1869
1870	static void
1871	slf_init_transfer (pTHX_ struct CoroSLF *frame, SV **arg, int items)
1872	{
1873	if (items != 2)
1874	croak ("Coro::State::transfer (prev, next) expects two arguments, not %d,", items);
1875
1876	frame->prepare = slf_prepare_transfer;
1877	frame->check = slf_check_nop;
1878	frame->data = (void *)arg; /* let's hope it will stay valid */
1879	}
1880
1881	static void
1882	slf_init_schedule (pTHX_ struct CoroSLF *frame, SV **arg, int items)
1883	{
1884	frame->prepare = prepare_schedule;
1885	frame->check = slf_check_nop;
1886	}
1887
1888	static void
1889	slf_init_cede (pTHX_ struct CoroSLF *frame, SV **arg, int items)
1890	{
1891	frame->prepare = prepare_cede;
1892	frame->check = slf_check_nop;
1893	}
1894
1895	static void
1896	slf_init_cede_notself (pTHX_ struct CoroSLF *frame, SV **arg, int items)
1897	{
1898	frame->prepare = prepare_cede_notself;
1899	frame->check = slf_check_nop;
1900	}
1901
1902	/* we hijack an hopefully unused CV flag for our purposes */
1903	#define CVf_SLF 0x4000
1904
1905	/*
1906	* these not obviously related functions are all rolled into one
1907	* function to increase chances that they all will call transfer with the same
1908	* stack offset
1909	* SLF stands for "schedule-like-function".
1910	*/
1911	static OP *
1912	pp_slf (pTHX)
1913	{
1914	I32 checkmark; /* mark SP to see how many elements check has pushed */
1915
1916	/* set up the slf frame, unless it has already been set-up */
1917	/* the latter happens when a new coro has been started */
1918	/* or when a new cctx was attached to an existing coroutine */
1919	if (expect_true (!slf_frame.prepare))
1920	{
1921	/* first iteration */
1922	dSP;
1923	SV **arg = PL_stack_base + TOPMARK + 1;
1924	int items = SP - arg; /* args without function object */
1925	SV *gv = *sp;
1926
1927	/* do a quick consistency check on the "function" object, and if it isn't */
1928	/* for us, divert to the real entersub */
1929	if (SvTYPE (gv) != SVt_PVGV || !(CvFLAGS (GvCV (gv)) & CVf_SLF))
1930	return PL_ppaddr[OP_ENTERSUB](aTHX);
1931
1932	/* pop args */
1933	SP = PL_stack_base + POPMARK;
1934
1935	if (!(PL_op->op_flags & OPf_STACKED))
1936	{
1937	/* ampersand-form of call, use @_ instead of stack */
1938	AV *av = GvAV (PL_defgv);
1939	arg = AvARRAY (av);
1940	items = AvFILLp (av) + 1;
1941	}
1942
1943	PUTBACK;
1944
1945	((coro_slf_cb)CvXSUBANY (GvCV (gv)).any_ptr) (aTHX_ &slf_frame, arg, items);
1946	}
1947
1948	/* now interpret the slf_frame */
1949	/* we use a callback system not to make the code needlessly */
1950	/* complicated, but so we can run multiple perl coros from one cctx */
1951
1952	do
1953	{
1954	struct coro_transfer_args ta;
1955
1956	slf_frame.prepare (aTHX_ &ta);
1957	TRANSFER (ta, 0);
1958
1959	checkmark = PL_stack_sp - PL_stack_base;
1960	}
1961	while (slf_frame.check (aTHX_ &slf_frame));
1962
1963	{
1964	dSP;
1965	SV **bot = PL_stack_base + checkmark;
1966	int gimme = GIMME_V;
1967
1968	slf_frame.prepare = 0; /* signal pp_slf that we need a new frame */
1969
1970	/* make sure we put something on the stack in scalar context */
1971	if (gimme == G_SCALAR)
1972	{	2170	{
1973	if (sp == bot)	2171	SV *cb;
1974	XPUSHs (&PL_sv_undef);
1975		2172
1976	SP = bot + 1;	2173	/* swap first two elements so we can shift a waiter */
		2174	AvARRAY (av)[0] = AvARRAY (av)[1];
		2175	AvARRAY (av)[1] = count_sv;
		2176	cb = av_shift (av);
		2177
		2178	if (SvOBJECT (cb))
		2179	api_ready (aTHX_ cb);
		2180	else
		2181	croak ("callbacks not yet supported");
		2182
		2183	SvREFCNT_dec (cb);
		2184
		2185	--count;
1977	}	2186	}
1978
1979	PUTBACK;
1980	}
1981
1982	return NORMAL;
1983	}	2187	}
1984		2188
1985	static void	2189	static void
1986	api_execute_slf (pTHX_ CV *cv, coro_slf_cb init_cb, SV **arg, int items)	2190	coro_semaphore_on_destroy (pTHX_ struct coro *coro)
1987	{	2191	{
1988	assert (("FATAL: SLF call with illegal CV value", !CvANON (cv)));	2192	/* call $sem->adjust (0) to possibly wake up some other waiters */
1989		2193	coro_semaphore_adjust (aTHX_ (AV *)coro->slf_frame.data, 0);
1990	if (PL_op->op_ppaddr != PL_ppaddr [OP_ENTERSUB]
1991	&& PL_op->op_ppaddr != pp_slf)
1992	croak ("FATAL: Coro SLF calls can only be made normally, not via goto or any other means, caught");
1993
1994	if (items > 3)
1995	croak ("Coro only supports up to three arguments to SLF functions currently (not %d), caught", items);
1996
1997	CvFLAGS (cv) |= CVf_SLF;
1998	CvXSUBANY (cv).any_ptr = (void *)init_cb;
1999	slf_cv = cv;
2000
2001	/* we patch the op, and then re-run the whole call */
2002	/* we have to put the same argument on the stack for this to work */
2003	/* and this will be done by pp_restore */
2004	slf_restore.op_next = (OP *)&slf_restore;
2005	slf_restore.op_type = OP_NULL;
2006	slf_restore.op_ppaddr = pp_restore;
2007	slf_restore.op_first = PL_op;
2008
2009	slf_arg0 = items > 0 ? SvREFCNT_inc (arg [0]) : 0;
2010	slf_arg1 = items > 1 ? SvREFCNT_inc (arg [1]) : 0;
2011	slf_arg2 = items > 2 ? SvREFCNT_inc (arg [2]) : 0;
2012
2013	PL_op->op_ppaddr = pp_slf;
2014
2015	PL_op = (OP *)&slf_restore;
2016	}	2194	}
2017
2018	/*****************************************************************************/
2019		2195
2020	static int	2196	static int
2021	slf_check_semaphore_down (pTHX_ struct CoroSLF *frame)	2197	slf_check_semaphore_down_or_wait (pTHX_ struct CoroSLF *frame, int acquire)
2022	{	2198	{
2023	AV *av = (AV *)frame->data;	2199	AV *av = (AV *)frame->data;
2024	SV *count_sv = AvARRAY (av)[0];	2200	SV *count_sv = AvARRAY (av)[0];
2025		2201
		2202	/* if we are about to throw, don't actually acquire the lock, just throw */
		2203	if (CORO_THROW)
		2204	return 0;
2026	if (SvIVX (count_sv) > 0)	2205	else if (SvIVX (count_sv) > 0)
2027	{	2206	{
		2207	SvSTATE_current->on_destroy = 0;
		2208
		2209	if (acquire)
2028	SvIVX (count_sv) = SvIVX (count_sv) - 1;	2210	SvIVX (count_sv) = SvIVX (count_sv) - 1;
		2211	else
		2212	coro_semaphore_adjust (aTHX_ av, 0);
		2213
2029	return 0;	2214	return 0;
2030	}	2215	}
2031	else	2216	else
2032	{	2217	{
2033	int i;	2218	int i;
…		…
2042	av_push (av, SvREFCNT_inc (SvRV (coro_current)));	2227	av_push (av, SvREFCNT_inc (SvRV (coro_current)));
2043	return 1;	2228	return 1;
2044	}	2229	}
2045	}	2230	}
2046		2231
2047	static void	2232	static int
		2233	slf_check_semaphore_down (pTHX_ struct CoroSLF *frame)
		2234	{
		2235	return slf_check_semaphore_down_or_wait (aTHX_ frame, 1);
		2236	}
		2237
		2238	static int
		2239	slf_check_semaphore_wait (pTHX_ struct CoroSLF *frame)
		2240	{
		2241	return slf_check_semaphore_down_or_wait (aTHX_ frame, 0);
		2242	}
		2243
		2244	static void
2048	slf_init_semaphore_down (pTHX_ struct CoroSLF *frame, SV **arg, int items)	2245	slf_init_semaphore_down_or_wait (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
2049	{	2246	{
2050	AV *av = (AV *)SvRV (arg [0]);	2247	AV *av = (AV *)SvRV (arg [0]);
2051		2248
2052	if (SvIVX (AvARRAY (av)[0]) > 0)	2249	if (SvIVX (AvARRAY (av)[0]) > 0)
2053	{	2250	{
…		…
2058	{	2255	{
2059	av_push (av, SvREFCNT_inc (SvRV (coro_current)));	2256	av_push (av, SvREFCNT_inc (SvRV (coro_current)));
2060		2257
2061	frame->data = (void *)sv_2mortal (SvREFCNT_inc ((SV *)av));	2258	frame->data = (void *)sv_2mortal (SvREFCNT_inc ((SV *)av));
2062	frame->prepare = prepare_schedule;	2259	frame->prepare = prepare_schedule;
2063	}
2064		2260
		2261	/* to avoid race conditions when a woken-up coro gets terminated */
		2262	/* we arrange for a temporary on_destroy that calls adjust (0) */
		2263	SvSTATE_current->on_destroy = coro_semaphore_on_destroy;
		2264	}
		2265	}
		2266
		2267	static void
		2268	slf_init_semaphore_down (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		2269	{
		2270	slf_init_semaphore_down_or_wait (aTHX_ frame, cv, arg, items);
2065	frame->check = slf_check_semaphore_down;	2271	frame->check = slf_check_semaphore_down;
2066
2067	}	2272	}
		2273
		2274	static void
		2275	slf_init_semaphore_wait (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		2276	{
		2277	slf_init_semaphore_down_or_wait (aTHX_ frame, cv, arg, items);
		2278	frame->check = slf_check_semaphore_wait;
		2279	}
		2280
		2281	/* signal */
		2282
		2283	static void
		2284	coro_signal_wake (pTHX_ AV *av, int count)
		2285	{
		2286	SvIVX (AvARRAY (av)[0]) = 0;
		2287
		2288	/* now signal count waiters */
		2289	while (count > 0 && AvFILLp (av) > 0)
		2290	{
		2291	SV *cb;
		2292
		2293	/* swap first two elements so we can shift a waiter */
		2294	cb = AvARRAY (av)[0];
		2295	AvARRAY (av)[0] = AvARRAY (av)[1];
		2296	AvARRAY (av)[1] = cb;
		2297
		2298	cb = av_shift (av);
		2299
		2300	api_ready (aTHX_ cb);
		2301	sv_setiv (cb, 0); /* signal waiter */
		2302	SvREFCNT_dec (cb);
		2303
		2304	--count;
		2305	}
		2306	}
		2307
		2308	static int
		2309	slf_check_signal_wait (pTHX_ struct CoroSLF *frame)
		2310	{
		2311	/* if we are about to throw, also stop waiting */
		2312	return SvROK ((SV *)frame->data) && !CORO_THROW;
		2313	}
		2314
		2315	static void
		2316	slf_init_signal_wait (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		2317	{
		2318	AV *av = (AV *)SvRV (arg [0]);
		2319
		2320	if (SvIVX (AvARRAY (av)[0]))
		2321	{
		2322	SvIVX (AvARRAY (av)[0]) = 0;
		2323	frame->prepare = prepare_nop;
		2324	frame->check = slf_check_nop;
		2325	}
		2326	else
		2327	{
		2328	SV *waiter = newRV_inc (SvRV (coro_current)); /* owned by signal av */
		2329
		2330	av_push (av, waiter);
		2331
		2332	frame->data = (void *)sv_2mortal (SvREFCNT_inc_NN (waiter)); /* owned by process */
		2333	frame->prepare = prepare_schedule;
		2334	frame->check = slf_check_signal_wait;
		2335	}
		2336	}
		2337
		2338	/*****************************************************************************/
		2339	/* Coro::AIO */
		2340
		2341	#define CORO_MAGIC_type_aio PERL_MAGIC_ext
		2342
		2343	/* helper storage struct */
		2344	struct io_state
		2345	{
		2346	int errorno;
		2347	I32 laststype; /* U16 in 5.10.0 */
		2348	int laststatval;
		2349	Stat_t statcache;
		2350	};
		2351
		2352	static void
		2353	coro_aio_callback (pTHX_ CV *cv)
		2354	{
		2355	dXSARGS;
		2356	AV *state = (AV *)GENSUB_ARG;
		2357	SV *coro = av_pop (state);
		2358	SV *data_sv = newSV (sizeof (struct io_state));
		2359
		2360	av_extend (state, items);
		2361
		2362	sv_upgrade (data_sv, SVt_PV);
		2363	SvCUR_set (data_sv, sizeof (struct io_state));
		2364	SvPOK_only (data_sv);
		2365
		2366	{
		2367	struct io_state *data = (struct io_state *)SvPVX (data_sv);
		2368
		2369	data->errorno = errno;
		2370	data->laststype = PL_laststype;
		2371	data->laststatval = PL_laststatval;
		2372	data->statcache = PL_statcache;
		2373	}
		2374
		2375	/* now build the result vector out of all the parameters and the data_sv */
		2376	{
		2377	int i;
		2378
		2379	for (i = 0; i < items; ++i)
		2380	av_push (state, SvREFCNT_inc_NN (ST (i)));
		2381	}
		2382
		2383	av_push (state, data_sv);
		2384
		2385	api_ready (aTHX_ coro);
		2386	SvREFCNT_dec (coro);
		2387	SvREFCNT_dec ((AV *)state);
		2388	}
		2389
		2390	static int
		2391	slf_check_aio_req (pTHX_ struct CoroSLF *frame)
		2392	{
		2393	AV *state = (AV *)frame->data;
		2394
		2395	/* if we are about to throw, return early */
		2396	/* this does not cancel the aio request, but at least */
		2397	/* it quickly returns */
		2398	if (CORO_THROW)
		2399	return 0;
		2400
		2401	/* one element that is an RV? repeat! */
		2402	if (AvFILLp (state) == 0 && SvROK (AvARRAY (state)[0]))
		2403	return 1;
		2404
		2405	/* restore status */
		2406	{
		2407	SV *data_sv = av_pop (state);
		2408	struct io_state *data = (struct io_state *)SvPVX (data_sv);
		2409
		2410	errno = data->errorno;
		2411	PL_laststype = data->laststype;
		2412	PL_laststatval = data->laststatval;
		2413	PL_statcache = data->statcache;
		2414
		2415	SvREFCNT_dec (data_sv);
		2416	}
		2417
		2418	/* push result values */
		2419	{
		2420	dSP;
		2421	int i;
		2422
		2423	EXTEND (SP, AvFILLp (state) + 1);
		2424	for (i = 0; i <= AvFILLp (state); ++i)
		2425	PUSHs (sv_2mortal (SvREFCNT_inc_NN (AvARRAY (state)[i])));
		2426
		2427	PUTBACK;
		2428	}
		2429
		2430	return 0;
		2431	}
		2432
		2433	static void
		2434	slf_init_aio_req (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		2435	{
		2436	AV *state = (AV *)sv_2mortal ((SV *)newAV ());
		2437	SV *coro_hv = SvRV (coro_current);
		2438	struct coro *coro = SvSTATE_hv (coro_hv);
		2439
		2440	/* put our coroutine id on the state arg */
		2441	av_push (state, SvREFCNT_inc_NN (coro_hv));
		2442
		2443	/* first see whether we have a non-zero priority and set it as AIO prio */
		2444	if (coro->prio)
		2445	{
		2446	dSP;
		2447
		2448	static SV *prio_cv;
		2449	static SV *prio_sv;
		2450
		2451	if (expect_false (!prio_cv))
		2452	{
		2453	prio_cv = (SV *)get_cv ("IO::AIO::aioreq_pri", 0);
		2454	prio_sv = newSViv (0);
		2455	}
		2456
		2457	PUSHMARK (SP);
		2458	sv_setiv (prio_sv, coro->prio);
		2459	XPUSHs (prio_sv);
		2460
		2461	PUTBACK;
		2462	call_sv (prio_cv, G_VOID | G_DISCARD);
		2463	}
		2464
		2465	/* now call the original request */
		2466	{
		2467	dSP;
		2468	CV *req = (CV *)CORO_MAGIC_NN ((SV *)cv, CORO_MAGIC_type_aio)->mg_obj;
		2469	int i;
		2470
		2471	PUSHMARK (SP);
		2472
		2473	/* first push all args to the stack */
		2474	EXTEND (SP, items + 1);
		2475
		2476	for (i = 0; i < items; ++i)
		2477	PUSHs (arg [i]);
		2478
		2479	/* now push the callback closure */
		2480	PUSHs (sv_2mortal (gensub (aTHX_ coro_aio_callback, (void *)SvREFCNT_inc_NN ((SV *)state))));
		2481
		2482	/* now call the AIO function - we assume our request is uncancelable */
		2483	PUTBACK;
		2484	call_sv ((SV *)req, G_VOID | G_DISCARD);
		2485	}
		2486
		2487	/* now that the requets is going, we loop toll we have a result */
		2488	frame->data = (void *)state;
		2489	frame->prepare = prepare_schedule;
		2490	frame->check = slf_check_aio_req;
		2491	}
		2492
		2493	static void
		2494	coro_aio_req_xs (pTHX_ CV *cv)
		2495	{
		2496	dXSARGS;
		2497
		2498	CORO_EXECUTE_SLF_XS (slf_init_aio_req);
		2499
		2500	XSRETURN_EMPTY;
		2501	}
		2502
		2503	/*****************************************************************************/
2068		2504
2069	MODULE = Coro::State PACKAGE = Coro::State PREFIX = api_	2505	MODULE = Coro::State PACKAGE = Coro::State PREFIX = api_
2070		2506
2071	PROTOTYPES: DISABLE	2507	PROTOTYPES: DISABLE
2072		2508
2073	BOOT:	2509	BOOT:
2074	{	2510	{
2075	#ifdef USE_ITHREADS	2511	#ifdef USE_ITHREADS
2076	MUTEX_INIT (&coro_lock);
2077	# if CORO_PTHREAD	2512	# if CORO_PTHREAD
2078	coro_thx = PERL_GET_CONTEXT;	2513	coro_thx = PERL_GET_CONTEXT;
2079	# endif	2514	# endif
2080	#endif	2515	#endif
2081	BOOT_PAGESIZE;	2516	BOOT_PAGESIZE;
…		…
2101	main_mainstack = PL_mainstack;	2536	main_mainstack = PL_mainstack;
2102	main_top_env = PL_top_env;	2537	main_top_env = PL_top_env;
2103		2538
2104	while (main_top_env->je_prev)	2539	while (main_top_env->je_prev)
2105	main_top_env = main_top_env->je_prev;	2540	main_top_env = main_top_env->je_prev;
		2541
		2542	{
		2543	SV *slf = sv_2mortal (newSViv (PTR2IV (pp_slf)));
		2544
		2545	if (!PL_custom_op_names) PL_custom_op_names = newHV ();
		2546	hv_store_ent (PL_custom_op_names, slf,
		2547	newSVpv ("coro_slf", 0), 0);
		2548
		2549	if (!PL_custom_op_descs) PL_custom_op_descs = newHV ();
		2550	hv_store_ent (PL_custom_op_descs, slf,
		2551	newSVpv ("coro schedule like function", 0), 0);
		2552	}
2106		2553
2107	coroapi.ver = CORO_API_VERSION;	2554	coroapi.ver = CORO_API_VERSION;
2108	coroapi.rev = CORO_API_REVISION;	2555	coroapi.rev = CORO_API_REVISION;
2109		2556
2110	coroapi.transfer = api_transfer;	2557	coroapi.transfer = api_transfer;
…		…
2156	}	2603	}
2157	OUTPUT:	2604	OUTPUT:
2158	RETVAL	2605	RETVAL
2159		2606
2160	void	2607	void
2161	_set_stacklevel (...)
2162	CODE:
2163	api_execute_slf (aTHX_ cv, slf_init_set_stacklevel, &ST (0), items);
2164
2165	void
2166	transfer (...)	2608	transfer (...)
2167	PROTOTYPE: $$	2609	PROTOTYPE: $$
2168	CODE:	2610	CODE:
2169	api_execute_slf (aTHX_ cv, slf_init_transfer, &ST (0), items);	2611	CORO_EXECUTE_SLF_XS (slf_init_transfer);
2170		2612
2171	bool	2613	bool
2172	_destroy (SV *coro_sv)	2614	_destroy (SV *coro_sv)
2173	CODE:	2615	CODE:
2174	RETVAL = coro_state_destroy (aTHX_ SvSTATE (coro_sv));	2616	RETVAL = coro_state_destroy (aTHX_ SvSTATE (coro_sv));
…		…
2292		2734
2293	void	2735	void
2294	throw (Coro::State self, SV *throw = &PL_sv_undef)	2736	throw (Coro::State self, SV *throw = &PL_sv_undef)
2295	PROTOTYPE: $;$	2737	PROTOTYPE: $;$
2296	CODE:	2738	CODE:
		2739	{
		2740	struct coro *current = SvSTATE_current;
		2741	SV **throwp = self == current ? &CORO_THROW : &self->except;
2297	SvREFCNT_dec (self->throw);	2742	SvREFCNT_dec (*throwp);
2298	self->throw = SvOK (throw) ? newSVsv (throw) : 0;	2743	*throwp = SvOK (throw) ? newSVsv (throw) : 0;
		2744	}
2299		2745
2300	void	2746	void
2301	api_trace (SV *coro, int flags = CC_TRACE | CC_TRACE_SUB)	2747	api_trace (SV *coro, int flags = CC_TRACE | CC_TRACE_SUB)
2302	PROTOTYPE: $;$	2748	PROTOTYPE: $;$
2303	C_ARGS: aTHX_ coro, flags	2749	C_ARGS: aTHX_ coro, flags
…		…
2334		2780
2335	void	2781	void
2336	force_cctx ()	2782	force_cctx ()
2337	PROTOTYPE:	2783	PROTOTYPE:
2338	CODE:	2784	CODE:
2339	struct coro *coro = SvSTATE (coro_current);
2340	coro->cctx->idle_sp = 0;	2785	SvSTATE_current->cctx->idle_sp = 0;
2341		2786
2342	void	2787	void
2343	swap_defsv (Coro::State self)	2788	swap_defsv (Coro::State self)
2344	PROTOTYPE: $	2789	PROTOTYPE: $
2345	ALIAS:	2790	ALIAS:
…		…
2353	SV **dst = ix ? (SV **)&self->slot->defav : (SV **)&self->slot->defsv;	2798	SV **dst = ix ? (SV **)&self->slot->defav : (SV **)&self->slot->defsv;
2354		2799
2355	SV *tmp = *src; *src = *dst; *dst = tmp;	2800	SV *tmp = *src; *src = *dst; *dst = tmp;
2356	}	2801	}
2357		2802
		2803
2358	MODULE = Coro::State PACKAGE = Coro	2804	MODULE = Coro::State PACKAGE = Coro
2359		2805
2360	BOOT:	2806	BOOT:
2361	{	2807	{
2362	int i;	2808	int i;
…		…
2389	coroapi.ready = api_ready;	2835	coroapi.ready = api_ready;
2390	coroapi.is_ready = api_is_ready;	2836	coroapi.is_ready = api_is_ready;
2391	coroapi.nready = coro_nready;	2837	coroapi.nready = coro_nready;
2392	coroapi.current = coro_current;	2838	coroapi.current = coro_current;
2393		2839
2394	GCoroAPI = &coroapi;	2840	/*GCoroAPI = &coroapi;*/
2395	sv_setiv (sv, (IV)&coroapi);	2841	sv_setiv (sv, (IV)&coroapi);
2396	SvREADONLY_on (sv);	2842	SvREADONLY_on (sv);
2397	}	2843	}
2398	}	2844	}
2399		2845
2400	void	2846	void
2401	schedule (...)	2847	schedule (...)
2402	CODE:	2848	CODE:
2403	api_execute_slf (aTHX_ cv, slf_init_schedule, &ST (0), 0);	2849	CORO_EXECUTE_SLF_XS (slf_init_schedule);
2404		2850
2405	void	2851	void
2406	cede (...)	2852	cede (...)
2407	CODE:	2853	CODE:
2408	api_execute_slf (aTHX_ cv, slf_init_cede, &ST (0), 0);	2854	CORO_EXECUTE_SLF_XS (slf_init_cede);
2409		2855
2410	void	2856	void
2411	cede_notself (...)	2857	cede_notself (...)
2412	CODE:	2858	CODE:
2413	api_execute_slf (aTHX_ cv, slf_init_cede_notself, &ST (0), 0);	2859	CORO_EXECUTE_SLF_XS (slf_init_cede_notself);
2414		2860
2415	void	2861	void
2416	_set_current (SV *current)	2862	_set_current (SV *current)
2417	PROTOTYPE: $	2863	PROTOTYPE: $
2418	CODE:	2864	CODE:
…		…
2421		2867
2422	void	2868	void
2423	_set_readyhook (SV *hook)	2869	_set_readyhook (SV *hook)
2424	PROTOTYPE: $	2870	PROTOTYPE: $
2425	CODE:	2871	CODE:
2426	LOCK;
2427	SvREFCNT_dec (coro_readyhook);	2872	SvREFCNT_dec (coro_readyhook);
2428	coro_readyhook = SvOK (hook) ? newSVsv (hook) : 0;	2873	coro_readyhook = SvOK (hook) ? newSVsv (hook) : 0;
2429	UNLOCK;
2430		2874
2431	int	2875	int
2432	prio (Coro::State coro, int newprio = 0)	2876	prio (Coro::State coro, int newprio = 0)
2433	PROTOTYPE: $;$	2877	PROTOTYPE: $;$
2434	ALIAS:	2878	ALIAS:
…		…
2470	# for async_pool speedup	2914	# for async_pool speedup
2471	void	2915	void
2472	_pool_1 (SV *cb)	2916	_pool_1 (SV *cb)
2473	CODE:	2917	CODE:
2474	{	2918	{
2475	struct coro *coro = SvSTATE (coro_current);
2476	HV *hv = (HV *)SvRV (coro_current);	2919	HV *hv = (HV *)SvRV (coro_current);
		2920	struct coro *coro = SvSTATE_hv ((SV *)hv);
2477	AV *defav = GvAV (PL_defgv);	2921	AV *defav = GvAV (PL_defgv);
2478	SV *invoke = hv_delete (hv, "_invoke", sizeof ("_invoke") - 1, 0);	2922	SV *invoke = hv_delete (hv, "_invoke", sizeof ("_invoke") - 1, 0);
2479	AV *invoke_av;	2923	AV *invoke_av;
2480	int i, len;	2924	int i, len;
2481		2925
…		…
2508		2952
2509	void	2953	void
2510	_pool_2 (SV *cb)	2954	_pool_2 (SV *cb)
2511	CODE:	2955	CODE:
2512	{	2956	{
2513	struct coro *coro = SvSTATE (coro_current);	2957	HV *hv = (HV *)SvRV (coro_current);
		2958	struct coro *coro = SvSTATE_hv ((SV *)hv);
2514		2959
2515	sv_setsv (cb, &PL_sv_undef);	2960	sv_setsv (cb, &PL_sv_undef);
2516		2961
2517	SvREFCNT_dec ((SV *)PL_defoutgv); PL_defoutgv = (GV *)coro->saved_deffh;	2962	SvREFCNT_dec ((SV *)PL_defoutgv); PL_defoutgv = (GV *)coro->saved_deffh;
2518	coro->saved_deffh = 0;	2963	coro->saved_deffh = 0;
…		…
2525	SvREFCNT_dec (old);	2970	SvREFCNT_dec (old);
2526	croak ("\3async_pool terminate\2\n");	2971	croak ("\3async_pool terminate\2\n");
2527	}	2972	}
2528		2973
2529	av_clear (GvAV (PL_defgv));	2974	av_clear (GvAV (PL_defgv));
2530	hv_store ((HV *)SvRV (coro_current), "desc", sizeof ("desc") - 1,	2975	hv_store (hv, "desc", sizeof ("desc") - 1,
2531	newSVpvn ("[async_pool idle]", sizeof ("[async_pool idle]") - 1), 0);	2976	newSVpvn ("[async_pool idle]", sizeof ("[async_pool idle]") - 1), 0);
2532		2977
2533	coro->prio = 0;	2978	coro->prio = 0;
2534		2979
2535	if (coro->cctx && (coro->cctx->flags & CC_TRACE))	2980	if (coro->cctx && (coro->cctx->flags & CC_TRACE))
2536	api_trace (aTHX_ coro_current, 0);	2981	api_trace (aTHX_ coro_current, 0);
2537		2982
2538	av_push (av_async_pool, newSVsv (coro_current));	2983	av_push (av_async_pool, newSVsv (coro_current));
2539	}	2984	}
2540		2985
2541	#if 0
2542
2543	void
2544	_generator_call (...)
2545	PROTOTYPE: @
2546	PPCODE:
2547	fprintf (stderr, "call %p\n", CvXSUBANY(cv).any_ptr);
2548	xxxx
2549	abort ();
2550
2551	SV *	2986	SV *
2552	gensub (SV *sub, ...)	2987	rouse_cb ()
2553	PROTOTYPE: &;@	2988	PROTOTYPE:
2554	CODE:	2989	CODE:
2555	{	2990	RETVAL = coro_new_rouse_cb (aTHX);
2556	struct coro *coro;
2557	MAGIC *mg;
2558	CV *xcv;
2559	CV *ncv = (CV *)newSV_type (SVt_PVCV);
2560	int i;
2561
2562	CvGV (ncv) = CvGV (cv);
2563	CvFILE (ncv) = CvFILE (cv);
2564
2565	Newz (0, coro, 1, struct coro);
2566	coro->args = newAV ();
2567	coro->flags = CF_NEW;
2568
2569	av_extend (coro->args, items - 1);
2570	for (i = 1; i < items; i++)
2571	av_push (coro->args, newSVsv (ST (i)));
2572
2573	CvISXSUB_on (ncv);
2574	CvXSUBANY (ncv).any_ptr = (void *)coro;
2575
2576	xcv = GvCV (gv_fetchpv ("Coro::_generator_call", 0, SVt_PVCV));
2577
2578	CvXSUB (ncv) = CvXSUB (xcv);
2579	CvANON_on (ncv);
2580
2581	mg = sv_magicext ((SV *)ncv, 0, CORO_MAGIC_type_state, &coro_gensub_vtbl, (char *)coro, 0);
2582	RETVAL = newRV_noinc ((SV *)ncv);
2583	}
2584	OUTPUT:	2991	OUTPUT:
2585	RETVAL	2992	RETVAL
2586		2993
2587	#endif
2588
2589
2590	MODULE = Coro::State PACKAGE = Coro::AIO
2591
2592	void	2994	void
2593	_get_state (SV *self)	2995	rouse_wait (SV *cb = 0)
2594	PROTOTYPE: $	2996	PROTOTYPE: ;$
2595	PPCODE:	2997	PPCODE:
2596	{	2998	CORO_EXECUTE_SLF_XS (slf_init_rouse_wait);
2597	AV *defav = GvAV (PL_defgv);
2598	AV *av = newAV ();
2599	int i;
2600	SV *data_sv = newSV (sizeof (struct io_state));
2601	struct io_state *data = (struct io_state *)SvPVX (data_sv);
2602	SvCUR_set (data_sv, sizeof (struct io_state));
2603	SvPOK_only (data_sv);
2604
2605	data->errorno = errno;
2606	data->laststype = PL_laststype;
2607	data->laststatval = PL_laststatval;
2608	data->statcache = PL_statcache;
2609
2610	av_extend (av, AvFILLp (defav) + 1 + 1);
2611
2612	for (i = 0; i <= AvFILLp (defav); ++i)
2613	av_push (av, SvREFCNT_inc_NN (AvARRAY (defav)[i]));
2614
2615	av_push (av, data_sv);
2616
2617	XPUSHs (sv_2mortal (newRV_noinc ((SV *)av)));
2618
2619	api_ready (aTHX_ self);
2620	}
2621
2622	void
2623	_set_state (SV *state)
2624	PROTOTYPE: $
2625	PPCODE:
2626	{
2627	AV *av = (AV *)SvRV (state);
2628	struct io_state *data = (struct io_state *)SvPVX (AvARRAY (av)[AvFILLp (av)]);
2629	int i;
2630
2631	errno = data->errorno;
2632	PL_laststype = data->laststype;
2633	PL_laststatval = data->laststatval;
2634	PL_statcache = data->statcache;
2635
2636	EXTEND (SP, AvFILLp (av));
2637	for (i = 0; i < AvFILLp (av); ++i)
2638	PUSHs (sv_2mortal (SvREFCNT_inc_NN (AvARRAY (av)[i])));
2639	}
2640
2641
2642	MODULE = Coro::State PACKAGE = Coro::AnyEvent
2643
2644	BOOT:
2645	sv_activity = coro_get_sv (aTHX_ "Coro::AnyEvent::ACTIVITY", TRUE);
2646
2647	SV *
2648	_schedule (...)
2649	CODE:
2650	{
2651	static int incede;
2652
2653	api_cede_notself (aTHX);
2654
2655	++incede;
2656	while (coro_nready >= incede && api_cede (aTHX))
2657	;
2658
2659	sv_setsv (sv_activity, &PL_sv_undef);
2660	if (coro_nready >= incede)
2661	{
2662	PUSHMARK (SP);
2663	PUTBACK;
2664	call_pv ("Coro::AnyEvent::_activity", G_DISCARD | G_EVAL);
2665	SPAGAIN;
2666	}
2667
2668	--incede;
2669	}
2670		2999
2671		3000
2672	MODULE = Coro::State PACKAGE = PerlIO::cede	3001	MODULE = Coro::State PACKAGE = PerlIO::cede
2673		3002
2674	BOOT:	3003	BOOT:
2675	PerlIO_define_layer (aTHX_ &PerlIO_cede);	3004	PerlIO_define_layer (aTHX_ &PerlIO_cede);
2676		3005
		3006
2677	MODULE = Coro::State PACKAGE = Coro::Semaphore	3007	MODULE = Coro::State PACKAGE = Coro::Semaphore
2678		3008
2679	SV *	3009	SV *
2680	new (SV *klass, SV *count_ = 0)	3010	new (SV *klass, SV *count = 0)
2681	CODE:	3011	CODE:
2682	{	3012	RETVAL = sv_bless (
2683	/* a semaphore contains a counter IV in $sem->[0] and any waiters after that */	3013	coro_waitarray_new (aTHX_ count && SvOK (count) ? SvIV (count) : 1),
2684	AV *av = newAV ();	3014	GvSTASH (CvGV (cv))
2685	av_push (av, newSViv (count_ && SvOK (count_) ? SvIV (count_) : 1));	3015	);
2686	RETVAL = sv_bless (newRV_noinc ((SV *)av), GvSTASH (CvGV (cv)));	3016	OUTPUT:
2687	}	3017	RETVAL
		3018
		3019	# helper for Coro::Channel
		3020	SV *
		3021	_alloc (int count)
		3022	CODE:
		3023	RETVAL = coro_waitarray_new (aTHX_ count);
2688	OUTPUT:	3024	OUTPUT:
2689	RETVAL	3025	RETVAL
2690		3026
2691	SV *	3027	SV *
2692	count (SV *self)	3028	count (SV *self)
…		…
2698	void	3034	void
2699	up (SV *self, int adjust = 1)	3035	up (SV *self, int adjust = 1)
2700	ALIAS:	3036	ALIAS:
2701	adjust = 1	3037	adjust = 1
2702	CODE:	3038	CODE:
2703	{	3039	coro_semaphore_adjust (aTHX_ (AV *)SvRV (self), ix ? adjust : 1);
2704	AV *av = (AV *)SvRV (self);
2705	SV *count_sv = AvARRAY (av)[0];
2706	IV count = SvIVX (count_sv);
2707
2708	count += ix ? adjust : 1;
2709	SvIVX (count_sv) = count;
2710
2711	/* now wake up as many waiters as possible */
2712	while (count > 0 && AvFILLp (av) >= count)
2713	{
2714	SV *cb;
2715
2716	/* swap first two elements so we can shift a waiter */
2717	AvARRAY (av)[0] = AvARRAY (av)[1];
2718	AvARRAY (av)[1] = count_sv;
2719	cb = av_shift (av);
2720
2721	if (SvOBJECT (cb))
2722	api_ready (cb);
2723	else
2724	croak ("callbacks not yet supported");
2725
2726	SvREFCNT_dec (cb);
2727	}
2728	}
2729		3040
2730	void	3041	void
2731	down (SV *self)	3042	down (SV *self)
2732	CODE:	3043	CODE:
2733	api_execute_slf (aTHX_ cv, slf_init_semaphore_down, &ST (0), 1);	3044	CORO_EXECUTE_SLF_XS (slf_init_semaphore_down);
		3045
		3046	void
		3047	wait (SV *self)
		3048	CODE:
		3049	CORO_EXECUTE_SLF_XS (slf_init_semaphore_wait);
2734		3050
2735	void	3051	void
2736	try (SV *self)	3052	try (SV *self)
2737	PPCODE:	3053	PPCODE:
2738	{	3054	{
…		…
2750	XSRETURN_NO;	3066	XSRETURN_NO;
2751	}	3067	}
2752		3068
2753	void	3069	void
2754	waiters (SV *self)	3070	waiters (SV *self)
2755	CODE:	3071	PPCODE:
2756	{	3072	{
2757	AV *av = (AV *)SvRV (self);	3073	AV *av = (AV *)SvRV (self);
		3074	int wcount = AvFILLp (av) + 1 - 1;
2758		3075
2759	if (GIMME_V == G_SCALAR)	3076	if (GIMME_V == G_SCALAR)
2760	XPUSHs (sv_2mortal (newSVsv (AvARRAY (av)[0])));	3077	XPUSHs (sv_2mortal (newSViv (wcount)));
2761	else	3078	else
2762	{	3079	{
2763	int i;	3080	int i;
2764	EXTEND (SP, AvFILLp (av) + 1 - 1);	3081	EXTEND (SP, wcount);
2765	for (i = 1; i <= AvFILLp (av); ++i)	3082	for (i = 1; i <= wcount; ++i)
2766	PUSHs (newSVsv (AvARRAY (av)[i]));	3083	PUSHs (sv_2mortal (newRV_inc (AvARRAY (av)[i])));
2767	}	3084	}
2768	}	3085	}
2769		3086
		3087	MODULE = Coro::State PACKAGE = Coro::Signal
		3088
		3089	SV *
		3090	new (SV *klass)
		3091	CODE:
		3092	RETVAL = sv_bless (
		3093	coro_waitarray_new (aTHX_ 0),
		3094	GvSTASH (CvGV (cv))
		3095	);
		3096	OUTPUT:
		3097	RETVAL
		3098
		3099	void
		3100	wait (SV *self)
		3101	CODE:
		3102	CORO_EXECUTE_SLF_XS (slf_init_signal_wait);
		3103
		3104	void
		3105	broadcast (SV *self)
		3106	CODE:
		3107	{
		3108	AV *av = (AV *)SvRV (self);
		3109	coro_signal_wake (aTHX_ av, AvFILLp (av));
		3110	}
		3111
		3112	void
		3113	send (SV *self)
		3114	CODE:
		3115	{
		3116	AV *av = (AV *)SvRV (self);
		3117
		3118	if (AvFILLp (av))
		3119	coro_signal_wake (aTHX_ av, 1);
		3120	else
		3121	SvIVX (AvARRAY (av)[0]) = 1; /* remember the signal */
		3122	}
		3123
		3124	IV
		3125	awaited (SV *self)
		3126	CODE:
		3127	RETVAL = AvFILLp ((AV *)SvRV (self)) + 1 - 1;
		3128	OUTPUT:
		3129	RETVAL
		3130
		3131
		3132	MODULE = Coro::State PACKAGE = Coro::AnyEvent
		3133
		3134	BOOT:
		3135	sv_activity = coro_get_sv (aTHX_ "Coro::AnyEvent::ACTIVITY", TRUE);
		3136
		3137	void
		3138	_schedule (...)
		3139	CODE:
		3140	{
		3141	static int incede;
		3142
		3143	api_cede_notself (aTHX);
		3144
		3145	++incede;
		3146	while (coro_nready >= incede && api_cede (aTHX))
		3147	;
		3148
		3149	sv_setsv (sv_activity, &PL_sv_undef);
		3150	if (coro_nready >= incede)
		3151	{
		3152	PUSHMARK (SP);
		3153	PUTBACK;
		3154	call_pv ("Coro::AnyEvent::_activity", G_KEEPERR | G_EVAL | G_VOID | G_DISCARD);
		3155	}
		3156
		3157	--incede;
		3158	}
		3159
		3160
		3161	MODULE = Coro::State PACKAGE = Coro::AIO
		3162
		3163	void
		3164	_register (char *target, char *proto, SV *req)
		3165	CODE:
		3166	{
		3167	HV *st;
		3168	GV *gvp;
		3169	CV *req_cv = sv_2cv (req, &st, &gvp, 0);
		3170	/* newXSproto doesn't return the CV on 5.8 */
		3171	CV *slf_cv = newXS (target, coro_aio_req_xs, __FILE__);
		3172	sv_setpv ((SV *)slf_cv, proto);
		3173	sv_magicext ((SV *)slf_cv, (SV *)req_cv, CORO_MAGIC_type_aio, 0, 0, 0);
		3174	}
		3175

Diff Legend

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