[ViewVC] Diff of: cvs/Coro/Coro/State.xs

Comparing Coro/Coro/State.xs (file contents):
Revision 1.260 by root, Mon Nov 10 04:37:23 2008 UTC vs.
Revision 1.278 by root, Sun Nov 16 07:09:28 2008 UTC

…		…
46	# define BOOT_PAGESIZE (void)0	46	# define BOOT_PAGESIZE (void)0
47	#endif	47	#endif
48		48
49	#if CORO_USE_VALGRIND	49	#if CORO_USE_VALGRIND
50	# include <valgrind/valgrind.h>	50	# include <valgrind/valgrind.h>
51	# define REGISTER_STACK(cctx,start,end) (cctx)->valgrind_id = VALGRIND_STACK_REGISTER ((start), (end))
52	#else
53	# define REGISTER_STACK(cctx,start,end)
54	#endif	51	#endif
55		52
56	/* the maximum number of idle cctx that will be pooled */	53	/* the maximum number of idle cctx that will be pooled */
57	static int cctx_max_idle = 4;	54	static int cctx_max_idle = 4;
58		55
…		…
119	# define CORO_PREFER_PERL_FUNCTIONS 0	116	# define CORO_PREFER_PERL_FUNCTIONS 0
120	#endif	117	#endif
121		118
122	/* The next macros try to return the current stack pointer, in an as	119	/* The next macros try to return the current stack pointer, in an as
123	* portable way as possible. */	120	* portable way as possible. */
124	#define dSTACKLEVEL volatile char stacklevel	121	#if __GNUC__ >= 4
125	#define STACKLEVEL ((void *)&stacklevel)	122	# define dSTACKLEVEL void *stacklevel = __builtin_frame_address (0)
		123	#else
		124	# define dSTACKLEVEL volatile void stacklevel = (volatile void )&stacklevel
		125	#endif
126		126
127	#define IN_DESTRUCT (PL_main_cv == Nullcv)	127	#define IN_DESTRUCT (PL_main_cv == Nullcv)
128		128
129	#if __GNUC__ >= 3	129	#if __GNUC__ >= 3
130	# define attribute(x) __attribute__(x)	130	# define attribute(x) __attribute__(x)
131	# define BARRIER __asm__ __volatile__ ("" : : : "memory")
132	# define expect(expr,value) __builtin_expect ((expr),(value))	131	# define expect(expr,value) __builtin_expect ((expr),(value))
		132	# define INLINE static inline
133	#else	133	#else
134	# define attribute(x)	134	# define attribute(x)
135	# define BARRIER
136	# define expect(expr,value) (expr)	135	# define expect(expr,value) (expr)
		136	# define INLINE static
137	#endif	137	#endif
138		138
139	#define expect_false(expr) expect ((expr) != 0, 0)	139	#define expect_false(expr) expect ((expr) != 0, 0)
140	#define expect_true(expr) expect ((expr) != 0, 1)	140	#define expect_true(expr) expect ((expr) != 0, 1)
141		141
…		…
183	static JMPENV *main_top_env;	183	static JMPENV *main_top_env;
184	static HV coro_state_stash, coro_stash;	184	static HV coro_state_stash, coro_stash;
185	static volatile SV coro_mortal; / will be freed/thrown after next transfer */	185	static volatile SV coro_mortal; / will be freed/thrown after next transfer */
186	static volatile struct coro *transfer_next;	186	static volatile struct coro *transfer_next;
187		187
188	struct transfer_args
189	{
190	struct coro prev, next;
191	};
192
193	static GV irsgv; / $/ */	188	static GV irsgv; / $/ */
194	static GV stdoutgv; / STDOUT /	189	static GV stdoutgv; / STDOUT /
195	static SV *rv_diehook;	190	static SV *rv_diehook;
196	static SV *rv_warnhook;	191	static SV *rv_warnhook;
197	static HV hv_sig; / %SIG */	192	static HV hv_sig; / %SIG */
…		…
215	CC_TRACE_LINE = 0x10, /* trace each statement */	210	CC_TRACE_LINE = 0x10, /* trace each statement */
216	CC_TRACE_ALL = CC_TRACE_SUB \| CC_TRACE_LINE,	211	CC_TRACE_ALL = CC_TRACE_SUB \| CC_TRACE_LINE,
217	};	212	};
218		213
219	/* this is a structure representing a c-level coroutine */	214	/* this is a structure representing a c-level coroutine */
220	typedef struct coro_cctx {	215	typedef struct coro_cctx
		216	{
221	struct coro_cctx *next;	217	struct coro_cctx *next;
222		218
223	/* the stack */	219	/* the stack */
224	void *sptr;	220	void *sptr;
225	size_t ssize;	221	size_t ssize;
…		…
243	CF_NEW = 0x0004, /* has never been switched to */	239	CF_NEW = 0x0004, /* has never been switched to */
244	CF_DESTROYED = 0x0008, /* coroutine data has been freed */	240	CF_DESTROYED = 0x0008, /* coroutine data has been freed */
245	};	241	};
246		242
247	/* the structure where most of the perl state is stored, overlaid on the cxstack */	243	/* the structure where most of the perl state is stored, overlaid on the cxstack */
248	typedef struct {	244	typedef struct
		245	{
249	SV *defsv;	246	SV *defsv;
250	AV *defav;	247	AV *defav;
251	SV *errsv;	248	SV *errsv;
252	SV *irsgv;	249	SV *irsgv;
253	#define VAR(name,type) type name;	250	#define VAR(name,type) type name;
…		…
257		254
258	#define SLOT_COUNT ((sizeof (perl_slots) + sizeof (PERL_CONTEXT) - 1) / sizeof (PERL_CONTEXT))	255	#define SLOT_COUNT ((sizeof (perl_slots) + sizeof (PERL_CONTEXT) - 1) / sizeof (PERL_CONTEXT))
259		256
260	/* this is a structure representing a perl-level coroutine */	257	/* this is a structure representing a perl-level coroutine */
261	struct coro {	258	struct coro {
262	/* the c coroutine allocated to this perl coroutine, if any */	259	/* the C coroutine allocated to this perl coroutine, if any */
263	coro_cctx *cctx;	260	coro_cctx *cctx;
264		261
265	/* process data */	262	/* process data */
		263	struct CoroSLF slf_frame; /* saved slf frame */
266	AV *mainstack;	264	AV *mainstack;
267	perl_slots slot; / basically the saved sp */	265	perl_slots slot; / basically the saved sp */
268		266
269	AV args; / data associated with this coroutine (initial args) */	267	AV args; / data associated with this coroutine (initial args) */
270	int refcnt; /* coroutines are refcounted, yes */	268	int refcnt; /* coroutines are refcounted, yes */
…		…
285	struct coro next, prev;	283	struct coro next, prev;
286	};	284	};
287		285
288	typedef struct coro *Coro__State;	286	typedef struct coro *Coro__State;
289	typedef struct coro *Coro__State_or_hashref;	287	typedef struct coro *Coro__State_or_hashref;
		288
		289	static struct CoroSLF slf_frame; /* the current slf frame */
290		290
291	/ Coro ******************************************************************/	291	/ Coro ******************************************************************/
292		292
293	#define PRIO_MAX 3	293	#define PRIO_MAX 3
294	#define PRIO_HIGH 1	294	#define PRIO_HIGH 1
…		…
299		299
300	/* for Coro.pm */	300	/* for Coro.pm */
301	static SV *coro_current;	301	static SV *coro_current;
302	static SV *coro_readyhook;	302	static SV *coro_readyhook;
303	static AV *coro_ready [PRIO_MAX - PRIO_MIN + 1];	303	static AV *coro_ready [PRIO_MAX - PRIO_MIN + 1];
304	static int coro_nready;
305	static struct coro *coro_first;	304	static struct coro *coro_first;
		305	#define coro_nready coroapi.nready
306		306
307	/ lowlevel stuff ********************************************************/	307	/ lowlevel stuff ********************************************************/
308		308
309	static SV *	309	static SV *
310	coro_get_sv (pTHX_ const char *name, int create)	310	coro_get_sv (pTHX_ const char *name, int create)
…		…
413	: 0	413	: 0
414		414
415	#define CORO_MAGIC_cv(cv) CORO_MAGIC (((SV *)(cv)), CORO_MAGIC_type_cv)	415	#define CORO_MAGIC_cv(cv) CORO_MAGIC (((SV *)(cv)), CORO_MAGIC_type_cv)
416	#define CORO_MAGIC_state(sv) CORO_MAGIC (((SV *)(sv)), CORO_MAGIC_type_state)	416	#define CORO_MAGIC_state(sv) CORO_MAGIC (((SV *)(sv)), CORO_MAGIC_type_state)
417		417
418	static struct coro *	418	INLINE struct coro *
419	SvSTATE_ (pTHX_ SV *coro)	419	SvSTATE_ (pTHX_ SV *coro)
420	{	420	{
421	HV *stash;	421	HV *stash;
422	MAGIC *mg;	422	MAGIC *mg;
423		423
…		…
514	CvPADLIST (cv) = (AV *)POPs;	514	CvPADLIST (cv) = (AV *)POPs;
515	}	515	}
516		516
517	PUTBACK;	517	PUTBACK;
518	}	518	}
		519
		520	slf_frame = c->slf_frame;
519	}	521	}
520		522
521	static void	523	static void
522	save_perl (pTHX_ Coro__State c)	524	save_perl (pTHX_ Coro__State c)
523	{	525	{
		526	c->slf_frame = slf_frame;
		527
524	{	528	{
525	dSP;	529	dSP;
526	I32 cxix = cxstack_ix;	530	I32 cxix = cxstack_ix;
527	PERL_CONTEXT *ccstk = cxstack;	531	PERL_CONTEXT *ccstk = cxstack;
528	PERL_SI *top_si = PL_curstackinfo;	532	PERL_SI *top_si = PL_curstackinfo;
…		…
595	#undef VAR	599	#undef VAR
596	}	600	}
597	}	601	}
598		602
599	/*	603	/*
600	* allocate various perl stacks. This is an exact copy	604	* allocate various perl stacks. This is almost an exact copy
601	* of perl.c:init_stacks, except that it uses less memory	605	* of perl.c:init_stacks, except that it uses less memory
602	* on the (sometimes correct) assumption that coroutines do	606	* on the (sometimes correct) assumption that coroutines do
603	* not usually need a lot of stackspace.	607	* not usually need a lot of stackspace.
604	*/	608	*/
605	#if CORO_PREFER_PERL_FUNCTIONS	609	#if CORO_PREFER_PERL_FUNCTIONS
…		…
807		811
808	return orig_sigelem_set ? orig_sigelem_set (aTHX_ sv, mg) : 0;	812	return orig_sigelem_set ? orig_sigelem_set (aTHX_ sv, mg) : 0;
809	}	813	}
810		814
811	static void	815	static void
		816	prepare_nop (pTHX_ struct coro_transfer_args *ta)
		817	{
		818	/* kind of mega-hacky, but works */
		819	ta->next = ta->prev = (struct coro *)ta;
		820	}
		821
		822	static int
		823	slf_check_nop (pTHX_ struct CoroSLF *frame)
		824	{
		825	return 0;
		826	}
		827
		828	static void
812	coro_setup (pTHX_ struct coro *coro)	829	coro_setup (pTHX_ struct coro *coro)
813	{	830	{
814	/*	831	/*
815	* emulate part of the perl startup here.	832	* emulate part of the perl startup here.
816	*/	833	*/
…		…
855	PL_op = PL_ppaddr[OP_ENTERSUB](aTHX);	872	PL_op = PL_ppaddr[OP_ENTERSUB](aTHX);
856	SPAGAIN;	873	SPAGAIN;
857	}	874	}
858		875
859	/* this newly created coroutine might be run on an existing cctx which most	876	/* this newly created coroutine might be run on an existing cctx which most
860	* likely was suspended in set_stacklevel, called from entersub.	877	* likely was suspended in pp_slf, so we have to emulate entering pp_slf here.
861	* set_stacklevl doesn't do anything on return, but entersub does LEAVE,
862	* so we ENTER here for symmetry.
863	*/	878	*/
864	ENTER;	879	slf_frame.prepare = prepare_nop; /* provide a nop function for an eventual pp_slf */
		880	slf_frame.check = slf_check_nop; /* signal pp_slf to not repeat */
865	}	881	}
866		882
867	static void	883	static void
868	coro_destruct (pTHX_ struct coro *coro)	884	coro_destruct (pTHX_ struct coro *coro)
869	{	885	{
…		…
898	SvREFCNT_dec (coro->throw);	914	SvREFCNT_dec (coro->throw);
899		915
900	coro_destruct_stacks (aTHX);	916	coro_destruct_stacks (aTHX);
901	}	917	}
902		918
903	static void	919	INLINE void
904	free_coro_mortal (pTHX)	920	free_coro_mortal (pTHX)
905	{	921	{
906	if (expect_true (coro_mortal))	922	if (expect_true (coro_mortal))
907	{	923	{
908	SvREFCNT_dec (coro_mortal);	924	SvREFCNT_dec (coro_mortal);
…		…
1033	TAINT_NOT;	1049	TAINT_NOT;
1034	return 0;	1050	return 0;
1035	}	1051	}
1036		1052
1037	static void	1053	static void
1038	prepare_set_stacklevel (struct transfer_args ta, struct coro_cctx cctx)	1054	prepare_set_stacklevel (struct coro_transfer_args ta, struct coro_cctx cctx)
1039	{	1055	{
1040	ta->prev = (struct coro *)cctx;	1056	ta->prev = (struct coro *)cctx;
1041	ta->next = 0;	1057	ta->next = 0;
1042	}	1058	}
1043		1059
…		…
1068	PL_op = (OP *)&myop;	1084	PL_op = (OP *)&myop;
1069	PL_op = PL_ppaddr[OP_ENTERSUB](aTHX);	1085	PL_op = PL_ppaddr[OP_ENTERSUB](aTHX);
1070	SPAGAIN;	1086	SPAGAIN;
1071	}	1087	}
1072		1088
1073	/* the tail of transfer: execute stuff we can onyl do afetr a transfer */	1089	/* the tail of transfer: execute stuff we can only do after a transfer */
1074	static void	1090	INLINE void
1075	transfer_tail (void)	1091	transfer_tail (pTHX)
1076	{	1092	{
1077	struct coro next = (struct coro )transfer_next;	1093	struct coro next = (struct coro )transfer_next;
1078	transfer_next = 0; //D for temporary assertion in transfer	1094	assert (!(transfer_next = 0)); /* just used for the side effect when asserts are enabled */
1079	assert (("FATAL ERROR: internal error 1067 in Coro module, please report", next));//D	1095	assert (("FATAL: next coroutine was zero in transfer_tail (please report)", next));
1080		1096
1081	free_coro_mortal (aTHX);	1097	free_coro_mortal (aTHX);
1082	UNLOCK;	1098	UNLOCK;
1083		1099
1084	if (expect_false (next->throw))	1100	if (expect_false (next->throw))
…		…
1103	# endif	1119	# endif
1104	#endif	1120	#endif
1105	{	1121	{
1106	dTHX;	1122	dTHX;
1107		1123
1108	/* entersub called ENTER, but we never 'returned', undo that here */	1124	/* normally we would need to skip the entersub here */
1109	LEAVE;	1125	/* not doing so will re-execute it, which is exactly what we want */
1110
1111	/* we now skip the entersub that did lead to transfer() */
1112	PL_op = PL_op->op_next;	1126	/* PL_nop = PL_nop->op_next */
1113		1127
1114	/* inject a fake subroutine call to cctx_init */	1128	/* inject a fake subroutine call to cctx_init */
1115	cctx_prepare (aTHX_ (coro_cctx *)arg);	1129	cctx_prepare (aTHX_ (coro_cctx *)arg);
1116		1130
1117	/* cctx_run is the alternative tail of transfer() */	1131	/* cctx_run is the alternative tail of transfer() */
		1132	/* TODO: throwing an exception here might be deadly, VERIFY */
1118	transfer_tail ();	1133	transfer_tail (aTHX);
1119		1134
1120	/* somebody or something will hit me for both perl_run and PL_restartop */	1135	/* somebody or something will hit me for both perl_run and PL_restartop */
1121	PL_restartop = PL_op;	1136	PL_restartop = PL_op;
1122	perl_run (PL_curinterp);	1137	perl_run (PL_curinterp);
1123		1138
…		…
1173	/* mmap supposedly does allocate-on-write for us */	1188	/* mmap supposedly does allocate-on-write for us */
1174	cctx->sptr = mmap (0, cctx->ssize, PROT_EXEC\|PROT_READ\|PROT_WRITE, MAP_PRIVATE\|MAP_ANONYMOUS, 0, 0);	1189	cctx->sptr = mmap (0, cctx->ssize, PROT_EXEC\|PROT_READ\|PROT_WRITE, MAP_PRIVATE\|MAP_ANONYMOUS, 0, 0);
1175		1190
1176	if (cctx->sptr != (void *)-1)	1191	if (cctx->sptr != (void *)-1)
1177	{	1192	{
1178	# if CORO_STACKGUARD	1193	#if CORO_STACKGUARD
1179	mprotect (cctx->sptr, CORO_STACKGUARD * PAGESIZE, PROT_NONE);	1194	mprotect (cctx->sptr, CORO_STACKGUARD * PAGESIZE, PROT_NONE);
1180	# endif	1195	#endif
1181	stack_start = CORO_STACKGUARD * PAGESIZE + (char *)cctx->sptr;	1196	stack_start = (char )cctx->sptr + CORO_STACKGUARD PAGESIZE;
1182	stack_size = cctx->ssize - CORO_STACKGUARD * PAGESIZE;	1197	stack_size = cctx->ssize - CORO_STACKGUARD * PAGESIZE;
1183	cctx->flags \|= CC_MAPPED;	1198	cctx->flags \|= CC_MAPPED;
1184	}	1199	}
1185	else	1200	else
1186	#endif	1201	#endif
1187	{	1202	{
1188	cctx->ssize = cctx_stacksize * (long)sizeof (long);	1203	cctx->ssize = cctx_stacksize * (long)sizeof (long);
1189	New (0, cctx->sptr, cctx_stacksize, long);	1204	New (0, cctx->sptr, cctx_stacksize, long);
1190		1205
1191	if (!cctx->sptr)	1206	if (!cctx->sptr)
1192	{	1207	{
1193	perror ("FATAL: unable to allocate stack for coroutine");	1208	perror ("FATAL: unable to allocate stack for coroutine, exiting.");
1194	_exit (EXIT_FAILURE);	1209	_exit (EXIT_FAILURE);
1195	}	1210	}
1196		1211
1197	stack_start = cctx->sptr;	1212	stack_start = cctx->sptr;
1198	stack_size = cctx->ssize;	1213	stack_size = cctx->ssize;
1199	}	1214	}
1200		1215
1201	REGISTER_STACK (cctx, (char )stack_start, (char )stack_start + stack_size);	1216	#if CORO_USE_VALGRIND
		1217	cctx->valgrind_id = VALGRIND_STACK_REGISTER ((char )stack_start, (char )stack_start + stack_size);
		1218	#endif
		1219
1202	coro_create (&cctx->cctx, cctx_run, (void *)cctx, stack_start, stack_size);	1220	coro_create (&cctx->cctx, cctx_run, (void *)cctx, stack_start, stack_size);
1203		1221
1204	return cctx;	1222	return cctx;
1205	}	1223	}
1206		1224
…		…
1214	coro_destroy (&cctx->cctx);	1232	coro_destroy (&cctx->cctx);
1215		1233
1216	/* coro_transfer creates new, empty cctx's */	1234	/* coro_transfer creates new, empty cctx's */
1217	if (cctx->sptr)	1235	if (cctx->sptr)
1218	{	1236	{
1219	#if CORO_USE_VALGRIND	1237	#if CORO_USE_VALGRIND
1220	VALGRIND_STACK_DEREGISTER (cctx->valgrind_id);	1238	VALGRIND_STACK_DEREGISTER (cctx->valgrind_id);
1221	#endif	1239	#endif
1222		1240
1223	#if HAVE_MMAP	1241	#if HAVE_MMAP
1224	if (cctx->flags & CC_MAPPED)	1242	if (cctx->flags & CC_MAPPED)
1225	munmap (cctx->sptr, cctx->ssize);	1243	munmap (cctx->sptr, cctx->ssize);
1226	else	1244	else
…		…
1253	}	1271	}
1254		1272
1255	static void	1273	static void
1256	cctx_put (coro_cctx *cctx)	1274	cctx_put (coro_cctx *cctx)
1257	{	1275	{
1258	assert (("cctx_put called on non-initialised cctx", cctx->sptr));	1276	assert (("FATAL: cctx_put called on non-initialised cctx in Coro (please report)", cctx->sptr));
1259		1277
1260	/* free another cctx if overlimit */	1278	/* free another cctx if overlimit */
1261	if (expect_false (cctx_idle >= cctx_max_idle))	1279	if (expect_false (cctx_idle >= cctx_max_idle))
1262	{	1280	{
1263	coro_cctx *first = cctx_first;	1281	coro_cctx *first = cctx_first;
…		…
1278	transfer_check (pTHX_ struct coro prev, struct coro next)	1296	transfer_check (pTHX_ struct coro prev, struct coro next)
1279	{	1297	{
1280	if (expect_true (prev != next))	1298	if (expect_true (prev != next))
1281	{	1299	{
1282	if (expect_false (!(prev->flags & (CF_RUNNING \| CF_NEW))))	1300	if (expect_false (!(prev->flags & (CF_RUNNING \| CF_NEW))))
1283	croak ("Coro::State::transfer called with non-running/new prev Coro::State, but can only transfer from running or new states");	1301	croak ("Coro::State::transfer called with non-running/new prev Coro::State, but can only transfer from running or new states,");
1284		1302
1285	if (expect_false (next->flags & CF_RUNNING))	1303	if (expect_false (next->flags & CF_RUNNING))
1286	croak ("Coro::State::transfer called with running next Coro::State, but can only transfer to inactive states");	1304	croak ("Coro::State::transfer called with running next Coro::State, but can only transfer to inactive states,");
1287		1305
1288	if (expect_false (next->flags & CF_DESTROYED))	1306	if (expect_false (next->flags & CF_DESTROYED))
1289	croak ("Coro::State::transfer called with destroyed next Coro::State, but can only transfer to inactive states");	1307	croak ("Coro::State::transfer called with destroyed next Coro::State, but can only transfer to inactive states,");
1290		1308
1291	#if !PERL_VERSION_ATLEAST (5,10,0)	1309	#if !PERL_VERSION_ATLEAST (5,10,0)
1292	if (expect_false (PL_lex_state != LEX_NOTPARSING))	1310	if (expect_false (PL_lex_state != LEX_NOTPARSING))
1293	croak ("Coro::State::transfer called while parsing, but this is not supported in your perl version");	1311	croak ("Coro::State::transfer called while parsing, but this is not supported in your perl version,");
1294	#endif	1312	#endif
1295	}	1313	}
1296	}	1314	}
1297		1315
1298	/* always use the TRANSFER macro */	1316	/* always use the TRANSFER macro */
…		…
1302	dSTACKLEVEL;	1320	dSTACKLEVEL;
1303		1321
1304	/* sometimes transfer is only called to set idle_sp */	1322	/* sometimes transfer is only called to set idle_sp */
1305	if (expect_false (!next))	1323	if (expect_false (!next))
1306	{	1324	{
1307	((coro_cctx *)prev)->idle_sp = STACKLEVEL;	1325	((coro_cctx )prev)->idle_sp = (void )stacklevel;
1308	assert (((coro_cctx )prev)->idle_te = PL_top_env); / just for the side-effect when asserts are enabled */	1326	assert (((coro_cctx )prev)->idle_te = PL_top_env); / just for the side-effect when asserts are enabled */
1309	}	1327	}
1310	else if (expect_true (prev != next))	1328	else if (expect_true (prev != next))
1311	{	1329	{
1312	coro_cctx *prev__cctx;	1330	coro_cctx *prev__cctx;
…		…
1337	else	1355	else
1338	load_perl (aTHX_ next);	1356	load_perl (aTHX_ next);
1339		1357
1340	prev__cctx = prev->cctx;	1358	prev__cctx = prev->cctx;
1341		1359
1342	/* possibly "free" the cctx */	1360	/* possibly untie and reuse the cctx */
1343	if (expect_true (	1361	if (expect_true (
1344	prev__cctx->idle_sp == STACKLEVEL	1362	prev__cctx->idle_sp == (void *)stacklevel
1345	&& !(prev__cctx->flags & CC_TRACE)	1363	&& !(prev__cctx->flags & CC_TRACE)
1346	&& !force_cctx	1364	&& !force_cctx
1347	))	1365	))
1348	{	1366	{
1349	/* I assume that STACKLEVEL is a stronger indicator than PL_top_env changes */	1367	/* I assume that stacklevel is a stronger indicator than PL_top_env changes */
1350	assert (("ERROR: current top_env must equal previous top_env", PL_top_env == prev__cctx->idle_te));	1368	assert (("FATAL: current top_env must equal previous top_env in Coro (please report)", PL_top_env == prev__cctx->idle_te));
1351		1369
1352	prev->cctx = 0;	1370	prev->cctx = 0;
1353		1371
1354	/* if the cctx is about to be destroyed we need to make sure we won't see it in cctx_get */	1372	/* if the cctx is about to be destroyed we need to make sure we won't see it in cctx_get */
1355	/* without this the next cctx_get might destroy the prev__cctx while still in use */	1373	/* without this the next cctx_get might destroy the prev__cctx while still in use */
…		…
1363	++next->usecount;	1381	++next->usecount;
1364		1382
1365	if (expect_true (!next->cctx))	1383	if (expect_true (!next->cctx))
1366	next->cctx = cctx_get (aTHX);	1384	next->cctx = cctx_get (aTHX);
1367		1385
1368	assert (("FATAL ERROR: internal error 1352 in Coro, please report", !transfer_next));//D	1386	assert (("FATAL: transfer_next already nonzero in Coro (please report)", !transfer_next));
1369	transfer_next = next;	1387	transfer_next = next;
1370		1388
1371	if (expect_false (prev__cctx != next->cctx))	1389	if (expect_false (prev__cctx != next->cctx))
1372	{	1390	{
1373	prev__cctx->top_env = PL_top_env;	1391	prev__cctx->top_env = PL_top_env;
1374	PL_top_env = next->cctx->top_env;	1392	PL_top_env = next->cctx->top_env;
1375	coro_transfer (&prev__cctx->cctx, &next->cctx->cctx);	1393	coro_transfer (&prev__cctx->cctx, &next->cctx->cctx);
1376	}	1394	}
1377		1395
1378	transfer_tail ();	1396	transfer_tail (aTHX);
1379	}	1397	}
1380	}	1398	}
1381		1399
1382	#define TRANSFER(ta, force_cctx) transfer (aTHX_ (ta).prev, (ta).next, (force_cctx))	1400	#define TRANSFER(ta, force_cctx) transfer (aTHX_ (ta).prev, (ta).next, (force_cctx))
1383	#define TRANSFER_CHECK(ta) transfer_check (aTHX_ (ta).prev, (ta).next)	1401	#define TRANSFER_CHECK(ta) transfer_check (aTHX_ (ta).prev, (ta).next)
…		…
1405		1423
1406	if (coro->mainstack && coro->mainstack != main_mainstack)	1424	if (coro->mainstack && coro->mainstack != main_mainstack)
1407	{	1425	{
1408	struct coro temp;	1426	struct coro temp;
1409		1427
1410	if (coro->flags & CF_RUNNING)	1428	assert (("FATAL: tried to destroy currently running coroutine (please report)", !(coro->flags & CF_RUNNING)));
1411	croak ("FATAL: tried to destroy currently running coroutine");
1412		1429
1413	save_perl (aTHX_ &temp);	1430	save_perl (aTHX_ &temp);
1414	load_perl (aTHX_ coro);	1431	load_perl (aTHX_ coro);
1415		1432
1416	coro_destruct (aTHX_ coro);	1433	coro_destruct (aTHX_ coro);
…		…
1467	# define MGf_DUP 0	1484	# define MGf_DUP 0
1468	#endif	1485	#endif
1469	};	1486	};
1470		1487
1471	static void	1488	static void
1472	prepare_transfer (pTHX_ struct transfer_args ta, SV prev_sv, SV *next_sv)	1489	prepare_transfer (pTHX_ struct coro_transfer_args ta, SV prev_sv, SV *next_sv)
1473	{	1490	{
1474	ta->prev = SvSTATE (prev_sv);	1491	ta->prev = SvSTATE (prev_sv);
1475	ta->next = SvSTATE (next_sv);	1492	ta->next = SvSTATE (next_sv);
1476	TRANSFER_CHECK (*ta);	1493	TRANSFER_CHECK (*ta);
1477	}	1494	}
1478		1495
1479	static void	1496	static void
1480	api_transfer (SV prev_sv, SV next_sv)	1497	api_transfer (pTHX_ SV prev_sv, SV next_sv)
1481	{	1498	{
1482	dTHX;
1483	struct transfer_args ta;	1499	struct coro_transfer_args ta;
1484		1500
1485	prepare_transfer (aTHX_ &ta, prev_sv, next_sv);	1501	prepare_transfer (aTHX_ &ta, prev_sv, next_sv);
1486	TRANSFER (ta, 1);	1502	TRANSFER (ta, 1);
1487	}	1503	}
1488		1504
…		…
1505		1521
1506	return 0;	1522	return 0;
1507	}	1523	}
1508		1524
1509	static int	1525	static int
1510	api_ready (SV *coro_sv)	1526	api_ready (pTHX_ SV *coro_sv)
1511	{	1527	{
1512	dTHX;
1513	struct coro *coro;	1528	struct coro *coro;
1514	SV *sv_hook;	1529	SV *sv_hook;
1515	void (*xs_hook)(void);	1530	void (*xs_hook)(void);
1516		1531
1517	if (SvROK (coro_sv))	1532	if (SvROK (coro_sv))
…		…
1555		1570
1556	return 1;	1571	return 1;
1557	}	1572	}
1558		1573
1559	static int	1574	static int
1560	api_is_ready (SV *coro_sv)	1575	api_is_ready (pTHX_ SV *coro_sv)
1561	{	1576	{
1562	dTHX;
1563	return !!(SvSTATE (coro_sv)->flags & CF_READY);	1577	return !!(SvSTATE (coro_sv)->flags & CF_READY);
1564	}	1578	}
1565		1579
1566	static void	1580	INLINE void
1567	prepare_schedule (pTHX_ struct transfer_args *ta)	1581	prepare_schedule (pTHX_ struct coro_transfer_args *ta)
1568	{	1582	{
1569	SV prev_sv, next_sv;	1583	SV prev_sv, next_sv;
1570		1584
1571	for (;;)	1585	for (;;)
1572	{	1586	{
…		…
1610		1624
1611	/* free this only after the transfer */	1625	/* free this only after the transfer */
1612	prev_sv = SvRV (coro_current);	1626	prev_sv = SvRV (coro_current);
1613	ta->prev = SvSTATE (prev_sv);	1627	ta->prev = SvSTATE (prev_sv);
1614	TRANSFER_CHECK (*ta);	1628	TRANSFER_CHECK (*ta);
1615	assert (ta->next->flags & CF_READY);	1629	assert (("FATAL: next coroutine isn't marked as ready in Coro (please report)", ta->next->flags & CF_READY));
1616	ta->next->flags &= ~CF_READY;	1630	ta->next->flags &= ~CF_READY;
1617	SvRV_set (coro_current, next_sv);	1631	SvRV_set (coro_current, next_sv);
1618		1632
1619	LOCK;	1633	LOCK;
1620	free_coro_mortal (aTHX);	1634	free_coro_mortal (aTHX);
1621	coro_mortal = prev_sv;	1635	coro_mortal = prev_sv;
1622	UNLOCK;	1636	UNLOCK;
1623	}	1637	}
1624		1638
1625	static void	1639	INLINE void
1626	prepare_cede (pTHX_ struct transfer_args *ta)	1640	prepare_cede (pTHX_ struct coro_transfer_args *ta)
1627	{	1641	{
1628	api_ready (coro_current);	1642	api_ready (aTHX_ coro_current);
1629	prepare_schedule (aTHX_ ta);	1643	prepare_schedule (aTHX_ ta);
1630	}	1644	}
1631		1645
		1646	INLINE void
		1647	prepare_cede_notself (pTHX_ struct coro_transfer_args *ta)
		1648	{
		1649	SV *prev = SvRV (coro_current);
		1650
		1651	if (coro_nready)
		1652	{
		1653	prepare_schedule (aTHX_ ta);
		1654	api_ready (aTHX_ prev);
		1655	}
		1656	else
		1657	prepare_nop (aTHX_ ta);
		1658	}
		1659
		1660	static void
		1661	api_schedule (pTHX)
		1662	{
		1663	struct coro_transfer_args ta;
		1664
		1665	prepare_schedule (aTHX_ &ta);
		1666	TRANSFER (ta, 1);
		1667	}
		1668
1632	static int	1669	static int
1633	prepare_cede_notself (pTHX_ struct transfer_args *ta)	1670	api_cede (pTHX)
1634	{	1671	{
1635	if (coro_nready)	1672	struct coro_transfer_args ta;
1636	{	1673
1637	SV *prev = SvRV (coro_current);
1638	prepare_schedule (aTHX_ ta);	1674	prepare_cede (aTHX_ &ta);
1639	api_ready (prev);	1675
		1676	if (expect_true (ta.prev != ta.next))
		1677	{
		1678	TRANSFER (ta, 1);
1640	return 1;	1679	return 1;
1641	}	1680	}
1642	else	1681	else
1643	return 0;	1682	return 0;
1644	}	1683	}
1645		1684
1646	static void
1647	api_schedule (void)
1648	{
1649	dTHX;
1650	struct transfer_args ta;
1651
1652	prepare_schedule (aTHX_ &ta);
1653	TRANSFER (ta, 1);
1654	}
1655
1656	static int	1685	static int
1657	api_cede (void)	1686	api_cede_notself (pTHX)
1658	{	1687	{
1659	dTHX;	1688	if (coro_nready)
		1689	{
1660	struct transfer_args ta;	1690	struct coro_transfer_args ta;
1661		1691
1662	prepare_cede (aTHX_ &ta);	1692	prepare_cede_notself (aTHX_ &ta);
1663
1664	if (expect_true (ta.prev != ta.next))
1665	{
1666	TRANSFER (ta, 1);	1693	TRANSFER (ta, 1);
1667	return 1;	1694	return 1;
1668	}	1695	}
1669	else	1696	else
1670	return 0;	1697	return 0;
1671	}	1698	}
1672		1699
1673	static int	1700	static void
1674	api_cede_notself (void)
1675	{
1676	dTHX;
1677	struct transfer_args ta;
1678
1679	if (prepare_cede_notself (aTHX_ &ta))
1680	{
1681	TRANSFER (ta, 1);
1682	return 1;
1683	}
1684	else
1685	return 0;
1686	}
1687
1688	static void
1689	api_trace (SV *coro_sv, int flags)	1701	api_trace (pTHX_ SV *coro_sv, int flags)
1690	{	1702	{
1691	dTHX;
1692	struct coro *coro = SvSTATE (coro_sv);	1703	struct coro *coro = SvSTATE (coro_sv);
1693		1704
1694	if (flags & CC_TRACE)	1705	if (flags & CC_TRACE)
1695	{	1706	{
1696	if (!coro->cctx)	1707	if (!coro->cctx)
1697	coro->cctx = cctx_new_run ();	1708	coro->cctx = cctx_new_run ();
1698	else if (!(coro->cctx->flags & CC_TRACE))	1709	else if (!(coro->cctx->flags & CC_TRACE))
1699	croak ("cannot enable tracing on coroutine with custom stack");	1710	croak ("cannot enable tracing on coroutine with custom stack,");
1700		1711
1701	coro->cctx->flags \|= CC_NOREUSE \| (flags & (CC_TRACE \| CC_TRACE_ALL));	1712	coro->cctx->flags \|= CC_NOREUSE \| (flags & (CC_TRACE \| CC_TRACE_ALL));
1702	}	1713	}
1703	else if (coro->cctx && coro->cctx->flags & CC_TRACE)	1714	else if (coro->cctx && coro->cctx->flags & CC_TRACE)
1704	{	1715	{
…		…
1709	else	1720	else
1710	coro->slot->runops = RUNOPS_DEFAULT;	1721	coro->slot->runops = RUNOPS_DEFAULT;
1711	}	1722	}
1712	}	1723	}
1713		1724
1714	#if 0
1715	static int
1716	coro_gensub_free (pTHX_ SV sv, MAGIC mg)
1717	{
1718	AV *padlist;
1719	AV av = (AV )mg->mg_obj;
1720
1721	abort ();
1722
1723	return 0;
1724	}
1725
1726	static MGVTBL coro_gensub_vtbl = {
1727	0, 0, 0, 0,
1728	coro_gensub_free
1729	};
1730	#endif
1731
1732	/*****************************************************************************/	1725	/*****************************************************************************/
1733	/* PerlIO::cede */	1726	/* PerlIO::cede */
1734		1727
1735	typedef struct	1728	typedef struct
1736	{	1729	{
…		…
1763	PerlIOCede *self = PerlIOSelf (f, PerlIOCede);	1756	PerlIOCede *self = PerlIOSelf (f, PerlIOCede);
1764	double now = nvtime ();	1757	double now = nvtime ();
1765		1758
1766	if (now >= self->next)	1759	if (now >= self->next)
1767	{	1760	{
1768	api_cede ();	1761	api_cede (aTHX);
1769	self->next = now + self->every;	1762	self->next = now + self->every;
1770	}	1763	}
1771		1764
1772	return PerlIOBuf_flush (aTHX_ f);	1765	return PerlIOBuf_flush (aTHX_ f);
1773	}	1766	}
…		…
1802	PerlIOBuf_get_ptr,	1795	PerlIOBuf_get_ptr,
1803	PerlIOBuf_get_cnt,	1796	PerlIOBuf_get_cnt,
1804	PerlIOBuf_set_ptrcnt,	1797	PerlIOBuf_set_ptrcnt,
1805	};	1798	};
1806		1799
		1800	/*****************************************************************************/
		1801
		1802	static const CV slf_cv; / for quick consistency check */
		1803
		1804	static UNOP slf_restore; /* restore stack as entersub did, for first-re-run */
		1805	static SV *slf_arg0;
		1806	static SV *slf_arg1;
		1807	static SV *slf_arg2;
		1808
		1809	/* this restores the stack in the case we patched the entersub, to */
		1810	/* recreate the stack frame as perl will on following calls */
		1811	/* since entersub cleared the stack */
		1812	static OP *
		1813	pp_restore (pTHX)
		1814	{
		1815	dSP;
		1816
		1817	PUSHMARK (SP);
		1818
		1819	EXTEND (SP, 3);
		1820	if (slf_arg0) PUSHs (sv_2mortal (slf_arg0));
		1821	if (slf_arg1) PUSHs (sv_2mortal (slf_arg1));
		1822	if (slf_arg2) PUSHs (sv_2mortal (slf_arg2));
		1823	PUSHs ((SV *)CvGV (slf_cv));
		1824
		1825	RETURNOP (slf_restore.op_first);
		1826	}
		1827
		1828	static void
		1829	slf_prepare_set_stacklevel (pTHX_ struct coro_transfer_args *ta)
		1830	{
		1831	prepare_set_stacklevel (ta, (struct coro_cctx *)slf_frame.data);
		1832	}
		1833
		1834	static void
		1835	slf_init_set_stacklevel (pTHX_ struct CoroSLF frame, SV *arg, int items)
		1836	{
		1837	assert (("FATAL: set_stacklevel needs the coro cctx as sole argument", items == 1));
		1838
		1839	frame->prepare = slf_prepare_set_stacklevel;
		1840	frame->check = slf_check_nop;
		1841	frame->data = (void *)SvIV (arg [0]);
		1842	}
		1843
		1844	static void
		1845	slf_prepare_transfer (pTHX_ struct coro_transfer_args *ta)
		1846	{
		1847	SV arg = (SV )slf_frame.data;
		1848
		1849	prepare_transfer (aTHX_ ta, arg [0], arg [1]);
		1850	}
		1851
		1852	static void
		1853	slf_init_transfer (pTHX_ struct CoroSLF frame, SV *arg, int items)
		1854	{
		1855	if (items != 2)
		1856	croak ("Coro::State::transfer (prev, next) expects two arguments, not %d,", items);
		1857
		1858	frame->prepare = slf_prepare_transfer;
		1859	frame->check = slf_check_nop;
		1860	frame->data = (void )arg; / let's hope it will stay valid */
		1861	}
		1862
		1863	static void
		1864	slf_init_schedule (pTHX_ struct CoroSLF frame, SV *arg, int items)
		1865	{
		1866	frame->prepare = prepare_schedule;
		1867	frame->check = slf_check_nop;
		1868	}
		1869
		1870	static void
		1871	slf_init_cede (pTHX_ struct CoroSLF frame, SV *arg, int items)
		1872	{
		1873	frame->prepare = prepare_cede;
		1874	frame->check = slf_check_nop;
		1875	}
		1876
		1877	static void
		1878	slf_init_cede_notself (pTHX_ struct CoroSLF frame, SV *arg, int items)
		1879	{
		1880	frame->prepare = prepare_cede_notself;
		1881	frame->check = slf_check_nop;
		1882	}
		1883
		1884	/* we hijack an hopefully unused CV flag for our purposes */
		1885	#define CVf_SLF 0x4000
		1886
		1887	/*
		1888	* these not obviously related functions are all rolled into one
		1889	* function to increase chances that they all will call transfer with the same
		1890	* stack offset
		1891	* SLF stands for "schedule-like-function".
		1892	*/
		1893	static OP *
		1894	pp_slf (pTHX)
		1895	{
		1896	I32 checkmark; /* mark SP to see how many elements check has pushed */
		1897
		1898	/* set up the slf frame, unless it has already been set-up */
		1899	/* the latter happens when a new coro has been started */
		1900	/* or when a new cctx was attached to an existing coroutine */
		1901	if (expect_true (!slf_frame.prepare))
		1902	{
		1903	/* first iteration */
		1904	dSP;
		1905	SV **arg = PL_stack_base + TOPMARK + 1;
		1906	int items = SP - arg; /* args without function object */
		1907	SV gv = sp;
		1908
		1909	/* do a quick consistency check on the "function" object, and if it isn't */
		1910	/* for us, divert to the real entersub */
		1911	if (SvTYPE (gv) != SVt_PVGV \|\| !(CvFLAGS (GvCV (gv)) & CVf_SLF))
		1912	return PL_ppaddr[OP_ENTERSUB](aTHX);
		1913
		1914	/* pop args */
		1915	SP = PL_stack_base + POPMARK;
		1916
		1917	if (!(PL_op->op_flags & OPf_STACKED))
		1918	{
		1919	/* ampersand-form of call, use @_ instead of stack */
		1920	AV *av = GvAV (PL_defgv);
		1921	arg = AvARRAY (av);
		1922	items = AvFILLp (av) + 1;
		1923	}
		1924
		1925	PUTBACK;
		1926
		1927	((coro_slf_cb)CvXSUBANY (GvCV (gv)).any_ptr) (aTHX_ &slf_frame, arg, items);
		1928	}
		1929
		1930	/* now interpret the slf_frame */
		1931	/* we use a callback system not to make the code needlessly */
		1932	/* complicated, but so we can run multiple perl coros from one cctx */
		1933
		1934	do
		1935	{
		1936	struct coro_transfer_args ta;
		1937
		1938	slf_frame.prepare (aTHX_ &ta);
		1939	TRANSFER (ta, 0);
		1940
		1941	checkmark = PL_stack_sp - PL_stack_base;
		1942	}
		1943	while (slf_frame.check (aTHX_ &slf_frame));
		1944
		1945	{
		1946	dSP;
		1947	SV **bot = PL_stack_base + checkmark;
		1948	int gimme = GIMME_V;
		1949
		1950	slf_frame.prepare = 0; /* signal pp_slf that we need a new frame */
		1951
		1952	/* make sure we put something on the stack in scalar context */
		1953	if (gimme == G_SCALAR)
		1954	{
		1955	if (sp == bot)
		1956	XPUSHs (&PL_sv_undef);
		1957
		1958	SP = bot + 1;
		1959	}
		1960
		1961	PUTBACK;
		1962	}
		1963
		1964	return NORMAL;
		1965	}
		1966
		1967	static void
		1968	api_execute_slf (pTHX_ CV cv, coro_slf_cb init_cb, SV *arg, int items)
		1969	{
		1970	assert (("FATAL: SLF call with illegal CV value", !CvANON (cv)));
		1971
		1972	if (PL_op->op_ppaddr != PL_ppaddr [OP_ENTERSUB]
		1973	&& PL_op->op_ppaddr != pp_slf)
		1974	croak ("FATAL: Coro SLF calls can only be made normally, not via goto or any other means, caught");
		1975
		1976	if (items > 3)
		1977	croak ("Coro only supports up to three arguments to SLF functions currently (not %d), caught", items);
		1978
		1979	CvFLAGS (cv) \|= CVf_SLF;
		1980	CvXSUBANY (cv).any_ptr = (void *)init_cb;
		1981	slf_cv = cv;
		1982
		1983	/* we patch the op, and then re-run the whole call */
		1984	/* we have to put the same argument on the stack for this to work */
		1985	/* and this will be done by pp_restore */
		1986	slf_restore.op_next = (OP *)&slf_restore;
		1987	slf_restore.op_type = OP_NULL;
		1988	slf_restore.op_ppaddr = pp_restore;
		1989	slf_restore.op_first = PL_op;
		1990
		1991	slf_arg0 = items > 0 ? SvREFCNT_inc (arg [0]) : 0;
		1992	slf_arg1 = items > 1 ? SvREFCNT_inc (arg [1]) : 0;
		1993	slf_arg2 = items > 2 ? SvREFCNT_inc (arg [2]) : 0;
		1994
		1995	PL_op->op_ppaddr = pp_slf;
		1996
		1997	PL_op = (OP *)&slf_restore;
		1998	}
		1999
		2000	/*****************************************************************************/
		2001
		2002	static int
		2003	slf_check_semaphore_down (pTHX_ struct CoroSLF *frame)
		2004	{
		2005	AV av = (AV )frame->data;
		2006	SV *count_sv = AvARRAY (av)[0];
		2007
		2008	if (SvIVX (count_sv) > 0)
		2009	{
		2010	SvIVX (count_sv) = SvIVX (count_sv) - 1;
		2011	return 0;
		2012	}
		2013	else
		2014	{
		2015	int i;
		2016	/* if we were woken up but can't down, we look through the whole */
		2017	/* waiters list and only add us if we aren't in there already */
		2018	/* this avoids some degenerate memory usage cases */
		2019
		2020	for (i = 1; i <= AvFILLp (av); ++i)
		2021	if (AvARRAY (av)[i] == SvRV (coro_current))
		2022	return 1;
		2023
		2024	av_push (av, SvREFCNT_inc (SvRV (coro_current)));
		2025	return 1;
		2026	}
		2027	}
		2028
		2029	static void
		2030	slf_init_semaphore_down (pTHX_ struct CoroSLF frame, SV *arg, int items)
		2031	{
		2032	AV av = (AV )SvRV (arg [0]);
		2033
		2034	if (SvIVX (AvARRAY (av)[0]) > 0)
		2035	{
		2036	frame->data = (void *)av;
		2037	frame->prepare = prepare_nop;
		2038	}
		2039	else
		2040	{
		2041	av_push (av, SvREFCNT_inc (SvRV (coro_current)));
		2042
		2043	frame->data = (void )sv_2mortal (SvREFCNT_inc ((SV )av));
		2044	frame->prepare = prepare_schedule;
		2045	}
		2046
		2047	frame->check = slf_check_semaphore_down;
		2048
		2049	}
		2050
		2051	/*****************************************************************************/
		2052
		2053	#define GENSUB_ARG CvXSUBANY (cv).any_ptr
		2054
		2055	/* create a closure from XS, returns a code reference */
		2056	/* the arg can be accessed via GENSUB_ARG from the callback */
		2057	/* the callback must use dXSARGS/XSRETURN */
		2058	static SV *
		2059	gensub (pTHX_ void (xsub)(pTHX_ CV ), void *arg)
		2060	{
		2061	CV cv = (CV )NEWSV (0, 0);
		2062
		2063	sv_upgrade ((SV *)cv, SVt_PVCV);
		2064
		2065	CvANON_on (cv);
		2066	CvISXSUB_on (cv);
		2067	CvXSUB (cv) = xsub;
		2068	GENSUB_ARG = arg;
		2069
		2070	return newRV_noinc ((SV *)cv);
		2071	}
		2072
		2073	/*****************************************************************************/
1807		2074
1808	MODULE = Coro::State PACKAGE = Coro::State PREFIX = api_	2075	MODULE = Coro::State PACKAGE = Coro::State PREFIX = api_
1809		2076
1810	PROTOTYPES: DISABLE	2077	PROTOTYPES: DISABLE
1811		2078
…		…
1841	main_top_env = PL_top_env;	2108	main_top_env = PL_top_env;
1842		2109
1843	while (main_top_env->je_prev)	2110	while (main_top_env->je_prev)
1844	main_top_env = main_top_env->je_prev;	2111	main_top_env = main_top_env->je_prev;
1845		2112
1846	coroapi.ver = CORO_API_VERSION;	2113	coroapi.ver = CORO_API_VERSION;
1847	coroapi.rev = CORO_API_REVISION;	2114	coroapi.rev = CORO_API_REVISION;
		2115
1848	coroapi.transfer = api_transfer;	2116	coroapi.transfer = api_transfer;
		2117
		2118	coroapi.sv_state = SvSTATE_;
		2119	coroapi.execute_slf = api_execute_slf;
		2120	coroapi.prepare_nop = prepare_nop;
		2121	coroapi.prepare_schedule = prepare_schedule;
		2122	coroapi.prepare_cede = prepare_cede;
		2123	coroapi.prepare_cede_notself = prepare_cede_notself;
1849		2124
1850	{	2125	{
1851	SV **svp = hv_fetch (PL_modglobal, "Time::NVtime", 12, 0);	2126	SV **svp = hv_fetch (PL_modglobal, "Time::NVtime", 12, 0);
1852		2127
1853	if (!svp) croak ("Time::HiRes is required");	2128	if (!svp) croak ("Time::HiRes is required");
…		…
1886	av_push (coro->args, newSVsv (ST (i)));	2161	av_push (coro->args, newSVsv (ST (i)));
1887	}	2162	}
1888	OUTPUT:	2163	OUTPUT:
1889	RETVAL	2164	RETVAL
1890		2165
1891	# these not obviously related functions are all rolled into the same xs
1892	# function to increase chances that they all will call transfer with the same
1893	# stack offset
1894	void	2166	void
1895	_set_stacklevel (...)	2167	_set_stacklevel (...)
1896	ALIAS:	2168	CODE:
1897	Coro::State::transfer = 1	2169	api_execute_slf (aTHX_ cv, slf_init_set_stacklevel, &ST (0), items);
1898	Coro::schedule = 2
1899	Coro::cede = 3
1900	Coro::cede_notself = 4
1901	CODE:
1902	{
1903	struct transfer_args ta;
1904		2170
1905	PUTBACK;	2171	void
1906	switch (ix)	2172	transfer (...)
1907	{	2173	PROTOTYPE: $$
1908	case 0:	2174	CODE:
1909	prepare_set_stacklevel (&ta, (struct coro_cctx *)SvIV (ST (0)));	2175	api_execute_slf (aTHX_ cv, slf_init_transfer, &ST (0), items);
1910	break;
1911
1912	case 1:
1913	if (items != 2)
1914	croak ("Coro::State::transfer (prev, next) expects two arguments, not %d", items);
1915
1916	prepare_transfer (aTHX_ &ta, ST (0), ST (1));
1917	break;
1918
1919	case 2:
1920	prepare_schedule (aTHX_ &ta);
1921	break;
1922
1923	case 3:
1924	prepare_cede (aTHX_ &ta);
1925	break;
1926
1927	case 4:
1928	if (!prepare_cede_notself (aTHX_ &ta))
1929	XSRETURN_EMPTY;
1930
1931	break;
1932	}
1933	SPAGAIN;
1934
1935	BARRIER;
1936	PUTBACK;
1937	TRANSFER (ta, 0);
1938	SPAGAIN; /* might be the sp of a different coroutine now */
1939	/* be extra careful not to ever do anything after TRANSFER */
1940	}
1941		2176
1942	bool	2177	bool
1943	_destroy (SV *coro_sv)	2178	_destroy (SV *coro_sv)
1944	CODE:	2179	CODE:
1945	RETVAL = coro_state_destroy (aTHX_ SvSTATE (coro_sv));	2180	RETVAL = coro_state_destroy (aTHX_ SvSTATE (coro_sv));
…		…
1952	CODE:	2187	CODE:
1953	_exit (code);	2188	_exit (code);
1954		2189
1955	int	2190	int
1956	cctx_stacksize (int new_stacksize = 0)	2191	cctx_stacksize (int new_stacksize = 0)
		2192	PROTOTYPE: ;$
1957	CODE:	2193	CODE:
1958	RETVAL = cctx_stacksize;	2194	RETVAL = cctx_stacksize;
1959	if (new_stacksize)	2195	if (new_stacksize)
1960	{	2196	{
1961	cctx_stacksize = new_stacksize;	2197	cctx_stacksize = new_stacksize;
…		…
1964	OUTPUT:	2200	OUTPUT:
1965	RETVAL	2201	RETVAL
1966		2202
1967	int	2203	int
1968	cctx_max_idle (int max_idle = 0)	2204	cctx_max_idle (int max_idle = 0)
		2205	PROTOTYPE: ;$
1969	CODE:	2206	CODE:
1970	RETVAL = cctx_max_idle;	2207	RETVAL = cctx_max_idle;
1971	if (max_idle > 1)	2208	if (max_idle > 1)
1972	cctx_max_idle = max_idle;	2209	cctx_max_idle = max_idle;
1973	OUTPUT:	2210	OUTPUT:
1974	RETVAL	2211	RETVAL
1975		2212
1976	int	2213	int
1977	cctx_count ()	2214	cctx_count ()
		2215	PROTOTYPE:
1978	CODE:	2216	CODE:
1979	RETVAL = cctx_count;	2217	RETVAL = cctx_count;
1980	OUTPUT:	2218	OUTPUT:
1981	RETVAL	2219	RETVAL
1982		2220
1983	int	2221	int
1984	cctx_idle ()	2222	cctx_idle ()
		2223	PROTOTYPE:
1985	CODE:	2224	CODE:
1986	RETVAL = cctx_idle;	2225	RETVAL = cctx_idle;
1987	OUTPUT:	2226	OUTPUT:
1988	RETVAL	2227	RETVAL
1989		2228
1990	void	2229	void
1991	list ()	2230	list ()
		2231	PROTOTYPE:
1992	PPCODE:	2232	PPCODE:
1993	{	2233	{
1994	struct coro *coro;	2234	struct coro *coro;
1995	for (coro = coro_first; coro; coro = coro->next)	2235	for (coro = coro_first; coro; coro = coro->next)
1996	if (coro->hv)	2236	if (coro->hv)
…		…
2063	SvREFCNT_dec (self->throw);	2303	SvREFCNT_dec (self->throw);
2064	self->throw = SvOK (throw) ? newSVsv (throw) : 0;	2304	self->throw = SvOK (throw) ? newSVsv (throw) : 0;
2065		2305
2066	void	2306	void
2067	api_trace (SV *coro, int flags = CC_TRACE \| CC_TRACE_SUB)	2307	api_trace (SV *coro, int flags = CC_TRACE \| CC_TRACE_SUB)
		2308	PROTOTYPE: $;$
		2309	C_ARGS: aTHX_ coro, flags
2068		2310
2069	SV *	2311	SV *
2070	has_cctx (Coro::State coro)	2312	has_cctx (Coro::State coro)
2071	PROTOTYPE: $	2313	PROTOTYPE: $
2072	CODE:	2314	CODE:
…		…
2096	OUTPUT:	2338	OUTPUT:
2097	RETVAL	2339	RETVAL
2098		2340
2099	void	2341	void
2100	force_cctx ()	2342	force_cctx ()
		2343	PROTOTYPE:
2101	CODE:	2344	CODE:
2102	struct coro *coro = SvSTATE (coro_current);	2345	struct coro *coro = SvSTATE (coro_current);
2103	coro->cctx->idle_sp = 0;	2346	coro->cctx->idle_sp = 0;
2104		2347
2105	void	2348	void
…		…
2107	PROTOTYPE: $	2350	PROTOTYPE: $
2108	ALIAS:	2351	ALIAS:
2109	swap_defav = 1	2352	swap_defav = 1
2110	CODE:	2353	CODE:
2111	if (!self->slot)	2354	if (!self->slot)
2112	croak ("cannot swap state with coroutine that has no saved state");	2355	croak ("cannot swap state with coroutine that has no saved state,");
2113	else	2356	else
2114	{	2357	{
2115	SV src = ix ? (SV )&GvAV (PL_defgv) : &GvSV (PL_defgv);	2358	SV src = ix ? (SV )&GvAV (PL_defgv) : &GvSV (PL_defgv);
2116	SV dst = ix ? (SV )&self->slot->defav : (SV **)&self->slot->defsv;	2359	SV dst = ix ? (SV )&self->slot->defav : (SV **)&self->slot->defsv;
2117		2360
…		…
2142		2385
2143	for (i = PRIO_MAX - PRIO_MIN + 1; i--; )	2386	for (i = PRIO_MAX - PRIO_MIN + 1; i--; )
2144	coro_ready[i] = newAV ();	2387	coro_ready[i] = newAV ();
2145		2388
2146	{	2389	{
2147	SV *sv = perl_get_sv ("Coro::API", TRUE);	2390	SV *sv = coro_get_sv (aTHX_ "Coro::API", TRUE);
2148	perl_get_sv ("Coro::API", TRUE); /* silence 5.10 warning */
2149		2391
2150	coroapi.schedule = api_schedule;	2392	coroapi.schedule = api_schedule;
2151	coroapi.cede = api_cede;	2393	coroapi.cede = api_cede;
2152	coroapi.cede_notself = api_cede_notself;	2394	coroapi.cede_notself = api_cede_notself;
2153	coroapi.ready = api_ready;	2395	coroapi.ready = api_ready;
2154	coroapi.is_ready = api_is_ready;	2396	coroapi.is_ready = api_is_ready;
2155	coroapi.nready = &coro_nready;	2397	coroapi.nready = coro_nready;
2156	coroapi.current = coro_current;	2398	coroapi.current = coro_current;
2157		2399
2158	GCoroAPI = &coroapi;	2400	GCoroAPI = &coroapi;
2159	sv_setiv (sv, (IV)&coroapi);	2401	sv_setiv (sv, (IV)&coroapi);
2160	SvREADONLY_on (sv);	2402	SvREADONLY_on (sv);
2161	}	2403	}
2162	}	2404	}
		2405
		2406	void
		2407	schedule (...)
		2408	CODE:
		2409	api_execute_slf (aTHX_ cv, slf_init_schedule, &ST (0), 0);
		2410
		2411	void
		2412	cede (...)
		2413	CODE:
		2414	api_execute_slf (aTHX_ cv, slf_init_cede, &ST (0), 0);
		2415
		2416	void
		2417	cede_notself (...)
		2418	CODE:
		2419	api_execute_slf (aTHX_ cv, slf_init_cede_notself, &ST (0), 0);
2163		2420
2164	void	2421	void
2165	_set_current (SV *current)	2422	_set_current (SV *current)
2166	PROTOTYPE: $	2423	PROTOTYPE: $
2167	CODE:	2424	CODE:
…		…
2177	coro_readyhook = SvOK (hook) ? newSVsv (hook) : 0;	2434	coro_readyhook = SvOK (hook) ? newSVsv (hook) : 0;
2178	UNLOCK;	2435	UNLOCK;
2179		2436
2180	int	2437	int
2181	prio (Coro::State coro, int newprio = 0)	2438	prio (Coro::State coro, int newprio = 0)
		2439	PROTOTYPE: $;$
2182	ALIAS:	2440	ALIAS:
2183	nice = 1	2441	nice = 1
2184	CODE:	2442	CODE:
2185	{	2443	{
2186	RETVAL = coro->prio;	2444	RETVAL = coro->prio;
…		…
2201		2459
2202	SV *	2460	SV *
2203	ready (SV *self)	2461	ready (SV *self)
2204	PROTOTYPE: $	2462	PROTOTYPE: $
2205	CODE:	2463	CODE:
2206	RETVAL = boolSV (api_ready (self));	2464	RETVAL = boolSV (api_ready (aTHX_ self));
2207	OUTPUT:	2465	OUTPUT:
2208	RETVAL	2466	RETVAL
2209		2467
2210	int	2468	int
2211	nready (...)	2469	nready (...)
…		…
2250	{	2508	{
2251	av_fill (defav, len - 1);	2509	av_fill (defav, len - 1);
2252	for (i = 0; i < len; ++i)	2510	for (i = 0; i < len; ++i)
2253	av_store (defav, i, SvREFCNT_inc_NN (AvARRAY (invoke_av)[i + 1]));	2511	av_store (defav, i, SvREFCNT_inc_NN (AvARRAY (invoke_av)[i + 1]));
2254	}	2512	}
2255
2256	SvREFCNT_dec (invoke);
2257	}	2513	}
2258		2514
2259	void	2515	void
2260	_pool_2 (SV *cb)	2516	_pool_2 (SV *cb)
2261	CODE:	2517	CODE:
…		…
2281	newSVpvn ("[async_pool idle]", sizeof ("[async_pool idle]") - 1), 0);	2537	newSVpvn ("[async_pool idle]", sizeof ("[async_pool idle]") - 1), 0);
2282		2538
2283	coro->prio = 0;	2539	coro->prio = 0;
2284		2540
2285	if (coro->cctx && (coro->cctx->flags & CC_TRACE))	2541	if (coro->cctx && (coro->cctx->flags & CC_TRACE))
2286	api_trace (coro_current, 0);	2542	api_trace (aTHX_ coro_current, 0);
2287		2543
2288	av_push (av_async_pool, newSVsv (coro_current));	2544	av_push (av_async_pool, newSVsv (coro_current));
2289	}	2545	}
2290
2291	#if 0
2292
2293	void
2294	_generator_call (...)
2295	PROTOTYPE: @
2296	PPCODE:
2297	fprintf (stderr, "call %p\n", CvXSUBANY(cv).any_ptr);
2298	xxxx
2299	abort ();
2300
2301	SV *
2302	gensub (SV *sub, ...)
2303	PROTOTYPE: &;@
2304	CODE:
2305	{
2306	struct coro *coro;
2307	MAGIC *mg;
2308	CV *xcv;
2309	CV ncv = (CV )newSV_type (SVt_PVCV);
2310	int i;
2311
2312	CvGV (ncv) = CvGV (cv);
2313	CvFILE (ncv) = CvFILE (cv);
2314
2315	Newz (0, coro, 1, struct coro);
2316	coro->args = newAV ();
2317	coro->flags = CF_NEW;
2318
2319	av_extend (coro->args, items - 1);
2320	for (i = 1; i < items; i++)
2321	av_push (coro->args, newSVsv (ST (i)));
2322
2323	CvISXSUB_on (ncv);
2324	CvXSUBANY (ncv).any_ptr = (void *)coro;
2325
2326	xcv = GvCV (gv_fetchpv ("Coro::_generator_call", 0, SVt_PVCV));
2327
2328	CvXSUB (ncv) = CvXSUB (xcv);
2329	CvANON_on (ncv);
2330
2331	mg = sv_magicext ((SV )ncv, 0, CORO_MAGIC_type_state, &coro_gensub_vtbl, (char )coro, 0);
2332	RETVAL = newRV_noinc ((SV *)ncv);
2333	}
2334	OUTPUT:
2335	RETVAL
2336
2337	#endif
2338		2546
2339		2547
2340	MODULE = Coro::State PACKAGE = Coro::AIO	2548	MODULE = Coro::State PACKAGE = Coro::AIO
2341		2549
2342	void	2550	void
2343	_get_state (SV *self)	2551	_get_state (SV *self)
		2552	PROTOTYPE: $
2344	PPCODE:	2553	PPCODE:
2345	{	2554	{
2346	AV *defav = GvAV (PL_defgv);	2555	AV *defav = GvAV (PL_defgv);
2347	AV *av = newAV ();	2556	AV *av = newAV ();
2348	int i;	2557	int i;
…		…
2363		2572
2364	av_push (av, data_sv);	2573	av_push (av, data_sv);
2365		2574
2366	XPUSHs (sv_2mortal (newRV_noinc ((SV *)av)));	2575	XPUSHs (sv_2mortal (newRV_noinc ((SV *)av)));
2367		2576
2368	api_ready (self);	2577	api_ready (aTHX_ self);
2369	}	2578	}
2370		2579
2371	void	2580	void
2372	_set_state (SV *state)	2581	_set_state (SV *state)
2373	PROTOTYPE: $	2582	PROTOTYPE: $
…		…
2391	MODULE = Coro::State PACKAGE = Coro::AnyEvent	2600	MODULE = Coro::State PACKAGE = Coro::AnyEvent
2392		2601
2393	BOOT:	2602	BOOT:
2394	sv_activity = coro_get_sv (aTHX_ "Coro::AnyEvent::ACTIVITY", TRUE);	2603	sv_activity = coro_get_sv (aTHX_ "Coro::AnyEvent::ACTIVITY", TRUE);
2395		2604
2396	SV *	2605	void
2397	_schedule (...)	2606	_schedule (...)
2398	PROTOTYPE: @
2399	CODE:	2607	CODE:
2400	{	2608	{
2401	static int incede;	2609	static int incede;
2402		2610
2403	api_cede_notself ();	2611	api_cede_notself (aTHX);
2404		2612
2405	++incede;	2613	++incede;
2406	while (coro_nready >= incede && api_cede ())	2614	while (coro_nready >= incede && api_cede (aTHX))
2407	;	2615	;
2408		2616
2409	sv_setsv (sv_activity, &PL_sv_undef);	2617	sv_setsv (sv_activity, &PL_sv_undef);
2410	if (coro_nready >= incede)	2618	if (coro_nready >= incede)
2411	{	2619	{
…		…
2421		2629
2422	MODULE = Coro::State PACKAGE = PerlIO::cede	2630	MODULE = Coro::State PACKAGE = PerlIO::cede
2423		2631
2424	BOOT:	2632	BOOT:
2425	PerlIO_define_layer (aTHX_ &PerlIO_cede);	2633	PerlIO_define_layer (aTHX_ &PerlIO_cede);
		2634
		2635	MODULE = Coro::State PACKAGE = Coro::Semaphore
		2636
		2637	SV *
		2638	new (SV klass, SV count_ = 0)
		2639	CODE:
		2640	{
		2641	/* a semaphore contains a counter IV in $sem->[0] and any waiters after that */
		2642	AV *av = newAV ();
		2643	av_push (av, newSViv (count_ && SvOK (count_) ? SvIV (count_) : 1));
		2644	RETVAL = sv_bless (newRV_noinc ((SV *)av), GvSTASH (CvGV (cv)));
		2645	}
		2646	OUTPUT:
		2647	RETVAL
		2648
		2649	SV *
		2650	count (SV *self)
		2651	CODE:
		2652	RETVAL = newSVsv (AvARRAY ((AV *)SvRV (self))[0]);
		2653	OUTPUT:
		2654	RETVAL
		2655
		2656	void
		2657	up (SV *self, int adjust = 1)
		2658	ALIAS:
		2659	adjust = 1
		2660	CODE:
		2661	{
		2662	AV av = (AV )SvRV (self);
		2663	SV *count_sv = AvARRAY (av)[0];
		2664	IV count = SvIVX (count_sv);
		2665
		2666	count += ix ? adjust : 1;
		2667	SvIVX (count_sv) = count;
		2668
		2669	/* now wake up as many waiters as possible */
		2670	while (count > 0 && AvFILLp (av) >= count)
		2671	{
		2672	SV *cb;
		2673
		2674	/* swap first two elements so we can shift a waiter */
		2675	AvARRAY (av)[0] = AvARRAY (av)[1];
		2676	AvARRAY (av)[1] = count_sv;
		2677	cb = av_shift (av);
		2678
		2679	if (SvOBJECT (cb))
		2680	api_ready (aTHX_ cb);
		2681	else
		2682	croak ("callbacks not yet supported");
		2683
		2684	SvREFCNT_dec (cb);
		2685	}
		2686	}
		2687
		2688	void
		2689	down (SV *self)
		2690	CODE:
		2691	api_execute_slf (aTHX_ cv, slf_init_semaphore_down, &ST (0), 1);
		2692
		2693	void
		2694	try (SV *self)
		2695	PPCODE:
		2696	{
		2697	AV av = (AV )SvRV (self);
		2698	SV *count_sv = AvARRAY (av)[0];
		2699	IV count = SvIVX (count_sv);
		2700
		2701	if (count > 0)
		2702	{
		2703	--count;
		2704	SvIVX (count_sv) = count;
		2705	XSRETURN_YES;
		2706	}
		2707	else
		2708	XSRETURN_NO;
		2709	}
		2710
		2711	void
		2712	waiters (SV *self)
		2713	CODE:
		2714	{
		2715	AV av = (AV )SvRV (self);
		2716
		2717	if (GIMME_V == G_SCALAR)
		2718	XPUSHs (sv_2mortal (newSVsv (AvARRAY (av)[0])));
		2719	else
		2720	{
		2721	int i;
		2722	EXTEND (SP, AvFILLp (av) + 1 - 1);
		2723	for (i = 1; i <= AvFILLp (av); ++i)
		2724	PUSHs (newSVsv (AvARRAY (av)[i]));
		2725	}
		2726	}
		2727

Diff Legend

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

Comparing Coro/Coro/State.xs (file contents): Revision 1.260 by root, Mon Nov 10 04:37:23 2008 UTC vs. Revision 1.278 by root, Sun Nov 16 07:09:28 2008 UTC

Diff Legend

Comparing Coro/Coro/State.xs (file contents):
Revision 1.260 by root, Mon Nov 10 04:37:23 2008 UTC vs.
Revision 1.278 by root, Sun Nov 16 07:09:28 2008 UTC