ViewVC Help
View File | Revision Log | Show Annotations | Download File
/cvs/EV-ADNS/ADNS.xs
(Generate patch)

Comparing EV-ADNS/ADNS.xs (file contents):
Revision 1.1 by root, Sat Dec 1 13:53:11 2007 UTC vs.
Revision 1.9 by root, Sat Dec 1 22:36:10 2007 UTC

… …
5#include <poll.h> 5#include <poll.h>
6#include <adns.h> 6#include <adns.h>
7 7
8#include "EVAPI.h" 8#include "EVAPI.h"
9 9
10static HV *stash;
10static adns_state ads; 11static adns_state ads;
11 12
12struct ctx 13struct ctx
13{ 14{
14 SV *req; 15 SV *self;
16 adns_query query;
15 SV *cb; 17 SV *cb;
16}; 18};
17 19
20static SV *
21ha2sv (adns_rr_hostaddr *rr)
22{
23 AV *av = newAV ();
24 av_push (av, newSVpv (rr->host, 0));
25 // TODO: add addresses
26
27 return newRV_noinc ((SV *)av);
28}
29
18static void 30static void
19process () 31process ()
20{ 32{
33 dSP;
34
21 for (;;) 35 for (;;)
22 { 36 {
37 int i;
23 adns_query q; 38 adns_query q = 0;
24 adns_answer *a; 39 adns_answer *a;
25 void *c; 40 void *ctx;
41 SV *cb;
42 struct ctx *c;
26 int r = adns_check (ads, &q, &a, &c); 43 int r = adns_check (ads, &q, &a, &ctx);
27 44
28 if (r) 45 if (r)
29 break; 46 break;
47
48 c = (struct ctx *)ctx;
49 cb = c->cb;
50 c->cb = 0;
51 ev_unref ();
52 SvREFCNT_dec (c->self);
53
54 PUSHMARK (SP);
55
56 EXTEND (SP, a->nrrs + 2);
57 PUSHs (sv_2mortal (newSViv (a->status)));
58 PUSHs (sv_2mortal (newSViv (a->expires)));
59
60 for (i = 0; i < a->nrrs; ++i)
61 {
62 SV *sv;
63
64 switch (a->type & adns_r_unknown ? adns_r_unknown : a->type)
65 {
66 case adns_r_ns_raw:
67 case adns_r_cname:
68 case adns_r_ptr:
69 case adns_r_ptr_raw:
70 sv = newSVpv (a->rrs.str [i], 0);
71 break;
72
73 case adns_r_txt:
74 {
75 AV *av = newAV ();
76 adns_rr_intstr *rr = a->rrs.manyistr [i];
77
78 while (rr->str)
79 {
80 av_push (av, newSVpvn (rr->str, rr->i));
81 ++rr;
82 }
83
84 sv = newRV_noinc ((SV *)av);
85 }
86 break;
87
88 case adns_r_a:
89 sv = newSVpv (inet_ntoa (a->rrs.inaddr [i]), 0);
90 break;
91
92 case adns_r_ns:
93 sv = ha2sv (a->rrs.hostaddr + i);
94 break;
95
96 case adns_r_hinfo:
97 {
98 /* untested */
99 AV *av = newAV ();
100 adns_rr_intstrpair *rr = a->rrs.intstrpair + i;
101
102 av_push (av, newSVpvn (rr->array [0].str, rr->array [0].i));
103 av_push (av, newSVpvn (rr->array [1].str, rr->array [1].i));
104
105 sv = newRV_noinc ((SV *)av);
106 }
107 break;
108
109 case adns_r_rp:
110 case adns_r_rp_raw:
111 {
112 /* untested */
113 AV *av = newAV ();
114 adns_rr_strpair *rr = a->rrs.strpair + i;
115
116 av_push (av, newSVpv (rr->array [0], 0));
117 av_push (av, newSVpv (rr->array [1], 0));
118
119 sv = newRV_noinc ((SV *)av);
120 }
121 break;
122
123 case adns_r_mx:
124 {
125 AV *av = newAV ();
126 adns_rr_inthostaddr *rr = a->rrs.inthostaddr + i;
127
128 av_push (av, newSViv (rr->i));
129 av_push (av, ha2sv (&rr->ha));
130
131 sv = newRV_noinc ((SV *)av);
132 }
133 break;
134
135 case adns_r_mx_raw:
136 {
137 AV *av = newAV ();
138 adns_rr_intstr *rr = a->rrs.intstr + i;
139
140 av_push (av, newSViv (rr->i));
141 av_push (av, newSVpv (rr->str, 0));
142
143 sv = newRV_noinc ((SV *)av);
144 }
145 break;
146
147 case adns_r_soa:
148 case adns_r_soa_raw:
149 {
150 AV *av = newAV ();
151 adns_rr_soa *rr = a->rrs.soa + i;
152
153 av_push (av, newSVpv (rr->mname, 0));
154 av_push (av, newSVpv (rr->rname, 0));
155 av_push (av, newSVuv (rr->serial));
156 av_push (av, newSVuv (rr->refresh));
157 av_push (av, newSVuv (rr->retry));
158 av_push (av, newSVuv (rr->expire));
159 av_push (av, newSVuv (rr->minimum));
160
161 sv = newRV_noinc ((SV *)av);
162 }
163 break;
164
165 case adns_r_srv_raw:
166 {
167 AV *av = newAV ();
168 adns_rr_srvraw *rr = a->rrs.srvraw + i;
169
170 av_push (av, newSViv (rr->priority));
171 av_push (av, newSViv (rr->weight));
172 av_push (av, newSViv (rr->port));
173 av_push (av, newSVpv (rr->host, 0));
174
175 sv = newRV_noinc ((SV *)av);
176 }
177 break;
178
179 case adns_r_srv:
180 {
181 AV *av = newAV ();
182 adns_rr_srvha *rr = a->rrs.srvha + i;
183
184 av_push (av, newSViv (rr->priority));
185 av_push (av, newSViv (rr->weight));
186 av_push (av, newSViv (rr->port));
187 av_push (av, ha2sv (&rr->ha));
188
189 sv = newRV_noinc ((SV *)av);
190 }
191 break;
192
193 case adns_r_unknown:
194 sv = newSVpvn (a->rrs.byteblock [i].data, a->rrs.byteblock [i].len);
195 break;
196
197 default:
198 case adns_r_addr:
199 sv = &PL_sv_undef; /* not supported */
200 break;
201 }
202
203 PUSHs (sv_2mortal (sv));
204 }
205
206 free (a);
207
208 PUTBACK;
209 call_sv (cb, G_VOID | G_DISCARD | G_EVAL);
210 SPAGAIN;
211
212 SvREFCNT_dec (cb);
30 } 213 }
31} 214}
32 215
33static struct pollfd *fds; 216static struct pollfd *fds;
34static int nfd; 217static int nfd, mfd;
35static ev_io *iow; 218static ev_io *iow;
36static ev_timer tw; 219static ev_timer tw;
220static ev_idle iw;
37static ev_prepare prepare_ev; 221static ev_prepare prepare_ev;
38static struct timeval *tv_now; 222static struct timeval tv_now;
39 223
40static void 224static void
41update_now (EV_P) 225update_now (EV_P)
42{ 226{
43 ev_tstamp t = ev_now (EV_P); 227 ev_tstamp t = ev_now ();
44 228
45 tv.tv_sec = (long)t; 229 tv_now.tv_sec = (long)t;
46 tv.tv_usec = (long)((t - (ev_tstamp)tv.tv_sec) * 1e-6); 230 tv_now.tv_usec = (long)((t - (ev_tstamp)tv_now.tv_sec) * 1e-6);
231}
232
233static void
234idle_cb (EV_P_ ev_idle *w, int revents)
235{
236 ev_idle_stop (EV_A_ w);
47} 237}
48 238
49static void 239static void
50timer_cb (EV_P_ ev_timer *w, int revents) 240timer_cb (EV_P_ ev_timer *w, int revents)
51{ 241{
… …
65prepare_cb (EV_P_ ev_prepare *w, int revents) 255prepare_cb (EV_P_ ev_prepare *w, int revents)
66{ 256{
67 int i; 257 int i;
68 int timeout = 3600000; 258 int timeout = 3600000;
69 259
260 if (ev_is_active (&tw))
261 {
262 ev_ref ();
70 ev_timer_stop (EV_A_ &tw); 263 ev_timer_stop (EV_A_ &tw);
264 }
265
266 if (ev_is_active (&iw))
267 ev_idle_stop (EV_A_ &iw);
71 268
72 for (i = 0; i < nfd; ++i) 269 for (i = 0; i < nfd; ++i)
270 {
271 ev_ref ();
73 ev_io_stop (EV_A_ iow + i); 272 ev_io_stop (EV_A_ iow + i);
273 }
74 274
75 process (); 275 process ();
76 276
77 update_now (EV_A); 277 update_now (EV_A);
278
279 nfd = mfd;
78 280
79 while (adns_beforepoll (ads, fds, &nfd, &timeout, &tv_now)) 281 while (adns_beforepoll (ads, fds, &nfd, &timeout, &tv_now))
80 { 282 {
283 mfd = nfd;
284
285 free (iow); iow = malloc (mfd * sizeof (ev_io));
81 free (iow); iow = malloc (nfd * sizeof (struct pollfd)); 286 free (fds); fds = malloc (mfd * sizeof (struct pollfd));
82 free (fds); fds = malloc (nfd * sizeof (ev_io));
83 } 287 }
84 288
85 ev_timer_set (&tw, timeout * 1e-3, 0.); 289 ev_timer_set (&tw, timeout * 1e-3, 0.);
86 ev_timer_start (EV_A_ &tw); 290 ev_timer_start (EV_A_ &tw);
291 ev_unref ();
87 292
88 // create on ev_io per pollfd 293 // create one ev_io per pollfd
89 for (i = 0; i < nfd; ++i) 294 for (i = 0; i < nfd; ++i)
90 { 295 {
91 ev_io_init (iow + i, io_cb, fds [i].fd, 296 ev_io_init (iow + i, io_cb, fds [i].fd,
92 ((fds [i].events & POLLIN ? EV_READ : 0) 297 ((fds [i].events & POLLIN ? EV_READ : 0)
93 | (fds [i].events & POLLOUT ? EV_WRITE : 0))); 298 | (fds [i].events & POLLOUT ? EV_WRITE : 0)));
94 299
95 ev_io_start (EV_A_ iow + i); 300 ev_io_start (EV_A_ iow + i);
301 ev_unref ();
96 } 302 }
97} 303}
98 304
99MODULE = EV::ADNS PACKAGE = EV::ADNS 305MODULE = EV::ADNS PACKAGE = EV::ADNS
100 306
101PROTOTYPES: ENABLE 307PROTOTYPES: ENABLE
102 308
103BOOT: 309BOOT:
104{ 310{
105#if 0
106 HV *stash = gv_stashpv ("EV::ADNS", 1); 311 stash = gv_stashpv ("EV::ADNS", 1);
107 312
108 static const struct { 313 static const struct {
109 const char *name; 314 const char *name;
110 IV iv; 315 IV iv;
111 } *civ, const_iv[] = { 316 } *civ, const_iv[] = {
112# define const_iv(pfx, name) { # name, (IV) adns_ ## name }, 317# define const_iv(name) { # name, (IV) adns_ ## name },
113 const_iv (if_none) 318 const_iv (if_none)
114 const_iv (if_noenv) 319 const_iv (if_noenv)
115 const_iv (if_noerrprint) 320 const_iv (if_noerrprint)
116 const_iv (if_noserverwarn) 321 const_iv (if_noserverwarn)
117 const_iv (if_debug) 322 const_iv (if_debug)
… …
188 const_iv (s_max_permfail) 393 const_iv (s_max_permfail)
189 }; 394 };
190 395
191 for (civ = const_iv + sizeof (const_iv) / sizeof (const_iv [0]); civ-- > const_iv; ) 396 for (civ = const_iv + sizeof (const_iv) / sizeof (const_iv [0]); civ-- > const_iv; )
192 newCONSTSUB (stash, (char *)civ->name, newSViv (civ->iv)); 397 newCONSTSUB (stash, (char *)civ->name, newSViv (civ->iv));
193#endif
194 398
195 I_EV_API ("EV::ADNS"); 399 I_EV_API ("EV::ADNS");
196 400
197 ev_prepare_init (&prepare_ev, prepare_cb); ev_prepare_start (EV_DEFAULT_ &prepare_ev); 401 ev_prepare_init (&prepare_ev, prepare_cb);
402 ev_prepare_start (EV_DEFAULT_ &prepare_ev);
198 ev_unref (); 403 ev_unref ();
199 404
405 ev_init (&iw, idle_cb);
200 ev_init (&tw, timer_cb); 406 ev_init (&tw, timer_cb);
201 407
202 adns_init (&ads, adns_if_noenv | adns_if_noerrprint | adns_if_noserverwarn | adns_if_noautosys, 0); 408 adns_init (&ads, adns_if_noenv | adns_if_noerrprint | adns_if_noserverwarn | adns_if_noautosys, 0);
203} 409}
204 410
205int adns_submit (char *owner, int type, int flags, SV *cb) 411void submit (char *owner, int type, int flags, SV *cb)
206 CODE: 412 PPCODE:
207{ 413{
414 SV *csv = NEWSV (0, sizeof (struct ctx));
208 struct ctx *c = (struct ctx *)malloc (sizeof (ctx)); 415 struct ctx *c = (struct ctx *)SvPVX (csv);
209 adns_query q;
210 int r = adns_submit (owner, type, flags, (void *)c, &q); 416 int r = adns_submit (ads, owner, type, flags, (void *)c, &c->query);
211 417
212 if (r) 418 if (r)
213 { 419 {
214 free (c); 420 SvREFCNT_dec (csv);
421 errno = r;
215 XSRETURN_EMPTY; 422 XSRETURN_EMPTY;
216 } 423 }
217} 424 else
425 {
426 ev_ref ();
427 SvPOK_only (csv);
428 SvCUR_set (csv, sizeof (struct ctx));
218 429
430 c->self = csv;
431 c->cb = newSVsv (cb);
219 432
433 if (!ev_is_active (&iw))
434 ev_idle_start (EV_A_ &iw);
220 435
436 if (GIMME_V != G_VOID)
437 {
438 csv = sv_2mortal (newRV_inc (csv));
439 sv_bless (csv, stash);
440 XPUSHs (csv);
441 }
442 }
443}
444
445void DESTROY (SV *req)
446 ALIAS:
447 cancel = 1
448 CODE:
449{
450 struct ctx *c;
451
452 if (!(SvROK (req) && SvOBJECT (SvRV (req))
453 && (SvSTASH (SvRV (req)) == stash)))
454 croak ("object is not of type EV::ADNS");
455
456 c = (struct ctx *)SvPVX (SvRV (req));
457
458 if (c->cb)
459 {
460 ev_unref ();
461 SvREFCNT_dec (c->cb);
462 c->cb = 0;
463 adns_cancel (c->query);
464 SvREFCNT_dec (c->self);
465 }
466}
467
468
469
470

Diff Legend

– Removed lines
+ Added lines
< Changed lines
> Changed lines