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.231 by root, Thu Apr 24 10:31:59 2008 UTC vs.
Revision 1.242 by root, Thu Aug 14 12:48:20 2008 UTC

15 15
16#ifdef WIN32 16#ifdef WIN32
17# undef setjmp 17# undef setjmp
18# undef longjmp 18# undef longjmp
19# undef _exit 19# undef _exit
20#include <setjmp.h> 20# define setjmp _setjmp // deep magic, don't ask
21#else 21#else
22# include <inttypes.h> /* most portable stdint.h */ 22# include <inttypes.h> /* most portable stdint.h */
23#endif 23#endif
24 24
25#ifdef HAVE_MMAP 25#ifdef HAVE_MMAP
171/* async_pool helper stuff */ 171/* async_pool helper stuff */
172static SV *sv_pool_rss; 172static SV *sv_pool_rss;
173static SV *sv_pool_size; 173static SV *sv_pool_size;
174static AV *av_async_pool; 174static AV *av_async_pool;
175 175
176/* Coro::AnyEvent */
177static SV *sv_activity;
178
176static struct coro_cctx *cctx_first; 179static struct coro_cctx *cctx_first;
177static int cctx_count, cctx_idle; 180static int cctx_count, cctx_idle;
178 181
179enum { 182enum {
180 CC_MAPPED = 0x01, 183 CC_MAPPED = 0x01,
265#define PRIO_IDLE -3 268#define PRIO_IDLE -3
266#define PRIO_MIN -4 269#define PRIO_MIN -4
267 270
268/* for Coro.pm */ 271/* for Coro.pm */
269static SV *coro_current; 272static SV *coro_current;
273static SV *coro_readyhook;
270static AV *coro_ready [PRIO_MAX-PRIO_MIN+1]; 274static AV *coro_ready [PRIO_MAX-PRIO_MIN+1];
271static int coro_nready; 275static int coro_nready;
272static struct coro *coro_first; 276static struct coro *coro_first;
273 277
274/** lowlevel stuff **********************************************************/ 278/** lowlevel stuff **********************************************************/
510 514
511 if (expect_true (CvDEPTH (cv))) 515 if (expect_true (CvDEPTH (cv)))
512 { 516 {
513 EXTEND (SP, 3); 517 EXTEND (SP, 3);
514 PUSHs ((SV *)CvPADLIST (cv)); 518 PUSHs ((SV *)CvPADLIST (cv));
515 PUSHs (INT2PTR (SV *, CvDEPTH (cv))); 519 PUSHs (INT2PTR (SV *, (IV)CvDEPTH (cv)));
516 PUSHs ((SV *)cv); 520 PUSHs ((SV *)cv);
517 521
518 CvDEPTH (cv) = 0; 522 CvDEPTH (cv) = 0;
519 get_padlist (aTHX_ cv); 523 get_padlist (aTHX_ cv);
520 } 524 }
679 683
680/** coroutine stack handling ************************************************/ 684/** coroutine stack handling ************************************************/
681 685
682static int (*orig_sigelem_get) (pTHX_ SV *sv, MAGIC *mg); 686static int (*orig_sigelem_get) (pTHX_ SV *sv, MAGIC *mg);
683static int (*orig_sigelem_set) (pTHX_ SV *sv, MAGIC *mg); 687static int (*orig_sigelem_set) (pTHX_ SV *sv, MAGIC *mg);
688static int (*orig_sigelem_clr) (pTHX_ SV *sv, MAGIC *mg);
689
690/* apparently < 5.8.8 */
691#ifndef MgPV_nolen_const
692#define MgPV_nolen_const(mg) (((((int)(mg)->mg_len)) == HEf_SVKEY) ? \
693 SvPV_nolen((SV*)((mg)->mg_ptr)) : \
694 (const char*)(mg)->mg_ptr)
695#endif
684 696
685/* 697/*
686 * This overrides the default magic get method of %SIG elements. 698 * This overrides the default magic get method of %SIG elements.
687 * The original one doesn't provide for reading back of PL_diehook/PL_warnhook 699 * The original one doesn't provide for reading back of PL_diehook/PL_warnhook
688 * and instead of tryign to save and restore the hash elements, we just provide 700 * and instead of tryign to save and restore the hash elements, we just provide
696{ 708{
697 const char *s = MgPV_nolen_const (mg); 709 const char *s = MgPV_nolen_const (mg);
698 710
699 if (*s == '_') 711 if (*s == '_')
700 { 712 {
701 if (strEQ (s, "__DIE__" ) && PL_diehook ) return sv_setsv (sv, PL_diehook ), 0; 713 SV **svp = 0;
702 if (strEQ (s, "__WARN__") && PL_warnhook) return sv_setsv (sv, PL_warnhook), 0; 714
715 if (strEQ (s, "__DIE__" )) svp = &PL_diehook;
716 if (strEQ (s, "__WARN__")) svp = &PL_warnhook;
717
718 if (svp)
719 {
720 sv_setsv (sv, *svp ? *svp : &PL_sv_undef);
721 return 0;
722 }
703 } 723 }
704 724
705 return orig_sigelem_get ? orig_sigelem_get (aTHX_ sv, mg) : 0; 725 return orig_sigelem_get ? orig_sigelem_get (aTHX_ sv, mg) : 0;
726}
727
728static int
729coro_sigelem_clr (pTHX_ SV *sv, MAGIC *mg)
730{
731 const char *s = MgPV_nolen_const (mg);
732
733 if (*s == '_')
734 {
735 SV **svp = 0;
736
737 if (strEQ (s, "__DIE__" )) svp = &PL_diehook;
738 if (strEQ (s, "__WARN__")) svp = &PL_warnhook;
739
740 if (svp)
741 {
742 SV *old = *svp;
743 *svp = 0;
744 SvREFCNT_dec (old);
745 return 0;
746 }
747 }
748
749 return orig_sigelem_clr ? orig_sigelem_clr (aTHX_ sv, mg) : 0;
706} 750}
707 751
708static int 752static int
709coro_sigelem_set (pTHX_ SV *sv, MAGIC *mg) 753coro_sigelem_set (pTHX_ SV *sv, MAGIC *mg)
710{ 754{
1379static int 1423static int
1380api_ready (SV *coro_sv) 1424api_ready (SV *coro_sv)
1381{ 1425{
1382 dTHX; 1426 dTHX;
1383 struct coro *coro; 1427 struct coro *coro;
1428 SV *sv_hook;
1429 void (*xs_hook)(void);
1384 1430
1385 if (SvROK (coro_sv)) 1431 if (SvROK (coro_sv))
1386 coro_sv = SvRV (coro_sv); 1432 coro_sv = SvRV (coro_sv);
1387 1433
1388 coro = SvSTATE (coro_sv); 1434 coro = SvSTATE (coro_sv);
1391 return 0; 1437 return 0;
1392 1438
1393 coro->flags |= CF_READY; 1439 coro->flags |= CF_READY;
1394 1440
1395 LOCK; 1441 LOCK;
1442
1443 sv_hook = coro_nready ? 0 : coro_readyhook;
1444 xs_hook = coro_nready ? 0 : coroapi.readyhook;
1445
1396 coro_enq (aTHX_ SvREFCNT_inc (coro_sv)); 1446 coro_enq (aTHX_ SvREFCNT_inc (coro_sv));
1397 ++coro_nready; 1447 ++coro_nready;
1448
1398 UNLOCK; 1449 UNLOCK;
1450
1451 if (sv_hook)
1452 {
1453 dSP;
1454
1455 ENTER;
1456 SAVETMPS;
1457
1458 PUSHMARK (SP);
1459 PUTBACK;
1460 call_sv (sv_hook, G_DISCARD);
1461 SPAGAIN;
1462
1463 FREETMPS;
1464 LEAVE;
1465 }
1466
1467 if (xs_hook)
1468 xs_hook ();
1399 1469
1400 return 1; 1470 return 1;
1401} 1471}
1402 1472
1403static int 1473static int
1567 BOOT_PAGESIZE; 1637 BOOT_PAGESIZE;
1568 1638
1569 irsgv = gv_fetchpv ("/" , GV_ADD|GV_NOTQUAL, SVt_PV); 1639 irsgv = gv_fetchpv ("/" , GV_ADD|GV_NOTQUAL, SVt_PV);
1570 stdoutgv = gv_fetchpv ("STDOUT", GV_ADD|GV_NOTQUAL, SVt_PVIO); 1640 stdoutgv = gv_fetchpv ("STDOUT", GV_ADD|GV_NOTQUAL, SVt_PVIO);
1571 1641
1572 orig_sigelem_get = PL_vtbl_sigelem.svt_get; 1642 orig_sigelem_get = PL_vtbl_sigelem.svt_get; PL_vtbl_sigelem.svt_get = coro_sigelem_get;
1573 PL_vtbl_sigelem.svt_get = coro_sigelem_get; 1643 orig_sigelem_set = PL_vtbl_sigelem.svt_set; PL_vtbl_sigelem.svt_set = coro_sigelem_set;
1574 orig_sigelem_set = PL_vtbl_sigelem.svt_set; 1644 orig_sigelem_clr = PL_vtbl_sigelem.svt_clear; PL_vtbl_sigelem.svt_clear = coro_sigelem_clr;
1575 PL_vtbl_sigelem.svt_set = coro_sigelem_set;
1576 1645
1577 hv_sig = coro_get_hv (aTHX_ "SIG", TRUE); 1646 hv_sig = coro_get_hv (aTHX_ "SIG", TRUE);
1578 rv_diehook = newRV_inc ((SV *)gv_fetchpv ("Coro::State::diehook" , 0, SVt_PVCV)); 1647 rv_diehook = newRV_inc ((SV *)gv_fetchpv ("Coro::State::diehook" , 0, SVt_PVCV));
1579 rv_warnhook = newRV_inc ((SV *)gv_fetchpv ("Coro::State::warnhook", 0, SVt_PVCV)); 1648 rv_warnhook = newRV_inc ((SV *)gv_fetchpv ("Coro::State::warnhook", 0, SVt_PVCV));
1580 1649
1589 main_top_env = PL_top_env; 1658 main_top_env = PL_top_env;
1590 1659
1591 while (main_top_env->je_prev) 1660 while (main_top_env->je_prev)
1592 main_top_env = main_top_env->je_prev; 1661 main_top_env = main_top_env->je_prev;
1593 1662
1594 coroapi.ver = CORO_API_VERSION; 1663 coroapi.ver = CORO_API_VERSION;
1595 coroapi.rev = CORO_API_REVISION; 1664 coroapi.rev = CORO_API_REVISION;
1596 coroapi.transfer = api_transfer; 1665 coroapi.transfer = api_transfer;
1597 1666
1598 assert (("PRIO_NORMAL must be 0", !PRIO_NORMAL)); 1667 assert (("PRIO_NORMAL must be 0", !PRIO_NORMAL));
1599} 1668}
1600 1669
1601SV * 1670SV *
1685 RETVAL = coro_state_destroy (aTHX_ SvSTATE (coro_sv)); 1754 RETVAL = coro_state_destroy (aTHX_ SvSTATE (coro_sv));
1686 OUTPUT: 1755 OUTPUT:
1687 RETVAL 1756 RETVAL
1688 1757
1689void 1758void
1690_exit (code) 1759_exit (int code)
1691 int code
1692 PROTOTYPE: $ 1760 PROTOTYPE: $
1693 CODE: 1761 CODE:
1694 _exit (code); 1762 _exit (code);
1695 1763
1696int 1764int
1787 1855
1788void 1856void
1789api_trace (SV *coro, int flags = CC_TRACE | CC_TRACE_SUB) 1857api_trace (SV *coro, int flags = CC_TRACE | CC_TRACE_SUB)
1790 1858
1791SV * 1859SV *
1792has_stack (Coro::State coro) 1860has_cctx (Coro::State coro)
1793 PROTOTYPE: $ 1861 PROTOTYPE: $
1794 CODE: 1862 CODE:
1795 RETVAL = boolSV (!!coro->cctx); 1863 RETVAL = boolSV (!!coro->cctx);
1796 OUTPUT: 1864 OUTPUT:
1797 RETVAL 1865 RETVAL
1822force_cctx () 1890force_cctx ()
1823 CODE: 1891 CODE:
1824 struct coro *coro = SvSTATE (coro_current); 1892 struct coro *coro = SvSTATE (coro_current);
1825 coro->cctx->idle_sp = 0; 1893 coro->cctx->idle_sp = 0;
1826 1894
1827MODULE = Coro::State PACKAGE = Coro
1828
1829BOOT:
1830{
1831 int i;
1832
1833 sv_pool_rss = coro_get_sv (aTHX_ "Coro::POOL_RSS" , TRUE);
1834 sv_pool_size = coro_get_sv (aTHX_ "Coro::POOL_SIZE" , TRUE);
1835 av_async_pool = coro_get_av (aTHX_ "Coro::async_pool", TRUE);
1836
1837 coro_current = coro_get_sv (aTHX_ "Coro::current", FALSE);
1838 SvREADONLY_on (coro_current);
1839
1840 coro_stash = gv_stashpv ("Coro", TRUE);
1841
1842 newCONSTSUB (coro_stash, "PRIO_MAX", newSViv (PRIO_MAX));
1843 newCONSTSUB (coro_stash, "PRIO_HIGH", newSViv (PRIO_HIGH));
1844 newCONSTSUB (coro_stash, "PRIO_NORMAL", newSViv (PRIO_NORMAL));
1845 newCONSTSUB (coro_stash, "PRIO_LOW", newSViv (PRIO_LOW));
1846 newCONSTSUB (coro_stash, "PRIO_IDLE", newSViv (PRIO_IDLE));
1847 newCONSTSUB (coro_stash, "PRIO_MIN", newSViv (PRIO_MIN));
1848
1849 for (i = PRIO_MAX - PRIO_MIN + 1; i--; )
1850 coro_ready[i] = newAV ();
1851
1852 {
1853 SV *sv = perl_get_sv ("Coro::API", TRUE);
1854 perl_get_sv ("Coro::API", TRUE); /* silence 5.10 warning */
1855
1856 coroapi.schedule = api_schedule;
1857 coroapi.cede = api_cede;
1858 coroapi.cede_notself = api_cede_notself;
1859 coroapi.ready = api_ready;
1860 coroapi.is_ready = api_is_ready;
1861 coroapi.nready = &coro_nready;
1862 coroapi.current = coro_current;
1863
1864 GCoroAPI = &coroapi;
1865 sv_setiv (sv, (IV)&coroapi);
1866 SvREADONLY_on (sv);
1867 }
1868}
1869
1870void
1871_set_current (SV *current)
1872 PROTOTYPE: $
1873 CODE:
1874 SvREFCNT_dec (SvRV (coro_current));
1875 SvRV_set (coro_current, SvREFCNT_inc (SvRV (current)));
1876
1877int
1878prio (Coro::State coro, int newprio = 0)
1879 ALIAS:
1880 nice = 1
1881 CODE:
1882{
1883 RETVAL = coro->prio;
1884
1885 if (items > 1)
1886 {
1887 if (ix)
1888 newprio = coro->prio - newprio;
1889
1890 if (newprio < PRIO_MIN) newprio = PRIO_MIN;
1891 if (newprio > PRIO_MAX) newprio = PRIO_MAX;
1892
1893 coro->prio = newprio;
1894 }
1895}
1896 OUTPUT:
1897 RETVAL
1898
1899SV *
1900ready (SV *self)
1901 PROTOTYPE: $
1902 CODE:
1903 RETVAL = boolSV (api_ready (self));
1904 OUTPUT:
1905 RETVAL
1906
1907int
1908nready (...)
1909 PROTOTYPE:
1910 CODE:
1911 RETVAL = coro_nready;
1912 OUTPUT:
1913 RETVAL
1914
1915void 1895void
1916throw (Coro::State self, SV *throw = &PL_sv_undef) 1896throw (Coro::State self, SV *throw = &PL_sv_undef)
1917 PROTOTYPE: $;$ 1897 PROTOTYPE: $;$
1918 CODE: 1898 CODE:
1919 SvREFCNT_dec (self->throw); 1899 SvREFCNT_dec (self->throw);
1933 SV **dst = ix ? (SV **)&self->slot->defav : (SV **)&self->slot->defsv; 1913 SV **dst = ix ? (SV **)&self->slot->defav : (SV **)&self->slot->defsv;
1934 1914
1935 SV *tmp = *src; *src = *dst; *dst = tmp; 1915 SV *tmp = *src; *src = *dst; *dst = tmp;
1936 } 1916 }
1937 1917
1918MODULE = Coro::State PACKAGE = Coro
1919
1920BOOT:
1921{
1922 int i;
1923
1924 av_async_pool = coro_get_av (aTHX_ "Coro::async_pool", TRUE);
1925 sv_pool_rss = coro_get_sv (aTHX_ "Coro::POOL_RSS" , TRUE);
1926 sv_pool_size = coro_get_sv (aTHX_ "Coro::POOL_SIZE" , TRUE);
1927
1928 coro_current = coro_get_sv (aTHX_ "Coro::current", FALSE);
1929 SvREADONLY_on (coro_current);
1930
1931 coro_stash = gv_stashpv ("Coro", TRUE);
1932
1933 newCONSTSUB (coro_stash, "PRIO_MAX", newSViv (PRIO_MAX));
1934 newCONSTSUB (coro_stash, "PRIO_HIGH", newSViv (PRIO_HIGH));
1935 newCONSTSUB (coro_stash, "PRIO_NORMAL", newSViv (PRIO_NORMAL));
1936 newCONSTSUB (coro_stash, "PRIO_LOW", newSViv (PRIO_LOW));
1937 newCONSTSUB (coro_stash, "PRIO_IDLE", newSViv (PRIO_IDLE));
1938 newCONSTSUB (coro_stash, "PRIO_MIN", newSViv (PRIO_MIN));
1939
1940 for (i = PRIO_MAX - PRIO_MIN + 1; i--; )
1941 coro_ready[i] = newAV ();
1942
1943 {
1944 SV *sv = perl_get_sv ("Coro::API", TRUE);
1945 perl_get_sv ("Coro::API", TRUE); /* silence 5.10 warning */
1946
1947 coroapi.schedule = api_schedule;
1948 coroapi.cede = api_cede;
1949 coroapi.cede_notself = api_cede_notself;
1950 coroapi.ready = api_ready;
1951 coroapi.is_ready = api_is_ready;
1952 coroapi.nready = &coro_nready;
1953 coroapi.current = coro_current;
1954
1955 GCoroAPI = &coroapi;
1956 sv_setiv (sv, (IV)&coroapi);
1957 SvREADONLY_on (sv);
1958 }
1959}
1960
1961void
1962_set_current (SV *current)
1963 PROTOTYPE: $
1964 CODE:
1965 SvREFCNT_dec (SvRV (coro_current));
1966 SvRV_set (coro_current, SvREFCNT_inc (SvRV (current)));
1967
1968void
1969_set_readyhook (SV *hook)
1970 PROTOTYPE: $
1971 CODE:
1972 LOCK;
1973 SvREFCNT_dec (coro_readyhook);
1974 coro_readyhook = SvOK (hook) ? newSVsv (hook) : 0;
1975 UNLOCK;
1976
1977int
1978prio (Coro::State coro, int newprio = 0)
1979 ALIAS:
1980 nice = 1
1981 CODE:
1982{
1983 RETVAL = coro->prio;
1984
1985 if (items > 1)
1986 {
1987 if (ix)
1988 newprio = coro->prio - newprio;
1989
1990 if (newprio < PRIO_MIN) newprio = PRIO_MIN;
1991 if (newprio > PRIO_MAX) newprio = PRIO_MAX;
1992
1993 coro->prio = newprio;
1994 }
1995}
1996 OUTPUT:
1997 RETVAL
1998
1999SV *
2000ready (SV *self)
2001 PROTOTYPE: $
2002 CODE:
2003 RETVAL = boolSV (api_ready (self));
2004 OUTPUT:
2005 RETVAL
2006
2007int
2008nready (...)
2009 PROTOTYPE:
2010 CODE:
2011 RETVAL = coro_nready;
2012 OUTPUT:
2013 RETVAL
2014
1938# for async_pool speedup 2015# for async_pool speedup
1939void 2016void
1940_pool_1 (SV *cb) 2017_pool_1 (SV *cb)
1941 CODE: 2018 CODE:
1942{ 2019{
1947 AV *invoke_av; 2024 AV *invoke_av;
1948 int i, len; 2025 int i, len;
1949 2026
1950 if (!invoke) 2027 if (!invoke)
1951 { 2028 {
1952 SvREFCNT_dec (PL_diehook); PL_diehook = 0; 2029 SV *old = PL_diehook;
2030 PL_diehook = 0;
2031 SvREFCNT_dec (old);
1953 croak ("\3async_pool terminate\2\n"); 2032 croak ("\3async_pool terminate\2\n");
1954 } 2033 }
1955 2034
1956 SvREFCNT_dec (coro->saved_deffh); 2035 SvREFCNT_dec (coro->saved_deffh);
1957 coro->saved_deffh = SvREFCNT_inc ((SV *)PL_defoutgv); 2036 coro->saved_deffh = SvREFCNT_inc ((SV *)PL_defoutgv);
1986 coro->saved_deffh = 0; 2065 coro->saved_deffh = 0;
1987 2066
1988 if (coro_rss (aTHX_ coro) > SvIV (sv_pool_rss) 2067 if (coro_rss (aTHX_ coro) > SvIV (sv_pool_rss)
1989 || av_len (av_async_pool) + 1 >= SvIV (sv_pool_size)) 2068 || av_len (av_async_pool) + 1 >= SvIV (sv_pool_size))
1990 { 2069 {
1991 SvREFCNT_dec (PL_diehook); PL_diehook = 0; 2070 SV *old = PL_diehook;
2071 PL_diehook = 0;
2072 SvREFCNT_dec (old);
1992 croak ("\3async_pool terminate\2\n"); 2073 croak ("\3async_pool terminate\2\n");
1993 } 2074 }
1994 2075
1995 av_clear (GvAV (PL_defgv)); 2076 av_clear (GvAV (PL_defgv));
1996 hv_store ((HV *)SvRV (coro_current), "desc", sizeof ("desc") - 1, 2077 hv_store ((HV *)SvRV (coro_current), "desc", sizeof ("desc") - 1,
2037 PL_laststype = data->laststype; 2118 PL_laststype = data->laststype;
2038 PL_laststatval = data->laststatval; 2119 PL_laststatval = data->laststatval;
2039 PL_statcache = data->statcache; 2120 PL_statcache = data->statcache;
2040} 2121}
2041 2122
2123
2124MODULE = Coro::State PACKAGE = Coro::AnyEvent
2125
2126BOOT:
2127 sv_activity = coro_get_sv (aTHX_ "Coro::AnyEvent::ACTIVITY", TRUE);
2128
2129SV *
2130_schedule (...)
2131 PROTOTYPE: @
2132 CODE:
2133{
2134 static int incede;
2135
2136 api_cede_notself ();
2137
2138 ++incede;
2139 while (coro_nready >= incede && api_cede ())
2140 ;
2141
2142 sv_setsv (sv_activity, &PL_sv_undef);
2143 if (coro_nready >= incede)
2144 {
2145 PUSHMARK (SP);
2146 PUTBACK;
2147 call_pv ("Coro::AnyEvent::_activity", G_DISCARD | G_EVAL);
2148 SPAGAIN;
2149 }
2150
2151 --incede;
2152}
2153

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines