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

Diff Legend

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