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