|
|
1 | /* this works around a bug in mingw32 providing a non-working setjmp */ |
|
|
2 | #define USE_NO_MINGW_SETJMP_TWO_ARGS |
|
|
3 | |
|
|
4 | #define NDEBUG 1 |
|
|
5 | |
1 | #include "libcoro/coro.c" |
6 | #include "libcoro/coro.c" |
2 | |
7 | |
3 | #define PERL_NO_GET_CONTEXT |
8 | #define PERL_NO_GET_CONTEXT |
4 | #define PERL_EXT |
9 | #define PERL_EXT |
5 | |
10 | |
… | |
… | |
11 | #include "schmorp.h" |
16 | #include "schmorp.h" |
12 | |
17 | |
13 | #include <stdio.h> |
18 | #include <stdio.h> |
14 | #include <errno.h> |
19 | #include <errno.h> |
15 | #include <assert.h> |
20 | #include <assert.h> |
|
|
21 | |
|
|
22 | #ifndef SVs_PADSTALE |
|
|
23 | # define SVs_PADSTALE 0 |
|
|
24 | #endif |
16 | |
25 | |
17 | #ifdef WIN32 |
26 | #ifdef WIN32 |
18 | # undef setjmp |
27 | # undef setjmp |
19 | # undef longjmp |
28 | # undef longjmp |
20 | # undef _exit |
29 | # undef _exit |
… | |
… | |
250 | |
259 | |
251 | /* on_enter/on_leave */ |
260 | /* on_enter/on_leave */ |
252 | AV *on_enter; |
261 | AV *on_enter; |
253 | AV *on_leave; |
262 | AV *on_leave; |
254 | |
263 | |
|
|
264 | /* swap_sv */ |
|
|
265 | AV *swap_sv; |
|
|
266 | |
255 | /* times */ |
267 | /* times */ |
256 | coro_ts t_cpu, t_real; |
268 | coro_ts t_cpu, t_real; |
257 | |
269 | |
258 | /* linked list */ |
270 | /* linked list */ |
259 | struct coro *next, *prev; |
271 | struct coro *next, *prev; |
… | |
… | |
289 | static OP *(*coro_old_pp_sselect) (pTHX); |
301 | static OP *(*coro_old_pp_sselect) (pTHX); |
290 | static SV *coro_select_select; |
302 | static SV *coro_select_select; |
291 | |
303 | |
292 | /* horrible hack, but if it works... */ |
304 | /* horrible hack, but if it works... */ |
293 | static OP * |
305 | static OP * |
294 | coro_pp_sselect (aTHX) |
306 | coro_pp_sselect (pTHX) |
295 | { |
307 | { |
296 | dSP; |
308 | dSP; |
297 | PUSHMARK (SP - 4); /* fake argument list */ |
309 | PUSHMARK (SP - 4); /* fake argument list */ |
298 | XPUSHs (coro_select_select); |
310 | XPUSHs (coro_select_select); |
299 | PUTBACK; |
311 | PUTBACK; |
… | |
… | |
550 | AvARRAY (av)[++AvFILLp (av)] = (SV *)CvPADLIST (cv); |
562 | AvARRAY (av)[++AvFILLp (av)] = (SV *)CvPADLIST (cv); |
551 | } |
563 | } |
552 | |
564 | |
553 | /** load & save, init *******************************************************/ |
565 | /** load & save, init *******************************************************/ |
554 | |
566 | |
|
|
567 | /* swap sv heads, at least logically */ |
|
|
568 | static void |
|
|
569 | swap_svs (pTHX_ Coro__State c) |
|
|
570 | { |
|
|
571 | int i; |
|
|
572 | |
|
|
573 | for (i = 0; i <= AvFILLp (c->swap_sv); ) |
|
|
574 | { |
|
|
575 | SV *a = AvARRAY (c->swap_sv)[i++]; |
|
|
576 | SV *b = AvARRAY (c->swap_sv)[i++]; |
|
|
577 | |
|
|
578 | const U32 keep = SVs_PADSTALE | SVs_PADTMP | SVs_PADMY; /* keep these flags */ |
|
|
579 | SV tmp; |
|
|
580 | |
|
|
581 | /* swap sv_any */ |
|
|
582 | SvANY (&tmp) = SvANY (a); SvANY (a) = SvANY (b); SvANY (b) = SvANY (&tmp); |
|
|
583 | |
|
|
584 | /* swap sv_flags */ |
|
|
585 | SvFLAGS (&tmp) = SvFLAGS (a); |
|
|
586 | SvFLAGS (a) = (SvFLAGS (a) & keep) | (SvFLAGS (b ) & ~keep); |
|
|
587 | SvFLAGS (b) = (SvFLAGS (b) & keep) | (SvFLAGS (&tmp) & ~keep); |
|
|
588 | |
|
|
589 | #if PERL_VERSION_ATLEAST (5,10,0) |
|
|
590 | /* perl 5.10 complicates this _quite_ a bit, but it also is |
|
|
591 | * is much faster, so no quarrels here. alternatively, we could |
|
|
592 | * sv_upgrade to avoid this. |
|
|
593 | */ |
|
|
594 | { |
|
|
595 | /* swap sv_u */ |
|
|
596 | tmp.sv_u = a->sv_u; a->sv_u = b->sv_u; b->sv_u = tmp.sv_u; |
|
|
597 | |
|
|
598 | /* if SvANY points to the head, we need to adjust the pointers, |
|
|
599 | * as the pointer for a still points to b, and maybe vice versa. |
|
|
600 | */ |
|
|
601 | #define svany_in_head(type) \ |
|
|
602 | (((1 << SVt_NULL) | (1 << SVt_BIND) | (1 << SVt_IV) | (1 << SVt_RV)) & (1 << (type))) |
|
|
603 | |
|
|
604 | if (svany_in_head (SvTYPE (a))) |
|
|
605 | SvANY (a) = (void *)((PTRV)SvANY (a) - (PTRV)b + (PTRV)a); |
|
|
606 | |
|
|
607 | if (svany_in_head (SvTYPE (b))) |
|
|
608 | SvANY (b) = (void *)((PTRV)SvANY (b) - (PTRV)a + (PTRV)b); |
|
|
609 | } |
|
|
610 | #endif |
|
|
611 | } |
|
|
612 | } |
|
|
613 | |
|
|
614 | #define SWAP_SVS(coro) \ |
|
|
615 | if (expect_false ((coro)->swap_sv)) \ |
|
|
616 | swap_svs (aTHX_ (coro)) |
|
|
617 | |
555 | static void |
618 | static void |
556 | on_enterleave_call (pTHX_ SV *cb); |
619 | on_enterleave_call (pTHX_ SV *cb); |
557 | |
620 | |
558 | static void |
621 | static void |
559 | load_perl (pTHX_ Coro__State c) |
622 | load_perl (pTHX_ Coro__State c) |
… | |
… | |
605 | int i; |
668 | int i; |
606 | |
669 | |
607 | for (i = 0; i <= AvFILLp (c->on_enter); ++i) |
670 | for (i = 0; i <= AvFILLp (c->on_enter); ++i) |
608 | on_enterleave_call (aTHX_ AvARRAY (c->on_enter)[i]); |
671 | on_enterleave_call (aTHX_ AvARRAY (c->on_enter)[i]); |
609 | } |
672 | } |
|
|
673 | |
|
|
674 | SWAP_SVS (c); |
610 | } |
675 | } |
611 | |
676 | |
612 | static void |
677 | static void |
613 | save_perl (pTHX_ Coro__State c) |
678 | save_perl (pTHX_ Coro__State c) |
614 | { |
679 | { |
|
|
680 | SWAP_SVS (c); |
|
|
681 | |
615 | if (expect_false (c->on_leave)) |
682 | if (expect_false (c->on_leave)) |
616 | { |
683 | { |
617 | int i; |
684 | int i; |
618 | |
685 | |
619 | for (i = AvFILLp (c->on_leave); i >= 0; --i) |
686 | for (i = AvFILLp (c->on_leave); i >= 0; --i) |
… | |
… | |
926 | slf_check_repeat (pTHX_ struct CoroSLF *frame) |
993 | slf_check_repeat (pTHX_ struct CoroSLF *frame) |
927 | { |
994 | { |
928 | return 1; |
995 | return 1; |
929 | } |
996 | } |
930 | |
997 | |
931 | static UNOP coro_setup_op; |
998 | static UNOP init_perl_op; |
932 | |
999 | |
933 | static void NOINLINE /* noinline to keep it out of the transfer fast path */ |
1000 | static void NOINLINE /* noinline to keep it out of the transfer fast path */ |
934 | coro_setup (pTHX_ struct coro *coro) |
1001 | init_perl (pTHX_ struct coro *coro) |
935 | { |
1002 | { |
936 | /* |
1003 | /* |
937 | * emulate part of the perl startup here. |
1004 | * emulate part of the perl startup here. |
938 | */ |
1005 | */ |
939 | coro_init_stacks (aTHX); |
1006 | coro_init_stacks (aTHX); |
… | |
… | |
946 | PL_comppad_name_fill = 0; |
1013 | PL_comppad_name_fill = 0; |
947 | PL_comppad_name_floor = 0; |
1014 | PL_comppad_name_floor = 0; |
948 | PL_curpm = 0; |
1015 | PL_curpm = 0; |
949 | PL_curpad = 0; |
1016 | PL_curpad = 0; |
950 | PL_localizing = 0; |
1017 | PL_localizing = 0; |
951 | PL_dirty = 0; |
|
|
952 | PL_restartop = 0; |
1018 | PL_restartop = 0; |
953 | #if PERL_VERSION_ATLEAST (5,10,0) |
1019 | #if PERL_VERSION_ATLEAST (5,10,0) |
954 | PL_parser = 0; |
1020 | PL_parser = 0; |
955 | #endif |
1021 | #endif |
956 | PL_hints = 0; |
1022 | PL_hints = 0; |
… | |
… | |
988 | */ |
1054 | */ |
989 | slf_frame.prepare = prepare_nop; /* provide a nop function for an eventual pp_slf */ |
1055 | slf_frame.prepare = prepare_nop; /* provide a nop function for an eventual pp_slf */ |
990 | slf_frame.check = slf_check_nop; /* signal pp_slf to not repeat */ |
1056 | slf_frame.check = slf_check_nop; /* signal pp_slf to not repeat */ |
991 | |
1057 | |
992 | /* and we have to provide the pp_slf op in any case, so pp_slf can skip it */ |
1058 | /* and we have to provide the pp_slf op in any case, so pp_slf can skip it */ |
993 | coro_setup_op.op_next = PL_op; |
1059 | init_perl_op.op_next = PL_op; |
994 | coro_setup_op.op_type = OP_ENTERSUB; |
1060 | init_perl_op.op_type = OP_ENTERSUB; |
995 | coro_setup_op.op_ppaddr = pp_slf; |
1061 | init_perl_op.op_ppaddr = pp_slf; |
996 | /* no flags etc. required, as an init function won't be called */ |
1062 | /* no flags etc. required, as an init function won't be called */ |
997 | |
1063 | |
998 | PL_op = (OP *)&coro_setup_op; |
1064 | PL_op = (OP *)&init_perl_op; |
999 | |
1065 | |
1000 | /* copy throw, in case it was set before coro_setup */ |
1066 | /* copy throw, in case it was set before init_perl */ |
1001 | CORO_THROW = coro->except; |
1067 | CORO_THROW = coro->except; |
|
|
1068 | |
|
|
1069 | SWAP_SVS (coro); |
1002 | |
1070 | |
1003 | if (expect_false (enable_times)) |
1071 | if (expect_false (enable_times)) |
1004 | { |
1072 | { |
1005 | coro_times_update (); |
1073 | coro_times_update (); |
1006 | coro_times_sub (coro); |
1074 | coro_times_sub (coro); |
… | |
… | |
1027 | dounwind (-1); |
1095 | dounwind (-1); |
1028 | } |
1096 | } |
1029 | } |
1097 | } |
1030 | |
1098 | |
1031 | static void |
1099 | static void |
1032 | coro_destruct_perl (pTHX_ struct coro *coro) |
1100 | destroy_perl (pTHX_ struct coro *coro) |
1033 | { |
1101 | { |
1034 | SV *svf [9]; |
1102 | SV *svf [9]; |
1035 | |
1103 | |
1036 | { |
1104 | { |
1037 | struct coro *current = SvSTATE_current; |
1105 | struct coro *current = SvSTATE_current; |
… | |
… | |
1041 | save_perl (aTHX_ current); |
1109 | save_perl (aTHX_ current); |
1042 | load_perl (aTHX_ coro); |
1110 | load_perl (aTHX_ coro); |
1043 | |
1111 | |
1044 | coro_unwind_stacks (aTHX); |
1112 | coro_unwind_stacks (aTHX); |
1045 | coro_destruct_stacks (aTHX); |
1113 | coro_destruct_stacks (aTHX); |
|
|
1114 | |
|
|
1115 | /* restore swapped sv's */ |
|
|
1116 | SWAP_SVS (coro); |
1046 | |
1117 | |
1047 | // now save some sv's to be free'd later |
1118 | // now save some sv's to be free'd later |
1048 | svf [0] = GvSV (PL_defgv); |
1119 | svf [0] = GvSV (PL_defgv); |
1049 | svf [1] = (SV *)GvAV (PL_defgv); |
1120 | svf [1] = (SV *)GvAV (PL_defgv); |
1050 | svf [2] = GvSV (PL_errgv); |
1121 | svf [2] = GvSV (PL_errgv); |
… | |
… | |
1492 | if (expect_false (next->flags & CF_NEW)) |
1563 | if (expect_false (next->flags & CF_NEW)) |
1493 | { |
1564 | { |
1494 | /* need to start coroutine */ |
1565 | /* need to start coroutine */ |
1495 | next->flags &= ~CF_NEW; |
1566 | next->flags &= ~CF_NEW; |
1496 | /* setup coroutine call */ |
1567 | /* setup coroutine call */ |
1497 | coro_setup (aTHX_ next); |
1568 | init_perl (aTHX_ next); |
1498 | } |
1569 | } |
1499 | else |
1570 | else |
1500 | load_perl (aTHX_ next); |
1571 | load_perl (aTHX_ next); |
1501 | |
1572 | |
1502 | /* possibly untie and reuse the cctx */ |
1573 | /* possibly untie and reuse the cctx */ |
… | |
… | |
1565 | |
1636 | |
1566 | if (coro->mainstack |
1637 | if (coro->mainstack |
1567 | && coro->mainstack != main_mainstack |
1638 | && coro->mainstack != main_mainstack |
1568 | && coro->slot |
1639 | && coro->slot |
1569 | && !PL_dirty) |
1640 | && !PL_dirty) |
1570 | coro_destruct_perl (aTHX_ coro); |
1641 | destroy_perl (aTHX_ coro); |
|
|
1642 | |
|
|
1643 | if (coro->next) coro->next->prev = coro->prev; |
|
|
1644 | if (coro->prev) coro->prev->next = coro->next; |
|
|
1645 | if (coro == coro_first) coro_first = coro->next; |
1571 | |
1646 | |
1572 | cctx_destroy (coro->cctx); |
1647 | cctx_destroy (coro->cctx); |
1573 | SvREFCNT_dec (coro->startcv); |
1648 | SvREFCNT_dec (coro->startcv); |
1574 | SvREFCNT_dec (coro->args); |
1649 | SvREFCNT_dec (coro->args); |
|
|
1650 | SvREFCNT_dec (coro->swap_sv); |
1575 | SvREFCNT_dec (CORO_THROW); |
1651 | SvREFCNT_dec (CORO_THROW); |
1576 | |
|
|
1577 | if (coro->next) coro->next->prev = coro->prev; |
|
|
1578 | if (coro->prev) coro->prev->next = coro->next; |
|
|
1579 | if (coro == coro_first) coro_first = coro->next; |
|
|
1580 | |
1652 | |
1581 | return 1; |
1653 | return 1; |
1582 | } |
1654 | } |
1583 | |
1655 | |
1584 | static int |
1656 | static int |
… | |
… | |
1754 | { |
1826 | { |
1755 | /* nothing to schedule: call the idle handler */ |
1827 | /* nothing to schedule: call the idle handler */ |
1756 | if (SvROK (sv_idle) |
1828 | if (SvROK (sv_idle) |
1757 | && SvOBJECT (SvRV (sv_idle))) |
1829 | && SvOBJECT (SvRV (sv_idle))) |
1758 | { |
1830 | { |
|
|
1831 | if (SvRV (sv_idle) == SvRV (coro_current)) |
|
|
1832 | croak ("FATAL: $Coro::IDLE blocked itself - did you try to block inside an event loop callback? Caught"); |
|
|
1833 | |
1759 | ++coro_nready; /* hack so that api_ready doesn't invoke ready hook */ |
1834 | ++coro_nready; /* hack so that api_ready doesn't invoke ready hook */ |
1760 | api_ready (aTHX_ SvRV (sv_idle)); |
1835 | api_ready (aTHX_ SvRV (sv_idle)); |
1761 | --coro_nready; |
1836 | --coro_nready; |
1762 | } |
1837 | } |
1763 | else |
1838 | else |
1764 | { |
1839 | { |
|
|
1840 | /* TODO: deprecated, remove, cannot work reliably *//*D*/ |
1765 | dSP; |
1841 | dSP; |
1766 | |
1842 | |
1767 | ENTER; |
1843 | ENTER; |
1768 | SAVETMPS; |
1844 | SAVETMPS; |
1769 | |
1845 | |
… | |
… | |
2348 | if (PL_op->op_flags & OPf_STACKED) |
2424 | if (PL_op->op_flags & OPf_STACKED) |
2349 | { |
2425 | { |
2350 | if (items > slf_arga) |
2426 | if (items > slf_arga) |
2351 | { |
2427 | { |
2352 | slf_arga = items; |
2428 | slf_arga = items; |
2353 | free (slf_argv); |
2429 | Safefree (slf_argv); |
2354 | slf_argv = malloc (slf_arga * sizeof (SV *)); |
2430 | New (0, slf_argv, slf_arga, SV *); |
2355 | } |
2431 | } |
2356 | |
2432 | |
2357 | slf_argc = items; |
2433 | slf_argc = items; |
2358 | |
2434 | |
2359 | for (i = 0; i < items; ++i) |
2435 | for (i = 0; i < items; ++i) |
… | |
… | |
2720 | { |
2796 | { |
2721 | SV *cb_cv = s_get_cv_croak (arg [1]); |
2797 | SV *cb_cv = s_get_cv_croak (arg [1]); |
2722 | av_push (av, SvREFCNT_inc_NN (cb_cv)); |
2798 | av_push (av, SvREFCNT_inc_NN (cb_cv)); |
2723 | |
2799 | |
2724 | if (SvIVX (AvARRAY (av)[0])) |
2800 | if (SvIVX (AvARRAY (av)[0])) |
2725 | coro_signal_wake (aTHX_ av, 1); /* ust be the only waiter */ |
2801 | coro_signal_wake (aTHX_ av, 1); /* must be the only waiter */ |
2726 | |
2802 | |
2727 | frame->prepare = prepare_nop; |
2803 | frame->prepare = prepare_nop; |
2728 | frame->check = slf_check_nop; |
2804 | frame->check = slf_check_nop; |
2729 | } |
2805 | } |
2730 | else if (SvIVX (AvARRAY (av)[0])) |
2806 | else if (SvIVX (AvARRAY (av)[0])) |
… | |
… | |
2892 | /* now call the AIO function - we assume our request is uncancelable */ |
2968 | /* now call the AIO function - we assume our request is uncancelable */ |
2893 | PUTBACK; |
2969 | PUTBACK; |
2894 | call_sv ((SV *)req, G_VOID | G_DISCARD); |
2970 | call_sv ((SV *)req, G_VOID | G_DISCARD); |
2895 | } |
2971 | } |
2896 | |
2972 | |
2897 | /* now that the requets is going, we loop toll we have a result */ |
2973 | /* now that the request is going, we loop till we have a result */ |
2898 | frame->data = (void *)state; |
2974 | frame->data = (void *)state; |
2899 | frame->prepare = prepare_schedule; |
2975 | frame->prepare = prepare_schedule; |
2900 | frame->check = slf_check_aio_req; |
2976 | frame->check = slf_check_aio_req; |
2901 | } |
2977 | } |
2902 | |
2978 | |
… | |
… | |
3201 | RETVAL = boolSV (coro->flags & ix); |
3277 | RETVAL = boolSV (coro->flags & ix); |
3202 | OUTPUT: |
3278 | OUTPUT: |
3203 | RETVAL |
3279 | RETVAL |
3204 | |
3280 | |
3205 | void |
3281 | void |
3206 | throw (Coro::State self, SV *throw = &PL_sv_undef) |
3282 | throw (Coro::State self, SV *exception = &PL_sv_undef) |
3207 | PROTOTYPE: $;$ |
3283 | PROTOTYPE: $;$ |
3208 | CODE: |
3284 | CODE: |
3209 | { |
3285 | { |
3210 | struct coro *current = SvSTATE_current; |
3286 | struct coro *current = SvSTATE_current; |
3211 | SV **throwp = self == current ? &CORO_THROW : &self->except; |
3287 | SV **exceptionp = self == current ? &CORO_THROW : &self->except; |
3212 | SvREFCNT_dec (*throwp); |
3288 | SvREFCNT_dec (*exceptionp); |
3213 | SvGETMAGIC (throw); |
3289 | SvGETMAGIC (exception); |
3214 | *throwp = SvOK (throw) ? newSVsv (throw) : 0; |
3290 | *exceptionp = SvOK (exception) ? newSVsv (exception) : 0; |
3215 | } |
3291 | } |
3216 | |
3292 | |
3217 | void |
3293 | void |
3218 | api_trace (SV *coro, int flags = CC_TRACE | CC_TRACE_SUB) |
3294 | api_trace (SV *coro, int flags = CC_TRACE | CC_TRACE_SUB) |
3219 | PROTOTYPE: $;$ |
3295 | PROTOTYPE: $;$ |
… | |
… | |
3314 | |
3390 | |
3315 | if (expect_false (current == self)) |
3391 | if (expect_false (current == self)) |
3316 | coro_times_sub (SvSTATE (coro_current)); |
3392 | coro_times_sub (SvSTATE (coro_current)); |
3317 | } |
3393 | } |
3318 | |
3394 | |
|
|
3395 | void |
|
|
3396 | swap_sv (Coro::State coro, SV *sv, SV *swapsv) |
|
|
3397 | CODE: |
|
|
3398 | { |
|
|
3399 | struct coro *current = SvSTATE_current; |
|
|
3400 | |
|
|
3401 | if (current == coro) |
|
|
3402 | SWAP_SVS (current); |
|
|
3403 | |
|
|
3404 | if (!coro->swap_sv) |
|
|
3405 | coro->swap_sv = newAV (); |
|
|
3406 | |
|
|
3407 | av_push (coro->swap_sv, SvREFCNT_inc_NN (SvRV (sv ))); |
|
|
3408 | av_push (coro->swap_sv, SvREFCNT_inc_NN (SvRV (swapsv))); |
|
|
3409 | |
|
|
3410 | if (current == coro) |
|
|
3411 | SWAP_SVS (current); |
|
|
3412 | } |
|
|
3413 | |
|
|
3414 | |
3319 | MODULE = Coro::State PACKAGE = Coro |
3415 | MODULE = Coro::State PACKAGE = Coro |
3320 | |
3416 | |
3321 | BOOT: |
3417 | BOOT: |
3322 | { |
3418 | { |
3323 | sv_pool_rss = coro_get_sv (aTHX_ "Coro::POOL_RSS" , TRUE); |
3419 | sv_pool_rss = coro_get_sv (aTHX_ "Coro::POOL_RSS" , TRUE); |
… | |
… | |
3365 | SV * |
3461 | SV * |
3366 | async (...) |
3462 | async (...) |
3367 | PROTOTYPE: &@ |
3463 | PROTOTYPE: &@ |
3368 | CODE: |
3464 | CODE: |
3369 | RETVAL = coro_new (aTHX_ coro_stash, &ST (0), items, 1); |
3465 | RETVAL = coro_new (aTHX_ coro_stash, &ST (0), items, 1); |
3370 | api_ready (RETVAL); |
3466 | api_ready (aTHX_ RETVAL); |
3371 | OUTPUT: |
3467 | OUTPUT: |
3372 | RETVAL |
3468 | RETVAL |
3373 | |
3469 | |
3374 | void |
3470 | void |
3375 | terminate (...) |
3471 | terminate (...) |
… | |
… | |
3779 | { |
3875 | { |
3780 | PL_ppaddr [OP_SSELECT] = coro_old_pp_sselect; |
3876 | PL_ppaddr [OP_SSELECT] = coro_old_pp_sselect; |
3781 | coro_old_pp_sselect = 0; |
3877 | coro_old_pp_sselect = 0; |
3782 | } |
3878 | } |
3783 | |
3879 | |
3784 | |
|
|