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

Comparing Coro/Coro/State.xs (file contents):
Revision 1.270 by root, Fri Nov 14 07:22:11 2008 UTC vs.
Revision 1.283 by root, Sun Nov 16 11:12:57 2008 UTC

…		…
142	#define NOINLINE attribute ((noinline))	142	#define NOINLINE attribute ((noinline))
143		143
144	#include "CoroAPI.h"	144	#include "CoroAPI.h"
145		145
146	#ifdef USE_ITHREADS	146	#ifdef USE_ITHREADS
147
148	static perl_mutex coro_lock;
149	# define LOCK do { MUTEX_LOCK (&coro_lock); } while (0)
150	# define UNLOCK do { MUTEX_UNLOCK (&coro_lock); } while (0)
151	# if CORO_PTHREAD	147	# if CORO_PTHREAD
152	static void *coro_thx;	148	static void *coro_thx;
153	# endif	149	# endif
154
155	#else
156
157	# define LOCK (void)0
158	# define UNLOCK (void)0
159
160	#endif	150	#endif
161
162	# undef LOCK
163	# define LOCK (void)0
164	# undef UNLOCK
165	# define UNLOCK (void)0
166		151
167	/* helper storage struct for Coro::AIO */	152	/* helper storage struct for Coro::AIO */
168	struct io_state	153	struct io_state
169	{	154	{
170	AV *res;	155	AV *res;
…		…
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
188	struct transfer_args
189	{
190	struct coro prev, next;
191	};
192		171
193	static GV irsgv; / $/ */	172	static GV irsgv; / $/ */
194	static GV stdoutgv; / STDOUT /	173	static GV stdoutgv; / STDOUT /
195	static SV *rv_diehook;	174	static SV *rv_diehook;
196	static SV *rv_warnhook;	175	static SV *rv_warnhook;
…		…
215	CC_TRACE_LINE = 0x10, /* trace each statement */	194	CC_TRACE_LINE = 0x10, /* trace each statement */
216	CC_TRACE_ALL = CC_TRACE_SUB \| CC_TRACE_LINE,	195	CC_TRACE_ALL = CC_TRACE_SUB \| CC_TRACE_LINE,
217	};	196	};
218		197
219	/* this is a structure representing a c-level coroutine */	198	/* this is a structure representing a c-level coroutine */
220	typedef struct coro_cctx {	199	typedef struct coro_cctx
		200	{
221	struct coro_cctx *next;	201	struct coro_cctx *next;
222		202
223	/* the stack */	203	/* the stack */
224	void *sptr;	204	void *sptr;
225	size_t ssize;	205	size_t ssize;
…		…
243	CF_NEW = 0x0004, /* has never been switched to */	223	CF_NEW = 0x0004, /* has never been switched to */
244	CF_DESTROYED = 0x0008, /* coroutine data has been freed */	224	CF_DESTROYED = 0x0008, /* coroutine data has been freed */
245	};	225	};
246		226
247	/* the structure where most of the perl state is stored, overlaid on the cxstack */	227	/* the structure where most of the perl state is stored, overlaid on the cxstack */
248	typedef struct {	228	typedef struct
		229	{
249	SV *defsv;	230	SV *defsv;
250	AV *defav;	231	AV *defav;
251	SV *errsv;	232	SV *errsv;
252	SV *irsgv;	233	SV *irsgv;
253	#define VAR(name,type) type name;	234	#define VAR(name,type) type name;
…		…
257		238
258	#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))
259		240
260	/* this is a structure representing a perl-level coroutine */	241	/* this is a structure representing a perl-level coroutine */
261	struct coro {	242	struct coro {
262	/* the c coroutine allocated to this perl coroutine, if any */	243	/* the C coroutine allocated to this perl coroutine, if any */
263	coro_cctx *cctx;	244	coro_cctx *cctx;
264		245
265	/* process data */	246	/* process data */
		247	struct CoroSLF slf_frame; /* saved slf frame */
266	AV *mainstack;	248	AV *mainstack;
267	perl_slots slot; / basically the saved sp */	249	perl_slots slot; / basically the saved sp */
268		250
269	AV args; / data associated with this coroutine (initial args) */	251	AV args; / data associated with this coroutine (initial args) */
270	int refcnt; /* coroutines are refcounted, yes */	252	int refcnt; /* coroutines are refcounted, yes */
271	int flags; /* CF_ flags */	253	int flags; /* CF_ flags */
272	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);
273		256
274	/* statistics */	257	/* statistics */
275	int usecount; /* number of transfers to this coro */	258	int usecount; /* number of transfers to this coro */
276		259
277	/* coro process data */	260	/* coro process data */
…		…
285	struct coro next, prev;	268	struct coro next, prev;
286	};	269	};
287		270
288	typedef struct coro *Coro__State;	271	typedef struct coro *Coro__State;
289	typedef struct coro *Coro__State_or_hashref;	272	typedef struct coro *Coro__State_or_hashref;
		273
		274	static struct CoroSLF slf_frame; /* the current slf frame */
290		275
291	/ Coro ******************************************************************/	276	/ Coro ******************************************************************/
292		277
293	#define PRIO_MAX 3	278	#define PRIO_MAX 3
294	#define PRIO_HIGH 1	279	#define PRIO_HIGH 1
…		…
403	static MGVTBL coro_cv_vtbl = {	388	static MGVTBL coro_cv_vtbl = {
404	0, 0, 0, 0,	389	0, 0, 0, 0,
405	coro_cv_free	390	coro_cv_free
406	};	391	};
407		392
408	#define CORO_MAGIC(sv, type) \	393	#define CORO_MAGIC(sv, type) \
409	SvMAGIC (sv) \	394	expect_true (SvMAGIC (sv)) \
410	? SvMAGIC (sv)->mg_type == type \	395	? expect_true (SvMAGIC (sv)->mg_type == type) \
411	? SvMAGIC (sv) \	396	? SvMAGIC (sv) \
412	: mg_find (sv, type) \	397	: mg_find (sv, type) \
413	: 0	398	: 0
414		399
415	#define CORO_MAGIC_cv(cv) CORO_MAGIC (((SV *)(cv)), CORO_MAGIC_type_cv)	400	#define CORO_MAGIC_cv(cv) CORO_MAGIC (((SV *)(cv)), CORO_MAGIC_type_cv)
416	#define CORO_MAGIC_state(sv) CORO_MAGIC (((SV *)(sv)), CORO_MAGIC_type_state)	401	#define CORO_MAGIC_state(sv) CORO_MAGIC (((SV *)(sv)), CORO_MAGIC_type_state)
417		402
…		…
438	mg = CORO_MAGIC_state (coro);	423	mg = CORO_MAGIC_state (coro);
439	return (struct coro *)mg->mg_ptr;	424	return (struct coro *)mg->mg_ptr;
440	}	425	}
441		426
442	#define SvSTATE(sv) SvSTATE_ (aTHX_ (sv))	427	#define SvSTATE(sv) SvSTATE_ (aTHX_ (sv))
		428
		429	/* fastert than SvSTATE, but expects a coroutine hv */
		430	INLINE struct coro *
		431	SvSTATE_hv (SV *sv)
		432	{
		433	MAGIC *mg = expect_true (SvMAGIC (sv)->mg_type == CORO_MAGIC_type_state)
		434	? SvMAGIC (sv)
		435	: mg_find (sv, CORO_MAGIC_type_state);
		436
		437	return (struct coro *)mg->mg_ptr;
		438	}
		439
		440	#define SvSTATE_current SvSTATE_hv (SvRV (coro_current))
443		441
444	/* the next two functions merely cache the padlists */	442	/* the next two functions merely cache the padlists */
445	static void	443	static void
446	get_padlist (pTHX_ CV *cv)	444	get_padlist (pTHX_ CV *cv)
447	{	445	{
…		…
514	CvPADLIST (cv) = (AV *)POPs;	512	CvPADLIST (cv) = (AV *)POPs;
515	}	513	}
516		514
517	PUTBACK;	515	PUTBACK;
518	}	516	}
		517
		518	slf_frame = c->slf_frame;
519	}	519	}
520		520
521	static void	521	static void
522	save_perl (pTHX_ Coro__State c)	522	save_perl (pTHX_ Coro__State c)
523	{	523	{
		524	c->slf_frame = slf_frame;
		525
524	{	526	{
525	dSP;	527	dSP;
526	I32 cxix = cxstack_ix;	528	I32 cxix = cxstack_ix;
527	PERL_CONTEXT *ccstk = cxstack;	529	PERL_CONTEXT *ccstk = cxstack;
528	PERL_SI *top_si = PL_curstackinfo;	530	PERL_SI *top_si = PL_curstackinfo;
…		…
595	#undef VAR	597	#undef VAR
596	}	598	}
597	}	599	}
598		600
599	/*	601	/*
600	* allocate various perl stacks. This is an exact copy	602	* allocate various perl stacks. This is almost an exact copy
601	* of perl.c:init_stacks, except that it uses less memory	603	* of perl.c:init_stacks, except that it uses less memory
602	* on the (sometimes correct) assumption that coroutines do	604	* on the (sometimes correct) assumption that coroutines do
603	* not usually need a lot of stackspace.	605	* not usually need a lot of stackspace.
604	*/	606	*/
605	#if CORO_PREFER_PERL_FUNCTIONS	607	#if CORO_PREFER_PERL_FUNCTIONS
…		…
712	#endif	714	#endif
713	}	715	}
714	}	716	}
715		717
716	return rss;	718	return rss;
717	}
718
719	/ set stacklevel support ************************************************/
720
721	/* we sometimes need to create the effect of pp_slf calling us */
722	#define SLF_HEAD (void)0
723	/* we sometimes need to create the effect of leaving via pp_slf */
724	#define SLF_TAIL slf_tail (aTHX)
725
726	INLINE void
727	slf_tail (pTHX)
728	{
729	dSP;
730	SV **bot = SP;
731
732	int gimme = GIMME_V;
733
734	/* make sure we put something on the stack in scalar context */
735	if (gimme == G_SCALAR)
736	{
737	if (sp == bot)
738	XPUSHs (&PL_sv_undef);
739
740	SP = bot + 1;
741	}
742
743	PUTBACK;
744	}	719	}
745		720
746	/ coroutine stack handling **********************************************/	721	/ coroutine stack handling **********************************************/
747		722
748	static int (orig_sigelem_get) (pTHX_ SV sv, MAGIC *mg);	723	static int (orig_sigelem_get) (pTHX_ SV sv, MAGIC *mg);
…		…
834		809
835	return orig_sigelem_set ? orig_sigelem_set (aTHX_ sv, mg) : 0;	810	return orig_sigelem_set ? orig_sigelem_set (aTHX_ sv, mg) : 0;
836	}	811	}
837		812
838	static void	813	static void
		814	prepare_nop (pTHX_ struct coro_transfer_args *ta)
		815	{
		816	/* kind of mega-hacky, but works */
		817	ta->next = ta->prev = (struct coro *)ta;
		818	}
		819
		820	static int
		821	slf_check_nop (pTHX_ struct CoroSLF *frame)
		822	{
		823	return 0;
		824	}
		825
		826	static void
839	coro_setup (pTHX_ struct coro *coro)	827	coro_setup (pTHX_ struct coro *coro)
840	{	828	{
841	/*	829	/*
842	* emulate part of the perl startup here.	830	* emulate part of the perl startup here.
843	*/	831	*/
…		…
882	PL_op = PL_ppaddr[OP_ENTERSUB](aTHX);	870	PL_op = PL_ppaddr[OP_ENTERSUB](aTHX);
883	SPAGAIN;	871	SPAGAIN;
884	}	872	}
885		873
886	/* this newly created coroutine might be run on an existing cctx which most	874	/* this newly created coroutine might be run on an existing cctx which most
887	* likely was suspended in set_stacklevel, called from pp_set_stacklevel,	875	* likely was suspended in pp_slf, so we have to emulate entering pp_slf here.
888	* so we have to emulate entering pp_set_stacklevel here.
889	*/	876	*/
890	SLF_HEAD;	877	slf_frame.prepare = prepare_nop; /* provide a nop function for an eventual pp_slf */
		878	slf_frame.check = slf_check_nop; /* signal pp_slf to not repeat */
891	}	879	}
892		880
893	static void	881	static void
894	coro_destruct (pTHX_ struct coro *coro)	882	coro_destruct (pTHX_ struct coro *coro)
895	{	883	{
…		…
939	static int	927	static int
940	runops_trace (pTHX)	928	runops_trace (pTHX)
941	{	929	{
942	COP *oldcop = 0;	930	COP *oldcop = 0;
943	int oldcxix = -2;	931	int oldcxix = -2;
944	struct coro coro = SvSTATE (coro_current); / trace cctx is tied to specific coro */	932	struct coro coro = SvSTATE_current; / trace cctx is tied to specific coro */
945	coro_cctx *cctx = coro->cctx;	933	coro_cctx *cctx = coro->cctx;
946		934
947	while ((PL_op = CALL_FPTR (PL_op->op_ppaddr) (aTHX)))	935	while ((PL_op = CALL_FPTR (PL_op->op_ppaddr) (aTHX)))
948	{	936	{
949	PERL_ASYNC_CHECK ();	937	PERL_ASYNC_CHECK ();
…		…
1059	TAINT_NOT;	1047	TAINT_NOT;
1060	return 0;	1048	return 0;
1061	}	1049	}
1062		1050
1063	static void	1051	static void
1064	prepare_set_stacklevel (struct transfer_args ta, struct coro_cctx cctx)	1052	prepare_set_stacklevel (struct coro_transfer_args ta, struct coro_cctx cctx)
1065	{	1053	{
1066	ta->prev = (struct coro *)cctx;	1054	ta->prev = (struct coro *)cctx;
1067	ta->next = 0;	1055	ta->next = 0;
1068	}	1056	}
1069		1057
…		…
1098		1086
1099	/* the tail of transfer: execute stuff we can only do after a transfer */	1087	/* the tail of transfer: execute stuff we can only do after a transfer */
1100	INLINE void	1088	INLINE void
1101	transfer_tail (pTHX)	1089	transfer_tail (pTHX)
1102	{	1090	{
1103	struct coro next = (struct coro )transfer_next;
1104	assert (!(transfer_next = 0)); /* just used for the side effect when asserts are enabled */
1105	assert (("FATAL: next coroutine was zero in transfer_tail (please report)", next));
1106
1107	free_coro_mortal (aTHX);	1091	free_coro_mortal (aTHX);
1108	UNLOCK;
1109
1110	if (expect_false (next->throw))
1111	{
1112	SV *exception = sv_2mortal (next->throw);
1113
1114	next->throw = 0;
1115	sv_setsv (ERRSV, exception);
1116	croak (0);
1117	}
1118	}	1092	}
1119		1093
1120	/*	1094	/*
1121	* this is a _very_ stripped down perl interpreter ;)	1095	* this is a _very_ stripped down perl interpreter ;)
1122	*/	1096	*/
…		…
1129	# endif	1103	# endif
1130	#endif	1104	#endif
1131	{	1105	{
1132	dTHX;	1106	dTHX;
1133		1107
1134	/* we are the alternative tail to pp_set_stacklevel */	1108	/* normally we would need to skip the entersub here */
1135	/* so do the same things here */	1109	/* not doing so will re-execute it, which is exactly what we want */
1136	SLF_TAIL;
1137
1138	/* we now skip the op that did lead to transfer() */
1139	PL_op = PL_op->op_next;	1110	/* PL_nop = PL_nop->op_next */
1140		1111
1141	/* inject a fake subroutine call to cctx_init */	1112	/* inject a fake subroutine call to cctx_init */
1142	cctx_prepare (aTHX_ (coro_cctx *)arg);	1113	cctx_prepare (aTHX_ (coro_cctx *)arg);
1143		1114
1144	/* cctx_run is the alternative tail of transfer() */	1115	/* cctx_run is the alternative tail of transfer() */
…		…
1308	transfer_check (pTHX_ struct coro prev, struct coro next)	1279	transfer_check (pTHX_ struct coro prev, struct coro next)
1309	{	1280	{
1310	if (expect_true (prev != next))	1281	if (expect_true (prev != next))
1311	{	1282	{
1312	if (expect_false (!(prev->flags & (CF_RUNNING \| CF_NEW))))	1283	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");	1284	croak ("Coro::State::transfer called with non-running/new prev Coro::State, but can only transfer from running or new states,");
1314		1285
1315	if (expect_false (next->flags & CF_RUNNING))	1286	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");	1287	croak ("Coro::State::transfer called with running next Coro::State, but can only transfer to inactive states,");
1317		1288
1318	if (expect_false (next->flags & CF_DESTROYED))	1289	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");	1290	croak ("Coro::State::transfer called with destroyed next Coro::State, but can only transfer to inactive states,");
1320		1291
1321	#if !PERL_VERSION_ATLEAST (5,10,0)	1292	#if !PERL_VERSION_ATLEAST (5,10,0)
1322	if (expect_false (PL_lex_state != LEX_NOTPARSING))	1293	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");	1294	croak ("Coro::State::transfer called while parsing, but this is not supported in your perl version,");
1324	#endif	1295	#endif
1325	}	1296	}
1326	}	1297	}
1327		1298
1328	/* always use the TRANSFER macro */	1299	/* always use the TRANSFER macro */
…		…
1332	dSTACKLEVEL;	1303	dSTACKLEVEL;
1333		1304
1334	/* sometimes transfer is only called to set idle_sp */	1305	/* sometimes transfer is only called to set idle_sp */
1335	if (expect_false (!next))	1306	if (expect_false (!next))
1336	{	1307	{
1337	((coro_cctx *)prev)->idle_sp = stacklevel;	1308	((coro_cctx )prev)->idle_sp = (void )stacklevel;
1338	assert (((coro_cctx )prev)->idle_te = PL_top_env); / just for the side-effect when asserts are enabled */	1309	assert (((coro_cctx )prev)->idle_te = PL_top_env); / just for the side-effect when asserts are enabled */
1339	}	1310	}
1340	else if (expect_true (prev != next))	1311	else if (expect_true (prev != next))
1341	{	1312	{
1342	coro_cctx *prev__cctx;	1313	coro_cctx *prev__cctx;
…		…
1349	prev->flags \|= CF_RUNNING;	1320	prev->flags \|= CF_RUNNING;
1350	}	1321	}
1351		1322
1352	prev->flags &= ~CF_RUNNING;	1323	prev->flags &= ~CF_RUNNING;
1353	next->flags \|= CF_RUNNING;	1324	next->flags \|= CF_RUNNING;
1354
1355	LOCK;
1356		1325
1357	/* first get rid of the old state */	1326	/* first get rid of the old state */
1358	save_perl (aTHX_ prev);	1327	save_perl (aTHX_ prev);
1359		1328
1360	if (expect_false (next->flags & CF_NEW))	1329	if (expect_false (next->flags & CF_NEW))
…		…
1369		1338
1370	prev__cctx = prev->cctx;	1339	prev__cctx = prev->cctx;
1371		1340
1372	/* possibly untie and reuse the cctx */	1341	/* possibly untie and reuse the cctx */
1373	if (expect_true (	1342	if (expect_true (
1374	prev__cctx->idle_sp == stacklevel	1343	prev__cctx->idle_sp == (void *)stacklevel
1375	&& !(prev__cctx->flags & CC_TRACE)	1344	&& !(prev__cctx->flags & CC_TRACE)
1376	&& !force_cctx	1345	&& !force_cctx
1377	))	1346	))
1378	{	1347	{
1379	/* I assume that stacklevel is a stronger indicator than PL_top_env changes */	1348	/* I assume that stacklevel is a stronger indicator than PL_top_env changes */
…		…
1393	++next->usecount;	1362	++next->usecount;
1394		1363
1395	if (expect_true (!next->cctx))	1364	if (expect_true (!next->cctx))
1396	next->cctx = cctx_get (aTHX);	1365	next->cctx = cctx_get (aTHX);
1397		1366
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))	1367	if (expect_false (prev__cctx != next->cctx))
1402	{	1368	{
1403	prev__cctx->top_env = PL_top_env;	1369	prev__cctx->top_env = PL_top_env;
1404	PL_top_env = next->cctx->top_env;	1370	PL_top_env = next->cctx->top_env;
1405	coro_transfer (&prev__cctx->cctx, &next->cctx->cctx);	1371	coro_transfer (&prev__cctx->cctx, &next->cctx->cctx);
…		…
1418	coro_state_destroy (pTHX_ struct coro *coro)	1384	coro_state_destroy (pTHX_ struct coro *coro)
1419	{	1385	{
1420	if (coro->flags & CF_DESTROYED)	1386	if (coro->flags & CF_DESTROYED)
1421	return 0;	1387	return 0;
1422		1388
		1389	if (coro->on_destroy)
		1390	coro->on_destroy (aTHX_ coro);
		1391
1423	coro->flags \|= CF_DESTROYED;	1392	coro->flags \|= CF_DESTROYED;
1424		1393
1425	if (coro->flags & CF_READY)	1394	if (coro->flags & CF_READY)
1426	{	1395	{
1427	/* reduce nready, as destroying a ready coro effectively unreadies it */	1396	/* reduce nready, as destroying a ready coro effectively unreadies it */
1428	/* alternative: look through all ready queues and remove the coro */	1397	/* alternative: look through all ready queues and remove the coro */
1429	LOCK;
1430	--coro_nready;	1398	--coro_nready;
1431	UNLOCK;
1432	}	1399	}
1433	else	1400	else
1434	coro->flags \|= CF_READY; /* make sure it is NOT put into the readyqueue */	1401	coro->flags \|= CF_READY; /* make sure it is NOT put into the readyqueue */
1435		1402
1436	if (coro->mainstack && coro->mainstack != main_mainstack)	1403	if (coro->mainstack && coro->mainstack != main_mainstack)
1437	{	1404	{
1438	struct coro temp;	1405	struct coro temp;
1439		1406
1440	if (coro->flags & CF_RUNNING)	1407	assert (("FATAL: tried to destroy currently running coroutine (please report)", !(coro->flags & CF_RUNNING)));
1441	croak ("FATAL: tried to destroy currently running coroutine");
1442		1408
1443	save_perl (aTHX_ &temp);	1409	save_perl (aTHX_ &temp);
1444	load_perl (aTHX_ coro);	1410	load_perl (aTHX_ coro);
1445		1411
1446	coro_destruct (aTHX_ coro);	1412	coro_destruct (aTHX_ coro);
…		…
1497	# define MGf_DUP 0	1463	# define MGf_DUP 0
1498	#endif	1464	#endif
1499	};	1465	};
1500		1466
1501	static void	1467	static void
1502	prepare_transfer (pTHX_ struct transfer_args ta, SV prev_sv, SV *next_sv)	1468	prepare_transfer (pTHX_ struct coro_transfer_args ta, SV prev_sv, SV *next_sv)
1503	{	1469	{
1504	ta->prev = SvSTATE (prev_sv);	1470	ta->prev = SvSTATE (prev_sv);
1505	ta->next = SvSTATE (next_sv);	1471	ta->next = SvSTATE (next_sv);
1506	TRANSFER_CHECK (*ta);	1472	TRANSFER_CHECK (*ta);
1507	}	1473	}
1508		1474
1509	static void	1475	static void
1510	api_transfer (pTHX_ SV prev_sv, SV next_sv)	1476	api_transfer (pTHX_ SV prev_sv, SV next_sv)
1511	{	1477	{
1512	struct transfer_args ta;	1478	struct coro_transfer_args ta;
1513		1479
1514	prepare_transfer (aTHX_ &ta, prev_sv, next_sv);	1480	prepare_transfer (aTHX_ &ta, prev_sv, next_sv);
1515	TRANSFER (ta, 1);	1481	TRANSFER (ta, 1);
1516	}	1482	}
1517		1483
1518	/ Coro ******************************************************************/	1484	/ Coro ******************************************************************/
1519		1485
1520	static void	1486	INLINE void
1521	coro_enq (pTHX_ SV *coro_sv)	1487	coro_enq (pTHX_ struct coro *coro)
1522	{	1488	{
1523	av_push (coro_ready [SvSTATE (coro_sv)->prio - PRIO_MIN], coro_sv);	1489	av_push (coro_ready [coro->prio - PRIO_MIN], SvREFCNT_inc_NN (coro->hv));
1524	}	1490	}
1525		1491
1526	static SV *	1492	INLINE SV *
1527	coro_deq (pTHX)	1493	coro_deq (pTHX)
1528	{	1494	{
1529	int prio;	1495	int prio;
1530		1496
1531	for (prio = PRIO_MAX - PRIO_MIN + 1; --prio >= 0; )	1497	for (prio = PRIO_MAX - PRIO_MIN + 1; --prio >= 0; )
…		…
1550	if (coro->flags & CF_READY)	1516	if (coro->flags & CF_READY)
1551	return 0;	1517	return 0;
1552		1518
1553	coro->flags \|= CF_READY;	1519	coro->flags \|= CF_READY;
1554		1520
1555	LOCK;
1556
1557	sv_hook = coro_nready ? 0 : coro_readyhook;	1521	sv_hook = coro_nready ? 0 : coro_readyhook;
1558	xs_hook = coro_nready ? 0 : coroapi.readyhook;	1522	xs_hook = coro_nready ? 0 : coroapi.readyhook;
1559		1523
1560	coro_enq (aTHX_ SvREFCNT_inc_NN (coro_sv));	1524	coro_enq (aTHX_ coro);
1561	++coro_nready;	1525	++coro_nready;
1562		1526
1563	UNLOCK;
1564
1565	if (sv_hook)	1527	if (sv_hook)
1566	{	1528	{
1567	dSP;	1529	dSP;
1568		1530
1569	ENTER;	1531	ENTER;
…		…
1589	{	1551	{
1590	return !!(SvSTATE (coro_sv)->flags & CF_READY);	1552	return !!(SvSTATE (coro_sv)->flags & CF_READY);
1591	}	1553	}
1592		1554
1593	INLINE void	1555	INLINE void
1594	prepare_schedule (pTHX_ struct transfer_args *ta)	1556	prepare_schedule (pTHX_ struct coro_transfer_args *ta)
1595	{	1557	{
1596	SV prev_sv, next_sv;	1558	SV prev_sv, next_sv;
1597		1559
1598	for (;;)	1560	for (;;)
1599	{	1561	{
1600	LOCK;
1601	next_sv = coro_deq (aTHX);	1562	next_sv = coro_deq (aTHX);
1602		1563
1603	/* nothing to schedule: call the idle handler */	1564	/* nothing to schedule: call the idle handler */
1604	if (expect_false (!next_sv))	1565	if (expect_false (!next_sv))
1605	{	1566	{
1606	dSP;	1567	dSP;
1607	UNLOCK;
1608		1568
1609	ENTER;	1569	ENTER;
1610	SAVETMPS;	1570	SAVETMPS;
1611		1571
1612	PUSHMARK (SP);	1572	PUSHMARK (SP);
…		…
1617	FREETMPS;	1577	FREETMPS;
1618	LEAVE;	1578	LEAVE;
1619	continue;	1579	continue;
1620	}	1580	}
1621		1581
1622	ta->next = SvSTATE (next_sv);	1582	ta->next = SvSTATE_hv (next_sv);
1623		1583
1624	/* cannot transfer to destroyed coros, skip and look for next */	1584	/* cannot transfer to destroyed coros, skip and look for next */
1625	if (expect_false (ta->next->flags & CF_DESTROYED))	1585	if (expect_false (ta->next->flags & CF_DESTROYED))
1626	{	1586	{
1627	UNLOCK;
1628	SvREFCNT_dec (next_sv);	1587	SvREFCNT_dec (next_sv);
1629	/* coro_nready has already been taken care of by destroy */	1588	/* coro_nready has already been taken care of by destroy */
1630	continue;	1589	continue;
1631	}	1590	}
1632		1591
1633	--coro_nready;	1592	--coro_nready;
1634	UNLOCK;
1635	break;	1593	break;
1636	}	1594	}
1637		1595
1638	/* free this only after the transfer */	1596	/* free this only after the transfer */
1639	prev_sv = SvRV (coro_current);	1597	prev_sv = SvRV (coro_current);
1640	ta->prev = SvSTATE (prev_sv);	1598	ta->prev = SvSTATE_hv (prev_sv);
1641	TRANSFER_CHECK (*ta);	1599	TRANSFER_CHECK (*ta);
1642	assert (("FATAL: next coroutine isn't marked as ready in Coro (please report)", ta->next->flags & CF_READY));	1600	assert (("FATAL: next coroutine isn't marked as ready in Coro (please report)", ta->next->flags & CF_READY));
1643	ta->next->flags &= ~CF_READY;	1601	ta->next->flags &= ~CF_READY;
1644	SvRV_set (coro_current, next_sv);	1602	SvRV_set (coro_current, next_sv);
1645		1603
1646	LOCK;
1647	free_coro_mortal (aTHX);	1604	free_coro_mortal (aTHX);
1648	coro_mortal = prev_sv;	1605	coro_mortal = prev_sv;
1649	UNLOCK;
1650	}	1606	}
1651		1607
1652	INLINE void	1608	INLINE void
1653	prepare_cede (pTHX_ struct transfer_args *ta)	1609	prepare_cede (pTHX_ struct coro_transfer_args *ta)
1654	{	1610	{
1655	api_ready (aTHX_ coro_current);	1611	api_ready (aTHX_ coro_current);
1656	prepare_schedule (aTHX_ ta);	1612	prepare_schedule (aTHX_ ta);
1657	}	1613	}
1658		1614
1659	static void	1615	INLINE void
1660	prepare_cede_notself (pTHX_ struct transfer_args *ta)	1616	prepare_cede_notself (pTHX_ struct coro_transfer_args *ta)
1661	{	1617	{
1662	SV *prev = SvRV (coro_current);	1618	SV *prev = SvRV (coro_current);
1663		1619
1664	if (coro_nready)	1620	if (coro_nready)
1665	{	1621	{
1666	prepare_schedule (aTHX_ ta);	1622	prepare_schedule (aTHX_ ta);
1667	api_ready (aTHX_ prev);	1623	api_ready (aTHX_ prev);
1668	}	1624	}
1669	else	1625	else
1670	ta->prev = ta->next = SvSTATE (prev);	1626	prepare_nop (aTHX_ ta);
1671	}	1627	}
1672		1628
1673	static void	1629	static void
1674	api_schedule (pTHX)	1630	api_schedule (pTHX)
1675	{	1631	{
1676	struct transfer_args ta;	1632	struct coro_transfer_args ta;
1677		1633
1678	prepare_schedule (aTHX_ &ta);	1634	prepare_schedule (aTHX_ &ta);
1679	TRANSFER (ta, 1);	1635	TRANSFER (ta, 1);
1680	}	1636	}
1681		1637
1682	static int	1638	static int
1683	api_cede (pTHX)	1639	api_cede (pTHX)
1684	{	1640	{
1685	struct transfer_args ta;	1641	struct coro_transfer_args ta;
1686		1642
1687	prepare_cede (aTHX_ &ta);	1643	prepare_cede (aTHX_ &ta);
1688		1644
1689	if (expect_true (ta.prev != ta.next))	1645	if (expect_true (ta.prev != ta.next))
1690	{	1646	{
…		…
1698	static int	1654	static int
1699	api_cede_notself (pTHX)	1655	api_cede_notself (pTHX)
1700	{	1656	{
1701	if (coro_nready)	1657	if (coro_nready)
1702	{	1658	{
1703	struct transfer_args ta;	1659	struct coro_transfer_args ta;
1704		1660
1705	prepare_cede_notself (aTHX_ &ta);	1661	prepare_cede_notself (aTHX_ &ta);
1706	TRANSFER (ta, 1);	1662	TRANSFER (ta, 1);
1707	return 1;	1663	return 1;
1708	}	1664	}
…		…
1718	if (flags & CC_TRACE)	1674	if (flags & CC_TRACE)
1719	{	1675	{
1720	if (!coro->cctx)	1676	if (!coro->cctx)
1721	coro->cctx = cctx_new_run ();	1677	coro->cctx = cctx_new_run ();
1722	else if (!(coro->cctx->flags & CC_TRACE))	1678	else if (!(coro->cctx->flags & CC_TRACE))
1723	croak ("cannot enable tracing on coroutine with custom stack");	1679	croak ("cannot enable tracing on coroutine with custom stack,");
1724		1680
1725	coro->cctx->flags \|= CC_NOREUSE \| (flags & (CC_TRACE \| CC_TRACE_ALL));	1681	coro->cctx->flags \|= CC_NOREUSE \| (flags & (CC_TRACE \| CC_TRACE_ALL));
1726	}	1682	}
1727	else if (coro->cctx && coro->cctx->flags & CC_TRACE)	1683	else if (coro->cctx && coro->cctx->flags & CC_TRACE)
1728	{	1684	{
…		…
1732	PL_runops = RUNOPS_DEFAULT;	1688	PL_runops = RUNOPS_DEFAULT;
1733	else	1689	else
1734	coro->slot->runops = RUNOPS_DEFAULT;	1690	coro->slot->runops = RUNOPS_DEFAULT;
1735	}	1691	}
1736	}	1692	}
1737
1738	#if 0
1739	static int
1740	coro_gensub_free (pTHX_ SV sv, MAGIC mg)
1741	{
1742	AV *padlist;
1743	AV av = (AV )mg->mg_obj;
1744
1745	abort ();
1746
1747	return 0;
1748	}
1749
1750	static MGVTBL coro_gensub_vtbl = {
1751	0, 0, 0, 0,
1752	coro_gensub_free
1753	};
1754	#endif
1755		1693
1756	/*****************************************************************************/	1694	/*****************************************************************************/
1757	/* PerlIO::cede */	1695	/* PerlIO::cede */
1758		1696
1759	typedef struct	1697	typedef struct
…		…
1833	static const CV slf_cv; / for quick consistency check */	1771	static const CV slf_cv; / for quick consistency check */
1834		1772
1835	static UNOP slf_restore; /* restore stack as entersub did, for first-re-run */	1773	static UNOP slf_restore; /* restore stack as entersub did, for first-re-run */
1836	static SV *slf_arg0;	1774	static SV *slf_arg0;
1837	static SV *slf_arg1;	1775	static SV *slf_arg1;
		1776	static SV *slf_arg2;
1838		1777
1839	/* this restores the stack in the case we patched the entersub, to */	1778	/* this restores the stack in the case we patched the entersub, to */
1840	/* recreate the stack frame as perl will on following calls */	1779	/* recreate the stack frame as perl will on following calls */
1841	/* since entersub cleared the stack */	1780	/* since entersub cleared the stack */
1842	static OP *	1781	static OP *
…		…
1847	PUSHMARK (SP);	1786	PUSHMARK (SP);
1848		1787
1849	EXTEND (SP, 3);	1788	EXTEND (SP, 3);
1850	if (slf_arg0) PUSHs (sv_2mortal (slf_arg0));	1789	if (slf_arg0) PUSHs (sv_2mortal (slf_arg0));
1851	if (slf_arg1) PUSHs (sv_2mortal (slf_arg1));	1790	if (slf_arg1) PUSHs (sv_2mortal (slf_arg1));
		1791	if (slf_arg2) PUSHs (sv_2mortal (slf_arg2));
1852	PUSHs ((SV *)CvGV (slf_cv));	1792	PUSHs ((SV *)CvGV (slf_cv));
1853		1793
1854	RETURNOP (slf_restore.op_first);	1794	RETURNOP (slf_restore.op_first);
1855	}	1795	}
1856		1796
1857	#define OPpENTERSUB_SLF 15 /* the part of op_private entersub hopefully doesn't use */	1797	static void
		1798	slf_prepare_set_stacklevel (pTHX_ struct coro_transfer_args *ta)
		1799	{
		1800	prepare_set_stacklevel (ta, (struct coro_cctx *)slf_frame.data);
		1801	}
1858		1802
1859	enum {	1803	static void
1860	CORO_SLF_CUSTOM = 0,	1804	slf_init_set_stacklevel (pTHX_ struct CoroSLF frame, CV cv, SV **arg, int items)
1861	CORO_SLF_SET_STACKLEVEL = 1,	1805	{
1862	CORO_SLF_TRANSFER = 2	1806	assert (("FATAL: set_stacklevel needs the coro cctx as sole argument", items == 1));
1863	};
1864		1807
1865	/* declare prototype */	1808	frame->prepare = slf_prepare_set_stacklevel;
1866	XS(XS_Coro__State__set_stacklevel);	1809	frame->check = slf_check_nop;
		1810	frame->data = (void *)SvIV (arg [0]);
		1811	}
		1812
		1813	static void
		1814	slf_prepare_transfer (pTHX_ struct coro_transfer_args *ta)
		1815	{
		1816	SV arg = (SV )slf_frame.data;
		1817
		1818	prepare_transfer (aTHX_ ta, arg [0], arg [1]);
		1819
		1820	/* if the destination has ->throw set, then copy it */
		1821	/* into the current coro's throw slot, so it will be raised */
		1822	/* after the schedule */
		1823	if (expect_false (ta->next->throw))
		1824	{
		1825	struct coro *coro = SvSTATE_current;
		1826	SvREFCNT_dec (coro->throw);
		1827	coro->throw = ta->next->throw;
		1828	ta->next->throw = 0;
		1829	}
		1830	}
		1831
		1832	static void
		1833	slf_init_transfer (pTHX_ struct CoroSLF frame, CV cv, SV **arg, int items)
		1834	{
		1835	if (items != 2)
		1836	croak ("Coro::State::transfer (prev, next) expects two arguments, not %d,", items);
		1837
		1838	frame->prepare = slf_prepare_transfer;
		1839	frame->check = slf_check_nop;
		1840	frame->data = (void )arg; / let's hope it will stay valid */
		1841	}
		1842
		1843	static void
		1844	slf_init_schedule (pTHX_ struct CoroSLF frame, CV cv, SV **arg, int items)
		1845	{
		1846	frame->prepare = prepare_schedule;
		1847	frame->check = slf_check_nop;
		1848	}
		1849
		1850	static void
		1851	slf_init_cede (pTHX_ struct CoroSLF frame, CV cv, SV **arg, int items)
		1852	{
		1853	frame->prepare = prepare_cede;
		1854	frame->check = slf_check_nop;
		1855	}
		1856
		1857	static void
		1858	slf_init_cede_notself (pTHX_ struct CoroSLF frame, CV cv, SV **arg, int items)
		1859	{
		1860	frame->prepare = prepare_cede_notself;
		1861	frame->check = slf_check_nop;
		1862	}
		1863
		1864	/* we hijack an hopefully unused CV flag for our purposes */
		1865	#define CVf_SLF 0x4000
1867		1866
1868	/*	1867	/*
1869	* these not obviously related functions are all rolled into one	1868	* these not obviously related functions are all rolled into one
1870	* function to increase chances that they all will call transfer with the same	1869	* function to increase chances that they all will call transfer with the same
1871	* stack offset	1870	* stack offset
1872	* SLF stands for "schedule-like-function".	1871	* SLF stands for "schedule-like-function".
1873	*/	1872	*/
1874	static OP *	1873	static OP *
1875	pp_slf (pTHX)	1874	pp_slf (pTHX)
1876	{	1875	{
		1876	I32 checkmark; /* mark SP to see how many elements check has pushed */
		1877
		1878	/* set up the slf frame, unless it has already been set-up */
		1879	/* the latter happens when a new coro has been started */
		1880	/* or when a new cctx was attached to an existing coroutine */
		1881	if (expect_true (!slf_frame.prepare))
		1882	{
		1883	/* first iteration */
1877	dSP;	1884	dSP;
1878	struct transfer_args ta;
1879	SV **arg = PL_stack_base + TOPMARK + 1;	1885	SV **arg = PL_stack_base + TOPMARK + 1;
1880	int items = SP - arg; /* args without function object */	1886	int items = SP - arg; /* args without function object */
1881	int ix = PL_op->op_private & OPpENTERSUB_SLF;	1887	SV gv = sp;
1882	struct CoroSLF *slf = 0;
1883		1888
1884	/* do a quick consistency check on the "function" object, and if it isn't */	1889	/* do a quick consistency check on the "function" object, and if it isn't */
1885	/* for us, divert to the real entersub */	1890	/* for us, divert to the real entersub */
1886	if (SvTYPE (sp) != SVt_PVGV \|\| CvXSUB (GvCV (sp)) != XS_Coro__State__set_stacklevel)	1891	if (SvTYPE (gv) != SVt_PVGV \|\| !(CvFLAGS (GvCV (gv)) & CVf_SLF))
1887	return PL_ppaddr[OP_ENTERSUB](aTHX);	1892	return PL_ppaddr[OP_ENTERSUB](aTHX);
1888		1893
1889	/* pop args */
1890	SP = PL_stack_base + POPMARK;
1891
1892	if (!(PL_op->op_flags & OPf_STACKED))	1894	if (!(PL_op->op_flags & OPf_STACKED))
1893	{	1895	{
1894	/* ampersand-form of call, use @_ instead of stack */	1896	/* ampersand-form of call, use @_ instead of stack */
1895	AV *av = GvAV (PL_defgv);	1897	AV *av = GvAV (PL_defgv);
1896	arg = AvARRAY (av);	1898	arg = AvARRAY (av);
1897	items = AvFILLp (av) + 1;	1899	items = AvFILLp (av) + 1;
1898	}	1900	}
1899		1901
1900	PUTBACK;	1902	PUTBACK;
1901		1903
1902	if (!ix)	1904	/* now call the init function, which needs to set up slf_frame */
		1905	((coro_slf_cb)CvXSUBANY (GvCV (gv)).any_ptr)
		1906	(aTHX_ &slf_frame, GvCV (gv), arg, items);
		1907
		1908	/* pop args */
		1909	SP = PL_stack_base + POPMARK;
		1910
		1911	PUTBACK;
1903	{	1912	}
1904	slf = (struct CoroSLF )CvSTART (GvCV (sp));
1905	ix = slf->prepare (aTHX_ arg, items);
1906	}
1907		1913
1908	switch (ix)	1914	/* now that we have a slf_frame, interpret it! */
1909	{	1915	/* we use a callback system not to make the code needlessly */
1910	case CORO_SLF_SET_STACKLEVEL:	1916	/* complicated, but so we can run multiple perl coros from one cctx */
1911	prepare_set_stacklevel (&ta, (struct coro_cctx *)SvIV (arg [0]));
1912	break;
1913
1914	case CORO_SLF_TRANSFER:
1915	if (items != 2)
1916	croak ("Coro::State::transfer (prev, next) expects two arguments, not %d.", items);
1917
1918	prepare_transfer (aTHX_ &ta, arg [0], arg [1]);
1919	break;
1920
1921	case CORO_SLF_SCHEDULE:
1922	prepare_schedule (aTHX_ &ta);
1923	break;
1924
1925	case CORO_SLF_CEDE:
1926	prepare_cede (aTHX_ &ta);
1927	break;
1928
1929	case CORO_SLF_CEDE_NOTSELF:
1930	prepare_cede_notself (aTHX_ &ta);
1931	break;
1932
1933	default:
1934	abort ();
1935	}
1936		1917
1937	do	1918	do
		1919	{
		1920	struct coro_transfer_args ta;
		1921
		1922	slf_frame.prepare (aTHX_ &ta);
1938	TRANSFER (ta, 0);	1923	TRANSFER (ta, 0);
1939	while (slf && slf->check (aTHX));
1940		1924
1941	SPAGAIN;	1925	checkmark = PL_stack_sp - PL_stack_base;
		1926	}
		1927	while (slf_frame.check (aTHX_ &slf_frame));
1942		1928
		1929	{
		1930	dSP;
		1931	SV **bot = PL_stack_base + checkmark;
		1932	int gimme = GIMME_V;
		1933
		1934	slf_frame.prepare = 0; /* invalidate the frame, so it gets initialised again next time */
		1935
		1936	/* make sure we put something on the stack in scalar context */
		1937	if (gimme == G_SCALAR)
		1938	{
		1939	if (sp == bot)
		1940	XPUSHs (&PL_sv_undef);
		1941
		1942	SP = bot + 1;
		1943	}
		1944
1943	PUTBACK;	1945	PUTBACK;
1944	SLF_TAIL;	1946	}
1945	SPAGAIN;
1946	RETURN;
1947	}
1948		1947
1949	static void	1948	{
1950	coro_slf_patch (pTHX_ CV cv, int ix, SV *args, int items)	1949	struct coro *coro = SvSTATE_current;
1951	{
1952	assert (("FATAL: SLF call recursion in Coro module (please report)", PL_op->op_ppaddr != pp_slf));
1953		1950
		1951	if (expect_false (coro->throw))
		1952	{
		1953	SV *exception = sv_2mortal (coro->throw);
		1954
		1955	coro->throw = 0;
		1956	sv_setsv (ERRSV, exception);
		1957	croak (0);
		1958	}
		1959	}
		1960
		1961	return NORMAL;
		1962	}
		1963
		1964	static void
		1965	api_execute_slf (pTHX_ CV cv, coro_slf_cb init_cb, SV *arg, int items)
		1966	{
1954	assert (("FATAL: SLF call with illegal CV value", CvGV (cv)));	1967	assert (("FATAL: SLF call with illegal CV value", !CvANON (cv)));
		1968
		1969	if (PL_op->op_ppaddr != PL_ppaddr [OP_ENTERSUB]
		1970	&& PL_op->op_ppaddr != pp_slf)
		1971	croak ("FATAL: Coro SLF calls can only be made normally, not via goto or any other means, caught");
		1972
		1973	if (items > 3)
		1974	croak ("Coro only supports up to three arguments to SLF functions currently (not %d), caught", items);
		1975
		1976	CvFLAGS (cv) \|= CVf_SLF;
		1977	CvXSUBANY (cv).any_ptr = (void *)init_cb;
1955	slf_cv = cv;	1978	slf_cv = cv;
1956		1979
1957	/* we patch the op, and then re-run the whole call */	1980	/* we patch the op, and then re-run the whole call */
1958	/* we have to put the same argument on the stack for this to work */	1981	/* we have to put the same argument on the stack for this to work */
1959	/* and this will be done by pp_restore */	1982	/* and this will be done by pp_restore */
1960	slf_restore.op_next = (OP *)&slf_restore;	1983	slf_restore.op_next = (OP *)&slf_restore;
1961	slf_restore.op_type = OP_NULL;	1984	slf_restore.op_type = OP_CUSTOM;
1962	slf_restore.op_ppaddr = pp_restore;	1985	slf_restore.op_ppaddr = pp_restore;
1963	slf_restore.op_first = PL_op;	1986	slf_restore.op_first = PL_op;
1964		1987
1965	slf_arg0 = items > 0 ? SvREFCNT_inc (args [0]) : 0;	1988	slf_arg0 = items > 0 ? SvREFCNT_inc (arg [0]) : 0;
1966	slf_arg1 = items > 1 ? SvREFCNT_inc (args [1]) : 0;	1989	slf_arg1 = items > 1 ? SvREFCNT_inc (arg [1]) : 0;
		1990	slf_arg2 = items > 2 ? SvREFCNT_inc (arg [2]) : 0;
1967		1991
1968	PL_op->op_ppaddr = pp_slf;	1992	PL_op->op_ppaddr = pp_slf;
1969	PL_op->op_private = PL_op->op_private & ~OPpENTERSUB_SLF \| ix; /* we potentially share our private flags with entersub */
1970		1993
1971	PL_op = (OP *)&slf_restore;	1994	PL_op = (OP *)&slf_restore;
1972	}	1995	}
1973		1996
		1997	/*****************************************************************************/
		1998
		1999	static void
		2000	coro_semaphore_adjust (pTHX_ AV *av, IV adjust)
		2001	{
		2002	SV *count_sv = AvARRAY (av)[0];
		2003	IV count = SvIVX (count_sv);
		2004
		2005	count += adjust;
		2006	SvIVX (count_sv) = count;
		2007
		2008	/* now wake up as many waiters as are expected to lock */
		2009	while (count > 0 && AvFILLp (av) > 0)
		2010	{
		2011	SV *cb;
		2012
		2013	/* swap first two elements so we can shift a waiter */
		2014	AvARRAY (av)[0] = AvARRAY (av)[1];
		2015	AvARRAY (av)[1] = count_sv;
		2016	cb = av_shift (av);
		2017
		2018	if (SvOBJECT (cb))
		2019	api_ready (aTHX_ cb);
		2020	else
		2021	croak ("callbacks not yet supported");
		2022
		2023	SvREFCNT_dec (cb);
		2024
		2025	--count;
		2026	}
		2027	}
		2028
		2029	static void
		2030	coro_semaphore_on_destroy (pTHX_ struct coro *coro)
		2031	{
		2032	/* call $sem->adjust (0) to possibly wake up some other waiters */
		2033	coro_semaphore_adjust (aTHX_ (AV *)coro->slf_frame.data, 0);
		2034	}
		2035
		2036	static int
		2037	slf_check_semaphore_down (pTHX_ struct CoroSLF *frame)
		2038	{
		2039	AV av = (AV )frame->data;
		2040	SV *count_sv = AvARRAY (av)[0];
		2041
		2042	if (SvIVX (count_sv) > 0)
		2043	{
		2044	SvSTATE_current->on_destroy = 0;
		2045	SvIVX (count_sv) = SvIVX (count_sv) - 1;
		2046	return 0;
		2047	}
		2048	else
		2049	{
		2050	int i;
		2051	/* if we were woken up but can't down, we look through the whole */
		2052	/* waiters list and only add us if we aren't in there already */
		2053	/* this avoids some degenerate memory usage cases */
		2054
		2055	for (i = 1; i <= AvFILLp (av); ++i)
		2056	if (AvARRAY (av)[i] == SvRV (coro_current))
		2057	return 1;
		2058
		2059	av_push (av, SvREFCNT_inc (SvRV (coro_current)));
		2060	return 1;
		2061	}
		2062	}
		2063
		2064	static void
		2065	slf_init_semaphore_down (pTHX_ struct CoroSLF frame, CV cv, SV **arg, int items)
		2066	{
		2067	AV av = (AV )SvRV (arg [0]);
		2068
		2069	if (SvIVX (AvARRAY (av)[0]) > 0)
		2070	{
		2071	frame->data = (void *)av;
		2072	frame->prepare = prepare_nop;
		2073	SvSTATE_current->on_destroy = coro_semaphore_on_destroy;
		2074	}
		2075	else
		2076	{
		2077	av_push (av, SvREFCNT_inc (SvRV (coro_current)));
		2078
		2079	frame->data = (void )sv_2mortal (SvREFCNT_inc ((SV )av));
		2080	frame->prepare = prepare_schedule;
		2081
		2082	/* to avoid race conditions when a woken-up coro gets terminated */
		2083	/* we arrange for a temporary on_destroy that calls adjust (0) */
		2084	assert (!SvSTATE_current->on_destroy);//D
		2085	SvSTATE_current->on_destroy = coro_semaphore_on_destroy;
		2086	}
		2087
		2088	frame->check = slf_check_semaphore_down;
		2089
		2090	}
		2091
		2092	/*****************************************************************************/
		2093
		2094	#define GENSUB_ARG CvXSUBANY (cv).any_ptr
		2095
		2096	/* create a closure from XS, returns a code reference */
		2097	/* the arg can be accessed via GENSUB_ARG from the callback */
		2098	/* the callback must use dXSARGS/XSRETURN */
		2099	static SV *
		2100	gensub (pTHX_ void (xsub)(pTHX_ CV ), void *arg)
		2101	{
		2102	CV cv = (CV )NEWSV (0, 0);
		2103
		2104	sv_upgrade ((SV *)cv, SVt_PVCV);
		2105
		2106	CvANON_on (cv);
		2107	CvISXSUB_on (cv);
		2108	CvXSUB (cv) = xsub;
		2109	GENSUB_ARG = arg;
		2110
		2111	return newRV_noinc ((SV *)cv);
		2112	}
		2113
		2114	/*****************************************************************************/
		2115
1974	MODULE = Coro::State PACKAGE = Coro::State PREFIX = api_	2116	MODULE = Coro::State PACKAGE = Coro::State PREFIX = api_
1975		2117
1976	PROTOTYPES: DISABLE	2118	PROTOTYPES: DISABLE
1977		2119
1978	BOOT:	2120	BOOT:
1979	{	2121	{
1980	#ifdef USE_ITHREADS	2122	#ifdef USE_ITHREADS
1981	MUTEX_INIT (&coro_lock);
1982	# if CORO_PTHREAD	2123	# if CORO_PTHREAD
1983	coro_thx = PERL_GET_CONTEXT;	2124	coro_thx = PERL_GET_CONTEXT;
1984	# endif	2125	# endif
1985	#endif	2126	#endif
1986	BOOT_PAGESIZE;	2127	BOOT_PAGESIZE;
…		…
2007	main_top_env = PL_top_env;	2148	main_top_env = PL_top_env;
2008		2149
2009	while (main_top_env->je_prev)	2150	while (main_top_env->je_prev)
2010	main_top_env = main_top_env->je_prev;	2151	main_top_env = main_top_env->je_prev;
2011		2152
		2153	{
		2154	SV *slf = sv_2mortal (newSViv (PTR2IV (pp_slf)));
		2155
		2156	if (!PL_custom_op_names) PL_custom_op_names = newHV ();
		2157	hv_store_ent (PL_custom_op_names, slf,
		2158	newSVpv ("coro_slf", 0), 0);
		2159
		2160	if (!PL_custom_op_descs) PL_custom_op_descs = newHV ();
		2161	hv_store_ent (PL_custom_op_descs, slf,
		2162	newSVpv ("coro schedule like function", 0), 0);
		2163	}
		2164
2012	coroapi.ver = CORO_API_VERSION;	2165	coroapi.ver = CORO_API_VERSION;
2013	coroapi.rev = CORO_API_REVISION;	2166	coroapi.rev = CORO_API_REVISION;
		2167
2014	coroapi.transfer = api_transfer;	2168	coroapi.transfer = api_transfer;
		2169
		2170	coroapi.sv_state = SvSTATE_;
		2171	coroapi.execute_slf = api_execute_slf;
		2172	coroapi.prepare_nop = prepare_nop;
		2173	coroapi.prepare_schedule = prepare_schedule;
		2174	coroapi.prepare_cede = prepare_cede;
		2175	coroapi.prepare_cede_notself = prepare_cede_notself;
2015		2176
2016	{	2177	{
2017	SV **svp = hv_fetch (PL_modglobal, "Time::NVtime", 12, 0);	2178	SV **svp = hv_fetch (PL_modglobal, "Time::NVtime", 12, 0);
2018		2179
2019	if (!svp) croak ("Time::HiRes is required");	2180	if (!svp) croak ("Time::HiRes is required");
…		…
2054	OUTPUT:	2215	OUTPUT:
2055	RETVAL	2216	RETVAL
2056		2217
2057	void	2218	void
2058	_set_stacklevel (...)	2219	_set_stacklevel (...)
2059	ALIAS:	2220	CODE:
2060	_set_stacklevel = CORO_SLF_SET_STACKLEVEL	2221	api_execute_slf (aTHX_ cv, slf_init_set_stacklevel, &ST (0), items);
2061	Coro::State::transfer = CORO_SLF_TRANSFER	2222
2062	Coro::schedule = CORO_SLF_SCHEDULE	2223	void
2063	Coro::cede = CORO_SLF_CEDE	2224	transfer (...)
2064	Coro::cede_notself = CORO_SLF_CEDE_NOTSELF	2225	PROTOTYPE: $$
2065	CODE:	2226	CODE:
2066	coro_slf_patch (aTHX_ cv, ix, &ST (0), items);	2227	api_execute_slf (aTHX_ cv, slf_init_transfer, &ST (0), items);
2067		2228
2068	bool	2229	bool
2069	_destroy (SV *coro_sv)	2230	_destroy (SV *coro_sv)
2070	CODE:	2231	CODE:
2071	RETVAL = coro_state_destroy (aTHX_ SvSTATE (coro_sv));	2232	RETVAL = coro_state_destroy (aTHX_ SvSTATE (coro_sv));
…		…
2078	CODE:	2239	CODE:
2079	_exit (code);	2240	_exit (code);
2080		2241
2081	int	2242	int
2082	cctx_stacksize (int new_stacksize = 0)	2243	cctx_stacksize (int new_stacksize = 0)
		2244	PROTOTYPE: ;$
2083	CODE:	2245	CODE:
2084	RETVAL = cctx_stacksize;	2246	RETVAL = cctx_stacksize;
2085	if (new_stacksize)	2247	if (new_stacksize)
2086	{	2248	{
2087	cctx_stacksize = new_stacksize;	2249	cctx_stacksize = new_stacksize;
…		…
2090	OUTPUT:	2252	OUTPUT:
2091	RETVAL	2253	RETVAL
2092		2254
2093	int	2255	int
2094	cctx_max_idle (int max_idle = 0)	2256	cctx_max_idle (int max_idle = 0)
		2257	PROTOTYPE: ;$
2095	CODE:	2258	CODE:
2096	RETVAL = cctx_max_idle;	2259	RETVAL = cctx_max_idle;
2097	if (max_idle > 1)	2260	if (max_idle > 1)
2098	cctx_max_idle = max_idle;	2261	cctx_max_idle = max_idle;
2099	OUTPUT:	2262	OUTPUT:
2100	RETVAL	2263	RETVAL
2101		2264
2102	int	2265	int
2103	cctx_count ()	2266	cctx_count ()
		2267	PROTOTYPE:
2104	CODE:	2268	CODE:
2105	RETVAL = cctx_count;	2269	RETVAL = cctx_count;
2106	OUTPUT:	2270	OUTPUT:
2107	RETVAL	2271	RETVAL
2108		2272
2109	int	2273	int
2110	cctx_idle ()	2274	cctx_idle ()
		2275	PROTOTYPE:
2111	CODE:	2276	CODE:
2112	RETVAL = cctx_idle;	2277	RETVAL = cctx_idle;
2113	OUTPUT:	2278	OUTPUT:
2114	RETVAL	2279	RETVAL
2115		2280
2116	void	2281	void
2117	list ()	2282	list ()
		2283	PROTOTYPE:
2118	PPCODE:	2284	PPCODE:
2119	{	2285	{
2120	struct coro *coro;	2286	struct coro *coro;
2121	for (coro = coro_first; coro; coro = coro->next)	2287	for (coro = coro_first; coro; coro = coro->next)
2122	if (coro->hv)	2288	if (coro->hv)
…		…
2189	SvREFCNT_dec (self->throw);	2355	SvREFCNT_dec (self->throw);
2190	self->throw = SvOK (throw) ? newSVsv (throw) : 0;	2356	self->throw = SvOK (throw) ? newSVsv (throw) : 0;
2191		2357
2192	void	2358	void
2193	api_trace (SV *coro, int flags = CC_TRACE \| CC_TRACE_SUB)	2359	api_trace (SV *coro, int flags = CC_TRACE \| CC_TRACE_SUB)
		2360	PROTOTYPE: $;$
2194	C_ARGS: aTHX_ coro, flags	2361	C_ARGS: aTHX_ coro, flags
2195		2362
2196	SV *	2363	SV *
2197	has_cctx (Coro::State coro)	2364	has_cctx (Coro::State coro)
2198	PROTOTYPE: $	2365	PROTOTYPE: $
…		…
2223	OUTPUT:	2390	OUTPUT:
2224	RETVAL	2391	RETVAL
2225		2392
2226	void	2393	void
2227	force_cctx ()	2394	force_cctx ()
		2395	PROTOTYPE:
2228	CODE:	2396	CODE:
2229	struct coro *coro = SvSTATE (coro_current);
2230	coro->cctx->idle_sp = 0;	2397	SvSTATE_current->cctx->idle_sp = 0;
2231		2398
2232	void	2399	void
2233	swap_defsv (Coro::State self)	2400	swap_defsv (Coro::State self)
2234	PROTOTYPE: $	2401	PROTOTYPE: $
2235	ALIAS:	2402	ALIAS:
2236	swap_defav = 1	2403	swap_defav = 1
2237	CODE:	2404	CODE:
2238	if (!self->slot)	2405	if (!self->slot)
2239	croak ("cannot swap state with coroutine that has no saved state");	2406	croak ("cannot swap state with coroutine that has no saved state,");
2240	else	2407	else
2241	{	2408	{
2242	SV src = ix ? (SV )&GvAV (PL_defgv) : &GvSV (PL_defgv);	2409	SV src = ix ? (SV )&GvAV (PL_defgv) : &GvSV (PL_defgv);
2243	SV dst = ix ? (SV )&self->slot->defav : (SV **)&self->slot->defsv;	2410	SV dst = ix ? (SV )&self->slot->defav : (SV **)&self->slot->defsv;
2244		2411
…		…
2286	SvREADONLY_on (sv);	2453	SvREADONLY_on (sv);
2287	}	2454	}
2288	}	2455	}
2289		2456
2290	void	2457	void
		2458	schedule (...)
		2459	CODE:
		2460	api_execute_slf (aTHX_ cv, slf_init_schedule, &ST (0), 0);
		2461
		2462	void
		2463	cede (...)
		2464	CODE:
		2465	api_execute_slf (aTHX_ cv, slf_init_cede, &ST (0), 0);
		2466
		2467	void
		2468	cede_notself (...)
		2469	CODE:
		2470	api_execute_slf (aTHX_ cv, slf_init_cede_notself, &ST (0), 0);
		2471
		2472	void
2291	_set_current (SV *current)	2473	_set_current (SV *current)
2292	PROTOTYPE: $	2474	PROTOTYPE: $
2293	CODE:	2475	CODE:
2294	SvREFCNT_dec (SvRV (coro_current));	2476	SvREFCNT_dec (SvRV (coro_current));
2295	SvRV_set (coro_current, SvREFCNT_inc_NN (SvRV (current)));	2477	SvRV_set (coro_current, SvREFCNT_inc_NN (SvRV (current)));
2296		2478
2297	void	2479	void
2298	_set_readyhook (SV *hook)	2480	_set_readyhook (SV *hook)
2299	PROTOTYPE: $	2481	PROTOTYPE: $
2300	CODE:	2482	CODE:
2301	LOCK;
2302	SvREFCNT_dec (coro_readyhook);	2483	SvREFCNT_dec (coro_readyhook);
2303	coro_readyhook = SvOK (hook) ? newSVsv (hook) : 0;	2484	coro_readyhook = SvOK (hook) ? newSVsv (hook) : 0;
2304	UNLOCK;
2305		2485
2306	int	2486	int
2307	prio (Coro::State coro, int newprio = 0)	2487	prio (Coro::State coro, int newprio = 0)
		2488	PROTOTYPE: $;$
2308	ALIAS:	2489	ALIAS:
2309	nice = 1	2490	nice = 1
2310	CODE:	2491	CODE:
2311	{	2492	{
2312	RETVAL = coro->prio;	2493	RETVAL = coro->prio;
…		…
2344	# for async_pool speedup	2525	# for async_pool speedup
2345	void	2526	void
2346	_pool_1 (SV *cb)	2527	_pool_1 (SV *cb)
2347	CODE:	2528	CODE:
2348	{	2529	{
2349	struct coro *coro = SvSTATE (coro_current);
2350	HV hv = (HV )SvRV (coro_current);	2530	HV hv = (HV )SvRV (coro_current);
		2531	struct coro coro = SvSTATE_hv ((SV )hv);
2351	AV *defav = GvAV (PL_defgv);	2532	AV *defav = GvAV (PL_defgv);
2352	SV *invoke = hv_delete (hv, "_invoke", sizeof ("_invoke") - 1, 0);	2533	SV *invoke = hv_delete (hv, "_invoke", sizeof ("_invoke") - 1, 0);
2353	AV *invoke_av;	2534	AV *invoke_av;
2354	int i, len;	2535	int i, len;
2355		2536
…		…
2376	{	2557	{
2377	av_fill (defav, len - 1);	2558	av_fill (defav, len - 1);
2378	for (i = 0; i < len; ++i)	2559	for (i = 0; i < len; ++i)
2379	av_store (defav, i, SvREFCNT_inc_NN (AvARRAY (invoke_av)[i + 1]));	2560	av_store (defav, i, SvREFCNT_inc_NN (AvARRAY (invoke_av)[i + 1]));
2380	}	2561	}
2381
2382	SvREFCNT_dec (invoke);
2383	}	2562	}
2384		2563
2385	void	2564	void
2386	_pool_2 (SV *cb)	2565	_pool_2 (SV *cb)
2387	CODE:	2566	CODE:
2388	{	2567	{
2389	struct coro *coro = SvSTATE (coro_current);	2568	struct coro *coro = SvSTATE_current;
2390		2569
2391	sv_setsv (cb, &PL_sv_undef);	2570	sv_setsv (cb, &PL_sv_undef);
2392		2571
2393	SvREFCNT_dec ((SV )PL_defoutgv); PL_defoutgv = (GV )coro->saved_deffh;	2572	SvREFCNT_dec ((SV )PL_defoutgv); PL_defoutgv = (GV )coro->saved_deffh;
2394	coro->saved_deffh = 0;	2573	coro->saved_deffh = 0;
…		…
2412	api_trace (aTHX_ coro_current, 0);	2591	api_trace (aTHX_ coro_current, 0);
2413		2592
2414	av_push (av_async_pool, newSVsv (coro_current));	2593	av_push (av_async_pool, newSVsv (coro_current));
2415	}	2594	}
2416		2595
2417	#if 0
2418
2419	void
2420	_generator_call (...)
2421	PROTOTYPE: @
2422	PPCODE:
2423	fprintf (stderr, "call %p\n", CvXSUBANY(cv).any_ptr);
2424	xxxx
2425	abort ();
2426
2427	SV *
2428	gensub (SV *sub, ...)
2429	PROTOTYPE: &;@
2430	CODE:
2431	{
2432	struct coro *coro;
2433	MAGIC *mg;
2434	CV *xcv;
2435	CV ncv = (CV )newSV_type (SVt_PVCV);
2436	int i;
2437
2438	CvGV (ncv) = CvGV (cv);
2439	CvFILE (ncv) = CvFILE (cv);
2440
2441	Newz (0, coro, 1, struct coro);
2442	coro->args = newAV ();
2443	coro->flags = CF_NEW;
2444
2445	av_extend (coro->args, items - 1);
2446	for (i = 1; i < items; i++)
2447	av_push (coro->args, newSVsv (ST (i)));
2448
2449	CvISXSUB_on (ncv);
2450	CvXSUBANY (ncv).any_ptr = (void *)coro;
2451
2452	xcv = GvCV (gv_fetchpv ("Coro::_generator_call", 0, SVt_PVCV));
2453
2454	CvXSUB (ncv) = CvXSUB (xcv);
2455	CvANON_on (ncv);
2456
2457	mg = sv_magicext ((SV )ncv, 0, CORO_MAGIC_type_state, &coro_gensub_vtbl, (char )coro, 0);
2458	RETVAL = newRV_noinc ((SV *)ncv);
2459	}
2460	OUTPUT:
2461	RETVAL
2462
2463	#endif
2464
2465		2596
2466	MODULE = Coro::State PACKAGE = Coro::AIO	2597	MODULE = Coro::State PACKAGE = Coro::AIO
2467		2598
2468	void	2599	void
2469	_get_state (SV *self)	2600	_get_state (SV *self)
		2601	PROTOTYPE: $
2470	PPCODE:	2602	PPCODE:
2471	{	2603	{
2472	AV *defav = GvAV (PL_defgv);	2604	AV *defav = GvAV (PL_defgv);
2473	AV *av = newAV ();	2605	AV *av = newAV ();
2474	int i;	2606	int i;
…		…
2517	MODULE = Coro::State PACKAGE = Coro::AnyEvent	2649	MODULE = Coro::State PACKAGE = Coro::AnyEvent
2518		2650
2519	BOOT:	2651	BOOT:
2520	sv_activity = coro_get_sv (aTHX_ "Coro::AnyEvent::ACTIVITY", TRUE);	2652	sv_activity = coro_get_sv (aTHX_ "Coro::AnyEvent::ACTIVITY", TRUE);
2521		2653
2522	SV *	2654	void
2523	_schedule (...)	2655	_schedule (...)
2524	PROTOTYPE: @
2525	CODE:	2656	CODE:
2526	{	2657	{
2527	static int incede;	2658	static int incede;
2528		2659
2529	api_cede_notself (aTHX);	2660	api_cede_notself (aTHX);
…		…
2548	MODULE = Coro::State PACKAGE = PerlIO::cede	2679	MODULE = Coro::State PACKAGE = PerlIO::cede
2549		2680
2550	BOOT:	2681	BOOT:
2551	PerlIO_define_layer (aTHX_ &PerlIO_cede);	2682	PerlIO_define_layer (aTHX_ &PerlIO_cede);
2552		2683
		2684	MODULE = Coro::State PACKAGE = Coro::Semaphore
		2685
		2686	SV *
		2687	new (SV klass, SV count_ = 0)
		2688	CODE:
		2689	{
		2690	/* a semaphore contains a counter IV in $sem->[0] and any waiters after that */
		2691	AV *av = newAV ();
		2692	av_push (av, newSViv (count_ && SvOK (count_) ? SvIV (count_) : 1));
		2693	RETVAL = sv_bless (newRV_noinc ((SV *)av), GvSTASH (CvGV (cv)));
		2694	}
		2695	OUTPUT:
		2696	RETVAL
		2697
		2698	SV *
		2699	count (SV *self)
		2700	CODE:
		2701	RETVAL = newSVsv (AvARRAY ((AV *)SvRV (self))[0]);
		2702	OUTPUT:
		2703	RETVAL
		2704
		2705	void
		2706	up (SV *self, int adjust = 1)
		2707	ALIAS:
		2708	adjust = 1
		2709	CODE:
		2710	coro_semaphore_adjust (aTHX_ (AV *)SvRV (self), ix ? adjust : 1);
		2711
		2712	void
		2713	down (SV *self)
		2714	CODE:
		2715	api_execute_slf (aTHX_ cv, slf_init_semaphore_down, &ST (0), 1);
		2716
		2717	void
		2718	try (SV *self)
		2719	PPCODE:
		2720	{
		2721	AV av = (AV )SvRV (self);
		2722	SV *count_sv = AvARRAY (av)[0];
		2723	IV count = SvIVX (count_sv);
		2724
		2725	if (count > 0)
		2726	{
		2727	--count;
		2728	SvIVX (count_sv) = count;
		2729	XSRETURN_YES;
		2730	}
		2731	else
		2732	XSRETURN_NO;
		2733	}
		2734
		2735	void
		2736	waiters (SV *self)
		2737	CODE:
		2738	{
		2739	AV av = (AV )SvRV (self);
		2740
		2741	if (GIMME_V == G_SCALAR)
		2742	XPUSHs (sv_2mortal (newSVsv (AvARRAY (av)[0])));
		2743	else
		2744	{
		2745	int i;
		2746	EXTEND (SP, AvFILLp (av) + 1 - 1);
		2747	for (i = 1; i <= AvFILLp (av); ++i)
		2748	PUSHs (newSVsv (AvARRAY (av)[i]));
		2749	}
		2750	}
		2751

Diff Legend

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

Comparing Coro/Coro/State.xs (file contents): Revision 1.270 by root, Fri Nov 14 07:22:11 2008 UTC vs. Revision 1.283 by root, Sun Nov 16 11:12:57 2008 UTC

Diff Legend

Comparing Coro/Coro/State.xs (file contents):
Revision 1.270 by root, Fri Nov 14 07:22:11 2008 UTC vs.
Revision 1.283 by root, Sun Nov 16 11:12:57 2008 UTC