ViewVC Help
View File | Revision Log | Show Annotations | Download File
/cvs/IO-AIO/AIO.xs
Revision: 1.44
Committed: Sat Oct 21 23:20:29 2006 UTC (17 years, 7 months ago) by root
Branch: MAIN
Changes since 1.43: +19 -62 lines
Log Message:
*** empty log message ***

File Contents

# User Rev Content
1 root 1.20 #define _REENTRANT 1
2 root 1.19 #include <errno.h>
3    
4 root 1.15 #include "EXTERN.h"
5 root 1.1 #include "perl.h"
6     #include "XSUB.h"
7    
8 root 1.16 #include "autoconf/config.h"
9    
10 root 1.32 #include <pthread.h>
11    
12 root 1.37 #include <stddef.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 <unistd.h>
18     #include <fcntl.h>
19     #include <signal.h>
20     #include <sched.h>
21    
22 root 1.32 #if HAVE_SENDFILE
23     # if __linux
24     # include <sys/sendfile.h>
25     # elif __freebsd
26     # include <sys/socket.h>
27     # include <sys/uio.h>
28     # elif __hpux
29     # include <sys/socket.h>
30 root 1.35 # elif __solaris /* not yet */
31     # include <sys/sendfile.h>
32 root 1.34 # else
33     # error sendfile support requested but not available
34 root 1.32 # endif
35     #endif
36 root 1.1
37 root 1.39 /* used for struct dirent, AIX doesn't provide it */
38     #ifndef NAME_MAX
39     # define NAME_MAX 4096
40     #endif
41    
42 root 1.4 #if __ia64
43     # define STACKSIZE 65536
44 root 1.1 #else
45 root 1.37 # define STACKSIZE 8192
46 root 1.1 #endif
47    
48     enum {
49     REQ_QUIT,
50     REQ_OPEN, REQ_CLOSE,
51     REQ_READ, REQ_WRITE, REQ_READAHEAD,
52 root 1.32 REQ_SENDFILE,
53 root 1.22 REQ_STAT, REQ_LSTAT, REQ_FSTAT,
54 root 1.1 REQ_FSYNC, REQ_FDATASYNC,
55 root 1.40 REQ_UNLINK, REQ_RMDIR, REQ_RENAME,
56 root 1.37 REQ_READDIR,
57 root 1.40 REQ_LINK, REQ_SYMLINK,
58 root 1.44 REQ_GROUP,
59 root 1.1 };
60    
61 root 1.44 #define AIO_REQ_KLASS "IO::AIO::REQ"
62     #define AIO_GRP_KLASS "IO::AIO::GRP"
63 root 1.43
64     typedef struct aio_cb
65     {
66 root 1.44 struct aio_cb *grp, *grp_prev, *grp_next;
67 root 1.43
68 root 1.3 struct aio_cb *volatile next;
69 root 1.1
70 root 1.43 SV *self; /* the perl counterpart of this request, if any */
71 root 1.1
72 root 1.43 SV *data, *callback;
73     SV *fh, *fh2;
74     void *dataptr, *data2ptr;
75     Stat_t *statdata;
76 root 1.1 off_t offset;
77     size_t length;
78     ssize_t result;
79 root 1.43
80     int type;
81     int fd, fd2;
82 root 1.1 int errorno;
83     STRLEN dataoffset;
84 root 1.43 mode_t mode; /* open */
85     unsigned char cancelled;
86 root 1.1 } aio_cb;
87    
88     typedef aio_cb *aio_req;
89 root 1.43 typedef aio_cb *aio_req_ornot;
90 root 1.44 typedef aio_cb *aio_group;
91 root 1.1
92 root 1.30 static int started, wanted;
93 root 1.11 static volatile int nreqs;
94 root 1.4 static int max_outstanding = 1<<30;
95 root 1.3 static int respipe [2];
96 root 1.1
97 root 1.3 static pthread_mutex_t reslock = PTHREAD_MUTEX_INITIALIZER;
98     static pthread_mutex_t reqlock = PTHREAD_MUTEX_INITIALIZER;
99     static pthread_cond_t reqwait = PTHREAD_COND_INITIALIZER;
100    
101     static volatile aio_req reqs, reqe; /* queue start, queue end */
102     static volatile aio_req ress, rese; /* queue start, queue end */
103 root 1.1
104 root 1.43 /* must be called at most once */
105 root 1.44 static SV *req_sv (aio_req req, const char *klass)
106 root 1.43 {
107     req->self = (SV *)newHV ();
108     sv_magic (req->self, 0, PERL_MAGIC_ext, (char *)req, 0);
109    
110 root 1.44 return sv_2mortal (sv_bless (newRV_noinc (req->self), gv_stashpv (klass, 1)));
111 root 1.43 }
112    
113 root 1.44 static aio_req SvAIO_REQ (SV *sv, const char *klass)
114 root 1.26 {
115 root 1.44 if (!sv_derived_from (sv, klass) || !SvROK (sv))
116     croak ("object of class %s expected", klass);
117 root 1.43
118     MAGIC *mg = mg_find (SvRV (sv), PERL_MAGIC_ext);
119    
120     return mg ? (aio_req)mg->mg_ptr : 0;
121     }
122    
123     static void req_free (aio_req req)
124     {
125     if (req->self)
126     {
127     sv_unmagic (req->self, PERL_MAGIC_ext);
128     SvREFCNT_dec (req->self);
129     }
130    
131 root 1.26 if (req->data)
132     SvREFCNT_dec (req->data);
133    
134     if (req->fh)
135     SvREFCNT_dec (req->fh);
136    
137 root 1.32 if (req->fh2)
138     SvREFCNT_dec (req->fh2);
139    
140 root 1.27 if (req->statdata)
141 root 1.26 Safefree (req->statdata);
142    
143     if (req->callback)
144     SvREFCNT_dec (req->callback);
145    
146 root 1.37 if (req->type == REQ_READDIR && req->result >= 0)
147     free (req->data2ptr);
148    
149 root 1.26 Safefree (req);
150     }
151    
152 root 1.1 static void
153     poll_wait ()
154     {
155 root 1.11 if (nreqs && !ress)
156     {
157     fd_set rfd;
158     FD_ZERO(&rfd);
159     FD_SET(respipe [0], &rfd);
160 root 1.3
161 root 1.11 select (respipe [0] + 1, &rfd, 0, 0, 0);
162     }
163 root 1.1 }
164    
165     static int
166 root 1.5 poll_cb ()
167 root 1.1 {
168     dSP;
169     int count = 0;
170 root 1.24 int do_croak = 0;
171 root 1.27 aio_req req;
172    
173     for (;;)
174     {
175     pthread_mutex_lock (&reslock);
176     req = ress;
177    
178     if (req)
179     {
180     ress = req->next;
181 root 1.11
182 root 1.27 if (!ress)
183     {
184     /* read any signals sent by the worker threads */
185     char buf [32];
186     while (read (respipe [0], buf, 32) == 32)
187     ;
188 root 1.30
189     rese = 0;
190 root 1.27 }
191     }
192 root 1.1
193 root 1.27 pthread_mutex_unlock (&reslock);
194 root 1.3
195 root 1.27 if (!req)
196     break;
197 root 1.3
198 root 1.1 nreqs--;
199    
200     if (req->type == REQ_QUIT)
201     started--;
202     else
203     {
204     int errorno = errno;
205     errno = req->errorno;
206    
207     if (req->type == REQ_READ)
208 root 1.27 SvCUR_set (req->data, req->dataoffset + (req->result > 0 ? req->result : 0));
209 root 1.1
210 root 1.28 if (req->data2ptr && (req->type == REQ_READ || req->type == REQ_WRITE))
211     SvREADONLY_off (req->data);
212    
213 root 1.27 if (req->statdata)
214 root 1.1 {
215     PL_laststype = req->type == REQ_LSTAT ? OP_LSTAT : OP_STAT;
216     PL_laststatval = req->result;
217     PL_statcache = *(req->statdata);
218     }
219    
220 root 1.13 ENTER;
221 root 1.1 PUSHMARK (SP);
222 root 1.2
223 root 1.37 if (req->type == REQ_READDIR)
224     {
225     SV *rv = &PL_sv_undef;
226    
227     if (req->result >= 0)
228     {
229     char *buf = req->data2ptr;
230     AV *av = newAV ();
231    
232     while (req->result)
233     {
234     SV *sv = newSVpv (buf, 0);
235    
236     av_push (av, sv);
237     buf += SvCUR (sv) + 1;
238     req->result--;
239     }
240    
241     rv = sv_2mortal (newRV_noinc ((SV *)av));
242     }
243    
244     XPUSHs (rv);
245     }
246     else
247 root 1.2 {
248 root 1.37 XPUSHs (sv_2mortal (newSViv (req->result)));
249    
250     if (req->type == REQ_OPEN)
251     {
252     /* convert fd to fh */
253     SV *fh;
254 root 1.2
255 root 1.37 PUTBACK;
256     call_pv ("IO::AIO::_fd2fh", G_SCALAR | G_EVAL);
257     SPAGAIN;
258 root 1.2
259 root 1.37 fh = SvREFCNT_inc (POPs);
260 root 1.2
261 root 1.37 PUSHMARK (SP);
262     XPUSHs (sv_2mortal (fh));
263     }
264 root 1.2 }
265    
266 root 1.43 if (SvOK (req->callback) && !req->cancelled)
267 root 1.8 {
268     PUTBACK;
269     call_sv (req->callback, G_VOID | G_EVAL);
270     SPAGAIN;
271 root 1.27
272     if (SvTRUE (ERRSV))
273     {
274 root 1.43 req_free (req);
275 root 1.27 croak (0);
276     }
277 root 1.8 }
278 root 1.13
279 root 1.26 LEAVE;
280    
281 root 1.1 errno = errorno;
282     count++;
283     }
284    
285 root 1.43 req_free (req);
286 root 1.1 }
287    
288     return count;
289     }
290    
291 root 1.4 static void *aio_proc(void *arg);
292    
293     static void
294     start_thread (void)
295     {
296     sigset_t fullsigset, oldsigset;
297     pthread_t tid;
298     pthread_attr_t attr;
299    
300     pthread_attr_init (&attr);
301     pthread_attr_setstacksize (&attr, STACKSIZE);
302 root 1.31 pthread_attr_setdetachstate (&attr, PTHREAD_CREATE_DETACHED);
303 root 1.4
304     sigfillset (&fullsigset);
305     sigprocmask (SIG_SETMASK, &fullsigset, &oldsigset);
306    
307     if (pthread_create (&tid, &attr, aio_proc, 0) == 0)
308     started++;
309    
310     sigprocmask (SIG_SETMASK, &oldsigset, 0);
311     }
312    
313     static void
314 root 1.43 req_send (aio_req req)
315 root 1.4 {
316 root 1.30 while (started < wanted && nreqs >= started)
317     start_thread ();
318    
319 root 1.4 nreqs++;
320    
321     pthread_mutex_lock (&reqlock);
322    
323     req->next = 0;
324    
325     if (reqe)
326     {
327     reqe->next = req;
328     reqe = req;
329     }
330     else
331     reqe = reqs = req;
332    
333     pthread_cond_signal (&reqwait);
334     pthread_mutex_unlock (&reqlock);
335    
336 root 1.27 if (nreqs > max_outstanding)
337     for (;;)
338     {
339     poll_cb ();
340    
341     if (nreqs <= max_outstanding)
342     break;
343    
344     poll_wait ();
345     }
346 root 1.4 }
347    
348     static void
349     end_thread (void)
350     {
351     aio_req req;
352 root 1.26 Newz (0, req, 1, aio_cb);
353 root 1.4 req->type = REQ_QUIT;
354    
355 root 1.43 req_send (req);
356 root 1.4 }
357    
358 root 1.22 static void min_parallel (int nthreads)
359     {
360 root 1.30 if (wanted < nthreads)
361     wanted = nthreads;
362 root 1.22 }
363    
364     static void max_parallel (int nthreads)
365     {
366     int cur = started;
367 root 1.27
368 root 1.30 if (wanted > nthreads)
369     wanted = nthreads;
370    
371     while (cur > wanted)
372     {
373 root 1.22 end_thread ();
374     cur--;
375     }
376    
377 root 1.30 while (started > wanted)
378 root 1.22 {
379     poll_wait ();
380     poll_cb ();
381     }
382     }
383    
384 root 1.26 static void create_pipe ()
385 root 1.22 {
386 root 1.26 if (pipe (respipe))
387     croak ("unable to initialize result pipe");
388 root 1.22
389 root 1.26 if (fcntl (respipe [0], F_SETFL, O_NONBLOCK))
390     croak ("cannot set result pipe to nonblocking mode");
391 root 1.22
392 root 1.26 if (fcntl (respipe [1], F_SETFL, O_NONBLOCK))
393     croak ("cannot set result pipe to nonblocking mode");
394     }
395 root 1.22
396     /*****************************************************************************/
397 root 1.17 /* work around various missing functions */
398    
399     #if !HAVE_PREADWRITE
400     # define pread aio_pread
401     # define pwrite aio_pwrite
402    
403     /*
404     * make our pread/pwrite safe against themselves, but not against
405     * normal read/write by using a mutex. slows down execution a lot,
406     * but that's your problem, not mine.
407     */
408 root 1.37 static pthread_mutex_t preadwritelock = PTHREAD_MUTEX_INITIALIZER;
409 root 1.17
410     static ssize_t
411     pread (int fd, void *buf, size_t count, off_t offset)
412     {
413     ssize_t res;
414     off_t ooffset;
415    
416 root 1.37 pthread_mutex_lock (&preadwritelock);
417 root 1.17 ooffset = lseek (fd, 0, SEEK_CUR);
418     lseek (fd, offset, SEEK_SET);
419     res = read (fd, buf, count);
420     lseek (fd, ooffset, SEEK_SET);
421 root 1.37 pthread_mutex_unlock (&preadwritelock);
422 root 1.17
423     return res;
424     }
425    
426     static ssize_t
427     pwrite (int fd, void *buf, size_t count, off_t offset)
428     {
429     ssize_t res;
430     off_t ooffset;
431    
432 root 1.37 pthread_mutex_lock (&preadwritelock);
433 root 1.17 ooffset = lseek (fd, 0, SEEK_CUR);
434     lseek (fd, offset, SEEK_SET);
435     res = write (fd, buf, count);
436     lseek (fd, offset, SEEK_SET);
437 root 1.37 pthread_mutex_unlock (&preadwritelock);
438 root 1.17
439     return res;
440     }
441     #endif
442    
443     #if !HAVE_FDATASYNC
444     # define fdatasync fsync
445     #endif
446    
447     #if !HAVE_READAHEAD
448     # define readahead aio_readahead
449    
450     static ssize_t
451     readahead (int fd, off_t offset, size_t count)
452     {
453 root 1.37 char readahead_buf[4096];
454    
455 root 1.17 while (count > 0)
456     {
457     size_t len = count < sizeof (readahead_buf) ? count : sizeof (readahead_buf);
458    
459     pread (fd, readahead_buf, len, offset);
460     offset += len;
461     count -= len;
462     }
463    
464     errno = 0;
465     }
466     #endif
467    
468 root 1.37 #if !HAVE_READDIR_R
469     # define readdir_r aio_readdir_r
470    
471     static pthread_mutex_t readdirlock = PTHREAD_MUTEX_INITIALIZER;
472    
473     static int
474     readdir_r (DIR *dirp, struct dirent *ent, struct dirent **res)
475     {
476     struct dirent *e;
477     int errorno;
478    
479     pthread_mutex_lock (&readdirlock);
480    
481     e = readdir (dirp);
482     errorno = errno;
483    
484     if (e)
485     {
486     *res = ent;
487     strcpy (ent->d_name, e->d_name);
488     }
489     else
490     *res = 0;
491    
492     pthread_mutex_unlock (&readdirlock);
493    
494     errno = errorno;
495     return e ? 0 : -1;
496     }
497     #endif
498    
499 root 1.32 /* sendfile always needs emulation */
500     static ssize_t
501     sendfile_ (int ofd, int ifd, off_t offset, size_t count)
502     {
503 root 1.37 ssize_t res;
504 root 1.32
505 root 1.37 if (!count)
506     return 0;
507 root 1.32
508 root 1.35 #if HAVE_SENDFILE
509     # if __linux
510 root 1.37 res = sendfile (ofd, ifd, &offset, count);
511 root 1.32
512 root 1.35 # elif __freebsd
513 root 1.37 /*
514     * Of course, the freebsd sendfile is a dire hack with no thoughts
515     * wasted on making it similar to other I/O functions.
516     */
517     {
518     off_t sbytes;
519     res = sendfile (ifd, ofd, offset, count, 0, &sbytes, 0);
520    
521     if (res < 0 && sbytes)
522     /* maybe only on EAGAIN only: as usual, the manpage leaves you guessing */
523     res = sbytes;
524     }
525 root 1.32
526 root 1.35 # elif __hpux
527 root 1.37 res = sendfile (ofd, ifd, offset, count, 0, 0);
528 root 1.32
529 root 1.35 # elif __solaris
530 root 1.37 {
531     struct sendfilevec vec;
532     size_t sbytes;
533    
534     vec.sfv_fd = ifd;
535     vec.sfv_flag = 0;
536     vec.sfv_off = offset;
537     vec.sfv_len = count;
538    
539     res = sendfilev (ofd, &vec, 1, &sbytes);
540    
541     if (res < 0 && sbytes)
542     res = sbytes;
543     }
544 root 1.35
545 root 1.38 # endif
546     #else
547 root 1.37 res = -1;
548     errno = ENOSYS;
549 root 1.32 #endif
550    
551 root 1.37 if (res < 0
552     && (errno == ENOSYS || errno == EINVAL || errno == ENOTSOCK
553 root 1.35 #if __solaris
554 root 1.37 || errno == EAFNOSUPPORT || errno == EPROTOTYPE
555 root 1.35 #endif
556 root 1.37 )
557     )
558     {
559     /* emulate sendfile. this is a major pain in the ass */
560     char buf[4096];
561     res = 0;
562    
563 root 1.38 while (count)
564 root 1.37 {
565     ssize_t cnt;
566    
567 root 1.38 cnt = pread (ifd, buf, count > 4096 ? 4096 : count, offset);
568 root 1.32
569 root 1.37 if (cnt <= 0)
570     {
571     if (cnt && !res) res = -1;
572     break;
573     }
574    
575     cnt = write (ofd, buf, cnt);
576    
577     if (cnt <= 0)
578     {
579     if (cnt && !res) res = -1;
580     break;
581     }
582    
583     offset += cnt;
584     res += cnt;
585 root 1.38 count -= cnt;
586 root 1.37 }
587     }
588    
589     return res;
590     }
591    
592     /* read a full directory */
593     static int
594     scandir_ (const char *path, void **namesp)
595     {
596     DIR *dirp = opendir (path);
597     union
598     {
599     struct dirent d;
600     char b [offsetof (struct dirent, d_name) + NAME_MAX + 1];
601     } u;
602     struct dirent *entp;
603     char *name, *names;
604     int memlen = 4096;
605     int memofs = 0;
606     int res = 0;
607     int errorno;
608    
609     if (!dirp)
610     return -1;
611    
612     names = malloc (memlen);
613    
614     for (;;)
615     {
616     errno = 0, readdir_r (dirp, &u.d, &entp);
617    
618     if (!entp)
619     break;
620    
621     name = entp->d_name;
622    
623     if (name [0] != '.' || (name [1] && (name [1] != '.' || name [2])))
624     {
625     int len = strlen (name) + 1;
626    
627     res++;
628    
629     while (memofs + len > memlen)
630     {
631     memlen *= 2;
632     names = realloc (names, memlen);
633     if (!names)
634     break;
635     }
636    
637     memcpy (names + memofs, name, len);
638     memofs += len;
639     }
640     }
641    
642     errorno = errno;
643     closedir (dirp);
644    
645     if (errorno)
646     {
647     free (names);
648     errno = errorno;
649     res = -1;
650     }
651    
652     *namesp = (void *)names;
653     return res;
654 root 1.32 }
655    
656 root 1.22 /*****************************************************************************/
657    
658 root 1.1 static void *
659     aio_proc (void *thr_arg)
660     {
661     aio_req req;
662 root 1.3 int type;
663 root 1.1
664 root 1.3 do
665 root 1.1 {
666 root 1.3 pthread_mutex_lock (&reqlock);
667    
668     for (;;)
669     {
670     req = reqs;
671    
672     if (reqs)
673     {
674     reqs = reqs->next;
675     if (!reqs) reqe = 0;
676     }
677    
678     if (req)
679     break;
680    
681     pthread_cond_wait (&reqwait, &reqlock);
682     }
683    
684     pthread_mutex_unlock (&reqlock);
685    
686 root 1.1 errno = 0; /* strictly unnecessary */
687    
688 root 1.43 if (!req->cancelled)
689     switch (req->type)
690     {
691     case REQ_READ: req->result = pread (req->fd, req->dataptr, req->length, req->offset); break;
692     case REQ_WRITE: req->result = pwrite (req->fd, req->dataptr, req->length, req->offset); break;
693 root 1.3
694 root 1.43 case REQ_READAHEAD: req->result = readahead (req->fd, req->offset, req->length); break;
695     case REQ_SENDFILE: req->result = sendfile_ (req->fd, req->fd2, req->offset, req->length); break;
696 root 1.16
697 root 1.43 case REQ_STAT: req->result = stat (req->dataptr, req->statdata); break;
698     case REQ_LSTAT: req->result = lstat (req->dataptr, req->statdata); break;
699     case REQ_FSTAT: req->result = fstat (req->fd , req->statdata); break;
700    
701     case REQ_OPEN: req->result = open (req->dataptr, req->fd, req->mode); break;
702     case REQ_CLOSE: req->result = close (req->fd); break;
703     case REQ_UNLINK: req->result = unlink (req->dataptr); break;
704     case REQ_RMDIR: req->result = rmdir (req->dataptr); break;
705     case REQ_RENAME: req->result = rename (req->data2ptr, req->dataptr); break;
706     case REQ_LINK: req->result = link (req->data2ptr, req->dataptr); break;
707     case REQ_SYMLINK: req->result = symlink (req->data2ptr, req->dataptr); break;
708    
709     case REQ_FDATASYNC: req->result = fdatasync (req->fd); break;
710     case REQ_FSYNC: req->result = fsync (req->fd); break;
711     case REQ_READDIR: req->result = scandir_ (req->dataptr, &req->data2ptr); break;
712 root 1.1
713 root 1.43 case REQ_QUIT:
714     break;
715 root 1.1
716 root 1.43 default:
717     req->result = ENOSYS;
718     break;
719     }
720 root 1.1
721     req->errorno = errno;
722 root 1.3
723     pthread_mutex_lock (&reslock);
724    
725     req->next = 0;
726    
727     if (rese)
728     {
729     rese->next = req;
730     rese = req;
731     }
732     else
733     {
734     rese = ress = req;
735    
736     /* write a dummy byte to the pipe so fh becomes ready */
737     write (respipe [1], &respipe, 1);
738     }
739    
740     pthread_mutex_unlock (&reslock);
741 root 1.1 }
742 root 1.3 while (type != REQ_QUIT);
743 root 1.1
744     return 0;
745     }
746    
747 root 1.37 /*****************************************************************************/
748    
749     static void atfork_prepare (void)
750     {
751     pthread_mutex_lock (&reqlock);
752     pthread_mutex_lock (&reslock);
753     #if !HAVE_PREADWRITE
754     pthread_mutex_lock (&preadwritelock);
755     #endif
756     #if !HAVE_READDIR_R
757     pthread_mutex_lock (&readdirlock);
758     #endif
759     }
760    
761     static void atfork_parent (void)
762     {
763     #if !HAVE_READDIR_R
764     pthread_mutex_unlock (&readdirlock);
765     #endif
766     #if !HAVE_PREADWRITE
767     pthread_mutex_unlock (&preadwritelock);
768     #endif
769     pthread_mutex_unlock (&reslock);
770     pthread_mutex_unlock (&reqlock);
771     }
772    
773     static void atfork_child (void)
774     {
775     aio_req prv;
776    
777     started = 0;
778    
779     while (reqs)
780     {
781     prv = reqs;
782     reqs = prv->next;
783 root 1.43 req_free (prv);
784 root 1.37 }
785    
786     reqs = reqe = 0;
787    
788     while (ress)
789     {
790     prv = ress;
791     ress = prv->next;
792 root 1.43 req_free (prv);
793 root 1.37 }
794    
795     ress = rese = 0;
796    
797     close (respipe [0]);
798     close (respipe [1]);
799     create_pipe ();
800    
801     atfork_parent ();
802     }
803    
804 root 1.22 #define dREQ \
805     aio_req req; \
806     \
807     if (SvOK (callback) && !SvROK (callback)) \
808 root 1.43 croak ("callback must be undef or of reference type"); \
809 root 1.22 \
810     Newz (0, req, 1, aio_cb); \
811     if (!req) \
812     croak ("out of memory during aio_req allocation"); \
813     \
814 root 1.43 req->callback = newSVsv (callback)
815    
816     #define REQ_SEND \
817     req_send (req); \
818     \
819     if (GIMME_V != G_VOID) \
820 root 1.44 XPUSHs (req_sv (req, AIO_REQ_KLASS));
821 root 1.22
822 root 1.1 MODULE = IO::AIO PACKAGE = IO::AIO
823    
824 root 1.8 PROTOTYPES: ENABLE
825    
826 root 1.1 BOOT:
827     {
828 root 1.41 HV *stash = gv_stashpv ("IO::AIO", 1);
829     newCONSTSUB (stash, "EXDEV", newSViv (EXDEV));
830     newCONSTSUB (stash, "O_RDONLY", newSViv (O_RDONLY));
831     newCONSTSUB (stash, "O_WRONLY", newSViv (O_WRONLY));
832    
833 root 1.26 create_pipe ();
834 root 1.22 pthread_atfork (atfork_prepare, atfork_parent, atfork_child);
835 root 1.1 }
836    
837     void
838 root 1.40 min_parallel (nthreads)
839 root 1.1 int nthreads
840     PROTOTYPE: $
841    
842     void
843 root 1.40 max_parallel (nthreads)
844 root 1.1 int nthreads
845     PROTOTYPE: $
846    
847 root 1.4 int
848 root 1.40 max_outstanding (nreqs)
849 root 1.4 int nreqs
850     PROTOTYPE: $
851     CODE:
852     RETVAL = max_outstanding;
853     max_outstanding = nreqs;
854    
855 root 1.1 void
856 root 1.40 aio_open (pathname,flags,mode,callback=&PL_sv_undef)
857 root 1.1 SV * pathname
858     int flags
859     int mode
860     SV * callback
861 root 1.8 PROTOTYPE: $$$;$
862 root 1.43 PPCODE:
863 root 1.1 {
864 root 1.22 dREQ;
865 root 1.1
866     req->type = REQ_OPEN;
867     req->data = newSVsv (pathname);
868 root 1.22 req->dataptr = SvPVbyte_nolen (req->data);
869 root 1.1 req->fd = flags;
870     req->mode = mode;
871    
872 root 1.43 REQ_SEND;
873 root 1.1 }
874    
875     void
876 root 1.40 aio_close (fh,callback=&PL_sv_undef)
877 root 1.13 SV * fh
878     SV * callback
879 root 1.8 PROTOTYPE: $;$
880 root 1.1 ALIAS:
881     aio_close = REQ_CLOSE
882     aio_fsync = REQ_FSYNC
883     aio_fdatasync = REQ_FDATASYNC
884 root 1.43 PPCODE:
885 root 1.1 {
886 root 1.22 dREQ;
887 root 1.1
888     req->type = ix;
889 root 1.13 req->fh = newSVsv (fh);
890     req->fd = PerlIO_fileno (IoIFP (sv_2io (fh)));
891 root 1.1
892 root 1.43 REQ_SEND (req);
893 root 1.1 }
894    
895     void
896 root 1.40 aio_read (fh,offset,length,data,dataoffset,callback=&PL_sv_undef)
897 root 1.13 SV * fh
898     UV offset
899 root 1.32 UV length
900 root 1.13 SV * data
901 root 1.32 UV dataoffset
902 root 1.13 SV * callback
903     ALIAS:
904     aio_read = REQ_READ
905     aio_write = REQ_WRITE
906 root 1.8 PROTOTYPE: $$$$$;$
907 root 1.43 PPCODE:
908 root 1.13 {
909     aio_req req;
910     STRLEN svlen;
911 root 1.21 char *svptr = SvPVbyte (data, svlen);
912 root 1.13
913     SvUPGRADE (data, SVt_PV);
914     SvPOK_on (data);
915 root 1.1
916 root 1.13 if (dataoffset < 0)
917     dataoffset += svlen;
918    
919     if (dataoffset < 0 || dataoffset > svlen)
920     croak ("data offset outside of string");
921    
922     if (ix == REQ_WRITE)
923     {
924     /* write: check length and adjust. */
925     if (length < 0 || length + dataoffset > svlen)
926     length = svlen - dataoffset;
927     }
928     else
929     {
930     /* read: grow scalar as necessary */
931     svptr = SvGROW (data, length + dataoffset);
932     }
933    
934     if (length < 0)
935     croak ("length must not be negative");
936    
937 root 1.22 {
938     dREQ;
939 root 1.13
940 root 1.22 req->type = ix;
941     req->fh = newSVsv (fh);
942     req->fd = PerlIO_fileno (ix == REQ_READ ? IoIFP (sv_2io (fh))
943     : IoOFP (sv_2io (fh)));
944     req->offset = offset;
945     req->length = length;
946     req->data = SvREFCNT_inc (data);
947     req->dataptr = (char *)svptr + dataoffset;
948 root 1.13
949 root 1.28 if (!SvREADONLY (data))
950     {
951     SvREADONLY_on (data);
952     req->data2ptr = (void *)data;
953     }
954    
955 root 1.43 REQ_SEND;
956 root 1.22 }
957 root 1.13 }
958 root 1.1
959     void
960 root 1.40 aio_sendfile (out_fh,in_fh,in_offset,length,callback=&PL_sv_undef)
961 root 1.32 SV * out_fh
962     SV * in_fh
963     UV in_offset
964     UV length
965     SV * callback
966     PROTOTYPE: $$$$;$
967 root 1.43 PPCODE:
968 root 1.32 {
969     dREQ;
970    
971     req->type = REQ_SENDFILE;
972     req->fh = newSVsv (out_fh);
973     req->fd = PerlIO_fileno (IoIFP (sv_2io (out_fh)));
974     req->fh2 = newSVsv (in_fh);
975     req->fd2 = PerlIO_fileno (IoIFP (sv_2io (in_fh)));
976     req->offset = in_offset;
977     req->length = length;
978    
979 root 1.43 REQ_SEND;
980 root 1.32 }
981    
982     void
983 root 1.40 aio_readahead (fh,offset,length,callback=&PL_sv_undef)
984 root 1.13 SV * fh
985     UV offset
986     IV length
987     SV * callback
988 root 1.8 PROTOTYPE: $$$;$
989 root 1.43 PPCODE:
990 root 1.1 {
991 root 1.22 dREQ;
992 root 1.1
993     req->type = REQ_READAHEAD;
994 root 1.13 req->fh = newSVsv (fh);
995     req->fd = PerlIO_fileno (IoIFP (sv_2io (fh)));
996 root 1.1 req->offset = offset;
997     req->length = length;
998    
999 root 1.43 REQ_SEND;
1000 root 1.1 }
1001    
1002     void
1003 root 1.40 aio_stat (fh_or_path,callback=&PL_sv_undef)
1004 root 1.1 SV * fh_or_path
1005     SV * callback
1006     ALIAS:
1007 root 1.8 aio_stat = REQ_STAT
1008     aio_lstat = REQ_LSTAT
1009 root 1.43 PPCODE:
1010 root 1.1 {
1011 root 1.22 dREQ;
1012 root 1.1
1013     New (0, req->statdata, 1, Stat_t);
1014     if (!req->statdata)
1015 root 1.27 {
1016 root 1.43 req_free (req);
1017 root 1.27 croak ("out of memory during aio_req->statdata allocation");
1018     }
1019 root 1.1
1020     if (SvPOK (fh_or_path))
1021     {
1022 root 1.8 req->type = ix;
1023 root 1.1 req->data = newSVsv (fh_or_path);
1024 root 1.22 req->dataptr = SvPVbyte_nolen (req->data);
1025 root 1.1 }
1026     else
1027     {
1028     req->type = REQ_FSTAT;
1029 root 1.13 req->fh = newSVsv (fh_or_path);
1030 root 1.1 req->fd = PerlIO_fileno (IoIFP (sv_2io (fh_or_path)));
1031     }
1032    
1033 root 1.43 REQ_SEND;
1034 root 1.1 }
1035    
1036     void
1037 root 1.40 aio_unlink (pathname,callback=&PL_sv_undef)
1038 root 1.1 SV * pathname
1039     SV * callback
1040 root 1.22 ALIAS:
1041 root 1.40 aio_unlink = REQ_UNLINK
1042     aio_rmdir = REQ_RMDIR
1043     aio_readdir = REQ_READDIR
1044 root 1.43 PPCODE:
1045 root 1.1 {
1046 root 1.22 dREQ;
1047 root 1.1
1048 root 1.22 req->type = ix;
1049     req->data = newSVsv (pathname);
1050     req->dataptr = SvPVbyte_nolen (req->data);
1051 root 1.1
1052 root 1.43 REQ_SEND;
1053 root 1.22 }
1054    
1055     void
1056 root 1.40 aio_link (oldpath,newpath,callback=&PL_sv_undef)
1057 root 1.22 SV * oldpath
1058     SV * newpath
1059     SV * callback
1060 root 1.40 ALIAS:
1061     aio_link = REQ_LINK
1062     aio_symlink = REQ_SYMLINK
1063     aio_rename = REQ_RENAME
1064 root 1.43 PPCODE:
1065 root 1.22 {
1066     dREQ;
1067 root 1.1
1068 root 1.40 req->type = ix;
1069 root 1.22 req->fh = newSVsv (oldpath);
1070     req->data2ptr = SvPVbyte_nolen (req->fh);
1071     req->data = newSVsv (newpath);
1072     req->dataptr = SvPVbyte_nolen (req->data);
1073 root 1.1
1074 root 1.43 REQ_SEND;
1075 root 1.1 }
1076    
1077 root 1.42 void
1078 root 1.44 aio_group (callback=&PL_sv_undef)
1079     SV * callback
1080     PROTOTYPE: ;&
1081     PPCODE:
1082 root 1.42 {
1083 root 1.44 dREQ;
1084     req->type = REQ_GROUP;
1085     XPUSHs (req_sv (req, AIO_GRP_KLASS));
1086 root 1.42 }
1087    
1088 root 1.6 void
1089 root 1.40 flush ()
1090 root 1.6 PROTOTYPE:
1091     CODE:
1092     while (nreqs)
1093     {
1094     poll_wait ();
1095     poll_cb ();
1096     }
1097    
1098 root 1.7 void
1099     poll()
1100     PROTOTYPE:
1101     CODE:
1102     if (nreqs)
1103     {
1104     poll_wait ();
1105     poll_cb ();
1106     }
1107    
1108 root 1.1 int
1109     poll_fileno()
1110     PROTOTYPE:
1111     CODE:
1112 root 1.3 RETVAL = respipe [0];
1113 root 1.1 OUTPUT:
1114     RETVAL
1115    
1116     int
1117     poll_cb(...)
1118     PROTOTYPE:
1119     CODE:
1120 root 1.5 RETVAL = poll_cb ();
1121 root 1.1 OUTPUT:
1122     RETVAL
1123    
1124     void
1125     poll_wait()
1126     PROTOTYPE:
1127     CODE:
1128 root 1.3 if (nreqs)
1129     poll_wait ();
1130 root 1.1
1131     int
1132     nreqs()
1133     PROTOTYPE:
1134     CODE:
1135     RETVAL = nreqs;
1136     OUTPUT:
1137     RETVAL
1138    
1139 root 1.44 MODULE = IO::AIO PACKAGE = IO::AIO::REQ
1140 root 1.43
1141     void
1142     cancel (aio_req_ornot req)
1143     PROTOTYPE:
1144     CODE:
1145     req->cancelled = 1;
1146