| … | |
… | |
| 28 | # define IS_PADCONST(v) 0 |
28 | # define IS_PADCONST(v) 0 |
| 29 | # endif |
29 | # endif |
| 30 | #endif |
30 | #endif |
| 31 | |
31 | |
| 32 | #include <errno.h> |
32 | #include <errno.h> |
| 33 | #include <signal.h> |
33 | |
|
|
34 | #if !__i386 && !__x86_64 && !__powerpc && !__m68k && !__alpha && !__mips && !__sparc64 |
|
|
35 | # undef STACKGUARD |
|
|
36 | #endif |
|
|
37 | |
|
|
38 | #ifndef STACKGUARD |
|
|
39 | # define STACKGUARD 0 |
|
|
40 | #endif |
| 34 | |
41 | |
| 35 | #ifdef HAVE_MMAP |
42 | #ifdef HAVE_MMAP |
| 36 | # include <unistd.h> |
43 | # include <unistd.h> |
| 37 | # include <sys/mman.h> |
44 | # include <sys/mman.h> |
| 38 | # ifndef MAP_ANONYMOUS |
45 | # ifndef MAP_ANONYMOUS |
| … | |
… | |
| 40 | # define MAP_ANONYMOUS MAP_ANON |
47 | # define MAP_ANONYMOUS MAP_ANON |
| 41 | # else |
48 | # else |
| 42 | # undef HAVE_MMAP |
49 | # undef HAVE_MMAP |
| 43 | # endif |
50 | # endif |
| 44 | # endif |
51 | # endif |
|
|
52 | # include <limits.h> |
|
|
53 | # ifndef PAGESIZE |
|
|
54 | # define PAGESIZE pagesize |
|
|
55 | # define BOOT_PAGESIZE pagesize = sysconf (_SC_PAGESIZE) |
|
|
56 | static long pagesize; |
|
|
57 | # else |
|
|
58 | # define BOOT_PAGESIZE |
|
|
59 | # endif |
| 45 | #endif |
60 | #endif |
| 46 | |
61 | |
| 47 | #define SUB_INIT "Coro::State::initialize" |
62 | #define SUB_INIT "Coro::State::initialize" |
| 48 | #define UCORO_STATE "_coro_state" |
63 | #define UCORO_STATE "_coro_state" |
| 49 | |
64 | |
| … | |
… | |
| 85 | void *sptr; |
100 | void *sptr; |
| 86 | long ssize; /* positive == mmap, otherwise malloc */ |
101 | long ssize; /* positive == mmap, otherwise malloc */ |
| 87 | } coro_stack; |
102 | } coro_stack; |
| 88 | |
103 | |
| 89 | struct coro { |
104 | struct coro { |
| 90 | /* the top-level JMPENV for each coroutine, needed to catch dies. */ |
|
|
| 91 | JMPENV start_env; |
|
|
| 92 | |
|
|
| 93 | /* the optional C context */ |
105 | /* the optional C context */ |
| 94 | coro_stack *stack; |
106 | coro_stack *stack; |
| 95 | void *cursp; |
107 | void *cursp; |
| 96 | int gencnt; |
108 | int gencnt; |
| 97 | |
109 | |
| … | |
… | |
| 133 | I32 retstack_max; |
145 | I32 retstack_max; |
| 134 | PMOP *curpm; |
146 | PMOP *curpm; |
| 135 | COP *curcop; |
147 | COP *curcop; |
| 136 | JMPENV *top_env; |
148 | JMPENV *top_env; |
| 137 | |
149 | |
|
|
150 | /* coro process data */ |
|
|
151 | int prio; |
|
|
152 | |
| 138 | /* data associated with this coroutine (initial args) */ |
153 | /* data associated with this coroutine (initial args) */ |
| 139 | AV *args; |
154 | AV *args; |
|
|
155 | int refcnt; |
| 140 | }; |
156 | }; |
| 141 | |
157 | |
| 142 | typedef struct coro *Coro__State; |
158 | typedef struct coro *Coro__State; |
| 143 | typedef struct coro *Coro__State_or_hashref; |
159 | typedef struct coro *Coro__State_or_hashref; |
| 144 | |
160 | |
| 145 | /* mostly copied from op.c:cv_clone2 */ |
161 | static AV * |
| 146 | STATIC AV * |
162 | coro_clone_padlist (pTHX_ CV *cv) |
| 147 | clone_padlist (pTHX_ AV *protopadlist) |
|
|
| 148 | { |
163 | { |
| 149 | AV *av; |
164 | AV *padlist = CvPADLIST (cv); |
| 150 | I32 ix; |
|
|
| 151 | AV *protopad_name = (AV *) * av_fetch (protopadlist, 0, FALSE); |
|
|
| 152 | AV *protopad = (AV *) * av_fetch (protopadlist, 1, FALSE); |
|
|
| 153 | SV **pname = AvARRAY (protopad_name); |
|
|
| 154 | SV **ppad = AvARRAY (protopad); |
|
|
| 155 | I32 fname = AvFILLp (protopad_name); |
|
|
| 156 | I32 fpad = AvFILLp (protopad); |
|
|
| 157 | AV *newpadlist, *newpad_name, *newpad; |
165 | AV *newpadlist, *newpad; |
| 158 | SV **npad; |
|
|
| 159 | |
|
|
| 160 | newpad_name = newAV (); |
|
|
| 161 | for (ix = fname; ix >= 0; ix--) |
|
|
| 162 | av_store (newpad_name, ix, SvREFCNT_inc (pname[ix])); |
|
|
| 163 | |
|
|
| 164 | newpad = newAV (); |
|
|
| 165 | av_fill (newpad, AvFILLp (protopad)); |
|
|
| 166 | npad = AvARRAY (newpad); |
|
|
| 167 | |
166 | |
| 168 | newpadlist = newAV (); |
167 | newpadlist = newAV (); |
| 169 | AvREAL_off (newpadlist); |
168 | AvREAL_off (newpadlist); |
| 170 | av_store (newpadlist, 0, (SV *) newpad_name); |
169 | #if PERL_VERSION < 9 |
|
|
170 | Perl_pad_push (aTHX_ padlist, AvFILLp (padlist) + 1, 1); |
|
|
171 | #else |
|
|
172 | Perl_pad_push (aTHX_ padlist, AvFILLp (padlist) + 1); |
|
|
173 | #endif |
|
|
174 | newpad = (AV *)AvARRAY (padlist)[AvFILLp (padlist)]; |
|
|
175 | --AvFILLp (padlist); |
|
|
176 | |
|
|
177 | av_store (newpadlist, 0, SvREFCNT_inc (*av_fetch (padlist, 0, FALSE))); |
| 171 | av_store (newpadlist, 1, (SV *) newpad); |
178 | av_store (newpadlist, 1, (SV *)newpad); |
| 172 | |
|
|
| 173 | av = newAV (); /* will be @_ */ |
|
|
| 174 | av_extend (av, 0); |
|
|
| 175 | av_store (newpad, 0, (SV *) av); |
|
|
| 176 | AvREIFY_on (av); |
|
|
| 177 | |
|
|
| 178 | for (ix = fpad; ix > 0; ix--) |
|
|
| 179 | { |
|
|
| 180 | SV *namesv = (ix <= fname) ? pname[ix] : Nullsv; |
|
|
| 181 | |
|
|
| 182 | if (namesv && namesv != &PL_sv_undef) |
|
|
| 183 | { |
|
|
| 184 | char *name = SvPVX (namesv); /* XXX */ |
|
|
| 185 | |
|
|
| 186 | if (SvFLAGS (namesv) & SVf_FAKE || *name == '&') |
|
|
| 187 | { /* lexical from outside? */ |
|
|
| 188 | npad[ix] = SvREFCNT_inc (ppad[ix]); |
|
|
| 189 | } |
|
|
| 190 | else |
|
|
| 191 | { /* our own lexical */ |
|
|
| 192 | SV *sv; |
|
|
| 193 | if (*name == '&') |
|
|
| 194 | sv = SvREFCNT_inc (ppad[ix]); |
|
|
| 195 | else if (*name == '@') |
|
|
| 196 | sv = (SV *) newAV (); |
|
|
| 197 | else if (*name == '%') |
|
|
| 198 | sv = (SV *) newHV (); |
|
|
| 199 | else |
|
|
| 200 | sv = NEWSV (0, 0); |
|
|
| 201 | |
|
|
| 202 | #ifdef SvPADBUSY |
|
|
| 203 | if (!SvPADBUSY (sv)) |
|
|
| 204 | #endif |
|
|
| 205 | SvPADMY_on (sv); |
|
|
| 206 | |
|
|
| 207 | npad[ix] = sv; |
|
|
| 208 | } |
|
|
| 209 | } |
|
|
| 210 | else if (IS_PADGV (ppad[ix]) || IS_PADCONST (ppad[ix])) |
|
|
| 211 | { |
|
|
| 212 | npad[ix] = SvREFCNT_inc (ppad[ix]); |
|
|
| 213 | } |
|
|
| 214 | else |
|
|
| 215 | { |
|
|
| 216 | SV *sv = NEWSV (0, 0); |
|
|
| 217 | SvPADTMP_on (sv); |
|
|
| 218 | npad[ix] = sv; |
|
|
| 219 | } |
|
|
| 220 | } |
|
|
| 221 | |
|
|
| 222 | #if 0 /* return -ENOTUNDERSTOOD */ |
|
|
| 223 | /* Now that vars are all in place, clone nested closures. */ |
|
|
| 224 | |
|
|
| 225 | for (ix = fpad; ix > 0; ix--) { |
|
|
| 226 | SV* namesv = (ix <= fname) ? pname[ix] : Nullsv; |
|
|
| 227 | if (namesv |
|
|
| 228 | && namesv != &PL_sv_undef |
|
|
| 229 | && !(SvFLAGS(namesv) & SVf_FAKE) |
|
|
| 230 | && *SvPVX(namesv) == '&' |
|
|
| 231 | && CvCLONE(ppad[ix])) |
|
|
| 232 | { |
|
|
| 233 | CV *kid = cv_clone((CV*)ppad[ix]); |
|
|
| 234 | SvREFCNT_dec(ppad[ix]); |
|
|
| 235 | CvCLONE_on(kid); |
|
|
| 236 | SvPADMY_on(kid); |
|
|
| 237 | npad[ix] = (SV*)kid; |
|
|
| 238 | } |
|
|
| 239 | } |
|
|
| 240 | #endif |
|
|
| 241 | |
179 | |
| 242 | return newpadlist; |
180 | return newpadlist; |
| 243 | } |
181 | } |
| 244 | |
182 | |
| 245 | STATIC void |
183 | static void |
| 246 | free_padlist (pTHX_ AV *padlist) |
184 | free_padlist (pTHX_ AV *padlist) |
| 247 | { |
185 | { |
| 248 | /* may be during global destruction */ |
186 | /* may be during global destruction */ |
| 249 | if (SvREFCNT (padlist)) |
187 | if (SvREFCNT (padlist)) |
| 250 | { |
188 | { |
| … | |
… | |
| 264 | |
202 | |
| 265 | SvREFCNT_dec ((SV*)padlist); |
203 | SvREFCNT_dec ((SV*)padlist); |
| 266 | } |
204 | } |
| 267 | } |
205 | } |
| 268 | |
206 | |
| 269 | STATIC int |
207 | static int |
| 270 | coro_cv_free (pTHX_ SV *sv, MAGIC *mg) |
208 | coro_cv_free (pTHX_ SV *sv, MAGIC *mg) |
| 271 | { |
209 | { |
| 272 | AV *padlist; |
210 | AV *padlist; |
| 273 | AV *av = (AV *)mg->mg_obj; |
211 | AV *av = (AV *)mg->mg_obj; |
| 274 | |
212 | |
| 275 | /* casting is fun. */ |
213 | /* casting is fun. */ |
| 276 | while (&PL_sv_undef != (SV *)(padlist = (AV *)av_pop (av))) |
214 | while (&PL_sv_undef != (SV *)(padlist = (AV *)av_pop (av))) |
| 277 | free_padlist (aTHX_ padlist); |
215 | free_padlist (aTHX_ padlist); |
| 278 | |
216 | |
| 279 | SvREFCNT_dec (av); |
217 | SvREFCNT_dec (av); |
|
|
218 | |
|
|
219 | return 0; |
| 280 | } |
220 | } |
| 281 | |
221 | |
| 282 | #define PERL_MAGIC_coro PERL_MAGIC_ext |
222 | #define PERL_MAGIC_coro PERL_MAGIC_ext |
| 283 | |
223 | |
| 284 | static MGVTBL vtbl_coro = {0, 0, 0, 0, coro_cv_free}; |
224 | static MGVTBL vtbl_coro = {0, 0, 0, 0, coro_cv_free}; |
| 285 | |
225 | |
| 286 | /* the next two functions merely cache the padlists */ |
226 | /* the next two functions merely cache the padlists */ |
| 287 | STATIC void |
227 | static void |
| 288 | get_padlist (pTHX_ CV *cv) |
228 | get_padlist (pTHX_ CV *cv) |
| 289 | { |
229 | { |
| 290 | MAGIC *mg = mg_find ((SV *)cv, PERL_MAGIC_coro); |
230 | MAGIC *mg = mg_find ((SV *)cv, PERL_MAGIC_coro); |
| 291 | |
231 | |
| 292 | if (mg && AvFILLp ((AV *)mg->mg_obj) >= 0) |
232 | if (mg && AvFILLp ((AV *)mg->mg_obj) >= 0) |
| 293 | CvPADLIST (cv) = (AV *)av_pop ((AV *)mg->mg_obj); |
233 | CvPADLIST (cv) = (AV *)av_pop ((AV *)mg->mg_obj); |
| 294 | else |
234 | else |
|
|
235 | { |
|
|
236 | #if 0 |
|
|
237 | /* this should work - but it doesn't :( */ |
|
|
238 | CV *cp = Perl_cv_clone (aTHX_ cv); |
|
|
239 | CvPADLIST (cv) = CvPADLIST (cp); |
|
|
240 | CvPADLIST (cp) = 0; |
|
|
241 | SvREFCNT_dec (cp); |
|
|
242 | #else |
| 295 | CvPADLIST (cv) = clone_padlist (aTHX_ CvPADLIST (cv)); |
243 | CvPADLIST (cv) = coro_clone_padlist (aTHX_ cv); |
|
|
244 | #endif |
|
|
245 | } |
| 296 | } |
246 | } |
| 297 | |
247 | |
| 298 | STATIC void |
248 | static void |
| 299 | put_padlist (pTHX_ CV *cv) |
249 | put_padlist (pTHX_ CV *cv) |
| 300 | { |
250 | { |
| 301 | MAGIC *mg = mg_find ((SV *)cv, PERL_MAGIC_coro); |
251 | MAGIC *mg = mg_find ((SV *)cv, PERL_MAGIC_coro); |
| 302 | |
252 | |
| 303 | if (!mg) |
253 | if (!mg) |
| … | |
… | |
| 420 | } |
370 | } |
| 421 | |
371 | |
| 422 | PUSHs ((SV *)CvPADLIST(cv)); |
372 | PUSHs ((SV *)CvPADLIST(cv)); |
| 423 | PUSHs ((SV *)cv); |
373 | PUSHs ((SV *)cv); |
| 424 | |
374 | |
| 425 | get_padlist (aTHX_ cv); /* this is a monster */ |
375 | get_padlist (aTHX_ cv); |
| 426 | } |
376 | } |
| 427 | } |
377 | } |
| 428 | #ifdef CXt_FORMAT |
378 | #ifdef CXt_FORMAT |
| 429 | else if (CxTYPE(cx) == CXt_FORMAT) |
379 | else if (CxTYPE(cx) == CXt_FORMAT) |
| 430 | { |
380 | { |
| … | |
… | |
| 490 | * allocate various perl stacks. This is an exact copy |
440 | * allocate various perl stacks. This is an exact copy |
| 491 | * of perl.c:init_stacks, except that it uses less memory |
441 | * of perl.c:init_stacks, except that it uses less memory |
| 492 | * on the (sometimes correct) assumption that coroutines do |
442 | * on the (sometimes correct) assumption that coroutines do |
| 493 | * not usually need a lot of stackspace. |
443 | * not usually need a lot of stackspace. |
| 494 | */ |
444 | */ |
| 495 | STATIC void |
445 | static void |
| 496 | coro_init_stacks (pTHX) |
446 | coro_init_stacks (pTHX) |
| 497 | { |
447 | { |
| 498 | LOCK; |
448 | LOCK; |
| 499 | |
449 | |
| 500 | PL_curstackinfo = new_stackinfo(96, 1024/sizeof(PERL_CONTEXT) - 1); |
450 | PL_curstackinfo = new_stackinfo(96, 1024/sizeof(PERL_CONTEXT) - 1); |
| … | |
… | |
| 537 | } |
487 | } |
| 538 | |
488 | |
| 539 | /* |
489 | /* |
| 540 | * destroy the stacks, the callchain etc... |
490 | * destroy the stacks, the callchain etc... |
| 541 | */ |
491 | */ |
| 542 | STATIC void |
492 | static void |
| 543 | destroy_stacks(pTHX) |
493 | destroy_stacks(pTHX) |
| 544 | { |
494 | { |
| 545 | if (!IN_DESTRUCT) |
495 | if (!IN_DESTRUCT) |
| 546 | { |
496 | { |
| 547 | /* is this ugly, I ask? */ |
497 | /* is this ugly, I ask? */ |
| … | |
… | |
| 598 | stack->gencnt = ctx->gencnt = 0; |
548 | stack->gencnt = ctx->gencnt = 0; |
| 599 | |
549 | |
| 600 | if (alloc) |
550 | if (alloc) |
| 601 | { |
551 | { |
| 602 | #if HAVE_MMAP |
552 | #if HAVE_MMAP |
| 603 | stack->ssize = STACKSIZE * sizeof (long); /* mmap should do allocate-on-write for us */ |
553 | stack->ssize = ((STACKSIZE * sizeof (long) + PAGESIZE - 1) / PAGESIZE + STACKGUARD) * PAGESIZE; /* mmap should do allocate-on-write for us */ |
| 604 | stack->sptr = mmap (0, stack->ssize, PROT_EXEC|PROT_READ|PROT_WRITE, MAP_PRIVATE|MAP_ANONYMOUS, 0, 0); |
554 | stack->sptr = mmap (0, stack->ssize, PROT_EXEC|PROT_READ|PROT_WRITE, MAP_PRIVATE|MAP_ANONYMOUS, 0, 0); |
| 605 | if (stack->sptr == (void *)-1) |
555 | if (stack->sptr != (void *)-1) |
| 606 | #endif |
|
|
| 607 | { |
556 | { |
| 608 | /*FIXME*//*D*//* reasonable stack size! */ |
557 | # if STACKGUARD |
|
|
558 | mprotect (stack->sptr, STACKGUARD * PAGESIZE, PROT_NONE); |
|
|
559 | # endif |
|
|
560 | } |
|
|
561 | else |
|
|
562 | #endif |
|
|
563 | { |
| 609 | stack->ssize = - (STACKSIZE * sizeof (long)); |
564 | stack->ssize = - (STACKSIZE * (long)sizeof (long)); |
| 610 | New (0, stack->sptr, STACKSIZE, long); |
565 | New (0, stack->sptr, STACKSIZE, long); |
| 611 | } |
566 | } |
| 612 | } |
567 | } |
| 613 | else |
568 | else |
| 614 | stack->sptr = 0; |
569 | stack->sptr = 0; |
| … | |
… | |
| 675 | } |
630 | } |
| 676 | else |
631 | else |
| 677 | { |
632 | { |
| 678 | UNOP myop; |
633 | UNOP myop; |
| 679 | |
634 | |
| 680 | PL_op = (OP *)&myop; |
|
|
| 681 | |
|
|
| 682 | Zero(&myop, 1, UNOP); |
635 | Zero(&myop, 1, UNOP); |
| 683 | myop.op_next = Nullop; |
636 | myop.op_next = Nullop; |
| 684 | myop.op_flags = OPf_WANT_VOID; |
637 | myop.op_flags = OPf_WANT_VOID; |
|
|
638 | |
|
|
639 | PL_op = (OP *)&myop; |
| 685 | |
640 | |
| 686 | PUSHMARK(SP); |
641 | PUSHMARK(SP); |
| 687 | XPUSHs (sub_init); |
642 | XPUSHs (sub_init); |
| 688 | /* |
643 | /* |
| 689 | * the next line is slightly wrong, as PL_op->op_next |
644 | * the next line is slightly wrong, as PL_op->op_next |
| … | |
… | |
| 706 | /* |
661 | /* |
| 707 | * this is a _very_ stripped down perl interpreter ;) |
662 | * this is a _very_ stripped down perl interpreter ;) |
| 708 | */ |
663 | */ |
| 709 | dTHX; |
664 | dTHX; |
| 710 | Coro__State ctx = (Coro__State)arg; |
665 | Coro__State ctx = (Coro__State)arg; |
| 711 | JMPENV coro_start_env; |
|
|
| 712 | |
666 | |
| 713 | PL_top_env = &ctx->start_env; |
667 | PL_top_env = &PL_start_env; |
| 714 | |
668 | |
| 715 | ctx->cursp = 0; |
669 | ctx->cursp = 0; |
| 716 | PL_op = PL_op->op_next; |
670 | PL_op = PL_op->op_next; |
| 717 | CALLRUNOPS(aTHX); |
671 | CALLRUNOPS(aTHX); |
| 718 | |
672 | |
| 719 | abort (); |
673 | abort (); |
| 720 | } |
674 | } |
| 721 | |
675 | |
| 722 | STATIC void |
676 | /* never call directly, always through the coro_state_transfer global variable */ |
|
|
677 | static void |
| 723 | transfer (pTHX_ struct coro *prev, struct coro *next, int flags) |
678 | transfer_impl (pTHX_ struct coro *prev, struct coro *next, int flags) |
| 724 | { |
679 | { |
| 725 | dSTACKLEVEL; |
680 | dSTACKLEVEL; |
| 726 | |
681 | |
| 727 | if (prev != next) |
682 | if (prev != next) |
| 728 | { |
683 | { |
|
|
684 | /* has this coro been created yet? */ |
| 729 | if (next->mainstack) |
685 | if (next->mainstack) |
| 730 | { |
686 | { |
| 731 | LOCK; |
687 | LOCK; |
| 732 | SAVE (prev, flags); |
688 | SAVE (prev, flags); |
| 733 | LOAD (next); |
689 | LOAD (next); |
| … | |
… | |
| 772 | { |
728 | { |
| 773 | LOCK; |
729 | LOCK; |
| 774 | SAVE (prev, -1); /* first get rid of the old state */ |
730 | SAVE (prev, -1); /* first get rid of the old state */ |
| 775 | UNLOCK; |
731 | UNLOCK; |
| 776 | |
732 | |
|
|
733 | /* create the coroutine for the first time */ |
| 777 | if (flags & TRANSFER_SAVE_CCTXT) |
734 | if (flags & TRANSFER_SAVE_CCTXT) |
| 778 | { |
735 | { |
| 779 | if (!prev->stack) |
736 | if (!prev->stack) |
| 780 | allocate_stack (prev, 0); |
737 | allocate_stack (prev, 0); |
| 781 | |
738 | |
|
|
739 | /* the new coroutine starts with start_env again */ |
|
|
740 | PL_top_env = &PL_start_env; |
|
|
741 | |
| 782 | if (prev->stack->sptr && flags & TRANSFER_LAZY_STACK) |
742 | if (prev->stack->sptr && flags & TRANSFER_LAZY_STACK) |
| 783 | { |
743 | { |
| 784 | PL_top_env = &next->start_env; |
|
|
| 785 | |
|
|
| 786 | setup_coro (next); |
744 | setup_coro (next); |
| 787 | next->cursp = stacklevel; |
745 | next->cursp = stacklevel; |
| 788 | |
746 | |
| 789 | prev->stack->refcnt++; |
747 | prev->stack->refcnt++; |
| 790 | prev->stack->usecnt++; |
748 | prev->stack->usecnt++; |
| … | |
… | |
| 818 | coro_mortal = 0; |
776 | coro_mortal = 0; |
| 819 | } |
777 | } |
| 820 | UNLOCK; |
778 | UNLOCK; |
| 821 | } |
779 | } |
| 822 | |
780 | |
|
|
781 | /* use this function pointer to call the above function */ |
|
|
782 | /* this is done to increase chances of the compiler not inlining the call */ |
|
|
783 | void (*coro_state_transfer)(pTHX_ struct coro *prev, struct coro *next, int flags) = transfer_impl; |
|
|
784 | |
| 823 | #define SV_CORO(sv,func) \ |
785 | #define SV_CORO(sv,func) \ |
| 824 | do { \ |
786 | do { \ |
| 825 | if (SvROK (sv)) \ |
787 | if (SvROK (sv)) \ |
| 826 | sv = SvRV (sv); \ |
788 | sv = SvRV (sv); \ |
| 827 | \ |
789 | \ |
| 828 | if (SvTYPE (sv) == SVt_PVHV) \ |
790 | if (SvTYPE (sv) == SVt_PVHV && SvSTASH (sv) != coro_state_stash) \ |
| 829 | { \ |
791 | { \ |
| 830 | HE *he = hv_fetch_ent ((HV *)sv, ucoro_state_sv, 0, ucoro_state_hash); \ |
792 | HE *he = hv_fetch_ent ((HV *)sv, ucoro_state_sv, 0, ucoro_state_hash); \ |
| 831 | \ |
793 | \ |
| 832 | if (!he) \ |
794 | if (!he) \ |
| 833 | croak ("%s() -- %s is a hashref but lacks the " UCORO_STATE " key", func, # sv); \ |
795 | croak ("%s() -- %s is a hashref but lacks the " UCORO_STATE " key", func, # sv); \ |
| … | |
… | |
| 839 | if (!SvOBJECT (sv) || SvSTASH (sv) != coro_state_stash) \ |
801 | if (!SvOBJECT (sv) || SvSTASH (sv) != coro_state_stash) \ |
| 840 | croak ("%s() -- %s is not (and contains not) a Coro::State object", func, # sv); \ |
802 | croak ("%s() -- %s is not (and contains not) a Coro::State object", func, # sv); \ |
| 841 | \ |
803 | \ |
| 842 | } while(0) |
804 | } while(0) |
| 843 | |
805 | |
| 844 | #define SvSTATE(sv) (struct coro *)SvIV (sv) |
806 | static void |
|
|
807 | coro_state_destroy (struct coro *coro) |
|
|
808 | { |
|
|
809 | if (coro->refcnt--) |
|
|
810 | return; |
| 845 | |
811 | |
|
|
812 | if (coro->mainstack && coro->mainstack != main_mainstack) |
|
|
813 | { |
|
|
814 | struct coro temp; |
|
|
815 | |
|
|
816 | SAVE (aTHX_ (&temp), TRANSFER_SAVE_ALL); |
|
|
817 | LOAD (aTHX_ coro); |
|
|
818 | |
|
|
819 | destroy_stacks (aTHX); |
|
|
820 | |
|
|
821 | LOAD ((&temp)); /* this will get rid of defsv etc.. */ |
|
|
822 | |
|
|
823 | coro->mainstack = 0; |
|
|
824 | } |
|
|
825 | |
|
|
826 | deallocate_stack (coro); |
|
|
827 | SvREFCNT_dec (coro->args); |
|
|
828 | Safefree (coro); |
|
|
829 | } |
|
|
830 | |
| 846 | static void |
831 | static int |
|
|
832 | coro_state_clear (SV *sv, MAGIC *mg) |
|
|
833 | { |
|
|
834 | struct coro *coro = (struct coro *)mg->mg_ptr; |
|
|
835 | mg->mg_ptr = 0; |
|
|
836 | |
|
|
837 | coro_state_destroy (coro); |
|
|
838 | |
|
|
839 | return 0; |
|
|
840 | } |
|
|
841 | |
|
|
842 | static int |
|
|
843 | coro_state_dup (MAGIC *mg, CLONE_PARAMS *params) |
|
|
844 | { |
|
|
845 | struct coro *coro = (struct coro *)mg->mg_ptr; |
|
|
846 | |
|
|
847 | ++coro->refcnt; |
|
|
848 | |
|
|
849 | return 0; |
|
|
850 | } |
|
|
851 | |
|
|
852 | static MGVTBL coro_state_vtbl = { 0, 0, 0, 0, coro_state_clear, 0, coro_state_dup, 0 }; |
|
|
853 | |
|
|
854 | static struct coro * |
|
|
855 | SvSTATE (SV *coro) |
|
|
856 | { |
|
|
857 | MAGIC *mg = SvMAGIC (SvROK (coro) ? SvRV (coro) : coro); |
|
|
858 | assert (mg->mg_type == PERL_MAGIC_ext); |
|
|
859 | return (struct coro *)mg->mg_ptr; |
|
|
860 | } |
|
|
861 | |
|
|
862 | static void |
| 847 | api_transfer(pTHX_ SV *prev, SV *next, int flags) |
863 | api_transfer (pTHX_ SV *prev, SV *next, int flags) |
| 848 | { |
864 | { |
| 849 | SV_CORO (prev, "Coro::transfer"); |
865 | SV_CORO (prev, "Coro::transfer"); |
| 850 | SV_CORO (next, "Coro::transfer"); |
866 | SV_CORO (next, "Coro::transfer"); |
| 851 | |
867 | |
| 852 | transfer (aTHX_ SvSTATE (prev), SvSTATE (next), flags); |
868 | coro_state_transfer (aTHX_ SvSTATE (prev), SvSTATE (next), flags); |
| 853 | } |
869 | } |
| 854 | |
870 | |
| 855 | /** Coro ********************************************************************/ |
871 | /** Coro ********************************************************************/ |
| 856 | |
872 | |
| 857 | #define PRIO_MAX 3 |
873 | #define PRIO_MAX 3 |
| … | |
… | |
| 861 | #define PRIO_IDLE -3 |
877 | #define PRIO_IDLE -3 |
| 862 | #define PRIO_MIN -4 |
878 | #define PRIO_MIN -4 |
| 863 | |
879 | |
| 864 | /* for Coro.pm */ |
880 | /* for Coro.pm */ |
| 865 | static GV *coro_current, *coro_idle; |
881 | static GV *coro_current, *coro_idle; |
| 866 | static AV *coro_ready[PRIO_MAX-PRIO_MIN+1]; |
882 | static AV *coro_ready [PRIO_MAX-PRIO_MIN+1]; |
| 867 | static int coro_nready; |
883 | static int coro_nready; |
| 868 | |
884 | |
| 869 | static void |
885 | static void |
| 870 | coro_enq (pTHX_ SV *sv) |
886 | coro_enq (pTHX_ SV *sv) |
| 871 | { |
887 | { |
|
|
888 | int prio; |
|
|
889 | |
| 872 | if (SvTYPE (sv) != SVt_PVHV) |
890 | if (SvTYPE (sv) != SVt_PVHV) |
| 873 | croak ("Coro::ready tried to enqueue something that is not a coroutine"); |
891 | croak ("Coro::ready tried to enqueue something that is not a coroutine"); |
| 874 | |
892 | |
| 875 | SV **xprio = hv_fetch ((HV *)sv, "prio", 4, 0); |
893 | { |
| 876 | int prio = xprio ? SvIV (*xprio) : PRIO_NORMAL; |
894 | SV *coro = sv; |
| 877 | |
895 | SV_CORO (coro, "omg"); |
| 878 | prio = prio > PRIO_MAX ? PRIO_MAX |
896 | prio = SvSTATE (coro)->prio; |
| 879 | : prio < PRIO_MIN ? PRIO_MIN |
897 | } |
| 880 | : prio; |
|
|
| 881 | |
898 | |
| 882 | av_push (coro_ready [prio - PRIO_MIN], sv); |
899 | av_push (coro_ready [prio - PRIO_MIN], sv); |
| 883 | coro_nready++; |
900 | coro_nready++; |
| 884 | } |
901 | } |
| 885 | |
902 | |
| … | |
… | |
| 891 | min_prio -= PRIO_MIN; |
908 | min_prio -= PRIO_MIN; |
| 892 | if (min_prio < 0) |
909 | if (min_prio < 0) |
| 893 | min_prio = 0; |
910 | min_prio = 0; |
| 894 | |
911 | |
| 895 | for (prio = PRIO_MAX - PRIO_MIN + 1; --prio >= min_prio; ) |
912 | for (prio = PRIO_MAX - PRIO_MIN + 1; --prio >= min_prio; ) |
| 896 | if (av_len (coro_ready[prio]) >= 0) |
913 | if (AvFILLp (coro_ready [prio]) >= 0) |
| 897 | { |
914 | { |
| 898 | coro_nready--; |
915 | coro_nready--; |
| 899 | return av_shift (coro_ready[prio]); |
916 | return av_shift (coro_ready [prio]); |
| 900 | } |
917 | } |
| 901 | |
918 | |
| 902 | return 0; |
919 | return 0; |
| 903 | } |
920 | } |
| 904 | |
921 | |
| … | |
… | |
| 938 | |
955 | |
| 939 | SV_CORO (next, "Coro::schedule"); |
956 | SV_CORO (next, "Coro::schedule"); |
| 940 | |
957 | |
| 941 | UNLOCK; |
958 | UNLOCK; |
| 942 | |
959 | |
| 943 | transfer (aTHX_ SvSTATE (prev), SvSTATE (next), |
960 | coro_state_transfer (aTHX_ SvSTATE (prev), SvSTATE (next), |
| 944 | TRANSFER_SAVE_ALL | TRANSFER_LAZY_STACK); |
961 | TRANSFER_SAVE_ALL | TRANSFER_LAZY_STACK); |
| 945 | } |
962 | } |
| 946 | |
963 | |
| 947 | static void |
964 | static void |
| 948 | api_cede (void) |
965 | api_cede (void) |
| 949 | { |
966 | { |
| … | |
… | |
| 956 | api_schedule (); |
973 | api_schedule (); |
| 957 | } |
974 | } |
| 958 | |
975 | |
| 959 | MODULE = Coro::State PACKAGE = Coro::State |
976 | MODULE = Coro::State PACKAGE = Coro::State |
| 960 | |
977 | |
| 961 | PROTOTYPES: ENABLE |
978 | PROTOTYPES: DISABLE |
| 962 | |
979 | |
| 963 | BOOT: |
980 | BOOT: |
| 964 | { /* {} necessary for stoopid perl-5.6.x */ |
981 | { /* {} necessary for stoopid perl-5.6.x */ |
| 965 | #ifdef USE_ITHREADS |
982 | #ifdef USE_ITHREADS |
| 966 | MUTEX_INIT (&coro_mutex); |
983 | MUTEX_INIT (&coro_mutex); |
| 967 | #endif |
984 | #endif |
|
|
985 | BOOT_PAGESIZE; |
| 968 | |
986 | |
| 969 | ucoro_state_sv = newSVpv (UCORO_STATE, sizeof(UCORO_STATE) - 1); |
987 | ucoro_state_sv = newSVpv (UCORO_STATE, sizeof(UCORO_STATE) - 1); |
| 970 | PERL_HASH(ucoro_state_hash, UCORO_STATE, sizeof(UCORO_STATE) - 1); |
988 | PERL_HASH(ucoro_state_hash, UCORO_STATE, sizeof(UCORO_STATE) - 1); |
| 971 | coro_state_stash = gv_stashpv ("Coro::State", TRUE); |
989 | coro_state_stash = gv_stashpv ("Coro::State", TRUE); |
| 972 | |
990 | |
| … | |
… | |
| 977 | |
995 | |
| 978 | main_mainstack = PL_mainstack; |
996 | main_mainstack = PL_mainstack; |
| 979 | |
997 | |
| 980 | coroapi.ver = CORO_API_VERSION; |
998 | coroapi.ver = CORO_API_VERSION; |
| 981 | coroapi.transfer = api_transfer; |
999 | coroapi.transfer = api_transfer; |
| 982 | } |
|
|
| 983 | |
1000 | |
| 984 | Coro::State |
1001 | assert (("PRIO_NORMAL must be 0", !PRIO_NORMAL)); |
| 985 | _newprocess(args) |
1002 | } |
| 986 | SV * args |
1003 | |
| 987 | PROTOTYPE: $ |
1004 | SV * |
|
|
1005 | new (char *klass, ...) |
| 988 | CODE: |
1006 | CODE: |
| 989 | Coro__State coro; |
1007 | { |
|
|
1008 | struct coro *coro; |
|
|
1009 | HV *hv; |
|
|
1010 | int i; |
| 990 | |
1011 | |
| 991 | if (!SvROK (args) || SvTYPE (SvRV (args)) != SVt_PVAV) |
|
|
| 992 | croak ("Coro::State::_newprocess expects an arrayref"); |
|
|
| 993 | |
|
|
| 994 | Newz (0, coro, 1, struct coro); |
1012 | Newz (0, coro, 1, struct coro); |
|
|
1013 | coro->args = newAV (); |
| 995 | |
1014 | |
| 996 | coro->args = (AV *)SvREFCNT_inc (SvRV (args)); |
1015 | for (i = 1; i < items; i++) |
|
|
1016 | av_push (coro->args, newSVsv (ST (i))); |
|
|
1017 | |
| 997 | /*coro->mainstack = 0; *//*actual work is done inside transfer */ |
1018 | /*coro->mainstack = 0; *//*actual work is done inside transfer */ |
| 998 | /*coro->stack = 0;*/ |
1019 | /*coro->stack = 0;*/ |
| 999 | |
1020 | |
| 1000 | /* same as JMPENV_BOOTSTRAP */ |
1021 | hv = newHV (); |
| 1001 | /* we might be able to recycle start_env, but safe is safe */ |
1022 | sv_magicext ((SV *)hv, 0, PERL_MAGIC_ext, &coro_state_vtbl, (char *)coro, 0)->mg_flags |= MGf_DUP; |
| 1002 | /*Zero(&coro->start_env, 1, JMPENV);*/ |
1023 | RETVAL = sv_bless (newRV_noinc ((SV *)hv), gv_stashpv (klass, 1)); |
| 1003 | coro->start_env.je_ret = -1; |
1024 | } |
| 1004 | coro->start_env.je_mustcatch = TRUE; |
|
|
| 1005 | |
|
|
| 1006 | RETVAL = coro; |
|
|
| 1007 | OUTPUT: |
1025 | OUTPUT: |
| 1008 | RETVAL |
1026 | RETVAL |
| 1009 | |
1027 | |
| 1010 | void |
1028 | void |
| 1011 | transfer(prev, next, flags) |
1029 | transfer(prev, next, flags) |
| 1012 | SV *prev |
1030 | SV *prev |
| 1013 | SV *next |
1031 | SV *next |
| 1014 | int flags |
1032 | int flags |
| 1015 | PROTOTYPE: @ |
|
|
| 1016 | CODE: |
1033 | CODE: |
| 1017 | PUTBACK; |
1034 | PUTBACK; |
| 1018 | SV_CORO (next, "Coro::transfer"); |
1035 | SV_CORO (next, "Coro::transfer"); |
| 1019 | SV_CORO (prev, "Coro::transfer"); |
1036 | SV_CORO (prev, "Coro::transfer"); |
| 1020 | transfer (aTHX_ SvSTATE (prev), SvSTATE (next), flags); |
1037 | coro_state_transfer (aTHX_ SvSTATE (prev), SvSTATE (next), flags); |
| 1021 | SPAGAIN; |
1038 | SPAGAIN; |
| 1022 | |
1039 | |
|
|
1040 | int |
|
|
1041 | prio (Coro::State coro, int newprio = 0) |
|
|
1042 | ALIAS: |
|
|
1043 | nice = 1 |
|
|
1044 | CODE: |
|
|
1045 | { |
|
|
1046 | RETVAL = coro->prio; |
|
|
1047 | |
|
|
1048 | if (items > 1) |
|
|
1049 | { |
|
|
1050 | if (ix) |
|
|
1051 | newprio += coro->prio; |
|
|
1052 | |
|
|
1053 | if (newprio < PRIO_MIN) newprio = PRIO_MIN; |
|
|
1054 | if (newprio > PRIO_MAX) newprio = PRIO_MAX; |
|
|
1055 | |
|
|
1056 | coro->prio = newprio; |
|
|
1057 | } |
|
|
1058 | } |
|
|
1059 | |
| 1023 | void |
1060 | void |
| 1024 | DESTROY(coro) |
1061 | _clone_state_from (SV *dst, SV *src) |
| 1025 | Coro::State coro |
1062 | CODE: |
| 1026 | CODE: |
1063 | { |
|
|
1064 | struct coro *coro_src = SvSTATE (src); |
| 1027 | |
1065 | |
| 1028 | if (coro->mainstack && coro->mainstack != main_mainstack) |
1066 | sv_unmagic (SvRV (dst), PERL_MAGIC_ext); |
| 1029 | { |
|
|
| 1030 | struct coro temp; |
|
|
| 1031 | |
1067 | |
| 1032 | PUTBACK; |
1068 | ++coro_src->refcnt; |
| 1033 | SAVE (aTHX_ (&temp), TRANSFER_SAVE_ALL); |
1069 | sv_magicext (SvRV (dst), 0, PERL_MAGIC_ext, &coro_state_vtbl, (char *)coro_src, 0)->mg_flags |= MGf_DUP; |
| 1034 | LOAD (aTHX_ coro); |
1070 | } |
| 1035 | SPAGAIN; |
|
|
| 1036 | |
|
|
| 1037 | destroy_stacks (aTHX); |
|
|
| 1038 | |
|
|
| 1039 | LOAD ((&temp)); /* this will get rid of defsv etc.. */ |
|
|
| 1040 | SPAGAIN; |
|
|
| 1041 | |
|
|
| 1042 | coro->mainstack = 0; |
|
|
| 1043 | } |
|
|
| 1044 | |
|
|
| 1045 | deallocate_stack (coro); |
|
|
| 1046 | SvREFCNT_dec (coro->args); |
|
|
| 1047 | Safefree (coro); |
|
|
| 1048 | |
1071 | |
| 1049 | void |
1072 | void |
| 1050 | _exit(code) |
1073 | _exit (code) |
| 1051 | int code |
1074 | int code |
| 1052 | PROTOTYPE: $ |
1075 | PROTOTYPE: $ |
| 1053 | CODE: |
1076 | CODE: |
| 1054 | _exit (code); |
1077 | _exit (code); |
| 1055 | |
1078 | |
| 1056 | MODULE = Coro::State PACKAGE = Coro::Cont |
1079 | MODULE = Coro::State PACKAGE = Coro::Cont |
| 1057 | |
1080 | |
| 1058 | # this is slightly dirty (should expose a c-level api) |
|
|
| 1059 | |
|
|
| 1060 | void |
1081 | void |
| 1061 | yield(...) |
1082 | yield (...) |
| 1062 | PROTOTYPE: @ |
1083 | PROTOTYPE: @ |
| 1063 | CODE: |
1084 | CODE: |
| 1064 | SV *yieldstack; |
1085 | SV *yieldstack; |
| 1065 | SV *sv; |
1086 | SV *sv; |
| 1066 | AV *defav = GvAV (PL_defgv); |
1087 | AV *defav = GvAV (PL_defgv); |
| … | |
… | |
| 1077 | av_fill (defav, items - 1); |
1098 | av_fill (defav, items - 1); |
| 1078 | while (items--) |
1099 | while (items--) |
| 1079 | av_store (defav, items, SvREFCNT_inc (ST(items))); |
1100 | av_store (defav, items, SvREFCNT_inc (ST(items))); |
| 1080 | |
1101 | |
| 1081 | sv = av_pop ((AV *)SvRV (yieldstack)); |
1102 | sv = av_pop ((AV *)SvRV (yieldstack)); |
| 1082 | prev = (struct coro *)SvIV ((SV*)SvRV (*av_fetch ((AV *)SvRV (sv), 0, 0))); |
1103 | prev = SvSTATE ((SV*)SvRV (*av_fetch ((AV *)SvRV (sv), 0, 0))); |
| 1083 | next = (struct coro *)SvIV ((SV*)SvRV (*av_fetch ((AV *)SvRV (sv), 1, 0))); |
1104 | next = SvSTATE ((SV*)SvRV (*av_fetch ((AV *)SvRV (sv), 1, 0))); |
| 1084 | SvREFCNT_dec (sv); |
1105 | SvREFCNT_dec (sv); |
| 1085 | |
1106 | |
| 1086 | transfer (aTHX_ prev, next, 0); |
1107 | coro_state_transfer (aTHX_ prev, next, 0); |
| 1087 | |
1108 | |
| 1088 | MODULE = Coro::State PACKAGE = Coro |
1109 | MODULE = Coro::State PACKAGE = Coro |
| 1089 | |
|
|
| 1090 | # this is slightly dirty (should expose a c-level api) |
|
|
| 1091 | |
1110 | |
| 1092 | BOOT: |
1111 | BOOT: |
| 1093 | { |
1112 | { |
| 1094 | int i; |
1113 | int i; |
| 1095 | HV *stash = gv_stashpv ("Coro", TRUE); |
1114 | HV *stash = gv_stashpv ("Coro", TRUE); |
| … | |
… | |
| 1115 | coroapi.ready = api_ready; |
1134 | coroapi.ready = api_ready; |
| 1116 | coroapi.nready = &coro_nready; |
1135 | coroapi.nready = &coro_nready; |
| 1117 | coroapi.current = coro_current; |
1136 | coroapi.current = coro_current; |
| 1118 | |
1137 | |
| 1119 | GCoroAPI = &coroapi; |
1138 | GCoroAPI = &coroapi; |
| 1120 | sv_setiv(sv, (IV)&coroapi); |
1139 | sv_setiv (sv, (IV)&coroapi); |
| 1121 | SvREADONLY_on(sv); |
1140 | SvREADONLY_on (sv); |
| 1122 | } |
1141 | } |
| 1123 | } |
1142 | } |
| 1124 | |
1143 | |
| 1125 | #if !PERL_MICRO |
1144 | #if !PERL_MICRO |
| 1126 | |
1145 | |
| 1127 | void |
1146 | void |
| 1128 | ready(self) |
1147 | ready (self) |
| 1129 | SV * self |
1148 | SV * self |
| 1130 | PROTOTYPE: $ |
1149 | PROTOTYPE: $ |
| 1131 | CODE: |
1150 | CODE: |
| 1132 | api_ready (self); |
1151 | api_ready (self); |
| 1133 | |
1152 | |
| 1134 | #endif |
1153 | #endif |
| 1135 | |
1154 | |
| 1136 | int |
1155 | int |
| 1137 | nready(...) |
1156 | nready (...) |
| 1138 | PROTOTYPE: |
1157 | PROTOTYPE: |
| 1139 | CODE: |
1158 | CODE: |
| 1140 | RETVAL = coro_nready; |
1159 | RETVAL = coro_nready; |
| 1141 | OUTPUT: |
1160 | OUTPUT: |
| 1142 | RETVAL |
1161 | RETVAL |
| 1143 | |
1162 | |
| 1144 | void |
1163 | void |
| 1145 | schedule(...) |
1164 | schedule (...) |
| 1146 | PROTOTYPE: |
1165 | PROTOTYPE: |
| 1147 | CODE: |
1166 | CODE: |
| 1148 | api_schedule (); |
1167 | api_schedule (); |
| 1149 | |
1168 | |
| 1150 | void |
1169 | void |
| 1151 | cede(...) |
1170 | cede (...) |
| 1152 | PROTOTYPE: |
1171 | PROTOTYPE: |
| 1153 | CODE: |
1172 | CODE: |
| 1154 | api_cede (); |
1173 | api_cede (); |
| 1155 | |
1174 | |
| 1156 | # and these are hacks |
1175 | # and these are hacks |
