… | |
… | |
55 | |
55 | |
56 | #define PERL_VERSION_ATLEAST(a,b,c) \ |
56 | #define PERL_VERSION_ATLEAST(a,b,c) \ |
57 | (PERL_REVISION > (a) \ |
57 | (PERL_REVISION > (a) \ |
58 | || (PERL_REVISION == (a) \ |
58 | || (PERL_REVISION == (a) \ |
59 | && (PERL_VERSION > (b) \ |
59 | && (PERL_VERSION > (b) \ |
60 | || (PERL_VERSION == (b) && PERLSUBVERSION >= (c))))) |
60 | || (PERL_VERSION == (b) && PERL_SUBVERSION >= (c))))) |
61 | |
61 | |
62 | #if !PERL_VERSION_ATLEAST (5,6,0) |
62 | #if !PERL_VERSION_ATLEAST (5,6,0) |
63 | # ifndef PL_ppaddr |
63 | # ifndef PL_ppaddr |
64 | # define PL_ppaddr ppaddr |
64 | # define PL_ppaddr ppaddr |
65 | # endif |
65 | # endif |
… | |
… | |
98 | # define newSV(l) NEWSV(0,l) |
98 | # define newSV(l) NEWSV(0,l) |
99 | #endif |
99 | #endif |
100 | #ifndef CvISXSUB_on |
100 | #ifndef CvISXSUB_on |
101 | # define CvISXSUB_on(cv) (void)cv |
101 | # define CvISXSUB_on(cv) (void)cv |
102 | #endif |
102 | #endif |
|
|
103 | #ifndef CvISXSUB |
|
|
104 | # define CvISXSUB(cv) (CvXSUB (cv) ? TRUE : FALSE) |
|
|
105 | #endif |
103 | |
106 | |
104 | /* 5.8.7 */ |
107 | /* 5.8.7 */ |
105 | #ifndef SvRV_set |
108 | #ifndef SvRV_set |
106 | # define SvRV_set(s,v) SvRV(s) = (v) |
109 | # define SvRV_set(s,v) SvRV(s) = (v) |
107 | #endif |
110 | #endif |
… | |
… | |
167 | static AV *main_mainstack; /* used to differentiate between $main and others */ |
170 | static AV *main_mainstack; /* used to differentiate between $main and others */ |
168 | static JMPENV *main_top_env; |
171 | static JMPENV *main_top_env; |
169 | static HV *coro_state_stash, *coro_stash; |
172 | static HV *coro_state_stash, *coro_stash; |
170 | static volatile SV *coro_mortal; /* will be freed/thrown after next transfer */ |
173 | static volatile SV *coro_mortal; /* will be freed/thrown after next transfer */ |
171 | |
174 | |
|
|
175 | static AV *av_destroy; /* destruction queue */ |
|
|
176 | static SV *sv_manager; /* the manager coro */ |
|
|
177 | |
172 | static GV *irsgv; /* $/ */ |
178 | static GV *irsgv; /* $/ */ |
173 | static GV *stdoutgv; /* *STDOUT */ |
179 | static GV *stdoutgv; /* *STDOUT */ |
174 | static SV *rv_diehook; |
180 | static SV *rv_diehook; |
175 | static SV *rv_warnhook; |
181 | static SV *rv_warnhook; |
176 | static HV *hv_sig; /* %SIG */ |
182 | static HV *hv_sig; /* %SIG */ |
177 | |
183 | |
178 | /* async_pool helper stuff */ |
184 | /* async_pool helper stuff */ |
179 | static SV *sv_pool_rss; |
185 | static SV *sv_pool_rss; |
180 | static SV *sv_pool_size; |
186 | static SV *sv_pool_size; |
|
|
187 | static SV *sv_async_pool_idle; /* description string */ |
181 | static AV *av_async_pool; |
188 | static AV *av_async_pool; /* idle pool */ |
|
|
189 | static SV *sv_Coro; /* class string */ |
|
|
190 | static CV *cv_pool_handler; |
|
|
191 | static CV *cv_coro_state_new; |
182 | |
192 | |
183 | /* Coro::AnyEvent */ |
193 | /* Coro::AnyEvent */ |
184 | static SV *sv_activity; |
194 | static SV *sv_activity; |
185 | |
195 | |
186 | static struct coro_cctx *cctx_first; |
196 | static struct coro_cctx *cctx_first; |
… | |
… | |
263 | SV *except; /* exception to be thrown */ |
273 | SV *except; /* exception to be thrown */ |
264 | SV *rouse_cb; |
274 | SV *rouse_cb; |
265 | |
275 | |
266 | /* async_pool */ |
276 | /* async_pool */ |
267 | SV *saved_deffh; |
277 | SV *saved_deffh; |
|
|
278 | SV *invoke_cb; |
|
|
279 | AV *invoke_av; |
268 | |
280 | |
269 | /* linked list */ |
281 | /* linked list */ |
270 | struct coro *next, *prev; |
282 | struct coro *next, *prev; |
271 | }; |
283 | }; |
272 | |
284 | |
… | |
… | |
289 | |
301 | |
290 | /* for Coro.pm */ |
302 | /* for Coro.pm */ |
291 | static SV *coro_current; |
303 | static SV *coro_current; |
292 | static SV *coro_readyhook; |
304 | static SV *coro_readyhook; |
293 | static AV *coro_ready [PRIO_MAX - PRIO_MIN + 1]; |
305 | static AV *coro_ready [PRIO_MAX - PRIO_MIN + 1]; |
294 | static CV *cv_coro_run; |
306 | static CV *cv_coro_run, *cv_coro_terminate; |
295 | static struct coro *coro_first; |
307 | static struct coro *coro_first; |
296 | #define coro_nready coroapi.nready |
308 | #define coro_nready coroapi.nready |
297 | |
309 | |
298 | /** lowlevel stuff **********************************************************/ |
310 | /** lowlevel stuff **********************************************************/ |
299 | |
311 | |
… | |
… | |
687 | #if !PERL_VERSION_ATLEAST (5,10,0) |
699 | #if !PERL_VERSION_ATLEAST (5,10,0) |
688 | Safefree (PL_retstack); |
700 | Safefree (PL_retstack); |
689 | #endif |
701 | #endif |
690 | } |
702 | } |
691 | |
703 | |
|
|
704 | #define CORO_RSS \ |
|
|
705 | rss += sizeof (SYM (curstackinfo)); \ |
|
|
706 | rss += (SYM (curstackinfo->si_cxmax) + 1) * sizeof (PERL_CONTEXT); \ |
|
|
707 | rss += sizeof (SV) + sizeof (struct xpvav) + (1 + AvMAX (SYM (curstack))) * sizeof (SV *); \ |
|
|
708 | rss += SYM (tmps_max) * sizeof (SV *); \ |
|
|
709 | rss += (SYM (markstack_max) - SYM (markstack_ptr)) * sizeof (I32); \ |
|
|
710 | rss += SYM (scopestack_max) * sizeof (I32); \ |
|
|
711 | rss += SYM (savestack_max) * sizeof (ANY); |
|
|
712 | |
692 | static size_t |
713 | static size_t |
693 | coro_rss (pTHX_ struct coro *coro) |
714 | coro_rss (pTHX_ struct coro *coro) |
694 | { |
715 | { |
695 | size_t rss = sizeof (*coro); |
716 | size_t rss = sizeof (*coro); |
696 | |
717 | |
697 | if (coro->mainstack) |
718 | if (coro->mainstack) |
698 | { |
719 | { |
699 | perl_slots tmp_slot; |
|
|
700 | perl_slots *slot; |
|
|
701 | |
|
|
702 | if (coro->flags & CF_RUNNING) |
720 | if (coro->flags & CF_RUNNING) |
703 | { |
721 | { |
704 | slot = &tmp_slot; |
722 | #define SYM(sym) PL_ ## sym |
705 | |
723 | CORO_RSS; |
706 | #define VAR(name,type) slot->name = PL_ ## name; |
|
|
707 | # include "state.h" |
|
|
708 | #undef VAR |
724 | #undef SYM |
709 | } |
725 | } |
710 | else |
726 | else |
711 | slot = coro->slot; |
|
|
712 | |
|
|
713 | if (slot) |
|
|
714 | { |
727 | { |
715 | rss += sizeof (slot->curstackinfo); |
728 | #define SYM(sym) coro->slot->sym |
716 | rss += (slot->curstackinfo->si_cxmax + 1) * sizeof (PERL_CONTEXT); |
729 | CORO_RSS; |
717 | rss += sizeof (SV) + sizeof (struct xpvav) + (1 + AvMAX (slot->curstack)) * sizeof (SV *); |
730 | #undef SYM |
718 | rss += slot->tmps_max * sizeof (SV *); |
|
|
719 | rss += (slot->markstack_max - slot->markstack_ptr) * sizeof (I32); |
|
|
720 | rss += slot->scopestack_max * sizeof (I32); |
|
|
721 | rss += slot->savestack_max * sizeof (ANY); |
|
|
722 | |
|
|
723 | #if !PERL_VERSION_ATLEAST (5,10,0) |
|
|
724 | rss += slot->retstack_max * sizeof (OP *); |
|
|
725 | #endif |
|
|
726 | } |
731 | } |
727 | } |
732 | } |
728 | |
733 | |
729 | return rss; |
734 | return rss; |
730 | } |
735 | } |
… | |
… | |
830 | |
835 | |
831 | static int |
836 | static int |
832 | slf_check_nop (pTHX_ struct CoroSLF *frame) |
837 | slf_check_nop (pTHX_ struct CoroSLF *frame) |
833 | { |
838 | { |
834 | return 0; |
839 | return 0; |
|
|
840 | } |
|
|
841 | |
|
|
842 | static int |
|
|
843 | slf_check_repeat (pTHX_ struct CoroSLF *frame) |
|
|
844 | { |
|
|
845 | return 1; |
835 | } |
846 | } |
836 | |
847 | |
837 | static UNOP coro_setup_op; |
848 | static UNOP coro_setup_op; |
838 | |
849 | |
839 | static void NOINLINE /* noinline to keep it out of the transfer fast path */ |
850 | static void NOINLINE /* noinline to keep it out of the transfer fast path */ |
… | |
… | |
932 | SvREFCNT_dec (PL_diehook); |
943 | SvREFCNT_dec (PL_diehook); |
933 | SvREFCNT_dec (PL_warnhook); |
944 | SvREFCNT_dec (PL_warnhook); |
934 | |
945 | |
935 | SvREFCNT_dec (coro->saved_deffh); |
946 | SvREFCNT_dec (coro->saved_deffh); |
936 | SvREFCNT_dec (coro->rouse_cb); |
947 | SvREFCNT_dec (coro->rouse_cb); |
|
|
948 | SvREFCNT_dec (coro->invoke_cb); |
|
|
949 | SvREFCNT_dec (coro->invoke_av); |
937 | |
950 | |
938 | coro_destruct_stacks (aTHX); |
951 | coro_destruct_stacks (aTHX); |
939 | } |
952 | } |
940 | |
953 | |
941 | INLINE void |
954 | INLINE void |
… | |
… | |
1013 | { |
1026 | { |
1014 | PERL_CONTEXT *cx = &cxstack[cxstack_ix]; |
1027 | PERL_CONTEXT *cx = &cxstack[cxstack_ix]; |
1015 | |
1028 | |
1016 | if (CxTYPE (cx) == CXt_SUB && oldcxix < cxstack_ix) |
1029 | if (CxTYPE (cx) == CXt_SUB && oldcxix < cxstack_ix) |
1017 | { |
1030 | { |
1018 | runops_proc_t old_runops = PL_runops; |
|
|
1019 | dSP; |
1031 | dSP; |
1020 | GV *gv = CvGV (cx->blk_sub.cv); |
1032 | GV *gv = CvGV (cx->blk_sub.cv); |
1021 | SV *fullname = sv_2mortal (newSV (0)); |
1033 | SV *fullname = sv_2mortal (newSV (0)); |
1022 | |
1034 | |
1023 | if (isGV (gv)) |
1035 | if (isGV (gv)) |
… | |
… | |
1608 | api_is_ready (pTHX_ SV *coro_sv) |
1620 | api_is_ready (pTHX_ SV *coro_sv) |
1609 | { |
1621 | { |
1610 | return !!(SvSTATE (coro_sv)->flags & CF_READY); |
1622 | return !!(SvSTATE (coro_sv)->flags & CF_READY); |
1611 | } |
1623 | } |
1612 | |
1624 | |
|
|
1625 | /* expects to own a reference to next->hv */ |
1613 | INLINE void |
1626 | INLINE void |
|
|
1627 | prepare_schedule_to (pTHX_ struct coro_transfer_args *ta, struct coro *next) |
|
|
1628 | { |
|
|
1629 | SV *prev_sv = SvRV (coro_current); |
|
|
1630 | |
|
|
1631 | ta->prev = SvSTATE_hv (prev_sv); |
|
|
1632 | ta->next = next; |
|
|
1633 | |
|
|
1634 | TRANSFER_CHECK (*ta); |
|
|
1635 | |
|
|
1636 | SvRV_set (coro_current, (SV *)next->hv); |
|
|
1637 | |
|
|
1638 | free_coro_mortal (aTHX); |
|
|
1639 | coro_mortal = prev_sv; |
|
|
1640 | } |
|
|
1641 | |
|
|
1642 | static void |
1614 | prepare_schedule (pTHX_ struct coro_transfer_args *ta) |
1643 | prepare_schedule (pTHX_ struct coro_transfer_args *ta) |
1615 | { |
1644 | { |
1616 | SV *prev_sv, *next_sv; |
|
|
1617 | |
|
|
1618 | for (;;) |
1645 | for (;;) |
1619 | { |
1646 | { |
1620 | next_sv = coro_deq (aTHX); |
1647 | SV *next_sv = coro_deq (aTHX); |
1621 | |
1648 | |
1622 | /* nothing to schedule: call the idle handler */ |
|
|
1623 | if (expect_false (!next_sv)) |
1649 | if (expect_true (next_sv)) |
1624 | { |
1650 | { |
|
|
1651 | struct coro *next = SvSTATE_hv (next_sv); |
|
|
1652 | |
|
|
1653 | /* cannot transfer to destroyed coros, skip and look for next */ |
|
|
1654 | if (expect_false (next->flags & CF_DESTROYED)) |
|
|
1655 | SvREFCNT_dec (next_sv); /* coro_nready has already been taken care of by destroy */ |
|
|
1656 | else |
|
|
1657 | { |
|
|
1658 | next->flags &= ~CF_READY; |
|
|
1659 | --coro_nready; |
|
|
1660 | |
|
|
1661 | prepare_schedule_to (aTHX_ ta, next); |
|
|
1662 | break; |
|
|
1663 | } |
|
|
1664 | } |
|
|
1665 | else |
|
|
1666 | { |
|
|
1667 | /* nothing to schedule: call the idle handler */ |
1625 | dSP; |
1668 | dSP; |
1626 | |
1669 | |
1627 | ENTER; |
1670 | ENTER; |
1628 | SAVETMPS; |
1671 | SAVETMPS; |
1629 | |
1672 | |
… | |
… | |
1631 | PUTBACK; |
1674 | PUTBACK; |
1632 | call_sv (get_sv ("Coro::idle", FALSE), G_VOID | G_DISCARD); |
1675 | call_sv (get_sv ("Coro::idle", FALSE), G_VOID | G_DISCARD); |
1633 | |
1676 | |
1634 | FREETMPS; |
1677 | FREETMPS; |
1635 | LEAVE; |
1678 | LEAVE; |
1636 | continue; |
|
|
1637 | } |
1679 | } |
1638 | |
|
|
1639 | ta->next = SvSTATE_hv (next_sv); |
|
|
1640 | |
|
|
1641 | /* cannot transfer to destroyed coros, skip and look for next */ |
|
|
1642 | if (expect_false (ta->next->flags & CF_DESTROYED)) |
|
|
1643 | { |
|
|
1644 | SvREFCNT_dec (next_sv); |
|
|
1645 | /* coro_nready has already been taken care of by destroy */ |
|
|
1646 | continue; |
|
|
1647 | } |
|
|
1648 | |
|
|
1649 | --coro_nready; |
|
|
1650 | break; |
|
|
1651 | } |
1680 | } |
1652 | |
|
|
1653 | /* free this only after the transfer */ |
|
|
1654 | prev_sv = SvRV (coro_current); |
|
|
1655 | ta->prev = SvSTATE_hv (prev_sv); |
|
|
1656 | TRANSFER_CHECK (*ta); |
|
|
1657 | assert (("FATAL: next coroutine isn't marked as ready in Coro (please report)", ta->next->flags & CF_READY)); |
|
|
1658 | ta->next->flags &= ~CF_READY; |
|
|
1659 | SvRV_set (coro_current, next_sv); |
|
|
1660 | |
|
|
1661 | free_coro_mortal (aTHX); |
|
|
1662 | coro_mortal = prev_sv; |
|
|
1663 | } |
1681 | } |
1664 | |
1682 | |
1665 | INLINE void |
1683 | INLINE void |
1666 | prepare_cede (pTHX_ struct coro_transfer_args *ta) |
1684 | prepare_cede (pTHX_ struct coro_transfer_args *ta) |
1667 | { |
1685 | { |
… | |
… | |
1688 | { |
1706 | { |
1689 | struct coro_transfer_args ta; |
1707 | struct coro_transfer_args ta; |
1690 | |
1708 | |
1691 | prepare_schedule (aTHX_ &ta); |
1709 | prepare_schedule (aTHX_ &ta); |
1692 | TRANSFER (ta, 1); |
1710 | TRANSFER (ta, 1); |
|
|
1711 | } |
|
|
1712 | |
|
|
1713 | static void |
|
|
1714 | api_schedule_to (pTHX_ SV *coro_sv) |
|
|
1715 | { |
|
|
1716 | struct coro_transfer_args ta; |
|
|
1717 | struct coro *next = SvSTATE (coro_sv); |
|
|
1718 | |
|
|
1719 | SvREFCNT_inc_NN (coro_sv); |
|
|
1720 | prepare_schedule_to (aTHX_ &ta, next); |
1693 | } |
1721 | } |
1694 | |
1722 | |
1695 | static int |
1723 | static int |
1696 | api_cede (pTHX) |
1724 | api_cede (pTHX) |
1697 | { |
1725 | { |
… | |
… | |
1746 | else |
1774 | else |
1747 | coro->slot->runops = RUNOPS_DEFAULT; |
1775 | coro->slot->runops = RUNOPS_DEFAULT; |
1748 | } |
1776 | } |
1749 | } |
1777 | } |
1750 | |
1778 | |
|
|
1779 | static void |
|
|
1780 | coro_call_on_destroy (pTHX_ struct coro *coro) |
|
|
1781 | { |
|
|
1782 | SV **on_destroyp = hv_fetch (coro->hv, "_on_destroy", sizeof ("_on_destroy") - 1, 0); |
|
|
1783 | SV **statusp = hv_fetch (coro->hv, "_status", sizeof ("_status") - 1, 0); |
|
|
1784 | |
|
|
1785 | if (on_destroyp) |
|
|
1786 | { |
|
|
1787 | AV *on_destroy = (AV *)SvRV (*on_destroyp); |
|
|
1788 | |
|
|
1789 | while (AvFILLp (on_destroy) >= 0) |
|
|
1790 | { |
|
|
1791 | dSP; /* don't disturb outer sp */ |
|
|
1792 | SV *cb = av_pop (on_destroy); |
|
|
1793 | |
|
|
1794 | PUSHMARK (SP); |
|
|
1795 | |
|
|
1796 | if (statusp) |
|
|
1797 | { |
|
|
1798 | int i; |
|
|
1799 | AV *status = (AV *)SvRV (*statusp); |
|
|
1800 | EXTEND (SP, AvFILLp (status) + 1); |
|
|
1801 | |
|
|
1802 | for (i = 0; i <= AvFILLp (status); ++i) |
|
|
1803 | PUSHs (AvARRAY (status)[i]); |
|
|
1804 | } |
|
|
1805 | |
|
|
1806 | PUTBACK; |
|
|
1807 | call_sv (sv_2mortal (cb), G_VOID | G_DISCARD); |
|
|
1808 | } |
|
|
1809 | } |
|
|
1810 | } |
|
|
1811 | |
|
|
1812 | static void |
|
|
1813 | slf_init_terminate (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items) |
|
|
1814 | { |
|
|
1815 | int i; |
|
|
1816 | HV *hv = (HV *)SvRV (coro_current); |
|
|
1817 | AV *av = newAV (); |
|
|
1818 | |
|
|
1819 | av_extend (av, items - 1); |
|
|
1820 | for (i = 0; i < items; ++i) |
|
|
1821 | av_push (av, SvREFCNT_inc_NN (arg [i])); |
|
|
1822 | |
|
|
1823 | hv_store (hv, "_status", sizeof ("_status") - 1, newRV_noinc ((SV *)av), 0); |
|
|
1824 | |
|
|
1825 | av_push (av_destroy, (SV *)newRV_inc ((SV *)hv)); /* RVinc for perl */ |
|
|
1826 | api_ready (aTHX_ sv_manager); |
|
|
1827 | |
|
|
1828 | frame->prepare = prepare_schedule; |
|
|
1829 | frame->check = slf_check_repeat; |
|
|
1830 | } |
|
|
1831 | |
|
|
1832 | /*****************************************************************************/ |
|
|
1833 | /* async pool handler */ |
|
|
1834 | |
|
|
1835 | static int |
|
|
1836 | slf_check_pool_handler (pTHX_ struct CoroSLF *frame) |
|
|
1837 | { |
|
|
1838 | HV *hv = (HV *)SvRV (coro_current); |
|
|
1839 | struct coro *coro = (struct coro *)frame->data; |
|
|
1840 | |
|
|
1841 | if (!coro->invoke_cb) |
|
|
1842 | return 1; /* loop till we have invoke */ |
|
|
1843 | else |
|
|
1844 | { |
|
|
1845 | hv_store (hv, "desc", sizeof ("desc") - 1, |
|
|
1846 | newSVpvn ("[async_pool]", sizeof ("[async_pool]") - 1), 0); |
|
|
1847 | |
|
|
1848 | coro->saved_deffh = SvREFCNT_inc_NN ((SV *)PL_defoutgv); |
|
|
1849 | |
|
|
1850 | { |
|
|
1851 | dSP; |
|
|
1852 | XPUSHs (sv_2mortal (coro->invoke_cb)); coro->invoke_cb = 0; |
|
|
1853 | PUTBACK; |
|
|
1854 | } |
|
|
1855 | |
|
|
1856 | SvREFCNT_dec (GvAV (PL_defgv)); |
|
|
1857 | GvAV (PL_defgv) = coro->invoke_av; |
|
|
1858 | coro->invoke_av = 0; |
|
|
1859 | |
|
|
1860 | return 0; |
|
|
1861 | } |
|
|
1862 | } |
|
|
1863 | |
|
|
1864 | static void |
|
|
1865 | slf_init_pool_handler (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items) |
|
|
1866 | { |
|
|
1867 | HV *hv = (HV *)SvRV (coro_current); |
|
|
1868 | struct coro *coro = SvSTATE_hv ((SV *)hv); |
|
|
1869 | |
|
|
1870 | if (expect_true (coro->saved_deffh)) |
|
|
1871 | { |
|
|
1872 | /* subsequent iteration */ |
|
|
1873 | SvREFCNT_dec ((SV *)PL_defoutgv); PL_defoutgv = (GV *)coro->saved_deffh; |
|
|
1874 | coro->saved_deffh = 0; |
|
|
1875 | |
|
|
1876 | if (coro_rss (aTHX_ coro) > SvUV (sv_pool_rss) |
|
|
1877 | || av_len (av_async_pool) + 1 >= SvIV (sv_pool_size)) |
|
|
1878 | { |
|
|
1879 | coro->invoke_cb = SvREFCNT_inc_NN ((SV *)cv_coro_terminate); |
|
|
1880 | coro->invoke_av = newAV (); |
|
|
1881 | |
|
|
1882 | frame->prepare = prepare_nop; |
|
|
1883 | } |
|
|
1884 | else |
|
|
1885 | { |
|
|
1886 | av_clear (GvAV (PL_defgv)); |
|
|
1887 | hv_store (hv, "desc", sizeof ("desc") - 1, SvREFCNT_inc_NN (sv_async_pool_idle), 0); |
|
|
1888 | |
|
|
1889 | coro->prio = 0; |
|
|
1890 | |
|
|
1891 | if (coro->cctx && (coro->cctx->flags & CC_TRACE)) |
|
|
1892 | api_trace (aTHX_ coro_current, 0); |
|
|
1893 | |
|
|
1894 | frame->prepare = prepare_schedule; |
|
|
1895 | av_push (av_async_pool, SvREFCNT_inc (hv)); |
|
|
1896 | } |
|
|
1897 | } |
|
|
1898 | else |
|
|
1899 | { |
|
|
1900 | /* first iteration, simply fall through */ |
|
|
1901 | frame->prepare = prepare_nop; |
|
|
1902 | } |
|
|
1903 | |
|
|
1904 | frame->check = slf_check_pool_handler; |
|
|
1905 | frame->data = (void *)coro; |
|
|
1906 | } |
|
|
1907 | |
1751 | /*****************************************************************************/ |
1908 | /*****************************************************************************/ |
1752 | /* rouse callback */ |
1909 | /* rouse callback */ |
1753 | |
1910 | |
1754 | #define CORO_MAGIC_type_rouse PERL_MAGIC_ext |
1911 | #define CORO_MAGIC_type_rouse PERL_MAGIC_ext |
1755 | |
1912 | |
… | |
… | |
1760 | SV *data = (SV *)GENSUB_ARG; |
1917 | SV *data = (SV *)GENSUB_ARG; |
1761 | |
1918 | |
1762 | if (SvTYPE (SvRV (data)) != SVt_PVAV) |
1919 | if (SvTYPE (SvRV (data)) != SVt_PVAV) |
1763 | { |
1920 | { |
1764 | /* first call, set args */ |
1921 | /* first call, set args */ |
1765 | int i; |
|
|
1766 | AV *av = newAV (); |
1922 | AV *av = newAV (); |
1767 | SV *coro = SvRV (data); |
1923 | SV *coro = SvRV (data); |
1768 | |
1924 | |
1769 | SvRV_set (data, (SV *)av); |
1925 | SvRV_set (data, (SV *)av); |
1770 | api_ready (aTHX_ coro); |
1926 | api_ready (aTHX_ coro); |
… | |
… | |
1911 | static void |
2067 | static void |
1912 | slf_init_schedule (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items) |
2068 | slf_init_schedule (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items) |
1913 | { |
2069 | { |
1914 | frame->prepare = prepare_schedule; |
2070 | frame->prepare = prepare_schedule; |
1915 | frame->check = slf_check_nop; |
2071 | frame->check = slf_check_nop; |
|
|
2072 | } |
|
|
2073 | |
|
|
2074 | static void |
|
|
2075 | slf_prepare_schedule_to (pTHX_ struct coro_transfer_args *ta) |
|
|
2076 | { |
|
|
2077 | struct coro *next = (struct coro *)slf_frame.data; |
|
|
2078 | |
|
|
2079 | SvREFCNT_inc_NN (next->hv); |
|
|
2080 | prepare_schedule_to (aTHX_ ta, next); |
|
|
2081 | } |
|
|
2082 | |
|
|
2083 | static void |
|
|
2084 | slf_init_schedule_to (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items) |
|
|
2085 | { |
|
|
2086 | if (!items) |
|
|
2087 | croak ("Coro::schedule_to expects a coroutine argument, caught"); |
|
|
2088 | |
|
|
2089 | frame->data = (void *)SvSTATE (arg [0]); |
|
|
2090 | frame->prepare = slf_prepare_schedule_to; |
|
|
2091 | frame->check = slf_check_nop; |
|
|
2092 | } |
|
|
2093 | |
|
|
2094 | static void |
|
|
2095 | slf_init_cede_to (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items) |
|
|
2096 | { |
|
|
2097 | api_ready (aTHX_ SvRV (coro_current)); |
|
|
2098 | |
|
|
2099 | slf_init_schedule_to (aTHX_ frame, cv, arg, items); |
1916 | } |
2100 | } |
1917 | |
2101 | |
1918 | static void |
2102 | static void |
1919 | slf_init_cede (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items) |
2103 | slf_init_cede (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items) |
1920 | { |
2104 | { |
… | |
… | |
2395 | dXSARGS; |
2579 | dXSARGS; |
2396 | AV *state = (AV *)GENSUB_ARG; |
2580 | AV *state = (AV *)GENSUB_ARG; |
2397 | SV *coro = av_pop (state); |
2581 | SV *coro = av_pop (state); |
2398 | SV *data_sv = newSV (sizeof (struct io_state)); |
2582 | SV *data_sv = newSV (sizeof (struct io_state)); |
2399 | |
2583 | |
2400 | av_extend (state, items); |
2584 | av_extend (state, items - 1); |
2401 | |
2585 | |
2402 | sv_upgrade (data_sv, SVt_PV); |
2586 | sv_upgrade (data_sv, SVt_PV); |
2403 | SvCUR_set (data_sv, sizeof (struct io_state)); |
2587 | SvCUR_set (data_sv, sizeof (struct io_state)); |
2404 | SvPOK_only (data_sv); |
2588 | SvPOK_only (data_sv); |
2405 | |
2589 | |
… | |
… | |
2581 | |
2765 | |
2582 | { |
2766 | { |
2583 | SV *slf = sv_2mortal (newSViv (PTR2IV (pp_slf))); |
2767 | SV *slf = sv_2mortal (newSViv (PTR2IV (pp_slf))); |
2584 | |
2768 | |
2585 | if (!PL_custom_op_names) PL_custom_op_names = newHV (); |
2769 | if (!PL_custom_op_names) PL_custom_op_names = newHV (); |
2586 | hv_store_ent (PL_custom_op_names, slf, |
2770 | hv_store_ent (PL_custom_op_names, slf, newSVpv ("coro_slf", 0), 0); |
2587 | newSVpv ("coro_slf", 0), 0); |
|
|
2588 | |
2771 | |
2589 | if (!PL_custom_op_descs) PL_custom_op_descs = newHV (); |
2772 | if (!PL_custom_op_descs) PL_custom_op_descs = newHV (); |
2590 | hv_store_ent (PL_custom_op_descs, slf, |
2773 | hv_store_ent (PL_custom_op_descs, slf, newSVpv ("coro schedule like function", 0), 0); |
2591 | newSVpv ("coro schedule like function", 0), 0); |
|
|
2592 | } |
2774 | } |
2593 | |
2775 | |
2594 | coroapi.ver = CORO_API_VERSION; |
2776 | coroapi.ver = CORO_API_VERSION; |
2595 | coroapi.rev = CORO_API_REVISION; |
2777 | coroapi.rev = CORO_API_REVISION; |
2596 | |
2778 | |
… | |
… | |
2654 | mg->mg_flags |= MGf_DUP; |
2836 | mg->mg_flags |= MGf_DUP; |
2655 | RETVAL = sv_bless (newRV_noinc ((SV *)hv), gv_stashpv (klass, 1)); |
2837 | RETVAL = sv_bless (newRV_noinc ((SV *)hv), gv_stashpv (klass, 1)); |
2656 | |
2838 | |
2657 | if (items > 1) |
2839 | if (items > 1) |
2658 | { |
2840 | { |
2659 | av_extend (coro->args, items - 1 + ix); |
2841 | av_extend (coro->args, items - 1 + ix - 1); |
2660 | |
2842 | |
2661 | if (ix) |
2843 | if (ix) |
2662 | { |
2844 | { |
2663 | av_push (coro->args, SvREFCNT_inc_NN ((SV *)cb)); |
2845 | av_push (coro->args, SvREFCNT_inc_NN ((SV *)cb)); |
2664 | cb = cv_coro_run; |
2846 | cb = cv_coro_run; |
… | |
… | |
2874 | |
3056 | |
2875 | BOOT: |
3057 | BOOT: |
2876 | { |
3058 | { |
2877 | int i; |
3059 | int i; |
2878 | |
3060 | |
2879 | av_async_pool = coro_get_av (aTHX_ "Coro::async_pool", TRUE); |
|
|
2880 | sv_pool_rss = coro_get_sv (aTHX_ "Coro::POOL_RSS" , TRUE); |
3061 | sv_pool_rss = coro_get_sv (aTHX_ "Coro::POOL_RSS" , TRUE); |
2881 | sv_pool_size = coro_get_sv (aTHX_ "Coro::POOL_SIZE" , TRUE); |
3062 | sv_pool_size = coro_get_sv (aTHX_ "Coro::POOL_SIZE" , TRUE); |
2882 | cv_coro_run = get_cv ( "Coro::_terminate", GV_ADD); |
3063 | cv_coro_run = get_cv ( "Coro::_terminate", GV_ADD); |
|
|
3064 | cv_coro_terminate = get_cv ( "Coro::terminate" , GV_ADD); |
2883 | coro_current = coro_get_sv (aTHX_ "Coro::current" , FALSE); |
3065 | coro_current = coro_get_sv (aTHX_ "Coro::current" , FALSE); SvREADONLY_on (coro_current); |
2884 | SvREADONLY_on (coro_current); |
3066 | av_async_pool = coro_get_av (aTHX_ "Coro::async_pool", TRUE); |
|
|
3067 | av_destroy = coro_get_av (aTHX_ "Coro::destroy" , TRUE); |
|
|
3068 | sv_manager = coro_get_sv (aTHX_ "Coro::manager" , TRUE); |
|
|
3069 | |
|
|
3070 | sv_async_pool_idle = newSVpv ("[async pool idle]", 0); SvREADONLY_on (sv_async_pool_idle); |
|
|
3071 | sv_Coro = newSVpv ("Coro", 0); SvREADONLY_on (sv_Coro); |
|
|
3072 | cv_pool_handler = get_cv ("Coro::pool_handler", GV_ADD); SvREADONLY_on (cv_pool_handler); |
|
|
3073 | cv_coro_state_new = get_cv ("Coro::State::new", 0); SvREADONLY_on (cv_coro_state_new); |
2885 | |
3074 | |
2886 | coro_stash = gv_stashpv ("Coro", TRUE); |
3075 | coro_stash = gv_stashpv ("Coro", TRUE); |
2887 | |
3076 | |
2888 | newCONSTSUB (coro_stash, "PRIO_MAX", newSViv (PRIO_MAX)); |
3077 | newCONSTSUB (coro_stash, "PRIO_MAX", newSViv (PRIO_MAX)); |
2889 | newCONSTSUB (coro_stash, "PRIO_HIGH", newSViv (PRIO_HIGH)); |
3078 | newCONSTSUB (coro_stash, "PRIO_HIGH", newSViv (PRIO_HIGH)); |
… | |
… | |
2897 | |
3086 | |
2898 | { |
3087 | { |
2899 | SV *sv = coro_get_sv (aTHX_ "Coro::API", TRUE); |
3088 | SV *sv = coro_get_sv (aTHX_ "Coro::API", TRUE); |
2900 | |
3089 | |
2901 | coroapi.schedule = api_schedule; |
3090 | coroapi.schedule = api_schedule; |
|
|
3091 | coroapi.schedule_to = api_schedule_to; |
2902 | coroapi.cede = api_cede; |
3092 | coroapi.cede = api_cede; |
2903 | coroapi.cede_notself = api_cede_notself; |
3093 | coroapi.cede_notself = api_cede_notself; |
2904 | coroapi.ready = api_ready; |
3094 | coroapi.ready = api_ready; |
2905 | coroapi.is_ready = api_is_ready; |
3095 | coroapi.is_ready = api_is_ready; |
2906 | coroapi.nready = coro_nready; |
3096 | coroapi.nready = coro_nready; |
… | |
… | |
2911 | SvREADONLY_on (sv); |
3101 | SvREADONLY_on (sv); |
2912 | } |
3102 | } |
2913 | } |
3103 | } |
2914 | |
3104 | |
2915 | void |
3105 | void |
|
|
3106 | terminate (...) |
|
|
3107 | CODE: |
|
|
3108 | CORO_EXECUTE_SLF_XS (slf_init_terminate); |
|
|
3109 | |
|
|
3110 | void |
2916 | schedule (...) |
3111 | schedule (...) |
2917 | CODE: |
3112 | CODE: |
2918 | CORO_EXECUTE_SLF_XS (slf_init_schedule); |
3113 | CORO_EXECUTE_SLF_XS (slf_init_schedule); |
|
|
3114 | |
|
|
3115 | void |
|
|
3116 | schedule_to (...) |
|
|
3117 | CODE: |
|
|
3118 | CORO_EXECUTE_SLF_XS (slf_init_schedule_to); |
|
|
3119 | |
|
|
3120 | void |
|
|
3121 | cede_to (...) |
|
|
3122 | CODE: |
|
|
3123 | CORO_EXECUTE_SLF_XS (slf_init_cede_to); |
2919 | |
3124 | |
2920 | void |
3125 | void |
2921 | cede (...) |
3126 | cede (...) |
2922 | CODE: |
3127 | CODE: |
2923 | CORO_EXECUTE_SLF_XS (slf_init_cede); |
3128 | CORO_EXECUTE_SLF_XS (slf_init_cede); |
2924 | |
3129 | |
2925 | void |
3130 | void |
2926 | cede_notself (...) |
3131 | cede_notself (...) |
2927 | CODE: |
3132 | CODE: |
2928 | CORO_EXECUTE_SLF_XS (slf_init_cede_notself); |
3133 | CORO_EXECUTE_SLF_XS (slf_init_cede_notself); |
|
|
3134 | |
|
|
3135 | void |
|
|
3136 | _cancel (Coro::State self) |
|
|
3137 | CODE: |
|
|
3138 | coro_state_destroy (aTHX_ self); |
|
|
3139 | coro_call_on_destroy (aTHX_ self); |
2929 | |
3140 | |
2930 | void |
3141 | void |
2931 | _set_current (SV *current) |
3142 | _set_current (SV *current) |
2932 | PROTOTYPE: $ |
3143 | PROTOTYPE: $ |
2933 | CODE: |
3144 | CODE: |
… | |
… | |
2978 | CODE: |
3189 | CODE: |
2979 | RETVAL = coro_nready; |
3190 | RETVAL = coro_nready; |
2980 | OUTPUT: |
3191 | OUTPUT: |
2981 | RETVAL |
3192 | RETVAL |
2982 | |
3193 | |
2983 | # for async_pool speedup |
|
|
2984 | void |
3194 | void |
2985 | _pool_1 (SV *cb) |
3195 | _pool_handler (...) |
2986 | CODE: |
3196 | CODE: |
2987 | { |
3197 | CORO_EXECUTE_SLF_XS (slf_init_pool_handler); |
2988 | HV *hv = (HV *)SvRV (coro_current); |
|
|
2989 | struct coro *coro = SvSTATE_hv ((SV *)hv); |
|
|
2990 | AV *defav = GvAV (PL_defgv); |
|
|
2991 | SV *invoke = hv_delete (hv, "_invoke", sizeof ("_invoke") - 1, 0); |
|
|
2992 | AV *invoke_av; |
|
|
2993 | int i, len; |
|
|
2994 | |
3198 | |
2995 | if (!invoke) |
3199 | void |
|
|
3200 | async_pool (SV *cv, ...) |
|
|
3201 | PROTOTYPE: &@ |
|
|
3202 | PPCODE: |
|
|
3203 | { |
|
|
3204 | HV *hv = (HV *)av_pop (av_async_pool); |
|
|
3205 | AV *av = newAV (); |
|
|
3206 | SV *cb = ST (0); |
|
|
3207 | int i; |
|
|
3208 | |
|
|
3209 | av_extend (av, items - 2); |
|
|
3210 | for (i = 1; i < items; ++i) |
|
|
3211 | av_push (av, SvREFCNT_inc_NN (ST (i))); |
|
|
3212 | |
|
|
3213 | if ((SV *)hv == &PL_sv_undef) |
2996 | { |
3214 | { |
2997 | SV *old = PL_diehook; |
3215 | PUSHMARK (SP); |
2998 | PL_diehook = 0; |
3216 | EXTEND (SP, 2); |
2999 | SvREFCNT_dec (old); |
3217 | PUSHs (sv_Coro); |
3000 | croak ("\3async_pool terminate\2\n"); |
3218 | PUSHs ((SV *)cv_pool_handler); |
|
|
3219 | PUTBACK; |
|
|
3220 | call_sv ((SV *)cv_coro_state_new, G_SCALAR); |
|
|
3221 | SPAGAIN; |
|
|
3222 | |
|
|
3223 | hv = (HV *)SvREFCNT_inc_NN (SvRV (POPs)); |
3001 | } |
3224 | } |
3002 | |
3225 | |
3003 | SvREFCNT_dec (coro->saved_deffh); |
|
|
3004 | coro->saved_deffh = SvREFCNT_inc_NN ((SV *)PL_defoutgv); |
|
|
3005 | |
|
|
3006 | hv_store (hv, "desc", sizeof ("desc") - 1, |
|
|
3007 | newSVpvn ("[async_pool]", sizeof ("[async_pool]") - 1), 0); |
|
|
3008 | |
|
|
3009 | invoke_av = (AV *)SvRV (invoke); |
|
|
3010 | len = av_len (invoke_av); |
|
|
3011 | |
|
|
3012 | sv_setsv (cb, AvARRAY (invoke_av)[0]); |
|
|
3013 | |
|
|
3014 | if (len > 0) |
|
|
3015 | { |
3226 | { |
3016 | av_fill (defav, len - 1); |
3227 | struct coro *coro = SvSTATE_hv (hv); |
3017 | for (i = 0; i < len; ++i) |
3228 | |
3018 | av_store (defav, i, SvREFCNT_inc_NN (AvARRAY (invoke_av)[i + 1])); |
3229 | assert (!coro->invoke_cb); |
|
|
3230 | assert (!coro->invoke_av); |
|
|
3231 | coro->invoke_cb = SvREFCNT_inc (cb); |
|
|
3232 | coro->invoke_av = av; |
3019 | } |
3233 | } |
3020 | } |
|
|
3021 | |
3234 | |
3022 | void |
3235 | api_ready (aTHX_ (SV *)hv); |
3023 | _pool_2 (SV *cb) |
|
|
3024 | CODE: |
|
|
3025 | { |
|
|
3026 | HV *hv = (HV *)SvRV (coro_current); |
|
|
3027 | struct coro *coro = SvSTATE_hv ((SV *)hv); |
|
|
3028 | |
3236 | |
3029 | sv_setsv (cb, &PL_sv_undef); |
3237 | if (GIMME_V != G_VOID) |
3030 | |
3238 | XPUSHs (sv_2mortal (newRV_noinc ((SV *)hv))); |
3031 | SvREFCNT_dec ((SV *)PL_defoutgv); PL_defoutgv = (GV *)coro->saved_deffh; |
3239 | else |
3032 | coro->saved_deffh = 0; |
|
|
3033 | |
|
|
3034 | if (coro_rss (aTHX_ coro) > SvUV (sv_pool_rss) |
|
|
3035 | || av_len (av_async_pool) + 1 >= SvIV (sv_pool_size)) |
|
|
3036 | { |
|
|
3037 | SV *old = PL_diehook; |
|
|
3038 | PL_diehook = 0; |
|
|
3039 | SvREFCNT_dec (old); |
3240 | SvREFCNT_dec (hv); |
3040 | croak ("\3async_pool terminate\2\n"); |
|
|
3041 | } |
|
|
3042 | |
|
|
3043 | av_clear (GvAV (PL_defgv)); |
|
|
3044 | hv_store (hv, "desc", sizeof ("desc") - 1, |
|
|
3045 | newSVpvn ("[async_pool idle]", sizeof ("[async_pool idle]") - 1), 0); |
|
|
3046 | |
|
|
3047 | coro->prio = 0; |
|
|
3048 | |
|
|
3049 | if (coro->cctx && (coro->cctx->flags & CC_TRACE)) |
|
|
3050 | api_trace (aTHX_ coro_current, 0); |
|
|
3051 | |
|
|
3052 | av_push (av_async_pool, newSVsv (coro_current)); |
|
|
3053 | } |
3241 | } |
3054 | |
3242 | |
3055 | SV * |
3243 | SV * |
3056 | rouse_cb () |
3244 | rouse_cb () |
3057 | PROTOTYPE: |
3245 | PROTOTYPE: |