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.3 by root, Sat Dec 1 14:37:46 2007 UTC vs.
Revision 1.9 by root, Sat Dec 1 22:36:10 2007 UTC

15 SV *self; 15 SV *self;
16 adns_query query; 16 adns_query query;
17 SV *cb; 17 SV *cb;
18}; 18};
19 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
20static void 30static void
21process () 31process ()
22{ 32{
23 dSP; 33 dSP;
24 34
25 for (;;) 35 for (;;)
26 { 36 {
37 int i;
27 adns_query q; 38 adns_query q = 0;
28 adns_answer *a; 39 adns_answer *a;
29 void *ctx; 40 void *ctx;
30 SV *cb; 41 SV *cb;
31 struct ctx *c; 42 struct ctx *c;
32 int r = adns_check (ads, &q, &a, &ctx); 43 int r = adns_check (ads, &q, &a, &ctx);
36 47
37 c = (struct ctx *)ctx; 48 c = (struct ctx *)ctx;
38 cb = c->cb; 49 cb = c->cb;
39 c->cb = 0; 50 c->cb = 0;
40 ev_unref (); 51 ev_unref ();
52 SvREFCNT_dec (c->self);
41 53
42 PUSHMARK (SP); 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;
43 call_sv (cb, G_VOID | G_DISCARD | G_EVAL); 209 call_sv (cb, G_VOID | G_DISCARD | G_EVAL);
210 SPAGAIN;
44 211
45 SvREFCNT_dec (cb); 212 SvREFCNT_dec (cb);
46 } 213 }
47} 214}
48 215
49static struct pollfd *fds; 216static struct pollfd *fds;
50static int nfd; 217static int nfd, mfd;
51static ev_io *iow; 218static ev_io *iow;
52static ev_timer tw; 219static ev_timer tw;
220static ev_idle iw;
53static ev_prepare prepare_ev; 221static ev_prepare prepare_ev;
54static struct timeval tv_now; 222static struct timeval tv_now;
55 223
56static void 224static void
57update_now (EV_P) 225update_now (EV_P)
61 tv_now.tv_sec = (long)t; 229 tv_now.tv_sec = (long)t;
62 tv_now.tv_usec = (long)((t - (ev_tstamp)tv_now.tv_sec) * 1e-6); 230 tv_now.tv_usec = (long)((t - (ev_tstamp)tv_now.tv_sec) * 1e-6);
63} 231}
64 232
65static void 233static void
234idle_cb (EV_P_ ev_idle *w, int revents)
235{
236 ev_idle_stop (EV_A_ w);
237}
238
239static void
66timer_cb (EV_P_ ev_timer *w, int revents) 240timer_cb (EV_P_ ev_timer *w, int revents)
67{ 241{
68} 242}
69 243
70static void 244static void
81prepare_cb (EV_P_ ev_prepare *w, int revents) 255prepare_cb (EV_P_ ev_prepare *w, int revents)
82{ 256{
83 int i; 257 int i;
84 int timeout = 3600000; 258 int timeout = 3600000;
85 259
260 if (ev_is_active (&tw))
261 {
262 ev_ref ();
86 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);
87 268
88 for (i = 0; i < nfd; ++i) 269 for (i = 0; i < nfd; ++i)
270 {
271 ev_ref ();
89 ev_io_stop (EV_A_ iow + i); 272 ev_io_stop (EV_A_ iow + i);
273 }
90 274
91 process (); 275 process ();
92 276
93 update_now (EV_A); 277 update_now (EV_A);
278
279 nfd = mfd;
94 280
95 while (adns_beforepoll (ads, fds, &nfd, &timeout, &tv_now)) 281 while (adns_beforepoll (ads, fds, &nfd, &timeout, &tv_now))
96 { 282 {
283 mfd = nfd;
284
285 free (iow); iow = malloc (mfd * sizeof (ev_io));
97 free (iow); iow = malloc (nfd * sizeof (struct pollfd)); 286 free (fds); fds = malloc (mfd * sizeof (struct pollfd));
98 free (fds); fds = malloc (nfd * sizeof (ev_io));
99 } 287 }
100 288
101 ev_timer_set (&tw, timeout * 1e-3, 0.); 289 ev_timer_set (&tw, timeout * 1e-3, 0.);
102 ev_timer_start (EV_A_ &tw); 290 ev_timer_start (EV_A_ &tw);
291 ev_unref ();
103 292
104 // create on ev_io per pollfd 293 // create one ev_io per pollfd
105 for (i = 0; i < nfd; ++i) 294 for (i = 0; i < nfd; ++i)
106 { 295 {
107 ev_io_init (iow + i, io_cb, fds [i].fd, 296 ev_io_init (iow + i, io_cb, fds [i].fd,
108 ((fds [i].events & POLLIN ? EV_READ : 0) 297 ((fds [i].events & POLLIN ? EV_READ : 0)
109 | (fds [i].events & POLLOUT ? EV_WRITE : 0))); 298 | (fds [i].events & POLLOUT ? EV_WRITE : 0)));
110 299
111 ev_io_start (EV_A_ iow + i); 300 ev_io_start (EV_A_ iow + i);
301 ev_unref ();
112 } 302 }
113} 303}
114 304
115MODULE = EV::ADNS PACKAGE = EV::ADNS 305MODULE = EV::ADNS PACKAGE = EV::ADNS
116 306
206 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; )
207 newCONSTSUB (stash, (char *)civ->name, newSViv (civ->iv)); 397 newCONSTSUB (stash, (char *)civ->name, newSViv (civ->iv));
208 398
209 I_EV_API ("EV::ADNS"); 399 I_EV_API ("EV::ADNS");
210 400
211 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);
212 ev_unref (); 403 ev_unref ();
213 404
405 ev_init (&iw, idle_cb);
214 ev_init (&tw, timer_cb); 406 ev_init (&tw, timer_cb);
215 407
216 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);
217} 409}
218 410
223 struct ctx *c = (struct ctx *)SvPVX (csv); 415 struct ctx *c = (struct ctx *)SvPVX (csv);
224 int r = adns_submit (ads, owner, type, flags, (void *)c, &c->query); 416 int r = adns_submit (ads, owner, type, flags, (void *)c, &c->query);
225 417
226 if (r) 418 if (r)
227 { 419 {
228 printf ("errn %d\n", r);//D
229 SvREFCNT_dec (csv); 420 SvREFCNT_dec (csv);
421 errno = r;
230 XSRETURN_EMPTY; 422 XSRETURN_EMPTY;
231 } 423 }
232 else 424 else
233 { 425 {
234 ev_ref (); 426 ev_ref ();
235 SvPOK_only (csv); 427 SvPOK_only (csv);
236 SvCUR_set (csv, sizeof (struct ctx)); 428 SvCUR_set (csv, sizeof (struct ctx));
429
237 c->self = csv; 430 c->self = csv;
238 c->cb = newSVsv (cb); 431 c->cb = newSVsv (cb);
432
433 if (!ev_is_active (&iw))
434 ev_idle_start (EV_A_ &iw);
239 435
240 if (GIMME_V != G_VOID) 436 if (GIMME_V != G_VOID)
241 { 437 {
242 csv = sv_2mortal (newRV_inc (csv)); 438 csv = sv_2mortal (newRV_inc (csv));
243 sv_bless (csv, stash); 439 sv_bless (csv, stash);
245 } 441 }
246 } 442 }
247} 443}
248 444
249void DESTROY (SV *req) 445void DESTROY (SV *req)
446 ALIAS:
447 cancel = 1
250 CODE: 448 CODE:
251{ 449{
252 struct ctx *c; 450 struct ctx *c;
253 451
254 if (!(SvROK (req) && SvOBJECT (SvRV (req)) 452 if (!(SvROK (req) && SvOBJECT (SvRV (req))
258 c = (struct ctx *)SvPVX (SvRV (req)); 456 c = (struct ctx *)SvPVX (SvRV (req));
259 457
260 if (c->cb) 458 if (c->cb)
261 { 459 {
262 ev_unref (); 460 ev_unref ();
461 SvREFCNT_dec (c->cb);
462 c->cb = 0;
263 adns_cancel (c->query); 463 adns_cancel (c->query);
264 SvREFCNT_dec (c->cb); 464 SvREFCNT_dec (c->self);
265 } 465 }
266
267 SvREFCNT_dec (c->self);
268} 466}
269 467
270 468
271 469
272 470

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines