… | |
… | |
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 |
72 | #else |
81 | #else |
73 | # define SvVAL64 SvNV |
82 | # define SvVAL64 SvNV |
74 | #endif |
83 | #endif |
75 | |
84 | |
|
|
85 | /*****************************************************************************/ |
|
|
86 | |
|
|
87 | #if __GNUC__ >= 3 |
|
|
88 | # define expect(expr,value) __builtin_expect ((expr),(value)) |
|
|
89 | #else |
|
|
90 | # define expect(expr,value) (expr) |
|
|
91 | #endif |
|
|
92 | |
|
|
93 | #define expect_false(expr) expect ((expr) != 0, 0) |
|
|
94 | #define expect_true(expr) expect ((expr) != 0, 1) |
|
|
95 | |
|
|
96 | /*****************************************************************************/ |
|
|
97 | |
76 | static HV *stash; |
98 | static HV *stash; |
77 | typedef SV SV8; /* byte-sv, used for argument-checking */ |
99 | typedef SV SV8; /* byte-sv, used for argument-checking */ |
78 | |
100 | |
79 | #define AIO_REQ_KLASS "IO::AIO::REQ" |
101 | #define AIO_REQ_KLASS "IO::AIO::REQ" |
80 | #define AIO_GRP_KLASS "IO::AIO::GRP" |
102 | #define AIO_GRP_KLASS "IO::AIO::GRP" |
81 | |
103 | |
82 | #define EIO_COMMON \ |
104 | #define EIO_REQ_MEMBERS \ |
83 | SV *callback; \ |
105 | SV *callback; \ |
84 | SV *sv1, *sv2; \ |
106 | SV *sv1, *sv2; \ |
85 | STRLEN stroffset; \ |
107 | STRLEN stroffset; \ |
86 | SV *self |
108 | SV *self; |
|
|
109 | |
|
|
110 | #define EIO_NO_WRAPPERS 1 |
87 | |
111 | |
88 | #include "libeio/eio.h" |
112 | #include "libeio/eio.h" |
89 | |
113 | |
90 | static int req_invoke (eio_req *req); |
114 | static int req_invoke (eio_req *req); |
91 | #define EIO_FINISH(req) req_invoke (req) |
115 | #define EIO_FINISH(req) req_invoke (req) |
… | |
… | |
101 | #include "libeio/eio.c" |
125 | #include "libeio/eio.c" |
102 | |
126 | |
103 | typedef eio_req *aio_req; |
127 | typedef eio_req *aio_req; |
104 | typedef eio_req *aio_req_ornot; |
128 | typedef eio_req *aio_req_ornot; |
105 | |
129 | |
106 | static unsigned int max_outstanding = 0x7fffffff; |
130 | static SV *on_next_submit; |
107 | |
131 | static int next_pri = EIO_PRI_DEFAULT; |
108 | static int next_pri = EIO_DEFAULT_PRI + EIO_PRI_BIAS; |
132 | static int max_outstanding; |
109 | |
133 | |
110 | static int respipe_osf [2], respipe [2] = { -1, -1 }; |
134 | static int respipe_osf [2], respipe [2] = { -1, -1 }; |
111 | |
135 | |
112 | static void req_destroy (aio_req req); |
136 | static void req_destroy (aio_req req); |
113 | static void req_cancel (aio_req req); |
137 | static void req_cancel (aio_req req); |
… | |
… | |
166 | FREETMPS; |
190 | FREETMPS; |
167 | LEAVE; |
191 | LEAVE; |
168 | } |
192 | } |
169 | } |
193 | } |
170 | |
194 | |
|
|
195 | static void req_submit (eio_req *req) |
|
|
196 | { |
|
|
197 | eio_submit (req); |
|
|
198 | |
|
|
199 | if (expect_false (on_next_submit)) |
|
|
200 | { |
|
|
201 | dSP; |
|
|
202 | SV *cb = sv_2mortal (on_next_submit); |
|
|
203 | |
|
|
204 | on_next_submit = 0; |
|
|
205 | |
|
|
206 | PUSHMARK (SP); |
|
|
207 | PUTBACK; |
|
|
208 | call_sv (cb, G_DISCARD | G_EVAL); |
|
|
209 | } |
|
|
210 | } |
|
|
211 | |
171 | static int req_invoke (eio_req *req) |
212 | static int req_invoke (eio_req *req) |
172 | { |
213 | { |
173 | dSP; |
214 | dSP; |
174 | |
215 | |
175 | if (req->flags & FLAG_SV2_RO_OFF) |
216 | if (req->flags & FLAG_SV2_RO_OFF) |
… | |
… | |
260 | case EIO_BUSY: |
301 | case EIO_BUSY: |
261 | break; |
302 | break; |
262 | |
303 | |
263 | case EIO_READLINK: |
304 | case EIO_READLINK: |
264 | if (req->result > 0) |
305 | if (req->result > 0) |
265 | { |
306 | 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; |
307 | break; |
271 | |
308 | |
272 | case EIO_STAT: |
309 | case EIO_STAT: |
273 | case EIO_LSTAT: |
310 | case EIO_LSTAT: |
274 | case EIO_FSTAT: |
311 | case EIO_FSTAT: |
… | |
… | |
394 | X_UNLOCK (reslock); |
431 | X_UNLOCK (reslock); |
395 | |
432 | |
396 | if (size) |
433 | if (size) |
397 | return; |
434 | return; |
398 | |
435 | |
399 | maybe_start_thread (); |
436 | etp_maybe_start_thread (); |
400 | |
437 | |
401 | FD_ZERO (&rfd); |
438 | FD_ZERO (&rfd); |
402 | FD_SET (respipe [0], &rfd); |
439 | FD_SET (respipe [0], &rfd); |
403 | |
440 | |
404 | PerlSock_select (respipe [0] + 1, &rfd, 0, 0, 0); |
441 | PerlSock_select (respipe [0] + 1, &rfd, 0, 0, 0); |
405 | } |
442 | } |
406 | } |
443 | } |
407 | |
444 | |
408 | static int poll_cb (void) |
445 | static int poll_cb (void) |
409 | { |
446 | { |
|
|
447 | for (;;) |
|
|
448 | { |
410 | int res = eio_poll (); |
449 | int res = eio_poll (); |
411 | |
450 | |
412 | if (res > 0) |
451 | if (res > 0) |
413 | croak (0); |
452 | croak (0); |
414 | |
453 | |
|
|
454 | if (!max_outstanding || max_outstanding > eio_nreqs ()) |
415 | return res; |
455 | return res; |
|
|
456 | |
|
|
457 | poll_wait (); |
|
|
458 | } |
|
|
459 | } |
|
|
460 | |
|
|
461 | static void atfork_child (void) |
|
|
462 | { |
|
|
463 | create_respipe (); |
416 | } |
464 | } |
417 | |
465 | |
418 | #define dREQ \ |
466 | #define dREQ \ |
419 | aio_req req; \ |
467 | aio_req req; \ |
420 | int req_pri = next_pri; \ |
468 | int req_pri = next_pri; \ |
421 | next_pri = EIO_DEFAULT_PRI + EIO_PRI_BIAS; \ |
469 | next_pri = EIO_PRI_DEFAULT; \ |
422 | \ |
470 | \ |
423 | if (SvOK (callback) && !SvROK (callback)) \ |
471 | if (SvOK (callback) && !SvROK (callback)) \ |
424 | croak ("callback must be undef or of reference type"); \ |
472 | croak ("callback must be undef or of reference type"); \ |
425 | \ |
473 | \ |
426 | Newz (0, req, 1, eio_req); \ |
474 | Newz (0, req, 1, eio_req); \ |
… | |
… | |
429 | \ |
477 | \ |
430 | req->callback = newSVsv (callback); \ |
478 | req->callback = newSVsv (callback); \ |
431 | req->pri = req_pri |
479 | req->pri = req_pri |
432 | |
480 | |
433 | #define REQ_SEND \ |
481 | #define REQ_SEND \ |
|
|
482 | PUTBACK; \ |
434 | eio_submit (req); \ |
483 | req_submit (req); \ |
|
|
484 | SPAGAIN; \ |
435 | \ |
485 | \ |
436 | if (GIMME_V != G_VOID) \ |
486 | if (GIMME_V != G_VOID) \ |
437 | XPUSHs (req_sv (req, AIO_REQ_KLASS)); |
487 | XPUSHs (req_sv (req, AIO_REQ_KLASS)); |
438 | |
488 | |
439 | MODULE = IO::AIO PACKAGE = IO::AIO |
489 | MODULE = IO::AIO PACKAGE = IO::AIO |
… | |
… | |
447 | newCONSTSUB (stash, "EXDEV", newSViv (EXDEV)); |
497 | newCONSTSUB (stash, "EXDEV", newSViv (EXDEV)); |
448 | newCONSTSUB (stash, "O_RDONLY", newSViv (O_RDONLY)); |
498 | newCONSTSUB (stash, "O_RDONLY", newSViv (O_RDONLY)); |
449 | newCONSTSUB (stash, "O_WRONLY", newSViv (O_WRONLY)); |
499 | newCONSTSUB (stash, "O_WRONLY", newSViv (O_WRONLY)); |
450 | newCONSTSUB (stash, "O_CREAT", newSViv (O_CREAT)); |
500 | newCONSTSUB (stash, "O_CREAT", newSViv (O_CREAT)); |
451 | newCONSTSUB (stash, "O_TRUNC", newSViv (O_TRUNC)); |
501 | newCONSTSUB (stash, "O_TRUNC", newSViv (O_TRUNC)); |
452 | #ifdef _WIN32 |
502 | #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)); |
503 | newCONSTSUB (stash, "S_IFIFO", newSViv (S_IFIFO)); |
463 | #endif |
504 | #endif |
464 | |
505 | |
465 | create_respipe (); |
506 | create_respipe (); |
466 | |
507 | |
467 | if (eio_init (want_poll, done_poll) < 0) |
508 | if (eio_init (want_poll, done_poll) < 0) |
468 | croak ("IO::AIO: unable to initialise eio library"); |
509 | croak ("IO::AIO: unable to initialise eio library"); |
|
|
510 | |
|
|
511 | /* atfork child called in fifo order, so before eio's handler */ |
|
|
512 | X_THREAD_ATFORK (0, 0, atfork_child); |
469 | } |
513 | } |
470 | |
514 | |
471 | void |
515 | void |
472 | max_poll_reqs (int nreqs) |
516 | max_poll_reqs (int nreqs) |
473 | PROTOTYPE: $ |
517 | PROTOTYPE: $ |
… | |
… | |
630 | PPCODE: |
674 | PPCODE: |
631 | { |
675 | { |
632 | SV *data; |
676 | SV *data; |
633 | dREQ; |
677 | dREQ; |
634 | |
678 | |
635 | data = newSV (NAME_MAX); |
|
|
636 | SvPOK_on (data); |
|
|
637 | |
|
|
638 | req->type = EIO_READLINK; |
679 | req->type = EIO_READLINK; |
639 | req->sv1 = newSVsv (path); |
680 | req->sv1 = newSVsv (path); |
640 | req->ptr1 = SvPVbyte_nolen (req->sv1); |
681 | req->ptr1 = SvPVbyte_nolen (req->sv1); |
641 | req->sv2 = data; |
|
|
642 | req->ptr2 = SvPVbyte_nolen (data); |
|
|
643 | |
682 | |
644 | REQ_SEND; |
683 | REQ_SEND; |
645 | } |
684 | } |
646 | |
685 | |
647 | void |
686 | void |
… | |
… | |
685 | aio_lstat = EIO_LSTAT |
724 | aio_lstat = EIO_LSTAT |
686 | PPCODE: |
725 | PPCODE: |
687 | { |
726 | { |
688 | dREQ; |
727 | dREQ; |
689 | |
728 | |
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); |
729 | req->sv1 = newSVsv (fh_or_path); |
699 | |
730 | |
700 | if (SvPOK (fh_or_path)) |
731 | if (SvPOK (req->sv1)) |
701 | { |
732 | { |
702 | req->type = ix; |
733 | req->type = ix; |
703 | req->ptr1 = SvPVbyte_nolen (req->sv1); |
734 | req->ptr1 = SvPVbyte_nolen (req->sv1); |
704 | } |
735 | } |
705 | else |
736 | else |
… | |
… | |
719 | |
750 | |
720 | req->nv1 = SvOK (atime) ? SvNV (atime) : -1.; |
751 | req->nv1 = SvOK (atime) ? SvNV (atime) : -1.; |
721 | req->nv2 = SvOK (mtime) ? SvNV (mtime) : -1.; |
752 | req->nv2 = SvOK (mtime) ? SvNV (mtime) : -1.; |
722 | req->sv1 = newSVsv (fh_or_path); |
753 | req->sv1 = newSVsv (fh_or_path); |
723 | |
754 | |
724 | if (SvPOK (fh_or_path)) |
755 | if (SvPOK (req->sv1)) |
725 | { |
756 | { |
726 | req->type = EIO_UTIME; |
757 | req->type = EIO_UTIME; |
727 | req->ptr1 = SvPVbyte_nolen (req->sv1); |
758 | req->ptr1 = SvPVbyte_nolen (req->sv1); |
728 | } |
759 | } |
729 | else |
760 | else |
… | |
… | |
742 | dREQ; |
773 | dREQ; |
743 | |
774 | |
744 | req->sv1 = newSVsv (fh_or_path); |
775 | req->sv1 = newSVsv (fh_or_path); |
745 | req->offs = SvOK (offset) ? SvVAL64 (offset) : -1; |
776 | req->offs = SvOK (offset) ? SvVAL64 (offset) : -1; |
746 | |
777 | |
747 | if (SvPOK (fh_or_path)) |
778 | if (SvPOK (req->sv1)) |
748 | { |
779 | { |
749 | req->type = EIO_TRUNCATE; |
780 | req->type = EIO_TRUNCATE; |
750 | req->ptr1 = SvPVbyte_nolen (req->sv1); |
781 | req->ptr1 = SvPVbyte_nolen (req->sv1); |
751 | } |
782 | } |
752 | else |
783 | else |
… | |
… | |
760 | |
791 | |
761 | void |
792 | void |
762 | aio_chmod (SV8 *fh_or_path, int mode, SV *callback=&PL_sv_undef) |
793 | aio_chmod (SV8 *fh_or_path, int mode, SV *callback=&PL_sv_undef) |
763 | ALIAS: |
794 | ALIAS: |
764 | aio_chmod = EIO_CHMOD |
795 | aio_chmod = EIO_CHMOD |
765 | aio_fchmod = EIO_FCHMOD |
|
|
766 | aio_mkdir = EIO_MKDIR |
796 | aio_mkdir = EIO_MKDIR |
767 | PPCODE: |
797 | PPCODE: |
768 | { |
798 | { |
769 | dREQ; |
799 | dREQ; |
770 | |
800 | |
771 | req->type = ix; |
|
|
772 | req->int2 = mode; |
801 | req->int2 = mode; |
773 | req->sv1 = newSVsv (fh_or_path); |
802 | req->sv1 = newSVsv (fh_or_path); |
774 | |
803 | |
775 | if (ix == EIO_FCHMOD) |
804 | if (SvPOK (req->sv1)) |
776 | req->int1 = PerlIO_fileno (IoIFP (sv_2io (fh_or_path))); |
805 | { |
|
|
806 | req->type = ix; |
|
|
807 | req->ptr1 = SvPVbyte_nolen (req->sv1); |
|
|
808 | } |
777 | else |
809 | else |
778 | req->ptr1 = SvPVbyte_nolen (req->sv1); |
810 | { |
|
|
811 | req->type = EIO_FCHMOD; |
|
|
812 | req->int1 = PerlIO_fileno (IoIFP (sv_2io (fh_or_path))); |
|
|
813 | } |
779 | |
814 | |
780 | REQ_SEND; |
815 | REQ_SEND; |
781 | } |
816 | } |
782 | |
817 | |
783 | void |
818 | void |
… | |
… | |
788 | |
823 | |
789 | req->int2 = SvOK (uid) ? SvIV (uid) : -1; |
824 | req->int2 = SvOK (uid) ? SvIV (uid) : -1; |
790 | req->int3 = SvOK (gid) ? SvIV (gid) : -1; |
825 | req->int3 = SvOK (gid) ? SvIV (gid) : -1; |
791 | req->sv1 = newSVsv (fh_or_path); |
826 | req->sv1 = newSVsv (fh_or_path); |
792 | |
827 | |
793 | if (SvPOK (fh_or_path)) |
828 | if (SvPOK (req->sv1)) |
794 | { |
829 | { |
795 | req->type = EIO_CHOWN; |
830 | req->type = EIO_CHOWN; |
796 | req->ptr1 = SvPVbyte_nolen (req->sv1); |
831 | req->ptr1 = SvPVbyte_nolen (req->sv1); |
797 | } |
832 | } |
798 | else |
833 | else |
… | |
… | |
874 | { |
909 | { |
875 | dREQ; |
910 | dREQ; |
876 | |
911 | |
877 | req->type = EIO_GROUP; |
912 | req->type = EIO_GROUP; |
878 | |
913 | |
879 | eio_submit (req); |
914 | req_submit (req); |
880 | XPUSHs (req_sv (req, AIO_GRP_KLASS)); |
915 | XPUSHs (req_sv (req, AIO_GRP_KLASS)); |
881 | } |
916 | } |
882 | |
917 | |
883 | void |
918 | void |
884 | aio_nop (SV *callback=&PL_sv_undef) |
919 | aio_nop (SV *callback=&PL_sv_undef) |
… | |
… | |
896 | |
931 | |
897 | int |
932 | int |
898 | aioreq_pri (int pri = 0) |
933 | aioreq_pri (int pri = 0) |
899 | PROTOTYPE: ;$ |
934 | PROTOTYPE: ;$ |
900 | CODE: |
935 | CODE: |
901 | RETVAL = next_pri - EIO_PRI_BIAS; |
936 | RETVAL = next_pri; |
902 | if (items > 0) |
937 | if (items > 0) |
903 | { |
938 | { |
904 | if (pri < EIO_PRI_MIN) pri = EIO_PRI_MIN; |
939 | if (pri < EIO_PRI_MIN) pri = EIO_PRI_MIN; |
905 | if (pri > EIO_PRI_MAX) pri = EIO_PRI_MAX; |
940 | if (pri > EIO_PRI_MAX) pri = EIO_PRI_MAX; |
906 | next_pri = pri + EIO_PRI_BIAS; |
941 | next_pri = pri; |
907 | } |
942 | } |
908 | OUTPUT: |
943 | OUTPUT: |
909 | RETVAL |
944 | RETVAL |
910 | |
945 | |
911 | void |
946 | void |
912 | aioreq_nice (int nice = 0) |
947 | aioreq_nice (int nice = 0) |
913 | CODE: |
948 | CODE: |
914 | nice = next_pri - nice; |
949 | nice = next_pri - nice; |
915 | if (nice < EIO_PRI_MIN) nice = EIO_PRI_MIN; |
950 | if (nice < EIO_PRI_MIN) nice = EIO_PRI_MIN; |
916 | if (nice > EIO_PRI_MAX) nice = EIO_PRI_MAX; |
951 | if (nice > EIO_PRI_MAX) nice = EIO_PRI_MAX; |
917 | next_pri = nice + EIO_PRI_BIAS; |
952 | next_pri = nice; |
918 | |
953 | |
919 | void |
954 | void |
920 | flush () |
955 | flush () |
921 | PROTOTYPE: |
956 | PROTOTYPE: |
922 | CODE: |
957 | CODE: |
923 | while (nreqs) |
958 | while (eio_nreqs ()) |
924 | { |
959 | { |
925 | poll_wait (); |
960 | poll_wait (); |
926 | poll_cb (); |
961 | poll_cb (); |
927 | } |
962 | } |
928 | |
963 | |
… | |
… | |
983 | |
1018 | |
984 | int |
1019 | int |
985 | nthreads() |
1020 | nthreads() |
986 | PROTOTYPE: |
1021 | PROTOTYPE: |
987 | CODE: |
1022 | CODE: |
988 | if (WORDACCESS_UNSAFE) X_LOCK (wrklock); |
1023 | RETVAL = eio_nthreads (); |
989 | RETVAL = started; |
|
|
990 | if (WORDACCESS_UNSAFE) X_UNLOCK (wrklock); |
|
|
991 | OUTPUT: |
1024 | OUTPUT: |
992 | RETVAL |
1025 | RETVAL |
|
|
1026 | |
|
|
1027 | void _on_next_submit (SV *cb) |
|
|
1028 | CODE: |
|
|
1029 | SvREFCNT_dec (on_next_submit); |
|
|
1030 | on_next_submit = SvOK (cb) ? newSVsv (cb) : 0; |
993 | |
1031 | |
994 | PROTOTYPES: DISABLE |
1032 | PROTOTYPES: DISABLE |
995 | |
1033 | |
996 | MODULE = IO::AIO PACKAGE = IO::AIO::REQ |
1034 | MODULE = IO::AIO PACKAGE = IO::AIO::REQ |
997 | |
1035 | |