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.115 by root, Sat May 10 18:06:41 2008 UTC vs.
Revision 1.126 by root, Thu Sep 25 07:56:07 2008 UTC

12#include <sys/types.h> 12#include <sys/types.h>
13#include <sys/stat.h> 13#include <sys/stat.h>
14#include <limits.h> 14#include <limits.h>
15#include <fcntl.h> 15#include <fcntl.h>
16#include <sched.h> 16#include <sched.h>
17
18/* perl namespace pollution */
19#undef VERSION
17 20
18#ifdef _WIN32 21#ifdef _WIN32
19 22
20# define EIO_STRUCT_DIRENT Direntry_t 23# define EIO_STRUCT_DIRENT Direntry_t
21# undef malloc 24# undef malloc
62# include <signal.h> 65# include <signal.h>
63# define EIO_STRUCT_DIRENT struct dirent 66# define EIO_STRUCT_DIRENT struct dirent
64 67
65#endif 68#endif
66 69
70/* perl stupidly overrides readdir and maybe others */
71/* with thread-unsafe versions, imagine that :( */
72#undef readdir
73#undef opendir
74#undef closedir
75
67#define EIO_STRUCT_STAT Stat_t 76#define EIO_STRUCT_STAT Stat_t
68 77
69/* use NV for 32 bit perls as it allows larger offsets */ 78/* use NV for 32 bit perls as it allows larger offsets */
70#if IVSIZE >= 8 79#if IVSIZE >= 8
71# define SvVAL64 SvIV 80# define SvVAL64 SvIV
77typedef SV SV8; /* byte-sv, used for argument-checking */ 86typedef SV SV8; /* byte-sv, used for argument-checking */
78 87
79#define AIO_REQ_KLASS "IO::AIO::REQ" 88#define AIO_REQ_KLASS "IO::AIO::REQ"
80#define AIO_GRP_KLASS "IO::AIO::GRP" 89#define AIO_GRP_KLASS "IO::AIO::GRP"
81 90
82#define EIO_COMMON \ 91#define EIO_REQ_MEMBERS \
83 SV *callback; \ 92 SV *callback; \
84 SV *sv1, *sv2; \ 93 SV *sv1, *sv2; \
85 STRLEN stroffset; \ 94 STRLEN stroffset; \
86 SV *self 95 SV *self;
96
97#define EIO_NO_WRAPPERS 1
87 98
88#include "libeio/eio.h" 99#include "libeio/eio.h"
89 100
90static int req_invoke (eio_req *req); 101static int req_invoke (eio_req *req);
91#define EIO_FINISH(req) req_invoke (req) 102#define EIO_FINISH(req) req_invoke (req)
101#include "libeio/eio.c" 112#include "libeio/eio.c"
102 113
103typedef eio_req *aio_req; 114typedef eio_req *aio_req;
104typedef eio_req *aio_req_ornot; 115typedef eio_req *aio_req_ornot;
105 116
106static unsigned int max_outstanding = 0x7fffffff; 117static SV *on_next_submit;
107 118static int next_pri = EIO_PRI_DEFAULT;
108static int next_pri = EIO_DEFAULT_PRI + EIO_PRI_BIAS; 119static int max_outstanding;
109 120
110static int respipe_osf [2], respipe [2] = { -1, -1 }; 121static int respipe_osf [2], respipe [2] = { -1, -1 };
111 122
112static void req_destroy (aio_req req); 123static void req_destroy (aio_req req);
113static void req_cancel (aio_req req); 124static void req_cancel (aio_req req);
166 FREETMPS; 177 FREETMPS;
167 LEAVE; 178 LEAVE;
168 } 179 }
169} 180}
170 181
182static void req_submit (eio_req *req)
183{
184 eio_submit (req);
185
186 if (on_next_submit)
187 {
188 dSP;
189 SV *cb = sv_2mortal (on_next_submit);
190
191 on_next_submit = 0;
192
193 PUSHMARK (SP);
194 PUTBACK;
195 call_sv (cb, G_DISCARD | G_EVAL);
196 }
197}
198
171static int req_invoke (eio_req *req) 199static int req_invoke (eio_req *req)
172{ 200{
173 dSP; 201 dSP;
174 202
175 if (req->flags & FLAG_SV2_RO_OFF) 203 if (req->flags & FLAG_SV2_RO_OFF)
260 case EIO_BUSY: 288 case EIO_BUSY:
261 break; 289 break;
262 290
263 case EIO_READLINK: 291 case EIO_READLINK:
264 if (req->result > 0) 292 if (req->result > 0)
265 { 293 PUSHs (sv_2mortal (newSVpvn (req->ptr2, req->result)));
266 SvCUR_set (req->sv2, req->result);
267 *SvEND (req->sv2) = 0;
268 PUSHs (req->sv2);
269 }
270 break; 294 break;
271 295
272 case EIO_STAT: 296 case EIO_STAT:
273 case EIO_LSTAT: 297 case EIO_LSTAT:
274 case EIO_FSTAT: 298 case EIO_FSTAT:
394 X_UNLOCK (reslock); 418 X_UNLOCK (reslock);
395 419
396 if (size) 420 if (size)
397 return; 421 return;
398 422
399 maybe_start_thread (); 423 etp_maybe_start_thread ();
400 424
401 FD_ZERO (&rfd); 425 FD_ZERO (&rfd);
402 FD_SET (respipe [0], &rfd); 426 FD_SET (respipe [0], &rfd);
403 427
404 PerlSock_select (respipe [0] + 1, &rfd, 0, 0, 0); 428 PerlSock_select (respipe [0] + 1, &rfd, 0, 0, 0);
405 } 429 }
406} 430}
407 431
408static int poll_cb (void) 432static int poll_cb (void)
409{ 433{
434 for (;;)
435 {
410 int res = eio_poll (); 436 int res = eio_poll ();
411 437
412 if (res > 0) 438 if (res > 0)
413 croak (0); 439 croak (0);
414 440
441 if (!max_outstanding || max_outstanding > eio_nreqs ())
415 return res; 442 return res;
443
444 poll_wait ();
445 }
446}
447
448static void atfork_child (void)
449{
450 create_respipe ();
416} 451}
417 452
418#define dREQ \ 453#define dREQ \
419 aio_req req; \ 454 aio_req req; \
420 int req_pri = next_pri; \ 455 int req_pri = next_pri; \
421 next_pri = EIO_DEFAULT_PRI + EIO_PRI_BIAS; \ 456 next_pri = EIO_PRI_DEFAULT; \
422 \ 457 \
423 if (SvOK (callback) && !SvROK (callback)) \ 458 if (SvOK (callback) && !SvROK (callback)) \
424 croak ("callback must be undef or of reference type"); \ 459 croak ("callback must be undef or of reference type"); \
425 \ 460 \
426 Newz (0, req, 1, eio_req); \ 461 Newz (0, req, 1, eio_req); \
429 \ 464 \
430 req->callback = newSVsv (callback); \ 465 req->callback = newSVsv (callback); \
431 req->pri = req_pri 466 req->pri = req_pri
432 467
433#define REQ_SEND \ 468#define REQ_SEND \
469 PUTBACK; \
434 eio_submit (req); \ 470 req_submit (req); \
471 SPAGAIN; \
435 \ 472 \
436 if (GIMME_V != G_VOID) \ 473 if (GIMME_V != G_VOID) \
437 XPUSHs (req_sv (req, AIO_REQ_KLASS)); 474 XPUSHs (req_sv (req, AIO_REQ_KLASS));
438 475
439MODULE = IO::AIO PACKAGE = IO::AIO 476MODULE = IO::AIO PACKAGE = IO::AIO
447 newCONSTSUB (stash, "EXDEV", newSViv (EXDEV)); 484 newCONSTSUB (stash, "EXDEV", newSViv (EXDEV));
448 newCONSTSUB (stash, "O_RDONLY", newSViv (O_RDONLY)); 485 newCONSTSUB (stash, "O_RDONLY", newSViv (O_RDONLY));
449 newCONSTSUB (stash, "O_WRONLY", newSViv (O_WRONLY)); 486 newCONSTSUB (stash, "O_WRONLY", newSViv (O_WRONLY));
450 newCONSTSUB (stash, "O_CREAT", newSViv (O_CREAT)); 487 newCONSTSUB (stash, "O_CREAT", newSViv (O_CREAT));
451 newCONSTSUB (stash, "O_TRUNC", newSViv (O_TRUNC)); 488 newCONSTSUB (stash, "O_TRUNC", newSViv (O_TRUNC));
452#ifdef _WIN32 489#ifndef _WIN32
453 X_MUTEX_CHECK (wrklock);
454 X_MUTEX_CHECK (reslock);
455 X_MUTEX_CHECK (reqlock);
456 X_MUTEX_CHECK (reqwait);
457 X_MUTEX_CHECK (preadwritelock);
458 X_MUTEX_CHECK (readdirlock);
459
460 X_COND_CHECK (reqwait);
461#else
462 newCONSTSUB (stash, "S_IFIFO", newSViv (S_IFIFO)); 490 newCONSTSUB (stash, "S_IFIFO", newSViv (S_IFIFO));
463#endif 491#endif
464 492
465 create_respipe (); 493 create_respipe ();
466 494
467 if (eio_init (want_poll, done_poll) < 0) 495 if (eio_init (want_poll, done_poll) < 0)
468 croak ("IO::AIO: unable to initialise eio library"); 496 croak ("IO::AIO: unable to initialise eio library");
497
498 /* atfork child called in fifo order, so before eio's handler */
499 X_THREAD_ATFORK (0, 0, atfork_child);
469} 500}
470 501
471void 502void
472max_poll_reqs (int nreqs) 503max_poll_reqs (int nreqs)
473 PROTOTYPE: $ 504 PROTOTYPE: $
630 PPCODE: 661 PPCODE:
631{ 662{
632 SV *data; 663 SV *data;
633 dREQ; 664 dREQ;
634 665
635 data = newSV (NAME_MAX);
636 SvPOK_on (data);
637
638 req->type = EIO_READLINK; 666 req->type = EIO_READLINK;
639 req->sv1 = newSVsv (path); 667 req->sv1 = newSVsv (path);
640 req->ptr1 = SvPVbyte_nolen (req->sv1); 668 req->ptr1 = SvPVbyte_nolen (req->sv1);
641 req->sv2 = data;
642 req->ptr2 = SvPVbyte_nolen (data);
643 669
644 REQ_SEND; 670 REQ_SEND;
645} 671}
646 672
647void 673void
685 aio_lstat = EIO_LSTAT 711 aio_lstat = EIO_LSTAT
686 PPCODE: 712 PPCODE:
687{ 713{
688 dREQ; 714 dREQ;
689 715
690 req->ptr2 = malloc (sizeof (EIO_STRUCT_STAT));
691 if (!req->ptr2)
692 {
693 req_destroy (req);
694 croak ("out of memory during aio_stat statdata allocation");
695 }
696
697 req->flags |= EIO_FLAG_PTR2_FREE;
698 req->sv1 = newSVsv (fh_or_path); 716 req->sv1 = newSVsv (fh_or_path);
699 717
700 if (SvPOK (fh_or_path)) 718 if (SvPOK (req->sv1))
701 { 719 {
702 req->type = ix; 720 req->type = ix;
703 req->ptr1 = SvPVbyte_nolen (req->sv1); 721 req->ptr1 = SvPVbyte_nolen (req->sv1);
704 } 722 }
705 else 723 else
719 737
720 req->nv1 = SvOK (atime) ? SvNV (atime) : -1.; 738 req->nv1 = SvOK (atime) ? SvNV (atime) : -1.;
721 req->nv2 = SvOK (mtime) ? SvNV (mtime) : -1.; 739 req->nv2 = SvOK (mtime) ? SvNV (mtime) : -1.;
722 req->sv1 = newSVsv (fh_or_path); 740 req->sv1 = newSVsv (fh_or_path);
723 741
724 if (SvPOK (fh_or_path)) 742 if (SvPOK (req->sv1))
725 { 743 {
726 req->type = EIO_UTIME; 744 req->type = EIO_UTIME;
727 req->ptr1 = SvPVbyte_nolen (req->sv1); 745 req->ptr1 = SvPVbyte_nolen (req->sv1);
728 } 746 }
729 else 747 else
742 dREQ; 760 dREQ;
743 761
744 req->sv1 = newSVsv (fh_or_path); 762 req->sv1 = newSVsv (fh_or_path);
745 req->offs = SvOK (offset) ? SvVAL64 (offset) : -1; 763 req->offs = SvOK (offset) ? SvVAL64 (offset) : -1;
746 764
747 if (SvPOK (fh_or_path)) 765 if (SvPOK (req->sv1))
748 { 766 {
749 req->type = EIO_TRUNCATE; 767 req->type = EIO_TRUNCATE;
750 req->ptr1 = SvPVbyte_nolen (req->sv1); 768 req->ptr1 = SvPVbyte_nolen (req->sv1);
751 } 769 }
752 else 770 else
760 778
761void 779void
762aio_chmod (SV8 *fh_or_path, int mode, SV *callback=&PL_sv_undef) 780aio_chmod (SV8 *fh_or_path, int mode, SV *callback=&PL_sv_undef)
763 ALIAS: 781 ALIAS:
764 aio_chmod = EIO_CHMOD 782 aio_chmod = EIO_CHMOD
765 aio_fchmod = EIO_FCHMOD
766 aio_mkdir = EIO_MKDIR 783 aio_mkdir = EIO_MKDIR
767 PPCODE: 784 PPCODE:
768{ 785{
769 dREQ; 786 dREQ;
770 787
771 req->type = ix;
772 req->int2 = mode; 788 req->int2 = mode;
773 req->sv1 = newSVsv (fh_or_path); 789 req->sv1 = newSVsv (fh_or_path);
774 790
775 if (ix == EIO_FCHMOD) 791 if (SvPOK (req->sv1))
776 req->int1 = PerlIO_fileno (IoIFP (sv_2io (fh_or_path))); 792 {
793 req->type = ix;
794 req->ptr1 = SvPVbyte_nolen (req->sv1);
795 }
777 else 796 else
778 req->ptr1 = SvPVbyte_nolen (req->sv1); 797 {
798 req->type = EIO_FCHMOD;
799 req->int1 = PerlIO_fileno (IoIFP (sv_2io (fh_or_path)));
800 }
779 801
780 REQ_SEND; 802 REQ_SEND;
781} 803}
782 804
783void 805void
788 810
789 req->int2 = SvOK (uid) ? SvIV (uid) : -1; 811 req->int2 = SvOK (uid) ? SvIV (uid) : -1;
790 req->int3 = SvOK (gid) ? SvIV (gid) : -1; 812 req->int3 = SvOK (gid) ? SvIV (gid) : -1;
791 req->sv1 = newSVsv (fh_or_path); 813 req->sv1 = newSVsv (fh_or_path);
792 814
793 if (SvPOK (fh_or_path)) 815 if (SvPOK (req->sv1))
794 { 816 {
795 req->type = EIO_CHOWN; 817 req->type = EIO_CHOWN;
796 req->ptr1 = SvPVbyte_nolen (req->sv1); 818 req->ptr1 = SvPVbyte_nolen (req->sv1);
797 } 819 }
798 else 820 else
896 918
897int 919int
898aioreq_pri (int pri = 0) 920aioreq_pri (int pri = 0)
899 PROTOTYPE: ;$ 921 PROTOTYPE: ;$
900 CODE: 922 CODE:
901 RETVAL = next_pri - EIO_PRI_BIAS; 923 RETVAL = next_pri;
902 if (items > 0) 924 if (items > 0)
903 { 925 {
904 if (pri < EIO_PRI_MIN) pri = EIO_PRI_MIN; 926 if (pri < EIO_PRI_MIN) pri = EIO_PRI_MIN;
905 if (pri > EIO_PRI_MAX) pri = EIO_PRI_MAX; 927 if (pri > EIO_PRI_MAX) pri = EIO_PRI_MAX;
906 next_pri = pri + EIO_PRI_BIAS; 928 next_pri = pri;
907 } 929 }
908 OUTPUT: 930 OUTPUT:
909 RETVAL 931 RETVAL
910 932
911void 933void
912aioreq_nice (int nice = 0) 934aioreq_nice (int nice = 0)
913 CODE: 935 CODE:
914 nice = next_pri - nice; 936 nice = next_pri - nice;
915 if (nice < EIO_PRI_MIN) nice = EIO_PRI_MIN; 937 if (nice < EIO_PRI_MIN) nice = EIO_PRI_MIN;
916 if (nice > EIO_PRI_MAX) nice = EIO_PRI_MAX; 938 if (nice > EIO_PRI_MAX) nice = EIO_PRI_MAX;
917 next_pri = nice + EIO_PRI_BIAS; 939 next_pri = nice;
918 940
919void 941void
920flush () 942flush ()
921 PROTOTYPE: 943 PROTOTYPE:
922 CODE: 944 CODE:
923 while (nreqs) 945 while (eio_nreqs ())
924 { 946 {
925 poll_wait (); 947 poll_wait ();
926 poll_cb (); 948 poll_cb ();
927 } 949 }
928 950
983 1005
984int 1006int
985nthreads() 1007nthreads()
986 PROTOTYPE: 1008 PROTOTYPE:
987 CODE: 1009 CODE:
988 if (WORDACCESS_UNSAFE) X_LOCK (wrklock); 1010 RETVAL = eio_nthreads ();
989 RETVAL = started;
990 if (WORDACCESS_UNSAFE) X_UNLOCK (wrklock);
991 OUTPUT: 1011 OUTPUT:
992 RETVAL 1012 RETVAL
1013
1014void _on_next_submit (SV *cb)
1015 CODE:
1016 SvREFCNT_dec (on_next_submit);
1017 on_next_submit = SvOK (cb) ? newSVsv (cb) : 0;
993 1018
994PROTOTYPES: DISABLE 1019PROTOTYPES: DISABLE
995 1020
996MODULE = IO::AIO PACKAGE = IO::AIO::REQ 1021MODULE = IO::AIO PACKAGE = IO::AIO::REQ
997 1022

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines