ViewVC Help

View File | Revision Log | Show Annotations | Download File

/cvs/Coro/Coro/State.xs

Comparing Coro/Coro/State.xs (file contents):
Revision 1.285 by root, Mon Nov 17 04:17:20 2008 UTC vs.
Revision 1.300 by root, Wed Nov 19 02:07:48 2008 UTC

…		…
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 */
…		…
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		263
264	/* async_pool */	264	/* async_pool */
265	SV *saved_deffh;	265	SV *saved_deffh;
266		266
267	/* linked list */	267	/* linked list */
…		…
273		273
274	/* the following variables are effectively part of the perl context */	274	/* the following variables are effectively part of the perl context */
275	/* and get copied between struct coro and these variables */	275	/* and get copied between struct coro and these variables */
276	/* the mainr easonw e don't support windows process emulation */	276	/* the mainr easonw e don't support windows process emulation */
277	static struct CoroSLF slf_frame; /* the current slf frame */	277	static struct CoroSLF slf_frame; /* the current slf frame */
278	static SV *coro_throw;
279		278
280	/** Coro ********************************************************************/	279	/** Coro ********************************************************************/
281		280
282	#define PRIO_MAX 3	281	#define PRIO_MAX 3
283	#define PRIO_HIGH 1	282	#define PRIO_HIGH 1
…		…
384	SvREFCNT_dec (av); /* sv_magicext increased the refcount */	383	SvREFCNT_dec (av); /* sv_magicext increased the refcount */
385		384
386	return 0;	385	return 0;
387	}	386	}
388		387
389	#define CORO_MAGIC_type_cv PERL_MAGIC_ext	388	#define CORO_MAGIC_type_cv 26
390	#define CORO_MAGIC_type_state PERL_MAGIC_ext	389	#define CORO_MAGIC_type_state PERL_MAGIC_ext
391		390
392	static MGVTBL coro_cv_vtbl = {	391	static MGVTBL coro_cv_vtbl = {
393	0, 0, 0, 0,	392	0, 0, 0, 0,
394	coro_cv_free	393	coro_cv_free
395	};	394	};
396		395
		396	#define CORO_MAGIC_NN(sv, type) \
		397	(expect_true (SvMAGIC (sv)->mg_type == type) \
		398	? SvMAGIC (sv) \
		399	: mg_find (sv, type))
		400
397	#define CORO_MAGIC(sv, type) \	401	#define CORO_MAGIC(sv, type) \
398	expect_true (SvMAGIC (sv)) \	402	(expect_true (SvMAGIC (sv)) \
399	? expect_true (SvMAGIC (sv)->mg_type == type) \	403	? CORO_MAGIC_NN (sv, type) \
400	? SvMAGIC (sv) \
401	: mg_find (sv, type) \
402	: 0	404	: 0)
403		405
404	#define CORO_MAGIC_cv(cv) CORO_MAGIC (((SV *)(cv)), CORO_MAGIC_type_cv)	406	#define CORO_MAGIC_cv(cv) CORO_MAGIC (((SV *)(cv)), CORO_MAGIC_type_cv)
405	#define CORO_MAGIC_state(sv) CORO_MAGIC (((SV *)(sv)), CORO_MAGIC_type_state)	407	#define CORO_MAGIC_state(sv) CORO_MAGIC_NN (((SV *)(sv)), CORO_MAGIC_type_state)
406		408
407	INLINE struct coro *	409	INLINE struct coro *
408	SvSTATE_ (pTHX_ SV *coro)	410	SvSTATE_ (pTHX_ SV *coro)
409	{	411	{
410	HV *stash;	412	HV *stash;
…		…
428	return (struct coro *)mg->mg_ptr;	430	return (struct coro *)mg->mg_ptr;
429	}	431	}
430		432
431	#define SvSTATE(sv) SvSTATE_ (aTHX_ (sv))	433	#define SvSTATE(sv) SvSTATE_ (aTHX_ (sv))
432		434
433	/* fastert than SvSTATE, but expects a coroutine hv */	435	/* faster than SvSTATE, but expects a coroutine hv */
434	INLINE struct coro *	436	#define SvSTATE_hv(hv) ((struct coro *)CORO_MAGIC_NN ((SV *)hv, CORO_MAGIC_type_state)->mg_ptr)
435	SvSTATE_hv (SV *sv)
436	{
437	MAGIC *mg = expect_true (SvMAGIC (sv)->mg_type == CORO_MAGIC_type_state)
438	? SvMAGIC (sv)
439	: mg_find (sv, CORO_MAGIC_type_state);
440
441	return (struct coro *)mg->mg_ptr;
442	}
443
444	#define SvSTATE_current SvSTATE_hv (SvRV (coro_current))	437	#define SvSTATE_current SvSTATE_hv (SvRV (coro_current))
445		438
446	/* the next two functions merely cache the padlists */	439	/* the next two functions merely cache the padlists */
447	static void	440	static void
448	get_padlist (pTHX_ CV *cv)	441	get_padlist (pTHX_ CV *cv)
…		…
518		511
519	PUTBACK;	512	PUTBACK;
520	}	513	}
521		514
522	slf_frame = c->slf_frame;	515	slf_frame = c->slf_frame;
523	coro_throw = c->throw;	516	CORO_THROW = c->except;
524	}	517	}
525		518
526	static void	519	static void
527	save_perl (pTHX_ Coro__State c)	520	save_perl (pTHX_ Coro__State c)
528	{	521	{
529	c->throw = coro_throw;	522	c->except = CORO_THROW;
530	c->slf_frame = slf_frame;	523	c->slf_frame = slf_frame;
531		524
532	{	525	{
533	dSP;	526	dSP;
534	I32 cxix = cxstack_ix;	527	I32 cxix = cxstack_ix;
…		…
827	slf_check_nop (pTHX_ struct CoroSLF *frame)	820	slf_check_nop (pTHX_ struct CoroSLF *frame)
828	{	821	{
829	return 0;	822	return 0;
830	}	823	}
831		824
		825	static UNOP coro_setup_op;
		826
832	static void NOINLINE /* noinline to keep it out of the transfer fast path */	827	static void NOINLINE /* noinline to keep it out of the transfer fast path */
833	coro_setup (pTHX_ struct coro *coro)	828	coro_setup (pTHX_ struct coro *coro)
834	{	829	{
835	/*	830	/*
836	* emulate part of the perl startup here.	831	* emulate part of the perl startup here.
…		…
864	{	859	{
865	dSP;	860	dSP;
866	UNOP myop;	861	UNOP myop;
867		862
868	Zero (&myop, 1, UNOP);	863	Zero (&myop, 1, UNOP);
869	myop.op_next = Nullop;	864	myop.op_next = Nullop;
870	myop.op_flags = OPf_WANT_VOID;	865	myop.op_flags = OPf_WANT_VOID;
871		866
872	PUSHMARK (SP);	867	PUSHMARK (SP);
873	XPUSHs (sv_2mortal (av_shift (GvAV (PL_defgv))));	868	XPUSHs (sv_2mortal (av_shift (GvAV (PL_defgv))));
874	PUTBACK;	869	PUTBACK;
…		…
881	* likely was suspended in pp_slf, so we have to emulate entering pp_slf here.	876	* likely was suspended in pp_slf, so we have to emulate entering pp_slf here.
882	*/	877	*/
883	slf_frame.prepare = prepare_nop; /* provide a nop function for an eventual pp_slf */	878	slf_frame.prepare = prepare_nop; /* provide a nop function for an eventual pp_slf */
884	slf_frame.check = slf_check_nop; /* signal pp_slf to not repeat */	879	slf_frame.check = slf_check_nop; /* signal pp_slf to not repeat */
885		880
886	coro_throw = coro->throw;	881	/* and we have to provide the pp_slf op in any case, so pp_slf can skip it */
		882	coro_setup_op.op_next = PL_op;
		883	coro_setup_op.op_type = OP_CUSTOM;
		884	coro_setup_op.op_ppaddr = pp_slf;
		885	/* no flags required, as an init function won't be called */
		886
		887	PL_op = (OP *)&coro_setup_op;
		888
		889	/* copy throw, in case it was set before coro_setup */
		890	CORO_THROW = coro->except;
887	}	891	}
888		892
889	static void	893	static void
890	coro_destruct (pTHX_ struct coro *coro)	894	coro_destruct (pTHX_ struct coro *coro)
891	{	895	{
…		…
915		919
916	SvREFCNT_dec (PL_diehook);	920	SvREFCNT_dec (PL_diehook);
917	SvREFCNT_dec (PL_warnhook);	921	SvREFCNT_dec (PL_warnhook);
918		922
919	SvREFCNT_dec (coro->saved_deffh);	923	SvREFCNT_dec (coro->saved_deffh);
920	SvREFCNT_dec (coro_throw);	924	SvREFCNT_dec (CORO_THROW);
921		925
922	coro_destruct_stacks (aTHX);	926	coro_destruct_stacks (aTHX);
923	}	927	}
924		928
925	INLINE void	929	INLINE void
…		…
1054		1058
1055	TAINT_NOT;	1059	TAINT_NOT;
1056	return 0;	1060	return 0;
1057	}	1061	}
1058		1062
		1063	static struct coro_cctx *cctx_ssl_cctx;
		1064	static struct CoroSLF cctx_ssl_frame;
		1065
1059	static void	1066	static void
1060	prepare_set_stacklevel (struct coro_transfer_args *ta, struct coro_cctx *cctx)	1067	slf_prepare_set_stacklevel (pTHX_ struct coro_transfer_args *ta)
1061	{	1068	{
1062	ta->prev = (struct coro *)cctx;	1069	ta->prev = (struct coro *)cctx_ssl_cctx;
1063	ta->next = 0;	1070	ta->next = 0;
1064	}	1071	}
1065		1072
1066	/* inject a fake call to Coro::State::_cctx_init into the execution */	1073	static int
1067	/* _cctx_init should be careful, as it could be called at almost any time */	1074	slf_check_set_stacklevel (pTHX_ struct CoroSLF *frame)
1068	/* during execution of a perl program */	1075	{
1069	/* also initialises PL_top_env */	1076	*frame = cctx_ssl_frame;
		1077
		1078	return frame->check (aTHX_ frame); /* execute the restored frame - there must be one */
		1079	}
		1080
		1081	/* initialises PL_top_env and injects a pseudo-slf-call to set the stacklevel */
1070	static void NOINLINE	1082	static void NOINLINE
1071	cctx_prepare (pTHX_ coro_cctx *cctx)	1083	cctx_prepare (pTHX_ coro_cctx *cctx)
1072	{	1084	{
1073	dSP;
1074	UNOP myop;
1075
1076	PL_top_env = &PL_start_env;	1085	PL_top_env = &PL_start_env;
1077		1086
1078	if (cctx->flags & CC_TRACE)	1087	if (cctx->flags & CC_TRACE)
1079	PL_runops = runops_trace;	1088	PL_runops = runops_trace;
1080		1089
1081	Zero (&myop, 1, UNOP);	1090	/* we already must be executing an SLF op, there is no other valid way
1082	myop.op_next = PL_op;	1091	* that can lead to creation of a new cctx */
1083	myop.op_flags = OPf_WANT_VOID | OPf_STACKED;	1092	assert (("FATAL: can't prepare slf-less cctx in Coro module (please report)",
		1093	slf_frame.prepare && PL_op->op_ppaddr == pp_slf));
1084		1094
1085	PUSHMARK (SP);	1095	/* we must emulate leaving pp_slf, which is done inside slf_check_set_stacklevel */
1086	EXTEND (SP, 2);	1096	cctx_ssl_cctx = cctx;
1087	PUSHs (sv_2mortal (newSViv ((IV)cctx)));	1097	cctx_ssl_frame = slf_frame;
1088	PUSHs ((SV *)get_cv ("Coro::State::_cctx_init", FALSE));	1098
1089	PUTBACK;	1099	slf_frame.prepare = slf_prepare_set_stacklevel;
1090	PL_op = (OP *)&myop;	1100	slf_frame.check = slf_check_set_stacklevel;
1091	PL_op = PL_ppaddr[OP_ENTERSUB](aTHX);
1092	SPAGAIN;
1093	}	1101	}
1094		1102
1095	/* the tail of transfer: execute stuff we can only do after a transfer */	1103	/* the tail of transfer: execute stuff we can only do after a transfer */
1096	INLINE void	1104	INLINE void
1097	transfer_tail (pTHX)	1105	transfer_tail (pTHX)
…		…
1313	dSTACKLEVEL;	1321	dSTACKLEVEL;
1314		1322
1315	/* sometimes transfer is only called to set idle_sp */	1323	/* sometimes transfer is only called to set idle_sp */
1316	if (expect_false (!next))	1324	if (expect_false (!next))
1317	{	1325	{
1318	((coro_cctx *)prev)->idle_sp = (void *)stacklevel;	1326	((coro_cctx *)prev)->idle_sp = STACKLEVEL;
1319	assert (((coro_cctx *)prev)->idle_te = PL_top_env); /* just for the side-effect when asserts are enabled */	1327	assert (((coro_cctx *)prev)->idle_te = PL_top_env); /* just for the side-effect when asserts are enabled */
1320	}	1328	}
1321	else if (expect_true (prev != next))	1329	else if (expect_true (prev != next))
1322	{	1330	{
1323	coro_cctx *prev__cctx;	1331	coro_cctx *prev__cctx;
…		…
1348		1356
1349	prev__cctx = prev->cctx;	1357	prev__cctx = prev->cctx;
1350		1358
1351	/* possibly untie and reuse the cctx */	1359	/* possibly untie and reuse the cctx */
1352	if (expect_true (	1360	if (expect_true (
1353	prev__cctx->idle_sp == (void *)stacklevel	1361	prev__cctx->idle_sp == STACKLEVEL
1354	&& !(prev__cctx->flags & CC_TRACE)	1362	&& !(prev__cctx->flags & CC_TRACE)
1355	&& !force_cctx	1363	&& !force_cctx
1356	))	1364	))
1357	{	1365	{
1358	/* I assume that stacklevel is a stronger indicator than PL_top_env changes */	1366	/* I assume that stacklevel is a stronger indicator than PL_top_env changes */
…		…
1541	ENTER;	1549	ENTER;
1542	SAVETMPS;	1550	SAVETMPS;
1543		1551
1544	PUSHMARK (SP);	1552	PUSHMARK (SP);
1545	PUTBACK;	1553	PUTBACK;
1546	call_sv (sv_hook, G_DISCARD);	1554	call_sv (sv_hook, G_VOID | G_DISCARD);
1547	SPAGAIN;
1548		1555
1549	FREETMPS;	1556	FREETMPS;
1550	LEAVE;	1557	LEAVE;
1551	}	1558	}
1552		1559
…		…
1579	ENTER;	1586	ENTER;
1580	SAVETMPS;	1587	SAVETMPS;
1581		1588
1582	PUSHMARK (SP);	1589	PUSHMARK (SP);
1583	PUTBACK;	1590	PUTBACK;
1584	call_sv (get_sv ("Coro::idle", FALSE), G_DISCARD);	1591	call_sv (get_sv ("Coro::idle", FALSE), G_VOID | G_DISCARD);
1585	SPAGAIN;
1586		1592
1587	FREETMPS;	1593	FREETMPS;
1588	LEAVE;	1594	LEAVE;
1589	continue;	1595	continue;
1590	}	1596	}
…		…
1697	if (coro->flags & CF_RUNNING)	1703	if (coro->flags & CF_RUNNING)
1698	PL_runops = RUNOPS_DEFAULT;	1704	PL_runops = RUNOPS_DEFAULT;
1699	else	1705	else
1700	coro->slot->runops = RUNOPS_DEFAULT;	1706	coro->slot->runops = RUNOPS_DEFAULT;
1701	}	1707	}
		1708	}
		1709
		1710	/*****************************************************************************/
		1711	/* schedule-like-function opcode (SLF) */
		1712
		1713	static UNOP slf_restore; /* restore stack as entersub did, for first-re-run */
		1714	static const CV *slf_cv;
		1715	static SV **slf_argv;
		1716	static int slf_argc, slf_arga; /* count, allocated */
		1717	static I32 slf_ax; /* top of stack, for restore */
		1718
		1719	/* this restores the stack in the case we patched the entersub, to */
		1720	/* recreate the stack frame as perl will on following calls */
		1721	/* since entersub cleared the stack */
		1722	static OP *
		1723	pp_restore (pTHX)
		1724	{
		1725	int i;
		1726	SV **SP = PL_stack_base + slf_ax;
		1727
		1728	PUSHMARK (SP);
		1729
		1730	EXTEND (SP, slf_argc + 1);
		1731
		1732	for (i = 0; i < slf_argc; ++i)
		1733	PUSHs (sv_2mortal (slf_argv [i]));
		1734
		1735	PUSHs ((SV *)CvGV (slf_cv));
		1736
		1737	RETURNOP (slf_restore.op_first);
		1738	}
		1739
		1740	static void
		1741	slf_prepare_transfer (pTHX_ struct coro_transfer_args *ta)
		1742	{
		1743	SV **arg = (SV **)slf_frame.data;
		1744
		1745	prepare_transfer (aTHX_ ta, arg [0], arg [1]);
		1746	}
		1747
		1748	static void
		1749	slf_init_transfer (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		1750	{
		1751	if (items != 2)
		1752	croak ("Coro::State::transfer (prev, next) expects two arguments, not %d,", items);
		1753
		1754	frame->prepare = slf_prepare_transfer;
		1755	frame->check = slf_check_nop;
		1756	frame->data = (void *)arg; /* let's hope it will stay valid */
		1757	}
		1758
		1759	static void
		1760	slf_init_schedule (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		1761	{
		1762	frame->prepare = prepare_schedule;
		1763	frame->check = slf_check_nop;
		1764	}
		1765
		1766	static void
		1767	slf_init_cede (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		1768	{
		1769	frame->prepare = prepare_cede;
		1770	frame->check = slf_check_nop;
		1771	}
		1772
		1773	static void
		1774	slf_init_cede_notself (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		1775	{
		1776	frame->prepare = prepare_cede_notself;
		1777	frame->check = slf_check_nop;
		1778	}
		1779
		1780	/*
		1781	* these not obviously related functions are all rolled into one
		1782	* function to increase chances that they all will call transfer with the same
		1783	* stack offset
		1784	* SLF stands for "schedule-like-function".
		1785	*/
		1786	static OP *
		1787	pp_slf (pTHX)
		1788	{
		1789	I32 checkmark; /* mark SP to see how many elements check has pushed */
		1790
		1791	/* set up the slf frame, unless it has already been set-up */
		1792	/* the latter happens when a new coro has been started */
		1793	/* or when a new cctx was attached to an existing coroutine */
		1794	if (expect_true (!slf_frame.prepare))
		1795	{
		1796	/* first iteration */
		1797	dSP;
		1798	SV **arg = PL_stack_base + TOPMARK + 1;
		1799	int items = SP - arg; /* args without function object */
		1800	SV *gv = *sp;
		1801
		1802	/* do a quick consistency check on the "function" object, and if it isn't */
		1803	/* for us, divert to the real entersub */
		1804	if (SvTYPE (gv) != SVt_PVGV
		1805	|| !GvCV (gv)
		1806	|| !(CvFLAGS (GvCV (gv)) & CVf_SLF))
		1807	return PL_ppaddr[OP_ENTERSUB](aTHX);
		1808
		1809	if (!(PL_op->op_flags & OPf_STACKED))
		1810	{
		1811	/* ampersand-form of call, use @_ instead of stack */
		1812	AV *av = GvAV (PL_defgv);
		1813	arg = AvARRAY (av);
		1814	items = AvFILLp (av) + 1;
		1815	}
		1816
		1817	/* now call the init function, which needs to set up slf_frame */
		1818	((coro_slf_cb)CvXSUBANY (GvCV (gv)).any_ptr)
		1819	(aTHX_ &slf_frame, GvCV (gv), arg, items);
		1820
		1821	/* pop args */
		1822	SP = PL_stack_base + POPMARK;
		1823
		1824	PUTBACK;
		1825	}
		1826
		1827	/* now that we have a slf_frame, interpret it! */
		1828	/* we use a callback system not to make the code needlessly */
		1829	/* complicated, but so we can run multiple perl coros from one cctx */
		1830
		1831	do
		1832	{
		1833	struct coro_transfer_args ta;
		1834
		1835	slf_frame.prepare (aTHX_ &ta);
		1836	TRANSFER (ta, 0);
		1837
		1838	checkmark = PL_stack_sp - PL_stack_base;
		1839	}
		1840	while (slf_frame.check (aTHX_ &slf_frame));
		1841
		1842	slf_frame.prepare = 0; /* invalidate the frame, we are done processing it */
		1843
		1844	/* exception handling */
		1845	if (expect_false (CORO_THROW))
		1846	{
		1847	SV *exception = sv_2mortal (CORO_THROW);
		1848
		1849	CORO_THROW = 0;
		1850	sv_setsv (ERRSV, exception);
		1851	croak (0);
		1852	}
		1853
		1854	/* return value handling - mostly like entersub */
		1855	/* make sure we put something on the stack in scalar context */
		1856	if (GIMME_V == G_SCALAR)
		1857	{
		1858	dSP;
		1859	SV **bot = PL_stack_base + checkmark;
		1860
		1861	if (sp == bot) /* too few, push undef */
		1862	bot [1] = &PL_sv_undef;
		1863	else if (sp != bot + 1) /* too many, take last one */
		1864	bot [1] = *sp;
		1865
		1866	SP = bot + 1;
		1867
		1868	PUTBACK;
		1869	}
		1870
		1871	return NORMAL;
		1872	}
		1873
		1874	static void
		1875	api_execute_slf (pTHX_ CV *cv, coro_slf_cb init_cb, I32 ax)
		1876	{
		1877	int i;
		1878	SV **arg = PL_stack_base + ax;
		1879	int items = PL_stack_sp - arg + 1;
		1880
		1881	assert (("FATAL: SLF call with illegal CV value", !CvANON (cv)));
		1882
		1883	if (PL_op->op_ppaddr != PL_ppaddr [OP_ENTERSUB]
		1884	&& PL_op->op_ppaddr != pp_slf)
		1885	croak ("FATAL: Coro SLF calls can only be made normally, not via goto or any other means, caught");
		1886
		1887	CvFLAGS (cv) |= CVf_SLF;
		1888	CvXSUBANY (cv).any_ptr = (void *)init_cb;
		1889	slf_cv = cv;
		1890
		1891	/* we patch the op, and then re-run the whole call */
		1892	/* we have to put the same argument on the stack for this to work */
		1893	/* and this will be done by pp_restore */
		1894	slf_restore.op_next = (OP *)&slf_restore;
		1895	slf_restore.op_type = OP_CUSTOM;
		1896	slf_restore.op_ppaddr = pp_restore;
		1897	slf_restore.op_first = PL_op;
		1898
		1899	slf_ax = ax - 1; /* undo the ax++ inside dAXMARK */
		1900
		1901	if (PL_op->op_flags & OPf_STACKED)
		1902	{
		1903	if (items > slf_arga)
		1904	{
		1905	slf_arga = items;
		1906	free (slf_argv);
		1907	slf_argv = malloc (slf_arga * sizeof (SV *));
		1908	}
		1909
		1910	slf_argc = items;
		1911
		1912	for (i = 0; i < items; ++i)
		1913	slf_argv [i] = SvREFCNT_inc (arg [i]);
		1914	}
		1915	else
		1916	slf_argc = 0;
		1917
		1918	PL_op->op_ppaddr = pp_slf;
		1919	PL_op->op_type = OP_CUSTOM; /* maybe we should leave it at entersub? */
		1920
		1921	PL_op = (OP *)&slf_restore;
1702	}	1922	}
1703		1923
1704	/*****************************************************************************/	1924	/*****************************************************************************/
1705	/* PerlIO::cede */	1925	/* PerlIO::cede */
1706		1926
…		…
1775	PerlIOBuf_get_cnt,	1995	PerlIOBuf_get_cnt,
1776	PerlIOBuf_set_ptrcnt,	1996	PerlIOBuf_set_ptrcnt,
1777	};	1997	};
1778		1998
1779	/*****************************************************************************/	1999	/*****************************************************************************/
		2000	/* Coro::Semaphore & Coro::Signal */
1780		2001
1781	static UNOP slf_restore; /* restore stack as entersub did, for first-re-run */
1782	static const CV *slf_cv;
1783	static SV **slf_argv;
1784	static int slf_argc, slf_arga; /* count, allocated */
1785	static I32 slf_ax; /* top of stack, for restore */
1786
1787	/* this restores the stack in the case we patched the entersub, to */
1788	/* recreate the stack frame as perl will on following calls */
1789	/* since entersub cleared the stack */
1790	static OP *	2002	static SV *
1791	pp_restore (pTHX)	2003	coro_waitarray_new (pTHX_ int count)
1792	{	2004	{
1793	int i;	2005	/* a semaphore contains a counter IV in $sem->[0] and any waiters after that */
1794	SV **SP = PL_stack_base + slf_ax;	2006	AV *av = newAV ();
		2007	SV **ary;
1795		2008
1796	PUSHMARK (SP);	2009	/* unfortunately, building manually saves memory */
		2010	Newx (ary, 2, SV *);
		2011	AvALLOC (av) = ary;
		2012	AvARRAY (av) = ary;
		2013	AvMAX (av) = 1;
		2014	AvFILLp (av) = 0;
		2015	ary [0] = newSViv (count);
1797		2016
1798	EXTEND (SP, slf_argc + 1);	2017	return newRV_noinc ((SV *)av);
1799
1800	for (i = 0; i < slf_argc; ++i)
1801	PUSHs (sv_2mortal (slf_argv [i]));
1802
1803	PUSHs ((SV *)CvGV (slf_cv));
1804
1805	RETURNOP (slf_restore.op_first);
1806	}	2018	}
1807		2019
1808	static void	2020	/* semaphore */
1809	slf_prepare_set_stacklevel (pTHX_ struct coro_transfer_args *ta)
1810	{
1811	prepare_set_stacklevel (ta, (struct coro_cctx *)slf_frame.data);
1812	}
1813
1814	static void
1815	slf_init_set_stacklevel (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
1816	{
1817	assert (("FATAL: set_stacklevel needs the coro cctx as sole argument", items == 1));
1818
1819	frame->prepare = slf_prepare_set_stacklevel;
1820	frame->check = slf_check_nop;
1821	frame->data = (void *)SvIV (arg [0]);
1822	}
1823
1824	static void
1825	slf_prepare_transfer (pTHX_ struct coro_transfer_args *ta)
1826	{
1827	SV **arg = (SV **)slf_frame.data;
1828
1829	prepare_transfer (aTHX_ ta, arg [0], arg [1]);
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	slf_frame.prepare = 0; /* invalidate the frame, we are done processing it */
1928
1929	/* return value handling - mostly like entersub */
1930	{
1931	dSP;
1932	SV **bot = PL_stack_base + checkmark;
1933	int gimme = GIMME_V;
1934
1935	/* make sure we put something on the stack in scalar context */
1936	if (gimme == G_SCALAR)
1937	{
1938	if (sp == bot)
1939	XPUSHs (&PL_sv_undef);
1940
1941	SP = bot + 1;
1942	}
1943
1944	PUTBACK;
1945	}
1946
1947	/* exception handling */
1948	if (expect_false (coro_throw))
1949	{
1950	SV *exception = sv_2mortal (coro_throw);
1951
1952	coro_throw = 0;
1953	sv_setsv (ERRSV, exception);
1954	croak (0);
1955	}
1956
1957	return NORMAL;
1958	}
1959
1960	static void
1961	api_execute_slf (pTHX_ CV *cv, coro_slf_cb init_cb, I32 ax)
1962	{
1963	int i;
1964	SV **arg = PL_stack_base + ax;
1965	int items = PL_stack_sp - arg + 1;
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	CvFLAGS (cv) |= CVf_SLF;
1974	CvXSUBANY (cv).any_ptr = (void *)init_cb;
1975	slf_cv = cv;
1976
1977	/* we patch the op, and then re-run the whole call */
1978	/* we have to put the same argument on the stack for this to work */
1979	/* and this will be done by pp_restore */
1980	slf_restore.op_next = (OP *)&slf_restore;
1981	slf_restore.op_type = OP_CUSTOM;
1982	slf_restore.op_ppaddr = pp_restore;
1983	slf_restore.op_first = PL_op;
1984
1985	slf_ax = ax - 1; /* undo the ax++ inside dAXMARK */
1986
1987	if (items > slf_arga)
1988	{
1989	slf_arga = items;
1990	free (slf_argv);
1991	slf_argv = malloc (slf_arga * sizeof (SV *));
1992	}
1993
1994	slf_argc = items;
1995
1996	for (i = 0; i < items; ++i)
1997	slf_argv [i] = SvREFCNT_inc (arg [i]);
1998
1999	PL_op->op_ppaddr = pp_slf;
2000	PL_op->op_type = OP_CUSTOM; /* maybe we should leave it at entersub? */
2001
2002	PL_op = (OP *)&slf_restore;
2003	}
2004
2005	/*****************************************************************************/
2006		2021
2007	static void	2022	static void
2008	coro_semaphore_adjust (pTHX_ AV *av, IV adjust)	2023	coro_semaphore_adjust (pTHX_ AV *av, IV adjust)
2009	{	2024	{
2010	SV *count_sv = AvARRAY (av)[0];	2025	SV *count_sv = AvARRAY (av)[0];
…		…
2040	/* call $sem->adjust (0) to possibly wake up some other waiters */	2055	/* call $sem->adjust (0) to possibly wake up some other waiters */
2041	coro_semaphore_adjust (aTHX_ (AV *)coro->slf_frame.data, 0);	2056	coro_semaphore_adjust (aTHX_ (AV *)coro->slf_frame.data, 0);
2042	}	2057	}
2043		2058
2044	static int	2059	static int
2045	slf_check_semaphore_down (pTHX_ struct CoroSLF *frame)	2060	slf_check_semaphore_down_or_wait (pTHX_ struct CoroSLF *frame, int acquire)
2046	{	2061	{
2047	AV *av = (AV *)frame->data;	2062	AV *av = (AV *)frame->data;
2048	SV *count_sv = AvARRAY (av)[0];	2063	SV *count_sv = AvARRAY (av)[0];
2049		2064
		2065	/* if we are about to throw, don't actually acquire the lock, just throw */
		2066	if (CORO_THROW)
		2067	return 0;
2050	if (SvIVX (count_sv) > 0)	2068	else if (SvIVX (count_sv) > 0)
2051	{	2069	{
2052	SvSTATE_current->on_destroy = 0;	2070	SvSTATE_current->on_destroy = 0;
		2071
		2072	if (acquire)
2053	SvIVX (count_sv) = SvIVX (count_sv) - 1;	2073	SvIVX (count_sv) = SvIVX (count_sv) - 1;
		2074	else
		2075	coro_semaphore_adjust (aTHX_ av, 0);
		2076
2054	return 0;	2077	return 0;
2055	}	2078	}
2056	else	2079	else
2057	{	2080	{
2058	int i;	2081	int i;
…		…
2067	av_push (av, SvREFCNT_inc (SvRV (coro_current)));	2090	av_push (av, SvREFCNT_inc (SvRV (coro_current)));
2068	return 1;	2091	return 1;
2069	}	2092	}
2070	}	2093	}
2071		2094
2072	static void	2095	static int
		2096	slf_check_semaphore_down (pTHX_ struct CoroSLF *frame)
		2097	{
		2098	return slf_check_semaphore_down_or_wait (aTHX_ frame, 1);
		2099	}
		2100
		2101	static int
		2102	slf_check_semaphore_wait (pTHX_ struct CoroSLF *frame)
		2103	{
		2104	return slf_check_semaphore_down_or_wait (aTHX_ frame, 0);
		2105	}
		2106
		2107	static void
2073	slf_init_semaphore_down (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)	2108	slf_init_semaphore_down_or_wait (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
2074	{	2109	{
2075	AV *av = (AV *)SvRV (arg [0]);	2110	AV *av = (AV *)SvRV (arg [0]);
2076		2111
2077	if (SvIVX (AvARRAY (av)[0]) > 0)	2112	if (SvIVX (AvARRAY (av)[0]) > 0)
2078	{	2113	{
2079	frame->data = (void *)av;	2114	frame->data = (void *)av;
2080	frame->prepare = prepare_nop;	2115	frame->prepare = prepare_nop;
2081	SvSTATE_current->on_destroy = coro_semaphore_on_destroy;
2082	}	2116	}
2083	else	2117	else
2084	{	2118	{
2085	av_push (av, SvREFCNT_inc (SvRV (coro_current)));	2119	av_push (av, SvREFCNT_inc (SvRV (coro_current)));
2086		2120
2087	frame->data = (void *)sv_2mortal (SvREFCNT_inc ((SV *)av));	2121	frame->data = (void *)sv_2mortal (SvREFCNT_inc ((SV *)av));
2088	frame->prepare = prepare_schedule;	2122	frame->prepare = prepare_schedule;
2089		2123
2090	/* to avoid race conditions when a woken-up coro gets terminated */	2124	/* to avoid race conditions when a woken-up coro gets terminated */
2091	/* we arrange for a temporary on_destroy that calls adjust (0) */	2125	/* we arrange for a temporary on_destroy that calls adjust (0) */
2092	assert (!SvSTATE_current->on_destroy);//D
2093	SvSTATE_current->on_destroy = coro_semaphore_on_destroy;	2126	SvSTATE_current->on_destroy = coro_semaphore_on_destroy;
2094	}	2127	}
		2128	}
2095		2129
		2130	static void
		2131	slf_init_semaphore_down (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		2132	{
		2133	slf_init_semaphore_down_or_wait (aTHX_ frame, cv, arg, items);
2096	frame->check = slf_check_semaphore_down;	2134	frame->check = slf_check_semaphore_down;
		2135	}
2097		2136
		2137	static void
		2138	slf_init_semaphore_wait (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		2139	{
		2140	slf_init_semaphore_down_or_wait (aTHX_ frame, cv, arg, items);
		2141	frame->check = slf_check_semaphore_wait;
		2142	}
		2143
		2144	/* signal */
		2145
		2146	static void
		2147	coro_signal_wake (pTHX_ AV *av, int count)
		2148	{
		2149	SvIVX (AvARRAY (av)[0]) = 0;
		2150
		2151	/* now signal count waiters */
		2152	while (count > 0 && AvFILLp (av) > 0)
		2153	{
		2154	SV *cb;
		2155
		2156	/* swap first two elements so we can shift a waiter */
		2157	cb = AvARRAY (av)[0];
		2158	AvARRAY (av)[0] = AvARRAY (av)[1];
		2159	AvARRAY (av)[1] = cb;
		2160
		2161	cb = av_shift (av);
		2162
		2163	api_ready (cb);
		2164	sv_setiv (cb, 0); /* signal waiter */
		2165	SvREFCNT_dec (cb);
		2166
		2167	--count;
		2168	}
		2169	}
		2170
		2171	static int
		2172	slf_check_signal_wait (pTHX_ struct CoroSLF *frame)
		2173	{
		2174	/* if we are about to throw, also stop waiting */
		2175	return SvROK ((SV *)frame->data) && !CORO_THROW;
		2176	}
		2177
		2178	static void
		2179	slf_init_signal_wait (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		2180	{
		2181	AV *av = (AV *)SvRV (arg [0]);
		2182
		2183	if (SvIVX (AvARRAY (av)[0]))
		2184	{
		2185	SvIVX (AvARRAY (av)[0]) = 0;
		2186	frame->prepare = prepare_nop;
		2187	frame->check = slf_check_nop;
		2188	}
		2189	else
		2190	{
		2191	SV *waiter = newRV_inc (SvRV (coro_current)); /* owned by signal av */
		2192
		2193	av_push (av, waiter);
		2194
		2195	frame->data = (void *)sv_2mortal (SvREFCNT_inc_NN (waiter)); /* owned by process */
		2196	frame->prepare = prepare_schedule;
		2197	frame->check = slf_check_signal_wait;
		2198	}
2098	}	2199	}
2099		2200
2100	/*****************************************************************************/	2201	/*****************************************************************************/
		2202	/* gensub: simple closure generation utility */
2101		2203
2102	#define GENSUB_ARG CvXSUBANY (cv).any_ptr	2204	#define GENSUB_ARG CvXSUBANY (cv).any_ptr
2103		2205
2104	/* create a closure from XS, returns a code reference */	2206	/* create a closure from XS, returns a code reference */
2105	/* the arg can be accessed via GENSUB_ARG from the callback */	2207	/* the arg can be accessed via GENSUB_ARG from the callback */
2106	/* the callback must use dXSARGS/XSRETURN */	2208	/* the callback must use dXSARGS/XSRETURN */
2107	static SV *	2209	static SV *
2108	gensub (pTHX_ void (*xsub)(pTHX_ CV *), void *arg)	2210	gensub (pTHX_ void (*xsub)(pTHX_ CV *), void *arg)
2109	{	2211	{
2110	CV *cv = (CV *)NEWSV (0, 0);	2212	CV *cv = (CV *)newSV (0);
2111		2213
2112	sv_upgrade ((SV *)cv, SVt_PVCV);	2214	sv_upgrade ((SV *)cv, SVt_PVCV);
2113		2215
2114	CvANON_on (cv);	2216	CvANON_on (cv);
2115	CvISXSUB_on (cv);	2217	CvISXSUB_on (cv);
2116	CvXSUB (cv) = xsub;	2218	CvXSUB (cv) = xsub;
2117	GENSUB_ARG = arg;	2219	GENSUB_ARG = arg;
2118		2220
2119	return newRV_noinc ((SV *)cv);	2221	return newRV_noinc ((SV *)cv);
		2222	}
		2223
		2224	/*****************************************************************************/
		2225	/* Coro::AIO */
		2226
		2227	#define CORO_MAGIC_type_aio PERL_MAGIC_ext
		2228
		2229	/* helper storage struct */
		2230	struct io_state
		2231	{
		2232	int errorno;
		2233	I32 laststype; /* U16 in 5.10.0 */
		2234	int laststatval;
		2235	Stat_t statcache;
		2236	};
		2237
		2238	static void
		2239	coro_aio_callback (pTHX_ CV *cv)
		2240	{
		2241	dXSARGS;
		2242	AV *state = (AV *)GENSUB_ARG;
		2243	SV *coro = av_pop (state);
		2244	SV *data_sv = newSV (sizeof (struct io_state));
		2245
		2246	av_extend (state, items);
		2247
		2248	sv_upgrade (data_sv, SVt_PV);
		2249	SvCUR_set (data_sv, sizeof (struct io_state));
		2250	SvPOK_only (data_sv);
		2251
		2252	{
		2253	struct io_state *data = (struct io_state *)SvPVX (data_sv);
		2254
		2255	data->errorno = errno;
		2256	data->laststype = PL_laststype;
		2257	data->laststatval = PL_laststatval;
		2258	data->statcache = PL_statcache;
		2259	}
		2260
		2261	/* now build the result vector out of all the parameters and the data_sv */
		2262	{
		2263	int i;
		2264
		2265	for (i = 0; i < items; ++i)
		2266	av_push (state, SvREFCNT_inc_NN (ST (i)));
		2267	}
		2268
		2269	av_push (state, data_sv);
		2270
		2271	api_ready (aTHX_ coro);
		2272	SvREFCNT_dec (coro);
		2273	SvREFCNT_dec ((AV *)state);
		2274	}
		2275
		2276	static int
		2277	slf_check_aio_req (pTHX_ struct CoroSLF *frame)
		2278	{
		2279	AV *state = (AV *)frame->data;
		2280
		2281	/* if we are about to throw, return early */
		2282	/* this does not cancel the aio request, but at least */
		2283	/* it quickly returns */
		2284	if (CORO_THROW)
		2285	return 0;
		2286
		2287	/* one element that is an RV? repeat! */
		2288	if (AvFILLp (state) == 0 && SvROK (AvARRAY (state)[0]))
		2289	return 1;
		2290
		2291	/* restore status */
		2292	{
		2293	SV *data_sv = av_pop (state);
		2294	struct io_state *data = (struct io_state *)SvPVX (data_sv);
		2295
		2296	errno = data->errorno;
		2297	PL_laststype = data->laststype;
		2298	PL_laststatval = data->laststatval;
		2299	PL_statcache = data->statcache;
		2300
		2301	SvREFCNT_dec (data_sv);
		2302	}
		2303
		2304	/* push result values */
		2305	{
		2306	dSP;
		2307	int i;
		2308
		2309	EXTEND (SP, AvFILLp (state) + 1);
		2310	for (i = 0; i <= AvFILLp (state); ++i)
		2311	PUSHs (sv_2mortal (SvREFCNT_inc_NN (AvARRAY (state)[i])));
		2312
		2313	PUTBACK;
		2314	}
		2315
		2316	return 0;
		2317	}
		2318
		2319	static void
		2320	slf_init_aio_req (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		2321	{
		2322	AV *state = (AV *)sv_2mortal ((SV *)newAV ());
		2323	SV *coro_hv = SvRV (coro_current);
		2324	struct coro *coro = SvSTATE_hv (coro_hv);
		2325
		2326	/* put our coroutine id on the state arg */
		2327	av_push (state, SvREFCNT_inc_NN (coro_hv));
		2328
		2329	/* first see whether we have a non-zero priority and set it as AIO prio */
		2330	if (coro->prio)
		2331	{
		2332	dSP;
		2333
		2334	static SV *prio_cv;
		2335	static SV *prio_sv;
		2336
		2337	if (expect_false (!prio_cv))
		2338	{
		2339	prio_cv = (SV *)get_cv ("IO::AIO::aioreq_pri", 0);
		2340	prio_sv = newSViv (0);
		2341	}
		2342
		2343	PUSHMARK (SP);
		2344	sv_setiv (prio_sv, coro->prio);
		2345	XPUSHs (prio_sv);
		2346
		2347	PUTBACK;
		2348	call_sv (prio_cv, G_VOID | G_DISCARD);
		2349	}
		2350
		2351	/* now call the original request */
		2352	{
		2353	dSP;
		2354	CV *req = (CV *)CORO_MAGIC_NN ((SV *)cv, CORO_MAGIC_type_aio)->mg_obj;
		2355	int i;
		2356
		2357	PUSHMARK (SP);
		2358
		2359	/* first push all args to the stack */
		2360	EXTEND (SP, items + 1);
		2361
		2362	for (i = 0; i < items; ++i)
		2363	PUSHs (arg [i]);
		2364
		2365	/* now push the callback closure */
		2366	PUSHs (sv_2mortal (gensub (aTHX_ coro_aio_callback, (void *)SvREFCNT_inc_NN ((SV *)state))));
		2367
		2368	/* now call the AIO function - we assume our request is uncancelable */
		2369	PUTBACK;
		2370	call_sv ((SV *)req, G_VOID | G_DISCARD);
		2371	}
		2372
		2373	/* now that the requets is going, we loop toll we have a result */
		2374	frame->data = (void *)state;
		2375	frame->prepare = prepare_schedule;
		2376	frame->check = slf_check_aio_req;
		2377	}
		2378
		2379	static void
		2380	coro_aio_req_xs (pTHX_ CV *cv)
		2381	{
		2382	dXSARGS;
		2383
		2384	CORO_EXECUTE_SLF_XS (slf_init_aio_req);
		2385
		2386	XSRETURN_EMPTY;
2120	}	2387	}
2121		2388
2122	/*****************************************************************************/	2389	/*****************************************************************************/
2123		2390
2124	MODULE = Coro::State PACKAGE = Coro::State PREFIX = api_	2391	MODULE = Coro::State PACKAGE = Coro::State PREFIX = api_
…		…
2220	for (i = 1; i < items; i++)	2487	for (i = 1; i < items; i++)
2221	av_push (coro->args, newSVsv (ST (i)));	2488	av_push (coro->args, newSVsv (ST (i)));
2222	}	2489	}
2223	OUTPUT:	2490	OUTPUT:
2224	RETVAL	2491	RETVAL
2225
2226	void
2227	_set_stacklevel (...)
2228	CODE:
2229	CORO_EXECUTE_SLF_XS (slf_init_set_stacklevel);
2230		2492
2231	void	2493	void
2232	transfer (...)	2494	transfer (...)
2233	PROTOTYPE: $$	2495	PROTOTYPE: $$
2234	CODE:	2496	CODE:
…		…
2360	throw (Coro::State self, SV *throw = &PL_sv_undef)	2622	throw (Coro::State self, SV *throw = &PL_sv_undef)
2361	PROTOTYPE: $;$	2623	PROTOTYPE: $;$
2362	CODE:	2624	CODE:
2363	{	2625	{
2364	struct coro *current = SvSTATE_current;	2626	struct coro *current = SvSTATE_current;
2365	SV **throwp = self == current ? &coro_throw : &self->throw;	2627	SV **throwp = self == current ? &CORO_THROW : &self->except;
2366	SvREFCNT_dec (*throwp);	2628	SvREFCNT_dec (*throwp);
2367	*throwp = SvOK (throw) ? newSVsv (throw) : 0;	2629	*throwp = SvOK (throw) ? newSVsv (throw) : 0;
2368	}	2630	}
2369		2631
2370	void	2632	void
…		…
2422	SV **dst = ix ? (SV **)&self->slot->defav : (SV **)&self->slot->defsv;	2684	SV **dst = ix ? (SV **)&self->slot->defav : (SV **)&self->slot->defsv;
2423		2685
2424	SV *tmp = *src; *src = *dst; *dst = tmp;	2686	SV *tmp = *src; *src = *dst; *dst = tmp;
2425	}	2687	}
2426		2688
		2689
2427	MODULE = Coro::State PACKAGE = Coro	2690	MODULE = Coro::State PACKAGE = Coro
2428		2691
2429	BOOT:	2692	BOOT:
2430	{	2693	{
2431	int i;	2694	int i;
…		…
2458	coroapi.ready = api_ready;	2721	coroapi.ready = api_ready;
2459	coroapi.is_ready = api_is_ready;	2722	coroapi.is_ready = api_is_ready;
2460	coroapi.nready = coro_nready;	2723	coroapi.nready = coro_nready;
2461	coroapi.current = coro_current;	2724	coroapi.current = coro_current;
2462		2725
2463	GCoroAPI = &coroapi;	2726	/*GCoroAPI = &coroapi;*/
2464	sv_setiv (sv, (IV)&coroapi);	2727	sv_setiv (sv, (IV)&coroapi);
2465	SvREADONLY_on (sv);	2728	SvREADONLY_on (sv);
2466	}	2729	}
2467	}	2730	}
2468		2731
…		…
2605		2868
2606	av_push (av_async_pool, newSVsv (coro_current));	2869	av_push (av_async_pool, newSVsv (coro_current));
2607	}	2870	}
2608		2871
2609		2872
2610	MODULE = Coro::State PACKAGE = Coro::AIO
2611
2612	void
2613	_get_state (SV *self)
2614	PROTOTYPE: $
2615	PPCODE:
2616	{
2617	AV *defav = GvAV (PL_defgv);
2618	AV *av = newAV ();
2619	int i;
2620	SV *data_sv = newSV (sizeof (struct io_state));
2621	struct io_state *data = (struct io_state *)SvPVX (data_sv);
2622	SvCUR_set (data_sv, sizeof (struct io_state));
2623	SvPOK_only (data_sv);
2624
2625	data->errorno = errno;
2626	data->laststype = PL_laststype;
2627	data->laststatval = PL_laststatval;
2628	data->statcache = PL_statcache;
2629
2630	av_extend (av, AvFILLp (defav) + 1 + 1);
2631
2632	for (i = 0; i <= AvFILLp (defav); ++i)
2633	av_push (av, SvREFCNT_inc_NN (AvARRAY (defav)[i]));
2634
2635	av_push (av, data_sv);
2636
2637	XPUSHs (sv_2mortal (newRV_noinc ((SV *)av)));
2638
2639	api_ready (aTHX_ self);
2640	}
2641
2642	void
2643	_set_state (SV *state)
2644	PROTOTYPE: $
2645	PPCODE:
2646	{
2647	AV *av = (AV *)SvRV (state);
2648	struct io_state *data = (struct io_state *)SvPVX (AvARRAY (av)[AvFILLp (av)]);
2649	int i;
2650
2651	errno = data->errorno;
2652	PL_laststype = data->laststype;
2653	PL_laststatval = data->laststatval;
2654	PL_statcache = data->statcache;
2655
2656	EXTEND (SP, AvFILLp (av));
2657	for (i = 0; i < AvFILLp (av); ++i)
2658	PUSHs (sv_2mortal (SvREFCNT_inc_NN (AvARRAY (av)[i])));
2659	}
2660
2661
2662	MODULE = Coro::State PACKAGE = Coro::AnyEvent
2663
2664	BOOT:
2665	sv_activity = coro_get_sv (aTHX_ "Coro::AnyEvent::ACTIVITY", TRUE);
2666
2667	void
2668	_schedule (...)
2669	CODE:
2670	{
2671	static int incede;
2672
2673	api_cede_notself (aTHX);
2674
2675	++incede;
2676	while (coro_nready >= incede && api_cede (aTHX))
2677	;
2678
2679	sv_setsv (sv_activity, &PL_sv_undef);
2680	if (coro_nready >= incede)
2681	{
2682	PUSHMARK (SP);
2683	PUTBACK;
2684	call_pv ("Coro::AnyEvent::_activity", G_DISCARD | G_EVAL);
2685	SPAGAIN;
2686	}
2687
2688	--incede;
2689	}
2690
2691
2692	MODULE = Coro::State PACKAGE = PerlIO::cede	2873	MODULE = Coro::State PACKAGE = PerlIO::cede
2693		2874
2694	BOOT:	2875	BOOT:
2695	PerlIO_define_layer (aTHX_ &PerlIO_cede);	2876	PerlIO_define_layer (aTHX_ &PerlIO_cede);
2696		2877
		2878
2697	MODULE = Coro::State PACKAGE = Coro::Semaphore	2879	MODULE = Coro::State PACKAGE = Coro::Semaphore
2698		2880
2699	SV *	2881	SV *
2700	new (SV *klass, SV *count_ = 0)	2882	new (SV *klass, SV *count = 0)
2701	CODE:	2883	CODE:
2702	{	2884	RETVAL = sv_bless (
2703	/* a semaphore contains a counter IV in $sem->[0] and any waiters after that */	2885	coro_waitarray_new (aTHX_ count && SvOK (count) ? SvIV (count) : 1),
2704	AV *av = newAV ();	2886	GvSTASH (CvGV (cv))
2705	av_push (av, newSViv (count_ && SvOK (count_) ? SvIV (count_) : 1));	2887	);
2706	RETVAL = sv_bless (newRV_noinc ((SV *)av), GvSTASH (CvGV (cv)));	2888	OUTPUT:
2707	}	2889	RETVAL
		2890
		2891	# helper for Coro::Channel
		2892	SV *
		2893	_alloc (int count)
		2894	CODE:
		2895	RETVAL = coro_waitarray_new (aTHX_ count);
2708	OUTPUT:	2896	OUTPUT:
2709	RETVAL	2897	RETVAL
2710		2898
2711	SV *	2899	SV *
2712	count (SV *self)	2900	count (SV *self)
…		…
2724		2912
2725	void	2913	void
2726	down (SV *self)	2914	down (SV *self)
2727	CODE:	2915	CODE:
2728	CORO_EXECUTE_SLF_XS (slf_init_semaphore_down);	2916	CORO_EXECUTE_SLF_XS (slf_init_semaphore_down);
		2917
		2918	void
		2919	wait (SV *self)
		2920	CODE:
		2921	CORO_EXECUTE_SLF_XS (slf_init_semaphore_wait);
2729		2922
2730	void	2923	void
2731	try (SV *self)	2924	try (SV *self)
2732	PPCODE:	2925	PPCODE:
2733	{	2926	{
…		…
2745	XSRETURN_NO;	2938	XSRETURN_NO;
2746	}	2939	}
2747		2940
2748	void	2941	void
2749	waiters (SV *self)	2942	waiters (SV *self)
2750	CODE:	2943	PPCODE:
2751	{	2944	{
2752	AV *av = (AV *)SvRV (self);	2945	AV *av = (AV *)SvRV (self);
		2946	int wcount = AvFILLp (av) + 1 - 1;
2753		2947
2754	if (GIMME_V == G_SCALAR)	2948	if (GIMME_V == G_SCALAR)
2755	XPUSHs (sv_2mortal (newSVsv (AvARRAY (av)[0])));	2949	XPUSHs (sv_2mortal (newSViv (wcount)));
2756	else	2950	else
2757	{	2951	{
2758	int i;	2952	int i;
2759	EXTEND (SP, AvFILLp (av) + 1 - 1);	2953	EXTEND (SP, wcount);
2760	for (i = 1; i <= AvFILLp (av); ++i)	2954	for (i = 1; i <= wcount; ++i)
2761	PUSHs (newSVsv (AvARRAY (av)[i]));	2955	PUSHs (sv_2mortal (newRV_inc (AvARRAY (av)[i])));
2762	}	2956	}
2763	}	2957	}
2764		2958
		2959	MODULE = Coro::State PACKAGE = Coro::Signal
		2960
		2961	SV *
		2962	new (SV *klass)
		2963	CODE:
		2964	RETVAL = sv_bless (
		2965	coro_waitarray_new (aTHX_ 0),
		2966	GvSTASH (CvGV (cv))
		2967	);
		2968	OUTPUT:
		2969	RETVAL
		2970
		2971	void
		2972	wait (SV *self)
		2973	CODE:
		2974	CORO_EXECUTE_SLF_XS (slf_init_signal_wait);
		2975
		2976	void
		2977	broadcast (SV *self)
		2978	CODE:
		2979	{
		2980	AV *av = (AV *)SvRV (self);
		2981	coro_signal_wake (aTHX_ av, AvFILLp (av));
		2982	}
		2983
		2984	void
		2985	send (SV *self)
		2986	CODE:
		2987	{
		2988	AV *av = (AV *)SvRV (self);
		2989
		2990	if (AvFILLp (av))
		2991	coro_signal_wake (av, 1);
		2992	else
		2993	SvIVX (AvARRAY (av)[0]) = 1; /* remember the signal */
		2994	}
		2995
		2996	IV
		2997	awaited (SV *self)
		2998	CODE:
		2999	RETVAL = AvFILLp ((AV *)SvRV (self)) + 1 - 1;
		3000	OUTPUT:
		3001	RETVAL
		3002
		3003
		3004	MODULE = Coro::State PACKAGE = Coro::AnyEvent
		3005
		3006	BOOT:
		3007	sv_activity = coro_get_sv (aTHX_ "Coro::AnyEvent::ACTIVITY", TRUE);
		3008
		3009	void
		3010	_schedule (...)
		3011	CODE:
		3012	{
		3013	static int incede;
		3014
		3015	api_cede_notself (aTHX);
		3016
		3017	++incede;
		3018	while (coro_nready >= incede && api_cede (aTHX))
		3019	;
		3020
		3021	sv_setsv (sv_activity, &PL_sv_undef);
		3022	if (coro_nready >= incede)
		3023	{
		3024	PUSHMARK (SP);
		3025	PUTBACK;
		3026	call_pv ("Coro::AnyEvent::_activity", G_KEEPERR | G_EVAL | G_VOID | G_DISCARD);
		3027	}
		3028
		3029	--incede;
		3030	}
		3031
		3032
		3033	MODULE = Coro::State PACKAGE = Coro::AIO
		3034
		3035	void
		3036	_register (char *target, char *proto, SV *req)
		3037	CODE:
		3038	{
		3039	HV *st;
		3040	GV *gvp;
		3041	CV *req_cv = sv_2cv (req, &st, &gvp, 0);
		3042	/* newXSproto doesn't return the CV on 5.8 */
		3043	CV *slf_cv = newXS (target, coro_aio_req_xs, __FILE__);
		3044	sv_setpv ((SV *)slf_cv, proto);
		3045	sv_magicext ((SV *)slf_cv, (SV *)req_cv, CORO_MAGIC_type_aio, 0, 0, 0);
		3046	}
		3047

Diff Legend

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