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