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

Comparing Coro/Coro/State.xs (file contents):
Revision 1.330 by root, Wed Nov 26 09:29:06 2008 UTC vs.
Revision 1.352 by root, Sat Jun 20 08:58:50 2009 UTC

…		…
133	#else	133	#else
134	# define dSTACKLEVEL volatile void *stacklevel	134	# define dSTACKLEVEL volatile void *stacklevel
135	# define STACKLEVEL ((void *)&stacklevel)	135	# define STACKLEVEL ((void *)&stacklevel)
136	#endif	136	#endif
137		137
138	#define IN_DESTRUCT (PL_main_cv == Nullcv)	138	#define IN_DESTRUCT PL_dirty
139		139
140	#if __GNUC__ >= 3	140	#if __GNUC__ >= 3
141	# define attribute(x) __attribute__(x)	141	# define attribute(x) __attribute__(x)
142	# define expect(expr,value) __builtin_expect ((expr),(value))	142	# define expect(expr,value) __builtin_expect ((expr),(value))
143	# define INLINE static inline	143	# define INLINE static inline
…		…
238	enum {	238	enum {
239	CF_RUNNING = 0x0001, /* coroutine is running */	239	CF_RUNNING = 0x0001, /* coroutine is running */
240	CF_READY = 0x0002, /* coroutine is ready */	240	CF_READY = 0x0002, /* coroutine is ready */
241	CF_NEW = 0x0004, /* has never been switched to */	241	CF_NEW = 0x0004, /* has never been switched to */
242	CF_DESTROYED = 0x0008, /* coroutine data has been freed */	242	CF_DESTROYED = 0x0008, /* coroutine data has been freed */
		243	CF_SUSPENDED = 0x0010, /* coroutine can't be scheduled */
243	};	244	};
244		245
245	/* the structure where most of the perl state is stored, overlaid on the cxstack */	246	/* the structure where most of the perl state is stored, overlaid on the cxstack */
246	typedef struct	247	typedef struct
247	{	248	{
…		…
260	/* this is a structure representing a perl-level coroutine */	261	/* this is a structure representing a perl-level coroutine */
261	struct coro {	262	struct coro {
262	/* the C coroutine allocated to this perl coroutine, if any */	263	/* the C coroutine allocated to this perl coroutine, if any */
263	coro_cctx *cctx;	264	coro_cctx *cctx;
264		265
		266	/* ready queue */
		267	struct coro *next_ready;
		268
265	/* state data */	269	/* state data */
266	struct CoroSLF slf_frame; /* saved slf frame */	270	struct CoroSLF slf_frame; /* saved slf frame */
267	AV *mainstack;	271	AV *mainstack;
268	perl_slots slot; / basically the saved sp */	272	perl_slots slot; / basically the saved sp */
269		273
…		…
285	/* async_pool */	289	/* async_pool */
286	SV *saved_deffh;	290	SV *saved_deffh;
287	SV *invoke_cb;	291	SV *invoke_cb;
288	AV *invoke_av;	292	AV *invoke_av;
289		293
		294	/* on_enter/on_leave */
		295	AV *on_enter;
		296	AV *on_leave;
		297
290	/* linked list */	298	/* linked list */
291	struct coro next, prev;	299	struct coro next, prev;
292	};	300	};
293		301
294	typedef struct coro *Coro__State;	302	typedef struct coro *Coro__State;
…		…
309	#define PRIO_MIN -4	317	#define PRIO_MIN -4
310		318
311	/* for Coro.pm */	319	/* for Coro.pm */
312	static SV *coro_current;	320	static SV *coro_current;
313	static SV *coro_readyhook;	321	static SV *coro_readyhook;
314	static AV *coro_ready [PRIO_MAX - PRIO_MIN + 1];	322	static struct coro coro_ready [PRIO_MAX - PRIO_MIN + 1][2]; / head\|tail */
315	static CV cv_coro_run, cv_coro_terminate;	323	static CV cv_coro_run, cv_coro_terminate;
316	static struct coro *coro_first;	324	static struct coro *coro_first;
317	#define coro_nready coroapi.nready	325	#define coro_nready coroapi.nready
318		326
319	/ lowlevel stuff ********************************************************/	327	/ lowlevel stuff ********************************************************/
…		…
352	INLINE CV *	360	INLINE CV *
353	coro_sv_2cv (pTHX_ SV *sv)	361	coro_sv_2cv (pTHX_ SV *sv)
354	{	362	{
355	HV *st;	363	HV *st;
356	GV *gvp;	364	GV *gvp;
357	return sv_2cv (sv, &st, &gvp, 0);	365	CV *cv = sv_2cv (sv, &st, &gvp, 0);
		366
		367	if (!cv)
		368	croak ("code reference expected");
		369
		370	return cv;
358	}	371	}
		372
		373	/*****************************************************************************/
		374	/* magic glue */
		375
		376	#define CORO_MAGIC_type_cv 26
		377	#define CORO_MAGIC_type_state PERL_MAGIC_ext
		378
		379	#define CORO_MAGIC_NN(sv, type) \
		380	(expect_true (SvMAGIC (sv)->mg_type == type) \
		381	? SvMAGIC (sv) \
		382	: mg_find (sv, type))
		383
		384	#define CORO_MAGIC(sv, type) \
		385	(expect_true (SvMAGIC (sv)) \
		386	? CORO_MAGIC_NN (sv, type) \
		387	: 0)
		388
		389	#define CORO_MAGIC_cv(cv) CORO_MAGIC (((SV *)(cv)), CORO_MAGIC_type_cv)
		390	#define CORO_MAGIC_state(sv) CORO_MAGIC_NN (((SV *)(sv)), CORO_MAGIC_type_state)
		391
		392	INLINE struct coro *
		393	SvSTATE_ (pTHX_ SV *coro)
		394	{
		395	HV *stash;
		396	MAGIC *mg;
		397
		398	if (SvROK (coro))
		399	coro = SvRV (coro);
		400
		401	if (expect_false (SvTYPE (coro) != SVt_PVHV))
		402	croak ("Coro::State object required");
		403
		404	stash = SvSTASH (coro);
		405	if (expect_false (stash != coro_stash && stash != coro_state_stash))
		406	{
		407	/* very slow, but rare, check */
		408	if (!sv_derived_from (sv_2mortal (newRV_inc (coro)), "Coro::State"))
		409	croak ("Coro::State object required");
		410	}
		411
		412	mg = CORO_MAGIC_state (coro);
		413	return (struct coro *)mg->mg_ptr;
		414	}
		415
		416	#define SvSTATE(sv) SvSTATE_ (aTHX_ (sv))
		417
		418	/* faster than SvSTATE, but expects a coroutine hv */
		419	#define SvSTATE_hv(hv) ((struct coro )CORO_MAGIC_NN ((SV )hv, CORO_MAGIC_type_state)->mg_ptr)
		420	#define SvSTATE_current SvSTATE_hv (SvRV (coro_current))
		421
		422	/*****************************************************************************/
		423	/* padlist management and caching */
359		424
360	static AV *	425	static AV *
361	coro_derive_padlist (pTHX_ CV *cv)	426	coro_derive_padlist (pTHX_ CV *cv)
362	{	427	{
363	AV *padlist = CvPADLIST (cv);	428	AV *padlist = CvPADLIST (cv);
…		…
371	Perl_pad_push (aTHX_ padlist, AvFILLp (padlist) + 1, 1);	436	Perl_pad_push (aTHX_ padlist, AvFILLp (padlist) + 1, 1);
372	#endif	437	#endif
373	newpad = (AV *)AvARRAY (padlist)[AvFILLp (padlist)];	438	newpad = (AV *)AvARRAY (padlist)[AvFILLp (padlist)];
374	--AvFILLp (padlist);	439	--AvFILLp (padlist);
375		440
376	av_store (newpadlist, 0, SvREFCNT_inc_NN (*av_fetch (padlist, 0, FALSE)));	441	av_store (newpadlist, 0, SvREFCNT_inc_NN (AvARRAY (padlist)[0]));
377	av_store (newpadlist, 1, (SV *)newpad);	442	av_store (newpadlist, 1, (SV *)newpad);
378		443
379	return newpadlist;	444	return newpadlist;
380	}	445	}
381		446
382	static void	447	static void
383	free_padlist (pTHX_ AV *padlist)	448	free_padlist (pTHX_ AV *padlist)
384	{	449	{
385	/* may be during global destruction */	450	/* may be during global destruction */
386	if (SvREFCNT (padlist))	451	if (!IN_DESTRUCT)
387	{	452	{
388	I32 i = AvFILLp (padlist);	453	I32 i = AvFILLp (padlist);
389	while (i >= 0)	454
		455	while (i > 0) /* special-case index 0 */
390	{	456	{
		457	/* we try to be extra-careful here */
391	SV **svp = AvARRAY (padlist)[i--];	458	AV av = (AV )AvARRAY (padlist)[i--];
392	if (svp)	459	I32 j = AvFILLp (av);
393	{	460
394	AvREAL_on (*svp);	461	while (j >= 0)
395	av_undef (*svp);	462	SvREFCNT_dec (AvARRAY (av)[j--]);
		463
		464	AvFILLp (av) = -1;
396	SvREFCNT_dec (*svp);	465	SvREFCNT_dec (av);
397	}
398	}	466	}
399		467
		468	SvREFCNT_dec (AvARRAY (padlist)[0]);
		469
		470	AvFILLp (padlist) = -1;
400	SvREFCNT_dec ((SV*)padlist);	471	SvREFCNT_dec ((SV*)padlist);
401	}	472	}
402	}	473	}
403		474
404	static int	475	static int
…		…
413		484
414	SvREFCNT_dec (av); /* sv_magicext increased the refcount */	485	SvREFCNT_dec (av); /* sv_magicext increased the refcount */
415		486
416	return 0;	487	return 0;
417	}	488	}
418
419	#define CORO_MAGIC_type_cv 26
420	#define CORO_MAGIC_type_state PERL_MAGIC_ext
421		489
422	static MGVTBL coro_cv_vtbl = {	490	static MGVTBL coro_cv_vtbl = {
423	0, 0, 0, 0,	491	0, 0, 0, 0,
424	coro_cv_free	492	coro_cv_free
425	};	493	};
426
427	#define CORO_MAGIC_NN(sv, type) \
428	(expect_true (SvMAGIC (sv)->mg_type == type) \
429	? SvMAGIC (sv) \
430	: mg_find (sv, type))
431
432	#define CORO_MAGIC(sv, type) \
433	(expect_true (SvMAGIC (sv)) \
434	? CORO_MAGIC_NN (sv, type) \
435	: 0)
436
437	#define CORO_MAGIC_cv(cv) CORO_MAGIC (((SV *)(cv)), CORO_MAGIC_type_cv)
438	#define CORO_MAGIC_state(sv) CORO_MAGIC_NN (((SV *)(sv)), CORO_MAGIC_type_state)
439
440	INLINE struct coro *
441	SvSTATE_ (pTHX_ SV *coro)
442	{
443	HV *stash;
444	MAGIC *mg;
445
446	if (SvROK (coro))
447	coro = SvRV (coro);
448
449	if (expect_false (SvTYPE (coro) != SVt_PVHV))
450	croak ("Coro::State object required");
451
452	stash = SvSTASH (coro);
453	if (expect_false (stash != coro_stash && stash != coro_state_stash))
454	{
455	/* very slow, but rare, check */
456	if (!sv_derived_from (sv_2mortal (newRV_inc (coro)), "Coro::State"))
457	croak ("Coro::State object required");
458	}
459
460	mg = CORO_MAGIC_state (coro);
461	return (struct coro *)mg->mg_ptr;
462	}
463
464	#define SvSTATE(sv) SvSTATE_ (aTHX_ (sv))
465
466	/* faster than SvSTATE, but expects a coroutine hv */
467	#define SvSTATE_hv(hv) ((struct coro )CORO_MAGIC_NN ((SV )hv, CORO_MAGIC_type_state)->mg_ptr)
468	#define SvSTATE_current SvSTATE_hv (SvRV (coro_current))
469		494
470	/* the next two functions merely cache the padlists */	495	/* the next two functions merely cache the padlists */
471	static void	496	static void
472	get_padlist (pTHX_ CV *cv)	497	get_padlist (pTHX_ CV *cv)
473	{	498	{
…		…
509	}	534	}
510		535
511	/ load & save, init *****************************************************/	536	/ load & save, init *****************************************************/
512		537
513	static void	538	static void
		539	on_enterleave_call (pTHX_ SV *cb);
		540
		541	static void
514	load_perl (pTHX_ Coro__State c)	542	load_perl (pTHX_ Coro__State c)
515	{	543	{
516	perl_slots *slot = c->slot;	544	perl_slots *slot = c->slot;
517	c->slot = 0;	545	c->slot = 0;
518		546
…		…
544	PUTBACK;	572	PUTBACK;
545	}	573	}
546		574
547	slf_frame = c->slf_frame;	575	slf_frame = c->slf_frame;
548	CORO_THROW = c->except;	576	CORO_THROW = c->except;
		577
		578	if (expect_false (c->on_enter))
		579	{
		580	int i;
		581
		582	for (i = 0; i <= AvFILLp (c->on_enter); ++i)
		583	on_enterleave_call (aTHX_ AvARRAY (c->on_enter)[i]);
		584	}
549	}	585	}
550		586
551	static void	587	static void
552	save_perl (pTHX_ Coro__State c)	588	save_perl (pTHX_ Coro__State c)
553	{	589	{
		590	if (expect_false (c->on_leave))
		591	{
		592	int i;
		593
		594	for (i = AvFILLp (c->on_leave); i >= 0; --i)
		595	on_enterleave_call (aTHX_ AvARRAY (c->on_leave)[i]);
		596	}
		597
554	c->except = CORO_THROW;	598	c->except = CORO_THROW;
555	c->slf_frame = slf_frame;	599	c->slf_frame = slf_frame;
556		600
557	{	601	{
558	dSP;	602	dSP;
…		…
757	#endif	801	#endif
758		802
759	/*	803	/*
760	* This overrides the default magic get method of %SIG elements.	804	* This overrides the default magic get method of %SIG elements.
761	* The original one doesn't provide for reading back of PL_diehook/PL_warnhook	805	* The original one doesn't provide for reading back of PL_diehook/PL_warnhook
762	* and instead of tryign to save and restore the hash elements, we just provide	806	* and instead of trying to save and restore the hash elements, we just provide
763	* readback here.	807	* readback here.
764	* We only do this when the hook is != 0, as they are often set to 0 temporarily,
765	* not expecting this to actually change the hook. This is a potential problem
766	* when a schedule happens then, but we ignore this.
767	*/	808	*/
768	static int	809	static int
769	coro_sigelem_get (pTHX_ SV sv, MAGIC mg)	810	coro_sigelem_get (pTHX_ SV sv, MAGIC mg)
770	{	811	{
771	const char *s = MgPV_nolen_const (mg);	812	const char *s = MgPV_nolen_const (mg);
…		…
824	if (strEQ (s, "__WARN__")) svp = &PL_warnhook;	865	if (strEQ (s, "__WARN__")) svp = &PL_warnhook;
825		866
826	if (svp)	867	if (svp)
827	{	868	{
828	SV old = svp;	869	SV old = svp;
829	*svp = newSVsv (sv);	870	*svp = SvOK (sv) ? newSVsv (sv) : 0;
830	SvREFCNT_dec (old);	871	SvREFCNT_dec (old);
831	return 0;	872	return 0;
832	}	873	}
833	}	874	}
834		875
…		…
926	/* copy throw, in case it was set before coro_setup */	967	/* copy throw, in case it was set before coro_setup */
927	CORO_THROW = coro->except;	968	CORO_THROW = coro->except;
928	}	969	}
929		970
930	static void	971	static void
931	coro_destruct (pTHX_ struct coro *coro)	972	coro_unwind_stacks (pTHX)
932	{	973	{
933	if (!IN_DESTRUCT)	974	if (!IN_DESTRUCT)
934	{	975	{
935	/* restore all saved variables and stuff */	976	/* restore all saved variables and stuff */
936	LEAVE_SCOPE (0);	977	LEAVE_SCOPE (0);
…		…
944	POPSTACK_TO (PL_mainstack);	985	POPSTACK_TO (PL_mainstack);
945		986
946	/* unwind main stack */	987	/* unwind main stack */
947	dounwind (-1);	988	dounwind (-1);
948	}	989	}
		990	}
949		991
950	SvREFCNT_dec (GvSV (PL_defgv));	992	static void
951	SvREFCNT_dec (GvAV (PL_defgv));	993	coro_destruct_perl (pTHX_ struct coro *coro)
952	SvREFCNT_dec (GvSV (PL_errgv));	994	{
953	SvREFCNT_dec (PL_defoutgv);	995	SV *svf [9];
954	SvREFCNT_dec (PL_rs);
955	SvREFCNT_dec (GvSV (irsgv));
956	SvREFCNT_dec (GvHV (PL_hintgv));
957		996
958	SvREFCNT_dec (PL_diehook);
959	SvREFCNT_dec (PL_warnhook);
960		997	{
		998	struct coro *current = SvSTATE_current;
		999
		1000	assert (("FATAL: tried to destroy currently running coroutine", coro->mainstack != PL_mainstack));
		1001
		1002	save_perl (aTHX_ current);
		1003	load_perl (aTHX_ coro);
		1004
		1005	coro_unwind_stacks (aTHX);
		1006	coro_destruct_stacks (aTHX);
		1007
		1008	// now save some sv's to be free'd later
		1009	svf [0] = GvSV (PL_defgv);
		1010	svf [1] = (SV *)GvAV (PL_defgv);
		1011	svf [2] = GvSV (PL_errgv);
		1012	svf [3] = (SV *)PL_defoutgv;
		1013	svf [4] = PL_rs;
		1014	svf [5] = GvSV (irsgv);
		1015	svf [6] = (SV *)GvHV (PL_hintgv);
		1016	svf [7] = PL_diehook;
		1017	svf [8] = PL_warnhook;
		1018	assert (9 == sizeof (svf) / sizeof (*svf));
		1019
		1020	load_perl (aTHX_ current);
		1021	}
		1022
		1023	{
		1024	int i;
		1025
		1026	for (i = 0; i < sizeof (svf) / sizeof (*svf); ++i)
		1027	SvREFCNT_dec (svf [i]);
		1028
961	SvREFCNT_dec (coro->saved_deffh);	1029	SvREFCNT_dec (coro->saved_deffh);
962	SvREFCNT_dec (coro->rouse_cb);	1030	SvREFCNT_dec (coro->rouse_cb);
963	SvREFCNT_dec (coro->invoke_cb);	1031	SvREFCNT_dec (coro->invoke_cb);
964	SvREFCNT_dec (coro->invoke_av);	1032	SvREFCNT_dec (coro->invoke_av);
965		1033	}
966	coro_destruct_stacks (aTHX);
967	}	1034	}
968		1035
969	INLINE void	1036	INLINE void
970	free_coro_mortal (pTHX)	1037	free_coro_mortal (pTHX)
971	{	1038	{
…		…
1265	cctx_destroy (coro_cctx *cctx)	1332	cctx_destroy (coro_cctx *cctx)
1266	{	1333	{
1267	if (!cctx)	1334	if (!cctx)
1268	return;	1335	return;
1269		1336
1270	assert (cctx != cctx_current);//D temporary	1337	assert (("FATAL: tried to destroy current cctx", cctx != cctx_current));//D temporary?
1271		1338
1272	--cctx_count;	1339	--cctx_count;
1273	coro_destroy (&cctx->cctx);	1340	coro_destroy (&cctx->cctx);
1274		1341
1275	/* coro_transfer creates new, empty cctx's */	1342	/* coro_transfer creates new, empty cctx's */
…		…
1339	/* TODO: throwing up here is considered harmful */	1406	/* TODO: throwing up here is considered harmful */
1340		1407
1341	if (expect_true (prev != next))	1408	if (expect_true (prev != next))
1342	{	1409	{
1343	if (expect_false (!(prev->flags & (CF_RUNNING \| CF_NEW))))	1410	if (expect_false (!(prev->flags & (CF_RUNNING \| CF_NEW))))
1344	croak ("Coro::State::transfer called with a suspended prev Coro::State, but can only transfer from running or new states,");	1411	croak ("Coro::State::transfer called with a blocked prev Coro::State, but can only transfer from running or new states,");
1345		1412
1346	if (expect_false (next->flags & CF_RUNNING))
1347	croak ("Coro::State::transfer called with running next Coro::State, but can only transfer to inactive states,");
1348
1349	if (expect_false (next->flags & CF_DESTROYED))	1413	if (expect_false (next->flags & (CF_RUNNING \| CF_DESTROYED \| CF_SUSPENDED)))
1350	croak ("Coro::State::transfer called with destroyed next Coro::State, but can only transfer to inactive states,");	1414	croak ("Coro::State::transfer called with running, destroyed or suspended next Coro::State, but can only transfer to inactive states,");
1351		1415
1352	#if !PERL_VERSION_ATLEAST (5,10,0)	1416	#if !PERL_VERSION_ATLEAST (5,10,0)
1353	if (expect_false (PL_lex_state != LEX_NOTPARSING))	1417	if (expect_false (PL_lex_state != LEX_NOTPARSING))
1354	croak ("Coro::State::transfer called while parsing, but this is not supported in your perl version,");	1418	croak ("Coro::State::transfer called while parsing, but this is not supported in your perl version,");
1355	#endif	1419	#endif
…		…
1393	coro_setup (aTHX_ next);	1457	coro_setup (aTHX_ next);
1394	}	1458	}
1395	else	1459	else
1396	load_perl (aTHX_ next);	1460	load_perl (aTHX_ next);
1397		1461
1398	assert (!prev->cctx);//D temporary
1399
1400	/* possibly untie and reuse the cctx */	1462	/* possibly untie and reuse the cctx */
1401	if (expect_true (	1463	if (expect_true (
1402	cctx_current->idle_sp == STACKLEVEL	1464	cctx_current->idle_sp == STACKLEVEL
1403	&& !(cctx_current->flags & CC_TRACE)	1465	&& !(cctx_current->flags & CC_TRACE)
1404	&& !force_cctx	1466	&& !force_cctx
…		…
1445	coro_state_destroy (pTHX_ struct coro *coro)	1507	coro_state_destroy (pTHX_ struct coro *coro)
1446	{	1508	{
1447	if (coro->flags & CF_DESTROYED)	1509	if (coro->flags & CF_DESTROYED)
1448	return 0;	1510	return 0;
1449		1511
1450	if (coro->on_destroy)	1512	if (coro->on_destroy && !PL_dirty)
1451	coro->on_destroy (aTHX_ coro);	1513	coro->on_destroy (aTHX_ coro);
1452		1514
1453	coro->flags \|= CF_DESTROYED;	1515	coro->flags \|= CF_DESTROYED;
1454		1516
1455	if (coro->flags & CF_READY)	1517	if (coro->flags & CF_READY)
…		…
1459	--coro_nready;	1521	--coro_nready;
1460	}	1522	}
1461	else	1523	else
1462	coro->flags \|= CF_READY; /* make sure it is NOT put into the readyqueue */	1524	coro->flags \|= CF_READY; /* make sure it is NOT put into the readyqueue */
1463		1525
1464	if (coro->mainstack && coro->mainstack != main_mainstack)	1526	if (coro->mainstack
1465	{	1527	&& coro->mainstack != main_mainstack
1466	struct coro temp;	1528	&& coro->slot
1467		1529	&& !PL_dirty)
1468	assert (("FATAL: tried to destroy currently running coroutine (please report)", !(coro->flags & CF_RUNNING)));
1469
1470	save_perl (aTHX_ &temp);
1471	load_perl (aTHX_ coro);
1472
1473	coro_destruct (aTHX_ coro);	1530	coro_destruct_perl (aTHX_ coro);
1474
1475	load_perl (aTHX_ &temp);
1476
1477	coro->slot = 0;
1478	}
1479		1531
1480	cctx_destroy (coro->cctx);	1532	cctx_destroy (coro->cctx);
1481	SvREFCNT_dec (coro->startcv);	1533	SvREFCNT_dec (coro->startcv);
1482	SvREFCNT_dec (coro->args);	1534	SvREFCNT_dec (coro->args);
1483	SvREFCNT_dec (CORO_THROW);	1535	SvREFCNT_dec (CORO_THROW);
…		…
1570	/ Coro ******************************************************************/	1622	/ Coro ******************************************************************/
1571		1623
1572	INLINE void	1624	INLINE void
1573	coro_enq (pTHX_ struct coro *coro)	1625	coro_enq (pTHX_ struct coro *coro)
1574	{	1626	{
1575	av_push (coro_ready [coro->prio - PRIO_MIN], SvREFCNT_inc_NN (coro->hv));	1627	struct coro **ready = coro_ready [coro->prio - PRIO_MIN];
1576	}
1577		1628
1578	INLINE SV *	1629	SvREFCNT_inc_NN (coro->hv);
		1630
		1631	coro->next_ready = 0;
		1632	*(ready [0] ? &ready [1]->next_ready : &ready [0]) = coro;
		1633	ready [1] = coro;
		1634	}
		1635
		1636	INLINE struct coro *
1579	coro_deq (pTHX)	1637	coro_deq (pTHX)
1580	{	1638	{
1581	int prio;	1639	int prio;
1582		1640
1583	for (prio = PRIO_MAX - PRIO_MIN + 1; --prio >= 0; )	1641	for (prio = PRIO_MAX - PRIO_MIN + 1; --prio >= 0; )
1584	if (AvFILLp (coro_ready [prio]) >= 0)	1642	{
1585	return av_shift (coro_ready [prio]);	1643	struct coro **ready = coro_ready [prio];
		1644
		1645	if (ready [0])
		1646	{
		1647	struct coro *coro = ready [0];
		1648	ready [0] = coro->next_ready;
		1649	return coro;
		1650	}
		1651	}
1586		1652
1587	return 0;	1653	return 0;
1588	}	1654	}
1589		1655
1590	static int	1656	static int
…		…
1654	static void	1720	static void
1655	prepare_schedule (pTHX_ struct coro_transfer_args *ta)	1721	prepare_schedule (pTHX_ struct coro_transfer_args *ta)
1656	{	1722	{
1657	for (;;)	1723	for (;;)
1658	{	1724	{
1659	SV *next_sv = coro_deq (aTHX);	1725	struct coro *next = coro_deq (aTHX);
1660		1726
1661	if (expect_true (next_sv))	1727	if (expect_true (next))
1662	{	1728	{
1663	struct coro *next = SvSTATE_hv (next_sv);
1664
1665	/* cannot transfer to destroyed coros, skip and look for next */	1729	/* cannot transfer to destroyed coros, skip and look for next */
1666	if (expect_false (next->flags & CF_DESTROYED))	1730	if (expect_false (next->flags & (CF_DESTROYED \| CF_SUSPENDED)))
1667	SvREFCNT_dec (next_sv); /* coro_nready has already been taken care of by destroy */	1731	SvREFCNT_dec (next->hv); /* coro_nready has already been taken care of by destroy */
1668	else	1732	else
1669	{	1733	{
1670	next->flags &= ~CF_READY;	1734	next->flags &= ~CF_READY;
1671	--coro_nready;	1735	--coro_nready;
1672		1736
…		…
1850	av_push (av_destroy, (SV )newRV_inc ((SV )hv)); /* RVinc for perl */	1914	av_push (av_destroy, (SV )newRV_inc ((SV )hv)); /* RVinc for perl */
1851	api_ready (aTHX_ sv_manager);	1915	api_ready (aTHX_ sv_manager);
1852		1916
1853	frame->prepare = prepare_schedule;	1917	frame->prepare = prepare_schedule;
1854	frame->check = slf_check_repeat;	1918	frame->check = slf_check_repeat;
		1919
		1920	/* as a minor optimisation, we could unwind all stacks here */
		1921	/* but that puts extra pressure on pp_slf, and is not worth much */
		1922	/coro_unwind_stacks (aTHX);/
1855	}	1923	}
1856		1924
1857	/*****************************************************************************/	1925	/*****************************************************************************/
1858	/* async pool handler */	1926	/* async pool handler */
1859		1927
…		…
1942	SV data = (SV )GENSUB_ARG;	2010	SV data = (SV )GENSUB_ARG;
1943		2011
1944	if (SvTYPE (SvRV (data)) != SVt_PVAV)	2012	if (SvTYPE (SvRV (data)) != SVt_PVAV)
1945	{	2013	{
1946	/* first call, set args */	2014	/* first call, set args */
		2015	SV *coro = SvRV (data);
1947	AV *av = newAV ();	2016	AV *av = newAV ();
1948	SV *coro = SvRV (data);
1949		2017
1950	SvRV_set (data, (SV *)av);	2018	SvRV_set (data, (SV *)av);
1951	api_ready (aTHX_ coro);
1952	SvREFCNT_dec (coro);
1953		2019
1954	/* better take a full copy of the arguments */	2020	/* better take a full copy of the arguments */
1955	while (items--)	2021	while (items--)
1956	av_store (av, items, newSVsv (ST (items)));	2022	av_store (av, items, newSVsv (ST (items)));
		2023
		2024	api_ready (aTHX_ coro);
		2025	SvREFCNT_dec (coro);
1957	}	2026	}
1958		2027
1959	XSRETURN_EMPTY;	2028	XSRETURN_EMPTY;
1960	}	2029	}
1961		2030
…		…
1978		2047
1979	EXTEND (SP, AvFILLp (av) + 1);	2048	EXTEND (SP, AvFILLp (av) + 1);
1980	for (i = 0; i <= AvFILLp (av); ++i)	2049	for (i = 0; i <= AvFILLp (av); ++i)
1981	PUSHs (sv_2mortal (AvARRAY (av)[i]));	2050	PUSHs (sv_2mortal (AvARRAY (av)[i]));
1982		2051
1983	/* we have stolen the elements, so ste length to zero and free */	2052	/* we have stolen the elements, so set length to zero and free */
1984	AvFILLp (av) = -1;	2053	AvFILLp (av) = -1;
1985	av_undef (av);	2054	av_undef (av);
1986		2055
1987	PUTBACK;	2056	PUTBACK;
1988	}	2057	}
…		…
2278		2347
2279	PL_op->op_ppaddr = pp_slf;	2348	PL_op->op_ppaddr = pp_slf;
2280	/PL_op->op_type = OP_CUSTOM; / we do behave like entersub still */	2349	/PL_op->op_type = OP_CUSTOM; / we do behave like entersub still */
2281		2350
2282	PL_op = (OP *)&slf_restore;	2351	PL_op = (OP *)&slf_restore;
		2352	}
		2353
		2354	/*****************************************************************************/
		2355	/* dynamic wind */
		2356
		2357	static void
		2358	on_enterleave_call (pTHX_ SV *cb)
		2359	{
		2360	dSP;
		2361
		2362	PUSHSTACK;
		2363
		2364	PUSHMARK (SP);
		2365	PUTBACK;
		2366	call_sv (cb, G_VOID \| G_DISCARD);
		2367	SPAGAIN;
		2368
		2369	POPSTACK;
		2370	}
		2371
		2372	static SV *
		2373	coro_avp_pop_and_free (pTHX_ AV **avp)
		2374	{
		2375	AV av = avp;
		2376	SV *res = av_pop (av);
		2377
		2378	if (AvFILLp (av) < 0)
		2379	{
		2380	*avp = 0;
		2381	SvREFCNT_dec (av);
		2382	}
		2383
		2384	return res;
		2385	}
		2386
		2387	static void
		2388	coro_pop_on_enter (pTHX_ void *coro)
		2389	{
		2390	SV cb = coro_avp_pop_and_free (aTHX_ &((struct coro )coro)->on_enter);
		2391	SvREFCNT_dec (cb);
		2392	}
		2393
		2394	static void
		2395	coro_pop_on_leave (pTHX_ void *coro)
		2396	{
		2397	SV cb = coro_avp_pop_and_free (aTHX_ &((struct coro )coro)->on_leave);
		2398	on_enterleave_call (aTHX_ sv_2mortal (cb));
2283	}	2399	}
2284		2400
2285	/*****************************************************************************/	2401	/*****************************************************************************/
2286	/* PerlIO::cede */	2402	/* PerlIO::cede */
2287		2403
…		…
2913	SV *	3029	SV *
2914	clone (Coro::State coro)	3030	clone (Coro::State coro)
2915	CODE:	3031	CODE:
2916	{	3032	{
2917	#if CORO_CLONE	3033	#if CORO_CLONE
2918	struct coro *ncoro = coro_clone (coro);	3034	struct coro *ncoro = coro_clone (aTHX_ coro);
2919	MAGIC *mg;	3035	MAGIC *mg;
2920	/* TODO: too much duplication */	3036	/* TODO: too much duplication */
2921	ncoro->hv = newHV ();	3037	ncoro->hv = newHV ();
2922	mg = sv_magicext ((SV )ncoro->hv, 0, CORO_MAGIC_type_state, &coro_state_vtbl, (char )ncoro, 0);	3038	mg = sv_magicext ((SV )ncoro->hv, 0, CORO_MAGIC_type_state, &coro_state_vtbl, (char )ncoro, 0);
2923	mg->mg_flags \|= MGf_DUP;	3039	mg->mg_flags \|= MGf_DUP;
…		…
2985	eval = 1	3101	eval = 1
2986	CODE:	3102	CODE:
2987	{	3103	{
2988	if (coro->mainstack && ((coro->flags & CF_RUNNING) \|\| coro->slot))	3104	if (coro->mainstack && ((coro->flags & CF_RUNNING) \|\| coro->slot))
2989	{	3105	{
2990	struct coro temp;	3106	struct coro *current = SvSTATE_current;
2991		3107
2992	if (!(coro->flags & CF_RUNNING))	3108	if (current != coro)
2993	{	3109	{
2994	PUTBACK;	3110	PUTBACK;
2995	save_perl (aTHX_ &temp);	3111	save_perl (aTHX_ current);
2996	load_perl (aTHX_ coro);	3112	load_perl (aTHX_ coro);
		3113	SPAGAIN;
2997	}	3114	}
2998		3115
2999	{
3000	dSP;
3001	ENTER;
3002	SAVETMPS;
3003	PUTBACK;
3004	PUSHSTACK;	3116	PUSHSTACK;
		3117
3005	PUSHMARK (SP);	3118	PUSHMARK (SP);
		3119	PUTBACK;
3006		3120
3007	if (ix)	3121	if (ix)
3008	eval_sv (coderef, 0);	3122	eval_sv (coderef, 0);
3009	else	3123	else
3010	call_sv (coderef, G_KEEPERR \| G_EVAL \| G_VOID \| G_DISCARD);	3124	call_sv (coderef, G_KEEPERR \| G_EVAL \| G_VOID \| G_DISCARD);
3011		3125
3012	POPSTACK;	3126	POPSTACK;
3013	SPAGAIN;	3127	SPAGAIN;
3014	FREETMPS;
3015	LEAVE;
3016	PUTBACK;
3017	}
3018		3128
3019	if (!(coro->flags & CF_RUNNING))	3129	if (current != coro)
3020	{	3130	{
		3131	PUTBACK;
3021	save_perl (aTHX_ coro);	3132	save_perl (aTHX_ coro);
3022	load_perl (aTHX_ &temp);	3133	load_perl (aTHX_ current);
3023	SPAGAIN;	3134	SPAGAIN;
3024	}	3135	}
3025	}	3136	}
3026	}	3137	}
3027		3138
…		…
3031	ALIAS:	3142	ALIAS:
3032	is_ready = CF_READY	3143	is_ready = CF_READY
3033	is_running = CF_RUNNING	3144	is_running = CF_RUNNING
3034	is_new = CF_NEW	3145	is_new = CF_NEW
3035	is_destroyed = CF_DESTROYED	3146	is_destroyed = CF_DESTROYED
		3147	is_suspended = CF_SUSPENDED
3036	CODE:	3148	CODE:
3037	RETVAL = boolSV (coro->flags & ix);	3149	RETVAL = boolSV (coro->flags & ix);
3038	OUTPUT:	3150	OUTPUT:
3039	RETVAL	3151	RETVAL
3040		3152
…		…
3105	SV dst = ix ? (SV )&self->slot->defav : (SV **)&self->slot->defsv;	3217	SV dst = ix ? (SV )&self->slot->defav : (SV **)&self->slot->defsv;
3106		3218
3107	SV tmp = src; src = dst; *dst = tmp;	3219	SV tmp = src; src = dst; *dst = tmp;
3108	}	3220	}
3109		3221
		3222	void
		3223	cancel (Coro::State self)
		3224	CODE:
		3225	coro_state_destroy (aTHX_ self);
		3226	coro_call_on_destroy (aTHX_ self); /* actually only for Coro objects */
		3227
3110		3228
3111	MODULE = Coro::State PACKAGE = Coro	3229	MODULE = Coro::State PACKAGE = Coro
3112		3230
3113	BOOT:	3231	BOOT:
3114	{	3232	{
3115	int i;
3116
3117	sv_pool_rss = coro_get_sv (aTHX_ "Coro::POOL_RSS" , TRUE);	3233	sv_pool_rss = coro_get_sv (aTHX_ "Coro::POOL_RSS" , TRUE);
3118	sv_pool_size = coro_get_sv (aTHX_ "Coro::POOL_SIZE" , TRUE);	3234	sv_pool_size = coro_get_sv (aTHX_ "Coro::POOL_SIZE" , TRUE);
3119	cv_coro_run = get_cv ( "Coro::_terminate", GV_ADD);	3235	cv_coro_run = get_cv ( "Coro::_coro_run" , GV_ADD);
3120	cv_coro_terminate = get_cv ( "Coro::terminate" , GV_ADD);	3236	cv_coro_terminate = get_cv ( "Coro::terminate" , GV_ADD);
3121	coro_current = coro_get_sv (aTHX_ "Coro::current" , FALSE); SvREADONLY_on (coro_current);	3237	coro_current = coro_get_sv (aTHX_ "Coro::current" , FALSE); SvREADONLY_on (coro_current);
3122	av_async_pool = coro_get_av (aTHX_ "Coro::async_pool", TRUE);	3238	av_async_pool = coro_get_av (aTHX_ "Coro::async_pool", TRUE);
3123	av_destroy = coro_get_av (aTHX_ "Coro::destroy" , TRUE);	3239	av_destroy = coro_get_av (aTHX_ "Coro::destroy" , TRUE);
3124	sv_manager = coro_get_sv (aTHX_ "Coro::manager" , TRUE);	3240	sv_manager = coro_get_sv (aTHX_ "Coro::manager" , TRUE);
…		…
3135	newCONSTSUB (coro_stash, "PRIO_HIGH", newSViv (PRIO_HIGH));	3251	newCONSTSUB (coro_stash, "PRIO_HIGH", newSViv (PRIO_HIGH));
3136	newCONSTSUB (coro_stash, "PRIO_NORMAL", newSViv (PRIO_NORMAL));	3252	newCONSTSUB (coro_stash, "PRIO_NORMAL", newSViv (PRIO_NORMAL));
3137	newCONSTSUB (coro_stash, "PRIO_LOW", newSViv (PRIO_LOW));	3253	newCONSTSUB (coro_stash, "PRIO_LOW", newSViv (PRIO_LOW));
3138	newCONSTSUB (coro_stash, "PRIO_IDLE", newSViv (PRIO_IDLE));	3254	newCONSTSUB (coro_stash, "PRIO_IDLE", newSViv (PRIO_IDLE));
3139	newCONSTSUB (coro_stash, "PRIO_MIN", newSViv (PRIO_MIN));	3255	newCONSTSUB (coro_stash, "PRIO_MIN", newSViv (PRIO_MIN));
3140
3141	for (i = PRIO_MAX - PRIO_MIN + 1; i--; )
3142	coro_ready[i] = newAV ();
3143		3256
3144	{	3257	{
3145	SV *sv = coro_get_sv (aTHX_ "Coro::API", TRUE);	3258	SV *sv = coro_get_sv (aTHX_ "Coro::API", TRUE);
3146		3259
3147	coroapi.schedule = api_schedule;	3260	coroapi.schedule = api_schedule;
…		…
3188	cede_notself (...)	3301	cede_notself (...)
3189	CODE:	3302	CODE:
3190	CORO_EXECUTE_SLF_XS (slf_init_cede_notself);	3303	CORO_EXECUTE_SLF_XS (slf_init_cede_notself);
3191		3304
3192	void	3305	void
3193	_cancel (Coro::State self)
3194	CODE:
3195	coro_state_destroy (aTHX_ self);
3196	coro_call_on_destroy (aTHX_ self);
3197
3198	void
3199	_set_current (SV *current)	3306	_set_current (SV *current)
3200	PROTOTYPE: $	3307	PROTOTYPE: $
3201	CODE:	3308	CODE:
3202	SvREFCNT_dec (SvRV (coro_current));	3309	SvREFCNT_dec (SvRV (coro_current));
3203	SvRV_set (coro_current, SvREFCNT_inc_NN (SvRV (current)));	3310	SvRV_set (coro_current, SvREFCNT_inc_NN (SvRV (current)));
…		…
3245	PROTOTYPE:	3352	PROTOTYPE:
3246	CODE:	3353	CODE:
3247	RETVAL = coro_nready;	3354	RETVAL = coro_nready;
3248	OUTPUT:	3355	OUTPUT:
3249	RETVAL	3356	RETVAL
		3357
		3358	void
		3359	suspend (Coro::State self)
		3360	PROTOTYPE: $
		3361	CODE:
		3362	self->flags \|= CF_SUSPENDED;
		3363
		3364	void
		3365	resume (Coro::State self)
		3366	PROTOTYPE: $
		3367	CODE:
		3368	self->flags &= ~CF_SUSPENDED;
3250		3369
3251	void	3370	void
3252	_pool_handler (...)	3371	_pool_handler (...)
3253	CODE:	3372	CODE:
3254	CORO_EXECUTE_SLF_XS (slf_init_pool_handler);	3373	CORO_EXECUTE_SLF_XS (slf_init_pool_handler);
…		…
3309	rouse_wait (...)	3428	rouse_wait (...)
3310	PROTOTYPE: ;$	3429	PROTOTYPE: ;$
3311	PPCODE:	3430	PPCODE:
3312	CORO_EXECUTE_SLF_XS (slf_init_rouse_wait);	3431	CORO_EXECUTE_SLF_XS (slf_init_rouse_wait);
3313		3432
		3433	void
		3434	on_enter (SV *block)
		3435	ALIAS:
		3436	on_leave = 1
		3437	PROTOTYPE: &
		3438	CODE:
		3439	{
		3440	struct coro *coro = SvSTATE_current;
		3441	AV **avp = ix ? &coro->on_leave : &coro->on_enter;
		3442
		3443	block = (SV *)coro_sv_2cv (aTHX_ block);
		3444
		3445	if (!*avp)
		3446	*avp = newAV ();
		3447
		3448	av_push (*avp, SvREFCNT_inc (block));
		3449
		3450	if (!ix)
		3451	on_enterleave_call (aTHX_ block);
		3452
		3453	LEAVE; /* pp_entersub unfortunately forces an ENTER/LEAVE around XS calls */
		3454	SAVEDESTRUCTOR_X (ix ? coro_pop_on_leave : coro_pop_on_enter, (void *)coro);
		3455	ENTER; /* pp_entersub unfortunately forces an ENTER/LEAVE around XS calls */
		3456	}
		3457
3314		3458
3315	MODULE = Coro::State PACKAGE = PerlIO::cede	3459	MODULE = Coro::State PACKAGE = PerlIO::cede
3316		3460
3317	BOOT:	3461	BOOT:
3318	PerlIO_define_layer (aTHX_ &PerlIO_cede);	3462	PerlIO_define_layer (aTHX_ &PerlIO_cede);
…		…
3328	GvSTASH (CvGV (cv))	3472	GvSTASH (CvGV (cv))
3329	);	3473	);
3330	OUTPUT:	3474	OUTPUT:
3331	RETVAL	3475	RETVAL
3332		3476
3333	# helper for Coro::Channel	3477	# helper for Coro::Channel and others
3334	SV *	3478	SV *
3335	_alloc (int count)	3479	_alloc (int count)
3336	CODE:	3480	CODE:
3337	RETVAL = coro_waitarray_new (aTHX_ count);	3481	RETVAL = coro_waitarray_new (aTHX_ count);
3338	OUTPUT:	3482	OUTPUT:
…		…
3396	for (i = 1; i <= wcount; ++i)	3540	for (i = 1; i <= wcount; ++i)
3397	PUSHs (sv_2mortal (newRV_inc (AvARRAY (av)[i])));	3541	PUSHs (sv_2mortal (newRV_inc (AvARRAY (av)[i])));
3398	}	3542	}
3399	}	3543	}
3400		3544
		3545	MODULE = Coro::State PACKAGE = Coro::SemaphoreSet
		3546
		3547	void
		3548	_may_delete (SV *sem, int count, int extra_refs)
		3549	PPCODE:
		3550	{
		3551	AV av = (AV )SvRV (sem);
		3552
		3553	if (SvREFCNT ((SV *)av) == 1 + extra_refs
		3554	&& AvFILLp (av) == 0 /* no waiters, just count */
		3555	&& SvIV (AvARRAY (av)[0]) == count)
		3556	XSRETURN_YES;
		3557
		3558	XSRETURN_NO;
		3559	}
		3560
3401	MODULE = Coro::State PACKAGE = Coro::Signal	3561	MODULE = Coro::State PACKAGE = Coro::Signal
3402		3562
3403	SV *	3563	SV *
3404	new (SV *klass)	3564	new (SV *klass)
3405	CODE:	3565	CODE:

Diff Legend

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

Comparing Coro/Coro/State.xs (file contents): Revision 1.330 by root, Wed Nov 26 09:29:06 2008 UTC vs. Revision 1.352 by root, Sat Jun 20 08:58:50 2009 UTC

Diff Legend

Comparing Coro/Coro/State.xs (file contents):
Revision 1.330 by root, Wed Nov 26 09:29:06 2008 UTC vs.
Revision 1.352 by root, Sat Jun 20 08:58:50 2009 UTC