ViewVC Help

View File | Revision Log | Show Annotations | Download File

/cvs/Coro/Coro/State.xs

Comparing Coro/Coro/State.xs (file contents):
Revision 1.297 by root, Tue Nov 18 11:04:42 2008 UTC vs.
Revision 1.320 by root, Fri Nov 21 07:29:04 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
…		…
167	static AV *main_mainstack; /* used to differentiate between $main and others */	170	static AV *main_mainstack; /* used to differentiate between $main and others */
168	static JMPENV *main_top_env;	171	static JMPENV *main_top_env;
169	static HV *coro_state_stash, *coro_stash;	172	static HV *coro_state_stash, *coro_stash;
170	static volatile SV *coro_mortal; /* will be freed/thrown after next transfer */	173	static volatile SV *coro_mortal; /* will be freed/thrown after next transfer */
171		174
		175	static AV *av_destroy; /* destruction queue */
		176	static SV *sv_manager; /* the manager coro */
		177
172	static GV *irsgv; /* $/ */	178	static GV *irsgv; /* $/ */
173	static GV *stdoutgv; /* *STDOUT */	179	static GV *stdoutgv; /* *STDOUT */
174	static SV *rv_diehook;	180	static SV *rv_diehook;
175	static SV *rv_warnhook;	181	static SV *rv_warnhook;
176	static HV *hv_sig; /* %SIG */	182	static HV *hv_sig; /* %SIG */
177		183
178	/* async_pool helper stuff */	184	/* async_pool helper stuff */
179	static SV *sv_pool_rss;	185	static SV *sv_pool_rss;
180	static SV *sv_pool_size;	186	static SV *sv_pool_size;
		187	static SV *sv_async_pool_idle; /* description string */
181	static AV *av_async_pool;	188	static AV *av_async_pool; /* idle pool */
		189	static SV *sv_Coro; /* class string */
		190	static CV *cv_pool_handler;
		191	static CV *cv_coro_state_new;
182		192
183	/* Coro::AnyEvent */	193	/* Coro::AnyEvent */
184	static SV *sv_activity;	194	static SV *sv_activity;
185		195
186	static struct coro_cctx *cctx_first;	196	static struct coro_cctx *cctx_first;
…		…
246	/* state data */	256	/* state data */
247	struct CoroSLF slf_frame; /* saved slf frame */	257	struct CoroSLF slf_frame; /* saved slf frame */
248	AV *mainstack;	258	AV *mainstack;
249	perl_slots *slot; /* basically the saved sp */	259	perl_slots *slot; /* basically the saved sp */
250		260
		261	CV *startcv; /* the CV to execute */
251	AV *args; /* data associated with this coroutine (initial args) */	262	AV *args; /* data associated with this coroutine (initial args) */
252	int refcnt; /* coroutines are refcounted, yes */	263	int refcnt; /* coroutines are refcounted, yes */
253	int flags; /* CF_ flags */	264	int flags; /* CF_ flags */
254	HV *hv; /* the perl hash associated with this coro, if any */	265	HV *hv; /* the perl hash associated with this coro, if any */
255	void (*on_destroy)(pTHX_ struct coro *coro);	266	void (*on_destroy)(pTHX_ struct coro *coro);
256		267
257	/* statistics */	268	/* statistics */
258	int usecount; /* number of transfers to this coro */	269	int usecount; /* number of transfers to this coro */
259		270
260	/* coro process data */	271	/* coro process data */
261	int prio;	272	int prio;
262	SV *except; /* exception to be thrown */	273	SV *except; /* exception to be thrown */
		274	SV *rouse_cb;
263		275
264	/* async_pool */	276	/* async_pool */
265	SV *saved_deffh;	277	SV *saved_deffh;
		278	SV *invoke_cb;
		279	AV *invoke_av;
266		280
267	/* linked list */	281	/* linked list */
268	struct coro *next, *prev;	282	struct coro *next, *prev;
269	};	283	};
270		284
…		…
287		301
288	/* for Coro.pm */	302	/* for Coro.pm */
289	static SV *coro_current;	303	static SV *coro_current;
290	static SV *coro_readyhook;	304	static SV *coro_readyhook;
291	static AV *coro_ready [PRIO_MAX - PRIO_MIN + 1];	305	static AV *coro_ready [PRIO_MAX - PRIO_MIN + 1];
		306	static CV *cv_coro_run, *cv_coro_terminate;
292	static struct coro *coro_first;	307	static struct coro *coro_first;
293	#define coro_nready coroapi.nready	308	#define coro_nready coroapi.nready
294		309
295	/** lowlevel stuff **********************************************************/	310	/** lowlevel stuff **********************************************************/
296		311
…		…
320	#if PERL_VERSION_ATLEAST (5,10,0)	335	#if PERL_VERSION_ATLEAST (5,10,0)
321	/* silence stupid and wrong 5.10 warning that I am unable to switch off */	336	/* silence stupid and wrong 5.10 warning that I am unable to switch off */
322	get_hv (name, create);	337	get_hv (name, create);
323	#endif	338	#endif
324	return get_hv (name, create);	339	return get_hv (name, create);
		340	}
		341
		342	/* may croak */
		343	INLINE CV *
		344	coro_sv_2cv (pTHX_ SV *sv)
		345	{
		346	HV *st;
		347	GV *gvp;
		348	return sv_2cv (sv, &st, &gvp, 0);
325	}	349	}
326		350
327	static AV *	351	static AV *
328	coro_clone_padlist (pTHX_ CV *cv)	352	coro_clone_padlist (pTHX_ CV *cv)
329	{	353	{
…		…
448	else	472	else
449	{	473	{
450	#if CORO_PREFER_PERL_FUNCTIONS	474	#if CORO_PREFER_PERL_FUNCTIONS
451	/* this is probably cleaner? but also slower! */	475	/* this is probably cleaner? but also slower! */
452	/* in practise, it seems to be less stable */	476	/* in practise, it seems to be less stable */
453	CV *cp = Perl_cv_clone (cv);	477	CV *cp = Perl_cv_clone (aTHX_ cv);
454	CvPADLIST (cv) = CvPADLIST (cp);	478	CvPADLIST (cv) = CvPADLIST (cp);
455	CvPADLIST (cp) = 0;	479	CvPADLIST (cp) = 0;
456	SvREFCNT_dec (cp);	480	SvREFCNT_dec (cp);
457	#else	481	#else
458	CvPADLIST (cv) = coro_clone_padlist (aTHX_ cv);	482	CvPADLIST (cv) = coro_clone_padlist (aTHX_ cv);
…		…
602	* of perl.c:init_stacks, except that it uses less memory	626	* of perl.c:init_stacks, except that it uses less memory
603	* on the (sometimes correct) assumption that coroutines do	627	* on the (sometimes correct) assumption that coroutines do
604	* not usually need a lot of stackspace.	628	* not usually need a lot of stackspace.
605	*/	629	*/
606	#if CORO_PREFER_PERL_FUNCTIONS	630	#if CORO_PREFER_PERL_FUNCTIONS
607	# define coro_init_stacks init_stacks	631	# define coro_init_stacks(thx) init_stacks ()
608	#else	632	#else
609	static void	633	static void
610	coro_init_stacks (pTHX)	634	coro_init_stacks (pTHX)
611	{	635	{
612	PL_curstackinfo = new_stackinfo(32, 8);	636	PL_curstackinfo = new_stackinfo(32, 8);
…		…
675	#if !PERL_VERSION_ATLEAST (5,10,0)	699	#if !PERL_VERSION_ATLEAST (5,10,0)
676	Safefree (PL_retstack);	700	Safefree (PL_retstack);
677	#endif	701	#endif
678	}	702	}
679		703
		704	#define CORO_RSS \
		705	rss += sizeof (SYM (curstackinfo)); \
		706	rss += (SYM (curstackinfo->si_cxmax) + 1) * sizeof (PERL_CONTEXT); \
		707	rss += sizeof (SV) + sizeof (struct xpvav) + (1 + AvMAX (SYM (curstack))) * sizeof (SV *); \
		708	rss += SYM (tmps_max) * sizeof (SV *); \
		709	rss += (SYM (markstack_max) - SYM (markstack_ptr)) * sizeof (I32); \
		710	rss += SYM (scopestack_max) * sizeof (I32); \
		711	rss += SYM (savestack_max) * sizeof (ANY);
		712
680	static size_t	713	static size_t
681	coro_rss (pTHX_ struct coro *coro)	714	coro_rss (pTHX_ struct coro *coro)
682	{	715	{
683	size_t rss = sizeof (*coro);	716	size_t rss = sizeof (*coro);
684		717
685	if (coro->mainstack)	718	if (coro->mainstack)
686	{	719	{
687	perl_slots tmp_slot;
688	perl_slots *slot;
689
690	if (coro->flags & CF_RUNNING)	720	if (coro->flags & CF_RUNNING)
691	{	721	{
692	slot = &tmp_slot;	722	#define SYM(sym) PL_ ## sym
693		723	CORO_RSS;
694	#define VAR(name,type) slot->name = PL_ ## name;
695	# include "state.h"
696	#undef VAR	724	#undef SYM
697	}	725	}
698	else	726	else
699	slot = coro->slot;
700
701	if (slot)
702	{	727	{
703	rss += sizeof (slot->curstackinfo);	728	#define SYM(sym) coro->slot->sym
704	rss += (slot->curstackinfo->si_cxmax + 1) * sizeof (PERL_CONTEXT);	729	CORO_RSS;
705	rss += sizeof (SV) + sizeof (struct xpvav) + (1 + AvMAX (slot->curstack)) * sizeof (SV *);	730	#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	}	731	}
715	}	732	}
716		733
717	return rss;	734	return rss;
718	}	735	}
…		…
818		835
819	static int	836	static int
820	slf_check_nop (pTHX_ struct CoroSLF *frame)	837	slf_check_nop (pTHX_ struct CoroSLF *frame)
821	{	838	{
822	return 0;	839	return 0;
		840	}
		841
		842	static int
		843	slf_check_repeat (pTHX_ struct CoroSLF *frame)
		844	{
		845	return 1;
823	}	846	}
824		847
825	static UNOP coro_setup_op;	848	static UNOP coro_setup_op;
826		849
827	static void NOINLINE /* noinline to keep it out of the transfer fast path */	850	static void NOINLINE /* noinline to keep it out of the transfer fast path */
…		…
860	dSP;	883	dSP;
861	UNOP myop;	884	UNOP myop;
862		885
863	Zero (&myop, 1, UNOP);	886	Zero (&myop, 1, UNOP);
864	myop.op_next = Nullop;	887	myop.op_next = Nullop;
		888	myop.op_type = OP_ENTERSUB;
865	myop.op_flags = OPf_WANT_VOID;	889	myop.op_flags = OPf_WANT_VOID;
866		890
867	PUSHMARK (SP);	891	PUSHMARK (SP);
868	XPUSHs (sv_2mortal (av_shift (GvAV (PL_defgv))));	892	PUSHs ((SV *)coro->startcv);
869	PUTBACK;	893	PUTBACK;
870	PL_op = (OP *)&myop;	894	PL_op = (OP *)&myop;
871	PL_op = PL_ppaddr[OP_ENTERSUB](aTHX);	895	PL_op = PL_ppaddr[OP_ENTERSUB](aTHX);
872	SPAGAIN;
873	}	896	}
874		897
875	/* this newly created coroutine might be run on an existing cctx which most	898	/* 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.	899	* likely was suspended in pp_slf, so we have to emulate entering pp_slf here.
877	*/	900	*/
878	slf_frame.prepare = prepare_nop; /* provide a nop function for an eventual pp_slf */	901	slf_frame.prepare = prepare_nop; /* provide a nop function for an eventual pp_slf */
879	slf_frame.check = slf_check_nop; /* signal pp_slf to not repeat */	902	slf_frame.check = slf_check_nop; /* signal pp_slf to not repeat */
880		903
881	/* and we have to provide the pp_slf op in any case, so pp_slf can skip it */	904	/* 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;	905	coro_setup_op.op_next = PL_op;
883	coro_setup_op.op_type = OP_CUSTOM;	906	coro_setup_op.op_type = OP_ENTERSUB;
884	coro_setup_op.op_ppaddr = pp_slf;	907	coro_setup_op.op_ppaddr = pp_slf;
885	/* no flags required, as an init function won't be called */	908	/* no flags etc. required, as an init function won't be called */
886		909
887	PL_op = (OP *)&coro_setup_op;	910	PL_op = (OP *)&coro_setup_op;
888		911
889	/* copy throw, in case it was set before coro_setup */	912	/* copy throw, in case it was set before coro_setup */
890	CORO_THROW = coro->except;	913	CORO_THROW = coro->except;
…		…
919		942
920	SvREFCNT_dec (PL_diehook);	943	SvREFCNT_dec (PL_diehook);
921	SvREFCNT_dec (PL_warnhook);	944	SvREFCNT_dec (PL_warnhook);
922		945
923	SvREFCNT_dec (coro->saved_deffh);	946	SvREFCNT_dec (coro->saved_deffh);
924	SvREFCNT_dec (CORO_THROW);	947	SvREFCNT_dec (coro->rouse_cb);
		948	SvREFCNT_dec (coro->invoke_cb);
		949	SvREFCNT_dec (coro->invoke_av);
925		950
926	coro_destruct_stacks (aTHX);	951	coro_destruct_stacks (aTHX);
927	}	952	}
928		953
929	INLINE void	954	INLINE void
…		…
1001	{	1026	{
1002	PERL_CONTEXT *cx = &cxstack[cxstack_ix];	1027	PERL_CONTEXT *cx = &cxstack[cxstack_ix];
1003		1028
1004	if (CxTYPE (cx) == CXt_SUB && oldcxix < cxstack_ix)	1029	if (CxTYPE (cx) == CXt_SUB && oldcxix < cxstack_ix)
1005	{	1030	{
1006	runops_proc_t old_runops = PL_runops;
1007	dSP;	1031	dSP;
1008	GV *gv = CvGV (cx->blk_sub.cv);	1032	GV *gv = CvGV (cx->blk_sub.cv);
1009	SV *fullname = sv_2mortal (newSV (0));	1033	SV *fullname = sv_2mortal (newSV (0));
1010		1034
1011	if (isGV (gv))	1035	if (isGV (gv))
…		…
1016	SAVETMPS;	1040	SAVETMPS;
1017	EXTEND (SP, 3);	1041	EXTEND (SP, 3);
1018	PUSHMARK (SP);	1042	PUSHMARK (SP);
1019	PUSHs (&PL_sv_yes);	1043	PUSHs (&PL_sv_yes);
1020	PUSHs (fullname);	1044	PUSHs (fullname);
1021	PUSHs (CxHASARGS (cx) ? sv_2mortal (newRV_inc ((SV *)cx->blk_sub.argarray)) : &PL_sv_undef);	1045	PUSHs (CxHASARGS (cx) ? sv_2mortal (newRV_inc ((SV *)cx->blk_sub.argarray)) : &PL_sv_undef);
1022	PUTBACK;	1046	PUTBACK;
1023	cb = hv_fetch ((HV *)SvRV (coro_current), "_trace_sub_cb", sizeof ("_trace_sub_cb") - 1, 0);	1047	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);	1048	if (cb) call_sv (*cb, G_KEEPERR | G_EVAL | G_VOID | G_DISCARD);
1025	SPAGAIN;	1049	SPAGAIN;
1026	FREETMPS;	1050	FREETMPS;
…		…
1132	transfer_tail (aTHX);	1156	transfer_tail (aTHX);
1133		1157
1134	/* somebody or something will hit me for both perl_run and PL_restartop */	1158	/* somebody or something will hit me for both perl_run and PL_restartop */
1135	PL_restartop = PL_op;	1159	PL_restartop = PL_op;
1136	perl_run (PL_curinterp);	1160	perl_run (PL_curinterp);
		1161	/*
		1162	* Unfortunately, there is no way to get at the return values of the
		1163	* coro body here, as perl_run destroys these
		1164	*/
1137		1165
1138	/*	1166	/*
1139	* If perl-run returns we assume exit() was being called or the coro	1167	* 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)	1168	* 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	1169	* reason for perl_run to return. We try to exit by jumping to the
…		…
1433		1461
1434	coro->slot = 0;	1462	coro->slot = 0;
1435	}	1463	}
1436		1464
1437	cctx_destroy (coro->cctx);	1465	cctx_destroy (coro->cctx);
		1466	SvREFCNT_dec (coro->startcv);
1438	SvREFCNT_dec (coro->args);	1467	SvREFCNT_dec (coro->args);
		1468	SvREFCNT_dec (CORO_THROW);
1439		1469
1440	if (coro->next) coro->next->prev = coro->prev;	1470	if (coro->next) coro->next->prev = coro->prev;
1441	if (coro->prev) coro->prev->next = coro->next;	1471	if (coro->prev) coro->prev->next = coro->next;
1442	if (coro == coro_first) coro_first = coro->next;	1472	if (coro == coro_first) coro_first = coro->next;
1443		1473
…		…
1497		1527
1498	prepare_transfer (aTHX_ &ta, prev_sv, next_sv);	1528	prepare_transfer (aTHX_ &ta, prev_sv, next_sv);
1499	TRANSFER (ta, 1);	1529	TRANSFER (ta, 1);
1500	}	1530	}
1501		1531
		1532	/*****************************************************************************/
		1533	/* gensub: simple closure generation utility */
		1534
		1535	#define GENSUB_ARG CvXSUBANY (cv).any_ptr
		1536
		1537	/* create a closure from XS, returns a code reference */
		1538	/* the arg can be accessed via GENSUB_ARG from the callback */
		1539	/* the callback must use dXSARGS/XSRETURN */
		1540	static SV *
		1541	gensub (pTHX_ void (*xsub)(pTHX_ CV *), void *arg)
		1542	{
		1543	CV *cv = (CV *)newSV (0);
		1544
		1545	sv_upgrade ((SV *)cv, SVt_PVCV);
		1546
		1547	CvANON_on (cv);
		1548	CvISXSUB_on (cv);
		1549	CvXSUB (cv) = xsub;
		1550	GENSUB_ARG = arg;
		1551
		1552	return newRV_noinc ((SV *)cv);
		1553	}
		1554
1502	/** Coro ********************************************************************/	1555	/** Coro ********************************************************************/
1503		1556
1504	INLINE void	1557	INLINE void
1505	coro_enq (pTHX_ struct coro *coro)	1558	coro_enq (pTHX_ struct coro *coro)
1506	{	1559	{
…		…
1567	api_is_ready (pTHX_ SV *coro_sv)	1620	api_is_ready (pTHX_ SV *coro_sv)
1568	{	1621	{
1569	return !!(SvSTATE (coro_sv)->flags & CF_READY);	1622	return !!(SvSTATE (coro_sv)->flags & CF_READY);
1570	}	1623	}
1571		1624
		1625	/* expects to own a reference to next->hv */
1572	INLINE void	1626	INLINE void
		1627	prepare_schedule_to (pTHX_ struct coro_transfer_args *ta, struct coro *next)
		1628	{
		1629	SV *prev_sv = SvRV (coro_current);
		1630
		1631	ta->prev = SvSTATE_hv (prev_sv);
		1632	ta->next = next;
		1633
		1634	TRANSFER_CHECK (*ta);
		1635
		1636	SvRV_set (coro_current, (SV *)next->hv);
		1637
		1638	free_coro_mortal (aTHX);
		1639	coro_mortal = prev_sv;
		1640	}
		1641
		1642	static void
1573	prepare_schedule (pTHX_ struct coro_transfer_args *ta)	1643	prepare_schedule (pTHX_ struct coro_transfer_args *ta)
1574	{	1644	{
1575	SV *prev_sv, *next_sv;
1576
1577	for (;;)	1645	for (;;)
1578	{	1646	{
1579	next_sv = coro_deq (aTHX);	1647	SV *next_sv = coro_deq (aTHX);
1580		1648
1581	/* nothing to schedule: call the idle handler */
1582	if (expect_false (!next_sv))	1649	if (expect_true (next_sv))
1583	{	1650	{
		1651	struct coro *next = SvSTATE_hv (next_sv);
		1652
		1653	/* cannot transfer to destroyed coros, skip and look for next */
		1654	if (expect_false (next->flags & CF_DESTROYED))
		1655	SvREFCNT_dec (next_sv); /* coro_nready has already been taken care of by destroy */
		1656	else
		1657	{
		1658	next->flags &= ~CF_READY;
		1659	--coro_nready;
		1660
		1661	prepare_schedule_to (aTHX_ ta, next);
		1662	break;
		1663	}
		1664	}
		1665	else
		1666	{
		1667	/* nothing to schedule: call the idle handler */
1584	dSP;	1668	dSP;
1585		1669
1586	ENTER;	1670	ENTER;
1587	SAVETMPS;	1671	SAVETMPS;
1588		1672
…		…
1590	PUTBACK;	1674	PUTBACK;
1591	call_sv (get_sv ("Coro::idle", FALSE), G_VOID | G_DISCARD);	1675	call_sv (get_sv ("Coro::idle", FALSE), G_VOID | G_DISCARD);
1592		1676
1593	FREETMPS;	1677	FREETMPS;
1594	LEAVE;	1678	LEAVE;
1595	continue;
1596	}	1679	}
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	}	1680	}
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	}	1681	}
1623		1682
1624	INLINE void	1683	INLINE void
1625	prepare_cede (pTHX_ struct coro_transfer_args *ta)	1684	prepare_cede (pTHX_ struct coro_transfer_args *ta)
1626	{	1685	{
…		…
1647	{	1706	{
1648	struct coro_transfer_args ta;	1707	struct coro_transfer_args ta;
1649		1708
1650	prepare_schedule (aTHX_ &ta);	1709	prepare_schedule (aTHX_ &ta);
1651	TRANSFER (ta, 1);	1710	TRANSFER (ta, 1);
		1711	}
		1712
		1713	static void
		1714	api_schedule_to (pTHX_ SV *coro_sv)
		1715	{
		1716	struct coro_transfer_args ta;
		1717	struct coro *next = SvSTATE (coro_sv);
		1718
		1719	SvREFCNT_inc_NN (coro_sv);
		1720	prepare_schedule_to (aTHX_ &ta, next);
1652	}	1721	}
1653		1722
1654	static int	1723	static int
1655	api_cede (pTHX)	1724	api_cede (pTHX)
1656	{	1725	{
…		…
1703	if (coro->flags & CF_RUNNING)	1772	if (coro->flags & CF_RUNNING)
1704	PL_runops = RUNOPS_DEFAULT;	1773	PL_runops = RUNOPS_DEFAULT;
1705	else	1774	else
1706	coro->slot->runops = RUNOPS_DEFAULT;	1775	coro->slot->runops = RUNOPS_DEFAULT;
1707	}	1776	}
		1777	}
		1778
		1779	static void
		1780	coro_call_on_destroy (pTHX_ struct coro *coro)
		1781	{
		1782	SV **on_destroyp = hv_fetch (coro->hv, "_on_destroy", sizeof ("_on_destroy") - 1, 0);
		1783	SV **statusp = hv_fetch (coro->hv, "_status", sizeof ("_status") - 1, 0);
		1784
		1785	if (on_destroyp)
		1786	{
		1787	AV *on_destroy = (AV *)SvRV (*on_destroyp);
		1788
		1789	while (AvFILLp (on_destroy) >= 0)
		1790	{
		1791	dSP; /* don't disturb outer sp */
		1792	SV *cb = av_pop (on_destroy);
		1793
		1794	PUSHMARK (SP);
		1795
		1796	if (statusp)
		1797	{
		1798	int i;
		1799	AV *status = (AV *)SvRV (*statusp);
		1800	EXTEND (SP, AvFILLp (status) + 1);
		1801
		1802	for (i = 0; i <= AvFILLp (status); ++i)
		1803	PUSHs (AvARRAY (status)[i]);
		1804	}
		1805
		1806	PUTBACK;
		1807	call_sv (sv_2mortal (cb), G_VOID | G_DISCARD);
		1808	}
		1809	}
		1810	}
		1811
		1812	static void
		1813	slf_init_terminate (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		1814	{
		1815	int i;
		1816	HV *hv = (HV *)SvRV (coro_current);
		1817	AV *av = newAV ();
		1818
		1819	av_extend (av, items - 1);
		1820	for (i = 0; i < items; ++i)
		1821	av_push (av, SvREFCNT_inc_NN (arg [i]));
		1822
		1823	hv_store (hv, "_status", sizeof ("_status") - 1, newRV_noinc ((SV *)av), 0);
		1824
		1825	av_push (av_destroy, (SV *)newRV_inc ((SV *)hv)); /* RVinc for perl */
		1826	api_ready (aTHX_ sv_manager);
		1827
		1828	frame->prepare = prepare_schedule;
		1829	frame->check = slf_check_repeat;
		1830	}
		1831
		1832	/*****************************************************************************/
		1833	/* async pool handler */
		1834
		1835	static int
		1836	slf_check_pool_handler (pTHX_ struct CoroSLF *frame)
		1837	{
		1838	HV *hv = (HV *)SvRV (coro_current);
		1839	struct coro *coro = (struct coro *)frame->data;
		1840
		1841	if (!coro->invoke_cb)
		1842	return 1; /* loop till we have invoke */
		1843	else
		1844	{
		1845	hv_store (hv, "desc", sizeof ("desc") - 1,
		1846	newSVpvn ("[async_pool]", sizeof ("[async_pool]") - 1), 0);
		1847
		1848	coro->saved_deffh = SvREFCNT_inc_NN ((SV *)PL_defoutgv);
		1849
		1850	{
		1851	dSP;
		1852	XPUSHs (sv_2mortal (coro->invoke_cb)); coro->invoke_cb = 0;
		1853	PUTBACK;
		1854	}
		1855
		1856	SvREFCNT_dec (GvAV (PL_defgv));
		1857	GvAV (PL_defgv) = coro->invoke_av;
		1858	coro->invoke_av = 0;
		1859
		1860	return 0;
		1861	}
		1862	}
		1863
		1864	static void
		1865	slf_init_pool_handler (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		1866	{
		1867	HV *hv = (HV *)SvRV (coro_current);
		1868	struct coro *coro = SvSTATE_hv ((SV *)hv);
		1869
		1870	if (expect_true (coro->saved_deffh))
		1871	{
		1872	/* subsequent iteration */
		1873	SvREFCNT_dec ((SV *)PL_defoutgv); PL_defoutgv = (GV *)coro->saved_deffh;
		1874	coro->saved_deffh = 0;
		1875
		1876	if (coro_rss (aTHX_ coro) > SvUV (sv_pool_rss)
		1877	|| av_len (av_async_pool) + 1 >= SvIV (sv_pool_size))
		1878	{
		1879	coro->invoke_cb = SvREFCNT_inc_NN ((SV *)cv_coro_terminate);
		1880	coro->invoke_av = newAV ();
		1881
		1882	frame->prepare = prepare_nop;
		1883	}
		1884	else
		1885	{
		1886	av_clear (GvAV (PL_defgv));
		1887	hv_store (hv, "desc", sizeof ("desc") - 1, SvREFCNT_inc_NN (sv_async_pool_idle), 0);
		1888
		1889	coro->prio = 0;
		1890
		1891	if (coro->cctx && (coro->cctx->flags & CC_TRACE))
		1892	api_trace (aTHX_ coro_current, 0);
		1893
		1894	frame->prepare = prepare_schedule;
		1895	av_push (av_async_pool, SvREFCNT_inc (hv));
		1896	}
		1897	}
		1898	else
		1899	{
		1900	/* first iteration, simply fall through */
		1901	frame->prepare = prepare_nop;
		1902	}
		1903
		1904	frame->check = slf_check_pool_handler;
		1905	frame->data = (void *)coro;
		1906	}
		1907
		1908	/*****************************************************************************/
		1909	/* rouse callback */
		1910
		1911	#define CORO_MAGIC_type_rouse PERL_MAGIC_ext
		1912
		1913	static void
		1914	coro_rouse_callback (pTHX_ CV *cv)
		1915	{
		1916	dXSARGS;
		1917	SV *data = (SV *)GENSUB_ARG;
		1918
		1919	if (SvTYPE (SvRV (data)) != SVt_PVAV)
		1920	{
		1921	/* first call, set args */
		1922	AV *av = newAV ();
		1923	SV *coro = SvRV (data);
		1924
		1925	SvRV_set (data, (SV *)av);
		1926	api_ready (aTHX_ coro);
		1927	SvREFCNT_dec (coro);
		1928
		1929	/* better take a full copy of the arguments */
		1930	while (items--)
		1931	av_store (av, items, newSVsv (ST (items)));
		1932	}
		1933
		1934	XSRETURN_EMPTY;
		1935	}
		1936
		1937	static int
		1938	slf_check_rouse_wait (pTHX_ struct CoroSLF *frame)
		1939	{
		1940	SV *data = (SV *)frame->data;
		1941
		1942	if (CORO_THROW)
		1943	return 0;
		1944
		1945	if (SvTYPE (SvRV (data)) != SVt_PVAV)
		1946	return 1;
		1947
		1948	/* now push all results on the stack */
		1949	{
		1950	dSP;
		1951	AV *av = (AV *)SvRV (data);
		1952	int i;
		1953
		1954	EXTEND (SP, AvFILLp (av) + 1);
		1955	for (i = 0; i <= AvFILLp (av); ++i)
		1956	PUSHs (sv_2mortal (AvARRAY (av)[i]));
		1957
		1958	/* we have stolen the elements, so ste length to zero and free */
		1959	AvFILLp (av) = -1;
		1960	av_undef (av);
		1961
		1962	PUTBACK;
		1963	}
		1964
		1965	return 0;
		1966	}
		1967
		1968	static void
		1969	slf_init_rouse_wait (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		1970	{
		1971	SV *cb;
		1972
		1973	if (items)
		1974	cb = arg [0];
		1975	else
		1976	{
		1977	struct coro *coro = SvSTATE_current;
		1978
		1979	if (!coro->rouse_cb)
		1980	croak ("Coro::rouse_wait called without rouse callback, and no default rouse callback found either,");
		1981
		1982	cb = sv_2mortal (coro->rouse_cb);
		1983	coro->rouse_cb = 0;
		1984	}
		1985
		1986	if (!SvROK (cb)
		1987	|| SvTYPE (SvRV (cb)) != SVt_PVCV
		1988	|| CvXSUB ((CV *)SvRV (cb)) != coro_rouse_callback)
		1989	croak ("Coro::rouse_wait called with illegal callback argument,");
		1990
		1991	{
		1992	CV *cv = (CV *)SvRV (cb); /* for GENSUB_ARG */
		1993	SV *data = (SV *)GENSUB_ARG;
		1994
		1995	frame->data = (void *)data;
		1996	frame->prepare = SvTYPE (SvRV (data)) == SVt_PVAV ? prepare_nop : prepare_schedule;
		1997	frame->check = slf_check_rouse_wait;
		1998	}
		1999	}
		2000
		2001	static SV *
		2002	coro_new_rouse_cb (pTHX)
		2003	{
		2004	HV *hv = (HV *)SvRV (coro_current);
		2005	struct coro *coro = SvSTATE_hv (hv);
		2006	SV *data = newRV_inc ((SV *)hv);
		2007	SV *cb = gensub (aTHX_ coro_rouse_callback, (void *)data);
		2008
		2009	sv_magicext (SvRV (cb), data, CORO_MAGIC_type_rouse, 0, 0, 0);
		2010	SvREFCNT_dec (data); /* magicext increases the refcount */
		2011
		2012	SvREFCNT_dec (coro->rouse_cb);
		2013	coro->rouse_cb = SvREFCNT_inc_NN (cb);
		2014
		2015	return cb;
1708	}	2016	}
1709		2017
1710	/*****************************************************************************/	2018	/*****************************************************************************/
1711	/* schedule-like-function opcode (SLF) */	2019	/* schedule-like-function opcode (SLF) */
1712		2020
…		…
1759	static void	2067	static void
1760	slf_init_schedule (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)	2068	slf_init_schedule (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
1761	{	2069	{
1762	frame->prepare = prepare_schedule;	2070	frame->prepare = prepare_schedule;
1763	frame->check = slf_check_nop;	2071	frame->check = slf_check_nop;
		2072	}
		2073
		2074	static void
		2075	slf_prepare_schedule_to (pTHX_ struct coro_transfer_args *ta)
		2076	{
		2077	struct coro *next = (struct coro *)slf_frame.data;
		2078
		2079	SvREFCNT_inc_NN (next->hv);
		2080	prepare_schedule_to (aTHX_ ta, next);
		2081	}
		2082
		2083	static void
		2084	slf_init_schedule_to (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		2085	{
		2086	if (!items)
		2087	croak ("Coro::schedule_to expects a coroutine argument, caught");
		2088
		2089	frame->data = (void *)SvSTATE (arg [0]);
		2090	frame->prepare = slf_prepare_schedule_to;
		2091	frame->check = slf_check_nop;
		2092	}
		2093
		2094	static void
		2095	slf_init_cede_to (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		2096	{
		2097	api_ready (aTHX_ SvRV (coro_current));
		2098
		2099	slf_init_schedule_to (aTHX_ frame, cv, arg, items);
1764	}	2100	}
1765		2101
1766	static void	2102	static void
1767	slf_init_cede (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)	2103	slf_init_cede (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
1768	{	2104	{
…		…
1914	}	2250	}
1915	else	2251	else
1916	slf_argc = 0;	2252	slf_argc = 0;
1917		2253
1918	PL_op->op_ppaddr = pp_slf;	2254	PL_op->op_ppaddr = pp_slf;
1919	PL_op->op_type = OP_CUSTOM; /* maybe we should leave it at entersub? */	2255	/*PL_op->op_type = OP_CUSTOM; /* we do behave like entersub still */
1920		2256
1921	PL_op = (OP *)&slf_restore;	2257	PL_op = (OP *)&slf_restore;
1922	}	2258	}
1923		2259
1924	/*****************************************************************************/	2260	/*****************************************************************************/
…		…
1995	PerlIOBuf_get_cnt,	2331	PerlIOBuf_get_cnt,
1996	PerlIOBuf_set_ptrcnt,	2332	PerlIOBuf_set_ptrcnt,
1997	};	2333	};
1998		2334
1999	/*****************************************************************************/	2335	/*****************************************************************************/
		2336	/* Coro::Semaphore & Coro::Signal */
		2337
		2338	static SV *
		2339	coro_waitarray_new (pTHX_ int count)
		2340	{
		2341	/* a semaphore contains a counter IV in $sem->[0] and any waiters after that */
		2342	AV *av = newAV ();
		2343	SV **ary;
		2344
		2345	/* unfortunately, building manually saves memory */
		2346	Newx (ary, 2, SV *);
		2347	AvALLOC (av) = ary;
		2348	/*AvARRAY (av) = ary;*/
		2349	SvPVX ((SV *)av) = (char *)ary; /* 5.8.8 needs this syntax instead of AvARRAY = ary */
		2350	AvMAX (av) = 1;
		2351	AvFILLp (av) = 0;
		2352	ary [0] = newSViv (count);
		2353
		2354	return newRV_noinc ((SV *)av);
		2355	}
		2356
2000	/* Coro::Semaphore */	2357	/* semaphore */
2001		2358
2002	static void	2359	static void
2003	coro_semaphore_adjust (pTHX_ AV *av, IV adjust)	2360	coro_semaphore_adjust (pTHX_ AV *av, IV adjust)
2004	{	2361	{
2005	SV *count_sv = AvARRAY (av)[0];	2362	SV *count_sv = AvARRAY (av)[0];
…		…
2017	AvARRAY (av)[0] = AvARRAY (av)[1];	2374	AvARRAY (av)[0] = AvARRAY (av)[1];
2018	AvARRAY (av)[1] = count_sv;	2375	AvARRAY (av)[1] = count_sv;
2019	cb = av_shift (av);	2376	cb = av_shift (av);
2020		2377
2021	if (SvOBJECT (cb))	2378	if (SvOBJECT (cb))
		2379	{
2022	api_ready (aTHX_ cb);	2380	api_ready (aTHX_ cb);
2023	else	2381	--count;
2024	croak ("callbacks not yet supported");	2382	}
		2383	else if (SvTYPE (cb) == SVt_PVCV)
		2384	{
		2385	dSP;
		2386	PUSHMARK (SP);
		2387	XPUSHs (sv_2mortal (newRV_inc ((SV *)av)));
		2388	PUTBACK;
		2389	call_sv (cb, G_VOID | G_DISCARD | G_EVAL | G_KEEPERR);
		2390	}
2025		2391
2026	SvREFCNT_dec (cb);	2392	SvREFCNT_dec (cb);
2027
2028	--count;
2029	}	2393	}
2030	}	2394	}
2031		2395
2032	static void	2396	static void
2033	coro_semaphore_on_destroy (pTHX_ struct coro *coro)	2397	coro_semaphore_on_destroy (pTHX_ struct coro *coro)
…		…
2035	/* call $sem->adjust (0) to possibly wake up some other waiters */	2399	/* call $sem->adjust (0) to possibly wake up some other waiters */
2036	coro_semaphore_adjust (aTHX_ (AV *)coro->slf_frame.data, 0);	2400	coro_semaphore_adjust (aTHX_ (AV *)coro->slf_frame.data, 0);
2037	}	2401	}
2038		2402
2039	static int	2403	static int
2040	slf_check_semaphore_down (pTHX_ struct CoroSLF *frame)	2404	slf_check_semaphore_down_or_wait (pTHX_ struct CoroSLF *frame, int acquire)
2041	{	2405	{
2042	AV *av = (AV *)frame->data;	2406	AV *av = (AV *)frame->data;
2043	SV *count_sv = AvARRAY (av)[0];	2407	SV *count_sv = AvARRAY (av)[0];
2044		2408
2045	/* if we are about to throw, don't actually acquire the lock, just throw */	2409	/* if we are about to throw, don't actually acquire the lock, just throw */
2046	if (CORO_THROW)	2410	if (CORO_THROW)
2047	return 0;	2411	return 0;
2048	else if (SvIVX (count_sv) > 0)	2412	else if (SvIVX (count_sv) > 0)
2049	{	2413	{
2050	SvSTATE_current->on_destroy = 0;	2414	SvSTATE_current->on_destroy = 0;
		2415
		2416	if (acquire)
2051	SvIVX (count_sv) = SvIVX (count_sv) - 1;	2417	SvIVX (count_sv) = SvIVX (count_sv) - 1;
		2418	else
		2419	coro_semaphore_adjust (aTHX_ av, 0);
		2420
2052	return 0;	2421	return 0;
2053	}	2422	}
2054	else	2423	else
2055	{	2424	{
2056	int i;	2425	int i;
…		…
2065	av_push (av, SvREFCNT_inc (SvRV (coro_current)));	2434	av_push (av, SvREFCNT_inc (SvRV (coro_current)));
2066	return 1;	2435	return 1;
2067	}	2436	}
2068	}	2437	}
2069		2438
2070	static void	2439	static int
		2440	slf_check_semaphore_down (pTHX_ struct CoroSLF *frame)
		2441	{
		2442	return slf_check_semaphore_down_or_wait (aTHX_ frame, 1);
		2443	}
		2444
		2445	static int
		2446	slf_check_semaphore_wait (pTHX_ struct CoroSLF *frame)
		2447	{
		2448	return slf_check_semaphore_down_or_wait (aTHX_ frame, 0);
		2449	}
		2450
		2451	static void
2071	slf_init_semaphore_down (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)	2452	slf_init_semaphore_down_or_wait (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
2072	{	2453	{
2073	AV *av = (AV *)SvRV (arg [0]);	2454	AV *av = (AV *)SvRV (arg [0]);
2074		2455
2075	if (SvIVX (AvARRAY (av)[0]) > 0)	2456	if (SvIVX (AvARRAY (av)[0]) > 0)
2076	{	2457	{
…		…
2086		2467
2087	/* to avoid race conditions when a woken-up coro gets terminated */	2468	/* to avoid race conditions when a woken-up coro gets terminated */
2088	/* we arrange for a temporary on_destroy that calls adjust (0) */	2469	/* we arrange for a temporary on_destroy that calls adjust (0) */
2089	SvSTATE_current->on_destroy = coro_semaphore_on_destroy;	2470	SvSTATE_current->on_destroy = coro_semaphore_on_destroy;
2090	}	2471	}
		2472	}
2091		2473
		2474	static void
		2475	slf_init_semaphore_down (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		2476	{
		2477	slf_init_semaphore_down_or_wait (aTHX_ frame, cv, arg, items);
2092	frame->check = slf_check_semaphore_down;	2478	frame->check = slf_check_semaphore_down;
2093
2094	}	2479	}
2095		2480
2096	/*****************************************************************************/	2481	static void
2097	/* gensub: simple closure generation utility */	2482	slf_init_semaphore_wait (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
2098
2099	#define GENSUB_ARG CvXSUBANY (cv).any_ptr
2100
2101	/* create a closure from XS, returns a code reference */
2102	/* the arg can be accessed via GENSUB_ARG from the callback */
2103	/* the callback must use dXSARGS/XSRETURN */
2104	static SV *
2105	gensub (pTHX_ void (*xsub)(pTHX_ CV *), void *arg)
2106	{	2483	{
2107	CV *cv = (CV *)newSV (0);	2484	if (items >= 2)
		2485	{
		2486	/* callback form */
		2487	AV *av = (AV *)SvRV (arg [0]);
		2488	CV *cb_cv = coro_sv_2cv (aTHX_ arg [1]);
2108		2489
2109	sv_upgrade ((SV *)cv, SVt_PVCV);	2490	av_push (av, (SV *)SvREFCNT_inc_NN (cb_cv));
2110		2491
2111	CvANON_on (cv);	2492	if (SvIVX (AvARRAY (av)[0]) > 0)
2112	CvISXSUB_on (cv);	2493	coro_semaphore_adjust (aTHX_ av, 0);
2113	CvXSUB (cv) = xsub;
2114	GENSUB_ARG = arg;
2115		2494
2116	return newRV_noinc ((SV *)cv);	2495	frame->prepare = prepare_nop;
		2496	frame->check = slf_check_nop;
		2497	}
		2498	else
		2499	{
		2500	slf_init_semaphore_down_or_wait (aTHX_ frame, cv, arg, items);
		2501	frame->check = slf_check_semaphore_wait;
		2502	}
		2503	}
		2504
		2505	/* signal */
		2506
		2507	static void
		2508	coro_signal_wake (pTHX_ AV *av, int count)
		2509	{
		2510	SvIVX (AvARRAY (av)[0]) = 0;
		2511
		2512	/* now signal count waiters */
		2513	while (count > 0 && AvFILLp (av) > 0)
		2514	{
		2515	SV *cb;
		2516
		2517	/* swap first two elements so we can shift a waiter */
		2518	cb = AvARRAY (av)[0];
		2519	AvARRAY (av)[0] = AvARRAY (av)[1];
		2520	AvARRAY (av)[1] = cb;
		2521
		2522	cb = av_shift (av);
		2523
		2524	api_ready (aTHX_ cb);
		2525	sv_setiv (cb, 0); /* signal waiter */
		2526	SvREFCNT_dec (cb);
		2527
		2528	--count;
		2529	}
		2530	}
		2531
		2532	static int
		2533	slf_check_signal_wait (pTHX_ struct CoroSLF *frame)
		2534	{
		2535	/* if we are about to throw, also stop waiting */
		2536	return SvROK ((SV *)frame->data) && !CORO_THROW;
		2537	}
		2538
		2539	static void
		2540	slf_init_signal_wait (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		2541	{
		2542	AV *av = (AV *)SvRV (arg [0]);
		2543
		2544	if (SvIVX (AvARRAY (av)[0]))
		2545	{
		2546	SvIVX (AvARRAY (av)[0]) = 0;
		2547	frame->prepare = prepare_nop;
		2548	frame->check = slf_check_nop;
		2549	}
		2550	else
		2551	{
		2552	SV *waiter = newRV_inc (SvRV (coro_current)); /* owned by signal av */
		2553
		2554	av_push (av, waiter);
		2555
		2556	frame->data = (void *)sv_2mortal (SvREFCNT_inc_NN (waiter)); /* owned by process */
		2557	frame->prepare = prepare_schedule;
		2558	frame->check = slf_check_signal_wait;
		2559	}
2117	}	2560	}
2118		2561
2119	/*****************************************************************************/	2562	/*****************************************************************************/
2120	/* Coro::AIO */	2563	/* Coro::AIO */
2121		2564
…		…
2136	dXSARGS;	2579	dXSARGS;
2137	AV *state = (AV *)GENSUB_ARG;	2580	AV *state = (AV *)GENSUB_ARG;
2138	SV *coro = av_pop (state);	2581	SV *coro = av_pop (state);
2139	SV *data_sv = newSV (sizeof (struct io_state));	2582	SV *data_sv = newSV (sizeof (struct io_state));
2140		2583
2141	av_extend (state, items);	2584	av_extend (state, items - 1);
2142		2585
2143	sv_upgrade (data_sv, SVt_PV);	2586	sv_upgrade (data_sv, SVt_PV);
2144	SvCUR_set (data_sv, sizeof (struct io_state));	2587	SvCUR_set (data_sv, sizeof (struct io_state));
2145	SvPOK_only (data_sv);	2588	SvPOK_only (data_sv);
2146		2589
…		…
2322		2765
2323	{	2766	{
2324	SV *slf = sv_2mortal (newSViv (PTR2IV (pp_slf)));	2767	SV *slf = sv_2mortal (newSViv (PTR2IV (pp_slf)));
2325		2768
2326	if (!PL_custom_op_names) PL_custom_op_names = newHV ();	2769	if (!PL_custom_op_names) PL_custom_op_names = newHV ();
2327	hv_store_ent (PL_custom_op_names, slf,	2770	hv_store_ent (PL_custom_op_names, slf, newSVpv ("coro_slf", 0), 0);
2328	newSVpv ("coro_slf", 0), 0);
2329		2771
2330	if (!PL_custom_op_descs) PL_custom_op_descs = newHV ();	2772	if (!PL_custom_op_descs) PL_custom_op_descs = newHV ();
2331	hv_store_ent (PL_custom_op_descs, slf,	2773	hv_store_ent (PL_custom_op_descs, slf, newSVpv ("coro schedule like function", 0), 0);
2332	newSVpv ("coro schedule like function", 0), 0);
2333	}	2774	}
2334		2775
2335	coroapi.ver = CORO_API_VERSION;	2776	coroapi.ver = CORO_API_VERSION;
2336	coroapi.rev = CORO_API_REVISION;	2777	coroapi.rev = CORO_API_REVISION;
2337		2778
…		…
2356	assert (("PRIO_NORMAL must be 0", !PRIO_NORMAL));	2797	assert (("PRIO_NORMAL must be 0", !PRIO_NORMAL));
2357	}	2798	}
2358		2799
2359	SV *	2800	SV *
2360	new (char *klass, ...)	2801	new (char *klass, ...)
		2802	ALIAS:
		2803	Coro::new = 1
2361	CODE:	2804	CODE:
2362	{	2805	{
2363	struct coro *coro;	2806	struct coro *coro;
2364	MAGIC *mg;	2807	MAGIC *mg;
2365	HV *hv;	2808	HV *hv;
		2809	CV *cb;
2366	int i;	2810	int i;
		2811
		2812	if (items > 1)
		2813	{
		2814	cb = coro_sv_2cv (aTHX_ ST (1));
		2815
		2816	if (!ix)
		2817	{
		2818	if (CvISXSUB (cb))
		2819	croak ("Coro::State doesn't support XS functions as coroutine start, caught");
		2820
		2821	if (!CvROOT (cb))
		2822	croak ("Coro::State doesn't support autoloaded or undefined functions as coroutine start, caught");
		2823	}
		2824	}
2367		2825
2368	Newz (0, coro, 1, struct coro);	2826	Newz (0, coro, 1, struct coro);
2369	coro->args = newAV ();	2827	coro->args = newAV ();
2370	coro->flags = CF_NEW;	2828	coro->flags = CF_NEW;
2371		2829
…		…
2376	coro->hv = hv = newHV ();	2834	coro->hv = hv = newHV ();
2377	mg = sv_magicext ((SV *)hv, 0, CORO_MAGIC_type_state, &coro_state_vtbl, (char *)coro, 0);	2835	mg = sv_magicext ((SV *)hv, 0, CORO_MAGIC_type_state, &coro_state_vtbl, (char *)coro, 0);
2378	mg->mg_flags |= MGf_DUP;	2836	mg->mg_flags |= MGf_DUP;
2379	RETVAL = sv_bless (newRV_noinc ((SV *)hv), gv_stashpv (klass, 1));	2837	RETVAL = sv_bless (newRV_noinc ((SV *)hv), gv_stashpv (klass, 1));
2380		2838
		2839	if (items > 1)
		2840	{
2381	av_extend (coro->args, items - 1);	2841	av_extend (coro->args, items - 1 + ix - 1);
		2842
		2843	if (ix)
		2844	{
		2845	av_push (coro->args, SvREFCNT_inc_NN ((SV *)cb));
		2846	cb = cv_coro_run;
		2847	}
		2848
		2849	coro->startcv = (CV *)SvREFCNT_inc_NN ((SV *)cb);
		2850
2382	for (i = 1; i < items; i++)	2851	for (i = 2; i < items; i++)
2383	av_push (coro->args, newSVsv (ST (i)));	2852	av_push (coro->args, newSVsv (ST (i)));
		2853	}
2384	}	2854	}
2385	OUTPUT:	2855	OUTPUT:
2386	RETVAL	2856	RETVAL
2387		2857
2388	void	2858	void
…		…
2586		3056
2587	BOOT:	3057	BOOT:
2588	{	3058	{
2589	int i;	3059	int i;
2590		3060
2591	av_async_pool = coro_get_av (aTHX_ "Coro::async_pool", TRUE);
2592	sv_pool_rss = coro_get_sv (aTHX_ "Coro::POOL_RSS" , TRUE);	3061	sv_pool_rss = coro_get_sv (aTHX_ "Coro::POOL_RSS" , TRUE);
2593	sv_pool_size = coro_get_sv (aTHX_ "Coro::POOL_SIZE" , TRUE);	3062	sv_pool_size = coro_get_sv (aTHX_ "Coro::POOL_SIZE" , TRUE);
2594		3063	cv_coro_run = get_cv ( "Coro::_terminate", GV_ADD);
		3064	cv_coro_terminate = get_cv ( "Coro::terminate" , GV_ADD);
2595	coro_current = coro_get_sv (aTHX_ "Coro::current", FALSE);	3065	coro_current = coro_get_sv (aTHX_ "Coro::current" , FALSE); SvREADONLY_on (coro_current);
2596	SvREADONLY_on (coro_current);	3066	av_async_pool = coro_get_av (aTHX_ "Coro::async_pool", TRUE);
		3067	av_destroy = coro_get_av (aTHX_ "Coro::destroy" , TRUE);
		3068	sv_manager = coro_get_sv (aTHX_ "Coro::manager" , TRUE);
		3069
		3070	sv_async_pool_idle = newSVpv ("[async pool idle]", 0); SvREADONLY_on (sv_async_pool_idle);
		3071	sv_Coro = newSVpv ("Coro", 0); SvREADONLY_on (sv_Coro);
		3072	cv_pool_handler = get_cv ("Coro::pool_handler", GV_ADD); SvREADONLY_on (cv_pool_handler);
		3073	cv_coro_state_new = get_cv ("Coro::State::new", 0); SvREADONLY_on (cv_coro_state_new);
2597		3074
2598	coro_stash = gv_stashpv ("Coro", TRUE);	3075	coro_stash = gv_stashpv ("Coro", TRUE);
2599		3076
2600	newCONSTSUB (coro_stash, "PRIO_MAX", newSViv (PRIO_MAX));	3077	newCONSTSUB (coro_stash, "PRIO_MAX", newSViv (PRIO_MAX));
2601	newCONSTSUB (coro_stash, "PRIO_HIGH", newSViv (PRIO_HIGH));	3078	newCONSTSUB (coro_stash, "PRIO_HIGH", newSViv (PRIO_HIGH));
…		…
2609		3086
2610	{	3087	{
2611	SV *sv = coro_get_sv (aTHX_ "Coro::API", TRUE);	3088	SV *sv = coro_get_sv (aTHX_ "Coro::API", TRUE);
2612		3089
2613	coroapi.schedule = api_schedule;	3090	coroapi.schedule = api_schedule;
		3091	coroapi.schedule_to = api_schedule_to;
2614	coroapi.cede = api_cede;	3092	coroapi.cede = api_cede;
2615	coroapi.cede_notself = api_cede_notself;	3093	coroapi.cede_notself = api_cede_notself;
2616	coroapi.ready = api_ready;	3094	coroapi.ready = api_ready;
2617	coroapi.is_ready = api_is_ready;	3095	coroapi.is_ready = api_is_ready;
2618	coroapi.nready = coro_nready;	3096	coroapi.nready = coro_nready;
…		…
2623	SvREADONLY_on (sv);	3101	SvREADONLY_on (sv);
2624	}	3102	}
2625	}	3103	}
2626		3104
2627	void	3105	void
		3106	terminate (...)
		3107	CODE:
		3108	CORO_EXECUTE_SLF_XS (slf_init_terminate);
		3109
		3110	void
2628	schedule (...)	3111	schedule (...)
2629	CODE:	3112	CODE:
2630	CORO_EXECUTE_SLF_XS (slf_init_schedule);	3113	CORO_EXECUTE_SLF_XS (slf_init_schedule);
		3114
		3115	void
		3116	schedule_to (...)
		3117	CODE:
		3118	CORO_EXECUTE_SLF_XS (slf_init_schedule_to);
		3119
		3120	void
		3121	cede_to (...)
		3122	CODE:
		3123	CORO_EXECUTE_SLF_XS (slf_init_cede_to);
2631		3124
2632	void	3125	void
2633	cede (...)	3126	cede (...)
2634	CODE:	3127	CODE:
2635	CORO_EXECUTE_SLF_XS (slf_init_cede);	3128	CORO_EXECUTE_SLF_XS (slf_init_cede);
2636		3129
2637	void	3130	void
2638	cede_notself (...)	3131	cede_notself (...)
2639	CODE:	3132	CODE:
2640	CORO_EXECUTE_SLF_XS (slf_init_cede_notself);	3133	CORO_EXECUTE_SLF_XS (slf_init_cede_notself);
		3134
		3135	void
		3136	_cancel (Coro::State self)
		3137	CODE:
		3138	coro_state_destroy (aTHX_ self);
		3139	coro_call_on_destroy (aTHX_ self);
2641		3140
2642	void	3141	void
2643	_set_current (SV *current)	3142	_set_current (SV *current)
2644	PROTOTYPE: $	3143	PROTOTYPE: $
2645	CODE:	3144	CODE:
…		…
2690	CODE:	3189	CODE:
2691	RETVAL = coro_nready;	3190	RETVAL = coro_nready;
2692	OUTPUT:	3191	OUTPUT:
2693	RETVAL	3192	RETVAL
2694		3193
2695	# for async_pool speedup
2696	void	3194	void
2697	_pool_1 (SV *cb)	3195	_pool_handler (...)
2698	CODE:	3196	CODE:
2699	{	3197	CORO_EXECUTE_SLF_XS (slf_init_pool_handler);
2700	HV *hv = (HV *)SvRV (coro_current);
2701	struct coro *coro = SvSTATE_hv ((SV *)hv);
2702	AV *defav = GvAV (PL_defgv);
2703	SV *invoke = hv_delete (hv, "_invoke", sizeof ("_invoke") - 1, 0);
2704	AV *invoke_av;
2705	int i, len;
2706		3198
2707	if (!invoke)	3199	void
		3200	async_pool (SV *cv, ...)
		3201	PROTOTYPE: &@
		3202	PPCODE:
		3203	{
		3204	HV *hv = (HV *)av_pop (av_async_pool);
		3205	AV *av = newAV ();
		3206	SV *cb = ST (0);
		3207	int i;
		3208
		3209	av_extend (av, items - 2);
		3210	for (i = 1; i < items; ++i)
		3211	av_push (av, SvREFCNT_inc_NN (ST (i)));
		3212
		3213	if ((SV *)hv == &PL_sv_undef)
2708	{	3214	{
2709	SV *old = PL_diehook;	3215	PUSHMARK (SP);
2710	PL_diehook = 0;	3216	EXTEND (SP, 2);
2711	SvREFCNT_dec (old);	3217	PUSHs (sv_Coro);
2712	croak ("\3async_pool terminate\2\n");	3218	PUSHs ((SV *)cv_pool_handler);
		3219	PUTBACK;
		3220	call_sv ((SV *)cv_coro_state_new, G_SCALAR);
		3221	SPAGAIN;
		3222
		3223	hv = (HV *)SvREFCNT_inc_NN (SvRV (POPs));
2713	}	3224	}
2714		3225
2715	SvREFCNT_dec (coro->saved_deffh);
2716	coro->saved_deffh = SvREFCNT_inc_NN ((SV *)PL_defoutgv);
2717
2718	hv_store (hv, "desc", sizeof ("desc") - 1,
2719	newSVpvn ("[async_pool]", sizeof ("[async_pool]") - 1), 0);
2720
2721	invoke_av = (AV *)SvRV (invoke);
2722	len = av_len (invoke_av);
2723
2724	sv_setsv (cb, AvARRAY (invoke_av)[0]);
2725
2726	if (len > 0)
2727	{	3226	{
2728	av_fill (defav, len - 1);	3227	struct coro *coro = SvSTATE_hv (hv);
2729	for (i = 0; i < len; ++i)	3228
2730	av_store (defav, i, SvREFCNT_inc_NN (AvARRAY (invoke_av)[i + 1]));	3229	assert (!coro->invoke_cb);
		3230	assert (!coro->invoke_av);
		3231	coro->invoke_cb = SvREFCNT_inc (cb);
		3232	coro->invoke_av = av;
2731	}	3233	}
		3234
		3235	api_ready (aTHX_ (SV *)hv);
		3236
		3237	if (GIMME_V != G_VOID)
		3238	XPUSHs (sv_2mortal (newRV_noinc ((SV *)hv)));
		3239	else
		3240	SvREFCNT_dec (hv);
2732	}	3241	}
		3242
		3243	SV *
		3244	rouse_cb ()
		3245	PROTOTYPE:
		3246	CODE:
		3247	RETVAL = coro_new_rouse_cb (aTHX);
		3248	OUTPUT:
		3249	RETVAL
2733		3250
2734	void	3251	void
2735	_pool_2 (SV *cb)	3252	rouse_wait (...)
		3253	PROTOTYPE: ;$
2736	CODE:	3254	PPCODE:
2737	{	3255	CORO_EXECUTE_SLF_XS (slf_init_rouse_wait);
2738	HV *hv = (HV *)SvRV (coro_current);
2739	struct coro *coro = SvSTATE_hv ((SV *)hv);
2740
2741	sv_setsv (cb, &PL_sv_undef);
2742
2743	SvREFCNT_dec ((SV *)PL_defoutgv); PL_defoutgv = (GV *)coro->saved_deffh;
2744	coro->saved_deffh = 0;
2745
2746	if (coro_rss (aTHX_ coro) > SvUV (sv_pool_rss)
2747	|| av_len (av_async_pool) + 1 >= SvIV (sv_pool_size))
2748	{
2749	SV *old = PL_diehook;
2750	PL_diehook = 0;
2751	SvREFCNT_dec (old);
2752	croak ("\3async_pool terminate\2\n");
2753	}
2754
2755	av_clear (GvAV (PL_defgv));
2756	hv_store (hv, "desc", sizeof ("desc") - 1,
2757	newSVpvn ("[async_pool idle]", sizeof ("[async_pool idle]") - 1), 0);
2758
2759	coro->prio = 0;
2760
2761	if (coro->cctx && (coro->cctx->flags & CC_TRACE))
2762	api_trace (aTHX_ coro_current, 0);
2763
2764	av_push (av_async_pool, newSVsv (coro_current));
2765	}
2766		3256
2767		3257
2768	MODULE = Coro::State PACKAGE = PerlIO::cede	3258	MODULE = Coro::State PACKAGE = PerlIO::cede
2769		3259
2770	BOOT:	3260	BOOT:
…		…
2772		3262
2773		3263
2774	MODULE = Coro::State PACKAGE = Coro::Semaphore	3264	MODULE = Coro::State PACKAGE = Coro::Semaphore
2775		3265
2776	SV *	3266	SV *
2777	new (SV *klass, SV *count_ = 0)	3267	new (SV *klass, SV *count = 0)
2778	CODE:	3268	CODE:
2779	{	3269	RETVAL = sv_bless (
2780	/* a semaphore contains a counter IV in $sem->[0] and any waiters after that */	3270	coro_waitarray_new (aTHX_ count && SvOK (count) ? SvIV (count) : 1),
2781	AV *av = newAV ();	3271	GvSTASH (CvGV (cv))
2782	SV **ary;	3272	);
		3273	OUTPUT:
		3274	RETVAL
2783		3275
2784	/* unfortunately, building manually saves memory */	3276	# helper for Coro::Channel
2785	Newx (ary, 2, SV *);	3277	SV *
2786	AvALLOC (av) = ary;	3278	_alloc (int count)
2787	AvARRAY (av) = ary;	3279	CODE:
2788	AvMAX (av) = 1;	3280	RETVAL = coro_waitarray_new (aTHX_ count);
2789	AvFILLp (av) = 0;
2790	ary [0] = newSViv (count_ && SvOK (count_) ? SvIV (count_) : 1);
2791
2792	RETVAL = sv_bless (newRV_noinc ((SV *)av), GvSTASH (CvGV (cv)));
2793	}
2794	OUTPUT:	3281	OUTPUT:
2795	RETVAL	3282	RETVAL
2796		3283
2797	SV *	3284	SV *
2798	count (SV *self)	3285	count (SV *self)
…		…
2807	adjust = 1	3294	adjust = 1
2808	CODE:	3295	CODE:
2809	coro_semaphore_adjust (aTHX_ (AV *)SvRV (self), ix ? adjust : 1);	3296	coro_semaphore_adjust (aTHX_ (AV *)SvRV (self), ix ? adjust : 1);
2810		3297
2811	void	3298	void
2812	down (SV *self)	3299	down (...)
2813	CODE:	3300	CODE:
2814	CORO_EXECUTE_SLF_XS (slf_init_semaphore_down);	3301	CORO_EXECUTE_SLF_XS (slf_init_semaphore_down);
		3302
		3303	void
		3304	wait (...)
		3305	CODE:
		3306	CORO_EXECUTE_SLF_XS (slf_init_semaphore_wait);
2815		3307
2816	void	3308	void
2817	try (SV *self)	3309	try (SV *self)
2818	PPCODE:	3310	PPCODE:
2819	{	3311	{
…		…
2831	XSRETURN_NO;	3323	XSRETURN_NO;
2832	}	3324	}
2833		3325
2834	void	3326	void
2835	waiters (SV *self)	3327	waiters (SV *self)
2836	CODE:	3328	PPCODE:
2837	{	3329	{
2838	AV *av = (AV *)SvRV (self);	3330	AV *av = (AV *)SvRV (self);
		3331	int wcount = AvFILLp (av) + 1 - 1;
2839		3332
2840	if (GIMME_V == G_SCALAR)	3333	if (GIMME_V == G_SCALAR)
2841	XPUSHs (sv_2mortal (newSVsv (AvARRAY (av)[0])));	3334	XPUSHs (sv_2mortal (newSViv (wcount)));
2842	else	3335	else
2843	{	3336	{
2844	int i;	3337	int i;
2845	EXTEND (SP, AvFILLp (av) + 1 - 1);	3338	EXTEND (SP, wcount);
2846	for (i = 1; i <= AvFILLp (av); ++i)	3339	for (i = 1; i <= wcount; ++i)
2847	PUSHs (sv_2mortal (newRV_inc (AvARRAY (av)[i])));	3340	PUSHs (sv_2mortal (newRV_inc (AvARRAY (av)[i])));
2848	}	3341	}
2849	}	3342	}
		3343
		3344	MODULE = Coro::State PACKAGE = Coro::Signal
		3345
		3346	SV *
		3347	new (SV *klass)
		3348	CODE:
		3349	RETVAL = sv_bless (
		3350	coro_waitarray_new (aTHX_ 0),
		3351	GvSTASH (CvGV (cv))
		3352	);
		3353	OUTPUT:
		3354	RETVAL
		3355
		3356	void
		3357	wait (...)
		3358	CODE:
		3359	CORO_EXECUTE_SLF_XS (slf_init_signal_wait);
		3360
		3361	void
		3362	broadcast (SV *self)
		3363	CODE:
		3364	{
		3365	AV *av = (AV *)SvRV (self);
		3366	coro_signal_wake (aTHX_ av, AvFILLp (av));
		3367	}
		3368
		3369	void
		3370	send (SV *self)
		3371	CODE:
		3372	{
		3373	AV *av = (AV *)SvRV (self);
		3374
		3375	if (AvFILLp (av))
		3376	coro_signal_wake (aTHX_ av, 1);
		3377	else
		3378	SvIVX (AvARRAY (av)[0]) = 1; /* remember the signal */
		3379	}
		3380
		3381	IV
		3382	awaited (SV *self)
		3383	CODE:
		3384	RETVAL = AvFILLp ((AV *)SvRV (self)) + 1 - 1;
		3385	OUTPUT:
		3386	RETVAL
2850		3387
2851		3388
2852	MODULE = Coro::State PACKAGE = Coro::AnyEvent	3389	MODULE = Coro::State PACKAGE = Coro::AnyEvent
2853		3390
2854	BOOT:	3391	BOOT:
…		…
2882		3419
2883	void	3420	void
2884	_register (char *target, char *proto, SV *req)	3421	_register (char *target, char *proto, SV *req)
2885	CODE:	3422	CODE:
2886	{	3423	{
2887	HV *st;	3424	CV *req_cv = coro_sv_2cv (aTHX_ req);
2888	GV *gvp;
2889	CV *req_cv = sv_2cv (req, &st, &gvp, 0);
2890	/* newXSproto doesn't return the CV on 5.8 */	3425	/* newXSproto doesn't return the CV on 5.8 */
2891	CV *slf_cv = newXS (target, coro_aio_req_xs, __FILE__);	3426	CV *slf_cv = newXS (target, coro_aio_req_xs, __FILE__);
2892	sv_setpv ((SV *)slf_cv, proto);	3427	sv_setpv ((SV *)slf_cv, proto);
2893	sv_magicext ((SV *)slf_cv, (SV *)req_cv, CORO_MAGIC_type_aio, 0, 0, 0);	3428	sv_magicext ((SV *)slf_cv, (SV *)req_cv, CORO_MAGIC_type_aio, 0, 0, 0);
2894	}	3429	}

Diff Legend

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