ViewVC Help

View File | Revision Log | Show Annotations | Download File

/cvs/Coro/Coro/State.xs

Comparing Coro/Coro/State.xs (file contents):
Revision 1.264 by root, Fri Nov 14 02:29:09 2008 UTC vs.
Revision 1.294 by root, Tue Nov 18 08:31:03 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
…		…
116	# define CORO_PREFER_PERL_FUNCTIONS 0	119	# define CORO_PREFER_PERL_FUNCTIONS 0
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. */
		124	#if __GNUC__ >= 4
		125	# define dSTACKLEVEL int stacklevel_dummy
		126	# define STACKLEVEL __builtin_frame_address (0)
		127	#else
121	#define dSTACKLEVEL volatile char stacklevel	128	# define dSTACKLEVEL volatile void *stacklevel
122	#define STACKLEVEL ((void *)&stacklevel)	129	# define STACKLEVEL ((void *)&stacklevel)
		130	#endif
123		131
124	#define IN_DESTRUCT (PL_main_cv == Nullcv)	132	#define IN_DESTRUCT (PL_main_cv == Nullcv)
125		133
126	#if __GNUC__ >= 3	134	#if __GNUC__ >= 3
127	# define attribute(x) __attribute__(x)	135	# define attribute(x) __attribute__(x)
…		…
139	#define NOINLINE attribute ((noinline))	147	#define NOINLINE attribute ((noinline))
140		148
141	#include "CoroAPI.h"	149	#include "CoroAPI.h"
142		150
143	#ifdef USE_ITHREADS	151	#ifdef USE_ITHREADS
144
145	static perl_mutex coro_lock;
146	# define LOCK do { MUTEX_LOCK (&coro_lock); } while (0)
147	# define UNLOCK do { MUTEX_UNLOCK (&coro_lock); } while (0)
148	# if CORO_PTHREAD	152	# if CORO_PTHREAD
149	static void *coro_thx;	153	static void *coro_thx;
150	# endif	154	# endif
151
152	#else
153
154	# define LOCK (void)0
155	# define UNLOCK (void)0
156
157	#endif	155	#endif
158
159	# undef LOCK
160	# define LOCK (void)0
161	# undef UNLOCK
162	# define UNLOCK (void)0
163
164	/* helper storage struct for Coro::AIO */
165	struct io_state
166	{
167	AV *res;
168	int errorno;
169	I32 laststype; /* U16 in 5.10.0 */
170	int laststatval;
171	Stat_t statcache;
172	};
173		156
174	static double (*nvtime)(); /* so why doesn't it take void? */	157	static double (*nvtime)(); /* so why doesn't it take void? */
		158
		159	/* we hijack an hopefully unused CV flag for our purposes */
		160	#define CVf_SLF 0x4000
		161	static OP *pp_slf (pTHX);
175		162
176	static U32 cctx_gen;	163	static U32 cctx_gen;
177	static size_t cctx_stacksize = CORO_STACKSIZE;	164	static size_t cctx_stacksize = CORO_STACKSIZE;
178	static struct CoroAPI coroapi;	165	static struct CoroAPI coroapi;
179	static AV *main_mainstack; /* used to differentiate between $main and others */	166	static AV *main_mainstack; /* used to differentiate between $main and others */
180	static JMPENV *main_top_env;	167	static JMPENV *main_top_env;
181	static HV *coro_state_stash, *coro_stash;	168	static HV *coro_state_stash, *coro_stash;
182	static volatile SV *coro_mortal; /* will be freed/thrown after next transfer */	169	static volatile SV *coro_mortal; /* will be freed/thrown after next transfer */
183	static volatile struct coro *transfer_next;
184
185	struct transfer_args
186	{
187	struct coro *prev, *next;
188	};
189		170
190	static GV *irsgv; /* $/ */	171	static GV *irsgv; /* $/ */
191	static GV *stdoutgv; /* *STDOUT */	172	static GV *stdoutgv; /* *STDOUT */
192	static SV *rv_diehook;	173	static SV *rv_diehook;
193	static SV *rv_warnhook;	174	static SV *rv_warnhook;
…		…
212	CC_TRACE_LINE = 0x10, /* trace each statement */	193	CC_TRACE_LINE = 0x10, /* trace each statement */
213	CC_TRACE_ALL = CC_TRACE_SUB | CC_TRACE_LINE,	194	CC_TRACE_ALL = CC_TRACE_SUB | CC_TRACE_LINE,
214	};	195	};
215		196
216	/* this is a structure representing a c-level coroutine */	197	/* this is a structure representing a c-level coroutine */
217	typedef struct coro_cctx {	198	typedef struct coro_cctx
		199	{
218	struct coro_cctx *next;	200	struct coro_cctx *next;
219		201
220	/* the stack */	202	/* the stack */
221	void *sptr;	203	void *sptr;
222	size_t ssize;	204	size_t ssize;
…		…
240	CF_NEW = 0x0004, /* has never been switched to */	222	CF_NEW = 0x0004, /* has never been switched to */
241	CF_DESTROYED = 0x0008, /* coroutine data has been freed */	223	CF_DESTROYED = 0x0008, /* coroutine data has been freed */
242	};	224	};
243		225
244	/* the structure where most of the perl state is stored, overlaid on the cxstack */	226	/* the structure where most of the perl state is stored, overlaid on the cxstack */
245	typedef struct {	227	typedef struct
		228	{
246	SV *defsv;	229	SV *defsv;
247	AV *defav;	230	AV *defav;
248	SV *errsv;	231	SV *errsv;
249	SV *irsgv;	232	SV *irsgv;
250	#define VAR(name,type) type name;	233	#define VAR(name,type) type name;
…		…
254		237
255	#define SLOT_COUNT ((sizeof (perl_slots) + sizeof (PERL_CONTEXT) - 1) / sizeof (PERL_CONTEXT))	238	#define SLOT_COUNT ((sizeof (perl_slots) + sizeof (PERL_CONTEXT) - 1) / sizeof (PERL_CONTEXT))
256		239
257	/* this is a structure representing a perl-level coroutine */	240	/* this is a structure representing a perl-level coroutine */
258	struct coro {	241	struct coro {
259	/* the c coroutine allocated to this perl coroutine, if any */	242	/* the C coroutine allocated to this perl coroutine, if any */
260	coro_cctx *cctx;	243	coro_cctx *cctx;
261		244
262	/* process data */	245	/* state data */
		246	struct CoroSLF slf_frame; /* saved slf frame */
263	AV *mainstack;	247	AV *mainstack;
264	perl_slots *slot; /* basically the saved sp */	248	perl_slots *slot; /* basically the saved sp */
265		249
266	AV *args; /* data associated with this coroutine (initial args) */	250	AV *args; /* data associated with this coroutine (initial args) */
267	int refcnt; /* coroutines are refcounted, yes */	251	int refcnt; /* coroutines are refcounted, yes */
268	int flags; /* CF_ flags */	252	int flags; /* CF_ flags */
269	HV *hv; /* the perl hash associated with this coro, if any */	253	HV *hv; /* the perl hash associated with this coro, if any */
		254	void (*on_destroy)(pTHX_ struct coro *coro);
270		255
271	/* statistics */	256	/* statistics */
272	int usecount; /* number of transfers to this coro */	257	int usecount; /* number of transfers to this coro */
273		258
274	/* coro process data */	259	/* coro process data */
…		…
282	struct coro *next, *prev;	267	struct coro *next, *prev;
283	};	268	};
284		269
285	typedef struct coro *Coro__State;	270	typedef struct coro *Coro__State;
286	typedef struct coro *Coro__State_or_hashref;	271	typedef struct coro *Coro__State_or_hashref;
		272
		273	/* the following variables are effectively part of the perl context */
		274	/* and get copied between struct coro and these variables */
		275	/* the mainr easonw e don't support windows process emulation */
		276	static struct CoroSLF slf_frame; /* the current slf frame */
		277	static SV *coro_throw;
287		278
288	/** Coro ********************************************************************/	279	/** Coro ********************************************************************/
289		280
290	#define PRIO_MAX 3	281	#define PRIO_MAX 3
291	#define PRIO_HIGH 1	282	#define PRIO_HIGH 1
…		…
296		287
297	/* for Coro.pm */	288	/* for Coro.pm */
298	static SV *coro_current;	289	static SV *coro_current;
299	static SV *coro_readyhook;	290	static SV *coro_readyhook;
300	static AV *coro_ready [PRIO_MAX - PRIO_MIN + 1];	291	static AV *coro_ready [PRIO_MAX - PRIO_MIN + 1];
301	static int coro_nready;
302	static struct coro *coro_first;	292	static struct coro *coro_first;
		293	#define coro_nready coroapi.nready
303		294
304	/** lowlevel stuff **********************************************************/	295	/** lowlevel stuff **********************************************************/
305		296
306	static SV *	297	static SV *
307	coro_get_sv (pTHX_ const char *name, int create)	298	coro_get_sv (pTHX_ const char *name, int create)
…		…
392	SvREFCNT_dec (av); /* sv_magicext increased the refcount */	383	SvREFCNT_dec (av); /* sv_magicext increased the refcount */
393		384
394	return 0;	385	return 0;
395	}	386	}
396		387
397	#define CORO_MAGIC_type_cv PERL_MAGIC_ext	388	#define CORO_MAGIC_type_cv 26
398	#define CORO_MAGIC_type_state PERL_MAGIC_ext	389	#define CORO_MAGIC_type_state PERL_MAGIC_ext
399		390
400	static MGVTBL coro_cv_vtbl = {	391	static MGVTBL coro_cv_vtbl = {
401	0, 0, 0, 0,	392	0, 0, 0, 0,
402	coro_cv_free	393	coro_cv_free
403	};	394	};
404		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
405	#define CORO_MAGIC(sv, type) \	401	#define CORO_MAGIC(sv, type) \
406	SvMAGIC (sv) \	402	(expect_true (SvMAGIC (sv)) \
407	? SvMAGIC (sv)->mg_type == type \	403	? CORO_MAGIC_NN (sv, type) \
408	? SvMAGIC (sv) \
409	: mg_find (sv, type) \
410	: 0	404	: 0)
411		405
412	#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)
413	#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)
414		408
415	INLINE struct coro *	409	INLINE struct coro *
416	SvSTATE_ (pTHX_ SV *coro)	410	SvSTATE_ (pTHX_ SV *coro)
417	{	411	{
418	HV *stash;	412	HV *stash;
…		…
435	mg = CORO_MAGIC_state (coro);	429	mg = CORO_MAGIC_state (coro);
436	return (struct coro *)mg->mg_ptr;	430	return (struct coro *)mg->mg_ptr;
437	}	431	}
438		432
439	#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))
440		438
441	/* the next two functions merely cache the padlists */	439	/* the next two functions merely cache the padlists */
442	static void	440	static void
443	get_padlist (pTHX_ CV *cv)	441	get_padlist (pTHX_ CV *cv)
444	{	442	{
…		…
511	CvPADLIST (cv) = (AV *)POPs;	509	CvPADLIST (cv) = (AV *)POPs;
512	}	510	}
513		511
514	PUTBACK;	512	PUTBACK;
515	}	513	}
		514
		515	slf_frame = c->slf_frame;
		516	coro_throw = c->throw;
516	}	517	}
517		518
518	static void	519	static void
519	save_perl (pTHX_ Coro__State c)	520	save_perl (pTHX_ Coro__State c)
520	{	521	{
		522	c->throw = coro_throw;
		523	c->slf_frame = slf_frame;
		524
521	{	525	{
522	dSP;	526	dSP;
523	I32 cxix = cxstack_ix;	527	I32 cxix = cxstack_ix;
524	PERL_CONTEXT *ccstk = cxstack;	528	PERL_CONTEXT *ccstk = cxstack;
525	PERL_SI *top_si = PL_curstackinfo;	529	PERL_SI *top_si = PL_curstackinfo;
…		…
592	#undef VAR	596	#undef VAR
593	}	597	}
594	}	598	}
595		599
596	/*	600	/*
597	* allocate various perl stacks. This is an exact copy	601	* allocate various perl stacks. This is almost an exact copy
598	* of perl.c:init_stacks, except that it uses less memory	602	* of perl.c:init_stacks, except that it uses less memory
599	* on the (sometimes correct) assumption that coroutines do	603	* on the (sometimes correct) assumption that coroutines do
600	* not usually need a lot of stackspace.	604	* not usually need a lot of stackspace.
601	*/	605	*/
602	#if CORO_PREFER_PERL_FUNCTIONS	606	#if CORO_PREFER_PERL_FUNCTIONS
…		…
723	#ifndef MgPV_nolen_const	727	#ifndef MgPV_nolen_const
724	#define MgPV_nolen_const(mg) (((((int)(mg)->mg_len)) == HEf_SVKEY) ? \	728	#define MgPV_nolen_const(mg) (((((int)(mg)->mg_len)) == HEf_SVKEY) ? \
725	SvPV_nolen((SV*)((mg)->mg_ptr)) : \	729	SvPV_nolen((SV*)((mg)->mg_ptr)) : \
726	(const char*)(mg)->mg_ptr)	730	(const char*)(mg)->mg_ptr)
727	#endif	731	#endif
728
729	/* we sometimes need to create the effect of entersub calling us */
730	#define SSL_HEAD (void)0
731	/* we somtimes need to create the effect of leaving via entersub */
732	#define SSL_TAIL (void)0
733		732
734	/*	733	/*
735	* This overrides the default magic get method of %SIG elements.	734	* This overrides the default magic get method of %SIG elements.
736	* The original one doesn't provide for reading back of PL_diehook/PL_warnhook	735	* The original one doesn't provide for reading back of PL_diehook/PL_warnhook
737	* and instead of tryign to save and restore the hash elements, we just provide	736	* and instead of tryign to save and restore the hash elements, we just provide
…		…
809		808
810	return orig_sigelem_set ? orig_sigelem_set (aTHX_ sv, mg) : 0;	809	return orig_sigelem_set ? orig_sigelem_set (aTHX_ sv, mg) : 0;
811	}	810	}
812		811
813	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 */
814	coro_setup (pTHX_ struct coro *coro)	828	coro_setup (pTHX_ struct coro *coro)
815	{	829	{
816	/*	830	/*
817	* emulate part of the perl startup here.	831	* emulate part of the perl startup here.
818	*/	832	*/
…		…
845	{	859	{
846	dSP;	860	dSP;
847	UNOP myop;	861	UNOP myop;
848		862
849	Zero (&myop, 1, UNOP);	863	Zero (&myop, 1, UNOP);
850	myop.op_next = Nullop;	864	myop.op_next = Nullop;
851	myop.op_flags = OPf_WANT_VOID;	865	myop.op_flags = OPf_WANT_VOID;
852		866
853	PUSHMARK (SP);	867	PUSHMARK (SP);
854	XPUSHs (sv_2mortal (av_shift (GvAV (PL_defgv))));	868	XPUSHs (sv_2mortal (av_shift (GvAV (PL_defgv))));
855	PUTBACK;	869	PUTBACK;
…		…
857	PL_op = PL_ppaddr[OP_ENTERSUB](aTHX);	871	PL_op = PL_ppaddr[OP_ENTERSUB](aTHX);
858	SPAGAIN;	872	SPAGAIN;
859	}	873	}
860		874
861	/* 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
862	* likely was suspended in set_stacklevel, called from entersub.	876	* likely was suspended in pp_slf, so we have to emulate entering pp_slf here.
863	* set_stacklevel doesn't do anything on return, but entersub does LEAVE,
864	* so we ENTER here for symmetry.
865	*/	877	*/
866	SSL_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->throw;
867	}	891	}
868		892
869	static void	893	static void
870	coro_destruct (pTHX_ struct coro *coro)	894	coro_destruct (pTHX_ struct coro *coro)
871	{	895	{
…		…
895		919
896	SvREFCNT_dec (PL_diehook);	920	SvREFCNT_dec (PL_diehook);
897	SvREFCNT_dec (PL_warnhook);	921	SvREFCNT_dec (PL_warnhook);
898		922
899	SvREFCNT_dec (coro->saved_deffh);	923	SvREFCNT_dec (coro->saved_deffh);
900	SvREFCNT_dec (coro->throw);	924	SvREFCNT_dec (coro_throw);
901		925
902	coro_destruct_stacks (aTHX);	926	coro_destruct_stacks (aTHX);
903	}	927	}
904		928
905	INLINE void	929	INLINE void
…		…
915	static int	939	static int
916	runops_trace (pTHX)	940	runops_trace (pTHX)
917	{	941	{
918	COP *oldcop = 0;	942	COP *oldcop = 0;
919	int oldcxix = -2;	943	int oldcxix = -2;
920	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 */
921	coro_cctx *cctx = coro->cctx;	945	coro_cctx *cctx = coro->cctx;
922		946
923	while ((PL_op = CALL_FPTR (PL_op->op_ppaddr) (aTHX)))	947	while ((PL_op = CALL_FPTR (PL_op->op_ppaddr) (aTHX)))
924	{	948	{
925	PERL_ASYNC_CHECK ();	949	PERL_ASYNC_CHECK ();
…		…
1034		1058
1035	TAINT_NOT;	1059	TAINT_NOT;
1036	return 0;	1060	return 0;
1037	}	1061	}
1038		1062
		1063	static struct coro_cctx *cctx_ssl_cctx;
		1064	static struct CoroSLF cctx_ssl_frame;
		1065
1039	static void	1066	static void
1040	prepare_set_stacklevel (struct transfer_args *ta, struct coro_cctx *cctx)	1067	slf_prepare_set_stacklevel (pTHX_ struct coro_transfer_args *ta)
1041	{	1068	{
1042	ta->prev = (struct coro *)cctx;	1069	ta->prev = (struct coro *)cctx_ssl_cctx;
1043	ta->next = 0;	1070	ta->next = 0;
1044	}	1071	}
1045		1072
1046	/* inject a fake call to Coro::State::_cctx_init into the execution */	1073	static int
1047	/* _cctx_init should be careful, as it could be called at almost any time */	1074	slf_check_set_stacklevel (pTHX_ struct CoroSLF *frame)
1048	/* during execution of a perl program */	1075	{
1049	/* 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 */
1050	static void NOINLINE	1082	static void NOINLINE
1051	cctx_prepare (pTHX_ coro_cctx *cctx)	1083	cctx_prepare (pTHX_ coro_cctx *cctx)
1052	{	1084	{
1053	dSP;
1054	UNOP myop;
1055
1056	PL_top_env = &PL_start_env;	1085	PL_top_env = &PL_start_env;
1057		1086
1058	if (cctx->flags & CC_TRACE)	1087	if (cctx->flags & CC_TRACE)
1059	PL_runops = runops_trace;	1088	PL_runops = runops_trace;
1060		1089
1061	Zero (&myop, 1, UNOP);	1090	/* we already must be executing an SLF op, there is no other valid way
1062	myop.op_next = PL_op;	1091	* that can lead to creation of a new cctx */
1063	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));
1064		1094
1065	PUSHMARK (SP);	1095	/* we must emulate leaving pp_slf, which is done inside slf_check_set_stacklevel */
1066	EXTEND (SP, 2);	1096	cctx_ssl_cctx = cctx;
1067	PUSHs (sv_2mortal (newSViv ((IV)cctx)));	1097	cctx_ssl_frame = slf_frame;
1068	PUSHs ((SV *)get_cv ("Coro::State::_cctx_init", FALSE));	1098
1069	PUTBACK;	1099	slf_frame.prepare = slf_prepare_set_stacklevel;
1070	PL_op = (OP *)&myop;	1100	slf_frame.check = slf_check_set_stacklevel;
1071	PL_op = PL_ppaddr[OP_ENTERSUB](aTHX);
1072	SPAGAIN;
1073	}	1101	}
1074		1102
1075	/* 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 */
1076	INLINE void	1104	INLINE void
1077	transfer_tail (pTHX)	1105	transfer_tail (pTHX)
1078	{	1106	{
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);	1107	free_coro_mortal (aTHX);
1084	UNLOCK;
1085
1086	if (expect_false (next->throw))
1087	{
1088	SV *exception = sv_2mortal (next->throw);
1089
1090	next->throw = 0;
1091	sv_setsv (ERRSV, exception);
1092	croak (0);
1093	}
1094	}	1108	}
1095		1109
1096	/*	1110	/*
1097	* this is a _very_ stripped down perl interpreter ;)	1111	* this is a _very_ stripped down perl interpreter ;)
1098	*/	1112	*/
…		…
1105	# endif	1119	# endif
1106	#endif	1120	#endif
1107	{	1121	{
1108	dTHX;	1122	dTHX;
1109		1123
1110	/* entersub called ENTER, but we never 'returned', undo that here */	1124	/* normally we would need to skip the entersub here */
1111	SSL_TAIL;	1125	/* not doing so will re-execute it, which is exactly what we want */
1112
1113	/* we now skip the op that did lead to transfer() */
1114	PL_op = PL_op->op_next;	1126	/* PL_nop = PL_nop->op_next */
1115		1127
1116	/* inject a fake subroutine call to cctx_init */	1128	/* inject a fake subroutine call to cctx_init */
1117	cctx_prepare (aTHX_ (coro_cctx *)arg);	1129	cctx_prepare (aTHX_ (coro_cctx *)arg);
1118		1130
1119	/* cctx_run is the alternative tail of transfer() */	1131	/* cctx_run is the alternative tail of transfer() */
…		…
1280	/** coroutine switching *****************************************************/	1292	/** coroutine switching *****************************************************/
1281		1293
1282	static void	1294	static void
1283	transfer_check (pTHX_ struct coro *prev, struct coro *next)	1295	transfer_check (pTHX_ struct coro *prev, struct coro *next)
1284	{	1296	{
		1297	/* TODO: throwing up here is considered harmful */
		1298
1285	if (expect_true (prev != next))	1299	if (expect_true (prev != next))
1286	{	1300	{
1287	if (expect_false (!(prev->flags & (CF_RUNNING | CF_NEW))))	1301	if (expect_false (!(prev->flags & (CF_RUNNING | CF_NEW))))
1288	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,");
1289		1303
1290	if (expect_false (next->flags & CF_RUNNING))	1304	if (expect_false (next->flags & CF_RUNNING))
1291	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,");
1292		1306
1293	if (expect_false (next->flags & CF_DESTROYED))	1307	if (expect_false (next->flags & CF_DESTROYED))
1294	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,");
1295		1309
1296	#if !PERL_VERSION_ATLEAST (5,10,0)	1310	#if !PERL_VERSION_ATLEAST (5,10,0)
1297	if (expect_false (PL_lex_state != LEX_NOTPARSING))	1311	if (expect_false (PL_lex_state != LEX_NOTPARSING))
1298	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,");
1299	#endif	1313	#endif
1300	}	1314	}
1301	}	1315	}
1302		1316
1303	/* always use the TRANSFER macro */	1317	/* always use the TRANSFER macro */
1304	static void NOINLINE	1318	static void NOINLINE /* noinline so we have a fixed stackframe */
1305	transfer (pTHX_ struct coro *prev, struct coro *next, int force_cctx)	1319	transfer (pTHX_ struct coro *prev, struct coro *next, int force_cctx)
1306	{	1320	{
1307	dSTACKLEVEL;	1321	dSTACKLEVEL;
1308		1322
1309	/* sometimes transfer is only called to set idle_sp */	1323	/* sometimes transfer is only called to set idle_sp */
…		…
1324	prev->flags |= CF_RUNNING;	1338	prev->flags |= CF_RUNNING;
1325	}	1339	}
1326		1340
1327	prev->flags &= ~CF_RUNNING;	1341	prev->flags &= ~CF_RUNNING;
1328	next->flags |= CF_RUNNING;	1342	next->flags |= CF_RUNNING;
1329
1330	LOCK;
1331		1343
1332	/* first get rid of the old state */	1344	/* first get rid of the old state */
1333	save_perl (aTHX_ prev);	1345	save_perl (aTHX_ prev);
1334		1346
1335	if (expect_false (next->flags & CF_NEW))	1347	if (expect_false (next->flags & CF_NEW))
…		…
1349	prev__cctx->idle_sp == STACKLEVEL	1361	prev__cctx->idle_sp == STACKLEVEL
1350	&& !(prev__cctx->flags & CC_TRACE)	1362	&& !(prev__cctx->flags & CC_TRACE)
1351	&& !force_cctx	1363	&& !force_cctx
1352	))	1364	))
1353	{	1365	{
1354	/* 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 */
1355	assert (("FATAL: current top_env must equal previous top_env in Coro (please report)", PL_top_env == prev__cctx->idle_te));	1367	assert (("FATAL: current top_env must equal previous top_env in Coro (please report)", PL_top_env == prev__cctx->idle_te));
1356		1368
1357	prev->cctx = 0;	1369	prev->cctx = 0;
1358		1370
1359	/* if the cctx is about to be destroyed we need to make sure we won't see it in cctx_get */	1371	/* if the cctx is about to be destroyed we need to make sure we won't see it in cctx_get */
…		…
1367		1379
1368	++next->usecount;	1380	++next->usecount;
1369		1381
1370	if (expect_true (!next->cctx))	1382	if (expect_true (!next->cctx))
1371	next->cctx = cctx_get (aTHX);	1383	next->cctx = cctx_get (aTHX);
1372
1373	assert (("FATAL: transfer_next already nonzero in Coro (please report)", !transfer_next));
1374	transfer_next = next;
1375		1384
1376	if (expect_false (prev__cctx != next->cctx))	1385	if (expect_false (prev__cctx != next->cctx))
1377	{	1386	{
1378	prev__cctx->top_env = PL_top_env;	1387	prev__cctx->top_env = PL_top_env;
1379	PL_top_env = next->cctx->top_env;	1388	PL_top_env = next->cctx->top_env;
…		…
1393	coro_state_destroy (pTHX_ struct coro *coro)	1402	coro_state_destroy (pTHX_ struct coro *coro)
1394	{	1403	{
1395	if (coro->flags & CF_DESTROYED)	1404	if (coro->flags & CF_DESTROYED)
1396	return 0;	1405	return 0;
1397		1406
		1407	if (coro->on_destroy)
		1408	coro->on_destroy (aTHX_ coro);
		1409
1398	coro->flags |= CF_DESTROYED;	1410	coro->flags |= CF_DESTROYED;
1399		1411
1400	if (coro->flags & CF_READY)	1412	if (coro->flags & CF_READY)
1401	{	1413	{
1402	/* reduce nready, as destroying a ready coro effectively unreadies it */	1414	/* reduce nready, as destroying a ready coro effectively unreadies it */
1403	/* alternative: look through all ready queues and remove the coro */	1415	/* alternative: look through all ready queues and remove the coro */
1404	LOCK;
1405	--coro_nready;	1416	--coro_nready;
1406	UNLOCK;
1407	}	1417	}
1408	else	1418	else
1409	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 */
1410		1420
1411	if (coro->mainstack && coro->mainstack != main_mainstack)	1421	if (coro->mainstack && coro->mainstack != main_mainstack)
1412	{	1422	{
1413	struct coro temp;	1423	struct coro temp;
1414		1424
1415	if (coro->flags & CF_RUNNING)	1425	assert (("FATAL: tried to destroy currently running coroutine (please report)", !(coro->flags & CF_RUNNING)));
1416	croak ("FATAL: tried to destroy currently running coroutine");
1417		1426
1418	save_perl (aTHX_ &temp);	1427	save_perl (aTHX_ &temp);
1419	load_perl (aTHX_ coro);	1428	load_perl (aTHX_ coro);
1420		1429
1421	coro_destruct (aTHX_ coro);	1430	coro_destruct (aTHX_ coro);
…		…
1472	# define MGf_DUP 0	1481	# define MGf_DUP 0
1473	#endif	1482	#endif
1474	};	1483	};
1475		1484
1476	static void	1485	static void
1477	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)
1478	{	1487	{
1479	ta->prev = SvSTATE (prev_sv);	1488	ta->prev = SvSTATE (prev_sv);
1480	ta->next = SvSTATE (next_sv);	1489	ta->next = SvSTATE (next_sv);
1481	TRANSFER_CHECK (*ta);	1490	TRANSFER_CHECK (*ta);
1482	}	1491	}
1483		1492
1484	static void	1493	static void
1485	api_transfer (SV *prev_sv, SV *next_sv)	1494	api_transfer (pTHX_ SV *prev_sv, SV *next_sv)
1486	{	1495	{
1487	dTHX;
1488	struct transfer_args ta;	1496	struct coro_transfer_args ta;
1489		1497
1490	prepare_transfer (aTHX_ &ta, prev_sv, next_sv);	1498	prepare_transfer (aTHX_ &ta, prev_sv, next_sv);
1491	TRANSFER (ta, 1);	1499	TRANSFER (ta, 1);
1492	}	1500	}
1493		1501
1494	/** Coro ********************************************************************/	1502	/** Coro ********************************************************************/
1495		1503
1496	static void	1504	INLINE void
1497	coro_enq (pTHX_ SV *coro_sv)	1505	coro_enq (pTHX_ struct coro *coro)
1498	{	1506	{
1499	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));
1500	}	1508	}
1501		1509
1502	static SV *	1510	INLINE SV *
1503	coro_deq (pTHX)	1511	coro_deq (pTHX)
1504	{	1512	{
1505	int prio;	1513	int prio;
1506		1514
1507	for (prio = PRIO_MAX - PRIO_MIN + 1; --prio >= 0; )	1515	for (prio = PRIO_MAX - PRIO_MIN + 1; --prio >= 0; )
…		…
1510		1518
1511	return 0;	1519	return 0;
1512	}	1520	}
1513		1521
1514	static int	1522	static int
1515	api_ready (SV *coro_sv)	1523	api_ready (pTHX_ SV *coro_sv)
1516	{	1524	{
1517	dTHX;
1518	struct coro *coro;	1525	struct coro *coro;
1519	SV *sv_hook;	1526	SV *sv_hook;
1520	void (*xs_hook)(void);	1527	void (*xs_hook)(void);
1521		1528
1522	if (SvROK (coro_sv))	1529	if (SvROK (coro_sv))
…		…
1527	if (coro->flags & CF_READY)	1534	if (coro->flags & CF_READY)
1528	return 0;	1535	return 0;
1529		1536
1530	coro->flags |= CF_READY;	1537	coro->flags |= CF_READY;
1531		1538
1532	LOCK;
1533
1534	sv_hook = coro_nready ? 0 : coro_readyhook;	1539	sv_hook = coro_nready ? 0 : coro_readyhook;
1535	xs_hook = coro_nready ? 0 : coroapi.readyhook;	1540	xs_hook = coro_nready ? 0 : coroapi.readyhook;
1536		1541
1537	coro_enq (aTHX_ SvREFCNT_inc_NN (coro_sv));	1542	coro_enq (aTHX_ coro);
1538	++coro_nready;	1543	++coro_nready;
1539		1544
1540	UNLOCK;
1541
1542	if (sv_hook)	1545	if (sv_hook)
1543	{	1546	{
1544	dSP;	1547	dSP;
1545		1548
1546	ENTER;	1549	ENTER;
1547	SAVETMPS;	1550	SAVETMPS;
1548		1551
1549	PUSHMARK (SP);	1552	PUSHMARK (SP);
1550	PUTBACK;	1553	PUTBACK;
1551	call_sv (sv_hook, G_DISCARD);	1554	call_sv (sv_hook, G_VOID | G_DISCARD);
1552	SPAGAIN;
1553		1555
1554	FREETMPS;	1556	FREETMPS;
1555	LEAVE;	1557	LEAVE;
1556	}	1558	}
1557		1559
…		…
1560		1562
1561	return 1;	1563	return 1;
1562	}	1564	}
1563		1565
1564	static int	1566	static int
1565	api_is_ready (SV *coro_sv)	1567	api_is_ready (pTHX_ SV *coro_sv)
1566	{	1568	{
1567	dTHX;
1568
1569	return !!(SvSTATE (coro_sv)->flags & CF_READY);	1569	return !!(SvSTATE (coro_sv)->flags & CF_READY);
1570	}	1570	}
1571		1571
1572	INLINE void	1572	INLINE void
1573	prepare_schedule (pTHX_ struct transfer_args *ta)	1573	prepare_schedule (pTHX_ struct coro_transfer_args *ta)
1574	{	1574	{
1575	SV *prev_sv, *next_sv;	1575	SV *prev_sv, *next_sv;
1576		1576
1577	for (;;)	1577	for (;;)
1578	{	1578	{
1579	LOCK;
1580	next_sv = coro_deq (aTHX);	1579	next_sv = coro_deq (aTHX);
1581		1580
1582	/* nothing to schedule: call the idle handler */	1581	/* nothing to schedule: call the idle handler */
1583	if (expect_false (!next_sv))	1582	if (expect_false (!next_sv))
1584	{	1583	{
1585	dSP;	1584	dSP;
1586	UNLOCK;
1587		1585
1588	ENTER;	1586	ENTER;
1589	SAVETMPS;	1587	SAVETMPS;
1590		1588
1591	PUSHMARK (SP);	1589	PUSHMARK (SP);
1592	PUTBACK;	1590	PUTBACK;
1593	call_sv (get_sv ("Coro::idle", FALSE), G_DISCARD);	1591	call_sv (get_sv ("Coro::idle", FALSE), G_VOID | G_DISCARD);
1594	SPAGAIN;
1595		1592
1596	FREETMPS;	1593	FREETMPS;
1597	LEAVE;	1594	LEAVE;
1598	continue;	1595	continue;
1599	}	1596	}
1600		1597
1601	ta->next = SvSTATE (next_sv);	1598	ta->next = SvSTATE_hv (next_sv);
1602		1599
1603	/* cannot transfer to destroyed coros, skip and look for next */	1600	/* cannot transfer to destroyed coros, skip and look for next */
1604	if (expect_false (ta->next->flags & CF_DESTROYED))	1601	if (expect_false (ta->next->flags & CF_DESTROYED))
1605	{	1602	{
1606	UNLOCK;
1607	SvREFCNT_dec (next_sv);	1603	SvREFCNT_dec (next_sv);
1608	/* coro_nready has already been taken care of by destroy */	1604	/* coro_nready has already been taken care of by destroy */
1609	continue;	1605	continue;
1610	}	1606	}
1611		1607
1612	--coro_nready;	1608	--coro_nready;
1613	UNLOCK;
1614	break;	1609	break;
1615	}	1610	}
1616		1611
1617	/* free this only after the transfer */	1612	/* free this only after the transfer */
1618	prev_sv = SvRV (coro_current);	1613	prev_sv = SvRV (coro_current);
1619	ta->prev = SvSTATE (prev_sv);	1614	ta->prev = SvSTATE_hv (prev_sv);
1620	TRANSFER_CHECK (*ta);	1615	TRANSFER_CHECK (*ta);
1621	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));
1622	ta->next->flags &= ~CF_READY;	1617	ta->next->flags &= ~CF_READY;
1623	SvRV_set (coro_current, next_sv);	1618	SvRV_set (coro_current, next_sv);
1624		1619
1625	LOCK;
1626	free_coro_mortal (aTHX);	1620	free_coro_mortal (aTHX);
1627	coro_mortal = prev_sv;	1621	coro_mortal = prev_sv;
1628	UNLOCK;
1629	}	1622	}
1630		1623
1631	INLINE void	1624	INLINE void
1632	prepare_cede (pTHX_ struct transfer_args *ta)	1625	prepare_cede (pTHX_ struct coro_transfer_args *ta)
1633	{	1626	{
1634	api_ready (coro_current);	1627	api_ready (aTHX_ coro_current);
1635	prepare_schedule (aTHX_ ta);	1628	prepare_schedule (aTHX_ ta);
1636	}	1629	}
1637		1630
		1631	INLINE void
		1632	prepare_cede_notself (pTHX_ struct coro_transfer_args *ta)
		1633	{
		1634	SV *prev = SvRV (coro_current);
		1635
		1636	if (coro_nready)
		1637	{
		1638	prepare_schedule (aTHX_ ta);
		1639	api_ready (aTHX_ prev);
		1640	}
		1641	else
		1642	prepare_nop (aTHX_ ta);
		1643	}
		1644
		1645	static void
		1646	api_schedule (pTHX)
		1647	{
		1648	struct coro_transfer_args ta;
		1649
		1650	prepare_schedule (aTHX_ &ta);
		1651	TRANSFER (ta, 1);
		1652	}
		1653
1638	static int	1654	static int
1639	prepare_cede_notself (pTHX_ struct transfer_args *ta)	1655	api_cede (pTHX)
1640	{	1656	{
1641	if (coro_nready)	1657	struct coro_transfer_args ta;
1642	{	1658
1643	SV *prev = SvRV (coro_current);
1644	prepare_schedule (aTHX_ ta);	1659	prepare_cede (aTHX_ &ta);
1645	api_ready (prev);	1660
		1661	if (expect_true (ta.prev != ta.next))
		1662	{
		1663	TRANSFER (ta, 1);
1646	return 1;	1664	return 1;
1647	}	1665	}
1648	else	1666	else
1649	return 0;	1667	return 0;
1650	}	1668	}
1651		1669
1652	static void
1653	api_schedule (void)
1654	{
1655	dTHX;
1656	struct transfer_args ta;
1657
1658	prepare_schedule (aTHX_ &ta);
1659	TRANSFER (ta, 1);
1660	}
1661
1662	static int	1670	static int
1663	api_cede (void)	1671	api_cede_notself (pTHX)
1664	{	1672	{
1665	dTHX;	1673	if (coro_nready)
		1674	{
1666	struct transfer_args ta;	1675	struct coro_transfer_args ta;
1667		1676
1668	prepare_cede (aTHX_ &ta);	1677	prepare_cede_notself (aTHX_ &ta);
1669
1670	if (expect_true (ta.prev != ta.next))
1671	{
1672	TRANSFER (ta, 1);	1678	TRANSFER (ta, 1);
1673	return 1;	1679	return 1;
1674	}	1680	}
1675	else	1681	else
1676	return 0;	1682	return 0;
1677	}	1683	}
1678		1684
1679	static int	1685	static void
1680	api_cede_notself (void)
1681	{
1682	dTHX;
1683	struct transfer_args ta;
1684
1685	if (prepare_cede_notself (aTHX_ &ta))
1686	{
1687	TRANSFER (ta, 1);
1688	return 1;
1689	}
1690	else
1691	return 0;
1692	}
1693
1694	static void
1695	api_trace (SV *coro_sv, int flags)	1686	api_trace (pTHX_ SV *coro_sv, int flags)
1696	{	1687	{
1697	dTHX;
1698	struct coro *coro = SvSTATE (coro_sv);	1688	struct coro *coro = SvSTATE (coro_sv);
1699		1689
1700	if (flags & CC_TRACE)	1690	if (flags & CC_TRACE)
1701	{	1691	{
1702	if (!coro->cctx)	1692	if (!coro->cctx)
1703	coro->cctx = cctx_new_run ();	1693	coro->cctx = cctx_new_run ();
1704	else if (!(coro->cctx->flags & CC_TRACE))	1694	else if (!(coro->cctx->flags & CC_TRACE))
1705	croak ("cannot enable tracing on coroutine with custom stack");	1695	croak ("cannot enable tracing on coroutine with custom stack,");
1706		1696
1707	coro->cctx->flags |= CC_NOREUSE | (flags & (CC_TRACE | CC_TRACE_ALL));	1697	coro->cctx->flags |= CC_NOREUSE | (flags & (CC_TRACE | CC_TRACE_ALL));
1708	}	1698	}
1709	else if (coro->cctx && coro->cctx->flags & CC_TRACE)	1699	else if (coro->cctx && coro->cctx->flags & CC_TRACE)
1710	{	1700	{
…		…
1715	else	1705	else
1716	coro->slot->runops = RUNOPS_DEFAULT;	1706	coro->slot->runops = RUNOPS_DEFAULT;
1717	}	1707	}
1718	}	1708	}
1719		1709
1720	#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
1721	static int	1740	static void
1722	coro_gensub_free (pTHX_ SV *sv, MAGIC *mg)	1741	slf_prepare_transfer (pTHX_ struct coro_transfer_args *ta)
1723	{	1742	{
1724	AV *padlist;	1743	SV **arg = (SV **)slf_frame.data;
1725	AV *av = (AV *)mg->mg_obj;
1726		1744
1727	abort ();	1745	prepare_transfer (aTHX_ ta, arg [0], arg [1]);
1728
1729	return 0;
1730	}	1746	}
1731		1747
1732	static MGVTBL coro_gensub_vtbl = {	1748	static void
1733	0, 0, 0, 0,	1749	slf_init_transfer (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
1734	coro_gensub_free	1750	{
1735	};	1751	if (items != 2)
1736	#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	/* return value handling - mostly like entersub */
		1845	/* make sure we put something on the stack in scalar context */
		1846	if (GIMME_V == G_SCALAR)
		1847	{
		1848	dSP;
		1849	SV **bot = PL_stack_base + checkmark;
		1850
		1851	if (sp == bot) /* too few, push undef */
		1852	bot [1] = &PL_sv_undef;
		1853	else if (sp != bot + 1) /* too many, take last one */
		1854	bot [1] = *sp;
		1855
		1856	SP = bot + 1;
		1857
		1858	PUTBACK;
		1859	}
		1860
		1861	/* exception handling */
		1862	if (expect_false (coro_throw))
		1863	{
		1864	SV *exception = sv_2mortal (coro_throw);
		1865
		1866	coro_throw = 0;
		1867	sv_setsv (ERRSV, exception);
		1868	croak (0);
		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	}
1737		1923
1738	/*****************************************************************************/	1924	/*****************************************************************************/
1739	/* PerlIO::cede */	1925	/* PerlIO::cede */
1740		1926
1741	typedef struct	1927	typedef struct
…		…
1769	PerlIOCede *self = PerlIOSelf (f, PerlIOCede);	1955	PerlIOCede *self = PerlIOSelf (f, PerlIOCede);
1770	double now = nvtime ();	1956	double now = nvtime ();
1771		1957
1772	if (now >= self->next)	1958	if (now >= self->next)
1773	{	1959	{
1774	api_cede ();	1960	api_cede (aTHX);
1775	self->next = now + self->every;	1961	self->next = now + self->every;
1776	}	1962	}
1777		1963
1778	return PerlIOBuf_flush (aTHX_ f);	1964	return PerlIOBuf_flush (aTHX_ f);
1779	}	1965	}
…		…
1809	PerlIOBuf_get_cnt,	1995	PerlIOBuf_get_cnt,
1810	PerlIOBuf_set_ptrcnt,	1996	PerlIOBuf_set_ptrcnt,
1811	};	1997	};
1812		1998
1813	/*****************************************************************************/	1999	/*****************************************************************************/
		2000	/* Coro::Semaphore */
1814		2001
1815	static const CV *ssl_cv; /* for quick consistency check */	2002	static void
		2003	coro_semaphore_adjust (pTHX_ AV *av, IV adjust)
		2004	{
		2005	SV *count_sv = AvARRAY (av)[0];
		2006	IV count = SvIVX (count_sv);
1816		2007
1817	static UNOP ssl_restore; /* restore stack as entersub did, for first-re-run */	2008	count += adjust;
1818	static SV *ssl_arg0;	2009	SvIVX (count_sv) = count;
1819	static SV *ssl_arg1;
1820		2010
1821	/* this restores the stack in the case we patched the entersub, to */	2011	/* now wake up as many waiters as are expected to lock */
1822	/* recreate the stack frame as perl will on following calls */	2012	while (count > 0 && AvFILLp (av) > 0)
1823	/* since entersub cleared the stack */	2013	{
		2014	SV *cb;
		2015
		2016	/* swap first two elements so we can shift a waiter */
		2017	AvARRAY (av)[0] = AvARRAY (av)[1];
		2018	AvARRAY (av)[1] = count_sv;
		2019	cb = av_shift (av);
		2020
		2021	if (SvOBJECT (cb))
		2022	api_ready (aTHX_ cb);
		2023	else
		2024	croak ("callbacks not yet supported");
		2025
		2026	SvREFCNT_dec (cb);
		2027
		2028	--count;
		2029	}
		2030	}
		2031
		2032	static void
		2033	coro_semaphore_on_destroy (pTHX_ struct coro *coro)
		2034	{
		2035	/* call $sem->adjust (0) to possibly wake up some other waiters */
		2036	coro_semaphore_adjust (aTHX_ (AV *)coro->slf_frame.data, 0);
		2037	}
		2038
		2039	static int
		2040	slf_check_semaphore_down (pTHX_ struct CoroSLF *frame)
		2041	{
		2042	AV *av = (AV *)frame->data;
		2043	SV *count_sv = AvARRAY (av)[0];
		2044
		2045	if (SvIVX (count_sv) > 0)
		2046	{
		2047	SvSTATE_current->on_destroy = 0;
		2048	SvIVX (count_sv) = SvIVX (count_sv) - 1;
		2049	return 0;
		2050	}
		2051	else
		2052	{
		2053	int i;
		2054	/* if we were woken up but can't down, we look through the whole */
		2055	/* waiters list and only add us if we aren't in there already */
		2056	/* this avoids some degenerate memory usage cases */
		2057
		2058	for (i = 1; i <= AvFILLp (av); ++i)
		2059	if (AvARRAY (av)[i] == SvRV (coro_current))
		2060	return 1;
		2061
		2062	av_push (av, SvREFCNT_inc (SvRV (coro_current)));
		2063	return 1;
		2064	}
		2065	}
		2066
		2067	static void
		2068	slf_init_semaphore_down (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		2069	{
		2070	AV *av = (AV *)SvRV (arg [0]);
		2071
		2072	if (SvIVX (AvARRAY (av)[0]) > 0)
		2073	{
		2074	frame->data = (void *)av;
		2075	frame->prepare = prepare_nop;
		2076	SvSTATE_current->on_destroy = coro_semaphore_on_destroy;
		2077	}
		2078	else
		2079	{
		2080	av_push (av, SvREFCNT_inc (SvRV (coro_current)));
		2081
		2082	frame->data = (void *)sv_2mortal (SvREFCNT_inc ((SV *)av));
		2083	frame->prepare = prepare_schedule;
		2084
		2085	/* to avoid race conditions when a woken-up coro gets terminated */
		2086	/* we arrange for a temporary on_destroy that calls adjust (0) */
		2087	assert (!SvSTATE_current->on_destroy);//D
		2088	SvSTATE_current->on_destroy = coro_semaphore_on_destroy;
		2089	}
		2090
		2091	frame->check = slf_check_semaphore_down;
		2092
		2093	}
		2094
		2095	/*****************************************************************************/
		2096	/* gensub: simple closure generation utility */
		2097
		2098	#define GENSUB_ARG CvXSUBANY (cv).any_ptr
		2099
		2100	/* create a closure from XS, returns a code reference */
		2101	/* the arg can be accessed via GENSUB_ARG from the callback */
		2102	/* the callback must use dXSARGS/XSRETURN */
1824	static OP *	2103	static SV *
1825	pp_restore (pTHX)	2104	gensub (pTHX_ void (*xsub)(pTHX_ CV *), void *arg)
1826	{	2105	{
		2106	CV *cv = (CV *)newSV (0);
		2107
		2108	sv_upgrade ((SV *)cv, SVt_PVCV);
		2109
		2110	CvANON_on (cv);
		2111	CvISXSUB_on (cv);
		2112	CvXSUB (cv) = xsub;
		2113	GENSUB_ARG = arg;
		2114
		2115	return newRV_noinc ((SV *)cv);
		2116	}
		2117
		2118	/*****************************************************************************/
		2119	/* Coro::AIO */
		2120
		2121	#define CORO_MAGIC_type_aio PERL_MAGIC_ext
		2122
		2123	/* helper storage struct */
		2124	struct io_state
		2125	{
		2126	int errorno;
		2127	I32 laststype; /* U16 in 5.10.0 */
		2128	int laststatval;
		2129	Stat_t statcache;
		2130	};
		2131
		2132	static void
		2133	coro_aio_callback (pTHX_ CV *cv)
		2134	{
		2135	dXSARGS;
		2136	AV *state = (AV *)GENSUB_ARG;
		2137	SV *coro = av_pop (state);
		2138	SV *data_sv = newSV (sizeof (struct io_state));
		2139
		2140	av_extend (state, items);
		2141
		2142	sv_upgrade (data_sv, SVt_PV);
		2143	SvCUR_set (data_sv, sizeof (struct io_state));
		2144	SvPOK_only (data_sv);
		2145
		2146	{
		2147	struct io_state *data = (struct io_state *)SvPVX (data_sv);
		2148
		2149	data->errorno = errno;
		2150	data->laststype = PL_laststype;
		2151	data->laststatval = PL_laststatval;
		2152	data->statcache = PL_statcache;
		2153	}
		2154
		2155	/* now build the result vector out of all the parameters and the data_sv */
		2156	{
		2157	int i;
		2158
		2159	for (i = 0; i < items; ++i)
		2160	av_push (state, SvREFCNT_inc_NN (ST (i)));
		2161	}
		2162
		2163	av_push (state, data_sv);
		2164
		2165	api_ready (aTHX_ coro);
		2166	SvREFCNT_dec (coro);
		2167	SvREFCNT_dec ((AV *)state);
		2168	}
		2169
		2170	static int
		2171	slf_check_aio_req (pTHX_ struct CoroSLF *frame)
		2172	{
		2173	AV *state = (AV *)frame->data;
		2174
		2175	/* one element that is an RV? repeat! */
		2176	if (AvFILLp (state) == 0 && SvROK (AvARRAY (state)[0]))
		2177	return 1;
		2178
		2179	/* restore status */
		2180	{
		2181	SV *data_sv = av_pop (state);
		2182	struct io_state *data = (struct io_state *)SvPVX (data_sv);
		2183
		2184	errno = data->errorno;
		2185	PL_laststype = data->laststype;
		2186	PL_laststatval = data->laststatval;
		2187	PL_statcache = data->statcache;
		2188
		2189	SvREFCNT_dec (data_sv);
		2190	}
		2191
		2192	/* push result values */
		2193	{
1827	dSP;	2194	dSP;
		2195	int i;
1828		2196
		2197	EXTEND (SP, AvFILLp (state) + 1);
		2198	for (i = 0; i <= AvFILLp (state); ++i)
		2199	PUSHs (sv_2mortal (SvREFCNT_inc_NN (AvARRAY (state)[i])));
		2200
		2201	PUTBACK;
		2202	}
		2203
		2204	return 0;
		2205	}
		2206
		2207	static void
		2208	slf_init_aio_req (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		2209	{
		2210	AV *state = (AV *)sv_2mortal ((SV *)newAV ());
		2211	SV *coro_hv = SvRV (coro_current);
		2212	struct coro *coro = SvSTATE_hv (coro_hv);
		2213
		2214	/* put our coroutine id on the state arg */
		2215	av_push (state, SvREFCNT_inc_NN (coro_hv));
		2216
		2217	/* first see whether we have a non-zero priority and set it as AIO prio */
		2218	if (coro->prio)
		2219	{
		2220	dSP;
		2221
		2222	static SV *prio_cv;
		2223	static SV *prio_sv;
		2224
		2225	if (expect_false (!prio_cv))
		2226	{
		2227	prio_cv = (SV *)get_cv ("IO::AIO::aioreq_pri", 0);
		2228	prio_sv = newSViv (0);
		2229	}
		2230
		2231	PUSHMARK (SP);
		2232	sv_setiv (prio_sv, coro->prio);
		2233	XPUSHs (prio_sv);
		2234
		2235	PUTBACK;
		2236	call_sv (prio_cv, G_VOID | G_DISCARD);
		2237	}
		2238
		2239	/* now call the original request */
		2240	{
		2241	dSP;
		2242	CV *req = (CV *)CORO_MAGIC_NN ((SV *)cv, CORO_MAGIC_type_aio)->mg_obj;
		2243	int i;
		2244
1829	PUSHMARK (SP);	2245	PUSHMARK (SP);
1830		2246
1831	EXTEND (SP, 3);	2247	/* first push all args to the stack */
1832	if (ssl_arg0) PUSHs (sv_2mortal (ssl_arg0)), ssl_arg0 = 0;	2248	EXTEND (SP, items + 1);
1833	if (ssl_arg1) PUSHs (sv_2mortal (ssl_arg1)), ssl_arg1 = 0;
1834	PUSHs ((SV *)CvGV (ssl_cv));
1835		2249
1836	RETURNOP (ssl_restore.op_first);	2250	for (i = 0; i < items; ++i)
1837	}	2251	PUSHs (arg [i]);
1838		2252
1839	/* declare prototype */	2253	/* now push the callback closure */
1840	XS(XS_Coro__State__set_stacklevel);	2254	PUSHs (sv_2mortal (gensub (aTHX_ coro_aio_callback, (void *)SvREFCNT_inc_NN ((SV *)state))));
1841		2255
1842	static OP *	2256	/* now call the AIO function - we assume our request is uncancelable */
1843	pp_set_stacklevel (pTHX)
1844	{
1845	dSP;
1846	struct transfer_args ta;
1847	SV **arg = PL_stack_base + TOPMARK + 1;
1848	int items = SP - arg; /* args without function object */
1849
1850	/* do a quick consistency check on the "function" object, and if it isn't */
1851	/* for us, divert to the real entersub */
1852	if (SvTYPE (*sp) != SVt_PVGV || CvXSUB (GvCV (*sp)) != XS_Coro__State__set_stacklevel)
1853	return PL_ppaddr[OP_ENTERSUB](aTHX);
1854
1855	/* pop args */
1856	SP = PL_stack_base + POPMARK;
1857
1858	if (!(PL_op->op_flags & OPf_STACKED))
1859	{
1860	/* ampersand-form of call, use @_ instead of stack */
1861	AV *av = GvAV (PL_defgv);
1862	arg = AvARRAY (av);
1863	items = AvFILLp (av) + 1;
1864	}
1865
1866	PUTBACK;	2257	PUTBACK;
1867	switch (PL_op->op_private & 7)	2258	call_sv ((SV *)req, G_VOID | G_DISCARD);
1868	{
1869	case 0:
1870	prepare_set_stacklevel (&ta, (struct coro_cctx *)SvIV (arg [0]));
1871	break;
1872
1873	case 1:
1874	if (items != 2)
1875	croak ("Coro::State::transfer (prev, next) expects two arguments, not %d.", items);
1876
1877	prepare_transfer (aTHX_ &ta, arg [0], arg [1]);
1878	break;
1879
1880	case 2:
1881	prepare_schedule (aTHX_ &ta);
1882	break;
1883
1884	case 3:
1885	prepare_cede (aTHX_ &ta);
1886	break;
1887
1888	case 4:
1889	if (!prepare_cede_notself (aTHX_ &ta))
1890	RETURN;
1891
1892	break;
1893	}	2259	}
1894		2260
1895	TRANSFER (ta, 0);	2261	/* now that the requets is going, we loop toll we have a result */
1896	SPAGAIN;	2262	frame->data = (void *)state;
1897		2263	frame->prepare = prepare_schedule;
1898	skip:	2264	frame->check = slf_check_aio_req;
1899
1900	RETURN;
1901	}	2265	}
		2266
		2267	static void
		2268	coro_aio_req_xs (pTHX_ CV *cv)
		2269	{
		2270	dXSARGS;
		2271
		2272	CORO_EXECUTE_SLF_XS (slf_init_aio_req);
		2273
		2274	XSRETURN_EMPTY;
		2275	}
		2276
		2277	/*****************************************************************************/
1902		2278
1903	MODULE = Coro::State PACKAGE = Coro::State PREFIX = api_	2279	MODULE = Coro::State PACKAGE = Coro::State PREFIX = api_
1904		2280
1905	PROTOTYPES: DISABLE	2281	PROTOTYPES: DISABLE
1906		2282
1907	# these not obviously related functions are all rolled into the same xs
1908	# function to increase chances that they all will call transfer with the same
1909	# stack offset
1910	void
1911	_set_stacklevel (...)
1912	ALIAS:
1913	Coro::State::transfer = 1
1914	Coro::schedule = 2
1915	Coro::cede = 3
1916	Coro::cede_notself = 4
1917	CODE:
1918	{
1919	assert (("FATAL: ssl call recursion in Coro module (please report)", PL_op->op_ppaddr != pp_set_stacklevel));
1920
1921	/* we patch the op, and then re-run the whole call */
1922	/* we have to put some dummy argument on the stack for this to work */
1923	/* TODO: walk back the opcode chain (but how?), nuke the pp_gv etc. */
1924	ssl_restore.op_next = (OP *)&ssl_restore;
1925	ssl_restore.op_type = OP_NULL;
1926	ssl_restore.op_ppaddr = pp_restore;
1927	ssl_restore.op_first = PL_op;
1928
1929	ssl_arg0 = items > 0 ? SvREFCNT_inc (ST (0)) : 0;
1930	ssl_arg1 = items > 1 ? SvREFCNT_inc (ST (1)) : 0;
1931
1932	PL_op->op_ppaddr = pp_set_stacklevel;
1933	PL_op->op_private = PL_op->op_private & ~7 | ix; /* we potentially share our private flags with entersub */
1934
1935	PL_op = (OP *)&ssl_restore;
1936	}
1937
1938	BOOT:	2283	BOOT:
1939	{	2284	{
1940	#ifdef USE_ITHREADS	2285	#ifdef USE_ITHREADS
1941	MUTEX_INIT (&coro_lock);
1942	# if CORO_PTHREAD	2286	# if CORO_PTHREAD
1943	coro_thx = PERL_GET_CONTEXT;	2287	coro_thx = PERL_GET_CONTEXT;
1944	# endif	2288	# endif
1945	#endif	2289	#endif
1946	BOOT_PAGESIZE;	2290	BOOT_PAGESIZE;
1947
1948	ssl_cv = get_cv ("Coro::State::_set_stacklevel", 0);
1949		2291
1950	irsgv = gv_fetchpv ("/" , GV_ADD|GV_NOTQUAL, SVt_PV);	2292	irsgv = gv_fetchpv ("/" , GV_ADD|GV_NOTQUAL, SVt_PV);
1951	stdoutgv = gv_fetchpv ("STDOUT", GV_ADD|GV_NOTQUAL, SVt_PVIO);	2293	stdoutgv = gv_fetchpv ("STDOUT", GV_ADD|GV_NOTQUAL, SVt_PVIO);
1952		2294
1953	orig_sigelem_get = PL_vtbl_sigelem.svt_get; PL_vtbl_sigelem.svt_get = coro_sigelem_get;	2295	orig_sigelem_get = PL_vtbl_sigelem.svt_get; PL_vtbl_sigelem.svt_get = coro_sigelem_get;
…		…
1969	main_top_env = PL_top_env;	2311	main_top_env = PL_top_env;
1970		2312
1971	while (main_top_env->je_prev)	2313	while (main_top_env->je_prev)
1972	main_top_env = main_top_env->je_prev;	2314	main_top_env = main_top_env->je_prev;
1973		2315
		2316	{
		2317	SV *slf = sv_2mortal (newSViv (PTR2IV (pp_slf)));
		2318
		2319	if (!PL_custom_op_names) PL_custom_op_names = newHV ();
		2320	hv_store_ent (PL_custom_op_names, slf,
		2321	newSVpv ("coro_slf", 0), 0);
		2322
		2323	if (!PL_custom_op_descs) PL_custom_op_descs = newHV ();
		2324	hv_store_ent (PL_custom_op_descs, slf,
		2325	newSVpv ("coro schedule like function", 0), 0);
		2326	}
		2327
1974	coroapi.ver = CORO_API_VERSION;	2328	coroapi.ver = CORO_API_VERSION;
1975	coroapi.rev = CORO_API_REVISION;	2329	coroapi.rev = CORO_API_REVISION;
		2330
1976	coroapi.transfer = api_transfer;	2331	coroapi.transfer = api_transfer;
		2332
		2333	coroapi.sv_state = SvSTATE_;
		2334	coroapi.execute_slf = api_execute_slf;
		2335	coroapi.prepare_nop = prepare_nop;
		2336	coroapi.prepare_schedule = prepare_schedule;
		2337	coroapi.prepare_cede = prepare_cede;
		2338	coroapi.prepare_cede_notself = prepare_cede_notself;
1977		2339
1978	{	2340	{
1979	SV **svp = hv_fetch (PL_modglobal, "Time::NVtime", 12, 0);	2341	SV **svp = hv_fetch (PL_modglobal, "Time::NVtime", 12, 0);
1980		2342
1981	if (!svp) croak ("Time::HiRes is required");	2343	if (!svp) croak ("Time::HiRes is required");
…		…
2013	for (i = 1; i < items; i++)	2375	for (i = 1; i < items; i++)
2014	av_push (coro->args, newSVsv (ST (i)));	2376	av_push (coro->args, newSVsv (ST (i)));
2015	}	2377	}
2016	OUTPUT:	2378	OUTPUT:
2017	RETVAL	2379	RETVAL
		2380
		2381	void
		2382	transfer (...)
		2383	PROTOTYPE: $$
		2384	CODE:
		2385	CORO_EXECUTE_SLF_XS (slf_init_transfer);
2018		2386
2019	bool	2387	bool
2020	_destroy (SV *coro_sv)	2388	_destroy (SV *coro_sv)
2021	CODE:	2389	CODE:
2022	RETVAL = coro_state_destroy (aTHX_ SvSTATE (coro_sv));	2390	RETVAL = coro_state_destroy (aTHX_ SvSTATE (coro_sv));
…		…
2029	CODE:	2397	CODE:
2030	_exit (code);	2398	_exit (code);
2031		2399
2032	int	2400	int
2033	cctx_stacksize (int new_stacksize = 0)	2401	cctx_stacksize (int new_stacksize = 0)
		2402	PROTOTYPE: ;$
2034	CODE:	2403	CODE:
2035	RETVAL = cctx_stacksize;	2404	RETVAL = cctx_stacksize;
2036	if (new_stacksize)	2405	if (new_stacksize)
2037	{	2406	{
2038	cctx_stacksize = new_stacksize;	2407	cctx_stacksize = new_stacksize;
…		…
2041	OUTPUT:	2410	OUTPUT:
2042	RETVAL	2411	RETVAL
2043		2412
2044	int	2413	int
2045	cctx_max_idle (int max_idle = 0)	2414	cctx_max_idle (int max_idle = 0)
		2415	PROTOTYPE: ;$
2046	CODE:	2416	CODE:
2047	RETVAL = cctx_max_idle;	2417	RETVAL = cctx_max_idle;
2048	if (max_idle > 1)	2418	if (max_idle > 1)
2049	cctx_max_idle = max_idle;	2419	cctx_max_idle = max_idle;
2050	OUTPUT:	2420	OUTPUT:
2051	RETVAL	2421	RETVAL
2052		2422
2053	int	2423	int
2054	cctx_count ()	2424	cctx_count ()
		2425	PROTOTYPE:
2055	CODE:	2426	CODE:
2056	RETVAL = cctx_count;	2427	RETVAL = cctx_count;
2057	OUTPUT:	2428	OUTPUT:
2058	RETVAL	2429	RETVAL
2059		2430
2060	int	2431	int
2061	cctx_idle ()	2432	cctx_idle ()
		2433	PROTOTYPE:
2062	CODE:	2434	CODE:
2063	RETVAL = cctx_idle;	2435	RETVAL = cctx_idle;
2064	OUTPUT:	2436	OUTPUT:
2065	RETVAL	2437	RETVAL
2066		2438
2067	void	2439	void
2068	list ()	2440	list ()
		2441	PROTOTYPE:
2069	PPCODE:	2442	PPCODE:
2070	{	2443	{
2071	struct coro *coro;	2444	struct coro *coro;
2072	for (coro = coro_first; coro; coro = coro->next)	2445	for (coro = coro_first; coro; coro = coro->next)
2073	if (coro->hv)	2446	if (coro->hv)
…		…
2135		2508
2136	void	2509	void
2137	throw (Coro::State self, SV *throw = &PL_sv_undef)	2510	throw (Coro::State self, SV *throw = &PL_sv_undef)
2138	PROTOTYPE: $;$	2511	PROTOTYPE: $;$
2139	CODE:	2512	CODE:
		2513	{
		2514	struct coro *current = SvSTATE_current;
		2515	SV **throwp = self == current ? &coro_throw : &self->throw;
2140	SvREFCNT_dec (self->throw);	2516	SvREFCNT_dec (*throwp);
2141	self->throw = SvOK (throw) ? newSVsv (throw) : 0;	2517	*throwp = SvOK (throw) ? newSVsv (throw) : 0;
		2518	}
2142		2519
2143	void	2520	void
2144	api_trace (SV *coro, int flags = CC_TRACE | CC_TRACE_SUB)	2521	api_trace (SV *coro, int flags = CC_TRACE | CC_TRACE_SUB)
		2522	PROTOTYPE: $;$
		2523	C_ARGS: aTHX_ coro, flags
2145		2524
2146	SV *	2525	SV *
2147	has_cctx (Coro::State coro)	2526	has_cctx (Coro::State coro)
2148	PROTOTYPE: $	2527	PROTOTYPE: $
2149	CODE:	2528	CODE:
…		…
2173	OUTPUT:	2552	OUTPUT:
2174	RETVAL	2553	RETVAL
2175		2554
2176	void	2555	void
2177	force_cctx ()	2556	force_cctx ()
		2557	PROTOTYPE:
2178	CODE:	2558	CODE:
2179	struct coro *coro = SvSTATE (coro_current);
2180	coro->cctx->idle_sp = 0;	2559	SvSTATE_current->cctx->idle_sp = 0;
2181		2560
2182	void	2561	void
2183	swap_defsv (Coro::State self)	2562	swap_defsv (Coro::State self)
2184	PROTOTYPE: $	2563	PROTOTYPE: $
2185	ALIAS:	2564	ALIAS:
2186	swap_defav = 1	2565	swap_defav = 1
2187	CODE:	2566	CODE:
2188	if (!self->slot)	2567	if (!self->slot)
2189	croak ("cannot swap state with coroutine that has no saved state");	2568	croak ("cannot swap state with coroutine that has no saved state,");
2190	else	2569	else
2191	{	2570	{
2192	SV **src = ix ? (SV **)&GvAV (PL_defgv) : &GvSV (PL_defgv);	2571	SV **src = ix ? (SV **)&GvAV (PL_defgv) : &GvSV (PL_defgv);
2193	SV **dst = ix ? (SV **)&self->slot->defav : (SV **)&self->slot->defsv;	2572	SV **dst = ix ? (SV **)&self->slot->defav : (SV **)&self->slot->defsv;
2194		2573
2195	SV *tmp = *src; *src = *dst; *dst = tmp;	2574	SV *tmp = *src; *src = *dst; *dst = tmp;
2196	}	2575	}
		2576
2197		2577
2198	MODULE = Coro::State PACKAGE = Coro	2578	MODULE = Coro::State PACKAGE = Coro
2199		2579
2200	BOOT:	2580	BOOT:
2201	{	2581	{
…		…
2226	coroapi.schedule = api_schedule;	2606	coroapi.schedule = api_schedule;
2227	coroapi.cede = api_cede;	2607	coroapi.cede = api_cede;
2228	coroapi.cede_notself = api_cede_notself;	2608	coroapi.cede_notself = api_cede_notself;
2229	coroapi.ready = api_ready;	2609	coroapi.ready = api_ready;
2230	coroapi.is_ready = api_is_ready;	2610	coroapi.is_ready = api_is_ready;
2231	coroapi.nready = &coro_nready;	2611	coroapi.nready = coro_nready;
2232	coroapi.current = coro_current;	2612	coroapi.current = coro_current;
2233		2613
2234	GCoroAPI = &coroapi;	2614	GCoroAPI = &coroapi;
2235	sv_setiv (sv, (IV)&coroapi);	2615	sv_setiv (sv, (IV)&coroapi);
2236	SvREADONLY_on (sv);	2616	SvREADONLY_on (sv);
2237	}	2617	}
2238	}	2618	}
		2619
		2620	void
		2621	schedule (...)
		2622	CODE:
		2623	CORO_EXECUTE_SLF_XS (slf_init_schedule);
		2624
		2625	void
		2626	cede (...)
		2627	CODE:
		2628	CORO_EXECUTE_SLF_XS (slf_init_cede);
		2629
		2630	void
		2631	cede_notself (...)
		2632	CODE:
		2633	CORO_EXECUTE_SLF_XS (slf_init_cede_notself);
2239		2634
2240	void	2635	void
2241	_set_current (SV *current)	2636	_set_current (SV *current)
2242	PROTOTYPE: $	2637	PROTOTYPE: $
2243	CODE:	2638	CODE:
…		…
2246		2641
2247	void	2642	void
2248	_set_readyhook (SV *hook)	2643	_set_readyhook (SV *hook)
2249	PROTOTYPE: $	2644	PROTOTYPE: $
2250	CODE:	2645	CODE:
2251	LOCK;
2252	SvREFCNT_dec (coro_readyhook);	2646	SvREFCNT_dec (coro_readyhook);
2253	coro_readyhook = SvOK (hook) ? newSVsv (hook) : 0;	2647	coro_readyhook = SvOK (hook) ? newSVsv (hook) : 0;
2254	UNLOCK;
2255		2648
2256	int	2649	int
2257	prio (Coro::State coro, int newprio = 0)	2650	prio (Coro::State coro, int newprio = 0)
		2651	PROTOTYPE: $;$
2258	ALIAS:	2652	ALIAS:
2259	nice = 1	2653	nice = 1
2260	CODE:	2654	CODE:
2261	{	2655	{
2262	RETVAL = coro->prio;	2656	RETVAL = coro->prio;
…		…
2277		2671
2278	SV *	2672	SV *
2279	ready (SV *self)	2673	ready (SV *self)
2280	PROTOTYPE: $	2674	PROTOTYPE: $
2281	CODE:	2675	CODE:
2282	RETVAL = boolSV (api_ready (self));	2676	RETVAL = boolSV (api_ready (aTHX_ self));
2283	OUTPUT:	2677	OUTPUT:
2284	RETVAL	2678	RETVAL
2285		2679
2286	int	2680	int
2287	nready (...)	2681	nready (...)
…		…
2294	# for async_pool speedup	2688	# for async_pool speedup
2295	void	2689	void
2296	_pool_1 (SV *cb)	2690	_pool_1 (SV *cb)
2297	CODE:	2691	CODE:
2298	{	2692	{
2299	struct coro *coro = SvSTATE (coro_current);
2300	HV *hv = (HV *)SvRV (coro_current);	2693	HV *hv = (HV *)SvRV (coro_current);
		2694	struct coro *coro = SvSTATE_hv ((SV *)hv);
2301	AV *defav = GvAV (PL_defgv);	2695	AV *defav = GvAV (PL_defgv);
2302	SV *invoke = hv_delete (hv, "_invoke", sizeof ("_invoke") - 1, 0);	2696	SV *invoke = hv_delete (hv, "_invoke", sizeof ("_invoke") - 1, 0);
2303	AV *invoke_av;	2697	AV *invoke_av;
2304	int i, len;	2698	int i, len;
2305		2699
…		…
2326	{	2720	{
2327	av_fill (defav, len - 1);	2721	av_fill (defav, len - 1);
2328	for (i = 0; i < len; ++i)	2722	for (i = 0; i < len; ++i)
2329	av_store (defav, i, SvREFCNT_inc_NN (AvARRAY (invoke_av)[i + 1]));	2723	av_store (defav, i, SvREFCNT_inc_NN (AvARRAY (invoke_av)[i + 1]));
2330	}	2724	}
2331
2332	SvREFCNT_dec (invoke);
2333	}	2725	}
2334		2726
2335	void	2727	void
2336	_pool_2 (SV *cb)	2728	_pool_2 (SV *cb)
2337	CODE:	2729	CODE:
2338	{	2730	{
2339	struct coro *coro = SvSTATE (coro_current);	2731	HV *hv = (HV *)SvRV (coro_current);
		2732	struct coro *coro = SvSTATE_hv ((SV *)hv);
2340		2733
2341	sv_setsv (cb, &PL_sv_undef);	2734	sv_setsv (cb, &PL_sv_undef);
2342		2735
2343	SvREFCNT_dec ((SV *)PL_defoutgv); PL_defoutgv = (GV *)coro->saved_deffh;	2736	SvREFCNT_dec ((SV *)PL_defoutgv); PL_defoutgv = (GV *)coro->saved_deffh;
2344	coro->saved_deffh = 0;	2737	coro->saved_deffh = 0;
…		…
2351	SvREFCNT_dec (old);	2744	SvREFCNT_dec (old);
2352	croak ("\3async_pool terminate\2\n");	2745	croak ("\3async_pool terminate\2\n");
2353	}	2746	}
2354		2747
2355	av_clear (GvAV (PL_defgv));	2748	av_clear (GvAV (PL_defgv));
2356	hv_store ((HV *)SvRV (coro_current), "desc", sizeof ("desc") - 1,	2749	hv_store (hv, "desc", sizeof ("desc") - 1,
2357	newSVpvn ("[async_pool idle]", sizeof ("[async_pool idle]") - 1), 0);	2750	newSVpvn ("[async_pool idle]", sizeof ("[async_pool idle]") - 1), 0);
2358		2751
2359	coro->prio = 0;	2752	coro->prio = 0;
2360		2753
2361	if (coro->cctx && (coro->cctx->flags & CC_TRACE))	2754	if (coro->cctx && (coro->cctx->flags & CC_TRACE))
2362	api_trace (coro_current, 0);	2755	api_trace (aTHX_ coro_current, 0);
2363		2756
2364	av_push (av_async_pool, newSVsv (coro_current));	2757	av_push (av_async_pool, newSVsv (coro_current));
2365	}	2758	}
2366		2759
2367	#if 0
2368		2760
2369	void	2761	MODULE = Coro::State PACKAGE = PerlIO::cede
2370	_generator_call (...)	2762
2371	PROTOTYPE: @	2763	BOOT:
2372	PPCODE:	2764	PerlIO_define_layer (aTHX_ &PerlIO_cede);
2373	fprintf (stderr, "call %p\n", CvXSUBANY(cv).any_ptr);	2765
2374	xxxx	2766
2375	abort ();	2767	MODULE = Coro::State PACKAGE = Coro::Semaphore
2376		2768
2377	SV *	2769	SV *
2378	gensub (SV *sub, ...)	2770	new (SV *klass, SV *count_ = 0)
2379	PROTOTYPE: &;@	2771	CODE:
2380	CODE:
2381	{	2772	{
2382	struct coro *coro;	2773	/* a semaphore contains a counter IV in $sem->[0] and any waiters after that */
2383	MAGIC *mg;	2774	AV *av = newAV ();
2384	CV *xcv;	2775	SV **ary;
2385	CV *ncv = (CV *)newSV_type (SVt_PVCV);
2386	int i;
2387		2776
2388	CvGV (ncv) = CvGV (cv);	2777	/* unfortunately, building manually saves memory */
2389	CvFILE (ncv) = CvFILE (cv);	2778	Newx (ary, 2, SV *);
		2779	AvALLOC (av) = ary;
		2780	AvARRAY (av) = ary;
		2781	AvMAX (av) = 1;
		2782	AvFILLp (av) = 0;
		2783	ary [0] = newSViv (count_ && SvOK (count_) ? SvIV (count_) : 1);
2390		2784
2391	Newz (0, coro, 1, struct coro);	2785	RETVAL = sv_bless (newRV_noinc ((SV *)av), GvSTASH (CvGV (cv)));
2392	coro->args = newAV ();
2393	coro->flags = CF_NEW;
2394
2395	av_extend (coro->args, items - 1);
2396	for (i = 1; i < items; i++)
2397	av_push (coro->args, newSVsv (ST (i)));
2398
2399	CvISXSUB_on (ncv);
2400	CvXSUBANY (ncv).any_ptr = (void *)coro;
2401
2402	xcv = GvCV (gv_fetchpv ("Coro::_generator_call", 0, SVt_PVCV));
2403
2404	CvXSUB (ncv) = CvXSUB (xcv);
2405	CvANON_on (ncv);
2406
2407	mg = sv_magicext ((SV *)ncv, 0, CORO_MAGIC_type_state, &coro_gensub_vtbl, (char *)coro, 0);
2408	RETVAL = newRV_noinc ((SV *)ncv);
2409	}	2786	}
2410	OUTPUT:	2787	OUTPUT:
2411	RETVAL	2788	RETVAL
2412		2789
2413	#endif	2790	SV *
2414		2791	count (SV *self)
2415		2792	CODE:
2416	MODULE = Coro::State PACKAGE = Coro::AIO	2793	RETVAL = newSVsv (AvARRAY ((AV *)SvRV (self))[0]);
		2794	OUTPUT:
		2795	RETVAL
2417		2796
2418	void	2797	void
2419	_get_state (SV *self)	2798	up (SV *self, int adjust = 1)
2420	PPCODE:	2799	ALIAS:
2421	{	2800	adjust = 1
2422	AV *defav = GvAV (PL_defgv);	2801	CODE:
2423	AV *av = newAV ();	2802	coro_semaphore_adjust (aTHX_ (AV *)SvRV (self), ix ? adjust : 1);
2424	int i;
2425	SV *data_sv = newSV (sizeof (struct io_state));
2426	struct io_state *data = (struct io_state *)SvPVX (data_sv);
2427	SvCUR_set (data_sv, sizeof (struct io_state));
2428	SvPOK_only (data_sv);
2429
2430	data->errorno = errno;
2431	data->laststype = PL_laststype;
2432	data->laststatval = PL_laststatval;
2433	data->statcache = PL_statcache;
2434
2435	av_extend (av, AvFILLp (defav) + 1 + 1);
2436
2437	for (i = 0; i <= AvFILLp (defav); ++i)
2438	av_push (av, SvREFCNT_inc_NN (AvARRAY (defav)[i]));
2439
2440	av_push (av, data_sv);
2441
2442	XPUSHs (sv_2mortal (newRV_noinc ((SV *)av)));
2443
2444	api_ready (self);
2445	}
2446		2803
2447	void	2804	void
2448	_set_state (SV *state)	2805	down (SV *self)
2449	PROTOTYPE: $	2806	CODE:
2450	PPCODE:	2807	CORO_EXECUTE_SLF_XS (slf_init_semaphore_down);
		2808
		2809	void
		2810	try (SV *self)
		2811	PPCODE:
2451	{	2812	{
2452	AV *av = (AV *)SvRV (state);	2813	AV *av = (AV *)SvRV (self);
2453	struct io_state *data = (struct io_state *)SvPVX (AvARRAY (av)[AvFILLp (av)]);	2814	SV *count_sv = AvARRAY (av)[0];
		2815	IV count = SvIVX (count_sv);
		2816
		2817	if (count > 0)
		2818	{
		2819	--count;
		2820	SvIVX (count_sv) = count;
		2821	XSRETURN_YES;
		2822	}
		2823	else
		2824	XSRETURN_NO;
		2825	}
		2826
		2827	void
		2828	waiters (SV *self)
		2829	CODE:
		2830	{
		2831	AV *av = (AV *)SvRV (self);
		2832
		2833	if (GIMME_V == G_SCALAR)
		2834	XPUSHs (sv_2mortal (newSVsv (AvARRAY (av)[0])));
		2835	else
		2836	{
2454	int i;	2837	int i;
2455
2456	errno = data->errorno;
2457	PL_laststype = data->laststype;
2458	PL_laststatval = data->laststatval;
2459	PL_statcache = data->statcache;
2460
2461	EXTEND (SP, AvFILLp (av));	2838	EXTEND (SP, AvFILLp (av) + 1 - 1);
2462	for (i = 0; i < AvFILLp (av); ++i)	2839	for (i = 1; i <= AvFILLp (av); ++i)
2463	PUSHs (sv_2mortal (SvREFCNT_inc_NN (AvARRAY (av)[i])));	2840	PUSHs (sv_2mortal (newRV_inc (AvARRAY (av)[i])));
		2841	}
2464	}	2842	}
2465		2843
2466		2844
2467	MODULE = Coro::State PACKAGE = Coro::AnyEvent	2845	MODULE = Coro::State PACKAGE = Coro::AnyEvent
2468		2846
2469	BOOT:	2847	BOOT:
2470	sv_activity = coro_get_sv (aTHX_ "Coro::AnyEvent::ACTIVITY", TRUE);	2848	sv_activity = coro_get_sv (aTHX_ "Coro::AnyEvent::ACTIVITY", TRUE);
2471		2849
2472	SV *	2850	void
2473	_schedule (...)	2851	_schedule (...)
2474	PROTOTYPE: @
2475	CODE:	2852	CODE:
2476	{	2853	{
2477	static int incede;	2854	static int incede;
2478		2855
2479	api_cede_notself ();	2856	api_cede_notself (aTHX);
2480		2857
2481	++incede;	2858	++incede;
2482	while (coro_nready >= incede && api_cede ())	2859	while (coro_nready >= incede && api_cede (aTHX))
2483	;	2860	;
2484		2861
2485	sv_setsv (sv_activity, &PL_sv_undef);	2862	sv_setsv (sv_activity, &PL_sv_undef);
2486	if (coro_nready >= incede)	2863	if (coro_nready >= incede)
2487	{	2864	{
2488	PUSHMARK (SP);	2865	PUSHMARK (SP);
2489	PUTBACK;	2866	PUTBACK;
2490	call_pv ("Coro::AnyEvent::_activity", G_DISCARD | G_EVAL);	2867	call_pv ("Coro::AnyEvent::_activity", G_KEEPERR | G_EVAL | G_VOID | G_DISCARD);
2491	SPAGAIN;
2492	}	2868	}
2493		2869
2494	--incede;	2870	--incede;
2495	}	2871	}
2496		2872
2497		2873
2498	MODULE = Coro::State PACKAGE = PerlIO::cede	2874	MODULE = Coro::State PACKAGE = Coro::AIO
2499		2875
2500	BOOT:	2876	void
2501	PerlIO_define_layer (aTHX_ &PerlIO_cede);	2877	_register (char *target, char *proto, SV *req)
		2878	CODE:
		2879	{
		2880	HV *st;
		2881	GV *gvp;
		2882	CV *req_cv = sv_2cv (req, &st, &gvp, 0);
		2883	/* newXSproto doesn't return the CV on 5.8 */
		2884	CV *slf_cv = newXS (target, coro_aio_req_xs, __FILE__);
		2885	sv_setpv ((SV *)slf_cv, proto);
		2886	sv_magicext ((SV *)slf_cv, (SV *)req_cv, CORO_MAGIC_type_aio, 0, 0, 0);
		2887	}
2502		2888

Diff Legend

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