ViewVC Help

View File | Revision Log | Show Annotations | Download File

/cvs/Coro/Coro/State.xs

Comparing Coro/Coro/State.xs (file contents):
Revision 1.280 by root, Sun Nov 16 09:43:18 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 */
168	static JMPENV *main_top_env;	165	static JMPENV *main_top_env;
169	static HV *coro_state_stash, *coro_stash;	166	static HV *coro_state_stash, *coro_stash;
170	static volatile SV *coro_mortal; /* will be freed/thrown after next transfer */	167	static volatile SV *coro_mortal; /* will be freed/thrown after next transfer */
171	static volatile struct coro *transfer_next;
172		168
173	static GV *irsgv; /* $/ */	169	static GV *irsgv; /* $/ */
174	static GV *stdoutgv; /* *STDOUT */	170	static GV *stdoutgv; /* *STDOUT */
175	static SV *rv_diehook;	171	static SV *rv_diehook;
176	static SV *rv_warnhook;	172	static SV *rv_warnhook;
…		…
242	/* this is a structure representing a perl-level coroutine */	238	/* this is a structure representing a perl-level coroutine */
243	struct coro {	239	struct coro {
244	/* the C coroutine allocated to this perl coroutine, if any */	240	/* the C coroutine allocated to this perl coroutine, if any */
245	coro_cctx *cctx;	241	coro_cctx *cctx;
246		242
247	/* process data */	243	/* state data */
248	struct CoroSLF slf_frame; /* saved slf frame */	244	struct CoroSLF slf_frame; /* saved slf frame */
249	AV *mainstack;	245	AV *mainstack;
250	perl_slots *slot; /* basically the saved sp */	246	perl_slots *slot; /* basically the saved sp */
251		247
252	AV *args; /* data associated with this coroutine (initial args) */	248	AV *args; /* data associated with this coroutine (initial args) */
…		…
270	};	266	};
271		267
272	typedef struct coro *Coro__State;	268	typedef struct coro *Coro__State;
273	typedef struct coro *Coro__State_or_hashref;	269	typedef struct coro *Coro__State_or_hashref;
274		270
		271	/* the following variables are effectively part of the perl context */
		272	/* and get copied between struct coro and these variables */
		273	/* the mainr easonw e don't support windows process emulation */
275	static struct CoroSLF slf_frame; /* the current slf frame */	274	static struct CoroSLF slf_frame; /* the current slf frame */
		275	static SV *coro_throw;
276		276
277	/** Coro ********************************************************************/	277	/** Coro ********************************************************************/
278		278
279	#define PRIO_MAX 3	279	#define PRIO_MAX 3
280	#define PRIO_HIGH 1	280	#define PRIO_HIGH 1
…		…
381	SvREFCNT_dec (av); /* sv_magicext increased the refcount */	381	SvREFCNT_dec (av); /* sv_magicext increased the refcount */
382		382
383	return 0;	383	return 0;
384	}	384	}
385		385
386	#define CORO_MAGIC_type_cv PERL_MAGIC_ext	386	#define CORO_MAGIC_type_cv 26
387	#define CORO_MAGIC_type_state PERL_MAGIC_ext	387	#define CORO_MAGIC_type_state PERL_MAGIC_ext
388		388
389	static MGVTBL coro_cv_vtbl = {	389	static MGVTBL coro_cv_vtbl = {
390	0, 0, 0, 0,	390	0, 0, 0, 0,
391	coro_cv_free	391	coro_cv_free
392	};	392	};
393		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
394	#define CORO_MAGIC(sv, type) \	399	#define CORO_MAGIC(sv, type) \
395	SvMAGIC (sv) \	400	(expect_true (SvMAGIC (sv)) \
396	? SvMAGIC (sv)->mg_type == type \	401	? CORO_MAGIC_NN (sv, type) \
397	? SvMAGIC (sv) \
398	: mg_find (sv, type) \
399	: 0	402	: 0)
400		403
401	#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)
402	#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)
403		406
404	INLINE struct coro *	407	INLINE struct coro *
405	SvSTATE_ (pTHX_ SV *coro)	408	SvSTATE_ (pTHX_ SV *coro)
406	{	409	{
407	HV *stash;	410	HV *stash;
…		…
424	mg = CORO_MAGIC_state (coro);	427	mg = CORO_MAGIC_state (coro);
425	return (struct coro *)mg->mg_ptr;	428	return (struct coro *)mg->mg_ptr;
426	}	429	}
427		430
428	#define SvSTATE(sv) SvSTATE_ (aTHX_ (sv))	431	#define SvSTATE(sv) SvSTATE_ (aTHX_ (sv))
		432
		433	/* faster than SvSTATE, but expects a coroutine hv */
		434	#define SvSTATE_hv(hv) ((struct coro *)CORO_MAGIC_NN ((SV *)hv, CORO_MAGIC_type_state)->mg_ptr)
		435	#define SvSTATE_current SvSTATE_hv (SvRV (coro_current))
429		436
430	/* the next two functions merely cache the padlists */	437	/* the next two functions merely cache the padlists */
431	static void	438	static void
432	get_padlist (pTHX_ CV *cv)	439	get_padlist (pTHX_ CV *cv)
433	{	440	{
…		…
501	}	508	}
502		509
503	PUTBACK;	510	PUTBACK;
504	}	511	}
505		512
506	slf_frame = c->slf_frame;	513	slf_frame = c->slf_frame;
		514	coro_throw = c->throw;
507	}	515	}
508		516
509	static void	517	static void
510	save_perl (pTHX_ Coro__State c)	518	save_perl (pTHX_ Coro__State c)
511	{	519	{
		520	c->throw = coro_throw;
512	c->slf_frame = slf_frame;	521	c->slf_frame = slf_frame;
513		522
514	{	523	{
515	dSP;	524	dSP;
516	I32 cxix = cxstack_ix;	525	I32 cxix = cxstack_ix;
…		…
809	slf_check_nop (pTHX_ struct CoroSLF *frame)	818	slf_check_nop (pTHX_ struct CoroSLF *frame)
810	{	819	{
811	return 0;	820	return 0;
812	}	821	}
813		822
814	static void	823	static UNOP coro_setup_op;
		824
		825	static void NOINLINE /* noinline to keep it out of the transfer fast path */
815	coro_setup (pTHX_ struct coro *coro)	826	coro_setup (pTHX_ struct coro *coro)
816	{	827	{
817	/*	828	/*
818	* emulate part of the perl startup here.	829	* emulate part of the perl startup here.
819	*/	830	*/
…		…
862	/* this newly created coroutine might be run on an existing cctx which most	873	/* this newly created coroutine might be run on an existing cctx which most
863	* 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.
864	*/	875	*/
865	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 */
866	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 */
		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 */
		887	coro_throw = coro->throw;
867	}	888	}
868		889
869	static void	890	static void
870	coro_destruct (pTHX_ struct coro *coro)	891	coro_destruct (pTHX_ struct coro *coro)
871	{	892	{
…		…
895		916
896	SvREFCNT_dec (PL_diehook);	917	SvREFCNT_dec (PL_diehook);
897	SvREFCNT_dec (PL_warnhook);	918	SvREFCNT_dec (PL_warnhook);
898		919
899	SvREFCNT_dec (coro->saved_deffh);	920	SvREFCNT_dec (coro->saved_deffh);
900	SvREFCNT_dec (coro->throw);	921	SvREFCNT_dec (coro_throw);
901		922
902	coro_destruct_stacks (aTHX);	923	coro_destruct_stacks (aTHX);
903	}	924	}
904		925
905	INLINE void	926	INLINE void
…		…
915	static int	936	static int
916	runops_trace (pTHX)	937	runops_trace (pTHX)
917	{	938	{
918	COP *oldcop = 0;	939	COP *oldcop = 0;
919	int oldcxix = -2;	940	int oldcxix = -2;
920	struct coro *coro = SvSTATE (coro_current); /* trace cctx is tied to specific coro */	941	struct coro *coro = SvSTATE_current; /* trace cctx is tied to specific coro */
921	coro_cctx *cctx = coro->cctx;	942	coro_cctx *cctx = coro->cctx;
922		943
923	while ((PL_op = CALL_FPTR (PL_op->op_ppaddr) (aTHX)))	944	while ((PL_op = CALL_FPTR (PL_op->op_ppaddr) (aTHX)))
924	{	945	{
925	PERL_ASYNC_CHECK ();	946	PERL_ASYNC_CHECK ();
…		…
1034		1055
1035	TAINT_NOT;	1056	TAINT_NOT;
1036	return 0;	1057	return 0;
1037	}	1058	}
1038		1059
		1060	static struct coro_cctx *cctx_ssl_cctx;
		1061	static struct CoroSLF cctx_ssl_frame;
		1062
1039	static void	1063	static void
1040	prepare_set_stacklevel (struct coro_transfer_args *ta, struct coro_cctx *cctx)	1064	slf_prepare_set_stacklevel (pTHX_ struct coro_transfer_args *ta)
1041	{	1065	{
1042	ta->prev = (struct coro *)cctx;	1066	ta->prev = (struct coro *)cctx_ssl_cctx;
1043	ta->next = 0;	1067	ta->next = 0;
1044	}	1068	}
1045		1069
1046	/* inject a fake call to Coro::State::_cctx_init into the execution */	1070	static int
1047	/* _cctx_init should be careful, as it could be called at almost any time */	1071	slf_check_set_stacklevel (pTHX_ struct CoroSLF *frame)
1048	/* during execution of a perl program */	1072	{
1049	/* 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 */
1050	static void NOINLINE	1079	static void NOINLINE
1051	cctx_prepare (pTHX_ coro_cctx *cctx)	1080	cctx_prepare (pTHX_ coro_cctx *cctx)
1052	{	1081	{
1053	dSP;
1054	UNOP myop;
1055
1056	PL_top_env = &PL_start_env;	1082	PL_top_env = &PL_start_env;
1057		1083
1058	if (cctx->flags & CC_TRACE)	1084	if (cctx->flags & CC_TRACE)
1059	PL_runops = runops_trace;	1085	PL_runops = runops_trace;
1060		1086
1061	Zero (&myop, 1, UNOP);	1087	/* we already must be in an SLF call, there is no other valid way
1062	myop.op_next = PL_op;	1088	* that can lead to creation of a new cctx */
1063	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));
1064		1091
1065	PUSHMARK (SP);	1092	cctx_ssl_cctx = cctx;
1066	EXTEND (SP, 2);	1093	cctx_ssl_frame = slf_frame;
1067	PUSHs (sv_2mortal (newSViv ((IV)cctx)));	1094
1068	PUSHs ((SV *)get_cv ("Coro::State::_cctx_init", FALSE));	1095	slf_frame.prepare = slf_prepare_set_stacklevel;
1069	PUTBACK;	1096	slf_frame.check = slf_check_set_stacklevel;
1070	PL_op = (OP *)&myop;
1071	PL_op = PL_ppaddr[OP_ENTERSUB](aTHX);
1072	SPAGAIN;
1073	}	1097	}
1074		1098
1075	/* 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 */
1076	INLINE void	1100	INLINE void
1077	transfer_tail (pTHX)	1101	transfer_tail (pTHX)
1078	{	1102	{
1079	struct coro *next = (struct coro *)transfer_next;
1080	assert (!(transfer_next = 0)); /* just used for the side effect when asserts are enabled */
1081	assert (("FATAL: next coroutine was zero in transfer_tail (please report)", next));
1082
1083	free_coro_mortal (aTHX);	1103	free_coro_mortal (aTHX);
1084
1085	if (expect_false (next->throw))
1086	{
1087	SV *exception = sv_2mortal (next->throw);
1088
1089	next->throw = 0;
1090	sv_setsv (ERRSV, exception);
1091	croak (0);
1092	}
1093	}	1104	}
1094		1105
1095	/*	1106	/*
1096	* this is a _very_ stripped down perl interpreter ;)	1107	* this is a _very_ stripped down perl interpreter ;)
1097	*/	1108	*/
…		…
1112		1123
1113	/* inject a fake subroutine call to cctx_init */	1124	/* inject a fake subroutine call to cctx_init */
1114	cctx_prepare (aTHX_ (coro_cctx *)arg);	1125	cctx_prepare (aTHX_ (coro_cctx *)arg);
1115		1126
1116	/* cctx_run is the alternative tail of transfer() */	1127	/* cctx_run is the alternative tail of transfer() */
1117	/* TODO: throwing an exception here might be deadly, VERIFY */
1118	transfer_tail (aTHX);	1128	transfer_tail (aTHX);
1119		1129
1120	/* somebody or something will hit me for both perl_run and PL_restartop */	1130	/* somebody or something will hit me for both perl_run and PL_restartop */
1121	PL_restartop = PL_op;	1131	PL_restartop = PL_op;
1122	perl_run (PL_curinterp);	1132	perl_run (PL_curinterp);
…		…
1278	/** coroutine switching *****************************************************/	1288	/** coroutine switching *****************************************************/
1279		1289
1280	static void	1290	static void
1281	transfer_check (pTHX_ struct coro *prev, struct coro *next)	1291	transfer_check (pTHX_ struct coro *prev, struct coro *next)
1282	{	1292	{
		1293	/* TODO: throwing up here is considered harmful */
		1294
1283	if (expect_true (prev != next))	1295	if (expect_true (prev != next))
1284	{	1296	{
1285	if (expect_false (!(prev->flags & (CF_RUNNING | CF_NEW))))	1297	if (expect_false (!(prev->flags & (CF_RUNNING | CF_NEW))))
1286	croak ("Coro::State::transfer called with non-running/new prev Coro::State, but can only transfer from running or new states,");	1298	croak ("Coro::State::transfer called with non-running/new prev Coro::State, but can only transfer from running or new states,");
1287		1299
…		…
1297	#endif	1309	#endif
1298	}	1310	}
1299	}	1311	}
1300		1312
1301	/* always use the TRANSFER macro */	1313	/* always use the TRANSFER macro */
1302	static void NOINLINE	1314	static void NOINLINE /* noinline so we have a fixed stackframe */
1303	transfer (pTHX_ struct coro *prev, struct coro *next, int force_cctx)	1315	transfer (pTHX_ struct coro *prev, struct coro *next, int force_cctx)
1304	{	1316	{
1305	dSTACKLEVEL;	1317	dSTACKLEVEL;
1306		1318
1307	/* sometimes transfer is only called to set idle_sp */	1319	/* sometimes transfer is only called to set idle_sp */
…		…
1364	++next->usecount;	1376	++next->usecount;
1365		1377
1366	if (expect_true (!next->cctx))	1378	if (expect_true (!next->cctx))
1367	next->cctx = cctx_get (aTHX);	1379	next->cctx = cctx_get (aTHX);
1368		1380
1369	assert (("FATAL: transfer_next already nonzero in Coro (please report)", !transfer_next));
1370	transfer_next = next;
1371
1372	if (expect_false (prev__cctx != next->cctx))	1381	if (expect_false (prev__cctx != next->cctx))
1373	{	1382	{
1374	prev__cctx->top_env = PL_top_env;	1383	prev__cctx->top_env = PL_top_env;
1375	PL_top_env = next->cctx->top_env;	1384	PL_top_env = next->cctx->top_env;
1376	coro_transfer (&prev__cctx->cctx, &next->cctx->cctx);	1385	coro_transfer (&prev__cctx->cctx, &next->cctx->cctx);
…		…
1486	TRANSFER (ta, 1);	1495	TRANSFER (ta, 1);
1487	}	1496	}
1488		1497
1489	/** Coro ********************************************************************/	1498	/** Coro ********************************************************************/
1490		1499
1491	static void	1500	INLINE void
1492	coro_enq (pTHX_ SV *coro_sv)	1501	coro_enq (pTHX_ struct coro *coro)
1493	{	1502	{
1494	av_push (coro_ready [SvSTATE (coro_sv)->prio - PRIO_MIN], coro_sv);	1503	av_push (coro_ready [coro->prio - PRIO_MIN], SvREFCNT_inc_NN (coro->hv));
1495	}	1504	}
1496		1505
1497	static SV *	1506	INLINE SV *
1498	coro_deq (pTHX)	1507	coro_deq (pTHX)
1499	{	1508	{
1500	int prio;	1509	int prio;
1501		1510
1502	for (prio = PRIO_MAX - PRIO_MIN + 1; --prio >= 0; )	1511	for (prio = PRIO_MAX - PRIO_MIN + 1; --prio >= 0; )
…		…
1524	coro->flags |= CF_READY;	1533	coro->flags |= CF_READY;
1525		1534
1526	sv_hook = coro_nready ? 0 : coro_readyhook;	1535	sv_hook = coro_nready ? 0 : coro_readyhook;
1527	xs_hook = coro_nready ? 0 : coroapi.readyhook;	1536	xs_hook = coro_nready ? 0 : coroapi.readyhook;
1528		1537
1529	coro_enq (aTHX_ SvREFCNT_inc_NN (coro_sv));	1538	coro_enq (aTHX_ coro);
1530	++coro_nready;	1539	++coro_nready;
1531		1540
1532	if (sv_hook)	1541	if (sv_hook)
1533	{	1542	{
1534	dSP;	1543	dSP;
…		…
1536	ENTER;	1545	ENTER;
1537	SAVETMPS;	1546	SAVETMPS;
1538		1547
1539	PUSHMARK (SP);	1548	PUSHMARK (SP);
1540	PUTBACK;	1549	PUTBACK;
1541	call_sv (sv_hook, G_DISCARD);	1550	call_sv (sv_hook, G_VOID | G_DISCARD);
1542	SPAGAIN;
1543		1551
1544	FREETMPS;	1552	FREETMPS;
1545	LEAVE;	1553	LEAVE;
1546	}	1554	}
1547		1555
…		…
1574	ENTER;	1582	ENTER;
1575	SAVETMPS;	1583	SAVETMPS;
1576		1584
1577	PUSHMARK (SP);	1585	PUSHMARK (SP);
1578	PUTBACK;	1586	PUTBACK;
1579	call_sv (get_sv ("Coro::idle", FALSE), G_DISCARD);	1587	call_sv (get_sv ("Coro::idle", FALSE), G_VOID | G_DISCARD);
1580	SPAGAIN;
1581		1588
1582	FREETMPS;	1589	FREETMPS;
1583	LEAVE;	1590	LEAVE;
1584	continue;	1591	continue;
1585	}	1592	}
1586		1593
1587	ta->next = SvSTATE (next_sv);	1594	ta->next = SvSTATE_hv (next_sv);
1588		1595
1589	/* cannot transfer to destroyed coros, skip and look for next */	1596	/* cannot transfer to destroyed coros, skip and look for next */
1590	if (expect_false (ta->next->flags & CF_DESTROYED))	1597	if (expect_false (ta->next->flags & CF_DESTROYED))
1591	{	1598	{
1592	SvREFCNT_dec (next_sv);	1599	SvREFCNT_dec (next_sv);
…		…
1598	break;	1605	break;
1599	}	1606	}
1600		1607
1601	/* free this only after the transfer */	1608	/* free this only after the transfer */
1602	prev_sv = SvRV (coro_current);	1609	prev_sv = SvRV (coro_current);
1603	ta->prev = SvSTATE (prev_sv);	1610	ta->prev = SvSTATE_hv (prev_sv);
1604	TRANSFER_CHECK (*ta);	1611	TRANSFER_CHECK (*ta);
1605	assert (("FATAL: next coroutine isn't marked as ready in Coro (please report)", ta->next->flags & CF_READY));	1612	assert (("FATAL: next coroutine isn't marked as ready in Coro (please report)", ta->next->flags & CF_READY));
1606	ta->next->flags &= ~CF_READY;	1613	ta->next->flags &= ~CF_READY;
1607	SvRV_set (coro_current, next_sv);	1614	SvRV_set (coro_current, next_sv);
1608		1615
…		…
1692	if (coro->flags & CF_RUNNING)	1699	if (coro->flags & CF_RUNNING)
1693	PL_runops = RUNOPS_DEFAULT;	1700	PL_runops = RUNOPS_DEFAULT;
1694	else	1701	else
1695	coro->slot->runops = RUNOPS_DEFAULT;	1702	coro->slot->runops = RUNOPS_DEFAULT;
1696	}	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;
1697	}	1918	}
1698		1919
1699	/*****************************************************************************/	1920	/*****************************************************************************/
1700	/* PerlIO::cede */	1921	/* PerlIO::cede */
1701		1922
…		…
1770	PerlIOBuf_get_cnt,	1991	PerlIOBuf_get_cnt,
1771	PerlIOBuf_set_ptrcnt,	1992	PerlIOBuf_set_ptrcnt,
1772	};	1993	};
1773		1994
1774	/*****************************************************************************/	1995	/*****************************************************************************/
		1996	/* Coro::Semaphore */
1775		1997
1776	static const CV *slf_cv; /* for quick consistency check */
1777
1778	static UNOP slf_restore; /* restore stack as entersub did, for first-re-run */
1779	static SV *slf_arg0;
1780	static SV *slf_arg1;
1781	static SV *slf_arg2;
1782
1783	/* this restores the stack in the case we patched the entersub, to */
1784	/* recreate the stack frame as perl will on following calls */
1785	/* since entersub cleared the stack */
1786	static OP *
1787	pp_restore (pTHX)
1788	{
1789	dSP;
1790
1791	PUSHMARK (SP);
1792
1793	EXTEND (SP, 3);
1794	if (slf_arg0) PUSHs (sv_2mortal (slf_arg0));
1795	if (slf_arg1) PUSHs (sv_2mortal (slf_arg1));
1796	if (slf_arg2) PUSHs (sv_2mortal (slf_arg2));
1797	PUSHs ((SV *)CvGV (slf_cv));
1798
1799	RETURNOP (slf_restore.op_first);
1800	}
1801
1802	static void	1998	static void
1803	slf_prepare_set_stacklevel (pTHX_ struct coro_transfer_args *ta)
1804	{
1805	prepare_set_stacklevel (ta, (struct coro_cctx *)slf_frame.data);
1806	}
1807
1808	static void
1809	slf_init_set_stacklevel (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
1810	{
1811	assert (("FATAL: set_stacklevel needs the coro cctx as sole argument", items == 1));
1812
1813	frame->prepare = slf_prepare_set_stacklevel;
1814	frame->check = slf_check_nop;
1815	frame->data = (void *)SvIV (arg [0]);
1816	}
1817
1818	static void
1819	slf_prepare_transfer (pTHX_ struct coro_transfer_args *ta)
1820	{
1821	SV **arg = (SV **)slf_frame.data;
1822
1823	prepare_transfer (aTHX_ ta, arg [0], arg [1]);
1824	}
1825
1826	static void
1827	slf_init_transfer (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
1828	{
1829	if (items != 2)
1830	croak ("Coro::State::transfer (prev, next) expects two arguments, not %d,", items);
1831
1832	frame->prepare = slf_prepare_transfer;
1833	frame->check = slf_check_nop;
1834	frame->data = (void *)arg; /* let's hope it will stay valid */
1835	}
1836
1837	static void
1838	slf_init_schedule (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
1839	{
1840	frame->prepare = prepare_schedule;
1841	frame->check = slf_check_nop;
1842	}
1843
1844	static void
1845	slf_init_cede (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
1846	{
1847	frame->prepare = prepare_cede;
1848	frame->check = slf_check_nop;
1849	}
1850
1851	static void
1852	slf_init_cede_notself (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items)
1853	{
1854	frame->prepare = prepare_cede_notself;
1855	frame->check = slf_check_nop;
1856	}
1857
1858	/* we hijack an hopefully unused CV flag for our purposes */
1859	#define CVf_SLF 0x4000
1860
1861	/*
1862	* these not obviously related functions are all rolled into one
1863	* function to increase chances that they all will call transfer with the same
1864	* stack offset
1865	* SLF stands for "schedule-like-function".
1866	*/
1867	static OP *
1868	pp_slf (pTHX)
1869	{
1870	I32 checkmark; /* mark SP to see how many elements check has pushed */
1871
1872	/* set up the slf frame, unless it has already been set-up */
1873	/* the latter happens when a new coro has been started */
1874	/* or when a new cctx was attached to an existing coroutine */
1875	if (expect_true (!slf_frame.prepare))
1876	{
1877	/* first iteration */
1878	dSP;
1879	SV **arg = PL_stack_base + TOPMARK + 1;
1880	int items = SP - arg; /* args without function object */
1881	SV *gv = *sp;
1882
1883	/* do a quick consistency check on the "function" object, and if it isn't */
1884	/* for us, divert to the real entersub */
1885	if (SvTYPE (gv) != SVt_PVGV || !(CvFLAGS (GvCV (gv)) & CVf_SLF))
1886	return PL_ppaddr[OP_ENTERSUB](aTHX);
1887
1888	/* pop args */
1889	SP = PL_stack_base + POPMARK;
1890
1891	if (!(PL_op->op_flags & OPf_STACKED))
1892	{
1893	/* ampersand-form of call, use @_ instead of stack */
1894	AV *av = GvAV (PL_defgv);
1895	arg = AvARRAY (av);
1896	items = AvFILLp (av) + 1;
1897	}
1898
1899	PUTBACK;
1900
1901	/* now call the init function, which needs to set up slf_frame */
1902	((coro_slf_cb)CvXSUBANY (GvCV (gv)).any_ptr)
1903	(aTHX_ &slf_frame, GvCV (gv), arg, items);
1904	}
1905
1906	/* now that we have a slf_frame, interpret it! */
1907	/* we use a callback system not to make the code needlessly */
1908	/* complicated, but so we can run multiple perl coros from one cctx */
1909
1910	do
1911	{
1912	struct coro_transfer_args ta;
1913
1914	slf_frame.prepare (aTHX_ &ta);
1915	TRANSFER (ta, 0);
1916
1917	checkmark = PL_stack_sp - PL_stack_base;
1918	}
1919	while (slf_frame.check (aTHX_ &slf_frame));
1920
1921	{
1922	dSP;
1923	SV **bot = PL_stack_base + checkmark;
1924	int gimme = GIMME_V;
1925
1926	slf_frame.prepare = 0; /* invalidate the frame, so it gets initialised again next time */
1927
1928	/* make sure we put something on the stack in scalar context */
1929	if (gimme == G_SCALAR)
1930	{
1931	if (sp == bot)
1932	XPUSHs (&PL_sv_undef);
1933
1934	SP = bot + 1;
1935	}
1936
1937	PUTBACK;
1938	}
1939
1940	return NORMAL;
1941	}
1942
1943	static void
1944	api_execute_slf (pTHX_ CV *cv, coro_slf_cb init_cb, SV **arg, int items)
1945	{
1946	assert (("FATAL: SLF call with illegal CV value", !CvANON (cv)));
1947
1948	if (PL_op->op_ppaddr != PL_ppaddr [OP_ENTERSUB]
1949	&& PL_op->op_ppaddr != pp_slf)
1950	croak ("FATAL: Coro SLF calls can only be made normally, not via goto or any other means, caught");
1951
1952	if (items > 3)
1953	croak ("Coro only supports up to three arguments to SLF functions currently (not %d), caught", items);
1954
1955	CvFLAGS (cv) |= CVf_SLF;
1956	CvXSUBANY (cv).any_ptr = (void *)init_cb;
1957	slf_cv = cv;
1958
1959	/* we patch the op, and then re-run the whole call */
1960	/* we have to put the same argument on the stack for this to work */
1961	/* and this will be done by pp_restore */
1962	slf_restore.op_next = (OP *)&slf_restore;
1963	slf_restore.op_type = OP_CUSTOM;
1964	slf_restore.op_ppaddr = pp_restore;
1965	slf_restore.op_first = PL_op;
1966
1967	slf_arg0 = items > 0 ? SvREFCNT_inc (arg [0]) : 0;
1968	slf_arg1 = items > 1 ? SvREFCNT_inc (arg [1]) : 0;
1969	slf_arg2 = items > 2 ? SvREFCNT_inc (arg [2]) : 0;
1970
1971	PL_op->op_ppaddr = pp_slf;
1972
1973	PL_op = (OP *)&slf_restore;
1974	}
1975
1976	/*****************************************************************************/
1977
1978	static void
1979	coro_semaphore_adjust (AV *av, int adjust)	1999	coro_semaphore_adjust (pTHX_ AV *av, IV adjust)
1980	{	2000	{
1981	SV *count_sv = AvARRAY (av)[0];	2001	SV *count_sv = AvARRAY (av)[0];
1982	IV count = SvIVX (count_sv);	2002	IV count = SvIVX (count_sv);
1983		2003
1984	count += adjust;	2004	count += adjust;
1985	SvIVX (count_sv) = count;	2005	SvIVX (count_sv) = count;
1986		2006
1987	/* now wake up as many waiters as possible */	2007	/* now wake up as many waiters as are expected to lock */
1988	while (count > 0 && AvFILLp (av) >= count)	2008	while (count > 0 && AvFILLp (av) > 0)
1989	{	2009	{
1990	SV *cb;	2010	SV *cb;
1991		2011
1992	/* swap first two elements so we can shift a waiter */	2012	/* swap first two elements so we can shift a waiter */
1993	AvARRAY (av)[0] = AvARRAY (av)[1];	2013	AvARRAY (av)[0] = AvARRAY (av)[1];
…		…
2006	}	2026	}
2007		2027
2008	static void	2028	static void
2009	coro_semaphore_on_destroy (pTHX_ struct coro *coro)	2029	coro_semaphore_on_destroy (pTHX_ struct coro *coro)
2010	{	2030	{
2011	/* call $sem->adjust (0) to possibly wake up some waiters */	2031	/* call $sem->adjust (0) to possibly wake up some other waiters */
2012	coro_semaphore_adjust ((AV *)coro->slf_frame.data, 0);	2032	coro_semaphore_adjust (aTHX_ (AV *)coro->slf_frame.data, 0);
2013	}	2033	}
2014		2034
2015	static int	2035	static int
2016	slf_check_semaphore_down (pTHX_ struct CoroSLF *frame)	2036	slf_check_semaphore_down (pTHX_ struct CoroSLF *frame)
2017	{	2037	{
2018	AV *av = (AV *)frame->data;	2038	AV *av = (AV *)frame->data;
2019	SV *count_sv = AvARRAY (av)[0];	2039	SV *count_sv = AvARRAY (av)[0];
2020		2040
2021	if (SvIVX (count_sv) > 0)	2041	if (SvIVX (count_sv) > 0)
2022	{	2042	{
2023	SvSTATE (coro_current)->on_destroy = 0;	2043	SvSTATE_current->on_destroy = 0;
2024	SvIVX (count_sv) = SvIVX (count_sv) - 1;	2044	SvIVX (count_sv) = SvIVX (count_sv) - 1;
2025	return 0;	2045	return 0;
2026	}	2046	}
2027	else	2047	else
2028	{	2048	{
…		…
2047		2067
2048	if (SvIVX (AvARRAY (av)[0]) > 0)	2068	if (SvIVX (AvARRAY (av)[0]) > 0)
2049	{	2069	{
2050	frame->data = (void *)av;	2070	frame->data = (void *)av;
2051	frame->prepare = prepare_nop;	2071	frame->prepare = prepare_nop;
		2072	SvSTATE_current->on_destroy = coro_semaphore_on_destroy;
2052	}	2073	}
2053	else	2074	else
2054	{	2075	{
2055	av_push (av, SvREFCNT_inc (SvRV (coro_current)));	2076	av_push (av, SvREFCNT_inc (SvRV (coro_current)));
2056		2077
2057	frame->data = (void *)sv_2mortal (SvREFCNT_inc ((SV *)av));	2078	frame->data = (void *)sv_2mortal (SvREFCNT_inc ((SV *)av));
2058	frame->prepare = prepare_schedule;	2079	frame->prepare = prepare_schedule;
2059		2080
2060	/* to avoid race conditions when a woken-up coro gets terminated */	2081	/* to avoid race conditions when a woken-up coro gets terminated */
2061	/* we arrange for a temporary on_destroy that calls adjust (0) */	2082	/* we arrange for a temporary on_destroy that calls adjust (0) */
		2083	assert (!SvSTATE_current->on_destroy);//D
2062	SvSTATE (coro_current)->on_destroy = coro_semaphore_on_destroy;	2084	SvSTATE_current->on_destroy = coro_semaphore_on_destroy;
2063	}	2085	}
2064		2086
2065	frame->check = slf_check_semaphore_down;	2087	frame->check = slf_check_semaphore_down;
2066		2088
2067	}	2089	}
2068		2090
2069	/*****************************************************************************/	2091	/*****************************************************************************/
		2092	/* gensub: simple closure generation utility */
2070		2093
2071	#define GENSUB_ARG CvXSUBANY (cv).any_ptr	2094	#define GENSUB_ARG CvXSUBANY (cv).any_ptr
2072		2095
2073	/* create a closure from XS, returns a code reference */	2096	/* create a closure from XS, returns a code reference */
2074	/* the arg can be accessed via GENSUB_ARG from the callback */	2097	/* the arg can be accessed via GENSUB_ARG from the callback */
2075	/* the callback must use dXSARGS/XSRETURN */	2098	/* the callback must use dXSARGS/XSRETURN */
2076	static SV *	2099	static SV *
2077	gensub (pTHX_ void (*xsub)(pTHX_ CV *), void *arg)	2100	gensub (pTHX_ void (*xsub)(pTHX_ CV *), void *arg)
2078	{	2101	{
2079	CV *cv = (CV *)NEWSV (0, 0);	2102	CV *cv = (CV *)newSV (0);
2080		2103
2081	sv_upgrade ((SV *)cv, SVt_PVCV);	2104	sv_upgrade ((SV *)cv, SVt_PVCV);
2082		2105
2083	CvANON_on (cv);	2106	CvANON_on (cv);
2084	CvISXSUB_on (cv);	2107	CvISXSUB_on (cv);
…		…
2087		2110
2088	return newRV_noinc ((SV *)cv);	2111	return newRV_noinc ((SV *)cv);
2089	}	2112	}
2090		2113
2091	/*****************************************************************************/	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;
		2271	}
		2272
		2273	/*****************************************************************************/
2092		2274
2093	MODULE = Coro::State PACKAGE = Coro::State PREFIX = api_	2275	MODULE = Coro::State PACKAGE = Coro::State PREFIX = api_
2094		2276
2095	PROTOTYPES: DISABLE	2277	PROTOTYPES: DISABLE
2096		2278
2097	BOOT:	2279	BOOT:
2098	{	2280	{
2099	#ifdef USE_ITHREADS	2281	#ifdef USE_ITHREADS
2100	MUTEX_INIT (&coro_lock);
2101	# if CORO_PTHREAD	2282	# if CORO_PTHREAD
2102	coro_thx = PERL_GET_CONTEXT;	2283	coro_thx = PERL_GET_CONTEXT;
2103	# endif	2284	# endif
2104	#endif	2285	#endif
2105	BOOT_PAGESIZE;	2286	BOOT_PAGESIZE;
…		…
2192	}	2373	}
2193	OUTPUT:	2374	OUTPUT:
2194	RETVAL	2375	RETVAL
2195		2376
2196	void	2377	void
2197	_set_stacklevel (...)
2198	CODE:
2199	api_execute_slf (aTHX_ cv, slf_init_set_stacklevel, &ST (0), items);
2200
2201	void
2202	transfer (...)	2378	transfer (...)
2203	PROTOTYPE: $$	2379	PROTOTYPE: $$
2204	CODE:	2380	CODE:
2205	api_execute_slf (aTHX_ cv, slf_init_transfer, &ST (0), items);	2381	CORO_EXECUTE_SLF_XS (slf_init_transfer);
2206		2382
2207	bool	2383	bool
2208	_destroy (SV *coro_sv)	2384	_destroy (SV *coro_sv)
2209	CODE:	2385	CODE:
2210	RETVAL = coro_state_destroy (aTHX_ SvSTATE (coro_sv));	2386	RETVAL = coro_state_destroy (aTHX_ SvSTATE (coro_sv));
…		…
2328		2504
2329	void	2505	void
2330	throw (Coro::State self, SV *throw = &PL_sv_undef)	2506	throw (Coro::State self, SV *throw = &PL_sv_undef)
2331	PROTOTYPE: $;$	2507	PROTOTYPE: $;$
2332	CODE:	2508	CODE:
		2509	{
		2510	struct coro *current = SvSTATE_current;
		2511	SV **throwp = self == current ? &coro_throw : &self->throw;
2333	SvREFCNT_dec (self->throw);	2512	SvREFCNT_dec (*throwp);
2334	self->throw = SvOK (throw) ? newSVsv (throw) : 0;	2513	*throwp = SvOK (throw) ? newSVsv (throw) : 0;
		2514	}
2335		2515
2336	void	2516	void
2337	api_trace (SV *coro, int flags = CC_TRACE | CC_TRACE_SUB)	2517	api_trace (SV *coro, int flags = CC_TRACE | CC_TRACE_SUB)
2338	PROTOTYPE: $;$	2518	PROTOTYPE: $;$
2339	C_ARGS: aTHX_ coro, flags	2519	C_ARGS: aTHX_ coro, flags
…		…
2370		2550
2371	void	2551	void
2372	force_cctx ()	2552	force_cctx ()
2373	PROTOTYPE:	2553	PROTOTYPE:
2374	CODE:	2554	CODE:
2375	struct coro *coro = SvSTATE (coro_current);
2376	coro->cctx->idle_sp = 0;	2555	SvSTATE_current->cctx->idle_sp = 0;
2377		2556
2378	void	2557	void
2379	swap_defsv (Coro::State self)	2558	swap_defsv (Coro::State self)
2380	PROTOTYPE: $	2559	PROTOTYPE: $
2381	ALIAS:	2560	ALIAS:
…		…
2389	SV **dst = ix ? (SV **)&self->slot->defav : (SV **)&self->slot->defsv;	2568	SV **dst = ix ? (SV **)&self->slot->defav : (SV **)&self->slot->defsv;
2390		2569
2391	SV *tmp = *src; *src = *dst; *dst = tmp;	2570	SV *tmp = *src; *src = *dst; *dst = tmp;
2392	}	2571	}
2393		2572
		2573
2394	MODULE = Coro::State PACKAGE = Coro	2574	MODULE = Coro::State PACKAGE = Coro
2395		2575
2396	BOOT:	2576	BOOT:
2397	{	2577	{
2398	int i;	2578	int i;
…		…
2434	}	2614	}
2435		2615
2436	void	2616	void
2437	schedule (...)	2617	schedule (...)
2438	CODE:	2618	CODE:
2439	api_execute_slf (aTHX_ cv, slf_init_schedule, &ST (0), 0);	2619	CORO_EXECUTE_SLF_XS (slf_init_schedule);
2440		2620
2441	void	2621	void
2442	cede (...)	2622	cede (...)
2443	CODE:	2623	CODE:
2444	api_execute_slf (aTHX_ cv, slf_init_cede, &ST (0), 0);	2624	CORO_EXECUTE_SLF_XS (slf_init_cede);
2445		2625
2446	void	2626	void
2447	cede_notself (...)	2627	cede_notself (...)
2448	CODE:	2628	CODE:
2449	api_execute_slf (aTHX_ cv, slf_init_cede_notself, &ST (0), 0);	2629	CORO_EXECUTE_SLF_XS (slf_init_cede_notself);
2450		2630
2451	void	2631	void
2452	_set_current (SV *current)	2632	_set_current (SV *current)
2453	PROTOTYPE: $	2633	PROTOTYPE: $
2454	CODE:	2634	CODE:
…		…
2504	# for async_pool speedup	2684	# for async_pool speedup
2505	void	2685	void
2506	_pool_1 (SV *cb)	2686	_pool_1 (SV *cb)
2507	CODE:	2687	CODE:
2508	{	2688	{
2509	struct coro *coro = SvSTATE (coro_current);
2510	HV *hv = (HV *)SvRV (coro_current);	2689	HV *hv = (HV *)SvRV (coro_current);
		2690	struct coro *coro = SvSTATE_hv ((SV *)hv);
2511	AV *defav = GvAV (PL_defgv);	2691	AV *defav = GvAV (PL_defgv);
2512	SV *invoke = hv_delete (hv, "_invoke", sizeof ("_invoke") - 1, 0);	2692	SV *invoke = hv_delete (hv, "_invoke", sizeof ("_invoke") - 1, 0);
2513	AV *invoke_av;	2693	AV *invoke_av;
2514	int i, len;	2694	int i, len;
2515		2695
…		…
2542		2722
2543	void	2723	void
2544	_pool_2 (SV *cb)	2724	_pool_2 (SV *cb)
2545	CODE:	2725	CODE:
2546	{	2726	{
2547	struct coro *coro = SvSTATE (coro_current);	2727	HV *hv = (HV *)SvRV (coro_current);
		2728	struct coro *coro = SvSTATE_hv ((SV *)hv);
2548		2729
2549	sv_setsv (cb, &PL_sv_undef);	2730	sv_setsv (cb, &PL_sv_undef);
2550		2731
2551	SvREFCNT_dec ((SV *)PL_defoutgv); PL_defoutgv = (GV *)coro->saved_deffh;	2732	SvREFCNT_dec ((SV *)PL_defoutgv); PL_defoutgv = (GV *)coro->saved_deffh;
2552	coro->saved_deffh = 0;	2733	coro->saved_deffh = 0;
…		…
2559	SvREFCNT_dec (old);	2740	SvREFCNT_dec (old);
2560	croak ("\3async_pool terminate\2\n");	2741	croak ("\3async_pool terminate\2\n");
2561	}	2742	}
2562		2743
2563	av_clear (GvAV (PL_defgv));	2744	av_clear (GvAV (PL_defgv));
2564	hv_store ((HV *)SvRV (coro_current), "desc", sizeof ("desc") - 1,	2745	hv_store (hv, "desc", sizeof ("desc") - 1,
2565	newSVpvn ("[async_pool idle]", sizeof ("[async_pool idle]") - 1), 0);	2746	newSVpvn ("[async_pool idle]", sizeof ("[async_pool idle]") - 1), 0);
2566		2747
2567	coro->prio = 0;	2748	coro->prio = 0;
2568		2749
2569	if (coro->cctx && (coro->cctx->flags & CC_TRACE))	2750	if (coro->cctx && (coro->cctx->flags & CC_TRACE))
…		…
2571		2752
2572	av_push (av_async_pool, newSVsv (coro_current));	2753	av_push (av_async_pool, newSVsv (coro_current));
2573	}	2754	}
2574		2755
2575		2756
2576	MODULE = Coro::State PACKAGE = Coro::AIO
2577
2578	void
2579	_get_state (SV *self)
2580	PROTOTYPE: $
2581	PPCODE:
2582	{
2583	AV *defav = GvAV (PL_defgv);
2584	AV *av = newAV ();
2585	int i;
2586	SV *data_sv = newSV (sizeof (struct io_state));
2587	struct io_state *data = (struct io_state *)SvPVX (data_sv);
2588	SvCUR_set (data_sv, sizeof (struct io_state));
2589	SvPOK_only (data_sv);
2590
2591	data->errorno = errno;
2592	data->laststype = PL_laststype;
2593	data->laststatval = PL_laststatval;
2594	data->statcache = PL_statcache;
2595
2596	av_extend (av, AvFILLp (defav) + 1 + 1);
2597
2598	for (i = 0; i <= AvFILLp (defav); ++i)
2599	av_push (av, SvREFCNT_inc_NN (AvARRAY (defav)[i]));
2600
2601	av_push (av, data_sv);
2602
2603	XPUSHs (sv_2mortal (newRV_noinc ((SV *)av)));
2604
2605	api_ready (aTHX_ self);
2606	}
2607
2608	void
2609	_set_state (SV *state)
2610	PROTOTYPE: $
2611	PPCODE:
2612	{
2613	AV *av = (AV *)SvRV (state);
2614	struct io_state *data = (struct io_state *)SvPVX (AvARRAY (av)[AvFILLp (av)]);
2615	int i;
2616
2617	errno = data->errorno;
2618	PL_laststype = data->laststype;
2619	PL_laststatval = data->laststatval;
2620	PL_statcache = data->statcache;
2621
2622	EXTEND (SP, AvFILLp (av));
2623	for (i = 0; i < AvFILLp (av); ++i)
2624	PUSHs (sv_2mortal (SvREFCNT_inc_NN (AvARRAY (av)[i])));
2625	}
2626
2627
2628	MODULE = Coro::State PACKAGE = Coro::AnyEvent
2629
2630	BOOT:
2631	sv_activity = coro_get_sv (aTHX_ "Coro::AnyEvent::ACTIVITY", TRUE);
2632
2633	void
2634	_schedule (...)
2635	CODE:
2636	{
2637	static int incede;
2638
2639	api_cede_notself (aTHX);
2640
2641	++incede;
2642	while (coro_nready >= incede && api_cede (aTHX))
2643	;
2644
2645	sv_setsv (sv_activity, &PL_sv_undef);
2646	if (coro_nready >= incede)
2647	{
2648	PUSHMARK (SP);
2649	PUTBACK;
2650	call_pv ("Coro::AnyEvent::_activity", G_DISCARD | G_EVAL);
2651	SPAGAIN;
2652	}
2653
2654	--incede;
2655	}
2656
2657
2658	MODULE = Coro::State PACKAGE = PerlIO::cede	2757	MODULE = Coro::State PACKAGE = PerlIO::cede
2659		2758
2660	BOOT:	2759	BOOT:
2661	PerlIO_define_layer (aTHX_ &PerlIO_cede);	2760	PerlIO_define_layer (aTHX_ &PerlIO_cede);
		2761
2662		2762
2663	MODULE = Coro::State PACKAGE = Coro::Semaphore	2763	MODULE = Coro::State PACKAGE = Coro::Semaphore
2664		2764
2665	SV *	2765	SV *
2666	new (SV *klass, SV *count_ = 0)	2766	new (SV *klass, SV *count_ = 0)
2667	CODE:	2767	CODE:
2668	{	2768	{
2669	/* 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 */
2670	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;
2671	av_push (av, newSViv (count_ && SvOK (count_) ? SvIV (count_) : 1));	2779	ary [0] = newSViv (count_ && SvOK (count_) ? SvIV (count_) : 1);
		2780
2672	RETVAL = sv_bless (newRV_noinc ((SV *)av), GvSTASH (CvGV (cv)));	2781	RETVAL = sv_bless (newRV_noinc ((SV *)av), GvSTASH (CvGV (cv)));
2673	}	2782	}
2674	OUTPUT:	2783	OUTPUT:
2675	RETVAL	2784	RETVAL
2676		2785
…		…
2684	void	2793	void
2685	up (SV *self, int adjust = 1)	2794	up (SV *self, int adjust = 1)
2686	ALIAS:	2795	ALIAS:
2687	adjust = 1	2796	adjust = 1
2688	CODE:	2797	CODE:
2689	coro_semaphore_adjust ((AV *)SvRV (self), ix ? adjust : 1);	2798	coro_semaphore_adjust (aTHX_ (AV *)SvRV (self), ix ? adjust : 1);
2690		2799
2691	void	2800	void
2692	down (SV *self)	2801	down (SV *self)
2693	CODE:	2802	CODE:
2694	api_execute_slf (aTHX_ cv, slf_init_semaphore_down, &ST (0), 1);	2803	CORO_EXECUTE_SLF_XS (slf_init_semaphore_down);
2695		2804
2696	void	2805	void
2697	try (SV *self)	2806	try (SV *self)
2698	PPCODE:	2807	PPCODE:
2699	{	2808	{
…		…
2722	else	2831	else
2723	{	2832	{
2724	int i;	2833	int i;
2725	EXTEND (SP, AvFILLp (av) + 1 - 1);	2834	EXTEND (SP, AvFILLp (av) + 1 - 1);
2726	for (i = 1; i <= AvFILLp (av); ++i)	2835	for (i = 1; i <= AvFILLp (av); ++i)
2727	PUSHs (newSVsv (AvARRAY (av)[i]));	2836	PUSHs (sv_2mortal (newRV_inc (AvARRAY (av)[i])));
2728	}	2837	}
2729	}	2838	}
2730		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