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.126 by root, Thu Sep 25 07:56:07 2008 UTC vs.
Revision 1.135 by root, Sun Apr 19 13:06:15 2009 UTC

… …
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
85static HV *stash; 98static HV *stash;
86typedef SV SV8; /* byte-sv, used for argument-checking */ 99typedef 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
101static int req_invoke (eio_req *req); 114static int req_invoke (eio_req *req);
102#define EIO_FINISH(req) req_invoke (req) 115#define EIO_FINISH(req) req_invoke (req)
103static void aio_grp_feed (eio_req *grp);
104#define EIO_FEED(req) aio_grp_feed (req)
105static void req_destroy (eio_req *grp); 116static void req_destroy (eio_req *grp);
106#define EIO_DESTROY(req) req_destroy (req) 117#define EIO_DESTROY(req) req_destroy (req)
107 118
108enum { 119enum {
109 FLAG_SV2_RO_OFF = 0x40, /* data was set readonly */ 120 FLAG_SV2_RO_OFF = 0x40, /* data was set readonly */
… …
181 192
182static void req_submit (eio_req *req) 193static 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);
… …
448static void atfork_child (void) 459static void atfork_child (void)
449{ 460{
450 create_respipe (); 461 create_respipe ();
451} 462}
452 463
464static SV *
465get_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 \
469 PUTBACK; \ 498 PUTBACK; \
470 req_submit (req); \ 499 req_submit (req); \
… …
487 newCONSTSUB (stash, "O_CREAT", newSViv (O_CREAT)); 516 newCONSTSUB (stash, "O_CREAT", newSViv (O_CREAT));
488 newCONSTSUB (stash, "O_TRUNC", newSViv (O_TRUNC)); 517 newCONSTSUB (stash, "O_TRUNC", newSViv (O_TRUNC));
489#ifndef _WIN32 518#ifndef _WIN32
490 newCONSTSUB (stash, "S_IFIFO", newSViv (S_IFIFO)); 519 newCONSTSUB (stash, "S_IFIFO", newSViv (S_IFIFO));
491#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));
492 525
493 create_respipe (); 526 create_respipe ();
494 527
495 if (eio_init (want_poll, done_poll) < 0) 528 if (eio_init (want_poll, done_poll) < 0)
496 croak ("IO::AIO: unable to initialise eio library"); 529 croak ("IO::AIO: unable to initialise eio library");
… …
562 dREQ; 595 dREQ;
563 596
564 req->type = ix; 597 req->type = ix;
565 req->sv1 = newSVsv (fh); 598 req->sv1 = newSVsv (fh);
566 req->int1 = PerlIO_fileno (IoIFP (sv_2io (fh))); 599 req->int1 = PerlIO_fileno (IoIFP (sv_2io (fh)));
600
601 REQ_SEND (req);
602}
603
604void
605aio_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;
567 617
568 REQ_SEND (req); 618 REQ_SEND (req);
569} 619}
570 620
571void 621void
… …
605 PPCODE: 655 PPCODE:
606{ 656{
607 STRLEN svlen; 657 STRLEN svlen;
608 char *svptr = SvPVbyte (data, svlen); 658 char *svptr = SvPVbyte (data, svlen);
609 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));
610 664
611 SvUPGRADE (data, SVt_PV); 665 SvUPGRADE (data, SVt_PV);
612 SvPOK_on (data); 666 SvPOK_only (data);
613 667
614 if (dataoffset < 0) 668 if (dataoffset < 0)
615 dataoffset += svlen; 669 dataoffset += svlen;
616 670
617 if (dataoffset < 0 || dataoffset > svlen) 671 if (dataoffset < 0 || dataoffset > svlen)
… …
630 } 684 }
631 685
632 if (len < 0) 686 if (len < 0)
633 croak ("length must not be negative"); 687 croak ("length must not be negative");
634 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
635 { 695 {
636 dREQ; 696 dREQ;
637 697
638 req->type = ix; 698 req->type = ix;
639 req->sv1 = newSVsv (fh); 699 req->sv1 = newSVsv (fh);
640 req->int1 = PerlIO_fileno (ix == EIO_READ ? IoIFP (sv_2io (fh)) 700 req->int1 = fd;
641 : IoOFP (sv_2io (fh)));
642 req->offs = SvOK (offset) ? SvVAL64 (offset) : -1; 701 req->offs = SvOK (offset) ? SvVAL64 (offset) : -1;
643 req->size = len; 702 req->size = len;
644 req->sv2 = SvREFCNT_inc (data); 703 req->sv2 = SvREFCNT_inc (data);
645 req->ptr2 = (char *)svptr + dataoffset; 704 req->ptr2 = (char *)svptr + dataoffset;
646 req->stroffset = dataoffset; 705 req->stroffset = dataoffset;
… …
896{ 955{
897 dREQ; 956 dREQ;
898 957
899 req->type = EIO_GROUP; 958 req->type = EIO_GROUP;
900 959
901 eio_submit (req); 960 req_submit (req);
902 XPUSHs (req_sv (req, AIO_GRP_KLASS)); 961 XPUSHs (req_sv (req, AIO_GRP_KLASS));
903} 962}
904 963
905void 964void
906aio_nop (SV *callback=&PL_sv_undef) 965aio_nop (SV *callback=&PL_sv_undef)
… …
1025 CODE: 1084 CODE:
1026 eio_cancel (req); 1085 eio_cancel (req);
1027 1086
1028void 1087void
1029cb (aio_req_ornot req, SV *callback=&PL_sv_undef) 1088cb (aio_req_ornot req, SV *callback=&PL_sv_undef)
1030 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
1031 SvREFCNT_dec (req->callback); 1098 SvREFCNT_dec (req->callback);
1032 req->callback = newSVsv (callback); 1099 req->callback = SvREFCNT_inc (cb_cv);
1100 }
1101}
1033 1102
1034MODULE = IO::AIO PACKAGE = IO::AIO::GRP 1103MODULE = IO::AIO PACKAGE = IO::AIO::GRP
1035 1104
1036void 1105void
1037add (aio_req grp, ...) 1106add (aio_req grp, ...)
… …
1092void 1161void
1093feed (aio_req grp, SV *callback=&PL_sv_undef) 1162feed (aio_req grp, SV *callback=&PL_sv_undef)
1094 CODE: 1163 CODE:
1095{ 1164{
1096 SvREFCNT_dec (grp->sv2); 1165 SvREFCNT_dec (grp->sv2);
1097 grp->sv2 = newSVsv (callback); 1166 grp->sv2 = newSVsv (callback);
1167 grp->feed = aio_grp_feed;
1098 1168
1099 if (grp->int2 <= 0) 1169 if (grp->int2 <= 0)
1100 grp->int2 = 2; 1170 grp->int2 = 2;
1101 1171
1102 eio_grp_limit (grp, grp->int2); 1172 eio_grp_limit (grp, grp->int2);

Diff Legend

– Removed lines
+ Added lines
< Changed lines
> Changed lines