[ViewVC] Diff of: cvs/Coro/Coro/State.xs

Comparing Coro/Coro/State.xs (file contents):
Revision 1.272 by root, Fri Nov 14 20:35:49 2008 UTC vs.
Revision 1.297 by root, Tue Nov 18 11:04:42 2008 UTC

…		…
95	# define GV_NOTQUAL 0	95	# define GV_NOTQUAL 0
96	#endif	96	#endif
97	#ifndef newSV	97	#ifndef newSV
98	# define newSV(l) NEWSV(0,l)	98	# define newSV(l) NEWSV(0,l)
99	#endif	99	#endif
		100	#ifndef CvISXSUB_on
		101	# define CvISXSUB_on(cv) (void)cv
		102	#endif
100		103
101	/* 5.8.7 */	104	/* 5.8.7 */
102	#ifndef SvRV_set	105	#ifndef SvRV_set
103	# define SvRV_set(s,v) SvRV(s) = (v)	106	# define SvRV_set(s,v) SvRV(s) = (v)
104	#endif	107	#endif
…		…
117	#endif	120	#endif
118		121
119	/* The next macros try to return the current stack pointer, in an as	122	/* The next macros try to return the current stack pointer, in an as
120	* portable way as possible. */	123	* portable way as possible. */
121	#if __GNUC__ >= 4	124	#if __GNUC__ >= 4
		125	# define dSTACKLEVEL int stacklevel_dummy
122	# define dSTACKLEVEL void *stacklevel = __builtin_frame_address (0)	126	# define STACKLEVEL __builtin_frame_address (0)
123	#else	127	#else
124	# define dSTACKLEVEL volatile void stacklevel = (volatile void )&stacklevel	128	# define dSTACKLEVEL volatile void *stacklevel
		129	# define STACKLEVEL ((void *)&stacklevel)
125	#endif	130	#endif
126		131
127	#define IN_DESTRUCT (PL_main_cv == Nullcv)	132	#define IN_DESTRUCT (PL_main_cv == Nullcv)
128		133
129	#if __GNUC__ >= 3	134	#if __GNUC__ >= 3
…		…
140	#define expect_true(expr) expect ((expr) != 0, 1)	145	#define expect_true(expr) expect ((expr) != 0, 1)
141		146
142	#define NOINLINE attribute ((noinline))	147	#define NOINLINE attribute ((noinline))
143		148
144	#include "CoroAPI.h"	149	#include "CoroAPI.h"
		150	#define GCoroAPI (&coroapi) /* very sneaky */
145		151
146	#ifdef USE_ITHREADS	152	#ifdef USE_ITHREADS
147
148	static perl_mutex coro_lock;
149	# define LOCK do { MUTEX_LOCK (&coro_lock); } while (0)
150	# define UNLOCK do { MUTEX_UNLOCK (&coro_lock); } while (0)
151	# if CORO_PTHREAD	153	# if CORO_PTHREAD
152	static void *coro_thx;	154	static void *coro_thx;
153	# endif	155	# endif
154
155	#else
156
157	# define LOCK (void)0
158	# define UNLOCK (void)0
159
160	#endif	156	#endif
161
162	# undef LOCK
163	# define LOCK (void)0
164	# undef UNLOCK
165	# define UNLOCK (void)0
166
167	/* helper storage struct for Coro::AIO */
168	struct io_state
169	{
170	AV *res;
171	int errorno;
172	I32 laststype; /* U16 in 5.10.0 */
173	int laststatval;
174	Stat_t statcache;
175	};
176		157
177	static double (nvtime)(); / so why doesn't it take void? */	158	static double (nvtime)(); / so why doesn't it take void? */
		159
		160	/* we hijack an hopefully unused CV flag for our purposes */
		161	#define CVf_SLF 0x4000
		162	static OP *pp_slf (pTHX);
178		163
179	static U32 cctx_gen;	164	static U32 cctx_gen;
180	static size_t cctx_stacksize = CORO_STACKSIZE;	165	static size_t cctx_stacksize = CORO_STACKSIZE;
181	static struct CoroAPI coroapi;	166	static struct CoroAPI coroapi;
182	static AV main_mainstack; / used to differentiate between $main and others */	167	static AV main_mainstack; / used to differentiate between $main and others */
183	static JMPENV *main_top_env;	168	static JMPENV *main_top_env;
184	static HV coro_state_stash, coro_stash;	169	static HV coro_state_stash, coro_stash;
185	static volatile SV coro_mortal; / will be freed/thrown after next transfer */	170	static volatile SV coro_mortal; / will be freed/thrown after next transfer */
186	static volatile struct coro *transfer_next;
187		171
188	static GV irsgv; / $/ */	172	static GV irsgv; / $/ */
189	static GV stdoutgv; / STDOUT /	173	static GV stdoutgv; / STDOUT /
190	static SV *rv_diehook;	174	static SV *rv_diehook;
191	static SV *rv_warnhook;	175	static SV *rv_warnhook;
…		…
252	#undef VAR	236	#undef VAR
253	} perl_slots;	237	} perl_slots;
254		238
255	#define SLOT_COUNT ((sizeof (perl_slots) + sizeof (PERL_CONTEXT) - 1) / sizeof (PERL_CONTEXT))	239	#define SLOT_COUNT ((sizeof (perl_slots) + sizeof (PERL_CONTEXT) - 1) / sizeof (PERL_CONTEXT))
256		240
257	/* this is the per-perl-coro slf frame info */
258	/* it is treated like other "global" interpreter data */
259	/* and unfortunately is copied around, so kepe it small */
260	struct slf_frame
261	{
262	void (prepare) (struct coro_transfer_args ta); /* 0 means not yet initialised */
263	int (*check) (pTHX);
264	};
265
266	/* this is a structure representing a perl-level coroutine */	241	/* this is a structure representing a perl-level coroutine */
267	struct coro {	242	struct coro {
268	/* the C coroutine allocated to this perl coroutine, if any */	243	/* the C coroutine allocated to this perl coroutine, if any */
269	coro_cctx *cctx;	244	coro_cctx *cctx;
270		245
271	/* process data */	246	/* state data */
272	struct slf_frame slf_frame; /* saved slf frame */	247	struct CoroSLF slf_frame; /* saved slf frame */
273	void *slf_data;
274	AV *mainstack;	248	AV *mainstack;
275	perl_slots slot; / basically the saved sp */	249	perl_slots slot; / basically the saved sp */
276		250
277	AV args; / data associated with this coroutine (initial args) */	251	AV args; / data associated with this coroutine (initial args) */
278	int refcnt; /* coroutines are refcounted, yes */	252	int refcnt; /* coroutines are refcounted, yes */
279	int flags; /* CF_ flags */	253	int flags; /* CF_ flags */
280	HV hv; / the perl hash associated with this coro, if any */	254	HV hv; / the perl hash associated with this coro, if any */
		255	void (on_destroy)(pTHX_ struct coro coro);
281		256
282	/* statistics */	257	/* statistics */
283	int usecount; /* number of transfers to this coro */	258	int usecount; /* number of transfers to this coro */
284		259
285	/* coro process data */	260	/* coro process data */
286	int prio;	261	int prio;
287	SV throw; / exception to be thrown */	262	SV except; / exception to be thrown */
288		263
289	/* async_pool */	264	/* async_pool */
290	SV *saved_deffh;	265	SV *saved_deffh;
291		266
292	/* linked list */	267	/* linked list */
…		…
294	};	269	};
295		270
296	typedef struct coro *Coro__State;	271	typedef struct coro *Coro__State;
297	typedef struct coro *Coro__State_or_hashref;	272	typedef struct coro *Coro__State_or_hashref;
298		273
		274	/* the following variables are effectively part of the perl context */
		275	/* and get copied between struct coro and these variables */
		276	/* the mainr easonw e don't support windows process emulation */
299	static struct slf_frame slf_frame; /* the current slf frame */	277	static struct CoroSLF slf_frame; /* the current slf frame */
300		278
301	/ Coro ******************************************************************/	279	/ Coro ******************************************************************/
302		280
303	#define PRIO_MAX 3	281	#define PRIO_MAX 3
304	#define PRIO_HIGH 1	282	#define PRIO_HIGH 1
…		…
405	SvREFCNT_dec (av); /* sv_magicext increased the refcount */	383	SvREFCNT_dec (av); /* sv_magicext increased the refcount */
406		384
407	return 0;	385	return 0;
408	}	386	}
409		387
410	#define CORO_MAGIC_type_cv PERL_MAGIC_ext	388	#define CORO_MAGIC_type_cv 26
411	#define CORO_MAGIC_type_state PERL_MAGIC_ext	389	#define CORO_MAGIC_type_state PERL_MAGIC_ext
412		390
413	static MGVTBL coro_cv_vtbl = {	391	static MGVTBL coro_cv_vtbl = {
414	0, 0, 0, 0,	392	0, 0, 0, 0,
415	coro_cv_free	393	coro_cv_free
416	};	394	};
417		395
		396	#define CORO_MAGIC_NN(sv, type) \
		397	(expect_true (SvMAGIC (sv)->mg_type == type) \
		398	? SvMAGIC (sv) \
		399	: mg_find (sv, type))
		400
418	#define CORO_MAGIC(sv, type) \	401	#define CORO_MAGIC(sv, type) \
419	SvMAGIC (sv) \	402	(expect_true (SvMAGIC (sv)) \
420	? SvMAGIC (sv)->mg_type == type \	403	? CORO_MAGIC_NN (sv, type) \
421	? SvMAGIC (sv) \
422	: mg_find (sv, type) \
423	: 0	404	: 0)
424		405
425	#define CORO_MAGIC_cv(cv) CORO_MAGIC (((SV *)(cv)), CORO_MAGIC_type_cv)	406	#define CORO_MAGIC_cv(cv) CORO_MAGIC (((SV *)(cv)), CORO_MAGIC_type_cv)
426	#define CORO_MAGIC_state(sv) CORO_MAGIC (((SV *)(sv)), CORO_MAGIC_type_state)	407	#define CORO_MAGIC_state(sv) CORO_MAGIC_NN (((SV *)(sv)), CORO_MAGIC_type_state)
427		408
428	INLINE struct coro *	409	INLINE struct coro *
429	SvSTATE_ (pTHX_ SV *coro)	410	SvSTATE_ (pTHX_ SV *coro)
430	{	411	{
431	HV *stash;	412	HV *stash;
…		…
448	mg = CORO_MAGIC_state (coro);	429	mg = CORO_MAGIC_state (coro);
449	return (struct coro *)mg->mg_ptr;	430	return (struct coro *)mg->mg_ptr;
450	}	431	}
451		432
452	#define SvSTATE(sv) SvSTATE_ (aTHX_ (sv))	433	#define SvSTATE(sv) SvSTATE_ (aTHX_ (sv))
		434
		435	/* faster than SvSTATE, but expects a coroutine hv */
		436	#define SvSTATE_hv(hv) ((struct coro )CORO_MAGIC_NN ((SV )hv, CORO_MAGIC_type_state)->mg_ptr)
		437	#define SvSTATE_current SvSTATE_hv (SvRV (coro_current))
453		438
454	/* the next two functions merely cache the padlists */	439	/* the next two functions merely cache the padlists */
455	static void	440	static void
456	get_padlist (pTHX_ CV *cv)	441	get_padlist (pTHX_ CV *cv)
457	{	442	{
…		…
525	}	510	}
526		511
527	PUTBACK;	512	PUTBACK;
528	}	513	}
529		514
530	slf_frame = c->slf_frame;	515	slf_frame = c->slf_frame;
531	coroapi.slf_data = c->slf_data;	516	CORO_THROW = c->except;
532	}	517	}
533		518
534	static void	519	static void
535	save_perl (pTHX_ Coro__State c)	520	save_perl (pTHX_ Coro__State c)
536	{	521	{
537	c->slf_data = coroapi.slf_data;	522	c->except = CORO_THROW;
538	c->slf_frame = slf_frame;	523	c->slf_frame = slf_frame;
539		524
540	{	525	{
541	dSP;	526	dSP;
542	I32 cxix = cxstack_ix;	527	I32 cxix = cxstack_ix;
…		…
823		808
824	return orig_sigelem_set ? orig_sigelem_set (aTHX_ sv, mg) : 0;	809	return orig_sigelem_set ? orig_sigelem_set (aTHX_ sv, mg) : 0;
825	}	810	}
826		811
827	static void	812	static void
828	prepare_nop (aTHX_ struct coro_transfer_args *ta)	813	prepare_nop (pTHX_ struct coro_transfer_args *ta)
829	{	814	{
830	/* kind of mega-hacky, but works */	815	/* kind of mega-hacky, but works */
831	ta->next = ta->prev = (struct coro *)ta;	816	ta->next = ta->prev = (struct coro *)ta;
832	}	817	}
833		818
834	static int	819	static int
835	slf_check_nop (aTHX)	820	slf_check_nop (pTHX_ struct CoroSLF *frame)
836	{	821	{
837	return 0;	822	return 0;
838	}	823	}
839		824
840	static void	825	static UNOP coro_setup_op;
		826
		827	static void NOINLINE /* noinline to keep it out of the transfer fast path */
841	coro_setup (pTHX_ struct coro *coro)	828	coro_setup (pTHX_ struct coro *coro)
842	{	829	{
843	/*	830	/*
844	* emulate part of the perl startup here.	831	* emulate part of the perl startup here.
845	*/	832	*/
…		…
872	{	859	{
873	dSP;	860	dSP;
874	UNOP myop;	861	UNOP myop;
875		862
876	Zero (&myop, 1, UNOP);	863	Zero (&myop, 1, UNOP);
877	myop.op_next = Nullop;	864	myop.op_next = Nullop;
878	myop.op_flags = OPf_WANT_VOID;	865	myop.op_flags = OPf_WANT_VOID;
879		866
880	PUSHMARK (SP);	867	PUSHMARK (SP);
881	XPUSHs (sv_2mortal (av_shift (GvAV (PL_defgv))));	868	XPUSHs (sv_2mortal (av_shift (GvAV (PL_defgv))));
882	PUTBACK;	869	PUTBACK;
…		…
888	/* this newly created coroutine might be run on an existing cctx which most	875	/* this newly created coroutine might be run on an existing cctx which most
889	* likely was suspended in pp_slf, so we have to emulate entering pp_slf here.	876	* likely was suspended in pp_slf, so we have to emulate entering pp_slf here.
890	*/	877	*/
891	slf_frame.prepare = prepare_nop; /* provide a nop function for an eventual pp_slf */	878	slf_frame.prepare = prepare_nop; /* provide a nop function for an eventual pp_slf */
892	slf_frame.check = slf_check_nop; /* signal pp_slf to not repeat */	879	slf_frame.check = slf_check_nop; /* signal pp_slf to not repeat */
		880
		881	/* and we have to provide the pp_slf op in any case, so pp_slf can skip it */
		882	coro_setup_op.op_next = PL_op;
		883	coro_setup_op.op_type = OP_CUSTOM;
		884	coro_setup_op.op_ppaddr = pp_slf;
		885	/* no flags required, as an init function won't be called */
		886
		887	PL_op = (OP *)&coro_setup_op;
		888
		889	/* copy throw, in case it was set before coro_setup */
		890	CORO_THROW = coro->except;
893	}	891	}
894		892
895	static void	893	static void
896	coro_destruct (pTHX_ struct coro *coro)	894	coro_destruct (pTHX_ struct coro *coro)
897	{	895	{
…		…
921		919
922	SvREFCNT_dec (PL_diehook);	920	SvREFCNT_dec (PL_diehook);
923	SvREFCNT_dec (PL_warnhook);	921	SvREFCNT_dec (PL_warnhook);
924		922
925	SvREFCNT_dec (coro->saved_deffh);	923	SvREFCNT_dec (coro->saved_deffh);
926	SvREFCNT_dec (coro->throw);	924	SvREFCNT_dec (CORO_THROW);
927		925
928	coro_destruct_stacks (aTHX);	926	coro_destruct_stacks (aTHX);
929	}	927	}
930		928
931	INLINE void	929	INLINE void
…		…
941	static int	939	static int
942	runops_trace (pTHX)	940	runops_trace (pTHX)
943	{	941	{
944	COP *oldcop = 0;	942	COP *oldcop = 0;
945	int oldcxix = -2;	943	int oldcxix = -2;
946	struct coro coro = SvSTATE (coro_current); / trace cctx is tied to specific coro */	944	struct coro coro = SvSTATE_current; / trace cctx is tied to specific coro */
947	coro_cctx *cctx = coro->cctx;	945	coro_cctx *cctx = coro->cctx;
948		946
949	while ((PL_op = CALL_FPTR (PL_op->op_ppaddr) (aTHX)))	947	while ((PL_op = CALL_FPTR (PL_op->op_ppaddr) (aTHX)))
950	{	948	{
951	PERL_ASYNC_CHECK ();	949	PERL_ASYNC_CHECK ();
…		…
1060		1058
1061	TAINT_NOT;	1059	TAINT_NOT;
1062	return 0;	1060	return 0;
1063	}	1061	}
1064		1062
		1063	static struct coro_cctx *cctx_ssl_cctx;
		1064	static struct CoroSLF cctx_ssl_frame;
		1065
1065	static void	1066	static void
1066	prepare_set_stacklevel (struct coro_transfer_args ta, struct coro_cctx cctx)	1067	slf_prepare_set_stacklevel (pTHX_ struct coro_transfer_args *ta)
1067	{	1068	{
1068	ta->prev = (struct coro *)cctx;	1069	ta->prev = (struct coro *)cctx_ssl_cctx;
1069	ta->next = 0;	1070	ta->next = 0;
1070	}	1071	}
1071		1072
1072	/* inject a fake call to Coro::State::_cctx_init into the execution */	1073	static int
1073	/* _cctx_init should be careful, as it could be called at almost any time */	1074	slf_check_set_stacklevel (pTHX_ struct CoroSLF *frame)
1074	/* during execution of a perl program */	1075	{
1075	/* also initialises PL_top_env */	1076	*frame = cctx_ssl_frame;
		1077
		1078	return frame->check (aTHX_ frame); /* execute the restored frame - there must be one */
		1079	}
		1080
		1081	/* initialises PL_top_env and injects a pseudo-slf-call to set the stacklevel */
1076	static void NOINLINE	1082	static void NOINLINE
1077	cctx_prepare (pTHX_ coro_cctx *cctx)	1083	cctx_prepare (pTHX_ coro_cctx *cctx)
1078	{	1084	{
1079	dSP;
1080	UNOP myop;
1081
1082	PL_top_env = &PL_start_env;	1085	PL_top_env = &PL_start_env;
1083		1086
1084	if (cctx->flags & CC_TRACE)	1087	if (cctx->flags & CC_TRACE)
1085	PL_runops = runops_trace;	1088	PL_runops = runops_trace;
1086		1089
1087	Zero (&myop, 1, UNOP);	1090	/* we already must be executing an SLF op, there is no other valid way
1088	myop.op_next = PL_op;	1091	* that can lead to creation of a new cctx */
1089	myop.op_flags = OPf_WANT_VOID \| OPf_STACKED;	1092	assert (("FATAL: can't prepare slf-less cctx in Coro module (please report)",
		1093	slf_frame.prepare && PL_op->op_ppaddr == pp_slf));
1090		1094
1091	PUSHMARK (SP);	1095	/* we must emulate leaving pp_slf, which is done inside slf_check_set_stacklevel */
1092	EXTEND (SP, 2);	1096	cctx_ssl_cctx = cctx;
1093	PUSHs (sv_2mortal (newSViv ((IV)cctx)));	1097	cctx_ssl_frame = slf_frame;
1094	PUSHs ((SV *)get_cv ("Coro::State::_cctx_init", FALSE));	1098
1095	PUTBACK;	1099	slf_frame.prepare = slf_prepare_set_stacklevel;
1096	PL_op = (OP *)&myop;	1100	slf_frame.check = slf_check_set_stacklevel;
1097	PL_op = PL_ppaddr[OP_ENTERSUB](aTHX);
1098	SPAGAIN;
1099	}	1101	}
1100		1102
1101	/* the tail of transfer: execute stuff we can only do after a transfer */	1103	/* the tail of transfer: execute stuff we can only do after a transfer */
1102	INLINE void	1104	INLINE void
1103	transfer_tail (pTHX)	1105	transfer_tail (pTHX)
1104	{	1106	{
1105	struct coro next = (struct coro )transfer_next;
1106	assert (!(transfer_next = 0)); /* just used for the side effect when asserts are enabled */
1107	assert (("FATAL: next coroutine was zero in transfer_tail (please report)", next));
1108
1109	free_coro_mortal (aTHX);	1107	free_coro_mortal (aTHX);
1110	UNLOCK;
1111
1112	if (expect_false (next->throw))
1113	{
1114	SV *exception = sv_2mortal (next->throw);
1115
1116	next->throw = 0;
1117	sv_setsv (ERRSV, exception);
1118	croak (0);
1119	}
1120	}	1108	}
1121		1109
1122	/*	1110	/*
1123	* this is a _very_ stripped down perl interpreter ;)	1111	* this is a _very_ stripped down perl interpreter ;)
1124	*/	1112	*/
…		…
1139		1127
1140	/* inject a fake subroutine call to cctx_init */	1128	/* inject a fake subroutine call to cctx_init */
1141	cctx_prepare (aTHX_ (coro_cctx *)arg);	1129	cctx_prepare (aTHX_ (coro_cctx *)arg);
1142		1130
1143	/* cctx_run is the alternative tail of transfer() */	1131	/* cctx_run is the alternative tail of transfer() */
1144	/* TODO: throwing an exception here might be deadly, VERIFY */
1145	transfer_tail (aTHX);	1132	transfer_tail (aTHX);
1146		1133
1147	/* somebody or something will hit me for both perl_run and PL_restartop */	1134	/* somebody or something will hit me for both perl_run and PL_restartop */
1148	PL_restartop = PL_op;	1135	PL_restartop = PL_op;
1149	perl_run (PL_curinterp);	1136	perl_run (PL_curinterp);
…		…
1305	/ coroutine switching ***************************************************/	1292	/ coroutine switching ***************************************************/
1306		1293
1307	static void	1294	static void
1308	transfer_check (pTHX_ struct coro prev, struct coro next)	1295	transfer_check (pTHX_ struct coro prev, struct coro next)
1309	{	1296	{
		1297	/* TODO: throwing up here is considered harmful */
		1298
1310	if (expect_true (prev != next))	1299	if (expect_true (prev != next))
1311	{	1300	{
1312	if (expect_false (!(prev->flags & (CF_RUNNING \| CF_NEW))))	1301	if (expect_false (!(prev->flags & (CF_RUNNING \| CF_NEW))))
1313	croak ("Coro::State::transfer called with non-running/new prev Coro::State, but can only transfer from running or new states");	1302	croak ("Coro::State::transfer called with non-running/new prev Coro::State, but can only transfer from running or new states,");
1314		1303
1315	if (expect_false (next->flags & CF_RUNNING))	1304	if (expect_false (next->flags & CF_RUNNING))
1316	croak ("Coro::State::transfer called with running next Coro::State, but can only transfer to inactive states");	1305	croak ("Coro::State::transfer called with running next Coro::State, but can only transfer to inactive states,");
1317		1306
1318	if (expect_false (next->flags & CF_DESTROYED))	1307	if (expect_false (next->flags & CF_DESTROYED))
1319	croak ("Coro::State::transfer called with destroyed next Coro::State, but can only transfer to inactive states");	1308	croak ("Coro::State::transfer called with destroyed next Coro::State, but can only transfer to inactive states,");
1320		1309
1321	#if !PERL_VERSION_ATLEAST (5,10,0)	1310	#if !PERL_VERSION_ATLEAST (5,10,0)
1322	if (expect_false (PL_lex_state != LEX_NOTPARSING))	1311	if (expect_false (PL_lex_state != LEX_NOTPARSING))
1323	croak ("Coro::State::transfer called while parsing, but this is not supported in your perl version");	1312	croak ("Coro::State::transfer called while parsing, but this is not supported in your perl version,");
1324	#endif	1313	#endif
1325	}	1314	}
1326	}	1315	}
1327		1316
1328	/* always use the TRANSFER macro */	1317	/* always use the TRANSFER macro */
1329	static void NOINLINE	1318	static void NOINLINE /* noinline so we have a fixed stackframe */
1330	transfer (pTHX_ struct coro prev, struct coro next, int force_cctx)	1319	transfer (pTHX_ struct coro prev, struct coro next, int force_cctx)
1331	{	1320	{
1332	dSTACKLEVEL;	1321	dSTACKLEVEL;
1333		1322
1334	/* sometimes transfer is only called to set idle_sp */	1323	/* sometimes transfer is only called to set idle_sp */
1335	if (expect_false (!next))	1324	if (expect_false (!next))
1336	{	1325	{
1337	((coro_cctx *)prev)->idle_sp = stacklevel;	1326	((coro_cctx *)prev)->idle_sp = STACKLEVEL;
1338	assert (((coro_cctx )prev)->idle_te = PL_top_env); / just for the side-effect when asserts are enabled */	1327	assert (((coro_cctx )prev)->idle_te = PL_top_env); / just for the side-effect when asserts are enabled */
1339	}	1328	}
1340	else if (expect_true (prev != next))	1329	else if (expect_true (prev != next))
1341	{	1330	{
1342	coro_cctx *prev__cctx;	1331	coro_cctx *prev__cctx;
…		…
1349	prev->flags \|= CF_RUNNING;	1338	prev->flags \|= CF_RUNNING;
1350	}	1339	}
1351		1340
1352	prev->flags &= ~CF_RUNNING;	1341	prev->flags &= ~CF_RUNNING;
1353	next->flags \|= CF_RUNNING;	1342	next->flags \|= CF_RUNNING;
1354
1355	LOCK;
1356		1343
1357	/* first get rid of the old state */	1344	/* first get rid of the old state */
1358	save_perl (aTHX_ prev);	1345	save_perl (aTHX_ prev);
1359		1346
1360	if (expect_false (next->flags & CF_NEW))	1347	if (expect_false (next->flags & CF_NEW))
…		…
1369		1356
1370	prev__cctx = prev->cctx;	1357	prev__cctx = prev->cctx;
1371		1358
1372	/* possibly untie and reuse the cctx */	1359	/* possibly untie and reuse the cctx */
1373	if (expect_true (	1360	if (expect_true (
1374	prev__cctx->idle_sp == stacklevel	1361	prev__cctx->idle_sp == STACKLEVEL
1375	&& !(prev__cctx->flags & CC_TRACE)	1362	&& !(prev__cctx->flags & CC_TRACE)
1376	&& !force_cctx	1363	&& !force_cctx
1377	))	1364	))
1378	{	1365	{
1379	/* I assume that stacklevel is a stronger indicator than PL_top_env changes */	1366	/* I assume that stacklevel is a stronger indicator than PL_top_env changes */
…		…
1393	++next->usecount;	1380	++next->usecount;
1394		1381
1395	if (expect_true (!next->cctx))	1382	if (expect_true (!next->cctx))
1396	next->cctx = cctx_get (aTHX);	1383	next->cctx = cctx_get (aTHX);
1397		1384
1398	assert (("FATAL: transfer_next already nonzero in Coro (please report)", !transfer_next));
1399	transfer_next = next;
1400
1401	if (expect_false (prev__cctx != next->cctx))	1385	if (expect_false (prev__cctx != next->cctx))
1402	{	1386	{
1403	prev__cctx->top_env = PL_top_env;	1387	prev__cctx->top_env = PL_top_env;
1404	PL_top_env = next->cctx->top_env;	1388	PL_top_env = next->cctx->top_env;
1405	coro_transfer (&prev__cctx->cctx, &next->cctx->cctx);	1389	coro_transfer (&prev__cctx->cctx, &next->cctx->cctx);
…		…
1418	coro_state_destroy (pTHX_ struct coro *coro)	1402	coro_state_destroy (pTHX_ struct coro *coro)
1419	{	1403	{
1420	if (coro->flags & CF_DESTROYED)	1404	if (coro->flags & CF_DESTROYED)
1421	return 0;	1405	return 0;
1422		1406
		1407	if (coro->on_destroy)
		1408	coro->on_destroy (aTHX_ coro);
		1409
1423	coro->flags \|= CF_DESTROYED;	1410	coro->flags \|= CF_DESTROYED;
1424		1411
1425	if (coro->flags & CF_READY)	1412	if (coro->flags & CF_READY)
1426	{	1413	{
1427	/* reduce nready, as destroying a ready coro effectively unreadies it */	1414	/* reduce nready, as destroying a ready coro effectively unreadies it */
1428	/* alternative: look through all ready queues and remove the coro */	1415	/* alternative: look through all ready queues and remove the coro */
1429	LOCK;
1430	--coro_nready;	1416	--coro_nready;
1431	UNLOCK;
1432	}	1417	}
1433	else	1418	else
1434	coro->flags \|= CF_READY; /* make sure it is NOT put into the readyqueue */	1419	coro->flags \|= CF_READY; /* make sure it is NOT put into the readyqueue */
1435		1420
1436	if (coro->mainstack && coro->mainstack != main_mainstack)	1421	if (coro->mainstack && coro->mainstack != main_mainstack)
1437	{	1422	{
1438	struct coro temp;	1423	struct coro temp;
1439		1424
1440	if (coro->flags & CF_RUNNING)	1425	assert (("FATAL: tried to destroy currently running coroutine (please report)", !(coro->flags & CF_RUNNING)));
1441	croak ("FATAL: tried to destroy currently running coroutine");
1442		1426
1443	save_perl (aTHX_ &temp);	1427	save_perl (aTHX_ &temp);
1444	load_perl (aTHX_ coro);	1428	load_perl (aTHX_ coro);
1445		1429
1446	coro_destruct (aTHX_ coro);	1430	coro_destruct (aTHX_ coro);
…		…
1515	TRANSFER (ta, 1);	1499	TRANSFER (ta, 1);
1516	}	1500	}
1517		1501
1518	/ Coro ******************************************************************/	1502	/ Coro ******************************************************************/
1519		1503
1520	static void	1504	INLINE void
1521	coro_enq (pTHX_ SV *coro_sv)	1505	coro_enq (pTHX_ struct coro *coro)
1522	{	1506	{
1523	av_push (coro_ready [SvSTATE (coro_sv)->prio - PRIO_MIN], coro_sv);	1507	av_push (coro_ready [coro->prio - PRIO_MIN], SvREFCNT_inc_NN (coro->hv));
1524	}	1508	}
1525		1509
1526	static SV *	1510	INLINE SV *
1527	coro_deq (pTHX)	1511	coro_deq (pTHX)
1528	{	1512	{
1529	int prio;	1513	int prio;
1530		1514
1531	for (prio = PRIO_MAX - PRIO_MIN + 1; --prio >= 0; )	1515	for (prio = PRIO_MAX - PRIO_MIN + 1; --prio >= 0; )
…		…
1550	if (coro->flags & CF_READY)	1534	if (coro->flags & CF_READY)
1551	return 0;	1535	return 0;
1552		1536
1553	coro->flags \|= CF_READY;	1537	coro->flags \|= CF_READY;
1554		1538
1555	LOCK;
1556
1557	sv_hook = coro_nready ? 0 : coro_readyhook;	1539	sv_hook = coro_nready ? 0 : coro_readyhook;
1558	xs_hook = coro_nready ? 0 : coroapi.readyhook;	1540	xs_hook = coro_nready ? 0 : coroapi.readyhook;
1559		1541
1560	coro_enq (aTHX_ SvREFCNT_inc_NN (coro_sv));	1542	coro_enq (aTHX_ coro);
1561	++coro_nready;	1543	++coro_nready;
1562		1544
1563	UNLOCK;
1564
1565	if (sv_hook)	1545	if (sv_hook)
1566	{	1546	{
1567	dSP;	1547	dSP;
1568		1548
1569	ENTER;	1549	ENTER;
1570	SAVETMPS;	1550	SAVETMPS;
1571		1551
1572	PUSHMARK (SP);	1552	PUSHMARK (SP);
1573	PUTBACK;	1553	PUTBACK;
1574	call_sv (sv_hook, G_DISCARD);	1554	call_sv (sv_hook, G_VOID \| G_DISCARD);
1575	SPAGAIN;
1576		1555
1577	FREETMPS;	1556	FREETMPS;
1578	LEAVE;	1557	LEAVE;
1579	}	1558	}
1580		1559
…		…
1595	{	1574	{
1596	SV prev_sv, next_sv;	1575	SV prev_sv, next_sv;
1597		1576
1598	for (;;)	1577	for (;;)
1599	{	1578	{
1600	LOCK;
1601	next_sv = coro_deq (aTHX);	1579	next_sv = coro_deq (aTHX);
1602		1580
1603	/* nothing to schedule: call the idle handler */	1581	/* nothing to schedule: call the idle handler */
1604	if (expect_false (!next_sv))	1582	if (expect_false (!next_sv))
1605	{	1583	{
1606	dSP;	1584	dSP;
1607	UNLOCK;
1608		1585
1609	ENTER;	1586	ENTER;
1610	SAVETMPS;	1587	SAVETMPS;
1611		1588
1612	PUSHMARK (SP);	1589	PUSHMARK (SP);
1613	PUTBACK;	1590	PUTBACK;
1614	call_sv (get_sv ("Coro::idle", FALSE), G_DISCARD);	1591	call_sv (get_sv ("Coro::idle", FALSE), G_VOID \| G_DISCARD);
1615	SPAGAIN;
1616		1592
1617	FREETMPS;	1593	FREETMPS;
1618	LEAVE;	1594	LEAVE;
1619	continue;	1595	continue;
1620	}	1596	}
1621		1597
1622	ta->next = SvSTATE (next_sv);	1598	ta->next = SvSTATE_hv (next_sv);
1623		1599
1624	/* cannot transfer to destroyed coros, skip and look for next */	1600	/* cannot transfer to destroyed coros, skip and look for next */
1625	if (expect_false (ta->next->flags & CF_DESTROYED))	1601	if (expect_false (ta->next->flags & CF_DESTROYED))
1626	{	1602	{
1627	UNLOCK;
1628	SvREFCNT_dec (next_sv);	1603	SvREFCNT_dec (next_sv);
1629	/* coro_nready has already been taken care of by destroy */	1604	/* coro_nready has already been taken care of by destroy */
1630	continue;	1605	continue;
1631	}	1606	}
1632		1607
1633	--coro_nready;	1608	--coro_nready;
1634	UNLOCK;
1635	break;	1609	break;
1636	}	1610	}
1637		1611
1638	/* free this only after the transfer */	1612	/* free this only after the transfer */
1639	prev_sv = SvRV (coro_current);	1613	prev_sv = SvRV (coro_current);
1640	ta->prev = SvSTATE (prev_sv);	1614	ta->prev = SvSTATE_hv (prev_sv);
1641	TRANSFER_CHECK (*ta);	1615	TRANSFER_CHECK (*ta);
1642	assert (("FATAL: next coroutine isn't marked as ready in Coro (please report)", ta->next->flags & CF_READY));	1616	assert (("FATAL: next coroutine isn't marked as ready in Coro (please report)", ta->next->flags & CF_READY));
1643	ta->next->flags &= ~CF_READY;	1617	ta->next->flags &= ~CF_READY;
1644	SvRV_set (coro_current, next_sv);	1618	SvRV_set (coro_current, next_sv);
1645		1619
1646	LOCK;
1647	free_coro_mortal (aTHX);	1620	free_coro_mortal (aTHX);
1648	coro_mortal = prev_sv;	1621	coro_mortal = prev_sv;
1649	UNLOCK;
1650	}	1622	}
1651		1623
1652	INLINE void	1624	INLINE void
1653	prepare_cede (pTHX_ struct coro_transfer_args *ta)	1625	prepare_cede (pTHX_ struct coro_transfer_args *ta)
1654	{	1626	{
…		…
1718	if (flags & CC_TRACE)	1690	if (flags & CC_TRACE)
1719	{	1691	{
1720	if (!coro->cctx)	1692	if (!coro->cctx)
1721	coro->cctx = cctx_new_run ();	1693	coro->cctx = cctx_new_run ();
1722	else if (!(coro->cctx->flags & CC_TRACE))	1694	else if (!(coro->cctx->flags & CC_TRACE))
1723	croak ("cannot enable tracing on coroutine with custom stack");	1695	croak ("cannot enable tracing on coroutine with custom stack,");
1724		1696
1725	coro->cctx->flags \|= CC_NOREUSE \| (flags & (CC_TRACE \| CC_TRACE_ALL));	1697	coro->cctx->flags \|= CC_NOREUSE \| (flags & (CC_TRACE \| CC_TRACE_ALL));
1726	}	1698	}
1727	else if (coro->cctx && coro->cctx->flags & CC_TRACE)	1699	else if (coro->cctx && coro->cctx->flags & CC_TRACE)
1728	{	1700	{
…		…
1733	else	1705	else
1734	coro->slot->runops = RUNOPS_DEFAULT;	1706	coro->slot->runops = RUNOPS_DEFAULT;
1735	}	1707	}
1736	}	1708	}
1737		1709
1738	#if 0	1710	/*****************************************************************************/
		1711	/* schedule-like-function opcode (SLF) */
		1712
		1713	static UNOP slf_restore; /* restore stack as entersub did, for first-re-run */
		1714	static const CV *slf_cv;
		1715	static SV **slf_argv;
		1716	static int slf_argc, slf_arga; /* count, allocated */
		1717	static I32 slf_ax; /* top of stack, for restore */
		1718
		1719	/* this restores the stack in the case we patched the entersub, to */
		1720	/* recreate the stack frame as perl will on following calls */
		1721	/* since entersub cleared the stack */
		1722	static OP *
		1723	pp_restore (pTHX)
		1724	{
		1725	int i;
		1726	SV **SP = PL_stack_base + slf_ax;
		1727
		1728	PUSHMARK (SP);
		1729
		1730	EXTEND (SP, slf_argc + 1);
		1731
		1732	for (i = 0; i < slf_argc; ++i)
		1733	PUSHs (sv_2mortal (slf_argv [i]));
		1734
		1735	PUSHs ((SV *)CvGV (slf_cv));
		1736
		1737	RETURNOP (slf_restore.op_first);
		1738	}
		1739
1739	static int	1740	static void
1740	coro_gensub_free (pTHX_ SV sv, MAGIC mg)	1741	slf_prepare_transfer (pTHX_ struct coro_transfer_args *ta)
1741	{	1742	{
1742	AV *padlist;	1743	SV arg = (SV )slf_frame.data;
1743	AV av = (AV )mg->mg_obj;
1744		1744
1745	abort ();	1745	prepare_transfer (aTHX_ ta, arg [0], arg [1]);
1746
1747	return 0;
1748	}	1746	}
1749		1747
1750	static MGVTBL coro_gensub_vtbl = {	1748	static void
1751	0, 0, 0, 0,	1749	slf_init_transfer (pTHX_ struct CoroSLF frame, CV cv, SV **arg, int items)
1752	coro_gensub_free	1750	{
1753	};	1751	if (items != 2)
1754	#endif	1752	croak ("Coro::State::transfer (prev, next) expects two arguments, not %d,", items);
		1753
		1754	frame->prepare = slf_prepare_transfer;
		1755	frame->check = slf_check_nop;
		1756	frame->data = (void )arg; / let's hope it will stay valid */
		1757	}
		1758
		1759	static void
		1760	slf_init_schedule (pTHX_ struct CoroSLF frame, CV cv, SV **arg, int items)
		1761	{
		1762	frame->prepare = prepare_schedule;
		1763	frame->check = slf_check_nop;
		1764	}
		1765
		1766	static void
		1767	slf_init_cede (pTHX_ struct CoroSLF frame, CV cv, SV **arg, int items)
		1768	{
		1769	frame->prepare = prepare_cede;
		1770	frame->check = slf_check_nop;
		1771	}
		1772
		1773	static void
		1774	slf_init_cede_notself (pTHX_ struct CoroSLF frame, CV cv, SV **arg, int items)
		1775	{
		1776	frame->prepare = prepare_cede_notself;
		1777	frame->check = slf_check_nop;
		1778	}
		1779
		1780	/*
		1781	* these not obviously related functions are all rolled into one
		1782	* function to increase chances that they all will call transfer with the same
		1783	* stack offset
		1784	* SLF stands for "schedule-like-function".
		1785	*/
		1786	static OP *
		1787	pp_slf (pTHX)
		1788	{
		1789	I32 checkmark; /* mark SP to see how many elements check has pushed */
		1790
		1791	/* set up the slf frame, unless it has already been set-up */
		1792	/* the latter happens when a new coro has been started */
		1793	/* or when a new cctx was attached to an existing coroutine */
		1794	if (expect_true (!slf_frame.prepare))
		1795	{
		1796	/* first iteration */
		1797	dSP;
		1798	SV **arg = PL_stack_base + TOPMARK + 1;
		1799	int items = SP - arg; /* args without function object */
		1800	SV gv = sp;
		1801
		1802	/* do a quick consistency check on the "function" object, and if it isn't */
		1803	/* for us, divert to the real entersub */
		1804	if (SvTYPE (gv) != SVt_PVGV
		1805	\|\| !GvCV (gv)
		1806	\|\| !(CvFLAGS (GvCV (gv)) & CVf_SLF))
		1807	return PL_ppaddr[OP_ENTERSUB](aTHX);
		1808
		1809	if (!(PL_op->op_flags & OPf_STACKED))
		1810	{
		1811	/* ampersand-form of call, use @_ instead of stack */
		1812	AV *av = GvAV (PL_defgv);
		1813	arg = AvARRAY (av);
		1814	items = AvFILLp (av) + 1;
		1815	}
		1816
		1817	/* now call the init function, which needs to set up slf_frame */
		1818	((coro_slf_cb)CvXSUBANY (GvCV (gv)).any_ptr)
		1819	(aTHX_ &slf_frame, GvCV (gv), arg, items);
		1820
		1821	/* pop args */
		1822	SP = PL_stack_base + POPMARK;
		1823
		1824	PUTBACK;
		1825	}
		1826
		1827	/* now that we have a slf_frame, interpret it! */
		1828	/* we use a callback system not to make the code needlessly */
		1829	/* complicated, but so we can run multiple perl coros from one cctx */
		1830
		1831	do
		1832	{
		1833	struct coro_transfer_args ta;
		1834
		1835	slf_frame.prepare (aTHX_ &ta);
		1836	TRANSFER (ta, 0);
		1837
		1838	checkmark = PL_stack_sp - PL_stack_base;
		1839	}
		1840	while (slf_frame.check (aTHX_ &slf_frame));
		1841
		1842	slf_frame.prepare = 0; /* invalidate the frame, we are done processing it */
		1843
		1844	/* exception handling */
		1845	if (expect_false (CORO_THROW))
		1846	{
		1847	SV *exception = sv_2mortal (CORO_THROW);
		1848
		1849	CORO_THROW = 0;
		1850	sv_setsv (ERRSV, exception);
		1851	croak (0);
		1852	}
		1853
		1854	/* return value handling - mostly like entersub */
		1855	/* make sure we put something on the stack in scalar context */
		1856	if (GIMME_V == G_SCALAR)
		1857	{
		1858	dSP;
		1859	SV **bot = PL_stack_base + checkmark;
		1860
		1861	if (sp == bot) /* too few, push undef */
		1862	bot [1] = &PL_sv_undef;
		1863	else if (sp != bot + 1) /* too many, take last one */
		1864	bot [1] = *sp;
		1865
		1866	SP = bot + 1;
		1867
		1868	PUTBACK;
		1869	}
		1870
		1871	return NORMAL;
		1872	}
		1873
		1874	static void
		1875	api_execute_slf (pTHX_ CV *cv, coro_slf_cb init_cb, I32 ax)
		1876	{
		1877	int i;
		1878	SV **arg = PL_stack_base + ax;
		1879	int items = PL_stack_sp - arg + 1;
		1880
		1881	assert (("FATAL: SLF call with illegal CV value", !CvANON (cv)));
		1882
		1883	if (PL_op->op_ppaddr != PL_ppaddr [OP_ENTERSUB]
		1884	&& PL_op->op_ppaddr != pp_slf)
		1885	croak ("FATAL: Coro SLF calls can only be made normally, not via goto or any other means, caught");
		1886
		1887	CvFLAGS (cv) \|= CVf_SLF;
		1888	CvXSUBANY (cv).any_ptr = (void *)init_cb;
		1889	slf_cv = cv;
		1890
		1891	/* we patch the op, and then re-run the whole call */
		1892	/* we have to put the same argument on the stack for this to work */
		1893	/* and this will be done by pp_restore */
		1894	slf_restore.op_next = (OP *)&slf_restore;
		1895	slf_restore.op_type = OP_CUSTOM;
		1896	slf_restore.op_ppaddr = pp_restore;
		1897	slf_restore.op_first = PL_op;
		1898
		1899	slf_ax = ax - 1; /* undo the ax++ inside dAXMARK */
		1900
		1901	if (PL_op->op_flags & OPf_STACKED)
		1902	{
		1903	if (items > slf_arga)
		1904	{
		1905	slf_arga = items;
		1906	free (slf_argv);
		1907	slf_argv = malloc (slf_arga * sizeof (SV *));
		1908	}
		1909
		1910	slf_argc = items;
		1911
		1912	for (i = 0; i < items; ++i)
		1913	slf_argv [i] = SvREFCNT_inc (arg [i]);
		1914	}
		1915	else
		1916	slf_argc = 0;
		1917
		1918	PL_op->op_ppaddr = pp_slf;
		1919	PL_op->op_type = OP_CUSTOM; /* maybe we should leave it at entersub? */
		1920
		1921	PL_op = (OP *)&slf_restore;
		1922	}
1755		1923
1756	/*****************************************************************************/	1924	/*****************************************************************************/
1757	/* PerlIO::cede */	1925	/* PerlIO::cede */
1758		1926
1759	typedef struct	1927	typedef struct
…		…
1827	PerlIOBuf_get_cnt,	1995	PerlIOBuf_get_cnt,
1828	PerlIOBuf_set_ptrcnt,	1996	PerlIOBuf_set_ptrcnt,
1829	};	1997	};
1830		1998
1831	/*****************************************************************************/	1999	/*****************************************************************************/
		2000	/* Coro::Semaphore */
1832		2001
1833	static const CV slf_cv; / for quick consistency check */	2002	static void
		2003	coro_semaphore_adjust (pTHX_ AV *av, IV adjust)
		2004	{
		2005	SV *count_sv = AvARRAY (av)[0];
		2006	IV count = SvIVX (count_sv);
1834		2007
1835	static UNOP slf_restore; /* restore stack as entersub did, for first-re-run */	2008	count += adjust;
1836	static SV *slf_arg0;	2009	SvIVX (count_sv) = count;
1837	static SV *slf_arg1;
1838		2010
1839	/* this restores the stack in the case we patched the entersub, to */	2011	/* now wake up as many waiters as are expected to lock */
1840	/* recreate the stack frame as perl will on following calls */	2012	while (count > 0 && AvFILLp (av) > 0)
1841	/* since entersub cleared the stack */	2013	{
		2014	SV *cb;
		2015
		2016	/* swap first two elements so we can shift a waiter */
		2017	AvARRAY (av)[0] = AvARRAY (av)[1];
		2018	AvARRAY (av)[1] = count_sv;
		2019	cb = av_shift (av);
		2020
		2021	if (SvOBJECT (cb))
		2022	api_ready (aTHX_ cb);
		2023	else
		2024	croak ("callbacks not yet supported");
		2025
		2026	SvREFCNT_dec (cb);
		2027
		2028	--count;
		2029	}
		2030	}
		2031
		2032	static void
		2033	coro_semaphore_on_destroy (pTHX_ struct coro *coro)
		2034	{
		2035	/* call $sem->adjust (0) to possibly wake up some other waiters */
		2036	coro_semaphore_adjust (aTHX_ (AV *)coro->slf_frame.data, 0);
		2037	}
		2038
		2039	static int
		2040	slf_check_semaphore_down (pTHX_ struct CoroSLF *frame)
		2041	{
		2042	AV av = (AV )frame->data;
		2043	SV *count_sv = AvARRAY (av)[0];
		2044
		2045	/* if we are about to throw, don't actually acquire the lock, just throw */
		2046	if (CORO_THROW)
		2047	return 0;
		2048	else if (SvIVX (count_sv) > 0)
		2049	{
		2050	SvSTATE_current->on_destroy = 0;
		2051	SvIVX (count_sv) = SvIVX (count_sv) - 1;
		2052	return 0;
		2053	}
		2054	else
		2055	{
		2056	int i;
		2057	/* if we were woken up but can't down, we look through the whole */
		2058	/* waiters list and only add us if we aren't in there already */
		2059	/* this avoids some degenerate memory usage cases */
		2060
		2061	for (i = 1; i <= AvFILLp (av); ++i)
		2062	if (AvARRAY (av)[i] == SvRV (coro_current))
		2063	return 1;
		2064
		2065	av_push (av, SvREFCNT_inc (SvRV (coro_current)));
		2066	return 1;
		2067	}
		2068	}
		2069
		2070	static void
		2071	slf_init_semaphore_down (pTHX_ struct CoroSLF frame, CV cv, SV **arg, int items)
		2072	{
		2073	AV av = (AV )SvRV (arg [0]);
		2074
		2075	if (SvIVX (AvARRAY (av)[0]) > 0)
		2076	{
		2077	frame->data = (void *)av;
		2078	frame->prepare = prepare_nop;
		2079	}
		2080	else
		2081	{
		2082	av_push (av, SvREFCNT_inc (SvRV (coro_current)));
		2083
		2084	frame->data = (void )sv_2mortal (SvREFCNT_inc ((SV )av));
		2085	frame->prepare = prepare_schedule;
		2086
		2087	/* to avoid race conditions when a woken-up coro gets terminated */
		2088	/* we arrange for a temporary on_destroy that calls adjust (0) */
		2089	SvSTATE_current->on_destroy = coro_semaphore_on_destroy;
		2090	}
		2091
		2092	frame->check = slf_check_semaphore_down;
		2093
		2094	}
		2095
		2096	/*****************************************************************************/
		2097	/* gensub: simple closure generation utility */
		2098
		2099	#define GENSUB_ARG CvXSUBANY (cv).any_ptr
		2100
		2101	/* create a closure from XS, returns a code reference */
		2102	/* the arg can be accessed via GENSUB_ARG from the callback */
		2103	/* the callback must use dXSARGS/XSRETURN */
1842	static OP *	2104	static SV *
1843	pp_restore (pTHX)	2105	gensub (pTHX_ void (xsub)(pTHX_ CV ), void *arg)
1844	{	2106	{
1845	dSP;	2107	CV cv = (CV )newSV (0);
1846		2108
1847	PUSHMARK (SP);	2109	sv_upgrade ((SV *)cv, SVt_PVCV);
1848		2110
1849	EXTEND (SP, 3);	2111	CvANON_on (cv);
1850	if (slf_arg0) PUSHs (sv_2mortal (slf_arg0));	2112	CvISXSUB_on (cv);
1851	if (slf_arg1) PUSHs (sv_2mortal (slf_arg1));	2113	CvXSUB (cv) = xsub;
1852	PUSHs ((SV *)CvGV (slf_cv));	2114	GENSUB_ARG = arg;
1853		2115
1854	RETURNOP (slf_restore.op_first);	2116	return newRV_noinc ((SV *)cv);
1855	}	2117	}
1856		2118
		2119	/*****************************************************************************/
		2120	/* Coro::AIO */
		2121
		2122	#define CORO_MAGIC_type_aio PERL_MAGIC_ext
		2123
		2124	/* helper storage struct */
		2125	struct io_state
		2126	{
		2127	int errorno;
		2128	I32 laststype; /* U16 in 5.10.0 */
		2129	int laststatval;
		2130	Stat_t statcache;
		2131	};
		2132
		2133	static void
		2134	coro_aio_callback (pTHX_ CV *cv)
		2135	{
		2136	dXSARGS;
		2137	AV state = (AV )GENSUB_ARG;
		2138	SV *coro = av_pop (state);
		2139	SV *data_sv = newSV (sizeof (struct io_state));
		2140
		2141	av_extend (state, items);
		2142
		2143	sv_upgrade (data_sv, SVt_PV);
		2144	SvCUR_set (data_sv, sizeof (struct io_state));
		2145	SvPOK_only (data_sv);
		2146
		2147	{
		2148	struct io_state data = (struct io_state )SvPVX (data_sv);
		2149
		2150	data->errorno = errno;
		2151	data->laststype = PL_laststype;
		2152	data->laststatval = PL_laststatval;
		2153	data->statcache = PL_statcache;
		2154	}
		2155
		2156	/* now build the result vector out of all the parameters and the data_sv */
		2157	{
		2158	int i;
		2159
		2160	for (i = 0; i < items; ++i)
		2161	av_push (state, SvREFCNT_inc_NN (ST (i)));
		2162	}
		2163
		2164	av_push (state, data_sv);
		2165
		2166	api_ready (aTHX_ coro);
		2167	SvREFCNT_dec (coro);
		2168	SvREFCNT_dec ((AV *)state);
		2169	}
		2170
1857	static void	2171	static int
1858	slf_init_set_stacklevel (pTHX_ SV **arg, int items)	2172	slf_check_aio_req (pTHX_ struct CoroSLF *frame)
1859	{	2173	{
1860	assert (("FATAL: set_stacklevel needs the coro cctx as sole argument", items == 1));	2174	AV state = (AV )frame->data;
1861	CORO_SLF_DATA = (void *)SvIV (arg [0]);
1862	}
1863		2175
1864	static void	2176	/* if we are about to throw, return early */
1865	slf_prepare_set_stacklevel (pTHX_ struct coro_transfer_args *ta)	2177	/* this does not cancel the aio request, but at least */
1866	{	2178	/* it quickly returns */
1867	prepare_set_stacklevel (ta, (struct coro_cctx *)CORO_SLF_DATA);	2179	if (CORO_THROW)
1868	}	2180	return 0;
1869		2181
1870	static void	2182	/* one element that is an RV? repeat! */
1871	slf_init_transfer (pTHX_ SV **arg, int items)	2183	if (AvFILLp (state) == 0 && SvROK (AvARRAY (state)[0]))
1872	{	2184	return 1;
1873	if (items != 2)
1874	croak ("Coro::State::transfer (prev, next) expects two arguments, not %d.", items);
1875		2185
1876	CORO_SLF_DATA = (void )arg; / let's hope it will stay valid */	2186	/* restore status */
1877	}
1878
1879	static void
1880	slf_prepare_transfer (pTHX_ struct coro_transfer_args *ta)
1881	{
1882	SV arg = (SV )CORO_SLF_DATA;
1883
1884	prepare_transfer (ta, arg [0], arg [1]);
1885	}
1886
1887	static void
1888	slf_init_nop (pTHX_ SV **arg, int items)
1889	{
1890	}
1891
1892	/* slf_prepare_schedule == prepare_schedule */
1893	/* slf_prepare_cede == prepare_cede */
1894	/* slf_prepare_notself == prepare_notself */
1895
1896	/* we hijack an hopefully unused CV flag for our purposes */
1897	#define CVf_SLF 0x4000
1898
1899	/*
1900	* these not obviously related functions are all rolled into one
1901	* function to increase chances that they all will call transfer with the same
1902	* stack offset
1903	* SLF stands for "schedule-like-function".
1904	*/
1905	static OP *
1906	pp_slf (pTHX)
1907	{
1908	I32 checkmark; /* mark SP to see how many elements check has pushed */
1909
1910	if (expect_true (!slf_frame.prepare))
1911	{	2187	{
1912	/* first iteration */	2188	SV *data_sv = av_pop (state);
1913	dSP;	2189	struct io_state data = (struct io_state )SvPVX (data_sv);
1914	SV **arg = PL_stack_base + TOPMARK + 1;
1915	int items = SP - arg; /* args without function object */
1916	SV gv = sp;
1917	struct CoroSLF *slf;
1918		2190
1919	/* do a quick consistency check on the "function" object, and if it isn't */	2191	errno = data->errorno;
1920	/* for us, divert to the real entersub */	2192	PL_laststype = data->laststype;
1921	if (SvTYPE (gv) != SVt_PVGV \|\| !(CvFLAGS (GvCV (gv)) & CVf_SLF))	2193	PL_laststatval = data->laststatval;
1922	return PL_ppaddr[OP_ENTERSUB](aTHX);	2194	PL_statcache = data->statcache;
1923		2195
1924	/* pop args */	2196	SvREFCNT_dec (data_sv);
1925	SP = PL_stack_base + POPMARK;
1926
1927	if (!(PL_op->op_flags & OPf_STACKED))
1928	{
1929	/* ampersand-form of call, use @_ instead of stack */
1930	AV *av = GvAV (PL_defgv);
1931	arg = AvARRAY (av);
1932	items = AvFILLp (av) + 1;
1933	}
1934
1935	PUTBACK;
1936
1937	slf = (struct CoroSLF *)CvXSUBANY (GvCV (gv)).any_ptr;
1938	slf_frame.prepare = slf->prepare;
1939	slf_frame.check = slf->check;
1940	slf->init (aTHX_ arg, items);
1941	}	2197	}
1942		2198
1943	/* now interpret the slf_frame */	2199	/* push result values */
1944	/* we use a callback system not to make the code needlessly */
1945	/* complicated, but so we can run multiple perl coros from one cctx */
1946
1947	do
1948	{
1949	struct coro_transfer_args ta;
1950
1951	slf_frame.prepare (aTHX_ &ta);
1952	TRANSFER (ta, 0);
1953
1954	checkmark = PL_stack_sp - PL_stack_base;
1955	}
1956	while (slf_frame.check (aTHX));
1957
1958	{	2200	{
1959	dSP;	2201	dSP;
1960	SV **bot = PL_stack_base + checkmark;	2202	int i;
1961	int gimme = GIMME_V;
1962		2203
1963	slf_frame.prepare = 0; /* signal pp_slf that we need a new frame */	2204	EXTEND (SP, AvFILLp (state) + 1);
1964		2205	for (i = 0; i <= AvFILLp (state); ++i)
1965	/* make sure we put something on the stack in scalar context */	2206	PUSHs (sv_2mortal (SvREFCNT_inc_NN (AvARRAY (state)[i])));
1966	if (gimme == G_SCALAR)
1967	{
1968	if (sp == bot)
1969	XPUSHs (&PL_sv_undef);
1970
1971	SP = bot + 1;
1972	}
1973		2207
1974	PUTBACK;	2208	PUTBACK;
1975	}	2209	}
1976		2210
1977	return NORMAL;	2211	return 0;
1978	}	2212	}
1979		2213
1980	static void	2214	static void
1981	api_execute_slf (pTHX_ CV cv, const struct CoroSLF slf, SV **arg, int items)	2215	slf_init_aio_req (pTHX_ struct CoroSLF frame, CV cv, SV **arg, int items)
1982	{	2216	{
1983	assert (("FATAL: SLF call recursion in Coro module (please report)", PL_op->op_ppaddr != pp_slf));	2217	AV state = (AV )sv_2mortal ((SV *)newAV ());
1984	assert (("FATAL: SLF call with illegal CV value", !CvANON (cv)));	2218	SV *coro_hv = SvRV (coro_current);
		2219	struct coro *coro = SvSTATE_hv (coro_hv);
1985		2220
1986	if (items > 2)	2221	/* put our coroutine id on the state arg */
1987	croak ("Coro only supports a max of two arguments to SLF functions.");	2222	av_push (state, SvREFCNT_inc_NN (coro_hv));
1988		2223
1989	CvFLAGS (cv) \|= CVf_SLF;	2224	/* first see whether we have a non-zero priority and set it as AIO prio */
1990	CvXSUBANY (cv).any_ptr = (void *)slf;	2225	if (coro->prio)
1991	slf_cv = cv;	2226	{
		2227	dSP;
1992		2228
1993	/* we patch the op, and then re-run the whole call */	2229	static SV *prio_cv;
1994	/* we have to put the same argument on the stack for this to work */	2230	static SV *prio_sv;
1995	/* and this will be done by pp_restore */
1996	slf_restore.op_next = (OP *)&slf_restore;
1997	slf_restore.op_type = OP_NULL;
1998	slf_restore.op_ppaddr = pp_restore;
1999	slf_restore.op_first = PL_op;
2000		2231
2001	slf_arg0 = items > 0 ? SvREFCNT_inc (arg [0]) : 0;	2232	if (expect_false (!prio_cv))
2002	slf_arg1 = items > 1 ? SvREFCNT_inc (arg [1]) : 0;	2233	{
		2234	prio_cv = (SV *)get_cv ("IO::AIO::aioreq_pri", 0);
		2235	prio_sv = newSViv (0);
		2236	}
2003		2237
2004	PL_op->op_ppaddr = pp_slf;	2238	PUSHMARK (SP);
		2239	sv_setiv (prio_sv, coro->prio);
		2240	XPUSHs (prio_sv);
2005		2241
2006	PL_op = (OP *)&slf_restore;	2242	PUTBACK;
		2243	call_sv (prio_cv, G_VOID \| G_DISCARD);
		2244	}
		2245
		2246	/* now call the original request */
		2247	{
		2248	dSP;
		2249	CV req = (CV )CORO_MAGIC_NN ((SV *)cv, CORO_MAGIC_type_aio)->mg_obj;
		2250	int i;
		2251
		2252	PUSHMARK (SP);
		2253
		2254	/* first push all args to the stack */
		2255	EXTEND (SP, items + 1);
		2256
		2257	for (i = 0; i < items; ++i)
		2258	PUSHs (arg [i]);
		2259
		2260	/* now push the callback closure */
		2261	PUSHs (sv_2mortal (gensub (aTHX_ coro_aio_callback, (void )SvREFCNT_inc_NN ((SV )state))));
		2262
		2263	/* now call the AIO function - we assume our request is uncancelable */
		2264	PUTBACK;
		2265	call_sv ((SV *)req, G_VOID \| G_DISCARD);
		2266	}
		2267
		2268	/* now that the requets is going, we loop toll we have a result */
		2269	frame->data = (void *)state;
		2270	frame->prepare = prepare_schedule;
		2271	frame->check = slf_check_aio_req;
2007	}	2272	}
		2273
		2274	static void
		2275	coro_aio_req_xs (pTHX_ CV *cv)
		2276	{
		2277	dXSARGS;
		2278
		2279	CORO_EXECUTE_SLF_XS (slf_init_aio_req);
		2280
		2281	XSRETURN_EMPTY;
		2282	}
		2283
		2284	/*****************************************************************************/
2008		2285
2009	MODULE = Coro::State PACKAGE = Coro::State PREFIX = api_	2286	MODULE = Coro::State PACKAGE = Coro::State PREFIX = api_
2010		2287
2011	PROTOTYPES: DISABLE	2288	PROTOTYPES: DISABLE
2012		2289
2013	BOOT:	2290	BOOT:
2014	{	2291	{
2015	#ifdef USE_ITHREADS	2292	#ifdef USE_ITHREADS
2016	MUTEX_INIT (&coro_lock);
2017	# if CORO_PTHREAD	2293	# if CORO_PTHREAD
2018	coro_thx = PERL_GET_CONTEXT;	2294	coro_thx = PERL_GET_CONTEXT;
2019	# endif	2295	# endif
2020	#endif	2296	#endif
2021	BOOT_PAGESIZE;	2297	BOOT_PAGESIZE;
…		…
2042	main_top_env = PL_top_env;	2318	main_top_env = PL_top_env;
2043		2319
2044	while (main_top_env->je_prev)	2320	while (main_top_env->je_prev)
2045	main_top_env = main_top_env->je_prev;	2321	main_top_env = main_top_env->je_prev;
2046		2322
		2323	{
		2324	SV *slf = sv_2mortal (newSViv (PTR2IV (pp_slf)));
		2325
		2326	if (!PL_custom_op_names) PL_custom_op_names = newHV ();
		2327	hv_store_ent (PL_custom_op_names, slf,
		2328	newSVpv ("coro_slf", 0), 0);
		2329
		2330	if (!PL_custom_op_descs) PL_custom_op_descs = newHV ();
		2331	hv_store_ent (PL_custom_op_descs, slf,
		2332	newSVpv ("coro schedule like function", 0), 0);
		2333	}
		2334
2047	coroapi.ver = CORO_API_VERSION;	2335	coroapi.ver = CORO_API_VERSION;
2048	coroapi.rev = CORO_API_REVISION;	2336	coroapi.rev = CORO_API_REVISION;
		2337
2049	coroapi.transfer = api_transfer;	2338	coroapi.transfer = api_transfer;
		2339
		2340	coroapi.sv_state = SvSTATE_;
2050	coroapi.execute_slf = api_execute_slf;	2341	coroapi.execute_slf = api_execute_slf;
2051	coroapi.sv_state = SvSTATE_;	2342	coroapi.prepare_nop = prepare_nop;
		2343	coroapi.prepare_schedule = prepare_schedule;
		2344	coroapi.prepare_cede = prepare_cede;
		2345	coroapi.prepare_cede_notself = prepare_cede_notself;
2052		2346
2053	{	2347	{
2054	SV **svp = hv_fetch (PL_modglobal, "Time::NVtime", 12, 0);	2348	SV **svp = hv_fetch (PL_modglobal, "Time::NVtime", 12, 0);
2055		2349
2056	if (!svp) croak ("Time::HiRes is required");	2350	if (!svp) croak ("Time::HiRes is required");
…		…
2090	}	2384	}
2091	OUTPUT:	2385	OUTPUT:
2092	RETVAL	2386	RETVAL
2093		2387
2094	void	2388	void
2095	_set_stacklevel (...)	2389	transfer (...)
		2390	PROTOTYPE: $$
2096	CODE:	2391	CODE:
2097	{	2392	CORO_EXECUTE_SLF_XS (slf_init_transfer);
2098	static struct CoroSLF slf = { slf_init_set_stacklevel, slf_prepare_set_stacklevel, slf_check_nop };
2099	api_execute_slf (aTHX_ cv, &slf, &ST (0), items);
2100	}
2101
2102	void
2103	transfer (...)
2104	CODE:
2105	{
2106	static struct CoroSLF slf = { slf_init_transfer, slf_prepare_transfer, slf_check_nop };
2107	api_execute_slf (aTHX_ cv, &slf, &ST (0), items);
2108	}
2109		2393
2110	bool	2394	bool
2111	_destroy (SV *coro_sv)	2395	_destroy (SV *coro_sv)
2112	CODE:	2396	CODE:
2113	RETVAL = coro_state_destroy (aTHX_ SvSTATE (coro_sv));	2397	RETVAL = coro_state_destroy (aTHX_ SvSTATE (coro_sv));
…		…
2120	CODE:	2404	CODE:
2121	_exit (code);	2405	_exit (code);
2122		2406
2123	int	2407	int
2124	cctx_stacksize (int new_stacksize = 0)	2408	cctx_stacksize (int new_stacksize = 0)
		2409	PROTOTYPE: ;$
2125	CODE:	2410	CODE:
2126	RETVAL = cctx_stacksize;	2411	RETVAL = cctx_stacksize;
2127	if (new_stacksize)	2412	if (new_stacksize)
2128	{	2413	{
2129	cctx_stacksize = new_stacksize;	2414	cctx_stacksize = new_stacksize;
…		…
2132	OUTPUT:	2417	OUTPUT:
2133	RETVAL	2418	RETVAL
2134		2419
2135	int	2420	int
2136	cctx_max_idle (int max_idle = 0)	2421	cctx_max_idle (int max_idle = 0)
		2422	PROTOTYPE: ;$
2137	CODE:	2423	CODE:
2138	RETVAL = cctx_max_idle;	2424	RETVAL = cctx_max_idle;
2139	if (max_idle > 1)	2425	if (max_idle > 1)
2140	cctx_max_idle = max_idle;	2426	cctx_max_idle = max_idle;
2141	OUTPUT:	2427	OUTPUT:
2142	RETVAL	2428	RETVAL
2143		2429
2144	int	2430	int
2145	cctx_count ()	2431	cctx_count ()
		2432	PROTOTYPE:
2146	CODE:	2433	CODE:
2147	RETVAL = cctx_count;	2434	RETVAL = cctx_count;
2148	OUTPUT:	2435	OUTPUT:
2149	RETVAL	2436	RETVAL
2150		2437
2151	int	2438	int
2152	cctx_idle ()	2439	cctx_idle ()
		2440	PROTOTYPE:
2153	CODE:	2441	CODE:
2154	RETVAL = cctx_idle;	2442	RETVAL = cctx_idle;
2155	OUTPUT:	2443	OUTPUT:
2156	RETVAL	2444	RETVAL
2157		2445
2158	void	2446	void
2159	list ()	2447	list ()
		2448	PROTOTYPE:
2160	PPCODE:	2449	PPCODE:
2161	{	2450	{
2162	struct coro *coro;	2451	struct coro *coro;
2163	for (coro = coro_first; coro; coro = coro->next)	2452	for (coro = coro_first; coro; coro = coro->next)
2164	if (coro->hv)	2453	if (coro->hv)
…		…
2226		2515
2227	void	2516	void
2228	throw (Coro::State self, SV *throw = &PL_sv_undef)	2517	throw (Coro::State self, SV *throw = &PL_sv_undef)
2229	PROTOTYPE: $;$	2518	PROTOTYPE: $;$
2230	CODE:	2519	CODE:
		2520	{
		2521	struct coro *current = SvSTATE_current;
		2522	SV **throwp = self == current ? &CORO_THROW : &self->except;
2231	SvREFCNT_dec (self->throw);	2523	SvREFCNT_dec (*throwp);
2232	self->throw = SvOK (throw) ? newSVsv (throw) : 0;	2524	*throwp = SvOK (throw) ? newSVsv (throw) : 0;
		2525	}
2233		2526
2234	void	2527	void
2235	api_trace (SV *coro, int flags = CC_TRACE \| CC_TRACE_SUB)	2528	api_trace (SV *coro, int flags = CC_TRACE \| CC_TRACE_SUB)
		2529	PROTOTYPE: $;$
2236	C_ARGS: aTHX_ coro, flags	2530	C_ARGS: aTHX_ coro, flags
2237		2531
2238	SV *	2532	SV *
2239	has_cctx (Coro::State coro)	2533	has_cctx (Coro::State coro)
2240	PROTOTYPE: $	2534	PROTOTYPE: $
…		…
2265	OUTPUT:	2559	OUTPUT:
2266	RETVAL	2560	RETVAL
2267		2561
2268	void	2562	void
2269	force_cctx ()	2563	force_cctx ()
		2564	PROTOTYPE:
2270	CODE:	2565	CODE:
2271	struct coro *coro = SvSTATE (coro_current);
2272	coro->cctx->idle_sp = 0;	2566	SvSTATE_current->cctx->idle_sp = 0;
2273		2567
2274	void	2568	void
2275	swap_defsv (Coro::State self)	2569	swap_defsv (Coro::State self)
2276	PROTOTYPE: $	2570	PROTOTYPE: $
2277	ALIAS:	2571	ALIAS:
2278	swap_defav = 1	2572	swap_defav = 1
2279	CODE:	2573	CODE:
2280	if (!self->slot)	2574	if (!self->slot)
2281	croak ("cannot swap state with coroutine that has no saved state");	2575	croak ("cannot swap state with coroutine that has no saved state,");
2282	else	2576	else
2283	{	2577	{
2284	SV src = ix ? (SV )&GvAV (PL_defgv) : &GvSV (PL_defgv);	2578	SV src = ix ? (SV )&GvAV (PL_defgv) : &GvSV (PL_defgv);
2285	SV dst = ix ? (SV )&self->slot->defav : (SV **)&self->slot->defsv;	2579	SV dst = ix ? (SV )&self->slot->defav : (SV **)&self->slot->defsv;
2286		2580
2287	SV tmp = src; src = dst; *dst = tmp;	2581	SV tmp = src; src = dst; *dst = tmp;
2288	}	2582	}
		2583
2289		2584
2290	MODULE = Coro::State PACKAGE = Coro	2585	MODULE = Coro::State PACKAGE = Coro
2291		2586
2292	BOOT:	2587	BOOT:
2293	{	2588	{
…		…
2321	coroapi.ready = api_ready;	2616	coroapi.ready = api_ready;
2322	coroapi.is_ready = api_is_ready;	2617	coroapi.is_ready = api_is_ready;
2323	coroapi.nready = coro_nready;	2618	coroapi.nready = coro_nready;
2324	coroapi.current = coro_current;	2619	coroapi.current = coro_current;
2325		2620
2326	GCoroAPI = &coroapi;	2621	/GCoroAPI = &coroapi;/
2327	sv_setiv (sv, (IV)&coroapi);	2622	sv_setiv (sv, (IV)&coroapi);
2328	SvREADONLY_on (sv);	2623	SvREADONLY_on (sv);
2329	}	2624	}
2330	}	2625	}
2331		2626
2332	void	2627	void
2333	schedule (...)	2628	schedule (...)
2334	CODE:	2629	CODE:
2335	{	2630	CORO_EXECUTE_SLF_XS (slf_init_schedule);
2336	static struct CoroSLF slf = { slf_init_nop, prepare_schedule, slf_check_nop };
2337	api_execute_slf (aTHX_ cv, &slf, &ST (0), items);
2338	}
2339		2631
2340	void	2632	void
2341	cede (...)	2633	cede (...)
2342	CODE:	2634	CODE:
2343	{	2635	CORO_EXECUTE_SLF_XS (slf_init_cede);
2344	static struct CoroSLF slf = { slf_init_nop, prepare_cede, slf_check_nop };
2345	api_execute_slf (aTHX_ cv, &slf, &ST (0), items);
2346	}
2347		2636
2348	void	2637	void
2349	cede_notself (...)	2638	cede_notself (...)
2350	CODE:	2639	CODE:
2351	{	2640	CORO_EXECUTE_SLF_XS (slf_init_cede_notself);
2352	static struct CoroSLF slf = { slf_init_nop, prepare_cede_notself, slf_check_nop };
2353	api_execute_slf (aTHX_ cv, &slf, &ST (0), items);
2354	}
2355		2641
2356	void	2642	void
2357	_set_current (SV *current)	2643	_set_current (SV *current)
2358	PROTOTYPE: $	2644	PROTOTYPE: $
2359	CODE:	2645	CODE:
…		…
2362		2648
2363	void	2649	void
2364	_set_readyhook (SV *hook)	2650	_set_readyhook (SV *hook)
2365	PROTOTYPE: $	2651	PROTOTYPE: $
2366	CODE:	2652	CODE:
2367	LOCK;
2368	SvREFCNT_dec (coro_readyhook);	2653	SvREFCNT_dec (coro_readyhook);
2369	coro_readyhook = SvOK (hook) ? newSVsv (hook) : 0;	2654	coro_readyhook = SvOK (hook) ? newSVsv (hook) : 0;
2370	UNLOCK;
2371		2655
2372	int	2656	int
2373	prio (Coro::State coro, int newprio = 0)	2657	prio (Coro::State coro, int newprio = 0)
		2658	PROTOTYPE: $;$
2374	ALIAS:	2659	ALIAS:
2375	nice = 1	2660	nice = 1
2376	CODE:	2661	CODE:
2377	{	2662	{
2378	RETVAL = coro->prio;	2663	RETVAL = coro->prio;
…		…
2410	# for async_pool speedup	2695	# for async_pool speedup
2411	void	2696	void
2412	_pool_1 (SV *cb)	2697	_pool_1 (SV *cb)
2413	CODE:	2698	CODE:
2414	{	2699	{
2415	struct coro *coro = SvSTATE (coro_current);
2416	HV hv = (HV )SvRV (coro_current);	2700	HV hv = (HV )SvRV (coro_current);
		2701	struct coro coro = SvSTATE_hv ((SV )hv);
2417	AV *defav = GvAV (PL_defgv);	2702	AV *defav = GvAV (PL_defgv);
2418	SV *invoke = hv_delete (hv, "_invoke", sizeof ("_invoke") - 1, 0);	2703	SV *invoke = hv_delete (hv, "_invoke", sizeof ("_invoke") - 1, 0);
2419	AV *invoke_av;	2704	AV *invoke_av;
2420	int i, len;	2705	int i, len;
2421		2706
…		…
2442	{	2727	{
2443	av_fill (defav, len - 1);	2728	av_fill (defav, len - 1);
2444	for (i = 0; i < len; ++i)	2729	for (i = 0; i < len; ++i)
2445	av_store (defav, i, SvREFCNT_inc_NN (AvARRAY (invoke_av)[i + 1]));	2730	av_store (defav, i, SvREFCNT_inc_NN (AvARRAY (invoke_av)[i + 1]));
2446	}	2731	}
2447
2448	SvREFCNT_dec (invoke);
2449	}	2732	}
2450		2733
2451	void	2734	void
2452	_pool_2 (SV *cb)	2735	_pool_2 (SV *cb)
2453	CODE:	2736	CODE:
2454	{	2737	{
2455	struct coro *coro = SvSTATE (coro_current);	2738	HV hv = (HV )SvRV (coro_current);
		2739	struct coro coro = SvSTATE_hv ((SV )hv);
2456		2740
2457	sv_setsv (cb, &PL_sv_undef);	2741	sv_setsv (cb, &PL_sv_undef);
2458		2742
2459	SvREFCNT_dec ((SV )PL_defoutgv); PL_defoutgv = (GV )coro->saved_deffh;	2743	SvREFCNT_dec ((SV )PL_defoutgv); PL_defoutgv = (GV )coro->saved_deffh;
2460	coro->saved_deffh = 0;	2744	coro->saved_deffh = 0;
…		…
2467	SvREFCNT_dec (old);	2751	SvREFCNT_dec (old);
2468	croak ("\3async_pool terminate\2\n");	2752	croak ("\3async_pool terminate\2\n");
2469	}	2753	}
2470		2754
2471	av_clear (GvAV (PL_defgv));	2755	av_clear (GvAV (PL_defgv));
2472	hv_store ((HV *)SvRV (coro_current), "desc", sizeof ("desc") - 1,	2756	hv_store (hv, "desc", sizeof ("desc") - 1,
2473	newSVpvn ("[async_pool idle]", sizeof ("[async_pool idle]") - 1), 0);	2757	newSVpvn ("[async_pool idle]", sizeof ("[async_pool idle]") - 1), 0);
2474		2758
2475	coro->prio = 0;	2759	coro->prio = 0;
2476		2760
2477	if (coro->cctx && (coro->cctx->flags & CC_TRACE))	2761	if (coro->cctx && (coro->cctx->flags & CC_TRACE))
2478	api_trace (aTHX_ coro_current, 0);	2762	api_trace (aTHX_ coro_current, 0);
2479		2763
2480	av_push (av_async_pool, newSVsv (coro_current));	2764	av_push (av_async_pool, newSVsv (coro_current));
2481	}	2765	}
2482		2766
2483	#if 0
2484		2767
2485	void	2768	MODULE = Coro::State PACKAGE = PerlIO::cede
2486	_generator_call (...)	2769
2487	PROTOTYPE: @	2770	BOOT:
2488	PPCODE:	2771	PerlIO_define_layer (aTHX_ &PerlIO_cede);
2489	fprintf (stderr, "call %p\n", CvXSUBANY(cv).any_ptr);	2772
2490	xxxx	2773
2491	abort ();	2774	MODULE = Coro::State PACKAGE = Coro::Semaphore
2492		2775
2493	SV *	2776	SV *
2494	gensub (SV *sub, ...)	2777	new (SV klass, SV count_ = 0)
2495	PROTOTYPE: &;@	2778	CODE:
2496	CODE:
2497	{	2779	{
2498	struct coro *coro;	2780	/* a semaphore contains a counter IV in $sem->[0] and any waiters after that */
2499	MAGIC *mg;	2781	AV *av = newAV ();
2500	CV *xcv;	2782	SV **ary;
2501	CV ncv = (CV )newSV_type (SVt_PVCV);
2502	int i;
2503		2783
2504	CvGV (ncv) = CvGV (cv);	2784	/* unfortunately, building manually saves memory */
2505	CvFILE (ncv) = CvFILE (cv);	2785	Newx (ary, 2, SV *);
		2786	AvALLOC (av) = ary;
		2787	AvARRAY (av) = ary;
		2788	AvMAX (av) = 1;
		2789	AvFILLp (av) = 0;
		2790	ary [0] = newSViv (count_ && SvOK (count_) ? SvIV (count_) : 1);
2506		2791
2507	Newz (0, coro, 1, struct coro);	2792	RETVAL = sv_bless (newRV_noinc ((SV *)av), GvSTASH (CvGV (cv)));
2508	coro->args = newAV ();
2509	coro->flags = CF_NEW;
2510
2511	av_extend (coro->args, items - 1);
2512	for (i = 1; i < items; i++)
2513	av_push (coro->args, newSVsv (ST (i)));
2514
2515	CvISXSUB_on (ncv);
2516	CvXSUBANY (ncv).any_ptr = (void *)coro;
2517
2518	xcv = GvCV (gv_fetchpv ("Coro::_generator_call", 0, SVt_PVCV));
2519
2520	CvXSUB (ncv) = CvXSUB (xcv);
2521	CvANON_on (ncv);
2522
2523	mg = sv_magicext ((SV )ncv, 0, CORO_MAGIC_type_state, &coro_gensub_vtbl, (char )coro, 0);
2524	RETVAL = newRV_noinc ((SV *)ncv);
2525	}	2793	}
2526	OUTPUT:	2794	OUTPUT:
2527	RETVAL	2795	RETVAL
2528		2796
2529	#endif	2797	SV *
2530		2798	count (SV *self)
2531		2799	CODE:
2532	MODULE = Coro::State PACKAGE = Coro::AIO	2800	RETVAL = newSVsv (AvARRAY ((AV *)SvRV (self))[0]);
		2801	OUTPUT:
		2802	RETVAL
2533		2803
2534	void	2804	void
2535	_get_state (SV *self)	2805	up (SV *self, int adjust = 1)
2536	PPCODE:	2806	ALIAS:
2537	{	2807	adjust = 1
2538	AV *defav = GvAV (PL_defgv);	2808	CODE:
2539	AV *av = newAV ();	2809	coro_semaphore_adjust (aTHX_ (AV *)SvRV (self), ix ? adjust : 1);
2540	int i;
2541	SV *data_sv = newSV (sizeof (struct io_state));
2542	struct io_state data = (struct io_state )SvPVX (data_sv);
2543	SvCUR_set (data_sv, sizeof (struct io_state));
2544	SvPOK_only (data_sv);
2545
2546	data->errorno = errno;
2547	data->laststype = PL_laststype;
2548	data->laststatval = PL_laststatval;
2549	data->statcache = PL_statcache;
2550
2551	av_extend (av, AvFILLp (defav) + 1 + 1);
2552
2553	for (i = 0; i <= AvFILLp (defav); ++i)
2554	av_push (av, SvREFCNT_inc_NN (AvARRAY (defav)[i]));
2555
2556	av_push (av, data_sv);
2557
2558	XPUSHs (sv_2mortal (newRV_noinc ((SV *)av)));
2559
2560	api_ready (aTHX_ self);
2561	}
2562		2810
2563	void	2811	void
2564	_set_state (SV *state)	2812	down (SV *self)
2565	PROTOTYPE: $	2813	CODE:
2566	PPCODE:	2814	CORO_EXECUTE_SLF_XS (slf_init_semaphore_down);
		2815
		2816	void
		2817	try (SV *self)
		2818	PPCODE:
2567	{	2819	{
2568	AV av = (AV )SvRV (state);	2820	AV av = (AV )SvRV (self);
2569	struct io_state data = (struct io_state )SvPVX (AvARRAY (av)[AvFILLp (av)]);	2821	SV *count_sv = AvARRAY (av)[0];
		2822	IV count = SvIVX (count_sv);
		2823
		2824	if (count > 0)
		2825	{
		2826	--count;
		2827	SvIVX (count_sv) = count;
		2828	XSRETURN_YES;
		2829	}
		2830	else
		2831	XSRETURN_NO;
		2832	}
		2833
		2834	void
		2835	waiters (SV *self)
		2836	CODE:
		2837	{
		2838	AV av = (AV )SvRV (self);
		2839
		2840	if (GIMME_V == G_SCALAR)
		2841	XPUSHs (sv_2mortal (newSVsv (AvARRAY (av)[0])));
		2842	else
		2843	{
2570	int i;	2844	int i;
2571
2572	errno = data->errorno;
2573	PL_laststype = data->laststype;
2574	PL_laststatval = data->laststatval;
2575	PL_statcache = data->statcache;
2576
2577	EXTEND (SP, AvFILLp (av));	2845	EXTEND (SP, AvFILLp (av) + 1 - 1);
2578	for (i = 0; i < AvFILLp (av); ++i)	2846	for (i = 1; i <= AvFILLp (av); ++i)
2579	PUSHs (sv_2mortal (SvREFCNT_inc_NN (AvARRAY (av)[i])));	2847	PUSHs (sv_2mortal (newRV_inc (AvARRAY (av)[i])));
		2848	}
2580	}	2849	}
2581		2850
2582		2851
2583	MODULE = Coro::State PACKAGE = Coro::AnyEvent	2852	MODULE = Coro::State PACKAGE = Coro::AnyEvent
2584		2853
2585	BOOT:	2854	BOOT:
2586	sv_activity = coro_get_sv (aTHX_ "Coro::AnyEvent::ACTIVITY", TRUE);	2855	sv_activity = coro_get_sv (aTHX_ "Coro::AnyEvent::ACTIVITY", TRUE);
2587		2856
2588	SV *	2857	void
2589	_schedule (...)	2858	_schedule (...)
2590	PROTOTYPE: @
2591	CODE:	2859	CODE:
2592	{	2860	{
2593	static int incede;	2861	static int incede;
2594		2862
2595	api_cede_notself (aTHX);	2863	api_cede_notself (aTHX);
…		…
2601	sv_setsv (sv_activity, &PL_sv_undef);	2869	sv_setsv (sv_activity, &PL_sv_undef);
2602	if (coro_nready >= incede)	2870	if (coro_nready >= incede)
2603	{	2871	{
2604	PUSHMARK (SP);	2872	PUSHMARK (SP);
2605	PUTBACK;	2873	PUTBACK;
2606	call_pv ("Coro::AnyEvent::_activity", G_DISCARD \| G_EVAL);	2874	call_pv ("Coro::AnyEvent::_activity", G_KEEPERR \| G_EVAL \| G_VOID \| G_DISCARD);
2607	SPAGAIN;
2608	}	2875	}
2609		2876
2610	--incede;	2877	--incede;
2611	}	2878	}
2612		2879
2613		2880
2614	MODULE = Coro::State PACKAGE = PerlIO::cede	2881	MODULE = Coro::State PACKAGE = Coro::AIO
2615		2882
2616	BOOT:	2883	void
2617	PerlIO_define_layer (aTHX_ &PerlIO_cede);	2884	_register (char target, char proto, SV *req)
		2885	CODE:
		2886	{
		2887	HV *st;
		2888	GV *gvp;
		2889	CV *req_cv = sv_2cv (req, &st, &gvp, 0);
		2890	/* newXSproto doesn't return the CV on 5.8 */
		2891	CV *slf_cv = newXS (target, coro_aio_req_xs, __FILE__);
		2892	sv_setpv ((SV *)slf_cv, proto);
		2893	sv_magicext ((SV )slf_cv, (SV )req_cv, CORO_MAGIC_type_aio, 0, 0, 0);
		2894	}
2618		2895

Diff Legend

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

Comparing Coro/Coro/State.xs (file contents): Revision 1.272 by root, Fri Nov 14 20:35:49 2008 UTC vs. Revision 1.297 by root, Tue Nov 18 11:04:42 2008 UTC

Diff Legend

Comparing Coro/Coro/State.xs (file contents):
Revision 1.272 by root, Fri Nov 14 20:35:49 2008 UTC vs.
Revision 1.297 by root, Tue Nov 18 11:04:42 2008 UTC