ViewVC Help

View File | Revision Log | Show Annotations | Download File

/cvs/Coro/Coro/State.xs

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

Diff Legend

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