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.341 by root, Mon Dec 15 15:03:31 2008 UTC vs.
Revision 1.352 by root, Sat Jun 20 08:58:50 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);
1004
992 coro_unwind_stacks (aTHX); 1005 coro_unwind_stacks (aTHX);
1006 coro_destruct_stacks (aTHX);
993 1007
994 SvREFCNT_dec (GvSV (PL_defgv)); 1008 // now save some sv's to be free'd later
995 SvREFCNT_dec (GvAV (PL_defgv)); 1009 svf [0] = GvSV (PL_defgv);
996 SvREFCNT_dec (GvSV (PL_errgv)); 1010 svf [1] = (SV *)GvAV (PL_defgv);
997 SvREFCNT_dec (PL_defoutgv); 1011 svf [2] = GvSV (PL_errgv);
998 SvREFCNT_dec (PL_rs); 1012 svf [3] = (SV *)PL_defoutgv;
999 SvREFCNT_dec (GvSV (irsgv)); 1013 svf [4] = PL_rs;
1000 SvREFCNT_dec (GvHV (PL_hintgv)); 1014 svf [5] = GvSV (irsgv);
1015 svf [6] = (SV *)GvHV (PL_hintgv);
1016 svf [7] = PL_diehook;
1017 svf [8] = PL_warnhook;
1018 assert (9 == sizeof (svf) / sizeof (*svf));
1001 1019
1002 SvREFCNT_dec (PL_diehook); 1020 load_perl (aTHX_ current);
1003 SvREFCNT_dec (PL_warnhook);
1004 1021 }
1022
1023 {
1024 int i;
1025
1026 for (i = 0; i < sizeof (svf) / sizeof (*svf); ++i)
1027 SvREFCNT_dec (svf [i]);
1028
1005 SvREFCNT_dec (coro->saved_deffh); 1029 SvREFCNT_dec (coro->saved_deffh);
1006 SvREFCNT_dec (coro->rouse_cb); 1030 SvREFCNT_dec (coro->rouse_cb);
1007 SvREFCNT_dec (coro->invoke_cb); 1031 SvREFCNT_dec (coro->invoke_cb);
1008 SvREFCNT_dec (coro->invoke_av); 1032 SvREFCNT_dec (coro->invoke_av);
1009 1033 }
1010 coro_destruct_stacks (aTHX);
1011} 1034}
1012 1035
1013INLINE void 1036INLINE void
1014free_coro_mortal (pTHX) 1037free_coro_mortal (pTHX)
1015{ 1038{
… …
1484coro_state_destroy (pTHX_ struct coro *coro) 1507coro_state_destroy (pTHX_ struct coro *coro)
1485{ 1508{
1486 if (coro->flags & CF_DESTROYED) 1509 if (coro->flags & CF_DESTROYED)
1487 return 0; 1510 return 0;
1488 1511
1489 if (coro->on_destroy) 1512 if (coro->on_destroy && !PL_dirty)
1490 coro->on_destroy (aTHX_ coro); 1513 coro->on_destroy (aTHX_ coro);
1491 1514
1492 coro->flags |= CF_DESTROYED; 1515 coro->flags |= CF_DESTROYED;
1493 1516
1494 if (coro->flags & CF_READY) 1517 if (coro->flags & CF_READY)
… …
1502 1525
1503 if (coro->mainstack 1526 if (coro->mainstack
1504 && coro->mainstack != main_mainstack 1527 && coro->mainstack != main_mainstack
1505 && coro->slot 1528 && coro->slot
1506 && !PL_dirty) 1529 && !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); 1530 coro_destruct_perl (aTHX_ coro);
1516
1517 load_perl (aTHX_ current);
1518
1519 coro->slot = 0;
1520 }
1521 1531
1522 cctx_destroy (coro->cctx); 1532 cctx_destroy (coro->cctx);
1523 SvREFCNT_dec (coro->startcv); 1533 SvREFCNT_dec (coro->startcv);
1524 SvREFCNT_dec (coro->args); 1534 SvREFCNT_dec (coro->args);
1525 SvREFCNT_dec (CORO_THROW); 1535 SvREFCNT_dec (CORO_THROW);
… …
1612/** Coro ********************************************************************/ 1622/** Coro ********************************************************************/
1613 1623
1614INLINE void 1624INLINE void
1615coro_enq (pTHX_ struct coro *coro) 1625coro_enq (pTHX_ struct coro *coro)
1616{ 1626{
1617 av_push (coro_ready [coro->prio - PRIO_MIN], SvREFCNT_inc_NN (coro->hv)); 1627 struct coro **ready = coro_ready [coro->prio - PRIO_MIN];
1618}
1619 1628
1620INLINE SV * 1629 SvREFCNT_inc_NN (coro->hv);
1630
1631 coro->next_ready = 0;
1632 *(ready [0] ? &ready [1]->next_ready : &ready [0]) = coro;
1633 ready [1] = coro;
1634}
1635
1636INLINE struct coro *
1621coro_deq (pTHX) 1637coro_deq (pTHX)
1622{ 1638{
1623 int prio; 1639 int prio;
1624 1640
1625 for (prio = PRIO_MAX - PRIO_MIN + 1; --prio >= 0; ) 1641 for (prio = PRIO_MAX - PRIO_MIN + 1; --prio >= 0; )
1626 if (AvFILLp (coro_ready [prio]) >= 0) 1642 {
1627 return av_shift (coro_ready [prio]); 1643 struct coro **ready = coro_ready [prio];
1644
1645 if (ready [0])
1646 {
1647 struct coro *coro = ready [0];
1648 ready [0] = coro->next_ready;
1649 return coro;
1650 }
1651 }
1628 1652
1629 return 0; 1653 return 0;
1630} 1654}
1631 1655
1632static int 1656static int
… …
1696static void 1720static void
1697prepare_schedule (pTHX_ struct coro_transfer_args *ta) 1721prepare_schedule (pTHX_ struct coro_transfer_args *ta)
1698{ 1722{
1699 for (;;) 1723 for (;;)
1700 { 1724 {
1701 SV *next_sv = coro_deq (aTHX); 1725 struct coro *next = coro_deq (aTHX);
1702 1726
1703 if (expect_true (next_sv)) 1727 if (expect_true (next))
1704 { 1728 {
1705 struct coro *next = SvSTATE_hv (next_sv);
1706
1707 /* cannot transfer to destroyed coros, skip and look for next */ 1729 /* cannot transfer to destroyed coros, skip and look for next */
1708 if (expect_false (next->flags & (CF_DESTROYED | CF_SUSPENDED))) 1730 if (expect_false (next->flags & (CF_DESTROYED | CF_SUSPENDED)))
1709 SvREFCNT_dec (next_sv); /* coro_nready has already been taken care of by destroy */ 1731 SvREFCNT_dec (next->hv); /* coro_nready has already been taken care of by destroy */
1710 else 1732 else
1711 { 1733 {
1712 next->flags &= ~CF_READY; 1734 next->flags &= ~CF_READY;
1713 --coro_nready; 1735 --coro_nready;
1714 1736
… …
1988 SV *data = (SV *)GENSUB_ARG; 2010 SV *data = (SV *)GENSUB_ARG;
1989 2011
1990 if (SvTYPE (SvRV (data)) != SVt_PVAV) 2012 if (SvTYPE (SvRV (data)) != SVt_PVAV)
1991 { 2013 {
1992 /* first call, set args */ 2014 /* first call, set args */
2015 SV *coro = SvRV (data);
1993 AV *av = newAV (); 2016 AV *av = newAV ();
1994 SV *coro = SvRV (data);
1995 2017
1996 SvRV_set (data, (SV *)av); 2018 SvRV_set (data, (SV *)av);
1997 api_ready (aTHX_ coro);
1998 SvREFCNT_dec (coro);
1999 2019
2000 /* better take a full copy of the arguments */ 2020 /* better take a full copy of the arguments */
2001 while (items--) 2021 while (items--)
2002 av_store (av, items, newSVsv (ST (items))); 2022 av_store (av, items, newSVsv (ST (items)));
2023
2024 api_ready (aTHX_ coro);
2025 SvREFCNT_dec (coro);
2003 } 2026 }
2004 2027
2005 XSRETURN_EMPTY; 2028 XSRETURN_EMPTY;
2006} 2029}
2007 2030
… …
2024 2047
2025 EXTEND (SP, AvFILLp (av) + 1); 2048 EXTEND (SP, AvFILLp (av) + 1);
2026 for (i = 0; i <= AvFILLp (av); ++i) 2049 for (i = 0; i <= AvFILLp (av); ++i)
2027 PUSHs (sv_2mortal (AvARRAY (av)[i])); 2050 PUSHs (sv_2mortal (AvARRAY (av)[i]));
2028 2051
2029 /* we have stolen the elements, so ste length to zero and free */ 2052 /* we have stolen the elements, so set length to zero and free */
2030 AvFILLp (av) = -1; 2053 AvFILLp (av) = -1;
2031 av_undef (av); 2054 av_undef (av);
2032 2055
2033 PUTBACK; 2056 PUTBACK;
2034 } 2057 }
… …
3098 if (ix) 3121 if (ix)
3099 eval_sv (coderef, 0); 3122 eval_sv (coderef, 0);
3100 else 3123 else
3101 call_sv (coderef, G_KEEPERR | G_EVAL | G_VOID | G_DISCARD); 3124 call_sv (coderef, G_KEEPERR | G_EVAL | G_VOID | G_DISCARD);
3102 3125
3126 POPSTACK;
3103 SPAGAIN; 3127 SPAGAIN;
3104 POPSTACK;
3105 3128
3106 if (current != coro) 3129 if (current != coro)
3107 { 3130 {
3108 PUTBACK; 3131 PUTBACK;
3109 save_perl (aTHX_ coro); 3132 save_perl (aTHX_ coro);
… …
3205 3228
3206MODULE = Coro::State PACKAGE = Coro 3229MODULE = Coro::State PACKAGE = Coro
3207 3230
3208BOOT: 3231BOOT:
3209{ 3232{
3210 int i;
3211
3212 sv_pool_rss = coro_get_sv (aTHX_ "Coro::POOL_RSS" , TRUE); 3233 sv_pool_rss = coro_get_sv (aTHX_ "Coro::POOL_RSS" , TRUE);
3213 sv_pool_size = coro_get_sv (aTHX_ "Coro::POOL_SIZE" , TRUE); 3234 sv_pool_size = coro_get_sv (aTHX_ "Coro::POOL_SIZE" , TRUE);
3214 cv_coro_run = get_cv ( "Coro::_coro_run" , GV_ADD); 3235 cv_coro_run = get_cv ( "Coro::_coro_run" , GV_ADD);
3215 cv_coro_terminate = get_cv ( "Coro::terminate" , GV_ADD); 3236 cv_coro_terminate = get_cv ( "Coro::terminate" , GV_ADD);
3216 coro_current = coro_get_sv (aTHX_ "Coro::current" , FALSE); SvREADONLY_on (coro_current); 3237 coro_current = coro_get_sv (aTHX_ "Coro::current" , FALSE); SvREADONLY_on (coro_current);
… …
3231 newCONSTSUB (coro_stash, "PRIO_NORMAL", newSViv (PRIO_NORMAL)); 3252 newCONSTSUB (coro_stash, "PRIO_NORMAL", newSViv (PRIO_NORMAL));
3232 newCONSTSUB (coro_stash, "PRIO_LOW", newSViv (PRIO_LOW)); 3253 newCONSTSUB (coro_stash, "PRIO_LOW", newSViv (PRIO_LOW));
3233 newCONSTSUB (coro_stash, "PRIO_IDLE", newSViv (PRIO_IDLE)); 3254 newCONSTSUB (coro_stash, "PRIO_IDLE", newSViv (PRIO_IDLE));
3234 newCONSTSUB (coro_stash, "PRIO_MIN", newSViv (PRIO_MIN)); 3255 newCONSTSUB (coro_stash, "PRIO_MIN", newSViv (PRIO_MIN));
3235 3256
3236 for (i = PRIO_MAX - PRIO_MIN + 1; i--; )
3237 coro_ready[i] = newAV ();
3238
3239 { 3257 {
3240 SV *sv = coro_get_sv (aTHX_ "Coro::API", TRUE); 3258 SV *sv = coro_get_sv (aTHX_ "Coro::API", TRUE);
3241 3259
3242 coroapi.schedule = api_schedule; 3260 coroapi.schedule = api_schedule;
3243 coroapi.schedule_to = api_schedule_to; 3261 coroapi.schedule_to = api_schedule_to;
… …
3334 PROTOTYPE: 3352 PROTOTYPE:
3335 CODE: 3353 CODE:
3336 RETVAL = coro_nready; 3354 RETVAL = coro_nready;
3337 OUTPUT: 3355 OUTPUT:
3338 RETVAL 3356 RETVAL
3357
3358void
3359suspend (Coro::State self)
3360 PROTOTYPE: $
3361 CODE:
3362 self->flags |= CF_SUSPENDED;
3363
3364void
3365resume (Coro::State self)
3366 PROTOTYPE: $
3367 CODE:
3368 self->flags &= ~CF_SUSPENDED;
3339 3369
3340void 3370void
3341_pool_handler (...) 3371_pool_handler (...)
3342 CODE: 3372 CODE:
3343 CORO_EXECUTE_SLF_XS (slf_init_pool_handler); 3373 CORO_EXECUTE_SLF_XS (slf_init_pool_handler);
… …
3418 av_push (*avp, SvREFCNT_inc (block)); 3448 av_push (*avp, SvREFCNT_inc (block));
3419 3449
3420 if (!ix) 3450 if (!ix)
3421 on_enterleave_call (aTHX_ block); 3451 on_enterleave_call (aTHX_ block);
3422 3452
3423 LEAVE; /* pp_entersub unfortunately forces an ENTER/LEAVE around xs calls */ 3453 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); 3454 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 */ 3455 ENTER; /* pp_entersub unfortunately forces an ENTER/LEAVE around XS calls */
3426} 3456}
3427 3457
3428 3458
3429MODULE = Coro::State PACKAGE = PerlIO::cede 3459MODULE = Coro::State PACKAGE = PerlIO::cede
3430 3460
… …
3442 GvSTASH (CvGV (cv)) 3472 GvSTASH (CvGV (cv))
3443 ); 3473 );
3444 OUTPUT: 3474 OUTPUT:
3445 RETVAL 3475 RETVAL
3446 3476
3447# helper for Coro::Channel 3477# helper for Coro::Channel and others
3448SV * 3478SV *
3449_alloc (int count) 3479_alloc (int count)
3450 CODE: 3480 CODE:
3451 RETVAL = coro_waitarray_new (aTHX_ count); 3481 RETVAL = coro_waitarray_new (aTHX_ count);
3452 OUTPUT: 3482 OUTPUT:
… …
3510 for (i = 1; i <= wcount; ++i) 3540 for (i = 1; i <= wcount; ++i)
3511 PUSHs (sv_2mortal (newRV_inc (AvARRAY (av)[i]))); 3541 PUSHs (sv_2mortal (newRV_inc (AvARRAY (av)[i])));
3512 } 3542 }
3513} 3543}
3514 3544
3545MODULE = Coro::State PACKAGE = Coro::SemaphoreSet
3546
3547void
3548_may_delete (SV *sem, int count, int extra_refs)
3549 PPCODE:
3550{
3551 AV *av = (AV *)SvRV (sem);
3552
3553 if (SvREFCNT ((SV *)av) == 1 + extra_refs
3554 && AvFILLp (av) == 0 /* no waiters, just count */
3555 && SvIV (AvARRAY (av)[0]) == count)
3556 XSRETURN_YES;
3557
3558 XSRETURN_NO;
3559}
3560
3515MODULE = Coro::State PACKAGE = Coro::Signal 3561MODULE = Coro::State PACKAGE = Coro::Signal
3516 3562
3517SV * 3563SV *
3518new (SV *klass) 3564new (SV *klass)
3519 CODE: 3565 CODE:

Diff Legend

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