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

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines