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.84 by root, Fri Nov 24 00:31:21 2006 UTC vs.
Revision 1.106 by root, Mon Nov 27 02:08:55 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#define UCORO_STATE "_coro_state"
64 73
65/* 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
66 * 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
67 * and should be unique. */ 76 * and should be unique. */
77#define dSTACKLEVEL int stacklevel
68#define dSTACKLEVEL void *stacklevel = &stacklevel 78#define STACKLEVEL ((void *)&stacklevel)
69 79
70#define IN_DESTRUCT (PL_main_cv == Nullcv) 80#define IN_DESTRUCT (PL_main_cv == Nullcv)
71 81
72#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))
73 89
74#include "CoroAPI.h" 90#include "CoroAPI.h"
91
92#define TRANSFER_SET_STACKLEVEL 0x8bfbfbfb /* magic cookie */
75 93
76#ifdef USE_ITHREADS 94#ifdef USE_ITHREADS
77static perl_mutex coro_mutex; 95static perl_mutex coro_mutex;
78# define LOCK do { MUTEX_LOCK (&coro_mutex); } while (0) 96# define LOCK do { MUTEX_LOCK (&coro_mutex); } while (0)
79# define UNLOCK do { MUTEX_UNLOCK (&coro_mutex); } while (0) 97# define UNLOCK do { MUTEX_UNLOCK (&coro_mutex); } while (0)
82# define UNLOCK (void)0 100# define UNLOCK (void)0
83#endif 101#endif
84 102
85static struct CoroAPI coroapi; 103static struct CoroAPI coroapi;
86static AV *main_mainstack; /* used to differentiate between $main and others */ 104static AV *main_mainstack; /* used to differentiate between $main and others */
87static HV *coro_state_stash; 105static HV *coro_state_stash, *coro_stash;
88static SV *ucoro_state_sv;
89static U32 ucoro_state_hash;
90static SV *coro_mortal; /* will be freed after next transfer */ 106static SV *coro_mortal; /* will be freed after next transfer */
91 107
92/* this is actually not only the c stack but also c registers etc... */ 108/* this is a structure representing a c-level coroutine */
93typedef struct { 109typedef struct coro_cctx {
94 int refcnt; /* pointer reference counter */ 110 struct coro_cctx *next;
95 int usecnt; /* shared by how many coroutines */
96 int gencnt; /* generation counter */
97 111
98 coro_context cctx; 112 /* the stack */
99
100 void *sptr; 113 void *sptr;
101 long ssize; /* positive == mmap, otherwise malloc */ 114 long ssize; /* positive == mmap, otherwise malloc */
102} coro_stack;
103 115
116 /* cpu state */
117 void *idle_sp; /* sp of top-level transfer/schedule/cede call */
118 JMPENV *top_env;
119 coro_context cctx;
120
121#if USE_VALGRIND
122 int valgrind_id;
123#endif
124} coro_cctx;
125
126/* this is a structure representing a perl-level coroutine */
104struct coro { 127struct coro {
105 /* the optional C context */ 128 /* the c coroutine allocated to this perl coroutine, if any */
106 coro_stack *stack; 129 coro_cctx *cctx;
107 void *cursp; 130
131 /* data associated with this coroutine (initial args) */
132 AV *args;
108 int gencnt; 133 int refcnt;
109 134
110 /* optionally saved, might be zero */ 135 /* optionally saved, might be zero */
111 AV *defav; 136 AV *defav;
112 SV *defsv; 137 SV *defsv;
113 SV *errsv; 138 SV *errsv;
143 OP **retstack; 168 OP **retstack;
144 I32 retstack_ix; 169 I32 retstack_ix;
145 I32 retstack_max; 170 I32 retstack_max;
146 PMOP *curpm; 171 PMOP *curpm;
147 COP *curcop; 172 COP *curcop;
148 JMPENV *top_env;
149 173
150 /* data associated with this coroutine (initial args) */ 174 /* coro process data */
151 AV *args; 175 int prio;
152}; 176};
153 177
154typedef struct coro *Coro__State; 178typedef struct coro *Coro__State;
155typedef struct coro *Coro__State_or_hashref; 179typedef struct coro *Coro__State_or_hashref;
156 180
157static AV * 181static AV *
158coro_clone_padlist (pTHX_ CV *cv) 182coro_clone_padlist (CV *cv)
159{ 183{
160 AV *padlist = CvPADLIST (cv); 184 AV *padlist = CvPADLIST (cv);
161 AV *newpadlist, *newpad; 185 AV *newpadlist, *newpad;
162 186
163 newpadlist = newAV (); 187 newpadlist = newAV ();
164 AvREAL_off (newpadlist); 188 AvREAL_off (newpadlist);
165#if PERL_VERSION < 9 189#if PERL_VERSION_ATLEAST (5,9,0)
190 Perl_pad_push (aTHX_ padlist, AvFILLp (padlist) + 1);
191#else
166 Perl_pad_push (aTHX_ padlist, AvFILLp (padlist) + 1, 1); 192 Perl_pad_push (aTHX_ padlist, AvFILLp (padlist) + 1, 1);
167#else
168 Perl_pad_push (aTHX_ padlist, AvFILLp (padlist) + 1);
169#endif 193#endif
170 newpad = (AV *)AvARRAY (padlist)[AvFILLp (padlist)]; 194 newpad = (AV *)AvARRAY (padlist)[AvFILLp (padlist)];
171 --AvFILLp (padlist); 195 --AvFILLp (padlist);
172 196
173 av_store (newpadlist, 0, SvREFCNT_inc (*av_fetch (padlist, 0, FALSE))); 197 av_store (newpadlist, 0, SvREFCNT_inc (*av_fetch (padlist, 0, FALSE)));
175 199
176 return newpadlist; 200 return newpadlist;
177} 201}
178 202
179static void 203static void
180free_padlist (pTHX_ AV *padlist) 204free_padlist (AV *padlist)
181{ 205{
182 /* may be during global destruction */ 206 /* may be during global destruction */
183 if (SvREFCNT (padlist)) 207 if (SvREFCNT (padlist))
184 { 208 {
185 I32 i = AvFILLp (padlist); 209 I32 i = AvFILLp (padlist);
206 AV *padlist; 230 AV *padlist;
207 AV *av = (AV *)mg->mg_obj; 231 AV *av = (AV *)mg->mg_obj;
208 232
209 /* casting is fun. */ 233 /* casting is fun. */
210 while (&PL_sv_undef != (SV *)(padlist = (AV *)av_pop (av))) 234 while (&PL_sv_undef != (SV *)(padlist = (AV *)av_pop (av)))
211 free_padlist (aTHX_ padlist); 235 free_padlist (padlist);
212 236
213 SvREFCNT_dec (av); 237 SvREFCNT_dec (av);
214 238
215 return 0; 239 return 0;
216} 240}
219 243
220static MGVTBL vtbl_coro = {0, 0, 0, 0, coro_cv_free}; 244static MGVTBL vtbl_coro = {0, 0, 0, 0, coro_cv_free};
221 245
222/* the next two functions merely cache the padlists */ 246/* the next two functions merely cache the padlists */
223static void 247static void
224get_padlist (pTHX_ CV *cv) 248get_padlist (CV *cv)
225{ 249{
226 MAGIC *mg = mg_find ((SV *)cv, PERL_MAGIC_coro); 250 MAGIC *mg = mg_find ((SV *)cv, PERL_MAGIC_coro);
227 251
228 if (mg && AvFILLp ((AV *)mg->mg_obj) >= 0) 252 if (mg && AvFILLp ((AV *)mg->mg_obj) >= 0)
229 CvPADLIST (cv) = (AV *)av_pop ((AV *)mg->mg_obj); 253 CvPADLIST (cv) = (AV *)av_pop ((AV *)mg->mg_obj);
230 else 254 else
231 { 255 {
232#if 0 256#if 0
233 /* this should work - but it doesn't :( */ 257 /* this is probably cleaner, but also slower? */
234 CV *cp = Perl_cv_clone (aTHX_ cv); 258 CV *cp = Perl_cv_clone (cv);
235 CvPADLIST (cv) = CvPADLIST (cp); 259 CvPADLIST (cv) = CvPADLIST (cp);
236 CvPADLIST (cp) = 0; 260 CvPADLIST (cp) = 0;
237 SvREFCNT_dec (cp); 261 SvREFCNT_dec (cp);
238#else 262#else
239 CvPADLIST (cv) = coro_clone_padlist (aTHX_ cv); 263 CvPADLIST (cv) = coro_clone_padlist (cv);
240#endif 264#endif
241 } 265 }
242} 266}
243 267
244static void 268static void
245put_padlist (pTHX_ CV *cv) 269put_padlist (CV *cv)
246{ 270{
247 MAGIC *mg = mg_find ((SV *)cv, PERL_MAGIC_coro); 271 MAGIC *mg = mg_find ((SV *)cv, PERL_MAGIC_coro);
248 272
249 if (!mg) 273 if (!mg)
250 { 274 {
258} 282}
259 283
260#define SB do { 284#define SB do {
261#define SE } while (0) 285#define SE } while (0)
262 286
263#define LOAD(state) load_state(aTHX_ (state)); 287#define LOAD(state) load_state((state));
264#define SAVE(state,flags) save_state(aTHX_ (state),(flags)); 288#define SAVE(state,flags) save_state((state),(flags));
265 289
266#define REPLACE_SV(sv,val) SB SvREFCNT_dec(sv); (sv) = (val); (val) = 0; SE 290#define REPLACE_SV(sv,val) SB SvREFCNT_dec(sv); (sv) = (val); (val) = 0; SE
267 291
268static void 292static void
269load_state(pTHX_ Coro__State c) 293load_state(Coro__State c)
270{ 294{
271 PL_dowarn = c->dowarn; 295 PL_dowarn = c->dowarn;
272 PL_in_eval = c->in_eval; 296 PL_in_eval = c->in_eval;
273 297
274 PL_curstackinfo = c->curstackinfo; 298 PL_curstackinfo = c->curstackinfo;
292 PL_scopestack_ix = c->scopestack_ix; 316 PL_scopestack_ix = c->scopestack_ix;
293 PL_scopestack_max = c->scopestack_max; 317 PL_scopestack_max = c->scopestack_max;
294 PL_savestack = c->savestack; 318 PL_savestack = c->savestack;
295 PL_savestack_ix = c->savestack_ix; 319 PL_savestack_ix = c->savestack_ix;
296 PL_savestack_max = c->savestack_max; 320 PL_savestack_max = c->savestack_max;
297#if PERL_VERSION < 9 321#if !PERL_VERSION_ATLEAST (5,9,0)
298 PL_retstack = c->retstack; 322 PL_retstack = c->retstack;
299 PL_retstack_ix = c->retstack_ix; 323 PL_retstack_ix = c->retstack_ix;
300 PL_retstack_max = c->retstack_max; 324 PL_retstack_max = c->retstack_max;
301#endif 325#endif
302 PL_curpm = c->curpm; 326 PL_curpm = c->curpm;
303 PL_curcop = c->curcop; 327 PL_curcop = c->curcop;
304 PL_top_env = c->top_env;
305 328
306 if (c->defav) REPLACE_SV (GvAV (PL_defgv), c->defav); 329 if (c->defav) REPLACE_SV (GvAV (PL_defgv), c->defav);
307 if (c->defsv) REPLACE_SV (DEFSV , c->defsv); 330 if (c->defsv) REPLACE_SV (DEFSV , c->defsv);
308 if (c->errsv) REPLACE_SV (ERRSV , c->errsv); 331 if (c->errsv) REPLACE_SV (ERRSV , c->errsv);
309 332
316 { 339 {
317 AV *padlist = (AV *)POPs; 340 AV *padlist = (AV *)POPs;
318 341
319 if (padlist) 342 if (padlist)
320 { 343 {
321 put_padlist (aTHX_ cv); /* mark this padlist as available */ 344 put_padlist (cv); /* mark this padlist as available */
322 CvPADLIST(cv) = padlist; 345 CvPADLIST(cv) = padlist;
323 } 346 }
324 347
325 ++CvDEPTH(cv); 348 ++CvDEPTH(cv);
326 } 349 }
328 PUTBACK; 351 PUTBACK;
329 } 352 }
330} 353}
331 354
332static void 355static void
333save_state(pTHX_ Coro__State c, int flags) 356save_state(Coro__State c, int flags)
334{ 357{
335 { 358 {
336 dSP; 359 dSP;
337 I32 cxix = cxstack_ix; 360 I32 cxix = cxstack_ix;
338 PERL_CONTEXT *ccstk = cxstack; 361 PERL_CONTEXT *ccstk = cxstack;
366 } 389 }
367 390
368 PUSHs ((SV *)CvPADLIST(cv)); 391 PUSHs ((SV *)CvPADLIST(cv));
369 PUSHs ((SV *)cv); 392 PUSHs ((SV *)cv);
370 393
371 get_padlist (aTHX_ cv); 394 get_padlist (cv);
372 } 395 }
373 } 396 }
374#ifdef CXt_FORMAT 397#ifdef CXt_FORMAT
375 else if (CxTYPE(cx) == CXt_FORMAT) 398 else if (CxTYPE(cx) == CXt_FORMAT)
376 { 399 {
420 c->scopestack_ix = PL_scopestack_ix; 443 c->scopestack_ix = PL_scopestack_ix;
421 c->scopestack_max = PL_scopestack_max; 444 c->scopestack_max = PL_scopestack_max;
422 c->savestack = PL_savestack; 445 c->savestack = PL_savestack;
423 c->savestack_ix = PL_savestack_ix; 446 c->savestack_ix = PL_savestack_ix;
424 c->savestack_max = PL_savestack_max; 447 c->savestack_max = PL_savestack_max;
425#if PERL_VERSION < 9 448#if !PERL_VERSION_ATLEAST (5,9,0)
426 c->retstack = PL_retstack; 449 c->retstack = PL_retstack;
427 c->retstack_ix = PL_retstack_ix; 450 c->retstack_ix = PL_retstack_ix;
428 c->retstack_max = PL_retstack_max; 451 c->retstack_max = PL_retstack_max;
429#endif 452#endif
430 c->curpm = PL_curpm; 453 c->curpm = PL_curpm;
431 c->curcop = PL_curcop; 454 c->curcop = PL_curcop;
432 c->top_env = PL_top_env;
433} 455}
434 456
435/* 457/*
436 * allocate various perl stacks. This is an exact copy 458 * allocate various perl stacks. This is an exact copy
437 * of perl.c:init_stacks, except that it uses less memory 459 * of perl.c:init_stacks, except that it uses less memory
438 * on the (sometimes correct) assumption that coroutines do 460 * on the (sometimes correct) assumption that coroutines do
439 * not usually need a lot of stackspace. 461 * not usually need a lot of stackspace.
440 */ 462 */
441static void 463static void
442coro_init_stacks (pTHX) 464coro_init_stacks ()
443{ 465{
444 LOCK;
445
446 PL_curstackinfo = new_stackinfo(96, 1024/sizeof(PERL_CONTEXT) - 1); 466 PL_curstackinfo = new_stackinfo(96, 1024/sizeof(PERL_CONTEXT) - 1);
447 PL_curstackinfo->si_type = PERLSI_MAIN; 467 PL_curstackinfo->si_type = PERLSI_MAIN;
448 PL_curstack = PL_curstackinfo->si_stack; 468 PL_curstack = PL_curstackinfo->si_stack;
449 PL_mainstack = PL_curstack; /* remember in case we switch stacks */ 469 PL_mainstack = PL_curstack; /* remember in case we switch stacks */
450 470
471 491
472 New(54,PL_savestack,96,ANY); 492 New(54,PL_savestack,96,ANY);
473 PL_savestack_ix = 0; 493 PL_savestack_ix = 0;
474 PL_savestack_max = 96; 494 PL_savestack_max = 96;
475 495
476#if PERL_VERSION < 9 496#if !PERL_VERSION_ATLEAST (5,9,0)
477 New(54,PL_retstack,8,OP*); 497 New(54,PL_retstack,8,OP*);
478 PL_retstack_ix = 0; 498 PL_retstack_ix = 0;
479 PL_retstack_max = 8; 499 PL_retstack_max = 8;
480#endif 500#endif
481
482 UNLOCK;
483} 501}
484 502
485/* 503/*
486 * destroy the stacks, the callchain etc... 504 * destroy the stacks, the callchain etc...
487 */ 505 */
488static void 506static void
489destroy_stacks(pTHX) 507coro_destroy_stacks ()
490{ 508{
491 if (!IN_DESTRUCT) 509 if (!IN_DESTRUCT)
492 { 510 {
493 /* is this ugly, I ask? */ 511 /* is this ugly, I ask? */
494 LEAVE_SCOPE (0); 512 LEAVE_SCOPE (0);
525 543
526 Safefree (PL_tmps_stack); 544 Safefree (PL_tmps_stack);
527 Safefree (PL_markstack); 545 Safefree (PL_markstack);
528 Safefree (PL_scopestack); 546 Safefree (PL_scopestack);
529 Safefree (PL_savestack); 547 Safefree (PL_savestack);
530#if PERL_VERSION < 9 548#if !PERL_VERSION_ATLEAST (5,9,0)
531 Safefree (PL_retstack); 549 Safefree (PL_retstack);
532#endif 550#endif
533} 551}
534 552
535static void 553static void
536allocate_stack (Coro__State ctx, int alloc) 554setup_coro (struct coro *coro)
537{
538 coro_stack *stack;
539
540 New (0, stack, 1, coro_stack);
541
542 stack->refcnt = 1;
543 stack->usecnt = 1;
544 stack->gencnt = ctx->gencnt = 0;
545
546 if (alloc)
547 {
548#if HAVE_MMAP
549 stack->ssize = ((STACKSIZE * sizeof (long) + PAGESIZE - 1) / PAGESIZE + STACKGUARD) * PAGESIZE; /* mmap should do allocate-on-write for us */
550 stack->sptr = mmap (0, stack->ssize, PROT_EXEC|PROT_READ|PROT_WRITE, MAP_PRIVATE|MAP_ANONYMOUS, 0, 0);
551 if (stack->sptr != (void *)-1)
552 {
553# if STACKGUARD
554 mprotect (stack->sptr, STACKGUARD * PAGESIZE, PROT_NONE);
555# endif
556 }
557 else
558#endif
559 {
560 stack->ssize = - (STACKSIZE * (long)sizeof (long));
561 New (0, stack->sptr, STACKSIZE, long);
562 }
563 }
564 else
565 stack->sptr = 0;
566
567 ctx->stack = stack;
568}
569
570static void
571deallocate_stack (Coro__State ctx)
572{
573 coro_stack *stack = ctx->stack;
574
575 ctx->stack = 0;
576
577 if (stack)
578 {
579 if (!--stack->refcnt)
580 {
581#ifdef HAVE_MMAP
582 if (stack->ssize > 0 && stack->sptr)
583 munmap (stack->sptr, stack->ssize);
584 else
585#endif
586 Safefree (stack->sptr);
587
588 Safefree (stack);
589 }
590 else if (ctx->gencnt == stack->gencnt)
591 --stack->usecnt;
592 }
593}
594
595static void
596setup_coro (void *arg)
597{ 555{
598 /* 556 /*
599 * emulate part of the perl startup here. 557 * emulate part of the perl startup here.
600 */ 558 */
601 dTHX;
602 dSP;
603 Coro__State ctx = (Coro__State)arg;
604 SV *sub_init = (SV *)get_cv (SUB_INIT, FALSE);
605 559
606 coro_init_stacks (aTHX); 560 coro_init_stacks ();
561
607 /*PL_curcop = 0;*/ 562 PL_curcop = 0;
608 /*PL_in_eval = PL_in_eval;*/ /* inherit */ 563 PL_in_eval = 0;
564 PL_curpm = 0;
565
566 {
567 dSP;
568 LOGOP myop;
569
570 /* I have no idea why this is needed, but it is */
571 PUSHMARK (SP);
572
609 SvREFCNT_dec (GvAV (PL_defgv)); 573 SvREFCNT_dec (GvAV (PL_defgv));
610 GvAV (PL_defgv) = ctx->args; ctx->args = 0; 574 GvAV (PL_defgv) = coro->args; coro->args = 0;
611 575
612 SPAGAIN;
613
614 if (ctx->stack)
615 {
616 ctx->cursp = 0;
617
618 PUSHMARK(SP);
619 PUTBACK;
620 (void) call_sv (sub_init, G_VOID|G_NOARGS|G_EVAL);
621
622 if (SvTRUE (ERRSV))
623 croak (NULL);
624 else
625 croak ("FATAL: CCTXT coroutine returned!");
626 }
627 else
628 {
629 UNOP myop;
630
631 Zero(&myop, 1, UNOP); 576 Zero (&myop, 1, LOGOP);
632 myop.op_next = Nullop; 577 myop.op_next = Nullop;
633 myop.op_flags = OPf_WANT_VOID; 578 myop.op_flags = OPf_WANT_VOID;
634 579
635 PL_op = (OP *)&myop; 580 PL_op = (OP *)&myop;
636 581
637 PUSHMARK(SP); 582 PUSHMARK (SP);
638 XPUSHs (sub_init); 583 XPUSHs ((SV *)get_cv ("Coro::State::coro_init", FALSE));
639 /*
640 * the next line is slightly wrong, as PL_op->op_next
641 * is actually being executed so we skip the first op.
642 * that doesn't matter, though, since it is only
643 * pp_nextstate and we never return...
644 * ah yes, and I don't care anyways ;)
645 */
646 PUTBACK; 584 PUTBACK;
647 PL_op = PL_ppaddr[OP_ENTERSUB](aTHX); 585 PL_op = PL_ppaddr[OP_ENTERSUB](aTHX);
648 SPAGAIN; 586 SPAGAIN;
649 587
650 ENTER; /* necessary e.g. for dounwind */ 588 ENTER; /* necessary e.g. for dounwind */
651 } 589 }
652} 590}
653 591
654static void 592static void
655continue_coro (void *arg) 593free_coro_mortal ()
656{ 594{
657 /*
658 * this is a _very_ stripped down perl interpreter ;)
659 */
660 dTHX;
661 Coro__State ctx = (Coro__State)arg;
662
663 PL_top_env = &PL_start_env;
664
665 ctx->cursp = 0;
666 PL_op = PL_op->op_next;
667 CALLRUNOPS(aTHX);
668
669 abort ();
670}
671
672/* never call directly, always through the coro_state_transfer global variable */
673static void
674transfer_impl (pTHX_ struct coro *prev, struct coro *next, int flags)
675{
676 dSTACKLEVEL;
677
678 if (prev != next)
679 {
680 /* has this coro been created yet? */
681 if (next->mainstack)
682 {
683 LOCK;
684 SAVE (prev, flags);
685 LOAD (next);
686 UNLOCK;
687
688 /* mark this state as in-use */
689 next->mainstack = 0;
690 next->tmps_ix = -2;
691
692 /* stacklevel changed? if yes, grab the stack for us! */
693 if (flags & TRANSFER_SAVE_CCTXT)
694 {
695 if (!prev->stack)
696 allocate_stack (prev, 0);
697 else if (prev->cursp != stacklevel
698 && prev->stack->usecnt > 1)
699 {
700 prev->gencnt = ++prev->stack->gencnt;
701 prev->stack->usecnt = 1;
702 }
703
704 /* has our stack been invalidated? */
705 if (next->stack && next->stack->gencnt != next->gencnt)
706 {
707 deallocate_stack (next);
708 allocate_stack (next, 1);
709 coro_create (&(next->stack->cctx),
710 continue_coro, (void *)next,
711 next->stack->sptr, labs (next->stack->ssize));
712 }
713
714 coro_transfer (&(prev->stack->cctx), &(next->stack->cctx));
715 prev->cursp = stacklevel;
716 /* don't add any code here */
717 }
718 else
719 next->cursp = stacklevel;
720 }
721 else if (next->tmps_ix == -2)
722 croak ("tried to transfer to running coroutine");
723 else
724 {
725 LOCK;
726 SAVE (prev, -1); /* first get rid of the old state */
727 UNLOCK;
728
729 /* create the coroutine for the first time */
730 if (flags & TRANSFER_SAVE_CCTXT)
731 {
732 if (!prev->stack)
733 allocate_stack (prev, 0);
734
735 /* the new coroutine starts with start_env again */
736 PL_top_env = &PL_start_env;
737
738 if (prev->stack->sptr && flags & TRANSFER_LAZY_STACK)
739 {
740 setup_coro (next);
741 next->cursp = stacklevel;
742
743 prev->stack->refcnt++;
744 prev->stack->usecnt++;
745 next->stack = prev->stack;
746 next->gencnt = prev->gencnt;
747 }
748 else
749 {
750 assert (!next->stack);
751 allocate_stack (next, 1);
752 coro_create (&(next->stack->cctx),
753 setup_coro, (void *)next,
754 next->stack->sptr, labs (next->stack->ssize));
755 coro_transfer (&(prev->stack->cctx), &(next->stack->cctx));
756 prev->cursp = stacklevel;
757 /* don't add any code here */
758 }
759 }
760 else
761 {
762 setup_coro (next);
763 next->cursp = stacklevel;
764 }
765 }
766 }
767
768 LOCK;
769 if (coro_mortal) 595 if (coro_mortal)
770 { 596 {
771 SvREFCNT_dec (coro_mortal); 597 SvREFCNT_dec (coro_mortal);
772 coro_mortal = 0; 598 coro_mortal = 0;
773 } 599 }
600}
601
602static void NOINLINE
603prepare_cctx (coro_cctx *cctx)
604{
605 dSP;
606 LOGOP myop;
607
608 Zero (&myop, 1, LOGOP);
609 myop.op_next = PL_op;
610 myop.op_flags = OPf_WANT_VOID;
611
612 sv_setiv (get_sv ("Coro::State::cctx", FALSE), PTR2IV (cctx));
613
614 PUSHMARK (SP);
615 XPUSHs ((SV *)get_cv ("Coro::State::cctx_init", FALSE));
616 PUTBACK;
617 PL_restartop = PL_ppaddr[OP_ENTERSUB](aTHX);
618 SPAGAIN;
619}
620
621static void
622coro_run (void *arg)
623{
624 /* coro_run is the alternative epilogue of transfer() */
774 UNLOCK; 625 UNLOCK;
775}
776 626
777void (*coro_state_transfer)(pTHX_ struct coro *prev, struct coro *next, int flags) = transfer_impl; 627 /*
628 * this is a _very_ stripped down perl interpreter ;)
629 */
630 PL_top_env = &PL_start_env;
631 /* inject call to cctx_init */
632 prepare_cctx ((coro_cctx *)arg);
778 633
779#define SV_CORO(sv,func) \ 634 /* somebody will hit me for both perl_run and PL_restartop */
780 do { \ 635 perl_run (PL_curinterp);
781 if (SvROK (sv)) \
782 sv = SvRV (sv); \
783 \
784 if (SvTYPE (sv) == SVt_PVHV) \
785 { \
786 HE *he = hv_fetch_ent ((HV *)sv, ucoro_state_sv, 0, ucoro_state_hash); \
787 \
788 if (!he) \
789 croak ("%s() -- %s is a hashref but lacks the " UCORO_STATE " key", func, # sv); \
790 \
791 (sv) = SvRV (HeVAL(he)); \
792 } \
793 \
794 /* must also be changed inside Coro::Cont::yield */ \
795 if (!SvOBJECT (sv) || SvSTASH (sv) != coro_state_stash) \
796 croak ("%s() -- %s is not (and contains not) a Coro::State object", func, # sv); \
797 \
798 } while(0)
799 636
800#define SvSTATE(sv) INT2PTR (struct coro *, SvIVX (sv)) 637 fputs ("FATAL: C coroutine fell over the edge of the world, aborting. Did you call exit in a coroutine?\n", stderr);
638 abort ();
639}
801 640
641static coro_cctx *
642cctx_new ()
643{
644 coro_cctx *cctx;
645
646 New (0, cctx, 1, coro_cctx);
647
648#if HAVE_MMAP
649
650 cctx->ssize = ((STACKSIZE * sizeof (long) + PAGESIZE - 1) / PAGESIZE + STACKGUARD) * PAGESIZE;
651 /* mmap suppsedly does allocate-on-write for us */
652 cctx->sptr = mmap (0, cctx->ssize, PROT_EXEC|PROT_READ|PROT_WRITE, MAP_PRIVATE|MAP_ANONYMOUS, 0, 0);
653
654 if (cctx->sptr == (void *)-1)
655 {
656 perror ("FATAL: unable to mmap stack for coroutine");
657 _exit (EXIT_FAILURE);
658 }
659
660# if STACKGUARD
661 mprotect (cctx->sptr, STACKGUARD * PAGESIZE, PROT_NONE);
662# endif
663
664#else
665
666 cctx->ssize = STACKSIZE * (long)sizeof (long);
667 New (0, cctx->sptr, STACKSIZE, long);
668
669 if (!cctx->sptr)
670 {
671 perror ("FATAL: unable to malloc stack for coroutine");
672 _exit (EXIT_FAILURE);
673 }
674
675#endif
676
677#if USE_VALGRIND
678 cctx->valgrind_id = VALGRIND_STACK_REGISTER (
679 STACKGUARD * PAGESIZE + (char *)cctx->sptr,
680 cctx->ssize + (char *)cctx->sptr
681 );
682#endif
683
684 coro_create (&cctx->cctx, coro_run, (void *)cctx, cctx->sptr, cctx->ssize);
685
686 return cctx;
687}
688
689static void
690cctx_free (coro_cctx *cctx)
691{
692 if (!cctx)
693 return;
694
695#if USE_VALGRIND
696 VALGRIND_STACK_DEREGISTER (cctx->valgrind_id);
697#endif
698
699#if HAVE_MMAP
700 munmap (cctx->sptr, cctx->ssize);
701#else
702 Safefree (cctx->sptr);
703#endif
704
705 Safefree (cctx);
706}
707
708static coro_cctx *cctx_first;
709static int cctx_count, cctx_idle;
710
711static coro_cctx *
712cctx_get ()
713{
714 coro_cctx *cctx;
715
716 if (cctx_first)
717 {
718 --cctx_idle;
719 cctx = cctx_first;
720 cctx_first = cctx->next;
721 }
722 else
723 {
724 ++cctx_count;
725 cctx = cctx_new ();
726 PL_op = PL_op->op_next;
727 }
728
729 return cctx;
730}
731
732static void
733cctx_put (coro_cctx *cctx)
734{
735 ++cctx_idle;
736 cctx->next = cctx_first;
737 cctx_first = cctx;
738}
739
740/* never call directly, always through the coro_state_transfer global variable */
741static void NOINLINE
742transfer (struct coro *prev, struct coro *next, int flags)
743{
744 dSTACKLEVEL;
745
746 /* sometimes transfer is only called to set idle_sp */
747 if (flags == TRANSFER_SET_STACKLEVEL)
748 ((coro_cctx *)prev)->idle_sp = STACKLEVEL;
749 else if (prev != next)
750 {
751 coro_cctx *prev__cctx;
752
753 LOCK;
754
755 if (next->mainstack)
756 {
757 /* coroutine already started */
758 SAVE (prev, flags);
759 LOAD (next);
760 }
761 else
762 {
763 /* need to start coroutine */
764 /* first get rid of the old state */
765 SAVE (prev, -1);
766 /* setup coroutine call */
767 setup_coro (next);
768 /* need a stack */
769 next->cctx = 0;
770 }
771
772 if (!prev->cctx)
773 /* create a new empty context */
774 Newz (0, prev->cctx, 1, coro_cctx);
775
776 prev__cctx = prev->cctx;
777
778 /* possibly "free" the cctx */
779 if (prev__cctx->idle_sp == STACKLEVEL)
780 {
781 cctx_put (prev__cctx);
782 prev->cctx = 0;
783 }
784
785 if (!next->cctx)
786 next->cctx = cctx_get ();
787
788 if (prev__cctx != next->cctx)
789 {
790 prev__cctx->top_env = PL_top_env;
791 PL_top_env = next->cctx->top_env;
792 coro_transfer (&prev__cctx->cctx, &next->cctx->cctx);
793 }
794
795 free_coro_mortal ();
796
797 UNLOCK;
798 }
799}
800
801struct transfer_args
802{
803 struct coro *prev, *next;
804 int flags;
805};
806
807#define TRANSFER(ta) transfer ((ta).prev, (ta).next, (ta).flags)
808
809static void
810coro_state_destroy (struct coro *coro)
811{
812 if (coro->refcnt--)
813 return;
814
815 if (coro->mainstack && coro->mainstack != main_mainstack)
816 {
817 struct coro temp;
818
819 SAVE ((&temp), TRANSFER_SAVE_ALL);
820 LOAD (coro);
821
822 coro_destroy_stacks ();
823
824 LOAD ((&temp)); /* this will get rid of defsv etc.. */
825
826 coro->mainstack = 0;
827 }
828
829 cctx_free (coro->cctx);
830 SvREFCNT_dec (coro->args);
831 Safefree (coro);
832}
833
802static void 834static int
835coro_state_clear (pTHX_ SV *sv, MAGIC *mg)
836{
837 struct coro *coro = (struct coro *)mg->mg_ptr;
838 mg->mg_ptr = 0;
839
840 coro_state_destroy (coro);
841
842 return 0;
843}
844
845static int
846coro_state_dup (pTHX_ MAGIC *mg, CLONE_PARAMS *params)
847{
848 struct coro *coro = (struct coro *)mg->mg_ptr;
849
850 ++coro->refcnt;
851
852 return 0;
853}
854
855static MGVTBL coro_state_vtbl = {
856 0, 0, 0, 0,
857 coro_state_clear,
858 0,
859#ifdef MGf_DUP
860 coro_state_dup,
861#else
862# define MGf_DUP 0
863#endif
864};
865
866static struct coro *
867SvSTATE (SV *coro)
868{
869 HV *stash;
870 MAGIC *mg;
871
872 if (SvROK (coro))
873 coro = SvRV (coro);
874
875 stash = SvSTASH (coro);
876 if (stash != coro_stash && stash != coro_state_stash)
877 {
878 /* very slow, but rare, check */
879 if (!sv_derived_from (sv_2mortal (newRV_inc (coro)), "Coro::State"))
880 croak ("Coro::State object required");
881 }
882
883 mg = SvMAGIC (coro);
884 assert (mg->mg_type == PERL_MAGIC_ext);
885 return (struct coro *)mg->mg_ptr;
886}
887
888static void
889prepare_transfer (struct transfer_args *ta, SV *prev, SV *next, int flags)
890{
891 ta->prev = SvSTATE (prev);
892 ta->next = SvSTATE (next);
893 ta->flags = flags;
894}
895
896static void
803api_transfer (pTHX_ SV *prev, SV *next, int flags) 897api_transfer (SV *prev, SV *next, int flags)
804{ 898{
805 SV_CORO (prev, "Coro::transfer"); 899 dTHX;
806 SV_CORO (next, "Coro::transfer"); 900 struct transfer_args ta;
807 901
808 coro_state_transfer (aTHX_ SvSTATE (prev), SvSTATE (next), flags); 902 prepare_transfer (&ta, prev, next, flags);
903 TRANSFER (ta);
809} 904}
810 905
811/** Coro ********************************************************************/ 906/** Coro ********************************************************************/
812 907
813#define PRIO_MAX 3 908#define PRIO_MAX 3
821static GV *coro_current, *coro_idle; 916static GV *coro_current, *coro_idle;
822static AV *coro_ready [PRIO_MAX-PRIO_MIN+1]; 917static AV *coro_ready [PRIO_MAX-PRIO_MIN+1];
823static int coro_nready; 918static int coro_nready;
824 919
825static void 920static void
826coro_enq (pTHX_ SV *sv) 921coro_enq (SV *sv)
827{ 922{
828 SV **xprio;
829 int prio; 923 int prio;
830 924
831 if (SvTYPE (sv) != SVt_PVHV) 925 if (SvTYPE (sv) != SVt_PVHV)
832 croak ("Coro::ready tried to enqueue something that is not a coroutine"); 926 croak ("Coro::ready tried to enqueue something that is not a coroutine");
833 927
834 xprio = hv_fetch ((HV *)sv, "prio", 4, 0); 928 prio = SvSTATE (sv)->prio;
835 prio = xprio ? SvIV (*xprio) : PRIO_NORMAL;
836
837 prio = prio > PRIO_MAX ? PRIO_MAX
838 : prio < PRIO_MIN ? PRIO_MIN
839 : prio;
840 929
841 av_push (coro_ready [prio - PRIO_MIN], sv); 930 av_push (coro_ready [prio - PRIO_MIN], sv);
842 coro_nready++; 931 coro_nready++;
843} 932}
844 933
845static SV * 934static SV *
846coro_deq (pTHX_ int min_prio) 935coro_deq (int min_prio)
847{ 936{
848 int prio = PRIO_MAX - PRIO_MIN; 937 int prio = PRIO_MAX - PRIO_MIN;
849 938
850 min_prio -= PRIO_MIN; 939 min_prio -= PRIO_MIN;
851 if (min_prio < 0) 940 if (min_prio < 0)
868 957
869 if (SvROK (coro)) 958 if (SvROK (coro))
870 coro = SvRV (coro); 959 coro = SvRV (coro);
871 960
872 LOCK; 961 LOCK;
873 coro_enq (aTHX_ SvREFCNT_inc (coro)); 962 coro_enq (SvREFCNT_inc (coro));
874 UNLOCK; 963 UNLOCK;
875} 964}
876 965
877static void 966static void
967prepare_schedule (struct transfer_args *ta)
968{
969 SV *current, *prev, *next;
970
971 current = GvSV (coro_current);
972
973 for (;;)
974 {
975 LOCK;
976 next = coro_deq (PRIO_MIN);
977 UNLOCK;
978
979 if (next)
980 break;
981
982 {
983 dSP;
984
985 ENTER;
986 SAVETMPS;
987
988 PUSHMARK (SP);
989 PUTBACK;
990 call_sv (GvSV (coro_idle), G_DISCARD);
991
992 FREETMPS;
993 LEAVE;
994 }
995 }
996
997 prev = SvRV (current);
998 SvRV (current) = next;
999
1000 /* free this only after the transfer */
1001 LOCK;
1002 free_coro_mortal ();
1003 UNLOCK;
1004 coro_mortal = prev;
1005
1006 ta->prev = SvSTATE (prev);
1007 ta->next = SvSTATE (next);
1008 ta->flags = TRANSFER_SAVE_ALL;
1009}
1010
1011static void
1012prepare_cede (struct transfer_args *ta)
1013{
1014 LOCK;
1015 coro_enq (SvREFCNT_inc (SvRV (GvSV (coro_current))));
1016 UNLOCK;
1017
1018 prepare_schedule (ta);
1019}
1020
1021static void
878api_schedule (void) 1022api_schedule (void)
879{ 1023{
880 dTHX; 1024 dTHX;
1025 struct transfer_args ta;
881 1026
882 SV *prev, *next; 1027 prepare_schedule (&ta);
883 1028 TRANSFER (ta);
884 LOCK;
885
886 prev = SvRV (GvSV (coro_current));
887 next = coro_deq (aTHX_ PRIO_MIN);
888
889 if (!next)
890 next = SvREFCNT_inc (SvRV (GvSV (coro_idle)));
891
892 /* free this only after the transfer */
893 coro_mortal = prev;
894 SV_CORO (prev, "Coro::schedule");
895
896 SvRV (GvSV (coro_current)) = next;
897
898 SV_CORO (next, "Coro::schedule");
899
900 UNLOCK;
901
902 coro_state_transfer (aTHX_ SvSTATE (prev), SvSTATE (next),
903 TRANSFER_SAVE_ALL | TRANSFER_LAZY_STACK);
904} 1029}
905 1030
906static void 1031static void
907api_cede (void) 1032api_cede (void)
908{ 1033{
909 dTHX; 1034 dTHX;
1035 struct transfer_args ta;
910 1036
911 LOCK; 1037 prepare_cede (&ta);
912 coro_enq (aTHX_ SvREFCNT_inc (SvRV (GvSV (coro_current)))); 1038 TRANSFER (ta);
913 UNLOCK;
914
915 api_schedule ();
916} 1039}
917 1040
918MODULE = Coro::State PACKAGE = Coro::State 1041MODULE = Coro::State PACKAGE = Coro::State
919 1042
920PROTOTYPES: ENABLE 1043PROTOTYPES: DISABLE
921 1044
922BOOT: 1045BOOT:
923{ /* {} necessary for stoopid perl-5.6.x */ 1046{
924#ifdef USE_ITHREADS 1047#ifdef USE_ITHREADS
925 MUTEX_INIT (&coro_mutex); 1048 MUTEX_INIT (&coro_mutex);
926#endif 1049#endif
927 BOOT_PAGESIZE; 1050 BOOT_PAGESIZE;
928 1051
929 ucoro_state_sv = newSVpv (UCORO_STATE, sizeof(UCORO_STATE) - 1);
930 PERL_HASH(ucoro_state_hash, UCORO_STATE, sizeof(UCORO_STATE) - 1);
931 coro_state_stash = gv_stashpv ("Coro::State", TRUE); 1052 coro_state_stash = gv_stashpv ("Coro::State", TRUE);
932 1053
933 newCONSTSUB (coro_state_stash, "SAVE_DEFAV", newSViv (TRANSFER_SAVE_DEFAV)); 1054 newCONSTSUB (coro_state_stash, "SAVE_DEFAV", newSViv (TRANSFER_SAVE_DEFAV));
934 newCONSTSUB (coro_state_stash, "SAVE_DEFSV", newSViv (TRANSFER_SAVE_DEFSV)); 1055 newCONSTSUB (coro_state_stash, "SAVE_DEFSV", newSViv (TRANSFER_SAVE_DEFSV));
935 newCONSTSUB (coro_state_stash, "SAVE_ERRSV", newSViv (TRANSFER_SAVE_ERRSV)); 1056 newCONSTSUB (coro_state_stash, "SAVE_ERRSV", newSViv (TRANSFER_SAVE_ERRSV));
936 newCONSTSUB (coro_state_stash, "SAVE_CCTXT", newSViv (TRANSFER_SAVE_CCTXT));
937 1057
938 main_mainstack = PL_mainstack; 1058 main_mainstack = PL_mainstack;
939 1059
940 coroapi.ver = CORO_API_VERSION; 1060 coroapi.ver = CORO_API_VERSION;
941 coroapi.transfer = api_transfer; 1061 coroapi.transfer = api_transfer;
942}
943 1062
944Coro::State 1063 assert (("PRIO_NORMAL must be 0", !PRIO_NORMAL));
945_newprocess(args) 1064}
946 SV * args 1065
947 PROTOTYPE: $ 1066SV *
1067new (char *klass, ...)
948 CODE: 1068 CODE:
949 Coro__State coro; 1069{
1070 struct coro *coro;
1071 HV *hv;
1072 int i;
950 1073
951 if (!SvROK (args) || SvTYPE (SvRV (args)) != SVt_PVAV)
952 croak ("Coro::State::_newprocess expects an arrayref");
953
954 Newz (0, coro, 1, struct coro); 1074 Newz (0, coro, 1, struct coro);
1075 coro->args = newAV ();
955 1076
956 coro->args = (AV *)SvREFCNT_inc (SvRV (args)); 1077 hv = newHV ();
957 /*coro->mainstack = 0; *//*actual work is done inside transfer */ 1078 sv_magicext ((SV *)hv, 0, PERL_MAGIC_ext, &coro_state_vtbl, (char *)coro, 0)->mg_flags |= MGf_DUP;
958 /*coro->stack = 0;*/ 1079 RETVAL = sv_bless (newRV_noinc ((SV *)hv), gv_stashpv (klass, 1));
959 1080
960 RETVAL = coro; 1081 for (i = 1; i < items; i++)
1082 av_push (coro->args, newSVsv (ST (i)));
1083}
961 OUTPUT: 1084 OUTPUT:
962 RETVAL 1085 RETVAL
963 1086
964void 1087void
965transfer(prev, next, flags) 1088_set_stacklevel (...)
966 SV *prev 1089 ALIAS:
967 SV *next 1090 Coro::State::transfer = 1
968 int flags 1091 Coro::schedule = 2
969 PROTOTYPE: @ 1092 Coro::cede = 3
1093 Coro::Cont::yield = 4
970 CODE: 1094 CODE:
971 PUTBACK; 1095{
972 SV_CORO (next, "Coro::transfer"); 1096 struct transfer_args ta;
973 SV_CORO (prev, "Coro::transfer"); 1097
974 coro_state_transfer (aTHX_ SvSTATE (prev), SvSTATE (next), flags); 1098 switch (ix)
975 SPAGAIN; 1099 {
1100 case 0:
1101 ta.prev = (struct coro *)INT2PTR (coro_cctx *, SvIV (ST (0)));
1102 ta.next = 0;
1103 ta.flags = TRANSFER_SET_STACKLEVEL;
1104 break;
1105
1106 case 1:
1107 if (items != 3)
1108 croak ("Coro::State::transfer(prev,next,flags) expects three arguments, not %d", items);
1109
1110 prepare_transfer (&ta, ST (0), ST (1), SvIV (ST (2)));
1111 break;
1112
1113 case 2:
1114 prepare_schedule (&ta);
1115 break;
1116
1117 case 3:
1118 prepare_cede (&ta);
1119 break;
1120
1121 case 4:
1122 {
1123 SV *yieldstack;
1124 SV *sv;
1125 AV *defav = GvAV (PL_defgv);
1126
1127 yieldstack = *hv_fetch (
1128 (HV *)SvRV (GvSV (coro_current)),
1129 "yieldstack", sizeof ("yieldstack") - 1,
1130 0
1131 );
1132
1133 /* set up @_ -- ugly */
1134 av_clear (defav);
1135 av_fill (defav, items - 1);
1136 while (items--)
1137 av_store (defav, items, SvREFCNT_inc (ST(items)));
1138
1139 sv = av_pop ((AV *)SvRV (yieldstack));
1140 ta.prev = SvSTATE (*av_fetch ((AV *)SvRV (sv), 0, 0));
1141 ta.next = SvSTATE (*av_fetch ((AV *)SvRV (sv), 1, 0));
1142 ta.flags = 0;
1143 SvREFCNT_dec (sv);
1144 }
1145 break;
1146
1147 }
1148
1149 TRANSFER (ta);
1150}
976 1151
977void 1152void
978DESTROY(coro) 1153_clone_state_from (SV *dst, SV *src)
979 Coro::State coro 1154 CODE:
980 CODE: 1155{
1156 struct coro *coro_src = SvSTATE (src);
981 1157
982 if (coro->mainstack && coro->mainstack != main_mainstack) 1158 sv_unmagic (SvRV (dst), PERL_MAGIC_ext);
983 {
984 struct coro temp;
985 1159
986 PUTBACK; 1160 ++coro_src->refcnt;
987 SAVE (aTHX_ (&temp), TRANSFER_SAVE_ALL); 1161 sv_magicext (SvRV (dst), 0, PERL_MAGIC_ext, &coro_state_vtbl, (char *)coro_src, 0)->mg_flags |= MGf_DUP;
988 LOAD (aTHX_ coro); 1162}
989 SPAGAIN;
990
991 destroy_stacks (aTHX);
992
993 LOAD ((&temp)); /* this will get rid of defsv etc.. */
994 SPAGAIN;
995
996 coro->mainstack = 0;
997 }
998
999 deallocate_stack (coro);
1000 SvREFCNT_dec (coro->args);
1001 Safefree (coro);
1002 1163
1003void 1164void
1004_exit(code) 1165_exit (code)
1005 int code 1166 int code
1006 PROTOTYPE: $ 1167 PROTOTYPE: $
1007 CODE: 1168 CODE:
1008 _exit (code); 1169 _exit (code);
1009 1170
1010MODULE = Coro::State PACKAGE = Coro::Cont 1171int
1172cctx_count ()
1173 CODE:
1174 RETVAL = cctx_count;
1175 OUTPUT:
1176 RETVAL
1011 1177
1012void 1178int
1013yield(...) 1179cctx_idle ()
1014 PROTOTYPE: @ 1180 CODE:
1015 CODE: 1181 RETVAL = cctx_idle;
1016 SV *yieldstack; 1182 OUTPUT:
1017 SV *sv; 1183 RETVAL
1018 AV *defav = GvAV (PL_defgv);
1019 struct coro *prev, *next;
1020
1021 yieldstack = *hv_fetch (
1022 (HV *)SvRV (GvSV (coro_current)),
1023 "yieldstack", sizeof ("yieldstack") - 1,
1024 0
1025 );
1026
1027 /* set up @_ -- ugly */
1028 av_clear (defav);
1029 av_fill (defav, items - 1);
1030 while (items--)
1031 av_store (defav, items, SvREFCNT_inc (ST(items)));
1032
1033 sv = av_pop ((AV *)SvRV (yieldstack));
1034 prev = SvSTATE ((SV*)SvRV (*av_fetch ((AV *)SvRV (sv), 0, 0)));
1035 next = SvSTATE ((SV*)SvRV (*av_fetch ((AV *)SvRV (sv), 1, 0)));
1036 SvREFCNT_dec (sv);
1037
1038 coro_state_transfer (aTHX_ prev, next, 0);
1039 1184
1040MODULE = Coro::State PACKAGE = Coro 1185MODULE = Coro::State PACKAGE = Coro
1041 1186
1042BOOT: 1187BOOT:
1043{ 1188{
1044 int i; 1189 int i;
1190
1045 HV *stash = gv_stashpv ("Coro", TRUE); 1191 coro_stash = gv_stashpv ("Coro", TRUE);
1046 1192
1047 newCONSTSUB (stash, "PRIO_MAX", newSViv (PRIO_MAX)); 1193 newCONSTSUB (coro_stash, "PRIO_MAX", newSViv (PRIO_MAX));
1048 newCONSTSUB (stash, "PRIO_HIGH", newSViv (PRIO_HIGH)); 1194 newCONSTSUB (coro_stash, "PRIO_HIGH", newSViv (PRIO_HIGH));
1049 newCONSTSUB (stash, "PRIO_NORMAL", newSViv (PRIO_NORMAL)); 1195 newCONSTSUB (coro_stash, "PRIO_NORMAL", newSViv (PRIO_NORMAL));
1050 newCONSTSUB (stash, "PRIO_LOW", newSViv (PRIO_LOW)); 1196 newCONSTSUB (coro_stash, "PRIO_LOW", newSViv (PRIO_LOW));
1051 newCONSTSUB (stash, "PRIO_IDLE", newSViv (PRIO_IDLE)); 1197 newCONSTSUB (coro_stash, "PRIO_IDLE", newSViv (PRIO_IDLE));
1052 newCONSTSUB (stash, "PRIO_MIN", newSViv (PRIO_MIN)); 1198 newCONSTSUB (coro_stash, "PRIO_MIN", newSViv (PRIO_MIN));
1053 1199
1054 coro_current = gv_fetchpv ("Coro::current", TRUE, SVt_PV); 1200 coro_current = gv_fetchpv ("Coro::current", TRUE, SVt_PV);
1055 coro_idle = gv_fetchpv ("Coro::idle" , TRUE, SVt_PV); 1201 coro_idle = gv_fetchpv ("Coro::idle" , TRUE, SVt_PV);
1056 1202
1057 for (i = PRIO_MAX - PRIO_MIN + 1; i--; ) 1203 for (i = PRIO_MAX - PRIO_MIN + 1; i--; )
1070 sv_setiv (sv, (IV)&coroapi); 1216 sv_setiv (sv, (IV)&coroapi);
1071 SvREADONLY_on (sv); 1217 SvREADONLY_on (sv);
1072 } 1218 }
1073} 1219}
1074 1220
1075#if !PERL_MICRO 1221int
1222prio (Coro::State coro, int newprio = 0)
1223 ALIAS:
1224 nice = 1
1225 CODE:
1226{
1227 RETVAL = coro->prio;
1228
1229 if (items > 1)
1230 {
1231 if (ix)
1232 newprio += coro->prio;
1233
1234 if (newprio < PRIO_MIN) newprio = PRIO_MIN;
1235 if (newprio > PRIO_MAX) newprio = PRIO_MAX;
1236
1237 coro->prio = newprio;
1238 }
1239}
1076 1240
1077void 1241void
1078ready(self) 1242ready (SV *self)
1079 SV * self
1080 PROTOTYPE: $ 1243 PROTOTYPE: $
1081 CODE: 1244 CODE:
1082 api_ready (self); 1245 api_ready (self);
1083 1246
1084#endif
1085
1086int 1247int
1087nready(...) 1248nready (...)
1088 PROTOTYPE: 1249 PROTOTYPE:
1089 CODE: 1250 CODE:
1090 RETVAL = coro_nready; 1251 RETVAL = coro_nready;
1091 OUTPUT: 1252 OUTPUT:
1092 RETVAL 1253 RETVAL
1093 1254
1094void 1255MODULE = Coro::State PACKAGE = Coro::AIO
1095schedule(...)
1096 PROTOTYPE:
1097 CODE:
1098 api_schedule ();
1099 1256
1100void
1101cede(...)
1102 PROTOTYPE:
1103 CODE:
1104 api_cede ();
1105
1106# and these are hacks
1107SV * 1257SV *
1108_aio_get_state () 1258_get_state ()
1109 CODE: 1259 CODE:
1110{ 1260{
1111 struct { 1261 struct {
1112 int errorno; 1262 int errorno;
1113 int laststype; 1263 int laststype;
1124} 1274}
1125 OUTPUT: 1275 OUTPUT:
1126 RETVAL 1276 RETVAL
1127 1277
1128void 1278void
1129_aio_set_state (char *data_) 1279_set_state (char *data_)
1130 PROTOTYPE: $ 1280 PROTOTYPE: $
1131 CODE: 1281 CODE:
1132{ 1282{
1133 struct { 1283 struct {
1134 int errorno; 1284 int errorno;

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines