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

Comparing Coro/Coro/State.xs (file contents):
Revision 1.284 by root, Mon Nov 17 01:05:47 2008 UTC vs.
Revision 1.289 by root, Mon Nov 17 08:25:06 2008 UTC

…		…
95	# define GV_NOTQUAL 0	95	# define GV_NOTQUAL 0
96	#endif	96	#endif
97	#ifndef newSV	97	#ifndef newSV
98	# define newSV(l) NEWSV(0,l)	98	# define newSV(l) NEWSV(0,l)
99	#endif	99	#endif
		100	#ifndef CvISXSUB_on
		101	# define CvISXSUB_on(cv) (void)cv
		102	#endif
100		103
101	/* 5.8.7 */	104	/* 5.8.7 */
102	#ifndef SvRV_set	105	#ifndef SvRV_set
103	# define SvRV_set(s,v) SvRV(s) = (v)	106	# define SvRV_set(s,v) SvRV(s) = (v)
104	#endif	107	#endif
…		…
147	# if CORO_PTHREAD	150	# if CORO_PTHREAD
148	static void *coro_thx;	151	static void *coro_thx;
149	# endif	152	# endif
150	#endif	153	#endif
151		154
152	/* helper storage struct for Coro::AIO */
153	struct io_state
154	{
155	AV *res;
156	int errorno;
157	I32 laststype; /* U16 in 5.10.0 */
158	int laststatval;
159	Stat_t statcache;
160	};
161
162	static double (nvtime)(); / so why doesn't it take void? */	155	static double (nvtime)(); / so why doesn't it take void? */
163		156
164	static U32 cctx_gen;	157	static U32 cctx_gen;
165	static size_t cctx_stacksize = CORO_STACKSIZE;	158	static size_t cctx_stacksize = CORO_STACKSIZE;
166	static struct CoroAPI coroapi;	159	static struct CoroAPI coroapi;
…		…
241	/* this is a structure representing a perl-level coroutine */	234	/* this is a structure representing a perl-level coroutine */
242	struct coro {	235	struct coro {
243	/* the C coroutine allocated to this perl coroutine, if any */	236	/* the C coroutine allocated to this perl coroutine, if any */
244	coro_cctx *cctx;	237	coro_cctx *cctx;
245		238
246	/* process data */	239	/* state data */
247	struct CoroSLF slf_frame; /* saved slf frame */	240	struct CoroSLF slf_frame; /* saved slf frame */
248	AV *mainstack;	241	AV *mainstack;
249	perl_slots slot; / basically the saved sp */	242	perl_slots slot; / basically the saved sp */
250		243
251	AV args; / data associated with this coroutine (initial args) */	244	AV args; / data associated with this coroutine (initial args) */
…		…
269	};	262	};
270		263
271	typedef struct coro *Coro__State;	264	typedef struct coro *Coro__State;
272	typedef struct coro *Coro__State_or_hashref;	265	typedef struct coro *Coro__State_or_hashref;
273		266
		267	/* the following variables are effectively part of the perl context */
		268	/* and get copied between struct coro and these variables */
		269	/* the mainr easonw e don't support windows process emulation */
274	static struct CoroSLF slf_frame; /* the current slf frame */	270	static struct CoroSLF slf_frame; /* the current slf frame */
		271	static SV *coro_throw;
275		272
276	/ Coro ******************************************************************/	273	/ Coro ******************************************************************/
277		274
278	#define PRIO_MAX 3	275	#define PRIO_MAX 3
279	#define PRIO_HIGH 1	276	#define PRIO_HIGH 1
…		…
380	SvREFCNT_dec (av); /* sv_magicext increased the refcount */	377	SvREFCNT_dec (av); /* sv_magicext increased the refcount */
381		378
382	return 0;	379	return 0;
383	}	380	}
384		381
385	#define CORO_MAGIC_type_cv PERL_MAGIC_ext	382	#define CORO_MAGIC_type_cv 26
386	#define CORO_MAGIC_type_state PERL_MAGIC_ext	383	#define CORO_MAGIC_type_state PERL_MAGIC_ext
387		384
388	static MGVTBL coro_cv_vtbl = {	385	static MGVTBL coro_cv_vtbl = {
389	0, 0, 0, 0,	386	0, 0, 0, 0,
390	coro_cv_free	387	coro_cv_free
391	};	388	};
392		389
		390	#define CORO_MAGIC_NN(sv, type) \
		391	(expect_true (SvMAGIC (sv)->mg_type == type) \
		392	? SvMAGIC (sv) \
		393	: mg_find (sv, type))
		394
393	#define CORO_MAGIC(sv, type) \	395	#define CORO_MAGIC(sv, type) \
394	expect_true (SvMAGIC (sv)) \	396	(expect_true (SvMAGIC (sv)) \
395	? expect_true (SvMAGIC (sv)->mg_type == type) \	397	? CORO_MAGIC_NN (sv, type) \
396	? SvMAGIC (sv) \
397	: mg_find (sv, type) \
398	: 0	398	: 0)
399		399
400	#define CORO_MAGIC_cv(cv) CORO_MAGIC (((SV *)(cv)), CORO_MAGIC_type_cv)	400	#define CORO_MAGIC_cv(cv) CORO_MAGIC (((SV *)(cv)), CORO_MAGIC_type_cv)
401	#define CORO_MAGIC_state(sv) CORO_MAGIC (((SV *)(sv)), CORO_MAGIC_type_state)	401	#define CORO_MAGIC_state(sv) CORO_MAGIC_NN (((SV *)(sv)), CORO_MAGIC_type_state)
402		402
403	INLINE struct coro *	403	INLINE struct coro *
404	SvSTATE_ (pTHX_ SV *coro)	404	SvSTATE_ (pTHX_ SV *coro)
405	{	405	{
406	HV *stash;	406	HV *stash;
…		…
424	return (struct coro *)mg->mg_ptr;	424	return (struct coro *)mg->mg_ptr;
425	}	425	}
426		426
427	#define SvSTATE(sv) SvSTATE_ (aTHX_ (sv))	427	#define SvSTATE(sv) SvSTATE_ (aTHX_ (sv))
428		428
429	/* fastert than SvSTATE, but expects a coroutine hv */	429	/* faster than SvSTATE, but expects a coroutine hv */
430	INLINE struct coro *	430	#define SvSTATE_hv(hv) ((struct coro )CORO_MAGIC_NN ((SV )hv, CORO_MAGIC_type_state)->mg_ptr)
431	SvSTATE_hv (SV *sv)
432	{
433	MAGIC *mg = expect_true (SvMAGIC (sv)->mg_type == CORO_MAGIC_type_state)
434	? SvMAGIC (sv)
435	: mg_find (sv, CORO_MAGIC_type_state);
436
437	return (struct coro *)mg->mg_ptr;
438	}
439
440	#define SvSTATE_current SvSTATE_hv (SvRV (coro_current))	431	#define SvSTATE_current SvSTATE_hv (SvRV (coro_current))
441		432
442	/* the next two functions merely cache the padlists */	433	/* the next two functions merely cache the padlists */
443	static void	434	static void
444	get_padlist (pTHX_ CV *cv)	435	get_padlist (pTHX_ CV *cv)
…		…
513	}	504	}
514		505
515	PUTBACK;	506	PUTBACK;
516	}	507	}
517		508
518	slf_frame = c->slf_frame;	509	slf_frame = c->slf_frame;
		510	coro_throw = c->throw;
519	}	511	}
520		512
521	static void	513	static void
522	save_perl (pTHX_ Coro__State c)	514	save_perl (pTHX_ Coro__State c)
523	{	515	{
		516	c->throw = coro_throw;
524	c->slf_frame = slf_frame;	517	c->slf_frame = slf_frame;
525		518
526	{	519	{
527	dSP;	520	dSP;
528	I32 cxix = cxstack_ix;	521	I32 cxix = cxstack_ix;
…		…
821	slf_check_nop (pTHX_ struct CoroSLF *frame)	814	slf_check_nop (pTHX_ struct CoroSLF *frame)
822	{	815	{
823	return 0;	816	return 0;
824	}	817	}
825		818
826	static void	819	static void NOINLINE /* noinline to keep it out of the transfer fast path */
827	coro_setup (pTHX_ struct coro *coro)	820	coro_setup (pTHX_ struct coro *coro)
828	{	821	{
829	/*	822	/*
830	* emulate part of the perl startup here.	823	* emulate part of the perl startup here.
831	*/	824	*/
…		…
874	/* this newly created coroutine might be run on an existing cctx which most	867	/* this newly created coroutine might be run on an existing cctx which most
875	* likely was suspended in pp_slf, so we have to emulate entering pp_slf here.	868	* likely was suspended in pp_slf, so we have to emulate entering pp_slf here.
876	*/	869	*/
877	slf_frame.prepare = prepare_nop; /* provide a nop function for an eventual pp_slf */	870	slf_frame.prepare = prepare_nop; /* provide a nop function for an eventual pp_slf */
878	slf_frame.check = slf_check_nop; /* signal pp_slf to not repeat */	871	slf_frame.check = slf_check_nop; /* signal pp_slf to not repeat */
		872
		873	coro_throw = coro->throw;
879	}	874	}
880		875
881	static void	876	static void
882	coro_destruct (pTHX_ struct coro *coro)	877	coro_destruct (pTHX_ struct coro *coro)
883	{	878	{
…		…
907		902
908	SvREFCNT_dec (PL_diehook);	903	SvREFCNT_dec (PL_diehook);
909	SvREFCNT_dec (PL_warnhook);	904	SvREFCNT_dec (PL_warnhook);
910		905
911	SvREFCNT_dec (coro->saved_deffh);	906	SvREFCNT_dec (coro->saved_deffh);
912	SvREFCNT_dec (coro->throw);	907	SvREFCNT_dec (coro_throw);
913		908
914	coro_destruct_stacks (aTHX);	909	coro_destruct_stacks (aTHX);
915	}	910	}
916		911
917	INLINE void	912	INLINE void
…		…
1276	/ coroutine switching ***************************************************/	1271	/ coroutine switching ***************************************************/
1277		1272
1278	static void	1273	static void
1279	transfer_check (pTHX_ struct coro prev, struct coro next)	1274	transfer_check (pTHX_ struct coro prev, struct coro next)
1280	{	1275	{
		1276	/* TODO: throwing up here is considered harmful */
		1277
1281	if (expect_true (prev != next))	1278	if (expect_true (prev != next))
1282	{	1279	{
1283	if (expect_false (!(prev->flags & (CF_RUNNING \| CF_NEW))))	1280	if (expect_false (!(prev->flags & (CF_RUNNING \| CF_NEW))))
1284	croak ("Coro::State::transfer called with non-running/new prev Coro::State, but can only transfer from running or new states,");	1281	croak ("Coro::State::transfer called with non-running/new prev Coro::State, but can only transfer from running or new states,");
1285		1282
…		…
1295	#endif	1292	#endif
1296	}	1293	}
1297	}	1294	}
1298		1295
1299	/* always use the TRANSFER macro */	1296	/* always use the TRANSFER macro */
1300	static void NOINLINE	1297	static void NOINLINE /* noinline so we have a fixed stackframe */
1301	transfer (pTHX_ struct coro prev, struct coro next, int force_cctx)	1298	transfer (pTHX_ struct coro prev, struct coro next, int force_cctx)
1302	{	1299	{
1303	dSTACKLEVEL;	1300	dSTACKLEVEL;
1304		1301
1305	/* sometimes transfer is only called to set idle_sp */	1302	/* sometimes transfer is only called to set idle_sp */
…		…
1687	if (coro->flags & CF_RUNNING)	1684	if (coro->flags & CF_RUNNING)
1688	PL_runops = RUNOPS_DEFAULT;	1685	PL_runops = RUNOPS_DEFAULT;
1689	else	1686	else
1690	coro->slot->runops = RUNOPS_DEFAULT;	1687	coro->slot->runops = RUNOPS_DEFAULT;
1691	}	1688	}
		1689	}
		1690
		1691	/*****************************************************************************/
		1692	/* schedule-like-function opcode (SLF) */
		1693
		1694	static UNOP slf_restore; /* restore stack as entersub did, for first-re-run */
		1695	static const CV *slf_cv;
		1696	static SV **slf_argv;
		1697	static int slf_argc, slf_arga; /* count, allocated */
		1698	static I32 slf_ax; /* top of stack, for restore */
		1699
		1700	/* this restores the stack in the case we patched the entersub, to */
		1701	/* recreate the stack frame as perl will on following calls */
		1702	/* since entersub cleared the stack */
		1703	static OP *
		1704	pp_restore (pTHX)
		1705	{
		1706	int i;
		1707	SV **SP = PL_stack_base + slf_ax;
		1708
		1709	PUSHMARK (SP);
		1710
		1711	EXTEND (SP, slf_argc + 1);
		1712
		1713	for (i = 0; i < slf_argc; ++i)
		1714	PUSHs (sv_2mortal (slf_argv [i]));
		1715
		1716	PUSHs ((SV *)CvGV (slf_cv));
		1717
		1718	RETURNOP (slf_restore.op_first);
		1719	}
		1720
		1721	static void
		1722	slf_prepare_set_stacklevel (pTHX_ struct coro_transfer_args *ta)
		1723	{
		1724	prepare_set_stacklevel (ta, (struct coro_cctx *)slf_frame.data);
		1725	}
		1726
		1727	static void
		1728	slf_init_set_stacklevel (pTHX_ struct CoroSLF frame, CV cv, SV **arg, int items)
		1729	{
		1730	assert (("FATAL: set_stacklevel needs the coro cctx as sole argument", items == 1));
		1731
		1732	frame->prepare = slf_prepare_set_stacklevel;
		1733	frame->check = slf_check_nop;
		1734	frame->data = (void *)SvIV (arg [0]);
		1735	}
		1736
		1737	static void
		1738	slf_prepare_transfer (pTHX_ struct coro_transfer_args *ta)
		1739	{
		1740	SV arg = (SV )slf_frame.data;
		1741
		1742	prepare_transfer (aTHX_ ta, arg [0], arg [1]);
		1743	}
		1744
		1745	static void
		1746	slf_init_transfer (pTHX_ struct CoroSLF frame, CV cv, SV **arg, int items)
		1747	{
		1748	if (items != 2)
		1749	croak ("Coro::State::transfer (prev, next) expects two arguments, not %d,", items);
		1750
		1751	frame->prepare = slf_prepare_transfer;
		1752	frame->check = slf_check_nop;
		1753	frame->data = (void )arg; / let's hope it will stay valid */
		1754	}
		1755
		1756	static void
		1757	slf_init_schedule (pTHX_ struct CoroSLF frame, CV cv, SV **arg, int items)
		1758	{
		1759	frame->prepare = prepare_schedule;
		1760	frame->check = slf_check_nop;
		1761	}
		1762
		1763	static void
		1764	slf_init_cede (pTHX_ struct CoroSLF frame, CV cv, SV **arg, int items)
		1765	{
		1766	frame->prepare = prepare_cede;
		1767	frame->check = slf_check_nop;
		1768	}
		1769
		1770	static void
		1771	slf_init_cede_notself (pTHX_ struct CoroSLF frame, CV cv, SV **arg, int items)
		1772	{
		1773	frame->prepare = prepare_cede_notself;
		1774	frame->check = slf_check_nop;
		1775	}
		1776
		1777	/* we hijack an hopefully unused CV flag for our purposes */
		1778	#define CVf_SLF 0x4000
		1779
		1780	/*
		1781	* these not obviously related functions are all rolled into one
		1782	* function to increase chances that they all will call transfer with the same
		1783	* stack offset
		1784	* SLF stands for "schedule-like-function".
		1785	*/
		1786	static OP *
		1787	pp_slf (pTHX)
		1788	{
		1789	I32 checkmark; /* mark SP to see how many elements check has pushed */
		1790
		1791	/* set up the slf frame, unless it has already been set-up */
		1792	/* the latter happens when a new coro has been started */
		1793	/* or when a new cctx was attached to an existing coroutine */
		1794	if (expect_true (!slf_frame.prepare))
		1795	{
		1796	/* first iteration */
		1797	dSP;
		1798	SV **arg = PL_stack_base + TOPMARK + 1;
		1799	int items = SP - arg; /* args without function object */
		1800	SV gv = sp;
		1801
		1802	/* do a quick consistency check on the "function" object, and if it isn't */
		1803	/* for us, divert to the real entersub */
		1804	if (SvTYPE (gv) != SVt_PVGV
		1805	\|\| !GvCV (gv)
		1806	\|\| !(CvFLAGS (GvCV (gv)) & CVf_SLF))
		1807	return PL_ppaddr[OP_ENTERSUB](aTHX);
		1808
		1809	if (!(PL_op->op_flags & OPf_STACKED))
		1810	{
		1811	/* ampersand-form of call, use @_ instead of stack */
		1812	AV *av = GvAV (PL_defgv);
		1813	arg = AvARRAY (av);
		1814	items = AvFILLp (av) + 1;
		1815	}
		1816
		1817	/* now call the init function, which needs to set up slf_frame */
		1818	((coro_slf_cb)CvXSUBANY (GvCV (gv)).any_ptr)
		1819	(aTHX_ &slf_frame, GvCV (gv), arg, items);
		1820
		1821	/* pop args */
		1822	SP = PL_stack_base + POPMARK;
		1823
		1824	PUTBACK;
		1825	}
		1826
		1827	/* now that we have a slf_frame, interpret it! */
		1828	/* we use a callback system not to make the code needlessly */
		1829	/* complicated, but so we can run multiple perl coros from one cctx */
		1830
		1831	do
		1832	{
		1833	struct coro_transfer_args ta;
		1834
		1835	slf_frame.prepare (aTHX_ &ta);
		1836	TRANSFER (ta, 0);
		1837
		1838	checkmark = PL_stack_sp - PL_stack_base;
		1839	}
		1840	while (slf_frame.check (aTHX_ &slf_frame));
		1841
		1842	slf_frame.prepare = 0; /* invalidate the frame, we are done processing it */
		1843
		1844	/* return value handling - mostly like entersub */
		1845	{
		1846	dSP;
		1847	SV **bot = PL_stack_base + checkmark;
		1848	int gimme = GIMME_V;
		1849
		1850	/* make sure we put something on the stack in scalar context */
		1851	if (gimme == G_SCALAR)
		1852	{
		1853	if (sp == bot)
		1854	XPUSHs (&PL_sv_undef);
		1855
		1856	SP = bot + 1;
		1857	}
		1858
		1859	PUTBACK;
		1860	}
		1861
		1862	/* exception handling */
		1863	if (expect_false (coro_throw))
		1864	{
		1865	SV *exception = sv_2mortal (coro_throw);
		1866
		1867	coro_throw = 0;
		1868	sv_setsv (ERRSV, exception);
		1869	croak (0);
		1870	}
		1871
		1872	return NORMAL;
		1873	}
		1874
		1875	static void
		1876	api_execute_slf (pTHX_ CV *cv, coro_slf_cb init_cb, I32 ax)
		1877	{
		1878	int i;
		1879	SV **arg = PL_stack_base + ax;
		1880	int items = PL_stack_sp - arg + 1;
		1881
		1882	assert (("FATAL: SLF call with illegal CV value", !CvANON (cv)));
		1883
		1884	if (PL_op->op_ppaddr != PL_ppaddr [OP_ENTERSUB]
		1885	&& PL_op->op_ppaddr != pp_slf)
		1886	croak ("FATAL: Coro SLF calls can only be made normally, not via goto or any other means, caught");
		1887
		1888	CvFLAGS (cv) \|= CVf_SLF;
		1889	CvXSUBANY (cv).any_ptr = (void *)init_cb;
		1890	slf_cv = cv;
		1891
		1892	/* we patch the op, and then re-run the whole call */
		1893	/* we have to put the same argument on the stack for this to work */
		1894	/* and this will be done by pp_restore */
		1895	slf_restore.op_next = (OP *)&slf_restore;
		1896	slf_restore.op_type = OP_CUSTOM;
		1897	slf_restore.op_ppaddr = pp_restore;
		1898	slf_restore.op_first = PL_op;
		1899
		1900	slf_ax = ax - 1; /* undo the ax++ inside dAXMARK */
		1901
		1902	if (PL_op->op_flags & OPf_STACKED)
		1903	{
		1904	if (items > slf_arga)
		1905	{
		1906	slf_arga = items;
		1907	free (slf_argv);
		1908	slf_argv = malloc (slf_arga * sizeof (SV *));
		1909	}
		1910
		1911	slf_argc = items;
		1912
		1913	for (i = 0; i < items; ++i)
		1914	slf_argv [i] = SvREFCNT_inc (arg [i]);
		1915	}
		1916	else
		1917	slf_argc = 0;
		1918
		1919	PL_op->op_ppaddr = pp_slf;
		1920	PL_op->op_type = OP_CUSTOM; /* maybe we should leave it at entersub? */
		1921
		1922	PL_op = (OP *)&slf_restore;
1692	}	1923	}
1693		1924
1694	/*****************************************************************************/	1925	/*****************************************************************************/
1695	/* PerlIO::cede */	1926	/* PerlIO::cede */
1696		1927
…		…
1765	PerlIOBuf_get_cnt,	1996	PerlIOBuf_get_cnt,
1766	PerlIOBuf_set_ptrcnt,	1997	PerlIOBuf_set_ptrcnt,
1767	};	1998	};
1768		1999
1769	/*****************************************************************************/	2000	/*****************************************************************************/
1770		2001	/* Coro::Semaphore */
1771	static UNOP slf_restore; /* restore stack as entersub did, for first-re-run */
1772	static const CV *slf_cv;
1773	static SV *slf_arg0;
1774	static SV *slf_arg1;
1775	static SV *slf_arg2;
1776	static I32 slf_ax; /* top of stack, for restore */
1777
1778	/* this restores the stack in the case we patched the entersub, to */
1779	/* recreate the stack frame as perl will on following calls */
1780	/* since entersub cleared the stack */
1781	static OP *
1782	pp_restore (pTHX)
1783	{
1784	SV **SP = PL_stack_base + slf_ax;
1785
1786	PUSHMARK (SP);
1787
1788	EXTEND (SP, 3);
1789	if (slf_arg0) PUSHs (sv_2mortal (slf_arg0));
1790	if (slf_arg1) PUSHs (sv_2mortal (slf_arg1));
1791	if (slf_arg2) PUSHs (sv_2mortal (slf_arg2));
1792	PUSHs ((SV *)CvGV (slf_cv));
1793
1794	RETURNOP (slf_restore.op_first);
1795	}
1796
1797	static void
1798	slf_prepare_set_stacklevel (pTHX_ struct coro_transfer_args *ta)
1799	{
1800	prepare_set_stacklevel (ta, (struct coro_cctx *)slf_frame.data);
1801	}
1802
1803	static void
1804	slf_init_set_stacklevel (pTHX_ struct CoroSLF frame, CV cv, SV **arg, int items)
1805	{
1806	assert (("FATAL: set_stacklevel needs the coro cctx as sole argument", items == 1));
1807
1808	frame->prepare = slf_prepare_set_stacklevel;
1809	frame->check = slf_check_nop;
1810	frame->data = (void *)SvIV (arg [0]);
1811	}
1812
1813	static void
1814	slf_prepare_transfer (pTHX_ struct coro_transfer_args *ta)
1815	{
1816	SV arg = (SV )slf_frame.data;
1817
1818	prepare_transfer (aTHX_ ta, arg [0], arg [1]);
1819
1820	/* if the destination has ->throw set, then copy it */
1821	/* into the current coro's throw slot, so it will be raised */
1822	/* after the schedule */
1823	if (expect_false (ta->next->throw))
1824	{
1825	struct coro *coro = SvSTATE_current;
1826	SvREFCNT_dec (coro->throw);
1827	coro->throw = ta->next->throw;
1828	ta->next->throw = 0;
1829	}
1830	}
1831
1832	static void
1833	slf_init_transfer (pTHX_ struct CoroSLF frame, CV cv, SV **arg, int items)
1834	{
1835	if (items != 2)
1836	croak ("Coro::State::transfer (prev, next) expects two arguments, not %d,", items);
1837
1838	frame->prepare = slf_prepare_transfer;
1839	frame->check = slf_check_nop;
1840	frame->data = (void )arg; / let's hope it will stay valid */
1841	}
1842
1843	static void
1844	slf_init_schedule (pTHX_ struct CoroSLF frame, CV cv, SV **arg, int items)
1845	{
1846	frame->prepare = prepare_schedule;
1847	frame->check = slf_check_nop;
1848	}
1849
1850	static void
1851	slf_init_cede (pTHX_ struct CoroSLF frame, CV cv, SV **arg, int items)
1852	{
1853	frame->prepare = prepare_cede;
1854	frame->check = slf_check_nop;
1855	}
1856
1857	static void
1858	slf_init_cede_notself (pTHX_ struct CoroSLF frame, CV cv, SV **arg, int items)
1859	{
1860	frame->prepare = prepare_cede_notself;
1861	frame->check = slf_check_nop;
1862	}
1863
1864	/* we hijack an hopefully unused CV flag for our purposes */
1865	#define CVf_SLF 0x4000
1866
1867	/*
1868	* these not obviously related functions are all rolled into one
1869	* function to increase chances that they all will call transfer with the same
1870	* stack offset
1871	* SLF stands for "schedule-like-function".
1872	*/
1873	static OP *
1874	pp_slf (pTHX)
1875	{
1876	I32 checkmark; /* mark SP to see how many elements check has pushed */
1877
1878	/* set up the slf frame, unless it has already been set-up */
1879	/* the latter happens when a new coro has been started */
1880	/* or when a new cctx was attached to an existing coroutine */
1881	if (expect_true (!slf_frame.prepare))
1882	{
1883	/* first iteration */
1884	dSP;
1885	SV **arg = PL_stack_base + TOPMARK + 1;
1886	int items = SP - arg; /* args without function object */
1887	SV gv = sp;
1888
1889	/* do a quick consistency check on the "function" object, and if it isn't */
1890	/* for us, divert to the real entersub */
1891	if (SvTYPE (gv) != SVt_PVGV \|\| !(CvFLAGS (GvCV (gv)) & CVf_SLF))
1892	return PL_ppaddr[OP_ENTERSUB](aTHX);
1893
1894	if (!(PL_op->op_flags & OPf_STACKED))
1895	{
1896	/* ampersand-form of call, use @_ instead of stack */
1897	AV *av = GvAV (PL_defgv);
1898	arg = AvARRAY (av);
1899	items = AvFILLp (av) + 1;
1900	}
1901
1902	/* now call the init function, which needs to set up slf_frame */
1903	((coro_slf_cb)CvXSUBANY (GvCV (gv)).any_ptr)
1904	(aTHX_ &slf_frame, GvCV (gv), arg, items);
1905
1906	/* pop args */
1907	SP = PL_stack_base + POPMARK;
1908
1909	PUTBACK;
1910	}
1911
1912	/* now that we have a slf_frame, interpret it! */
1913	/* we use a callback system not to make the code needlessly */
1914	/* complicated, but so we can run multiple perl coros from one cctx */
1915
1916	do
1917	{
1918	struct coro_transfer_args ta;
1919
1920	slf_frame.prepare (aTHX_ &ta);
1921	TRANSFER (ta, 0);
1922
1923	checkmark = PL_stack_sp - PL_stack_base;
1924	}
1925	while (slf_frame.check (aTHX_ &slf_frame));
1926
1927	{
1928	dSP;
1929	SV **bot = PL_stack_base + checkmark;
1930	int gimme = GIMME_V;
1931
1932	slf_frame.prepare = 0; /* invalidate the frame, so it gets initialised again next time */
1933
1934	/* make sure we put something on the stack in scalar context */
1935	if (gimme == G_SCALAR)
1936	{
1937	if (sp == bot)
1938	XPUSHs (&PL_sv_undef);
1939
1940	SP = bot + 1;
1941	}
1942
1943	PUTBACK;
1944	}
1945
1946	{
1947	struct coro *coro = SvSTATE_current;
1948
1949	if (expect_false (coro->throw))
1950	{
1951	SV *exception = sv_2mortal (coro->throw);
1952
1953	coro->throw = 0;
1954	sv_setsv (ERRSV, exception);
1955	croak (0);
1956	}
1957	}
1958
1959	return NORMAL;
1960	}
1961
1962	static void
1963	api_execute_slf (pTHX_ CV *cv, coro_slf_cb init_cb, I32 ax)
1964	{
1965	SV **arg = PL_stack_base + ax;
1966	int items = PL_stack_sp - arg + 1;
1967
1968	assert (("FATAL: SLF call with illegal CV value", !CvANON (cv)));
1969
1970	if (PL_op->op_ppaddr != PL_ppaddr [OP_ENTERSUB]
1971	&& PL_op->op_ppaddr != pp_slf)
1972	croak ("FATAL: Coro SLF calls can only be made normally, not via goto or any other means, caught");
1973
1974	#if 0
1975	if (items > 3)
1976	croak ("Coro only supports up to three arguments to SLF functions currently (not %d), caught", items);
1977	#endif
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_CUSTOM;
1988	slf_restore.op_ppaddr = pp_restore;
1989	slf_restore.op_first = PL_op;
1990
1991	slf_ax = ax - 1; /* undo the ax++ inside dAXMARK */
1992	slf_arg0 = items > 0 ? SvREFCNT_inc (arg [0]) : 0;
1993	slf_arg1 = items > 1 ? SvREFCNT_inc (arg [1]) : 0;
1994	slf_arg2 = items > 2 ? SvREFCNT_inc (arg [2]) : 0;
1995
1996	PL_op->op_ppaddr = pp_slf;
1997	PL_op->op_type = OP_CUSTOM; /* maybe we should leave it at entersub? */
1998
1999	PL_op = (OP *)&slf_restore;
2000	}
2001
2002	/*****************************************************************************/
2003		2002
2004	static void	2003	static void
2005	coro_semaphore_adjust (pTHX_ AV *av, IV adjust)	2004	coro_semaphore_adjust (pTHX_ AV *av, IV adjust)
2006	{	2005	{
2007	SV *count_sv = AvARRAY (av)[0];	2006	SV *count_sv = AvARRAY (av)[0];
…		…
2093	frame->check = slf_check_semaphore_down;	2092	frame->check = slf_check_semaphore_down;
2094		2093
2095	}	2094	}
2096		2095
2097	/*****************************************************************************/	2096	/*****************************************************************************/
		2097	/* gensub: simple closure generation utility */
2098		2098
2099	#define GENSUB_ARG CvXSUBANY (cv).any_ptr	2099	#define GENSUB_ARG CvXSUBANY (cv).any_ptr
2100		2100
2101	/* create a closure from XS, returns a code reference */	2101	/* create a closure from XS, returns a code reference */
2102	/* the arg can be accessed via GENSUB_ARG from the callback */	2102	/* the arg can be accessed via GENSUB_ARG from the callback */
2103	/* the callback must use dXSARGS/XSRETURN */	2103	/* the callback must use dXSARGS/XSRETURN */
2104	static SV *	2104	static SV *
2105	gensub (pTHX_ void (xsub)(pTHX_ CV ), void *arg)	2105	gensub (pTHX_ void (xsub)(pTHX_ CV ), void *arg)
2106	{	2106	{
2107	CV cv = (CV )NEWSV (0, 0);	2107	CV cv = (CV )newSV (0);
2108		2108
2109	sv_upgrade ((SV *)cv, SVt_PVCV);	2109	sv_upgrade ((SV *)cv, SVt_PVCV);
2110		2110
2111	CvANON_on (cv);	2111	CvANON_on (cv);
2112	CvISXSUB_on (cv);	2112	CvISXSUB_on (cv);
2113	CvXSUB (cv) = xsub;	2113	CvXSUB (cv) = xsub;
2114	GENSUB_ARG = arg;	2114	GENSUB_ARG = arg;
2115		2115
2116	return newRV_noinc ((SV *)cv);	2116	return newRV_noinc ((SV *)cv);
		2117	}
		2118
		2119	/*****************************************************************************/
		2120	/* Coro::AIO */
		2121
		2122	#define CORO_MAGIC_type_aio PERL_MAGIC_ext
		2123
		2124	/* helper storage struct */
		2125	struct io_state
		2126	{
		2127	int errorno;
		2128	I32 laststype; /* U16 in 5.10.0 */
		2129	int laststatval;
		2130	Stat_t statcache;
		2131	};
		2132
		2133	static void
		2134	coro_aio_callback (pTHX_ CV *cv)
		2135	{
		2136	dXSARGS;
		2137	AV state = (AV )GENSUB_ARG;
		2138	SV *coro = av_pop (state);
		2139	SV *data_sv = newSV (sizeof (struct io_state));
		2140
		2141	av_extend (state, items);
		2142
		2143	sv_upgrade (data_sv, SVt_PV);
		2144	SvCUR_set (data_sv, sizeof (struct io_state));
		2145	SvPOK_only (data_sv);
		2146
		2147	{
		2148	struct io_state data = (struct io_state )SvPVX (data_sv);
		2149
		2150	data->errorno = errno;
		2151	data->laststype = PL_laststype;
		2152	data->laststatval = PL_laststatval;
		2153	data->statcache = PL_statcache;
		2154	}
		2155
		2156	/* now build the result vector out of all the parameters and the data_sv */
		2157	{
		2158	int i;
		2159
		2160	for (i = 0; i < items; ++i)
		2161	av_push (state, SvREFCNT_inc_NN (ST (i)));
		2162	}
		2163
		2164	av_push (state, data_sv);
		2165
		2166	api_ready (aTHX_ coro);
		2167	SvREFCNT_dec (coro);
		2168	SvREFCNT_dec ((AV *)state);
		2169	}
		2170
		2171	static int
		2172	slf_check_aio_req (pTHX_ struct CoroSLF *frame)
		2173	{
		2174	AV state = (AV )frame->data;
		2175
		2176	/* one element that is an RV? repeat! */
		2177	if (AvFILLp (state) == 0 && SvROK (AvARRAY (state)[0]))
		2178	return 1;
		2179
		2180	/* restore status */
		2181	{
		2182	SV *data_sv = av_pop (state);
		2183	struct io_state data = (struct io_state )SvPVX (data_sv);
		2184
		2185	errno = data->errorno;
		2186	PL_laststype = data->laststype;
		2187	PL_laststatval = data->laststatval;
		2188	PL_statcache = data->statcache;
		2189
		2190	SvREFCNT_dec (data_sv);
		2191	}
		2192
		2193	/* push result values */
		2194	{
		2195	dSP;
		2196	int i;
		2197
		2198	EXTEND (SP, AvFILLp (state) + 1);
		2199	for (i = 0; i <= AvFILLp (state); ++i)
		2200	PUSHs (sv_2mortal (SvREFCNT_inc_NN (AvARRAY (state)[i])));
		2201
		2202	PUTBACK;
		2203	}
		2204
		2205	return 0;
		2206	}
		2207
		2208	static void
		2209	slf_init_aio_req (pTHX_ struct CoroSLF frame, CV cv, SV **arg, int items)
		2210	{
		2211	AV state = (AV )sv_2mortal ((SV *)newAV ());
		2212	SV *coro_hv = SvRV (coro_current);
		2213	struct coro *coro = SvSTATE_hv (coro_hv);
		2214
		2215	/* put our coroutine id on the state arg */
		2216	av_push (state, SvREFCNT_inc_NN (coro_hv));
		2217
		2218	/* first see whether we have a non-zero priority and set it as AIO prio */
		2219	if (coro->prio)
		2220	{
		2221	dSP;
		2222
		2223	static SV *prio_cv;
		2224	static SV *prio_sv;
		2225
		2226	if (expect_false (!prio_cv))
		2227	{
		2228	prio_cv = (SV *)get_cv ("IO::AIO::aioreq_pri", 0);
		2229	prio_sv = newSViv (0);
		2230	}
		2231
		2232	PUSHMARK (SP);
		2233	sv_setiv (prio_sv, coro->prio);
		2234	XPUSHs (prio_sv);
		2235
		2236	PUTBACK;
		2237	call_sv (prio_cv, G_VOID \| G_DISCARD);
		2238	}
		2239
		2240	/* now call the original request */
		2241	{
		2242	dSP;
		2243	CV req = (CV )CORO_MAGIC_NN ((SV *)cv, CORO_MAGIC_type_aio)->mg_obj;
		2244	int i;
		2245
		2246	PUSHMARK (SP);
		2247
		2248	/* first push all args to the stack */
		2249	EXTEND (SP, items + 1);
		2250
		2251	for (i = 0; i < items; ++i)
		2252	PUSHs (arg [i]);
		2253
		2254	/* now push the callback closure */
		2255	PUSHs (sv_2mortal (gensub (aTHX_ coro_aio_callback, (void )SvREFCNT_inc_NN ((SV )state))));
		2256
		2257	/* now call the AIO function - we assume our request is uncancelable */
		2258	PUTBACK;
		2259	call_sv ((SV *)req, G_VOID \| G_DISCARD);
		2260	}
		2261
		2262	/* now that the requets is going, we loop toll we have a result */
		2263	frame->data = (void *)state;
		2264	frame->prepare = prepare_schedule;
		2265	frame->check = slf_check_aio_req;
		2266	}
		2267
		2268	static void
		2269	coro_aio_req_xs (pTHX_ CV *cv)
		2270	{
		2271	dXSARGS;
		2272
		2273	CORO_EXECUTE_SLF_XS (slf_init_aio_req);
		2274
		2275	XSRETURN_EMPTY;
2117	}	2276	}
2118		2277
2119	/*****************************************************************************/	2278	/*****************************************************************************/
2120		2279
2121	MODULE = Coro::State PACKAGE = Coro::State PREFIX = api_	2280	MODULE = Coro::State PACKAGE = Coro::State PREFIX = api_
…		…
2355		2514
2356	void	2515	void
2357	throw (Coro::State self, SV *throw = &PL_sv_undef)	2516	throw (Coro::State self, SV *throw = &PL_sv_undef)
2358	PROTOTYPE: $;$	2517	PROTOTYPE: $;$
2359	CODE:	2518	CODE:
		2519	{
		2520	struct coro *current = SvSTATE_current;
		2521	SV **throwp = self == current ? &coro_throw : &self->throw;
2360	SvREFCNT_dec (self->throw);	2522	SvREFCNT_dec (*throwp);
2361	self->throw = SvOK (throw) ? newSVsv (throw) : 0;	2523	*throwp = SvOK (throw) ? newSVsv (throw) : 0;
		2524	}
2362		2525
2363	void	2526	void
2364	api_trace (SV *coro, int flags = CC_TRACE \| CC_TRACE_SUB)	2527	api_trace (SV *coro, int flags = CC_TRACE \| CC_TRACE_SUB)
2365	PROTOTYPE: $;$	2528	PROTOTYPE: $;$
2366	C_ARGS: aTHX_ coro, flags	2529	C_ARGS: aTHX_ coro, flags
…		…
2415	SV dst = ix ? (SV )&self->slot->defav : (SV **)&self->slot->defsv;	2578	SV dst = ix ? (SV )&self->slot->defav : (SV **)&self->slot->defsv;
2416		2579
2417	SV tmp = src; src = dst; *dst = tmp;	2580	SV tmp = src; src = dst; *dst = tmp;
2418	}	2581	}
2419		2582
		2583
2420	MODULE = Coro::State PACKAGE = Coro	2584	MODULE = Coro::State PACKAGE = Coro
2421		2585
2422	BOOT:	2586	BOOT:
2423	{	2587	{
2424	int i;	2588	int i;
…		…
2568		2732
2569	void	2733	void
2570	_pool_2 (SV *cb)	2734	_pool_2 (SV *cb)
2571	CODE:	2735	CODE:
2572	{	2736	{
2573	struct coro *coro = SvSTATE_current;	2737	HV hv = (HV )SvRV (coro_current);
		2738	struct coro coro = SvSTATE_hv ((SV )hv);
2574		2739
2575	sv_setsv (cb, &PL_sv_undef);	2740	sv_setsv (cb, &PL_sv_undef);
2576		2741
2577	SvREFCNT_dec ((SV )PL_defoutgv); PL_defoutgv = (GV )coro->saved_deffh;	2742	SvREFCNT_dec ((SV )PL_defoutgv); PL_defoutgv = (GV )coro->saved_deffh;
2578	coro->saved_deffh = 0;	2743	coro->saved_deffh = 0;
…		…
2585	SvREFCNT_dec (old);	2750	SvREFCNT_dec (old);
2586	croak ("\3async_pool terminate\2\n");	2751	croak ("\3async_pool terminate\2\n");
2587	}	2752	}
2588		2753
2589	av_clear (GvAV (PL_defgv));	2754	av_clear (GvAV (PL_defgv));
2590	hv_store ((HV *)SvRV (coro_current), "desc", sizeof ("desc") - 1,	2755	hv_store (hv, "desc", sizeof ("desc") - 1,
2591	newSVpvn ("[async_pool idle]", sizeof ("[async_pool idle]") - 1), 0);	2756	newSVpvn ("[async_pool idle]", sizeof ("[async_pool idle]") - 1), 0);
2592		2757
2593	coro->prio = 0;	2758	coro->prio = 0;
2594		2759
2595	if (coro->cctx && (coro->cctx->flags & CC_TRACE))	2760	if (coro->cctx && (coro->cctx->flags & CC_TRACE))
…		…
2597		2762
2598	av_push (av_async_pool, newSVsv (coro_current));	2763	av_push (av_async_pool, newSVsv (coro_current));
2599	}	2764	}
2600		2765
2601		2766
2602	MODULE = Coro::State PACKAGE = Coro::AIO
2603
2604	void
2605	_get_state (SV *self)
2606	PROTOTYPE: $
2607	PPCODE:
2608	{
2609	AV *defav = GvAV (PL_defgv);
2610	AV *av = newAV ();
2611	int i;
2612	SV *data_sv = newSV (sizeof (struct io_state));
2613	struct io_state data = (struct io_state )SvPVX (data_sv);
2614	SvCUR_set (data_sv, sizeof (struct io_state));
2615	SvPOK_only (data_sv);
2616
2617	data->errorno = errno;
2618	data->laststype = PL_laststype;
2619	data->laststatval = PL_laststatval;
2620	data->statcache = PL_statcache;
2621
2622	av_extend (av, AvFILLp (defav) + 1 + 1);
2623
2624	for (i = 0; i <= AvFILLp (defav); ++i)
2625	av_push (av, SvREFCNT_inc_NN (AvARRAY (defav)[i]));
2626
2627	av_push (av, data_sv);
2628
2629	XPUSHs (sv_2mortal (newRV_noinc ((SV *)av)));
2630
2631	api_ready (aTHX_ self);
2632	}
2633
2634	void
2635	_set_state (SV *state)
2636	PROTOTYPE: $
2637	PPCODE:
2638	{
2639	AV av = (AV )SvRV (state);
2640	struct io_state data = (struct io_state )SvPVX (AvARRAY (av)[AvFILLp (av)]);
2641	int i;
2642
2643	errno = data->errorno;
2644	PL_laststype = data->laststype;
2645	PL_laststatval = data->laststatval;
2646	PL_statcache = data->statcache;
2647
2648	EXTEND (SP, AvFILLp (av));
2649	for (i = 0; i < AvFILLp (av); ++i)
2650	PUSHs (sv_2mortal (SvREFCNT_inc_NN (AvARRAY (av)[i])));
2651	}
2652
2653
2654	MODULE = Coro::State PACKAGE = Coro::AnyEvent
2655
2656	BOOT:
2657	sv_activity = coro_get_sv (aTHX_ "Coro::AnyEvent::ACTIVITY", TRUE);
2658
2659	void
2660	_schedule (...)
2661	CODE:
2662	{
2663	static int incede;
2664
2665	api_cede_notself (aTHX);
2666
2667	++incede;
2668	while (coro_nready >= incede && api_cede (aTHX))
2669	;
2670
2671	sv_setsv (sv_activity, &PL_sv_undef);
2672	if (coro_nready >= incede)
2673	{
2674	PUSHMARK (SP);
2675	PUTBACK;
2676	call_pv ("Coro::AnyEvent::_activity", G_DISCARD \| G_EVAL);
2677	SPAGAIN;
2678	}
2679
2680	--incede;
2681	}
2682
2683
2684	MODULE = Coro::State PACKAGE = PerlIO::cede	2767	MODULE = Coro::State PACKAGE = PerlIO::cede
2685		2768
2686	BOOT:	2769	BOOT:
2687	PerlIO_define_layer (aTHX_ &PerlIO_cede);	2770	PerlIO_define_layer (aTHX_ &PerlIO_cede);
		2771
2688		2772
2689	MODULE = Coro::State PACKAGE = Coro::Semaphore	2773	MODULE = Coro::State PACKAGE = Coro::Semaphore
2690		2774
2691	SV *	2775	SV *
2692	new (SV klass, SV count_ = 0)	2776	new (SV klass, SV count_ = 0)
2693	CODE:	2777	CODE:
2694	{	2778	{
2695	/* a semaphore contains a counter IV in $sem->[0] and any waiters after that */	2779	/* a semaphore contains a counter IV in $sem->[0] and any waiters after that */
2696	AV *av = newAV ();	2780	AV *av = newAV ();
		2781	SV **ary;
		2782
		2783	/* unfortunately, building manually saves memory */
		2784	Newx (ary, 2, SV *);
		2785	AvALLOC (av) = ary;
		2786	AvARRAY (av) = ary;
		2787	AvMAX (av) = 1;
		2788	AvFILLp (av) = 0;
2697	av_push (av, newSViv (count_ && SvOK (count_) ? SvIV (count_) : 1));	2789	ary [0] = newSViv (count_ && SvOK (count_) ? SvIV (count_) : 1);
		2790
2698	RETVAL = sv_bless (newRV_noinc ((SV *)av), GvSTASH (CvGV (cv)));	2791	RETVAL = sv_bless (newRV_noinc ((SV *)av), GvSTASH (CvGV (cv)));
2699	}	2792	}
2700	OUTPUT:	2793	OUTPUT:
2701	RETVAL	2794	RETVAL
2702		2795
…		…
2752	for (i = 1; i <= AvFILLp (av); ++i)	2845	for (i = 1; i <= AvFILLp (av); ++i)
2753	PUSHs (newSVsv (AvARRAY (av)[i]));	2846	PUSHs (newSVsv (AvARRAY (av)[i]));
2754	}	2847	}
2755	}	2848	}
2756		2849
		2850
		2851	MODULE = Coro::State PACKAGE = Coro::AnyEvent
		2852
		2853	BOOT:
		2854	sv_activity = coro_get_sv (aTHX_ "Coro::AnyEvent::ACTIVITY", TRUE);
		2855
		2856	void
		2857	_schedule (...)
		2858	CODE:
		2859	{
		2860	static int incede;
		2861
		2862	api_cede_notself (aTHX);
		2863
		2864	++incede;
		2865	while (coro_nready >= incede && api_cede (aTHX))
		2866	;
		2867
		2868	sv_setsv (sv_activity, &PL_sv_undef);
		2869	if (coro_nready >= incede)
		2870	{
		2871	PUSHMARK (SP);
		2872	PUTBACK;
		2873	call_pv ("Coro::AnyEvent::_activity", G_DISCARD \| G_EVAL);
		2874	SPAGAIN;
		2875	}
		2876
		2877	--incede;
		2878	}
		2879
		2880
		2881	MODULE = Coro::State PACKAGE = Coro::AIO
		2882
		2883	void
		2884	_register (char target, char proto, SV *req)
		2885	CODE:
		2886	{
		2887	HV *st;
		2888	GV *gvp;
		2889	CV *req_cv = sv_2cv (req, &st, &gvp, 0);
		2890	/* newXSproto doesn't return the CV on 5.8 */
		2891	CV *slf_cv = newXS (target, coro_aio_req_xs, __FILE__);
		2892	sv_setpv ((SV *)slf_cv, proto);
		2893	sv_magicext ((SV )slf_cv, (SV )req_cv, CORO_MAGIC_type_aio, 0, 0, 0);
		2894	}
		2895

Diff Legend

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

Comparing Coro/Coro/State.xs (file contents): Revision 1.284 by root, Mon Nov 17 01:05:47 2008 UTC vs. Revision 1.289 by root, Mon Nov 17 08:25:06 2008 UTC

Diff Legend

Comparing Coro/Coro/State.xs (file contents):
Revision 1.284 by root, Mon Nov 17 01:05:47 2008 UTC vs.
Revision 1.289 by root, Mon Nov 17 08:25:06 2008 UTC