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

Comparing IO-AIO/AIO.xs (file contents):
Revision 1.59 by root, Mon Oct 23 18:56:27 2006 UTC vs.
Revision 1.82 by root, Fri Oct 27 20:11:58 2006 UTC

1/* solaris */
2#define _POSIX_PTHREAD_SEMANTICS 1
3
4#if __linux && !defined(_GNU_SOURCE)
5# define _GNU_SOURCE
6#endif
7
8/* just in case */
1#define _REENTRANT 1 9#define _REENTRANT 1
10
2#include <errno.h> 11#include <errno.h>
3 12
4#include "EXTERN.h" 13#include "EXTERN.h"
5#include "perl.h" 14#include "perl.h"
6#include "XSUB.h" 15#include "XSUB.h"
39/* used for struct dirent, AIX doesn't provide it */ 48/* used for struct dirent, AIX doesn't provide it */
40#ifndef NAME_MAX 49#ifndef NAME_MAX
41# define NAME_MAX 4096 50# define NAME_MAX 4096
42#endif 51#endif
43 52
53#ifndef PTHREAD_STACK_MIN
54/* care for broken platforms, e.g. windows */
55# define PTHREAD_STACK_MIN 16384
56#endif
57
44#if __ia64 58#if __ia64
45# define STACKSIZE 65536 59# define STACKSIZE 65536
60#elif __i386 || __x86_64 /* 16k is unreasonably high :( */
61# define STACKSIZE PTHREAD_STACK_MIN
46#else 62#else
47# define STACKSIZE 8192 63# define STACKSIZE 16384
64#endif
65
66/* wether word reads are potentially non-atomic.
67 * this is conservatice, likely most arches this runs
68 * on have atomic word read/writes.
69 */
70#ifndef WORDREAD_UNSAFE
71# if __i386 || __x86_64
72# define WORDREAD_UNSAFE 0
73# else
74# define WORDREAD_UNSAFE 1
48#endif 75# endif
76#endif
77
78/* buffer size for various temporary buffers */
79#define AIO_BUFSIZE 65536
80
81#define dBUF \
82 char *aio_buf; \
83 LOCK (wrklock); \
84 self->dbuf = aio_buf = malloc (AIO_BUFSIZE); \
85 UNLOCK (wrklock); \
86 if (!aio_buf) \
87 return -1;
49 88
50enum { 89enum {
51 REQ_QUIT, 90 REQ_QUIT,
52 REQ_OPEN, REQ_CLOSE, 91 REQ_OPEN, REQ_CLOSE,
53 REQ_READ, REQ_WRITE, REQ_READAHEAD, 92 REQ_READ, REQ_WRITE, REQ_READAHEAD,
54 REQ_SENDFILE, 93 REQ_SENDFILE,
55 REQ_STAT, REQ_LSTAT, REQ_FSTAT, 94 REQ_STAT, REQ_LSTAT, REQ_FSTAT,
56 REQ_FSYNC, REQ_FDATASYNC, 95 REQ_FSYNC, REQ_FDATASYNC,
57 REQ_UNLINK, REQ_RMDIR, REQ_RENAME, 96 REQ_UNLINK, REQ_RMDIR, REQ_RENAME,
58 REQ_READDIR, 97 REQ_MKNOD, REQ_READDIR,
59 REQ_LINK, REQ_SYMLINK, 98 REQ_LINK, REQ_SYMLINK,
60 REQ_GROUP, REQ_NOP, 99 REQ_GROUP, REQ_NOP,
61 REQ_SLEEP, 100 REQ_BUSY,
62}; 101};
63 102
64#define AIO_REQ_KLASS "IO::AIO::REQ" 103#define AIO_REQ_KLASS "IO::AIO::REQ"
65#define AIO_GRP_KLASS "IO::AIO::GRP" 104#define AIO_GRP_KLASS "IO::AIO::GRP"
66 105
67typedef struct aio_cb 106typedef struct aio_cb
68{ 107{
69 struct aio_cb *volatile next; 108 struct aio_cb *volatile next;
70
71 struct aio_cb *grp, *grp_prev, *grp_next, *grp_first;
72
73 SV *self; /* the perl counterpart of this request, if any */
74 109
75 SV *data, *callback; 110 SV *data, *callback;
76 SV *fh, *fh2; 111 SV *fh, *fh2;
77 void *dataptr, *data2ptr; 112 void *dataptr, *data2ptr;
78 Stat_t *statdata; 113 Stat_t *statdata;
79 off_t offset; 114 off_t offset;
80 size_t length; 115 size_t length;
81 ssize_t result; 116 ssize_t result;
82 117
118 STRLEN dataoffset;
83 int type; 119 int type;
84 int fd, fd2; 120 int fd, fd2;
85 int errorno; 121 int errorno;
86 STRLEN dataoffset;
87 mode_t mode; /* open */ 122 mode_t mode; /* open */
123
124 unsigned char flags;
88 unsigned char pri; 125 unsigned char pri;
89 unsigned char flags; 126
127 SV *self; /* the perl counterpart of this request, if any */
128 struct aio_cb *grp, *grp_prev, *grp_next, *grp_first;
90} aio_cb; 129} aio_cb;
91 130
92enum { 131enum {
93 FLAG_CANCELLED = 0x01, 132 FLAG_CANCELLED = 0x01,
94}; 133};
95 134
96typedef aio_cb *aio_req; 135typedef aio_cb *aio_req;
97typedef aio_cb *aio_req_ornot; 136typedef aio_cb *aio_req_ornot;
98 137
138enum {
139 PRI_MIN = -4,
140 PRI_MAX = 4,
141
142 DEFAULT_PRI = 0,
143 PRI_BIAS = -PRI_MIN,
144 NUM_PRI = PRI_MAX + PRI_BIAS + 1,
145};
146
147static int next_pri = DEFAULT_PRI + PRI_BIAS;
148
99static int started, wanted; 149static unsigned int started, wanted;
100static volatile int nreqs; 150
101static int max_outstanding = 1<<30; 151#if __linux && defined (PTHREAD_ADAPTIVE_MUTEX_INITIALIZER_NP)
152# define AIO_MUTEX_INIT PTHREAD_ADAPTIVE_MUTEX_INITIALIZER_NP
153#else
154# define AIO_MUTEX_INIT PTHREAD_MUTEX_INITIALIZER
155#endif
156
157#define LOCK(mutex) pthread_mutex_lock (&(mutex))
158#define UNLOCK(mutex) pthread_mutex_unlock (&(mutex))
159
160/* worker threads management */
161static pthread_mutex_t wrklock = AIO_MUTEX_INIT;
162
163typedef struct worker {
164 /* locked by wrklock */
165 struct worker *prev, *next;
166
167 pthread_t tid;
168
169 /* locked by reslock, reqlock or wrklock */
170 aio_req req; /* currently processed request */
171 void *dbuf;
172 DIR *dirp;
173} worker;
174
175static worker wrk_first = { &wrk_first, &wrk_first, 0 };
176
177static void worker_clear (worker *wrk)
178{
179 if (wrk->dirp)
180 {
181 closedir (wrk->dirp);
182 wrk->dirp = 0;
183 }
184
185 if (wrk->dbuf)
186 {
187 free (wrk->dbuf);
188 wrk->dbuf = 0;
189 }
190}
191
192static void worker_free (worker *wrk)
193{
194 wrk->next->prev = wrk->prev;
195 wrk->prev->next = wrk->next;
196
197 free (wrk);
198}
199
200static volatile unsigned int nreqs, nready, npending;
201static volatile unsigned int max_outstanding = 0xffffffff;
102static int respipe [2]; 202static int respipe [2];
103 203
104static pthread_mutex_t reslock = PTHREAD_MUTEX_INITIALIZER; 204static pthread_mutex_t reslock = AIO_MUTEX_INIT;
105static pthread_mutex_t reqlock = PTHREAD_MUTEX_INITIALIZER; 205static pthread_mutex_t reqlock = AIO_MUTEX_INIT;
106static pthread_cond_t reqwait = PTHREAD_COND_INITIALIZER; 206static pthread_cond_t reqwait = PTHREAD_COND_INITIALIZER;
107 207
108static volatile aio_req reqs, reqe; /* queue start, queue end */ 208#if WORDREAD_UNSAFE
109static volatile aio_req ress, rese; /* queue start, queue end */
110 209
210static unsigned int get_nready ()
211{
212 unsigned int retval;
213
214 LOCK (reqlock);
215 retval = nready;
216 UNLOCK (reqlock);
217
218 return retval;
219}
220
221static unsigned int get_npending ()
222{
223 unsigned int retval;
224
225 LOCK (reslock);
226 retval = npending;
227 UNLOCK (reslock);
228
229 return retval;
230}
231
232#else
233
234# define get_nready() nready
235# define get_npending() npending
236
237#endif
238
239/*
240 * a somewhat faster data structure might be nice, but
241 * with 8 priorities this actually needs <20 insns
242 * per shift, the most expensive operation.
243 */
244typedef struct {
245 aio_req qs[NUM_PRI], qe[NUM_PRI]; /* qstart, qend */
246 int size;
247} reqq;
248
249static reqq req_queue;
250static reqq res_queue;
251
252int reqq_push (reqq *q, aio_req req)
253{
254 int pri = req->pri;
255 req->next = 0;
256
257 if (q->qe[pri])
258 {
259 q->qe[pri]->next = req;
260 q->qe[pri] = req;
261 }
262 else
263 q->qe[pri] = q->qs[pri] = req;
264
265 return q->size++;
266}
267
268aio_req reqq_shift (reqq *q)
269{
270 int pri;
271
272 if (!q->size)
273 return 0;
274
275 --q->size;
276
277 for (pri = NUM_PRI; pri--; )
278 {
279 aio_req req = q->qs[pri];
280
281 if (req)
282 {
283 if (!(q->qs[pri] = req->next))
284 q->qe[pri] = 0;
285
286 return req;
287 }
288 }
289
290 abort ();
291}
292
293static int poll_cb (int max);
111static void req_invoke (aio_req req); 294static void req_invoke (aio_req req);
112static void req_free (aio_req req); 295static void req_free (aio_req req);
296static void req_cancel (aio_req req);
113 297
114/* must be called at most once */ 298/* must be called at most once */
115static SV *req_sv (aio_req req, const char *klass) 299static SV *req_sv (aio_req req, const char *klass)
116{ 300{
117 if (!req->self) 301 if (!req->self)
148 ENTER; 332 ENTER;
149 SAVETMPS; 333 SAVETMPS;
150 PUSHMARK (SP); 334 PUSHMARK (SP);
151 XPUSHs (req_sv (grp, AIO_GRP_KLASS)); 335 XPUSHs (req_sv (grp, AIO_GRP_KLASS));
152 PUTBACK; 336 PUTBACK;
153 call_sv (grp->fh2, G_VOID | G_EVAL); 337 call_sv (grp->fh2, G_VOID | G_EVAL | G_KEEPERR);
154 SPAGAIN; 338 SPAGAIN;
155 FREETMPS; 339 FREETMPS;
156 LEAVE; 340 LEAVE;
157 } 341 }
158 342
179 req_invoke (grp); 363 req_invoke (grp);
180 req_free (grp); 364 req_free (grp);
181 } 365 }
182} 366}
183 367
184static void poll_wait ()
185{
186 if (nreqs && !ress)
187 {
188 fd_set rfd;
189 FD_ZERO(&rfd);
190 FD_SET(respipe [0], &rfd);
191
192 select (respipe [0] + 1, &rfd, 0, 0, 0);
193 }
194}
195
196static void req_invoke (aio_req req) 368static void req_invoke (aio_req req)
197{ 369{
198 dSP; 370 dSP;
199 int errorno = errno;
200 371
201 if (req->flags & FLAG_CANCELLED || !SvOK (req->callback)) 372 if (!(req->flags & FLAG_CANCELLED) && SvOK (req->callback))
202 return; 373 {
203
204 errno = req->errorno;
205
206 ENTER; 374 ENTER;
207 SAVETMPS; 375 SAVETMPS;
208 PUSHMARK (SP); 376 PUSHMARK (SP);
209 EXTEND (SP, 1); 377 EXTEND (SP, 1);
210 378
211 switch (req->type) 379 switch (req->type)
212 {
213 case REQ_READDIR:
214 { 380 {
215 SV *rv = &PL_sv_undef; 381 case REQ_READDIR:
216
217 if (req->result >= 0)
218 { 382 {
219 char *buf = req->data2ptr; 383 SV *rv = &PL_sv_undef;
220 AV *av = newAV ();
221 384
222 while (req->result) 385 if (req->result >= 0)
223 { 386 {
387 int i;
388 char *buf = req->data2ptr;
389 AV *av = newAV ();
390
391 av_extend (av, req->result - 1);
392
393 for (i = 0; i < req->result; ++i)
394 {
224 SV *sv = newSVpv (buf, 0); 395 SV *sv = newSVpv (buf, 0);
225 396
226 av_push (av, sv); 397 av_store (av, i, sv);
227 buf += SvCUR (sv) + 1; 398 buf += SvCUR (sv) + 1;
228 req->result--; 399 }
400
401 rv = sv_2mortal (newRV_noinc ((SV *)av));
229 } 402 }
230 403
231 rv = sv_2mortal (newRV_noinc ((SV *)av)); 404 PUSHs (rv);
232 } 405 }
406 break;
233 407
234 PUSHs (rv); 408 case REQ_OPEN:
409 {
410 /* convert fd to fh */
411 SV *fh;
412
413 PUSHs (sv_2mortal (newSViv (req->result)));
414 PUTBACK;
415 call_pv ("IO::AIO::_fd2fh", G_SCALAR | G_EVAL);
416 SPAGAIN;
417
418 fh = SvREFCNT_inc (POPs);
419
420 PUSHMARK (SP);
421 XPUSHs (sv_2mortal (fh));
422 }
423 break;
424
425 case REQ_GROUP:
426 req->fd = 2; /* mark group as finished */
427
428 if (req->data)
429 {
430 int i;
431 AV *av = (AV *)req->data;
432
433 EXTEND (SP, AvFILL (av) + 1);
434 for (i = 0; i <= AvFILL (av); ++i)
435 PUSHs (*av_fetch (av, i, 0));
436 }
437 break;
438
439 case REQ_NOP:
440 case REQ_BUSY:
441 break;
442
443 default:
444 PUSHs (sv_2mortal (newSViv (req->result)));
445 break;
235 } 446 }
236 break;
237 447
238 case REQ_OPEN: 448 errno = req->errorno;
239 {
240 /* convert fd to fh */
241 SV *fh;
242 449
243 PUSHs (sv_2mortal (newSViv (req->result)));
244 PUTBACK; 450 PUTBACK;
245 call_pv ("IO::AIO::_fd2fh", G_SCALAR | G_EVAL);
246 SPAGAIN;
247
248 fh = SvREFCNT_inc (POPs);
249
250 PUSHMARK (SP);
251 XPUSHs (sv_2mortal (fh));
252 }
253 break;
254
255 case REQ_GROUP:
256 req->fd = 2; /* mark group as finished */
257
258 if (req->data)
259 {
260 int i;
261 AV *av = (AV *)req->data;
262
263 EXTEND (SP, AvFILL (av) + 1);
264 for (i = 0; i <= AvFILL (av); ++i)
265 PUSHs (*av_fetch (av, i, 0));
266 }
267 break;
268
269 case REQ_NOP:
270 case REQ_SLEEP:
271 break;
272
273 default:
274 PUSHs (sv_2mortal (newSViv (req->result)));
275 break;
276 }
277
278
279 PUTBACK;
280 call_sv (req->callback, G_VOID | G_EVAL); 451 call_sv (req->callback, G_VOID | G_EVAL);
281 SPAGAIN; 452 SPAGAIN;
282 453
283 FREETMPS; 454 FREETMPS;
284 LEAVE; 455 LEAVE;
285
286 errno = errorno;
287
288 if (SvTRUE (ERRSV))
289 { 456 }
290 req_free (req);
291 croak (0);
292 }
293}
294 457
295static void req_free (aio_req req)
296{
297 if (req->grp) 458 if (req->grp)
298 { 459 {
299 aio_req grp = req->grp; 460 aio_req grp = req->grp;
300 461
301 /* unlink request */ 462 /* unlink request */
306 grp->grp_first = req->grp_next; 467 grp->grp_first = req->grp_next;
307 468
308 aio_grp_dec (grp); 469 aio_grp_dec (grp);
309 } 470 }
310 471
472 if (SvTRUE (ERRSV))
473 {
474 req_free (req);
475 croak (0);
476 }
477}
478
479static void req_free (aio_req req)
480{
311 if (req->self) 481 if (req->self)
312 { 482 {
313 sv_unmagic (req->self, PERL_MAGIC_ext); 483 sv_unmagic (req->self, PERL_MAGIC_ext);
314 SvREFCNT_dec (req->self); 484 SvREFCNT_dec (req->self);
315 } 485 }
318 SvREFCNT_dec (req->fh); 488 SvREFCNT_dec (req->fh);
319 SvREFCNT_dec (req->fh2); 489 SvREFCNT_dec (req->fh2);
320 SvREFCNT_dec (req->callback); 490 SvREFCNT_dec (req->callback);
321 Safefree (req->statdata); 491 Safefree (req->statdata);
322 492
323 if (req->type == REQ_READDIR && req->result >= 0) 493 if (req->type == REQ_READDIR)
324 free (req->data2ptr); 494 free (req->data2ptr);
325 495
326 Safefree (req); 496 Safefree (req);
327} 497}
328 498
499static void req_cancel_subs (aio_req grp)
500{
501 aio_req sub;
502
503 if (grp->type != REQ_GROUP)
504 return;
505
506 SvREFCNT_dec (grp->fh2);
507 grp->fh2 = 0;
508
509 for (sub = grp->grp_first; sub; sub = sub->grp_next)
510 req_cancel (sub);
511}
512
329static void req_cancel (aio_req req) 513static void req_cancel (aio_req req)
330{ 514{
331 req->flags |= FLAG_CANCELLED; 515 req->flags |= FLAG_CANCELLED;
332 516
333 if (req->type == REQ_GROUP) 517 req_cancel_subs (req);
334 { 518}
335 aio_req sub;
336 519
337 for (sub = req->grp_first; sub; sub = sub->grp_next) 520static void *aio_proc(void *arg);
338 req_cancel (sub); 521
522static void start_thread (void)
523{
524 sigset_t fullsigset, oldsigset;
525 pthread_attr_t attr;
526
527 worker *wrk = calloc (1, sizeof (worker));
528
529 if (!wrk)
530 croak ("unable to allocate worker thread data");
531
532 pthread_attr_init (&attr);
533 pthread_attr_setstacksize (&attr, STACKSIZE);
534 pthread_attr_setdetachstate (&attr, PTHREAD_CREATE_DETACHED);
535#ifdef PTHREAD_SCOPE_PROCESS
536 pthread_attr_setscope (&attr, PTHREAD_SCOPE_PROCESS);
537#endif
538
539 sigfillset (&fullsigset);
540
541 LOCK (wrklock);
542 sigprocmask (SIG_SETMASK, &fullsigset, &oldsigset);
543
544 if (pthread_create (&wrk->tid, &attr, aio_proc, (void *)wrk) == 0)
339 } 545 {
340} 546 wrk->prev = &wrk_first;
547 wrk->next = wrk_first.next;
548 wrk_first.next->prev = wrk;
549 wrk_first.next = wrk;
550 ++started;
551 }
552 else
553 free (wrk);
341 554
555 sigprocmask (SIG_SETMASK, &oldsigset, 0);
556 UNLOCK (wrklock);
557}
558
559static void maybe_start_thread ()
560{
561#if 0
562 static struct timeval last;
563 struct timeval diff, now;
564#endif
565
566 if (started >= wanted)
567 return;
568
569 if (nready <= nreqs - get_nready () - get_npending ())
570 return;
571
572#if 0
573 gettimeofday (&now, 0);
574
575 diff.tv_sec = now.tv_sec - last.tv_sec;
576 diff.tv_usec = now.tv_usec - last.tv_usec;
577
578 if (diff.tv_usec < 0)
579 {
580 --diff.tv_sec;
581 diff.tv_usec += 1000000;
582 }
583
584 if (!diff.tv_sec && diff.tv_usec < 10000)
585 return;
586
587 last = now;
588#endif
589
590 start_thread ();
591}
592
593static void req_send (aio_req req)
594{
595 ++nreqs;
596
597 LOCK (reqlock);
598 ++nready;
599 reqq_push (&req_queue, req);
600 pthread_cond_signal (&reqwait);
601 UNLOCK (reqlock);
602
603 maybe_start_thread ();
604}
605
606static void end_thread (void)
607{
608 aio_req req;
609
610 Newz (0, req, 1, aio_cb);
611
612 req->type = REQ_QUIT;
613 req->pri = PRI_MAX + PRI_BIAS;
614
615 req_send (req);
616
617 LOCK (wrklock);
618 --started;
619 UNLOCK (wrklock);
620}
621
622static void min_parallel (int nthreads)
623{
624 if (wanted < nthreads)
625 wanted = nthreads;
626}
627
628static void max_parallel (int nthreads)
629{
630 if (wanted > nthreads)
631 wanted = nthreads;
632
633 while (started > wanted)
634 end_thread ();
635}
636
637static void poll_wait ()
638{
639 fd_set rfd;
640
641 while (nreqs)
642 {
643 int size;
644 if (WORDREAD_UNSAFE) LOCK (reslock);
645 size = res_queue.size;
646 if (WORDREAD_UNSAFE) UNLOCK (reslock);
647
648 if (size)
649 return;
650
651 maybe_start_thread ();
652
653 FD_ZERO(&rfd);
654 FD_SET(respipe [0], &rfd);
655
656 select (respipe [0] + 1, &rfd, 0, 0, 0);
657 }
658}
659
342static int poll_cb () 660static int poll_cb (int max)
343{ 661{
344 dSP; 662 dSP;
345 int count = 0; 663 int count = 0;
346 int do_croak = 0; 664 int do_croak = 0;
347 aio_req req; 665 aio_req req;
348 666
349 for (;;) 667 for (;;)
350 { 668 {
351 pthread_mutex_lock (&reslock); 669 while (max <= 0 || count < max)
352 req = ress;
353
354 if (req)
355 { 670 {
356 ress = req->next; 671 maybe_start_thread ();
357 672
673 LOCK (reslock);
674 req = reqq_shift (&res_queue);
675
358 if (!ress) 676 if (req)
359 { 677 {
678 --npending;
679
680 if (!res_queue.size)
681 {
360 /* read any signals sent by the worker threads */ 682 /* read any signals sent by the worker threads */
361 char buf [32]; 683 char buf [32];
362 while (read (respipe [0], buf, 32) == 32) 684 while (read (respipe [0], buf, 32) == 32)
685 ;
363 ; 686 }
364
365 rese = 0;
366 } 687 }
688
689 UNLOCK (reslock);
690
691 if (!req)
692 break;
693
694 --nreqs;
695
696 if (req->type == REQ_GROUP && req->length)
697 {
698 req->fd = 1; /* mark request as delayed */
699 continue;
700 }
701 else
702 {
703 if (req->type == REQ_READ)
704 SvCUR_set (req->data, req->dataoffset + (req->result > 0 ? req->result : 0));
705
706 if (req->data2ptr && (req->type == REQ_READ || req->type == REQ_WRITE))
707 SvREADONLY_off (req->data);
708
709 if (req->statdata)
710 {
711 PL_laststype = req->type == REQ_LSTAT ? OP_LSTAT : OP_STAT;
712 PL_laststatval = req->result;
713 PL_statcache = *(req->statdata);
714 }
715
716 req_invoke (req);
717
718 count++;
719 }
720
721 req_free (req);
367 } 722 }
368 723
369 pthread_mutex_unlock (&reslock); 724 if (nreqs <= max_outstanding)
370
371 if (!req)
372 break; 725 break;
373 726
374 --nreqs; 727 poll_wait ();
375 728
376 if (req->type == REQ_QUIT) 729 max = 0;
377 started--;
378 else if (req->type == REQ_GROUP && req->length)
379 {
380 req->fd = 1; /* mark request as delayed */
381 continue;
382 }
383 else
384 {
385 if (req->type == REQ_READ)
386 SvCUR_set (req->data, req->dataoffset + (req->result > 0 ? req->result : 0));
387
388 if (req->data2ptr && (req->type == REQ_READ || req->type == REQ_WRITE))
389 SvREADONLY_off (req->data);
390
391 if (req->statdata)
392 {
393 PL_laststype = req->type == REQ_LSTAT ? OP_LSTAT : OP_STAT;
394 PL_laststatval = req->result;
395 PL_statcache = *(req->statdata);
396 }
397
398 req_invoke (req);
399
400 count++;
401 }
402
403 req_free (req);
404 } 730 }
405 731
406 return count; 732 return count;
407}
408
409static void *aio_proc(void *arg);
410
411static void start_thread (void)
412{
413 sigset_t fullsigset, oldsigset;
414 pthread_t tid;
415 pthread_attr_t attr;
416
417 pthread_attr_init (&attr);
418 pthread_attr_setstacksize (&attr, STACKSIZE);
419 pthread_attr_setdetachstate (&attr, PTHREAD_CREATE_DETACHED);
420
421 sigfillset (&fullsigset);
422 sigprocmask (SIG_SETMASK, &fullsigset, &oldsigset);
423
424 if (pthread_create (&tid, &attr, aio_proc, 0) == 0)
425 started++;
426
427 sigprocmask (SIG_SETMASK, &oldsigset, 0);
428}
429
430static void req_send (aio_req req)
431{
432 while (started < wanted && nreqs >= started)
433 start_thread ();
434
435 ++nreqs;
436
437 pthread_mutex_lock (&reqlock);
438
439 req->next = 0;
440
441 if (reqe)
442 {
443 reqe->next = req;
444 reqe = req;
445 }
446 else
447 reqe = reqs = req;
448
449 pthread_cond_signal (&reqwait);
450 pthread_mutex_unlock (&reqlock);
451
452 if (nreqs > max_outstanding)
453 for (;;)
454 {
455 poll_cb ();
456
457 if (nreqs <= max_outstanding)
458 break;
459
460 poll_wait ();
461 }
462}
463
464static void end_thread (void)
465{
466 aio_req req;
467 Newz (0, req, 1, aio_cb);
468 req->type = REQ_QUIT;
469
470 req_send (req);
471}
472
473static void min_parallel (int nthreads)
474{
475 if (wanted < nthreads)
476 wanted = nthreads;
477}
478
479static void max_parallel (int nthreads)
480{
481 int cur = started;
482
483 if (wanted > nthreads)
484 wanted = nthreads;
485
486 while (cur > wanted)
487 {
488 end_thread ();
489 cur--;
490 }
491
492 while (started > wanted)
493 {
494 poll_wait ();
495 poll_cb ();
496 }
497} 733}
498 734
499static void create_pipe () 735static void create_pipe ()
500{ 736{
501 if (pipe (respipe)) 737 if (pipe (respipe))
525static ssize_t pread (int fd, void *buf, size_t count, off_t offset) 761static ssize_t pread (int fd, void *buf, size_t count, off_t offset)
526{ 762{
527 ssize_t res; 763 ssize_t res;
528 off_t ooffset; 764 off_t ooffset;
529 765
530 pthread_mutex_lock (&preadwritelock); 766 LOCK (preadwritelock);
531 ooffset = lseek (fd, 0, SEEK_CUR); 767 ooffset = lseek (fd, 0, SEEK_CUR);
532 lseek (fd, offset, SEEK_SET); 768 lseek (fd, offset, SEEK_SET);
533 res = read (fd, buf, count); 769 res = read (fd, buf, count);
534 lseek (fd, ooffset, SEEK_SET); 770 lseek (fd, ooffset, SEEK_SET);
535 pthread_mutex_unlock (&preadwritelock); 771 UNLOCK (preadwritelock);
536 772
537 return res; 773 return res;
538} 774}
539 775
540static ssize_t pwrite (int fd, void *buf, size_t count, off_t offset) 776static ssize_t pwrite (int fd, void *buf, size_t count, off_t offset)
541{ 777{
542 ssize_t res; 778 ssize_t res;
543 off_t ooffset; 779 off_t ooffset;
544 780
545 pthread_mutex_lock (&preadwritelock); 781 LOCK (preadwritelock);
546 ooffset = lseek (fd, 0, SEEK_CUR); 782 ooffset = lseek (fd, 0, SEEK_CUR);
547 lseek (fd, offset, SEEK_SET); 783 lseek (fd, offset, SEEK_SET);
548 res = write (fd, buf, count); 784 res = write (fd, buf, count);
549 lseek (fd, offset, SEEK_SET); 785 lseek (fd, offset, SEEK_SET);
550 pthread_mutex_unlock (&preadwritelock); 786 UNLOCK (preadwritelock);
551 787
552 return res; 788 return res;
553} 789}
554#endif 790#endif
555 791
556#if !HAVE_FDATASYNC 792#if !HAVE_FDATASYNC
557# define fdatasync fsync 793# define fdatasync fsync
558#endif 794#endif
559 795
560#if !HAVE_READAHEAD 796#if !HAVE_READAHEAD
561# define readahead aio_readahead 797# define readahead(fd,offset,count) aio_readahead (fd, offset, count, self)
562 798
563static ssize_t readahead (int fd, off_t offset, size_t count) 799static ssize_t aio_readahead (int fd, off_t offset, size_t count, worker *self)
564{ 800{
565 char readahead_buf[4096]; 801 dBUF;
566 802
567 while (count > 0) 803 while (count > 0)
568 { 804 {
569 size_t len = count < sizeof (readahead_buf) ? count : sizeof (readahead_buf); 805 size_t len = count < AIO_BUFSIZE ? count : AIO_BUFSIZE;
570 806
571 pread (fd, readahead_buf, len, offset); 807 pread (fd, aio_buf, len, offset);
572 offset += len; 808 offset += len;
573 count -= len; 809 count -= len;
574 } 810 }
575 811
576 errno = 0; 812 errno = 0;
577} 813}
814
578#endif 815#endif
579 816
580#if !HAVE_READDIR_R 817#if !HAVE_READDIR_R
581# define readdir_r aio_readdir_r 818# define readdir_r aio_readdir_r
582 819
585static int readdir_r (DIR *dirp, struct dirent *ent, struct dirent **res) 822static int readdir_r (DIR *dirp, struct dirent *ent, struct dirent **res)
586{ 823{
587 struct dirent *e; 824 struct dirent *e;
588 int errorno; 825 int errorno;
589 826
590 pthread_mutex_lock (&readdirlock); 827 LOCK (readdirlock);
591 828
592 e = readdir (dirp); 829 e = readdir (dirp);
593 errorno = errno; 830 errorno = errno;
594 831
595 if (e) 832 if (e)
598 strcpy (ent->d_name, e->d_name); 835 strcpy (ent->d_name, e->d_name);
599 } 836 }
600 else 837 else
601 *res = 0; 838 *res = 0;
602 839
603 pthread_mutex_unlock (&readdirlock); 840 UNLOCK (readdirlock);
604 841
605 errno = errorno; 842 errno = errorno;
606 return e ? 0 : -1; 843 return e ? 0 : -1;
607} 844}
608#endif 845#endif
609 846
610/* sendfile always needs emulation */ 847/* sendfile always needs emulation */
611static ssize_t sendfile_ (int ofd, int ifd, off_t offset, size_t count) 848static ssize_t sendfile_ (int ofd, int ifd, off_t offset, size_t count, worker *self)
612{ 849{
613 ssize_t res; 850 ssize_t res;
614 851
615 if (!count) 852 if (!count)
616 return 0; 853 return 0;
627 { 864 {
628 off_t sbytes; 865 off_t sbytes;
629 res = sendfile (ifd, ofd, offset, count, 0, &sbytes, 0); 866 res = sendfile (ifd, ofd, offset, count, 0, &sbytes, 0);
630 867
631 if (res < 0 && sbytes) 868 if (res < 0 && sbytes)
632 /* maybe only on EAGAIN only: as usual, the manpage leaves you guessing */ 869 /* maybe only on EAGAIN: as usual, the manpage leaves you guessing */
633 res = sbytes; 870 res = sbytes;
634 } 871 }
635 872
636# elif __hpux 873# elif __hpux
637 res = sendfile (ofd, ifd, offset, count, 0, 0); 874 res = sendfile (ofd, ifd, offset, count, 0, 0);
665#endif 902#endif
666 ) 903 )
667 ) 904 )
668 { 905 {
669 /* emulate sendfile. this is a major pain in the ass */ 906 /* emulate sendfile. this is a major pain in the ass */
670 char buf[4096]; 907 dBUF;
908
671 res = 0; 909 res = 0;
672 910
673 while (count) 911 while (count)
674 { 912 {
675 ssize_t cnt; 913 ssize_t cnt;
676 914
677 cnt = pread (ifd, buf, count > 4096 ? 4096 : count, offset); 915 cnt = pread (ifd, aio_buf, count > AIO_BUFSIZE ? AIO_BUFSIZE : count, offset);
678 916
679 if (cnt <= 0) 917 if (cnt <= 0)
680 { 918 {
681 if (cnt && !res) res = -1; 919 if (cnt && !res) res = -1;
682 break; 920 break;
683 } 921 }
684 922
685 cnt = write (ofd, buf, cnt); 923 cnt = write (ofd, aio_buf, cnt);
686 924
687 if (cnt <= 0) 925 if (cnt <= 0)
688 { 926 {
689 if (cnt && !res) res = -1; 927 if (cnt && !res) res = -1;
690 break; 928 break;
698 936
699 return res; 937 return res;
700} 938}
701 939
702/* read a full directory */ 940/* read a full directory */
703static int scandir_ (const char *path, void **namesp) 941static void scandir_ (aio_req req, worker *self)
704{ 942{
705 DIR *dirp = opendir (path); 943 DIR *dirp;
706 union 944 union
707 { 945 {
708 struct dirent d; 946 struct dirent d;
709 char b [offsetof (struct dirent, d_name) + NAME_MAX + 1]; 947 char b [offsetof (struct dirent, d_name) + NAME_MAX + 1];
710 } u; 948 } *u;
711 struct dirent *entp; 949 struct dirent *entp;
712 char *name, *names; 950 char *name, *names;
713 int memlen = 4096; 951 int memlen = 4096;
714 int memofs = 0; 952 int memofs = 0;
715 int res = 0; 953 int res = 0;
716 int errorno; 954 int errorno;
717 955
718 if (!dirp) 956 LOCK (wrklock);
719 return -1; 957 self->dirp = dirp = opendir (req->dataptr);
720 958 self->dbuf = u = malloc (sizeof (*u));
721 names = malloc (memlen); 959 req->data2ptr = names = malloc (memlen);
960 UNLOCK (wrklock);
961
962 if (dirp && u && names)
963 for (;;)
964 {
965 errno = 0;
966 readdir_r (dirp, &u->d, &entp);
967
968 if (!entp)
969 break;
970
971 name = entp->d_name;
972
973 if (name [0] != '.' || (name [1] && (name [1] != '.' || name [2])))
974 {
975 int len = strlen (name) + 1;
976
977 res++;
978
979 while (memofs + len > memlen)
980 {
981 memlen *= 2;
982 LOCK (wrklock);
983 req->data2ptr = names = realloc (names, memlen);
984 UNLOCK (wrklock);
985
986 if (!names)
987 break;
988 }
989
990 memcpy (names + memofs, name, len);
991 memofs += len;
992 }
993 }
994
995 if (errno)
996 res = -1;
997
998 req->result = res;
999}
1000
1001/*****************************************************************************/
1002
1003static void *aio_proc (void *thr_arg)
1004{
1005 aio_req req;
1006 worker *self = (worker *)thr_arg;
722 1007
723 for (;;) 1008 for (;;)
724 { 1009 {
725 errno = 0, readdir_r (dirp, &u.d, &entp); 1010 LOCK (reqlock);
726
727 if (!entp)
728 break;
729
730 name = entp->d_name;
731
732 if (name [0] != '.' || (name [1] && (name [1] != '.' || name [2])))
733 {
734 int len = strlen (name) + 1;
735
736 res++;
737
738 while (memofs + len > memlen)
739 {
740 memlen *= 2;
741 names = realloc (names, memlen);
742 if (!names)
743 break;
744 }
745
746 memcpy (names + memofs, name, len);
747 memofs += len;
748 }
749 }
750
751 errorno = errno;
752 closedir (dirp);
753
754 if (errorno)
755 {
756 free (names);
757 errno = errorno;
758 res = -1;
759 }
760
761 *namesp = (void *)names;
762 return res;
763}
764
765/*****************************************************************************/
766
767static void *aio_proc (void *thr_arg)
768{
769 aio_req req;
770 int type;
771
772 do
773 {
774 pthread_mutex_lock (&reqlock);
775 1011
776 for (;;) 1012 for (;;)
777 { 1013 {
778 req = reqs; 1014 self->req = req = reqq_shift (&req_queue);
779
780 if (reqs)
781 {
782 reqs = reqs->next;
783 if (!reqs) reqe = 0;
784 }
785 1015
786 if (req) 1016 if (req)
787 break; 1017 break;
788 1018
789 pthread_cond_wait (&reqwait, &reqlock); 1019 pthread_cond_wait (&reqwait, &reqlock);
790 } 1020 }
791 1021
792 pthread_mutex_unlock (&reqlock); 1022 --nready;
1023
1024 UNLOCK (reqlock);
793 1025
794 errno = 0; /* strictly unnecessary */ 1026 errno = 0; /* strictly unnecessary */
795 1027
796 if (!(req->flags & FLAG_CANCELLED)) 1028 if (!(req->flags & FLAG_CANCELLED))
797 switch (type = req->type) /* remember type for QUIT check */ 1029 switch (req->type)
798 { 1030 {
799 case REQ_READ: req->result = pread (req->fd, req->dataptr, req->length, req->offset); break; 1031 case REQ_READ: req->result = pread (req->fd, req->dataptr, req->length, req->offset); break;
800 case REQ_WRITE: req->result = pwrite (req->fd, req->dataptr, req->length, req->offset); break; 1032 case REQ_WRITE: req->result = pwrite (req->fd, req->dataptr, req->length, req->offset); break;
801 1033
802 case REQ_READAHEAD: req->result = readahead (req->fd, req->offset, req->length); break; 1034 case REQ_READAHEAD: req->result = readahead (req->fd, req->offset, req->length); break;
803 case REQ_SENDFILE: req->result = sendfile_ (req->fd, req->fd2, req->offset, req->length); break; 1035 case REQ_SENDFILE: req->result = sendfile_ (req->fd, req->fd2, req->offset, req->length, self); break;
804 1036
805 case REQ_STAT: req->result = stat (req->dataptr, req->statdata); break; 1037 case REQ_STAT: req->result = stat (req->dataptr, req->statdata); break;
806 case REQ_LSTAT: req->result = lstat (req->dataptr, req->statdata); break; 1038 case REQ_LSTAT: req->result = lstat (req->dataptr, req->statdata); break;
807 case REQ_FSTAT: req->result = fstat (req->fd , req->statdata); break; 1039 case REQ_FSTAT: req->result = fstat (req->fd , req->statdata); break;
808 1040
811 case REQ_UNLINK: req->result = unlink (req->dataptr); break; 1043 case REQ_UNLINK: req->result = unlink (req->dataptr); break;
812 case REQ_RMDIR: req->result = rmdir (req->dataptr); break; 1044 case REQ_RMDIR: req->result = rmdir (req->dataptr); break;
813 case REQ_RENAME: req->result = rename (req->data2ptr, req->dataptr); break; 1045 case REQ_RENAME: req->result = rename (req->data2ptr, req->dataptr); break;
814 case REQ_LINK: req->result = link (req->data2ptr, req->dataptr); break; 1046 case REQ_LINK: req->result = link (req->data2ptr, req->dataptr); break;
815 case REQ_SYMLINK: req->result = symlink (req->data2ptr, req->dataptr); break; 1047 case REQ_SYMLINK: req->result = symlink (req->data2ptr, req->dataptr); break;
1048 case REQ_MKNOD: req->result = mknod (req->data2ptr, req->mode, (dev_t)req->offset); break;
816 1049
817 case REQ_FDATASYNC: req->result = fdatasync (req->fd); break; 1050 case REQ_FDATASYNC: req->result = fdatasync (req->fd); break;
818 case REQ_FSYNC: req->result = fsync (req->fd); break; 1051 case REQ_FSYNC: req->result = fsync (req->fd); break;
819 case REQ_READDIR: req->result = scandir_ (req->dataptr, &req->data2ptr); break; 1052 case REQ_READDIR: scandir_ (req, self); break;
820 1053
821 case REQ_SLEEP: 1054 case REQ_BUSY:
822 { 1055 {
823 struct timeval tv; 1056 struct timeval tv;
824 1057
825 tv.tv_sec = req->fd; 1058 tv.tv_sec = req->fd;
826 tv.tv_usec = req->fd2; 1059 tv.tv_usec = req->fd2;
828 req->result = select (0, 0, 0, 0, &tv); 1061 req->result = select (0, 0, 0, 0, &tv);
829 } 1062 }
830 1063
831 case REQ_GROUP: 1064 case REQ_GROUP:
832 case REQ_NOP: 1065 case REQ_NOP:
1066 break;
1067
833 case REQ_QUIT: 1068 case REQ_QUIT:
1069 LOCK (wrklock);
1070 worker_free (self);
1071 --started;
1072 UNLOCK (wrklock);
834 break; 1073 return 0;
835 1074
836 default: 1075 default:
837 req->result = ENOSYS; 1076 req->result = ENOSYS;
838 break; 1077 break;
839 } 1078 }
840 1079
841 req->errorno = errno; 1080 req->errorno = errno;
842 1081
843 pthread_mutex_lock (&reslock); 1082 LOCK (reslock);
844 1083
845 req->next = 0; 1084 ++npending;
846 1085
847 if (rese) 1086 if (!reqq_push (&res_queue, req))
848 {
849 rese->next = req;
850 rese = req;
851 }
852 else
853 {
854 rese = ress = req;
855
856 /* write a dummy byte to the pipe so fh becomes ready */ 1087 /* write a dummy byte to the pipe so fh becomes ready */
857 write (respipe [1], &respipe, 1); 1088 write (respipe [1], &respipe, 1);
858 }
859 1089
860 pthread_mutex_unlock (&reslock); 1090 self->req = 0;
861 } 1091 worker_clear (self);
862 while (type != REQ_QUIT);
863 1092
864 return 0; 1093 UNLOCK (reslock);
1094 }
865} 1095}
866 1096
867/*****************************************************************************/ 1097/*****************************************************************************/
868 1098
869static void atfork_prepare (void) 1099static void atfork_prepare (void)
870{ 1100{
871 pthread_mutex_lock (&reqlock); 1101 LOCK (wrklock);
872 pthread_mutex_lock (&reslock); 1102 LOCK (reqlock);
1103 LOCK (reslock);
873#if !HAVE_PREADWRITE 1104#if !HAVE_PREADWRITE
874 pthread_mutex_lock (&preadwritelock); 1105 LOCK (preadwritelock);
875#endif 1106#endif
876#if !HAVE_READDIR_R 1107#if !HAVE_READDIR_R
877 pthread_mutex_lock (&readdirlock); 1108 LOCK (readdirlock);
878#endif 1109#endif
879} 1110}
880 1111
881static void atfork_parent (void) 1112static void atfork_parent (void)
882{ 1113{
883#if !HAVE_READDIR_R 1114#if !HAVE_READDIR_R
884 pthread_mutex_unlock (&readdirlock); 1115 UNLOCK (readdirlock);
885#endif 1116#endif
886#if !HAVE_PREADWRITE 1117#if !HAVE_PREADWRITE
887 pthread_mutex_unlock (&preadwritelock); 1118 UNLOCK (preadwritelock);
888#endif 1119#endif
889 pthread_mutex_unlock (&reslock); 1120 UNLOCK (reslock);
890 pthread_mutex_unlock (&reqlock); 1121 UNLOCK (reqlock);
1122 UNLOCK (wrklock);
891} 1123}
892 1124
893static void atfork_child (void) 1125static void atfork_child (void)
894{ 1126{
895 aio_req prv; 1127 aio_req prv;
896 1128
1129 while (prv = reqq_shift (&req_queue))
1130 req_free (prv);
1131
1132 while (prv = reqq_shift (&res_queue))
1133 req_free (prv);
1134
1135 while (wrk_first.next != &wrk_first)
1136 {
1137 worker *wrk = wrk_first.next;
1138
1139 if (wrk->req)
1140 req_free (wrk->req);
1141
1142 worker_clear (wrk);
1143 worker_free (wrk);
1144 }
1145
897 started = 0; 1146 started = 0;
898 1147 nreqs = 0;
899 while (reqs)
900 {
901 prv = reqs;
902 reqs = prv->next;
903 req_free (prv);
904 }
905
906 reqs = reqe = 0;
907
908 while (ress)
909 {
910 prv = ress;
911 ress = prv->next;
912 req_free (prv);
913 }
914
915 ress = rese = 0;
916 1148
917 close (respipe [0]); 1149 close (respipe [0]);
918 close (respipe [1]); 1150 close (respipe [1]);
919 create_pipe (); 1151 create_pipe ();
920 1152
921 atfork_parent (); 1153 atfork_parent ();
922} 1154}
923 1155
924#define dREQ \ 1156#define dREQ \
925 aio_req req; \ 1157 aio_req req; \
1158 int req_pri = next_pri; \
1159 next_pri = DEFAULT_PRI + PRI_BIAS; \
926 \ 1160 \
927 if (SvOK (callback) && !SvROK (callback)) \ 1161 if (SvOK (callback) && !SvROK (callback)) \
928 croak ("callback must be undef or of reference type"); \ 1162 croak ("callback must be undef or of reference type"); \
929 \ 1163 \
930 Newz (0, req, 1, aio_cb); \ 1164 Newz (0, req, 1, aio_cb); \
931 if (!req) \ 1165 if (!req) \
932 croak ("out of memory during aio_req allocation"); \ 1166 croak ("out of memory during aio_req allocation"); \
933 \ 1167 \
934 req->callback = newSVsv (callback) 1168 req->callback = newSVsv (callback); \
1169 req->pri = req_pri
935 1170
936#define REQ_SEND \ 1171#define REQ_SEND \
937 req_send (req); \ 1172 req_send (req); \
938 \ 1173 \
939 if (GIMME_V != G_VOID) \ 1174 if (GIMME_V != G_VOID) \
944PROTOTYPES: ENABLE 1179PROTOTYPES: ENABLE
945 1180
946BOOT: 1181BOOT:
947{ 1182{
948 HV *stash = gv_stashpv ("IO::AIO", 1); 1183 HV *stash = gv_stashpv ("IO::AIO", 1);
1184
949 newCONSTSUB (stash, "EXDEV", newSViv (EXDEV)); 1185 newCONSTSUB (stash, "EXDEV", newSViv (EXDEV));
950 newCONSTSUB (stash, "O_RDONLY", newSViv (O_RDONLY)); 1186 newCONSTSUB (stash, "O_RDONLY", newSViv (O_RDONLY));
951 newCONSTSUB (stash, "O_WRONLY", newSViv (O_WRONLY)); 1187 newCONSTSUB (stash, "O_WRONLY", newSViv (O_WRONLY));
1188 newCONSTSUB (stash, "O_CREAT", newSViv (O_CREAT));
1189 newCONSTSUB (stash, "O_TRUNC", newSViv (O_TRUNC));
1190 newCONSTSUB (stash, "S_IFIFO", newSViv (S_IFIFO));
952 1191
953 create_pipe (); 1192 create_pipe ();
954 pthread_atfork (atfork_prepare, atfork_parent, atfork_child); 1193 pthread_atfork (atfork_prepare, atfork_parent, atfork_child);
955}
956 1194
1195 start_thread ();
1196}
1197
957void 1198void
958min_parallel (nthreads) 1199min_parallel (int nthreads)
959 int nthreads
960 PROTOTYPE: $ 1200 PROTOTYPE: $
961 1201
962void 1202void
963max_parallel (nthreads) 1203max_parallel (int nthreads)
964 int nthreads
965 PROTOTYPE: $ 1204 PROTOTYPE: $
966 1205
967int 1206int
968max_outstanding (nreqs) 1207max_outstanding (int maxreqs)
969 int nreqs 1208 PROTOTYPE: $
970 PROTOTYPE: $
971 CODE: 1209 CODE:
972 RETVAL = max_outstanding; 1210 RETVAL = max_outstanding;
973 max_outstanding = nreqs; 1211 max_outstanding = maxreqs;
1212 OUTPUT:
1213 RETVAL
974 1214
975void 1215void
976aio_open (pathname,flags,mode,callback=&PL_sv_undef) 1216aio_open (pathname,flags,mode,callback=&PL_sv_undef)
977 SV * pathname 1217 SV * pathname
978 int flags 1218 int flags
1193 1433
1194 REQ_SEND; 1434 REQ_SEND;
1195} 1435}
1196 1436
1197void 1437void
1438aio_mknod (pathname,mode,dev,callback=&PL_sv_undef)
1439 SV * pathname
1440 SV * callback
1441 UV mode
1442 UV dev
1443 PPCODE:
1444{
1445 dREQ;
1446
1447 req->type = REQ_MKNOD;
1448 req->data = newSVsv (pathname);
1449 req->dataptr = SvPVbyte_nolen (req->data);
1450 req->mode = (mode_t)mode;
1451 req->offset = dev;
1452
1453 REQ_SEND;
1454}
1455
1456void
1198aio_sleep (delay,callback=&PL_sv_undef) 1457aio_busy (delay,callback=&PL_sv_undef)
1199 double delay 1458 double delay
1200 SV * callback 1459 SV * callback
1201 PPCODE: 1460 PPCODE:
1202{ 1461{
1203 dREQ; 1462 dREQ;
1204 1463
1205 req->type = REQ_SLEEP; 1464 req->type = REQ_BUSY;
1206 req->fd = delay < 0. ? 0 : delay; 1465 req->fd = delay < 0. ? 0 : delay;
1207 req->fd2 = delay < 0. ? 0 : 1000. * (delay - req->fd); 1466 req->fd2 = delay < 0. ? 0 : 1000. * (delay - req->fd);
1208 1467
1209 REQ_SEND; 1468 REQ_SEND;
1210} 1469}
1214 SV * callback 1473 SV * callback
1215 PROTOTYPE: ;$ 1474 PROTOTYPE: ;$
1216 PPCODE: 1475 PPCODE:
1217{ 1476{
1218 dREQ; 1477 dREQ;
1478
1219 req->type = REQ_GROUP; 1479 req->type = REQ_GROUP;
1220 req_send (req); 1480 req_send (req);
1481
1221 XPUSHs (req_sv (req, AIO_GRP_KLASS)); 1482 XPUSHs (req_sv (req, AIO_GRP_KLASS));
1222} 1483}
1223 1484
1224void 1485void
1225aio_nop (callback=&PL_sv_undef) 1486aio_nop (callback=&PL_sv_undef)
1230 1491
1231 req->type = REQ_NOP; 1492 req->type = REQ_NOP;
1232 1493
1233 REQ_SEND; 1494 REQ_SEND;
1234} 1495}
1496
1497int
1498aioreq_pri (int pri = 0)
1499 PROTOTYPE: ;$
1500 CODE:
1501 RETVAL = next_pri - PRI_BIAS;
1502 if (items > 0)
1503 {
1504 if (pri < PRI_MIN) pri = PRI_MIN;
1505 if (pri > PRI_MAX) pri = PRI_MAX;
1506 next_pri = pri + PRI_BIAS;
1507 }
1508 OUTPUT:
1509 RETVAL
1510
1511void
1512aioreq_nice (int nice = 0)
1513 CODE:
1514 nice = next_pri - nice;
1515 if (nice < PRI_MIN) nice = PRI_MIN;
1516 if (nice > PRI_MAX) nice = PRI_MAX;
1517 next_pri = nice + PRI_BIAS;
1235 1518
1236void 1519void
1237flush () 1520flush ()
1238 PROTOTYPE: 1521 PROTOTYPE:
1239 CODE: 1522 CODE:
1240 while (nreqs) 1523 while (nreqs)
1241 { 1524 {
1242 poll_wait (); 1525 poll_wait ();
1243 poll_cb (); 1526 poll_cb (0);
1244 } 1527 }
1245 1528
1246void 1529void
1247poll() 1530poll()
1248 PROTOTYPE: 1531 PROTOTYPE:
1249 CODE: 1532 CODE:
1250 if (nreqs) 1533 if (nreqs)
1251 { 1534 {
1252 poll_wait (); 1535 poll_wait ();
1253 poll_cb (); 1536 poll_cb (0);
1254 } 1537 }
1255 1538
1256int 1539int
1257poll_fileno() 1540poll_fileno()
1258 PROTOTYPE: 1541 PROTOTYPE:
1263 1546
1264int 1547int
1265poll_cb(...) 1548poll_cb(...)
1266 PROTOTYPE: 1549 PROTOTYPE:
1267 CODE: 1550 CODE:
1268 RETVAL = poll_cb (); 1551 RETVAL = poll_cb (0);
1552 OUTPUT:
1553 RETVAL
1554
1555int
1556poll_some(int max = 0)
1557 PROTOTYPE: $
1558 CODE:
1559 RETVAL = poll_cb (max);
1269 OUTPUT: 1560 OUTPUT:
1270 RETVAL 1561 RETVAL
1271 1562
1272void 1563void
1273poll_wait() 1564poll_wait()
1282 CODE: 1573 CODE:
1283 RETVAL = nreqs; 1574 RETVAL = nreqs;
1284 OUTPUT: 1575 OUTPUT:
1285 RETVAL 1576 RETVAL
1286 1577
1578int
1579nready()
1580 PROTOTYPE:
1581 CODE:
1582 RETVAL = get_nready ();
1583 OUTPUT:
1584 RETVAL
1585
1586int
1587npending()
1588 PROTOTYPE:
1589 CODE:
1590 RETVAL = get_npending ();
1591 OUTPUT:
1592 RETVAL
1593
1287PROTOTYPES: DISABLE 1594PROTOTYPES: DISABLE
1288 1595
1289MODULE = IO::AIO PACKAGE = IO::AIO::REQ 1596MODULE = IO::AIO PACKAGE = IO::AIO::REQ
1290 1597
1291void 1598void
1292cancel (aio_req_ornot req) 1599cancel (aio_req_ornot req)
1293 PROTOTYPE:
1294 CODE: 1600 CODE:
1295 req_cancel (req); 1601 req_cancel (req);
1296 1602
1297void 1603void
1298cb (aio_req_ornot req, SV *callback=&PL_sv_undef) 1604cb (aio_req_ornot req, SV *callback=&PL_sv_undef)
1334 } 1640 }
1335 } 1641 }
1336} 1642}
1337 1643
1338void 1644void
1645cancel_subs (aio_req_ornot req)
1646 CODE:
1647 req_cancel_subs (req);
1648
1649void
1339result (aio_req grp, ...) 1650result (aio_req grp, ...)
1340 CODE: 1651 CODE:
1341{ 1652{
1342 int i; 1653 int i;
1654 AV *av;
1655
1656 grp->errorno = errno;
1657
1343 AV *av = newAV (); 1658 av = newAV ();
1344 1659
1345 for (i = 1; i < items; ++i ) 1660 for (i = 1; i < items; ++i )
1346 av_push (av, newSVsv (ST (i))); 1661 av_push (av, newSVsv (ST (i)));
1347 1662
1348 SvREFCNT_dec (grp->data); 1663 SvREFCNT_dec (grp->data);
1349 grp->data = (SV *)av; 1664 grp->data = (SV *)av;
1350} 1665}
1351 1666
1352void 1667void
1668errno (aio_req grp, int errorno = errno)
1669 CODE:
1670 grp->errorno = errorno;
1671
1672void
1353feed_limit (aio_req grp, int limit) 1673limit (aio_req grp, int limit)
1354 CODE: 1674 CODE:
1355 grp->fd2 = limit; 1675 grp->fd2 = limit;
1356 aio_grp_feed (grp); 1676 aio_grp_feed (grp);
1357 1677
1358void 1678void

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines