… | |
… | |
261 | /* this is a structure representing a perl-level coroutine */ |
261 | /* this is a structure representing a perl-level coroutine */ |
262 | struct coro { |
262 | struct coro { |
263 | /* the C coroutine allocated to this perl coroutine, if any */ |
263 | /* the C coroutine allocated to this perl coroutine, if any */ |
264 | coro_cctx *cctx; |
264 | coro_cctx *cctx; |
265 | |
265 | |
|
|
266 | /* ready queue */ |
|
|
267 | struct coro *next_ready; |
|
|
268 | |
266 | /* state data */ |
269 | /* state data */ |
267 | struct CoroSLF slf_frame; /* saved slf frame */ |
270 | struct CoroSLF slf_frame; /* saved slf frame */ |
268 | AV *mainstack; |
271 | AV *mainstack; |
269 | perl_slots *slot; /* basically the saved sp */ |
272 | perl_slots *slot; /* basically the saved sp */ |
270 | |
273 | |
… | |
… | |
314 | #define PRIO_MIN -4 |
317 | #define PRIO_MIN -4 |
315 | |
318 | |
316 | /* for Coro.pm */ |
319 | /* for Coro.pm */ |
317 | static SV *coro_current; |
320 | static SV *coro_current; |
318 | static SV *coro_readyhook; |
321 | static SV *coro_readyhook; |
319 | static AV *coro_ready [PRIO_MAX - PRIO_MIN + 1]; |
322 | static struct coro *coro_ready [PRIO_MAX - PRIO_MIN + 1][2]; /* head|tail */ |
320 | static CV *cv_coro_run, *cv_coro_terminate; |
323 | static CV *cv_coro_run, *cv_coro_terminate; |
321 | static struct coro *coro_first; |
324 | static struct coro *coro_first; |
322 | #define coro_nready coroapi.nready |
325 | #define coro_nready coroapi.nready |
323 | |
326 | |
324 | /** lowlevel stuff **********************************************************/ |
327 | /** lowlevel stuff **********************************************************/ |
… | |
… | |
987 | } |
990 | } |
988 | |
991 | |
989 | static void |
992 | static void |
990 | coro_destruct_perl (pTHX_ struct coro *coro) |
993 | coro_destruct_perl (pTHX_ struct coro *coro) |
991 | { |
994 | { |
|
|
995 | SV *svf [9]; |
|
|
996 | |
|
|
997 | { |
|
|
998 | struct coro *current = SvSTATE_current; |
|
|
999 | |
|
|
1000 | assert (("FATAL: tried to destroy currently running coroutine", coro->mainstack != PL_mainstack)); |
|
|
1001 | |
|
|
1002 | save_perl (aTHX_ current); |
|
|
1003 | load_perl (aTHX_ coro); |
992 | coro_unwind_stacks (aTHX); |
1004 | coro_unwind_stacks (aTHX); |
|
|
1005 | coro_destruct_stacks (aTHX); |
993 | |
1006 | |
994 | SvREFCNT_dec (GvSV (PL_defgv)); |
1007 | // now save some sv's to be free'd later |
995 | SvREFCNT_dec (GvAV (PL_defgv)); |
1008 | svf [0] = GvSV (PL_defgv); |
996 | SvREFCNT_dec (GvSV (PL_errgv)); |
1009 | svf [1] = (SV *)GvAV (PL_defgv); |
997 | SvREFCNT_dec (PL_defoutgv); |
1010 | svf [2] = GvSV (PL_errgv); |
998 | SvREFCNT_dec (PL_rs); |
1011 | svf [3] = (SV *)PL_defoutgv; |
999 | SvREFCNT_dec (GvSV (irsgv)); |
1012 | svf [4] = PL_rs; |
1000 | SvREFCNT_dec (GvHV (PL_hintgv)); |
1013 | svf [5] = GvSV (irsgv); |
|
|
1014 | svf [6] = (SV *)GvHV (PL_hintgv); |
|
|
1015 | svf [7] = PL_diehook; |
|
|
1016 | svf [8] = PL_warnhook; |
|
|
1017 | assert (9 == sizeof (svf) / sizeof (*svf)); |
1001 | |
1018 | |
1002 | SvREFCNT_dec (PL_diehook); |
1019 | load_perl (aTHX_ current); |
1003 | SvREFCNT_dec (PL_warnhook); |
|
|
1004 | |
1020 | } |
|
|
1021 | |
|
|
1022 | { |
|
|
1023 | int i; |
|
|
1024 | |
|
|
1025 | for (i = 0; i < sizeof (svf) / sizeof (*svf); ++i) |
|
|
1026 | SvREFCNT_dec (svf [i]); |
|
|
1027 | |
1005 | SvREFCNT_dec (coro->saved_deffh); |
1028 | SvREFCNT_dec (coro->saved_deffh); |
1006 | SvREFCNT_dec (coro->rouse_cb); |
1029 | SvREFCNT_dec (coro->rouse_cb); |
1007 | SvREFCNT_dec (coro->invoke_cb); |
1030 | SvREFCNT_dec (coro->invoke_cb); |
1008 | SvREFCNT_dec (coro->invoke_av); |
1031 | SvREFCNT_dec (coro->invoke_av); |
1009 | |
1032 | } |
1010 | coro_destruct_stacks (aTHX); |
|
|
1011 | } |
1033 | } |
1012 | |
1034 | |
1013 | INLINE void |
1035 | INLINE void |
1014 | free_coro_mortal (pTHX) |
1036 | free_coro_mortal (pTHX) |
1015 | { |
1037 | { |
… | |
… | |
1484 | coro_state_destroy (pTHX_ struct coro *coro) |
1506 | coro_state_destroy (pTHX_ struct coro *coro) |
1485 | { |
1507 | { |
1486 | if (coro->flags & CF_DESTROYED) |
1508 | if (coro->flags & CF_DESTROYED) |
1487 | return 0; |
1509 | return 0; |
1488 | |
1510 | |
1489 | if (coro->on_destroy) |
1511 | if (coro->on_destroy && !PL_dirty) |
1490 | coro->on_destroy (aTHX_ coro); |
1512 | coro->on_destroy (aTHX_ coro); |
1491 | |
1513 | |
1492 | coro->flags |= CF_DESTROYED; |
1514 | coro->flags |= CF_DESTROYED; |
1493 | |
1515 | |
1494 | if (coro->flags & CF_READY) |
1516 | if (coro->flags & CF_READY) |
… | |
… | |
1502 | |
1524 | |
1503 | if (coro->mainstack |
1525 | if (coro->mainstack |
1504 | && coro->mainstack != main_mainstack |
1526 | && coro->mainstack != main_mainstack |
1505 | && coro->slot |
1527 | && coro->slot |
1506 | && !PL_dirty) |
1528 | && !PL_dirty) |
1507 | { |
|
|
1508 | struct coro *current = SvSTATE_current; |
|
|
1509 | |
|
|
1510 | assert (("FATAL: tried to destroy currently running coroutine", coro->mainstack != PL_mainstack)); |
|
|
1511 | |
|
|
1512 | save_perl (aTHX_ current); |
|
|
1513 | load_perl (aTHX_ coro); |
|
|
1514 | |
|
|
1515 | coro_destruct_perl (aTHX_ coro); |
1529 | coro_destruct_perl (aTHX_ coro); |
1516 | |
|
|
1517 | load_perl (aTHX_ current); |
|
|
1518 | |
|
|
1519 | coro->slot = 0; |
|
|
1520 | } |
|
|
1521 | |
1530 | |
1522 | cctx_destroy (coro->cctx); |
1531 | cctx_destroy (coro->cctx); |
1523 | SvREFCNT_dec (coro->startcv); |
1532 | SvREFCNT_dec (coro->startcv); |
1524 | SvREFCNT_dec (coro->args); |
1533 | SvREFCNT_dec (coro->args); |
1525 | SvREFCNT_dec (CORO_THROW); |
1534 | SvREFCNT_dec (CORO_THROW); |
… | |
… | |
1612 | /** Coro ********************************************************************/ |
1621 | /** Coro ********************************************************************/ |
1613 | |
1622 | |
1614 | INLINE void |
1623 | INLINE void |
1615 | coro_enq (pTHX_ struct coro *coro) |
1624 | coro_enq (pTHX_ struct coro *coro) |
1616 | { |
1625 | { |
1617 | av_push (coro_ready [coro->prio - PRIO_MIN], SvREFCNT_inc_NN (coro->hv)); |
1626 | struct coro **ready = coro_ready [coro->prio - PRIO_MIN]; |
1618 | } |
|
|
1619 | |
1627 | |
1620 | INLINE SV * |
1628 | SvREFCNT_inc_NN (coro->hv); |
|
|
1629 | |
|
|
1630 | coro->next_ready = 0; |
|
|
1631 | *(ready [0] ? &ready [1]->next_ready : &ready [0]) = coro; |
|
|
1632 | ready [1] = coro; |
|
|
1633 | } |
|
|
1634 | |
|
|
1635 | INLINE struct coro * |
1621 | coro_deq (pTHX) |
1636 | coro_deq (pTHX) |
1622 | { |
1637 | { |
1623 | int prio; |
1638 | int prio; |
1624 | |
1639 | |
1625 | for (prio = PRIO_MAX - PRIO_MIN + 1; --prio >= 0; ) |
1640 | for (prio = PRIO_MAX - PRIO_MIN + 1; --prio >= 0; ) |
1626 | if (AvFILLp (coro_ready [prio]) >= 0) |
1641 | { |
1627 | return av_shift (coro_ready [prio]); |
1642 | struct coro **ready = coro_ready [prio]; |
|
|
1643 | |
|
|
1644 | if (ready [0]) |
|
|
1645 | { |
|
|
1646 | struct coro *coro = ready [0]; |
|
|
1647 | ready [0] = coro->next_ready; |
|
|
1648 | return coro; |
|
|
1649 | } |
|
|
1650 | } |
1628 | |
1651 | |
1629 | return 0; |
1652 | return 0; |
1630 | } |
1653 | } |
1631 | |
1654 | |
1632 | static int |
1655 | static int |
… | |
… | |
1696 | static void |
1719 | static void |
1697 | prepare_schedule (pTHX_ struct coro_transfer_args *ta) |
1720 | prepare_schedule (pTHX_ struct coro_transfer_args *ta) |
1698 | { |
1721 | { |
1699 | for (;;) |
1722 | for (;;) |
1700 | { |
1723 | { |
1701 | SV *next_sv = coro_deq (aTHX); |
1724 | struct coro *next = coro_deq (aTHX); |
1702 | |
1725 | |
1703 | if (expect_true (next_sv)) |
1726 | if (expect_true (next)) |
1704 | { |
1727 | { |
1705 | struct coro *next = SvSTATE_hv (next_sv); |
|
|
1706 | |
|
|
1707 | /* cannot transfer to destroyed coros, skip and look for next */ |
1728 | /* cannot transfer to destroyed coros, skip and look for next */ |
1708 | if (expect_false (next->flags & (CF_DESTROYED | CF_SUSPENDED))) |
1729 | if (expect_false (next->flags & (CF_DESTROYED | CF_SUSPENDED))) |
1709 | SvREFCNT_dec (next_sv); /* coro_nready has already been taken care of by destroy */ |
1730 | SvREFCNT_dec (next->hv); /* coro_nready has already been taken care of by destroy */ |
1710 | else |
1731 | else |
1711 | { |
1732 | { |
1712 | next->flags &= ~CF_READY; |
1733 | next->flags &= ~CF_READY; |
1713 | --coro_nready; |
1734 | --coro_nready; |
1714 | |
1735 | |
… | |
… | |
1988 | SV *data = (SV *)GENSUB_ARG; |
2009 | SV *data = (SV *)GENSUB_ARG; |
1989 | |
2010 | |
1990 | if (SvTYPE (SvRV (data)) != SVt_PVAV) |
2011 | if (SvTYPE (SvRV (data)) != SVt_PVAV) |
1991 | { |
2012 | { |
1992 | /* first call, set args */ |
2013 | /* first call, set args */ |
|
|
2014 | SV *coro = SvRV (data); |
1993 | AV *av = newAV (); |
2015 | AV *av = newAV (); |
1994 | SV *coro = SvRV (data); |
|
|
1995 | |
2016 | |
1996 | SvRV_set (data, (SV *)av); |
2017 | SvRV_set (data, (SV *)av); |
1997 | api_ready (aTHX_ coro); |
|
|
1998 | SvREFCNT_dec (coro); |
|
|
1999 | |
2018 | |
2000 | /* better take a full copy of the arguments */ |
2019 | /* better take a full copy of the arguments */ |
2001 | while (items--) |
2020 | while (items--) |
2002 | av_store (av, items, newSVsv (ST (items))); |
2021 | av_store (av, items, newSVsv (ST (items))); |
|
|
2022 | |
|
|
2023 | api_ready (aTHX_ coro); |
|
|
2024 | SvREFCNT_dec (coro); |
2003 | } |
2025 | } |
2004 | |
2026 | |
2005 | XSRETURN_EMPTY; |
2027 | XSRETURN_EMPTY; |
2006 | } |
2028 | } |
2007 | |
2029 | |
… | |
… | |
2024 | |
2046 | |
2025 | EXTEND (SP, AvFILLp (av) + 1); |
2047 | EXTEND (SP, AvFILLp (av) + 1); |
2026 | for (i = 0; i <= AvFILLp (av); ++i) |
2048 | for (i = 0; i <= AvFILLp (av); ++i) |
2027 | PUSHs (sv_2mortal (AvARRAY (av)[i])); |
2049 | PUSHs (sv_2mortal (AvARRAY (av)[i])); |
2028 | |
2050 | |
2029 | /* we have stolen the elements, so ste length to zero and free */ |
2051 | /* we have stolen the elements, so set length to zero and free */ |
2030 | AvFILLp (av) = -1; |
2052 | AvFILLp (av) = -1; |
2031 | av_undef (av); |
2053 | av_undef (av); |
2032 | |
2054 | |
2033 | PUTBACK; |
2055 | PUTBACK; |
2034 | } |
2056 | } |
… | |
… | |
3098 | if (ix) |
3120 | if (ix) |
3099 | eval_sv (coderef, 0); |
3121 | eval_sv (coderef, 0); |
3100 | else |
3122 | else |
3101 | call_sv (coderef, G_KEEPERR | G_EVAL | G_VOID | G_DISCARD); |
3123 | call_sv (coderef, G_KEEPERR | G_EVAL | G_VOID | G_DISCARD); |
3102 | |
3124 | |
|
|
3125 | POPSTACK; |
3103 | SPAGAIN; |
3126 | SPAGAIN; |
3104 | POPSTACK; |
|
|
3105 | |
3127 | |
3106 | if (current != coro) |
3128 | if (current != coro) |
3107 | { |
3129 | { |
3108 | PUTBACK; |
3130 | PUTBACK; |
3109 | save_perl (aTHX_ coro); |
3131 | save_perl (aTHX_ coro); |
… | |
… | |
3205 | |
3227 | |
3206 | MODULE = Coro::State PACKAGE = Coro |
3228 | MODULE = Coro::State PACKAGE = Coro |
3207 | |
3229 | |
3208 | BOOT: |
3230 | BOOT: |
3209 | { |
3231 | { |
3210 | int i; |
|
|
3211 | |
|
|
3212 | sv_pool_rss = coro_get_sv (aTHX_ "Coro::POOL_RSS" , TRUE); |
3232 | sv_pool_rss = coro_get_sv (aTHX_ "Coro::POOL_RSS" , TRUE); |
3213 | sv_pool_size = coro_get_sv (aTHX_ "Coro::POOL_SIZE" , TRUE); |
3233 | sv_pool_size = coro_get_sv (aTHX_ "Coro::POOL_SIZE" , TRUE); |
3214 | cv_coro_run = get_cv ( "Coro::_coro_run" , GV_ADD); |
3234 | cv_coro_run = get_cv ( "Coro::_coro_run" , GV_ADD); |
3215 | cv_coro_terminate = get_cv ( "Coro::terminate" , GV_ADD); |
3235 | cv_coro_terminate = get_cv ( "Coro::terminate" , GV_ADD); |
3216 | coro_current = coro_get_sv (aTHX_ "Coro::current" , FALSE); SvREADONLY_on (coro_current); |
3236 | coro_current = coro_get_sv (aTHX_ "Coro::current" , FALSE); SvREADONLY_on (coro_current); |
… | |
… | |
3231 | newCONSTSUB (coro_stash, "PRIO_NORMAL", newSViv (PRIO_NORMAL)); |
3251 | newCONSTSUB (coro_stash, "PRIO_NORMAL", newSViv (PRIO_NORMAL)); |
3232 | newCONSTSUB (coro_stash, "PRIO_LOW", newSViv (PRIO_LOW)); |
3252 | newCONSTSUB (coro_stash, "PRIO_LOW", newSViv (PRIO_LOW)); |
3233 | newCONSTSUB (coro_stash, "PRIO_IDLE", newSViv (PRIO_IDLE)); |
3253 | newCONSTSUB (coro_stash, "PRIO_IDLE", newSViv (PRIO_IDLE)); |
3234 | newCONSTSUB (coro_stash, "PRIO_MIN", newSViv (PRIO_MIN)); |
3254 | newCONSTSUB (coro_stash, "PRIO_MIN", newSViv (PRIO_MIN)); |
3235 | |
3255 | |
3236 | for (i = PRIO_MAX - PRIO_MIN + 1; i--; ) |
|
|
3237 | coro_ready[i] = newAV (); |
|
|
3238 | |
|
|
3239 | { |
3256 | { |
3240 | SV *sv = coro_get_sv (aTHX_ "Coro::API", TRUE); |
3257 | SV *sv = coro_get_sv (aTHX_ "Coro::API", TRUE); |
3241 | |
3258 | |
3242 | coroapi.schedule = api_schedule; |
3259 | coroapi.schedule = api_schedule; |
3243 | coroapi.schedule_to = api_schedule_to; |
3260 | coroapi.schedule_to = api_schedule_to; |
… | |
… | |
3334 | PROTOTYPE: |
3351 | PROTOTYPE: |
3335 | CODE: |
3352 | CODE: |
3336 | RETVAL = coro_nready; |
3353 | RETVAL = coro_nready; |
3337 | OUTPUT: |
3354 | OUTPUT: |
3338 | RETVAL |
3355 | RETVAL |
|
|
3356 | |
|
|
3357 | void |
|
|
3358 | suspend (Coro::State self) |
|
|
3359 | PROTOTYPE: $ |
|
|
3360 | CODE: |
|
|
3361 | self->flags |= CF_SUSPENDED; |
|
|
3362 | |
|
|
3363 | void |
|
|
3364 | resume (Coro::State self) |
|
|
3365 | PROTOTYPE: $ |
|
|
3366 | CODE: |
|
|
3367 | self->flags &= ~CF_SUSPENDED; |
3339 | |
3368 | |
3340 | void |
3369 | void |
3341 | _pool_handler (...) |
3370 | _pool_handler (...) |
3342 | CODE: |
3371 | CODE: |
3343 | CORO_EXECUTE_SLF_XS (slf_init_pool_handler); |
3372 | CORO_EXECUTE_SLF_XS (slf_init_pool_handler); |
… | |
… | |
3418 | av_push (*avp, SvREFCNT_inc (block)); |
3447 | av_push (*avp, SvREFCNT_inc (block)); |
3419 | |
3448 | |
3420 | if (!ix) |
3449 | if (!ix) |
3421 | on_enterleave_call (aTHX_ block); |
3450 | on_enterleave_call (aTHX_ block); |
3422 | |
3451 | |
3423 | LEAVE; /* pp_entersub unfortunately forces an ENTER/LEAVE around xs calls */ |
3452 | LEAVE; /* pp_entersub unfortunately forces an ENTER/LEAVE around XS calls */ |
3424 | SAVEDESTRUCTOR_X (ix ? coro_pop_on_leave : coro_pop_on_enter, (void *)coro); |
3453 | SAVEDESTRUCTOR_X (ix ? coro_pop_on_leave : coro_pop_on_enter, (void *)coro); |
3425 | ENTER; /* pp_entersub unfortunately forces an ENTER/LEAVE around xs calls */ |
3454 | ENTER; /* pp_entersub unfortunately forces an ENTER/LEAVE around XS calls */ |
3426 | } |
3455 | } |
3427 | |
3456 | |
3428 | |
3457 | |
3429 | MODULE = Coro::State PACKAGE = PerlIO::cede |
3458 | MODULE = Coro::State PACKAGE = PerlIO::cede |
3430 | |
3459 | |
… | |
… | |
3442 | GvSTASH (CvGV (cv)) |
3471 | GvSTASH (CvGV (cv)) |
3443 | ); |
3472 | ); |
3444 | OUTPUT: |
3473 | OUTPUT: |
3445 | RETVAL |
3474 | RETVAL |
3446 | |
3475 | |
3447 | # helper for Coro::Channel |
3476 | # helper for Coro::Channel and others |
3448 | SV * |
3477 | SV * |
3449 | _alloc (int count) |
3478 | _alloc (int count) |
3450 | CODE: |
3479 | CODE: |
3451 | RETVAL = coro_waitarray_new (aTHX_ count); |
3480 | RETVAL = coro_waitarray_new (aTHX_ count); |
3452 | OUTPUT: |
3481 | OUTPUT: |
… | |
… | |
3510 | for (i = 1; i <= wcount; ++i) |
3539 | for (i = 1; i <= wcount; ++i) |
3511 | PUSHs (sv_2mortal (newRV_inc (AvARRAY (av)[i]))); |
3540 | PUSHs (sv_2mortal (newRV_inc (AvARRAY (av)[i]))); |
3512 | } |
3541 | } |
3513 | } |
3542 | } |
3514 | |
3543 | |
|
|
3544 | MODULE = Coro::State PACKAGE = Coro::SemaphoreSet |
|
|
3545 | |
|
|
3546 | void |
|
|
3547 | _may_delete (SV *sem, int count, int extra_refs) |
|
|
3548 | PPCODE: |
|
|
3549 | { |
|
|
3550 | AV *av = (AV *)SvRV (sem); |
|
|
3551 | |
|
|
3552 | if (SvREFCNT ((SV *)av) == 1 + extra_refs |
|
|
3553 | && AvFILLp (av) == 0 /* no waiters, just count */ |
|
|
3554 | && SvIV (AvARRAY (av)[0]) == count) |
|
|
3555 | XSRETURN_YES; |
|
|
3556 | |
|
|
3557 | XSRETURN_NO; |
|
|
3558 | } |
|
|
3559 | |
3515 | MODULE = Coro::State PACKAGE = Coro::Signal |
3560 | MODULE = Coro::State PACKAGE = Coro::Signal |
3516 | |
3561 | |
3517 | SV * |
3562 | SV * |
3518 | new (SV *klass) |
3563 | new (SV *klass) |
3519 | CODE: |
3564 | CODE: |