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

Comparing cvsroot/EV-ADNS/ADNS.xs (file contents):
Revision 1.11 by root, Fri Dec 7 23:41:12 2007 UTC vs.
Revision 1.16 by root, Thu Dec 20 07:13:36 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 struct pollfd *fds;
11static adns_state ads; 11static int nfd, mfd;
12static ev_io *iow;
13static ev_timer tw;
14static ev_idle iw;
15static ev_prepare pw;
16static struct timeval tv_now;
17static int outstanding;
18
19static void
20outstanding_inc (adns_state ads)
21{
22 if (!outstanding++)
23 ev_prepare_start (EV_DEFAULT, &pw);
24}
25
26static void
27outstanding_dec (adns_state ads)
28{
29 --outstanding;
30}
12 31
13struct ctx 32struct ctx
14{ 33{
15 SV *self; 34 SV *self;
35 adns_state ads;
16 adns_query query; 36 adns_query query;
17 SV *cb; 37 SV *cb;
18}; 38};
19 39
20static SV * 40static SV *
26 46
27 return newRV_noinc ((SV *)av); 47 return newRV_noinc ((SV *)av);
28} 48}
29 49
30static void 50static void
31process () 51process (adns_state ads)
32{ 52{
33 dSP; 53 dSP;
34 54
35 for (;;) 55 for (;;)
36 { 56 {
45 if (r) 65 if (r)
46 break; 66 break;
47 67
48 c = (struct ctx *)ctx; 68 c = (struct ctx *)ctx;
49 cb = c->cb; 69 cb = c->cb;
50 c->cb = 0; 70 c->cb = 0; outstanding_dec (ads);
51 ev_unref ();
52 SvREFCNT_dec (c->self); 71 SvREFCNT_dec (c->self);
72
73 assert (cb);
53 74
54 PUSHMARK (SP); 75 PUSHMARK (SP);
55 76
56 EXTEND (SP, a->nrrs + 2); 77 EXTEND (SP, a->nrrs + 2);
57 PUSHs (sv_2mortal (newSViv (a->status))); 78 PUSHs (sv_2mortal (newSViv (a->status)));
207 228
208 PUTBACK; 229 PUTBACK;
209 call_sv (cb, G_VOID | G_DISCARD | G_EVAL); 230 call_sv (cb, G_VOID | G_DISCARD | G_EVAL);
210 SPAGAIN; 231 SPAGAIN;
211 232
233 if (SvTRUE (ERRSV))
234 warn ("%s", SvPV_nolen (ERRSV));
235
212 SvREFCNT_dec (cb); 236 SvREFCNT_dec (cb);
213 } 237 }
214} 238}
215 239
216static struct pollfd *fds;
217static int nfd, mfd;
218static ev_io *iow;
219static ev_timer tw;
220static ev_idle iw;
221static ev_prepare prepare_ev;
222static struct timeval tv_now;
223
224static void 240static void
225update_now (EV_P) 241update_now (EV_P)
226{ 242{
227 ev_tstamp t = ev_now (); 243 ev_tstamp t = ev_now (EV_A);
228 244
229 tv_now.tv_sec = (long)t; 245 tv_now.tv_sec = (long)t;
230 tv_now.tv_usec = (long)((t - (ev_tstamp)tv_now.tv_sec) * 1e6); 246 tv_now.tv_usec = (long)((t - (ev_tstamp)tv_now.tv_sec) * 1e6);
231} 247}
232 248
233static void 249static void
234idle_cb (EV_P_ ev_idle *w, int revents) 250idle_cb (EV_P_ ev_idle *w, int revents)
235{ 251{
236 ev_idle_stop (EV_A_ w); 252 ev_idle_stop (EV_A, w);
237} 253}
238 254
239static void 255static void
240timer_cb (EV_P_ ev_timer *w, int revents) 256timer_cb (EV_P_ ev_timer *w, int revents)
241{ 257{
258 adns_state ads = (adns_state)w->data;
242 update_now (); 259 update_now (EV_A);
243 260
244 adns_processtimeouts (ads, &tv_now); 261 adns_processtimeouts (ads, &tv_now);
245} 262}
246 263
247static void 264static void
248io_cb (EV_P_ ev_io *w, int revents) 265io_cb (EV_P_ ev_io *w, int revents)
249{ 266{
267 adns_state ads = (adns_state)w->data;
250 update_now (EV_A); 268 update_now (EV_A);
251 269
252 if (revents & EV_READ ) adns_processreadable (ads, w->fd, &tv_now); 270 if (revents & EV_READ ) adns_processreadable (ads, w->fd, &tv_now);
253 if (revents & EV_WRITE) adns_processwriteable (ads, w->fd, &tv_now); 271 if (revents & EV_WRITE) adns_processwriteable (ads, w->fd, &tv_now);
254} 272}
257static void 275static void
258prepare_cb (EV_P_ ev_prepare *w, int revents) 276prepare_cb (EV_P_ ev_prepare *w, int revents)
259{ 277{
260 int i; 278 int i;
261 int timeout = 3600000; 279 int timeout = 3600000;
280 adns_state ads = (adns_state)w->data;
262 281
263 if (ev_is_active (&tw)) 282 if (ev_is_active (&tw))
283 ev_timer_stop (EV_A, &tw);
284
285 if (ev_is_active (&iw))
286 ev_idle_stop (EV_A, &iw);
287
288 for (i = 0; i < nfd; ++i)
289 ev_io_stop (EV_A, iow + i);
290
291 process (ads);
292
293 if (!outstanding)
264 { 294 {
265 ev_ref ();
266 ev_timer_stop (EV_A_ &tw); 295 ev_prepare_stop (EV_A, w);
296 return;
267 } 297 }
268
269 if (ev_is_active (&iw))
270 ev_idle_stop (EV_A_ &iw);
271
272 for (i = 0; i < nfd; ++i)
273 {
274 ev_ref ();
275 ev_io_stop (EV_A_ iow + i);
276 }
277
278 process ();
279 298
280 update_now (EV_A); 299 update_now (EV_A);
281 300
282 nfd = mfd; 301 nfd = mfd;
283 302
288 free (iow); iow = malloc (mfd * sizeof (ev_io)); 307 free (iow); iow = malloc (mfd * sizeof (ev_io));
289 free (fds); fds = malloc (mfd * sizeof (struct pollfd)); 308 free (fds); fds = malloc (mfd * sizeof (struct pollfd));
290 } 309 }
291 310
292 ev_timer_set (&tw, timeout * 1e-3, 0.); 311 ev_timer_set (&tw, timeout * 1e-3, 0.);
293 ev_timer_start (EV_A_ &tw); 312 ev_timer_start (EV_A, &tw);
294 ev_unref ();
295 313
296 // create one ev_io per pollfd 314 // create one ev_io per pollfd
297 for (i = 0; i < nfd; ++i) 315 for (i = 0; i < nfd; ++i)
298 { 316 {
317 ev_io *w = iow + i;
318
299 ev_io_init (iow + i, io_cb, fds [i].fd, 319 ev_io_init (w, io_cb, fds [i].fd,
300 ((fds [i].events & POLLIN ? EV_READ : 0) 320 ((fds [i].events & POLLIN ? EV_READ : 0)
301 | (fds [i].events & POLLOUT ? EV_WRITE : 0))); 321 | (fds [i].events & POLLOUT ? EV_WRITE : 0)));
302 322
323 w->data = (void *)ads;
303 ev_io_start (EV_A_ iow + i); 324 ev_io_start (EV_A, w);
304 ev_unref ();
305 } 325 }
306} 326}
327
328static HV *stash;
329static adns_state ads;
307 330
308MODULE = EV::ADNS PACKAGE = EV::ADNS 331MODULE = EV::ADNS PACKAGE = EV::ADNS
309 332
310PROTOTYPES: ENABLE 333PROTOTYPES: ENABLE
311 334
399 for (civ = const_iv + sizeof (const_iv) / sizeof (const_iv [0]); civ-- > const_iv; ) 422 for (civ = const_iv + sizeof (const_iv) / sizeof (const_iv [0]); civ-- > const_iv; )
400 newCONSTSUB (stash, (char *)civ->name, newSViv (civ->iv)); 423 newCONSTSUB (stash, (char *)civ->name, newSViv (civ->iv));
401 424
402 I_EV_API ("EV::ADNS"); 425 I_EV_API ("EV::ADNS");
403 426
427 adns_init (&ads, adns_if_noenv | adns_if_noerrprint | adns_if_noserverwarn | adns_if_noautosys, 0);
428
404 ev_prepare_init (&prepare_ev, prepare_cb); 429 ev_prepare_init (&pw, prepare_cb);
405 ev_prepare_start (EV_DEFAULT_ &prepare_ev); 430 pw.data = (void *)ads;
406 ev_unref ();
407 431
408 ev_init (&iw, idle_cb); ev_set_priority (&iw, EV_MINPRI); 432 ev_init (&iw, idle_cb); ev_set_priority (&iw, EV_MINPRI);
433 iw.data = (void *)ads;
409 ev_init (&tw, timer_cb); 434 ev_init (&tw, timer_cb);
410 435 tw.data = (void *)ads;
411 adns_init (&ads, adns_if_noenv | adns_if_noerrprint | adns_if_noserverwarn | adns_if_noautosys, 0);
412} 436}
413 437
414void submit (char *owner, int type, int flags, SV *cb) 438void submit (char *owner, int type, int flags, SV *cb)
415 PPCODE: 439 PPCODE:
416{ 440{
417 SV *csv = NEWSV (0, sizeof (struct ctx)); 441 SV *csv = NEWSV (0, sizeof (struct ctx));
418 struct ctx *c = (struct ctx *)SvPVX (csv); 442 struct ctx *c = (struct ctx *)SvPVX (csv);
419 int r = adns_submit (ads, owner, type, flags, (void *)c, &c->query); 443 int r = adns_submit (ads, owner, type, flags, (void *)c, &c->query);
444
445 outstanding_inc (ads);
420 446
421 if (r) 447 if (r)
422 { 448 {
423 SvREFCNT_dec (csv); 449 SvREFCNT_dec (csv);
424 errno = r; 450 errno = r;
425 XSRETURN_EMPTY; 451 XSRETURN_EMPTY;
426 } 452 }
427 else 453 else
428 { 454 {
429 ev_ref ();
430 SvPOK_only (csv); 455 SvPOK_only (csv);
431 SvCUR_set (csv, sizeof (struct ctx)); 456 SvCUR_set (csv, sizeof (struct ctx));
432 457
433 c->self = csv; 458 c->self = csv;
434 c->cb = newSVsv (cb); 459 c->cb = newSVsv (cb);
460 c->ads = ads;
435 461
436 if (!ev_is_active (&iw)) 462 if (!ev_is_active (&iw))
437 ev_idle_start (EV_A_ &iw); 463 ev_idle_start (EV_DEFAULT, &iw);
438 464
439 if (GIMME_V != G_VOID) 465 if (GIMME_V != G_VOID)
440 { 466 {
441 csv = sv_2mortal (newRV_inc (csv)); 467 csv = sv_2mortal (newRV_inc (csv));
442 sv_bless (csv, stash); 468 sv_bless (csv, stash);
458 484
459 c = (struct ctx *)SvPVX (SvRV (req)); 485 c = (struct ctx *)SvPVX (SvRV (req));
460 486
461 if (c->cb) 487 if (c->cb)
462 { 488 {
463 ev_unref ();
464 SvREFCNT_dec (c->cb); 489 SvREFCNT_dec (c->cb);
465 c->cb = 0; 490 c->cb = 0; outstanding_dec (c->ads);
466 adns_cancel (c->query); 491 adns_cancel (c->query);
467 SvREFCNT_dec (c->self); 492 SvREFCNT_dec (c->self);
468 } 493 }
469} 494}
470 495

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines