ViewVC Help

View File | Revision Log | Show Annotations | Download File

/cvs/Coro/Coro/State.xs

Comparing Coro/Coro/State.xs (file contents):
Revision 1.280 by root, Sun Nov 16 09:43:18 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	# if CORO_PTHREAD	153	# if CORO_PTHREAD
148	static void *coro_thx;	154	static void *coro_thx;
149	# endif	155	# endif
150	#endif	156	#endif
151		157
152	/* helper storage struct for Coro::AIO */
153	struct io_state
154	{
155	AV *res;
156	int errorno;
157	I32 laststype; /* U16 in 5.10.0 */
158	int laststatval;
159	Stat_t statcache;
160	};
161
162	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);
163		163
164	static U32 cctx_gen;	164	static U32 cctx_gen;
165	static size_t cctx_stacksize = CORO_STACKSIZE;	165	static size_t cctx_stacksize = CORO_STACKSIZE;
166	static struct CoroAPI coroapi;	166	static struct CoroAPI coroapi;
167	static AV *main_mainstack; /* used to differentiate between $main and others */	167	static AV *main_mainstack; /* used to differentiate between $main and others */
168	static JMPENV *main_top_env;	168	static JMPENV *main_top_env;
169	static HV *coro_state_stash, *coro_stash;	169	static HV *coro_state_stash, *coro_stash;
170	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 */
171	static volatile struct coro *transfer_next;
172		171
173	static GV *irsgv; /* $/ */	172	static GV *irsgv; /* $/ */
174	static GV *stdoutgv; /* *STDOUT */	173	static GV *stdoutgv; /* *STDOUT */
175	static SV *rv_diehook;	174	static SV *rv_diehook;
176	static SV *rv_warnhook;	175	static SV *rv_warnhook;
…		…
242	/* this is a structure representing a perl-level coroutine */	241	/* this is a structure representing a perl-level coroutine */
243	struct coro {	242	struct coro {
244	/* the C coroutine allocated to this perl coroutine, if any */	243	/* the C coroutine allocated to this perl coroutine, if any */
245	coro_cctx *cctx;	244	coro_cctx *cctx;
246		245
247	/* process data */	246	/* state data */
248	struct CoroSLF slf_frame; /* saved slf frame */	247	struct CoroSLF slf_frame; /* saved slf frame */
249	AV *mainstack;	248	AV *mainstack;
250	perl_slots *slot; /* basically the saved sp */	249	perl_slots *slot; /* basically the saved sp */
251		250
252	AV *args; /* data associated with this coroutine (initial args) */	251	AV *args; /* data associated with this coroutine (initial args) */
…		…
258	/* statistics */	257	/* statistics */
259	int usecount; /* number of transfers to this coro */	258	int usecount; /* number of transfers to this coro */
260		259
261	/* coro process data */	260	/* coro process data */
262	int prio;	261	int prio;
263	SV *throw; /* exception to be thrown */	262	SV *except; /* exception to be thrown */
264		263
265	/* async_pool */	264	/* async_pool */
266	SV *saved_deffh;	265	SV *saved_deffh;
267		266
268	/* linked list */	267	/* linked list */
…		…
270	};	269	};
271		270
272	typedef struct coro *Coro__State;	271	typedef struct coro *Coro__State;
273	typedef struct coro *Coro__State_or_hashref;	272	typedef struct coro *Coro__State_or_hashref;
274		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 */
275	static struct CoroSLF slf_frame; /* the current slf frame */	277	static struct CoroSLF slf_frame; /* the current slf frame */
276		278
277	/** Coro ********************************************************************/	279	/** Coro ********************************************************************/
278		280
279	#define PRIO_MAX 3	281	#define PRIO_MAX 3
…		…
381	SvREFCNT_dec (av); /* sv_magicext increased the refcount */	383	SvREFCNT_dec (av); /* sv_magicext increased the refcount */
382		384
383	return 0;	385	return 0;
384	}	386	}
385		387
386	#define CORO_MAGIC_type_cv PERL_MAGIC_ext	388	#define CORO_MAGIC_type_cv 26
387	#define CORO_MAGIC_type_state PERL_MAGIC_ext	389	#define CORO_MAGIC_type_state PERL_MAGIC_ext
388		390
389	static MGVTBL coro_cv_vtbl = {	391	static MGVTBL coro_cv_vtbl = {
390	0, 0, 0, 0,	392	0, 0, 0, 0,
391	coro_cv_free	393	coro_cv_free
392	};	394	};
393		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
394	#define CORO_MAGIC(sv, type) \	401	#define CORO_MAGIC(sv, type) \
395	SvMAGIC (sv) \	402	(expect_true (SvMAGIC (sv)) \
396	? SvMAGIC (sv)->mg_type == type \	403	? CORO_MAGIC_NN (sv, type) \
397	? SvMAGIC (sv) \
398	: mg_find (sv, type) \
399	: 0	404	: 0)
400		405
401	#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)
402	#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)
403		408
404	INLINE struct coro *	409	INLINE struct coro *
405	SvSTATE_ (pTHX_ SV *coro)	410	SvSTATE_ (pTHX_ SV *coro)
406	{	411	{
407	HV *stash;	412	HV *stash;
…		…
424	mg = CORO_MAGIC_state (coro);	429	mg = CORO_MAGIC_state (coro);
425	return (struct coro *)mg->mg_ptr;	430	return (struct coro *)mg->mg_ptr;
426	}	431	}
427		432
428	#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))
429		438
430	/* the next two functions merely cache the padlists */	439	/* the next two functions merely cache the padlists */
431	static void	440	static void
432	get_padlist (pTHX_ CV *cv)	441	get_padlist (pTHX_ CV *cv)
433	{	442	{
…		…
501	}	510	}
502		511
503	PUTBACK;	512	PUTBACK;
504	}	513	}
505		514
506	slf_frame = c->slf_frame;	515	slf_frame = c->slf_frame;
		516	CORO_THROW = c->except;
507	}	517	}
508		518
509	static void	519	static void
510	save_perl (pTHX_ Coro__State c)	520	save_perl (pTHX_ Coro__State c)
511	{	521	{
		522	c->except = CORO_THROW;
512	c->slf_frame = slf_frame;	523	c->slf_frame = slf_frame;
513		524
514	{	525	{
515	dSP;	526	dSP;
516	I32 cxix = cxstack_ix;	527	I32 cxix = cxstack_ix;
…		…
809	slf_check_nop (pTHX_ struct CoroSLF *frame)	820	slf_check_nop (pTHX_ struct CoroSLF *frame)
810	{	821	{
811	return 0;	822	return 0;
812	}	823	}
813		824
814	static void	825	static UNOP coro_setup_op;
		826
		827	static void NOINLINE /* noinline to keep it out of the transfer fast path */
815	coro_setup (pTHX_ struct coro *coro)	828	coro_setup (pTHX_ struct coro *coro)
816	{	829	{
817	/*	830	/*
818	* emulate part of the perl startup here.	831	* emulate part of the perl startup here.
819	*/	832	*/
…		…
846	{	859	{
847	dSP;	860	dSP;
848	UNOP myop;	861	UNOP myop;
849		862
850	Zero (&myop, 1, UNOP);	863	Zero (&myop, 1, UNOP);
851	myop.op_next = Nullop;	864	myop.op_next = Nullop;
852	myop.op_flags = OPf_WANT_VOID;	865	myop.op_flags = OPf_WANT_VOID;
853		866
854	PUSHMARK (SP);	867	PUSHMARK (SP);
855	XPUSHs (sv_2mortal (av_shift (GvAV (PL_defgv))));	868	XPUSHs (sv_2mortal (av_shift (GvAV (PL_defgv))));
856	PUTBACK;	869	PUTBACK;
…		…
862	/* 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
863	* 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.
864	*/	877	*/
865	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 */
866	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;
867	}	891	}
868		892
869	static void	893	static void
870	coro_destruct (pTHX_ struct coro *coro)	894	coro_destruct (pTHX_ struct coro *coro)
871	{	895	{
…		…
895		919
896	SvREFCNT_dec (PL_diehook);	920	SvREFCNT_dec (PL_diehook);
897	SvREFCNT_dec (PL_warnhook);	921	SvREFCNT_dec (PL_warnhook);
898		922
899	SvREFCNT_dec (coro->saved_deffh);	923	SvREFCNT_dec (coro->saved_deffh);
900	SvREFCNT_dec (coro->throw);	924	SvREFCNT_dec (CORO_THROW);
901		925
902	coro_destruct_stacks (aTHX);	926	coro_destruct_stacks (aTHX);
903	}	927	}
904		928
905	INLINE void	929	INLINE void
…		…
915	static int	939	static int
916	runops_trace (pTHX)	940	runops_trace (pTHX)
917	{	941	{
918	COP *oldcop = 0;	942	COP *oldcop = 0;
919	int oldcxix = -2;	943	int oldcxix = -2;
920	struct coro *coro = SvSTATE (coro_current); /* trace cctx is tied to specific coro */	944	struct coro *coro = SvSTATE_current; /* trace cctx is tied to specific coro */
921	coro_cctx *cctx = coro->cctx;	945	coro_cctx *cctx = coro->cctx;
922		946
923	while ((PL_op = CALL_FPTR (PL_op->op_ppaddr) (aTHX)))	947	while ((PL_op = CALL_FPTR (PL_op->op_ppaddr) (aTHX)))
924	{	948	{
925	PERL_ASYNC_CHECK ();	949	PERL_ASYNC_CHECK ();
…		…
1034		1058
1035	TAINT_NOT;	1059	TAINT_NOT;
1036	return 0;	1060	return 0;
1037	}	1061	}
1038		1062
		1063	static struct coro_cctx *cctx_ssl_cctx;
		1064	static struct CoroSLF cctx_ssl_frame;
		1065
1039	static void	1066	static void
1040	prepare_set_stacklevel (struct coro_transfer_args *ta, struct coro_cctx *cctx)	1067	slf_prepare_set_stacklevel (pTHX_ struct coro_transfer_args *ta)
1041	{	1068	{
1042	ta->prev = (struct coro *)cctx;	1069	ta->prev = (struct coro *)cctx_ssl_cctx;
1043	ta->next = 0;	1070	ta->next = 0;
1044	}	1071	}
1045		1072
1046	/* inject a fake call to Coro::State::_cctx_init into the execution */	1073	static int
1047	/* _cctx_init should be careful, as it could be called at almost any time */	1074	slf_check_set_stacklevel (pTHX_ struct CoroSLF *frame)
1048	/* during execution of a perl program */	1075	{
1049	/* also initialises PL_top_env */	1076	*frame = cctx_ssl_frame;
		1077
		1078	return frame->check (aTHX_ frame); /* execute the restored frame - there must be one */
		1079	}
		1080
		1081	/* initialises PL_top_env and injects a pseudo-slf-call to set the stacklevel */
1050	static void NOINLINE	1082	static void NOINLINE
1051	cctx_prepare (pTHX_ coro_cctx *cctx)	1083	cctx_prepare (pTHX_ coro_cctx *cctx)
1052	{	1084	{
1053	dSP;
1054	UNOP myop;
1055
1056	PL_top_env = &PL_start_env;	1085	PL_top_env = &PL_start_env;
1057		1086
1058	if (cctx->flags & CC_TRACE)	1087	if (cctx->flags & CC_TRACE)
1059	PL_runops = runops_trace;	1088	PL_runops = runops_trace;
1060		1089
1061	Zero (&myop, 1, UNOP);	1090	/* we already must be executing an SLF op, there is no other valid way
1062	myop.op_next = PL_op;	1091	* that can lead to creation of a new cctx */
1063	myop.op_flags = OPf_WANT_VOID | OPf_STACKED;	1092	assert (("FATAL: can't prepare slf-less cctx in Coro module (please report)",
		1093	slf_frame.prepare && PL_op->op_ppaddr == pp_slf));
1064		1094
1065	PUSHMARK (SP);	1095	/* we must emulate leaving pp_slf, which is done inside slf_check_set_stacklevel */
1066	EXTEND (SP, 2);	1096	cctx_ssl_cctx = cctx;
1067	PUSHs (sv_2mortal (newSViv ((IV)cctx)));	1097	cctx_ssl_frame = slf_frame;
1068	PUSHs ((SV *)get_cv ("Coro::State::_cctx_init", FALSE));	1098
1069	PUTBACK;	1099	slf_frame.prepare = slf_prepare_set_stacklevel;
1070	PL_op = (OP *)&myop;	1100	slf_frame.check = slf_check_set_stacklevel;
1071	PL_op = PL_ppaddr[OP_ENTERSUB](aTHX);
1072	SPAGAIN;
1073	}	1101	}
1074		1102
1075	/* the tail of transfer: execute stuff we can only do after a transfer */	1103	/* the tail of transfer: execute stuff we can only do after a transfer */
1076	INLINE void	1104	INLINE void
1077	transfer_tail (pTHX)	1105	transfer_tail (pTHX)
1078	{	1106	{
1079	struct coro *next = (struct coro *)transfer_next;
1080	assert (!(transfer_next = 0)); /* just used for the side effect when asserts are enabled */
1081	assert (("FATAL: next coroutine was zero in transfer_tail (please report)", next));
1082
1083	free_coro_mortal (aTHX);	1107	free_coro_mortal (aTHX);
1084
1085	if (expect_false (next->throw))
1086	{
1087	SV *exception = sv_2mortal (next->throw);
1088
1089	next->throw = 0;
1090	sv_setsv (ERRSV, exception);
1091	croak (0);
1092	}
1093	}	1108	}
1094		1109
1095	/*	1110	/*
1096	* this is a _very_ stripped down perl interpreter ;)	1111	* this is a _very_ stripped down perl interpreter ;)
1097	*/	1112	*/
…		…
1112		1127
1113	/* inject a fake subroutine call to cctx_init */	1128	/* inject a fake subroutine call to cctx_init */
1114	cctx_prepare (aTHX_ (coro_cctx *)arg);	1129	cctx_prepare (aTHX_ (coro_cctx *)arg);
1115		1130
1116	/* cctx_run is the alternative tail of transfer() */	1131	/* cctx_run is the alternative tail of transfer() */
1117	/* TODO: throwing an exception here might be deadly, VERIFY */
1118	transfer_tail (aTHX);	1132	transfer_tail (aTHX);
1119		1133
1120	/* 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 */
1121	PL_restartop = PL_op;	1135	PL_restartop = PL_op;
1122	perl_run (PL_curinterp);	1136	perl_run (PL_curinterp);
…		…
1278	/** coroutine switching *****************************************************/	1292	/** coroutine switching *****************************************************/
1279		1293
1280	static void	1294	static void
1281	transfer_check (pTHX_ struct coro *prev, struct coro *next)	1295	transfer_check (pTHX_ struct coro *prev, struct coro *next)
1282	{	1296	{
		1297	/* TODO: throwing up here is considered harmful */
		1298
1283	if (expect_true (prev != next))	1299	if (expect_true (prev != next))
1284	{	1300	{
1285	if (expect_false (!(prev->flags & (CF_RUNNING | CF_NEW))))	1301	if (expect_false (!(prev->flags & (CF_RUNNING | CF_NEW))))
1286	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,");
1287		1303
…		…
1297	#endif	1313	#endif
1298	}	1314	}
1299	}	1315	}
1300		1316
1301	/* always use the TRANSFER macro */	1317	/* always use the TRANSFER macro */
1302	static void NOINLINE	1318	static void NOINLINE /* noinline so we have a fixed stackframe */
1303	transfer (pTHX_ struct coro *prev, struct coro *next, int force_cctx)	1319	transfer (pTHX_ struct coro *prev, struct coro *next, int force_cctx)
1304	{	1320	{
1305	dSTACKLEVEL;	1321	dSTACKLEVEL;
1306		1322
1307	/* sometimes transfer is only called to set idle_sp */	1323	/* sometimes transfer is only called to set idle_sp */
1308	if (expect_false (!next))	1324	if (expect_false (!next))
1309	{	1325	{
1310	((coro_cctx *)prev)->idle_sp = (void *)stacklevel;	1326	((coro_cctx *)prev)->idle_sp = STACKLEVEL;
1311	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 */
1312	}	1328	}
1313	else if (expect_true (prev != next))	1329	else if (expect_true (prev != next))
1314	{	1330	{
1315	coro_cctx *prev__cctx;	1331	coro_cctx *prev__cctx;
…		…
1340		1356
1341	prev__cctx = prev->cctx;	1357	prev__cctx = prev->cctx;
1342		1358
1343	/* possibly untie and reuse the cctx */	1359	/* possibly untie and reuse the cctx */
1344	if (expect_true (	1360	if (expect_true (
1345	prev__cctx->idle_sp == (void *)stacklevel	1361	prev__cctx->idle_sp == STACKLEVEL
1346	&& !(prev__cctx->flags & CC_TRACE)	1362	&& !(prev__cctx->flags & CC_TRACE)
1347	&& !force_cctx	1363	&& !force_cctx
1348	))	1364	))
1349	{	1365	{
1350	/* 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 */
…		…
1363		1379
1364	++next->usecount;	1380	++next->usecount;
1365		1381
1366	if (expect_true (!next->cctx))	1382	if (expect_true (!next->cctx))
1367	next->cctx = cctx_get (aTHX);	1383	next->cctx = cctx_get (aTHX);
1368
1369	assert (("FATAL: transfer_next already nonzero in Coro (please report)", !transfer_next));
1370	transfer_next = next;
1371		1384
1372	if (expect_false (prev__cctx != next->cctx))	1385	if (expect_false (prev__cctx != next->cctx))
1373	{	1386	{
1374	prev__cctx->top_env = PL_top_env;	1387	prev__cctx->top_env = PL_top_env;
1375	PL_top_env = next->cctx->top_env;	1388	PL_top_env = next->cctx->top_env;
…		…
1486	TRANSFER (ta, 1);	1499	TRANSFER (ta, 1);
1487	}	1500	}
1488		1501
1489	/** Coro ********************************************************************/	1502	/** Coro ********************************************************************/
1490		1503
1491	static void	1504	INLINE void
1492	coro_enq (pTHX_ SV *coro_sv)	1505	coro_enq (pTHX_ struct coro *coro)
1493	{	1506	{
1494	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));
1495	}	1508	}
1496		1509
1497	static SV *	1510	INLINE SV *
1498	coro_deq (pTHX)	1511	coro_deq (pTHX)
1499	{	1512	{
1500	int prio;	1513	int prio;
1501		1514
1502	for (prio = PRIO_MAX - PRIO_MIN + 1; --prio >= 0; )	1515	for (prio = PRIO_MAX - PRIO_MIN + 1; --prio >= 0; )
…		…
1524	coro->flags |= CF_READY;	1537	coro->flags |= CF_READY;
1525		1538
1526	sv_hook = coro_nready ? 0 : coro_readyhook;	1539	sv_hook = coro_nready ? 0 : coro_readyhook;
1527	xs_hook = coro_nready ? 0 : coroapi.readyhook;	1540	xs_hook = coro_nready ? 0 : coroapi.readyhook;
1528		1541
1529	coro_enq (aTHX_ SvREFCNT_inc_NN (coro_sv));	1542	coro_enq (aTHX_ coro);
1530	++coro_nready;	1543	++coro_nready;
1531		1544
1532	if (sv_hook)	1545	if (sv_hook)
1533	{	1546	{
1534	dSP;	1547	dSP;
…		…
1536	ENTER;	1549	ENTER;
1537	SAVETMPS;	1550	SAVETMPS;
1538		1551
1539	PUSHMARK (SP);	1552	PUSHMARK (SP);
1540	PUTBACK;	1553	PUTBACK;
1541	call_sv (sv_hook, G_DISCARD);	1554	call_sv (sv_hook, G_VOID | G_DISCARD);
1542	SPAGAIN;
1543		1555
1544	FREETMPS;	1556	FREETMPS;
1545	LEAVE;	1557	LEAVE;
1546	}	1558	}
1547		1559
…		…
1574	ENTER;	1586	ENTER;
1575	SAVETMPS;	1587	SAVETMPS;
1576		1588
1577	PUSHMARK (SP);	1589	PUSHMARK (SP);
1578	PUTBACK;	1590	PUTBACK;
1579	call_sv (get_sv ("Coro::idle", FALSE), G_DISCARD);	1591	call_sv (get_sv ("Coro::idle", FALSE), G_VOID | G_DISCARD);
1580	SPAGAIN;
1581		1592
1582	FREETMPS;	1593	FREETMPS;
1583	LEAVE;	1594	LEAVE;
1584	continue;	1595	continue;
1585	}	1596	}
1586		1597
1587	ta->next = SvSTATE (next_sv);	1598	ta->next = SvSTATE_hv (next_sv);
1588		1599
1589	/* cannot transfer to destroyed coros, skip and look for next */	1600	/* cannot transfer to destroyed coros, skip and look for next */
1590	if (expect_false (ta->next->flags & CF_DESTROYED))	1601	if (expect_false (ta->next->flags & CF_DESTROYED))
1591	{	1602	{
1592	SvREFCNT_dec (next_sv);	1603	SvREFCNT_dec (next_sv);
…		…
1598	break;	1609	break;
1599	}	1610	}
1600		1611
1601	/* free this only after the transfer */	1612	/* free this only after the transfer */
1602	prev_sv = SvRV (coro_current);	1613	prev_sv = SvRV (coro_current);
1603	ta->prev = SvSTATE (prev_sv);	1614	ta->prev = SvSTATE_hv (prev_sv);
1604	TRANSFER_CHECK (*ta);	1615	TRANSFER_CHECK (*ta);
1605	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));
1606	ta->next->flags &= ~CF_READY;	1617	ta->next->flags &= ~CF_READY;
1607	SvRV_set (coro_current, next_sv);	1618	SvRV_set (coro_current, next_sv);
1608		1619
…		…
1692	if (coro->flags & CF_RUNNING)	1703	if (coro->flags & CF_RUNNING)
1693	PL_runops = RUNOPS_DEFAULT;	1704	PL_runops = RUNOPS_DEFAULT;
1694	else	1705	else
1695	coro->slot->runops = RUNOPS_DEFAULT;	1706	coro->slot->runops = RUNOPS_DEFAULT;
1696	}	1707	}
		1708	}
		1709
		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
		1740	static void
		1741	slf_prepare_transfer (pTHX_ struct coro_transfer_args *ta)
		1742	{
		1743	SV **arg = (SV **)slf_frame.data;
		1744
		1745	prepare_transfer (aTHX_ ta, arg [0], arg [1]);
		1746	}
		1747
		1748	static void
		1749	slf_init_transfer (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		1750	{
		1751	if (items != 2)
		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;
1697	}	1922	}
1698		1923
1699	/*****************************************************************************/	1924	/*****************************************************************************/
1700	/* PerlIO::cede */	1925	/* PerlIO::cede */
1701		1926
…		…
1770	PerlIOBuf_get_cnt,	1995	PerlIOBuf_get_cnt,
1771	PerlIOBuf_set_ptrcnt,	1996	PerlIOBuf_set_ptrcnt,
1772	};	1997	};
1773		1998
1774	/*****************************************************************************/	1999	/*****************************************************************************/
		2000	/* Coro::Semaphore */
1775		2001
1776	static const CV *slf_cv; /* for quick consistency check */
1777
1778	static UNOP slf_restore; /* restore stack as entersub did, for first-re-run */
1779	static SV *slf_arg0;
1780	static SV *slf_arg1;
1781	static SV *slf_arg2;
1782
1783	/* this restores the stack in the case we patched the entersub, to */
1784	/* recreate the stack frame as perl will on following calls */
1785	/* since entersub cleared the stack */
1786	static OP *
1787	pp_restore (pTHX)
1788	{
1789	dSP;
1790
1791	PUSHMARK (SP);
1792
1793	EXTEND (SP, 3);
1794	if (slf_arg0) PUSHs (sv_2mortal (slf_arg0));
1795	if (slf_arg1) PUSHs (sv_2mortal (slf_arg1));
1796	if (slf_arg2) PUSHs (sv_2mortal (slf_arg2));
1797	PUSHs ((SV *)CvGV (slf_cv));
1798
1799	RETURNOP (slf_restore.op_first);
1800	}
1801
1802	static void	2002	static void
1803	slf_prepare_set_stacklevel (pTHX_ struct coro_transfer_args *ta)
1804	{
1805	prepare_set_stacklevel (ta, (struct coro_cctx *)slf_frame.data);
1806	}
1807
1808	static void
1809	slf_init_set_stacklevel (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
1810	{
1811	assert (("FATAL: set_stacklevel needs the coro cctx as sole argument", items == 1));
1812
1813	frame->prepare = slf_prepare_set_stacklevel;
1814	frame->check = slf_check_nop;
1815	frame->data = (void *)SvIV (arg [0]);
1816	}
1817
1818	static void
1819	slf_prepare_transfer (pTHX_ struct coro_transfer_args *ta)
1820	{
1821	SV **arg = (SV **)slf_frame.data;
1822
1823	prepare_transfer (aTHX_ ta, arg [0], arg [1]);
1824	}
1825
1826	static void
1827	slf_init_transfer (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
1828	{
1829	if (items != 2)
1830	croak ("Coro::State::transfer (prev, next) expects two arguments, not %d,", items);
1831
1832	frame->prepare = slf_prepare_transfer;
1833	frame->check = slf_check_nop;
1834	frame->data = (void *)arg; /* let's hope it will stay valid */
1835	}
1836
1837	static void
1838	slf_init_schedule (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
1839	{
1840	frame->prepare = prepare_schedule;
1841	frame->check = slf_check_nop;
1842	}
1843
1844	static void
1845	slf_init_cede (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
1846	{
1847	frame->prepare = prepare_cede;
1848	frame->check = slf_check_nop;
1849	}
1850
1851	static void
1852	slf_init_cede_notself (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
1853	{
1854	frame->prepare = prepare_cede_notself;
1855	frame->check = slf_check_nop;
1856	}
1857
1858	/* we hijack an hopefully unused CV flag for our purposes */
1859	#define CVf_SLF 0x4000
1860
1861	/*
1862	* these not obviously related functions are all rolled into one
1863	* function to increase chances that they all will call transfer with the same
1864	* stack offset
1865	* SLF stands for "schedule-like-function".
1866	*/
1867	static OP *
1868	pp_slf (pTHX)
1869	{
1870	I32 checkmark; /* mark SP to see how many elements check has pushed */
1871
1872	/* set up the slf frame, unless it has already been set-up */
1873	/* the latter happens when a new coro has been started */
1874	/* or when a new cctx was attached to an existing coroutine */
1875	if (expect_true (!slf_frame.prepare))
1876	{
1877	/* first iteration */
1878	dSP;
1879	SV **arg = PL_stack_base + TOPMARK + 1;
1880	int items = SP - arg; /* args without function object */
1881	SV *gv = *sp;
1882
1883	/* do a quick consistency check on the "function" object, and if it isn't */
1884	/* for us, divert to the real entersub */
1885	if (SvTYPE (gv) != SVt_PVGV || !(CvFLAGS (GvCV (gv)) & CVf_SLF))
1886	return PL_ppaddr[OP_ENTERSUB](aTHX);
1887
1888	/* pop args */
1889	SP = PL_stack_base + POPMARK;
1890
1891	if (!(PL_op->op_flags & OPf_STACKED))
1892	{
1893	/* ampersand-form of call, use @_ instead of stack */
1894	AV *av = GvAV (PL_defgv);
1895	arg = AvARRAY (av);
1896	items = AvFILLp (av) + 1;
1897	}
1898
1899	PUTBACK;
1900
1901	/* now call the init function, which needs to set up slf_frame */
1902	((coro_slf_cb)CvXSUBANY (GvCV (gv)).any_ptr)
1903	(aTHX_ &slf_frame, GvCV (gv), arg, items);
1904	}
1905
1906	/* now that we have a slf_frame, interpret it! */
1907	/* we use a callback system not to make the code needlessly */
1908	/* complicated, but so we can run multiple perl coros from one cctx */
1909
1910	do
1911	{
1912	struct coro_transfer_args ta;
1913
1914	slf_frame.prepare (aTHX_ &ta);
1915	TRANSFER (ta, 0);
1916
1917	checkmark = PL_stack_sp - PL_stack_base;
1918	}
1919	while (slf_frame.check (aTHX_ &slf_frame));
1920
1921	{
1922	dSP;
1923	SV **bot = PL_stack_base + checkmark;
1924	int gimme = GIMME_V;
1925
1926	slf_frame.prepare = 0; /* invalidate the frame, so it gets initialised again next time */
1927
1928	/* make sure we put something on the stack in scalar context */
1929	if (gimme == G_SCALAR)
1930	{
1931	if (sp == bot)
1932	XPUSHs (&PL_sv_undef);
1933
1934	SP = bot + 1;
1935	}
1936
1937	PUTBACK;
1938	}
1939
1940	return NORMAL;
1941	}
1942
1943	static void
1944	api_execute_slf (pTHX_ CV *cv, coro_slf_cb init_cb, SV **arg, int items)
1945	{
1946	assert (("FATAL: SLF call with illegal CV value", !CvANON (cv)));
1947
1948	if (PL_op->op_ppaddr != PL_ppaddr [OP_ENTERSUB]
1949	&& PL_op->op_ppaddr != pp_slf)
1950	croak ("FATAL: Coro SLF calls can only be made normally, not via goto or any other means, caught");
1951
1952	if (items > 3)
1953	croak ("Coro only supports up to three arguments to SLF functions currently (not %d), caught", items);
1954
1955	CvFLAGS (cv) |= CVf_SLF;
1956	CvXSUBANY (cv).any_ptr = (void *)init_cb;
1957	slf_cv = cv;
1958
1959	/* we patch the op, and then re-run the whole call */
1960	/* we have to put the same argument on the stack for this to work */
1961	/* and this will be done by pp_restore */
1962	slf_restore.op_next = (OP *)&slf_restore;
1963	slf_restore.op_type = OP_CUSTOM;
1964	slf_restore.op_ppaddr = pp_restore;
1965	slf_restore.op_first = PL_op;
1966
1967	slf_arg0 = items > 0 ? SvREFCNT_inc (arg [0]) : 0;
1968	slf_arg1 = items > 1 ? SvREFCNT_inc (arg [1]) : 0;
1969	slf_arg2 = items > 2 ? SvREFCNT_inc (arg [2]) : 0;
1970
1971	PL_op->op_ppaddr = pp_slf;
1972
1973	PL_op = (OP *)&slf_restore;
1974	}
1975
1976	/*****************************************************************************/
1977
1978	static void
1979	coro_semaphore_adjust (AV *av, int adjust)	2003	coro_semaphore_adjust (pTHX_ AV *av, IV adjust)
1980	{	2004	{
1981	SV *count_sv = AvARRAY (av)[0];	2005	SV *count_sv = AvARRAY (av)[0];
1982	IV count = SvIVX (count_sv);	2006	IV count = SvIVX (count_sv);
1983		2007
1984	count += adjust;	2008	count += adjust;
1985	SvIVX (count_sv) = count;	2009	SvIVX (count_sv) = count;
1986		2010
1987	/* now wake up as many waiters as possible */	2011	/* now wake up as many waiters as are expected to lock */
1988	while (count > 0 && AvFILLp (av) >= count)	2012	while (count > 0 && AvFILLp (av) > 0)
1989	{	2013	{
1990	SV *cb;	2014	SV *cb;
1991		2015
1992	/* swap first two elements so we can shift a waiter */	2016	/* swap first two elements so we can shift a waiter */
1993	AvARRAY (av)[0] = AvARRAY (av)[1];	2017	AvARRAY (av)[0] = AvARRAY (av)[1];
…		…
2006	}	2030	}
2007		2031
2008	static void	2032	static void
2009	coro_semaphore_on_destroy (pTHX_ struct coro *coro)	2033	coro_semaphore_on_destroy (pTHX_ struct coro *coro)
2010	{	2034	{
2011	/* call $sem->adjust (0) to possibly wake up some waiters */	2035	/* call $sem->adjust (0) to possibly wake up some other waiters */
2012	coro_semaphore_adjust ((AV *)coro->slf_frame.data, 0);	2036	coro_semaphore_adjust (aTHX_ (AV *)coro->slf_frame.data, 0);
2013	}	2037	}
2014		2038
2015	static int	2039	static int
2016	slf_check_semaphore_down (pTHX_ struct CoroSLF *frame)	2040	slf_check_semaphore_down (pTHX_ struct CoroSLF *frame)
2017	{	2041	{
2018	AV *av = (AV *)frame->data;	2042	AV *av = (AV *)frame->data;
2019	SV *count_sv = AvARRAY (av)[0];	2043	SV *count_sv = AvARRAY (av)[0];
2020		2044
		2045	/* if we are about to throw, don't actually acquire the lock, just throw */
		2046	if (CORO_THROW)
		2047	return 0;
2021	if (SvIVX (count_sv) > 0)	2048	else if (SvIVX (count_sv) > 0)
2022	{	2049	{
2023	SvSTATE (coro_current)->on_destroy = 0;	2050	SvSTATE_current->on_destroy = 0;
2024	SvIVX (count_sv) = SvIVX (count_sv) - 1;	2051	SvIVX (count_sv) = SvIVX (count_sv) - 1;
2025	return 0;	2052	return 0;
2026	}	2053	}
2027	else	2054	else
2028	{	2055	{
…		…
2057	frame->data = (void *)sv_2mortal (SvREFCNT_inc ((SV *)av));	2084	frame->data = (void *)sv_2mortal (SvREFCNT_inc ((SV *)av));
2058	frame->prepare = prepare_schedule;	2085	frame->prepare = prepare_schedule;
2059		2086
2060	/* to avoid race conditions when a woken-up coro gets terminated */	2087	/* to avoid race conditions when a woken-up coro gets terminated */
2061	/* we arrange for a temporary on_destroy that calls adjust (0) */	2088	/* we arrange for a temporary on_destroy that calls adjust (0) */
2062	SvSTATE (coro_current)->on_destroy = coro_semaphore_on_destroy;	2089	SvSTATE_current->on_destroy = coro_semaphore_on_destroy;
2063	}	2090	}
2064		2091
2065	frame->check = slf_check_semaphore_down;	2092	frame->check = slf_check_semaphore_down;
2066		2093
2067	}	2094	}
2068		2095
2069	/*****************************************************************************/	2096	/*****************************************************************************/
		2097	/* gensub: simple closure generation utility */
2070		2098
2071	#define GENSUB_ARG CvXSUBANY (cv).any_ptr	2099	#define GENSUB_ARG CvXSUBANY (cv).any_ptr
2072		2100
2073	/* create a closure from XS, returns a code reference */	2101	/* create a closure from XS, returns a code reference */
2074	/* the arg can be accessed via GENSUB_ARG from the callback */	2102	/* the arg can be accessed via GENSUB_ARG from the callback */
2075	/* the callback must use dXSARGS/XSRETURN */	2103	/* the callback must use dXSARGS/XSRETURN */
2076	static SV *	2104	static SV *
2077	gensub (pTHX_ void (*xsub)(pTHX_ CV *), void *arg)	2105	gensub (pTHX_ void (*xsub)(pTHX_ CV *), void *arg)
2078	{	2106	{
2079	CV *cv = (CV *)NEWSV (0, 0);	2107	CV *cv = (CV *)newSV (0);
2080		2108
2081	sv_upgrade ((SV *)cv, SVt_PVCV);	2109	sv_upgrade ((SV *)cv, SVt_PVCV);
2082		2110
2083	CvANON_on (cv);	2111	CvANON_on (cv);
2084	CvISXSUB_on (cv);	2112	CvISXSUB_on (cv);
…		…
2087		2115
2088	return newRV_noinc ((SV *)cv);	2116	return newRV_noinc ((SV *)cv);
2089	}	2117	}
2090		2118
2091	/*****************************************************************************/	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
		2171	static int
		2172	slf_check_aio_req (pTHX_ struct CoroSLF *frame)
		2173	{
		2174	AV *state = (AV *)frame->data;
		2175
		2176	/* if we are about to throw, return early */
		2177	/* this does not cancel the aio request, but at least */
		2178	/* it quickly returns */
		2179	if (CORO_THROW)
		2180	return 0;
		2181
		2182	/* one element that is an RV? repeat! */
		2183	if (AvFILLp (state) == 0 && SvROK (AvARRAY (state)[0]))
		2184	return 1;
		2185
		2186	/* restore status */
		2187	{
		2188	SV *data_sv = av_pop (state);
		2189	struct io_state *data = (struct io_state *)SvPVX (data_sv);
		2190
		2191	errno = data->errorno;
		2192	PL_laststype = data->laststype;
		2193	PL_laststatval = data->laststatval;
		2194	PL_statcache = data->statcache;
		2195
		2196	SvREFCNT_dec (data_sv);
		2197	}
		2198
		2199	/* push result values */
		2200	{
		2201	dSP;
		2202	int i;
		2203
		2204	EXTEND (SP, AvFILLp (state) + 1);
		2205	for (i = 0; i <= AvFILLp (state); ++i)
		2206	PUSHs (sv_2mortal (SvREFCNT_inc_NN (AvARRAY (state)[i])));
		2207
		2208	PUTBACK;
		2209	}
		2210
		2211	return 0;
		2212	}
		2213
		2214	static void
		2215	slf_init_aio_req (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		2216	{
		2217	AV *state = (AV *)sv_2mortal ((SV *)newAV ());
		2218	SV *coro_hv = SvRV (coro_current);
		2219	struct coro *coro = SvSTATE_hv (coro_hv);
		2220
		2221	/* put our coroutine id on the state arg */
		2222	av_push (state, SvREFCNT_inc_NN (coro_hv));
		2223
		2224	/* first see whether we have a non-zero priority and set it as AIO prio */
		2225	if (coro->prio)
		2226	{
		2227	dSP;
		2228
		2229	static SV *prio_cv;
		2230	static SV *prio_sv;
		2231
		2232	if (expect_false (!prio_cv))
		2233	{
		2234	prio_cv = (SV *)get_cv ("IO::AIO::aioreq_pri", 0);
		2235	prio_sv = newSViv (0);
		2236	}
		2237
		2238	PUSHMARK (SP);
		2239	sv_setiv (prio_sv, coro->prio);
		2240	XPUSHs (prio_sv);
		2241
		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;
		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	/*****************************************************************************/
2092		2285
2093	MODULE = Coro::State PACKAGE = Coro::State PREFIX = api_	2286	MODULE = Coro::State PACKAGE = Coro::State PREFIX = api_
2094		2287
2095	PROTOTYPES: DISABLE	2288	PROTOTYPES: DISABLE
2096		2289
2097	BOOT:	2290	BOOT:
2098	{	2291	{
2099	#ifdef USE_ITHREADS	2292	#ifdef USE_ITHREADS
2100	MUTEX_INIT (&coro_lock);
2101	# if CORO_PTHREAD	2293	# if CORO_PTHREAD
2102	coro_thx = PERL_GET_CONTEXT;	2294	coro_thx = PERL_GET_CONTEXT;
2103	# endif	2295	# endif
2104	#endif	2296	#endif
2105	BOOT_PAGESIZE;	2297	BOOT_PAGESIZE;
…		…
2192	}	2384	}
2193	OUTPUT:	2385	OUTPUT:
2194	RETVAL	2386	RETVAL
2195		2387
2196	void	2388	void
2197	_set_stacklevel (...)
2198	CODE:
2199	api_execute_slf (aTHX_ cv, slf_init_set_stacklevel, &ST (0), items);
2200
2201	void
2202	transfer (...)	2389	transfer (...)
2203	PROTOTYPE: $$	2390	PROTOTYPE: $$
2204	CODE:	2391	CODE:
2205	api_execute_slf (aTHX_ cv, slf_init_transfer, &ST (0), items);	2392	CORO_EXECUTE_SLF_XS (slf_init_transfer);
2206		2393
2207	bool	2394	bool
2208	_destroy (SV *coro_sv)	2395	_destroy (SV *coro_sv)
2209	CODE:	2396	CODE:
2210	RETVAL = coro_state_destroy (aTHX_ SvSTATE (coro_sv));	2397	RETVAL = coro_state_destroy (aTHX_ SvSTATE (coro_sv));
…		…
2328		2515
2329	void	2516	void
2330	throw (Coro::State self, SV *throw = &PL_sv_undef)	2517	throw (Coro::State self, SV *throw = &PL_sv_undef)
2331	PROTOTYPE: $;$	2518	PROTOTYPE: $;$
2332	CODE:	2519	CODE:
		2520	{
		2521	struct coro *current = SvSTATE_current;
		2522	SV **throwp = self == current ? &CORO_THROW : &self->except;
2333	SvREFCNT_dec (self->throw);	2523	SvREFCNT_dec (*throwp);
2334	self->throw = SvOK (throw) ? newSVsv (throw) : 0;	2524	*throwp = SvOK (throw) ? newSVsv (throw) : 0;
		2525	}
2335		2526
2336	void	2527	void
2337	api_trace (SV *coro, int flags = CC_TRACE | CC_TRACE_SUB)	2528	api_trace (SV *coro, int flags = CC_TRACE | CC_TRACE_SUB)
2338	PROTOTYPE: $;$	2529	PROTOTYPE: $;$
2339	C_ARGS: aTHX_ coro, flags	2530	C_ARGS: aTHX_ coro, flags
…		…
2370		2561
2371	void	2562	void
2372	force_cctx ()	2563	force_cctx ()
2373	PROTOTYPE:	2564	PROTOTYPE:
2374	CODE:	2565	CODE:
2375	struct coro *coro = SvSTATE (coro_current);
2376	coro->cctx->idle_sp = 0;	2566	SvSTATE_current->cctx->idle_sp = 0;
2377		2567
2378	void	2568	void
2379	swap_defsv (Coro::State self)	2569	swap_defsv (Coro::State self)
2380	PROTOTYPE: $	2570	PROTOTYPE: $
2381	ALIAS:	2571	ALIAS:
…		…
2389	SV **dst = ix ? (SV **)&self->slot->defav : (SV **)&self->slot->defsv;	2579	SV **dst = ix ? (SV **)&self->slot->defav : (SV **)&self->slot->defsv;
2390		2580
2391	SV *tmp = *src; *src = *dst; *dst = tmp;	2581	SV *tmp = *src; *src = *dst; *dst = tmp;
2392	}	2582	}
2393		2583
		2584
2394	MODULE = Coro::State PACKAGE = Coro	2585	MODULE = Coro::State PACKAGE = Coro
2395		2586
2396	BOOT:	2587	BOOT:
2397	{	2588	{
2398	int i;	2589	int i;
…		…
2425	coroapi.ready = api_ready;	2616	coroapi.ready = api_ready;
2426	coroapi.is_ready = api_is_ready;	2617	coroapi.is_ready = api_is_ready;
2427	coroapi.nready = coro_nready;	2618	coroapi.nready = coro_nready;
2428	coroapi.current = coro_current;	2619	coroapi.current = coro_current;
2429		2620
2430	GCoroAPI = &coroapi;	2621	/*GCoroAPI = &coroapi;*/
2431	sv_setiv (sv, (IV)&coroapi);	2622	sv_setiv (sv, (IV)&coroapi);
2432	SvREADONLY_on (sv);	2623	SvREADONLY_on (sv);
2433	}	2624	}
2434	}	2625	}
2435		2626
2436	void	2627	void
2437	schedule (...)	2628	schedule (...)
2438	CODE:	2629	CODE:
2439	api_execute_slf (aTHX_ cv, slf_init_schedule, &ST (0), 0);	2630	CORO_EXECUTE_SLF_XS (slf_init_schedule);
2440		2631
2441	void	2632	void
2442	cede (...)	2633	cede (...)
2443	CODE:	2634	CODE:
2444	api_execute_slf (aTHX_ cv, slf_init_cede, &ST (0), 0);	2635	CORO_EXECUTE_SLF_XS (slf_init_cede);
2445		2636
2446	void	2637	void
2447	cede_notself (...)	2638	cede_notself (...)
2448	CODE:	2639	CODE:
2449	api_execute_slf (aTHX_ cv, slf_init_cede_notself, &ST (0), 0);	2640	CORO_EXECUTE_SLF_XS (slf_init_cede_notself);
2450		2641
2451	void	2642	void
2452	_set_current (SV *current)	2643	_set_current (SV *current)
2453	PROTOTYPE: $	2644	PROTOTYPE: $
2454	CODE:	2645	CODE:
…		…
2504	# for async_pool speedup	2695	# for async_pool speedup
2505	void	2696	void
2506	_pool_1 (SV *cb)	2697	_pool_1 (SV *cb)
2507	CODE:	2698	CODE:
2508	{	2699	{
2509	struct coro *coro = SvSTATE (coro_current);
2510	HV *hv = (HV *)SvRV (coro_current);	2700	HV *hv = (HV *)SvRV (coro_current);
		2701	struct coro *coro = SvSTATE_hv ((SV *)hv);
2511	AV *defav = GvAV (PL_defgv);	2702	AV *defav = GvAV (PL_defgv);
2512	SV *invoke = hv_delete (hv, "_invoke", sizeof ("_invoke") - 1, 0);	2703	SV *invoke = hv_delete (hv, "_invoke", sizeof ("_invoke") - 1, 0);
2513	AV *invoke_av;	2704	AV *invoke_av;
2514	int i, len;	2705	int i, len;
2515		2706
…		…
2542		2733
2543	void	2734	void
2544	_pool_2 (SV *cb)	2735	_pool_2 (SV *cb)
2545	CODE:	2736	CODE:
2546	{	2737	{
2547	struct coro *coro = SvSTATE (coro_current);	2738	HV *hv = (HV *)SvRV (coro_current);
		2739	struct coro *coro = SvSTATE_hv ((SV *)hv);
2548		2740
2549	sv_setsv (cb, &PL_sv_undef);	2741	sv_setsv (cb, &PL_sv_undef);
2550		2742
2551	SvREFCNT_dec ((SV *)PL_defoutgv); PL_defoutgv = (GV *)coro->saved_deffh;	2743	SvREFCNT_dec ((SV *)PL_defoutgv); PL_defoutgv = (GV *)coro->saved_deffh;
2552	coro->saved_deffh = 0;	2744	coro->saved_deffh = 0;
…		…
2559	SvREFCNT_dec (old);	2751	SvREFCNT_dec (old);
2560	croak ("\3async_pool terminate\2\n");	2752	croak ("\3async_pool terminate\2\n");
2561	}	2753	}
2562		2754
2563	av_clear (GvAV (PL_defgv));	2755	av_clear (GvAV (PL_defgv));
2564	hv_store ((HV *)SvRV (coro_current), "desc", sizeof ("desc") - 1,	2756	hv_store (hv, "desc", sizeof ("desc") - 1,
2565	newSVpvn ("[async_pool idle]", sizeof ("[async_pool idle]") - 1), 0);	2757	newSVpvn ("[async_pool idle]", sizeof ("[async_pool idle]") - 1), 0);
2566		2758
2567	coro->prio = 0;	2759	coro->prio = 0;
2568		2760
2569	if (coro->cctx && (coro->cctx->flags & CC_TRACE))	2761	if (coro->cctx && (coro->cctx->flags & CC_TRACE))
…		…
2571		2763
2572	av_push (av_async_pool, newSVsv (coro_current));	2764	av_push (av_async_pool, newSVsv (coro_current));
2573	}	2765	}
2574		2766
2575		2767
2576	MODULE = Coro::State PACKAGE = Coro::AIO
2577
2578	void
2579	_get_state (SV *self)
2580	PROTOTYPE: $
2581	PPCODE:
2582	{
2583	AV *defav = GvAV (PL_defgv);
2584	AV *av = newAV ();
2585	int i;
2586	SV *data_sv = newSV (sizeof (struct io_state));
2587	struct io_state *data = (struct io_state *)SvPVX (data_sv);
2588	SvCUR_set (data_sv, sizeof (struct io_state));
2589	SvPOK_only (data_sv);
2590
2591	data->errorno = errno;
2592	data->laststype = PL_laststype;
2593	data->laststatval = PL_laststatval;
2594	data->statcache = PL_statcache;
2595
2596	av_extend (av, AvFILLp (defav) + 1 + 1);
2597
2598	for (i = 0; i <= AvFILLp (defav); ++i)
2599	av_push (av, SvREFCNT_inc_NN (AvARRAY (defav)[i]));
2600
2601	av_push (av, data_sv);
2602
2603	XPUSHs (sv_2mortal (newRV_noinc ((SV *)av)));
2604
2605	api_ready (aTHX_ self);
2606	}
2607
2608	void
2609	_set_state (SV *state)
2610	PROTOTYPE: $
2611	PPCODE:
2612	{
2613	AV *av = (AV *)SvRV (state);
2614	struct io_state *data = (struct io_state *)SvPVX (AvARRAY (av)[AvFILLp (av)]);
2615	int i;
2616
2617	errno = data->errorno;
2618	PL_laststype = data->laststype;
2619	PL_laststatval = data->laststatval;
2620	PL_statcache = data->statcache;
2621
2622	EXTEND (SP, AvFILLp (av));
2623	for (i = 0; i < AvFILLp (av); ++i)
2624	PUSHs (sv_2mortal (SvREFCNT_inc_NN (AvARRAY (av)[i])));
2625	}
2626
2627
2628	MODULE = Coro::State PACKAGE = Coro::AnyEvent
2629
2630	BOOT:
2631	sv_activity = coro_get_sv (aTHX_ "Coro::AnyEvent::ACTIVITY", TRUE);
2632
2633	void
2634	_schedule (...)
2635	CODE:
2636	{
2637	static int incede;
2638
2639	api_cede_notself (aTHX);
2640
2641	++incede;
2642	while (coro_nready >= incede && api_cede (aTHX))
2643	;
2644
2645	sv_setsv (sv_activity, &PL_sv_undef);
2646	if (coro_nready >= incede)
2647	{
2648	PUSHMARK (SP);
2649	PUTBACK;
2650	call_pv ("Coro::AnyEvent::_activity", G_DISCARD | G_EVAL);
2651	SPAGAIN;
2652	}
2653
2654	--incede;
2655	}
2656
2657
2658	MODULE = Coro::State PACKAGE = PerlIO::cede	2768	MODULE = Coro::State PACKAGE = PerlIO::cede
2659		2769
2660	BOOT:	2770	BOOT:
2661	PerlIO_define_layer (aTHX_ &PerlIO_cede);	2771	PerlIO_define_layer (aTHX_ &PerlIO_cede);
		2772
2662		2773
2663	MODULE = Coro::State PACKAGE = Coro::Semaphore	2774	MODULE = Coro::State PACKAGE = Coro::Semaphore
2664		2775
2665	SV *	2776	SV *
2666	new (SV *klass, SV *count_ = 0)	2777	new (SV *klass, SV *count_ = 0)
2667	CODE:	2778	CODE:
2668	{	2779	{
2669	/* a semaphore contains a counter IV in $sem->[0] and any waiters after that */	2780	/* a semaphore contains a counter IV in $sem->[0] and any waiters after that */
2670	AV *av = newAV ();	2781	AV *av = newAV ();
		2782	SV **ary;
		2783
		2784	/* unfortunately, building manually saves memory */
		2785	Newx (ary, 2, SV *);
		2786	AvALLOC (av) = ary;
		2787	AvARRAY (av) = ary;
		2788	AvMAX (av) = 1;
		2789	AvFILLp (av) = 0;
2671	av_push (av, newSViv (count_ && SvOK (count_) ? SvIV (count_) : 1));	2790	ary [0] = newSViv (count_ && SvOK (count_) ? SvIV (count_) : 1);
		2791
2672	RETVAL = sv_bless (newRV_noinc ((SV *)av), GvSTASH (CvGV (cv)));	2792	RETVAL = sv_bless (newRV_noinc ((SV *)av), GvSTASH (CvGV (cv)));
2673	}	2793	}
2674	OUTPUT:	2794	OUTPUT:
2675	RETVAL	2795	RETVAL
2676		2796
…		…
2684	void	2804	void
2685	up (SV *self, int adjust = 1)	2805	up (SV *self, int adjust = 1)
2686	ALIAS:	2806	ALIAS:
2687	adjust = 1	2807	adjust = 1
2688	CODE:	2808	CODE:
2689	coro_semaphore_adjust ((AV *)SvRV (self), ix ? adjust : 1);	2809	coro_semaphore_adjust (aTHX_ (AV *)SvRV (self), ix ? adjust : 1);
2690		2810
2691	void	2811	void
2692	down (SV *self)	2812	down (SV *self)
2693	CODE:	2813	CODE:
2694	api_execute_slf (aTHX_ cv, slf_init_semaphore_down, &ST (0), 1);	2814	CORO_EXECUTE_SLF_XS (slf_init_semaphore_down);
2695		2815
2696	void	2816	void
2697	try (SV *self)	2817	try (SV *self)
2698	PPCODE:	2818	PPCODE:
2699	{	2819	{
…		…
2722	else	2842	else
2723	{	2843	{
2724	int i;	2844	int i;
2725	EXTEND (SP, AvFILLp (av) + 1 - 1);	2845	EXTEND (SP, AvFILLp (av) + 1 - 1);
2726	for (i = 1; i <= AvFILLp (av); ++i)	2846	for (i = 1; i <= AvFILLp (av); ++i)
2727	PUSHs (newSVsv (AvARRAY (av)[i]));	2847	PUSHs (sv_2mortal (newRV_inc (AvARRAY (av)[i])));
2728	}	2848	}
2729	}	2849	}
2730		2850
		2851
		2852	MODULE = Coro::State PACKAGE = Coro::AnyEvent
		2853
		2854	BOOT:
		2855	sv_activity = coro_get_sv (aTHX_ "Coro::AnyEvent::ACTIVITY", TRUE);
		2856
		2857	void
		2858	_schedule (...)
		2859	CODE:
		2860	{
		2861	static int incede;
		2862
		2863	api_cede_notself (aTHX);
		2864
		2865	++incede;
		2866	while (coro_nready >= incede && api_cede (aTHX))
		2867	;
		2868
		2869	sv_setsv (sv_activity, &PL_sv_undef);
		2870	if (coro_nready >= incede)
		2871	{
		2872	PUSHMARK (SP);
		2873	PUTBACK;
		2874	call_pv ("Coro::AnyEvent::_activity", G_KEEPERR | G_EVAL | G_VOID | G_DISCARD);
		2875	}
		2876
		2877	--incede;
		2878	}
		2879
		2880
		2881	MODULE = Coro::State PACKAGE = Coro::AIO
		2882
		2883	void
		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	}
		2895

Diff Legend

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