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.239 by root, Sat Jun 28 23:14:19 2008 UTC vs.
Revision 1.247 by root, Wed Sep 24 21:31:29 2008 UTC

4#define PERL_EXT 4#define PERL_EXT
5 5
6#include "EXTERN.h" 6#include "EXTERN.h"
7#include "perl.h" 7#include "perl.h"
8#include "XSUB.h" 8#include "XSUB.h"
9#include "perliol.h"
9 10
10#include "patchlevel.h" 11#include "patchlevel.h"
11 12
12#include <stdio.h> 13#include <stdio.h>
13#include <errno.h> 14#include <errno.h>
153 I32 laststype; 154 I32 laststype;
154 int laststatval; 155 int laststatval;
155 Stat_t statcache; 156 Stat_t statcache;
156}; 157};
157 158
159static double (*nvtime)(); /* so why doesn't it take void? */
160
158static size_t coro_stacksize = CORO_STACKSIZE; 161static size_t coro_stacksize = CORO_STACKSIZE;
159static struct CoroAPI coroapi; 162static struct CoroAPI coroapi;
160static AV *main_mainstack; /* used to differentiate between $main and others */ 163static AV *main_mainstack; /* used to differentiate between $main and others */
161static JMPENV *main_top_env; 164static JMPENV *main_top_env;
162static HV *coro_state_stash, *coro_stash; 165static HV *coro_state_stash, *coro_stash;
361 364
362 /* casting is fun. */ 365 /* casting is fun. */
363 while (&PL_sv_undef != (SV *)(padlist = (AV *)av_pop (av))) 366 while (&PL_sv_undef != (SV *)(padlist = (AV *)av_pop (av)))
364 free_padlist (aTHX_ padlist); 367 free_padlist (aTHX_ padlist);
365 368
369 SvREFCNT_dec (av); /* sv_magicext increased the refcount */
370
366 return 0; 371 return 0;
367} 372}
368 373
369#define CORO_MAGIC_type_cv PERL_MAGIC_ext 374#define CORO_MAGIC_type_cv PERL_MAGIC_ext
370#define CORO_MAGIC_type_state PERL_MAGIC_ext 375#define CORO_MAGIC_type_state PERL_MAGIC_ext
420 if (expect_true (mg && AvFILLp ((av = (AV *)mg->mg_obj)) >= 0)) 425 if (expect_true (mg && AvFILLp ((av = (AV *)mg->mg_obj)) >= 0))
421 CvPADLIST (cv) = (AV *)AvARRAY (av)[AvFILLp (av)--]; 426 CvPADLIST (cv) = (AV *)AvARRAY (av)[AvFILLp (av)--];
422 else 427 else
423 { 428 {
424#if CORO_PREFER_PERL_FUNCTIONS 429#if CORO_PREFER_PERL_FUNCTIONS
425 /* this is probably cleaner, but also slower? */ 430 /* this is probably cleaner? but also slower! */
431 /* in practise, it seems to be less stable */
426 CV *cp = Perl_cv_clone (cv); 432 CV *cp = Perl_cv_clone (cv);
427 CvPADLIST (cv) = CvPADLIST (cp); 433 CvPADLIST (cv) = CvPADLIST (cp);
428 CvPADLIST (cp) = 0; 434 CvPADLIST (cp) = 0;
429 SvREFCNT_dec (cp); 435 SvREFCNT_dec (cp);
430#else 436#else
663 #undef VAR 669 #undef VAR
664 } 670 }
665 else 671 else
666 slot = coro->slot; 672 slot = coro->slot;
667 673
674 if (slot)
675 {
668 rss += sizeof (slot->curstackinfo); 676 rss += sizeof (slot->curstackinfo);
669 rss += (slot->curstackinfo->si_cxmax + 1) * sizeof (PERL_CONTEXT); 677 rss += (slot->curstackinfo->si_cxmax + 1) * sizeof (PERL_CONTEXT);
670 rss += sizeof (SV) + sizeof (struct xpvav) + (1 + AvMAX (slot->curstack)) * sizeof (SV *); 678 rss += sizeof (SV) + sizeof (struct xpvav) + (1 + AvMAX (slot->curstack)) * sizeof (SV *);
671 rss += slot->tmps_max * sizeof (SV *); 679 rss += slot->tmps_max * sizeof (SV *);
672 rss += (slot->markstack_max - slot->markstack_ptr) * sizeof (I32); 680 rss += (slot->markstack_max - slot->markstack_ptr) * sizeof (I32);
673 rss += slot->scopestack_max * sizeof (I32); 681 rss += slot->scopestack_max * sizeof (I32);
674 rss += slot->savestack_max * sizeof (ANY); 682 rss += slot->savestack_max * sizeof (ANY);
675 683
676#if !PERL_VERSION_ATLEAST (5,10,0) 684#if !PERL_VERSION_ATLEAST (5,10,0)
677 rss += slot->retstack_max * sizeof (OP *); 685 rss += slot->retstack_max * sizeof (OP *);
678#endif 686#endif
687 }
679 } 688 }
680 689
681 return rss; 690 return rss;
682} 691}
683 692
686static int (*orig_sigelem_get) (pTHX_ SV *sv, MAGIC *mg); 695static int (*orig_sigelem_get) (pTHX_ SV *sv, MAGIC *mg);
687static int (*orig_sigelem_set) (pTHX_ SV *sv, MAGIC *mg); 696static int (*orig_sigelem_set) (pTHX_ SV *sv, MAGIC *mg);
688static int (*orig_sigelem_clr) (pTHX_ SV *sv, MAGIC *mg); 697static int (*orig_sigelem_clr) (pTHX_ SV *sv, MAGIC *mg);
689 698
690/* apparently < 5.8.8 */ 699/* apparently < 5.8.8 */
691#undef MgPV_nolen_const
692#ifndef MgPV_nolen_const 700#ifndef MgPV_nolen_const
693#define MgPV_nolen_const(mg) (((((int)(mg)->mg_len)) == HEf_SVKEY) ? \ 701#define MgPV_nolen_const(mg) (((((int)(mg)->mg_len)) == HEf_SVKEY) ? \
694 SvPV_nolen_const((SV*)((mg)->mg_ptr)) : \ 702 SvPV_nolen((SV*)((mg)->mg_ptr)) : \
695 (const char*)(mg)->mg_ptr) 703 (const char*)(mg)->mg_ptr)
696#endif 704#endif
697 705
698/* 706/*
699 * This overrides the default magic get method of %SIG elements. 707 * This overrides the default magic get method of %SIG elements.
1624 else 1632 else
1625 coro->slot->runops = RUNOPS_DEFAULT; 1633 coro->slot->runops = RUNOPS_DEFAULT;
1626 } 1634 }
1627} 1635}
1628 1636
1637static int
1638coro_gensub_free (pTHX_ SV *sv, MAGIC *mg)
1639{
1640 AV *padlist;
1641 AV *av = (AV *)mg->mg_obj;
1642
1643 abort ();
1644
1645 return 0;
1646}
1647
1648static MGVTBL coro_gensub_vtbl = {
1649 0, 0, 0, 0,
1650 coro_gensub_free
1651};
1652
1653/*****************************************************************************/
1654/* PerlIO::cede */
1655
1656typedef struct
1657{
1658 PerlIOBuf base;
1659 NV next, every;
1660} PerlIOCede;
1661
1662static IV
1663PerlIOCede_pushed (pTHX_ PerlIO *f, const char *mode, SV *arg, PerlIO_funcs *tab)
1664{
1665 PerlIOCede *self = PerlIOSelf (f, PerlIOCede);
1666
1667 self->every = SvCUR (arg) ? SvNV (arg) : 0.01;
1668 self->next = nvtime () + self->every;
1669
1670 return PerlIOBuf_pushed (aTHX_ f, mode, Nullsv, tab);
1671}
1672
1673static SV *
1674PerlIOCede_getarg (pTHX_ PerlIO *f, CLONE_PARAMS *param, int flags)
1675{
1676 PerlIOCede *self = PerlIOSelf (f, PerlIOCede);
1677
1678 return newSVnv (self->every);
1679}
1680
1681static IV
1682PerlIOCede_flush (pTHX_ PerlIO *f)
1683{
1684 PerlIOCede *self = PerlIOSelf (f, PerlIOCede);
1685 double now = nvtime ();
1686
1687 if (now >= self->next)
1688 {
1689 api_cede ();
1690 self->next = now + self->every;
1691 }
1692
1693 return PerlIOBuf_flush (f);
1694}
1695
1696static PerlIO_funcs PerlIO_cede =
1697{
1698 sizeof(PerlIO_funcs),
1699 "cede",
1700 sizeof(PerlIOCede),
1701 PERLIO_K_DESTRUCT | PERLIO_K_RAW,
1702 PerlIOCede_pushed,
1703 PerlIOBuf_popped,
1704 PerlIOBuf_open,
1705 PerlIOBase_binmode,
1706 PerlIOCede_getarg,
1707 PerlIOBase_fileno,
1708 PerlIOBuf_dup,
1709 PerlIOBuf_read,
1710 PerlIOBuf_unread,
1711 PerlIOBuf_write,
1712 PerlIOBuf_seek,
1713 PerlIOBuf_tell,
1714 PerlIOBuf_close,
1715 PerlIOCede_flush,
1716 PerlIOBuf_fill,
1717 PerlIOBase_eof,
1718 PerlIOBase_error,
1719 PerlIOBase_clearerr,
1720 PerlIOBase_setlinebuf,
1721 PerlIOBuf_get_base,
1722 PerlIOBuf_bufsiz,
1723 PerlIOBuf_get_ptr,
1724 PerlIOBuf_get_cnt,
1725 PerlIOBuf_set_ptrcnt,
1726};
1727
1728
1629MODULE = Coro::State PACKAGE = Coro::State PREFIX = api_ 1729MODULE = Coro::State PACKAGE = Coro::State PREFIX = api_
1630 1730
1631PROTOTYPES: DISABLE 1731PROTOTYPES: DISABLE
1632 1732
1633BOOT: 1733BOOT:
1662 main_top_env = main_top_env->je_prev; 1762 main_top_env = main_top_env->je_prev;
1663 1763
1664 coroapi.ver = CORO_API_VERSION; 1764 coroapi.ver = CORO_API_VERSION;
1665 coroapi.rev = CORO_API_REVISION; 1765 coroapi.rev = CORO_API_REVISION;
1666 coroapi.transfer = api_transfer; 1766 coroapi.transfer = api_transfer;
1767
1768 {
1769 SV **svp = hv_fetch (PL_modglobal, "Time::NVtime", 12, 0);
1770
1771 if (!svp) croak ("Time::HiRes is required");
1772 if (!SvIOK (*svp)) croak ("Time::NVtime isn't a function pointer");
1773
1774 nvtime = INT2PTR (double (*)(), SvIV (*svp));
1775 }
1667 1776
1668 assert (("PRIO_NORMAL must be 0", !PRIO_NORMAL)); 1777 assert (("PRIO_NORMAL must be 0", !PRIO_NORMAL));
1669} 1778}
1670 1779
1671SV * 1780SV *
1799call (Coro::State coro, SV *coderef) 1908call (Coro::State coro, SV *coderef)
1800 ALIAS: 1909 ALIAS:
1801 eval = 1 1910 eval = 1
1802 CODE: 1911 CODE:
1803{ 1912{
1804 if (coro->mainstack) 1913 if (coro->mainstack && ((coro->flags & CF_RUNNING) || coro->slot))
1805 { 1914 {
1806 struct coro temp; 1915 struct coro temp;
1807 1916
1808 if (!(coro->flags & CF_RUNNING)) 1917 if (!(coro->flags & CF_RUNNING))
1809 { 1918 {
1891force_cctx () 2000force_cctx ()
1892 CODE: 2001 CODE:
1893 struct coro *coro = SvSTATE (coro_current); 2002 struct coro *coro = SvSTATE (coro_current);
1894 coro->cctx->idle_sp = 0; 2003 coro->cctx->idle_sp = 0;
1895 2004
1896MODULE = Coro::State PACKAGE = Coro
1897
1898BOOT:
1899{
1900 int i;
1901
1902 av_async_pool = coro_get_av (aTHX_ "Coro::async_pool", TRUE);
1903 sv_pool_rss = coro_get_sv (aTHX_ "Coro::POOL_RSS" , TRUE);
1904 sv_pool_size = coro_get_sv (aTHX_ "Coro::POOL_SIZE" , TRUE);
1905
1906 coro_current = coro_get_sv (aTHX_ "Coro::current", FALSE);
1907 SvREADONLY_on (coro_current);
1908
1909 coro_stash = gv_stashpv ("Coro", TRUE);
1910
1911 newCONSTSUB (coro_stash, "PRIO_MAX", newSViv (PRIO_MAX));
1912 newCONSTSUB (coro_stash, "PRIO_HIGH", newSViv (PRIO_HIGH));
1913 newCONSTSUB (coro_stash, "PRIO_NORMAL", newSViv (PRIO_NORMAL));
1914 newCONSTSUB (coro_stash, "PRIO_LOW", newSViv (PRIO_LOW));
1915 newCONSTSUB (coro_stash, "PRIO_IDLE", newSViv (PRIO_IDLE));
1916 newCONSTSUB (coro_stash, "PRIO_MIN", newSViv (PRIO_MIN));
1917
1918 for (i = PRIO_MAX - PRIO_MIN + 1; i--; )
1919 coro_ready[i] = newAV ();
1920
1921 {
1922 SV *sv = perl_get_sv ("Coro::API", TRUE);
1923 perl_get_sv ("Coro::API", TRUE); /* silence 5.10 warning */
1924
1925 coroapi.schedule = api_schedule;
1926 coroapi.cede = api_cede;
1927 coroapi.cede_notself = api_cede_notself;
1928 coroapi.ready = api_ready;
1929 coroapi.is_ready = api_is_ready;
1930 coroapi.nready = &coro_nready;
1931 coroapi.current = coro_current;
1932
1933 GCoroAPI = &coroapi;
1934 sv_setiv (sv, (IV)&coroapi);
1935 SvREADONLY_on (sv);
1936 }
1937}
1938
1939void
1940_set_current (SV *current)
1941 PROTOTYPE: $
1942 CODE:
1943 SvREFCNT_dec (SvRV (coro_current));
1944 SvRV_set (coro_current, SvREFCNT_inc (SvRV (current)));
1945
1946void
1947_set_readyhook (SV *hook)
1948 PROTOTYPE: $
1949 CODE:
1950 LOCK;
1951 SvREFCNT_dec (coro_readyhook);
1952 coro_readyhook = SvOK (hook) ? newSVsv (hook) : 0;
1953 UNLOCK;
1954
1955int
1956prio (Coro::State coro, int newprio = 0)
1957 ALIAS:
1958 nice = 1
1959 CODE:
1960{
1961 RETVAL = coro->prio;
1962
1963 if (items > 1)
1964 {
1965 if (ix)
1966 newprio = coro->prio - newprio;
1967
1968 if (newprio < PRIO_MIN) newprio = PRIO_MIN;
1969 if (newprio > PRIO_MAX) newprio = PRIO_MAX;
1970
1971 coro->prio = newprio;
1972 }
1973}
1974 OUTPUT:
1975 RETVAL
1976
1977SV *
1978ready (SV *self)
1979 PROTOTYPE: $
1980 CODE:
1981 RETVAL = boolSV (api_ready (self));
1982 OUTPUT:
1983 RETVAL
1984
1985int
1986nready (...)
1987 PROTOTYPE:
1988 CODE:
1989 RETVAL = coro_nready;
1990 OUTPUT:
1991 RETVAL
1992
1993void 2005void
1994throw (Coro::State self, SV *throw = &PL_sv_undef) 2006throw (Coro::State self, SV *throw = &PL_sv_undef)
1995 PROTOTYPE: $;$ 2007 PROTOTYPE: $;$
1996 CODE: 2008 CODE:
1997 SvREFCNT_dec (self->throw); 2009 SvREFCNT_dec (self->throw);
2011 SV **dst = ix ? (SV **)&self->slot->defav : (SV **)&self->slot->defsv; 2023 SV **dst = ix ? (SV **)&self->slot->defav : (SV **)&self->slot->defsv;
2012 2024
2013 SV *tmp = *src; *src = *dst; *dst = tmp; 2025 SV *tmp = *src; *src = *dst; *dst = tmp;
2014 } 2026 }
2015 2027
2028MODULE = Coro::State PACKAGE = Coro
2029
2030BOOT:
2031{
2032 int i;
2033
2034 av_async_pool = coro_get_av (aTHX_ "Coro::async_pool", TRUE);
2035 sv_pool_rss = coro_get_sv (aTHX_ "Coro::POOL_RSS" , TRUE);
2036 sv_pool_size = coro_get_sv (aTHX_ "Coro::POOL_SIZE" , TRUE);
2037
2038 coro_current = coro_get_sv (aTHX_ "Coro::current", FALSE);
2039 SvREADONLY_on (coro_current);
2040
2041 coro_stash = gv_stashpv ("Coro", TRUE);
2042
2043 newCONSTSUB (coro_stash, "PRIO_MAX", newSViv (PRIO_MAX));
2044 newCONSTSUB (coro_stash, "PRIO_HIGH", newSViv (PRIO_HIGH));
2045 newCONSTSUB (coro_stash, "PRIO_NORMAL", newSViv (PRIO_NORMAL));
2046 newCONSTSUB (coro_stash, "PRIO_LOW", newSViv (PRIO_LOW));
2047 newCONSTSUB (coro_stash, "PRIO_IDLE", newSViv (PRIO_IDLE));
2048 newCONSTSUB (coro_stash, "PRIO_MIN", newSViv (PRIO_MIN));
2049
2050 for (i = PRIO_MAX - PRIO_MIN + 1; i--; )
2051 coro_ready[i] = newAV ();
2052
2053 {
2054 SV *sv = perl_get_sv ("Coro::API", TRUE);
2055 perl_get_sv ("Coro::API", TRUE); /* silence 5.10 warning */
2056
2057 coroapi.schedule = api_schedule;
2058 coroapi.cede = api_cede;
2059 coroapi.cede_notself = api_cede_notself;
2060 coroapi.ready = api_ready;
2061 coroapi.is_ready = api_is_ready;
2062 coroapi.nready = &coro_nready;
2063 coroapi.current = coro_current;
2064
2065 GCoroAPI = &coroapi;
2066 sv_setiv (sv, (IV)&coroapi);
2067 SvREADONLY_on (sv);
2068 }
2069}
2070
2071void
2072_set_current (SV *current)
2073 PROTOTYPE: $
2074 CODE:
2075 SvREFCNT_dec (SvRV (coro_current));
2076 SvRV_set (coro_current, SvREFCNT_inc (SvRV (current)));
2077
2078void
2079_set_readyhook (SV *hook)
2080 PROTOTYPE: $
2081 CODE:
2082 LOCK;
2083 SvREFCNT_dec (coro_readyhook);
2084 coro_readyhook = SvOK (hook) ? newSVsv (hook) : 0;
2085 UNLOCK;
2086
2087int
2088prio (Coro::State coro, int newprio = 0)
2089 ALIAS:
2090 nice = 1
2091 CODE:
2092{
2093 RETVAL = coro->prio;
2094
2095 if (items > 1)
2096 {
2097 if (ix)
2098 newprio = coro->prio - newprio;
2099
2100 if (newprio < PRIO_MIN) newprio = PRIO_MIN;
2101 if (newprio > PRIO_MAX) newprio = PRIO_MAX;
2102
2103 coro->prio = newprio;
2104 }
2105}
2106 OUTPUT:
2107 RETVAL
2108
2109SV *
2110ready (SV *self)
2111 PROTOTYPE: $
2112 CODE:
2113 RETVAL = boolSV (api_ready (self));
2114 OUTPUT:
2115 RETVAL
2116
2117int
2118nready (...)
2119 PROTOTYPE:
2120 CODE:
2121 RETVAL = coro_nready;
2122 OUTPUT:
2123 RETVAL
2124
2016# for async_pool speedup 2125# for async_pool speedup
2017void 2126void
2018_pool_1 (SV *cb) 2127_pool_1 (SV *cb)
2019 CODE: 2128 CODE:
2020{ 2129{
2025 AV *invoke_av; 2134 AV *invoke_av;
2026 int i, len; 2135 int i, len;
2027 2136
2028 if (!invoke) 2137 if (!invoke)
2029 { 2138 {
2030 SvREFCNT_dec (PL_diehook); PL_diehook = 0; 2139 SV *old = PL_diehook;
2140 PL_diehook = 0;
2141 SvREFCNT_dec (old);
2031 croak ("\3async_pool terminate\2\n"); 2142 croak ("\3async_pool terminate\2\n");
2032 } 2143 }
2033 2144
2034 SvREFCNT_dec (coro->saved_deffh); 2145 SvREFCNT_dec (coro->saved_deffh);
2035 coro->saved_deffh = SvREFCNT_inc ((SV *)PL_defoutgv); 2146 coro->saved_deffh = SvREFCNT_inc ((SV *)PL_defoutgv);
2064 coro->saved_deffh = 0; 2175 coro->saved_deffh = 0;
2065 2176
2066 if (coro_rss (aTHX_ coro) > SvIV (sv_pool_rss) 2177 if (coro_rss (aTHX_ coro) > SvIV (sv_pool_rss)
2067 || av_len (av_async_pool) + 1 >= SvIV (sv_pool_size)) 2178 || av_len (av_async_pool) + 1 >= SvIV (sv_pool_size))
2068 { 2179 {
2069 SvREFCNT_dec (PL_diehook); PL_diehook = 0; 2180 SV *old = PL_diehook;
2181 PL_diehook = 0;
2182 SvREFCNT_dec (old);
2070 croak ("\3async_pool terminate\2\n"); 2183 croak ("\3async_pool terminate\2\n");
2071 } 2184 }
2072 2185
2073 av_clear (GvAV (PL_defgv)); 2186 av_clear (GvAV (PL_defgv));
2074 hv_store ((HV *)SvRV (coro_current), "desc", sizeof ("desc") - 1, 2187 hv_store ((HV *)SvRV (coro_current), "desc", sizeof ("desc") - 1,
2079 if (coro->cctx && (coro->cctx->flags & CC_TRACE)) 2192 if (coro->cctx && (coro->cctx->flags & CC_TRACE))
2080 api_trace (coro_current, 0); 2193 api_trace (coro_current, 0);
2081 2194
2082 av_push (av_async_pool, newSVsv (coro_current)); 2195 av_push (av_async_pool, newSVsv (coro_current));
2083} 2196}
2197
2198#if 0
2199
2200void
2201_generator_call (...)
2202 PROTOTYPE: @
2203 PPCODE:
2204 fprintf (stderr, "call %p\n", CvXSUBANY(cv).any_ptr);
2205 xxxx
2206 abort ();
2207
2208SV *
2209gensub (SV *sub, ...)
2210 PROTOTYPE: &;@
2211 CODE:
2212{
2213 struct coro *coro;
2214 MAGIC *mg;
2215 CV *xcv;
2216 CV *ncv = (CV *)newSV_type (SVt_PVCV);
2217 int i;
2218
2219 CvGV (ncv) = CvGV (cv);
2220 CvFILE (ncv) = CvFILE (cv);
2221
2222 Newz (0, coro, 1, struct coro);
2223 coro->args = newAV ();
2224 coro->flags = CF_NEW;
2225
2226 av_extend (coro->args, items - 1);
2227 for (i = 1; i < items; i++)
2228 av_push (coro->args, newSVsv (ST (i)));
2229
2230 CvISXSUB_on (ncv);
2231 CvXSUBANY (ncv).any_ptr = (void *)coro;
2232
2233 xcv = GvCV (gv_fetchpv ("Coro::_generator_call", 0, SVt_PVCV));
2234
2235 CvXSUB (ncv) = CvXSUB (xcv);
2236 CvANON_on (ncv);
2237
2238 mg = sv_magicext ((SV *)ncv, 0, CORO_MAGIC_type_state, &coro_gensub_vtbl, (char *)coro, 0);
2239 RETVAL = newRV_noinc ((SV *)ncv);
2240}
2241 OUTPUT:
2242 RETVAL
2243
2244#endif
2084 2245
2085 2246
2086MODULE = Coro::State PACKAGE = Coro::AIO 2247MODULE = Coro::State PACKAGE = Coro::AIO
2087 2248
2088SV * 2249SV *
2146 } 2307 }
2147 2308
2148 --incede; 2309 --incede;
2149} 2310}
2150 2311
2312
2313MODULE = Coro::State PACKAGE = PerlIO::cede
2314
2315BOOT:
2316 PerlIO_define_layer (aTHX_ &PerlIO_cede);

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines