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.9 by root, Sat Jul 21 02:49:09 2001 UTC vs.
Revision 1.29 by root, Sat Aug 25 21:40:34 2001 UTC

1#include "EXTERN.h" 1#include "EXTERN.h"
2#include "perl.h" 2#include "perl.h"
3#include "XSUB.h" 3#include "XSUB.h"
4 4
5#if 1 5#include "libcoro/coro.c"
6# define CHK(x) (void *)0 6
7#include <signal.h>
8
9#ifdef HAVE_MMAP
10# include <unistd.h>
11# include <sys/mman.h>
12# ifndef MAP_ANON
13# ifdef MAP_ANONYMOUS
14# define MAP_ANON MAP_ANONYMOUS
7#else 15# else
8# define CHK(x) if (!(x)) croak("FATAL, CHK: " #x) 16# undef HAVE_MMAP
17# endif
18# endif
9#endif 19#endif
10 20
11#define MAY_FLUSH /* increases codesize */ 21#define MAY_FLUSH /* increases codesize and is rarely used */
12 22
13#define SUB_INIT "Coro::State::_newcoro" 23#define SUB_INIT "Coro::State::initialize"
24#define UCORO_STATE "_coro_state"
14 25
15#define SAVE_DEFAV 0x00000001 26/* The next macro should declare a variable stacklevel that contains and approximation
16#define SAVE_DEFSV 0x00000002 27 * to the current C stack pointer. Its property is that it changes with each call
17#define SAVE_ERRSV 0x00000004 28 * and should be unique. */
29#define dSTACKLEVEL void *stacklevel = &stacklevel
18 30
19#define SAVE_ALL -1 31#define labs(l) ((l) >= 0 ? (l) : -(l))
32
33#include "CoroAPI.h"
34
35static struct CoroAPI coroapi;
36
37/* this is actually not only the c stack but also c registers etc... */
38typedef struct {
39 int refcnt; /* pointer reference counter */
40 int usecnt; /* shared by how many coroutines */
41 int gencnt; /* generation counter */
42
43 coro_context cctx;
44
45 void *sptr;
46 long ssize; /* positive == mmap, otherwise malloc */
47} coro_stack;
20 48
21struct coro { 49struct coro {
50 /* the optional C context */
51 coro_stack *stack;
52 void *cursp;
53 int gencnt;
54
22 /* optionally saved, might be zero */ 55 /* optionally saved, might be zero */
23 AV *defav; 56 AV *defav;
24 SV *defsv; 57 SV *defsv;
25 SV *errsv; 58 SV *errsv;
26 59
27 /* saved global state not related to stacks */ 60 /* saved global state not related to stacks */
28 U8 dowarn; 61 U8 dowarn;
62 I32 in_eval;
29 63
30 /* the stacks and related info (callchain etc..) */ 64 /* the stacks and related info (callchain etc..) */
31 PERL_SI *curstackinfo; 65 PERL_SI *curstackinfo;
32 AV *curstack; 66 AV *curstack;
33 AV *mainstack; 67 AV *mainstack;
51 I32 savestack_max; 85 I32 savestack_max;
52 OP **retstack; 86 OP **retstack;
53 I32 retstack_ix; 87 I32 retstack_ix;
54 I32 retstack_max; 88 I32 retstack_max;
55 COP *curcop; 89 COP *curcop;
90 JMPENV *top_env;
56 91
57 /* data associated with this coroutine (initial args) */ 92 /* data associated with this coroutine (initial args) */
58 AV *args; 93 AV *args;
59}; 94};
60 95
61typedef struct coro *Coro__State; 96typedef struct coro *Coro__State;
62typedef struct coro *Coro__State_or_hashref; 97typedef struct coro *Coro__State_or_hashref;
63 98
99static AV *main_mainstack; /* used to differentiate between $main and others */
100static HV *coro_state_stash;
101static SV *ucoro_state_sv;
102static U32 ucoro_state_hash;
64static HV *padlist_cache; 103static HV *padlist_cache;
65 104
66/* mostly copied from op.c:cv_clone2 */ 105/* mostly copied from op.c:cv_clone2 */
67STATIC AV * 106STATIC AV *
68clone_padlist (AV *protopadlist) 107clone_padlist (AV *protopadlist)
156 195
157 return newpadlist; 196 return newpadlist;
158} 197}
159 198
160#ifdef MAY_FLUSH 199#ifdef MAY_FLUSH
161STATIC AV * 200STATIC void
162free_padlist (AV *padlist) 201free_padlist (AV *padlist)
163{ 202{
164 /* may be during global destruction */ 203 /* may be during global destruction */
165 if (SvREFCNT(padlist)) 204 if (SvREFCNT(padlist))
166 { 205 {
231#endif 270#endif
232 271
233#define SB do { 272#define SB do {
234#define SE } while (0) 273#define SE } while (0)
235 274
236#define LOAD(state) SB load_state(aTHX_ state); SPAGAIN; SE 275#define LOAD(state) load_state(aTHX_ (state));
237#define SAVE(state,flags) SB PUTBACK; save_state(aTHX_ state,flags); SE 276#define SAVE(state,flags) save_state(aTHX_ (state),(flags));
238 277
239#define REPLACE_SV(sv,val) SB SvREFCNT_dec(sv); (sv) = (val); SE 278#define REPLACE_SV(sv,val) SB SvREFCNT_dec(sv); (sv) = (val); SE
240 279
241static void 280static void
242load_state(pTHX_ Coro__State c) 281load_state(pTHX_ Coro__State c)
243{ 282{
244 PL_dowarn = c->dowarn; 283 PL_dowarn = c->dowarn;
284 PL_in_eval = c->in_eval;
245 285
246 PL_curstackinfo = c->curstackinfo; 286 PL_curstackinfo = c->curstackinfo;
247 PL_curstack = c->curstack; 287 PL_curstack = c->curstack;
248 PL_mainstack = c->mainstack; 288 PL_mainstack = c->mainstack;
249 PL_stack_sp = c->stack_sp; 289 PL_stack_sp = c->stack_sp;
266 PL_savestack_max = c->savestack_max; 306 PL_savestack_max = c->savestack_max;
267 PL_retstack = c->retstack; 307 PL_retstack = c->retstack;
268 PL_retstack_ix = c->retstack_ix; 308 PL_retstack_ix = c->retstack_ix;
269 PL_retstack_max = c->retstack_max; 309 PL_retstack_max = c->retstack_max;
270 PL_curcop = c->curcop; 310 PL_curcop = c->curcop;
311 PL_top_env = c->top_env;
271 312
272 if (c->defav) REPLACE_SV (GvAV (PL_defgv), c->defav); 313 if (c->defav) REPLACE_SV (GvAV (PL_defgv), c->defav);
273 if (c->defsv) REPLACE_SV (DEFSV , c->defsv); 314 if (c->defsv) REPLACE_SV (DEFSV , c->defsv);
274 if (c->errsv) REPLACE_SV (ERRSV , c->errsv); 315 if (c->errsv) REPLACE_SV (ERRSV , c->errsv);
275 316
302save_state(pTHX_ Coro__State c, int flags) 343save_state(pTHX_ Coro__State c, int flags)
303{ 344{
304 { 345 {
305 dSP; 346 dSP;
306 I32 cxix = cxstack_ix; 347 I32 cxix = cxstack_ix;
348 PERL_CONTEXT *ccstk = cxstack;
307 PERL_SI *top_si = PL_curstackinfo; 349 PERL_SI *top_si = PL_curstackinfo;
308 PERL_CONTEXT *ccstk = cxstack;
309 350
310 /* 351 /*
311 * the worst thing you can imagine happens first - we have to save 352 * the worst thing you can imagine happens first - we have to save
312 * (and reinitialize) all cv's in the whole callchain :( 353 * (and reinitialize) all cv's in the whole callchain :(
313 */ 354 */
314 355
315 PUSHs (Nullsv); 356 PUSHs (Nullsv);
316 /* this loop was inspired by pp_caller */ 357 /* this loop was inspired by pp_caller */
317 for (;;) 358 for (;;)
318 { 359 {
319 do 360 do
320 { 361 {
321 PERL_CONTEXT *cx = &ccstk[cxix--]; 362 PERL_CONTEXT *cx = &ccstk[cxix--];
322 363
323 if (CxTYPE(cx) == CXt_SUB) 364 if (CxTYPE(cx) == CXt_SUB)
324 { 365 {
363 } 404 }
364 405
365 PUTBACK; 406 PUTBACK;
366 } 407 }
367 408
368 c->defav = flags & SAVE_DEFAV ? (AV *)SvREFCNT_inc (GvAV (PL_defgv)) : 0; 409 c->defav = flags & TRANSFER_SAVE_DEFAV ? (AV *)SvREFCNT_inc (GvAV (PL_defgv)) : 0;
369 c->defsv = flags & SAVE_DEFSV ? SvREFCNT_inc (DEFSV) : 0; 410 c->defsv = flags & TRANSFER_SAVE_DEFSV ? SvREFCNT_inc (DEFSV) : 0;
370 c->errsv = flags & SAVE_ERRSV ? SvREFCNT_inc (ERRSV) : 0; 411 c->errsv = flags & TRANSFER_SAVE_ERRSV ? SvREFCNT_inc (ERRSV) : 0;
371 412
372 c->dowarn = PL_dowarn; 413 c->dowarn = PL_dowarn;
414 c->in_eval = PL_in_eval;
373 415
374 c->curstackinfo = PL_curstackinfo; 416 c->curstackinfo = PL_curstackinfo;
375 c->curstack = PL_curstack; 417 c->curstack = PL_curstack;
376 c->mainstack = PL_mainstack; 418 c->mainstack = PL_mainstack;
377 c->stack_sp = PL_stack_sp; 419 c->stack_sp = PL_stack_sp;
394 c->savestack_max = PL_savestack_max; 436 c->savestack_max = PL_savestack_max;
395 c->retstack = PL_retstack; 437 c->retstack = PL_retstack;
396 c->retstack_ix = PL_retstack_ix; 438 c->retstack_ix = PL_retstack_ix;
397 c->retstack_max = PL_retstack_max; 439 c->retstack_max = PL_retstack_max;
398 c->curcop = PL_curcop; 440 c->curcop = PL_curcop;
441 c->top_env = PL_top_env;
442}
443
444/*
445 * allocate various perl stacks. This is an exact copy
446 * of perl.c:init_stacks, except that it uses less memory
447 * on the assumption that coroutines do not usually need
448 * a lot of stackspace.
449 */
450STATIC void
451coro_init_stacks (pTHX)
452{
453 PL_curstackinfo = new_stackinfo(96, 1024/sizeof(PERL_CONTEXT) - 1);
454 PL_curstackinfo->si_type = PERLSI_MAIN;
455 PL_curstack = PL_curstackinfo->si_stack;
456 PL_mainstack = PL_curstack; /* remember in case we switch stacks */
457
458 PL_stack_base = AvARRAY(PL_curstack);
459 PL_stack_sp = PL_stack_base;
460 PL_stack_max = PL_stack_base + AvMAX(PL_curstack);
461
462 New(50,PL_tmps_stack,96,SV*);
463 PL_tmps_floor = -1;
464 PL_tmps_ix = -1;
465 PL_tmps_max = 96;
466
467 New(54,PL_markstack,16,I32);
468 PL_markstack_ptr = PL_markstack;
469 PL_markstack_max = PL_markstack + 16;
470
471 SET_MARK_OFFSET;
472
473 New(54,PL_scopestack,16,I32);
474 PL_scopestack_ix = 0;
475 PL_scopestack_max = 16;
476
477 New(54,PL_savestack,96,ANY);
478 PL_savestack_ix = 0;
479 PL_savestack_max = 96;
480
481 New(54,PL_retstack,8,OP*);
482 PL_retstack_ix = 0;
483 PL_retstack_max = 8;
399} 484}
400 485
401/* 486/*
402 * destroy the stacks, the callchain etc... 487 * destroy the stacks, the callchain etc...
403 * still there is a memleak of 128 bytes... 488 * still there is a memleak of 128 bytes...
424 dSP; 509 dSP;
425 SWITCHSTACK (PL_curstack, PL_curstackinfo->si_stack); 510 SWITCHSTACK (PL_curstack, PL_curstackinfo->si_stack);
426 PUTBACK; /* possibly superfluous */ 511 PUTBACK; /* possibly superfluous */
427 } 512 }
428 513
514 if (PL_main_cv != Nullcv) /* don't during destruction. hack? */
429 dounwind(-1); 515 dounwind(-1);
430 516
431 SvREFCNT_dec(PL_curstackinfo->si_stack); 517 SvREFCNT_dec(PL_curstackinfo->si_stack);
432 Safefree(PL_curstackinfo->si_cxstack); 518 Safefree(PL_curstackinfo->si_cxstack);
433 Safefree(PL_curstackinfo); 519 Safefree(PL_curstackinfo);
434 PL_curstackinfo = p; 520 PL_curstackinfo = p;
439 Safefree(PL_scopestack); 525 Safefree(PL_scopestack);
440 Safefree(PL_savestack); 526 Safefree(PL_savestack);
441 Safefree(PL_retstack); 527 Safefree(PL_retstack);
442} 528}
443 529
530static void
531allocate_stack (Coro__State ctx, int alloc)
532{
533 coro_stack *stack;
534
535 New (0, stack, 1, coro_stack);
536
537 stack->refcnt = 1;
538 stack->usecnt = 1;
539 stack->gencnt = ctx->gencnt = 0;
540 if (alloc)
541 {
542#ifdef HAVE_MMAP
543 stack->ssize = 128 * 1024 * sizeof (long); /* mmap should do allocate-on-write for us */
544 stack->sptr = mmap (0, stack->ssize, PROT_EXEC|PROT_READ|PROT_WRITE, MAP_PRIVATE|MAP_ANON, 0, 0);
545 if (stack->sptr == (void *)-1)
546#endif
547 {
548 /*FIXME*//*D*//* reasonable stack size! */
549 stack->ssize = -4096 * sizeof (long);
550 New (0, stack->sptr, 4096, long);
551 }
552 }
553 else
554 stack->sptr = 0;
555
556 ctx->stack = stack;
557}
558
559static void
560deallocate_stack (Coro__State ctx)
561{
562 coro_stack *stack = ctx->stack;
563
564 ctx->stack = 0;
565
566 if (stack)
567 {
568 if (!--stack->refcnt)
569 {
570#ifdef HAVE_MMAP
571 if (stack->ssize > 0 && stack->sptr)
572 munmap (stack->sptr, stack->ssize);
573 else
574#else
575 Safefree (stack->sptr);
576#endif
577 Safefree (stack);
578 }
579 else if (ctx->gencnt == stack->gencnt)
580 --stack->usecnt;
581 }
582}
583
584static void
585setup_coro (void *arg)
586{
587 /*
588 * emulate part of the perl startup here.
589 */
590 dSP;
591 Coro__State ctx = (Coro__State)arg;
592 SV *sub_init = (SV*)get_cv(SUB_INIT, FALSE);
593
594 coro_init_stacks (aTHX);
595 /*PL_curcop = 0;*/
596 /*PL_in_eval = PL_in_eval;*/ /* inherit */
597 SvREFCNT_dec (GvAV (PL_defgv));
598 GvAV (PL_defgv) = ctx->args;
599
600 SPAGAIN;
601
602 if (ctx->stack)
603 {
604 ctx->cursp = 0;
605
606 PUSHMARK(SP);
607 PUTBACK;
608 (void) call_sv (sub_init, G_VOID|G_NOARGS|G_EVAL);
609
610 if (SvTRUE (ERRSV))
611 croak (NULL);
612 else
613 croak ("FATAL: CCTXT coroutine returned!");
614 }
615 else
616 {
617 UNOP myop;
618
619 PL_op = (OP *)&myop;
620
621 Zero(&myop, 1, UNOP);
622 myop.op_next = Nullop;
623 myop.op_flags = OPf_WANT_VOID;
624
625 PUSHMARK(SP);
626 XPUSHs (sub_init);
627 /*
628 * the next line is slightly wrong, as PL_op->op_next
629 * is actually being executed so we skip the first op.
630 * that doesn't matter, though, since it is only
631 * pp_nextstate and we never return...
632 * ah yes, and I don't care anyways ;)
633 */
634 PUTBACK;
635 PL_op = pp_entersub();
636 SPAGAIN;
637
638 ENTER; /* necessary e.g. for dounwind */
639 }
640}
641
642static void
643continue_coro (void *arg)
644{
645 /*
646 * this is a _very_ stripped down perl interpreter ;)
647 */
648 Coro__State ctx = (Coro__State)arg;
649
650 /*FIXME*//* must set up top_env here */
651 ctx->cursp = 0;
652 PL_op = PL_op->op_next;
653 CALLRUNOPS(aTHX);
654
655 abort ();
656}
657
444STATIC void 658STATIC void
445transfer(pTHX_ struct coro *prev, struct coro *next, int flags) 659transfer(pTHX_ struct coro *prev, struct coro *next, int flags)
446{ 660{
447 dSP; 661 dSTACKLEVEL;
662 static struct coro *xnext;
448 663
449 if (prev != next) 664 if (prev != next)
450 { 665 {
451 /* 666 xnext = next;
452 * this could be done in newprocess which would lead to 667
453 * extremely elegant and fast (just SAVE/LOAD)
454 * code here, but lazy allocation of stacks has also
455 * some virtues and the overhead of the if() is nil.
456 */
457 if (next->mainstack) 668 if (next->mainstack)
458 { 669 {
459 SAVE (prev, flags); 670 SAVE (prev, flags);
460 LOAD (next); 671 LOAD (next);
672
461 /* mark this state as in-use */ 673 /* mark this state as in-use */
462 next->mainstack = 0; 674 next->mainstack = 0;
463 next->tmps_ix = -2; 675 next->tmps_ix = -2;
676
677 /* stacklevel changed? if yes, grab the stack for us! */
678 if (flags & TRANSFER_SAVE_CCTXT)
679 {
680 if (!prev->stack)
681 allocate_stack (prev, 0);
682 else if (prev->cursp != stacklevel
683 && prev->stack->usecnt > 1)
684 {
685 prev->gencnt = ++prev->stack->gencnt;
686 prev->stack->usecnt = 1;
687 }
688
689 /* has our stack been invalidated? */
690 if (next->stack && next->stack->gencnt != next->gencnt)
691 {
692 deallocate_stack (next);
693 allocate_stack (next, 1);
694 coro_create (&(next->stack->cctx),
695 continue_coro, (void *)next,
696 next->stack->sptr, labs (next->stack->ssize));
697 }
698
699 coro_transfer (&(prev->stack->cctx), &(next->stack->cctx));
700 /* don't add any code here */
701 }
702
464 } 703 }
465 else if (next->tmps_ix == -2) 704 else if (next->tmps_ix == -2)
466 {
467 croak ("tried to transfer to running coroutine"); 705 croak ("tried to transfer to running coroutine");
468 }
469 else 706 else
470 { 707 {
471 /*
472 * emulate part of the perl startup here.
473 */
474 UNOP myop;
475
476 SAVE (prev, -1); /* first get rid of the old state */ 708 SAVE (prev, -1); /* first get rid of the old state */
477 709
478 init_stacks (); /* from perl.c */ 710 if (flags & TRANSFER_SAVE_CCTXT)
479 SPAGAIN;
480
481 PL_op = (OP *)&myop;
482 /*PL_curcop = 0;*/
483 SvREFCNT_dec (GvAV (PL_defgv));
484 GvAV (PL_defgv) = next->args;
485
486 Zero(&myop, 1, UNOP);
487 myop.op_next = Nullop;
488 myop.op_flags = OPf_WANT_VOID;
489
490 PUSHMARK(SP);
491 XPUSHs ((SV*)get_cv(SUB_INIT, TRUE));
492 /*
493 * the next line is slightly wrong, as PL_op->op_next
494 * is actually being executed so we skip the first op.
495 * that doesn't matter, though, since it is only
496 * pp_nextstate and we never return...
497 * ah yes, and I don't care anyways ;)
498 */ 711 {
499 PUTBACK; 712 if (!prev->stack)
500 PL_op = pp_entersub(aTHX); 713 allocate_stack (prev, 0);
501 SPAGAIN;
502 714
503 ENTER; /* necessary e.g. for dounwind */ 715 if (prev->stack->sptr && flags & TRANSFER_LAZY_STACK)
716 {
717 setup_coro (next);
718
719 prev->stack->refcnt++;
720 prev->stack->usecnt++;
721 next->stack = prev->stack;
722 next->gencnt = prev->gencnt;
723 }
724 else
725 {
726 allocate_stack (next, 1);
727 coro_create (&(next->stack->cctx),
728 setup_coro, (void *)next,
729 next->stack->sptr, labs (next->stack->ssize));
730 coro_transfer (&(prev->stack->cctx), &(next->stack->cctx));
731 /* don't add any code here */
732 }
733 }
734 else
735 setup_coro (next);
504 } 736 }
505 } 737 }
738
739 /*
740 * xnext is now either prev or next, depending on wether
741 * we switched the c stack or not. that's why i use a global
742 * variable, that should become thread-specific at one point.
743 */
744 xnext->cursp = stacklevel;
745}
746
747static struct coro *
748sv_to_coro (SV *arg, const char *funcname, const char *varname)
749{
750 if (SvROK(arg) && SvTYPE(SvRV(arg)) == SVt_PVHV)
751 {
752 HE *he = hv_fetch_ent((HV *)SvRV(arg), ucoro_state_sv, 0, ucoro_state_hash);
753
754 if (!he)
755 croak ("%s() -- %s is a hashref but lacks the " UCORO_STATE " key", funcname, varname);
756
757 arg = HeVAL(he);
758 }
759
760 /* must also be changed inside Coro::Cont::yield */
761 if (SvROK(arg) && SvSTASH(SvRV(arg)) == coro_state_stash)
762 return (struct coro *) SvIV((SV*)SvRV(arg));
763
764 croak ("%s() -- %s is not (and contains not) a Coro::State object", funcname, varname);
765 /*NORETURN*/
766}
767
768static void
769api_transfer(pTHX_ SV *prev, SV *next, int flags)
770{
771 transfer(aTHX_
772 sv_to_coro (prev, "Coro::transfer", "prev"),
773 sv_to_coro (next, "Coro::transfer", "next"),
774 flags);
775}
776
777/** Coro ********************************************************************/
778
779#define PRIO_MAX 3
780#define PRIO_HIGH 1
781#define PRIO_NORMAL 0
782#define PRIO_LOW -1
783#define PRIO_IDLE -3
784#define PRIO_MIN -4
785
786/* for Coro.pm */
787static GV *coro_current, *coro_idle;
788static AV *coro_ready[PRIO_MAX-PRIO_MIN+1];
789static int coro_nready;
790
791static void
792coro_enq (SV *sv)
793{
794 if (SvROK (sv))
795 {
796 SV *hv = SvRV (sv);
797 if (SvTYPE (hv) == SVt_PVHV)
798 {
799 SV **xprio = hv_fetch ((HV *)hv, "prio", 4, 0);
800 int prio = xprio ? SvIV (*xprio) : PRIO_NORMAL;
801
802 prio = prio > PRIO_MAX ? PRIO_MAX
803 : prio < PRIO_MIN ? PRIO_MIN
804 : prio;
805
806 av_push (coro_ready [prio - PRIO_MIN], sv);
807 coro_nready++;
808
809 return;
810 }
811 }
812
813 croak ("Coro::ready tried to enqueue something that is not a coroutine");
814}
815
816static SV *
817coro_deq (int min_prio)
818{
819 int prio = PRIO_MAX - PRIO_MIN;
820
821 min_prio -= PRIO_MIN;
822 if (min_prio < 0)
823 min_prio = 0;
824
825 for (prio = PRIO_MAX - PRIO_MIN + 1; --prio >= min_prio; )
826 if (av_len (coro_ready[prio]) >= 0)
827 {
828 coro_nready--;
829 return av_shift (coro_ready[prio]);
830 }
831
832 return 0;
833}
834
835static void
836api_ready (SV *coro)
837{
838 coro_enq (SvREFCNT_inc (coro));
839}
840
841static void
842api_schedule (int cede)
843{
844 SV *prev, *next;
845
846 prev = GvSV (coro_current);
847
848 if (cede)
849 coro_enq (SvREFCNT_inc (prev));
850
851 next = coro_deq (PRIO_MIN);
852
853 if (!next)
854 next = SvREFCNT_inc (GvSV (coro_idle));
855
856 GvSV (coro_current) = SvREFCNT_inc (next);
857 transfer (aTHX_
858 sv_to_coro (prev, "Coro::schedule", "current coroutine"),
859 sv_to_coro (next, "Coro::schedule", "next coroutine"),
860 TRANSFER_SAVE_ALL | TRANSFER_LAZY_STACK);
861 SvREFCNT_dec (next);
862 SvREFCNT_dec (prev);
506} 863}
507 864
508MODULE = Coro::State PACKAGE = Coro::State 865MODULE = Coro::State PACKAGE = Coro::State
509 866
510PROTOTYPES: ENABLE 867PROTOTYPES: ENABLE
511 868
512BOOT: 869BOOT:
513{ /* {} necessary for stoopid perl-5.6.x */ 870{ /* {} necessary for stoopid perl-5.6.x */
871 ucoro_state_sv = newSVpv (UCORO_STATE, sizeof(UCORO_STATE) - 1);
872 PERL_HASH(ucoro_state_hash, UCORO_STATE, sizeof(UCORO_STATE) - 1);
514 HV * stash = gv_stashpvn("Coro::State", 10, TRUE); 873 coro_state_stash = gv_stashpv ("Coro::State", TRUE);
515 874
516 newCONSTSUB (stash, "SAVE_DEFAV", newSViv (SAVE_DEFAV)); 875 newCONSTSUB (coro_state_stash, "SAVE_DEFAV", newSViv (TRANSFER_SAVE_DEFAV));
517 newCONSTSUB (stash, "SAVE_DEFSV", newSViv (SAVE_DEFSV)); 876 newCONSTSUB (coro_state_stash, "SAVE_DEFSV", newSViv (TRANSFER_SAVE_DEFSV));
518 newCONSTSUB (stash, "SAVE_ERRSV", newSViv (SAVE_ERRSV)); 877 newCONSTSUB (coro_state_stash, "SAVE_ERRSV", newSViv (TRANSFER_SAVE_ERRSV));
878 newCONSTSUB (coro_state_stash, "SAVE_CCTXT", newSViv (TRANSFER_SAVE_CCTXT));
519 879
520 if (!padlist_cache) 880 if (!padlist_cache)
521 padlist_cache = newHV (); 881 padlist_cache = newHV ();
882
883 main_mainstack = PL_mainstack;
884
885 coroapi.ver = CORO_API_VERSION;
886 coroapi.transfer = api_transfer;
522} 887}
523 888
524Coro::State 889Coro::State
525_newprocess(args) 890_newprocess(args)
526 SV * args 891 SV * args
531 if (!SvROK (args) || SvTYPE (SvRV (args)) != SVt_PVAV) 896 if (!SvROK (args) || SvTYPE (SvRV (args)) != SVt_PVAV)
532 croak ("Coro::State::_newprocess expects an arrayref"); 897 croak ("Coro::State::_newprocess expects an arrayref");
533 898
534 New (0, coro, 1, struct coro); 899 New (0, coro, 1, struct coro);
535 900
901 coro->args = (AV *)SvREFCNT_inc (SvRV (args));
536 coro->mainstack = 0; /* actual work is done inside transfer */ 902 coro->mainstack = 0; /* actual work is done inside transfer */
537 coro->args = (AV *)SvREFCNT_inc (SvRV (args)); 903 coro->stack = 0;
538 904
539 RETVAL = coro; 905 RETVAL = coro;
540 OUTPUT: 906 OUTPUT:
541 RETVAL 907 RETVAL
542 908
543void 909void
544transfer(prev, next, flags = SAVE_DEFAV) 910transfer(prev, next, flags)
545 Coro::State_or_hashref prev 911 Coro::State_or_hashref prev
546 Coro::State_or_hashref next 912 Coro::State_or_hashref next
547 int flags 913 int flags
548 PROTOTYPE: @ 914 PROTOTYPE: @
549 CODE: 915 CODE:
550 916 PUTBACK;
551 transfer (aTHX_ prev, next, flags); 917 transfer (aTHX_ prev, next, flags);
918 SPAGAIN;
552 919
553void 920void
554DESTROY(coro) 921DESTROY(coro)
555 Coro::State coro 922 Coro::State coro
556 CODE: 923 CODE:
557 924
558 if (coro->mainstack) 925 if (coro->mainstack && coro->mainstack != main_mainstack)
559 { 926 {
560 struct coro temp; 927 struct coro temp;
561 928
929 PUTBACK;
562 SAVE(aTHX_ (&temp), SAVE_ALL); 930 SAVE(aTHX_ (&temp), TRANSFER_SAVE_ALL);
563 LOAD(aTHX_ coro); 931 LOAD(aTHX_ coro);
932 SPAGAIN;
564 933
565 destroy_stacks (); 934 destroy_stacks (aTHX);
566 935
567 LOAD((&temp)); /* this will get rid of defsv etc.. */ 936 LOAD((&temp)); /* this will get rid of defsv etc.. */
937 SPAGAIN;
938
939 coro->mainstack = 0;
568 } 940 }
941
942 deallocate_stack (coro);
569 943
570 Safefree (coro); 944 Safefree (coro);
571 945
572void 946void
573flush() 947flush()
574 CODE: 948 CODE:
575#ifdef MAY_FLUSH 949#ifdef MAY_FLUSH
576 flush_padlist_cache (); 950 flush_padlist_cache ();
577#endif 951#endif
578 952
953void
954_exit(code)
955 int code
956 PROTOTYPE: $
957 CODE:
958#if defined(__GLIBC__) || _POSIX_C_SOURCE
959 _exit (code);
960#else
961 signal (SIGTERM, SIG_DFL);
962 raise (SIGTERM);
963 exit (code);
964#endif
965
579MODULE = Coro::State PACKAGE = Coro::Cont 966MODULE = Coro::State PACKAGE = Coro::Cont
580 967
581# this is dirty and should be in it's own .xs 968# this is slightly dirty (should expose a c-level api)
582 969
583void 970void
584result(...) 971yield(...)
585 PROTOTYPE: @ 972 PROTOTYPE: @
586 CODE: 973 CODE:
587 static SV *returnstk; 974 static SV *returnstk;
588 SV *sv; 975 SV *sv;
589 AV *defav = GvAV (PL_defgv); 976 AV *defav = GvAV (PL_defgv);
590 struct coro *prev, *next; 977 struct coro *prev, *next;
591 978
592 if (!returnstk) 979 if (!returnstk)
593 returnstk = SvRV (get_sv ("Coro::Cont::return", FALSE)); 980 returnstk = SvRV (get_sv ("Coro::Cont::return", FALSE));
594 981
595 /* set up @_ */ 982 /* set up @_ -- ugly */
596 av_clear (defav); 983 av_clear (defav);
597 av_fill (defav, items - 1); 984 av_fill (defav, items - 1);
598 while (items--) 985 while (items--)
599 av_store (defav, items, SvREFCNT_inc (ST(items))); 986 av_store (defav, items, SvREFCNT_inc (ST(items)));
600 987
601 mg_get (returnstk); /* isn't documentation wrong for mg_get? */ 988 mg_get (returnstk); /* isn't documentation wrong for mg_get? */
602 sv = av_pop ((AV *)SvRV (returnstk)); 989 sv = av_pop ((AV *)SvRV (returnstk));
603 prev = (struct coro *)SvIV ((SV*)SvRV (*av_fetch ((AV *)SvRV (sv), 0, 0))); 990 prev = (struct coro *)SvIV ((SV*)SvRV (*av_fetch ((AV *)SvRV (sv), 0, 0)));
604 next = (struct coro *)SvIV ((SV*)SvRV (*av_fetch ((AV *)SvRV (sv), 1, 0))); 991 next = (struct coro *)SvIV ((SV*)SvRV (*av_fetch ((AV *)SvRV (sv), 1, 0)));
605 SvREFCNT_dec (sv); 992 SvREFCNT_dec (sv);
993
606 transfer(prev, next, 0); 994 transfer(aTHX_ prev, next, 0);
607 995
996MODULE = Coro::State PACKAGE = Coro
997
998# this is slightly dirty (should expose a c-level api)
999
1000BOOT:
1001{
1002 int i;
1003 HV *stash = gv_stashpv ("Coro", TRUE);
1004
1005 newCONSTSUB (stash, "PRIO_MAX", newSViv (PRIO_MAX));
1006 newCONSTSUB (stash, "PRIO_HIGH", newSViv (PRIO_HIGH));
1007 newCONSTSUB (stash, "PRIO_NORMAL", newSViv (PRIO_NORMAL));
1008 newCONSTSUB (stash, "PRIO_LOW", newSViv (PRIO_LOW));
1009 newCONSTSUB (stash, "PRIO_IDLE", newSViv (PRIO_IDLE));
1010 newCONSTSUB (stash, "PRIO_MIN", newSViv (PRIO_MIN));
1011
1012 coro_current = gv_fetchpv ("Coro::current", TRUE, SVt_PV);
1013 coro_idle = gv_fetchpv ("Coro::idle" , TRUE, SVt_PV);
1014
1015 for (i = PRIO_MAX - PRIO_MIN + 1; i--; )
1016 coro_ready[i] = newAV ();
1017
1018 {
1019 SV *sv = perl_get_sv("Coro::API", 1);
1020
1021 coroapi.schedule = api_schedule;
1022 coroapi.ready = api_ready;
1023 coroapi.nready = &coro_nready;
1024 coroapi.current = coro_current;
1025
1026 GCoroAPI = &coroapi;
1027 sv_setiv(sv, (IV)&coroapi);
1028 SvREADONLY_on(sv);
1029 }
1030}
1031
1032void
1033ready(self)
1034 SV * self
1035 CODE:
1036 api_ready (self);
1037
1038int
1039nready(...)
1040 PROTOTYPE:
1041 CODE:
1042 RETVAL = coro_nready;
1043 OUTPUT:
1044 RETVAL
1045
1046void
1047schedule(...)
1048 PROTOTYPE:
1049 ALIAS:
1050 cede = 1
1051 CODE:
1052 api_schedule (ix);
1053

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines