ViewVC Help
View File | Revision Log | Show Annotations | Download File
/cvs/Coro/Coro/State.xs
(Generate patch)

Comparing Coro/Coro/State.xs (file contents):
Revision 1.88 by root, Fri Nov 24 15:34:33 2006 UTC vs.
Revision 1.110 by root, Tue Nov 28 23:08:07 2006 UTC

1#define PERL_NO_GET_CONTEXT
2
3#include "libcoro/coro.c" 1#include "libcoro/coro.c"
4 2
5#include "EXTERN.h" 3#include "EXTERN.h"
6#include "perl.h" 4#include "perl.h"
7#include "XSUB.h" 5#include "XSUB.h"
8 6
9#include "patchlevel.h" 7#include "patchlevel.h"
10 8
11#if PERL_VERSION < 6 9#if USE_VALGRIND
10# include <valgrind/valgrind.h>
11#endif
12
13#define PERL_VERSION_ATLEAST(a,b,c) \
14 (PERL_REVISION > (a) \
15 || (PERL_REVISION == (a) \
16 && (PERL_VERSION > (b) \
17 || (PERL_VERSION == (b) && PERLSUBVERSION >= (c)))))
18
19#if !PERL_VERSION_ATLEAST (5,6,0)
12# ifndef PL_ppaddr 20# ifndef PL_ppaddr
13# define PL_ppaddr ppaddr 21# define PL_ppaddr ppaddr
14# endif 22# endif
15# ifndef call_sv 23# ifndef call_sv
16# define call_sv perl_call_sv 24# define call_sv perl_call_sv
27# ifndef IS_PADCONST 35# ifndef IS_PADCONST
28# define IS_PADCONST(v) 0 36# define IS_PADCONST(v) 0
29# endif 37# endif
30#endif 38#endif
31 39
40#include <stdio.h>
32#include <errno.h> 41#include <errno.h>
33 42
34#if !__i386 && !__x86_64 && !__powerpc && !__m68k && !__alpha && !__mips && !__sparc64 43#if !__i386 && !__x86_64 && !__powerpc && !__m68k && !__alpha && !__mips && !__sparc64
35# undef STACKGUARD 44# undef STACKGUARD
36#endif 45#endif
53# ifndef PAGESIZE 62# ifndef PAGESIZE
54# define PAGESIZE pagesize 63# define PAGESIZE pagesize
55# define BOOT_PAGESIZE pagesize = sysconf (_SC_PAGESIZE) 64# define BOOT_PAGESIZE pagesize = sysconf (_SC_PAGESIZE)
56static long pagesize; 65static long pagesize;
57# else 66# else
58# define BOOT_PAGESIZE 67# define BOOT_PAGESIZE (void)0
59# endif 68# endif
69#else
70# define PAGESIZE 0
71# define BOOT_PAGESIZE (void)0
60#endif 72#endif
61
62#define SUB_INIT "Coro::State::initialize"
63 73
64/* The next macro should declare a variable stacklevel that contains and approximation 74/* The next macro should declare a variable stacklevel that contains and approximation
65 * to the current C stack pointer. Its property is that it changes with each call 75 * to the current C stack pointer. Its property is that it changes with each call
66 * and should be unique. */ 76 * and should be unique. */
77#define dSTACKLEVEL int stacklevel
67#define dSTACKLEVEL void *stacklevel = &stacklevel 78#define STACKLEVEL ((void *)&stacklevel)
68 79
69#define IN_DESTRUCT (PL_main_cv == Nullcv) 80#define IN_DESTRUCT (PL_main_cv == Nullcv)
70 81
71#define labs(l) ((l) >= 0 ? (l) : -(l)) 82#if __GNUC__ >= 3
83# define attribute(x) __attribute__(x)
84#else
85# define attribute(x)
86#endif
87
88#define NOINLINE attribute ((noinline))
72 89
73#include "CoroAPI.h" 90#include "CoroAPI.h"
91
92#define TRANSFER_SET_STACKLEVEL 0x8bfbfbfb /* magic cookie */
74 93
75#ifdef USE_ITHREADS 94#ifdef USE_ITHREADS
76static perl_mutex coro_mutex; 95static perl_mutex coro_mutex;
77# define LOCK do { MUTEX_LOCK (&coro_mutex); } while (0) 96# define LOCK do { MUTEX_LOCK (&coro_mutex); } while (0)
78# define UNLOCK do { MUTEX_UNLOCK (&coro_mutex); } while (0) 97# define UNLOCK do { MUTEX_UNLOCK (&coro_mutex); } while (0)
84static struct CoroAPI coroapi; 103static struct CoroAPI coroapi;
85static AV *main_mainstack; /* used to differentiate between $main and others */ 104static AV *main_mainstack; /* used to differentiate between $main and others */
86static HV *coro_state_stash, *coro_stash; 105static HV *coro_state_stash, *coro_stash;
87static SV *coro_mortal; /* will be freed after next transfer */ 106static SV *coro_mortal; /* will be freed after next transfer */
88 107
89/* this is actually not only the c stack but also c registers etc... */ 108static struct coro_cctx *cctx_first;
109static int cctx_count, cctx_idle;
110
111/* this is a structure representing a c-level coroutine */
90typedef struct { 112typedef struct coro_cctx {
91 int refcnt; /* pointer reference counter */ 113 struct coro_cctx *next;
92 int usecnt; /* shared by how many coroutines */
93 int gencnt; /* generation counter */
94 114
95 coro_context cctx; 115 /* the stack */
96
97 void *sptr; 116 void *sptr;
98 long ssize; /* positive == mmap, otherwise malloc */ 117 long ssize; /* positive == mmap, otherwise malloc */
99} coro_stack;
100 118
119 /* cpu state */
120 void *idle_sp; /* sp of top-level transfer/schedule/cede call */
121 JMPENV *top_env;
122 coro_context cctx;
123
124#if USE_VALGRIND
125 int valgrind_id;
126#endif
127} coro_cctx;
128
129/* this is a structure representing a perl-level coroutine */
101struct coro { 130struct coro {
102 /* the optional C context */ 131 /* the c coroutine allocated to this perl coroutine, if any */
103 coro_stack *stack; 132 coro_cctx *cctx;
104 void *cursp; 133
134 /* data associated with this coroutine (initial args) */
135 AV *args;
105 int gencnt; 136 int refcnt;
106 137
107 /* optionally saved, might be zero */ 138 /* optionally saved, might be zero */
108 AV *defav; 139 AV *defav;
109 SV *defsv; 140 SV *defsv;
110 SV *errsv; 141 SV *errsv;
111 142
112 /* saved global state not related to stacks */ 143#define VAR(name,type) type name;
113 U8 dowarn; 144# include "state.h"
114 I32 in_eval; 145#undef VAR
115
116 /* the stacks and related info (callchain etc..) */
117 PERL_SI *curstackinfo;
118 AV *curstack;
119 AV *mainstack;
120 SV **stack_sp;
121 OP *op;
122 SV **curpad;
123 AV *comppad;
124 CV *compcv;
125 SV **stack_base;
126 SV **stack_max;
127 SV **tmps_stack;
128 I32 tmps_floor;
129 I32 tmps_ix;
130 I32 tmps_max;
131 I32 *markstack;
132 I32 *markstack_ptr;
133 I32 *markstack_max;
134 I32 *scopestack;
135 I32 scopestack_ix;
136 I32 scopestack_max;
137 ANY *savestack;
138 I32 savestack_ix;
139 I32 savestack_max;
140 OP **retstack;
141 I32 retstack_ix;
142 I32 retstack_max;
143 PMOP *curpm;
144 COP *curcop;
145 JMPENV *top_env;
146 146
147 /* coro process data */ 147 /* coro process data */
148 int prio; 148 int prio;
149
150 /* data associated with this coroutine (initial args) */
151 AV *args;
152 int refcnt;
153}; 149};
154 150
155typedef struct coro *Coro__State; 151typedef struct coro *Coro__State;
156typedef struct coro *Coro__State_or_hashref; 152typedef struct coro *Coro__State_or_hashref;
157 153
158static AV * 154static AV *
159coro_clone_padlist (pTHX_ CV *cv) 155coro_clone_padlist (CV *cv)
160{ 156{
161 AV *padlist = CvPADLIST (cv); 157 AV *padlist = CvPADLIST (cv);
162 AV *newpadlist, *newpad; 158 AV *newpadlist, *newpad;
163 159
164 newpadlist = newAV (); 160 newpadlist = newAV ();
165 AvREAL_off (newpadlist); 161 AvREAL_off (newpadlist);
166#if PERL_VERSION < 9 162#if PERL_VERSION_ATLEAST (5,9,0)
163 Perl_pad_push (aTHX_ padlist, AvFILLp (padlist) + 1);
164#else
167 Perl_pad_push (aTHX_ padlist, AvFILLp (padlist) + 1, 1); 165 Perl_pad_push (aTHX_ padlist, AvFILLp (padlist) + 1, 1);
168#else
169 Perl_pad_push (aTHX_ padlist, AvFILLp (padlist) + 1);
170#endif 166#endif
171 newpad = (AV *)AvARRAY (padlist)[AvFILLp (padlist)]; 167 newpad = (AV *)AvARRAY (padlist)[AvFILLp (padlist)];
172 --AvFILLp (padlist); 168 --AvFILLp (padlist);
173 169
174 av_store (newpadlist, 0, SvREFCNT_inc (*av_fetch (padlist, 0, FALSE))); 170 av_store (newpadlist, 0, SvREFCNT_inc (*av_fetch (padlist, 0, FALSE)));
176 172
177 return newpadlist; 173 return newpadlist;
178} 174}
179 175
180static void 176static void
181free_padlist (pTHX_ AV *padlist) 177free_padlist (AV *padlist)
182{ 178{
183 /* may be during global destruction */ 179 /* may be during global destruction */
184 if (SvREFCNT (padlist)) 180 if (SvREFCNT (padlist))
185 { 181 {
186 I32 i = AvFILLp (padlist); 182 I32 i = AvFILLp (padlist);
207 AV *padlist; 203 AV *padlist;
208 AV *av = (AV *)mg->mg_obj; 204 AV *av = (AV *)mg->mg_obj;
209 205
210 /* casting is fun. */ 206 /* casting is fun. */
211 while (&PL_sv_undef != (SV *)(padlist = (AV *)av_pop (av))) 207 while (&PL_sv_undef != (SV *)(padlist = (AV *)av_pop (av)))
212 free_padlist (aTHX_ padlist); 208 free_padlist (padlist);
213 209
214 SvREFCNT_dec (av); 210 SvREFCNT_dec (av);
215 211
216 return 0; 212 return 0;
217} 213}
218 214
219#define PERL_MAGIC_coro PERL_MAGIC_ext 215#define PERL_MAGIC_coro PERL_MAGIC_ext
220 216
221static MGVTBL vtbl_coro = {0, 0, 0, 0, coro_cv_free}; 217static MGVTBL vtbl_coro = {0, 0, 0, 0, coro_cv_free};
222 218
219#define CORO_MAGIC(cv) \
220 SvMAGIC (cv) \
221 ? SvMAGIC (cv)->mg_type == PERL_MAGIC_coro \
222 ? SvMAGIC (cv) \
223 : mg_find ((SV *)cv, PERL_MAGIC_coro) \
224 : 0
225
223/* the next two functions merely cache the padlists */ 226/* the next two functions merely cache the padlists */
224static void 227static void
225get_padlist (pTHX_ CV *cv) 228get_padlist (CV *cv)
226{ 229{
227 MAGIC *mg = mg_find ((SV *)cv, PERL_MAGIC_coro); 230 MAGIC *mg = CORO_MAGIC (cv);
231 AV *av;
228 232
229 if (mg && AvFILLp ((AV *)mg->mg_obj) >= 0) 233 if (mg && AvFILLp ((av = (AV *)mg->mg_obj)) >= 0)
230 CvPADLIST (cv) = (AV *)av_pop ((AV *)mg->mg_obj); 234 CvPADLIST (cv) = (AV *)AvARRAY (av)[AvFILLp (av)--];
231 else 235 else
232 { 236 {
233#if 0 237#if 0
234 /* this should work - but it doesn't :( */ 238 /* this is probably cleaner, but also slower? */
235 CV *cp = Perl_cv_clone (aTHX_ cv); 239 CV *cp = Perl_cv_clone (cv);
236 CvPADLIST (cv) = CvPADLIST (cp); 240 CvPADLIST (cv) = CvPADLIST (cp);
237 CvPADLIST (cp) = 0; 241 CvPADLIST (cp) = 0;
238 SvREFCNT_dec (cp); 242 SvREFCNT_dec (cp);
239#else 243#else
240 CvPADLIST (cv) = coro_clone_padlist (aTHX_ cv); 244 CvPADLIST (cv) = coro_clone_padlist (cv);
241#endif 245#endif
242 } 246 }
243} 247}
244 248
245static void 249static void
246put_padlist (pTHX_ CV *cv) 250put_padlist (CV *cv)
247{ 251{
248 MAGIC *mg = mg_find ((SV *)cv, PERL_MAGIC_coro); 252 MAGIC *mg = CORO_MAGIC (cv);
253 AV *av;
249 254
250 if (!mg) 255 if (!mg)
251 { 256 {
252 sv_magic ((SV *)cv, 0, PERL_MAGIC_coro, 0, 0); 257 sv_magic ((SV *)cv, 0, PERL_MAGIC_coro, 0, 0);
253 mg = mg_find ((SV *)cv, PERL_MAGIC_coro); 258 mg = mg_find ((SV *)cv, PERL_MAGIC_coro);
254 mg->mg_virtual = &vtbl_coro; 259 mg->mg_virtual = &vtbl_coro;
255 mg->mg_obj = (SV *)newAV (); 260 mg->mg_obj = (SV *)newAV ();
256 } 261 }
257 262
258 av_push ((AV *)mg->mg_obj, (SV *)CvPADLIST (cv)); 263 av = (AV *)mg->mg_obj;
264
265 if (AvFILLp (av) >= AvMAX (av))
266 av_extend (av, AvMAX (av) + 1);
267
268 AvARRAY (av)[++AvFILLp (av)] = (SV *)CvPADLIST (cv);
259} 269}
260 270
261#define SB do { 271#define SB do {
262#define SE } while (0) 272#define SE } while (0)
263 273
264#define LOAD(state) load_state(aTHX_ (state)); 274#define LOAD(state) load_state((state));
265#define SAVE(state,flags) save_state(aTHX_ (state),(flags)); 275#define SAVE(state,flags) save_state((state),(flags));
266 276
267#define REPLACE_SV(sv,val) SB SvREFCNT_dec(sv); (sv) = (val); (val) = 0; SE 277#define REPLACE_SV(sv,val) SB SvREFCNT_dec(sv); (sv) = (val); (val) = 0; SE
268 278
269static void 279static void
270load_state(pTHX_ Coro__State c) 280load_state(Coro__State c)
271{ 281{
272 PL_dowarn = c->dowarn; 282#define VAR(name,type) PL_ ## name = c->name;
273 PL_in_eval = c->in_eval; 283# include "state.h"
274 284#undef VAR
275 PL_curstackinfo = c->curstackinfo;
276 PL_curstack = c->curstack;
277 PL_mainstack = c->mainstack;
278 PL_stack_sp = c->stack_sp;
279 PL_op = c->op;
280 PL_curpad = c->curpad;
281 PL_comppad = c->comppad;
282 PL_compcv = c->compcv;
283 PL_stack_base = c->stack_base;
284 PL_stack_max = c->stack_max;
285 PL_tmps_stack = c->tmps_stack;
286 PL_tmps_floor = c->tmps_floor;
287 PL_tmps_ix = c->tmps_ix;
288 PL_tmps_max = c->tmps_max;
289 PL_markstack = c->markstack;
290 PL_markstack_ptr = c->markstack_ptr;
291 PL_markstack_max = c->markstack_max;
292 PL_scopestack = c->scopestack;
293 PL_scopestack_ix = c->scopestack_ix;
294 PL_scopestack_max = c->scopestack_max;
295 PL_savestack = c->savestack;
296 PL_savestack_ix = c->savestack_ix;
297 PL_savestack_max = c->savestack_max;
298#if PERL_VERSION < 9
299 PL_retstack = c->retstack;
300 PL_retstack_ix = c->retstack_ix;
301 PL_retstack_max = c->retstack_max;
302#endif
303 PL_curpm = c->curpm;
304 PL_curcop = c->curcop;
305 PL_top_env = c->top_env;
306 285
307 if (c->defav) REPLACE_SV (GvAV (PL_defgv), c->defav); 286 if (c->defav) REPLACE_SV (GvAV (PL_defgv), c->defav);
308 if (c->defsv) REPLACE_SV (DEFSV , c->defsv); 287 if (c->defsv) REPLACE_SV (DEFSV , c->defsv);
309 if (c->errsv) REPLACE_SV (ERRSV , c->errsv); 288 if (c->errsv) REPLACE_SV (ERRSV , c->errsv);
310 289
313 CV *cv; 292 CV *cv;
314 293
315 /* now do the ugly restore mess */ 294 /* now do the ugly restore mess */
316 while ((cv = (CV *)POPs)) 295 while ((cv = (CV *)POPs))
317 { 296 {
318 AV *padlist = (AV *)POPs;
319
320 if (padlist)
321 {
322 put_padlist (aTHX_ cv); /* mark this padlist as available */ 297 put_padlist (cv); /* mark this padlist as available */
323 CvPADLIST(cv) = padlist; 298 CvDEPTH (cv) = PTR2IV (POPs);
324 } 299 CvPADLIST (cv) = (AV *)POPs;
325
326 ++CvDEPTH(cv);
327 } 300 }
328 301
329 PUTBACK; 302 PUTBACK;
330 } 303 }
331} 304}
332 305
333static void 306static void
334save_state(pTHX_ Coro__State c, int flags) 307save_state(Coro__State c, int flags)
335{ 308{
336 { 309 {
337 dSP; 310 dSP;
338 I32 cxix = cxstack_ix; 311 I32 cxix = cxstack_ix;
339 PERL_CONTEXT *ccstk = cxstack; 312 PERL_CONTEXT *ccstk = cxstack;
353 PERL_CONTEXT *cx = &ccstk[cxix--]; 326 PERL_CONTEXT *cx = &ccstk[cxix--];
354 327
355 if (CxTYPE(cx) == CXt_SUB) 328 if (CxTYPE(cx) == CXt_SUB)
356 { 329 {
357 CV *cv = cx->blk_sub.cv; 330 CV *cv = cx->blk_sub.cv;
331
358 if (CvDEPTH(cv)) 332 if (CvDEPTH (cv))
359 { 333 {
360 EXTEND (SP, CvDEPTH(cv)*2); 334 EXTEND (SP, 3);
361
362 while (--CvDEPTH(cv))
363 {
364 /* this tells the restore code to increment CvDEPTH */
365 PUSHs (Nullsv);
366 PUSHs ((SV *)cv);
367 }
368 335
369 PUSHs ((SV *)CvPADLIST(cv)); 336 PUSHs ((SV *)CvPADLIST(cv));
337 PUSHs (INT2PTR (SV *, CvDEPTH (cv)));
370 PUSHs ((SV *)cv); 338 PUSHs ((SV *)cv);
371 339
340 CvDEPTH (cv) = 0;
372 get_padlist (aTHX_ cv); 341 get_padlist (cv);
373 } 342 }
374 } 343 }
375#ifdef CXt_FORMAT 344#ifdef CXt_FORMAT
376 else if (CxTYPE(cx) == CXt_FORMAT) 345 else if (CxTYPE(cx) == CXt_FORMAT)
377 { 346 {
395 364
396 c->defav = flags & TRANSFER_SAVE_DEFAV ? (AV *)SvREFCNT_inc (GvAV (PL_defgv)) : 0; 365 c->defav = flags & TRANSFER_SAVE_DEFAV ? (AV *)SvREFCNT_inc (GvAV (PL_defgv)) : 0;
397 c->defsv = flags & TRANSFER_SAVE_DEFSV ? SvREFCNT_inc (DEFSV) : 0; 366 c->defsv = flags & TRANSFER_SAVE_DEFSV ? SvREFCNT_inc (DEFSV) : 0;
398 c->errsv = flags & TRANSFER_SAVE_ERRSV ? SvREFCNT_inc (ERRSV) : 0; 367 c->errsv = flags & TRANSFER_SAVE_ERRSV ? SvREFCNT_inc (ERRSV) : 0;
399 368
400 c->dowarn = PL_dowarn; 369#define VAR(name,type)c->name = PL_ ## name;
401 c->in_eval = PL_in_eval; 370# include "state.h"
402 371#undef VAR
403 c->curstackinfo = PL_curstackinfo;
404 c->curstack = PL_curstack;
405 c->mainstack = PL_mainstack;
406 c->stack_sp = PL_stack_sp;
407 c->op = PL_op;
408 c->curpad = PL_curpad;
409 c->comppad = PL_comppad;
410 c->compcv = PL_compcv;
411 c->stack_base = PL_stack_base;
412 c->stack_max = PL_stack_max;
413 c->tmps_stack = PL_tmps_stack;
414 c->tmps_floor = PL_tmps_floor;
415 c->tmps_ix = PL_tmps_ix;
416 c->tmps_max = PL_tmps_max;
417 c->markstack = PL_markstack;
418 c->markstack_ptr = PL_markstack_ptr;
419 c->markstack_max = PL_markstack_max;
420 c->scopestack = PL_scopestack;
421 c->scopestack_ix = PL_scopestack_ix;
422 c->scopestack_max = PL_scopestack_max;
423 c->savestack = PL_savestack;
424 c->savestack_ix = PL_savestack_ix;
425 c->savestack_max = PL_savestack_max;
426#if PERL_VERSION < 9
427 c->retstack = PL_retstack;
428 c->retstack_ix = PL_retstack_ix;
429 c->retstack_max = PL_retstack_max;
430#endif
431 c->curpm = PL_curpm;
432 c->curcop = PL_curcop;
433 c->top_env = PL_top_env;
434} 372}
435 373
436/* 374/*
437 * allocate various perl stacks. This is an exact copy 375 * allocate various perl stacks. This is an exact copy
438 * of perl.c:init_stacks, except that it uses less memory 376 * of perl.c:init_stacks, except that it uses less memory
439 * on the (sometimes correct) assumption that coroutines do 377 * on the (sometimes correct) assumption that coroutines do
440 * not usually need a lot of stackspace. 378 * not usually need a lot of stackspace.
441 */ 379 */
442static void 380static void
443coro_init_stacks (pTHX) 381coro_init_stacks ()
444{ 382{
445 LOCK;
446
447 PL_curstackinfo = new_stackinfo(96, 1024/sizeof(PERL_CONTEXT) - 1); 383 PL_curstackinfo = new_stackinfo(96, 1024/sizeof(PERL_CONTEXT) - 1);
448 PL_curstackinfo->si_type = PERLSI_MAIN; 384 PL_curstackinfo->si_type = PERLSI_MAIN;
449 PL_curstack = PL_curstackinfo->si_stack; 385 PL_curstack = PL_curstackinfo->si_stack;
450 PL_mainstack = PL_curstack; /* remember in case we switch stacks */ 386 PL_mainstack = PL_curstack; /* remember in case we switch stacks */
451 387
472 408
473 New(54,PL_savestack,96,ANY); 409 New(54,PL_savestack,96,ANY);
474 PL_savestack_ix = 0; 410 PL_savestack_ix = 0;
475 PL_savestack_max = 96; 411 PL_savestack_max = 96;
476 412
477#if PERL_VERSION < 9 413#if !PERL_VERSION_ATLEAST (5,9,0)
478 New(54,PL_retstack,8,OP*); 414 New(54,PL_retstack,8,OP*);
479 PL_retstack_ix = 0; 415 PL_retstack_ix = 0;
480 PL_retstack_max = 8; 416 PL_retstack_max = 8;
481#endif 417#endif
482
483 UNLOCK;
484} 418}
485 419
486/* 420/*
487 * destroy the stacks, the callchain etc... 421 * destroy the stacks, the callchain etc...
488 */ 422 */
489static void 423static void
490destroy_stacks(pTHX) 424coro_destroy_stacks ()
491{ 425{
492 if (!IN_DESTRUCT) 426 if (!IN_DESTRUCT)
493 { 427 {
494 /* is this ugly, I ask? */ 428 /* is this ugly, I ask? */
495 LEAVE_SCOPE (0); 429 LEAVE_SCOPE (0);
526 460
527 Safefree (PL_tmps_stack); 461 Safefree (PL_tmps_stack);
528 Safefree (PL_markstack); 462 Safefree (PL_markstack);
529 Safefree (PL_scopestack); 463 Safefree (PL_scopestack);
530 Safefree (PL_savestack); 464 Safefree (PL_savestack);
531#if PERL_VERSION < 9 465#if !PERL_VERSION_ATLEAST (5,9,0)
532 Safefree (PL_retstack); 466 Safefree (PL_retstack);
533#endif 467#endif
534} 468}
535 469
536static void 470static void
537allocate_stack (Coro__State ctx, int alloc) 471setup_coro (struct coro *coro)
538{
539 coro_stack *stack;
540
541 New (0, stack, 1, coro_stack);
542
543 stack->refcnt = 1;
544 stack->usecnt = 1;
545 stack->gencnt = ctx->gencnt = 0;
546
547 if (alloc)
548 {
549#if HAVE_MMAP
550 stack->ssize = ((STACKSIZE * sizeof (long) + PAGESIZE - 1) / PAGESIZE + STACKGUARD) * PAGESIZE; /* mmap should do allocate-on-write for us */
551 stack->sptr = mmap (0, stack->ssize, PROT_EXEC|PROT_READ|PROT_WRITE, MAP_PRIVATE|MAP_ANONYMOUS, 0, 0);
552 if (stack->sptr != (void *)-1)
553 {
554# if STACKGUARD
555 mprotect (stack->sptr, STACKGUARD * PAGESIZE, PROT_NONE);
556# endif
557 }
558 else
559#endif
560 {
561 stack->ssize = - (STACKSIZE * (long)sizeof (long));
562 New (0, stack->sptr, STACKSIZE, long);
563 }
564 }
565 else
566 stack->sptr = 0;
567
568 ctx->stack = stack;
569}
570
571static void
572deallocate_stack (Coro__State ctx)
573{
574 coro_stack *stack = ctx->stack;
575
576 ctx->stack = 0;
577
578 if (stack)
579 {
580 if (!--stack->refcnt)
581 {
582#ifdef HAVE_MMAP
583 if (stack->ssize > 0 && stack->sptr)
584 munmap (stack->sptr, stack->ssize);
585 else
586#endif
587 Safefree (stack->sptr);
588
589 Safefree (stack);
590 }
591 else if (ctx->gencnt == stack->gencnt)
592 --stack->usecnt;
593 }
594}
595
596static void
597setup_coro (void *arg)
598{ 472{
599 /* 473 /*
600 * emulate part of the perl startup here. 474 * emulate part of the perl startup here.
601 */ 475 */
602 dTHX;
603 dSP;
604 Coro__State ctx = (Coro__State)arg;
605 SV *sub_init = (SV *)get_cv (SUB_INIT, FALSE);
606 476
607 coro_init_stacks (aTHX); 477 coro_init_stacks ();
478
608 /*PL_curcop = 0;*/ 479 PL_curcop = 0;
609 /*PL_in_eval = PL_in_eval;*/ /* inherit */ 480 PL_in_eval = 0;
481 PL_curpm = 0;
482
483 {
484 dSP;
485 LOGOP myop;
486
487 /* I have no idea why this is needed, but it is */
488 PUSHMARK (SP);
489
610 SvREFCNT_dec (GvAV (PL_defgv)); 490 SvREFCNT_dec (GvAV (PL_defgv));
611 GvAV (PL_defgv) = ctx->args; ctx->args = 0; 491 GvAV (PL_defgv) = coro->args; coro->args = 0;
612 492
613 SPAGAIN;
614
615 if (ctx->stack)
616 {
617 ctx->cursp = 0;
618
619 PUSHMARK(SP);
620 PUTBACK;
621 (void) call_sv (sub_init, G_VOID|G_NOARGS|G_EVAL);
622
623 if (SvTRUE (ERRSV))
624 croak (NULL);
625 else
626 croak ("FATAL: CCTXT coroutine returned!");
627 }
628 else
629 {
630 UNOP myop;
631
632 Zero(&myop, 1, UNOP); 493 Zero (&myop, 1, LOGOP);
633 myop.op_next = Nullop; 494 myop.op_next = Nullop;
634 myop.op_flags = OPf_WANT_VOID; 495 myop.op_flags = OPf_WANT_VOID;
635 496
636 PL_op = (OP *)&myop; 497 PL_op = (OP *)&myop;
637 498
638 PUSHMARK(SP); 499 PUSHMARK (SP);
639 XPUSHs (sub_init); 500 XPUSHs ((SV *)get_cv ("Coro::State::_coro_init", FALSE));
640 /*
641 * the next line is slightly wrong, as PL_op->op_next
642 * is actually being executed so we skip the first op.
643 * that doesn't matter, though, since it is only
644 * pp_nextstate and we never return...
645 * ah yes, and I don't care anyways ;)
646 */
647 PUTBACK; 501 PUTBACK;
648 PL_op = PL_ppaddr[OP_ENTERSUB](aTHX); 502 PL_op = PL_ppaddr[OP_ENTERSUB](aTHX);
649 SPAGAIN; 503 SPAGAIN;
650 504
651 ENTER; /* necessary e.g. for dounwind */ 505 ENTER; /* necessary e.g. for dounwind */
652 } 506 }
653} 507}
654 508
655static void 509static void
656continue_coro (void *arg) 510free_coro_mortal ()
657{ 511{
658 /*
659 * this is a _very_ stripped down perl interpreter ;)
660 */
661 dTHX;
662 Coro__State ctx = (Coro__State)arg;
663
664 PL_top_env = &PL_start_env;
665
666 ctx->cursp = 0;
667 PL_op = PL_op->op_next;
668 CALLRUNOPS(aTHX);
669
670 abort ();
671}
672
673/* never call directly, always through the coro_state_transfer global variable */
674static void
675transfer_impl (pTHX_ struct coro *prev, struct coro *next, int flags)
676{
677 dSTACKLEVEL;
678
679 if (prev != next)
680 {
681 /* has this coro been created yet? */
682 if (next->mainstack)
683 {
684 LOCK;
685 SAVE (prev, flags);
686 LOAD (next);
687 UNLOCK;
688
689 /* mark this state as in-use */
690 next->mainstack = 0;
691 next->tmps_ix = -2;
692
693 /* stacklevel changed? if yes, grab the stack for us! */
694 if (flags & TRANSFER_SAVE_CCTXT)
695 {
696 if (!prev->stack)
697 allocate_stack (prev, 0);
698 else if (prev->cursp != stacklevel
699 && prev->stack->usecnt > 1)
700 {
701 prev->gencnt = ++prev->stack->gencnt;
702 prev->stack->usecnt = 1;
703 }
704
705 /* has our stack been invalidated? */
706 if (next->stack && next->stack->gencnt != next->gencnt)
707 {
708 deallocate_stack (next);
709 allocate_stack (next, 1);
710 coro_create (&(next->stack->cctx),
711 continue_coro, (void *)next,
712 next->stack->sptr, labs (next->stack->ssize));
713 }
714
715 coro_transfer (&(prev->stack->cctx), &(next->stack->cctx));
716 prev->cursp = stacklevel;
717 /* don't add any code here */
718 }
719 else
720 next->cursp = stacklevel;
721 }
722 else if (next->tmps_ix == -2)
723 croak ("tried to transfer to running coroutine");
724 else
725 {
726 LOCK;
727 SAVE (prev, -1); /* first get rid of the old state */
728 UNLOCK;
729
730 /* create the coroutine for the first time */
731 if (flags & TRANSFER_SAVE_CCTXT)
732 {
733 if (!prev->stack)
734 allocate_stack (prev, 0);
735
736 /* the new coroutine starts with start_env again */
737 PL_top_env = &PL_start_env;
738
739 if (prev->stack->sptr && flags & TRANSFER_LAZY_STACK)
740 {
741 setup_coro (next);
742 next->cursp = stacklevel;
743
744 prev->stack->refcnt++;
745 prev->stack->usecnt++;
746 next->stack = prev->stack;
747 next->gencnt = prev->gencnt;
748 }
749 else
750 {
751 assert (!next->stack);
752 allocate_stack (next, 1);
753 coro_create (&(next->stack->cctx),
754 setup_coro, (void *)next,
755 next->stack->sptr, labs (next->stack->ssize));
756 coro_transfer (&(prev->stack->cctx), &(next->stack->cctx));
757 prev->cursp = stacklevel;
758 /* don't add any code here */
759 }
760 }
761 else
762 {
763 setup_coro (next);
764 next->cursp = stacklevel;
765 }
766 }
767 }
768
769 LOCK;
770 if (coro_mortal) 512 if (coro_mortal)
771 { 513 {
772 SvREFCNT_dec (coro_mortal); 514 SvREFCNT_dec (coro_mortal);
773 coro_mortal = 0; 515 coro_mortal = 0;
774 } 516 }
517}
518
519static void NOINLINE
520prepare_cctx (coro_cctx *cctx)
521{
522 dSP;
523 LOGOP myop;
524
525 Zero (&myop, 1, LOGOP);
526 myop.op_next = PL_op;
527 myop.op_flags = OPf_WANT_VOID;
528
529 sv_setiv (get_sv ("Coro::State::_cctx", FALSE), PTR2IV (cctx));
530
531 PUSHMARK (SP);
532 XPUSHs ((SV *)get_cv ("Coro::State::_cctx_init", FALSE));
533 PUTBACK;
534 PL_restartop = PL_ppaddr[OP_ENTERSUB](aTHX);
535 SPAGAIN;
536}
537
538static void
539coro_run (void *arg)
540{
541 /* coro_run is the alternative epilogue of transfer() */
775 UNLOCK; 542 UNLOCK;
776}
777 543
778/* use this function pointer to call the above function */ 544 /*
779/* this is done to increase chances of the compiler not inlining the call */ 545 * this is a _very_ stripped down perl interpreter ;)
780void (*coro_state_transfer)(pTHX_ struct coro *prev, struct coro *next, int flags) = transfer_impl; 546 */
547 PL_top_env = &PL_start_env;
548
549 /* inject call to cctx_init */
550 prepare_cctx ((coro_cctx *)arg);
551
552 /* somebody will hit me for both perl_run and PL_restartop */
553 perl_run (PL_curinterp);
554
555 fputs ("FATAL: C coroutine fell over the edge of the world, aborting. Did you call exit in a coroutine?\n", stderr);
556 abort ();
557}
558
559static coro_cctx *
560cctx_new ()
561{
562 coro_cctx *cctx;
563
564 ++cctx_count;
565
566 New (0, cctx, 1, coro_cctx);
567
568#if HAVE_MMAP
569
570 cctx->ssize = ((STACKSIZE * sizeof (long) + PAGESIZE - 1) / PAGESIZE + STACKGUARD) * PAGESIZE;
571 /* mmap suppsedly does allocate-on-write for us */
572 cctx->sptr = mmap (0, cctx->ssize, PROT_EXEC|PROT_READ|PROT_WRITE, MAP_PRIVATE|MAP_ANONYMOUS, 0, 0);
573
574 if (cctx->sptr == (void *)-1)
575 {
576 perror ("FATAL: unable to mmap stack for coroutine");
577 _exit (EXIT_FAILURE);
578 }
579
580# if STACKGUARD
581 mprotect (cctx->sptr, STACKGUARD * PAGESIZE, PROT_NONE);
582# endif
583
584#else
585
586 cctx->ssize = STACKSIZE * (long)sizeof (long);
587 New (0, cctx->sptr, STACKSIZE, long);
588
589 if (!cctx->sptr)
590 {
591 perror ("FATAL: unable to malloc stack for coroutine");
592 _exit (EXIT_FAILURE);
593 }
594
595#endif
596
597#if USE_VALGRIND
598 cctx->valgrind_id = VALGRIND_STACK_REGISTER (
599 STACKGUARD * PAGESIZE + (char *)cctx->sptr,
600 cctx->ssize + (char *)cctx->sptr
601 );
602#endif
603
604 coro_create (&cctx->cctx, coro_run, (void *)cctx, cctx->sptr, cctx->ssize);
605
606 return cctx;
607}
608
609static void
610cctx_free (coro_cctx *cctx)
611{
612 if (!cctx)
613 return;
614
615 --cctx_count;
616
617#if USE_VALGRIND
618 VALGRIND_STACK_DEREGISTER (cctx->valgrind_id);
619#endif
620
621#if HAVE_MMAP
622 munmap (cctx->sptr, cctx->ssize);
623#else
624 Safefree (cctx->sptr);
625#endif
626
627 Safefree (cctx);
628}
629
630static coro_cctx *
631cctx_get ()
632{
633 coro_cctx *cctx;
634
635 if (cctx_first)
636 {
637 --cctx_idle;
638 cctx = cctx_first;
639 cctx_first = cctx->next;
640 }
641 else
642 {
643 cctx = cctx_new ();
644 PL_op = PL_op->op_next;
645 }
646
647 return cctx;
648}
649
650static void
651cctx_put (coro_cctx *cctx)
652{
653 ++cctx_idle;
654 cctx->next = cctx_first;
655 cctx_first = cctx;
656}
657
658/* never call directly, always through the coro_state_transfer global variable */
659static void NOINLINE
660transfer (struct coro *prev, struct coro *next, int flags)
661{
662 dSTACKLEVEL;
663
664 /* sometimes transfer is only called to set idle_sp */
665 if (flags == TRANSFER_SET_STACKLEVEL)
666 ((coro_cctx *)prev)->idle_sp = STACKLEVEL;
667 else if (prev != next)
668 {
669 coro_cctx *prev__cctx;
670
671 LOCK;
672
673 if (next->mainstack)
674 {
675 /* coroutine already started */
676 SAVE (prev, flags);
677 LOAD (next);
678 }
679 else
680 {
681 /* need to start coroutine */
682 /* first get rid of the old state */
683 SAVE (prev, -1);
684 /* setup coroutine call */
685 setup_coro (next);
686 /* need a stack */
687 next->cctx = 0;
688 }
689
690 if (!prev->cctx)
691 /* create a new empty context */
692 Newz (0, prev->cctx, 1, coro_cctx);
693
694 prev__cctx = prev->cctx;
695
696 /* possibly "free" the cctx */
697 if (prev__cctx->idle_sp == STACKLEVEL)
698 {
699 assert (PL_top_env == prev__cctx->top_env);//D
700 cctx_put (prev__cctx);
701 prev->cctx = 0;
702 }
703
704 if (!next->cctx)
705 next->cctx = cctx_get ();
706
707 if (prev__cctx != next->cctx)
708 {
709 prev__cctx->top_env = PL_top_env;
710 PL_top_env = next->cctx->top_env;
711 coro_transfer (&prev__cctx->cctx, &next->cctx->cctx);
712 }
713
714 free_coro_mortal ();
715
716 UNLOCK;
717 }
718}
719
720struct transfer_args
721{
722 struct coro *prev, *next;
723 int flags;
724};
725
726#define TRANSFER(ta) transfer ((ta).prev, (ta).next, (ta).flags)
781 727
782static void 728static void
783coro_state_destroy (struct coro *coro) 729coro_state_destroy (struct coro *coro)
784{ 730{
785 if (coro->refcnt--) 731 if (coro->refcnt--)
787 733
788 if (coro->mainstack && coro->mainstack != main_mainstack) 734 if (coro->mainstack && coro->mainstack != main_mainstack)
789 { 735 {
790 struct coro temp; 736 struct coro temp;
791 737
792 SAVE (aTHX_ (&temp), TRANSFER_SAVE_ALL); 738 SAVE ((&temp), TRANSFER_SAVE_ALL);
793 LOAD (aTHX_ coro); 739 LOAD (coro);
794 740
795 destroy_stacks (aTHX); 741 coro_destroy_stacks ();
796 742
797 LOAD ((&temp)); /* this will get rid of defsv etc.. */ 743 LOAD ((&temp)); /* this will get rid of defsv etc.. */
798 744
799 coro->mainstack = 0; 745 coro->mainstack = 0;
800 } 746 }
801 747
802 deallocate_stack (coro); 748 cctx_free (coro->cctx);
803 SvREFCNT_dec (coro->args); 749 SvREFCNT_dec (coro->args);
804 Safefree (coro); 750 Safefree (coro);
805} 751}
806 752
807static int 753static int
808coro_state_clear (SV *sv, MAGIC *mg) 754coro_state_clear (pTHX_ SV *sv, MAGIC *mg)
809{ 755{
810 struct coro *coro = (struct coro *)mg->mg_ptr; 756 struct coro *coro = (struct coro *)mg->mg_ptr;
811 mg->mg_ptr = 0; 757 mg->mg_ptr = 0;
812 758
813 coro_state_destroy (coro); 759 coro_state_destroy (coro);
814 760
815 return 0; 761 return 0;
816} 762}
817 763
818static int 764static int
819coro_state_dup (MAGIC *mg, CLONE_PARAMS *params) 765coro_state_dup (pTHX_ MAGIC *mg, CLONE_PARAMS *params)
820{ 766{
821 struct coro *coro = (struct coro *)mg->mg_ptr; 767 struct coro *coro = (struct coro *)mg->mg_ptr;
822 768
823 ++coro->refcnt; 769 ++coro->refcnt;
824 770
825 return 0; 771 return 0;
826} 772}
827 773
828static MGVTBL coro_state_vtbl = { 0, 0, 0, 0, coro_state_clear, 0, coro_state_dup, 0 }; 774static MGVTBL coro_state_vtbl = {
775 0, 0, 0, 0,
776 coro_state_clear,
777 0,
778#ifdef MGf_DUP
779 coro_state_dup,
780#else
781# define MGf_DUP 0
782#endif
783};
829 784
830static struct coro * 785static struct coro *
831SvSTATE (SV *coro) 786SvSTATE (SV *coro)
832{ 787{
833 HV *stash; 788 HV *stash;
848 assert (mg->mg_type == PERL_MAGIC_ext); 803 assert (mg->mg_type == PERL_MAGIC_ext);
849 return (struct coro *)mg->mg_ptr; 804 return (struct coro *)mg->mg_ptr;
850} 805}
851 806
852static void 807static void
808prepare_transfer (struct transfer_args *ta, SV *prev, SV *next, int flags)
809{
810 ta->prev = SvSTATE (prev);
811 ta->next = SvSTATE (next);
812 ta->flags = flags;
813}
814
815static void
853api_transfer (pTHX_ SV *prev, SV *next, int flags) 816api_transfer (SV *prev, SV *next, int flags)
854{ 817{
855 coro_state_transfer (aTHX_ SvSTATE (prev), SvSTATE (next), flags); 818 dTHX;
819 struct transfer_args ta;
820
821 prepare_transfer (&ta, prev, next, flags);
822 TRANSFER (ta);
856} 823}
857 824
858/** Coro ********************************************************************/ 825/** Coro ********************************************************************/
859 826
860#define PRIO_MAX 3 827#define PRIO_MAX 3
868static GV *coro_current, *coro_idle; 835static GV *coro_current, *coro_idle;
869static AV *coro_ready [PRIO_MAX-PRIO_MIN+1]; 836static AV *coro_ready [PRIO_MAX-PRIO_MIN+1];
870static int coro_nready; 837static int coro_nready;
871 838
872static void 839static void
873coro_enq (pTHX_ SV *sv) 840coro_enq (SV *sv)
874{ 841{
875 int prio; 842 int prio;
876 843
877 if (SvTYPE (sv) != SVt_PVHV) 844 if (SvTYPE (sv) != SVt_PVHV)
878 croak ("Coro::ready tried to enqueue something that is not a coroutine"); 845 croak ("Coro::ready tried to enqueue something that is not a coroutine");
882 av_push (coro_ready [prio - PRIO_MIN], sv); 849 av_push (coro_ready [prio - PRIO_MIN], sv);
883 coro_nready++; 850 coro_nready++;
884} 851}
885 852
886static SV * 853static SV *
887coro_deq (pTHX_ int min_prio) 854coro_deq (int min_prio)
888{ 855{
889 int prio = PRIO_MAX - PRIO_MIN; 856 int prio = PRIO_MAX - PRIO_MIN;
890 857
891 min_prio -= PRIO_MIN; 858 min_prio -= PRIO_MIN;
892 if (min_prio < 0) 859 if (min_prio < 0)
909 876
910 if (SvROK (coro)) 877 if (SvROK (coro))
911 coro = SvRV (coro); 878 coro = SvRV (coro);
912 879
913 LOCK; 880 LOCK;
914 coro_enq (aTHX_ SvREFCNT_inc (coro)); 881 coro_enq (SvREFCNT_inc (coro));
915 UNLOCK; 882 UNLOCK;
916} 883}
917 884
918static void 885static void
919api_schedule (void) 886prepare_schedule (struct transfer_args *ta)
920{ 887{
921 dTHX;
922
923 SV *prev, *next; 888 SV *current, *prev, *next;
889
924 SV *current = GvSV (coro_current); 890 current = GvSV (coro_current);
925 891
926 for (;;) 892 for (;;)
927 { 893 {
928 LOCK; 894 LOCK;
929
930 next = coro_deq (aTHX_ PRIO_MIN); 895 next = coro_deq (PRIO_MIN);
896 UNLOCK;
931 897
932 if (next) 898 if (next)
933 break; 899 break;
934
935 UNLOCK;
936 900
937 { 901 {
938 dSP; 902 dSP;
939 903
940 ENTER; 904 ENTER;
951 915
952 prev = SvRV (current); 916 prev = SvRV (current);
953 SvRV (current) = next; 917 SvRV (current) = next;
954 918
955 /* free this only after the transfer */ 919 /* free this only after the transfer */
920 LOCK;
921 free_coro_mortal ();
922 UNLOCK;
956 coro_mortal = prev; 923 coro_mortal = prev;
957 924
925 ta->prev = SvSTATE (prev);
926 ta->next = SvSTATE (next);
927 ta->flags = TRANSFER_SAVE_ALL;
928}
929
930static void
931prepare_cede (struct transfer_args *ta)
932{
933 LOCK;
934 coro_enq (SvREFCNT_inc (SvRV (GvSV (coro_current))));
958 UNLOCK; 935 UNLOCK;
959 936
960 coro_state_transfer (aTHX_ SvSTATE (prev), SvSTATE (next), 937 prepare_schedule (ta);
961 TRANSFER_SAVE_ALL | TRANSFER_LAZY_STACK); 938}
939
940static void
941api_schedule (void)
942{
943 dTHX;
944 struct transfer_args ta;
945
946 prepare_schedule (&ta);
947 TRANSFER (ta);
962} 948}
963 949
964static void 950static void
965api_cede (void) 951api_cede (void)
966{ 952{
967 dTHX; 953 dTHX;
954 struct transfer_args ta;
968 955
969 LOCK; 956 prepare_cede (&ta);
970 coro_enq (aTHX_ SvREFCNT_inc (SvRV (GvSV (coro_current)))); 957 TRANSFER (ta);
971 UNLOCK;
972
973 api_schedule ();
974} 958}
975 959
976MODULE = Coro::State PACKAGE = Coro::State 960MODULE = Coro::State PACKAGE = Coro::State
977 961
978PROTOTYPES: DISABLE 962PROTOTYPES: DISABLE
987 coro_state_stash = gv_stashpv ("Coro::State", TRUE); 971 coro_state_stash = gv_stashpv ("Coro::State", TRUE);
988 972
989 newCONSTSUB (coro_state_stash, "SAVE_DEFAV", newSViv (TRANSFER_SAVE_DEFAV)); 973 newCONSTSUB (coro_state_stash, "SAVE_DEFAV", newSViv (TRANSFER_SAVE_DEFAV));
990 newCONSTSUB (coro_state_stash, "SAVE_DEFSV", newSViv (TRANSFER_SAVE_DEFSV)); 974 newCONSTSUB (coro_state_stash, "SAVE_DEFSV", newSViv (TRANSFER_SAVE_DEFSV));
991 newCONSTSUB (coro_state_stash, "SAVE_ERRSV", newSViv (TRANSFER_SAVE_ERRSV)); 975 newCONSTSUB (coro_state_stash, "SAVE_ERRSV", newSViv (TRANSFER_SAVE_ERRSV));
992 newCONSTSUB (coro_state_stash, "SAVE_CCTXT", newSViv (TRANSFER_SAVE_CCTXT));
993 976
994 main_mainstack = PL_mainstack; 977 main_mainstack = PL_mainstack;
995 978
996 coroapi.ver = CORO_API_VERSION; 979 coroapi.ver = CORO_API_VERSION;
997 coroapi.transfer = api_transfer; 980 coroapi.transfer = api_transfer;
1008 int i; 991 int i;
1009 992
1010 Newz (0, coro, 1, struct coro); 993 Newz (0, coro, 1, struct coro);
1011 coro->args = newAV (); 994 coro->args = newAV ();
1012 995
1013 for (i = 1; i < items; i++)
1014 av_push (coro->args, newSVsv (ST (i)));
1015
1016 /*coro->mainstack = 0; *//*actual work is done inside transfer */
1017 /*coro->stack = 0;*/
1018
1019 hv = newHV (); 996 hv = newHV ();
1020 sv_magicext ((SV *)hv, 0, PERL_MAGIC_ext, &coro_state_vtbl, (char *)coro, 0)->mg_flags |= MGf_DUP; 997 sv_magicext ((SV *)hv, 0, PERL_MAGIC_ext, &coro_state_vtbl, (char *)coro, 0)->mg_flags |= MGf_DUP;
1021 RETVAL = sv_bless (newRV_noinc ((SV *)hv), gv_stashpv (klass, 1)); 998 RETVAL = sv_bless (newRV_noinc ((SV *)hv), gv_stashpv (klass, 1));
999
1000 for (i = 1; i < items; i++)
1001 av_push (coro->args, newSVsv (ST (i)));
1022} 1002}
1023 OUTPUT: 1003 OUTPUT:
1024 RETVAL 1004 RETVAL
1025 1005
1026void 1006void
1027transfer(prev, next, flags) 1007_set_stacklevel (...)
1028 SV *prev 1008 ALIAS:
1029 SV *next 1009 Coro::State::transfer = 1
1030 int flags 1010 Coro::schedule = 2
1011 Coro::cede = 3
1012 Coro::Cont::yield = 4
1031 CODE: 1013 CODE:
1032 PUTBACK;
1033 coro_state_transfer (aTHX_ SvSTATE (prev), SvSTATE (next), flags);
1034 SPAGAIN;
1035
1036int
1037prio (Coro::State coro, int newprio = 0)
1038 ALIAS:
1039 nice = 1
1040 CODE:
1041{ 1014{
1042 RETVAL = coro->prio; 1015 struct transfer_args ta;
1043 1016
1044 if (items > 1) 1017 switch (ix)
1045 { 1018 {
1019 case 0:
1020 ta.prev = (struct coro *)INT2PTR (coro_cctx *, SvIV (ST (0)));
1021 ta.next = 0;
1022 ta.flags = TRANSFER_SET_STACKLEVEL;
1023 break;
1024
1025 case 1:
1046 if (ix) 1026 if (items != 3)
1047 newprio += coro->prio; 1027 croak ("Coro::State::transfer(prev,next,flags) expects three arguments, not %d", items);
1048 1028
1049 if (newprio < PRIO_MIN) newprio = PRIO_MIN; 1029 prepare_transfer (&ta, ST (0), ST (1), SvIV (ST (2)));
1050 if (newprio > PRIO_MAX) newprio = PRIO_MAX; 1030 break;
1051 1031
1052 coro->prio = newprio; 1032 case 2:
1033 prepare_schedule (&ta);
1034 break;
1035
1036 case 3:
1037 prepare_cede (&ta);
1038 break;
1039
1040 case 4:
1041 {
1042 SV *yieldstack;
1043 SV *sv;
1044 AV *defav = GvAV (PL_defgv);
1045
1046 yieldstack = *hv_fetch (
1047 (HV *)SvRV (GvSV (coro_current)),
1048 "yieldstack", sizeof ("yieldstack") - 1,
1049 0
1050 );
1051
1052 /* set up @_ -- ugly */
1053 av_clear (defav);
1054 av_fill (defav, items - 1);
1055 while (items--)
1056 av_store (defav, items, SvREFCNT_inc (ST(items)));
1057
1058 sv = av_pop ((AV *)SvRV (yieldstack));
1059 ta.prev = SvSTATE (*av_fetch ((AV *)SvRV (sv), 0, 0));
1060 ta.next = SvSTATE (*av_fetch ((AV *)SvRV (sv), 1, 0));
1061 ta.flags = 0;
1062 SvREFCNT_dec (sv);
1063 }
1064 break;
1065
1053 } 1066 }
1067
1068 TRANSFER (ta);
1054} 1069}
1055 1070
1056void 1071void
1057_clone_state_from (SV *dst, SV *src) 1072_clone_state_from (SV *dst, SV *src)
1058 CODE: 1073 CODE:
1070 int code 1085 int code
1071 PROTOTYPE: $ 1086 PROTOTYPE: $
1072 CODE: 1087 CODE:
1073 _exit (code); 1088 _exit (code);
1074 1089
1075MODULE = Coro::State PACKAGE = Coro::Cont 1090int
1091cctx_count ()
1092 CODE:
1093 RETVAL = cctx_count;
1094 OUTPUT:
1095 RETVAL
1076 1096
1077void 1097int
1078yield (...) 1098cctx_idle ()
1079 PROTOTYPE: @ 1099 CODE:
1080 CODE: 1100 RETVAL = cctx_idle;
1081{ 1101 OUTPUT:
1082 SV *yieldstack; 1102 RETVAL
1083 SV *sv;
1084 AV *defav = GvAV (PL_defgv);
1085 struct coro *prev, *next;
1086
1087 yieldstack = *hv_fetch (
1088 (HV *)SvRV (GvSV (coro_current)),
1089 "yieldstack", sizeof ("yieldstack") - 1,
1090 0
1091 );
1092
1093 /* set up @_ -- ugly */
1094 av_clear (defav);
1095 av_fill (defav, items - 1);
1096 while (items--)
1097 av_store (defav, items, SvREFCNT_inc (ST(items)));
1098
1099 sv = av_pop ((AV *)SvRV (yieldstack));
1100 prev = SvSTATE (*av_fetch ((AV *)SvRV (sv), 0, 0));
1101 next = SvSTATE (*av_fetch ((AV *)SvRV (sv), 1, 0));
1102 SvREFCNT_dec (sv);
1103
1104 coro_state_transfer (aTHX_ prev, next, 0);
1105}
1106 1103
1107MODULE = Coro::State PACKAGE = Coro 1104MODULE = Coro::State PACKAGE = Coro
1108 1105
1109BOOT: 1106BOOT:
1110{ 1107{
1138 sv_setiv (sv, (IV)&coroapi); 1135 sv_setiv (sv, (IV)&coroapi);
1139 SvREADONLY_on (sv); 1136 SvREADONLY_on (sv);
1140 } 1137 }
1141} 1138}
1142 1139
1143#if !PERL_MICRO 1140int
1141prio (Coro::State coro, int newprio = 0)
1142 ALIAS:
1143 nice = 1
1144 CODE:
1145{
1146 RETVAL = coro->prio;
1147
1148 if (items > 1)
1149 {
1150 if (ix)
1151 newprio += coro->prio;
1152
1153 if (newprio < PRIO_MIN) newprio = PRIO_MIN;
1154 if (newprio > PRIO_MAX) newprio = PRIO_MAX;
1155
1156 coro->prio = newprio;
1157 }
1158}
1144 1159
1145void 1160void
1146ready (self) 1161ready (SV *self)
1147 SV * self
1148 PROTOTYPE: $ 1162 PROTOTYPE: $
1149 CODE: 1163 CODE:
1150 api_ready (self); 1164 api_ready (self);
1151
1152#endif
1153 1165
1154int 1166int
1155nready (...) 1167nready (...)
1156 PROTOTYPE: 1168 PROTOTYPE:
1157 CODE: 1169 CODE:
1158 RETVAL = coro_nready; 1170 RETVAL = coro_nready;
1159 OUTPUT: 1171 OUTPUT:
1160 RETVAL 1172 RETVAL
1161 1173
1162void 1174MODULE = Coro::State PACKAGE = Coro::AIO
1163schedule (...)
1164 PROTOTYPE:
1165 CODE:
1166 api_schedule ();
1167 1175
1168void
1169cede (...)
1170 PROTOTYPE:
1171 CODE:
1172 api_cede ();
1173
1174# and these are hacks
1175SV * 1176SV *
1176_aio_get_state () 1177_get_state ()
1177 CODE: 1178 CODE:
1178{ 1179{
1179 struct { 1180 struct {
1180 int errorno; 1181 int errorno;
1181 int laststype; 1182 int laststype;
1192} 1193}
1193 OUTPUT: 1194 OUTPUT:
1194 RETVAL 1195 RETVAL
1195 1196
1196void 1197void
1197_aio_set_state (char *data_) 1198_set_state (char *data_)
1198 PROTOTYPE: $ 1199 PROTOTYPE: $
1199 CODE: 1200 CODE:
1200{ 1201{
1201 struct { 1202 struct {
1202 int errorno; 1203 int errorno;

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines