ViewVC Help

View File | Revision Log | Show Annotations | Download File

/cvs/Coro/Coro/State.xs

Comparing Coro/Coro/State.xs (file contents):
Revision 1.280 by root, Sun Nov 16 09:43:18 2008 UTC vs.
Revision 1.310 by root, Wed Nov 19 16:01:32 2008 UTC

…		…
16		16
17	#ifdef WIN32	17	#ifdef WIN32
18	# undef setjmp	18	# undef setjmp
19	# undef longjmp	19	# undef longjmp
20	# undef _exit	20	# undef _exit
21	# define setjmp _setjmp // deep magic, don't ask	21	# define setjmp _setjmp /* deep magic */
22	#else	22	#else
23	# include <inttypes.h> /* most portable stdint.h */	23	# include <inttypes.h> /* most portable stdint.h */
24	#endif	24	#endif
25		25
26	#ifdef HAVE_MMAP	26	#ifdef HAVE_MMAP
…		…
95	# define GV_NOTQUAL 0	95	# define GV_NOTQUAL 0
96	#endif	96	#endif
97	#ifndef newSV	97	#ifndef newSV
98	# define newSV(l) NEWSV(0,l)	98	# define newSV(l) NEWSV(0,l)
99	#endif	99	#endif
		100	#ifndef CvISXSUB_on
		101	# define CvISXSUB_on(cv) (void)cv
		102	#endif
100		103
101	/* 5.8.7 */	104	/* 5.8.7 */
102	#ifndef SvRV_set	105	#ifndef SvRV_set
103	# define SvRV_set(s,v) SvRV(s) = (v)	106	# define SvRV_set(s,v) SvRV(s) = (v)
104	#endif	107	#endif
…		…
117	#endif	120	#endif
118		121
119	/* The next macros try to return the current stack pointer, in an as	122	/* The next macros try to return the current stack pointer, in an as
120	* portable way as possible. */	123	* portable way as possible. */
121	#if __GNUC__ >= 4	124	#if __GNUC__ >= 4
		125	# define dSTACKLEVEL int stacklevel_dummy
122	# define dSTACKLEVEL void *stacklevel = __builtin_frame_address (0)	126	# define STACKLEVEL __builtin_frame_address (0)
123	#else	127	#else
124	# define dSTACKLEVEL volatile void *stacklevel = (volatile void *)&stacklevel	128	# define dSTACKLEVEL volatile void *stacklevel
		129	# define STACKLEVEL ((void *)&stacklevel)
125	#endif	130	#endif
126		131
127	#define IN_DESTRUCT (PL_main_cv == Nullcv)	132	#define IN_DESTRUCT (PL_main_cv == Nullcv)
128		133
129	#if __GNUC__ >= 3	134	#if __GNUC__ >= 3
…		…
140	#define expect_true(expr) expect ((expr) != 0, 1)	145	#define expect_true(expr) expect ((expr) != 0, 1)
141		146
142	#define NOINLINE attribute ((noinline))	147	#define NOINLINE attribute ((noinline))
143		148
144	#include "CoroAPI.h"	149	#include "CoroAPI.h"
		150	#define GCoroAPI (&coroapi) /* very sneaky */
145		151
146	#ifdef USE_ITHREADS	152	#ifdef USE_ITHREADS
147	# if CORO_PTHREAD	153	# if CORO_PTHREAD
148	static void *coro_thx;	154	static void *coro_thx;
149	# endif	155	# endif
150	#endif	156	#endif
151		157
152	/* helper storage struct for Coro::AIO */
153	struct io_state
154	{
155	AV *res;
156	int errorno;
157	I32 laststype; /* U16 in 5.10.0 */
158	int laststatval;
159	Stat_t statcache;
160	};
161
162	static double (*nvtime)(); /* so why doesn't it take void? */	158	static double (*nvtime)(); /* so why doesn't it take void? */
		159
		160	/* we hijack an hopefully unused CV flag for our purposes */
		161	#define CVf_SLF 0x4000
		162	static OP *pp_slf (pTHX);
163		163
164	static U32 cctx_gen;	164	static U32 cctx_gen;
165	static size_t cctx_stacksize = CORO_STACKSIZE;	165	static size_t cctx_stacksize = CORO_STACKSIZE;
166	static struct CoroAPI coroapi;	166	static struct CoroAPI coroapi;
167	static AV *main_mainstack; /* used to differentiate between $main and others */	167	static AV *main_mainstack; /* used to differentiate between $main and others */
168	static JMPENV *main_top_env;	168	static JMPENV *main_top_env;
169	static HV *coro_state_stash, *coro_stash;	169	static HV *coro_state_stash, *coro_stash;
170	static volatile SV *coro_mortal; /* will be freed/thrown after next transfer */	170	static volatile SV *coro_mortal; /* will be freed/thrown after next transfer */
171	static volatile struct coro *transfer_next;
172		171
173	static GV *irsgv; /* $/ */	172	static GV *irsgv; /* $/ */
174	static GV *stdoutgv; /* *STDOUT */	173	static GV *stdoutgv; /* *STDOUT */
175	static SV *rv_diehook;	174	static SV *rv_diehook;
176	static SV *rv_warnhook;	175	static SV *rv_warnhook;
…		…
242	/* this is a structure representing a perl-level coroutine */	241	/* this is a structure representing a perl-level coroutine */
243	struct coro {	242	struct coro {
244	/* the C coroutine allocated to this perl coroutine, if any */	243	/* the C coroutine allocated to this perl coroutine, if any */
245	coro_cctx *cctx;	244	coro_cctx *cctx;
246		245
247	/* process data */	246	/* state data */
248	struct CoroSLF slf_frame; /* saved slf frame */	247	struct CoroSLF slf_frame; /* saved slf frame */
249	AV *mainstack;	248	AV *mainstack;
250	perl_slots *slot; /* basically the saved sp */	249	perl_slots *slot; /* basically the saved sp */
251		250
		251	CV *startcv; /* the CV to execute */
252	AV *args; /* data associated with this coroutine (initial args) */	252	AV *args; /* data associated with this coroutine (initial args) */
253	int refcnt; /* coroutines are refcounted, yes */	253	int refcnt; /* coroutines are refcounted, yes */
254	int flags; /* CF_ flags */	254	int flags; /* CF_ flags */
255	HV *hv; /* the perl hash associated with this coro, if any */	255	HV *hv; /* the perl hash associated with this coro, if any */
256	void (*on_destroy)(pTHX_ struct coro *coro);	256	void (*on_destroy)(pTHX_ struct coro *coro);
257		257
258	/* statistics */	258	/* statistics */
259	int usecount; /* number of transfers to this coro */	259	int usecount; /* number of transfers to this coro */
260		260
261	/* coro process data */	261	/* coro process data */
262	int prio;	262	int prio;
263	SV *throw; /* exception to be thrown */	263	SV *except; /* exception to be thrown */
		264	SV *rouse_cb;
264		265
265	/* async_pool */	266	/* async_pool */
266	SV *saved_deffh;	267	SV *saved_deffh;
267		268
268	/* linked list */	269	/* linked list */
…		…
270	};	271	};
271		272
272	typedef struct coro *Coro__State;	273	typedef struct coro *Coro__State;
273	typedef struct coro *Coro__State_or_hashref;	274	typedef struct coro *Coro__State_or_hashref;
274		275
		276	/* the following variables are effectively part of the perl context */
		277	/* and get copied between struct coro and these variables */
		278	/* the mainr easonw e don't support windows process emulation */
275	static struct CoroSLF slf_frame; /* the current slf frame */	279	static struct CoroSLF slf_frame; /* the current slf frame */
276		280
277	/** Coro ********************************************************************/	281	/** Coro ********************************************************************/
278		282
279	#define PRIO_MAX 3	283	#define PRIO_MAX 3
…		…
285		289
286	/* for Coro.pm */	290	/* for Coro.pm */
287	static SV *coro_current;	291	static SV *coro_current;
288	static SV *coro_readyhook;	292	static SV *coro_readyhook;
289	static AV *coro_ready [PRIO_MAX - PRIO_MIN + 1];	293	static AV *coro_ready [PRIO_MAX - PRIO_MIN + 1];
		294	static CV *cv_coro_run;
290	static struct coro *coro_first;	295	static struct coro *coro_first;
291	#define coro_nready coroapi.nready	296	#define coro_nready coroapi.nready
292		297
293	/** lowlevel stuff **********************************************************/	298	/** lowlevel stuff **********************************************************/
294		299
…		…
318	#if PERL_VERSION_ATLEAST (5,10,0)	323	#if PERL_VERSION_ATLEAST (5,10,0)
319	/* silence stupid and wrong 5.10 warning that I am unable to switch off */	324	/* silence stupid and wrong 5.10 warning that I am unable to switch off */
320	get_hv (name, create);	325	get_hv (name, create);
321	#endif	326	#endif
322	return get_hv (name, create);	327	return get_hv (name, create);
		328	}
		329
		330	/* may croak */
		331	INLINE CV *
		332	coro_sv_2cv (SV *sv)
		333	{
		334	HV *st;
		335	GV *gvp;
		336	return sv_2cv (sv, &st, &gvp, 0);
323	}	337	}
324		338
325	static AV *	339	static AV *
326	coro_clone_padlist (pTHX_ CV *cv)	340	coro_clone_padlist (pTHX_ CV *cv)
327	{	341	{
…		…
381	SvREFCNT_dec (av); /* sv_magicext increased the refcount */	395	SvREFCNT_dec (av); /* sv_magicext increased the refcount */
382		396
383	return 0;	397	return 0;
384	}	398	}
385		399
386	#define CORO_MAGIC_type_cv PERL_MAGIC_ext	400	#define CORO_MAGIC_type_cv 26
387	#define CORO_MAGIC_type_state PERL_MAGIC_ext	401	#define CORO_MAGIC_type_state PERL_MAGIC_ext
388		402
389	static MGVTBL coro_cv_vtbl = {	403	static MGVTBL coro_cv_vtbl = {
390	0, 0, 0, 0,	404	0, 0, 0, 0,
391	coro_cv_free	405	coro_cv_free
392	};	406	};
393		407
		408	#define CORO_MAGIC_NN(sv, type) \
		409	(expect_true (SvMAGIC (sv)->mg_type == type) \
		410	? SvMAGIC (sv) \
		411	: mg_find (sv, type))
		412
394	#define CORO_MAGIC(sv, type) \	413	#define CORO_MAGIC(sv, type) \
395	SvMAGIC (sv) \	414	(expect_true (SvMAGIC (sv)) \
396	? SvMAGIC (sv)->mg_type == type \	415	? CORO_MAGIC_NN (sv, type) \
397	? SvMAGIC (sv) \
398	: mg_find (sv, type) \
399	: 0	416	: 0)
400		417
401	#define CORO_MAGIC_cv(cv) CORO_MAGIC (((SV *)(cv)), CORO_MAGIC_type_cv)	418	#define CORO_MAGIC_cv(cv) CORO_MAGIC (((SV *)(cv)), CORO_MAGIC_type_cv)
402	#define CORO_MAGIC_state(sv) CORO_MAGIC (((SV *)(sv)), CORO_MAGIC_type_state)	419	#define CORO_MAGIC_state(sv) CORO_MAGIC_NN (((SV *)(sv)), CORO_MAGIC_type_state)
403		420
404	INLINE struct coro *	421	INLINE struct coro *
405	SvSTATE_ (pTHX_ SV *coro)	422	SvSTATE_ (pTHX_ SV *coro)
406	{	423	{
407	HV *stash;	424	HV *stash;
…		…
424	mg = CORO_MAGIC_state (coro);	441	mg = CORO_MAGIC_state (coro);
425	return (struct coro *)mg->mg_ptr;	442	return (struct coro *)mg->mg_ptr;
426	}	443	}
427		444
428	#define SvSTATE(sv) SvSTATE_ (aTHX_ (sv))	445	#define SvSTATE(sv) SvSTATE_ (aTHX_ (sv))
		446
		447	/* faster than SvSTATE, but expects a coroutine hv */
		448	#define SvSTATE_hv(hv) ((struct coro *)CORO_MAGIC_NN ((SV *)hv, CORO_MAGIC_type_state)->mg_ptr)
		449	#define SvSTATE_current SvSTATE_hv (SvRV (coro_current))
429		450
430	/* the next two functions merely cache the padlists */	451	/* the next two functions merely cache the padlists */
431	static void	452	static void
432	get_padlist (pTHX_ CV *cv)	453	get_padlist (pTHX_ CV *cv)
433	{	454	{
…		…
439	else	460	else
440	{	461	{
441	#if CORO_PREFER_PERL_FUNCTIONS	462	#if CORO_PREFER_PERL_FUNCTIONS
442	/* this is probably cleaner? but also slower! */	463	/* this is probably cleaner? but also slower! */
443	/* in practise, it seems to be less stable */	464	/* in practise, it seems to be less stable */
444	CV *cp = Perl_cv_clone (cv);	465	CV *cp = Perl_cv_clone (aTHX_ cv);
445	CvPADLIST (cv) = CvPADLIST (cp);	466	CvPADLIST (cv) = CvPADLIST (cp);
446	CvPADLIST (cp) = 0;	467	CvPADLIST (cp) = 0;
447	SvREFCNT_dec (cp);	468	SvREFCNT_dec (cp);
448	#else	469	#else
449	CvPADLIST (cv) = coro_clone_padlist (aTHX_ cv);	470	CvPADLIST (cv) = coro_clone_padlist (aTHX_ cv);
…		…
501	}	522	}
502		523
503	PUTBACK;	524	PUTBACK;
504	}	525	}
505		526
506	slf_frame = c->slf_frame;	527	slf_frame = c->slf_frame;
		528	CORO_THROW = c->except;
507	}	529	}
508		530
509	static void	531	static void
510	save_perl (pTHX_ Coro__State c)	532	save_perl (pTHX_ Coro__State c)
511	{	533	{
		534	c->except = CORO_THROW;
512	c->slf_frame = slf_frame;	535	c->slf_frame = slf_frame;
513		536
514	{	537	{
515	dSP;	538	dSP;
516	I32 cxix = cxstack_ix;	539	I32 cxix = cxstack_ix;
…		…
591	* of perl.c:init_stacks, except that it uses less memory	614	* of perl.c:init_stacks, except that it uses less memory
592	* on the (sometimes correct) assumption that coroutines do	615	* on the (sometimes correct) assumption that coroutines do
593	* not usually need a lot of stackspace.	616	* not usually need a lot of stackspace.
594	*/	617	*/
595	#if CORO_PREFER_PERL_FUNCTIONS	618	#if CORO_PREFER_PERL_FUNCTIONS
596	# define coro_init_stacks init_stacks	619	# define coro_init_stacks(thx) init_stacks ()
597	#else	620	#else
598	static void	621	static void
599	coro_init_stacks (pTHX)	622	coro_init_stacks (pTHX)
600	{	623	{
601	PL_curstackinfo = new_stackinfo(32, 8);	624	PL_curstackinfo = new_stackinfo(32, 8);
…		…
809	slf_check_nop (pTHX_ struct CoroSLF *frame)	832	slf_check_nop (pTHX_ struct CoroSLF *frame)
810	{	833	{
811	return 0;	834	return 0;
812	}	835	}
813		836
814	static void	837	static UNOP coro_setup_op;
		838
		839	static void NOINLINE /* noinline to keep it out of the transfer fast path */
815	coro_setup (pTHX_ struct coro *coro)	840	coro_setup (pTHX_ struct coro *coro)
816	{	841	{
817	/*	842	/*
818	* emulate part of the perl startup here.	843	* emulate part of the perl startup here.
819	*/	844	*/
…		…
846	{	871	{
847	dSP;	872	dSP;
848	UNOP myop;	873	UNOP myop;
849		874
850	Zero (&myop, 1, UNOP);	875	Zero (&myop, 1, UNOP);
851	myop.op_next = Nullop;	876	myop.op_next = Nullop;
		877	myop.op_type = OP_ENTERSUB;
852	myop.op_flags = OPf_WANT_VOID;	878	myop.op_flags = OPf_WANT_VOID;
853		879
854	PUSHMARK (SP);	880	PUSHMARK (SP);
855	XPUSHs (sv_2mortal (av_shift (GvAV (PL_defgv))));	881	PUSHs ((SV *)coro->startcv);
856	PUTBACK;	882	PUTBACK;
857	PL_op = (OP *)&myop;	883	PL_op = (OP *)&myop;
858	PL_op = PL_ppaddr[OP_ENTERSUB](aTHX);	884	PL_op = PL_ppaddr[OP_ENTERSUB](aTHX);
859	SPAGAIN;
860	}	885	}
861		886
862	/* this newly created coroutine might be run on an existing cctx which most	887	/* this newly created coroutine might be run on an existing cctx which most
863	* likely was suspended in pp_slf, so we have to emulate entering pp_slf here.	888	* likely was suspended in pp_slf, so we have to emulate entering pp_slf here.
864	*/	889	*/
865	slf_frame.prepare = prepare_nop; /* provide a nop function for an eventual pp_slf */	890	slf_frame.prepare = prepare_nop; /* provide a nop function for an eventual pp_slf */
866	slf_frame.check = slf_check_nop; /* signal pp_slf to not repeat */	891	slf_frame.check = slf_check_nop; /* signal pp_slf to not repeat */
		892
		893	/* and we have to provide the pp_slf op in any case, so pp_slf can skip it */
		894	coro_setup_op.op_next = PL_op;
		895	coro_setup_op.op_type = OP_CUSTOM;
		896	coro_setup_op.op_ppaddr = pp_slf;
		897	/* no flags etc. required, as an init function won't be called */
		898
		899	PL_op = (OP *)&coro_setup_op;
		900
		901	/* copy throw, in case it was set before coro_setup */
		902	CORO_THROW = coro->except;
867	}	903	}
868		904
869	static void	905	static void
870	coro_destruct (pTHX_ struct coro *coro)	906	coro_destruct (pTHX_ struct coro *coro)
871	{	907	{
…		…
895		931
896	SvREFCNT_dec (PL_diehook);	932	SvREFCNT_dec (PL_diehook);
897	SvREFCNT_dec (PL_warnhook);	933	SvREFCNT_dec (PL_warnhook);
898		934
899	SvREFCNT_dec (coro->saved_deffh);	935	SvREFCNT_dec (coro->saved_deffh);
900	SvREFCNT_dec (coro->throw);	936	SvREFCNT_dec (coro->rouse_cb);
901		937
902	coro_destruct_stacks (aTHX);	938	coro_destruct_stacks (aTHX);
903	}	939	}
904		940
905	INLINE void	941	INLINE void
…		…
915	static int	951	static int
916	runops_trace (pTHX)	952	runops_trace (pTHX)
917	{	953	{
918	COP *oldcop = 0;	954	COP *oldcop = 0;
919	int oldcxix = -2;	955	int oldcxix = -2;
920	struct coro *coro = SvSTATE (coro_current); /* trace cctx is tied to specific coro */	956	struct coro *coro = SvSTATE_current; /* trace cctx is tied to specific coro */
921	coro_cctx *cctx = coro->cctx;	957	coro_cctx *cctx = coro->cctx;
922		958
923	while ((PL_op = CALL_FPTR (PL_op->op_ppaddr) (aTHX)))	959	while ((PL_op = CALL_FPTR (PL_op->op_ppaddr) (aTHX)))
924	{	960	{
925	PERL_ASYNC_CHECK ();	961	PERL_ASYNC_CHECK ();
…		…
992	SAVETMPS;	1028	SAVETMPS;
993	EXTEND (SP, 3);	1029	EXTEND (SP, 3);
994	PUSHMARK (SP);	1030	PUSHMARK (SP);
995	PUSHs (&PL_sv_yes);	1031	PUSHs (&PL_sv_yes);
996	PUSHs (fullname);	1032	PUSHs (fullname);
997	PUSHs (CxHASARGS (cx) ? sv_2mortal (newRV_inc ((SV *)cx->blk_sub.argarray)) : &PL_sv_undef);	1033	PUSHs (CxHASARGS (cx) ? sv_2mortal (newRV_inc ((SV *)cx->blk_sub.argarray)) : &PL_sv_undef);
998	PUTBACK;	1034	PUTBACK;
999	cb = hv_fetch ((HV *)SvRV (coro_current), "_trace_sub_cb", sizeof ("_trace_sub_cb") - 1, 0);	1035	cb = hv_fetch ((HV *)SvRV (coro_current), "_trace_sub_cb", sizeof ("_trace_sub_cb") - 1, 0);
1000	if (cb) call_sv (*cb, G_KEEPERR | G_EVAL | G_VOID | G_DISCARD);	1036	if (cb) call_sv (*cb, G_KEEPERR | G_EVAL | G_VOID | G_DISCARD);
1001	SPAGAIN;	1037	SPAGAIN;
1002	FREETMPS;	1038	FREETMPS;
…		…
1034		1070
1035	TAINT_NOT;	1071	TAINT_NOT;
1036	return 0;	1072	return 0;
1037	}	1073	}
1038		1074
		1075	static struct coro_cctx *cctx_ssl_cctx;
		1076	static struct CoroSLF cctx_ssl_frame;
		1077
1039	static void	1078	static void
1040	prepare_set_stacklevel (struct coro_transfer_args *ta, struct coro_cctx *cctx)	1079	slf_prepare_set_stacklevel (pTHX_ struct coro_transfer_args *ta)
1041	{	1080	{
1042	ta->prev = (struct coro *)cctx;	1081	ta->prev = (struct coro *)cctx_ssl_cctx;
1043	ta->next = 0;	1082	ta->next = 0;
1044	}	1083	}
1045		1084
1046	/* inject a fake call to Coro::State::_cctx_init into the execution */	1085	static int
1047	/* _cctx_init should be careful, as it could be called at almost any time */	1086	slf_check_set_stacklevel (pTHX_ struct CoroSLF *frame)
1048	/* during execution of a perl program */	1087	{
1049	/* also initialises PL_top_env */	1088	*frame = cctx_ssl_frame;
		1089
		1090	return frame->check (aTHX_ frame); /* execute the restored frame - there must be one */
		1091	}
		1092
		1093	/* initialises PL_top_env and injects a pseudo-slf-call to set the stacklevel */
1050	static void NOINLINE	1094	static void NOINLINE
1051	cctx_prepare (pTHX_ coro_cctx *cctx)	1095	cctx_prepare (pTHX_ coro_cctx *cctx)
1052	{	1096	{
1053	dSP;
1054	UNOP myop;
1055
1056	PL_top_env = &PL_start_env;	1097	PL_top_env = &PL_start_env;
1057		1098
1058	if (cctx->flags & CC_TRACE)	1099	if (cctx->flags & CC_TRACE)
1059	PL_runops = runops_trace;	1100	PL_runops = runops_trace;
1060		1101
1061	Zero (&myop, 1, UNOP);	1102	/* we already must be executing an SLF op, there is no other valid way
1062	myop.op_next = PL_op;	1103	* that can lead to creation of a new cctx */
1063	myop.op_flags = OPf_WANT_VOID | OPf_STACKED;	1104	assert (("FATAL: can't prepare slf-less cctx in Coro module (please report)",
		1105	slf_frame.prepare && PL_op->op_ppaddr == pp_slf));
1064		1106
1065	PUSHMARK (SP);	1107	/* we must emulate leaving pp_slf, which is done inside slf_check_set_stacklevel */
1066	EXTEND (SP, 2);	1108	cctx_ssl_cctx = cctx;
1067	PUSHs (sv_2mortal (newSViv ((IV)cctx)));	1109	cctx_ssl_frame = slf_frame;
1068	PUSHs ((SV *)get_cv ("Coro::State::_cctx_init", FALSE));	1110
1069	PUTBACK;	1111	slf_frame.prepare = slf_prepare_set_stacklevel;
1070	PL_op = (OP *)&myop;	1112	slf_frame.check = slf_check_set_stacklevel;
1071	PL_op = PL_ppaddr[OP_ENTERSUB](aTHX);
1072	SPAGAIN;
1073	}	1113	}
1074		1114
1075	/* the tail of transfer: execute stuff we can only do after a transfer */	1115	/* the tail of transfer: execute stuff we can only do after a transfer */
1076	INLINE void	1116	INLINE void
1077	transfer_tail (pTHX)	1117	transfer_tail (pTHX)
1078	{	1118	{
1079	struct coro *next = (struct coro *)transfer_next;
1080	assert (!(transfer_next = 0)); /* just used for the side effect when asserts are enabled */
1081	assert (("FATAL: next coroutine was zero in transfer_tail (please report)", next));
1082
1083	free_coro_mortal (aTHX);	1119	free_coro_mortal (aTHX);
1084
1085	if (expect_false (next->throw))
1086	{
1087	SV *exception = sv_2mortal (next->throw);
1088
1089	next->throw = 0;
1090	sv_setsv (ERRSV, exception);
1091	croak (0);
1092	}
1093	}	1120	}
1094		1121
1095	/*	1122	/*
1096	* this is a _very_ stripped down perl interpreter ;)	1123	* this is a _very_ stripped down perl interpreter ;)
1097	*/	1124	*/
…		…
1112		1139
1113	/* inject a fake subroutine call to cctx_init */	1140	/* inject a fake subroutine call to cctx_init */
1114	cctx_prepare (aTHX_ (coro_cctx *)arg);	1141	cctx_prepare (aTHX_ (coro_cctx *)arg);
1115		1142
1116	/* cctx_run is the alternative tail of transfer() */	1143	/* cctx_run is the alternative tail of transfer() */
1117	/* TODO: throwing an exception here might be deadly, VERIFY */
1118	transfer_tail (aTHX);	1144	transfer_tail (aTHX);
1119		1145
1120	/* somebody or something will hit me for both perl_run and PL_restartop */	1146	/* somebody or something will hit me for both perl_run and PL_restartop */
1121	PL_restartop = PL_op;	1147	PL_restartop = PL_op;
1122	perl_run (PL_curinterp);	1148	perl_run (PL_curinterp);
		1149	/*
		1150	* Unfortunately, there is no way to get at the return values of the
		1151	* coro body here, as perl_run destroys these
		1152	*/
1123		1153
1124	/*	1154	/*
1125	* If perl-run returns we assume exit() was being called or the coro	1155	* If perl-run returns we assume exit() was being called or the coro
1126	* fell off the end, which seems to be the only valid (non-bug)	1156	* fell off the end, which seems to be the only valid (non-bug)
1127	* reason for perl_run to return. We try to exit by jumping to the	1157	* reason for perl_run to return. We try to exit by jumping to the
…		…
1278	/** coroutine switching *****************************************************/	1308	/** coroutine switching *****************************************************/
1279		1309
1280	static void	1310	static void
1281	transfer_check (pTHX_ struct coro *prev, struct coro *next)	1311	transfer_check (pTHX_ struct coro *prev, struct coro *next)
1282	{	1312	{
		1313	/* TODO: throwing up here is considered harmful */
		1314
1283	if (expect_true (prev != next))	1315	if (expect_true (prev != next))
1284	{	1316	{
1285	if (expect_false (!(prev->flags & (CF_RUNNING | CF_NEW))))	1317	if (expect_false (!(prev->flags & (CF_RUNNING | CF_NEW))))
1286	croak ("Coro::State::transfer called with non-running/new prev Coro::State, but can only transfer from running or new states,");	1318	croak ("Coro::State::transfer called with non-running/new prev Coro::State, but can only transfer from running or new states,");
1287		1319
…		…
1297	#endif	1329	#endif
1298	}	1330	}
1299	}	1331	}
1300		1332
1301	/* always use the TRANSFER macro */	1333	/* always use the TRANSFER macro */
1302	static void NOINLINE	1334	static void NOINLINE /* noinline so we have a fixed stackframe */
1303	transfer (pTHX_ struct coro *prev, struct coro *next, int force_cctx)	1335	transfer (pTHX_ struct coro *prev, struct coro *next, int force_cctx)
1304	{	1336	{
1305	dSTACKLEVEL;	1337	dSTACKLEVEL;
1306		1338
1307	/* sometimes transfer is only called to set idle_sp */	1339	/* sometimes transfer is only called to set idle_sp */
1308	if (expect_false (!next))	1340	if (expect_false (!next))
1309	{	1341	{
1310	((coro_cctx *)prev)->idle_sp = (void *)stacklevel;	1342	((coro_cctx *)prev)->idle_sp = STACKLEVEL;
1311	assert (((coro_cctx *)prev)->idle_te = PL_top_env); /* just for the side-effect when asserts are enabled */	1343	assert (((coro_cctx *)prev)->idle_te = PL_top_env); /* just for the side-effect when asserts are enabled */
1312	}	1344	}
1313	else if (expect_true (prev != next))	1345	else if (expect_true (prev != next))
1314	{	1346	{
1315	coro_cctx *prev__cctx;	1347	coro_cctx *prev__cctx;
…		…
1340		1372
1341	prev__cctx = prev->cctx;	1373	prev__cctx = prev->cctx;
1342		1374
1343	/* possibly untie and reuse the cctx */	1375	/* possibly untie and reuse the cctx */
1344	if (expect_true (	1376	if (expect_true (
1345	prev__cctx->idle_sp == (void *)stacklevel	1377	prev__cctx->idle_sp == STACKLEVEL
1346	&& !(prev__cctx->flags & CC_TRACE)	1378	&& !(prev__cctx->flags & CC_TRACE)
1347	&& !force_cctx	1379	&& !force_cctx
1348	))	1380	))
1349	{	1381	{
1350	/* I assume that stacklevel is a stronger indicator than PL_top_env changes */	1382	/* I assume that stacklevel is a stronger indicator than PL_top_env changes */
…		…
1364	++next->usecount;	1396	++next->usecount;
1365		1397
1366	if (expect_true (!next->cctx))	1398	if (expect_true (!next->cctx))
1367	next->cctx = cctx_get (aTHX);	1399	next->cctx = cctx_get (aTHX);
1368		1400
1369	assert (("FATAL: transfer_next already nonzero in Coro (please report)", !transfer_next));
1370	transfer_next = next;
1371
1372	if (expect_false (prev__cctx != next->cctx))	1401	if (expect_false (prev__cctx != next->cctx))
1373	{	1402	{
1374	prev__cctx->top_env = PL_top_env;	1403	prev__cctx->top_env = PL_top_env;
1375	PL_top_env = next->cctx->top_env;	1404	PL_top_env = next->cctx->top_env;
1376	coro_transfer (&prev__cctx->cctx, &next->cctx->cctx);	1405	coro_transfer (&prev__cctx->cctx, &next->cctx->cctx);
…		…
1420		1449
1421	coro->slot = 0;	1450	coro->slot = 0;
1422	}	1451	}
1423		1452
1424	cctx_destroy (coro->cctx);	1453	cctx_destroy (coro->cctx);
		1454	SvREFCNT_dec (coro->startcv);
1425	SvREFCNT_dec (coro->args);	1455	SvREFCNT_dec (coro->args);
		1456	SvREFCNT_dec (CORO_THROW);
1426		1457
1427	if (coro->next) coro->next->prev = coro->prev;	1458	if (coro->next) coro->next->prev = coro->prev;
1428	if (coro->prev) coro->prev->next = coro->next;	1459	if (coro->prev) coro->prev->next = coro->next;
1429	if (coro == coro_first) coro_first = coro->next;	1460	if (coro == coro_first) coro_first = coro->next;
1430		1461
…		…
1484		1515
1485	prepare_transfer (aTHX_ &ta, prev_sv, next_sv);	1516	prepare_transfer (aTHX_ &ta, prev_sv, next_sv);
1486	TRANSFER (ta, 1);	1517	TRANSFER (ta, 1);
1487	}	1518	}
1488		1519
		1520	/*****************************************************************************/
		1521	/* gensub: simple closure generation utility */
		1522
		1523	#define GENSUB_ARG CvXSUBANY (cv).any_ptr
		1524
		1525	/* create a closure from XS, returns a code reference */
		1526	/* the arg can be accessed via GENSUB_ARG from the callback */
		1527	/* the callback must use dXSARGS/XSRETURN */
		1528	static SV *
		1529	gensub (pTHX_ void (*xsub)(pTHX_ CV *), void *arg)
		1530	{
		1531	CV *cv = (CV *)newSV (0);
		1532
		1533	sv_upgrade ((SV *)cv, SVt_PVCV);
		1534
		1535	CvANON_on (cv);
		1536	CvISXSUB_on (cv);
		1537	CvXSUB (cv) = xsub;
		1538	GENSUB_ARG = arg;
		1539
		1540	return newRV_noinc ((SV *)cv);
		1541	}
		1542
1489	/** Coro ********************************************************************/	1543	/** Coro ********************************************************************/
1490		1544
1491	static void	1545	INLINE void
1492	coro_enq (pTHX_ SV *coro_sv)	1546	coro_enq (pTHX_ struct coro *coro)
1493	{	1547	{
1494	av_push (coro_ready [SvSTATE (coro_sv)->prio - PRIO_MIN], coro_sv);	1548	av_push (coro_ready [coro->prio - PRIO_MIN], SvREFCNT_inc_NN (coro->hv));
1495	}	1549	}
1496		1550
1497	static SV *	1551	INLINE SV *
1498	coro_deq (pTHX)	1552	coro_deq (pTHX)
1499	{	1553	{
1500	int prio;	1554	int prio;
1501		1555
1502	for (prio = PRIO_MAX - PRIO_MIN + 1; --prio >= 0; )	1556	for (prio = PRIO_MAX - PRIO_MIN + 1; --prio >= 0; )
…		…
1524	coro->flags |= CF_READY;	1578	coro->flags |= CF_READY;
1525		1579
1526	sv_hook = coro_nready ? 0 : coro_readyhook;	1580	sv_hook = coro_nready ? 0 : coro_readyhook;
1527	xs_hook = coro_nready ? 0 : coroapi.readyhook;	1581	xs_hook = coro_nready ? 0 : coroapi.readyhook;
1528		1582
1529	coro_enq (aTHX_ SvREFCNT_inc_NN (coro_sv));	1583	coro_enq (aTHX_ coro);
1530	++coro_nready;	1584	++coro_nready;
1531		1585
1532	if (sv_hook)	1586	if (sv_hook)
1533	{	1587	{
1534	dSP;	1588	dSP;
…		…
1536	ENTER;	1590	ENTER;
1537	SAVETMPS;	1591	SAVETMPS;
1538		1592
1539	PUSHMARK (SP);	1593	PUSHMARK (SP);
1540	PUTBACK;	1594	PUTBACK;
1541	call_sv (sv_hook, G_DISCARD);	1595	call_sv (sv_hook, G_VOID | G_DISCARD);
1542	SPAGAIN;
1543		1596
1544	FREETMPS;	1597	FREETMPS;
1545	LEAVE;	1598	LEAVE;
1546	}	1599	}
1547		1600
…		…
1574	ENTER;	1627	ENTER;
1575	SAVETMPS;	1628	SAVETMPS;
1576		1629
1577	PUSHMARK (SP);	1630	PUSHMARK (SP);
1578	PUTBACK;	1631	PUTBACK;
1579	call_sv (get_sv ("Coro::idle", FALSE), G_DISCARD);	1632	call_sv (get_sv ("Coro::idle", FALSE), G_VOID | G_DISCARD);
1580	SPAGAIN;
1581		1633
1582	FREETMPS;	1634	FREETMPS;
1583	LEAVE;	1635	LEAVE;
1584	continue;	1636	continue;
1585	}	1637	}
1586		1638
1587	ta->next = SvSTATE (next_sv);	1639	ta->next = SvSTATE_hv (next_sv);
1588		1640
1589	/* cannot transfer to destroyed coros, skip and look for next */	1641	/* cannot transfer to destroyed coros, skip and look for next */
1590	if (expect_false (ta->next->flags & CF_DESTROYED))	1642	if (expect_false (ta->next->flags & CF_DESTROYED))
1591	{	1643	{
1592	SvREFCNT_dec (next_sv);	1644	SvREFCNT_dec (next_sv);
…		…
1598	break;	1650	break;
1599	}	1651	}
1600		1652
1601	/* free this only after the transfer */	1653	/* free this only after the transfer */
1602	prev_sv = SvRV (coro_current);	1654	prev_sv = SvRV (coro_current);
1603	ta->prev = SvSTATE (prev_sv);	1655	ta->prev = SvSTATE_hv (prev_sv);
1604	TRANSFER_CHECK (*ta);	1656	TRANSFER_CHECK (*ta);
1605	assert (("FATAL: next coroutine isn't marked as ready in Coro (please report)", ta->next->flags & CF_READY));	1657	assert (("FATAL: next coroutine isn't marked as ready in Coro (please report)", ta->next->flags & CF_READY));
1606	ta->next->flags &= ~CF_READY;	1658	ta->next->flags &= ~CF_READY;
1607	SvRV_set (coro_current, next_sv);	1659	SvRV_set (coro_current, next_sv);
1608		1660
…		…
1692	if (coro->flags & CF_RUNNING)	1744	if (coro->flags & CF_RUNNING)
1693	PL_runops = RUNOPS_DEFAULT;	1745	PL_runops = RUNOPS_DEFAULT;
1694	else	1746	else
1695	coro->slot->runops = RUNOPS_DEFAULT;	1747	coro->slot->runops = RUNOPS_DEFAULT;
1696	}	1748	}
		1749	}
		1750
		1751	/*****************************************************************************/
		1752	/* rouse callback */
		1753
		1754	#define CORO_MAGIC_type_rouse PERL_MAGIC_ext
		1755
		1756	static void
		1757	coro_rouse_callback (pTHX_ CV *cv)
		1758	{
		1759	dXSARGS;
		1760	SV *data = (SV *)GENSUB_ARG;
		1761
		1762	if (SvTYPE (SvRV (data)) != SVt_PVAV)
		1763	{
		1764	/* first call, set args */
		1765	int i;
		1766	AV *av = newAV ();
		1767	SV *coro = SvRV (data);
		1768
		1769	SvRV_set (data, (SV *)av);
		1770	api_ready (aTHX_ coro);
		1771	SvREFCNT_dec (coro);
		1772
		1773	/* better take a full copy of the arguments */
		1774	while (items--)
		1775	av_store (av, items, newSVsv (ST (items)));
		1776	}
		1777
		1778	XSRETURN_EMPTY;
		1779	}
		1780
		1781	static int
		1782	slf_check_rouse_wait (pTHX_ struct CoroSLF *frame)
		1783	{
		1784	SV *data = (SV *)frame->data;
		1785
		1786	if (CORO_THROW)
		1787	return 0;
		1788
		1789	if (SvTYPE (SvRV (data)) != SVt_PVAV)
		1790	return 1;
		1791
		1792	/* now push all results on the stack */
		1793	{
		1794	dSP;
		1795	AV *av = (AV *)SvRV (data);
		1796	int i;
		1797
		1798	EXTEND (SP, AvFILLp (av) + 1);
		1799	for (i = 0; i <= AvFILLp (av); ++i)
		1800	PUSHs (sv_2mortal (AvARRAY (av)[i]));
		1801
		1802	/* we have stolen the elements, so ste length to zero and free */
		1803	AvFILLp (av) = -1;
		1804	av_undef (av);
		1805
		1806	PUTBACK;
		1807	}
		1808
		1809	return 0;
		1810	}
		1811
		1812	static void
		1813	slf_init_rouse_wait (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		1814	{
		1815	SV *cb;
		1816
		1817	if (items)
		1818	cb = arg [0];
		1819	else
		1820	{
		1821	struct coro *coro = SvSTATE_current;
		1822
		1823	if (!coro->rouse_cb)
		1824	croak ("Coro::rouse_wait called without rouse callback, and no default rouse callback found either,");
		1825
		1826	cb = sv_2mortal (coro->rouse_cb);
		1827	coro->rouse_cb = 0;
		1828	}
		1829
		1830	if (!SvROK (cb)
		1831	|| SvTYPE (SvRV (cb)) != SVt_PVCV
		1832	|| CvXSUB ((CV *)SvRV (cb)) != coro_rouse_callback)
		1833	croak ("Coro::rouse_wait called with illegal callback argument,");
		1834
		1835	{
		1836	CV *cv = (CV *)SvRV (cb); /* for GENSUB_ARG */
		1837	SV *data = (SV *)GENSUB_ARG;
		1838
		1839	frame->data = (void *)data;
		1840	frame->prepare = SvTYPE (SvRV (data)) == SVt_PVAV ? prepare_nop : prepare_schedule;
		1841	frame->check = slf_check_rouse_wait;
		1842	}
		1843	}
		1844
		1845	static SV *
		1846	coro_new_rouse_cb (pTHX)
		1847	{
		1848	HV *hv = (HV *)SvRV (coro_current);
		1849	struct coro *coro = SvSTATE_hv (hv);
		1850	SV *data = newRV_inc ((SV *)hv);
		1851	SV *cb = gensub (aTHX_ coro_rouse_callback, (void *)data);
		1852
		1853	sv_magicext (SvRV (cb), data, CORO_MAGIC_type_rouse, 0, 0, 0);
		1854	SvREFCNT_dec (data); /* magicext increases the refcount */
		1855
		1856	SvREFCNT_dec (coro->rouse_cb);
		1857	coro->rouse_cb = SvREFCNT_inc_NN (cb);
		1858
		1859	return cb;
		1860	}
		1861
		1862	/*****************************************************************************/
		1863	/* schedule-like-function opcode (SLF) */
		1864
		1865	static UNOP slf_restore; /* restore stack as entersub did, for first-re-run */
		1866	static const CV *slf_cv;
		1867	static SV **slf_argv;
		1868	static int slf_argc, slf_arga; /* count, allocated */
		1869	static I32 slf_ax; /* top of stack, for restore */
		1870
		1871	/* this restores the stack in the case we patched the entersub, to */
		1872	/* recreate the stack frame as perl will on following calls */
		1873	/* since entersub cleared the stack */
		1874	static OP *
		1875	pp_restore (pTHX)
		1876	{
		1877	int i;
		1878	SV **SP = PL_stack_base + slf_ax;
		1879
		1880	PUSHMARK (SP);
		1881
		1882	EXTEND (SP, slf_argc + 1);
		1883
		1884	for (i = 0; i < slf_argc; ++i)
		1885	PUSHs (sv_2mortal (slf_argv [i]));
		1886
		1887	PUSHs ((SV *)CvGV (slf_cv));
		1888
		1889	RETURNOP (slf_restore.op_first);
		1890	}
		1891
		1892	static void
		1893	slf_prepare_transfer (pTHX_ struct coro_transfer_args *ta)
		1894	{
		1895	SV **arg = (SV **)slf_frame.data;
		1896
		1897	prepare_transfer (aTHX_ ta, arg [0], arg [1]);
		1898	}
		1899
		1900	static void
		1901	slf_init_transfer (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		1902	{
		1903	if (items != 2)
		1904	croak ("Coro::State::transfer (prev, next) expects two arguments, not %d,", items);
		1905
		1906	frame->prepare = slf_prepare_transfer;
		1907	frame->check = slf_check_nop;
		1908	frame->data = (void *)arg; /* let's hope it will stay valid */
		1909	}
		1910
		1911	static void
		1912	slf_init_schedule (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		1913	{
		1914	frame->prepare = prepare_schedule;
		1915	frame->check = slf_check_nop;
		1916	}
		1917
		1918	static void
		1919	slf_init_cede (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		1920	{
		1921	frame->prepare = prepare_cede;
		1922	frame->check = slf_check_nop;
		1923	}
		1924
		1925	static void
		1926	slf_init_cede_notself (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		1927	{
		1928	frame->prepare = prepare_cede_notself;
		1929	frame->check = slf_check_nop;
		1930	}
		1931
		1932	/*
		1933	* these not obviously related functions are all rolled into one
		1934	* function to increase chances that they all will call transfer with the same
		1935	* stack offset
		1936	* SLF stands for "schedule-like-function".
		1937	*/
		1938	static OP *
		1939	pp_slf (pTHX)
		1940	{
		1941	I32 checkmark; /* mark SP to see how many elements check has pushed */
		1942
		1943	/* set up the slf frame, unless it has already been set-up */
		1944	/* the latter happens when a new coro has been started */
		1945	/* or when a new cctx was attached to an existing coroutine */
		1946	if (expect_true (!slf_frame.prepare))
		1947	{
		1948	/* first iteration */
		1949	dSP;
		1950	SV **arg = PL_stack_base + TOPMARK + 1;
		1951	int items = SP - arg; /* args without function object */
		1952	SV *gv = *sp;
		1953
		1954	/* do a quick consistency check on the "function" object, and if it isn't */
		1955	/* for us, divert to the real entersub */
		1956	if (SvTYPE (gv) != SVt_PVGV
		1957	|| !GvCV (gv)
		1958	|| !(CvFLAGS (GvCV (gv)) & CVf_SLF))
		1959	return PL_ppaddr[OP_ENTERSUB](aTHX);
		1960
		1961	if (!(PL_op->op_flags & OPf_STACKED))
		1962	{
		1963	/* ampersand-form of call, use @_ instead of stack */
		1964	AV *av = GvAV (PL_defgv);
		1965	arg = AvARRAY (av);
		1966	items = AvFILLp (av) + 1;
		1967	}
		1968
		1969	/* now call the init function, which needs to set up slf_frame */
		1970	((coro_slf_cb)CvXSUBANY (GvCV (gv)).any_ptr)
		1971	(aTHX_ &slf_frame, GvCV (gv), arg, items);
		1972
		1973	/* pop args */
		1974	SP = PL_stack_base + POPMARK;
		1975
		1976	PUTBACK;
		1977	}
		1978
		1979	/* now that we have a slf_frame, interpret it! */
		1980	/* we use a callback system not to make the code needlessly */
		1981	/* complicated, but so we can run multiple perl coros from one cctx */
		1982
		1983	do
		1984	{
		1985	struct coro_transfer_args ta;
		1986
		1987	slf_frame.prepare (aTHX_ &ta);
		1988	TRANSFER (ta, 0);
		1989
		1990	checkmark = PL_stack_sp - PL_stack_base;
		1991	}
		1992	while (slf_frame.check (aTHX_ &slf_frame));
		1993
		1994	slf_frame.prepare = 0; /* invalidate the frame, we are done processing it */
		1995
		1996	/* exception handling */
		1997	if (expect_false (CORO_THROW))
		1998	{
		1999	SV *exception = sv_2mortal (CORO_THROW);
		2000
		2001	CORO_THROW = 0;
		2002	sv_setsv (ERRSV, exception);
		2003	croak (0);
		2004	}
		2005
		2006	/* return value handling - mostly like entersub */
		2007	/* make sure we put something on the stack in scalar context */
		2008	if (GIMME_V == G_SCALAR)
		2009	{
		2010	dSP;
		2011	SV **bot = PL_stack_base + checkmark;
		2012
		2013	if (sp == bot) /* too few, push undef */
		2014	bot [1] = &PL_sv_undef;
		2015	else if (sp != bot + 1) /* too many, take last one */
		2016	bot [1] = *sp;
		2017
		2018	SP = bot + 1;
		2019
		2020	PUTBACK;
		2021	}
		2022
		2023	return NORMAL;
		2024	}
		2025
		2026	static void
		2027	api_execute_slf (pTHX_ CV *cv, coro_slf_cb init_cb, I32 ax)
		2028	{
		2029	int i;
		2030	SV **arg = PL_stack_base + ax;
		2031	int items = PL_stack_sp - arg + 1;
		2032
		2033	assert (("FATAL: SLF call with illegal CV value", !CvANON (cv)));
		2034
		2035	if (PL_op->op_ppaddr != PL_ppaddr [OP_ENTERSUB]
		2036	&& PL_op->op_ppaddr != pp_slf)
		2037	croak ("FATAL: Coro SLF calls can only be made normally, not via goto or any other means, caught");
		2038
		2039	CvFLAGS (cv) |= CVf_SLF;
		2040	CvXSUBANY (cv).any_ptr = (void *)init_cb;
		2041	slf_cv = cv;
		2042
		2043	/* we patch the op, and then re-run the whole call */
		2044	/* we have to put the same argument on the stack for this to work */
		2045	/* and this will be done by pp_restore */
		2046	slf_restore.op_next = (OP *)&slf_restore;
		2047	slf_restore.op_type = OP_CUSTOM;
		2048	slf_restore.op_ppaddr = pp_restore;
		2049	slf_restore.op_first = PL_op;
		2050
		2051	slf_ax = ax - 1; /* undo the ax++ inside dAXMARK */
		2052
		2053	if (PL_op->op_flags & OPf_STACKED)
		2054	{
		2055	if (items > slf_arga)
		2056	{
		2057	slf_arga = items;
		2058	free (slf_argv);
		2059	slf_argv = malloc (slf_arga * sizeof (SV *));
		2060	}
		2061
		2062	slf_argc = items;
		2063
		2064	for (i = 0; i < items; ++i)
		2065	slf_argv [i] = SvREFCNT_inc (arg [i]);
		2066	}
		2067	else
		2068	slf_argc = 0;
		2069
		2070	PL_op->op_ppaddr = pp_slf;
		2071	PL_op->op_type = OP_CUSTOM; /* maybe we should leave it at entersub? */
		2072
		2073	PL_op = (OP *)&slf_restore;
1697	}	2074	}
1698		2075
1699	/*****************************************************************************/	2076	/*****************************************************************************/
1700	/* PerlIO::cede */	2077	/* PerlIO::cede */
1701		2078
…		…
1770	PerlIOBuf_get_cnt,	2147	PerlIOBuf_get_cnt,
1771	PerlIOBuf_set_ptrcnt,	2148	PerlIOBuf_set_ptrcnt,
1772	};	2149	};
1773		2150
1774	/*****************************************************************************/	2151	/*****************************************************************************/
		2152	/* Coro::Semaphore & Coro::Signal */
1775		2153
1776	static const CV *slf_cv; /* for quick consistency check */
1777
1778	static UNOP slf_restore; /* restore stack as entersub did, for first-re-run */
1779	static SV *slf_arg0;
1780	static SV *slf_arg1;
1781	static SV *slf_arg2;
1782
1783	/* this restores the stack in the case we patched the entersub, to */
1784	/* recreate the stack frame as perl will on following calls */
1785	/* since entersub cleared the stack */
1786	static OP *	2154	static SV *
1787	pp_restore (pTHX)	2155	coro_waitarray_new (pTHX_ int count)
1788	{	2156	{
1789	dSP;	2157	/* a semaphore contains a counter IV in $sem->[0] and any waiters after that */
		2158	AV *av = newAV ();
		2159	SV **ary;
1790		2160
1791	PUSHMARK (SP);	2161	/* unfortunately, building manually saves memory */
		2162	Newx (ary, 2, SV *);
		2163	AvALLOC (av) = ary;
		2164	/*AvARRAY (av) = ary;*/
		2165	SvPVX ((SV *)av) = (char *)ary; /* 5.8.8 needs this syntax instead of AvARRAY = ary */
		2166	AvMAX (av) = 1;
		2167	AvFILLp (av) = 0;
		2168	ary [0] = newSViv (count);
1792		2169
1793	EXTEND (SP, 3);	2170	return newRV_noinc ((SV *)av);
1794	if (slf_arg0) PUSHs (sv_2mortal (slf_arg0));
1795	if (slf_arg1) PUSHs (sv_2mortal (slf_arg1));
1796	if (slf_arg2) PUSHs (sv_2mortal (slf_arg2));
1797	PUSHs ((SV *)CvGV (slf_cv));
1798
1799	RETURNOP (slf_restore.op_first);
1800	}	2171	}
1801		2172
1802	static void	2173	/* semaphore */
1803	slf_prepare_set_stacklevel (pTHX_ struct coro_transfer_args *ta)
1804	{
1805	prepare_set_stacklevel (ta, (struct coro_cctx *)slf_frame.data);
1806	}
1807		2174
1808	static void	2175	static void
1809	slf_init_set_stacklevel (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
1810	{
1811	assert (("FATAL: set_stacklevel needs the coro cctx as sole argument", items == 1));
1812
1813	frame->prepare = slf_prepare_set_stacklevel;
1814	frame->check = slf_check_nop;
1815	frame->data = (void *)SvIV (arg [0]);
1816	}
1817
1818	static void
1819	slf_prepare_transfer (pTHX_ struct coro_transfer_args *ta)
1820	{
1821	SV **arg = (SV **)slf_frame.data;
1822
1823	prepare_transfer (aTHX_ ta, arg [0], arg [1]);
1824	}
1825
1826	static void
1827	slf_init_transfer (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
1828	{
1829	if (items != 2)
1830	croak ("Coro::State::transfer (prev, next) expects two arguments, not %d,", items);
1831
1832	frame->prepare = slf_prepare_transfer;
1833	frame->check = slf_check_nop;
1834	frame->data = (void *)arg; /* let's hope it will stay valid */
1835	}
1836
1837	static void
1838	slf_init_schedule (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
1839	{
1840	frame->prepare = prepare_schedule;
1841	frame->check = slf_check_nop;
1842	}
1843
1844	static void
1845	slf_init_cede (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
1846	{
1847	frame->prepare = prepare_cede;
1848	frame->check = slf_check_nop;
1849	}
1850
1851	static void
1852	slf_init_cede_notself (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
1853	{
1854	frame->prepare = prepare_cede_notself;
1855	frame->check = slf_check_nop;
1856	}
1857
1858	/* we hijack an hopefully unused CV flag for our purposes */
1859	#define CVf_SLF 0x4000
1860
1861	/*
1862	* these not obviously related functions are all rolled into one
1863	* function to increase chances that they all will call transfer with the same
1864	* stack offset
1865	* SLF stands for "schedule-like-function".
1866	*/
1867	static OP *
1868	pp_slf (pTHX)
1869	{
1870	I32 checkmark; /* mark SP to see how many elements check has pushed */
1871
1872	/* set up the slf frame, unless it has already been set-up */
1873	/* the latter happens when a new coro has been started */
1874	/* or when a new cctx was attached to an existing coroutine */
1875	if (expect_true (!slf_frame.prepare))
1876	{
1877	/* first iteration */
1878	dSP;
1879	SV **arg = PL_stack_base + TOPMARK + 1;
1880	int items = SP - arg; /* args without function object */
1881	SV *gv = *sp;
1882
1883	/* do a quick consistency check on the "function" object, and if it isn't */
1884	/* for us, divert to the real entersub */
1885	if (SvTYPE (gv) != SVt_PVGV || !(CvFLAGS (GvCV (gv)) & CVf_SLF))
1886	return PL_ppaddr[OP_ENTERSUB](aTHX);
1887
1888	/* pop args */
1889	SP = PL_stack_base + POPMARK;
1890
1891	if (!(PL_op->op_flags & OPf_STACKED))
1892	{
1893	/* ampersand-form of call, use @_ instead of stack */
1894	AV *av = GvAV (PL_defgv);
1895	arg = AvARRAY (av);
1896	items = AvFILLp (av) + 1;
1897	}
1898
1899	PUTBACK;
1900
1901	/* now call the init function, which needs to set up slf_frame */
1902	((coro_slf_cb)CvXSUBANY (GvCV (gv)).any_ptr)
1903	(aTHX_ &slf_frame, GvCV (gv), arg, items);
1904	}
1905
1906	/* now that we have a slf_frame, interpret it! */
1907	/* we use a callback system not to make the code needlessly */
1908	/* complicated, but so we can run multiple perl coros from one cctx */
1909
1910	do
1911	{
1912	struct coro_transfer_args ta;
1913
1914	slf_frame.prepare (aTHX_ &ta);
1915	TRANSFER (ta, 0);
1916
1917	checkmark = PL_stack_sp - PL_stack_base;
1918	}
1919	while (slf_frame.check (aTHX_ &slf_frame));
1920
1921	{
1922	dSP;
1923	SV **bot = PL_stack_base + checkmark;
1924	int gimme = GIMME_V;
1925
1926	slf_frame.prepare = 0; /* invalidate the frame, so it gets initialised again next time */
1927
1928	/* make sure we put something on the stack in scalar context */
1929	if (gimme == G_SCALAR)
1930	{
1931	if (sp == bot)
1932	XPUSHs (&PL_sv_undef);
1933
1934	SP = bot + 1;
1935	}
1936
1937	PUTBACK;
1938	}
1939
1940	return NORMAL;
1941	}
1942
1943	static void
1944	api_execute_slf (pTHX_ CV *cv, coro_slf_cb init_cb, SV **arg, int items)
1945	{
1946	assert (("FATAL: SLF call with illegal CV value", !CvANON (cv)));
1947
1948	if (PL_op->op_ppaddr != PL_ppaddr [OP_ENTERSUB]
1949	&& PL_op->op_ppaddr != pp_slf)
1950	croak ("FATAL: Coro SLF calls can only be made normally, not via goto or any other means, caught");
1951
1952	if (items > 3)
1953	croak ("Coro only supports up to three arguments to SLF functions currently (not %d), caught", items);
1954
1955	CvFLAGS (cv) |= CVf_SLF;
1956	CvXSUBANY (cv).any_ptr = (void *)init_cb;
1957	slf_cv = cv;
1958
1959	/* we patch the op, and then re-run the whole call */
1960	/* we have to put the same argument on the stack for this to work */
1961	/* and this will be done by pp_restore */
1962	slf_restore.op_next = (OP *)&slf_restore;
1963	slf_restore.op_type = OP_CUSTOM;
1964	slf_restore.op_ppaddr = pp_restore;
1965	slf_restore.op_first = PL_op;
1966
1967	slf_arg0 = items > 0 ? SvREFCNT_inc (arg [0]) : 0;
1968	slf_arg1 = items > 1 ? SvREFCNT_inc (arg [1]) : 0;
1969	slf_arg2 = items > 2 ? SvREFCNT_inc (arg [2]) : 0;
1970
1971	PL_op->op_ppaddr = pp_slf;
1972
1973	PL_op = (OP *)&slf_restore;
1974	}
1975
1976	/*****************************************************************************/
1977
1978	static void
1979	coro_semaphore_adjust (AV *av, int adjust)	2176	coro_semaphore_adjust (pTHX_ AV *av, IV adjust)
1980	{	2177	{
1981	SV *count_sv = AvARRAY (av)[0];	2178	SV *count_sv = AvARRAY (av)[0];
1982	IV count = SvIVX (count_sv);	2179	IV count = SvIVX (count_sv);
1983		2180
1984	count += adjust;	2181	count += adjust;
1985	SvIVX (count_sv) = count;	2182	SvIVX (count_sv) = count;
1986		2183
1987	/* now wake up as many waiters as possible */	2184	/* now wake up as many waiters as are expected to lock */
1988	while (count > 0 && AvFILLp (av) >= count)	2185	while (count > 0 && AvFILLp (av) > 0)
1989	{	2186	{
1990	SV *cb;	2187	SV *cb;
1991		2188
1992	/* swap first two elements so we can shift a waiter */	2189	/* swap first two elements so we can shift a waiter */
1993	AvARRAY (av)[0] = AvARRAY (av)[1];	2190	AvARRAY (av)[0] = AvARRAY (av)[1];
1994	AvARRAY (av)[1] = count_sv;	2191	AvARRAY (av)[1] = count_sv;
1995	cb = av_shift (av);	2192	cb = av_shift (av);
1996		2193
1997	if (SvOBJECT (cb))	2194	if (SvOBJECT (cb))
		2195	{
1998	api_ready (aTHX_ cb);	2196	api_ready (aTHX_ cb);
1999	else	2197	--count;
2000	croak ("callbacks not yet supported");	2198	}
		2199	else if (SvTYPE (cb) == SVt_PVCV)
		2200	{
		2201	dSP;
		2202	PUSHMARK (SP);
		2203	XPUSHs (sv_2mortal (newRV_inc ((SV *)av)));
		2204	PUTBACK;
		2205	call_sv (cb, G_VOID | G_DISCARD | G_EVAL | G_KEEPERR);
		2206	}
2001		2207
2002	SvREFCNT_dec (cb);	2208	SvREFCNT_dec (cb);
2003
2004	--count;
2005	}	2209	}
2006	}	2210	}
2007		2211
2008	static void	2212	static void
2009	coro_semaphore_on_destroy (pTHX_ struct coro *coro)	2213	coro_semaphore_on_destroy (pTHX_ struct coro *coro)
2010	{	2214	{
2011	/* call $sem->adjust (0) to possibly wake up some waiters */	2215	/* call $sem->adjust (0) to possibly wake up some other waiters */
2012	coro_semaphore_adjust ((AV *)coro->slf_frame.data, 0);	2216	coro_semaphore_adjust (aTHX_ (AV *)coro->slf_frame.data, 0);
2013	}	2217	}
2014		2218
2015	static int	2219	static int
2016	slf_check_semaphore_down (pTHX_ struct CoroSLF *frame)	2220	slf_check_semaphore_down_or_wait (pTHX_ struct CoroSLF *frame, int acquire)
2017	{	2221	{
2018	AV *av = (AV *)frame->data;	2222	AV *av = (AV *)frame->data;
2019	SV *count_sv = AvARRAY (av)[0];	2223	SV *count_sv = AvARRAY (av)[0];
2020		2224
		2225	/* if we are about to throw, don't actually acquire the lock, just throw */
		2226	if (CORO_THROW)
		2227	return 0;
2021	if (SvIVX (count_sv) > 0)	2228	else if (SvIVX (count_sv) > 0)
2022	{	2229	{
2023	SvSTATE (coro_current)->on_destroy = 0;	2230	SvSTATE_current->on_destroy = 0;
		2231
		2232	if (acquire)
2024	SvIVX (count_sv) = SvIVX (count_sv) - 1;	2233	SvIVX (count_sv) = SvIVX (count_sv) - 1;
		2234	else
		2235	coro_semaphore_adjust (aTHX_ av, 0);
		2236
2025	return 0;	2237	return 0;
2026	}	2238	}
2027	else	2239	else
2028	{	2240	{
2029	int i;	2241	int i;
…		…
2038	av_push (av, SvREFCNT_inc (SvRV (coro_current)));	2250	av_push (av, SvREFCNT_inc (SvRV (coro_current)));
2039	return 1;	2251	return 1;
2040	}	2252	}
2041	}	2253	}
2042		2254
2043	static void	2255	static int
		2256	slf_check_semaphore_down (pTHX_ struct CoroSLF *frame)
		2257	{
		2258	return slf_check_semaphore_down_or_wait (aTHX_ frame, 1);
		2259	}
		2260
		2261	static int
		2262	slf_check_semaphore_wait (pTHX_ struct CoroSLF *frame)
		2263	{
		2264	return slf_check_semaphore_down_or_wait (aTHX_ frame, 0);
		2265	}
		2266
		2267	static void
2044	slf_init_semaphore_down (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)	2268	slf_init_semaphore_down_or_wait (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
2045	{	2269	{
2046	AV *av = (AV *)SvRV (arg [0]);	2270	AV *av = (AV *)SvRV (arg [0]);
2047		2271
2048	if (SvIVX (AvARRAY (av)[0]) > 0)	2272	if (SvIVX (AvARRAY (av)[0]) > 0)
2049	{	2273	{
…		…
2057	frame->data = (void *)sv_2mortal (SvREFCNT_inc ((SV *)av));	2281	frame->data = (void *)sv_2mortal (SvREFCNT_inc ((SV *)av));
2058	frame->prepare = prepare_schedule;	2282	frame->prepare = prepare_schedule;
2059		2283
2060	/* to avoid race conditions when a woken-up coro gets terminated */	2284	/* to avoid race conditions when a woken-up coro gets terminated */
2061	/* we arrange for a temporary on_destroy that calls adjust (0) */	2285	/* we arrange for a temporary on_destroy that calls adjust (0) */
2062	SvSTATE (coro_current)->on_destroy = coro_semaphore_on_destroy;	2286	SvSTATE_current->on_destroy = coro_semaphore_on_destroy;
2063	}	2287	}
		2288	}
2064		2289
		2290	static void
		2291	slf_init_semaphore_down (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		2292	{
		2293	slf_init_semaphore_down_or_wait (aTHX_ frame, cv, arg, items);
2065	frame->check = slf_check_semaphore_down;	2294	frame->check = slf_check_semaphore_down;
		2295	}
2066		2296
		2297	static void
		2298	slf_init_semaphore_wait (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		2299	{
		2300	if (items >= 2)
		2301	{
		2302	/* callback form */
		2303	AV *av = (AV *)SvRV (arg [0]);
		2304	CV *cb_cv = coro_sv_2cv (arg [1]);
		2305
		2306	av_push (av, (SV *)SvREFCNT_inc_NN (cb_cv));
		2307
		2308	if (SvIVX (AvARRAY (av)[0]) > 0)
		2309	coro_semaphore_adjust (aTHX_ av, 0);
		2310
		2311	frame->prepare = prepare_nop;
		2312	frame->check = slf_check_nop;
		2313	}
		2314	else
		2315	{
		2316	slf_init_semaphore_down_or_wait (aTHX_ frame, cv, arg, items);
		2317	frame->check = slf_check_semaphore_wait;
		2318	}
		2319	}
		2320
		2321	/* signal */
		2322
		2323	static void
		2324	coro_signal_wake (pTHX_ AV *av, int count)
		2325	{
		2326	SvIVX (AvARRAY (av)[0]) = 0;
		2327
		2328	/* now signal count waiters */
		2329	while (count > 0 && AvFILLp (av) > 0)
		2330	{
		2331	SV *cb;
		2332
		2333	/* swap first two elements so we can shift a waiter */
		2334	cb = AvARRAY (av)[0];
		2335	AvARRAY (av)[0] = AvARRAY (av)[1];
		2336	AvARRAY (av)[1] = cb;
		2337
		2338	cb = av_shift (av);
		2339
		2340	api_ready (aTHX_ cb);
		2341	sv_setiv (cb, 0); /* signal waiter */
		2342	SvREFCNT_dec (cb);
		2343
		2344	--count;
		2345	}
		2346	}
		2347
		2348	static int
		2349	slf_check_signal_wait (pTHX_ struct CoroSLF *frame)
		2350	{
		2351	/* if we are about to throw, also stop waiting */
		2352	return SvROK ((SV *)frame->data) && !CORO_THROW;
		2353	}
		2354
		2355	static void
		2356	slf_init_signal_wait (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		2357	{
		2358	AV *av = (AV *)SvRV (arg [0]);
		2359
		2360	if (SvIVX (AvARRAY (av)[0]))
		2361	{
		2362	SvIVX (AvARRAY (av)[0]) = 0;
		2363	frame->prepare = prepare_nop;
		2364	frame->check = slf_check_nop;
		2365	}
		2366	else
		2367	{
		2368	SV *waiter = newRV_inc (SvRV (coro_current)); /* owned by signal av */
		2369
		2370	av_push (av, waiter);
		2371
		2372	frame->data = (void *)sv_2mortal (SvREFCNT_inc_NN (waiter)); /* owned by process */
		2373	frame->prepare = prepare_schedule;
		2374	frame->check = slf_check_signal_wait;
		2375	}
2067	}	2376	}
2068		2377
2069	/*****************************************************************************/	2378	/*****************************************************************************/
		2379	/* Coro::AIO */
2070		2380
2071	#define GENSUB_ARG CvXSUBANY (cv).any_ptr	2381	#define CORO_MAGIC_type_aio PERL_MAGIC_ext
2072		2382
2073	/* create a closure from XS, returns a code reference */	2383	/* helper storage struct */
2074	/* the arg can be accessed via GENSUB_ARG from the callback */	2384	struct io_state
2075	/* the callback must use dXSARGS/XSRETURN */
2076	static SV *
2077	gensub (pTHX_ void (*xsub)(pTHX_ CV *), void *arg)
2078	{	2385	{
2079	CV *cv = (CV *)NEWSV (0, 0);	2386	int errorno;
		2387	I32 laststype; /* U16 in 5.10.0 */
		2388	int laststatval;
		2389	Stat_t statcache;
		2390	};
2080		2391
		2392	static void
		2393	coro_aio_callback (pTHX_ CV *cv)
		2394	{
		2395	dXSARGS;
		2396	AV *state = (AV *)GENSUB_ARG;
		2397	SV *coro = av_pop (state);
		2398	SV *data_sv = newSV (sizeof (struct io_state));
		2399
		2400	av_extend (state, items);
		2401
2081	sv_upgrade ((SV *)cv, SVt_PVCV);	2402	sv_upgrade (data_sv, SVt_PV);
		2403	SvCUR_set (data_sv, sizeof (struct io_state));
		2404	SvPOK_only (data_sv);
2082		2405
2083	CvANON_on (cv);	2406	{
2084	CvISXSUB_on (cv);	2407	struct io_state *data = (struct io_state *)SvPVX (data_sv);
2085	CvXSUB (cv) = xsub;
2086	GENSUB_ARG = arg;
2087		2408
2088	return newRV_noinc ((SV *)cv);	2409	data->errorno = errno;
		2410	data->laststype = PL_laststype;
		2411	data->laststatval = PL_laststatval;
		2412	data->statcache = PL_statcache;
		2413	}
		2414
		2415	/* now build the result vector out of all the parameters and the data_sv */
		2416	{
		2417	int i;
		2418
		2419	for (i = 0; i < items; ++i)
		2420	av_push (state, SvREFCNT_inc_NN (ST (i)));
		2421	}
		2422
		2423	av_push (state, data_sv);
		2424
		2425	api_ready (aTHX_ coro);
		2426	SvREFCNT_dec (coro);
		2427	SvREFCNT_dec ((AV *)state);
		2428	}
		2429
		2430	static int
		2431	slf_check_aio_req (pTHX_ struct CoroSLF *frame)
		2432	{
		2433	AV *state = (AV *)frame->data;
		2434
		2435	/* if we are about to throw, return early */
		2436	/* this does not cancel the aio request, but at least */
		2437	/* it quickly returns */
		2438	if (CORO_THROW)
		2439	return 0;
		2440
		2441	/* one element that is an RV? repeat! */
		2442	if (AvFILLp (state) == 0 && SvROK (AvARRAY (state)[0]))
		2443	return 1;
		2444
		2445	/* restore status */
		2446	{
		2447	SV *data_sv = av_pop (state);
		2448	struct io_state *data = (struct io_state *)SvPVX (data_sv);
		2449
		2450	errno = data->errorno;
		2451	PL_laststype = data->laststype;
		2452	PL_laststatval = data->laststatval;
		2453	PL_statcache = data->statcache;
		2454
		2455	SvREFCNT_dec (data_sv);
		2456	}
		2457
		2458	/* push result values */
		2459	{
		2460	dSP;
		2461	int i;
		2462
		2463	EXTEND (SP, AvFILLp (state) + 1);
		2464	for (i = 0; i <= AvFILLp (state); ++i)
		2465	PUSHs (sv_2mortal (SvREFCNT_inc_NN (AvARRAY (state)[i])));
		2466
		2467	PUTBACK;
		2468	}
		2469
		2470	return 0;
		2471	}
		2472
		2473	static void
		2474	slf_init_aio_req (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		2475	{
		2476	AV *state = (AV *)sv_2mortal ((SV *)newAV ());
		2477	SV *coro_hv = SvRV (coro_current);
		2478	struct coro *coro = SvSTATE_hv (coro_hv);
		2479
		2480	/* put our coroutine id on the state arg */
		2481	av_push (state, SvREFCNT_inc_NN (coro_hv));
		2482
		2483	/* first see whether we have a non-zero priority and set it as AIO prio */
		2484	if (coro->prio)
		2485	{
		2486	dSP;
		2487
		2488	static SV *prio_cv;
		2489	static SV *prio_sv;
		2490
		2491	if (expect_false (!prio_cv))
		2492	{
		2493	prio_cv = (SV *)get_cv ("IO::AIO::aioreq_pri", 0);
		2494	prio_sv = newSViv (0);
		2495	}
		2496
		2497	PUSHMARK (SP);
		2498	sv_setiv (prio_sv, coro->prio);
		2499	XPUSHs (prio_sv);
		2500
		2501	PUTBACK;
		2502	call_sv (prio_cv, G_VOID | G_DISCARD);
		2503	}
		2504
		2505	/* now call the original request */
		2506	{
		2507	dSP;
		2508	CV *req = (CV *)CORO_MAGIC_NN ((SV *)cv, CORO_MAGIC_type_aio)->mg_obj;
		2509	int i;
		2510
		2511	PUSHMARK (SP);
		2512
		2513	/* first push all args to the stack */
		2514	EXTEND (SP, items + 1);
		2515
		2516	for (i = 0; i < items; ++i)
		2517	PUSHs (arg [i]);
		2518
		2519	/* now push the callback closure */
		2520	PUSHs (sv_2mortal (gensub (aTHX_ coro_aio_callback, (void *)SvREFCNT_inc_NN ((SV *)state))));
		2521
		2522	/* now call the AIO function - we assume our request is uncancelable */
		2523	PUTBACK;
		2524	call_sv ((SV *)req, G_VOID | G_DISCARD);
		2525	}
		2526
		2527	/* now that the requets is going, we loop toll we have a result */
		2528	frame->data = (void *)state;
		2529	frame->prepare = prepare_schedule;
		2530	frame->check = slf_check_aio_req;
		2531	}
		2532
		2533	static void
		2534	coro_aio_req_xs (pTHX_ CV *cv)
		2535	{
		2536	dXSARGS;
		2537
		2538	CORO_EXECUTE_SLF_XS (slf_init_aio_req);
		2539
		2540	XSRETURN_EMPTY;
2089	}	2541	}
2090		2542
2091	/*****************************************************************************/	2543	/*****************************************************************************/
2092		2544
2093	MODULE = Coro::State PACKAGE = Coro::State PREFIX = api_	2545	MODULE = Coro::State PACKAGE = Coro::State PREFIX = api_
…		…
2095	PROTOTYPES: DISABLE	2547	PROTOTYPES: DISABLE
2096		2548
2097	BOOT:	2549	BOOT:
2098	{	2550	{
2099	#ifdef USE_ITHREADS	2551	#ifdef USE_ITHREADS
2100	MUTEX_INIT (&coro_lock);
2101	# if CORO_PTHREAD	2552	# if CORO_PTHREAD
2102	coro_thx = PERL_GET_CONTEXT;	2553	coro_thx = PERL_GET_CONTEXT;
2103	# endif	2554	# endif
2104	#endif	2555	#endif
2105	BOOT_PAGESIZE;	2556	BOOT_PAGESIZE;
…		…
2164	assert (("PRIO_NORMAL must be 0", !PRIO_NORMAL));	2615	assert (("PRIO_NORMAL must be 0", !PRIO_NORMAL));
2165	}	2616	}
2166		2617
2167	SV *	2618	SV *
2168	new (char *klass, ...)	2619	new (char *klass, ...)
		2620	ALIAS:
		2621	Coro::new = 1
2169	CODE:	2622	CODE:
2170	{	2623	{
2171	struct coro *coro;	2624	struct coro *coro;
2172	MAGIC *mg;	2625	MAGIC *mg;
2173	HV *hv;	2626	HV *hv;
		2627	CV *cb;
2174	int i;	2628	int i;
		2629
		2630	if (items > 1)
		2631	{
		2632	cb = coro_sv_2cv (ST (1));
		2633
		2634	if (!ix)
		2635	{
		2636	if (CvISXSUB (cb))
		2637	croak ("Coro::State doesn't support XS functions as coroutine start, caught");
		2638
		2639	if (!CvROOT (cb))
		2640	croak ("Coro::State doesn't support autoloaded or undefined functions as coroutine start, caught");
		2641	}
		2642	}
2175		2643
2176	Newz (0, coro, 1, struct coro);	2644	Newz (0, coro, 1, struct coro);
2177	coro->args = newAV ();	2645	coro->args = newAV ();
2178	coro->flags = CF_NEW;	2646	coro->flags = CF_NEW;
2179		2647
…		…
2184	coro->hv = hv = newHV ();	2652	coro->hv = hv = newHV ();
2185	mg = sv_magicext ((SV *)hv, 0, CORO_MAGIC_type_state, &coro_state_vtbl, (char *)coro, 0);	2653	mg = sv_magicext ((SV *)hv, 0, CORO_MAGIC_type_state, &coro_state_vtbl, (char *)coro, 0);
2186	mg->mg_flags |= MGf_DUP;	2654	mg->mg_flags |= MGf_DUP;
2187	RETVAL = sv_bless (newRV_noinc ((SV *)hv), gv_stashpv (klass, 1));	2655	RETVAL = sv_bless (newRV_noinc ((SV *)hv), gv_stashpv (klass, 1));
2188		2656
		2657	if (items > 1)
		2658	{
2189	av_extend (coro->args, items - 1);	2659	av_extend (coro->args, items - 1 + ix);
		2660
		2661	if (ix)
		2662	{
		2663	av_push (coro->args, SvREFCNT_inc_NN ((SV *)cb));
		2664	cb = cv_coro_run;
		2665	}
		2666
		2667	coro->startcv = (CV *)SvREFCNT_inc_NN ((SV *)cb);
		2668
2190	for (i = 1; i < items; i++)	2669	for (i = 2; i < items; i++)
2191	av_push (coro->args, newSVsv (ST (i)));	2670	av_push (coro->args, newSVsv (ST (i)));
		2671	}
2192	}	2672	}
2193	OUTPUT:	2673	OUTPUT:
2194	RETVAL	2674	RETVAL
2195
2196	void
2197	_set_stacklevel (...)
2198	CODE:
2199	api_execute_slf (aTHX_ cv, slf_init_set_stacklevel, &ST (0), items);
2200		2675
2201	void	2676	void
2202	transfer (...)	2677	transfer (...)
2203	PROTOTYPE: $$	2678	PROTOTYPE: $$
2204	CODE:	2679	CODE:
2205	api_execute_slf (aTHX_ cv, slf_init_transfer, &ST (0), items);	2680	CORO_EXECUTE_SLF_XS (slf_init_transfer);
2206		2681
2207	bool	2682	bool
2208	_destroy (SV *coro_sv)	2683	_destroy (SV *coro_sv)
2209	CODE:	2684	CODE:
2210	RETVAL = coro_state_destroy (aTHX_ SvSTATE (coro_sv));	2685	RETVAL = coro_state_destroy (aTHX_ SvSTATE (coro_sv));
…		…
2328		2803
2329	void	2804	void
2330	throw (Coro::State self, SV *throw = &PL_sv_undef)	2805	throw (Coro::State self, SV *throw = &PL_sv_undef)
2331	PROTOTYPE: $;$	2806	PROTOTYPE: $;$
2332	CODE:	2807	CODE:
		2808	{
		2809	struct coro *current = SvSTATE_current;
		2810	SV **throwp = self == current ? &CORO_THROW : &self->except;
2333	SvREFCNT_dec (self->throw);	2811	SvREFCNT_dec (*throwp);
2334	self->throw = SvOK (throw) ? newSVsv (throw) : 0;	2812	*throwp = SvOK (throw) ? newSVsv (throw) : 0;
		2813	}
2335		2814
2336	void	2815	void
2337	api_trace (SV *coro, int flags = CC_TRACE | CC_TRACE_SUB)	2816	api_trace (SV *coro, int flags = CC_TRACE | CC_TRACE_SUB)
2338	PROTOTYPE: $;$	2817	PROTOTYPE: $;$
2339	C_ARGS: aTHX_ coro, flags	2818	C_ARGS: aTHX_ coro, flags
…		…
2370		2849
2371	void	2850	void
2372	force_cctx ()	2851	force_cctx ()
2373	PROTOTYPE:	2852	PROTOTYPE:
2374	CODE:	2853	CODE:
2375	struct coro *coro = SvSTATE (coro_current);
2376	coro->cctx->idle_sp = 0;	2854	SvSTATE_current->cctx->idle_sp = 0;
2377		2855
2378	void	2856	void
2379	swap_defsv (Coro::State self)	2857	swap_defsv (Coro::State self)
2380	PROTOTYPE: $	2858	PROTOTYPE: $
2381	ALIAS:	2859	ALIAS:
…		…
2389	SV **dst = ix ? (SV **)&self->slot->defav : (SV **)&self->slot->defsv;	2867	SV **dst = ix ? (SV **)&self->slot->defav : (SV **)&self->slot->defsv;
2390		2868
2391	SV *tmp = *src; *src = *dst; *dst = tmp;	2869	SV *tmp = *src; *src = *dst; *dst = tmp;
2392	}	2870	}
2393		2871
		2872
2394	MODULE = Coro::State PACKAGE = Coro	2873	MODULE = Coro::State PACKAGE = Coro
2395		2874
2396	BOOT:	2875	BOOT:
2397	{	2876	{
2398	int i;	2877	int i;
2399		2878
2400	av_async_pool = coro_get_av (aTHX_ "Coro::async_pool", TRUE);	2879	av_async_pool = coro_get_av (aTHX_ "Coro::async_pool", TRUE);
2401	sv_pool_rss = coro_get_sv (aTHX_ "Coro::POOL_RSS" , TRUE);	2880	sv_pool_rss = coro_get_sv (aTHX_ "Coro::POOL_RSS" , TRUE);
2402	sv_pool_size = coro_get_sv (aTHX_ "Coro::POOL_SIZE" , TRUE);	2881	sv_pool_size = coro_get_sv (aTHX_ "Coro::POOL_SIZE" , TRUE);
2403		2882	cv_coro_run = get_cv (aTHX_ "Coro::_terminate", GV_ADD);
2404	coro_current = coro_get_sv (aTHX_ "Coro::current", FALSE);	2883	coro_current = coro_get_sv (aTHX_ "Coro::current" , FALSE);
2405	SvREADONLY_on (coro_current);	2884	SvREADONLY_on (coro_current);
2406		2885
2407	coro_stash = gv_stashpv ("Coro", TRUE);	2886	coro_stash = gv_stashpv ("Coro", TRUE);
2408		2887
2409	newCONSTSUB (coro_stash, "PRIO_MAX", newSViv (PRIO_MAX));	2888	newCONSTSUB (coro_stash, "PRIO_MAX", newSViv (PRIO_MAX));
…		…
2425	coroapi.ready = api_ready;	2904	coroapi.ready = api_ready;
2426	coroapi.is_ready = api_is_ready;	2905	coroapi.is_ready = api_is_ready;
2427	coroapi.nready = coro_nready;	2906	coroapi.nready = coro_nready;
2428	coroapi.current = coro_current;	2907	coroapi.current = coro_current;
2429		2908
2430	GCoroAPI = &coroapi;	2909	/*GCoroAPI = &coroapi;*/
2431	sv_setiv (sv, (IV)&coroapi);	2910	sv_setiv (sv, (IV)&coroapi);
2432	SvREADONLY_on (sv);	2911	SvREADONLY_on (sv);
2433	}	2912	}
2434	}	2913	}
2435		2914
2436	void	2915	void
2437	schedule (...)	2916	schedule (...)
2438	CODE:	2917	CODE:
2439	api_execute_slf (aTHX_ cv, slf_init_schedule, &ST (0), 0);	2918	CORO_EXECUTE_SLF_XS (slf_init_schedule);
2440		2919
2441	void	2920	void
2442	cede (...)	2921	cede (...)
2443	CODE:	2922	CODE:
2444	api_execute_slf (aTHX_ cv, slf_init_cede, &ST (0), 0);	2923	CORO_EXECUTE_SLF_XS (slf_init_cede);
2445		2924
2446	void	2925	void
2447	cede_notself (...)	2926	cede_notself (...)
2448	CODE:	2927	CODE:
2449	api_execute_slf (aTHX_ cv, slf_init_cede_notself, &ST (0), 0);	2928	CORO_EXECUTE_SLF_XS (slf_init_cede_notself);
2450		2929
2451	void	2930	void
2452	_set_current (SV *current)	2931	_set_current (SV *current)
2453	PROTOTYPE: $	2932	PROTOTYPE: $
2454	CODE:	2933	CODE:
…		…
2504	# for async_pool speedup	2983	# for async_pool speedup
2505	void	2984	void
2506	_pool_1 (SV *cb)	2985	_pool_1 (SV *cb)
2507	CODE:	2986	CODE:
2508	{	2987	{
2509	struct coro *coro = SvSTATE (coro_current);
2510	HV *hv = (HV *)SvRV (coro_current);	2988	HV *hv = (HV *)SvRV (coro_current);
		2989	struct coro *coro = SvSTATE_hv ((SV *)hv);
2511	AV *defav = GvAV (PL_defgv);	2990	AV *defav = GvAV (PL_defgv);
2512	SV *invoke = hv_delete (hv, "_invoke", sizeof ("_invoke") - 1, 0);	2991	SV *invoke = hv_delete (hv, "_invoke", sizeof ("_invoke") - 1, 0);
2513	AV *invoke_av;	2992	AV *invoke_av;
2514	int i, len;	2993	int i, len;
2515		2994
…		…
2542		3021
2543	void	3022	void
2544	_pool_2 (SV *cb)	3023	_pool_2 (SV *cb)
2545	CODE:	3024	CODE:
2546	{	3025	{
2547	struct coro *coro = SvSTATE (coro_current);	3026	HV *hv = (HV *)SvRV (coro_current);
		3027	struct coro *coro = SvSTATE_hv ((SV *)hv);
2548		3028
2549	sv_setsv (cb, &PL_sv_undef);	3029	sv_setsv (cb, &PL_sv_undef);
2550		3030
2551	SvREFCNT_dec ((SV *)PL_defoutgv); PL_defoutgv = (GV *)coro->saved_deffh;	3031	SvREFCNT_dec ((SV *)PL_defoutgv); PL_defoutgv = (GV *)coro->saved_deffh;
2552	coro->saved_deffh = 0;	3032	coro->saved_deffh = 0;
…		…
2559	SvREFCNT_dec (old);	3039	SvREFCNT_dec (old);
2560	croak ("\3async_pool terminate\2\n");	3040	croak ("\3async_pool terminate\2\n");
2561	}	3041	}
2562		3042
2563	av_clear (GvAV (PL_defgv));	3043	av_clear (GvAV (PL_defgv));
2564	hv_store ((HV *)SvRV (coro_current), "desc", sizeof ("desc") - 1,	3044	hv_store (hv, "desc", sizeof ("desc") - 1,
2565	newSVpvn ("[async_pool idle]", sizeof ("[async_pool idle]") - 1), 0);	3045	newSVpvn ("[async_pool idle]", sizeof ("[async_pool idle]") - 1), 0);
2566		3046
2567	coro->prio = 0;	3047	coro->prio = 0;
2568		3048
2569	if (coro->cctx && (coro->cctx->flags & CC_TRACE))	3049	if (coro->cctx && (coro->cctx->flags & CC_TRACE))
2570	api_trace (aTHX_ coro_current, 0);	3050	api_trace (aTHX_ coro_current, 0);
2571		3051
2572	av_push (av_async_pool, newSVsv (coro_current));	3052	av_push (av_async_pool, newSVsv (coro_current));
2573	}	3053	}
2574		3054
2575		3055	SV *
2576	MODULE = Coro::State PACKAGE = Coro::AIO	3056	rouse_cb ()
		3057	PROTOTYPE:
		3058	CODE:
		3059	RETVAL = coro_new_rouse_cb (aTHX);
		3060	OUTPUT:
		3061	RETVAL
2577		3062
2578	void	3063	void
2579	_get_state (SV *self)	3064	rouse_wait (...)
2580	PROTOTYPE: $	3065	PROTOTYPE: ;$
2581	PPCODE:	3066	PPCODE:
2582	{	3067	CORO_EXECUTE_SLF_XS (slf_init_rouse_wait);
2583	AV *defav = GvAV (PL_defgv);
2584	AV *av = newAV ();
2585	int i;
2586	SV *data_sv = newSV (sizeof (struct io_state));
2587	struct io_state *data = (struct io_state *)SvPVX (data_sv);
2588	SvCUR_set (data_sv, sizeof (struct io_state));
2589	SvPOK_only (data_sv);
2590
2591	data->errorno = errno;
2592	data->laststype = PL_laststype;
2593	data->laststatval = PL_laststatval;
2594	data->statcache = PL_statcache;
2595
2596	av_extend (av, AvFILLp (defav) + 1 + 1);
2597
2598	for (i = 0; i <= AvFILLp (defav); ++i)
2599	av_push (av, SvREFCNT_inc_NN (AvARRAY (defav)[i]));
2600
2601	av_push (av, data_sv);
2602
2603	XPUSHs (sv_2mortal (newRV_noinc ((SV *)av)));
2604
2605	api_ready (aTHX_ self);
2606	}
2607
2608	void
2609	_set_state (SV *state)
2610	PROTOTYPE: $
2611	PPCODE:
2612	{
2613	AV *av = (AV *)SvRV (state);
2614	struct io_state *data = (struct io_state *)SvPVX (AvARRAY (av)[AvFILLp (av)]);
2615	int i;
2616
2617	errno = data->errorno;
2618	PL_laststype = data->laststype;
2619	PL_laststatval = data->laststatval;
2620	PL_statcache = data->statcache;
2621
2622	EXTEND (SP, AvFILLp (av));
2623	for (i = 0; i < AvFILLp (av); ++i)
2624	PUSHs (sv_2mortal (SvREFCNT_inc_NN (AvARRAY (av)[i])));
2625	}
2626
2627
2628	MODULE = Coro::State PACKAGE = Coro::AnyEvent
2629
2630	BOOT:
2631	sv_activity = coro_get_sv (aTHX_ "Coro::AnyEvent::ACTIVITY", TRUE);
2632
2633	void
2634	_schedule (...)
2635	CODE:
2636	{
2637	static int incede;
2638
2639	api_cede_notself (aTHX);
2640
2641	++incede;
2642	while (coro_nready >= incede && api_cede (aTHX))
2643	;
2644
2645	sv_setsv (sv_activity, &PL_sv_undef);
2646	if (coro_nready >= incede)
2647	{
2648	PUSHMARK (SP);
2649	PUTBACK;
2650	call_pv ("Coro::AnyEvent::_activity", G_DISCARD | G_EVAL);
2651	SPAGAIN;
2652	}
2653
2654	--incede;
2655	}
2656		3068
2657		3069
2658	MODULE = Coro::State PACKAGE = PerlIO::cede	3070	MODULE = Coro::State PACKAGE = PerlIO::cede
2659		3071
2660	BOOT:	3072	BOOT:
2661	PerlIO_define_layer (aTHX_ &PerlIO_cede);	3073	PerlIO_define_layer (aTHX_ &PerlIO_cede);
2662		3074
		3075
2663	MODULE = Coro::State PACKAGE = Coro::Semaphore	3076	MODULE = Coro::State PACKAGE = Coro::Semaphore
2664		3077
2665	SV *	3078	SV *
2666	new (SV *klass, SV *count_ = 0)	3079	new (SV *klass, SV *count = 0)
2667	CODE:	3080	CODE:
2668	{	3081	RETVAL = sv_bless (
2669	/* a semaphore contains a counter IV in $sem->[0] and any waiters after that */	3082	coro_waitarray_new (aTHX_ count && SvOK (count) ? SvIV (count) : 1),
2670	AV *av = newAV ();	3083	GvSTASH (CvGV (cv))
2671	av_push (av, newSViv (count_ && SvOK (count_) ? SvIV (count_) : 1));	3084	);
2672	RETVAL = sv_bless (newRV_noinc ((SV *)av), GvSTASH (CvGV (cv)));	3085	OUTPUT:
2673	}	3086	RETVAL
		3087
		3088	# helper for Coro::Channel
		3089	SV *
		3090	_alloc (int count)
		3091	CODE:
		3092	RETVAL = coro_waitarray_new (aTHX_ count);
2674	OUTPUT:	3093	OUTPUT:
2675	RETVAL	3094	RETVAL
2676		3095
2677	SV *	3096	SV *
2678	count (SV *self)	3097	count (SV *self)
…		…
2684	void	3103	void
2685	up (SV *self, int adjust = 1)	3104	up (SV *self, int adjust = 1)
2686	ALIAS:	3105	ALIAS:
2687	adjust = 1	3106	adjust = 1
2688	CODE:	3107	CODE:
2689	coro_semaphore_adjust ((AV *)SvRV (self), ix ? adjust : 1);	3108	coro_semaphore_adjust (aTHX_ (AV *)SvRV (self), ix ? adjust : 1);
2690		3109
2691	void	3110	void
2692	down (SV *self)	3111	down (...)
2693	CODE:	3112	CODE:
2694	api_execute_slf (aTHX_ cv, slf_init_semaphore_down, &ST (0), 1);	3113	CORO_EXECUTE_SLF_XS (slf_init_semaphore_down);
		3114
		3115	void
		3116	wait (...)
		3117	CODE:
		3118	CORO_EXECUTE_SLF_XS (slf_init_semaphore_wait);
2695		3119
2696	void	3120	void
2697	try (SV *self)	3121	try (SV *self)
2698	PPCODE:	3122	PPCODE:
2699	{	3123	{
…		…
2711	XSRETURN_NO;	3135	XSRETURN_NO;
2712	}	3136	}
2713		3137
2714	void	3138	void
2715	waiters (SV *self)	3139	waiters (SV *self)
2716	CODE:	3140	PPCODE:
2717	{	3141	{
2718	AV *av = (AV *)SvRV (self);	3142	AV *av = (AV *)SvRV (self);
		3143	int wcount = AvFILLp (av) + 1 - 1;
2719		3144
2720	if (GIMME_V == G_SCALAR)	3145	if (GIMME_V == G_SCALAR)
2721	XPUSHs (sv_2mortal (newSVsv (AvARRAY (av)[0])));	3146	XPUSHs (sv_2mortal (newSViv (wcount)));
2722	else	3147	else
2723	{	3148	{
2724	int i;	3149	int i;
2725	EXTEND (SP, AvFILLp (av) + 1 - 1);	3150	EXTEND (SP, wcount);
2726	for (i = 1; i <= AvFILLp (av); ++i)	3151	for (i = 1; i <= wcount; ++i)
2727	PUSHs (newSVsv (AvARRAY (av)[i]));	3152	PUSHs (sv_2mortal (newRV_inc (AvARRAY (av)[i])));
2728	}	3153	}
2729	}	3154	}
2730		3155
		3156	MODULE = Coro::State PACKAGE = Coro::Signal
		3157
		3158	SV *
		3159	new (SV *klass)
		3160	CODE:
		3161	RETVAL = sv_bless (
		3162	coro_waitarray_new (aTHX_ 0),
		3163	GvSTASH (CvGV (cv))
		3164	);
		3165	OUTPUT:
		3166	RETVAL
		3167
		3168	void
		3169	wait (...)
		3170	CODE:
		3171	CORO_EXECUTE_SLF_XS (slf_init_signal_wait);
		3172
		3173	void
		3174	broadcast (SV *self)
		3175	CODE:
		3176	{
		3177	AV *av = (AV *)SvRV (self);
		3178	coro_signal_wake (aTHX_ av, AvFILLp (av));
		3179	}
		3180
		3181	void
		3182	send (SV *self)
		3183	CODE:
		3184	{
		3185	AV *av = (AV *)SvRV (self);
		3186
		3187	if (AvFILLp (av))
		3188	coro_signal_wake (aTHX_ av, 1);
		3189	else
		3190	SvIVX (AvARRAY (av)[0]) = 1; /* remember the signal */
		3191	}
		3192
		3193	IV
		3194	awaited (SV *self)
		3195	CODE:
		3196	RETVAL = AvFILLp ((AV *)SvRV (self)) + 1 - 1;
		3197	OUTPUT:
		3198	RETVAL
		3199
		3200
		3201	MODULE = Coro::State PACKAGE = Coro::AnyEvent
		3202
		3203	BOOT:
		3204	sv_activity = coro_get_sv (aTHX_ "Coro::AnyEvent::ACTIVITY", TRUE);
		3205
		3206	void
		3207	_schedule (...)
		3208	CODE:
		3209	{
		3210	static int incede;
		3211
		3212	api_cede_notself (aTHX);
		3213
		3214	++incede;
		3215	while (coro_nready >= incede && api_cede (aTHX))
		3216	;
		3217
		3218	sv_setsv (sv_activity, &PL_sv_undef);
		3219	if (coro_nready >= incede)
		3220	{
		3221	PUSHMARK (SP);
		3222	PUTBACK;
		3223	call_pv ("Coro::AnyEvent::_activity", G_KEEPERR | G_EVAL | G_VOID | G_DISCARD);
		3224	}
		3225
		3226	--incede;
		3227	}
		3228
		3229
		3230	MODULE = Coro::State PACKAGE = Coro::AIO
		3231
		3232	void
		3233	_register (char *target, char *proto, SV *req)
		3234	CODE:
		3235	{
		3236	CV *req_cv = coro_sv_2cv (req);
		3237	/* newXSproto doesn't return the CV on 5.8 */
		3238	CV *slf_cv = newXS (target, coro_aio_req_xs, __FILE__);
		3239	sv_setpv ((SV *)slf_cv, proto);
		3240	sv_magicext ((SV *)slf_cv, (SV *)req_cv, CORO_MAGIC_type_aio, 0, 0, 0);
		3241	}
		3242

Diff Legend

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