[ViewVC] Diff of: cvs/Coro/Coro/State.xs

Comparing Coro/Coro/State.xs (file contents):
Revision 1.294 by root, Tue Nov 18 08:31:03 2008 UTC vs.
Revision 1.317 by root, Thu Nov 20 06:28:52 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
…		…
55		55
56	#define PERL_VERSION_ATLEAST(a,b,c) \	56	#define PERL_VERSION_ATLEAST(a,b,c) \
57	(PERL_REVISION > (a) \	57	(PERL_REVISION > (a) \
58	\|\| (PERL_REVISION == (a) \	58	\|\| (PERL_REVISION == (a) \
59	&& (PERL_VERSION > (b) \	59	&& (PERL_VERSION > (b) \
60	\|\| (PERL_VERSION == (b) && PERLSUBVERSION >= (c)))))	60	\|\| (PERL_VERSION == (b) && PERL_SUBVERSION >= (c)))))
61		61
62	#if !PERL_VERSION_ATLEAST (5,6,0)	62	#if !PERL_VERSION_ATLEAST (5,6,0)
63	# ifndef PL_ppaddr	63	# ifndef PL_ppaddr
64	# define PL_ppaddr ppaddr	64	# define PL_ppaddr ppaddr
65	# endif	65	# endif
…		…
98	# define newSV(l) NEWSV(0,l)	98	# define newSV(l) NEWSV(0,l)
99	#endif	99	#endif
100	#ifndef CvISXSUB_on	100	#ifndef CvISXSUB_on
101	# define CvISXSUB_on(cv) (void)cv	101	# define CvISXSUB_on(cv) (void)cv
102	#endif	102	#endif
		103	#ifndef CvISXSUB
		104	# define CvISXSUB(cv) (CvXSUB (cv) ? TRUE : FALSE)
		105	#endif
103		106
104	/* 5.8.7 */	107	/* 5.8.7 */
105	#ifndef SvRV_set	108	#ifndef SvRV_set
106	# define SvRV_set(s,v) SvRV(s) = (v)	109	# define SvRV_set(s,v) SvRV(s) = (v)
107	#endif	110	#endif
…		…
145	#define expect_true(expr) expect ((expr) != 0, 1)	148	#define expect_true(expr) expect ((expr) != 0, 1)
146		149
147	#define NOINLINE attribute ((noinline))	150	#define NOINLINE attribute ((noinline))
148		151
149	#include "CoroAPI.h"	152	#include "CoroAPI.h"
		153	#define GCoroAPI (&coroapi) /* very sneaky */
150		154
151	#ifdef USE_ITHREADS	155	#ifdef USE_ITHREADS
152	# if CORO_PTHREAD	156	# if CORO_PTHREAD
153	static void *coro_thx;	157	static void *coro_thx;
154	# endif	158	# endif
…		…
175	static HV hv_sig; / %SIG */	179	static HV hv_sig; / %SIG */
176		180
177	/* async_pool helper stuff */	181	/* async_pool helper stuff */
178	static SV *sv_pool_rss;	182	static SV *sv_pool_rss;
179	static SV *sv_pool_size;	183	static SV *sv_pool_size;
		184	static SV *sv_async_pool_idle;
180	static AV *av_async_pool;	185	static AV *av_async_pool;
		186	static SV *sv_Coro;
		187	static CV *cv_pool_handler;
		188	static CV *cv_coro_state_new;
181		189
182	/* Coro::AnyEvent */	190	/* Coro::AnyEvent */
183	static SV *sv_activity;	191	static SV *sv_activity;
184		192
185	static struct coro_cctx *cctx_first;	193	static struct coro_cctx *cctx_first;
…		…
245	/* state data */	253	/* state data */
246	struct CoroSLF slf_frame; /* saved slf frame */	254	struct CoroSLF slf_frame; /* saved slf frame */
247	AV *mainstack;	255	AV *mainstack;
248	perl_slots slot; / basically the saved sp */	256	perl_slots slot; / basically the saved sp */
249		257
		258	CV startcv; / the CV to execute */
250	AV args; / data associated with this coroutine (initial args) */	259	AV args; / data associated with this coroutine (initial args) */
251	int refcnt; /* coroutines are refcounted, yes */	260	int refcnt; /* coroutines are refcounted, yes */
252	int flags; /* CF_ flags */	261	int flags; /* CF_ flags */
253	HV hv; / the perl hash associated with this coro, if any */	262	HV hv; / the perl hash associated with this coro, if any */
254	void (on_destroy)(pTHX_ struct coro coro);	263	void (on_destroy)(pTHX_ struct coro coro);
255		264
256	/* statistics */	265	/* statistics */
257	int usecount; /* number of transfers to this coro */	266	int usecount; /* number of transfers to this coro */
258		267
259	/* coro process data */	268	/* coro process data */
260	int prio;	269	int prio;
261	SV throw; / exception to be thrown */	270	SV except; / exception to be thrown */
		271	SV *rouse_cb;
262		272
263	/* async_pool */	273	/* async_pool */
264	SV *saved_deffh;	274	SV *saved_deffh;
		275	SV *invoke_cb;
		276	AV *invoke_av;
265		277
266	/* linked list */	278	/* linked list */
267	struct coro next, prev;	279	struct coro next, prev;
268	};	280	};
269		281
…		…
272		284
273	/* the following variables are effectively part of the perl context */	285	/* the following variables are effectively part of the perl context */
274	/* and get copied between struct coro and these variables */	286	/* and get copied between struct coro and these variables */
275	/* the mainr easonw e don't support windows process emulation */	287	/* the mainr easonw e don't support windows process emulation */
276	static struct CoroSLF slf_frame; /* the current slf frame */	288	static struct CoroSLF slf_frame; /* the current slf frame */
277	static SV *coro_throw;
278		289
279	/ Coro ******************************************************************/	290	/ Coro ******************************************************************/
280		291
281	#define PRIO_MAX 3	292	#define PRIO_MAX 3
282	#define PRIO_HIGH 1	293	#define PRIO_HIGH 1
…		…
287		298
288	/* for Coro.pm */	299	/* for Coro.pm */
289	static SV *coro_current;	300	static SV *coro_current;
290	static SV *coro_readyhook;	301	static SV *coro_readyhook;
291	static AV *coro_ready [PRIO_MAX - PRIO_MIN + 1];	302	static AV *coro_ready [PRIO_MAX - PRIO_MIN + 1];
		303	static CV cv_coro_run, cv_coro_terminate;
292	static struct coro *coro_first;	304	static struct coro *coro_first;
293	#define coro_nready coroapi.nready	305	#define coro_nready coroapi.nready
294		306
295	/ lowlevel stuff ********************************************************/	307	/ lowlevel stuff ********************************************************/
296		308
…		…
320	#if PERL_VERSION_ATLEAST (5,10,0)	332	#if PERL_VERSION_ATLEAST (5,10,0)
321	/* silence stupid and wrong 5.10 warning that I am unable to switch off */	333	/* silence stupid and wrong 5.10 warning that I am unable to switch off */
322	get_hv (name, create);	334	get_hv (name, create);
323	#endif	335	#endif
324	return get_hv (name, create);	336	return get_hv (name, create);
		337	}
		338
		339	/* may croak */
		340	INLINE CV *
		341	coro_sv_2cv (pTHX_ SV *sv)
		342	{
		343	HV *st;
		344	GV *gvp;
		345	return sv_2cv (sv, &st, &gvp, 0);
325	}	346	}
326		347
327	static AV *	348	static AV *
328	coro_clone_padlist (pTHX_ CV *cv)	349	coro_clone_padlist (pTHX_ CV *cv)
329	{	350	{
…		…
448	else	469	else
449	{	470	{
450	#if CORO_PREFER_PERL_FUNCTIONS	471	#if CORO_PREFER_PERL_FUNCTIONS
451	/* this is probably cleaner? but also slower! */	472	/* this is probably cleaner? but also slower! */
452	/* in practise, it seems to be less stable */	473	/* in practise, it seems to be less stable */
453	CV *cp = Perl_cv_clone (cv);	474	CV *cp = Perl_cv_clone (aTHX_ cv);
454	CvPADLIST (cv) = CvPADLIST (cp);	475	CvPADLIST (cv) = CvPADLIST (cp);
455	CvPADLIST (cp) = 0;	476	CvPADLIST (cp) = 0;
456	SvREFCNT_dec (cp);	477	SvREFCNT_dec (cp);
457	#else	478	#else
458	CvPADLIST (cv) = coro_clone_padlist (aTHX_ cv);	479	CvPADLIST (cv) = coro_clone_padlist (aTHX_ cv);
…		…
511		532
512	PUTBACK;	533	PUTBACK;
513	}	534	}
514		535
515	slf_frame = c->slf_frame;	536	slf_frame = c->slf_frame;
516	coro_throw = c->throw;	537	CORO_THROW = c->except;
517	}	538	}
518		539
519	static void	540	static void
520	save_perl (pTHX_ Coro__State c)	541	save_perl (pTHX_ Coro__State c)
521	{	542	{
522	c->throw = coro_throw;	543	c->except = CORO_THROW;
523	c->slf_frame = slf_frame;	544	c->slf_frame = slf_frame;
524		545
525	{	546	{
526	dSP;	547	dSP;
527	I32 cxix = cxstack_ix;	548	I32 cxix = cxstack_ix;
…		…
602	* of perl.c:init_stacks, except that it uses less memory	623	* of perl.c:init_stacks, except that it uses less memory
603	* on the (sometimes correct) assumption that coroutines do	624	* on the (sometimes correct) assumption that coroutines do
604	* not usually need a lot of stackspace.	625	* not usually need a lot of stackspace.
605	*/	626	*/
606	#if CORO_PREFER_PERL_FUNCTIONS	627	#if CORO_PREFER_PERL_FUNCTIONS
607	# define coro_init_stacks init_stacks	628	# define coro_init_stacks(thx) init_stacks ()
608	#else	629	#else
609	static void	630	static void
610	coro_init_stacks (pTHX)	631	coro_init_stacks (pTHX)
611	{	632	{
612	PL_curstackinfo = new_stackinfo(32, 8);	633	PL_curstackinfo = new_stackinfo(32, 8);
…		…
675	#if !PERL_VERSION_ATLEAST (5,10,0)	696	#if !PERL_VERSION_ATLEAST (5,10,0)
676	Safefree (PL_retstack);	697	Safefree (PL_retstack);
677	#endif	698	#endif
678	}	699	}
679		700
		701	#define CORO_RSS \
		702	rss += sizeof (SYM (curstackinfo)); \
		703	rss += (SYM (curstackinfo->si_cxmax) + 1) * sizeof (PERL_CONTEXT); \
		704	rss += sizeof (SV) + sizeof (struct xpvav) + (1 + AvMAX (SYM (curstack))) * sizeof (SV *); \
		705	rss += SYM (tmps_max) * sizeof (SV *); \
		706	rss += (SYM (markstack_max) - SYM (markstack_ptr)) * sizeof (I32); \
		707	rss += SYM (scopestack_max) * sizeof (I32); \
		708	rss += SYM (savestack_max) * sizeof (ANY);
		709
680	static size_t	710	static size_t
681	coro_rss (pTHX_ struct coro *coro)	711	coro_rss (pTHX_ struct coro *coro)
682	{	712	{
683	size_t rss = sizeof (*coro);	713	size_t rss = sizeof (*coro);
684		714
685	if (coro->mainstack)	715	if (coro->mainstack)
686	{	716	{
687	perl_slots tmp_slot;
688	perl_slots *slot;
689
690	if (coro->flags & CF_RUNNING)	717	if (coro->flags & CF_RUNNING)
691	{	718	{
692	slot = &tmp_slot;	719	#define SYM(sym) PL_ ## sym
693		720	CORO_RSS;
694	#define VAR(name,type) slot->name = PL_ ## name;
695	# include "state.h"
696	#undef VAR	721	#undef SYM
697	}	722	}
698	else	723	else
699	slot = coro->slot;
700
701	if (slot)
702	{	724	{
703	rss += sizeof (slot->curstackinfo);	725	#define SYM(sym) coro->slot->sym
704	rss += (slot->curstackinfo->si_cxmax + 1) * sizeof (PERL_CONTEXT);	726	CORO_RSS;
705	rss += sizeof (SV) + sizeof (struct xpvav) + (1 + AvMAX (slot->curstack)) * sizeof (SV *);	727	#undef SYM
706	rss += slot->tmps_max * sizeof (SV *);
707	rss += (slot->markstack_max - slot->markstack_ptr) * sizeof (I32);
708	rss += slot->scopestack_max * sizeof (I32);
709	rss += slot->savestack_max * sizeof (ANY);
710
711	#if !PERL_VERSION_ATLEAST (5,10,0)
712	rss += slot->retstack_max * sizeof (OP *);
713	#endif
714	}	728	}
715	}	729	}
716		730
717	return rss;	731	return rss;
718	}	732	}
…		…
860	dSP;	874	dSP;
861	UNOP myop;	875	UNOP myop;
862		876
863	Zero (&myop, 1, UNOP);	877	Zero (&myop, 1, UNOP);
864	myop.op_next = Nullop;	878	myop.op_next = Nullop;
		879	myop.op_type = OP_ENTERSUB;
865	myop.op_flags = OPf_WANT_VOID;	880	myop.op_flags = OPf_WANT_VOID;
866		881
867	PUSHMARK (SP);	882	PUSHMARK (SP);
868	XPUSHs (sv_2mortal (av_shift (GvAV (PL_defgv))));	883	PUSHs ((SV *)coro->startcv);
869	PUTBACK;	884	PUTBACK;
870	PL_op = (OP *)&myop;	885	PL_op = (OP *)&myop;
871	PL_op = PL_ppaddr[OP_ENTERSUB](aTHX);	886	PL_op = PL_ppaddr[OP_ENTERSUB](aTHX);
872	SPAGAIN;
873	}	887	}
874		888
875	/* this newly created coroutine might be run on an existing cctx which most	889	/* this newly created coroutine might be run on an existing cctx which most
876	* likely was suspended in pp_slf, so we have to emulate entering pp_slf here.	890	* likely was suspended in pp_slf, so we have to emulate entering pp_slf here.
877	*/	891	*/
…		…
880		894
881	/* and we have to provide the pp_slf op in any case, so pp_slf can skip it */	895	/* 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;	896	coro_setup_op.op_next = PL_op;
883	coro_setup_op.op_type = OP_CUSTOM;	897	coro_setup_op.op_type = OP_CUSTOM;
884	coro_setup_op.op_ppaddr = pp_slf;	898	coro_setup_op.op_ppaddr = pp_slf;
885	/* no flags required, as an init function won't be called */	899	/* no flags etc. required, as an init function won't be called */
886		900
887	PL_op = (OP *)&coro_setup_op;	901	PL_op = (OP *)&coro_setup_op;
888		902
889	/* copy throw, in case it was set before coro_setup */	903	/* copy throw, in case it was set before coro_setup */
890	coro_throw = coro->throw;	904	CORO_THROW = coro->except;
891	}	905	}
892		906
893	static void	907	static void
894	coro_destruct (pTHX_ struct coro *coro)	908	coro_destruct (pTHX_ struct coro *coro)
895	{	909	{
…		…
919		933
920	SvREFCNT_dec (PL_diehook);	934	SvREFCNT_dec (PL_diehook);
921	SvREFCNT_dec (PL_warnhook);	935	SvREFCNT_dec (PL_warnhook);
922		936
923	SvREFCNT_dec (coro->saved_deffh);	937	SvREFCNT_dec (coro->saved_deffh);
924	SvREFCNT_dec (coro_throw);	938	SvREFCNT_dec (coro->rouse_cb);
		939	SvREFCNT_dec (coro->invoke_cb);
		940	SvREFCNT_dec (coro->invoke_av);
925		941
926	coro_destruct_stacks (aTHX);	942	coro_destruct_stacks (aTHX);
927	}	943	}
928		944
929	INLINE void	945	INLINE void
…		…
1016	SAVETMPS;	1032	SAVETMPS;
1017	EXTEND (SP, 3);	1033	EXTEND (SP, 3);
1018	PUSHMARK (SP);	1034	PUSHMARK (SP);
1019	PUSHs (&PL_sv_yes);	1035	PUSHs (&PL_sv_yes);
1020	PUSHs (fullname);	1036	PUSHs (fullname);
1021	PUSHs (CxHASARGS (cx) ? sv_2mortal (newRV_inc ((SV *)cx->blk_sub.argarray)) : &PL_sv_undef);	1037	PUSHs (CxHASARGS (cx) ? sv_2mortal (newRV_inc ((SV *)cx->blk_sub.argarray)) : &PL_sv_undef);
1022	PUTBACK;	1038	PUTBACK;
1023	cb = hv_fetch ((HV *)SvRV (coro_current), "_trace_sub_cb", sizeof ("_trace_sub_cb") - 1, 0);	1039	cb = hv_fetch ((HV *)SvRV (coro_current), "_trace_sub_cb", sizeof ("_trace_sub_cb") - 1, 0);
1024	if (cb) call_sv (*cb, G_KEEPERR \| G_EVAL \| G_VOID \| G_DISCARD);	1040	if (cb) call_sv (*cb, G_KEEPERR \| G_EVAL \| G_VOID \| G_DISCARD);
1025	SPAGAIN;	1041	SPAGAIN;
1026	FREETMPS;	1042	FREETMPS;
…		…
1132	transfer_tail (aTHX);	1148	transfer_tail (aTHX);
1133		1149
1134	/* somebody or something will hit me for both perl_run and PL_restartop */	1150	/* somebody or something will hit me for both perl_run and PL_restartop */
1135	PL_restartop = PL_op;	1151	PL_restartop = PL_op;
1136	perl_run (PL_curinterp);	1152	perl_run (PL_curinterp);
		1153	/*
		1154	* Unfortunately, there is no way to get at the return values of the
		1155	* coro body here, as perl_run destroys these
		1156	*/
1137		1157
1138	/*	1158	/*
1139	* If perl-run returns we assume exit() was being called or the coro	1159	* If perl-run returns we assume exit() was being called or the coro
1140	* fell off the end, which seems to be the only valid (non-bug)	1160	* fell off the end, which seems to be the only valid (non-bug)
1141	* reason for perl_run to return. We try to exit by jumping to the	1161	* reason for perl_run to return. We try to exit by jumping to the
…		…
1433		1453
1434	coro->slot = 0;	1454	coro->slot = 0;
1435	}	1455	}
1436		1456
1437	cctx_destroy (coro->cctx);	1457	cctx_destroy (coro->cctx);
		1458	SvREFCNT_dec (coro->startcv);
1438	SvREFCNT_dec (coro->args);	1459	SvREFCNT_dec (coro->args);
		1460	SvREFCNT_dec (CORO_THROW);
1439		1461
1440	if (coro->next) coro->next->prev = coro->prev;	1462	if (coro->next) coro->next->prev = coro->prev;
1441	if (coro->prev) coro->prev->next = coro->next;	1463	if (coro->prev) coro->prev->next = coro->next;
1442	if (coro == coro_first) coro_first = coro->next;	1464	if (coro == coro_first) coro_first = coro->next;
1443		1465
…		…
1497		1519
1498	prepare_transfer (aTHX_ &ta, prev_sv, next_sv);	1520	prepare_transfer (aTHX_ &ta, prev_sv, next_sv);
1499	TRANSFER (ta, 1);	1521	TRANSFER (ta, 1);
1500	}	1522	}
1501		1523
		1524	/*****************************************************************************/
		1525	/* gensub: simple closure generation utility */
		1526
		1527	#define GENSUB_ARG CvXSUBANY (cv).any_ptr
		1528
		1529	/* create a closure from XS, returns a code reference */
		1530	/* the arg can be accessed via GENSUB_ARG from the callback */
		1531	/* the callback must use dXSARGS/XSRETURN */
		1532	static SV *
		1533	gensub (pTHX_ void (xsub)(pTHX_ CV ), void *arg)
		1534	{
		1535	CV cv = (CV )newSV (0);
		1536
		1537	sv_upgrade ((SV *)cv, SVt_PVCV);
		1538
		1539	CvANON_on (cv);
		1540	CvISXSUB_on (cv);
		1541	CvXSUB (cv) = xsub;
		1542	GENSUB_ARG = arg;
		1543
		1544	return newRV_noinc ((SV *)cv);
		1545	}
		1546
1502	/ Coro ******************************************************************/	1547	/ Coro ******************************************************************/
1503		1548
1504	INLINE void	1549	INLINE void
1505	coro_enq (pTHX_ struct coro *coro)	1550	coro_enq (pTHX_ struct coro *coro)
1506	{	1551	{
…		…
1567	api_is_ready (pTHX_ SV *coro_sv)	1612	api_is_ready (pTHX_ SV *coro_sv)
1568	{	1613	{
1569	return !!(SvSTATE (coro_sv)->flags & CF_READY);	1614	return !!(SvSTATE (coro_sv)->flags & CF_READY);
1570	}	1615	}
1571		1616
		1617	/* expects to own a reference to next->hv */
1572	INLINE void	1618	INLINE void
		1619	prepare_schedule_to (pTHX_ struct coro_transfer_args ta, struct coro next)
		1620	{
		1621	SV *prev_sv = SvRV (coro_current);
		1622
		1623	ta->prev = SvSTATE_hv (prev_sv);
		1624	ta->next = next;
		1625
		1626	TRANSFER_CHECK (*ta);
		1627
		1628	SvRV_set (coro_current, (SV *)next->hv);
		1629
		1630	free_coro_mortal (aTHX);
		1631	coro_mortal = prev_sv;
		1632	}
		1633
		1634	static void
1573	prepare_schedule (pTHX_ struct coro_transfer_args *ta)	1635	prepare_schedule (pTHX_ struct coro_transfer_args *ta)
1574	{	1636	{
1575	SV prev_sv, next_sv;
1576
1577	for (;;)	1637	for (;;)
1578	{	1638	{
1579	next_sv = coro_deq (aTHX);	1639	SV *next_sv = coro_deq (aTHX);
1580		1640
1581	/* nothing to schedule: call the idle handler */
1582	if (expect_false (!next_sv))	1641	if (expect_true (next_sv))
1583	{	1642	{
		1643	struct coro *next = SvSTATE_hv (next_sv);
		1644
		1645	/* cannot transfer to destroyed coros, skip and look for next */
		1646	if (expect_false (next->flags & CF_DESTROYED))
		1647	SvREFCNT_dec (next_sv); /* coro_nready has already been taken care of by destroy */
		1648	else
		1649	{
		1650	next->flags &= ~CF_READY;
		1651	--coro_nready;
		1652
		1653	return prepare_schedule_to (aTHX_ ta, next);
		1654	}
		1655	}
		1656	else
		1657	{
		1658	/* nothing to schedule: call the idle handler */
1584	dSP;	1659	dSP;
1585		1660
1586	ENTER;	1661	ENTER;
1587	SAVETMPS;	1662	SAVETMPS;
1588		1663
…		…
1590	PUTBACK;	1665	PUTBACK;
1591	call_sv (get_sv ("Coro::idle", FALSE), G_VOID \| G_DISCARD);	1666	call_sv (get_sv ("Coro::idle", FALSE), G_VOID \| G_DISCARD);
1592		1667
1593	FREETMPS;	1668	FREETMPS;
1594	LEAVE;	1669	LEAVE;
1595	continue;
1596	}	1670	}
1597
1598	ta->next = SvSTATE_hv (next_sv);
1599
1600	/* cannot transfer to destroyed coros, skip and look for next */
1601	if (expect_false (ta->next->flags & CF_DESTROYED))
1602	{
1603	SvREFCNT_dec (next_sv);
1604	/* coro_nready has already been taken care of by destroy */
1605	continue;
1606	}
1607
1608	--coro_nready;
1609	break;
1610	}	1671	}
1611
1612	/* free this only after the transfer */
1613	prev_sv = SvRV (coro_current);
1614	ta->prev = SvSTATE_hv (prev_sv);
1615	TRANSFER_CHECK (*ta);
1616	assert (("FATAL: next coroutine isn't marked as ready in Coro (please report)", ta->next->flags & CF_READY));
1617	ta->next->flags &= ~CF_READY;
1618	SvRV_set (coro_current, next_sv);
1619
1620	free_coro_mortal (aTHX);
1621	coro_mortal = prev_sv;
1622	}	1672	}
1623		1673
1624	INLINE void	1674	INLINE void
1625	prepare_cede (pTHX_ struct coro_transfer_args *ta)	1675	prepare_cede (pTHX_ struct coro_transfer_args *ta)
1626	{	1676	{
…		…
1647	{	1697	{
1648	struct coro_transfer_args ta;	1698	struct coro_transfer_args ta;
1649		1699
1650	prepare_schedule (aTHX_ &ta);	1700	prepare_schedule (aTHX_ &ta);
1651	TRANSFER (ta, 1);	1701	TRANSFER (ta, 1);
		1702	}
		1703
		1704	static void
		1705	api_schedule_to (pTHX_ SV *coro_sv)
		1706	{
		1707	struct coro_transfer_args ta;
		1708	struct coro *next = SvSTATE (coro_sv);
		1709
		1710	SvREFCNT_inc_NN (coro_sv);
		1711	prepare_schedule_to (aTHX_ &ta, next);
1652	}	1712	}
1653		1713
1654	static int	1714	static int
1655	api_cede (pTHX)	1715	api_cede (pTHX)
1656	{	1716	{
…		…
1703	if (coro->flags & CF_RUNNING)	1763	if (coro->flags & CF_RUNNING)
1704	PL_runops = RUNOPS_DEFAULT;	1764	PL_runops = RUNOPS_DEFAULT;
1705	else	1765	else
1706	coro->slot->runops = RUNOPS_DEFAULT;	1766	coro->slot->runops = RUNOPS_DEFAULT;
1707	}	1767	}
		1768	}
		1769
		1770	/*****************************************************************************/
		1771	/* async pool handler */
		1772
		1773	static int
		1774	slf_check_pool_handler (pTHX_ struct CoroSLF *frame)
		1775	{
		1776	HV hv = (HV )SvRV (coro_current);
		1777	struct coro coro = (struct coro )frame->data;
		1778
		1779	if (!coro->invoke_cb)
		1780	return 1; /* loop till we have invoke */
		1781	else
		1782	{
		1783	hv_store (hv, "desc", sizeof ("desc") - 1,
		1784	newSVpvn ("[async_pool]", sizeof ("[async_pool]") - 1), 0);
		1785
		1786	coro->saved_deffh = SvREFCNT_inc_NN ((SV *)PL_defoutgv);
		1787
		1788	{
		1789	dSP;
		1790	XPUSHs (sv_2mortal (coro->invoke_cb)); coro->invoke_cb = 0;
		1791	PUTBACK;
		1792	}
		1793
		1794	SvREFCNT_dec (GvAV (PL_defgv));
		1795	GvAV (PL_defgv) = coro->invoke_av;
		1796	coro->invoke_av = 0;
		1797
		1798	return 0;
		1799	}
		1800	}
		1801
		1802	static void
		1803	slf_init_pool_handler (pTHX_ struct CoroSLF frame, CV cv, SV **arg, int items)
		1804	{
		1805	HV hv = (HV )SvRV (coro_current);
		1806	struct coro coro = SvSTATE_hv ((SV )hv);
		1807
		1808	if (expect_true (coro->saved_deffh))
		1809	{
		1810	/* subsequent iteration */
		1811	SvREFCNT_dec ((SV )PL_defoutgv); PL_defoutgv = (GV )coro->saved_deffh;
		1812	coro->saved_deffh = 0;
		1813
		1814	if (coro_rss (aTHX_ coro) > SvUV (sv_pool_rss)
		1815	\|\| av_len (av_async_pool) + 1 >= SvIV (sv_pool_size))
		1816	{
		1817	coro->invoke_cb = SvREFCNT_inc_NN ((SV *)cv_coro_terminate);
		1818	coro->invoke_av = newAV ();
		1819
		1820	frame->prepare = prepare_nop;
		1821	}
		1822	else
		1823	{
		1824	av_clear (GvAV (PL_defgv));
		1825	hv_store (hv, "desc", sizeof ("desc") - 1, SvREFCNT_inc_NN (sv_async_pool_idle), 0);
		1826
		1827	coro->prio = 0;
		1828
		1829	if (coro->cctx && (coro->cctx->flags & CC_TRACE))
		1830	api_trace (aTHX_ coro_current, 0);
		1831
		1832	frame->prepare = prepare_schedule;
		1833	av_push (av_async_pool, SvREFCNT_inc (hv));
		1834	}
		1835	}
		1836	else
		1837	{
		1838	/* first iteration, simply fall through */
		1839	frame->prepare = prepare_nop;
		1840	}
		1841
		1842	frame->check = slf_check_pool_handler;
		1843	frame->data = (void *)coro;
		1844	}
		1845
		1846	/*****************************************************************************/
		1847	/* rouse callback */
		1848
		1849	#define CORO_MAGIC_type_rouse PERL_MAGIC_ext
		1850
		1851	static void
		1852	coro_rouse_callback (pTHX_ CV *cv)
		1853	{
		1854	dXSARGS;
		1855	SV data = (SV )GENSUB_ARG;
		1856
		1857	if (SvTYPE (SvRV (data)) != SVt_PVAV)
		1858	{
		1859	/* first call, set args */
		1860	AV *av = newAV ();
		1861	SV *coro = SvRV (data);
		1862
		1863	SvRV_set (data, (SV *)av);
		1864	api_ready (aTHX_ coro);
		1865	SvREFCNT_dec (coro);
		1866
		1867	/* better take a full copy of the arguments */
		1868	while (items--)
		1869	av_store (av, items, newSVsv (ST (items)));
		1870	}
		1871
		1872	XSRETURN_EMPTY;
		1873	}
		1874
		1875	static int
		1876	slf_check_rouse_wait (pTHX_ struct CoroSLF *frame)
		1877	{
		1878	SV data = (SV )frame->data;
		1879
		1880	if (CORO_THROW)
		1881	return 0;
		1882
		1883	if (SvTYPE (SvRV (data)) != SVt_PVAV)
		1884	return 1;
		1885
		1886	/* now push all results on the stack */
		1887	{
		1888	dSP;
		1889	AV av = (AV )SvRV (data);
		1890	int i;
		1891
		1892	EXTEND (SP, AvFILLp (av) + 1);
		1893	for (i = 0; i <= AvFILLp (av); ++i)
		1894	PUSHs (sv_2mortal (AvARRAY (av)[i]));
		1895
		1896	/* we have stolen the elements, so ste length to zero and free */
		1897	AvFILLp (av) = -1;
		1898	av_undef (av);
		1899
		1900	PUTBACK;
		1901	}
		1902
		1903	return 0;
		1904	}
		1905
		1906	static void
		1907	slf_init_rouse_wait (pTHX_ struct CoroSLF frame, CV cv, SV **arg, int items)
		1908	{
		1909	SV *cb;
		1910
		1911	if (items)
		1912	cb = arg [0];
		1913	else
		1914	{
		1915	struct coro *coro = SvSTATE_current;
		1916
		1917	if (!coro->rouse_cb)
		1918	croak ("Coro::rouse_wait called without rouse callback, and no default rouse callback found either,");
		1919
		1920	cb = sv_2mortal (coro->rouse_cb);
		1921	coro->rouse_cb = 0;
		1922	}
		1923
		1924	if (!SvROK (cb)
		1925	\|\| SvTYPE (SvRV (cb)) != SVt_PVCV
		1926	\|\| CvXSUB ((CV *)SvRV (cb)) != coro_rouse_callback)
		1927	croak ("Coro::rouse_wait called with illegal callback argument,");
		1928
		1929	{
		1930	CV cv = (CV )SvRV (cb); /* for GENSUB_ARG */
		1931	SV data = (SV )GENSUB_ARG;
		1932
		1933	frame->data = (void *)data;
		1934	frame->prepare = SvTYPE (SvRV (data)) == SVt_PVAV ? prepare_nop : prepare_schedule;
		1935	frame->check = slf_check_rouse_wait;
		1936	}
		1937	}
		1938
		1939	static SV *
		1940	coro_new_rouse_cb (pTHX)
		1941	{
		1942	HV hv = (HV )SvRV (coro_current);
		1943	struct coro *coro = SvSTATE_hv (hv);
		1944	SV data = newRV_inc ((SV )hv);
		1945	SV cb = gensub (aTHX_ coro_rouse_callback, (void )data);
		1946
		1947	sv_magicext (SvRV (cb), data, CORO_MAGIC_type_rouse, 0, 0, 0);
		1948	SvREFCNT_dec (data); /* magicext increases the refcount */
		1949
		1950	SvREFCNT_dec (coro->rouse_cb);
		1951	coro->rouse_cb = SvREFCNT_inc_NN (cb);
		1952
		1953	return cb;
1708	}	1954	}
1709		1955
1710	/*****************************************************************************/	1956	/*****************************************************************************/
1711	/* schedule-like-function opcode (SLF) */	1957	/* schedule-like-function opcode (SLF) */
1712		1958
…		…
1759	static void	2005	static void
1760	slf_init_schedule (pTHX_ struct CoroSLF frame, CV cv, SV **arg, int items)	2006	slf_init_schedule (pTHX_ struct CoroSLF frame, CV cv, SV **arg, int items)
1761	{	2007	{
1762	frame->prepare = prepare_schedule;	2008	frame->prepare = prepare_schedule;
1763	frame->check = slf_check_nop;	2009	frame->check = slf_check_nop;
		2010	}
		2011
		2012	static void
		2013	slf_prepare_schedule_to (pTHX_ struct coro_transfer_args *ta)
		2014	{
		2015	struct coro next = (struct coro )slf_frame.data;
		2016
		2017	SvREFCNT_inc_NN (next->hv);
		2018	prepare_schedule_to (aTHX_ ta, next);
		2019	}
		2020
		2021	static void
		2022	slf_init_schedule_to (pTHX_ struct CoroSLF frame, CV cv, SV **arg, int items)
		2023	{
		2024	if (!items)
		2025	croak ("Coro::schedule_to expects a coroutine argument, caught");
		2026
		2027	frame->data = (void *)SvSTATE (arg [0]);
		2028	frame->prepare = slf_prepare_schedule_to;
		2029	frame->check = slf_check_nop;
		2030	}
		2031
		2032	static void
		2033	slf_init_cede_to (pTHX_ struct CoroSLF frame, CV cv, SV **arg, int items)
		2034	{
		2035	api_ready (aTHX_ SvRV (coro_current));
		2036
		2037	slf_init_schedule_to (aTHX_ frame, cv, arg, items);
1764	}	2038	}
1765		2039
1766	static void	2040	static void
1767	slf_init_cede (pTHX_ struct CoroSLF frame, CV cv, SV **arg, int items)	2041	slf_init_cede (pTHX_ struct CoroSLF frame, CV cv, SV **arg, int items)
1768	{	2042	{
…		…
1839	}	2113	}
1840	while (slf_frame.check (aTHX_ &slf_frame));	2114	while (slf_frame.check (aTHX_ &slf_frame));
1841		2115
1842	slf_frame.prepare = 0; /* invalidate the frame, we are done processing it */	2116	slf_frame.prepare = 0; /* invalidate the frame, we are done processing it */
1843		2117
		2118	/* exception handling */
		2119	if (expect_false (CORO_THROW))
		2120	{
		2121	SV *exception = sv_2mortal (CORO_THROW);
		2122
		2123	CORO_THROW = 0;
		2124	sv_setsv (ERRSV, exception);
		2125	croak (0);
		2126	}
		2127
1844	/* return value handling - mostly like entersub */	2128	/* return value handling - mostly like entersub */
1845	/* make sure we put something on the stack in scalar context */	2129	/* make sure we put something on the stack in scalar context */
1846	if (GIMME_V == G_SCALAR)	2130	if (GIMME_V == G_SCALAR)
1847	{	2131	{
1848	dSP;	2132	dSP;
…		…
1854	bot [1] = *sp;	2138	bot [1] = *sp;
1855		2139
1856	SP = bot + 1;	2140	SP = bot + 1;
1857		2141
1858	PUTBACK;	2142	PUTBACK;
1859	}
1860
1861	/* exception handling */
1862	if (expect_false (coro_throw))
1863	{
1864	SV *exception = sv_2mortal (coro_throw);
1865
1866	coro_throw = 0;
1867	sv_setsv (ERRSV, exception);
1868	croak (0);
1869	}	2143	}
1870		2144
1871	return NORMAL;	2145	return NORMAL;
1872	}	2146	}
1873		2147
…		…
1995	PerlIOBuf_get_cnt,	2269	PerlIOBuf_get_cnt,
1996	PerlIOBuf_set_ptrcnt,	2270	PerlIOBuf_set_ptrcnt,
1997	};	2271	};
1998		2272
1999	/*****************************************************************************/	2273	/*****************************************************************************/
		2274	/* Coro::Semaphore & Coro::Signal */
		2275
		2276	static SV *
		2277	coro_waitarray_new (pTHX_ int count)
		2278	{
		2279	/* a semaphore contains a counter IV in $sem->[0] and any waiters after that */
		2280	AV *av = newAV ();
		2281	SV **ary;
		2282
		2283	/* unfortunately, building manually saves memory */
		2284	Newx (ary, 2, SV *);
		2285	AvALLOC (av) = ary;
		2286	/AvARRAY (av) = ary;/
		2287	SvPVX ((SV )av) = (char )ary; /* 5.8.8 needs this syntax instead of AvARRAY = ary */
		2288	AvMAX (av) = 1;
		2289	AvFILLp (av) = 0;
		2290	ary [0] = newSViv (count);
		2291
		2292	return newRV_noinc ((SV *)av);
		2293	}
		2294
2000	/* Coro::Semaphore */	2295	/* semaphore */
2001		2296
2002	static void	2297	static void
2003	coro_semaphore_adjust (pTHX_ AV *av, IV adjust)	2298	coro_semaphore_adjust (pTHX_ AV *av, IV adjust)
2004	{	2299	{
2005	SV *count_sv = AvARRAY (av)[0];	2300	SV *count_sv = AvARRAY (av)[0];
…		…
2017	AvARRAY (av)[0] = AvARRAY (av)[1];	2312	AvARRAY (av)[0] = AvARRAY (av)[1];
2018	AvARRAY (av)[1] = count_sv;	2313	AvARRAY (av)[1] = count_sv;
2019	cb = av_shift (av);	2314	cb = av_shift (av);
2020		2315
2021	if (SvOBJECT (cb))	2316	if (SvOBJECT (cb))
		2317	{
2022	api_ready (aTHX_ cb);	2318	api_ready (aTHX_ cb);
2023	else	2319	--count;
2024	croak ("callbacks not yet supported");	2320	}
		2321	else if (SvTYPE (cb) == SVt_PVCV)
		2322	{
		2323	dSP;
		2324	PUSHMARK (SP);
		2325	XPUSHs (sv_2mortal (newRV_inc ((SV *)av)));
		2326	PUTBACK;
		2327	call_sv (cb, G_VOID \| G_DISCARD \| G_EVAL \| G_KEEPERR);
		2328	}
2025		2329
2026	SvREFCNT_dec (cb);	2330	SvREFCNT_dec (cb);
2027
2028	--count;
2029	}	2331	}
2030	}	2332	}
2031		2333
2032	static void	2334	static void
2033	coro_semaphore_on_destroy (pTHX_ struct coro *coro)	2335	coro_semaphore_on_destroy (pTHX_ struct coro *coro)
…		…
2035	/* call $sem->adjust (0) to possibly wake up some other waiters */	2337	/* call $sem->adjust (0) to possibly wake up some other waiters */
2036	coro_semaphore_adjust (aTHX_ (AV *)coro->slf_frame.data, 0);	2338	coro_semaphore_adjust (aTHX_ (AV *)coro->slf_frame.data, 0);
2037	}	2339	}
2038		2340
2039	static int	2341	static int
2040	slf_check_semaphore_down (pTHX_ struct CoroSLF *frame)	2342	slf_check_semaphore_down_or_wait (pTHX_ struct CoroSLF *frame, int acquire)
2041	{	2343	{
2042	AV av = (AV )frame->data;	2344	AV av = (AV )frame->data;
2043	SV *count_sv = AvARRAY (av)[0];	2345	SV *count_sv = AvARRAY (av)[0];
2044		2346
		2347	/* if we are about to throw, don't actually acquire the lock, just throw */
		2348	if (CORO_THROW)
		2349	return 0;
2045	if (SvIVX (count_sv) > 0)	2350	else if (SvIVX (count_sv) > 0)
2046	{	2351	{
2047	SvSTATE_current->on_destroy = 0;	2352	SvSTATE_current->on_destroy = 0;
		2353
		2354	if (acquire)
2048	SvIVX (count_sv) = SvIVX (count_sv) - 1;	2355	SvIVX (count_sv) = SvIVX (count_sv) - 1;
		2356	else
		2357	coro_semaphore_adjust (aTHX_ av, 0);
		2358
2049	return 0;	2359	return 0;
2050	}	2360	}
2051	else	2361	else
2052	{	2362	{
2053	int i;	2363	int i;
…		…
2062	av_push (av, SvREFCNT_inc (SvRV (coro_current)));	2372	av_push (av, SvREFCNT_inc (SvRV (coro_current)));
2063	return 1;	2373	return 1;
2064	}	2374	}
2065	}	2375	}
2066		2376
2067	static void	2377	static int
		2378	slf_check_semaphore_down (pTHX_ struct CoroSLF *frame)
		2379	{
		2380	return slf_check_semaphore_down_or_wait (aTHX_ frame, 1);
		2381	}
		2382
		2383	static int
		2384	slf_check_semaphore_wait (pTHX_ struct CoroSLF *frame)
		2385	{
		2386	return slf_check_semaphore_down_or_wait (aTHX_ frame, 0);
		2387	}
		2388
		2389	static void
2068	slf_init_semaphore_down (pTHX_ struct CoroSLF frame, CV cv, SV **arg, int items)	2390	slf_init_semaphore_down_or_wait (pTHX_ struct CoroSLF frame, CV cv, SV **arg, int items)
2069	{	2391	{
2070	AV av = (AV )SvRV (arg [0]);	2392	AV av = (AV )SvRV (arg [0]);
2071		2393
2072	if (SvIVX (AvARRAY (av)[0]) > 0)	2394	if (SvIVX (AvARRAY (av)[0]) > 0)
2073	{	2395	{
2074	frame->data = (void *)av;	2396	frame->data = (void *)av;
2075	frame->prepare = prepare_nop;	2397	frame->prepare = prepare_nop;
2076	SvSTATE_current->on_destroy = coro_semaphore_on_destroy;
2077	}	2398	}
2078	else	2399	else
2079	{	2400	{
2080	av_push (av, SvREFCNT_inc (SvRV (coro_current)));	2401	av_push (av, SvREFCNT_inc (SvRV (coro_current)));
2081		2402
2082	frame->data = (void )sv_2mortal (SvREFCNT_inc ((SV )av));	2403	frame->data = (void )sv_2mortal (SvREFCNT_inc ((SV )av));
2083	frame->prepare = prepare_schedule;	2404	frame->prepare = prepare_schedule;
2084		2405
2085	/* to avoid race conditions when a woken-up coro gets terminated */	2406	/* to avoid race conditions when a woken-up coro gets terminated */
2086	/* we arrange for a temporary on_destroy that calls adjust (0) */	2407	/* we arrange for a temporary on_destroy that calls adjust (0) */
2087	assert (!SvSTATE_current->on_destroy);//D
2088	SvSTATE_current->on_destroy = coro_semaphore_on_destroy;	2408	SvSTATE_current->on_destroy = coro_semaphore_on_destroy;
2089	}	2409	}
		2410	}
2090		2411
		2412	static void
		2413	slf_init_semaphore_down (pTHX_ struct CoroSLF frame, CV cv, SV **arg, int items)
		2414	{
		2415	slf_init_semaphore_down_or_wait (aTHX_ frame, cv, arg, items);
2091	frame->check = slf_check_semaphore_down;	2416	frame->check = slf_check_semaphore_down;
2092
2093	}	2417	}
2094		2418
2095	/*****************************************************************************/	2419	static void
2096	/* gensub: simple closure generation utility */	2420	slf_init_semaphore_wait (pTHX_ struct CoroSLF frame, CV cv, SV **arg, int items)
2097
2098	#define GENSUB_ARG CvXSUBANY (cv).any_ptr
2099
2100	/* create a closure from XS, returns a code reference */
2101	/* the arg can be accessed via GENSUB_ARG from the callback */
2102	/* the callback must use dXSARGS/XSRETURN */
2103	static SV *
2104	gensub (pTHX_ void (xsub)(pTHX_ CV ), void *arg)
2105	{	2421	{
2106	CV cv = (CV )newSV (0);	2422	if (items >= 2)
		2423	{
		2424	/* callback form */
		2425	AV av = (AV )SvRV (arg [0]);
		2426	CV *cb_cv = coro_sv_2cv (aTHX_ arg [1]);
2107		2427
2108	sv_upgrade ((SV *)cv, SVt_PVCV);	2428	av_push (av, (SV *)SvREFCNT_inc_NN (cb_cv));
2109		2429
2110	CvANON_on (cv);	2430	if (SvIVX (AvARRAY (av)[0]) > 0)
2111	CvISXSUB_on (cv);	2431	coro_semaphore_adjust (aTHX_ av, 0);
2112	CvXSUB (cv) = xsub;
2113	GENSUB_ARG = arg;
2114		2432
2115	return newRV_noinc ((SV *)cv);	2433	frame->prepare = prepare_nop;
		2434	frame->check = slf_check_nop;
		2435	}
		2436	else
		2437	{
		2438	slf_init_semaphore_down_or_wait (aTHX_ frame, cv, arg, items);
		2439	frame->check = slf_check_semaphore_wait;
		2440	}
		2441	}
		2442
		2443	/* signal */
		2444
		2445	static void
		2446	coro_signal_wake (pTHX_ AV *av, int count)
		2447	{
		2448	SvIVX (AvARRAY (av)[0]) = 0;
		2449
		2450	/* now signal count waiters */
		2451	while (count > 0 && AvFILLp (av) > 0)
		2452	{
		2453	SV *cb;
		2454
		2455	/* swap first two elements so we can shift a waiter */
		2456	cb = AvARRAY (av)[0];
		2457	AvARRAY (av)[0] = AvARRAY (av)[1];
		2458	AvARRAY (av)[1] = cb;
		2459
		2460	cb = av_shift (av);
		2461
		2462	api_ready (aTHX_ cb);
		2463	sv_setiv (cb, 0); /* signal waiter */
		2464	SvREFCNT_dec (cb);
		2465
		2466	--count;
		2467	}
		2468	}
		2469
		2470	static int
		2471	slf_check_signal_wait (pTHX_ struct CoroSLF *frame)
		2472	{
		2473	/* if we are about to throw, also stop waiting */
		2474	return SvROK ((SV *)frame->data) && !CORO_THROW;
		2475	}
		2476
		2477	static void
		2478	slf_init_signal_wait (pTHX_ struct CoroSLF frame, CV cv, SV **arg, int items)
		2479	{
		2480	AV av = (AV )SvRV (arg [0]);
		2481
		2482	if (SvIVX (AvARRAY (av)[0]))
		2483	{
		2484	SvIVX (AvARRAY (av)[0]) = 0;
		2485	frame->prepare = prepare_nop;
		2486	frame->check = slf_check_nop;
		2487	}
		2488	else
		2489	{
		2490	SV waiter = newRV_inc (SvRV (coro_current)); / owned by signal av */
		2491
		2492	av_push (av, waiter);
		2493
		2494	frame->data = (void )sv_2mortal (SvREFCNT_inc_NN (waiter)); / owned by process */
		2495	frame->prepare = prepare_schedule;
		2496	frame->check = slf_check_signal_wait;
		2497	}
2116	}	2498	}
2117		2499
2118	/*****************************************************************************/	2500	/*****************************************************************************/
2119	/* Coro::AIO */	2501	/* Coro::AIO */
2120		2502
…		…
2135	dXSARGS;	2517	dXSARGS;
2136	AV state = (AV )GENSUB_ARG;	2518	AV state = (AV )GENSUB_ARG;
2137	SV *coro = av_pop (state);	2519	SV *coro = av_pop (state);
2138	SV *data_sv = newSV (sizeof (struct io_state));	2520	SV *data_sv = newSV (sizeof (struct io_state));
2139		2521
2140	av_extend (state, items);	2522	av_extend (state, items - 1);
2141		2523
2142	sv_upgrade (data_sv, SVt_PV);	2524	sv_upgrade (data_sv, SVt_PV);
2143	SvCUR_set (data_sv, sizeof (struct io_state));	2525	SvCUR_set (data_sv, sizeof (struct io_state));
2144	SvPOK_only (data_sv);	2526	SvPOK_only (data_sv);
2145		2527
…		…
2169		2551
2170	static int	2552	static int
2171	slf_check_aio_req (pTHX_ struct CoroSLF *frame)	2553	slf_check_aio_req (pTHX_ struct CoroSLF *frame)
2172	{	2554	{
2173	AV state = (AV )frame->data;	2555	AV state = (AV )frame->data;
		2556
		2557	/* if we are about to throw, return early */
		2558	/* this does not cancel the aio request, but at least */
		2559	/* it quickly returns */
		2560	if (CORO_THROW)
		2561	return 0;
2174		2562
2175	/* one element that is an RV? repeat! */	2563	/* one element that is an RV? repeat! */
2176	if (AvFILLp (state) == 0 && SvROK (AvARRAY (state)[0]))	2564	if (AvFILLp (state) == 0 && SvROK (AvARRAY (state)[0]))
2177	return 1;	2565	return 1;
2178		2566
…		…
2349	assert (("PRIO_NORMAL must be 0", !PRIO_NORMAL));	2737	assert (("PRIO_NORMAL must be 0", !PRIO_NORMAL));
2350	}	2738	}
2351		2739
2352	SV *	2740	SV *
2353	new (char *klass, ...)	2741	new (char *klass, ...)
		2742	ALIAS:
		2743	Coro::new = 1
2354	CODE:	2744	CODE:
2355	{	2745	{
2356	struct coro *coro;	2746	struct coro *coro;
2357	MAGIC *mg;	2747	MAGIC *mg;
2358	HV *hv;	2748	HV *hv;
		2749	CV *cb;
2359	int i;	2750	int i;
		2751
		2752	if (items > 1)
		2753	{
		2754	cb = coro_sv_2cv (aTHX_ ST (1));
		2755
		2756	if (!ix)
		2757	{
		2758	if (CvISXSUB (cb))
		2759	croak ("Coro::State doesn't support XS functions as coroutine start, caught");
		2760
		2761	if (!CvROOT (cb))
		2762	croak ("Coro::State doesn't support autoloaded or undefined functions as coroutine start, caught");
		2763	}
		2764	}
2360		2765
2361	Newz (0, coro, 1, struct coro);	2766	Newz (0, coro, 1, struct coro);
2362	coro->args = newAV ();	2767	coro->args = newAV ();
2363	coro->flags = CF_NEW;	2768	coro->flags = CF_NEW;
2364		2769
…		…
2369	coro->hv = hv = newHV ();	2774	coro->hv = hv = newHV ();
2370	mg = sv_magicext ((SV )hv, 0, CORO_MAGIC_type_state, &coro_state_vtbl, (char )coro, 0);	2775	mg = sv_magicext ((SV )hv, 0, CORO_MAGIC_type_state, &coro_state_vtbl, (char )coro, 0);
2371	mg->mg_flags \|= MGf_DUP;	2776	mg->mg_flags \|= MGf_DUP;
2372	RETVAL = sv_bless (newRV_noinc ((SV *)hv), gv_stashpv (klass, 1));	2777	RETVAL = sv_bless (newRV_noinc ((SV *)hv), gv_stashpv (klass, 1));
2373		2778
		2779	if (items > 1)
		2780	{
2374	av_extend (coro->args, items - 1);	2781	av_extend (coro->args, items - 1 + ix - 1);
		2782
		2783	if (ix)
		2784	{
		2785	av_push (coro->args, SvREFCNT_inc_NN ((SV *)cb));
		2786	cb = cv_coro_run;
		2787	}
		2788
		2789	coro->startcv = (CV )SvREFCNT_inc_NN ((SV )cb);
		2790
2375	for (i = 1; i < items; i++)	2791	for (i = 2; i < items; i++)
2376	av_push (coro->args, newSVsv (ST (i)));	2792	av_push (coro->args, newSVsv (ST (i)));
		2793	}
2377	}	2794	}
2378	OUTPUT:	2795	OUTPUT:
2379	RETVAL	2796	RETVAL
2380		2797
2381	void	2798	void
…		…
2510	throw (Coro::State self, SV *throw = &PL_sv_undef)	2927	throw (Coro::State self, SV *throw = &PL_sv_undef)
2511	PROTOTYPE: $;$	2928	PROTOTYPE: $;$
2512	CODE:	2929	CODE:
2513	{	2930	{
2514	struct coro *current = SvSTATE_current;	2931	struct coro *current = SvSTATE_current;
2515	SV **throwp = self == current ? &coro_throw : &self->throw;	2932	SV **throwp = self == current ? &CORO_THROW : &self->except;
2516	SvREFCNT_dec (*throwp);	2933	SvREFCNT_dec (*throwp);
2517	*throwp = SvOK (throw) ? newSVsv (throw) : 0;	2934	*throwp = SvOK (throw) ? newSVsv (throw) : 0;
2518	}	2935	}
2519		2936
2520	void	2937	void
…		…
2579		2996
2580	BOOT:	2997	BOOT:
2581	{	2998	{
2582	int i;	2999	int i;
2583		3000
2584	av_async_pool = coro_get_av (aTHX_ "Coro::async_pool", TRUE);	3001	av_async_pool = coro_get_av (aTHX_ "Coro::async_pool", TRUE);
2585	sv_pool_rss = coro_get_sv (aTHX_ "Coro::POOL_RSS" , TRUE);	3002	sv_pool_rss = coro_get_sv (aTHX_ "Coro::POOL_RSS" , TRUE);
2586	sv_pool_size = coro_get_sv (aTHX_ "Coro::POOL_SIZE" , TRUE);	3003	sv_pool_size = coro_get_sv (aTHX_ "Coro::POOL_SIZE" , TRUE);
2587		3004	cv_coro_run = get_cv ( "Coro::_terminate", GV_ADD);
		3005	cv_coro_terminate = get_cv ( "Coro::terminate" , GV_ADD);
2588	coro_current = coro_get_sv (aTHX_ "Coro::current", FALSE);	3006	coro_current = coro_get_sv (aTHX_ "Coro::current" , FALSE); SvREADONLY_on (coro_current);
2589	SvREADONLY_on (coro_current);	3007
		3008	sv_async_pool_idle = newSVpv ("[async pool idle]", 0); SvREADONLY_on (sv_async_pool_idle);
		3009	sv_Coro = newSVpv ("Coro", 0); SvREADONLY_on (sv_Coro);
		3010	cv_pool_handler = get_cv ("Coro::pool_handler", GV_ADD); SvREADONLY_on (cv_pool_handler);
		3011	cv_coro_state_new = get_cv ("Coro::State::new", 0); SvREADONLY_on (cv_coro_state_new);
2590		3012
2591	coro_stash = gv_stashpv ("Coro", TRUE);	3013	coro_stash = gv_stashpv ("Coro", TRUE);
2592		3014
2593	newCONSTSUB (coro_stash, "PRIO_MAX", newSViv (PRIO_MAX));	3015	newCONSTSUB (coro_stash, "PRIO_MAX", newSViv (PRIO_MAX));
2594	newCONSTSUB (coro_stash, "PRIO_HIGH", newSViv (PRIO_HIGH));	3016	newCONSTSUB (coro_stash, "PRIO_HIGH", newSViv (PRIO_HIGH));
…		…
2602		3024
2603	{	3025	{
2604	SV *sv = coro_get_sv (aTHX_ "Coro::API", TRUE);	3026	SV *sv = coro_get_sv (aTHX_ "Coro::API", TRUE);
2605		3027
2606	coroapi.schedule = api_schedule;	3028	coroapi.schedule = api_schedule;
		3029	coroapi.schedule_to = api_schedule_to;
2607	coroapi.cede = api_cede;	3030	coroapi.cede = api_cede;
2608	coroapi.cede_notself = api_cede_notself;	3031	coroapi.cede_notself = api_cede_notself;
2609	coroapi.ready = api_ready;	3032	coroapi.ready = api_ready;
2610	coroapi.is_ready = api_is_ready;	3033	coroapi.is_ready = api_is_ready;
2611	coroapi.nready = coro_nready;	3034	coroapi.nready = coro_nready;
2612	coroapi.current = coro_current;	3035	coroapi.current = coro_current;
2613		3036
2614	GCoroAPI = &coroapi;	3037	/GCoroAPI = &coroapi;/
2615	sv_setiv (sv, (IV)&coroapi);	3038	sv_setiv (sv, (IV)&coroapi);
2616	SvREADONLY_on (sv);	3039	SvREADONLY_on (sv);
2617	}	3040	}
2618	}	3041	}
2619		3042
2620	void	3043	void
2621	schedule (...)	3044	schedule (...)
2622	CODE:	3045	CODE:
2623	CORO_EXECUTE_SLF_XS (slf_init_schedule);	3046	CORO_EXECUTE_SLF_XS (slf_init_schedule);
		3047
		3048	void
		3049	schedule_to (...)
		3050	CODE:
		3051	CORO_EXECUTE_SLF_XS (slf_init_schedule_to);
		3052
		3053	void
		3054	cede_to (...)
		3055	CODE:
		3056	CORO_EXECUTE_SLF_XS (slf_init_cede_to);
2624		3057
2625	void	3058	void
2626	cede (...)	3059	cede (...)
2627	CODE:	3060	CODE:
2628	CORO_EXECUTE_SLF_XS (slf_init_cede);	3061	CORO_EXECUTE_SLF_XS (slf_init_cede);
…		…
2683	CODE:	3116	CODE:
2684	RETVAL = coro_nready;	3117	RETVAL = coro_nready;
2685	OUTPUT:	3118	OUTPUT:
2686	RETVAL	3119	RETVAL
2687		3120
2688	# for async_pool speedup
2689	void	3121	void
2690	_pool_1 (SV *cb)	3122	_pool_handler (...)
2691	CODE:	3123	CODE:
2692	{	3124	CORO_EXECUTE_SLF_XS (slf_init_pool_handler);
2693	HV hv = (HV )SvRV (coro_current);
2694	struct coro coro = SvSTATE_hv ((SV )hv);
2695	AV *defav = GvAV (PL_defgv);
2696	SV *invoke = hv_delete (hv, "_invoke", sizeof ("_invoke") - 1, 0);
2697	AV *invoke_av;
2698	int i, len;
2699		3125
2700	if (!invoke)	3126	void
		3127	async_pool (SV *cv, ...)
		3128	PROTOTYPE: &@
		3129	PPCODE:
		3130	{
		3131	HV hv = (HV )av_pop (av_async_pool);
		3132	AV *av = newAV ();
		3133	SV *cb = ST (0);
		3134	int i;
		3135
		3136	av_extend (av, items - 2);
		3137	for (i = 1; i < items; ++i)
		3138	av_push (av, SvREFCNT_inc_NN (ST (i)));
		3139
		3140	if ((SV *)hv == &PL_sv_undef)
2701	{	3141	{
2702	SV *old = PL_diehook;	3142	PUSHMARK (SP);
2703	PL_diehook = 0;	3143	EXTEND (SP, 2);
2704	SvREFCNT_dec (old);	3144	PUSHs (sv_Coro);
2705	croak ("\3async_pool terminate\2\n");	3145	PUSHs ((SV *)cv_pool_handler);
		3146	PUTBACK;
		3147	call_sv ((SV *)cv_coro_state_new, G_SCALAR);
		3148	SPAGAIN;
		3149
		3150	hv = (HV *)SvREFCNT_inc_NN (SvRV (POPs));
2706	}	3151	}
2707		3152
2708	SvREFCNT_dec (coro->saved_deffh);
2709	coro->saved_deffh = SvREFCNT_inc_NN ((SV *)PL_defoutgv);
2710
2711	hv_store (hv, "desc", sizeof ("desc") - 1,
2712	newSVpvn ("[async_pool]", sizeof ("[async_pool]") - 1), 0);
2713
2714	invoke_av = (AV *)SvRV (invoke);
2715	len = av_len (invoke_av);
2716
2717	sv_setsv (cb, AvARRAY (invoke_av)[0]);
2718
2719	if (len > 0)
2720	{	3153	{
2721	av_fill (defav, len - 1);	3154	struct coro *coro = SvSTATE_hv (hv);
2722	for (i = 0; i < len; ++i)	3155
2723	av_store (defav, i, SvREFCNT_inc_NN (AvARRAY (invoke_av)[i + 1]));	3156	assert (!coro->invoke_cb);
		3157	assert (!coro->invoke_av);
		3158	coro->invoke_cb = SvREFCNT_inc (cb);
		3159	coro->invoke_av = av;
2724	}	3160	}
		3161
		3162	api_ready (aTHX_ (SV *)hv);
		3163
		3164	if (GIMME_V != G_VOID)
		3165	XPUSHs (sv_2mortal (newRV_noinc ((SV *)hv)));
		3166	else
		3167	SvREFCNT_dec (hv);
2725	}	3168	}
		3169
		3170	SV *
		3171	rouse_cb ()
		3172	PROTOTYPE:
		3173	CODE:
		3174	RETVAL = coro_new_rouse_cb (aTHX);
		3175	OUTPUT:
		3176	RETVAL
2726		3177
2727	void	3178	void
2728	_pool_2 (SV *cb)	3179	rouse_wait (...)
		3180	PROTOTYPE: ;$
2729	CODE:	3181	PPCODE:
2730	{	3182	CORO_EXECUTE_SLF_XS (slf_init_rouse_wait);
2731	HV hv = (HV )SvRV (coro_current);
2732	struct coro coro = SvSTATE_hv ((SV )hv);
2733
2734	sv_setsv (cb, &PL_sv_undef);
2735
2736	SvREFCNT_dec ((SV )PL_defoutgv); PL_defoutgv = (GV )coro->saved_deffh;
2737	coro->saved_deffh = 0;
2738
2739	if (coro_rss (aTHX_ coro) > SvUV (sv_pool_rss)
2740	\|\| av_len (av_async_pool) + 1 >= SvIV (sv_pool_size))
2741	{
2742	SV *old = PL_diehook;
2743	PL_diehook = 0;
2744	SvREFCNT_dec (old);
2745	croak ("\3async_pool terminate\2\n");
2746	}
2747
2748	av_clear (GvAV (PL_defgv));
2749	hv_store (hv, "desc", sizeof ("desc") - 1,
2750	newSVpvn ("[async_pool idle]", sizeof ("[async_pool idle]") - 1), 0);
2751
2752	coro->prio = 0;
2753
2754	if (coro->cctx && (coro->cctx->flags & CC_TRACE))
2755	api_trace (aTHX_ coro_current, 0);
2756
2757	av_push (av_async_pool, newSVsv (coro_current));
2758	}
2759		3183
2760		3184
2761	MODULE = Coro::State PACKAGE = PerlIO::cede	3185	MODULE = Coro::State PACKAGE = PerlIO::cede
2762		3186
2763	BOOT:	3187	BOOT:
…		…
2765		3189
2766		3190
2767	MODULE = Coro::State PACKAGE = Coro::Semaphore	3191	MODULE = Coro::State PACKAGE = Coro::Semaphore
2768		3192
2769	SV *	3193	SV *
2770	new (SV klass, SV count_ = 0)	3194	new (SV klass, SV count = 0)
2771	CODE:	3195	CODE:
2772	{	3196	RETVAL = sv_bless (
2773	/* a semaphore contains a counter IV in $sem->[0] and any waiters after that */	3197	coro_waitarray_new (aTHX_ count && SvOK (count) ? SvIV (count) : 1),
2774	AV *av = newAV ();	3198	GvSTASH (CvGV (cv))
2775	SV **ary;	3199	);
		3200	OUTPUT:
		3201	RETVAL
2776		3202
2777	/* unfortunately, building manually saves memory */	3203	# helper for Coro::Channel
2778	Newx (ary, 2, SV *);	3204	SV *
2779	AvALLOC (av) = ary;	3205	_alloc (int count)
2780	AvARRAY (av) = ary;	3206	CODE:
2781	AvMAX (av) = 1;	3207	RETVAL = coro_waitarray_new (aTHX_ count);
2782	AvFILLp (av) = 0;
2783	ary [0] = newSViv (count_ && SvOK (count_) ? SvIV (count_) : 1);
2784
2785	RETVAL = sv_bless (newRV_noinc ((SV *)av), GvSTASH (CvGV (cv)));
2786	}
2787	OUTPUT:	3208	OUTPUT:
2788	RETVAL	3209	RETVAL
2789		3210
2790	SV *	3211	SV *
2791	count (SV *self)	3212	count (SV *self)
…		…
2800	adjust = 1	3221	adjust = 1
2801	CODE:	3222	CODE:
2802	coro_semaphore_adjust (aTHX_ (AV *)SvRV (self), ix ? adjust : 1);	3223	coro_semaphore_adjust (aTHX_ (AV *)SvRV (self), ix ? adjust : 1);
2803		3224
2804	void	3225	void
2805	down (SV *self)	3226	down (...)
2806	CODE:	3227	CODE:
2807	CORO_EXECUTE_SLF_XS (slf_init_semaphore_down);	3228	CORO_EXECUTE_SLF_XS (slf_init_semaphore_down);
		3229
		3230	void
		3231	wait (...)
		3232	CODE:
		3233	CORO_EXECUTE_SLF_XS (slf_init_semaphore_wait);
2808		3234
2809	void	3235	void
2810	try (SV *self)	3236	try (SV *self)
2811	PPCODE:	3237	PPCODE:
2812	{	3238	{
…		…
2824	XSRETURN_NO;	3250	XSRETURN_NO;
2825	}	3251	}
2826		3252
2827	void	3253	void
2828	waiters (SV *self)	3254	waiters (SV *self)
2829	CODE:	3255	PPCODE:
2830	{	3256	{
2831	AV av = (AV )SvRV (self);	3257	AV av = (AV )SvRV (self);
		3258	int wcount = AvFILLp (av) + 1 - 1;
2832		3259
2833	if (GIMME_V == G_SCALAR)	3260	if (GIMME_V == G_SCALAR)
2834	XPUSHs (sv_2mortal (newSVsv (AvARRAY (av)[0])));	3261	XPUSHs (sv_2mortal (newSViv (wcount)));
2835	else	3262	else
2836	{	3263	{
2837	int i;	3264	int i;
2838	EXTEND (SP, AvFILLp (av) + 1 - 1);	3265	EXTEND (SP, wcount);
2839	for (i = 1; i <= AvFILLp (av); ++i)	3266	for (i = 1; i <= wcount; ++i)
2840	PUSHs (sv_2mortal (newRV_inc (AvARRAY (av)[i])));	3267	PUSHs (sv_2mortal (newRV_inc (AvARRAY (av)[i])));
2841	}	3268	}
2842	}	3269	}
		3270
		3271	MODULE = Coro::State PACKAGE = Coro::Signal
		3272
		3273	SV *
		3274	new (SV *klass)
		3275	CODE:
		3276	RETVAL = sv_bless (
		3277	coro_waitarray_new (aTHX_ 0),
		3278	GvSTASH (CvGV (cv))
		3279	);
		3280	OUTPUT:
		3281	RETVAL
		3282
		3283	void
		3284	wait (...)
		3285	CODE:
		3286	CORO_EXECUTE_SLF_XS (slf_init_signal_wait);
		3287
		3288	void
		3289	broadcast (SV *self)
		3290	CODE:
		3291	{
		3292	AV av = (AV )SvRV (self);
		3293	coro_signal_wake (aTHX_ av, AvFILLp (av));
		3294	}
		3295
		3296	void
		3297	send (SV *self)
		3298	CODE:
		3299	{
		3300	AV av = (AV )SvRV (self);
		3301
		3302	if (AvFILLp (av))
		3303	coro_signal_wake (aTHX_ av, 1);
		3304	else
		3305	SvIVX (AvARRAY (av)[0]) = 1; /* remember the signal */
		3306	}
		3307
		3308	IV
		3309	awaited (SV *self)
		3310	CODE:
		3311	RETVAL = AvFILLp ((AV *)SvRV (self)) + 1 - 1;
		3312	OUTPUT:
		3313	RETVAL
2843		3314
2844		3315
2845	MODULE = Coro::State PACKAGE = Coro::AnyEvent	3316	MODULE = Coro::State PACKAGE = Coro::AnyEvent
2846		3317
2847	BOOT:	3318	BOOT:
…		…
2875		3346
2876	void	3347	void
2877	_register (char target, char proto, SV *req)	3348	_register (char target, char proto, SV *req)
2878	CODE:	3349	CODE:
2879	{	3350	{
2880	HV *st;	3351	CV *req_cv = coro_sv_2cv (aTHX_ req);
2881	GV *gvp;
2882	CV *req_cv = sv_2cv (req, &st, &gvp, 0);
2883	/* newXSproto doesn't return the CV on 5.8 */	3352	/* newXSproto doesn't return the CV on 5.8 */
2884	CV *slf_cv = newXS (target, coro_aio_req_xs, __FILE__);	3353	CV *slf_cv = newXS (target, coro_aio_req_xs, __FILE__);
2885	sv_setpv ((SV *)slf_cv, proto);	3354	sv_setpv ((SV *)slf_cv, proto);
2886	sv_magicext ((SV )slf_cv, (SV )req_cv, CORO_MAGIC_type_aio, 0, 0, 0);	3355	sv_magicext ((SV )slf_cv, (SV )req_cv, CORO_MAGIC_type_aio, 0, 0, 0);
2887	}	3356	}

Diff Legend

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

Comparing Coro/Coro/State.xs (file contents): Revision 1.294 by root, Tue Nov 18 08:31:03 2008 UTC vs. Revision 1.317 by root, Thu Nov 20 06:28:52 2008 UTC

Diff Legend

Comparing Coro/Coro/State.xs (file contents):
Revision 1.294 by root, Tue Nov 18 08:31:03 2008 UTC vs.
Revision 1.317 by root, Thu Nov 20 06:28:52 2008 UTC