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.125 by root, Mon Dec 11 20:54:45 2006 UTC vs.
Revision 1.139 by root, Sun Jan 14 21:13:41 2007 UTC

4#include "perl.h" 4#include "perl.h"
5#include "XSUB.h" 5#include "XSUB.h"
6 6
7#include "patchlevel.h" 7#include "patchlevel.h"
8 8
9#if USE_VALGRIND
10# include <valgrind/valgrind.h>
11#endif
12
13/* the maximum number of idle cctx that will be pooled */
14#define MAX_IDLE_CCTX 8
15
16#define PERL_VERSION_ATLEAST(a,b,c) \
17 (PERL_REVISION > (a) \
18 || (PERL_REVISION == (a) \
19 && (PERL_VERSION > (b) \
20 || (PERL_VERSION == (b) && PERLSUBVERSION >= (c)))))
21
22#if !PERL_VERSION_ATLEAST (5,6,0)
23# ifndef PL_ppaddr
24# define PL_ppaddr ppaddr
25# endif
26# ifndef call_sv
27# define call_sv perl_call_sv
28# endif
29# ifndef get_sv
30# define get_sv perl_get_sv
31# endif
32# ifndef get_cv
33# define get_cv perl_get_cv
34# endif
35# ifndef IS_PADGV
36# define IS_PADGV(v) 0
37# endif
38# ifndef IS_PADCONST
39# define IS_PADCONST(v) 0
40# endif
41#endif
42
43/* 5.8.7 */
44#ifndef SvRV_set
45# define SvRV_set(s,v) SvRV(s) = (v)
46#endif
47
48#include <stdio.h> 9#include <stdio.h>
49#include <errno.h> 10#include <errno.h>
50#include <assert.h> 11#include <assert.h>
51
52#if !__i386 && !__x86_64 && !__powerpc && !__m68k && !__alpha && !__mips && !__sparc64
53# undef STACKGUARD
54#endif
55
56#ifndef STACKGUARD
57# define STACKGUARD 0
58#endif
59 12
60#ifdef HAVE_MMAP 13#ifdef HAVE_MMAP
61# include <unistd.h> 14# include <unistd.h>
62# include <sys/mman.h> 15# include <sys/mman.h>
63# ifndef MAP_ANONYMOUS 16# ifndef MAP_ANONYMOUS
78#else 31#else
79# define PAGESIZE 0 32# define PAGESIZE 0
80# define BOOT_PAGESIZE (void)0 33# define BOOT_PAGESIZE (void)0
81#endif 34#endif
82 35
36#if USE_VALGRIND
37# include <valgrind/valgrind.h>
38#endif
39
40/* the maximum number of idle cctx that will be pooled */
41#define MAX_IDLE_CCTX 8
42
43#define PERL_VERSION_ATLEAST(a,b,c) \
44 (PERL_REVISION > (a) \
45 || (PERL_REVISION == (a) \
46 && (PERL_VERSION > (b) \
47 || (PERL_VERSION == (b) && PERLSUBVERSION >= (c)))))
48
49#if !PERL_VERSION_ATLEAST (5,6,0)
50# ifndef PL_ppaddr
51# define PL_ppaddr ppaddr
52# endif
53# ifndef call_sv
54# define call_sv perl_call_sv
55# endif
56# ifndef get_sv
57# define get_sv perl_get_sv
58# endif
59# ifndef get_cv
60# define get_cv perl_get_cv
61# endif
62# ifndef IS_PADGV
63# define IS_PADGV(v) 0
64# endif
65# ifndef IS_PADCONST
66# define IS_PADCONST(v) 0
67# endif
68#endif
69
70/* 5.8.7 */
71#ifndef SvRV_set
72# define SvRV_set(s,v) SvRV(s) = (v)
73#endif
74
75#if !__i386 && !__x86_64 && !__powerpc && !__m68k && !__alpha && !__mips && !__sparc64
76# undef STACKGUARD
77#endif
78
79#ifndef STACKGUARD
80# define STACKGUARD 0
81#endif
82
83/* prefer perl internal functions over our own? */
84#ifndef PREFER_PERL_FUNCTIONS
85# define PREFER_PERL_FUNCTIONS 0
86#endif
87
83/* The next macro should declare a variable stacklevel that contains and approximation 88/* The next macro should declare a variable stacklevel that contains and approximation
84 * to the current C stack pointer. Its property is that it changes with each call 89 * to the current C stack pointer. Its property is that it changes with each call
85 * and should be unique. */ 90 * and should be unique. */
86#define dSTACKLEVEL int stacklevel 91#define dSTACKLEVEL int stacklevel
87#define STACKLEVEL ((void *)&stacklevel) 92#define STACKLEVEL ((void *)&stacklevel)
88 93
89#define IN_DESTRUCT (PL_main_cv == Nullcv) 94#define IN_DESTRUCT (PL_main_cv == Nullcv)
90 95
91#if __GNUC__ >= 3 96#if __GNUC__ >= 3
92# define attribute(x) __attribute__(x) 97# define attribute(x) __attribute__(x)
98# define BARRIER __asm__ __volatile__ ("" : : : "memory")
93#else 99#else
94# define attribute(x) 100# define attribute(x)
101# define BARRIER
95#endif 102#endif
96 103
97#define NOINLINE attribute ((noinline)) 104#define NOINLINE attribute ((noinline))
98 105
99#include "CoroAPI.h" 106#include "CoroAPI.h"
105#else 112#else
106# define LOCK (void)0 113# define LOCK (void)0
107# define UNLOCK (void)0 114# define UNLOCK (void)0
108#endif 115#endif
109 116
117/* helper storage struct for Coro::AIO */
110struct io_state 118struct io_state
111{ 119{
112 int errorno; 120 int errorno;
113 I32 laststype; 121 I32 laststype;
114 int laststatval; 122 int laststatval;
130 /* the stack */ 138 /* the stack */
131 void *sptr; 139 void *sptr;
132 long ssize; /* positive == mmap, otherwise malloc */ 140 long ssize; /* positive == mmap, otherwise malloc */
133 141
134 /* cpu state */ 142 /* cpu state */
135 void *idle_sp; /* sp of top-level transfer/schedule/cede call */ 143 void *idle_sp; /* sp of top-level transfer/schedule/cede call */
144 JMPENV *idle_te; /* same as idle_sp, but for top_env, TODO: remove once stable */
136 JMPENV *top_env; 145 JMPENV *top_env;
137 coro_context cctx; 146 coro_context cctx;
138 147
139 int inuse; 148 int inuse;
140 149
142 int valgrind_id; 151 int valgrind_id;
143#endif 152#endif
144} coro_cctx; 153} coro_cctx;
145 154
146enum { 155enum {
147 CF_RUNNING = 0x0001, /* coroutine is running */ 156 CF_RUNNING = 0x0001, /* coroutine is running */
148 CF_READY = 0x0002, /* coroutine is ready */ 157 CF_READY = 0x0002, /* coroutine is ready */
149 CF_NEW = 0x0004, /* ahs never been switched to */ 158 CF_NEW = 0x0004, /* has never been switched to */
159 CF_DESTROYED = 0x0008, /* coroutine data has been freed */
150}; 160};
151 161
152/* this is a structure representing a perl-level coroutine */ 162/* this is a structure representing a perl-level coroutine */
153struct coro { 163struct coro {
154 /* the c coroutine allocated to this perl coroutine, if any */ 164 /* the c coroutine allocated to this perl coroutine, if any */
176}; 186};
177 187
178typedef struct coro *Coro__State; 188typedef struct coro *Coro__State;
179typedef struct coro *Coro__State_or_hashref; 189typedef struct coro *Coro__State_or_hashref;
180 190
191/** Coro ********************************************************************/
192
193#define PRIO_MAX 3
194#define PRIO_HIGH 1
195#define PRIO_NORMAL 0
196#define PRIO_LOW -1
197#define PRIO_IDLE -3
198#define PRIO_MIN -4
199
200/* for Coro.pm */
201static SV *coro_current;
202static AV *coro_ready [PRIO_MAX-PRIO_MIN+1];
203static int coro_nready;
204
205/** lowlevel stuff **********************************************************/
206
181static AV * 207static AV *
182coro_clone_padlist (CV *cv) 208coro_clone_padlist (CV *cv)
183{ 209{
184 AV *padlist = CvPADLIST (cv); 210 AV *padlist = CvPADLIST (cv);
185 AV *newpadlist, *newpad; 211 AV *newpadlist, *newpad;
259 285
260 if (mg && AvFILLp ((av = (AV *)mg->mg_obj)) >= 0) 286 if (mg && AvFILLp ((av = (AV *)mg->mg_obj)) >= 0)
261 CvPADLIST (cv) = (AV *)AvARRAY (av)[AvFILLp (av)--]; 287 CvPADLIST (cv) = (AV *)AvARRAY (av)[AvFILLp (av)--];
262 else 288 else
263 { 289 {
264#if 0 290#if PREFER_PERL_FUNCTIONS
265 /* this is probably cleaner, but also slower? */ 291 /* this is probably cleaner, but also slower? */
266 CV *cp = Perl_cv_clone (cv); 292 CV *cp = Perl_cv_clone (cv);
267 CvPADLIST (cv) = CvPADLIST (cp); 293 CvPADLIST (cv) = CvPADLIST (cp);
268 CvPADLIST (cp) = 0; 294 CvPADLIST (cp) = 0;
269 SvREFCNT_dec (cp); 295 SvREFCNT_dec (cp);
292 if (AvFILLp (av) >= AvMAX (av)) 318 if (AvFILLp (av) >= AvMAX (av))
293 av_extend (av, AvMAX (av) + 1); 319 av_extend (av, AvMAX (av) + 1);
294 320
295 AvARRAY (av)[++AvFILLp (av)] = (SV *)CvPADLIST (cv); 321 AvARRAY (av)[++AvFILLp (av)] = (SV *)CvPADLIST (cv);
296} 322}
323
324/** load & save, init *******************************************************/
297 325
298#define SB do { 326#define SB do {
299#define SE } while (0) 327#define SE } while (0)
300 328
301#define REPLACE_SV(sv,val) SB SvREFCNT_dec (sv); (sv) = (val); (val) = 0; SE 329#define REPLACE_SV(sv,val) SB SvREFCNT_dec (sv); (sv) = (val); (val) = 0; SE
350 /* 378 /*
351 * the worst thing you can imagine happens first - we have to save 379 * the worst thing you can imagine happens first - we have to save
352 * (and reinitialize) all cv's in the whole callchain :( 380 * (and reinitialize) all cv's in the whole callchain :(
353 */ 381 */
354 382
383 EXTEND (SP, 3 + 1);
355 PUSHs (Nullsv); 384 PUSHs (Nullsv);
356 /* this loop was inspired by pp_caller */ 385 /* this loop was inspired by pp_caller */
357 for (;;) 386 for (;;)
358 { 387 {
359 while (cxix >= 0) 388 while (cxix >= 0)
365 CV *cv = cx->blk_sub.cv; 394 CV *cv = cx->blk_sub.cv;
366 395
367 if (CvDEPTH (cv)) 396 if (CvDEPTH (cv))
368 { 397 {
369 EXTEND (SP, 3); 398 EXTEND (SP, 3);
370
371 PUSHs ((SV *)CvPADLIST (cv)); 399 PUSHs ((SV *)CvPADLIST (cv));
372 PUSHs (INT2PTR (SV *, CvDEPTH (cv))); 400 PUSHs (INT2PTR (SV *, CvDEPTH (cv)));
373 PUSHs ((SV *)cv); 401 PUSHs ((SV *)cv);
374 402
375 CvDEPTH (cv) = 0; 403 CvDEPTH (cv) = 0;
376 get_padlist (cv); 404 get_padlist (cv);
377 } 405 }
378 } 406 }
379#ifdef CXt_FORMAT
380 else if (CxTYPE (cx) == CXt_FORMAT)
381 {
382 /* I never used formats, so how should I know how these are implemented? */
383 /* my bold guess is as a simple, plain sub... */
384 croak ("CXt_FORMAT not yet handled. Don't switch coroutines from within formats");
385 }
386#endif
387 } 407 }
388 408
389 if (top_si->si_type == PERLSI_MAIN) 409 if (top_si->si_type == PERLSI_MAIN)
390 break; 410 break;
391 411
411 * allocate various perl stacks. This is an exact copy 431 * allocate various perl stacks. This is an exact copy
412 * of perl.c:init_stacks, except that it uses less memory 432 * of perl.c:init_stacks, except that it uses less memory
413 * on the (sometimes correct) assumption that coroutines do 433 * on the (sometimes correct) assumption that coroutines do
414 * not usually need a lot of stackspace. 434 * not usually need a lot of stackspace.
415 */ 435 */
436#if PREFER_PERL_FUNCTIONS
437# define coro_init_stacks init_stacks
438#else
416static void 439static void
417coro_init_stacks () 440coro_init_stacks ()
418{ 441{
419 PL_curstackinfo = new_stackinfo(128, 1024/sizeof(PERL_CONTEXT)); 442 PL_curstackinfo = new_stackinfo(128, 1024/sizeof(PERL_CONTEXT));
420 PL_curstackinfo->si_type = PERLSI_MAIN; 443 PL_curstackinfo->si_type = PERLSI_MAIN;
450 New(54,PL_retstack,16,OP*); 473 New(54,PL_retstack,16,OP*);
451 PL_retstack_ix = 0; 474 PL_retstack_ix = 0;
452 PL_retstack_max = 16; 475 PL_retstack_max = 16;
453#endif 476#endif
454} 477}
478#endif
455 479
456/* 480/*
457 * destroy the stacks, the callchain etc... 481 * destroy the stacks, the callchain etc...
458 */ 482 */
459static void 483static void
460coro_destroy_stacks () 484coro_destroy_stacks ()
461{ 485{
462 if (!IN_DESTRUCT) 486 if (!IN_DESTRUCT)
463 { 487 {
464 /* is this ugly, I ask? */ 488 /* restore all saved variables and stuff */
465 LEAVE_SCOPE (0); 489 LEAVE_SCOPE (0);
490 assert (PL_tmps_floor == -1);
466 491
467 /* sure it is, but more important: is it correct?? :/ */ 492 /* free all temporaries */
468 FREETMPS; 493 FREETMPS;
494 assert (PL_tmps_ix == -1);
469 495
496 /* unwind all extra stacks */
470 /*POPSTACK_TO (PL_mainstack);*//*D*//*use*/ 497 POPSTACK_TO (PL_mainstack);
498
499 /* unwind main stack */
500 dounwind (-1);
471 } 501 }
472 502
473 while (PL_curstackinfo->si_next) 503 while (PL_curstackinfo->si_next)
474 PL_curstackinfo = PL_curstackinfo->si_next; 504 PL_curstackinfo = PL_curstackinfo->si_next;
475 505
476 while (PL_curstackinfo) 506 while (PL_curstackinfo)
477 { 507 {
478 PERL_SI *p = PL_curstackinfo->si_prev; 508 PERL_SI *p = PL_curstackinfo->si_prev;
479 509
480 { /*D*//*remove*/
481 dSP;
482 SWITCHSTACK (PL_curstack, PL_curstackinfo->si_stack);
483 PUTBACK; /* possibly superfluous */
484 }
485
486 if (!IN_DESTRUCT) 510 if (!IN_DESTRUCT)
487 {
488 dounwind (-1);/*D*//*remove*/
489 SvREFCNT_dec (PL_curstackinfo->si_stack); 511 SvREFCNT_dec (PL_curstackinfo->si_stack);
490 }
491 512
492 Safefree (PL_curstackinfo->si_cxstack); 513 Safefree (PL_curstackinfo->si_cxstack);
493 Safefree (PL_curstackinfo); 514 Safefree (PL_curstackinfo);
494 PL_curstackinfo = p; 515 PL_curstackinfo = p;
495 } 516 }
500 Safefree (PL_savestack); 521 Safefree (PL_savestack);
501#if !PERL_VERSION_ATLEAST (5,9,0) 522#if !PERL_VERSION_ATLEAST (5,9,0)
502 Safefree (PL_retstack); 523 Safefree (PL_retstack);
503#endif 524#endif
504} 525}
526
527/** coroutine stack handling ************************************************/
505 528
506static void 529static void
507setup_coro (struct coro *coro) 530setup_coro (struct coro *coro)
508{ 531{
509 /* 532 /*
702 ++cctx_idle; 725 ++cctx_idle;
703 cctx->next = cctx_first; 726 cctx->next = cctx_first;
704 cctx_first = cctx; 727 cctx_first = cctx;
705} 728}
706 729
730/** coroutine switching *****************************************************/
731
707/* never call directly, always through the coro_state_transfer global variable */ 732/* never call directly, always through the coro_state_transfer global variable */
708static void NOINLINE 733static void NOINLINE
709transfer (struct coro *prev, struct coro *next) 734transfer (struct coro *prev, struct coro *next)
710{ 735{
711 dSTACKLEVEL; 736 dSTACKLEVEL;
712 737
713 /* sometimes transfer is only called to set idle_sp */ 738 /* sometimes transfer is only called to set idle_sp */
714 if (!next) 739 if (!next)
715 { 740 {
716 ((coro_cctx *)prev)->idle_sp = STACKLEVEL; 741 ((coro_cctx *)prev)->idle_sp = STACKLEVEL;
717 assert (((coro_cctx *)prev)->top_env = PL_top_env); /* just for the side effetc when assert is enabled */ 742 assert (((coro_cctx *)prev)->idle_te = PL_top_env); /* just for the side-effect when asserts are enabled */
718 } 743 }
719 else if (prev != next) 744 else if (prev != next)
720 { 745 {
721 coro_cctx *prev__cctx; 746 coro_cctx *prev__cctx;
722 747
733 if (!prev->flags & CF_RUNNING) 758 if (!prev->flags & CF_RUNNING)
734 croak ("Coro::State::transfer called with non-running prev Coro::State, but can only transfer from running states"); 759 croak ("Coro::State::transfer called with non-running prev Coro::State, but can only transfer from running states");
735 760
736 if (next->flags & CF_RUNNING) 761 if (next->flags & CF_RUNNING)
737 croak ("Coro::State::transfer called with running next Coro::State, but can only transfer to inactive states"); 762 croak ("Coro::State::transfer called with running next Coro::State, but can only transfer to inactive states");
763
764 if (next->flags & CF_DESTROYED)
765 croak ("Coro::State::transfer called with destroyed next Coro::State, but can only transfer to inactive states");
738 766
739 prev->flags &= ~CF_RUNNING; 767 prev->flags &= ~CF_RUNNING;
740 next->flags |= CF_RUNNING; 768 next->flags |= CF_RUNNING;
741 769
742 LOCK; 770 LOCK;
763 791
764 /* possibly "free" the cctx */ 792 /* possibly "free" the cctx */
765 if (prev__cctx->idle_sp == STACKLEVEL) 793 if (prev__cctx->idle_sp == STACKLEVEL)
766 { 794 {
767 /* I assume that STACKLEVEL is a stronger indicator than PL_top_env changes */ 795 /* I assume that STACKLEVEL is a stronger indicator than PL_top_env changes */
768 assert (PL_top_env == prev__cctx->top_env); 796 assert (("ERROR: current top_env must equal previous top_env", PL_top_env == prev__cctx->idle_te));
769 797
770 prev->cctx = 0; 798 prev->cctx = 0;
771 799
772 cctx_put (prev__cctx); 800 cctx_put (prev__cctx);
773 prev__cctx->inuse = 0; 801 prev__cctx->inuse = 0;
786 PL_top_env = next->cctx->top_env; 814 PL_top_env = next->cctx->top_env;
787 coro_transfer (&prev__cctx->cctx, &next->cctx->cctx); 815 coro_transfer (&prev__cctx->cctx, &next->cctx->cctx);
788 } 816 }
789 817
790 free_coro_mortal (); 818 free_coro_mortal ();
791
792 UNLOCK; 819 UNLOCK;
793 } 820 }
794} 821}
795 822
796struct transfer_args 823struct transfer_args
798 struct coro *prev, *next; 825 struct coro *prev, *next;
799}; 826};
800 827
801#define TRANSFER(ta) transfer ((ta).prev, (ta).next) 828#define TRANSFER(ta) transfer ((ta).prev, (ta).next)
802 829
830/** high level stuff ********************************************************/
831
803static void 832static int
804coro_state_destroy (struct coro *coro) 833coro_state_destroy (struct coro *coro)
805{ 834{
806 if (coro->refcnt--) 835 if (coro->flags & CF_DESTROYED)
807 return; 836 return 0;
837
838 coro->flags |= CF_DESTROYED;
839
840 if (coro->flags & CF_READY)
841 {
842 /* reduce nready, as destroying a ready coro effectively unreadies it */
843 /* alternative: look through all ready queues and remove the coro */
844 LOCK;
845 --coro_nready;
846 UNLOCK;
847 }
848 else
849 coro->flags |= CF_READY; /* make sure it is NOT put into the readyqueue */
808 850
809 if (coro->mainstack && coro->mainstack != main_mainstack) 851 if (coro->mainstack && coro->mainstack != main_mainstack)
810 { 852 {
853 assert (!(coro->flags & CF_RUNNING));
854
811 struct coro temp; 855 struct coro temp;
812 Zero (&temp, 1, struct coro); 856 Zero (&temp, 1, struct coro);
813 temp.save = CORO_SAVE_ALL; 857 temp.save = CORO_SAVE_ALL;
814 858
815 if (coro->flags & CF_RUNNING) 859 if (coro->flags & CF_RUNNING)
825 coro->mainstack = 0; 869 coro->mainstack = 0;
826 } 870 }
827 871
828 cctx_destroy (coro->cctx); 872 cctx_destroy (coro->cctx);
829 SvREFCNT_dec (coro->args); 873 SvREFCNT_dec (coro->args);
830 Safefree (coro); 874
875 return 1;
831} 876}
832 877
833static int 878static int
834coro_state_clear (pTHX_ SV *sv, MAGIC *mg) 879coro_state_free (pTHX_ SV *sv, MAGIC *mg)
835{ 880{
836 struct coro *coro = (struct coro *)mg->mg_ptr; 881 struct coro *coro = (struct coro *)mg->mg_ptr;
837 mg->mg_ptr = 0; 882 mg->mg_ptr = 0;
838 883
884 if (--coro->refcnt < 0)
885 {
839 coro_state_destroy (coro); 886 coro_state_destroy (coro);
887 Safefree (coro);
888 }
840 889
841 return 0; 890 return 0;
842} 891}
843 892
844static int 893static int
851 return 0; 900 return 0;
852} 901}
853 902
854static MGVTBL coro_state_vtbl = { 903static MGVTBL coro_state_vtbl = {
855 0, 0, 0, 0, 904 0, 0, 0, 0,
856 coro_state_clear, 905 coro_state_free,
857 0, 906 0,
858#ifdef MGf_DUP 907#ifdef MGf_DUP
859 coro_state_dup, 908 coro_state_dup,
860#else 909#else
861# define MGf_DUP 0 910# define MGf_DUP 0
912 return old_save; 961 return old_save;
913} 962}
914 963
915/** Coro ********************************************************************/ 964/** Coro ********************************************************************/
916 965
917#define PRIO_MAX 3
918#define PRIO_HIGH 1
919#define PRIO_NORMAL 0
920#define PRIO_LOW -1
921#define PRIO_IDLE -3
922#define PRIO_MIN -4
923
924/* for Coro.pm */
925static SV *coro_current;
926static AV *coro_ready [PRIO_MAX-PRIO_MIN+1];
927static int coro_nready;
928
929static void 966static void
930coro_enq (SV *coro_sv) 967coro_enq (SV *coro_sv)
931{ 968{
932 av_push (coro_ready [SvSTATE (coro_sv)->prio - PRIO_MIN], coro_sv); 969 av_push (coro_ready [SvSTATE (coro_sv)->prio - PRIO_MIN], coro_sv);
933 coro_nready++;
934} 970}
935 971
936static SV * 972static SV *
937coro_deq (int min_prio) 973coro_deq (int min_prio)
938{ 974{
942 if (min_prio < 0) 978 if (min_prio < 0)
943 min_prio = 0; 979 min_prio = 0;
944 980
945 for (prio = PRIO_MAX - PRIO_MIN + 1; --prio >= min_prio; ) 981 for (prio = PRIO_MAX - PRIO_MIN + 1; --prio >= min_prio; )
946 if (AvFILLp (coro_ready [prio]) >= 0) 982 if (AvFILLp (coro_ready [prio]) >= 0)
947 {
948 coro_nready--;
949 return av_shift (coro_ready [prio]); 983 return av_shift (coro_ready [prio]);
950 }
951 984
952 return 0; 985 return 0;
953} 986}
954 987
955static int 988static int
963 coro = SvSTATE (coro_sv); 996 coro = SvSTATE (coro_sv);
964 997
965 if (coro->flags & CF_READY) 998 if (coro->flags & CF_READY)
966 return 0; 999 return 0;
967 1000
968#if 0 /* this is actually harmless */
969 if (coro->flags & CF_RUNNING)
970 croak ("Coro::ready called on currently running coroutine");
971#endif
972
973 coro->flags |= CF_READY; 1001 coro->flags |= CF_READY;
974 1002
975 LOCK; 1003 LOCK;
976 coro_enq (SvREFCNT_inc (coro_sv)); 1004 coro_enq (SvREFCNT_inc (coro_sv));
1005 ++coro_nready;
977 UNLOCK; 1006 UNLOCK;
978 1007
979 return 1; 1008 return 1;
980} 1009}
981 1010
982static int 1011static int
983api_is_ready (SV *coro_sv) 1012api_is_ready (SV *coro_sv)
984{ 1013{
985 return !!SvSTATE (coro_sv)->flags & CF_READY; 1014 return !!(SvSTATE (coro_sv)->flags & CF_READY);
986} 1015}
987 1016
988static void 1017static void
989prepare_schedule (struct transfer_args *ta) 1018prepare_schedule (struct transfer_args *ta)
990{ 1019{
991 SV *prev, *next; 1020 SV *prev_sv, *next_sv;
992 1021
993 for (;;) 1022 for (;;)
994 { 1023 {
995 LOCK; 1024 LOCK;
996 next = coro_deq (PRIO_MIN); 1025 next_sv = coro_deq (PRIO_MIN);
1026
1027 /* nothing to schedule: call the idle handler */
1028 if (!next_sv)
1029 {
1030 UNLOCK;
1031 dSP;
1032
1033 ENTER;
1034 SAVETMPS;
1035
1036 PUSHMARK (SP);
1037 PUTBACK;
1038 call_sv (get_sv ("Coro::idle", FALSE), G_DISCARD);
1039
1040 FREETMPS;
1041 LEAVE;
1042 continue;
1043 }
1044
1045 ta->next = SvSTATE (next_sv);
1046
1047 /* cannot transfer to destroyed coros, skip and look for next */
1048 if (ta->next->flags & CF_DESTROYED)
1049 {
1050 UNLOCK;
1051 SvREFCNT_dec (next_sv);
1052 /* coro_nready is already taken care of by destroy */
1053 continue;
1054 }
1055
1056 --coro_nready;
997 UNLOCK; 1057 UNLOCK;
998
999 if (next)
1000 break; 1058 break;
1001
1002 {
1003 dSP;
1004
1005 ENTER;
1006 SAVETMPS;
1007
1008 PUSHMARK (SP);
1009 PUTBACK;
1010 call_sv (get_sv ("Coro::idle", FALSE), G_DISCARD);
1011
1012 FREETMPS;
1013 LEAVE;
1014 } 1059 }
1015 }
1016
1017 prev = SvRV (coro_current);
1018 SvRV_set (coro_current, next);
1019 1060
1020 /* free this only after the transfer */ 1061 /* free this only after the transfer */
1062 prev_sv = SvRV (coro_current);
1063 SvRV_set (coro_current, next_sv);
1064 ta->prev = SvSTATE (prev_sv);
1065
1066 assert (ta->next->flags & CF_READY);
1067 ta->next->flags &= ~CF_READY;
1068
1021 LOCK; 1069 LOCK;
1022 free_coro_mortal (); 1070 free_coro_mortal ();
1071 coro_mortal = prev_sv;
1023 UNLOCK; 1072 UNLOCK;
1024 coro_mortal = prev;
1025
1026 assert (!SvROK(prev));//D
1027 assert (!SvROK(next));//D
1028
1029 ta->prev = SvSTATE (prev);
1030 ta->next = SvSTATE (next);
1031
1032 assert (ta->next->flags & CF_READY);
1033 ta->next->flags &= ~CF_READY;
1034} 1073}
1035 1074
1036static void 1075static void
1037prepare_cede (struct transfer_args *ta) 1076prepare_cede (struct transfer_args *ta)
1038{ 1077{
1039 api_ready (coro_current); 1078 api_ready (coro_current);
1040
1041 prepare_schedule (ta); 1079 prepare_schedule (ta);
1080}
1081
1082static int
1083prepare_cede_notself (struct transfer_args *ta)
1084{
1085 if (coro_nready)
1086 {
1087 SV *prev = SvRV (coro_current);
1088 prepare_schedule (ta);
1089 api_ready (prev);
1090 return 1;
1091 }
1092 else
1093 return 0;
1042} 1094}
1043 1095
1044static void 1096static void
1045api_schedule (void) 1097api_schedule (void)
1046{ 1098{
1048 1100
1049 prepare_schedule (&ta); 1101 prepare_schedule (&ta);
1050 TRANSFER (ta); 1102 TRANSFER (ta);
1051} 1103}
1052 1104
1053static void 1105static int
1054api_cede (void) 1106api_cede (void)
1055{ 1107{
1056 struct transfer_args ta; 1108 struct transfer_args ta;
1057 1109
1058 prepare_cede (&ta); 1110 prepare_cede (&ta);
1111
1112 if (ta.prev != ta.next)
1113 {
1059 TRANSFER (ta); 1114 TRANSFER (ta);
1115 return 1;
1116 }
1117 else
1118 return 0;
1119}
1120
1121static int
1122api_cede_notself (void)
1123{
1124 struct transfer_args ta;
1125
1126 if (prepare_cede_notself (&ta))
1127 {
1128 TRANSFER (ta);
1129 return 1;
1130 }
1131 else
1132 return 0;
1060} 1133}
1061 1134
1062MODULE = Coro::State PACKAGE = Coro::State 1135MODULE = Coro::State PACKAGE = Coro::State
1063 1136
1064PROTOTYPES: DISABLE 1137PROTOTYPES: DISABLE
1120_set_stacklevel (...) 1193_set_stacklevel (...)
1121 ALIAS: 1194 ALIAS:
1122 Coro::State::transfer = 1 1195 Coro::State::transfer = 1
1123 Coro::schedule = 2 1196 Coro::schedule = 2
1124 Coro::cede = 3 1197 Coro::cede = 3
1198 Coro::cede_notself = 4
1125 CODE: 1199 CODE:
1126{ 1200{
1127 struct transfer_args ta; 1201 struct transfer_args ta;
1128 1202
1129 switch (ix) 1203 switch (ix)
1145 break; 1219 break;
1146 1220
1147 case 3: 1221 case 3:
1148 prepare_cede (&ta); 1222 prepare_cede (&ta);
1149 break; 1223 break;
1224
1225 case 4:
1226 if (!prepare_cede_notself (&ta))
1227 XSRETURN_EMPTY;
1228
1229 break;
1150 } 1230 }
1151 1231
1232 BARRIER;
1152 TRANSFER (ta); 1233 TRANSFER (ta);
1153}
1154 1234
1155void 1235 if (GIMME_V != G_VOID && ta.next != ta.prev)
1156_clone_state_from (SV *dst, SV *src) 1236 XSRETURN_YES;
1237}
1238
1239bool
1240_destroy (SV *coro_sv)
1157 CODE: 1241 CODE:
1158{ 1242 RETVAL = coro_state_destroy (SvSTATE (coro_sv));
1159 struct coro *coro_src = SvSTATE (src); 1243 OUTPUT:
1160 1244 RETVAL
1161 sv_unmagic (SvRV (dst), PERL_MAGIC_ext);
1162
1163 ++coro_src->refcnt;
1164 sv_magicext (SvRV (dst), 0, PERL_MAGIC_ext, &coro_state_vtbl, (char *)coro_src, 0)->mg_flags |= MGf_DUP;
1165}
1166 1245
1167void 1246void
1168_exit (code) 1247_exit (code)
1169 int code 1248 int code
1170 PROTOTYPE: $ 1249 PROTOTYPE: $
1207 coro_ready[i] = newAV (); 1286 coro_ready[i] = newAV ();
1208 1287
1209 { 1288 {
1210 SV *sv = perl_get_sv("Coro::API", 1); 1289 SV *sv = perl_get_sv("Coro::API", 1);
1211 1290
1212 coroapi.schedule = api_schedule; 1291 coroapi.schedule = api_schedule;
1213 coroapi.save = api_save; 1292 coroapi.save = api_save;
1214 coroapi.cede = api_cede; 1293 coroapi.cede = api_cede;
1294 coroapi.cede_notself = api_cede_notself;
1215 coroapi.ready = api_ready; 1295 coroapi.ready = api_ready;
1216 coroapi.is_ready = api_is_ready; 1296 coroapi.is_ready = api_is_ready;
1217 coroapi.nready = &coro_nready; 1297 coroapi.nready = &coro_nready;
1218 coroapi.current = coro_current; 1298 coroapi.current = coro_current;
1219 1299
1220 GCoroAPI = &coroapi; 1300 GCoroAPI = &coroapi;
1221 sv_setiv (sv, (IV)&coroapi); 1301 sv_setiv (sv, (IV)&coroapi);
1222 SvREADONLY_on (sv); 1302 SvREADONLY_on (sv);
1223 } 1303 }
1239 RETVAL = coro->prio; 1319 RETVAL = coro->prio;
1240 1320
1241 if (items > 1) 1321 if (items > 1)
1242 { 1322 {
1243 if (ix) 1323 if (ix)
1244 newprio += coro->prio; 1324 newprio = coro->prio - newprio;
1245 1325
1246 if (newprio < PRIO_MIN) newprio = PRIO_MIN; 1326 if (newprio < PRIO_MIN) newprio = PRIO_MIN;
1247 if (newprio > PRIO_MAX) newprio = PRIO_MAX; 1327 if (newprio > PRIO_MAX) newprio = PRIO_MAX;
1248 1328
1249 coro->prio = newprio; 1329 coro->prio = newprio;
1250 } 1330 }
1251} 1331}
1332 OUTPUT:
1333 RETVAL
1252 1334
1253SV * 1335SV *
1254ready (SV *self) 1336ready (SV *self)
1255 PROTOTYPE: $ 1337 PROTOTYPE: $
1256 CODE: 1338 CODE:

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines