… | |
… | |
80 | # define SvVAL64 SvIV |
80 | # define SvVAL64 SvIV |
81 | #else |
81 | #else |
82 | # define SvVAL64 SvNV |
82 | # define SvVAL64 SvNV |
83 | #endif |
83 | #endif |
84 | |
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 | |
85 | static HV *stash; |
98 | static HV *stash; |
86 | typedef SV SV8; /* byte-sv, used for argument-checking */ |
99 | typedef SV SV8; /* byte-sv, used for argument-checking */ |
87 | |
100 | |
88 | #define AIO_REQ_KLASS "IO::AIO::REQ" |
101 | #define AIO_REQ_KLASS "IO::AIO::REQ" |
89 | #define AIO_GRP_KLASS "IO::AIO::GRP" |
102 | #define AIO_GRP_KLASS "IO::AIO::GRP" |
… | |
… | |
98 | |
111 | |
99 | #include "libeio/eio.h" |
112 | #include "libeio/eio.h" |
100 | |
113 | |
101 | static int req_invoke (eio_req *req); |
114 | static int req_invoke (eio_req *req); |
102 | #define EIO_FINISH(req) req_invoke (req) |
115 | #define EIO_FINISH(req) req_invoke (req) |
103 | static void aio_grp_feed (eio_req *grp); |
|
|
104 | #define EIO_FEED(req) aio_grp_feed (req) |
|
|
105 | static void req_destroy (eio_req *grp); |
116 | static void req_destroy (eio_req *grp); |
106 | #define EIO_DESTROY(req) req_destroy (req) |
117 | #define EIO_DESTROY(req) req_destroy (req) |
107 | |
118 | |
108 | enum { |
119 | enum { |
109 | FLAG_SV2_RO_OFF = 0x40, /* data was set readonly */ |
120 | FLAG_SV2_RO_OFF = 0x40, /* data was set readonly */ |
… | |
… | |
181 | |
192 | |
182 | static void req_submit (eio_req *req) |
193 | static void req_submit (eio_req *req) |
183 | { |
194 | { |
184 | eio_submit (req); |
195 | eio_submit (req); |
185 | |
196 | |
186 | if (on_next_submit) |
197 | if (expect_false (on_next_submit)) |
187 | { |
198 | { |
188 | dSP; |
199 | dSP; |
189 | SV *cb = sv_2mortal (on_next_submit); |
200 | SV *cb = sv_2mortal (on_next_submit); |
190 | |
201 | |
191 | on_next_submit = 0; |
202 | on_next_submit = 0; |
… | |
… | |
201 | dSP; |
212 | dSP; |
202 | |
213 | |
203 | if (req->flags & FLAG_SV2_RO_OFF) |
214 | if (req->flags & FLAG_SV2_RO_OFF) |
204 | SvREADONLY_off (req->sv2); |
215 | SvREADONLY_off (req->sv2); |
205 | |
216 | |
206 | if (!EIO_CANCELLED (req) && SvOK (req->callback)) |
217 | if (!EIO_CANCELLED (req) && req->callback) |
207 | { |
218 | { |
208 | ENTER; |
219 | ENTER; |
209 | SAVETMPS; |
220 | SAVETMPS; |
210 | PUSHMARK (SP); |
221 | PUSHMARK (SP); |
211 | EXTEND (SP, 1); |
222 | EXTEND (SP, 1); |
… | |
… | |
448 | static void atfork_child (void) |
459 | static void atfork_child (void) |
449 | { |
460 | { |
450 | create_respipe (); |
461 | create_respipe (); |
451 | } |
462 | } |
452 | |
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 | |
453 | #define dREQ \ |
482 | #define dREQ \ |
|
|
483 | SV *cb_cv; \ |
454 | aio_req req; \ |
484 | aio_req req; \ |
455 | int req_pri = next_pri; \ |
485 | int req_pri = next_pri; \ |
456 | next_pri = EIO_PRI_DEFAULT; \ |
486 | next_pri = EIO_PRI_DEFAULT; \ |
457 | \ |
487 | \ |
458 | if (SvOK (callback) && !SvROK (callback)) \ |
488 | cb_cv = get_cb (callback); \ |
459 | croak ("callback must be undef or of reference type"); \ |
|
|
460 | \ |
489 | \ |
461 | Newz (0, req, 1, eio_req); \ |
490 | Newz (0, req, 1, eio_req); \ |
462 | if (!req) \ |
491 | if (!req) \ |
463 | croak ("out of memory during eio_req allocation"); \ |
492 | croak ("out of memory during eio_req allocation"); \ |
464 | \ |
493 | \ |
465 | req->callback = newSVsv (callback); \ |
494 | req->callback = SvREFCNT_inc (cb_cv); \ |
466 | req->pri = req_pri |
495 | req->pri = req_pri |
467 | |
496 | |
468 | #define REQ_SEND \ |
497 | #define REQ_SEND \ |
|
|
498 | PUTBACK; \ |
469 | req_submit (req); \ |
499 | req_submit (req); \ |
|
|
500 | SPAGAIN; \ |
470 | \ |
501 | \ |
471 | if (GIMME_V != G_VOID) \ |
502 | if (GIMME_V != G_VOID) \ |
472 | XPUSHs (req_sv (req, AIO_REQ_KLASS)); |
503 | XPUSHs (req_sv (req, AIO_REQ_KLASS)); |
473 | |
504 | |
474 | MODULE = IO::AIO PACKAGE = IO::AIO |
505 | MODULE = IO::AIO PACKAGE = IO::AIO |
… | |
… | |
637 | req->sv1 = newSVsv (fh); |
668 | req->sv1 = newSVsv (fh); |
638 | req->int1 = PerlIO_fileno (ix == EIO_READ ? IoIFP (sv_2io (fh)) |
669 | req->int1 = PerlIO_fileno (ix == EIO_READ ? IoIFP (sv_2io (fh)) |
639 | : IoOFP (sv_2io (fh))); |
670 | : IoOFP (sv_2io (fh))); |
640 | req->offs = SvOK (offset) ? SvVAL64 (offset) : -1; |
671 | req->offs = SvOK (offset) ? SvVAL64 (offset) : -1; |
641 | req->size = len; |
672 | req->size = len; |
642 | req->sv2 = SvREFCNT_inc (data); |
673 | req->sv2 = SvREFCNT_inc_NN (data); |
643 | req->ptr2 = (char *)svptr + dataoffset; |
674 | req->ptr2 = (char *)svptr + dataoffset; |
644 | req->stroffset = dataoffset; |
675 | req->stroffset = dataoffset; |
645 | |
676 | |
646 | if (!SvREADONLY (data)) |
677 | if (!SvREADONLY (data)) |
647 | { |
678 | { |
… | |
… | |
894 | { |
925 | { |
895 | dREQ; |
926 | dREQ; |
896 | |
927 | |
897 | req->type = EIO_GROUP; |
928 | req->type = EIO_GROUP; |
898 | |
929 | |
899 | eio_submit (req); |
930 | req_submit (req); |
900 | XPUSHs (req_sv (req, AIO_GRP_KLASS)); |
931 | XPUSHs (req_sv (req, AIO_GRP_KLASS)); |
901 | } |
932 | } |
902 | |
933 | |
903 | void |
934 | void |
904 | aio_nop (SV *callback=&PL_sv_undef) |
935 | aio_nop (SV *callback=&PL_sv_undef) |
… | |
… | |
1023 | CODE: |
1054 | CODE: |
1024 | eio_cancel (req); |
1055 | eio_cancel (req); |
1025 | |
1056 | |
1026 | void |
1057 | void |
1027 | cb (aio_req_ornot req, SV *callback=&PL_sv_undef) |
1058 | cb (aio_req_ornot req, SV *callback=&PL_sv_undef) |
1028 | CODE: |
1059 | PPCODE: |
|
|
1060 | { |
|
|
1061 | if (GIMME_V != G_VOID) |
|
|
1062 | XPUSHs (req->callback ? sv_2mortal (newRV_inc (req->callback)) : &PL_sv_undef); |
|
|
1063 | |
|
|
1064 | if (items > 1) |
|
|
1065 | { |
|
|
1066 | SV *cb_cv = get_cb (callback); |
|
|
1067 | |
1029 | SvREFCNT_dec (req->callback); |
1068 | SvREFCNT_dec (req->callback); |
1030 | req->callback = newSVsv (callback); |
1069 | req->callback = SvREFCNT_inc (cb_cv); |
|
|
1070 | } |
|
|
1071 | } |
1031 | |
1072 | |
1032 | MODULE = IO::AIO PACKAGE = IO::AIO::GRP |
1073 | MODULE = IO::AIO PACKAGE = IO::AIO::GRP |
1033 | |
1074 | |
1034 | void |
1075 | void |
1035 | add (aio_req grp, ...) |
1076 | add (aio_req grp, ...) |
… | |
… | |
1090 | void |
1131 | void |
1091 | feed (aio_req grp, SV *callback=&PL_sv_undef) |
1132 | feed (aio_req grp, SV *callback=&PL_sv_undef) |
1092 | CODE: |
1133 | CODE: |
1093 | { |
1134 | { |
1094 | SvREFCNT_dec (grp->sv2); |
1135 | SvREFCNT_dec (grp->sv2); |
1095 | grp->sv2 = newSVsv (callback); |
1136 | grp->sv2 = newSVsv (callback); |
|
|
1137 | grp->feed = aio_grp_feed; |
1096 | |
1138 | |
1097 | if (grp->int2 <= 0) |
1139 | if (grp->int2 <= 0) |
1098 | grp->int2 = 2; |
1140 | grp->int2 = 2; |
1099 | |
1141 | |
1100 | eio_grp_limit (grp, grp->int2); |
1142 | eio_grp_limit (grp, grp->int2); |