ViewVC Help

View File | Revision Log | Show Annotations | Download File

/cvs/Coro/Coro/State.xs

Comparing Coro/Coro/State.xs (file contents):
Revision 1.254 by root, Fri Nov 7 20:27:47 2008 UTC vs.
Revision 1.285 by root, Mon Nov 17 04:17:20 2008 UTC

…		…
46	# define BOOT_PAGESIZE (void)0	46	# define BOOT_PAGESIZE (void)0
47	#endif	47	#endif
48		48
49	#if CORO_USE_VALGRIND	49	#if CORO_USE_VALGRIND
50	# include <valgrind/valgrind.h>	50	# include <valgrind/valgrind.h>
51	# define REGISTER_STACK(cctx,start,end) (cctx)->valgrind_id = VALGRIND_STACK_REGISTER ((start), (end))
52	#else
53	# define REGISTER_STACK(cctx,start,end)
54	#endif	51	#endif
55		52
56	/* the maximum number of idle cctx that will be pooled */	53	/* the maximum number of idle cctx that will be pooled */
57	static int cctx_max_idle = 4;	54	static int cctx_max_idle = 4;
58		55
…		…
119	# define CORO_PREFER_PERL_FUNCTIONS 0	116	# define CORO_PREFER_PERL_FUNCTIONS 0
120	#endif	117	#endif
121		118
122	/* The next macros try to return the current stack pointer, in an as	119	/* The next macros try to return the current stack pointer, in an as
123	* portable way as possible. */	120	* portable way as possible. */
124	#define dSTACKLEVEL volatile char stacklevel	121	#if __GNUC__ >= 4
125	#define STACKLEVEL ((void *)&stacklevel)	122	# define dSTACKLEVEL void *stacklevel = __builtin_frame_address (0)
		123	#else
		124	# define dSTACKLEVEL volatile void *stacklevel = (volatile void *)&stacklevel
		125	#endif
126		126
127	#define IN_DESTRUCT (PL_main_cv == Nullcv)	127	#define IN_DESTRUCT (PL_main_cv == Nullcv)
128		128
129	#if __GNUC__ >= 3	129	#if __GNUC__ >= 3
130	# define attribute(x) __attribute__(x)	130	# define attribute(x) __attribute__(x)
131	# define BARRIER __asm__ __volatile__ ("" : : : "memory")
132	# define expect(expr,value) __builtin_expect ((expr),(value))	131	# define expect(expr,value) __builtin_expect ((expr),(value))
		132	# define INLINE static inline
133	#else	133	#else
134	# define attribute(x)	134	# define attribute(x)
135	# define BARRIER
136	# define expect(expr,value) (expr)	135	# define expect(expr,value) (expr)
		136	# define INLINE static
137	#endif	137	#endif
138		138
139	#define expect_false(expr) expect ((expr) != 0, 0)	139	#define expect_false(expr) expect ((expr) != 0, 0)
140	#define expect_true(expr) expect ((expr) != 0, 1)	140	#define expect_true(expr) expect ((expr) != 0, 1)
141		141
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	static perl_mutex coro_lock;	147	# if CORO_PTHREAD
148	# define LOCK do { MUTEX_LOCK (&coro_lock); } while (0)	148	static void *coro_thx;
149	# define UNLOCK do { MUTEX_UNLOCK (&coro_lock); } while (0)	149	# endif
150	#else
151	# define LOCK (void)0
152	# define UNLOCK (void)0
153	#endif	150	#endif
154		151
155	/* helper storage struct for Coro::AIO */	152	/* helper storage struct for Coro::AIO */
156	struct io_state	153	struct io_state
157	{	154	{
158	AV *res;	155	AV *res;
159	int errorno;	156	int errorno;
160	I32 laststype;	157	I32 laststype; /* U16 in 5.10.0 */
161	int laststatval;	158	int laststatval;
162	Stat_t statcache;	159	Stat_t statcache;
163	};	160	};
164		161
165	static double (*nvtime)(); /* so why doesn't it take void? */	162	static double (*nvtime)(); /* so why doesn't it take void? */
…		…
197	CC_TRACE_LINE = 0x10, /* trace each statement */	194	CC_TRACE_LINE = 0x10, /* trace each statement */
198	CC_TRACE_ALL = CC_TRACE_SUB | CC_TRACE_LINE,	195	CC_TRACE_ALL = CC_TRACE_SUB | CC_TRACE_LINE,
199	};	196	};
200		197
201	/* this is a structure representing a c-level coroutine */	198	/* this is a structure representing a c-level coroutine */
202	typedef struct coro_cctx {	199	typedef struct coro_cctx
		200	{
203	struct coro_cctx *next;	201	struct coro_cctx *next;
204		202
205	/* the stack */	203	/* the stack */
206	void *sptr;	204	void *sptr;
207	size_t ssize;	205	size_t ssize;
…		…
225	CF_NEW = 0x0004, /* has never been switched to */	223	CF_NEW = 0x0004, /* has never been switched to */
226	CF_DESTROYED = 0x0008, /* coroutine data has been freed */	224	CF_DESTROYED = 0x0008, /* coroutine data has been freed */
227	};	225	};
228		226
229	/* 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 */
230	typedef struct {	228	typedef struct
		229	{
231	SV *defsv;	230	SV *defsv;
232	AV *defav;	231	AV *defav;
233	SV *errsv;	232	SV *errsv;
234	SV *irsgv;	233	SV *irsgv;
235	#define VAR(name,type) type name;	234	#define VAR(name,type) type name;
…		…
239		238
240	#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))
241		240
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 */
		247	struct CoroSLF slf_frame; /* saved slf frame */
248	AV *mainstack;	248	AV *mainstack;
249	perl_slots *slot; /* basically the saved sp */	249	perl_slots *slot; /* basically the saved sp */
250		250
251	AV *args; /* data associated with this coroutine (initial args) */	251	AV *args; /* data associated with this coroutine (initial args) */
252	int refcnt; /* coroutines are refcounted, yes */	252	int refcnt; /* coroutines are refcounted, yes */
253	int flags; /* CF_ flags */	253	int flags; /* CF_ flags */
254	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);
255		256
256	/* statistics */	257	/* statistics */
257	int usecount; /* number of transfers to this coro */	258	int usecount; /* number of transfers to this coro */
258		259
259	/* coro process data */	260	/* coro process data */
…		…
267	struct coro *next, *prev;	268	struct coro *next, *prev;
268	};	269	};
269		270
270	typedef struct coro *Coro__State;	271	typedef struct coro *Coro__State;
271	typedef struct coro *Coro__State_or_hashref;	272	typedef struct coro *Coro__State_or_hashref;
		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 */
		277	static struct CoroSLF slf_frame; /* the current slf frame */
		278	static SV *coro_throw;
272		279
273	/** Coro ********************************************************************/	280	/** Coro ********************************************************************/
274		281
275	#define PRIO_MAX 3	282	#define PRIO_MAX 3
276	#define PRIO_HIGH 1	283	#define PRIO_HIGH 1
…		…
280	#define PRIO_MIN -4	287	#define PRIO_MIN -4
281		288
282	/* for Coro.pm */	289	/* for Coro.pm */
283	static SV *coro_current;	290	static SV *coro_current;
284	static SV *coro_readyhook;	291	static SV *coro_readyhook;
285	static AV *coro_ready [PRIO_MAX-PRIO_MIN+1];	292	static AV *coro_ready [PRIO_MAX - PRIO_MIN + 1];
286	static int coro_nready;
287	static struct coro *coro_first;	293	static struct coro *coro_first;
		294	#define coro_nready coroapi.nready
288		295
289	/** lowlevel stuff **********************************************************/	296	/** lowlevel stuff **********************************************************/
290		297
291	static SV *	298	static SV *
292	coro_get_sv (pTHX_ const char *name, int create)	299	coro_get_sv (pTHX_ const char *name, int create)
…		…
385	static MGVTBL coro_cv_vtbl = {	392	static MGVTBL coro_cv_vtbl = {
386	0, 0, 0, 0,	393	0, 0, 0, 0,
387	coro_cv_free	394	coro_cv_free
388	};	395	};
389		396
390	#define CORO_MAGIC(sv, type) \	397	#define CORO_MAGIC(sv, type) \
391	SvMAGIC (sv) \	398	expect_true (SvMAGIC (sv)) \
392	? SvMAGIC (sv)->mg_type == type \	399	? expect_true (SvMAGIC (sv)->mg_type == type) \
393	? SvMAGIC (sv) \	400	? SvMAGIC (sv) \
394	: mg_find (sv, type) \	401	: mg_find (sv, type) \
395	: 0	402	: 0
396		403
397	#define CORO_MAGIC_cv(cv) CORO_MAGIC (((SV *)(cv)), CORO_MAGIC_type_cv)	404	#define CORO_MAGIC_cv(cv) CORO_MAGIC (((SV *)(cv)), CORO_MAGIC_type_cv)
398	#define CORO_MAGIC_state(sv) CORO_MAGIC (((SV *)(sv)), CORO_MAGIC_type_state)	405	#define CORO_MAGIC_state(sv) CORO_MAGIC (((SV *)(sv)), CORO_MAGIC_type_state)
399		406
400	static struct coro *	407	INLINE struct coro *
401	SvSTATE_ (pTHX_ SV *coro)	408	SvSTATE_ (pTHX_ SV *coro)
402	{	409	{
403	HV *stash;	410	HV *stash;
404	MAGIC *mg;	411	MAGIC *mg;
405		412
…		…
420	mg = CORO_MAGIC_state (coro);	427	mg = CORO_MAGIC_state (coro);
421	return (struct coro *)mg->mg_ptr;	428	return (struct coro *)mg->mg_ptr;
422	}	429	}
423		430
424	#define SvSTATE(sv) SvSTATE_ (aTHX_ (sv))	431	#define SvSTATE(sv) SvSTATE_ (aTHX_ (sv))
		432
		433	/* fastert than SvSTATE, but expects a coroutine hv */
		434	INLINE struct coro *
		435	SvSTATE_hv (SV *sv)
		436	{
		437	MAGIC *mg = expect_true (SvMAGIC (sv)->mg_type == CORO_MAGIC_type_state)
		438	? SvMAGIC (sv)
		439	: mg_find (sv, CORO_MAGIC_type_state);
		440
		441	return (struct coro *)mg->mg_ptr;
		442	}
		443
		444	#define SvSTATE_current SvSTATE_hv (SvRV (coro_current))
425		445
426	/* the next two functions merely cache the padlists */	446	/* the next two functions merely cache the padlists */
427	static void	447	static void
428	get_padlist (pTHX_ CV *cv)	448	get_padlist (pTHX_ CV *cv)
429	{	449	{
…		…
496	CvPADLIST (cv) = (AV *)POPs;	516	CvPADLIST (cv) = (AV *)POPs;
497	}	517	}
498		518
499	PUTBACK;	519	PUTBACK;
500	}	520	}
		521
		522	slf_frame = c->slf_frame;
		523	coro_throw = c->throw;
501	}	524	}
502		525
503	static void	526	static void
504	save_perl (pTHX_ Coro__State c)	527	save_perl (pTHX_ Coro__State c)
505	{	528	{
		529	c->throw = coro_throw;
		530	c->slf_frame = slf_frame;
		531
506	{	532	{
507	dSP;	533	dSP;
508	I32 cxix = cxstack_ix;	534	I32 cxix = cxstack_ix;
509	PERL_CONTEXT *ccstk = cxstack;	535	PERL_CONTEXT *ccstk = cxstack;
510	PERL_SI *top_si = PL_curstackinfo;	536	PERL_SI *top_si = PL_curstackinfo;
…		…
577	#undef VAR	603	#undef VAR
578	}	604	}
579	}	605	}
580		606
581	/*	607	/*
582	* allocate various perl stacks. This is an exact copy	608	* allocate various perl stacks. This is almost an exact copy
583	* of perl.c:init_stacks, except that it uses less memory	609	* of perl.c:init_stacks, except that it uses less memory
584	* on the (sometimes correct) assumption that coroutines do	610	* on the (sometimes correct) assumption that coroutines do
585	* not usually need a lot of stackspace.	611	* not usually need a lot of stackspace.
586	*/	612	*/
587	#if CORO_PREFER_PERL_FUNCTIONS	613	#if CORO_PREFER_PERL_FUNCTIONS
…		…
789		815
790	return orig_sigelem_set ? orig_sigelem_set (aTHX_ sv, mg) : 0;	816	return orig_sigelem_set ? orig_sigelem_set (aTHX_ sv, mg) : 0;
791	}	817	}
792		818
793	static void	819	static void
		820	prepare_nop (pTHX_ struct coro_transfer_args *ta)
		821	{
		822	/* kind of mega-hacky, but works */
		823	ta->next = ta->prev = (struct coro *)ta;
		824	}
		825
		826	static int
		827	slf_check_nop (pTHX_ struct CoroSLF *frame)
		828	{
		829	return 0;
		830	}
		831
		832	static void NOINLINE /* noinline to keep it out of the transfer fast path */
794	coro_setup (pTHX_ struct coro *coro)	833	coro_setup (pTHX_ struct coro *coro)
795	{	834	{
796	/*	835	/*
797	* emulate part of the perl startup here.	836	* emulate part of the perl startup here.
798	*/	837	*/
…		…
822	PL_rs = newSVsv (GvSV (irsgv));	861	PL_rs = newSVsv (GvSV (irsgv));
823	PL_defoutgv = (GV *)SvREFCNT_inc_NN (stdoutgv);	862	PL_defoutgv = (GV *)SvREFCNT_inc_NN (stdoutgv);
824		863
825	{	864	{
826	dSP;	865	dSP;
827	LOGOP myop;	866	UNOP myop;
828		867
829	Zero (&myop, 1, LOGOP);	868	Zero (&myop, 1, UNOP);
830	myop.op_next = Nullop;	869	myop.op_next = Nullop;
831	myop.op_flags = OPf_WANT_VOID;	870	myop.op_flags = OPf_WANT_VOID;
832		871
833	PUSHMARK (SP);	872	PUSHMARK (SP);
834	XPUSHs (sv_2mortal (av_shift (GvAV (PL_defgv))));	873	XPUSHs (sv_2mortal (av_shift (GvAV (PL_defgv))));
…		…
837	PL_op = PL_ppaddr[OP_ENTERSUB](aTHX);	876	PL_op = PL_ppaddr[OP_ENTERSUB](aTHX);
838	SPAGAIN;	877	SPAGAIN;
839	}	878	}
840		879
841	/* this newly created coroutine might be run on an existing cctx which most	880	/* this newly created coroutine might be run on an existing cctx which most
842	* likely was suspended in set_stacklevel, called from entersub.	881	* likely was suspended in pp_slf, so we have to emulate entering pp_slf here.
843	* set_stacklevl doesn't do anything on return, but entersub does LEAVE,
844	* so we ENTER here for symmetry
845	*/	882	*/
846	ENTER;	883	slf_frame.prepare = prepare_nop; /* provide a nop function for an eventual pp_slf */
		884	slf_frame.check = slf_check_nop; /* signal pp_slf to not repeat */
		885
		886	coro_throw = coro->throw;
847	}	887	}
848		888
849	static void	889	static void
850	coro_destruct (pTHX_ struct coro *coro)	890	coro_destruct (pTHX_ struct coro *coro)
851	{	891	{
…		…
875		915
876	SvREFCNT_dec (PL_diehook);	916	SvREFCNT_dec (PL_diehook);
877	SvREFCNT_dec (PL_warnhook);	917	SvREFCNT_dec (PL_warnhook);
878		918
879	SvREFCNT_dec (coro->saved_deffh);	919	SvREFCNT_dec (coro->saved_deffh);
880	SvREFCNT_dec (coro->throw);	920	SvREFCNT_dec (coro_throw);
881		921
882	coro_destruct_stacks (aTHX);	922	coro_destruct_stacks (aTHX);
883	}	923	}
884		924
885	static void	925	INLINE void
886	free_coro_mortal (pTHX)	926	free_coro_mortal (pTHX)
887	{	927	{
888	if (expect_true (coro_mortal))	928	if (expect_true (coro_mortal))
889	{	929	{
890	SvREFCNT_dec (coro_mortal);	930	SvREFCNT_dec (coro_mortal);
…		…
895	static int	935	static int
896	runops_trace (pTHX)	936	runops_trace (pTHX)
897	{	937	{
898	COP *oldcop = 0;	938	COP *oldcop = 0;
899	int oldcxix = -2;	939	int oldcxix = -2;
900	struct coro *coro = SvSTATE (coro_current); /* trace cctx is tied to specific coro */	940	struct coro *coro = SvSTATE_current; /* trace cctx is tied to specific coro */
901	coro_cctx *cctx = coro->cctx;	941	coro_cctx *cctx = coro->cctx;
902		942
903	while ((PL_op = CALL_FPTR (PL_op->op_ppaddr) (aTHX)))	943	while ((PL_op = CALL_FPTR (PL_op->op_ppaddr) (aTHX)))
904	{	944	{
905	PERL_ASYNC_CHECK ();	945	PERL_ASYNC_CHECK ();
…		…
1014		1054
1015	TAINT_NOT;	1055	TAINT_NOT;
1016	return 0;	1056	return 0;
1017	}	1057	}
1018		1058
		1059	static void
		1060	prepare_set_stacklevel (struct coro_transfer_args *ta, struct coro_cctx *cctx)
		1061	{
		1062	ta->prev = (struct coro *)cctx;
		1063	ta->next = 0;
		1064	}
		1065
1019	/* inject a fake call to Coro::State::_cctx_init into the execution */	1066	/* inject a fake call to Coro::State::_cctx_init into the execution */
1020	/* _cctx_init should be careful, as it could be called at almost any time */	1067	/* _cctx_init should be careful, as it could be called at almost any time */
1021	/* during execution of a perl program */	1068	/* during execution of a perl program */
		1069	/* also initialises PL_top_env */
1022	static void NOINLINE	1070	static void NOINLINE
1023	cctx_prepare (pTHX_ coro_cctx *cctx)	1071	cctx_prepare (pTHX_ coro_cctx *cctx)
1024	{	1072	{
1025	dSP;	1073	dSP;
1026	LOGOP myop;	1074	UNOP myop;
1027		1075
1028	PL_top_env = &PL_start_env;	1076	PL_top_env = &PL_start_env;
1029		1077
1030	if (cctx->flags & CC_TRACE)	1078	if (cctx->flags & CC_TRACE)
1031	PL_runops = runops_trace;	1079	PL_runops = runops_trace;
1032		1080
1033	Zero (&myop, 1, LOGOP);	1081	Zero (&myop, 1, UNOP);
1034	myop.op_next = PL_op;	1082	myop.op_next = PL_op;
1035	myop.op_flags = OPf_WANT_VOID | OPf_STACKED;	1083	myop.op_flags = OPf_WANT_VOID | OPf_STACKED;
1036		1084
1037	PUSHMARK (SP);	1085	PUSHMARK (SP);
1038	EXTEND (SP, 2);	1086	EXTEND (SP, 2);
1039	PUSHs (sv_2mortal (newSViv (PTR2IV (cctx))));	1087	PUSHs (sv_2mortal (newSViv ((IV)cctx)));
1040	PUSHs ((SV *)get_cv ("Coro::State::_cctx_init", FALSE));	1088	PUSHs ((SV *)get_cv ("Coro::State::_cctx_init", FALSE));
1041	PUTBACK;	1089	PUTBACK;
1042	PL_op = (OP *)&myop;	1090	PL_op = (OP *)&myop;
1043	PL_op = PL_ppaddr[OP_ENTERSUB](aTHX);	1091	PL_op = PL_ppaddr[OP_ENTERSUB](aTHX);
1044	SPAGAIN;	1092	SPAGAIN;
1045	}	1093	}
1046		1094
		1095	/* the tail of transfer: execute stuff we can only do after a transfer */
		1096	INLINE void
		1097	transfer_tail (pTHX)
		1098	{
		1099	free_coro_mortal (aTHX);
		1100	}
		1101
1047	/*	1102	/*
1048	* this is a _very_ stripped down perl interpreter ;)	1103	* this is a _very_ stripped down perl interpreter ;)
1049	*/	1104	*/
1050	static void	1105	static void
1051	cctx_run (void *arg)	1106	cctx_run (void *arg)
1052	{	1107	{
		1108	#ifdef USE_ITHREADS
		1109	# if CORO_PTHREAD
		1110	PERL_SET_CONTEXT (coro_thx);
		1111	# endif
		1112	#endif
		1113	{
1053	dTHX;	1114	dTHX;
1054		1115
1055	/* cctx_run is the alternative tail of transfer(), so unlock here. */	1116	/* normally we would need to skip the entersub here */
1056	UNLOCK;	1117	/* not doing so will re-execute it, which is exactly what we want */
1057
1058	/* we now skip the entersub that lead to transfer() */
1059	PL_op = PL_op->op_next;	1118	/* PL_nop = PL_nop->op_next */
1060		1119
1061	/* inject a fake subroutine call to cctx_init */	1120	/* inject a fake subroutine call to cctx_init */
1062	cctx_prepare (aTHX_ (coro_cctx *)arg);	1121	cctx_prepare (aTHX_ (coro_cctx *)arg);
1063		1122
		1123	/* cctx_run is the alternative tail of transfer() */
		1124	transfer_tail (aTHX);
		1125
1064	/* somebody or something will hit me for both perl_run and PL_restartop */	1126	/* somebody or something will hit me for both perl_run and PL_restartop */
1065	PL_restartop = PL_op;	1127	PL_restartop = PL_op;
1066	perl_run (PL_curinterp);	1128	perl_run (PL_curinterp);
1067		1129
1068	/*	1130	/*
1069	* If perl-run returns we assume exit() was being called or the coro	1131	* If perl-run returns we assume exit() was being called or the coro
1070	* fell off the end, which seems to be the only valid (non-bug)	1132	* fell off the end, which seems to be the only valid (non-bug)
1071	* reason for perl_run to return. We try to exit by jumping to the	1133	* reason for perl_run to return. We try to exit by jumping to the
1072	* bootstrap-time "top" top_env, as we cannot restore the "main"	1134	* bootstrap-time "top" top_env, as we cannot restore the "main"
1073	* coroutine as Coro has no such concept	1135	* coroutine as Coro has no such concept
1074	*/	1136	*/
1075	PL_top_env = main_top_env;	1137	PL_top_env = main_top_env;
1076	JMPENV_JUMP (2); /* I do not feel well about the hardcoded 2 at all */	1138	JMPENV_JUMP (2); /* I do not feel well about the hardcoded 2 at all */
		1139	}
1077	}	1140	}
1078		1141
1079	static coro_cctx *	1142	static coro_cctx *
1080	cctx_new ()	1143	cctx_new ()
1081	{	1144	{
1082	coro_cctx *cctx;	1145	coro_cctx *cctx;
		1146
		1147	++cctx_count;
		1148	New (0, cctx, 1, coro_cctx);
		1149
		1150	cctx->gen = cctx_gen;
		1151	cctx->flags = 0;
		1152	cctx->idle_sp = 0; /* can be accessed by transfer between cctx_run and set_stacklevel, on throw */
		1153
		1154	return cctx;
		1155	}
		1156
		1157	/* create a new cctx only suitable as source */
		1158	static coro_cctx *
		1159	cctx_new_empty ()
		1160	{
		1161	coro_cctx *cctx = cctx_new ();
		1162
		1163	cctx->sptr = 0;
		1164	coro_create (&cctx->cctx, 0, 0, 0, 0);
		1165
		1166	return cctx;
		1167	}
		1168
		1169	/* create a new cctx suitable as destination/running a perl interpreter */
		1170	static coro_cctx *
		1171	cctx_new_run ()
		1172	{
		1173	coro_cctx *cctx = cctx_new ();
1083	void *stack_start;	1174	void *stack_start;
1084	size_t stack_size;	1175	size_t stack_size;
1085
1086	++cctx_count;
1087	Newz (0, cctx, 1, coro_cctx);
1088
1089	cctx->gen = cctx_gen;
1090		1176
1091	#if HAVE_MMAP	1177	#if HAVE_MMAP
1092	cctx->ssize = ((cctx_stacksize * sizeof (long) + PAGESIZE - 1) / PAGESIZE + CORO_STACKGUARD) * PAGESIZE;	1178	cctx->ssize = ((cctx_stacksize * sizeof (long) + PAGESIZE - 1) / PAGESIZE + CORO_STACKGUARD) * PAGESIZE;
1093	/* mmap supposedly does allocate-on-write for us */	1179	/* mmap supposedly does allocate-on-write for us */
1094	cctx->sptr = mmap (0, cctx->ssize, PROT_EXEC|PROT_READ|PROT_WRITE, MAP_PRIVATE|MAP_ANONYMOUS, 0, 0);	1180	cctx->sptr = mmap (0, cctx->ssize, PROT_EXEC|PROT_READ|PROT_WRITE, MAP_PRIVATE|MAP_ANONYMOUS, 0, 0);
1095		1181
1096	if (cctx->sptr != (void *)-1)	1182	if (cctx->sptr != (void *)-1)
1097	{	1183	{
1098	# if CORO_STACKGUARD	1184	#if CORO_STACKGUARD
1099	mprotect (cctx->sptr, CORO_STACKGUARD * PAGESIZE, PROT_NONE);	1185	mprotect (cctx->sptr, CORO_STACKGUARD * PAGESIZE, PROT_NONE);
1100	# endif	1186	#endif
1101	stack_start = CORO_STACKGUARD * PAGESIZE + (char *)cctx->sptr;	1187	stack_start = (char *)cctx->sptr + CORO_STACKGUARD * PAGESIZE;
1102	stack_size = cctx->ssize - CORO_STACKGUARD * PAGESIZE;	1188	stack_size = cctx->ssize - CORO_STACKGUARD * PAGESIZE;
1103	cctx->flags |= CC_MAPPED;	1189	cctx->flags |= CC_MAPPED;
1104	}	1190	}
1105	else	1191	else
1106	#endif	1192	#endif
1107	{	1193	{
1108	cctx->ssize = cctx_stacksize * (long)sizeof (long);	1194	cctx->ssize = cctx_stacksize * (long)sizeof (long);
1109	New (0, cctx->sptr, cctx_stacksize, long);	1195	New (0, cctx->sptr, cctx_stacksize, long);
1110		1196
1111	if (!cctx->sptr)	1197	if (!cctx->sptr)
1112	{	1198	{
1113	perror ("FATAL: unable to allocate stack for coroutine");	1199	perror ("FATAL: unable to allocate stack for coroutine, exiting.");
1114	_exit (EXIT_FAILURE);	1200	_exit (EXIT_FAILURE);
1115	}	1201	}
1116		1202
1117	stack_start = cctx->sptr;	1203	stack_start = cctx->sptr;
1118	stack_size = cctx->ssize;	1204	stack_size = cctx->ssize;
1119	}	1205	}
1120		1206
1121	REGISTER_STACK (cctx, (char *)stack_start, (char *)stack_start + stack_size);	1207	#if CORO_USE_VALGRIND
		1208	cctx->valgrind_id = VALGRIND_STACK_REGISTER ((char *)stack_start, (char *)stack_start + stack_size);
		1209	#endif
		1210
1122	coro_create (&cctx->cctx, cctx_run, (void *)cctx, stack_start, stack_size);	1211	coro_create (&cctx->cctx, cctx_run, (void *)cctx, stack_start, stack_size);
1123		1212
1124	return cctx;	1213	return cctx;
1125	}	1214	}
1126		1215
…		…
1134	coro_destroy (&cctx->cctx);	1223	coro_destroy (&cctx->cctx);
1135		1224
1136	/* coro_transfer creates new, empty cctx's */	1225	/* coro_transfer creates new, empty cctx's */
1137	if (cctx->sptr)	1226	if (cctx->sptr)
1138	{	1227	{
1139	#if CORO_USE_VALGRIND	1228	#if CORO_USE_VALGRIND
1140	VALGRIND_STACK_DEREGISTER (cctx->valgrind_id);	1229	VALGRIND_STACK_DEREGISTER (cctx->valgrind_id);
1141	#endif	1230	#endif
1142		1231
1143	#if HAVE_MMAP	1232	#if HAVE_MMAP
1144	if (cctx->flags & CC_MAPPED)	1233	if (cctx->flags & CC_MAPPED)
1145	munmap (cctx->sptr, cctx->ssize);	1234	munmap (cctx->sptr, cctx->ssize);
1146	else	1235	else
…		…
1167	return cctx;	1256	return cctx;
1168		1257
1169	cctx_destroy (cctx);	1258	cctx_destroy (cctx);
1170	}	1259	}
1171		1260
1172	return cctx_new ();	1261	return cctx_new_run ();
1173	}	1262	}
1174		1263
1175	static void	1264	static void
1176	cctx_put (coro_cctx *cctx)	1265	cctx_put (coro_cctx *cctx)
1177	{	1266	{
1178	assert (("cctx_put called on non-initialised cctx", cctx->sptr));	1267	assert (("FATAL: cctx_put called on non-initialised cctx in Coro (please report)", cctx->sptr));
1179		1268
1180	/* free another cctx if overlimit */	1269	/* free another cctx if overlimit */
1181	if (expect_false (cctx_idle >= cctx_max_idle))	1270	if (expect_false (cctx_idle >= cctx_max_idle))
1182	{	1271	{
1183	coro_cctx *first = cctx_first;	1272	coro_cctx *first = cctx_first;
…		…
1195	/** coroutine switching *****************************************************/	1284	/** coroutine switching *****************************************************/
1196		1285
1197	static void	1286	static void
1198	transfer_check (pTHX_ struct coro *prev, struct coro *next)	1287	transfer_check (pTHX_ struct coro *prev, struct coro *next)
1199	{	1288	{
		1289	/* TODO: throwing up here is considered harmful */
		1290
1200	if (expect_true (prev != next))	1291	if (expect_true (prev != next))
1201	{	1292	{
1202	if (expect_false (!(prev->flags & (CF_RUNNING | CF_NEW))))	1293	if (expect_false (!(prev->flags & (CF_RUNNING | CF_NEW))))
1203	croak ("Coro::State::transfer called with non-running/new prev Coro::State, but can only transfer from running or new states");	1294	croak ("Coro::State::transfer called with non-running/new prev Coro::State, but can only transfer from running or new states,");
1204		1295
1205	if (expect_false (next->flags & CF_RUNNING))	1296	if (expect_false (next->flags & CF_RUNNING))
1206	croak ("Coro::State::transfer called with running next Coro::State, but can only transfer to inactive states");	1297	croak ("Coro::State::transfer called with running next Coro::State, but can only transfer to inactive states,");
1207		1298
1208	if (expect_false (next->flags & CF_DESTROYED))	1299	if (expect_false (next->flags & CF_DESTROYED))
1209	croak ("Coro::State::transfer called with destroyed next Coro::State, but can only transfer to inactive states");	1300	croak ("Coro::State::transfer called with destroyed next Coro::State, but can only transfer to inactive states,");
1210		1301
1211	#if !PERL_VERSION_ATLEAST (5,10,0)	1302	#if !PERL_VERSION_ATLEAST (5,10,0)
1212	if (expect_false (PL_lex_state != LEX_NOTPARSING))	1303	if (expect_false (PL_lex_state != LEX_NOTPARSING))
1213	croak ("Coro::State::transfer called while parsing, but this is not supported in your perl version");	1304	croak ("Coro::State::transfer called while parsing, but this is not supported in your perl version,");
1214	#endif	1305	#endif
1215	}	1306	}
1216	}	1307	}
1217		1308
1218	/* always use the TRANSFER macro */	1309	/* always use the TRANSFER macro */
1219	static void NOINLINE	1310	static void NOINLINE /* noinline so we have a fixed stackframe */
1220	transfer (pTHX_ struct coro *prev, struct coro *next, int force_cctx)	1311	transfer (pTHX_ struct coro *prev, struct coro *next, int force_cctx)
1221	{	1312	{
1222	dSTACKLEVEL;	1313	dSTACKLEVEL;
1223		1314
1224	/* sometimes transfer is only called to set idle_sp */	1315	/* sometimes transfer is only called to set idle_sp */
1225	if (expect_false (!next))	1316	if (expect_false (!next))
1226	{	1317	{
1227	((coro_cctx *)prev)->idle_sp = STACKLEVEL;	1318	((coro_cctx *)prev)->idle_sp = (void *)stacklevel;
1228	assert (((coro_cctx *)prev)->idle_te = PL_top_env); /* just for the side-effect when asserts are enabled */	1319	assert (((coro_cctx *)prev)->idle_te = PL_top_env); /* just for the side-effect when asserts are enabled */
1229	}	1320	}
1230	else if (expect_true (prev != next))	1321	else if (expect_true (prev != next))
1231	{	1322	{
1232	static volatile int has_throw;
1233	coro_cctx *prev__cctx;	1323	coro_cctx *prev__cctx;
1234		1324
1235	if (expect_false (prev->flags & CF_NEW))	1325	if (expect_false (prev->flags & CF_NEW))
1236	{	1326	{
1237	/* create a new empty/source context */	1327	/* create a new empty/source context */
1238	++cctx_count;	1328	prev->cctx = cctx_new_empty ();
1239	New (0, prev->cctx, 1, coro_cctx);
1240	prev->cctx->sptr = 0;
1241	coro_create (&prev->cctx->cctx, 0, 0, 0, 0);
1242
1243	prev->flags &= ~CF_NEW;	1329	prev->flags &= ~CF_NEW;
1244	prev->flags |= CF_RUNNING;	1330	prev->flags |= CF_RUNNING;
1245	}	1331	}
1246		1332
1247	prev->flags &= ~CF_RUNNING;	1333	prev->flags &= ~CF_RUNNING;
1248	next->flags |= CF_RUNNING;	1334	next->flags |= CF_RUNNING;
1249
1250	LOCK;
1251		1335
1252	/* first get rid of the old state */	1336	/* first get rid of the old state */
1253	save_perl (aTHX_ prev);	1337	save_perl (aTHX_ prev);
1254		1338
1255	if (expect_false (next->flags & CF_NEW))	1339	if (expect_false (next->flags & CF_NEW))
…		…
1262	else	1346	else
1263	load_perl (aTHX_ next);	1347	load_perl (aTHX_ next);
1264		1348
1265	prev__cctx = prev->cctx;	1349	prev__cctx = prev->cctx;
1266		1350
1267	/* possibly "free" the cctx */	1351	/* possibly untie and reuse the cctx */
1268	if (expect_true (	1352	if (expect_true (
1269	prev__cctx->idle_sp == STACKLEVEL	1353	prev__cctx->idle_sp == (void *)stacklevel
1270	&& !(prev__cctx->flags & CC_TRACE)	1354	&& !(prev__cctx->flags & CC_TRACE)
1271	&& !force_cctx	1355	&& !force_cctx
1272	))	1356	))
1273	{	1357	{
1274	/* I assume that STACKLEVEL is a stronger indicator than PL_top_env changes */	1358	/* I assume that stacklevel is a stronger indicator than PL_top_env changes */
1275	assert (("ERROR: current top_env must equal previous top_env", PL_top_env == prev__cctx->idle_te));	1359	assert (("FATAL: current top_env must equal previous top_env in Coro (please report)", PL_top_env == prev__cctx->idle_te));
1276		1360
1277	prev->cctx = 0;	1361	prev->cctx = 0;
1278		1362
1279	/* if the cctx is about to be destroyed we need to make sure we won't see it in cctx_get */	1363	/* if the cctx is about to be destroyed we need to make sure we won't see it in cctx_get */
1280	/* without this the next cctx_get might destroy the prev__cctx while still in use */	1364	/* without this the next cctx_get might destroy the prev__cctx while still in use */
…		…
1287		1371
1288	++next->usecount;	1372	++next->usecount;
1289		1373
1290	if (expect_true (!next->cctx))	1374	if (expect_true (!next->cctx))
1291	next->cctx = cctx_get (aTHX);	1375	next->cctx = cctx_get (aTHX);
1292
1293	has_throw = !!next->throw;
1294		1376
1295	if (expect_false (prev__cctx != next->cctx))	1377	if (expect_false (prev__cctx != next->cctx))
1296	{	1378	{
1297	prev__cctx->top_env = PL_top_env;	1379	prev__cctx->top_env = PL_top_env;
1298	PL_top_env = next->cctx->top_env;	1380	PL_top_env = next->cctx->top_env;
1299	coro_transfer (&prev__cctx->cctx, &next->cctx->cctx);	1381	coro_transfer (&prev__cctx->cctx, &next->cctx->cctx);
1300	}	1382	}
1301		1383
1302	free_coro_mortal (aTHX);	1384	transfer_tail (aTHX);
1303	UNLOCK;
1304
1305	if (expect_false (has_throw))
1306	{
1307	struct coro *coro = SvSTATE (coro_current);
1308
1309	if (coro->throw)
1310	{
1311	SV *exception = coro->throw;
1312	coro->throw = 0;
1313	sv_setsv (ERRSV, exception);
1314	croak (0);
1315	}
1316	}
1317	}	1385	}
1318	}	1386	}
1319
1320	struct transfer_args
1321	{
1322	struct coro *prev, *next;
1323	};
1324		1387
1325	#define TRANSFER(ta, force_cctx) transfer (aTHX_ (ta).prev, (ta).next, (force_cctx))	1388	#define TRANSFER(ta, force_cctx) transfer (aTHX_ (ta).prev, (ta).next, (force_cctx))
1326	#define TRANSFER_CHECK(ta) transfer_check (aTHX_ (ta).prev, (ta).next)	1389	#define TRANSFER_CHECK(ta) transfer_check (aTHX_ (ta).prev, (ta).next)
1327		1390
1328	/** high level stuff ********************************************************/	1391	/** high level stuff ********************************************************/
…		…
1330	static int	1393	static int
1331	coro_state_destroy (pTHX_ struct coro *coro)	1394	coro_state_destroy (pTHX_ struct coro *coro)
1332	{	1395	{
1333	if (coro->flags & CF_DESTROYED)	1396	if (coro->flags & CF_DESTROYED)
1334	return 0;	1397	return 0;
		1398
		1399	if (coro->on_destroy)
		1400	coro->on_destroy (aTHX_ coro);
1335		1401
1336	coro->flags |= CF_DESTROYED;	1402	coro->flags |= CF_DESTROYED;
1337		1403
1338	if (coro->flags & CF_READY)	1404	if (coro->flags & CF_READY)
1339	{	1405	{
1340	/* reduce nready, as destroying a ready coro effectively unreadies it */	1406	/* reduce nready, as destroying a ready coro effectively unreadies it */
1341	/* alternative: look through all ready queues and remove the coro */	1407	/* alternative: look through all ready queues and remove the coro */
1342	LOCK;
1343	--coro_nready;	1408	--coro_nready;
1344	UNLOCK;
1345	}	1409	}
1346	else	1410	else
1347	coro->flags |= CF_READY; /* make sure it is NOT put into the readyqueue */	1411	coro->flags |= CF_READY; /* make sure it is NOT put into the readyqueue */
1348		1412
1349	if (coro->mainstack && coro->mainstack != main_mainstack)	1413	if (coro->mainstack && coro->mainstack != main_mainstack)
1350	{	1414	{
1351	struct coro temp;	1415	struct coro temp;
1352		1416
1353	if (coro->flags & CF_RUNNING)	1417	assert (("FATAL: tried to destroy currently running coroutine (please report)", !(coro->flags & CF_RUNNING)));
1354	croak ("FATAL: tried to destroy currently running coroutine");
1355		1418
1356	save_perl (aTHX_ &temp);	1419	save_perl (aTHX_ &temp);
1357	load_perl (aTHX_ coro);	1420	load_perl (aTHX_ coro);
1358		1421
1359	coro_destruct (aTHX_ coro);	1422	coro_destruct (aTHX_ coro);
…		…
1410	# define MGf_DUP 0	1473	# define MGf_DUP 0
1411	#endif	1474	#endif
1412	};	1475	};
1413		1476
1414	static void	1477	static void
1415	prepare_transfer (pTHX_ struct transfer_args *ta, SV *prev_sv, SV *next_sv)	1478	prepare_transfer (pTHX_ struct coro_transfer_args *ta, SV *prev_sv, SV *next_sv)
1416	{	1479	{
1417	ta->prev = SvSTATE (prev_sv);	1480	ta->prev = SvSTATE (prev_sv);
1418	ta->next = SvSTATE (next_sv);	1481	ta->next = SvSTATE (next_sv);
1419	TRANSFER_CHECK (*ta);	1482	TRANSFER_CHECK (*ta);
1420	}	1483	}
1421		1484
1422	static void	1485	static void
1423	api_transfer (SV *prev_sv, SV *next_sv)	1486	api_transfer (pTHX_ SV *prev_sv, SV *next_sv)
1424	{	1487	{
1425	dTHX;
1426	struct transfer_args ta;	1488	struct coro_transfer_args ta;
1427		1489
1428	prepare_transfer (aTHX_ &ta, prev_sv, next_sv);	1490	prepare_transfer (aTHX_ &ta, prev_sv, next_sv);
1429	TRANSFER (ta, 1);	1491	TRANSFER (ta, 1);
1430	}	1492	}
1431		1493
1432	/** Coro ********************************************************************/	1494	/** Coro ********************************************************************/
1433		1495
1434	static void	1496	INLINE void
1435	coro_enq (pTHX_ SV *coro_sv)	1497	coro_enq (pTHX_ struct coro *coro)
1436	{	1498	{
1437	av_push (coro_ready [SvSTATE (coro_sv)->prio - PRIO_MIN], coro_sv);	1499	av_push (coro_ready [coro->prio - PRIO_MIN], SvREFCNT_inc_NN (coro->hv));
1438	}	1500	}
1439		1501
1440	static SV *	1502	INLINE SV *
1441	coro_deq (pTHX)	1503	coro_deq (pTHX)
1442	{	1504	{
1443	int prio;	1505	int prio;
1444		1506
1445	for (prio = PRIO_MAX - PRIO_MIN + 1; --prio >= 0; )	1507	for (prio = PRIO_MAX - PRIO_MIN + 1; --prio >= 0; )
…		…
1448		1510
1449	return 0;	1511	return 0;
1450	}	1512	}
1451		1513
1452	static int	1514	static int
1453	api_ready (SV *coro_sv)	1515	api_ready (pTHX_ SV *coro_sv)
1454	{	1516	{
1455	dTHX;
1456	struct coro *coro;	1517	struct coro *coro;
1457	SV *sv_hook;	1518	SV *sv_hook;
1458	void (*xs_hook)(void);	1519	void (*xs_hook)(void);
1459		1520
1460	if (SvROK (coro_sv))	1521	if (SvROK (coro_sv))
…		…
1465	if (coro->flags & CF_READY)	1526	if (coro->flags & CF_READY)
1466	return 0;	1527	return 0;
1467		1528
1468	coro->flags |= CF_READY;	1529	coro->flags |= CF_READY;
1469		1530
1470	LOCK;
1471
1472	sv_hook = coro_nready ? 0 : coro_readyhook;	1531	sv_hook = coro_nready ? 0 : coro_readyhook;
1473	xs_hook = coro_nready ? 0 : coroapi.readyhook;	1532	xs_hook = coro_nready ? 0 : coroapi.readyhook;
1474		1533
1475	coro_enq (aTHX_ SvREFCNT_inc_NN (coro_sv));	1534	coro_enq (aTHX_ coro);
1476	++coro_nready;	1535	++coro_nready;
1477		1536
1478	UNLOCK;
1479
1480	if (sv_hook)	1537	if (sv_hook)
1481	{	1538	{
1482	dSP;	1539	dSP;
1483		1540
1484	ENTER;	1541	ENTER;
…		…
1498		1555
1499	return 1;	1556	return 1;
1500	}	1557	}
1501		1558
1502	static int	1559	static int
1503	api_is_ready (SV *coro_sv)	1560	api_is_ready (pTHX_ SV *coro_sv)
1504	{	1561	{
1505	dTHX;
1506	return !!(SvSTATE (coro_sv)->flags & CF_READY);	1562	return !!(SvSTATE (coro_sv)->flags & CF_READY);
1507	}	1563	}
1508		1564
1509	static void	1565	INLINE void
1510	prepare_schedule (pTHX_ struct transfer_args *ta)	1566	prepare_schedule (pTHX_ struct coro_transfer_args *ta)
1511	{	1567	{
1512	SV *prev_sv, *next_sv;	1568	SV *prev_sv, *next_sv;
1513		1569
1514	for (;;)	1570	for (;;)
1515	{	1571	{
1516	LOCK;
1517	next_sv = coro_deq (aTHX);	1572	next_sv = coro_deq (aTHX);
1518		1573
1519	/* nothing to schedule: call the idle handler */	1574	/* nothing to schedule: call the idle handler */
1520	if (expect_false (!next_sv))	1575	if (expect_false (!next_sv))
1521	{	1576	{
1522	dSP;	1577	dSP;
1523	UNLOCK;
1524		1578
1525	ENTER;	1579	ENTER;
1526	SAVETMPS;	1580	SAVETMPS;
1527		1581
1528	PUSHMARK (SP);	1582	PUSHMARK (SP);
…		…
1533	FREETMPS;	1587	FREETMPS;
1534	LEAVE;	1588	LEAVE;
1535	continue;	1589	continue;
1536	}	1590	}
1537		1591
1538	ta->next = SvSTATE (next_sv);	1592	ta->next = SvSTATE_hv (next_sv);
1539		1593
1540	/* cannot transfer to destroyed coros, skip and look for next */	1594	/* cannot transfer to destroyed coros, skip and look for next */
1541	if (expect_false (ta->next->flags & CF_DESTROYED))	1595	if (expect_false (ta->next->flags & CF_DESTROYED))
1542	{	1596	{
1543	UNLOCK;
1544	SvREFCNT_dec (next_sv);	1597	SvREFCNT_dec (next_sv);
1545	/* coro_nready is already taken care of by destroy */	1598	/* coro_nready has already been taken care of by destroy */
1546	continue;	1599	continue;
1547	}	1600	}
1548		1601
1549	--coro_nready;	1602	--coro_nready;
1550	UNLOCK;
1551	break;	1603	break;
1552	}	1604	}
1553		1605
1554	/* free this only after the transfer */	1606	/* free this only after the transfer */
1555	prev_sv = SvRV (coro_current);	1607	prev_sv = SvRV (coro_current);
1556	ta->prev = SvSTATE (prev_sv);	1608	ta->prev = SvSTATE_hv (prev_sv);
1557	TRANSFER_CHECK (*ta);	1609	TRANSFER_CHECK (*ta);
1558	assert (ta->next->flags & CF_READY);	1610	assert (("FATAL: next coroutine isn't marked as ready in Coro (please report)", ta->next->flags & CF_READY));
1559	ta->next->flags &= ~CF_READY;	1611	ta->next->flags &= ~CF_READY;
1560	SvRV_set (coro_current, next_sv);	1612	SvRV_set (coro_current, next_sv);
1561		1613
1562	LOCK;
1563	free_coro_mortal (aTHX);	1614	free_coro_mortal (aTHX);
1564	coro_mortal = prev_sv;	1615	coro_mortal = prev_sv;
1565	UNLOCK;
1566	}	1616	}
1567		1617
1568	static void	1618	INLINE void
1569	prepare_cede (pTHX_ struct transfer_args *ta)	1619	prepare_cede (pTHX_ struct coro_transfer_args *ta)
1570	{	1620	{
1571	api_ready (coro_current);	1621	api_ready (aTHX_ coro_current);
1572	prepare_schedule (aTHX_ ta);	1622	prepare_schedule (aTHX_ ta);
1573	}	1623	}
1574		1624
		1625	INLINE void
		1626	prepare_cede_notself (pTHX_ struct coro_transfer_args *ta)
		1627	{
		1628	SV *prev = SvRV (coro_current);
		1629
		1630	if (coro_nready)
		1631	{
		1632	prepare_schedule (aTHX_ ta);
		1633	api_ready (aTHX_ prev);
		1634	}
		1635	else
		1636	prepare_nop (aTHX_ ta);
		1637	}
		1638
		1639	static void
		1640	api_schedule (pTHX)
		1641	{
		1642	struct coro_transfer_args ta;
		1643
		1644	prepare_schedule (aTHX_ &ta);
		1645	TRANSFER (ta, 1);
		1646	}
		1647
1575	static int	1648	static int
1576	prepare_cede_notself (pTHX_ struct transfer_args *ta)	1649	api_cede (pTHX)
1577	{	1650	{
1578	if (coro_nready)	1651	struct coro_transfer_args ta;
1579	{	1652
1580	SV *prev = SvRV (coro_current);
1581	prepare_schedule (aTHX_ ta);	1653	prepare_cede (aTHX_ &ta);
1582	api_ready (prev);	1654
		1655	if (expect_true (ta.prev != ta.next))
		1656	{
		1657	TRANSFER (ta, 1);
1583	return 1;	1658	return 1;
1584	}	1659	}
1585	else	1660	else
1586	return 0;	1661	return 0;
1587	}	1662	}
1588		1663
1589	static void
1590	api_schedule (void)
1591	{
1592	dTHX;
1593	struct transfer_args ta;
1594
1595	prepare_schedule (aTHX_ &ta);
1596	TRANSFER (ta, 1);
1597	}
1598
1599	static int	1664	static int
1600	api_cede (void)	1665	api_cede_notself (pTHX)
1601	{	1666	{
1602	dTHX;	1667	if (coro_nready)
		1668	{
1603	struct transfer_args ta;	1669	struct coro_transfer_args ta;
1604		1670
1605	prepare_cede (aTHX_ &ta);	1671	prepare_cede_notself (aTHX_ &ta);
1606
1607	if (expect_true (ta.prev != ta.next))
1608	{
1609	TRANSFER (ta, 1);	1672	TRANSFER (ta, 1);
1610	return 1;	1673	return 1;
1611	}	1674	}
1612	else	1675	else
1613	return 0;	1676	return 0;
1614	}	1677	}
1615		1678
1616	static int	1679	static void
1617	api_cede_notself (void)
1618	{
1619	dTHX;
1620	struct transfer_args ta;
1621
1622	if (prepare_cede_notself (aTHX_ &ta))
1623	{
1624	TRANSFER (ta, 1);
1625	return 1;
1626	}
1627	else
1628	return 0;
1629	}
1630
1631	static void
1632	api_trace (SV *coro_sv, int flags)	1680	api_trace (pTHX_ SV *coro_sv, int flags)
1633	{	1681	{
1634	dTHX;
1635	struct coro *coro = SvSTATE (coro_sv);	1682	struct coro *coro = SvSTATE (coro_sv);
1636		1683
1637	if (flags & CC_TRACE)	1684	if (flags & CC_TRACE)
1638	{	1685	{
1639	if (!coro->cctx)	1686	if (!coro->cctx)
1640	coro->cctx = cctx_new ();	1687	coro->cctx = cctx_new_run ();
1641	else if (!(coro->cctx->flags & CC_TRACE))	1688	else if (!(coro->cctx->flags & CC_TRACE))
1642	croak ("cannot enable tracing on coroutine with custom stack");	1689	croak ("cannot enable tracing on coroutine with custom stack,");
1643		1690
1644	coro->cctx->flags |= CC_NOREUSE | (flags & (CC_TRACE | CC_TRACE_ALL));	1691	coro->cctx->flags |= CC_NOREUSE | (flags & (CC_TRACE | CC_TRACE_ALL));
1645	}	1692	}
1646	else if (coro->cctx && coro->cctx->flags & CC_TRACE)	1693	else if (coro->cctx && coro->cctx->flags & CC_TRACE)
1647	{	1694	{
…		…
1652	else	1699	else
1653	coro->slot->runops = RUNOPS_DEFAULT;	1700	coro->slot->runops = RUNOPS_DEFAULT;
1654	}	1701	}
1655	}	1702	}
1656		1703
1657	static int
1658	coro_gensub_free (pTHX_ SV *sv, MAGIC *mg)
1659	{
1660	AV *padlist;
1661	AV *av = (AV *)mg->mg_obj;
1662
1663	abort ();
1664
1665	return 0;
1666	}
1667
1668	static MGVTBL coro_gensub_vtbl = {
1669	0, 0, 0, 0,
1670	coro_gensub_free
1671	};
1672
1673	/*****************************************************************************/	1704	/*****************************************************************************/
1674	/* PerlIO::cede */	1705	/* PerlIO::cede */
1675		1706
1676	typedef struct	1707	typedef struct
1677	{	1708	{
…		…
1704	PerlIOCede *self = PerlIOSelf (f, PerlIOCede);	1735	PerlIOCede *self = PerlIOSelf (f, PerlIOCede);
1705	double now = nvtime ();	1736	double now = nvtime ();
1706		1737
1707	if (now >= self->next)	1738	if (now >= self->next)
1708	{	1739	{
1709	api_cede ();	1740	api_cede (aTHX);
1710	self->next = now + self->every;	1741	self->next = now + self->every;
1711	}	1742	}
1712		1743
1713	return PerlIOBuf_flush (aTHX_ f);	1744	return PerlIOBuf_flush (aTHX_ f);
1714	}	1745	}
…		…
1743	PerlIOBuf_get_ptr,	1774	PerlIOBuf_get_ptr,
1744	PerlIOBuf_get_cnt,	1775	PerlIOBuf_get_cnt,
1745	PerlIOBuf_set_ptrcnt,	1776	PerlIOBuf_set_ptrcnt,
1746	};	1777	};
1747		1778
		1779	/*****************************************************************************/
		1780
		1781	static UNOP slf_restore; /* restore stack as entersub did, for first-re-run */
		1782	static const CV *slf_cv;
		1783	static SV **slf_argv;
		1784	static int slf_argc, slf_arga; /* count, allocated */
		1785	static I32 slf_ax; /* top of stack, for restore */
		1786
		1787	/* this restores the stack in the case we patched the entersub, to */
		1788	/* recreate the stack frame as perl will on following calls */
		1789	/* since entersub cleared the stack */
		1790	static OP *
		1791	pp_restore (pTHX)
		1792	{
		1793	int i;
		1794	SV **SP = PL_stack_base + slf_ax;
		1795
		1796	PUSHMARK (SP);
		1797
		1798	EXTEND (SP, slf_argc + 1);
		1799
		1800	for (i = 0; i < slf_argc; ++i)
		1801	PUSHs (sv_2mortal (slf_argv [i]));
		1802
		1803	PUSHs ((SV *)CvGV (slf_cv));
		1804
		1805	RETURNOP (slf_restore.op_first);
		1806	}
		1807
		1808	static void
		1809	slf_prepare_set_stacklevel (pTHX_ struct coro_transfer_args *ta)
		1810	{
		1811	prepare_set_stacklevel (ta, (struct coro_cctx *)slf_frame.data);
		1812	}
		1813
		1814	static void
		1815	slf_init_set_stacklevel (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		1816	{
		1817	assert (("FATAL: set_stacklevel needs the coro cctx as sole argument", items == 1));
		1818
		1819	frame->prepare = slf_prepare_set_stacklevel;
		1820	frame->check = slf_check_nop;
		1821	frame->data = (void *)SvIV (arg [0]);
		1822	}
		1823
		1824	static void
		1825	slf_prepare_transfer (pTHX_ struct coro_transfer_args *ta)
		1826	{
		1827	SV **arg = (SV **)slf_frame.data;
		1828
		1829	prepare_transfer (aTHX_ ta, arg [0], arg [1]);
		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
		1866
		1867	/*
		1868	* these not obviously related functions are all rolled into one
		1869	* function to increase chances that they all will call transfer with the same
		1870	* stack offset
		1871	* SLF stands for "schedule-like-function".
		1872	*/
		1873	static OP *
		1874	pp_slf (pTHX)
		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 */
		1884	dSP;
		1885	SV **arg = PL_stack_base + TOPMARK + 1;
		1886	int items = SP - arg; /* args without function object */
		1887	SV *gv = *sp;
		1888
		1889	/* do a quick consistency check on the "function" object, and if it isn't */
		1890	/* for us, divert to the real entersub */
		1891	if (SvTYPE (gv) != SVt_PVGV || !(CvFLAGS (GvCV (gv)) & CVf_SLF))
		1892	return PL_ppaddr[OP_ENTERSUB](aTHX);
		1893
		1894	if (!(PL_op->op_flags & OPf_STACKED))
		1895	{
		1896	/* ampersand-form of call, use @_ instead of stack */
		1897	AV *av = GvAV (PL_defgv);
		1898	arg = AvARRAY (av);
		1899	items = AvFILLp (av) + 1;
		1900	}
		1901
		1902	/* now call the init function, which needs to set up slf_frame */
		1903	((coro_slf_cb)CvXSUBANY (GvCV (gv)).any_ptr)
		1904	(aTHX_ &slf_frame, GvCV (gv), arg, items);
		1905
		1906	/* pop args */
		1907	SP = PL_stack_base + POPMARK;
		1908
		1909	PUTBACK;
		1910	}
		1911
		1912	/* now that we have a slf_frame, interpret it! */
		1913	/* we use a callback system not to make the code needlessly */
		1914	/* complicated, but so we can run multiple perl coros from one cctx */
		1915
		1916	do
		1917	{
		1918	struct coro_transfer_args ta;
		1919
		1920	slf_frame.prepare (aTHX_ &ta);
		1921	TRANSFER (ta, 0);
		1922
		1923	checkmark = PL_stack_sp - PL_stack_base;
		1924	}
		1925	while (slf_frame.check (aTHX_ &slf_frame));
		1926
		1927	slf_frame.prepare = 0; /* invalidate the frame, we are done processing it */
		1928
		1929	/* return value handling - mostly like entersub */
		1930	{
		1931	dSP;
		1932	SV **bot = PL_stack_base + checkmark;
		1933	int gimme = GIMME_V;
		1934
		1935	/* make sure we put something on the stack in scalar context */
		1936	if (gimme == G_SCALAR)
		1937	{
		1938	if (sp == bot)
		1939	XPUSHs (&PL_sv_undef);
		1940
		1941	SP = bot + 1;
		1942	}
		1943
		1944	PUTBACK;
		1945	}
		1946
		1947	/* exception handling */
		1948	if (expect_false (coro_throw))
		1949	{
		1950	SV *exception = sv_2mortal (coro_throw);
		1951
		1952	coro_throw = 0;
		1953	sv_setsv (ERRSV, exception);
		1954	croak (0);
		1955	}
		1956
		1957	return NORMAL;
		1958	}
		1959
		1960	static void
		1961	api_execute_slf (pTHX_ CV *cv, coro_slf_cb init_cb, I32 ax)
		1962	{
		1963	int i;
		1964	SV **arg = PL_stack_base + ax;
		1965	int items = PL_stack_sp - arg + 1;
		1966
		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	CvFLAGS (cv) |= CVf_SLF;
		1974	CvXSUBANY (cv).any_ptr = (void *)init_cb;
		1975	slf_cv = cv;
		1976
		1977	/* we patch the op, and then re-run the whole call */
		1978	/* we have to put the same argument on the stack for this to work */
		1979	/* and this will be done by pp_restore */
		1980	slf_restore.op_next = (OP *)&slf_restore;
		1981	slf_restore.op_type = OP_CUSTOM;
		1982	slf_restore.op_ppaddr = pp_restore;
		1983	slf_restore.op_first = PL_op;
		1984
		1985	slf_ax = ax - 1; /* undo the ax++ inside dAXMARK */
		1986
		1987	if (items > slf_arga)
		1988	{
		1989	slf_arga = items;
		1990	free (slf_argv);
		1991	slf_argv = malloc (slf_arga * sizeof (SV *));
		1992	}
		1993
		1994	slf_argc = items;
		1995
		1996	for (i = 0; i < items; ++i)
		1997	slf_argv [i] = SvREFCNT_inc (arg [i]);
		1998
		1999	PL_op->op_ppaddr = pp_slf;
		2000	PL_op->op_type = OP_CUSTOM; /* maybe we should leave it at entersub? */
		2001
		2002	PL_op = (OP *)&slf_restore;
		2003	}
		2004
		2005	/*****************************************************************************/
		2006
		2007	static void
		2008	coro_semaphore_adjust (pTHX_ AV *av, IV adjust)
		2009	{
		2010	SV *count_sv = AvARRAY (av)[0];
		2011	IV count = SvIVX (count_sv);
		2012
		2013	count += adjust;
		2014	SvIVX (count_sv) = count;
		2015
		2016	/* now wake up as many waiters as are expected to lock */
		2017	while (count > 0 && AvFILLp (av) > 0)
		2018	{
		2019	SV *cb;
		2020
		2021	/* swap first two elements so we can shift a waiter */
		2022	AvARRAY (av)[0] = AvARRAY (av)[1];
		2023	AvARRAY (av)[1] = count_sv;
		2024	cb = av_shift (av);
		2025
		2026	if (SvOBJECT (cb))
		2027	api_ready (aTHX_ cb);
		2028	else
		2029	croak ("callbacks not yet supported");
		2030
		2031	SvREFCNT_dec (cb);
		2032
		2033	--count;
		2034	}
		2035	}
		2036
		2037	static void
		2038	coro_semaphore_on_destroy (pTHX_ struct coro *coro)
		2039	{
		2040	/* call $sem->adjust (0) to possibly wake up some other waiters */
		2041	coro_semaphore_adjust (aTHX_ (AV *)coro->slf_frame.data, 0);
		2042	}
		2043
		2044	static int
		2045	slf_check_semaphore_down (pTHX_ struct CoroSLF *frame)
		2046	{
		2047	AV *av = (AV *)frame->data;
		2048	SV *count_sv = AvARRAY (av)[0];
		2049
		2050	if (SvIVX (count_sv) > 0)
		2051	{
		2052	SvSTATE_current->on_destroy = 0;
		2053	SvIVX (count_sv) = SvIVX (count_sv) - 1;
		2054	return 0;
		2055	}
		2056	else
		2057	{
		2058	int i;
		2059	/* if we were woken up but can't down, we look through the whole */
		2060	/* waiters list and only add us if we aren't in there already */
		2061	/* this avoids some degenerate memory usage cases */
		2062
		2063	for (i = 1; i <= AvFILLp (av); ++i)
		2064	if (AvARRAY (av)[i] == SvRV (coro_current))
		2065	return 1;
		2066
		2067	av_push (av, SvREFCNT_inc (SvRV (coro_current)));
		2068	return 1;
		2069	}
		2070	}
		2071
		2072	static void
		2073	slf_init_semaphore_down (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		2074	{
		2075	AV *av = (AV *)SvRV (arg [0]);
		2076
		2077	if (SvIVX (AvARRAY (av)[0]) > 0)
		2078	{
		2079	frame->data = (void *)av;
		2080	frame->prepare = prepare_nop;
		2081	SvSTATE_current->on_destroy = coro_semaphore_on_destroy;
		2082	}
		2083	else
		2084	{
		2085	av_push (av, SvREFCNT_inc (SvRV (coro_current)));
		2086
		2087	frame->data = (void *)sv_2mortal (SvREFCNT_inc ((SV *)av));
		2088	frame->prepare = prepare_schedule;
		2089
		2090	/* to avoid race conditions when a woken-up coro gets terminated */
		2091	/* we arrange for a temporary on_destroy that calls adjust (0) */
		2092	assert (!SvSTATE_current->on_destroy);//D
		2093	SvSTATE_current->on_destroy = coro_semaphore_on_destroy;
		2094	}
		2095
		2096	frame->check = slf_check_semaphore_down;
		2097
		2098	}
		2099
		2100	/*****************************************************************************/
		2101
		2102	#define GENSUB_ARG CvXSUBANY (cv).any_ptr
		2103
		2104	/* create a closure from XS, returns a code reference */
		2105	/* the arg can be accessed via GENSUB_ARG from the callback */
		2106	/* the callback must use dXSARGS/XSRETURN */
		2107	static SV *
		2108	gensub (pTHX_ void (*xsub)(pTHX_ CV *), void *arg)
		2109	{
		2110	CV *cv = (CV *)NEWSV (0, 0);
		2111
		2112	sv_upgrade ((SV *)cv, SVt_PVCV);
		2113
		2114	CvANON_on (cv);
		2115	CvISXSUB_on (cv);
		2116	CvXSUB (cv) = xsub;
		2117	GENSUB_ARG = arg;
		2118
		2119	return newRV_noinc ((SV *)cv);
		2120	}
		2121
		2122	/*****************************************************************************/
1748		2123
1749	MODULE = Coro::State PACKAGE = Coro::State PREFIX = api_	2124	MODULE = Coro::State PACKAGE = Coro::State PREFIX = api_
1750		2125
1751	PROTOTYPES: DISABLE	2126	PROTOTYPES: DISABLE
1752		2127
1753	BOOT:	2128	BOOT:
1754	{	2129	{
1755	#ifdef USE_ITHREADS	2130	#ifdef USE_ITHREADS
1756	MUTEX_INIT (&coro_lock);	2131	# if CORO_PTHREAD
		2132	coro_thx = PERL_GET_CONTEXT;
		2133	# endif
1757	#endif	2134	#endif
1758	BOOT_PAGESIZE;	2135	BOOT_PAGESIZE;
1759		2136
1760	irsgv = gv_fetchpv ("/" , GV_ADD|GV_NOTQUAL, SVt_PV);	2137	irsgv = gv_fetchpv ("/" , GV_ADD|GV_NOTQUAL, SVt_PV);
1761	stdoutgv = gv_fetchpv ("STDOUT", GV_ADD|GV_NOTQUAL, SVt_PVIO);	2138	stdoutgv = gv_fetchpv ("STDOUT", GV_ADD|GV_NOTQUAL, SVt_PVIO);
…		…
1779	main_top_env = PL_top_env;	2156	main_top_env = PL_top_env;
1780		2157
1781	while (main_top_env->je_prev)	2158	while (main_top_env->je_prev)
1782	main_top_env = main_top_env->je_prev;	2159	main_top_env = main_top_env->je_prev;
1783		2160
		2161	{
		2162	SV *slf = sv_2mortal (newSViv (PTR2IV (pp_slf)));
		2163
		2164	if (!PL_custom_op_names) PL_custom_op_names = newHV ();
		2165	hv_store_ent (PL_custom_op_names, slf,
		2166	newSVpv ("coro_slf", 0), 0);
		2167
		2168	if (!PL_custom_op_descs) PL_custom_op_descs = newHV ();
		2169	hv_store_ent (PL_custom_op_descs, slf,
		2170	newSVpv ("coro schedule like function", 0), 0);
		2171	}
		2172
1784	coroapi.ver = CORO_API_VERSION;	2173	coroapi.ver = CORO_API_VERSION;
1785	coroapi.rev = CORO_API_REVISION;	2174	coroapi.rev = CORO_API_REVISION;
		2175
1786	coroapi.transfer = api_transfer;	2176	coroapi.transfer = api_transfer;
		2177
		2178	coroapi.sv_state = SvSTATE_;
		2179	coroapi.execute_slf = api_execute_slf;
		2180	coroapi.prepare_nop = prepare_nop;
		2181	coroapi.prepare_schedule = prepare_schedule;
		2182	coroapi.prepare_cede = prepare_cede;
		2183	coroapi.prepare_cede_notself = prepare_cede_notself;
1787		2184
1788	{	2185	{
1789	SV **svp = hv_fetch (PL_modglobal, "Time::NVtime", 12, 0);	2186	SV **svp = hv_fetch (PL_modglobal, "Time::NVtime", 12, 0);
1790		2187
1791	if (!svp) croak ("Time::HiRes is required");	2188	if (!svp) croak ("Time::HiRes is required");
…		…
1824	av_push (coro->args, newSVsv (ST (i)));	2221	av_push (coro->args, newSVsv (ST (i)));
1825	}	2222	}
1826	OUTPUT:	2223	OUTPUT:
1827	RETVAL	2224	RETVAL
1828		2225
1829	# these not obviously related functions are all rolled into the same xs
1830	# function to increase chances that they all will call transfer with the same
1831	# stack offset
1832	void	2226	void
1833	_set_stacklevel (...)	2227	_set_stacklevel (...)
1834	ALIAS:	2228	CODE:
1835	Coro::State::transfer = 1	2229	CORO_EXECUTE_SLF_XS (slf_init_set_stacklevel);
1836	Coro::schedule = 2
1837	Coro::cede = 3
1838	Coro::cede_notself = 4
1839	CODE:
1840	{
1841	struct transfer_args ta;
1842		2230
1843	PUTBACK;	2231	void
1844	switch (ix)	2232	transfer (...)
1845	{	2233	PROTOTYPE: $$
1846	case 0:	2234	CODE:
1847	ta.prev = (struct coro *)INT2PTR (coro_cctx *, SvIV (ST (0)));	2235	CORO_EXECUTE_SLF_XS (slf_init_transfer);
1848	ta.next = 0;
1849	break;
1850
1851	case 1:
1852	if (items != 2)
1853	croak ("Coro::State::transfer (prev, next) expects two arguments, not %d", items);
1854
1855	prepare_transfer (aTHX_ &ta, ST (0), ST (1));
1856	break;
1857
1858	case 2:
1859	prepare_schedule (aTHX_ &ta);
1860	break;
1861
1862	case 3:
1863	prepare_cede (aTHX_ &ta);
1864	break;
1865
1866	case 4:
1867	if (!prepare_cede_notself (aTHX_ &ta))
1868	XSRETURN_EMPTY;
1869
1870	break;
1871	}
1872	SPAGAIN;
1873
1874	BARRIER;
1875	PUTBACK;
1876	TRANSFER (ta, 0);
1877	SPAGAIN; /* might be the sp of a different coroutine now */
1878	/* be extra careful not to ever do anything after TRANSFER */
1879	}
1880		2236
1881	bool	2237	bool
1882	_destroy (SV *coro_sv)	2238	_destroy (SV *coro_sv)
1883	CODE:	2239	CODE:
1884	RETVAL = coro_state_destroy (aTHX_ SvSTATE (coro_sv));	2240	RETVAL = coro_state_destroy (aTHX_ SvSTATE (coro_sv));
…		…
1891	CODE:	2247	CODE:
1892	_exit (code);	2248	_exit (code);
1893		2249
1894	int	2250	int
1895	cctx_stacksize (int new_stacksize = 0)	2251	cctx_stacksize (int new_stacksize = 0)
		2252	PROTOTYPE: ;$
1896	CODE:	2253	CODE:
1897	RETVAL = cctx_stacksize;	2254	RETVAL = cctx_stacksize;
1898	if (new_stacksize)	2255	if (new_stacksize)
1899	{	2256	{
1900	cctx_stacksize = new_stacksize;	2257	cctx_stacksize = new_stacksize;
…		…
1903	OUTPUT:	2260	OUTPUT:
1904	RETVAL	2261	RETVAL
1905		2262
1906	int	2263	int
1907	cctx_max_idle (int max_idle = 0)	2264	cctx_max_idle (int max_idle = 0)
		2265	PROTOTYPE: ;$
1908	CODE:	2266	CODE:
1909	RETVAL = cctx_max_idle;	2267	RETVAL = cctx_max_idle;
1910	if (max_idle > 1)	2268	if (max_idle > 1)
1911	cctx_max_idle = max_idle;	2269	cctx_max_idle = max_idle;
1912	OUTPUT:	2270	OUTPUT:
1913	RETVAL	2271	RETVAL
1914		2272
1915	int	2273	int
1916	cctx_count ()	2274	cctx_count ()
		2275	PROTOTYPE:
1917	CODE:	2276	CODE:
1918	RETVAL = cctx_count;	2277	RETVAL = cctx_count;
1919	OUTPUT:	2278	OUTPUT:
1920	RETVAL	2279	RETVAL
1921		2280
1922	int	2281	int
1923	cctx_idle ()	2282	cctx_idle ()
		2283	PROTOTYPE:
1924	CODE:	2284	CODE:
1925	RETVAL = cctx_idle;	2285	RETVAL = cctx_idle;
1926	OUTPUT:	2286	OUTPUT:
1927	RETVAL	2287	RETVAL
1928		2288
1929	void	2289	void
1930	list ()	2290	list ()
		2291	PROTOTYPE:
1931	PPCODE:	2292	PPCODE:
1932	{	2293	{
1933	struct coro *coro;	2294	struct coro *coro;
1934	for (coro = coro_first; coro; coro = coro->next)	2295	for (coro = coro_first; coro; coro = coro->next)
1935	if (coro->hv)	2296	if (coro->hv)
…		…
1994	RETVAL = boolSV (coro->flags & ix);	2355	RETVAL = boolSV (coro->flags & ix);
1995	OUTPUT:	2356	OUTPUT:
1996	RETVAL	2357	RETVAL
1997		2358
1998	void	2359	void
		2360	throw (Coro::State self, SV *throw = &PL_sv_undef)
		2361	PROTOTYPE: $;$
		2362	CODE:
		2363	{
		2364	struct coro *current = SvSTATE_current;
		2365	SV **throwp = self == current ? &coro_throw : &self->throw;
		2366	SvREFCNT_dec (*throwp);
		2367	*throwp = SvOK (throw) ? newSVsv (throw) : 0;
		2368	}
		2369
		2370	void
1999	api_trace (SV *coro, int flags = CC_TRACE | CC_TRACE_SUB)	2371	api_trace (SV *coro, int flags = CC_TRACE | CC_TRACE_SUB)
		2372	PROTOTYPE: $;$
		2373	C_ARGS: aTHX_ coro, flags
2000		2374
2001	SV *	2375	SV *
2002	has_cctx (Coro::State coro)	2376	has_cctx (Coro::State coro)
2003	PROTOTYPE: $	2377	PROTOTYPE: $
2004	CODE:	2378	CODE:
…		…
2012	CODE:	2386	CODE:
2013	RETVAL = (coro->cctx ? coro->cctx->flags : 0) & CC_TRACE_ALL;	2387	RETVAL = (coro->cctx ? coro->cctx->flags : 0) & CC_TRACE_ALL;
2014	OUTPUT:	2388	OUTPUT:
2015	RETVAL	2389	RETVAL
2016		2390
2017	IV	2391	UV
2018	rss (Coro::State coro)	2392	rss (Coro::State coro)
2019	PROTOTYPE: $	2393	PROTOTYPE: $
2020	ALIAS:	2394	ALIAS:
2021	usecount = 1	2395	usecount = 1
2022	CODE:	2396	CODE:
…		…
2028	OUTPUT:	2402	OUTPUT:
2029	RETVAL	2403	RETVAL
2030		2404
2031	void	2405	void
2032	force_cctx ()	2406	force_cctx ()
		2407	PROTOTYPE:
2033	CODE:	2408	CODE:
2034	struct coro *coro = SvSTATE (coro_current);
2035	coro->cctx->idle_sp = 0;	2409	SvSTATE_current->cctx->idle_sp = 0;
2036		2410
2037	void	2411	void
2038	swap_defsv (Coro::State self)	2412	swap_defsv (Coro::State self)
2039	PROTOTYPE: $	2413	PROTOTYPE: $
2040	ALIAS:	2414	ALIAS:
2041	swap_defav = 1	2415	swap_defav = 1
2042	CODE:	2416	CODE:
2043	if (!self->slot)	2417	if (!self->slot)
2044	croak ("cannot swap state with coroutine that has no saved state");	2418	croak ("cannot swap state with coroutine that has no saved state,");
2045	else	2419	else
2046	{	2420	{
2047	SV **src = ix ? (SV **)&GvAV (PL_defgv) : &GvSV (PL_defgv);	2421	SV **src = ix ? (SV **)&GvAV (PL_defgv) : &GvSV (PL_defgv);
2048	SV **dst = ix ? (SV **)&self->slot->defav : (SV **)&self->slot->defsv;	2422	SV **dst = ix ? (SV **)&self->slot->defav : (SV **)&self->slot->defsv;
2049		2423
…		…
2074		2448
2075	for (i = PRIO_MAX - PRIO_MIN + 1; i--; )	2449	for (i = PRIO_MAX - PRIO_MIN + 1; i--; )
2076	coro_ready[i] = newAV ();	2450	coro_ready[i] = newAV ();
2077		2451
2078	{	2452	{
2079	SV *sv = perl_get_sv ("Coro::API", TRUE);	2453	SV *sv = coro_get_sv (aTHX_ "Coro::API", TRUE);
2080	perl_get_sv ("Coro::API", TRUE); /* silence 5.10 warning */
2081		2454
2082	coroapi.schedule = api_schedule;	2455	coroapi.schedule = api_schedule;
2083	coroapi.cede = api_cede;	2456	coroapi.cede = api_cede;
2084	coroapi.cede_notself = api_cede_notself;	2457	coroapi.cede_notself = api_cede_notself;
2085	coroapi.ready = api_ready;	2458	coroapi.ready = api_ready;
2086	coroapi.is_ready = api_is_ready;	2459	coroapi.is_ready = api_is_ready;
2087	coroapi.nready = &coro_nready;	2460	coroapi.nready = coro_nready;
2088	coroapi.current = coro_current;	2461	coroapi.current = coro_current;
2089		2462
2090	GCoroAPI = &coroapi;	2463	GCoroAPI = &coroapi;
2091	sv_setiv (sv, (IV)&coroapi);	2464	sv_setiv (sv, (IV)&coroapi);
2092	SvREADONLY_on (sv);	2465	SvREADONLY_on (sv);
2093	}	2466	}
2094	}	2467	}
		2468
		2469	void
		2470	schedule (...)
		2471	CODE:
		2472	CORO_EXECUTE_SLF_XS (slf_init_schedule);
		2473
		2474	void
		2475	cede (...)
		2476	CODE:
		2477	CORO_EXECUTE_SLF_XS (slf_init_cede);
		2478
		2479	void
		2480	cede_notself (...)
		2481	CODE:
		2482	CORO_EXECUTE_SLF_XS (slf_init_cede_notself);
2095		2483
2096	void	2484	void
2097	_set_current (SV *current)	2485	_set_current (SV *current)
2098	PROTOTYPE: $	2486	PROTOTYPE: $
2099	CODE:	2487	CODE:
…		…
2102		2490
2103	void	2491	void
2104	_set_readyhook (SV *hook)	2492	_set_readyhook (SV *hook)
2105	PROTOTYPE: $	2493	PROTOTYPE: $
2106	CODE:	2494	CODE:
2107	LOCK;
2108	SvREFCNT_dec (coro_readyhook);	2495	SvREFCNT_dec (coro_readyhook);
2109	coro_readyhook = SvOK (hook) ? newSVsv (hook) : 0;	2496	coro_readyhook = SvOK (hook) ? newSVsv (hook) : 0;
2110	UNLOCK;
2111		2497
2112	int	2498	int
2113	prio (Coro::State coro, int newprio = 0)	2499	prio (Coro::State coro, int newprio = 0)
		2500	PROTOTYPE: $;$
2114	ALIAS:	2501	ALIAS:
2115	nice = 1	2502	nice = 1
2116	CODE:	2503	CODE:
2117	{	2504	{
2118	RETVAL = coro->prio;	2505	RETVAL = coro->prio;
…		…
2133		2520
2134	SV *	2521	SV *
2135	ready (SV *self)	2522	ready (SV *self)
2136	PROTOTYPE: $	2523	PROTOTYPE: $
2137	CODE:	2524	CODE:
2138	RETVAL = boolSV (api_ready (self));	2525	RETVAL = boolSV (api_ready (aTHX_ self));
2139	OUTPUT:	2526	OUTPUT:
2140	RETVAL	2527	RETVAL
2141		2528
2142	int	2529	int
2143	nready (...)	2530	nready (...)
…		…
2145	CODE:	2532	CODE:
2146	RETVAL = coro_nready;	2533	RETVAL = coro_nready;
2147	OUTPUT:	2534	OUTPUT:
2148	RETVAL	2535	RETVAL
2149		2536
2150	void
2151	throw (Coro::State self, SV *throw = &PL_sv_undef)
2152	PROTOTYPE: $;$
2153	CODE:
2154	SvREFCNT_dec (self->throw);
2155	self->throw = SvOK (throw) ? newSVsv (throw) : 0;
2156
2157	# for async_pool speedup	2537	# for async_pool speedup
2158	void	2538	void
2159	_pool_1 (SV *cb)	2539	_pool_1 (SV *cb)
2160	CODE:	2540	CODE:
2161	{	2541	{
2162	struct coro *coro = SvSTATE (coro_current);
2163	HV *hv = (HV *)SvRV (coro_current);	2542	HV *hv = (HV *)SvRV (coro_current);
		2543	struct coro *coro = SvSTATE_hv ((SV *)hv);
2164	AV *defav = GvAV (PL_defgv);	2544	AV *defav = GvAV (PL_defgv);
2165	SV *invoke = hv_delete (hv, "_invoke", sizeof ("_invoke") - 1, 0);	2545	SV *invoke = hv_delete (hv, "_invoke", sizeof ("_invoke") - 1, 0);
2166	AV *invoke_av;	2546	AV *invoke_av;
2167	int i, len;	2547	int i, len;
2168		2548
…		…
2189	{	2569	{
2190	av_fill (defav, len - 1);	2570	av_fill (defav, len - 1);
2191	for (i = 0; i < len; ++i)	2571	for (i = 0; i < len; ++i)
2192	av_store (defav, i, SvREFCNT_inc_NN (AvARRAY (invoke_av)[i + 1]));	2572	av_store (defav, i, SvREFCNT_inc_NN (AvARRAY (invoke_av)[i + 1]));
2193	}	2573	}
2194
2195	SvREFCNT_dec (invoke);
2196	}	2574	}
2197		2575
2198	void	2576	void
2199	_pool_2 (SV *cb)	2577	_pool_2 (SV *cb)
2200	CODE:	2578	CODE:
2201	{	2579	{
2202	struct coro *coro = SvSTATE (coro_current);	2580	HV *hv = (HV *)SvRV (coro_current);
		2581	struct coro *coro = SvSTATE_hv ((SV *)hv);
2203		2582
2204	sv_setsv (cb, &PL_sv_undef);	2583	sv_setsv (cb, &PL_sv_undef);
2205		2584
2206	SvREFCNT_dec ((SV *)PL_defoutgv); PL_defoutgv = (GV *)coro->saved_deffh;	2585	SvREFCNT_dec ((SV *)PL_defoutgv); PL_defoutgv = (GV *)coro->saved_deffh;
2207	coro->saved_deffh = 0;	2586	coro->saved_deffh = 0;
2208		2587
2209	if (coro_rss (aTHX_ coro) > SvIV (sv_pool_rss)	2588	if (coro_rss (aTHX_ coro) > SvUV (sv_pool_rss)
2210	|| av_len (av_async_pool) + 1 >= SvIV (sv_pool_size))	2589	|| av_len (av_async_pool) + 1 >= SvIV (sv_pool_size))
2211	{	2590	{
2212	SV *old = PL_diehook;	2591	SV *old = PL_diehook;
2213	PL_diehook = 0;	2592	PL_diehook = 0;
2214	SvREFCNT_dec (old);	2593	SvREFCNT_dec (old);
2215	croak ("\3async_pool terminate\2\n");	2594	croak ("\3async_pool terminate\2\n");
2216	}	2595	}
2217		2596
2218	av_clear (GvAV (PL_defgv));	2597	av_clear (GvAV (PL_defgv));
2219	hv_store ((HV *)SvRV (coro_current), "desc", sizeof ("desc") - 1,	2598	hv_store (hv, "desc", sizeof ("desc") - 1,
2220	newSVpvn ("[async_pool idle]", sizeof ("[async_pool idle]") - 1), 0);	2599	newSVpvn ("[async_pool idle]", sizeof ("[async_pool idle]") - 1), 0);
2221		2600
2222	coro->prio = 0;	2601	coro->prio = 0;
2223		2602
2224	if (coro->cctx && (coro->cctx->flags & CC_TRACE))	2603	if (coro->cctx && (coro->cctx->flags & CC_TRACE))
2225	api_trace (coro_current, 0);	2604	api_trace (aTHX_ coro_current, 0);
2226		2605
2227	av_push (av_async_pool, newSVsv (coro_current));	2606	av_push (av_async_pool, newSVsv (coro_current));
2228	}	2607	}
2229
2230	#if 0
2231
2232	void
2233	_generator_call (...)
2234	PROTOTYPE: @
2235	PPCODE:
2236	fprintf (stderr, "call %p\n", CvXSUBANY(cv).any_ptr);
2237	xxxx
2238	abort ();
2239
2240	SV *
2241	gensub (SV *sub, ...)
2242	PROTOTYPE: &;@
2243	CODE:
2244	{
2245	struct coro *coro;
2246	MAGIC *mg;
2247	CV *xcv;
2248	CV *ncv = (CV *)newSV_type (SVt_PVCV);
2249	int i;
2250
2251	CvGV (ncv) = CvGV (cv);
2252	CvFILE (ncv) = CvFILE (cv);
2253
2254	Newz (0, coro, 1, struct coro);
2255	coro->args = newAV ();
2256	coro->flags = CF_NEW;
2257
2258	av_extend (coro->args, items - 1);
2259	for (i = 1; i < items; i++)
2260	av_push (coro->args, newSVsv (ST (i)));
2261
2262	CvISXSUB_on (ncv);
2263	CvXSUBANY (ncv).any_ptr = (void *)coro;
2264
2265	xcv = GvCV (gv_fetchpv ("Coro::_generator_call", 0, SVt_PVCV));
2266
2267	CvXSUB (ncv) = CvXSUB (xcv);
2268	CvANON_on (ncv);
2269
2270	mg = sv_magicext ((SV *)ncv, 0, CORO_MAGIC_type_state, &coro_gensub_vtbl, (char *)coro, 0);
2271	RETVAL = newRV_noinc ((SV *)ncv);
2272	}
2273	OUTPUT:
2274	RETVAL
2275
2276	#endif
2277		2608
2278		2609
2279	MODULE = Coro::State PACKAGE = Coro::AIO	2610	MODULE = Coro::State PACKAGE = Coro::AIO
2280		2611
2281	void	2612	void
2282	_get_state (SV *self)	2613	_get_state (SV *self)
		2614	PROTOTYPE: $
2283	PPCODE:	2615	PPCODE:
2284	{	2616	{
2285	AV *defav = GvAV (PL_defgv);	2617	AV *defav = GvAV (PL_defgv);
2286	AV *av = newAV ();	2618	AV *av = newAV ();
2287	int i;	2619	int i;
…		…
2302		2634
2303	av_push (av, data_sv);	2635	av_push (av, data_sv);
2304		2636
2305	XPUSHs (sv_2mortal (newRV_noinc ((SV *)av)));	2637	XPUSHs (sv_2mortal (newRV_noinc ((SV *)av)));
2306		2638
2307	api_ready (self);	2639	api_ready (aTHX_ self);
2308	}	2640	}
2309		2641
2310	void	2642	void
2311	_set_state (SV *state)	2643	_set_state (SV *state)
2312	PROTOTYPE: $	2644	PROTOTYPE: $
…		…
2330	MODULE = Coro::State PACKAGE = Coro::AnyEvent	2662	MODULE = Coro::State PACKAGE = Coro::AnyEvent
2331		2663
2332	BOOT:	2664	BOOT:
2333	sv_activity = coro_get_sv (aTHX_ "Coro::AnyEvent::ACTIVITY", TRUE);	2665	sv_activity = coro_get_sv (aTHX_ "Coro::AnyEvent::ACTIVITY", TRUE);
2334		2666
2335	SV *	2667	void
2336	_schedule (...)	2668	_schedule (...)
2337	PROTOTYPE: @
2338	CODE:	2669	CODE:
2339	{	2670	{
2340	static int incede;	2671	static int incede;
2341		2672
2342	api_cede_notself ();	2673	api_cede_notself (aTHX);
2343		2674
2344	++incede;	2675	++incede;
2345	while (coro_nready >= incede && api_cede ())	2676	while (coro_nready >= incede && api_cede (aTHX))
2346	;	2677	;
2347		2678
2348	sv_setsv (sv_activity, &PL_sv_undef);	2679	sv_setsv (sv_activity, &PL_sv_undef);
2349	if (coro_nready >= incede)	2680	if (coro_nready >= incede)
2350	{	2681	{
…		…
2360		2691
2361	MODULE = Coro::State PACKAGE = PerlIO::cede	2692	MODULE = Coro::State PACKAGE = PerlIO::cede
2362		2693
2363	BOOT:	2694	BOOT:
2364	PerlIO_define_layer (aTHX_ &PerlIO_cede);	2695	PerlIO_define_layer (aTHX_ &PerlIO_cede);
		2696
		2697	MODULE = Coro::State PACKAGE = Coro::Semaphore
		2698
		2699	SV *
		2700	new (SV *klass, SV *count_ = 0)
		2701	CODE:
		2702	{
		2703	/* a semaphore contains a counter IV in $sem->[0] and any waiters after that */
		2704	AV *av = newAV ();
		2705	av_push (av, newSViv (count_ && SvOK (count_) ? SvIV (count_) : 1));
		2706	RETVAL = sv_bless (newRV_noinc ((SV *)av), GvSTASH (CvGV (cv)));
		2707	}
		2708	OUTPUT:
		2709	RETVAL
		2710
		2711	SV *
		2712	count (SV *self)
		2713	CODE:
		2714	RETVAL = newSVsv (AvARRAY ((AV *)SvRV (self))[0]);
		2715	OUTPUT:
		2716	RETVAL
		2717
		2718	void
		2719	up (SV *self, int adjust = 1)
		2720	ALIAS:
		2721	adjust = 1
		2722	CODE:
		2723	coro_semaphore_adjust (aTHX_ (AV *)SvRV (self), ix ? adjust : 1);
		2724
		2725	void
		2726	down (SV *self)
		2727	CODE:
		2728	CORO_EXECUTE_SLF_XS (slf_init_semaphore_down);
		2729
		2730	void
		2731	try (SV *self)
		2732	PPCODE:
		2733	{
		2734	AV *av = (AV *)SvRV (self);
		2735	SV *count_sv = AvARRAY (av)[0];
		2736	IV count = SvIVX (count_sv);
		2737
		2738	if (count > 0)
		2739	{
		2740	--count;
		2741	SvIVX (count_sv) = count;
		2742	XSRETURN_YES;
		2743	}
		2744	else
		2745	XSRETURN_NO;
		2746	}
		2747
		2748	void
		2749	waiters (SV *self)
		2750	CODE:
		2751	{
		2752	AV *av = (AV *)SvRV (self);
		2753
		2754	if (GIMME_V == G_SCALAR)
		2755	XPUSHs (sv_2mortal (newSVsv (AvARRAY (av)[0])));
		2756	else
		2757	{
		2758	int i;
		2759	EXTEND (SP, AvFILLp (av) + 1 - 1);
		2760	for (i = 1; i <= AvFILLp (av); ++i)
		2761	PUSHs (newSVsv (AvARRAY (av)[i]));
		2762	}
		2763	}
		2764

Diff Legend

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