ViewVC Help

View File | Revision Log | Show Annotations | Download File

/cvs/Coro/Coro/State.xs

Comparing Coro/Coro/State.xs (file contents):
Revision 1.286 by root, Mon Nov 17 04:19:49 2008 UTC vs.
Revision 1.302 by root, Wed Nov 19 04:48:24 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 */
…		…
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 */
…		…
273		274
274	/* the following variables are effectively part of the perl context */	275	/* the following variables are effectively part of the perl context */
275	/* and get copied between struct coro and these variables */	276	/* and get copied between struct coro and these variables */
276	/* the mainr easonw e don't support windows process emulation */	277	/* the mainr easonw e don't support windows process emulation */
277	static struct CoroSLF slf_frame; /* the current slf frame */	278	static struct CoroSLF slf_frame; /* the current slf frame */
278	static SV *coro_throw;
279		279
280	/** Coro ********************************************************************/	280	/** Coro ********************************************************************/
281		281
282	#define PRIO_MAX 3	282	#define PRIO_MAX 3
283	#define PRIO_HIGH 1	283	#define PRIO_HIGH 1
…		…
384	SvREFCNT_dec (av); /* sv_magicext increased the refcount */	384	SvREFCNT_dec (av); /* sv_magicext increased the refcount */
385		385
386	return 0;	386	return 0;
387	}	387	}
388		388
389	#define CORO_MAGIC_type_cv PERL_MAGIC_ext	389	#define CORO_MAGIC_type_cv 26
390	#define CORO_MAGIC_type_state PERL_MAGIC_ext	390	#define CORO_MAGIC_type_state PERL_MAGIC_ext
391		391
392	static MGVTBL coro_cv_vtbl = {	392	static MGVTBL coro_cv_vtbl = {
393	0, 0, 0, 0,	393	0, 0, 0, 0,
394	coro_cv_free	394	coro_cv_free
395	};	395	};
396		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
397	#define CORO_MAGIC(sv, type) \	402	#define CORO_MAGIC(sv, type) \
398	expect_true (SvMAGIC (sv)) \	403	(expect_true (SvMAGIC (sv)) \
399	? expect_true (SvMAGIC (sv)->mg_type == type) \	404	? CORO_MAGIC_NN (sv, type) \
400	? SvMAGIC (sv) \
401	: mg_find (sv, type) \
402	: 0	405	: 0)
403		406
404	#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)
405	#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)
406		409
407	INLINE struct coro *	410	INLINE struct coro *
408	SvSTATE_ (pTHX_ SV *coro)	411	SvSTATE_ (pTHX_ SV *coro)
409	{	412	{
410	HV *stash;	413	HV *stash;
…		…
428	return (struct coro *)mg->mg_ptr;	431	return (struct coro *)mg->mg_ptr;
429	}	432	}
430		433
431	#define SvSTATE(sv) SvSTATE_ (aTHX_ (sv))	434	#define SvSTATE(sv) SvSTATE_ (aTHX_ (sv))
432		435
433	/* fastert than SvSTATE, but expects a coroutine hv */	436	/* faster than SvSTATE, but expects a coroutine hv */
434	INLINE struct coro *	437	#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))	438	#define SvSTATE_current SvSTATE_hv (SvRV (coro_current))
445		439
446	/* the next two functions merely cache the padlists */	440	/* the next two functions merely cache the padlists */
447	static void	441	static void
448	get_padlist (pTHX_ CV *cv)	442	get_padlist (pTHX_ CV *cv)
…		…
518		512
519	PUTBACK;	513	PUTBACK;
520	}	514	}
521		515
522	slf_frame = c->slf_frame;	516	slf_frame = c->slf_frame;
523	coro_throw = c->throw;	517	CORO_THROW = c->except;
524	}	518	}
525		519
526	static void	520	static void
527	save_perl (pTHX_ Coro__State c)	521	save_perl (pTHX_ Coro__State c)
528	{	522	{
529	c->throw = coro_throw;	523	c->except = CORO_THROW;
530	c->slf_frame = slf_frame;	524	c->slf_frame = slf_frame;
531		525
532	{	526	{
533	dSP;	527	dSP;
534	I32 cxix = cxstack_ix;	528	I32 cxix = cxstack_ix;
…		…
827	slf_check_nop (pTHX_ struct CoroSLF *frame)	821	slf_check_nop (pTHX_ struct CoroSLF *frame)
828	{	822	{
829	return 0;	823	return 0;
830	}	824	}
831		825
		826	static UNOP coro_setup_op;
		827
832	static void NOINLINE /* noinline to keep it out of the transfer fast path */	828	static void NOINLINE /* noinline to keep it out of the transfer fast path */
833	coro_setup (pTHX_ struct coro *coro)	829	coro_setup (pTHX_ struct coro *coro)
834	{	830	{
835	/*	831	/*
836	* emulate part of the perl startup here.	832	* emulate part of the perl startup here.
…		…
864	{	860	{
865	dSP;	861	dSP;
866	UNOP myop;	862	UNOP myop;
867		863
868	Zero (&myop, 1, UNOP);	864	Zero (&myop, 1, UNOP);
869	myop.op_next = Nullop;	865	myop.op_next = Nullop;
870	myop.op_flags = OPf_WANT_VOID;	866	myop.op_flags = OPf_WANT_VOID;
871		867
872	PUSHMARK (SP);	868	PUSHMARK (SP);
873	XPUSHs (sv_2mortal (av_shift (GvAV (PL_defgv))));	869	XPUSHs (sv_2mortal (av_shift (GvAV (PL_defgv))));
874	PUTBACK;	870	PUTBACK;
…		…
881	* 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.
882	*/	878	*/
883	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 */
884	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 */
885		881
886	coro_throw = coro->throw;	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;
887	}	892	}
888		893
889	static void	894	static void
890	coro_destruct (pTHX_ struct coro *coro)	895	coro_destruct (pTHX_ struct coro *coro)
891	{	896	{
…		…
914	SvREFCNT_dec (GvSV (irsgv));	919	SvREFCNT_dec (GvSV (irsgv));
915		920
916	SvREFCNT_dec (PL_diehook);	921	SvREFCNT_dec (PL_diehook);
917	SvREFCNT_dec (PL_warnhook);	922	SvREFCNT_dec (PL_warnhook);
918		923
		924	SvREFCNT_dec (CORO_THROW);
919	SvREFCNT_dec (coro->saved_deffh);	925	SvREFCNT_dec (coro->saved_deffh);
920	SvREFCNT_dec (coro_throw);	926	SvREFCNT_dec (coro->rouse_cb);
921		927
922	coro_destruct_stacks (aTHX);	928	coro_destruct_stacks (aTHX);
923	}	929	}
924		930
925	INLINE void	931	INLINE void
…		…
1054		1060
1055	TAINT_NOT;	1061	TAINT_NOT;
1056	return 0;	1062	return 0;
1057	}	1063	}
1058		1064
		1065	static struct coro_cctx *cctx_ssl_cctx;
		1066	static struct CoroSLF cctx_ssl_frame;
		1067
1059	static void	1068	static void
1060	prepare_set_stacklevel (struct coro_transfer_args *ta, struct coro_cctx *cctx)	1069	slf_prepare_set_stacklevel (pTHX_ struct coro_transfer_args *ta)
1061	{	1070	{
1062	ta->prev = (struct coro *)cctx;	1071	ta->prev = (struct coro *)cctx_ssl_cctx;
1063	ta->next = 0;	1072	ta->next = 0;
1064	}	1073	}
1065		1074
1066	/* inject a fake call to Coro::State::_cctx_init into the execution */	1075	static int
1067	/* _cctx_init should be careful, as it could be called at almost any time */	1076	slf_check_set_stacklevel (pTHX_ struct CoroSLF *frame)
1068	/* during execution of a perl program */	1077	{
1069	/* 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 */
1070	static void NOINLINE	1084	static void NOINLINE
1071	cctx_prepare (pTHX_ coro_cctx *cctx)	1085	cctx_prepare (pTHX_ coro_cctx *cctx)
1072	{	1086	{
1073	dSP;
1074	UNOP myop;
1075
1076	PL_top_env = &PL_start_env;	1087	PL_top_env = &PL_start_env;
1077		1088
1078	if (cctx->flags & CC_TRACE)	1089	if (cctx->flags & CC_TRACE)
1079	PL_runops = runops_trace;	1090	PL_runops = runops_trace;
1080		1091
1081	Zero (&myop, 1, UNOP);	1092	/* we already must be executing an SLF op, there is no other valid way
1082	myop.op_next = PL_op;	1093	* that can lead to creation of a new cctx */
1083	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));
1084		1096
1085	PUSHMARK (SP);	1097	/* we must emulate leaving pp_slf, which is done inside slf_check_set_stacklevel */
1086	EXTEND (SP, 2);	1098	cctx_ssl_cctx = cctx;
1087	PUSHs (sv_2mortal (newSViv ((IV)cctx)));	1099	cctx_ssl_frame = slf_frame;
1088	PUSHs ((SV *)get_cv ("Coro::State::_cctx_init", FALSE));	1100
1089	PUTBACK;	1101	slf_frame.prepare = slf_prepare_set_stacklevel;
1090	PL_op = (OP *)&myop;	1102	slf_frame.check = slf_check_set_stacklevel;
1091	PL_op = PL_ppaddr[OP_ENTERSUB](aTHX);
1092	SPAGAIN;
1093	}	1103	}
1094		1104
1095	/* 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 */
1096	INLINE void	1106	INLINE void
1097	transfer_tail (pTHX)	1107	transfer_tail (pTHX)
…		…
1313	dSTACKLEVEL;	1323	dSTACKLEVEL;
1314		1324
1315	/* sometimes transfer is only called to set idle_sp */	1325	/* sometimes transfer is only called to set idle_sp */
1316	if (expect_false (!next))	1326	if (expect_false (!next))
1317	{	1327	{
1318	((coro_cctx *)prev)->idle_sp = (void *)stacklevel;	1328	((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 */	1329	assert (((coro_cctx *)prev)->idle_te = PL_top_env); /* just for the side-effect when asserts are enabled */
1320	}	1330	}
1321	else if (expect_true (prev != next))	1331	else if (expect_true (prev != next))
1322	{	1332	{
1323	coro_cctx *prev__cctx;	1333	coro_cctx *prev__cctx;
…		…
1348		1358
1349	prev__cctx = prev->cctx;	1359	prev__cctx = prev->cctx;
1350		1360
1351	/* possibly untie and reuse the cctx */	1361	/* possibly untie and reuse the cctx */
1352	if (expect_true (	1362	if (expect_true (
1353	prev__cctx->idle_sp == (void *)stacklevel	1363	prev__cctx->idle_sp == STACKLEVEL
1354	&& !(prev__cctx->flags & CC_TRACE)	1364	&& !(prev__cctx->flags & CC_TRACE)
1355	&& !force_cctx	1365	&& !force_cctx
1356	))	1366	))
1357	{	1367	{
1358	/* 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 */
…		…
1489		1499
1490	prepare_transfer (aTHX_ &ta, prev_sv, next_sv);	1500	prepare_transfer (aTHX_ &ta, prev_sv, next_sv);
1491	TRANSFER (ta, 1);	1501	TRANSFER (ta, 1);
1492	}	1502	}
1493		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
1494	/** Coro ********************************************************************/	1527	/** Coro ********************************************************************/
1495		1528
1496	INLINE void	1529	INLINE void
1497	coro_enq (pTHX_ struct coro *coro)	1530	coro_enq (pTHX_ struct coro *coro)
1498	{	1531	{
…		…
1541	ENTER;	1574	ENTER;
1542	SAVETMPS;	1575	SAVETMPS;
1543		1576
1544	PUSHMARK (SP);	1577	PUSHMARK (SP);
1545	PUTBACK;	1578	PUTBACK;
1546	call_sv (sv_hook, G_DISCARD);	1579	call_sv (sv_hook, G_VOID | G_DISCARD);
1547	SPAGAIN;
1548		1580
1549	FREETMPS;	1581	FREETMPS;
1550	LEAVE;	1582	LEAVE;
1551	}	1583	}
1552		1584
…		…
1579	ENTER;	1611	ENTER;
1580	SAVETMPS;	1612	SAVETMPS;
1581		1613
1582	PUSHMARK (SP);	1614	PUSHMARK (SP);
1583	PUTBACK;	1615	PUTBACK;
1584	call_sv (get_sv ("Coro::idle", FALSE), G_DISCARD);	1616	call_sv (get_sv ("Coro::idle", FALSE), G_VOID | G_DISCARD);
1585	SPAGAIN;
1586		1617
1587	FREETMPS;	1618	FREETMPS;
1588	LEAVE;	1619	LEAVE;
1589	continue;	1620	continue;
1590	}	1621	}
…		…
1697	if (coro->flags & CF_RUNNING)	1728	if (coro->flags & CF_RUNNING)
1698	PL_runops = RUNOPS_DEFAULT;	1729	PL_runops = RUNOPS_DEFAULT;
1699	else	1730	else
1700	coro->slot->runops = RUNOPS_DEFAULT;	1731	coro->slot->runops = RUNOPS_DEFAULT;
1701	}	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	while (items--)
		1758	av_store (av, items, SvREFCNT_inc_NN (ST (items)));
		1759	}
		1760
		1761	XSRETURN_EMPTY;
		1762	}
		1763
		1764	static int
		1765	slf_check_rouse_wait (pTHX_ struct CoroSLF *frame)
		1766	{
		1767	SV *data = (SV *)frame->data;
		1768
		1769	if (CORO_THROW)
		1770	return 0;
		1771
		1772	if (SvTYPE (SvRV (data)) != SVt_PVAV)
		1773	return 1;
		1774
		1775	/* now push all results on the stack */
		1776	{
		1777	dSP;
		1778	AV *av = (AV *)SvRV (data);
		1779	int i;
		1780
		1781	EXTEND (SP, AvFILLp (av) + 1);
		1782	for (i = 0; i <= AvFILLp (av); ++i)
		1783	PUSHs (AvARRAY (av)[i]);
		1784
		1785	PUTBACK;
		1786	}
		1787
		1788	return 0;
		1789	}
		1790
		1791	static void
		1792	slf_init_rouse_wait (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		1793	{
		1794	SV *cb;
		1795
		1796	if (items)
		1797	cb = arg [0];
		1798	else
		1799	{
		1800	struct coro *coro = SvSTATE_current;
		1801
		1802	if (!coro->rouse_cb)
		1803	croak ("Coro::rouse_wait called without rouse callback, and no default rouse callback found either,");
		1804
		1805	cb = sv_2mortal (coro->rouse_cb);
		1806	coro->rouse_cb = 0;
		1807	}
		1808
		1809	if (!SvROK (cb)
		1810	|| SvTYPE (SvRV (cb)) != SVt_PVCV
		1811	|| CvXSUB ((CV *)SvRV (cb)) != coro_rouse_callback)
		1812	croak ("Coro::rouse_wait called with illegal callback argument,");
		1813
		1814	{
		1815	CV *cv = (CV *)SvRV (cb); /* for GENSUB_ARG */
		1816	SV *data = (SV *)GENSUB_ARG;
		1817
		1818	frame->data = (void *)data;
		1819	frame->prepare = SvTYPE (SvRV (data)) == SVt_PVAV ? prepare_nop : prepare_schedule;
		1820	frame->check = slf_check_rouse_wait;
		1821	}
		1822	}
		1823
		1824	static SV *
		1825	coro_new_rouse_cb (pTHX)
		1826	{
		1827	HV *hv = (HV *)SvRV (coro_current);
		1828	struct coro *coro = SvSTATE_hv (hv);
		1829	SV *data = newRV_inc ((SV *)hv);
		1830	SV *cb = gensub (aTHX_ coro_rouse_callback, (void *)data);
		1831
		1832	sv_magicext (SvRV (cb), data, CORO_MAGIC_type_rouse, 0, 0, 0);
		1833
		1834	SvREFCNT_dec (coro->rouse_cb);
		1835	coro->rouse_cb = SvREFCNT_inc_NN (cb);
		1836
		1837	return cb;
		1838	}
		1839
		1840	/*****************************************************************************/
		1841	/* schedule-like-function opcode (SLF) */
		1842
		1843	static UNOP slf_restore; /* restore stack as entersub did, for first-re-run */
		1844	static const CV *slf_cv;
		1845	static SV **slf_argv;
		1846	static int slf_argc, slf_arga; /* count, allocated */
		1847	static I32 slf_ax; /* top of stack, for restore */
		1848
		1849	/* this restores the stack in the case we patched the entersub, to */
		1850	/* recreate the stack frame as perl will on following calls */
		1851	/* since entersub cleared the stack */
		1852	static OP *
		1853	pp_restore (pTHX)
		1854	{
		1855	int i;
		1856	SV **SP = PL_stack_base + slf_ax;
		1857
		1858	PUSHMARK (SP);
		1859
		1860	EXTEND (SP, slf_argc + 1);
		1861
		1862	for (i = 0; i < slf_argc; ++i)
		1863	PUSHs (sv_2mortal (slf_argv [i]));
		1864
		1865	PUSHs ((SV *)CvGV (slf_cv));
		1866
		1867	RETURNOP (slf_restore.op_first);
		1868	}
		1869
		1870	static void
		1871	slf_prepare_transfer (pTHX_ struct coro_transfer_args *ta)
		1872	{
		1873	SV **arg = (SV **)slf_frame.data;
		1874
		1875	prepare_transfer (aTHX_ ta, arg [0], arg [1]);
		1876	}
		1877
		1878	static void
		1879	slf_init_transfer (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		1880	{
		1881	if (items != 2)
		1882	croak ("Coro::State::transfer (prev, next) expects two arguments, not %d,", items);
		1883
		1884	frame->prepare = slf_prepare_transfer;
		1885	frame->check = slf_check_nop;
		1886	frame->data = (void *)arg; /* let's hope it will stay valid */
		1887	}
		1888
		1889	static void
		1890	slf_init_schedule (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		1891	{
		1892	frame->prepare = prepare_schedule;
		1893	frame->check = slf_check_nop;
		1894	}
		1895
		1896	static void
		1897	slf_init_cede (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		1898	{
		1899	frame->prepare = prepare_cede;
		1900	frame->check = slf_check_nop;
		1901	}
		1902
		1903	static void
		1904	slf_init_cede_notself (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		1905	{
		1906	frame->prepare = prepare_cede_notself;
		1907	frame->check = slf_check_nop;
		1908	}
		1909
		1910	/*
		1911	* these not obviously related functions are all rolled into one
		1912	* function to increase chances that they all will call transfer with the same
		1913	* stack offset
		1914	* SLF stands for "schedule-like-function".
		1915	*/
		1916	static OP *
		1917	pp_slf (pTHX)
		1918	{
		1919	I32 checkmark; /* mark SP to see how many elements check has pushed */
		1920
		1921	/* set up the slf frame, unless it has already been set-up */
		1922	/* the latter happens when a new coro has been started */
		1923	/* or when a new cctx was attached to an existing coroutine */
		1924	if (expect_true (!slf_frame.prepare))
		1925	{
		1926	/* first iteration */
		1927	dSP;
		1928	SV **arg = PL_stack_base + TOPMARK + 1;
		1929	int items = SP - arg; /* args without function object */
		1930	SV *gv = *sp;
		1931
		1932	/* do a quick consistency check on the "function" object, and if it isn't */
		1933	/* for us, divert to the real entersub */
		1934	if (SvTYPE (gv) != SVt_PVGV
		1935	|| !GvCV (gv)
		1936	|| !(CvFLAGS (GvCV (gv)) & CVf_SLF))
		1937	return PL_ppaddr[OP_ENTERSUB](aTHX);
		1938
		1939	if (!(PL_op->op_flags & OPf_STACKED))
		1940	{
		1941	/* ampersand-form of call, use @_ instead of stack */
		1942	AV *av = GvAV (PL_defgv);
		1943	arg = AvARRAY (av);
		1944	items = AvFILLp (av) + 1;
		1945	}
		1946
		1947	/* now call the init function, which needs to set up slf_frame */
		1948	((coro_slf_cb)CvXSUBANY (GvCV (gv)).any_ptr)
		1949	(aTHX_ &slf_frame, GvCV (gv), arg, items);
		1950
		1951	/* pop args */
		1952	SP = PL_stack_base + POPMARK;
		1953
		1954	PUTBACK;
		1955	}
		1956
		1957	/* now that we have a slf_frame, interpret it! */
		1958	/* we use a callback system not to make the code needlessly */
		1959	/* complicated, but so we can run multiple perl coros from one cctx */
		1960
		1961	do
		1962	{
		1963	struct coro_transfer_args ta;
		1964
		1965	slf_frame.prepare (aTHX_ &ta);
		1966	TRANSFER (ta, 0);
		1967
		1968	checkmark = PL_stack_sp - PL_stack_base;
		1969	}
		1970	while (slf_frame.check (aTHX_ &slf_frame));
		1971
		1972	slf_frame.prepare = 0; /* invalidate the frame, we are done processing it */
		1973
		1974	/* exception handling */
		1975	if (expect_false (CORO_THROW))
		1976	{
		1977	SV *exception = sv_2mortal (CORO_THROW);
		1978
		1979	CORO_THROW = 0;
		1980	sv_setsv (ERRSV, exception);
		1981	croak (0);
		1982	}
		1983
		1984	/* return value handling - mostly like entersub */
		1985	/* make sure we put something on the stack in scalar context */
		1986	if (GIMME_V == G_SCALAR)
		1987	{
		1988	dSP;
		1989	SV **bot = PL_stack_base + checkmark;
		1990
		1991	if (sp == bot) /* too few, push undef */
		1992	bot [1] = &PL_sv_undef;
		1993	else if (sp != bot + 1) /* too many, take last one */
		1994	bot [1] = *sp;
		1995
		1996	SP = bot + 1;
		1997
		1998	PUTBACK;
		1999	}
		2000
		2001	return NORMAL;
		2002	}
		2003
		2004	static void
		2005	api_execute_slf (pTHX_ CV *cv, coro_slf_cb init_cb, I32 ax)
		2006	{
		2007	int i;
		2008	SV **arg = PL_stack_base + ax;
		2009	int items = PL_stack_sp - arg + 1;
		2010
		2011	assert (("FATAL: SLF call with illegal CV value", !CvANON (cv)));
		2012
		2013	if (PL_op->op_ppaddr != PL_ppaddr [OP_ENTERSUB]
		2014	&& PL_op->op_ppaddr != pp_slf)
		2015	croak ("FATAL: Coro SLF calls can only be made normally, not via goto or any other means, caught");
		2016
		2017	CvFLAGS (cv) |= CVf_SLF;
		2018	CvXSUBANY (cv).any_ptr = (void *)init_cb;
		2019	slf_cv = cv;
		2020
		2021	/* we patch the op, and then re-run the whole call */
		2022	/* we have to put the same argument on the stack for this to work */
		2023	/* and this will be done by pp_restore */
		2024	slf_restore.op_next = (OP *)&slf_restore;
		2025	slf_restore.op_type = OP_CUSTOM;
		2026	slf_restore.op_ppaddr = pp_restore;
		2027	slf_restore.op_first = PL_op;
		2028
		2029	slf_ax = ax - 1; /* undo the ax++ inside dAXMARK */
		2030
		2031	if (PL_op->op_flags & OPf_STACKED)
		2032	{
		2033	if (items > slf_arga)
		2034	{
		2035	slf_arga = items;
		2036	free (slf_argv);
		2037	slf_argv = malloc (slf_arga * sizeof (SV *));
		2038	}
		2039
		2040	slf_argc = items;
		2041
		2042	for (i = 0; i < items; ++i)
		2043	slf_argv [i] = SvREFCNT_inc (arg [i]);
		2044	}
		2045	else
		2046	slf_argc = 0;
		2047
		2048	PL_op->op_ppaddr = pp_slf;
		2049	PL_op->op_type = OP_CUSTOM; /* maybe we should leave it at entersub? */
		2050
		2051	PL_op = (OP *)&slf_restore;
1702	}	2052	}
1703		2053
1704	/*****************************************************************************/	2054	/*****************************************************************************/
1705	/* PerlIO::cede */	2055	/* PerlIO::cede */
1706		2056
…		…
1775	PerlIOBuf_get_cnt,	2125	PerlIOBuf_get_cnt,
1776	PerlIOBuf_set_ptrcnt,	2126	PerlIOBuf_set_ptrcnt,
1777	};	2127	};
1778		2128
1779	/*****************************************************************************/	2129	/*****************************************************************************/
		2130	/* Coro::Semaphore & Coro::Signal */
1780		2131
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 *	2132	static SV *
1791	pp_restore (pTHX)	2133	coro_waitarray_new (pTHX_ int count)
1792	{	2134	{
1793	int i;	2135	/* a semaphore contains a counter IV in $sem->[0] and any waiters after that */
1794	SV **SP = PL_stack_base + slf_ax;	2136	AV *av = newAV ();
		2137	SV **ary;
1795		2138
1796	PUSHMARK (SP);	2139	/* unfortunately, building manually saves memory */
		2140	Newx (ary, 2, SV *);
		2141	AvALLOC (av) = ary;
		2142	AvARRAY (av) = ary;
		2143	AvMAX (av) = 1;
		2144	AvFILLp (av) = 0;
		2145	ary [0] = newSViv (count);
1797		2146
1798	EXTEND (SP, slf_argc + 1);	2147	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	}	2148	}
1807		2149
1808	static void	2150	/* 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 (PL_op->op_flags & OPf_STACKED)
1988	{
1989	if (items > slf_arga)
1990	{
1991	slf_arga = items;
1992	free (slf_argv);
1993	slf_argv = malloc (slf_arga * sizeof (SV *));
1994	}
1995
1996	slf_argc = items;
1997
1998	for (i = 0; i < items; ++i)
1999	slf_argv [i] = SvREFCNT_inc (arg [i]);
2000	}
2001	else
2002	slf_argc = 0;
2003
2004	PL_op->op_ppaddr = pp_slf;
2005	PL_op->op_type = OP_CUSTOM; /* maybe we should leave it at entersub? */
2006
2007	PL_op = (OP *)&slf_restore;
2008	}
2009
2010	/*****************************************************************************/
2011		2151
2012	static void	2152	static void
2013	coro_semaphore_adjust (pTHX_ AV *av, IV adjust)	2153	coro_semaphore_adjust (pTHX_ AV *av, IV adjust)
2014	{	2154	{
2015	SV *count_sv = AvARRAY (av)[0];	2155	SV *count_sv = AvARRAY (av)[0];
…		…
2045	/* call $sem->adjust (0) to possibly wake up some other waiters */	2185	/* call $sem->adjust (0) to possibly wake up some other waiters */
2046	coro_semaphore_adjust (aTHX_ (AV *)coro->slf_frame.data, 0);	2186	coro_semaphore_adjust (aTHX_ (AV *)coro->slf_frame.data, 0);
2047	}	2187	}
2048		2188
2049	static int	2189	static int
2050	slf_check_semaphore_down (pTHX_ struct CoroSLF *frame)	2190	slf_check_semaphore_down_or_wait (pTHX_ struct CoroSLF *frame, int acquire)
2051	{	2191	{
2052	AV *av = (AV *)frame->data;	2192	AV *av = (AV *)frame->data;
2053	SV *count_sv = AvARRAY (av)[0];	2193	SV *count_sv = AvARRAY (av)[0];
2054		2194
		2195	/* if we are about to throw, don't actually acquire the lock, just throw */
		2196	if (CORO_THROW)
		2197	return 0;
2055	if (SvIVX (count_sv) > 0)	2198	else if (SvIVX (count_sv) > 0)
2056	{	2199	{
2057	SvSTATE_current->on_destroy = 0;	2200	SvSTATE_current->on_destroy = 0;
		2201
		2202	if (acquire)
2058	SvIVX (count_sv) = SvIVX (count_sv) - 1;	2203	SvIVX (count_sv) = SvIVX (count_sv) - 1;
		2204	else
		2205	coro_semaphore_adjust (aTHX_ av, 0);
		2206
2059	return 0;	2207	return 0;
2060	}	2208	}
2061	else	2209	else
2062	{	2210	{
2063	int i;	2211	int i;
…		…
2072	av_push (av, SvREFCNT_inc (SvRV (coro_current)));	2220	av_push (av, SvREFCNT_inc (SvRV (coro_current)));
2073	return 1;	2221	return 1;
2074	}	2222	}
2075	}	2223	}
2076		2224
2077	static void	2225	static int
		2226	slf_check_semaphore_down (pTHX_ struct CoroSLF *frame)
		2227	{
		2228	return slf_check_semaphore_down_or_wait (aTHX_ frame, 1);
		2229	}
		2230
		2231	static int
		2232	slf_check_semaphore_wait (pTHX_ struct CoroSLF *frame)
		2233	{
		2234	return slf_check_semaphore_down_or_wait (aTHX_ frame, 0);
		2235	}
		2236
		2237	static void
2078	slf_init_semaphore_down (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)	2238	slf_init_semaphore_down_or_wait (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
2079	{	2239	{
2080	AV *av = (AV *)SvRV (arg [0]);	2240	AV *av = (AV *)SvRV (arg [0]);
2081		2241
2082	if (SvIVX (AvARRAY (av)[0]) > 0)	2242	if (SvIVX (AvARRAY (av)[0]) > 0)
2083	{	2243	{
2084	frame->data = (void *)av;	2244	frame->data = (void *)av;
2085	frame->prepare = prepare_nop;	2245	frame->prepare = prepare_nop;
2086	SvSTATE_current->on_destroy = coro_semaphore_on_destroy;
2087	}	2246	}
2088	else	2247	else
2089	{	2248	{
2090	av_push (av, SvREFCNT_inc (SvRV (coro_current)));	2249	av_push (av, SvREFCNT_inc (SvRV (coro_current)));
2091		2250
2092	frame->data = (void *)sv_2mortal (SvREFCNT_inc ((SV *)av));	2251	frame->data = (void *)sv_2mortal (SvREFCNT_inc ((SV *)av));
2093	frame->prepare = prepare_schedule;	2252	frame->prepare = prepare_schedule;
2094		2253
2095	/* to avoid race conditions when a woken-up coro gets terminated */	2254	/* to avoid race conditions when a woken-up coro gets terminated */
2096	/* we arrange for a temporary on_destroy that calls adjust (0) */	2255	/* we arrange for a temporary on_destroy that calls adjust (0) */
2097	assert (!SvSTATE_current->on_destroy);//D
2098	SvSTATE_current->on_destroy = coro_semaphore_on_destroy;	2256	SvSTATE_current->on_destroy = coro_semaphore_on_destroy;
2099	}	2257	}
		2258	}
2100		2259
		2260	static void
		2261	slf_init_semaphore_down (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		2262	{
		2263	slf_init_semaphore_down_or_wait (aTHX_ frame, cv, arg, items);
2101	frame->check = slf_check_semaphore_down;	2264	frame->check = slf_check_semaphore_down;
		2265	}
2102		2266
		2267	static void
		2268	slf_init_semaphore_wait (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		2269	{
		2270	slf_init_semaphore_down_or_wait (aTHX_ frame, cv, arg, items);
		2271	frame->check = slf_check_semaphore_wait;
		2272	}
		2273
		2274	/* signal */
		2275
		2276	static void
		2277	coro_signal_wake (pTHX_ AV *av, int count)
		2278	{
		2279	SvIVX (AvARRAY (av)[0]) = 0;
		2280
		2281	/* now signal count waiters */
		2282	while (count > 0 && AvFILLp (av) > 0)
		2283	{
		2284	SV *cb;
		2285
		2286	/* swap first two elements so we can shift a waiter */
		2287	cb = AvARRAY (av)[0];
		2288	AvARRAY (av)[0] = AvARRAY (av)[1];
		2289	AvARRAY (av)[1] = cb;
		2290
		2291	cb = av_shift (av);
		2292
		2293	api_ready (aTHX_ cb);
		2294	sv_setiv (cb, 0); /* signal waiter */
		2295	SvREFCNT_dec (cb);
		2296
		2297	--count;
		2298	}
		2299	}
		2300
		2301	static int
		2302	slf_check_signal_wait (pTHX_ struct CoroSLF *frame)
		2303	{
		2304	/* if we are about to throw, also stop waiting */
		2305	return SvROK ((SV *)frame->data) && !CORO_THROW;
		2306	}
		2307
		2308	static void
		2309	slf_init_signal_wait (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		2310	{
		2311	AV *av = (AV *)SvRV (arg [0]);
		2312
		2313	if (SvIVX (AvARRAY (av)[0]))
		2314	{
		2315	SvIVX (AvARRAY (av)[0]) = 0;
		2316	frame->prepare = prepare_nop;
		2317	frame->check = slf_check_nop;
		2318	}
		2319	else
		2320	{
		2321	SV *waiter = newRV_inc (SvRV (coro_current)); /* owned by signal av */
		2322
		2323	av_push (av, waiter);
		2324
		2325	frame->data = (void *)sv_2mortal (SvREFCNT_inc_NN (waiter)); /* owned by process */
		2326	frame->prepare = prepare_schedule;
		2327	frame->check = slf_check_signal_wait;
		2328	}
2103	}	2329	}
2104		2330
2105	/*****************************************************************************/	2331	/*****************************************************************************/
		2332	/* Coro::AIO */
2106		2333
2107	#define GENSUB_ARG CvXSUBANY (cv).any_ptr	2334	#define CORO_MAGIC_type_aio PERL_MAGIC_ext
2108		2335
2109	/* create a closure from XS, returns a code reference */	2336	/* helper storage struct */
2110	/* the arg can be accessed via GENSUB_ARG from the callback */	2337	struct io_state
2111	/* the callback must use dXSARGS/XSRETURN */
2112	static SV *
2113	gensub (pTHX_ void (*xsub)(pTHX_ CV *), void *arg)
2114	{	2338	{
2115	CV *cv = (CV *)NEWSV (0, 0);	2339	int errorno;
		2340	I32 laststype; /* U16 in 5.10.0 */
		2341	int laststatval;
		2342	Stat_t statcache;
		2343	};
2116		2344
		2345	static void
		2346	coro_aio_callback (pTHX_ CV *cv)
		2347	{
		2348	dXSARGS;
		2349	AV *state = (AV *)GENSUB_ARG;
		2350	SV *coro = av_pop (state);
		2351	SV *data_sv = newSV (sizeof (struct io_state));
		2352
		2353	av_extend (state, items);
		2354
2117	sv_upgrade ((SV *)cv, SVt_PVCV);	2355	sv_upgrade (data_sv, SVt_PV);
		2356	SvCUR_set (data_sv, sizeof (struct io_state));
		2357	SvPOK_only (data_sv);
2118		2358
2119	CvANON_on (cv);	2359	{
2120	CvISXSUB_on (cv);	2360	struct io_state *data = (struct io_state *)SvPVX (data_sv);
2121	CvXSUB (cv) = xsub;
2122	GENSUB_ARG = arg;
2123		2361
2124	return newRV_noinc ((SV *)cv);	2362	data->errorno = errno;
		2363	data->laststype = PL_laststype;
		2364	data->laststatval = PL_laststatval;
		2365	data->statcache = PL_statcache;
		2366	}
		2367
		2368	/* now build the result vector out of all the parameters and the data_sv */
		2369	{
		2370	int i;
		2371
		2372	for (i = 0; i < items; ++i)
		2373	av_push (state, SvREFCNT_inc_NN (ST (i)));
		2374	}
		2375
		2376	av_push (state, data_sv);
		2377
		2378	api_ready (aTHX_ coro);
		2379	SvREFCNT_dec (coro);
		2380	SvREFCNT_dec ((AV *)state);
		2381	}
		2382
		2383	static int
		2384	slf_check_aio_req (pTHX_ struct CoroSLF *frame)
		2385	{
		2386	AV *state = (AV *)frame->data;
		2387
		2388	/* if we are about to throw, return early */
		2389	/* this does not cancel the aio request, but at least */
		2390	/* it quickly returns */
		2391	if (CORO_THROW)
		2392	return 0;
		2393
		2394	/* one element that is an RV? repeat! */
		2395	if (AvFILLp (state) == 0 && SvROK (AvARRAY (state)[0]))
		2396	return 1;
		2397
		2398	/* restore status */
		2399	{
		2400	SV *data_sv = av_pop (state);
		2401	struct io_state *data = (struct io_state *)SvPVX (data_sv);
		2402
		2403	errno = data->errorno;
		2404	PL_laststype = data->laststype;
		2405	PL_laststatval = data->laststatval;
		2406	PL_statcache = data->statcache;
		2407
		2408	SvREFCNT_dec (data_sv);
		2409	}
		2410
		2411	/* push result values */
		2412	{
		2413	dSP;
		2414	int i;
		2415
		2416	EXTEND (SP, AvFILLp (state) + 1);
		2417	for (i = 0; i <= AvFILLp (state); ++i)
		2418	PUSHs (sv_2mortal (SvREFCNT_inc_NN (AvARRAY (state)[i])));
		2419
		2420	PUTBACK;
		2421	}
		2422
		2423	return 0;
		2424	}
		2425
		2426	static void
		2427	slf_init_aio_req (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		2428	{
		2429	AV *state = (AV *)sv_2mortal ((SV *)newAV ());
		2430	SV *coro_hv = SvRV (coro_current);
		2431	struct coro *coro = SvSTATE_hv (coro_hv);
		2432
		2433	/* put our coroutine id on the state arg */
		2434	av_push (state, SvREFCNT_inc_NN (coro_hv));
		2435
		2436	/* first see whether we have a non-zero priority and set it as AIO prio */
		2437	if (coro->prio)
		2438	{
		2439	dSP;
		2440
		2441	static SV *prio_cv;
		2442	static SV *prio_sv;
		2443
		2444	if (expect_false (!prio_cv))
		2445	{
		2446	prio_cv = (SV *)get_cv ("IO::AIO::aioreq_pri", 0);
		2447	prio_sv = newSViv (0);
		2448	}
		2449
		2450	PUSHMARK (SP);
		2451	sv_setiv (prio_sv, coro->prio);
		2452	XPUSHs (prio_sv);
		2453
		2454	PUTBACK;
		2455	call_sv (prio_cv, G_VOID | G_DISCARD);
		2456	}
		2457
		2458	/* now call the original request */
		2459	{
		2460	dSP;
		2461	CV *req = (CV *)CORO_MAGIC_NN ((SV *)cv, CORO_MAGIC_type_aio)->mg_obj;
		2462	int i;
		2463
		2464	PUSHMARK (SP);
		2465
		2466	/* first push all args to the stack */
		2467	EXTEND (SP, items + 1);
		2468
		2469	for (i = 0; i < items; ++i)
		2470	PUSHs (arg [i]);
		2471
		2472	/* now push the callback closure */
		2473	PUSHs (sv_2mortal (gensub (aTHX_ coro_aio_callback, (void *)SvREFCNT_inc_NN ((SV *)state))));
		2474
		2475	/* now call the AIO function - we assume our request is uncancelable */
		2476	PUTBACK;
		2477	call_sv ((SV *)req, G_VOID | G_DISCARD);
		2478	}
		2479
		2480	/* now that the requets is going, we loop toll we have a result */
		2481	frame->data = (void *)state;
		2482	frame->prepare = prepare_schedule;
		2483	frame->check = slf_check_aio_req;
		2484	}
		2485
		2486	static void
		2487	coro_aio_req_xs (pTHX_ CV *cv)
		2488	{
		2489	dXSARGS;
		2490
		2491	CORO_EXECUTE_SLF_XS (slf_init_aio_req);
		2492
		2493	XSRETURN_EMPTY;
2125	}	2494	}
2126		2495
2127	/*****************************************************************************/	2496	/*****************************************************************************/
2128		2497
2129	MODULE = Coro::State PACKAGE = Coro::State PREFIX = api_	2498	MODULE = Coro::State PACKAGE = Coro::State PREFIX = api_
…		…
2225	for (i = 1; i < items; i++)	2594	for (i = 1; i < items; i++)
2226	av_push (coro->args, newSVsv (ST (i)));	2595	av_push (coro->args, newSVsv (ST (i)));
2227	}	2596	}
2228	OUTPUT:	2597	OUTPUT:
2229	RETVAL	2598	RETVAL
2230
2231	void
2232	_set_stacklevel (...)
2233	CODE:
2234	CORO_EXECUTE_SLF_XS (slf_init_set_stacklevel);
2235		2599
2236	void	2600	void
2237	transfer (...)	2601	transfer (...)
2238	PROTOTYPE: $$	2602	PROTOTYPE: $$
2239	CODE:	2603	CODE:
…		…
2365	throw (Coro::State self, SV *throw = &PL_sv_undef)	2729	throw (Coro::State self, SV *throw = &PL_sv_undef)
2366	PROTOTYPE: $;$	2730	PROTOTYPE: $;$
2367	CODE:	2731	CODE:
2368	{	2732	{
2369	struct coro *current = SvSTATE_current;	2733	struct coro *current = SvSTATE_current;
2370	SV **throwp = self == current ? &coro_throw : &self->throw;	2734	SV **throwp = self == current ? &CORO_THROW : &self->except;
2371	SvREFCNT_dec (*throwp);	2735	SvREFCNT_dec (*throwp);
2372	*throwp = SvOK (throw) ? newSVsv (throw) : 0;	2736	*throwp = SvOK (throw) ? newSVsv (throw) : 0;
2373	}	2737	}
2374		2738
2375	void	2739	void
…		…
2427	SV **dst = ix ? (SV **)&self->slot->defav : (SV **)&self->slot->defsv;	2791	SV **dst = ix ? (SV **)&self->slot->defav : (SV **)&self->slot->defsv;
2428		2792
2429	SV *tmp = *src; *src = *dst; *dst = tmp;	2793	SV *tmp = *src; *src = *dst; *dst = tmp;
2430	}	2794	}
2431		2795
		2796
2432	MODULE = Coro::State PACKAGE = Coro	2797	MODULE = Coro::State PACKAGE = Coro
2433		2798
2434	BOOT:	2799	BOOT:
2435	{	2800	{
2436	int i;	2801	int i;
…		…
2463	coroapi.ready = api_ready;	2828	coroapi.ready = api_ready;
2464	coroapi.is_ready = api_is_ready;	2829	coroapi.is_ready = api_is_ready;
2465	coroapi.nready = coro_nready;	2830	coroapi.nready = coro_nready;
2466	coroapi.current = coro_current;	2831	coroapi.current = coro_current;
2467		2832
2468	GCoroAPI = &coroapi;	2833	/*GCoroAPI = &coroapi;*/
2469	sv_setiv (sv, (IV)&coroapi);	2834	sv_setiv (sv, (IV)&coroapi);
2470	SvREADONLY_on (sv);	2835	SvREADONLY_on (sv);
2471	}	2836	}
2472	}	2837	}
2473		2838
…		…
2609	api_trace (aTHX_ coro_current, 0);	2974	api_trace (aTHX_ coro_current, 0);
2610		2975
2611	av_push (av_async_pool, newSVsv (coro_current));	2976	av_push (av_async_pool, newSVsv (coro_current));
2612	}	2977	}
2613		2978
2614		2979	SV *
2615	MODULE = Coro::State PACKAGE = Coro::AIO	2980	rouse_cb ()
		2981	PROTOTYPE:
		2982	CODE:
		2983	RETVAL = coro_new_rouse_cb (aTHX);
		2984	OUTPUT:
		2985	RETVAL
2616		2986
2617	void	2987	void
2618	_get_state (SV *self)	2988	rouse_wait (SV *cb = 0)
2619	PROTOTYPE: $	2989	PROTOTYPE: ;$
2620	PPCODE:	2990	PPCODE:
2621	{	2991	CORO_EXECUTE_SLF_XS (slf_init_rouse_wait);
2622	AV *defav = GvAV (PL_defgv);
2623	AV *av = newAV ();
2624	int i;
2625	SV *data_sv = newSV (sizeof (struct io_state));
2626	struct io_state *data = (struct io_state *)SvPVX (data_sv);
2627	SvCUR_set (data_sv, sizeof (struct io_state));
2628	SvPOK_only (data_sv);
2629
2630	data->errorno = errno;
2631	data->laststype = PL_laststype;
2632	data->laststatval = PL_laststatval;
2633	data->statcache = PL_statcache;
2634
2635	av_extend (av, AvFILLp (defav) + 1 + 1);
2636
2637	for (i = 0; i <= AvFILLp (defav); ++i)
2638	av_push (av, SvREFCNT_inc_NN (AvARRAY (defav)[i]));
2639
2640	av_push (av, data_sv);
2641
2642	XPUSHs (sv_2mortal (newRV_noinc ((SV *)av)));
2643
2644	api_ready (aTHX_ self);
2645	}
2646
2647	void
2648	_set_state (SV *state)
2649	PROTOTYPE: $
2650	PPCODE:
2651	{
2652	AV *av = (AV *)SvRV (state);
2653	struct io_state *data = (struct io_state *)SvPVX (AvARRAY (av)[AvFILLp (av)]);
2654	int i;
2655
2656	errno = data->errorno;
2657	PL_laststype = data->laststype;
2658	PL_laststatval = data->laststatval;
2659	PL_statcache = data->statcache;
2660
2661	EXTEND (SP, AvFILLp (av));
2662	for (i = 0; i < AvFILLp (av); ++i)
2663	PUSHs (sv_2mortal (SvREFCNT_inc_NN (AvARRAY (av)[i])));
2664	}
2665
2666
2667	MODULE = Coro::State PACKAGE = Coro::AnyEvent
2668
2669	BOOT:
2670	sv_activity = coro_get_sv (aTHX_ "Coro::AnyEvent::ACTIVITY", TRUE);
2671
2672	void
2673	_schedule (...)
2674	CODE:
2675	{
2676	static int incede;
2677
2678	api_cede_notself (aTHX);
2679
2680	++incede;
2681	while (coro_nready >= incede && api_cede (aTHX))
2682	;
2683
2684	sv_setsv (sv_activity, &PL_sv_undef);
2685	if (coro_nready >= incede)
2686	{
2687	PUSHMARK (SP);
2688	PUTBACK;
2689	call_pv ("Coro::AnyEvent::_activity", G_DISCARD | G_EVAL);
2690	SPAGAIN;
2691	}
2692
2693	--incede;
2694	}
2695		2992
2696		2993
2697	MODULE = Coro::State PACKAGE = PerlIO::cede	2994	MODULE = Coro::State PACKAGE = PerlIO::cede
2698		2995
2699	BOOT:	2996	BOOT:
2700	PerlIO_define_layer (aTHX_ &PerlIO_cede);	2997	PerlIO_define_layer (aTHX_ &PerlIO_cede);
2701		2998
		2999
2702	MODULE = Coro::State PACKAGE = Coro::Semaphore	3000	MODULE = Coro::State PACKAGE = Coro::Semaphore
2703		3001
2704	SV *	3002	SV *
2705	new (SV *klass, SV *count_ = 0)	3003	new (SV *klass, SV *count = 0)
2706	CODE:	3004	CODE:
2707	{	3005	RETVAL = sv_bless (
2708	/* a semaphore contains a counter IV in $sem->[0] and any waiters after that */	3006	coro_waitarray_new (aTHX_ count && SvOK (count) ? SvIV (count) : 1),
2709	AV *av = newAV ();	3007	GvSTASH (CvGV (cv))
2710	av_push (av, newSViv (count_ && SvOK (count_) ? SvIV (count_) : 1));	3008	);
2711	RETVAL = sv_bless (newRV_noinc ((SV *)av), GvSTASH (CvGV (cv)));	3009	OUTPUT:
2712	}	3010	RETVAL
		3011
		3012	# helper for Coro::Channel
		3013	SV *
		3014	_alloc (int count)
		3015	CODE:
		3016	RETVAL = coro_waitarray_new (aTHX_ count);
2713	OUTPUT:	3017	OUTPUT:
2714	RETVAL	3018	RETVAL
2715		3019
2716	SV *	3020	SV *
2717	count (SV *self)	3021	count (SV *self)
…		…
2729		3033
2730	void	3034	void
2731	down (SV *self)	3035	down (SV *self)
2732	CODE:	3036	CODE:
2733	CORO_EXECUTE_SLF_XS (slf_init_semaphore_down);	3037	CORO_EXECUTE_SLF_XS (slf_init_semaphore_down);
		3038
		3039	void
		3040	wait (SV *self)
		3041	CODE:
		3042	CORO_EXECUTE_SLF_XS (slf_init_semaphore_wait);
2734		3043
2735	void	3044	void
2736	try (SV *self)	3045	try (SV *self)
2737	PPCODE:	3046	PPCODE:
2738	{	3047	{
…		…
2750	XSRETURN_NO;	3059	XSRETURN_NO;
2751	}	3060	}
2752		3061
2753	void	3062	void
2754	waiters (SV *self)	3063	waiters (SV *self)
2755	CODE:	3064	PPCODE:
2756	{	3065	{
2757	AV *av = (AV *)SvRV (self);	3066	AV *av = (AV *)SvRV (self);
		3067	int wcount = AvFILLp (av) + 1 - 1;
2758		3068
2759	if (GIMME_V == G_SCALAR)	3069	if (GIMME_V == G_SCALAR)
2760	XPUSHs (sv_2mortal (newSVsv (AvARRAY (av)[0])));	3070	XPUSHs (sv_2mortal (newSViv (wcount)));
2761	else	3071	else
2762	{	3072	{
2763	int i;	3073	int i;
2764	EXTEND (SP, AvFILLp (av) + 1 - 1);	3074	EXTEND (SP, wcount);
2765	for (i = 1; i <= AvFILLp (av); ++i)	3075	for (i = 1; i <= wcount; ++i)
2766	PUSHs (newSVsv (AvARRAY (av)[i]));	3076	PUSHs (sv_2mortal (newRV_inc (AvARRAY (av)[i])));
2767	}	3077	}
2768	}	3078	}
2769		3079
		3080	MODULE = Coro::State PACKAGE = Coro::Signal
		3081
		3082	SV *
		3083	new (SV *klass)
		3084	CODE:
		3085	RETVAL = sv_bless (
		3086	coro_waitarray_new (aTHX_ 0),
		3087	GvSTASH (CvGV (cv))
		3088	);
		3089	OUTPUT:
		3090	RETVAL
		3091
		3092	void
		3093	wait (SV *self)
		3094	CODE:
		3095	CORO_EXECUTE_SLF_XS (slf_init_signal_wait);
		3096
		3097	void
		3098	broadcast (SV *self)
		3099	CODE:
		3100	{
		3101	AV *av = (AV *)SvRV (self);
		3102	coro_signal_wake (aTHX_ av, AvFILLp (av));
		3103	}
		3104
		3105	void
		3106	send (SV *self)
		3107	CODE:
		3108	{
		3109	AV *av = (AV *)SvRV (self);
		3110
		3111	if (AvFILLp (av))
		3112	coro_signal_wake (aTHX_ av, 1);
		3113	else
		3114	SvIVX (AvARRAY (av)[0]) = 1; /* remember the signal */
		3115	}
		3116
		3117	IV
		3118	awaited (SV *self)
		3119	CODE:
		3120	RETVAL = AvFILLp ((AV *)SvRV (self)) + 1 - 1;
		3121	OUTPUT:
		3122	RETVAL
		3123
		3124
		3125	MODULE = Coro::State PACKAGE = Coro::AnyEvent
		3126
		3127	BOOT:
		3128	sv_activity = coro_get_sv (aTHX_ "Coro::AnyEvent::ACTIVITY", TRUE);
		3129
		3130	void
		3131	_schedule (...)
		3132	CODE:
		3133	{
		3134	static int incede;
		3135
		3136	api_cede_notself (aTHX);
		3137
		3138	++incede;
		3139	while (coro_nready >= incede && api_cede (aTHX))
		3140	;
		3141
		3142	sv_setsv (sv_activity, &PL_sv_undef);
		3143	if (coro_nready >= incede)
		3144	{
		3145	PUSHMARK (SP);
		3146	PUTBACK;
		3147	call_pv ("Coro::AnyEvent::_activity", G_KEEPERR | G_EVAL | G_VOID | G_DISCARD);
		3148	}
		3149
		3150	--incede;
		3151	}
		3152
		3153
		3154	MODULE = Coro::State PACKAGE = Coro::AIO
		3155
		3156	void
		3157	_register (char *target, char *proto, SV *req)
		3158	CODE:
		3159	{
		3160	HV *st;
		3161	GV *gvp;
		3162	CV *req_cv = sv_2cv (req, &st, &gvp, 0);
		3163	/* newXSproto doesn't return the CV on 5.8 */
		3164	CV *slf_cv = newXS (target, coro_aio_req_xs, __FILE__);
		3165	sv_setpv ((SV *)slf_cv, proto);
		3166	sv_magicext ((SV *)slf_cv, (SV *)req_cv, CORO_MAGIC_type_aio, 0, 0, 0);
		3167	}
		3168

Diff Legend

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