… | |
… | |
65 | # include <signal.h> |
65 | # include <signal.h> |
66 | # define EIO_STRUCT_DIRENT struct dirent |
66 | # define EIO_STRUCT_DIRENT struct dirent |
67 | |
67 | |
68 | #endif |
68 | #endif |
69 | |
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 | |
70 | #define EIO_STRUCT_STAT Stat_t |
76 | #define EIO_STRUCT_STAT Stat_t |
71 | |
77 | |
72 | /* use NV for 32 bit perls as it allows larger offsets */ |
78 | /* use NV for 32 bit perls as it allows larger offsets */ |
73 | #if IVSIZE >= 8 |
79 | #if IVSIZE >= 8 |
74 | # define SvVAL64 SvIV |
80 | # define SvVAL64 SvIV |
75 | #else |
81 | #else |
76 | # define SvVAL64 SvNV |
82 | # define SvVAL64 SvNV |
77 | #endif |
83 | #endif |
78 | |
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 | |
79 | static HV *stash; |
98 | static HV *stash; |
80 | typedef SV SV8; /* byte-sv, used for argument-checking */ |
99 | typedef SV SV8; /* byte-sv, used for argument-checking */ |
81 | |
100 | |
82 | #define AIO_REQ_KLASS "IO::AIO::REQ" |
101 | #define AIO_REQ_KLASS "IO::AIO::REQ" |
83 | #define AIO_GRP_KLASS "IO::AIO::GRP" |
102 | #define AIO_GRP_KLASS "IO::AIO::GRP" |
… | |
… | |
92 | |
111 | |
93 | #include "libeio/eio.h" |
112 | #include "libeio/eio.h" |
94 | |
113 | |
95 | static int req_invoke (eio_req *req); |
114 | static int req_invoke (eio_req *req); |
96 | #define EIO_FINISH(req) req_invoke (req) |
115 | #define EIO_FINISH(req) req_invoke (req) |
97 | static void aio_grp_feed (eio_req *grp); |
|
|
98 | #define EIO_FEED(req) aio_grp_feed (req) |
|
|
99 | static void req_destroy (eio_req *grp); |
116 | static void req_destroy (eio_req *grp); |
100 | #define EIO_DESTROY(req) req_destroy (req) |
117 | #define EIO_DESTROY(req) req_destroy (req) |
101 | |
118 | |
102 | enum { |
119 | enum { |
103 | FLAG_SV2_RO_OFF = 0x40, /* data was set readonly */ |
120 | FLAG_SV2_RO_OFF = 0x40, /* data was set readonly */ |
… | |
… | |
175 | |
192 | |
176 | static void req_submit (eio_req *req) |
193 | static void req_submit (eio_req *req) |
177 | { |
194 | { |
178 | eio_submit (req); |
195 | eio_submit (req); |
179 | |
196 | |
180 | if (on_next_submit) |
197 | if (expect_false (on_next_submit)) |
181 | { |
198 | { |
182 | dSP; |
199 | dSP; |
183 | SV *cb = sv_2mortal (on_next_submit); |
200 | SV *cb = sv_2mortal (on_next_submit); |
184 | |
201 | |
185 | on_next_submit = 0; |
202 | on_next_submit = 0; |
… | |
… | |
195 | dSP; |
212 | dSP; |
196 | |
213 | |
197 | if (req->flags & FLAG_SV2_RO_OFF) |
214 | if (req->flags & FLAG_SV2_RO_OFF) |
198 | SvREADONLY_off (req->sv2); |
215 | SvREADONLY_off (req->sv2); |
199 | |
216 | |
200 | if (!EIO_CANCELLED (req) && SvOK (req->callback)) |
217 | if (!EIO_CANCELLED (req) && req->callback) |
201 | { |
218 | { |
202 | ENTER; |
219 | ENTER; |
203 | SAVETMPS; |
220 | SAVETMPS; |
204 | PUSHMARK (SP); |
221 | PUSHMARK (SP); |
205 | EXTEND (SP, 1); |
222 | EXTEND (SP, 1); |
… | |
… | |
412 | X_UNLOCK (reslock); |
429 | X_UNLOCK (reslock); |
413 | |
430 | |
414 | if (size) |
431 | if (size) |
415 | return; |
432 | return; |
416 | |
433 | |
417 | maybe_start_thread (); |
434 | etp_maybe_start_thread (); |
418 | |
435 | |
419 | FD_ZERO (&rfd); |
436 | FD_ZERO (&rfd); |
420 | FD_SET (respipe [0], &rfd); |
437 | FD_SET (respipe [0], &rfd); |
421 | |
438 | |
422 | PerlSock_select (respipe [0] + 1, &rfd, 0, 0, 0); |
439 | PerlSock_select (respipe [0] + 1, &rfd, 0, 0, 0); |
423 | } |
440 | } |
424 | } |
441 | } |
425 | |
442 | |
426 | static int poll_cb (void) |
443 | static int poll_cb (void) |
427 | { |
444 | { |
428 | int res; |
445 | for (;;) |
429 | |
|
|
430 | do |
|
|
431 | { |
446 | { |
432 | res = eio_poll (); |
447 | int res = eio_poll (); |
433 | |
448 | |
434 | if (res > 0) |
449 | if (res > 0) |
435 | croak (0); |
450 | croak (0); |
|
|
451 | |
|
|
452 | if (!max_outstanding || max_outstanding > eio_nreqs ()) |
|
|
453 | return res; |
|
|
454 | |
|
|
455 | poll_wait (); |
436 | } |
456 | } |
437 | while (max_outstanding && max_outstanding <= eio_nreqs ()); |
|
|
438 | |
|
|
439 | return res; |
|
|
440 | } |
457 | } |
441 | |
458 | |
442 | static void atfork_child (void) |
459 | static void atfork_child (void) |
443 | { |
460 | { |
444 | create_respipe (); |
461 | create_respipe (); |
445 | } |
462 | } |
446 | |
463 | |
|
|
464 | static SV * |
|
|
465 | get_cb (SV *cb_sv) |
|
|
466 | { |
|
|
467 | HV *st; |
|
|
468 | GV *gvp; |
|
|
469 | CV *cv; |
|
|
470 | |
|
|
471 | if (!SvOK (cb_sv)) |
|
|
472 | return 0; |
|
|
473 | |
|
|
474 | cv = sv_2cv (cb_sv, &st, &gvp, 0); |
|
|
475 | |
|
|
476 | if (!cv) |
|
|
477 | croak ("IO::AIO callback must be undef or a CODE reference"); |
|
|
478 | |
|
|
479 | return (SV *)cv; |
|
|
480 | } |
|
|
481 | |
447 | #define dREQ \ |
482 | #define dREQ \ |
|
|
483 | SV *cb_cv; \ |
448 | aio_req req; \ |
484 | aio_req req; \ |
449 | int req_pri = next_pri; \ |
485 | int req_pri = next_pri; \ |
450 | next_pri = EIO_PRI_DEFAULT; \ |
486 | next_pri = EIO_PRI_DEFAULT; \ |
451 | \ |
487 | \ |
452 | if (SvOK (callback) && !SvROK (callback)) \ |
488 | cb_cv = get_cb (callback); \ |
453 | croak ("callback must be undef or of reference type"); \ |
|
|
454 | \ |
489 | \ |
455 | Newz (0, req, 1, eio_req); \ |
490 | Newz (0, req, 1, eio_req); \ |
456 | if (!req) \ |
491 | if (!req) \ |
457 | croak ("out of memory during eio_req allocation"); \ |
492 | croak ("out of memory during eio_req allocation"); \ |
458 | \ |
493 | \ |
459 | req->callback = newSVsv (callback); \ |
494 | req->callback = SvREFCNT_inc (cb_cv); \ |
460 | req->pri = req_pri |
495 | req->pri = req_pri |
461 | |
496 | |
462 | #define REQ_SEND \ |
497 | #define REQ_SEND \ |
|
|
498 | PUTBACK; \ |
463 | req_submit (req); \ |
499 | req_submit (req); \ |
|
|
500 | SPAGAIN; \ |
464 | \ |
501 | \ |
465 | if (GIMME_V != G_VOID) \ |
502 | if (GIMME_V != G_VOID) \ |
466 | XPUSHs (req_sv (req, AIO_REQ_KLASS)); |
503 | XPUSHs (req_sv (req, AIO_REQ_KLASS)); |
467 | |
504 | |
468 | MODULE = IO::AIO PACKAGE = IO::AIO |
505 | MODULE = IO::AIO PACKAGE = IO::AIO |
… | |
… | |
479 | newCONSTSUB (stash, "O_CREAT", newSViv (O_CREAT)); |
516 | newCONSTSUB (stash, "O_CREAT", newSViv (O_CREAT)); |
480 | newCONSTSUB (stash, "O_TRUNC", newSViv (O_TRUNC)); |
517 | newCONSTSUB (stash, "O_TRUNC", newSViv (O_TRUNC)); |
481 | #ifndef _WIN32 |
518 | #ifndef _WIN32 |
482 | newCONSTSUB (stash, "S_IFIFO", newSViv (S_IFIFO)); |
519 | newCONSTSUB (stash, "S_IFIFO", newSViv (S_IFIFO)); |
483 | #endif |
520 | #endif |
|
|
521 | newCONSTSUB (stash, "S_IFIFO", newSViv (S_IFIFO)); |
|
|
522 | newCONSTSUB (stash, "SYNC_FILE_RANGE_WAIT_BEFORE", newSViv (EIO_SYNC_FILE_RANGE_WAIT_BEFORE)); |
|
|
523 | newCONSTSUB (stash, "SYNC_FILE_RANGE_WRITE" , newSViv (EIO_SYNC_FILE_RANGE_WRITE)); |
|
|
524 | newCONSTSUB (stash, "SYNC_FILE_RANGE_WAIT_AFTER" , newSViv (EIO_SYNC_FILE_RANGE_WAIT_AFTER)); |
484 | |
525 | |
485 | create_respipe (); |
526 | create_respipe (); |
486 | |
527 | |
487 | if (eio_init (want_poll, done_poll) < 0) |
528 | if (eio_init (want_poll, done_poll) < 0) |
488 | croak ("IO::AIO: unable to initialise eio library"); |
529 | croak ("IO::AIO: unable to initialise eio library"); |
… | |
… | |
554 | dREQ; |
595 | dREQ; |
555 | |
596 | |
556 | req->type = ix; |
597 | req->type = ix; |
557 | req->sv1 = newSVsv (fh); |
598 | req->sv1 = newSVsv (fh); |
558 | req->int1 = PerlIO_fileno (IoIFP (sv_2io (fh))); |
599 | req->int1 = PerlIO_fileno (IoIFP (sv_2io (fh))); |
|
|
600 | |
|
|
601 | REQ_SEND (req); |
|
|
602 | } |
|
|
603 | |
|
|
604 | void |
|
|
605 | aio_sync_file_range (SV *fh, SV *offset, SV *nbytes, IV flags, SV *callback=&PL_sv_undef) |
|
|
606 | PROTOTYPE: $$$$;$ |
|
|
607 | PPCODE: |
|
|
608 | { |
|
|
609 | dREQ; |
|
|
610 | |
|
|
611 | req->type = EIO_SYNC_FILE_RANGE; |
|
|
612 | req->sv1 = newSVsv (fh); |
|
|
613 | req->int1 = PerlIO_fileno (IoIFP (sv_2io (fh))); |
|
|
614 | req->offs = SvVAL64 (offset); |
|
|
615 | req->size = SvVAL64 (nbytes); |
|
|
616 | req->int2 = flags; |
559 | |
617 | |
560 | REQ_SEND (req); |
618 | REQ_SEND (req); |
561 | } |
619 | } |
562 | |
620 | |
563 | void |
621 | void |
… | |
… | |
597 | PPCODE: |
655 | PPCODE: |
598 | { |
656 | { |
599 | STRLEN svlen; |
657 | STRLEN svlen; |
600 | char *svptr = SvPVbyte (data, svlen); |
658 | char *svptr = SvPVbyte (data, svlen); |
601 | UV len = SvUV (length); |
659 | UV len = SvUV (length); |
|
|
660 | int fd; |
|
|
661 | |
|
|
662 | if (SvTYPE (data) > SVt_PVMG) |
|
|
663 | croak ("illegal data argument '%s', must be plain scalar string", SvPV_nolen (data)); |
602 | |
664 | |
603 | SvUPGRADE (data, SVt_PV); |
665 | SvUPGRADE (data, SVt_PV); |
604 | SvPOK_on (data); |
666 | SvPOK_only (data); |
605 | |
667 | |
606 | if (dataoffset < 0) |
668 | if (dataoffset < 0) |
607 | dataoffset += svlen; |
669 | dataoffset += svlen; |
608 | |
670 | |
609 | if (dataoffset < 0 || dataoffset > svlen) |
671 | if (dataoffset < 0 || dataoffset > svlen) |
… | |
… | |
622 | } |
684 | } |
623 | |
685 | |
624 | if (len < 0) |
686 | if (len < 0) |
625 | croak ("length must not be negative"); |
687 | croak ("length must not be negative"); |
626 | |
688 | |
|
|
689 | fd = PerlIO_fileno (ix == EIO_READ ? IoIFP (sv_2io (fh)) |
|
|
690 | : IoOFP (sv_2io (fh))); |
|
|
691 | |
|
|
692 | if (fd < 0) |
|
|
693 | croak ("illegal fh argument, either not an OS file or read/write mode mismatch"); |
|
|
694 | |
627 | { |
695 | { |
628 | dREQ; |
696 | dREQ; |
629 | |
697 | |
630 | req->type = ix; |
698 | req->type = ix; |
631 | req->sv1 = newSVsv (fh); |
699 | req->sv1 = newSVsv (fh); |
632 | req->int1 = PerlIO_fileno (ix == EIO_READ ? IoIFP (sv_2io (fh)) |
700 | req->int1 = df; |
633 | : IoOFP (sv_2io (fh))); |
|
|
634 | req->offs = SvOK (offset) ? SvVAL64 (offset) : -1; |
701 | req->offs = SvOK (offset) ? SvVAL64 (offset) : -1; |
635 | req->size = len; |
702 | req->size = len; |
636 | req->sv2 = SvREFCNT_inc (data); |
703 | req->sv2 = SvREFCNT_inc (data); |
637 | req->ptr2 = (char *)svptr + dataoffset; |
704 | req->ptr2 = (char *)svptr + dataoffset; |
638 | req->stroffset = dataoffset; |
705 | req->stroffset = dataoffset; |
… | |
… | |
888 | { |
955 | { |
889 | dREQ; |
956 | dREQ; |
890 | |
957 | |
891 | req->type = EIO_GROUP; |
958 | req->type = EIO_GROUP; |
892 | |
959 | |
893 | eio_submit (req); |
960 | req_submit (req); |
894 | XPUSHs (req_sv (req, AIO_GRP_KLASS)); |
961 | XPUSHs (req_sv (req, AIO_GRP_KLASS)); |
895 | } |
962 | } |
896 | |
963 | |
897 | void |
964 | void |
898 | aio_nop (SV *callback=&PL_sv_undef) |
965 | aio_nop (SV *callback=&PL_sv_undef) |
… | |
… | |
1017 | CODE: |
1084 | CODE: |
1018 | eio_cancel (req); |
1085 | eio_cancel (req); |
1019 | |
1086 | |
1020 | void |
1087 | void |
1021 | cb (aio_req_ornot req, SV *callback=&PL_sv_undef) |
1088 | cb (aio_req_ornot req, SV *callback=&PL_sv_undef) |
1022 | CODE: |
1089 | PPCODE: |
|
|
1090 | { |
|
|
1091 | if (GIMME_V != G_VOID) |
|
|
1092 | XPUSHs (req->callback ? sv_2mortal (newRV_inc (req->callback)) : &PL_sv_undef); |
|
|
1093 | |
|
|
1094 | if (items > 1) |
|
|
1095 | { |
|
|
1096 | SV *cb_cv = get_cb (callback); |
|
|
1097 | |
1023 | SvREFCNT_dec (req->callback); |
1098 | SvREFCNT_dec (req->callback); |
1024 | req->callback = newSVsv (callback); |
1099 | req->callback = SvREFCNT_inc (cb_cv); |
|
|
1100 | } |
|
|
1101 | } |
1025 | |
1102 | |
1026 | MODULE = IO::AIO PACKAGE = IO::AIO::GRP |
1103 | MODULE = IO::AIO PACKAGE = IO::AIO::GRP |
1027 | |
1104 | |
1028 | void |
1105 | void |
1029 | add (aio_req grp, ...) |
1106 | add (aio_req grp, ...) |
… | |
… | |
1084 | void |
1161 | void |
1085 | feed (aio_req grp, SV *callback=&PL_sv_undef) |
1162 | feed (aio_req grp, SV *callback=&PL_sv_undef) |
1086 | CODE: |
1163 | CODE: |
1087 | { |
1164 | { |
1088 | SvREFCNT_dec (grp->sv2); |
1165 | SvREFCNT_dec (grp->sv2); |
1089 | grp->sv2 = newSVsv (callback); |
1166 | grp->sv2 = newSVsv (callback); |
|
|
1167 | grp->feed = aio_grp_feed; |
1090 | |
1168 | |
1091 | if (grp->int2 <= 0) |
1169 | if (grp->int2 <= 0) |
1092 | grp->int2 = 2; |
1170 | grp->int2 = 2; |
1093 | |
1171 | |
1094 | eio_grp_limit (grp, grp->int2); |
1172 | eio_grp_limit (grp, grp->int2); |