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

Comparing Coro/Coro/State.xs (file contents):
Revision 1.286 by root, Mon Nov 17 04:19:49 2008 UTC vs.
Revision 1.292 by root, Tue Nov 18 06:09:35 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? */
		156
		157	/* we hijack an hopefully unused CV flag for our purposes */
		158	#define CVf_SLF 0x4000
		159	static OP *pp_slf (pTHX);
163		160
164	static U32 cctx_gen;	161	static U32 cctx_gen;
165	static size_t cctx_stacksize = CORO_STACKSIZE;	162	static size_t cctx_stacksize = CORO_STACKSIZE;
166	static struct CoroAPI coroapi;	163	static struct CoroAPI coroapi;
167	static AV main_mainstack; / used to differentiate between $main and others */	164	static AV main_mainstack; / used to differentiate between $main and others */
…		…
384	SvREFCNT_dec (av); /* sv_magicext increased the refcount */	381	SvREFCNT_dec (av); /* sv_magicext increased the refcount */
385		382
386	return 0;	383	return 0;
387	}	384	}
388		385
389	#define CORO_MAGIC_type_cv PERL_MAGIC_ext	386	#define CORO_MAGIC_type_cv 26
390	#define CORO_MAGIC_type_state PERL_MAGIC_ext	387	#define CORO_MAGIC_type_state PERL_MAGIC_ext
391		388
392	static MGVTBL coro_cv_vtbl = {	389	static MGVTBL coro_cv_vtbl = {
393	0, 0, 0, 0,	390	0, 0, 0, 0,
394	coro_cv_free	391	coro_cv_free
395	};	392	};
396		393
		394	#define CORO_MAGIC_NN(sv, type) \
		395	(expect_true (SvMAGIC (sv)->mg_type == type) \
		396	? SvMAGIC (sv) \
		397	: mg_find (sv, type))
		398
397	#define CORO_MAGIC(sv, type) \	399	#define CORO_MAGIC(sv, type) \
398	expect_true (SvMAGIC (sv)) \	400	(expect_true (SvMAGIC (sv)) \
399	? expect_true (SvMAGIC (sv)->mg_type == type) \	401	? CORO_MAGIC_NN (sv, type) \
400	? SvMAGIC (sv) \
401	: mg_find (sv, type) \
402	: 0	402	: 0)
403		403
404	#define CORO_MAGIC_cv(cv) CORO_MAGIC (((SV *)(cv)), CORO_MAGIC_type_cv)	404	#define CORO_MAGIC_cv(cv) CORO_MAGIC (((SV *)(cv)), CORO_MAGIC_type_cv)
405	#define CORO_MAGIC_state(sv) CORO_MAGIC (((SV *)(sv)), CORO_MAGIC_type_state)	405	#define CORO_MAGIC_state(sv) CORO_MAGIC_NN (((SV *)(sv)), CORO_MAGIC_type_state)
406		406
407	INLINE struct coro *	407	INLINE struct coro *
408	SvSTATE_ (pTHX_ SV *coro)	408	SvSTATE_ (pTHX_ SV *coro)
409	{	409	{
410	HV *stash;	410	HV *stash;
…		…
428	return (struct coro *)mg->mg_ptr;	428	return (struct coro *)mg->mg_ptr;
429	}	429	}
430		430
431	#define SvSTATE(sv) SvSTATE_ (aTHX_ (sv))	431	#define SvSTATE(sv) SvSTATE_ (aTHX_ (sv))
432		432
433	/* fastert than SvSTATE, but expects a coroutine hv */	433	/* faster than SvSTATE, but expects a coroutine hv */
434	INLINE struct coro *	434	#define SvSTATE_hv(hv) ((struct coro )CORO_MAGIC_NN ((SV )hv, CORO_MAGIC_type_state)->mg_ptr)
435	SvSTATE_hv (SV *sv)
436	{
437	MAGIC *mg = expect_true (SvMAGIC (sv)->mg_type == CORO_MAGIC_type_state)
438	? SvMAGIC (sv)
439	: mg_find (sv, CORO_MAGIC_type_state);
440
441	return (struct coro *)mg->mg_ptr;
442	}
443
444	#define SvSTATE_current SvSTATE_hv (SvRV (coro_current))	435	#define SvSTATE_current SvSTATE_hv (SvRV (coro_current))
445		436
446	/* the next two functions merely cache the padlists */	437	/* the next two functions merely cache the padlists */
447	static void	438	static void
448	get_padlist (pTHX_ CV *cv)	439	get_padlist (pTHX_ CV *cv)
…		…
827	slf_check_nop (pTHX_ struct CoroSLF *frame)	818	slf_check_nop (pTHX_ struct CoroSLF *frame)
828	{	819	{
829	return 0;	820	return 0;
830	}	821	}
831		822
		823	static UNOP coro_setup_op;
		824
832	static void NOINLINE /* noinline to keep it out of the transfer fast path */	825	static void NOINLINE /* noinline to keep it out of the transfer fast path */
833	coro_setup (pTHX_ struct coro *coro)	826	coro_setup (pTHX_ struct coro *coro)
834	{	827	{
835	/*	828	/*
836	* emulate part of the perl startup here.	829	* emulate part of the perl startup here.
…		…
881	* likely was suspended in pp_slf, so we have to emulate entering pp_slf here.	874	* likely was suspended in pp_slf, so we have to emulate entering pp_slf here.
882	*/	875	*/
883	slf_frame.prepare = prepare_nop; /* provide a nop function for an eventual pp_slf */	876	slf_frame.prepare = prepare_nop; /* provide a nop function for an eventual pp_slf */
884	slf_frame.check = slf_check_nop; /* signal pp_slf to not repeat */	877	slf_frame.check = slf_check_nop; /* signal pp_slf to not repeat */
885		878
		879	/* and we have to provide the pp_slf op in any case, so pp_slf can skip it */
		880	coro_setup_op.op_type = OP_CUSTOM;
		881	coro_setup_op.op_ppaddr = pp_slf;
		882	coro_setup_op.op_next = PL_op;
		883
		884	PL_op = (OP *)&coro_setup_op;
		885
		886	/* copy throw, in case it was set before coro_setup */
886	coro_throw = coro->throw;	887	coro_throw = coro->throw;
887	}	888	}
888		889
889	static void	890	static void
890	coro_destruct (pTHX_ struct coro *coro)	891	coro_destruct (pTHX_ struct coro *coro)
…		…
1054		1055
1055	TAINT_NOT;	1056	TAINT_NOT;
1056	return 0;	1057	return 0;
1057	}	1058	}
1058		1059
		1060	static struct coro_cctx *cctx_ssl_cctx;
		1061	static struct CoroSLF cctx_ssl_frame;
		1062
1059	static void	1063	static void
1060	prepare_set_stacklevel (struct coro_transfer_args ta, struct coro_cctx cctx)	1064	slf_prepare_set_stacklevel (pTHX_ struct coro_transfer_args *ta)
1061	{	1065	{
1062	ta->prev = (struct coro *)cctx;	1066	ta->prev = (struct coro *)cctx_ssl_cctx;
1063	ta->next = 0;	1067	ta->next = 0;
1064	}	1068	}
1065		1069
1066	/* inject a fake call to Coro::State::_cctx_init into the execution */	1070	static int
1067	/* _cctx_init should be careful, as it could be called at almost any time */	1071	slf_check_set_stacklevel (pTHX_ struct CoroSLF *frame)
1068	/* during execution of a perl program */	1072	{
1069	/* also initialises PL_top_env */	1073	*frame = cctx_ssl_frame;
		1074
		1075	return 1; /* execute the restored frame - there must be one */
		1076	}
		1077
		1078	/* initialises PL_top_env and injects a pseudo-slf-call to sett he stacklevel */
1070	static void NOINLINE	1079	static void NOINLINE
1071	cctx_prepare (pTHX_ coro_cctx *cctx)	1080	cctx_prepare (pTHX_ coro_cctx *cctx)
1072	{	1081	{
1073	dSP;
1074	UNOP myop;
1075
1076	PL_top_env = &PL_start_env;	1082	PL_top_env = &PL_start_env;
1077		1083
1078	if (cctx->flags & CC_TRACE)	1084	if (cctx->flags & CC_TRACE)
1079	PL_runops = runops_trace;	1085	PL_runops = runops_trace;
1080		1086
1081	Zero (&myop, 1, UNOP);	1087	/* we already must be in an SLF call, there is no other valid way
1082	myop.op_next = PL_op;	1088	* that can lead to creation of a new cctx */
1083	myop.op_flags = OPf_WANT_VOID \| OPf_STACKED;	1089	assert (("FATAL: can't prepare slf-less cctx in Coro module (please report)",
		1090	slf_frame.prepare && PL_op->op_ppaddr == pp_slf));
1084		1091
1085	PUSHMARK (SP);	1092	cctx_ssl_cctx = cctx;
1086	EXTEND (SP, 2);	1093	cctx_ssl_frame = slf_frame;
1087	PUSHs (sv_2mortal (newSViv ((IV)cctx)));	1094
1088	PUSHs ((SV *)get_cv ("Coro::State::_cctx_init", FALSE));	1095	slf_frame.prepare = slf_prepare_set_stacklevel;
1089	PUTBACK;	1096	slf_frame.check = slf_check_set_stacklevel;
1090	PL_op = (OP *)&myop;
1091	PL_op = PL_ppaddr[OP_ENTERSUB](aTHX);
1092	SPAGAIN;
1093	}	1097	}
1094		1098
1095	/* the tail of transfer: execute stuff we can only do after a transfer */	1099	/* the tail of transfer: execute stuff we can only do after a transfer */
1096	INLINE void	1100	INLINE void
1097	transfer_tail (pTHX)	1101	transfer_tail (pTHX)
…		…
1541	ENTER;	1545	ENTER;
1542	SAVETMPS;	1546	SAVETMPS;
1543		1547
1544	PUSHMARK (SP);	1548	PUSHMARK (SP);
1545	PUTBACK;	1549	PUTBACK;
1546	call_sv (sv_hook, G_DISCARD);	1550	call_sv (sv_hook, G_VOID \| G_DISCARD);
1547	SPAGAIN;
1548		1551
1549	FREETMPS;	1552	FREETMPS;
1550	LEAVE;	1553	LEAVE;
1551	}	1554	}
1552		1555
…		…
1579	ENTER;	1582	ENTER;
1580	SAVETMPS;	1583	SAVETMPS;
1581		1584
1582	PUSHMARK (SP);	1585	PUSHMARK (SP);
1583	PUTBACK;	1586	PUTBACK;
1584	call_sv (get_sv ("Coro::idle", FALSE), G_DISCARD);	1587	call_sv (get_sv ("Coro::idle", FALSE), G_VOID \| G_DISCARD);
1585	SPAGAIN;
1586		1588
1587	FREETMPS;	1589	FREETMPS;
1588	LEAVE;	1590	LEAVE;
1589	continue;	1591	continue;
1590	}	1592	}
…		…
1697	if (coro->flags & CF_RUNNING)	1699	if (coro->flags & CF_RUNNING)
1698	PL_runops = RUNOPS_DEFAULT;	1700	PL_runops = RUNOPS_DEFAULT;
1699	else	1701	else
1700	coro->slot->runops = RUNOPS_DEFAULT;	1702	coro->slot->runops = RUNOPS_DEFAULT;
1701	}	1703	}
		1704	}
		1705
		1706	/*****************************************************************************/
		1707	/* schedule-like-function opcode (SLF) */
		1708
		1709	static UNOP slf_restore; /* restore stack as entersub did, for first-re-run */
		1710	static const CV *slf_cv;
		1711	static SV **slf_argv;
		1712	static int slf_argc, slf_arga; /* count, allocated */
		1713	static I32 slf_ax; /* top of stack, for restore */
		1714
		1715	/* this restores the stack in the case we patched the entersub, to */
		1716	/* recreate the stack frame as perl will on following calls */
		1717	/* since entersub cleared the stack */
		1718	static OP *
		1719	pp_restore (pTHX)
		1720	{
		1721	int i;
		1722	SV **SP = PL_stack_base + slf_ax;
		1723
		1724	PUSHMARK (SP);
		1725
		1726	EXTEND (SP, slf_argc + 1);
		1727
		1728	for (i = 0; i < slf_argc; ++i)
		1729	PUSHs (sv_2mortal (slf_argv [i]));
		1730
		1731	PUSHs ((SV *)CvGV (slf_cv));
		1732
		1733	RETURNOP (slf_restore.op_first);
		1734	}
		1735
		1736	static void
		1737	slf_prepare_transfer (pTHX_ struct coro_transfer_args *ta)
		1738	{
		1739	SV arg = (SV )slf_frame.data;
		1740
		1741	prepare_transfer (aTHX_ ta, arg [0], arg [1]);
		1742	}
		1743
		1744	static void
		1745	slf_init_transfer (pTHX_ struct CoroSLF frame, CV cv, SV **arg, int items)
		1746	{
		1747	if (items != 2)
		1748	croak ("Coro::State::transfer (prev, next) expects two arguments, not %d,", items);
		1749
		1750	frame->prepare = slf_prepare_transfer;
		1751	frame->check = slf_check_nop;
		1752	frame->data = (void )arg; / let's hope it will stay valid */
		1753	}
		1754
		1755	static void
		1756	slf_init_schedule (pTHX_ struct CoroSLF frame, CV cv, SV **arg, int items)
		1757	{
		1758	frame->prepare = prepare_schedule;
		1759	frame->check = slf_check_nop;
		1760	}
		1761
		1762	static void
		1763	slf_init_cede (pTHX_ struct CoroSLF frame, CV cv, SV **arg, int items)
		1764	{
		1765	frame->prepare = prepare_cede;
		1766	frame->check = slf_check_nop;
		1767	}
		1768
		1769	static void
		1770	slf_init_cede_notself (pTHX_ struct CoroSLF frame, CV cv, SV **arg, int items)
		1771	{
		1772	frame->prepare = prepare_cede_notself;
		1773	frame->check = slf_check_nop;
		1774	}
		1775
		1776	/*
		1777	* these not obviously related functions are all rolled into one
		1778	* function to increase chances that they all will call transfer with the same
		1779	* stack offset
		1780	* SLF stands for "schedule-like-function".
		1781	*/
		1782	static OP *
		1783	pp_slf (pTHX)
		1784	{
		1785	I32 checkmark; /* mark SP to see how many elements check has pushed */
		1786
		1787	/* set up the slf frame, unless it has already been set-up */
		1788	/* the latter happens when a new coro has been started */
		1789	/* or when a new cctx was attached to an existing coroutine */
		1790	if (expect_true (!slf_frame.prepare))
		1791	{
		1792	/* first iteration */
		1793	dSP;
		1794	SV **arg = PL_stack_base + TOPMARK + 1;
		1795	int items = SP - arg; /* args without function object */
		1796	SV gv = sp;
		1797
		1798	/* do a quick consistency check on the "function" object, and if it isn't */
		1799	/* for us, divert to the real entersub */
		1800	if (SvTYPE (gv) != SVt_PVGV
		1801	\|\| !GvCV (gv)
		1802	\|\| !(CvFLAGS (GvCV (gv)) & CVf_SLF))
		1803	return PL_ppaddr[OP_ENTERSUB](aTHX);
		1804
		1805	if (!(PL_op->op_flags & OPf_STACKED))
		1806	{
		1807	/* ampersand-form of call, use @_ instead of stack */
		1808	AV *av = GvAV (PL_defgv);
		1809	arg = AvARRAY (av);
		1810	items = AvFILLp (av) + 1;
		1811	}
		1812
		1813	/* now call the init function, which needs to set up slf_frame */
		1814	((coro_slf_cb)CvXSUBANY (GvCV (gv)).any_ptr)
		1815	(aTHX_ &slf_frame, GvCV (gv), arg, items);
		1816
		1817	/* pop args */
		1818	SP = PL_stack_base + POPMARK;
		1819
		1820	PUTBACK;
		1821	}
		1822
		1823	/* now that we have a slf_frame, interpret it! */
		1824	/* we use a callback system not to make the code needlessly */
		1825	/* complicated, but so we can run multiple perl coros from one cctx */
		1826
		1827	do
		1828	{
		1829	struct coro_transfer_args ta;
		1830
		1831	slf_frame.prepare (aTHX_ &ta);
		1832	TRANSFER (ta, 0);
		1833
		1834	checkmark = PL_stack_sp - PL_stack_base;
		1835	}
		1836	while (slf_frame.check (aTHX_ &slf_frame));
		1837
		1838	slf_frame.prepare = 0; /* invalidate the frame, we are done processing it */
		1839
		1840	/* return value handling - mostly like entersub */
		1841	/* make sure we put something on the stack in scalar context */
		1842	if (GIMME_V == G_SCALAR)
		1843	{
		1844	dSP;
		1845	SV **bot = PL_stack_base + checkmark;
		1846
		1847	if (sp == bot) /* too few, push undef */
		1848	bot [1] = &PL_sv_undef;
		1849	else if (sp != bot + 1) /* too many, take last one */
		1850	bot [1] = *sp;
		1851
		1852	SP = bot + 1;
		1853
		1854	PUTBACK;
		1855	}
		1856
		1857	/* exception handling */
		1858	if (expect_false (coro_throw))
		1859	{
		1860	SV *exception = sv_2mortal (coro_throw);
		1861
		1862	coro_throw = 0;
		1863	sv_setsv (ERRSV, exception);
		1864	croak (0);
		1865	}
		1866
		1867	return NORMAL;
		1868	}
		1869
		1870	static void
		1871	api_execute_slf (pTHX_ CV *cv, coro_slf_cb init_cb, I32 ax)
		1872	{
		1873	int i;
		1874	SV **arg = PL_stack_base + ax;
		1875	int items = PL_stack_sp - arg + 1;
		1876
		1877	assert (("FATAL: SLF call with illegal CV value", !CvANON (cv)));
		1878
		1879	if (PL_op->op_ppaddr != PL_ppaddr [OP_ENTERSUB]
		1880	&& PL_op->op_ppaddr != pp_slf)
		1881	croak ("FATAL: Coro SLF calls can only be made normally, not via goto or any other means, caught");
		1882
		1883	CvFLAGS (cv) \|= CVf_SLF;
		1884	CvXSUBANY (cv).any_ptr = (void *)init_cb;
		1885	slf_cv = cv;
		1886
		1887	/* we patch the op, and then re-run the whole call */
		1888	/* we have to put the same argument on the stack for this to work */
		1889	/* and this will be done by pp_restore */
		1890	slf_restore.op_next = (OP *)&slf_restore;
		1891	slf_restore.op_type = OP_CUSTOM;
		1892	slf_restore.op_ppaddr = pp_restore;
		1893	slf_restore.op_first = PL_op;
		1894
		1895	slf_ax = ax - 1; /* undo the ax++ inside dAXMARK */
		1896
		1897	if (PL_op->op_flags & OPf_STACKED)
		1898	{
		1899	if (items > slf_arga)
		1900	{
		1901	slf_arga = items;
		1902	free (slf_argv);
		1903	slf_argv = malloc (slf_arga * sizeof (SV *));
		1904	}
		1905
		1906	slf_argc = items;
		1907
		1908	for (i = 0; i < items; ++i)
		1909	slf_argv [i] = SvREFCNT_inc (arg [i]);
		1910	}
		1911	else
		1912	slf_argc = 0;
		1913
		1914	PL_op->op_ppaddr = pp_slf;
		1915	PL_op->op_type = OP_CUSTOM; /* maybe we should leave it at entersub? */
		1916
		1917	PL_op = (OP *)&slf_restore;
1702	}	1918	}
1703		1919
1704	/*****************************************************************************/	1920	/*****************************************************************************/
1705	/* PerlIO::cede */	1921	/* PerlIO::cede */
1706		1922
…		…
1775	PerlIOBuf_get_cnt,	1991	PerlIOBuf_get_cnt,
1776	PerlIOBuf_set_ptrcnt,	1992	PerlIOBuf_set_ptrcnt,
1777	};	1993	};
1778		1994
1779	/*****************************************************************************/	1995	/*****************************************************************************/
1780		1996	/* Coro::Semaphore */
1781	static UNOP slf_restore; /* restore stack as entersub did, for first-re-run */
1782	static const CV *slf_cv;
1783	static SV **slf_argv;
1784	static int slf_argc, slf_arga; /* count, allocated */
1785	static I32 slf_ax; /* top of stack, for restore */
1786
1787	/* this restores the stack in the case we patched the entersub, to */
1788	/* recreate the stack frame as perl will on following calls */
1789	/* since entersub cleared the stack */
1790	static OP *
1791	pp_restore (pTHX)
1792	{
1793	int i;
1794	SV **SP = PL_stack_base + slf_ax;
1795
1796	PUSHMARK (SP);
1797
1798	EXTEND (SP, slf_argc + 1);
1799
1800	for (i = 0; i < slf_argc; ++i)
1801	PUSHs (sv_2mortal (slf_argv [i]));
1802
1803	PUSHs ((SV *)CvGV (slf_cv));
1804
1805	RETURNOP (slf_restore.op_first);
1806	}
1807
1808	static void
1809	slf_prepare_set_stacklevel (pTHX_ struct coro_transfer_args *ta)
1810	{
1811	prepare_set_stacklevel (ta, (struct coro_cctx *)slf_frame.data);
1812	}
1813
1814	static void
1815	slf_init_set_stacklevel (pTHX_ struct CoroSLF frame, CV cv, SV **arg, int items)
1816	{
1817	assert (("FATAL: set_stacklevel needs the coro cctx as sole argument", items == 1));
1818
1819	frame->prepare = slf_prepare_set_stacklevel;
1820	frame->check = slf_check_nop;
1821	frame->data = (void *)SvIV (arg [0]);
1822	}
1823
1824	static void
1825	slf_prepare_transfer (pTHX_ struct coro_transfer_args *ta)
1826	{
1827	SV arg = (SV )slf_frame.data;
1828
1829	prepare_transfer (aTHX_ ta, arg [0], arg [1]);
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	slf_frame.prepare = 0; /* invalidate the frame, we are done processing it */
1928
1929	/* return value handling - mostly like entersub */
1930	{
1931	dSP;
1932	SV **bot = PL_stack_base + checkmark;
1933	int gimme = GIMME_V;
1934
1935	/* make sure we put something on the stack in scalar context */
1936	if (gimme == G_SCALAR)
1937	{
1938	if (sp == bot)
1939	XPUSHs (&PL_sv_undef);
1940
1941	SP = bot + 1;
1942	}
1943
1944	PUTBACK;
1945	}
1946
1947	/* exception handling */
1948	if (expect_false (coro_throw))
1949	{
1950	SV *exception = sv_2mortal (coro_throw);
1951
1952	coro_throw = 0;
1953	sv_setsv (ERRSV, exception);
1954	croak (0);
1955	}
1956
1957	return NORMAL;
1958	}
1959
1960	static void
1961	api_execute_slf (pTHX_ CV *cv, coro_slf_cb init_cb, I32 ax)
1962	{
1963	int i;
1964	SV **arg = PL_stack_base + ax;
1965	int items = PL_stack_sp - arg + 1;
1966
1967	assert (("FATAL: SLF call with illegal CV value", !CvANON (cv)));
1968
1969	if (PL_op->op_ppaddr != PL_ppaddr [OP_ENTERSUB]
1970	&& PL_op->op_ppaddr != pp_slf)
1971	croak ("FATAL: Coro SLF calls can only be made normally, not via goto or any other means, caught");
1972
1973	CvFLAGS (cv) \|= CVf_SLF;
1974	CvXSUBANY (cv).any_ptr = (void *)init_cb;
1975	slf_cv = cv;
1976
1977	/* we patch the op, and then re-run the whole call */
1978	/* we have to put the same argument on the stack for this to work */
1979	/* and this will be done by pp_restore */
1980	slf_restore.op_next = (OP *)&slf_restore;
1981	slf_restore.op_type = OP_CUSTOM;
1982	slf_restore.op_ppaddr = pp_restore;
1983	slf_restore.op_first = PL_op;
1984
1985	slf_ax = ax - 1; /* undo the ax++ inside dAXMARK */
1986
1987	if (PL_op->op_flags & OPf_STACKED)
1988	{
1989	if (items > slf_arga)
1990	{
1991	slf_arga = items;
1992	free (slf_argv);
1993	slf_argv = malloc (slf_arga * sizeof (SV *));
1994	}
1995
1996	slf_argc = items;
1997
1998	for (i = 0; i < items; ++i)
1999	slf_argv [i] = SvREFCNT_inc (arg [i]);
2000	}
2001	else
2002	slf_argc = 0;
2003
2004	PL_op->op_ppaddr = pp_slf;
2005	PL_op->op_type = OP_CUSTOM; /* maybe we should leave it at entersub? */
2006
2007	PL_op = (OP *)&slf_restore;
2008	}
2009
2010	/*****************************************************************************/
2011		1997
2012	static void	1998	static void
2013	coro_semaphore_adjust (pTHX_ AV *av, IV adjust)	1999	coro_semaphore_adjust (pTHX_ AV *av, IV adjust)
2014	{	2000	{
2015	SV *count_sv = AvARRAY (av)[0];	2001	SV *count_sv = AvARRAY (av)[0];
…		…
2101	frame->check = slf_check_semaphore_down;	2087	frame->check = slf_check_semaphore_down;
2102		2088
2103	}	2089	}
2104		2090
2105	/*****************************************************************************/	2091	/*****************************************************************************/
		2092	/* gensub: simple closure generation utility */
2106		2093
2107	#define GENSUB_ARG CvXSUBANY (cv).any_ptr	2094	#define GENSUB_ARG CvXSUBANY (cv).any_ptr
2108		2095
2109	/* create a closure from XS, returns a code reference */	2096	/* create a closure from XS, returns a code reference */
2110	/* the arg can be accessed via GENSUB_ARG from the callback */	2097	/* the arg can be accessed via GENSUB_ARG from the callback */
2111	/* the callback must use dXSARGS/XSRETURN */	2098	/* the callback must use dXSARGS/XSRETURN */
2112	static SV *	2099	static SV *
2113	gensub (pTHX_ void (xsub)(pTHX_ CV ), void *arg)	2100	gensub (pTHX_ void (xsub)(pTHX_ CV ), void *arg)
2114	{	2101	{
2115	CV cv = (CV )NEWSV (0, 0);	2102	CV cv = (CV )newSV (0);
2116		2103
2117	sv_upgrade ((SV *)cv, SVt_PVCV);	2104	sv_upgrade ((SV *)cv, SVt_PVCV);
2118		2105
2119	CvANON_on (cv);	2106	CvANON_on (cv);
2120	CvISXSUB_on (cv);	2107	CvISXSUB_on (cv);
2121	CvXSUB (cv) = xsub;	2108	CvXSUB (cv) = xsub;
2122	GENSUB_ARG = arg;	2109	GENSUB_ARG = arg;
2123		2110
2124	return newRV_noinc ((SV *)cv);	2111	return newRV_noinc ((SV *)cv);
		2112	}
		2113
		2114	/*****************************************************************************/
		2115	/* Coro::AIO */
		2116
		2117	#define CORO_MAGIC_type_aio PERL_MAGIC_ext
		2118
		2119	/* helper storage struct */
		2120	struct io_state
		2121	{
		2122	int errorno;
		2123	I32 laststype; /* U16 in 5.10.0 */
		2124	int laststatval;
		2125	Stat_t statcache;
		2126	};
		2127
		2128	static void
		2129	coro_aio_callback (pTHX_ CV *cv)
		2130	{
		2131	dXSARGS;
		2132	AV state = (AV )GENSUB_ARG;
		2133	SV *coro = av_pop (state);
		2134	SV *data_sv = newSV (sizeof (struct io_state));
		2135
		2136	av_extend (state, items);
		2137
		2138	sv_upgrade (data_sv, SVt_PV);
		2139	SvCUR_set (data_sv, sizeof (struct io_state));
		2140	SvPOK_only (data_sv);
		2141
		2142	{
		2143	struct io_state data = (struct io_state )SvPVX (data_sv);
		2144
		2145	data->errorno = errno;
		2146	data->laststype = PL_laststype;
		2147	data->laststatval = PL_laststatval;
		2148	data->statcache = PL_statcache;
		2149	}
		2150
		2151	/* now build the result vector out of all the parameters and the data_sv */
		2152	{
		2153	int i;
		2154
		2155	for (i = 0; i < items; ++i)
		2156	av_push (state, SvREFCNT_inc_NN (ST (i)));
		2157	}
		2158
		2159	av_push (state, data_sv);
		2160
		2161	api_ready (aTHX_ coro);
		2162	SvREFCNT_dec (coro);
		2163	SvREFCNT_dec ((AV *)state);
		2164	}
		2165
		2166	static int
		2167	slf_check_aio_req (pTHX_ struct CoroSLF *frame)
		2168	{
		2169	AV state = (AV )frame->data;
		2170
		2171	/* one element that is an RV? repeat! */
		2172	if (AvFILLp (state) == 0 && SvROK (AvARRAY (state)[0]))
		2173	return 1;
		2174
		2175	/* restore status */
		2176	{
		2177	SV *data_sv = av_pop (state);
		2178	struct io_state data = (struct io_state )SvPVX (data_sv);
		2179
		2180	errno = data->errorno;
		2181	PL_laststype = data->laststype;
		2182	PL_laststatval = data->laststatval;
		2183	PL_statcache = data->statcache;
		2184
		2185	SvREFCNT_dec (data_sv);
		2186	}
		2187
		2188	/* push result values */
		2189	{
		2190	dSP;
		2191	int i;
		2192
		2193	EXTEND (SP, AvFILLp (state) + 1);
		2194	for (i = 0; i <= AvFILLp (state); ++i)
		2195	PUSHs (sv_2mortal (SvREFCNT_inc_NN (AvARRAY (state)[i])));
		2196
		2197	PUTBACK;
		2198	}
		2199
		2200	return 0;
		2201	}
		2202
		2203	static void
		2204	slf_init_aio_req (pTHX_ struct CoroSLF frame, CV cv, SV **arg, int items)
		2205	{
		2206	AV state = (AV )sv_2mortal ((SV *)newAV ());
		2207	SV *coro_hv = SvRV (coro_current);
		2208	struct coro *coro = SvSTATE_hv (coro_hv);
		2209
		2210	/* put our coroutine id on the state arg */
		2211	av_push (state, SvREFCNT_inc_NN (coro_hv));
		2212
		2213	/* first see whether we have a non-zero priority and set it as AIO prio */
		2214	if (coro->prio)
		2215	{
		2216	dSP;
		2217
		2218	static SV *prio_cv;
		2219	static SV *prio_sv;
		2220
		2221	if (expect_false (!prio_cv))
		2222	{
		2223	prio_cv = (SV *)get_cv ("IO::AIO::aioreq_pri", 0);
		2224	prio_sv = newSViv (0);
		2225	}
		2226
		2227	PUSHMARK (SP);
		2228	sv_setiv (prio_sv, coro->prio);
		2229	XPUSHs (prio_sv);
		2230
		2231	PUTBACK;
		2232	call_sv (prio_cv, G_VOID \| G_DISCARD);
		2233	}
		2234
		2235	/* now call the original request */
		2236	{
		2237	dSP;
		2238	CV req = (CV )CORO_MAGIC_NN ((SV *)cv, CORO_MAGIC_type_aio)->mg_obj;
		2239	int i;
		2240
		2241	PUSHMARK (SP);
		2242
		2243	/* first push all args to the stack */
		2244	EXTEND (SP, items + 1);
		2245
		2246	for (i = 0; i < items; ++i)
		2247	PUSHs (arg [i]);
		2248
		2249	/* now push the callback closure */
		2250	PUSHs (sv_2mortal (gensub (aTHX_ coro_aio_callback, (void )SvREFCNT_inc_NN ((SV )state))));
		2251
		2252	/* now call the AIO function - we assume our request is uncancelable */
		2253	PUTBACK;
		2254	call_sv ((SV *)req, G_VOID \| G_DISCARD);
		2255	}
		2256
		2257	/* now that the requets is going, we loop toll we have a result */
		2258	frame->data = (void *)state;
		2259	frame->prepare = prepare_schedule;
		2260	frame->check = slf_check_aio_req;
		2261	}
		2262
		2263	static void
		2264	coro_aio_req_xs (pTHX_ CV *cv)
		2265	{
		2266	dXSARGS;
		2267
		2268	CORO_EXECUTE_SLF_XS (slf_init_aio_req);
		2269
		2270	XSRETURN_EMPTY;
2125	}	2271	}
2126		2272
2127	/*****************************************************************************/	2273	/*****************************************************************************/
2128		2274
2129	MODULE = Coro::State PACKAGE = Coro::State PREFIX = api_	2275	MODULE = Coro::State PACKAGE = Coro::State PREFIX = api_
…		…
2225	for (i = 1; i < items; i++)	2371	for (i = 1; i < items; i++)
2226	av_push (coro->args, newSVsv (ST (i)));	2372	av_push (coro->args, newSVsv (ST (i)));
2227	}	2373	}
2228	OUTPUT:	2374	OUTPUT:
2229	RETVAL	2375	RETVAL
2230
2231	void
2232	_set_stacklevel (...)
2233	CODE:
2234	CORO_EXECUTE_SLF_XS (slf_init_set_stacklevel);
2235		2376
2236	void	2377	void
2237	transfer (...)	2378	transfer (...)
2238	PROTOTYPE: $$	2379	PROTOTYPE: $$
2239	CODE:	2380	CODE:
…		…
2427	SV dst = ix ? (SV )&self->slot->defav : (SV **)&self->slot->defsv;	2568	SV dst = ix ? (SV )&self->slot->defav : (SV **)&self->slot->defsv;
2428		2569
2429	SV tmp = src; src = dst; *dst = tmp;	2570	SV tmp = src; src = dst; *dst = tmp;
2430	}	2571	}
2431		2572
		2573
2432	MODULE = Coro::State PACKAGE = Coro	2574	MODULE = Coro::State PACKAGE = Coro
2433		2575
2434	BOOT:	2576	BOOT:
2435	{	2577	{
2436	int i;	2578	int i;
…		…
2610		2752
2611	av_push (av_async_pool, newSVsv (coro_current));	2753	av_push (av_async_pool, newSVsv (coro_current));
2612	}	2754	}
2613		2755
2614		2756
2615	MODULE = Coro::State PACKAGE = Coro::AIO
2616
2617	void
2618	_get_state (SV *self)
2619	PROTOTYPE: $
2620	PPCODE:
2621	{
2622	AV *defav = GvAV (PL_defgv);
2623	AV *av = newAV ();
2624	int i;
2625	SV *data_sv = newSV (sizeof (struct io_state));
2626	struct io_state data = (struct io_state )SvPVX (data_sv);
2627	SvCUR_set (data_sv, sizeof (struct io_state));
2628	SvPOK_only (data_sv);
2629
2630	data->errorno = errno;
2631	data->laststype = PL_laststype;
2632	data->laststatval = PL_laststatval;
2633	data->statcache = PL_statcache;
2634
2635	av_extend (av, AvFILLp (defav) + 1 + 1);
2636
2637	for (i = 0; i <= AvFILLp (defav); ++i)
2638	av_push (av, SvREFCNT_inc_NN (AvARRAY (defav)[i]));
2639
2640	av_push (av, data_sv);
2641
2642	XPUSHs (sv_2mortal (newRV_noinc ((SV *)av)));
2643
2644	api_ready (aTHX_ self);
2645	}
2646
2647	void
2648	_set_state (SV *state)
2649	PROTOTYPE: $
2650	PPCODE:
2651	{
2652	AV av = (AV )SvRV (state);
2653	struct io_state data = (struct io_state )SvPVX (AvARRAY (av)[AvFILLp (av)]);
2654	int i;
2655
2656	errno = data->errorno;
2657	PL_laststype = data->laststype;
2658	PL_laststatval = data->laststatval;
2659	PL_statcache = data->statcache;
2660
2661	EXTEND (SP, AvFILLp (av));
2662	for (i = 0; i < AvFILLp (av); ++i)
2663	PUSHs (sv_2mortal (SvREFCNT_inc_NN (AvARRAY (av)[i])));
2664	}
2665
2666
2667	MODULE = Coro::State PACKAGE = Coro::AnyEvent
2668
2669	BOOT:
2670	sv_activity = coro_get_sv (aTHX_ "Coro::AnyEvent::ACTIVITY", TRUE);
2671
2672	void
2673	_schedule (...)
2674	CODE:
2675	{
2676	static int incede;
2677
2678	api_cede_notself (aTHX);
2679
2680	++incede;
2681	while (coro_nready >= incede && api_cede (aTHX))
2682	;
2683
2684	sv_setsv (sv_activity, &PL_sv_undef);
2685	if (coro_nready >= incede)
2686	{
2687	PUSHMARK (SP);
2688	PUTBACK;
2689	call_pv ("Coro::AnyEvent::_activity", G_DISCARD \| G_EVAL);
2690	SPAGAIN;
2691	}
2692
2693	--incede;
2694	}
2695
2696
2697	MODULE = Coro::State PACKAGE = PerlIO::cede	2757	MODULE = Coro::State PACKAGE = PerlIO::cede
2698		2758
2699	BOOT:	2759	BOOT:
2700	PerlIO_define_layer (aTHX_ &PerlIO_cede);	2760	PerlIO_define_layer (aTHX_ &PerlIO_cede);
		2761
2701		2762
2702	MODULE = Coro::State PACKAGE = Coro::Semaphore	2763	MODULE = Coro::State PACKAGE = Coro::Semaphore
2703		2764
2704	SV *	2765	SV *
2705	new (SV klass, SV count_ = 0)	2766	new (SV klass, SV count_ = 0)
2706	CODE:	2767	CODE:
2707	{	2768	{
2708	/* a semaphore contains a counter IV in $sem->[0] and any waiters after that */	2769	/* a semaphore contains a counter IV in $sem->[0] and any waiters after that */
2709	AV *av = newAV ();	2770	AV *av = newAV ();
		2771	SV **ary;
		2772
		2773	/* unfortunately, building manually saves memory */
		2774	Newx (ary, 2, SV *);
		2775	AvALLOC (av) = ary;
		2776	AvARRAY (av) = ary;
		2777	AvMAX (av) = 1;
		2778	AvFILLp (av) = 0;
2710	av_push (av, newSViv (count_ && SvOK (count_) ? SvIV (count_) : 1));	2779	ary [0] = newSViv (count_ && SvOK (count_) ? SvIV (count_) : 1);
		2780
2711	RETVAL = sv_bless (newRV_noinc ((SV *)av), GvSTASH (CvGV (cv)));	2781	RETVAL = sv_bless (newRV_noinc ((SV *)av), GvSTASH (CvGV (cv)));
2712	}	2782	}
2713	OUTPUT:	2783	OUTPUT:
2714	RETVAL	2784	RETVAL
2715		2785
…		…
2761	else	2831	else
2762	{	2832	{
2763	int i;	2833	int i;
2764	EXTEND (SP, AvFILLp (av) + 1 - 1);	2834	EXTEND (SP, AvFILLp (av) + 1 - 1);
2765	for (i = 1; i <= AvFILLp (av); ++i)	2835	for (i = 1; i <= AvFILLp (av); ++i)
2766	PUSHs (newSVsv (AvARRAY (av)[i]));	2836	PUSHs (sv_2mortal (newRV_inc (AvARRAY (av)[i])));
2767	}	2837	}
2768	}	2838	}
2769		2839
		2840
		2841	MODULE = Coro::State PACKAGE = Coro::AnyEvent
		2842
		2843	BOOT:
		2844	sv_activity = coro_get_sv (aTHX_ "Coro::AnyEvent::ACTIVITY", TRUE);
		2845
		2846	void
		2847	_schedule (...)
		2848	CODE:
		2849	{
		2850	static int incede;
		2851
		2852	api_cede_notself (aTHX);
		2853
		2854	++incede;
		2855	while (coro_nready >= incede && api_cede (aTHX))
		2856	;
		2857
		2858	sv_setsv (sv_activity, &PL_sv_undef);
		2859	if (coro_nready >= incede)
		2860	{
		2861	PUSHMARK (SP);
		2862	PUTBACK;
		2863	call_pv ("Coro::AnyEvent::_activity", G_KEEPERR \| G_EVAL \| G_VOID \| G_DISCARD);
		2864	}
		2865
		2866	--incede;
		2867	}
		2868
		2869
		2870	MODULE = Coro::State PACKAGE = Coro::AIO
		2871
		2872	void
		2873	_register (char target, char proto, SV *req)
		2874	CODE:
		2875	{
		2876	HV *st;
		2877	GV *gvp;
		2878	CV *req_cv = sv_2cv (req, &st, &gvp, 0);
		2879	/* newXSproto doesn't return the CV on 5.8 */
		2880	CV *slf_cv = newXS (target, coro_aio_req_xs, __FILE__);
		2881	sv_setpv ((SV *)slf_cv, proto);
		2882	sv_magicext ((SV )slf_cv, (SV )req_cv, CORO_MAGIC_type_aio, 0, 0, 0);
		2883	}
		2884

Diff Legend

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

Comparing Coro/Coro/State.xs (file contents): Revision 1.286 by root, Mon Nov 17 04:19:49 2008 UTC vs. Revision 1.292 by root, Tue Nov 18 06:09:35 2008 UTC

Diff Legend

Comparing Coro/Coro/State.xs (file contents):
Revision 1.286 by root, Mon Nov 17 04:19:49 2008 UTC vs.
Revision 1.292 by root, Tue Nov 18 06:09:35 2008 UTC