… | |
… | |
13 | #include "XSUB.h" |
13 | #include "XSUB.h" |
14 | #include "perliol.h" |
14 | #include "perliol.h" |
15 | |
15 | |
16 | #include "schmorp.h" |
16 | #include "schmorp.h" |
17 | |
17 | |
|
|
18 | #include <stddef.h> |
18 | #include <stdio.h> |
19 | #include <stdio.h> |
19 | #include <errno.h> |
20 | #include <errno.h> |
20 | #include <assert.h> |
21 | #include <assert.h> |
21 | |
22 | |
22 | #ifndef SVs_PADSTALE |
23 | #ifndef SVs_PADSTALE |
… | |
… | |
126 | static void (*u2time)(pTHX_ UV ret[2]); |
127 | static void (*u2time)(pTHX_ UV ret[2]); |
127 | |
128 | |
128 | /* we hijack an hopefully unused CV flag for our purposes */ |
129 | /* we hijack an hopefully unused CV flag for our purposes */ |
129 | #define CVf_SLF 0x4000 |
130 | #define CVf_SLF 0x4000 |
130 | static OP *pp_slf (pTHX); |
131 | static OP *pp_slf (pTHX); |
|
|
132 | static void slf_destroy (pTHX_ struct coro *coro); |
131 | |
133 | |
132 | static U32 cctx_gen; |
134 | static U32 cctx_gen; |
133 | static size_t cctx_stacksize = CORO_STACKSIZE; |
135 | static size_t cctx_stacksize = CORO_STACKSIZE; |
134 | static struct CoroAPI coroapi; |
136 | static struct CoroAPI coroapi; |
135 | static AV *main_mainstack; /* used to differentiate between $main and others */ |
137 | static AV *main_mainstack; /* used to differentiate between $main and others */ |
… | |
… | |
165 | static char times_valid; |
167 | static char times_valid; |
166 | |
168 | |
167 | static struct coro_cctx *cctx_first; |
169 | static struct coro_cctx *cctx_first; |
168 | static int cctx_count, cctx_idle; |
170 | static int cctx_count, cctx_idle; |
169 | |
171 | |
170 | enum { |
172 | enum |
|
|
173 | { |
171 | CC_MAPPED = 0x01, |
174 | CC_MAPPED = 0x01, |
172 | CC_NOREUSE = 0x02, /* throw this away after tracing */ |
175 | CC_NOREUSE = 0x02, /* throw this away after tracing */ |
173 | CC_TRACE = 0x04, |
176 | CC_TRACE = 0x04, |
174 | CC_TRACE_SUB = 0x08, /* trace sub calls */ |
177 | CC_TRACE_SUB = 0x08, /* trace sub calls */ |
175 | CC_TRACE_LINE = 0x10, /* trace each statement */ |
178 | CC_TRACE_LINE = 0x10, /* trace each statement */ |
… | |
… | |
200 | |
203 | |
201 | static coro_cctx *cctx_current; /* the currently running cctx */ |
204 | static coro_cctx *cctx_current; /* the currently running cctx */ |
202 | |
205 | |
203 | /*****************************************************************************/ |
206 | /*****************************************************************************/ |
204 | |
207 | |
|
|
208 | static MGVTBL coro_state_vtbl; |
|
|
209 | |
205 | enum { |
210 | enum |
|
|
211 | { |
206 | CF_RUNNING = 0x0001, /* coroutine is running */ |
212 | CF_RUNNING = 0x0001, /* coroutine is running */ |
207 | CF_READY = 0x0002, /* coroutine is ready */ |
213 | CF_READY = 0x0002, /* coroutine is ready */ |
208 | CF_NEW = 0x0004, /* has never been switched to */ |
214 | CF_NEW = 0x0004, /* has never been switched to */ |
209 | CF_DESTROYED = 0x0008, /* coroutine data has been freed */ |
215 | CF_ZOMBIE = 0x0008, /* coroutine data has been freed */ |
210 | CF_SUSPENDED = 0x0010, /* coroutine can't be scheduled */ |
216 | CF_SUSPENDED = 0x0010, /* coroutine can't be scheduled */ |
211 | CF_NOCANCEL = 0x0020, /* cannot cancel, set slf_frame.data to 1 (hackish) */ |
217 | CF_NOCANCEL = 0x0020, /* cannot cancel, set slf_frame.data to 1 (hackish) */ |
212 | }; |
218 | }; |
213 | |
219 | |
214 | /* the structure where most of the perl state is stored, overlaid on the cxstack */ |
220 | /* the structure where most of the perl state is stored, overlaid on the cxstack */ |
… | |
… | |
222 | #define VAR(name,type) type name; |
228 | #define VAR(name,type) type name; |
223 | # include "state.h" |
229 | # include "state.h" |
224 | #undef VAR |
230 | #undef VAR |
225 | } perl_slots; |
231 | } perl_slots; |
226 | |
232 | |
|
|
233 | // how many context stack entries do we need for perl_slots |
227 | #define SLOT_COUNT ((sizeof (perl_slots) + sizeof (PERL_CONTEXT) - 1) / sizeof (PERL_CONTEXT)) |
234 | #define SLOT_COUNT ((sizeof (perl_slots) + sizeof (PERL_CONTEXT) - 1) / sizeof (PERL_CONTEXT)) |
228 | |
235 | |
229 | /* this is a structure representing a perl-level coroutine */ |
236 | /* this is a structure representing a perl-level coroutine */ |
230 | struct coro { |
237 | struct coro |
|
|
238 | { |
231 | /* the C coroutine allocated to this perl coroutine, if any */ |
239 | /* the C coroutine allocated to this perl coroutine, if any */ |
232 | coro_cctx *cctx; |
240 | coro_cctx *cctx; |
233 | |
241 | |
234 | /* ready queue */ |
242 | /* ready queue */ |
235 | struct coro *next_ready; |
243 | struct coro *next_ready; |
… | |
… | |
237 | /* state data */ |
245 | /* state data */ |
238 | struct CoroSLF slf_frame; /* saved slf frame */ |
246 | struct CoroSLF slf_frame; /* saved slf frame */ |
239 | AV *mainstack; |
247 | AV *mainstack; |
240 | perl_slots *slot; /* basically the saved sp */ |
248 | perl_slots *slot; /* basically the saved sp */ |
241 | |
249 | |
242 | CV *startcv; /* the CV to execute */ |
250 | CV *startcv; /* the CV to execute */ |
243 | AV *args; /* data associated with this coroutine (initial args) */ |
251 | AV *args; /* data associated with this coroutine (initial args) */ |
244 | int refcnt; /* coroutines are refcounted, yes */ |
|
|
245 | int flags; /* CF_ flags */ |
252 | int flags; /* CF_ flags */ |
246 | HV *hv; /* the perl hash associated with this coro, if any */ |
253 | HV *hv; /* the perl hash associated with this coro, if any */ |
247 | void (*on_destroy)(pTHX_ struct coro *coro); /* for temporary use by xs in critical sections */ |
|
|
248 | |
254 | |
249 | /* statistics */ |
255 | /* statistics */ |
250 | int usecount; /* number of transfers to this coro */ |
256 | int usecount; /* number of transfers to this coro */ |
251 | |
257 | |
252 | /* coro process data */ |
258 | /* coro process data */ |
253 | int prio; |
259 | int prio; |
254 | SV *except; /* exception to be thrown */ |
260 | SV *except; /* exception to be thrown */ |
255 | SV *rouse_cb; |
261 | SV *rouse_cb; /* last rouse callback */ |
|
|
262 | AV *on_destroy; /* callbacks or coros to notify on destroy */ |
|
|
263 | AV *status; /* the exit status list */ |
256 | |
264 | |
257 | /* async_pool */ |
265 | /* async_pool */ |
258 | SV *saved_deffh; |
266 | SV *saved_deffh; |
259 | SV *invoke_cb; |
267 | SV *invoke_cb; |
260 | AV *invoke_av; |
268 | AV *invoke_av; |
… | |
… | |
292 | |
300 | |
293 | /* for Coro.pm */ |
301 | /* for Coro.pm */ |
294 | static SV *coro_current; |
302 | static SV *coro_current; |
295 | static SV *coro_readyhook; |
303 | static SV *coro_readyhook; |
296 | static struct coro *coro_ready [CORO_PRIO_MAX - CORO_PRIO_MIN + 1][2]; /* head|tail */ |
304 | static struct coro *coro_ready [CORO_PRIO_MAX - CORO_PRIO_MIN + 1][2]; /* head|tail */ |
297 | static CV *cv_coro_run, *cv_coro_terminate; |
305 | static CV *cv_coro_run; |
298 | static struct coro *coro_first; |
306 | static struct coro *coro_first; |
299 | #define coro_nready coroapi.nready |
307 | #define coro_nready coroapi.nready |
300 | |
308 | |
301 | /** Coro::Select ************************************************************/ |
309 | /** Coro::Select ************************************************************/ |
302 | |
310 | |
… | |
… | |
466 | : 0) |
474 | : 0) |
467 | |
475 | |
468 | #define CORO_MAGIC_cv(cv) CORO_MAGIC (((SV *)(cv)), CORO_MAGIC_type_cv) |
476 | #define CORO_MAGIC_cv(cv) CORO_MAGIC (((SV *)(cv)), CORO_MAGIC_type_cv) |
469 | #define CORO_MAGIC_state(sv) CORO_MAGIC_NN (((SV *)(sv)), CORO_MAGIC_type_state) |
477 | #define CORO_MAGIC_state(sv) CORO_MAGIC_NN (((SV *)(sv)), CORO_MAGIC_type_state) |
470 | |
478 | |
|
|
479 | INLINE MAGIC * |
|
|
480 | SvSTATEhv_p (pTHX_ SV *coro) |
|
|
481 | { |
|
|
482 | MAGIC *mg; |
|
|
483 | |
|
|
484 | if (expect_true ( |
|
|
485 | SvTYPE (coro) == SVt_PVHV |
|
|
486 | && (mg = CORO_MAGIC_state (coro)) |
|
|
487 | && mg->mg_virtual == &coro_state_vtbl |
|
|
488 | )) |
|
|
489 | return mg; |
|
|
490 | |
|
|
491 | return 0; |
|
|
492 | } |
|
|
493 | |
471 | INLINE struct coro * |
494 | INLINE struct coro * |
472 | SvSTATE_ (pTHX_ SV *coro) |
495 | SvSTATE_ (pTHX_ SV *coro) |
473 | { |
496 | { |
474 | HV *stash; |
|
|
475 | MAGIC *mg; |
497 | MAGIC *mg; |
476 | |
498 | |
477 | if (SvROK (coro)) |
499 | if (SvROK (coro)) |
478 | coro = SvRV (coro); |
500 | coro = SvRV (coro); |
479 | |
501 | |
480 | if (expect_false (SvTYPE (coro) != SVt_PVHV)) |
502 | mg = SvSTATEhv_p (aTHX_ coro); |
|
|
503 | if (!mg) |
481 | croak ("Coro::State object required"); |
504 | croak ("Coro::State object required"); |
482 | |
505 | |
483 | stash = SvSTASH (coro); |
|
|
484 | if (expect_false (stash != coro_stash && stash != coro_state_stash)) |
|
|
485 | { |
|
|
486 | /* very slow, but rare, check */ |
|
|
487 | if (!sv_derived_from (sv_2mortal (newRV_inc (coro)), "Coro::State")) |
|
|
488 | croak ("Coro::State object required"); |
|
|
489 | } |
|
|
490 | |
|
|
491 | mg = CORO_MAGIC_state (coro); |
|
|
492 | return (struct coro *)mg->mg_ptr; |
506 | return (struct coro *)mg->mg_ptr; |
493 | } |
507 | } |
494 | |
508 | |
495 | #define SvSTATE(sv) SvSTATE_ (aTHX_ (sv)) |
509 | #define SvSTATE(sv) SvSTATE_ (aTHX_ (sv)) |
496 | |
510 | |
… | |
… | |
798 | |
812 | |
799 | PUTBACK; |
813 | PUTBACK; |
800 | } |
814 | } |
801 | |
815 | |
802 | /* allocate some space on the context stack for our purposes */ |
816 | /* allocate some space on the context stack for our purposes */ |
803 | /* we manually unroll here, as usually 2 slots is enough */ |
817 | if (expect_false (cxstack_ix + SLOT_COUNT >= cxstack_max)) |
804 | if (SLOT_COUNT >= 1) CXINC; |
|
|
805 | if (SLOT_COUNT >= 2) CXINC; |
|
|
806 | if (SLOT_COUNT >= 3) CXINC; |
|
|
807 | { |
818 | { |
808 | unsigned int i; |
819 | unsigned int i; |
|
|
820 | |
809 | for (i = 3; i < SLOT_COUNT; ++i) |
821 | for (i = 0; i < SLOT_COUNT; ++i) |
810 | CXINC; |
822 | CXINC; |
811 | } |
823 | |
812 | cxstack_ix -= SLOT_COUNT; /* undo allocation */ |
824 | cxstack_ix -= SLOT_COUNT; /* undo allocation */ |
|
|
825 | } |
813 | |
826 | |
814 | c->mainstack = PL_mainstack; |
827 | c->mainstack = PL_mainstack; |
815 | |
828 | |
816 | { |
829 | { |
817 | perl_slots *slot = c->slot = (perl_slots *)(cxstack + cxstack_ix + 1); |
830 | perl_slots *slot = c->slot = (perl_slots *)(cxstack + cxstack_ix + 1); |
… | |
… | |
838 | # define coro_init_stacks(thx) init_stacks () |
851 | # define coro_init_stacks(thx) init_stacks () |
839 | #else |
852 | #else |
840 | static void |
853 | static void |
841 | coro_init_stacks (pTHX) |
854 | coro_init_stacks (pTHX) |
842 | { |
855 | { |
843 | PL_curstackinfo = new_stackinfo(32, 8); |
856 | PL_curstackinfo = new_stackinfo(32, 4 + SLOT_COUNT); /* 3 is minimum due to perl rounding down in scope.c:GROW() */ |
844 | PL_curstackinfo->si_type = PERLSI_MAIN; |
857 | PL_curstackinfo->si_type = PERLSI_MAIN; |
845 | PL_curstack = PL_curstackinfo->si_stack; |
858 | PL_curstack = PL_curstackinfo->si_stack; |
846 | PL_mainstack = PL_curstack; /* remember in case we switch stacks */ |
859 | PL_mainstack = PL_curstack; /* remember in case we switch stacks */ |
847 | |
860 | |
848 | PL_stack_base = AvARRAY(PL_curstack); |
861 | PL_stack_base = AvARRAY(PL_curstack); |
… | |
… | |
955 | #endif |
968 | #endif |
956 | |
969 | |
957 | /* |
970 | /* |
958 | * This overrides the default magic get method of %SIG elements. |
971 | * This overrides the default magic get method of %SIG elements. |
959 | * The original one doesn't provide for reading back of PL_diehook/PL_warnhook |
972 | * The original one doesn't provide for reading back of PL_diehook/PL_warnhook |
960 | * and instead of trying to save and restore the hash elements, we just provide |
973 | * and instead of trying to save and restore the hash elements (extremely slow), |
961 | * readback here. |
974 | * we just provide our own readback here. |
962 | */ |
975 | */ |
963 | static int |
976 | static int |
964 | coro_sigelem_get (pTHX_ SV *sv, MAGIC *mg) |
977 | coro_sigelem_get (pTHX_ SV *sv, MAGIC *mg) |
965 | { |
978 | { |
966 | const char *s = MgPV_nolen_const (mg); |
979 | const char *s = MgPV_nolen_const (mg); |
… | |
… | |
1074 | PL_parser = 0; |
1087 | PL_parser = 0; |
1075 | #endif |
1088 | #endif |
1076 | PL_hints = 0; |
1089 | PL_hints = 0; |
1077 | |
1090 | |
1078 | /* recreate the die/warn hooks */ |
1091 | /* recreate the die/warn hooks */ |
1079 | PL_diehook = 0; SvSetMagicSV (*hv_fetch (hv_sig, "__DIE__" , sizeof ("__DIE__" ) - 1, 1), rv_diehook ); |
1092 | PL_diehook = SvREFCNT_inc (rv_diehook); |
1080 | PL_warnhook = 0; SvSetMagicSV (*hv_fetch (hv_sig, "__WARN__", sizeof ("__WARN__") - 1, 1), rv_warnhook); |
1093 | PL_warnhook = SvREFCNT_inc (rv_warnhook); |
1081 | |
1094 | |
1082 | GvSV (PL_defgv) = newSV (0); |
1095 | GvSV (PL_defgv) = newSV (0); |
1083 | GvAV (PL_defgv) = coro->args; coro->args = 0; |
1096 | GvAV (PL_defgv) = coro->args; coro->args = 0; |
1084 | GvSV (PL_errgv) = newSV (0); |
1097 | GvSV (PL_errgv) = newSV (0); |
1085 | GvSV (irsgv) = newSVpvn ("\n", 1); sv_magic (GvSV (irsgv), (SV *)irsgv, PERL_MAGIC_sv, "/", 0); |
1098 | GvSV (irsgv) = newSVpvn ("\n", 1); sv_magic (GvSV (irsgv), (SV *)irsgv, PERL_MAGIC_sv, "/", 0); |
… | |
… | |
1106 | /* this newly created coroutine might be run on an existing cctx which most |
1119 | /* this newly created coroutine might be run on an existing cctx which most |
1107 | * likely was suspended in pp_slf, so we have to emulate entering pp_slf here. |
1120 | * likely was suspended in pp_slf, so we have to emulate entering pp_slf here. |
1108 | */ |
1121 | */ |
1109 | slf_frame.prepare = prepare_nop; /* provide a nop function for an eventual pp_slf */ |
1122 | slf_frame.prepare = prepare_nop; /* provide a nop function for an eventual pp_slf */ |
1110 | slf_frame.check = slf_check_nop; /* signal pp_slf to not repeat */ |
1123 | slf_frame.check = slf_check_nop; /* signal pp_slf to not repeat */ |
|
|
1124 | slf_frame.destroy = 0; |
1111 | |
1125 | |
1112 | /* and we have to provide the pp_slf op in any case, so pp_slf can skip it */ |
1126 | /* and we have to provide the pp_slf op in any case, so pp_slf can skip it */ |
1113 | init_perl_op.op_next = PL_op; |
1127 | init_perl_op.op_next = PL_op; |
1114 | init_perl_op.op_type = OP_ENTERSUB; |
1128 | init_perl_op.op_type = OP_ENTERSUB; |
1115 | init_perl_op.op_ppaddr = pp_slf; |
1129 | init_perl_op.op_ppaddr = pp_slf; |
… | |
… | |
1154 | destroy_perl (pTHX_ struct coro *coro) |
1168 | destroy_perl (pTHX_ struct coro *coro) |
1155 | { |
1169 | { |
1156 | SV *svf [9]; |
1170 | SV *svf [9]; |
1157 | |
1171 | |
1158 | { |
1172 | { |
|
|
1173 | SV *old_current = SvRV (coro_current); |
1159 | struct coro *current = SvSTATE_current; |
1174 | struct coro *current = SvSTATE (old_current); |
1160 | |
1175 | |
1161 | assert (("FATAL: tried to destroy currently running coroutine", coro->mainstack != PL_mainstack)); |
1176 | assert (("FATAL: tried to destroy currently running coroutine", coro->mainstack != PL_mainstack)); |
1162 | |
1177 | |
1163 | save_perl (aTHX_ current); |
1178 | save_perl (aTHX_ current); |
1164 | SvRV_set (coro_current, (SV *)coro->hv); /* this will cause acroak in transfer_check */ |
1179 | |
|
|
1180 | /* this will cause transfer_check to croak on block*/ |
|
|
1181 | SvRV_set (coro_current, (SV *)coro->hv); |
|
|
1182 | |
1165 | load_perl (aTHX_ coro); |
1183 | load_perl (aTHX_ coro); |
1166 | |
1184 | |
1167 | coro_unwind_stacks (aTHX); |
1185 | coro_unwind_stacks (aTHX); |
1168 | coro_destruct_stacks (aTHX); |
|
|
1169 | |
1186 | |
1170 | /* restore swapped sv's */ |
1187 | /* restore swapped sv's */ |
1171 | SWAP_SVS (coro); |
1188 | SWAP_SVS (coro); |
|
|
1189 | |
|
|
1190 | coro_destruct_stacks (aTHX); |
1172 | |
1191 | |
1173 | // now save some sv's to be free'd later |
1192 | // now save some sv's to be free'd later |
1174 | svf [0] = GvSV (PL_defgv); |
1193 | svf [0] = GvSV (PL_defgv); |
1175 | svf [1] = (SV *)GvAV (PL_defgv); |
1194 | svf [1] = (SV *)GvAV (PL_defgv); |
1176 | svf [2] = GvSV (PL_errgv); |
1195 | svf [2] = GvSV (PL_errgv); |
… | |
… | |
1180 | svf [6] = (SV *)GvHV (PL_hintgv); |
1199 | svf [6] = (SV *)GvHV (PL_hintgv); |
1181 | svf [7] = PL_diehook; |
1200 | svf [7] = PL_diehook; |
1182 | svf [8] = PL_warnhook; |
1201 | svf [8] = PL_warnhook; |
1183 | assert (9 == sizeof (svf) / sizeof (*svf)); |
1202 | assert (9 == sizeof (svf) / sizeof (*svf)); |
1184 | |
1203 | |
1185 | SvRV_set (coro_current, (SV *)current->hv); |
1204 | SvRV_set (coro_current, old_current); |
|
|
1205 | |
1186 | load_perl (aTHX_ current); |
1206 | load_perl (aTHX_ current); |
1187 | } |
1207 | } |
1188 | |
1208 | |
1189 | { |
1209 | { |
1190 | unsigned int i; |
1210 | unsigned int i; |
… | |
… | |
1202 | INLINE void |
1222 | INLINE void |
1203 | free_coro_mortal (pTHX) |
1223 | free_coro_mortal (pTHX) |
1204 | { |
1224 | { |
1205 | if (expect_true (coro_mortal)) |
1225 | if (expect_true (coro_mortal)) |
1206 | { |
1226 | { |
1207 | SvREFCNT_dec (coro_mortal); |
1227 | SvREFCNT_dec ((SV *)coro_mortal); |
1208 | coro_mortal = 0; |
1228 | coro_mortal = 0; |
1209 | } |
1229 | } |
1210 | } |
1230 | } |
1211 | |
1231 | |
1212 | static int |
1232 | static int |
… | |
… | |
1575 | if (expect_true (prev != next)) |
1595 | if (expect_true (prev != next)) |
1576 | { |
1596 | { |
1577 | if (expect_false (!(prev->flags & (CF_RUNNING | CF_NEW)))) |
1597 | if (expect_false (!(prev->flags & (CF_RUNNING | CF_NEW)))) |
1578 | croak ("Coro::State::transfer called with a blocked prev Coro::State, but can only transfer from running or new states,"); |
1598 | croak ("Coro::State::transfer called with a blocked prev Coro::State, but can only transfer from running or new states,"); |
1579 | |
1599 | |
1580 | if (expect_false (next->flags & (CF_RUNNING | CF_DESTROYED | CF_SUSPENDED))) |
1600 | if (expect_false (next->flags & (CF_RUNNING | CF_ZOMBIE | CF_SUSPENDED))) |
1581 | croak ("Coro::State::transfer called with running, destroyed or suspended next Coro::State, but can only transfer to inactive states,"); |
1601 | croak ("Coro::State::transfer called with running, destroyed or suspended next Coro::State, but can only transfer to inactive states,"); |
1582 | |
1602 | |
1583 | #if !PERL_VERSION_ATLEAST (5,10,0) |
1603 | #if !PERL_VERSION_ATLEAST (5,10,0) |
1584 | if (expect_false (PL_lex_state != LEX_NOTPARSING)) |
1604 | if (expect_false (PL_lex_state != LEX_NOTPARSING)) |
1585 | croak ("Coro::State::transfer called while parsing, but this is not supported in your perl version,"); |
1605 | croak ("Coro::State::transfer called while parsing, but this is not supported in your perl version,"); |
… | |
… | |
1668 | #define TRANSFER(ta, force_cctx) transfer (aTHX_ (ta).prev, (ta).next, (force_cctx)) |
1688 | #define TRANSFER(ta, force_cctx) transfer (aTHX_ (ta).prev, (ta).next, (force_cctx)) |
1669 | #define TRANSFER_CHECK(ta) transfer_check (aTHX_ (ta).prev, (ta).next) |
1689 | #define TRANSFER_CHECK(ta) transfer_check (aTHX_ (ta).prev, (ta).next) |
1670 | |
1690 | |
1671 | /** high level stuff ********************************************************/ |
1691 | /** high level stuff ********************************************************/ |
1672 | |
1692 | |
|
|
1693 | /* this function is actually Coro, not Coro::State, but we call it from here */ |
|
|
1694 | /* because it is convenient - but it hasn't been declared yet for that reason */ |
|
|
1695 | static void |
|
|
1696 | coro_call_on_destroy (pTHX_ struct coro *coro); |
|
|
1697 | |
1673 | static void |
1698 | static void |
1674 | coro_state_destroy (pTHX_ struct coro *coro) |
1699 | coro_state_destroy (pTHX_ struct coro *coro) |
1675 | { |
1700 | { |
1676 | if (coro->flags & CF_DESTROYED) |
1701 | if (coro->flags & CF_ZOMBIE) |
1677 | return; |
1702 | return; |
1678 | |
1703 | |
1679 | /* this callback is reserved for slf functions needing to do cleanup */ |
|
|
1680 | if (coro->on_destroy && !PL_dirty) |
|
|
1681 | coro->on_destroy (aTHX_ coro); |
1704 | slf_destroy (aTHX_ coro); |
1682 | |
1705 | |
1683 | /* |
|
|
1684 | * The on_destroy above most likely is from an SLF call. |
|
|
1685 | * Since by definition the SLF call will not finish when we destroy |
|
|
1686 | * the coro, we will have to force-finish it here, otherwise |
|
|
1687 | * cleanup functions cannot call SLF functions. |
|
|
1688 | */ |
|
|
1689 | coro->slf_frame.prepare = 0; |
|
|
1690 | |
|
|
1691 | coro->flags |= CF_DESTROYED; |
1706 | coro->flags |= CF_ZOMBIE; |
1692 | |
1707 | |
1693 | if (coro->flags & CF_READY) |
1708 | if (coro->flags & CF_READY) |
1694 | { |
1709 | { |
1695 | /* reduce nready, as destroying a ready coro effectively unreadies it */ |
1710 | /* reduce nready, as destroying a ready coro effectively unreadies it */ |
1696 | /* alternative: look through all ready queues and remove the coro */ |
1711 | /* alternative: look through all ready queues and remove the coro */ |
1697 | --coro_nready; |
1712 | --coro_nready; |
1698 | } |
1713 | } |
1699 | else |
1714 | else |
1700 | coro->flags |= CF_READY; /* make sure it is NOT put into the readyqueue */ |
1715 | coro->flags |= CF_READY; /* make sure it is NOT put into the readyqueue */ |
|
|
1716 | |
|
|
1717 | if (coro->next) coro->next->prev = coro->prev; |
|
|
1718 | if (coro->prev) coro->prev->next = coro->next; |
|
|
1719 | if (coro == coro_first) coro_first = coro->next; |
1701 | |
1720 | |
1702 | if (coro->mainstack |
1721 | if (coro->mainstack |
1703 | && coro->mainstack != main_mainstack |
1722 | && coro->mainstack != main_mainstack |
1704 | && coro->slot |
1723 | && coro->slot |
1705 | && !PL_dirty) |
1724 | && !PL_dirty) |
1706 | destroy_perl (aTHX_ coro); |
1725 | destroy_perl (aTHX_ coro); |
1707 | |
1726 | |
1708 | if (coro->next) coro->next->prev = coro->prev; |
|
|
1709 | if (coro->prev) coro->prev->next = coro->next; |
|
|
1710 | if (coro == coro_first) coro_first = coro->next; |
|
|
1711 | |
|
|
1712 | cctx_destroy (coro->cctx); |
1727 | cctx_destroy (coro->cctx); |
1713 | SvREFCNT_dec (coro->startcv); |
1728 | SvREFCNT_dec (coro->startcv); |
1714 | SvREFCNT_dec (coro->args); |
1729 | SvREFCNT_dec (coro->args); |
1715 | SvREFCNT_dec (coro->swap_sv); |
1730 | SvREFCNT_dec (coro->swap_sv); |
1716 | SvREFCNT_dec (CORO_THROW); |
1731 | SvREFCNT_dec (CORO_THROW); |
|
|
1732 | |
|
|
1733 | coro_call_on_destroy (aTHX_ coro); |
|
|
1734 | |
|
|
1735 | /* more destruction mayhem in coro_state_free */ |
1717 | } |
1736 | } |
1718 | |
1737 | |
1719 | static int |
1738 | static int |
1720 | coro_state_free (pTHX_ SV *sv, MAGIC *mg) |
1739 | coro_state_free (pTHX_ SV *sv, MAGIC *mg) |
1721 | { |
1740 | { |
1722 | struct coro *coro = (struct coro *)mg->mg_ptr; |
1741 | struct coro *coro = (struct coro *)mg->mg_ptr; |
1723 | mg->mg_ptr = 0; |
1742 | mg->mg_ptr = 0; |
1724 | |
1743 | |
1725 | coro->hv = 0; |
|
|
1726 | |
|
|
1727 | if (--coro->refcnt < 0) |
|
|
1728 | { |
|
|
1729 | coro_state_destroy (aTHX_ coro); |
1744 | coro_state_destroy (aTHX_ coro); |
|
|
1745 | SvREFCNT_dec (coro->on_destroy); |
|
|
1746 | SvREFCNT_dec (coro->status); |
|
|
1747 | |
1730 | Safefree (coro); |
1748 | Safefree (coro); |
1731 | } |
|
|
1732 | |
1749 | |
1733 | return 0; |
1750 | return 0; |
1734 | } |
1751 | } |
1735 | |
1752 | |
1736 | static int |
1753 | static int |
1737 | coro_state_dup (pTHX_ MAGIC *mg, CLONE_PARAMS *params) |
1754 | coro_state_dup (pTHX_ MAGIC *mg, CLONE_PARAMS *params) |
1738 | { |
1755 | { |
1739 | struct coro *coro = (struct coro *)mg->mg_ptr; |
1756 | /* called when perl clones the current process the slow way (windows process emulation) */ |
1740 | |
1757 | /* WE SIMply nuke the pointers in the copy, causing perl to croak */ |
1741 | ++coro->refcnt; |
1758 | mg->mg_ptr = 0; |
|
|
1759 | mg->mg_virtual = 0; |
1742 | |
1760 | |
1743 | return 0; |
1761 | return 0; |
1744 | } |
1762 | } |
1745 | |
1763 | |
1746 | static MGVTBL coro_state_vtbl = { |
1764 | static MGVTBL coro_state_vtbl = { |
… | |
… | |
1872 | struct coro *next = coro_deq (aTHX); |
1890 | struct coro *next = coro_deq (aTHX); |
1873 | |
1891 | |
1874 | if (expect_true (next)) |
1892 | if (expect_true (next)) |
1875 | { |
1893 | { |
1876 | /* cannot transfer to destroyed coros, skip and look for next */ |
1894 | /* cannot transfer to destroyed coros, skip and look for next */ |
1877 | if (expect_false (next->flags & (CF_DESTROYED | CF_SUSPENDED))) |
1895 | if (expect_false (next->flags & (CF_ZOMBIE | CF_SUSPENDED))) |
1878 | SvREFCNT_dec (next->hv); /* coro_nready has already been taken care of by destroy */ |
1896 | SvREFCNT_dec (next->hv); /* coro_nready has already been taken care of by destroy */ |
1879 | else |
1897 | else |
1880 | { |
1898 | { |
1881 | next->flags &= ~CF_READY; |
1899 | next->flags &= ~CF_READY; |
1882 | --coro_nready; |
1900 | --coro_nready; |
… | |
… | |
2015 | coro->slot->runops = RUNOPS_DEFAULT; |
2033 | coro->slot->runops = RUNOPS_DEFAULT; |
2016 | } |
2034 | } |
2017 | } |
2035 | } |
2018 | |
2036 | |
2019 | static void |
2037 | static void |
|
|
2038 | coro_push_av (pTHX_ AV *av, I32 gimme_v) |
|
|
2039 | { |
|
|
2040 | if (AvFILLp (av) >= 0 && gimme_v != G_VOID) |
|
|
2041 | { |
|
|
2042 | dSP; |
|
|
2043 | |
|
|
2044 | if (gimme_v == G_SCALAR) |
|
|
2045 | XPUSHs (AvARRAY (av)[AvFILLp (av)]); |
|
|
2046 | else |
|
|
2047 | { |
|
|
2048 | int i; |
|
|
2049 | EXTEND (SP, AvFILLp (av) + 1); |
|
|
2050 | |
|
|
2051 | for (i = 0; i <= AvFILLp (av); ++i) |
|
|
2052 | PUSHs (AvARRAY (av)[i]); |
|
|
2053 | } |
|
|
2054 | |
|
|
2055 | PUTBACK; |
|
|
2056 | } |
|
|
2057 | } |
|
|
2058 | |
|
|
2059 | static void |
|
|
2060 | coro_push_on_destroy (pTHX_ struct coro *coro, SV *cb) |
|
|
2061 | { |
|
|
2062 | if (!coro->on_destroy) |
|
|
2063 | coro->on_destroy = newAV (); |
|
|
2064 | |
|
|
2065 | av_push (coro->on_destroy, cb); |
|
|
2066 | } |
|
|
2067 | |
|
|
2068 | static void |
|
|
2069 | slf_destroy_join (pTHX_ struct CoroSLF *frame) |
|
|
2070 | { |
|
|
2071 | SvREFCNT_dec ((SV *)((struct coro *)frame->data)->hv); |
|
|
2072 | } |
|
|
2073 | |
|
|
2074 | static int |
|
|
2075 | slf_check_join (pTHX_ struct CoroSLF *frame) |
|
|
2076 | { |
|
|
2077 | struct coro *coro = (struct coro *)frame->data; |
|
|
2078 | |
|
|
2079 | if (!coro->status) |
|
|
2080 | return 1; |
|
|
2081 | |
|
|
2082 | frame->destroy = 0; |
|
|
2083 | |
|
|
2084 | coro_push_av (aTHX_ coro->status, GIMME_V); |
|
|
2085 | |
|
|
2086 | SvREFCNT_dec ((SV *)coro->hv); |
|
|
2087 | |
|
|
2088 | return 0; |
|
|
2089 | } |
|
|
2090 | |
|
|
2091 | static void |
|
|
2092 | slf_init_join (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items) |
|
|
2093 | { |
|
|
2094 | struct coro *coro = SvSTATE (items > 0 ? arg [0] : &PL_sv_undef); |
|
|
2095 | |
|
|
2096 | if (items > 1) |
|
|
2097 | croak ("join called with too many arguments"); |
|
|
2098 | |
|
|
2099 | if (coro->status) |
|
|
2100 | frame->prepare = prepare_nop; |
|
|
2101 | else |
|
|
2102 | { |
|
|
2103 | coro_push_on_destroy (aTHX_ coro, SvREFCNT_inc_NN (SvRV (coro_current))); |
|
|
2104 | frame->prepare = prepare_schedule; |
|
|
2105 | } |
|
|
2106 | |
|
|
2107 | frame->check = slf_check_join; |
|
|
2108 | frame->destroy = slf_destroy_join; |
|
|
2109 | frame->data = (void *)coro; |
|
|
2110 | SvREFCNT_inc (coro->hv); |
|
|
2111 | } |
|
|
2112 | |
|
|
2113 | static void |
2020 | coro_call_on_destroy (pTHX_ struct coro *coro) |
2114 | coro_call_on_destroy (pTHX_ struct coro *coro) |
2021 | { |
2115 | { |
2022 | SV **on_destroyp = hv_fetch (coro->hv, "_on_destroy", sizeof ("_on_destroy") - 1, 0); |
2116 | AV *od = coro->on_destroy; |
2023 | |
2117 | |
2024 | if (on_destroyp) |
2118 | if (!od) |
2025 | { |
2119 | return; |
2026 | SV **statusp = hv_fetch (coro->hv, "_status", sizeof ("_status") - 1, 0); |
|
|
2027 | AV *on_destroy = sv_2mortal (SvREFCNT_inc ((AV *)SvRV (*on_destroyp))); |
|
|
2028 | AV *status = statusp ? sv_2mortal (SvREFCNT_inc ((AV *)SvRV (*statusp))) : 0; |
|
|
2029 | |
2120 | |
2030 | while (AvFILLp (on_destroy) >= 0) |
2121 | while (AvFILLp (od) >= 0) |
|
|
2122 | { |
|
|
2123 | SV *cb = sv_2mortal (av_pop (od)); |
|
|
2124 | |
|
|
2125 | /* coro hv's (and only hv's at the moment) are supported as well */ |
|
|
2126 | if (SvSTATEhv_p (aTHX_ cb)) |
|
|
2127 | api_ready (aTHX_ cb); |
|
|
2128 | else |
2031 | { |
2129 | { |
2032 | dSP; /* don't disturb outer sp */ |
2130 | dSP; /* don't disturb outer sp */ |
2033 | SV *cb = av_pop (on_destroy); |
|
|
2034 | |
|
|
2035 | PUSHMARK (SP); |
2131 | PUSHMARK (SP); |
2036 | |
2132 | |
2037 | if (statusp) |
2133 | if (coro->status) |
2038 | { |
2134 | { |
2039 | int i; |
2135 | PUTBACK; |
2040 | EXTEND (SP, AvFILLp (status) + 1); |
2136 | coro_push_av (aTHX_ coro->status, G_ARRAY); |
2041 | |
2137 | SPAGAIN; |
2042 | for (i = 0; i <= AvFILLp (status); ++i) |
|
|
2043 | PUSHs (AvARRAY (status)[i]); |
|
|
2044 | } |
2138 | } |
2045 | |
2139 | |
2046 | PUTBACK; |
2140 | PUTBACK; |
2047 | call_sv (sv_2mortal (cb), G_VOID | G_DISCARD); |
2141 | call_sv (cb, G_VOID | G_DISCARD); |
2048 | } |
2142 | } |
2049 | } |
2143 | } |
2050 | } |
2144 | } |
2051 | |
2145 | |
2052 | static void |
2146 | static void |
2053 | coro_set_status (HV *coro_hv, SV **arg, int items) |
2147 | coro_set_status (pTHX_ struct coro *coro, SV **arg, int items) |
2054 | { |
2148 | { |
2055 | AV *av = newAV (); |
2149 | AV *av; |
|
|
2150 | |
|
|
2151 | if (coro->status) |
|
|
2152 | { |
|
|
2153 | av = coro->status; |
|
|
2154 | av_clear (av); |
|
|
2155 | } |
|
|
2156 | else |
|
|
2157 | av = coro->status = newAV (); |
2056 | |
2158 | |
2057 | /* items are actually not so common, so optimise for this case */ |
2159 | /* items are actually not so common, so optimise for this case */ |
2058 | if (items) |
2160 | if (items) |
2059 | { |
2161 | { |
2060 | int i; |
2162 | int i; |
… | |
… | |
2062 | av_extend (av, items - 1); |
2164 | av_extend (av, items - 1); |
2063 | |
2165 | |
2064 | for (i = 0; i < items; ++i) |
2166 | for (i = 0; i < items; ++i) |
2065 | av_push (av, SvREFCNT_inc_NN (arg [i])); |
2167 | av_push (av, SvREFCNT_inc_NN (arg [i])); |
2066 | } |
2168 | } |
2067 | |
|
|
2068 | hv_store (coro_hv, "_status", sizeof ("_status") - 1, newRV_noinc ((SV *)av), 0); |
|
|
2069 | } |
2169 | } |
2070 | |
2170 | |
2071 | static void |
2171 | static void |
2072 | slf_init_terminate_cancel_common (pTHX_ struct CoroSLF *frame, HV *coro_hv) |
2172 | slf_init_terminate_cancel_common (pTHX_ struct CoroSLF *frame, HV *coro_hv) |
2073 | { |
2173 | { |
… | |
… | |
2085 | static void |
2185 | static void |
2086 | slf_init_terminate (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items) |
2186 | slf_init_terminate (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items) |
2087 | { |
2187 | { |
2088 | HV *coro_hv = (HV *)SvRV (coro_current); |
2188 | HV *coro_hv = (HV *)SvRV (coro_current); |
2089 | |
2189 | |
2090 | coro_set_status (coro_hv, arg, items); |
2190 | coro_set_status (aTHX_ SvSTATE ((SV *)coro_hv), arg, items); |
2091 | slf_init_terminate_cancel_common (frame, coro_hv); |
2191 | slf_init_terminate_cancel_common (aTHX_ frame, coro_hv); |
2092 | } |
2192 | } |
2093 | |
2193 | |
2094 | static void |
2194 | static void |
2095 | slf_init_cancel (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items) |
2195 | slf_init_cancel (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items) |
2096 | { |
2196 | { |
… | |
… | |
2101 | croak ("Coro::cancel called without coro object,"); |
2201 | croak ("Coro::cancel called without coro object,"); |
2102 | |
2202 | |
2103 | coro = SvSTATE (arg [0]); |
2203 | coro = SvSTATE (arg [0]); |
2104 | coro_hv = coro->hv; |
2204 | coro_hv = coro->hv; |
2105 | |
2205 | |
2106 | coro_set_status (coro_hv, arg + 1, items - 1); |
2206 | coro_set_status (aTHX_ coro, arg + 1, items - 1); |
2107 | |
2207 | |
2108 | if (expect_false (coro->flags & CF_NOCANCEL)) |
2208 | if (expect_false (coro->flags & CF_NOCANCEL)) |
2109 | { |
2209 | { |
2110 | /* coro currently busy cancelling something, so just notify it */ |
2210 | /* coro currently busy cancelling something, so just notify it */ |
2111 | coro->slf_frame.data = (void *)coro; |
2211 | coro->slf_frame.data = (void *)coro; |
… | |
… | |
2114 | frame->check = slf_check_nop; |
2214 | frame->check = slf_check_nop; |
2115 | } |
2215 | } |
2116 | else if (coro_hv == (HV *)SvRV (coro_current)) |
2216 | else if (coro_hv == (HV *)SvRV (coro_current)) |
2117 | { |
2217 | { |
2118 | /* cancelling the current coro is allowed, and equals terminate */ |
2218 | /* cancelling the current coro is allowed, and equals terminate */ |
2119 | slf_init_terminate_cancel_common (frame, coro_hv); |
2219 | slf_init_terminate_cancel_common (aTHX_ frame, coro_hv); |
2120 | } |
2220 | } |
2121 | else |
2221 | else |
2122 | { |
2222 | { |
2123 | struct coro *self = SvSTATE_current; |
2223 | struct coro *self = SvSTATE_current; |
2124 | |
2224 | |
… | |
… | |
2128 | * this is ugly, and hopefully fully worth the extra speed. |
2228 | * this is ugly, and hopefully fully worth the extra speed. |
2129 | * besides, I can't get the slow-but-safe version working... |
2229 | * besides, I can't get the slow-but-safe version working... |
2130 | */ |
2230 | */ |
2131 | slf_frame.data = 0; |
2231 | slf_frame.data = 0; |
2132 | self->flags |= CF_NOCANCEL; |
2232 | self->flags |= CF_NOCANCEL; |
2133 | |
|
|
2134 | coro_state_destroy (aTHX_ coro); |
2233 | coro_state_destroy (aTHX_ coro); |
2135 | coro_call_on_destroy (aTHX_ coro); |
|
|
2136 | |
|
|
2137 | self->flags &= ~CF_NOCANCEL; |
2234 | self->flags &= ~CF_NOCANCEL; |
2138 | |
2235 | |
2139 | if (slf_frame.data) |
2236 | if (slf_frame.data) |
2140 | { |
2237 | { |
2141 | /* while we were busy we have been cancelled, so terminate */ |
2238 | /* while we were busy we have been cancelled, so terminate */ |
2142 | slf_init_terminate_cancel_common (frame, self->hv); |
2239 | slf_init_terminate_cancel_common (aTHX_ frame, self->hv); |
2143 | } |
2240 | } |
2144 | else |
2241 | else |
2145 | { |
2242 | { |
2146 | frame->prepare = prepare_nop; |
2243 | frame->prepare = prepare_nop; |
2147 | frame->check = slf_check_nop; |
2244 | frame->check = slf_check_nop; |
2148 | } |
2245 | } |
2149 | } |
2246 | } |
2150 | } |
2247 | } |
2151 | |
2248 | |
|
|
2249 | static int |
|
|
2250 | slf_check_safe_cancel (pTHX_ struct CoroSLF *frame) |
|
|
2251 | { |
|
|
2252 | frame->prepare = 0; |
|
|
2253 | coro_unwind_stacks (aTHX); |
|
|
2254 | |
|
|
2255 | slf_init_terminate_cancel_common (aTHX_ frame, (HV *)SvRV (coro_current)); |
|
|
2256 | |
|
|
2257 | return 1; |
|
|
2258 | } |
|
|
2259 | |
|
|
2260 | static int |
|
|
2261 | safe_cancel (pTHX_ struct coro *coro, SV **arg, int items) |
|
|
2262 | { |
|
|
2263 | if (coro->cctx) |
|
|
2264 | croak ("coro inside C callback, unable to cancel at this time, caught"); |
|
|
2265 | |
|
|
2266 | if (coro->flags & CF_NEW) |
|
|
2267 | { |
|
|
2268 | coro_set_status (aTHX_ coro, arg, items); |
|
|
2269 | coro_state_destroy (aTHX_ coro); |
|
|
2270 | } |
|
|
2271 | else |
|
|
2272 | { |
|
|
2273 | if (!coro->slf_frame.prepare) |
|
|
2274 | croak ("coro outside an SLF function, unable to cancel at this time, caught"); |
|
|
2275 | |
|
|
2276 | slf_destroy (aTHX_ coro); |
|
|
2277 | |
|
|
2278 | coro_set_status (aTHX_ coro, arg, items); |
|
|
2279 | coro->slf_frame.prepare = prepare_nop; |
|
|
2280 | coro->slf_frame.check = slf_check_safe_cancel; |
|
|
2281 | |
|
|
2282 | api_ready (aTHX_ (SV *)coro->hv); |
|
|
2283 | } |
|
|
2284 | |
|
|
2285 | return 1; |
|
|
2286 | } |
|
|
2287 | |
2152 | /*****************************************************************************/ |
2288 | /*****************************************************************************/ |
2153 | /* async pool handler */ |
2289 | /* async pool handler */ |
2154 | |
2290 | |
2155 | static int |
2291 | static int |
2156 | slf_check_pool_handler (pTHX_ struct CoroSLF *frame) |
2292 | slf_check_pool_handler (pTHX_ struct CoroSLF *frame) |
… | |
… | |
2194 | coro->saved_deffh = 0; |
2330 | coro->saved_deffh = 0; |
2195 | |
2331 | |
2196 | if (coro_rss (aTHX_ coro) > SvUV (sv_pool_rss) |
2332 | if (coro_rss (aTHX_ coro) > SvUV (sv_pool_rss) |
2197 | || av_len (av_async_pool) + 1 >= SvIV (sv_pool_size)) |
2333 | || av_len (av_async_pool) + 1 >= SvIV (sv_pool_size)) |
2198 | { |
2334 | { |
2199 | coro->invoke_cb = SvREFCNT_inc_NN ((SV *)cv_coro_terminate); |
2335 | slf_init_terminate_cancel_common (aTHX_ frame, hv); |
2200 | coro->invoke_av = newAV (); |
2336 | return; |
2201 | |
|
|
2202 | frame->prepare = prepare_nop; |
|
|
2203 | } |
2337 | } |
2204 | else |
2338 | else |
2205 | { |
2339 | { |
2206 | av_clear (GvAV (PL_defgv)); |
2340 | av_clear (GvAV (PL_defgv)); |
2207 | hv_store (hv, "desc", sizeof ("desc") - 1, SvREFCNT_inc_NN (sv_async_pool_idle), 0); |
2341 | hv_store (hv, "desc", sizeof ("desc") - 1, SvREFCNT_inc_NN (sv_async_pool_idle), 0); |
… | |
… | |
2430 | static void |
2564 | static void |
2431 | slf_init_cede_notself (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items) |
2565 | slf_init_cede_notself (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items) |
2432 | { |
2566 | { |
2433 | frame->prepare = prepare_cede_notself; |
2567 | frame->prepare = prepare_cede_notself; |
2434 | frame->check = slf_check_nop; |
2568 | frame->check = slf_check_nop; |
|
|
2569 | } |
|
|
2570 | |
|
|
2571 | /* "undo"/cancel a running slf call - used when cancelling a coro, mainly */ |
|
|
2572 | static void |
|
|
2573 | slf_destroy (pTHX_ struct coro *coro) |
|
|
2574 | { |
|
|
2575 | /* this callback is reserved for slf functions needing to do cleanup */ |
|
|
2576 | if (coro->slf_frame.destroy && coro->slf_frame.prepare && !PL_dirty) |
|
|
2577 | coro->slf_frame.destroy (aTHX_ &coro->slf_frame); |
|
|
2578 | |
|
|
2579 | /* |
|
|
2580 | * The on_destroy above most likely is from an SLF call. |
|
|
2581 | * Since by definition the SLF call will not finish when we destroy |
|
|
2582 | * the coro, we will have to force-finish it here, otherwise |
|
|
2583 | * cleanup functions cannot call SLF functions. |
|
|
2584 | */ |
|
|
2585 | coro->slf_frame.prepare = 0; |
2435 | } |
2586 | } |
2436 | |
2587 | |
2437 | /* |
2588 | /* |
2438 | * these not obviously related functions are all rolled into one |
2589 | * these not obviously related functions are all rolled into one |
2439 | * function to increase chances that they all will call transfer with the same |
2590 | * function to increase chances that they all will call transfer with the same |
… | |
… | |
2508 | croak (0); |
2659 | croak (0); |
2509 | } |
2660 | } |
2510 | |
2661 | |
2511 | /* return value handling - mostly like entersub */ |
2662 | /* return value handling - mostly like entersub */ |
2512 | /* make sure we put something on the stack in scalar context */ |
2663 | /* make sure we put something on the stack in scalar context */ |
2513 | if (GIMME_V == G_SCALAR) |
2664 | if (GIMME_V == G_SCALAR |
|
|
2665 | && expect_false (PL_stack_sp != PL_stack_base + checkmark + 1)) |
2514 | { |
2666 | { |
2515 | dSP; |
2667 | dSP; |
2516 | SV **bot = PL_stack_base + checkmark; |
2668 | SV **bot = PL_stack_base + checkmark; |
2517 | |
2669 | |
2518 | if (sp == bot) /* too few, push undef */ |
2670 | if (sp == bot) /* too few, push undef */ |
2519 | bot [1] = &PL_sv_undef; |
2671 | bot [1] = &PL_sv_undef; |
2520 | else if (sp != bot + 1) /* too many, take last one */ |
2672 | else /* too many, take last one */ |
2521 | bot [1] = *sp; |
2673 | bot [1] = *sp; |
2522 | |
2674 | |
2523 | SP = bot + 1; |
2675 | SP = bot + 1; |
2524 | |
2676 | |
2525 | PUTBACK; |
2677 | PUTBACK; |
… | |
… | |
2765 | SvREFCNT_dec (cb); |
2917 | SvREFCNT_dec (cb); |
2766 | } |
2918 | } |
2767 | } |
2919 | } |
2768 | |
2920 | |
2769 | static void |
2921 | static void |
2770 | coro_semaphore_on_destroy (pTHX_ struct coro *coro) |
2922 | coro_semaphore_destroy (pTHX_ struct CoroSLF *frame) |
2771 | { |
2923 | { |
2772 | /* call $sem->adjust (0) to possibly wake up some other waiters */ |
2924 | /* call $sem->adjust (0) to possibly wake up some other waiters */ |
2773 | coro_semaphore_adjust (aTHX_ (AV *)coro->slf_frame.data, 0); |
2925 | coro_semaphore_adjust (aTHX_ (AV *)frame->data, 0); |
2774 | } |
2926 | } |
2775 | |
2927 | |
2776 | static int |
2928 | static int |
2777 | slf_check_semaphore_down_or_wait (pTHX_ struct CoroSLF *frame, int acquire) |
2929 | slf_check_semaphore_down_or_wait (pTHX_ struct CoroSLF *frame, int acquire) |
2778 | { |
2930 | { |
2779 | AV *av = (AV *)frame->data; |
2931 | AV *av = (AV *)frame->data; |
2780 | SV *count_sv = AvARRAY (av)[0]; |
2932 | SV *count_sv = AvARRAY (av)[0]; |
|
|
2933 | SV *coro_hv = SvRV (coro_current); |
2781 | |
2934 | |
2782 | /* if we are about to throw, don't actually acquire the lock, just throw */ |
2935 | /* if we are about to throw, don't actually acquire the lock, just throw */ |
2783 | if (CORO_THROW) |
2936 | if (CORO_THROW) |
2784 | return 0; |
2937 | return 0; |
2785 | else if (SvIVX (count_sv) > 0) |
2938 | else if (SvIVX (count_sv) > 0) |
2786 | { |
2939 | { |
2787 | SvSTATE_current->on_destroy = 0; |
2940 | frame->destroy = 0; |
2788 | |
2941 | |
2789 | if (acquire) |
2942 | if (acquire) |
2790 | SvIVX (count_sv) = SvIVX (count_sv) - 1; |
2943 | SvIVX (count_sv) = SvIVX (count_sv) - 1; |
2791 | else |
2944 | else |
2792 | coro_semaphore_adjust (aTHX_ av, 0); |
2945 | coro_semaphore_adjust (aTHX_ av, 0); |
… | |
… | |
2797 | { |
2950 | { |
2798 | int i; |
2951 | int i; |
2799 | /* if we were woken up but can't down, we look through the whole */ |
2952 | /* if we were woken up but can't down, we look through the whole */ |
2800 | /* waiters list and only add us if we aren't in there already */ |
2953 | /* waiters list and only add us if we aren't in there already */ |
2801 | /* this avoids some degenerate memory usage cases */ |
2954 | /* this avoids some degenerate memory usage cases */ |
2802 | |
2955 | for (i = AvFILLp (av); i > 0; --i) // i > 0 is not an off-by-one bug |
2803 | for (i = 1; i <= AvFILLp (av); ++i) |
|
|
2804 | if (AvARRAY (av)[i] == SvRV (coro_current)) |
2956 | if (AvARRAY (av)[i] == coro_hv) |
2805 | return 1; |
2957 | return 1; |
2806 | |
2958 | |
2807 | av_push (av, SvREFCNT_inc (SvRV (coro_current))); |
2959 | av_push (av, SvREFCNT_inc (coro_hv)); |
2808 | return 1; |
2960 | return 1; |
2809 | } |
2961 | } |
2810 | } |
2962 | } |
2811 | |
2963 | |
2812 | static int |
2964 | static int |
… | |
… | |
2835 | { |
2987 | { |
2836 | av_push (av, SvREFCNT_inc (SvRV (coro_current))); |
2988 | av_push (av, SvREFCNT_inc (SvRV (coro_current))); |
2837 | |
2989 | |
2838 | frame->data = (void *)sv_2mortal (SvREFCNT_inc ((SV *)av)); |
2990 | frame->data = (void *)sv_2mortal (SvREFCNT_inc ((SV *)av)); |
2839 | frame->prepare = prepare_schedule; |
2991 | frame->prepare = prepare_schedule; |
2840 | |
|
|
2841 | /* to avoid race conditions when a woken-up coro gets terminated */ |
2992 | /* to avoid race conditions when a woken-up coro gets terminated */ |
2842 | /* we arrange for a temporary on_destroy that calls adjust (0) */ |
2993 | /* we arrange for a temporary on_destroy that calls adjust (0) */ |
2843 | SvSTATE_current->on_destroy = coro_semaphore_on_destroy; |
2994 | frame->destroy = coro_semaphore_destroy; |
2844 | } |
2995 | } |
2845 | } |
2996 | } |
2846 | |
2997 | |
2847 | static void |
2998 | static void |
2848 | slf_init_semaphore_down (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items) |
2999 | slf_init_semaphore_down (pTHX_ struct CoroSLF *frame, CV *cv, SV **arg, int items) |
… | |
… | |
3397 | PROTOTYPE: $ |
3548 | PROTOTYPE: $ |
3398 | ALIAS: |
3549 | ALIAS: |
3399 | is_ready = CF_READY |
3550 | is_ready = CF_READY |
3400 | is_running = CF_RUNNING |
3551 | is_running = CF_RUNNING |
3401 | is_new = CF_NEW |
3552 | is_new = CF_NEW |
3402 | is_destroyed = CF_DESTROYED |
3553 | is_destroyed = CF_ZOMBIE |
|
|
3554 | is_zombie = CF_ZOMBIE |
3403 | is_suspended = CF_SUSPENDED |
3555 | is_suspended = CF_SUSPENDED |
3404 | CODE: |
3556 | CODE: |
3405 | RETVAL = boolSV (coro->flags & ix); |
3557 | RETVAL = boolSV (coro->flags & ix); |
3406 | OUTPUT: |
3558 | OUTPUT: |
3407 | RETVAL |
3559 | RETVAL |
… | |
… | |
3543 | BOOT: |
3695 | BOOT: |
3544 | { |
3696 | { |
3545 | sv_pool_rss = coro_get_sv (aTHX_ "Coro::POOL_RSS" , TRUE); |
3697 | sv_pool_rss = coro_get_sv (aTHX_ "Coro::POOL_RSS" , TRUE); |
3546 | sv_pool_size = coro_get_sv (aTHX_ "Coro::POOL_SIZE" , TRUE); |
3698 | sv_pool_size = coro_get_sv (aTHX_ "Coro::POOL_SIZE" , TRUE); |
3547 | cv_coro_run = get_cv ( "Coro::_coro_run" , GV_ADD); |
3699 | cv_coro_run = get_cv ( "Coro::_coro_run" , GV_ADD); |
3548 | cv_coro_terminate = get_cv ( "Coro::terminate" , GV_ADD); |
|
|
3549 | coro_current = coro_get_sv (aTHX_ "Coro::current" , FALSE); SvREADONLY_on (coro_current); |
3700 | coro_current = coro_get_sv (aTHX_ "Coro::current" , FALSE); SvREADONLY_on (coro_current); |
3550 | av_async_pool = coro_get_av (aTHX_ "Coro::async_pool", TRUE); |
3701 | av_async_pool = coro_get_av (aTHX_ "Coro::async_pool", TRUE); |
3551 | av_destroy = coro_get_av (aTHX_ "Coro::destroy" , TRUE); |
3702 | av_destroy = coro_get_av (aTHX_ "Coro::destroy" , TRUE); |
3552 | sv_manager = coro_get_sv (aTHX_ "Coro::manager" , TRUE); |
3703 | sv_manager = coro_get_sv (aTHX_ "Coro::manager" , TRUE); |
3553 | sv_idle = coro_get_sv (aTHX_ "Coro::idle" , TRUE); |
3704 | sv_idle = coro_get_sv (aTHX_ "Coro::idle" , TRUE); |
… | |
… | |
3596 | void |
3747 | void |
3597 | _destroy (Coro::State coro) |
3748 | _destroy (Coro::State coro) |
3598 | CODE: |
3749 | CODE: |
3599 | /* used by the manager thread */ |
3750 | /* used by the manager thread */ |
3600 | coro_state_destroy (aTHX_ coro); |
3751 | coro_state_destroy (aTHX_ coro); |
|
|
3752 | |
|
|
3753 | void |
|
|
3754 | on_destroy (Coro::State coro, SV *cb) |
|
|
3755 | CODE: |
3601 | coro_call_on_destroy (aTHX_ coro); |
3756 | coro_push_on_destroy (aTHX_ coro, newSVsv (cb)); |
|
|
3757 | |
|
|
3758 | void |
|
|
3759 | join (...) |
|
|
3760 | CODE: |
|
|
3761 | CORO_EXECUTE_SLF_XS (slf_init_join); |
3602 | |
3762 | |
3603 | void |
3763 | void |
3604 | terminate (...) |
3764 | terminate (...) |
3605 | CODE: |
3765 | CODE: |
3606 | CORO_EXECUTE_SLF_XS (slf_init_terminate); |
3766 | CORO_EXECUTE_SLF_XS (slf_init_terminate); |
3607 | |
3767 | |
3608 | void |
3768 | void |
3609 | cancel (...) |
3769 | cancel (...) |
3610 | CODE: |
3770 | CODE: |
3611 | CORO_EXECUTE_SLF_XS (slf_init_cancel); |
3771 | CORO_EXECUTE_SLF_XS (slf_init_cancel); |
|
|
3772 | |
|
|
3773 | int |
|
|
3774 | safe_cancel (Coro::State self, ...) |
|
|
3775 | C_ARGS: aTHX_ self, &ST (1), items - 1 |
3612 | |
3776 | |
3613 | void |
3777 | void |
3614 | schedule (...) |
3778 | schedule (...) |
3615 | CODE: |
3779 | CODE: |
3616 | CORO_EXECUTE_SLF_XS (slf_init_schedule); |
3780 | CORO_EXECUTE_SLF_XS (slf_init_schedule); |