… | |
… | |
3 | #include <errno.h> |
3 | #include <errno.h> |
4 | |
4 | |
5 | #include "EXTERN.h" |
5 | #include "EXTERN.h" |
6 | #include "perl.h" |
6 | #include "perl.h" |
7 | #include "XSUB.h" |
7 | #include "XSUB.h" |
|
|
8 | |
|
|
9 | #include "schmorp.h" |
8 | |
10 | |
9 | #include <stddef.h> |
11 | #include <stddef.h> |
10 | #include <stdlib.h> |
12 | #include <stdlib.h> |
11 | #include <errno.h> |
13 | #include <errno.h> |
12 | #include <sys/types.h> |
14 | #include <sys/types.h> |
… | |
… | |
117 | |
119 | |
118 | #include "libeio/eio.h" |
120 | #include "libeio/eio.h" |
119 | |
121 | |
120 | #ifndef POSIX_FADV_NORMAL |
122 | #ifndef POSIX_FADV_NORMAL |
121 | # define POSIX_FADV_NORMAL 0 |
123 | # define POSIX_FADV_NORMAL 0 |
|
|
124 | # define NO_FADVISE 1 |
122 | #endif |
125 | #endif |
123 | |
126 | |
124 | #ifndef POSIX_FADV_SEQUENTIAL |
127 | #ifndef POSIX_FADV_SEQUENTIAL |
125 | # define POSIX_FADV_SEQUENTIAL 0 |
128 | # define POSIX_FADV_SEQUENTIAL 0 |
126 | #endif |
129 | #endif |
… | |
… | |
157 | |
160 | |
158 | static SV *on_next_submit; |
161 | static SV *on_next_submit; |
159 | static int next_pri = EIO_PRI_DEFAULT; |
162 | static int next_pri = EIO_PRI_DEFAULT; |
160 | static int max_outstanding; |
163 | static int max_outstanding; |
161 | |
164 | |
162 | static int respipe_osf [2], respipe [2] = { -1, -1 }; |
165 | static s_epipe respipe; |
163 | |
166 | |
164 | static void req_destroy (aio_req req); |
167 | static void req_destroy (aio_req req); |
165 | static void req_cancel (aio_req req); |
168 | static void req_cancel (aio_req req); |
166 | |
169 | |
167 | static void want_poll (void) |
170 | static void want_poll (void) |
168 | { |
171 | { |
169 | /* write a dummy byte to the pipe so fh becomes ready */ |
172 | /* write a dummy byte to the pipe so fh becomes ready */ |
170 | respipe_write (respipe_osf [1], (const void *)&respipe_osf, 1); |
173 | s_epipe_signal (&respipe); |
171 | } |
174 | } |
172 | |
175 | |
173 | static void done_poll (void) |
176 | static void done_poll (void) |
174 | { |
177 | { |
175 | /* read any signals sent by the worker threads */ |
178 | /* read any signals sent by the worker threads */ |
176 | char buf [4]; |
179 | s_epipe_drain (&respipe); |
177 | while (respipe_read (respipe [0], buf, 4) == 4) |
|
|
178 | ; |
|
|
179 | } |
180 | } |
180 | |
181 | |
181 | /* must be called at most once */ |
182 | /* must be called at most once */ |
182 | static SV *req_sv (aio_req req, const char *klass) |
183 | static SV *req_sv (aio_req req, const char *klass) |
183 | { |
184 | { |
… | |
… | |
438 | grp->sv2 = 0; |
439 | grp->sv2 = 0; |
439 | |
440 | |
440 | eio_grp_cancel (grp); |
441 | eio_grp_cancel (grp); |
441 | } |
442 | } |
442 | |
443 | |
443 | #ifdef USE_SOCKETS_AS_HANDLES |
|
|
444 | # define TO_SOCKET(x) (win32_get_osfhandle (x)) |
|
|
445 | #else |
|
|
446 | # define TO_SOCKET(x) (x) |
|
|
447 | #endif |
|
|
448 | |
|
|
449 | static void |
444 | static void |
450 | create_respipe (void) |
445 | create_respipe (void) |
451 | { |
446 | { |
452 | int old_readfd = respipe [0]; |
|
|
453 | |
|
|
454 | if (respipe [1] >= 0) |
|
|
455 | respipe_close (TO_SOCKET (respipe [1])); |
|
|
456 | |
|
|
457 | #ifdef _WIN32 |
|
|
458 | if (PerlSock_socketpair (AF_UNIX, SOCK_STREAM, 0, respipe)) |
|
|
459 | #else |
|
|
460 | if (pipe (respipe)) |
447 | if (s_epipe_renew (&respipe)) |
461 | #endif |
|
|
462 | croak ("unable to initialize result pipe"); |
448 | croak ("unable to initialize result pipe"); |
463 | |
|
|
464 | if (old_readfd >= 0) |
|
|
465 | { |
|
|
466 | if (dup2 (TO_SOCKET (respipe [0]), TO_SOCKET (old_readfd)) < 0) |
|
|
467 | croak ("unable to initialize result pipe(2)"); |
|
|
468 | |
|
|
469 | respipe_close (respipe [0]); |
|
|
470 | respipe [0] = old_readfd; |
|
|
471 | } |
|
|
472 | |
|
|
473 | #ifdef _WIN32 |
|
|
474 | int arg = 1; |
|
|
475 | if (ioctlsocket (TO_SOCKET (respipe [0]), FIONBIO, &arg) |
|
|
476 | || ioctlsocket (TO_SOCKET (respipe [1]), FIONBIO, &arg)) |
|
|
477 | #else |
|
|
478 | if (fcntl (respipe [0], F_SETFL, O_NONBLOCK) |
|
|
479 | || fcntl (respipe [1], F_SETFL, O_NONBLOCK)) |
|
|
480 | #endif |
|
|
481 | croak ("unable to initialize result pipe(3)"); |
|
|
482 | |
|
|
483 | respipe_osf [0] = TO_SOCKET (respipe [0]); |
|
|
484 | respipe_osf [1] = TO_SOCKET (respipe [1]); |
|
|
485 | } |
449 | } |
486 | |
450 | |
487 | static void poll_wait (void) |
451 | static void poll_wait (void) |
488 | { |
452 | { |
489 | fd_set rfd; |
|
|
490 | |
|
|
491 | while (eio_nreqs ()) |
453 | while (eio_nreqs ()) |
492 | { |
454 | { |
493 | int size; |
455 | int size; |
494 | |
456 | |
495 | X_LOCK (reslock); |
457 | X_LOCK (reslock); |
… | |
… | |
499 | if (size) |
461 | if (size) |
500 | return; |
462 | return; |
501 | |
463 | |
502 | etp_maybe_start_thread (); |
464 | etp_maybe_start_thread (); |
503 | |
465 | |
504 | FD_ZERO (&rfd); |
466 | s_epipe_wait (&respipe); |
505 | FD_SET (respipe [0], &rfd); |
|
|
506 | |
|
|
507 | PerlSock_select (respipe [0] + 1, &rfd, 0, 0, 0); |
|
|
508 | } |
467 | } |
509 | } |
468 | } |
510 | |
469 | |
511 | static int poll_cb (void) |
470 | static int poll_cb (void) |
512 | { |
471 | { |
… | |
… | |
530 | } |
489 | } |
531 | |
490 | |
532 | static SV * |
491 | static SV * |
533 | get_cb (SV *cb_sv) |
492 | get_cb (SV *cb_sv) |
534 | { |
493 | { |
535 | HV *st; |
494 | SvGETMAGIC (cb_sv); |
536 | GV *gvp; |
495 | return SvOK (cb_sv) ? s_get_cv_croak (cb_sv) : 0; |
537 | CV *cv; |
|
|
538 | |
|
|
539 | if (!SvOK (cb_sv)) |
|
|
540 | return 0; |
|
|
541 | |
|
|
542 | cv = sv_2cv (cb_sv, &st, &gvp, 0); |
|
|
543 | |
|
|
544 | if (!cv) |
|
|
545 | croak ("IO::AIO callback must be undef or a CODE reference"); |
|
|
546 | |
|
|
547 | return (SV *)cv; |
|
|
548 | } |
496 | } |
549 | |
497 | |
550 | #define dREQ \ |
498 | #define dREQ \ |
551 | SV *cb_cv; \ |
499 | SV *cb_cv; \ |
552 | aio_req req; \ |
500 | aio_req req; \ |
… | |
… | |
567 | req_submit (req); \ |
515 | req_submit (req); \ |
568 | SPAGAIN; \ |
516 | SPAGAIN; \ |
569 | \ |
517 | \ |
570 | if (GIMME_V != G_VOID) \ |
518 | if (GIMME_V != G_VOID) \ |
571 | XPUSHs (req_sv (req, AIO_REQ_KLASS)); |
519 | XPUSHs (req_sv (req, AIO_REQ_KLASS)); |
572 | |
|
|
573 | static int |
|
|
574 | extract_fd (SV *fh, int wr) |
|
|
575 | { |
|
|
576 | int fd = PerlIO_fileno (wr ? IoOFP (sv_2io (fh)) : IoIFP (sv_2io (fh))); |
|
|
577 | |
|
|
578 | if (fd < 0) |
|
|
579 | croak ("illegal fh argument, either not an OS file or read/write mode mismatch"); |
|
|
580 | |
|
|
581 | return fd; |
|
|
582 | } |
|
|
583 | |
520 | |
584 | MODULE = IO::AIO PACKAGE = IO::AIO |
521 | MODULE = IO::AIO PACKAGE = IO::AIO |
585 | |
522 | |
586 | PROTOTYPES: ENABLE |
523 | PROTOTYPES: ENABLE |
587 | |
524 | |
… | |
… | |
701 | ALIAS: |
638 | ALIAS: |
702 | aio_fsync = EIO_FSYNC |
639 | aio_fsync = EIO_FSYNC |
703 | aio_fdatasync = EIO_FDATASYNC |
640 | aio_fdatasync = EIO_FDATASYNC |
704 | PPCODE: |
641 | PPCODE: |
705 | { |
642 | { |
706 | int fd = extract_fd (fh, 0); |
643 | int fd = s_fileno_croak (fh, 0); |
707 | dREQ; |
644 | dREQ; |
708 | |
645 | |
709 | req->type = ix; |
646 | req->type = ix; |
710 | req->sv1 = newSVsv (fh); |
647 | req->sv1 = newSVsv (fh); |
711 | req->int1 = fd; |
648 | req->int1 = fd; |
… | |
… | |
716 | void |
653 | void |
717 | aio_sync_file_range (SV *fh, off_t offset, size_t nbytes, UV flags, SV *callback=&PL_sv_undef) |
654 | aio_sync_file_range (SV *fh, off_t offset, size_t nbytes, UV flags, SV *callback=&PL_sv_undef) |
718 | PROTOTYPE: $$$$;$ |
655 | PROTOTYPE: $$$$;$ |
719 | PPCODE: |
656 | PPCODE: |
720 | { |
657 | { |
721 | int fd = extract_fd (fh, 0); |
658 | int fd = s_fileno_croak (fh, 0); |
722 | dREQ; |
659 | dREQ; |
723 | |
660 | |
724 | req->type = EIO_SYNC_FILE_RANGE; |
661 | req->type = EIO_SYNC_FILE_RANGE; |
725 | req->sv1 = newSVsv (fh); |
662 | req->sv1 = newSVsv (fh); |
726 | req->int1 = fd; |
663 | req->int1 = fd; |
… | |
… | |
735 | aio_close (SV *fh, SV *callback=&PL_sv_undef) |
672 | aio_close (SV *fh, SV *callback=&PL_sv_undef) |
736 | PROTOTYPE: $;$ |
673 | PROTOTYPE: $;$ |
737 | PPCODE: |
674 | PPCODE: |
738 | { |
675 | { |
739 | static int close_pipe = -1; /* dummy fd to close fds via dup2 */ |
676 | static int close_pipe = -1; /* dummy fd to close fds via dup2 */ |
740 | int fd = extract_fd (fh, 0); |
677 | int fd = s_fileno_croak (fh, 0); |
741 | dREQ; |
678 | dREQ; |
742 | |
679 | |
743 | if (close_pipe < 0) |
680 | if (close_pipe < 0) |
744 | { |
681 | { |
745 | int pipefd [2]; |
682 | int pipefd [2]; |
… | |
… | |
767 | aio_write = EIO_WRITE |
704 | aio_write = EIO_WRITE |
768 | PROTOTYPE: $$$$$;$ |
705 | PROTOTYPE: $$$$$;$ |
769 | PPCODE: |
706 | PPCODE: |
770 | { |
707 | { |
771 | STRLEN svlen; |
708 | STRLEN svlen; |
772 | int fd = extract_fd (fh, ix == EIO_WRITE); |
709 | int fd = s_fileno_croak (fh, ix == EIO_WRITE); |
773 | char *svptr = SvPVbyte (data, svlen); |
710 | char *svptr = SvPVbyte (data, svlen); |
774 | UV len = SvUV (length); |
711 | UV len = SvUV (length); |
775 | |
712 | |
776 | if (dataoffset < 0) |
713 | if (dataoffset < 0) |
777 | dataoffset += svlen; |
714 | dataoffset += svlen; |
… | |
… | |
832 | void |
769 | void |
833 | aio_sendfile (SV *out_fh, SV *in_fh, off_t in_offset, size_t length, SV *callback=&PL_sv_undef) |
770 | aio_sendfile (SV *out_fh, SV *in_fh, off_t in_offset, size_t length, SV *callback=&PL_sv_undef) |
834 | PROTOTYPE: $$$$;$ |
771 | PROTOTYPE: $$$$;$ |
835 | PPCODE: |
772 | PPCODE: |
836 | { |
773 | { |
837 | int ifd = extract_fd (in_fh , 0); |
774 | int ifd = s_fileno_croak (in_fh , 0); |
838 | int ofd = extract_fd (out_fh, 0); |
775 | int ofd = s_fileno_croak (out_fh, 1); |
839 | dREQ; |
776 | dREQ; |
840 | |
777 | |
841 | req->type = EIO_SENDFILE; |
778 | req->type = EIO_SENDFILE; |
842 | req->sv1 = newSVsv (out_fh); |
779 | req->sv1 = newSVsv (out_fh); |
843 | req->int1 = ofd; |
780 | req->int1 = ofd; |
… | |
… | |
852 | void |
789 | void |
853 | aio_readahead (SV *fh, off_t offset, size_t length, SV *callback=&PL_sv_undef) |
790 | aio_readahead (SV *fh, off_t offset, size_t length, SV *callback=&PL_sv_undef) |
854 | PROTOTYPE: $$$;$ |
791 | PROTOTYPE: $$$;$ |
855 | PPCODE: |
792 | PPCODE: |
856 | { |
793 | { |
857 | int fd = extract_fd (fh, 0); |
794 | int fd = s_fileno_croak (fh, 0); |
858 | dREQ; |
795 | dREQ; |
859 | |
796 | |
860 | req->type = EIO_READAHEAD; |
797 | req->type = EIO_READAHEAD; |
861 | req->sv1 = newSVsv (fh); |
798 | req->sv1 = newSVsv (fh); |
862 | req->int1 = fd; |
799 | req->int1 = fd; |
… | |
… | |
1138 | |
1075 | |
1139 | int |
1076 | int |
1140 | poll_fileno() |
1077 | poll_fileno() |
1141 | PROTOTYPE: |
1078 | PROTOTYPE: |
1142 | CODE: |
1079 | CODE: |
1143 | RETVAL = respipe [0]; |
1080 | RETVAL = s_epipe_fd (&respipe); |
1144 | OUTPUT: |
1081 | OUTPUT: |
1145 | RETVAL |
1082 | RETVAL |
1146 | |
1083 | |
1147 | int |
1084 | int |
1148 | poll_cb(...) |
1085 | poll_cb(...) |
… | |
… | |
1192 | |
1129 | |
1193 | int |
1130 | int |
1194 | fadvise (aio_rfd fh, off_t offset, off_t length, IV advice) |
1131 | fadvise (aio_rfd fh, off_t offset, off_t length, IV advice) |
1195 | PROTOTYPE: $$$$ |
1132 | PROTOTYPE: $$$$ |
1196 | CODE: |
1133 | CODE: |
1197 | { |
|
|
1198 | #if _XOPEN_SOURCE >= 600 |
1134 | #if _XOPEN_SOURCE >= 600 && !NO_FADVISE |
1199 | RETVAL = posix_fadvise (fh, offset, length, advice); |
1135 | RETVAL = posix_fadvise (fh, offset, length, advice); |
1200 | #else |
1136 | #else |
1201 | RETVAL = errno = ENOSYS; |
1137 | RETVAL = errno = ENOSYS; |
1202 | #endif |
1138 | #endif |
1203 | } |
|
|
1204 | OUTPUT: |
1139 | OUTPUT: |
1205 | RETVAL |
1140 | RETVAL |
1206 | |
1141 | |
1207 | ssize_t |
1142 | ssize_t |
1208 | sendfile (aio_wfd ofh, aio_rfd ifh, off_t offset, size_t count) |
1143 | sendfile (aio_wfd ofh, aio_rfd ifh, off_t offset, size_t count) |
… | |
… | |
1231 | if (GIMME_V != G_VOID) |
1166 | if (GIMME_V != G_VOID) |
1232 | XPUSHs (req->callback ? sv_2mortal (newRV_inc (req->callback)) : &PL_sv_undef); |
1167 | XPUSHs (req->callback ? sv_2mortal (newRV_inc (req->callback)) : &PL_sv_undef); |
1233 | |
1168 | |
1234 | if (items > 1) |
1169 | if (items > 1) |
1235 | { |
1170 | { |
1236 | SV *cb_cv = get_cb (callback); |
1171 | SV *cb_cv =get_cb (callback); |
1237 | |
1172 | |
1238 | SvREFCNT_dec (req->callback); |
1173 | SvREFCNT_dec (req->callback); |
1239 | req->callback = SvREFCNT_inc (cb_cv); |
1174 | req->callback = SvREFCNT_inc (cb_cv); |
1240 | } |
1175 | } |
1241 | } |
1176 | } |