… | |
… | |
100 | void *sptr; |
100 | void *sptr; |
101 | long ssize; /* positive == mmap, otherwise malloc */ |
101 | long ssize; /* positive == mmap, otherwise malloc */ |
102 | } coro_stack; |
102 | } coro_stack; |
103 | |
103 | |
104 | struct coro { |
104 | struct coro { |
105 | /* the top-level JMPENV for each coroutine, needed to catch dies. */ |
|
|
106 | JMPENV start_env; |
|
|
107 | |
|
|
108 | /* the optional C context */ |
105 | /* the optional C context */ |
109 | coro_stack *stack; |
106 | coro_stack *stack; |
110 | void *cursp; |
107 | void *cursp; |
111 | int gencnt; |
108 | int gencnt; |
112 | |
109 | |
… | |
… | |
629 | } |
626 | } |
630 | else |
627 | else |
631 | { |
628 | { |
632 | UNOP myop; |
629 | UNOP myop; |
633 | |
630 | |
634 | PL_op = (OP *)&myop; |
|
|
635 | |
|
|
636 | Zero(&myop, 1, UNOP); |
631 | Zero(&myop, 1, UNOP); |
637 | myop.op_next = Nullop; |
632 | myop.op_next = Nullop; |
638 | myop.op_flags = OPf_WANT_VOID; |
633 | myop.op_flags = OPf_WANT_VOID; |
|
|
634 | |
|
|
635 | PL_op = (OP *)&myop; |
639 | |
636 | |
640 | PUSHMARK(SP); |
637 | PUSHMARK(SP); |
641 | XPUSHs (sub_init); |
638 | XPUSHs (sub_init); |
642 | /* |
639 | /* |
643 | * the next line is slightly wrong, as PL_op->op_next |
640 | * the next line is slightly wrong, as PL_op->op_next |
… | |
… | |
661 | * this is a _very_ stripped down perl interpreter ;) |
658 | * this is a _very_ stripped down perl interpreter ;) |
662 | */ |
659 | */ |
663 | dTHX; |
660 | dTHX; |
664 | Coro__State ctx = (Coro__State)arg; |
661 | Coro__State ctx = (Coro__State)arg; |
665 | |
662 | |
666 | PL_top_env = &ctx->start_env; |
663 | PL_top_env = &PL_start_env; |
667 | |
664 | |
668 | ctx->cursp = 0; |
665 | ctx->cursp = 0; |
669 | PL_op = PL_op->op_next; |
666 | PL_op = PL_op->op_next; |
670 | CALLRUNOPS(aTHX); |
667 | CALLRUNOPS(aTHX); |
671 | |
668 | |
672 | abort (); |
669 | abort (); |
673 | } |
670 | } |
674 | |
671 | |
|
|
672 | /* never call directly, always through the coro_state_transfer global variable */ |
675 | static void |
673 | static void |
676 | transfer (pTHX_ struct coro *prev, struct coro *next, int flags) |
674 | transfer_impl (pTHX_ struct coro *prev, struct coro *next, int flags) |
677 | { |
675 | { |
678 | dSTACKLEVEL; |
676 | dSTACKLEVEL; |
679 | |
677 | |
680 | if (prev != next) |
678 | if (prev != next) |
681 | { |
679 | { |
|
|
680 | /* has this coro been created yet? */ |
682 | if (next->mainstack) |
681 | if (next->mainstack) |
683 | { |
682 | { |
684 | LOCK; |
683 | LOCK; |
685 | SAVE (prev, flags); |
684 | SAVE (prev, flags); |
686 | LOAD (next); |
685 | LOAD (next); |
… | |
… | |
725 | { |
724 | { |
726 | LOCK; |
725 | LOCK; |
727 | SAVE (prev, -1); /* first get rid of the old state */ |
726 | SAVE (prev, -1); /* first get rid of the old state */ |
728 | UNLOCK; |
727 | UNLOCK; |
729 | |
728 | |
|
|
729 | /* create the coroutine for the first time */ |
730 | if (flags & TRANSFER_SAVE_CCTXT) |
730 | if (flags & TRANSFER_SAVE_CCTXT) |
731 | { |
731 | { |
732 | if (!prev->stack) |
732 | if (!prev->stack) |
733 | allocate_stack (prev, 0); |
733 | allocate_stack (prev, 0); |
734 | |
734 | |
|
|
735 | /* the new coroutine starts with start_env again */ |
|
|
736 | PL_top_env = &PL_start_env; |
|
|
737 | |
735 | if (prev->stack->sptr && flags & TRANSFER_LAZY_STACK) |
738 | if (prev->stack->sptr && flags & TRANSFER_LAZY_STACK) |
736 | { |
739 | { |
737 | PL_top_env = &next->start_env; |
|
|
738 | |
|
|
739 | setup_coro (next); |
740 | setup_coro (next); |
740 | next->cursp = stacklevel; |
741 | next->cursp = stacklevel; |
741 | |
742 | |
742 | prev->stack->refcnt++; |
743 | prev->stack->refcnt++; |
743 | prev->stack->usecnt++; |
744 | prev->stack->usecnt++; |
… | |
… | |
771 | coro_mortal = 0; |
772 | coro_mortal = 0; |
772 | } |
773 | } |
773 | UNLOCK; |
774 | UNLOCK; |
774 | } |
775 | } |
775 | |
776 | |
|
|
777 | void (*coro_state_transfer)(pTHX_ struct coro *prev, struct coro *next, int flags) = transfer_impl; |
|
|
778 | |
776 | #define SV_CORO(sv,func) \ |
779 | #define SV_CORO(sv,func) \ |
777 | do { \ |
780 | do { \ |
778 | if (SvROK (sv)) \ |
781 | if (SvROK (sv)) \ |
779 | sv = SvRV (sv); \ |
782 | sv = SvRV (sv); \ |
780 | \ |
783 | \ |
… | |
… | |
792 | if (!SvOBJECT (sv) || SvSTASH (sv) != coro_state_stash) \ |
795 | if (!SvOBJECT (sv) || SvSTASH (sv) != coro_state_stash) \ |
793 | croak ("%s() -- %s is not (and contains not) a Coro::State object", func, # sv); \ |
796 | croak ("%s() -- %s is not (and contains not) a Coro::State object", func, # sv); \ |
794 | \ |
797 | \ |
795 | } while(0) |
798 | } while(0) |
796 | |
799 | |
797 | #define SvSTATE(sv) INT2PTR (struct coro *, SvIV (sv)) |
800 | #define SvSTATE(sv) INT2PTR (struct coro *, SvIVX (sv)) |
798 | |
801 | |
799 | static void |
802 | static void |
800 | api_transfer(pTHX_ SV *prev, SV *next, int flags) |
803 | api_transfer (pTHX_ SV *prev, SV *next, int flags) |
801 | { |
804 | { |
802 | SV_CORO (prev, "Coro::transfer"); |
805 | SV_CORO (prev, "Coro::transfer"); |
803 | SV_CORO (next, "Coro::transfer"); |
806 | SV_CORO (next, "Coro::transfer"); |
804 | |
807 | |
805 | transfer (aTHX_ SvSTATE (prev), SvSTATE (next), flags); |
808 | coro_state_transfer (aTHX_ SvSTATE (prev), SvSTATE (next), flags); |
806 | } |
809 | } |
807 | |
810 | |
808 | /** Coro ********************************************************************/ |
811 | /** Coro ********************************************************************/ |
809 | |
812 | |
810 | #define PRIO_MAX 3 |
813 | #define PRIO_MAX 3 |
… | |
… | |
814 | #define PRIO_IDLE -3 |
817 | #define PRIO_IDLE -3 |
815 | #define PRIO_MIN -4 |
818 | #define PRIO_MIN -4 |
816 | |
819 | |
817 | /* for Coro.pm */ |
820 | /* for Coro.pm */ |
818 | static GV *coro_current, *coro_idle; |
821 | static GV *coro_current, *coro_idle; |
819 | static AV *coro_ready[PRIO_MAX-PRIO_MIN+1]; |
822 | static AV *coro_ready [PRIO_MAX-PRIO_MIN+1]; |
820 | static int coro_nready; |
823 | static int coro_nready; |
821 | |
824 | |
822 | static void |
825 | static void |
823 | coro_enq (pTHX_ SV *sv) |
826 | coro_enq (pTHX_ SV *sv) |
824 | { |
827 | { |
… | |
… | |
847 | min_prio -= PRIO_MIN; |
850 | min_prio -= PRIO_MIN; |
848 | if (min_prio < 0) |
851 | if (min_prio < 0) |
849 | min_prio = 0; |
852 | min_prio = 0; |
850 | |
853 | |
851 | for (prio = PRIO_MAX - PRIO_MIN + 1; --prio >= min_prio; ) |
854 | for (prio = PRIO_MAX - PRIO_MIN + 1; --prio >= min_prio; ) |
852 | if (av_len (coro_ready[prio]) >= 0) |
855 | if (AvFILLp (coro_ready [prio]) >= 0) |
853 | { |
856 | { |
854 | coro_nready--; |
857 | coro_nready--; |
855 | return av_shift (coro_ready[prio]); |
858 | return av_shift (coro_ready [prio]); |
856 | } |
859 | } |
857 | |
860 | |
858 | return 0; |
861 | return 0; |
859 | } |
862 | } |
860 | |
863 | |
… | |
… | |
894 | |
897 | |
895 | SV_CORO (next, "Coro::schedule"); |
898 | SV_CORO (next, "Coro::schedule"); |
896 | |
899 | |
897 | UNLOCK; |
900 | UNLOCK; |
898 | |
901 | |
899 | transfer (aTHX_ SvSTATE (prev), SvSTATE (next), |
902 | coro_state_transfer (aTHX_ SvSTATE (prev), SvSTATE (next), |
900 | TRANSFER_SAVE_ALL | TRANSFER_LAZY_STACK); |
903 | TRANSFER_SAVE_ALL | TRANSFER_LAZY_STACK); |
901 | } |
904 | } |
902 | |
905 | |
903 | static void |
906 | static void |
904 | api_cede (void) |
907 | api_cede (void) |
905 | { |
908 | { |
… | |
… | |
951 | Newz (0, coro, 1, struct coro); |
954 | Newz (0, coro, 1, struct coro); |
952 | |
955 | |
953 | coro->args = (AV *)SvREFCNT_inc (SvRV (args)); |
956 | coro->args = (AV *)SvREFCNT_inc (SvRV (args)); |
954 | /*coro->mainstack = 0; *//*actual work is done inside transfer */ |
957 | /*coro->mainstack = 0; *//*actual work is done inside transfer */ |
955 | /*coro->stack = 0;*/ |
958 | /*coro->stack = 0;*/ |
956 | |
|
|
957 | /* same as JMPENV_BOOTSTRAP */ |
|
|
958 | /* we might be able to recycle start_env, but safe is safe */ |
|
|
959 | /*Zero(&coro->start_env, 1, JMPENV);*/ |
|
|
960 | coro->start_env.je_ret = -1; |
|
|
961 | coro->start_env.je_mustcatch = TRUE; |
|
|
962 | |
959 | |
963 | RETVAL = coro; |
960 | RETVAL = coro; |
964 | OUTPUT: |
961 | OUTPUT: |
965 | RETVAL |
962 | RETVAL |
966 | |
963 | |
… | |
… | |
972 | PROTOTYPE: @ |
969 | PROTOTYPE: @ |
973 | CODE: |
970 | CODE: |
974 | PUTBACK; |
971 | PUTBACK; |
975 | SV_CORO (next, "Coro::transfer"); |
972 | SV_CORO (next, "Coro::transfer"); |
976 | SV_CORO (prev, "Coro::transfer"); |
973 | SV_CORO (prev, "Coro::transfer"); |
977 | transfer (aTHX_ SvSTATE (prev), SvSTATE (next), flags); |
974 | coro_state_transfer (aTHX_ SvSTATE (prev), SvSTATE (next), flags); |
978 | SPAGAIN; |
975 | SPAGAIN; |
979 | |
976 | |
980 | void |
977 | void |
981 | DESTROY(coro) |
978 | DESTROY(coro) |
982 | Coro::State coro |
979 | Coro::State coro |
… | |
… | |
1009 | PROTOTYPE: $ |
1006 | PROTOTYPE: $ |
1010 | CODE: |
1007 | CODE: |
1011 | _exit (code); |
1008 | _exit (code); |
1012 | |
1009 | |
1013 | MODULE = Coro::State PACKAGE = Coro::Cont |
1010 | MODULE = Coro::State PACKAGE = Coro::Cont |
1014 | |
|
|
1015 | # this is slightly dirty (should expose a c-level api) |
|
|
1016 | |
1011 | |
1017 | void |
1012 | void |
1018 | yield(...) |
1013 | yield(...) |
1019 | PROTOTYPE: @ |
1014 | PROTOTYPE: @ |
1020 | CODE: |
1015 | CODE: |
… | |
… | |
1034 | av_fill (defav, items - 1); |
1029 | av_fill (defav, items - 1); |
1035 | while (items--) |
1030 | while (items--) |
1036 | av_store (defav, items, SvREFCNT_inc (ST(items))); |
1031 | av_store (defav, items, SvREFCNT_inc (ST(items))); |
1037 | |
1032 | |
1038 | sv = av_pop ((AV *)SvRV (yieldstack)); |
1033 | sv = av_pop ((AV *)SvRV (yieldstack)); |
1039 | prev = INT2PTR (struct coro *, SvIV ((SV*)SvRV (*av_fetch ((AV *)SvRV (sv), 0, 0)))); |
1034 | prev = SvSTATE ((SV*)SvRV (*av_fetch ((AV *)SvRV (sv), 0, 0))); |
1040 | next = INT2PTR (struct coro *, SvIV ((SV*)SvRV (*av_fetch ((AV *)SvRV (sv), 1, 0)))); |
1035 | next = SvSTATE ((SV*)SvRV (*av_fetch ((AV *)SvRV (sv), 1, 0))); |
1041 | SvREFCNT_dec (sv); |
1036 | SvREFCNT_dec (sv); |
1042 | |
1037 | |
1043 | transfer (aTHX_ prev, next, 0); |
1038 | coro_state_transfer (aTHX_ prev, next, 0); |
1044 | |
1039 | |
1045 | MODULE = Coro::State PACKAGE = Coro |
1040 | MODULE = Coro::State PACKAGE = Coro |
1046 | |
|
|
1047 | # this is slightly dirty (should expose a c-level api) |
|
|
1048 | |
1041 | |
1049 | BOOT: |
1042 | BOOT: |
1050 | { |
1043 | { |
1051 | int i; |
1044 | int i; |
1052 | HV *stash = gv_stashpv ("Coro", TRUE); |
1045 | HV *stash = gv_stashpv ("Coro", TRUE); |
… | |
… | |
1072 | coroapi.ready = api_ready; |
1065 | coroapi.ready = api_ready; |
1073 | coroapi.nready = &coro_nready; |
1066 | coroapi.nready = &coro_nready; |
1074 | coroapi.current = coro_current; |
1067 | coroapi.current = coro_current; |
1075 | |
1068 | |
1076 | GCoroAPI = &coroapi; |
1069 | GCoroAPI = &coroapi; |
1077 | sv_setiv(sv, (IV)&coroapi); |
1070 | sv_setiv (sv, (IV)&coroapi); |
1078 | SvREADONLY_on(sv); |
1071 | SvREADONLY_on (sv); |
1079 | } |
1072 | } |
1080 | } |
1073 | } |
1081 | |
1074 | |
1082 | #if !PERL_MICRO |
1075 | #if !PERL_MICRO |
1083 | |
1076 | |