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.229 by root, Thu Apr 10 07:45:33 2008 UTC vs.
Revision 1.241 by root, Mon Jul 21 02:39:15 2008 UTC

10#include "patchlevel.h" 10#include "patchlevel.h"
11 11
12#include <stdio.h> 12#include <stdio.h>
13#include <errno.h> 13#include <errno.h>
14#include <assert.h> 14#include <assert.h>
15
16#ifdef WIN32
17# undef setjmp
18# undef longjmp
19# undef _exit
20# define setjmp _setjmp // deep magic, don't ask
21#else
15#include <inttypes.h> /* portable stdint.h */ 22# include <inttypes.h> /* most portable stdint.h */
23#endif
16 24
17#ifdef HAVE_MMAP 25#ifdef HAVE_MMAP
18# include <unistd.h> 26# include <unistd.h>
19# include <sys/mman.h> 27# include <sys/mman.h>
20# ifndef MAP_ANONYMOUS 28# ifndef MAP_ANONYMOUS
163/* async_pool helper stuff */ 171/* async_pool helper stuff */
164static SV *sv_pool_rss; 172static SV *sv_pool_rss;
165static SV *sv_pool_size; 173static SV *sv_pool_size;
166static AV *av_async_pool; 174static AV *av_async_pool;
167 175
176/* Coro::AnyEvent */
177static SV *sv_activity;
178
168static struct coro_cctx *cctx_first; 179static struct coro_cctx *cctx_first;
169static int cctx_count, cctx_idle; 180static int cctx_count, cctx_idle;
170 181
171enum { 182enum {
172 CC_MAPPED = 0x01, 183 CC_MAPPED = 0x01,
257#define PRIO_IDLE -3 268#define PRIO_IDLE -3
258#define PRIO_MIN -4 269#define PRIO_MIN -4
259 270
260/* for Coro.pm */ 271/* for Coro.pm */
261static SV *coro_current; 272static SV *coro_current;
273static SV *coro_readyhook;
262static AV *coro_ready [PRIO_MAX-PRIO_MIN+1]; 274static AV *coro_ready [PRIO_MAX-PRIO_MIN+1];
263static int coro_nready; 275static int coro_nready;
264static struct coro *coro_first; 276static struct coro *coro_first;
265 277
266/** lowlevel stuff **********************************************************/ 278/** lowlevel stuff **********************************************************/
502 514
503 if (expect_true (CvDEPTH (cv))) 515 if (expect_true (CvDEPTH (cv)))
504 { 516 {
505 EXTEND (SP, 3); 517 EXTEND (SP, 3);
506 PUSHs ((SV *)CvPADLIST (cv)); 518 PUSHs ((SV *)CvPADLIST (cv));
507 PUSHs (INT2PTR (SV *, CvDEPTH (cv))); 519 PUSHs (INT2PTR (SV *, (IV)CvDEPTH (cv)));
508 PUSHs ((SV *)cv); 520 PUSHs ((SV *)cv);
509 521
510 CvDEPTH (cv) = 0; 522 CvDEPTH (cv) = 0;
511 get_padlist (aTHX_ cv); 523 get_padlist (aTHX_ cv);
512 } 524 }
671 683
672/** coroutine stack handling ************************************************/ 684/** coroutine stack handling ************************************************/
673 685
674static int (*orig_sigelem_get) (pTHX_ SV *sv, MAGIC *mg); 686static int (*orig_sigelem_get) (pTHX_ SV *sv, MAGIC *mg);
675static 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
676 696
677/* 697/*
678 * This overrides the default magic get method of %SIG elements. 698 * This overrides the default magic get method of %SIG elements.
679 * 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
680 * 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
688{ 708{
689 const char *s = MgPV_nolen_const (mg); 709 const char *s = MgPV_nolen_const (mg);
690 710
691 if (*s == '_') 711 if (*s == '_')
692 { 712 {
693 if (strEQ (s, "__DIE__" ) && PL_diehook ) return sv_setsv (sv, PL_diehook ), 0; 713 SV **svp = 0;
694 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 }
695 } 723 }
696 724
697 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;
698} 750}
699 751
700static int 752static int
701coro_sigelem_set (pTHX_ SV *sv, MAGIC *mg) 753coro_sigelem_set (pTHX_ SV *sv, MAGIC *mg)
702{ 754{
712 if (svp) 764 if (svp)
713 { 765 {
714 SV *old = *svp; 766 SV *old = *svp;
715 *svp = newSVsv (sv); 767 *svp = newSVsv (sv);
716 SvREFCNT_dec (old); 768 SvREFCNT_dec (old);
717 return; 769 return 0;
718 } 770 }
719 } 771 }
720 772
721 return orig_sigelem_set ? orig_sigelem_set (aTHX_ sv, mg) : 0; 773 return orig_sigelem_set ? orig_sigelem_set (aTHX_ sv, mg) : 0;
722} 774}
1130 1182
1131 if (expect_false (next->flags & CF_DESTROYED)) 1183 if (expect_false (next->flags & CF_DESTROYED))
1132 croak ("Coro::State::transfer called with destroyed next Coro::State, but can only transfer to inactive states"); 1184 croak ("Coro::State::transfer called with destroyed next Coro::State, but can only transfer to inactive states");
1133 1185
1134#if !PERL_VERSION_ATLEAST (5,10,0) 1186#if !PERL_VERSION_ATLEAST (5,10,0)
1135 if (expect_false (PL_lex_state != LEX_NOTPARSING) 1187 if (expect_false (PL_lex_state != LEX_NOTPARSING))
1136 croak ("Coro::State::transfer called while parsing, but this is not supported in your perl version"); 1188 croak ("Coro::State::transfer called while parsing, but this is not supported in your perl version");
1137#endif 1189#endif
1138 } 1190 }
1139} 1191}
1140 1192
1371static int 1423static int
1372api_ready (SV *coro_sv) 1424api_ready (SV *coro_sv)
1373{ 1425{
1374 dTHX; 1426 dTHX;
1375 struct coro *coro; 1427 struct coro *coro;
1428 SV *sv_hook;
1429 void (*xs_hook)(void);
1376 1430
1377 if (SvROK (coro_sv)) 1431 if (SvROK (coro_sv))
1378 coro_sv = SvRV (coro_sv); 1432 coro_sv = SvRV (coro_sv);
1379 1433
1380 coro = SvSTATE (coro_sv); 1434 coro = SvSTATE (coro_sv);
1383 return 0; 1437 return 0;
1384 1438
1385 coro->flags |= CF_READY; 1439 coro->flags |= CF_READY;
1386 1440
1387 LOCK; 1441 LOCK;
1442
1443 sv_hook = coro_nready ? 0 : coro_readyhook;
1444 xs_hook = coro_nready ? 0 : coroapi.readyhook;
1445
1388 coro_enq (aTHX_ SvREFCNT_inc (coro_sv)); 1446 coro_enq (aTHX_ SvREFCNT_inc (coro_sv));
1389 ++coro_nready; 1447 ++coro_nready;
1448
1390 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 ();
1391 1469
1392 return 1; 1470 return 1;
1393} 1471}
1394 1472
1395static int 1473static int
1559 BOOT_PAGESIZE; 1637 BOOT_PAGESIZE;
1560 1638
1561 irsgv = gv_fetchpv ("/" , GV_ADD|GV_NOTQUAL, SVt_PV); 1639 irsgv = gv_fetchpv ("/" , GV_ADD|GV_NOTQUAL, SVt_PV);
1562 stdoutgv = gv_fetchpv ("STDOUT", GV_ADD|GV_NOTQUAL, SVt_PVIO); 1640 stdoutgv = gv_fetchpv ("STDOUT", GV_ADD|GV_NOTQUAL, SVt_PVIO);
1563 1641
1564 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;
1565 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;
1566 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;
1567 PL_vtbl_sigelem.svt_set = coro_sigelem_set;
1568 1645
1569 hv_sig = coro_get_hv (aTHX_ "SIG", TRUE); 1646 hv_sig = coro_get_hv (aTHX_ "SIG", TRUE);
1570 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));
1571 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));
1572 1649
1581 main_top_env = PL_top_env; 1658 main_top_env = PL_top_env;
1582 1659
1583 while (main_top_env->je_prev) 1660 while (main_top_env->je_prev)
1584 main_top_env = main_top_env->je_prev; 1661 main_top_env = main_top_env->je_prev;
1585 1662
1586 coroapi.ver = CORO_API_VERSION; 1663 coroapi.ver = CORO_API_VERSION;
1587 coroapi.rev = CORO_API_REVISION; 1664 coroapi.rev = CORO_API_REVISION;
1588 coroapi.transfer = api_transfer; 1665 coroapi.transfer = api_transfer;
1589 1666
1590 assert (("PRIO_NORMAL must be 0", !PRIO_NORMAL)); 1667 assert (("PRIO_NORMAL must be 0", !PRIO_NORMAL));
1591} 1668}
1592 1669
1593SV * 1670SV *
1677 RETVAL = coro_state_destroy (aTHX_ SvSTATE (coro_sv)); 1754 RETVAL = coro_state_destroy (aTHX_ SvSTATE (coro_sv));
1678 OUTPUT: 1755 OUTPUT:
1679 RETVAL 1756 RETVAL
1680 1757
1681void 1758void
1682_exit (code) 1759_exit (int code)
1683 int code
1684 PROTOTYPE: $ 1760 PROTOTYPE: $
1685 CODE: 1761 CODE:
1686 _exit (code); 1762 _exit (code);
1687 1763
1688int 1764int
1779 1855
1780void 1856void
1781api_trace (SV *coro, int flags = CC_TRACE | CC_TRACE_SUB) 1857api_trace (SV *coro, int flags = CC_TRACE | CC_TRACE_SUB)
1782 1858
1783SV * 1859SV *
1784has_stack (Coro::State coro) 1860has_cctx (Coro::State coro)
1785 PROTOTYPE: $ 1861 PROTOTYPE: $
1786 CODE: 1862 CODE:
1787 RETVAL = boolSV (!!coro->cctx); 1863 RETVAL = boolSV (!!coro->cctx);
1788 OUTPUT: 1864 OUTPUT:
1789 RETVAL 1865 RETVAL
1820 1896
1821BOOT: 1897BOOT:
1822{ 1898{
1823 int i; 1899 int i;
1824 1900
1901 av_async_pool = coro_get_av (aTHX_ "Coro::async_pool", TRUE);
1825 sv_pool_rss = coro_get_sv (aTHX_ "Coro::POOL_RSS" , TRUE); 1902 sv_pool_rss = coro_get_sv (aTHX_ "Coro::POOL_RSS" , TRUE);
1826 sv_pool_size = coro_get_sv (aTHX_ "Coro::POOL_SIZE" , TRUE); 1903 sv_pool_size = coro_get_sv (aTHX_ "Coro::POOL_SIZE" , TRUE);
1827 av_async_pool = coro_get_av (aTHX_ "Coro::async_pool", TRUE);
1828 1904
1829 coro_current = coro_get_sv (aTHX_ "Coro::current", FALSE); 1905 coro_current = coro_get_sv (aTHX_ "Coro::current", FALSE);
1830 SvREADONLY_on (coro_current); 1906 SvREADONLY_on (coro_current);
1831 1907
1832 coro_stash = gv_stashpv ("Coro", TRUE); 1908 coro_stash = gv_stashpv ("Coro", TRUE);
1863_set_current (SV *current) 1939_set_current (SV *current)
1864 PROTOTYPE: $ 1940 PROTOTYPE: $
1865 CODE: 1941 CODE:
1866 SvREFCNT_dec (SvRV (coro_current)); 1942 SvREFCNT_dec (SvRV (coro_current));
1867 SvRV_set (coro_current, SvREFCNT_inc (SvRV (current))); 1943 SvRV_set (coro_current, SvREFCNT_inc (SvRV (current)));
1944
1945void
1946_set_readyhook (SV *hook)
1947 PROTOTYPE: $
1948 CODE:
1949 LOCK;
1950 SvREFCNT_dec (coro_readyhook);
1951 coro_readyhook = SvOK (hook) ? newSVsv (hook) : 0;
1952 UNLOCK;
1868 1953
1869int 1954int
1870prio (Coro::State coro, int newprio = 0) 1955prio (Coro::State coro, int newprio = 0)
1871 ALIAS: 1956 ALIAS:
1872 nice = 1 1957 nice = 1
1939 AV *invoke_av; 2024 AV *invoke_av;
1940 int i, len; 2025 int i, len;
1941 2026
1942 if (!invoke) 2027 if (!invoke)
1943 { 2028 {
1944 SvREFCNT_dec (PL_diehook); PL_diehook = 0; 2029 SV *old = PL_diehook;
2030 PL_diehook = 0;
2031 SvREFCNT_dec (old);
1945 croak ("\3async_pool terminate\2\n"); 2032 croak ("\3async_pool terminate\2\n");
1946 } 2033 }
1947 2034
1948 SvREFCNT_dec (coro->saved_deffh); 2035 SvREFCNT_dec (coro->saved_deffh);
1949 coro->saved_deffh = SvREFCNT_inc ((SV *)PL_defoutgv); 2036 coro->saved_deffh = SvREFCNT_inc ((SV *)PL_defoutgv);
1978 coro->saved_deffh = 0; 2065 coro->saved_deffh = 0;
1979 2066
1980 if (coro_rss (aTHX_ coro) > SvIV (sv_pool_rss) 2067 if (coro_rss (aTHX_ coro) > SvIV (sv_pool_rss)
1981 || av_len (av_async_pool) + 1 >= SvIV (sv_pool_size)) 2068 || av_len (av_async_pool) + 1 >= SvIV (sv_pool_size))
1982 { 2069 {
1983 SvREFCNT_dec (PL_diehook); PL_diehook = 0; 2070 SV *old = PL_diehook;
2071 PL_diehook = 0;
2072 SvREFCNT_dec (old);
1984 croak ("\3async_pool terminate\2\n"); 2073 croak ("\3async_pool terminate\2\n");
1985 } 2074 }
1986 2075
1987 av_clear (GvAV (PL_defgv)); 2076 av_clear (GvAV (PL_defgv));
1988 hv_store ((HV *)SvRV (coro_current), "desc", sizeof ("desc") - 1, 2077 hv_store ((HV *)SvRV (coro_current), "desc", sizeof ("desc") - 1,
2029 PL_laststype = data->laststype; 2118 PL_laststype = data->laststype;
2030 PL_laststatval = data->laststatval; 2119 PL_laststatval = data->laststatval;
2031 PL_statcache = data->statcache; 2120 PL_statcache = data->statcache;
2032} 2121}
2033 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