ViewVC Help
View File | Revision Log | Show Annotations | Download File
/cvs/IO-AIO/AIO.xs
Revision: 1.209
Committed: Sun Apr 1 17:46:02 2012 UTC (12 years, 1 month ago) by root
Branch: MAIN
Changes since 1.208: +29 -4 lines
Log Message:
*** empty log message ***

File Contents

# User Rev Content
1 root 1.115 #include "libeio/xthread.h"
2 root 1.63
3 root 1.19 #include <errno.h>
4    
5 root 1.15 #include "EXTERN.h"
6 root 1.1 #include "perl.h"
7     #include "XSUB.h"
8    
9 root 1.151 #include "schmorp.h"
10    
11 root 1.37 #include <stddef.h>
12 root 1.94 #include <stdlib.h>
13 root 1.41 #include <errno.h>
14 root 1.1 #include <sys/types.h>
15     #include <sys/stat.h>
16 root 1.37 #include <limits.h>
17 root 1.1 #include <fcntl.h>
18     #include <sched.h>
19 root 1.103
20 root 1.178 #if _POSIX_MEMLOCK || _POSIX_MEMLOCK_RANGE || _POSIX_MAPPED_FILES
21 root 1.161 # include <sys/mman.h>
22     #endif
23    
24 root 1.118 /* perl namespace pollution */
25     #undef VERSION
26    
27 root 1.187 /* perl stupidly overrides readdir and maybe others */
28     /* with thread-unsafe versions, imagine that :( */
29     #undef readdir
30     #undef opendir
31     #undef closedir
32    
33 root 1.103 #ifdef _WIN32
34    
35 root 1.189 // perl overrides all those nice libc functions
36    
37 root 1.188 #undef malloc
38     #undef free
39     #undef open
40     #undef read
41     #undef write
42     #undef send
43     #undef recv
44     #undef stat
45     #undef lstat
46     #undef fstat
47     #undef truncate
48     #undef ftruncate
49     #undef open
50     #undef link
51     #undef close
52     #undef unlink
53     #undef mkdir
54     #undef rmdir
55     #undef rename
56     #undef lseek
57     #undef opendir
58     #undef readdir
59     #undef closedir
60     #undef chmod
61     #undef fchmod
62     #undef dup
63     #undef dup2
64     #undef abort
65     #undef pipe
66 root 1.103
67 root 1.208 #define EIO_STRUCT_STAT struct _stati64
68     #define EIO_STRUCT_STATI64
69    
70 root 1.103 #else
71    
72 root 1.188 #include <sys/time.h>
73     #include <sys/select.h>
74     #include <unistd.h>
75     #include <utime.h>
76     #include <signal.h>
77 root 1.103
78 root 1.208 #define EIO_STRUCT_STAT Stat_t
79    
80 root 1.103 #endif
81 root 1.1
82 root 1.101 /* use NV for 32 bit perls as it allows larger offsets */
83     #if IVSIZE >= 8
84 root 1.148 # define VAL64 IV
85 root 1.101 # define SvVAL64 SvIV
86 root 1.148 # define newSVval64 newSViv
87 root 1.101 #else
88 root 1.148 # define VAL64 NV
89 root 1.101 # define SvVAL64 SvNV
90 root 1.148 # define newSVval64 newSVnv
91 root 1.101 #endif
92    
93 root 1.127 /*****************************************************************************/
94    
95     #if __GNUC__ >= 3
96     # define expect(expr,value) __builtin_expect ((expr),(value))
97     #else
98     # define expect(expr,value) (expr)
99     #endif
100    
101     #define expect_false(expr) expect ((expr) != 0, 0)
102     #define expect_true(expr) expect ((expr) != 0, 1)
103    
104     /*****************************************************************************/
105    
106 root 1.90 typedef SV SV8; /* byte-sv, used for argument-checking */
107 root 1.148 typedef int aio_rfd; /* read file desriptor */
108     typedef int aio_wfd; /* write file descriptor */
109 root 1.90
110 root 1.195 static HV *aio_stash, *aio_req_stash, *aio_grp_stash, *aio_wd_stash;
111 root 1.43
112 root 1.121 #define EIO_REQ_MEMBERS \
113 root 1.115 SV *callback; \
114     SV *sv1, *sv2; \
115 root 1.195 SV *sv3, *sv4; \
116 root 1.115 STRLEN stroffset; \
117 root 1.121 SV *self;
118 root 1.115
119 root 1.122 #define EIO_NO_WRAPPERS 1
120    
121 root 1.176 #include "libeio/config.h"
122 root 1.115 #include "libeio/eio.h"
123    
124 root 1.188 static int req_invoke (eio_req *req);
125     #define EIO_FINISH(req) req_invoke (req)
126     static void req_destroy (eio_req *grp);
127     #define EIO_DESTROY(req) req_destroy (req)
128    
129     #include "libeio/eio.c"
130    
131 root 1.182 /* Linux/others */
132     #ifndef O_ASYNC
133     # define O_ASYNC 0
134     #endif
135     #ifndef O_DIRECT
136     # define O_DIRECT 0
137     #endif
138     #ifndef O_NOATIME
139     # define O_NOATIME 0
140     #endif
141    
142     /* POSIX */
143     #ifndef O_CLOEXEC
144     # define O_CLOEXEC 0
145     #endif
146     #ifndef O_NOFOLLOW
147     # define O_NOFOLLOW 0
148     #endif
149     #ifndef O_NOCTTY
150     # define O_NOCTTY 0
151     #endif
152     #ifndef O_NONBLOCK
153     # define O_NONBLOCK 0
154     #endif
155     #ifndef O_EXEC
156     # define O_EXEC 0
157     #endif
158     #ifndef O_SEARCH
159     # define O_SEARCH 0
160     #endif
161     #ifndef O_DIRECTORY
162     # define O_DIRECTORY 0
163     #endif
164     #ifndef O_DSYNC
165     # define O_DSYNC 0
166     #endif
167     #ifndef O_RSYNC
168     # define O_RSYNC 0
169     #endif
170     #ifndef O_SYNC
171     # define O_SYNC 0
172     #endif
173     #ifndef O_TTY_INIT
174     # define O_TTY_INIT 0
175     #endif
176    
177 root 1.148 #ifndef POSIX_FADV_NORMAL
178     # define POSIX_FADV_NORMAL 0
179     #endif
180     #ifndef POSIX_FADV_SEQUENTIAL
181     # define POSIX_FADV_SEQUENTIAL 0
182     #endif
183     #ifndef POSIX_FADV_RANDOM
184     # define POSIX_FADV_RANDOM 0
185     #endif
186     #ifndef POSIX_FADV_NOREUSE
187     # define POSIX_FADV_NOREUSE 0
188     #endif
189     #ifndef POSIX_FADV_WILLNEED
190     # define POSIX_FADV_WILLNEED 0
191     #endif
192     #ifndef POSIX_FADV_DONTNEED
193     # define POSIX_FADV_DONTNEED 0
194     #endif
195    
196 root 1.175 #if !HAVE_POSIX_FADVISE
197 root 1.170 # define posix_fadvise(a,b,c,d) errno = ENOSYS /* also return ENOSYS */
198     #endif
199    
200 root 1.169 #ifndef POSIX_MADV_NORMAL
201     # define POSIX_MADV_NORMAL 0
202     #endif
203     #ifndef POSIX_MADV_SEQUENTIAL
204     # define POSIX_MADV_SEQUENTIAL 0
205     #endif
206     #ifndef POSIX_MADV_RANDOM
207     # define POSIX_MADV_RANDOM 0
208     #endif
209     #ifndef POSIX_MADV_WILLNEED
210     # define POSIX_MADV_WILLNEED 0
211     #endif
212     #ifndef POSIX_MADV_DONTNEED
213     # define POSIX_MADV_DONTNEED 0
214     #endif
215    
216 root 1.175 #if !HAVE_POSIX_MADVISE
217 root 1.170 # define posix_madvise(a,b,c) errno = ENOSYS /* also return ENOSYS */
218     #endif
219    
220 root 1.171 #ifndef PROT_NONE
221     # define PROT_NONE 0
222     #endif
223     #ifndef PROT_READ
224     # define PROT_READ 0
225     #endif
226     #ifndef PROT_WRITE
227     # define PROT_READ 0
228     #endif
229     #ifndef PROT_EXEC
230     # define PROT_EXEC 0
231     #endif
232    
233 root 1.187 #ifndef ST_RDONLY
234     # define ST_RDONLY 0
235     #endif
236     #ifndef ST_NOSUID
237     # define ST_NOSUID 0
238     #endif
239 root 1.159 #ifndef ST_NODEV
240     # define ST_NODEV 0
241     #endif
242     #ifndef ST_NOEXEC
243     # define ST_NOEXEC 0
244     #endif
245     #ifndef ST_SYNCHRONOUS
246     # define ST_SYNCHRONOUS 0
247     #endif
248     #ifndef ST_MANDLOCK
249     # define ST_MANDLOCK 0
250     #endif
251     #ifndef ST_WRITE
252     # define ST_WRITE 0
253     #endif
254     #ifndef ST_APPEND
255     # define ST_APPEND 0
256     #endif
257     #ifndef ST_IMMUTABLE
258     # define ST_IMMUTABLE 0
259     #endif
260     #ifndef ST_NOATIME
261     # define ST_NOATIME 0
262     #endif
263     #ifndef ST_NODIRATIME
264     # define ST_NODIRATIME 0
265     #endif
266     #ifndef ST_RELATIME
267     # define ST_RELATIME 0
268     #endif
269    
270 root 1.180 #ifndef S_IFIFO
271     # define S_IFIFO 0
272     #endif
273     #ifndef S_IFCHR
274     # define S_IFCHR 0
275     #endif
276     #ifndef S_IFBLK
277     # define S_IFBLK 0
278     #endif
279     #ifndef S_IFLNK
280     # define S_IFLNK 0
281     #endif
282     #ifndef S_IFREG
283     # define S_IFREG 0
284     #endif
285     #ifndef S_IFDIR
286     # define S_IFDIR 0
287     #endif
288     #ifndef S_IFWHT
289     # define S_IFWHT 0
290     #endif
291     #ifndef S_IFSOCK
292     # define S_IFSOCK 0
293     #endif
294    
295 root 1.162 #ifndef MAP_ANONYMOUS
296     # ifdef MAP_ANON
297     # define MAP_ANONYMOUS MAP_ANON
298     # else
299     # define MAP_ANONYMOUS MAP_FIXED /* and hope this fails */
300     # endif
301     #endif
302     #ifndef MAP_HUGETLB
303     # define MAP_HUGETLB 0
304     #endif
305     #ifndef MAP_LOCKED
306     # define MAP_LOCKED 0
307     #endif
308     #ifndef MAP_NORESERVE
309     # define MAP_NORESERVE 0
310     #endif
311     #ifndef MAP_POPULATE
312     # define MAP_POPULATE 0
313     #endif
314     #ifndef MAP_NONBLOCK
315     # define MAP_NONBLOCK 0
316     #endif
317    
318 root 1.180 #ifndef makedev
319     # define makedev(maj,min) (((maj) << 8) | (min))
320     #endif
321     #ifndef major
322     # define major(dev) ((dev) >> 8)
323     #endif
324     #ifndef minor
325     # define minor(dev) ((dev) & 0xff)
326     #endif
327    
328 root 1.162 #ifndef PAGESIZE
329     # define PAGESIZE sysconf (_SC_PAGESIZE)
330     #endif
331    
332 root 1.58 enum {
333 root 1.115 FLAG_SV2_RO_OFF = 0x40, /* data was set readonly */
334 root 1.58 };
335    
336 root 1.115 typedef eio_req *aio_req;
337     typedef eio_req *aio_req_ornot;
338 root 1.196 typedef eio_wd aio_wd;
339 root 1.60
340 root 1.117 static SV *on_next_submit;
341 root 1.121 static int next_pri = EIO_PRI_DEFAULT;
342 root 1.117 static int max_outstanding;
343 root 1.94
344 root 1.153 static s_epipe respipe;
345 root 1.80
346 root 1.115 static void req_destroy (aio_req req);
347     static void req_cancel (aio_req req);
348 root 1.80
349 root 1.201 static void
350     want_poll (void)
351 root 1.80 {
352 root 1.115 /* write a dummy byte to the pipe so fh becomes ready */
353 root 1.153 s_epipe_signal (&respipe);
354 root 1.67 }
355    
356 root 1.201 static void
357     done_poll (void)
358 root 1.67 {
359 root 1.115 /* read any signals sent by the worker threads */
360 root 1.153 s_epipe_drain (&respipe);
361 root 1.67 }
362 root 1.1
363 root 1.43 /* must be called at most once */
364 root 1.201 static SV *
365     req_sv (aio_req req, HV *stash)
366 root 1.43 {
367 root 1.49 if (!req->self)
368     {
369     req->self = (SV *)newHV ();
370     sv_magic (req->self, 0, PERL_MAGIC_ext, (char *)req, 0);
371     }
372 root 1.43
373 root 1.156 return sv_2mortal (sv_bless (newRV_inc (req->self), stash));
374 root 1.43 }
375    
376 root 1.196 static SV *
377     newSVaio_wd (aio_wd wd)
378     {
379 root 1.203 return sv_bless (newRV_noinc (newSViv ((IV)wd)), aio_wd_stash);
380 root 1.196 }
381    
382 root 1.201 static aio_req
383     SvAIO_REQ (SV *sv)
384 root 1.26 {
385 root 1.53 MAGIC *mg;
386    
387 root 1.156 if (!SvROK (sv)
388 root 1.204 /* for speed reasons, we do not verify that SvROK actually has a stash ptr */
389 root 1.156 || (SvSTASH (SvRV (sv)) != aio_grp_stash
390     && SvSTASH (SvRV (sv)) != aio_req_stash
391     && !sv_derived_from (sv, "IO::AIO::REQ")))
392     croak ("object of class IO::AIO::REQ expected");
393 root 1.43
394 root 1.53 mg = mg_find (SvRV (sv), PERL_MAGIC_ext);
395 root 1.43
396     return mg ? (aio_req)mg->mg_ptr : 0;
397     }
398    
399 root 1.201 static aio_wd
400     SvAIO_WD (SV *sv)
401 root 1.195 {
402     if (!SvROK (sv)
403 root 1.204 || SvTYPE (SvRV (sv)) != SVt_PVMG
404     || SvSTASH (SvRV (sv)) != aio_wd_stash)
405 root 1.195 croak ("IO::AIO: expected a working directory object as returned by aio_wd");
406    
407     return (aio_wd)(long)SvIVX (SvRV (sv));
408     }
409    
410 root 1.201 static void
411     aio_grp_feed (aio_req grp)
412 root 1.49 {
413 root 1.115 if (grp->sv2 && SvOK (grp->sv2))
414 root 1.49 {
415 root 1.115 dSP;
416 root 1.49
417 root 1.115 ENTER;
418     SAVETMPS;
419     PUSHMARK (SP);
420 root 1.156 XPUSHs (req_sv (grp, aio_grp_stash));
421 root 1.115 PUTBACK;
422     call_sv (grp->sv2, G_VOID | G_EVAL | G_KEEPERR);
423     SPAGAIN;
424     FREETMPS;
425     LEAVE;
426 root 1.50 }
427     }
428    
429 root 1.201 static void
430     req_submit (eio_req *req)
431 root 1.117 {
432     eio_submit (req);
433    
434 root 1.127 if (expect_false (on_next_submit))
435 root 1.117 {
436     dSP;
437     SV *cb = sv_2mortal (on_next_submit);
438    
439     on_next_submit = 0;
440    
441     PUSHMARK (SP);
442     PUTBACK;
443     call_sv (cb, G_DISCARD | G_EVAL);
444     }
445     }
446    
447 root 1.201 static int
448     req_invoke (eio_req *req)
449 root 1.45 {
450 root 1.100 if (req->flags & FLAG_SV2_RO_OFF)
451     SvREADONLY_off (req->sv2);
452 root 1.86
453 root 1.128 if (!EIO_CANCELLED (req) && req->callback)
454 root 1.67 {
455 root 1.166 dSP;
456 root 1.165 static SV *sv_result_cache; /* caches the result integer SV */
457     SV *sv_result;
458    
459 root 1.67 ENTER;
460     SAVETMPS;
461     PUSHMARK (SP);
462     EXTEND (SP, 1);
463 root 1.45
464 root 1.165 /* do not recreate the result IV from scratch each time */
465     if (expect_true (sv_result_cache))
466     {
467     sv_result = sv_result_cache; sv_result_cache = 0;
468     SvIV_set (sv_result, req->result);
469 root 1.177 SvIOK_only (sv_result);
470 root 1.165 }
471     else
472     {
473     sv_result = newSViv (req->result);
474     SvREADONLY_on (sv_result);
475     }
476    
477 root 1.67 switch (req->type)
478 root 1.45 {
479 root 1.195 case EIO_WD_OPEN:
480 root 1.203 PUSHs (req->result ? &PL_sv_undef : sv_2mortal (newSVaio_wd (req->wd)));
481 root 1.195 break;
482    
483 root 1.115 case EIO_READDIR:
484 root 1.45 {
485 root 1.67 SV *rv = &PL_sv_undef;
486 root 1.45
487 root 1.67 if (req->result >= 0)
488 root 1.45 {
489 root 1.71 int i;
490 root 1.147 char *names = (char *)req->ptr2;
491     eio_dirent *ent = (eio_dirent *)req->ptr1; /* might be 0 */
492 root 1.67 AV *av = newAV ();
493    
494 root 1.71 av_extend (av, req->result - 1);
495    
496     for (i = 0; i < req->result; ++i)
497 root 1.67 {
498 root 1.141 if (req->int1 & EIO_READDIR_DENTS)
499     {
500 root 1.147 SV *namesv = newSVpvn (names + ent->nameofs, ent->namelen);
501 root 1.141
502     if (req->int1 & EIO_READDIR_CUSTOM2)
503     {
504     static SV *sv_type [EIO_DT_MAX + 1]; /* type sv cache */
505     AV *avent = newAV ();
506    
507     av_extend (avent, 2);
508    
509     if (!sv_type [ent->type])
510     {
511     sv_type [ent->type] = newSViv (ent->type);
512     SvREADONLY_on (sv_type [ent->type]);
513     }
514    
515     av_store (avent, 0, namesv);
516 root 1.143 av_store (avent, 1, SvREFCNT_inc (sv_type [ent->type]));
517     av_store (avent, 2, IVSIZE >= 8 ? newSVuv (ent->inode) : newSVnv (ent->inode));
518 root 1.141
519     av_store (av, i, newRV_noinc ((SV *)avent));
520     }
521     else
522     av_store (av, i, namesv);
523    
524 root 1.147 ++ent;
525 root 1.141 }
526     else
527     {
528 root 1.147 SV *name = newSVpv (names, 0);
529 root 1.141 av_store (av, i, name);
530 root 1.147 names += SvCUR (name) + 1;
531 root 1.141 }
532 root 1.67 }
533 root 1.45
534 root 1.67 rv = sv_2mortal (newRV_noinc ((SV *)av));
535 root 1.45 }
536    
537 root 1.67 PUSHs (rv);
538 root 1.141
539     if (req->int1 & EIO_READDIR_CUSTOM1)
540     XPUSHs (sv_2mortal (newSViv (req->int1 & ~(EIO_READDIR_CUSTOM1 | EIO_READDIR_CUSTOM2))));
541 root 1.45 }
542 root 1.67 break;
543 root 1.45
544 root 1.115 case EIO_OPEN:
545 root 1.67 {
546     /* convert fd to fh */
547 root 1.107 SV *fh = &PL_sv_undef;
548    
549     if (req->result >= 0)
550     {
551     GV *gv = (GV *)sv_newmortal ();
552     int flags = req->int1 & (O_RDONLY | O_WRONLY | O_RDWR);
553     char sym [64];
554     int symlen;
555    
556 root 1.146 symlen = snprintf (sym, sizeof (sym), "fd#%d", (int)req->result);
557 root 1.156 gv_init (gv, aio_stash, sym, symlen, 0);
558 root 1.107
559     symlen = snprintf (
560     sym,
561     sizeof (sym),
562 root 1.146 "%s&=%d",
563 root 1.107 flags == O_RDONLY ? "<" : flags == O_WRONLY ? ">" : "+<",
564 root 1.146 (int)req->result
565 root 1.107 );
566 root 1.45
567 root 1.107 if (do_open (gv, sym, symlen, 0, 0, 0, 0))
568 root 1.108 fh = (SV *)gv;
569 root 1.107 }
570 root 1.45
571 root 1.109 PUSHs (fh);
572 root 1.67 }
573     break;
574 root 1.45
575 root 1.159 case EIO_STATVFS:
576     case EIO_FSTATVFS:
577     {
578     SV *rv = &PL_sv_undef;
579    
580 root 1.187 #ifndef _WIN32
581 root 1.159 if (req->result >= 0)
582     {
583     EIO_STRUCT_STATVFS *f = EIO_STATVFS_BUF (req);
584     HV *hv = newHV ();
585    
586 root 1.160 rv = sv_2mortal (newRV_noinc ((SV *)hv));
587 root 1.159
588     hv_store (hv, "bsize" , sizeof ("bsize" ) - 1, newSVval64 (f->f_bsize ), 0);
589     hv_store (hv, "frsize" , sizeof ("frsize" ) - 1, newSVval64 (f->f_frsize ), 0);
590     hv_store (hv, "blocks" , sizeof ("blocks" ) - 1, newSVval64 (f->f_blocks ), 0);
591     hv_store (hv, "bfree" , sizeof ("bfree" ) - 1, newSVval64 (f->f_bfree ), 0);
592     hv_store (hv, "bavail" , sizeof ("bavail" ) - 1, newSVval64 (f->f_bavail ), 0);
593     hv_store (hv, "files" , sizeof ("files" ) - 1, newSVval64 (f->f_files ), 0);
594     hv_store (hv, "ffree" , sizeof ("ffree" ) - 1, newSVval64 (f->f_ffree ), 0);
595     hv_store (hv, "favail" , sizeof ("favail" ) - 1, newSVval64 (f->f_favail ), 0);
596     hv_store (hv, "fsid" , sizeof ("fsid" ) - 1, newSVval64 (f->f_fsid ), 0);
597     hv_store (hv, "flag" , sizeof ("flag" ) - 1, newSVval64 (f->f_flag ), 0);
598     hv_store (hv, "namemax", sizeof ("namemax") - 1, newSVval64 (f->f_namemax), 0);
599     }
600 root 1.187 #endif
601 root 1.159
602     PUSHs (rv);
603     }
604    
605     break;
606    
607 root 1.115 case EIO_GROUP:
608 root 1.86 req->int1 = 2; /* mark group as finished */
609 root 1.45
610 root 1.86 if (req->sv1)
611 root 1.67 {
612     int i;
613 root 1.86 AV *av = (AV *)req->sv1;
614 root 1.49
615 root 1.67 EXTEND (SP, AvFILL (av) + 1);
616     for (i = 0; i <= AvFILL (av); ++i)
617     PUSHs (*av_fetch (av, i, 0));
618     }
619     break;
620 root 1.48
621 root 1.115 case EIO_NOP:
622     case EIO_BUSY:
623 root 1.67 break;
624 root 1.48
625 root 1.115 case EIO_READLINK:
626 root 1.183 case EIO_REALPATH:
627 root 1.86 if (req->result > 0)
628 root 1.119 PUSHs (sv_2mortal (newSVpvn (req->ptr2, req->result)));
629 root 1.86 break;
630    
631 root 1.115 case EIO_STAT:
632     case EIO_LSTAT:
633     case EIO_FSTAT:
634 root 1.197 PL_laststype = req->type == EIO_LSTAT ? OP_LSTAT : OP_STAT;
635    
636     if (!(PL_laststatval = req->result))
637     /* if compilation fails here then perl's Stat_t is not struct _stati64 */
638     PL_statcache = *(EIO_STRUCT_STAT *)(req->ptr2);
639    
640 root 1.165 PUSHs (sv_result);
641 root 1.87 break;
642    
643 root 1.209 case EIO_SEEK:
644     PUSHs (req->result ? sv_result : sv_2mortal (newSVval64 (req->offs)));
645     break;
646    
647 root 1.115 case EIO_READ:
648 root 1.138 {
649     SvCUR_set (req->sv2, req->stroffset + (req->result > 0 ? req->result : 0));
650     *SvEND (req->sv2) = 0;
651     SvPOK_only (req->sv2);
652     SvSETMAGIC (req->sv2);
653 root 1.165 PUSHs (sv_result);
654 root 1.138 }
655 root 1.87 break;
656    
657 root 1.177 case EIO_DUP2: /* EIO_DUP2 actually means aio_close(), so fudge result value */
658 root 1.115 if (req->result > 0)
659 root 1.165 SvIV_set (sv_result, 0);
660 root 1.115 /* FALLTHROUGH */
661    
662 root 1.67 default:
663 root 1.165 PUSHs (sv_result);
664 root 1.67 break;
665     }
666 root 1.45
667 root 1.79 errno = req->errorno;
668 root 1.51
669 root 1.67 PUTBACK;
670 root 1.109 call_sv (req->callback, G_VOID | G_EVAL | G_DISCARD);
671 root 1.67 SPAGAIN;
672 root 1.51
673 root 1.165 if (expect_false (SvREFCNT (sv_result) != 1 || sv_result_cache))
674     SvREFCNT_dec (sv_result);
675     else
676     sv_result_cache = sv_result;
677    
678 root 1.67 FREETMPS;
679     LEAVE;
680 root 1.109
681     PUTBACK;
682 root 1.45 }
683    
684 root 1.115 return !!SvTRUE (ERRSV);
685 root 1.67 }
686    
687 root 1.201 static void
688     req_destroy (aio_req req)
689 root 1.67 {
690 root 1.43 if (req->self)
691     {
692     sv_unmagic (req->self, PERL_MAGIC_ext);
693     SvREFCNT_dec (req->self);
694     }
695    
696 root 1.86 SvREFCNT_dec (req->sv1);
697     SvREFCNT_dec (req->sv2);
698 root 1.195 SvREFCNT_dec (req->sv3);
699     SvREFCNT_dec (req->sv4);
700 root 1.49 SvREFCNT_dec (req->callback);
701 root 1.26
702 root 1.204 free (req);
703 root 1.26 }
704    
705 root 1.201 static void
706     req_cancel_subs (aio_req grp)
707 root 1.72 {
708 root 1.115 if (grp->type != EIO_GROUP)
709 root 1.72 return;
710    
711 root 1.86 SvREFCNT_dec (grp->sv2);
712     grp->sv2 = 0;
713 root 1.72
714 root 1.115 eio_grp_cancel (grp);
715 root 1.1 }
716    
717 root 1.201 static void ecb_cold
718     create_respipe (void)
719 root 1.104 {
720 root 1.153 if (s_epipe_renew (&respipe))
721 root 1.155 croak ("IO::AIO: unable to initialize result pipe");
722 root 1.104 }
723    
724 root 1.201 static void
725     poll_wait (void)
726 root 1.80 {
727 root 1.115 while (eio_nreqs ())
728 root 1.30 {
729 root 1.80 int size;
730 root 1.115
731     X_LOCK (reslock);
732 root 1.80 size = res_queue.size;
733 root 1.115 X_UNLOCK (reslock);
734 root 1.80
735     if (size)
736     return;
737    
738 root 1.123 etp_maybe_start_thread ();
739 root 1.80
740 root 1.153 s_epipe_wait (&respipe);
741 root 1.22 }
742 root 1.80 }
743    
744 root 1.201 static int
745     poll_cb (void)
746 root 1.80 {
747 root 1.124 for (;;)
748 root 1.116 {
749 root 1.124 int res = eio_poll ();
750 root 1.116
751     if (res > 0)
752     croak (0);
753 root 1.124
754     if (!max_outstanding || max_outstanding > eio_nreqs ())
755     return res;
756    
757     poll_wait ();
758 root 1.116 }
759 root 1.17 }
760 root 1.37
761 root 1.189 static void ecb_cold
762 root 1.190 reinit (void)
763 root 1.116 {
764     create_respipe ();
765 root 1.189
766     if (eio_init (want_poll, done_poll) < 0)
767     croak ("IO::AIO: unable to initialise eio library");
768 root 1.116 }
769    
770 root 1.162 /*****************************************************************************/
771    
772 root 1.163 #if !_POSIX_MAPPED_FILES
773     # define mmap(addr,length,prot,flags,fd,offs) (errno = ENOSYS, -1)
774     # define munmap(addr,length) (errno = ENOSYS, -1)
775 root 1.179 #endif
776    
777     #if !_POSIX_MEMORY_PROTECTION
778 root 1.171 # define mprotect(addr,len,prot) (errno = ENOSYS, -1)
779 root 1.174 # define PROT_NONE 0
780     # define PROT_WRITE 0
781     # define MAP_PRIVATE 0
782     # define MAP_SHARED 0
783     # define MAP_FIXED 0
784 root 1.163 #endif
785    
786 root 1.162 #define MMAP_MAGIC PERL_MAGIC_ext
787    
788 root 1.201 static int ecb_cold
789     mmap_free (pTHX_ SV *sv, MAGIC *mg)
790 root 1.162 {
791     int old_errno = errno;
792     munmap (mg->mg_ptr, (size_t)mg->mg_obj);
793     errno = old_errno;
794    
795     mg->mg_obj = 0; /* just in case */
796    
797     SvREADONLY_off (sv);
798 root 1.168
799     if (SvPVX (sv) != mg->mg_ptr)
800     croak ("ERROR: IO::AIO::mmap-mapped scalar changed location, detected");
801    
802 root 1.162 SvCUR_set (sv, 0);
803     SvPVX (sv) = 0;
804     SvOK_off (sv);
805    
806     return 0;
807     }
808    
809     static MGVTBL mmap_vtbl = {
810     0, 0, 0, 0, mmap_free
811     };
812    
813     /*****************************************************************************/
814    
815 root 1.201 static SV *
816     get_cb (SV *cb_sv)
817 root 1.128 {
818 root 1.152 SvGETMAGIC (cb_sv);
819     return SvOK (cb_sv) ? s_get_cv_croak (cb_sv) : 0;
820 root 1.128 }
821    
822 root 1.203 static aio_req ecb_noinline
823     dreq (SV *callback)
824     {
825     SV *cb_cv;
826     aio_req req;
827     int req_pri = next_pri;
828     next_pri = EIO_PRI_DEFAULT;
829    
830     cb_cv = get_cb (callback);
831    
832 root 1.204 req = calloc (sizeof (*req), 1);
833 root 1.203 if (!req)
834     croak ("out of memory during eio_req allocation");
835    
836     req->callback = SvREFCNT_inc (cb_cv);
837     req->pri = req_pri;
838    
839     return req;
840     }
841    
842 root 1.22 #define dREQ \
843 root 1.203 aio_req req = dreq (callback); \
844 root 1.43
845     #define REQ_SEND \
846 root 1.126 PUTBACK; \
847 root 1.117 req_submit (req); \
848 root 1.126 SPAGAIN; \
849 root 1.43 \
850     if (GIMME_V != G_VOID) \
851 root 1.156 XPUSHs (req_sv (req, aio_req_stash));
852 root 1.136
853 root 1.201 ecb_inline void
854 root 1.195 req_set_path (aio_req req, SV *path, SV **wdsv, SV **pathsv, eio_wd *wd, void **ptr)
855     {
856 root 1.198 if (expect_false (SvROK (path)))
857 root 1.195 {
858 root 1.204 SV *rv = SvRV (path);
859 root 1.195 SV *wdob;
860    
861 root 1.204 if (SvTYPE (rv) == SVt_PVAV && AvFILLp (rv) == 1)
862     {
863     path = AvARRAY (rv)[1];
864     wdob = AvARRAY (rv)[0];
865 root 1.195
866 root 1.204 if (SvOK (wdob))
867     {
868     *wd = SvAIO_WD (wdob);
869     *wdsv = SvREFCNT_inc_NN (SvRV (wdob));
870     }
871     else
872     *wd = EIO_INVALID_WD;
873     }
874     else if (SvTYPE (rv) == SVt_PVMG && SvSTASH (rv) == aio_wd_stash)
875 root 1.196 {
876 root 1.206 *wd = (aio_wd)(long)SvIVX (rv);
877 root 1.204 *wdsv = SvREFCNT_inc_NN (rv);
878     *ptr = ".";
879     return; /* path set to "." */
880 root 1.196 }
881     else
882 root 1.205 croak ("IO::AIO: pathname arguments must be specified as a string, an IO::AIO::WD object or a [IO::AIO::WD, path] pair");
883 root 1.195 }
884    
885     *pathsv = newSVsv (path);
886     *ptr = SvPVbyte_nolen (*pathsv);
887     }
888    
889 root 1.201 static void ecb_noinline
890 root 1.195 req_set_path1 (aio_req req, SV *path)
891     {
892     req_set_path (req, path, &req->sv1, &req->sv3, &req->wd, &req->ptr1);
893     }
894    
895 root 1.201 static void ecb_noinline
896 root 1.195 req_set_fh_or_path (aio_req req, int type_path, int type_fh, SV *fh_or_path)
897     {
898     SV *rv = SvROK (fh_or_path) ? SvRV (fh_or_path) : fh_or_path;
899    
900     switch (SvTYPE (rv))
901     {
902     case SVt_PVIO:
903     case SVt_PVLV:
904     case SVt_PVGV:
905     req->type = type_fh;
906     req->sv1 = newSVsv (fh_or_path);
907     req->int1 = PerlIO_fileno (IoIFP (sv_2io (fh_or_path)));
908     break;
909    
910     default:
911     req->type = type_path;
912     req_set_path1 (req, fh_or_path);
913     break;
914     }
915     }
916    
917 root 1.202 XS(boot_IO__AIO) ecb_cold;
918    
919 root 1.1 MODULE = IO::AIO PACKAGE = IO::AIO
920    
921 root 1.8 PROTOTYPES: ENABLE
922    
923 root 1.1 BOOT:
924     {
925 root 1.141 static const struct {
926     const char *name;
927     IV iv;
928     } *civ, const_iv[] = {
929 root 1.162 # define const_niv(name, value) { # name, (IV) value },
930     # define const_iv(name) { # name, (IV) name },
931 root 1.141 # define const_eio(name) { # name, (IV) EIO_ ## name },
932 root 1.162 const_iv (EXDEV)
933     const_iv (ENOSYS)
934     const_iv (O_RDONLY)
935     const_iv (O_WRONLY)
936 root 1.164 const_iv (O_RDWR)
937 root 1.162 const_iv (O_CREAT)
938     const_iv (O_TRUNC)
939 root 1.164 const_iv (O_EXCL)
940     const_iv (O_APPEND)
941 root 1.180
942 root 1.182 const_iv (O_ASYNC)
943     const_iv (O_DIRECT)
944     const_iv (O_NOATIME)
945    
946     const_iv (O_CLOEXEC)
947     const_iv (O_NOCTTY)
948     const_iv (O_NOFOLLOW)
949     const_iv (O_NONBLOCK)
950     const_iv (O_EXEC)
951     const_iv (O_SEARCH)
952     const_iv (O_DIRECTORY)
953     const_iv (O_DSYNC)
954     const_iv (O_RSYNC)
955     const_iv (O_SYNC)
956     const_iv (O_TTY_INIT)
957    
958 root 1.162 const_iv (S_IFIFO)
959 root 1.180 const_iv (S_IFCHR)
960     const_iv (S_IFBLK)
961     const_iv (S_IFLNK)
962     const_iv (S_IFREG)
963     const_iv (S_IFDIR)
964     const_iv (S_IFWHT)
965     const_iv (S_IFSOCK)
966     const_iv (S_IFMT)
967    
968 root 1.162 const_niv (FADV_NORMAL , POSIX_FADV_NORMAL)
969     const_niv (FADV_SEQUENTIAL, POSIX_FADV_SEQUENTIAL)
970     const_niv (FADV_RANDOM , POSIX_FADV_RANDOM)
971     const_niv (FADV_NOREUSE , POSIX_FADV_NOREUSE)
972     const_niv (FADV_WILLNEED , POSIX_FADV_WILLNEED)
973     const_niv (FADV_DONTNEED , POSIX_FADV_DONTNEED)
974    
975 root 1.169 const_niv (MADV_NORMAL , POSIX_MADV_NORMAL)
976     const_niv (MADV_SEQUENTIAL, POSIX_MADV_SEQUENTIAL)
977     const_niv (MADV_RANDOM , POSIX_MADV_RANDOM)
978     const_niv (MADV_WILLNEED , POSIX_MADV_WILLNEED)
979     const_niv (MADV_DONTNEED , POSIX_MADV_DONTNEED)
980    
981 root 1.162 const_iv (ST_RDONLY)
982     const_iv (ST_NOSUID)
983     const_iv (ST_NODEV)
984     const_iv (ST_NOEXEC)
985     const_iv (ST_SYNCHRONOUS)
986     const_iv (ST_MANDLOCK)
987     const_iv (ST_WRITE)
988     const_iv (ST_APPEND)
989     const_iv (ST_IMMUTABLE)
990     const_iv (ST_NOATIME)
991     const_iv (ST_NODIRATIME)
992     const_iv (ST_RELATIME)
993    
994 root 1.171 const_iv (PROT_NONE)
995 root 1.162 const_iv (PROT_EXEC)
996     const_iv (PROT_READ)
997     const_iv (PROT_WRITE)
998    
999     /*const_iv (MAP_FIXED)*/
1000     const_iv (MAP_PRIVATE)
1001     const_iv (MAP_SHARED)
1002     const_iv (MAP_ANONYMOUS)
1003    
1004     /* linuxish */
1005     const_iv (MAP_HUGETLB)
1006     const_iv (MAP_LOCKED)
1007     const_iv (MAP_NORESERVE)
1008     const_iv (MAP_POPULATE)
1009     const_iv (MAP_NONBLOCK)
1010 root 1.159
1011 root 1.209 const_eio (SEEK_SET)
1012     const_eio (SEEK_CUR)
1013     const_eio (SEEK_END)
1014    
1015 root 1.172 const_eio (MCL_FUTURE)
1016     const_eio (MCL_CURRENT)
1017 root 1.161
1018 root 1.158 const_eio (MS_ASYNC)
1019     const_eio (MS_INVALIDATE)
1020     const_eio (MS_SYNC)
1021    
1022     const_eio (MT_MODIFY)
1023    
1024 root 1.141 const_eio (SYNC_FILE_RANGE_WAIT_BEFORE)
1025     const_eio (SYNC_FILE_RANGE_WRITE)
1026     const_eio (SYNC_FILE_RANGE_WAIT_AFTER)
1027    
1028 root 1.185 const_eio (FALLOC_FL_KEEP_SIZE)
1029    
1030 root 1.141 const_eio (READDIR_DENTS)
1031     const_eio (READDIR_DIRS_FIRST)
1032     const_eio (READDIR_STAT_ORDER)
1033     const_eio (READDIR_FOUND_UNKNOWN)
1034    
1035     const_eio (DT_UNKNOWN)
1036     const_eio (DT_FIFO)
1037     const_eio (DT_CHR)
1038     const_eio (DT_DIR)
1039     const_eio (DT_BLK)
1040     const_eio (DT_REG)
1041     const_eio (DT_LNK)
1042     const_eio (DT_SOCK)
1043     const_eio (DT_WHT)
1044     };
1045 root 1.103
1046 root 1.156 aio_stash = gv_stashpv ("IO::AIO" , 1);
1047     aio_req_stash = gv_stashpv ("IO::AIO::REQ", 1);
1048     aio_grp_stash = gv_stashpv ("IO::AIO::GRP", 1);
1049 root 1.195 aio_wd_stash = gv_stashpv ("IO::AIO::WD" , 1);
1050 root 1.146
1051 root 1.194 for (civ = const_iv + sizeof (const_iv) / sizeof (const_iv [0]); civ > const_iv; civ--)
1052     newCONSTSUB (aio_stash, (char *)civ[-1].name, newSViv (civ[-1].iv));
1053 root 1.141
1054 root 1.162 newCONSTSUB (aio_stash, "PAGESIZE", newSViv (PAGESIZE));
1055    
1056 root 1.190 reinit ();
1057 root 1.1 }
1058    
1059     void
1060 root 1.190 reinit ()
1061     PROTOTYPE:
1062    
1063     void
1064 root 1.181 max_poll_reqs (unsigned int nreqs)
1065 root 1.85 PROTOTYPE: $
1066     CODE:
1067 root 1.115 eio_set_max_poll_reqs (nreqs);
1068 root 1.85
1069     void
1070     max_poll_time (double nseconds)
1071     PROTOTYPE: $
1072     CODE:
1073 root 1.115 eio_set_max_poll_time (nseconds);
1074 root 1.85
1075     void
1076 root 1.181 min_parallel (unsigned int nthreads)
1077 root 1.1 PROTOTYPE: $
1078 root 1.115 CODE:
1079     eio_set_min_parallel (nthreads);
1080 root 1.1
1081     void
1082 root 1.181 max_parallel (unsigned int nthreads)
1083 root 1.78 PROTOTYPE: $
1084 root 1.115 CODE:
1085     eio_set_max_parallel (nthreads);
1086 root 1.78
1087 root 1.85 void
1088 root 1.181 max_idle (unsigned int nthreads)
1089 root 1.85 PROTOTYPE: $
1090     CODE:
1091 root 1.115 eio_set_max_idle (nthreads);
1092 root 1.85
1093 root 1.115 void
1094 root 1.181 idle_timeout (unsigned int seconds)
1095     PROTOTYPE: $
1096     CODE:
1097     eio_set_idle_timeout (seconds);
1098    
1099     void
1100     max_outstanding (unsigned int maxreqs)
1101 root 1.1 PROTOTYPE: $
1102 root 1.78 CODE:
1103     max_outstanding = maxreqs;
1104 root 1.1
1105     void
1106 root 1.195 aio_wd (SV8 *pathname, SV *callback=&PL_sv_undef)
1107     PPCODE:
1108     {
1109     dREQ;
1110    
1111     req->type = EIO_WD_OPEN;
1112     req_set_path1 (req, pathname);
1113    
1114     REQ_SEND;
1115     }
1116    
1117     void
1118 root 1.99 aio_open (SV8 *pathname, int flags, int mode, SV *callback=&PL_sv_undef)
1119 root 1.43 PPCODE:
1120 root 1.1 {
1121 root 1.22 dREQ;
1122 root 1.1
1123 root 1.115 req->type = EIO_OPEN;
1124 root 1.195 req_set_path1 (req, pathname);
1125 root 1.86 req->int1 = flags;
1126 root 1.115 req->int2 = mode;
1127 root 1.1
1128 root 1.43 REQ_SEND;
1129 root 1.1 }
1130    
1131     void
1132 root 1.107 aio_fsync (SV *fh, SV *callback=&PL_sv_undef)
1133 root 1.1 ALIAS:
1134 root 1.115 aio_fsync = EIO_FSYNC
1135     aio_fdatasync = EIO_FDATASYNC
1136 root 1.192 aio_syncfs = EIO_SYNCFS
1137 root 1.43 PPCODE:
1138 root 1.1 {
1139 root 1.151 int fd = s_fileno_croak (fh, 0);
1140 root 1.22 dREQ;
1141 root 1.1
1142     req->type = ix;
1143 root 1.99 req->sv1 = newSVsv (fh);
1144 root 1.136 req->int1 = fd;
1145 root 1.1
1146 root 1.209 REQ_SEND;
1147 root 1.1 }
1148    
1149     void
1150 root 1.148 aio_sync_file_range (SV *fh, off_t offset, size_t nbytes, UV flags, SV *callback=&PL_sv_undef)
1151 root 1.133 PPCODE:
1152     {
1153 root 1.151 int fd = s_fileno_croak (fh, 0);
1154 root 1.133 dREQ;
1155    
1156     req->type = EIO_SYNC_FILE_RANGE;
1157     req->sv1 = newSVsv (fh);
1158 root 1.136 req->int1 = fd;
1159 root 1.148 req->offs = offset;
1160     req->size = nbytes;
1161 root 1.133 req->int2 = flags;
1162    
1163 root 1.209 REQ_SEND;
1164 root 1.133 }
1165    
1166     void
1167 root 1.185 aio_fallocate (SV *fh, int mode, off_t offset, size_t len, SV *callback=&PL_sv_undef)
1168     PPCODE:
1169     {
1170     int fd = s_fileno_croak (fh, 0);
1171     dREQ;
1172    
1173     req->type = EIO_FALLOCATE;
1174     req->sv1 = newSVsv (fh);
1175     req->int1 = fd;
1176     req->int2 = mode;
1177     req->offs = offset;
1178     req->size = len;
1179    
1180 root 1.209 REQ_SEND;
1181 root 1.185 }
1182    
1183     void
1184 root 1.113 aio_close (SV *fh, SV *callback=&PL_sv_undef)
1185 root 1.107 PPCODE:
1186     {
1187 root 1.187 static int close_fd = -1; /* dummy fd to close fds via dup2 */
1188 root 1.151 int fd = s_fileno_croak (fh, 0);
1189 root 1.109 dREQ;
1190 root 1.107
1191 root 1.187 if (expect_false (close_fd < 0))
1192 root 1.115 {
1193 root 1.188 int pipefd [2];
1194    
1195     if (
1196 root 1.187 #ifdef _WIN32
1197 root 1.188 _pipe (pipefd, 1, _O_BINARY) < 0
1198 root 1.187 #else
1199 root 1.188 pipe (pipefd) < 0
1200     || fcntl (pipefd [0], F_SETFD, FD_CLOEXEC) < 0
1201     #endif
1202     || close (pipefd [1]) < 0
1203     )
1204 root 1.115 abort (); /*D*/
1205    
1206 root 1.187 close_fd = pipefd [0];
1207 root 1.115 }
1208    
1209     req->type = EIO_DUP2;
1210 root 1.187 req->int1 = close_fd;
1211 root 1.115 req->sv2 = newSVsv (fh);
1212 root 1.136 req->int2 = fd;
1213 root 1.107
1214 root 1.209 REQ_SEND;
1215     }
1216    
1217     void
1218     aio_seek (SV *fh, SV *offset, int whence, SV *callback=&PL_sv_undef)
1219     PPCODE:
1220     {
1221     STRLEN svlen;
1222     int fd = s_fileno_croak (fh, 0);
1223     dREQ;
1224    
1225     req->type = EIO_SEEK;
1226     req->sv1 = newSVsv (fh);
1227     req->int1 = fd;
1228     req->offs = SvVAL64 (offset);
1229     req->int2 = whence;
1230    
1231     REQ_SEND;
1232 root 1.107 }
1233    
1234     void
1235 root 1.102 aio_read (SV *fh, SV *offset, SV *length, SV8 *data, IV dataoffset, SV *callback=&PL_sv_undef)
1236 root 1.13 ALIAS:
1237 root 1.115 aio_read = EIO_READ
1238     aio_write = EIO_WRITE
1239 root 1.43 PPCODE:
1240 root 1.13 {
1241     STRLEN svlen;
1242 root 1.151 int fd = s_fileno_croak (fh, ix == EIO_WRITE);
1243 root 1.21 char *svptr = SvPVbyte (data, svlen);
1244 root 1.102 UV len = SvUV (length);
1245 root 1.134
1246 root 1.13 if (dataoffset < 0)
1247     dataoffset += svlen;
1248    
1249     if (dataoffset < 0 || dataoffset > svlen)
1250 root 1.102 croak ("dataoffset outside of data scalar");
1251 root 1.13
1252 root 1.115 if (ix == EIO_WRITE)
1253 root 1.13 {
1254     /* write: check length and adjust. */
1255 root 1.102 if (!SvOK (length) || len + dataoffset > svlen)
1256     len = svlen - dataoffset;
1257 root 1.13 }
1258     else
1259     {
1260 root 1.138 /* read: check type and grow scalar as necessary */
1261     SvUPGRADE (data, SVt_PV);
1262 root 1.102 svptr = SvGROW (data, len + dataoffset + 1);
1263 root 1.13 }
1264    
1265 root 1.22 {
1266     dREQ;
1267 root 1.13
1268 root 1.22 req->type = ix;
1269 root 1.99 req->sv1 = newSVsv (fh);
1270 root 1.135 req->int1 = fd;
1271 root 1.101 req->offs = SvOK (offset) ? SvVAL64 (offset) : -1;
1272 root 1.102 req->size = len;
1273 root 1.132 req->sv2 = SvREFCNT_inc (data);
1274 root 1.115 req->ptr2 = (char *)svptr + dataoffset;
1275 root 1.86 req->stroffset = dataoffset;
1276 root 1.13
1277 root 1.28 if (!SvREADONLY (data))
1278     {
1279     SvREADONLY_on (data);
1280 root 1.100 req->flags |= FLAG_SV2_RO_OFF;
1281 root 1.28 }
1282    
1283 root 1.43 REQ_SEND;
1284 root 1.22 }
1285 root 1.13 }
1286 root 1.1
1287     void
1288 root 1.195 aio_readlink (SV8 *pathname, SV *callback=&PL_sv_undef)
1289 root 1.183 ALIAS:
1290     aio_readlink = EIO_READLINK
1291     aio_realpath = EIO_REALPATH
1292 root 1.86 PPCODE:
1293     {
1294     dREQ;
1295    
1296 root 1.183 req->type = ix;
1297 root 1.195 req_set_path1 (req, pathname);
1298 root 1.86
1299     REQ_SEND;
1300     }
1301    
1302     void
1303 root 1.148 aio_sendfile (SV *out_fh, SV *in_fh, off_t in_offset, size_t length, SV *callback=&PL_sv_undef)
1304 root 1.43 PPCODE:
1305 root 1.32 {
1306 root 1.151 int ifd = s_fileno_croak (in_fh , 0);
1307     int ofd = s_fileno_croak (out_fh, 1);
1308 root 1.32 dREQ;
1309    
1310 root 1.115 req->type = EIO_SENDFILE;
1311 root 1.99 req->sv1 = newSVsv (out_fh);
1312 root 1.136 req->int1 = ofd;
1313 root 1.86 req->sv2 = newSVsv (in_fh);
1314 root 1.136 req->int2 = ifd;
1315 root 1.148 req->offs = in_offset;
1316 root 1.86 req->size = length;
1317 root 1.32
1318 root 1.43 REQ_SEND;
1319 root 1.32 }
1320    
1321     void
1322 root 1.148 aio_readahead (SV *fh, off_t offset, size_t length, SV *callback=&PL_sv_undef)
1323 root 1.43 PPCODE:
1324 root 1.1 {
1325 root 1.151 int fd = s_fileno_croak (fh, 0);
1326 root 1.22 dREQ;
1327 root 1.1
1328 root 1.115 req->type = EIO_READAHEAD;
1329 root 1.99 req->sv1 = newSVsv (fh);
1330 root 1.136 req->int1 = fd;
1331 root 1.148 req->offs = offset;
1332 root 1.86 req->size = length;
1333 root 1.1
1334 root 1.43 REQ_SEND;
1335 root 1.1 }
1336    
1337     void
1338 root 1.99 aio_stat (SV8 *fh_or_path, SV *callback=&PL_sv_undef)
1339 root 1.1 ALIAS:
1340 root 1.159 aio_stat = EIO_STAT
1341     aio_lstat = EIO_LSTAT
1342     aio_statvfs = EIO_STATVFS
1343 root 1.43 PPCODE:
1344 root 1.1 {
1345 root 1.22 dREQ;
1346 root 1.1
1347 root 1.195 req_set_fh_or_path (req, ix, ix == EIO_STATVFS ? EIO_FSTATVFS : EIO_FSTAT, fh_or_path);
1348 root 1.203
1349 root 1.43 REQ_SEND;
1350 root 1.1 }
1351    
1352 root 1.180 UV
1353     major (UV dev)
1354     ALIAS:
1355     minor = 1
1356     CODE:
1357     RETVAL = ix ? major (dev) : minor (dev);
1358     OUTPUT:
1359     RETVAL
1360    
1361     UV
1362     makedev (UV maj, UV min)
1363     CODE:
1364     RETVAL = makedev (maj, min);
1365     OUTPUT:
1366     RETVAL
1367    
1368 root 1.1 void
1369 root 1.99 aio_utime (SV8 *fh_or_path, SV *atime, SV *mtime, SV *callback=&PL_sv_undef)
1370     PPCODE:
1371     {
1372     dREQ;
1373    
1374     req->nv1 = SvOK (atime) ? SvNV (atime) : -1.;
1375     req->nv2 = SvOK (mtime) ? SvNV (mtime) : -1.;
1376 root 1.195 req_set_fh_or_path (req, EIO_UTIME, EIO_FUTIME, fh_or_path);
1377 root 1.99
1378     REQ_SEND;
1379     }
1380    
1381     void
1382 root 1.103 aio_truncate (SV8 *fh_or_path, SV *offset, SV *callback=&PL_sv_undef)
1383     PPCODE:
1384     {
1385     dREQ;
1386    
1387     req->offs = SvOK (offset) ? SvVAL64 (offset) : -1;
1388 root 1.195 req_set_fh_or_path (req, EIO_TRUNCATE, EIO_FTRUNCATE, fh_or_path);
1389 root 1.103
1390     REQ_SEND;
1391     }
1392    
1393     void
1394 root 1.99 aio_chmod (SV8 *fh_or_path, int mode, SV *callback=&PL_sv_undef)
1395     PPCODE:
1396     {
1397     dREQ;
1398    
1399 root 1.115 req->int2 = mode;
1400 root 1.195 req_set_fh_or_path (req, EIO_CHMOD, EIO_FCHMOD, fh_or_path);
1401 root 1.99
1402     REQ_SEND;
1403     }
1404    
1405     void
1406     aio_chown (SV8 *fh_or_path, SV *uid, SV *gid, SV *callback=&PL_sv_undef)
1407     PPCODE:
1408     {
1409     dREQ;
1410    
1411     req->int2 = SvOK (uid) ? SvIV (uid) : -1;
1412     req->int3 = SvOK (gid) ? SvIV (gid) : -1;
1413 root 1.195 req_set_fh_or_path (req, EIO_CHOWN, EIO_FCHOWN, fh_or_path);
1414 root 1.99
1415     REQ_SEND;
1416     }
1417    
1418     void
1419 root 1.141 aio_readdirx (SV8 *pathname, IV flags, SV *callback=&PL_sv_undef)
1420     PPCODE:
1421     {
1422     dREQ;
1423    
1424     req->type = EIO_READDIR;
1425     req->int1 = flags | EIO_READDIR_DENTS | EIO_READDIR_CUSTOM1;
1426    
1427     if (flags & EIO_READDIR_DENTS)
1428 root 1.142 req->int1 |= EIO_READDIR_CUSTOM2;
1429 root 1.141
1430 root 1.197 req_set_path1 (req, pathname);
1431    
1432 root 1.141 REQ_SEND;
1433     }
1434    
1435     void
1436 root 1.200 aio_mkdir (SV8 *pathname, int mode, SV *callback=&PL_sv_undef)
1437     PPCODE:
1438     {
1439     dREQ;
1440    
1441     req->type = EIO_MKDIR;
1442     req->int2 = mode;
1443     req_set_path1 (req, pathname);
1444    
1445     REQ_SEND;
1446     }
1447    
1448     void
1449 root 1.99 aio_unlink (SV8 *pathname, SV *callback=&PL_sv_undef)
1450 root 1.22 ALIAS:
1451 root 1.115 aio_unlink = EIO_UNLINK
1452     aio_rmdir = EIO_RMDIR
1453     aio_readdir = EIO_READDIR
1454 root 1.43 PPCODE:
1455 root 1.1 {
1456 root 1.22 dREQ;
1457 root 1.1
1458 root 1.22 req->type = ix;
1459 root 1.195 req_set_path1 (req, pathname);
1460 root 1.87
1461 root 1.43 REQ_SEND;
1462 root 1.22 }
1463    
1464     void
1465 root 1.99 aio_link (SV8 *oldpath, SV8 *newpath, SV *callback=&PL_sv_undef)
1466 root 1.40 ALIAS:
1467 root 1.115 aio_link = EIO_LINK
1468     aio_symlink = EIO_SYMLINK
1469     aio_rename = EIO_RENAME
1470 root 1.43 PPCODE:
1471 root 1.22 {
1472 root 1.207 eio_wd wd2 = 0;
1473 root 1.22 dREQ;
1474 root 1.1
1475 root 1.40 req->type = ix;
1476 root 1.195 req_set_path1 (req, oldpath);
1477     req_set_path (req, newpath, &req->sv2, &req->sv4, &wd2, &req->ptr2);
1478     req->int3 = (long)wd2;
1479 root 1.1
1480 root 1.43 REQ_SEND;
1481 root 1.1 }
1482    
1483 root 1.42 void
1484 root 1.99 aio_mknod (SV8 *pathname, int mode, UV dev, SV *callback=&PL_sv_undef)
1485 root 1.81 PPCODE:
1486     {
1487     dREQ;
1488    
1489 root 1.115 req->type = EIO_MKNOD;
1490     req->int2 = (mode_t)mode;
1491 root 1.86 req->offs = dev;
1492 root 1.195 req_set_path1 (req, pathname);
1493 root 1.81
1494     REQ_SEND;
1495     }
1496    
1497     void
1498 root 1.158 aio_mtouch (SV8 *data, IV offset = 0, SV *length = &PL_sv_undef, int flags = 0, SV *callback=&PL_sv_undef)
1499     ALIAS:
1500     aio_mtouch = EIO_MTOUCH
1501     aio_msync = EIO_MSYNC
1502     PPCODE:
1503     {
1504     STRLEN svlen;
1505 root 1.172 char *svptr = SvPVbyte (data, svlen);
1506 root 1.158 UV len = SvUV (length);
1507    
1508     if (offset < 0)
1509     offset += svlen;
1510    
1511     if (offset < 0 || offset > svlen)
1512     croak ("offset outside of scalar");
1513    
1514     if (!SvOK (length) || len + offset > svlen)
1515     len = svlen - offset;
1516    
1517     {
1518     dREQ;
1519    
1520     req->type = ix;
1521 root 1.172 req->sv2 = SvREFCNT_inc (data);
1522     req->ptr2 = (char *)svptr + offset;
1523 root 1.158 req->size = len;
1524 root 1.172 req->int1 = flags;
1525    
1526     REQ_SEND;
1527     }
1528     }
1529    
1530     void
1531     aio_mlock (SV8 *data, IV offset = 0, SV *length = &PL_sv_undef, SV *callback=&PL_sv_undef)
1532     PPCODE:
1533     {
1534     STRLEN svlen;
1535     char *svptr = SvPVbyte (data, svlen);
1536     UV len = SvUV (length);
1537    
1538     if (offset < 0)
1539     offset += svlen;
1540    
1541     if (offset < 0 || offset > svlen)
1542     croak ("offset outside of scalar");
1543    
1544     if (!SvOK (length) || len + offset > svlen)
1545     len = svlen - offset;
1546    
1547     {
1548     dREQ;
1549    
1550     req->type = EIO_MLOCK;
1551 root 1.158 req->sv2 = SvREFCNT_inc (data);
1552     req->ptr2 = (char *)svptr + offset;
1553 root 1.172 req->size = len;
1554 root 1.158
1555     REQ_SEND;
1556     }
1557     }
1558    
1559     void
1560 root 1.172 aio_mlockall (IV flags, SV *callback=&PL_sv_undef)
1561     PPCODE:
1562     {
1563     dREQ;
1564    
1565     req->type = EIO_MLOCKALL;
1566     req->int1 = flags;
1567    
1568     REQ_SEND;
1569     }
1570    
1571     void
1572 root 1.99 aio_busy (double delay, SV *callback=&PL_sv_undef)
1573 root 1.45 PPCODE:
1574     {
1575     dREQ;
1576    
1577 root 1.115 req->type = EIO_BUSY;
1578 root 1.99 req->nv1 = delay < 0. ? 0. : delay;
1579 root 1.45
1580     REQ_SEND;
1581     }
1582    
1583     void
1584 root 1.99 aio_group (SV *callback=&PL_sv_undef)
1585 root 1.44 PPCODE:
1586 root 1.42 {
1587 root 1.44 dREQ;
1588 root 1.60
1589 root 1.115 req->type = EIO_GROUP;
1590 root 1.86
1591 root 1.127 req_submit (req);
1592 root 1.156 XPUSHs (req_sv (req, aio_grp_stash));
1593 root 1.42 }
1594    
1595 root 1.6 void
1596 root 1.99 aio_nop (SV *callback=&PL_sv_undef)
1597 root 1.110 ALIAS:
1598 root 1.115 aio_nop = EIO_NOP
1599     aio_sync = EIO_SYNC
1600 root 1.54 PPCODE:
1601     {
1602     dREQ;
1603    
1604 root 1.110 req->type = ix;
1605 root 1.54
1606     REQ_SEND;
1607     }
1608    
1609 root 1.79 int
1610     aioreq_pri (int pri = 0)
1611     CODE:
1612 root 1.121 RETVAL = next_pri;
1613 root 1.79 if (items > 0)
1614     {
1615 root 1.115 if (pri < EIO_PRI_MIN) pri = EIO_PRI_MIN;
1616     if (pri > EIO_PRI_MAX) pri = EIO_PRI_MAX;
1617 root 1.121 next_pri = pri;
1618 root 1.79 }
1619     OUTPUT:
1620     RETVAL
1621 root 1.68
1622     void
1623     aioreq_nice (int nice = 0)
1624 root 1.79 CODE:
1625     nice = next_pri - nice;
1626 root 1.115 if (nice < EIO_PRI_MIN) nice = EIO_PRI_MIN;
1627     if (nice > EIO_PRI_MAX) nice = EIO_PRI_MAX;
1628 root 1.121 next_pri = nice;
1629 root 1.60
1630 root 1.54 void
1631 root 1.40 flush ()
1632 root 1.6 CODE:
1633 root 1.116 while (eio_nreqs ())
1634 root 1.6 {
1635     poll_wait ();
1636 root 1.91 poll_cb ();
1637 root 1.6 }
1638    
1639 root 1.91 int
1640 root 1.172 poll ()
1641 root 1.7 CODE:
1642 root 1.92 poll_wait ();
1643     RETVAL = poll_cb ();
1644 root 1.91 OUTPUT:
1645     RETVAL
1646 root 1.7
1647 root 1.1 int
1648 root 1.172 poll_fileno ()
1649 root 1.1 CODE:
1650 root 1.153 RETVAL = s_epipe_fd (&respipe);
1651 root 1.1 OUTPUT:
1652     RETVAL
1653    
1654     int
1655 root 1.172 poll_cb (...)
1656 root 1.1 PROTOTYPE:
1657     CODE:
1658 root 1.85 RETVAL = poll_cb ();
1659 root 1.1 OUTPUT:
1660     RETVAL
1661    
1662     void
1663 root 1.172 poll_wait ()
1664 root 1.1 CODE:
1665 root 1.92 poll_wait ();
1666 root 1.1
1667     int
1668 root 1.172 nreqs ()
1669 root 1.1 CODE:
1670 root 1.115 RETVAL = eio_nreqs ();
1671 root 1.1 OUTPUT:
1672     RETVAL
1673    
1674 root 1.79 int
1675 root 1.172 nready ()
1676 root 1.79 CODE:
1677 root 1.115 RETVAL = eio_nready ();
1678 root 1.79 OUTPUT:
1679     RETVAL
1680    
1681     int
1682 root 1.172 npending ()
1683 root 1.79 CODE:
1684 root 1.115 RETVAL = eio_npending ();
1685 root 1.79 OUTPUT:
1686     RETVAL
1687    
1688 root 1.85 int
1689 root 1.172 nthreads ()
1690 root 1.85 CODE:
1691 root 1.122 RETVAL = eio_nthreads ();
1692 root 1.85 OUTPUT:
1693     RETVAL
1694    
1695 root 1.148 int
1696     fadvise (aio_rfd fh, off_t offset, off_t length, IV advice)
1697     CODE:
1698     RETVAL = posix_fadvise (fh, offset, length, advice);
1699     OUTPUT:
1700     RETVAL
1701    
1702 root 1.191 IV
1703 root 1.148 sendfile (aio_wfd ofh, aio_rfd ifh, off_t offset, size_t count)
1704     CODE:
1705 root 1.157 RETVAL = eio_sendfile_sync (ofh, ifh, offset, count);
1706     OUTPUT:
1707     RETVAL
1708 root 1.148
1709 root 1.162 void
1710     mmap (SV *scalar, size_t length, int prot, int flags, SV *fh, off_t offset = 0)
1711     PPCODE:
1712     sv_unmagic (scalar, MMAP_MAGIC);
1713     {
1714     int fd = SvOK (fh) ? s_fileno_croak (fh, flags & PROT_WRITE) : -1;
1715     void *addr = (void *)mmap (0, length, prot, flags, fd, offset);
1716     if (addr == (void *)-1)
1717     XSRETURN_NO;
1718    
1719 root 1.168 sv_force_normal (scalar);
1720    
1721 root 1.162 /* we store the length in mg_obj, as namlen is I32 :/ */
1722     sv_magicext (scalar, 0, MMAP_MAGIC, &mmap_vtbl, (char *)addr, 0)
1723     ->mg_obj = (SV *)length;
1724    
1725     SvUPGRADE (scalar, SVt_PV); /* nop... */
1726 root 1.168
1727 root 1.162 if (!(prot & PROT_WRITE))
1728     SvREADONLY_on (scalar);
1729    
1730 root 1.168 if (SvLEN (scalar))
1731     Safefree (SvPVX (scalar));
1732    
1733 root 1.162 SvPVX (scalar) = (char *)addr;
1734     SvCUR_set (scalar, length);
1735     SvLEN_set (scalar, 0);
1736     SvPOK_only (scalar);
1737    
1738     XSRETURN_YES;
1739     }
1740    
1741     void
1742     munmap (SV *scalar)
1743     CODE:
1744     sv_unmagic (scalar, MMAP_MAGIC);
1745    
1746 root 1.161 int
1747 root 1.172 madvise (SV *scalar, off_t offset = 0, SV *length = &PL_sv_undef, IV advice_or_prot)
1748 root 1.171 ALIAS:
1749     mprotect = 1
1750 root 1.169 CODE:
1751     {
1752 root 1.172 STRLEN svlen;
1753     void *addr = SvPVbyte (scalar, svlen);
1754     size_t len = SvUV (length);
1755    
1756     if (offset < 0)
1757     offset += svlen;
1758    
1759     if (offset < 0 || offset > svlen)
1760     croak ("offset outside of scalar");
1761    
1762     if (!SvOK (length) || len + offset > svlen)
1763     len = svlen - offset;
1764 root 1.169
1765 root 1.172 addr = (void *)(((intptr_t)addr) + offset);
1766     eio_page_align (&addr, &len);
1767    
1768     switch (ix)
1769 root 1.171 {
1770 root 1.172 case 0: RETVAL = posix_madvise (addr, len, advice_or_prot); break;
1771     case 1: RETVAL = mprotect (addr, len, advice_or_prot); break;
1772 root 1.171 }
1773 root 1.169 }
1774     OUTPUT:
1775     RETVAL
1776    
1777     int
1778 root 1.172 munlock (SV *scalar, off_t offset = 0, SV *length = &PL_sv_undef)
1779 root 1.161 CODE:
1780 root 1.172 {
1781     STRLEN svlen;
1782     void *addr = SvPVbyte (scalar, svlen);
1783     size_t len = SvUV (length);
1784    
1785     if (offset < 0)
1786     offset += svlen;
1787    
1788     if (offset < 0 || offset > svlen)
1789     croak ("offset outside of scalar");
1790    
1791     if (!SvOK (length) || len + offset > svlen)
1792     len = svlen - offset;
1793    
1794     addr = (void *)(((intptr_t)addr) + offset);
1795     eio_page_align (&addr, &len);
1796 root 1.178 #if _POSIX_MEMLOCK_RANGE
1797 root 1.172 RETVAL = munlock (addr, len);
1798 root 1.161 #else
1799 root 1.172 RETVAL = ((errno = ENOSYS), -1);
1800 root 1.161 #endif
1801 root 1.172 }
1802 root 1.161 OUTPUT:
1803     RETVAL
1804    
1805     int
1806     munlockall ()
1807     CODE:
1808     #if _POSIX_MEMLOCK
1809     munlockall ();
1810     #else
1811 root 1.162 RETVAL = -1;
1812 root 1.161 errno = ENOSYS;
1813     #endif
1814     OUTPUT:
1815     RETVAL
1816    
1817 root 1.117 void _on_next_submit (SV *cb)
1818     CODE:
1819     SvREFCNT_dec (on_next_submit);
1820     on_next_submit = SvOK (cb) ? newSVsv (cb) : 0;
1821    
1822 root 1.48 PROTOTYPES: DISABLE
1823    
1824 root 1.195 MODULE = IO::AIO PACKAGE = IO::AIO::WD
1825    
1826 root 1.196 BOOT:
1827     {
1828     newCONSTSUB (aio_stash, "CWD" , newSVaio_wd (EIO_CWD ));
1829     newCONSTSUB (aio_stash, "INVALID_WD", newSVaio_wd (EIO_INVALID_WD));
1830     }
1831    
1832 root 1.195 void
1833     DESTROY (SV *self)
1834     CODE:
1835     {
1836     aio_wd wd = SvAIO_WD (self);
1837     #if HAVE_AT
1838 root 1.203 {
1839     SV *callback = &PL_sv_undef;
1840     dREQ; /* clobbers next_pri :/ */
1841     next_pri = req->pri; /* restore next_pri */
1842     req->pri = EIO_PRI_MAX; /* better use max. priority to conserve fds */
1843     req->type = EIO_WD_CLOSE;
1844     req->wd = wd;
1845     REQ_SEND;
1846     }
1847 root 1.195 #else
1848     eio_wd_close_sync (wd);
1849     #endif
1850     }
1851    
1852 root 1.44 MODULE = IO::AIO PACKAGE = IO::AIO::REQ
1853 root 1.43
1854     void
1855     cancel (aio_req_ornot req)
1856     CODE:
1857 root 1.115 eio_cancel (req);
1858 root 1.45
1859 root 1.56 void
1860 root 1.59 cb (aio_req_ornot req, SV *callback=&PL_sv_undef)
1861 root 1.128 PPCODE:
1862     {
1863     if (GIMME_V != G_VOID)
1864     XPUSHs (req->callback ? sv_2mortal (newRV_inc (req->callback)) : &PL_sv_undef);
1865    
1866     if (items > 1)
1867     {
1868 root 1.152 SV *cb_cv =get_cb (callback);
1869 root 1.128
1870     SvREFCNT_dec (req->callback);
1871     req->callback = SvREFCNT_inc (cb_cv);
1872     }
1873     }
1874 root 1.56
1875 root 1.45 MODULE = IO::AIO PACKAGE = IO::AIO::GRP
1876    
1877     void
1878     add (aio_req grp, ...)
1879     PPCODE:
1880     {
1881     int i;
1882 root 1.94
1883 root 1.86 if (grp->int1 == 2)
1884 root 1.49 croak ("cannot add requests to IO::AIO::GRP after the group finished");
1885    
1886 root 1.45 for (i = 1; i < items; ++i )
1887     {
1888 root 1.115 aio_req req;
1889    
1890 root 1.46 if (GIMME_V != G_VOID)
1891     XPUSHs (sv_2mortal (newSVsv (ST (i))));
1892    
1893 root 1.53 req = SvAIO_REQ (ST (i));
1894 root 1.45
1895 root 1.46 if (req)
1896 root 1.115 eio_grp_add (grp, req);
1897 root 1.45 }
1898     }
1899 root 1.43
1900 root 1.48 void
1901 root 1.72 cancel_subs (aio_req_ornot req)
1902     CODE:
1903     req_cancel_subs (req);
1904    
1905     void
1906 root 1.51 result (aio_req grp, ...)
1907     CODE:
1908     {
1909     int i;
1910 root 1.79 AV *av;
1911    
1912     grp->errorno = errno;
1913    
1914     av = newAV ();
1915 root 1.141 av_extend (av, items - 1);
1916 root 1.51
1917     for (i = 1; i < items; ++i )
1918     av_push (av, newSVsv (ST (i)));
1919    
1920 root 1.86 SvREFCNT_dec (grp->sv1);
1921     grp->sv1 = (SV *)av;
1922 root 1.51 }
1923    
1924     void
1925 root 1.79 errno (aio_req grp, int errorno = errno)
1926     CODE:
1927     grp->errorno = errorno;
1928    
1929     void
1930 root 1.67 limit (aio_req grp, int limit)
1931 root 1.49 CODE:
1932 root 1.115 eio_grp_limit (grp, limit);
1933 root 1.49
1934     void
1935 root 1.56 feed (aio_req grp, SV *callback=&PL_sv_undef)
1936 root 1.49 CODE:
1937     {
1938 root 1.86 SvREFCNT_dec (grp->sv2);
1939 root 1.131 grp->sv2 = newSVsv (callback);
1940     grp->feed = aio_grp_feed;
1941 root 1.49
1942 root 1.86 if (grp->int2 <= 0)
1943     grp->int2 = 2;
1944 root 1.49
1945 root 1.115 eio_grp_limit (grp, grp->int2);
1946 root 1.49 }
1947