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.83 by root, Sat Oct 28 00:17:30 2006 UTC vs.
Revision 1.88 by root, Mon Oct 30 23:43:17 2006 UTC

43# else 43# else
44# error sendfile support requested but not available 44# error sendfile support requested but not available
45# endif 45# endif
46#endif 46#endif
47 47
48/* number of seconds after which idle threads exit */
49#define IDLE_TIMEOUT 10
50
48/* used for struct dirent, AIX doesn't provide it */ 51/* used for struct dirent, AIX doesn't provide it */
49#ifndef NAME_MAX 52#ifndef NAME_MAX
50# define NAME_MAX 4096 53# define NAME_MAX 4096
51#endif 54#endif
52 55
65 68
66/* wether word reads are potentially non-atomic. 69/* wether word reads are potentially non-atomic.
67 * this is conservatice, likely most arches this runs 70 * this is conservatice, likely most arches this runs
68 * on have atomic word read/writes. 71 * on have atomic word read/writes.
69 */ 72 */
70#ifndef WORDREAD_UNSAFE 73#ifndef WORDACCESS_UNSAFE
71# if __i386 || __x86_64 74# if __i386 || __x86_64
72# define WORDREAD_UNSAFE 0 75# define WORDACCESS_UNSAFE 0
73# else 76# else
74# define WORDREAD_UNSAFE 1 77# define WORDACCESS_UNSAFE 1
75# endif 78# endif
76#endif 79#endif
77 80
78/* buffer size for various temporary buffers */ 81/* buffer size for various temporary buffers */
79#define AIO_BUFSIZE 65536 82#define AIO_BUFSIZE 65536
93 REQ_SENDFILE, 96 REQ_SENDFILE,
94 REQ_STAT, REQ_LSTAT, REQ_FSTAT, 97 REQ_STAT, REQ_LSTAT, REQ_FSTAT,
95 REQ_FSYNC, REQ_FDATASYNC, 98 REQ_FSYNC, REQ_FDATASYNC,
96 REQ_UNLINK, REQ_RMDIR, REQ_RENAME, 99 REQ_UNLINK, REQ_RMDIR, REQ_RENAME,
97 REQ_MKNOD, REQ_READDIR, 100 REQ_MKNOD, REQ_READDIR,
98 REQ_LINK, REQ_SYMLINK, 101 REQ_LINK, REQ_SYMLINK, REQ_READLINK,
99 REQ_GROUP, REQ_NOP, 102 REQ_GROUP, REQ_NOP,
100 REQ_BUSY, 103 REQ_BUSY,
101}; 104};
102 105
103#define AIO_REQ_KLASS "IO::AIO::REQ" 106#define AIO_REQ_KLASS "IO::AIO::REQ"
105 108
106typedef struct aio_cb 109typedef struct aio_cb
107{ 110{
108 struct aio_cb *volatile next; 111 struct aio_cb *volatile next;
109 112
110 SV *data, *callback; 113 SV *callback, *fh;
111 SV *fh, *fh2; 114 SV *sv1, *sv2;
112 void *dataptr, *data2ptr; 115 void *ptr1, *ptr2;
113 Stat_t *statdata;
114 off_t offset; 116 off_t offs;
115 size_t length; 117 size_t size;
116 ssize_t result; 118 ssize_t result;
117 119
118 STRLEN dataoffset; 120 STRLEN stroffset;
119 int type; 121 int type;
120 int fd, fd2; 122 int int1, int2;
121 int errorno; 123 int errorno;
122 mode_t mode; /* open */ 124 mode_t mode; /* open */
123 125
124 unsigned char flags; 126 unsigned char flags;
125 unsigned char pri; 127 unsigned char pri;
127 SV *self; /* the perl counterpart of this request, if any */ 129 SV *self; /* the perl counterpart of this request, if any */
128 struct aio_cb *grp, *grp_prev, *grp_next, *grp_first; 130 struct aio_cb *grp, *grp_prev, *grp_next, *grp_first;
129} aio_cb; 131} aio_cb;
130 132
131enum { 133enum {
132 FLAG_CANCELLED = 0x01, 134 FLAG_CANCELLED = 0x01,
135 FLAG_SV1_RO_OFF = 0x40, /* data was set readonly */
136 FLAG_PTR2_FREE = 0x80, /* need free(ptr2) */
133}; 137};
134 138
135typedef aio_cb *aio_req; 139typedef aio_cb *aio_req;
136typedef aio_cb *aio_req_ornot; 140typedef aio_cb *aio_req_ornot;
137 141
142 DEFAULT_PRI = 0, 146 DEFAULT_PRI = 0,
143 PRI_BIAS = -PRI_MIN, 147 PRI_BIAS = -PRI_MIN,
144 NUM_PRI = PRI_MAX + PRI_BIAS + 1, 148 NUM_PRI = PRI_MAX + PRI_BIAS + 1,
145}; 149};
146 150
151#define AIO_TICKS ((1000000 + 1023) >> 10)
152
153static unsigned int max_poll_time = 0;
154static unsigned int max_poll_reqs = 0;
155
156/* calculcate time difference in ~1/AIO_TICKS of a second */
157static int tvdiff (struct timeval *tv1, struct timeval *tv2)
158{
159 return (tv2->tv_sec - tv1->tv_sec ) * AIO_TICKS
160 + ((tv2->tv_usec - tv1->tv_usec) >> 10);
161}
162
147static int next_pri = DEFAULT_PRI + PRI_BIAS; 163static int next_pri = DEFAULT_PRI + PRI_BIAS;
148 164
149static unsigned int started, wanted; 165static unsigned int started, idle, wanted;
150 166
151#if __linux && defined (PTHREAD_ADAPTIVE_MUTEX_INITIALIZER_NP) 167#if __linux && defined (PTHREAD_ADAPTIVE_MUTEX_INITIALIZER_NP)
152# define AIO_MUTEX_INIT PTHREAD_ADAPTIVE_MUTEX_INITIALIZER_NP 168# define AIO_MUTEX_INIT PTHREAD_ADAPTIVE_MUTEX_INITIALIZER_NP
153#else 169#else
154# define AIO_MUTEX_INIT PTHREAD_MUTEX_INITIALIZER 170# define AIO_MUTEX_INIT PTHREAD_MUTEX_INITIALIZER
196 212
197 free (wrk); 213 free (wrk);
198} 214}
199 215
200static volatile unsigned int nreqs, nready, npending; 216static volatile unsigned int nreqs, nready, npending;
217static volatile unsigned int max_idle = 4;
201static volatile unsigned int max_outstanding = 0xffffffff; 218static volatile unsigned int max_outstanding = 0xffffffff;
202static int respipe [2]; 219static int respipe [2];
203 220
204static pthread_mutex_t reslock = AIO_MUTEX_INIT; 221static pthread_mutex_t reslock = AIO_MUTEX_INIT;
205static pthread_mutex_t reqlock = AIO_MUTEX_INIT; 222static pthread_mutex_t reqlock = AIO_MUTEX_INIT;
206static pthread_cond_t reqwait = PTHREAD_COND_INITIALIZER; 223static pthread_cond_t reqwait = PTHREAD_COND_INITIALIZER;
207 224
208#if WORDREAD_UNSAFE 225#if WORDACCESS_UNSAFE
209 226
210static unsigned int get_nready () 227static unsigned int get_nready ()
211{ 228{
212 unsigned int retval; 229 unsigned int retval;
213 230
227 UNLOCK (reslock); 244 UNLOCK (reslock);
228 245
229 return retval; 246 return retval;
230} 247}
231 248
249static unsigned int get_nthreads ()
250{
251 unsigned int retval;
252
253 LOCK (wrklock);
254 retval = started;
255 UNLOCK (wrklock);
256
257 return retval;
258}
259
232#else 260#else
233 261
234# define get_nready() nready 262# define get_nready() nready
235# define get_npending() npending 263# define get_npending() npending
264# define get_nthreads() started
236 265
237#endif 266#endif
238 267
239/* 268/*
240 * a somewhat faster data structure might be nice, but 269 * a somewhat faster data structure might be nice, but
288 } 317 }
289 318
290 abort (); 319 abort ();
291} 320}
292 321
293static int poll_cb (int max); 322static int poll_cb ();
294static void req_invoke (aio_req req); 323static void req_invoke (aio_req req);
295static void req_free (aio_req req); 324static void req_free (aio_req req);
296static void req_cancel (aio_req req); 325static void req_cancel (aio_req req);
297 326
298/* must be called at most once */ 327/* must be called at most once */
319 return mg ? (aio_req)mg->mg_ptr : 0; 348 return mg ? (aio_req)mg->mg_ptr : 0;
320} 349}
321 350
322static void aio_grp_feed (aio_req grp) 351static void aio_grp_feed (aio_req grp)
323{ 352{
324 while (grp->length < grp->fd2 && !(grp->flags & FLAG_CANCELLED)) 353 while (grp->size < grp->int2 && !(grp->flags & FLAG_CANCELLED))
325 { 354 {
326 int old_len = grp->length; 355 int old_len = grp->size;
327 356
328 if (grp->fh2 && SvOK (grp->fh2)) 357 if (grp->sv2 && SvOK (grp->sv2))
329 { 358 {
330 dSP; 359 dSP;
331 360
332 ENTER; 361 ENTER;
333 SAVETMPS; 362 SAVETMPS;
334 PUSHMARK (SP); 363 PUSHMARK (SP);
335 XPUSHs (req_sv (grp, AIO_GRP_KLASS)); 364 XPUSHs (req_sv (grp, AIO_GRP_KLASS));
336 PUTBACK; 365 PUTBACK;
337 call_sv (grp->fh2, G_VOID | G_EVAL | G_KEEPERR); 366 call_sv (grp->sv2, G_VOID | G_EVAL | G_KEEPERR);
338 SPAGAIN; 367 SPAGAIN;
339 FREETMPS; 368 FREETMPS;
340 LEAVE; 369 LEAVE;
341 } 370 }
342 371
343 /* stop if no progress has been made */ 372 /* stop if no progress has been made */
344 if (old_len == grp->length) 373 if (old_len == grp->size)
345 { 374 {
346 SvREFCNT_dec (grp->fh2); 375 SvREFCNT_dec (grp->sv2);
347 grp->fh2 = 0; 376 grp->sv2 = 0;
348 break; 377 break;
349 } 378 }
350 } 379 }
351} 380}
352 381
353static void aio_grp_dec (aio_req grp) 382static void aio_grp_dec (aio_req grp)
354{ 383{
355 --grp->length; 384 --grp->size;
356 385
357 /* call feeder, if applicable */ 386 /* call feeder, if applicable */
358 aio_grp_feed (grp); 387 aio_grp_feed (grp);
359 388
360 /* finish, if done */ 389 /* finish, if done */
361 if (!grp->length && grp->fd) 390 if (!grp->size && grp->int1)
362 { 391 {
363 req_invoke (grp); 392 req_invoke (grp);
364 req_free (grp); 393 req_free (grp);
365 } 394 }
366} 395}
367 396
368static void req_invoke (aio_req req) 397static void req_invoke (aio_req req)
369{ 398{
370 dSP; 399 dSP;
400
401 if (req->flags & FLAG_SV1_RO_OFF)
402 SvREADONLY_off (req->sv1);
371 403
372 if (!(req->flags & FLAG_CANCELLED) && SvOK (req->callback)) 404 if (!(req->flags & FLAG_CANCELLED) && SvOK (req->callback))
373 { 405 {
374 ENTER; 406 ENTER;
375 SAVETMPS; 407 SAVETMPS;
383 SV *rv = &PL_sv_undef; 415 SV *rv = &PL_sv_undef;
384 416
385 if (req->result >= 0) 417 if (req->result >= 0)
386 { 418 {
387 int i; 419 int i;
388 char *buf = req->data2ptr; 420 char *buf = req->ptr2;
389 AV *av = newAV (); 421 AV *av = newAV ();
390 422
391 av_extend (av, req->result - 1); 423 av_extend (av, req->result - 1);
392 424
393 for (i = 0; i < req->result; ++i) 425 for (i = 0; i < req->result; ++i)
421 XPUSHs (sv_2mortal (fh)); 453 XPUSHs (sv_2mortal (fh));
422 } 454 }
423 break; 455 break;
424 456
425 case REQ_GROUP: 457 case REQ_GROUP:
426 req->fd = 2; /* mark group as finished */ 458 req->int1 = 2; /* mark group as finished */
427 459
428 if (req->data) 460 if (req->sv1)
429 { 461 {
430 int i; 462 int i;
431 AV *av = (AV *)req->data; 463 AV *av = (AV *)req->sv1;
432 464
433 EXTEND (SP, AvFILL (av) + 1); 465 EXTEND (SP, AvFILL (av) + 1);
434 for (i = 0; i <= AvFILL (av); ++i) 466 for (i = 0; i <= AvFILL (av); ++i)
435 PUSHs (*av_fetch (av, i, 0)); 467 PUSHs (*av_fetch (av, i, 0));
436 } 468 }
438 470
439 case REQ_NOP: 471 case REQ_NOP:
440 case REQ_BUSY: 472 case REQ_BUSY:
441 break; 473 break;
442 474
475 case REQ_READLINK:
476 if (req->result > 0)
477 {
478 SvCUR_set (req->sv1, req->result);
479 *SvEND (req->sv1) = 0;
480 PUSHs (req->sv1);
481 }
482 break;
483
484 case REQ_STAT:
485 case REQ_LSTAT:
486 case REQ_FSTAT:
487 PL_laststype = req->type == REQ_LSTAT ? OP_LSTAT : OP_STAT;
488 PL_laststatval = req->result;
489 PL_statcache = *(Stat_t *)(req->ptr2);
490 PUSHs (sv_2mortal (newSViv (req->result)));
491 break;
492
493 case REQ_READ:
494 SvCUR_set (req->sv1, req->stroffset + (req->result > 0 ? req->result : 0));
495 *SvEND (req->sv1) = 0;
496 PUSHs (sv_2mortal (newSViv (req->result)));
497 break;
498
443 default: 499 default:
444 PUSHs (sv_2mortal (newSViv (req->result))); 500 PUSHs (sv_2mortal (newSViv (req->result)));
445 break; 501 break;
446 } 502 }
447 503
482 { 538 {
483 sv_unmagic (req->self, PERL_MAGIC_ext); 539 sv_unmagic (req->self, PERL_MAGIC_ext);
484 SvREFCNT_dec (req->self); 540 SvREFCNT_dec (req->self);
485 } 541 }
486 542
487 SvREFCNT_dec (req->data);
488 SvREFCNT_dec (req->fh); 543 SvREFCNT_dec (req->fh);
544 SvREFCNT_dec (req->sv1);
489 SvREFCNT_dec (req->fh2); 545 SvREFCNT_dec (req->sv2);
490 SvREFCNT_dec (req->callback); 546 SvREFCNT_dec (req->callback);
491 Safefree (req->statdata);
492 547
493 if (req->type == REQ_READDIR) 548 if (req->flags & FLAG_PTR2_FREE)
494 free (req->data2ptr); 549 free (req->ptr2);
495 550
496 Safefree (req); 551 Safefree (req);
497} 552}
498 553
499static void req_cancel_subs (aio_req grp) 554static void req_cancel_subs (aio_req grp)
501 aio_req sub; 556 aio_req sub;
502 557
503 if (grp->type != REQ_GROUP) 558 if (grp->type != REQ_GROUP)
504 return; 559 return;
505 560
506 SvREFCNT_dec (grp->fh2); 561 SvREFCNT_dec (grp->sv2);
507 grp->fh2 = 0; 562 grp->sv2 = 0;
508 563
509 for (sub = grp->grp_first; sub; sub = sub->grp_next) 564 for (sub = grp->grp_first; sub; sub = sub->grp_next)
510 req_cancel (sub); 565 req_cancel (sub);
511} 566}
512 567
556 UNLOCK (wrklock); 611 UNLOCK (wrklock);
557} 612}
558 613
559static void maybe_start_thread () 614static void maybe_start_thread ()
560{ 615{
561#if 0 616 if (get_nthreads () >= wanted)
562 static struct timeval last;
563 struct timeval diff, now;
564#endif
565
566 if (started >= wanted)
567 return; 617 return;
568 618
569 if (nready <= nreqs - get_nready () - get_npending ()) 619 /* todo: maybe use idle here, but might be less exact */
620 if (0 <= (int)get_nthreads () + (int)get_npending () - (int)nreqs)
570 return; 621 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 622
590 start_thread (); 623 start_thread ();
591} 624}
592 625
593static void req_send (aio_req req) 626static void req_send (aio_req req)
620 LOCK (wrklock); 653 LOCK (wrklock);
621 --started; 654 --started;
622 UNLOCK (wrklock); 655 UNLOCK (wrklock);
623} 656}
624 657
658static void set_max_idle (int nthreads)
659{
660 if (WORDACCESS_UNSAFE) LOCK (reqlock);
661 max_idle = nthreads <= 0 ? 1 : nthreads;
662 if (WORDACCESS_UNSAFE) UNLOCK (reqlock);
663}
664
625static void min_parallel (int nthreads) 665static void min_parallel (int nthreads)
626{ 666{
627 if (wanted < nthreads) 667 if (wanted < nthreads)
628 wanted = nthreads; 668 wanted = nthreads;
629} 669}
642 fd_set rfd; 682 fd_set rfd;
643 683
644 while (nreqs) 684 while (nreqs)
645 { 685 {
646 int size; 686 int size;
647 if (WORDREAD_UNSAFE) LOCK (reslock); 687 if (WORDACCESS_UNSAFE) LOCK (reslock);
648 size = res_queue.size; 688 size = res_queue.size;
649 if (WORDREAD_UNSAFE) UNLOCK (reslock); 689 if (WORDACCESS_UNSAFE) UNLOCK (reslock);
650 690
651 if (size) 691 if (size)
652 return; 692 return;
653 693
654 maybe_start_thread (); 694 maybe_start_thread ();
658 698
659 select (respipe [0] + 1, &rfd, 0, 0, 0); 699 select (respipe [0] + 1, &rfd, 0, 0, 0);
660 } 700 }
661} 701}
662 702
663static int poll_cb (int max) 703static int poll_cb ()
664{ 704{
665 dSP; 705 dSP;
666 int count = 0; 706 int count = 0;
707 int maxreqs = max_poll_reqs;
667 int do_croak = 0; 708 int do_croak = 0;
709 struct timeval tv_start, tv_now;
668 aio_req req; 710 aio_req req;
711
712 if (max_poll_time)
713 gettimeofday (&tv_start, 0);
669 714
670 for (;;) 715 for (;;)
671 { 716 {
672 while (max <= 0 || count < max) 717 for (;;)
673 { 718 {
674 maybe_start_thread (); 719 maybe_start_thread ();
675 720
676 LOCK (reslock); 721 LOCK (reslock);
677 req = reqq_shift (&res_queue); 722 req = reqq_shift (&res_queue);
694 if (!req) 739 if (!req)
695 break; 740 break;
696 741
697 --nreqs; 742 --nreqs;
698 743
699 if (req->type == REQ_GROUP && req->length) 744 if (req->type == REQ_GROUP && req->size)
700 { 745 {
701 req->fd = 1; /* mark request as delayed */ 746 req->int1 = 1; /* mark request as delayed */
702 continue; 747 continue;
703 } 748 }
704 else 749 else
705 { 750 {
706 if (req->type == REQ_READ)
707 SvCUR_set (req->data, req->dataoffset + (req->result > 0 ? req->result : 0));
708
709 if (req->data2ptr && (req->type == REQ_READ || req->type == REQ_WRITE))
710 SvREADONLY_off (req->data);
711
712 if (req->statdata)
713 {
714 PL_laststype = req->type == REQ_LSTAT ? OP_LSTAT : OP_STAT;
715 PL_laststatval = req->result;
716 PL_statcache = *(req->statdata);
717 }
718
719 req_invoke (req); 751 req_invoke (req);
720 752
721 count++; 753 count++;
722 } 754 }
723 755
724 req_free (req); 756 req_free (req);
757
758 if (maxreqs && !--maxreqs)
759 break;
760
761 if (max_poll_time)
762 {
763 gettimeofday (&tv_now, 0);
764
765 if (tvdiff (&tv_start, &tv_now) >= max_poll_time)
766 break;
767 }
725 } 768 }
726 769
727 if (nreqs <= max_outstanding) 770 if (nreqs <= max_outstanding)
728 break; 771 break;
729 772
730 poll_wait (); 773 poll_wait ();
731 774
732 max = 0; 775 ++maxreqs;
733 } 776 }
734 777
735 return count; 778 return count;
736} 779}
737 780
955 int memofs = 0; 998 int memofs = 0;
956 int res = 0; 999 int res = 0;
957 int errorno; 1000 int errorno;
958 1001
959 LOCK (wrklock); 1002 LOCK (wrklock);
960 self->dirp = dirp = opendir (req->dataptr); 1003 self->dirp = dirp = opendir (req->ptr1);
961 self->dbuf = u = malloc (sizeof (*u)); 1004 self->dbuf = u = malloc (sizeof (*u));
1005 req->flags |= FLAG_PTR2_FREE;
962 req->data2ptr = names = malloc (memlen); 1006 req->ptr2 = names = malloc (memlen);
963 UNLOCK (wrklock); 1007 UNLOCK (wrklock);
964 1008
965 if (dirp && u && names) 1009 if (dirp && u && names)
966 for (;;) 1010 for (;;)
967 { 1011 {
981 1025
982 while (memofs + len > memlen) 1026 while (memofs + len > memlen)
983 { 1027 {
984 memlen *= 2; 1028 memlen *= 2;
985 LOCK (wrklock); 1029 LOCK (wrklock);
986 req->data2ptr = names = realloc (names, memlen); 1030 req->ptr2 = names = realloc (names, memlen);
987 UNLOCK (wrklock); 1031 UNLOCK (wrklock);
988 1032
989 if (!names) 1033 if (!names)
990 break; 1034 break;
991 } 1035 }
1004/*****************************************************************************/ 1048/*****************************************************************************/
1005 1049
1006static void *aio_proc (void *thr_arg) 1050static void *aio_proc (void *thr_arg)
1007{ 1051{
1008 aio_req req; 1052 aio_req req;
1053 struct timespec ts;
1009 worker *self = (worker *)thr_arg; 1054 worker *self = (worker *)thr_arg;
1055
1056 /* try to distribute timeouts somewhat evenly */
1057 ts.tv_nsec = (((unsigned long)self + (unsigned long)ts.tv_sec) & 1023UL)
1058 * (1000000000UL / 1024UL);
1010 1059
1011 for (;;) 1060 for (;;)
1012 { 1061 {
1062 ts.tv_sec = time (0) + IDLE_TIMEOUT;
1063
1013 LOCK (reqlock); 1064 LOCK (reqlock);
1014 1065
1015 for (;;) 1066 for (;;)
1016 { 1067 {
1017 self->req = req = reqq_shift (&req_queue); 1068 self->req = req = reqq_shift (&req_queue);
1018 1069
1019 if (req) 1070 if (req)
1020 break; 1071 break;
1021 1072
1073 ++idle;
1074
1075 if (pthread_cond_timedwait (&reqwait, &reqlock, &ts)
1076 == ETIMEDOUT)
1077 {
1078 if (idle > max_idle)
1079 {
1080 --idle;
1081 UNLOCK (reqlock);
1082 LOCK (wrklock);
1083 --started;
1084 UNLOCK (wrklock);
1085 goto quit;
1086 }
1087
1088 /* we are allowed to idle, so do so without any timeout */
1022 pthread_cond_wait (&reqwait, &reqlock); 1089 pthread_cond_wait (&reqwait, &reqlock);
1090 ts.tv_sec = time (0) + IDLE_TIMEOUT;
1091 }
1092
1093 --idle;
1023 } 1094 }
1024 1095
1025 --nready; 1096 --nready;
1026 1097
1027 UNLOCK (reqlock); 1098 UNLOCK (reqlock);
1029 errno = 0; /* strictly unnecessary */ 1100 errno = 0; /* strictly unnecessary */
1030 1101
1031 if (!(req->flags & FLAG_CANCELLED)) 1102 if (!(req->flags & FLAG_CANCELLED))
1032 switch (req->type) 1103 switch (req->type)
1033 { 1104 {
1034 case REQ_READ: req->result = pread (req->fd, req->dataptr, req->length, req->offset); break; 1105 case REQ_READ: req->result = pread (req->int1, req->ptr1, req->size, req->offs); break;
1035 case REQ_WRITE: req->result = pwrite (req->fd, req->dataptr, req->length, req->offset); break; 1106 case REQ_WRITE: req->result = pwrite (req->int1, req->ptr1, req->size, req->offs); break;
1036 1107
1037 case REQ_READAHEAD: req->result = readahead (req->fd, req->offset, req->length); break; 1108 case REQ_READAHEAD: req->result = readahead (req->int1, req->offs, req->size); break;
1038 case REQ_SENDFILE: req->result = sendfile_ (req->fd, req->fd2, req->offset, req->length, self); break; 1109 case REQ_SENDFILE: req->result = sendfile_ (req->int1, req->int2, req->offs, req->size, self); break;
1039 1110
1040 case REQ_STAT: req->result = stat (req->dataptr, req->statdata); break; 1111 case REQ_STAT: req->result = stat (req->ptr1, (Stat_t *)req->ptr2); break;
1041 case REQ_LSTAT: req->result = lstat (req->dataptr, req->statdata); break; 1112 case REQ_LSTAT: req->result = lstat (req->ptr1, (Stat_t *)req->ptr2); break;
1042 case REQ_FSTAT: req->result = fstat (req->fd , req->statdata); break; 1113 case REQ_FSTAT: req->result = fstat (req->int1, (Stat_t *)req->ptr2); break;
1043 1114
1044 case REQ_OPEN: req->result = open (req->dataptr, req->fd, req->mode); break; 1115 case REQ_OPEN: req->result = open (req->ptr1, req->int1, req->mode); break;
1045 case REQ_CLOSE: req->result = close (req->fd); break; 1116 case REQ_CLOSE: req->result = close (req->int1); break;
1046 case REQ_UNLINK: req->result = unlink (req->dataptr); break; 1117 case REQ_UNLINK: req->result = unlink (req->ptr1); break;
1047 case REQ_RMDIR: req->result = rmdir (req->dataptr); break; 1118 case REQ_RMDIR: req->result = rmdir (req->ptr1); break;
1048 case REQ_RENAME: req->result = rename (req->data2ptr, req->dataptr); break; 1119 case REQ_RENAME: req->result = rename (req->ptr2, req->ptr1); break;
1049 case REQ_LINK: req->result = link (req->data2ptr, req->dataptr); break; 1120 case REQ_LINK: req->result = link (req->ptr2, req->ptr1); break;
1050 case REQ_SYMLINK: req->result = symlink (req->data2ptr, req->dataptr); break; 1121 case REQ_SYMLINK: req->result = symlink (req->ptr2, req->ptr1); break;
1051 case REQ_MKNOD: req->result = mknod (req->data2ptr, req->mode, (dev_t)req->offset); break; 1122 case REQ_MKNOD: req->result = mknod (req->ptr2, req->mode, (dev_t)req->offs); break;
1123 case REQ_READLINK: req->result = readlink (req->ptr2, req->ptr1, NAME_MAX); break;
1052 1124
1053 case REQ_FDATASYNC: req->result = fdatasync (req->fd); break; 1125 case REQ_FDATASYNC: req->result = fdatasync (req->int1); break;
1054 case REQ_FSYNC: req->result = fsync (req->fd); break; 1126 case REQ_FSYNC: req->result = fsync (req->int1); break;
1055 case REQ_READDIR: scandir_ (req, self); break; 1127 case REQ_READDIR: scandir_ (req, self); break;
1056 1128
1057 case REQ_BUSY: 1129 case REQ_BUSY:
1058 { 1130 {
1059 struct timeval tv; 1131 struct timeval tv;
1060 1132
1061 tv.tv_sec = req->fd; 1133 tv.tv_sec = req->int1;
1062 tv.tv_usec = req->fd2; 1134 tv.tv_usec = req->int2;
1063 1135
1064 req->result = select (0, 0, 0, 0, &tv); 1136 req->result = select (0, 0, 0, 0, &tv);
1065 } 1137 }
1066 1138
1067 case REQ_GROUP: 1139 case REQ_GROUP:
1068 case REQ_NOP: 1140 case REQ_NOP:
1069 break; 1141 break;
1070 1142
1071 case REQ_QUIT: 1143 case REQ_QUIT:
1072 LOCK (wrklock); 1144 goto quit;
1073 worker_free (self);
1074 --started;
1075 UNLOCK (wrklock);
1076 return 0;
1077 1145
1078 default: 1146 default:
1079 req->result = ENOSYS; 1147 req->result = ENOSYS;
1080 break; 1148 break;
1081 } 1149 }
1093 self->req = 0; 1161 self->req = 0;
1094 worker_clear (self); 1162 worker_clear (self);
1095 1163
1096 UNLOCK (reslock); 1164 UNLOCK (reslock);
1097 } 1165 }
1166
1167quit:
1168 LOCK (wrklock);
1169 worker_free (self);
1170 UNLOCK (wrklock);
1171
1172 return 0;
1098} 1173}
1099 1174
1100/*****************************************************************************/ 1175/*****************************************************************************/
1101 1176
1102static void atfork_prepare (void) 1177static void atfork_prepare (void)
1144 1219
1145 worker_clear (wrk); 1220 worker_clear (wrk);
1146 worker_free (wrk); 1221 worker_free (wrk);
1147 } 1222 }
1148 1223
1149 started = 0; 1224 started = 0;
1225 idle = 0;
1150 nreqs = 0; 1226 nreqs = 0;
1227 nready = 0;
1228 npending = 0;
1151 1229
1152 close (respipe [0]); 1230 close (respipe [0]);
1153 close (respipe [1]); 1231 close (respipe [1]);
1154 create_pipe (); 1232 create_pipe ();
1155 1233
1192 newCONSTSUB (stash, "O_TRUNC", newSViv (O_TRUNC)); 1270 newCONSTSUB (stash, "O_TRUNC", newSViv (O_TRUNC));
1193 newCONSTSUB (stash, "S_IFIFO", newSViv (S_IFIFO)); 1271 newCONSTSUB (stash, "S_IFIFO", newSViv (S_IFIFO));
1194 1272
1195 create_pipe (); 1273 create_pipe ();
1196 pthread_atfork (atfork_prepare, atfork_parent, atfork_child); 1274 pthread_atfork (atfork_prepare, atfork_parent, atfork_child);
1197
1198 start_thread ();
1199} 1275}
1276
1277void
1278max_poll_reqs (int nreqs)
1279 PROTOTYPE: $
1280 CODE:
1281 max_poll_reqs = nreqs;
1282
1283void
1284max_poll_time (double nseconds)
1285 PROTOTYPE: $
1286 CODE:
1287 max_poll_time = nseconds * AIO_TICKS;
1200 1288
1201void 1289void
1202min_parallel (int nthreads) 1290min_parallel (int nthreads)
1203 PROTOTYPE: $ 1291 PROTOTYPE: $
1204 1292
1205void 1293void
1206max_parallel (int nthreads) 1294max_parallel (int nthreads)
1207 PROTOTYPE: $ 1295 PROTOTYPE: $
1296
1297void
1298max_idle (int nthreads)
1299 PROTOTYPE: $
1300 CODE:
1301 set_max_idle (nthreads);
1208 1302
1209int 1303int
1210max_outstanding (int maxreqs) 1304max_outstanding (int maxreqs)
1211 PROTOTYPE: $ 1305 PROTOTYPE: $
1212 CODE: 1306 CODE:
1225 PPCODE: 1319 PPCODE:
1226{ 1320{
1227 dREQ; 1321 dREQ;
1228 1322
1229 req->type = REQ_OPEN; 1323 req->type = REQ_OPEN;
1230 req->data = newSVsv (pathname); 1324 req->sv1 = newSVsv (pathname);
1231 req->dataptr = SvPVbyte_nolen (req->data); 1325 req->ptr1 = SvPVbyte_nolen (pathname);
1232 req->fd = flags; 1326 req->int1 = flags;
1233 req->mode = mode; 1327 req->mode = mode;
1234 1328
1235 REQ_SEND; 1329 REQ_SEND;
1236} 1330}
1237 1331
1247 PPCODE: 1341 PPCODE:
1248{ 1342{
1249 dREQ; 1343 dREQ;
1250 1344
1251 req->type = ix; 1345 req->type = ix;
1252 req->fh = newSVsv (fh); 1346 req->fh = newSVsv (fh);
1253 req->fd = PerlIO_fileno (IoIFP (sv_2io (fh))); 1347 req->int1 = PerlIO_fileno (IoIFP (sv_2io (fh)));
1254 1348
1255 REQ_SEND (req); 1349 REQ_SEND (req);
1256} 1350}
1257 1351
1258void 1352void
1267 aio_read = REQ_READ 1361 aio_read = REQ_READ
1268 aio_write = REQ_WRITE 1362 aio_write = REQ_WRITE
1269 PROTOTYPE: $$$$$;$ 1363 PROTOTYPE: $$$$$;$
1270 PPCODE: 1364 PPCODE:
1271{ 1365{
1272 aio_req req;
1273 STRLEN svlen; 1366 STRLEN svlen;
1274 char *svptr = SvPVbyte (data, svlen); 1367 char *svptr = SvPVbyte (data, svlen);
1275 1368
1276 SvUPGRADE (data, SVt_PV); 1369 SvUPGRADE (data, SVt_PV);
1277 SvPOK_on (data); 1370 SvPOK_on (data);
1299 1392
1300 { 1393 {
1301 dREQ; 1394 dREQ;
1302 1395
1303 req->type = ix; 1396 req->type = ix;
1304 req->fh = newSVsv (fh); 1397 req->fh = newSVsv (fh);
1305 req->fd = PerlIO_fileno (ix == REQ_READ ? IoIFP (sv_2io (fh)) 1398 req->int1 = PerlIO_fileno (ix == REQ_READ ? IoIFP (sv_2io (fh))
1306 : IoOFP (sv_2io (fh))); 1399 : IoOFP (sv_2io (fh)));
1307 req->offset = offset; 1400 req->offs = offset;
1308 req->length = length; 1401 req->size = length;
1309 req->data = SvREFCNT_inc (data); 1402 req->sv1 = SvREFCNT_inc (data);
1310 req->dataptr = (char *)svptr + dataoffset; 1403 req->ptr1 = (char *)svptr + dataoffset;
1404 req->stroffset = dataoffset;
1311 1405
1312 if (!SvREADONLY (data)) 1406 if (!SvREADONLY (data))
1313 { 1407 {
1314 SvREADONLY_on (data); 1408 SvREADONLY_on (data);
1315 req->data2ptr = (void *)data; 1409 req->flags |= FLAG_SV1_RO_OFF;
1316 } 1410 }
1317 1411
1318 REQ_SEND; 1412 REQ_SEND;
1319 } 1413 }
1414}
1415
1416void
1417aio_readlink (path,callback=&PL_sv_undef)
1418 SV * path
1419 SV * callback
1420 PROTOTYPE: $$;$
1421 PPCODE:
1422{
1423 SV *data;
1424 dREQ;
1425
1426 data = newSV (NAME_MAX);
1427 SvPOK_on (data);
1428
1429 req->type = REQ_READLINK;
1430 req->fh = newSVsv (path);
1431 req->ptr2 = SvPVbyte_nolen (path);
1432 req->sv1 = data;
1433 req->ptr1 = SvPVbyte_nolen (data);
1434
1435 REQ_SEND;
1320} 1436}
1321 1437
1322void 1438void
1323aio_sendfile (out_fh,in_fh,in_offset,length,callback=&PL_sv_undef) 1439aio_sendfile (out_fh,in_fh,in_offset,length,callback=&PL_sv_undef)
1324 SV * out_fh 1440 SV * out_fh
1330 PPCODE: 1446 PPCODE:
1331{ 1447{
1332 dREQ; 1448 dREQ;
1333 1449
1334 req->type = REQ_SENDFILE; 1450 req->type = REQ_SENDFILE;
1335 req->fh = newSVsv (out_fh); 1451 req->fh = newSVsv (out_fh);
1336 req->fd = PerlIO_fileno (IoIFP (sv_2io (out_fh))); 1452 req->int1 = PerlIO_fileno (IoIFP (sv_2io (out_fh)));
1337 req->fh2 = newSVsv (in_fh); 1453 req->sv2 = newSVsv (in_fh);
1338 req->fd2 = PerlIO_fileno (IoIFP (sv_2io (in_fh))); 1454 req->int2 = PerlIO_fileno (IoIFP (sv_2io (in_fh)));
1339 req->offset = in_offset; 1455 req->offs = in_offset;
1340 req->length = length; 1456 req->size = length;
1341 1457
1342 REQ_SEND; 1458 REQ_SEND;
1343} 1459}
1344 1460
1345void 1461void
1352 PPCODE: 1468 PPCODE:
1353{ 1469{
1354 dREQ; 1470 dREQ;
1355 1471
1356 req->type = REQ_READAHEAD; 1472 req->type = REQ_READAHEAD;
1357 req->fh = newSVsv (fh); 1473 req->fh = newSVsv (fh);
1358 req->fd = PerlIO_fileno (IoIFP (sv_2io (fh))); 1474 req->int1 = PerlIO_fileno (IoIFP (sv_2io (fh)));
1359 req->offset = offset; 1475 req->offs = offset;
1360 req->length = length; 1476 req->size = length;
1361 1477
1362 REQ_SEND; 1478 REQ_SEND;
1363} 1479}
1364 1480
1365void 1481void
1371 aio_lstat = REQ_LSTAT 1487 aio_lstat = REQ_LSTAT
1372 PPCODE: 1488 PPCODE:
1373{ 1489{
1374 dREQ; 1490 dREQ;
1375 1491
1376 New (0, req->statdata, 1, Stat_t); 1492 req->ptr2 = malloc (sizeof (Stat_t));
1377 if (!req->statdata) 1493 if (!req->ptr2)
1378 { 1494 {
1379 req_free (req); 1495 req_free (req);
1380 croak ("out of memory during aio_req->statdata allocation"); 1496 croak ("out of memory during aio_stat statdata allocation");
1381 } 1497 }
1498
1499 req->flags |= FLAG_PTR2_FREE;
1382 1500
1383 if (SvPOK (fh_or_path)) 1501 if (SvPOK (fh_or_path))
1384 { 1502 {
1385 req->type = ix; 1503 req->type = ix;
1386 req->data = newSVsv (fh_or_path); 1504 req->sv1 = newSVsv (fh_or_path);
1387 req->dataptr = SvPVbyte_nolen (req->data); 1505 req->ptr1 = SvPVbyte_nolen (fh_or_path);
1388 } 1506 }
1389 else 1507 else
1390 { 1508 {
1391 req->type = REQ_FSTAT; 1509 req->type = REQ_FSTAT;
1392 req->fh = newSVsv (fh_or_path); 1510 req->fh = newSVsv (fh_or_path);
1393 req->fd = PerlIO_fileno (IoIFP (sv_2io (fh_or_path))); 1511 req->int1 = PerlIO_fileno (IoIFP (sv_2io (fh_or_path)));
1394 } 1512 }
1395 1513
1396 REQ_SEND; 1514 REQ_SEND;
1397} 1515}
1398 1516
1399void 1517void
1400aio_unlink (pathname,callback=&PL_sv_undef) 1518aio_unlink (pathname,callback=&PL_sv_undef)
1401 SV * pathname 1519 SV * pathname
1402 SV * callback 1520 SV * callback
1403 ALIAS: 1521 ALIAS:
1404 aio_unlink = REQ_UNLINK 1522 aio_unlink = REQ_UNLINK
1405 aio_rmdir = REQ_RMDIR 1523 aio_rmdir = REQ_RMDIR
1406 aio_readdir = REQ_READDIR 1524 aio_readdir = REQ_READDIR
1407 PPCODE: 1525 PPCODE:
1408{ 1526{
1409 dREQ; 1527 dREQ;
1410 1528
1411 req->type = ix; 1529 req->type = ix;
1412 req->data = newSVsv (pathname); 1530 req->sv1 = newSVsv (pathname);
1413 req->dataptr = SvPVbyte_nolen (req->data); 1531 req->ptr1 = SvPVbyte_nolen (pathname);
1414 1532
1415 REQ_SEND; 1533 REQ_SEND;
1416} 1534}
1417 1535
1418void 1536void
1419aio_link (oldpath,newpath,callback=&PL_sv_undef) 1537aio_link (oldpath,newpath,callback=&PL_sv_undef)
1420 SV * oldpath 1538 SV * oldpath
1421 SV * newpath 1539 SV * newpath
1422 SV * callback 1540 SV * callback
1423 ALIAS: 1541 ALIAS:
1424 aio_link = REQ_LINK 1542 aio_link = REQ_LINK
1425 aio_symlink = REQ_SYMLINK 1543 aio_symlink = REQ_SYMLINK
1426 aio_rename = REQ_RENAME 1544 aio_rename = REQ_RENAME
1427 PPCODE: 1545 PPCODE:
1428{ 1546{
1429 dREQ; 1547 dREQ;
1430 1548
1431 req->type = ix; 1549 req->type = ix;
1432 req->fh = newSVsv (oldpath); 1550 req->fh = newSVsv (oldpath);
1433 req->data2ptr = SvPVbyte_nolen (req->fh); 1551 req->ptr2 = SvPVbyte_nolen (req->fh);
1434 req->data = newSVsv (newpath); 1552 req->sv1 = newSVsv (newpath);
1435 req->dataptr = SvPVbyte_nolen (req->data); 1553 req->ptr1 = SvPVbyte_nolen (newpath);
1436 1554
1437 REQ_SEND; 1555 REQ_SEND;
1438} 1556}
1439 1557
1440void 1558void
1441aio_mknod (pathname,mode,dev,callback=&PL_sv_undef) 1559aio_mknod (pathname,mode,dev,callback=&PL_sv_undef)
1442 SV * pathname 1560 SV * pathname
1443 SV * callback 1561 SV * callback
1444 UV mode 1562 UV mode
1445 UV dev 1563 UV dev
1446 PPCODE: 1564 PPCODE:
1447{ 1565{
1448 dREQ; 1566 dREQ;
1449 1567
1450 req->type = REQ_MKNOD; 1568 req->type = REQ_MKNOD;
1451 req->data = newSVsv (pathname); 1569 req->sv1 = newSVsv (pathname);
1452 req->dataptr = SvPVbyte_nolen (req->data); 1570 req->ptr1 = SvPVbyte_nolen (pathname);
1453 req->mode = (mode_t)mode; 1571 req->mode = (mode_t)mode;
1454 req->offset = dev; 1572 req->offs = dev;
1455 1573
1456 REQ_SEND; 1574 REQ_SEND;
1457} 1575}
1458 1576
1459void 1577void
1460aio_busy (delay,callback=&PL_sv_undef) 1578aio_busy (delay,callback=&PL_sv_undef)
1461 double delay 1579 double delay
1462 SV * callback 1580 SV * callback
1463 PPCODE: 1581 PPCODE:
1464{ 1582{
1465 dREQ; 1583 dREQ;
1466 1584
1467 req->type = REQ_BUSY; 1585 req->type = REQ_BUSY;
1468 req->fd = delay < 0. ? 0 : delay; 1586 req->int1 = delay < 0. ? 0 : delay;
1469 req->fd2 = delay < 0. ? 0 : 1000. * (delay - req->fd); 1587 req->int2 = delay < 0. ? 0 : 1000. * (delay - req->int1);
1470 1588
1471 REQ_SEND; 1589 REQ_SEND;
1472} 1590}
1473 1591
1474void 1592void
1478 PPCODE: 1596 PPCODE:
1479{ 1597{
1480 dREQ; 1598 dREQ;
1481 1599
1482 req->type = REQ_GROUP; 1600 req->type = REQ_GROUP;
1601
1483 req_send (req); 1602 req_send (req);
1484
1485 XPUSHs (req_sv (req, AIO_GRP_KLASS)); 1603 XPUSHs (req_sv (req, AIO_GRP_KLASS));
1486} 1604}
1487 1605
1488void 1606void
1489aio_nop (callback=&PL_sv_undef) 1607aio_nop (callback=&PL_sv_undef)
1490 SV * callback 1608 SV * callback
1491 PPCODE: 1609 PPCODE:
1492{ 1610{
1493 dREQ; 1611 dREQ;
1494 1612
1495 req->type = REQ_NOP; 1613 req->type = REQ_NOP;
1549 1667
1550int 1668int
1551poll_cb(...) 1669poll_cb(...)
1552 PROTOTYPE: 1670 PROTOTYPE:
1553 CODE: 1671 CODE:
1554 RETVAL = poll_cb (0); 1672 RETVAL = poll_cb ();
1555 OUTPUT:
1556 RETVAL
1557
1558int
1559poll_some(int max = 0)
1560 PROTOTYPE: $
1561 CODE:
1562 RETVAL = poll_cb (max);
1563 OUTPUT: 1673 OUTPUT:
1564 RETVAL 1674 RETVAL
1565 1675
1566void 1676void
1567poll_wait() 1677poll_wait()
1592 CODE: 1702 CODE:
1593 RETVAL = get_npending (); 1703 RETVAL = get_npending ();
1594 OUTPUT: 1704 OUTPUT:
1595 RETVAL 1705 RETVAL
1596 1706
1707int
1708nthreads()
1709 PROTOTYPE:
1710 CODE:
1711 if (WORDACCESS_UNSAFE) LOCK (wrklock);
1712 RETVAL = started;
1713 if (WORDACCESS_UNSAFE) UNLOCK (wrklock);
1714 OUTPUT:
1715 RETVAL
1716
1597PROTOTYPES: DISABLE 1717PROTOTYPES: DISABLE
1598 1718
1599MODULE = IO::AIO PACKAGE = IO::AIO::REQ 1719MODULE = IO::AIO PACKAGE = IO::AIO::REQ
1600 1720
1601void 1721void
1616 PPCODE: 1736 PPCODE:
1617{ 1737{
1618 int i; 1738 int i;
1619 aio_req req; 1739 aio_req req;
1620 1740
1621 if (grp->fd == 2) 1741 if (grp->int1 == 2)
1622 croak ("cannot add requests to IO::AIO::GRP after the group finished"); 1742 croak ("cannot add requests to IO::AIO::GRP after the group finished");
1623 1743
1624 for (i = 1; i < items; ++i ) 1744 for (i = 1; i < items; ++i )
1625 { 1745 {
1626 if (GIMME_V != G_VOID) 1746 if (GIMME_V != G_VOID)
1628 1748
1629 req = SvAIO_REQ (ST (i)); 1749 req = SvAIO_REQ (ST (i));
1630 1750
1631 if (req) 1751 if (req)
1632 { 1752 {
1633 ++grp->length; 1753 ++grp->size;
1634 req->grp = grp; 1754 req->grp = grp;
1635 1755
1636 req->grp_prev = 0; 1756 req->grp_prev = 0;
1637 req->grp_next = grp->grp_first; 1757 req->grp_next = grp->grp_first;
1638 1758
1661 av = newAV (); 1781 av = newAV ();
1662 1782
1663 for (i = 1; i < items; ++i ) 1783 for (i = 1; i < items; ++i )
1664 av_push (av, newSVsv (ST (i))); 1784 av_push (av, newSVsv (ST (i)));
1665 1785
1666 SvREFCNT_dec (grp->data); 1786 SvREFCNT_dec (grp->sv1);
1667 grp->data = (SV *)av; 1787 grp->sv1 = (SV *)av;
1668} 1788}
1669 1789
1670void 1790void
1671errno (aio_req grp, int errorno = errno) 1791errno (aio_req grp, int errorno = errno)
1672 CODE: 1792 CODE:
1673 grp->errorno = errorno; 1793 grp->errorno = errorno;
1674 1794
1675void 1795void
1676limit (aio_req grp, int limit) 1796limit (aio_req grp, int limit)
1677 CODE: 1797 CODE:
1678 grp->fd2 = limit; 1798 grp->int2 = limit;
1679 aio_grp_feed (grp); 1799 aio_grp_feed (grp);
1680 1800
1681void 1801void
1682feed (aio_req grp, SV *callback=&PL_sv_undef) 1802feed (aio_req grp, SV *callback=&PL_sv_undef)
1683 CODE: 1803 CODE:
1684{ 1804{
1685 SvREFCNT_dec (grp->fh2); 1805 SvREFCNT_dec (grp->sv2);
1686 grp->fh2 = newSVsv (callback); 1806 grp->sv2 = newSVsv (callback);
1687 1807
1688 if (grp->fd2 <= 0) 1808 if (grp->int2 <= 0)
1689 grp->fd2 = 2; 1809 grp->int2 = 2;
1690 1810
1691 aio_grp_feed (grp); 1811 aio_grp_feed (grp);
1692} 1812}
1693 1813

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines