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.343 by root, Mon Dec 15 16:41:43 2008 UTC vs.
Revision 1.351 by root, Sat Jun 20 08:58:25 2009 UTC

… …
261/* this is a structure representing a perl-level coroutine */ 261/* this is a structure representing a perl-level coroutine */
262struct coro { 262struct 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 */
317static SV *coro_current; 320static SV *coro_current;
318static SV *coro_readyhook; 321static SV *coro_readyhook;
319static AV *coro_ready [PRIO_MAX - PRIO_MIN + 1]; 322static struct coro *coro_ready [PRIO_MAX - PRIO_MIN + 1][2]; /* head|tail */
320static CV *cv_coro_run, *cv_coro_terminate; 323static CV *cv_coro_run, *cv_coro_terminate;
321static struct coro *coro_first; 324static 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
989static void 992static void
990coro_destruct_perl (pTHX_ struct coro *coro) 993coro_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
1013INLINE void 1035INLINE void
1014free_coro_mortal (pTHX) 1036free_coro_mortal (pTHX)
1015{ 1037{
… …
1484coro_state_destroy (pTHX_ struct coro *coro) 1506coro_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
1614INLINE void 1623INLINE void
1615coro_enq (pTHX_ struct coro *coro) 1624coro_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
1620INLINE 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
1635INLINE struct coro *
1621coro_deq (pTHX) 1636coro_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
1632static int 1655static int
… …
1696static void 1719static void
1697prepare_schedule (pTHX_ struct coro_transfer_args *ta) 1720prepare_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 }
… …
3205 3227
3206MODULE = Coro::State PACKAGE = Coro 3228MODULE = Coro::State PACKAGE = Coro
3207 3229
3208BOOT: 3230BOOT:
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
3357void
3358suspend (Coro::State self)
3359 PROTOTYPE: $
3360 CODE:
3361 self->flags |= CF_SUSPENDED;
3362
3363void
3364resume (Coro::State self)
3365 PROTOTYPE: $
3366 CODE:
3367 self->flags &= ~CF_SUSPENDED;
3339 3368
3340void 3369void
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
3429MODULE = Coro::State PACKAGE = PerlIO::cede 3458MODULE = Coro::State PACKAGE = PerlIO::cede
3430 3459
… …
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
3544MODULE = Coro::State PACKAGE = Coro::SemaphoreSet
3545
3546void
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
3515MODULE = Coro::State PACKAGE = Coro::Signal 3560MODULE = Coro::State PACKAGE = Coro::Signal
3516 3561
3517SV * 3562SV *
3518new (SV *klass) 3563new (SV *klass)
3519 CODE: 3564 CODE:

Diff Legend

– Removed lines
+ Added lines
< Changed lines
> Changed lines