ViewVC Help

View File | Revision Log | Show Annotations | Download File

/cvs/Coro/Coro/State.xs

Comparing Coro/Coro/State.xs (file contents):
Revision 1.287 by root, Mon Nov 17 07:03:12 2008 UTC vs.
Revision 1.318 by root, Thu Nov 20 07:02:43 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
…		…
95	# define GV_NOTQUAL 0	95	# define GV_NOTQUAL 0
96	#endif	96	#endif
97	#ifndef newSV	97	#ifndef newSV
98	# define newSV(l) NEWSV(0,l)	98	# define newSV(l) NEWSV(0,l)
99	#endif	99	#endif
		100	#ifndef CvISXSUB_on
		101	# define CvISXSUB_on(cv) (void)cv
		102	#endif
		103	#ifndef CvISXSUB
		104	# define CvISXSUB(cv) (CvXSUB (cv) ? TRUE : FALSE)
		105	#endif
100		106
101	/* 5.8.7 */	107	/* 5.8.7 */
102	#ifndef SvRV_set	108	#ifndef SvRV_set
103	# define SvRV_set(s,v) SvRV(s) = (v)	109	# define SvRV_set(s,v) SvRV(s) = (v)
104	#endif	110	#endif
…		…
117	#endif	123	#endif
118		124
119	/* The next macros try to return the current stack pointer, in an as	125	/* The next macros try to return the current stack pointer, in an as
120	* portable way as possible. */	126	* portable way as possible. */
121	#if __GNUC__ >= 4	127	#if __GNUC__ >= 4
		128	# define dSTACKLEVEL int stacklevel_dummy
122	# define dSTACKLEVEL void *stacklevel = __builtin_frame_address (0)	129	# define STACKLEVEL __builtin_frame_address (0)
123	#else	130	#else
124	# define dSTACKLEVEL volatile void *stacklevel = (volatile void *)&stacklevel	131	# define dSTACKLEVEL volatile void *stacklevel
		132	# define STACKLEVEL ((void *)&stacklevel)
125	#endif	133	#endif
126		134
127	#define IN_DESTRUCT (PL_main_cv == Nullcv)	135	#define IN_DESTRUCT (PL_main_cv == Nullcv)
128		136
129	#if __GNUC__ >= 3	137	#if __GNUC__ >= 3
…		…
140	#define expect_true(expr) expect ((expr) != 0, 1)	148	#define expect_true(expr) expect ((expr) != 0, 1)
141		149
142	#define NOINLINE attribute ((noinline))	150	#define NOINLINE attribute ((noinline))
143		151
144	#include "CoroAPI.h"	152	#include "CoroAPI.h"
		153	#define GCoroAPI (&coroapi) /* very sneaky */
145		154
146	#ifdef USE_ITHREADS	155	#ifdef USE_ITHREADS
147	# if CORO_PTHREAD	156	# if CORO_PTHREAD
148	static void *coro_thx;	157	static void *coro_thx;
149	# endif	158	# endif
150	#endif	159	#endif
151		160
152	static double (*nvtime)(); /* so why doesn't it take void? */	161	static double (*nvtime)(); /* so why doesn't it take void? */
		162
		163	/* we hijack an hopefully unused CV flag for our purposes */
		164	#define CVf_SLF 0x4000
		165	static OP *pp_slf (pTHX);
153		166
154	static U32 cctx_gen;	167	static U32 cctx_gen;
155	static size_t cctx_stacksize = CORO_STACKSIZE;	168	static size_t cctx_stacksize = CORO_STACKSIZE;
156	static struct CoroAPI coroapi;	169	static struct CoroAPI coroapi;
157	static AV *main_mainstack; /* used to differentiate between $main and others */	170	static AV *main_mainstack; /* used to differentiate between $main and others */
158	static JMPENV *main_top_env;	171	static JMPENV *main_top_env;
159	static HV *coro_state_stash, *coro_stash;	172	static HV *coro_state_stash, *coro_stash;
160	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 */
161		174
		175	static AV *av_destroy; /* destruction queue */
		176	static SV *sv_manager; /* the manager coro */
		177
162	static GV *irsgv; /* $/ */	178	static GV *irsgv; /* $/ */
163	static GV *stdoutgv; /* *STDOUT */	179	static GV *stdoutgv; /* *STDOUT */
164	static SV *rv_diehook;	180	static SV *rv_diehook;
165	static SV *rv_warnhook;	181	static SV *rv_warnhook;
166	static HV *hv_sig; /* %SIG */	182	static HV *hv_sig; /* %SIG */
167		183
168	/* async_pool helper stuff */	184	/* async_pool helper stuff */
169	static SV *sv_pool_rss;	185	static SV *sv_pool_rss;
170	static SV *sv_pool_size;	186	static SV *sv_pool_size;
		187	static SV *sv_async_pool_idle; /* description string */
171	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;
172		192
173	/* Coro::AnyEvent */	193	/* Coro::AnyEvent */
174	static SV *sv_activity;	194	static SV *sv_activity;
175		195
176	static struct coro_cctx *cctx_first;	196	static struct coro_cctx *cctx_first;
…		…
236	/* state data */	256	/* state data */
237	struct CoroSLF slf_frame; /* saved slf frame */	257	struct CoroSLF slf_frame; /* saved slf frame */
238	AV *mainstack;	258	AV *mainstack;
239	perl_slots *slot; /* basically the saved sp */	259	perl_slots *slot; /* basically the saved sp */
240		260
		261	CV *startcv; /* the CV to execute */
241	AV *args; /* data associated with this coroutine (initial args) */	262	AV *args; /* data associated with this coroutine (initial args) */
242	int refcnt; /* coroutines are refcounted, yes */	263	int refcnt; /* coroutines are refcounted, yes */
243	int flags; /* CF_ flags */	264	int flags; /* CF_ flags */
244	HV *hv; /* the perl hash associated with this coro, if any */	265	HV *hv; /* the perl hash associated with this coro, if any */
245	void (*on_destroy)(pTHX_ struct coro *coro);	266	void (*on_destroy)(pTHX_ struct coro *coro);
246		267
247	/* statistics */	268	/* statistics */
248	int usecount; /* number of transfers to this coro */	269	int usecount; /* number of transfers to this coro */
249		270
250	/* coro process data */	271	/* coro process data */
251	int prio;	272	int prio;
252	SV *throw; /* exception to be thrown */	273	SV *except; /* exception to be thrown */
		274	SV *rouse_cb;
253		275
254	/* async_pool */	276	/* async_pool */
255	SV *saved_deffh;	277	SV *saved_deffh;
		278	SV *invoke_cb;
		279	AV *invoke_av;
256		280
257	/* linked list */	281	/* linked list */
258	struct coro *next, *prev;	282	struct coro *next, *prev;
259	};	283	};
260		284
…		…
263		287
264	/* the following variables are effectively part of the perl context */	288	/* the following variables are effectively part of the perl context */
265	/* and get copied between struct coro and these variables */	289	/* and get copied between struct coro and these variables */
266	/* the mainr easonw e don't support windows process emulation */	290	/* the mainr easonw e don't support windows process emulation */
267	static struct CoroSLF slf_frame; /* the current slf frame */	291	static struct CoroSLF slf_frame; /* the current slf frame */
268	static SV *coro_throw;
269		292
270	/** Coro ********************************************************************/	293	/** Coro ********************************************************************/
271		294
272	#define PRIO_MAX 3	295	#define PRIO_MAX 3
273	#define PRIO_HIGH 1	296	#define PRIO_HIGH 1
…		…
278		301
279	/* for Coro.pm */	302	/* for Coro.pm */
280	static SV *coro_current;	303	static SV *coro_current;
281	static SV *coro_readyhook;	304	static SV *coro_readyhook;
282	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;
283	static struct coro *coro_first;	307	static struct coro *coro_first;
284	#define coro_nready coroapi.nready	308	#define coro_nready coroapi.nready
285		309
286	/** lowlevel stuff **********************************************************/	310	/** lowlevel stuff **********************************************************/
287		311
…		…
311	#if PERL_VERSION_ATLEAST (5,10,0)	335	#if PERL_VERSION_ATLEAST (5,10,0)
312	/* 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 */
313	get_hv (name, create);	337	get_hv (name, create);
314	#endif	338	#endif
315	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);
316	}	349	}
317		350
318	static AV *	351	static AV *
319	coro_clone_padlist (pTHX_ CV *cv)	352	coro_clone_padlist (pTHX_ CV *cv)
320	{	353	{
…		…
439	else	472	else
440	{	473	{
441	#if CORO_PREFER_PERL_FUNCTIONS	474	#if CORO_PREFER_PERL_FUNCTIONS
442	/* this is probably cleaner? but also slower! */	475	/* this is probably cleaner? but also slower! */
443	/* in practise, it seems to be less stable */	476	/* in practise, it seems to be less stable */
444	CV *cp = Perl_cv_clone (cv);	477	CV *cp = Perl_cv_clone (aTHX_ cv);
445	CvPADLIST (cv) = CvPADLIST (cp);	478	CvPADLIST (cv) = CvPADLIST (cp);
446	CvPADLIST (cp) = 0;	479	CvPADLIST (cp) = 0;
447	SvREFCNT_dec (cp);	480	SvREFCNT_dec (cp);
448	#else	481	#else
449	CvPADLIST (cv) = coro_clone_padlist (aTHX_ cv);	482	CvPADLIST (cv) = coro_clone_padlist (aTHX_ cv);
…		…
502		535
503	PUTBACK;	536	PUTBACK;
504	}	537	}
505		538
506	slf_frame = c->slf_frame;	539	slf_frame = c->slf_frame;
507	coro_throw = c->throw;	540	CORO_THROW = c->except;
508	}	541	}
509		542
510	static void	543	static void
511	save_perl (pTHX_ Coro__State c)	544	save_perl (pTHX_ Coro__State c)
512	{	545	{
513	c->throw = coro_throw;	546	c->except = CORO_THROW;
514	c->slf_frame = slf_frame;	547	c->slf_frame = slf_frame;
515		548
516	{	549	{
517	dSP;	550	dSP;
518	I32 cxix = cxstack_ix;	551	I32 cxix = cxstack_ix;
…		…
593	* of perl.c:init_stacks, except that it uses less memory	626	* of perl.c:init_stacks, except that it uses less memory
594	* on the (sometimes correct) assumption that coroutines do	627	* on the (sometimes correct) assumption that coroutines do
595	* not usually need a lot of stackspace.	628	* not usually need a lot of stackspace.
596	*/	629	*/
597	#if CORO_PREFER_PERL_FUNCTIONS	630	#if CORO_PREFER_PERL_FUNCTIONS
598	# define coro_init_stacks init_stacks	631	# define coro_init_stacks(thx) init_stacks ()
599	#else	632	#else
600	static void	633	static void
601	coro_init_stacks (pTHX)	634	coro_init_stacks (pTHX)
602	{	635	{
603	PL_curstackinfo = new_stackinfo(32, 8);	636	PL_curstackinfo = new_stackinfo(32, 8);
…		…
666	#if !PERL_VERSION_ATLEAST (5,10,0)	699	#if !PERL_VERSION_ATLEAST (5,10,0)
667	Safefree (PL_retstack);	700	Safefree (PL_retstack);
668	#endif	701	#endif
669	}	702	}
670		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
671	static size_t	713	static size_t
672	coro_rss (pTHX_ struct coro *coro)	714	coro_rss (pTHX_ struct coro *coro)
673	{	715	{
674	size_t rss = sizeof (*coro);	716	size_t rss = sizeof (*coro);
675		717
676	if (coro->mainstack)	718	if (coro->mainstack)
677	{	719	{
678	perl_slots tmp_slot;
679	perl_slots *slot;
680
681	if (coro->flags & CF_RUNNING)	720	if (coro->flags & CF_RUNNING)
682	{	721	{
683	slot = &tmp_slot;	722	#define SYM(sym) PL_ ## sym
684		723	CORO_RSS;
685	#define VAR(name,type) slot->name = PL_ ## name;
686	# include "state.h"
687	#undef VAR	724	#undef SYM
688	}	725	}
689	else	726	else
690	slot = coro->slot;
691
692	if (slot)
693	{	727	{
694	rss += sizeof (slot->curstackinfo);	728	#define SYM(sym) coro->slot->sym
695	rss += (slot->curstackinfo->si_cxmax + 1) * sizeof (PERL_CONTEXT);	729	CORO_RSS;
696	rss += sizeof (SV) + sizeof (struct xpvav) + (1 + AvMAX (slot->curstack)) * sizeof (SV *);	730	#undef SYM
697	rss += slot->tmps_max * sizeof (SV *);
698	rss += (slot->markstack_max - slot->markstack_ptr) * sizeof (I32);
699	rss += slot->scopestack_max * sizeof (I32);
700	rss += slot->savestack_max * sizeof (ANY);
701
702	#if !PERL_VERSION_ATLEAST (5,10,0)
703	rss += slot->retstack_max * sizeof (OP *);
704	#endif
705	}	731	}
706	}	732	}
707		733
708	return rss;	734	return rss;
709	}	735	}
…		…
810	static int	836	static int
811	slf_check_nop (pTHX_ struct CoroSLF *frame)	837	slf_check_nop (pTHX_ struct CoroSLF *frame)
812	{	838	{
813	return 0;	839	return 0;
814	}	840	}
		841
		842	static int
		843	slf_check_repeat (pTHX_ struct CoroSLF *frame)
		844	{
		845	return 1;
		846	}
		847
		848	static UNOP coro_setup_op;
815		849
816	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 */
817	coro_setup (pTHX_ struct coro *coro)	851	coro_setup (pTHX_ struct coro *coro)
818	{	852	{
819	/*	853	/*
…		…
848	{	882	{
849	dSP;	883	dSP;
850	UNOP myop;	884	UNOP myop;
851		885
852	Zero (&myop, 1, UNOP);	886	Zero (&myop, 1, UNOP);
853	myop.op_next = Nullop;	887	myop.op_next = Nullop;
		888	myop.op_type = OP_ENTERSUB;
854	myop.op_flags = OPf_WANT_VOID;	889	myop.op_flags = OPf_WANT_VOID;
855		890
856	PUSHMARK (SP);	891	PUSHMARK (SP);
857	XPUSHs (sv_2mortal (av_shift (GvAV (PL_defgv))));	892	PUSHs ((SV *)coro->startcv);
858	PUTBACK;	893	PUTBACK;
859	PL_op = (OP *)&myop;	894	PL_op = (OP *)&myop;
860	PL_op = PL_ppaddr[OP_ENTERSUB](aTHX);	895	PL_op = PL_ppaddr[OP_ENTERSUB](aTHX);
861	SPAGAIN;
862	}	896	}
863		897
864	/* 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
865	* 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.
866	*/	900	*/
867	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 */
868	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 */
869		903
870	coro_throw = coro->throw;	904	/* and we have to provide the pp_slf op in any case, so pp_slf can skip it */
		905	coro_setup_op.op_next = PL_op;
		906	coro_setup_op.op_type = OP_CUSTOM;
		907	coro_setup_op.op_ppaddr = pp_slf;
		908	/* no flags etc. required, as an init function won't be called */
		909
		910	PL_op = (OP *)&coro_setup_op;
		911
		912	/* copy throw, in case it was set before coro_setup */
		913	CORO_THROW = coro->except;
871	}	914	}
872		915
873	static void	916	static void
874	coro_destruct (pTHX_ struct coro *coro)	917	coro_destruct (pTHX_ struct coro *coro)
875	{	918	{
…		…
899		942
900	SvREFCNT_dec (PL_diehook);	943	SvREFCNT_dec (PL_diehook);
901	SvREFCNT_dec (PL_warnhook);	944	SvREFCNT_dec (PL_warnhook);
902		945
903	SvREFCNT_dec (coro->saved_deffh);	946	SvREFCNT_dec (coro->saved_deffh);
904	SvREFCNT_dec (coro_throw);	947	SvREFCNT_dec (coro->rouse_cb);
		948	SvREFCNT_dec (coro->invoke_cb);
		949	SvREFCNT_dec (coro->invoke_av);
905		950
906	coro_destruct_stacks (aTHX);	951	coro_destruct_stacks (aTHX);
907	}	952	}
908		953
909	INLINE void	954	INLINE void
…		…
996	SAVETMPS;	1041	SAVETMPS;
997	EXTEND (SP, 3);	1042	EXTEND (SP, 3);
998	PUSHMARK (SP);	1043	PUSHMARK (SP);
999	PUSHs (&PL_sv_yes);	1044	PUSHs (&PL_sv_yes);
1000	PUSHs (fullname);	1045	PUSHs (fullname);
1001	PUSHs (CxHASARGS (cx) ? sv_2mortal (newRV_inc ((SV *)cx->blk_sub.argarray)) : &PL_sv_undef);	1046	PUSHs (CxHASARGS (cx) ? sv_2mortal (newRV_inc ((SV *)cx->blk_sub.argarray)) : &PL_sv_undef);
1002	PUTBACK;	1047	PUTBACK;
1003	cb = hv_fetch ((HV *)SvRV (coro_current), "_trace_sub_cb", sizeof ("_trace_sub_cb") - 1, 0);	1048	cb = hv_fetch ((HV *)SvRV (coro_current), "_trace_sub_cb", sizeof ("_trace_sub_cb") - 1, 0);
1004	if (cb) call_sv (*cb, G_KEEPERR | G_EVAL | G_VOID | G_DISCARD);	1049	if (cb) call_sv (*cb, G_KEEPERR | G_EVAL | G_VOID | G_DISCARD);
1005	SPAGAIN;	1050	SPAGAIN;
1006	FREETMPS;	1051	FREETMPS;
…		…
1038		1083
1039	TAINT_NOT;	1084	TAINT_NOT;
1040	return 0;	1085	return 0;
1041	}	1086	}
1042		1087
		1088	static struct coro_cctx *cctx_ssl_cctx;
		1089	static struct CoroSLF cctx_ssl_frame;
		1090
1043	static void	1091	static void
1044	prepare_set_stacklevel (struct coro_transfer_args *ta, struct coro_cctx *cctx)	1092	slf_prepare_set_stacklevel (pTHX_ struct coro_transfer_args *ta)
1045	{	1093	{
1046	ta->prev = (struct coro *)cctx;	1094	ta->prev = (struct coro *)cctx_ssl_cctx;
1047	ta->next = 0;	1095	ta->next = 0;
1048	}	1096	}
1049		1097
1050	/* inject a fake call to Coro::State::_cctx_init into the execution */	1098	static int
1051	/* _cctx_init should be careful, as it could be called at almost any time */	1099	slf_check_set_stacklevel (pTHX_ struct CoroSLF *frame)
1052	/* during execution of a perl program */	1100	{
1053	/* also initialises PL_top_env */	1101	*frame = cctx_ssl_frame;
		1102
		1103	return frame->check (aTHX_ frame); /* execute the restored frame - there must be one */
		1104	}
		1105
		1106	/* initialises PL_top_env and injects a pseudo-slf-call to set the stacklevel */
1054	static void NOINLINE	1107	static void NOINLINE
1055	cctx_prepare (pTHX_ coro_cctx *cctx)	1108	cctx_prepare (pTHX_ coro_cctx *cctx)
1056	{	1109	{
1057	dSP;
1058	UNOP myop;
1059
1060	PL_top_env = &PL_start_env;	1110	PL_top_env = &PL_start_env;
1061		1111
1062	if (cctx->flags & CC_TRACE)	1112	if (cctx->flags & CC_TRACE)
1063	PL_runops = runops_trace;	1113	PL_runops = runops_trace;
1064		1114
1065	Zero (&myop, 1, UNOP);	1115	/* we already must be executing an SLF op, there is no other valid way
1066	myop.op_next = PL_op;	1116	* that can lead to creation of a new cctx */
1067	myop.op_flags = OPf_WANT_VOID | OPf_STACKED;	1117	assert (("FATAL: can't prepare slf-less cctx in Coro module (please report)",
		1118	slf_frame.prepare && PL_op->op_ppaddr == pp_slf));
1068		1119
1069	PUSHMARK (SP);	1120	/* we must emulate leaving pp_slf, which is done inside slf_check_set_stacklevel */
1070	EXTEND (SP, 2);	1121	cctx_ssl_cctx = cctx;
1071	PUSHs (sv_2mortal (newSViv ((IV)cctx)));	1122	cctx_ssl_frame = slf_frame;
1072	PUSHs ((SV *)get_cv ("Coro::State::_cctx_init", FALSE));	1123
1073	PUTBACK;	1124	slf_frame.prepare = slf_prepare_set_stacklevel;
1074	PL_op = (OP *)&myop;	1125	slf_frame.check = slf_check_set_stacklevel;
1075	PL_op = PL_ppaddr[OP_ENTERSUB](aTHX);
1076	SPAGAIN;
1077	}	1126	}
1078		1127
1079	/* the tail of transfer: execute stuff we can only do after a transfer */	1128	/* the tail of transfer: execute stuff we can only do after a transfer */
1080	INLINE void	1129	INLINE void
1081	transfer_tail (pTHX)	1130	transfer_tail (pTHX)
…		…
1108	transfer_tail (aTHX);	1157	transfer_tail (aTHX);
1109		1158
1110	/* somebody or something will hit me for both perl_run and PL_restartop */	1159	/* somebody or something will hit me for both perl_run and PL_restartop */
1111	PL_restartop = PL_op;	1160	PL_restartop = PL_op;
1112	perl_run (PL_curinterp);	1161	perl_run (PL_curinterp);
		1162	/*
		1163	* Unfortunately, there is no way to get at the return values of the
		1164	* coro body here, as perl_run destroys these
		1165	*/
1113		1166
1114	/*	1167	/*
1115	* If perl-run returns we assume exit() was being called or the coro	1168	* If perl-run returns we assume exit() was being called or the coro
1116	* fell off the end, which seems to be the only valid (non-bug)	1169	* fell off the end, which seems to be the only valid (non-bug)
1117	* reason for perl_run to return. We try to exit by jumping to the	1170	* reason for perl_run to return. We try to exit by jumping to the
…		…
1297	dSTACKLEVEL;	1350	dSTACKLEVEL;
1298		1351
1299	/* sometimes transfer is only called to set idle_sp */	1352	/* sometimes transfer is only called to set idle_sp */
1300	if (expect_false (!next))	1353	if (expect_false (!next))
1301	{	1354	{
1302	((coro_cctx *)prev)->idle_sp = (void *)stacklevel;	1355	((coro_cctx *)prev)->idle_sp = STACKLEVEL;
1303	assert (((coro_cctx *)prev)->idle_te = PL_top_env); /* just for the side-effect when asserts are enabled */	1356	assert (((coro_cctx *)prev)->idle_te = PL_top_env); /* just for the side-effect when asserts are enabled */
1304	}	1357	}
1305	else if (expect_true (prev != next))	1358	else if (expect_true (prev != next))
1306	{	1359	{
1307	coro_cctx *prev__cctx;	1360	coro_cctx *prev__cctx;
…		…
1332		1385
1333	prev__cctx = prev->cctx;	1386	prev__cctx = prev->cctx;
1334		1387
1335	/* possibly untie and reuse the cctx */	1388	/* possibly untie and reuse the cctx */
1336	if (expect_true (	1389	if (expect_true (
1337	prev__cctx->idle_sp == (void *)stacklevel	1390	prev__cctx->idle_sp == STACKLEVEL
1338	&& !(prev__cctx->flags & CC_TRACE)	1391	&& !(prev__cctx->flags & CC_TRACE)
1339	&& !force_cctx	1392	&& !force_cctx
1340	))	1393	))
1341	{	1394	{
1342	/* I assume that stacklevel is a stronger indicator than PL_top_env changes */	1395	/* I assume that stacklevel is a stronger indicator than PL_top_env changes */
…		…
1409		1462
1410	coro->slot = 0;	1463	coro->slot = 0;
1411	}	1464	}
1412		1465
1413	cctx_destroy (coro->cctx);	1466	cctx_destroy (coro->cctx);
		1467	SvREFCNT_dec (coro->startcv);
1414	SvREFCNT_dec (coro->args);	1468	SvREFCNT_dec (coro->args);
		1469	SvREFCNT_dec (CORO_THROW);
1415		1470
1416	if (coro->next) coro->next->prev = coro->prev;	1471	if (coro->next) coro->next->prev = coro->prev;
1417	if (coro->prev) coro->prev->next = coro->next;	1472	if (coro->prev) coro->prev->next = coro->next;
1418	if (coro == coro_first) coro_first = coro->next;	1473	if (coro == coro_first) coro_first = coro->next;
1419		1474
…		…
1473		1528
1474	prepare_transfer (aTHX_ &ta, prev_sv, next_sv);	1529	prepare_transfer (aTHX_ &ta, prev_sv, next_sv);
1475	TRANSFER (ta, 1);	1530	TRANSFER (ta, 1);
1476	}	1531	}
1477		1532
		1533	/*****************************************************************************/
		1534	/* gensub: simple closure generation utility */
		1535
		1536	#define GENSUB_ARG CvXSUBANY (cv).any_ptr
		1537
		1538	/* create a closure from XS, returns a code reference */
		1539	/* the arg can be accessed via GENSUB_ARG from the callback */
		1540	/* the callback must use dXSARGS/XSRETURN */
		1541	static SV *
		1542	gensub (pTHX_ void (*xsub)(pTHX_ CV *), void *arg)
		1543	{
		1544	CV *cv = (CV *)newSV (0);
		1545
		1546	sv_upgrade ((SV *)cv, SVt_PVCV);
		1547
		1548	CvANON_on (cv);
		1549	CvISXSUB_on (cv);
		1550	CvXSUB (cv) = xsub;
		1551	GENSUB_ARG = arg;
		1552
		1553	return newRV_noinc ((SV *)cv);
		1554	}
		1555
1478	/** Coro ********************************************************************/	1556	/** Coro ********************************************************************/
1479		1557
1480	INLINE void	1558	INLINE void
1481	coro_enq (pTHX_ struct coro *coro)	1559	coro_enq (pTHX_ struct coro *coro)
1482	{	1560	{
…		…
1525	ENTER;	1603	ENTER;
1526	SAVETMPS;	1604	SAVETMPS;
1527		1605
1528	PUSHMARK (SP);	1606	PUSHMARK (SP);
1529	PUTBACK;	1607	PUTBACK;
1530	call_sv (sv_hook, G_DISCARD);	1608	call_sv (sv_hook, G_VOID | G_DISCARD);
1531	SPAGAIN;
1532		1609
1533	FREETMPS;	1610	FREETMPS;
1534	LEAVE;	1611	LEAVE;
1535	}	1612	}
1536		1613
…		…
1544	api_is_ready (pTHX_ SV *coro_sv)	1621	api_is_ready (pTHX_ SV *coro_sv)
1545	{	1622	{
1546	return !!(SvSTATE (coro_sv)->flags & CF_READY);	1623	return !!(SvSTATE (coro_sv)->flags & CF_READY);
1547	}	1624	}
1548		1625
		1626	/* expects to own a reference to next->hv */
1549	INLINE void	1627	INLINE void
		1628	prepare_schedule_to (pTHX_ struct coro_transfer_args *ta, struct coro *next)
		1629	{
		1630	SV *prev_sv = SvRV (coro_current);
		1631
		1632	ta->prev = SvSTATE_hv (prev_sv);
		1633	ta->next = next;
		1634
		1635	TRANSFER_CHECK (*ta);
		1636
		1637	SvRV_set (coro_current, (SV *)next->hv);
		1638
		1639	free_coro_mortal (aTHX);
		1640	coro_mortal = prev_sv;
		1641	}
		1642
		1643	static void
1550	prepare_schedule (pTHX_ struct coro_transfer_args *ta)	1644	prepare_schedule (pTHX_ struct coro_transfer_args *ta)
1551	{	1645	{
1552	SV *prev_sv, *next_sv;
1553
1554	for (;;)	1646	for (;;)
1555	{	1647	{
1556	next_sv = coro_deq (aTHX);	1648	SV *next_sv = coro_deq (aTHX);
1557		1649
1558	/* nothing to schedule: call the idle handler */
1559	if (expect_false (!next_sv))	1650	if (expect_true (next_sv))
1560	{	1651	{
		1652	struct coro *next = SvSTATE_hv (next_sv);
		1653
		1654	/* cannot transfer to destroyed coros, skip and look for next */
		1655	if (expect_false (next->flags & CF_DESTROYED))
		1656	SvREFCNT_dec (next_sv); /* coro_nready has already been taken care of by destroy */
		1657	else
		1658	{
		1659	next->flags &= ~CF_READY;
		1660	--coro_nready;
		1661
		1662	return prepare_schedule_to (aTHX_ ta, next);
		1663	}
		1664	}
		1665	else
		1666	{
		1667	/* nothing to schedule: call the idle handler */
1561	dSP;	1668	dSP;
1562		1669
1563	ENTER;	1670	ENTER;
1564	SAVETMPS;	1671	SAVETMPS;
1565		1672
1566	PUSHMARK (SP);	1673	PUSHMARK (SP);
1567	PUTBACK;	1674	PUTBACK;
1568	call_sv (get_sv ("Coro::idle", FALSE), G_DISCARD);	1675	call_sv (get_sv ("Coro::idle", FALSE), G_VOID | G_DISCARD);
1569	SPAGAIN;
1570		1676
1571	FREETMPS;	1677	FREETMPS;
1572	LEAVE;	1678	LEAVE;
1573	continue;
1574	}	1679	}
1575
1576	ta->next = SvSTATE_hv (next_sv);
1577
1578	/* cannot transfer to destroyed coros, skip and look for next */
1579	if (expect_false (ta->next->flags & CF_DESTROYED))
1580	{
1581	SvREFCNT_dec (next_sv);
1582	/* coro_nready has already been taken care of by destroy */
1583	continue;
1584	}
1585
1586	--coro_nready;
1587	break;
1588	}	1680	}
1589
1590	/* free this only after the transfer */
1591	prev_sv = SvRV (coro_current);
1592	ta->prev = SvSTATE_hv (prev_sv);
1593	TRANSFER_CHECK (*ta);
1594	assert (("FATAL: next coroutine isn't marked as ready in Coro (please report)", ta->next->flags & CF_READY));
1595	ta->next->flags &= ~CF_READY;
1596	SvRV_set (coro_current, next_sv);
1597
1598	free_coro_mortal (aTHX);
1599	coro_mortal = prev_sv;
1600	}	1681	}
1601		1682
1602	INLINE void	1683	INLINE void
1603	prepare_cede (pTHX_ struct coro_transfer_args *ta)	1684	prepare_cede (pTHX_ struct coro_transfer_args *ta)
1604	{	1685	{
…		…
1625	{	1706	{
1626	struct coro_transfer_args ta;	1707	struct coro_transfer_args ta;
1627		1708
1628	prepare_schedule (aTHX_ &ta);	1709	prepare_schedule (aTHX_ &ta);
1629	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);
1630	}	1721	}
1631		1722
1632	static int	1723	static int
1633	api_cede (pTHX)	1724	api_cede (pTHX)
1634	{	1725	{
…		…
1681	if (coro->flags & CF_RUNNING)	1772	if (coro->flags & CF_RUNNING)
1682	PL_runops = RUNOPS_DEFAULT;	1773	PL_runops = RUNOPS_DEFAULT;
1683	else	1774	else
1684	coro->slot->runops = RUNOPS_DEFAULT;	1775	coro->slot->runops = RUNOPS_DEFAULT;
1685	}	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;
1686	}	2016	}
1687		2017
1688	/*****************************************************************************/	2018	/*****************************************************************************/
1689	/* schedule-like-function opcode (SLF) */	2019	/* schedule-like-function opcode (SLF) */
1690		2020
…		…
1714		2044
1715	RETURNOP (slf_restore.op_first);	2045	RETURNOP (slf_restore.op_first);
1716	}	2046	}
1717		2047
1718	static void	2048	static void
1719	slf_prepare_set_stacklevel (pTHX_ struct coro_transfer_args *ta)
1720	{
1721	prepare_set_stacklevel (ta, (struct coro_cctx *)slf_frame.data);
1722	}
1723
1724	static void
1725	slf_init_set_stacklevel (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
1726	{
1727	assert (("FATAL: set_stacklevel needs the coro cctx as sole argument", items == 1));
1728
1729	frame->prepare = slf_prepare_set_stacklevel;
1730	frame->check = slf_check_nop;
1731	frame->data = (void *)SvIV (arg [0]);
1732	}
1733
1734	static void
1735	slf_prepare_transfer (pTHX_ struct coro_transfer_args *ta)	2049	slf_prepare_transfer (pTHX_ struct coro_transfer_args *ta)
1736	{	2050	{
1737	SV **arg = (SV **)slf_frame.data;	2051	SV **arg = (SV **)slf_frame.data;
1738		2052
1739	prepare_transfer (aTHX_ ta, arg [0], arg [1]);	2053	prepare_transfer (aTHX_ ta, arg [0], arg [1]);
…		…
1756	frame->prepare = prepare_schedule;	2070	frame->prepare = prepare_schedule;
1757	frame->check = slf_check_nop;	2071	frame->check = slf_check_nop;
1758	}	2072	}
1759		2073
1760	static void	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);
		2100	}
		2101
		2102	static void
1761	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)
1762	{	2104	{
1763	frame->prepare = prepare_cede;	2105	frame->prepare = prepare_cede;
1764	frame->check = slf_check_nop;	2106	frame->check = slf_check_nop;
1765	}	2107	}
…		…
1768	slf_init_cede_notself (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)	2110	slf_init_cede_notself (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
1769	{	2111	{
1770	frame->prepare = prepare_cede_notself;	2112	frame->prepare = prepare_cede_notself;
1771	frame->check = slf_check_nop;	2113	frame->check = slf_check_nop;
1772	}	2114	}
1773
1774	/* we hijack an hopefully unused CV flag for our purposes */
1775	#define CVf_SLF 0x4000
1776		2115
1777	/*	2116	/*
1778	* these not obviously related functions are all rolled into one	2117	* these not obviously related functions are all rolled into one
1779	* function to increase chances that they all will call transfer with the same	2118	* function to increase chances that they all will call transfer with the same
1780	* stack offset	2119	* stack offset
…		…
1836	}	2175	}
1837	while (slf_frame.check (aTHX_ &slf_frame));	2176	while (slf_frame.check (aTHX_ &slf_frame));
1838		2177
1839	slf_frame.prepare = 0; /* invalidate the frame, we are done processing it */	2178	slf_frame.prepare = 0; /* invalidate the frame, we are done processing it */
1840		2179
1841	/* return value handling - mostly like entersub */
1842	{
1843	dSP;
1844	SV **bot = PL_stack_base + checkmark;
1845	int gimme = GIMME_V;
1846
1847	/* make sure we put something on the stack in scalar context */
1848	if (gimme == G_SCALAR)
1849	{
1850	if (sp == bot)
1851	XPUSHs (&PL_sv_undef);
1852
1853	SP = bot + 1;
1854	}
1855
1856	PUTBACK;
1857	}
1858
1859	/* exception handling */	2180	/* exception handling */
1860	if (expect_false (coro_throw))	2181	if (expect_false (CORO_THROW))
1861	{	2182	{
1862	SV *exception = sv_2mortal (coro_throw);	2183	SV *exception = sv_2mortal (CORO_THROW);
1863		2184
1864	coro_throw = 0;	2185	CORO_THROW = 0;
1865	sv_setsv (ERRSV, exception);	2186	sv_setsv (ERRSV, exception);
1866	croak (0);	2187	croak (0);
		2188	}
		2189
		2190	/* return value handling - mostly like entersub */
		2191	/* make sure we put something on the stack in scalar context */
		2192	if (GIMME_V == G_SCALAR)
		2193	{
		2194	dSP;
		2195	SV **bot = PL_stack_base + checkmark;
		2196
		2197	if (sp == bot) /* too few, push undef */
		2198	bot [1] = &PL_sv_undef;
		2199	else if (sp != bot + 1) /* too many, take last one */
		2200	bot [1] = *sp;
		2201
		2202	SP = bot + 1;
		2203
		2204	PUTBACK;
1867	}	2205	}
1868		2206
1869	return NORMAL;	2207	return NORMAL;
1870	}	2208	}
1871		2209
…		…
1993	PerlIOBuf_get_cnt,	2331	PerlIOBuf_get_cnt,
1994	PerlIOBuf_set_ptrcnt,	2332	PerlIOBuf_set_ptrcnt,
1995	};	2333	};
1996		2334
1997	/*****************************************************************************/	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
1998	/* Coro::Semaphore */	2357	/* semaphore */
1999		2358
2000	static void	2359	static void
2001	coro_semaphore_adjust (pTHX_ AV *av, IV adjust)	2360	coro_semaphore_adjust (pTHX_ AV *av, IV adjust)
2002	{	2361	{
2003	SV *count_sv = AvARRAY (av)[0];	2362	SV *count_sv = AvARRAY (av)[0];
…		…
2015	AvARRAY (av)[0] = AvARRAY (av)[1];	2374	AvARRAY (av)[0] = AvARRAY (av)[1];
2016	AvARRAY (av)[1] = count_sv;	2375	AvARRAY (av)[1] = count_sv;
2017	cb = av_shift (av);	2376	cb = av_shift (av);
2018		2377
2019	if (SvOBJECT (cb))	2378	if (SvOBJECT (cb))
		2379	{
2020	api_ready (aTHX_ cb);	2380	api_ready (aTHX_ cb);
2021	else	2381	--count;
2022	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	}
2023		2391
2024	SvREFCNT_dec (cb);	2392	SvREFCNT_dec (cb);
2025
2026	--count;
2027	}	2393	}
2028	}	2394	}
2029		2395
2030	static void	2396	static void
2031	coro_semaphore_on_destroy (pTHX_ struct coro *coro)	2397	coro_semaphore_on_destroy (pTHX_ struct coro *coro)
…		…
2033	/* call $sem->adjust (0) to possibly wake up some other waiters */	2399	/* call $sem->adjust (0) to possibly wake up some other waiters */
2034	coro_semaphore_adjust (aTHX_ (AV *)coro->slf_frame.data, 0);	2400	coro_semaphore_adjust (aTHX_ (AV *)coro->slf_frame.data, 0);
2035	}	2401	}
2036		2402
2037	static int	2403	static int
2038	slf_check_semaphore_down (pTHX_ struct CoroSLF *frame)	2404	slf_check_semaphore_down_or_wait (pTHX_ struct CoroSLF *frame, int acquire)
2039	{	2405	{
2040	AV *av = (AV *)frame->data;	2406	AV *av = (AV *)frame->data;
2041	SV *count_sv = AvARRAY (av)[0];	2407	SV *count_sv = AvARRAY (av)[0];
2042		2408
		2409	/* if we are about to throw, don't actually acquire the lock, just throw */
		2410	if (CORO_THROW)
		2411	return 0;
2043	if (SvIVX (count_sv) > 0)	2412	else if (SvIVX (count_sv) > 0)
2044	{	2413	{
2045	SvSTATE_current->on_destroy = 0;	2414	SvSTATE_current->on_destroy = 0;
		2415
		2416	if (acquire)
2046	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
2047	return 0;	2421	return 0;
2048	}	2422	}
2049	else	2423	else
2050	{	2424	{
2051	int i;	2425	int i;
…		…
2060	av_push (av, SvREFCNT_inc (SvRV (coro_current)));	2434	av_push (av, SvREFCNT_inc (SvRV (coro_current)));
2061	return 1;	2435	return 1;
2062	}	2436	}
2063	}	2437	}
2064		2438
2065	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
2066	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)
2067	{	2453	{
2068	AV *av = (AV *)SvRV (arg [0]);	2454	AV *av = (AV *)SvRV (arg [0]);
2069		2455
2070	if (SvIVX (AvARRAY (av)[0]) > 0)	2456	if (SvIVX (AvARRAY (av)[0]) > 0)
2071	{	2457	{
2072	frame->data = (void *)av;	2458	frame->data = (void *)av;
2073	frame->prepare = prepare_nop;	2459	frame->prepare = prepare_nop;
2074	SvSTATE_current->on_destroy = coro_semaphore_on_destroy;
2075	}	2460	}
2076	else	2461	else
2077	{	2462	{
2078	av_push (av, SvREFCNT_inc (SvRV (coro_current)));	2463	av_push (av, SvREFCNT_inc (SvRV (coro_current)));
2079		2464
2080	frame->data = (void *)sv_2mortal (SvREFCNT_inc ((SV *)av));	2465	frame->data = (void *)sv_2mortal (SvREFCNT_inc ((SV *)av));
2081	frame->prepare = prepare_schedule;	2466	frame->prepare = prepare_schedule;
2082		2467
2083	/* to avoid race conditions when a woken-up coro gets terminated */	2468	/* to avoid race conditions when a woken-up coro gets terminated */
2084	/* we arrange for a temporary on_destroy that calls adjust (0) */	2469	/* we arrange for a temporary on_destroy that calls adjust (0) */
2085	assert (!SvSTATE_current->on_destroy);//D
2086	SvSTATE_current->on_destroy = coro_semaphore_on_destroy;	2470	SvSTATE_current->on_destroy = coro_semaphore_on_destroy;
2087	}	2471	}
		2472	}
2088		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);
2089	frame->check = slf_check_semaphore_down;	2478	frame->check = slf_check_semaphore_down;
2090
2091	}	2479	}
2092		2480
2093	/*****************************************************************************/	2481	static void
2094	/* gensub: simple closure generation utility */	2482	slf_init_semaphore_wait (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
2095
2096	#define GENSUB_ARG CvXSUBANY (cv).any_ptr
2097
2098	/* create a closure from XS, returns a code reference */
2099	/* the arg can be accessed via GENSUB_ARG from the callback */
2100	/* the callback must use dXSARGS/XSRETURN */
2101	static SV *
2102	gensub (pTHX_ void (*xsub)(pTHX_ CV *), void *arg)
2103	{	2483	{
2104	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]);
2105		2489
2106	sv_upgrade ((SV *)cv, SVt_PVCV);	2490	av_push (av, (SV *)SvREFCNT_inc_NN (cb_cv));
2107		2491
2108	CvANON_on (cv);	2492	if (SvIVX (AvARRAY (av)[0]) > 0)
2109	CvISXSUB_on (cv);	2493	coro_semaphore_adjust (aTHX_ av, 0);
2110	CvXSUB (cv) = xsub;
2111	GENSUB_ARG = arg;
2112		2494
2113	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	}
2114	}	2560	}
2115		2561
2116	/*****************************************************************************/	2562	/*****************************************************************************/
2117	/* Coro::AIO */	2563	/* Coro::AIO */
2118		2564
…		…
2133	dXSARGS;	2579	dXSARGS;
2134	AV *state = (AV *)GENSUB_ARG;	2580	AV *state = (AV *)GENSUB_ARG;
2135	SV *coro = av_pop (state);	2581	SV *coro = av_pop (state);
2136	SV *data_sv = newSV (sizeof (struct io_state));	2582	SV *data_sv = newSV (sizeof (struct io_state));
2137		2583
2138	av_extend (state, items);	2584	av_extend (state, items - 1);
2139		2585
2140	sv_upgrade (data_sv, SVt_PV);	2586	sv_upgrade (data_sv, SVt_PV);
2141	SvCUR_set (data_sv, sizeof (struct io_state));	2587	SvCUR_set (data_sv, sizeof (struct io_state));
2142	SvPOK_only (data_sv);	2588	SvPOK_only (data_sv);
2143		2589
…		…
2168	static int	2614	static int
2169	slf_check_aio_req (pTHX_ struct CoroSLF *frame)	2615	slf_check_aio_req (pTHX_ struct CoroSLF *frame)
2170	{	2616	{
2171	AV *state = (AV *)frame->data;	2617	AV *state = (AV *)frame->data;
2172		2618
		2619	/* if we are about to throw, return early */
		2620	/* this does not cancel the aio request, but at least */
		2621	/* it quickly returns */
		2622	if (CORO_THROW)
		2623	return 0;
		2624
2173	/* one element that is an RV? repeat! */	2625	/* one element that is an RV? repeat! */
2174	if (AvFILLp (state) == 0 && SvROK (AvARRAY (state)[0]))	2626	if (AvFILLp (state) == 0 && SvROK (AvARRAY (state)[0]))
2175	return 1;	2627	return 1;
2176		2628
2177	/* restore status */	2629	/* restore status */
…		…
2247		2699
2248	for (i = 0; i < items; ++i)	2700	for (i = 0; i < items; ++i)
2249	PUSHs (arg [i]);	2701	PUSHs (arg [i]);
2250		2702
2251	/* now push the callback closure */	2703	/* now push the callback closure */
2252	PUSHs (sv_2mortal (gensub (coro_aio_callback, (void *)SvREFCNT_inc_NN ((SV *)state))));	2704	PUSHs (sv_2mortal (gensub (aTHX_ coro_aio_callback, (void *)SvREFCNT_inc_NN ((SV *)state))));
2253		2705
2254	/* now call the AIO function - we assume our request is uncancelable */	2706	/* now call the AIO function - we assume our request is uncancelable */
2255	PUTBACK;	2707	PUTBACK;
2256	call_sv ((SV *)req, G_VOID | G_DISCARD);	2708	call_sv ((SV *)req, G_VOID | G_DISCARD);
2257	}	2709	}
…		…
2263	}	2715	}
2264		2716
2265	static void	2717	static void
2266	coro_aio_req_xs (pTHX_ CV *cv)	2718	coro_aio_req_xs (pTHX_ CV *cv)
2267	{	2719	{
2268	dVAR;
2269	dXSARGS;	2720	dXSARGS;
2270		2721
2271	CORO_EXECUTE_SLF_XS (slf_init_aio_req);	2722	CORO_EXECUTE_SLF_XS (slf_init_aio_req);
2272		2723
2273	XSRETURN_EMPTY;	2724	XSRETURN_EMPTY;
…		…
2348	assert (("PRIO_NORMAL must be 0", !PRIO_NORMAL));	2799	assert (("PRIO_NORMAL must be 0", !PRIO_NORMAL));
2349	}	2800	}
2350		2801
2351	SV *	2802	SV *
2352	new (char *klass, ...)	2803	new (char *klass, ...)
		2804	ALIAS:
		2805	Coro::new = 1
2353	CODE:	2806	CODE:
2354	{	2807	{
2355	struct coro *coro;	2808	struct coro *coro;
2356	MAGIC *mg;	2809	MAGIC *mg;
2357	HV *hv;	2810	HV *hv;
		2811	CV *cb;
2358	int i;	2812	int i;
		2813
		2814	if (items > 1)
		2815	{
		2816	cb = coro_sv_2cv (aTHX_ ST (1));
		2817
		2818	if (!ix)
		2819	{
		2820	if (CvISXSUB (cb))
		2821	croak ("Coro::State doesn't support XS functions as coroutine start, caught");
		2822
		2823	if (!CvROOT (cb))
		2824	croak ("Coro::State doesn't support autoloaded or undefined functions as coroutine start, caught");
		2825	}
		2826	}
2359		2827
2360	Newz (0, coro, 1, struct coro);	2828	Newz (0, coro, 1, struct coro);
2361	coro->args = newAV ();	2829	coro->args = newAV ();
2362	coro->flags = CF_NEW;	2830	coro->flags = CF_NEW;
2363		2831
…		…
2368	coro->hv = hv = newHV ();	2836	coro->hv = hv = newHV ();
2369	mg = sv_magicext ((SV *)hv, 0, CORO_MAGIC_type_state, &coro_state_vtbl, (char *)coro, 0);	2837	mg = sv_magicext ((SV *)hv, 0, CORO_MAGIC_type_state, &coro_state_vtbl, (char *)coro, 0);
2370	mg->mg_flags |= MGf_DUP;	2838	mg->mg_flags |= MGf_DUP;
2371	RETVAL = sv_bless (newRV_noinc ((SV *)hv), gv_stashpv (klass, 1));	2839	RETVAL = sv_bless (newRV_noinc ((SV *)hv), gv_stashpv (klass, 1));
2372		2840
		2841	if (items > 1)
		2842	{
2373	av_extend (coro->args, items - 1);	2843	av_extend (coro->args, items - 1 + ix - 1);
		2844
		2845	if (ix)
		2846	{
		2847	av_push (coro->args, SvREFCNT_inc_NN ((SV *)cb));
		2848	cb = cv_coro_run;
		2849	}
		2850
		2851	coro->startcv = (CV *)SvREFCNT_inc_NN ((SV *)cb);
		2852
2374	for (i = 1; i < items; i++)	2853	for (i = 2; i < items; i++)
2375	av_push (coro->args, newSVsv (ST (i)));	2854	av_push (coro->args, newSVsv (ST (i)));
		2855	}
2376	}	2856	}
2377	OUTPUT:	2857	OUTPUT:
2378	RETVAL	2858	RETVAL
2379
2380	void
2381	_set_stacklevel (...)
2382	CODE:
2383	CORO_EXECUTE_SLF_XS (slf_init_set_stacklevel);
2384		2859
2385	void	2860	void
2386	transfer (...)	2861	transfer (...)
2387	PROTOTYPE: $$	2862	PROTOTYPE: $$
2388	CODE:	2863	CODE:
…		…
2514	throw (Coro::State self, SV *throw = &PL_sv_undef)	2989	throw (Coro::State self, SV *throw = &PL_sv_undef)
2515	PROTOTYPE: $;$	2990	PROTOTYPE: $;$
2516	CODE:	2991	CODE:
2517	{	2992	{
2518	struct coro *current = SvSTATE_current;	2993	struct coro *current = SvSTATE_current;
2519	SV **throwp = self == current ? &coro_throw : &self->throw;	2994	SV **throwp = self == current ? &CORO_THROW : &self->except;
2520	SvREFCNT_dec (*throwp);	2995	SvREFCNT_dec (*throwp);
2521	*throwp = SvOK (throw) ? newSVsv (throw) : 0;	2996	*throwp = SvOK (throw) ? newSVsv (throw) : 0;
2522	}	2997	}
2523		2998
2524	void	2999	void
…		…
2583		3058
2584	BOOT:	3059	BOOT:
2585	{	3060	{
2586	int i;	3061	int i;
2587		3062
2588	av_async_pool = coro_get_av (aTHX_ "Coro::async_pool", TRUE);
2589	sv_pool_rss = coro_get_sv (aTHX_ "Coro::POOL_RSS" , TRUE);	3063	sv_pool_rss = coro_get_sv (aTHX_ "Coro::POOL_RSS" , TRUE);
2590	sv_pool_size = coro_get_sv (aTHX_ "Coro::POOL_SIZE" , TRUE);	3064	sv_pool_size = coro_get_sv (aTHX_ "Coro::POOL_SIZE" , TRUE);
2591		3065	cv_coro_run = get_cv ( "Coro::_terminate", GV_ADD);
		3066	cv_coro_terminate = get_cv ( "Coro::terminate" , GV_ADD);
2592	coro_current = coro_get_sv (aTHX_ "Coro::current", FALSE);	3067	coro_current = coro_get_sv (aTHX_ "Coro::current" , FALSE); SvREADONLY_on (coro_current);
2593	SvREADONLY_on (coro_current);	3068	av_async_pool = coro_get_av (aTHX_ "Coro::async_pool", TRUE);
		3069	av_destroy = coro_get_av (aTHX_ "Coro::destroy" , TRUE);
		3070	sv_manager = coro_get_sv (aTHX_ "Coro::manager" , TRUE);
		3071
		3072	sv_async_pool_idle = newSVpv ("[async pool idle]", 0); SvREADONLY_on (sv_async_pool_idle);
		3073	sv_Coro = newSVpv ("Coro", 0); SvREADONLY_on (sv_Coro);
		3074	cv_pool_handler = get_cv ("Coro::pool_handler", GV_ADD); SvREADONLY_on (cv_pool_handler);
		3075	cv_coro_state_new = get_cv ("Coro::State::new", 0); SvREADONLY_on (cv_coro_state_new);
2594		3076
2595	coro_stash = gv_stashpv ("Coro", TRUE);	3077	coro_stash = gv_stashpv ("Coro", TRUE);
2596		3078
2597	newCONSTSUB (coro_stash, "PRIO_MAX", newSViv (PRIO_MAX));	3079	newCONSTSUB (coro_stash, "PRIO_MAX", newSViv (PRIO_MAX));
2598	newCONSTSUB (coro_stash, "PRIO_HIGH", newSViv (PRIO_HIGH));	3080	newCONSTSUB (coro_stash, "PRIO_HIGH", newSViv (PRIO_HIGH));
…		…
2606		3088
2607	{	3089	{
2608	SV *sv = coro_get_sv (aTHX_ "Coro::API", TRUE);	3090	SV *sv = coro_get_sv (aTHX_ "Coro::API", TRUE);
2609		3091
2610	coroapi.schedule = api_schedule;	3092	coroapi.schedule = api_schedule;
		3093	coroapi.schedule_to = api_schedule_to;
2611	coroapi.cede = api_cede;	3094	coroapi.cede = api_cede;
2612	coroapi.cede_notself = api_cede_notself;	3095	coroapi.cede_notself = api_cede_notself;
2613	coroapi.ready = api_ready;	3096	coroapi.ready = api_ready;
2614	coroapi.is_ready = api_is_ready;	3097	coroapi.is_ready = api_is_ready;
2615	coroapi.nready = coro_nready;	3098	coroapi.nready = coro_nready;
2616	coroapi.current = coro_current;	3099	coroapi.current = coro_current;
2617		3100
2618	GCoroAPI = &coroapi;	3101	/*GCoroAPI = &coroapi;*/
2619	sv_setiv (sv, (IV)&coroapi);	3102	sv_setiv (sv, (IV)&coroapi);
2620	SvREADONLY_on (sv);	3103	SvREADONLY_on (sv);
2621	}	3104	}
2622	}	3105	}
		3106
		3107	void
		3108	terminate (...)
		3109	CODE:
		3110	CORO_EXECUTE_SLF_XS (slf_init_terminate);
2623		3111
2624	void	3112	void
2625	schedule (...)	3113	schedule (...)
2626	CODE:	3114	CODE:
2627	CORO_EXECUTE_SLF_XS (slf_init_schedule);	3115	CORO_EXECUTE_SLF_XS (slf_init_schedule);
2628		3116
2629	void	3117	void
		3118	schedule_to (...)
		3119	CODE:
		3120	CORO_EXECUTE_SLF_XS (slf_init_schedule_to);
		3121
		3122	void
		3123	cede_to (...)
		3124	CODE:
		3125	CORO_EXECUTE_SLF_XS (slf_init_cede_to);
		3126
		3127	void
2630	cede (...)	3128	cede (...)
2631	CODE:	3129	CODE:
2632	CORO_EXECUTE_SLF_XS (slf_init_cede);	3130	CORO_EXECUTE_SLF_XS (slf_init_cede);
2633		3131
2634	void	3132	void
2635	cede_notself (...)	3133	cede_notself (...)
2636	CODE:	3134	CODE:
2637	CORO_EXECUTE_SLF_XS (slf_init_cede_notself);	3135	CORO_EXECUTE_SLF_XS (slf_init_cede_notself);
		3136
		3137	void
		3138	_cancel (Coro::State self)
		3139	CODE:
		3140	coro_state_destroy (aTHX_ self);
		3141	coro_call_on_destroy (aTHX_ self);
2638		3142
2639	void	3143	void
2640	_set_current (SV *current)	3144	_set_current (SV *current)
2641	PROTOTYPE: $	3145	PROTOTYPE: $
2642	CODE:	3146	CODE:
…		…
2687	CODE:	3191	CODE:
2688	RETVAL = coro_nready;	3192	RETVAL = coro_nready;
2689	OUTPUT:	3193	OUTPUT:
2690	RETVAL	3194	RETVAL
2691		3195
2692	# for async_pool speedup
2693	void	3196	void
2694	_pool_1 (SV *cb)	3197	_pool_handler (...)
2695	CODE:	3198	CODE:
2696	{	3199	CORO_EXECUTE_SLF_XS (slf_init_pool_handler);
2697	HV *hv = (HV *)SvRV (coro_current);
2698	struct coro *coro = SvSTATE_hv ((SV *)hv);
2699	AV *defav = GvAV (PL_defgv);
2700	SV *invoke = hv_delete (hv, "_invoke", sizeof ("_invoke") - 1, 0);
2701	AV *invoke_av;
2702	int i, len;
2703		3200
2704	if (!invoke)	3201	void
		3202	async_pool (SV *cv, ...)
		3203	PROTOTYPE: &@
		3204	PPCODE:
		3205	{
		3206	HV *hv = (HV *)av_pop (av_async_pool);
		3207	AV *av = newAV ();
		3208	SV *cb = ST (0);
		3209	int i;
		3210
		3211	av_extend (av, items - 2);
		3212	for (i = 1; i < items; ++i)
		3213	av_push (av, SvREFCNT_inc_NN (ST (i)));
		3214
		3215	if ((SV *)hv == &PL_sv_undef)
2705	{	3216	{
2706	SV *old = PL_diehook;	3217	PUSHMARK (SP);
2707	PL_diehook = 0;	3218	EXTEND (SP, 2);
2708	SvREFCNT_dec (old);	3219	PUSHs (sv_Coro);
2709	croak ("\3async_pool terminate\2\n");	3220	PUSHs ((SV *)cv_pool_handler);
		3221	PUTBACK;
		3222	call_sv ((SV *)cv_coro_state_new, G_SCALAR);
		3223	SPAGAIN;
		3224
		3225	hv = (HV *)SvREFCNT_inc_NN (SvRV (POPs));
2710	}	3226	}
2711		3227
2712	SvREFCNT_dec (coro->saved_deffh);
2713	coro->saved_deffh = SvREFCNT_inc_NN ((SV *)PL_defoutgv);
2714
2715	hv_store (hv, "desc", sizeof ("desc") - 1,
2716	newSVpvn ("[async_pool]", sizeof ("[async_pool]") - 1), 0);
2717
2718	invoke_av = (AV *)SvRV (invoke);
2719	len = av_len (invoke_av);
2720
2721	sv_setsv (cb, AvARRAY (invoke_av)[0]);
2722
2723	if (len > 0)
2724	{	3228	{
2725	av_fill (defav, len - 1);	3229	struct coro *coro = SvSTATE_hv (hv);
2726	for (i = 0; i < len; ++i)	3230
2727	av_store (defav, i, SvREFCNT_inc_NN (AvARRAY (invoke_av)[i + 1]));	3231	assert (!coro->invoke_cb);
		3232	assert (!coro->invoke_av);
		3233	coro->invoke_cb = SvREFCNT_inc (cb);
		3234	coro->invoke_av = av;
2728	}	3235	}
		3236
		3237	api_ready (aTHX_ (SV *)hv);
		3238
		3239	if (GIMME_V != G_VOID)
		3240	XPUSHs (sv_2mortal (newRV_noinc ((SV *)hv)));
		3241	else
		3242	SvREFCNT_dec (hv);
2729	}	3243	}
		3244
		3245	SV *
		3246	rouse_cb ()
		3247	PROTOTYPE:
		3248	CODE:
		3249	RETVAL = coro_new_rouse_cb (aTHX);
		3250	OUTPUT:
		3251	RETVAL
2730		3252
2731	void	3253	void
2732	_pool_2 (SV *cb)	3254	rouse_wait (...)
		3255	PROTOTYPE: ;$
2733	CODE:	3256	PPCODE:
2734	{	3257	CORO_EXECUTE_SLF_XS (slf_init_rouse_wait);
2735	HV *hv = (HV *)SvRV (coro_current);
2736	struct coro *coro = SvSTATE_hv ((SV *)hv);
2737
2738	sv_setsv (cb, &PL_sv_undef);
2739
2740	SvREFCNT_dec ((SV *)PL_defoutgv); PL_defoutgv = (GV *)coro->saved_deffh;
2741	coro->saved_deffh = 0;
2742
2743	if (coro_rss (aTHX_ coro) > SvUV (sv_pool_rss)
2744	|| av_len (av_async_pool) + 1 >= SvIV (sv_pool_size))
2745	{
2746	SV *old = PL_diehook;
2747	PL_diehook = 0;
2748	SvREFCNT_dec (old);
2749	croak ("\3async_pool terminate\2\n");
2750	}
2751
2752	av_clear (GvAV (PL_defgv));
2753	hv_store (hv, "desc", sizeof ("desc") - 1,
2754	newSVpvn ("[async_pool idle]", sizeof ("[async_pool idle]") - 1), 0);
2755
2756	coro->prio = 0;
2757
2758	if (coro->cctx && (coro->cctx->flags & CC_TRACE))
2759	api_trace (aTHX_ coro_current, 0);
2760
2761	av_push (av_async_pool, newSVsv (coro_current));
2762	}
2763		3258
2764		3259
2765	MODULE = Coro::State PACKAGE = PerlIO::cede	3260	MODULE = Coro::State PACKAGE = PerlIO::cede
2766		3261
2767	BOOT:	3262	BOOT:
…		…
2769		3264
2770		3265
2771	MODULE = Coro::State PACKAGE = Coro::Semaphore	3266	MODULE = Coro::State PACKAGE = Coro::Semaphore
2772		3267
2773	SV *	3268	SV *
2774	new (SV *klass, SV *count_ = 0)	3269	new (SV *klass, SV *count = 0)
2775	CODE:	3270	CODE:
2776	{	3271	RETVAL = sv_bless (
2777	/* a semaphore contains a counter IV in $sem->[0] and any waiters after that */	3272	coro_waitarray_new (aTHX_ count && SvOK (count) ? SvIV (count) : 1),
2778	AV *av = newAV ();	3273	GvSTASH (CvGV (cv))
2779	av_push (av, newSViv (count_ && SvOK (count_) ? SvIV (count_) : 1));	3274	);
2780	RETVAL = sv_bless (newRV_noinc ((SV *)av), GvSTASH (CvGV (cv)));	3275	OUTPUT:
2781	}	3276	RETVAL
		3277
		3278	# helper for Coro::Channel
		3279	SV *
		3280	_alloc (int count)
		3281	CODE:
		3282	RETVAL = coro_waitarray_new (aTHX_ count);
2782	OUTPUT:	3283	OUTPUT:
2783	RETVAL	3284	RETVAL
2784		3285
2785	SV *	3286	SV *
2786	count (SV *self)	3287	count (SV *self)
…		…
2795	adjust = 1	3296	adjust = 1
2796	CODE:	3297	CODE:
2797	coro_semaphore_adjust (aTHX_ (AV *)SvRV (self), ix ? adjust : 1);	3298	coro_semaphore_adjust (aTHX_ (AV *)SvRV (self), ix ? adjust : 1);
2798		3299
2799	void	3300	void
2800	down (SV *self)	3301	down (...)
2801	CODE:	3302	CODE:
2802	CORO_EXECUTE_SLF_XS (slf_init_semaphore_down);	3303	CORO_EXECUTE_SLF_XS (slf_init_semaphore_down);
		3304
		3305	void
		3306	wait (...)
		3307	CODE:
		3308	CORO_EXECUTE_SLF_XS (slf_init_semaphore_wait);
2803		3309
2804	void	3310	void
2805	try (SV *self)	3311	try (SV *self)
2806	PPCODE:	3312	PPCODE:
2807	{	3313	{
…		…
2819	XSRETURN_NO;	3325	XSRETURN_NO;
2820	}	3326	}
2821		3327
2822	void	3328	void
2823	waiters (SV *self)	3329	waiters (SV *self)
2824	CODE:	3330	PPCODE:
2825	{	3331	{
2826	AV *av = (AV *)SvRV (self);	3332	AV *av = (AV *)SvRV (self);
		3333	int wcount = AvFILLp (av) + 1 - 1;
2827		3334
2828	if (GIMME_V == G_SCALAR)	3335	if (GIMME_V == G_SCALAR)
2829	XPUSHs (sv_2mortal (newSVsv (AvARRAY (av)[0])));	3336	XPUSHs (sv_2mortal (newSViv (wcount)));
2830	else	3337	else
2831	{	3338	{
2832	int i;	3339	int i;
2833	EXTEND (SP, AvFILLp (av) + 1 - 1);	3340	EXTEND (SP, wcount);
2834	for (i = 1; i <= AvFILLp (av); ++i)	3341	for (i = 1; i <= wcount; ++i)
2835	PUSHs (newSVsv (AvARRAY (av)[i]));	3342	PUSHs (sv_2mortal (newRV_inc (AvARRAY (av)[i])));
2836	}	3343	}
2837	}	3344	}
		3345
		3346	MODULE = Coro::State PACKAGE = Coro::Signal
		3347
		3348	SV *
		3349	new (SV *klass)
		3350	CODE:
		3351	RETVAL = sv_bless (
		3352	coro_waitarray_new (aTHX_ 0),
		3353	GvSTASH (CvGV (cv))
		3354	);
		3355	OUTPUT:
		3356	RETVAL
		3357
		3358	void
		3359	wait (...)
		3360	CODE:
		3361	CORO_EXECUTE_SLF_XS (slf_init_signal_wait);
		3362
		3363	void
		3364	broadcast (SV *self)
		3365	CODE:
		3366	{
		3367	AV *av = (AV *)SvRV (self);
		3368	coro_signal_wake (aTHX_ av, AvFILLp (av));
		3369	}
		3370
		3371	void
		3372	send (SV *self)
		3373	CODE:
		3374	{
		3375	AV *av = (AV *)SvRV (self);
		3376
		3377	if (AvFILLp (av))
		3378	coro_signal_wake (aTHX_ av, 1);
		3379	else
		3380	SvIVX (AvARRAY (av)[0]) = 1; /* remember the signal */
		3381	}
		3382
		3383	IV
		3384	awaited (SV *self)
		3385	CODE:
		3386	RETVAL = AvFILLp ((AV *)SvRV (self)) + 1 - 1;
		3387	OUTPUT:
		3388	RETVAL
2838		3389
2839		3390
2840	MODULE = Coro::State PACKAGE = Coro::AnyEvent	3391	MODULE = Coro::State PACKAGE = Coro::AnyEvent
2841		3392
2842	BOOT:	3393	BOOT:
…		…
2857	sv_setsv (sv_activity, &PL_sv_undef);	3408	sv_setsv (sv_activity, &PL_sv_undef);
2858	if (coro_nready >= incede)	3409	if (coro_nready >= incede)
2859	{	3410	{
2860	PUSHMARK (SP);	3411	PUSHMARK (SP);
2861	PUTBACK;	3412	PUTBACK;
2862	call_pv ("Coro::AnyEvent::_activity", G_DISCARD | G_EVAL);	3413	call_pv ("Coro::AnyEvent::_activity", G_KEEPERR | G_EVAL | G_VOID | G_DISCARD);
2863	SPAGAIN;
2864	}	3414	}
2865		3415
2866	--incede;	3416	--incede;
2867	}	3417	}
2868		3418
…		…
2871		3421
2872	void	3422	void
2873	_register (char *target, char *proto, SV *req)	3423	_register (char *target, char *proto, SV *req)
2874	CODE:	3424	CODE:
2875	{	3425	{
2876	HV *st;	3426	CV *req_cv = coro_sv_2cv (aTHX_ req);
2877	GV *gvp;	3427	/* newXSproto doesn't return the CV on 5.8 */
2878	CV *req_cv = sv_2cv (req, &st, &gvp, 0);
2879	CV *slf_cv = newXSproto (target, coro_aio_req_xs, __FILE__, proto);	3428	CV *slf_cv = newXS (target, coro_aio_req_xs, __FILE__);
		3429	sv_setpv ((SV *)slf_cv, proto);
2880	sv_magicext ((SV *)slf_cv, (SV *)req_cv, CORO_MAGIC_type_aio, 0, 0, 0);	3430	sv_magicext ((SV *)slf_cv, (SV *)req_cv, CORO_MAGIC_type_aio, 0, 0, 0);
2881	}	3431	}
2882		3432

Diff Legend

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