ViewVC Help

View File | Revision Log | Show Annotations | Download File

/cvs/Coro/Coro/State.xs

Comparing Coro/Coro/State.xs (file contents):
Revision 1.271 by root, Fri Nov 14 07:35:32 2008 UTC vs.
Revision 1.301 by root, Wed Nov 19 02:41:31 2008 UTC

…		…
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
148	static perl_mutex coro_lock;
149	# define LOCK do { MUTEX_LOCK (&coro_lock); } while (0)
150	# define UNLOCK do { MUTEX_UNLOCK (&coro_lock); } while (0)
151	# if CORO_PTHREAD	153	# if CORO_PTHREAD
152	static void *coro_thx;	154	static void *coro_thx;
153	# endif	155	# endif
154
155	#else
156
157	# define LOCK (void)0
158	# define UNLOCK (void)0
159
160	#endif	156	#endif
161
162	# undef LOCK
163	# define LOCK (void)0
164	# undef UNLOCK
165	# define UNLOCK (void)0
166
167	/* helper storage struct for Coro::AIO */
168	struct io_state
169	{
170	AV *res;
171	int errorno;
172	I32 laststype; /* U16 in 5.10.0 */
173	int laststatval;
174	Stat_t statcache;
175	};
176		157
177	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);
178		163
179	static U32 cctx_gen;	164	static U32 cctx_gen;
180	static size_t cctx_stacksize = CORO_STACKSIZE;	165	static size_t cctx_stacksize = CORO_STACKSIZE;
181	static struct CoroAPI coroapi;	166	static struct CoroAPI coroapi;
182	static AV *main_mainstack; /* used to differentiate between $main and others */	167	static AV *main_mainstack; /* used to differentiate between $main and others */
183	static JMPENV *main_top_env;	168	static JMPENV *main_top_env;
184	static HV *coro_state_stash, *coro_stash;	169	static HV *coro_state_stash, *coro_stash;
185	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 */
186	static volatile struct coro *transfer_next;
187
188	struct transfer_args
189	{
190	struct coro *prev, *next;
191	};
192		171
193	static GV *irsgv; /* $/ */	172	static GV *irsgv; /* $/ */
194	static GV *stdoutgv; /* *STDOUT */	173	static GV *stdoutgv; /* *STDOUT */
195	static SV *rv_diehook;	174	static SV *rv_diehook;
196	static SV *rv_warnhook;	175	static SV *rv_warnhook;
…		…
215	CC_TRACE_LINE = 0x10, /* trace each statement */	194	CC_TRACE_LINE = 0x10, /* trace each statement */
216	CC_TRACE_ALL = CC_TRACE_SUB | CC_TRACE_LINE,	195	CC_TRACE_ALL = CC_TRACE_SUB | CC_TRACE_LINE,
217	};	196	};
218		197
219	/* this is a structure representing a c-level coroutine */	198	/* this is a structure representing a c-level coroutine */
220	typedef struct coro_cctx {	199	typedef struct coro_cctx
		200	{
221	struct coro_cctx *next;	201	struct coro_cctx *next;
222		202
223	/* the stack */	203	/* the stack */
224	void *sptr;	204	void *sptr;
225	size_t ssize;	205	size_t ssize;
…		…
243	CF_NEW = 0x0004, /* has never been switched to */	223	CF_NEW = 0x0004, /* has never been switched to */
244	CF_DESTROYED = 0x0008, /* coroutine data has been freed */	224	CF_DESTROYED = 0x0008, /* coroutine data has been freed */
245	};	225	};
246		226
247	/* the structure where most of the perl state is stored, overlaid on the cxstack */	227	/* the structure where most of the perl state is stored, overlaid on the cxstack */
248	typedef struct {	228	typedef struct
		229	{
249	SV *defsv;	230	SV *defsv;
250	AV *defav;	231	AV *defav;
251	SV *errsv;	232	SV *errsv;
252	SV *irsgv;	233	SV *irsgv;
253	#define VAR(name,type) type name;	234	#define VAR(name,type) type name;
…		…
257		238
258	#define SLOT_COUNT ((sizeof (perl_slots) + sizeof (PERL_CONTEXT) - 1) / sizeof (PERL_CONTEXT))	239	#define SLOT_COUNT ((sizeof (perl_slots) + sizeof (PERL_CONTEXT) - 1) / sizeof (PERL_CONTEXT))
259		240
260	/* this is a structure representing a perl-level coroutine */	241	/* this is a structure representing a perl-level coroutine */
261	struct coro {	242	struct coro {
262	/* the c coroutine allocated to this perl coroutine, if any */	243	/* the C coroutine allocated to this perl coroutine, if any */
263	coro_cctx *cctx;	244	coro_cctx *cctx;
264		245
265	/* process data */	246	/* state data */
		247	struct CoroSLF slf_frame; /* saved slf frame */
266	AV *mainstack;	248	AV *mainstack;
267	perl_slots *slot; /* basically the saved sp */	249	perl_slots *slot; /* basically the saved sp */
268		250
269	AV *args; /* data associated with this coroutine (initial args) */	251	AV *args; /* data associated with this coroutine (initial args) */
270	int refcnt; /* coroutines are refcounted, yes */	252	int refcnt; /* coroutines are refcounted, yes */
271	int flags; /* CF_ flags */	253	int flags; /* CF_ flags */
272	HV *hv; /* the perl hash associated with this coro, if any */	254	HV *hv; /* the perl hash associated with this coro, if any */
		255	void (*on_destroy)(pTHX_ struct coro *coro);
273		256
274	/* statistics */	257	/* statistics */
275	int usecount; /* number of transfers to this coro */	258	int usecount; /* number of transfers to this coro */
276		259
277	/* coro process data */	260	/* coro process data */
278	int prio;	261	int prio;
279	SV *throw; /* exception to be thrown */	262	SV *except; /* exception to be thrown */
280		263
281	/* async_pool */	264	/* async_pool */
282	SV *saved_deffh;	265	SV *saved_deffh;
283		266
284	/* linked list */	267	/* linked list */
285	struct coro *next, *prev;	268	struct coro *next, *prev;
286	};	269	};
287		270
288	typedef struct coro *Coro__State;	271	typedef struct coro *Coro__State;
289	typedef struct coro *Coro__State_or_hashref;	272	typedef struct coro *Coro__State_or_hashref;
		273
		274	/* the following variables are effectively part of the perl context */
		275	/* and get copied between struct coro and these variables */
		276	/* the mainr easonw e don't support windows process emulation */
		277	static struct CoroSLF slf_frame; /* the current slf frame */
290		278
291	/** Coro ********************************************************************/	279	/** Coro ********************************************************************/
292		280
293	#define PRIO_MAX 3	281	#define PRIO_MAX 3
294	#define PRIO_HIGH 1	282	#define PRIO_HIGH 1
…		…
395	SvREFCNT_dec (av); /* sv_magicext increased the refcount */	383	SvREFCNT_dec (av); /* sv_magicext increased the refcount */
396		384
397	return 0;	385	return 0;
398	}	386	}
399		387
400	#define CORO_MAGIC_type_cv PERL_MAGIC_ext	388	#define CORO_MAGIC_type_cv 26
401	#define CORO_MAGIC_type_state PERL_MAGIC_ext	389	#define CORO_MAGIC_type_state PERL_MAGIC_ext
402		390
403	static MGVTBL coro_cv_vtbl = {	391	static MGVTBL coro_cv_vtbl = {
404	0, 0, 0, 0,	392	0, 0, 0, 0,
405	coro_cv_free	393	coro_cv_free
406	};	394	};
407		395
		396	#define CORO_MAGIC_NN(sv, type) \
		397	(expect_true (SvMAGIC (sv)->mg_type == type) \
		398	? SvMAGIC (sv) \
		399	: mg_find (sv, type))
		400
408	#define CORO_MAGIC(sv, type) \	401	#define CORO_MAGIC(sv, type) \
409	SvMAGIC (sv) \	402	(expect_true (SvMAGIC (sv)) \
410	? SvMAGIC (sv)->mg_type == type \	403	? CORO_MAGIC_NN (sv, type) \
411	? SvMAGIC (sv) \
412	: mg_find (sv, type) \
413	: 0	404	: 0)
414		405
415	#define CORO_MAGIC_cv(cv) CORO_MAGIC (((SV *)(cv)), CORO_MAGIC_type_cv)	406	#define CORO_MAGIC_cv(cv) CORO_MAGIC (((SV *)(cv)), CORO_MAGIC_type_cv)
416	#define CORO_MAGIC_state(sv) CORO_MAGIC (((SV *)(sv)), CORO_MAGIC_type_state)	407	#define CORO_MAGIC_state(sv) CORO_MAGIC_NN (((SV *)(sv)), CORO_MAGIC_type_state)
417		408
418	INLINE struct coro *	409	INLINE struct coro *
419	SvSTATE_ (pTHX_ SV *coro)	410	SvSTATE_ (pTHX_ SV *coro)
420	{	411	{
421	HV *stash;	412	HV *stash;
…		…
438	mg = CORO_MAGIC_state (coro);	429	mg = CORO_MAGIC_state (coro);
439	return (struct coro *)mg->mg_ptr;	430	return (struct coro *)mg->mg_ptr;
440	}	431	}
441		432
442	#define SvSTATE(sv) SvSTATE_ (aTHX_ (sv))	433	#define SvSTATE(sv) SvSTATE_ (aTHX_ (sv))
		434
		435	/* faster than SvSTATE, but expects a coroutine hv */
		436	#define SvSTATE_hv(hv) ((struct coro *)CORO_MAGIC_NN ((SV *)hv, CORO_MAGIC_type_state)->mg_ptr)
		437	#define SvSTATE_current SvSTATE_hv (SvRV (coro_current))
443		438
444	/* the next two functions merely cache the padlists */	439	/* the next two functions merely cache the padlists */
445	static void	440	static void
446	get_padlist (pTHX_ CV *cv)	441	get_padlist (pTHX_ CV *cv)
447	{	442	{
…		…
514	CvPADLIST (cv) = (AV *)POPs;	509	CvPADLIST (cv) = (AV *)POPs;
515	}	510	}
516		511
517	PUTBACK;	512	PUTBACK;
518	}	513	}
		514
		515	slf_frame = c->slf_frame;
		516	CORO_THROW = c->except;
519	}	517	}
520		518
521	static void	519	static void
522	save_perl (pTHX_ Coro__State c)	520	save_perl (pTHX_ Coro__State c)
523	{	521	{
		522	c->except = CORO_THROW;
		523	c->slf_frame = slf_frame;
		524
524	{	525	{
525	dSP;	526	dSP;
526	I32 cxix = cxstack_ix;	527	I32 cxix = cxstack_ix;
527	PERL_CONTEXT *ccstk = cxstack;	528	PERL_CONTEXT *ccstk = cxstack;
528	PERL_SI *top_si = PL_curstackinfo;	529	PERL_SI *top_si = PL_curstackinfo;
…		…
595	#undef VAR	596	#undef VAR
596	}	597	}
597	}	598	}
598		599
599	/*	600	/*
600	* allocate various perl stacks. This is an exact copy	601	* allocate various perl stacks. This is almost an exact copy
601	* of perl.c:init_stacks, except that it uses less memory	602	* of perl.c:init_stacks, except that it uses less memory
602	* on the (sometimes correct) assumption that coroutines do	603	* on the (sometimes correct) assumption that coroutines do
603	* not usually need a lot of stackspace.	604	* not usually need a lot of stackspace.
604	*/	605	*/
605	#if CORO_PREFER_PERL_FUNCTIONS	606	#if CORO_PREFER_PERL_FUNCTIONS
…		…
712	#endif	713	#endif
713	}	714	}
714	}	715	}
715		716
716	return rss;	717	return rss;
717	}
718
719	/** set stacklevel support **************************************************/
720
721	/* we sometimes need to create the effect of pp_slf calling us */
722	#define SLF_HEAD (void)0
723	/* we sometimes need to create the effect of leaving via pp_slf */
724	#define SLF_TAIL slf_tail (aTHX)
725
726	INLINE void
727	slf_tail (pTHX)
728	{
729	dSP;
730	SV **bot = SP;
731
732	int gimme = GIMME_V;
733
734	/* make sure we put something on the stack in scalar context */
735	if (gimme == G_SCALAR)
736	{
737	if (sp == bot)
738	XPUSHs (&PL_sv_undef);
739
740	SP = bot + 1;
741	}
742
743	PUTBACK;
744	}	718	}
745		719
746	/** coroutine stack handling ************************************************/	720	/** coroutine stack handling ************************************************/
747		721
748	static int (*orig_sigelem_get) (pTHX_ SV *sv, MAGIC *mg);	722	static int (*orig_sigelem_get) (pTHX_ SV *sv, MAGIC *mg);
…		…
834		808
835	return orig_sigelem_set ? orig_sigelem_set (aTHX_ sv, mg) : 0;	809	return orig_sigelem_set ? orig_sigelem_set (aTHX_ sv, mg) : 0;
836	}	810	}
837		811
838	static void	812	static void
		813	prepare_nop (pTHX_ struct coro_transfer_args *ta)
		814	{
		815	/* kind of mega-hacky, but works */
		816	ta->next = ta->prev = (struct coro *)ta;
		817	}
		818
		819	static int
		820	slf_check_nop (pTHX_ struct CoroSLF *frame)
		821	{
		822	return 0;
		823	}
		824
		825	static UNOP coro_setup_op;
		826
		827	static void NOINLINE /* noinline to keep it out of the transfer fast path */
839	coro_setup (pTHX_ struct coro *coro)	828	coro_setup (pTHX_ struct coro *coro)
840	{	829	{
841	/*	830	/*
842	* emulate part of the perl startup here.	831	* emulate part of the perl startup here.
843	*/	832	*/
…		…
870	{	859	{
871	dSP;	860	dSP;
872	UNOP myop;	861	UNOP myop;
873		862
874	Zero (&myop, 1, UNOP);	863	Zero (&myop, 1, UNOP);
875	myop.op_next = Nullop;	864	myop.op_next = Nullop;
876	myop.op_flags = OPf_WANT_VOID;	865	myop.op_flags = OPf_WANT_VOID;
877		866
878	PUSHMARK (SP);	867	PUSHMARK (SP);
879	XPUSHs (sv_2mortal (av_shift (GvAV (PL_defgv))));	868	XPUSHs (sv_2mortal (av_shift (GvAV (PL_defgv))));
880	PUTBACK;	869	PUTBACK;
…		…
882	PL_op = PL_ppaddr[OP_ENTERSUB](aTHX);	871	PL_op = PL_ppaddr[OP_ENTERSUB](aTHX);
883	SPAGAIN;	872	SPAGAIN;
884	}	873	}
885		874
886	/* this newly created coroutine might be run on an existing cctx which most	875	/* this newly created coroutine might be run on an existing cctx which most
887	* likely was suspended in set_stacklevel, called from pp_set_stacklevel,	876	* likely was suspended in pp_slf, so we have to emulate entering pp_slf here.
888	* so we have to emulate entering pp_set_stacklevel here.
889	*/	877	*/
890	SLF_HEAD;	878	slf_frame.prepare = prepare_nop; /* provide a nop function for an eventual pp_slf */
		879	slf_frame.check = slf_check_nop; /* signal pp_slf to not repeat */
		880
		881	/* and we have to provide the pp_slf op in any case, so pp_slf can skip it */
		882	coro_setup_op.op_next = PL_op;
		883	coro_setup_op.op_type = OP_CUSTOM;
		884	coro_setup_op.op_ppaddr = pp_slf;
		885	/* no flags required, as an init function won't be called */
		886
		887	PL_op = (OP *)&coro_setup_op;
		888
		889	/* copy throw, in case it was set before coro_setup */
		890	CORO_THROW = coro->except;
891	}	891	}
892		892
893	static void	893	static void
894	coro_destruct (pTHX_ struct coro *coro)	894	coro_destruct (pTHX_ struct coro *coro)
895	{	895	{
…		…
919		919
920	SvREFCNT_dec (PL_diehook);	920	SvREFCNT_dec (PL_diehook);
921	SvREFCNT_dec (PL_warnhook);	921	SvREFCNT_dec (PL_warnhook);
922		922
923	SvREFCNT_dec (coro->saved_deffh);	923	SvREFCNT_dec (coro->saved_deffh);
924	SvREFCNT_dec (coro->throw);	924	SvREFCNT_dec (CORO_THROW);
925		925
926	coro_destruct_stacks (aTHX);	926	coro_destruct_stacks (aTHX);
927	}	927	}
928		928
929	INLINE void	929	INLINE void
…		…
939	static int	939	static int
940	runops_trace (pTHX)	940	runops_trace (pTHX)
941	{	941	{
942	COP *oldcop = 0;	942	COP *oldcop = 0;
943	int oldcxix = -2;	943	int oldcxix = -2;
944	struct coro *coro = SvSTATE (coro_current); /* trace cctx is tied to specific coro */	944	struct coro *coro = SvSTATE_current; /* trace cctx is tied to specific coro */
945	coro_cctx *cctx = coro->cctx;	945	coro_cctx *cctx = coro->cctx;
946		946
947	while ((PL_op = CALL_FPTR (PL_op->op_ppaddr) (aTHX)))	947	while ((PL_op = CALL_FPTR (PL_op->op_ppaddr) (aTHX)))
948	{	948	{
949	PERL_ASYNC_CHECK ();	949	PERL_ASYNC_CHECK ();
…		…
1058		1058
1059	TAINT_NOT;	1059	TAINT_NOT;
1060	return 0;	1060	return 0;
1061	}	1061	}
1062		1062
		1063	static struct coro_cctx *cctx_ssl_cctx;
		1064	static struct CoroSLF cctx_ssl_frame;
		1065
1063	static void	1066	static void
1064	prepare_set_stacklevel (struct transfer_args *ta, struct coro_cctx *cctx)	1067	slf_prepare_set_stacklevel (pTHX_ struct coro_transfer_args *ta)
1065	{	1068	{
1066	ta->prev = (struct coro *)cctx;	1069	ta->prev = (struct coro *)cctx_ssl_cctx;
1067	ta->next = 0;	1070	ta->next = 0;
1068	}	1071	}
1069		1072
1070	/* inject a fake call to Coro::State::_cctx_init into the execution */	1073	static int
1071	/* _cctx_init should be careful, as it could be called at almost any time */	1074	slf_check_set_stacklevel (pTHX_ struct CoroSLF *frame)
1072	/* during execution of a perl program */	1075	{
1073	/* also initialises PL_top_env */	1076	*frame = cctx_ssl_frame;
		1077
		1078	return frame->check (aTHX_ frame); /* execute the restored frame - there must be one */
		1079	}
		1080
		1081	/* initialises PL_top_env and injects a pseudo-slf-call to set the stacklevel */
1074	static void NOINLINE	1082	static void NOINLINE
1075	cctx_prepare (pTHX_ coro_cctx *cctx)	1083	cctx_prepare (pTHX_ coro_cctx *cctx)
1076	{	1084	{
1077	dSP;
1078	UNOP myop;
1079
1080	PL_top_env = &PL_start_env;	1085	PL_top_env = &PL_start_env;
1081		1086
1082	if (cctx->flags & CC_TRACE)	1087	if (cctx->flags & CC_TRACE)
1083	PL_runops = runops_trace;	1088	PL_runops = runops_trace;
1084		1089
1085	Zero (&myop, 1, UNOP);	1090	/* we already must be executing an SLF op, there is no other valid way
1086	myop.op_next = PL_op;	1091	* that can lead to creation of a new cctx */
1087	myop.op_flags = OPf_WANT_VOID | OPf_STACKED;	1092	assert (("FATAL: can't prepare slf-less cctx in Coro module (please report)",
		1093	slf_frame.prepare && PL_op->op_ppaddr == pp_slf));
1088		1094
1089	PUSHMARK (SP);	1095	/* we must emulate leaving pp_slf, which is done inside slf_check_set_stacklevel */
1090	EXTEND (SP, 2);	1096	cctx_ssl_cctx = cctx;
1091	PUSHs (sv_2mortal (newSViv ((IV)cctx)));	1097	cctx_ssl_frame = slf_frame;
1092	PUSHs ((SV *)get_cv ("Coro::State::_cctx_init", FALSE));	1098
1093	PUTBACK;	1099	slf_frame.prepare = slf_prepare_set_stacklevel;
1094	PL_op = (OP *)&myop;	1100	slf_frame.check = slf_check_set_stacklevel;
1095	PL_op = PL_ppaddr[OP_ENTERSUB](aTHX);
1096	SPAGAIN;
1097	}	1101	}
1098		1102
1099	/* the tail of transfer: execute stuff we can only do after a transfer */	1103	/* the tail of transfer: execute stuff we can only do after a transfer */
1100	INLINE void	1104	INLINE void
1101	transfer_tail (pTHX)	1105	transfer_tail (pTHX)
1102	{	1106	{
1103	struct coro *next = (struct coro *)transfer_next;
1104	assert (!(transfer_next = 0)); /* just used for the side effect when asserts are enabled */
1105	assert (("FATAL: next coroutine was zero in transfer_tail (please report)", next));
1106
1107	free_coro_mortal (aTHX);	1107	free_coro_mortal (aTHX);
1108	UNLOCK;
1109
1110	if (expect_false (next->throw))
1111	{
1112	SV *exception = sv_2mortal (next->throw);
1113
1114	next->throw = 0;
1115	sv_setsv (ERRSV, exception);
1116	croak (0);
1117	}
1118	}	1108	}
1119		1109
1120	/*	1110	/*
1121	* this is a _very_ stripped down perl interpreter ;)	1111	* this is a _very_ stripped down perl interpreter ;)
1122	*/	1112	*/
…		…
1129	# endif	1119	# endif
1130	#endif	1120	#endif
1131	{	1121	{
1132	dTHX;	1122	dTHX;
1133		1123
1134	/* we are the alternative tail to pp_set_stacklevel */	1124	/* normally we would need to skip the entersub here */
1135	/* so do the same things here */	1125	/* not doing so will re-execute it, which is exactly what we want */
1136	SLF_TAIL;
1137
1138	/* we now skip the op that did lead to transfer() */
1139	PL_op = PL_op->op_next;	1126	/* PL_nop = PL_nop->op_next */
1140		1127
1141	/* inject a fake subroutine call to cctx_init */	1128	/* inject a fake subroutine call to cctx_init */
1142	cctx_prepare (aTHX_ (coro_cctx *)arg);	1129	cctx_prepare (aTHX_ (coro_cctx *)arg);
1143		1130
1144	/* cctx_run is the alternative tail of transfer() */	1131	/* cctx_run is the alternative tail of transfer() */
…		…
1305	/** coroutine switching *****************************************************/	1292	/** coroutine switching *****************************************************/
1306		1293
1307	static void	1294	static void
1308	transfer_check (pTHX_ struct coro *prev, struct coro *next)	1295	transfer_check (pTHX_ struct coro *prev, struct coro *next)
1309	{	1296	{
		1297	/* TODO: throwing up here is considered harmful */
		1298
1310	if (expect_true (prev != next))	1299	if (expect_true (prev != next))
1311	{	1300	{
1312	if (expect_false (!(prev->flags & (CF_RUNNING | CF_NEW))))	1301	if (expect_false (!(prev->flags & (CF_RUNNING | CF_NEW))))
1313	croak ("Coro::State::transfer called with non-running/new prev Coro::State, but can only transfer from running or new states");	1302	croak ("Coro::State::transfer called with non-running/new prev Coro::State, but can only transfer from running or new states,");
1314		1303
1315	if (expect_false (next->flags & CF_RUNNING))	1304	if (expect_false (next->flags & CF_RUNNING))
1316	croak ("Coro::State::transfer called with running next Coro::State, but can only transfer to inactive states");	1305	croak ("Coro::State::transfer called with running next Coro::State, but can only transfer to inactive states,");
1317		1306
1318	if (expect_false (next->flags & CF_DESTROYED))	1307	if (expect_false (next->flags & CF_DESTROYED))
1319	croak ("Coro::State::transfer called with destroyed next Coro::State, but can only transfer to inactive states");	1308	croak ("Coro::State::transfer called with destroyed next Coro::State, but can only transfer to inactive states,");
1320		1309
1321	#if !PERL_VERSION_ATLEAST (5,10,0)	1310	#if !PERL_VERSION_ATLEAST (5,10,0)
1322	if (expect_false (PL_lex_state != LEX_NOTPARSING))	1311	if (expect_false (PL_lex_state != LEX_NOTPARSING))
1323	croak ("Coro::State::transfer called while parsing, but this is not supported in your perl version");	1312	croak ("Coro::State::transfer called while parsing, but this is not supported in your perl version,");
1324	#endif	1313	#endif
1325	}	1314	}
1326	}	1315	}
1327		1316
1328	/* always use the TRANSFER macro */	1317	/* always use the TRANSFER macro */
1329	static void NOINLINE	1318	static void NOINLINE /* noinline so we have a fixed stackframe */
1330	transfer (pTHX_ struct coro *prev, struct coro *next, int force_cctx)	1319	transfer (pTHX_ struct coro *prev, struct coro *next, int force_cctx)
1331	{	1320	{
1332	dSTACKLEVEL;	1321	dSTACKLEVEL;
1333		1322
1334	/* sometimes transfer is only called to set idle_sp */	1323	/* sometimes transfer is only called to set idle_sp */
1335	if (expect_false (!next))	1324	if (expect_false (!next))
1336	{	1325	{
1337	((coro_cctx *)prev)->idle_sp = stacklevel;	1326	((coro_cctx *)prev)->idle_sp = STACKLEVEL;
1338	assert (((coro_cctx *)prev)->idle_te = PL_top_env); /* just for the side-effect when asserts are enabled */	1327	assert (((coro_cctx *)prev)->idle_te = PL_top_env); /* just for the side-effect when asserts are enabled */
1339	}	1328	}
1340	else if (expect_true (prev != next))	1329	else if (expect_true (prev != next))
1341	{	1330	{
1342	coro_cctx *prev__cctx;	1331	coro_cctx *prev__cctx;
…		…
1349	prev->flags |= CF_RUNNING;	1338	prev->flags |= CF_RUNNING;
1350	}	1339	}
1351		1340
1352	prev->flags &= ~CF_RUNNING;	1341	prev->flags &= ~CF_RUNNING;
1353	next->flags |= CF_RUNNING;	1342	next->flags |= CF_RUNNING;
1354
1355	LOCK;
1356		1343
1357	/* first get rid of the old state */	1344	/* first get rid of the old state */
1358	save_perl (aTHX_ prev);	1345	save_perl (aTHX_ prev);
1359		1346
1360	if (expect_false (next->flags & CF_NEW))	1347	if (expect_false (next->flags & CF_NEW))
…		…
1369		1356
1370	prev__cctx = prev->cctx;	1357	prev__cctx = prev->cctx;
1371		1358
1372	/* possibly untie and reuse the cctx */	1359	/* possibly untie and reuse the cctx */
1373	if (expect_true (	1360	if (expect_true (
1374	prev__cctx->idle_sp == stacklevel	1361	prev__cctx->idle_sp == STACKLEVEL
1375	&& !(prev__cctx->flags & CC_TRACE)	1362	&& !(prev__cctx->flags & CC_TRACE)
1376	&& !force_cctx	1363	&& !force_cctx
1377	))	1364	))
1378	{	1365	{
1379	/* I assume that stacklevel is a stronger indicator than PL_top_env changes */	1366	/* I assume that stacklevel is a stronger indicator than PL_top_env changes */
…		…
1393	++next->usecount;	1380	++next->usecount;
1394		1381
1395	if (expect_true (!next->cctx))	1382	if (expect_true (!next->cctx))
1396	next->cctx = cctx_get (aTHX);	1383	next->cctx = cctx_get (aTHX);
1397		1384
1398	assert (("FATAL: transfer_next already nonzero in Coro (please report)", !transfer_next));
1399	transfer_next = next;
1400
1401	if (expect_false (prev__cctx != next->cctx))	1385	if (expect_false (prev__cctx != next->cctx))
1402	{	1386	{
1403	prev__cctx->top_env = PL_top_env;	1387	prev__cctx->top_env = PL_top_env;
1404	PL_top_env = next->cctx->top_env;	1388	PL_top_env = next->cctx->top_env;
1405	coro_transfer (&prev__cctx->cctx, &next->cctx->cctx);	1389	coro_transfer (&prev__cctx->cctx, &next->cctx->cctx);
…		…
1418	coro_state_destroy (pTHX_ struct coro *coro)	1402	coro_state_destroy (pTHX_ struct coro *coro)
1419	{	1403	{
1420	if (coro->flags & CF_DESTROYED)	1404	if (coro->flags & CF_DESTROYED)
1421	return 0;	1405	return 0;
1422		1406
		1407	if (coro->on_destroy)
		1408	coro->on_destroy (aTHX_ coro);
		1409
1423	coro->flags |= CF_DESTROYED;	1410	coro->flags |= CF_DESTROYED;
1424		1411
1425	if (coro->flags & CF_READY)	1412	if (coro->flags & CF_READY)
1426	{	1413	{
1427	/* reduce nready, as destroying a ready coro effectively unreadies it */	1414	/* reduce nready, as destroying a ready coro effectively unreadies it */
1428	/* alternative: look through all ready queues and remove the coro */	1415	/* alternative: look through all ready queues and remove the coro */
1429	LOCK;
1430	--coro_nready;	1416	--coro_nready;
1431	UNLOCK;
1432	}	1417	}
1433	else	1418	else
1434	coro->flags |= CF_READY; /* make sure it is NOT put into the readyqueue */	1419	coro->flags |= CF_READY; /* make sure it is NOT put into the readyqueue */
1435		1420
1436	if (coro->mainstack && coro->mainstack != main_mainstack)	1421	if (coro->mainstack && coro->mainstack != main_mainstack)
1437	{	1422	{
1438	struct coro temp;	1423	struct coro temp;
1439		1424
1440	if (coro->flags & CF_RUNNING)	1425	assert (("FATAL: tried to destroy currently running coroutine (please report)", !(coro->flags & CF_RUNNING)));
1441	croak ("FATAL: tried to destroy currently running coroutine");
1442		1426
1443	save_perl (aTHX_ &temp);	1427	save_perl (aTHX_ &temp);
1444	load_perl (aTHX_ coro);	1428	load_perl (aTHX_ coro);
1445		1429
1446	coro_destruct (aTHX_ coro);	1430	coro_destruct (aTHX_ coro);
…		…
1497	# define MGf_DUP 0	1481	# define MGf_DUP 0
1498	#endif	1482	#endif
1499	};	1483	};
1500		1484
1501	static void	1485	static void
1502	prepare_transfer (pTHX_ struct transfer_args *ta, SV *prev_sv, SV *next_sv)	1486	prepare_transfer (pTHX_ struct coro_transfer_args *ta, SV *prev_sv, SV *next_sv)
1503	{	1487	{
1504	ta->prev = SvSTATE (prev_sv);	1488	ta->prev = SvSTATE (prev_sv);
1505	ta->next = SvSTATE (next_sv);	1489	ta->next = SvSTATE (next_sv);
1506	TRANSFER_CHECK (*ta);	1490	TRANSFER_CHECK (*ta);
1507	}	1491	}
1508		1492
1509	static void	1493	static void
1510	api_transfer (pTHX_ SV *prev_sv, SV *next_sv)	1494	api_transfer (pTHX_ SV *prev_sv, SV *next_sv)
1511	{	1495	{
1512	struct transfer_args ta;	1496	struct coro_transfer_args ta;
1513		1497
1514	prepare_transfer (aTHX_ &ta, prev_sv, next_sv);	1498	prepare_transfer (aTHX_ &ta, prev_sv, next_sv);
1515	TRANSFER (ta, 1);	1499	TRANSFER (ta, 1);
1516	}	1500	}
1517		1501
1518	/** Coro ********************************************************************/	1502	/** Coro ********************************************************************/
1519		1503
1520	static void	1504	INLINE void
1521	coro_enq (pTHX_ SV *coro_sv)	1505	coro_enq (pTHX_ struct coro *coro)
1522	{	1506	{
1523	av_push (coro_ready [SvSTATE (coro_sv)->prio - PRIO_MIN], coro_sv);	1507	av_push (coro_ready [coro->prio - PRIO_MIN], SvREFCNT_inc_NN (coro->hv));
1524	}	1508	}
1525		1509
1526	static SV *	1510	INLINE SV *
1527	coro_deq (pTHX)	1511	coro_deq (pTHX)
1528	{	1512	{
1529	int prio;	1513	int prio;
1530		1514
1531	for (prio = PRIO_MAX - PRIO_MIN + 1; --prio >= 0; )	1515	for (prio = PRIO_MAX - PRIO_MIN + 1; --prio >= 0; )
…		…
1550	if (coro->flags & CF_READY)	1534	if (coro->flags & CF_READY)
1551	return 0;	1535	return 0;
1552		1536
1553	coro->flags |= CF_READY;	1537	coro->flags |= CF_READY;
1554		1538
1555	LOCK;
1556
1557	sv_hook = coro_nready ? 0 : coro_readyhook;	1539	sv_hook = coro_nready ? 0 : coro_readyhook;
1558	xs_hook = coro_nready ? 0 : coroapi.readyhook;	1540	xs_hook = coro_nready ? 0 : coroapi.readyhook;
1559		1541
1560	coro_enq (aTHX_ SvREFCNT_inc_NN (coro_sv));	1542	coro_enq (aTHX_ coro);
1561	++coro_nready;	1543	++coro_nready;
1562		1544
1563	UNLOCK;
1564
1565	if (sv_hook)	1545	if (sv_hook)
1566	{	1546	{
1567	dSP;	1547	dSP;
1568		1548
1569	ENTER;	1549	ENTER;
1570	SAVETMPS;	1550	SAVETMPS;
1571		1551
1572	PUSHMARK (SP);	1552	PUSHMARK (SP);
1573	PUTBACK;	1553	PUTBACK;
1574	call_sv (sv_hook, G_DISCARD);	1554	call_sv (sv_hook, G_VOID | G_DISCARD);
1575	SPAGAIN;
1576		1555
1577	FREETMPS;	1556	FREETMPS;
1578	LEAVE;	1557	LEAVE;
1579	}	1558	}
1580		1559
…		…
1589	{	1568	{
1590	return !!(SvSTATE (coro_sv)->flags & CF_READY);	1569	return !!(SvSTATE (coro_sv)->flags & CF_READY);
1591	}	1570	}
1592		1571
1593	INLINE void	1572	INLINE void
1594	prepare_schedule (pTHX_ struct transfer_args *ta)	1573	prepare_schedule (pTHX_ struct coro_transfer_args *ta)
1595	{	1574	{
1596	SV *prev_sv, *next_sv;	1575	SV *prev_sv, *next_sv;
1597		1576
1598	for (;;)	1577	for (;;)
1599	{	1578	{
1600	LOCK;
1601	next_sv = coro_deq (aTHX);	1579	next_sv = coro_deq (aTHX);
1602		1580
1603	/* nothing to schedule: call the idle handler */	1581	/* nothing to schedule: call the idle handler */
1604	if (expect_false (!next_sv))	1582	if (expect_false (!next_sv))
1605	{	1583	{
1606	dSP;	1584	dSP;
1607	UNLOCK;
1608		1585
1609	ENTER;	1586	ENTER;
1610	SAVETMPS;	1587	SAVETMPS;
1611		1588
1612	PUSHMARK (SP);	1589	PUSHMARK (SP);
1613	PUTBACK;	1590	PUTBACK;
1614	call_sv (get_sv ("Coro::idle", FALSE), G_DISCARD);	1591	call_sv (get_sv ("Coro::idle", FALSE), G_VOID | G_DISCARD);
1615	SPAGAIN;
1616		1592
1617	FREETMPS;	1593	FREETMPS;
1618	LEAVE;	1594	LEAVE;
1619	continue;	1595	continue;
1620	}	1596	}
1621		1597
1622	ta->next = SvSTATE (next_sv);	1598	ta->next = SvSTATE_hv (next_sv);
1623		1599
1624	/* cannot transfer to destroyed coros, skip and look for next */	1600	/* cannot transfer to destroyed coros, skip and look for next */
1625	if (expect_false (ta->next->flags & CF_DESTROYED))	1601	if (expect_false (ta->next->flags & CF_DESTROYED))
1626	{	1602	{
1627	UNLOCK;
1628	SvREFCNT_dec (next_sv);	1603	SvREFCNT_dec (next_sv);
1629	/* coro_nready has already been taken care of by destroy */	1604	/* coro_nready has already been taken care of by destroy */
1630	continue;	1605	continue;
1631	}	1606	}
1632		1607
1633	--coro_nready;	1608	--coro_nready;
1634	UNLOCK;
1635	break;	1609	break;
1636	}	1610	}
1637		1611
1638	/* free this only after the transfer */	1612	/* free this only after the transfer */
1639	prev_sv = SvRV (coro_current);	1613	prev_sv = SvRV (coro_current);
1640	ta->prev = SvSTATE (prev_sv);	1614	ta->prev = SvSTATE_hv (prev_sv);
1641	TRANSFER_CHECK (*ta);	1615	TRANSFER_CHECK (*ta);
1642	assert (("FATAL: next coroutine isn't marked as ready in Coro (please report)", ta->next->flags & CF_READY));	1616	assert (("FATAL: next coroutine isn't marked as ready in Coro (please report)", ta->next->flags & CF_READY));
1643	ta->next->flags &= ~CF_READY;	1617	ta->next->flags &= ~CF_READY;
1644	SvRV_set (coro_current, next_sv);	1618	SvRV_set (coro_current, next_sv);
1645		1619
1646	LOCK;
1647	free_coro_mortal (aTHX);	1620	free_coro_mortal (aTHX);
1648	coro_mortal = prev_sv;	1621	coro_mortal = prev_sv;
1649	UNLOCK;
1650	}	1622	}
1651		1623
1652	INLINE void	1624	INLINE void
1653	prepare_cede (pTHX_ struct transfer_args *ta)	1625	prepare_cede (pTHX_ struct coro_transfer_args *ta)
1654	{	1626	{
1655	api_ready (aTHX_ coro_current);	1627	api_ready (aTHX_ coro_current);
1656	prepare_schedule (aTHX_ ta);	1628	prepare_schedule (aTHX_ ta);
1657	}	1629	}
1658		1630
1659	static void	1631	INLINE void
1660	prepare_cede_notself (pTHX_ struct transfer_args *ta)	1632	prepare_cede_notself (pTHX_ struct coro_transfer_args *ta)
1661	{	1633	{
1662	SV *prev = SvRV (coro_current);	1634	SV *prev = SvRV (coro_current);
1663		1635
1664	if (coro_nready)	1636	if (coro_nready)
1665	{	1637	{
1666	prepare_schedule (aTHX_ ta);	1638	prepare_schedule (aTHX_ ta);
1667	api_ready (aTHX_ prev);	1639	api_ready (aTHX_ prev);
1668	}	1640	}
1669	else	1641	else
1670	ta->prev = ta->next = SvSTATE (prev);	1642	prepare_nop (aTHX_ ta);
1671	}	1643	}
1672		1644
1673	static void	1645	static void
1674	api_schedule (pTHX)	1646	api_schedule (pTHX)
1675	{	1647	{
1676	struct transfer_args ta;	1648	struct coro_transfer_args ta;
1677		1649
1678	prepare_schedule (aTHX_ &ta);	1650	prepare_schedule (aTHX_ &ta);
1679	TRANSFER (ta, 1);	1651	TRANSFER (ta, 1);
1680	}	1652	}
1681		1653
1682	static int	1654	static int
1683	api_cede (pTHX)	1655	api_cede (pTHX)
1684	{	1656	{
1685	struct transfer_args ta;	1657	struct coro_transfer_args ta;
1686		1658
1687	prepare_cede (aTHX_ &ta);	1659	prepare_cede (aTHX_ &ta);
1688		1660
1689	if (expect_true (ta.prev != ta.next))	1661	if (expect_true (ta.prev != ta.next))
1690	{	1662	{
…		…
1698	static int	1670	static int
1699	api_cede_notself (pTHX)	1671	api_cede_notself (pTHX)
1700	{	1672	{
1701	if (coro_nready)	1673	if (coro_nready)
1702	{	1674	{
1703	struct transfer_args ta;	1675	struct coro_transfer_args ta;
1704		1676
1705	prepare_cede_notself (aTHX_ &ta);	1677	prepare_cede_notself (aTHX_ &ta);
1706	TRANSFER (ta, 1);	1678	TRANSFER (ta, 1);
1707	return 1;	1679	return 1;
1708	}	1680	}
…		…
1718	if (flags & CC_TRACE)	1690	if (flags & CC_TRACE)
1719	{	1691	{
1720	if (!coro->cctx)	1692	if (!coro->cctx)
1721	coro->cctx = cctx_new_run ();	1693	coro->cctx = cctx_new_run ();
1722	else if (!(coro->cctx->flags & CC_TRACE))	1694	else if (!(coro->cctx->flags & CC_TRACE))
1723	croak ("cannot enable tracing on coroutine with custom stack");	1695	croak ("cannot enable tracing on coroutine with custom stack,");
1724		1696
1725	coro->cctx->flags |= CC_NOREUSE | (flags & (CC_TRACE | CC_TRACE_ALL));	1697	coro->cctx->flags |= CC_NOREUSE | (flags & (CC_TRACE | CC_TRACE_ALL));
1726	}	1698	}
1727	else if (coro->cctx && coro->cctx->flags & CC_TRACE)	1699	else if (coro->cctx && coro->cctx->flags & CC_TRACE)
1728	{	1700	{
…		…
1733	else	1705	else
1734	coro->slot->runops = RUNOPS_DEFAULT;	1706	coro->slot->runops = RUNOPS_DEFAULT;
1735	}	1707	}
1736	}	1708	}
1737		1709
1738	#if 0	1710	/*****************************************************************************/
		1711	/* schedule-like-function opcode (SLF) */
		1712
		1713	static UNOP slf_restore; /* restore stack as entersub did, for first-re-run */
		1714	static const CV *slf_cv;
		1715	static SV **slf_argv;
		1716	static int slf_argc, slf_arga; /* count, allocated */
		1717	static I32 slf_ax; /* top of stack, for restore */
		1718
		1719	/* this restores the stack in the case we patched the entersub, to */
		1720	/* recreate the stack frame as perl will on following calls */
		1721	/* since entersub cleared the stack */
		1722	static OP *
		1723	pp_restore (pTHX)
		1724	{
		1725	int i;
		1726	SV **SP = PL_stack_base + slf_ax;
		1727
		1728	PUSHMARK (SP);
		1729
		1730	EXTEND (SP, slf_argc + 1);
		1731
		1732	for (i = 0; i < slf_argc; ++i)
		1733	PUSHs (sv_2mortal (slf_argv [i]));
		1734
		1735	PUSHs ((SV *)CvGV (slf_cv));
		1736
		1737	RETURNOP (slf_restore.op_first);
		1738	}
		1739
1739	static int	1740	static void
1740	coro_gensub_free (pTHX_ SV *sv, MAGIC *mg)	1741	slf_prepare_transfer (pTHX_ struct coro_transfer_args *ta)
1741	{	1742	{
1742	AV *padlist;	1743	SV **arg = (SV **)slf_frame.data;
1743	AV *av = (AV *)mg->mg_obj;
1744		1744
1745	abort ();	1745	prepare_transfer (aTHX_ ta, arg [0], arg [1]);
1746
1747	return 0;
1748	}	1746	}
1749		1747
1750	static MGVTBL coro_gensub_vtbl = {	1748	static void
1751	0, 0, 0, 0,	1749	slf_init_transfer (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
1752	coro_gensub_free	1750	{
1753	};	1751	if (items != 2)
1754	#endif	1752	croak ("Coro::State::transfer (prev, next) expects two arguments, not %d,", items);
		1753
		1754	frame->prepare = slf_prepare_transfer;
		1755	frame->check = slf_check_nop;
		1756	frame->data = (void *)arg; /* let's hope it will stay valid */
		1757	}
		1758
		1759	static void
		1760	slf_init_schedule (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		1761	{
		1762	frame->prepare = prepare_schedule;
		1763	frame->check = slf_check_nop;
		1764	}
		1765
		1766	static void
		1767	slf_init_cede (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		1768	{
		1769	frame->prepare = prepare_cede;
		1770	frame->check = slf_check_nop;
		1771	}
		1772
		1773	static void
		1774	slf_init_cede_notself (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		1775	{
		1776	frame->prepare = prepare_cede_notself;
		1777	frame->check = slf_check_nop;
		1778	}
		1779
		1780	/*
		1781	* these not obviously related functions are all rolled into one
		1782	* function to increase chances that they all will call transfer with the same
		1783	* stack offset
		1784	* SLF stands for "schedule-like-function".
		1785	*/
		1786	static OP *
		1787	pp_slf (pTHX)
		1788	{
		1789	I32 checkmark; /* mark SP to see how many elements check has pushed */
		1790
		1791	/* set up the slf frame, unless it has already been set-up */
		1792	/* the latter happens when a new coro has been started */
		1793	/* or when a new cctx was attached to an existing coroutine */
		1794	if (expect_true (!slf_frame.prepare))
		1795	{
		1796	/* first iteration */
		1797	dSP;
		1798	SV **arg = PL_stack_base + TOPMARK + 1;
		1799	int items = SP - arg; /* args without function object */
		1800	SV *gv = *sp;
		1801
		1802	/* do a quick consistency check on the "function" object, and if it isn't */
		1803	/* for us, divert to the real entersub */
		1804	if (SvTYPE (gv) != SVt_PVGV
		1805	|| !GvCV (gv)
		1806	|| !(CvFLAGS (GvCV (gv)) & CVf_SLF))
		1807	return PL_ppaddr[OP_ENTERSUB](aTHX);
		1808
		1809	if (!(PL_op->op_flags & OPf_STACKED))
		1810	{
		1811	/* ampersand-form of call, use @_ instead of stack */
		1812	AV *av = GvAV (PL_defgv);
		1813	arg = AvARRAY (av);
		1814	items = AvFILLp (av) + 1;
		1815	}
		1816
		1817	/* now call the init function, which needs to set up slf_frame */
		1818	((coro_slf_cb)CvXSUBANY (GvCV (gv)).any_ptr)
		1819	(aTHX_ &slf_frame, GvCV (gv), arg, items);
		1820
		1821	/* pop args */
		1822	SP = PL_stack_base + POPMARK;
		1823
		1824	PUTBACK;
		1825	}
		1826
		1827	/* now that we have a slf_frame, interpret it! */
		1828	/* we use a callback system not to make the code needlessly */
		1829	/* complicated, but so we can run multiple perl coros from one cctx */
		1830
		1831	do
		1832	{
		1833	struct coro_transfer_args ta;
		1834
		1835	slf_frame.prepare (aTHX_ &ta);
		1836	TRANSFER (ta, 0);
		1837
		1838	checkmark = PL_stack_sp - PL_stack_base;
		1839	}
		1840	while (slf_frame.check (aTHX_ &slf_frame));
		1841
		1842	slf_frame.prepare = 0; /* invalidate the frame, we are done processing it */
		1843
		1844	/* exception handling */
		1845	if (expect_false (CORO_THROW))
		1846	{
		1847	SV *exception = sv_2mortal (CORO_THROW);
		1848
		1849	CORO_THROW = 0;
		1850	sv_setsv (ERRSV, exception);
		1851	croak (0);
		1852	}
		1853
		1854	/* return value handling - mostly like entersub */
		1855	/* make sure we put something on the stack in scalar context */
		1856	if (GIMME_V == G_SCALAR)
		1857	{
		1858	dSP;
		1859	SV **bot = PL_stack_base + checkmark;
		1860
		1861	if (sp == bot) /* too few, push undef */
		1862	bot [1] = &PL_sv_undef;
		1863	else if (sp != bot + 1) /* too many, take last one */
		1864	bot [1] = *sp;
		1865
		1866	SP = bot + 1;
		1867
		1868	PUTBACK;
		1869	}
		1870
		1871	return NORMAL;
		1872	}
		1873
		1874	static void
		1875	api_execute_slf (pTHX_ CV *cv, coro_slf_cb init_cb, I32 ax)
		1876	{
		1877	int i;
		1878	SV **arg = PL_stack_base + ax;
		1879	int items = PL_stack_sp - arg + 1;
		1880
		1881	assert (("FATAL: SLF call with illegal CV value", !CvANON (cv)));
		1882
		1883	if (PL_op->op_ppaddr != PL_ppaddr [OP_ENTERSUB]
		1884	&& PL_op->op_ppaddr != pp_slf)
		1885	croak ("FATAL: Coro SLF calls can only be made normally, not via goto or any other means, caught");
		1886
		1887	CvFLAGS (cv) |= CVf_SLF;
		1888	CvXSUBANY (cv).any_ptr = (void *)init_cb;
		1889	slf_cv = cv;
		1890
		1891	/* we patch the op, and then re-run the whole call */
		1892	/* we have to put the same argument on the stack for this to work */
		1893	/* and this will be done by pp_restore */
		1894	slf_restore.op_next = (OP *)&slf_restore;
		1895	slf_restore.op_type = OP_CUSTOM;
		1896	slf_restore.op_ppaddr = pp_restore;
		1897	slf_restore.op_first = PL_op;
		1898
		1899	slf_ax = ax - 1; /* undo the ax++ inside dAXMARK */
		1900
		1901	if (PL_op->op_flags & OPf_STACKED)
		1902	{
		1903	if (items > slf_arga)
		1904	{
		1905	slf_arga = items;
		1906	free (slf_argv);
		1907	slf_argv = malloc (slf_arga * sizeof (SV *));
		1908	}
		1909
		1910	slf_argc = items;
		1911
		1912	for (i = 0; i < items; ++i)
		1913	slf_argv [i] = SvREFCNT_inc (arg [i]);
		1914	}
		1915	else
		1916	slf_argc = 0;
		1917
		1918	PL_op->op_ppaddr = pp_slf;
		1919	PL_op->op_type = OP_CUSTOM; /* maybe we should leave it at entersub? */
		1920
		1921	PL_op = (OP *)&slf_restore;
		1922	}
1755		1923
1756	/*****************************************************************************/	1924	/*****************************************************************************/
1757	/* PerlIO::cede */	1925	/* PerlIO::cede */
1758		1926
1759	typedef struct	1927	typedef struct
…		…
1827	PerlIOBuf_get_cnt,	1995	PerlIOBuf_get_cnt,
1828	PerlIOBuf_set_ptrcnt,	1996	PerlIOBuf_set_ptrcnt,
1829	};	1997	};
1830		1998
1831	/*****************************************************************************/	1999	/*****************************************************************************/
		2000	/* Coro::Semaphore & Coro::Signal */
1832		2001
1833	static const CV *slf_cv; /* for quick consistency check */
1834
1835	static UNOP slf_restore; /* restore stack as entersub did, for first-re-run */
1836	static SV *slf_arg0;
1837	static SV *slf_arg1;
1838
1839	/* this restores the stack in the case we patched the entersub, to */
1840	/* recreate the stack frame as perl will on following calls */
1841	/* since entersub cleared the stack */
1842	static OP *	2002	static SV *
1843	pp_restore (pTHX)	2003	coro_waitarray_new (pTHX_ int count)
1844	{	2004	{
1845	dSP;	2005	/* a semaphore contains a counter IV in $sem->[0] and any waiters after that */
		2006	AV *av = newAV ();
		2007	SV **ary;
1846		2008
1847	PUSHMARK (SP);	2009	/* unfortunately, building manually saves memory */
		2010	Newx (ary, 2, SV *);
		2011	AvALLOC (av) = ary;
		2012	AvARRAY (av) = ary;
		2013	AvMAX (av) = 1;
		2014	AvFILLp (av) = 0;
		2015	ary [0] = newSViv (count);
1848		2016
1849	EXTEND (SP, 3);	2017	return newRV_noinc ((SV *)av);
1850	if (slf_arg0) PUSHs (sv_2mortal (slf_arg0));
1851	if (slf_arg1) PUSHs (sv_2mortal (slf_arg1));
1852	PUSHs ((SV *)CvGV (slf_cv));
1853
1854	RETURNOP (slf_restore.op_first);
1855	}	2018	}
1856		2019
1857	#define OPpENTERSUB_SLF 15 /* the part of op_private entersub hopefully doesn't use */	2020	/* semaphore */
1858		2021
1859	enum {	2022	static void
1860	CORO_SLF_CUSTOM = 0,	2023	coro_semaphore_adjust (pTHX_ AV *av, IV adjust)
1861	CORO_SLF_SET_STACKLEVEL = 1,	2024	{
1862	CORO_SLF_TRANSFER = 2	2025	SV *count_sv = AvARRAY (av)[0];
		2026	IV count = SvIVX (count_sv);
		2027
		2028	count += adjust;
		2029	SvIVX (count_sv) = count;
		2030
		2031	/* now wake up as many waiters as are expected to lock */
		2032	while (count > 0 && AvFILLp (av) > 0)
		2033	{
		2034	SV *cb;
		2035
		2036	/* swap first two elements so we can shift a waiter */
		2037	AvARRAY (av)[0] = AvARRAY (av)[1];
		2038	AvARRAY (av)[1] = count_sv;
		2039	cb = av_shift (av);
		2040
		2041	if (SvOBJECT (cb))
		2042	api_ready (aTHX_ cb);
		2043	else
		2044	croak ("callbacks not yet supported");
		2045
		2046	SvREFCNT_dec (cb);
		2047
		2048	--count;
		2049	}
		2050	}
		2051
		2052	static void
		2053	coro_semaphore_on_destroy (pTHX_ struct coro *coro)
		2054	{
		2055	/* call $sem->adjust (0) to possibly wake up some other waiters */
		2056	coro_semaphore_adjust (aTHX_ (AV *)coro->slf_frame.data, 0);
		2057	}
		2058
		2059	static int
		2060	slf_check_semaphore_down_or_wait (pTHX_ struct CoroSLF *frame, int acquire)
		2061	{
		2062	AV *av = (AV *)frame->data;
		2063	SV *count_sv = AvARRAY (av)[0];
		2064
		2065	/* if we are about to throw, don't actually acquire the lock, just throw */
		2066	if (CORO_THROW)
		2067	return 0;
		2068	else if (SvIVX (count_sv) > 0)
		2069	{
		2070	SvSTATE_current->on_destroy = 0;
		2071
		2072	if (acquire)
		2073	SvIVX (count_sv) = SvIVX (count_sv) - 1;
		2074	else
		2075	coro_semaphore_adjust (aTHX_ av, 0);
		2076
		2077	return 0;
		2078	}
		2079	else
		2080	{
		2081	int i;
		2082	/* if we were woken up but can't down, we look through the whole */
		2083	/* waiters list and only add us if we aren't in there already */
		2084	/* this avoids some degenerate memory usage cases */
		2085
		2086	for (i = 1; i <= AvFILLp (av); ++i)
		2087	if (AvARRAY (av)[i] == SvRV (coro_current))
		2088	return 1;
		2089
		2090	av_push (av, SvREFCNT_inc (SvRV (coro_current)));
		2091	return 1;
		2092	}
		2093	}
		2094
		2095	static int
		2096	slf_check_semaphore_down (pTHX_ struct CoroSLF *frame)
		2097	{
		2098	return slf_check_semaphore_down_or_wait (aTHX_ frame, 1);
		2099	}
		2100
		2101	static int
		2102	slf_check_semaphore_wait (pTHX_ struct CoroSLF *frame)
		2103	{
		2104	return slf_check_semaphore_down_or_wait (aTHX_ frame, 0);
		2105	}
		2106
		2107	static void
		2108	slf_init_semaphore_down_or_wait (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		2109	{
		2110	AV *av = (AV *)SvRV (arg [0]);
		2111
		2112	if (SvIVX (AvARRAY (av)[0]) > 0)
		2113	{
		2114	frame->data = (void *)av;
		2115	frame->prepare = prepare_nop;
		2116	}
		2117	else
		2118	{
		2119	av_push (av, SvREFCNT_inc (SvRV (coro_current)));
		2120
		2121	frame->data = (void *)sv_2mortal (SvREFCNT_inc ((SV *)av));
		2122	frame->prepare = prepare_schedule;
		2123
		2124	/* to avoid race conditions when a woken-up coro gets terminated */
		2125	/* we arrange for a temporary on_destroy that calls adjust (0) */
		2126	SvSTATE_current->on_destroy = coro_semaphore_on_destroy;
		2127	}
		2128	}
		2129
		2130	static void
		2131	slf_init_semaphore_down (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		2132	{
		2133	slf_init_semaphore_down_or_wait (aTHX_ frame, cv, arg, items);
		2134	frame->check = slf_check_semaphore_down;
		2135	}
		2136
		2137	static void
		2138	slf_init_semaphore_wait (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		2139	{
		2140	slf_init_semaphore_down_or_wait (aTHX_ frame, cv, arg, items);
		2141	frame->check = slf_check_semaphore_wait;
		2142	}
		2143
		2144	/* signal */
		2145
		2146	static void
		2147	coro_signal_wake (pTHX_ AV *av, int count)
		2148	{
		2149	SvIVX (AvARRAY (av)[0]) = 0;
		2150
		2151	/* now signal count waiters */
		2152	while (count > 0 && AvFILLp (av) > 0)
		2153	{
		2154	SV *cb;
		2155
		2156	/* swap first two elements so we can shift a waiter */
		2157	cb = AvARRAY (av)[0];
		2158	AvARRAY (av)[0] = AvARRAY (av)[1];
		2159	AvARRAY (av)[1] = cb;
		2160
		2161	cb = av_shift (av);
		2162
		2163	api_ready (aTHX_ cb);
		2164	sv_setiv (cb, 0); /* signal waiter */
		2165	SvREFCNT_dec (cb);
		2166
		2167	--count;
		2168	}
		2169	}
		2170
		2171	static int
		2172	slf_check_signal_wait (pTHX_ struct CoroSLF *frame)
		2173	{
		2174	/* if we are about to throw, also stop waiting */
		2175	return SvROK ((SV *)frame->data) && !CORO_THROW;
		2176	}
		2177
		2178	static void
		2179	slf_init_signal_wait (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		2180	{
		2181	AV *av = (AV *)SvRV (arg [0]);
		2182
		2183	if (SvIVX (AvARRAY (av)[0]))
		2184	{
		2185	SvIVX (AvARRAY (av)[0]) = 0;
		2186	frame->prepare = prepare_nop;
		2187	frame->check = slf_check_nop;
		2188	}
		2189	else
		2190	{
		2191	SV *waiter = newRV_inc (SvRV (coro_current)); /* owned by signal av */
		2192
		2193	av_push (av, waiter);
		2194
		2195	frame->data = (void *)sv_2mortal (SvREFCNT_inc_NN (waiter)); /* owned by process */
		2196	frame->prepare = prepare_schedule;
		2197	frame->check = slf_check_signal_wait;
		2198	}
		2199	}
		2200
		2201	/*****************************************************************************/
		2202	/* gensub: simple closure generation utility */
		2203
		2204	#define GENSUB_ARG CvXSUBANY (cv).any_ptr
		2205
		2206	/* create a closure from XS, returns a code reference */
		2207	/* the arg can be accessed via GENSUB_ARG from the callback */
		2208	/* the callback must use dXSARGS/XSRETURN */
		2209	static SV *
		2210	gensub (pTHX_ void (*xsub)(pTHX_ CV *), void *arg)
		2211	{
		2212	CV *cv = (CV *)newSV (0);
		2213
		2214	sv_upgrade ((SV *)cv, SVt_PVCV);
		2215
		2216	CvANON_on (cv);
		2217	CvISXSUB_on (cv);
		2218	CvXSUB (cv) = xsub;
		2219	GENSUB_ARG = arg;
		2220
		2221	return newRV_noinc ((SV *)cv);
		2222	}
		2223
		2224	/*****************************************************************************/
		2225	/* Coro::AIO */
		2226
		2227	#define CORO_MAGIC_type_aio PERL_MAGIC_ext
		2228
		2229	/* helper storage struct */
		2230	struct io_state
		2231	{
		2232	int errorno;
		2233	I32 laststype; /* U16 in 5.10.0 */
		2234	int laststatval;
		2235	Stat_t statcache;
1863	};	2236	};
1864		2237
1865	/* declare prototype */	2238	static void
1866	XS(XS_Coro__State__set_stacklevel);	2239	coro_aio_callback (pTHX_ CV *cv)
1867
1868	/*
1869	* these not obviously related functions are all rolled into one
1870	* function to increase chances that they all will call transfer with the same
1871	* stack offset
1872	* SLF stands for "schedule-like-function".
1873	*/
1874	static OP *
1875	pp_slf (pTHX)
1876	{	2240	{
		2241	dXSARGS;
		2242	AV *state = (AV *)GENSUB_ARG;
		2243	SV *coro = av_pop (state);
		2244	SV *data_sv = newSV (sizeof (struct io_state));
		2245
		2246	av_extend (state, items);
		2247
		2248	sv_upgrade (data_sv, SVt_PV);
		2249	SvCUR_set (data_sv, sizeof (struct io_state));
		2250	SvPOK_only (data_sv);
		2251
		2252	{
		2253	struct io_state *data = (struct io_state *)SvPVX (data_sv);
		2254
		2255	data->errorno = errno;
		2256	data->laststype = PL_laststype;
		2257	data->laststatval = PL_laststatval;
		2258	data->statcache = PL_statcache;
		2259	}
		2260
		2261	/* now build the result vector out of all the parameters and the data_sv */
		2262	{
		2263	int i;
		2264
		2265	for (i = 0; i < items; ++i)
		2266	av_push (state, SvREFCNT_inc_NN (ST (i)));
		2267	}
		2268
		2269	av_push (state, data_sv);
		2270
		2271	api_ready (aTHX_ coro);
		2272	SvREFCNT_dec (coro);
		2273	SvREFCNT_dec ((AV *)state);
		2274	}
		2275
		2276	static int
		2277	slf_check_aio_req (pTHX_ struct CoroSLF *frame)
		2278	{
		2279	AV *state = (AV *)frame->data;
		2280
		2281	/* if we are about to throw, return early */
		2282	/* this does not cancel the aio request, but at least */
		2283	/* it quickly returns */
		2284	if (CORO_THROW)
		2285	return 0;
		2286
		2287	/* one element that is an RV? repeat! */
		2288	if (AvFILLp (state) == 0 && SvROK (AvARRAY (state)[0]))
		2289	return 1;
		2290
		2291	/* restore status */
		2292	{
		2293	SV *data_sv = av_pop (state);
		2294	struct io_state *data = (struct io_state *)SvPVX (data_sv);
		2295
		2296	errno = data->errorno;
		2297	PL_laststype = data->laststype;
		2298	PL_laststatval = data->laststatval;
		2299	PL_statcache = data->statcache;
		2300
		2301	SvREFCNT_dec (data_sv);
		2302	}
		2303
		2304	/* push result values */
		2305	{
1877	dSP;	2306	dSP;
1878	struct transfer_args ta;	2307	int i;
1879	SV **arg = PL_stack_base + TOPMARK + 1;
1880	int items = SP - arg; /* args without function object */
1881	int ix = PL_op->op_private & OPpENTERSUB_SLF;
1882	struct CoroSLF *slf = 0;
1883	SV *gv = *sp;
1884		2308
1885	/* do a quick consistency check on the "function" object, and if it isn't */	2309	EXTEND (SP, AvFILLp (state) + 1);
1886	/* for us, divert to the real entersub */	2310	for (i = 0; i <= AvFILLp (state); ++i)
1887	if (SvTYPE (gv) != SVt_PVGV || CvXSUB (GvCV (gv)) != XS_Coro__State__set_stacklevel)	2311	PUSHs (sv_2mortal (SvREFCNT_inc_NN (AvARRAY (state)[i])));
1888	return PL_ppaddr[OP_ENTERSUB](aTHX);
1889		2312
1890	/* pop args */
1891	SP = PL_stack_base + POPMARK;
1892
1893	if (!(PL_op->op_flags & OPf_STACKED))
1894	{
1895	/* ampersand-form of call, use @_ instead of stack */
1896	AV *av = GvAV (PL_defgv);
1897	arg = AvARRAY (av);
1898	items = AvFILLp (av) + 1;
1899	}
1900
1901	PUTBACK;	2313	PUTBACK;
		2314	}
1902		2315
1903	if (!ix)	2316	return 0;
1904	{	2317	}
1905	slf = (struct CoroSLF *)CvXSUBANY (GvCV (gv)).any_ptr;	2318
1906	ix = slf->prepare (aTHX_ arg, items);	2319	static void
		2320	slf_init_aio_req (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		2321	{
		2322	AV *state = (AV *)sv_2mortal ((SV *)newAV ());
		2323	SV *coro_hv = SvRV (coro_current);
		2324	struct coro *coro = SvSTATE_hv (coro_hv);
		2325
		2326	/* put our coroutine id on the state arg */
		2327	av_push (state, SvREFCNT_inc_NN (coro_hv));
		2328
		2329	/* first see whether we have a non-zero priority and set it as AIO prio */
		2330	if (coro->prio)
1907	}	2331	{
		2332	dSP;
1908		2333
1909	switch (ix)	2334	static SV *prio_cv;
		2335	static SV *prio_sv;
		2336
		2337	if (expect_false (!prio_cv))
		2338	{
		2339	prio_cv = (SV *)get_cv ("IO::AIO::aioreq_pri", 0);
		2340	prio_sv = newSViv (0);
		2341	}
		2342
		2343	PUSHMARK (SP);
		2344	sv_setiv (prio_sv, coro->prio);
		2345	XPUSHs (prio_sv);
		2346
		2347	PUTBACK;
		2348	call_sv (prio_cv, G_VOID | G_DISCARD);
1910	{	2349	}
1911	case CORO_SLF_SET_STACKLEVEL:
1912	prepare_set_stacklevel (&ta, (struct coro_cctx *)SvIV (arg [0]));
1913	break;
1914		2350
1915	case CORO_SLF_TRANSFER:	2351	/* now call the original request */
1916	if (items != 2)	2352	{
1917	croak ("Coro::State::transfer (prev, next) expects two arguments, not %d.", items);	2353	dSP;
		2354	CV *req = (CV *)CORO_MAGIC_NN ((SV *)cv, CORO_MAGIC_type_aio)->mg_obj;
		2355	int i;
1918		2356
1919	prepare_transfer (aTHX_ &ta, arg [0], arg [1]);	2357	PUSHMARK (SP);
1920	break;
1921		2358
1922	case CORO_SLF_SCHEDULE:	2359	/* first push all args to the stack */
1923	prepare_schedule (aTHX_ &ta);	2360	EXTEND (SP, items + 1);
1924	break;
1925		2361
1926	case CORO_SLF_CEDE:	2362	for (i = 0; i < items; ++i)
1927	prepare_cede (aTHX_ &ta);	2363	PUSHs (arg [i]);
1928	break;
1929		2364
1930	case CORO_SLF_CEDE_NOTSELF:	2365	/* now push the callback closure */
1931	prepare_cede_notself (aTHX_ &ta);	2366	PUSHs (sv_2mortal (gensub (aTHX_ coro_aio_callback, (void *)SvREFCNT_inc_NN ((SV *)state))));
1932	break;
1933		2367
1934	default:	2368	/* now call the AIO function - we assume our request is uncancelable */
1935	abort ();
1936	}
1937
1938	do
1939	TRANSFER (ta, 0);
1940	while (slf && slf->check (aTHX));
1941
1942	SPAGAIN;
1943
1944	PUTBACK;	2369	PUTBACK;
1945	SLF_TAIL;	2370	call_sv ((SV *)req, G_VOID | G_DISCARD);
1946	SPAGAIN;	2371	}
1947	RETURN;
1948	}
1949		2372
1950	static void	2373	/* now that the requets is going, we loop toll we have a result */
1951	coro_slf_patch (pTHX_ CV *cv, int ix, SV **args, int items)	2374	frame->data = (void *)state;
1952	{	2375	frame->prepare = prepare_schedule;
1953	assert (("FATAL: SLF call recursion in Coro module (please report)", PL_op->op_ppaddr != pp_slf));	2376	frame->check = slf_check_aio_req;
1954
1955	assert (("FATAL: SLF call with illegal CV value", CvGV (cv)));
1956	slf_cv = cv;
1957
1958	/* we patch the op, and then re-run the whole call */
1959	/* we have to put the same argument on the stack for this to work */
1960	/* and this will be done by pp_restore */
1961	slf_restore.op_next = (OP *)&slf_restore;
1962	slf_restore.op_type = OP_NULL;
1963	slf_restore.op_ppaddr = pp_restore;
1964	slf_restore.op_first = PL_op;
1965
1966	slf_arg0 = items > 0 ? SvREFCNT_inc (args [0]) : 0;
1967	slf_arg1 = items > 1 ? SvREFCNT_inc (args [1]) : 0;
1968
1969	PL_op->op_ppaddr = pp_slf;
1970	PL_op->op_private = PL_op->op_private & ~OPpENTERSUB_SLF | ix; /* we potentially share our private flags with entersub */
1971
1972	PL_op = (OP *)&slf_restore;
1973	}	2377	}
1974		2378
1975	static void	2379	static void
1976	api_execute_slf (pTHX_ CV *cv, const struct CoroSLF *slf, SV **arg, int items)	2380	coro_aio_req_xs (pTHX_ CV *cv)
1977	{	2381	{
1978	CvXSUBANY (cv).any_ptr = (void *)slf;	2382	dXSARGS;
1979	coro_slf_patch (aTHX_ cv, CORO_SLF_CUSTOM, arg, items);	2383
		2384	CORO_EXECUTE_SLF_XS (slf_init_aio_req);
		2385
		2386	XSRETURN_EMPTY;
1980	}	2387	}
		2388
		2389	/*****************************************************************************/
1981		2390
1982	MODULE = Coro::State PACKAGE = Coro::State PREFIX = api_	2391	MODULE = Coro::State PACKAGE = Coro::State PREFIX = api_
1983		2392
1984	PROTOTYPES: DISABLE	2393	PROTOTYPES: DISABLE
1985		2394
1986	BOOT:	2395	BOOT:
1987	{	2396	{
1988	#ifdef USE_ITHREADS	2397	#ifdef USE_ITHREADS
1989	MUTEX_INIT (&coro_lock);
1990	# if CORO_PTHREAD	2398	# if CORO_PTHREAD
1991	coro_thx = PERL_GET_CONTEXT;	2399	coro_thx = PERL_GET_CONTEXT;
1992	# endif	2400	# endif
1993	#endif	2401	#endif
1994	BOOT_PAGESIZE;	2402	BOOT_PAGESIZE;
…		…
2015	main_top_env = PL_top_env;	2423	main_top_env = PL_top_env;
2016		2424
2017	while (main_top_env->je_prev)	2425	while (main_top_env->je_prev)
2018	main_top_env = main_top_env->je_prev;	2426	main_top_env = main_top_env->je_prev;
2019		2427
		2428	{
		2429	SV *slf = sv_2mortal (newSViv (PTR2IV (pp_slf)));
		2430
		2431	if (!PL_custom_op_names) PL_custom_op_names = newHV ();
		2432	hv_store_ent (PL_custom_op_names, slf,
		2433	newSVpv ("coro_slf", 0), 0);
		2434
		2435	if (!PL_custom_op_descs) PL_custom_op_descs = newHV ();
		2436	hv_store_ent (PL_custom_op_descs, slf,
		2437	newSVpv ("coro schedule like function", 0), 0);
		2438	}
		2439
2020	coroapi.ver = CORO_API_VERSION;	2440	coroapi.ver = CORO_API_VERSION;
2021	coroapi.rev = CORO_API_REVISION;	2441	coroapi.rev = CORO_API_REVISION;
		2442
2022	coroapi.transfer = api_transfer;	2443	coroapi.transfer = api_transfer;
		2444
		2445	coroapi.sv_state = SvSTATE_;
2023	coroapi.execute_slf = api_execute_slf;	2446	coroapi.execute_slf = api_execute_slf;
		2447	coroapi.prepare_nop = prepare_nop;
		2448	coroapi.prepare_schedule = prepare_schedule;
		2449	coroapi.prepare_cede = prepare_cede;
		2450	coroapi.prepare_cede_notself = prepare_cede_notself;
2024		2451
2025	{	2452	{
2026	SV **svp = hv_fetch (PL_modglobal, "Time::NVtime", 12, 0);	2453	SV **svp = hv_fetch (PL_modglobal, "Time::NVtime", 12, 0);
2027		2454
2028	if (!svp) croak ("Time::HiRes is required");	2455	if (!svp) croak ("Time::HiRes is required");
…		…
2062	}	2489	}
2063	OUTPUT:	2490	OUTPUT:
2064	RETVAL	2491	RETVAL
2065		2492
2066	void	2493	void
2067	_set_stacklevel (...)	2494	transfer (...)
2068	ALIAS:	2495	PROTOTYPE: $$
2069	_set_stacklevel = CORO_SLF_SET_STACKLEVEL	2496	CODE:
2070	Coro::State::transfer = CORO_SLF_TRANSFER	2497	CORO_EXECUTE_SLF_XS (slf_init_transfer);
2071	Coro::schedule = CORO_SLF_SCHEDULE
2072	Coro::cede = CORO_SLF_CEDE
2073	Coro::cede_notself = CORO_SLF_CEDE_NOTSELF
2074	CODE:
2075	coro_slf_patch (aTHX_ cv, ix, &ST (0), items);
2076		2498
2077	bool	2499	bool
2078	_destroy (SV *coro_sv)	2500	_destroy (SV *coro_sv)
2079	CODE:	2501	CODE:
2080	RETVAL = coro_state_destroy (aTHX_ SvSTATE (coro_sv));	2502	RETVAL = coro_state_destroy (aTHX_ SvSTATE (coro_sv));
…		…
2087	CODE:	2509	CODE:
2088	_exit (code);	2510	_exit (code);
2089		2511
2090	int	2512	int
2091	cctx_stacksize (int new_stacksize = 0)	2513	cctx_stacksize (int new_stacksize = 0)
		2514	PROTOTYPE: ;$
2092	CODE:	2515	CODE:
2093	RETVAL = cctx_stacksize;	2516	RETVAL = cctx_stacksize;
2094	if (new_stacksize)	2517	if (new_stacksize)
2095	{	2518	{
2096	cctx_stacksize = new_stacksize;	2519	cctx_stacksize = new_stacksize;
…		…
2099	OUTPUT:	2522	OUTPUT:
2100	RETVAL	2523	RETVAL
2101		2524
2102	int	2525	int
2103	cctx_max_idle (int max_idle = 0)	2526	cctx_max_idle (int max_idle = 0)
		2527	PROTOTYPE: ;$
2104	CODE:	2528	CODE:
2105	RETVAL = cctx_max_idle;	2529	RETVAL = cctx_max_idle;
2106	if (max_idle > 1)	2530	if (max_idle > 1)
2107	cctx_max_idle = max_idle;	2531	cctx_max_idle = max_idle;
2108	OUTPUT:	2532	OUTPUT:
2109	RETVAL	2533	RETVAL
2110		2534
2111	int	2535	int
2112	cctx_count ()	2536	cctx_count ()
		2537	PROTOTYPE:
2113	CODE:	2538	CODE:
2114	RETVAL = cctx_count;	2539	RETVAL = cctx_count;
2115	OUTPUT:	2540	OUTPUT:
2116	RETVAL	2541	RETVAL
2117		2542
2118	int	2543	int
2119	cctx_idle ()	2544	cctx_idle ()
		2545	PROTOTYPE:
2120	CODE:	2546	CODE:
2121	RETVAL = cctx_idle;	2547	RETVAL = cctx_idle;
2122	OUTPUT:	2548	OUTPUT:
2123	RETVAL	2549	RETVAL
2124		2550
2125	void	2551	void
2126	list ()	2552	list ()
		2553	PROTOTYPE:
2127	PPCODE:	2554	PPCODE:
2128	{	2555	{
2129	struct coro *coro;	2556	struct coro *coro;
2130	for (coro = coro_first; coro; coro = coro->next)	2557	for (coro = coro_first; coro; coro = coro->next)
2131	if (coro->hv)	2558	if (coro->hv)
…		…
2193		2620
2194	void	2621	void
2195	throw (Coro::State self, SV *throw = &PL_sv_undef)	2622	throw (Coro::State self, SV *throw = &PL_sv_undef)
2196	PROTOTYPE: $;$	2623	PROTOTYPE: $;$
2197	CODE:	2624	CODE:
		2625	{
		2626	struct coro *current = SvSTATE_current;
		2627	SV **throwp = self == current ? &CORO_THROW : &self->except;
2198	SvREFCNT_dec (self->throw);	2628	SvREFCNT_dec (*throwp);
2199	self->throw = SvOK (throw) ? newSVsv (throw) : 0;	2629	*throwp = SvOK (throw) ? newSVsv (throw) : 0;
		2630	}
2200		2631
2201	void	2632	void
2202	api_trace (SV *coro, int flags = CC_TRACE | CC_TRACE_SUB)	2633	api_trace (SV *coro, int flags = CC_TRACE | CC_TRACE_SUB)
		2634	PROTOTYPE: $;$
2203	C_ARGS: aTHX_ coro, flags	2635	C_ARGS: aTHX_ coro, flags
2204		2636
2205	SV *	2637	SV *
2206	has_cctx (Coro::State coro)	2638	has_cctx (Coro::State coro)
2207	PROTOTYPE: $	2639	PROTOTYPE: $
…		…
2232	OUTPUT:	2664	OUTPUT:
2233	RETVAL	2665	RETVAL
2234		2666
2235	void	2667	void
2236	force_cctx ()	2668	force_cctx ()
		2669	PROTOTYPE:
2237	CODE:	2670	CODE:
2238	struct coro *coro = SvSTATE (coro_current);
2239	coro->cctx->idle_sp = 0;	2671	SvSTATE_current->cctx->idle_sp = 0;
2240		2672
2241	void	2673	void
2242	swap_defsv (Coro::State self)	2674	swap_defsv (Coro::State self)
2243	PROTOTYPE: $	2675	PROTOTYPE: $
2244	ALIAS:	2676	ALIAS:
2245	swap_defav = 1	2677	swap_defav = 1
2246	CODE:	2678	CODE:
2247	if (!self->slot)	2679	if (!self->slot)
2248	croak ("cannot swap state with coroutine that has no saved state");	2680	croak ("cannot swap state with coroutine that has no saved state,");
2249	else	2681	else
2250	{	2682	{
2251	SV **src = ix ? (SV **)&GvAV (PL_defgv) : &GvSV (PL_defgv);	2683	SV **src = ix ? (SV **)&GvAV (PL_defgv) : &GvSV (PL_defgv);
2252	SV **dst = ix ? (SV **)&self->slot->defav : (SV **)&self->slot->defsv;	2684	SV **dst = ix ? (SV **)&self->slot->defav : (SV **)&self->slot->defsv;
2253		2685
2254	SV *tmp = *src; *src = *dst; *dst = tmp;	2686	SV *tmp = *src; *src = *dst; *dst = tmp;
2255	}	2687	}
		2688
2256		2689
2257	MODULE = Coro::State PACKAGE = Coro	2690	MODULE = Coro::State PACKAGE = Coro
2258		2691
2259	BOOT:	2692	BOOT:
2260	{	2693	{
…		…
2288	coroapi.ready = api_ready;	2721	coroapi.ready = api_ready;
2289	coroapi.is_ready = api_is_ready;	2722	coroapi.is_ready = api_is_ready;
2290	coroapi.nready = coro_nready;	2723	coroapi.nready = coro_nready;
2291	coroapi.current = coro_current;	2724	coroapi.current = coro_current;
2292		2725
2293	GCoroAPI = &coroapi;	2726	/*GCoroAPI = &coroapi;*/
2294	sv_setiv (sv, (IV)&coroapi);	2727	sv_setiv (sv, (IV)&coroapi);
2295	SvREADONLY_on (sv);	2728	SvREADONLY_on (sv);
2296	}	2729	}
2297	}	2730	}
		2731
		2732	void
		2733	schedule (...)
		2734	CODE:
		2735	CORO_EXECUTE_SLF_XS (slf_init_schedule);
		2736
		2737	void
		2738	cede (...)
		2739	CODE:
		2740	CORO_EXECUTE_SLF_XS (slf_init_cede);
		2741
		2742	void
		2743	cede_notself (...)
		2744	CODE:
		2745	CORO_EXECUTE_SLF_XS (slf_init_cede_notself);
2298		2746
2299	void	2747	void
2300	_set_current (SV *current)	2748	_set_current (SV *current)
2301	PROTOTYPE: $	2749	PROTOTYPE: $
2302	CODE:	2750	CODE:
…		…
2305		2753
2306	void	2754	void
2307	_set_readyhook (SV *hook)	2755	_set_readyhook (SV *hook)
2308	PROTOTYPE: $	2756	PROTOTYPE: $
2309	CODE:	2757	CODE:
2310	LOCK;
2311	SvREFCNT_dec (coro_readyhook);	2758	SvREFCNT_dec (coro_readyhook);
2312	coro_readyhook = SvOK (hook) ? newSVsv (hook) : 0;	2759	coro_readyhook = SvOK (hook) ? newSVsv (hook) : 0;
2313	UNLOCK;
2314		2760
2315	int	2761	int
2316	prio (Coro::State coro, int newprio = 0)	2762	prio (Coro::State coro, int newprio = 0)
		2763	PROTOTYPE: $;$
2317	ALIAS:	2764	ALIAS:
2318	nice = 1	2765	nice = 1
2319	CODE:	2766	CODE:
2320	{	2767	{
2321	RETVAL = coro->prio;	2768	RETVAL = coro->prio;
…		…
2353	# for async_pool speedup	2800	# for async_pool speedup
2354	void	2801	void
2355	_pool_1 (SV *cb)	2802	_pool_1 (SV *cb)
2356	CODE:	2803	CODE:
2357	{	2804	{
2358	struct coro *coro = SvSTATE (coro_current);
2359	HV *hv = (HV *)SvRV (coro_current);	2805	HV *hv = (HV *)SvRV (coro_current);
		2806	struct coro *coro = SvSTATE_hv ((SV *)hv);
2360	AV *defav = GvAV (PL_defgv);	2807	AV *defav = GvAV (PL_defgv);
2361	SV *invoke = hv_delete (hv, "_invoke", sizeof ("_invoke") - 1, 0);	2808	SV *invoke = hv_delete (hv, "_invoke", sizeof ("_invoke") - 1, 0);
2362	AV *invoke_av;	2809	AV *invoke_av;
2363	int i, len;	2810	int i, len;
2364		2811
…		…
2385	{	2832	{
2386	av_fill (defav, len - 1);	2833	av_fill (defav, len - 1);
2387	for (i = 0; i < len; ++i)	2834	for (i = 0; i < len; ++i)
2388	av_store (defav, i, SvREFCNT_inc_NN (AvARRAY (invoke_av)[i + 1]));	2835	av_store (defav, i, SvREFCNT_inc_NN (AvARRAY (invoke_av)[i + 1]));
2389	}	2836	}
2390
2391	SvREFCNT_dec (invoke);
2392	}	2837	}
2393		2838
2394	void	2839	void
2395	_pool_2 (SV *cb)	2840	_pool_2 (SV *cb)
2396	CODE:	2841	CODE:
2397	{	2842	{
2398	struct coro *coro = SvSTATE (coro_current);	2843	HV *hv = (HV *)SvRV (coro_current);
		2844	struct coro *coro = SvSTATE_hv ((SV *)hv);
2399		2845
2400	sv_setsv (cb, &PL_sv_undef);	2846	sv_setsv (cb, &PL_sv_undef);
2401		2847
2402	SvREFCNT_dec ((SV *)PL_defoutgv); PL_defoutgv = (GV *)coro->saved_deffh;	2848	SvREFCNT_dec ((SV *)PL_defoutgv); PL_defoutgv = (GV *)coro->saved_deffh;
2403	coro->saved_deffh = 0;	2849	coro->saved_deffh = 0;
…		…
2410	SvREFCNT_dec (old);	2856	SvREFCNT_dec (old);
2411	croak ("\3async_pool terminate\2\n");	2857	croak ("\3async_pool terminate\2\n");
2412	}	2858	}
2413		2859
2414	av_clear (GvAV (PL_defgv));	2860	av_clear (GvAV (PL_defgv));
2415	hv_store ((HV *)SvRV (coro_current), "desc", sizeof ("desc") - 1,	2861	hv_store (hv, "desc", sizeof ("desc") - 1,
2416	newSVpvn ("[async_pool idle]", sizeof ("[async_pool idle]") - 1), 0);	2862	newSVpvn ("[async_pool idle]", sizeof ("[async_pool idle]") - 1), 0);
2417		2863
2418	coro->prio = 0;	2864	coro->prio = 0;
2419		2865
2420	if (coro->cctx && (coro->cctx->flags & CC_TRACE))	2866	if (coro->cctx && (coro->cctx->flags & CC_TRACE))
2421	api_trace (aTHX_ coro_current, 0);	2867	api_trace (aTHX_ coro_current, 0);
2422		2868
2423	av_push (av_async_pool, newSVsv (coro_current));	2869	av_push (av_async_pool, newSVsv (coro_current));
2424	}	2870	}
2425		2871
2426	#if 0
2427		2872
2428	void	2873	MODULE = Coro::State PACKAGE = PerlIO::cede
2429	_generator_call (...)	2874
2430	PROTOTYPE: @	2875	BOOT:
2431	PPCODE:	2876	PerlIO_define_layer (aTHX_ &PerlIO_cede);
2432	fprintf (stderr, "call %p\n", CvXSUBANY(cv).any_ptr);	2877
2433	xxxx	2878
2434	abort ();	2879	MODULE = Coro::State PACKAGE = Coro::Semaphore
2435		2880
2436	SV *	2881	SV *
2437	gensub (SV *sub, ...)	2882	new (SV *klass, SV *count = 0)
2438	PROTOTYPE: &;@	2883	CODE:
2439	CODE:	2884	RETVAL = sv_bless (
2440	{	2885	coro_waitarray_new (aTHX_ count && SvOK (count) ? SvIV (count) : 1),
2441	struct coro *coro;	2886	GvSTASH (CvGV (cv))
2442	MAGIC *mg;	2887	);
2443	CV *xcv;
2444	CV *ncv = (CV *)newSV_type (SVt_PVCV);
2445	int i;
2446
2447	CvGV (ncv) = CvGV (cv);
2448	CvFILE (ncv) = CvFILE (cv);
2449
2450	Newz (0, coro, 1, struct coro);
2451	coro->args = newAV ();
2452	coro->flags = CF_NEW;
2453
2454	av_extend (coro->args, items - 1);
2455	for (i = 1; i < items; i++)
2456	av_push (coro->args, newSVsv (ST (i)));
2457
2458	CvISXSUB_on (ncv);
2459	CvXSUBANY (ncv).any_ptr = (void *)coro;
2460
2461	xcv = GvCV (gv_fetchpv ("Coro::_generator_call", 0, SVt_PVCV));
2462
2463	CvXSUB (ncv) = CvXSUB (xcv);
2464	CvANON_on (ncv);
2465
2466	mg = sv_magicext ((SV *)ncv, 0, CORO_MAGIC_type_state, &coro_gensub_vtbl, (char *)coro, 0);
2467	RETVAL = newRV_noinc ((SV *)ncv);
2468	}
2469	OUTPUT:	2888	OUTPUT:
2470	RETVAL	2889	RETVAL
2471		2890
2472	#endif	2891	# helper for Coro::Channel
		2892	SV *
		2893	_alloc (int count)
		2894	CODE:
		2895	RETVAL = coro_waitarray_new (aTHX_ count);
		2896	OUTPUT:
		2897	RETVAL
2473		2898
2474		2899	SV *
2475	MODULE = Coro::State PACKAGE = Coro::AIO	2900	count (SV *self)
		2901	CODE:
		2902	RETVAL = newSVsv (AvARRAY ((AV *)SvRV (self))[0]);
		2903	OUTPUT:
		2904	RETVAL
2476		2905
2477	void	2906	void
2478	_get_state (SV *self)	2907	up (SV *self, int adjust = 1)
2479	PPCODE:	2908	ALIAS:
2480	{	2909	adjust = 1
2481	AV *defav = GvAV (PL_defgv);	2910	CODE:
2482	AV *av = newAV ();	2911	coro_semaphore_adjust (aTHX_ (AV *)SvRV (self), ix ? adjust : 1);
2483	int i;
2484	SV *data_sv = newSV (sizeof (struct io_state));
2485	struct io_state *data = (struct io_state *)SvPVX (data_sv);
2486	SvCUR_set (data_sv, sizeof (struct io_state));
2487	SvPOK_only (data_sv);
2488
2489	data->errorno = errno;
2490	data->laststype = PL_laststype;
2491	data->laststatval = PL_laststatval;
2492	data->statcache = PL_statcache;
2493
2494	av_extend (av, AvFILLp (defav) + 1 + 1);
2495
2496	for (i = 0; i <= AvFILLp (defav); ++i)
2497	av_push (av, SvREFCNT_inc_NN (AvARRAY (defav)[i]));
2498
2499	av_push (av, data_sv);
2500
2501	XPUSHs (sv_2mortal (newRV_noinc ((SV *)av)));
2502
2503	api_ready (aTHX_ self);
2504	}
2505		2912
2506	void	2913	void
2507	_set_state (SV *state)	2914	down (SV *self)
2508	PROTOTYPE: $	2915	CODE:
		2916	CORO_EXECUTE_SLF_XS (slf_init_semaphore_down);
		2917
		2918	void
		2919	wait (SV *self)
		2920	CODE:
		2921	CORO_EXECUTE_SLF_XS (slf_init_semaphore_wait);
		2922
		2923	void
		2924	try (SV *self)
		2925	PPCODE:
		2926	{
		2927	AV *av = (AV *)SvRV (self);
		2928	SV *count_sv = AvARRAY (av)[0];
		2929	IV count = SvIVX (count_sv);
		2930
		2931	if (count > 0)
		2932	{
		2933	--count;
		2934	SvIVX (count_sv) = count;
		2935	XSRETURN_YES;
		2936	}
		2937	else
		2938	XSRETURN_NO;
		2939	}
		2940
		2941	void
		2942	waiters (SV *self)
		2943	PPCODE:
		2944	{
		2945	AV *av = (AV *)SvRV (self);
		2946	int wcount = AvFILLp (av) + 1 - 1;
		2947
		2948	if (GIMME_V == G_SCALAR)
		2949	XPUSHs (sv_2mortal (newSViv (wcount)));
		2950	else
		2951	{
		2952	int i;
		2953	EXTEND (SP, wcount);
		2954	for (i = 1; i <= wcount; ++i)
		2955	PUSHs (sv_2mortal (newRV_inc (AvARRAY (av)[i])));
		2956	}
		2957	}
		2958
		2959	MODULE = Coro::State PACKAGE = Coro::Signal
		2960
		2961	SV *
		2962	new (SV *klass)
2509	PPCODE:	2963	CODE:
		2964	RETVAL = sv_bless (
		2965	coro_waitarray_new (aTHX_ 0),
		2966	GvSTASH (CvGV (cv))
		2967	);
		2968	OUTPUT:
		2969	RETVAL
		2970
		2971	void
		2972	wait (SV *self)
		2973	CODE:
		2974	CORO_EXECUTE_SLF_XS (slf_init_signal_wait);
		2975
		2976	void
		2977	broadcast (SV *self)
		2978	CODE:
2510	{	2979	{
2511	AV *av = (AV *)SvRV (state);	2980	AV *av = (AV *)SvRV (self);
2512	struct io_state *data = (struct io_state *)SvPVX (AvARRAY (av)[AvFILLp (av)]);	2981	coro_signal_wake (aTHX_ av, AvFILLp (av));
2513	int i;	2982	}
2514		2983
2515	errno = data->errorno;	2984	void
2516	PL_laststype = data->laststype;	2985	send (SV *self)
2517	PL_laststatval = data->laststatval;	2986	CODE:
2518	PL_statcache = data->statcache;	2987	{
		2988	AV *av = (AV *)SvRV (self);
2519		2989
2520	EXTEND (SP, AvFILLp (av));	2990	if (AvFILLp (av))
2521	for (i = 0; i < AvFILLp (av); ++i)	2991	coro_signal_wake (aTHX_ av, 1);
2522	PUSHs (sv_2mortal (SvREFCNT_inc_NN (AvARRAY (av)[i])));	2992	else
		2993	SvIVX (AvARRAY (av)[0]) = 1; /* remember the signal */
2523	}	2994	}
		2995
		2996	IV
		2997	awaited (SV *self)
		2998	CODE:
		2999	RETVAL = AvFILLp ((AV *)SvRV (self)) + 1 - 1;
		3000	OUTPUT:
		3001	RETVAL
2524		3002
2525		3003
2526	MODULE = Coro::State PACKAGE = Coro::AnyEvent	3004	MODULE = Coro::State PACKAGE = Coro::AnyEvent
2527		3005
2528	BOOT:	3006	BOOT:
2529	sv_activity = coro_get_sv (aTHX_ "Coro::AnyEvent::ACTIVITY", TRUE);	3007	sv_activity = coro_get_sv (aTHX_ "Coro::AnyEvent::ACTIVITY", TRUE);
2530		3008
2531	SV *	3009	void
2532	_schedule (...)	3010	_schedule (...)
2533	PROTOTYPE: @
2534	CODE:	3011	CODE:
2535	{	3012	{
2536	static int incede;	3013	static int incede;
2537		3014
2538	api_cede_notself (aTHX);	3015	api_cede_notself (aTHX);
…		…
2544	sv_setsv (sv_activity, &PL_sv_undef);	3021	sv_setsv (sv_activity, &PL_sv_undef);
2545	if (coro_nready >= incede)	3022	if (coro_nready >= incede)
2546	{	3023	{
2547	PUSHMARK (SP);	3024	PUSHMARK (SP);
2548	PUTBACK;	3025	PUTBACK;
2549	call_pv ("Coro::AnyEvent::_activity", G_DISCARD | G_EVAL);	3026	call_pv ("Coro::AnyEvent::_activity", G_KEEPERR | G_EVAL | G_VOID | G_DISCARD);
2550	SPAGAIN;
2551	}	3027	}
2552		3028
2553	--incede;	3029	--incede;
2554	}	3030	}
2555		3031
2556		3032
2557	MODULE = Coro::State PACKAGE = PerlIO::cede	3033	MODULE = Coro::State PACKAGE = Coro::AIO
2558		3034
2559	BOOT:	3035	void
2560	PerlIO_define_layer (aTHX_ &PerlIO_cede);	3036	_register (char *target, char *proto, SV *req)
		3037	CODE:
		3038	{
		3039	HV *st;
		3040	GV *gvp;
		3041	CV *req_cv = sv_2cv (req, &st, &gvp, 0);
		3042	/* newXSproto doesn't return the CV on 5.8 */
		3043	CV *slf_cv = newXS (target, coro_aio_req_xs, __FILE__);
		3044	sv_setpv ((SV *)slf_cv, proto);
		3045	sv_magicext ((SV *)slf_cv, (SV *)req_cv, CORO_MAGIC_type_aio, 0, 0, 0);
		3046	}
2561		3047

Diff Legend

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