ViewVC Help

View File | Revision Log | Show Annotations | Download File

/cvs/Coro/Coro/State.xs

Comparing Coro/Coro/State.xs (file contents):
Revision 1.282 by root, Sun Nov 16 10:33:08 2008 UTC vs.
Revision 1.308 by root, Wed Nov 19 15:09:57 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	# if CORO_PTHREAD	153	# if CORO_PTHREAD
148	static void *coro_thx;	154	static void *coro_thx;
149	# endif	155	# endif
150	#endif	156	#endif
151		157
152	/* helper storage struct for Coro::AIO */
153	struct io_state
154	{
155	AV *res;
156	int errorno;
157	I32 laststype; /* U16 in 5.10.0 */
158	int laststatval;
159	Stat_t statcache;
160	};
161
162	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);
163		163
164	static U32 cctx_gen;	164	static U32 cctx_gen;
165	static size_t cctx_stacksize = CORO_STACKSIZE;	165	static size_t cctx_stacksize = CORO_STACKSIZE;
166	static struct CoroAPI coroapi;	166	static struct CoroAPI coroapi;
167	static AV *main_mainstack; /* used to differentiate between $main and others */	167	static AV *main_mainstack; /* used to differentiate between $main and others */
168	static JMPENV *main_top_env;	168	static JMPENV *main_top_env;
169	static HV *coro_state_stash, *coro_stash;	169	static HV *coro_state_stash, *coro_stash;
170	static volatile SV *coro_mortal; /* will be freed/thrown after next transfer */	170	static volatile SV *coro_mortal; /* will be freed/thrown after next transfer */
171	static volatile struct coro *transfer_next;
172		171
173	static GV *irsgv; /* $/ */	172	static GV *irsgv; /* $/ */
174	static GV *stdoutgv; /* *STDOUT */	173	static GV *stdoutgv; /* *STDOUT */
175	static SV *rv_diehook;	174	static SV *rv_diehook;
176	static SV *rv_warnhook;	175	static SV *rv_warnhook;
…		…
242	/* this is a structure representing a perl-level coroutine */	241	/* this is a structure representing a perl-level coroutine */
243	struct coro {	242	struct coro {
244	/* the C coroutine allocated to this perl coroutine, if any */	243	/* the C coroutine allocated to this perl coroutine, if any */
245	coro_cctx *cctx;	244	coro_cctx *cctx;
246		245
247	/* process data */	246	/* state data */
248	struct CoroSLF slf_frame; /* saved slf frame */	247	struct CoroSLF slf_frame; /* saved slf frame */
249	AV *mainstack;	248	AV *mainstack;
250	perl_slots *slot; /* basically the saved sp */	249	perl_slots *slot; /* basically the saved sp */
251		250
		251	CV *startcv; /* the CV to execute */
252	AV *args; /* data associated with this coroutine (initial args) */	252	AV *args; /* data associated with this coroutine (initial args) */
253	int refcnt; /* coroutines are refcounted, yes */	253	int refcnt; /* coroutines are refcounted, yes */
254	int flags; /* CF_ flags */	254	int flags; /* CF_ flags */
255	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);	256	void (*on_destroy)(pTHX_ struct coro *coro);
257		257
258	/* statistics */	258	/* statistics */
259	int usecount; /* number of transfers to this coro */	259	int usecount; /* number of transfers to this coro */
260		260
261	/* coro process data */	261	/* coro process data */
262	int prio;	262	int prio;
263	SV *throw; /* exception to be thrown */	263	SV *except; /* exception to be thrown */
		264	SV *rouse_cb;
264		265
265	/* async_pool */	266	/* async_pool */
266	SV *saved_deffh;	267	SV *saved_deffh;
267		268
268	/* linked list */	269	/* linked list */
…		…
270	};	271	};
271		272
272	typedef struct coro *Coro__State;	273	typedef struct coro *Coro__State;
273	typedef struct coro *Coro__State_or_hashref;	274	typedef struct coro *Coro__State_or_hashref;
274		275
		276	/* the following variables are effectively part of the perl context */
		277	/* and get copied between struct coro and these variables */
		278	/* the mainr easonw e don't support windows process emulation */
275	static struct CoroSLF slf_frame; /* the current slf frame */	279	static struct CoroSLF slf_frame; /* the current slf frame */
276		280
277	/** Coro ********************************************************************/	281	/** Coro ********************************************************************/
278		282
279	#define PRIO_MAX 3	283	#define PRIO_MAX 3
…		…
318	#if PERL_VERSION_ATLEAST (5,10,0)	322	#if PERL_VERSION_ATLEAST (5,10,0)
319	/* silence stupid and wrong 5.10 warning that I am unable to switch off */	323	/* silence stupid and wrong 5.10 warning that I am unable to switch off */
320	get_hv (name, create);	324	get_hv (name, create);
321	#endif	325	#endif
322	return get_hv (name, create);	326	return get_hv (name, create);
		327	}
		328
		329	/* may croak */
		330	INLINE CV *
		331	coro_sv_2cv (SV *sv)
		332	{
		333	HV *st;
		334	GV *gvp;
		335	return sv_2cv (sv, &st, &gvp, 0);
323	}	336	}
324		337
325	static AV *	338	static AV *
326	coro_clone_padlist (pTHX_ CV *cv)	339	coro_clone_padlist (pTHX_ CV *cv)
327	{	340	{
…		…
381	SvREFCNT_dec (av); /* sv_magicext increased the refcount */	394	SvREFCNT_dec (av); /* sv_magicext increased the refcount */
382		395
383	return 0;	396	return 0;
384	}	397	}
385		398
386	#define CORO_MAGIC_type_cv PERL_MAGIC_ext	399	#define CORO_MAGIC_type_cv 26
387	#define CORO_MAGIC_type_state PERL_MAGIC_ext	400	#define CORO_MAGIC_type_state PERL_MAGIC_ext
388		401
389	static MGVTBL coro_cv_vtbl = {	402	static MGVTBL coro_cv_vtbl = {
390	0, 0, 0, 0,	403	0, 0, 0, 0,
391	coro_cv_free	404	coro_cv_free
392	};	405	};
393		406
		407	#define CORO_MAGIC_NN(sv, type) \
		408	(expect_true (SvMAGIC (sv)->mg_type == type) \
		409	? SvMAGIC (sv) \
		410	: mg_find (sv, type))
		411
394	#define CORO_MAGIC(sv, type) \	412	#define CORO_MAGIC(sv, type) \
395	expect_true (SvMAGIC (sv)) \	413	(expect_true (SvMAGIC (sv)) \
396	? expect_true (SvMAGIC (sv)->mg_type == type) \	414	? CORO_MAGIC_NN (sv, type) \
397	? SvMAGIC (sv) \
398	: mg_find (sv, type) \
399	: 0	415	: 0)
400		416
401	#define CORO_MAGIC_cv(cv) CORO_MAGIC (((SV *)(cv)), CORO_MAGIC_type_cv)	417	#define CORO_MAGIC_cv(cv) CORO_MAGIC (((SV *)(cv)), CORO_MAGIC_type_cv)
402	#define CORO_MAGIC_state(sv) CORO_MAGIC (((SV *)(sv)), CORO_MAGIC_type_state)	418	#define CORO_MAGIC_state(sv) CORO_MAGIC_NN (((SV *)(sv)), CORO_MAGIC_type_state)
403		419
404	INLINE struct coro *	420	INLINE struct coro *
405	SvSTATE_ (pTHX_ SV *coro)	421	SvSTATE_ (pTHX_ SV *coro)
406	{	422	{
407	HV *stash;	423	HV *stash;
…		…
425	return (struct coro *)mg->mg_ptr;	441	return (struct coro *)mg->mg_ptr;
426	}	442	}
427		443
428	#define SvSTATE(sv) SvSTATE_ (aTHX_ (sv))	444	#define SvSTATE(sv) SvSTATE_ (aTHX_ (sv))
429		445
430	/* fastert than SvSTATE, but expects a coroutine hv */	446	/* faster than SvSTATE, but expects a coroutine hv */
431	INLINE struct coro *	447	#define SvSTATE_hv(hv) ((struct coro *)CORO_MAGIC_NN ((SV *)hv, CORO_MAGIC_type_state)->mg_ptr)
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))	448	#define SvSTATE_current SvSTATE_hv (SvRV (coro_current))
442		449
443	/* the next two functions merely cache the padlists */	450	/* the next two functions merely cache the padlists */
444	static void	451	static void
445	get_padlist (pTHX_ CV *cv)	452	get_padlist (pTHX_ CV *cv)
…		…
452	else	459	else
453	{	460	{
454	#if CORO_PREFER_PERL_FUNCTIONS	461	#if CORO_PREFER_PERL_FUNCTIONS
455	/* this is probably cleaner? but also slower! */	462	/* this is probably cleaner? but also slower! */
456	/* in practise, it seems to be less stable */	463	/* in practise, it seems to be less stable */
457	CV *cp = Perl_cv_clone (cv);	464	CV *cp = Perl_cv_clone (aTHX_ cv);
458	CvPADLIST (cv) = CvPADLIST (cp);	465	CvPADLIST (cv) = CvPADLIST (cp);
459	CvPADLIST (cp) = 0;	466	CvPADLIST (cp) = 0;
460	SvREFCNT_dec (cp);	467	SvREFCNT_dec (cp);
461	#else	468	#else
462	CvPADLIST (cv) = coro_clone_padlist (aTHX_ cv);	469	CvPADLIST (cv) = coro_clone_padlist (aTHX_ cv);
…		…
514	}	521	}
515		522
516	PUTBACK;	523	PUTBACK;
517	}	524	}
518		525
519	slf_frame = c->slf_frame;	526	slf_frame = c->slf_frame;
		527	CORO_THROW = c->except;
520	}	528	}
521		529
522	static void	530	static void
523	save_perl (pTHX_ Coro__State c)	531	save_perl (pTHX_ Coro__State c)
524	{	532	{
		533	c->except = CORO_THROW;
525	c->slf_frame = slf_frame;	534	c->slf_frame = slf_frame;
526		535
527	{	536	{
528	dSP;	537	dSP;
529	I32 cxix = cxstack_ix;	538	I32 cxix = cxstack_ix;
…		…
604	* of perl.c:init_stacks, except that it uses less memory	613	* of perl.c:init_stacks, except that it uses less memory
605	* on the (sometimes correct) assumption that coroutines do	614	* on the (sometimes correct) assumption that coroutines do
606	* not usually need a lot of stackspace.	615	* not usually need a lot of stackspace.
607	*/	616	*/
608	#if CORO_PREFER_PERL_FUNCTIONS	617	#if CORO_PREFER_PERL_FUNCTIONS
609	# define coro_init_stacks init_stacks	618	# define coro_init_stacks(thx) init_stacks ()
610	#else	619	#else
611	static void	620	static void
612	coro_init_stacks (pTHX)	621	coro_init_stacks (pTHX)
613	{	622	{
614	PL_curstackinfo = new_stackinfo(32, 8);	623	PL_curstackinfo = new_stackinfo(32, 8);
…		…
822	slf_check_nop (pTHX_ struct CoroSLF *frame)	831	slf_check_nop (pTHX_ struct CoroSLF *frame)
823	{	832	{
824	return 0;	833	return 0;
825	}	834	}
826		835
827	static void	836	static UNOP coro_setup_op;
		837
		838	static void NOINLINE /* noinline to keep it out of the transfer fast path */
828	coro_setup (pTHX_ struct coro *coro)	839	coro_setup (pTHX_ struct coro *coro)
829	{	840	{
830	/*	841	/*
831	* emulate part of the perl startup here.	842	* emulate part of the perl startup here.
832	*/	843	*/
…		…
859	{	870	{
860	dSP;	871	dSP;
861	UNOP myop;	872	UNOP myop;
862		873
863	Zero (&myop, 1, UNOP);	874	Zero (&myop, 1, UNOP);
864	myop.op_next = Nullop;	875	myop.op_next = Nullop;
		876	myop.op_type = OP_ENTERSUB;
865	myop.op_flags = OPf_WANT_VOID;	877	myop.op_flags = OPf_WANT_VOID;
866		878
867	PUSHMARK (SP);	879	PUSHMARK (SP);
868	XPUSHs (sv_2mortal (av_shift (GvAV (PL_defgv))));	880	PUSHs ((SV *)coro->startcv);
869	PUTBACK;	881	PUTBACK;
870	PL_op = (OP *)&myop;	882	PL_op = (OP *)&myop;
871	PL_op = PL_ppaddr[OP_ENTERSUB](aTHX);	883	PL_op = PL_ppaddr[OP_ENTERSUB](aTHX);
872	SPAGAIN;
873	}	884	}
874		885
875	/* this newly created coroutine might be run on an existing cctx which most	886	/* this newly created coroutine might be run on an existing cctx which most
876	* likely was suspended in pp_slf, so we have to emulate entering pp_slf here.	887	* likely was suspended in pp_slf, so we have to emulate entering pp_slf here.
877	*/	888	*/
878	slf_frame.prepare = prepare_nop; /* provide a nop function for an eventual pp_slf */	889	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 */	890	slf_frame.check = slf_check_nop; /* signal pp_slf to not repeat */
		891
		892	/* and we have to provide the pp_slf op in any case, so pp_slf can skip it */
		893	coro_setup_op.op_next = PL_op;
		894	coro_setup_op.op_type = OP_CUSTOM;
		895	coro_setup_op.op_ppaddr = pp_slf;
		896	/* no flags etc. required, as an init function won't be called */
		897
		898	PL_op = (OP *)&coro_setup_op;
		899
		900	/* copy throw, in case it was set before coro_setup */
		901	CORO_THROW = coro->except;
880	}	902	}
881		903
882	static void	904	static void
883	coro_destruct (pTHX_ struct coro *coro)	905	coro_destruct (pTHX_ struct coro *coro)
884	{	906	{
…		…
908		930
909	SvREFCNT_dec (PL_diehook);	931	SvREFCNT_dec (PL_diehook);
910	SvREFCNT_dec (PL_warnhook);	932	SvREFCNT_dec (PL_warnhook);
911		933
912	SvREFCNT_dec (coro->saved_deffh);	934	SvREFCNT_dec (coro->saved_deffh);
913	SvREFCNT_dec (coro->throw);	935	SvREFCNT_dec (coro->rouse_cb);
914		936
915	coro_destruct_stacks (aTHX);	937	coro_destruct_stacks (aTHX);
916	}	938	}
917		939
918	INLINE void	940	INLINE void
…		…
1005	SAVETMPS;	1027	SAVETMPS;
1006	EXTEND (SP, 3);	1028	EXTEND (SP, 3);
1007	PUSHMARK (SP);	1029	PUSHMARK (SP);
1008	PUSHs (&PL_sv_yes);	1030	PUSHs (&PL_sv_yes);
1009	PUSHs (fullname);	1031	PUSHs (fullname);
1010	PUSHs (CxHASARGS (cx) ? sv_2mortal (newRV_inc ((SV *)cx->blk_sub.argarray)) : &PL_sv_undef);	1032	PUSHs (CxHASARGS (cx) ? sv_2mortal (newRV_inc ((SV *)cx->blk_sub.argarray)) : &PL_sv_undef);
1011	PUTBACK;	1033	PUTBACK;
1012	cb = hv_fetch ((HV *)SvRV (coro_current), "_trace_sub_cb", sizeof ("_trace_sub_cb") - 1, 0);	1034	cb = hv_fetch ((HV *)SvRV (coro_current), "_trace_sub_cb", sizeof ("_trace_sub_cb") - 1, 0);
1013	if (cb) call_sv (*cb, G_KEEPERR | G_EVAL | G_VOID | G_DISCARD);	1035	if (cb) call_sv (*cb, G_KEEPERR | G_EVAL | G_VOID | G_DISCARD);
1014	SPAGAIN;	1036	SPAGAIN;
1015	FREETMPS;	1037	FREETMPS;
…		…
1047		1069
1048	TAINT_NOT;	1070	TAINT_NOT;
1049	return 0;	1071	return 0;
1050	}	1072	}
1051		1073
		1074	static struct coro_cctx *cctx_ssl_cctx;
		1075	static struct CoroSLF cctx_ssl_frame;
		1076
1052	static void	1077	static void
1053	prepare_set_stacklevel (struct coro_transfer_args *ta, struct coro_cctx *cctx)	1078	slf_prepare_set_stacklevel (pTHX_ struct coro_transfer_args *ta)
1054	{	1079	{
1055	ta->prev = (struct coro *)cctx;	1080	ta->prev = (struct coro *)cctx_ssl_cctx;
1056	ta->next = 0;	1081	ta->next = 0;
1057	}	1082	}
1058		1083
1059	/* inject a fake call to Coro::State::_cctx_init into the execution */	1084	static int
1060	/* _cctx_init should be careful, as it could be called at almost any time */	1085	slf_check_set_stacklevel (pTHX_ struct CoroSLF *frame)
1061	/* during execution of a perl program */	1086	{
1062	/* also initialises PL_top_env */	1087	*frame = cctx_ssl_frame;
		1088
		1089	return frame->check (aTHX_ frame); /* execute the restored frame - there must be one */
		1090	}
		1091
		1092	/* initialises PL_top_env and injects a pseudo-slf-call to set the stacklevel */
1063	static void NOINLINE	1093	static void NOINLINE
1064	cctx_prepare (pTHX_ coro_cctx *cctx)	1094	cctx_prepare (pTHX_ coro_cctx *cctx)
1065	{	1095	{
1066	dSP;
1067	UNOP myop;
1068
1069	PL_top_env = &PL_start_env;	1096	PL_top_env = &PL_start_env;
1070		1097
1071	if (cctx->flags & CC_TRACE)	1098	if (cctx->flags & CC_TRACE)
1072	PL_runops = runops_trace;	1099	PL_runops = runops_trace;
1073		1100
1074	Zero (&myop, 1, UNOP);	1101	/* we already must be executing an SLF op, there is no other valid way
1075	myop.op_next = PL_op;	1102	* that can lead to creation of a new cctx */
1076	myop.op_flags = OPf_WANT_VOID | OPf_STACKED;	1103	assert (("FATAL: can't prepare slf-less cctx in Coro module (please report)",
		1104	slf_frame.prepare && PL_op->op_ppaddr == pp_slf));
1077		1105
1078	PUSHMARK (SP);	1106	/* we must emulate leaving pp_slf, which is done inside slf_check_set_stacklevel */
1079	EXTEND (SP, 2);	1107	cctx_ssl_cctx = cctx;
1080	PUSHs (sv_2mortal (newSViv ((IV)cctx)));	1108	cctx_ssl_frame = slf_frame;
1081	PUSHs ((SV *)get_cv ("Coro::State::_cctx_init", FALSE));	1109
1082	PUTBACK;	1110	slf_frame.prepare = slf_prepare_set_stacklevel;
1083	PL_op = (OP *)&myop;	1111	slf_frame.check = slf_check_set_stacklevel;
1084	PL_op = PL_ppaddr[OP_ENTERSUB](aTHX);
1085	SPAGAIN;
1086	}	1112	}
1087		1113
1088	/* the tail of transfer: execute stuff we can only do after a transfer */	1114	/* the tail of transfer: execute stuff we can only do after a transfer */
1089	INLINE void	1115	INLINE void
1090	transfer_tail (pTHX)	1116	transfer_tail (pTHX)
1091	{	1117	{
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);	1118	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	}	1119	}
1107		1120
1108	/*	1121	/*
1109	* this is a _very_ stripped down perl interpreter ;)	1122	* this is a _very_ stripped down perl interpreter ;)
1110	*/	1123	*/
…		…
1125		1138
1126	/* inject a fake subroutine call to cctx_init */	1139	/* inject a fake subroutine call to cctx_init */
1127	cctx_prepare (aTHX_ (coro_cctx *)arg);	1140	cctx_prepare (aTHX_ (coro_cctx *)arg);
1128		1141
1129	/* cctx_run is the alternative tail of transfer() */	1142	/* cctx_run is the alternative tail of transfer() */
1130	/* TODO: throwing an exception here might be deadly, VERIFY */
1131	transfer_tail (aTHX);	1143	transfer_tail (aTHX);
1132		1144
1133	/* somebody or something will hit me for both perl_run and PL_restartop */	1145	/* somebody or something will hit me for both perl_run and PL_restartop */
1134	PL_restartop = PL_op;	1146	PL_restartop = PL_op;
1135	perl_run (PL_curinterp);	1147	perl_run (PL_curinterp);
		1148	/*
		1149	* Unfortunately, there is no way to get at the return values of the
		1150	* coro body here, as perl_run destroys these
		1151	*/
1136		1152
1137	/*	1153	/*
1138	* If perl-run returns we assume exit() was being called or the coro	1154	* If perl-run returns we assume exit() was being called or the coro
1139	* fell off the end, which seems to be the only valid (non-bug)	1155	* fell off the end, which seems to be the only valid (non-bug)
1140	* reason for perl_run to return. We try to exit by jumping to the	1156	* reason for perl_run to return. We try to exit by jumping to the
…		…
1291	/** coroutine switching *****************************************************/	1307	/** coroutine switching *****************************************************/
1292		1308
1293	static void	1309	static void
1294	transfer_check (pTHX_ struct coro *prev, struct coro *next)	1310	transfer_check (pTHX_ struct coro *prev, struct coro *next)
1295	{	1311	{
		1312	/* TODO: throwing up here is considered harmful */
		1313
1296	if (expect_true (prev != next))	1314	if (expect_true (prev != next))
1297	{	1315	{
1298	if (expect_false (!(prev->flags & (CF_RUNNING | CF_NEW))))	1316	if (expect_false (!(prev->flags & (CF_RUNNING | CF_NEW))))
1299	croak ("Coro::State::transfer called with non-running/new prev Coro::State, but can only transfer from running or new states,");	1317	croak ("Coro::State::transfer called with non-running/new prev Coro::State, but can only transfer from running or new states,");
1300		1318
…		…
1310	#endif	1328	#endif
1311	}	1329	}
1312	}	1330	}
1313		1331
1314	/* always use the TRANSFER macro */	1332	/* always use the TRANSFER macro */
1315	static void NOINLINE	1333	static void NOINLINE /* noinline so we have a fixed stackframe */
1316	transfer (pTHX_ struct coro *prev, struct coro *next, int force_cctx)	1334	transfer (pTHX_ struct coro *prev, struct coro *next, int force_cctx)
1317	{	1335	{
1318	dSTACKLEVEL;	1336	dSTACKLEVEL;
1319		1337
1320	/* sometimes transfer is only called to set idle_sp */	1338	/* sometimes transfer is only called to set idle_sp */
1321	if (expect_false (!next))	1339	if (expect_false (!next))
1322	{	1340	{
1323	((coro_cctx *)prev)->idle_sp = (void *)stacklevel;	1341	((coro_cctx *)prev)->idle_sp = STACKLEVEL;
1324	assert (((coro_cctx *)prev)->idle_te = PL_top_env); /* just for the side-effect when asserts are enabled */	1342	assert (((coro_cctx *)prev)->idle_te = PL_top_env); /* just for the side-effect when asserts are enabled */
1325	}	1343	}
1326	else if (expect_true (prev != next))	1344	else if (expect_true (prev != next))
1327	{	1345	{
1328	coro_cctx *prev__cctx;	1346	coro_cctx *prev__cctx;
…		…
1353		1371
1354	prev__cctx = prev->cctx;	1372	prev__cctx = prev->cctx;
1355		1373
1356	/* possibly untie and reuse the cctx */	1374	/* possibly untie and reuse the cctx */
1357	if (expect_true (	1375	if (expect_true (
1358	prev__cctx->idle_sp == (void *)stacklevel	1376	prev__cctx->idle_sp == STACKLEVEL
1359	&& !(prev__cctx->flags & CC_TRACE)	1377	&& !(prev__cctx->flags & CC_TRACE)
1360	&& !force_cctx	1378	&& !force_cctx
1361	))	1379	))
1362	{	1380	{
1363	/* I assume that stacklevel is a stronger indicator than PL_top_env changes */	1381	/* I assume that stacklevel is a stronger indicator than PL_top_env changes */
…		…
1377	++next->usecount;	1395	++next->usecount;
1378		1396
1379	if (expect_true (!next->cctx))	1397	if (expect_true (!next->cctx))
1380	next->cctx = cctx_get (aTHX);	1398	next->cctx = cctx_get (aTHX);
1381		1399
1382	assert (("FATAL: transfer_next already nonzero in Coro (please report)", !transfer_next));
1383	transfer_next = next;
1384
1385	if (expect_false (prev__cctx != next->cctx))	1400	if (expect_false (prev__cctx != next->cctx))
1386	{	1401	{
1387	prev__cctx->top_env = PL_top_env;	1402	prev__cctx->top_env = PL_top_env;
1388	PL_top_env = next->cctx->top_env;	1403	PL_top_env = next->cctx->top_env;
1389	coro_transfer (&prev__cctx->cctx, &next->cctx->cctx);	1404	coro_transfer (&prev__cctx->cctx, &next->cctx->cctx);
…		…
1433		1448
1434	coro->slot = 0;	1449	coro->slot = 0;
1435	}	1450	}
1436		1451
1437	cctx_destroy (coro->cctx);	1452	cctx_destroy (coro->cctx);
		1453	SvREFCNT_dec (coro->startcv);
1438	SvREFCNT_dec (coro->args);	1454	SvREFCNT_dec (coro->args);
		1455	SvREFCNT_dec (CORO_THROW);
1439		1456
1440	if (coro->next) coro->next->prev = coro->prev;	1457	if (coro->next) coro->next->prev = coro->prev;
1441	if (coro->prev) coro->prev->next = coro->next;	1458	if (coro->prev) coro->prev->next = coro->next;
1442	if (coro == coro_first) coro_first = coro->next;	1459	if (coro == coro_first) coro_first = coro->next;
1443		1460
…		…
1497		1514
1498	prepare_transfer (aTHX_ &ta, prev_sv, next_sv);	1515	prepare_transfer (aTHX_ &ta, prev_sv, next_sv);
1499	TRANSFER (ta, 1);	1516	TRANSFER (ta, 1);
1500	}	1517	}
1501		1518
		1519	/*****************************************************************************/
		1520	/* gensub: simple closure generation utility */
		1521
		1522	#define GENSUB_ARG CvXSUBANY (cv).any_ptr
		1523
		1524	/* create a closure from XS, returns a code reference */
		1525	/* the arg can be accessed via GENSUB_ARG from the callback */
		1526	/* the callback must use dXSARGS/XSRETURN */
		1527	static SV *
		1528	gensub (pTHX_ void (*xsub)(pTHX_ CV *), void *arg)
		1529	{
		1530	CV *cv = (CV *)newSV (0);
		1531
		1532	sv_upgrade ((SV *)cv, SVt_PVCV);
		1533
		1534	CvANON_on (cv);
		1535	CvISXSUB_on (cv);
		1536	CvXSUB (cv) = xsub;
		1537	GENSUB_ARG = arg;
		1538
		1539	return newRV_noinc ((SV *)cv);
		1540	}
		1541
1502	/** Coro ********************************************************************/	1542	/** Coro ********************************************************************/
1503		1543
1504	INLINE void	1544	INLINE void
1505	coro_enq (pTHX_ struct coro *coro)	1545	coro_enq (pTHX_ struct coro *coro)
1506	{	1546	{
…		…
1549	ENTER;	1589	ENTER;
1550	SAVETMPS;	1590	SAVETMPS;
1551		1591
1552	PUSHMARK (SP);	1592	PUSHMARK (SP);
1553	PUTBACK;	1593	PUTBACK;
1554	call_sv (sv_hook, G_DISCARD);	1594	call_sv (sv_hook, G_VOID | G_DISCARD);
1555	SPAGAIN;
1556		1595
1557	FREETMPS;	1596	FREETMPS;
1558	LEAVE;	1597	LEAVE;
1559	}	1598	}
1560		1599
…		…
1587	ENTER;	1626	ENTER;
1588	SAVETMPS;	1627	SAVETMPS;
1589		1628
1590	PUSHMARK (SP);	1629	PUSHMARK (SP);
1591	PUTBACK;	1630	PUTBACK;
1592	call_sv (get_sv ("Coro::idle", FALSE), G_DISCARD);	1631	call_sv (get_sv ("Coro::idle", FALSE), G_VOID | G_DISCARD);
1593	SPAGAIN;
1594		1632
1595	FREETMPS;	1633	FREETMPS;
1596	LEAVE;	1634	LEAVE;
1597	continue;	1635	continue;
1598	}	1636	}
…		…
1705	if (coro->flags & CF_RUNNING)	1743	if (coro->flags & CF_RUNNING)
1706	PL_runops = RUNOPS_DEFAULT;	1744	PL_runops = RUNOPS_DEFAULT;
1707	else	1745	else
1708	coro->slot->runops = RUNOPS_DEFAULT;	1746	coro->slot->runops = RUNOPS_DEFAULT;
1709	}	1747	}
		1748	}
		1749
		1750	/*****************************************************************************/
		1751	/* rouse callback */
		1752
		1753	#define CORO_MAGIC_type_rouse PERL_MAGIC_ext
		1754
		1755	static void
		1756	coro_rouse_callback (pTHX_ CV *cv)
		1757	{
		1758	dXSARGS;
		1759	SV *data = (SV *)GENSUB_ARG;
		1760
		1761	if (SvTYPE (SvRV (data)) != SVt_PVAV)
		1762	{
		1763	/* first call, set args */
		1764	int i;
		1765	AV *av = newAV ();
		1766	SV *coro = SvRV (data);
		1767
		1768	SvRV_set (data, (SV *)av);
		1769	api_ready (aTHX_ coro);
		1770	SvREFCNT_dec (coro);
		1771
		1772	/* better take a full copy of the arguments */
		1773	while (items--)
		1774	av_store (av, items, newSVsv (ST (items)));
		1775	}
		1776
		1777	XSRETURN_EMPTY;
		1778	}
		1779
		1780	static int
		1781	slf_check_rouse_wait (pTHX_ struct CoroSLF *frame)
		1782	{
		1783	SV *data = (SV *)frame->data;
		1784
		1785	if (CORO_THROW)
		1786	return 0;
		1787
		1788	if (SvTYPE (SvRV (data)) != SVt_PVAV)
		1789	return 1;
		1790
		1791	/* now push all results on the stack */
		1792	{
		1793	dSP;
		1794	AV *av = (AV *)SvRV (data);
		1795	int i;
		1796
		1797	EXTEND (SP, AvFILLp (av) + 1);
		1798	for (i = 0; i <= AvFILLp (av); ++i)
		1799	PUSHs (sv_2mortal (AvARRAY (av)[i]));
		1800
		1801	/* we have stolen the elements, so ste length to zero and free */
		1802	AvFILLp (av) = -1;
		1803	av_undef (av);
		1804
		1805	PUTBACK;
		1806	}
		1807
		1808	return 0;
		1809	}
		1810
		1811	static void
		1812	slf_init_rouse_wait (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		1813	{
		1814	SV *cb;
		1815
		1816	if (items)
		1817	cb = arg [0];
		1818	else
		1819	{
		1820	struct coro *coro = SvSTATE_current;
		1821
		1822	if (!coro->rouse_cb)
		1823	croak ("Coro::rouse_wait called without rouse callback, and no default rouse callback found either,");
		1824
		1825	cb = sv_2mortal (coro->rouse_cb);
		1826	coro->rouse_cb = 0;
		1827	}
		1828
		1829	if (!SvROK (cb)
		1830	|| SvTYPE (SvRV (cb)) != SVt_PVCV
		1831	|| CvXSUB ((CV *)SvRV (cb)) != coro_rouse_callback)
		1832	croak ("Coro::rouse_wait called with illegal callback argument,");
		1833
		1834	{
		1835	CV *cv = (CV *)SvRV (cb); /* for GENSUB_ARG */
		1836	SV *data = (SV *)GENSUB_ARG;
		1837
		1838	frame->data = (void *)data;
		1839	frame->prepare = SvTYPE (SvRV (data)) == SVt_PVAV ? prepare_nop : prepare_schedule;
		1840	frame->check = slf_check_rouse_wait;
		1841	}
		1842	}
		1843
		1844	static SV *
		1845	coro_new_rouse_cb (pTHX)
		1846	{
		1847	HV *hv = (HV *)SvRV (coro_current);
		1848	struct coro *coro = SvSTATE_hv (hv);
		1849	SV *data = newRV_inc ((SV *)hv);
		1850	SV *cb = gensub (aTHX_ coro_rouse_callback, (void *)data);
		1851
		1852	sv_magicext (SvRV (cb), data, CORO_MAGIC_type_rouse, 0, 0, 0);
		1853	SvREFCNT_dec (data); /* magicext increases the refcount */
		1854
		1855	SvREFCNT_dec (coro->rouse_cb);
		1856	coro->rouse_cb = SvREFCNT_inc_NN (cb);
		1857
		1858	return cb;
		1859	}
		1860
		1861	/*****************************************************************************/
		1862	/* schedule-like-function opcode (SLF) */
		1863
		1864	static UNOP slf_restore; /* restore stack as entersub did, for first-re-run */
		1865	static const CV *slf_cv;
		1866	static SV **slf_argv;
		1867	static int slf_argc, slf_arga; /* count, allocated */
		1868	static I32 slf_ax; /* top of stack, for restore */
		1869
		1870	/* this restores the stack in the case we patched the entersub, to */
		1871	/* recreate the stack frame as perl will on following calls */
		1872	/* since entersub cleared the stack */
		1873	static OP *
		1874	pp_restore (pTHX)
		1875	{
		1876	int i;
		1877	SV **SP = PL_stack_base + slf_ax;
		1878
		1879	PUSHMARK (SP);
		1880
		1881	EXTEND (SP, slf_argc + 1);
		1882
		1883	for (i = 0; i < slf_argc; ++i)
		1884	PUSHs (sv_2mortal (slf_argv [i]));
		1885
		1886	PUSHs ((SV *)CvGV (slf_cv));
		1887
		1888	RETURNOP (slf_restore.op_first);
		1889	}
		1890
		1891	static void
		1892	slf_prepare_transfer (pTHX_ struct coro_transfer_args *ta)
		1893	{
		1894	SV **arg = (SV **)slf_frame.data;
		1895
		1896	prepare_transfer (aTHX_ ta, arg [0], arg [1]);
		1897	}
		1898
		1899	static void
		1900	slf_init_transfer (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		1901	{
		1902	if (items != 2)
		1903	croak ("Coro::State::transfer (prev, next) expects two arguments, not %d,", items);
		1904
		1905	frame->prepare = slf_prepare_transfer;
		1906	frame->check = slf_check_nop;
		1907	frame->data = (void *)arg; /* let's hope it will stay valid */
		1908	}
		1909
		1910	static void
		1911	slf_init_schedule (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		1912	{
		1913	frame->prepare = prepare_schedule;
		1914	frame->check = slf_check_nop;
		1915	}
		1916
		1917	static void
		1918	slf_init_cede (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		1919	{
		1920	frame->prepare = prepare_cede;
		1921	frame->check = slf_check_nop;
		1922	}
		1923
		1924	static void
		1925	slf_init_cede_notself (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		1926	{
		1927	frame->prepare = prepare_cede_notself;
		1928	frame->check = slf_check_nop;
		1929	}
		1930
		1931	/*
		1932	* these not obviously related functions are all rolled into one
		1933	* function to increase chances that they all will call transfer with the same
		1934	* stack offset
		1935	* SLF stands for "schedule-like-function".
		1936	*/
		1937	static OP *
		1938	pp_slf (pTHX)
		1939	{
		1940	I32 checkmark; /* mark SP to see how many elements check has pushed */
		1941
		1942	/* set up the slf frame, unless it has already been set-up */
		1943	/* the latter happens when a new coro has been started */
		1944	/* or when a new cctx was attached to an existing coroutine */
		1945	if (expect_true (!slf_frame.prepare))
		1946	{
		1947	/* first iteration */
		1948	dSP;
		1949	SV **arg = PL_stack_base + TOPMARK + 1;
		1950	int items = SP - arg; /* args without function object */
		1951	SV *gv = *sp;
		1952
		1953	/* do a quick consistency check on the "function" object, and if it isn't */
		1954	/* for us, divert to the real entersub */
		1955	if (SvTYPE (gv) != SVt_PVGV
		1956	|| !GvCV (gv)
		1957	|| !(CvFLAGS (GvCV (gv)) & CVf_SLF))
		1958	return PL_ppaddr[OP_ENTERSUB](aTHX);
		1959
		1960	if (!(PL_op->op_flags & OPf_STACKED))
		1961	{
		1962	/* ampersand-form of call, use @_ instead of stack */
		1963	AV *av = GvAV (PL_defgv);
		1964	arg = AvARRAY (av);
		1965	items = AvFILLp (av) + 1;
		1966	}
		1967
		1968	/* now call the init function, which needs to set up slf_frame */
		1969	((coro_slf_cb)CvXSUBANY (GvCV (gv)).any_ptr)
		1970	(aTHX_ &slf_frame, GvCV (gv), arg, items);
		1971
		1972	/* pop args */
		1973	SP = PL_stack_base + POPMARK;
		1974
		1975	PUTBACK;
		1976	}
		1977
		1978	/* now that we have a slf_frame, interpret it! */
		1979	/* we use a callback system not to make the code needlessly */
		1980	/* complicated, but so we can run multiple perl coros from one cctx */
		1981
		1982	do
		1983	{
		1984	struct coro_transfer_args ta;
		1985
		1986	slf_frame.prepare (aTHX_ &ta);
		1987	TRANSFER (ta, 0);
		1988
		1989	checkmark = PL_stack_sp - PL_stack_base;
		1990	}
		1991	while (slf_frame.check (aTHX_ &slf_frame));
		1992
		1993	slf_frame.prepare = 0; /* invalidate the frame, we are done processing it */
		1994
		1995	/* exception handling */
		1996	if (expect_false (CORO_THROW))
		1997	{
		1998	SV *exception = sv_2mortal (CORO_THROW);
		1999
		2000	CORO_THROW = 0;
		2001	sv_setsv (ERRSV, exception);
		2002	croak (0);
		2003	}
		2004
		2005	/* return value handling - mostly like entersub */
		2006	/* make sure we put something on the stack in scalar context */
		2007	if (GIMME_V == G_SCALAR)
		2008	{
		2009	dSP;
		2010	SV **bot = PL_stack_base + checkmark;
		2011
		2012	if (sp == bot) /* too few, push undef */
		2013	bot [1] = &PL_sv_undef;
		2014	else if (sp != bot + 1) /* too many, take last one */
		2015	bot [1] = *sp;
		2016
		2017	SP = bot + 1;
		2018
		2019	PUTBACK;
		2020	}
		2021
		2022	return NORMAL;
		2023	}
		2024
		2025	static void
		2026	api_execute_slf (pTHX_ CV *cv, coro_slf_cb init_cb, I32 ax)
		2027	{
		2028	int i;
		2029	SV **arg = PL_stack_base + ax;
		2030	int items = PL_stack_sp - arg + 1;
		2031
		2032	assert (("FATAL: SLF call with illegal CV value", !CvANON (cv)));
		2033
		2034	if (PL_op->op_ppaddr != PL_ppaddr [OP_ENTERSUB]
		2035	&& PL_op->op_ppaddr != pp_slf)
		2036	croak ("FATAL: Coro SLF calls can only be made normally, not via goto or any other means, caught");
		2037
		2038	CvFLAGS (cv) |= CVf_SLF;
		2039	CvXSUBANY (cv).any_ptr = (void *)init_cb;
		2040	slf_cv = cv;
		2041
		2042	/* we patch the op, and then re-run the whole call */
		2043	/* we have to put the same argument on the stack for this to work */
		2044	/* and this will be done by pp_restore */
		2045	slf_restore.op_next = (OP *)&slf_restore;
		2046	slf_restore.op_type = OP_CUSTOM;
		2047	slf_restore.op_ppaddr = pp_restore;
		2048	slf_restore.op_first = PL_op;
		2049
		2050	slf_ax = ax - 1; /* undo the ax++ inside dAXMARK */
		2051
		2052	if (PL_op->op_flags & OPf_STACKED)
		2053	{
		2054	if (items > slf_arga)
		2055	{
		2056	slf_arga = items;
		2057	free (slf_argv);
		2058	slf_argv = malloc (slf_arga * sizeof (SV *));
		2059	}
		2060
		2061	slf_argc = items;
		2062
		2063	for (i = 0; i < items; ++i)
		2064	slf_argv [i] = SvREFCNT_inc (arg [i]);
		2065	}
		2066	else
		2067	slf_argc = 0;
		2068
		2069	PL_op->op_ppaddr = pp_slf;
		2070	PL_op->op_type = OP_CUSTOM; /* maybe we should leave it at entersub? */
		2071
		2072	PL_op = (OP *)&slf_restore;
1710	}	2073	}
1711		2074
1712	/*****************************************************************************/	2075	/*****************************************************************************/
1713	/* PerlIO::cede */	2076	/* PerlIO::cede */
1714		2077
…		…
1783	PerlIOBuf_get_cnt,	2146	PerlIOBuf_get_cnt,
1784	PerlIOBuf_set_ptrcnt,	2147	PerlIOBuf_set_ptrcnt,
1785	};	2148	};
1786		2149
1787	/*****************************************************************************/	2150	/*****************************************************************************/
		2151	/* Coro::Semaphore & Coro::Signal */
1788		2152
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 *	2153	static SV *
1800	pp_restore (pTHX)	2154	coro_waitarray_new (pTHX_ int count)
1801	{	2155	{
1802	dSP;	2156	/* a semaphore contains a counter IV in $sem->[0] and any waiters after that */
		2157	AV *av = newAV ();
		2158	SV **ary;
1803		2159
1804	PUSHMARK (SP);	2160	/* unfortunately, building manually saves memory */
		2161	Newx (ary, 2, SV *);
		2162	AvALLOC (av) = ary;
		2163	/*AvARRAY (av) = ary;*/
		2164	SvPVX ((SV *)av) = (char *)ary; /* 5.8.8 needs this syntax instead of AvARRAY = ary */
		2165	AvMAX (av) = 1;
		2166	AvFILLp (av) = 0;
		2167	ary [0] = newSViv (count);
1805		2168
1806	EXTEND (SP, 3);	2169	return newRV_noinc ((SV *)av);
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	}	2170	}
1814		2171
1815	static void	2172	/* semaphore */
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		2173
1991	static void	2174	static void
1992	coro_semaphore_adjust (pTHX_ AV *av, IV adjust)	2175	coro_semaphore_adjust (pTHX_ AV *av, IV adjust)
1993	{	2176	{
1994	SV *count_sv = AvARRAY (av)[0];	2177	SV *count_sv = AvARRAY (av)[0];
…		…
2006	AvARRAY (av)[0] = AvARRAY (av)[1];	2189	AvARRAY (av)[0] = AvARRAY (av)[1];
2007	AvARRAY (av)[1] = count_sv;	2190	AvARRAY (av)[1] = count_sv;
2008	cb = av_shift (av);	2191	cb = av_shift (av);
2009		2192
2010	if (SvOBJECT (cb))	2193	if (SvOBJECT (cb))
		2194	{
2011	api_ready (aTHX_ cb);	2195	api_ready (aTHX_ cb);
2012	else	2196	--count;
2013	croak ("callbacks not yet supported");	2197	}
		2198	else if (SvTYPE (cb) == SVt_PVCV)
		2199	{
		2200	dSP;
		2201	PUSHMARK (SP);
		2202	XPUSHs (sv_2mortal (newRV_inc ((SV *)av)));
		2203	PUTBACK;
		2204	call_sv (cb, G_VOID | G_DISCARD | G_EVAL | G_KEEPERR);
		2205	}
2014		2206
2015	SvREFCNT_dec (cb);	2207	SvREFCNT_dec (cb);
2016
2017	--count;
2018	}	2208	}
2019	}	2209	}
2020		2210
2021	static void	2211	static void
2022	coro_semaphore_on_destroy (pTHX_ struct coro *coro)	2212	coro_semaphore_on_destroy (pTHX_ struct coro *coro)
…		…
2024	/* call $sem->adjust (0) to possibly wake up some other waiters */	2214	/* call $sem->adjust (0) to possibly wake up some other waiters */
2025	coro_semaphore_adjust (aTHX_ (AV *)coro->slf_frame.data, 0);	2215	coro_semaphore_adjust (aTHX_ (AV *)coro->slf_frame.data, 0);
2026	}	2216	}
2027		2217
2028	static int	2218	static int
2029	slf_check_semaphore_down (pTHX_ struct CoroSLF *frame)	2219	slf_check_semaphore_down_or_wait (pTHX_ struct CoroSLF *frame, int acquire)
2030	{	2220	{
2031	AV *av = (AV *)frame->data;	2221	AV *av = (AV *)frame->data;
2032	SV *count_sv = AvARRAY (av)[0];	2222	SV *count_sv = AvARRAY (av)[0];
2033		2223
		2224	/* if we are about to throw, don't actually acquire the lock, just throw */
		2225	if (CORO_THROW)
		2226	return 0;
2034	if (SvIVX (count_sv) > 0)	2227	else if (SvIVX (count_sv) > 0)
2035	{	2228	{
2036	SvSTATE_current->on_destroy = 0;	2229	SvSTATE_current->on_destroy = 0;
		2230
		2231	if (acquire)
2037	SvIVX (count_sv) = SvIVX (count_sv) - 1;	2232	SvIVX (count_sv) = SvIVX (count_sv) - 1;
		2233	else
		2234	coro_semaphore_adjust (aTHX_ av, 0);
		2235
2038	return 0;	2236	return 0;
2039	}	2237	}
2040	else	2238	else
2041	{	2239	{
2042	int i;	2240	int i;
…		…
2051	av_push (av, SvREFCNT_inc (SvRV (coro_current)));	2249	av_push (av, SvREFCNT_inc (SvRV (coro_current)));
2052	return 1;	2250	return 1;
2053	}	2251	}
2054	}	2252	}
2055		2253
2056	static void	2254	static int
		2255	slf_check_semaphore_down (pTHX_ struct CoroSLF *frame)
		2256	{
		2257	return slf_check_semaphore_down_or_wait (aTHX_ frame, 1);
		2258	}
		2259
		2260	static int
		2261	slf_check_semaphore_wait (pTHX_ struct CoroSLF *frame)
		2262	{
		2263	return slf_check_semaphore_down_or_wait (aTHX_ frame, 0);
		2264	}
		2265
		2266	static void
2057	slf_init_semaphore_down (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)	2267	slf_init_semaphore_down_or_wait (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
2058	{	2268	{
2059	AV *av = (AV *)SvRV (arg [0]);	2269	AV *av = (AV *)SvRV (arg [0]);
2060		2270
2061	if (SvIVX (AvARRAY (av)[0]) > 0)	2271	if (SvIVX (AvARRAY (av)[0]) > 0)
2062	{	2272	{
2063	frame->data = (void *)av;	2273	frame->data = (void *)av;
2064	frame->prepare = prepare_nop;	2274	frame->prepare = prepare_nop;
2065	SvSTATE_current->on_destroy = coro_semaphore_on_destroy;
2066	}	2275	}
2067	else	2276	else
2068	{	2277	{
2069	av_push (av, SvREFCNT_inc (SvRV (coro_current)));	2278	av_push (av, SvREFCNT_inc (SvRV (coro_current)));
2070		2279
2071	frame->data = (void *)sv_2mortal (SvREFCNT_inc ((SV *)av));	2280	frame->data = (void *)sv_2mortal (SvREFCNT_inc ((SV *)av));
2072	frame->prepare = prepare_schedule;	2281	frame->prepare = prepare_schedule;
2073		2282
2074	/* to avoid race conditions when a woken-up coro gets terminated */	2283	/* to avoid race conditions when a woken-up coro gets terminated */
2075	/* we arrange for a temporary on_destroy that calls adjust (0) */	2284	/* we arrange for a temporary on_destroy that calls adjust (0) */
2076	assert (!SvSTATE_current->on_destroy);//D
2077	SvSTATE_current->on_destroy = coro_semaphore_on_destroy;	2285	SvSTATE_current->on_destroy = coro_semaphore_on_destroy;
2078	}	2286	}
		2287	}
2079		2288
		2289	static void
		2290	slf_init_semaphore_down (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		2291	{
		2292	slf_init_semaphore_down_or_wait (aTHX_ frame, cv, arg, items);
2080	frame->check = slf_check_semaphore_down;	2293	frame->check = slf_check_semaphore_down;
		2294	}
2081		2295
		2296	static void
		2297	slf_init_semaphore_wait (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		2298	{
		2299	if (items >= 2)
		2300	{
		2301	/* callback form */
		2302	AV *av = (AV *)SvRV (arg [0]);
		2303	CV *cb_cv = coro_sv_2cv (arg [1]);
		2304
		2305	av_push (av, (SV *)SvREFCNT_inc_NN (cb_cv));
		2306
		2307	if (SvIVX (AvARRAY (av)[0]) > 0)
		2308	coro_semaphore_adjust (aTHX_ av, 0);
		2309
		2310	frame->prepare = prepare_nop;
		2311	frame->check = slf_check_nop;
		2312	}
		2313	else
		2314	{
		2315	slf_init_semaphore_down_or_wait (aTHX_ frame, cv, arg, items);
		2316	frame->check = slf_check_semaphore_wait;
		2317	}
		2318	}
		2319
		2320	/* signal */
		2321
		2322	static void
		2323	coro_signal_wake (pTHX_ AV *av, int count)
		2324	{
		2325	SvIVX (AvARRAY (av)[0]) = 0;
		2326
		2327	/* now signal count waiters */
		2328	while (count > 0 && AvFILLp (av) > 0)
		2329	{
		2330	SV *cb;
		2331
		2332	/* swap first two elements so we can shift a waiter */
		2333	cb = AvARRAY (av)[0];
		2334	AvARRAY (av)[0] = AvARRAY (av)[1];
		2335	AvARRAY (av)[1] = cb;
		2336
		2337	cb = av_shift (av);
		2338
		2339	api_ready (aTHX_ cb);
		2340	sv_setiv (cb, 0); /* signal waiter */
		2341	SvREFCNT_dec (cb);
		2342
		2343	--count;
		2344	}
		2345	}
		2346
		2347	static int
		2348	slf_check_signal_wait (pTHX_ struct CoroSLF *frame)
		2349	{
		2350	/* if we are about to throw, also stop waiting */
		2351	return SvROK ((SV *)frame->data) && !CORO_THROW;
		2352	}
		2353
		2354	static void
		2355	slf_init_signal_wait (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		2356	{
		2357	AV *av = (AV *)SvRV (arg [0]);
		2358
		2359	if (SvIVX (AvARRAY (av)[0]))
		2360	{
		2361	SvIVX (AvARRAY (av)[0]) = 0;
		2362	frame->prepare = prepare_nop;
		2363	frame->check = slf_check_nop;
		2364	}
		2365	else
		2366	{
		2367	SV *waiter = newRV_inc (SvRV (coro_current)); /* owned by signal av */
		2368
		2369	av_push (av, waiter);
		2370
		2371	frame->data = (void *)sv_2mortal (SvREFCNT_inc_NN (waiter)); /* owned by process */
		2372	frame->prepare = prepare_schedule;
		2373	frame->check = slf_check_signal_wait;
		2374	}
2082	}	2375	}
2083		2376
2084	/*****************************************************************************/	2377	/*****************************************************************************/
		2378	/* Coro::AIO */
2085		2379
2086	#define GENSUB_ARG CvXSUBANY (cv).any_ptr	2380	#define CORO_MAGIC_type_aio PERL_MAGIC_ext
2087		2381
2088	/* create a closure from XS, returns a code reference */	2382	/* helper storage struct */
2089	/* the arg can be accessed via GENSUB_ARG from the callback */	2383	struct io_state
2090	/* the callback must use dXSARGS/XSRETURN */
2091	static SV *
2092	gensub (pTHX_ void (*xsub)(pTHX_ CV *), void *arg)
2093	{	2384	{
2094	CV *cv = (CV *)NEWSV (0, 0);	2385	int errorno;
		2386	I32 laststype; /* U16 in 5.10.0 */
		2387	int laststatval;
		2388	Stat_t statcache;
		2389	};
2095		2390
		2391	static void
		2392	coro_aio_callback (pTHX_ CV *cv)
		2393	{
		2394	dXSARGS;
		2395	AV *state = (AV *)GENSUB_ARG;
		2396	SV *coro = av_pop (state);
		2397	SV *data_sv = newSV (sizeof (struct io_state));
		2398
		2399	av_extend (state, items);
		2400
2096	sv_upgrade ((SV *)cv, SVt_PVCV);	2401	sv_upgrade (data_sv, SVt_PV);
		2402	SvCUR_set (data_sv, sizeof (struct io_state));
		2403	SvPOK_only (data_sv);
2097		2404
2098	CvANON_on (cv);	2405	{
2099	CvISXSUB_on (cv);	2406	struct io_state *data = (struct io_state *)SvPVX (data_sv);
2100	CvXSUB (cv) = xsub;
2101	GENSUB_ARG = arg;
2102		2407
2103	return newRV_noinc ((SV *)cv);	2408	data->errorno = errno;
		2409	data->laststype = PL_laststype;
		2410	data->laststatval = PL_laststatval;
		2411	data->statcache = PL_statcache;
		2412	}
		2413
		2414	/* now build the result vector out of all the parameters and the data_sv */
		2415	{
		2416	int i;
		2417
		2418	for (i = 0; i < items; ++i)
		2419	av_push (state, SvREFCNT_inc_NN (ST (i)));
		2420	}
		2421
		2422	av_push (state, data_sv);
		2423
		2424	api_ready (aTHX_ coro);
		2425	SvREFCNT_dec (coro);
		2426	SvREFCNT_dec ((AV *)state);
		2427	}
		2428
		2429	static int
		2430	slf_check_aio_req (pTHX_ struct CoroSLF *frame)
		2431	{
		2432	AV *state = (AV *)frame->data;
		2433
		2434	/* if we are about to throw, return early */
		2435	/* this does not cancel the aio request, but at least */
		2436	/* it quickly returns */
		2437	if (CORO_THROW)
		2438	return 0;
		2439
		2440	/* one element that is an RV? repeat! */
		2441	if (AvFILLp (state) == 0 && SvROK (AvARRAY (state)[0]))
		2442	return 1;
		2443
		2444	/* restore status */
		2445	{
		2446	SV *data_sv = av_pop (state);
		2447	struct io_state *data = (struct io_state *)SvPVX (data_sv);
		2448
		2449	errno = data->errorno;
		2450	PL_laststype = data->laststype;
		2451	PL_laststatval = data->laststatval;
		2452	PL_statcache = data->statcache;
		2453
		2454	SvREFCNT_dec (data_sv);
		2455	}
		2456
		2457	/* push result values */
		2458	{
		2459	dSP;
		2460	int i;
		2461
		2462	EXTEND (SP, AvFILLp (state) + 1);
		2463	for (i = 0; i <= AvFILLp (state); ++i)
		2464	PUSHs (sv_2mortal (SvREFCNT_inc_NN (AvARRAY (state)[i])));
		2465
		2466	PUTBACK;
		2467	}
		2468
		2469	return 0;
		2470	}
		2471
		2472	static void
		2473	slf_init_aio_req (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		2474	{
		2475	AV *state = (AV *)sv_2mortal ((SV *)newAV ());
		2476	SV *coro_hv = SvRV (coro_current);
		2477	struct coro *coro = SvSTATE_hv (coro_hv);
		2478
		2479	/* put our coroutine id on the state arg */
		2480	av_push (state, SvREFCNT_inc_NN (coro_hv));
		2481
		2482	/* first see whether we have a non-zero priority and set it as AIO prio */
		2483	if (coro->prio)
		2484	{
		2485	dSP;
		2486
		2487	static SV *prio_cv;
		2488	static SV *prio_sv;
		2489
		2490	if (expect_false (!prio_cv))
		2491	{
		2492	prio_cv = (SV *)get_cv ("IO::AIO::aioreq_pri", 0);
		2493	prio_sv = newSViv (0);
		2494	}
		2495
		2496	PUSHMARK (SP);
		2497	sv_setiv (prio_sv, coro->prio);
		2498	XPUSHs (prio_sv);
		2499
		2500	PUTBACK;
		2501	call_sv (prio_cv, G_VOID | G_DISCARD);
		2502	}
		2503
		2504	/* now call the original request */
		2505	{
		2506	dSP;
		2507	CV *req = (CV *)CORO_MAGIC_NN ((SV *)cv, CORO_MAGIC_type_aio)->mg_obj;
		2508	int i;
		2509
		2510	PUSHMARK (SP);
		2511
		2512	/* first push all args to the stack */
		2513	EXTEND (SP, items + 1);
		2514
		2515	for (i = 0; i < items; ++i)
		2516	PUSHs (arg [i]);
		2517
		2518	/* now push the callback closure */
		2519	PUSHs (sv_2mortal (gensub (aTHX_ coro_aio_callback, (void *)SvREFCNT_inc_NN ((SV *)state))));
		2520
		2521	/* now call the AIO function - we assume our request is uncancelable */
		2522	PUTBACK;
		2523	call_sv ((SV *)req, G_VOID | G_DISCARD);
		2524	}
		2525
		2526	/* now that the requets is going, we loop toll we have a result */
		2527	frame->data = (void *)state;
		2528	frame->prepare = prepare_schedule;
		2529	frame->check = slf_check_aio_req;
		2530	}
		2531
		2532	static void
		2533	coro_aio_req_xs (pTHX_ CV *cv)
		2534	{
		2535	dXSARGS;
		2536
		2537	CORO_EXECUTE_SLF_XS (slf_init_aio_req);
		2538
		2539	XSRETURN_EMPTY;
2104	}	2540	}
2105		2541
2106	/*****************************************************************************/	2542	/*****************************************************************************/
2107		2543
2108	MODULE = Coro::State PACKAGE = Coro::State PREFIX = api_	2544	MODULE = Coro::State PACKAGE = Coro::State PREFIX = api_
…		…
2183	CODE:	2619	CODE:
2184	{	2620	{
2185	struct coro *coro;	2621	struct coro *coro;
2186	MAGIC *mg;	2622	MAGIC *mg;
2187	HV *hv;	2623	HV *hv;
		2624	CV *cb;
2188	int i;	2625	int i;
		2626
		2627	if (items > 1)
		2628	{
		2629	cb = coro_sv_2cv (ST (1));
		2630
		2631	if (CvISXSUB (cb))
		2632	croak ("Coro::State doesn't support XS functions as coroutine start, caught");
		2633
		2634	if (!CvROOT (cb))
		2635	croak ("Coro::State doesn't support autoloaded or undefined functions as coroutine start, caught");
		2636	}
2189		2637
2190	Newz (0, coro, 1, struct coro);	2638	Newz (0, coro, 1, struct coro);
2191	coro->args = newAV ();	2639	coro->args = newAV ();
2192	coro->flags = CF_NEW;	2640	coro->flags = CF_NEW;
2193		2641
…		…
2198	coro->hv = hv = newHV ();	2646	coro->hv = hv = newHV ();
2199	mg = sv_magicext ((SV *)hv, 0, CORO_MAGIC_type_state, &coro_state_vtbl, (char *)coro, 0);	2647	mg = sv_magicext ((SV *)hv, 0, CORO_MAGIC_type_state, &coro_state_vtbl, (char *)coro, 0);
2200	mg->mg_flags |= MGf_DUP;	2648	mg->mg_flags |= MGf_DUP;
2201	RETVAL = sv_bless (newRV_noinc ((SV *)hv), gv_stashpv (klass, 1));	2649	RETVAL = sv_bless (newRV_noinc ((SV *)hv), gv_stashpv (klass, 1));
2202		2650
		2651	if (items > 1)
		2652	{
		2653	coro->startcv = SvREFCNT_inc_NN (cb);
		2654
2203	av_extend (coro->args, items - 1);	2655	av_extend (coro->args, items - 1);
2204	for (i = 1; i < items; i++)	2656	for (i = 2; i < items; i++)
2205	av_push (coro->args, newSVsv (ST (i)));	2657	av_push (coro->args, newSVsv (ST (i)));
		2658	}
2206	}	2659	}
2207	OUTPUT:	2660	OUTPUT:
2208	RETVAL	2661	RETVAL
2209
2210	void
2211	_set_stacklevel (...)
2212	CODE:
2213	api_execute_slf (aTHX_ cv, slf_init_set_stacklevel, &ST (0), items);
2214		2662
2215	void	2663	void
2216	transfer (...)	2664	transfer (...)
2217	PROTOTYPE: $$	2665	PROTOTYPE: $$
2218	CODE:	2666	CODE:
2219	api_execute_slf (aTHX_ cv, slf_init_transfer, &ST (0), items);	2667	CORO_EXECUTE_SLF_XS (slf_init_transfer);
2220		2668
2221	bool	2669	bool
2222	_destroy (SV *coro_sv)	2670	_destroy (SV *coro_sv)
2223	CODE:	2671	CODE:
2224	RETVAL = coro_state_destroy (aTHX_ SvSTATE (coro_sv));	2672	RETVAL = coro_state_destroy (aTHX_ SvSTATE (coro_sv));
…		…
2342		2790
2343	void	2791	void
2344	throw (Coro::State self, SV *throw = &PL_sv_undef)	2792	throw (Coro::State self, SV *throw = &PL_sv_undef)
2345	PROTOTYPE: $;$	2793	PROTOTYPE: $;$
2346	CODE:	2794	CODE:
		2795	{
		2796	struct coro *current = SvSTATE_current;
		2797	SV **throwp = self == current ? &CORO_THROW : &self->except;
2347	SvREFCNT_dec (self->throw);	2798	SvREFCNT_dec (*throwp);
2348	self->throw = SvOK (throw) ? newSVsv (throw) : 0;	2799	*throwp = SvOK (throw) ? newSVsv (throw) : 0;
		2800	}
2349		2801
2350	void	2802	void
2351	api_trace (SV *coro, int flags = CC_TRACE | CC_TRACE_SUB)	2803	api_trace (SV *coro, int flags = CC_TRACE | CC_TRACE_SUB)
2352	PROTOTYPE: $;$	2804	PROTOTYPE: $;$
2353	C_ARGS: aTHX_ coro, flags	2805	C_ARGS: aTHX_ coro, flags
…		…
2402	SV **dst = ix ? (SV **)&self->slot->defav : (SV **)&self->slot->defsv;	2854	SV **dst = ix ? (SV **)&self->slot->defav : (SV **)&self->slot->defsv;
2403		2855
2404	SV *tmp = *src; *src = *dst; *dst = tmp;	2856	SV *tmp = *src; *src = *dst; *dst = tmp;
2405	}	2857	}
2406		2858
		2859
2407	MODULE = Coro::State PACKAGE = Coro	2860	MODULE = Coro::State PACKAGE = Coro
2408		2861
2409	BOOT:	2862	BOOT:
2410	{	2863	{
2411	int i;	2864	int i;
…		…
2438	coroapi.ready = api_ready;	2891	coroapi.ready = api_ready;
2439	coroapi.is_ready = api_is_ready;	2892	coroapi.is_ready = api_is_ready;
2440	coroapi.nready = coro_nready;	2893	coroapi.nready = coro_nready;
2441	coroapi.current = coro_current;	2894	coroapi.current = coro_current;
2442		2895
2443	GCoroAPI = &coroapi;	2896	/*GCoroAPI = &coroapi;*/
2444	sv_setiv (sv, (IV)&coroapi);	2897	sv_setiv (sv, (IV)&coroapi);
2445	SvREADONLY_on (sv);	2898	SvREADONLY_on (sv);
2446	}	2899	}
2447	}	2900	}
2448		2901
2449	void	2902	void
2450	schedule (...)	2903	schedule (...)
2451	CODE:	2904	CODE:
2452	api_execute_slf (aTHX_ cv, slf_init_schedule, &ST (0), 0);	2905	CORO_EXECUTE_SLF_XS (slf_init_schedule);
2453		2906
2454	void	2907	void
2455	cede (...)	2908	cede (...)
2456	CODE:	2909	CODE:
2457	api_execute_slf (aTHX_ cv, slf_init_cede, &ST (0), 0);	2910	CORO_EXECUTE_SLF_XS (slf_init_cede);
2458		2911
2459	void	2912	void
2460	cede_notself (...)	2913	cede_notself (...)
2461	CODE:	2914	CODE:
2462	api_execute_slf (aTHX_ cv, slf_init_cede_notself, &ST (0), 0);	2915	CORO_EXECUTE_SLF_XS (slf_init_cede_notself);
2463		2916
2464	void	2917	void
2465	_set_current (SV *current)	2918	_set_current (SV *current)
2466	PROTOTYPE: $	2919	PROTOTYPE: $
2467	CODE:	2920	CODE:
…		…
2555		3008
2556	void	3009	void
2557	_pool_2 (SV *cb)	3010	_pool_2 (SV *cb)
2558	CODE:	3011	CODE:
2559	{	3012	{
2560	struct coro *coro = SvSTATE_current;	3013	HV *hv = (HV *)SvRV (coro_current);
		3014	struct coro *coro = SvSTATE_hv ((SV *)hv);
2561		3015
2562	sv_setsv (cb, &PL_sv_undef);	3016	sv_setsv (cb, &PL_sv_undef);
2563		3017
2564	SvREFCNT_dec ((SV *)PL_defoutgv); PL_defoutgv = (GV *)coro->saved_deffh;	3018	SvREFCNT_dec ((SV *)PL_defoutgv); PL_defoutgv = (GV *)coro->saved_deffh;
2565	coro->saved_deffh = 0;	3019	coro->saved_deffh = 0;
…		…
2572	SvREFCNT_dec (old);	3026	SvREFCNT_dec (old);
2573	croak ("\3async_pool terminate\2\n");	3027	croak ("\3async_pool terminate\2\n");
2574	}	3028	}
2575		3029
2576	av_clear (GvAV (PL_defgv));	3030	av_clear (GvAV (PL_defgv));
2577	hv_store ((HV *)SvRV (coro_current), "desc", sizeof ("desc") - 1,	3031	hv_store (hv, "desc", sizeof ("desc") - 1,
2578	newSVpvn ("[async_pool idle]", sizeof ("[async_pool idle]") - 1), 0);	3032	newSVpvn ("[async_pool idle]", sizeof ("[async_pool idle]") - 1), 0);
2579		3033
2580	coro->prio = 0;	3034	coro->prio = 0;
2581		3035
2582	if (coro->cctx && (coro->cctx->flags & CC_TRACE))	3036	if (coro->cctx && (coro->cctx->flags & CC_TRACE))
2583	api_trace (aTHX_ coro_current, 0);	3037	api_trace (aTHX_ coro_current, 0);
2584		3038
2585	av_push (av_async_pool, newSVsv (coro_current));	3039	av_push (av_async_pool, newSVsv (coro_current));
2586	}	3040	}
2587		3041
2588		3042	SV *
2589	MODULE = Coro::State PACKAGE = Coro::AIO	3043	rouse_cb ()
		3044	PROTOTYPE:
		3045	CODE:
		3046	RETVAL = coro_new_rouse_cb (aTHX);
		3047	OUTPUT:
		3048	RETVAL
2590		3049
2591	void	3050	void
2592	_get_state (SV *self)	3051	rouse_wait (...)
2593	PROTOTYPE: $	3052	PROTOTYPE: ;$
2594	PPCODE:	3053	PPCODE:
2595	{	3054	CORO_EXECUTE_SLF_XS (slf_init_rouse_wait);
2596	AV *defav = GvAV (PL_defgv);
2597	AV *av = newAV ();
2598	int i;
2599	SV *data_sv = newSV (sizeof (struct io_state));
2600	struct io_state *data = (struct io_state *)SvPVX (data_sv);
2601	SvCUR_set (data_sv, sizeof (struct io_state));
2602	SvPOK_only (data_sv);
2603
2604	data->errorno = errno;
2605	data->laststype = PL_laststype;
2606	data->laststatval = PL_laststatval;
2607	data->statcache = PL_statcache;
2608
2609	av_extend (av, AvFILLp (defav) + 1 + 1);
2610
2611	for (i = 0; i <= AvFILLp (defav); ++i)
2612	av_push (av, SvREFCNT_inc_NN (AvARRAY (defav)[i]));
2613
2614	av_push (av, data_sv);
2615
2616	XPUSHs (sv_2mortal (newRV_noinc ((SV *)av)));
2617
2618	api_ready (aTHX_ self);
2619	}
2620
2621	void
2622	_set_state (SV *state)
2623	PROTOTYPE: $
2624	PPCODE:
2625	{
2626	AV *av = (AV *)SvRV (state);
2627	struct io_state *data = (struct io_state *)SvPVX (AvARRAY (av)[AvFILLp (av)]);
2628	int i;
2629
2630	errno = data->errorno;
2631	PL_laststype = data->laststype;
2632	PL_laststatval = data->laststatval;
2633	PL_statcache = data->statcache;
2634
2635	EXTEND (SP, AvFILLp (av));
2636	for (i = 0; i < AvFILLp (av); ++i)
2637	PUSHs (sv_2mortal (SvREFCNT_inc_NN (AvARRAY (av)[i])));
2638	}
2639
2640
2641	MODULE = Coro::State PACKAGE = Coro::AnyEvent
2642
2643	BOOT:
2644	sv_activity = coro_get_sv (aTHX_ "Coro::AnyEvent::ACTIVITY", TRUE);
2645
2646	void
2647	_schedule (...)
2648	CODE:
2649	{
2650	static int incede;
2651
2652	api_cede_notself (aTHX);
2653
2654	++incede;
2655	while (coro_nready >= incede && api_cede (aTHX))
2656	;
2657
2658	sv_setsv (sv_activity, &PL_sv_undef);
2659	if (coro_nready >= incede)
2660	{
2661	PUSHMARK (SP);
2662	PUTBACK;
2663	call_pv ("Coro::AnyEvent::_activity", G_DISCARD | G_EVAL);
2664	SPAGAIN;
2665	}
2666
2667	--incede;
2668	}
2669		3055
2670		3056
2671	MODULE = Coro::State PACKAGE = PerlIO::cede	3057	MODULE = Coro::State PACKAGE = PerlIO::cede
2672		3058
2673	BOOT:	3059	BOOT:
2674	PerlIO_define_layer (aTHX_ &PerlIO_cede);	3060	PerlIO_define_layer (aTHX_ &PerlIO_cede);
2675		3061
		3062
2676	MODULE = Coro::State PACKAGE = Coro::Semaphore	3063	MODULE = Coro::State PACKAGE = Coro::Semaphore
2677		3064
2678	SV *	3065	SV *
2679	new (SV *klass, SV *count_ = 0)	3066	new (SV *klass, SV *count = 0)
2680	CODE:	3067	CODE:
2681	{	3068	RETVAL = sv_bless (
2682	/* a semaphore contains a counter IV in $sem->[0] and any waiters after that */	3069	coro_waitarray_new (aTHX_ count && SvOK (count) ? SvIV (count) : 1),
2683	AV *av = newAV ();	3070	GvSTASH (CvGV (cv))
2684	av_push (av, newSViv (count_ && SvOK (count_) ? SvIV (count_) : 1));	3071	);
2685	RETVAL = sv_bless (newRV_noinc ((SV *)av), GvSTASH (CvGV (cv)));	3072	OUTPUT:
2686	}	3073	RETVAL
		3074
		3075	# helper for Coro::Channel
		3076	SV *
		3077	_alloc (int count)
		3078	CODE:
		3079	RETVAL = coro_waitarray_new (aTHX_ count);
2687	OUTPUT:	3080	OUTPUT:
2688	RETVAL	3081	RETVAL
2689		3082
2690	SV *	3083	SV *
2691	count (SV *self)	3084	count (SV *self)
…		…
2700	adjust = 1	3093	adjust = 1
2701	CODE:	3094	CODE:
2702	coro_semaphore_adjust (aTHX_ (AV *)SvRV (self), ix ? adjust : 1);	3095	coro_semaphore_adjust (aTHX_ (AV *)SvRV (self), ix ? adjust : 1);
2703		3096
2704	void	3097	void
2705	down (SV *self)	3098	down (...)
2706	CODE:	3099	CODE:
2707	api_execute_slf (aTHX_ cv, slf_init_semaphore_down, &ST (0), 1);	3100	CORO_EXECUTE_SLF_XS (slf_init_semaphore_down);
		3101
		3102	void
		3103	wait (...)
		3104	CODE:
		3105	CORO_EXECUTE_SLF_XS (slf_init_semaphore_wait);
2708		3106
2709	void	3107	void
2710	try (SV *self)	3108	try (SV *self)
2711	PPCODE:	3109	PPCODE:
2712	{	3110	{
…		…
2724	XSRETURN_NO;	3122	XSRETURN_NO;
2725	}	3123	}
2726		3124
2727	void	3125	void
2728	waiters (SV *self)	3126	waiters (SV *self)
2729	CODE:	3127	PPCODE:
2730	{	3128	{
2731	AV *av = (AV *)SvRV (self);	3129	AV *av = (AV *)SvRV (self);
		3130	int wcount = AvFILLp (av) + 1 - 1;
2732		3131
2733	if (GIMME_V == G_SCALAR)	3132	if (GIMME_V == G_SCALAR)
2734	XPUSHs (sv_2mortal (newSVsv (AvARRAY (av)[0])));	3133	XPUSHs (sv_2mortal (newSViv (wcount)));
2735	else	3134	else
2736	{	3135	{
2737	int i;	3136	int i;
2738	EXTEND (SP, AvFILLp (av) + 1 - 1);	3137	EXTEND (SP, wcount);
2739	for (i = 1; i <= AvFILLp (av); ++i)	3138	for (i = 1; i <= wcount; ++i)
2740	PUSHs (newSVsv (AvARRAY (av)[i]));	3139	PUSHs (sv_2mortal (newRV_inc (AvARRAY (av)[i])));
2741	}	3140	}
2742	}	3141	}
2743		3142
		3143	MODULE = Coro::State PACKAGE = Coro::Signal
		3144
		3145	SV *
		3146	new (SV *klass)
		3147	CODE:
		3148	RETVAL = sv_bless (
		3149	coro_waitarray_new (aTHX_ 0),
		3150	GvSTASH (CvGV (cv))
		3151	);
		3152	OUTPUT:
		3153	RETVAL
		3154
		3155	void
		3156	wait (...)
		3157	CODE:
		3158	CORO_EXECUTE_SLF_XS (slf_init_signal_wait);
		3159
		3160	void
		3161	broadcast (SV *self)
		3162	CODE:
		3163	{
		3164	AV *av = (AV *)SvRV (self);
		3165	coro_signal_wake (aTHX_ av, AvFILLp (av));
		3166	}
		3167
		3168	void
		3169	send (SV *self)
		3170	CODE:
		3171	{
		3172	AV *av = (AV *)SvRV (self);
		3173
		3174	if (AvFILLp (av))
		3175	coro_signal_wake (aTHX_ av, 1);
		3176	else
		3177	SvIVX (AvARRAY (av)[0]) = 1; /* remember the signal */
		3178	}
		3179
		3180	IV
		3181	awaited (SV *self)
		3182	CODE:
		3183	RETVAL = AvFILLp ((AV *)SvRV (self)) + 1 - 1;
		3184	OUTPUT:
		3185	RETVAL
		3186
		3187
		3188	MODULE = Coro::State PACKAGE = Coro::AnyEvent
		3189
		3190	BOOT:
		3191	sv_activity = coro_get_sv (aTHX_ "Coro::AnyEvent::ACTIVITY", TRUE);
		3192
		3193	void
		3194	_schedule (...)
		3195	CODE:
		3196	{
		3197	static int incede;
		3198
		3199	api_cede_notself (aTHX);
		3200
		3201	++incede;
		3202	while (coro_nready >= incede && api_cede (aTHX))
		3203	;
		3204
		3205	sv_setsv (sv_activity, &PL_sv_undef);
		3206	if (coro_nready >= incede)
		3207	{
		3208	PUSHMARK (SP);
		3209	PUTBACK;
		3210	call_pv ("Coro::AnyEvent::_activity", G_KEEPERR | G_EVAL | G_VOID | G_DISCARD);
		3211	}
		3212
		3213	--incede;
		3214	}
		3215
		3216
		3217	MODULE = Coro::State PACKAGE = Coro::AIO
		3218
		3219	void
		3220	_register (char *target, char *proto, SV *req)
		3221	CODE:
		3222	{
		3223	CV *req_cv = coro_sv_2cv (req);
		3224	/* newXSproto doesn't return the CV on 5.8 */
		3225	CV *slf_cv = newXS (target, coro_aio_req_xs, __FILE__);
		3226	sv_setpv ((SV *)slf_cv, proto);
		3227	sv_magicext ((SV *)slf_cv, (SV *)req_cv, CORO_MAGIC_type_aio, 0, 0, 0);
		3228	}
		3229

Diff Legend

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