ViewVC Help

View File | Revision Log | Show Annotations | Download File

/cvs/Coro/Coro/State.xs

Comparing Coro/Coro/State.xs (file contents):
Revision 1.284 by root, Mon Nov 17 01:05:47 2008 UTC vs.
Revision 1.307 by root, Wed Nov 19 14:34:46 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 */
…		…
241	/* this is a structure representing a perl-level coroutine */	241	/* this is a structure representing a perl-level coroutine */
242	struct coro {	242	struct coro {
243	/* the C coroutine allocated to this perl coroutine, if any */	243	/* the C coroutine allocated to this perl coroutine, if any */
244	coro_cctx *cctx;	244	coro_cctx *cctx;
245		245
246	/* process data */	246	/* state data */
247	struct CoroSLF slf_frame; /* saved slf frame */	247	struct CoroSLF slf_frame; /* saved slf frame */
248	AV *mainstack;	248	AV *mainstack;
249	perl_slots *slot; /* basically the saved sp */	249	perl_slots *slot; /* basically the saved sp */
250		250
		251	CV *startcv; /* the CV to execute */
251	AV *args; /* data associated with this coroutine (initial args) */	252	AV *args; /* data associated with this coroutine (initial args) */
252	int refcnt; /* coroutines are refcounted, yes */	253	int refcnt; /* coroutines are refcounted, yes */
253	int flags; /* CF_ flags */	254	int flags; /* CF_ flags */
254	HV *hv; /* the perl hash associated with this coro, if any */	255	HV *hv; /* the perl hash associated with this coro, if any */
255	void (*on_destroy)(pTHX_ struct coro *coro);	256	void (*on_destroy)(pTHX_ struct coro *coro);
256		257
257	/* statistics */	258	/* statistics */
258	int usecount; /* number of transfers to this coro */	259	int usecount; /* number of transfers to this coro */
259		260
260	/* coro process data */	261	/* coro process data */
261	int prio;	262	int prio;
262	SV *throw; /* exception to be thrown */	263	SV *except; /* exception to be thrown */
		264	SV *rouse_cb;
263		265
264	/* async_pool */	266	/* async_pool */
265	SV *saved_deffh;	267	SV *saved_deffh;
266		268
267	/* linked list */	269	/* linked list */
…		…
269	};	271	};
270		272
271	typedef struct coro *Coro__State;	273	typedef struct coro *Coro__State;
272	typedef struct coro *Coro__State_or_hashref;	274	typedef struct coro *Coro__State_or_hashref;
273		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 */
274	static struct CoroSLF slf_frame; /* the current slf frame */	279	static struct CoroSLF slf_frame; /* the current slf frame */
275		280
276	/** Coro ********************************************************************/	281	/** Coro ********************************************************************/
277		282
278	#define PRIO_MAX 3	283	#define PRIO_MAX 3
…		…
317	#if PERL_VERSION_ATLEAST (5,10,0)	322	#if PERL_VERSION_ATLEAST (5,10,0)
318	/* 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 */
319	get_hv (name, create);	324	get_hv (name, create);
320	#endif	325	#endif
321	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);
322	}	336	}
323		337
324	static AV *	338	static AV *
325	coro_clone_padlist (pTHX_ CV *cv)	339	coro_clone_padlist (pTHX_ CV *cv)
326	{	340	{
…		…
380	SvREFCNT_dec (av); /* sv_magicext increased the refcount */	394	SvREFCNT_dec (av); /* sv_magicext increased the refcount */
381		395
382	return 0;	396	return 0;
383	}	397	}
384		398
385	#define CORO_MAGIC_type_cv PERL_MAGIC_ext	399	#define CORO_MAGIC_type_cv 26
386	#define CORO_MAGIC_type_state PERL_MAGIC_ext	400	#define CORO_MAGIC_type_state PERL_MAGIC_ext
387		401
388	static MGVTBL coro_cv_vtbl = {	402	static MGVTBL coro_cv_vtbl = {
389	0, 0, 0, 0,	403	0, 0, 0, 0,
390	coro_cv_free	404	coro_cv_free
391	};	405	};
392		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
393	#define CORO_MAGIC(sv, type) \	412	#define CORO_MAGIC(sv, type) \
394	expect_true (SvMAGIC (sv)) \	413	(expect_true (SvMAGIC (sv)) \
395	? expect_true (SvMAGIC (sv)->mg_type == type) \	414	? CORO_MAGIC_NN (sv, type) \
396	? SvMAGIC (sv) \
397	: mg_find (sv, type) \
398	: 0	415	: 0)
399		416
400	#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)
401	#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)
402		419
403	INLINE struct coro *	420	INLINE struct coro *
404	SvSTATE_ (pTHX_ SV *coro)	421	SvSTATE_ (pTHX_ SV *coro)
405	{	422	{
406	HV *stash;	423	HV *stash;
…		…
424	return (struct coro *)mg->mg_ptr;	441	return (struct coro *)mg->mg_ptr;
425	}	442	}
426		443
427	#define SvSTATE(sv) SvSTATE_ (aTHX_ (sv))	444	#define SvSTATE(sv) SvSTATE_ (aTHX_ (sv))
428		445
429	/* fastert than SvSTATE, but expects a coroutine hv */	446	/* faster than SvSTATE, but expects a coroutine hv */
430	INLINE struct coro *	447	#define SvSTATE_hv(hv) ((struct coro *)CORO_MAGIC_NN ((SV *)hv, CORO_MAGIC_type_state)->mg_ptr)
431	SvSTATE_hv (SV *sv)
432	{
433	MAGIC *mg = expect_true (SvMAGIC (sv)->mg_type == CORO_MAGIC_type_state)
434	? SvMAGIC (sv)
435	: mg_find (sv, CORO_MAGIC_type_state);
436
437	return (struct coro *)mg->mg_ptr;
438	}
439
440	#define SvSTATE_current SvSTATE_hv (SvRV (coro_current))	448	#define SvSTATE_current SvSTATE_hv (SvRV (coro_current))
441		449
442	/* the next two functions merely cache the padlists */	450	/* the next two functions merely cache the padlists */
443	static void	451	static void
444	get_padlist (pTHX_ CV *cv)	452	get_padlist (pTHX_ CV *cv)
…		…
451	else	459	else
452	{	460	{
453	#if CORO_PREFER_PERL_FUNCTIONS	461	#if CORO_PREFER_PERL_FUNCTIONS
454	/* this is probably cleaner? but also slower! */	462	/* this is probably cleaner? but also slower! */
455	/* in practise, it seems to be less stable */	463	/* in practise, it seems to be less stable */
456	CV *cp = Perl_cv_clone (cv);	464	CV *cp = Perl_cv_clone (aTHX_ cv);
457	CvPADLIST (cv) = CvPADLIST (cp);	465	CvPADLIST (cv) = CvPADLIST (cp);
458	CvPADLIST (cp) = 0;	466	CvPADLIST (cp) = 0;
459	SvREFCNT_dec (cp);	467	SvREFCNT_dec (cp);
460	#else	468	#else
461	CvPADLIST (cv) = coro_clone_padlist (aTHX_ cv);	469	CvPADLIST (cv) = coro_clone_padlist (aTHX_ cv);
…		…
513	}	521	}
514		522
515	PUTBACK;	523	PUTBACK;
516	}	524	}
517		525
518	slf_frame = c->slf_frame;	526	slf_frame = c->slf_frame;
		527	CORO_THROW = c->except;
519	}	528	}
520		529
521	static void	530	static void
522	save_perl (pTHX_ Coro__State c)	531	save_perl (pTHX_ Coro__State c)
523	{	532	{
		533	c->except = CORO_THROW;
524	c->slf_frame = slf_frame;	534	c->slf_frame = slf_frame;
525		535
526	{	536	{
527	dSP;	537	dSP;
528	I32 cxix = cxstack_ix;	538	I32 cxix = cxstack_ix;
…		…
603	* of perl.c:init_stacks, except that it uses less memory	613	* of perl.c:init_stacks, except that it uses less memory
604	* on the (sometimes correct) assumption that coroutines do	614	* on the (sometimes correct) assumption that coroutines do
605	* not usually need a lot of stackspace.	615	* not usually need a lot of stackspace.
606	*/	616	*/
607	#if CORO_PREFER_PERL_FUNCTIONS	617	#if CORO_PREFER_PERL_FUNCTIONS
608	# define coro_init_stacks init_stacks	618	# define coro_init_stacks(thx) init_stacks ()
609	#else	619	#else
610	static void	620	static void
611	coro_init_stacks (pTHX)	621	coro_init_stacks (pTHX)
612	{	622	{
613	PL_curstackinfo = new_stackinfo(32, 8);	623	PL_curstackinfo = new_stackinfo(32, 8);
…		…
821	slf_check_nop (pTHX_ struct CoroSLF *frame)	831	slf_check_nop (pTHX_ struct CoroSLF *frame)
822	{	832	{
823	return 0;	833	return 0;
824	}	834	}
825		835
826	static void	836	static UNOP coro_setup_op;
		837
		838	static void NOINLINE /* noinline to keep it out of the transfer fast path */
827	coro_setup (pTHX_ struct coro *coro)	839	coro_setup (pTHX_ struct coro *coro)
828	{	840	{
829	/*	841	/*
830	* emulate part of the perl startup here.	842	* emulate part of the perl startup here.
831	*/	843	*/
…		…
858	{	870	{
859	dSP;	871	dSP;
860	UNOP myop;	872	UNOP myop;
861		873
862	Zero (&myop, 1, UNOP);	874	Zero (&myop, 1, UNOP);
863	myop.op_next = Nullop;	875	myop.op_next = Nullop;
864	myop.op_flags = OPf_WANT_VOID;	876	myop.op_flags = OPf_WANT_VOID;
865		877
866	PUSHMARK (SP);	878	PUSHMARK (SP);
867	XPUSHs (sv_2mortal (av_shift (GvAV (PL_defgv))));	879	XPUSHs (sv_2mortal (av_shift (GvAV (PL_defgv))));
868	PUTBACK;	880	PUTBACK;
…		…
874	/* 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
875	* 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.
876	*/	888	*/
877	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 */
878	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 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;
879	}	902	}
880		903
881	static void	904	static void
882	coro_destruct (pTHX_ struct coro *coro)	905	coro_destruct (pTHX_ struct coro *coro)
883	{	906	{
…		…
906	SvREFCNT_dec (GvSV (irsgv));	929	SvREFCNT_dec (GvSV (irsgv));
907		930
908	SvREFCNT_dec (PL_diehook);	931	SvREFCNT_dec (PL_diehook);
909	SvREFCNT_dec (PL_warnhook);	932	SvREFCNT_dec (PL_warnhook);
910		933
		934	SvREFCNT_dec (CORO_THROW);
911	SvREFCNT_dec (coro->saved_deffh);	935	SvREFCNT_dec (coro->saved_deffh);
912	SvREFCNT_dec (coro->throw);	936	SvREFCNT_dec (coro->rouse_cb);
913		937
914	coro_destruct_stacks (aTHX);	938	coro_destruct_stacks (aTHX);
915	}	939	}
916		940
917	INLINE void	941	INLINE void
…		…
1004	SAVETMPS;	1028	SAVETMPS;
1005	EXTEND (SP, 3);	1029	EXTEND (SP, 3);
1006	PUSHMARK (SP);	1030	PUSHMARK (SP);
1007	PUSHs (&PL_sv_yes);	1031	PUSHs (&PL_sv_yes);
1008	PUSHs (fullname);	1032	PUSHs (fullname);
1009	PUSHs (CxHASARGS (cx) ? sv_2mortal (newRV_inc ((SV *)cx->blk_sub.argarray)) : &PL_sv_undef);	1033	PUSHs (CxHASARGS (cx) ? sv_2mortal (newRV_inc ((SV *)cx->blk_sub.argarray)) : &PL_sv_undef);
1010	PUTBACK;	1034	PUTBACK;
1011	cb = hv_fetch ((HV *)SvRV (coro_current), "_trace_sub_cb", sizeof ("_trace_sub_cb") - 1, 0);	1035	cb = hv_fetch ((HV *)SvRV (coro_current), "_trace_sub_cb", sizeof ("_trace_sub_cb") - 1, 0);
1012	if (cb) call_sv (*cb, G_KEEPERR | G_EVAL | G_VOID | G_DISCARD);	1036	if (cb) call_sv (*cb, G_KEEPERR | G_EVAL | G_VOID | G_DISCARD);
1013	SPAGAIN;	1037	SPAGAIN;
1014	FREETMPS;	1038	FREETMPS;
…		…
1046		1070
1047	TAINT_NOT;	1071	TAINT_NOT;
1048	return 0;	1072	return 0;
1049	}	1073	}
1050		1074
		1075	static struct coro_cctx *cctx_ssl_cctx;
		1076	static struct CoroSLF cctx_ssl_frame;
		1077
1051	static void	1078	static void
1052	prepare_set_stacklevel (struct coro_transfer_args *ta, struct coro_cctx *cctx)	1079	slf_prepare_set_stacklevel (pTHX_ struct coro_transfer_args *ta)
1053	{	1080	{
1054	ta->prev = (struct coro *)cctx;	1081	ta->prev = (struct coro *)cctx_ssl_cctx;
1055	ta->next = 0;	1082	ta->next = 0;
1056	}	1083	}
1057		1084
1058	/* inject a fake call to Coro::State::_cctx_init into the execution */	1085	static int
1059	/* _cctx_init should be careful, as it could be called at almost any time */	1086	slf_check_set_stacklevel (pTHX_ struct CoroSLF *frame)
1060	/* during execution of a perl program */	1087	{
1061	/* also initialises PL_top_env */	1088	*frame = cctx_ssl_frame;
		1089
		1090	return frame->check (aTHX_ frame); /* execute the restored frame - there must be one */
		1091	}
		1092
		1093	/* initialises PL_top_env and injects a pseudo-slf-call to set the stacklevel */
1062	static void NOINLINE	1094	static void NOINLINE
1063	cctx_prepare (pTHX_ coro_cctx *cctx)	1095	cctx_prepare (pTHX_ coro_cctx *cctx)
1064	{	1096	{
1065	dSP;
1066	UNOP myop;
1067
1068	PL_top_env = &PL_start_env;	1097	PL_top_env = &PL_start_env;
1069		1098
1070	if (cctx->flags & CC_TRACE)	1099	if (cctx->flags & CC_TRACE)
1071	PL_runops = runops_trace;	1100	PL_runops = runops_trace;
1072		1101
1073	Zero (&myop, 1, UNOP);	1102	/* we already must be executing an SLF op, there is no other valid way
1074	myop.op_next = PL_op;	1103	* that can lead to creation of a new cctx */
1075	myop.op_flags = OPf_WANT_VOID | OPf_STACKED;	1104	assert (("FATAL: can't prepare slf-less cctx in Coro module (please report)",
		1105	slf_frame.prepare && PL_op->op_ppaddr == pp_slf));
1076		1106
1077	PUSHMARK (SP);	1107	/* we must emulate leaving pp_slf, which is done inside slf_check_set_stacklevel */
1078	EXTEND (SP, 2);	1108	cctx_ssl_cctx = cctx;
1079	PUSHs (sv_2mortal (newSViv ((IV)cctx)));	1109	cctx_ssl_frame = slf_frame;
1080	PUSHs ((SV *)get_cv ("Coro::State::_cctx_init", FALSE));	1110
1081	PUTBACK;	1111	slf_frame.prepare = slf_prepare_set_stacklevel;
1082	PL_op = (OP *)&myop;	1112	slf_frame.check = slf_check_set_stacklevel;
1083	PL_op = PL_ppaddr[OP_ENTERSUB](aTHX);
1084	SPAGAIN;
1085	}	1113	}
1086		1114
1087	/* the tail of transfer: execute stuff we can only do after a transfer */	1115	/* the tail of transfer: execute stuff we can only do after a transfer */
1088	INLINE void	1116	INLINE void
1089	transfer_tail (pTHX)	1117	transfer_tail (pTHX)
…		…
1276	/** coroutine switching *****************************************************/	1304	/** coroutine switching *****************************************************/
1277		1305
1278	static void	1306	static void
1279	transfer_check (pTHX_ struct coro *prev, struct coro *next)	1307	transfer_check (pTHX_ struct coro *prev, struct coro *next)
1280	{	1308	{
		1309	/* TODO: throwing up here is considered harmful */
		1310
1281	if (expect_true (prev != next))	1311	if (expect_true (prev != next))
1282	{	1312	{
1283	if (expect_false (!(prev->flags & (CF_RUNNING | CF_NEW))))	1313	if (expect_false (!(prev->flags & (CF_RUNNING | CF_NEW))))
1284	croak ("Coro::State::transfer called with non-running/new prev Coro::State, but can only transfer from running or new states,");	1314	croak ("Coro::State::transfer called with non-running/new prev Coro::State, but can only transfer from running or new states,");
1285		1315
…		…
1295	#endif	1325	#endif
1296	}	1326	}
1297	}	1327	}
1298		1328
1299	/* always use the TRANSFER macro */	1329	/* always use the TRANSFER macro */
1300	static void NOINLINE	1330	static void NOINLINE /* noinline so we have a fixed stackframe */
1301	transfer (pTHX_ struct coro *prev, struct coro *next, int force_cctx)	1331	transfer (pTHX_ struct coro *prev, struct coro *next, int force_cctx)
1302	{	1332	{
1303	dSTACKLEVEL;	1333	dSTACKLEVEL;
1304		1334
1305	/* sometimes transfer is only called to set idle_sp */	1335	/* sometimes transfer is only called to set idle_sp */
1306	if (expect_false (!next))	1336	if (expect_false (!next))
1307	{	1337	{
1308	((coro_cctx *)prev)->idle_sp = (void *)stacklevel;	1338	((coro_cctx *)prev)->idle_sp = STACKLEVEL;
1309	assert (((coro_cctx *)prev)->idle_te = PL_top_env); /* just for the side-effect when asserts are enabled */	1339	assert (((coro_cctx *)prev)->idle_te = PL_top_env); /* just for the side-effect when asserts are enabled */
1310	}	1340	}
1311	else if (expect_true (prev != next))	1341	else if (expect_true (prev != next))
1312	{	1342	{
1313	coro_cctx *prev__cctx;	1343	coro_cctx *prev__cctx;
…		…
1338		1368
1339	prev__cctx = prev->cctx;	1369	prev__cctx = prev->cctx;
1340		1370
1341	/* possibly untie and reuse the cctx */	1371	/* possibly untie and reuse the cctx */
1342	if (expect_true (	1372	if (expect_true (
1343	prev__cctx->idle_sp == (void *)stacklevel	1373	prev__cctx->idle_sp == STACKLEVEL
1344	&& !(prev__cctx->flags & CC_TRACE)	1374	&& !(prev__cctx->flags & CC_TRACE)
1345	&& !force_cctx	1375	&& !force_cctx
1346	))	1376	))
1347	{	1377	{
1348	/* I assume that stacklevel is a stronger indicator than PL_top_env changes */	1378	/* I assume that stacklevel is a stronger indicator than PL_top_env changes */
…		…
1415		1445
1416	coro->slot = 0;	1446	coro->slot = 0;
1417	}	1447	}
1418		1448
1419	cctx_destroy (coro->cctx);	1449	cctx_destroy (coro->cctx);
		1450	SvREFCNT_dec (coro->startcv);
1420	SvREFCNT_dec (coro->args);	1451	SvREFCNT_dec (coro->args);
1421		1452
1422	if (coro->next) coro->next->prev = coro->prev;	1453	if (coro->next) coro->next->prev = coro->prev;
1423	if (coro->prev) coro->prev->next = coro->next;	1454	if (coro->prev) coro->prev->next = coro->next;
1424	if (coro == coro_first) coro_first = coro->next;	1455	if (coro == coro_first) coro_first = coro->next;
…		…
1479		1510
1480	prepare_transfer (aTHX_ &ta, prev_sv, next_sv);	1511	prepare_transfer (aTHX_ &ta, prev_sv, next_sv);
1481	TRANSFER (ta, 1);	1512	TRANSFER (ta, 1);
1482	}	1513	}
1483		1514
		1515	/*****************************************************************************/
		1516	/* gensub: simple closure generation utility */
		1517
		1518	#define GENSUB_ARG CvXSUBANY (cv).any_ptr
		1519
		1520	/* create a closure from XS, returns a code reference */
		1521	/* the arg can be accessed via GENSUB_ARG from the callback */
		1522	/* the callback must use dXSARGS/XSRETURN */
		1523	static SV *
		1524	gensub (pTHX_ void (*xsub)(pTHX_ CV *), void *arg)
		1525	{
		1526	CV *cv = (CV *)newSV (0);
		1527
		1528	sv_upgrade ((SV *)cv, SVt_PVCV);
		1529
		1530	CvANON_on (cv);
		1531	CvISXSUB_on (cv);
		1532	CvXSUB (cv) = xsub;
		1533	GENSUB_ARG = arg;
		1534
		1535	return newRV_noinc ((SV *)cv);
		1536	}
		1537
1484	/** Coro ********************************************************************/	1538	/** Coro ********************************************************************/
1485		1539
1486	INLINE void	1540	INLINE void
1487	coro_enq (pTHX_ struct coro *coro)	1541	coro_enq (pTHX_ struct coro *coro)
1488	{	1542	{
…		…
1531	ENTER;	1585	ENTER;
1532	SAVETMPS;	1586	SAVETMPS;
1533		1587
1534	PUSHMARK (SP);	1588	PUSHMARK (SP);
1535	PUTBACK;	1589	PUTBACK;
1536	call_sv (sv_hook, G_DISCARD);	1590	call_sv (sv_hook, G_VOID | G_DISCARD);
1537	SPAGAIN;
1538		1591
1539	FREETMPS;	1592	FREETMPS;
1540	LEAVE;	1593	LEAVE;
1541	}	1594	}
1542		1595
…		…
1569	ENTER;	1622	ENTER;
1570	SAVETMPS;	1623	SAVETMPS;
1571		1624
1572	PUSHMARK (SP);	1625	PUSHMARK (SP);
1573	PUTBACK;	1626	PUTBACK;
1574	call_sv (get_sv ("Coro::idle", FALSE), G_DISCARD);	1627	call_sv (get_sv ("Coro::idle", FALSE), G_VOID | G_DISCARD);
1575	SPAGAIN;
1576		1628
1577	FREETMPS;	1629	FREETMPS;
1578	LEAVE;	1630	LEAVE;
1579	continue;	1631	continue;
1580	}	1632	}
…		…
1687	if (coro->flags & CF_RUNNING)	1739	if (coro->flags & CF_RUNNING)
1688	PL_runops = RUNOPS_DEFAULT;	1740	PL_runops = RUNOPS_DEFAULT;
1689	else	1741	else
1690	coro->slot->runops = RUNOPS_DEFAULT;	1742	coro->slot->runops = RUNOPS_DEFAULT;
1691	}	1743	}
		1744	}
		1745
		1746	/*****************************************************************************/
		1747	/* rouse callback */
		1748
		1749	#define CORO_MAGIC_type_rouse PERL_MAGIC_ext
		1750
		1751	static void
		1752	coro_rouse_callback (pTHX_ CV *cv)
		1753	{
		1754	dXSARGS;
		1755	SV *data = (SV *)GENSUB_ARG;
		1756
		1757	if (SvTYPE (SvRV (data)) != SVt_PVAV)
		1758	{
		1759	/* first call, set args */
		1760	int i;
		1761	AV *av = newAV ();
		1762	SV *coro = SvRV (data);
		1763
		1764	SvRV_set (data, (SV *)av);
		1765	api_ready (aTHX_ coro);
		1766	SvREFCNT_dec (coro);
		1767
		1768	/* better take a full copy of the arguments */
		1769	while (items--)
		1770	av_store (av, items, newSVsv (ST (items)));
		1771	}
		1772
		1773	XSRETURN_EMPTY;
		1774	}
		1775
		1776	static int
		1777	slf_check_rouse_wait (pTHX_ struct CoroSLF *frame)
		1778	{
		1779	SV *data = (SV *)frame->data;
		1780
		1781	if (CORO_THROW)
		1782	return 0;
		1783
		1784	if (SvTYPE (SvRV (data)) != SVt_PVAV)
		1785	return 1;
		1786
		1787	/* now push all results on the stack */
		1788	{
		1789	dSP;
		1790	AV *av = (AV *)SvRV (data);
		1791	int i;
		1792
		1793	EXTEND (SP, AvFILLp (av) + 1);
		1794	for (i = 0; i <= AvFILLp (av); ++i)
		1795	PUSHs (sv_2mortal (AvARRAY (av)[i]));
		1796
		1797	/* we have stolen the elements, so ste length to zero and free */
		1798	AvFILLp (av) = -1;
		1799	av_undef (av);
		1800
		1801	PUTBACK;
		1802	}
		1803
		1804	return 0;
		1805	}
		1806
		1807	static void
		1808	slf_init_rouse_wait (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		1809	{
		1810	SV *cb;
		1811
		1812	if (items)
		1813	cb = arg [0];
		1814	else
		1815	{
		1816	struct coro *coro = SvSTATE_current;
		1817
		1818	if (!coro->rouse_cb)
		1819	croak ("Coro::rouse_wait called without rouse callback, and no default rouse callback found either,");
		1820
		1821	cb = sv_2mortal (coro->rouse_cb);
		1822	coro->rouse_cb = 0;
		1823	}
		1824
		1825	if (!SvROK (cb)
		1826	|| SvTYPE (SvRV (cb)) != SVt_PVCV
		1827	|| CvXSUB ((CV *)SvRV (cb)) != coro_rouse_callback)
		1828	croak ("Coro::rouse_wait called with illegal callback argument,");
		1829
		1830	{
		1831	CV *cv = (CV *)SvRV (cb); /* for GENSUB_ARG */
		1832	SV *data = (SV *)GENSUB_ARG;
		1833
		1834	frame->data = (void *)data;
		1835	frame->prepare = SvTYPE (SvRV (data)) == SVt_PVAV ? prepare_nop : prepare_schedule;
		1836	frame->check = slf_check_rouse_wait;
		1837	}
		1838	}
		1839
		1840	static SV *
		1841	coro_new_rouse_cb (pTHX)
		1842	{
		1843	HV *hv = (HV *)SvRV (coro_current);
		1844	struct coro *coro = SvSTATE_hv (hv);
		1845	SV *data = newRV_inc ((SV *)hv);
		1846	SV *cb = gensub (aTHX_ coro_rouse_callback, (void *)data);
		1847
		1848	sv_magicext (SvRV (cb), data, CORO_MAGIC_type_rouse, 0, 0, 0);
		1849	SvREFCNT_dec (data); /* magicext increases the refcount */
		1850
		1851	SvREFCNT_dec (coro->rouse_cb);
		1852	coro->rouse_cb = SvREFCNT_inc_NN (cb);
		1853
		1854	return cb;
		1855	}
		1856
		1857	/*****************************************************************************/
		1858	/* schedule-like-function opcode (SLF) */
		1859
		1860	static UNOP slf_restore; /* restore stack as entersub did, for first-re-run */
		1861	static const CV *slf_cv;
		1862	static SV **slf_argv;
		1863	static int slf_argc, slf_arga; /* count, allocated */
		1864	static I32 slf_ax; /* top of stack, for restore */
		1865
		1866	/* this restores the stack in the case we patched the entersub, to */
		1867	/* recreate the stack frame as perl will on following calls */
		1868	/* since entersub cleared the stack */
		1869	static OP *
		1870	pp_restore (pTHX)
		1871	{
		1872	int i;
		1873	SV **SP = PL_stack_base + slf_ax;
		1874
		1875	PUSHMARK (SP);
		1876
		1877	EXTEND (SP, slf_argc + 1);
		1878
		1879	for (i = 0; i < slf_argc; ++i)
		1880	PUSHs (sv_2mortal (slf_argv [i]));
		1881
		1882	PUSHs ((SV *)CvGV (slf_cv));
		1883
		1884	RETURNOP (slf_restore.op_first);
		1885	}
		1886
		1887	static void
		1888	slf_prepare_transfer (pTHX_ struct coro_transfer_args *ta)
		1889	{
		1890	SV **arg = (SV **)slf_frame.data;
		1891
		1892	prepare_transfer (aTHX_ ta, arg [0], arg [1]);
		1893	}
		1894
		1895	static void
		1896	slf_init_transfer (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		1897	{
		1898	if (items != 2)
		1899	croak ("Coro::State::transfer (prev, next) expects two arguments, not %d,", items);
		1900
		1901	frame->prepare = slf_prepare_transfer;
		1902	frame->check = slf_check_nop;
		1903	frame->data = (void *)arg; /* let's hope it will stay valid */
		1904	}
		1905
		1906	static void
		1907	slf_init_schedule (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		1908	{
		1909	frame->prepare = prepare_schedule;
		1910	frame->check = slf_check_nop;
		1911	}
		1912
		1913	static void
		1914	slf_init_cede (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		1915	{
		1916	frame->prepare = prepare_cede;
		1917	frame->check = slf_check_nop;
		1918	}
		1919
		1920	static void
		1921	slf_init_cede_notself (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		1922	{
		1923	frame->prepare = prepare_cede_notself;
		1924	frame->check = slf_check_nop;
		1925	}
		1926
		1927	/*
		1928	* these not obviously related functions are all rolled into one
		1929	* function to increase chances that they all will call transfer with the same
		1930	* stack offset
		1931	* SLF stands for "schedule-like-function".
		1932	*/
		1933	static OP *
		1934	pp_slf (pTHX)
		1935	{
		1936	I32 checkmark; /* mark SP to see how many elements check has pushed */
		1937
		1938	/* set up the slf frame, unless it has already been set-up */
		1939	/* the latter happens when a new coro has been started */
		1940	/* or when a new cctx was attached to an existing coroutine */
		1941	if (expect_true (!slf_frame.prepare))
		1942	{
		1943	/* first iteration */
		1944	dSP;
		1945	SV **arg = PL_stack_base + TOPMARK + 1;
		1946	int items = SP - arg; /* args without function object */
		1947	SV *gv = *sp;
		1948
		1949	/* do a quick consistency check on the "function" object, and if it isn't */
		1950	/* for us, divert to the real entersub */
		1951	if (SvTYPE (gv) != SVt_PVGV
		1952	|| !GvCV (gv)
		1953	|| !(CvFLAGS (GvCV (gv)) & CVf_SLF))
		1954	return PL_ppaddr[OP_ENTERSUB](aTHX);
		1955
		1956	if (!(PL_op->op_flags & OPf_STACKED))
		1957	{
		1958	/* ampersand-form of call, use @_ instead of stack */
		1959	AV *av = GvAV (PL_defgv);
		1960	arg = AvARRAY (av);
		1961	items = AvFILLp (av) + 1;
		1962	}
		1963
		1964	/* now call the init function, which needs to set up slf_frame */
		1965	((coro_slf_cb)CvXSUBANY (GvCV (gv)).any_ptr)
		1966	(aTHX_ &slf_frame, GvCV (gv), arg, items);
		1967
		1968	/* pop args */
		1969	SP = PL_stack_base + POPMARK;
		1970
		1971	PUTBACK;
		1972	}
		1973
		1974	/* now that we have a slf_frame, interpret it! */
		1975	/* we use a callback system not to make the code needlessly */
		1976	/* complicated, but so we can run multiple perl coros from one cctx */
		1977
		1978	do
		1979	{
		1980	struct coro_transfer_args ta;
		1981
		1982	slf_frame.prepare (aTHX_ &ta);
		1983	TRANSFER (ta, 0);
		1984
		1985	checkmark = PL_stack_sp - PL_stack_base;
		1986	}
		1987	while (slf_frame.check (aTHX_ &slf_frame));
		1988
		1989	slf_frame.prepare = 0; /* invalidate the frame, we are done processing it */
		1990
		1991	/* exception handling */
		1992	if (expect_false (CORO_THROW))
		1993	{
		1994	SV *exception = sv_2mortal (CORO_THROW);
		1995
		1996	CORO_THROW = 0;
		1997	sv_setsv (ERRSV, exception);
		1998	croak (0);
		1999	}
		2000
		2001	/* return value handling - mostly like entersub */
		2002	/* make sure we put something on the stack in scalar context */
		2003	if (GIMME_V == G_SCALAR)
		2004	{
		2005	dSP;
		2006	SV **bot = PL_stack_base + checkmark;
		2007
		2008	if (sp == bot) /* too few, push undef */
		2009	bot [1] = &PL_sv_undef;
		2010	else if (sp != bot + 1) /* too many, take last one */
		2011	bot [1] = *sp;
		2012
		2013	SP = bot + 1;
		2014
		2015	PUTBACK;
		2016	}
		2017
		2018	return NORMAL;
		2019	}
		2020
		2021	static void
		2022	api_execute_slf (pTHX_ CV *cv, coro_slf_cb init_cb, I32 ax)
		2023	{
		2024	int i;
		2025	SV **arg = PL_stack_base + ax;
		2026	int items = PL_stack_sp - arg + 1;
		2027
		2028	assert (("FATAL: SLF call with illegal CV value", !CvANON (cv)));
		2029
		2030	if (PL_op->op_ppaddr != PL_ppaddr [OP_ENTERSUB]
		2031	&& PL_op->op_ppaddr != pp_slf)
		2032	croak ("FATAL: Coro SLF calls can only be made normally, not via goto or any other means, caught");
		2033
		2034	CvFLAGS (cv) |= CVf_SLF;
		2035	CvXSUBANY (cv).any_ptr = (void *)init_cb;
		2036	slf_cv = cv;
		2037
		2038	/* we patch the op, and then re-run the whole call */
		2039	/* we have to put the same argument on the stack for this to work */
		2040	/* and this will be done by pp_restore */
		2041	slf_restore.op_next = (OP *)&slf_restore;
		2042	slf_restore.op_type = OP_CUSTOM;
		2043	slf_restore.op_ppaddr = pp_restore;
		2044	slf_restore.op_first = PL_op;
		2045
		2046	slf_ax = ax - 1; /* undo the ax++ inside dAXMARK */
		2047
		2048	if (PL_op->op_flags & OPf_STACKED)
		2049	{
		2050	if (items > slf_arga)
		2051	{
		2052	slf_arga = items;
		2053	free (slf_argv);
		2054	slf_argv = malloc (slf_arga * sizeof (SV *));
		2055	}
		2056
		2057	slf_argc = items;
		2058
		2059	for (i = 0; i < items; ++i)
		2060	slf_argv [i] = SvREFCNT_inc (arg [i]);
		2061	}
		2062	else
		2063	slf_argc = 0;
		2064
		2065	PL_op->op_ppaddr = pp_slf;
		2066	PL_op->op_type = OP_CUSTOM; /* maybe we should leave it at entersub? */
		2067
		2068	PL_op = (OP *)&slf_restore;
1692	}	2069	}
1693		2070
1694	/*****************************************************************************/	2071	/*****************************************************************************/
1695	/* PerlIO::cede */	2072	/* PerlIO::cede */
1696		2073
…		…
1765	PerlIOBuf_get_cnt,	2142	PerlIOBuf_get_cnt,
1766	PerlIOBuf_set_ptrcnt,	2143	PerlIOBuf_set_ptrcnt,
1767	};	2144	};
1768		2145
1769	/*****************************************************************************/	2146	/*****************************************************************************/
		2147	/* Coro::Semaphore & Coro::Signal */
1770		2148
1771	static UNOP slf_restore; /* restore stack as entersub did, for first-re-run */
1772	static const CV *slf_cv;
1773	static SV *slf_arg0;
1774	static SV *slf_arg1;
1775	static SV *slf_arg2;
1776	static I32 slf_ax; /* top of stack, for restore */
1777
1778	/* this restores the stack in the case we patched the entersub, to */
1779	/* recreate the stack frame as perl will on following calls */
1780	/* since entersub cleared the stack */
1781	static OP *	2149	static SV *
1782	pp_restore (pTHX)	2150	coro_waitarray_new (pTHX_ int count)
1783	{	2151	{
1784	SV **SP = PL_stack_base + slf_ax;	2152	/* a semaphore contains a counter IV in $sem->[0] and any waiters after that */
		2153	AV *av = newAV ();
		2154	SV **ary;
1785		2155
1786	PUSHMARK (SP);	2156	/* unfortunately, building manually saves memory */
		2157	Newx (ary, 2, SV *);
		2158	AvALLOC (av) = ary;
		2159	/*AvARRAY (av) = ary;*/
		2160	SvPVX ((SV *)av) = (char *)ary; /* 5.8.8 needs this syntax instead of AvARRAY = ary */
		2161	AvMAX (av) = 1;
		2162	AvFILLp (av) = 0;
		2163	ary [0] = newSViv (count);
1787		2164
1788	EXTEND (SP, 3);	2165	return newRV_noinc ((SV *)av);
1789	if (slf_arg0) PUSHs (sv_2mortal (slf_arg0));
1790	if (slf_arg1) PUSHs (sv_2mortal (slf_arg1));
1791	if (slf_arg2) PUSHs (sv_2mortal (slf_arg2));
1792	PUSHs ((SV *)CvGV (slf_cv));
1793
1794	RETURNOP (slf_restore.op_first);
1795	}	2166	}
1796		2167
1797	static void	2168	/* semaphore */
1798	slf_prepare_set_stacklevel (pTHX_ struct coro_transfer_args *ta)
1799	{
1800	prepare_set_stacklevel (ta, (struct coro_cctx *)slf_frame.data);
1801	}
1802
1803	static void
1804	slf_init_set_stacklevel (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
1805	{
1806	assert (("FATAL: set_stacklevel needs the coro cctx as sole argument", items == 1));
1807
1808	frame->prepare = slf_prepare_set_stacklevel;
1809	frame->check = slf_check_nop;
1810	frame->data = (void *)SvIV (arg [0]);
1811	}
1812
1813	static void
1814	slf_prepare_transfer (pTHX_ struct coro_transfer_args *ta)
1815	{
1816	SV **arg = (SV **)slf_frame.data;
1817
1818	prepare_transfer (aTHX_ ta, arg [0], arg [1]);
1819
1820	/* if the destination has ->throw set, then copy it */
1821	/* into the current coro's throw slot, so it will be raised */
1822	/* after the schedule */
1823	if (expect_false (ta->next->throw))
1824	{
1825	struct coro *coro = SvSTATE_current;
1826	SvREFCNT_dec (coro->throw);
1827	coro->throw = ta->next->throw;
1828	ta->next->throw = 0;
1829	}
1830	}
1831
1832	static void
1833	slf_init_transfer (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
1834	{
1835	if (items != 2)
1836	croak ("Coro::State::transfer (prev, next) expects two arguments, not %d,", items);
1837
1838	frame->prepare = slf_prepare_transfer;
1839	frame->check = slf_check_nop;
1840	frame->data = (void *)arg; /* let's hope it will stay valid */
1841	}
1842
1843	static void
1844	slf_init_schedule (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
1845	{
1846	frame->prepare = prepare_schedule;
1847	frame->check = slf_check_nop;
1848	}
1849
1850	static void
1851	slf_init_cede (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
1852	{
1853	frame->prepare = prepare_cede;
1854	frame->check = slf_check_nop;
1855	}
1856
1857	static void
1858	slf_init_cede_notself (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
1859	{
1860	frame->prepare = prepare_cede_notself;
1861	frame->check = slf_check_nop;
1862	}
1863
1864	/* we hijack an hopefully unused CV flag for our purposes */
1865	#define CVf_SLF 0x4000
1866
1867	/*
1868	* these not obviously related functions are all rolled into one
1869	* function to increase chances that they all will call transfer with the same
1870	* stack offset
1871	* SLF stands for "schedule-like-function".
1872	*/
1873	static OP *
1874	pp_slf (pTHX)
1875	{
1876	I32 checkmark; /* mark SP to see how many elements check has pushed */
1877
1878	/* set up the slf frame, unless it has already been set-up */
1879	/* the latter happens when a new coro has been started */
1880	/* or when a new cctx was attached to an existing coroutine */
1881	if (expect_true (!slf_frame.prepare))
1882	{
1883	/* first iteration */
1884	dSP;
1885	SV **arg = PL_stack_base + TOPMARK + 1;
1886	int items = SP - arg; /* args without function object */
1887	SV *gv = *sp;
1888
1889	/* do a quick consistency check on the "function" object, and if it isn't */
1890	/* for us, divert to the real entersub */
1891	if (SvTYPE (gv) != SVt_PVGV || !(CvFLAGS (GvCV (gv)) & CVf_SLF))
1892	return PL_ppaddr[OP_ENTERSUB](aTHX);
1893
1894	if (!(PL_op->op_flags & OPf_STACKED))
1895	{
1896	/* ampersand-form of call, use @_ instead of stack */
1897	AV *av = GvAV (PL_defgv);
1898	arg = AvARRAY (av);
1899	items = AvFILLp (av) + 1;
1900	}
1901
1902	/* now call the init function, which needs to set up slf_frame */
1903	((coro_slf_cb)CvXSUBANY (GvCV (gv)).any_ptr)
1904	(aTHX_ &slf_frame, GvCV (gv), arg, items);
1905
1906	/* pop args */
1907	SP = PL_stack_base + POPMARK;
1908
1909	PUTBACK;
1910	}
1911
1912	/* now that we have a slf_frame, interpret it! */
1913	/* we use a callback system not to make the code needlessly */
1914	/* complicated, but so we can run multiple perl coros from one cctx */
1915
1916	do
1917	{
1918	struct coro_transfer_args ta;
1919
1920	slf_frame.prepare (aTHX_ &ta);
1921	TRANSFER (ta, 0);
1922
1923	checkmark = PL_stack_sp - PL_stack_base;
1924	}
1925	while (slf_frame.check (aTHX_ &slf_frame));
1926
1927	{
1928	dSP;
1929	SV **bot = PL_stack_base + checkmark;
1930	int gimme = GIMME_V;
1931
1932	slf_frame.prepare = 0; /* invalidate the frame, so it gets initialised again next time */
1933
1934	/* make sure we put something on the stack in scalar context */
1935	if (gimme == G_SCALAR)
1936	{
1937	if (sp == bot)
1938	XPUSHs (&PL_sv_undef);
1939
1940	SP = bot + 1;
1941	}
1942
1943	PUTBACK;
1944	}
1945
1946	{
1947	struct coro *coro = SvSTATE_current;
1948
1949	if (expect_false (coro->throw))
1950	{
1951	SV *exception = sv_2mortal (coro->throw);
1952
1953	coro->throw = 0;
1954	sv_setsv (ERRSV, exception);
1955	croak (0);
1956	}
1957	}
1958
1959	return NORMAL;
1960	}
1961
1962	static void
1963	api_execute_slf (pTHX_ CV *cv, coro_slf_cb init_cb, I32 ax)
1964	{
1965	SV **arg = PL_stack_base + ax;
1966	int items = PL_stack_sp - arg + 1;
1967
1968	assert (("FATAL: SLF call with illegal CV value", !CvANON (cv)));
1969
1970	if (PL_op->op_ppaddr != PL_ppaddr [OP_ENTERSUB]
1971	&& PL_op->op_ppaddr != pp_slf)
1972	croak ("FATAL: Coro SLF calls can only be made normally, not via goto or any other means, caught");
1973
1974	#if 0
1975	if (items > 3)
1976	croak ("Coro only supports up to three arguments to SLF functions currently (not %d), caught", items);
1977	#endif
1978
1979	CvFLAGS (cv) |= CVf_SLF;
1980	CvXSUBANY (cv).any_ptr = (void *)init_cb;
1981	slf_cv = cv;
1982
1983	/* we patch the op, and then re-run the whole call */
1984	/* we have to put the same argument on the stack for this to work */
1985	/* and this will be done by pp_restore */
1986	slf_restore.op_next = (OP *)&slf_restore;
1987	slf_restore.op_type = OP_CUSTOM;
1988	slf_restore.op_ppaddr = pp_restore;
1989	slf_restore.op_first = PL_op;
1990
1991	slf_ax = ax - 1; /* undo the ax++ inside dAXMARK */
1992	slf_arg0 = items > 0 ? SvREFCNT_inc (arg [0]) : 0;
1993	slf_arg1 = items > 1 ? SvREFCNT_inc (arg [1]) : 0;
1994	slf_arg2 = items > 2 ? SvREFCNT_inc (arg [2]) : 0;
1995
1996	PL_op->op_ppaddr = pp_slf;
1997	PL_op->op_type = OP_CUSTOM; /* maybe we should leave it at entersub? */
1998
1999	PL_op = (OP *)&slf_restore;
2000	}
2001
2002	/*****************************************************************************/
2003		2169
2004	static void	2170	static void
2005	coro_semaphore_adjust (pTHX_ AV *av, IV adjust)	2171	coro_semaphore_adjust (pTHX_ AV *av, IV adjust)
2006	{	2172	{
2007	SV *count_sv = AvARRAY (av)[0];	2173	SV *count_sv = AvARRAY (av)[0];
…		…
2019	AvARRAY (av)[0] = AvARRAY (av)[1];	2185	AvARRAY (av)[0] = AvARRAY (av)[1];
2020	AvARRAY (av)[1] = count_sv;	2186	AvARRAY (av)[1] = count_sv;
2021	cb = av_shift (av);	2187	cb = av_shift (av);
2022		2188
2023	if (SvOBJECT (cb))	2189	if (SvOBJECT (cb))
		2190	{
2024	api_ready (aTHX_ cb);	2191	api_ready (aTHX_ cb);
2025	else	2192	--count;
2026	croak ("callbacks not yet supported");	2193	}
		2194	else if (SvTYPE (cb) == SVt_PVCV)
		2195	{
		2196	dSP;
		2197	PUSHMARK (SP);
		2198	XPUSHs (sv_2mortal (newRV_inc ((SV *)av)));
		2199	PUTBACK;
		2200	call_sv (cb, G_VOID | G_DISCARD | G_EVAL | G_KEEPERR);
		2201	}
2027		2202
2028	SvREFCNT_dec (cb);	2203	SvREFCNT_dec (cb);
2029
2030	--count;
2031	}	2204	}
2032	}	2205	}
2033		2206
2034	static void	2207	static void
2035	coro_semaphore_on_destroy (pTHX_ struct coro *coro)	2208	coro_semaphore_on_destroy (pTHX_ struct coro *coro)
…		…
2037	/* call $sem->adjust (0) to possibly wake up some other waiters */	2210	/* call $sem->adjust (0) to possibly wake up some other waiters */
2038	coro_semaphore_adjust (aTHX_ (AV *)coro->slf_frame.data, 0);	2211	coro_semaphore_adjust (aTHX_ (AV *)coro->slf_frame.data, 0);
2039	}	2212	}
2040		2213
2041	static int	2214	static int
2042	slf_check_semaphore_down (pTHX_ struct CoroSLF *frame)	2215	slf_check_semaphore_down_or_wait (pTHX_ struct CoroSLF *frame, int acquire)
2043	{	2216	{
2044	AV *av = (AV *)frame->data;	2217	AV *av = (AV *)frame->data;
2045	SV *count_sv = AvARRAY (av)[0];	2218	SV *count_sv = AvARRAY (av)[0];
2046		2219
		2220	/* if we are about to throw, don't actually acquire the lock, just throw */
		2221	if (CORO_THROW)
		2222	return 0;
2047	if (SvIVX (count_sv) > 0)	2223	else if (SvIVX (count_sv) > 0)
2048	{	2224	{
2049	SvSTATE_current->on_destroy = 0;	2225	SvSTATE_current->on_destroy = 0;
		2226
		2227	if (acquire)
2050	SvIVX (count_sv) = SvIVX (count_sv) - 1;	2228	SvIVX (count_sv) = SvIVX (count_sv) - 1;
		2229	else
		2230	coro_semaphore_adjust (aTHX_ av, 0);
		2231
2051	return 0;	2232	return 0;
2052	}	2233	}
2053	else	2234	else
2054	{	2235	{
2055	int i;	2236	int i;
…		…
2064	av_push (av, SvREFCNT_inc (SvRV (coro_current)));	2245	av_push (av, SvREFCNT_inc (SvRV (coro_current)));
2065	return 1;	2246	return 1;
2066	}	2247	}
2067	}	2248	}
2068		2249
2069	static void	2250	static int
		2251	slf_check_semaphore_down (pTHX_ struct CoroSLF *frame)
		2252	{
		2253	return slf_check_semaphore_down_or_wait (aTHX_ frame, 1);
		2254	}
		2255
		2256	static int
		2257	slf_check_semaphore_wait (pTHX_ struct CoroSLF *frame)
		2258	{
		2259	return slf_check_semaphore_down_or_wait (aTHX_ frame, 0);
		2260	}
		2261
		2262	static void
2070	slf_init_semaphore_down (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)	2263	slf_init_semaphore_down_or_wait (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
2071	{	2264	{
2072	AV *av = (AV *)SvRV (arg [0]);	2265	AV *av = (AV *)SvRV (arg [0]);
2073		2266
2074	if (SvIVX (AvARRAY (av)[0]) > 0)	2267	if (SvIVX (AvARRAY (av)[0]) > 0)
2075	{	2268	{
2076	frame->data = (void *)av;	2269	frame->data = (void *)av;
2077	frame->prepare = prepare_nop;	2270	frame->prepare = prepare_nop;
2078	SvSTATE_current->on_destroy = coro_semaphore_on_destroy;
2079	}	2271	}
2080	else	2272	else
2081	{	2273	{
2082	av_push (av, SvREFCNT_inc (SvRV (coro_current)));	2274	av_push (av, SvREFCNT_inc (SvRV (coro_current)));
2083		2275
2084	frame->data = (void *)sv_2mortal (SvREFCNT_inc ((SV *)av));	2276	frame->data = (void *)sv_2mortal (SvREFCNT_inc ((SV *)av));
2085	frame->prepare = prepare_schedule;	2277	frame->prepare = prepare_schedule;
2086		2278
2087	/* to avoid race conditions when a woken-up coro gets terminated */	2279	/* to avoid race conditions when a woken-up coro gets terminated */
2088	/* we arrange for a temporary on_destroy that calls adjust (0) */	2280	/* we arrange for a temporary on_destroy that calls adjust (0) */
2089	assert (!SvSTATE_current->on_destroy);//D
2090	SvSTATE_current->on_destroy = coro_semaphore_on_destroy;	2281	SvSTATE_current->on_destroy = coro_semaphore_on_destroy;
2091	}	2282	}
		2283	}
2092		2284
		2285	static void
		2286	slf_init_semaphore_down (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		2287	{
		2288	slf_init_semaphore_down_or_wait (aTHX_ frame, cv, arg, items);
2093	frame->check = slf_check_semaphore_down;	2289	frame->check = slf_check_semaphore_down;
		2290	}
2094		2291
		2292	static void
		2293	slf_init_semaphore_wait (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		2294	{
		2295	if (items >= 2)
		2296	{
		2297	/* callback form */
		2298	AV *av = (AV *)SvRV (arg [0]);
		2299	CV *cb_cv = coro_sv_2cv (arg [1]);
		2300
		2301	av_push (av, (SV *)SvREFCNT_inc_NN (cb_cv));
		2302
		2303	if (SvIVX (AvARRAY (av)[0]) > 0)
		2304	coro_semaphore_adjust (aTHX_ av, 0);
		2305
		2306	frame->prepare = prepare_nop;
		2307	frame->check = slf_check_nop;
		2308	}
		2309	else
		2310	{
		2311	slf_init_semaphore_down_or_wait (aTHX_ frame, cv, arg, items);
		2312	frame->check = slf_check_semaphore_wait;
		2313	}
		2314	}
		2315
		2316	/* signal */
		2317
		2318	static void
		2319	coro_signal_wake (pTHX_ AV *av, int count)
		2320	{
		2321	SvIVX (AvARRAY (av)[0]) = 0;
		2322
		2323	/* now signal count waiters */
		2324	while (count > 0 && AvFILLp (av) > 0)
		2325	{
		2326	SV *cb;
		2327
		2328	/* swap first two elements so we can shift a waiter */
		2329	cb = AvARRAY (av)[0];
		2330	AvARRAY (av)[0] = AvARRAY (av)[1];
		2331	AvARRAY (av)[1] = cb;
		2332
		2333	cb = av_shift (av);
		2334
		2335	api_ready (aTHX_ cb);
		2336	sv_setiv (cb, 0); /* signal waiter */
		2337	SvREFCNT_dec (cb);
		2338
		2339	--count;
		2340	}
		2341	}
		2342
		2343	static int
		2344	slf_check_signal_wait (pTHX_ struct CoroSLF *frame)
		2345	{
		2346	/* if we are about to throw, also stop waiting */
		2347	return SvROK ((SV *)frame->data) && !CORO_THROW;
		2348	}
		2349
		2350	static void
		2351	slf_init_signal_wait (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		2352	{
		2353	AV *av = (AV *)SvRV (arg [0]);
		2354
		2355	if (SvIVX (AvARRAY (av)[0]))
		2356	{
		2357	SvIVX (AvARRAY (av)[0]) = 0;
		2358	frame->prepare = prepare_nop;
		2359	frame->check = slf_check_nop;
		2360	}
		2361	else
		2362	{
		2363	SV *waiter = newRV_inc (SvRV (coro_current)); /* owned by signal av */
		2364
		2365	av_push (av, waiter);
		2366
		2367	frame->data = (void *)sv_2mortal (SvREFCNT_inc_NN (waiter)); /* owned by process */
		2368	frame->prepare = prepare_schedule;
		2369	frame->check = slf_check_signal_wait;
		2370	}
2095	}	2371	}
2096		2372
2097	/*****************************************************************************/	2373	/*****************************************************************************/
		2374	/* Coro::AIO */
2098		2375
2099	#define GENSUB_ARG CvXSUBANY (cv).any_ptr	2376	#define CORO_MAGIC_type_aio PERL_MAGIC_ext
2100		2377
2101	/* create a closure from XS, returns a code reference */	2378	/* helper storage struct */
2102	/* the arg can be accessed via GENSUB_ARG from the callback */	2379	struct io_state
2103	/* the callback must use dXSARGS/XSRETURN */
2104	static SV *
2105	gensub (pTHX_ void (*xsub)(pTHX_ CV *), void *arg)
2106	{	2380	{
2107	CV *cv = (CV *)NEWSV (0, 0);	2381	int errorno;
		2382	I32 laststype; /* U16 in 5.10.0 */
		2383	int laststatval;
		2384	Stat_t statcache;
		2385	};
2108		2386
		2387	static void
		2388	coro_aio_callback (pTHX_ CV *cv)
		2389	{
		2390	dXSARGS;
		2391	AV *state = (AV *)GENSUB_ARG;
		2392	SV *coro = av_pop (state);
		2393	SV *data_sv = newSV (sizeof (struct io_state));
		2394
		2395	av_extend (state, items);
		2396
2109	sv_upgrade ((SV *)cv, SVt_PVCV);	2397	sv_upgrade (data_sv, SVt_PV);
		2398	SvCUR_set (data_sv, sizeof (struct io_state));
		2399	SvPOK_only (data_sv);
2110		2400
2111	CvANON_on (cv);	2401	{
2112	CvISXSUB_on (cv);	2402	struct io_state *data = (struct io_state *)SvPVX (data_sv);
2113	CvXSUB (cv) = xsub;
2114	GENSUB_ARG = arg;
2115		2403
2116	return newRV_noinc ((SV *)cv);	2404	data->errorno = errno;
		2405	data->laststype = PL_laststype;
		2406	data->laststatval = PL_laststatval;
		2407	data->statcache = PL_statcache;
		2408	}
		2409
		2410	/* now build the result vector out of all the parameters and the data_sv */
		2411	{
		2412	int i;
		2413
		2414	for (i = 0; i < items; ++i)
		2415	av_push (state, SvREFCNT_inc_NN (ST (i)));
		2416	}
		2417
		2418	av_push (state, data_sv);
		2419
		2420	api_ready (aTHX_ coro);
		2421	SvREFCNT_dec (coro);
		2422	SvREFCNT_dec ((AV *)state);
		2423	}
		2424
		2425	static int
		2426	slf_check_aio_req (pTHX_ struct CoroSLF *frame)
		2427	{
		2428	AV *state = (AV *)frame->data;
		2429
		2430	/* if we are about to throw, return early */
		2431	/* this does not cancel the aio request, but at least */
		2432	/* it quickly returns */
		2433	if (CORO_THROW)
		2434	return 0;
		2435
		2436	/* one element that is an RV? repeat! */
		2437	if (AvFILLp (state) == 0 && SvROK (AvARRAY (state)[0]))
		2438	return 1;
		2439
		2440	/* restore status */
		2441	{
		2442	SV *data_sv = av_pop (state);
		2443	struct io_state *data = (struct io_state *)SvPVX (data_sv);
		2444
		2445	errno = data->errorno;
		2446	PL_laststype = data->laststype;
		2447	PL_laststatval = data->laststatval;
		2448	PL_statcache = data->statcache;
		2449
		2450	SvREFCNT_dec (data_sv);
		2451	}
		2452
		2453	/* push result values */
		2454	{
		2455	dSP;
		2456	int i;
		2457
		2458	EXTEND (SP, AvFILLp (state) + 1);
		2459	for (i = 0; i <= AvFILLp (state); ++i)
		2460	PUSHs (sv_2mortal (SvREFCNT_inc_NN (AvARRAY (state)[i])));
		2461
		2462	PUTBACK;
		2463	}
		2464
		2465	return 0;
		2466	}
		2467
		2468	static void
		2469	slf_init_aio_req (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		2470	{
		2471	AV *state = (AV *)sv_2mortal ((SV *)newAV ());
		2472	SV *coro_hv = SvRV (coro_current);
		2473	struct coro *coro = SvSTATE_hv (coro_hv);
		2474
		2475	/* put our coroutine id on the state arg */
		2476	av_push (state, SvREFCNT_inc_NN (coro_hv));
		2477
		2478	/* first see whether we have a non-zero priority and set it as AIO prio */
		2479	if (coro->prio)
		2480	{
		2481	dSP;
		2482
		2483	static SV *prio_cv;
		2484	static SV *prio_sv;
		2485
		2486	if (expect_false (!prio_cv))
		2487	{
		2488	prio_cv = (SV *)get_cv ("IO::AIO::aioreq_pri", 0);
		2489	prio_sv = newSViv (0);
		2490	}
		2491
		2492	PUSHMARK (SP);
		2493	sv_setiv (prio_sv, coro->prio);
		2494	XPUSHs (prio_sv);
		2495
		2496	PUTBACK;
		2497	call_sv (prio_cv, G_VOID | G_DISCARD);
		2498	}
		2499
		2500	/* now call the original request */
		2501	{
		2502	dSP;
		2503	CV *req = (CV *)CORO_MAGIC_NN ((SV *)cv, CORO_MAGIC_type_aio)->mg_obj;
		2504	int i;
		2505
		2506	PUSHMARK (SP);
		2507
		2508	/* first push all args to the stack */
		2509	EXTEND (SP, items + 1);
		2510
		2511	for (i = 0; i < items; ++i)
		2512	PUSHs (arg [i]);
		2513
		2514	/* now push the callback closure */
		2515	PUSHs (sv_2mortal (gensub (aTHX_ coro_aio_callback, (void *)SvREFCNT_inc_NN ((SV *)state))));
		2516
		2517	/* now call the AIO function - we assume our request is uncancelable */
		2518	PUTBACK;
		2519	call_sv ((SV *)req, G_VOID | G_DISCARD);
		2520	}
		2521
		2522	/* now that the requets is going, we loop toll we have a result */
		2523	frame->data = (void *)state;
		2524	frame->prepare = prepare_schedule;
		2525	frame->check = slf_check_aio_req;
		2526	}
		2527
		2528	static void
		2529	coro_aio_req_xs (pTHX_ CV *cv)
		2530	{
		2531	dXSARGS;
		2532
		2533	CORO_EXECUTE_SLF_XS (slf_init_aio_req);
		2534
		2535	XSRETURN_EMPTY;
2117	}	2536	}
2118		2537
2119	/*****************************************************************************/	2538	/*****************************************************************************/
2120		2539
2121	MODULE = Coro::State PACKAGE = Coro::State PREFIX = api_	2540	MODULE = Coro::State PACKAGE = Coro::State PREFIX = api_
…		…
2196	CODE:	2615	CODE:
2197	{	2616	{
2198	struct coro *coro;	2617	struct coro *coro;
2199	MAGIC *mg;	2618	MAGIC *mg;
2200	HV *hv;	2619	HV *hv;
		2620	CV *cb;
2201	int i;	2621	int i;
		2622
		2623	if (items > 1)
		2624	{
		2625	cb = coro_sv_2cv (ST (1));
		2626
		2627	if (CvISXSUB (cb))
		2628	croak ("Coro::State doesn't support XS functions as coroutine start, caught");
		2629
		2630	if (!CvROOT (cb))
		2631	croak ("Coro::State doesn't support autoloaded or undefined functions as coroutine start, caught");
		2632	}
2202		2633
2203	Newz (0, coro, 1, struct coro);	2634	Newz (0, coro, 1, struct coro);
2204	coro->args = newAV ();	2635	coro->args = newAV ();
2205	coro->flags = CF_NEW;	2636	coro->flags = CF_NEW;
2206		2637
…		…
2211	coro->hv = hv = newHV ();	2642	coro->hv = hv = newHV ();
2212	mg = sv_magicext ((SV *)hv, 0, CORO_MAGIC_type_state, &coro_state_vtbl, (char *)coro, 0);	2643	mg = sv_magicext ((SV *)hv, 0, CORO_MAGIC_type_state, &coro_state_vtbl, (char *)coro, 0);
2213	mg->mg_flags |= MGf_DUP;	2644	mg->mg_flags |= MGf_DUP;
2214	RETVAL = sv_bless (newRV_noinc ((SV *)hv), gv_stashpv (klass, 1));	2645	RETVAL = sv_bless (newRV_noinc ((SV *)hv), gv_stashpv (klass, 1));
2215		2646
		2647	if (items > 1)
		2648	{
		2649	coro->startcv = SvREFCNT_inc_NN (cb);
		2650
2216	av_extend (coro->args, items - 1);	2651	av_extend (coro->args, items - 1);
		2652	av_push (coro->args, SvREFCNT_inc (cb));
2217	for (i = 1; i < items; i++)	2653	for (i = 2; i < items; i++)
2218	av_push (coro->args, newSVsv (ST (i)));	2654	av_push (coro->args, newSVsv (ST (i)));
		2655	}
2219	}	2656	}
2220	OUTPUT:	2657	OUTPUT:
2221	RETVAL	2658	RETVAL
2222
2223	void
2224	_set_stacklevel (...)
2225	CODE:
2226	CORO_EXECUTE_SLF_XS (slf_init_set_stacklevel);
2227		2659
2228	void	2660	void
2229	transfer (...)	2661	transfer (...)
2230	PROTOTYPE: $$	2662	PROTOTYPE: $$
2231	CODE:	2663	CODE:
…		…
2355		2787
2356	void	2788	void
2357	throw (Coro::State self, SV *throw = &PL_sv_undef)	2789	throw (Coro::State self, SV *throw = &PL_sv_undef)
2358	PROTOTYPE: $;$	2790	PROTOTYPE: $;$
2359	CODE:	2791	CODE:
		2792	{
		2793	struct coro *current = SvSTATE_current;
		2794	SV **throwp = self == current ? &CORO_THROW : &self->except;
2360	SvREFCNT_dec (self->throw);	2795	SvREFCNT_dec (*throwp);
2361	self->throw = SvOK (throw) ? newSVsv (throw) : 0;	2796	*throwp = SvOK (throw) ? newSVsv (throw) : 0;
		2797	}
2362		2798
2363	void	2799	void
2364	api_trace (SV *coro, int flags = CC_TRACE | CC_TRACE_SUB)	2800	api_trace (SV *coro, int flags = CC_TRACE | CC_TRACE_SUB)
2365	PROTOTYPE: $;$	2801	PROTOTYPE: $;$
2366	C_ARGS: aTHX_ coro, flags	2802	C_ARGS: aTHX_ coro, flags
…		…
2415	SV **dst = ix ? (SV **)&self->slot->defav : (SV **)&self->slot->defsv;	2851	SV **dst = ix ? (SV **)&self->slot->defav : (SV **)&self->slot->defsv;
2416		2852
2417	SV *tmp = *src; *src = *dst; *dst = tmp;	2853	SV *tmp = *src; *src = *dst; *dst = tmp;
2418	}	2854	}
2419		2855
		2856
2420	MODULE = Coro::State PACKAGE = Coro	2857	MODULE = Coro::State PACKAGE = Coro
2421		2858
2422	BOOT:	2859	BOOT:
2423	{	2860	{
2424	int i;	2861	int i;
…		…
2451	coroapi.ready = api_ready;	2888	coroapi.ready = api_ready;
2452	coroapi.is_ready = api_is_ready;	2889	coroapi.is_ready = api_is_ready;
2453	coroapi.nready = coro_nready;	2890	coroapi.nready = coro_nready;
2454	coroapi.current = coro_current;	2891	coroapi.current = coro_current;
2455		2892
2456	GCoroAPI = &coroapi;	2893	/*GCoroAPI = &coroapi;*/
2457	sv_setiv (sv, (IV)&coroapi);	2894	sv_setiv (sv, (IV)&coroapi);
2458	SvREADONLY_on (sv);	2895	SvREADONLY_on (sv);
2459	}	2896	}
2460	}	2897	}
2461		2898
…		…
2568		3005
2569	void	3006	void
2570	_pool_2 (SV *cb)	3007	_pool_2 (SV *cb)
2571	CODE:	3008	CODE:
2572	{	3009	{
2573	struct coro *coro = SvSTATE_current;	3010	HV *hv = (HV *)SvRV (coro_current);
		3011	struct coro *coro = SvSTATE_hv ((SV *)hv);
2574		3012
2575	sv_setsv (cb, &PL_sv_undef);	3013	sv_setsv (cb, &PL_sv_undef);
2576		3014
2577	SvREFCNT_dec ((SV *)PL_defoutgv); PL_defoutgv = (GV *)coro->saved_deffh;	3015	SvREFCNT_dec ((SV *)PL_defoutgv); PL_defoutgv = (GV *)coro->saved_deffh;
2578	coro->saved_deffh = 0;	3016	coro->saved_deffh = 0;
…		…
2585	SvREFCNT_dec (old);	3023	SvREFCNT_dec (old);
2586	croak ("\3async_pool terminate\2\n");	3024	croak ("\3async_pool terminate\2\n");
2587	}	3025	}
2588		3026
2589	av_clear (GvAV (PL_defgv));	3027	av_clear (GvAV (PL_defgv));
2590	hv_store ((HV *)SvRV (coro_current), "desc", sizeof ("desc") - 1,	3028	hv_store (hv, "desc", sizeof ("desc") - 1,
2591	newSVpvn ("[async_pool idle]", sizeof ("[async_pool idle]") - 1), 0);	3029	newSVpvn ("[async_pool idle]", sizeof ("[async_pool idle]") - 1), 0);
2592		3030
2593	coro->prio = 0;	3031	coro->prio = 0;
2594		3032
2595	if (coro->cctx && (coro->cctx->flags & CC_TRACE))	3033	if (coro->cctx && (coro->cctx->flags & CC_TRACE))
2596	api_trace (aTHX_ coro_current, 0);	3034	api_trace (aTHX_ coro_current, 0);
2597		3035
2598	av_push (av_async_pool, newSVsv (coro_current));	3036	av_push (av_async_pool, newSVsv (coro_current));
2599	}	3037	}
2600		3038
2601		3039	SV *
2602	MODULE = Coro::State PACKAGE = Coro::AIO	3040	rouse_cb ()
		3041	PROTOTYPE:
		3042	CODE:
		3043	RETVAL = coro_new_rouse_cb (aTHX);
		3044	OUTPUT:
		3045	RETVAL
2603		3046
2604	void	3047	void
2605	_get_state (SV *self)	3048	rouse_wait (...)
2606	PROTOTYPE: $	3049	PROTOTYPE: ;$
2607	PPCODE:	3050	PPCODE:
2608	{	3051	CORO_EXECUTE_SLF_XS (slf_init_rouse_wait);
2609	AV *defav = GvAV (PL_defgv);
2610	AV *av = newAV ();
2611	int i;
2612	SV *data_sv = newSV (sizeof (struct io_state));
2613	struct io_state *data = (struct io_state *)SvPVX (data_sv);
2614	SvCUR_set (data_sv, sizeof (struct io_state));
2615	SvPOK_only (data_sv);
2616
2617	data->errorno = errno;
2618	data->laststype = PL_laststype;
2619	data->laststatval = PL_laststatval;
2620	data->statcache = PL_statcache;
2621
2622	av_extend (av, AvFILLp (defav) + 1 + 1);
2623
2624	for (i = 0; i <= AvFILLp (defav); ++i)
2625	av_push (av, SvREFCNT_inc_NN (AvARRAY (defav)[i]));
2626
2627	av_push (av, data_sv);
2628
2629	XPUSHs (sv_2mortal (newRV_noinc ((SV *)av)));
2630
2631	api_ready (aTHX_ self);
2632	}
2633
2634	void
2635	_set_state (SV *state)
2636	PROTOTYPE: $
2637	PPCODE:
2638	{
2639	AV *av = (AV *)SvRV (state);
2640	struct io_state *data = (struct io_state *)SvPVX (AvARRAY (av)[AvFILLp (av)]);
2641	int i;
2642
2643	errno = data->errorno;
2644	PL_laststype = data->laststype;
2645	PL_laststatval = data->laststatval;
2646	PL_statcache = data->statcache;
2647
2648	EXTEND (SP, AvFILLp (av));
2649	for (i = 0; i < AvFILLp (av); ++i)
2650	PUSHs (sv_2mortal (SvREFCNT_inc_NN (AvARRAY (av)[i])));
2651	}
2652
2653
2654	MODULE = Coro::State PACKAGE = Coro::AnyEvent
2655
2656	BOOT:
2657	sv_activity = coro_get_sv (aTHX_ "Coro::AnyEvent::ACTIVITY", TRUE);
2658
2659	void
2660	_schedule (...)
2661	CODE:
2662	{
2663	static int incede;
2664
2665	api_cede_notself (aTHX);
2666
2667	++incede;
2668	while (coro_nready >= incede && api_cede (aTHX))
2669	;
2670
2671	sv_setsv (sv_activity, &PL_sv_undef);
2672	if (coro_nready >= incede)
2673	{
2674	PUSHMARK (SP);
2675	PUTBACK;
2676	call_pv ("Coro::AnyEvent::_activity", G_DISCARD | G_EVAL);
2677	SPAGAIN;
2678	}
2679
2680	--incede;
2681	}
2682		3052
2683		3053
2684	MODULE = Coro::State PACKAGE = PerlIO::cede	3054	MODULE = Coro::State PACKAGE = PerlIO::cede
2685		3055
2686	BOOT:	3056	BOOT:
2687	PerlIO_define_layer (aTHX_ &PerlIO_cede);	3057	PerlIO_define_layer (aTHX_ &PerlIO_cede);
2688		3058
		3059
2689	MODULE = Coro::State PACKAGE = Coro::Semaphore	3060	MODULE = Coro::State PACKAGE = Coro::Semaphore
2690		3061
2691	SV *	3062	SV *
2692	new (SV *klass, SV *count_ = 0)	3063	new (SV *klass, SV *count = 0)
2693	CODE:	3064	CODE:
2694	{	3065	RETVAL = sv_bless (
2695	/* a semaphore contains a counter IV in $sem->[0] and any waiters after that */	3066	coro_waitarray_new (aTHX_ count && SvOK (count) ? SvIV (count) : 1),
2696	AV *av = newAV ();	3067	GvSTASH (CvGV (cv))
2697	av_push (av, newSViv (count_ && SvOK (count_) ? SvIV (count_) : 1));	3068	);
2698	RETVAL = sv_bless (newRV_noinc ((SV *)av), GvSTASH (CvGV (cv)));	3069	OUTPUT:
2699	}	3070	RETVAL
		3071
		3072	# helper for Coro::Channel
		3073	SV *
		3074	_alloc (int count)
		3075	CODE:
		3076	RETVAL = coro_waitarray_new (aTHX_ count);
2700	OUTPUT:	3077	OUTPUT:
2701	RETVAL	3078	RETVAL
2702		3079
2703	SV *	3080	SV *
2704	count (SV *self)	3081	count (SV *self)
…		…
2713	adjust = 1	3090	adjust = 1
2714	CODE:	3091	CODE:
2715	coro_semaphore_adjust (aTHX_ (AV *)SvRV (self), ix ? adjust : 1);	3092	coro_semaphore_adjust (aTHX_ (AV *)SvRV (self), ix ? adjust : 1);
2716		3093
2717	void	3094	void
2718	down (SV *self)	3095	down (...)
2719	CODE:	3096	CODE:
2720	CORO_EXECUTE_SLF_XS (slf_init_semaphore_down);	3097	CORO_EXECUTE_SLF_XS (slf_init_semaphore_down);
		3098
		3099	void
		3100	wait (...)
		3101	CODE:
		3102	CORO_EXECUTE_SLF_XS (slf_init_semaphore_wait);
2721		3103
2722	void	3104	void
2723	try (SV *self)	3105	try (SV *self)
2724	PPCODE:	3106	PPCODE:
2725	{	3107	{
…		…
2737	XSRETURN_NO;	3119	XSRETURN_NO;
2738	}	3120	}
2739		3121
2740	void	3122	void
2741	waiters (SV *self)	3123	waiters (SV *self)
2742	CODE:	3124	PPCODE:
2743	{	3125	{
2744	AV *av = (AV *)SvRV (self);	3126	AV *av = (AV *)SvRV (self);
		3127	int wcount = AvFILLp (av) + 1 - 1;
2745		3128
2746	if (GIMME_V == G_SCALAR)	3129	if (GIMME_V == G_SCALAR)
2747	XPUSHs (sv_2mortal (newSVsv (AvARRAY (av)[0])));	3130	XPUSHs (sv_2mortal (newSViv (wcount)));
2748	else	3131	else
2749	{	3132	{
2750	int i;	3133	int i;
2751	EXTEND (SP, AvFILLp (av) + 1 - 1);	3134	EXTEND (SP, wcount);
2752	for (i = 1; i <= AvFILLp (av); ++i)	3135	for (i = 1; i <= wcount; ++i)
2753	PUSHs (newSVsv (AvARRAY (av)[i]));	3136	PUSHs (sv_2mortal (newRV_inc (AvARRAY (av)[i])));
2754	}	3137	}
2755	}	3138	}
2756		3139
		3140	MODULE = Coro::State PACKAGE = Coro::Signal
		3141
		3142	SV *
		3143	new (SV *klass)
		3144	CODE:
		3145	RETVAL = sv_bless (
		3146	coro_waitarray_new (aTHX_ 0),
		3147	GvSTASH (CvGV (cv))
		3148	);
		3149	OUTPUT:
		3150	RETVAL
		3151
		3152	void
		3153	wait (...)
		3154	CODE:
		3155	CORO_EXECUTE_SLF_XS (slf_init_signal_wait);
		3156
		3157	void
		3158	broadcast (SV *self)
		3159	CODE:
		3160	{
		3161	AV *av = (AV *)SvRV (self);
		3162	coro_signal_wake (aTHX_ av, AvFILLp (av));
		3163	}
		3164
		3165	void
		3166	send (SV *self)
		3167	CODE:
		3168	{
		3169	AV *av = (AV *)SvRV (self);
		3170
		3171	if (AvFILLp (av))
		3172	coro_signal_wake (aTHX_ av, 1);
		3173	else
		3174	SvIVX (AvARRAY (av)[0]) = 1; /* remember the signal */
		3175	}
		3176
		3177	IV
		3178	awaited (SV *self)
		3179	CODE:
		3180	RETVAL = AvFILLp ((AV *)SvRV (self)) + 1 - 1;
		3181	OUTPUT:
		3182	RETVAL
		3183
		3184
		3185	MODULE = Coro::State PACKAGE = Coro::AnyEvent
		3186
		3187	BOOT:
		3188	sv_activity = coro_get_sv (aTHX_ "Coro::AnyEvent::ACTIVITY", TRUE);
		3189
		3190	void
		3191	_schedule (...)
		3192	CODE:
		3193	{
		3194	static int incede;
		3195
		3196	api_cede_notself (aTHX);
		3197
		3198	++incede;
		3199	while (coro_nready >= incede && api_cede (aTHX))
		3200	;
		3201
		3202	sv_setsv (sv_activity, &PL_sv_undef);
		3203	if (coro_nready >= incede)
		3204	{
		3205	PUSHMARK (SP);
		3206	PUTBACK;
		3207	call_pv ("Coro::AnyEvent::_activity", G_KEEPERR | G_EVAL | G_VOID | G_DISCARD);
		3208	}
		3209
		3210	--incede;
		3211	}
		3212
		3213
		3214	MODULE = Coro::State PACKAGE = Coro::AIO
		3215
		3216	void
		3217	_register (char *target, char *proto, SV *req)
		3218	CODE:
		3219	{
		3220	CV *req_cv = coro_sv_2cv (req);
		3221	/* newXSproto doesn't return the CV on 5.8 */
		3222	CV *slf_cv = newXS (target, coro_aio_req_xs, __FILE__);
		3223	sv_setpv ((SV *)slf_cv, proto);
		3224	sv_magicext ((SV *)slf_cv, (SV *)req_cv, CORO_MAGIC_type_aio, 0, 0, 0);
		3225	}
		3226

Diff Legend

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