1 | #include "EXTERN.h" |
1 | #include "EXTERN.h" |
2 | #include "perl.h" |
2 | #include "perl.h" |
3 | #include "XSUB.h" |
3 | #include "XSUB.h" |
4 | |
4 | |
|
|
5 | #include <assert.h> |
5 | #include <string.h> |
6 | #include <string.h> |
6 | |
|
|
7 | /* this useful idiom is unfortunately missing... */ |
|
|
8 | static void |
|
|
9 | confess (const char *msg) |
|
|
10 | { |
|
|
11 | dSP; |
|
|
12 | |
|
|
13 | PUSHMARK(SP); |
|
|
14 | XPUSHs (sv_2mortal(newSVpv("only one coroutine can wait for an event at any given time",0))); |
|
|
15 | PUTBACK; |
|
|
16 | call_pv ("Carp::confess", G_VOID); |
|
|
17 | } |
|
|
18 | |
7 | |
19 | #include "EventAPI.h" |
8 | #include "EventAPI.h" |
20 | #include "../Coro/CoroAPI.h" |
9 | #include "../Coro/CoroAPI.h" |
21 | |
10 | |
22 | #ifndef PE_PERLCB |
11 | #define CD_WAIT 0 /* wait queue */ |
23 | # define PE_PERLCB 0x020 /* not public, but we need it :( */ |
|
|
24 | #endif |
|
|
25 | |
|
|
26 | #define CD_CORO 0 |
|
|
27 | #define CD_TYPE 1 |
12 | #define CD_TYPE 1 |
28 | #define CD_OK 2 |
13 | #define CD_OK 2 |
29 | #define CD_PRIO 3 /* hardcoded in Coro::Event */ |
|
|
30 | #define CD_HITS 4 /* hardcoded in Coro::Event */ |
14 | #define CD_HITS 3 /* hardcoded in Coro::Event */ |
31 | #define CD_GOT 5 /* hardcoded in Coro::Event, Coro::Handle */ |
15 | #define CD_GOT 4 /* hardcoded in Coro::Event, Coro::Handle */ |
32 | #define CD_MAX 5 |
16 | #define CD_MAX 4 |
33 | |
17 | |
34 | #define EV_CLASS "Coro::Event" |
18 | static HV *coro_event_event_stash; |
35 | |
19 | |
36 | static pe_idle *scheduler; |
20 | #define PERL_MAGIC_coro_event 0x18 /* to avoid clashes with e.g. event */ |
37 | static int do_schedule; |
|
|
38 | |
|
|
39 | #define NEED_SCHEDULE if (!do_schedule) \ |
|
|
40 | { \ |
|
|
41 | do_schedule = 1; \ |
|
|
42 | GEventAPI->now ((pe_watcher *)scheduler); \ |
|
|
43 | } |
|
|
44 | |
21 | |
45 | static void |
22 | static void |
46 | coro_std_cb(pe_event *pe) |
23 | coro_std_cb (pe_event *pe) |
47 | { |
24 | { |
48 | AV *priv = (AV *)pe->ext_data; |
25 | AV *priv = (AV *)pe->ext_data; |
49 | IV type = SvIV (*av_fetch (priv, CD_TYPE, 1)); |
26 | IV type = SvIV (AvARRAY (priv)[CD_TYPE]); |
50 | SV **cd_coro = &AvARRAY(priv)[CD_CORO]; |
27 | AV *cd_wait; |
|
|
28 | SV *coro; |
51 | |
29 | |
52 | sv_setiv (AvARRAY(priv)[CD_PRIO], pe->prio); |
|
|
53 | sv_setiv (AvARRAY(priv)[CD_HITS], pe->hits); |
30 | SvIV_set (AvARRAY (priv)[CD_HITS], pe->hits); |
|
|
31 | SvIV_set (AvARRAY (priv)[CD_GOT], type ? ((pe_ioevent *)pe)->got : 0); |
54 | |
32 | |
55 | if (type == 1) |
33 | AvARRAY (priv)[CD_OK] = &PL_sv_yes; |
56 | sv_setiv (AvARRAY(priv)[CD_GOT], ((pe_ioevent *)pe)->got); |
|
|
57 | |
34 | |
|
|
35 | cd_wait = (AV *)AvARRAY(priv)[CD_WAIT]; |
|
|
36 | |
|
|
37 | coro = av_shift (cd_wait); |
58 | if (*cd_coro != &PL_sv_undef) |
38 | if (coro != &PL_sv_undef) |
59 | { |
39 | { |
60 | CORO_READY (*cd_coro); |
40 | CORO_READY (coro); |
61 | SvREFCNT_dec (*cd_coro); |
41 | SvREFCNT_dec (coro); |
62 | *cd_coro = &PL_sv_undef; |
|
|
63 | NEED_SCHEDULE; |
|
|
64 | } |
42 | } |
65 | else |
43 | |
66 | { |
44 | if (av_len (cd_wait) < 0) |
67 | AvARRAY(priv)[CD_OK] = &PL_sv_yes; |
|
|
68 | GEventAPI->stop (pe->up, 0); |
45 | GEventAPI->stop (pe->up, 0); |
69 | } |
|
|
70 | } |
46 | } |
71 | |
47 | |
72 | static void |
48 | static void |
73 | scheduler_cb(pe_event *pe) |
49 | asynccheck_hook (void *data) |
74 | { |
50 | { |
|
|
51 | /* this loops as long as we have _other_ coros with the same or higher priority */ |
75 | while (CORO_NREADY) |
52 | while (CORO_NREADY && CORO_CEDE) |
76 | CORO_CEDE; |
53 | ; |
|
|
54 | } |
77 | |
55 | |
78 | do_schedule = 0; |
56 | static double |
|
|
57 | prepare_hook (void *data) |
|
|
58 | { |
|
|
59 | /* this yields once to another coro with any priority */ |
|
|
60 | if (CORO_NREADY) |
|
|
61 | { |
|
|
62 | CORO_CEDE_NOTSELF; |
|
|
63 | /* |
|
|
64 | * timers might have changed, and Event fails to notice this |
|
|
65 | * so we have to assume the worst. If Event didn't have that bug, |
|
|
66 | * we would only need to do this if CORO_NREADY is != 0 now. |
|
|
67 | */ |
|
|
68 | return 0.; |
|
|
69 | } |
|
|
70 | else |
|
|
71 | return 85197.73; /* this is as good as any value, but it factors badly with common values */ |
79 | } |
72 | } |
80 | |
73 | |
81 | MODULE = Coro::Event PACKAGE = Coro::Event |
74 | MODULE = Coro::Event PACKAGE = Coro::Event |
82 | |
75 | |
83 | PROTOTYPES: ENABLE |
76 | PROTOTYPES: ENABLE |
84 | |
77 | |
85 | BOOT: |
78 | BOOT: |
86 | { |
79 | { |
|
|
80 | coro_event_event_stash = gv_stashpv ("Coro::Event::Event", TRUE); |
|
|
81 | |
87 | I_EVENT_API ("Coro::Event"); |
82 | I_EVENT_API ("Coro::Event"); |
88 | I_CORO_API ("Coro::Event"); |
83 | I_CORO_API ("Coro::Event"); |
89 | |
84 | |
90 | /* create a fake idle handler (we only ever call now) */ |
85 | GEventAPI->add_hook ("asynccheck", (void *)asynccheck_hook, 0); |
91 | scheduler = GEventAPI->new_idle (0, 0); |
86 | GEventAPI->add_hook ("prepare", (void *)prepare_hook, 0); |
92 | scheduler->base.callback = scheduler_cb; |
|
|
93 | scheduler->base.prio = PE_PRIO_NORMAL; /* StarvePrio */ |
|
|
94 | scheduler->min_interval = newSVnv (0); |
|
|
95 | scheduler->max_interval = newSVnv (0); |
|
|
96 | GEventAPI->stop ((pe_watcher *)scheduler, 0); |
|
|
97 | } |
87 | } |
98 | |
88 | |
99 | void |
89 | void |
100 | _install_std_cb(self,type) |
90 | _install_std_cb (SV *self, int type) |
101 | SV * self |
|
|
102 | int type |
|
|
103 | CODE: |
91 | CODE: |
|
|
92 | { |
104 | pe_watcher *w = GEventAPI->sv_2watcher (self); |
93 | pe_watcher *w = GEventAPI->sv_2watcher (self); |
105 | |
94 | |
106 | if (WaFLAGS (w) & PE_PERLCB) |
95 | if (w->callback) |
107 | croak ("Coro::Event watchers must not have a perl callback (see Coro::Event), caught"); |
96 | croak ("Coro::Event watchers must not have a callback (see Coro::Event), caught"); |
|
|
97 | |
108 | { |
98 | { |
109 | AV *priv = newAV (); |
99 | AV *priv = newAV (); |
110 | SV *rv = newRV_noinc ((SV *)priv); |
|
|
111 | |
100 | |
112 | av_extend (priv, CD_MAX); |
101 | av_fill (priv, CD_MAX); |
113 | av_store (priv, CD_CORO, &PL_sv_undef); |
102 | AvARRAY (priv)[CD_WAIT] = (SV *)newAV (); /* AV in AV _should_ not be exposed to perl */ |
114 | av_store (priv, CD_TYPE, newSViv (type)); |
103 | AvARRAY (priv)[CD_TYPE] = newSViv (type); |
115 | av_store (priv, CD_OK , &PL_sv_no); |
104 | AvARRAY (priv)[CD_OK ] = &PL_sv_no; |
116 | av_store (priv, CD_PRIO, newSViv (0)); |
|
|
117 | av_store (priv, CD_HITS, newSViv (0)); |
105 | AvARRAY (priv)[CD_HITS] = newSViv (0); |
118 | av_store (priv, CD_GOT , type ? newSViv (0) : &PL_sv_undef); |
106 | AvARRAY (priv)[CD_GOT ] = newSViv (0); |
119 | SvREADONLY_on (priv); |
107 | SvREADONLY_on (priv); |
120 | |
108 | |
121 | w->callback = coro_std_cb; |
109 | w->callback = coro_std_cb; |
122 | w->ext_data = priv; |
110 | w->ext_data = priv; |
123 | |
111 | |
124 | hv_store ((HV *)SvRV (self), |
112 | { |
125 | EV_CLASS, strlen (EV_CLASS), |
113 | SV *mob = newRV_noinc ((SV *)priv); |
126 | rv, 0); |
114 | sv_magicext (SvRV (self), mob, PERL_MAGIC_coro_event, 0, (char *)w, 0); |
127 | |
115 | SvREFCNT_dec (mob); /* sv_magicext increments the refcount */ |
128 | GEventAPI->start (w, 0); |
116 | } |
129 | } |
117 | } |
|
|
118 | } |
130 | |
119 | |
131 | void |
120 | void |
132 | _next(self) |
121 | _next (SV *self) |
133 | SV * self |
|
|
134 | CODE: |
122 | CODE: |
|
|
123 | { |
135 | pe_watcher *w = GEventAPI->sv_2watcher (self); |
124 | pe_watcher *w = GEventAPI->sv_2watcher (self); |
136 | AV *priv = (AV *)w->ext_data; |
125 | AV *priv = (AV *)w->ext_data; |
|
|
126 | |
|
|
127 | if (AvARRAY (priv)[CD_OK] == &PL_sv_yes) |
|
|
128 | { |
|
|
129 | AvARRAY (priv)[CD_OK] = &PL_sv_no; |
|
|
130 | XSRETURN_NO; /* got an event */ |
|
|
131 | } |
|
|
132 | |
|
|
133 | av_push ((AV *)AvARRAY (priv)[CD_WAIT], SvREFCNT_inc (CORO_CURRENT)); |
137 | |
134 | |
138 | if (!w->running) |
135 | if (!w->running) |
139 | GEventAPI->start (w, 1); |
136 | GEventAPI->start (w, 1); |
140 | |
137 | |
141 | if (AvARRAY(priv)[CD_OK] == &PL_sv_yes) |
138 | XSRETURN_YES; /* schedule */ |
142 | { |
139 | } |
143 | AvARRAY(priv)[CD_OK] = &PL_sv_no; |
|
|
144 | XSRETURN_NO; |
|
|
145 | } |
|
|
146 | else |
|
|
147 | { |
|
|
148 | if (AvARRAY(priv)[CD_CORO] != &PL_sv_undef) |
|
|
149 | confess ("only one coroutine can wait for an event"); |
|
|
150 | |
140 | |
151 | AvARRAY(priv)[CD_CORO] = SvREFCNT_inc (CORO_CURRENT); |
141 | SV * |
|
|
142 | _event (SV *self) |
|
|
143 | CODE: |
|
|
144 | { |
|
|
145 | if (GIMME_V == G_VOID) |
152 | XSRETURN_YES; |
146 | XSRETURN_EMPTY; |
153 | } |
|
|
154 | |
147 | |
155 | MODULE = Coro::Event PACKAGE = Coro |
148 | { |
|
|
149 | pe_watcher *w = GEventAPI->sv_2watcher (self); |
|
|
150 | AV *priv = (AV *)w->ext_data; |
156 | |
151 | |
157 | # overwrite the ready function |
152 | RETVAL = newRV_inc ((SV *)priv); |
158 | void |
|
|
159 | ready(self) |
|
|
160 | SV * self |
|
|
161 | PROTOTYPE: $ |
|
|
162 | CODE: |
|
|
163 | NEED_SCHEDULE; |
|
|
164 | CORO_READY (self); |
|
|
165 | |
153 | |
|
|
154 | /* may need to bless it now */ |
|
|
155 | if (!SvOBJECT (priv)) |
|
|
156 | { |
|
|
157 | SvREADONLY_off ((SV *)priv); |
|
|
158 | sv_bless (RETVAL, coro_event_event_stash); |
|
|
159 | SvREADONLY_on ((SV *)priv); |
|
|
160 | } |
|
|
161 | } |
|
|
162 | } |
|
|
163 | OUTPUT: |
|
|
164 | RETVAL |
166 | |
165 | |