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.6 by root, Sat Dec 1 20:52:30 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 printf ("refrefclr\n");
54
55 PUSHMARK (SP);
56
57 EXTEND (SP, a->nrrs + 2);
58 PUSHs (sv_2mortal (newSViv (a->status)));
59 PUSHs (sv_2mortal (newSViv (a->expires)));
60
61 for (i = 0; i < a->nrrs; ++i)
62 {
63 SV *sv;
64
65 switch (a->type & adns_r_unknown ? adns_r_unknown : a->type)
66 {
67 case adns_r_ns_raw:
68 case adns_r_cname:
69 case adns_r_ptr:
70 case adns_r_ptr_raw:
71 sv = newSVpv (a->rrs.str [i], 0);
72 break;
73
74 case adns_r_txt:
75 sv = newSVpvn (a->rrs.manyistr [i]->str, a->rrs.manyistr [i]->i);
76 break;
77
78 case adns_r_a:
79 sv = newSVpv (inet_ntoa (a->rrs.inaddr [i]), 0);
80 break;
81
82 case adns_r_ns:
83 sv = ha2sv (a->rrs.hostaddr + i);
84 break;
85
86 case adns_r_hinfo:
87 {
88 /* untested */
89 AV *av = newAV ();
90 sv = newRV_noinc ((SV *)av);
91 adns_rr_intstrpair *rr = a->rrs.intstrpair + i;
92
93 av_push (av, newSVpvn (rr->array [0].str, rr->array [0].i));
94 av_push (av, newSVpvn (rr->array [1].str, rr->array [1].i));
95 }
96 break;
97
98 case adns_r_rp:
99 case adns_r_rp_raw:
100 {
101 /* untested */
102 AV *av = newAV ();
103 sv = newRV_noinc ((SV *)av);
104 adns_rr_strpair *rr = a->rrs.strpair + i;
105
106 av_push (av, newSVpv (rr->array [0], 0));
107 av_push (av, newSVpv (rr->array [1], 0));
108 }
109 break;
110
111 case adns_r_mx:
112 {
113 AV *av = newAV ();
114 sv = newRV_noinc ((SV *)av);
115 adns_rr_inthostaddr *rr = a->rrs.inthostaddr + i;
116
117 av_push (av, newSViv (rr->i));
118 av_push (av, ha2sv (&rr->ha));
119 }
120 break;
121
122 case adns_r_mx_raw:
123 {
124 AV *av = newAV ();
125 sv = newRV_noinc ((SV *)av);
126 adns_rr_intstr *rr = a->rrs.intstr + i;
127
128 av_push (av, newSViv (rr->i));
129 av_push (av, newSVpv (rr->str, 0));
130 }
131 break;
132
133 case adns_r_soa:
134 case adns_r_soa_raw:
135 {
136 AV *av = newAV ();
137 sv = newRV_noinc ((SV *)av);
138 adns_rr_soa *rr = a->rrs.soa + i;
139
140 av_push (av, newSVpv (rr->mname, 0));
141 av_push (av, newSVpv (rr->rname, 0));
142 av_push (av, newSVuv (rr->serial));
143 av_push (av, newSVuv (rr->refresh));
144 av_push (av, newSVuv (rr->retry));
145 av_push (av, newSVuv (rr->expire));
146 av_push (av, newSVuv (rr->minimum));
147 }
148 break;
149
150 case adns_r_srv_raw:
151 {
152 AV *av = newAV ();
153 sv = newRV_noinc ((SV *)av);
154 adns_rr_srvraw *rr = a->rrs.srvraw + i;
155
156 av_push (av, newSViv (rr->priority));
157 av_push (av, newSViv (rr->weight));
158 av_push (av, newSViv (rr->port));
159 av_push (av, newSVpv (rr->host, 0));
160 }
161 break;
162
163 case adns_r_srv:
164 {
165 AV *av = newAV ();
166 sv = newRV_noinc ((SV *)av);
167 adns_rr_srvha *rr = a->rrs.srvha + i;
168
169 av_push (av, newSViv (rr->priority));
170 av_push (av, newSViv (rr->weight));
171 av_push (av, newSViv (rr->port));
172 av_push (av, ha2sv (&rr->ha));
173 }
174 break;
175
176 case adns_r_unknown:
177 sv = newSVpvn (a->rrs.byteblock [i].data, a->rrs.byteblock [i].len);
178 break;
179
180 default:
181 case adns_r_addr:
182 sv = &PL_sv_undef; /* not supported */
183 break;
184 }
185
186 PUSHs (sv_2mortal (sv));
187 }
188
189 free (a);
190
191 PUTBACK;
192 call_sv (cb, G_VOID | G_DISCARD | G_EVAL);
193 SPAGAIN;
194
195 SvREFCNT_dec (cb);
30 } 196 }
31} 197}
32 198
33static struct pollfd *fds; 199static struct pollfd *fds;
34static int nfd; 200static int nfd, mfd;
35static ev_io *iow; 201static ev_io *iow;
36static ev_timer tw; 202static ev_timer tw;
37static ev_prepare prepare_ev; 203static ev_prepare prepare_ev;
38static struct timeval *tv_now; 204static struct timeval tv_now;
39 205
40static void 206static void
41update_now (EV_P) 207update_now (EV_P)
42{ 208{
43 ev_tstamp t = ev_now (EV_P); 209 ev_tstamp t = ev_now ();
44 210
45 tv.tv_sec = (long)t; 211 tv_now.tv_sec = (long)t;
46 tv.tv_usec = (long)((t - (ev_tstamp)tv.tv_sec) * 1e-6); 212 tv_now.tv_usec = (long)((t - (ev_tstamp)tv_now.tv_sec) * 1e-6);
47} 213}
48 214
49static void 215static void
50timer_cb (EV_P_ ev_timer *w, int revents) 216timer_cb (EV_P_ ev_timer *w, int revents)
51{ 217{
… …
65prepare_cb (EV_P_ ev_prepare *w, int revents) 231prepare_cb (EV_P_ ev_prepare *w, int revents)
66{ 232{
67 int i; 233 int i;
68 int timeout = 3600000; 234 int timeout = 3600000;
69 235
236 if (ev_is_active (&tw))
237 {
238 ev_ref ();
70 ev_timer_stop (EV_A_ &tw); 239 ev_timer_stop (EV_A_ &tw);
240 }
71 241
72 for (i = 0; i < nfd; ++i) 242 for (i = 0; i < nfd; ++i)
243 {
244 ev_ref ();
73 ev_io_stop (EV_A_ iow + i); 245 ev_io_stop (EV_A_ iow + i);
246 }
74 247
75 process (); 248 process ();
76 249
77 update_now (EV_A); 250 update_now (EV_A);
251
252 nfd = mfd;
78 253
79 while (adns_beforepoll (ads, fds, &nfd, &timeout, &tv_now)) 254 while (adns_beforepoll (ads, fds, &nfd, &timeout, &tv_now))
80 { 255 {
256 mfd = nfd;
257
81 free (iow); iow = malloc (nfd * sizeof (struct pollfd)); 258 free (iow); iow = malloc (mfd * sizeof (struct pollfd));
82 free (fds); fds = malloc (nfd * sizeof (ev_io)); 259 free (fds); fds = malloc (mfd * sizeof (ev_io));
83 } 260 }
84 261
85 ev_timer_set (&tw, timeout * 1e-3, 0.); 262 ev_timer_set (&tw, timeout * 1e-3, 0.);
86 ev_timer_start (EV_A_ &tw); 263 ev_timer_start (EV_A_ &tw);
264 ev_unref ();
87 265
88 // create on ev_io per pollfd 266 // create one ev_io per pollfd
89 for (i = 0; i < nfd; ++i) 267 for (i = 0; i < nfd; ++i)
90 { 268 {
91 ev_io_init (iow + i, io_cb, fds [i].fd, 269 ev_io_init (iow + i, io_cb, fds [i].fd,
92 ((fds [i].events & POLLIN ? EV_READ : 0) 270 ((fds [i].events & POLLIN ? EV_READ : 0)
93 | (fds [i].events & POLLOUT ? EV_WRITE : 0))); 271 | (fds [i].events & POLLOUT ? EV_WRITE : 0)));
94 272
95 ev_io_start (EV_A_ iow + i); 273 ev_io_start (EV_A_ iow + i);
274 ev_unref ();
96 } 275 }
97} 276}
98 277
99MODULE = EV::ADNS PACKAGE = EV::ADNS 278MODULE = EV::ADNS PACKAGE = EV::ADNS
100 279
101PROTOTYPES: ENABLE 280PROTOTYPES: ENABLE
102 281
103BOOT: 282BOOT:
104{ 283{
105#if 0
106 HV *stash = gv_stashpv ("EV::ADNS", 1); 284 stash = gv_stashpv ("EV::ADNS", 1);
107 285
108 static const struct { 286 static const struct {
109 const char *name; 287 const char *name;
110 IV iv; 288 IV iv;
111 } *civ, const_iv[] = { 289 } *civ, const_iv[] = {
112# define const_iv(pfx, name) { # name, (IV) adns_ ## name }, 290# define const_iv(name) { # name, (IV) adns_ ## name },
113 const_iv (if_none) 291 const_iv (if_none)
114 const_iv (if_noenv) 292 const_iv (if_noenv)
115 const_iv (if_noerrprint) 293 const_iv (if_noerrprint)
116 const_iv (if_noserverwarn) 294 const_iv (if_noserverwarn)
117 const_iv (if_debug) 295 const_iv (if_debug)
… …
188 const_iv (s_max_permfail) 366 const_iv (s_max_permfail)
189 }; 367 };
190 368
191 for (civ = const_iv + sizeof (const_iv) / sizeof (const_iv [0]); civ-- > const_iv; ) 369 for (civ = const_iv + sizeof (const_iv) / sizeof (const_iv [0]); civ-- > const_iv; )
192 newCONSTSUB (stash, (char *)civ->name, newSViv (civ->iv)); 370 newCONSTSUB (stash, (char *)civ->name, newSViv (civ->iv));
193#endif
194 371
195 I_EV_API ("EV::ADNS"); 372 I_EV_API ("EV::ADNS");
196 373
197 ev_prepare_init (&prepare_ev, prepare_cb); ev_prepare_start (EV_DEFAULT_ &prepare_ev); 374 ev_prepare_init (&prepare_ev, prepare_cb); ev_prepare_start (EV_DEFAULT_ &prepare_ev);
198 ev_unref (); 375 ev_unref ();
… …
200 ev_init (&tw, timer_cb); 377 ev_init (&tw, timer_cb);
201 378
202 adns_init (&ads, adns_if_noenv | adns_if_noerrprint | adns_if_noserverwarn | adns_if_noautosys, 0); 379 adns_init (&ads, adns_if_noenv | adns_if_noerrprint | adns_if_noserverwarn | adns_if_noautosys, 0);
203} 380}
204 381
205int adns_submit (char *owner, int type, int flags, SV *cb) 382void submit (char *owner, int type, int flags, SV *cb)
206 CODE: 383 PPCODE:
207{ 384{
385 SV *csv = NEWSV (0, sizeof (struct ctx));
208 struct ctx *c = (struct ctx *)malloc (sizeof (ctx)); 386 struct ctx *c = (struct ctx *)SvPVX (csv);
209 adns_query q;
210 int r = adns_submit (owner, type, flags, (void *)c, &q); 387 int r = adns_submit (ads, owner, type, flags, (void *)c, &c->query);
211 388
212 if (r) 389 if (r)
213 { 390 {
214 free (c); 391 SvREFCNT_dec (csv);
392 errno = r;
215 XSRETURN_EMPTY; 393 XSRETURN_EMPTY;
216 } 394 }
217} 395 else
396 {
397 ev_ref ();
398 SvPOK_only (csv);
399 SvCUR_set (csv, sizeof (struct ctx));
218 400
401 c->self = csv;
402 c->cb = newSVsv (cb);
219 403
404 if (GIMME_V != G_VOID)
405 {
406 csv = sv_2mortal (newRV_inc (csv));
407 sv_bless (csv, stash);
408 XPUSHs (csv);
409 }
410 }
411}
220 412
413void DESTROY (SV *req)
414 ALIAS:
415 cancel = 1
416 CODE:
417{
418 struct ctx *c;
419
420 printf ("destroy\n");
421 if (!(SvROK (req) && SvOBJECT (SvRV (req))
422 && (SvSTASH (SvRV (req)) == stash)))
423 croak ("object is not of type EV::ADNS");
424
425 c = (struct ctx *)SvPVX (SvRV (req));
426
427 if (c->cb)
428 {
429 ev_unref ();
430 SvREFCNT_dec (c->cb);
431 c->cb = 0;
432 adns_cancel (c->query);
433 SvREFCNT_dec (c->self);
434 printf ("refrefcl2\n");
435 }
436}
437
438
439
440

Diff Legend

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