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.338 by root, Sun Dec 7 15:33:21 2008 UTC vs.
Revision 1.346 by root, Tue Jun 9 18:56:45 2009 UTC

… …
238enum { 238enum {
239 CF_RUNNING = 0x0001, /* coroutine is running */ 239 CF_RUNNING = 0x0001, /* coroutine is running */
240 CF_READY = 0x0002, /* coroutine is ready */ 240 CF_READY = 0x0002, /* coroutine is ready */
241 CF_NEW = 0x0004, /* has never been switched to */ 241 CF_NEW = 0x0004, /* has never been switched to */
242 CF_DESTROYED = 0x0008, /* coroutine data has been freed */ 242 CF_DESTROYED = 0x0008, /* coroutine data has been freed */
243 CF_SUSPENDED = 0x0010, /* coroutine can't be scheduled */
243}; 244};
244 245
245/* the structure where most of the perl state is stored, overlaid on the cxstack */ 246/* the structure where most of the perl state is stored, overlaid on the cxstack */
246typedef struct 247typedef struct
247{ 248{
… …
285 /* async_pool */ 286 /* async_pool */
286 SV *saved_deffh; 287 SV *saved_deffh;
287 SV *invoke_cb; 288 SV *invoke_cb;
288 AV *invoke_av; 289 AV *invoke_av;
289 290
291 /* on_enter/on_leave */
292 AV *on_enter;
293 AV *on_leave;
294
290 /* linked list */ 295 /* linked list */
291 struct coro *next, *prev; 296 struct coro *next, *prev;
292}; 297};
293 298
294typedef struct coro *Coro__State; 299typedef struct coro *Coro__State;
… …
352INLINE CV * 357INLINE CV *
353coro_sv_2cv (pTHX_ SV *sv) 358coro_sv_2cv (pTHX_ SV *sv)
354{ 359{
355 HV *st; 360 HV *st;
356 GV *gvp; 361 GV *gvp;
357 return sv_2cv (sv, &st, &gvp, 0); 362 CV *cv = sv_2cv (sv, &st, &gvp, 0);
363
364 if (!cv)
365 croak ("code reference expected");
366
367 return cv;
358} 368}
359 369
360/*****************************************************************************/ 370/*****************************************************************************/
361/* magic glue */ 371/* magic glue */
362 372
… …
521} 531}
522 532
523/** load & save, init *******************************************************/ 533/** load & save, init *******************************************************/
524 534
525static void 535static void
536on_enterleave_call (pTHX_ SV *cb);
537
538static void
526load_perl (pTHX_ Coro__State c) 539load_perl (pTHX_ Coro__State c)
527{ 540{
528 perl_slots *slot = c->slot; 541 perl_slots *slot = c->slot;
529 c->slot = 0; 542 c->slot = 0;
530 543
… …
556 PUTBACK; 569 PUTBACK;
557 } 570 }
558 571
559 slf_frame = c->slf_frame; 572 slf_frame = c->slf_frame;
560 CORO_THROW = c->except; 573 CORO_THROW = c->except;
574
575 if (expect_false (c->on_enter))
576 {
577 int i;
578
579 for (i = 0; i <= AvFILLp (c->on_enter); ++i)
580 on_enterleave_call (aTHX_ AvARRAY (c->on_enter)[i]);
581 }
561} 582}
562 583
563static void 584static void
564save_perl (pTHX_ Coro__State c) 585save_perl (pTHX_ Coro__State c)
565{ 586{
587 if (expect_false (c->on_leave))
588 {
589 int i;
590
591 for (i = AvFILLp (c->on_leave); i >= 0; --i)
592 on_enterleave_call (aTHX_ AvARRAY (c->on_leave)[i]);
593 }
594
566 c->except = CORO_THROW; 595 c->except = CORO_THROW;
567 c->slf_frame = slf_frame; 596 c->slf_frame = slf_frame;
568 597
569 { 598 {
570 dSP; 599 dSP;
… …
1354 /* TODO: throwing up here is considered harmful */ 1383 /* TODO: throwing up here is considered harmful */
1355 1384
1356 if (expect_true (prev != next)) 1385 if (expect_true (prev != next))
1357 { 1386 {
1358 if (expect_false (!(prev->flags & (CF_RUNNING | CF_NEW)))) 1387 if (expect_false (!(prev->flags & (CF_RUNNING | CF_NEW))))
1359 croak ("Coro::State::transfer called with a suspended prev Coro::State, but can only transfer from running or new states,"); 1388 croak ("Coro::State::transfer called with a blocked prev Coro::State, but can only transfer from running or new states,");
1360 1389
1361 if (expect_false (next->flags & CF_RUNNING))
1362 croak ("Coro::State::transfer called with running next Coro::State, but can only transfer to inactive states,");
1363
1364 if (expect_false (next->flags & CF_DESTROYED)) 1390 if (expect_false (next->flags & (CF_RUNNING | CF_DESTROYED | CF_SUSPENDED)))
1365 croak ("Coro::State::transfer called with destroyed next Coro::State, but can only transfer to inactive states,"); 1391 croak ("Coro::State::transfer called with running, destroyed or suspended next Coro::State, but can only transfer to inactive states,");
1366 1392
1367#if !PERL_VERSION_ATLEAST (5,10,0) 1393#if !PERL_VERSION_ATLEAST (5,10,0)
1368 if (expect_false (PL_lex_state != LEX_NOTPARSING)) 1394 if (expect_false (PL_lex_state != LEX_NOTPARSING))
1369 croak ("Coro::State::transfer called while parsing, but this is not supported in your perl version,"); 1395 croak ("Coro::State::transfer called while parsing, but this is not supported in your perl version,");
1370#endif 1396#endif
… …
1458coro_state_destroy (pTHX_ struct coro *coro) 1484coro_state_destroy (pTHX_ struct coro *coro)
1459{ 1485{
1460 if (coro->flags & CF_DESTROYED) 1486 if (coro->flags & CF_DESTROYED)
1461 return 0; 1487 return 0;
1462 1488
1463 if (coro->on_destroy) 1489 if (coro->on_destroy && !PL_dirty)
1464 coro->on_destroy (aTHX_ coro); 1490 coro->on_destroy (aTHX_ coro);
1465 1491
1466 coro->flags |= CF_DESTROYED; 1492 coro->flags |= CF_DESTROYED;
1467 1493
1468 if (coro->flags & CF_READY) 1494 if (coro->flags & CF_READY)
… …
1477 if (coro->mainstack 1503 if (coro->mainstack
1478 && coro->mainstack != main_mainstack 1504 && coro->mainstack != main_mainstack
1479 && coro->slot 1505 && coro->slot
1480 && !PL_dirty) 1506 && !PL_dirty)
1481 { 1507 {
1482 struct coro temp; 1508 struct coro *current = SvSTATE_current;
1483 1509
1484 assert (("FATAL: tried to destroy currently running coroutine", coro->mainstack != PL_mainstack)); 1510 assert (("FATAL: tried to destroy currently running coroutine", coro->mainstack != PL_mainstack));
1485 1511
1486 save_perl (aTHX_ &temp); 1512 save_perl (aTHX_ current);
1487 load_perl (aTHX_ coro); 1513 load_perl (aTHX_ coro);
1488 1514
1489 coro_destruct_perl (aTHX_ coro); 1515 coro_destruct_perl (aTHX_ coro);
1490 1516
1491 load_perl (aTHX_ &temp); 1517 load_perl (aTHX_ current);
1492 1518
1493 coro->slot = 0; 1519 coro->slot = 0;
1494 } 1520 }
1495 1521
1496 cctx_destroy (coro->cctx); 1522 cctx_destroy (coro->cctx);
… …
1677 if (expect_true (next_sv)) 1703 if (expect_true (next_sv))
1678 { 1704 {
1679 struct coro *next = SvSTATE_hv (next_sv); 1705 struct coro *next = SvSTATE_hv (next_sv);
1680 1706
1681 /* cannot transfer to destroyed coros, skip and look for next */ 1707 /* cannot transfer to destroyed coros, skip and look for next */
1682 if (expect_false (next->flags & CF_DESTROYED)) 1708 if (expect_false (next->flags & (CF_DESTROYED | CF_SUSPENDED)))
1683 SvREFCNT_dec (next_sv); /* coro_nready has already been taken care of by destroy */ 1709 SvREFCNT_dec (next_sv); /* coro_nready has already been taken care of by destroy */
1684 else 1710 else
1685 { 1711 {
1686 next->flags &= ~CF_READY; 1712 next->flags &= ~CF_READY;
1687 --coro_nready; 1713 --coro_nready;
… …
1962 SV *data = (SV *)GENSUB_ARG; 1988 SV *data = (SV *)GENSUB_ARG;
1963 1989
1964 if (SvTYPE (SvRV (data)) != SVt_PVAV) 1990 if (SvTYPE (SvRV (data)) != SVt_PVAV)
1965 { 1991 {
1966 /* first call, set args */ 1992 /* first call, set args */
1993 SV *coro = SvRV (data);
1967 AV *av = newAV (); 1994 AV *av = newAV ();
1968 SV *coro = SvRV (data);
1969 1995
1970 SvRV_set (data, (SV *)av); 1996 SvRV_set (data, (SV *)av);
1971 api_ready (aTHX_ coro);
1972 SvREFCNT_dec (coro);
1973 1997
1974 /* better take a full copy of the arguments */ 1998 /* better take a full copy of the arguments */
1975 while (items--) 1999 while (items--)
1976 av_store (av, items, newSVsv (ST (items))); 2000 av_store (av, items, newSVsv (ST (items)));
2001
2002 api_ready (aTHX_ coro);
2003 SvREFCNT_dec (coro);
1977 } 2004 }
1978 2005
1979 XSRETURN_EMPTY; 2006 XSRETURN_EMPTY;
1980} 2007}
1981 2008
… …
1998 2025
1999 EXTEND (SP, AvFILLp (av) + 1); 2026 EXTEND (SP, AvFILLp (av) + 1);
2000 for (i = 0; i <= AvFILLp (av); ++i) 2027 for (i = 0; i <= AvFILLp (av); ++i)
2001 PUSHs (sv_2mortal (AvARRAY (av)[i])); 2028 PUSHs (sv_2mortal (AvARRAY (av)[i]));
2002 2029
2003 /* we have stolen the elements, so ste length to zero and free */ 2030 /* we have stolen the elements, so set length to zero and free */
2004 AvFILLp (av) = -1; 2031 AvFILLp (av) = -1;
2005 av_undef (av); 2032 av_undef (av);
2006 2033
2007 PUTBACK; 2034 PUTBACK;
2008 } 2035 }
… …
2298 2325
2299 PL_op->op_ppaddr = pp_slf; 2326 PL_op->op_ppaddr = pp_slf;
2300 /*PL_op->op_type = OP_CUSTOM; /* we do behave like entersub still */ 2327 /*PL_op->op_type = OP_CUSTOM; /* we do behave like entersub still */
2301 2328
2302 PL_op = (OP *)&slf_restore; 2329 PL_op = (OP *)&slf_restore;
2330}
2331
2332/*****************************************************************************/
2333/* dynamic wind */
2334
2335static void
2336on_enterleave_call (pTHX_ SV *cb)
2337{
2338 dSP;
2339
2340 PUSHSTACK;
2341
2342 PUSHMARK (SP);
2343 PUTBACK;
2344 call_sv (cb, G_VOID | G_DISCARD);
2345 SPAGAIN;
2346
2347 POPSTACK;
2348}
2349
2350static SV *
2351coro_avp_pop_and_free (pTHX_ AV **avp)
2352{
2353 AV *av = *avp;
2354 SV *res = av_pop (av);
2355
2356 if (AvFILLp (av) < 0)
2357 {
2358 *avp = 0;
2359 SvREFCNT_dec (av);
2360 }
2361
2362 return res;
2363}
2364
2365static void
2366coro_pop_on_enter (pTHX_ void *coro)
2367{
2368 SV *cb = coro_avp_pop_and_free (aTHX_ &((struct coro *)coro)->on_enter);
2369 SvREFCNT_dec (cb);
2370}
2371
2372static void
2373coro_pop_on_leave (pTHX_ void *coro)
2374{
2375 SV *cb = coro_avp_pop_and_free (aTHX_ &((struct coro *)coro)->on_leave);
2376 on_enterleave_call (aTHX_ sv_2mortal (cb));
2303} 2377}
2304 2378
2305/*****************************************************************************/ 2379/*****************************************************************************/
2306/* PerlIO::cede */ 2380/* PerlIO::cede */
2307 2381
… …
3005 eval = 1 3079 eval = 1
3006 CODE: 3080 CODE:
3007{ 3081{
3008 if (coro->mainstack && ((coro->flags & CF_RUNNING) || coro->slot)) 3082 if (coro->mainstack && ((coro->flags & CF_RUNNING) || coro->slot))
3009 { 3083 {
3010 struct coro temp; 3084 struct coro *current = SvSTATE_current;
3011 3085
3012 if (!(coro->flags & CF_RUNNING)) 3086 if (current != coro)
3013 { 3087 {
3014 PUTBACK; 3088 PUTBACK;
3015 save_perl (aTHX_ &temp); 3089 save_perl (aTHX_ current);
3016 load_perl (aTHX_ coro); 3090 load_perl (aTHX_ coro);
3091 SPAGAIN;
3017 } 3092 }
3018 3093
3019 {
3020 dSP;
3021 ENTER;
3022 SAVETMPS;
3023 PUTBACK;
3024 PUSHSTACK; 3094 PUSHSTACK;
3095
3025 PUSHMARK (SP); 3096 PUSHMARK (SP);
3097 PUTBACK;
3026 3098
3027 if (ix) 3099 if (ix)
3028 eval_sv (coderef, 0); 3100 eval_sv (coderef, 0);
3029 else 3101 else
3030 call_sv (coderef, G_KEEPERR | G_EVAL | G_VOID | G_DISCARD); 3102 call_sv (coderef, G_KEEPERR | G_EVAL | G_VOID | G_DISCARD);
3031 3103
3032 POPSTACK; 3104 POPSTACK;
3033 SPAGAIN; 3105 SPAGAIN;
3034 FREETMPS;
3035 LEAVE;
3036 PUTBACK;
3037 }
3038 3106
3039 if (!(coro->flags & CF_RUNNING)) 3107 if (current != coro)
3040 { 3108 {
3109 PUTBACK;
3041 save_perl (aTHX_ coro); 3110 save_perl (aTHX_ coro);
3042 load_perl (aTHX_ &temp); 3111 load_perl (aTHX_ current);
3043 SPAGAIN; 3112 SPAGAIN;
3044 } 3113 }
3045 } 3114 }
3046} 3115}
3047 3116
… …
3051 ALIAS: 3120 ALIAS:
3052 is_ready = CF_READY 3121 is_ready = CF_READY
3053 is_running = CF_RUNNING 3122 is_running = CF_RUNNING
3054 is_new = CF_NEW 3123 is_new = CF_NEW
3055 is_destroyed = CF_DESTROYED 3124 is_destroyed = CF_DESTROYED
3125 is_suspended = CF_SUSPENDED
3056 CODE: 3126 CODE:
3057 RETVAL = boolSV (coro->flags & ix); 3127 RETVAL = boolSV (coro->flags & ix);
3058 OUTPUT: 3128 OUTPUT:
3059 RETVAL 3129 RETVAL
3060 3130
… …
3125 SV **dst = ix ? (SV **)&self->slot->defav : (SV **)&self->slot->defsv; 3195 SV **dst = ix ? (SV **)&self->slot->defav : (SV **)&self->slot->defsv;
3126 3196
3127 SV *tmp = *src; *src = *dst; *dst = tmp; 3197 SV *tmp = *src; *src = *dst; *dst = tmp;
3128 } 3198 }
3129 3199
3200void
3201cancel (Coro::State self)
3202 CODE:
3203 coro_state_destroy (aTHX_ self);
3204 coro_call_on_destroy (aTHX_ self); /* actually only for Coro objects */
3205
3130 3206
3131MODULE = Coro::State PACKAGE = Coro 3207MODULE = Coro::State PACKAGE = Coro
3132 3208
3133BOOT: 3209BOOT:
3134{ 3210{
… …
3208cede_notself (...) 3284cede_notself (...)
3209 CODE: 3285 CODE:
3210 CORO_EXECUTE_SLF_XS (slf_init_cede_notself); 3286 CORO_EXECUTE_SLF_XS (slf_init_cede_notself);
3211 3287
3212void 3288void
3213_cancel (Coro::State self)
3214 CODE:
3215 coro_state_destroy (aTHX_ self);
3216 coro_call_on_destroy (aTHX_ self);
3217
3218void
3219_set_current (SV *current) 3289_set_current (SV *current)
3220 PROTOTYPE: $ 3290 PROTOTYPE: $
3221 CODE: 3291 CODE:
3222 SvREFCNT_dec (SvRV (coro_current)); 3292 SvREFCNT_dec (SvRV (coro_current));
3223 SvRV_set (coro_current, SvREFCNT_inc_NN (SvRV (current))); 3293 SvRV_set (coro_current, SvREFCNT_inc_NN (SvRV (current)));
… …
3265 PROTOTYPE: 3335 PROTOTYPE:
3266 CODE: 3336 CODE:
3267 RETVAL = coro_nready; 3337 RETVAL = coro_nready;
3268 OUTPUT: 3338 OUTPUT:
3269 RETVAL 3339 RETVAL
3340
3341void
3342suspend (Coro::State self)
3343 PROTOTYPE: $
3344 CODE:
3345 self->flags |= CF_SUSPENDED;
3346
3347void
3348resume (Coro::State self)
3349 PROTOTYPE: $
3350 CODE:
3351 self->flags &= ~CF_SUSPENDED;
3270 3352
3271void 3353void
3272_pool_handler (...) 3354_pool_handler (...)
3273 CODE: 3355 CODE:
3274 CORO_EXECUTE_SLF_XS (slf_init_pool_handler); 3356 CORO_EXECUTE_SLF_XS (slf_init_pool_handler);
… …
3329rouse_wait (...) 3411rouse_wait (...)
3330 PROTOTYPE: ;$ 3412 PROTOTYPE: ;$
3331 PPCODE: 3413 PPCODE:
3332 CORO_EXECUTE_SLF_XS (slf_init_rouse_wait); 3414 CORO_EXECUTE_SLF_XS (slf_init_rouse_wait);
3333 3415
3416void
3417on_enter (SV *block)
3418 ALIAS:
3419 on_leave = 1
3420 PROTOTYPE: &
3421 CODE:
3422{
3423 struct coro *coro = SvSTATE_current;
3424 AV **avp = ix ? &coro->on_leave : &coro->on_enter;
3425
3426 block = (SV *)coro_sv_2cv (aTHX_ block);
3427
3428 if (!*avp)
3429 *avp = newAV ();
3430
3431 av_push (*avp, SvREFCNT_inc (block));
3432
3433 if (!ix)
3434 on_enterleave_call (aTHX_ block);
3435
3436 LEAVE; /* pp_entersub unfortunately forces an ENTER/LEAVE around XS calls */
3437 SAVEDESTRUCTOR_X (ix ? coro_pop_on_leave : coro_pop_on_enter, (void *)coro);
3438 ENTER; /* pp_entersub unfortunately forces an ENTER/LEAVE around XS calls */
3439}
3440
3334 3441
3335MODULE = Coro::State PACKAGE = PerlIO::cede 3442MODULE = Coro::State PACKAGE = PerlIO::cede
3336 3443
3337BOOT: 3444BOOT:
3338 PerlIO_define_layer (aTHX_ &PerlIO_cede); 3445 PerlIO_define_layer (aTHX_ &PerlIO_cede);
… …
3348 GvSTASH (CvGV (cv)) 3455 GvSTASH (CvGV (cv))
3349 ); 3456 );
3350 OUTPUT: 3457 OUTPUT:
3351 RETVAL 3458 RETVAL
3352 3459
3353# helper for Coro::Channel 3460# helper for Coro::Channel and others
3354SV * 3461SV *
3355_alloc (int count) 3462_alloc (int count)
3356 CODE: 3463 CODE:
3357 RETVAL = coro_waitarray_new (aTHX_ count); 3464 RETVAL = coro_waitarray_new (aTHX_ count);
3358 OUTPUT: 3465 OUTPUT:
… …
3416 for (i = 1; i <= wcount; ++i) 3523 for (i = 1; i <= wcount; ++i)
3417 PUSHs (sv_2mortal (newRV_inc (AvARRAY (av)[i]))); 3524 PUSHs (sv_2mortal (newRV_inc (AvARRAY (av)[i])));
3418 } 3525 }
3419} 3526}
3420 3527
3528MODULE = Coro::State PACKAGE = Coro::SemaphoreSet
3529
3530void
3531_may_delete (SV *sem, int count, int extra_refs)
3532 PPCODE:
3533{
3534 AV *av = (AV *)SvRV (sem);
3535
3536 if (SvREFCNT ((SV *)av) == 1 + extra_refs
3537 && AvFILLp (av) == 0 /* no waiters, just count */
3538 && SvIV (AvARRAY (av)[0]) == count)
3539 XSRETURN_YES;
3540
3541 XSRETURN_NO;
3542}
3543
3421MODULE = Coro::State PACKAGE = Coro::Signal 3544MODULE = Coro::State PACKAGE = Coro::Signal
3422 3545
3423SV * 3546SV *
3424new (SV *klass) 3547new (SV *klass)
3425 CODE: 3548 CODE:

Diff Legend

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