ViewVC Help

View File | Revision Log | Show Annotations | Download File

/cvs/Coro/Coro/State.xs

Comparing Coro/Coro/State.xs (file contents):
Revision 1.283 by root, Sun Nov 16 11:12:57 2008 UTC vs.
Revision 1.306 by root, Wed Nov 19 11:11:10 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	AV *args; /* data associated with this coroutine (initial args) */	251	AV *args; /* data associated with this coroutine (initial args) */
…		…
257	/* statistics */	257	/* statistics */
258	int usecount; /* number of transfers to this coro */	258	int usecount; /* number of transfers to this coro */
259		259
260	/* coro process data */	260	/* coro process data */
261	int prio;	261	int prio;
262	SV *throw; /* exception to be thrown */	262	SV *except; /* exception to be thrown */
		263	SV *rouse_cb;
263		264
264	/* async_pool */	265	/* async_pool */
265	SV *saved_deffh;	266	SV *saved_deffh;
266		267
267	/* linked list */	268	/* linked list */
…		…
269	};	270	};
270		271
271	typedef struct coro *Coro__State;	272	typedef struct coro *Coro__State;
272	typedef struct coro *Coro__State_or_hashref;	273	typedef struct coro *Coro__State_or_hashref;
273		274
		275	/* the following variables are effectively part of the perl context */
		276	/* and get copied between struct coro and these variables */
		277	/* the mainr easonw e don't support windows process emulation */
274	static struct CoroSLF slf_frame; /* the current slf frame */	278	static struct CoroSLF slf_frame; /* the current slf frame */
275		279
276	/** Coro ********************************************************************/	280	/** Coro ********************************************************************/
277		281
278	#define PRIO_MAX 3	282	#define PRIO_MAX 3
…		…
380	SvREFCNT_dec (av); /* sv_magicext increased the refcount */	384	SvREFCNT_dec (av); /* sv_magicext increased the refcount */
381		385
382	return 0;	386	return 0;
383	}	387	}
384		388
385	#define CORO_MAGIC_type_cv PERL_MAGIC_ext	389	#define CORO_MAGIC_type_cv 26
386	#define CORO_MAGIC_type_state PERL_MAGIC_ext	390	#define CORO_MAGIC_type_state PERL_MAGIC_ext
387		391
388	static MGVTBL coro_cv_vtbl = {	392	static MGVTBL coro_cv_vtbl = {
389	0, 0, 0, 0,	393	0, 0, 0, 0,
390	coro_cv_free	394	coro_cv_free
391	};	395	};
392		396
		397	#define CORO_MAGIC_NN(sv, type) \
		398	(expect_true (SvMAGIC (sv)->mg_type == type) \
		399	? SvMAGIC (sv) \
		400	: mg_find (sv, type))
		401
393	#define CORO_MAGIC(sv, type) \	402	#define CORO_MAGIC(sv, type) \
394	expect_true (SvMAGIC (sv)) \	403	(expect_true (SvMAGIC (sv)) \
395	? expect_true (SvMAGIC (sv)->mg_type == type) \	404	? CORO_MAGIC_NN (sv, type) \
396	? SvMAGIC (sv) \
397	: mg_find (sv, type) \
398	: 0	405	: 0)
399		406
400	#define CORO_MAGIC_cv(cv) CORO_MAGIC (((SV *)(cv)), CORO_MAGIC_type_cv)	407	#define CORO_MAGIC_cv(cv) CORO_MAGIC (((SV *)(cv)), CORO_MAGIC_type_cv)
401	#define CORO_MAGIC_state(sv) CORO_MAGIC (((SV *)(sv)), CORO_MAGIC_type_state)	408	#define CORO_MAGIC_state(sv) CORO_MAGIC_NN (((SV *)(sv)), CORO_MAGIC_type_state)
402		409
403	INLINE struct coro *	410	INLINE struct coro *
404	SvSTATE_ (pTHX_ SV *coro)	411	SvSTATE_ (pTHX_ SV *coro)
405	{	412	{
406	HV *stash;	413	HV *stash;
…		…
424	return (struct coro *)mg->mg_ptr;	431	return (struct coro *)mg->mg_ptr;
425	}	432	}
426		433
427	#define SvSTATE(sv) SvSTATE_ (aTHX_ (sv))	434	#define SvSTATE(sv) SvSTATE_ (aTHX_ (sv))
428		435
429	/* fastert than SvSTATE, but expects a coroutine hv */	436	/* faster than SvSTATE, but expects a coroutine hv */
430	INLINE struct coro *	437	#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))	438	#define SvSTATE_current SvSTATE_hv (SvRV (coro_current))
441		439
442	/* the next two functions merely cache the padlists */	440	/* the next two functions merely cache the padlists */
443	static void	441	static void
444	get_padlist (pTHX_ CV *cv)	442	get_padlist (pTHX_ CV *cv)
…		…
451	else	449	else
452	{	450	{
453	#if CORO_PREFER_PERL_FUNCTIONS	451	#if CORO_PREFER_PERL_FUNCTIONS
454	/* this is probably cleaner? but also slower! */	452	/* this is probably cleaner? but also slower! */
455	/* in practise, it seems to be less stable */	453	/* in practise, it seems to be less stable */
456	CV *cp = Perl_cv_clone (cv);	454	CV *cp = Perl_cv_clone (aTHX_ cv);
457	CvPADLIST (cv) = CvPADLIST (cp);	455	CvPADLIST (cv) = CvPADLIST (cp);
458	CvPADLIST (cp) = 0;	456	CvPADLIST (cp) = 0;
459	SvREFCNT_dec (cp);	457	SvREFCNT_dec (cp);
460	#else	458	#else
461	CvPADLIST (cv) = coro_clone_padlist (aTHX_ cv);	459	CvPADLIST (cv) = coro_clone_padlist (aTHX_ cv);
…		…
513	}	511	}
514		512
515	PUTBACK;	513	PUTBACK;
516	}	514	}
517		515
518	slf_frame = c->slf_frame;	516	slf_frame = c->slf_frame;
		517	CORO_THROW = c->except;
519	}	518	}
520		519
521	static void	520	static void
522	save_perl (pTHX_ Coro__State c)	521	save_perl (pTHX_ Coro__State c)
523	{	522	{
		523	c->except = CORO_THROW;
524	c->slf_frame = slf_frame;	524	c->slf_frame = slf_frame;
525		525
526	{	526	{
527	dSP;	527	dSP;
528	I32 cxix = cxstack_ix;	528	I32 cxix = cxstack_ix;
…		…
603	* of perl.c:init_stacks, except that it uses less memory	603	* of perl.c:init_stacks, except that it uses less memory
604	* on the (sometimes correct) assumption that coroutines do	604	* on the (sometimes correct) assumption that coroutines do
605	* not usually need a lot of stackspace.	605	* not usually need a lot of stackspace.
606	*/	606	*/
607	#if CORO_PREFER_PERL_FUNCTIONS	607	#if CORO_PREFER_PERL_FUNCTIONS
608	# define coro_init_stacks init_stacks	608	# define coro_init_stacks(thx) init_stacks ()
609	#else	609	#else
610	static void	610	static void
611	coro_init_stacks (pTHX)	611	coro_init_stacks (pTHX)
612	{	612	{
613	PL_curstackinfo = new_stackinfo(32, 8);	613	PL_curstackinfo = new_stackinfo(32, 8);
…		…
821	slf_check_nop (pTHX_ struct CoroSLF *frame)	821	slf_check_nop (pTHX_ struct CoroSLF *frame)
822	{	822	{
823	return 0;	823	return 0;
824	}	824	}
825		825
826	static void	826	static UNOP coro_setup_op;
		827
		828	static void NOINLINE /* noinline to keep it out of the transfer fast path */
827	coro_setup (pTHX_ struct coro *coro)	829	coro_setup (pTHX_ struct coro *coro)
828	{	830	{
829	/*	831	/*
830	* emulate part of the perl startup here.	832	* emulate part of the perl startup here.
831	*/	833	*/
…		…
858	{	860	{
859	dSP;	861	dSP;
860	UNOP myop;	862	UNOP myop;
861		863
862	Zero (&myop, 1, UNOP);	864	Zero (&myop, 1, UNOP);
863	myop.op_next = Nullop;	865	myop.op_next = Nullop;
864	myop.op_flags = OPf_WANT_VOID;	866	myop.op_flags = OPf_WANT_VOID;
865		867
866	PUSHMARK (SP);	868	PUSHMARK (SP);
867	XPUSHs (sv_2mortal (av_shift (GvAV (PL_defgv))));	869	XPUSHs (sv_2mortal (av_shift (GvAV (PL_defgv))));
868	PUTBACK;	870	PUTBACK;
…		…
874	/* this newly created coroutine might be run on an existing cctx which most	876	/* this newly created coroutine might be run on an existing cctx which most
875	* likely was suspended in pp_slf, so we have to emulate entering pp_slf here.	877	* likely was suspended in pp_slf, so we have to emulate entering pp_slf here.
876	*/	878	*/
877	slf_frame.prepare = prepare_nop; /* provide a nop function for an eventual pp_slf */	879	slf_frame.prepare = prepare_nop; /* provide a nop function for an eventual pp_slf */
878	slf_frame.check = slf_check_nop; /* signal pp_slf to not repeat */	880	slf_frame.check = slf_check_nop; /* signal pp_slf to not repeat */
		881
		882	/* and we have to provide the pp_slf op in any case, so pp_slf can skip it */
		883	coro_setup_op.op_next = PL_op;
		884	coro_setup_op.op_type = OP_CUSTOM;
		885	coro_setup_op.op_ppaddr = pp_slf;
		886	/* no flags required, as an init function won't be called */
		887
		888	PL_op = (OP *)&coro_setup_op;
		889
		890	/* copy throw, in case it was set before coro_setup */
		891	CORO_THROW = coro->except;
879	}	892	}
880		893
881	static void	894	static void
882	coro_destruct (pTHX_ struct coro *coro)	895	coro_destruct (pTHX_ struct coro *coro)
883	{	896	{
…		…
906	SvREFCNT_dec (GvSV (irsgv));	919	SvREFCNT_dec (GvSV (irsgv));
907		920
908	SvREFCNT_dec (PL_diehook);	921	SvREFCNT_dec (PL_diehook);
909	SvREFCNT_dec (PL_warnhook);	922	SvREFCNT_dec (PL_warnhook);
910		923
		924	SvREFCNT_dec (CORO_THROW);
911	SvREFCNT_dec (coro->saved_deffh);	925	SvREFCNT_dec (coro->saved_deffh);
912	SvREFCNT_dec (coro->throw);	926	SvREFCNT_dec (coro->rouse_cb);
913		927
914	coro_destruct_stacks (aTHX);	928	coro_destruct_stacks (aTHX);
915	}	929	}
916		930
917	INLINE void	931	INLINE void
…		…
1046		1060
1047	TAINT_NOT;	1061	TAINT_NOT;
1048	return 0;	1062	return 0;
1049	}	1063	}
1050		1064
		1065	static struct coro_cctx *cctx_ssl_cctx;
		1066	static struct CoroSLF cctx_ssl_frame;
		1067
1051	static void	1068	static void
1052	prepare_set_stacklevel (struct coro_transfer_args *ta, struct coro_cctx *cctx)	1069	slf_prepare_set_stacklevel (pTHX_ struct coro_transfer_args *ta)
1053	{	1070	{
1054	ta->prev = (struct coro *)cctx;	1071	ta->prev = (struct coro *)cctx_ssl_cctx;
1055	ta->next = 0;	1072	ta->next = 0;
1056	}	1073	}
1057		1074
1058	/* inject a fake call to Coro::State::_cctx_init into the execution */	1075	static int
1059	/* _cctx_init should be careful, as it could be called at almost any time */	1076	slf_check_set_stacklevel (pTHX_ struct CoroSLF *frame)
1060	/* during execution of a perl program */	1077	{
1061	/* also initialises PL_top_env */	1078	*frame = cctx_ssl_frame;
		1079
		1080	return frame->check (aTHX_ frame); /* execute the restored frame - there must be one */
		1081	}
		1082
		1083	/* initialises PL_top_env and injects a pseudo-slf-call to set the stacklevel */
1062	static void NOINLINE	1084	static void NOINLINE
1063	cctx_prepare (pTHX_ coro_cctx *cctx)	1085	cctx_prepare (pTHX_ coro_cctx *cctx)
1064	{	1086	{
1065	dSP;
1066	UNOP myop;
1067
1068	PL_top_env = &PL_start_env;	1087	PL_top_env = &PL_start_env;
1069		1088
1070	if (cctx->flags & CC_TRACE)	1089	if (cctx->flags & CC_TRACE)
1071	PL_runops = runops_trace;	1090	PL_runops = runops_trace;
1072		1091
1073	Zero (&myop, 1, UNOP);	1092	/* we already must be executing an SLF op, there is no other valid way
1074	myop.op_next = PL_op;	1093	* that can lead to creation of a new cctx */
1075	myop.op_flags = OPf_WANT_VOID | OPf_STACKED;	1094	assert (("FATAL: can't prepare slf-less cctx in Coro module (please report)",
		1095	slf_frame.prepare && PL_op->op_ppaddr == pp_slf));
1076		1096
1077	PUSHMARK (SP);	1097	/* we must emulate leaving pp_slf, which is done inside slf_check_set_stacklevel */
1078	EXTEND (SP, 2);	1098	cctx_ssl_cctx = cctx;
1079	PUSHs (sv_2mortal (newSViv ((IV)cctx)));	1099	cctx_ssl_frame = slf_frame;
1080	PUSHs ((SV *)get_cv ("Coro::State::_cctx_init", FALSE));	1100
1081	PUTBACK;	1101	slf_frame.prepare = slf_prepare_set_stacklevel;
1082	PL_op = (OP *)&myop;	1102	slf_frame.check = slf_check_set_stacklevel;
1083	PL_op = PL_ppaddr[OP_ENTERSUB](aTHX);
1084	SPAGAIN;
1085	}	1103	}
1086		1104
1087	/* the tail of transfer: execute stuff we can only do after a transfer */	1105	/* the tail of transfer: execute stuff we can only do after a transfer */
1088	INLINE void	1106	INLINE void
1089	transfer_tail (pTHX)	1107	transfer_tail (pTHX)
…		…
1276	/** coroutine switching *****************************************************/	1294	/** coroutine switching *****************************************************/
1277		1295
1278	static void	1296	static void
1279	transfer_check (pTHX_ struct coro *prev, struct coro *next)	1297	transfer_check (pTHX_ struct coro *prev, struct coro *next)
1280	{	1298	{
		1299	/* TODO: throwing up here is considered harmful */
		1300
1281	if (expect_true (prev != next))	1301	if (expect_true (prev != next))
1282	{	1302	{
1283	if (expect_false (!(prev->flags & (CF_RUNNING | CF_NEW))))	1303	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,");	1304	croak ("Coro::State::transfer called with non-running/new prev Coro::State, but can only transfer from running or new states,");
1285		1305
…		…
1295	#endif	1315	#endif
1296	}	1316	}
1297	}	1317	}
1298		1318
1299	/* always use the TRANSFER macro */	1319	/* always use the TRANSFER macro */
1300	static void NOINLINE	1320	static void NOINLINE /* noinline so we have a fixed stackframe */
1301	transfer (pTHX_ struct coro *prev, struct coro *next, int force_cctx)	1321	transfer (pTHX_ struct coro *prev, struct coro *next, int force_cctx)
1302	{	1322	{
1303	dSTACKLEVEL;	1323	dSTACKLEVEL;
1304		1324
1305	/* sometimes transfer is only called to set idle_sp */	1325	/* sometimes transfer is only called to set idle_sp */
1306	if (expect_false (!next))	1326	if (expect_false (!next))
1307	{	1327	{
1308	((coro_cctx *)prev)->idle_sp = (void *)stacklevel;	1328	((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 */	1329	assert (((coro_cctx *)prev)->idle_te = PL_top_env); /* just for the side-effect when asserts are enabled */
1310	}	1330	}
1311	else if (expect_true (prev != next))	1331	else if (expect_true (prev != next))
1312	{	1332	{
1313	coro_cctx *prev__cctx;	1333	coro_cctx *prev__cctx;
…		…
1338		1358
1339	prev__cctx = prev->cctx;	1359	prev__cctx = prev->cctx;
1340		1360
1341	/* possibly untie and reuse the cctx */	1361	/* possibly untie and reuse the cctx */
1342	if (expect_true (	1362	if (expect_true (
1343	prev__cctx->idle_sp == (void *)stacklevel	1363	prev__cctx->idle_sp == STACKLEVEL
1344	&& !(prev__cctx->flags & CC_TRACE)	1364	&& !(prev__cctx->flags & CC_TRACE)
1345	&& !force_cctx	1365	&& !force_cctx
1346	))	1366	))
1347	{	1367	{
1348	/* I assume that stacklevel is a stronger indicator than PL_top_env changes */	1368	/* I assume that stacklevel is a stronger indicator than PL_top_env changes */
…		…
1479		1499
1480	prepare_transfer (aTHX_ &ta, prev_sv, next_sv);	1500	prepare_transfer (aTHX_ &ta, prev_sv, next_sv);
1481	TRANSFER (ta, 1);	1501	TRANSFER (ta, 1);
1482	}	1502	}
1483		1503
		1504	/*****************************************************************************/
		1505	/* gensub: simple closure generation utility */
		1506
		1507	#define GENSUB_ARG CvXSUBANY (cv).any_ptr
		1508
		1509	/* create a closure from XS, returns a code reference */
		1510	/* the arg can be accessed via GENSUB_ARG from the callback */
		1511	/* the callback must use dXSARGS/XSRETURN */
		1512	static SV *
		1513	gensub (pTHX_ void (*xsub)(pTHX_ CV *), void *arg)
		1514	{
		1515	CV *cv = (CV *)newSV (0);
		1516
		1517	sv_upgrade ((SV *)cv, SVt_PVCV);
		1518
		1519	CvANON_on (cv);
		1520	CvISXSUB_on (cv);
		1521	CvXSUB (cv) = xsub;
		1522	GENSUB_ARG = arg;
		1523
		1524	return newRV_noinc ((SV *)cv);
		1525	}
		1526
1484	/** Coro ********************************************************************/	1527	/** Coro ********************************************************************/
1485		1528
1486	INLINE void	1529	INLINE void
1487	coro_enq (pTHX_ struct coro *coro)	1530	coro_enq (pTHX_ struct coro *coro)
1488	{	1531	{
…		…
1531	ENTER;	1574	ENTER;
1532	SAVETMPS;	1575	SAVETMPS;
1533		1576
1534	PUSHMARK (SP);	1577	PUSHMARK (SP);
1535	PUTBACK;	1578	PUTBACK;
1536	call_sv (sv_hook, G_DISCARD);	1579	call_sv (sv_hook, G_VOID | G_DISCARD);
1537	SPAGAIN;
1538		1580
1539	FREETMPS;	1581	FREETMPS;
1540	LEAVE;	1582	LEAVE;
1541	}	1583	}
1542		1584
…		…
1569	ENTER;	1611	ENTER;
1570	SAVETMPS;	1612	SAVETMPS;
1571		1613
1572	PUSHMARK (SP);	1614	PUSHMARK (SP);
1573	PUTBACK;	1615	PUTBACK;
1574	call_sv (get_sv ("Coro::idle", FALSE), G_DISCARD);	1616	call_sv (get_sv ("Coro::idle", FALSE), G_VOID | G_DISCARD);
1575	SPAGAIN;
1576		1617
1577	FREETMPS;	1618	FREETMPS;
1578	LEAVE;	1619	LEAVE;
1579	continue;	1620	continue;
1580	}	1621	}
…		…
1687	if (coro->flags & CF_RUNNING)	1728	if (coro->flags & CF_RUNNING)
1688	PL_runops = RUNOPS_DEFAULT;	1729	PL_runops = RUNOPS_DEFAULT;
1689	else	1730	else
1690	coro->slot->runops = RUNOPS_DEFAULT;	1731	coro->slot->runops = RUNOPS_DEFAULT;
1691	}	1732	}
		1733	}
		1734
		1735	/*****************************************************************************/
		1736	/* rouse callback */
		1737
		1738	#define CORO_MAGIC_type_rouse PERL_MAGIC_ext
		1739
		1740	static void
		1741	coro_rouse_callback (pTHX_ CV *cv)
		1742	{
		1743	dXSARGS;
		1744	SV *data = (SV *)GENSUB_ARG;
		1745
		1746	if (SvTYPE (SvRV (data)) != SVt_PVAV)
		1747	{
		1748	/* first call, set args */
		1749	int i;
		1750	AV *av = newAV ();
		1751	SV *coro = SvRV (data);
		1752
		1753	SvRV_set (data, (SV *)av);
		1754	api_ready (aTHX_ coro);
		1755	SvREFCNT_dec (coro);
		1756
		1757	/* better take a full copy of the arguments */
		1758	while (items--)
		1759	av_store (av, items, newSVsv (ST (items)));
		1760	}
		1761
		1762	XSRETURN_EMPTY;
		1763	}
		1764
		1765	static int
		1766	slf_check_rouse_wait (pTHX_ struct CoroSLF *frame)
		1767	{
		1768	SV *data = (SV *)frame->data;
		1769
		1770	if (CORO_THROW)
		1771	return 0;
		1772
		1773	if (SvTYPE (SvRV (data)) != SVt_PVAV)
		1774	return 1;
		1775
		1776	/* now push all results on the stack */
		1777	{
		1778	dSP;
		1779	AV *av = (AV *)SvRV (data);
		1780	int i;
		1781
		1782	EXTEND (SP, AvFILLp (av) + 1);
		1783	for (i = 0; i <= AvFILLp (av); ++i)
		1784	PUSHs (sv_2mortal (AvARRAY (av)[i]));
		1785
		1786	/* we have stolen the elements, so ste length to zero and free */
		1787	AvFILLp (av) = -1;
		1788	av_undef (av);
		1789
		1790	PUTBACK;
		1791	}
		1792
		1793	return 0;
		1794	}
		1795
		1796	static void
		1797	slf_init_rouse_wait (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		1798	{
		1799	SV *cb;
		1800
		1801	if (items)
		1802	cb = arg [0];
		1803	else
		1804	{
		1805	struct coro *coro = SvSTATE_current;
		1806
		1807	if (!coro->rouse_cb)
		1808	croak ("Coro::rouse_wait called without rouse callback, and no default rouse callback found either,");
		1809
		1810	cb = sv_2mortal (coro->rouse_cb);
		1811	coro->rouse_cb = 0;
		1812	}
		1813
		1814	if (!SvROK (cb)
		1815	|| SvTYPE (SvRV (cb)) != SVt_PVCV
		1816	|| CvXSUB ((CV *)SvRV (cb)) != coro_rouse_callback)
		1817	croak ("Coro::rouse_wait called with illegal callback argument,");
		1818
		1819	{
		1820	CV *cv = (CV *)SvRV (cb); /* for GENSUB_ARG */
		1821	SV *data = (SV *)GENSUB_ARG;
		1822
		1823	frame->data = (void *)data;
		1824	frame->prepare = SvTYPE (SvRV (data)) == SVt_PVAV ? prepare_nop : prepare_schedule;
		1825	frame->check = slf_check_rouse_wait;
		1826	}
		1827	}
		1828
		1829	static SV *
		1830	coro_new_rouse_cb (pTHX)
		1831	{
		1832	HV *hv = (HV *)SvRV (coro_current);
		1833	struct coro *coro = SvSTATE_hv (hv);
		1834	SV *data = newRV_inc ((SV *)hv);
		1835	SV *cb = gensub (aTHX_ coro_rouse_callback, (void *)data);
		1836
		1837	sv_magicext (SvRV (cb), data, CORO_MAGIC_type_rouse, 0, 0, 0);
		1838	SvREFCNT_dec (data); /* magicext increases the refcount */
		1839
		1840	SvREFCNT_dec (coro->rouse_cb);
		1841	coro->rouse_cb = SvREFCNT_inc_NN (cb);
		1842
		1843	return cb;
		1844	}
		1845
		1846	/*****************************************************************************/
		1847	/* schedule-like-function opcode (SLF) */
		1848
		1849	static UNOP slf_restore; /* restore stack as entersub did, for first-re-run */
		1850	static const CV *slf_cv;
		1851	static SV **slf_argv;
		1852	static int slf_argc, slf_arga; /* count, allocated */
		1853	static I32 slf_ax; /* top of stack, for restore */
		1854
		1855	/* this restores the stack in the case we patched the entersub, to */
		1856	/* recreate the stack frame as perl will on following calls */
		1857	/* since entersub cleared the stack */
		1858	static OP *
		1859	pp_restore (pTHX)
		1860	{
		1861	int i;
		1862	SV **SP = PL_stack_base + slf_ax;
		1863
		1864	PUSHMARK (SP);
		1865
		1866	EXTEND (SP, slf_argc + 1);
		1867
		1868	for (i = 0; i < slf_argc; ++i)
		1869	PUSHs (sv_2mortal (slf_argv [i]));
		1870
		1871	PUSHs ((SV *)CvGV (slf_cv));
		1872
		1873	RETURNOP (slf_restore.op_first);
		1874	}
		1875
		1876	static void
		1877	slf_prepare_transfer (pTHX_ struct coro_transfer_args *ta)
		1878	{
		1879	SV **arg = (SV **)slf_frame.data;
		1880
		1881	prepare_transfer (aTHX_ ta, arg [0], arg [1]);
		1882	}
		1883
		1884	static void
		1885	slf_init_transfer (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		1886	{
		1887	if (items != 2)
		1888	croak ("Coro::State::transfer (prev, next) expects two arguments, not %d,", items);
		1889
		1890	frame->prepare = slf_prepare_transfer;
		1891	frame->check = slf_check_nop;
		1892	frame->data = (void *)arg; /* let's hope it will stay valid */
		1893	}
		1894
		1895	static void
		1896	slf_init_schedule (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		1897	{
		1898	frame->prepare = prepare_schedule;
		1899	frame->check = slf_check_nop;
		1900	}
		1901
		1902	static void
		1903	slf_init_cede (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		1904	{
		1905	frame->prepare = prepare_cede;
		1906	frame->check = slf_check_nop;
		1907	}
		1908
		1909	static void
		1910	slf_init_cede_notself (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		1911	{
		1912	frame->prepare = prepare_cede_notself;
		1913	frame->check = slf_check_nop;
		1914	}
		1915
		1916	/*
		1917	* these not obviously related functions are all rolled into one
		1918	* function to increase chances that they all will call transfer with the same
		1919	* stack offset
		1920	* SLF stands for "schedule-like-function".
		1921	*/
		1922	static OP *
		1923	pp_slf (pTHX)
		1924	{
		1925	I32 checkmark; /* mark SP to see how many elements check has pushed */
		1926
		1927	/* set up the slf frame, unless it has already been set-up */
		1928	/* the latter happens when a new coro has been started */
		1929	/* or when a new cctx was attached to an existing coroutine */
		1930	if (expect_true (!slf_frame.prepare))
		1931	{
		1932	/* first iteration */
		1933	dSP;
		1934	SV **arg = PL_stack_base + TOPMARK + 1;
		1935	int items = SP - arg; /* args without function object */
		1936	SV *gv = *sp;
		1937
		1938	/* do a quick consistency check on the "function" object, and if it isn't */
		1939	/* for us, divert to the real entersub */
		1940	if (SvTYPE (gv) != SVt_PVGV
		1941	|| !GvCV (gv)
		1942	|| !(CvFLAGS (GvCV (gv)) & CVf_SLF))
		1943	return PL_ppaddr[OP_ENTERSUB](aTHX);
		1944
		1945	if (!(PL_op->op_flags & OPf_STACKED))
		1946	{
		1947	/* ampersand-form of call, use @_ instead of stack */
		1948	AV *av = GvAV (PL_defgv);
		1949	arg = AvARRAY (av);
		1950	items = AvFILLp (av) + 1;
		1951	}
		1952
		1953	/* now call the init function, which needs to set up slf_frame */
		1954	((coro_slf_cb)CvXSUBANY (GvCV (gv)).any_ptr)
		1955	(aTHX_ &slf_frame, GvCV (gv), arg, items);
		1956
		1957	/* pop args */
		1958	SP = PL_stack_base + POPMARK;
		1959
		1960	PUTBACK;
		1961	}
		1962
		1963	/* now that we have a slf_frame, interpret it! */
		1964	/* we use a callback system not to make the code needlessly */
		1965	/* complicated, but so we can run multiple perl coros from one cctx */
		1966
		1967	do
		1968	{
		1969	struct coro_transfer_args ta;
		1970
		1971	slf_frame.prepare (aTHX_ &ta);
		1972	TRANSFER (ta, 0);
		1973
		1974	checkmark = PL_stack_sp - PL_stack_base;
		1975	}
		1976	while (slf_frame.check (aTHX_ &slf_frame));
		1977
		1978	slf_frame.prepare = 0; /* invalidate the frame, we are done processing it */
		1979
		1980	/* exception handling */
		1981	if (expect_false (CORO_THROW))
		1982	{
		1983	SV *exception = sv_2mortal (CORO_THROW);
		1984
		1985	CORO_THROW = 0;
		1986	sv_setsv (ERRSV, exception);
		1987	croak (0);
		1988	}
		1989
		1990	/* return value handling - mostly like entersub */
		1991	/* make sure we put something on the stack in scalar context */
		1992	if (GIMME_V == G_SCALAR)
		1993	{
		1994	dSP;
		1995	SV **bot = PL_stack_base + checkmark;
		1996
		1997	if (sp == bot) /* too few, push undef */
		1998	bot [1] = &PL_sv_undef;
		1999	else if (sp != bot + 1) /* too many, take last one */
		2000	bot [1] = *sp;
		2001
		2002	SP = bot + 1;
		2003
		2004	PUTBACK;
		2005	}
		2006
		2007	return NORMAL;
		2008	}
		2009
		2010	static void
		2011	api_execute_slf (pTHX_ CV *cv, coro_slf_cb init_cb, I32 ax)
		2012	{
		2013	int i;
		2014	SV **arg = PL_stack_base + ax;
		2015	int items = PL_stack_sp - arg + 1;
		2016
		2017	assert (("FATAL: SLF call with illegal CV value", !CvANON (cv)));
		2018
		2019	if (PL_op->op_ppaddr != PL_ppaddr [OP_ENTERSUB]
		2020	&& PL_op->op_ppaddr != pp_slf)
		2021	croak ("FATAL: Coro SLF calls can only be made normally, not via goto or any other means, caught");
		2022
		2023	CvFLAGS (cv) |= CVf_SLF;
		2024	CvXSUBANY (cv).any_ptr = (void *)init_cb;
		2025	slf_cv = cv;
		2026
		2027	/* we patch the op, and then re-run the whole call */
		2028	/* we have to put the same argument on the stack for this to work */
		2029	/* and this will be done by pp_restore */
		2030	slf_restore.op_next = (OP *)&slf_restore;
		2031	slf_restore.op_type = OP_CUSTOM;
		2032	slf_restore.op_ppaddr = pp_restore;
		2033	slf_restore.op_first = PL_op;
		2034
		2035	slf_ax = ax - 1; /* undo the ax++ inside dAXMARK */
		2036
		2037	if (PL_op->op_flags & OPf_STACKED)
		2038	{
		2039	if (items > slf_arga)
		2040	{
		2041	slf_arga = items;
		2042	free (slf_argv);
		2043	slf_argv = malloc (slf_arga * sizeof (SV *));
		2044	}
		2045
		2046	slf_argc = items;
		2047
		2048	for (i = 0; i < items; ++i)
		2049	slf_argv [i] = SvREFCNT_inc (arg [i]);
		2050	}
		2051	else
		2052	slf_argc = 0;
		2053
		2054	PL_op->op_ppaddr = pp_slf;
		2055	PL_op->op_type = OP_CUSTOM; /* maybe we should leave it at entersub? */
		2056
		2057	PL_op = (OP *)&slf_restore;
1692	}	2058	}
1693		2059
1694	/*****************************************************************************/	2060	/*****************************************************************************/
1695	/* PerlIO::cede */	2061	/* PerlIO::cede */
1696		2062
…		…
1765	PerlIOBuf_get_cnt,	2131	PerlIOBuf_get_cnt,
1766	PerlIOBuf_set_ptrcnt,	2132	PerlIOBuf_set_ptrcnt,
1767	};	2133	};
1768		2134
1769	/*****************************************************************************/	2135	/*****************************************************************************/
		2136	/* Coro::Semaphore & Coro::Signal */
1770		2137
1771	static const CV *slf_cv; /* for quick consistency check */
1772
1773	static UNOP slf_restore; /* restore stack as entersub did, for first-re-run */
1774	static SV *slf_arg0;
1775	static SV *slf_arg1;
1776	static SV *slf_arg2;
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 *	2138	static SV *
1782	pp_restore (pTHX)	2139	coro_waitarray_new (pTHX_ int count)
1783	{	2140	{
1784	dSP;	2141	/* a semaphore contains a counter IV in $sem->[0] and any waiters after that */
		2142	AV *av = newAV ();
		2143	SV **ary;
1785		2144
1786	PUSHMARK (SP);	2145	/* unfortunately, building manually saves memory */
		2146	Newx (ary, 2, SV *);
		2147	AvALLOC (av) = ary;
		2148	/*AvARRAY (av) = ary;*/
		2149	SvPVX ((SV *)av) = (char *)ary; /* 5.8.8 needs this syntax instead of AvARRAY = ary */
		2150	AvMAX (av) = 1;
		2151	AvFILLp (av) = 0;
		2152	ary [0] = newSViv (count);
1787		2153
1788	EXTEND (SP, 3);	2154	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	}	2155	}
1796		2156
1797	static void	2157	/* 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	PUTBACK;
1903
1904	/* now call the init function, which needs to set up slf_frame */
1905	((coro_slf_cb)CvXSUBANY (GvCV (gv)).any_ptr)
1906	(aTHX_ &slf_frame, GvCV (gv), arg, items);
1907
1908	/* pop args */
1909	SP = PL_stack_base + POPMARK;
1910
1911	PUTBACK;
1912	}
1913
1914	/* now that we have a slf_frame, interpret it! */
1915	/* we use a callback system not to make the code needlessly */
1916	/* complicated, but so we can run multiple perl coros from one cctx */
1917
1918	do
1919	{
1920	struct coro_transfer_args ta;
1921
1922	slf_frame.prepare (aTHX_ &ta);
1923	TRANSFER (ta, 0);
1924
1925	checkmark = PL_stack_sp - PL_stack_base;
1926	}
1927	while (slf_frame.check (aTHX_ &slf_frame));
1928
1929	{
1930	dSP;
1931	SV **bot = PL_stack_base + checkmark;
1932	int gimme = GIMME_V;
1933
1934	slf_frame.prepare = 0; /* invalidate the frame, so it gets initialised again next time */
1935
1936	/* make sure we put something on the stack in scalar context */
1937	if (gimme == G_SCALAR)
1938	{
1939	if (sp == bot)
1940	XPUSHs (&PL_sv_undef);
1941
1942	SP = bot + 1;
1943	}
1944
1945	PUTBACK;
1946	}
1947
1948	{
1949	struct coro *coro = SvSTATE_current;
1950
1951	if (expect_false (coro->throw))
1952	{
1953	SV *exception = sv_2mortal (coro->throw);
1954
1955	coro->throw = 0;
1956	sv_setsv (ERRSV, exception);
1957	croak (0);
1958	}
1959	}
1960
1961	return NORMAL;
1962	}
1963
1964	static void
1965	api_execute_slf (pTHX_ CV *cv, coro_slf_cb init_cb, SV **arg, int items)
1966	{
1967	assert (("FATAL: SLF call with illegal CV value", !CvANON (cv)));
1968
1969	if (PL_op->op_ppaddr != PL_ppaddr [OP_ENTERSUB]
1970	&& PL_op->op_ppaddr != pp_slf)
1971	croak ("FATAL: Coro SLF calls can only be made normally, not via goto or any other means, caught");
1972
1973	if (items > 3)
1974	croak ("Coro only supports up to three arguments to SLF functions currently (not %d), caught", items);
1975
1976	CvFLAGS (cv) |= CVf_SLF;
1977	CvXSUBANY (cv).any_ptr = (void *)init_cb;
1978	slf_cv = cv;
1979
1980	/* we patch the op, and then re-run the whole call */
1981	/* we have to put the same argument on the stack for this to work */
1982	/* and this will be done by pp_restore */
1983	slf_restore.op_next = (OP *)&slf_restore;
1984	slf_restore.op_type = OP_CUSTOM;
1985	slf_restore.op_ppaddr = pp_restore;
1986	slf_restore.op_first = PL_op;
1987
1988	slf_arg0 = items > 0 ? SvREFCNT_inc (arg [0]) : 0;
1989	slf_arg1 = items > 1 ? SvREFCNT_inc (arg [1]) : 0;
1990	slf_arg2 = items > 2 ? SvREFCNT_inc (arg [2]) : 0;
1991
1992	PL_op->op_ppaddr = pp_slf;
1993
1994	PL_op = (OP *)&slf_restore;
1995	}
1996
1997	/*****************************************************************************/
1998		2158
1999	static void	2159	static void
2000	coro_semaphore_adjust (pTHX_ AV *av, IV adjust)	2160	coro_semaphore_adjust (pTHX_ AV *av, IV adjust)
2001	{	2161	{
2002	SV *count_sv = AvARRAY (av)[0];	2162	SV *count_sv = AvARRAY (av)[0];
…		…
2014	AvARRAY (av)[0] = AvARRAY (av)[1];	2174	AvARRAY (av)[0] = AvARRAY (av)[1];
2015	AvARRAY (av)[1] = count_sv;	2175	AvARRAY (av)[1] = count_sv;
2016	cb = av_shift (av);	2176	cb = av_shift (av);
2017		2177
2018	if (SvOBJECT (cb))	2178	if (SvOBJECT (cb))
		2179	{
2019	api_ready (aTHX_ cb);	2180	api_ready (aTHX_ cb);
2020	else	2181	--count;
2021	croak ("callbacks not yet supported");	2182	}
		2183	else if (SvTYPE (cb) == SVt_PVCV)
		2184	{
		2185	dSP;
		2186	PUSHMARK (SP);
		2187	XPUSHs (sv_2mortal (newRV_inc ((SV *)av)));
		2188	PUTBACK;
		2189	call_sv (cb, G_VOID | G_DISCARD | G_EVAL | G_KEEPERR);
		2190	}
2022		2191
2023	SvREFCNT_dec (cb);	2192	SvREFCNT_dec (cb);
2024
2025	--count;
2026	}	2193	}
2027	}	2194	}
2028		2195
2029	static void	2196	static void
2030	coro_semaphore_on_destroy (pTHX_ struct coro *coro)	2197	coro_semaphore_on_destroy (pTHX_ struct coro *coro)
…		…
2032	/* call $sem->adjust (0) to possibly wake up some other waiters */	2199	/* call $sem->adjust (0) to possibly wake up some other waiters */
2033	coro_semaphore_adjust (aTHX_ (AV *)coro->slf_frame.data, 0);	2200	coro_semaphore_adjust (aTHX_ (AV *)coro->slf_frame.data, 0);
2034	}	2201	}
2035		2202
2036	static int	2203	static int
2037	slf_check_semaphore_down (pTHX_ struct CoroSLF *frame)	2204	slf_check_semaphore_down_or_wait (pTHX_ struct CoroSLF *frame, int acquire)
2038	{	2205	{
2039	AV *av = (AV *)frame->data;	2206	AV *av = (AV *)frame->data;
2040	SV *count_sv = AvARRAY (av)[0];	2207	SV *count_sv = AvARRAY (av)[0];
2041		2208
		2209	/* if we are about to throw, don't actually acquire the lock, just throw */
		2210	if (CORO_THROW)
		2211	return 0;
2042	if (SvIVX (count_sv) > 0)	2212	else if (SvIVX (count_sv) > 0)
2043	{	2213	{
2044	SvSTATE_current->on_destroy = 0;	2214	SvSTATE_current->on_destroy = 0;
		2215
		2216	if (acquire)
2045	SvIVX (count_sv) = SvIVX (count_sv) - 1;	2217	SvIVX (count_sv) = SvIVX (count_sv) - 1;
		2218	else
		2219	coro_semaphore_adjust (aTHX_ av, 0);
		2220
2046	return 0;	2221	return 0;
2047	}	2222	}
2048	else	2223	else
2049	{	2224	{
2050	int i;	2225	int i;
…		…
2059	av_push (av, SvREFCNT_inc (SvRV (coro_current)));	2234	av_push (av, SvREFCNT_inc (SvRV (coro_current)));
2060	return 1;	2235	return 1;
2061	}	2236	}
2062	}	2237	}
2063		2238
2064	static void	2239	static int
		2240	slf_check_semaphore_down (pTHX_ struct CoroSLF *frame)
		2241	{
		2242	return slf_check_semaphore_down_or_wait (aTHX_ frame, 1);
		2243	}
		2244
		2245	static int
		2246	slf_check_semaphore_wait (pTHX_ struct CoroSLF *frame)
		2247	{
		2248	return slf_check_semaphore_down_or_wait (aTHX_ frame, 0);
		2249	}
		2250
		2251	static void
2065	slf_init_semaphore_down (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)	2252	slf_init_semaphore_down_or_wait (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
2066	{	2253	{
2067	AV *av = (AV *)SvRV (arg [0]);	2254	AV *av = (AV *)SvRV (arg [0]);
2068		2255
2069	if (SvIVX (AvARRAY (av)[0]) > 0)	2256	if (SvIVX (AvARRAY (av)[0]) > 0)
2070	{	2257	{
2071	frame->data = (void *)av;	2258	frame->data = (void *)av;
2072	frame->prepare = prepare_nop;	2259	frame->prepare = prepare_nop;
2073	SvSTATE_current->on_destroy = coro_semaphore_on_destroy;
2074	}	2260	}
2075	else	2261	else
2076	{	2262	{
2077	av_push (av, SvREFCNT_inc (SvRV (coro_current)));	2263	av_push (av, SvREFCNT_inc (SvRV (coro_current)));
2078		2264
2079	frame->data = (void *)sv_2mortal (SvREFCNT_inc ((SV *)av));	2265	frame->data = (void *)sv_2mortal (SvREFCNT_inc ((SV *)av));
2080	frame->prepare = prepare_schedule;	2266	frame->prepare = prepare_schedule;
2081		2267
2082	/* to avoid race conditions when a woken-up coro gets terminated */	2268	/* to avoid race conditions when a woken-up coro gets terminated */
2083	/* we arrange for a temporary on_destroy that calls adjust (0) */	2269	/* we arrange for a temporary on_destroy that calls adjust (0) */
2084	assert (!SvSTATE_current->on_destroy);//D
2085	SvSTATE_current->on_destroy = coro_semaphore_on_destroy;	2270	SvSTATE_current->on_destroy = coro_semaphore_on_destroy;
2086	}	2271	}
		2272	}
2087		2273
		2274	static void
		2275	slf_init_semaphore_down (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		2276	{
		2277	slf_init_semaphore_down_or_wait (aTHX_ frame, cv, arg, items);
2088	frame->check = slf_check_semaphore_down;	2278	frame->check = slf_check_semaphore_down;
		2279	}
2089		2280
		2281	static void
		2282	slf_init_semaphore_wait (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		2283	{
		2284	if (items >= 2)
		2285	{
		2286	/* callback form */
		2287	AV *av = (AV *)SvRV (arg [0]);
		2288	HV *st;
		2289	GV *gvp;
		2290	CV *cb_cv = sv_2cv (arg [1], &st, &gvp, 0);
		2291
		2292	av_push (av, (SV *)SvREFCNT_inc_NN (cb_cv));
		2293
		2294	if (SvIVX (AvARRAY (av)[0]) > 0)
		2295	coro_semaphore_adjust (aTHX_ av, 0);
		2296
		2297	frame->prepare = prepare_nop;
		2298	frame->check = slf_check_nop;
		2299	}
		2300	else
		2301	{
		2302	slf_init_semaphore_down_or_wait (aTHX_ frame, cv, arg, items);
		2303	frame->check = slf_check_semaphore_wait;
		2304	}
		2305	}
		2306
		2307	/* signal */
		2308
		2309	static void
		2310	coro_signal_wake (pTHX_ AV *av, int count)
		2311	{
		2312	SvIVX (AvARRAY (av)[0]) = 0;
		2313
		2314	/* now signal count waiters */
		2315	while (count > 0 && AvFILLp (av) > 0)
		2316	{
		2317	SV *cb;
		2318
		2319	/* swap first two elements so we can shift a waiter */
		2320	cb = AvARRAY (av)[0];
		2321	AvARRAY (av)[0] = AvARRAY (av)[1];
		2322	AvARRAY (av)[1] = cb;
		2323
		2324	cb = av_shift (av);
		2325
		2326	api_ready (aTHX_ cb);
		2327	sv_setiv (cb, 0); /* signal waiter */
		2328	SvREFCNT_dec (cb);
		2329
		2330	--count;
		2331	}
		2332	}
		2333
		2334	static int
		2335	slf_check_signal_wait (pTHX_ struct CoroSLF *frame)
		2336	{
		2337	/* if we are about to throw, also stop waiting */
		2338	return SvROK ((SV *)frame->data) && !CORO_THROW;
		2339	}
		2340
		2341	static void
		2342	slf_init_signal_wait (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		2343	{
		2344	AV *av = (AV *)SvRV (arg [0]);
		2345
		2346	if (SvIVX (AvARRAY (av)[0]))
		2347	{
		2348	SvIVX (AvARRAY (av)[0]) = 0;
		2349	frame->prepare = prepare_nop;
		2350	frame->check = slf_check_nop;
		2351	}
		2352	else
		2353	{
		2354	SV *waiter = newRV_inc (SvRV (coro_current)); /* owned by signal av */
		2355
		2356	av_push (av, waiter);
		2357
		2358	frame->data = (void *)sv_2mortal (SvREFCNT_inc_NN (waiter)); /* owned by process */
		2359	frame->prepare = prepare_schedule;
		2360	frame->check = slf_check_signal_wait;
		2361	}
2090	}	2362	}
2091		2363
2092	/*****************************************************************************/	2364	/*****************************************************************************/
		2365	/* Coro::AIO */
2093		2366
2094	#define GENSUB_ARG CvXSUBANY (cv).any_ptr	2367	#define CORO_MAGIC_type_aio PERL_MAGIC_ext
2095		2368
2096	/* create a closure from XS, returns a code reference */	2369	/* helper storage struct */
2097	/* the arg can be accessed via GENSUB_ARG from the callback */	2370	struct io_state
2098	/* the callback must use dXSARGS/XSRETURN */
2099	static SV *
2100	gensub (pTHX_ void (*xsub)(pTHX_ CV *), void *arg)
2101	{	2371	{
2102	CV *cv = (CV *)NEWSV (0, 0);	2372	int errorno;
		2373	I32 laststype; /* U16 in 5.10.0 */
		2374	int laststatval;
		2375	Stat_t statcache;
		2376	};
2103		2377
		2378	static void
		2379	coro_aio_callback (pTHX_ CV *cv)
		2380	{
		2381	dXSARGS;
		2382	AV *state = (AV *)GENSUB_ARG;
		2383	SV *coro = av_pop (state);
		2384	SV *data_sv = newSV (sizeof (struct io_state));
		2385
		2386	av_extend (state, items);
		2387
2104	sv_upgrade ((SV *)cv, SVt_PVCV);	2388	sv_upgrade (data_sv, SVt_PV);
		2389	SvCUR_set (data_sv, sizeof (struct io_state));
		2390	SvPOK_only (data_sv);
2105		2391
2106	CvANON_on (cv);	2392	{
2107	CvISXSUB_on (cv);	2393	struct io_state *data = (struct io_state *)SvPVX (data_sv);
2108	CvXSUB (cv) = xsub;
2109	GENSUB_ARG = arg;
2110		2394
2111	return newRV_noinc ((SV *)cv);	2395	data->errorno = errno;
		2396	data->laststype = PL_laststype;
		2397	data->laststatval = PL_laststatval;
		2398	data->statcache = PL_statcache;
		2399	}
		2400
		2401	/* now build the result vector out of all the parameters and the data_sv */
		2402	{
		2403	int i;
		2404
		2405	for (i = 0; i < items; ++i)
		2406	av_push (state, SvREFCNT_inc_NN (ST (i)));
		2407	}
		2408
		2409	av_push (state, data_sv);
		2410
		2411	api_ready (aTHX_ coro);
		2412	SvREFCNT_dec (coro);
		2413	SvREFCNT_dec ((AV *)state);
		2414	}
		2415
		2416	static int
		2417	slf_check_aio_req (pTHX_ struct CoroSLF *frame)
		2418	{
		2419	AV *state = (AV *)frame->data;
		2420
		2421	/* if we are about to throw, return early */
		2422	/* this does not cancel the aio request, but at least */
		2423	/* it quickly returns */
		2424	if (CORO_THROW)
		2425	return 0;
		2426
		2427	/* one element that is an RV? repeat! */
		2428	if (AvFILLp (state) == 0 && SvROK (AvARRAY (state)[0]))
		2429	return 1;
		2430
		2431	/* restore status */
		2432	{
		2433	SV *data_sv = av_pop (state);
		2434	struct io_state *data = (struct io_state *)SvPVX (data_sv);
		2435
		2436	errno = data->errorno;
		2437	PL_laststype = data->laststype;
		2438	PL_laststatval = data->laststatval;
		2439	PL_statcache = data->statcache;
		2440
		2441	SvREFCNT_dec (data_sv);
		2442	}
		2443
		2444	/* push result values */
		2445	{
		2446	dSP;
		2447	int i;
		2448
		2449	EXTEND (SP, AvFILLp (state) + 1);
		2450	for (i = 0; i <= AvFILLp (state); ++i)
		2451	PUSHs (sv_2mortal (SvREFCNT_inc_NN (AvARRAY (state)[i])));
		2452
		2453	PUTBACK;
		2454	}
		2455
		2456	return 0;
		2457	}
		2458
		2459	static void
		2460	slf_init_aio_req (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		2461	{
		2462	AV *state = (AV *)sv_2mortal ((SV *)newAV ());
		2463	SV *coro_hv = SvRV (coro_current);
		2464	struct coro *coro = SvSTATE_hv (coro_hv);
		2465
		2466	/* put our coroutine id on the state arg */
		2467	av_push (state, SvREFCNT_inc_NN (coro_hv));
		2468
		2469	/* first see whether we have a non-zero priority and set it as AIO prio */
		2470	if (coro->prio)
		2471	{
		2472	dSP;
		2473
		2474	static SV *prio_cv;
		2475	static SV *prio_sv;
		2476
		2477	if (expect_false (!prio_cv))
		2478	{
		2479	prio_cv = (SV *)get_cv ("IO::AIO::aioreq_pri", 0);
		2480	prio_sv = newSViv (0);
		2481	}
		2482
		2483	PUSHMARK (SP);
		2484	sv_setiv (prio_sv, coro->prio);
		2485	XPUSHs (prio_sv);
		2486
		2487	PUTBACK;
		2488	call_sv (prio_cv, G_VOID | G_DISCARD);
		2489	}
		2490
		2491	/* now call the original request */
		2492	{
		2493	dSP;
		2494	CV *req = (CV *)CORO_MAGIC_NN ((SV *)cv, CORO_MAGIC_type_aio)->mg_obj;
		2495	int i;
		2496
		2497	PUSHMARK (SP);
		2498
		2499	/* first push all args to the stack */
		2500	EXTEND (SP, items + 1);
		2501
		2502	for (i = 0; i < items; ++i)
		2503	PUSHs (arg [i]);
		2504
		2505	/* now push the callback closure */
		2506	PUSHs (sv_2mortal (gensub (aTHX_ coro_aio_callback, (void *)SvREFCNT_inc_NN ((SV *)state))));
		2507
		2508	/* now call the AIO function - we assume our request is uncancelable */
		2509	PUTBACK;
		2510	call_sv ((SV *)req, G_VOID | G_DISCARD);
		2511	}
		2512
		2513	/* now that the requets is going, we loop toll we have a result */
		2514	frame->data = (void *)state;
		2515	frame->prepare = prepare_schedule;
		2516	frame->check = slf_check_aio_req;
		2517	}
		2518
		2519	static void
		2520	coro_aio_req_xs (pTHX_ CV *cv)
		2521	{
		2522	dXSARGS;
		2523
		2524	CORO_EXECUTE_SLF_XS (slf_init_aio_req);
		2525
		2526	XSRETURN_EMPTY;
2112	}	2527	}
2113		2528
2114	/*****************************************************************************/	2529	/*****************************************************************************/
2115		2530
2116	MODULE = Coro::State PACKAGE = Coro::State PREFIX = api_	2531	MODULE = Coro::State PACKAGE = Coro::State PREFIX = api_
…		…
2214	}	2629	}
2215	OUTPUT:	2630	OUTPUT:
2216	RETVAL	2631	RETVAL
2217		2632
2218	void	2633	void
2219	_set_stacklevel (...)
2220	CODE:
2221	api_execute_slf (aTHX_ cv, slf_init_set_stacklevel, &ST (0), items);
2222
2223	void
2224	transfer (...)	2634	transfer (...)
2225	PROTOTYPE: $$	2635	PROTOTYPE: $$
2226	CODE:	2636	CODE:
2227	api_execute_slf (aTHX_ cv, slf_init_transfer, &ST (0), items);	2637	CORO_EXECUTE_SLF_XS (slf_init_transfer);
2228		2638
2229	bool	2639	bool
2230	_destroy (SV *coro_sv)	2640	_destroy (SV *coro_sv)
2231	CODE:	2641	CODE:
2232	RETVAL = coro_state_destroy (aTHX_ SvSTATE (coro_sv));	2642	RETVAL = coro_state_destroy (aTHX_ SvSTATE (coro_sv));
…		…
2350		2760
2351	void	2761	void
2352	throw (Coro::State self, SV *throw = &PL_sv_undef)	2762	throw (Coro::State self, SV *throw = &PL_sv_undef)
2353	PROTOTYPE: $;$	2763	PROTOTYPE: $;$
2354	CODE:	2764	CODE:
		2765	{
		2766	struct coro *current = SvSTATE_current;
		2767	SV **throwp = self == current ? &CORO_THROW : &self->except;
2355	SvREFCNT_dec (self->throw);	2768	SvREFCNT_dec (*throwp);
2356	self->throw = SvOK (throw) ? newSVsv (throw) : 0;	2769	*throwp = SvOK (throw) ? newSVsv (throw) : 0;
		2770	}
2357		2771
2358	void	2772	void
2359	api_trace (SV *coro, int flags = CC_TRACE | CC_TRACE_SUB)	2773	api_trace (SV *coro, int flags = CC_TRACE | CC_TRACE_SUB)
2360	PROTOTYPE: $;$	2774	PROTOTYPE: $;$
2361	C_ARGS: aTHX_ coro, flags	2775	C_ARGS: aTHX_ coro, flags
…		…
2410	SV **dst = ix ? (SV **)&self->slot->defav : (SV **)&self->slot->defsv;	2824	SV **dst = ix ? (SV **)&self->slot->defav : (SV **)&self->slot->defsv;
2411		2825
2412	SV *tmp = *src; *src = *dst; *dst = tmp;	2826	SV *tmp = *src; *src = *dst; *dst = tmp;
2413	}	2827	}
2414		2828
		2829
2415	MODULE = Coro::State PACKAGE = Coro	2830	MODULE = Coro::State PACKAGE = Coro
2416		2831
2417	BOOT:	2832	BOOT:
2418	{	2833	{
2419	int i;	2834	int i;
…		…
2446	coroapi.ready = api_ready;	2861	coroapi.ready = api_ready;
2447	coroapi.is_ready = api_is_ready;	2862	coroapi.is_ready = api_is_ready;
2448	coroapi.nready = coro_nready;	2863	coroapi.nready = coro_nready;
2449	coroapi.current = coro_current;	2864	coroapi.current = coro_current;
2450		2865
2451	GCoroAPI = &coroapi;	2866	/*GCoroAPI = &coroapi;*/
2452	sv_setiv (sv, (IV)&coroapi);	2867	sv_setiv (sv, (IV)&coroapi);
2453	SvREADONLY_on (sv);	2868	SvREADONLY_on (sv);
2454	}	2869	}
2455	}	2870	}
2456		2871
2457	void	2872	void
2458	schedule (...)	2873	schedule (...)
2459	CODE:	2874	CODE:
2460	api_execute_slf (aTHX_ cv, slf_init_schedule, &ST (0), 0);	2875	CORO_EXECUTE_SLF_XS (slf_init_schedule);
2461		2876
2462	void	2877	void
2463	cede (...)	2878	cede (...)
2464	CODE:	2879	CODE:
2465	api_execute_slf (aTHX_ cv, slf_init_cede, &ST (0), 0);	2880	CORO_EXECUTE_SLF_XS (slf_init_cede);
2466		2881
2467	void	2882	void
2468	cede_notself (...)	2883	cede_notself (...)
2469	CODE:	2884	CODE:
2470	api_execute_slf (aTHX_ cv, slf_init_cede_notself, &ST (0), 0);	2885	CORO_EXECUTE_SLF_XS (slf_init_cede_notself);
2471		2886
2472	void	2887	void
2473	_set_current (SV *current)	2888	_set_current (SV *current)
2474	PROTOTYPE: $	2889	PROTOTYPE: $
2475	CODE:	2890	CODE:
…		…
2563		2978
2564	void	2979	void
2565	_pool_2 (SV *cb)	2980	_pool_2 (SV *cb)
2566	CODE:	2981	CODE:
2567	{	2982	{
2568	struct coro *coro = SvSTATE_current;	2983	HV *hv = (HV *)SvRV (coro_current);
		2984	struct coro *coro = SvSTATE_hv ((SV *)hv);
2569		2985
2570	sv_setsv (cb, &PL_sv_undef);	2986	sv_setsv (cb, &PL_sv_undef);
2571		2987
2572	SvREFCNT_dec ((SV *)PL_defoutgv); PL_defoutgv = (GV *)coro->saved_deffh;	2988	SvREFCNT_dec ((SV *)PL_defoutgv); PL_defoutgv = (GV *)coro->saved_deffh;
2573	coro->saved_deffh = 0;	2989	coro->saved_deffh = 0;
…		…
2580	SvREFCNT_dec (old);	2996	SvREFCNT_dec (old);
2581	croak ("\3async_pool terminate\2\n");	2997	croak ("\3async_pool terminate\2\n");
2582	}	2998	}
2583		2999
2584	av_clear (GvAV (PL_defgv));	3000	av_clear (GvAV (PL_defgv));
2585	hv_store ((HV *)SvRV (coro_current), "desc", sizeof ("desc") - 1,	3001	hv_store (hv, "desc", sizeof ("desc") - 1,
2586	newSVpvn ("[async_pool idle]", sizeof ("[async_pool idle]") - 1), 0);	3002	newSVpvn ("[async_pool idle]", sizeof ("[async_pool idle]") - 1), 0);
2587		3003
2588	coro->prio = 0;	3004	coro->prio = 0;
2589		3005
2590	if (coro->cctx && (coro->cctx->flags & CC_TRACE))	3006	if (coro->cctx && (coro->cctx->flags & CC_TRACE))
2591	api_trace (aTHX_ coro_current, 0);	3007	api_trace (aTHX_ coro_current, 0);
2592		3008
2593	av_push (av_async_pool, newSVsv (coro_current));	3009	av_push (av_async_pool, newSVsv (coro_current));
2594	}	3010	}
2595		3011
2596		3012	SV *
2597	MODULE = Coro::State PACKAGE = Coro::AIO	3013	rouse_cb ()
		3014	PROTOTYPE:
		3015	CODE:
		3016	RETVAL = coro_new_rouse_cb (aTHX);
		3017	OUTPUT:
		3018	RETVAL
2598		3019
2599	void	3020	void
2600	_get_state (SV *self)	3021	rouse_wait (...)
2601	PROTOTYPE: $	3022	PROTOTYPE: ;$
2602	PPCODE:	3023	PPCODE:
2603	{	3024	CORO_EXECUTE_SLF_XS (slf_init_rouse_wait);
2604	AV *defav = GvAV (PL_defgv);
2605	AV *av = newAV ();
2606	int i;
2607	SV *data_sv = newSV (sizeof (struct io_state));
2608	struct io_state *data = (struct io_state *)SvPVX (data_sv);
2609	SvCUR_set (data_sv, sizeof (struct io_state));
2610	SvPOK_only (data_sv);
2611
2612	data->errorno = errno;
2613	data->laststype = PL_laststype;
2614	data->laststatval = PL_laststatval;
2615	data->statcache = PL_statcache;
2616
2617	av_extend (av, AvFILLp (defav) + 1 + 1);
2618
2619	for (i = 0; i <= AvFILLp (defav); ++i)
2620	av_push (av, SvREFCNT_inc_NN (AvARRAY (defav)[i]));
2621
2622	av_push (av, data_sv);
2623
2624	XPUSHs (sv_2mortal (newRV_noinc ((SV *)av)));
2625
2626	api_ready (aTHX_ self);
2627	}
2628
2629	void
2630	_set_state (SV *state)
2631	PROTOTYPE: $
2632	PPCODE:
2633	{
2634	AV *av = (AV *)SvRV (state);
2635	struct io_state *data = (struct io_state *)SvPVX (AvARRAY (av)[AvFILLp (av)]);
2636	int i;
2637
2638	errno = data->errorno;
2639	PL_laststype = data->laststype;
2640	PL_laststatval = data->laststatval;
2641	PL_statcache = data->statcache;
2642
2643	EXTEND (SP, AvFILLp (av));
2644	for (i = 0; i < AvFILLp (av); ++i)
2645	PUSHs (sv_2mortal (SvREFCNT_inc_NN (AvARRAY (av)[i])));
2646	}
2647
2648
2649	MODULE = Coro::State PACKAGE = Coro::AnyEvent
2650
2651	BOOT:
2652	sv_activity = coro_get_sv (aTHX_ "Coro::AnyEvent::ACTIVITY", TRUE);
2653
2654	void
2655	_schedule (...)
2656	CODE:
2657	{
2658	static int incede;
2659
2660	api_cede_notself (aTHX);
2661
2662	++incede;
2663	while (coro_nready >= incede && api_cede (aTHX))
2664	;
2665
2666	sv_setsv (sv_activity, &PL_sv_undef);
2667	if (coro_nready >= incede)
2668	{
2669	PUSHMARK (SP);
2670	PUTBACK;
2671	call_pv ("Coro::AnyEvent::_activity", G_DISCARD | G_EVAL);
2672	SPAGAIN;
2673	}
2674
2675	--incede;
2676	}
2677		3025
2678		3026
2679	MODULE = Coro::State PACKAGE = PerlIO::cede	3027	MODULE = Coro::State PACKAGE = PerlIO::cede
2680		3028
2681	BOOT:	3029	BOOT:
2682	PerlIO_define_layer (aTHX_ &PerlIO_cede);	3030	PerlIO_define_layer (aTHX_ &PerlIO_cede);
2683		3031
		3032
2684	MODULE = Coro::State PACKAGE = Coro::Semaphore	3033	MODULE = Coro::State PACKAGE = Coro::Semaphore
2685		3034
2686	SV *	3035	SV *
2687	new (SV *klass, SV *count_ = 0)	3036	new (SV *klass, SV *count = 0)
2688	CODE:	3037	CODE:
2689	{	3038	RETVAL = sv_bless (
2690	/* a semaphore contains a counter IV in $sem->[0] and any waiters after that */	3039	coro_waitarray_new (aTHX_ count && SvOK (count) ? SvIV (count) : 1),
2691	AV *av = newAV ();	3040	GvSTASH (CvGV (cv))
2692	av_push (av, newSViv (count_ && SvOK (count_) ? SvIV (count_) : 1));	3041	);
2693	RETVAL = sv_bless (newRV_noinc ((SV *)av), GvSTASH (CvGV (cv)));	3042	OUTPUT:
2694	}	3043	RETVAL
		3044
		3045	# helper for Coro::Channel
		3046	SV *
		3047	_alloc (int count)
		3048	CODE:
		3049	RETVAL = coro_waitarray_new (aTHX_ count);
2695	OUTPUT:	3050	OUTPUT:
2696	RETVAL	3051	RETVAL
2697		3052
2698	SV *	3053	SV *
2699	count (SV *self)	3054	count (SV *self)
…		…
2708	adjust = 1	3063	adjust = 1
2709	CODE:	3064	CODE:
2710	coro_semaphore_adjust (aTHX_ (AV *)SvRV (self), ix ? adjust : 1);	3065	coro_semaphore_adjust (aTHX_ (AV *)SvRV (self), ix ? adjust : 1);
2711		3066
2712	void	3067	void
2713	down (SV *self)	3068	down (...)
2714	CODE:	3069	CODE:
2715	api_execute_slf (aTHX_ cv, slf_init_semaphore_down, &ST (0), 1);	3070	CORO_EXECUTE_SLF_XS (slf_init_semaphore_down);
		3071
		3072	void
		3073	wait (...)
		3074	CODE:
		3075	CORO_EXECUTE_SLF_XS (slf_init_semaphore_wait);
2716		3076
2717	void	3077	void
2718	try (SV *self)	3078	try (SV *self)
2719	PPCODE:	3079	PPCODE:
2720	{	3080	{
…		…
2732	XSRETURN_NO;	3092	XSRETURN_NO;
2733	}	3093	}
2734		3094
2735	void	3095	void
2736	waiters (SV *self)	3096	waiters (SV *self)
2737	CODE:	3097	PPCODE:
2738	{	3098	{
2739	AV *av = (AV *)SvRV (self);	3099	AV *av = (AV *)SvRV (self);
		3100	int wcount = AvFILLp (av) + 1 - 1;
2740		3101
2741	if (GIMME_V == G_SCALAR)	3102	if (GIMME_V == G_SCALAR)
2742	XPUSHs (sv_2mortal (newSVsv (AvARRAY (av)[0])));	3103	XPUSHs (sv_2mortal (newSViv (wcount)));
2743	else	3104	else
2744	{	3105	{
2745	int i;	3106	int i;
2746	EXTEND (SP, AvFILLp (av) + 1 - 1);	3107	EXTEND (SP, wcount);
2747	for (i = 1; i <= AvFILLp (av); ++i)	3108	for (i = 1; i <= wcount; ++i)
2748	PUSHs (newSVsv (AvARRAY (av)[i]));	3109	PUSHs (sv_2mortal (newRV_inc (AvARRAY (av)[i])));
2749	}	3110	}
2750	}	3111	}
2751		3112
		3113	MODULE = Coro::State PACKAGE = Coro::Signal
		3114
		3115	SV *
		3116	new (SV *klass)
		3117	CODE:
		3118	RETVAL = sv_bless (
		3119	coro_waitarray_new (aTHX_ 0),
		3120	GvSTASH (CvGV (cv))
		3121	);
		3122	OUTPUT:
		3123	RETVAL
		3124
		3125	void
		3126	wait (...)
		3127	CODE:
		3128	CORO_EXECUTE_SLF_XS (slf_init_signal_wait);
		3129
		3130	void
		3131	broadcast (SV *self)
		3132	CODE:
		3133	{
		3134	AV *av = (AV *)SvRV (self);
		3135	coro_signal_wake (aTHX_ av, AvFILLp (av));
		3136	}
		3137
		3138	void
		3139	send (SV *self)
		3140	CODE:
		3141	{
		3142	AV *av = (AV *)SvRV (self);
		3143
		3144	if (AvFILLp (av))
		3145	coro_signal_wake (aTHX_ av, 1);
		3146	else
		3147	SvIVX (AvARRAY (av)[0]) = 1; /* remember the signal */
		3148	}
		3149
		3150	IV
		3151	awaited (SV *self)
		3152	CODE:
		3153	RETVAL = AvFILLp ((AV *)SvRV (self)) + 1 - 1;
		3154	OUTPUT:
		3155	RETVAL
		3156
		3157
		3158	MODULE = Coro::State PACKAGE = Coro::AnyEvent
		3159
		3160	BOOT:
		3161	sv_activity = coro_get_sv (aTHX_ "Coro::AnyEvent::ACTIVITY", TRUE);
		3162
		3163	void
		3164	_schedule (...)
		3165	CODE:
		3166	{
		3167	static int incede;
		3168
		3169	api_cede_notself (aTHX);
		3170
		3171	++incede;
		3172	while (coro_nready >= incede && api_cede (aTHX))
		3173	;
		3174
		3175	sv_setsv (sv_activity, &PL_sv_undef);
		3176	if (coro_nready >= incede)
		3177	{
		3178	PUSHMARK (SP);
		3179	PUTBACK;
		3180	call_pv ("Coro::AnyEvent::_activity", G_KEEPERR | G_EVAL | G_VOID | G_DISCARD);
		3181	}
		3182
		3183	--incede;
		3184	}
		3185
		3186
		3187	MODULE = Coro::State PACKAGE = Coro::AIO
		3188
		3189	void
		3190	_register (char *target, char *proto, SV *req)
		3191	CODE:
		3192	{
		3193	HV *st;
		3194	GV *gvp;
		3195	CV *req_cv = sv_2cv (req, &st, &gvp, 0);
		3196	/* newXSproto doesn't return the CV on 5.8 */
		3197	CV *slf_cv = newXS (target, coro_aio_req_xs, __FILE__);
		3198	sv_setpv ((SV *)slf_cv, proto);
		3199	sv_magicext ((SV *)slf_cv, (SV *)req_cv, CORO_MAGIC_type_aio, 0, 0, 0);
		3200	}
		3201

Diff Legend

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