ViewVC Help

View File | Revision Log | Show Annotations | Download File

/cvs/Coro/Coro/State.xs

Comparing Coro/Coro/State.xs (file contents):
Revision 1.222 by root, Sun Jan 20 10:22:07 2008 UTC vs.
Revision 1.265 by root, Fri Nov 14 02:42:26 2008 UTC

…		…
4	#define PERL_EXT	4	#define PERL_EXT
5		5
6	#include "EXTERN.h"	6	#include "EXTERN.h"
7	#include "perl.h"	7	#include "perl.h"
8	#include "XSUB.h"	8	#include "XSUB.h"
		9	#include "perliol.h"
9		10
10	#include "patchlevel.h"	11	#include "patchlevel.h"
11		12
12	#include <stdio.h>	13	#include <stdio.h>
13	#include <errno.h>	14	#include <errno.h>
14	#include <assert.h>	15	#include <assert.h>
		16
		17	#ifdef WIN32
		18	# undef setjmp
		19	# undef longjmp
		20	# undef _exit
		21	# define setjmp _setjmp // deep magic, don't ask
		22	#else
15	#include <inttypes.h> /* portable stdint.h */	23	# include <inttypes.h> /* most portable stdint.h */
		24	#endif
16		25
17	#ifdef HAVE_MMAP	26	#ifdef HAVE_MMAP
18	# include <unistd.h>	27	# include <unistd.h>
19	# include <sys/mman.h>	28	# include <sys/mman.h>
20	# ifndef MAP_ANONYMOUS	29	# ifndef MAP_ANONYMOUS
…		…
37	# define BOOT_PAGESIZE (void)0	46	# define BOOT_PAGESIZE (void)0
38	#endif	47	#endif
39		48
40	#if CORO_USE_VALGRIND	49	#if CORO_USE_VALGRIND
41	# include <valgrind/valgrind.h>	50	# include <valgrind/valgrind.h>
42	# define REGISTER_STACK(cctx,start,end) (cctx)->valgrind_id = VALGRIND_STACK_REGISTER ((start), (end))
43	#else
44	# define REGISTER_STACK(cctx,start,end)
45	#endif	51	#endif
46		52
47	/* the maximum number of idle cctx that will be pooled */	53	/* the maximum number of idle cctx that will be pooled */
48	#define MAX_IDLE_CCTX 8	54	static int cctx_max_idle = 4;
49		55
50	#define PERL_VERSION_ATLEAST(a,b,c) \	56	#define PERL_VERSION_ATLEAST(a,b,c) \
51	(PERL_REVISION > (a) \	57	(PERL_REVISION > (a) \
52	|| (PERL_REVISION == (a) \	58	|| (PERL_REVISION == (a) \
53	&& (PERL_VERSION > (b) \	59	&& (PERL_VERSION > (b) \
…		…
72	# ifndef IS_PADCONST	78	# ifndef IS_PADCONST
73	# define IS_PADCONST(v) 0	79	# define IS_PADCONST(v) 0
74	# endif	80	# endif
75	#endif	81	#endif
76		82
		83	/* 5.11 */
		84	#ifndef CxHASARGS
		85	# define CxHASARGS(cx) (cx)->blk_sub.hasargs
		86	#endif
		87
		88	/* 5.10.0 */
		89	#ifndef SvREFCNT_inc_NN
		90	# define SvREFCNT_inc_NN(sv) SvREFCNT_inc (sv)
		91	#endif
		92
		93	/* 5.8.8 */
		94	#ifndef GV_NOTQUAL
		95	# define GV_NOTQUAL 0
		96	#endif
		97	#ifndef newSV
		98	# define newSV(l) NEWSV(0,l)
		99	#endif
		100
77	/* 5.8.7 */	101	/* 5.8.7 */
78	#ifndef SvRV_set	102	#ifndef SvRV_set
79	# define SvRV_set(s,v) SvRV(s) = (v)	103	# define SvRV_set(s,v) SvRV(s) = (v)
80	#endif
81
82	/* 5.8.8 */
83	#ifndef GV_NOTQUAL
84	# define GV_NOTQUAL 0
85	#endif
86	#ifndef newSV
87	# define newSV(l) NEWSV(0,l)
88	#endif	104	#endif
89		105
90	#if !__i386 && !__x86_64 && !__powerpc && !__m68k && !__alpha && !__mips && !__sparc64	106	#if !__i386 && !__x86_64 && !__powerpc && !__m68k && !__alpha && !__mips && !__sparc64
91	# undef CORO_STACKGUARD	107	# undef CORO_STACKGUARD
92	#endif	108	#endif
…		…
107		123
108	#define IN_DESTRUCT (PL_main_cv == Nullcv)	124	#define IN_DESTRUCT (PL_main_cv == Nullcv)
109		125
110	#if __GNUC__ >= 3	126	#if __GNUC__ >= 3
111	# define attribute(x) __attribute__(x)	127	# define attribute(x) __attribute__(x)
112	# define BARRIER __asm__ __volatile__ ("" : : : "memory")
113	# define expect(expr,value) __builtin_expect ((expr),(value))	128	# define expect(expr,value) __builtin_expect ((expr),(value))
		129	# define INLINE static inline
114	#else	130	#else
115	# define attribute(x)	131	# define attribute(x)
116	# define BARRIER
117	# define expect(expr,value) (expr)	132	# define expect(expr,value) (expr)
		133	# define INLINE static
118	#endif	134	#endif
119		135
120	#define expect_false(expr) expect ((expr) != 0, 0)	136	#define expect_false(expr) expect ((expr) != 0, 0)
121	#define expect_true(expr) expect ((expr) != 0, 1)	137	#define expect_true(expr) expect ((expr) != 0, 1)
122		138
123	#define NOINLINE attribute ((noinline))	139	#define NOINLINE attribute ((noinline))
124		140
125	#include "CoroAPI.h"	141	#include "CoroAPI.h"
126		142
127	#ifdef USE_ITHREADS	143	#ifdef USE_ITHREADS
		144
128	static perl_mutex coro_mutex;	145	static perl_mutex coro_lock;
129	# define LOCK do { MUTEX_LOCK (&coro_mutex); } while (0)	146	# define LOCK do { MUTEX_LOCK (&coro_lock); } while (0)
130	# define UNLOCK do { MUTEX_UNLOCK (&coro_mutex); } while (0)	147	# define UNLOCK do { MUTEX_UNLOCK (&coro_lock); } while (0)
		148	# if CORO_PTHREAD
		149	static void *coro_thx;
		150	# endif
		151
131	#else	152	#else
		153
132	# define LOCK (void)0	154	# define LOCK (void)0
133	# define UNLOCK (void)0	155	# define UNLOCK (void)0
		156
134	#endif	157	#endif
		158
		159	# undef LOCK
		160	# define LOCK (void)0
		161	# undef UNLOCK
		162	# define UNLOCK (void)0
135		163
136	/* helper storage struct for Coro::AIO */	164	/* helper storage struct for Coro::AIO */
137	struct io_state	165	struct io_state
138	{	166	{
		167	AV *res;
139	int errorno;	168	int errorno;
140	I32 laststype;	169	I32 laststype; /* U16 in 5.10.0 */
141	int laststatval;	170	int laststatval;
142	Stat_t statcache;	171	Stat_t statcache;
143	};	172	};
144		173
		174	static double (*nvtime)(); /* so why doesn't it take void? */
		175
		176	static U32 cctx_gen;
145	static size_t coro_stacksize = CORO_STACKSIZE;	177	static size_t cctx_stacksize = CORO_STACKSIZE;
146	static struct CoroAPI coroapi;	178	static struct CoroAPI coroapi;
147	static AV *main_mainstack; /* used to differentiate between $main and others */	179	static AV *main_mainstack; /* used to differentiate between $main and others */
148	static JMPENV *main_top_env;	180	static JMPENV *main_top_env;
149	static HV *coro_state_stash, *coro_stash;	181	static HV *coro_state_stash, *coro_stash;
150	static volatile SV *coro_mortal; /* will be freed after next transfer */	182	static volatile SV *coro_mortal; /* will be freed/thrown after next transfer */
		183	static volatile struct coro *transfer_next;
		184
		185	struct transfer_args
		186	{
		187	struct coro *prev, *next;
		188	};
151		189
152	static GV *irsgv; /* $/ */	190	static GV *irsgv; /* $/ */
153	static GV *stdoutgv; /* *STDOUT */	191	static GV *stdoutgv; /* *STDOUT */
154	static SV *rv_diehook;	192	static SV *rv_diehook;
155	static SV *rv_warnhook;	193	static SV *rv_warnhook;
…		…
157		195
158	/* async_pool helper stuff */	196	/* async_pool helper stuff */
159	static SV *sv_pool_rss;	197	static SV *sv_pool_rss;
160	static SV *sv_pool_size;	198	static SV *sv_pool_size;
161	static AV *av_async_pool;	199	static AV *av_async_pool;
		200
		201	/* Coro::AnyEvent */
		202	static SV *sv_activity;
162		203
163	static struct coro_cctx *cctx_first;	204	static struct coro_cctx *cctx_first;
164	static int cctx_count, cctx_idle;	205	static int cctx_count, cctx_idle;
165		206
166	enum {	207	enum {
…		…
184	void *idle_sp; /* sp of top-level transfer/schedule/cede call */	225	void *idle_sp; /* sp of top-level transfer/schedule/cede call */
185	JMPENV *idle_te; /* same as idle_sp, but for top_env, TODO: remove once stable */	226	JMPENV *idle_te; /* same as idle_sp, but for top_env, TODO: remove once stable */
186	JMPENV *top_env;	227	JMPENV *top_env;
187	coro_context cctx;	228	coro_context cctx;
188		229
		230	U32 gen;
189	#if CORO_USE_VALGRIND	231	#if CORO_USE_VALGRIND
190	int valgrind_id;	232	int valgrind_id;
191	#endif	233	#endif
192	unsigned char flags;	234	unsigned char flags;
193	} coro_cctx;	235	} coro_cctx;
…		…
252	#define PRIO_IDLE -3	294	#define PRIO_IDLE -3
253	#define PRIO_MIN -4	295	#define PRIO_MIN -4
254		296
255	/* for Coro.pm */	297	/* for Coro.pm */
256	static SV *coro_current;	298	static SV *coro_current;
		299	static SV *coro_readyhook;
257	static AV *coro_ready [PRIO_MAX-PRIO_MIN+1];	300	static AV *coro_ready [PRIO_MAX - PRIO_MIN + 1];
258	static int coro_nready;	301	static int coro_nready;
259	static struct coro *coro_first;	302	static struct coro *coro_first;
260		303
261	/** lowlevel stuff **********************************************************/	304	/** lowlevel stuff **********************************************************/
262		305
263	static SV *	306	static SV *
264	coro_get_sv (pTHX_ const char *name, int create)	307	coro_get_sv (pTHX_ const char *name, int create)
265	{	308	{
266	#if PERL_VERSION_ATLEAST (5,9,0)	309	#if PERL_VERSION_ATLEAST (5,10,0)
267	/* silence stupid and wrong 5.10 warning that I am unable to switch off */	310	/* silence stupid and wrong 5.10 warning that I am unable to switch off */
268	get_sv (name, create);	311	get_sv (name, create);
269	#endif	312	#endif
270	return get_sv (name, create);	313	return get_sv (name, create);
271	}	314	}
272		315
273	static AV *	316	static AV *
274	coro_get_av (pTHX_ const char *name, int create)	317	coro_get_av (pTHX_ const char *name, int create)
275	{	318	{
276	#if PERL_VERSION_ATLEAST (5,9,0)	319	#if PERL_VERSION_ATLEAST (5,10,0)
277	/* silence stupid and wrong 5.10 warning that I am unable to switch off */	320	/* silence stupid and wrong 5.10 warning that I am unable to switch off */
278	get_av (name, create);	321	get_av (name, create);
279	#endif	322	#endif
280	return get_av (name, create);	323	return get_av (name, create);
281	}	324	}
282		325
283	static HV *	326	static HV *
284	coro_get_hv (pTHX_ const char *name, int create)	327	coro_get_hv (pTHX_ const char *name, int create)
285	{	328	{
286	#if PERL_VERSION_ATLEAST (5,9,0)	329	#if PERL_VERSION_ATLEAST (5,10,0)
287	/* silence stupid and wrong 5.10 warning that I am unable to switch off */	330	/* silence stupid and wrong 5.10 warning that I am unable to switch off */
288	get_hv (name, create);	331	get_hv (name, create);
289	#endif	332	#endif
290	return get_hv (name, create);	333	return get_hv (name, create);
291	}	334	}
…		…
296	AV *padlist = CvPADLIST (cv);	339	AV *padlist = CvPADLIST (cv);
297	AV *newpadlist, *newpad;	340	AV *newpadlist, *newpad;
298		341
299	newpadlist = newAV ();	342	newpadlist = newAV ();
300	AvREAL_off (newpadlist);	343	AvREAL_off (newpadlist);
301	#if PERL_VERSION_ATLEAST (5,9,0)	344	#if PERL_VERSION_ATLEAST (5,10,0)
302	Perl_pad_push (aTHX_ padlist, AvFILLp (padlist) + 1);	345	Perl_pad_push (aTHX_ padlist, AvFILLp (padlist) + 1);
303	#else	346	#else
304	Perl_pad_push (aTHX_ padlist, AvFILLp (padlist) + 1, 1);	347	Perl_pad_push (aTHX_ padlist, AvFILLp (padlist) + 1, 1);
305	#endif	348	#endif
306	newpad = (AV *)AvARRAY (padlist)[AvFILLp (padlist)];	349	newpad = (AV *)AvARRAY (padlist)[AvFILLp (padlist)];
307	--AvFILLp (padlist);	350	--AvFILLp (padlist);
308		351
309	av_store (newpadlist, 0, SvREFCNT_inc (*av_fetch (padlist, 0, FALSE)));	352	av_store (newpadlist, 0, SvREFCNT_inc_NN (*av_fetch (padlist, 0, FALSE)));
310	av_store (newpadlist, 1, (SV *)newpad);	353	av_store (newpadlist, 1, (SV *)newpad);
311		354
312	return newpadlist;	355	return newpadlist;
313	}	356	}
314		357
…		…
344		387
345	/* casting is fun. */	388	/* casting is fun. */
346	while (&PL_sv_undef != (SV *)(padlist = (AV *)av_pop (av)))	389	while (&PL_sv_undef != (SV *)(padlist = (AV *)av_pop (av)))
347	free_padlist (aTHX_ padlist);	390	free_padlist (aTHX_ padlist);
348		391
		392	SvREFCNT_dec (av); /* sv_magicext increased the refcount */
		393
349	return 0;	394	return 0;
350	}	395	}
351		396
352	#define CORO_MAGIC_type_cv PERL_MAGIC_ext	397	#define CORO_MAGIC_type_cv PERL_MAGIC_ext
353	#define CORO_MAGIC_type_state PERL_MAGIC_ext	398	#define CORO_MAGIC_type_state PERL_MAGIC_ext
…		…
355	static MGVTBL coro_cv_vtbl = {	400	static MGVTBL coro_cv_vtbl = {
356	0, 0, 0, 0,	401	0, 0, 0, 0,
357	coro_cv_free	402	coro_cv_free
358	};	403	};
359		404
360	#define CORO_MAGIC(sv,type) \	405	#define CORO_MAGIC(sv, type) \
361	SvMAGIC (sv) \	406	SvMAGIC (sv) \
362	? SvMAGIC (sv)->mg_type == type \	407	? SvMAGIC (sv)->mg_type == type \
363	? SvMAGIC (sv) \	408	? SvMAGIC (sv) \
364	: mg_find (sv, type) \	409	: mg_find (sv, type) \
365	: 0	410	: 0
366		411
367	#define CORO_MAGIC_cv(cv) CORO_MAGIC (((SV *)(cv)), CORO_MAGIC_type_cv)	412	#define CORO_MAGIC_cv(cv) CORO_MAGIC (((SV *)(cv)), CORO_MAGIC_type_cv)
368	#define CORO_MAGIC_state(sv) CORO_MAGIC (((SV *)(sv)), CORO_MAGIC_type_state)	413	#define CORO_MAGIC_state(sv) CORO_MAGIC (((SV *)(sv)), CORO_MAGIC_type_state)
369		414
370	static struct coro *	415	INLINE struct coro *
371	SvSTATE_ (pTHX_ SV *coro)	416	SvSTATE_ (pTHX_ SV *coro)
372	{	417	{
373	HV *stash;	418	HV *stash;
374	MAGIC *mg;	419	MAGIC *mg;
375		420
…		…
403	if (expect_true (mg && AvFILLp ((av = (AV *)mg->mg_obj)) >= 0))	448	if (expect_true (mg && AvFILLp ((av = (AV *)mg->mg_obj)) >= 0))
404	CvPADLIST (cv) = (AV *)AvARRAY (av)[AvFILLp (av)--];	449	CvPADLIST (cv) = (AV *)AvARRAY (av)[AvFILLp (av)--];
405	else	450	else
406	{	451	{
407	#if CORO_PREFER_PERL_FUNCTIONS	452	#if CORO_PREFER_PERL_FUNCTIONS
408	/* this is probably cleaner, but also slower? */	453	/* this is probably cleaner? but also slower! */
		454	/* in practise, it seems to be less stable */
409	CV *cp = Perl_cv_clone (cv);	455	CV *cp = Perl_cv_clone (cv);
410	CvPADLIST (cv) = CvPADLIST (cp);	456	CvPADLIST (cv) = CvPADLIST (cp);
411	CvPADLIST (cp) = 0;	457	CvPADLIST (cp) = 0;
412	SvREFCNT_dec (cp);	458	SvREFCNT_dec (cp);
413	#else	459	#else
…		…
497		543
498	if (expect_true (CvDEPTH (cv)))	544	if (expect_true (CvDEPTH (cv)))
499	{	545	{
500	EXTEND (SP, 3);	546	EXTEND (SP, 3);
501	PUSHs ((SV *)CvPADLIST (cv));	547	PUSHs ((SV *)CvPADLIST (cv));
502	PUSHs (INT2PTR (SV *, CvDEPTH (cv)));	548	PUSHs (INT2PTR (SV *, (IV)CvDEPTH (cv)));
503	PUSHs ((SV *)cv);	549	PUSHs ((SV *)cv);
504		550
505	CvDEPTH (cv) = 0;	551	CvDEPTH (cv) = 0;
506	get_padlist (aTHX_ cv);	552	get_padlist (aTHX_ cv);
507	}	553	}
…		…
587		633
588	New(54,PL_savestack,24,ANY);	634	New(54,PL_savestack,24,ANY);
589	PL_savestack_ix = 0;	635	PL_savestack_ix = 0;
590	PL_savestack_max = 24;	636	PL_savestack_max = 24;
591		637
592	#if !PERL_VERSION_ATLEAST (5,9,0)	638	#if !PERL_VERSION_ATLEAST (5,10,0)
593	New(54,PL_retstack,4,OP*);	639	New(54,PL_retstack,4,OP*);
594	PL_retstack_ix = 0;	640	PL_retstack_ix = 0;
595	PL_retstack_max = 4;	641	PL_retstack_max = 4;
596	#endif	642	#endif
597	}	643	}
…		…
599		645
600	/*	646	/*
601	* destroy the stacks, the callchain etc...	647	* destroy the stacks, the callchain etc...
602	*/	648	*/
603	static void	649	static void
604	coro_destroy_stacks (pTHX)	650	coro_destruct_stacks (pTHX)
605	{	651	{
606	while (PL_curstackinfo->si_next)	652	while (PL_curstackinfo->si_next)
607	PL_curstackinfo = PL_curstackinfo->si_next;	653	PL_curstackinfo = PL_curstackinfo->si_next;
608		654
609	while (PL_curstackinfo)	655	while (PL_curstackinfo)
…		…
620		666
621	Safefree (PL_tmps_stack);	667	Safefree (PL_tmps_stack);
622	Safefree (PL_markstack);	668	Safefree (PL_markstack);
623	Safefree (PL_scopestack);	669	Safefree (PL_scopestack);
624	Safefree (PL_savestack);	670	Safefree (PL_savestack);
625	#if !PERL_VERSION_ATLEAST (5,9,0)	671	#if !PERL_VERSION_ATLEAST (5,10,0)
626	Safefree (PL_retstack);	672	Safefree (PL_retstack);
627	#endif	673	#endif
628	}	674	}
629		675
630	static size_t	676	static size_t
…		…
646	#undef VAR	692	#undef VAR
647	}	693	}
648	else	694	else
649	slot = coro->slot;	695	slot = coro->slot;
650		696
		697	if (slot)
		698	{
651	rss += sizeof (slot->curstackinfo);	699	rss += sizeof (slot->curstackinfo);
652	rss += (slot->curstackinfo->si_cxmax + 1) * sizeof (PERL_CONTEXT);	700	rss += (slot->curstackinfo->si_cxmax + 1) * sizeof (PERL_CONTEXT);
653	rss += sizeof (SV) + sizeof (struct xpvav) + (1 + AvMAX (slot->curstack)) * sizeof (SV *);	701	rss += sizeof (SV) + sizeof (struct xpvav) + (1 + AvMAX (slot->curstack)) * sizeof (SV *);
654	rss += slot->tmps_max * sizeof (SV *);	702	rss += slot->tmps_max * sizeof (SV *);
655	rss += (slot->markstack_max - slot->markstack_ptr) * sizeof (I32);	703	rss += (slot->markstack_max - slot->markstack_ptr) * sizeof (I32);
656	rss += slot->scopestack_max * sizeof (I32);	704	rss += slot->scopestack_max * sizeof (I32);
657	rss += slot->savestack_max * sizeof (ANY);	705	rss += slot->savestack_max * sizeof (ANY);
658		706
659	#if !PERL_VERSION_ATLEAST (5,9,0)	707	#if !PERL_VERSION_ATLEAST (5,10,0)
660	rss += slot->retstack_max * sizeof (OP *);	708	rss += slot->retstack_max * sizeof (OP *);
661	#endif	709	#endif
		710	}
662	}	711	}
663		712
664	return rss;	713	return rss;
		714	}
		715
		716	/** set stacklevel support **************************************************/
		717
		718	/* we sometimes need to create the effect of pp_set_stacklevel calling us */
		719	#define SSL_HEAD (void)0
		720	/* we somtimes need to create the effect of leaving via pp_set_stacklevel */
		721	#define SSL_TAIL set_stacklevel_tail (aTHX)
		722
		723	INLINE void
		724	set_stacklevel_tail (pTHX)
		725	{
		726	dSP;
		727	int gimme = GIMME_V;
		728
		729	if (gimme == G_SCALAR)
		730	XPUSHs (&PL_sv_undef);
		731
		732	PUTBACK;
665	}	733	}
666		734
667	/** coroutine stack handling ************************************************/	735	/** coroutine stack handling ************************************************/
668		736
669	static int (*orig_sigelem_get) (pTHX_ SV *sv, MAGIC *mg);	737	static int (*orig_sigelem_get) (pTHX_ SV *sv, MAGIC *mg);
670	static int (*orig_sigelem_set) (pTHX_ SV *sv, MAGIC *mg);	738	static int (*orig_sigelem_set) (pTHX_ SV *sv, MAGIC *mg);
		739	static int (*orig_sigelem_clr) (pTHX_ SV *sv, MAGIC *mg);
		740
		741	/* apparently < 5.8.8 */
		742	#ifndef MgPV_nolen_const
		743	#define MgPV_nolen_const(mg) (((((int)(mg)->mg_len)) == HEf_SVKEY) ? \
		744	SvPV_nolen((SV*)((mg)->mg_ptr)) : \
		745	(const char*)(mg)->mg_ptr)
		746	#endif
671		747
672	/*	748	/*
673	* This overrides the default magic get method of %SIG elements.	749	* This overrides the default magic get method of %SIG elements.
674	* The original one doesn't provide for reading back of PL_diehook/PL_warnhook	750	* The original one doesn't provide for reading back of PL_diehook/PL_warnhook
675	* and instead of tryign to save and restore the hash elements, we just provide	751	* and instead of tryign to save and restore the hash elements, we just provide
…		…
683	{	759	{
684	const char *s = MgPV_nolen_const (mg);	760	const char *s = MgPV_nolen_const (mg);
685		761
686	if (*s == '_')	762	if (*s == '_')
687	{	763	{
688	if (strEQ (s, "__DIE__" ) && PL_diehook ) return sv_setsv (sv, PL_diehook ), 0;	764	SV **svp = 0;
689	if (strEQ (s, "__WARN__") && PL_warnhook) return sv_setsv (sv, PL_warnhook), 0;	765
		766	if (strEQ (s, "__DIE__" )) svp = &PL_diehook;
		767	if (strEQ (s, "__WARN__")) svp = &PL_warnhook;
		768
		769	if (svp)
		770	{
		771	sv_setsv (sv, *svp ? *svp : &PL_sv_undef);
		772	return 0;
		773	}
690	}	774	}
691		775
692	return orig_sigelem_get ? orig_sigelem_get (aTHX_ sv, mg) : 0;	776	return orig_sigelem_get ? orig_sigelem_get (aTHX_ sv, mg) : 0;
		777	}
		778
		779	static int
		780	coro_sigelem_clr (pTHX_ SV *sv, MAGIC *mg)
		781	{
		782	const char *s = MgPV_nolen_const (mg);
		783
		784	if (*s == '_')
		785	{
		786	SV **svp = 0;
		787
		788	if (strEQ (s, "__DIE__" )) svp = &PL_diehook;
		789	if (strEQ (s, "__WARN__")) svp = &PL_warnhook;
		790
		791	if (svp)
		792	{
		793	SV *old = *svp;
		794	*svp = 0;
		795	SvREFCNT_dec (old);
		796	return 0;
		797	}
		798	}
		799
		800	return orig_sigelem_clr ? orig_sigelem_clr (aTHX_ sv, mg) : 0;
693	}	801	}
694		802
695	static int	803	static int
696	coro_sigelem_set (pTHX_ SV *sv, MAGIC *mg)	804	coro_sigelem_set (pTHX_ SV *sv, MAGIC *mg)
697	{	805	{
…		…
707	if (svp)	815	if (svp)
708	{	816	{
709	SV *old = *svp;	817	SV *old = *svp;
710	*svp = newSVsv (sv);	818	*svp = newSVsv (sv);
711	SvREFCNT_dec (old);	819	SvREFCNT_dec (old);
712	return;	820	return 0;
713	}	821	}
714	}	822	}
715		823
716	return orig_sigelem_set ? orig_sigelem_set (aTHX_ sv, mg) : 0;	824	return orig_sigelem_set ? orig_sigelem_set (aTHX_ sv, mg) : 0;
717	}	825	}
…		…
731	PL_curpm = 0;	839	PL_curpm = 0;
732	PL_curpad = 0;	840	PL_curpad = 0;
733	PL_localizing = 0;	841	PL_localizing = 0;
734	PL_dirty = 0;	842	PL_dirty = 0;
735	PL_restartop = 0;	843	PL_restartop = 0;
		844	#if PERL_VERSION_ATLEAST (5,10,0)
		845	PL_parser = 0;
		846	#endif
736		847
737	/* recreate the die/warn hooks */	848	/* recreate the die/warn hooks */
738	PL_diehook = 0; SvSetMagicSV (*hv_fetch (hv_sig, "__DIE__" , sizeof ("__DIE__" ) - 1, 1), rv_diehook );	849	PL_diehook = 0; SvSetMagicSV (*hv_fetch (hv_sig, "__DIE__" , sizeof ("__DIE__" ) - 1, 1), rv_diehook );
739	PL_warnhook = 0; SvSetMagicSV (*hv_fetch (hv_sig, "__WARN__", sizeof ("__WARN__") - 1, 1), rv_warnhook);	850	PL_warnhook = 0; SvSetMagicSV (*hv_fetch (hv_sig, "__WARN__", sizeof ("__WARN__") - 1, 1), rv_warnhook);
740		851
741	GvSV (PL_defgv) = newSV (0);	852	GvSV (PL_defgv) = newSV (0);
742	GvAV (PL_defgv) = coro->args; coro->args = 0;	853	GvAV (PL_defgv) = coro->args; coro->args = 0;
743	GvSV (PL_errgv) = newSV (0);	854	GvSV (PL_errgv) = newSV (0);
744	GvSV (irsgv) = newSVpvn ("\n", 1); sv_magic (GvSV (irsgv), (SV *)irsgv, PERL_MAGIC_sv, "/", 0);	855	GvSV (irsgv) = newSVpvn ("\n", 1); sv_magic (GvSV (irsgv), (SV *)irsgv, PERL_MAGIC_sv, "/", 0);
745	PL_rs = newSVsv (GvSV (irsgv));	856	PL_rs = newSVsv (GvSV (irsgv));
746	PL_defoutgv = (GV *)SvREFCNT_inc (stdoutgv);	857	PL_defoutgv = (GV *)SvREFCNT_inc_NN (stdoutgv);
747		858
748	{	859	{
749	dSP;	860	dSP;
750	LOGOP myop;	861	UNOP myop;
751		862
752	Zero (&myop, 1, LOGOP);	863	Zero (&myop, 1, UNOP);
753	myop.op_next = Nullop;	864	myop.op_next = Nullop;
754	myop.op_flags = OPf_WANT_VOID;	865	myop.op_flags = OPf_WANT_VOID;
755		866
756	PUSHMARK (SP);	867	PUSHMARK (SP);
757	XPUSHs (sv_2mortal (av_shift (GvAV (PL_defgv))));	868	XPUSHs (sv_2mortal (av_shift (GvAV (PL_defgv))));
…		…
760	PL_op = PL_ppaddr[OP_ENTERSUB](aTHX);	871	PL_op = PL_ppaddr[OP_ENTERSUB](aTHX);
761	SPAGAIN;	872	SPAGAIN;
762	}	873	}
763		874
764	/* this newly created coroutine might be run on an existing cctx which most	875	/* this newly created coroutine might be run on an existing cctx which most
765	* likely was suspended in set_stacklevel, called from entersub.	876	* likely was suspended in set_stacklevel, called from pp_set_stacklevel,
766	* set_stacklevl doesn't do anything on return, but entersub does LEAVE,	877	* so we have to emulate entering pp_set_stacklevel here.
767	* so we ENTER here for symmetry
768	*/	878	*/
769	ENTER;	879	SSL_HEAD;
770	}	880	}
771		881
772	static void	882	static void
773	coro_destroy (pTHX_ struct coro *coro)	883	coro_destruct (pTHX_ struct coro *coro)
774	{	884	{
775	if (!IN_DESTRUCT)	885	if (!IN_DESTRUCT)
776	{	886	{
777	/* restore all saved variables and stuff */	887	/* restore all saved variables and stuff */
778	LEAVE_SCOPE (0);	888	LEAVE_SCOPE (0);
…		…
800	SvREFCNT_dec (PL_warnhook);	910	SvREFCNT_dec (PL_warnhook);
801		911
802	SvREFCNT_dec (coro->saved_deffh);	912	SvREFCNT_dec (coro->saved_deffh);
803	SvREFCNT_dec (coro->throw);	913	SvREFCNT_dec (coro->throw);
804		914
805	coro_destroy_stacks (aTHX);	915	coro_destruct_stacks (aTHX);
806	}	916	}
807		917
808	static void	918	INLINE void
809	free_coro_mortal (pTHX)	919	free_coro_mortal (pTHX)
810	{	920	{
811	if (expect_true (coro_mortal))	921	if (expect_true (coro_mortal))
812	{	922	{
813	SvREFCNT_dec (coro_mortal);	923	SvREFCNT_dec (coro_mortal);
…		…
847	: cx->blk_gimme == G_SCALAR ? bot + 1	957	: cx->blk_gimme == G_SCALAR ? bot + 1
848	: bot;	958	: bot;
849		959
850	av_extend (av, top - bot);	960	av_extend (av, top - bot);
851	while (bot < top)	961	while (bot < top)
852	av_push (av, SvREFCNT_inc (*bot++));	962	av_push (av, SvREFCNT_inc_NN (*bot++));
853		963
854	PL_runops = RUNOPS_DEFAULT;	964	PL_runops = RUNOPS_DEFAULT;
855	ENTER;	965	ENTER;
856	SAVETMPS;	966	SAVETMPS;
857	EXTEND (SP, 3);	967	EXTEND (SP, 3);
…		…
895	SAVETMPS;	1005	SAVETMPS;
896	EXTEND (SP, 3);	1006	EXTEND (SP, 3);
897	PUSHMARK (SP);	1007	PUSHMARK (SP);
898	PUSHs (&PL_sv_yes);	1008	PUSHs (&PL_sv_yes);
899	PUSHs (fullname);	1009	PUSHs (fullname);
900	PUSHs (cx->blk_sub.hasargs ? sv_2mortal (newRV_inc ((SV *)cx->blk_sub.argarray)) : &PL_sv_undef);	1010	PUSHs (CxHASARGS (cx) ? sv_2mortal (newRV_inc ((SV *)cx->blk_sub.argarray)) : &PL_sv_undef);
901	PUTBACK;	1011	PUTBACK;
902	cb = hv_fetch ((HV *)SvRV (coro_current), "_trace_sub_cb", sizeof ("_trace_sub_cb") - 1, 0);	1012	cb = hv_fetch ((HV *)SvRV (coro_current), "_trace_sub_cb", sizeof ("_trace_sub_cb") - 1, 0);
903	if (cb) call_sv (*cb, G_KEEPERR | G_EVAL | G_VOID | G_DISCARD);	1013	if (cb) call_sv (*cb, G_KEEPERR | G_EVAL | G_VOID | G_DISCARD);
904	SPAGAIN;	1014	SPAGAIN;
905	FREETMPS;	1015	FREETMPS;
…		…
937		1047
938	TAINT_NOT;	1048	TAINT_NOT;
939	return 0;	1049	return 0;
940	}	1050	}
941		1051
		1052	static void
		1053	prepare_set_stacklevel (struct transfer_args *ta, struct coro_cctx *cctx)
		1054	{
		1055	ta->prev = (struct coro *)cctx;
		1056	ta->next = 0;
		1057	}
		1058
942	/* inject a fake call to Coro::State::_cctx_init into the execution */	1059	/* inject a fake call to Coro::State::_cctx_init into the execution */
943	/* _cctx_init should be careful, as it could be called at almost any time */	1060	/* _cctx_init should be careful, as it could be called at almost any time */
944	/* during execution of a perl program */	1061	/* during execution of a perl program */
		1062	/* also initialises PL_top_env */
945	static void NOINLINE	1063	static void NOINLINE
946	cctx_prepare (pTHX_ coro_cctx *cctx)	1064	cctx_prepare (pTHX_ coro_cctx *cctx)
947	{	1065	{
948	dSP;	1066	dSP;
949	LOGOP myop;	1067	UNOP myop;
950		1068
951	PL_top_env = &PL_start_env;	1069	PL_top_env = &PL_start_env;
952		1070
953	if (cctx->flags & CC_TRACE)	1071	if (cctx->flags & CC_TRACE)
954	PL_runops = runops_trace;	1072	PL_runops = runops_trace;
955		1073
956	Zero (&myop, 1, LOGOP);	1074	Zero (&myop, 1, UNOP);
957	myop.op_next = PL_op;	1075	myop.op_next = PL_op;
958	myop.op_flags = OPf_WANT_VOID | OPf_STACKED;	1076	myop.op_flags = OPf_WANT_VOID | OPf_STACKED;
959		1077
960	PUSHMARK (SP);	1078	PUSHMARK (SP);
961	EXTEND (SP, 2);	1079	EXTEND (SP, 2);
962	PUSHs (sv_2mortal (newSViv (PTR2IV (cctx))));	1080	PUSHs (sv_2mortal (newSViv ((IV)cctx)));
963	PUSHs ((SV *)get_cv ("Coro::State::_cctx_init", FALSE));	1081	PUSHs ((SV *)get_cv ("Coro::State::_cctx_init", FALSE));
964	PUTBACK;	1082	PUTBACK;
965	PL_op = (OP *)&myop;	1083	PL_op = (OP *)&myop;
966	PL_op = PL_ppaddr[OP_ENTERSUB](aTHX);	1084	PL_op = PL_ppaddr[OP_ENTERSUB](aTHX);
967	SPAGAIN;	1085	SPAGAIN;
968	}	1086	}
969		1087
		1088	/* the tail of transfer: execute stuff we can only do after a transfer */
		1089	INLINE void
		1090	transfer_tail (pTHX)
		1091	{
		1092	struct coro *next = (struct coro *)transfer_next;
		1093	assert (!(transfer_next = 0)); /* just used for the side effect when asserts are enabled */
		1094	assert (("FATAL: next coroutine was zero in transfer_tail (please report)", next));
		1095
		1096	free_coro_mortal (aTHX);
		1097	UNLOCK;
		1098
		1099	if (expect_false (next->throw))
		1100	{
		1101	SV *exception = sv_2mortal (next->throw);
		1102
		1103	next->throw = 0;
		1104	sv_setsv (ERRSV, exception);
		1105	croak (0);
		1106	}
		1107	}
		1108
970	/*	1109	/*
971	* this is a _very_ stripped down perl interpreter ;)	1110	* this is a _very_ stripped down perl interpreter ;)
972	*/	1111	*/
973	static void	1112	static void
974	cctx_run (void *arg)	1113	cctx_run (void *arg)
975	{	1114	{
		1115	#ifdef USE_ITHREADS
		1116	# if CORO_PTHREAD
		1117	PERL_SET_CONTEXT (coro_thx);
		1118	# endif
		1119	#endif
		1120	{
976	dTHX;	1121	dTHX;
977		1122
978	/* cctx_run is the alternative tail of transfer(), so unlock here. */	1123	/* we are the alternative tail to pp_set_stacklevel */
979	UNLOCK;	1124	/* so do the same things here */
		1125	SSL_TAIL;
980		1126
981	/* we now skip the entersub that lead to transfer() */	1127	/* we now skip the op that did lead to transfer() */
982	PL_op = PL_op->op_next;	1128	PL_op = PL_op->op_next;
983		1129
984	/* inject a fake subroutine call to cctx_init */	1130	/* inject a fake subroutine call to cctx_init */
985	cctx_prepare (aTHX_ (coro_cctx *)arg);	1131	cctx_prepare (aTHX_ (coro_cctx *)arg);
986		1132
		1133	/* cctx_run is the alternative tail of transfer() */
		1134	transfer_tail (aTHX);
		1135
987	/* somebody or something will hit me for both perl_run and PL_restartop */	1136	/* somebody or something will hit me for both perl_run and PL_restartop */
988	PL_restartop = PL_op;	1137	PL_restartop = PL_op;
989	perl_run (PL_curinterp);	1138	perl_run (PL_curinterp);
990		1139
991	/*	1140	/*
992	* If perl-run returns we assume exit() was being called or the coro	1141	* If perl-run returns we assume exit() was being called or the coro
993	* fell off the end, which seems to be the only valid (non-bug)	1142	* fell off the end, which seems to be the only valid (non-bug)
994	* reason for perl_run to return. We try to exit by jumping to the	1143	* reason for perl_run to return. We try to exit by jumping to the
995	* bootstrap-time "top" top_env, as we cannot restore the "main"	1144	* bootstrap-time "top" top_env, as we cannot restore the "main"
996	* coroutine as Coro has no such concept	1145	* coroutine as Coro has no such concept
997	*/	1146	*/
998	PL_top_env = main_top_env;	1147	PL_top_env = main_top_env;
999	JMPENV_JUMP (2); /* I do not feel well about the hardcoded 2 at all */	1148	JMPENV_JUMP (2); /* I do not feel well about the hardcoded 2 at all */
		1149	}
1000	}	1150	}
1001		1151
1002	static coro_cctx *	1152	static coro_cctx *
1003	cctx_new ()	1153	cctx_new ()
1004	{	1154	{
1005	coro_cctx *cctx;	1155	coro_cctx *cctx;
		1156
		1157	++cctx_count;
		1158	New (0, cctx, 1, coro_cctx);
		1159
		1160	cctx->gen = cctx_gen;
		1161	cctx->flags = 0;
		1162	cctx->idle_sp = 0; /* can be accessed by transfer between cctx_run and set_stacklevel, on throw */
		1163
		1164	return cctx;
		1165	}
		1166
		1167	/* create a new cctx only suitable as source */
		1168	static coro_cctx *
		1169	cctx_new_empty ()
		1170	{
		1171	coro_cctx *cctx = cctx_new ();
		1172
		1173	cctx->sptr = 0;
		1174	coro_create (&cctx->cctx, 0, 0, 0, 0);
		1175
		1176	return cctx;
		1177	}
		1178
		1179	/* create a new cctx suitable as destination/running a perl interpreter */
		1180	static coro_cctx *
		1181	cctx_new_run ()
		1182	{
		1183	coro_cctx *cctx = cctx_new ();
1006	void *stack_start;	1184	void *stack_start;
1007	size_t stack_size;	1185	size_t stack_size;
1008		1186
1009	++cctx_count;
1010
1011	Newz (0, cctx, 1, coro_cctx);
1012
1013	#if HAVE_MMAP	1187	#if HAVE_MMAP
1014	cctx->ssize = ((coro_stacksize * sizeof (long) + PAGESIZE - 1) / PAGESIZE + CORO_STACKGUARD) * PAGESIZE;	1188	cctx->ssize = ((cctx_stacksize * sizeof (long) + PAGESIZE - 1) / PAGESIZE + CORO_STACKGUARD) * PAGESIZE;
1015	/* mmap supposedly does allocate-on-write for us */	1189	/* mmap supposedly does allocate-on-write for us */
1016	cctx->sptr = mmap (0, cctx->ssize, PROT_EXEC|PROT_READ|PROT_WRITE, MAP_PRIVATE|MAP_ANONYMOUS, 0, 0);	1190	cctx->sptr = mmap (0, cctx->ssize, PROT_EXEC|PROT_READ|PROT_WRITE, MAP_PRIVATE|MAP_ANONYMOUS, 0, 0);
1017		1191
1018	if (cctx->sptr != (void *)-1)	1192	if (cctx->sptr != (void *)-1)
1019	{	1193	{
1020	# if CORO_STACKGUARD	1194	#if CORO_STACKGUARD
1021	mprotect (cctx->sptr, CORO_STACKGUARD * PAGESIZE, PROT_NONE);	1195	mprotect (cctx->sptr, CORO_STACKGUARD * PAGESIZE, PROT_NONE);
1022	# endif	1196	#endif
1023	stack_start = CORO_STACKGUARD * PAGESIZE + (char *)cctx->sptr;	1197	stack_start = (char *)cctx->sptr + CORO_STACKGUARD * PAGESIZE;
1024	stack_size = cctx->ssize - CORO_STACKGUARD * PAGESIZE;	1198	stack_size = cctx->ssize - CORO_STACKGUARD * PAGESIZE;
1025	cctx->flags |= CC_MAPPED;	1199	cctx->flags |= CC_MAPPED;
1026	}	1200	}
1027	else	1201	else
1028	#endif	1202	#endif
1029	{	1203	{
1030	cctx->ssize = coro_stacksize * (long)sizeof (long);	1204	cctx->ssize = cctx_stacksize * (long)sizeof (long);
1031	New (0, cctx->sptr, coro_stacksize, long);	1205	New (0, cctx->sptr, cctx_stacksize, long);
1032		1206
1033	if (!cctx->sptr)	1207	if (!cctx->sptr)
1034	{	1208	{
1035	perror ("FATAL: unable to allocate stack for coroutine");	1209	perror ("FATAL: unable to allocate stack for coroutine, exiting.");
1036	_exit (EXIT_FAILURE);	1210	_exit (EXIT_FAILURE);
1037	}	1211	}
1038		1212
1039	stack_start = cctx->sptr;	1213	stack_start = cctx->sptr;
1040	stack_size = cctx->ssize;	1214	stack_size = cctx->ssize;
1041	}	1215	}
1042		1216
1043	REGISTER_STACK (cctx, (char *)stack_start, (char *)stack_start + stack_size);	1217	#if CORO_USE_VALGRIND
		1218	cctx->valgrind_id = VALGRIND_STACK_REGISTER ((char *)stack_start, (char *)stack_start + stack_size);
		1219	#endif
		1220
1044	coro_create (&cctx->cctx, cctx_run, (void *)cctx, stack_start, stack_size);	1221	coro_create (&cctx->cctx, cctx_run, (void *)cctx, stack_start, stack_size);
1045		1222
1046	return cctx;	1223	return cctx;
1047	}	1224	}
1048		1225
…		…
1051	{	1228	{
1052	if (!cctx)	1229	if (!cctx)
1053	return;	1230	return;
1054		1231
1055	--cctx_count;	1232	--cctx_count;
		1233	coro_destroy (&cctx->cctx);
1056		1234
		1235	/* coro_transfer creates new, empty cctx's */
		1236	if (cctx->sptr)
		1237	{
1057	#if CORO_USE_VALGRIND	1238	#if CORO_USE_VALGRIND
1058	VALGRIND_STACK_DEREGISTER (cctx->valgrind_id);	1239	VALGRIND_STACK_DEREGISTER (cctx->valgrind_id);
1059	#endif	1240	#endif
1060		1241
1061	#if HAVE_MMAP	1242	#if HAVE_MMAP
1062	if (cctx->flags & CC_MAPPED)	1243	if (cctx->flags & CC_MAPPED)
1063	munmap (cctx->sptr, cctx->ssize);	1244	munmap (cctx->sptr, cctx->ssize);
1064	else	1245	else
1065	#endif	1246	#endif
1066	Safefree (cctx->sptr);	1247	Safefree (cctx->sptr);
		1248	}
1067		1249
1068	Safefree (cctx);	1250	Safefree (cctx);
1069	}	1251	}
1070		1252
1071	/* wether this cctx should be destructed */	1253	/* wether this cctx should be destructed */
1072	#define CCTX_EXPIRED(cctx) ((cctx)->ssize < coro_stacksize || ((cctx)->flags & CC_NOREUSE))	1254	#define CCTX_EXPIRED(cctx) ((cctx)->gen != cctx_gen || ((cctx)->flags & CC_NOREUSE))
1073		1255
1074	static coro_cctx *	1256	static coro_cctx *
1075	cctx_get (pTHX)	1257	cctx_get (pTHX)
1076	{	1258	{
1077	while (expect_true (cctx_first))	1259	while (expect_true (cctx_first))
…		…
1084	return cctx;	1266	return cctx;
1085		1267
1086	cctx_destroy (cctx);	1268	cctx_destroy (cctx);
1087	}	1269	}
1088		1270
1089	return cctx_new ();	1271	return cctx_new_run ();
1090	}	1272	}
1091		1273
1092	static void	1274	static void
1093	cctx_put (coro_cctx *cctx)	1275	cctx_put (coro_cctx *cctx)
1094	{	1276	{
		1277	assert (("FATAL: cctx_put called on non-initialised cctx in Coro (please report)", cctx->sptr));
		1278
1095	/* free another cctx if overlimit */	1279	/* free another cctx if overlimit */
1096	if (expect_false (cctx_idle >= MAX_IDLE_CCTX))	1280	if (expect_false (cctx_idle >= cctx_max_idle))
1097	{	1281	{
1098	coro_cctx *first = cctx_first;	1282	coro_cctx *first = cctx_first;
1099	cctx_first = first->next;	1283	cctx_first = first->next;
1100	--cctx_idle;	1284	--cctx_idle;
1101		1285
…		…
1121	croak ("Coro::State::transfer called with running next Coro::State, but can only transfer to inactive states");	1305	croak ("Coro::State::transfer called with running next Coro::State, but can only transfer to inactive states");
1122		1306
1123	if (expect_false (next->flags & CF_DESTROYED))	1307	if (expect_false (next->flags & CF_DESTROYED))
1124	croak ("Coro::State::transfer called with destroyed next Coro::State, but can only transfer to inactive states");	1308	croak ("Coro::State::transfer called with destroyed next Coro::State, but can only transfer to inactive states");
1125		1309
1126	if (
1127	#if PERL_VERSION_ATLEAST (5,9,0)	1310	#if !PERL_VERSION_ATLEAST (5,10,0)
1128	expect_false (PL_parser && PL_parser->lex_state != LEX_NOTPARSING)
1129	#else
1130	expect_false (PL_lex_state != LEX_NOTPARSING)	1311	if (expect_false (PL_lex_state != LEX_NOTPARSING))
1131	#endif
1132	)
1133	croak ("Coro::State::transfer called while parsing, but this is not supported");	1312	croak ("Coro::State::transfer called while parsing, but this is not supported in your perl version");
		1313	#endif
1134	}	1314	}
1135	}	1315	}
1136		1316
1137	/* always use the TRANSFER macro */	1317	/* always use the TRANSFER macro */
1138	static void NOINLINE	1318	static void NOINLINE
1139	transfer (pTHX_ struct coro *prev, struct coro *next)	1319	transfer (pTHX_ struct coro *prev, struct coro *next, int force_cctx)
1140	{	1320	{
1141	dSTACKLEVEL;	1321	dSTACKLEVEL;
1142	static volatile int has_throw;
1143		1322
1144	/* sometimes transfer is only called to set idle_sp */	1323	/* sometimes transfer is only called to set idle_sp */
1145	if (expect_false (!next))	1324	if (expect_false (!next))
1146	{	1325	{
1147	((coro_cctx *)prev)->idle_sp = STACKLEVEL;	1326	((coro_cctx *)prev)->idle_sp = STACKLEVEL;
…		…
1151	{	1330	{
1152	coro_cctx *prev__cctx;	1331	coro_cctx *prev__cctx;
1153		1332
1154	if (expect_false (prev->flags & CF_NEW))	1333	if (expect_false (prev->flags & CF_NEW))
1155	{	1334	{
1156	/* create a new empty context */	1335	/* create a new empty/source context */
1157	Newz (0, prev->cctx, 1, coro_cctx);	1336	prev->cctx = cctx_new_empty ();
1158	prev->flags &= ~CF_NEW;	1337	prev->flags &= ~CF_NEW;
1159	prev->flags |= CF_RUNNING;	1338	prev->flags |= CF_RUNNING;
1160	}	1339	}
1161		1340
1162	prev->flags &= ~CF_RUNNING;	1341	prev->flags &= ~CF_RUNNING;
…		…
1177	else	1356	else
1178	load_perl (aTHX_ next);	1357	load_perl (aTHX_ next);
1179		1358
1180	prev__cctx = prev->cctx;	1359	prev__cctx = prev->cctx;
1181		1360
1182	/* possibly "free" the cctx */	1361	/* possibly untie and reuse the cctx */
1183	if (expect_true (prev__cctx->idle_sp == STACKLEVEL && !(prev__cctx->flags & CC_TRACE)))	1362	if (expect_true (
		1363	prev__cctx->idle_sp == STACKLEVEL
		1364	&& !(prev__cctx->flags & CC_TRACE)
		1365	&& !force_cctx
		1366	))
1184	{	1367	{
1185	/* I assume that STACKLEVEL is a stronger indicator than PL_top_env changes */	1368	/* I assume that STACKLEVEL is a stronger indicator than PL_top_env changes */
1186	assert (("ERROR: current top_env must equal previous top_env", PL_top_env == prev__cctx->idle_te));	1369	assert (("FATAL: current top_env must equal previous top_env in Coro (please report)", PL_top_env == prev__cctx->idle_te));
1187		1370
1188	prev->cctx = 0;	1371	prev->cctx = 0;
1189		1372
1190	/* if the cctx is about to be destroyed we need to make sure we won't see it in cctx_get */	1373	/* if the cctx is about to be destroyed we need to make sure we won't see it in cctx_get */
1191	/* without this the next cctx_get might destroy the prev__cctx while still in use */	1374	/* without this the next cctx_get might destroy the prev__cctx while still in use */
…		…
1199	++next->usecount;	1382	++next->usecount;
1200		1383
1201	if (expect_true (!next->cctx))	1384	if (expect_true (!next->cctx))
1202	next->cctx = cctx_get (aTHX);	1385	next->cctx = cctx_get (aTHX);
1203		1386
1204	has_throw = !!next->throw;	1387	assert (("FATAL: transfer_next already nonzero in Coro (please report)", !transfer_next));
		1388	transfer_next = next;
1205		1389
1206	if (expect_false (prev__cctx != next->cctx))	1390	if (expect_false (prev__cctx != next->cctx))
1207	{	1391	{
1208	prev__cctx->top_env = PL_top_env;	1392	prev__cctx->top_env = PL_top_env;
1209	PL_top_env = next->cctx->top_env;	1393	PL_top_env = next->cctx->top_env;
1210	coro_transfer (&prev__cctx->cctx, &next->cctx->cctx);	1394	coro_transfer (&prev__cctx->cctx, &next->cctx->cctx);
1211	}	1395	}
1212		1396
1213	free_coro_mortal (aTHX);	1397	transfer_tail (aTHX);
1214	UNLOCK;
1215
1216	if (expect_false (has_throw))
1217	{
1218	struct coro *coro = SvSTATE (coro_current);
1219
1220	if (coro->throw)
1221	{
1222	SV *exception = coro->throw;
1223	coro->throw = 0;
1224	sv_setsv (ERRSV, exception);
1225	croak (0);
1226	}
1227	}
1228	}	1398	}
1229	}	1399	}
1230		1400
1231	struct transfer_args
1232	{
1233	struct coro *prev, *next;
1234	};
1235
1236	#define TRANSFER(ta) transfer (aTHX_ (ta).prev, (ta).next)	1401	#define TRANSFER(ta, force_cctx) transfer (aTHX_ (ta).prev, (ta).next, (force_cctx))
1237	#define TRANSFER_CHECK(ta) transfer_check (aTHX_ (ta).prev, (ta).next)	1402	#define TRANSFER_CHECK(ta) transfer_check (aTHX_ (ta).prev, (ta).next)
1238		1403
1239	/** high level stuff ********************************************************/	1404	/** high level stuff ********************************************************/
1240		1405
1241	static int	1406	static int
…		…
1265	croak ("FATAL: tried to destroy currently running coroutine");	1430	croak ("FATAL: tried to destroy currently running coroutine");
1266		1431
1267	save_perl (aTHX_ &temp);	1432	save_perl (aTHX_ &temp);
1268	load_perl (aTHX_ coro);	1433	load_perl (aTHX_ coro);
1269		1434
1270	coro_destroy (aTHX_ coro);	1435	coro_destruct (aTHX_ coro);
1271		1436
1272	load_perl (aTHX_ &temp);	1437	load_perl (aTHX_ &temp);
1273		1438
1274	coro->slot = 0;	1439	coro->slot = 0;
1275	}	1440	}
…		…
1335	{	1500	{
1336	dTHX;	1501	dTHX;
1337	struct transfer_args ta;	1502	struct transfer_args ta;
1338		1503
1339	prepare_transfer (aTHX_ &ta, prev_sv, next_sv);	1504	prepare_transfer (aTHX_ &ta, prev_sv, next_sv);
1340	TRANSFER (ta);	1505	TRANSFER (ta, 1);
1341	}	1506	}
1342		1507
1343	/** Coro ********************************************************************/	1508	/** Coro ********************************************************************/
1344		1509
1345	static void	1510	static void
…		…
1363	static int	1528	static int
1364	api_ready (SV *coro_sv)	1529	api_ready (SV *coro_sv)
1365	{	1530	{
1366	dTHX;	1531	dTHX;
1367	struct coro *coro;	1532	struct coro *coro;
		1533	SV *sv_hook;
		1534	void (*xs_hook)(void);
1368		1535
1369	if (SvROK (coro_sv))	1536	if (SvROK (coro_sv))
1370	coro_sv = SvRV (coro_sv);	1537	coro_sv = SvRV (coro_sv);
1371		1538
1372	coro = SvSTATE (coro_sv);	1539	coro = SvSTATE (coro_sv);
…		…
1375	return 0;	1542	return 0;
1376		1543
1377	coro->flags |= CF_READY;	1544	coro->flags |= CF_READY;
1378		1545
1379	LOCK;	1546	LOCK;
		1547
		1548	sv_hook = coro_nready ? 0 : coro_readyhook;
		1549	xs_hook = coro_nready ? 0 : coroapi.readyhook;
		1550
1380	coro_enq (aTHX_ SvREFCNT_inc (coro_sv));	1551	coro_enq (aTHX_ SvREFCNT_inc_NN (coro_sv));
1381	++coro_nready;	1552	++coro_nready;
		1553
1382	UNLOCK;	1554	UNLOCK;
		1555
		1556	if (sv_hook)
		1557	{
		1558	dSP;
		1559
		1560	ENTER;
		1561	SAVETMPS;
		1562
		1563	PUSHMARK (SP);
		1564	PUTBACK;
		1565	call_sv (sv_hook, G_DISCARD);
		1566	SPAGAIN;
		1567
		1568	FREETMPS;
		1569	LEAVE;
		1570	}
		1571
		1572	if (xs_hook)
		1573	xs_hook ();
1383		1574
1384	return 1;	1575	return 1;
1385	}	1576	}
1386		1577
1387	static int	1578	static int
1388	api_is_ready (SV *coro_sv)	1579	api_is_ready (SV *coro_sv)
1389	{	1580	{
1390	dTHX;	1581	dTHX;
		1582
1391	return !!(SvSTATE (coro_sv)->flags & CF_READY);	1583	return !!(SvSTATE (coro_sv)->flags & CF_READY);
1392	}	1584	}
1393		1585
1394	static void	1586	INLINE void
1395	prepare_schedule (pTHX_ struct transfer_args *ta)	1587	prepare_schedule (pTHX_ struct transfer_args *ta)
1396	{	1588	{
1397	SV *prev_sv, *next_sv;	1589	SV *prev_sv, *next_sv;
1398		1590
1399	for (;;)	1591	for (;;)
…		…
1425	/* cannot transfer to destroyed coros, skip and look for next */	1617	/* cannot transfer to destroyed coros, skip and look for next */
1426	if (expect_false (ta->next->flags & CF_DESTROYED))	1618	if (expect_false (ta->next->flags & CF_DESTROYED))
1427	{	1619	{
1428	UNLOCK;	1620	UNLOCK;
1429	SvREFCNT_dec (next_sv);	1621	SvREFCNT_dec (next_sv);
1430	/* coro_nready is already taken care of by destroy */	1622	/* coro_nready has already been taken care of by destroy */
1431	continue;	1623	continue;
1432	}	1624	}
1433		1625
1434	--coro_nready;	1626	--coro_nready;
1435	UNLOCK;	1627	UNLOCK;
…		…
1438		1630
1439	/* free this only after the transfer */	1631	/* free this only after the transfer */
1440	prev_sv = SvRV (coro_current);	1632	prev_sv = SvRV (coro_current);
1441	ta->prev = SvSTATE (prev_sv);	1633	ta->prev = SvSTATE (prev_sv);
1442	TRANSFER_CHECK (*ta);	1634	TRANSFER_CHECK (*ta);
1443	assert (ta->next->flags & CF_READY);	1635	assert (("FATAL: next coroutine isn't marked as ready in Coro (please report)", ta->next->flags & CF_READY));
1444	ta->next->flags &= ~CF_READY;	1636	ta->next->flags &= ~CF_READY;
1445	SvRV_set (coro_current, next_sv);	1637	SvRV_set (coro_current, next_sv);
1446		1638
1447	LOCK;	1639	LOCK;
1448	free_coro_mortal (aTHX);	1640	free_coro_mortal (aTHX);
1449	coro_mortal = prev_sv;	1641	coro_mortal = prev_sv;
1450	UNLOCK;	1642	UNLOCK;
1451	}	1643	}
1452		1644
1453	static void	1645	INLINE void
1454	prepare_cede (pTHX_ struct transfer_args *ta)	1646	prepare_cede (pTHX_ struct transfer_args *ta)
1455	{	1647	{
1456	api_ready (coro_current);	1648	api_ready (coro_current);
1457	prepare_schedule (aTHX_ ta);	1649	prepare_schedule (aTHX_ ta);
1458	}	1650	}
…		…
1476	{	1668	{
1477	dTHX;	1669	dTHX;
1478	struct transfer_args ta;	1670	struct transfer_args ta;
1479		1671
1480	prepare_schedule (aTHX_ &ta);	1672	prepare_schedule (aTHX_ &ta);
1481	TRANSFER (ta);	1673	TRANSFER (ta, 1);
1482	}	1674	}
1483		1675
1484	static int	1676	static int
1485	api_cede (void)	1677	api_cede (void)
1486	{	1678	{
…		…
1489		1681
1490	prepare_cede (aTHX_ &ta);	1682	prepare_cede (aTHX_ &ta);
1491		1683
1492	if (expect_true (ta.prev != ta.next))	1684	if (expect_true (ta.prev != ta.next))
1493	{	1685	{
1494	TRANSFER (ta);	1686	TRANSFER (ta, 1);
1495	return 1;	1687	return 1;
1496	}	1688	}
1497	else	1689	else
1498	return 0;	1690	return 0;
1499	}	1691	}
…		…
1504	dTHX;	1696	dTHX;
1505	struct transfer_args ta;	1697	struct transfer_args ta;
1506		1698
1507	if (prepare_cede_notself (aTHX_ &ta))	1699	if (prepare_cede_notself (aTHX_ &ta))
1508	{	1700	{
1509	TRANSFER (ta);	1701	TRANSFER (ta, 1);
1510	return 1;	1702	return 1;
1511	}	1703	}
1512	else	1704	else
1513	return 0;	1705	return 0;
1514	}	1706	}
…		…
1520	struct coro *coro = SvSTATE (coro_sv);	1712	struct coro *coro = SvSTATE (coro_sv);
1521		1713
1522	if (flags & CC_TRACE)	1714	if (flags & CC_TRACE)
1523	{	1715	{
1524	if (!coro->cctx)	1716	if (!coro->cctx)
1525	coro->cctx = cctx_new ();	1717	coro->cctx = cctx_new_run ();
1526	else if (!(coro->cctx->flags & CC_TRACE))	1718	else if (!(coro->cctx->flags & CC_TRACE))
1527	croak ("cannot enable tracing on coroutine with custom stack");	1719	croak ("cannot enable tracing on coroutine with custom stack");
1528		1720
1529	coro->cctx->flags |= CC_NOREUSE | (flags & (CC_TRACE | CC_TRACE_ALL));	1721	coro->cctx->flags |= CC_NOREUSE | (flags & (CC_TRACE | CC_TRACE_ALL));
1530	}	1722	}
…		…
1537	else	1729	else
1538	coro->slot->runops = RUNOPS_DEFAULT;	1730	coro->slot->runops = RUNOPS_DEFAULT;
1539	}	1731	}
1540	}	1732	}
1541		1733
		1734	#if 0
		1735	static int
		1736	coro_gensub_free (pTHX_ SV *sv, MAGIC *mg)
		1737	{
		1738	AV *padlist;
		1739	AV *av = (AV *)mg->mg_obj;
		1740
		1741	abort ();
		1742
		1743	return 0;
		1744	}
		1745
		1746	static MGVTBL coro_gensub_vtbl = {
		1747	0, 0, 0, 0,
		1748	coro_gensub_free
		1749	};
		1750	#endif
		1751
		1752	/*****************************************************************************/
		1753	/* PerlIO::cede */
		1754
		1755	typedef struct
		1756	{
		1757	PerlIOBuf base;
		1758	NV next, every;
		1759	} PerlIOCede;
		1760
		1761	static IV
		1762	PerlIOCede_pushed (pTHX_ PerlIO *f, const char *mode, SV *arg, PerlIO_funcs *tab)
		1763	{
		1764	PerlIOCede *self = PerlIOSelf (f, PerlIOCede);
		1765
		1766	self->every = SvCUR (arg) ? SvNV (arg) : 0.01;
		1767	self->next = nvtime () + self->every;
		1768
		1769	return PerlIOBuf_pushed (aTHX_ f, mode, Nullsv, tab);
		1770	}
		1771
		1772	static SV *
		1773	PerlIOCede_getarg (pTHX_ PerlIO *f, CLONE_PARAMS *param, int flags)
		1774	{
		1775	PerlIOCede *self = PerlIOSelf (f, PerlIOCede);
		1776
		1777	return newSVnv (self->every);
		1778	}
		1779
		1780	static IV
		1781	PerlIOCede_flush (pTHX_ PerlIO *f)
		1782	{
		1783	PerlIOCede *self = PerlIOSelf (f, PerlIOCede);
		1784	double now = nvtime ();
		1785
		1786	if (now >= self->next)
		1787	{
		1788	api_cede ();
		1789	self->next = now + self->every;
		1790	}
		1791
		1792	return PerlIOBuf_flush (aTHX_ f);
		1793	}
		1794
		1795	static PerlIO_funcs PerlIO_cede =
		1796	{
		1797	sizeof(PerlIO_funcs),
		1798	"cede",
		1799	sizeof(PerlIOCede),
		1800	PERLIO_K_DESTRUCT | PERLIO_K_RAW,
		1801	PerlIOCede_pushed,
		1802	PerlIOBuf_popped,
		1803	PerlIOBuf_open,
		1804	PerlIOBase_binmode,
		1805	PerlIOCede_getarg,
		1806	PerlIOBase_fileno,
		1807	PerlIOBuf_dup,
		1808	PerlIOBuf_read,
		1809	PerlIOBuf_unread,
		1810	PerlIOBuf_write,
		1811	PerlIOBuf_seek,
		1812	PerlIOBuf_tell,
		1813	PerlIOBuf_close,
		1814	PerlIOCede_flush,
		1815	PerlIOBuf_fill,
		1816	PerlIOBase_eof,
		1817	PerlIOBase_error,
		1818	PerlIOBase_clearerr,
		1819	PerlIOBase_setlinebuf,
		1820	PerlIOBuf_get_base,
		1821	PerlIOBuf_bufsiz,
		1822	PerlIOBuf_get_ptr,
		1823	PerlIOBuf_get_cnt,
		1824	PerlIOBuf_set_ptrcnt,
		1825	};
		1826
		1827	/*****************************************************************************/
		1828
		1829	static const CV *ssl_cv; /* for quick consistency check */
		1830
		1831	static UNOP ssl_restore; /* restore stack as entersub did, for first-re-run */
		1832	static SV *ssl_arg0;
		1833	static SV *ssl_arg1;
		1834
		1835	/* this restores the stack in the case we patched the entersub, to */
		1836	/* recreate the stack frame as perl will on following calls */
		1837	/* since entersub cleared the stack */
		1838	static OP *
		1839	pp_restore (pTHX)
		1840	{
		1841	dSP;
		1842
		1843	PUSHMARK (SP);
		1844
		1845	EXTEND (SP, 3);
		1846	if (ssl_arg0) PUSHs (sv_2mortal (ssl_arg0)), ssl_arg0 = 0;
		1847	if (ssl_arg1) PUSHs (sv_2mortal (ssl_arg1)), ssl_arg1 = 0;
		1848	PUSHs ((SV *)CvGV (ssl_cv));
		1849
		1850	RETURNOP (ssl_restore.op_first);
		1851	}
		1852
		1853	/* declare prototype */
		1854	XS(XS_Coro__State__set_stacklevel);
		1855
		1856	static OP *
		1857	pp_set_stacklevel (pTHX)
		1858	{
		1859	dSP;
		1860	struct transfer_args ta;
		1861	SV **arg = PL_stack_base + TOPMARK + 1;
		1862	int items = SP - arg; /* args without function object */
		1863
		1864	/* do a quick consistency check on the "function" object, and if it isn't */
		1865	/* for us, divert to the real entersub */
		1866	if (SvTYPE (*sp) != SVt_PVGV || CvXSUB (GvCV (*sp)) != XS_Coro__State__set_stacklevel)
		1867	return PL_ppaddr[OP_ENTERSUB](aTHX);
		1868
		1869	/* pop args */
		1870	SP = PL_stack_base + POPMARK;
		1871
		1872	if (!(PL_op->op_flags & OPf_STACKED))
		1873	{
		1874	/* ampersand-form of call, use @_ instead of stack */
		1875	AV *av = GvAV (PL_defgv);
		1876	arg = AvARRAY (av);
		1877	items = AvFILLp (av) + 1;
		1878	}
		1879
		1880	PUTBACK;
		1881	switch (PL_op->op_private & 7)
		1882	{
		1883	case 0:
		1884	prepare_set_stacklevel (&ta, (struct coro_cctx *)SvIV (arg [0]));
		1885	break;
		1886
		1887	case 1:
		1888	if (items != 2)
		1889	croak ("Coro::State::transfer (prev, next) expects two arguments, not %d.", items);
		1890
		1891	prepare_transfer (aTHX_ &ta, arg [0], arg [1]);
		1892	break;
		1893
		1894	case 2:
		1895	prepare_schedule (aTHX_ &ta);
		1896	break;
		1897
		1898	case 3:
		1899	prepare_cede (aTHX_ &ta);
		1900	break;
		1901
		1902	case 4:
		1903	if (!prepare_cede_notself (aTHX_ &ta))
		1904	goto skip;
		1905
		1906	break;
		1907	}
		1908
		1909	TRANSFER (ta, 0);
		1910	SPAGAIN;
		1911
		1912	skip:
		1913	PUTBACK;
		1914	SSL_TAIL;
		1915	SPAGAIN;
		1916	RETURN;
		1917	}
		1918
1542	MODULE = Coro::State PACKAGE = Coro::State PREFIX = api_	1919	MODULE = Coro::State PACKAGE = Coro::State PREFIX = api_
1543		1920
1544	PROTOTYPES: DISABLE	1921	PROTOTYPES: DISABLE
1545
1546	BOOT:
1547	{
1548	#ifdef USE_ITHREADS
1549	MUTEX_INIT (&coro_mutex);
1550	#endif
1551	BOOT_PAGESIZE;
1552
1553	irsgv = gv_fetchpv ("/" , GV_ADD|GV_NOTQUAL, SVt_PV);
1554	stdoutgv = gv_fetchpv ("STDOUT", GV_ADD|GV_NOTQUAL, SVt_PVIO);
1555
1556	orig_sigelem_get = PL_vtbl_sigelem.svt_get;
1557	PL_vtbl_sigelem.svt_get = coro_sigelem_get;
1558	orig_sigelem_set = PL_vtbl_sigelem.svt_set;
1559	PL_vtbl_sigelem.svt_set = coro_sigelem_set;
1560
1561	hv_sig = coro_get_hv (aTHX_ "SIG", TRUE);
1562	rv_diehook = newRV_inc ((SV *)gv_fetchpv ("Coro::State::diehook" , 0, SVt_PVCV));
1563	rv_warnhook = newRV_inc ((SV *)gv_fetchpv ("Coro::State::warnhook", 0, SVt_PVCV));
1564
1565	coro_state_stash = gv_stashpv ("Coro::State", TRUE);
1566
1567	newCONSTSUB (coro_state_stash, "CC_TRACE" , newSViv (CC_TRACE));
1568	newCONSTSUB (coro_state_stash, "CC_TRACE_SUB" , newSViv (CC_TRACE_SUB));
1569	newCONSTSUB (coro_state_stash, "CC_TRACE_LINE", newSViv (CC_TRACE_LINE));
1570	newCONSTSUB (coro_state_stash, "CC_TRACE_ALL" , newSViv (CC_TRACE_ALL));
1571
1572	main_mainstack = PL_mainstack;
1573	main_top_env = PL_top_env;
1574
1575	while (main_top_env->je_prev)
1576	main_top_env = main_top_env->je_prev;
1577
1578	coroapi.ver = CORO_API_VERSION;
1579	coroapi.rev = CORO_API_REVISION;
1580	coroapi.transfer = api_transfer;
1581
1582	assert (("PRIO_NORMAL must be 0", !PRIO_NORMAL));
1583	}
1584
1585	SV *
1586	new (char *klass, ...)
1587	CODE:
1588	{
1589	struct coro *coro;
1590	MAGIC *mg;
1591	HV *hv;
1592	int i;
1593
1594	Newz (0, coro, 1, struct coro);
1595	coro->args = newAV ();
1596	coro->flags = CF_NEW;
1597
1598	if (coro_first) coro_first->prev = coro;
1599	coro->next = coro_first;
1600	coro_first = coro;
1601
1602	coro->hv = hv = newHV ();
1603	mg = sv_magicext ((SV *)hv, 0, CORO_MAGIC_type_state, &coro_state_vtbl, (char *)coro, 0);
1604	mg->mg_flags |= MGf_DUP;
1605	RETVAL = sv_bless (newRV_noinc ((SV *)hv), gv_stashpv (klass, 1));
1606
1607	av_extend (coro->args, items - 1);
1608	for (i = 1; i < items; i++)
1609	av_push (coro->args, newSVsv (ST (i)));
1610	}
1611	OUTPUT:
1612	RETVAL
1613		1922
1614	# these not obviously related functions are all rolled into the same xs	1923	# these not obviously related functions are all rolled into the same xs
1615	# function to increase chances that they all will call transfer with the same	1924	# function to increase chances that they all will call transfer with the same
1616	# stack offset	1925	# stack offset
1617	void	1926	void
…		…
1621	Coro::schedule = 2	1930	Coro::schedule = 2
1622	Coro::cede = 3	1931	Coro::cede = 3
1623	Coro::cede_notself = 4	1932	Coro::cede_notself = 4
1624	CODE:	1933	CODE:
1625	{	1934	{
1626	struct transfer_args ta;	1935	assert (("FATAL: ssl call recursion in Coro module (please report)", PL_op->op_ppaddr != pp_set_stacklevel));
1627		1936
1628	PUTBACK;	1937	/* we patch the op, and then re-run the whole call */
1629	switch (ix)	1938	/* we have to put some dummy argument on the stack for this to work */
		1939	/* TODO: walk back the opcode chain (but how?), nuke the pp_gv etc. */
		1940	ssl_restore.op_next = (OP *)&ssl_restore;
		1941	ssl_restore.op_type = OP_NULL;
		1942	ssl_restore.op_ppaddr = pp_restore;
		1943	ssl_restore.op_first = PL_op;
		1944
		1945	ssl_arg0 = items > 0 ? SvREFCNT_inc (ST (0)) : 0;
		1946	ssl_arg1 = items > 1 ? SvREFCNT_inc (ST (1)) : 0;
		1947
		1948	PL_op->op_ppaddr = pp_set_stacklevel;
		1949	PL_op->op_private = PL_op->op_private & ~7 | ix; /* we potentially share our private flags with entersub */
		1950
		1951	PL_op = (OP *)&ssl_restore;
		1952	}
		1953
		1954	BOOT:
		1955	{
		1956	#ifdef USE_ITHREADS
		1957	MUTEX_INIT (&coro_lock);
		1958	# if CORO_PTHREAD
		1959	coro_thx = PERL_GET_CONTEXT;
		1960	# endif
		1961	#endif
		1962	BOOT_PAGESIZE;
		1963
		1964	ssl_cv = get_cv ("Coro::State::_set_stacklevel", 0);
		1965
		1966	irsgv = gv_fetchpv ("/" , GV_ADD|GV_NOTQUAL, SVt_PV);
		1967	stdoutgv = gv_fetchpv ("STDOUT", GV_ADD|GV_NOTQUAL, SVt_PVIO);
		1968
		1969	orig_sigelem_get = PL_vtbl_sigelem.svt_get; PL_vtbl_sigelem.svt_get = coro_sigelem_get;
		1970	orig_sigelem_set = PL_vtbl_sigelem.svt_set; PL_vtbl_sigelem.svt_set = coro_sigelem_set;
		1971	orig_sigelem_clr = PL_vtbl_sigelem.svt_clear; PL_vtbl_sigelem.svt_clear = coro_sigelem_clr;
		1972
		1973	hv_sig = coro_get_hv (aTHX_ "SIG", TRUE);
		1974	rv_diehook = newRV_inc ((SV *)gv_fetchpv ("Coro::State::diehook" , 0, SVt_PVCV));
		1975	rv_warnhook = newRV_inc ((SV *)gv_fetchpv ("Coro::State::warnhook", 0, SVt_PVCV));
		1976
		1977	coro_state_stash = gv_stashpv ("Coro::State", TRUE);
		1978
		1979	newCONSTSUB (coro_state_stash, "CC_TRACE" , newSViv (CC_TRACE));
		1980	newCONSTSUB (coro_state_stash, "CC_TRACE_SUB" , newSViv (CC_TRACE_SUB));
		1981	newCONSTSUB (coro_state_stash, "CC_TRACE_LINE", newSViv (CC_TRACE_LINE));
		1982	newCONSTSUB (coro_state_stash, "CC_TRACE_ALL" , newSViv (CC_TRACE_ALL));
		1983
		1984	main_mainstack = PL_mainstack;
		1985	main_top_env = PL_top_env;
		1986
		1987	while (main_top_env->je_prev)
		1988	main_top_env = main_top_env->je_prev;
		1989
		1990	coroapi.ver = CORO_API_VERSION;
		1991	coroapi.rev = CORO_API_REVISION;
		1992	coroapi.transfer = api_transfer;
		1993
1630	{	1994	{
1631	case 0:	1995	SV **svp = hv_fetch (PL_modglobal, "Time::NVtime", 12, 0);
1632	ta.prev = (struct coro *)INT2PTR (coro_cctx *, SvIV (ST (0)));
1633	ta.next = 0;
1634	break;
1635		1996
1636	case 1:	1997	if (!svp) croak ("Time::HiRes is required");
1637	if (items != 2)	1998	if (!SvIOK (*svp)) croak ("Time::NVtime isn't a function pointer");
1638	croak ("Coro::State::transfer (prev,next) expects two arguments, not %d", items);
1639		1999
1640	prepare_transfer (aTHX_ &ta, ST (0), ST (1));	2000	nvtime = INT2PTR (double (*)(), SvIV (*svp));
1641	break;
1642
1643	case 2:
1644	prepare_schedule (aTHX_ &ta);
1645	break;
1646
1647	case 3:
1648	prepare_cede (aTHX_ &ta);
1649	break;
1650
1651	case 4:
1652	if (!prepare_cede_notself (aTHX_ &ta))
1653	XSRETURN_EMPTY;
1654
1655	break;
1656	}	2001	}
1657	SPAGAIN;
1658		2002
1659	BARRIER;	2003	assert (("PRIO_NORMAL must be 0", !PRIO_NORMAL));
1660	PUTBACK;
1661	TRANSFER (ta);
1662	SPAGAIN; /* might be the sp of a different coroutine now */
1663	/* be extra careful not to ever do anything after TRANSFER */
1664	}	2004	}
		2005
		2006	SV *
		2007	new (char *klass, ...)
		2008	CODE:
		2009	{
		2010	struct coro *coro;
		2011	MAGIC *mg;
		2012	HV *hv;
		2013	int i;
		2014
		2015	Newz (0, coro, 1, struct coro);
		2016	coro->args = newAV ();
		2017	coro->flags = CF_NEW;
		2018
		2019	if (coro_first) coro_first->prev = coro;
		2020	coro->next = coro_first;
		2021	coro_first = coro;
		2022
		2023	coro->hv = hv = newHV ();
		2024	mg = sv_magicext ((SV *)hv, 0, CORO_MAGIC_type_state, &coro_state_vtbl, (char *)coro, 0);
		2025	mg->mg_flags |= MGf_DUP;
		2026	RETVAL = sv_bless (newRV_noinc ((SV *)hv), gv_stashpv (klass, 1));
		2027
		2028	av_extend (coro->args, items - 1);
		2029	for (i = 1; i < items; i++)
		2030	av_push (coro->args, newSVsv (ST (i)));
		2031	}
		2032	OUTPUT:
		2033	RETVAL
1665		2034
1666	bool	2035	bool
1667	_destroy (SV *coro_sv)	2036	_destroy (SV *coro_sv)
1668	CODE:	2037	CODE:
1669	RETVAL = coro_state_destroy (aTHX_ SvSTATE (coro_sv));	2038	RETVAL = coro_state_destroy (aTHX_ SvSTATE (coro_sv));
1670	OUTPUT:	2039	OUTPUT:
1671	RETVAL	2040	RETVAL
1672		2041
1673	void	2042	void
1674	_exit (code)	2043	_exit (int code)
1675	int code
1676	PROTOTYPE: $	2044	PROTOTYPE: $
1677	CODE:	2045	CODE:
1678	_exit (code);	2046	_exit (code);
1679		2047
1680	int	2048	int
1681	cctx_stacksize (int new_stacksize = 0)	2049	cctx_stacksize (int new_stacksize = 0)
1682	CODE:	2050	CODE:
1683	RETVAL = coro_stacksize;	2051	RETVAL = cctx_stacksize;
1684	if (new_stacksize)	2052	if (new_stacksize)
		2053	{
1685	coro_stacksize = new_stacksize;	2054	cctx_stacksize = new_stacksize;
		2055	++cctx_gen;
		2056	}
		2057	OUTPUT:
		2058	RETVAL
		2059
		2060	int
		2061	cctx_max_idle (int max_idle = 0)
		2062	CODE:
		2063	RETVAL = cctx_max_idle;
		2064	if (max_idle > 1)
		2065	cctx_max_idle = max_idle;
1686	OUTPUT:	2066	OUTPUT:
1687	RETVAL	2067	RETVAL
1688		2068
1689	int	2069	int
1690	cctx_count ()	2070	cctx_count ()
…		…
1714	call (Coro::State coro, SV *coderef)	2094	call (Coro::State coro, SV *coderef)
1715	ALIAS:	2095	ALIAS:
1716	eval = 1	2096	eval = 1
1717	CODE:	2097	CODE:
1718	{	2098	{
1719	if (coro->mainstack)	2099	if (coro->mainstack && ((coro->flags & CF_RUNNING) || coro->slot))
1720	{	2100	{
1721	struct coro temp;	2101	struct coro temp;
1722		2102
1723	if (!(coro->flags & CF_RUNNING))	2103	if (!(coro->flags & CF_RUNNING))
1724	{	2104	{
…		…
1768	RETVAL = boolSV (coro->flags & ix);	2148	RETVAL = boolSV (coro->flags & ix);
1769	OUTPUT:	2149	OUTPUT:
1770	RETVAL	2150	RETVAL
1771		2151
1772	void	2152	void
		2153	throw (Coro::State self, SV *throw = &PL_sv_undef)
		2154	PROTOTYPE: $;$
		2155	CODE:
		2156	SvREFCNT_dec (self->throw);
		2157	self->throw = SvOK (throw) ? newSVsv (throw) : 0;
		2158
		2159	void
1773	api_trace (SV *coro, int flags = CC_TRACE | CC_TRACE_SUB)	2160	api_trace (SV *coro, int flags = CC_TRACE | CC_TRACE_SUB)
1774		2161
1775	SV *	2162	SV *
1776	has_stack (Coro::State coro)	2163	has_cctx (Coro::State coro)
1777	PROTOTYPE: $	2164	PROTOTYPE: $
1778	CODE:	2165	CODE:
1779	RETVAL = boolSV (!!coro->cctx);	2166	RETVAL = boolSV (!!coro->cctx);
1780	OUTPUT:	2167	OUTPUT:
1781	RETVAL	2168	RETVAL
…		…
1786	CODE:	2173	CODE:
1787	RETVAL = (coro->cctx ? coro->cctx->flags : 0) & CC_TRACE_ALL;	2174	RETVAL = (coro->cctx ? coro->cctx->flags : 0) & CC_TRACE_ALL;
1788	OUTPUT:	2175	OUTPUT:
1789	RETVAL	2176	RETVAL
1790		2177
1791	IV	2178	UV
1792	rss (Coro::State coro)	2179	rss (Coro::State coro)
1793	PROTOTYPE: $	2180	PROTOTYPE: $
1794	ALIAS:	2181	ALIAS:
1795	usecount = 1	2182	usecount = 1
1796	CODE:	2183	CODE:
…		…
1800	case 1: RETVAL = coro->usecount; break;	2187	case 1: RETVAL = coro->usecount; break;
1801	}	2188	}
1802	OUTPUT:	2189	OUTPUT:
1803	RETVAL	2190	RETVAL
1804		2191
		2192	void
		2193	force_cctx ()
		2194	CODE:
		2195	struct coro *coro = SvSTATE (coro_current);
		2196	coro->cctx->idle_sp = 0;
		2197
		2198	void
		2199	swap_defsv (Coro::State self)
		2200	PROTOTYPE: $
		2201	ALIAS:
		2202	swap_defav = 1
		2203	CODE:
		2204	if (!self->slot)
		2205	croak ("cannot swap state with coroutine that has no saved state");
		2206	else
		2207	{
		2208	SV **src = ix ? (SV **)&GvAV (PL_defgv) : &GvSV (PL_defgv);
		2209	SV **dst = ix ? (SV **)&self->slot->defav : (SV **)&self->slot->defsv;
		2210
		2211	SV *tmp = *src; *src = *dst; *dst = tmp;
		2212	}
1805		2213
1806	MODULE = Coro::State PACKAGE = Coro	2214	MODULE = Coro::State PACKAGE = Coro
1807		2215
1808	BOOT:	2216	BOOT:
1809	{	2217	{
1810	int i;	2218	int i;
1811		2219
		2220	av_async_pool = coro_get_av (aTHX_ "Coro::async_pool", TRUE);
1812	sv_pool_rss = coro_get_sv (aTHX_ "Coro::POOL_RSS" , TRUE);	2221	sv_pool_rss = coro_get_sv (aTHX_ "Coro::POOL_RSS" , TRUE);
1813	sv_pool_size = coro_get_sv (aTHX_ "Coro::POOL_SIZE" , TRUE);	2222	sv_pool_size = coro_get_sv (aTHX_ "Coro::POOL_SIZE" , TRUE);
1814	av_async_pool = coro_get_av (aTHX_ "Coro::async_pool", TRUE);
1815		2223
1816	coro_current = coro_get_sv (aTHX_ "Coro::current", FALSE);	2224	coro_current = coro_get_sv (aTHX_ "Coro::current", FALSE);
1817	SvREADONLY_on (coro_current);	2225	SvREADONLY_on (coro_current);
1818		2226
1819	coro_stash = gv_stashpv ("Coro", TRUE);	2227	coro_stash = gv_stashpv ("Coro", TRUE);
…		…
1827		2235
1828	for (i = PRIO_MAX - PRIO_MIN + 1; i--; )	2236	for (i = PRIO_MAX - PRIO_MIN + 1; i--; )
1829	coro_ready[i] = newAV ();	2237	coro_ready[i] = newAV ();
1830		2238
1831	{	2239	{
1832	SV *sv = perl_get_sv ("Coro::API", TRUE);	2240	SV *sv = coro_get_sv (aTHX_ "Coro::API", TRUE);
1833	perl_get_sv ("Coro::API", TRUE); /* silence 5.10 warning */
1834		2241
1835	coroapi.schedule = api_schedule;	2242	coroapi.schedule = api_schedule;
1836	coroapi.cede = api_cede;	2243	coroapi.cede = api_cede;
1837	coroapi.cede_notself = api_cede_notself;	2244	coroapi.cede_notself = api_cede_notself;
1838	coroapi.ready = api_ready;	2245	coroapi.ready = api_ready;
…		…
1849	void	2256	void
1850	_set_current (SV *current)	2257	_set_current (SV *current)
1851	PROTOTYPE: $	2258	PROTOTYPE: $
1852	CODE:	2259	CODE:
1853	SvREFCNT_dec (SvRV (coro_current));	2260	SvREFCNT_dec (SvRV (coro_current));
1854	SvRV_set (coro_current, SvREFCNT_inc (SvRV (current)));	2261	SvRV_set (coro_current, SvREFCNT_inc_NN (SvRV (current)));
		2262
		2263	void
		2264	_set_readyhook (SV *hook)
		2265	PROTOTYPE: $
		2266	CODE:
		2267	LOCK;
		2268	SvREFCNT_dec (coro_readyhook);
		2269	coro_readyhook = SvOK (hook) ? newSVsv (hook) : 0;
		2270	UNLOCK;
1855		2271
1856	int	2272	int
1857	prio (Coro::State coro, int newprio = 0)	2273	prio (Coro::State coro, int newprio = 0)
1858	ALIAS:	2274	ALIAS:
1859	nice = 1	2275	nice = 1
…		…
1889	CODE:	2305	CODE:
1890	RETVAL = coro_nready;	2306	RETVAL = coro_nready;
1891	OUTPUT:	2307	OUTPUT:
1892	RETVAL	2308	RETVAL
1893		2309
1894	void
1895	throw (Coro::State self, SV *throw = &PL_sv_undef)
1896	PROTOTYPE: $;$
1897	CODE:
1898	SvREFCNT_dec (self->throw);
1899	self->throw = SvOK (throw) ? newSVsv (throw) : 0;
1900
1901	# for async_pool speedup	2310	# for async_pool speedup
1902	void	2311	void
1903	_pool_1 (SV *cb)	2312	_pool_1 (SV *cb)
1904	CODE:	2313	CODE:
1905	{	2314	{
…		…
1910	AV *invoke_av;	2319	AV *invoke_av;
1911	int i, len;	2320	int i, len;
1912		2321
1913	if (!invoke)	2322	if (!invoke)
1914	{	2323	{
1915	SvREFCNT_dec (PL_diehook); PL_diehook = 0;	2324	SV *old = PL_diehook;
		2325	PL_diehook = 0;
		2326	SvREFCNT_dec (old);
1916	croak ("\3async_pool terminate\2\n");	2327	croak ("\3async_pool terminate\2\n");
1917	}	2328	}
1918		2329
1919	SvREFCNT_dec (coro->saved_deffh);	2330	SvREFCNT_dec (coro->saved_deffh);
1920	coro->saved_deffh = SvREFCNT_inc ((SV *)PL_defoutgv);	2331	coro->saved_deffh = SvREFCNT_inc_NN ((SV *)PL_defoutgv);
1921		2332
1922	hv_store (hv, "desc", sizeof ("desc") - 1,	2333	hv_store (hv, "desc", sizeof ("desc") - 1,
1923	newSVpvn ("[async_pool]", sizeof ("[async_pool]") - 1), 0);	2334	newSVpvn ("[async_pool]", sizeof ("[async_pool]") - 1), 0);
1924		2335
1925	invoke_av = (AV *)SvRV (invoke);	2336	invoke_av = (AV *)SvRV (invoke);
…		…
1929		2340
1930	if (len > 0)	2341	if (len > 0)
1931	{	2342	{
1932	av_fill (defav, len - 1);	2343	av_fill (defav, len - 1);
1933	for (i = 0; i < len; ++i)	2344	for (i = 0; i < len; ++i)
1934	av_store (defav, i, SvREFCNT_inc (AvARRAY (invoke_av)[i + 1]));	2345	av_store (defav, i, SvREFCNT_inc_NN (AvARRAY (invoke_av)[i + 1]));
1935	}	2346	}
1936		2347
1937	SvREFCNT_dec (invoke);	2348	SvREFCNT_dec (invoke);
1938	}	2349	}
1939		2350
…		…
1946	sv_setsv (cb, &PL_sv_undef);	2357	sv_setsv (cb, &PL_sv_undef);
1947		2358
1948	SvREFCNT_dec ((SV *)PL_defoutgv); PL_defoutgv = (GV *)coro->saved_deffh;	2359	SvREFCNT_dec ((SV *)PL_defoutgv); PL_defoutgv = (GV *)coro->saved_deffh;
1949	coro->saved_deffh = 0;	2360	coro->saved_deffh = 0;
1950		2361
1951	if (coro_rss (aTHX_ coro) > SvIV (sv_pool_rss)	2362	if (coro_rss (aTHX_ coro) > SvUV (sv_pool_rss)
1952	|| av_len (av_async_pool) + 1 >= SvIV (sv_pool_size))	2363	|| av_len (av_async_pool) + 1 >= SvIV (sv_pool_size))
1953	{	2364	{
1954	SvREFCNT_dec (PL_diehook); PL_diehook = 0;	2365	SV *old = PL_diehook;
		2366	PL_diehook = 0;
		2367	SvREFCNT_dec (old);
1955	croak ("\3async_pool terminate\2\n");	2368	croak ("\3async_pool terminate\2\n");
1956	}	2369	}
1957		2370
1958	av_clear (GvAV (PL_defgv));	2371	av_clear (GvAV (PL_defgv));
1959	hv_store ((HV *)SvRV (coro_current), "desc", sizeof ("desc") - 1,	2372	hv_store ((HV *)SvRV (coro_current), "desc", sizeof ("desc") - 1,
…		…
1965	api_trace (coro_current, 0);	2378	api_trace (coro_current, 0);
1966		2379
1967	av_push (av_async_pool, newSVsv (coro_current));	2380	av_push (av_async_pool, newSVsv (coro_current));
1968	}	2381	}
1969		2382
		2383	#if 0
		2384
		2385	void
		2386	_generator_call (...)
		2387	PROTOTYPE: @
		2388	PPCODE:
		2389	fprintf (stderr, "call %p\n", CvXSUBANY(cv).any_ptr);
		2390	xxxx
		2391	abort ();
		2392
		2393	SV *
		2394	gensub (SV *sub, ...)
		2395	PROTOTYPE: &;@
		2396	CODE:
		2397	{
		2398	struct coro *coro;
		2399	MAGIC *mg;
		2400	CV *xcv;
		2401	CV *ncv = (CV *)newSV_type (SVt_PVCV);
		2402	int i;
		2403
		2404	CvGV (ncv) = CvGV (cv);
		2405	CvFILE (ncv) = CvFILE (cv);
		2406
		2407	Newz (0, coro, 1, struct coro);
		2408	coro->args = newAV ();
		2409	coro->flags = CF_NEW;
		2410
		2411	av_extend (coro->args, items - 1);
		2412	for (i = 1; i < items; i++)
		2413	av_push (coro->args, newSVsv (ST (i)));
		2414
		2415	CvISXSUB_on (ncv);
		2416	CvXSUBANY (ncv).any_ptr = (void *)coro;
		2417
		2418	xcv = GvCV (gv_fetchpv ("Coro::_generator_call", 0, SVt_PVCV));
		2419
		2420	CvXSUB (ncv) = CvXSUB (xcv);
		2421	CvANON_on (ncv);
		2422
		2423	mg = sv_magicext ((SV *)ncv, 0, CORO_MAGIC_type_state, &coro_gensub_vtbl, (char *)coro, 0);
		2424	RETVAL = newRV_noinc ((SV *)ncv);
		2425	}
		2426	OUTPUT:
		2427	RETVAL
		2428
		2429	#endif
		2430
1970		2431
1971	MODULE = Coro::State PACKAGE = Coro::AIO	2432	MODULE = Coro::State PACKAGE = Coro::AIO
1972		2433
1973	SV *	2434	void
1974	_get_state ()	2435	_get_state (SV *self)
1975	CODE:	2436	PPCODE:
1976	{	2437	{
1977	struct io_state *data;	2438	AV *defav = GvAV (PL_defgv);
1978		2439	AV *av = newAV ();
		2440	int i;
1979	RETVAL = newSV (sizeof (struct io_state));	2441	SV *data_sv = newSV (sizeof (struct io_state));
1980	data = (struct io_state *)SvPVX (RETVAL);	2442	struct io_state *data = (struct io_state *)SvPVX (data_sv);
1981	SvCUR_set (RETVAL, sizeof (struct io_state));	2443	SvCUR_set (data_sv, sizeof (struct io_state));
1982	SvPOK_only (RETVAL);	2444	SvPOK_only (data_sv);
1983		2445
1984	data->errorno = errno;	2446	data->errorno = errno;
1985	data->laststype = PL_laststype;	2447	data->laststype = PL_laststype;
1986	data->laststatval = PL_laststatval;	2448	data->laststatval = PL_laststatval;
1987	data->statcache = PL_statcache;	2449	data->statcache = PL_statcache;
		2450
		2451	av_extend (av, AvFILLp (defav) + 1 + 1);
		2452
		2453	for (i = 0; i <= AvFILLp (defav); ++i)
		2454	av_push (av, SvREFCNT_inc_NN (AvARRAY (defav)[i]));
		2455
		2456	av_push (av, data_sv);
		2457
		2458	XPUSHs (sv_2mortal (newRV_noinc ((SV *)av)));
		2459
		2460	api_ready (self);
1988	}	2461	}
1989	OUTPUT:
1990	RETVAL
1991		2462
1992	void	2463	void
1993	_set_state (char *data_)	2464	_set_state (SV *state)
1994	PROTOTYPE: $	2465	PROTOTYPE: $
1995	CODE:	2466	PPCODE:
1996	{	2467	{
1997	struct io_state *data = (void *)data_;	2468	AV *av = (AV *)SvRV (state);
		2469	struct io_state *data = (struct io_state *)SvPVX (AvARRAY (av)[AvFILLp (av)]);
		2470	int i;
1998		2471
1999	errno = data->errorno;	2472	errno = data->errorno;
2000	PL_laststype = data->laststype;	2473	PL_laststype = data->laststype;
2001	PL_laststatval = data->laststatval;	2474	PL_laststatval = data->laststatval;
2002	PL_statcache = data->statcache;	2475	PL_statcache = data->statcache;
2003	}
2004		2476
		2477	EXTEND (SP, AvFILLp (av));
		2478	for (i = 0; i < AvFILLp (av); ++i)
		2479	PUSHs (sv_2mortal (SvREFCNT_inc_NN (AvARRAY (av)[i])));
		2480	}
		2481
		2482
		2483	MODULE = Coro::State PACKAGE = Coro::AnyEvent
		2484
		2485	BOOT:
		2486	sv_activity = coro_get_sv (aTHX_ "Coro::AnyEvent::ACTIVITY", TRUE);
		2487
		2488	SV *
		2489	_schedule (...)
		2490	PROTOTYPE: @
		2491	CODE:
		2492	{
		2493	static int incede;
		2494
		2495	api_cede_notself ();
		2496
		2497	++incede;
		2498	while (coro_nready >= incede && api_cede ())
		2499	;
		2500
		2501	sv_setsv (sv_activity, &PL_sv_undef);
		2502	if (coro_nready >= incede)
		2503	{
		2504	PUSHMARK (SP);
		2505	PUTBACK;
		2506	call_pv ("Coro::AnyEvent::_activity", G_DISCARD | G_EVAL);
		2507	SPAGAIN;
		2508	}
		2509
		2510	--incede;
		2511	}
		2512
		2513
		2514	MODULE = Coro::State PACKAGE = PerlIO::cede
		2515
		2516	BOOT:
		2517	PerlIO_define_layer (aTHX_ &PerlIO_cede);
		2518

Diff Legend

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