ViewVC Help

View File | Revision Log | Show Annotations | Download File

/cvs/Coro/Coro/State.xs

Comparing Coro/Coro/State.xs (file contents):
Revision 1.285 by root, Mon Nov 17 04:17:20 2008 UTC vs.
Revision 1.314 by root, Thu Nov 20 03:24:39 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	/* helper storage struct for Coro::AIO */
153	struct io_state
154	{
155	AV *res;
156	int errorno;
157	I32 laststype; /* U16 in 5.10.0 */
158	int laststatval;
159	Stat_t statcache;
160	};
161
162	static double (*nvtime)(); /* so why doesn't it take void? */	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);
163		166
164	static U32 cctx_gen;	167	static U32 cctx_gen;
165	static size_t cctx_stacksize = CORO_STACKSIZE;	168	static size_t cctx_stacksize = CORO_STACKSIZE;
166	static struct CoroAPI coroapi;	169	static struct CoroAPI coroapi;
167	static AV *main_mainstack; /* used to differentiate between $main and others */	170	static AV *main_mainstack; /* used to differentiate between $main and others */
…		…
176	static HV *hv_sig; /* %SIG */	179	static HV *hv_sig; /* %SIG */
177		180
178	/* async_pool helper stuff */	181	/* async_pool helper stuff */
179	static SV *sv_pool_rss;	182	static SV *sv_pool_rss;
180	static SV *sv_pool_size;	183	static SV *sv_pool_size;
		184	static SV *sv_async_pool_idle;
181	static AV *av_async_pool;	185	static AV *av_async_pool;
		186	static SV *sv_Coro;
		187	static CV *cv_pool_handler;
		188	static CV *cv_coro_new;
182		189
183	/* Coro::AnyEvent */	190	/* Coro::AnyEvent */
184	static SV *sv_activity;	191	static SV *sv_activity;
185		192
186	static struct coro_cctx *cctx_first;	193	static struct coro_cctx *cctx_first;
…		…
246	/* state data */	253	/* state data */
247	struct CoroSLF slf_frame; /* saved slf frame */	254	struct CoroSLF slf_frame; /* saved slf frame */
248	AV *mainstack;	255	AV *mainstack;
249	perl_slots *slot; /* basically the saved sp */	256	perl_slots *slot; /* basically the saved sp */
250		257
		258	CV *startcv; /* the CV to execute */
251	AV *args; /* data associated with this coroutine (initial args) */	259	AV *args; /* data associated with this coroutine (initial args) */
252	int refcnt; /* coroutines are refcounted, yes */	260	int refcnt; /* coroutines are refcounted, yes */
253	int flags; /* CF_ flags */	261	int flags; /* CF_ flags */
254	HV *hv; /* the perl hash associated with this coro, if any */	262	HV *hv; /* the perl hash associated with this coro, if any */
255	void (*on_destroy)(pTHX_ struct coro *coro);	263	void (*on_destroy)(pTHX_ struct coro *coro);
256		264
257	/* statistics */	265	/* statistics */
258	int usecount; /* number of transfers to this coro */	266	int usecount; /* number of transfers to this coro */
259		267
260	/* coro process data */	268	/* coro process data */
261	int prio;	269	int prio;
262	SV *throw; /* exception to be thrown */	270	SV *except; /* exception to be thrown */
		271	SV *rouse_cb;
263		272
264	/* async_pool */	273	/* async_pool */
265	SV *saved_deffh;	274	SV *saved_deffh;
		275	SV *invoke_cb;
		276	AV *invoke_av;
266		277
267	/* linked list */	278	/* linked list */
268	struct coro *next, *prev;	279	struct coro *next, *prev;
269	};	280	};
270		281
…		…
273		284
274	/* the following variables are effectively part of the perl context */	285	/* the following variables are effectively part of the perl context */
275	/* and get copied between struct coro and these variables */	286	/* and get copied between struct coro and these variables */
276	/* the mainr easonw e don't support windows process emulation */	287	/* the mainr easonw e don't support windows process emulation */
277	static struct CoroSLF slf_frame; /* the current slf frame */	288	static struct CoroSLF slf_frame; /* the current slf frame */
278	static SV *coro_throw;
279		289
280	/** Coro ********************************************************************/	290	/** Coro ********************************************************************/
281		291
282	#define PRIO_MAX 3	292	#define PRIO_MAX 3
283	#define PRIO_HIGH 1	293	#define PRIO_HIGH 1
…		…
288		298
289	/* for Coro.pm */	299	/* for Coro.pm */
290	static SV *coro_current;	300	static SV *coro_current;
291	static SV *coro_readyhook;	301	static SV *coro_readyhook;
292	static AV *coro_ready [PRIO_MAX - PRIO_MIN + 1];	302	static AV *coro_ready [PRIO_MAX - PRIO_MIN + 1];
		303	static CV *cv_coro_run, *cv_coro_terminate;
293	static struct coro *coro_first;	304	static struct coro *coro_first;
294	#define coro_nready coroapi.nready	305	#define coro_nready coroapi.nready
295		306
296	/** lowlevel stuff **********************************************************/	307	/** lowlevel stuff **********************************************************/
297		308
…		…
321	#if PERL_VERSION_ATLEAST (5,10,0)	332	#if PERL_VERSION_ATLEAST (5,10,0)
322	/* silence stupid and wrong 5.10 warning that I am unable to switch off */	333	/* silence stupid and wrong 5.10 warning that I am unable to switch off */
323	get_hv (name, create);	334	get_hv (name, create);
324	#endif	335	#endif
325	return get_hv (name, create);	336	return get_hv (name, create);
		337	}
		338
		339	/* may croak */
		340	INLINE CV *
		341	coro_sv_2cv (pTHX_ SV *sv)
		342	{
		343	HV *st;
		344	GV *gvp;
		345	return sv_2cv (sv, &st, &gvp, 0);
326	}	346	}
327		347
328	static AV *	348	static AV *
329	coro_clone_padlist (pTHX_ CV *cv)	349	coro_clone_padlist (pTHX_ CV *cv)
330	{	350	{
…		…
384	SvREFCNT_dec (av); /* sv_magicext increased the refcount */	404	SvREFCNT_dec (av); /* sv_magicext increased the refcount */
385		405
386	return 0;	406	return 0;
387	}	407	}
388		408
389	#define CORO_MAGIC_type_cv PERL_MAGIC_ext	409	#define CORO_MAGIC_type_cv 26
390	#define CORO_MAGIC_type_state PERL_MAGIC_ext	410	#define CORO_MAGIC_type_state PERL_MAGIC_ext
391		411
392	static MGVTBL coro_cv_vtbl = {	412	static MGVTBL coro_cv_vtbl = {
393	0, 0, 0, 0,	413	0, 0, 0, 0,
394	coro_cv_free	414	coro_cv_free
395	};	415	};
396		416
		417	#define CORO_MAGIC_NN(sv, type) \
		418	(expect_true (SvMAGIC (sv)->mg_type == type) \
		419	? SvMAGIC (sv) \
		420	: mg_find (sv, type))
		421
397	#define CORO_MAGIC(sv, type) \	422	#define CORO_MAGIC(sv, type) \
398	expect_true (SvMAGIC (sv)) \	423	(expect_true (SvMAGIC (sv)) \
399	? expect_true (SvMAGIC (sv)->mg_type == type) \	424	? CORO_MAGIC_NN (sv, type) \
400	? SvMAGIC (sv) \
401	: mg_find (sv, type) \
402	: 0	425	: 0)
403		426
404	#define CORO_MAGIC_cv(cv) CORO_MAGIC (((SV *)(cv)), CORO_MAGIC_type_cv)	427	#define CORO_MAGIC_cv(cv) CORO_MAGIC (((SV *)(cv)), CORO_MAGIC_type_cv)
405	#define CORO_MAGIC_state(sv) CORO_MAGIC (((SV *)(sv)), CORO_MAGIC_type_state)	428	#define CORO_MAGIC_state(sv) CORO_MAGIC_NN (((SV *)(sv)), CORO_MAGIC_type_state)
406		429
407	INLINE struct coro *	430	INLINE struct coro *
408	SvSTATE_ (pTHX_ SV *coro)	431	SvSTATE_ (pTHX_ SV *coro)
409	{	432	{
410	HV *stash;	433	HV *stash;
…		…
428	return (struct coro *)mg->mg_ptr;	451	return (struct coro *)mg->mg_ptr;
429	}	452	}
430		453
431	#define SvSTATE(sv) SvSTATE_ (aTHX_ (sv))	454	#define SvSTATE(sv) SvSTATE_ (aTHX_ (sv))
432		455
433	/* fastert than SvSTATE, but expects a coroutine hv */	456	/* faster than SvSTATE, but expects a coroutine hv */
434	INLINE struct coro *	457	#define SvSTATE_hv(hv) ((struct coro *)CORO_MAGIC_NN ((SV *)hv, CORO_MAGIC_type_state)->mg_ptr)
435	SvSTATE_hv (SV *sv)
436	{
437	MAGIC *mg = expect_true (SvMAGIC (sv)->mg_type == CORO_MAGIC_type_state)
438	? SvMAGIC (sv)
439	: mg_find (sv, CORO_MAGIC_type_state);
440
441	return (struct coro *)mg->mg_ptr;
442	}
443
444	#define SvSTATE_current SvSTATE_hv (SvRV (coro_current))	458	#define SvSTATE_current SvSTATE_hv (SvRV (coro_current))
445		459
446	/* the next two functions merely cache the padlists */	460	/* the next two functions merely cache the padlists */
447	static void	461	static void
448	get_padlist (pTHX_ CV *cv)	462	get_padlist (pTHX_ CV *cv)
…		…
455	else	469	else
456	{	470	{
457	#if CORO_PREFER_PERL_FUNCTIONS	471	#if CORO_PREFER_PERL_FUNCTIONS
458	/* this is probably cleaner? but also slower! */	472	/* this is probably cleaner? but also slower! */
459	/* in practise, it seems to be less stable */	473	/* in practise, it seems to be less stable */
460	CV *cp = Perl_cv_clone (cv);	474	CV *cp = Perl_cv_clone (aTHX_ cv);
461	CvPADLIST (cv) = CvPADLIST (cp);	475	CvPADLIST (cv) = CvPADLIST (cp);
462	CvPADLIST (cp) = 0;	476	CvPADLIST (cp) = 0;
463	SvREFCNT_dec (cp);	477	SvREFCNT_dec (cp);
464	#else	478	#else
465	CvPADLIST (cv) = coro_clone_padlist (aTHX_ cv);	479	CvPADLIST (cv) = coro_clone_padlist (aTHX_ cv);
…		…
518		532
519	PUTBACK;	533	PUTBACK;
520	}	534	}
521		535
522	slf_frame = c->slf_frame;	536	slf_frame = c->slf_frame;
523	coro_throw = c->throw;	537	CORO_THROW = c->except;
524	}	538	}
525		539
526	static void	540	static void
527	save_perl (pTHX_ Coro__State c)	541	save_perl (pTHX_ Coro__State c)
528	{	542	{
529	c->throw = coro_throw;	543	c->except = CORO_THROW;
530	c->slf_frame = slf_frame;	544	c->slf_frame = slf_frame;
531		545
532	{	546	{
533	dSP;	547	dSP;
534	I32 cxix = cxstack_ix;	548	I32 cxix = cxstack_ix;
…		…
609	* of perl.c:init_stacks, except that it uses less memory	623	* of perl.c:init_stacks, except that it uses less memory
610	* on the (sometimes correct) assumption that coroutines do	624	* on the (sometimes correct) assumption that coroutines do
611	* not usually need a lot of stackspace.	625	* not usually need a lot of stackspace.
612	*/	626	*/
613	#if CORO_PREFER_PERL_FUNCTIONS	627	#if CORO_PREFER_PERL_FUNCTIONS
614	# define coro_init_stacks init_stacks	628	# define coro_init_stacks(thx) init_stacks ()
615	#else	629	#else
616	static void	630	static void
617	coro_init_stacks (pTHX)	631	coro_init_stacks (pTHX)
618	{	632	{
619	PL_curstackinfo = new_stackinfo(32, 8);	633	PL_curstackinfo = new_stackinfo(32, 8);
…		…
682	#if !PERL_VERSION_ATLEAST (5,10,0)	696	#if !PERL_VERSION_ATLEAST (5,10,0)
683	Safefree (PL_retstack);	697	Safefree (PL_retstack);
684	#endif	698	#endif
685	}	699	}
686		700
		701	#define CORO_RSS \
		702	rss += sizeof (SYM (curstackinfo)); \
		703	rss += (SYM (curstackinfo->si_cxmax) + 1) * sizeof (PERL_CONTEXT); \
		704	rss += sizeof (SV) + sizeof (struct xpvav) + (1 + AvMAX (SYM (curstack))) * sizeof (SV *); \
		705	rss += SYM (tmps_max) * sizeof (SV *); \
		706	rss += (SYM (markstack_max) - SYM (markstack_ptr)) * sizeof (I32); \
		707	rss += SYM (scopestack_max) * sizeof (I32); \
		708	rss += SYM (savestack_max) * sizeof (ANY);
		709
687	static size_t	710	static size_t
688	coro_rss (pTHX_ struct coro *coro)	711	coro_rss (pTHX_ struct coro *coro)
689	{	712	{
690	size_t rss = sizeof (*coro);	713	size_t rss = sizeof (*coro);
691		714
692	if (coro->mainstack)	715	if (coro->mainstack)
693	{	716	{
694	perl_slots tmp_slot;
695	perl_slots *slot;
696
697	if (coro->flags & CF_RUNNING)	717	if (coro->flags & CF_RUNNING)
698	{	718	{
699	slot = &tmp_slot;	719	#define SYM(sym) PL_ ## sym
700		720	CORO_RSS;
701	#define VAR(name,type) slot->name = PL_ ## name;
702	# include "state.h"
703	#undef VAR	721	#undef SYM
704	}	722	}
705	else	723	else
706	slot = coro->slot;
707
708	if (slot)
709	{	724	{
710	rss += sizeof (slot->curstackinfo);	725	#define SYM(sym) coro->slot->sym
711	rss += (slot->curstackinfo->si_cxmax + 1) * sizeof (PERL_CONTEXT);	726	CORO_RSS;
712	rss += sizeof (SV) + sizeof (struct xpvav) + (1 + AvMAX (slot->curstack)) * sizeof (SV *);	727	#undef SYM
713	rss += slot->tmps_max * sizeof (SV *);
714	rss += (slot->markstack_max - slot->markstack_ptr) * sizeof (I32);
715	rss += slot->scopestack_max * sizeof (I32);
716	rss += slot->savestack_max * sizeof (ANY);
717
718	#if !PERL_VERSION_ATLEAST (5,10,0)
719	rss += slot->retstack_max * sizeof (OP *);
720	#endif
721	}	728	}
722	}	729	}
723		730
724	return rss;	731	return rss;
725	}	732	}
…		…
827	slf_check_nop (pTHX_ struct CoroSLF *frame)	834	slf_check_nop (pTHX_ struct CoroSLF *frame)
828	{	835	{
829	return 0;	836	return 0;
830	}	837	}
831		838
		839	static UNOP coro_setup_op;
		840
832	static void NOINLINE /* noinline to keep it out of the transfer fast path */	841	static void NOINLINE /* noinline to keep it out of the transfer fast path */
833	coro_setup (pTHX_ struct coro *coro)	842	coro_setup (pTHX_ struct coro *coro)
834	{	843	{
835	/*	844	/*
836	* emulate part of the perl startup here.	845	* emulate part of the perl startup here.
…		…
864	{	873	{
865	dSP;	874	dSP;
866	UNOP myop;	875	UNOP myop;
867		876
868	Zero (&myop, 1, UNOP);	877	Zero (&myop, 1, UNOP);
869	myop.op_next = Nullop;	878	myop.op_next = Nullop;
		879	myop.op_type = OP_ENTERSUB;
870	myop.op_flags = OPf_WANT_VOID;	880	myop.op_flags = OPf_WANT_VOID;
871		881
872	PUSHMARK (SP);	882	PUSHMARK (SP);
873	XPUSHs (sv_2mortal (av_shift (GvAV (PL_defgv))));	883	PUSHs ((SV *)coro->startcv);
874	PUTBACK;	884	PUTBACK;
875	PL_op = (OP *)&myop;	885	PL_op = (OP *)&myop;
876	PL_op = PL_ppaddr[OP_ENTERSUB](aTHX);	886	PL_op = PL_ppaddr[OP_ENTERSUB](aTHX);
877	SPAGAIN;
878	}	887	}
879		888
880	/* this newly created coroutine might be run on an existing cctx which most	889	/* this newly created coroutine might be run on an existing cctx which most
881	* likely was suspended in pp_slf, so we have to emulate entering pp_slf here.	890	* likely was suspended in pp_slf, so we have to emulate entering pp_slf here.
882	*/	891	*/
883	slf_frame.prepare = prepare_nop; /* provide a nop function for an eventual pp_slf */	892	slf_frame.prepare = prepare_nop; /* provide a nop function for an eventual pp_slf */
884	slf_frame.check = slf_check_nop; /* signal pp_slf to not repeat */	893	slf_frame.check = slf_check_nop; /* signal pp_slf to not repeat */
885		894
886	coro_throw = coro->throw;	895	/* and we have to provide the pp_slf op in any case, so pp_slf can skip it */
		896	coro_setup_op.op_next = PL_op;
		897	coro_setup_op.op_type = OP_CUSTOM;
		898	coro_setup_op.op_ppaddr = pp_slf;
		899	/* no flags etc. required, as an init function won't be called */
		900
		901	PL_op = (OP *)&coro_setup_op;
		902
		903	/* copy throw, in case it was set before coro_setup */
		904	CORO_THROW = coro->except;
887	}	905	}
888		906
889	static void	907	static void
890	coro_destruct (pTHX_ struct coro *coro)	908	coro_destruct (pTHX_ struct coro *coro)
891	{	909	{
…		…
915		933
916	SvREFCNT_dec (PL_diehook);	934	SvREFCNT_dec (PL_diehook);
917	SvREFCNT_dec (PL_warnhook);	935	SvREFCNT_dec (PL_warnhook);
918		936
919	SvREFCNT_dec (coro->saved_deffh);	937	SvREFCNT_dec (coro->saved_deffh);
920	SvREFCNT_dec (coro_throw);	938	SvREFCNT_dec (coro->rouse_cb);
		939	SvREFCNT_dec (coro->invoke_cb);
		940	SvREFCNT_dec (coro->invoke_av);
921		941
922	coro_destruct_stacks (aTHX);	942	coro_destruct_stacks (aTHX);
923	}	943	}
924		944
925	INLINE void	945	INLINE void
…		…
1012	SAVETMPS;	1032	SAVETMPS;
1013	EXTEND (SP, 3);	1033	EXTEND (SP, 3);
1014	PUSHMARK (SP);	1034	PUSHMARK (SP);
1015	PUSHs (&PL_sv_yes);	1035	PUSHs (&PL_sv_yes);
1016	PUSHs (fullname);	1036	PUSHs (fullname);
1017	PUSHs (CxHASARGS (cx) ? sv_2mortal (newRV_inc ((SV *)cx->blk_sub.argarray)) : &PL_sv_undef);	1037	PUSHs (CxHASARGS (cx) ? sv_2mortal (newRV_inc ((SV *)cx->blk_sub.argarray)) : &PL_sv_undef);
1018	PUTBACK;	1038	PUTBACK;
1019	cb = hv_fetch ((HV *)SvRV (coro_current), "_trace_sub_cb", sizeof ("_trace_sub_cb") - 1, 0);	1039	cb = hv_fetch ((HV *)SvRV (coro_current), "_trace_sub_cb", sizeof ("_trace_sub_cb") - 1, 0);
1020	if (cb) call_sv (*cb, G_KEEPERR | G_EVAL | G_VOID | G_DISCARD);	1040	if (cb) call_sv (*cb, G_KEEPERR | G_EVAL | G_VOID | G_DISCARD);
1021	SPAGAIN;	1041	SPAGAIN;
1022	FREETMPS;	1042	FREETMPS;
…		…
1054		1074
1055	TAINT_NOT;	1075	TAINT_NOT;
1056	return 0;	1076	return 0;
1057	}	1077	}
1058		1078
		1079	static struct coro_cctx *cctx_ssl_cctx;
		1080	static struct CoroSLF cctx_ssl_frame;
		1081
1059	static void	1082	static void
1060	prepare_set_stacklevel (struct coro_transfer_args *ta, struct coro_cctx *cctx)	1083	slf_prepare_set_stacklevel (pTHX_ struct coro_transfer_args *ta)
1061	{	1084	{
1062	ta->prev = (struct coro *)cctx;	1085	ta->prev = (struct coro *)cctx_ssl_cctx;
1063	ta->next = 0;	1086	ta->next = 0;
1064	}	1087	}
1065		1088
1066	/* inject a fake call to Coro::State::_cctx_init into the execution */	1089	static int
1067	/* _cctx_init should be careful, as it could be called at almost any time */	1090	slf_check_set_stacklevel (pTHX_ struct CoroSLF *frame)
1068	/* during execution of a perl program */	1091	{
1069	/* also initialises PL_top_env */	1092	*frame = cctx_ssl_frame;
		1093
		1094	return frame->check (aTHX_ frame); /* execute the restored frame - there must be one */
		1095	}
		1096
		1097	/* initialises PL_top_env and injects a pseudo-slf-call to set the stacklevel */
1070	static void NOINLINE	1098	static void NOINLINE
1071	cctx_prepare (pTHX_ coro_cctx *cctx)	1099	cctx_prepare (pTHX_ coro_cctx *cctx)
1072	{	1100	{
1073	dSP;
1074	UNOP myop;
1075
1076	PL_top_env = &PL_start_env;	1101	PL_top_env = &PL_start_env;
1077		1102
1078	if (cctx->flags & CC_TRACE)	1103	if (cctx->flags & CC_TRACE)
1079	PL_runops = runops_trace;	1104	PL_runops = runops_trace;
1080		1105
1081	Zero (&myop, 1, UNOP);	1106	/* we already must be executing an SLF op, there is no other valid way
1082	myop.op_next = PL_op;	1107	* that can lead to creation of a new cctx */
1083	myop.op_flags = OPf_WANT_VOID | OPf_STACKED;	1108	assert (("FATAL: can't prepare slf-less cctx in Coro module (please report)",
		1109	slf_frame.prepare && PL_op->op_ppaddr == pp_slf));
1084		1110
1085	PUSHMARK (SP);	1111	/* we must emulate leaving pp_slf, which is done inside slf_check_set_stacklevel */
1086	EXTEND (SP, 2);	1112	cctx_ssl_cctx = cctx;
1087	PUSHs (sv_2mortal (newSViv ((IV)cctx)));	1113	cctx_ssl_frame = slf_frame;
1088	PUSHs ((SV *)get_cv ("Coro::State::_cctx_init", FALSE));	1114
1089	PUTBACK;	1115	slf_frame.prepare = slf_prepare_set_stacklevel;
1090	PL_op = (OP *)&myop;	1116	slf_frame.check = slf_check_set_stacklevel;
1091	PL_op = PL_ppaddr[OP_ENTERSUB](aTHX);
1092	SPAGAIN;
1093	}	1117	}
1094		1118
1095	/* the tail of transfer: execute stuff we can only do after a transfer */	1119	/* the tail of transfer: execute stuff we can only do after a transfer */
1096	INLINE void	1120	INLINE void
1097	transfer_tail (pTHX)	1121	transfer_tail (pTHX)
…		…
1124	transfer_tail (aTHX);	1148	transfer_tail (aTHX);
1125		1149
1126	/* somebody or something will hit me for both perl_run and PL_restartop */	1150	/* somebody or something will hit me for both perl_run and PL_restartop */
1127	PL_restartop = PL_op;	1151	PL_restartop = PL_op;
1128	perl_run (PL_curinterp);	1152	perl_run (PL_curinterp);
		1153	/*
		1154	* Unfortunately, there is no way to get at the return values of the
		1155	* coro body here, as perl_run destroys these
		1156	*/
1129		1157
1130	/*	1158	/*
1131	* If perl-run returns we assume exit() was being called or the coro	1159	* If perl-run returns we assume exit() was being called or the coro
1132	* fell off the end, which seems to be the only valid (non-bug)	1160	* fell off the end, which seems to be the only valid (non-bug)
1133	* reason for perl_run to return. We try to exit by jumping to the	1161	* reason for perl_run to return. We try to exit by jumping to the
…		…
1313	dSTACKLEVEL;	1341	dSTACKLEVEL;
1314		1342
1315	/* sometimes transfer is only called to set idle_sp */	1343	/* sometimes transfer is only called to set idle_sp */
1316	if (expect_false (!next))	1344	if (expect_false (!next))
1317	{	1345	{
1318	((coro_cctx *)prev)->idle_sp = (void *)stacklevel;	1346	((coro_cctx *)prev)->idle_sp = STACKLEVEL;
1319	assert (((coro_cctx *)prev)->idle_te = PL_top_env); /* just for the side-effect when asserts are enabled */	1347	assert (((coro_cctx *)prev)->idle_te = PL_top_env); /* just for the side-effect when asserts are enabled */
1320	}	1348	}
1321	else if (expect_true (prev != next))	1349	else if (expect_true (prev != next))
1322	{	1350	{
1323	coro_cctx *prev__cctx;	1351	coro_cctx *prev__cctx;
…		…
1348		1376
1349	prev__cctx = prev->cctx;	1377	prev__cctx = prev->cctx;
1350		1378
1351	/* possibly untie and reuse the cctx */	1379	/* possibly untie and reuse the cctx */
1352	if (expect_true (	1380	if (expect_true (
1353	prev__cctx->idle_sp == (void *)stacklevel	1381	prev__cctx->idle_sp == STACKLEVEL
1354	&& !(prev__cctx->flags & CC_TRACE)	1382	&& !(prev__cctx->flags & CC_TRACE)
1355	&& !force_cctx	1383	&& !force_cctx
1356	))	1384	))
1357	{	1385	{
1358	/* I assume that stacklevel is a stronger indicator than PL_top_env changes */	1386	/* I assume that stacklevel is a stronger indicator than PL_top_env changes */
…		…
1425		1453
1426	coro->slot = 0;	1454	coro->slot = 0;
1427	}	1455	}
1428		1456
1429	cctx_destroy (coro->cctx);	1457	cctx_destroy (coro->cctx);
		1458	SvREFCNT_dec (coro->startcv);
1430	SvREFCNT_dec (coro->args);	1459	SvREFCNT_dec (coro->args);
		1460	SvREFCNT_dec (CORO_THROW);
1431		1461
1432	if (coro->next) coro->next->prev = coro->prev;	1462	if (coro->next) coro->next->prev = coro->prev;
1433	if (coro->prev) coro->prev->next = coro->next;	1463	if (coro->prev) coro->prev->next = coro->next;
1434	if (coro == coro_first) coro_first = coro->next;	1464	if (coro == coro_first) coro_first = coro->next;
1435		1465
…		…
1489		1519
1490	prepare_transfer (aTHX_ &ta, prev_sv, next_sv);	1520	prepare_transfer (aTHX_ &ta, prev_sv, next_sv);
1491	TRANSFER (ta, 1);	1521	TRANSFER (ta, 1);
1492	}	1522	}
1493		1523
		1524	/*****************************************************************************/
		1525	/* gensub: simple closure generation utility */
		1526
		1527	#define GENSUB_ARG CvXSUBANY (cv).any_ptr
		1528
		1529	/* create a closure from XS, returns a code reference */
		1530	/* the arg can be accessed via GENSUB_ARG from the callback */
		1531	/* the callback must use dXSARGS/XSRETURN */
		1532	static SV *
		1533	gensub (pTHX_ void (*xsub)(pTHX_ CV *), void *arg)
		1534	{
		1535	CV *cv = (CV *)newSV (0);
		1536
		1537	sv_upgrade ((SV *)cv, SVt_PVCV);
		1538
		1539	CvANON_on (cv);
		1540	CvISXSUB_on (cv);
		1541	CvXSUB (cv) = xsub;
		1542	GENSUB_ARG = arg;
		1543
		1544	return newRV_noinc ((SV *)cv);
		1545	}
		1546
1494	/** Coro ********************************************************************/	1547	/** Coro ********************************************************************/
1495		1548
1496	INLINE void	1549	INLINE void
1497	coro_enq (pTHX_ struct coro *coro)	1550	coro_enq (pTHX_ struct coro *coro)
1498	{	1551	{
…		…
1541	ENTER;	1594	ENTER;
1542	SAVETMPS;	1595	SAVETMPS;
1543		1596
1544	PUSHMARK (SP);	1597	PUSHMARK (SP);
1545	PUTBACK;	1598	PUTBACK;
1546	call_sv (sv_hook, G_DISCARD);	1599	call_sv (sv_hook, G_VOID | G_DISCARD);
1547	SPAGAIN;
1548		1600
1549	FREETMPS;	1601	FREETMPS;
1550	LEAVE;	1602	LEAVE;
1551	}	1603	}
1552		1604
…		…
1579	ENTER;	1631	ENTER;
1580	SAVETMPS;	1632	SAVETMPS;
1581		1633
1582	PUSHMARK (SP);	1634	PUSHMARK (SP);
1583	PUTBACK;	1635	PUTBACK;
1584	call_sv (get_sv ("Coro::idle", FALSE), G_DISCARD);	1636	call_sv (get_sv ("Coro::idle", FALSE), G_VOID | G_DISCARD);
1585	SPAGAIN;
1586		1637
1587	FREETMPS;	1638	FREETMPS;
1588	LEAVE;	1639	LEAVE;
1589	continue;	1640	continue;
1590	}	1641	}
…		…
1697	if (coro->flags & CF_RUNNING)	1748	if (coro->flags & CF_RUNNING)
1698	PL_runops = RUNOPS_DEFAULT;	1749	PL_runops = RUNOPS_DEFAULT;
1699	else	1750	else
1700	coro->slot->runops = RUNOPS_DEFAULT;	1751	coro->slot->runops = RUNOPS_DEFAULT;
1701	}	1752	}
		1753	}
		1754
		1755	/*****************************************************************************/
		1756	/* async pool handler */
		1757
		1758	static int
		1759	slf_check_pool_handler (pTHX_ struct CoroSLF *frame)
		1760	{
		1761	HV *hv = (HV *)SvRV (coro_current);
		1762	struct coro *coro = (struct coro *)frame->data;
		1763
		1764	if (!coro->invoke_cb)
		1765	return 1; /* loop till we have invoke */
		1766	else
		1767	{
		1768	hv_store (hv, "desc", sizeof ("desc") - 1,
		1769	newSVpvn ("[async_pool]", sizeof ("[async_pool]") - 1), 0);
		1770
		1771	coro->saved_deffh = SvREFCNT_inc_NN ((SV *)PL_defoutgv);
		1772
		1773	{
		1774	dSP;
		1775	XPUSHs (sv_2mortal (coro->invoke_cb)); coro->invoke_cb = 0;
		1776	PUTBACK;
		1777	}
		1778
		1779	SvREFCNT_dec (GvAV (PL_defgv));
		1780	GvAV (PL_defgv) = coro->invoke_av;
		1781	coro->invoke_av = 0;
		1782
		1783	return 0;
		1784	}
		1785	}
		1786
		1787	static void
		1788	slf_init_pool_handler (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		1789	{
		1790	HV *hv = (HV *)SvRV (coro_current);
		1791	struct coro *coro = SvSTATE_hv ((SV *)hv);
		1792
		1793	if (expect_true (coro->saved_deffh))
		1794	{
		1795	/* subsequent iteration */
		1796	SvREFCNT_dec ((SV *)PL_defoutgv); PL_defoutgv = (GV *)coro->saved_deffh;
		1797	coro->saved_deffh = 0;
		1798
		1799	if (coro_rss (aTHX_ coro) > SvUV (sv_pool_rss)
		1800	|| av_len (av_async_pool) + 1 >= SvIV (sv_pool_size))
		1801	{
		1802	coro->invoke_cb = SvREFCNT_inc_NN ((SV *)cv_coro_terminate);
		1803	coro->invoke_av = newAV ();
		1804
		1805	frame->prepare = prepare_nop;
		1806	}
		1807	else
		1808	{
		1809	av_clear (GvAV (PL_defgv));
		1810	hv_store (hv, "desc", sizeof ("desc") - 1, SvREFCNT_inc_NN (sv_async_pool_idle), 0);
		1811
		1812	coro->prio = 0;
		1813
		1814	if (coro->cctx && (coro->cctx->flags & CC_TRACE))
		1815	api_trace (aTHX_ coro_current, 0);
		1816
		1817	frame->prepare = prepare_schedule;
		1818	av_push (av_async_pool, SvREFCNT_inc (hv));
		1819	}
		1820	}
		1821	else
		1822	{
		1823	/* first iteration, simply fall through */
		1824	frame->prepare = prepare_nop;
		1825	}
		1826
		1827	frame->check = slf_check_pool_handler;
		1828	frame->data = (void *)coro;
		1829	}
		1830
		1831	/*****************************************************************************/
		1832	/* rouse callback */
		1833
		1834	#define CORO_MAGIC_type_rouse PERL_MAGIC_ext
		1835
		1836	static void
		1837	coro_rouse_callback (pTHX_ CV *cv)
		1838	{
		1839	dXSARGS;
		1840	SV *data = (SV *)GENSUB_ARG;
		1841
		1842	if (SvTYPE (SvRV (data)) != SVt_PVAV)
		1843	{
		1844	/* first call, set args */
		1845	AV *av = newAV ();
		1846	SV *coro = SvRV (data);
		1847
		1848	SvRV_set (data, (SV *)av);
		1849	api_ready (aTHX_ coro);
		1850	SvREFCNT_dec (coro);
		1851
		1852	/* better take a full copy of the arguments */
		1853	while (items--)
		1854	av_store (av, items, newSVsv (ST (items)));
		1855	}
		1856
		1857	XSRETURN_EMPTY;
		1858	}
		1859
		1860	static int
		1861	slf_check_rouse_wait (pTHX_ struct CoroSLF *frame)
		1862	{
		1863	SV *data = (SV *)frame->data;
		1864
		1865	if (CORO_THROW)
		1866	return 0;
		1867
		1868	if (SvTYPE (SvRV (data)) != SVt_PVAV)
		1869	return 1;
		1870
		1871	/* now push all results on the stack */
		1872	{
		1873	dSP;
		1874	AV *av = (AV *)SvRV (data);
		1875	int i;
		1876
		1877	EXTEND (SP, AvFILLp (av) + 1);
		1878	for (i = 0; i <= AvFILLp (av); ++i)
		1879	PUSHs (sv_2mortal (AvARRAY (av)[i]));
		1880
		1881	/* we have stolen the elements, so ste length to zero and free */
		1882	AvFILLp (av) = -1;
		1883	av_undef (av);
		1884
		1885	PUTBACK;
		1886	}
		1887
		1888	return 0;
		1889	}
		1890
		1891	static void
		1892	slf_init_rouse_wait (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		1893	{
		1894	SV *cb;
		1895
		1896	if (items)
		1897	cb = arg [0];
		1898	else
		1899	{
		1900	struct coro *coro = SvSTATE_current;
		1901
		1902	if (!coro->rouse_cb)
		1903	croak ("Coro::rouse_wait called without rouse callback, and no default rouse callback found either,");
		1904
		1905	cb = sv_2mortal (coro->rouse_cb);
		1906	coro->rouse_cb = 0;
		1907	}
		1908
		1909	if (!SvROK (cb)
		1910	|| SvTYPE (SvRV (cb)) != SVt_PVCV
		1911	|| CvXSUB ((CV *)SvRV (cb)) != coro_rouse_callback)
		1912	croak ("Coro::rouse_wait called with illegal callback argument,");
		1913
		1914	{
		1915	CV *cv = (CV *)SvRV (cb); /* for GENSUB_ARG */
		1916	SV *data = (SV *)GENSUB_ARG;
		1917
		1918	frame->data = (void *)data;
		1919	frame->prepare = SvTYPE (SvRV (data)) == SVt_PVAV ? prepare_nop : prepare_schedule;
		1920	frame->check = slf_check_rouse_wait;
		1921	}
		1922	}
		1923
		1924	static SV *
		1925	coro_new_rouse_cb (pTHX)
		1926	{
		1927	HV *hv = (HV *)SvRV (coro_current);
		1928	struct coro *coro = SvSTATE_hv (hv);
		1929	SV *data = newRV_inc ((SV *)hv);
		1930	SV *cb = gensub (aTHX_ coro_rouse_callback, (void *)data);
		1931
		1932	sv_magicext (SvRV (cb), data, CORO_MAGIC_type_rouse, 0, 0, 0);
		1933	SvREFCNT_dec (data); /* magicext increases the refcount */
		1934
		1935	SvREFCNT_dec (coro->rouse_cb);
		1936	coro->rouse_cb = SvREFCNT_inc_NN (cb);
		1937
		1938	return cb;
		1939	}
		1940
		1941	/*****************************************************************************/
		1942	/* schedule-like-function opcode (SLF) */
		1943
		1944	static UNOP slf_restore; /* restore stack as entersub did, for first-re-run */
		1945	static const CV *slf_cv;
		1946	static SV **slf_argv;
		1947	static int slf_argc, slf_arga; /* count, allocated */
		1948	static I32 slf_ax; /* top of stack, for restore */
		1949
		1950	/* this restores the stack in the case we patched the entersub, to */
		1951	/* recreate the stack frame as perl will on following calls */
		1952	/* since entersub cleared the stack */
		1953	static OP *
		1954	pp_restore (pTHX)
		1955	{
		1956	int i;
		1957	SV **SP = PL_stack_base + slf_ax;
		1958
		1959	PUSHMARK (SP);
		1960
		1961	EXTEND (SP, slf_argc + 1);
		1962
		1963	for (i = 0; i < slf_argc; ++i)
		1964	PUSHs (sv_2mortal (slf_argv [i]));
		1965
		1966	PUSHs ((SV *)CvGV (slf_cv));
		1967
		1968	RETURNOP (slf_restore.op_first);
		1969	}
		1970
		1971	static void
		1972	slf_prepare_transfer (pTHX_ struct coro_transfer_args *ta)
		1973	{
		1974	SV **arg = (SV **)slf_frame.data;
		1975
		1976	prepare_transfer (aTHX_ ta, arg [0], arg [1]);
		1977	}
		1978
		1979	static void
		1980	slf_init_transfer (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		1981	{
		1982	if (items != 2)
		1983	croak ("Coro::State::transfer (prev, next) expects two arguments, not %d,", items);
		1984
		1985	frame->prepare = slf_prepare_transfer;
		1986	frame->check = slf_check_nop;
		1987	frame->data = (void *)arg; /* let's hope it will stay valid */
		1988	}
		1989
		1990	static void
		1991	slf_init_schedule (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		1992	{
		1993	frame->prepare = prepare_schedule;
		1994	frame->check = slf_check_nop;
		1995	}
		1996
		1997	static void
		1998	slf_init_cede (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		1999	{
		2000	frame->prepare = prepare_cede;
		2001	frame->check = slf_check_nop;
		2002	}
		2003
		2004	static void
		2005	slf_init_cede_notself (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		2006	{
		2007	frame->prepare = prepare_cede_notself;
		2008	frame->check = slf_check_nop;
		2009	}
		2010
		2011	/*
		2012	* these not obviously related functions are all rolled into one
		2013	* function to increase chances that they all will call transfer with the same
		2014	* stack offset
		2015	* SLF stands for "schedule-like-function".
		2016	*/
		2017	static OP *
		2018	pp_slf (pTHX)
		2019	{
		2020	I32 checkmark; /* mark SP to see how many elements check has pushed */
		2021
		2022	/* set up the slf frame, unless it has already been set-up */
		2023	/* the latter happens when a new coro has been started */
		2024	/* or when a new cctx was attached to an existing coroutine */
		2025	if (expect_true (!slf_frame.prepare))
		2026	{
		2027	/* first iteration */
		2028	dSP;
		2029	SV **arg = PL_stack_base + TOPMARK + 1;
		2030	int items = SP - arg; /* args without function object */
		2031	SV *gv = *sp;
		2032
		2033	/* do a quick consistency check on the "function" object, and if it isn't */
		2034	/* for us, divert to the real entersub */
		2035	if (SvTYPE (gv) != SVt_PVGV
		2036	|| !GvCV (gv)
		2037	|| !(CvFLAGS (GvCV (gv)) & CVf_SLF))
		2038	return PL_ppaddr[OP_ENTERSUB](aTHX);
		2039
		2040	if (!(PL_op->op_flags & OPf_STACKED))
		2041	{
		2042	/* ampersand-form of call, use @_ instead of stack */
		2043	AV *av = GvAV (PL_defgv);
		2044	arg = AvARRAY (av);
		2045	items = AvFILLp (av) + 1;
		2046	}
		2047
		2048	/* now call the init function, which needs to set up slf_frame */
		2049	((coro_slf_cb)CvXSUBANY (GvCV (gv)).any_ptr)
		2050	(aTHX_ &slf_frame, GvCV (gv), arg, items);
		2051
		2052	/* pop args */
		2053	SP = PL_stack_base + POPMARK;
		2054
		2055	PUTBACK;
		2056	}
		2057
		2058	/* now that we have a slf_frame, interpret it! */
		2059	/* we use a callback system not to make the code needlessly */
		2060	/* complicated, but so we can run multiple perl coros from one cctx */
		2061
		2062	do
		2063	{
		2064	struct coro_transfer_args ta;
		2065
		2066	slf_frame.prepare (aTHX_ &ta);
		2067	TRANSFER (ta, 0);
		2068
		2069	checkmark = PL_stack_sp - PL_stack_base;
		2070	}
		2071	while (slf_frame.check (aTHX_ &slf_frame));
		2072
		2073	slf_frame.prepare = 0; /* invalidate the frame, we are done processing it */
		2074
		2075	/* exception handling */
		2076	if (expect_false (CORO_THROW))
		2077	{
		2078	SV *exception = sv_2mortal (CORO_THROW);
		2079
		2080	CORO_THROW = 0;
		2081	sv_setsv (ERRSV, exception);
		2082	croak (0);
		2083	}
		2084
		2085	/* return value handling - mostly like entersub */
		2086	/* make sure we put something on the stack in scalar context */
		2087	if (GIMME_V == G_SCALAR)
		2088	{
		2089	dSP;
		2090	SV **bot = PL_stack_base + checkmark;
		2091
		2092	if (sp == bot) /* too few, push undef */
		2093	bot [1] = &PL_sv_undef;
		2094	else if (sp != bot + 1) /* too many, take last one */
		2095	bot [1] = *sp;
		2096
		2097	SP = bot + 1;
		2098
		2099	PUTBACK;
		2100	}
		2101
		2102	return NORMAL;
		2103	}
		2104
		2105	static void
		2106	api_execute_slf (pTHX_ CV *cv, coro_slf_cb init_cb, I32 ax)
		2107	{
		2108	int i;
		2109	SV **arg = PL_stack_base + ax;
		2110	int items = PL_stack_sp - arg + 1;
		2111
		2112	assert (("FATAL: SLF call with illegal CV value", !CvANON (cv)));
		2113
		2114	if (PL_op->op_ppaddr != PL_ppaddr [OP_ENTERSUB]
		2115	&& PL_op->op_ppaddr != pp_slf)
		2116	croak ("FATAL: Coro SLF calls can only be made normally, not via goto or any other means, caught");
		2117
		2118	CvFLAGS (cv) |= CVf_SLF;
		2119	CvXSUBANY (cv).any_ptr = (void *)init_cb;
		2120	slf_cv = cv;
		2121
		2122	/* we patch the op, and then re-run the whole call */
		2123	/* we have to put the same argument on the stack for this to work */
		2124	/* and this will be done by pp_restore */
		2125	slf_restore.op_next = (OP *)&slf_restore;
		2126	slf_restore.op_type = OP_CUSTOM;
		2127	slf_restore.op_ppaddr = pp_restore;
		2128	slf_restore.op_first = PL_op;
		2129
		2130	slf_ax = ax - 1; /* undo the ax++ inside dAXMARK */
		2131
		2132	if (PL_op->op_flags & OPf_STACKED)
		2133	{
		2134	if (items > slf_arga)
		2135	{
		2136	slf_arga = items;
		2137	free (slf_argv);
		2138	slf_argv = malloc (slf_arga * sizeof (SV *));
		2139	}
		2140
		2141	slf_argc = items;
		2142
		2143	for (i = 0; i < items; ++i)
		2144	slf_argv [i] = SvREFCNT_inc (arg [i]);
		2145	}
		2146	else
		2147	slf_argc = 0;
		2148
		2149	PL_op->op_ppaddr = pp_slf;
		2150	PL_op->op_type = OP_CUSTOM; /* maybe we should leave it at entersub? */
		2151
		2152	PL_op = (OP *)&slf_restore;
1702	}	2153	}
1703		2154
1704	/*****************************************************************************/	2155	/*****************************************************************************/
1705	/* PerlIO::cede */	2156	/* PerlIO::cede */
1706		2157
…		…
1775	PerlIOBuf_get_cnt,	2226	PerlIOBuf_get_cnt,
1776	PerlIOBuf_set_ptrcnt,	2227	PerlIOBuf_set_ptrcnt,
1777	};	2228	};
1778		2229
1779	/*****************************************************************************/	2230	/*****************************************************************************/
		2231	/* Coro::Semaphore & Coro::Signal */
1780		2232
1781	static UNOP slf_restore; /* restore stack as entersub did, for first-re-run */
1782	static const CV *slf_cv;
1783	static SV **slf_argv;
1784	static int slf_argc, slf_arga; /* count, allocated */
1785	static I32 slf_ax; /* top of stack, for restore */
1786
1787	/* this restores the stack in the case we patched the entersub, to */
1788	/* recreate the stack frame as perl will on following calls */
1789	/* since entersub cleared the stack */
1790	static OP *	2233	static SV *
1791	pp_restore (pTHX)	2234	coro_waitarray_new (pTHX_ int count)
1792	{	2235	{
1793	int i;	2236	/* a semaphore contains a counter IV in $sem->[0] and any waiters after that */
1794	SV **SP = PL_stack_base + slf_ax;	2237	AV *av = newAV ();
		2238	SV **ary;
1795		2239
1796	PUSHMARK (SP);	2240	/* unfortunately, building manually saves memory */
		2241	Newx (ary, 2, SV *);
		2242	AvALLOC (av) = ary;
		2243	/*AvARRAY (av) = ary;*/
		2244	SvPVX ((SV *)av) = (char *)ary; /* 5.8.8 needs this syntax instead of AvARRAY = ary */
		2245	AvMAX (av) = 1;
		2246	AvFILLp (av) = 0;
		2247	ary [0] = newSViv (count);
1797		2248
1798	EXTEND (SP, slf_argc + 1);	2249	return newRV_noinc ((SV *)av);
1799
1800	for (i = 0; i < slf_argc; ++i)
1801	PUSHs (sv_2mortal (slf_argv [i]));
1802
1803	PUSHs ((SV *)CvGV (slf_cv));
1804
1805	RETURNOP (slf_restore.op_first);
1806	}	2250	}
1807		2251
1808	static void	2252	/* semaphore */
1809	slf_prepare_set_stacklevel (pTHX_ struct coro_transfer_args *ta)
1810	{
1811	prepare_set_stacklevel (ta, (struct coro_cctx *)slf_frame.data);
1812	}
1813
1814	static void
1815	slf_init_set_stacklevel (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
1816	{
1817	assert (("FATAL: set_stacklevel needs the coro cctx as sole argument", items == 1));
1818
1819	frame->prepare = slf_prepare_set_stacklevel;
1820	frame->check = slf_check_nop;
1821	frame->data = (void *)SvIV (arg [0]);
1822	}
1823
1824	static void
1825	slf_prepare_transfer (pTHX_ struct coro_transfer_args *ta)
1826	{
1827	SV **arg = (SV **)slf_frame.data;
1828
1829	prepare_transfer (aTHX_ ta, arg [0], arg [1]);
1830	}
1831
1832	static void
1833	slf_init_transfer (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
1834	{
1835	if (items != 2)
1836	croak ("Coro::State::transfer (prev, next) expects two arguments, not %d,", items);
1837
1838	frame->prepare = slf_prepare_transfer;
1839	frame->check = slf_check_nop;
1840	frame->data = (void *)arg; /* let's hope it will stay valid */
1841	}
1842
1843	static void
1844	slf_init_schedule (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
1845	{
1846	frame->prepare = prepare_schedule;
1847	frame->check = slf_check_nop;
1848	}
1849
1850	static void
1851	slf_init_cede (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
1852	{
1853	frame->prepare = prepare_cede;
1854	frame->check = slf_check_nop;
1855	}
1856
1857	static void
1858	slf_init_cede_notself (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
1859	{
1860	frame->prepare = prepare_cede_notself;
1861	frame->check = slf_check_nop;
1862	}
1863
1864	/* we hijack an hopefully unused CV flag for our purposes */
1865	#define CVf_SLF 0x4000
1866
1867	/*
1868	* these not obviously related functions are all rolled into one
1869	* function to increase chances that they all will call transfer with the same
1870	* stack offset
1871	* SLF stands for "schedule-like-function".
1872	*/
1873	static OP *
1874	pp_slf (pTHX)
1875	{
1876	I32 checkmark; /* mark SP to see how many elements check has pushed */
1877
1878	/* set up the slf frame, unless it has already been set-up */
1879	/* the latter happens when a new coro has been started */
1880	/* or when a new cctx was attached to an existing coroutine */
1881	if (expect_true (!slf_frame.prepare))
1882	{
1883	/* first iteration */
1884	dSP;
1885	SV **arg = PL_stack_base + TOPMARK + 1;
1886	int items = SP - arg; /* args without function object */
1887	SV *gv = *sp;
1888
1889	/* do a quick consistency check on the "function" object, and if it isn't */
1890	/* for us, divert to the real entersub */
1891	if (SvTYPE (gv) != SVt_PVGV || !(CvFLAGS (GvCV (gv)) & CVf_SLF))
1892	return PL_ppaddr[OP_ENTERSUB](aTHX);
1893
1894	if (!(PL_op->op_flags & OPf_STACKED))
1895	{
1896	/* ampersand-form of call, use @_ instead of stack */
1897	AV *av = GvAV (PL_defgv);
1898	arg = AvARRAY (av);
1899	items = AvFILLp (av) + 1;
1900	}
1901
1902	/* now call the init function, which needs to set up slf_frame */
1903	((coro_slf_cb)CvXSUBANY (GvCV (gv)).any_ptr)
1904	(aTHX_ &slf_frame, GvCV (gv), arg, items);
1905
1906	/* pop args */
1907	SP = PL_stack_base + POPMARK;
1908
1909	PUTBACK;
1910	}
1911
1912	/* now that we have a slf_frame, interpret it! */
1913	/* we use a callback system not to make the code needlessly */
1914	/* complicated, but so we can run multiple perl coros from one cctx */
1915
1916	do
1917	{
1918	struct coro_transfer_args ta;
1919
1920	slf_frame.prepare (aTHX_ &ta);
1921	TRANSFER (ta, 0);
1922
1923	checkmark = PL_stack_sp - PL_stack_base;
1924	}
1925	while (slf_frame.check (aTHX_ &slf_frame));
1926
1927	slf_frame.prepare = 0; /* invalidate the frame, we are done processing it */
1928
1929	/* return value handling - mostly like entersub */
1930	{
1931	dSP;
1932	SV **bot = PL_stack_base + checkmark;
1933	int gimme = GIMME_V;
1934
1935	/* make sure we put something on the stack in scalar context */
1936	if (gimme == G_SCALAR)
1937	{
1938	if (sp == bot)
1939	XPUSHs (&PL_sv_undef);
1940
1941	SP = bot + 1;
1942	}
1943
1944	PUTBACK;
1945	}
1946
1947	/* exception handling */
1948	if (expect_false (coro_throw))
1949	{
1950	SV *exception = sv_2mortal (coro_throw);
1951
1952	coro_throw = 0;
1953	sv_setsv (ERRSV, exception);
1954	croak (0);
1955	}
1956
1957	return NORMAL;
1958	}
1959
1960	static void
1961	api_execute_slf (pTHX_ CV *cv, coro_slf_cb init_cb, I32 ax)
1962	{
1963	int i;
1964	SV **arg = PL_stack_base + ax;
1965	int items = PL_stack_sp - arg + 1;
1966
1967	assert (("FATAL: SLF call with illegal CV value", !CvANON (cv)));
1968
1969	if (PL_op->op_ppaddr != PL_ppaddr [OP_ENTERSUB]
1970	&& PL_op->op_ppaddr != pp_slf)
1971	croak ("FATAL: Coro SLF calls can only be made normally, not via goto or any other means, caught");
1972
1973	CvFLAGS (cv) |= CVf_SLF;
1974	CvXSUBANY (cv).any_ptr = (void *)init_cb;
1975	slf_cv = cv;
1976
1977	/* we patch the op, and then re-run the whole call */
1978	/* we have to put the same argument on the stack for this to work */
1979	/* and this will be done by pp_restore */
1980	slf_restore.op_next = (OP *)&slf_restore;
1981	slf_restore.op_type = OP_CUSTOM;
1982	slf_restore.op_ppaddr = pp_restore;
1983	slf_restore.op_first = PL_op;
1984
1985	slf_ax = ax - 1; /* undo the ax++ inside dAXMARK */
1986
1987	if (items > slf_arga)
1988	{
1989	slf_arga = items;
1990	free (slf_argv);
1991	slf_argv = malloc (slf_arga * sizeof (SV *));
1992	}
1993
1994	slf_argc = items;
1995
1996	for (i = 0; i < items; ++i)
1997	slf_argv [i] = SvREFCNT_inc (arg [i]);
1998
1999	PL_op->op_ppaddr = pp_slf;
2000	PL_op->op_type = OP_CUSTOM; /* maybe we should leave it at entersub? */
2001
2002	PL_op = (OP *)&slf_restore;
2003	}
2004
2005	/*****************************************************************************/
2006		2253
2007	static void	2254	static void
2008	coro_semaphore_adjust (pTHX_ AV *av, IV adjust)	2255	coro_semaphore_adjust (pTHX_ AV *av, IV adjust)
2009	{	2256	{
2010	SV *count_sv = AvARRAY (av)[0];	2257	SV *count_sv = AvARRAY (av)[0];
…		…
2022	AvARRAY (av)[0] = AvARRAY (av)[1];	2269	AvARRAY (av)[0] = AvARRAY (av)[1];
2023	AvARRAY (av)[1] = count_sv;	2270	AvARRAY (av)[1] = count_sv;
2024	cb = av_shift (av);	2271	cb = av_shift (av);
2025		2272
2026	if (SvOBJECT (cb))	2273	if (SvOBJECT (cb))
		2274	{
2027	api_ready (aTHX_ cb);	2275	api_ready (aTHX_ cb);
2028	else	2276	--count;
2029	croak ("callbacks not yet supported");	2277	}
		2278	else if (SvTYPE (cb) == SVt_PVCV)
		2279	{
		2280	dSP;
		2281	PUSHMARK (SP);
		2282	XPUSHs (sv_2mortal (newRV_inc ((SV *)av)));
		2283	PUTBACK;
		2284	call_sv (cb, G_VOID | G_DISCARD | G_EVAL | G_KEEPERR);
		2285	}
2030		2286
2031	SvREFCNT_dec (cb);	2287	SvREFCNT_dec (cb);
2032
2033	--count;
2034	}	2288	}
2035	}	2289	}
2036		2290
2037	static void	2291	static void
2038	coro_semaphore_on_destroy (pTHX_ struct coro *coro)	2292	coro_semaphore_on_destroy (pTHX_ struct coro *coro)
…		…
2040	/* call $sem->adjust (0) to possibly wake up some other waiters */	2294	/* call $sem->adjust (0) to possibly wake up some other waiters */
2041	coro_semaphore_adjust (aTHX_ (AV *)coro->slf_frame.data, 0);	2295	coro_semaphore_adjust (aTHX_ (AV *)coro->slf_frame.data, 0);
2042	}	2296	}
2043		2297
2044	static int	2298	static int
2045	slf_check_semaphore_down (pTHX_ struct CoroSLF *frame)	2299	slf_check_semaphore_down_or_wait (pTHX_ struct CoroSLF *frame, int acquire)
2046	{	2300	{
2047	AV *av = (AV *)frame->data;	2301	AV *av = (AV *)frame->data;
2048	SV *count_sv = AvARRAY (av)[0];	2302	SV *count_sv = AvARRAY (av)[0];
2049		2303
		2304	/* if we are about to throw, don't actually acquire the lock, just throw */
		2305	if (CORO_THROW)
		2306	return 0;
2050	if (SvIVX (count_sv) > 0)	2307	else if (SvIVX (count_sv) > 0)
2051	{	2308	{
2052	SvSTATE_current->on_destroy = 0;	2309	SvSTATE_current->on_destroy = 0;
		2310
		2311	if (acquire)
2053	SvIVX (count_sv) = SvIVX (count_sv) - 1;	2312	SvIVX (count_sv) = SvIVX (count_sv) - 1;
		2313	else
		2314	coro_semaphore_adjust (aTHX_ av, 0);
		2315
2054	return 0;	2316	return 0;
2055	}	2317	}
2056	else	2318	else
2057	{	2319	{
2058	int i;	2320	int i;
…		…
2067	av_push (av, SvREFCNT_inc (SvRV (coro_current)));	2329	av_push (av, SvREFCNT_inc (SvRV (coro_current)));
2068	return 1;	2330	return 1;
2069	}	2331	}
2070	}	2332	}
2071		2333
2072	static void	2334	static int
		2335	slf_check_semaphore_down (pTHX_ struct CoroSLF *frame)
		2336	{
		2337	return slf_check_semaphore_down_or_wait (aTHX_ frame, 1);
		2338	}
		2339
		2340	static int
		2341	slf_check_semaphore_wait (pTHX_ struct CoroSLF *frame)
		2342	{
		2343	return slf_check_semaphore_down_or_wait (aTHX_ frame, 0);
		2344	}
		2345
		2346	static void
2073	slf_init_semaphore_down (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)	2347	slf_init_semaphore_down_or_wait (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
2074	{	2348	{
2075	AV *av = (AV *)SvRV (arg [0]);	2349	AV *av = (AV *)SvRV (arg [0]);
2076		2350
2077	if (SvIVX (AvARRAY (av)[0]) > 0)	2351	if (SvIVX (AvARRAY (av)[0]) > 0)
2078	{	2352	{
2079	frame->data = (void *)av;	2353	frame->data = (void *)av;
2080	frame->prepare = prepare_nop;	2354	frame->prepare = prepare_nop;
2081	SvSTATE_current->on_destroy = coro_semaphore_on_destroy;
2082	}	2355	}
2083	else	2356	else
2084	{	2357	{
2085	av_push (av, SvREFCNT_inc (SvRV (coro_current)));	2358	av_push (av, SvREFCNT_inc (SvRV (coro_current)));
2086		2359
2087	frame->data = (void *)sv_2mortal (SvREFCNT_inc ((SV *)av));	2360	frame->data = (void *)sv_2mortal (SvREFCNT_inc ((SV *)av));
2088	frame->prepare = prepare_schedule;	2361	frame->prepare = prepare_schedule;
2089		2362
2090	/* to avoid race conditions when a woken-up coro gets terminated */	2363	/* to avoid race conditions when a woken-up coro gets terminated */
2091	/* we arrange for a temporary on_destroy that calls adjust (0) */	2364	/* we arrange for a temporary on_destroy that calls adjust (0) */
2092	assert (!SvSTATE_current->on_destroy);//D
2093	SvSTATE_current->on_destroy = coro_semaphore_on_destroy;	2365	SvSTATE_current->on_destroy = coro_semaphore_on_destroy;
2094	}	2366	}
		2367	}
2095		2368
		2369	static void
		2370	slf_init_semaphore_down (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		2371	{
		2372	slf_init_semaphore_down_or_wait (aTHX_ frame, cv, arg, items);
2096	frame->check = slf_check_semaphore_down;	2373	frame->check = slf_check_semaphore_down;
		2374	}
2097		2375
		2376	static void
		2377	slf_init_semaphore_wait (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		2378	{
		2379	if (items >= 2)
		2380	{
		2381	/* callback form */
		2382	AV *av = (AV *)SvRV (arg [0]);
		2383	CV *cb_cv = coro_sv_2cv (aTHX_ arg [1]);
		2384
		2385	av_push (av, (SV *)SvREFCNT_inc_NN (cb_cv));
		2386
		2387	if (SvIVX (AvARRAY (av)[0]) > 0)
		2388	coro_semaphore_adjust (aTHX_ av, 0);
		2389
		2390	frame->prepare = prepare_nop;
		2391	frame->check = slf_check_nop;
		2392	}
		2393	else
		2394	{
		2395	slf_init_semaphore_down_or_wait (aTHX_ frame, cv, arg, items);
		2396	frame->check = slf_check_semaphore_wait;
		2397	}
		2398	}
		2399
		2400	/* signal */
		2401
		2402	static void
		2403	coro_signal_wake (pTHX_ AV *av, int count)
		2404	{
		2405	SvIVX (AvARRAY (av)[0]) = 0;
		2406
		2407	/* now signal count waiters */
		2408	while (count > 0 && AvFILLp (av) > 0)
		2409	{
		2410	SV *cb;
		2411
		2412	/* swap first two elements so we can shift a waiter */
		2413	cb = AvARRAY (av)[0];
		2414	AvARRAY (av)[0] = AvARRAY (av)[1];
		2415	AvARRAY (av)[1] = cb;
		2416
		2417	cb = av_shift (av);
		2418
		2419	api_ready (aTHX_ cb);
		2420	sv_setiv (cb, 0); /* signal waiter */
		2421	SvREFCNT_dec (cb);
		2422
		2423	--count;
		2424	}
		2425	}
		2426
		2427	static int
		2428	slf_check_signal_wait (pTHX_ struct CoroSLF *frame)
		2429	{
		2430	/* if we are about to throw, also stop waiting */
		2431	return SvROK ((SV *)frame->data) && !CORO_THROW;
		2432	}
		2433
		2434	static void
		2435	slf_init_signal_wait (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		2436	{
		2437	AV *av = (AV *)SvRV (arg [0]);
		2438
		2439	if (SvIVX (AvARRAY (av)[0]))
		2440	{
		2441	SvIVX (AvARRAY (av)[0]) = 0;
		2442	frame->prepare = prepare_nop;
		2443	frame->check = slf_check_nop;
		2444	}
		2445	else
		2446	{
		2447	SV *waiter = newRV_inc (SvRV (coro_current)); /* owned by signal av */
		2448
		2449	av_push (av, waiter);
		2450
		2451	frame->data = (void *)sv_2mortal (SvREFCNT_inc_NN (waiter)); /* owned by process */
		2452	frame->prepare = prepare_schedule;
		2453	frame->check = slf_check_signal_wait;
		2454	}
2098	}	2455	}
2099		2456
2100	/*****************************************************************************/	2457	/*****************************************************************************/
		2458	/* Coro::AIO */
2101		2459
2102	#define GENSUB_ARG CvXSUBANY (cv).any_ptr	2460	#define CORO_MAGIC_type_aio PERL_MAGIC_ext
2103		2461
2104	/* create a closure from XS, returns a code reference */	2462	/* helper storage struct */
2105	/* the arg can be accessed via GENSUB_ARG from the callback */	2463	struct io_state
2106	/* the callback must use dXSARGS/XSRETURN */
2107	static SV *
2108	gensub (pTHX_ void (*xsub)(pTHX_ CV *), void *arg)
2109	{	2464	{
2110	CV *cv = (CV *)NEWSV (0, 0);	2465	int errorno;
		2466	I32 laststype; /* U16 in 5.10.0 */
		2467	int laststatval;
		2468	Stat_t statcache;
		2469	};
2111		2470
		2471	static void
		2472	coro_aio_callback (pTHX_ CV *cv)
		2473	{
		2474	dXSARGS;
		2475	AV *state = (AV *)GENSUB_ARG;
		2476	SV *coro = av_pop (state);
		2477	SV *data_sv = newSV (sizeof (struct io_state));
		2478
		2479	av_extend (state, items - 1);
		2480
2112	sv_upgrade ((SV *)cv, SVt_PVCV);	2481	sv_upgrade (data_sv, SVt_PV);
		2482	SvCUR_set (data_sv, sizeof (struct io_state));
		2483	SvPOK_only (data_sv);
2113		2484
2114	CvANON_on (cv);	2485	{
2115	CvISXSUB_on (cv);	2486	struct io_state *data = (struct io_state *)SvPVX (data_sv);
2116	CvXSUB (cv) = xsub;
2117	GENSUB_ARG = arg;
2118		2487
2119	return newRV_noinc ((SV *)cv);	2488	data->errorno = errno;
		2489	data->laststype = PL_laststype;
		2490	data->laststatval = PL_laststatval;
		2491	data->statcache = PL_statcache;
		2492	}
		2493
		2494	/* now build the result vector out of all the parameters and the data_sv */
		2495	{
		2496	int i;
		2497
		2498	for (i = 0; i < items; ++i)
		2499	av_push (state, SvREFCNT_inc_NN (ST (i)));
		2500	}
		2501
		2502	av_push (state, data_sv);
		2503
		2504	api_ready (aTHX_ coro);
		2505	SvREFCNT_dec (coro);
		2506	SvREFCNT_dec ((AV *)state);
		2507	}
		2508
		2509	static int
		2510	slf_check_aio_req (pTHX_ struct CoroSLF *frame)
		2511	{
		2512	AV *state = (AV *)frame->data;
		2513
		2514	/* if we are about to throw, return early */
		2515	/* this does not cancel the aio request, but at least */
		2516	/* it quickly returns */
		2517	if (CORO_THROW)
		2518	return 0;
		2519
		2520	/* one element that is an RV? repeat! */
		2521	if (AvFILLp (state) == 0 && SvROK (AvARRAY (state)[0]))
		2522	return 1;
		2523
		2524	/* restore status */
		2525	{
		2526	SV *data_sv = av_pop (state);
		2527	struct io_state *data = (struct io_state *)SvPVX (data_sv);
		2528
		2529	errno = data->errorno;
		2530	PL_laststype = data->laststype;
		2531	PL_laststatval = data->laststatval;
		2532	PL_statcache = data->statcache;
		2533
		2534	SvREFCNT_dec (data_sv);
		2535	}
		2536
		2537	/* push result values */
		2538	{
		2539	dSP;
		2540	int i;
		2541
		2542	EXTEND (SP, AvFILLp (state) + 1);
		2543	for (i = 0; i <= AvFILLp (state); ++i)
		2544	PUSHs (sv_2mortal (SvREFCNT_inc_NN (AvARRAY (state)[i])));
		2545
		2546	PUTBACK;
		2547	}
		2548
		2549	return 0;
		2550	}
		2551
		2552	static void
		2553	slf_init_aio_req (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		2554	{
		2555	AV *state = (AV *)sv_2mortal ((SV *)newAV ());
		2556	SV *coro_hv = SvRV (coro_current);
		2557	struct coro *coro = SvSTATE_hv (coro_hv);
		2558
		2559	/* put our coroutine id on the state arg */
		2560	av_push (state, SvREFCNT_inc_NN (coro_hv));
		2561
		2562	/* first see whether we have a non-zero priority and set it as AIO prio */
		2563	if (coro->prio)
		2564	{
		2565	dSP;
		2566
		2567	static SV *prio_cv;
		2568	static SV *prio_sv;
		2569
		2570	if (expect_false (!prio_cv))
		2571	{
		2572	prio_cv = (SV *)get_cv ("IO::AIO::aioreq_pri", 0);
		2573	prio_sv = newSViv (0);
		2574	}
		2575
		2576	PUSHMARK (SP);
		2577	sv_setiv (prio_sv, coro->prio);
		2578	XPUSHs (prio_sv);
		2579
		2580	PUTBACK;
		2581	call_sv (prio_cv, G_VOID | G_DISCARD);
		2582	}
		2583
		2584	/* now call the original request */
		2585	{
		2586	dSP;
		2587	CV *req = (CV *)CORO_MAGIC_NN ((SV *)cv, CORO_MAGIC_type_aio)->mg_obj;
		2588	int i;
		2589
		2590	PUSHMARK (SP);
		2591
		2592	/* first push all args to the stack */
		2593	EXTEND (SP, items + 1);
		2594
		2595	for (i = 0; i < items; ++i)
		2596	PUSHs (arg [i]);
		2597
		2598	/* now push the callback closure */
		2599	PUSHs (sv_2mortal (gensub (aTHX_ coro_aio_callback, (void *)SvREFCNT_inc_NN ((SV *)state))));
		2600
		2601	/* now call the AIO function - we assume our request is uncancelable */
		2602	PUTBACK;
		2603	call_sv ((SV *)req, G_VOID | G_DISCARD);
		2604	}
		2605
		2606	/* now that the requets is going, we loop toll we have a result */
		2607	frame->data = (void *)state;
		2608	frame->prepare = prepare_schedule;
		2609	frame->check = slf_check_aio_req;
		2610	}
		2611
		2612	static void
		2613	coro_aio_req_xs (pTHX_ CV *cv)
		2614	{
		2615	dXSARGS;
		2616
		2617	CORO_EXECUTE_SLF_XS (slf_init_aio_req);
		2618
		2619	XSRETURN_EMPTY;
2120	}	2620	}
2121		2621
2122	/*****************************************************************************/	2622	/*****************************************************************************/
2123		2623
2124	MODULE = Coro::State PACKAGE = Coro::State PREFIX = api_	2624	MODULE = Coro::State PACKAGE = Coro::State PREFIX = api_
…		…
2194	assert (("PRIO_NORMAL must be 0", !PRIO_NORMAL));	2694	assert (("PRIO_NORMAL must be 0", !PRIO_NORMAL));
2195	}	2695	}
2196		2696
2197	SV *	2697	SV *
2198	new (char *klass, ...)	2698	new (char *klass, ...)
		2699	ALIAS:
		2700	Coro::new = 1
2199	CODE:	2701	CODE:
2200	{	2702	{
2201	struct coro *coro;	2703	struct coro *coro;
2202	MAGIC *mg;	2704	MAGIC *mg;
2203	HV *hv;	2705	HV *hv;
		2706	CV *cb;
2204	int i;	2707	int i;
		2708
		2709	if (items > 1)
		2710	{
		2711	cb = coro_sv_2cv (aTHX_ ST (1));
		2712
		2713	if (!ix)
		2714	{
		2715	if (CvISXSUB (cb))
		2716	croak ("Coro::State doesn't support XS functions as coroutine start, caught");
		2717
		2718	if (!CvROOT (cb))
		2719	croak ("Coro::State doesn't support autoloaded or undefined functions as coroutine start, caught");
		2720	}
		2721	}
2205		2722
2206	Newz (0, coro, 1, struct coro);	2723	Newz (0, coro, 1, struct coro);
2207	coro->args = newAV ();	2724	coro->args = newAV ();
2208	coro->flags = CF_NEW;	2725	coro->flags = CF_NEW;
2209		2726
…		…
2214	coro->hv = hv = newHV ();	2731	coro->hv = hv = newHV ();
2215	mg = sv_magicext ((SV *)hv, 0, CORO_MAGIC_type_state, &coro_state_vtbl, (char *)coro, 0);	2732	mg = sv_magicext ((SV *)hv, 0, CORO_MAGIC_type_state, &coro_state_vtbl, (char *)coro, 0);
2216	mg->mg_flags |= MGf_DUP;	2733	mg->mg_flags |= MGf_DUP;
2217	RETVAL = sv_bless (newRV_noinc ((SV *)hv), gv_stashpv (klass, 1));	2734	RETVAL = sv_bless (newRV_noinc ((SV *)hv), gv_stashpv (klass, 1));
2218		2735
		2736	if (items > 1)
		2737	{
2219	av_extend (coro->args, items - 1);	2738	av_extend (coro->args, items - 1 + ix - 1);
		2739
		2740	if (ix)
		2741	{
		2742	av_push (coro->args, SvREFCNT_inc_NN ((SV *)cb));
		2743	cb = cv_coro_run;
		2744	}
		2745
		2746	coro->startcv = (CV *)SvREFCNT_inc_NN ((SV *)cb);
		2747
2220	for (i = 1; i < items; i++)	2748	for (i = 2; i < items; i++)
2221	av_push (coro->args, newSVsv (ST (i)));	2749	av_push (coro->args, newSVsv (ST (i)));
		2750	}
2222	}	2751	}
2223	OUTPUT:	2752	OUTPUT:
2224	RETVAL	2753	RETVAL
2225
2226	void
2227	_set_stacklevel (...)
2228	CODE:
2229	CORO_EXECUTE_SLF_XS (slf_init_set_stacklevel);
2230		2754
2231	void	2755	void
2232	transfer (...)	2756	transfer (...)
2233	PROTOTYPE: $$	2757	PROTOTYPE: $$
2234	CODE:	2758	CODE:
…		…
2360	throw (Coro::State self, SV *throw = &PL_sv_undef)	2884	throw (Coro::State self, SV *throw = &PL_sv_undef)
2361	PROTOTYPE: $;$	2885	PROTOTYPE: $;$
2362	CODE:	2886	CODE:
2363	{	2887	{
2364	struct coro *current = SvSTATE_current;	2888	struct coro *current = SvSTATE_current;
2365	SV **throwp = self == current ? &coro_throw : &self->throw;	2889	SV **throwp = self == current ? &CORO_THROW : &self->except;
2366	SvREFCNT_dec (*throwp);	2890	SvREFCNT_dec (*throwp);
2367	*throwp = SvOK (throw) ? newSVsv (throw) : 0;	2891	*throwp = SvOK (throw) ? newSVsv (throw) : 0;
2368	}	2892	}
2369		2893
2370	void	2894	void
…		…
2422	SV **dst = ix ? (SV **)&self->slot->defav : (SV **)&self->slot->defsv;	2946	SV **dst = ix ? (SV **)&self->slot->defav : (SV **)&self->slot->defsv;
2423		2947
2424	SV *tmp = *src; *src = *dst; *dst = tmp;	2948	SV *tmp = *src; *src = *dst; *dst = tmp;
2425	}	2949	}
2426		2950
		2951
2427	MODULE = Coro::State PACKAGE = Coro	2952	MODULE = Coro::State PACKAGE = Coro
2428		2953
2429	BOOT:	2954	BOOT:
2430	{	2955	{
2431	int i;	2956	int i;
2432		2957
2433	av_async_pool = coro_get_av (aTHX_ "Coro::async_pool", TRUE);	2958	av_async_pool = coro_get_av (aTHX_ "Coro::async_pool", TRUE);
2434	sv_pool_rss = coro_get_sv (aTHX_ "Coro::POOL_RSS" , TRUE);	2959	sv_pool_rss = coro_get_sv (aTHX_ "Coro::POOL_RSS" , TRUE);
2435	sv_pool_size = coro_get_sv (aTHX_ "Coro::POOL_SIZE" , TRUE);	2960	sv_pool_size = coro_get_sv (aTHX_ "Coro::POOL_SIZE" , TRUE);
2436		2961	cv_coro_run = get_cv ( "Coro::_terminate", GV_ADD);
		2962	cv_coro_terminate = get_cv ( "Coro::terminate", GV_ADD);
2437	coro_current = coro_get_sv (aTHX_ "Coro::current", FALSE);	2963	coro_current = coro_get_sv (aTHX_ "Coro::current" , FALSE);
2438	SvREADONLY_on (coro_current);	2964	SvREADONLY_on (coro_current);
		2965
		2966	sv_async_pool_idle = newSVpv ("[async pool idle]", 0); SvREADONLY_on (sv_async_pool_idle);
		2967	sv_Coro = newSVpv ("Coro", 0); SvREADONLY_on (sv_Coro);
		2968	cv_pool_handler = get_cv ("Coro::_pool_handler", 0); SvREADONLY_on (cv_pool_handler);
		2969	cv_coro_new = get_cv ("Coro::new", 0); SvREADONLY_on (cv_coro_new);
2439		2970
2440	coro_stash = gv_stashpv ("Coro", TRUE);	2971	coro_stash = gv_stashpv ("Coro", TRUE);
2441		2972
2442	newCONSTSUB (coro_stash, "PRIO_MAX", newSViv (PRIO_MAX));	2973	newCONSTSUB (coro_stash, "PRIO_MAX", newSViv (PRIO_MAX));
2443	newCONSTSUB (coro_stash, "PRIO_HIGH", newSViv (PRIO_HIGH));	2974	newCONSTSUB (coro_stash, "PRIO_HIGH", newSViv (PRIO_HIGH));
…		…
2458	coroapi.ready = api_ready;	2989	coroapi.ready = api_ready;
2459	coroapi.is_ready = api_is_ready;	2990	coroapi.is_ready = api_is_ready;
2460	coroapi.nready = coro_nready;	2991	coroapi.nready = coro_nready;
2461	coroapi.current = coro_current;	2992	coroapi.current = coro_current;
2462		2993
2463	GCoroAPI = &coroapi;	2994	/*GCoroAPI = &coroapi;*/
2464	sv_setiv (sv, (IV)&coroapi);	2995	sv_setiv (sv, (IV)&coroapi);
2465	SvREADONLY_on (sv);	2996	SvREADONLY_on (sv);
2466	}	2997	}
2467	}	2998	}
2468		2999
…		…
2532	CODE:	3063	CODE:
2533	RETVAL = coro_nready;	3064	RETVAL = coro_nready;
2534	OUTPUT:	3065	OUTPUT:
2535	RETVAL	3066	RETVAL
2536		3067
2537	# for async_pool speedup
2538	void	3068	void
2539	_pool_1 (SV *cb)	3069	_pool_handler (...)
2540	CODE:	3070	CODE:
2541	{	3071	CORO_EXECUTE_SLF_XS (slf_init_pool_handler);
2542	HV *hv = (HV *)SvRV (coro_current);
2543	struct coro *coro = SvSTATE_hv ((SV *)hv);
2544	AV *defav = GvAV (PL_defgv);
2545	SV *invoke = hv_delete (hv, "_invoke", sizeof ("_invoke") - 1, 0);
2546	AV *invoke_av;
2547	int i, len;
2548
2549	if (!invoke)
2550	{
2551	SV *old = PL_diehook;
2552	PL_diehook = 0;
2553	SvREFCNT_dec (old);
2554	croak ("\3async_pool terminate\2\n");
2555	}
2556
2557	SvREFCNT_dec (coro->saved_deffh);
2558	coro->saved_deffh = SvREFCNT_inc_NN ((SV *)PL_defoutgv);
2559
2560	hv_store (hv, "desc", sizeof ("desc") - 1,
2561	newSVpvn ("[async_pool]", sizeof ("[async_pool]") - 1), 0);
2562
2563	invoke_av = (AV *)SvRV (invoke);
2564	len = av_len (invoke_av);
2565
2566	sv_setsv (cb, AvARRAY (invoke_av)[0]);
2567
2568	if (len > 0)
2569	{
2570	av_fill (defav, len - 1);
2571	for (i = 0; i < len; ++i)
2572	av_store (defav, i, SvREFCNT_inc_NN (AvARRAY (invoke_av)[i + 1]));
2573	}
2574	}
2575		3072
2576	void	3073	void
2577	_pool_2 (SV *cb)	3074	async_pool (SV *cv, ...)
2578	CODE:
2579	{
2580	HV *hv = (HV *)SvRV (coro_current);
2581	struct coro *coro = SvSTATE_hv ((SV *)hv);
2582
2583	sv_setsv (cb, &PL_sv_undef);
2584
2585	SvREFCNT_dec ((SV *)PL_defoutgv); PL_defoutgv = (GV *)coro->saved_deffh;
2586	coro->saved_deffh = 0;
2587
2588	if (coro_rss (aTHX_ coro) > SvUV (sv_pool_rss)
2589	|| av_len (av_async_pool) + 1 >= SvIV (sv_pool_size))
2590	{
2591	SV *old = PL_diehook;
2592	PL_diehook = 0;
2593	SvREFCNT_dec (old);
2594	croak ("\3async_pool terminate\2\n");
2595	}
2596
2597	av_clear (GvAV (PL_defgv));
2598	hv_store (hv, "desc", sizeof ("desc") - 1,
2599	newSVpvn ("[async_pool idle]", sizeof ("[async_pool idle]") - 1), 0);
2600
2601	coro->prio = 0;
2602
2603	if (coro->cctx && (coro->cctx->flags & CC_TRACE))
2604	api_trace (aTHX_ coro_current, 0);
2605
2606	av_push (av_async_pool, newSVsv (coro_current));
2607	}
2608
2609
2610	MODULE = Coro::State PACKAGE = Coro::AIO
2611
2612	void
2613	_get_state (SV *self)
2614	PROTOTYPE: $	3075	PROTOTYPE: &@
2615	PPCODE:	3076	PPCODE:
2616	{	3077	{
2617	AV *defav = GvAV (PL_defgv);	3078	HV *hv = (HV *)av_pop (av_async_pool);
2618	AV *av = newAV ();	3079	AV *av = newAV ();
		3080	SV *cb = ST (0);
2619	int i;	3081	int i;
2620	SV *data_sv = newSV (sizeof (struct io_state));
2621	struct io_state *data = (struct io_state *)SvPVX (data_sv);
2622	SvCUR_set (data_sv, sizeof (struct io_state));
2623	SvPOK_only (data_sv);
2624		3082
2625	data->errorno = errno;	3083	av_extend (av, items - 2);
2626	data->laststype = PL_laststype;	3084	for (i = 1; i < items; ++i)
2627	data->laststatval = PL_laststatval;
2628	data->statcache = PL_statcache;
2629
2630	av_extend (av, AvFILLp (defav) + 1 + 1);
2631
2632	for (i = 0; i <= AvFILLp (defav); ++i)
2633	av_push (av, SvREFCNT_inc_NN (AvARRAY (defav)[i]));	3085	av_push (av, SvREFCNT_inc_NN (ST (i)));
2634		3086
2635	av_push (av, data_sv);	3087	if ((SV *)hv == &PL_sv_undef)
2636
2637	XPUSHs (sv_2mortal (newRV_noinc ((SV *)av)));
2638
2639	api_ready (aTHX_ self);
2640	}
2641
2642	void
2643	_set_state (SV *state)
2644	PROTOTYPE: $
2645	PPCODE:
2646	{
2647	AV *av = (AV *)SvRV (state);
2648	struct io_state *data = (struct io_state *)SvPVX (AvARRAY (av)[AvFILLp (av)]);
2649	int i;
2650
2651	errno = data->errorno;
2652	PL_laststype = data->laststype;
2653	PL_laststatval = data->laststatval;
2654	PL_statcache = data->statcache;
2655
2656	EXTEND (SP, AvFILLp (av));
2657	for (i = 0; i < AvFILLp (av); ++i)
2658	PUSHs (sv_2mortal (SvREFCNT_inc_NN (AvARRAY (av)[i])));
2659	}
2660
2661
2662	MODULE = Coro::State PACKAGE = Coro::AnyEvent
2663
2664	BOOT:
2665	sv_activity = coro_get_sv (aTHX_ "Coro::AnyEvent::ACTIVITY", TRUE);
2666
2667	void
2668	_schedule (...)
2669	CODE:
2670	{
2671	static int incede;
2672
2673	api_cede_notself (aTHX);
2674
2675	++incede;
2676	while (coro_nready >= incede && api_cede (aTHX))
2677	;
2678
2679	sv_setsv (sv_activity, &PL_sv_undef);
2680	if (coro_nready >= incede)
2681	{	3088	{
2682	PUSHMARK (SP);	3089	PUSHMARK (SP);
		3090	EXTEND (SP, 2);
		3091	PUSHs (sv_Coro);
		3092	PUSHs ((SV *)cv_pool_handler);
2683	PUTBACK;	3093	PUTBACK;
2684	call_pv ("Coro::AnyEvent::_activity", G_DISCARD | G_EVAL);	3094	call_sv ((SV *)cv_coro_new, G_SCALAR);
2685	SPAGAIN;	3095	SPAGAIN;
		3096
		3097	hv = (HV *)SvREFCNT_inc_NN (SvRV (POPs));
2686	}	3098	}
2687		3099
2688	--incede;	3100	{
		3101	struct coro *coro = SvSTATE_hv (hv);
		3102
		3103	assert (!coro->invoke_cb);
		3104	assert (!coro->invoke_av);
		3105	coro->invoke_cb = SvREFCNT_inc (cb);
		3106	coro->invoke_av = av;
		3107	}
		3108
		3109	api_ready (aTHX_ (SV *)hv);
		3110
		3111	if (GIMME_V != G_VOID)
		3112	XPUSHs (sv_2mortal (newRV_noinc ((SV *)hv)));
		3113	else
		3114	SvREFCNT_dec (hv);
2689	}	3115	}
		3116
		3117	SV *
		3118	rouse_cb ()
		3119	PROTOTYPE:
		3120	CODE:
		3121	RETVAL = coro_new_rouse_cb (aTHX);
		3122	OUTPUT:
		3123	RETVAL
		3124
		3125	void
		3126	rouse_wait (...)
		3127	PROTOTYPE: ;$
		3128	PPCODE:
		3129	CORO_EXECUTE_SLF_XS (slf_init_rouse_wait);
2690		3130
2691		3131
2692	MODULE = Coro::State PACKAGE = PerlIO::cede	3132	MODULE = Coro::State PACKAGE = PerlIO::cede
2693		3133
2694	BOOT:	3134	BOOT:
2695	PerlIO_define_layer (aTHX_ &PerlIO_cede);	3135	PerlIO_define_layer (aTHX_ &PerlIO_cede);
2696		3136
		3137
2697	MODULE = Coro::State PACKAGE = Coro::Semaphore	3138	MODULE = Coro::State PACKAGE = Coro::Semaphore
2698		3139
2699	SV *	3140	SV *
2700	new (SV *klass, SV *count_ = 0)	3141	new (SV *klass, SV *count = 0)
2701	CODE:	3142	CODE:
2702	{	3143	RETVAL = sv_bless (
2703	/* a semaphore contains a counter IV in $sem->[0] and any waiters after that */	3144	coro_waitarray_new (aTHX_ count && SvOK (count) ? SvIV (count) : 1),
2704	AV *av = newAV ();	3145	GvSTASH (CvGV (cv))
2705	av_push (av, newSViv (count_ && SvOK (count_) ? SvIV (count_) : 1));	3146	);
2706	RETVAL = sv_bless (newRV_noinc ((SV *)av), GvSTASH (CvGV (cv)));	3147	OUTPUT:
2707	}	3148	RETVAL
		3149
		3150	# helper for Coro::Channel
		3151	SV *
		3152	_alloc (int count)
		3153	CODE:
		3154	RETVAL = coro_waitarray_new (aTHX_ count);
2708	OUTPUT:	3155	OUTPUT:
2709	RETVAL	3156	RETVAL
2710		3157
2711	SV *	3158	SV *
2712	count (SV *self)	3159	count (SV *self)
…		…
2721	adjust = 1	3168	adjust = 1
2722	CODE:	3169	CODE:
2723	coro_semaphore_adjust (aTHX_ (AV *)SvRV (self), ix ? adjust : 1);	3170	coro_semaphore_adjust (aTHX_ (AV *)SvRV (self), ix ? adjust : 1);
2724		3171
2725	void	3172	void
2726	down (SV *self)	3173	down (...)
2727	CODE:	3174	CODE:
2728	CORO_EXECUTE_SLF_XS (slf_init_semaphore_down);	3175	CORO_EXECUTE_SLF_XS (slf_init_semaphore_down);
		3176
		3177	void
		3178	wait (...)
		3179	CODE:
		3180	CORO_EXECUTE_SLF_XS (slf_init_semaphore_wait);
2729		3181
2730	void	3182	void
2731	try (SV *self)	3183	try (SV *self)
2732	PPCODE:	3184	PPCODE:
2733	{	3185	{
…		…
2745	XSRETURN_NO;	3197	XSRETURN_NO;
2746	}	3198	}
2747		3199
2748	void	3200	void
2749	waiters (SV *self)	3201	waiters (SV *self)
2750	CODE:	3202	PPCODE:
2751	{	3203	{
2752	AV *av = (AV *)SvRV (self);	3204	AV *av = (AV *)SvRV (self);
		3205	int wcount = AvFILLp (av) + 1 - 1;
2753		3206
2754	if (GIMME_V == G_SCALAR)	3207	if (GIMME_V == G_SCALAR)
2755	XPUSHs (sv_2mortal (newSVsv (AvARRAY (av)[0])));	3208	XPUSHs (sv_2mortal (newSViv (wcount)));
2756	else	3209	else
2757	{	3210	{
2758	int i;	3211	int i;
2759	EXTEND (SP, AvFILLp (av) + 1 - 1);	3212	EXTEND (SP, wcount);
2760	for (i = 1; i <= AvFILLp (av); ++i)	3213	for (i = 1; i <= wcount; ++i)
2761	PUSHs (newSVsv (AvARRAY (av)[i]));	3214	PUSHs (sv_2mortal (newRV_inc (AvARRAY (av)[i])));
2762	}	3215	}
2763	}	3216	}
2764		3217
		3218	MODULE = Coro::State PACKAGE = Coro::Signal
		3219
		3220	SV *
		3221	new (SV *klass)
		3222	CODE:
		3223	RETVAL = sv_bless (
		3224	coro_waitarray_new (aTHX_ 0),
		3225	GvSTASH (CvGV (cv))
		3226	);
		3227	OUTPUT:
		3228	RETVAL
		3229
		3230	void
		3231	wait (...)
		3232	CODE:
		3233	CORO_EXECUTE_SLF_XS (slf_init_signal_wait);
		3234
		3235	void
		3236	broadcast (SV *self)
		3237	CODE:
		3238	{
		3239	AV *av = (AV *)SvRV (self);
		3240	coro_signal_wake (aTHX_ av, AvFILLp (av));
		3241	}
		3242
		3243	void
		3244	send (SV *self)
		3245	CODE:
		3246	{
		3247	AV *av = (AV *)SvRV (self);
		3248
		3249	if (AvFILLp (av))
		3250	coro_signal_wake (aTHX_ av, 1);
		3251	else
		3252	SvIVX (AvARRAY (av)[0]) = 1; /* remember the signal */
		3253	}
		3254
		3255	IV
		3256	awaited (SV *self)
		3257	CODE:
		3258	RETVAL = AvFILLp ((AV *)SvRV (self)) + 1 - 1;
		3259	OUTPUT:
		3260	RETVAL
		3261
		3262
		3263	MODULE = Coro::State PACKAGE = Coro::AnyEvent
		3264
		3265	BOOT:
		3266	sv_activity = coro_get_sv (aTHX_ "Coro::AnyEvent::ACTIVITY", TRUE);
		3267
		3268	void
		3269	_schedule (...)
		3270	CODE:
		3271	{
		3272	static int incede;
		3273
		3274	api_cede_notself (aTHX);
		3275
		3276	++incede;
		3277	while (coro_nready >= incede && api_cede (aTHX))
		3278	;
		3279
		3280	sv_setsv (sv_activity, &PL_sv_undef);
		3281	if (coro_nready >= incede)
		3282	{
		3283	PUSHMARK (SP);
		3284	PUTBACK;
		3285	call_pv ("Coro::AnyEvent::_activity", G_KEEPERR | G_EVAL | G_VOID | G_DISCARD);
		3286	}
		3287
		3288	--incede;
		3289	}
		3290
		3291
		3292	MODULE = Coro::State PACKAGE = Coro::AIO
		3293
		3294	void
		3295	_register (char *target, char *proto, SV *req)
		3296	CODE:
		3297	{
		3298	CV *req_cv = coro_sv_2cv (aTHX_ req);
		3299	/* newXSproto doesn't return the CV on 5.8 */
		3300	CV *slf_cv = newXS (target, coro_aio_req_xs, __FILE__);
		3301	sv_setpv ((SV *)slf_cv, proto);
		3302	sv_magicext ((SV *)slf_cv, (SV *)req_cv, CORO_MAGIC_type_aio, 0, 0, 0);
		3303	}
		3304

Diff Legend

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