ViewVC Help

View File | Revision Log | Show Annotations | Download File

/cvs/Coro/Coro/State.xs

Comparing Coro/Coro/State.xs (file contents):
Revision 1.270 by root, Fri Nov 14 07:22:11 2008 UTC vs.
Revision 1.291 by root, Tue Nov 18 05:55:04 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
…		…
142	#define NOINLINE attribute ((noinline))	145	#define NOINLINE attribute ((noinline))
143		146
144	#include "CoroAPI.h"	147	#include "CoroAPI.h"
145		148
146	#ifdef USE_ITHREADS	149	#ifdef USE_ITHREADS
147
148	static perl_mutex coro_lock;
149	# define LOCK do { MUTEX_LOCK (&coro_lock); } while (0)
150	# define UNLOCK do { MUTEX_UNLOCK (&coro_lock); } while (0)
151	# if CORO_PTHREAD	150	# if CORO_PTHREAD
152	static void *coro_thx;	151	static void *coro_thx;
153	# endif	152	# endif
154
155	#else
156
157	# define LOCK (void)0
158	# define UNLOCK (void)0
159
160	#endif	153	#endif
161
162	# undef LOCK
163	# define LOCK (void)0
164	# undef UNLOCK
165	# define UNLOCK (void)0
166
167	/* helper storage struct for Coro::AIO */
168	struct io_state
169	{
170	AV *res;
171	int errorno;
172	I32 laststype; /* U16 in 5.10.0 */
173	int laststatval;
174	Stat_t statcache;
175	};
176		154
177	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);
178		160
179	static U32 cctx_gen;	161	static U32 cctx_gen;
180	static size_t cctx_stacksize = CORO_STACKSIZE;	162	static size_t cctx_stacksize = CORO_STACKSIZE;
181	static struct CoroAPI coroapi;	163	static struct CoroAPI coroapi;
182	static AV *main_mainstack; /* used to differentiate between $main and others */	164	static AV *main_mainstack; /* used to differentiate between $main and others */
183	static JMPENV *main_top_env;	165	static JMPENV *main_top_env;
184	static HV *coro_state_stash, *coro_stash;	166	static HV *coro_state_stash, *coro_stash;
185	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 */
186	static volatile struct coro *transfer_next;
187
188	struct transfer_args
189	{
190	struct coro *prev, *next;
191	};
192		168
193	static GV *irsgv; /* $/ */	169	static GV *irsgv; /* $/ */
194	static GV *stdoutgv; /* *STDOUT */	170	static GV *stdoutgv; /* *STDOUT */
195	static SV *rv_diehook;	171	static SV *rv_diehook;
196	static SV *rv_warnhook;	172	static SV *rv_warnhook;
…		…
215	CC_TRACE_LINE = 0x10, /* trace each statement */	191	CC_TRACE_LINE = 0x10, /* trace each statement */
216	CC_TRACE_ALL = CC_TRACE_SUB | CC_TRACE_LINE,	192	CC_TRACE_ALL = CC_TRACE_SUB | CC_TRACE_LINE,
217	};	193	};
218		194
219	/* this is a structure representing a c-level coroutine */	195	/* this is a structure representing a c-level coroutine */
220	typedef struct coro_cctx {	196	typedef struct coro_cctx
		197	{
221	struct coro_cctx *next;	198	struct coro_cctx *next;
222		199
223	/* the stack */	200	/* the stack */
224	void *sptr;	201	void *sptr;
225	size_t ssize;	202	size_t ssize;
…		…
243	CF_NEW = 0x0004, /* has never been switched to */	220	CF_NEW = 0x0004, /* has never been switched to */
244	CF_DESTROYED = 0x0008, /* coroutine data has been freed */	221	CF_DESTROYED = 0x0008, /* coroutine data has been freed */
245	};	222	};
246		223
247	/* 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 */
248	typedef struct {	225	typedef struct
		226	{
249	SV *defsv;	227	SV *defsv;
250	AV *defav;	228	AV *defav;
251	SV *errsv;	229	SV *errsv;
252	SV *irsgv;	230	SV *irsgv;
253	#define VAR(name,type) type name;	231	#define VAR(name,type) type name;
…		…
257		235
258	#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))
259		237
260	/* this is a structure representing a perl-level coroutine */	238	/* this is a structure representing a perl-level coroutine */
261	struct coro {	239	struct coro {
262	/* the c coroutine allocated to this perl coroutine, if any */	240	/* the C coroutine allocated to this perl coroutine, if any */
263	coro_cctx *cctx;	241	coro_cctx *cctx;
264		242
265	/* process data */	243	/* state data */
		244	struct CoroSLF slf_frame; /* saved slf frame */
266	AV *mainstack;	245	AV *mainstack;
267	perl_slots *slot; /* basically the saved sp */	246	perl_slots *slot; /* basically the saved sp */
268		247
269	AV *args; /* data associated with this coroutine (initial args) */	248	AV *args; /* data associated with this coroutine (initial args) */
270	int refcnt; /* coroutines are refcounted, yes */	249	int refcnt; /* coroutines are refcounted, yes */
271	int flags; /* CF_ flags */	250	int flags; /* CF_ flags */
272	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);
273		253
274	/* statistics */	254	/* statistics */
275	int usecount; /* number of transfers to this coro */	255	int usecount; /* number of transfers to this coro */
276		256
277	/* coro process data */	257	/* coro process data */
…		…
285	struct coro *next, *prev;	265	struct coro *next, *prev;
286	};	266	};
287		267
288	typedef struct coro *Coro__State;	268	typedef struct coro *Coro__State;
289	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;
290		276
291	/** Coro ********************************************************************/	277	/** Coro ********************************************************************/
292		278
293	#define PRIO_MAX 3	279	#define PRIO_MAX 3
294	#define PRIO_HIGH 1	280	#define PRIO_HIGH 1
…		…
395	SvREFCNT_dec (av); /* sv_magicext increased the refcount */	381	SvREFCNT_dec (av); /* sv_magicext increased the refcount */
396		382
397	return 0;	383	return 0;
398	}	384	}
399		385
400	#define CORO_MAGIC_type_cv PERL_MAGIC_ext	386	#define CORO_MAGIC_type_cv 26
401	#define CORO_MAGIC_type_state PERL_MAGIC_ext	387	#define CORO_MAGIC_type_state PERL_MAGIC_ext
402		388
403	static MGVTBL coro_cv_vtbl = {	389	static MGVTBL coro_cv_vtbl = {
404	0, 0, 0, 0,	390	0, 0, 0, 0,
405	coro_cv_free	391	coro_cv_free
406	};	392	};
407		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
408	#define CORO_MAGIC(sv, type) \	399	#define CORO_MAGIC(sv, type) \
409	SvMAGIC (sv) \	400	(expect_true (SvMAGIC (sv)) \
410	? SvMAGIC (sv)->mg_type == type \	401	? CORO_MAGIC_NN (sv, type) \
411	? SvMAGIC (sv) \
412	: mg_find (sv, type) \
413	: 0	402	: 0)
414		403
415	#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)
416	#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)
417		406
418	INLINE struct coro *	407	INLINE struct coro *
419	SvSTATE_ (pTHX_ SV *coro)	408	SvSTATE_ (pTHX_ SV *coro)
420	{	409	{
421	HV *stash;	410	HV *stash;
…		…
438	mg = CORO_MAGIC_state (coro);	427	mg = CORO_MAGIC_state (coro);
439	return (struct coro *)mg->mg_ptr;	428	return (struct coro *)mg->mg_ptr;
440	}	429	}
441		430
442	#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))
443		436
444	/* the next two functions merely cache the padlists */	437	/* the next two functions merely cache the padlists */
445	static void	438	static void
446	get_padlist (pTHX_ CV *cv)	439	get_padlist (pTHX_ CV *cv)
447	{	440	{
…		…
514	CvPADLIST (cv) = (AV *)POPs;	507	CvPADLIST (cv) = (AV *)POPs;
515	}	508	}
516		509
517	PUTBACK;	510	PUTBACK;
518	}	511	}
		512
		513	slf_frame = c->slf_frame;
		514	coro_throw = c->throw;
519	}	515	}
520		516
521	static void	517	static void
522	save_perl (pTHX_ Coro__State c)	518	save_perl (pTHX_ Coro__State c)
523	{	519	{
		520	c->throw = coro_throw;
		521	c->slf_frame = slf_frame;
		522
524	{	523	{
525	dSP;	524	dSP;
526	I32 cxix = cxstack_ix;	525	I32 cxix = cxstack_ix;
527	PERL_CONTEXT *ccstk = cxstack;	526	PERL_CONTEXT *ccstk = cxstack;
528	PERL_SI *top_si = PL_curstackinfo;	527	PERL_SI *top_si = PL_curstackinfo;
…		…
595	#undef VAR	594	#undef VAR
596	}	595	}
597	}	596	}
598		597
599	/*	598	/*
600	* allocate various perl stacks. This is an exact copy	599	* allocate various perl stacks. This is almost an exact copy
601	* of perl.c:init_stacks, except that it uses less memory	600	* of perl.c:init_stacks, except that it uses less memory
602	* on the (sometimes correct) assumption that coroutines do	601	* on the (sometimes correct) assumption that coroutines do
603	* not usually need a lot of stackspace.	602	* not usually need a lot of stackspace.
604	*/	603	*/
605	#if CORO_PREFER_PERL_FUNCTIONS	604	#if CORO_PREFER_PERL_FUNCTIONS
…		…
712	#endif	711	#endif
713	}	712	}
714	}	713	}
715		714
716	return rss;	715	return rss;
717	}
718
719	/** set stacklevel support **************************************************/
720
721	/* we sometimes need to create the effect of pp_slf calling us */
722	#define SLF_HEAD (void)0
723	/* we sometimes need to create the effect of leaving via pp_slf */
724	#define SLF_TAIL slf_tail (aTHX)
725
726	INLINE void
727	slf_tail (pTHX)
728	{
729	dSP;
730	SV **bot = SP;
731
732	int gimme = GIMME_V;
733
734	/* make sure we put something on the stack in scalar context */
735	if (gimme == G_SCALAR)
736	{
737	if (sp == bot)
738	XPUSHs (&PL_sv_undef);
739
740	SP = bot + 1;
741	}
742
743	PUTBACK;
744	}	716	}
745		717
746	/** coroutine stack handling ************************************************/	718	/** coroutine stack handling ************************************************/
747		719
748	static int (*orig_sigelem_get) (pTHX_ SV *sv, MAGIC *mg);	720	static int (*orig_sigelem_get) (pTHX_ SV *sv, MAGIC *mg);
…		…
834		806
835	return orig_sigelem_set ? orig_sigelem_set (aTHX_ sv, mg) : 0;	807	return orig_sigelem_set ? orig_sigelem_set (aTHX_ sv, mg) : 0;
836	}	808	}
837		809
838	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 */
839	coro_setup (pTHX_ struct coro *coro)	826	coro_setup (pTHX_ struct coro *coro)
840	{	827	{
841	/*	828	/*
842	* emulate part of the perl startup here.	829	* emulate part of the perl startup here.
843	*/	830	*/
…		…
882	PL_op = PL_ppaddr[OP_ENTERSUB](aTHX);	869	PL_op = PL_ppaddr[OP_ENTERSUB](aTHX);
883	SPAGAIN;	870	SPAGAIN;
884	}	871	}
885		872
886	/* 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
887	* likely was suspended in set_stacklevel, called from pp_set_stacklevel,	874	* likely was suspended in pp_slf, so we have to emulate entering pp_slf here.
888	* so we have to emulate entering pp_set_stacklevel here.
889	*/	875	*/
890	SLF_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;
891	}	888	}
892		889
893	static void	890	static void
894	coro_destruct (pTHX_ struct coro *coro)	891	coro_destruct (pTHX_ struct coro *coro)
895	{	892	{
…		…
919		916
920	SvREFCNT_dec (PL_diehook);	917	SvREFCNT_dec (PL_diehook);
921	SvREFCNT_dec (PL_warnhook);	918	SvREFCNT_dec (PL_warnhook);
922		919
923	SvREFCNT_dec (coro->saved_deffh);	920	SvREFCNT_dec (coro->saved_deffh);
924	SvREFCNT_dec (coro->throw);	921	SvREFCNT_dec (coro_throw);
925		922
926	coro_destruct_stacks (aTHX);	923	coro_destruct_stacks (aTHX);
927	}	924	}
928		925
929	INLINE void	926	INLINE void
…		…
939	static int	936	static int
940	runops_trace (pTHX)	937	runops_trace (pTHX)
941	{	938	{
942	COP *oldcop = 0;	939	COP *oldcop = 0;
943	int oldcxix = -2;	940	int oldcxix = -2;
944	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 */
945	coro_cctx *cctx = coro->cctx;	942	coro_cctx *cctx = coro->cctx;
946		943
947	while ((PL_op = CALL_FPTR (PL_op->op_ppaddr) (aTHX)))	944	while ((PL_op = CALL_FPTR (PL_op->op_ppaddr) (aTHX)))
948	{	945	{
949	PERL_ASYNC_CHECK ();	946	PERL_ASYNC_CHECK ();
…		…
1058		1055
1059	TAINT_NOT;	1056	TAINT_NOT;
1060	return 0;	1057	return 0;
1061	}	1058	}
1062		1059
		1060	static struct coro_cctx *cctx_ssl_cctx;
		1061	static struct CoroSLF cctx_ssl_frame;
		1062
1063	static void	1063	static void
1064	prepare_set_stacklevel (struct transfer_args *ta, struct coro_cctx *cctx)	1064	slf_prepare_set_stacklevel (pTHX_ struct coro_transfer_args *ta)
1065	{	1065	{
1066	ta->prev = (struct coro *)cctx;	1066	ta->prev = (struct coro *)cctx_ssl_cctx;
1067	ta->next = 0;	1067	ta->next = 0;
1068	}	1068	}
1069		1069
1070	/* inject a fake call to Coro::State::_cctx_init into the execution */	1070	static int
1071	/* _cctx_init should be careful, as it could be called at almost any time */	1071	slf_check_set_stacklevel (pTHX_ struct CoroSLF *frame)
1072	/* during execution of a perl program */	1072	{
1073	/* also initialises PL_top_env */	1073	*frame = cctx_ssl_frame;
		1074
		1075	return 1;
		1076	}
		1077
		1078	/* initialises PL_top_env and injects a pseudo-slf-call to sett he stacklevel */
1074	static void NOINLINE	1079	static void NOINLINE
1075	cctx_prepare (pTHX_ coro_cctx *cctx)	1080	cctx_prepare (pTHX_ coro_cctx *cctx)
1076	{	1081	{
1077	dSP;
1078	UNOP myop;
1079
1080	PL_top_env = &PL_start_env;	1082	PL_top_env = &PL_start_env;
1081		1083
1082	if (cctx->flags & CC_TRACE)	1084	if (cctx->flags & CC_TRACE)
1083	PL_runops = runops_trace;	1085	PL_runops = runops_trace;
1084		1086
1085	Zero (&myop, 1, UNOP);	1087	/* we already must be in an SLF call, there is no other valid way
1086	myop.op_next = PL_op;	1088	* that can lead to creation of a new cctx */
1087	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));
1088		1091
1089	PUSHMARK (SP);	1092	cctx_ssl_cctx = cctx;
1090	EXTEND (SP, 2);	1093	cctx_ssl_frame = slf_frame;
1091	PUSHs (sv_2mortal (newSViv ((IV)cctx)));	1094
1092	PUSHs ((SV *)get_cv ("Coro::State::_cctx_init", FALSE));	1095	slf_frame.prepare = slf_prepare_set_stacklevel;
1093	PUTBACK;	1096	slf_frame.check = slf_check_set_stacklevel;
1094	PL_op = (OP *)&myop;
1095	PL_op = PL_ppaddr[OP_ENTERSUB](aTHX);
1096	SPAGAIN;
1097	}	1097	}
1098		1098
1099	/* 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 */
1100	INLINE void	1100	INLINE void
1101	transfer_tail (pTHX)	1101	transfer_tail (pTHX)
1102	{	1102	{
1103	struct coro *next = (struct coro *)transfer_next;
1104	assert (!(transfer_next = 0)); /* just used for the side effect when asserts are enabled */
1105	assert (("FATAL: next coroutine was zero in transfer_tail (please report)", next));
1106
1107	free_coro_mortal (aTHX);	1103	free_coro_mortal (aTHX);
1108	UNLOCK;
1109
1110	if (expect_false (next->throw))
1111	{
1112	SV *exception = sv_2mortal (next->throw);
1113
1114	next->throw = 0;
1115	sv_setsv (ERRSV, exception);
1116	croak (0);
1117	}
1118	}	1104	}
1119		1105
1120	/*	1106	/*
1121	* this is a _very_ stripped down perl interpreter ;)	1107	* this is a _very_ stripped down perl interpreter ;)
1122	*/	1108	*/
…		…
1129	# endif	1115	# endif
1130	#endif	1116	#endif
1131	{	1117	{
1132	dTHX;	1118	dTHX;
1133		1119
1134	/* we are the alternative tail to pp_set_stacklevel */	1120	/* normally we would need to skip the entersub here */
1135	/* so do the same things here */	1121	/* not doing so will re-execute it, which is exactly what we want */
1136	SLF_TAIL;
1137
1138	/* we now skip the op that did lead to transfer() */
1139	PL_op = PL_op->op_next;	1122	/* PL_nop = PL_nop->op_next */
1140		1123
1141	/* inject a fake subroutine call to cctx_init */	1124	/* inject a fake subroutine call to cctx_init */
1142	cctx_prepare (aTHX_ (coro_cctx *)arg);	1125	cctx_prepare (aTHX_ (coro_cctx *)arg);
1143		1126
1144	/* cctx_run is the alternative tail of transfer() */	1127	/* cctx_run is the alternative tail of transfer() */
…		…
1305	/** coroutine switching *****************************************************/	1288	/** coroutine switching *****************************************************/
1306		1289
1307	static void	1290	static void
1308	transfer_check (pTHX_ struct coro *prev, struct coro *next)	1291	transfer_check (pTHX_ struct coro *prev, struct coro *next)
1309	{	1292	{
		1293	/* TODO: throwing up here is considered harmful */
		1294
1310	if (expect_true (prev != next))	1295	if (expect_true (prev != next))
1311	{	1296	{
1312	if (expect_false (!(prev->flags & (CF_RUNNING | CF_NEW))))	1297	if (expect_false (!(prev->flags & (CF_RUNNING | CF_NEW))))
1313	croak ("Coro::State::transfer called with non-running/new prev Coro::State, but can only transfer from running or new states");	1298	croak ("Coro::State::transfer called with non-running/new prev Coro::State, but can only transfer from running or new states,");
1314		1299
1315	if (expect_false (next->flags & CF_RUNNING))	1300	if (expect_false (next->flags & CF_RUNNING))
1316	croak ("Coro::State::transfer called with running next Coro::State, but can only transfer to inactive states");	1301	croak ("Coro::State::transfer called with running next Coro::State, but can only transfer to inactive states,");
1317		1302
1318	if (expect_false (next->flags & CF_DESTROYED))	1303	if (expect_false (next->flags & CF_DESTROYED))
1319	croak ("Coro::State::transfer called with destroyed next Coro::State, but can only transfer to inactive states");	1304	croak ("Coro::State::transfer called with destroyed next Coro::State, but can only transfer to inactive states,");
1320		1305
1321	#if !PERL_VERSION_ATLEAST (5,10,0)	1306	#if !PERL_VERSION_ATLEAST (5,10,0)
1322	if (expect_false (PL_lex_state != LEX_NOTPARSING))	1307	if (expect_false (PL_lex_state != LEX_NOTPARSING))
1323	croak ("Coro::State::transfer called while parsing, but this is not supported in your perl version");	1308	croak ("Coro::State::transfer called while parsing, but this is not supported in your perl version,");
1324	#endif	1309	#endif
1325	}	1310	}
1326	}	1311	}
1327		1312
1328	/* always use the TRANSFER macro */	1313	/* always use the TRANSFER macro */
1329	static void NOINLINE	1314	static void NOINLINE /* noinline so we have a fixed stackframe */
1330	transfer (pTHX_ struct coro *prev, struct coro *next, int force_cctx)	1315	transfer (pTHX_ struct coro *prev, struct coro *next, int force_cctx)
1331	{	1316	{
1332	dSTACKLEVEL;	1317	dSTACKLEVEL;
1333		1318
1334	/* sometimes transfer is only called to set idle_sp */	1319	/* sometimes transfer is only called to set idle_sp */
1335	if (expect_false (!next))	1320	if (expect_false (!next))
1336	{	1321	{
1337	((coro_cctx *)prev)->idle_sp = stacklevel;	1322	((coro_cctx *)prev)->idle_sp = (void *)stacklevel;
1338	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 */
1339	}	1324	}
1340	else if (expect_true (prev != next))	1325	else if (expect_true (prev != next))
1341	{	1326	{
1342	coro_cctx *prev__cctx;	1327	coro_cctx *prev__cctx;
…		…
1349	prev->flags |= CF_RUNNING;	1334	prev->flags |= CF_RUNNING;
1350	}	1335	}
1351		1336
1352	prev->flags &= ~CF_RUNNING;	1337	prev->flags &= ~CF_RUNNING;
1353	next->flags |= CF_RUNNING;	1338	next->flags |= CF_RUNNING;
1354
1355	LOCK;
1356		1339
1357	/* first get rid of the old state */	1340	/* first get rid of the old state */
1358	save_perl (aTHX_ prev);	1341	save_perl (aTHX_ prev);
1359		1342
1360	if (expect_false (next->flags & CF_NEW))	1343	if (expect_false (next->flags & CF_NEW))
…		…
1369		1352
1370	prev__cctx = prev->cctx;	1353	prev__cctx = prev->cctx;
1371		1354
1372	/* possibly untie and reuse the cctx */	1355	/* possibly untie and reuse the cctx */
1373	if (expect_true (	1356	if (expect_true (
1374	prev__cctx->idle_sp == stacklevel	1357	prev__cctx->idle_sp == (void *)stacklevel
1375	&& !(prev__cctx->flags & CC_TRACE)	1358	&& !(prev__cctx->flags & CC_TRACE)
1376	&& !force_cctx	1359	&& !force_cctx
1377	))	1360	))
1378	{	1361	{
1379	/* 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 */
…		…
1393	++next->usecount;	1376	++next->usecount;
1394		1377
1395	if (expect_true (!next->cctx))	1378	if (expect_true (!next->cctx))
1396	next->cctx = cctx_get (aTHX);	1379	next->cctx = cctx_get (aTHX);
1397		1380
1398	assert (("FATAL: transfer_next already nonzero in Coro (please report)", !transfer_next));
1399	transfer_next = next;
1400
1401	if (expect_false (prev__cctx != next->cctx))	1381	if (expect_false (prev__cctx != next->cctx))
1402	{	1382	{
1403	prev__cctx->top_env = PL_top_env;	1383	prev__cctx->top_env = PL_top_env;
1404	PL_top_env = next->cctx->top_env;	1384	PL_top_env = next->cctx->top_env;
1405	coro_transfer (&prev__cctx->cctx, &next->cctx->cctx);	1385	coro_transfer (&prev__cctx->cctx, &next->cctx->cctx);
…		…
1418	coro_state_destroy (pTHX_ struct coro *coro)	1398	coro_state_destroy (pTHX_ struct coro *coro)
1419	{	1399	{
1420	if (coro->flags & CF_DESTROYED)	1400	if (coro->flags & CF_DESTROYED)
1421	return 0;	1401	return 0;
1422		1402
		1403	if (coro->on_destroy)
		1404	coro->on_destroy (aTHX_ coro);
		1405
1423	coro->flags |= CF_DESTROYED;	1406	coro->flags |= CF_DESTROYED;
1424		1407
1425	if (coro->flags & CF_READY)	1408	if (coro->flags & CF_READY)
1426	{	1409	{
1427	/* reduce nready, as destroying a ready coro effectively unreadies it */	1410	/* reduce nready, as destroying a ready coro effectively unreadies it */
1428	/* alternative: look through all ready queues and remove the coro */	1411	/* alternative: look through all ready queues and remove the coro */
1429	LOCK;
1430	--coro_nready;	1412	--coro_nready;
1431	UNLOCK;
1432	}	1413	}
1433	else	1414	else
1434	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 */
1435		1416
1436	if (coro->mainstack && coro->mainstack != main_mainstack)	1417	if (coro->mainstack && coro->mainstack != main_mainstack)
1437	{	1418	{
1438	struct coro temp;	1419	struct coro temp;
1439		1420
1440	if (coro->flags & CF_RUNNING)	1421	assert (("FATAL: tried to destroy currently running coroutine (please report)", !(coro->flags & CF_RUNNING)));
1441	croak ("FATAL: tried to destroy currently running coroutine");
1442		1422
1443	save_perl (aTHX_ &temp);	1423	save_perl (aTHX_ &temp);
1444	load_perl (aTHX_ coro);	1424	load_perl (aTHX_ coro);
1445		1425
1446	coro_destruct (aTHX_ coro);	1426	coro_destruct (aTHX_ coro);
…		…
1497	# define MGf_DUP 0	1477	# define MGf_DUP 0
1498	#endif	1478	#endif
1499	};	1479	};
1500		1480
1501	static void	1481	static void
1502	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)
1503	{	1483	{
1504	ta->prev = SvSTATE (prev_sv);	1484	ta->prev = SvSTATE (prev_sv);
1505	ta->next = SvSTATE (next_sv);	1485	ta->next = SvSTATE (next_sv);
1506	TRANSFER_CHECK (*ta);	1486	TRANSFER_CHECK (*ta);
1507	}	1487	}
1508		1488
1509	static void	1489	static void
1510	api_transfer (pTHX_ SV *prev_sv, SV *next_sv)	1490	api_transfer (pTHX_ SV *prev_sv, SV *next_sv)
1511	{	1491	{
1512	struct transfer_args ta;	1492	struct coro_transfer_args ta;
1513		1493
1514	prepare_transfer (aTHX_ &ta, prev_sv, next_sv);	1494	prepare_transfer (aTHX_ &ta, prev_sv, next_sv);
1515	TRANSFER (ta, 1);	1495	TRANSFER (ta, 1);
1516	}	1496	}
1517		1497
1518	/** Coro ********************************************************************/	1498	/** Coro ********************************************************************/
1519		1499
1520	static void	1500	INLINE void
1521	coro_enq (pTHX_ SV *coro_sv)	1501	coro_enq (pTHX_ struct coro *coro)
1522	{	1502	{
1523	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));
1524	}	1504	}
1525		1505
1526	static SV *	1506	INLINE SV *
1527	coro_deq (pTHX)	1507	coro_deq (pTHX)
1528	{	1508	{
1529	int prio;	1509	int prio;
1530		1510
1531	for (prio = PRIO_MAX - PRIO_MIN + 1; --prio >= 0; )	1511	for (prio = PRIO_MAX - PRIO_MIN + 1; --prio >= 0; )
…		…
1550	if (coro->flags & CF_READY)	1530	if (coro->flags & CF_READY)
1551	return 0;	1531	return 0;
1552		1532
1553	coro->flags |= CF_READY;	1533	coro->flags |= CF_READY;
1554		1534
1555	LOCK;
1556
1557	sv_hook = coro_nready ? 0 : coro_readyhook;	1535	sv_hook = coro_nready ? 0 : coro_readyhook;
1558	xs_hook = coro_nready ? 0 : coroapi.readyhook;	1536	xs_hook = coro_nready ? 0 : coroapi.readyhook;
1559		1537
1560	coro_enq (aTHX_ SvREFCNT_inc_NN (coro_sv));	1538	coro_enq (aTHX_ coro);
1561	++coro_nready;	1539	++coro_nready;
1562		1540
1563	UNLOCK;
1564
1565	if (sv_hook)	1541	if (sv_hook)
1566	{	1542	{
1567	dSP;	1543	dSP;
1568		1544
1569	ENTER;	1545	ENTER;
1570	SAVETMPS;	1546	SAVETMPS;
1571		1547
1572	PUSHMARK (SP);	1548	PUSHMARK (SP);
1573	PUTBACK;	1549	PUTBACK;
1574	call_sv (sv_hook, G_DISCARD);	1550	call_sv (sv_hook, G_VOID | G_DISCARD);
1575	SPAGAIN;
1576		1551
1577	FREETMPS;	1552	FREETMPS;
1578	LEAVE;	1553	LEAVE;
1579	}	1554	}
1580		1555
…		…
1589	{	1564	{
1590	return !!(SvSTATE (coro_sv)->flags & CF_READY);	1565	return !!(SvSTATE (coro_sv)->flags & CF_READY);
1591	}	1566	}
1592		1567
1593	INLINE void	1568	INLINE void
1594	prepare_schedule (pTHX_ struct transfer_args *ta)	1569	prepare_schedule (pTHX_ struct coro_transfer_args *ta)
1595	{	1570	{
1596	SV *prev_sv, *next_sv;	1571	SV *prev_sv, *next_sv;
1597		1572
1598	for (;;)	1573	for (;;)
1599	{	1574	{
1600	LOCK;
1601	next_sv = coro_deq (aTHX);	1575	next_sv = coro_deq (aTHX);
1602		1576
1603	/* nothing to schedule: call the idle handler */	1577	/* nothing to schedule: call the idle handler */
1604	if (expect_false (!next_sv))	1578	if (expect_false (!next_sv))
1605	{	1579	{
1606	dSP;	1580	dSP;
1607	UNLOCK;
1608		1581
1609	ENTER;	1582	ENTER;
1610	SAVETMPS;	1583	SAVETMPS;
1611		1584
1612	PUSHMARK (SP);	1585	PUSHMARK (SP);
1613	PUTBACK;	1586	PUTBACK;
1614	call_sv (get_sv ("Coro::idle", FALSE), G_DISCARD);	1587	call_sv (get_sv ("Coro::idle", FALSE), G_VOID | G_DISCARD);
1615	SPAGAIN;
1616		1588
1617	FREETMPS;	1589	FREETMPS;
1618	LEAVE;	1590	LEAVE;
1619	continue;	1591	continue;
1620	}	1592	}
1621		1593
1622	ta->next = SvSTATE (next_sv);	1594	ta->next = SvSTATE_hv (next_sv);
1623		1595
1624	/* cannot transfer to destroyed coros, skip and look for next */	1596	/* cannot transfer to destroyed coros, skip and look for next */
1625	if (expect_false (ta->next->flags & CF_DESTROYED))	1597	if (expect_false (ta->next->flags & CF_DESTROYED))
1626	{	1598	{
1627	UNLOCK;
1628	SvREFCNT_dec (next_sv);	1599	SvREFCNT_dec (next_sv);
1629	/* coro_nready has already been taken care of by destroy */	1600	/* coro_nready has already been taken care of by destroy */
1630	continue;	1601	continue;
1631	}	1602	}
1632		1603
1633	--coro_nready;	1604	--coro_nready;
1634	UNLOCK;
1635	break;	1605	break;
1636	}	1606	}
1637		1607
1638	/* free this only after the transfer */	1608	/* free this only after the transfer */
1639	prev_sv = SvRV (coro_current);	1609	prev_sv = SvRV (coro_current);
1640	ta->prev = SvSTATE (prev_sv);	1610	ta->prev = SvSTATE_hv (prev_sv);
1641	TRANSFER_CHECK (*ta);	1611	TRANSFER_CHECK (*ta);
1642	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));
1643	ta->next->flags &= ~CF_READY;	1613	ta->next->flags &= ~CF_READY;
1644	SvRV_set (coro_current, next_sv);	1614	SvRV_set (coro_current, next_sv);
1645		1615
1646	LOCK;
1647	free_coro_mortal (aTHX);	1616	free_coro_mortal (aTHX);
1648	coro_mortal = prev_sv;	1617	coro_mortal = prev_sv;
1649	UNLOCK;
1650	}	1618	}
1651		1619
1652	INLINE void	1620	INLINE void
1653	prepare_cede (pTHX_ struct transfer_args *ta)	1621	prepare_cede (pTHX_ struct coro_transfer_args *ta)
1654	{	1622	{
1655	api_ready (aTHX_ coro_current);	1623	api_ready (aTHX_ coro_current);
1656	prepare_schedule (aTHX_ ta);	1624	prepare_schedule (aTHX_ ta);
1657	}	1625	}
1658		1626
1659	static void	1627	INLINE void
1660	prepare_cede_notself (pTHX_ struct transfer_args *ta)	1628	prepare_cede_notself (pTHX_ struct coro_transfer_args *ta)
1661	{	1629	{
1662	SV *prev = SvRV (coro_current);	1630	SV *prev = SvRV (coro_current);
1663		1631
1664	if (coro_nready)	1632	if (coro_nready)
1665	{	1633	{
1666	prepare_schedule (aTHX_ ta);	1634	prepare_schedule (aTHX_ ta);
1667	api_ready (aTHX_ prev);	1635	api_ready (aTHX_ prev);
1668	}	1636	}
1669	else	1637	else
1670	ta->prev = ta->next = SvSTATE (prev);	1638	prepare_nop (aTHX_ ta);
1671	}	1639	}
1672		1640
1673	static void	1641	static void
1674	api_schedule (pTHX)	1642	api_schedule (pTHX)
1675	{	1643	{
1676	struct transfer_args ta;	1644	struct coro_transfer_args ta;
1677		1645
1678	prepare_schedule (aTHX_ &ta);	1646	prepare_schedule (aTHX_ &ta);
1679	TRANSFER (ta, 1);	1647	TRANSFER (ta, 1);
1680	}	1648	}
1681		1649
1682	static int	1650	static int
1683	api_cede (pTHX)	1651	api_cede (pTHX)
1684	{	1652	{
1685	struct transfer_args ta;	1653	struct coro_transfer_args ta;
1686		1654
1687	prepare_cede (aTHX_ &ta);	1655	prepare_cede (aTHX_ &ta);
1688		1656
1689	if (expect_true (ta.prev != ta.next))	1657	if (expect_true (ta.prev != ta.next))
1690	{	1658	{
…		…
1698	static int	1666	static int
1699	api_cede_notself (pTHX)	1667	api_cede_notself (pTHX)
1700	{	1668	{
1701	if (coro_nready)	1669	if (coro_nready)
1702	{	1670	{
1703	struct transfer_args ta;	1671	struct coro_transfer_args ta;
1704		1672
1705	prepare_cede_notself (aTHX_ &ta);	1673	prepare_cede_notself (aTHX_ &ta);
1706	TRANSFER (ta, 1);	1674	TRANSFER (ta, 1);
1707	return 1;	1675	return 1;
1708	}	1676	}
…		…
1718	if (flags & CC_TRACE)	1686	if (flags & CC_TRACE)
1719	{	1687	{
1720	if (!coro->cctx)	1688	if (!coro->cctx)
1721	coro->cctx = cctx_new_run ();	1689	coro->cctx = cctx_new_run ();
1722	else if (!(coro->cctx->flags & CC_TRACE))	1690	else if (!(coro->cctx->flags & CC_TRACE))
1723	croak ("cannot enable tracing on coroutine with custom stack");	1691	croak ("cannot enable tracing on coroutine with custom stack,");
1724		1692
1725	coro->cctx->flags |= CC_NOREUSE | (flags & (CC_TRACE | CC_TRACE_ALL));	1693	coro->cctx->flags |= CC_NOREUSE | (flags & (CC_TRACE | CC_TRACE_ALL));
1726	}	1694	}
1727	else if (coro->cctx && coro->cctx->flags & CC_TRACE)	1695	else if (coro->cctx && coro->cctx->flags & CC_TRACE)
1728	{	1696	{
…		…
1733	else	1701	else
1734	coro->slot->runops = RUNOPS_DEFAULT;	1702	coro->slot->runops = RUNOPS_DEFAULT;
1735	}	1703	}
1736	}	1704	}
1737		1705
1738	#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
1739	static int	1736	static void
1740	coro_gensub_free (pTHX_ SV *sv, MAGIC *mg)	1737	slf_prepare_transfer (pTHX_ struct coro_transfer_args *ta)
1741	{	1738	{
1742	AV *padlist;	1739	SV **arg = (SV **)slf_frame.data;
1743	AV *av = (AV *)mg->mg_obj;
1744		1740
1745	abort ();	1741	prepare_transfer (aTHX_ ta, arg [0], arg [1]);
1746
1747	return 0;
1748	}	1742	}
1749		1743
1750	static MGVTBL coro_gensub_vtbl = {	1744	static void
1751	0, 0, 0, 0,	1745	slf_init_transfer (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
1752	coro_gensub_free	1746	{
1753	};	1747	if (items != 2)
1754	#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	}
1755		1919
1756	/*****************************************************************************/	1920	/*****************************************************************************/
1757	/* PerlIO::cede */	1921	/* PerlIO::cede */
1758		1922
1759	typedef struct	1923	typedef struct
…		…
1827	PerlIOBuf_get_cnt,	1991	PerlIOBuf_get_cnt,
1828	PerlIOBuf_set_ptrcnt,	1992	PerlIOBuf_set_ptrcnt,
1829	};	1993	};
1830		1994
1831	/*****************************************************************************/	1995	/*****************************************************************************/
		1996	/* Coro::Semaphore */
1832		1997
1833	static const CV *slf_cv; /* for quick consistency check */	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);
1834		2003
1835	static UNOP slf_restore; /* restore stack as entersub did, for first-re-run */	2004	count += adjust;
1836	static SV *slf_arg0;	2005	SvIVX (count_sv) = count;
1837	static SV *slf_arg1;
1838		2006
1839	/* this restores the stack in the case we patched the entersub, to */	2007	/* now wake up as many waiters as are expected to lock */
1840	/* recreate the stack frame as perl will on following calls */	2008	while (count > 0 && AvFILLp (av) > 0)
1841	/* since entersub cleared the stack */	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 */
1842	static OP *	2099	static SV *
1843	pp_restore (pTHX)	2100	gensub (pTHX_ void (*xsub)(pTHX_ CV *), void *arg)
1844	{	2101	{
1845	dSP;	2102	CV *cv = (CV *)newSV (0);
1846		2103
1847	PUSHMARK (SP);	2104	sv_upgrade ((SV *)cv, SVt_PVCV);
1848		2105
1849	EXTEND (SP, 3);	2106	CvANON_on (cv);
1850	if (slf_arg0) PUSHs (sv_2mortal (slf_arg0));	2107	CvISXSUB_on (cv);
1851	if (slf_arg1) PUSHs (sv_2mortal (slf_arg1));	2108	CvXSUB (cv) = xsub;
1852	PUSHs ((SV *)CvGV (slf_cv));	2109	GENSUB_ARG = arg;
1853		2110
1854	RETURNOP (slf_restore.op_first);	2111	return newRV_noinc ((SV *)cv);
1855	}	2112	}
1856		2113
1857	#define OPpENTERSUB_SLF 15 /* the part of op_private entersub hopefully doesn't use */	2114	/*****************************************************************************/
		2115	/* Coro::AIO */
1858		2116
1859	enum {	2117	#define CORO_MAGIC_type_aio PERL_MAGIC_ext
1860	CORO_SLF_CUSTOM = 0,	2118
1861	CORO_SLF_SET_STACKLEVEL = 1,	2119	/* helper storage struct */
1862	CORO_SLF_TRANSFER = 2	2120	struct io_state
		2121	{
		2122	int errorno;
		2123	I32 laststype; /* U16 in 5.10.0 */
		2124	int laststatval;
		2125	Stat_t statcache;
1863	};	2126	};
1864		2127
1865	/* declare prototype */	2128	static void
1866	XS(XS_Coro__State__set_stacklevel);	2129	coro_aio_callback (pTHX_ CV *cv)
1867
1868	/*
1869	* these not obviously related functions are all rolled into one
1870	* function to increase chances that they all will call transfer with the same
1871	* stack offset
1872	* SLF stands for "schedule-like-function".
1873	*/
1874	static OP *
1875	pp_slf (pTHX)
1876	{	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	{
1877	dSP;	2190	dSP;
1878	struct transfer_args ta;	2191	int i;
1879	SV **arg = PL_stack_base + TOPMARK + 1;
1880	int items = SP - arg; /* args without function object */
1881	int ix = PL_op->op_private & OPpENTERSUB_SLF;
1882	struct CoroSLF *slf = 0;
1883		2192
1884	/* do a quick consistency check on the "function" object, and if it isn't */	2193	EXTEND (SP, AvFILLp (state) + 1);
1885	/* for us, divert to the real entersub */	2194	for (i = 0; i <= AvFILLp (state); ++i)
1886	if (SvTYPE (*sp) != SVt_PVGV || CvXSUB (GvCV (*sp)) != XS_Coro__State__set_stacklevel)	2195	PUSHs (sv_2mortal (SvREFCNT_inc_NN (AvARRAY (state)[i])));
1887	return PL_ppaddr[OP_ENTERSUB](aTHX);
1888		2196
1889	/* pop args */
1890	SP = PL_stack_base + POPMARK;
1891
1892	if (!(PL_op->op_flags & OPf_STACKED))
1893	{
1894	/* ampersand-form of call, use @_ instead of stack */
1895	AV *av = GvAV (PL_defgv);
1896	arg = AvARRAY (av);
1897	items = AvFILLp (av) + 1;
1898	}
1899
1900	PUTBACK;	2197	PUTBACK;
		2198	}
1901		2199
1902	if (!ix)	2200	return 0;
1903	{	2201	}
1904	slf = (struct CoroSLF *)CvSTART (GvCV (*sp));	2202
1905	ix = slf->prepare (aTHX_ arg, items);	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)
1906	}	2215	{
		2216	dSP;
1907		2217
1908	switch (ix)	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);
1909	{	2233	}
1910	case CORO_SLF_SET_STACKLEVEL:
1911	prepare_set_stacklevel (&ta, (struct coro_cctx *)SvIV (arg [0]));
1912	break;
1913		2234
1914	case CORO_SLF_TRANSFER:	2235	/* now call the original request */
1915	if (items != 2)	2236	{
1916	croak ("Coro::State::transfer (prev, next) expects two arguments, not %d.", items);	2237	dSP;
		2238	CV *req = (CV *)CORO_MAGIC_NN ((SV *)cv, CORO_MAGIC_type_aio)->mg_obj;
		2239	int i;
1917		2240
1918	prepare_transfer (aTHX_ &ta, arg [0], arg [1]);	2241	PUSHMARK (SP);
1919	break;
1920		2242
1921	case CORO_SLF_SCHEDULE:	2243	/* first push all args to the stack */
1922	prepare_schedule (aTHX_ &ta);	2244	EXTEND (SP, items + 1);
1923	break;
1924		2245
1925	case CORO_SLF_CEDE:	2246	for (i = 0; i < items; ++i)
1926	prepare_cede (aTHX_ &ta);	2247	PUSHs (arg [i]);
1927	break;
1928		2248
1929	case CORO_SLF_CEDE_NOTSELF:	2249	/* now push the callback closure */
1930	prepare_cede_notself (aTHX_ &ta);	2250	PUSHs (sv_2mortal (gensub (aTHX_ coro_aio_callback, (void *)SvREFCNT_inc_NN ((SV *)state))));
1931	break;
1932		2251
1933	default:	2252	/* now call the AIO function - we assume our request is uncancelable */
1934	abort ();
1935	}
1936
1937	do
1938	TRANSFER (ta, 0);
1939	while (slf && slf->check (aTHX));
1940
1941	SPAGAIN;
1942
1943	PUTBACK;	2253	PUTBACK;
1944	SLF_TAIL;	2254	call_sv ((SV *)req, G_VOID | G_DISCARD);
1945	SPAGAIN;	2255	}
1946	RETURN;
1947	}
1948		2256
1949	static void	2257	/* now that the requets is going, we loop toll we have a result */
1950	coro_slf_patch (pTHX_ CV *cv, int ix, SV **args, int items)	2258	frame->data = (void *)state;
1951	{	2259	frame->prepare = prepare_schedule;
1952	assert (("FATAL: SLF call recursion in Coro module (please report)", PL_op->op_ppaddr != pp_slf));	2260	frame->check = slf_check_aio_req;
1953
1954	assert (("FATAL: SLF call with illegal CV value", CvGV (cv)));
1955	slf_cv = cv;
1956
1957	/* we patch the op, and then re-run the whole call */
1958	/* we have to put the same argument on the stack for this to work */
1959	/* and this will be done by pp_restore */
1960	slf_restore.op_next = (OP *)&slf_restore;
1961	slf_restore.op_type = OP_NULL;
1962	slf_restore.op_ppaddr = pp_restore;
1963	slf_restore.op_first = PL_op;
1964
1965	slf_arg0 = items > 0 ? SvREFCNT_inc (args [0]) : 0;
1966	slf_arg1 = items > 1 ? SvREFCNT_inc (args [1]) : 0;
1967
1968	PL_op->op_ppaddr = pp_slf;
1969	PL_op->op_private = PL_op->op_private & ~OPpENTERSUB_SLF | ix; /* we potentially share our private flags with entersub */
1970
1971	PL_op = (OP *)&slf_restore;
1972	}	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	/*****************************************************************************/
1973		2274
1974	MODULE = Coro::State PACKAGE = Coro::State PREFIX = api_	2275	MODULE = Coro::State PACKAGE = Coro::State PREFIX = api_
1975		2276
1976	PROTOTYPES: DISABLE	2277	PROTOTYPES: DISABLE
1977		2278
1978	BOOT:	2279	BOOT:
1979	{	2280	{
1980	#ifdef USE_ITHREADS	2281	#ifdef USE_ITHREADS
1981	MUTEX_INIT (&coro_lock);
1982	# if CORO_PTHREAD	2282	# if CORO_PTHREAD
1983	coro_thx = PERL_GET_CONTEXT;	2283	coro_thx = PERL_GET_CONTEXT;
1984	# endif	2284	# endif
1985	#endif	2285	#endif
1986	BOOT_PAGESIZE;	2286	BOOT_PAGESIZE;
…		…
2007	main_top_env = PL_top_env;	2307	main_top_env = PL_top_env;
2008		2308
2009	while (main_top_env->je_prev)	2309	while (main_top_env->je_prev)
2010	main_top_env = main_top_env->je_prev;	2310	main_top_env = main_top_env->je_prev;
2011		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
2012	coroapi.ver = CORO_API_VERSION;	2324	coroapi.ver = CORO_API_VERSION;
2013	coroapi.rev = CORO_API_REVISION;	2325	coroapi.rev = CORO_API_REVISION;
		2326
2014	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;
2015		2335
2016	{	2336	{
2017	SV **svp = hv_fetch (PL_modglobal, "Time::NVtime", 12, 0);	2337	SV **svp = hv_fetch (PL_modglobal, "Time::NVtime", 12, 0);
2018		2338
2019	if (!svp) croak ("Time::HiRes is required");	2339	if (!svp) croak ("Time::HiRes is required");
…		…
2053	}	2373	}
2054	OUTPUT:	2374	OUTPUT:
2055	RETVAL	2375	RETVAL
2056		2376
2057	void	2377	void
2058	_set_stacklevel (...)	2378	transfer (...)
2059	ALIAS:	2379	PROTOTYPE: $$
2060	_set_stacklevel = CORO_SLF_SET_STACKLEVEL	2380	CODE:
2061	Coro::State::transfer = CORO_SLF_TRANSFER	2381	CORO_EXECUTE_SLF_XS (slf_init_transfer);
2062	Coro::schedule = CORO_SLF_SCHEDULE
2063	Coro::cede = CORO_SLF_CEDE
2064	Coro::cede_notself = CORO_SLF_CEDE_NOTSELF
2065	CODE:
2066	coro_slf_patch (aTHX_ cv, ix, &ST (0), items);
2067		2382
2068	bool	2383	bool
2069	_destroy (SV *coro_sv)	2384	_destroy (SV *coro_sv)
2070	CODE:	2385	CODE:
2071	RETVAL = coro_state_destroy (aTHX_ SvSTATE (coro_sv));	2386	RETVAL = coro_state_destroy (aTHX_ SvSTATE (coro_sv));
…		…
2078	CODE:	2393	CODE:
2079	_exit (code);	2394	_exit (code);
2080		2395
2081	int	2396	int
2082	cctx_stacksize (int new_stacksize = 0)	2397	cctx_stacksize (int new_stacksize = 0)
		2398	PROTOTYPE: ;$
2083	CODE:	2399	CODE:
2084	RETVAL = cctx_stacksize;	2400	RETVAL = cctx_stacksize;
2085	if (new_stacksize)	2401	if (new_stacksize)
2086	{	2402	{
2087	cctx_stacksize = new_stacksize;	2403	cctx_stacksize = new_stacksize;
…		…
2090	OUTPUT:	2406	OUTPUT:
2091	RETVAL	2407	RETVAL
2092		2408
2093	int	2409	int
2094	cctx_max_idle (int max_idle = 0)	2410	cctx_max_idle (int max_idle = 0)
		2411	PROTOTYPE: ;$
2095	CODE:	2412	CODE:
2096	RETVAL = cctx_max_idle;	2413	RETVAL = cctx_max_idle;
2097	if (max_idle > 1)	2414	if (max_idle > 1)
2098	cctx_max_idle = max_idle;	2415	cctx_max_idle = max_idle;
2099	OUTPUT:	2416	OUTPUT:
2100	RETVAL	2417	RETVAL
2101		2418
2102	int	2419	int
2103	cctx_count ()	2420	cctx_count ()
		2421	PROTOTYPE:
2104	CODE:	2422	CODE:
2105	RETVAL = cctx_count;	2423	RETVAL = cctx_count;
2106	OUTPUT:	2424	OUTPUT:
2107	RETVAL	2425	RETVAL
2108		2426
2109	int	2427	int
2110	cctx_idle ()	2428	cctx_idle ()
		2429	PROTOTYPE:
2111	CODE:	2430	CODE:
2112	RETVAL = cctx_idle;	2431	RETVAL = cctx_idle;
2113	OUTPUT:	2432	OUTPUT:
2114	RETVAL	2433	RETVAL
2115		2434
2116	void	2435	void
2117	list ()	2436	list ()
		2437	PROTOTYPE:
2118	PPCODE:	2438	PPCODE:
2119	{	2439	{
2120	struct coro *coro;	2440	struct coro *coro;
2121	for (coro = coro_first; coro; coro = coro->next)	2441	for (coro = coro_first; coro; coro = coro->next)
2122	if (coro->hv)	2442	if (coro->hv)
…		…
2184		2504
2185	void	2505	void
2186	throw (Coro::State self, SV *throw = &PL_sv_undef)	2506	throw (Coro::State self, SV *throw = &PL_sv_undef)
2187	PROTOTYPE: $;$	2507	PROTOTYPE: $;$
2188	CODE:	2508	CODE:
		2509	{
		2510	struct coro *current = SvSTATE_current;
		2511	SV **throwp = self == current ? &coro_throw : &self->throw;
2189	SvREFCNT_dec (self->throw);	2512	SvREFCNT_dec (*throwp);
2190	self->throw = SvOK (throw) ? newSVsv (throw) : 0;	2513	*throwp = SvOK (throw) ? newSVsv (throw) : 0;
		2514	}
2191		2515
2192	void	2516	void
2193	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: $;$
2194	C_ARGS: aTHX_ coro, flags	2519	C_ARGS: aTHX_ coro, flags
2195		2520
2196	SV *	2521	SV *
2197	has_cctx (Coro::State coro)	2522	has_cctx (Coro::State coro)
2198	PROTOTYPE: $	2523	PROTOTYPE: $
…		…
2223	OUTPUT:	2548	OUTPUT:
2224	RETVAL	2549	RETVAL
2225		2550
2226	void	2551	void
2227	force_cctx ()	2552	force_cctx ()
		2553	PROTOTYPE:
2228	CODE:	2554	CODE:
2229	struct coro *coro = SvSTATE (coro_current);
2230	coro->cctx->idle_sp = 0;	2555	SvSTATE_current->cctx->idle_sp = 0;
2231		2556
2232	void	2557	void
2233	swap_defsv (Coro::State self)	2558	swap_defsv (Coro::State self)
2234	PROTOTYPE: $	2559	PROTOTYPE: $
2235	ALIAS:	2560	ALIAS:
2236	swap_defav = 1	2561	swap_defav = 1
2237	CODE:	2562	CODE:
2238	if (!self->slot)	2563	if (!self->slot)
2239	croak ("cannot swap state with coroutine that has no saved state");	2564	croak ("cannot swap state with coroutine that has no saved state,");
2240	else	2565	else
2241	{	2566	{
2242	SV **src = ix ? (SV **)&GvAV (PL_defgv) : &GvSV (PL_defgv);	2567	SV **src = ix ? (SV **)&GvAV (PL_defgv) : &GvSV (PL_defgv);
2243	SV **dst = ix ? (SV **)&self->slot->defav : (SV **)&self->slot->defsv;	2568	SV **dst = ix ? (SV **)&self->slot->defav : (SV **)&self->slot->defsv;
2244		2569
2245	SV *tmp = *src; *src = *dst; *dst = tmp;	2570	SV *tmp = *src; *src = *dst; *dst = tmp;
2246	}	2571	}
		2572
2247		2573
2248	MODULE = Coro::State PACKAGE = Coro	2574	MODULE = Coro::State PACKAGE = Coro
2249		2575
2250	BOOT:	2576	BOOT:
2251	{	2577	{
…		…
2286	SvREADONLY_on (sv);	2612	SvREADONLY_on (sv);
2287	}	2613	}
2288	}	2614	}
2289		2615
2290	void	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);
		2630
		2631	void
2291	_set_current (SV *current)	2632	_set_current (SV *current)
2292	PROTOTYPE: $	2633	PROTOTYPE: $
2293	CODE:	2634	CODE:
2294	SvREFCNT_dec (SvRV (coro_current));	2635	SvREFCNT_dec (SvRV (coro_current));
2295	SvRV_set (coro_current, SvREFCNT_inc_NN (SvRV (current)));	2636	SvRV_set (coro_current, SvREFCNT_inc_NN (SvRV (current)));
2296		2637
2297	void	2638	void
2298	_set_readyhook (SV *hook)	2639	_set_readyhook (SV *hook)
2299	PROTOTYPE: $	2640	PROTOTYPE: $
2300	CODE:	2641	CODE:
2301	LOCK;
2302	SvREFCNT_dec (coro_readyhook);	2642	SvREFCNT_dec (coro_readyhook);
2303	coro_readyhook = SvOK (hook) ? newSVsv (hook) : 0;	2643	coro_readyhook = SvOK (hook) ? newSVsv (hook) : 0;
2304	UNLOCK;
2305		2644
2306	int	2645	int
2307	prio (Coro::State coro, int newprio = 0)	2646	prio (Coro::State coro, int newprio = 0)
		2647	PROTOTYPE: $;$
2308	ALIAS:	2648	ALIAS:
2309	nice = 1	2649	nice = 1
2310	CODE:	2650	CODE:
2311	{	2651	{
2312	RETVAL = coro->prio;	2652	RETVAL = coro->prio;
…		…
2344	# for async_pool speedup	2684	# for async_pool speedup
2345	void	2685	void
2346	_pool_1 (SV *cb)	2686	_pool_1 (SV *cb)
2347	CODE:	2687	CODE:
2348	{	2688	{
2349	struct coro *coro = SvSTATE (coro_current);
2350	HV *hv = (HV *)SvRV (coro_current);	2689	HV *hv = (HV *)SvRV (coro_current);
		2690	struct coro *coro = SvSTATE_hv ((SV *)hv);
2351	AV *defav = GvAV (PL_defgv);	2691	AV *defav = GvAV (PL_defgv);
2352	SV *invoke = hv_delete (hv, "_invoke", sizeof ("_invoke") - 1, 0);	2692	SV *invoke = hv_delete (hv, "_invoke", sizeof ("_invoke") - 1, 0);
2353	AV *invoke_av;	2693	AV *invoke_av;
2354	int i, len;	2694	int i, len;
2355		2695
…		…
2376	{	2716	{
2377	av_fill (defav, len - 1);	2717	av_fill (defav, len - 1);
2378	for (i = 0; i < len; ++i)	2718	for (i = 0; i < len; ++i)
2379	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]));
2380	}	2720	}
2381
2382	SvREFCNT_dec (invoke);
2383	}	2721	}
2384		2722
2385	void	2723	void
2386	_pool_2 (SV *cb)	2724	_pool_2 (SV *cb)
2387	CODE:	2725	CODE:
2388	{	2726	{
2389	struct coro *coro = SvSTATE (coro_current);	2727	HV *hv = (HV *)SvRV (coro_current);
		2728	struct coro *coro = SvSTATE_hv ((SV *)hv);
2390		2729
2391	sv_setsv (cb, &PL_sv_undef);	2730	sv_setsv (cb, &PL_sv_undef);
2392		2731
2393	SvREFCNT_dec ((SV *)PL_defoutgv); PL_defoutgv = (GV *)coro->saved_deffh;	2732	SvREFCNT_dec ((SV *)PL_defoutgv); PL_defoutgv = (GV *)coro->saved_deffh;
2394	coro->saved_deffh = 0;	2733	coro->saved_deffh = 0;
…		…
2401	SvREFCNT_dec (old);	2740	SvREFCNT_dec (old);
2402	croak ("\3async_pool terminate\2\n");	2741	croak ("\3async_pool terminate\2\n");
2403	}	2742	}
2404		2743
2405	av_clear (GvAV (PL_defgv));	2744	av_clear (GvAV (PL_defgv));
2406	hv_store ((HV *)SvRV (coro_current), "desc", sizeof ("desc") - 1,	2745	hv_store (hv, "desc", sizeof ("desc") - 1,
2407	newSVpvn ("[async_pool idle]", sizeof ("[async_pool idle]") - 1), 0);	2746	newSVpvn ("[async_pool idle]", sizeof ("[async_pool idle]") - 1), 0);
2408		2747
2409	coro->prio = 0;	2748	coro->prio = 0;
2410		2749
2411	if (coro->cctx && (coro->cctx->flags & CC_TRACE))	2750	if (coro->cctx && (coro->cctx->flags & CC_TRACE))
2412	api_trace (aTHX_ coro_current, 0);	2751	api_trace (aTHX_ coro_current, 0);
2413		2752
2414	av_push (av_async_pool, newSVsv (coro_current));	2753	av_push (av_async_pool, newSVsv (coro_current));
2415	}	2754	}
2416		2755
2417	#if 0
2418		2756
2419	void	2757	MODULE = Coro::State PACKAGE = PerlIO::cede
2420	_generator_call (...)	2758
2421	PROTOTYPE: @	2759	BOOT:
2422	PPCODE:	2760	PerlIO_define_layer (aTHX_ &PerlIO_cede);
2423	fprintf (stderr, "call %p\n", CvXSUBANY(cv).any_ptr);	2761
2424	xxxx	2762
2425	abort ();	2763	MODULE = Coro::State PACKAGE = Coro::Semaphore
2426		2764
2427	SV *	2765	SV *
2428	gensub (SV *sub, ...)	2766	new (SV *klass, SV *count_ = 0)
2429	PROTOTYPE: &;@	2767	CODE:
2430	CODE:
2431	{	2768	{
2432	struct coro *coro;	2769	/* a semaphore contains a counter IV in $sem->[0] and any waiters after that */
2433	MAGIC *mg;	2770	AV *av = newAV ();
2434	CV *xcv;	2771	SV **ary;
2435	CV *ncv = (CV *)newSV_type (SVt_PVCV);
2436	int i;
2437		2772
2438	CvGV (ncv) = CvGV (cv);	2773	/* unfortunately, building manually saves memory */
2439	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);
2440		2780
2441	Newz (0, coro, 1, struct coro);	2781	RETVAL = sv_bless (newRV_noinc ((SV *)av), GvSTASH (CvGV (cv)));
2442	coro->args = newAV ();
2443	coro->flags = CF_NEW;
2444
2445	av_extend (coro->args, items - 1);
2446	for (i = 1; i < items; i++)
2447	av_push (coro->args, newSVsv (ST (i)));
2448
2449	CvISXSUB_on (ncv);
2450	CvXSUBANY (ncv).any_ptr = (void *)coro;
2451
2452	xcv = GvCV (gv_fetchpv ("Coro::_generator_call", 0, SVt_PVCV));
2453
2454	CvXSUB (ncv) = CvXSUB (xcv);
2455	CvANON_on (ncv);
2456
2457	mg = sv_magicext ((SV *)ncv, 0, CORO_MAGIC_type_state, &coro_gensub_vtbl, (char *)coro, 0);
2458	RETVAL = newRV_noinc ((SV *)ncv);
2459	}	2782	}
2460	OUTPUT:	2783	OUTPUT:
2461	RETVAL	2784	RETVAL
2462		2785
2463	#endif	2786	SV *
2464		2787	count (SV *self)
2465		2788	CODE:
2466	MODULE = Coro::State PACKAGE = Coro::AIO	2789	RETVAL = newSVsv (AvARRAY ((AV *)SvRV (self))[0]);
		2790	OUTPUT:
		2791	RETVAL
2467		2792
2468	void	2793	void
2469	_get_state (SV *self)	2794	up (SV *self, int adjust = 1)
2470	PPCODE:	2795	ALIAS:
2471	{	2796	adjust = 1
2472	AV *defav = GvAV (PL_defgv);	2797	CODE:
2473	AV *av = newAV ();	2798	coro_semaphore_adjust (aTHX_ (AV *)SvRV (self), ix ? adjust : 1);
2474	int i;
2475	SV *data_sv = newSV (sizeof (struct io_state));
2476	struct io_state *data = (struct io_state *)SvPVX (data_sv);
2477	SvCUR_set (data_sv, sizeof (struct io_state));
2478	SvPOK_only (data_sv);
2479
2480	data->errorno = errno;
2481	data->laststype = PL_laststype;
2482	data->laststatval = PL_laststatval;
2483	data->statcache = PL_statcache;
2484
2485	av_extend (av, AvFILLp (defav) + 1 + 1);
2486
2487	for (i = 0; i <= AvFILLp (defav); ++i)
2488	av_push (av, SvREFCNT_inc_NN (AvARRAY (defav)[i]));
2489
2490	av_push (av, data_sv);
2491
2492	XPUSHs (sv_2mortal (newRV_noinc ((SV *)av)));
2493
2494	api_ready (aTHX_ self);
2495	}
2496		2799
2497	void	2800	void
2498	_set_state (SV *state)	2801	down (SV *self)
2499	PROTOTYPE: $	2802	CODE:
2500	PPCODE:	2803	CORO_EXECUTE_SLF_XS (slf_init_semaphore_down);
		2804
		2805	void
		2806	try (SV *self)
		2807	PPCODE:
2501	{	2808	{
2502	AV *av = (AV *)SvRV (state);	2809	AV *av = (AV *)SvRV (self);
2503	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	{
2504	int i;	2833	int i;
2505
2506	errno = data->errorno;
2507	PL_laststype = data->laststype;
2508	PL_laststatval = data->laststatval;
2509	PL_statcache = data->statcache;
2510
2511	EXTEND (SP, AvFILLp (av));	2834	EXTEND (SP, AvFILLp (av) + 1 - 1);
2512	for (i = 0; i < AvFILLp (av); ++i)	2835	for (i = 1; i <= AvFILLp (av); ++i)
2513	PUSHs (sv_2mortal (SvREFCNT_inc_NN (AvARRAY (av)[i])));	2836	PUSHs (sv_2mortal (newRV_inc (AvARRAY (av)[i])));
		2837	}
2514	}	2838	}
2515		2839
2516		2840
2517	MODULE = Coro::State PACKAGE = Coro::AnyEvent	2841	MODULE = Coro::State PACKAGE = Coro::AnyEvent
2518		2842
2519	BOOT:	2843	BOOT:
2520	sv_activity = coro_get_sv (aTHX_ "Coro::AnyEvent::ACTIVITY", TRUE);	2844	sv_activity = coro_get_sv (aTHX_ "Coro::AnyEvent::ACTIVITY", TRUE);
2521		2845
2522	SV *	2846	void
2523	_schedule (...)	2847	_schedule (...)
2524	PROTOTYPE: @
2525	CODE:	2848	CODE:
2526	{	2849	{
2527	static int incede;	2850	static int incede;
2528		2851
2529	api_cede_notself (aTHX);	2852	api_cede_notself (aTHX);
…		…
2535	sv_setsv (sv_activity, &PL_sv_undef);	2858	sv_setsv (sv_activity, &PL_sv_undef);
2536	if (coro_nready >= incede)	2859	if (coro_nready >= incede)
2537	{	2860	{
2538	PUSHMARK (SP);	2861	PUSHMARK (SP);
2539	PUTBACK;	2862	PUTBACK;
2540	call_pv ("Coro::AnyEvent::_activity", G_DISCARD | G_EVAL);	2863	call_pv ("Coro::AnyEvent::_activity", G_KEEPERR | G_EVAL | G_VOID | G_DISCARD);
2541	SPAGAIN;
2542	}	2864	}
2543		2865
2544	--incede;	2866	--incede;
2545	}	2867	}
2546		2868
2547		2869
2548	MODULE = Coro::State PACKAGE = PerlIO::cede	2870	MODULE = Coro::State PACKAGE = Coro::AIO
2549		2871
2550	BOOT:	2872	void
2551	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	}
2552		2884

Diff Legend

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