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.429 by root, Fri Dec 7 14:21:09 2012 UTC vs.
Revision 1.473 by root, Sun Sep 3 23:36:01 2017 UTC

14#include "perliol.h" 14#include "perliol.h"
15 15
16#include "schmorp.h" 16#include "schmorp.h"
17 17
18#define ECB_NO_THREADS 1 18#define ECB_NO_THREADS 1
19#define ECB_NO_LIBM 1
19#include "ecb.h" 20#include "ecb.h"
20 21
21#include <stddef.h> 22#include <stddef.h>
22#include <stdio.h> 23#include <stdio.h>
23#include <errno.h> 24#include <errno.h>
24#include <assert.h> 25#include <assert.h>
25 26
27#ifndef SvREFCNT_dec_NN
28 #define SvREFCNT_dec_NN(sv) SvREFCNT_dec (sv)
29#endif
30
31#ifndef SvREFCNT_inc_NN
32 #define SvREFCNT_inc_NN(sv) SvREFCNT_inc (sv)
33#endif
34
26#ifndef SVs_PADSTALE 35#ifndef SVs_PADSTALE
27# define SVs_PADSTALE 0 36# define SVs_PADSTALE 0
28#endif 37#endif
29 38
30#ifdef PadARRAY 39#ifdef PadARRAY
31# define NEWPADAPI 40# define NEWPADAPI 1
32# define newPADLIST(var) (Newz (0, var, 1, PADLIST), Newx (PadlistARRAY (var), 2, PAD *)) 41# define newPADLIST(var) (Newz (0, var, 1, PADLIST), Newx (PadlistARRAY (var), 2, PAD *))
33#else 42#else
34typedef AV PADNAMELIST; 43typedef AV PADNAMELIST;
35# if !PERL_VERSION_ATLEAST(5,8,0) 44# if !PERL_VERSION_ATLEAST(5,8,0)
36typedef AV PADLIST; 45typedef AV PADLIST;
41# define PadlistNAMES(pl) (*PadlistARRAY (pl)) 50# define PadlistNAMES(pl) (*PadlistARRAY (pl))
42# define PadARRAY AvARRAY 51# define PadARRAY AvARRAY
43# define PadMAX AvFILLp 52# define PadMAX AvFILLp
44# define newPADLIST(var) ((var) = newAV (), av_extend (var, 1)) 53# define newPADLIST(var) ((var) = newAV (), av_extend (var, 1))
45#endif 54#endif
55#ifndef PadnamelistREFCNT
56# define PadnamelistREFCNT(pnl) SvREFCNT (pnl)
57#endif
58#ifndef PadnamelistREFCNT_dec
59# define PadnamelistREFCNT_dec(pnl) SvREFCNT_dec (pnl)
60#endif
61
62/* 5.19.something has replaced SVt_BIND by SVt_INVLIST */
63/* we just alias it to SVt_IV, as that is sufficient for swap_sv for now */
64#if PERL_VERSION_ATLEAST(5,19,0)
65# define SVt_BIND SVt_IV
66#endif
46 67
47#if defined(_WIN32) 68#if defined(_WIN32)
48# undef HAS_GETTIMEOFDAY 69# undef HAS_GETTIMEOFDAY
49# undef setjmp 70# undef setjmp
50# undef longjmp 71# undef longjmp
95# define CORO_CLOCK_MONOTONIC 1 116# define CORO_CLOCK_MONOTONIC 1
96# define CORO_CLOCK_THREAD_CPUTIME_ID 3 117# define CORO_CLOCK_THREAD_CPUTIME_ID 3
97# endif 118# endif
98#endif 119#endif
99 120
121/* one off bugfix for perl 5.22 */
122#if PERL_VERSION_ATLEAST(5,22,0) && !PERL_VERSION_ATLEAST(5,24,0)
123# undef PadlistNAMES
124# define PadlistNAMES(pl) *((PADNAMELIST **)PadlistARRAY (pl))
125#endif
126
127#if PERL_VERSION_ATLEAST(5,24,0)
128# define SUB_ARGARRAY PL_curpad[0]
129#else
130# define SUB_ARGARRAY (SV *)cx->blk_sub.argarray
131#endif
132
133/* perl usually suppresses asserts. for debugging, we sometimes force it to be on */
134#if 0
135# undef NDEBUG
136# include <assert.h>
137#endif
138
100static double (*nvtime)(); /* so why doesn't it take void? */ 139static double (*nvtime)(); /* so why doesn't it take void? */
101static void (*u2time)(pTHX_ UV ret[2]); 140static void (*u2time)(pTHX_ UV ret[2]);
102 141
103/* we hijack an hopefully unused CV flag for our purposes */ 142/* we hijack an hopefully unused CV flag for our purposes */
104#define CVf_SLF 0x4000 143#define CVf_SLF 0x4000
119 158
120static GV *irsgv; /* $/ */ 159static GV *irsgv; /* $/ */
121static GV *stdoutgv; /* *STDOUT */ 160static GV *stdoutgv; /* *STDOUT */
122static SV *rv_diehook; 161static SV *rv_diehook;
123static SV *rv_warnhook; 162static SV *rv_warnhook;
124static HV *hv_sig; /* %SIG */
125 163
126/* async_pool helper stuff */ 164/* async_pool helper stuff */
127static SV *sv_pool_rss; 165static SV *sv_pool_rss;
128static SV *sv_pool_size; 166static SV *sv_pool_size;
129static SV *sv_async_pool_idle; /* description string */ 167static SV *sv_async_pool_idle; /* description string */
235 SV *saved_deffh; 273 SV *saved_deffh;
236 SV *invoke_cb; 274 SV *invoke_cb;
237 AV *invoke_av; 275 AV *invoke_av;
238 276
239 /* on_enter/on_leave */ 277 /* on_enter/on_leave */
240 AV *on_enter; 278 AV *on_enter; AV *on_enter_xs;
241 AV *on_leave; 279 AV *on_leave; AV *on_leave_xs;
242 280
243 /* swap_sv */ 281 /* swap_sv */
244 AV *swap_sv; 282 AV *swap_sv;
245 283
246 /* times */ 284 /* times */
279 317
280#if CORO_JIT 318#if CORO_JIT
281 /* APPLE doesn't have mmap though */ 319 /* APPLE doesn't have mmap though */
282 #define CORO_JIT_UNIXY (__linux || __FreeBSD__ || __OpenBSD__ || __NetBSD__ || __solaris || __APPLE__) 320 #define CORO_JIT_UNIXY (__linux || __FreeBSD__ || __OpenBSD__ || __NetBSD__ || __solaris || __APPLE__)
283 #ifndef CORO_JIT_TYPE 321 #ifndef CORO_JIT_TYPE
284 #if __x86_64 && CORO_JIT_UNIXY 322 #if ECB_AMD64 && CORO_JIT_UNIXY
285 #define CORO_JIT_TYPE "amd64-unix" 323 #define CORO_JIT_TYPE "amd64-unix"
286 #elif __i386 && CORO_JIT_UNIXY 324 #elif __i386 && CORO_JIT_UNIXY
287 #define CORO_JIT_TYPE "x86-unix" 325 #define CORO_JIT_TYPE "x86-unix"
288 #endif 326 #endif
289 #endif 327 #endif
482 520
483 return 0; 521 return 0;
484} 522}
485 523
486ecb_inline struct coro * 524ecb_inline struct coro *
487SvSTATE_ (pTHX_ SV *coro) 525SvSTATE_ (pTHX_ SV *coro_sv)
488{ 526{
489 MAGIC *mg; 527 MAGIC *mg;
490 528
491 if (SvROK (coro)) 529 if (SvROK (coro_sv))
492 coro = SvRV (coro); 530 coro_sv = SvRV (coro_sv);
493 531
494 mg = SvSTATEhv_p (aTHX_ coro); 532 mg = SvSTATEhv_p (aTHX_ coro_sv);
495 if (!mg) 533 if (!mg)
496 croak ("Coro::State object required"); 534 croak ("Coro::State object required");
497 535
498 return (struct coro *)mg->mg_ptr; 536 return (struct coro *)mg->mg_ptr;
499} 537}
510ecb_inline PADLIST * 548ecb_inline PADLIST *
511coro_derive_padlist (pTHX_ CV *cv) 549coro_derive_padlist (pTHX_ CV *cv)
512{ 550{
513 PADLIST *padlist = CvPADLIST (cv); 551 PADLIST *padlist = CvPADLIST (cv);
514 PADLIST *newpadlist; 552 PADLIST *newpadlist;
553 PADNAMELIST *padnames;
515 PAD *newpad; 554 PAD *newpad;
516 PADOFFSET const off = PadlistMAX (padlist) + 1; 555 PADOFFSET off = PadlistMAX (padlist) + 1;
517 556
518 newPADLIST(newpadlist); 557#if NEWPADAPI
519#if !PERL_VERSION_ATLEAST(5,15,3) 558
520 /* Padlists are AvREAL as of 5.15.3. See perl bug #98092 and perl commit 7d953ba. */ 559 /* if we had the original CvDEPTH, we might be able to steal the CvDEPTH+1 entry instead */
521 AvREAL_off (newpadlist); 560 /* 20131102194744.GA6705@schmorp.de, 20131102195825.2013.qmail@lists-nntp.develooper.com */
522#endif 561 while (!PadlistARRAY (padlist)[off - 1])
562 --off;
563
564 Perl_pad_push (aTHX_ padlist, off);
565
566 newpad = PadlistARRAY (padlist)[off];
567 PadlistARRAY (padlist)[off] = 0;
568
569#else
570
523#if PERL_VERSION_ATLEAST (5,10,0) 571#if PERL_VERSION_ATLEAST (5,10,0)
524 Perl_pad_push (aTHX_ padlist, off); 572 Perl_pad_push (aTHX_ padlist, off);
525#else 573#else
526 Perl_pad_push (aTHX_ padlist, off, 1); 574 Perl_pad_push (aTHX_ padlist, off, 1);
527#endif 575#endif
576
528 newpad = PadlistARRAY (padlist)[off]; 577 newpad = PadlistARRAY (padlist)[off];
529 PadlistMAX (padlist) = off - 1; 578 PadlistMAX (padlist) = off - 1;
530 579
580#endif
581
582 newPADLIST (newpadlist);
583#if !PERL_VERSION_ATLEAST(5,15,3)
584 /* Padlists are AvREAL as of 5.15.3. See perl bug #98092 and perl commit 7d953ba. */
585 AvREAL_off (newpadlist);
586#endif
587
531 /* Already extended to 2 elements by newPADLIST. */ 588 /* Already extended to 2 elements by newPADLIST. */
532 PadlistMAX (newpadlist) = 1; 589 PadlistMAX (newpadlist) = 1;
533 PadlistNAMES (newpadlist) = (PADNAMELIST *)SvREFCNT_inc_NN (PadlistNAMES (padlist)); 590
591 padnames = PadlistNAMES (padlist);
592 ++PadnamelistREFCNT (padnames);
593 PadlistNAMES (newpadlist) = padnames;
594
534 PadlistARRAY (newpadlist)[1] = newpad; 595 PadlistARRAY (newpadlist)[1] = newpad;
535 596
536 return newpadlist; 597 return newpadlist;
537} 598}
538 599
546 607
547 while (i > 0) /* special-case index 0 */ 608 while (i > 0) /* special-case index 0 */
548 { 609 {
549 /* we try to be extra-careful here */ 610 /* we try to be extra-careful here */
550 PAD *pad = PadlistARRAY (padlist)[i--]; 611 PAD *pad = PadlistARRAY (padlist)[i--];
612
613 if (pad)
614 {
551 I32 j = PadMAX (pad); 615 I32 j = PadMAX (pad);
552 616
553 while (j >= 0) 617 while (j >= 0)
554 SvREFCNT_dec (PadARRAY (pad)[j--]); 618 SvREFCNT_dec (PadARRAY (pad)[j--]);
555 619
556 PadMAX (pad) = -1; 620 PadMAX (pad) = -1;
557 SvREFCNT_dec (pad); 621 SvREFCNT_dec (pad);
622 }
558 } 623 }
559 624
560 SvREFCNT_dec (PadlistNAMES (padlist)); 625 PadnamelistREFCNT_dec (PadlistNAMES (padlist));
561 626
562#ifdef NEWPADAPI 627#if NEWPADAPI
563 Safefree (PadlistARRAY (padlist)); 628 Safefree (PadlistARRAY (padlist));
564 Safefree (padlist); 629 Safefree (padlist);
565#else 630#else
566 AvFILLp (padlist) = -1; 631 AvFILLp (padlist) = -1;
567 AvREAL_off (padlist); 632 AvREAL_off (padlist);
661 tmp.sv_u = a->sv_u; a->sv_u = b->sv_u; b->sv_u = tmp.sv_u; 726 tmp.sv_u = a->sv_u; a->sv_u = b->sv_u; b->sv_u = tmp.sv_u;
662 727
663 /* if SvANY points to the head, we need to adjust the pointers, 728 /* if SvANY points to the head, we need to adjust the pointers,
664 * as the pointer for a still points to b, and maybe vice versa. 729 * as the pointer for a still points to b, and maybe vice versa.
665 */ 730 */
666 #define svany_in_head(type) \ 731 U32 svany_in_head_set = (1 << SVt_NULL) | (1 << SVt_BIND) | (1 << SVt_IV) | (1 << SVt_RV);
667 (((1 << SVt_NULL) | (1 << SVt_BIND) | (1 << SVt_IV) | (1 << SVt_RV)) & (1 << (type))) 732 #if NVSIZE <= IVSIZE && PERL_VERSION_ATLEAST(5,22,0)
733 svany_in_head_set |= 1 << SVt_NV;
734 #endif
735
736 #define svany_in_head(type) (svany_in_head_set & (1 << (type)))
668 737
669 if (svany_in_head (SvTYPE (a))) 738 if (svany_in_head (SvTYPE (a)))
670 SvANY (a) = (void *)((PTRV)SvANY (a) - (PTRV)b + (PTRV)a); 739 SvANY (a) = (void *)((PTRV)SvANY (a) - (PTRV)b + (PTRV)a);
671 740
672 if (svany_in_head (SvTYPE (b))) 741 if (svany_in_head (SvTYPE (b)))
675#endif 744#endif
676} 745}
677 746
678/* swap sv heads, at least logically */ 747/* swap sv heads, at least logically */
679static void 748static void
680swap_svs (pTHX_ Coro__State c) 749swap_svs_enter (pTHX_ Coro__State c)
681{ 750{
682 int i; 751 int i;
683 752
684 for (i = 0; i <= AvFILLp (c->swap_sv); i += 2) 753 for (i = 0; i <= AvFILLp (c->swap_sv); i += 2)
685 swap_sv (AvARRAY (c->swap_sv)[i], AvARRAY (c->swap_sv)[i + 1]); 754 swap_sv (AvARRAY (c->swap_sv)[i], AvARRAY (c->swap_sv)[i + 1]);
686} 755}
687 756
757static void
758swap_svs_leave (pTHX_ Coro__State c)
759{
760 int i;
761
762 for (i = AvFILLp (c->swap_sv) - 1; i >= 0; i -= 2)
763 swap_sv (AvARRAY (c->swap_sv)[i], AvARRAY (c->swap_sv)[i + 1]);
764}
765
688#define SWAP_SVS(coro) \ 766#define SWAP_SVS_ENTER(coro) \
689 if (ecb_expect_false ((coro)->swap_sv)) \ 767 if (ecb_expect_false ((coro)->swap_sv)) \
690 swap_svs (aTHX_ (coro)) 768 swap_svs_enter (aTHX_ (coro))
769
770#define SWAP_SVS_LEAVE(coro) \
771 if (ecb_expect_false ((coro)->swap_sv)) \
772 swap_svs_leave (aTHX_ (coro))
691 773
692static void 774static void
693on_enterleave_call (pTHX_ SV *cb); 775on_enterleave_call (pTHX_ SV *cb);
694 776
695static void 777static void
740 822
741 for (i = 0; i <= AvFILLp (c->on_enter); ++i) 823 for (i = 0; i <= AvFILLp (c->on_enter); ++i)
742 on_enterleave_call (aTHX_ AvARRAY (c->on_enter)[i]); 824 on_enterleave_call (aTHX_ AvARRAY (c->on_enter)[i]);
743 } 825 }
744 826
827 if (ecb_expect_false (c->on_enter_xs))
828 {
829 int i;
830
831 for (i = 0; i <= AvFILLp (c->on_enter_xs); i += 2)
832 ((coro_enterleave_hook)AvARRAY (c->on_enter_xs)[i]) (aTHX_ AvARRAY (c->on_enter_xs)[i + 1]);
833 }
834
745 SWAP_SVS (c); 835 SWAP_SVS_ENTER (c);
746} 836}
747 837
748static void 838static void
749save_perl (pTHX_ Coro__State c) 839save_perl (pTHX_ Coro__State c)
750{ 840{
751 SWAP_SVS (c); 841 SWAP_SVS_LEAVE (c);
842
843 if (ecb_expect_false (c->on_leave_xs))
844 {
845 int i;
846
847 for (i = AvFILLp (c->on_leave_xs) - 1; i >= 0; i -= 2)
848 ((coro_enterleave_hook)AvARRAY (c->on_leave_xs)[i]) (aTHX_ AvARRAY (c->on_leave_xs)[i + 1]);
849 }
752 850
753 if (ecb_expect_false (c->on_leave)) 851 if (ecb_expect_false (c->on_leave))
754 { 852 {
755 int i; 853 int i;
756 854
883#endif 981#endif
884 982
885 New(54,PL_savestack,24,ANY); 983 New(54,PL_savestack,24,ANY);
886 PL_savestack_ix = 0; 984 PL_savestack_ix = 0;
887 PL_savestack_max = 24; 985 PL_savestack_max = 24;
986#if PERL_VERSION_ATLEAST (5,24,0)
987 /* perl 5.24 moves SS_MAXPUSH optimisation from */
988 /* the header macros to PL_savestack_max */
989 PL_savestack_max -= SS_MAXPUSH;
990#endif
888 991
889#if !PERL_VERSION_ATLEAST (5,10,0) 992#if !PERL_VERSION_ATLEAST (5,10,0)
890 New(54,PL_retstack,4,OP*); 993 New(54,PL_retstack,4,OP*);
891 PL_retstack_ix = 0; 994 PL_retstack_ix = 0;
892 PL_retstack_max = 4; 995 PL_retstack_max = 4;
958 } 1061 }
959 1062
960 return rss; 1063 return rss;
961} 1064}
962 1065
963/** coroutine stack handling ************************************************/ 1066/** provide custom get/set/clear methods for %SIG elements ******************/
964
965static int (*orig_sigelem_get) (pTHX_ SV *sv, MAGIC *mg);
966static int (*orig_sigelem_set) (pTHX_ SV *sv, MAGIC *mg);
967static int (*orig_sigelem_clr) (pTHX_ SV *sv, MAGIC *mg);
968 1067
969/* apparently < 5.8.8 */ 1068/* apparently < 5.8.8 */
970#ifndef MgPV_nolen_const 1069#ifndef MgPV_nolen_const
971#define MgPV_nolen_const(mg) (((((int)(mg)->mg_len)) == HEf_SVKEY) ? \ 1070#define MgPV_nolen_const(mg) (((((int)(mg)->mg_len)) == HEf_SVKEY) ? \
972 SvPV_nolen((SV*)((mg)->mg_ptr)) : \ 1071 SvPV_nolen((SV*)((mg)->mg_ptr)) : \
973 (const char*)(mg)->mg_ptr) 1072 (const char*)(mg)->mg_ptr)
974#endif 1073#endif
1074
1075/* this will be a patched copy of PL_vtbl_sigelem */
1076static MGVTBL coro_sigelem_vtbl;
1077
1078static int ecb_cold
1079coro_sig_copy (pTHX_ SV *sv, MAGIC *mg, SV *nsv, const char *name, I32 namlen)
1080{
1081 char *key = SvPV_nolen ((SV *)name);
1082
1083 /* do what mg_copy normally does */
1084 sv_magic (nsv, mg->mg_obj, PERL_MAGIC_sigelem, name, namlen);
1085 assert (mg_find (nsv, PERL_MAGIC_sigelem)->mg_virtual == &PL_vtbl_sigelem);
1086
1087 /* patch sigelem vtbl, but only for __WARN__ and __DIE__ */
1088 if (*key == '_'
1089 && (strEQ (key, "__DIE__")
1090 || strEQ (key, "__WARN__")))
1091 mg_find (nsv, PERL_MAGIC_sigelem)->mg_virtual = &coro_sigelem_vtbl;
1092
1093 return 1;
1094}
1095
1096/* perl does not have a %SIG vtbl, we provide one so we can override */
1097/* the magic vtbl for the __DIE__ and __WARN__ members */
1098static const MGVTBL coro_sig_vtbl = {
1099 0, 0, 0, 0, 0,
1100 coro_sig_copy
1101};
975 1102
976/* 1103/*
977 * This overrides the default magic get method of %SIG elements. 1104 * This overrides the default magic get method of %SIG elements.
978 * The original one doesn't provide for reading back of PL_diehook/PL_warnhook 1105 * The original one doesn't provide for reading back of PL_diehook/PL_warnhook
979 * and instead of trying to save and restore the hash elements (extremely slow), 1106 * and instead of trying to save and restore the hash elements (extremely slow),
981 */ 1108 */
982static int ecb_cold 1109static int ecb_cold
983coro_sigelem_get (pTHX_ SV *sv, MAGIC *mg) 1110coro_sigelem_get (pTHX_ SV *sv, MAGIC *mg)
984{ 1111{
985 const char *s = MgPV_nolen_const (mg); 1112 const char *s = MgPV_nolen_const (mg);
1113 /* the key must be either __DIE__ or __WARN__ here */
1114 SV **svp = s[2] == 'D' ? &PL_diehook : &PL_warnhook;
986 1115
987 if (*s == '_') 1116 SV *ssv;
988 {
989 SV **svp = 0;
990 1117
991 if (strEQ (s, "__DIE__" )) svp = &PL_diehook;
992 if (strEQ (s, "__WARN__")) svp = &PL_warnhook;
993
994 if (svp) 1118 if (!*svp)
995 {
996 SV *ssv;
997
998 if (!*svp)
999 ssv = &PL_sv_undef; 1119 ssv = &PL_sv_undef;
1000 else if (SvTYPE (*svp) == SVt_PVCV) /* perlio directly stores a CV in warnhook. ugh. */ 1120 else if (SvTYPE (*svp) == SVt_PVCV) /* perlio directly stores a CV in warnhook. ugh. */
1001 ssv = sv_2mortal (newRV_inc (*svp)); 1121 ssv = sv_2mortal (newRV_inc (*svp));
1002 else 1122 else
1003 ssv = *svp; 1123 ssv = *svp;
1004 1124
1005 sv_setsv (sv, ssv); 1125 sv_setsv (sv, ssv);
1006 return 0; 1126 return 0;
1007 }
1008 }
1009
1010 return orig_sigelem_get ? orig_sigelem_get (aTHX_ sv, mg) : 0;
1011} 1127}
1012 1128
1013static int ecb_cold 1129static int ecb_cold
1014coro_sigelem_clr (pTHX_ SV *sv, MAGIC *mg) 1130coro_sigelem_clr (pTHX_ SV *sv, MAGIC *mg)
1015{ 1131{
1016 const char *s = MgPV_nolen_const (mg); 1132 const char *s = MgPV_nolen_const (mg);
1133 /* the key must be either __DIE__ or __WARN__ here */
1134 SV **svp = s[2] == 'D' ? &PL_diehook : &PL_warnhook;
1017 1135
1018 if (*s == '_')
1019 {
1020 SV **svp = 0;
1021
1022 if (strEQ (s, "__DIE__" )) svp = &PL_diehook;
1023 if (strEQ (s, "__WARN__")) svp = &PL_warnhook;
1024
1025 if (svp)
1026 {
1027 SV *old = *svp; 1136 SV *old = *svp;
1028 *svp = 0; 1137 *svp = 0;
1029 SvREFCNT_dec (old); 1138 SvREFCNT_dec (old);
1030 return 0; 1139 return 0;
1031 }
1032 }
1033
1034 return orig_sigelem_clr ? orig_sigelem_clr (aTHX_ sv, mg) : 0;
1035} 1140}
1036 1141
1037static int ecb_cold 1142static int ecb_cold
1038coro_sigelem_set (pTHX_ SV *sv, MAGIC *mg) 1143coro_sigelem_set (pTHX_ SV *sv, MAGIC *mg)
1039{ 1144{
1040 const char *s = MgPV_nolen_const (mg); 1145 const char *s = MgPV_nolen_const (mg);
1146 /* the key must be either __DIE__ or __WARN__ here */
1147 SV **svp = s[2] == 'D' ? &PL_diehook : &PL_warnhook;
1041 1148
1042 if (*s == '_')
1043 {
1044 SV **svp = 0;
1045
1046 if (strEQ (s, "__DIE__" )) svp = &PL_diehook;
1047 if (strEQ (s, "__WARN__")) svp = &PL_warnhook;
1048
1049 if (svp)
1050 {
1051 SV *old = *svp; 1149 SV *old = *svp;
1052 *svp = SvOK (sv) ? newSVsv (sv) : 0; 1150 *svp = SvOK (sv) ? newSVsv (sv) : 0;
1053 SvREFCNT_dec (old); 1151 SvREFCNT_dec (old);
1054 return 0; 1152 return 0;
1055 }
1056 }
1057
1058 return orig_sigelem_set ? orig_sigelem_set (aTHX_ sv, mg) : 0;
1059} 1153}
1060 1154
1061static void 1155static void
1062prepare_nop (pTHX_ struct coro_transfer_args *ta) 1156prepare_nop (pTHX_ struct coro_transfer_args *ta)
1063{ 1157{
1074static int 1168static int
1075slf_check_repeat (pTHX_ struct CoroSLF *frame) 1169slf_check_repeat (pTHX_ struct CoroSLF *frame)
1076{ 1170{
1077 return 1; 1171 return 1;
1078} 1172}
1173
1174/** coroutine stack handling ************************************************/
1079 1175
1080static UNOP init_perl_op; 1176static UNOP init_perl_op;
1081 1177
1082ecb_noinline static void /* noinline to keep it out of the transfer fast path */ 1178ecb_noinline static void /* noinline to keep it out of the transfer fast path */
1083init_perl (pTHX_ struct coro *coro) 1179init_perl (pTHX_ struct coro *coro)
1109 1205
1110 GvSV (PL_defgv) = newSV (0); 1206 GvSV (PL_defgv) = newSV (0);
1111 GvAV (PL_defgv) = coro->args; coro->args = 0; 1207 GvAV (PL_defgv) = coro->args; coro->args = 0;
1112 GvSV (PL_errgv) = newSV (0); 1208 GvSV (PL_errgv) = newSV (0);
1113 GvSV (irsgv) = newSVpvn ("\n", 1); sv_magic (GvSV (irsgv), (SV *)irsgv, PERL_MAGIC_sv, "/", 0); 1209 GvSV (irsgv) = newSVpvn ("\n", 1); sv_magic (GvSV (irsgv), (SV *)irsgv, PERL_MAGIC_sv, "/", 0);
1114 GvHV (PL_hintgv) = 0; 1210 GvHV (PL_hintgv) = newHV ();
1211#if PERL_VERSION_ATLEAST (5,10,0)
1212 hv_magic (GvHV (PL_hintgv), 0, PERL_MAGIC_hints);
1213#endif
1115 PL_rs = newSVsv (GvSV (irsgv)); 1214 PL_rs = newSVsv (GvSV (irsgv));
1116 PL_defoutgv = (GV *)SvREFCNT_inc_NN (stdoutgv); 1215 PL_defoutgv = (GV *)SvREFCNT_inc_NN (stdoutgv);
1117 1216
1118 { 1217 {
1119 dSP; 1218 dSP;
1123 myop.op_next = Nullop; 1222 myop.op_next = Nullop;
1124 myop.op_type = OP_ENTERSUB; 1223 myop.op_type = OP_ENTERSUB;
1125 myop.op_flags = OPf_WANT_VOID; 1224 myop.op_flags = OPf_WANT_VOID;
1126 1225
1127 PUSHMARK (SP); 1226 PUSHMARK (SP);
1128 PUSHs ((SV *)coro->startcv); 1227 XPUSHs ((SV *)coro->startcv);
1129 PUTBACK; 1228 PUTBACK;
1130 PL_op = (OP *)&myop; 1229 PL_op = (OP *)&myop;
1131 PL_op = PL_ppaddr[OP_ENTERSUB](aTHX); 1230 PL_op = PL_ppaddr[OP_ENTERSUB](aTHX);
1132 } 1231 }
1133 1232
1147 PL_op = (OP *)&init_perl_op; 1246 PL_op = (OP *)&init_perl_op;
1148 1247
1149 /* copy throw, in case it was set before init_perl */ 1248 /* copy throw, in case it was set before init_perl */
1150 CORO_THROW = coro->except; 1249 CORO_THROW = coro->except;
1151 1250
1152 SWAP_SVS (coro); 1251 SWAP_SVS_ENTER (coro);
1153 1252
1154 if (ecb_expect_false (enable_times)) 1253 if (ecb_expect_false (enable_times))
1155 { 1254 {
1156 coro_times_update (); 1255 coro_times_update ();
1157 coro_times_sub (coro); 1256 coro_times_sub (coro);
1195 /* this will cause transfer_check to croak on block*/ 1294 /* this will cause transfer_check to croak on block*/
1196 SvRV_set (coro_current, (SV *)coro->hv); 1295 SvRV_set (coro_current, (SV *)coro->hv);
1197 1296
1198 load_perl (aTHX_ coro); 1297 load_perl (aTHX_ coro);
1199 1298
1299 /* restore swapped sv's */
1300 SWAP_SVS_LEAVE (coro);
1301
1200 coro_unwind_stacks (aTHX); 1302 coro_unwind_stacks (aTHX);
1201
1202 /* restore swapped sv's */
1203 SWAP_SVS (coro);
1204 1303
1205 coro_destruct_stacks (aTHX); 1304 coro_destruct_stacks (aTHX);
1206 1305
1207 /* now save some sv's to be free'd later */ 1306 /* now save some sv's to be free'd later */
1208 svf [0] = GvSV (PL_defgv); 1307 svf [0] = GvSV (PL_defgv);
1229 1328
1230 SvREFCNT_dec (coro->saved_deffh); 1329 SvREFCNT_dec (coro->saved_deffh);
1231 SvREFCNT_dec (coro->rouse_cb); 1330 SvREFCNT_dec (coro->rouse_cb);
1232 SvREFCNT_dec (coro->invoke_cb); 1331 SvREFCNT_dec (coro->invoke_cb);
1233 SvREFCNT_dec (coro->invoke_av); 1332 SvREFCNT_dec (coro->invoke_av);
1333 SvREFCNT_dec (coro->on_enter_xs);
1334 SvREFCNT_dec (coro->on_leave_xs);
1234 } 1335 }
1235} 1336}
1236 1337
1237ecb_inline void 1338ecb_inline void
1238free_coro_mortal (pTHX) 1339free_coro_mortal (pTHX)
1279 av_push (av, SvREFCNT_inc_NN (*bot++)); 1380 av_push (av, SvREFCNT_inc_NN (*bot++));
1280 1381
1281 PL_runops = RUNOPS_DEFAULT; 1382 PL_runops = RUNOPS_DEFAULT;
1282 ENTER; 1383 ENTER;
1283 SAVETMPS; 1384 SAVETMPS;
1385 PUSHMARK (SP);
1284 EXTEND (SP, 3); 1386 EXTEND (SP, 3);
1285 PUSHMARK (SP);
1286 PUSHs (&PL_sv_no); 1387 PUSHs (&PL_sv_no);
1287 PUSHs (fullname); 1388 PUSHs (fullname);
1288 PUSHs (sv_2mortal (newRV_noinc ((SV *)av))); 1389 PUSHs (sv_2mortal (newRV_noinc ((SV *)av)));
1289 PUTBACK; 1390 PUTBACK;
1290 cb = hv_fetch ((HV *)SvRV (coro_current), "_trace_sub_cb", sizeof ("_trace_sub_cb") - 1, 0); 1391 cb = hv_fetch ((HV *)SvRV (coro_current), "_trace_sub_cb", sizeof ("_trace_sub_cb") - 1, 0);
1301 1402
1302 if (PL_curcop != &PL_compiling) 1403 if (PL_curcop != &PL_compiling)
1303 { 1404 {
1304 SV **cb; 1405 SV **cb;
1305 1406
1306 if (oldcxix != cxstack_ix && cctx_current->flags & CC_TRACE_SUB) 1407 if (oldcxix != cxstack_ix && cctx_current->flags & CC_TRACE_SUB && cxstack_ix >= 0)
1307 { 1408 {
1308 PERL_CONTEXT *cx = &cxstack[cxstack_ix]; 1409 PERL_CONTEXT *cx = &cxstack[cxstack_ix];
1309 1410
1310 if (CxTYPE (cx) == CXt_SUB && oldcxix < cxstack_ix) 1411 if (CxTYPE (cx) == CXt_SUB && oldcxix < cxstack_ix)
1311 { 1412 {
1317 gv_efullname3 (fullname, gv, 0); 1418 gv_efullname3 (fullname, gv, 0);
1318 1419
1319 PL_runops = RUNOPS_DEFAULT; 1420 PL_runops = RUNOPS_DEFAULT;
1320 ENTER; 1421 ENTER;
1321 SAVETMPS; 1422 SAVETMPS;
1423 PUSHMARK (SP);
1322 EXTEND (SP, 3); 1424 EXTEND (SP, 3);
1323 PUSHMARK (SP);
1324 PUSHs (&PL_sv_yes); 1425 PUSHs (&PL_sv_yes);
1325 PUSHs (fullname); 1426 PUSHs (fullname);
1326 PUSHs (CxHASARGS (cx) ? sv_2mortal (newRV_inc ((SV *)cx->blk_sub.argarray)) : &PL_sv_undef); 1427 PUSHs (CxHASARGS (cx) ? sv_2mortal (newRV_inc (SUB_ARGARRAY)) : &PL_sv_undef);
1327 PUTBACK; 1428 PUTBACK;
1328 cb = hv_fetch ((HV *)SvRV (coro_current), "_trace_sub_cb", sizeof ("_trace_sub_cb") - 1, 0); 1429 cb = hv_fetch ((HV *)SvRV (coro_current), "_trace_sub_cb", sizeof ("_trace_sub_cb") - 1, 0);
1329 if (cb) call_sv (*cb, G_KEEPERR | G_EVAL | G_VOID | G_DISCARD); 1430 if (cb) call_sv (*cb, G_KEEPERR | G_EVAL | G_VOID | G_DISCARD);
1330 SPAGAIN; 1431 SPAGAIN;
1331 FREETMPS; 1432 FREETMPS;
1341 dSP; 1442 dSP;
1342 1443
1343 PL_runops = RUNOPS_DEFAULT; 1444 PL_runops = RUNOPS_DEFAULT;
1344 ENTER; 1445 ENTER;
1345 SAVETMPS; 1446 SAVETMPS;
1346 EXTEND (SP, 3);
1347 PL_runops = RUNOPS_DEFAULT;
1348 PUSHMARK (SP); 1447 PUSHMARK (SP);
1448 EXTEND (SP, 2);
1349 PUSHs (sv_2mortal (newSVpv (OutCopFILE (oldcop), 0))); 1449 PUSHs (sv_2mortal (newSVpv (OutCopFILE (oldcop), 0)));
1350 PUSHs (sv_2mortal (newSViv (CopLINE (oldcop)))); 1450 PUSHs (sv_2mortal (newSViv (CopLINE (oldcop))));
1351 PUTBACK; 1451 PUTBACK;
1352 cb = hv_fetch ((HV *)SvRV (coro_current), "_trace_line_cb", sizeof ("_trace_line_cb") - 1, 0); 1452 cb = hv_fetch ((HV *)SvRV (coro_current), "_trace_line_cb", sizeof ("_trace_line_cb") - 1, 0);
1353 if (cb) call_sv (*cb, G_KEEPERR | G_EVAL | G_VOID | G_DISCARD); 1453 if (cb) call_sv (*cb, G_KEEPERR | G_EVAL | G_VOID | G_DISCARD);
1453 * coro body here, as perl_run destroys these. Likewise, we cannot catch 1553 * coro body here, as perl_run destroys these. Likewise, we cannot catch
1454 * runtime errors here, as this is just a random interpreter, not a thread. 1554 * runtime errors here, as this is just a random interpreter, not a thread.
1455 */ 1555 */
1456 1556
1457 /* 1557 /*
1558 * pp_entersub in 5.24 no longer ENTERs, but perl_destruct
1559 * requires PL_scopestack_ix, so do it here if required.
1560 */
1561 if (!PL_scopestack_ix)
1562 ENTER;
1563
1564 /*
1458 * If perl-run returns we assume exit() was being called or the coro 1565 * If perl-run returns we assume exit() was being called or the coro
1459 * fell off the end, which seems to be the only valid (non-bug) 1566 * fell off the end, which seems to be the only valid (non-bug)
1460 * reason for perl_run to return. We try to mimic whatever perl is normally 1567 * reason for perl_run to return. We try to mimic whatever perl is normally
1461 * doing in that case. YMMV. 1568 * doing in that case. YMMV.
1462 */ 1569 */
1888 /* nothing to schedule: call the idle handler */ 1995 /* nothing to schedule: call the idle handler */
1889 if (SvROK (sv_idle) 1996 if (SvROK (sv_idle)
1890 && SvOBJECT (SvRV (sv_idle))) 1997 && SvOBJECT (SvRV (sv_idle)))
1891 { 1998 {
1892 if (SvRV (sv_idle) == SvRV (coro_current)) 1999 if (SvRV (sv_idle) == SvRV (coro_current))
2000 {
2001 require_pv ("Carp");
2002
2003 {
2004 dSP;
2005
2006 ENTER;
2007 SAVETMPS;
2008
2009 PUSHMARK (SP);
1893 croak ("FATAL: $Coro::IDLE blocked itself - did you try to block inside an event loop callback? Caught"); 2010 XPUSHs (sv_2mortal (newSVpv ("FATAL: $Coro::idle blocked itself - did you try to block inside an event loop callback? Caught", 0)));
2011 PUTBACK;
2012 call_pv ("Carp::confess", G_VOID | G_DISCARD);
2013
2014 FREETMPS;
2015 LEAVE;
2016 }
2017 }
1894 2018
1895 ++coro_nready; /* hack so that api_ready doesn't invoke ready hook */ 2019 ++coro_nready; /* hack so that api_ready doesn't invoke ready hook */
1896 api_ready (aTHX_ SvRV (sv_idle)); 2020 api_ready (aTHX_ SvRV (sv_idle));
1897 --coro_nready; 2021 --coro_nready;
1898 } 2022 }
2096 AV *od = coro->on_destroy; 2220 AV *od = coro->on_destroy;
2097 2221
2098 if (!od) 2222 if (!od)
2099 return; 2223 return;
2100 2224
2225 coro->on_destroy = 0;
2226 sv_2mortal ((SV *)od);
2227
2101 while (AvFILLp (od) >= 0) 2228 while (AvFILLp (od) >= 0)
2102 { 2229 {
2103 SV *cb = sv_2mortal (av_pop (od)); 2230 SV *cb = sv_2mortal (av_pop (od));
2104 2231
2105 /* coro hv's (and only hv's at the moment) are supported as well */ 2232 /* coro hv's (and only hv's at the moment) are supported as well */
2199 slf_init_terminate_cancel_common (aTHX_ frame, coro_hv); 2326 slf_init_terminate_cancel_common (aTHX_ frame, coro_hv);
2200 } 2327 }
2201 else 2328 else
2202 { 2329 {
2203 struct coro *self = SvSTATE_current; 2330 struct coro *self = SvSTATE_current;
2331
2332 if (!self)
2333 croak ("Coro::cancel called outside of thread content,");
2204 2334
2205 /* otherwise we cancel directly, purely for speed reasons 2335 /* otherwise we cancel directly, purely for speed reasons
2206 * unfortunately, this requires some magic trickery, as 2336 * unfortunately, this requires some magic trickery, as
2207 * somebody else could cancel us, so we have to fight the cancellation. 2337 * somebody else could cancel us, so we have to fight the cancellation.
2208 * this is ugly, and hopefully fully worth the extra speed. 2338 * this is ugly, and hopefully fully worth the extra speed.
2241safe_cancel (pTHX_ struct coro *coro, SV **arg, int items) 2371safe_cancel (pTHX_ struct coro *coro, SV **arg, int items)
2242{ 2372{
2243 if (coro->cctx) 2373 if (coro->cctx)
2244 croak ("coro inside C callback, unable to cancel at this time, caught"); 2374 croak ("coro inside C callback, unable to cancel at this time, caught");
2245 2375
2246 if (coro->flags & CF_NEW) 2376 if (coro->flags & (CF_NEW | CF_ZOMBIE))
2247 { 2377 {
2248 coro_set_status (aTHX_ coro, arg, items); 2378 coro_set_status (aTHX_ coro, arg, items);
2249 coro_state_destroy (aTHX_ coro); 2379 coro_state_destroy (aTHX_ coro);
2250 } 2380 }
2251 else 2381 else
2318 else 2448 else
2319 { 2449 {
2320 av_clear (GvAV (PL_defgv)); 2450 av_clear (GvAV (PL_defgv));
2321 hv_store (hv, "desc", sizeof ("desc") - 1, SvREFCNT_inc_NN (sv_async_pool_idle), 0); 2451 hv_store (hv, "desc", sizeof ("desc") - 1, SvREFCNT_inc_NN (sv_async_pool_idle), 0);
2322 2452
2453 if (ecb_expect_false (coro->swap_sv))
2454 {
2455 SWAP_SVS_LEAVE (coro);
2456 SvREFCNT_dec_NN (coro->swap_sv);
2457 coro->swap_sv = 0;
2458 }
2459
2323 coro->prio = 0; 2460 coro->prio = 0;
2324 2461
2325 if (coro->cctx && (coro->cctx->flags & CC_TRACE)) 2462 if (ecb_expect_false (coro->cctx) && ecb_expect_false (coro->cctx->flags & CC_TRACE))
2326 api_trace (aTHX_ coro_current, 0); 2463 api_trace (aTHX_ coro_current, 0);
2327 2464
2328 frame->prepare = prepare_schedule; 2465 frame->prepare = prepare_schedule;
2329 av_push (av_async_pool, SvREFCNT_inc (hv)); 2466 av_push (av_async_pool, SvREFCNT_inc (hv));
2330 } 2467 }
2550 2687
2551/* "undo"/cancel a running slf call - used when cancelling a coro, mainly */ 2688/* "undo"/cancel a running slf call - used when cancelling a coro, mainly */
2552static void 2689static void
2553slf_destroy (pTHX_ struct coro *coro) 2690slf_destroy (pTHX_ struct coro *coro)
2554{ 2691{
2555 /* this callback is reserved for slf functions needing to do cleanup */ 2692 struct CoroSLF frame = coro->slf_frame;
2556 if (coro->slf_frame.destroy && coro->slf_frame.prepare && !PL_dirty)
2557 coro->slf_frame.destroy (aTHX_ &coro->slf_frame);
2558 2693
2559 /* 2694 /*
2560 * The on_destroy above most likely is from an SLF call. 2695 * The on_destroy below most likely is from an SLF call.
2561 * Since by definition the SLF call will not finish when we destroy 2696 * Since by definition the SLF call will not finish when we destroy
2562 * the coro, we will have to force-finish it here, otherwise 2697 * the coro, we will have to force-finish it here, otherwise
2563 * cleanup functions cannot call SLF functions. 2698 * cleanup functions cannot call SLF functions.
2564 */ 2699 */
2565 coro->slf_frame.prepare = 0; 2700 coro->slf_frame.prepare = 0;
2701
2702 /* this callback is reserved for slf functions needing to do cleanup */
2703 if (frame.destroy && frame.prepare && !PL_dirty)
2704 frame.destroy (aTHX_ &frame);
2566} 2705}
2567 2706
2568/* 2707/*
2569 * these not obviously related functions are all rolled into one 2708 * these not obviously related functions are all rolled into one
2570 * function to increase chances that they all will call transfer with the same 2709 * function to increase chances that they all will call transfer with the same
2753static void 2892static void
2754coro_pop_on_leave (pTHX_ void *coro) 2893coro_pop_on_leave (pTHX_ void *coro)
2755{ 2894{
2756 SV *cb = coro_avp_pop_and_free (aTHX_ &((struct coro *)coro)->on_leave); 2895 SV *cb = coro_avp_pop_and_free (aTHX_ &((struct coro *)coro)->on_leave);
2757 on_enterleave_call (aTHX_ sv_2mortal (cb)); 2896 on_enterleave_call (aTHX_ sv_2mortal (cb));
2897}
2898
2899static void
2900enterleave_hook_xs (pTHX_ struct coro *coro, AV **avp, coro_enterleave_hook hook, void *arg)
2901{
2902 if (!hook)
2903 return;
2904
2905 if (!*avp)
2906 {
2907 *avp = newAV ();
2908 AvREAL_off (*avp);
2909 }
2910
2911 av_push (*avp, (SV *)hook);
2912 av_push (*avp, (SV *)arg);
2913}
2914
2915static void
2916enterleave_unhook_xs (pTHX_ struct coro *coro, AV **avp, coro_enterleave_hook hook, int execute)
2917{
2918 AV *av = *avp;
2919 int i;
2920
2921 if (!av)
2922 return;
2923
2924 for (i = AvFILLp (av) - 1; i >= 0; i -= 2)
2925 if (AvARRAY (av)[i] == (SV *)hook)
2926 {
2927 if (execute)
2928 hook (aTHX_ (void *)AvARRAY (av)[i + 1]);
2929
2930 memmove (AvARRAY (av) + i, AvARRAY (av) + i + 2, AvFILLp (av) - i - 1);
2931 av_pop (av);
2932 av_pop (av);
2933 break;
2934 }
2935
2936 if (AvFILLp (av) >= 0)
2937 {
2938 *avp = 0;
2939 SvREFCNT_dec_NN (av);
2940 }
2941}
2942
2943static void
2944api_enterleave_hook (pTHX_ SV *coro_sv, coro_enterleave_hook enter, void *enter_arg, coro_enterleave_hook leave, void *leave_arg)
2945{
2946 struct coro *coro = SvSTATE (coro_sv);
2947
2948 if (SvSTATE_current == coro)
2949 if (enter)
2950 enter (aTHX_ enter_arg);
2951
2952 enterleave_hook_xs (aTHX_ coro, &coro->on_enter_xs, enter, enter_arg);
2953 enterleave_hook_xs (aTHX_ coro, &coro->on_leave_xs, leave, leave_arg);
2954}
2955
2956static void
2957api_enterleave_unhook (pTHX_ SV *coro_sv, coro_enterleave_hook enter, coro_enterleave_hook leave)
2958{
2959 struct coro *coro = SvSTATE (coro_sv);
2960
2961 enterleave_unhook_xs (aTHX_ coro, &coro->on_enter_xs, enter, 0);
2962 enterleave_unhook_xs (aTHX_ coro, &coro->on_leave_xs, leave, SvSTATE_current == coro);
2963}
2964
2965static void
2966savedestructor_unhook_enter (pTHX_ coro_enterleave_hook enter)
2967{
2968 struct coro *coro = SvSTATE_current;
2969
2970 enterleave_unhook_xs (aTHX_ coro, &coro->on_enter_xs, enter, 0);
2971}
2972
2973static void
2974savedestructor_unhook_leave (pTHX_ coro_enterleave_hook leave)
2975{
2976 struct coro *coro = SvSTATE_current;
2977
2978 enterleave_unhook_xs (aTHX_ coro, &coro->on_leave_xs, leave, 1);
2979}
2980
2981static void
2982api_enterleave_scope_hook (pTHX_ coro_enterleave_hook enter, void *enter_arg, coro_enterleave_hook leave, void *leave_arg)
2983{
2984 api_enterleave_hook (aTHX_ coro_current, enter, enter_arg, leave, leave_arg);
2985
2986 /* this ought to be much cheaper than malloc + a single destructor call */
2987 if (enter) SAVEDESTRUCTOR_X (savedestructor_unhook_enter, enter);
2988 if (leave) SAVEDESTRUCTOR_X (savedestructor_unhook_leave, leave);
2758} 2989}
2759 2990
2760/*****************************************************************************/ 2991/*****************************************************************************/
2761/* PerlIO::cede */ 2992/* PerlIO::cede */
2762 2993
2892 XPUSHs (sv_2mortal (newRV_inc ((SV *)av))); 3123 XPUSHs (sv_2mortal (newRV_inc ((SV *)av)));
2893 PUTBACK; 3124 PUTBACK;
2894 call_sv (cb, G_VOID | G_DISCARD | G_EVAL | G_KEEPERR); 3125 call_sv (cb, G_VOID | G_DISCARD | G_EVAL | G_KEEPERR);
2895 } 3126 }
2896 3127
2897 SvREFCNT_dec (cb); 3128 SvREFCNT_dec_NN (cb);
2898 } 3129 }
2899} 3130}
2900 3131
2901static void 3132static void
2902coro_semaphore_destroy (pTHX_ struct CoroSLF *frame) 3133coro_semaphore_destroy (pTHX_ struct CoroSLF *frame)
2910{ 3141{
2911 AV *av = (AV *)frame->data; 3142 AV *av = (AV *)frame->data;
2912 SV *count_sv = AvARRAY (av)[0]; 3143 SV *count_sv = AvARRAY (av)[0];
2913 SV *coro_hv = SvRV (coro_current); 3144 SV *coro_hv = SvRV (coro_current);
2914 3145
3146 frame->destroy = 0;
3147
2915 /* if we are about to throw, don't actually acquire the lock, just throw */ 3148 /* if we are about to throw, don't actually acquire the lock, just throw */
2916 if (CORO_THROW) 3149 if (ecb_expect_false (CORO_THROW))
3150 {
3151 /* we still might be responsible for the semaphore, so wake up others */
3152 coro_semaphore_adjust (aTHX_ av, 0);
3153
2917 return 0; 3154 return 0;
3155 }
2918 else if (SvIVX (count_sv) > 0) 3156 else if (SvIVX (count_sv) > 0)
2919 { 3157 {
2920 frame->destroy = 0;
2921
2922 if (acquire) 3158 if (acquire)
2923 SvIVX (count_sv) = SvIVX (count_sv) - 1; 3159 SvIVX (count_sv) = SvIVX (count_sv) - 1;
2924 else 3160 else
2925 coro_semaphore_adjust (aTHX_ av, 0); 3161 coro_semaphore_adjust (aTHX_ av, 0);
2926 3162
3037 { 3273 {
3038 api_ready (aTHX_ cb); 3274 api_ready (aTHX_ cb);
3039 sv_setiv (cb, 0); /* signal waiter */ 3275 sv_setiv (cb, 0); /* signal waiter */
3040 } 3276 }
3041 3277
3042 SvREFCNT_dec (cb); 3278 SvREFCNT_dec_NN (cb);
3043 3279
3044 --count; 3280 --count;
3045 } 3281 }
3046} 3282}
3047 3283
3132 } 3368 }
3133 3369
3134 av_push (state, data_sv); 3370 av_push (state, data_sv);
3135 3371
3136 api_ready (aTHX_ coro); 3372 api_ready (aTHX_ coro);
3137 SvREFCNT_dec (coro); 3373 SvREFCNT_dec_NN (coro);
3138 SvREFCNT_dec ((AV *)state); 3374 SvREFCNT_dec_NN ((AV *)state);
3139} 3375}
3140 3376
3141static int 3377static int
3142slf_check_aio_req (pTHX_ struct CoroSLF *frame) 3378slf_check_aio_req (pTHX_ struct CoroSLF *frame)
3143{ 3379{
3148 /* it quickly returns */ 3384 /* it quickly returns */
3149 if (CORO_THROW) 3385 if (CORO_THROW)
3150 return 0; 3386 return 0;
3151 3387
3152 /* one element that is an RV? repeat! */ 3388 /* one element that is an RV? repeat! */
3153 if (AvFILLp (state) == 0 && SvROK (AvARRAY (state)[0])) 3389 if (AvFILLp (state) == 0 && SvTYPE (AvARRAY (state)[0]) != SVt_PV)
3154 return 1; 3390 return 1;
3155 3391
3156 /* restore status */ 3392 /* restore status */
3157 { 3393 {
3158 SV *data_sv = av_pop (state); 3394 SV *data_sv = av_pop (state);
3161 errno = data->errorno; 3397 errno = data->errorno;
3162 PL_laststype = data->laststype; 3398 PL_laststype = data->laststype;
3163 PL_laststatval = data->laststatval; 3399 PL_laststatval = data->laststatval;
3164 PL_statcache = data->statcache; 3400 PL_statcache = data->statcache;
3165 3401
3166 SvREFCNT_dec (data_sv); 3402 SvREFCNT_dec_NN (data_sv);
3167 } 3403 }
3168 3404
3169 /* push result values */ 3405 /* push result values */
3170 { 3406 {
3171 dSP; 3407 dSP;
3360 3596
3361 map_len = load_len + save_len + 16; 3597 map_len = load_len + save_len + 16;
3362 3598
3363 map_base = mmap (0, map_len, PROT_READ | PROT_WRITE | PROT_EXEC, MAP_PRIVATE | MAP_ANONYMOUS, -1, 0); 3599 map_base = mmap (0, map_len, PROT_READ | PROT_WRITE | PROT_EXEC, MAP_PRIVATE | MAP_ANONYMOUS, -1, 0);
3364 3600
3601 if (map_base == (char *)MAP_FAILED)
3602 map_base = mmap (0, map_len, PROT_READ | PROT_WRITE, MAP_PRIVATE | MAP_ANONYMOUS, -1, 0);
3603
3365 assert (("Coro: unable to mmap jit code page, cannot continue.", map_base != (char *)MAP_FAILED)); 3604 assert (("Coro: unable to mmap jit code page, cannot continue.", map_base != (char *)MAP_FAILED));
3366 3605
3367 load_perl_slots = (load_save_perl_slots_type)map_base; 3606 load_perl_slots = (load_save_perl_slots_type)map_base;
3368 memcpy (map_base, load_ptr, load_len); 3607 memcpy (map_base, load_ptr, load_len);
3369 3608
3386 3625
3387PROTOTYPES: DISABLE 3626PROTOTYPES: DISABLE
3388 3627
3389BOOT: 3628BOOT:
3390{ 3629{
3630#define VARx(name,expr,type) if (sizeof (type) < sizeof (expr)) croak ("FATAL: Coro thread context slot '" # name "' too small for this version of perl.");
3631#include "state.h"
3391#ifdef USE_ITHREADS 3632#ifdef USE_ITHREADS
3392# if CORO_PTHREAD 3633# if CORO_PTHREAD
3393 coro_thx = PERL_GET_CONTEXT; 3634 coro_thx = PERL_GET_CONTEXT;
3394# endif 3635# endif
3395#endif 3636#endif
3403 cctx_current = cctx_new_empty (); 3644 cctx_current = cctx_new_empty ();
3404 3645
3405 irsgv = gv_fetchpv ("/" , GV_ADD|GV_NOTQUAL, SVt_PV); 3646 irsgv = gv_fetchpv ("/" , GV_ADD|GV_NOTQUAL, SVt_PV);
3406 stdoutgv = gv_fetchpv ("STDOUT", GV_ADD|GV_NOTQUAL, SVt_PVIO); 3647 stdoutgv = gv_fetchpv ("STDOUT", GV_ADD|GV_NOTQUAL, SVt_PVIO);
3407 3648
3408 orig_sigelem_get = PL_vtbl_sigelem.svt_get; PL_vtbl_sigelem.svt_get = coro_sigelem_get; 3649 {
3409 orig_sigelem_set = PL_vtbl_sigelem.svt_set; PL_vtbl_sigelem.svt_set = coro_sigelem_set; 3650 /*
3410 orig_sigelem_clr = PL_vtbl_sigelem.svt_clear; PL_vtbl_sigelem.svt_clear = coro_sigelem_clr; 3651 * we provide a vtbvl for %SIG magic that replaces PL_vtbl_sig
3652 * by coro_sig_vtbl in hash values.
3653 */
3654 MAGIC *mg = mg_find ((SV *)GvHV (gv_fetchpv ("SIG", GV_ADD | GV_NOTQUAL, SVt_PVHV)), PERL_MAGIC_sig);
3655
3656 /* this only works if perl doesn't have a vtbl for %SIG */
3657 assert (!mg->mg_virtual);
3658
3659 /*
3660 * The irony is that the perl API itself asserts that mg_virtual
3661 * must be non-const, yet perl5porters insisted on marking their
3662 * vtbls as read-only, just to thwart perl modules from patching
3663 * them.
3664 */
3665 mg->mg_virtual = (MGVTBL *)&coro_sig_vtbl;
3666 mg->mg_flags |= MGf_COPY;
3667
3668 coro_sigelem_vtbl = PL_vtbl_sigelem;
3669 coro_sigelem_vtbl.svt_get = coro_sigelem_get;
3670 coro_sigelem_vtbl.svt_set = coro_sigelem_set;
3671 coro_sigelem_vtbl.svt_clear = coro_sigelem_clr;
3672 }
3411 3673
3412 hv_sig = coro_get_hv (aTHX_ "SIG", TRUE);
3413 rv_diehook = newRV_inc ((SV *)gv_fetchpv ("Coro::State::diehook" , 0, SVt_PVCV)); 3674 rv_diehook = newRV_inc ((SV *)gv_fetchpv ("Coro::State::diehook" , 0, SVt_PVCV));
3414 rv_warnhook = newRV_inc ((SV *)gv_fetchpv ("Coro::State::warnhook", 0, SVt_PVCV)); 3675 rv_warnhook = newRV_inc ((SV *)gv_fetchpv ("Coro::State::warnhook", 0, SVt_PVCV));
3415 3676
3416 coro_state_stash = gv_stashpv ("Coro::State", TRUE); 3677 coro_state_stash = gv_stashpv ("Coro::State", TRUE);
3417 3678
3470void 3731void
3471transfer (...) 3732transfer (...)
3472 PROTOTYPE: $$ 3733 PROTOTYPE: $$
3473 CODE: 3734 CODE:
3474 CORO_EXECUTE_SLF_XS (slf_init_transfer); 3735 CORO_EXECUTE_SLF_XS (slf_init_transfer);
3475
3476void
3477_exit (int code)
3478 PROTOTYPE: $
3479 CODE:
3480 _exit (code);
3481 3736
3482SV * 3737SV *
3483clone (Coro::State coro) 3738clone (Coro::State coro)
3484 CODE: 3739 CODE:
3485{ 3740{
3729 if (ecb_expect_false (current == self)) 3984 if (ecb_expect_false (current == self))
3730 coro_times_sub (SvSTATE (coro_current)); 3985 coro_times_sub (SvSTATE (coro_current));
3731} 3986}
3732 3987
3733void 3988void
3734swap_sv (Coro::State coro, SV *sv, SV *swapsv) 3989swap_sv (Coro::State coro, SV *sva, SV *svb)
3735 CODE: 3990 CODE:
3736{ 3991{
3737 struct coro *current = SvSTATE_current; 3992 struct coro *current = SvSTATE_current;
3993 AV *swap_sv;
3994 int i;
3995
3996 sva = SvRV (sva);
3997 svb = SvRV (svb);
3738 3998
3739 if (current == coro) 3999 if (current == coro)
3740 SWAP_SVS (current); 4000 SWAP_SVS_LEAVE (current);
3741 4001
3742 if (!coro->swap_sv) 4002 if (!coro->swap_sv)
3743 coro->swap_sv = newAV (); 4003 coro->swap_sv = newAV ();
3744 4004
4005 swap_sv = coro->swap_sv;
4006
4007 for (i = AvFILLp (swap_sv) - 1; i >= 0; i -= 2)
4008 {
4009 SV *a = AvARRAY (swap_sv)[i ];
4010 SV *b = AvARRAY (swap_sv)[i + 1];
4011
4012 if (a == sva && b == svb)
4013 {
4014 SvREFCNT_dec_NN (a);
4015 SvREFCNT_dec_NN (b);
4016
4017 for (; i <= AvFILLp (swap_sv) - 2; i++)
4018 AvARRAY (swap_sv)[i] = AvARRAY (swap_sv)[i + 2];
4019
4020 AvFILLp (swap_sv) -= 2;
4021
4022 goto removed;
4023 }
4024 }
4025
3745 av_push (coro->swap_sv, SvREFCNT_inc_NN (SvRV (sv ))); 4026 av_push (swap_sv, SvREFCNT_inc_NN (sva));
3746 av_push (coro->swap_sv, SvREFCNT_inc_NN (SvRV (swapsv))); 4027 av_push (swap_sv, SvREFCNT_inc_NN (svb));
4028
4029 removed:
3747 4030
3748 if (current == coro) 4031 if (current == coro)
3749 SWAP_SVS (current); 4032 SWAP_SVS_ENTER (current);
3750} 4033}
3751 4034
3752 4035
3753MODULE = Coro::State PACKAGE = Coro 4036MODULE = Coro::State PACKAGE = Coro
3754 4037
3755BOOT: 4038BOOT:
3756{ 4039{
4040 if (SVt_LAST > 32)
4041 croak ("Coro internal error: SVt_LAST > 32, swap_sv might need adjustment");
4042
3757 sv_pool_rss = coro_get_sv (aTHX_ "Coro::POOL_RSS" , TRUE); 4043 sv_pool_rss = coro_get_sv (aTHX_ "Coro::POOL_RSS" , TRUE);
3758 sv_pool_size = coro_get_sv (aTHX_ "Coro::POOL_SIZE" , TRUE); 4044 sv_pool_size = coro_get_sv (aTHX_ "Coro::POOL_SIZE" , TRUE);
3759 cv_coro_run = get_cv ( "Coro::_coro_run" , GV_ADD); 4045 cv_coro_run = get_cv ( "Coro::_coro_run" , GV_ADD);
3760 coro_current = coro_get_sv (aTHX_ "Coro::current" , FALSE); SvREADONLY_on (coro_current); 4046 coro_current = coro_get_sv (aTHX_ "Coro::current" , FALSE); SvREADONLY_on (coro_current);
3761 av_async_pool = coro_get_av (aTHX_ "Coro::async_pool", TRUE); 4047 av_async_pool = coro_get_av (aTHX_ "Coro::async_pool", TRUE);
3787 coroapi.ready = api_ready; 4073 coroapi.ready = api_ready;
3788 coroapi.is_ready = api_is_ready; 4074 coroapi.is_ready = api_is_ready;
3789 coroapi.nready = coro_nready; 4075 coroapi.nready = coro_nready;
3790 coroapi.current = coro_current; 4076 coroapi.current = coro_current;
3791 4077
4078 coroapi.enterleave_hook = api_enterleave_hook;
4079 coroapi.enterleave_unhook = api_enterleave_unhook;
4080 coroapi.enterleave_scope_hook = api_enterleave_scope_hook;
4081
3792 /*GCoroAPI = &coroapi;*/ 4082 /*GCoroAPI = &coroapi;*/
3793 sv_setiv (sv, (IV)&coroapi); 4083 sv_setiv (sv, PTR2IV (&coroapi));
3794 SvREADONLY_on (sv); 4084 SvREADONLY_on (sv);
3795 } 4085 }
3796} 4086}
3797 4087
3798SV * 4088SV *
3861 4151
3862void 4152void
3863_set_current (SV *current) 4153_set_current (SV *current)
3864 PROTOTYPE: $ 4154 PROTOTYPE: $
3865 CODE: 4155 CODE:
3866 SvREFCNT_dec (SvRV (coro_current)); 4156 SvREFCNT_dec_NN (SvRV (coro_current));
3867 SvRV_set (coro_current, SvREFCNT_inc_NN (SvRV (current))); 4157 SvRV_set (coro_current, SvREFCNT_inc_NN (SvRV (current)));
3868 4158
3869void 4159void
3870_set_readyhook (SV *hook) 4160_set_readyhook (SV *hook)
3871 PROTOTYPE: $ 4161 PROTOTYPE: $
3955 4245
3956 if ((SV *)hv == &PL_sv_undef) 4246 if ((SV *)hv == &PL_sv_undef)
3957 { 4247 {
3958 SV *sv = coro_new (aTHX_ coro_stash, (SV **)&cv_pool_handler, 1, 1); 4248 SV *sv = coro_new (aTHX_ coro_stash, (SV **)&cv_pool_handler, 1, 1);
3959 hv = (HV *)SvREFCNT_inc_NN (SvRV (sv)); 4249 hv = (HV *)SvREFCNT_inc_NN (SvRV (sv));
3960 SvREFCNT_dec (sv); 4250 SvREFCNT_dec_NN (sv);
3961 } 4251 }
3962 4252
3963 { 4253 {
3964 struct coro *coro = SvSTATE_hv (hv); 4254 struct coro *coro = SvSTATE_hv (hv);
3965 4255
3972 api_ready (aTHX_ (SV *)hv); 4262 api_ready (aTHX_ (SV *)hv);
3973 4263
3974 if (GIMME_V != G_VOID) 4264 if (GIMME_V != G_VOID)
3975 XPUSHs (sv_2mortal (newRV_noinc ((SV *)hv))); 4265 XPUSHs (sv_2mortal (newRV_noinc ((SV *)hv)));
3976 else 4266 else
3977 SvREFCNT_dec (hv); 4267 SvREFCNT_dec_NN (hv);
3978} 4268}
3979 4269
3980SV * 4270SV *
3981rouse_cb () 4271rouse_cb ()
3982 PROTOTYPE: 4272 PROTOTYPE:
4061 RETVAL = newSVsv (AvARRAY ((AV *)SvRV (self))[0]); 4351 RETVAL = newSVsv (AvARRAY ((AV *)SvRV (self))[0]);
4062 OUTPUT: 4352 OUTPUT:
4063 RETVAL 4353 RETVAL
4064 4354
4065void 4355void
4066up (SV *self, int adjust = 1) 4356up (SV *self)
4067 ALIAS:
4068 adjust = 1
4069 CODE: 4357 CODE:
4358 coro_semaphore_adjust (aTHX_ (AV *)SvRV (self), 1);
4359
4360void
4361adjust (SV *self, int adjust)
4362 CODE:
4070 coro_semaphore_adjust (aTHX_ (AV *)SvRV (self), ix ? adjust : 1); 4363 coro_semaphore_adjust (aTHX_ (AV *)SvRV (self), adjust);
4071 4364
4072void 4365void
4073down (...) 4366down (...)
4074 CODE: 4367 CODE:
4075 CORO_EXECUTE_SLF_XS (slf_init_semaphore_down); 4368 CORO_EXECUTE_SLF_XS (slf_init_semaphore_down);
4237 { 4530 {
4238 PL_ppaddr [OP_SSELECT] = coro_old_pp_sselect; 4531 PL_ppaddr [OP_SSELECT] = coro_old_pp_sselect;
4239 coro_old_pp_sselect = 0; 4532 coro_old_pp_sselect = 0;
4240 } 4533 }
4241 4534
4535MODULE = Coro::State PACKAGE = Coro::Util
4536
4537void
4538_exit (int code)
4539 CODE:
4540 _exit (code);
4541
4542NV
4543time ()
4544 CODE:
4545 RETVAL = nvtime (aTHX);
4546 OUTPUT:
4547 RETVAL
4548
4549NV
4550gettimeofday ()
4551 PPCODE:
4552{
4553 UV tv [2];
4554 u2time (aTHX_ tv);
4555 EXTEND (SP, 2);
4556 PUSHs (sv_2mortal (newSVuv (tv [0])));
4557 PUSHs (sv_2mortal (newSVuv (tv [1])));
4558}
4559

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines