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.144 by root, Fri Jun 12 16:48:08 2009 UTC vs.
Revision 1.152 by root, Tue Jul 14 00:32:27 2009 UTC

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>
75 77
76#define EIO_STRUCT_STAT Stat_t 78#define EIO_STRUCT_STAT Stat_t
77 79
78/* use NV for 32 bit perls as it allows larger offsets */ 80/* use NV for 32 bit perls as it allows larger offsets */
79#if IVSIZE >= 8 81#if IVSIZE >= 8
82# define VAL64 IV
80# define SvVAL64 SvIV 83# define SvVAL64 SvIV
84# define newSVval64 newSViv
81#else 85#else
86# define VAL64 NV
82# define SvVAL64 SvNV 87# define SvVAL64 SvNV
88# define newSVval64 newSVnv
83#endif 89#endif
84 90
85/*****************************************************************************/ 91/*****************************************************************************/
86 92
87#if __GNUC__ >= 3 93#if __GNUC__ >= 3
95 101
96/*****************************************************************************/ 102/*****************************************************************************/
97 103
98static HV *stash; 104static HV *stash;
99typedef SV SV8; /* byte-sv, used for argument-checking */ 105typedef SV SV8; /* byte-sv, used for argument-checking */
106typedef int aio_rfd; /* read file desriptor */
107typedef int aio_wfd; /* write file descriptor */
100 108
101#define AIO_REQ_KLASS "IO::AIO::REQ" 109#define AIO_REQ_KLASS "IO::AIO::REQ"
102#define AIO_GRP_KLASS "IO::AIO::GRP" 110#define AIO_GRP_KLASS "IO::AIO::GRP"
103 111
104#define EIO_REQ_MEMBERS \ 112#define EIO_REQ_MEMBERS \
108 SV *self; 116 SV *self;
109 117
110#define EIO_NO_WRAPPERS 1 118#define EIO_NO_WRAPPERS 1
111 119
112#include "libeio/eio.h" 120#include "libeio/eio.h"
121
122#ifndef POSIX_FADV_NORMAL
123# define POSIX_FADV_NORMAL 0
124# define NO_FADVISE 1
125#endif
126
127#ifndef POSIX_FADV_SEQUENTIAL
128# define POSIX_FADV_SEQUENTIAL 0
129#endif
130
131#ifndef POSIX_FADV_RANDOM
132# define POSIX_FADV_RANDOM 0
133#endif
134
135#ifndef POSIX_FADV_NOREUSE
136# define POSIX_FADV_NOREUSE 0
137#endif
138
139#ifndef POSIX_FADV_WILLNEED
140# define POSIX_FADV_WILLNEED 0
141#endif
142
143#ifndef POSIX_FADV_DONTNEED
144# define POSIX_FADV_DONTNEED 0
145#endif
113 146
114static int req_invoke (eio_req *req); 147static int req_invoke (eio_req *req);
115#define EIO_FINISH(req) req_invoke (req) 148#define EIO_FINISH(req) req_invoke (req)
116static void req_destroy (eio_req *grp); 149static void req_destroy (eio_req *grp);
117#define EIO_DESTROY(req) req_destroy (req) 150#define EIO_DESTROY(req) req_destroy (req)
228 SV *rv = &PL_sv_undef; 261 SV *rv = &PL_sv_undef;
229 262
230 if (req->result >= 0) 263 if (req->result >= 0)
231 { 264 {
232 int i; 265 int i;
233 char *buf = req->ptr2; 266 char *names = (char *)req->ptr2;
267 eio_dirent *ent = (eio_dirent *)req->ptr1; /* might be 0 */
234 AV *av = newAV (); 268 AV *av = newAV ();
235 269
236 av_extend (av, req->result - 1); 270 av_extend (av, req->result - 1);
237 271
238 for (i = 0; i < req->result; ++i) 272 for (i = 0; i < req->result; ++i)
239 { 273 {
240 if (req->int1 & EIO_READDIR_DENTS) 274 if (req->int1 & EIO_READDIR_DENTS)
241 { 275 {
242 eio_dirent *ent = (eio_dirent *)buf;
243 SV *namesv = newSVpvn (ent->name, ent->namelen); 276 SV *namesv = newSVpvn (names + ent->nameofs, ent->namelen);
244 277
245 if (req->int1 & EIO_READDIR_CUSTOM2) 278 if (req->int1 & EIO_READDIR_CUSTOM2)
246 { 279 {
247 static SV *sv_type [EIO_DT_MAX + 1]; /* type sv cache */ 280 static SV *sv_type [EIO_DT_MAX + 1]; /* type sv cache */
248 AV *avent = newAV (); 281 AV *avent = newAV ();
262 av_store (av, i, newRV_noinc ((SV *)avent)); 295 av_store (av, i, newRV_noinc ((SV *)avent));
263 } 296 }
264 else 297 else
265 av_store (av, i, namesv); 298 av_store (av, i, namesv);
266 299
267 buf += sizeof (eio_dirent); 300 ++ent;
268 } 301 }
269 else 302 else
270 { 303 {
271 SV *name = newSVpv (buf, 0); 304 SV *name = newSVpv (names, 0);
272 av_store (av, i, name); 305 av_store (av, i, name);
273 buf += SvCUR (name) + 1; 306 names += SvCUR (name) + 1;
274 } 307 }
275 } 308 }
276 309
277 rv = sv_2mortal (newRV_noinc ((SV *)av)); 310 rv = sv_2mortal (newRV_noinc ((SV *)av));
278 } 311 }
294 GV *gv = (GV *)sv_newmortal (); 327 GV *gv = (GV *)sv_newmortal ();
295 int flags = req->int1 & (O_RDONLY | O_WRONLY | O_RDWR); 328 int flags = req->int1 & (O_RDONLY | O_WRONLY | O_RDWR);
296 char sym [64]; 329 char sym [64];
297 int symlen; 330 int symlen;
298 331
299 symlen = snprintf (sym, sizeof (sym), "fd#%u", (unsigned int)req->result); 332 symlen = snprintf (sym, sizeof (sym), "fd#%d", (int)req->result);
300 gv_init (gv, stash, sym, symlen, 0); 333 gv_init (gv, stash, sym, symlen, 0);
301 334
302 symlen = snprintf ( 335 symlen = snprintf (
303 sym, 336 sym,
304 sizeof (sym), 337 sizeof (sym),
305 "%s&=%u", 338 "%s&=%d",
306 flags == O_RDONLY ? "<" : flags == O_WRONLY ? ">" : "+<", 339 flags == O_RDONLY ? "<" : flags == O_WRONLY ? ">" : "+<",
307 (unsigned int)req->result 340 (int)req->result
308 ); 341 );
309 342
310 if (do_open (gv, sym, symlen, 0, 0, 0, 0)) 343 if (do_open (gv, sym, symlen, 0, 0, 0, 0))
311 fh = (SV *)gv; 344 fh = (SV *)gv;
312 } 345 }
500} 533}
501 534
502static SV * 535static SV *
503get_cb (SV *cb_sv) 536get_cb (SV *cb_sv)
504{ 537{
505 HV *st; 538 SvGETMAGIC (cb_sv);
506 GV *gvp; 539 return SvOK (cb_sv) ? s_get_cv_croak (cb_sv) : 0;
507 CV *cv;
508
509 if (!SvOK (cb_sv))
510 return 0;
511
512 cv = sv_2cv (cb_sv, &st, &gvp, 0);
513
514 if (!cv)
515 croak ("IO::AIO callback must be undef or a CODE reference");
516
517 return (SV *)cv;
518} 540}
519 541
520#define dREQ \ 542#define dREQ \
521 SV *cb_cv; \ 543 SV *cb_cv; \
522 aio_req req; \ 544 aio_req req; \
538 SPAGAIN; \ 560 SPAGAIN; \
539 \ 561 \
540 if (GIMME_V != G_VOID) \ 562 if (GIMME_V != G_VOID) \
541 XPUSHs (req_sv (req, AIO_REQ_KLASS)); 563 XPUSHs (req_sv (req, AIO_REQ_KLASS));
542 564
543static int
544extract_fd (SV *fh, int wr)
545{
546 int fd = PerlIO_fileno (wr ? IoOFP (sv_2io (fh)) : IoIFP (sv_2io (fh)));
547
548 if (fd < 0)
549 croak ("illegal fh argument, either not an OS file or read/write mode mismatch");
550
551 return fd;
552}
553
554MODULE = IO::AIO PACKAGE = IO::AIO 565MODULE = IO::AIO PACKAGE = IO::AIO
555 566
556PROTOTYPES: ENABLE 567PROTOTYPES: ENABLE
557 568
558BOOT: 569BOOT:
559{ 570{
560 stash = gv_stashpv ("IO::AIO", 1);
561
562 static const struct { 571 static const struct {
563 const char *name; 572 const char *name;
564 IV iv; 573 IV iv;
565 } *civ, const_iv[] = { 574 } *civ, const_iv[] = {
566# define const_iv(name, value) { # name, (IV) value }, 575# define const_iv(name, value) { # name, (IV) value },
572 const_iv (O_CREAT , O_CREAT) 581 const_iv (O_CREAT , O_CREAT)
573 const_iv (O_TRUNC , O_TRUNC) 582 const_iv (O_TRUNC , O_TRUNC)
574#ifndef _WIN32 583#ifndef _WIN32
575 const_iv (S_IFIFO , S_IFIFO) 584 const_iv (S_IFIFO , S_IFIFO)
576#endif 585#endif
586 const_iv (FADV_NORMAL , POSIX_FADV_NORMAL)
587 const_iv (FADV_SEQUENTIAL, POSIX_FADV_SEQUENTIAL)
588 const_iv (FADV_RANDOM , POSIX_FADV_RANDOM)
589 const_iv (FADV_NOREUSE , POSIX_FADV_NOREUSE)
590 const_iv (FADV_WILLNEED , POSIX_FADV_WILLNEED)
591 const_iv (FADV_DONTNEED , POSIX_FADV_DONTNEED)
592
577 const_eio (SYNC_FILE_RANGE_WAIT_BEFORE) 593 const_eio (SYNC_FILE_RANGE_WAIT_BEFORE)
578 const_eio (SYNC_FILE_RANGE_WRITE) 594 const_eio (SYNC_FILE_RANGE_WRITE)
579 const_eio (SYNC_FILE_RANGE_WAIT_AFTER) 595 const_eio (SYNC_FILE_RANGE_WAIT_AFTER)
580 596
581 const_eio (READDIR_DENTS) 597 const_eio (READDIR_DENTS)
592 const_eio (DT_LNK) 608 const_eio (DT_LNK)
593 const_eio (DT_SOCK) 609 const_eio (DT_SOCK)
594 const_eio (DT_WHT) 610 const_eio (DT_WHT)
595 }; 611 };
596 612
613 stash = gv_stashpv ("IO::AIO", 1);
614
597 for (civ = const_iv + sizeof (const_iv) / sizeof (const_iv [0]); civ-- > const_iv; ) 615 for (civ = const_iv + sizeof (const_iv) / sizeof (const_iv [0]); civ-- > const_iv; )
598 newCONSTSUB (stash, (char *)civ->name, newSViv (civ->iv)); 616 newCONSTSUB (stash, (char *)civ->name, newSViv (civ->iv));
599 617
600 create_respipe (); 618 create_respipe ();
601 619
664 ALIAS: 682 ALIAS:
665 aio_fsync = EIO_FSYNC 683 aio_fsync = EIO_FSYNC
666 aio_fdatasync = EIO_FDATASYNC 684 aio_fdatasync = EIO_FDATASYNC
667 PPCODE: 685 PPCODE:
668{ 686{
669 int fd = extract_fd (fh, 0); 687 int fd = s_fileno_croak (fh, 0);
670 dREQ; 688 dREQ;
671 689
672 req->type = ix; 690 req->type = ix;
673 req->sv1 = newSVsv (fh); 691 req->sv1 = newSVsv (fh);
674 req->int1 = fd; 692 req->int1 = fd;
675 693
676 REQ_SEND (req); 694 REQ_SEND (req);
677} 695}
678 696
679void 697void
680aio_sync_file_range (SV *fh, SV *offset, SV *nbytes, IV flags, SV *callback=&PL_sv_undef) 698aio_sync_file_range (SV *fh, off_t offset, size_t nbytes, UV flags, SV *callback=&PL_sv_undef)
681 PROTOTYPE: $$$$;$ 699 PROTOTYPE: $$$$;$
682 PPCODE: 700 PPCODE:
683{ 701{
684 int fd = extract_fd (fh, 0); 702 int fd = s_fileno_croak (fh, 0);
685 dREQ; 703 dREQ;
686 704
687 req->type = EIO_SYNC_FILE_RANGE; 705 req->type = EIO_SYNC_FILE_RANGE;
688 req->sv1 = newSVsv (fh); 706 req->sv1 = newSVsv (fh);
689 req->int1 = fd; 707 req->int1 = fd;
690 req->offs = SvVAL64 (offset); 708 req->offs = offset;
691 req->size = SvVAL64 (nbytes); 709 req->size = nbytes;
692 req->int2 = flags; 710 req->int2 = flags;
693 711
694 REQ_SEND (req); 712 REQ_SEND (req);
695} 713}
696 714
698aio_close (SV *fh, SV *callback=&PL_sv_undef) 716aio_close (SV *fh, SV *callback=&PL_sv_undef)
699 PROTOTYPE: $;$ 717 PROTOTYPE: $;$
700 PPCODE: 718 PPCODE:
701{ 719{
702 static int close_pipe = -1; /* dummy fd to close fds via dup2 */ 720 static int close_pipe = -1; /* dummy fd to close fds via dup2 */
703 int fd = extract_fd (fh, 0); 721 int fd = s_fileno_croak (fh, 0);
704 dREQ; 722 dREQ;
705 723
706 if (close_pipe < 0) 724 if (close_pipe < 0)
707 { 725 {
708 int pipefd [2]; 726 int pipefd [2];
730 aio_write = EIO_WRITE 748 aio_write = EIO_WRITE
731 PROTOTYPE: $$$$$;$ 749 PROTOTYPE: $$$$$;$
732 PPCODE: 750 PPCODE:
733{ 751{
734 STRLEN svlen; 752 STRLEN svlen;
735 int fd = extract_fd (fh, ix == EIO_WRITE); 753 int fd = s_fileno_croak (fh, ix == EIO_WRITE);
736 char *svptr = SvPVbyte (data, svlen); 754 char *svptr = SvPVbyte (data, svlen);
737 UV len = SvUV (length); 755 UV len = SvUV (length);
738 756
739 if (dataoffset < 0) 757 if (dataoffset < 0)
740 dataoffset += svlen; 758 dataoffset += svlen;
791 809
792 REQ_SEND; 810 REQ_SEND;
793} 811}
794 812
795void 813void
796aio_sendfile (SV *out_fh, SV *in_fh, SV *in_offset, UV length, SV *callback=&PL_sv_undef) 814aio_sendfile (SV *out_fh, SV *in_fh, off_t in_offset, size_t length, SV *callback=&PL_sv_undef)
797 PROTOTYPE: $$$$;$ 815 PROTOTYPE: $$$$;$
798 PPCODE: 816 PPCODE:
799{ 817{
800 int ifd = extract_fd (in_fh , 0); 818 int ifd = s_fileno_croak (in_fh , 0);
801 int ofd = extract_fd (out_fh, 0); 819 int ofd = s_fileno_croak (out_fh, 1);
802 dREQ; 820 dREQ;
803 821
804 req->type = EIO_SENDFILE; 822 req->type = EIO_SENDFILE;
805 req->sv1 = newSVsv (out_fh); 823 req->sv1 = newSVsv (out_fh);
806 req->int1 = ofd; 824 req->int1 = ofd;
807 req->sv2 = newSVsv (in_fh); 825 req->sv2 = newSVsv (in_fh);
808 req->int2 = ifd; 826 req->int2 = ifd;
809 req->offs = SvVAL64 (in_offset); 827 req->offs = in_offset;
810 req->size = length; 828 req->size = length;
811 829
812 REQ_SEND; 830 REQ_SEND;
813} 831}
814 832
815void 833void
816aio_readahead (SV *fh, SV *offset, IV length, SV *callback=&PL_sv_undef) 834aio_readahead (SV *fh, off_t offset, size_t length, SV *callback=&PL_sv_undef)
817 PROTOTYPE: $$$;$ 835 PROTOTYPE: $$$;$
818 PPCODE: 836 PPCODE:
819{ 837{
820 int fd = extract_fd (fh, 0); 838 int fd = s_fileno_croak (fh, 0);
821 dREQ; 839 dREQ;
822 840
823 req->type = EIO_READAHEAD; 841 req->type = EIO_READAHEAD;
824 req->sv1 = newSVsv (fh); 842 req->sv1 = newSVsv (fh);
825 req->int1 = fd; 843 req->int1 = fd;
826 req->offs = SvVAL64 (offset); 844 req->offs = offset;
827 req->size = length; 845 req->size = length;
828 846
829 REQ_SEND; 847 REQ_SEND;
830} 848}
831 849
1151 CODE: 1169 CODE:
1152 RETVAL = eio_nthreads (); 1170 RETVAL = eio_nthreads ();
1153 OUTPUT: 1171 OUTPUT:
1154 RETVAL 1172 RETVAL
1155 1173
1174int
1175fadvise (aio_rfd fh, off_t offset, off_t length, IV advice)
1176 PROTOTYPE: $$$$
1177 CODE:
1178#if _XOPEN_SOURCE >= 600 && !NO_FADVISE
1179 RETVAL = posix_fadvise (fh, offset, length, advice);
1180#else
1181 RETVAL = errno = ENOSYS;
1182#endif
1183 OUTPUT:
1184 RETVAL
1185
1186ssize_t
1187sendfile (aio_wfd ofh, aio_rfd ifh, off_t offset, size_t count)
1188 PROTOTYPE: $$$$
1189 CODE:
1190 eio_sendfile_sync (ofh, ifh, offset, count);
1191
1156void _on_next_submit (SV *cb) 1192void _on_next_submit (SV *cb)
1157 CODE: 1193 CODE:
1158 SvREFCNT_dec (on_next_submit); 1194 SvREFCNT_dec (on_next_submit);
1159 on_next_submit = SvOK (cb) ? newSVsv (cb) : 0; 1195 on_next_submit = SvOK (cb) ? newSVsv (cb) : 0;
1160 1196
1174 if (GIMME_V != G_VOID) 1210 if (GIMME_V != G_VOID)
1175 XPUSHs (req->callback ? sv_2mortal (newRV_inc (req->callback)) : &PL_sv_undef); 1211 XPUSHs (req->callback ? sv_2mortal (newRV_inc (req->callback)) : &PL_sv_undef);
1176 1212
1177 if (items > 1) 1213 if (items > 1)
1178 { 1214 {
1179 SV *cb_cv = get_cb (callback); 1215 SV *cb_cv =get_cb (callback);
1180 1216
1181 SvREFCNT_dec (req->callback); 1217 SvREFCNT_dec (req->callback);
1182 req->callback = SvREFCNT_inc (cb_cv); 1218 req->callback = SvREFCNT_inc (cb_cv);
1183 } 1219 }
1184} 1220}

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines