ViewVC Help

View File | Revision Log | Show Annotations | Download File

/cvs/Coro/Coro/State.xs

Comparing Coro/Coro/State.xs (file contents):
Revision 1.248 by root, Mon Sep 29 12:40:50 2008 UTC vs.
Revision 1.437 by root, Tue Nov 5 15:13:42 2013 UTC

		1	/* this works around a bug in mingw32 providing a non-working setjmp */
		2	#define USE_NO_MINGW_SETJMP_TWO_ARGS
		3
		4	#define NDEBUG 1 /* perl usually disables NDEBUG later */
		5
1	#include "libcoro/coro.c"	6	#include "libcoro/coro.c"
2		7
3	#define PERL_NO_GET_CONTEXT	8	#define PERL_NO_GET_CONTEXT
4	#define PERL_EXT	9	#define PERL_EXT
5		10
6	#include "EXTERN.h"	11	#include "EXTERN.h"
7	#include "perl.h"	12	#include "perl.h"
8	#include "XSUB.h"	13	#include "XSUB.h"
9	#include "perliol.h"	14	#include "perliol.h"
10		15
11	#include "patchlevel.h"	16	#include "schmorp.h"
12		17
		18	#define ECB_NO_THREADS 1
		19	#include "ecb.h"
		20
		21	#include <stddef.h>
13	#include <stdio.h>	22	#include <stdio.h>
14	#include <errno.h>	23	#include <errno.h>
15	#include <assert.h>	24	#include <assert.h>
16		25
17	#ifdef WIN32	26	#ifndef SVs_PADSTALE
		27	# define SVs_PADSTALE 0
		28	#endif
		29
		30	#ifdef PadARRAY
		31	# define NEWPADAPI 1
		32	# define newPADLIST(var) (Newz (0, var, 1, PADLIST), Newx (PadlistARRAY (var), 2, PAD *))
		33	#else
		34	typedef AV PADNAMELIST;
		35	# if !PERL_VERSION_ATLEAST(5,8,0)
		36	typedef AV PADLIST;
		37	typedef AV PAD;
		38	# endif
		39	# define PadlistARRAY(pl) ((PAD **)AvARRAY (pl))
		40	# define PadlistMAX(pl) AvFILLp (pl)
		41	# define PadlistNAMES(pl) (*PadlistARRAY (pl))
		42	# define PadARRAY AvARRAY
		43	# define PadMAX AvFILLp
		44	# define newPADLIST(var) ((var) = newAV (), av_extend (var, 1))
		45	#endif
		46
		47	/* 5.19.something has replaced SVt_BIND by SVt_INVLIST */
		48	/* we just alias it to SVt_IV, as that is sufficient for swap_sv for now */
		49	#if PERL_VERSION_ATLEAST(5,19,0)
		50	# define SVt_BIND SVt_IV
		51	#endif
		52
		53	#if defined(_WIN32)
		54	# undef HAS_GETTIMEOFDAY
18	# undef setjmp	55	# undef setjmp
19	# undef longjmp	56	# undef longjmp
20	# undef _exit	57	# undef _exit
21	# define setjmp _setjmp // deep magic, don't ask	58	# define setjmp _setjmp /* deep magic */
22	#else	59	#else
23	# include <inttypes.h> /* most portable stdint.h */	60	# include <inttypes.h> /* most portable stdint.h */
24	#endif	61	#endif
25		62
26	#ifdef HAVE_MMAP
27	# include <unistd.h>
28	# include <sys/mman.h>
29	# ifndef MAP_ANONYMOUS
30	# ifdef MAP_ANON
31	# define MAP_ANONYMOUS MAP_ANON
32	# else
33	# undef HAVE_MMAP
34	# endif
35	# endif
36	# include <limits.h>
37	# ifndef PAGESIZE
38	# define PAGESIZE pagesize
39	# define BOOT_PAGESIZE pagesize = sysconf (_SC_PAGESIZE)
40	static long pagesize;
41	# else
42	# define BOOT_PAGESIZE (void)0
43	# endif
44	#else
45	# define PAGESIZE 0
46	# define BOOT_PAGESIZE (void)0
47	#endif
48
49	#if CORO_USE_VALGRIND
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
55
56	/* the maximum number of idle cctx that will be pooled */	63	/* the maximum number of idle cctx that will be pooled */
57	#define MAX_IDLE_CCTX 8	64	static int cctx_max_idle = 4;
58		65
59	#define PERL_VERSION_ATLEAST(a,b,c) \	66	#if defined(DEBUGGING) && PERL_VERSION_ATLEAST(5,12,0)
60	(PERL_REVISION > (a) \	67	# define HAS_SCOPESTACK_NAME 1
61	|| (PERL_REVISION == (a) \
62	&& (PERL_VERSION > (b) \
63	|| (PERL_VERSION == (b) && PERLSUBVERSION >= (c)))))
64
65	#if !PERL_VERSION_ATLEAST (5,6,0)
66	# ifndef PL_ppaddr
67	# define PL_ppaddr ppaddr
68	# endif
69	# ifndef call_sv
70	# define call_sv perl_call_sv
71	# endif
72	# ifndef get_sv
73	# define get_sv perl_get_sv
74	# endif
75	# ifndef get_cv
76	# define get_cv perl_get_cv
77	# endif
78	# ifndef IS_PADGV
79	# define IS_PADGV(v) 0
80	# endif
81	# ifndef IS_PADCONST
82	# define IS_PADCONST(v) 0
83	# endif
84	#endif
85
86	/* 5.8.8 */
87	#ifndef GV_NOTQUAL
88	# define GV_NOTQUAL 0
89	#endif
90	#ifndef newSV
91	# define newSV(l) NEWSV(0,l)
92	#endif
93
94	/* 5.11 */
95	#ifndef CxHASARGS
96	# define CxHASARGS(cx) (cx)->blk_sub.hasargs
97	#endif
98
99	/* 5.8.7 */
100	#ifndef SvRV_set
101	# define SvRV_set(s,v) SvRV(s) = (v)
102	#endif
103
104	#if !__i386 && !__x86_64 && !__powerpc && !__m68k && !__alpha && !__mips && !__sparc64
105	# undef CORO_STACKGUARD
106	#endif
107
108	#ifndef CORO_STACKGUARD
109	# define CORO_STACKGUARD 0
110	#endif	68	#endif
111		69
112	/* prefer perl internal functions over our own? */	70	/* prefer perl internal functions over our own? */
113	#ifndef CORO_PREFER_PERL_FUNCTIONS	71	#ifndef CORO_PREFER_PERL_FUNCTIONS
114	# define CORO_PREFER_PERL_FUNCTIONS 0	72	# define CORO_PREFER_PERL_FUNCTIONS 0
115	#endif	73	#endif
116		74
117	/* The next macros try to return the current stack pointer, in an as	75	/* The next macros try to return the current stack pointer, in an as
118	* portable way as possible. */	76	* portable way as possible. */
119	#define dSTACKLEVEL volatile char stacklevel
120	#define STACKLEVEL ((void *)&stacklevel)
121
122	#define IN_DESTRUCT (PL_main_cv == Nullcv)
123
124	#if __GNUC__ >= 3	77	#if __GNUC__ >= 4
125	# define attribute(x) __attribute__(x)	78	# define dSTACKLEVEL int stacklevel_dummy
126	# define BARRIER __asm__ __volatile__ ("" : : : "memory")	79	# define STACKLEVEL __builtin_frame_address (0)
127	# define expect(expr,value) __builtin_expect ((expr),(value))
128	#else	80	#else
129	# define attribute(x)	81	# define dSTACKLEVEL volatile void *stacklevel
130	# define BARRIER	82	# define STACKLEVEL ((void *)&stacklevel)
131	# define expect(expr,value) (expr)
132	#endif	83	#endif
133		84
134	#define expect_false(expr) expect ((expr) != 0, 0)	85	#define IN_DESTRUCT PL_dirty
135	#define expect_true(expr) expect ((expr) != 0, 1)
136
137	#define NOINLINE attribute ((noinline))
138		86
139	#include "CoroAPI.h"	87	#include "CoroAPI.h"
		88	#define GCoroAPI (&coroapi) /* very sneaky */
140		89
141	#ifdef USE_ITHREADS	90	#ifdef USE_ITHREADS
142	static perl_mutex coro_mutex;	91	# if CORO_PTHREAD
143	# define LOCK do { MUTEX_LOCK (&coro_mutex); } while (0)	92	static void *coro_thx;
144	# define UNLOCK do { MUTEX_UNLOCK (&coro_mutex); } while (0)
145	#else
146	# define LOCK (void)0
147	# define UNLOCK (void)0
148	#endif	93	# endif
		94	#endif
149		95
150	/* helper storage struct for Coro::AIO */	96	#ifdef __linux
151	struct io_state	97	# include <time.h> /* for timespec */
152	{	98	# include <syscall.h> /* for SYS_* */
153	int errorno;	99	# ifdef SYS_clock_gettime
154	I32 laststype;	100	# define coro_clock_gettime(id, ts) syscall (SYS_clock_gettime, (id), (ts))
155	int laststatval;	101	# define CORO_CLOCK_MONOTONIC 1
156	Stat_t statcache;	102	# define CORO_CLOCK_THREAD_CPUTIME_ID 3
157	};	103	# endif
		104	#endif
158		105
159	static double (*nvtime)(); /* so why doesn't it take void? */	106	static double (*nvtime)(); /* so why doesn't it take void? */
		107	static void (*u2time)(pTHX_ UV ret[2]);
160		108
		109	/* we hijack an hopefully unused CV flag for our purposes */
		110	#define CVf_SLF 0x4000
		111	static OP *pp_slf (pTHX);
		112	static void slf_destroy (pTHX_ struct coro *coro);
		113
		114	static U32 cctx_gen;
161	static size_t coro_stacksize = CORO_STACKSIZE;	115	static size_t cctx_stacksize = CORO_STACKSIZE;
162	static struct CoroAPI coroapi;	116	static struct CoroAPI coroapi;
163	static AV *main_mainstack; /* used to differentiate between $main and others */	117	static AV *main_mainstack; /* used to differentiate between $main and others */
164	static JMPENV *main_top_env;	118	static JMPENV *main_top_env;
165	static HV *coro_state_stash, *coro_stash;	119	static HV *coro_state_stash, *coro_stash;
166	static volatile SV *coro_mortal; /* will be freed after next transfer */	120	static volatile SV *coro_mortal; /* will be freed/thrown after next transfer */
		121
		122	static AV *av_destroy; /* destruction queue */
		123	static SV *sv_manager; /* the manager coro */
		124	static SV *sv_idle; /* $Coro::idle */
167		125
168	static GV *irsgv; /* $/ */	126	static GV *irsgv; /* $/ */
169	static GV *stdoutgv; /* *STDOUT */	127	static GV *stdoutgv; /* *STDOUT */
170	static SV *rv_diehook;	128	static SV *rv_diehook;
171	static SV *rv_warnhook;	129	static SV *rv_warnhook;
172	static HV *hv_sig; /* %SIG */	130	static HV *hv_sig; /* %SIG */
173		131
174	/* async_pool helper stuff */	132	/* async_pool helper stuff */
175	static SV *sv_pool_rss;	133	static SV *sv_pool_rss;
176	static SV *sv_pool_size;	134	static SV *sv_pool_size;
		135	static SV *sv_async_pool_idle; /* description string */
177	static AV *av_async_pool;	136	static AV *av_async_pool; /* idle pool */
		137	static SV *sv_Coro; /* class string */
		138	static CV *cv_pool_handler;
178		139
179	/* Coro::AnyEvent */	140	/* Coro::AnyEvent */
180	static SV *sv_activity;	141	static SV *sv_activity;
181		142
		143	/* enable processtime/realtime profiling */
		144	static char enable_times;
		145	typedef U32 coro_ts[2];
		146	static coro_ts time_real, time_cpu;
		147	static char times_valid;
		148
182	static struct coro_cctx *cctx_first;	149	static struct coro_cctx *cctx_first;
183	static int cctx_count, cctx_idle;	150	static int cctx_count, cctx_idle;
184		151
185	enum {	152	enum
		153	{
186	CC_MAPPED = 0x01,	154	CC_MAPPED = 0x01,
187	CC_NOREUSE = 0x02, /* throw this away after tracing */	155	CC_NOREUSE = 0x02, /* throw this away after tracing */
188	CC_TRACE = 0x04,	156	CC_TRACE = 0x04,
189	CC_TRACE_SUB = 0x08, /* trace sub calls */	157	CC_TRACE_SUB = 0x08, /* trace sub calls */
190	CC_TRACE_LINE = 0x10, /* trace each statement */	158	CC_TRACE_LINE = 0x10, /* trace each statement */
191	CC_TRACE_ALL = CC_TRACE_SUB | CC_TRACE_LINE,	159	CC_TRACE_ALL = CC_TRACE_SUB | CC_TRACE_LINE,
192	};	160	};
193		161
194	/* this is a structure representing a c-level coroutine */	162	/* this is a structure representing a c-level coroutine */
195	typedef struct coro_cctx {	163	typedef struct coro_cctx
		164	{
196	struct coro_cctx *next;	165	struct coro_cctx *next;
197		166
198	/* the stack */	167	/* the stack */
199	void *sptr;	168	struct coro_stack stack;
200	size_t ssize;
201		169
202	/* cpu state */	170	/* cpu state */
203	void *idle_sp; /* sp of top-level transfer/schedule/cede call */	171	void *idle_sp; /* sp of top-level transfer/schedule/cede call */
		172	#ifndef NDEBUG
204	JMPENV *idle_te; /* same as idle_sp, but for top_env, TODO: remove once stable */	173	JMPENV *idle_te; /* same as idle_sp, but for top_env */
		174	#endif
205	JMPENV *top_env;	175	JMPENV *top_env;
206	coro_context cctx;	176	coro_context cctx;
207		177
		178	U32 gen;
208	#if CORO_USE_VALGRIND	179	#if CORO_USE_VALGRIND
209	int valgrind_id;	180	int valgrind_id;
210	#endif	181	#endif
211	unsigned char flags;	182	unsigned char flags;
212	} coro_cctx;	183	} coro_cctx;
213		184
		185	static coro_cctx *cctx_current; /* the currently running cctx */
		186
		187	/*****************************************************************************/
		188
		189	static MGVTBL coro_state_vtbl;
		190
214	enum {	191	enum
		192	{
215	CF_RUNNING = 0x0001, /* coroutine is running */	193	CF_RUNNING = 0x0001, /* coroutine is running */
216	CF_READY = 0x0002, /* coroutine is ready */	194	CF_READY = 0x0002, /* coroutine is ready */
217	CF_NEW = 0x0004, /* has never been switched to */	195	CF_NEW = 0x0004, /* has never been switched to */
218	CF_DESTROYED = 0x0008, /* coroutine data has been freed */	196	CF_ZOMBIE = 0x0008, /* coroutine data has been freed */
		197	CF_SUSPENDED = 0x0010, /* coroutine can't be scheduled */
		198	CF_NOCANCEL = 0x0020, /* cannot cancel, set slf_frame.data to 1 (hackish) */
219	};	199	};
220		200
221	/* the structure where most of the perl state is stored, overlaid on the cxstack */	201	/* the structure where most of the perl state is stored, overlaid on the cxstack */
222	typedef struct {	202	typedef struct
223	SV *defsv;	203	{
224	AV *defav;
225	SV *errsv;
226	SV *irsgv;
227	#define VAR(name,type) type name;	204	#define VARx(name,expr,type) type name;
228	# include "state.h"	205	#include "state.h"
229	#undef VAR
230	} perl_slots;	206	} perl_slots;
231		207
		208	/* how many context stack entries do we need for perl_slots */
232	#define SLOT_COUNT ((sizeof (perl_slots) + sizeof (PERL_CONTEXT) - 1) / sizeof (PERL_CONTEXT))	209	#define SLOT_COUNT ((sizeof (perl_slots) + sizeof (PERL_CONTEXT) - 1) / sizeof (PERL_CONTEXT))
233		210
234	/* this is a structure representing a perl-level coroutine */	211	/* this is a structure representing a perl-level coroutine */
235	struct coro {	212	struct coro
		213	{
236	/* the c coroutine allocated to this perl coroutine, if any */	214	/* the C coroutine allocated to this perl coroutine, if any */
237	coro_cctx *cctx;	215	coro_cctx *cctx;
238		216
239	/* process data */	217	/* ready queue */
		218	struct coro *next_ready;
		219
		220	/* state data */
		221	struct CoroSLF slf_frame; /* saved slf frame */
240	AV *mainstack;	222	AV *mainstack;
241	perl_slots *slot; /* basically the saved sp */	223	perl_slots *slot; /* basically the saved sp */
242		224
		225	CV *startcv; /* the CV to execute */
243	AV *args; /* data associated with this coroutine (initial args) */	226	AV *args; /* data associated with this coroutine (initial args) */
244	int refcnt; /* coroutines are refcounted, yes */
245	int flags; /* CF_ flags */	227	int flags; /* CF_ flags */
246	HV *hv; /* the perl hash associated with this coro, if any */	228	HV *hv; /* the perl hash associated with this coro, if any */
247		229
248	/* statistics */	230	/* statistics */
249	int usecount; /* number of transfers to this coro */	231	int usecount; /* number of transfers to this coro */
250		232
251	/* coro process data */	233	/* coro process data */
252	int prio;	234	int prio;
253	SV *throw; /* exception to be thrown */	235	SV *except; /* exception to be thrown */
		236	SV *rouse_cb; /* last rouse callback */
		237	AV *on_destroy; /* callbacks or coros to notify on destroy */
		238	AV *status; /* the exit status list */
254		239
255	/* async_pool */	240	/* async_pool */
256	SV *saved_deffh;	241	SV *saved_deffh;
		242	SV *invoke_cb;
		243	AV *invoke_av;
		244
		245	/* on_enter/on_leave */
		246	AV *on_enter;
		247	AV *on_leave;
		248
		249	/* swap_sv */
		250	AV *swap_sv;
		251
		252	/* times */
		253	coro_ts t_cpu, t_real;
257		254
258	/* linked list */	255	/* linked list */
259	struct coro *next, *prev;	256	struct coro *next, *prev;
260	};	257	};
261		258
262	typedef struct coro *Coro__State;	259	typedef struct coro *Coro__State;
263	typedef struct coro *Coro__State_or_hashref;	260	typedef struct coro *Coro__State_or_hashref;
264		261
		262	/* the following variables are effectively part of the perl context */
		263	/* and get copied between struct coro and these variables */
		264	/* the main reason we don't support windows process emulation */
		265	static struct CoroSLF slf_frame; /* the current slf frame */
		266
265	/** Coro ********************************************************************/	267	/** Coro ********************************************************************/
266		268
267	#define PRIO_MAX 3	269	#define CORO_PRIO_MAX 3
268	#define PRIO_HIGH 1	270	#define CORO_PRIO_HIGH 1
269	#define PRIO_NORMAL 0	271	#define CORO_PRIO_NORMAL 0
270	#define PRIO_LOW -1	272	#define CORO_PRIO_LOW -1
271	#define PRIO_IDLE -3	273	#define CORO_PRIO_IDLE -3
272	#define PRIO_MIN -4	274	#define CORO_PRIO_MIN -4
273		275
274	/* for Coro.pm */	276	/* for Coro.pm */
275	static SV *coro_current;	277	static SV *coro_current;
276	static SV *coro_readyhook;	278	static SV *coro_readyhook;
277	static AV *coro_ready [PRIO_MAX-PRIO_MIN+1];	279	static struct coro *coro_ready [CORO_PRIO_MAX - CORO_PRIO_MIN + 1][2]; /* head|tail */
278	static int coro_nready;	280	static CV *cv_coro_run;
279	static struct coro *coro_first;	281	static struct coro *coro_first;
		282	#define coro_nready coroapi.nready
		283
		284	/** JIT *********************************************************************/
		285
		286	#if CORO_JIT
		287	/* APPLE doesn't have mmap though */
		288	#define CORO_JIT_UNIXY (__linux || __FreeBSD__ || __OpenBSD__ || __NetBSD__ || __solaris || __APPLE__)
		289	#ifndef CORO_JIT_TYPE
		290	#if ECB_AMD64 && CORO_JIT_UNIXY
		291	#define CORO_JIT_TYPE "amd64-unix"
		292	#elif __i386 && CORO_JIT_UNIXY
		293	#define CORO_JIT_TYPE "x86-unix"
		294	#endif
		295	#endif
		296	#endif
		297
		298	#if !defined(CORO_JIT_TYPE) || _POSIX_MEMORY_PROTECTION <= 0
		299	#undef CORO_JIT
		300	#endif
		301
		302	#if CORO_JIT
		303	typedef void (*load_save_perl_slots_type)(perl_slots *);
		304	static load_save_perl_slots_type load_perl_slots, save_perl_slots;
		305	#endif
		306
		307	/** Coro::Select ************************************************************/
		308
		309	static OP *(*coro_old_pp_sselect) (pTHX);
		310	static SV *coro_select_select;
		311
		312	/* horrible hack, but if it works... */
		313	static OP *
		314	coro_pp_sselect (pTHX)
		315	{
		316	dSP;
		317	PUSHMARK (SP - 4); /* fake argument list */
		318	XPUSHs (coro_select_select);
		319	PUTBACK;
		320
		321	/* entersub is an UNOP, select a LISTOP... keep your fingers crossed */
		322	PL_op->op_flags |= OPf_STACKED;
		323	PL_op->op_private = 0;
		324	return PL_ppaddr [OP_ENTERSUB](aTHX);
		325	}
		326
		327	/** time stuff **************************************************************/
		328
		329	#ifdef HAS_GETTIMEOFDAY
		330
		331	ecb_inline void
		332	coro_u2time (pTHX_ UV ret[2])
		333	{
		334	struct timeval tv;
		335	gettimeofday (&tv, 0);
		336
		337	ret [0] = tv.tv_sec;
		338	ret [1] = tv.tv_usec;
		339	}
		340
		341	ecb_inline double
		342	coro_nvtime (void)
		343	{
		344	struct timeval tv;
		345	gettimeofday (&tv, 0);
		346
		347	return tv.tv_sec + tv.tv_usec * 1e-6;
		348	}
		349
		350	ecb_inline void
		351	time_init (pTHX)
		352	{
		353	nvtime = coro_nvtime;
		354	u2time = coro_u2time;
		355	}
		356
		357	#else
		358
		359	ecb_inline void
		360	time_init (pTHX)
		361	{
		362	SV **svp;
		363
		364	require_pv ("Time/HiRes.pm");
		365
		366	svp = hv_fetch (PL_modglobal, "Time::NVtime", 12, 0);
		367
		368	if (!svp) croak ("Time::HiRes is required, but missing. Caught");
		369	if (!SvIOK (*svp)) croak ("Time::NVtime isn't a function pointer. Caught");
		370
		371	nvtime = INT2PTR (double (*)(), SvIV (*svp));
		372
		373	svp = hv_fetch (PL_modglobal, "Time::U2time", 12, 0);
		374	u2time = INT2PTR (void (*)(pTHX_ UV ret[2]), SvIV (*svp));
		375	}
		376
		377	#endif
280		378
281	/** lowlevel stuff **********************************************************/	379	/** lowlevel stuff **********************************************************/
282		380
283	static SV *	381	static SV * ecb_noinline
284	coro_get_sv (pTHX_ const char *name, int create)	382	coro_get_sv (pTHX_ const char *name, int create)
285	{	383	{
286	#if PERL_VERSION_ATLEAST (5,10,0)	384	#if PERL_VERSION_ATLEAST (5,10,0)
287	/* silence stupid and wrong 5.10 warning that I am unable to switch off */	385	/* silence stupid and wrong 5.10 warning that I am unable to switch off */
288	get_sv (name, create);	386	get_sv (name, create);
289	#endif	387	#endif
290	return get_sv (name, create);	388	return get_sv (name, create);
291	}	389	}
292		390
293	static AV *	391	static AV * ecb_noinline
294	coro_get_av (pTHX_ const char *name, int create)	392	coro_get_av (pTHX_ const char *name, int create)
295	{	393	{
296	#if PERL_VERSION_ATLEAST (5,10,0)	394	#if PERL_VERSION_ATLEAST (5,10,0)
297	/* silence stupid and wrong 5.10 warning that I am unable to switch off */	395	/* silence stupid and wrong 5.10 warning that I am unable to switch off */
298	get_av (name, create);	396	get_av (name, create);
299	#endif	397	#endif
300	return get_av (name, create);	398	return get_av (name, create);
301	}	399	}
302		400
303	static HV *	401	static HV * ecb_noinline
304	coro_get_hv (pTHX_ const char *name, int create)	402	coro_get_hv (pTHX_ const char *name, int create)
305	{	403	{
306	#if PERL_VERSION_ATLEAST (5,10,0)	404	#if PERL_VERSION_ATLEAST (5,10,0)
307	/* silence stupid and wrong 5.10 warning that I am unable to switch off */	405	/* silence stupid and wrong 5.10 warning that I am unable to switch off */
308	get_hv (name, create);	406	get_hv (name, create);
309	#endif	407	#endif
310	return get_hv (name, create);	408	return get_hv (name, create);
311	}	409	}
312		410
313	static AV *	411	ecb_inline void
		412	coro_times_update (void)
		413	{
		414	#ifdef coro_clock_gettime
		415	struct timespec ts;
		416
		417	ts.tv_sec = ts.tv_nsec = 0;
		418	coro_clock_gettime (CORO_CLOCK_THREAD_CPUTIME_ID, &ts);
		419	time_cpu [0] = ts.tv_sec; time_cpu [1] = ts.tv_nsec;
		420
		421	ts.tv_sec = ts.tv_nsec = 0;
		422	coro_clock_gettime (CORO_CLOCK_MONOTONIC, &ts);
		423	time_real [0] = ts.tv_sec; time_real [1] = ts.tv_nsec;
		424	#else
		425	dTHX;
		426	UV tv[2];
		427
		428	u2time (aTHX_ tv);
		429	time_real [0] = tv [0];
		430	time_real [1] = tv [1] * 1000;
		431	#endif
		432	}
		433
		434	ecb_inline void
		435	coro_times_add (struct coro *c)
		436	{
		437	c->t_real [1] += time_real [1];
		438	if (c->t_real [1] > 1000000000) { c->t_real [1] -= 1000000000; ++c->t_real [0]; }
		439	c->t_real [0] += time_real [0];
		440
		441	c->t_cpu [1] += time_cpu [1];
		442	if (c->t_cpu [1] > 1000000000) { c->t_cpu [1] -= 1000000000; ++c->t_cpu [0]; }
		443	c->t_cpu [0] += time_cpu [0];
		444	}
		445
		446	ecb_inline void
		447	coro_times_sub (struct coro *c)
		448	{
		449	if (c->t_real [1] < time_real [1]) { c->t_real [1] += 1000000000; --c->t_real [0]; }
		450	c->t_real [1] -= time_real [1];
		451	c->t_real [0] -= time_real [0];
		452
		453	if (c->t_cpu [1] < time_cpu [1]) { c->t_cpu [1] += 1000000000; --c->t_cpu [0]; }
		454	c->t_cpu [1] -= time_cpu [1];
		455	c->t_cpu [0] -= time_cpu [0];
		456	}
		457
		458	/*****************************************************************************/
		459	/* magic glue */
		460
		461	#define CORO_MAGIC_type_cv 26
		462	#define CORO_MAGIC_type_state PERL_MAGIC_ext
		463
		464	#define CORO_MAGIC_NN(sv, type) \
		465	(ecb_expect_true (SvMAGIC (sv)->mg_type == type) \
		466	? SvMAGIC (sv) \
		467	: mg_find (sv, type))
		468
		469	#define CORO_MAGIC(sv, type) \
		470	(ecb_expect_true (SvMAGIC (sv)) \
		471	? CORO_MAGIC_NN (sv, type) \
		472	: 0)
		473
		474	#define CORO_MAGIC_cv(cv) CORO_MAGIC (((SV *)(cv)), CORO_MAGIC_type_cv)
		475	#define CORO_MAGIC_state(sv) CORO_MAGIC_NN (((SV *)(sv)), CORO_MAGIC_type_state)
		476
		477	ecb_inline MAGIC *
		478	SvSTATEhv_p (pTHX_ SV *coro)
		479	{
		480	MAGIC *mg;
		481
		482	if (ecb_expect_true (
		483	SvTYPE (coro) == SVt_PVHV
		484	&& (mg = CORO_MAGIC_state (coro))
		485	&& mg->mg_virtual == &coro_state_vtbl
		486	))
		487	return mg;
		488
		489	return 0;
		490	}
		491
		492	ecb_inline struct coro *
		493	SvSTATE_ (pTHX_ SV *coro)
		494	{
		495	MAGIC *mg;
		496
		497	if (SvROK (coro))
		498	coro = SvRV (coro);
		499
		500	mg = SvSTATEhv_p (aTHX_ coro);
		501	if (!mg)
		502	croak ("Coro::State object required");
		503
		504	return (struct coro *)mg->mg_ptr;
		505	}
		506
		507	#define SvSTATE(sv) SvSTATE_ (aTHX_ (sv))
		508
		509	/* faster than SvSTATE, but expects a coroutine hv */
		510	#define SvSTATE_hv(hv) ((struct coro *)CORO_MAGIC_NN ((SV *)hv, CORO_MAGIC_type_state)->mg_ptr)
		511	#define SvSTATE_current SvSTATE_hv (SvRV (coro_current))
		512
		513	/*****************************************************************************/
		514	/* padlist management and caching */
		515
		516	ecb_inline PADLIST *
314	coro_clone_padlist (pTHX_ CV *cv)	517	coro_derive_padlist (pTHX_ CV *cv)
315	{	518	{
316	AV *padlist = CvPADLIST (cv);	519	PADLIST *padlist = CvPADLIST (cv);
317	AV *newpadlist, *newpad;	520	PADLIST *newpadlist;
		521	PAD *newpad;
		522	PADOFFSET off = PadlistMAX (padlist) + 1;
318		523
319	newpadlist = newAV ();	524	#if NEWPADAPI
		525
		526	/* if we had the original CvDEPTH, we might be able to steal the CvDEPTH+1 entry instead */
		527	/* 20131102194744.GA6705@schmorp.de, 20131102195825.2013.qmail@lists-nntp.develooper.com */
		528	while (!PadlistARRAY (padlist)[off - 1])
		529	--off;
		530
		531	Perl_pad_push (aTHX_ padlist, off);
		532
		533	newpad = PadlistARRAY (padlist)[off];
		534	PadlistARRAY (padlist)[off] = 0;
		535
		536	#else
		537
		538	#if PERL_VERSION_ATLEAST (5,10,0)
		539	Perl_pad_push (aTHX_ padlist, off);
		540	#else
		541	Perl_pad_push (aTHX_ padlist, off, 1);
		542	#endif
		543
		544	newpad = PadlistARRAY (padlist)[off];
		545	PadlistMAX (padlist) = off - 1;
		546
		547	#endif
		548
		549	newPADLIST (newpadlist);
		550	#if !PERL_VERSION_ATLEAST(5,15,3)
		551	/* Padlists are AvREAL as of 5.15.3. See perl bug #98092 and perl commit 7d953ba. */
320	AvREAL_off (newpadlist);	552	AvREAL_off (newpadlist);
321	#if PERL_VERSION_ATLEAST (5,10,0)	553	#endif
322	Perl_pad_push (aTHX_ padlist, AvFILLp (padlist) + 1);	554
		555	/* Already extended to 2 elements by newPADLIST. */
		556	PadlistMAX (newpadlist) = 1;
		557	PadlistNAMES (newpadlist) = (PADNAMELIST *)SvREFCNT_inc_NN (PadlistNAMES (padlist));
		558	PadlistARRAY (newpadlist)[1] = newpad;
		559
		560	return newpadlist;
		561	}
		562
		563	ecb_inline void
		564	free_padlist (pTHX_ PADLIST *padlist)
		565	{
		566	/* may be during global destruction */
		567	if (!IN_DESTRUCT)
		568	{
		569	I32 i = PadlistMAX (padlist);
		570
		571	while (i > 0) /* special-case index 0 */
		572	{
		573	/* we try to be extra-careful here */
		574	PAD *pad = PadlistARRAY (padlist)[i--];
		575	I32 j = PadMAX (pad);
		576
		577	while (j >= 0)
		578	SvREFCNT_dec (PadARRAY (pad)[j--]);
		579
		580	PadMAX (pad) = -1;
		581	SvREFCNT_dec (pad);
		582	}
		583
		584	SvREFCNT_dec (PadlistNAMES (padlist));
		585
		586	#if NEWPADAPI
		587	Safefree (PadlistARRAY (padlist));
		588	Safefree (padlist);
323	#else	589	#else
324	Perl_pad_push (aTHX_ padlist, AvFILLp (padlist) + 1, 1);
325	#endif
326	newpad = (AV *)AvARRAY (padlist)[AvFILLp (padlist)];
327	--AvFILLp (padlist);
328
329	av_store (newpadlist, 0, SvREFCNT_inc (*av_fetch (padlist, 0, FALSE)));
330	av_store (newpadlist, 1, (SV *)newpad);
331
332	return newpadlist;
333	}
334
335	static void
336	free_padlist (pTHX_ AV *padlist)
337	{
338	/* may be during global destruction */
339	if (SvREFCNT (padlist))
340	{
341	I32 i = AvFILLp (padlist);	590	AvFILLp (padlist) = -1;
342	while (i >= 0)	591	AvREAL_off (padlist);
343	{
344	SV **svp = av_fetch (padlist, i--, FALSE);
345	if (svp)
346	{
347	SV *sv;
348	while (&PL_sv_undef != (sv = av_pop ((AV *)*svp)))
349	SvREFCNT_dec (sv);
350
351	SvREFCNT_dec (*svp);
352	}
353	}
354
355	SvREFCNT_dec ((SV*)padlist);	592	SvREFCNT_dec ((SV*)padlist);
		593	#endif
356	}	594	}
357	}	595	}
358		596
359	static int	597	static int
360	coro_cv_free (pTHX_ SV *sv, MAGIC *mg)	598	coro_cv_free (pTHX_ SV *sv, MAGIC *mg)
361	{	599	{
362	AV *padlist;	600	PADLIST *padlist;
363	AV *av = (AV *)mg->mg_obj;	601	PADLIST **padlists = (PADLIST **)(mg->mg_ptr + sizeof(size_t));
		602	size_t len = *(size_t *)mg->mg_ptr;
364		603
365	/* casting is fun. */	604	/* perl manages to free our internal AV and _then_ call us */
366	while (&PL_sv_undef != (SV *)(padlist = (AV *)av_pop (av)))	605	if (IN_DESTRUCT)
		606	return 0;
		607
		608	while (len--)
367	free_padlist (aTHX_ padlist);	609	free_padlist (aTHX_ padlists[len]);
368
369	SvREFCNT_dec (av); /* sv_magicext increased the refcount */
370		610
371	return 0;	611	return 0;
372	}	612	}
373
374	#define CORO_MAGIC_type_cv PERL_MAGIC_ext
375	#define CORO_MAGIC_type_state PERL_MAGIC_ext
376		613
377	static MGVTBL coro_cv_vtbl = {	614	static MGVTBL coro_cv_vtbl = {
378	0, 0, 0, 0,	615	0, 0, 0, 0,
379	coro_cv_free	616	coro_cv_free
380	};	617	};
381		618
382	#define CORO_MAGIC(sv,type) \
383	SvMAGIC (sv) \
384	? SvMAGIC (sv)->mg_type == type \
385	? SvMAGIC (sv) \
386	: mg_find (sv, type) \
387	: 0
388
389	#define CORO_MAGIC_cv(cv) CORO_MAGIC (((SV *)(cv)), CORO_MAGIC_type_cv)
390	#define CORO_MAGIC_state(sv) CORO_MAGIC (((SV *)(sv)), CORO_MAGIC_type_state)
391
392	static struct coro *
393	SvSTATE_ (pTHX_ SV *coro)
394	{
395	HV *stash;
396	MAGIC *mg;
397
398	if (SvROK (coro))
399	coro = SvRV (coro);
400
401	if (expect_false (SvTYPE (coro) != SVt_PVHV))
402	croak ("Coro::State object required");
403
404	stash = SvSTASH (coro);
405	if (expect_false (stash != coro_stash && stash != coro_state_stash))
406	{
407	/* very slow, but rare, check */
408	if (!sv_derived_from (sv_2mortal (newRV_inc (coro)), "Coro::State"))
409	croak ("Coro::State object required");
410	}
411
412	mg = CORO_MAGIC_state (coro);
413	return (struct coro *)mg->mg_ptr;
414	}
415
416	#define SvSTATE(sv) SvSTATE_ (aTHX_ (sv))
417
418	/* the next two functions merely cache the padlists */	619	/* the next two functions merely cache the padlists */
419	static void	620	ecb_inline void
420	get_padlist (pTHX_ CV *cv)	621	get_padlist (pTHX_ CV *cv)
421	{	622	{
422	MAGIC *mg = CORO_MAGIC_cv (cv);	623	MAGIC *mg = CORO_MAGIC_cv (cv);
423	AV *av;	624	size_t *lenp;
424		625
425	if (expect_true (mg && AvFILLp ((av = (AV *)mg->mg_obj)) >= 0))	626	if (ecb_expect_true (mg && *(lenp = (size_t *)mg->mg_ptr)))
426	CvPADLIST (cv) = (AV *)AvARRAY (av)[AvFILLp (av)--];	627	CvPADLIST (cv) = ((PADLIST **)(mg->mg_ptr + sizeof(size_t)))[--*lenp];
427	else	628	else
428	{	629	{
429	#if CORO_PREFER_PERL_FUNCTIONS	630	#if CORO_PREFER_PERL_FUNCTIONS
430	/* this is probably cleaner? but also slower! */	631	/* this is probably cleaner? but also slower! */
431	/* in practise, it seems to be less stable */	632	/* in practise, it seems to be less stable */
432	CV *cp = Perl_cv_clone (cv);	633	CV *cp = Perl_cv_clone (aTHX_ cv);
433	CvPADLIST (cv) = CvPADLIST (cp);	634	CvPADLIST (cv) = CvPADLIST (cp);
434	CvPADLIST (cp) = 0;	635	CvPADLIST (cp) = 0;
435	SvREFCNT_dec (cp);	636	SvREFCNT_dec (cp);
436	#else	637	#else
437	CvPADLIST (cv) = coro_clone_padlist (aTHX_ cv);	638	CvPADLIST (cv) = coro_derive_padlist (aTHX_ cv);
438	#endif	639	#endif
439	}	640	}
440	}	641	}
441		642
442	static void	643	ecb_inline void
443	put_padlist (pTHX_ CV *cv)	644	put_padlist (pTHX_ CV *cv)
444	{	645	{
445	MAGIC *mg = CORO_MAGIC_cv (cv);	646	MAGIC *mg = CORO_MAGIC_cv (cv);
446	AV *av;
447		647
448	if (expect_false (!mg))	648	if (ecb_expect_false (!mg))
		649	{
449	mg = sv_magicext ((SV *)cv, (SV *)newAV (), CORO_MAGIC_type_cv, &coro_cv_vtbl, 0, 0);	650	mg = sv_magicext ((SV *)cv, 0, CORO_MAGIC_type_cv, &coro_cv_vtbl, 0, 0);
		651	Newz (0, mg->mg_ptr ,sizeof (size_t) + sizeof (PADLIST *), char);
		652	mg->mg_len = 1; /* so mg_free frees mg_ptr */
		653	}
		654	else
		655	Renew (mg->mg_ptr,
		656	sizeof(size_t) + (*(size_t *)mg->mg_ptr + 1) * sizeof(PADLIST *),
		657	char);
450		658
451	av = (AV *)mg->mg_obj;	659	((PADLIST **)(mg->mg_ptr + sizeof (size_t))) [(*(size_t *)mg->mg_ptr)++] = CvPADLIST (cv);
452
453	if (expect_false (AvFILLp (av) >= AvMAX (av)))
454	av_extend (av, AvMAX (av) + 1);
455
456	AvARRAY (av)[++AvFILLp (av)] = (SV *)CvPADLIST (cv);
457	}	660	}
458		661
459	/** load & save, init *******************************************************/	662	/** load & save, init *******************************************************/
		663
		664	ecb_inline void
		665	swap_sv (SV *a, SV *b)
		666	{
		667	const U32 keep = SVs_PADSTALE | SVs_PADTMP | SVs_PADMY; /* keep these flags */
		668	SV tmp;
		669
		670	/* swap sv_any */
		671	SvANY (&tmp) = SvANY (a); SvANY (a) = SvANY (b); SvANY (b) = SvANY (&tmp);
		672
		673	/* swap sv_flags */
		674	SvFLAGS (&tmp) = SvFLAGS (a);
		675	SvFLAGS (a) = (SvFLAGS (a) & keep) | (SvFLAGS (b ) & ~keep);
		676	SvFLAGS (b) = (SvFLAGS (b) & keep) | (SvFLAGS (&tmp) & ~keep);
		677
		678	#if PERL_VERSION_ATLEAST (5,10,0)
		679	/* perl 5.10 and later complicates this _quite_ a bit, but it also
		680	* is much faster, so no quarrels here. alternatively, we could
		681	* sv_upgrade to avoid this.
		682	*/
		683	{
		684	/* swap sv_u */
		685	tmp.sv_u = a->sv_u; a->sv_u = b->sv_u; b->sv_u = tmp.sv_u;
		686
		687	/* if SvANY points to the head, we need to adjust the pointers,
		688	* as the pointer for a still points to b, and maybe vice versa.
		689	*/
		690	#define svany_in_head(type) \
		691	(((1 << SVt_NULL) | (1 << SVt_BIND) | (1 << SVt_IV) | (1 << SVt_RV)) & (1 << (type)))
		692
		693	if (svany_in_head (SvTYPE (a)))
		694	SvANY (a) = (void *)((PTRV)SvANY (a) - (PTRV)b + (PTRV)a);
		695
		696	if (svany_in_head (SvTYPE (b)))
		697	SvANY (b) = (void *)((PTRV)SvANY (b) - (PTRV)a + (PTRV)b);
		698	}
		699	#endif
		700	}
		701
		702	/* swap sv heads, at least logically */
		703	static void
		704	swap_svs (pTHX_ Coro__State c)
		705	{
		706	int i;
		707
		708	for (i = 0; i <= AvFILLp (c->swap_sv); i += 2)
		709	swap_sv (AvARRAY (c->swap_sv)[i], AvARRAY (c->swap_sv)[i + 1]);
		710	}
		711
		712	#define SWAP_SVS(coro) \
		713	if (ecb_expect_false ((coro)->swap_sv)) \
		714	swap_svs (aTHX_ (coro))
		715
		716	static void
		717	on_enterleave_call (pTHX_ SV *cb);
460		718
461	static void	719	static void
462	load_perl (pTHX_ Coro__State c)	720	load_perl (pTHX_ Coro__State c)
463	{	721	{
464	perl_slots *slot = c->slot;	722	perl_slots *slot = c->slot;
465	c->slot = 0;	723	c->slot = 0;
466		724
467	PL_mainstack = c->mainstack;	725	PL_mainstack = c->mainstack;
468		726
469	GvSV (PL_defgv) = slot->defsv;	727	#if CORO_JIT
470	GvAV (PL_defgv) = slot->defav;	728	load_perl_slots (slot);
471	GvSV (PL_errgv) = slot->errsv;	729	#else
472	GvSV (irsgv) = slot->irsgv;
473
474	#define VAR(name,type) PL_ ## name = slot->name;	730	#define VARx(name,expr,type) expr = slot->name;
475	# include "state.h"	731	#include "state.h"
476	#undef VAR	732	#endif
477		733
478	{	734	{
479	dSP;	735	dSP;
480		736
481	CV *cv;	737	CV *cv;
482		738
483	/* now do the ugly restore mess */	739	/* now do the ugly restore mess */
484	while (expect_true (cv = (CV *)POPs))	740	while (ecb_expect_true (cv = (CV *)POPs))
485	{	741	{
486	put_padlist (aTHX_ cv); /* mark this padlist as available */	742	put_padlist (aTHX_ cv); /* mark this padlist as available */
487	CvDEPTH (cv) = PTR2IV (POPs);	743	CvDEPTH (cv) = PTR2IV (POPs);
488	CvPADLIST (cv) = (AV *)POPs;	744	CvPADLIST (cv) = (PADLIST *)POPs;
489	}	745	}
490		746
491	PUTBACK;	747	PUTBACK;
492	}	748	}
		749
		750	slf_frame = c->slf_frame;
		751	CORO_THROW = c->except;
		752
		753	if (ecb_expect_false (enable_times))
		754	{
		755	if (ecb_expect_false (!times_valid))
		756	coro_times_update ();
		757
		758	coro_times_sub (c);
		759	}
		760
		761	if (ecb_expect_false (c->on_enter))
		762	{
		763	int i;
		764
		765	for (i = 0; i <= AvFILLp (c->on_enter); ++i)
		766	on_enterleave_call (aTHX_ AvARRAY (c->on_enter)[i]);
		767	}
		768
		769	SWAP_SVS (c);
493	}	770	}
494		771
495	static void	772	static void
496	save_perl (pTHX_ Coro__State c)	773	save_perl (pTHX_ Coro__State c)
497	{	774	{
		775	SWAP_SVS (c);
		776
		777	if (ecb_expect_false (c->on_leave))
		778	{
		779	int i;
		780
		781	for (i = AvFILLp (c->on_leave); i >= 0; --i)
		782	on_enterleave_call (aTHX_ AvARRAY (c->on_leave)[i]);
		783	}
		784
		785	times_valid = 0;
		786
		787	if (ecb_expect_false (enable_times))
		788	{
		789	coro_times_update (); times_valid = 1;
		790	coro_times_add (c);
		791	}
		792
		793	c->except = CORO_THROW;
		794	c->slf_frame = slf_frame;
		795
498	{	796	{
499	dSP;	797	dSP;
500	I32 cxix = cxstack_ix;	798	I32 cxix = cxstack_ix;
501	PERL_CONTEXT *ccstk = cxstack;	799	PERL_CONTEXT *ccstk = cxstack;
502	PERL_SI *top_si = PL_curstackinfo;	800	PERL_SI *top_si = PL_curstackinfo;
…		…
508		806
509	XPUSHs (Nullsv);	807	XPUSHs (Nullsv);
510	/* this loop was inspired by pp_caller */	808	/* this loop was inspired by pp_caller */
511	for (;;)	809	for (;;)
512	{	810	{
513	while (expect_true (cxix >= 0))	811	while (ecb_expect_true (cxix >= 0))
514	{	812	{
515	PERL_CONTEXT *cx = &ccstk[cxix--];	813	PERL_CONTEXT *cx = &ccstk[cxix--];
516		814
517	if (expect_true (CxTYPE (cx) == CXt_SUB || CxTYPE (cx) == CXt_FORMAT))	815	if (ecb_expect_true (CxTYPE (cx) == CXt_SUB) || ecb_expect_false (CxTYPE (cx) == CXt_FORMAT))
518	{	816	{
519	CV *cv = cx->blk_sub.cv;	817	CV *cv = cx->blk_sub.cv;
520		818
521	if (expect_true (CvDEPTH (cv)))	819	if (ecb_expect_true (CvDEPTH (cv)))
522	{	820	{
523	EXTEND (SP, 3);	821	EXTEND (SP, 3);
524	PUSHs ((SV *)CvPADLIST (cv));	822	PUSHs ((SV *)CvPADLIST (cv));
525	PUSHs (INT2PTR (SV *, (IV)CvDEPTH (cv)));	823	PUSHs (INT2PTR (SV *, (IV)CvDEPTH (cv)));
526	PUSHs ((SV *)cv);	824	PUSHs ((SV *)cv);
…		…
529	get_padlist (aTHX_ cv);	827	get_padlist (aTHX_ cv);
530	}	828	}
531	}	829	}
532	}	830	}
533		831
534	if (expect_true (top_si->si_type == PERLSI_MAIN))	832	if (ecb_expect_true (top_si->si_type == PERLSI_MAIN))
535	break;	833	break;
536		834
537	top_si = top_si->si_prev;	835	top_si = top_si->si_prev;
538	ccstk = top_si->si_cxstack;	836	ccstk = top_si->si_cxstack;
539	cxix = top_si->si_cxix;	837	cxix = top_si->si_cxix;
…		…
541		839
542	PUTBACK;	840	PUTBACK;
543	}	841	}
544		842
545	/* allocate some space on the context stack for our purposes */	843	/* allocate some space on the context stack for our purposes */
546	/* we manually unroll here, as usually 2 slots is enough */	844	if (ecb_expect_false (cxstack_ix + (int)SLOT_COUNT >= cxstack_max))
547	if (SLOT_COUNT >= 1) CXINC;
548	if (SLOT_COUNT >= 2) CXINC;
549	if (SLOT_COUNT >= 3) CXINC;
550	{	845	{
551	int i;	846	unsigned int i;
		847
552	for (i = 3; i < SLOT_COUNT; ++i)	848	for (i = 0; i < SLOT_COUNT; ++i)
553	CXINC;	849	CXINC;
554	}	850
555	cxstack_ix -= SLOT_COUNT; /* undo allocation */	851	cxstack_ix -= SLOT_COUNT; /* undo allocation */
		852	}
556		853
557	c->mainstack = PL_mainstack;	854	c->mainstack = PL_mainstack;
558		855
559	{	856	{
560	perl_slots *slot = c->slot = (perl_slots *)(cxstack + cxstack_ix + 1);	857	perl_slots *slot = c->slot = (perl_slots *)(cxstack + cxstack_ix + 1);
561		858
562	slot->defav = GvAV (PL_defgv);	859	#if CORO_JIT
563	slot->defsv = DEFSV;	860	save_perl_slots (slot);
564	slot->errsv = ERRSV;	861	#else
565	slot->irsgv = GvSV (irsgv);
566
567	#define VAR(name,type) slot->name = PL_ ## name;	862	#define VARx(name,expr,type) slot->name = expr;
568	# include "state.h"	863	#include "state.h"
569	#undef VAR	864	#endif
570	}	865	}
571	}	866	}
572		867
573	/*	868	/*
574	* allocate various perl stacks. This is an exact copy	869	* allocate various perl stacks. This is almost an exact copy
575	* of perl.c:init_stacks, except that it uses less memory	870	* of perl.c:init_stacks, except that it uses less memory
576	* on the (sometimes correct) assumption that coroutines do	871	* on the (sometimes correct) assumption that coroutines do
577	* not usually need a lot of stackspace.	872	* not usually need a lot of stackspace.
578	*/	873	*/
579	#if CORO_PREFER_PERL_FUNCTIONS	874	#if CORO_PREFER_PERL_FUNCTIONS
580	# define coro_init_stacks init_stacks	875	# define coro_init_stacks(thx) init_stacks ()
581	#else	876	#else
582	static void	877	static void
583	coro_init_stacks (pTHX)	878	coro_init_stacks (pTHX)
584	{	879	{
585	PL_curstackinfo = new_stackinfo(32, 8);	880	PL_curstackinfo = new_stackinfo(32, 4 + SLOT_COUNT); /* 3 is minimum due to perl rounding down in scope.c:GROW() */
586	PL_curstackinfo->si_type = PERLSI_MAIN;	881	PL_curstackinfo->si_type = PERLSI_MAIN;
587	PL_curstack = PL_curstackinfo->si_stack;	882	PL_curstack = PL_curstackinfo->si_stack;
588	PL_mainstack = PL_curstack; /* remember in case we switch stacks */	883	PL_mainstack = PL_curstack; /* remember in case we switch stacks */
589		884
590	PL_stack_base = AvARRAY(PL_curstack);	885	PL_stack_base = AvARRAY(PL_curstack);
…		…
605	#endif	900	#endif
606		901
607	New(54,PL_scopestack,8,I32);	902	New(54,PL_scopestack,8,I32);
608	PL_scopestack_ix = 0;	903	PL_scopestack_ix = 0;
609	PL_scopestack_max = 8;	904	PL_scopestack_max = 8;
		905	#if HAS_SCOPESTACK_NAME
		906	New(54,PL_scopestack_name,8,const char*);
		907	#endif
610		908
611	New(54,PL_savestack,24,ANY);	909	New(54,PL_savestack,24,ANY);
612	PL_savestack_ix = 0;	910	PL_savestack_ix = 0;
613	PL_savestack_max = 24;	911	PL_savestack_max = 24;
614		912
…		…
622		920
623	/*	921	/*
624	* destroy the stacks, the callchain etc...	922	* destroy the stacks, the callchain etc...
625	*/	923	*/
626	static void	924	static void
627	coro_destroy_stacks (pTHX)	925	coro_destruct_stacks (pTHX)
628	{	926	{
629	while (PL_curstackinfo->si_next)	927	while (PL_curstackinfo->si_next)
630	PL_curstackinfo = PL_curstackinfo->si_next;	928	PL_curstackinfo = PL_curstackinfo->si_next;
631		929
632	while (PL_curstackinfo)	930	while (PL_curstackinfo)
…		…
642	}	940	}
643		941
644	Safefree (PL_tmps_stack);	942	Safefree (PL_tmps_stack);
645	Safefree (PL_markstack);	943	Safefree (PL_markstack);
646	Safefree (PL_scopestack);	944	Safefree (PL_scopestack);
		945	#if HAS_SCOPESTACK_NAME
		946	Safefree (PL_scopestack_name);
		947	#endif
647	Safefree (PL_savestack);	948	Safefree (PL_savestack);
648	#if !PERL_VERSION_ATLEAST (5,10,0)	949	#if !PERL_VERSION_ATLEAST (5,10,0)
649	Safefree (PL_retstack);	950	Safefree (PL_retstack);
650	#endif	951	#endif
651	}	952	}
652		953
		954	#define CORO_RSS \
		955	rss += sizeof (SYM (curstackinfo)); \
		956	rss += (SYM (curstackinfo->si_cxmax) + 1) * sizeof (PERL_CONTEXT); \
		957	rss += sizeof (SV) + sizeof (struct xpvav) + (1 + AvMAX (SYM (curstack))) * sizeof (SV *); \
		958	rss += SYM (tmps_max) * sizeof (SV *); \
		959	rss += (SYM (markstack_max) - SYM (markstack_ptr)) * sizeof (I32); \
		960	rss += SYM (scopestack_max) * sizeof (I32); \
		961	rss += SYM (savestack_max) * sizeof (ANY);
		962
653	static size_t	963	static size_t
654	coro_rss (pTHX_ struct coro *coro)	964	coro_rss (pTHX_ struct coro *coro)
655	{	965	{
656	size_t rss = sizeof (*coro);	966	size_t rss = sizeof (*coro);
657		967
658	if (coro->mainstack)	968	if (coro->mainstack)
659	{	969	{
660	perl_slots tmp_slot;
661	perl_slots *slot;
662
663	if (coro->flags & CF_RUNNING)	970	if (coro->flags & CF_RUNNING)
664	{	971	{
665	slot = &tmp_slot;	972	#define SYM(sym) PL_ ## sym
666		973	CORO_RSS;
667	#define VAR(name,type) slot->name = PL_ ## name;
668	# include "state.h"
669	#undef VAR	974	#undef SYM
670	}	975	}
671	else	976	else
672	slot = coro->slot;
673
674	if (slot)
675	{	977	{
676	rss += sizeof (slot->curstackinfo);	978	#define SYM(sym) coro->slot->sym
677	rss += (slot->curstackinfo->si_cxmax + 1) * sizeof (PERL_CONTEXT);	979	CORO_RSS;
678	rss += sizeof (SV) + sizeof (struct xpvav) + (1 + AvMAX (slot->curstack)) * sizeof (SV *);	980	#undef SYM
679	rss += slot->tmps_max * sizeof (SV *);
680	rss += (slot->markstack_max - slot->markstack_ptr) * sizeof (I32);
681	rss += slot->scopestack_max * sizeof (I32);
682	rss += slot->savestack_max * sizeof (ANY);
683
684	#if !PERL_VERSION_ATLEAST (5,10,0)
685	rss += slot->retstack_max * sizeof (OP *);
686	#endif
687	}	981	}
688	}	982	}
689		983
690	return rss;	984	return rss;
691	}	985	}
…		…
704	#endif	998	#endif
705		999
706	/*	1000	/*
707	* This overrides the default magic get method of %SIG elements.	1001	* This overrides the default magic get method of %SIG elements.
708	* The original one doesn't provide for reading back of PL_diehook/PL_warnhook	1002	* The original one doesn't provide for reading back of PL_diehook/PL_warnhook
709	* and instead of tryign to save and restore the hash elements, we just provide	1003	* and instead of trying to save and restore the hash elements (extremely slow),
710	* readback here.	1004	* we just provide our own readback here.
711	* We only do this when the hook is != 0, as they are often set to 0 temporarily,
712	* not expecting this to actually change the hook. This is a potential problem
713	* when a schedule happens then, but we ignore this.
714	*/	1005	*/
715	static int	1006	static int ecb_cold
716	coro_sigelem_get (pTHX_ SV *sv, MAGIC *mg)	1007	coro_sigelem_get (pTHX_ SV *sv, MAGIC *mg)
717	{	1008	{
718	const char *s = MgPV_nolen_const (mg);	1009	const char *s = MgPV_nolen_const (mg);
719		1010
720	if (*s == '_')	1011	if (*s == '_')
721	{	1012	{
722	SV **svp = 0;	1013	SV **svp = 0;
723		1014
724	if (strEQ (s, "__DIE__" )) svp = &PL_diehook;	1015	if (strEQ (s, "__DIE__" )) svp = &PL_diehook;
725	if (strEQ (s, "__WARN__")) svp = &PL_warnhook;	1016	if (strEQ (s, "__WARN__")) svp = &PL_warnhook;
726		1017
727	if (svp)	1018	if (svp)
728	{	1019	{
729	sv_setsv (sv, *svp ? *svp : &PL_sv_undef);	1020	SV *ssv;
		1021
		1022	if (!*svp)
		1023	ssv = &PL_sv_undef;
		1024	else if (SvTYPE (*svp) == SVt_PVCV) /* perlio directly stores a CV in warnhook. ugh. */
		1025	ssv = sv_2mortal (newRV_inc (*svp));
		1026	else
		1027	ssv = *svp;
		1028
		1029	sv_setsv (sv, ssv);
730	return 0;	1030	return 0;
731	}	1031	}
732	}	1032	}
733		1033
734	return orig_sigelem_get ? orig_sigelem_get (aTHX_ sv, mg) : 0;	1034	return orig_sigelem_get ? orig_sigelem_get (aTHX_ sv, mg) : 0;
735	}	1035	}
736		1036
737	static int	1037	static int ecb_cold
738	coro_sigelem_clr (pTHX_ SV *sv, MAGIC *mg)	1038	coro_sigelem_clr (pTHX_ SV *sv, MAGIC *mg)
739	{	1039	{
740	const char *s = MgPV_nolen_const (mg);	1040	const char *s = MgPV_nolen_const (mg);
741		1041
742	if (*s == '_')	1042	if (*s == '_')
…		…
756	}	1056	}
757		1057
758	return orig_sigelem_clr ? orig_sigelem_clr (aTHX_ sv, mg) : 0;	1058	return orig_sigelem_clr ? orig_sigelem_clr (aTHX_ sv, mg) : 0;
759	}	1059	}
760		1060
761	static int	1061	static int ecb_cold
762	coro_sigelem_set (pTHX_ SV *sv, MAGIC *mg)	1062	coro_sigelem_set (pTHX_ SV *sv, MAGIC *mg)
763	{	1063	{
764	const char *s = MgPV_nolen_const (mg);	1064	const char *s = MgPV_nolen_const (mg);
765		1065
766	if (*s == '_')	1066	if (*s == '_')
…		…
771	if (strEQ (s, "__WARN__")) svp = &PL_warnhook;	1071	if (strEQ (s, "__WARN__")) svp = &PL_warnhook;
772		1072
773	if (svp)	1073	if (svp)
774	{	1074	{
775	SV *old = *svp;	1075	SV *old = *svp;
776	*svp = newSVsv (sv);	1076	*svp = SvOK (sv) ? newSVsv (sv) : 0;
777	SvREFCNT_dec (old);	1077	SvREFCNT_dec (old);
778	return 0;	1078	return 0;
779	}	1079	}
780	}	1080	}
781		1081
782	return orig_sigelem_set ? orig_sigelem_set (aTHX_ sv, mg) : 0;	1082	return orig_sigelem_set ? orig_sigelem_set (aTHX_ sv, mg) : 0;
783	}	1083	}
784		1084
785	static void	1085	static void
		1086	prepare_nop (pTHX_ struct coro_transfer_args *ta)
		1087	{
		1088	/* kind of mega-hacky, but works */
		1089	ta->next = ta->prev = (struct coro *)ta;
		1090	}
		1091
		1092	static int
		1093	slf_check_nop (pTHX_ struct CoroSLF *frame)
		1094	{
		1095	return 0;
		1096	}
		1097
		1098	static int
		1099	slf_check_repeat (pTHX_ struct CoroSLF *frame)
		1100	{
		1101	return 1;
		1102	}
		1103
		1104	static UNOP init_perl_op;
		1105
		1106	ecb_noinline static void /* noinline to keep it out of the transfer fast path */
786	coro_setup (pTHX_ struct coro *coro)	1107	init_perl (pTHX_ struct coro *coro)
787	{	1108	{
788	/*	1109	/*
789	* emulate part of the perl startup here.	1110	* emulate part of the perl startup here.
790	*/	1111	*/
791	coro_init_stacks (aTHX);	1112	coro_init_stacks (aTHX);
792		1113
793	PL_runops = RUNOPS_DEFAULT;	1114	PL_runops = RUNOPS_DEFAULT;
794	PL_curcop = &PL_compiling;	1115	PL_curcop = &PL_compiling;
795	PL_in_eval = EVAL_NULL;	1116	PL_in_eval = EVAL_NULL;
796	PL_comppad = 0;	1117	PL_comppad = 0;
		1118	PL_comppad_name = 0;
		1119	PL_comppad_name_fill = 0;
		1120	PL_comppad_name_floor = 0;
797	PL_curpm = 0;	1121	PL_curpm = 0;
798	PL_curpad = 0;	1122	PL_curpad = 0;
799	PL_localizing = 0;	1123	PL_localizing = 0;
800	PL_dirty = 0;
801	PL_restartop = 0;	1124	PL_restartop = 0;
802	#if PERL_VERSION_ATLEAST (5,10,0)	1125	#if PERL_VERSION_ATLEAST (5,10,0)
803	PL_parser = 0;	1126	PL_parser = 0;
804	#endif	1127	#endif
		1128	PL_hints = 0;
805		1129
806	/* recreate the die/warn hooks */	1130	/* recreate the die/warn hooks */
807	PL_diehook = 0; SvSetMagicSV (*hv_fetch (hv_sig, "__DIE__" , sizeof ("__DIE__" ) - 1, 1), rv_diehook );	1131	PL_diehook = SvREFCNT_inc (rv_diehook);
808	PL_warnhook = 0; SvSetMagicSV (*hv_fetch (hv_sig, "__WARN__", sizeof ("__WARN__") - 1, 1), rv_warnhook);	1132	PL_warnhook = SvREFCNT_inc (rv_warnhook);
809		1133
810	GvSV (PL_defgv) = newSV (0);	1134	GvSV (PL_defgv) = newSV (0);
811	GvAV (PL_defgv) = coro->args; coro->args = 0;	1135	GvAV (PL_defgv) = coro->args; coro->args = 0;
812	GvSV (PL_errgv) = newSV (0);	1136	GvSV (PL_errgv) = newSV (0);
813	GvSV (irsgv) = newSVpvn ("\n", 1); sv_magic (GvSV (irsgv), (SV *)irsgv, PERL_MAGIC_sv, "/", 0);	1137	GvSV (irsgv) = newSVpvn ("\n", 1); sv_magic (GvSV (irsgv), (SV *)irsgv, PERL_MAGIC_sv, "/", 0);
		1138	GvHV (PL_hintgv) = 0;
814	PL_rs = newSVsv (GvSV (irsgv));	1139	PL_rs = newSVsv (GvSV (irsgv));
815	PL_defoutgv = (GV *)SvREFCNT_inc (stdoutgv);	1140	PL_defoutgv = (GV *)SvREFCNT_inc_NN (stdoutgv);
816		1141
817	{	1142	{
818	dSP;	1143	dSP;
819	LOGOP myop;	1144	UNOP myop;
820		1145
821	Zero (&myop, 1, LOGOP);	1146	Zero (&myop, 1, UNOP);
822	myop.op_next = Nullop;	1147	myop.op_next = Nullop;
		1148	myop.op_type = OP_ENTERSUB;
823	myop.op_flags = OPf_WANT_VOID;	1149	myop.op_flags = OPf_WANT_VOID;
824		1150
825	PUSHMARK (SP);	1151	PUSHMARK (SP);
826	XPUSHs (sv_2mortal (av_shift (GvAV (PL_defgv))));	1152	PUSHs ((SV *)coro->startcv);
827	PUTBACK;	1153	PUTBACK;
828	PL_op = (OP *)&myop;	1154	PL_op = (OP *)&myop;
829	PL_op = PL_ppaddr[OP_ENTERSUB](aTHX);	1155	PL_op = PL_ppaddr[OP_ENTERSUB](aTHX);
830	SPAGAIN;
831	}	1156	}
832		1157
833	/* this newly created coroutine might be run on an existing cctx which most	1158	/* this newly created coroutine might be run on an existing cctx which most
834	* likely was suspended in set_stacklevel, called from entersub.	1159	* likely was suspended in pp_slf, so we have to emulate entering pp_slf here.
835	* set_stacklevl doesn't do anything on return, but entersub does LEAVE,
836	* so we ENTER here for symmetry
837	*/	1160	*/
838	ENTER;	1161	slf_frame.prepare = prepare_nop; /* provide a nop function for an eventual pp_slf */
839	}	1162	slf_frame.check = slf_check_nop; /* signal pp_slf to not repeat */
		1163	slf_frame.destroy = 0;
840		1164
		1165	/* and we have to provide the pp_slf op in any case, so pp_slf can skip it */
		1166	init_perl_op.op_next = PL_op;
		1167	init_perl_op.op_type = OP_ENTERSUB;
		1168	init_perl_op.op_ppaddr = pp_slf;
		1169	/* no flags etc. required, as an init function won't be called */
		1170
		1171	PL_op = (OP *)&init_perl_op;
		1172
		1173	/* copy throw, in case it was set before init_perl */
		1174	CORO_THROW = coro->except;
		1175
		1176	SWAP_SVS (coro);
		1177
		1178	if (ecb_expect_false (enable_times))
		1179	{
		1180	coro_times_update ();
		1181	coro_times_sub (coro);
		1182	}
		1183	}
		1184
841	static void	1185	static void
842	coro_destroy (pTHX_ struct coro *coro)	1186	coro_unwind_stacks (pTHX)
843	{	1187	{
844	if (!IN_DESTRUCT)	1188	if (!IN_DESTRUCT)
845	{	1189	{
846	/* restore all saved variables and stuff */	1190	/* restore all saved variables and stuff */
847	LEAVE_SCOPE (0);	1191	LEAVE_SCOPE (0);
…		…
855	POPSTACK_TO (PL_mainstack);	1199	POPSTACK_TO (PL_mainstack);
856		1200
857	/* unwind main stack */	1201	/* unwind main stack */
858	dounwind (-1);	1202	dounwind (-1);
859	}	1203	}
		1204	}
860		1205
861	SvREFCNT_dec (GvSV (PL_defgv));	1206	static void
862	SvREFCNT_dec (GvAV (PL_defgv));	1207	destroy_perl (pTHX_ struct coro *coro)
863	SvREFCNT_dec (GvSV (PL_errgv));	1208	{
864	SvREFCNT_dec (PL_defoutgv);	1209	SV *svf [9];
865	SvREFCNT_dec (PL_rs);
866	SvREFCNT_dec (GvSV (irsgv));
867		1210
868	SvREFCNT_dec (PL_diehook);
869	SvREFCNT_dec (PL_warnhook);
870		1211	{
		1212	SV *old_current = SvRV (coro_current);
		1213	struct coro *current = SvSTATE (old_current);
		1214
		1215	assert (("FATAL: tried to destroy currently running coroutine", coro->mainstack != PL_mainstack));
		1216
		1217	save_perl (aTHX_ current);
		1218
		1219	/* this will cause transfer_check to croak on block*/
		1220	SvRV_set (coro_current, (SV *)coro->hv);
		1221
		1222	load_perl (aTHX_ coro);
		1223
		1224	coro_unwind_stacks (aTHX);
		1225
		1226	/* restore swapped sv's */
		1227	SWAP_SVS (coro);
		1228
		1229	coro_destruct_stacks (aTHX);
		1230
		1231	/* now save some sv's to be free'd later */
		1232	svf [0] = GvSV (PL_defgv);
		1233	svf [1] = (SV *)GvAV (PL_defgv);
		1234	svf [2] = GvSV (PL_errgv);
		1235	svf [3] = (SV *)PL_defoutgv;
		1236	svf [4] = PL_rs;
		1237	svf [5] = GvSV (irsgv);
		1238	svf [6] = (SV *)GvHV (PL_hintgv);
		1239	svf [7] = PL_diehook;
		1240	svf [8] = PL_warnhook;
		1241	assert (9 == sizeof (svf) / sizeof (*svf));
		1242
		1243	SvRV_set (coro_current, old_current);
		1244
		1245	load_perl (aTHX_ current);
		1246	}
		1247
		1248	{
		1249	unsigned int i;
		1250
		1251	for (i = 0; i < sizeof (svf) / sizeof (*svf); ++i)
		1252	SvREFCNT_dec (svf [i]);
		1253
871	SvREFCNT_dec (coro->saved_deffh);	1254	SvREFCNT_dec (coro->saved_deffh);
872	SvREFCNT_dec (coro->throw);	1255	SvREFCNT_dec (coro->rouse_cb);
873		1256	SvREFCNT_dec (coro->invoke_cb);
874	coro_destroy_stacks (aTHX);	1257	SvREFCNT_dec (coro->invoke_av);
		1258	}
875	}	1259	}
876		1260
877	static void	1261	ecb_inline void
878	free_coro_mortal (pTHX)	1262	free_coro_mortal (pTHX)
879	{	1263	{
880	if (expect_true (coro_mortal))	1264	if (ecb_expect_true (coro_mortal))
881	{	1265	{
882	SvREFCNT_dec (coro_mortal);	1266	SvREFCNT_dec ((SV *)coro_mortal);
883	coro_mortal = 0;	1267	coro_mortal = 0;
884	}	1268	}
885	}	1269	}
886		1270
887	static int	1271	static int
888	runops_trace (pTHX)	1272	runops_trace (pTHX)
889	{	1273	{
890	COP *oldcop = 0;	1274	COP *oldcop = 0;
891	int oldcxix = -2;	1275	int oldcxix = -2;
892	struct coro *coro = SvSTATE (coro_current); /* trace cctx is tied to specific coro */
893	coro_cctx *cctx = coro->cctx;
894		1276
895	while ((PL_op = CALL_FPTR (PL_op->op_ppaddr) (aTHX)))	1277	while ((PL_op = CALL_FPTR (PL_op->op_ppaddr) (aTHX)))
896	{	1278	{
897	PERL_ASYNC_CHECK ();	1279	PERL_ASYNC_CHECK ();
898		1280
899	if (cctx->flags & CC_TRACE_ALL)	1281	if (cctx_current->flags & CC_TRACE_ALL)
900	{	1282	{
901	if (PL_op->op_type == OP_LEAVESUB && cctx->flags & CC_TRACE_SUB)	1283	if (PL_op->op_type == OP_LEAVESUB && cctx_current->flags & CC_TRACE_SUB)
902	{	1284	{
903	PERL_CONTEXT *cx = &cxstack[cxstack_ix];	1285	PERL_CONTEXT *cx = &cxstack[cxstack_ix];
904	SV **bot, **top;	1286	SV **bot, **top;
905	AV *av = newAV (); /* return values */	1287	AV *av = newAV (); /* return values */
906	SV **cb;	1288	SV **cb;
…		…
916	: cx->blk_gimme == G_SCALAR ? bot + 1	1298	: cx->blk_gimme == G_SCALAR ? bot + 1
917	: bot;	1299	: bot;
918		1300
919	av_extend (av, top - bot);	1301	av_extend (av, top - bot);
920	while (bot < top)	1302	while (bot < top)
921	av_push (av, SvREFCNT_inc (*bot++));	1303	av_push (av, SvREFCNT_inc_NN (*bot++));
922		1304
923	PL_runops = RUNOPS_DEFAULT;	1305	PL_runops = RUNOPS_DEFAULT;
924	ENTER;	1306	ENTER;
925	SAVETMPS;	1307	SAVETMPS;
926	EXTEND (SP, 3);	1308	EXTEND (SP, 3);
…		…
943		1325
944	if (PL_curcop != &PL_compiling)	1326	if (PL_curcop != &PL_compiling)
945	{	1327	{
946	SV **cb;	1328	SV **cb;
947		1329
948	if (oldcxix != cxstack_ix && cctx->flags & CC_TRACE_SUB)	1330	if (oldcxix != cxstack_ix && cctx_current->flags & CC_TRACE_SUB)
949	{	1331	{
950	PERL_CONTEXT *cx = &cxstack[cxstack_ix];	1332	PERL_CONTEXT *cx = &cxstack[cxstack_ix];
951		1333
952	if (CxTYPE (cx) == CXt_SUB && oldcxix < cxstack_ix)	1334	if (CxTYPE (cx) == CXt_SUB && oldcxix < cxstack_ix)
953	{	1335	{
954	runops_proc_t old_runops = PL_runops;
955	dSP;	1336	dSP;
956	GV *gv = CvGV (cx->blk_sub.cv);	1337	GV *gv = CvGV (cx->blk_sub.cv);
957	SV *fullname = sv_2mortal (newSV (0));	1338	SV *fullname = sv_2mortal (newSV (0));
958		1339
959	if (isGV (gv))	1340	if (isGV (gv))
…		…
964	SAVETMPS;	1345	SAVETMPS;
965	EXTEND (SP, 3);	1346	EXTEND (SP, 3);
966	PUSHMARK (SP);	1347	PUSHMARK (SP);
967	PUSHs (&PL_sv_yes);	1348	PUSHs (&PL_sv_yes);
968	PUSHs (fullname);	1349	PUSHs (fullname);
969	PUSHs (CxHASARGS (cx) ? sv_2mortal (newRV_inc ((SV *)cx->blk_sub.argarray)) : &PL_sv_undef);	1350	PUSHs (CxHASARGS (cx) ? sv_2mortal (newRV_inc ((SV *)cx->blk_sub.argarray)) : &PL_sv_undef);
970	PUTBACK;	1351	PUTBACK;
971	cb = hv_fetch ((HV *)SvRV (coro_current), "_trace_sub_cb", sizeof ("_trace_sub_cb") - 1, 0);	1352	cb = hv_fetch ((HV *)SvRV (coro_current), "_trace_sub_cb", sizeof ("_trace_sub_cb") - 1, 0);
972	if (cb) call_sv (*cb, G_KEEPERR | G_EVAL | G_VOID | G_DISCARD);	1353	if (cb) call_sv (*cb, G_KEEPERR | G_EVAL | G_VOID | G_DISCARD);
973	SPAGAIN;	1354	SPAGAIN;
974	FREETMPS;	1355	FREETMPS;
…		…
977	}	1358	}
978		1359
979	oldcxix = cxstack_ix;	1360	oldcxix = cxstack_ix;
980	}	1361	}
981		1362
982	if (cctx->flags & CC_TRACE_LINE)	1363	if (cctx_current->flags & CC_TRACE_LINE)
983	{	1364	{
984	dSP;	1365	dSP;
985		1366
986	PL_runops = RUNOPS_DEFAULT;	1367	PL_runops = RUNOPS_DEFAULT;
987	ENTER;	1368	ENTER;
…		…
1006		1387
1007	TAINT_NOT;	1388	TAINT_NOT;
1008	return 0;	1389	return 0;
1009	}	1390	}
1010		1391
1011	/* inject a fake call to Coro::State::_cctx_init into the execution */	1392	static struct CoroSLF cctx_ssl_frame;
1012	/* _cctx_init should be careful, as it could be called at almost any time */
1013	/* during execution of a perl program */
1014	static void NOINLINE
1015	cctx_prepare (pTHX_ coro_cctx *cctx)
1016	{
1017	dSP;
1018	LOGOP myop;
1019		1393
		1394	static void
		1395	slf_prepare_set_stacklevel (pTHX_ struct coro_transfer_args *ta)
		1396	{
		1397	ta->prev = 0;
		1398	}
		1399
		1400	static int
		1401	slf_check_set_stacklevel (pTHX_ struct CoroSLF *frame)
		1402	{
		1403	*frame = cctx_ssl_frame;
		1404
		1405	return frame->check (aTHX_ frame); /* execute the restored frame - there must be one */
		1406	}
		1407
		1408	/* initialises PL_top_env and injects a pseudo-slf-call to set the stacklevel */
		1409	static void ecb_noinline
		1410	cctx_prepare (pTHX)
		1411	{
1020	PL_top_env = &PL_start_env;	1412	PL_top_env = &PL_start_env;
1021		1413
1022	if (cctx->flags & CC_TRACE)	1414	if (cctx_current->flags & CC_TRACE)
1023	PL_runops = runops_trace;	1415	PL_runops = runops_trace;
1024		1416
1025	Zero (&myop, 1, LOGOP);	1417	/* we already must be executing an SLF op, there is no other valid way
1026	myop.op_next = PL_op;	1418	* that can lead to creation of a new cctx */
1027	myop.op_flags = OPf_WANT_VOID | OPf_STACKED;	1419	assert (("FATAL: can't prepare slf-less cctx in Coro module (please report)",
		1420	slf_frame.prepare && PL_op->op_ppaddr == pp_slf));
1028		1421
1029	PUSHMARK (SP);	1422	/* we must emulate leaving pp_slf, which is done inside slf_check_set_stacklevel */
1030	EXTEND (SP, 2);	1423	cctx_ssl_frame = slf_frame;
1031	PUSHs (sv_2mortal (newSViv (PTR2IV (cctx))));	1424
1032	PUSHs ((SV *)get_cv ("Coro::State::_cctx_init", FALSE));	1425	slf_frame.prepare = slf_prepare_set_stacklevel;
1033	PUTBACK;	1426	slf_frame.check = slf_check_set_stacklevel;
1034	PL_op = (OP *)&myop;	1427	}
1035	PL_op = PL_ppaddr[OP_ENTERSUB](aTHX);	1428
1036	SPAGAIN;	1429	/* the tail of transfer: execute stuff we can only do after a transfer */
		1430	ecb_inline void
		1431	transfer_tail (pTHX)
		1432	{
		1433	free_coro_mortal (aTHX);
		1434	}
		1435
		1436	/* try to exit the same way perl's main function would do */
		1437	/* we do not bother resetting the environment or other things *7
		1438	/* that are not, uhm, essential */
		1439	/* this obviously also doesn't work when perl is embedded */
		1440	static void ecb_noinline ecb_cold
		1441	perlish_exit (pTHX)
		1442	{
		1443	int exitstatus = perl_destruct (PL_curinterp);
		1444	perl_free (PL_curinterp);
		1445	exit (exitstatus);
1037	}	1446	}
1038		1447
1039	/*	1448	/*
1040	* this is a _very_ stripped down perl interpreter ;)	1449	* this is a _very_ stripped down perl interpreter ;)
1041	*/	1450	*/
1042	static void	1451	static void
1043	cctx_run (void *arg)	1452	cctx_run (void *arg)
1044	{	1453	{
		1454	#ifdef USE_ITHREADS
		1455	# if CORO_PTHREAD
		1456	PERL_SET_CONTEXT (coro_thx);
		1457	# endif
		1458	#endif
		1459	{
1045	dTHX;	1460	dTHX;
1046		1461
1047	/* cctx_run is the alternative tail of transfer(), so unlock here. */	1462	/* normally we would need to skip the entersub here */
1048	UNLOCK;	1463	/* not doing so will re-execute it, which is exactly what we want */
1049
1050	/* we now skip the entersub that lead to transfer() */
1051	PL_op = PL_op->op_next;	1464	/* PL_nop = PL_nop->op_next */
1052		1465
1053	/* inject a fake subroutine call to cctx_init */	1466	/* inject a fake subroutine call to cctx_init */
1054	cctx_prepare (aTHX_ (coro_cctx *)arg);	1467	cctx_prepare (aTHX);
1055		1468
		1469	/* cctx_run is the alternative tail of transfer() */
		1470	transfer_tail (aTHX);
		1471
1056	/* somebody or something will hit me for both perl_run and PL_restartop */	1472	/* somebody or something will hit me for both perl_run and PL_restartop */
1057	PL_restartop = PL_op;	1473	PL_restartop = PL_op;
1058	perl_run (PL_curinterp);	1474	perl_run (PL_curinterp);
1059
1060	/*	1475	/*
		1476	* Unfortunately, there is no way to get at the return values of the
		1477	* coro body here, as perl_run destroys these. Likewise, we cannot catch
		1478	* runtime errors here, as this is just a random interpreter, not a thread.
		1479	*/
		1480
		1481	/*
1061	* If perl-run returns we assume exit() was being called or the coro	1482	* If perl-run returns we assume exit() was being called or the coro
1062	* fell off the end, which seems to be the only valid (non-bug)	1483	* fell off the end, which seems to be the only valid (non-bug)
1063	* reason for perl_run to return. We try to exit by jumping to the	1484	* reason for perl_run to return. We try to mimic whatever perl is normally
1064	* bootstrap-time "top" top_env, as we cannot restore the "main"	1485	* doing in that case. YMMV.
1065	* coroutine as Coro has no such concept
1066	*/	1486	*/
1067	PL_top_env = main_top_env;	1487	perlish_exit (aTHX);
1068	JMPENV_JUMP (2); /* I do not feel well about the hardcoded 2 at all */	1488	}
1069	}	1489	}
1070		1490
1071	static coro_cctx *	1491	static coro_cctx *
1072	cctx_new ()	1492	cctx_new (void)
1073	{	1493	{
1074	coro_cctx *cctx;	1494	coro_cctx *cctx;
1075	void *stack_start;
1076	size_t stack_size;
1077		1495
1078	++cctx_count;	1496	++cctx_count;
1079
1080	Newz (0, cctx, 1, coro_cctx);	1497	New (0, cctx, 1, coro_cctx);
1081		1498
1082	#if HAVE_MMAP	1499	cctx->gen = cctx_gen;
1083	cctx->ssize = ((coro_stacksize * sizeof (long) + PAGESIZE - 1) / PAGESIZE + CORO_STACKGUARD) * PAGESIZE;	1500	cctx->flags = 0;
1084	/* mmap supposedly does allocate-on-write for us */	1501	cctx->idle_sp = 0; /* can be accessed by transfer between cctx_run and set_stacklevel, on throw */
1085	cctx->sptr = mmap (0, cctx->ssize, PROT_EXEC|PROT_READ|PROT_WRITE, MAP_PRIVATE|MAP_ANONYMOUS, 0, 0);
1086		1502
1087	if (cctx->sptr != (void *)-1)	1503	return cctx;
1088	{	1504	}
1089	# if CORO_STACKGUARD	1505
1090	mprotect (cctx->sptr, CORO_STACKGUARD * PAGESIZE, PROT_NONE);	1506	/* create a new cctx only suitable as source */
1091	# endif	1507	static coro_cctx *
1092	stack_start = CORO_STACKGUARD * PAGESIZE + (char *)cctx->sptr;	1508	cctx_new_empty (void)
1093	stack_size = cctx->ssize - CORO_STACKGUARD * PAGESIZE;	1509	{
1094	cctx->flags |= CC_MAPPED;	1510	coro_cctx *cctx = cctx_new ();
		1511
		1512	cctx->stack.sptr = 0;
		1513	coro_create (&cctx->cctx, 0, 0, 0, 0);
		1514
		1515	return cctx;
		1516	}
		1517
		1518	/* create a new cctx suitable as destination/running a perl interpreter */
		1519	static coro_cctx *
		1520	cctx_new_run (void)
		1521	{
		1522	coro_cctx *cctx = cctx_new ();
		1523
		1524	if (!coro_stack_alloc (&cctx->stack, cctx_stacksize))
1095	}	1525	{
1096	else
1097	#endif
1098	{
1099	cctx->ssize = coro_stacksize * (long)sizeof (long);
1100	New (0, cctx->sptr, coro_stacksize, long);
1101
1102	if (!cctx->sptr)
1103	{
1104	perror ("FATAL: unable to allocate stack for coroutine");	1526	perror ("FATAL: unable to allocate stack for coroutine, exiting.");
1105	_exit (EXIT_FAILURE);	1527	_exit (EXIT_FAILURE);
1106	}
1107
1108	stack_start = cctx->sptr;
1109	stack_size = cctx->ssize;
1110	}	1528	}
1111		1529
1112	REGISTER_STACK (cctx, (char *)stack_start, (char *)stack_start + stack_size);
1113	coro_create (&cctx->cctx, cctx_run, (void *)cctx, stack_start, stack_size);	1530	coro_create (&cctx->cctx, cctx_run, (void *)cctx, cctx->stack.sptr, cctx->stack.ssze);
1114		1531
1115	return cctx;	1532	return cctx;
1116	}	1533	}
1117		1534
1118	static void	1535	static void
1119	cctx_destroy (coro_cctx *cctx)	1536	cctx_destroy (coro_cctx *cctx)
1120	{	1537	{
1121	if (!cctx)	1538	if (!cctx)
1122	return;	1539	return;
1123		1540
		1541	assert (("FATAL: tried to destroy current cctx", cctx != cctx_current));
		1542
1124	--cctx_count;	1543	--cctx_count;
		1544	coro_destroy (&cctx->cctx);
1125		1545
1126	#if CORO_USE_VALGRIND	1546	coro_stack_free (&cctx->stack);
1127	VALGRIND_STACK_DEREGISTER (cctx->valgrind_id);
1128	#endif
1129
1130	#if HAVE_MMAP
1131	if (cctx->flags & CC_MAPPED)
1132	munmap (cctx->sptr, cctx->ssize);
1133	else
1134	#endif
1135	Safefree (cctx->sptr);
1136		1547
1137	Safefree (cctx);	1548	Safefree (cctx);
1138	}	1549	}
1139		1550
1140	/* wether this cctx should be destructed */	1551	/* wether this cctx should be destructed */
1141	#define CCTX_EXPIRED(cctx) ((cctx)->ssize < coro_stacksize || ((cctx)->flags & CC_NOREUSE))	1552	#define CCTX_EXPIRED(cctx) ((cctx)->gen != cctx_gen || ((cctx)->flags & CC_NOREUSE))
1142		1553
1143	static coro_cctx *	1554	static coro_cctx *
1144	cctx_get (pTHX)	1555	cctx_get (pTHX)
1145	{	1556	{
1146	while (expect_true (cctx_first))	1557	while (ecb_expect_true (cctx_first))
1147	{	1558	{
1148	coro_cctx *cctx = cctx_first;	1559	coro_cctx *cctx = cctx_first;
1149	cctx_first = cctx->next;	1560	cctx_first = cctx->next;
1150	--cctx_idle;	1561	--cctx_idle;
1151		1562
1152	if (expect_true (!CCTX_EXPIRED (cctx)))	1563	if (ecb_expect_true (!CCTX_EXPIRED (cctx)))
1153	return cctx;	1564	return cctx;
1154		1565
1155	cctx_destroy (cctx);	1566	cctx_destroy (cctx);
1156	}	1567	}
1157		1568
1158	return cctx_new ();	1569	return cctx_new_run ();
1159	}	1570	}
1160		1571
1161	static void	1572	static void
1162	cctx_put (coro_cctx *cctx)	1573	cctx_put (coro_cctx *cctx)
1163	{	1574	{
		1575	assert (("FATAL: cctx_put called on non-initialised cctx in Coro (please report)", cctx->stack.sptr));
		1576
1164	/* free another cctx if overlimit */	1577	/* free another cctx if overlimit */
1165	if (expect_false (cctx_idle >= MAX_IDLE_CCTX))	1578	if (ecb_expect_false (cctx_idle >= cctx_max_idle))
1166	{	1579	{
1167	coro_cctx *first = cctx_first;	1580	coro_cctx *first = cctx_first;
1168	cctx_first = first->next;	1581	cctx_first = first->next;
1169	--cctx_idle;	1582	--cctx_idle;
1170		1583
…		…
1179	/** coroutine switching *****************************************************/	1592	/** coroutine switching *****************************************************/
1180		1593
1181	static void	1594	static void
1182	transfer_check (pTHX_ struct coro *prev, struct coro *next)	1595	transfer_check (pTHX_ struct coro *prev, struct coro *next)
1183	{	1596	{
		1597	/* TODO: throwing up here is considered harmful */
		1598
1184	if (expect_true (prev != next))	1599	if (ecb_expect_true (prev != next))
1185	{	1600	{
1186	if (expect_false (!(prev->flags & (CF_RUNNING | CF_NEW))))	1601	if (ecb_expect_false (!(prev->flags & (CF_RUNNING | CF_NEW))))
1187	croak ("Coro::State::transfer called with non-running/new prev Coro::State, but can only transfer from running or new states");	1602	croak ("Coro::State::transfer called with a blocked prev Coro::State, but can only transfer from running or new states,");
1188		1603
1189	if (expect_false (next->flags & CF_RUNNING))	1604	if (ecb_expect_false (next->flags & (CF_RUNNING | CF_ZOMBIE | CF_SUSPENDED)))
1190	croak ("Coro::State::transfer called with running next Coro::State, but can only transfer to inactive states");
1191
1192	if (expect_false (next->flags & CF_DESTROYED))
1193	croak ("Coro::State::transfer called with destroyed next Coro::State, but can only transfer to inactive states");	1605	croak ("Coro::State::transfer called with running, destroyed or suspended next Coro::State, but can only transfer to inactive states,");
1194		1606
1195	#if !PERL_VERSION_ATLEAST (5,10,0)	1607	#if !PERL_VERSION_ATLEAST (5,10,0)
1196	if (expect_false (PL_lex_state != LEX_NOTPARSING))	1608	if (ecb_expect_false (PL_lex_state != LEX_NOTPARSING))
1197	croak ("Coro::State::transfer called while parsing, but this is not supported in your perl version");	1609	croak ("Coro::State::transfer called while parsing, but this is not supported in your perl version,");
1198	#endif	1610	#endif
1199	}	1611	}
1200	}	1612	}
1201		1613
1202	/* always use the TRANSFER macro */	1614	/* always use the TRANSFER macro */
1203	static void NOINLINE	1615	static void ecb_noinline /* noinline so we have a fixed stackframe */
1204	transfer (pTHX_ struct coro *prev, struct coro *next, int force_cctx)	1616	transfer (pTHX_ struct coro *prev, struct coro *next, int force_cctx)
1205	{	1617	{
1206	dSTACKLEVEL;	1618	dSTACKLEVEL;
1207	static volatile int has_throw;
1208		1619
1209	/* sometimes transfer is only called to set idle_sp */	1620	/* sometimes transfer is only called to set idle_sp */
1210	if (expect_false (!next))	1621	if (ecb_expect_false (!prev))
1211	{	1622	{
1212	((coro_cctx *)prev)->idle_sp = STACKLEVEL;	1623	cctx_current->idle_sp = STACKLEVEL;
1213	assert (((coro_cctx *)prev)->idle_te = PL_top_env); /* just for the side-effect when asserts are enabled */	1624	assert (cctx_current->idle_te = PL_top_env); /* just for the side-effect when asserts are enabled */
1214	}	1625	}
1215	else if (expect_true (prev != next))	1626	else if (ecb_expect_true (prev != next))
1216	{	1627	{
1217	coro_cctx *prev__cctx;	1628	coro_cctx *cctx_prev;
1218		1629
1219	if (expect_false (prev->flags & CF_NEW))	1630	if (ecb_expect_false (prev->flags & CF_NEW))
1220	{	1631	{
1221	/* create a new empty context */	1632	/* create a new empty/source context */
1222	Newz (0, prev->cctx, 1, coro_cctx);
1223	prev->flags &= ~CF_NEW;	1633	prev->flags &= ~CF_NEW;
1224	prev->flags |= CF_RUNNING;	1634	prev->flags |= CF_RUNNING;
1225	}	1635	}
1226		1636
1227	prev->flags &= ~CF_RUNNING;	1637	prev->flags &= ~CF_RUNNING;
1228	next->flags |= CF_RUNNING;	1638	next->flags |= CF_RUNNING;
1229		1639
1230	LOCK;
1231
1232	/* first get rid of the old state */	1640	/* first get rid of the old state */
1233	save_perl (aTHX_ prev);	1641	save_perl (aTHX_ prev);
1234		1642
1235	if (expect_false (next->flags & CF_NEW))	1643	if (ecb_expect_false (next->flags & CF_NEW))
1236	{	1644	{
1237	/* need to start coroutine */	1645	/* need to start coroutine */
1238	next->flags &= ~CF_NEW;	1646	next->flags &= ~CF_NEW;
1239	/* setup coroutine call */	1647	/* setup coroutine call */
1240	coro_setup (aTHX_ next);	1648	init_perl (aTHX_ next);
1241	}	1649	}
1242	else	1650	else
1243	load_perl (aTHX_ next);	1651	load_perl (aTHX_ next);
1244		1652
1245	prev__cctx = prev->cctx;
1246
1247	/* possibly "free" the cctx */	1653	/* possibly untie and reuse the cctx */
1248	if (expect_true (	1654	if (ecb_expect_true (
1249	prev__cctx->idle_sp == STACKLEVEL	1655	cctx_current->idle_sp == STACKLEVEL
1250	&& !(prev__cctx->flags & CC_TRACE)	1656	&& !(cctx_current->flags & CC_TRACE)
1251	&& !force_cctx	1657	&& !force_cctx
1252	))	1658	))
1253	{	1659	{
1254	/* I assume that STACKLEVEL is a stronger indicator than PL_top_env changes */	1660	/* I assume that stacklevel is a stronger indicator than PL_top_env changes */
1255	assert (("ERROR: current top_env must equal previous top_env", PL_top_env == prev__cctx->idle_te));	1661	assert (("FATAL: current top_env must equal previous top_env in Coro (please report)", PL_top_env == cctx_current->idle_te));
1256		1662
1257	prev->cctx = 0;
1258
1259	/* if the cctx is about to be destroyed we need to make sure we won't see it in cctx_get */	1663	/* if the cctx is about to be destroyed we need to make sure we won't see it in cctx_get. */
1260	/* without this the next cctx_get might destroy the prev__cctx while still in use */	1664	/* without this the next cctx_get might destroy the running cctx while still in use */
1261	if (expect_false (CCTX_EXPIRED (prev__cctx)))	1665	if (ecb_expect_false (CCTX_EXPIRED (cctx_current)))
1262	if (!next->cctx)	1666	if (ecb_expect_true (!next->cctx))
1263	next->cctx = cctx_get (aTHX);	1667	next->cctx = cctx_get (aTHX);
1264		1668
1265	cctx_put (prev__cctx);	1669	cctx_put (cctx_current);
1266	}	1670	}
		1671	else
		1672	prev->cctx = cctx_current;
1267		1673
1268	++next->usecount;	1674	++next->usecount;
1269		1675
1270	if (expect_true (!next->cctx))	1676	cctx_prev = cctx_current;
1271	next->cctx = cctx_get (aTHX);	1677	cctx_current = ecb_expect_false (next->cctx) ? next->cctx : cctx_get (aTHX);
1272		1678
1273	has_throw = !!next->throw;	1679	next->cctx = 0;
1274		1680
1275	if (expect_false (prev__cctx != next->cctx))	1681	if (ecb_expect_false (cctx_prev != cctx_current))
1276	{	1682	{
1277	prev__cctx->top_env = PL_top_env;	1683	cctx_prev->top_env = PL_top_env;
1278	PL_top_env = next->cctx->top_env;	1684	PL_top_env = cctx_current->top_env;
1279	coro_transfer (&prev__cctx->cctx, &next->cctx->cctx);	1685	coro_transfer (&cctx_prev->cctx, &cctx_current->cctx);
1280	}	1686	}
1281		1687
1282	free_coro_mortal (aTHX);	1688	transfer_tail (aTHX);
1283	UNLOCK;
1284
1285	if (expect_false (has_throw))
1286	{
1287	struct coro *coro = SvSTATE (coro_current);
1288
1289	if (coro->throw)
1290	{
1291	SV *exception = coro->throw;
1292	coro->throw = 0;
1293	sv_setsv (ERRSV, exception);
1294	croak (0);
1295	}
1296	}
1297	}	1689	}
1298	}	1690	}
1299
1300	struct transfer_args
1301	{
1302	struct coro *prev, *next;
1303	};
1304		1691
1305	#define TRANSFER(ta, force_cctx) transfer (aTHX_ (ta).prev, (ta).next, (force_cctx))	1692	#define TRANSFER(ta, force_cctx) transfer (aTHX_ (ta).prev, (ta).next, (force_cctx))
1306	#define TRANSFER_CHECK(ta) transfer_check (aTHX_ (ta).prev, (ta).next)	1693	#define TRANSFER_CHECK(ta) transfer_check (aTHX_ (ta).prev, (ta).next)
1307		1694
1308	/** high level stuff ********************************************************/	1695	/** high level stuff ********************************************************/
1309		1696
		1697	/* this function is actually Coro, not Coro::State, but we call it from here */
		1698	/* because it is convenient - but it hasn't been declared yet for that reason */
1310	static int	1699	static void
		1700	coro_call_on_destroy (pTHX_ struct coro *coro);
		1701
		1702	static void
1311	coro_state_destroy (pTHX_ struct coro *coro)	1703	coro_state_destroy (pTHX_ struct coro *coro)
1312	{	1704	{
1313	if (coro->flags & CF_DESTROYED)	1705	if (coro->flags & CF_ZOMBIE)
1314	return 0;	1706	return;
1315		1707
		1708	slf_destroy (aTHX_ coro);
		1709
1316	coro->flags |= CF_DESTROYED;	1710	coro->flags |= CF_ZOMBIE;
1317		1711
1318	if (coro->flags & CF_READY)	1712	if (coro->flags & CF_READY)
1319	{	1713	{
1320	/* reduce nready, as destroying a ready coro effectively unreadies it */	1714	/* reduce nready, as destroying a ready coro effectively unreadies it */
1321	/* alternative: look through all ready queues and remove the coro */	1715	/* alternative: look through all ready queues and remove the coro */
1322	LOCK;
1323	--coro_nready;	1716	--coro_nready;
1324	UNLOCK;
1325	}	1717	}
1326	else	1718	else
1327	coro->flags |= CF_READY; /* make sure it is NOT put into the readyqueue */	1719	coro->flags |= CF_READY; /* make sure it is NOT put into the readyqueue */
1328
1329	if (coro->mainstack && coro->mainstack != main_mainstack)
1330	{
1331	struct coro temp;
1332
1333	if (coro->flags & CF_RUNNING)
1334	croak ("FATAL: tried to destroy currently running coroutine");
1335
1336	save_perl (aTHX_ &temp);
1337	load_perl (aTHX_ coro);
1338
1339	coro_destroy (aTHX_ coro);
1340
1341	load_perl (aTHX_ &temp);
1342
1343	coro->slot = 0;
1344	}
1345
1346	cctx_destroy (coro->cctx);
1347	SvREFCNT_dec (coro->args);
1348		1720
1349	if (coro->next) coro->next->prev = coro->prev;	1721	if (coro->next) coro->next->prev = coro->prev;
1350	if (coro->prev) coro->prev->next = coro->next;	1722	if (coro->prev) coro->prev->next = coro->next;
1351	if (coro == coro_first) coro_first = coro->next;	1723	if (coro == coro_first) coro_first = coro->next;
1352		1724
1353	return 1;	1725	if (coro->mainstack
		1726	&& coro->mainstack != main_mainstack
		1727	&& coro->slot
		1728	&& !PL_dirty)
		1729	destroy_perl (aTHX_ coro);
		1730
		1731	cctx_destroy (coro->cctx);
		1732	SvREFCNT_dec (coro->startcv);
		1733	SvREFCNT_dec (coro->args);
		1734	SvREFCNT_dec (coro->swap_sv);
		1735	SvREFCNT_dec (CORO_THROW);
		1736
		1737	coro_call_on_destroy (aTHX_ coro);
		1738
		1739	/* more destruction mayhem in coro_state_free */
1354	}	1740	}
1355		1741
1356	static int	1742	static int
1357	coro_state_free (pTHX_ SV *sv, MAGIC *mg)	1743	coro_state_free (pTHX_ SV *sv, MAGIC *mg)
1358	{	1744	{
1359	struct coro *coro = (struct coro *)mg->mg_ptr;	1745	struct coro *coro = (struct coro *)mg->mg_ptr;
1360	mg->mg_ptr = 0;	1746	mg->mg_ptr = 0;
1361		1747
1362	coro->hv = 0;
1363
1364	if (--coro->refcnt < 0)
1365	{
1366	coro_state_destroy (aTHX_ coro);	1748	coro_state_destroy (aTHX_ coro);
		1749	SvREFCNT_dec (coro->on_destroy);
		1750	SvREFCNT_dec (coro->status);
		1751
1367	Safefree (coro);	1752	Safefree (coro);
1368	}
1369		1753
1370	return 0;	1754	return 0;
1371	}	1755	}
1372		1756
1373	static int	1757	static int ecb_cold
1374	coro_state_dup (pTHX_ MAGIC *mg, CLONE_PARAMS *params)	1758	coro_state_dup (pTHX_ MAGIC *mg, CLONE_PARAMS *params)
1375	{	1759	{
1376	struct coro *coro = (struct coro *)mg->mg_ptr;	1760	/* called when perl clones the current process the slow way (windows process emulation) */
1377		1761	/* WE SIMply nuke the pointers in the copy, causing perl to croak */
1378	++coro->refcnt;	1762	mg->mg_ptr = 0;
		1763	mg->mg_virtual = 0;
1379		1764
1380	return 0;	1765	return 0;
1381	}	1766	}
1382		1767
1383	static MGVTBL coro_state_vtbl = {	1768	static MGVTBL coro_state_vtbl = {
…		…
1390	# define MGf_DUP 0	1775	# define MGf_DUP 0
1391	#endif	1776	#endif
1392	};	1777	};
1393		1778
1394	static void	1779	static void
1395	prepare_transfer (pTHX_ struct transfer_args *ta, SV *prev_sv, SV *next_sv)	1780	prepare_transfer (pTHX_ struct coro_transfer_args *ta, SV *prev_sv, SV *next_sv)
1396	{	1781	{
1397	ta->prev = SvSTATE (prev_sv);	1782	ta->prev = SvSTATE (prev_sv);
1398	ta->next = SvSTATE (next_sv);	1783	ta->next = SvSTATE (next_sv);
1399	TRANSFER_CHECK (*ta);	1784	TRANSFER_CHECK (*ta);
1400	}	1785	}
1401		1786
1402	static void	1787	static void
1403	api_transfer (SV *prev_sv, SV *next_sv)	1788	api_transfer (pTHX_ SV *prev_sv, SV *next_sv)
1404	{	1789	{
1405	dTHX;
1406	struct transfer_args ta;	1790	struct coro_transfer_args ta;
1407		1791
1408	prepare_transfer (aTHX_ &ta, prev_sv, next_sv);	1792	prepare_transfer (aTHX_ &ta, prev_sv, next_sv);
1409	TRANSFER (ta, 1);	1793	TRANSFER (ta, 1);
1410	}	1794	}
1411		1795
1412	/** Coro ********************************************************************/	1796	/** Coro ********************************************************************/
1413		1797
1414	static void	1798	ecb_inline void
1415	coro_enq (pTHX_ SV *coro_sv)	1799	coro_enq (pTHX_ struct coro *coro)
1416	{	1800	{
1417	av_push (coro_ready [SvSTATE (coro_sv)->prio - PRIO_MIN], coro_sv);	1801	struct coro **ready = coro_ready [coro->prio - CORO_PRIO_MIN];
1418	}
1419		1802
1420	static SV *	1803	SvREFCNT_inc_NN (coro->hv);
		1804
		1805	coro->next_ready = 0;
		1806	*(ready [0] ? &ready [1]->next_ready : &ready [0]) = coro;
		1807	ready [1] = coro;
		1808	}
		1809
		1810	ecb_inline struct coro *
1421	coro_deq (pTHX)	1811	coro_deq (pTHX)
1422	{	1812	{
1423	int prio;	1813	int prio;
1424		1814
1425	for (prio = PRIO_MAX - PRIO_MIN + 1; --prio >= 0; )	1815	for (prio = CORO_PRIO_MAX - CORO_PRIO_MIN + 1; --prio >= 0; )
1426	if (AvFILLp (coro_ready [prio]) >= 0)	1816	{
1427	return av_shift (coro_ready [prio]);	1817	struct coro **ready = coro_ready [prio];
		1818
		1819	if (ready [0])
		1820	{
		1821	struct coro *coro = ready [0];
		1822	ready [0] = coro->next_ready;
		1823	return coro;
		1824	}
		1825	}
1428		1826
1429	return 0;	1827	return 0;
1430	}	1828	}
1431		1829
		1830	static void
		1831	invoke_sv_ready_hook_helper (void)
		1832	{
		1833	dTHX;
		1834	dSP;
		1835
		1836	ENTER;
		1837	SAVETMPS;
		1838
		1839	PUSHMARK (SP);
		1840	PUTBACK;
		1841	call_sv (coro_readyhook, G_VOID | G_DISCARD);
		1842
		1843	FREETMPS;
		1844	LEAVE;
		1845	}
		1846
1432	static int	1847	static int
1433	api_ready (SV *coro_sv)	1848	api_ready (pTHX_ SV *coro_sv)
1434	{	1849	{
1435	dTHX;
1436	struct coro *coro;
1437	SV *sv_hook;
1438	void (*xs_hook)(void);
1439
1440	if (SvROK (coro_sv))
1441	coro_sv = SvRV (coro_sv);
1442
1443	coro = SvSTATE (coro_sv);	1850	struct coro *coro = SvSTATE (coro_sv);
1444		1851
1445	if (coro->flags & CF_READY)	1852	if (coro->flags & CF_READY)
1446	return 0;	1853	return 0;
1447		1854
1448	coro->flags |= CF_READY;	1855	coro->flags |= CF_READY;
1449		1856
1450	LOCK;	1857	coro_enq (aTHX_ coro);
1451		1858
1452	sv_hook = coro_nready ? 0 : coro_readyhook;	1859	if (!coro_nready++)
1453	xs_hook = coro_nready ? 0 : coroapi.readyhook;	1860	if (coroapi.readyhook)
1454		1861	coroapi.readyhook ();
1455	coro_enq (aTHX_ SvREFCNT_inc (coro_sv));
1456	++coro_nready;
1457
1458	UNLOCK;
1459
1460	if (sv_hook)
1461	{
1462	dSP;
1463
1464	ENTER;
1465	SAVETMPS;
1466
1467	PUSHMARK (SP);
1468	PUTBACK;
1469	call_sv (sv_hook, G_DISCARD);
1470	SPAGAIN;
1471
1472	FREETMPS;
1473	LEAVE;
1474	}
1475
1476	if (xs_hook)
1477	xs_hook ();
1478		1862
1479	return 1;	1863	return 1;
1480	}	1864	}
1481		1865
1482	static int	1866	static int
1483	api_is_ready (SV *coro_sv)	1867	api_is_ready (pTHX_ SV *coro_sv)
1484	{	1868	{
1485	dTHX;
1486	return !!(SvSTATE (coro_sv)->flags & CF_READY);	1869	return !!(SvSTATE (coro_sv)->flags & CF_READY);
1487	}	1870	}
1488		1871
1489	static void	1872	/* expects to own a reference to next->hv */
		1873	ecb_inline void
1490	prepare_schedule (pTHX_ struct transfer_args *ta)	1874	prepare_schedule_to (pTHX_ struct coro_transfer_args *ta, struct coro *next)
1491	{	1875	{
1492	SV *prev_sv, *next_sv;
1493
1494	for (;;)
1495	{
1496	LOCK;
1497	next_sv = coro_deq (aTHX);
1498
1499	/* nothing to schedule: call the idle handler */
1500	if (expect_false (!next_sv))
1501	{
1502	dSP;
1503	UNLOCK;
1504
1505	ENTER;
1506	SAVETMPS;
1507
1508	PUSHMARK (SP);
1509	PUTBACK;
1510	call_sv (get_sv ("Coro::idle", FALSE), G_DISCARD);
1511	SPAGAIN;
1512
1513	FREETMPS;
1514	LEAVE;
1515	continue;
1516	}
1517
1518	ta->next = SvSTATE (next_sv);
1519
1520	/* cannot transfer to destroyed coros, skip and look for next */
1521	if (expect_false (ta->next->flags & CF_DESTROYED))
1522	{
1523	UNLOCK;
1524	SvREFCNT_dec (next_sv);
1525	/* coro_nready is already taken care of by destroy */
1526	continue;
1527	}
1528
1529	--coro_nready;
1530	UNLOCK;
1531	break;
1532	}
1533
1534	/* free this only after the transfer */
1535	prev_sv = SvRV (coro_current);	1876	SV *prev_sv = SvRV (coro_current);
		1877
1536	ta->prev = SvSTATE (prev_sv);	1878	ta->prev = SvSTATE_hv (prev_sv);
		1879	ta->next = next;
		1880
1537	TRANSFER_CHECK (*ta);	1881	TRANSFER_CHECK (*ta);
1538	assert (ta->next->flags & CF_READY);	1882
1539	ta->next->flags &= ~CF_READY;
1540	SvRV_set (coro_current, next_sv);	1883	SvRV_set (coro_current, (SV *)next->hv);
1541		1884
1542	LOCK;
1543	free_coro_mortal (aTHX);	1885	free_coro_mortal (aTHX);
1544	coro_mortal = prev_sv;	1886	coro_mortal = prev_sv;
1545	UNLOCK;
1546	}	1887	}
1547		1888
1548	static void	1889	static void
		1890	prepare_schedule (pTHX_ struct coro_transfer_args *ta)
		1891	{
		1892	for (;;)
		1893	{
		1894	struct coro *next = coro_deq (aTHX);
		1895
		1896	if (ecb_expect_true (next))
		1897	{
		1898	/* cannot transfer to destroyed coros, skip and look for next */
		1899	if (ecb_expect_false (next->flags & (CF_ZOMBIE | CF_SUSPENDED)))
		1900	SvREFCNT_dec (next->hv); /* coro_nready has already been taken care of by destroy */
		1901	else
		1902	{
		1903	next->flags &= ~CF_READY;
		1904	--coro_nready;
		1905
		1906	prepare_schedule_to (aTHX_ ta, next);
		1907	break;
		1908	}
		1909	}
		1910	else
		1911	{
		1912	/* nothing to schedule: call the idle handler */
		1913	if (SvROK (sv_idle)
		1914	&& SvOBJECT (SvRV (sv_idle)))
		1915	{
		1916	if (SvRV (sv_idle) == SvRV (coro_current))
		1917	croak ("FATAL: $Coro::IDLE blocked itself - did you try to block inside an event loop callback? Caught");
		1918
		1919	++coro_nready; /* hack so that api_ready doesn't invoke ready hook */
		1920	api_ready (aTHX_ SvRV (sv_idle));
		1921	--coro_nready;
		1922	}
		1923	else
		1924	{
		1925	/* TODO: deprecated, remove, cannot work reliably *//*D*/
		1926	dSP;
		1927
		1928	ENTER;
		1929	SAVETMPS;
		1930
		1931	PUSHMARK (SP);
		1932	PUTBACK;
		1933	call_sv (sv_idle, G_VOID | G_DISCARD);
		1934
		1935	FREETMPS;
		1936	LEAVE;
		1937	}
		1938	}
		1939	}
		1940	}
		1941
		1942	ecb_inline void
1549	prepare_cede (pTHX_ struct transfer_args *ta)	1943	prepare_cede (pTHX_ struct coro_transfer_args *ta)
1550	{	1944	{
1551	api_ready (coro_current);	1945	api_ready (aTHX_ coro_current);
1552	prepare_schedule (aTHX_ ta);	1946	prepare_schedule (aTHX_ ta);
1553	}	1947	}
1554		1948
		1949	ecb_inline void
		1950	prepare_cede_notself (pTHX_ struct coro_transfer_args *ta)
		1951	{
		1952	SV *prev = SvRV (coro_current);
		1953
		1954	if (coro_nready)
		1955	{
		1956	prepare_schedule (aTHX_ ta);
		1957	api_ready (aTHX_ prev);
		1958	}
		1959	else
		1960	prepare_nop (aTHX_ ta);
		1961	}
		1962
		1963	static void
		1964	api_schedule (pTHX)
		1965	{
		1966	struct coro_transfer_args ta;
		1967
		1968	prepare_schedule (aTHX_ &ta);
		1969	TRANSFER (ta, 1);
		1970	}
		1971
		1972	static void
		1973	api_schedule_to (pTHX_ SV *coro_sv)
		1974	{
		1975	struct coro_transfer_args ta;
		1976	struct coro *next = SvSTATE (coro_sv);
		1977
		1978	SvREFCNT_inc_NN (coro_sv);
		1979	prepare_schedule_to (aTHX_ &ta, next);
		1980	}
		1981
1555	static int	1982	static int
1556	prepare_cede_notself (pTHX_ struct transfer_args *ta)	1983	api_cede (pTHX)
1557	{	1984	{
1558	if (coro_nready)	1985	struct coro_transfer_args ta;
1559	{	1986
1560	SV *prev = SvRV (coro_current);
1561	prepare_schedule (aTHX_ ta);	1987	prepare_cede (aTHX_ &ta);
1562	api_ready (prev);	1988
		1989	if (ecb_expect_true (ta.prev != ta.next))
		1990	{
		1991	TRANSFER (ta, 1);
1563	return 1;	1992	return 1;
1564	}	1993	}
1565	else	1994	else
1566	return 0;	1995	return 0;
1567	}	1996	}
1568		1997
1569	static void
1570	api_schedule (void)
1571	{
1572	dTHX;
1573	struct transfer_args ta;
1574
1575	prepare_schedule (aTHX_ &ta);
1576	TRANSFER (ta, 1);
1577	}
1578
1579	static int	1998	static int
1580	api_cede (void)	1999	api_cede_notself (pTHX)
1581	{	2000	{
1582	dTHX;	2001	if (coro_nready)
		2002	{
1583	struct transfer_args ta;	2003	struct coro_transfer_args ta;
1584		2004
1585	prepare_cede (aTHX_ &ta);	2005	prepare_cede_notself (aTHX_ &ta);
1586
1587	if (expect_true (ta.prev != ta.next))
1588	{
1589	TRANSFER (ta, 1);	2006	TRANSFER (ta, 1);
1590	return 1;	2007	return 1;
1591	}	2008	}
1592	else	2009	else
1593	return 0;	2010	return 0;
1594	}	2011	}
1595		2012
1596	static int	2013	static void
1597	api_cede_notself (void)
1598	{
1599	dTHX;
1600	struct transfer_args ta;
1601
1602	if (prepare_cede_notself (aTHX_ &ta))
1603	{
1604	TRANSFER (ta, 1);
1605	return 1;
1606	}
1607	else
1608	return 0;
1609	}
1610
1611	static void
1612	api_trace (SV *coro_sv, int flags)	2014	api_trace (pTHX_ SV *coro_sv, int flags)
1613	{	2015	{
1614	dTHX;
1615	struct coro *coro = SvSTATE (coro_sv);	2016	struct coro *coro = SvSTATE (coro_sv);
1616		2017
		2018	if (coro->flags & CF_RUNNING)
		2019	croak ("cannot enable tracing on a running coroutine, caught");
		2020
1617	if (flags & CC_TRACE)	2021	if (flags & CC_TRACE)
1618	{	2022	{
1619	if (!coro->cctx)	2023	if (!coro->cctx)
1620	coro->cctx = cctx_new ();	2024	coro->cctx = cctx_new_run ();
1621	else if (!(coro->cctx->flags & CC_TRACE))	2025	else if (!(coro->cctx->flags & CC_TRACE))
1622	croak ("cannot enable tracing on coroutine with custom stack");	2026	croak ("cannot enable tracing on coroutine with custom stack, caught");
1623		2027
1624	coro->cctx->flags |= CC_NOREUSE | (flags & (CC_TRACE | CC_TRACE_ALL));	2028	coro->cctx->flags |= CC_NOREUSE | (flags & (CC_TRACE | CC_TRACE_ALL));
1625	}	2029	}
1626	else if (coro->cctx && coro->cctx->flags & CC_TRACE)	2030	else if (coro->cctx && coro->cctx->flags & CC_TRACE)
1627	{	2031	{
…		…
1632	else	2036	else
1633	coro->slot->runops = RUNOPS_DEFAULT;	2037	coro->slot->runops = RUNOPS_DEFAULT;
1634	}	2038	}
1635	}	2039	}
1636		2040
		2041	static void
		2042	coro_push_av (pTHX_ AV *av, I32 gimme_v)
		2043	{
		2044	if (AvFILLp (av) >= 0 && gimme_v != G_VOID)
		2045	{
		2046	dSP;
		2047
		2048	if (gimme_v == G_SCALAR)
		2049	XPUSHs (AvARRAY (av)[AvFILLp (av)]);
		2050	else
		2051	{
		2052	int i;
		2053	EXTEND (SP, AvFILLp (av) + 1);
		2054
		2055	for (i = 0; i <= AvFILLp (av); ++i)
		2056	PUSHs (AvARRAY (av)[i]);
		2057	}
		2058
		2059	PUTBACK;
		2060	}
		2061	}
		2062
		2063	static void
		2064	coro_push_on_destroy (pTHX_ struct coro *coro, SV *cb)
		2065	{
		2066	if (!coro->on_destroy)
		2067	coro->on_destroy = newAV ();
		2068
		2069	av_push (coro->on_destroy, cb);
		2070	}
		2071
		2072	static void
		2073	slf_destroy_join (pTHX_ struct CoroSLF *frame)
		2074	{
		2075	SvREFCNT_dec ((SV *)((struct coro *)frame->data)->hv);
		2076	}
		2077
1637	static int	2078	static int
1638	coro_gensub_free (pTHX_ SV *sv, MAGIC *mg)	2079	slf_check_join (pTHX_ struct CoroSLF *frame)
1639	{	2080	{
1640	AV *padlist;	2081	struct coro *coro = (struct coro *)frame->data;
1641	AV *av = (AV *)mg->mg_obj;
1642		2082
1643	abort ();	2083	if (!coro->status)
		2084	return 1;
		2085
		2086	frame->destroy = 0;
		2087
		2088	coro_push_av (aTHX_ coro->status, GIMME_V);
		2089
		2090	SvREFCNT_dec ((SV *)coro->hv);
1644		2091
1645	return 0;	2092	return 0;
1646	}	2093	}
1647		2094
1648	static MGVTBL coro_gensub_vtbl = {	2095	static void
1649	0, 0, 0, 0,	2096	slf_init_join (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
1650	coro_gensub_free	2097	{
1651	};	2098	struct coro *coro = SvSTATE (items > 0 ? arg [0] : &PL_sv_undef);
		2099
		2100	if (items > 1)
		2101	croak ("join called with too many arguments");
		2102
		2103	if (coro->status)
		2104	frame->prepare = prepare_nop;
		2105	else
		2106	{
		2107	coro_push_on_destroy (aTHX_ coro, SvREFCNT_inc_NN (SvRV (coro_current)));
		2108	frame->prepare = prepare_schedule;
		2109	}
		2110
		2111	frame->check = slf_check_join;
		2112	frame->destroy = slf_destroy_join;
		2113	frame->data = (void *)coro;
		2114	SvREFCNT_inc (coro->hv);
		2115	}
		2116
		2117	static void
		2118	coro_call_on_destroy (pTHX_ struct coro *coro)
		2119	{
		2120	AV *od = coro->on_destroy;
		2121
		2122	if (!od)
		2123	return;
		2124
		2125	coro->on_destroy = 0;
		2126	sv_2mortal ((SV *)od);
		2127
		2128	while (AvFILLp (od) >= 0)
		2129	{
		2130	SV *cb = sv_2mortal (av_pop (od));
		2131
		2132	/* coro hv's (and only hv's at the moment) are supported as well */
		2133	if (SvSTATEhv_p (aTHX_ cb))
		2134	api_ready (aTHX_ cb);
		2135	else
		2136	{
		2137	dSP; /* don't disturb outer sp */
		2138	PUSHMARK (SP);
		2139
		2140	if (coro->status)
		2141	{
		2142	PUTBACK;
		2143	coro_push_av (aTHX_ coro->status, G_ARRAY);
		2144	SPAGAIN;
		2145	}
		2146
		2147	PUTBACK;
		2148	call_sv (cb, G_VOID | G_DISCARD);
		2149	}
		2150	}
		2151	}
		2152
		2153	static void
		2154	coro_set_status (pTHX_ struct coro *coro, SV **arg, int items)
		2155	{
		2156	AV *av;
		2157
		2158	if (coro->status)
		2159	{
		2160	av = coro->status;
		2161	av_clear (av);
		2162	}
		2163	else
		2164	av = coro->status = newAV ();
		2165
		2166	/* items are actually not so common, so optimise for this case */
		2167	if (items)
		2168	{
		2169	int i;
		2170
		2171	av_extend (av, items - 1);
		2172
		2173	for (i = 0; i < items; ++i)
		2174	av_push (av, SvREFCNT_inc_NN (arg [i]));
		2175	}
		2176	}
		2177
		2178	static void
		2179	slf_init_terminate_cancel_common (pTHX_ struct CoroSLF *frame, HV *coro_hv)
		2180	{
		2181	av_push (av_destroy, (SV *)newRV_inc ((SV *)coro_hv)); /* RVinc for perl */
		2182	api_ready (aTHX_ sv_manager);
		2183
		2184	frame->prepare = prepare_schedule;
		2185	frame->check = slf_check_repeat;
		2186
		2187	/* as a minor optimisation, we could unwind all stacks here */
		2188	/* but that puts extra pressure on pp_slf, and is not worth much */
		2189	/*coro_unwind_stacks (aTHX);*/
		2190	}
		2191
		2192	static void
		2193	slf_init_terminate (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		2194	{
		2195	HV *coro_hv = (HV *)SvRV (coro_current);
		2196
		2197	coro_set_status (aTHX_ SvSTATE ((SV *)coro_hv), arg, items);
		2198	slf_init_terminate_cancel_common (aTHX_ frame, coro_hv);
		2199	}
		2200
		2201	static void
		2202	slf_init_cancel (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		2203	{
		2204	HV *coro_hv;
		2205	struct coro *coro;
		2206
		2207	if (items <= 0)
		2208	croak ("Coro::cancel called without coro object,");
		2209
		2210	coro = SvSTATE (arg [0]);
		2211	coro_hv = coro->hv;
		2212
		2213	coro_set_status (aTHX_ coro, arg + 1, items - 1);
		2214
		2215	if (ecb_expect_false (coro->flags & CF_NOCANCEL))
		2216	{
		2217	/* coro currently busy cancelling something, so just notify it */
		2218	coro->slf_frame.data = (void *)coro;
		2219
		2220	frame->prepare = prepare_nop;
		2221	frame->check = slf_check_nop;
		2222	}
		2223	else if (coro_hv == (HV *)SvRV (coro_current))
		2224	{
		2225	/* cancelling the current coro is allowed, and equals terminate */
		2226	slf_init_terminate_cancel_common (aTHX_ frame, coro_hv);
		2227	}
		2228	else
		2229	{
		2230	struct coro *self = SvSTATE_current;
		2231
		2232	/* otherwise we cancel directly, purely for speed reasons
		2233	* unfortunately, this requires some magic trickery, as
		2234	* somebody else could cancel us, so we have to fight the cancellation.
		2235	* this is ugly, and hopefully fully worth the extra speed.
		2236	* besides, I can't get the slow-but-safe version working...
		2237	*/
		2238	slf_frame.data = 0;
		2239	self->flags |= CF_NOCANCEL;
		2240	coro_state_destroy (aTHX_ coro);
		2241	self->flags &= ~CF_NOCANCEL;
		2242
		2243	if (slf_frame.data)
		2244	{
		2245	/* while we were busy we have been cancelled, so terminate */
		2246	slf_init_terminate_cancel_common (aTHX_ frame, self->hv);
		2247	}
		2248	else
		2249	{
		2250	frame->prepare = prepare_nop;
		2251	frame->check = slf_check_nop;
		2252	}
		2253	}
		2254	}
		2255
		2256	static int
		2257	slf_check_safe_cancel (pTHX_ struct CoroSLF *frame)
		2258	{
		2259	frame->prepare = 0;
		2260	coro_unwind_stacks (aTHX);
		2261
		2262	slf_init_terminate_cancel_common (aTHX_ frame, (HV *)SvRV (coro_current));
		2263
		2264	return 1;
		2265	}
		2266
		2267	static int
		2268	safe_cancel (pTHX_ struct coro *coro, SV **arg, int items)
		2269	{
		2270	if (coro->cctx)
		2271	croak ("coro inside C callback, unable to cancel at this time, caught");
		2272
		2273	if (coro->flags & CF_NEW)
		2274	{
		2275	coro_set_status (aTHX_ coro, arg, items);
		2276	coro_state_destroy (aTHX_ coro);
		2277	}
		2278	else
		2279	{
		2280	if (!coro->slf_frame.prepare)
		2281	croak ("coro outside an SLF function, unable to cancel at this time, caught");
		2282
		2283	slf_destroy (aTHX_ coro);
		2284
		2285	coro_set_status (aTHX_ coro, arg, items);
		2286	coro->slf_frame.prepare = prepare_nop;
		2287	coro->slf_frame.check = slf_check_safe_cancel;
		2288
		2289	api_ready (aTHX_ (SV *)coro->hv);
		2290	}
		2291
		2292	return 1;
		2293	}
		2294
		2295	/*****************************************************************************/
		2296	/* async pool handler */
		2297
		2298	static int
		2299	slf_check_pool_handler (pTHX_ struct CoroSLF *frame)
		2300	{
		2301	HV *hv = (HV *)SvRV (coro_current);
		2302	struct coro *coro = (struct coro *)frame->data;
		2303
		2304	if (!coro->invoke_cb)
		2305	return 1; /* loop till we have invoke */
		2306	else
		2307	{
		2308	hv_store (hv, "desc", sizeof ("desc") - 1,
		2309	newSVpvn ("[async_pool]", sizeof ("[async_pool]") - 1), 0);
		2310
		2311	coro->saved_deffh = SvREFCNT_inc_NN ((SV *)PL_defoutgv);
		2312
		2313	{
		2314	dSP;
		2315	XPUSHs (sv_2mortal (coro->invoke_cb)); coro->invoke_cb = 0;
		2316	PUTBACK;
		2317	}
		2318
		2319	SvREFCNT_dec (GvAV (PL_defgv));
		2320	GvAV (PL_defgv) = coro->invoke_av;
		2321	coro->invoke_av = 0;
		2322
		2323	return 0;
		2324	}
		2325	}
		2326
		2327	static void
		2328	slf_init_pool_handler (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		2329	{
		2330	HV *hv = (HV *)SvRV (coro_current);
		2331	struct coro *coro = SvSTATE_hv ((SV *)hv);
		2332
		2333	if (ecb_expect_true (coro->saved_deffh))
		2334	{
		2335	/* subsequent iteration */
		2336	SvREFCNT_dec ((SV *)PL_defoutgv); PL_defoutgv = (GV *)coro->saved_deffh;
		2337	coro->saved_deffh = 0;
		2338
		2339	if (coro_rss (aTHX_ coro) > SvUV (sv_pool_rss)
		2340	|| av_len (av_async_pool) + 1 >= SvIV (sv_pool_size))
		2341	{
		2342	slf_init_terminate_cancel_common (aTHX_ frame, hv);
		2343	return;
		2344	}
		2345	else
		2346	{
		2347	av_clear (GvAV (PL_defgv));
		2348	hv_store (hv, "desc", sizeof ("desc") - 1, SvREFCNT_inc_NN (sv_async_pool_idle), 0);
		2349
		2350	coro->prio = 0;
		2351
		2352	if (coro->cctx && (coro->cctx->flags & CC_TRACE))
		2353	api_trace (aTHX_ coro_current, 0);
		2354
		2355	frame->prepare = prepare_schedule;
		2356	av_push (av_async_pool, SvREFCNT_inc (hv));
		2357	}
		2358	}
		2359	else
		2360	{
		2361	/* first iteration, simply fall through */
		2362	frame->prepare = prepare_nop;
		2363	}
		2364
		2365	frame->check = slf_check_pool_handler;
		2366	frame->data = (void *)coro;
		2367	}
		2368
		2369	/*****************************************************************************/
		2370	/* rouse callback */
		2371
		2372	#define CORO_MAGIC_type_rouse PERL_MAGIC_ext
		2373
		2374	static void
		2375	coro_rouse_callback (pTHX_ CV *cv)
		2376	{
		2377	dXSARGS;
		2378	SV *data = (SV *)S_GENSUB_ARG;
		2379
		2380	if (SvTYPE (SvRV (data)) != SVt_PVAV)
		2381	{
		2382	/* first call, set args */
		2383	SV *coro = SvRV (data);
		2384	AV *av = newAV ();
		2385
		2386	SvRV_set (data, (SV *)av);
		2387
		2388	/* better take a full copy of the arguments */
		2389	while (items--)
		2390	av_store (av, items, newSVsv (ST (items)));
		2391
		2392	api_ready (aTHX_ coro);
		2393	SvREFCNT_dec (coro);
		2394	}
		2395
		2396	XSRETURN_EMPTY;
		2397	}
		2398
		2399	static int
		2400	slf_check_rouse_wait (pTHX_ struct CoroSLF *frame)
		2401	{
		2402	SV *data = (SV *)frame->data;
		2403
		2404	if (CORO_THROW)
		2405	return 0;
		2406
		2407	if (SvTYPE (SvRV (data)) != SVt_PVAV)
		2408	return 1;
		2409
		2410	/* now push all results on the stack */
		2411	{
		2412	dSP;
		2413	AV *av = (AV *)SvRV (data);
		2414	int i;
		2415
		2416	EXTEND (SP, AvFILLp (av) + 1);
		2417	for (i = 0; i <= AvFILLp (av); ++i)
		2418	PUSHs (sv_2mortal (AvARRAY (av)[i]));
		2419
		2420	/* we have stolen the elements, so set length to zero and free */
		2421	AvFILLp (av) = -1;
		2422	av_undef (av);
		2423
		2424	PUTBACK;
		2425	}
		2426
		2427	return 0;
		2428	}
		2429
		2430	static void
		2431	slf_init_rouse_wait (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		2432	{
		2433	SV *cb;
		2434
		2435	if (items)
		2436	cb = arg [0];
		2437	else
		2438	{
		2439	struct coro *coro = SvSTATE_current;
		2440
		2441	if (!coro->rouse_cb)
		2442	croak ("Coro::rouse_wait called without rouse callback, and no default rouse callback found either,");
		2443
		2444	cb = sv_2mortal (coro->rouse_cb);
		2445	coro->rouse_cb = 0;
		2446	}
		2447
		2448	if (!SvROK (cb)
		2449	|| SvTYPE (SvRV (cb)) != SVt_PVCV
		2450	|| CvXSUB ((CV *)SvRV (cb)) != coro_rouse_callback)
		2451	croak ("Coro::rouse_wait called with illegal callback argument,");
		2452
		2453	{
		2454	CV *cv = (CV *)SvRV (cb); /* for S_GENSUB_ARG */
		2455	SV *data = (SV *)S_GENSUB_ARG;
		2456
		2457	frame->data = (void *)data;
		2458	frame->prepare = SvTYPE (SvRV (data)) == SVt_PVAV ? prepare_nop : prepare_schedule;
		2459	frame->check = slf_check_rouse_wait;
		2460	}
		2461	}
		2462
		2463	static SV *
		2464	coro_new_rouse_cb (pTHX)
		2465	{
		2466	HV *hv = (HV *)SvRV (coro_current);
		2467	struct coro *coro = SvSTATE_hv (hv);
		2468	SV *data = newRV_inc ((SV *)hv);
		2469	SV *cb = s_gensub (aTHX_ coro_rouse_callback, (void *)data);
		2470
		2471	sv_magicext (SvRV (cb), data, CORO_MAGIC_type_rouse, 0, 0, 0);
		2472	SvREFCNT_dec (data); /* magicext increases the refcount */
		2473
		2474	SvREFCNT_dec (coro->rouse_cb);
		2475	coro->rouse_cb = SvREFCNT_inc_NN (cb);
		2476
		2477	return cb;
		2478	}
		2479
		2480	/*****************************************************************************/
		2481	/* schedule-like-function opcode (SLF) */
		2482
		2483	static UNOP slf_restore; /* restore stack as entersub did, for first-re-run */
		2484	static const CV *slf_cv;
		2485	static SV **slf_argv;
		2486	static int slf_argc, slf_arga; /* count, allocated */
		2487	static I32 slf_ax; /* top of stack, for restore */
		2488
		2489	/* this restores the stack in the case we patched the entersub, to */
		2490	/* recreate the stack frame as perl will on following calls */
		2491	/* since entersub cleared the stack */
		2492	static OP *
		2493	pp_restore (pTHX)
		2494	{
		2495	int i;
		2496	SV **SP = PL_stack_base + slf_ax;
		2497
		2498	PUSHMARK (SP);
		2499
		2500	EXTEND (SP, slf_argc + 1);
		2501
		2502	for (i = 0; i < slf_argc; ++i)
		2503	PUSHs (sv_2mortal (slf_argv [i]));
		2504
		2505	PUSHs ((SV *)CvGV (slf_cv));
		2506
		2507	RETURNOP (slf_restore.op_first);
		2508	}
		2509
		2510	static void
		2511	slf_prepare_transfer (pTHX_ struct coro_transfer_args *ta)
		2512	{
		2513	SV **arg = (SV **)slf_frame.data;
		2514
		2515	prepare_transfer (aTHX_ ta, arg [0], arg [1]);
		2516	}
		2517
		2518	static void
		2519	slf_init_transfer (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		2520	{
		2521	if (items != 2)
		2522	croak ("Coro::State::transfer (prev, next) expects two arguments, not %d,", items);
		2523
		2524	frame->prepare = slf_prepare_transfer;
		2525	frame->check = slf_check_nop;
		2526	frame->data = (void *)arg; /* let's hope it will stay valid */
		2527	}
		2528
		2529	static void
		2530	slf_init_schedule (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		2531	{
		2532	frame->prepare = prepare_schedule;
		2533	frame->check = slf_check_nop;
		2534	}
		2535
		2536	static void
		2537	slf_prepare_schedule_to (pTHX_ struct coro_transfer_args *ta)
		2538	{
		2539	struct coro *next = (struct coro *)slf_frame.data;
		2540
		2541	SvREFCNT_inc_NN (next->hv);
		2542	prepare_schedule_to (aTHX_ ta, next);
		2543	}
		2544
		2545	static void
		2546	slf_init_schedule_to (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		2547	{
		2548	if (!items)
		2549	croak ("Coro::schedule_to expects a coroutine argument, caught");
		2550
		2551	frame->data = (void *)SvSTATE (arg [0]);
		2552	frame->prepare = slf_prepare_schedule_to;
		2553	frame->check = slf_check_nop;
		2554	}
		2555
		2556	static void
		2557	slf_init_cede_to (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		2558	{
		2559	api_ready (aTHX_ SvRV (coro_current));
		2560
		2561	slf_init_schedule_to (aTHX_ frame, cv, arg, items);
		2562	}
		2563
		2564	static void
		2565	slf_init_cede (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		2566	{
		2567	frame->prepare = prepare_cede;
		2568	frame->check = slf_check_nop;
		2569	}
		2570
		2571	static void
		2572	slf_init_cede_notself (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		2573	{
		2574	frame->prepare = prepare_cede_notself;
		2575	frame->check = slf_check_nop;
		2576	}
		2577
		2578	/* "undo"/cancel a running slf call - used when cancelling a coro, mainly */
		2579	static void
		2580	slf_destroy (pTHX_ struct coro *coro)
		2581	{
		2582	/* this callback is reserved for slf functions needing to do cleanup */
		2583	if (coro->slf_frame.destroy && coro->slf_frame.prepare && !PL_dirty)
		2584	coro->slf_frame.destroy (aTHX_ &coro->slf_frame);
		2585
		2586	/*
		2587	* The on_destroy above most likely is from an SLF call.
		2588	* Since by definition the SLF call will not finish when we destroy
		2589	* the coro, we will have to force-finish it here, otherwise
		2590	* cleanup functions cannot call SLF functions.
		2591	*/
		2592	coro->slf_frame.prepare = 0;
		2593	}
		2594
		2595	/*
		2596	* these not obviously related functions are all rolled into one
		2597	* function to increase chances that they all will call transfer with the same
		2598	* stack offset
		2599	* SLF stands for "schedule-like-function".
		2600	*/
		2601	static OP *
		2602	pp_slf (pTHX)
		2603	{
		2604	I32 checkmark; /* mark SP to see how many elements check has pushed */
		2605
		2606	/* set up the slf frame, unless it has already been set-up */
		2607	/* the latter happens when a new coro has been started */
		2608	/* or when a new cctx was attached to an existing coroutine */
		2609	if (ecb_expect_true (!slf_frame.prepare))
		2610	{
		2611	/* first iteration */
		2612	dSP;
		2613	SV **arg = PL_stack_base + TOPMARK + 1;
		2614	int items = SP - arg; /* args without function object */
		2615	SV *gv = *sp;
		2616
		2617	/* do a quick consistency check on the "function" object, and if it isn't */
		2618	/* for us, divert to the real entersub */
		2619	if (SvTYPE (gv) != SVt_PVGV
		2620	|| !GvCV (gv)
		2621	|| !(CvFLAGS (GvCV (gv)) & CVf_SLF))
		2622	return PL_ppaddr[OP_ENTERSUB](aTHX);
		2623
		2624	if (!(PL_op->op_flags & OPf_STACKED))
		2625	{
		2626	/* ampersand-form of call, use @_ instead of stack */
		2627	AV *av = GvAV (PL_defgv);
		2628	arg = AvARRAY (av);
		2629	items = AvFILLp (av) + 1;
		2630	}
		2631
		2632	/* now call the init function, which needs to set up slf_frame */
		2633	((coro_slf_cb)CvXSUBANY (GvCV (gv)).any_ptr)
		2634	(aTHX_ &slf_frame, GvCV (gv), arg, items);
		2635
		2636	/* pop args */
		2637	SP = PL_stack_base + POPMARK;
		2638
		2639	PUTBACK;
		2640	}
		2641
		2642	/* now that we have a slf_frame, interpret it! */
		2643	/* we use a callback system not to make the code needlessly */
		2644	/* complicated, but so we can run multiple perl coros from one cctx */
		2645
		2646	do
		2647	{
		2648	struct coro_transfer_args ta;
		2649
		2650	slf_frame.prepare (aTHX_ &ta);
		2651	TRANSFER (ta, 0);
		2652
		2653	checkmark = PL_stack_sp - PL_stack_base;
		2654	}
		2655	while (slf_frame.check (aTHX_ &slf_frame));
		2656
		2657	slf_frame.prepare = 0; /* invalidate the frame, we are done processing it */
		2658
		2659	/* exception handling */
		2660	if (ecb_expect_false (CORO_THROW))
		2661	{
		2662	SV *exception = sv_2mortal (CORO_THROW);
		2663
		2664	CORO_THROW = 0;
		2665	sv_setsv (ERRSV, exception);
		2666	croak (0);
		2667	}
		2668
		2669	/* return value handling - mostly like entersub */
		2670	/* make sure we put something on the stack in scalar context */
		2671	if (GIMME_V == G_SCALAR
		2672	&& ecb_expect_false (PL_stack_sp != PL_stack_base + checkmark + 1))
		2673	{
		2674	dSP;
		2675	SV **bot = PL_stack_base + checkmark;
		2676
		2677	if (sp == bot) /* too few, push undef */
		2678	bot [1] = &PL_sv_undef;
		2679	else /* too many, take last one */
		2680	bot [1] = *sp;
		2681
		2682	SP = bot + 1;
		2683
		2684	PUTBACK;
		2685	}
		2686
		2687	return NORMAL;
		2688	}
		2689
		2690	static void
		2691	api_execute_slf (pTHX_ CV *cv, coro_slf_cb init_cb, I32 ax)
		2692	{
		2693	int i;
		2694	SV **arg = PL_stack_base + ax;
		2695	int items = PL_stack_sp - arg + 1;
		2696
		2697	assert (("FATAL: SLF call with illegal CV value", !CvANON (cv)));
		2698
		2699	if (PL_op->op_ppaddr != PL_ppaddr [OP_ENTERSUB]
		2700	&& PL_op->op_ppaddr != pp_slf)
		2701	croak ("FATAL: Coro SLF calls can only be made normally, not via goto or any other means, caught");
		2702
		2703	CvFLAGS (cv) |= CVf_SLF;
		2704	CvXSUBANY (cv).any_ptr = (void *)init_cb;
		2705	slf_cv = cv;
		2706
		2707	/* we patch the op, and then re-run the whole call */
		2708	/* we have to put the same argument on the stack for this to work */
		2709	/* and this will be done by pp_restore */
		2710	slf_restore.op_next = (OP *)&slf_restore;
		2711	slf_restore.op_type = OP_CUSTOM;
		2712	slf_restore.op_ppaddr = pp_restore;
		2713	slf_restore.op_first = PL_op;
		2714
		2715	slf_ax = ax - 1; /* undo the ax++ inside dAXMARK */
		2716
		2717	if (PL_op->op_flags & OPf_STACKED)
		2718	{
		2719	if (items > slf_arga)
		2720	{
		2721	slf_arga = items;
		2722	Safefree (slf_argv);
		2723	New (0, slf_argv, slf_arga, SV *);
		2724	}
		2725
		2726	slf_argc = items;
		2727
		2728	for (i = 0; i < items; ++i)
		2729	slf_argv [i] = SvREFCNT_inc (arg [i]);
		2730	}
		2731	else
		2732	slf_argc = 0;
		2733
		2734	PL_op->op_ppaddr = pp_slf;
		2735	/*PL_op->op_type = OP_CUSTOM; /* we do behave like entersub still */
		2736
		2737	PL_op = (OP *)&slf_restore;
		2738	}
		2739
		2740	/*****************************************************************************/
		2741	/* dynamic wind */
		2742
		2743	static void
		2744	on_enterleave_call (pTHX_ SV *cb)
		2745	{
		2746	dSP;
		2747
		2748	PUSHSTACK;
		2749
		2750	PUSHMARK (SP);
		2751	PUTBACK;
		2752	call_sv (cb, G_VOID | G_DISCARD);
		2753	SPAGAIN;
		2754
		2755	POPSTACK;
		2756	}
		2757
		2758	static SV *
		2759	coro_avp_pop_and_free (pTHX_ AV **avp)
		2760	{
		2761	AV *av = *avp;
		2762	SV *res = av_pop (av);
		2763
		2764	if (AvFILLp (av) < 0)
		2765	{
		2766	*avp = 0;
		2767	SvREFCNT_dec (av);
		2768	}
		2769
		2770	return res;
		2771	}
		2772
		2773	static void
		2774	coro_pop_on_enter (pTHX_ void *coro)
		2775	{
		2776	SV *cb = coro_avp_pop_and_free (aTHX_ &((struct coro *)coro)->on_enter);
		2777	SvREFCNT_dec (cb);
		2778	}
		2779
		2780	static void
		2781	coro_pop_on_leave (pTHX_ void *coro)
		2782	{
		2783	SV *cb = coro_avp_pop_and_free (aTHX_ &((struct coro *)coro)->on_leave);
		2784	on_enterleave_call (aTHX_ sv_2mortal (cb));
		2785	}
1652		2786
1653	/*****************************************************************************/	2787	/*****************************************************************************/
1654	/* PerlIO::cede */	2788	/* PerlIO::cede */
1655		2789
1656	typedef struct	2790	typedef struct
1657	{	2791	{
1658	PerlIOBuf base;	2792	PerlIOBuf base;
1659	NV next, every;	2793	NV next, every;
1660	} PerlIOCede;	2794	} PerlIOCede;
1661		2795
1662	static IV	2796	static IV ecb_cold
1663	PerlIOCede_pushed (pTHX_ PerlIO *f, const char *mode, SV *arg, PerlIO_funcs *tab)	2797	PerlIOCede_pushed (pTHX_ PerlIO *f, const char *mode, SV *arg, PerlIO_funcs *tab)
1664	{	2798	{
1665	PerlIOCede *self = PerlIOSelf (f, PerlIOCede);	2799	PerlIOCede *self = PerlIOSelf (f, PerlIOCede);
1666		2800
1667	self->every = SvCUR (arg) ? SvNV (arg) : 0.01;	2801	self->every = SvCUR (arg) ? SvNV (arg) : 0.01;
1668	self->next = nvtime () + self->every;	2802	self->next = nvtime () + self->every;
1669		2803
1670	return PerlIOBuf_pushed (aTHX_ f, mode, Nullsv, tab);	2804	return PerlIOBuf_pushed (aTHX_ f, mode, Nullsv, tab);
1671	}	2805	}
1672		2806
1673	static SV *	2807	static SV * ecb_cold
1674	PerlIOCede_getarg (pTHX_ PerlIO *f, CLONE_PARAMS *param, int flags)	2808	PerlIOCede_getarg (pTHX_ PerlIO *f, CLONE_PARAMS *param, int flags)
1675	{	2809	{
1676	PerlIOCede *self = PerlIOSelf (f, PerlIOCede);	2810	PerlIOCede *self = PerlIOSelf (f, PerlIOCede);
1677		2811
1678	return newSVnv (self->every);	2812	return newSVnv (self->every);
…		…
1684	PerlIOCede *self = PerlIOSelf (f, PerlIOCede);	2818	PerlIOCede *self = PerlIOSelf (f, PerlIOCede);
1685	double now = nvtime ();	2819	double now = nvtime ();
1686		2820
1687	if (now >= self->next)	2821	if (now >= self->next)
1688	{	2822	{
1689	api_cede ();	2823	api_cede (aTHX);
1690	self->next = now + self->every;	2824	self->next = now + self->every;
1691	}	2825	}
1692		2826
1693	return PerlIOBuf_flush (aTHX_ f);	2827	return PerlIOBuf_flush (aTHX_ f);
1694	}	2828	}
…		…
1723	PerlIOBuf_get_ptr,	2857	PerlIOBuf_get_ptr,
1724	PerlIOBuf_get_cnt,	2858	PerlIOBuf_get_cnt,
1725	PerlIOBuf_set_ptrcnt,	2859	PerlIOBuf_set_ptrcnt,
1726	};	2860	};
1727		2861
		2862	/*****************************************************************************/
		2863	/* Coro::Semaphore & Coro::Signal */
		2864
		2865	static SV *
		2866	coro_waitarray_new (pTHX_ int count)
		2867	{
		2868	/* a waitarray=semaphore contains a counter IV in $sem->[0] and any waiters after that */
		2869	AV *av = newAV ();
		2870	SV **ary;
		2871
		2872	/* unfortunately, building manually saves memory */
		2873	Newx (ary, 2, SV *);
		2874	AvALLOC (av) = ary;
		2875	#if PERL_VERSION_ATLEAST (5,10,0)
		2876	AvARRAY (av) = ary;
		2877	#else
		2878	/* 5.8.8 needs this syntax instead of AvARRAY = ary, yet */
		2879	/* -DDEBUGGING flags this as a bug, despite it perfectly working */
		2880	SvPVX ((SV *)av) = (char *)ary;
		2881	#endif
		2882	AvMAX (av) = 1;
		2883	AvFILLp (av) = 0;
		2884	ary [0] = newSViv (count);
		2885
		2886	return newRV_noinc ((SV *)av);
		2887	}
		2888
		2889	/* semaphore */
		2890
		2891	static void
		2892	coro_semaphore_adjust (pTHX_ AV *av, IV adjust)
		2893	{
		2894	SV *count_sv = AvARRAY (av)[0];
		2895	IV count = SvIVX (count_sv);
		2896
		2897	count += adjust;
		2898	SvIVX (count_sv) = count;
		2899
		2900	/* now wake up as many waiters as are expected to lock */
		2901	while (count > 0 && AvFILLp (av) > 0)
		2902	{
		2903	SV *cb;
		2904
		2905	/* swap first two elements so we can shift a waiter */
		2906	AvARRAY (av)[0] = AvARRAY (av)[1];
		2907	AvARRAY (av)[1] = count_sv;
		2908	cb = av_shift (av);
		2909
		2910	if (SvOBJECT (cb))
		2911	{
		2912	api_ready (aTHX_ cb);
		2913	--count;
		2914	}
		2915	else if (SvTYPE (cb) == SVt_PVCV)
		2916	{
		2917	dSP;
		2918	PUSHMARK (SP);
		2919	XPUSHs (sv_2mortal (newRV_inc ((SV *)av)));
		2920	PUTBACK;
		2921	call_sv (cb, G_VOID | G_DISCARD | G_EVAL | G_KEEPERR);
		2922	}
		2923
		2924	SvREFCNT_dec (cb);
		2925	}
		2926	}
		2927
		2928	static void
		2929	coro_semaphore_destroy (pTHX_ struct CoroSLF *frame)
		2930	{
		2931	/* call $sem->adjust (0) to possibly wake up some other waiters */
		2932	coro_semaphore_adjust (aTHX_ (AV *)frame->data, 0);
		2933	}
		2934
		2935	static int
		2936	slf_check_semaphore_down_or_wait (pTHX_ struct CoroSLF *frame, int acquire)
		2937	{
		2938	AV *av = (AV *)frame->data;
		2939	SV *count_sv = AvARRAY (av)[0];
		2940	SV *coro_hv = SvRV (coro_current);
		2941
		2942	/* if we are about to throw, don't actually acquire the lock, just throw */
		2943	if (CORO_THROW)
		2944	return 0;
		2945	else if (SvIVX (count_sv) > 0)
		2946	{
		2947	frame->destroy = 0;
		2948
		2949	if (acquire)
		2950	SvIVX (count_sv) = SvIVX (count_sv) - 1;
		2951	else
		2952	coro_semaphore_adjust (aTHX_ av, 0);
		2953
		2954	return 0;
		2955	}
		2956	else
		2957	{
		2958	int i;
		2959	/* if we were woken up but can't down, we look through the whole */
		2960	/* waiters list and only add us if we aren't in there already */
		2961	/* this avoids some degenerate memory usage cases */
		2962	for (i = AvFILLp (av); i > 0; --i) /* i > 0 is not an off-by-one bug */
		2963	if (AvARRAY (av)[i] == coro_hv)
		2964	return 1;
		2965
		2966	av_push (av, SvREFCNT_inc (coro_hv));
		2967	return 1;
		2968	}
		2969	}
		2970
		2971	static int
		2972	slf_check_semaphore_down (pTHX_ struct CoroSLF *frame)
		2973	{
		2974	return slf_check_semaphore_down_or_wait (aTHX_ frame, 1);
		2975	}
		2976
		2977	static int
		2978	slf_check_semaphore_wait (pTHX_ struct CoroSLF *frame)
		2979	{
		2980	return slf_check_semaphore_down_or_wait (aTHX_ frame, 0);
		2981	}
		2982
		2983	static void
		2984	slf_init_semaphore_down_or_wait (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		2985	{
		2986	AV *av = (AV *)SvRV (arg [0]);
		2987
		2988	if (SvIVX (AvARRAY (av)[0]) > 0)
		2989	{
		2990	frame->data = (void *)av;
		2991	frame->prepare = prepare_nop;
		2992	}
		2993	else
		2994	{
		2995	av_push (av, SvREFCNT_inc (SvRV (coro_current)));
		2996
		2997	frame->data = (void *)sv_2mortal (SvREFCNT_inc ((SV *)av));
		2998	frame->prepare = prepare_schedule;
		2999	/* to avoid race conditions when a woken-up coro gets terminated */
		3000	/* we arrange for a temporary on_destroy that calls adjust (0) */
		3001	frame->destroy = coro_semaphore_destroy;
		3002	}
		3003	}
		3004
		3005	static void
		3006	slf_init_semaphore_down (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		3007	{
		3008	slf_init_semaphore_down_or_wait (aTHX_ frame, cv, arg, items);
		3009	frame->check = slf_check_semaphore_down;
		3010	}
		3011
		3012	static void
		3013	slf_init_semaphore_wait (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		3014	{
		3015	if (items >= 2)
		3016	{
		3017	/* callback form */
		3018	AV *av = (AV *)SvRV (arg [0]);
		3019	SV *cb_cv = s_get_cv_croak (arg [1]);
		3020
		3021	av_push (av, SvREFCNT_inc_NN (cb_cv));
		3022
		3023	if (SvIVX (AvARRAY (av)[0]) > 0)
		3024	coro_semaphore_adjust (aTHX_ av, 0);
		3025
		3026	frame->prepare = prepare_nop;
		3027	frame->check = slf_check_nop;
		3028	}
		3029	else
		3030	{
		3031	slf_init_semaphore_down_or_wait (aTHX_ frame, cv, arg, items);
		3032	frame->check = slf_check_semaphore_wait;
		3033	}
		3034	}
		3035
		3036	/* signal */
		3037
		3038	static void
		3039	coro_signal_wake (pTHX_ AV *av, int count)
		3040	{
		3041	SvIVX (AvARRAY (av)[0]) = 0;
		3042
		3043	/* now signal count waiters */
		3044	while (count > 0 && AvFILLp (av) > 0)
		3045	{
		3046	SV *cb;
		3047
		3048	/* swap first two elements so we can shift a waiter */
		3049	cb = AvARRAY (av)[0];
		3050	AvARRAY (av)[0] = AvARRAY (av)[1];
		3051	AvARRAY (av)[1] = cb;
		3052
		3053	cb = av_shift (av);
		3054
		3055	if (SvTYPE (cb) == SVt_PVCV)
		3056	{
		3057	dSP;
		3058	PUSHMARK (SP);
		3059	XPUSHs (sv_2mortal (newRV_inc ((SV *)av)));
		3060	PUTBACK;
		3061	call_sv (cb, G_VOID | G_DISCARD | G_EVAL | G_KEEPERR);
		3062	}
		3063	else
		3064	{
		3065	api_ready (aTHX_ cb);
		3066	sv_setiv (cb, 0); /* signal waiter */
		3067	}
		3068
		3069	SvREFCNT_dec (cb);
		3070
		3071	--count;
		3072	}
		3073	}
		3074
		3075	static int
		3076	slf_check_signal_wait (pTHX_ struct CoroSLF *frame)
		3077	{
		3078	/* if we are about to throw, also stop waiting */
		3079	return SvROK ((SV *)frame->data) && !CORO_THROW;
		3080	}
		3081
		3082	static void
		3083	slf_init_signal_wait (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		3084	{
		3085	AV *av = (AV *)SvRV (arg [0]);
		3086
		3087	if (items >= 2)
		3088	{
		3089	SV *cb_cv = s_get_cv_croak (arg [1]);
		3090	av_push (av, SvREFCNT_inc_NN (cb_cv));
		3091
		3092	if (SvIVX (AvARRAY (av)[0]))
		3093	coro_signal_wake (aTHX_ av, 1); /* must be the only waiter */
		3094
		3095	frame->prepare = prepare_nop;
		3096	frame->check = slf_check_nop;
		3097	}
		3098	else if (SvIVX (AvARRAY (av)[0]))
		3099	{
		3100	SvIVX (AvARRAY (av)[0]) = 0;
		3101	frame->prepare = prepare_nop;
		3102	frame->check = slf_check_nop;
		3103	}
		3104	else
		3105	{
		3106	SV *waiter = newSVsv (coro_current); /* owned by signal av */
		3107
		3108	av_push (av, waiter);
		3109
		3110	frame->data = (void *)sv_2mortal (SvREFCNT_inc_NN (waiter)); /* owned by process */
		3111	frame->prepare = prepare_schedule;
		3112	frame->check = slf_check_signal_wait;
		3113	}
		3114	}
		3115
		3116	/*****************************************************************************/
		3117	/* Coro::AIO */
		3118
		3119	#define CORO_MAGIC_type_aio PERL_MAGIC_ext
		3120
		3121	/* helper storage struct */
		3122	struct io_state
		3123	{
		3124	int errorno;
		3125	I32 laststype; /* U16 in 5.10.0 */
		3126	int laststatval;
		3127	Stat_t statcache;
		3128	};
		3129
		3130	static void
		3131	coro_aio_callback (pTHX_ CV *cv)
		3132	{
		3133	dXSARGS;
		3134	AV *state = (AV *)S_GENSUB_ARG;
		3135	SV *coro = av_pop (state);
		3136	SV *data_sv = newSV (sizeof (struct io_state));
		3137
		3138	av_extend (state, items - 1);
		3139
		3140	sv_upgrade (data_sv, SVt_PV);
		3141	SvCUR_set (data_sv, sizeof (struct io_state));
		3142	SvPOK_only (data_sv);
		3143
		3144	{
		3145	struct io_state *data = (struct io_state *)SvPVX (data_sv);
		3146
		3147	data->errorno = errno;
		3148	data->laststype = PL_laststype;
		3149	data->laststatval = PL_laststatval;
		3150	data->statcache = PL_statcache;
		3151	}
		3152
		3153	/* now build the result vector out of all the parameters and the data_sv */
		3154	{
		3155	int i;
		3156
		3157	for (i = 0; i < items; ++i)
		3158	av_push (state, SvREFCNT_inc_NN (ST (i)));
		3159	}
		3160
		3161	av_push (state, data_sv);
		3162
		3163	api_ready (aTHX_ coro);
		3164	SvREFCNT_dec (coro);
		3165	SvREFCNT_dec ((AV *)state);
		3166	}
		3167
		3168	static int
		3169	slf_check_aio_req (pTHX_ struct CoroSLF *frame)
		3170	{
		3171	AV *state = (AV *)frame->data;
		3172
		3173	/* if we are about to throw, return early */
		3174	/* this does not cancel the aio request, but at least */
		3175	/* it quickly returns */
		3176	if (CORO_THROW)
		3177	return 0;
		3178
		3179	/* one element that is an RV? repeat! */
		3180	if (AvFILLp (state) == 0 && SvTYPE (AvARRAY (state)[0]) != SVt_PV)
		3181	return 1;
		3182
		3183	/* restore status */
		3184	{
		3185	SV *data_sv = av_pop (state);
		3186	struct io_state *data = (struct io_state *)SvPVX (data_sv);
		3187
		3188	errno = data->errorno;
		3189	PL_laststype = data->laststype;
		3190	PL_laststatval = data->laststatval;
		3191	PL_statcache = data->statcache;
		3192
		3193	SvREFCNT_dec (data_sv);
		3194	}
		3195
		3196	/* push result values */
		3197	{
		3198	dSP;
		3199	int i;
		3200
		3201	EXTEND (SP, AvFILLp (state) + 1);
		3202	for (i = 0; i <= AvFILLp (state); ++i)
		3203	PUSHs (sv_2mortal (SvREFCNT_inc_NN (AvARRAY (state)[i])));
		3204
		3205	PUTBACK;
		3206	}
		3207
		3208	return 0;
		3209	}
		3210
		3211	static void
		3212	slf_init_aio_req (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
		3213	{
		3214	AV *state = (AV *)sv_2mortal ((SV *)newAV ());
		3215	SV *coro_hv = SvRV (coro_current);
		3216	struct coro *coro = SvSTATE_hv (coro_hv);
		3217
		3218	/* put our coroutine id on the state arg */
		3219	av_push (state, SvREFCNT_inc_NN (coro_hv));
		3220
		3221	/* first see whether we have a non-zero priority and set it as AIO prio */
		3222	if (coro->prio)
		3223	{
		3224	dSP;
		3225
		3226	static SV *prio_cv;
		3227	static SV *prio_sv;
		3228
		3229	if (ecb_expect_false (!prio_cv))
		3230	{
		3231	prio_cv = (SV *)get_cv ("IO::AIO::aioreq_pri", 0);
		3232	prio_sv = newSViv (0);
		3233	}
		3234
		3235	PUSHMARK (SP);
		3236	sv_setiv (prio_sv, coro->prio);
		3237	XPUSHs (prio_sv);
		3238
		3239	PUTBACK;
		3240	call_sv (prio_cv, G_VOID | G_DISCARD);
		3241	}
		3242
		3243	/* now call the original request */
		3244	{
		3245	dSP;
		3246	CV *req = (CV *)CORO_MAGIC_NN ((SV *)cv, CORO_MAGIC_type_aio)->mg_obj;
		3247	int i;
		3248
		3249	PUSHMARK (SP);
		3250
		3251	/* first push all args to the stack */
		3252	EXTEND (SP, items + 1);
		3253
		3254	for (i = 0; i < items; ++i)
		3255	PUSHs (arg [i]);
		3256
		3257	/* now push the callback closure */
		3258	PUSHs (sv_2mortal (s_gensub (aTHX_ coro_aio_callback, (void *)SvREFCNT_inc_NN ((SV *)state))));
		3259
		3260	/* now call the AIO function - we assume our request is uncancelable */
		3261	PUTBACK;
		3262	call_sv ((SV *)req, G_VOID | G_DISCARD);
		3263	}
		3264
		3265	/* now that the request is going, we loop till we have a result */
		3266	frame->data = (void *)state;
		3267	frame->prepare = prepare_schedule;
		3268	frame->check = slf_check_aio_req;
		3269	}
		3270
		3271	static void
		3272	coro_aio_req_xs (pTHX_ CV *cv)
		3273	{
		3274	dXSARGS;
		3275
		3276	CORO_EXECUTE_SLF_XS (slf_init_aio_req);
		3277
		3278	XSRETURN_EMPTY;
		3279	}
		3280
		3281	/*****************************************************************************/
		3282
		3283	#if CORO_CLONE
		3284	# include "clone.c"
		3285	#endif
		3286
		3287	/*****************************************************************************/
		3288
		3289	static SV *
		3290	coro_new (pTHX_ HV *stash, SV **argv, int argc, int is_coro)
		3291	{
		3292	SV *coro_sv;
		3293	struct coro *coro;
		3294	MAGIC *mg;
		3295	HV *hv;
		3296	SV *cb;
		3297	int i;
		3298
		3299	if (argc > 0)
		3300	{
		3301	cb = s_get_cv_croak (argv [0]);
		3302
		3303	if (!is_coro)
		3304	{
		3305	if (CvISXSUB (cb))
		3306	croak ("Coro::State doesn't support XS functions as coroutine start, caught");
		3307
		3308	if (!CvROOT (cb))
		3309	croak ("Coro::State doesn't support autoloaded or undefined functions as coroutine start, caught");
		3310	}
		3311	}
		3312
		3313	Newz (0, coro, 1, struct coro);
		3314	coro->args = newAV ();
		3315	coro->flags = CF_NEW;
		3316
		3317	if (coro_first) coro_first->prev = coro;
		3318	coro->next = coro_first;
		3319	coro_first = coro;
		3320
		3321	coro->hv = hv = newHV ();
		3322	mg = sv_magicext ((SV *)hv, 0, CORO_MAGIC_type_state, &coro_state_vtbl, (char *)coro, 0);
		3323	mg->mg_flags |= MGf_DUP;
		3324	coro_sv = sv_bless (newRV_noinc ((SV *)hv), stash);
		3325
		3326	if (argc > 0)
		3327	{
		3328	av_extend (coro->args, argc + is_coro - 1);
		3329
		3330	if (is_coro)
		3331	{
		3332	av_push (coro->args, SvREFCNT_inc_NN ((SV *)cb));
		3333	cb = (SV *)cv_coro_run;
		3334	}
		3335
		3336	coro->startcv = (CV *)SvREFCNT_inc_NN ((SV *)cb);
		3337
		3338	for (i = 1; i < argc; i++)
		3339	av_push (coro->args, newSVsv (argv [i]));
		3340	}
		3341
		3342	return coro_sv;
		3343	}
		3344
		3345	#ifndef __cplusplus
		3346	ecb_cold XS(boot_Coro__State);
		3347	#endif
		3348
		3349	#if CORO_JIT
		3350
		3351	static void ecb_noinline ecb_cold
		3352	pushav_4uv (pTHX_ UV a, UV b, UV c, UV d)
		3353	{
		3354	dSP;
		3355	AV *av = newAV ();
		3356
		3357	av_store (av, 3, newSVuv (d));
		3358	av_store (av, 2, newSVuv (c));
		3359	av_store (av, 1, newSVuv (b));
		3360	av_store (av, 0, newSVuv (a));
		3361
		3362	XPUSHs (sv_2mortal (newRV_noinc ((SV *)av)));
		3363
		3364	PUTBACK;
		3365	}
		3366
		3367	static void ecb_noinline ecb_cold
		3368	jit_init (pTHX)
		3369	{
		3370	dSP;
		3371	SV *load, *save;
		3372	char *map_base;
		3373	char *load_ptr, *save_ptr;
		3374	STRLEN load_len, save_len, map_len;
		3375	int count;
		3376
		3377	eval_pv ("require 'Coro/jit-" CORO_JIT_TYPE ".pl'", 1);
		3378
		3379	PUSHMARK (SP);
		3380	#define VARx(name,expr,type) pushav_4uv (aTHX_ (UV)&(expr), sizeof (expr), offsetof (perl_slots, name), sizeof (type));
		3381	#include "state.h"
		3382	count = call_pv ("Coro::State::_jit", G_ARRAY);
		3383	SPAGAIN;
		3384
		3385	save = POPs; save_ptr = SvPVbyte (save, save_len);
		3386	load = POPs; load_ptr = SvPVbyte (load, load_len);
		3387
		3388	map_len = load_len + save_len + 16;
		3389
		3390	map_base = mmap (0, map_len, PROT_READ | PROT_WRITE | PROT_EXEC, MAP_PRIVATE | MAP_ANONYMOUS, -1, 0);
		3391
		3392	assert (("Coro: unable to mmap jit code page, cannot continue.", map_base != (char *)MAP_FAILED));
		3393
		3394	load_perl_slots = (load_save_perl_slots_type)map_base;
		3395	memcpy (map_base, load_ptr, load_len);
		3396
		3397	map_base += (load_len + 15) & ~15;
		3398
		3399	save_perl_slots = (load_save_perl_slots_type)map_base;
		3400	memcpy (map_base, save_ptr, save_len);
		3401
		3402	/* we are good citizens and try to make the page read-only, so the evil evil */
		3403	/* hackers might have it a bit more difficult */
		3404	mprotect (map_base, map_len, PROT_READ | PROT_EXEC);
		3405
		3406	PUTBACK;
		3407	eval_pv ("undef &Coro::State::_jit", 1);
		3408	}
		3409
		3410	#endif
1728		3411
1729	MODULE = Coro::State PACKAGE = Coro::State PREFIX = api_	3412	MODULE = Coro::State PACKAGE = Coro::State PREFIX = api_
1730		3413
1731	PROTOTYPES: DISABLE	3414	PROTOTYPES: DISABLE
1732		3415
1733	BOOT:	3416	BOOT:
1734	{	3417	{
1735	#ifdef USE_ITHREADS	3418	#ifdef USE_ITHREADS
1736	MUTEX_INIT (&coro_mutex);	3419	# if CORO_PTHREAD
		3420	coro_thx = PERL_GET_CONTEXT;
1737	#endif	3421	# endif
1738	BOOT_PAGESIZE;	3422	#endif
		3423	/* perl defines these to check for existance first, but why it doesn't */
		3424	/* just create them one at init time is not clear to me, except for */
		3425	/* programs trying to delete them, but... */
		3426	/* anyway, we declare this as invalid and make sure they are initialised here */
		3427	DEFSV;
		3428	ERRSV;
		3429
		3430	cctx_current = cctx_new_empty ();
1739		3431
1740	irsgv = gv_fetchpv ("/" , GV_ADD|GV_NOTQUAL, SVt_PV);	3432	irsgv = gv_fetchpv ("/" , GV_ADD|GV_NOTQUAL, SVt_PV);
1741	stdoutgv = gv_fetchpv ("STDOUT", GV_ADD|GV_NOTQUAL, SVt_PVIO);	3433	stdoutgv = gv_fetchpv ("STDOUT", GV_ADD|GV_NOTQUAL, SVt_PVIO);
1742		3434
1743	orig_sigelem_get = PL_vtbl_sigelem.svt_get; PL_vtbl_sigelem.svt_get = coro_sigelem_get;	3435	orig_sigelem_get = PL_vtbl_sigelem.svt_get; PL_vtbl_sigelem.svt_get = coro_sigelem_get;
…		…
1759	main_top_env = PL_top_env;	3451	main_top_env = PL_top_env;
1760		3452
1761	while (main_top_env->je_prev)	3453	while (main_top_env->je_prev)
1762	main_top_env = main_top_env->je_prev;	3454	main_top_env = main_top_env->je_prev;
1763		3455
1764	coroapi.ver = CORO_API_VERSION;
1765	coroapi.rev = CORO_API_REVISION;
1766	coroapi.transfer = api_transfer;
1767
1768	{	3456	{
1769	SV **svp = hv_fetch (PL_modglobal, "Time::NVtime", 12, 0);	3457	SV *slf = sv_2mortal (newSViv (PTR2IV (pp_slf)));
1770		3458
1771	if (!svp) croak ("Time::HiRes is required");	3459	if (!PL_custom_op_names) PL_custom_op_names = newHV ();
1772	if (!SvIOK (*svp)) croak ("Time::NVtime isn't a function pointer");	3460	hv_store_ent (PL_custom_op_names, slf, newSVpv ("coro_slf", 0), 0);
1773		3461
1774	nvtime = INT2PTR (double (*)(), SvIV (*svp));	3462	if (!PL_custom_op_descs) PL_custom_op_descs = newHV ();
		3463	hv_store_ent (PL_custom_op_descs, slf, newSVpv ("coro schedule like function", 0), 0);
1775	}	3464	}
1776		3465
		3466	coroapi.ver = CORO_API_VERSION;
		3467	coroapi.rev = CORO_API_REVISION;
		3468
		3469	coroapi.transfer = api_transfer;
		3470
		3471	coroapi.sv_state = SvSTATE_;
		3472	coroapi.execute_slf = api_execute_slf;
		3473	coroapi.prepare_nop = prepare_nop;
		3474	coroapi.prepare_schedule = prepare_schedule;
		3475	coroapi.prepare_cede = prepare_cede;
		3476	coroapi.prepare_cede_notself = prepare_cede_notself;
		3477
		3478	time_init (aTHX);
		3479
1777	assert (("PRIO_NORMAL must be 0", !PRIO_NORMAL));	3480	assert (("PRIO_NORMAL must be 0", !CORO_PRIO_NORMAL));
		3481	#if CORO_JIT
		3482	PUTBACK;
		3483	jit_init (aTHX);
		3484	SPAGAIN;
		3485	#endif
1778	}	3486	}
1779		3487
1780	SV *	3488	SV *
1781	new (char *klass, ...)	3489	new (SV *klass, ...)
		3490	ALIAS:
		3491	Coro::new = 1
1782	CODE:	3492	CODE:
1783	{	3493	RETVAL = coro_new (aTHX_ ix ? coro_stash : coro_state_stash, &ST (1), items - 1, ix);
1784	struct coro *coro;
1785	MAGIC *mg;
1786	HV *hv;
1787	int i;
1788
1789	Newz (0, coro, 1, struct coro);
1790	coro->args = newAV ();
1791	coro->flags = CF_NEW;
1792
1793	if (coro_first) coro_first->prev = coro;
1794	coro->next = coro_first;
1795	coro_first = coro;
1796
1797	coro->hv = hv = newHV ();
1798	mg = sv_magicext ((SV *)hv, 0, CORO_MAGIC_type_state, &coro_state_vtbl, (char *)coro, 0);
1799	mg->mg_flags |= MGf_DUP;
1800	RETVAL = sv_bless (newRV_noinc ((SV *)hv), gv_stashpv (klass, 1));
1801
1802	av_extend (coro->args, items - 1);
1803	for (i = 1; i < items; i++)
1804	av_push (coro->args, newSVsv (ST (i)));
1805	}
1806	OUTPUT:
1807	RETVAL
1808
1809	# these not obviously related functions are all rolled into the same xs
1810	# function to increase chances that they all will call transfer with the same
1811	# stack offset
1812	void
1813	_set_stacklevel (...)
1814	ALIAS:
1815	Coro::State::transfer = 1
1816	Coro::schedule = 2
1817	Coro::cede = 3
1818	Coro::cede_notself = 4
1819	CODE:
1820	{
1821	struct transfer_args ta;
1822
1823	PUTBACK;
1824	switch (ix)
1825	{
1826	case 0:
1827	ta.prev = (struct coro *)INT2PTR (coro_cctx *, SvIV (ST (0)));
1828	ta.next = 0;
1829	break;
1830
1831	case 1:
1832	if (items != 2)
1833	croak ("Coro::State::transfer (prev,next) expects two arguments, not %d", items);
1834
1835	prepare_transfer (aTHX_ &ta, ST (0), ST (1));
1836	break;
1837
1838	case 2:
1839	prepare_schedule (aTHX_ &ta);
1840	break;
1841
1842	case 3:
1843	prepare_cede (aTHX_ &ta);
1844	break;
1845
1846	case 4:
1847	if (!prepare_cede_notself (aTHX_ &ta))
1848	XSRETURN_EMPTY;
1849
1850	break;
1851	}
1852	SPAGAIN;
1853
1854	BARRIER;
1855	PUTBACK;
1856	TRANSFER (ta, 0);
1857	SPAGAIN; /* might be the sp of a different coroutine now */
1858	/* be extra careful not to ever do anything after TRANSFER */
1859	}
1860
1861	bool
1862	_destroy (SV *coro_sv)
1863	CODE:
1864	RETVAL = coro_state_destroy (aTHX_ SvSTATE (coro_sv));
1865	OUTPUT:	3494	OUTPUT:
1866	RETVAL	3495	RETVAL
1867		3496
1868	void	3497	void
1869	_exit (int code)	3498	transfer (...)
1870	PROTOTYPE: $	3499	PROTOTYPE: $$
1871	CODE:	3500	CODE:
1872	_exit (code);	3501	CORO_EXECUTE_SLF_XS (slf_init_transfer);
		3502
		3503	SV *
		3504	clone (Coro::State coro)
		3505	CODE:
		3506	{
		3507	#if CORO_CLONE
		3508	struct coro *ncoro = coro_clone (aTHX_ coro);
		3509	MAGIC *mg;
		3510	/* TODO: too much duplication */
		3511	ncoro->hv = newHV ();
		3512	mg = sv_magicext ((SV *)ncoro->hv, 0, CORO_MAGIC_type_state, &coro_state_vtbl, (char *)ncoro, 0);
		3513	mg->mg_flags |= MGf_DUP;
		3514	RETVAL = sv_bless (newRV_noinc ((SV *)ncoro->hv), SvSTASH (coro->hv));
		3515	#else
		3516	croak ("Coro::State->clone has not been configured into this installation of Coro, realised");
		3517	#endif
		3518	}
		3519	OUTPUT:
		3520	RETVAL
1873		3521
1874	int	3522	int
1875	cctx_stacksize (int new_stacksize = 0)	3523	cctx_stacksize (int new_stacksize = 0)
		3524	PROTOTYPE: ;$
1876	CODE:	3525	CODE:
1877	RETVAL = coro_stacksize;	3526	RETVAL = cctx_stacksize;
1878	if (new_stacksize)	3527	if (new_stacksize)
		3528	{
1879	coro_stacksize = new_stacksize;	3529	cctx_stacksize = new_stacksize;
		3530	++cctx_gen;
		3531	}
1880	OUTPUT:	3532	OUTPUT:
1881	RETVAL	3533	RETVAL
1882		3534
1883	int	3535	int
		3536	cctx_max_idle (int max_idle = 0)
		3537	PROTOTYPE: ;$
		3538	CODE:
		3539	RETVAL = cctx_max_idle;
		3540	if (max_idle > 1)
		3541	cctx_max_idle = max_idle;
		3542	OUTPUT:
		3543	RETVAL
		3544
		3545	int
1884	cctx_count ()	3546	cctx_count ()
		3547	PROTOTYPE:
1885	CODE:	3548	CODE:
1886	RETVAL = cctx_count;	3549	RETVAL = cctx_count;
1887	OUTPUT:	3550	OUTPUT:
1888	RETVAL	3551	RETVAL
1889		3552
1890	int	3553	int
1891	cctx_idle ()	3554	cctx_idle ()
		3555	PROTOTYPE:
1892	CODE:	3556	CODE:
1893	RETVAL = cctx_idle;	3557	RETVAL = cctx_idle;
1894	OUTPUT:	3558	OUTPUT:
1895	RETVAL	3559	RETVAL
1896		3560
1897	void	3561	void
1898	list ()	3562	list ()
		3563	PROTOTYPE:
1899	PPCODE:	3564	PPCODE:
1900	{	3565	{
1901	struct coro *coro;	3566	struct coro *coro;
1902	for (coro = coro_first; coro; coro = coro->next)	3567	for (coro = coro_first; coro; coro = coro->next)
1903	if (coro->hv)	3568	if (coro->hv)
1904	XPUSHs (sv_2mortal (newRV_inc ((SV *)coro->hv)));	3569	XPUSHs (sv_2mortal (newRV_inc ((SV *)coro->hv)));
1905	}	3570	}
1906		3571
…		…
1910	eval = 1	3575	eval = 1
1911	CODE:	3576	CODE:
1912	{	3577	{
1913	if (coro->mainstack && ((coro->flags & CF_RUNNING) || coro->slot))	3578	if (coro->mainstack && ((coro->flags & CF_RUNNING) || coro->slot))
1914	{	3579	{
1915	struct coro temp;	3580	struct coro *current = SvSTATE_current;
		3581	struct CoroSLF slf_save;
1916		3582
1917	if (!(coro->flags & CF_RUNNING))	3583	if (current != coro)
1918	{	3584	{
1919	PUTBACK;	3585	PUTBACK;
1920	save_perl (aTHX_ &temp);	3586	save_perl (aTHX_ current);
1921	load_perl (aTHX_ coro);	3587	load_perl (aTHX_ coro);
		3588	/* the coro is most likely in an active SLF call.
		3589	* while not strictly required (the code we execute is
		3590	* not allowed to call any SLF functions), it's cleaner
		3591	* to reinitialise the slf_frame and restore it later.
		3592	* This might one day allow us to actually do SLF calls
		3593	* from code executed here.
		3594	*/
		3595	slf_save = slf_frame;
		3596	slf_frame.prepare = 0;
		3597	SPAGAIN;
1922	}	3598	}
1923		3599
1924	{
1925	dSP;
1926	ENTER;
1927	SAVETMPS;
1928	PUTBACK;
1929	PUSHSTACK;	3600	PUSHSTACK;
		3601
1930	PUSHMARK (SP);	3602	PUSHMARK (SP);
		3603	PUTBACK;
1931		3604
1932	if (ix)	3605	if (ix)
1933	eval_sv (coderef, 0);	3606	eval_sv (coderef, 0);
1934	else	3607	else
1935	call_sv (coderef, G_KEEPERR | G_EVAL | G_VOID | G_DISCARD);	3608	call_sv (coderef, G_KEEPERR | G_EVAL | G_VOID | G_DISCARD);
1936		3609
1937	POPSTACK;	3610	POPSTACK;
1938	SPAGAIN;	3611	SPAGAIN;
1939	FREETMPS;
1940	LEAVE;
1941	PUTBACK;
1942	}
1943		3612
1944	if (!(coro->flags & CF_RUNNING))	3613	if (current != coro)
1945	{	3614	{
		3615	PUTBACK;
		3616	slf_frame = slf_save;
1946	save_perl (aTHX_ coro);	3617	save_perl (aTHX_ coro);
1947	load_perl (aTHX_ &temp);	3618	load_perl (aTHX_ current);
1948	SPAGAIN;	3619	SPAGAIN;
1949	}	3620	}
1950	}	3621	}
1951	}	3622	}
1952		3623
…		…
1955	PROTOTYPE: $	3626	PROTOTYPE: $
1956	ALIAS:	3627	ALIAS:
1957	is_ready = CF_READY	3628	is_ready = CF_READY
1958	is_running = CF_RUNNING	3629	is_running = CF_RUNNING
1959	is_new = CF_NEW	3630	is_new = CF_NEW
1960	is_destroyed = CF_DESTROYED	3631	is_destroyed = CF_ZOMBIE
		3632	is_zombie = CF_ZOMBIE
		3633	is_suspended = CF_SUSPENDED
1961	CODE:	3634	CODE:
1962	RETVAL = boolSV (coro->flags & ix);	3635	RETVAL = boolSV (coro->flags & ix);
1963	OUTPUT:	3636	OUTPUT:
1964	RETVAL	3637	RETVAL
1965		3638
1966	void	3639	void
		3640	throw (SV *self, SV *exception = &PL_sv_undef)
		3641	PROTOTYPE: $;$
		3642	CODE:
		3643	{
		3644	struct coro *coro = SvSTATE (self);
		3645	struct coro *current = SvSTATE_current;
		3646	SV **exceptionp = coro == current ? &CORO_THROW : &coro->except;
		3647	SvREFCNT_dec (*exceptionp);
		3648	SvGETMAGIC (exception);
		3649	*exceptionp = SvOK (exception) ? newSVsv (exception) : 0;
		3650
		3651	api_ready (aTHX_ self);
		3652	}
		3653
		3654	void
1967	api_trace (SV *coro, int flags = CC_TRACE | CC_TRACE_SUB)	3655	api_trace (SV *coro, int flags = CC_TRACE | CC_TRACE_SUB)
		3656	PROTOTYPE: $;$
		3657	C_ARGS: aTHX_ coro, flags
1968		3658
1969	SV *	3659	SV *
1970	has_cctx (Coro::State coro)	3660	has_cctx (Coro::State coro)
1971	PROTOTYPE: $	3661	PROTOTYPE: $
1972	CODE:	3662	CODE:
1973	RETVAL = boolSV (!!coro->cctx);	3663	/* maybe manage the running flag differently */
		3664	RETVAL = boolSV (!!coro->cctx || (coro->flags & CF_RUNNING));
1974	OUTPUT:	3665	OUTPUT:
1975	RETVAL	3666	RETVAL
1976		3667
1977	int	3668	int
1978	is_traced (Coro::State coro)	3669	is_traced (Coro::State coro)
…		…
1980	CODE:	3671	CODE:
1981	RETVAL = (coro->cctx ? coro->cctx->flags : 0) & CC_TRACE_ALL;	3672	RETVAL = (coro->cctx ? coro->cctx->flags : 0) & CC_TRACE_ALL;
1982	OUTPUT:	3673	OUTPUT:
1983	RETVAL	3674	RETVAL
1984		3675
1985	IV	3676	UV
1986	rss (Coro::State coro)	3677	rss (Coro::State coro)
1987	PROTOTYPE: $	3678	PROTOTYPE: $
1988	ALIAS:	3679	ALIAS:
1989	usecount = 1	3680	usecount = 1
1990	CODE:	3681	CODE:
…		…
1996	OUTPUT:	3687	OUTPUT:
1997	RETVAL	3688	RETVAL
1998		3689
1999	void	3690	void
2000	force_cctx ()	3691	force_cctx ()
		3692	PROTOTYPE:
2001	CODE:	3693	CODE:
2002	struct coro *coro = SvSTATE (coro_current);
2003	coro->cctx->idle_sp = 0;	3694	cctx_current->idle_sp = 0;
2004
2005	void
2006	throw (Coro::State self, SV *throw = &PL_sv_undef)
2007	PROTOTYPE: $;$
2008	CODE:
2009	SvREFCNT_dec (self->throw);
2010	self->throw = SvOK (throw) ? newSVsv (throw) : 0;
2011		3695
2012	void	3696	void
2013	swap_defsv (Coro::State self)	3697	swap_defsv (Coro::State self)
2014	PROTOTYPE: $	3698	PROTOTYPE: $
2015	ALIAS:	3699	ALIAS:
2016	swap_defav = 1	3700	swap_defav = 1
2017	CODE:	3701	CODE:
2018	if (!self->slot)	3702	if (!self->slot)
2019	croak ("cannot swap state with coroutine that has no saved state");	3703	croak ("cannot swap state with coroutine that has no saved state,");
2020	else	3704	else
2021	{	3705	{
2022	SV **src = ix ? (SV **)&GvAV (PL_defgv) : &GvSV (PL_defgv);	3706	SV **src = ix ? (SV **)&GvAV (PL_defgv) : &GvSV (PL_defgv);
2023	SV **dst = ix ? (SV **)&self->slot->defav : (SV **)&self->slot->defsv;	3707	SV **dst = ix ? (SV **)&self->slot->defav : (SV **)&self->slot->defsv;
2024		3708
2025	SV *tmp = *src; *src = *dst; *dst = tmp;	3709	SV *tmp = *src; *src = *dst; *dst = tmp;
2026	}	3710	}
2027		3711
		3712	void
		3713	cancel (Coro::State self)
		3714	CODE:
		3715	coro_state_destroy (aTHX_ self);
		3716
		3717	SV *
		3718	enable_times (int enabled = enable_times)
		3719	CODE:
		3720	{
		3721	RETVAL = boolSV (enable_times);
		3722
		3723	if (enabled != enable_times)
		3724	{
		3725	enable_times = enabled;
		3726
		3727	coro_times_update ();
		3728	(enabled ? coro_times_sub : coro_times_add)(SvSTATE (coro_current));
		3729	}
		3730	}
		3731	OUTPUT:
		3732	RETVAL
		3733
		3734	void
		3735	times (Coro::State self)
		3736	PPCODE:
		3737	{
		3738	struct coro *current = SvSTATE (coro_current);
		3739
		3740	if (ecb_expect_false (current == self))
		3741	{
		3742	coro_times_update ();
		3743	coro_times_add (SvSTATE (coro_current));
		3744	}
		3745
		3746	EXTEND (SP, 2);
		3747	PUSHs (sv_2mortal (newSVnv (self->t_real [0] + self->t_real [1] * 1e-9)));
		3748	PUSHs (sv_2mortal (newSVnv (self->t_cpu [0] + self->t_cpu [1] * 1e-9)));
		3749
		3750	if (ecb_expect_false (current == self))
		3751	coro_times_sub (SvSTATE (coro_current));
		3752	}
		3753
		3754	void
		3755	swap_sv (Coro::State coro, SV *sv, SV *swapsv)
		3756	CODE:
		3757	{
		3758	struct coro *current = SvSTATE_current;
		3759
		3760	if (current == coro)
		3761	SWAP_SVS (current);
		3762
		3763	if (!coro->swap_sv)
		3764	coro->swap_sv = newAV ();
		3765
		3766	av_push (coro->swap_sv, SvREFCNT_inc_NN (SvRV (sv )));
		3767	av_push (coro->swap_sv, SvREFCNT_inc_NN (SvRV (swapsv)));
		3768
		3769	if (current == coro)
		3770	SWAP_SVS (current);
		3771	}
		3772
		3773
2028	MODULE = Coro::State PACKAGE = Coro	3774	MODULE = Coro::State PACKAGE = Coro
2029		3775
2030	BOOT:	3776	BOOT:
2031	{	3777	{
2032	int i;
2033
2034	av_async_pool = coro_get_av (aTHX_ "Coro::async_pool", TRUE);
2035	sv_pool_rss = coro_get_sv (aTHX_ "Coro::POOL_RSS" , TRUE);	3778	sv_pool_rss = coro_get_sv (aTHX_ "Coro::POOL_RSS" , TRUE);
2036	sv_pool_size = coro_get_sv (aTHX_ "Coro::POOL_SIZE" , TRUE);	3779	sv_pool_size = coro_get_sv (aTHX_ "Coro::POOL_SIZE" , TRUE);
2037		3780	cv_coro_run = get_cv ( "Coro::_coro_run" , GV_ADD);
2038	coro_current = coro_get_sv (aTHX_ "Coro::current", FALSE);	3781	coro_current = coro_get_sv (aTHX_ "Coro::current" , FALSE); SvREADONLY_on (coro_current);
2039	SvREADONLY_on (coro_current);	3782	av_async_pool = coro_get_av (aTHX_ "Coro::async_pool", TRUE);
		3783	av_destroy = coro_get_av (aTHX_ "Coro::destroy" , TRUE);
		3784	sv_manager = coro_get_sv (aTHX_ "Coro::manager" , TRUE);
		3785	sv_idle = coro_get_sv (aTHX_ "Coro::idle" , TRUE);
		3786
		3787	sv_async_pool_idle = newSVpv ("[async pool idle]", 0); SvREADONLY_on (sv_async_pool_idle);
		3788	sv_Coro = newSVpv ("Coro", 0); SvREADONLY_on (sv_Coro);
		3789	cv_pool_handler = get_cv ("Coro::pool_handler", GV_ADD); SvREADONLY_on (cv_pool_handler);
		3790	CvNODEBUG_on (get_cv ("Coro::_pool_handler", 0)); /* work around a debugger bug */
2040		3791
2041	coro_stash = gv_stashpv ("Coro", TRUE);	3792	coro_stash = gv_stashpv ("Coro", TRUE);
2042		3793
2043	newCONSTSUB (coro_stash, "PRIO_MAX", newSViv (PRIO_MAX));	3794	newCONSTSUB (coro_stash, "PRIO_MAX", newSViv (CORO_PRIO_MAX));
2044	newCONSTSUB (coro_stash, "PRIO_HIGH", newSViv (PRIO_HIGH));	3795	newCONSTSUB (coro_stash, "PRIO_HIGH", newSViv (CORO_PRIO_HIGH));
2045	newCONSTSUB (coro_stash, "PRIO_NORMAL", newSViv (PRIO_NORMAL));	3796	newCONSTSUB (coro_stash, "PRIO_NORMAL", newSViv (CORO_PRIO_NORMAL));
2046	newCONSTSUB (coro_stash, "PRIO_LOW", newSViv (PRIO_LOW));	3797	newCONSTSUB (coro_stash, "PRIO_LOW", newSViv (CORO_PRIO_LOW));
2047	newCONSTSUB (coro_stash, "PRIO_IDLE", newSViv (PRIO_IDLE));	3798	newCONSTSUB (coro_stash, "PRIO_IDLE", newSViv (CORO_PRIO_IDLE));
2048	newCONSTSUB (coro_stash, "PRIO_MIN", newSViv (PRIO_MIN));	3799	newCONSTSUB (coro_stash, "PRIO_MIN", newSViv (CORO_PRIO_MIN));
2049
2050	for (i = PRIO_MAX - PRIO_MIN + 1; i--; )
2051	coro_ready[i] = newAV ();
2052		3800
2053	{	3801	{
2054	SV *sv = perl_get_sv ("Coro::API", TRUE);	3802	SV *sv = coro_get_sv (aTHX_ "Coro::API", TRUE);
2055	perl_get_sv ("Coro::API", TRUE); /* silence 5.10 warning */
2056		3803
2057	coroapi.schedule = api_schedule;	3804	coroapi.schedule = api_schedule;
		3805	coroapi.schedule_to = api_schedule_to;
2058	coroapi.cede = api_cede;	3806	coroapi.cede = api_cede;
2059	coroapi.cede_notself = api_cede_notself;	3807	coroapi.cede_notself = api_cede_notself;
2060	coroapi.ready = api_ready;	3808	coroapi.ready = api_ready;
2061	coroapi.is_ready = api_is_ready;	3809	coroapi.is_ready = api_is_ready;
2062	coroapi.nready = &coro_nready;	3810	coroapi.nready = coro_nready;
2063	coroapi.current = coro_current;	3811	coroapi.current = coro_current;
2064		3812
2065	GCoroAPI = &coroapi;	3813	/*GCoroAPI = &coroapi;*/
2066	sv_setiv (sv, (IV)&coroapi);	3814	sv_setiv (sv, (IV)&coroapi);
2067	SvREADONLY_on (sv);	3815	SvREADONLY_on (sv);
2068	}	3816	}
2069	}	3817	}
		3818
		3819	SV *
		3820	async (...)
		3821	PROTOTYPE: &@
		3822	CODE:
		3823	RETVAL = coro_new (aTHX_ coro_stash, &ST (0), items, 1);
		3824	api_ready (aTHX_ RETVAL);
		3825	OUTPUT:
		3826	RETVAL
		3827
		3828	void
		3829	_destroy (Coro::State coro)
		3830	CODE:
		3831	/* used by the manager thread */
		3832	coro_state_destroy (aTHX_ coro);
		3833
		3834	void
		3835	on_destroy (Coro::State coro, SV *cb)
		3836	CODE:
		3837	coro_push_on_destroy (aTHX_ coro, newSVsv (cb));
		3838
		3839	void
		3840	join (...)
		3841	CODE:
		3842	CORO_EXECUTE_SLF_XS (slf_init_join);
		3843
		3844	void
		3845	terminate (...)
		3846	CODE:
		3847	CORO_EXECUTE_SLF_XS (slf_init_terminate);
		3848
		3849	void
		3850	cancel (...)
		3851	CODE:
		3852	CORO_EXECUTE_SLF_XS (slf_init_cancel);
		3853
		3854	int
		3855	safe_cancel (Coro::State self, ...)
		3856	C_ARGS: aTHX_ self, &ST (1), items - 1
		3857
		3858	void
		3859	schedule (...)
		3860	CODE:
		3861	CORO_EXECUTE_SLF_XS (slf_init_schedule);
		3862
		3863	void
		3864	schedule_to (...)
		3865	CODE:
		3866	CORO_EXECUTE_SLF_XS (slf_init_schedule_to);
		3867
		3868	void
		3869	cede_to (...)
		3870	CODE:
		3871	CORO_EXECUTE_SLF_XS (slf_init_cede_to);
		3872
		3873	void
		3874	cede (...)
		3875	CODE:
		3876	CORO_EXECUTE_SLF_XS (slf_init_cede);
		3877
		3878	void
		3879	cede_notself (...)
		3880	CODE:
		3881	CORO_EXECUTE_SLF_XS (slf_init_cede_notself);
2070		3882
2071	void	3883	void
2072	_set_current (SV *current)	3884	_set_current (SV *current)
2073	PROTOTYPE: $	3885	PROTOTYPE: $
2074	CODE:	3886	CODE:
2075	SvREFCNT_dec (SvRV (coro_current));	3887	SvREFCNT_dec (SvRV (coro_current));
2076	SvRV_set (coro_current, SvREFCNT_inc (SvRV (current)));	3888	SvRV_set (coro_current, SvREFCNT_inc_NN (SvRV (current)));
2077		3889
2078	void	3890	void
2079	_set_readyhook (SV *hook)	3891	_set_readyhook (SV *hook)
2080	PROTOTYPE: $	3892	PROTOTYPE: $
2081	CODE:	3893	CODE:
2082	LOCK;
2083	SvREFCNT_dec (coro_readyhook);	3894	SvREFCNT_dec (coro_readyhook);
		3895	SvGETMAGIC (hook);
		3896	if (SvOK (hook))
		3897	{
2084	coro_readyhook = SvOK (hook) ? newSVsv (hook) : 0;	3898	coro_readyhook = newSVsv (hook);
2085	UNLOCK;	3899	CORO_READYHOOK = invoke_sv_ready_hook_helper;
		3900	}
		3901	else
		3902	{
		3903	coro_readyhook = 0;
		3904	CORO_READYHOOK = 0;
		3905	}
2086		3906
2087	int	3907	int
2088	prio (Coro::State coro, int newprio = 0)	3908	prio (Coro::State coro, int newprio = 0)
		3909	PROTOTYPE: $;$
2089	ALIAS:	3910	ALIAS:
2090	nice = 1	3911	nice = 1
2091	CODE:	3912	CODE:
2092	{	3913	{
2093	RETVAL = coro->prio;	3914	RETVAL = coro->prio;
…		…
2095	if (items > 1)	3916	if (items > 1)
2096	{	3917	{
2097	if (ix)	3918	if (ix)
2098	newprio = coro->prio - newprio;	3919	newprio = coro->prio - newprio;
2099		3920
2100	if (newprio < PRIO_MIN) newprio = PRIO_MIN;	3921	if (newprio < CORO_PRIO_MIN) newprio = CORO_PRIO_MIN;
2101	if (newprio > PRIO_MAX) newprio = PRIO_MAX;	3922	if (newprio > CORO_PRIO_MAX) newprio = CORO_PRIO_MAX;
2102		3923
2103	coro->prio = newprio;	3924	coro->prio = newprio;
2104	}	3925	}
2105	}	3926	}
2106	OUTPUT:	3927	OUTPUT:
…		…
2108		3929
2109	SV *	3930	SV *
2110	ready (SV *self)	3931	ready (SV *self)
2111	PROTOTYPE: $	3932	PROTOTYPE: $
2112	CODE:	3933	CODE:
2113	RETVAL = boolSV (api_ready (self));	3934	RETVAL = boolSV (api_ready (aTHX_ self));
2114	OUTPUT:	3935	OUTPUT:
2115	RETVAL	3936	RETVAL
2116		3937
2117	int	3938	int
2118	nready (...)	3939	nready (...)
…		…
2120	CODE:	3941	CODE:
2121	RETVAL = coro_nready;	3942	RETVAL = coro_nready;
2122	OUTPUT:	3943	OUTPUT:
2123	RETVAL	3944	RETVAL
2124		3945
2125	# for async_pool speedup
2126	void	3946	void
2127	_pool_1 (SV *cb)	3947	suspend (Coro::State self)
		3948	PROTOTYPE: $
2128	CODE:	3949	CODE:
2129	{	3950	self->flags |= CF_SUSPENDED;
2130	struct coro *coro = SvSTATE (coro_current);
2131	HV *hv = (HV *)SvRV (coro_current);
2132	AV *defav = GvAV (PL_defgv);
2133	SV *invoke = hv_delete (hv, "_invoke", sizeof ("_invoke") - 1, 0);
2134	AV *invoke_av;
2135	int i, len;
2136		3951
2137	if (!invoke)	3952	void
		3953	resume (Coro::State self)
		3954	PROTOTYPE: $
		3955	CODE:
		3956	self->flags &= ~CF_SUSPENDED;
		3957
		3958	void
		3959	_pool_handler (...)
		3960	CODE:
		3961	CORO_EXECUTE_SLF_XS (slf_init_pool_handler);
		3962
		3963	void
		3964	async_pool (SV *cv, ...)
		3965	PROTOTYPE: &@
		3966	PPCODE:
		3967	{
		3968	HV *hv = (HV *)av_pop (av_async_pool);
		3969	AV *av = newAV ();
		3970	SV *cb = ST (0);
		3971	int i;
		3972
		3973	av_extend (av, items - 2);
		3974	for (i = 1; i < items; ++i)
		3975	av_push (av, SvREFCNT_inc_NN (ST (i)));
		3976
		3977	if ((SV *)hv == &PL_sv_undef)
2138	{	3978	{
2139	SV *old = PL_diehook;	3979	SV *sv = coro_new (aTHX_ coro_stash, (SV **)&cv_pool_handler, 1, 1);
2140	PL_diehook = 0;	3980	hv = (HV *)SvREFCNT_inc_NN (SvRV (sv));
2141	SvREFCNT_dec (old);	3981	SvREFCNT_dec (sv);
2142	croak ("\3async_pool terminate\2\n");
2143	}	3982	}
2144		3983
2145	SvREFCNT_dec (coro->saved_deffh);
2146	coro->saved_deffh = SvREFCNT_inc ((SV *)PL_defoutgv);
2147
2148	hv_store (hv, "desc", sizeof ("desc") - 1,
2149	newSVpvn ("[async_pool]", sizeof ("[async_pool]") - 1), 0);
2150
2151	invoke_av = (AV *)SvRV (invoke);
2152	len = av_len (invoke_av);
2153
2154	sv_setsv (cb, AvARRAY (invoke_av)[0]);
2155
2156	if (len > 0)
2157	{	3984	{
2158	av_fill (defav, len - 1);	3985	struct coro *coro = SvSTATE_hv (hv);
2159	for (i = 0; i < len; ++i)	3986
2160	av_store (defav, i, SvREFCNT_inc (AvARRAY (invoke_av)[i + 1]));	3987	assert (!coro->invoke_cb);
		3988	assert (!coro->invoke_av);
		3989	coro->invoke_cb = SvREFCNT_inc (cb);
		3990	coro->invoke_av = av;
2161	}	3991	}
2162		3992
		3993	api_ready (aTHX_ (SV *)hv);
		3994
		3995	if (GIMME_V != G_VOID)
		3996	XPUSHs (sv_2mortal (newRV_noinc ((SV *)hv)));
		3997	else
2163	SvREFCNT_dec (invoke);	3998	SvREFCNT_dec (hv);
2164	}	3999	}
2165		4000
2166	void	4001	SV *
2167	_pool_2 (SV *cb)	4002	rouse_cb ()
		4003	PROTOTYPE:
2168	CODE:	4004	CODE:
2169	{	4005	RETVAL = coro_new_rouse_cb (aTHX);
2170	struct coro *coro = SvSTATE (coro_current);
2171
2172	sv_setsv (cb, &PL_sv_undef);
2173
2174	SvREFCNT_dec ((SV *)PL_defoutgv); PL_defoutgv = (GV *)coro->saved_deffh;
2175	coro->saved_deffh = 0;
2176
2177	if (coro_rss (aTHX_ coro) > SvIV (sv_pool_rss)
2178	|| av_len (av_async_pool) + 1 >= SvIV (sv_pool_size))
2179	{
2180	SV *old = PL_diehook;
2181	PL_diehook = 0;
2182	SvREFCNT_dec (old);
2183	croak ("\3async_pool terminate\2\n");
2184	}
2185
2186	av_clear (GvAV (PL_defgv));
2187	hv_store ((HV *)SvRV (coro_current), "desc", sizeof ("desc") - 1,
2188	newSVpvn ("[async_pool idle]", sizeof ("[async_pool idle]") - 1), 0);
2189
2190	coro->prio = 0;
2191
2192	if (coro->cctx && (coro->cctx->flags & CC_TRACE))
2193	api_trace (coro_current, 0);
2194
2195	av_push (av_async_pool, newSVsv (coro_current));
2196	}
2197
2198	#if 0
2199
2200	void
2201	_generator_call (...)
2202	PROTOTYPE: @
2203	PPCODE:
2204	fprintf (stderr, "call %p\n", CvXSUBANY(cv).any_ptr);
2205	xxxx
2206	abort ();
2207
2208	SV *
2209	gensub (SV *sub, ...)
2210	PROTOTYPE: &;@
2211	CODE:
2212	{
2213	struct coro *coro;
2214	MAGIC *mg;
2215	CV *xcv;
2216	CV *ncv = (CV *)newSV_type (SVt_PVCV);
2217	int i;
2218
2219	CvGV (ncv) = CvGV (cv);
2220	CvFILE (ncv) = CvFILE (cv);
2221
2222	Newz (0, coro, 1, struct coro);
2223	coro->args = newAV ();
2224	coro->flags = CF_NEW;
2225
2226	av_extend (coro->args, items - 1);
2227	for (i = 1; i < items; i++)
2228	av_push (coro->args, newSVsv (ST (i)));
2229
2230	CvISXSUB_on (ncv);
2231	CvXSUBANY (ncv).any_ptr = (void *)coro;
2232
2233	xcv = GvCV (gv_fetchpv ("Coro::_generator_call", 0, SVt_PVCV));
2234
2235	CvXSUB (ncv) = CvXSUB (xcv);
2236	CvANON_on (ncv);
2237
2238	mg = sv_magicext ((SV *)ncv, 0, CORO_MAGIC_type_state, &coro_gensub_vtbl, (char *)coro, 0);
2239	RETVAL = newRV_noinc ((SV *)ncv);
2240	}
2241	OUTPUT:	4006	OUTPUT:
2242	RETVAL	4007	RETVAL
2243		4008
2244	#endif	4009	void
		4010	rouse_wait (...)
		4011	PROTOTYPE: ;$
		4012	PPCODE:
		4013	CORO_EXECUTE_SLF_XS (slf_init_rouse_wait);
2245		4014
		4015	void
		4016	on_enter (SV *block)
		4017	ALIAS:
		4018	on_leave = 1
		4019	PROTOTYPE: &
		4020	CODE:
		4021	{
		4022	struct coro *coro = SvSTATE_current;
		4023	AV **avp = ix ? &coro->on_leave : &coro->on_enter;
2246		4024
		4025	block = s_get_cv_croak (block);
		4026
		4027	if (!*avp)
		4028	*avp = newAV ();
		4029
		4030	av_push (*avp, SvREFCNT_inc (block));
		4031
		4032	if (!ix)
		4033	on_enterleave_call (aTHX_ block);
		4034
		4035	LEAVE; /* pp_entersub unfortunately forces an ENTER/LEAVE around XS calls */
		4036	SAVEDESTRUCTOR_X (ix ? coro_pop_on_leave : coro_pop_on_enter, (void *)coro);
		4037	ENTER; /* pp_entersub unfortunately forces an ENTER/LEAVE around XS calls */
		4038	}
		4039
		4040
		4041	MODULE = Coro::State PACKAGE = PerlIO::cede
		4042
		4043	BOOT:
		4044	PerlIO_define_layer (aTHX_ &PerlIO_cede);
		4045
		4046
2247	MODULE = Coro::State PACKAGE = Coro::AIO	4047	MODULE = Coro::State PACKAGE = Coro::Semaphore
2248		4048
2249	SV *	4049	SV *
2250	_get_state ()	4050	new (SV *klass, SV *count = 0)
2251	CODE:	4051	CODE:
2252	{	4052	{
2253	struct io_state *data;	4053	int semcnt = 1;
2254		4054
2255	RETVAL = newSV (sizeof (struct io_state));	4055	if (count)
2256	data = (struct io_state *)SvPVX (RETVAL);	4056	{
2257	SvCUR_set (RETVAL, sizeof (struct io_state));	4057	SvGETMAGIC (count);
2258	SvPOK_only (RETVAL);
2259		4058
2260	data->errorno = errno;	4059	if (SvOK (count))
2261	data->laststype = PL_laststype;	4060	semcnt = SvIV (count);
2262	data->laststatval = PL_laststatval;	4061	}
2263	data->statcache = PL_statcache;	4062
		4063	RETVAL = sv_bless (
		4064	coro_waitarray_new (aTHX_ semcnt),
		4065	GvSTASH (CvGV (cv))
		4066	);
2264	}	4067	}
2265	OUTPUT:	4068	OUTPUT:
2266	RETVAL	4069	RETVAL
2267		4070
		4071	# helper for Coro::Channel and others
		4072	SV *
		4073	_alloc (int count)
		4074	CODE:
		4075	RETVAL = coro_waitarray_new (aTHX_ count);
		4076	OUTPUT:
		4077	RETVAL
		4078
		4079	SV *
		4080	count (SV *self)
		4081	CODE:
		4082	RETVAL = newSVsv (AvARRAY ((AV *)SvRV (self))[0]);
		4083	OUTPUT:
		4084	RETVAL
		4085
2268	void	4086	void
2269	_set_state (char *data_)	4087	up (SV *self, int adjust = 1)
2270	PROTOTYPE: $	4088	ALIAS:
		4089	adjust = 1
		4090	CODE:
		4091	coro_semaphore_adjust (aTHX_ (AV *)SvRV (self), ix ? adjust : 1);
		4092
		4093	void
		4094	down (...)
		4095	CODE:
		4096	CORO_EXECUTE_SLF_XS (slf_init_semaphore_down);
		4097
		4098	void
		4099	wait (...)
		4100	CODE:
		4101	CORO_EXECUTE_SLF_XS (slf_init_semaphore_wait);
		4102
		4103	void
		4104	try (SV *self)
		4105	PPCODE:
		4106	{
		4107	AV *av = (AV *)SvRV (self);
		4108	SV *count_sv = AvARRAY (av)[0];
		4109	IV count = SvIVX (count_sv);
		4110
		4111	if (count > 0)
		4112	{
		4113	--count;
		4114	SvIVX (count_sv) = count;
		4115	XSRETURN_YES;
		4116	}
		4117	else
		4118	XSRETURN_NO;
		4119	}
		4120
		4121	void
		4122	waiters (SV *self)
		4123	PPCODE:
		4124	{
		4125	AV *av = (AV *)SvRV (self);
		4126	int wcount = AvFILLp (av) + 1 - 1;
		4127
		4128	if (GIMME_V == G_SCALAR)
		4129	XPUSHs (sv_2mortal (newSViv (wcount)));
		4130	else
		4131	{
		4132	int i;
		4133	EXTEND (SP, wcount);
		4134	for (i = 1; i <= wcount; ++i)
		4135	PUSHs (sv_2mortal (newRV_inc (AvARRAY (av)[i])));
		4136	}
		4137	}
		4138
		4139	MODULE = Coro::State PACKAGE = Coro::SemaphoreSet
		4140
		4141	void
		4142	_may_delete (SV *sem, int count, unsigned int extra_refs)
		4143	PPCODE:
		4144	{
		4145	AV *av = (AV *)SvRV (sem);
		4146
		4147	if (SvREFCNT ((SV *)av) == 1 + extra_refs
		4148	&& AvFILLp (av) == 0 /* no waiters, just count */
		4149	&& SvIV (AvARRAY (av)[0]) == count)
		4150	XSRETURN_YES;
		4151
		4152	XSRETURN_NO;
		4153	}
		4154
		4155	MODULE = Coro::State PACKAGE = Coro::Signal
		4156
		4157	SV *
		4158	new (SV *klass)
2271	CODE:	4159	CODE:
2272	{	4160	RETVAL = sv_bless (
2273	struct io_state *data = (void *)data_;	4161	coro_waitarray_new (aTHX_ 0),
		4162	GvSTASH (CvGV (cv))
		4163	);
		4164	OUTPUT:
		4165	RETVAL
2274		4166
2275	errno = data->errorno;	4167	void
2276	PL_laststype = data->laststype;	4168	wait (...)
2277	PL_laststatval = data->laststatval;	4169	CODE:
2278	PL_statcache = data->statcache;	4170	CORO_EXECUTE_SLF_XS (slf_init_signal_wait);
		4171
		4172	void
		4173	broadcast (SV *self)
		4174	CODE:
		4175	{
		4176	AV *av = (AV *)SvRV (self);
		4177	coro_signal_wake (aTHX_ av, AvFILLp (av));
2279	}	4178	}
		4179
		4180	void
		4181	send (SV *self)
		4182	CODE:
		4183	{
		4184	AV *av = (AV *)SvRV (self);
		4185
		4186	if (AvFILLp (av))
		4187	coro_signal_wake (aTHX_ av, 1);
		4188	else
		4189	SvIVX (AvARRAY (av)[0]) = 1; /* remember the signal */
		4190	}
		4191
		4192	IV
		4193	awaited (SV *self)
		4194	CODE:
		4195	RETVAL = AvFILLp ((AV *)SvRV (self)) + 1 - 1;
		4196	OUTPUT:
		4197	RETVAL
2280		4198
2281		4199
2282	MODULE = Coro::State PACKAGE = Coro::AnyEvent	4200	MODULE = Coro::State PACKAGE = Coro::AnyEvent
2283		4201
2284	BOOT:	4202	BOOT:
2285	sv_activity = coro_get_sv (aTHX_ "Coro::AnyEvent::ACTIVITY", TRUE);	4203	sv_activity = coro_get_sv (aTHX_ "Coro::AnyEvent::ACTIVITY", TRUE);
2286		4204
2287	SV *	4205	void
2288	_schedule (...)	4206	_schedule (...)
2289	PROTOTYPE: @
2290	CODE:	4207	CODE:
2291	{	4208	{
2292	static int incede;	4209	static int incede;
2293		4210
2294	api_cede_notself ();	4211	api_cede_notself (aTHX);
2295		4212
2296	++incede;	4213	++incede;
2297	while (coro_nready >= incede && api_cede ())	4214	while (coro_nready >= incede && api_cede (aTHX))
2298	;	4215	;
2299		4216
2300	sv_setsv (sv_activity, &PL_sv_undef);	4217	sv_setsv (sv_activity, &PL_sv_undef);
2301	if (coro_nready >= incede)	4218	if (coro_nready >= incede)
2302	{	4219	{
2303	PUSHMARK (SP);	4220	PUSHMARK (SP);
2304	PUTBACK;	4221	PUTBACK;
2305	call_pv ("Coro::AnyEvent::_activity", G_DISCARD | G_EVAL);	4222	call_pv ("Coro::AnyEvent::_activity", G_KEEPERR | G_EVAL | G_VOID | G_DISCARD);
2306	SPAGAIN;
2307	}	4223	}
2308		4224
2309	--incede;	4225	--incede;
2310	}	4226	}
2311		4227
2312		4228
2313	MODULE = Coro::State PACKAGE = PerlIO::cede	4229	MODULE = Coro::State PACKAGE = Coro::AIO
2314		4230
2315	BOOT:	4231	void
2316	PerlIO_define_layer (aTHX_ &PerlIO_cede);	4232	_register (char *target, char *proto, SV *req)
		4233	CODE:
		4234	{
		4235	SV *req_cv = s_get_cv_croak (req);
		4236	/* newXSproto doesn't return the CV on 5.8 */
		4237	CV *slf_cv = newXS (target, coro_aio_req_xs, __FILE__);
		4238	sv_setpv ((SV *)slf_cv, proto);
		4239	sv_magicext ((SV *)slf_cv, (SV *)req_cv, CORO_MAGIC_type_aio, 0, 0, 0);
		4240	}
		4241
		4242	MODULE = Coro::State PACKAGE = Coro::Select
		4243
		4244	void
		4245	patch_pp_sselect ()
		4246	CODE:
		4247	if (!coro_old_pp_sselect)
		4248	{
		4249	coro_select_select = (SV *)get_cv ("Coro::Select::select", 0);
		4250	coro_old_pp_sselect = PL_ppaddr [OP_SSELECT];
		4251	PL_ppaddr [OP_SSELECT] = coro_pp_sselect;
		4252	}
		4253
		4254	void
		4255	unpatch_pp_sselect ()
		4256	CODE:
		4257	if (coro_old_pp_sselect)
		4258	{
		4259	PL_ppaddr [OP_SSELECT] = coro_old_pp_sselect;
		4260	coro_old_pp_sselect = 0;
		4261	}
		4262
		4263	MODULE = Coro::State PACKAGE = Coro::Util
		4264
		4265	void
		4266	_exit (int code)
		4267	CODE:
		4268	_exit (code);
		4269
		4270	NV
		4271	time ()
		4272	CODE:
		4273	RETVAL = nvtime (aTHX);
		4274	OUTPUT:
		4275	RETVAL
		4276
		4277	NV
		4278	gettimeofday ()
		4279	PPCODE:
		4280	{
		4281	UV tv [2];
		4282	u2time (aTHX_ tv);
		4283	EXTEND (SP, 2);
		4284	PUSHs (sv_2mortal (newSVuv (tv [0])));
		4285	PUSHs (sv_2mortal (newSVuv (tv [1])));
		4286	}
		4287

Diff Legend

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