ViewVC Help

View File | Revision Log | Show Annotations | Download File

/cvs/Coro/Coro/State.xs

Comparing Coro/Coro/State.xs (file contents):
Revision 1.259 by root, Mon Nov 10 00:02:29 2008 UTC vs.
Revision 1.286 by root, Mon Nov 17 04:19:49 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
148	static perl_mutex coro_lock;
149	# define LOCK do { MUTEX_LOCK (&coro_lock); } while (0)
150	# define UNLOCK do { MUTEX_UNLOCK (&coro_lock); } while (0)
151	# if CORO_PTHREAD	147	# if CORO_PTHREAD
152	static void *coro_thx;	148	static void *coro_thx;
153	# endif	149	# endif
154
155	#else
156
157	# define LOCK (void)0
158	# define UNLOCK (void)0
159
160	#endif	150	#endif
161
162	# undef LOCK
163	# define LOCK (void)0
164	# undef UNLOCK
165	# define UNLOCK (void)0
166		151
167	/* helper storage struct for Coro::AIO */	152	/* helper storage struct for Coro::AIO */
168	struct io_state	153	struct io_state
169	{	154	{
170	AV *res;	155	AV *res;
…		…
181	static struct CoroAPI coroapi;	166	static struct CoroAPI coroapi;
182	static AV *main_mainstack; /* used to differentiate between $main and others */	167	static AV *main_mainstack; /* used to differentiate between $main and others */
183	static JMPENV *main_top_env;	168	static JMPENV *main_top_env;
184	static HV *coro_state_stash, *coro_stash;	169	static HV *coro_state_stash, *coro_stash;
185	static volatile SV *coro_mortal; /* will be freed/thrown after next transfer */	170	static volatile SV *coro_mortal; /* will be freed/thrown after next transfer */
186	static volatile char next_has_throw; /* speedup flag for next->throw check */
187		171
188	static GV *irsgv; /* $/ */	172	static GV *irsgv; /* $/ */
189	static GV *stdoutgv; /* *STDOUT */	173	static GV *stdoutgv; /* *STDOUT */
190	static SV *rv_diehook;	174	static SV *rv_diehook;
191	static SV *rv_warnhook;	175	static SV *rv_warnhook;
…		…
210	CC_TRACE_LINE = 0x10, /* trace each statement */	194	CC_TRACE_LINE = 0x10, /* trace each statement */
211	CC_TRACE_ALL = CC_TRACE_SUB | CC_TRACE_LINE,	195	CC_TRACE_ALL = CC_TRACE_SUB | CC_TRACE_LINE,
212	};	196	};
213		197
214	/* this is a structure representing a c-level coroutine */	198	/* this is a structure representing a c-level coroutine */
215	typedef struct coro_cctx {	199	typedef struct coro_cctx
		200	{
216	struct coro_cctx *next;	201	struct coro_cctx *next;
217		202
218	/* the stack */	203	/* the stack */
219	void *sptr;	204	void *sptr;
220	size_t ssize;	205	size_t ssize;
…		…
238	CF_NEW = 0x0004, /* has never been switched to */	223	CF_NEW = 0x0004, /* has never been switched to */
239	CF_DESTROYED = 0x0008, /* coroutine data has been freed */	224	CF_DESTROYED = 0x0008, /* coroutine data has been freed */
240	};	225	};
241		226
242	/* 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 */
243	typedef struct {	228	typedef struct
		229	{
244	SV *defsv;	230	SV *defsv;
245	AV *defav;	231	AV *defav;
246	SV *errsv;	232	SV *errsv;
247	SV *irsgv;	233	SV *irsgv;
248	#define VAR(name,type) type name;	234	#define VAR(name,type) type name;
…		…
252		238
253	#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))
254		240
255	/* this is a structure representing a perl-level coroutine */	241	/* this is a structure representing a perl-level coroutine */
256	struct coro {	242	struct coro {
257	/* the c coroutine allocated to this perl coroutine, if any */	243	/* the C coroutine allocated to this perl coroutine, if any */
258	coro_cctx *cctx;	244	coro_cctx *cctx;
259		245
260	/* process data */	246	/* state data */
		247	struct CoroSLF slf_frame; /* saved slf frame */
261	AV *mainstack;	248	AV *mainstack;
262	perl_slots *slot; /* basically the saved sp */	249	perl_slots *slot; /* basically the saved sp */
263		250
264	AV *args; /* data associated with this coroutine (initial args) */	251	AV *args; /* data associated with this coroutine (initial args) */
265	int refcnt; /* coroutines are refcounted, yes */	252	int refcnt; /* coroutines are refcounted, yes */
266	int flags; /* CF_ flags */	253	int flags; /* CF_ flags */
267	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);
268		256
269	/* statistics */	257	/* statistics */
270	int usecount; /* number of transfers to this coro */	258	int usecount; /* number of transfers to this coro */
271		259
272	/* coro process data */	260	/* coro process data */
…		…
280	struct coro *next, *prev;	268	struct coro *next, *prev;
281	};	269	};
282		270
283	typedef struct coro *Coro__State;	271	typedef struct coro *Coro__State;
284	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;
285		279
286	/** Coro ********************************************************************/	280	/** Coro ********************************************************************/
287		281
288	#define PRIO_MAX 3	282	#define PRIO_MAX 3
289	#define PRIO_HIGH 1	283	#define PRIO_HIGH 1
…		…
294		288
295	/* for Coro.pm */	289	/* for Coro.pm */
296	static SV *coro_current;	290	static SV *coro_current;
297	static SV *coro_readyhook;	291	static SV *coro_readyhook;
298	static AV *coro_ready [PRIO_MAX - PRIO_MIN + 1];	292	static AV *coro_ready [PRIO_MAX - PRIO_MIN + 1];
299	static int coro_nready;
300	static struct coro *coro_first;	293	static struct coro *coro_first;
		294	#define coro_nready coroapi.nready
301		295
302	/** lowlevel stuff **********************************************************/	296	/** lowlevel stuff **********************************************************/
303		297
304	static SV *	298	static SV *
305	coro_get_sv (pTHX_ const char *name, int create)	299	coro_get_sv (pTHX_ const char *name, int create)
…		…
398	static MGVTBL coro_cv_vtbl = {	392	static MGVTBL coro_cv_vtbl = {
399	0, 0, 0, 0,	393	0, 0, 0, 0,
400	coro_cv_free	394	coro_cv_free
401	};	395	};
402		396
403	#define CORO_MAGIC(sv, type) \	397	#define CORO_MAGIC(sv, type) \
404	SvMAGIC (sv) \	398	expect_true (SvMAGIC (sv)) \
405	? SvMAGIC (sv)->mg_type == type \	399	? expect_true (SvMAGIC (sv)->mg_type == type) \
406	? SvMAGIC (sv) \	400	? SvMAGIC (sv) \
407	: mg_find (sv, type) \	401	: mg_find (sv, type) \
408	: 0	402	: 0
409		403
410	#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)
411	#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)
412		406
413	static struct coro *	407	INLINE struct coro *
414	SvSTATE_ (pTHX_ SV *coro)	408	SvSTATE_ (pTHX_ SV *coro)
415	{	409	{
416	HV *stash;	410	HV *stash;
417	MAGIC *mg;	411	MAGIC *mg;
418		412
…		…
433	mg = CORO_MAGIC_state (coro);	427	mg = CORO_MAGIC_state (coro);
434	return (struct coro *)mg->mg_ptr;	428	return (struct coro *)mg->mg_ptr;
435	}	429	}
436		430
437	#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))
438		445
439	/* the next two functions merely cache the padlists */	446	/* the next two functions merely cache the padlists */
440	static void	447	static void
441	get_padlist (pTHX_ CV *cv)	448	get_padlist (pTHX_ CV *cv)
442	{	449	{
…		…
509	CvPADLIST (cv) = (AV *)POPs;	516	CvPADLIST (cv) = (AV *)POPs;
510	}	517	}
511		518
512	PUTBACK;	519	PUTBACK;
513	}	520	}
		521
		522	slf_frame = c->slf_frame;
		523	coro_throw = c->throw;
514	}	524	}
515		525
516	static void	526	static void
517	save_perl (pTHX_ Coro__State c)	527	save_perl (pTHX_ Coro__State c)
518	{	528	{
		529	c->throw = coro_throw;
		530	c->slf_frame = slf_frame;
		531
519	{	532	{
520	dSP;	533	dSP;
521	I32 cxix = cxstack_ix;	534	I32 cxix = cxstack_ix;
522	PERL_CONTEXT *ccstk = cxstack;	535	PERL_CONTEXT *ccstk = cxstack;
523	PERL_SI *top_si = PL_curstackinfo;	536	PERL_SI *top_si = PL_curstackinfo;
…		…
590	#undef VAR	603	#undef VAR
591	}	604	}
592	}	605	}
593		606
594	/*	607	/*
595	* allocate various perl stacks. This is an exact copy	608	* allocate various perl stacks. This is almost an exact copy
596	* of perl.c:init_stacks, except that it uses less memory	609	* of perl.c:init_stacks, except that it uses less memory
597	* on the (sometimes correct) assumption that coroutines do	610	* on the (sometimes correct) assumption that coroutines do
598	* not usually need a lot of stackspace.	611	* not usually need a lot of stackspace.
599	*/	612	*/
600	#if CORO_PREFER_PERL_FUNCTIONS	613	#if CORO_PREFER_PERL_FUNCTIONS
…		…
802		815
803	return orig_sigelem_set ? orig_sigelem_set (aTHX_ sv, mg) : 0;	816	return orig_sigelem_set ? orig_sigelem_set (aTHX_ sv, mg) : 0;
804	}	817	}
805		818
806	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 */
807	coro_setup (pTHX_ struct coro *coro)	833	coro_setup (pTHX_ struct coro *coro)
808	{	834	{
809	/*	835	/*
810	* emulate part of the perl startup here.	836	* emulate part of the perl startup here.
811	*/	837	*/
…		…
835	PL_rs = newSVsv (GvSV (irsgv));	861	PL_rs = newSVsv (GvSV (irsgv));
836	PL_defoutgv = (GV *)SvREFCNT_inc_NN (stdoutgv);	862	PL_defoutgv = (GV *)SvREFCNT_inc_NN (stdoutgv);
837		863
838	{	864	{
839	dSP;	865	dSP;
840	LOGOP myop;	866	UNOP myop;
841		867
842	Zero (&myop, 1, LOGOP);	868	Zero (&myop, 1, UNOP);
843	myop.op_next = Nullop;	869	myop.op_next = Nullop;
844	myop.op_flags = OPf_WANT_VOID;	870	myop.op_flags = OPf_WANT_VOID;
845		871
846	PUSHMARK (SP);	872	PUSHMARK (SP);
847	XPUSHs (sv_2mortal (av_shift (GvAV (PL_defgv))));	873	XPUSHs (sv_2mortal (av_shift (GvAV (PL_defgv))));
…		…
850	PL_op = PL_ppaddr[OP_ENTERSUB](aTHX);	876	PL_op = PL_ppaddr[OP_ENTERSUB](aTHX);
851	SPAGAIN;	877	SPAGAIN;
852	}	878	}
853		879
854	/* 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
855	* 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.
856	* set_stacklevl doesn't do anything on return, but entersub does LEAVE,
857	* so we ENTER here for symmetry
858	*/	882	*/
859	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;
860	}	887	}
861		888
862	static void	889	static void
863	coro_destruct (pTHX_ struct coro *coro)	890	coro_destruct (pTHX_ struct coro *coro)
864	{	891	{
…		…
888		915
889	SvREFCNT_dec (PL_diehook);	916	SvREFCNT_dec (PL_diehook);
890	SvREFCNT_dec (PL_warnhook);	917	SvREFCNT_dec (PL_warnhook);
891		918
892	SvREFCNT_dec (coro->saved_deffh);	919	SvREFCNT_dec (coro->saved_deffh);
893	SvREFCNT_dec (coro->throw);	920	SvREFCNT_dec (coro_throw);
894		921
895	coro_destruct_stacks (aTHX);	922	coro_destruct_stacks (aTHX);
896	}	923	}
897		924
898	static void	925	INLINE void
899	free_coro_mortal (pTHX)	926	free_coro_mortal (pTHX)
900	{	927	{
901	if (expect_true (coro_mortal))	928	if (expect_true (coro_mortal))
902	{	929	{
903	SvREFCNT_dec (coro_mortal);	930	SvREFCNT_dec (coro_mortal);
…		…
908	static int	935	static int
909	runops_trace (pTHX)	936	runops_trace (pTHX)
910	{	937	{
911	COP *oldcop = 0;	938	COP *oldcop = 0;
912	int oldcxix = -2;	939	int oldcxix = -2;
913	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 */
914	coro_cctx *cctx = coro->cctx;	941	coro_cctx *cctx = coro->cctx;
915		942
916	while ((PL_op = CALL_FPTR (PL_op->op_ppaddr) (aTHX)))	943	while ((PL_op = CALL_FPTR (PL_op->op_ppaddr) (aTHX)))
917	{	944	{
918	PERL_ASYNC_CHECK ();	945	PERL_ASYNC_CHECK ();
…		…
1027		1054
1028	TAINT_NOT;	1055	TAINT_NOT;
1029	return 0;	1056	return 0;
1030	}	1057	}
1031		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
1032	/* 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 */
1033	/* _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 */
1034	/* during execution of a perl program */	1068	/* during execution of a perl program */
1035	/* also initialises PL_top_env */	1069	/* also initialises PL_top_env */
1036	static void NOINLINE	1070	static void NOINLINE
1037	cctx_prepare (pTHX_ coro_cctx *cctx)	1071	cctx_prepare (pTHX_ coro_cctx *cctx)
1038	{	1072	{
1039	dSP;	1073	dSP;
1040	LOGOP myop;	1074	UNOP myop;
1041		1075
1042	PL_top_env = &PL_start_env;	1076	PL_top_env = &PL_start_env;
1043		1077
1044	if (cctx->flags & CC_TRACE)	1078	if (cctx->flags & CC_TRACE)
1045	PL_runops = runops_trace;	1079	PL_runops = runops_trace;
1046		1080
1047	Zero (&myop, 1, LOGOP);	1081	Zero (&myop, 1, UNOP);
1048	myop.op_next = PL_op;	1082	myop.op_next = PL_op;
1049	myop.op_flags = OPf_WANT_VOID | OPf_STACKED;	1083	myop.op_flags = OPf_WANT_VOID | OPf_STACKED;
1050		1084
1051	PUSHMARK (SP);	1085	PUSHMARK (SP);
1052	EXTEND (SP, 2);	1086	EXTEND (SP, 2);
1053	PUSHs (sv_2mortal (newSViv (PTR2IV (cctx))));	1087	PUSHs (sv_2mortal (newSViv ((IV)cctx)));
1054	PUSHs ((SV *)get_cv ("Coro::State::_cctx_init", FALSE));	1088	PUSHs ((SV *)get_cv ("Coro::State::_cctx_init", FALSE));
1055	PUTBACK;	1089	PUTBACK;
1056	PL_op = (OP *)&myop;	1090	PL_op = (OP *)&myop;
1057	PL_op = PL_ppaddr[OP_ENTERSUB](aTHX);	1091	PL_op = PL_ppaddr[OP_ENTERSUB](aTHX);
1058	SPAGAIN;	1092	SPAGAIN;
1059	}	1093	}
1060		1094
1061	/* the tail of transfer: execute stuff we can onyl do afetr a transfer */	1095	/* the tail of transfer: execute stuff we can only do after a transfer */
1062	static void	1096	INLINE void
1063	transfer_tail (void)	1097	transfer_tail (pTHX)
1064	{	1098	{
1065	UNLOCK;	1099	free_coro_mortal (aTHX);
1066
1067	if (expect_false (next_has_throw))
1068	{
1069	struct coro *coro = SvSTATE (coro_current);
1070
1071	if (coro->throw)
1072	{
1073	SV *exception = coro->throw;
1074	coro->throw = 0;
1075	sv_setsv (ERRSV, exception);
1076	croak (0);
1077	}
1078	}
1079	}	1100	}
1080		1101
1081	/*	1102	/*
1082	* this is a _very_ stripped down perl interpreter ;)	1103	* this is a _very_ stripped down perl interpreter ;)
1083	*/	1104	*/
…		…
1090	# endif	1111	# endif
1091	#endif	1112	#endif
1092	{	1113	{
1093	dTHX;	1114	dTHX;
1094		1115
1095	/* we now skip the entersub that lead to transfer () */	1116	/* normally we would need to skip the entersub here */
		1117	/* not doing so will re-execute it, which is exactly what we want */
1096	PL_op = PL_op->op_next;	1118	/* PL_nop = PL_nop->op_next */
1097		1119
1098	/* inject a fake subroutine call to cctx_init */	1120	/* inject a fake subroutine call to cctx_init */
1099	cctx_prepare (aTHX_ (coro_cctx *)arg);	1121	cctx_prepare (aTHX_ (coro_cctx *)arg);
1100		1122
1101	/* cctx_run is the alternative tail of transfer () */	1123	/* cctx_run is the alternative tail of transfer() */
1102	transfer_tail ();	1124	transfer_tail (aTHX);
1103		1125
1104	/* 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 */
1105	PL_restartop = PL_op;	1127	PL_restartop = PL_op;
1106	perl_run (PL_curinterp);	1128	perl_run (PL_curinterp);
1107		1129
…		…
1125	++cctx_count;	1147	++cctx_count;
1126	New (0, cctx, 1, coro_cctx);	1148	New (0, cctx, 1, coro_cctx);
1127		1149
1128	cctx->gen = cctx_gen;	1150	cctx->gen = cctx_gen;
1129	cctx->flags = 0;	1151	cctx->flags = 0;
1130	cctx->idle_sp = 0; /* can be accessed by transfer between cctx_run and set_stacklevel */	1152	cctx->idle_sp = 0; /* can be accessed by transfer between cctx_run and set_stacklevel, on throw */
1131		1153
1132	return cctx;	1154	return cctx;
1133	}	1155	}
1134		1156
1135	/* create a new cctx only suitable as source */	1157	/* create a new cctx only suitable as source */
…		…
1157	/* mmap supposedly does allocate-on-write for us */	1179	/* mmap supposedly does allocate-on-write for us */
1158	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);
1159		1181
1160	if (cctx->sptr != (void *)-1)	1182	if (cctx->sptr != (void *)-1)
1161	{	1183	{
1162	# if CORO_STACKGUARD	1184	#if CORO_STACKGUARD
1163	mprotect (cctx->sptr, CORO_STACKGUARD * PAGESIZE, PROT_NONE);	1185	mprotect (cctx->sptr, CORO_STACKGUARD * PAGESIZE, PROT_NONE);
1164	# endif	1186	#endif
1165	stack_start = CORO_STACKGUARD * PAGESIZE + (char *)cctx->sptr;	1187	stack_start = (char *)cctx->sptr + CORO_STACKGUARD * PAGESIZE;
1166	stack_size = cctx->ssize - CORO_STACKGUARD * PAGESIZE;	1188	stack_size = cctx->ssize - CORO_STACKGUARD * PAGESIZE;
1167	cctx->flags |= CC_MAPPED;	1189	cctx->flags |= CC_MAPPED;
1168	}	1190	}
1169	else	1191	else
1170	#endif	1192	#endif
1171	{	1193	{
1172	cctx->ssize = cctx_stacksize * (long)sizeof (long);	1194	cctx->ssize = cctx_stacksize * (long)sizeof (long);
1173	New (0, cctx->sptr, cctx_stacksize, long);	1195	New (0, cctx->sptr, cctx_stacksize, long);
1174		1196
1175	if (!cctx->sptr)	1197	if (!cctx->sptr)
1176	{	1198	{
1177	perror ("FATAL: unable to allocate stack for coroutine");	1199	perror ("FATAL: unable to allocate stack for coroutine, exiting.");
1178	_exit (EXIT_FAILURE);	1200	_exit (EXIT_FAILURE);
1179	}	1201	}
1180		1202
1181	stack_start = cctx->sptr;	1203	stack_start = cctx->sptr;
1182	stack_size = cctx->ssize;	1204	stack_size = cctx->ssize;
1183	}	1205	}
1184		1206
1185	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
1186	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);
1187		1212
1188	return cctx;	1213	return cctx;
1189	}	1214	}
1190		1215
…		…
1198	coro_destroy (&cctx->cctx);	1223	coro_destroy (&cctx->cctx);
1199		1224
1200	/* coro_transfer creates new, empty cctx's */	1225	/* coro_transfer creates new, empty cctx's */
1201	if (cctx->sptr)	1226	if (cctx->sptr)
1202	{	1227	{
1203	#if CORO_USE_VALGRIND	1228	#if CORO_USE_VALGRIND
1204	VALGRIND_STACK_DEREGISTER (cctx->valgrind_id);	1229	VALGRIND_STACK_DEREGISTER (cctx->valgrind_id);
1205	#endif	1230	#endif
1206		1231
1207	#if HAVE_MMAP	1232	#if HAVE_MMAP
1208	if (cctx->flags & CC_MAPPED)	1233	if (cctx->flags & CC_MAPPED)
1209	munmap (cctx->sptr, cctx->ssize);	1234	munmap (cctx->sptr, cctx->ssize);
1210	else	1235	else
…		…
1237	}	1262	}
1238		1263
1239	static void	1264	static void
1240	cctx_put (coro_cctx *cctx)	1265	cctx_put (coro_cctx *cctx)
1241	{	1266	{
1242	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));
1243		1268
1244	/* free another cctx if overlimit */	1269	/* free another cctx if overlimit */
1245	if (expect_false (cctx_idle >= cctx_max_idle))	1270	if (expect_false (cctx_idle >= cctx_max_idle))
1246	{	1271	{
1247	coro_cctx *first = cctx_first;	1272	coro_cctx *first = cctx_first;
…		…
1259	/** coroutine switching *****************************************************/	1284	/** coroutine switching *****************************************************/
1260		1285
1261	static void	1286	static void
1262	transfer_check (pTHX_ struct coro *prev, struct coro *next)	1287	transfer_check (pTHX_ struct coro *prev, struct coro *next)
1263	{	1288	{
		1289	/* TODO: throwing up here is considered harmful */
		1290
1264	if (expect_true (prev != next))	1291	if (expect_true (prev != next))
1265	{	1292	{
1266	if (expect_false (!(prev->flags & (CF_RUNNING | CF_NEW))))	1293	if (expect_false (!(prev->flags & (CF_RUNNING | CF_NEW))))
1267	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,");
1268		1295
1269	if (expect_false (next->flags & CF_RUNNING))	1296	if (expect_false (next->flags & CF_RUNNING))
1270	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,");
1271		1298
1272	if (expect_false (next->flags & CF_DESTROYED))	1299	if (expect_false (next->flags & CF_DESTROYED))
1273	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,");
1274		1301
1275	#if !PERL_VERSION_ATLEAST (5,10,0)	1302	#if !PERL_VERSION_ATLEAST (5,10,0)
1276	if (expect_false (PL_lex_state != LEX_NOTPARSING))	1303	if (expect_false (PL_lex_state != LEX_NOTPARSING))
1277	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,");
1278	#endif	1305	#endif
1279	}	1306	}
1280	}	1307	}
1281		1308
1282	/* always use the TRANSFER macro */	1309	/* always use the TRANSFER macro */
1283	static void NOINLINE	1310	static void NOINLINE /* noinline so we have a fixed stackframe */
1284	transfer (pTHX_ struct coro *prev, struct coro *next, int force_cctx)	1311	transfer (pTHX_ struct coro *prev, struct coro *next, int force_cctx)
1285	{	1312	{
1286	dSTACKLEVEL;	1313	dSTACKLEVEL;
1287		1314
1288	/* sometimes transfer is only called to set idle_sp */	1315	/* sometimes transfer is only called to set idle_sp */
1289	if (expect_false (!next))	1316	if (expect_false (!next))
1290	{	1317	{
1291	((coro_cctx *)prev)->idle_sp = STACKLEVEL;	1318	((coro_cctx *)prev)->idle_sp = (void *)stacklevel;
1292	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 */
1293	}	1320	}
1294	else if (expect_true (prev != next))	1321	else if (expect_true (prev != next))
1295	{	1322	{
1296	coro_cctx *prev__cctx;	1323	coro_cctx *prev__cctx;
…		…
1303	prev->flags |= CF_RUNNING;	1330	prev->flags |= CF_RUNNING;
1304	}	1331	}
1305		1332
1306	prev->flags &= ~CF_RUNNING;	1333	prev->flags &= ~CF_RUNNING;
1307	next->flags |= CF_RUNNING;	1334	next->flags |= CF_RUNNING;
1308
1309	LOCK;
1310		1335
1311	/* first get rid of the old state */	1336	/* first get rid of the old state */
1312	save_perl (aTHX_ prev);	1337	save_perl (aTHX_ prev);
1313		1338
1314	if (expect_false (next->flags & CF_NEW))	1339	if (expect_false (next->flags & CF_NEW))
…		…
1321	else	1346	else
1322	load_perl (aTHX_ next);	1347	load_perl (aTHX_ next);
1323		1348
1324	prev__cctx = prev->cctx;	1349	prev__cctx = prev->cctx;
1325		1350
1326	if (prev__cctx->idle_sp == STACKLEVEL) asm volatile("");//D
1327
1328	/* possibly "free" the cctx */	1351	/* possibly untie and reuse the cctx */
1329	if (expect_true (	1352	if (expect_true (
1330	prev__cctx->idle_sp == STACKLEVEL	1353	prev__cctx->idle_sp == (void *)stacklevel
1331	&& !(prev__cctx->flags & CC_TRACE)	1354	&& !(prev__cctx->flags & CC_TRACE)
1332	&& !force_cctx	1355	&& !force_cctx
1333	))	1356	))
1334	{	1357	{
1335	/* 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 */
1336	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));
1337		1360
1338	prev->cctx = 0;	1361	prev->cctx = 0;
1339		1362
1340	/* 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 */
1341	/* 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 */
…		…
1348		1371
1349	++next->usecount;	1372	++next->usecount;
1350		1373
1351	if (expect_true (!next->cctx))	1374	if (expect_true (!next->cctx))
1352	next->cctx = cctx_get (aTHX);	1375	next->cctx = cctx_get (aTHX);
1353
1354	next_has_throw = !!next->throw;
1355		1376
1356	if (expect_false (prev__cctx != next->cctx))	1377	if (expect_false (prev__cctx != next->cctx))
1357	{	1378	{
1358	prev__cctx->top_env = PL_top_env;	1379	prev__cctx->top_env = PL_top_env;
1359	PL_top_env = next->cctx->top_env;	1380	PL_top_env = next->cctx->top_env;
1360	coro_transfer (&prev__cctx->cctx, &next->cctx->cctx);	1381	coro_transfer (&prev__cctx->cctx, &next->cctx->cctx);
1361	}	1382	}
1362		1383
1363	free_coro_mortal (aTHX);
1364	UNLOCK;
1365
1366	transfer_tail ();	1384	transfer_tail (aTHX);
1367	}	1385	}
1368	}	1386	}
1369
1370	struct transfer_args
1371	{
1372	struct coro *prev, *next;
1373	};
1374		1387
1375	#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))
1376	#define TRANSFER_CHECK(ta) transfer_check (aTHX_ (ta).prev, (ta).next)	1389	#define TRANSFER_CHECK(ta) transfer_check (aTHX_ (ta).prev, (ta).next)
1377		1390
1378	/** high level stuff ********************************************************/	1391	/** high level stuff ********************************************************/
…		…
1380	static int	1393	static int
1381	coro_state_destroy (pTHX_ struct coro *coro)	1394	coro_state_destroy (pTHX_ struct coro *coro)
1382	{	1395	{
1383	if (coro->flags & CF_DESTROYED)	1396	if (coro->flags & CF_DESTROYED)
1384	return 0;	1397	return 0;
		1398
		1399	if (coro->on_destroy)
		1400	coro->on_destroy (aTHX_ coro);
1385		1401
1386	coro->flags |= CF_DESTROYED;	1402	coro->flags |= CF_DESTROYED;
1387		1403
1388	if (coro->flags & CF_READY)	1404	if (coro->flags & CF_READY)
1389	{	1405	{
1390	/* reduce nready, as destroying a ready coro effectively unreadies it */	1406	/* reduce nready, as destroying a ready coro effectively unreadies it */
1391	/* alternative: look through all ready queues and remove the coro */	1407	/* alternative: look through all ready queues and remove the coro */
1392	LOCK;
1393	--coro_nready;	1408	--coro_nready;
1394	UNLOCK;
1395	}	1409	}
1396	else	1410	else
1397	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 */
1398		1412
1399	if (coro->mainstack && coro->mainstack != main_mainstack)	1413	if (coro->mainstack && coro->mainstack != main_mainstack)
1400	{	1414	{
1401	struct coro temp;	1415	struct coro temp;
1402		1416
1403	if (coro->flags & CF_RUNNING)	1417	assert (("FATAL: tried to destroy currently running coroutine (please report)", !(coro->flags & CF_RUNNING)));
1404	croak ("FATAL: tried to destroy currently running coroutine");
1405		1418
1406	save_perl (aTHX_ &temp);	1419	save_perl (aTHX_ &temp);
1407	load_perl (aTHX_ coro);	1420	load_perl (aTHX_ coro);
1408		1421
1409	coro_destruct (aTHX_ coro);	1422	coro_destruct (aTHX_ coro);
…		…
1460	# define MGf_DUP 0	1473	# define MGf_DUP 0
1461	#endif	1474	#endif
1462	};	1475	};
1463		1476
1464	static void	1477	static void
1465	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)
1466	{	1479	{
1467	ta->prev = SvSTATE (prev_sv);	1480	ta->prev = SvSTATE (prev_sv);
1468	ta->next = SvSTATE (next_sv);	1481	ta->next = SvSTATE (next_sv);
1469	TRANSFER_CHECK (*ta);	1482	TRANSFER_CHECK (*ta);
1470	}	1483	}
1471		1484
1472	static void	1485	static void
1473	api_transfer (SV *prev_sv, SV *next_sv)	1486	api_transfer (pTHX_ SV *prev_sv, SV *next_sv)
1474	{	1487	{
1475	dTHX;
1476	struct transfer_args ta;	1488	struct coro_transfer_args ta;
1477		1489
1478	prepare_transfer (aTHX_ &ta, prev_sv, next_sv);	1490	prepare_transfer (aTHX_ &ta, prev_sv, next_sv);
1479	TRANSFER (ta, 1);	1491	TRANSFER (ta, 1);
1480	}	1492	}
1481		1493
1482	/** Coro ********************************************************************/	1494	/** Coro ********************************************************************/
1483		1495
1484	static void	1496	INLINE void
1485	coro_enq (pTHX_ SV *coro_sv)	1497	coro_enq (pTHX_ struct coro *coro)
1486	{	1498	{
1487	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));
1488	}	1500	}
1489		1501
1490	static SV *	1502	INLINE SV *
1491	coro_deq (pTHX)	1503	coro_deq (pTHX)
1492	{	1504	{
1493	int prio;	1505	int prio;
1494		1506
1495	for (prio = PRIO_MAX - PRIO_MIN + 1; --prio >= 0; )	1507	for (prio = PRIO_MAX - PRIO_MIN + 1; --prio >= 0; )
…		…
1498		1510
1499	return 0;	1511	return 0;
1500	}	1512	}
1501		1513
1502	static int	1514	static int
1503	api_ready (SV *coro_sv)	1515	api_ready (pTHX_ SV *coro_sv)
1504	{	1516	{
1505	dTHX;
1506	struct coro *coro;	1517	struct coro *coro;
1507	SV *sv_hook;	1518	SV *sv_hook;
1508	void (*xs_hook)(void);	1519	void (*xs_hook)(void);
1509		1520
1510	if (SvROK (coro_sv))	1521	if (SvROK (coro_sv))
…		…
1515	if (coro->flags & CF_READY)	1526	if (coro->flags & CF_READY)
1516	return 0;	1527	return 0;
1517		1528
1518	coro->flags |= CF_READY;	1529	coro->flags |= CF_READY;
1519		1530
1520	LOCK;
1521
1522	sv_hook = coro_nready ? 0 : coro_readyhook;	1531	sv_hook = coro_nready ? 0 : coro_readyhook;
1523	xs_hook = coro_nready ? 0 : coroapi.readyhook;	1532	xs_hook = coro_nready ? 0 : coroapi.readyhook;
1524		1533
1525	coro_enq (aTHX_ SvREFCNT_inc_NN (coro_sv));	1534	coro_enq (aTHX_ coro);
1526	++coro_nready;	1535	++coro_nready;
1527		1536
1528	UNLOCK;
1529
1530	if (sv_hook)	1537	if (sv_hook)
1531	{	1538	{
1532	dSP;	1539	dSP;
1533		1540
1534	ENTER;	1541	ENTER;
…		…
1548		1555
1549	return 1;	1556	return 1;
1550	}	1557	}
1551		1558
1552	static int	1559	static int
1553	api_is_ready (SV *coro_sv)	1560	api_is_ready (pTHX_ SV *coro_sv)
1554	{	1561	{
1555	dTHX;
1556	return !!(SvSTATE (coro_sv)->flags & CF_READY);	1562	return !!(SvSTATE (coro_sv)->flags & CF_READY);
1557	}	1563	}
1558		1564
1559	static void	1565	INLINE void
1560	prepare_schedule (pTHX_ struct transfer_args *ta)	1566	prepare_schedule (pTHX_ struct coro_transfer_args *ta)
1561	{	1567	{
1562	SV *prev_sv, *next_sv;	1568	SV *prev_sv, *next_sv;
1563		1569
1564	for (;;)	1570	for (;;)
1565	{	1571	{
1566	LOCK;
1567	next_sv = coro_deq (aTHX);	1572	next_sv = coro_deq (aTHX);
1568		1573
1569	/* nothing to schedule: call the idle handler */	1574	/* nothing to schedule: call the idle handler */
1570	if (expect_false (!next_sv))	1575	if (expect_false (!next_sv))
1571	{	1576	{
1572	dSP;	1577	dSP;
1573	UNLOCK;
1574		1578
1575	ENTER;	1579	ENTER;
1576	SAVETMPS;	1580	SAVETMPS;
1577		1581
1578	PUSHMARK (SP);	1582	PUSHMARK (SP);
…		…
1583	FREETMPS;	1587	FREETMPS;
1584	LEAVE;	1588	LEAVE;
1585	continue;	1589	continue;
1586	}	1590	}
1587		1591
1588	ta->next = SvSTATE (next_sv);	1592	ta->next = SvSTATE_hv (next_sv);
1589		1593
1590	/* cannot transfer to destroyed coros, skip and look for next */	1594	/* cannot transfer to destroyed coros, skip and look for next */
1591	if (expect_false (ta->next->flags & CF_DESTROYED))	1595	if (expect_false (ta->next->flags & CF_DESTROYED))
1592	{	1596	{
1593	UNLOCK;
1594	SvREFCNT_dec (next_sv);	1597	SvREFCNT_dec (next_sv);
1595	/* coro_nready is already taken care of by destroy */	1598	/* coro_nready has already been taken care of by destroy */
1596	continue;	1599	continue;
1597	}	1600	}
1598		1601
1599	--coro_nready;	1602	--coro_nready;
1600	UNLOCK;
1601	break;	1603	break;
1602	}	1604	}
1603		1605
1604	/* free this only after the transfer */	1606	/* free this only after the transfer */
1605	prev_sv = SvRV (coro_current);	1607	prev_sv = SvRV (coro_current);
1606	ta->prev = SvSTATE (prev_sv);	1608	ta->prev = SvSTATE_hv (prev_sv);
1607	TRANSFER_CHECK (*ta);	1609	TRANSFER_CHECK (*ta);
1608	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));
1609	ta->next->flags &= ~CF_READY;	1611	ta->next->flags &= ~CF_READY;
1610	SvRV_set (coro_current, next_sv);	1612	SvRV_set (coro_current, next_sv);
1611		1613
1612	LOCK;
1613	free_coro_mortal (aTHX);	1614	free_coro_mortal (aTHX);
1614	coro_mortal = prev_sv;	1615	coro_mortal = prev_sv;
1615	UNLOCK;
1616	}	1616	}
1617		1617
1618	static void	1618	INLINE void
1619	prepare_cede (pTHX_ struct transfer_args *ta)	1619	prepare_cede (pTHX_ struct coro_transfer_args *ta)
1620	{	1620	{
1621	api_ready (coro_current);	1621	api_ready (aTHX_ coro_current);
1622	prepare_schedule (aTHX_ ta);	1622	prepare_schedule (aTHX_ ta);
1623	}	1623	}
1624		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
1625	static int	1648	static int
1626	prepare_cede_notself (pTHX_ struct transfer_args *ta)	1649	api_cede (pTHX)
1627	{	1650	{
1628	if (coro_nready)	1651	struct coro_transfer_args ta;
1629	{	1652
1630	SV *prev = SvRV (coro_current);
1631	prepare_schedule (aTHX_ ta);	1653	prepare_cede (aTHX_ &ta);
1632	api_ready (prev);	1654
		1655	if (expect_true (ta.prev != ta.next))
		1656	{
		1657	TRANSFER (ta, 1);
1633	return 1;	1658	return 1;
1634	}	1659	}
1635	else	1660	else
1636	return 0;	1661	return 0;
1637	}	1662	}
1638		1663
1639	static void
1640	api_schedule (void)
1641	{
1642	dTHX;
1643	struct transfer_args ta;
1644
1645	prepare_schedule (aTHX_ &ta);
1646	TRANSFER (ta, 1);
1647	}
1648
1649	static int	1664	static int
1650	api_cede (void)	1665	api_cede_notself (pTHX)
1651	{	1666	{
1652	dTHX;	1667	if (coro_nready)
		1668	{
1653	struct transfer_args ta;	1669	struct coro_transfer_args ta;
1654		1670
1655	prepare_cede (aTHX_ &ta);	1671	prepare_cede_notself (aTHX_ &ta);
1656
1657	if (expect_true (ta.prev != ta.next))
1658	{
1659	TRANSFER (ta, 1);	1672	TRANSFER (ta, 1);
1660	return 1;	1673	return 1;
1661	}	1674	}
1662	else	1675	else
1663	return 0;	1676	return 0;
1664	}	1677	}
1665		1678
1666	static int	1679	static void
1667	api_cede_notself (void)
1668	{
1669	dTHX;
1670	struct transfer_args ta;
1671
1672	if (prepare_cede_notself (aTHX_ &ta))
1673	{
1674	TRANSFER (ta, 1);
1675	return 1;
1676	}
1677	else
1678	return 0;
1679	}
1680
1681	static void
1682	api_trace (SV *coro_sv, int flags)	1680	api_trace (pTHX_ SV *coro_sv, int flags)
1683	{	1681	{
1684	dTHX;
1685	struct coro *coro = SvSTATE (coro_sv);	1682	struct coro *coro = SvSTATE (coro_sv);
1686		1683
1687	if (flags & CC_TRACE)	1684	if (flags & CC_TRACE)
1688	{	1685	{
1689	if (!coro->cctx)	1686	if (!coro->cctx)
1690	coro->cctx = cctx_new_run ();	1687	coro->cctx = cctx_new_run ();
1691	else if (!(coro->cctx->flags & CC_TRACE))	1688	else if (!(coro->cctx->flags & CC_TRACE))
1692	croak ("cannot enable tracing on coroutine with custom stack");	1689	croak ("cannot enable tracing on coroutine with custom stack,");
1693		1690
1694	coro->cctx->flags |= CC_NOREUSE | (flags & (CC_TRACE | CC_TRACE_ALL));	1691	coro->cctx->flags |= CC_NOREUSE | (flags & (CC_TRACE | CC_TRACE_ALL));
1695	}	1692	}
1696	else if (coro->cctx && coro->cctx->flags & CC_TRACE)	1693	else if (coro->cctx && coro->cctx->flags & CC_TRACE)
1697	{	1694	{
…		…
1702	else	1699	else
1703	coro->slot->runops = RUNOPS_DEFAULT;	1700	coro->slot->runops = RUNOPS_DEFAULT;
1704	}	1701	}
1705	}	1702	}
1706		1703
1707	#if 0
1708	static int
1709	coro_gensub_free (pTHX_ SV *sv, MAGIC *mg)
1710	{
1711	AV *padlist;
1712	AV *av = (AV *)mg->mg_obj;
1713
1714	abort ();
1715
1716	return 0;
1717	}
1718
1719	static MGVTBL coro_gensub_vtbl = {
1720	0, 0, 0, 0,
1721	coro_gensub_free
1722	};
1723	#endif
1724
1725	/*****************************************************************************/	1704	/*****************************************************************************/
1726	/* PerlIO::cede */	1705	/* PerlIO::cede */
1727		1706
1728	typedef struct	1707	typedef struct
1729	{	1708	{
…		…
1756	PerlIOCede *self = PerlIOSelf (f, PerlIOCede);	1735	PerlIOCede *self = PerlIOSelf (f, PerlIOCede);
1757	double now = nvtime ();	1736	double now = nvtime ();
1758		1737
1759	if (now >= self->next)	1738	if (now >= self->next)
1760	{	1739	{
1761	api_cede ();	1740	api_cede (aTHX);
1762	self->next = now + self->every;	1741	self->next = now + self->every;
1763	}	1742	}
1764		1743
1765	return PerlIOBuf_flush (aTHX_ f);	1744	return PerlIOBuf_flush (aTHX_ f);
1766	}	1745	}
…		…
1795	PerlIOBuf_get_ptr,	1774	PerlIOBuf_get_ptr,
1796	PerlIOBuf_get_cnt,	1775	PerlIOBuf_get_cnt,
1797	PerlIOBuf_set_ptrcnt,	1776	PerlIOBuf_set_ptrcnt,
1798	};	1777	};
1799		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 (PL_op->op_flags & OPf_STACKED)
		1988	{
		1989	if (items > slf_arga)
		1990	{
		1991	slf_arga = items;
		1992	free (slf_argv);
		1993	slf_argv = malloc (slf_arga * sizeof (SV *));
		1994	}
		1995
		1996	slf_argc = items;
		1997
		1998	for (i = 0; i < items; ++i)
		1999	slf_argv [i] = SvREFCNT_inc (arg [i]);
		2000	}
		2001	else
		2002	slf_argc = 0;
		2003
		2004	PL_op->op_ppaddr = pp_slf;
		2005	PL_op->op_type = OP_CUSTOM; /* maybe we should leave it at entersub? */
		2006
		2007	PL_op = (OP *)&slf_restore;
		2008	}
		2009
		2010	/*****************************************************************************/
		2011
		2012	static void
		2013	coro_semaphore_adjust (pTHX_ AV *av, IV adjust)
		2014	{
		2015	SV *count_sv = AvARRAY (av)[0];
		2016	IV count = SvIVX (count_sv);
		2017
		2018	count += adjust;
		2019	SvIVX (count_sv) = count;
		2020
		2021	/* now wake up as many waiters as are expected to lock */
		2022	while (count > 0 && AvFILLp (av) > 0)
		2023	{
		2024	SV *cb;
		2025
		2026	/* swap first two elements so we can shift a waiter */
		2027	AvARRAY (av)[0] = AvARRAY (av)[1];
		2028	AvARRAY (av)[1] = count_sv;
		2029	cb = av_shift (av);
		2030
		2031	if (SvOBJECT (cb))
		2032	api_ready (aTHX_ cb);
		2033	else
		2034	croak ("callbacks not yet supported");
		2035
		2036	SvREFCNT_dec (cb);
		2037
		2038	--count;
		2039	}
		2040	}
		2041
		2042	static void
		2043	coro_semaphore_on_destroy (pTHX_ struct coro *coro)
		2044	{
		2045	/* call $sem->adjust (0) to possibly wake up some other waiters */
		2046	coro_semaphore_adjust (aTHX_ (AV *)coro->slf_frame.data, 0);
		2047	}
		2048
		2049	static int
		2050	slf_check_semaphore_down (pTHX_ struct CoroSLF *frame)
		2051	{
		2052	AV *av = (AV *)frame->data;
		2053	SV *count_sv = AvARRAY (av)[0];
		2054
		2055	if (SvIVX (count_sv) > 0)
		2056	{
		2057	SvSTATE_current->on_destroy = 0;
		2058	SvIVX (count_sv) = SvIVX (count_sv) - 1;
		2059	return 0;
		2060	}
		2061	else
		2062	{
		2063	int i;
		2064	/* if we were woken up but can't down, we look through the whole */
		2065	/* waiters list and only add us if we aren't in there already */
		2066	/* this avoids some degenerate memory usage cases */
		2067
		2068	for (i = 1; i <= AvFILLp (av); ++i)
		2069	if (AvARRAY (av)[i] == SvRV (coro_current))
		2070	return 1;
		2071
		2072	av_push (av, SvREFCNT_inc (SvRV (coro_current)));
		2073	return 1;
		2074	}
		2075	}
		2076
		2077	static void
		2078	slf_init_semaphore_down (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		2079	{
		2080	AV *av = (AV *)SvRV (arg [0]);
		2081
		2082	if (SvIVX (AvARRAY (av)[0]) > 0)
		2083	{
		2084	frame->data = (void *)av;
		2085	frame->prepare = prepare_nop;
		2086	SvSTATE_current->on_destroy = coro_semaphore_on_destroy;
		2087	}
		2088	else
		2089	{
		2090	av_push (av, SvREFCNT_inc (SvRV (coro_current)));
		2091
		2092	frame->data = (void *)sv_2mortal (SvREFCNT_inc ((SV *)av));
		2093	frame->prepare = prepare_schedule;
		2094
		2095	/* to avoid race conditions when a woken-up coro gets terminated */
		2096	/* we arrange for a temporary on_destroy that calls adjust (0) */
		2097	assert (!SvSTATE_current->on_destroy);//D
		2098	SvSTATE_current->on_destroy = coro_semaphore_on_destroy;
		2099	}
		2100
		2101	frame->check = slf_check_semaphore_down;
		2102
		2103	}
		2104
		2105	/*****************************************************************************/
		2106
		2107	#define GENSUB_ARG CvXSUBANY (cv).any_ptr
		2108
		2109	/* create a closure from XS, returns a code reference */
		2110	/* the arg can be accessed via GENSUB_ARG from the callback */
		2111	/* the callback must use dXSARGS/XSRETURN */
		2112	static SV *
		2113	gensub (pTHX_ void (*xsub)(pTHX_ CV *), void *arg)
		2114	{
		2115	CV *cv = (CV *)NEWSV (0, 0);
		2116
		2117	sv_upgrade ((SV *)cv, SVt_PVCV);
		2118
		2119	CvANON_on (cv);
		2120	CvISXSUB_on (cv);
		2121	CvXSUB (cv) = xsub;
		2122	GENSUB_ARG = arg;
		2123
		2124	return newRV_noinc ((SV *)cv);
		2125	}
		2126
		2127	/*****************************************************************************/
1800		2128
1801	MODULE = Coro::State PACKAGE = Coro::State PREFIX = api_	2129	MODULE = Coro::State PACKAGE = Coro::State PREFIX = api_
1802		2130
1803	PROTOTYPES: DISABLE	2131	PROTOTYPES: DISABLE
1804		2132
1805	BOOT:	2133	BOOT:
1806	{	2134	{
1807	#ifdef USE_ITHREADS	2135	#ifdef USE_ITHREADS
1808	MUTEX_INIT (&coro_lock);
1809	# if CORO_PTHREAD	2136	# if CORO_PTHREAD
1810	coro_thx = PERL_GET_CONTEXT;	2137	coro_thx = PERL_GET_CONTEXT;
1811	# endif	2138	# endif
1812	#endif	2139	#endif
1813	BOOT_PAGESIZE;	2140	BOOT_PAGESIZE;
…		…
1834	main_top_env = PL_top_env;	2161	main_top_env = PL_top_env;
1835		2162
1836	while (main_top_env->je_prev)	2163	while (main_top_env->je_prev)
1837	main_top_env = main_top_env->je_prev;	2164	main_top_env = main_top_env->je_prev;
1838		2165
		2166	{
		2167	SV *slf = sv_2mortal (newSViv (PTR2IV (pp_slf)));
		2168
		2169	if (!PL_custom_op_names) PL_custom_op_names = newHV ();
		2170	hv_store_ent (PL_custom_op_names, slf,
		2171	newSVpv ("coro_slf", 0), 0);
		2172
		2173	if (!PL_custom_op_descs) PL_custom_op_descs = newHV ();
		2174	hv_store_ent (PL_custom_op_descs, slf,
		2175	newSVpv ("coro schedule like function", 0), 0);
		2176	}
		2177
1839	coroapi.ver = CORO_API_VERSION;	2178	coroapi.ver = CORO_API_VERSION;
1840	coroapi.rev = CORO_API_REVISION;	2179	coroapi.rev = CORO_API_REVISION;
		2180
1841	coroapi.transfer = api_transfer;	2181	coroapi.transfer = api_transfer;
		2182
		2183	coroapi.sv_state = SvSTATE_;
		2184	coroapi.execute_slf = api_execute_slf;
		2185	coroapi.prepare_nop = prepare_nop;
		2186	coroapi.prepare_schedule = prepare_schedule;
		2187	coroapi.prepare_cede = prepare_cede;
		2188	coroapi.prepare_cede_notself = prepare_cede_notself;
1842		2189
1843	{	2190	{
1844	SV **svp = hv_fetch (PL_modglobal, "Time::NVtime", 12, 0);	2191	SV **svp = hv_fetch (PL_modglobal, "Time::NVtime", 12, 0);
1845		2192
1846	if (!svp) croak ("Time::HiRes is required");	2193	if (!svp) croak ("Time::HiRes is required");
…		…
1879	av_push (coro->args, newSVsv (ST (i)));	2226	av_push (coro->args, newSVsv (ST (i)));
1880	}	2227	}
1881	OUTPUT:	2228	OUTPUT:
1882	RETVAL	2229	RETVAL
1883		2230
1884	# these not obviously related functions are all rolled into the same xs
1885	# function to increase chances that they all will call transfer with the same
1886	# stack offset
1887	void	2231	void
1888	_set_stacklevel (...)	2232	_set_stacklevel (...)
1889	ALIAS:	2233	CODE:
1890	Coro::State::transfer = 1	2234	CORO_EXECUTE_SLF_XS (slf_init_set_stacklevel);
1891	Coro::schedule = 2
1892	Coro::cede = 3
1893	Coro::cede_notself = 4
1894	CODE:
1895	{
1896	struct transfer_args ta;
1897		2235
1898	PUTBACK;	2236	void
1899	switch (ix)	2237	transfer (...)
1900	{	2238	PROTOTYPE: $$
1901	case 0:	2239	CODE:
1902	ta.prev = (struct coro *)INT2PTR (coro_cctx *, SvIV (ST (0)));	2240	CORO_EXECUTE_SLF_XS (slf_init_transfer);
1903	ta.next = 0;
1904	break;
1905
1906	case 1:
1907	if (items != 2)
1908	croak ("Coro::State::transfer (prev, next) expects two arguments, not %d", items);
1909
1910	prepare_transfer (aTHX_ &ta, ST (0), ST (1));
1911	break;
1912
1913	case 2:
1914	prepare_schedule (aTHX_ &ta);
1915	break;
1916
1917	case 3:
1918	prepare_cede (aTHX_ &ta);
1919	break;
1920
1921	case 4:
1922	if (!prepare_cede_notself (aTHX_ &ta))
1923	XSRETURN_EMPTY;
1924
1925	break;
1926	}
1927	SPAGAIN;
1928
1929	BARRIER;
1930	PUTBACK;
1931	TRANSFER (ta, 0);
1932	SPAGAIN; /* might be the sp of a different coroutine now */
1933	/* be extra careful not to ever do anything after TRANSFER */
1934	}
1935		2241
1936	bool	2242	bool
1937	_destroy (SV *coro_sv)	2243	_destroy (SV *coro_sv)
1938	CODE:	2244	CODE:
1939	RETVAL = coro_state_destroy (aTHX_ SvSTATE (coro_sv));	2245	RETVAL = coro_state_destroy (aTHX_ SvSTATE (coro_sv));
…		…
1946	CODE:	2252	CODE:
1947	_exit (code);	2253	_exit (code);
1948		2254
1949	int	2255	int
1950	cctx_stacksize (int new_stacksize = 0)	2256	cctx_stacksize (int new_stacksize = 0)
		2257	PROTOTYPE: ;$
1951	CODE:	2258	CODE:
1952	RETVAL = cctx_stacksize;	2259	RETVAL = cctx_stacksize;
1953	if (new_stacksize)	2260	if (new_stacksize)
1954	{	2261	{
1955	cctx_stacksize = new_stacksize;	2262	cctx_stacksize = new_stacksize;
…		…
1958	OUTPUT:	2265	OUTPUT:
1959	RETVAL	2266	RETVAL
1960		2267
1961	int	2268	int
1962	cctx_max_idle (int max_idle = 0)	2269	cctx_max_idle (int max_idle = 0)
		2270	PROTOTYPE: ;$
1963	CODE:	2271	CODE:
1964	RETVAL = cctx_max_idle;	2272	RETVAL = cctx_max_idle;
1965	if (max_idle > 1)	2273	if (max_idle > 1)
1966	cctx_max_idle = max_idle;	2274	cctx_max_idle = max_idle;
1967	OUTPUT:	2275	OUTPUT:
1968	RETVAL	2276	RETVAL
1969		2277
1970	int	2278	int
1971	cctx_count ()	2279	cctx_count ()
		2280	PROTOTYPE:
1972	CODE:	2281	CODE:
1973	RETVAL = cctx_count;	2282	RETVAL = cctx_count;
1974	OUTPUT:	2283	OUTPUT:
1975	RETVAL	2284	RETVAL
1976		2285
1977	int	2286	int
1978	cctx_idle ()	2287	cctx_idle ()
		2288	PROTOTYPE:
1979	CODE:	2289	CODE:
1980	RETVAL = cctx_idle;	2290	RETVAL = cctx_idle;
1981	OUTPUT:	2291	OUTPUT:
1982	RETVAL	2292	RETVAL
1983		2293
1984	void	2294	void
1985	list ()	2295	list ()
		2296	PROTOTYPE:
1986	PPCODE:	2297	PPCODE:
1987	{	2298	{
1988	struct coro *coro;	2299	struct coro *coro;
1989	for (coro = coro_first; coro; coro = coro->next)	2300	for (coro = coro_first; coro; coro = coro->next)
1990	if (coro->hv)	2301	if (coro->hv)
…		…
2049	RETVAL = boolSV (coro->flags & ix);	2360	RETVAL = boolSV (coro->flags & ix);
2050	OUTPUT:	2361	OUTPUT:
2051	RETVAL	2362	RETVAL
2052		2363
2053	void	2364	void
		2365	throw (Coro::State self, SV *throw = &PL_sv_undef)
		2366	PROTOTYPE: $;$
		2367	CODE:
		2368	{
		2369	struct coro *current = SvSTATE_current;
		2370	SV **throwp = self == current ? &coro_throw : &self->throw;
		2371	SvREFCNT_dec (*throwp);
		2372	*throwp = SvOK (throw) ? newSVsv (throw) : 0;
		2373	}
		2374
		2375	void
2054	api_trace (SV *coro, int flags = CC_TRACE | CC_TRACE_SUB)	2376	api_trace (SV *coro, int flags = CC_TRACE | CC_TRACE_SUB)
		2377	PROTOTYPE: $;$
		2378	C_ARGS: aTHX_ coro, flags
2055		2379
2056	SV *	2380	SV *
2057	has_cctx (Coro::State coro)	2381	has_cctx (Coro::State coro)
2058	PROTOTYPE: $	2382	PROTOTYPE: $
2059	CODE:	2383	CODE:
…		…
2083	OUTPUT:	2407	OUTPUT:
2084	RETVAL	2408	RETVAL
2085		2409
2086	void	2410	void
2087	force_cctx ()	2411	force_cctx ()
		2412	PROTOTYPE:
2088	CODE:	2413	CODE:
2089	struct coro *coro = SvSTATE (coro_current);
2090	coro->cctx->idle_sp = 0;	2414	SvSTATE_current->cctx->idle_sp = 0;
2091		2415
2092	void	2416	void
2093	swap_defsv (Coro::State self)	2417	swap_defsv (Coro::State self)
2094	PROTOTYPE: $	2418	PROTOTYPE: $
2095	ALIAS:	2419	ALIAS:
2096	swap_defav = 1	2420	swap_defav = 1
2097	CODE:	2421	CODE:
2098	if (!self->slot)	2422	if (!self->slot)
2099	croak ("cannot swap state with coroutine that has no saved state");	2423	croak ("cannot swap state with coroutine that has no saved state,");
2100	else	2424	else
2101	{	2425	{
2102	SV **src = ix ? (SV **)&GvAV (PL_defgv) : &GvSV (PL_defgv);	2426	SV **src = ix ? (SV **)&GvAV (PL_defgv) : &GvSV (PL_defgv);
2103	SV **dst = ix ? (SV **)&self->slot->defav : (SV **)&self->slot->defsv;	2427	SV **dst = ix ? (SV **)&self->slot->defav : (SV **)&self->slot->defsv;
2104		2428
…		…
2129		2453
2130	for (i = PRIO_MAX - PRIO_MIN + 1; i--; )	2454	for (i = PRIO_MAX - PRIO_MIN + 1; i--; )
2131	coro_ready[i] = newAV ();	2455	coro_ready[i] = newAV ();
2132		2456
2133	{	2457	{
2134	SV *sv = perl_get_sv ("Coro::API", TRUE);	2458	SV *sv = coro_get_sv (aTHX_ "Coro::API", TRUE);
2135	perl_get_sv ("Coro::API", TRUE); /* silence 5.10 warning */
2136		2459
2137	coroapi.schedule = api_schedule;	2460	coroapi.schedule = api_schedule;
2138	coroapi.cede = api_cede;	2461	coroapi.cede = api_cede;
2139	coroapi.cede_notself = api_cede_notself;	2462	coroapi.cede_notself = api_cede_notself;
2140	coroapi.ready = api_ready;	2463	coroapi.ready = api_ready;
2141	coroapi.is_ready = api_is_ready;	2464	coroapi.is_ready = api_is_ready;
2142	coroapi.nready = &coro_nready;	2465	coroapi.nready = coro_nready;
2143	coroapi.current = coro_current;	2466	coroapi.current = coro_current;
2144		2467
2145	GCoroAPI = &coroapi;	2468	GCoroAPI = &coroapi;
2146	sv_setiv (sv, (IV)&coroapi);	2469	sv_setiv (sv, (IV)&coroapi);
2147	SvREADONLY_on (sv);	2470	SvREADONLY_on (sv);
2148	}	2471	}
2149	}	2472	}
		2473
		2474	void
		2475	schedule (...)
		2476	CODE:
		2477	CORO_EXECUTE_SLF_XS (slf_init_schedule);
		2478
		2479	void
		2480	cede (...)
		2481	CODE:
		2482	CORO_EXECUTE_SLF_XS (slf_init_cede);
		2483
		2484	void
		2485	cede_notself (...)
		2486	CODE:
		2487	CORO_EXECUTE_SLF_XS (slf_init_cede_notself);
2150		2488
2151	void	2489	void
2152	_set_current (SV *current)	2490	_set_current (SV *current)
2153	PROTOTYPE: $	2491	PROTOTYPE: $
2154	CODE:	2492	CODE:
…		…
2157		2495
2158	void	2496	void
2159	_set_readyhook (SV *hook)	2497	_set_readyhook (SV *hook)
2160	PROTOTYPE: $	2498	PROTOTYPE: $
2161	CODE:	2499	CODE:
2162	LOCK;
2163	SvREFCNT_dec (coro_readyhook);	2500	SvREFCNT_dec (coro_readyhook);
2164	coro_readyhook = SvOK (hook) ? newSVsv (hook) : 0;	2501	coro_readyhook = SvOK (hook) ? newSVsv (hook) : 0;
2165	UNLOCK;
2166		2502
2167	int	2503	int
2168	prio (Coro::State coro, int newprio = 0)	2504	prio (Coro::State coro, int newprio = 0)
		2505	PROTOTYPE: $;$
2169	ALIAS:	2506	ALIAS:
2170	nice = 1	2507	nice = 1
2171	CODE:	2508	CODE:
2172	{	2509	{
2173	RETVAL = coro->prio;	2510	RETVAL = coro->prio;
…		…
2188		2525
2189	SV *	2526	SV *
2190	ready (SV *self)	2527	ready (SV *self)
2191	PROTOTYPE: $	2528	PROTOTYPE: $
2192	CODE:	2529	CODE:
2193	RETVAL = boolSV (api_ready (self));	2530	RETVAL = boolSV (api_ready (aTHX_ self));
2194	OUTPUT:	2531	OUTPUT:
2195	RETVAL	2532	RETVAL
2196		2533
2197	int	2534	int
2198	nready (...)	2535	nready (...)
…		…
2200	CODE:	2537	CODE:
2201	RETVAL = coro_nready;	2538	RETVAL = coro_nready;
2202	OUTPUT:	2539	OUTPUT:
2203	RETVAL	2540	RETVAL
2204		2541
2205	void
2206	throw (Coro::State self, SV *throw = &PL_sv_undef)
2207	PROTOTYPE: $;$
2208	CODE:
2209	SvREFCNT_dec (self->throw);
2210	self->throw = SvOK (throw) ? newSVsv (throw) : 0;
2211
2212	# for async_pool speedup	2542	# for async_pool speedup
2213	void	2543	void
2214	_pool_1 (SV *cb)	2544	_pool_1 (SV *cb)
2215	CODE:	2545	CODE:
2216	{	2546	{
2217	struct coro *coro = SvSTATE (coro_current);
2218	HV *hv = (HV *)SvRV (coro_current);	2547	HV *hv = (HV *)SvRV (coro_current);
		2548	struct coro *coro = SvSTATE_hv ((SV *)hv);
2219	AV *defav = GvAV (PL_defgv);	2549	AV *defav = GvAV (PL_defgv);
2220	SV *invoke = hv_delete (hv, "_invoke", sizeof ("_invoke") - 1, 0);	2550	SV *invoke = hv_delete (hv, "_invoke", sizeof ("_invoke") - 1, 0);
2221	AV *invoke_av;	2551	AV *invoke_av;
2222	int i, len;	2552	int i, len;
2223		2553
…		…
2244	{	2574	{
2245	av_fill (defav, len - 1);	2575	av_fill (defav, len - 1);
2246	for (i = 0; i < len; ++i)	2576	for (i = 0; i < len; ++i)
2247	av_store (defav, i, SvREFCNT_inc_NN (AvARRAY (invoke_av)[i + 1]));	2577	av_store (defav, i, SvREFCNT_inc_NN (AvARRAY (invoke_av)[i + 1]));
2248	}	2578	}
2249
2250	SvREFCNT_dec (invoke);
2251	}	2579	}
2252		2580
2253	void	2581	void
2254	_pool_2 (SV *cb)	2582	_pool_2 (SV *cb)
2255	CODE:	2583	CODE:
2256	{	2584	{
2257	struct coro *coro = SvSTATE (coro_current);	2585	HV *hv = (HV *)SvRV (coro_current);
		2586	struct coro *coro = SvSTATE_hv ((SV *)hv);
2258		2587
2259	sv_setsv (cb, &PL_sv_undef);	2588	sv_setsv (cb, &PL_sv_undef);
2260		2589
2261	SvREFCNT_dec ((SV *)PL_defoutgv); PL_defoutgv = (GV *)coro->saved_deffh;	2590	SvREFCNT_dec ((SV *)PL_defoutgv); PL_defoutgv = (GV *)coro->saved_deffh;
2262	coro->saved_deffh = 0;	2591	coro->saved_deffh = 0;
…		…
2269	SvREFCNT_dec (old);	2598	SvREFCNT_dec (old);
2270	croak ("\3async_pool terminate\2\n");	2599	croak ("\3async_pool terminate\2\n");
2271	}	2600	}
2272		2601
2273	av_clear (GvAV (PL_defgv));	2602	av_clear (GvAV (PL_defgv));
2274	hv_store ((HV *)SvRV (coro_current), "desc", sizeof ("desc") - 1,	2603	hv_store (hv, "desc", sizeof ("desc") - 1,
2275	newSVpvn ("[async_pool idle]", sizeof ("[async_pool idle]") - 1), 0);	2604	newSVpvn ("[async_pool idle]", sizeof ("[async_pool idle]") - 1), 0);
2276		2605
2277	coro->prio = 0;	2606	coro->prio = 0;
2278		2607
2279	if (coro->cctx && (coro->cctx->flags & CC_TRACE))	2608	if (coro->cctx && (coro->cctx->flags & CC_TRACE))
2280	api_trace (coro_current, 0);	2609	api_trace (aTHX_ coro_current, 0);
2281		2610
2282	av_push (av_async_pool, newSVsv (coro_current));	2611	av_push (av_async_pool, newSVsv (coro_current));
2283	}	2612	}
2284
2285	#if 0
2286
2287	void
2288	_generator_call (...)
2289	PROTOTYPE: @
2290	PPCODE:
2291	fprintf (stderr, "call %p\n", CvXSUBANY(cv).any_ptr);
2292	xxxx
2293	abort ();
2294
2295	SV *
2296	gensub (SV *sub, ...)
2297	PROTOTYPE: &;@
2298	CODE:
2299	{
2300	struct coro *coro;
2301	MAGIC *mg;
2302	CV *xcv;
2303	CV *ncv = (CV *)newSV_type (SVt_PVCV);
2304	int i;
2305
2306	CvGV (ncv) = CvGV (cv);
2307	CvFILE (ncv) = CvFILE (cv);
2308
2309	Newz (0, coro, 1, struct coro);
2310	coro->args = newAV ();
2311	coro->flags = CF_NEW;
2312
2313	av_extend (coro->args, items - 1);
2314	for (i = 1; i < items; i++)
2315	av_push (coro->args, newSVsv (ST (i)));
2316
2317	CvISXSUB_on (ncv);
2318	CvXSUBANY (ncv).any_ptr = (void *)coro;
2319
2320	xcv = GvCV (gv_fetchpv ("Coro::_generator_call", 0, SVt_PVCV));
2321
2322	CvXSUB (ncv) = CvXSUB (xcv);
2323	CvANON_on (ncv);
2324
2325	mg = sv_magicext ((SV *)ncv, 0, CORO_MAGIC_type_state, &coro_gensub_vtbl, (char *)coro, 0);
2326	RETVAL = newRV_noinc ((SV *)ncv);
2327	}
2328	OUTPUT:
2329	RETVAL
2330
2331	#endif
2332		2613
2333		2614
2334	MODULE = Coro::State PACKAGE = Coro::AIO	2615	MODULE = Coro::State PACKAGE = Coro::AIO
2335		2616
2336	void	2617	void
2337	_get_state (SV *self)	2618	_get_state (SV *self)
		2619	PROTOTYPE: $
2338	PPCODE:	2620	PPCODE:
2339	{	2621	{
2340	AV *defav = GvAV (PL_defgv);	2622	AV *defav = GvAV (PL_defgv);
2341	AV *av = newAV ();	2623	AV *av = newAV ();
2342	int i;	2624	int i;
…		…
2357		2639
2358	av_push (av, data_sv);	2640	av_push (av, data_sv);
2359		2641
2360	XPUSHs (sv_2mortal (newRV_noinc ((SV *)av)));	2642	XPUSHs (sv_2mortal (newRV_noinc ((SV *)av)));
2361		2643
2362	api_ready (self);	2644	api_ready (aTHX_ self);
2363	}	2645	}
2364		2646
2365	void	2647	void
2366	_set_state (SV *state)	2648	_set_state (SV *state)
2367	PROTOTYPE: $	2649	PROTOTYPE: $
…		…
2385	MODULE = Coro::State PACKAGE = Coro::AnyEvent	2667	MODULE = Coro::State PACKAGE = Coro::AnyEvent
2386		2668
2387	BOOT:	2669	BOOT:
2388	sv_activity = coro_get_sv (aTHX_ "Coro::AnyEvent::ACTIVITY", TRUE);	2670	sv_activity = coro_get_sv (aTHX_ "Coro::AnyEvent::ACTIVITY", TRUE);
2389		2671
2390	SV *	2672	void
2391	_schedule (...)	2673	_schedule (...)
2392	PROTOTYPE: @
2393	CODE:	2674	CODE:
2394	{	2675	{
2395	static int incede;	2676	static int incede;
2396		2677
2397	api_cede_notself ();	2678	api_cede_notself (aTHX);
2398		2679
2399	++incede;	2680	++incede;
2400	while (coro_nready >= incede && api_cede ())	2681	while (coro_nready >= incede && api_cede (aTHX))
2401	;	2682	;
2402		2683
2403	sv_setsv (sv_activity, &PL_sv_undef);	2684	sv_setsv (sv_activity, &PL_sv_undef);
2404	if (coro_nready >= incede)	2685	if (coro_nready >= incede)
2405	{	2686	{
…		…
2415		2696
2416	MODULE = Coro::State PACKAGE = PerlIO::cede	2697	MODULE = Coro::State PACKAGE = PerlIO::cede
2417		2698
2418	BOOT:	2699	BOOT:
2419	PerlIO_define_layer (aTHX_ &PerlIO_cede);	2700	PerlIO_define_layer (aTHX_ &PerlIO_cede);
		2701
		2702	MODULE = Coro::State PACKAGE = Coro::Semaphore
		2703
		2704	SV *
		2705	new (SV *klass, SV *count_ = 0)
		2706	CODE:
		2707	{
		2708	/* a semaphore contains a counter IV in $sem->[0] and any waiters after that */
		2709	AV *av = newAV ();
		2710	av_push (av, newSViv (count_ && SvOK (count_) ? SvIV (count_) : 1));
		2711	RETVAL = sv_bless (newRV_noinc ((SV *)av), GvSTASH (CvGV (cv)));
		2712	}
		2713	OUTPUT:
		2714	RETVAL
		2715
		2716	SV *
		2717	count (SV *self)
		2718	CODE:
		2719	RETVAL = newSVsv (AvARRAY ((AV *)SvRV (self))[0]);
		2720	OUTPUT:
		2721	RETVAL
		2722
		2723	void
		2724	up (SV *self, int adjust = 1)
		2725	ALIAS:
		2726	adjust = 1
		2727	CODE:
		2728	coro_semaphore_adjust (aTHX_ (AV *)SvRV (self), ix ? adjust : 1);
		2729
		2730	void
		2731	down (SV *self)
		2732	CODE:
		2733	CORO_EXECUTE_SLF_XS (slf_init_semaphore_down);
		2734
		2735	void
		2736	try (SV *self)
		2737	PPCODE:
		2738	{
		2739	AV *av = (AV *)SvRV (self);
		2740	SV *count_sv = AvARRAY (av)[0];
		2741	IV count = SvIVX (count_sv);
		2742
		2743	if (count > 0)
		2744	{
		2745	--count;
		2746	SvIVX (count_sv) = count;
		2747	XSRETURN_YES;
		2748	}
		2749	else
		2750	XSRETURN_NO;
		2751	}
		2752
		2753	void
		2754	waiters (SV *self)
		2755	CODE:
		2756	{
		2757	AV *av = (AV *)SvRV (self);
		2758
		2759	if (GIMME_V == G_SCALAR)
		2760	XPUSHs (sv_2mortal (newSVsv (AvARRAY (av)[0])));
		2761	else
		2762	{
		2763	int i;
		2764	EXTEND (SP, AvFILLp (av) + 1 - 1);
		2765	for (i = 1; i <= AvFILLp (av); ++i)
		2766	PUSHs (newSVsv (AvARRAY (av)[i]));
		2767	}
		2768	}
		2769

Diff Legend

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