ViewVC Help

View File | Revision Log | Show Annotations | Download File

/cvs/Coro/Coro/State.xs

Comparing Coro/Coro/State.xs (file contents):
Revision 1.269 by root, Fri Nov 14 06:50:11 2008 UTC vs.
Revision 1.287 by root, Mon Nov 17 07:03:12 2008 UTC

…		…
142	#define NOINLINE attribute ((noinline))	142	#define NOINLINE attribute ((noinline))
143		143
144	#include "CoroAPI.h"	144	#include "CoroAPI.h"
145		145
146	#ifdef USE_ITHREADS	146	#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	147	# if CORO_PTHREAD
152	static void *coro_thx;	148	static void *coro_thx;
153	# endif	149	# endif
154
155	#else
156
157	# define LOCK (void)0
158	# define UNLOCK (void)0
159
160	#endif	150	#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		151
177	static double (*nvtime)(); /* so why doesn't it take void? */	152	static double (*nvtime)(); /* so why doesn't it take void? */
178		153
179	static U32 cctx_gen;	154	static U32 cctx_gen;
180	static size_t cctx_stacksize = CORO_STACKSIZE;	155	static size_t cctx_stacksize = CORO_STACKSIZE;
181	static struct CoroAPI coroapi;	156	static struct CoroAPI coroapi;
182	static AV *main_mainstack; /* used to differentiate between $main and others */	157	static AV *main_mainstack; /* used to differentiate between $main and others */
183	static JMPENV *main_top_env;	158	static JMPENV *main_top_env;
184	static HV *coro_state_stash, *coro_stash;	159	static HV *coro_state_stash, *coro_stash;
185	static volatile SV *coro_mortal; /* will be freed/thrown after next transfer */	160	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		161
193	static GV *irsgv; /* $/ */	162	static GV *irsgv; /* $/ */
194	static GV *stdoutgv; /* *STDOUT */	163	static GV *stdoutgv; /* *STDOUT */
195	static SV *rv_diehook;	164	static SV *rv_diehook;
196	static SV *rv_warnhook;	165	static SV *rv_warnhook;
…		…
215	CC_TRACE_LINE = 0x10, /* trace each statement */	184	CC_TRACE_LINE = 0x10, /* trace each statement */
216	CC_TRACE_ALL = CC_TRACE_SUB | CC_TRACE_LINE,	185	CC_TRACE_ALL = CC_TRACE_SUB | CC_TRACE_LINE,
217	};	186	};
218		187
219	/* this is a structure representing a c-level coroutine */	188	/* this is a structure representing a c-level coroutine */
220	typedef struct coro_cctx {	189	typedef struct coro_cctx
		190	{
221	struct coro_cctx *next;	191	struct coro_cctx *next;
222		192
223	/* the stack */	193	/* the stack */
224	void *sptr;	194	void *sptr;
225	size_t ssize;	195	size_t ssize;
…		…
243	CF_NEW = 0x0004, /* has never been switched to */	213	CF_NEW = 0x0004, /* has never been switched to */
244	CF_DESTROYED = 0x0008, /* coroutine data has been freed */	214	CF_DESTROYED = 0x0008, /* coroutine data has been freed */
245	};	215	};
246		216
247	/* the structure where most of the perl state is stored, overlaid on the cxstack */	217	/* the structure where most of the perl state is stored, overlaid on the cxstack */
248	typedef struct {	218	typedef struct
		219	{
249	SV *defsv;	220	SV *defsv;
250	AV *defav;	221	AV *defav;
251	SV *errsv;	222	SV *errsv;
252	SV *irsgv;	223	SV *irsgv;
253	#define VAR(name,type) type name;	224	#define VAR(name,type) type name;
…		…
257		228
258	#define SLOT_COUNT ((sizeof (perl_slots) + sizeof (PERL_CONTEXT) - 1) / sizeof (PERL_CONTEXT))	229	#define SLOT_COUNT ((sizeof (perl_slots) + sizeof (PERL_CONTEXT) - 1) / sizeof (PERL_CONTEXT))
259		230
260	/* this is a structure representing a perl-level coroutine */	231	/* this is a structure representing a perl-level coroutine */
261	struct coro {	232	struct coro {
262	/* the c coroutine allocated to this perl coroutine, if any */	233	/* the C coroutine allocated to this perl coroutine, if any */
263	coro_cctx *cctx;	234	coro_cctx *cctx;
264		235
265	/* process data */	236	/* state data */
		237	struct CoroSLF slf_frame; /* saved slf frame */
266	AV *mainstack;	238	AV *mainstack;
267	perl_slots *slot; /* basically the saved sp */	239	perl_slots *slot; /* basically the saved sp */
268		240
269	AV *args; /* data associated with this coroutine (initial args) */	241	AV *args; /* data associated with this coroutine (initial args) */
270	int refcnt; /* coroutines are refcounted, yes */	242	int refcnt; /* coroutines are refcounted, yes */
271	int flags; /* CF_ flags */	243	int flags; /* CF_ flags */
272	HV *hv; /* the perl hash associated with this coro, if any */	244	HV *hv; /* the perl hash associated with this coro, if any */
		245	void (*on_destroy)(pTHX_ struct coro *coro);
273		246
274	/* statistics */	247	/* statistics */
275	int usecount; /* number of transfers to this coro */	248	int usecount; /* number of transfers to this coro */
276		249
277	/* coro process data */	250	/* coro process data */
…		…
285	struct coro *next, *prev;	258	struct coro *next, *prev;
286	};	259	};
287		260
288	typedef struct coro *Coro__State;	261	typedef struct coro *Coro__State;
289	typedef struct coro *Coro__State_or_hashref;	262	typedef struct coro *Coro__State_or_hashref;
		263
		264	/* the following variables are effectively part of the perl context */
		265	/* and get copied between struct coro and these variables */
		266	/* the mainr easonw e don't support windows process emulation */
		267	static struct CoroSLF slf_frame; /* the current slf frame */
		268	static SV *coro_throw;
290		269
291	/** Coro ********************************************************************/	270	/** Coro ********************************************************************/
292		271
293	#define PRIO_MAX 3	272	#define PRIO_MAX 3
294	#define PRIO_HIGH 1	273	#define PRIO_HIGH 1
…		…
299		278
300	/* for Coro.pm */	279	/* for Coro.pm */
301	static SV *coro_current;	280	static SV *coro_current;
302	static SV *coro_readyhook;	281	static SV *coro_readyhook;
303	static AV *coro_ready [PRIO_MAX - PRIO_MIN + 1];	282	static AV *coro_ready [PRIO_MAX - PRIO_MIN + 1];
304	static int coro_nready;
305	static struct coro *coro_first;	283	static struct coro *coro_first;
		284	#define coro_nready coroapi.nready
306		285
307	/** lowlevel stuff **********************************************************/	286	/** lowlevel stuff **********************************************************/
308		287
309	static SV *	288	static SV *
310	coro_get_sv (pTHX_ const char *name, int create)	289	coro_get_sv (pTHX_ const char *name, int create)
…		…
395	SvREFCNT_dec (av); /* sv_magicext increased the refcount */	374	SvREFCNT_dec (av); /* sv_magicext increased the refcount */
396		375
397	return 0;	376	return 0;
398	}	377	}
399		378
400	#define CORO_MAGIC_type_cv PERL_MAGIC_ext	379	#define CORO_MAGIC_type_cv 26
401	#define CORO_MAGIC_type_state PERL_MAGIC_ext	380	#define CORO_MAGIC_type_state PERL_MAGIC_ext
402		381
403	static MGVTBL coro_cv_vtbl = {	382	static MGVTBL coro_cv_vtbl = {
404	0, 0, 0, 0,	383	0, 0, 0, 0,
405	coro_cv_free	384	coro_cv_free
406	};	385	};
407		386
		387	#define CORO_MAGIC_NN(sv, type) \
		388	(expect_true (SvMAGIC (sv)->mg_type == type) \
		389	? SvMAGIC (sv) \
		390	: mg_find (sv, type))
		391
408	#define CORO_MAGIC(sv, type) \	392	#define CORO_MAGIC(sv, type) \
409	SvMAGIC (sv) \	393	(expect_true (SvMAGIC (sv)) \
410	? SvMAGIC (sv)->mg_type == type \	394	? CORO_MAGIC_NN (sv, type) \
411	? SvMAGIC (sv) \
412	: mg_find (sv, type) \
413	: 0	395	: 0)
414		396
415	#define CORO_MAGIC_cv(cv) CORO_MAGIC (((SV *)(cv)), CORO_MAGIC_type_cv)	397	#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)	398	#define CORO_MAGIC_state(sv) CORO_MAGIC_NN (((SV *)(sv)), CORO_MAGIC_type_state)
417		399
418	INLINE struct coro *	400	INLINE struct coro *
419	SvSTATE_ (pTHX_ SV *coro)	401	SvSTATE_ (pTHX_ SV *coro)
420	{	402	{
421	HV *stash;	403	HV *stash;
…		…
438	mg = CORO_MAGIC_state (coro);	420	mg = CORO_MAGIC_state (coro);
439	return (struct coro *)mg->mg_ptr;	421	return (struct coro *)mg->mg_ptr;
440	}	422	}
441		423
442	#define SvSTATE(sv) SvSTATE_ (aTHX_ (sv))	424	#define SvSTATE(sv) SvSTATE_ (aTHX_ (sv))
		425
		426	/* faster than SvSTATE, but expects a coroutine hv */
		427	#define SvSTATE_hv(hv) ((struct coro *)CORO_MAGIC_NN ((SV *)hv, CORO_MAGIC_type_state)->mg_ptr)
		428	#define SvSTATE_current SvSTATE_hv (SvRV (coro_current))
443		429
444	/* the next two functions merely cache the padlists */	430	/* the next two functions merely cache the padlists */
445	static void	431	static void
446	get_padlist (pTHX_ CV *cv)	432	get_padlist (pTHX_ CV *cv)
447	{	433	{
…		…
514	CvPADLIST (cv) = (AV *)POPs;	500	CvPADLIST (cv) = (AV *)POPs;
515	}	501	}
516		502
517	PUTBACK;	503	PUTBACK;
518	}	504	}
		505
		506	slf_frame = c->slf_frame;
		507	coro_throw = c->throw;
519	}	508	}
520		509
521	static void	510	static void
522	save_perl (pTHX_ Coro__State c)	511	save_perl (pTHX_ Coro__State c)
523	{	512	{
		513	c->throw = coro_throw;
		514	c->slf_frame = slf_frame;
		515
524	{	516	{
525	dSP;	517	dSP;
526	I32 cxix = cxstack_ix;	518	I32 cxix = cxstack_ix;
527	PERL_CONTEXT *ccstk = cxstack;	519	PERL_CONTEXT *ccstk = cxstack;
528	PERL_SI *top_si = PL_curstackinfo;	520	PERL_SI *top_si = PL_curstackinfo;
…		…
595	#undef VAR	587	#undef VAR
596	}	588	}
597	}	589	}
598		590
599	/*	591	/*
600	* allocate various perl stacks. This is an exact copy	592	* allocate various perl stacks. This is almost an exact copy
601	* of perl.c:init_stacks, except that it uses less memory	593	* of perl.c:init_stacks, except that it uses less memory
602	* on the (sometimes correct) assumption that coroutines do	594	* on the (sometimes correct) assumption that coroutines do
603	* not usually need a lot of stackspace.	595	* not usually need a lot of stackspace.
604	*/	596	*/
605	#if CORO_PREFER_PERL_FUNCTIONS	597	#if CORO_PREFER_PERL_FUNCTIONS
…		…
712	#endif	704	#endif
713	}	705	}
714	}	706	}
715		707
716	return rss;	708	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	}	709	}
745		710
746	/** coroutine stack handling ************************************************/	711	/** coroutine stack handling ************************************************/
747		712
748	static int (*orig_sigelem_get) (pTHX_ SV *sv, MAGIC *mg);	713	static int (*orig_sigelem_get) (pTHX_ SV *sv, MAGIC *mg);
…		…
834		799
835	return orig_sigelem_set ? orig_sigelem_set (aTHX_ sv, mg) : 0;	800	return orig_sigelem_set ? orig_sigelem_set (aTHX_ sv, mg) : 0;
836	}	801	}
837		802
838	static void	803	static void
		804	prepare_nop (pTHX_ struct coro_transfer_args *ta)
		805	{
		806	/* kind of mega-hacky, but works */
		807	ta->next = ta->prev = (struct coro *)ta;
		808	}
		809
		810	static int
		811	slf_check_nop (pTHX_ struct CoroSLF *frame)
		812	{
		813	return 0;
		814	}
		815
		816	static void NOINLINE /* noinline to keep it out of the transfer fast path */
839	coro_setup (pTHX_ struct coro *coro)	817	coro_setup (pTHX_ struct coro *coro)
840	{	818	{
841	/*	819	/*
842	* emulate part of the perl startup here.	820	* emulate part of the perl startup here.
843	*/	821	*/
…		…
882	PL_op = PL_ppaddr[OP_ENTERSUB](aTHX);	860	PL_op = PL_ppaddr[OP_ENTERSUB](aTHX);
883	SPAGAIN;	861	SPAGAIN;
884	}	862	}
885		863
886	/* this newly created coroutine might be run on an existing cctx which most	864	/* 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,	865	* 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	*/	866	*/
890	SLF_HEAD;	867	slf_frame.prepare = prepare_nop; /* provide a nop function for an eventual pp_slf */
		868	slf_frame.check = slf_check_nop; /* signal pp_slf to not repeat */
		869
		870	coro_throw = coro->throw;
891	}	871	}
892		872
893	static void	873	static void
894	coro_destruct (pTHX_ struct coro *coro)	874	coro_destruct (pTHX_ struct coro *coro)
895	{	875	{
…		…
919		899
920	SvREFCNT_dec (PL_diehook);	900	SvREFCNT_dec (PL_diehook);
921	SvREFCNT_dec (PL_warnhook);	901	SvREFCNT_dec (PL_warnhook);
922		902
923	SvREFCNT_dec (coro->saved_deffh);	903	SvREFCNT_dec (coro->saved_deffh);
924	SvREFCNT_dec (coro->throw);	904	SvREFCNT_dec (coro_throw);
925		905
926	coro_destruct_stacks (aTHX);	906	coro_destruct_stacks (aTHX);
927	}	907	}
928		908
929	INLINE void	909	INLINE void
…		…
939	static int	919	static int
940	runops_trace (pTHX)	920	runops_trace (pTHX)
941	{	921	{
942	COP *oldcop = 0;	922	COP *oldcop = 0;
943	int oldcxix = -2;	923	int oldcxix = -2;
944	struct coro *coro = SvSTATE (coro_current); /* trace cctx is tied to specific coro */	924	struct coro *coro = SvSTATE_current; /* trace cctx is tied to specific coro */
945	coro_cctx *cctx = coro->cctx;	925	coro_cctx *cctx = coro->cctx;
946		926
947	while ((PL_op = CALL_FPTR (PL_op->op_ppaddr) (aTHX)))	927	while ((PL_op = CALL_FPTR (PL_op->op_ppaddr) (aTHX)))
948	{	928	{
949	PERL_ASYNC_CHECK ();	929	PERL_ASYNC_CHECK ();
…		…
1059	TAINT_NOT;	1039	TAINT_NOT;
1060	return 0;	1040	return 0;
1061	}	1041	}
1062		1042
1063	static void	1043	static void
1064	prepare_set_stacklevel (struct transfer_args *ta, struct coro_cctx *cctx)	1044	prepare_set_stacklevel (struct coro_transfer_args *ta, struct coro_cctx *cctx)
1065	{	1045	{
1066	ta->prev = (struct coro *)cctx;	1046	ta->prev = (struct coro *)cctx;
1067	ta->next = 0;	1047	ta->next = 0;
1068	}	1048	}
1069		1049
…		…
1098		1078
1099	/* the tail of transfer: execute stuff we can only do after a transfer */	1079	/* the tail of transfer: execute stuff we can only do after a transfer */
1100	INLINE void	1080	INLINE void
1101	transfer_tail (pTHX)	1081	transfer_tail (pTHX)
1102	{	1082	{
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);	1083	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	}	1084	}
1119		1085
1120	/*	1086	/*
1121	* this is a _very_ stripped down perl interpreter ;)	1087	* this is a _very_ stripped down perl interpreter ;)
1122	*/	1088	*/
…		…
1129	# endif	1095	# endif
1130	#endif	1096	#endif
1131	{	1097	{
1132	dTHX;	1098	dTHX;
1133		1099
1134	/* we are the alternative tail to pp_set_stacklevel */	1100	/* normally we would need to skip the entersub here */
1135	/* so do the same things here */	1101	/* 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;	1102	/* PL_nop = PL_nop->op_next */
1140		1103
1141	/* inject a fake subroutine call to cctx_init */	1104	/* inject a fake subroutine call to cctx_init */
1142	cctx_prepare (aTHX_ (coro_cctx *)arg);	1105	cctx_prepare (aTHX_ (coro_cctx *)arg);
1143		1106
1144	/* cctx_run is the alternative tail of transfer() */	1107	/* cctx_run is the alternative tail of transfer() */
…		…
1305	/** coroutine switching *****************************************************/	1268	/** coroutine switching *****************************************************/
1306		1269
1307	static void	1270	static void
1308	transfer_check (pTHX_ struct coro *prev, struct coro *next)	1271	transfer_check (pTHX_ struct coro *prev, struct coro *next)
1309	{	1272	{
		1273	/* TODO: throwing up here is considered harmful */
		1274
1310	if (expect_true (prev != next))	1275	if (expect_true (prev != next))
1311	{	1276	{
1312	if (expect_false (!(prev->flags & (CF_RUNNING | CF_NEW))))	1277	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");	1278	croak ("Coro::State::transfer called with non-running/new prev Coro::State, but can only transfer from running or new states,");
1314		1279
1315	if (expect_false (next->flags & CF_RUNNING))	1280	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");	1281	croak ("Coro::State::transfer called with running next Coro::State, but can only transfer to inactive states,");
1317		1282
1318	if (expect_false (next->flags & CF_DESTROYED))	1283	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");	1284	croak ("Coro::State::transfer called with destroyed next Coro::State, but can only transfer to inactive states,");
1320		1285
1321	#if !PERL_VERSION_ATLEAST (5,10,0)	1286	#if !PERL_VERSION_ATLEAST (5,10,0)
1322	if (expect_false (PL_lex_state != LEX_NOTPARSING))	1287	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");	1288	croak ("Coro::State::transfer called while parsing, but this is not supported in your perl version,");
1324	#endif	1289	#endif
1325	}	1290	}
1326	}	1291	}
1327		1292
1328	/* always use the TRANSFER macro */	1293	/* always use the TRANSFER macro */
1329	static void NOINLINE	1294	static void NOINLINE /* noinline so we have a fixed stackframe */
1330	transfer (pTHX_ struct coro *prev, struct coro *next, int force_cctx)	1295	transfer (pTHX_ struct coro *prev, struct coro *next, int force_cctx)
1331	{	1296	{
1332	dSTACKLEVEL;	1297	dSTACKLEVEL;
1333		1298
1334	/* sometimes transfer is only called to set idle_sp */	1299	/* sometimes transfer is only called to set idle_sp */
1335	if (expect_false (!next))	1300	if (expect_false (!next))
1336	{	1301	{
1337	((coro_cctx *)prev)->idle_sp = stacklevel;	1302	((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 */	1303	assert (((coro_cctx *)prev)->idle_te = PL_top_env); /* just for the side-effect when asserts are enabled */
1339	}	1304	}
1340	else if (expect_true (prev != next))	1305	else if (expect_true (prev != next))
1341	{	1306	{
1342	coro_cctx *prev__cctx;	1307	coro_cctx *prev__cctx;
…		…
1349	prev->flags |= CF_RUNNING;	1314	prev->flags |= CF_RUNNING;
1350	}	1315	}
1351		1316
1352	prev->flags &= ~CF_RUNNING;	1317	prev->flags &= ~CF_RUNNING;
1353	next->flags |= CF_RUNNING;	1318	next->flags |= CF_RUNNING;
1354
1355	LOCK;
1356		1319
1357	/* first get rid of the old state */	1320	/* first get rid of the old state */
1358	save_perl (aTHX_ prev);	1321	save_perl (aTHX_ prev);
1359		1322
1360	if (expect_false (next->flags & CF_NEW))	1323	if (expect_false (next->flags & CF_NEW))
…		…
1369		1332
1370	prev__cctx = prev->cctx;	1333	prev__cctx = prev->cctx;
1371		1334
1372	/* possibly untie and reuse the cctx */	1335	/* possibly untie and reuse the cctx */
1373	if (expect_true (	1336	if (expect_true (
1374	prev__cctx->idle_sp == stacklevel	1337	prev__cctx->idle_sp == (void *)stacklevel
1375	&& !(prev__cctx->flags & CC_TRACE)	1338	&& !(prev__cctx->flags & CC_TRACE)
1376	&& !force_cctx	1339	&& !force_cctx
1377	))	1340	))
1378	{	1341	{
1379	/* I assume that stacklevel is a stronger indicator than PL_top_env changes */	1342	/* I assume that stacklevel is a stronger indicator than PL_top_env changes */
…		…
1393	++next->usecount;	1356	++next->usecount;
1394		1357
1395	if (expect_true (!next->cctx))	1358	if (expect_true (!next->cctx))
1396	next->cctx = cctx_get (aTHX);	1359	next->cctx = cctx_get (aTHX);
1397		1360
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))	1361	if (expect_false (prev__cctx != next->cctx))
1402	{	1362	{
1403	prev__cctx->top_env = PL_top_env;	1363	prev__cctx->top_env = PL_top_env;
1404	PL_top_env = next->cctx->top_env;	1364	PL_top_env = next->cctx->top_env;
1405	coro_transfer (&prev__cctx->cctx, &next->cctx->cctx);	1365	coro_transfer (&prev__cctx->cctx, &next->cctx->cctx);
…		…
1418	coro_state_destroy (pTHX_ struct coro *coro)	1378	coro_state_destroy (pTHX_ struct coro *coro)
1419	{	1379	{
1420	if (coro->flags & CF_DESTROYED)	1380	if (coro->flags & CF_DESTROYED)
1421	return 0;	1381	return 0;
1422		1382
		1383	if (coro->on_destroy)
		1384	coro->on_destroy (aTHX_ coro);
		1385
1423	coro->flags |= CF_DESTROYED;	1386	coro->flags |= CF_DESTROYED;
1424		1387
1425	if (coro->flags & CF_READY)	1388	if (coro->flags & CF_READY)
1426	{	1389	{
1427	/* reduce nready, as destroying a ready coro effectively unreadies it */	1390	/* reduce nready, as destroying a ready coro effectively unreadies it */
1428	/* alternative: look through all ready queues and remove the coro */	1391	/* alternative: look through all ready queues and remove the coro */
1429	LOCK;
1430	--coro_nready;	1392	--coro_nready;
1431	UNLOCK;
1432	}	1393	}
1433	else	1394	else
1434	coro->flags |= CF_READY; /* make sure it is NOT put into the readyqueue */	1395	coro->flags |= CF_READY; /* make sure it is NOT put into the readyqueue */
1435		1396
1436	if (coro->mainstack && coro->mainstack != main_mainstack)	1397	if (coro->mainstack && coro->mainstack != main_mainstack)
1437	{	1398	{
1438	struct coro temp;	1399	struct coro temp;
1439		1400
1440	if (coro->flags & CF_RUNNING)	1401	assert (("FATAL: tried to destroy currently running coroutine (please report)", !(coro->flags & CF_RUNNING)));
1441	croak ("FATAL: tried to destroy currently running coroutine");
1442		1402
1443	save_perl (aTHX_ &temp);	1403	save_perl (aTHX_ &temp);
1444	load_perl (aTHX_ coro);	1404	load_perl (aTHX_ coro);
1445		1405
1446	coro_destruct (aTHX_ coro);	1406	coro_destruct (aTHX_ coro);
…		…
1497	# define MGf_DUP 0	1457	# define MGf_DUP 0
1498	#endif	1458	#endif
1499	};	1459	};
1500		1460
1501	static void	1461	static void
1502	prepare_transfer (pTHX_ struct transfer_args *ta, SV *prev_sv, SV *next_sv)	1462	prepare_transfer (pTHX_ struct coro_transfer_args *ta, SV *prev_sv, SV *next_sv)
1503	{	1463	{
1504	ta->prev = SvSTATE (prev_sv);	1464	ta->prev = SvSTATE (prev_sv);
1505	ta->next = SvSTATE (next_sv);	1465	ta->next = SvSTATE (next_sv);
1506	TRANSFER_CHECK (*ta);	1466	TRANSFER_CHECK (*ta);
1507	}	1467	}
1508		1468
1509	static void	1469	static void
1510	api_transfer (SV *prev_sv, SV *next_sv)	1470	api_transfer (pTHX_ SV *prev_sv, SV *next_sv)
1511	{	1471	{
1512	dTHX;
1513	struct transfer_args ta;	1472	struct coro_transfer_args ta;
1514		1473
1515	prepare_transfer (aTHX_ &ta, prev_sv, next_sv);	1474	prepare_transfer (aTHX_ &ta, prev_sv, next_sv);
1516	TRANSFER (ta, 1);	1475	TRANSFER (ta, 1);
1517	}	1476	}
1518		1477
1519	/** Coro ********************************************************************/	1478	/** Coro ********************************************************************/
1520		1479
1521	static void	1480	INLINE void
1522	coro_enq (pTHX_ SV *coro_sv)	1481	coro_enq (pTHX_ struct coro *coro)
1523	{	1482	{
1524	av_push (coro_ready [SvSTATE (coro_sv)->prio - PRIO_MIN], coro_sv);	1483	av_push (coro_ready [coro->prio - PRIO_MIN], SvREFCNT_inc_NN (coro->hv));
1525	}	1484	}
1526		1485
1527	static SV *	1486	INLINE SV *
1528	coro_deq (pTHX)	1487	coro_deq (pTHX)
1529	{	1488	{
1530	int prio;	1489	int prio;
1531		1490
1532	for (prio = PRIO_MAX - PRIO_MIN + 1; --prio >= 0; )	1491	for (prio = PRIO_MAX - PRIO_MIN + 1; --prio >= 0; )
…		…
1535		1494
1536	return 0;	1495	return 0;
1537	}	1496	}
1538		1497
1539	static int	1498	static int
1540	api_ready (SV *coro_sv)	1499	api_ready (pTHX_ SV *coro_sv)
1541	{	1500	{
1542	dTHX;
1543	struct coro *coro;	1501	struct coro *coro;
1544	SV *sv_hook;	1502	SV *sv_hook;
1545	void (*xs_hook)(void);	1503	void (*xs_hook)(void);
1546		1504
1547	if (SvROK (coro_sv))	1505	if (SvROK (coro_sv))
…		…
1552	if (coro->flags & CF_READY)	1510	if (coro->flags & CF_READY)
1553	return 0;	1511	return 0;
1554		1512
1555	coro->flags |= CF_READY;	1513	coro->flags |= CF_READY;
1556		1514
1557	LOCK;
1558
1559	sv_hook = coro_nready ? 0 : coro_readyhook;	1515	sv_hook = coro_nready ? 0 : coro_readyhook;
1560	xs_hook = coro_nready ? 0 : coroapi.readyhook;	1516	xs_hook = coro_nready ? 0 : coroapi.readyhook;
1561		1517
1562	coro_enq (aTHX_ SvREFCNT_inc_NN (coro_sv));	1518	coro_enq (aTHX_ coro);
1563	++coro_nready;	1519	++coro_nready;
1564		1520
1565	UNLOCK;
1566
1567	if (sv_hook)	1521	if (sv_hook)
1568	{	1522	{
1569	dSP;	1523	dSP;
1570		1524
1571	ENTER;	1525	ENTER;
…		…
1585		1539
1586	return 1;	1540	return 1;
1587	}	1541	}
1588		1542
1589	static int	1543	static int
1590	api_is_ready (SV *coro_sv)	1544	api_is_ready (pTHX_ SV *coro_sv)
1591	{	1545	{
1592	dTHX;
1593
1594	return !!(SvSTATE (coro_sv)->flags & CF_READY);	1546	return !!(SvSTATE (coro_sv)->flags & CF_READY);
1595	}	1547	}
1596		1548
1597	INLINE void	1549	INLINE void
1598	prepare_schedule (pTHX_ struct transfer_args *ta)	1550	prepare_schedule (pTHX_ struct coro_transfer_args *ta)
1599	{	1551	{
1600	SV *prev_sv, *next_sv;	1552	SV *prev_sv, *next_sv;
1601		1553
1602	for (;;)	1554	for (;;)
1603	{	1555	{
1604	LOCK;
1605	next_sv = coro_deq (aTHX);	1556	next_sv = coro_deq (aTHX);
1606		1557
1607	/* nothing to schedule: call the idle handler */	1558	/* nothing to schedule: call the idle handler */
1608	if (expect_false (!next_sv))	1559	if (expect_false (!next_sv))
1609	{	1560	{
1610	dSP;	1561	dSP;
1611	UNLOCK;
1612		1562
1613	ENTER;	1563	ENTER;
1614	SAVETMPS;	1564	SAVETMPS;
1615		1565
1616	PUSHMARK (SP);	1566	PUSHMARK (SP);
…		…
1621	FREETMPS;	1571	FREETMPS;
1622	LEAVE;	1572	LEAVE;
1623	continue;	1573	continue;
1624	}	1574	}
1625		1575
1626	ta->next = SvSTATE (next_sv);	1576	ta->next = SvSTATE_hv (next_sv);
1627		1577
1628	/* cannot transfer to destroyed coros, skip and look for next */	1578	/* cannot transfer to destroyed coros, skip and look for next */
1629	if (expect_false (ta->next->flags & CF_DESTROYED))	1579	if (expect_false (ta->next->flags & CF_DESTROYED))
1630	{	1580	{
1631	UNLOCK;
1632	SvREFCNT_dec (next_sv);	1581	SvREFCNT_dec (next_sv);
1633	/* coro_nready has already been taken care of by destroy */	1582	/* coro_nready has already been taken care of by destroy */
1634	continue;	1583	continue;
1635	}	1584	}
1636		1585
1637	--coro_nready;	1586	--coro_nready;
1638	UNLOCK;
1639	break;	1587	break;
1640	}	1588	}
1641		1589
1642	/* free this only after the transfer */	1590	/* free this only after the transfer */
1643	prev_sv = SvRV (coro_current);	1591	prev_sv = SvRV (coro_current);
1644	ta->prev = SvSTATE (prev_sv);	1592	ta->prev = SvSTATE_hv (prev_sv);
1645	TRANSFER_CHECK (*ta);	1593	TRANSFER_CHECK (*ta);
1646	assert (("FATAL: next coroutine isn't marked as ready in Coro (please report)", ta->next->flags & CF_READY));	1594	assert (("FATAL: next coroutine isn't marked as ready in Coro (please report)", ta->next->flags & CF_READY));
1647	ta->next->flags &= ~CF_READY;	1595	ta->next->flags &= ~CF_READY;
1648	SvRV_set (coro_current, next_sv);	1596	SvRV_set (coro_current, next_sv);
1649		1597
1650	LOCK;
1651	free_coro_mortal (aTHX);	1598	free_coro_mortal (aTHX);
1652	coro_mortal = prev_sv;	1599	coro_mortal = prev_sv;
1653	UNLOCK;
1654	}	1600	}
1655		1601
1656	INLINE void	1602	INLINE void
1657	prepare_cede (pTHX_ struct transfer_args *ta)	1603	prepare_cede (pTHX_ struct coro_transfer_args *ta)
1658	{	1604	{
1659	api_ready (coro_current);	1605	api_ready (aTHX_ coro_current);
1660	prepare_schedule (aTHX_ ta);	1606	prepare_schedule (aTHX_ ta);
1661	}	1607	}
1662		1608
1663	static void	1609	INLINE void
1664	prepare_cede_notself (pTHX_ struct transfer_args *ta)	1610	prepare_cede_notself (pTHX_ struct coro_transfer_args *ta)
1665	{	1611	{
1666	SV *prev = SvRV (coro_current);	1612	SV *prev = SvRV (coro_current);
1667		1613
1668	if (coro_nready)	1614	if (coro_nready)
1669	{	1615	{
1670	prepare_schedule (aTHX_ ta);	1616	prepare_schedule (aTHX_ ta);
1671	api_ready (prev);	1617	api_ready (aTHX_ prev);
1672	}	1618	}
1673	else	1619	else
1674	ta->prev = ta->next = SvSTATE (prev);	1620	prepare_nop (aTHX_ ta);
1675	}	1621	}
1676		1622
1677	static void	1623	static void
1678	api_schedule (void)	1624	api_schedule (pTHX)
1679	{	1625	{
1680	dTHX;
1681	struct transfer_args ta;	1626	struct coro_transfer_args ta;
1682		1627
1683	prepare_schedule (aTHX_ &ta);	1628	prepare_schedule (aTHX_ &ta);
1684	TRANSFER (ta, 1);	1629	TRANSFER (ta, 1);
1685	}	1630	}
1686		1631
1687	static int	1632	static int
1688	api_cede (void)	1633	api_cede (pTHX)
1689	{	1634	{
1690	dTHX;
1691	struct transfer_args ta;	1635	struct coro_transfer_args ta;
1692		1636
1693	prepare_cede (aTHX_ &ta);	1637	prepare_cede (aTHX_ &ta);
1694		1638
1695	if (expect_true (ta.prev != ta.next))	1639	if (expect_true (ta.prev != ta.next))
1696	{	1640	{
…		…
1700	else	1644	else
1701	return 0;	1645	return 0;
1702	}	1646	}
1703		1647
1704	static int	1648	static int
1705	api_cede_notself (void)	1649	api_cede_notself (pTHX)
1706	{	1650	{
1707	if (coro_nready)	1651	if (coro_nready)
1708	{	1652	{
1709	dTHX;
1710	struct transfer_args ta;	1653	struct coro_transfer_args ta;
1711		1654
1712	prepare_cede_notself (aTHX_ &ta);	1655	prepare_cede_notself (aTHX_ &ta);
1713	TRANSFER (ta, 1);	1656	TRANSFER (ta, 1);
1714	return 1;	1657	return 1;
1715	}	1658	}
1716	else	1659	else
1717	return 0;	1660	return 0;
1718	}	1661	}
1719		1662
1720	static void	1663	static void
1721	api_trace (SV *coro_sv, int flags)	1664	api_trace (pTHX_ SV *coro_sv, int flags)
1722	{	1665	{
1723	dTHX;
1724	struct coro *coro = SvSTATE (coro_sv);	1666	struct coro *coro = SvSTATE (coro_sv);
1725		1667
1726	if (flags & CC_TRACE)	1668	if (flags & CC_TRACE)
1727	{	1669	{
1728	if (!coro->cctx)	1670	if (!coro->cctx)
1729	coro->cctx = cctx_new_run ();	1671	coro->cctx = cctx_new_run ();
1730	else if (!(coro->cctx->flags & CC_TRACE))	1672	else if (!(coro->cctx->flags & CC_TRACE))
1731	croak ("cannot enable tracing on coroutine with custom stack");	1673	croak ("cannot enable tracing on coroutine with custom stack,");
1732		1674
1733	coro->cctx->flags |= CC_NOREUSE | (flags & (CC_TRACE | CC_TRACE_ALL));	1675	coro->cctx->flags |= CC_NOREUSE | (flags & (CC_TRACE | CC_TRACE_ALL));
1734	}	1676	}
1735	else if (coro->cctx && coro->cctx->flags & CC_TRACE)	1677	else if (coro->cctx && coro->cctx->flags & CC_TRACE)
1736	{	1678	{
…		…
1741	else	1683	else
1742	coro->slot->runops = RUNOPS_DEFAULT;	1684	coro->slot->runops = RUNOPS_DEFAULT;
1743	}	1685	}
1744	}	1686	}
1745		1687
1746	#if 0	1688	/*****************************************************************************/
		1689	/* schedule-like-function opcode (SLF) */
		1690
		1691	static UNOP slf_restore; /* restore stack as entersub did, for first-re-run */
		1692	static const CV *slf_cv;
		1693	static SV **slf_argv;
		1694	static int slf_argc, slf_arga; /* count, allocated */
		1695	static I32 slf_ax; /* top of stack, for restore */
		1696
		1697	/* this restores the stack in the case we patched the entersub, to */
		1698	/* recreate the stack frame as perl will on following calls */
		1699	/* since entersub cleared the stack */
		1700	static OP *
		1701	pp_restore (pTHX)
		1702	{
		1703	int i;
		1704	SV **SP = PL_stack_base + slf_ax;
		1705
		1706	PUSHMARK (SP);
		1707
		1708	EXTEND (SP, slf_argc + 1);
		1709
		1710	for (i = 0; i < slf_argc; ++i)
		1711	PUSHs (sv_2mortal (slf_argv [i]));
		1712
		1713	PUSHs ((SV *)CvGV (slf_cv));
		1714
		1715	RETURNOP (slf_restore.op_first);
		1716	}
		1717
1747	static int	1718	static void
1748	coro_gensub_free (pTHX_ SV *sv, MAGIC *mg)	1719	slf_prepare_set_stacklevel (pTHX_ struct coro_transfer_args *ta)
1749	{	1720	{
1750	AV *padlist;	1721	prepare_set_stacklevel (ta, (struct coro_cctx *)slf_frame.data);
1751	AV *av = (AV *)mg->mg_obj;
1752
1753	abort ();
1754
1755	return 0;
1756	}	1722	}
1757		1723
1758	static MGVTBL coro_gensub_vtbl = {	1724	static void
1759	0, 0, 0, 0,	1725	slf_init_set_stacklevel (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
1760	coro_gensub_free	1726	{
1761	};	1727	assert (("FATAL: set_stacklevel needs the coro cctx as sole argument", items == 1));
1762	#endif	1728
		1729	frame->prepare = slf_prepare_set_stacklevel;
		1730	frame->check = slf_check_nop;
		1731	frame->data = (void *)SvIV (arg [0]);
		1732	}
		1733
		1734	static void
		1735	slf_prepare_transfer (pTHX_ struct coro_transfer_args *ta)
		1736	{
		1737	SV **arg = (SV **)slf_frame.data;
		1738
		1739	prepare_transfer (aTHX_ ta, arg [0], arg [1]);
		1740	}
		1741
		1742	static void
		1743	slf_init_transfer (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		1744	{
		1745	if (items != 2)
		1746	croak ("Coro::State::transfer (prev, next) expects two arguments, not %d,", items);
		1747
		1748	frame->prepare = slf_prepare_transfer;
		1749	frame->check = slf_check_nop;
		1750	frame->data = (void *)arg; /* let's hope it will stay valid */
		1751	}
		1752
		1753	static void
		1754	slf_init_schedule (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		1755	{
		1756	frame->prepare = prepare_schedule;
		1757	frame->check = slf_check_nop;
		1758	}
		1759
		1760	static void
		1761	slf_init_cede (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		1762	{
		1763	frame->prepare = prepare_cede;
		1764	frame->check = slf_check_nop;
		1765	}
		1766
		1767	static void
		1768	slf_init_cede_notself (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		1769	{
		1770	frame->prepare = prepare_cede_notself;
		1771	frame->check = slf_check_nop;
		1772	}
		1773
		1774	/* we hijack an hopefully unused CV flag for our purposes */
		1775	#define CVf_SLF 0x4000
		1776
		1777	/*
		1778	* these not obviously related functions are all rolled into one
		1779	* function to increase chances that they all will call transfer with the same
		1780	* stack offset
		1781	* SLF stands for "schedule-like-function".
		1782	*/
		1783	static OP *
		1784	pp_slf (pTHX)
		1785	{
		1786	I32 checkmark; /* mark SP to see how many elements check has pushed */
		1787
		1788	/* set up the slf frame, unless it has already been set-up */
		1789	/* the latter happens when a new coro has been started */
		1790	/* or when a new cctx was attached to an existing coroutine */
		1791	if (expect_true (!slf_frame.prepare))
		1792	{
		1793	/* first iteration */
		1794	dSP;
		1795	SV **arg = PL_stack_base + TOPMARK + 1;
		1796	int items = SP - arg; /* args without function object */
		1797	SV *gv = *sp;
		1798
		1799	/* do a quick consistency check on the "function" object, and if it isn't */
		1800	/* for us, divert to the real entersub */
		1801	if (SvTYPE (gv) != SVt_PVGV
		1802	|| !GvCV (gv)
		1803	|| !(CvFLAGS (GvCV (gv)) & CVf_SLF))
		1804	return PL_ppaddr[OP_ENTERSUB](aTHX);
		1805
		1806	if (!(PL_op->op_flags & OPf_STACKED))
		1807	{
		1808	/* ampersand-form of call, use @_ instead of stack */
		1809	AV *av = GvAV (PL_defgv);
		1810	arg = AvARRAY (av);
		1811	items = AvFILLp (av) + 1;
		1812	}
		1813
		1814	/* now call the init function, which needs to set up slf_frame */
		1815	((coro_slf_cb)CvXSUBANY (GvCV (gv)).any_ptr)
		1816	(aTHX_ &slf_frame, GvCV (gv), arg, items);
		1817
		1818	/* pop args */
		1819	SP = PL_stack_base + POPMARK;
		1820
		1821	PUTBACK;
		1822	}
		1823
		1824	/* now that we have a slf_frame, interpret it! */
		1825	/* we use a callback system not to make the code needlessly */
		1826	/* complicated, but so we can run multiple perl coros from one cctx */
		1827
		1828	do
		1829	{
		1830	struct coro_transfer_args ta;
		1831
		1832	slf_frame.prepare (aTHX_ &ta);
		1833	TRANSFER (ta, 0);
		1834
		1835	checkmark = PL_stack_sp - PL_stack_base;
		1836	}
		1837	while (slf_frame.check (aTHX_ &slf_frame));
		1838
		1839	slf_frame.prepare = 0; /* invalidate the frame, we are done processing it */
		1840
		1841	/* return value handling - mostly like entersub */
		1842	{
		1843	dSP;
		1844	SV **bot = PL_stack_base + checkmark;
		1845	int gimme = GIMME_V;
		1846
		1847	/* make sure we put something on the stack in scalar context */
		1848	if (gimme == G_SCALAR)
		1849	{
		1850	if (sp == bot)
		1851	XPUSHs (&PL_sv_undef);
		1852
		1853	SP = bot + 1;
		1854	}
		1855
		1856	PUTBACK;
		1857	}
		1858
		1859	/* exception handling */
		1860	if (expect_false (coro_throw))
		1861	{
		1862	SV *exception = sv_2mortal (coro_throw);
		1863
		1864	coro_throw = 0;
		1865	sv_setsv (ERRSV, exception);
		1866	croak (0);
		1867	}
		1868
		1869	return NORMAL;
		1870	}
		1871
		1872	static void
		1873	api_execute_slf (pTHX_ CV *cv, coro_slf_cb init_cb, I32 ax)
		1874	{
		1875	int i;
		1876	SV **arg = PL_stack_base + ax;
		1877	int items = PL_stack_sp - arg + 1;
		1878
		1879	assert (("FATAL: SLF call with illegal CV value", !CvANON (cv)));
		1880
		1881	if (PL_op->op_ppaddr != PL_ppaddr [OP_ENTERSUB]
		1882	&& PL_op->op_ppaddr != pp_slf)
		1883	croak ("FATAL: Coro SLF calls can only be made normally, not via goto or any other means, caught");
		1884
		1885	CvFLAGS (cv) |= CVf_SLF;
		1886	CvXSUBANY (cv).any_ptr = (void *)init_cb;
		1887	slf_cv = cv;
		1888
		1889	/* we patch the op, and then re-run the whole call */
		1890	/* we have to put the same argument on the stack for this to work */
		1891	/* and this will be done by pp_restore */
		1892	slf_restore.op_next = (OP *)&slf_restore;
		1893	slf_restore.op_type = OP_CUSTOM;
		1894	slf_restore.op_ppaddr = pp_restore;
		1895	slf_restore.op_first = PL_op;
		1896
		1897	slf_ax = ax - 1; /* undo the ax++ inside dAXMARK */
		1898
		1899	if (PL_op->op_flags & OPf_STACKED)
		1900	{
		1901	if (items > slf_arga)
		1902	{
		1903	slf_arga = items;
		1904	free (slf_argv);
		1905	slf_argv = malloc (slf_arga * sizeof (SV *));
		1906	}
		1907
		1908	slf_argc = items;
		1909
		1910	for (i = 0; i < items; ++i)
		1911	slf_argv [i] = SvREFCNT_inc (arg [i]);
		1912	}
		1913	else
		1914	slf_argc = 0;
		1915
		1916	PL_op->op_ppaddr = pp_slf;
		1917	PL_op->op_type = OP_CUSTOM; /* maybe we should leave it at entersub? */
		1918
		1919	PL_op = (OP *)&slf_restore;
		1920	}
1763		1921
1764	/*****************************************************************************/	1922	/*****************************************************************************/
1765	/* PerlIO::cede */	1923	/* PerlIO::cede */
1766		1924
1767	typedef struct	1925	typedef struct
…		…
1795	PerlIOCede *self = PerlIOSelf (f, PerlIOCede);	1953	PerlIOCede *self = PerlIOSelf (f, PerlIOCede);
1796	double now = nvtime ();	1954	double now = nvtime ();
1797		1955
1798	if (now >= self->next)	1956	if (now >= self->next)
1799	{	1957	{
1800	api_cede ();	1958	api_cede (aTHX);
1801	self->next = now + self->every;	1959	self->next = now + self->every;
1802	}	1960	}
1803		1961
1804	return PerlIOBuf_flush (aTHX_ f);	1962	return PerlIOBuf_flush (aTHX_ f);
1805	}	1963	}
…		…
1835	PerlIOBuf_get_cnt,	1993	PerlIOBuf_get_cnt,
1836	PerlIOBuf_set_ptrcnt,	1994	PerlIOBuf_set_ptrcnt,
1837	};	1995	};
1838		1996
1839	/*****************************************************************************/	1997	/*****************************************************************************/
		1998	/* Coro::Semaphore */
1840		1999
1841	static const CV *slf_cv; /* for quick consistency check */	2000	static void
		2001	coro_semaphore_adjust (pTHX_ AV *av, IV adjust)
		2002	{
		2003	SV *count_sv = AvARRAY (av)[0];
		2004	IV count = SvIVX (count_sv);
1842		2005
1843	static UNOP slf_restore; /* restore stack as entersub did, for first-re-run */	2006	count += adjust;
1844	static SV *slf_arg0;	2007	SvIVX (count_sv) = count;
1845	static SV *slf_arg1;
1846		2008
1847	/* this restores the stack in the case we patched the entersub, to */	2009	/* now wake up as many waiters as are expected to lock */
1848	/* recreate the stack frame as perl will on following calls */	2010	while (count > 0 && AvFILLp (av) > 0)
1849	/* since entersub cleared the stack */	2011	{
		2012	SV *cb;
		2013
		2014	/* swap first two elements so we can shift a waiter */
		2015	AvARRAY (av)[0] = AvARRAY (av)[1];
		2016	AvARRAY (av)[1] = count_sv;
		2017	cb = av_shift (av);
		2018
		2019	if (SvOBJECT (cb))
		2020	api_ready (aTHX_ cb);
		2021	else
		2022	croak ("callbacks not yet supported");
		2023
		2024	SvREFCNT_dec (cb);
		2025
		2026	--count;
		2027	}
		2028	}
		2029
		2030	static void
		2031	coro_semaphore_on_destroy (pTHX_ struct coro *coro)
		2032	{
		2033	/* call $sem->adjust (0) to possibly wake up some other waiters */
		2034	coro_semaphore_adjust (aTHX_ (AV *)coro->slf_frame.data, 0);
		2035	}
		2036
		2037	static int
		2038	slf_check_semaphore_down (pTHX_ struct CoroSLF *frame)
		2039	{
		2040	AV *av = (AV *)frame->data;
		2041	SV *count_sv = AvARRAY (av)[0];
		2042
		2043	if (SvIVX (count_sv) > 0)
		2044	{
		2045	SvSTATE_current->on_destroy = 0;
		2046	SvIVX (count_sv) = SvIVX (count_sv) - 1;
		2047	return 0;
		2048	}
		2049	else
		2050	{
		2051	int i;
		2052	/* if we were woken up but can't down, we look through the whole */
		2053	/* waiters list and only add us if we aren't in there already */
		2054	/* this avoids some degenerate memory usage cases */
		2055
		2056	for (i = 1; i <= AvFILLp (av); ++i)
		2057	if (AvARRAY (av)[i] == SvRV (coro_current))
		2058	return 1;
		2059
		2060	av_push (av, SvREFCNT_inc (SvRV (coro_current)));
		2061	return 1;
		2062	}
		2063	}
		2064
		2065	static void
		2066	slf_init_semaphore_down (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		2067	{
		2068	AV *av = (AV *)SvRV (arg [0]);
		2069
		2070	if (SvIVX (AvARRAY (av)[0]) > 0)
		2071	{
		2072	frame->data = (void *)av;
		2073	frame->prepare = prepare_nop;
		2074	SvSTATE_current->on_destroy = coro_semaphore_on_destroy;
		2075	}
		2076	else
		2077	{
		2078	av_push (av, SvREFCNT_inc (SvRV (coro_current)));
		2079
		2080	frame->data = (void *)sv_2mortal (SvREFCNT_inc ((SV *)av));
		2081	frame->prepare = prepare_schedule;
		2082
		2083	/* to avoid race conditions when a woken-up coro gets terminated */
		2084	/* we arrange for a temporary on_destroy that calls adjust (0) */
		2085	assert (!SvSTATE_current->on_destroy);//D
		2086	SvSTATE_current->on_destroy = coro_semaphore_on_destroy;
		2087	}
		2088
		2089	frame->check = slf_check_semaphore_down;
		2090
		2091	}
		2092
		2093	/*****************************************************************************/
		2094	/* gensub: simple closure generation utility */
		2095
		2096	#define GENSUB_ARG CvXSUBANY (cv).any_ptr
		2097
		2098	/* create a closure from XS, returns a code reference */
		2099	/* the arg can be accessed via GENSUB_ARG from the callback */
		2100	/* the callback must use dXSARGS/XSRETURN */
1850	static OP *	2101	static SV *
1851	pp_restore (pTHX)	2102	gensub (pTHX_ void (*xsub)(pTHX_ CV *), void *arg)
1852	{	2103	{
		2104	CV *cv = (CV *)newSV (0);
		2105
		2106	sv_upgrade ((SV *)cv, SVt_PVCV);
		2107
		2108	CvANON_on (cv);
		2109	CvISXSUB_on (cv);
		2110	CvXSUB (cv) = xsub;
		2111	GENSUB_ARG = arg;
		2112
		2113	return newRV_noinc ((SV *)cv);
		2114	}
		2115
		2116	/*****************************************************************************/
		2117	/* Coro::AIO */
		2118
		2119	#define CORO_MAGIC_type_aio PERL_MAGIC_ext
		2120
		2121	/* helper storage struct */
		2122	struct io_state
		2123	{
		2124	int errorno;
		2125	I32 laststype; /* U16 in 5.10.0 */
		2126	int laststatval;
		2127	Stat_t statcache;
		2128	};
		2129
		2130	static void
		2131	coro_aio_callback (pTHX_ CV *cv)
		2132	{
		2133	dXSARGS;
		2134	AV *state = (AV *)GENSUB_ARG;
		2135	SV *coro = av_pop (state);
		2136	SV *data_sv = newSV (sizeof (struct io_state));
		2137
		2138	av_extend (state, items);
		2139
		2140	sv_upgrade (data_sv, SVt_PV);
		2141	SvCUR_set (data_sv, sizeof (struct io_state));
		2142	SvPOK_only (data_sv);
		2143
		2144	{
		2145	struct io_state *data = (struct io_state *)SvPVX (data_sv);
		2146
		2147	data->errorno = errno;
		2148	data->laststype = PL_laststype;
		2149	data->laststatval = PL_laststatval;
		2150	data->statcache = PL_statcache;
		2151	}
		2152
		2153	/* now build the result vector out of all the parameters and the data_sv */
		2154	{
		2155	int i;
		2156
		2157	for (i = 0; i < items; ++i)
		2158	av_push (state, SvREFCNT_inc_NN (ST (i)));
		2159	}
		2160
		2161	av_push (state, data_sv);
		2162
		2163	api_ready (aTHX_ coro);
		2164	SvREFCNT_dec (coro);
		2165	SvREFCNT_dec ((AV *)state);
		2166	}
		2167
		2168	static int
		2169	slf_check_aio_req (pTHX_ struct CoroSLF *frame)
		2170	{
		2171	AV *state = (AV *)frame->data;
		2172
		2173	/* one element that is an RV? repeat! */
		2174	if (AvFILLp (state) == 0 && SvROK (AvARRAY (state)[0]))
		2175	return 1;
		2176
		2177	/* restore status */
		2178	{
		2179	SV *data_sv = av_pop (state);
		2180	struct io_state *data = (struct io_state *)SvPVX (data_sv);
		2181
		2182	errno = data->errorno;
		2183	PL_laststype = data->laststype;
		2184	PL_laststatval = data->laststatval;
		2185	PL_statcache = data->statcache;
		2186
		2187	SvREFCNT_dec (data_sv);
		2188	}
		2189
		2190	/* push result values */
		2191	{
1853	dSP;	2192	dSP;
		2193	int i;
1854		2194
		2195	EXTEND (SP, AvFILLp (state) + 1);
		2196	for (i = 0; i <= AvFILLp (state); ++i)
		2197	PUSHs (sv_2mortal (SvREFCNT_inc_NN (AvARRAY (state)[i])));
		2198
		2199	PUTBACK;
		2200	}
		2201
		2202	return 0;
		2203	}
		2204
		2205	static void
		2206	slf_init_aio_req (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		2207	{
		2208	AV *state = (AV *)sv_2mortal ((SV *)newAV ());
		2209	SV *coro_hv = SvRV (coro_current);
		2210	struct coro *coro = SvSTATE_hv (coro_hv);
		2211
		2212	/* put our coroutine id on the state arg */
		2213	av_push (state, SvREFCNT_inc_NN (coro_hv));
		2214
		2215	/* first see whether we have a non-zero priority and set it as AIO prio */
		2216	if (coro->prio)
		2217	{
		2218	dSP;
		2219
		2220	static SV *prio_cv;
		2221	static SV *prio_sv;
		2222
		2223	if (expect_false (!prio_cv))
		2224	{
		2225	prio_cv = (SV *)get_cv ("IO::AIO::aioreq_pri", 0);
		2226	prio_sv = newSViv (0);
		2227	}
		2228
		2229	PUSHMARK (SP);
		2230	sv_setiv (prio_sv, coro->prio);
		2231	XPUSHs (prio_sv);
		2232
		2233	PUTBACK;
		2234	call_sv (prio_cv, G_VOID | G_DISCARD);
		2235	}
		2236
		2237	/* now call the original request */
		2238	{
		2239	dSP;
		2240	CV *req = (CV *)CORO_MAGIC_NN ((SV *)cv, CORO_MAGIC_type_aio)->mg_obj;
		2241	int i;
		2242
1855	PUSHMARK (SP);	2243	PUSHMARK (SP);
1856		2244
1857	EXTEND (SP, 3);	2245	/* first push all args to the stack */
1858	if (slf_arg0) PUSHs (sv_2mortal (slf_arg0));	2246	EXTEND (SP, items + 1);
1859	if (slf_arg1) PUSHs (sv_2mortal (slf_arg1));
1860	PUSHs ((SV *)CvGV (slf_cv));
1861		2247
1862	RETURNOP (slf_restore.op_first);	2248	for (i = 0; i < items; ++i)
1863	}	2249	PUSHs (arg [i]);
1864		2250
1865	#define OPpENTERSUB_SLF 15 /* the part of op_private entersub hopefully doesn't use */	2251	/* now push the callback closure */
		2252	PUSHs (sv_2mortal (gensub (coro_aio_callback, (void *)SvREFCNT_inc_NN ((SV *)state))));
1866		2253
1867	/* declare prototype */	2254	/* now call the AIO function - we assume our request is uncancelable */
1868	XS(XS_Coro__State__set_stacklevel);
1869
1870	/*
1871	* these not obviously related functions are all rolled into one
1872	* function to increase chances that they all will call transfer with the same
1873	* stack offset
1874	* SLF stands for "schedule-like-function".
1875	*/
1876	static OP *
1877	pp_slf (pTHX)
1878	{
1879	dSP;
1880	struct transfer_args ta;
1881	SV **arg = PL_stack_base + TOPMARK + 1;
1882	int items = SP - arg; /* args without function object */
1883
1884	/* do a quick consistency check on the "function" object, and if it isn't */
1885	/* for us, divert to the real entersub */
1886	if (SvTYPE (*sp) != SVt_PVGV || CvXSUB (GvCV (*sp)) != XS_Coro__State__set_stacklevel)
1887	return PL_ppaddr[OP_ENTERSUB](aTHX);
1888
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;	2255	PUTBACK;
1901	switch (PL_op->op_private & OPpENTERSUB_SLF)	2256	call_sv ((SV *)req, G_VOID | G_DISCARD);
1902	{
1903	case 0:
1904	prepare_set_stacklevel (&ta, (struct coro_cctx *)SvIV (arg [0]));
1905	break;
1906
1907	case 1:
1908	if (items != 2)
1909	croak ("Coro::State::transfer (prev, next) expects two arguments, not %d.", items);
1910
1911	prepare_transfer (aTHX_ &ta, arg [0], arg [1]);
1912	break;
1913
1914	case 2:
1915	prepare_schedule (aTHX_ &ta);
1916	break;
1917
1918	case 3:
1919	prepare_cede (aTHX_ &ta);
1920	break;
1921
1922	case 4:
1923	prepare_cede_notself (aTHX_ &ta);
1924	break;
1925
1926	case 5:
1927	abort ();
1928
1929	default:
1930	abort ();
1931	}	2257	}
1932		2258
1933	TRANSFER (ta, 0);	2259	/* now that the requets is going, we loop toll we have a result */
1934	SPAGAIN;	2260	frame->data = (void *)state;
1935		2261	frame->prepare = prepare_schedule;
1936	PUTBACK;	2262	frame->check = slf_check_aio_req;
1937	SLF_TAIL;
1938	SPAGAIN;
1939	RETURN;
1940	}	2263	}
1941		2264
1942	static void	2265	static void
1943	coro_slf_patch (pTHX_ CV *cv, int ix, SV **args, int items)	2266	coro_aio_req_xs (pTHX_ CV *cv)
1944	{	2267	{
1945	assert (("FATAL: SLF call recursion in Coro module (please report)", PL_op->op_ppaddr != pp_slf));	2268	dVAR;
		2269	dXSARGS;
1946		2270
1947	assert (("FATAL: SLF call with illegal CV value", CvGV (cv)));	2271	CORO_EXECUTE_SLF_XS (slf_init_aio_req);
1948	slf_cv = cv;
1949		2272
1950	/* we patch the op, and then re-run the whole call */	2273	XSRETURN_EMPTY;
1951	/* we have to put the same argument on the stack for this to work */
1952	/* and this will be done by pp_restore */
1953	slf_restore.op_next = (OP *)&slf_restore;
1954	slf_restore.op_type = OP_NULL;
1955	slf_restore.op_ppaddr = pp_restore;
1956	slf_restore.op_first = PL_op;
1957
1958	slf_arg0 = items > 0 ? SvREFCNT_inc (args [0]) : 0;
1959	slf_arg1 = items > 1 ? SvREFCNT_inc (args [1]) : 0;
1960
1961	PL_op->op_ppaddr = pp_slf;
1962	PL_op->op_private = PL_op->op_private & ~OPpENTERSUB_SLF | ix; /* we potentially share our private flags with entersub */
1963
1964	PL_op = (OP *)&slf_restore;
1965	}	2274	}
		2275
		2276	/*****************************************************************************/
1966		2277
1967	MODULE = Coro::State PACKAGE = Coro::State PREFIX = api_	2278	MODULE = Coro::State PACKAGE = Coro::State PREFIX = api_
1968		2279
1969	PROTOTYPES: DISABLE	2280	PROTOTYPES: DISABLE
1970		2281
1971	BOOT:	2282	BOOT:
1972	{	2283	{
1973	#ifdef USE_ITHREADS	2284	#ifdef USE_ITHREADS
1974	MUTEX_INIT (&coro_lock);
1975	# if CORO_PTHREAD	2285	# if CORO_PTHREAD
1976	coro_thx = PERL_GET_CONTEXT;	2286	coro_thx = PERL_GET_CONTEXT;
1977	# endif	2287	# endif
1978	#endif	2288	#endif
1979	BOOT_PAGESIZE;	2289	BOOT_PAGESIZE;
…		…
2000	main_top_env = PL_top_env;	2310	main_top_env = PL_top_env;
2001		2311
2002	while (main_top_env->je_prev)	2312	while (main_top_env->je_prev)
2003	main_top_env = main_top_env->je_prev;	2313	main_top_env = main_top_env->je_prev;
2004		2314
		2315	{
		2316	SV *slf = sv_2mortal (newSViv (PTR2IV (pp_slf)));
		2317
		2318	if (!PL_custom_op_names) PL_custom_op_names = newHV ();
		2319	hv_store_ent (PL_custom_op_names, slf,
		2320	newSVpv ("coro_slf", 0), 0);
		2321
		2322	if (!PL_custom_op_descs) PL_custom_op_descs = newHV ();
		2323	hv_store_ent (PL_custom_op_descs, slf,
		2324	newSVpv ("coro schedule like function", 0), 0);
		2325	}
		2326
2005	coroapi.ver = CORO_API_VERSION;	2327	coroapi.ver = CORO_API_VERSION;
2006	coroapi.rev = CORO_API_REVISION;	2328	coroapi.rev = CORO_API_REVISION;
		2329
2007	coroapi.transfer = api_transfer;	2330	coroapi.transfer = api_transfer;
		2331
		2332	coroapi.sv_state = SvSTATE_;
		2333	coroapi.execute_slf = api_execute_slf;
		2334	coroapi.prepare_nop = prepare_nop;
		2335	coroapi.prepare_schedule = prepare_schedule;
		2336	coroapi.prepare_cede = prepare_cede;
		2337	coroapi.prepare_cede_notself = prepare_cede_notself;
2008		2338
2009	{	2339	{
2010	SV **svp = hv_fetch (PL_modglobal, "Time::NVtime", 12, 0);	2340	SV **svp = hv_fetch (PL_modglobal, "Time::NVtime", 12, 0);
2011		2341
2012	if (!svp) croak ("Time::HiRes is required");	2342	if (!svp) croak ("Time::HiRes is required");
…		…
2047	OUTPUT:	2377	OUTPUT:
2048	RETVAL	2378	RETVAL
2049		2379
2050	void	2380	void
2051	_set_stacklevel (...)	2381	_set_stacklevel (...)
2052	ALIAS:	2382	CODE:
2053	Coro::State::transfer = 1	2383	CORO_EXECUTE_SLF_XS (slf_init_set_stacklevel);
2054	Coro::schedule = 2	2384
2055	Coro::cede = 3	2385	void
2056	Coro::cede_notself = 4	2386	transfer (...)
2057	CODE:	2387	PROTOTYPE: $$
2058	coro_slf_patch (aTHX_ cv, ix, &ST (0), items);	2388	CODE:
		2389	CORO_EXECUTE_SLF_XS (slf_init_transfer);
2059		2390
2060	bool	2391	bool
2061	_destroy (SV *coro_sv)	2392	_destroy (SV *coro_sv)
2062	CODE:	2393	CODE:
2063	RETVAL = coro_state_destroy (aTHX_ SvSTATE (coro_sv));	2394	RETVAL = coro_state_destroy (aTHX_ SvSTATE (coro_sv));
…		…
2070	CODE:	2401	CODE:
2071	_exit (code);	2402	_exit (code);
2072		2403
2073	int	2404	int
2074	cctx_stacksize (int new_stacksize = 0)	2405	cctx_stacksize (int new_stacksize = 0)
		2406	PROTOTYPE: ;$
2075	CODE:	2407	CODE:
2076	RETVAL = cctx_stacksize;	2408	RETVAL = cctx_stacksize;
2077	if (new_stacksize)	2409	if (new_stacksize)
2078	{	2410	{
2079	cctx_stacksize = new_stacksize;	2411	cctx_stacksize = new_stacksize;
…		…
2082	OUTPUT:	2414	OUTPUT:
2083	RETVAL	2415	RETVAL
2084		2416
2085	int	2417	int
2086	cctx_max_idle (int max_idle = 0)	2418	cctx_max_idle (int max_idle = 0)
		2419	PROTOTYPE: ;$
2087	CODE:	2420	CODE:
2088	RETVAL = cctx_max_idle;	2421	RETVAL = cctx_max_idle;
2089	if (max_idle > 1)	2422	if (max_idle > 1)
2090	cctx_max_idle = max_idle;	2423	cctx_max_idle = max_idle;
2091	OUTPUT:	2424	OUTPUT:
2092	RETVAL	2425	RETVAL
2093		2426
2094	int	2427	int
2095	cctx_count ()	2428	cctx_count ()
		2429	PROTOTYPE:
2096	CODE:	2430	CODE:
2097	RETVAL = cctx_count;	2431	RETVAL = cctx_count;
2098	OUTPUT:	2432	OUTPUT:
2099	RETVAL	2433	RETVAL
2100		2434
2101	int	2435	int
2102	cctx_idle ()	2436	cctx_idle ()
		2437	PROTOTYPE:
2103	CODE:	2438	CODE:
2104	RETVAL = cctx_idle;	2439	RETVAL = cctx_idle;
2105	OUTPUT:	2440	OUTPUT:
2106	RETVAL	2441	RETVAL
2107		2442
2108	void	2443	void
2109	list ()	2444	list ()
		2445	PROTOTYPE:
2110	PPCODE:	2446	PPCODE:
2111	{	2447	{
2112	struct coro *coro;	2448	struct coro *coro;
2113	for (coro = coro_first; coro; coro = coro->next)	2449	for (coro = coro_first; coro; coro = coro->next)
2114	if (coro->hv)	2450	if (coro->hv)
…		…
2176		2512
2177	void	2513	void
2178	throw (Coro::State self, SV *throw = &PL_sv_undef)	2514	throw (Coro::State self, SV *throw = &PL_sv_undef)
2179	PROTOTYPE: $;$	2515	PROTOTYPE: $;$
2180	CODE:	2516	CODE:
		2517	{
		2518	struct coro *current = SvSTATE_current;
		2519	SV **throwp = self == current ? &coro_throw : &self->throw;
2181	SvREFCNT_dec (self->throw);	2520	SvREFCNT_dec (*throwp);
2182	self->throw = SvOK (throw) ? newSVsv (throw) : 0;	2521	*throwp = SvOK (throw) ? newSVsv (throw) : 0;
		2522	}
2183		2523
2184	void	2524	void
2185	api_trace (SV *coro, int flags = CC_TRACE | CC_TRACE_SUB)	2525	api_trace (SV *coro, int flags = CC_TRACE | CC_TRACE_SUB)
		2526	PROTOTYPE: $;$
		2527	C_ARGS: aTHX_ coro, flags
2186		2528
2187	SV *	2529	SV *
2188	has_cctx (Coro::State coro)	2530	has_cctx (Coro::State coro)
2189	PROTOTYPE: $	2531	PROTOTYPE: $
2190	CODE:	2532	CODE:
…		…
2214	OUTPUT:	2556	OUTPUT:
2215	RETVAL	2557	RETVAL
2216		2558
2217	void	2559	void
2218	force_cctx ()	2560	force_cctx ()
		2561	PROTOTYPE:
2219	CODE:	2562	CODE:
2220	struct coro *coro = SvSTATE (coro_current);
2221	coro->cctx->idle_sp = 0;	2563	SvSTATE_current->cctx->idle_sp = 0;
2222		2564
2223	void	2565	void
2224	swap_defsv (Coro::State self)	2566	swap_defsv (Coro::State self)
2225	PROTOTYPE: $	2567	PROTOTYPE: $
2226	ALIAS:	2568	ALIAS:
2227	swap_defav = 1	2569	swap_defav = 1
2228	CODE:	2570	CODE:
2229	if (!self->slot)	2571	if (!self->slot)
2230	croak ("cannot swap state with coroutine that has no saved state");	2572	croak ("cannot swap state with coroutine that has no saved state,");
2231	else	2573	else
2232	{	2574	{
2233	SV **src = ix ? (SV **)&GvAV (PL_defgv) : &GvSV (PL_defgv);	2575	SV **src = ix ? (SV **)&GvAV (PL_defgv) : &GvSV (PL_defgv);
2234	SV **dst = ix ? (SV **)&self->slot->defav : (SV **)&self->slot->defsv;	2576	SV **dst = ix ? (SV **)&self->slot->defav : (SV **)&self->slot->defsv;
2235		2577
2236	SV *tmp = *src; *src = *dst; *dst = tmp;	2578	SV *tmp = *src; *src = *dst; *dst = tmp;
2237	}	2579	}
		2580
2238		2581
2239	MODULE = Coro::State PACKAGE = Coro	2582	MODULE = Coro::State PACKAGE = Coro
2240		2583
2241	BOOT:	2584	BOOT:
2242	{	2585	{
…		…
2267	coroapi.schedule = api_schedule;	2610	coroapi.schedule = api_schedule;
2268	coroapi.cede = api_cede;	2611	coroapi.cede = api_cede;
2269	coroapi.cede_notself = api_cede_notself;	2612	coroapi.cede_notself = api_cede_notself;
2270	coroapi.ready = api_ready;	2613	coroapi.ready = api_ready;
2271	coroapi.is_ready = api_is_ready;	2614	coroapi.is_ready = api_is_ready;
2272	coroapi.nready = &coro_nready;	2615	coroapi.nready = coro_nready;
2273	coroapi.current = coro_current;	2616	coroapi.current = coro_current;
2274		2617
2275	GCoroAPI = &coroapi;	2618	GCoroAPI = &coroapi;
2276	sv_setiv (sv, (IV)&coroapi);	2619	sv_setiv (sv, (IV)&coroapi);
2277	SvREADONLY_on (sv);	2620	SvREADONLY_on (sv);
2278	}	2621	}
2279	}	2622	}
		2623
		2624	void
		2625	schedule (...)
		2626	CODE:
		2627	CORO_EXECUTE_SLF_XS (slf_init_schedule);
		2628
		2629	void
		2630	cede (...)
		2631	CODE:
		2632	CORO_EXECUTE_SLF_XS (slf_init_cede);
		2633
		2634	void
		2635	cede_notself (...)
		2636	CODE:
		2637	CORO_EXECUTE_SLF_XS (slf_init_cede_notself);
2280		2638
2281	void	2639	void
2282	_set_current (SV *current)	2640	_set_current (SV *current)
2283	PROTOTYPE: $	2641	PROTOTYPE: $
2284	CODE:	2642	CODE:
…		…
2287		2645
2288	void	2646	void
2289	_set_readyhook (SV *hook)	2647	_set_readyhook (SV *hook)
2290	PROTOTYPE: $	2648	PROTOTYPE: $
2291	CODE:	2649	CODE:
2292	LOCK;
2293	SvREFCNT_dec (coro_readyhook);	2650	SvREFCNT_dec (coro_readyhook);
2294	coro_readyhook = SvOK (hook) ? newSVsv (hook) : 0;	2651	coro_readyhook = SvOK (hook) ? newSVsv (hook) : 0;
2295	UNLOCK;
2296		2652
2297	int	2653	int
2298	prio (Coro::State coro, int newprio = 0)	2654	prio (Coro::State coro, int newprio = 0)
		2655	PROTOTYPE: $;$
2299	ALIAS:	2656	ALIAS:
2300	nice = 1	2657	nice = 1
2301	CODE:	2658	CODE:
2302	{	2659	{
2303	RETVAL = coro->prio;	2660	RETVAL = coro->prio;
…		…
2318		2675
2319	SV *	2676	SV *
2320	ready (SV *self)	2677	ready (SV *self)
2321	PROTOTYPE: $	2678	PROTOTYPE: $
2322	CODE:	2679	CODE:
2323	RETVAL = boolSV (api_ready (self));	2680	RETVAL = boolSV (api_ready (aTHX_ self));
2324	OUTPUT:	2681	OUTPUT:
2325	RETVAL	2682	RETVAL
2326		2683
2327	int	2684	int
2328	nready (...)	2685	nready (...)
…		…
2335	# for async_pool speedup	2692	# for async_pool speedup
2336	void	2693	void
2337	_pool_1 (SV *cb)	2694	_pool_1 (SV *cb)
2338	CODE:	2695	CODE:
2339	{	2696	{
2340	struct coro *coro = SvSTATE (coro_current);
2341	HV *hv = (HV *)SvRV (coro_current);	2697	HV *hv = (HV *)SvRV (coro_current);
		2698	struct coro *coro = SvSTATE_hv ((SV *)hv);
2342	AV *defav = GvAV (PL_defgv);	2699	AV *defav = GvAV (PL_defgv);
2343	SV *invoke = hv_delete (hv, "_invoke", sizeof ("_invoke") - 1, 0);	2700	SV *invoke = hv_delete (hv, "_invoke", sizeof ("_invoke") - 1, 0);
2344	AV *invoke_av;	2701	AV *invoke_av;
2345	int i, len;	2702	int i, len;
2346		2703
…		…
2367	{	2724	{
2368	av_fill (defav, len - 1);	2725	av_fill (defav, len - 1);
2369	for (i = 0; i < len; ++i)	2726	for (i = 0; i < len; ++i)
2370	av_store (defav, i, SvREFCNT_inc_NN (AvARRAY (invoke_av)[i + 1]));	2727	av_store (defav, i, SvREFCNT_inc_NN (AvARRAY (invoke_av)[i + 1]));
2371	}	2728	}
2372
2373	SvREFCNT_dec (invoke);
2374	}	2729	}
2375		2730
2376	void	2731	void
2377	_pool_2 (SV *cb)	2732	_pool_2 (SV *cb)
2378	CODE:	2733	CODE:
2379	{	2734	{
2380	struct coro *coro = SvSTATE (coro_current);	2735	HV *hv = (HV *)SvRV (coro_current);
		2736	struct coro *coro = SvSTATE_hv ((SV *)hv);
2381		2737
2382	sv_setsv (cb, &PL_sv_undef);	2738	sv_setsv (cb, &PL_sv_undef);
2383		2739
2384	SvREFCNT_dec ((SV *)PL_defoutgv); PL_defoutgv = (GV *)coro->saved_deffh;	2740	SvREFCNT_dec ((SV *)PL_defoutgv); PL_defoutgv = (GV *)coro->saved_deffh;
2385	coro->saved_deffh = 0;	2741	coro->saved_deffh = 0;
…		…
2392	SvREFCNT_dec (old);	2748	SvREFCNT_dec (old);
2393	croak ("\3async_pool terminate\2\n");	2749	croak ("\3async_pool terminate\2\n");
2394	}	2750	}
2395		2751
2396	av_clear (GvAV (PL_defgv));	2752	av_clear (GvAV (PL_defgv));
2397	hv_store ((HV *)SvRV (coro_current), "desc", sizeof ("desc") - 1,	2753	hv_store (hv, "desc", sizeof ("desc") - 1,
2398	newSVpvn ("[async_pool idle]", sizeof ("[async_pool idle]") - 1), 0);	2754	newSVpvn ("[async_pool idle]", sizeof ("[async_pool idle]") - 1), 0);
2399		2755
2400	coro->prio = 0;	2756	coro->prio = 0;
2401		2757
2402	if (coro->cctx && (coro->cctx->flags & CC_TRACE))	2758	if (coro->cctx && (coro->cctx->flags & CC_TRACE))
2403	api_trace (coro_current, 0);	2759	api_trace (aTHX_ coro_current, 0);
2404		2760
2405	av_push (av_async_pool, newSVsv (coro_current));	2761	av_push (av_async_pool, newSVsv (coro_current));
2406	}	2762	}
2407		2763
2408	#if 0
2409		2764
2410	void	2765	MODULE = Coro::State PACKAGE = PerlIO::cede
2411	_generator_call (...)	2766
2412	PROTOTYPE: @	2767	BOOT:
2413	PPCODE:	2768	PerlIO_define_layer (aTHX_ &PerlIO_cede);
2414	fprintf (stderr, "call %p\n", CvXSUBANY(cv).any_ptr);	2769
2415	xxxx	2770
2416	abort ();	2771	MODULE = Coro::State PACKAGE = Coro::Semaphore
2417		2772
2418	SV *	2773	SV *
2419	gensub (SV *sub, ...)	2774	new (SV *klass, SV *count_ = 0)
2420	PROTOTYPE: &;@	2775	CODE:
2421	CODE:
2422	{	2776	{
2423	struct coro *coro;	2777	/* a semaphore contains a counter IV in $sem->[0] and any waiters after that */
2424	MAGIC *mg;	2778	AV *av = newAV ();
2425	CV *xcv;	2779	av_push (av, newSViv (count_ && SvOK (count_) ? SvIV (count_) : 1));
2426	CV *ncv = (CV *)newSV_type (SVt_PVCV);	2780	RETVAL = sv_bless (newRV_noinc ((SV *)av), GvSTASH (CvGV (cv)));
2427	int i;
2428
2429	CvGV (ncv) = CvGV (cv);
2430	CvFILE (ncv) = CvFILE (cv);
2431
2432	Newz (0, coro, 1, struct coro);
2433	coro->args = newAV ();
2434	coro->flags = CF_NEW;
2435
2436	av_extend (coro->args, items - 1);
2437	for (i = 1; i < items; i++)
2438	av_push (coro->args, newSVsv (ST (i)));
2439
2440	CvISXSUB_on (ncv);
2441	CvXSUBANY (ncv).any_ptr = (void *)coro;
2442
2443	xcv = GvCV (gv_fetchpv ("Coro::_generator_call", 0, SVt_PVCV));
2444
2445	CvXSUB (ncv) = CvXSUB (xcv);
2446	CvANON_on (ncv);
2447
2448	mg = sv_magicext ((SV *)ncv, 0, CORO_MAGIC_type_state, &coro_gensub_vtbl, (char *)coro, 0);
2449	RETVAL = newRV_noinc ((SV *)ncv);
2450	}	2781	}
2451	OUTPUT:	2782	OUTPUT:
2452	RETVAL	2783	RETVAL
2453		2784
2454	#endif	2785	SV *
2455		2786	count (SV *self)
2456		2787	CODE:
2457	MODULE = Coro::State PACKAGE = Coro::AIO	2788	RETVAL = newSVsv (AvARRAY ((AV *)SvRV (self))[0]);
		2789	OUTPUT:
		2790	RETVAL
2458		2791
2459	void	2792	void
2460	_get_state (SV *self)	2793	up (SV *self, int adjust = 1)
2461	PPCODE:	2794	ALIAS:
2462	{	2795	adjust = 1
2463	AV *defav = GvAV (PL_defgv);	2796	CODE:
2464	AV *av = newAV ();	2797	coro_semaphore_adjust (aTHX_ (AV *)SvRV (self), ix ? adjust : 1);
2465	int i;
2466	SV *data_sv = newSV (sizeof (struct io_state));
2467	struct io_state *data = (struct io_state *)SvPVX (data_sv);
2468	SvCUR_set (data_sv, sizeof (struct io_state));
2469	SvPOK_only (data_sv);
2470
2471	data->errorno = errno;
2472	data->laststype = PL_laststype;
2473	data->laststatval = PL_laststatval;
2474	data->statcache = PL_statcache;
2475
2476	av_extend (av, AvFILLp (defav) + 1 + 1);
2477
2478	for (i = 0; i <= AvFILLp (defav); ++i)
2479	av_push (av, SvREFCNT_inc_NN (AvARRAY (defav)[i]));
2480
2481	av_push (av, data_sv);
2482
2483	XPUSHs (sv_2mortal (newRV_noinc ((SV *)av)));
2484
2485	api_ready (self);
2486	}
2487		2798
2488	void	2799	void
2489	_set_state (SV *state)	2800	down (SV *self)
2490	PROTOTYPE: $	2801	CODE:
2491	PPCODE:	2802	CORO_EXECUTE_SLF_XS (slf_init_semaphore_down);
		2803
		2804	void
		2805	try (SV *self)
		2806	PPCODE:
2492	{	2807	{
2493	AV *av = (AV *)SvRV (state);	2808	AV *av = (AV *)SvRV (self);
2494	struct io_state *data = (struct io_state *)SvPVX (AvARRAY (av)[AvFILLp (av)]);	2809	SV *count_sv = AvARRAY (av)[0];
		2810	IV count = SvIVX (count_sv);
		2811
		2812	if (count > 0)
		2813	{
		2814	--count;
		2815	SvIVX (count_sv) = count;
		2816	XSRETURN_YES;
		2817	}
		2818	else
		2819	XSRETURN_NO;
		2820	}
		2821
		2822	void
		2823	waiters (SV *self)
		2824	CODE:
		2825	{
		2826	AV *av = (AV *)SvRV (self);
		2827
		2828	if (GIMME_V == G_SCALAR)
		2829	XPUSHs (sv_2mortal (newSVsv (AvARRAY (av)[0])));
		2830	else
		2831	{
2495	int i;	2832	int i;
2496
2497	errno = data->errorno;
2498	PL_laststype = data->laststype;
2499	PL_laststatval = data->laststatval;
2500	PL_statcache = data->statcache;
2501
2502	EXTEND (SP, AvFILLp (av));	2833	EXTEND (SP, AvFILLp (av) + 1 - 1);
2503	for (i = 0; i < AvFILLp (av); ++i)	2834	for (i = 1; i <= AvFILLp (av); ++i)
2504	PUSHs (sv_2mortal (SvREFCNT_inc_NN (AvARRAY (av)[i])));	2835	PUSHs (newSVsv (AvARRAY (av)[i]));
		2836	}
2505	}	2837	}
2506		2838
2507		2839
2508	MODULE = Coro::State PACKAGE = Coro::AnyEvent	2840	MODULE = Coro::State PACKAGE = Coro::AnyEvent
2509		2841
2510	BOOT:	2842	BOOT:
2511	sv_activity = coro_get_sv (aTHX_ "Coro::AnyEvent::ACTIVITY", TRUE);	2843	sv_activity = coro_get_sv (aTHX_ "Coro::AnyEvent::ACTIVITY", TRUE);
2512		2844
2513	SV *	2845	void
2514	_schedule (...)	2846	_schedule (...)
2515	PROTOTYPE: @
2516	CODE:	2847	CODE:
2517	{	2848	{
2518	static int incede;	2849	static int incede;
2519		2850
2520	api_cede_notself ();	2851	api_cede_notself (aTHX);
2521		2852
2522	++incede;	2853	++incede;
2523	while (coro_nready >= incede && api_cede ())	2854	while (coro_nready >= incede && api_cede (aTHX))
2524	;	2855	;
2525		2856
2526	sv_setsv (sv_activity, &PL_sv_undef);	2857	sv_setsv (sv_activity, &PL_sv_undef);
2527	if (coro_nready >= incede)	2858	if (coro_nready >= incede)
2528	{	2859	{
…		…
2534		2865
2535	--incede;	2866	--incede;
2536	}	2867	}
2537		2868
2538		2869
2539	MODULE = Coro::State PACKAGE = PerlIO::cede	2870	MODULE = Coro::State PACKAGE = Coro::AIO
2540		2871
2541	BOOT:	2872	void
2542	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	CV *slf_cv = newXSproto (target, coro_aio_req_xs, __FILE__, proto);
		2880	sv_magicext ((SV *)slf_cv, (SV *)req_cv, CORO_MAGIC_type_aio, 0, 0, 0);
		2881	}
2543		2882

Diff Legend

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