ViewVC Help
View File | Revision Log | Show Annotations | Download File
/cvs/IO-AIO/AIO.xs
(Generate patch)

Comparing IO-AIO/AIO.xs (file contents):
Revision 1.2 by root, Sun Jul 10 18:16:49 2005 UTC vs.
Revision 1.18 by root, Sun Jul 31 19:00:31 2005 UTC

1#define PERL_NO_GET_CONTEXT
2
3#include "EXTERN.h" 1#include "EXTERN.h"
4#include "perl.h" 2#include "perl.h"
5#include "XSUB.h" 3#include "XSUB.h"
6 4
5#include "autoconf/config.h"
6
7#include <sys/types.h> 7#include <sys/types.h>
8#include <sys/stat.h> 8#include <sys/stat.h>
9
9#include <unistd.h> 10#include <unistd.h>
10#include <fcntl.h> 11#include <fcntl.h>
11#include <signal.h> 12#include <signal.h>
12#include <sched.h> 13#include <sched.h>
14
15#ifndef _REENTRANT
16# define _REENTRANT 1
17#endif
13#include <endian.h> 18#include <errno.h>
14 19
15#include <pthread.h> 20#include <pthread.h>
16#include <sys/syscall.h>
17 21
18typedef void *InputStream; /* hack, but 5.6.1 is simply toooo old ;) */ 22typedef void *InputStream; /* hack, but 5.6.1 is simply toooo old ;) */
19typedef void *OutputStream; /* hack, but 5.6.1 is simply toooo old ;) */ 23typedef void *OutputStream; /* hack, but 5.6.1 is simply toooo old ;) */
20typedef void *InOutStream; /* hack, but 5.6.1 is simply toooo old ;) */ 24typedef void *InOutStream; /* hack, but 5.6.1 is simply toooo old ;) */
21 25
22#if __i386 || __amd64
23# define STACKSIZE ( 256 * sizeof (long))
24#elif __ia64 26#if __ia64
25# define STACKSIZE (8192 * sizeof (long)) 27# define STACKSIZE 65536
26#else 28#else
27# define STACKSIZE ( 512 * sizeof (long)) 29# define STACKSIZE 4096
28#endif 30#endif
29 31
30enum { 32enum {
31 REQ_QUIT, 33 REQ_QUIT,
32 REQ_OPEN, REQ_CLOSE, 34 REQ_OPEN, REQ_CLOSE,
34 REQ_STAT, REQ_LSTAT, REQ_FSTAT, REQ_UNLINK, 36 REQ_STAT, REQ_LSTAT, REQ_FSTAT, REQ_UNLINK,
35 REQ_FSYNC, REQ_FDATASYNC, 37 REQ_FSYNC, REQ_FDATASYNC,
36}; 38};
37 39
38typedef struct aio_cb { 40typedef struct aio_cb {
39 struct aio_cb *next; 41 struct aio_cb *volatile next;
40 42
41 int type; 43 int type;
42 44
43 int fd; 45 int fd;
44 off_t offset; 46 off_t offset;
45 size_t length; 47 size_t length;
46 ssize_t result; 48 ssize_t result;
47 mode_t mode; /* open */ 49 mode_t mode; /* open */
48 int errorno; 50 int errorno;
49 SV *data, *callback; 51 SV *data, *callback, *fh;
50 void *dataptr; 52 void *dataptr;
51 STRLEN dataoffset; 53 STRLEN dataoffset;
52 54
53 Stat_t *statdata; 55 Stat_t *statdata;
54} aio_cb; 56} aio_cb;
55 57
56typedef aio_cb *aio_req; 58typedef aio_cb *aio_req;
57 59
58static int started; 60static int started;
59static int nreqs; 61static volatile int nreqs;
62static int max_outstanding = 1<<30;
60static int reqpipe[2], respipe[2]; 63static int respipe [2];
61 64
65static pthread_mutex_t reslock = PTHREAD_MUTEX_INITIALIZER;
66static pthread_mutex_t reqlock = PTHREAD_MUTEX_INITIALIZER;
67static pthread_cond_t reqwait = PTHREAD_COND_INITIALIZER;
68
62static aio_req qs, qe; /* queue start, queue end */ 69static volatile aio_req reqs, reqe; /* queue start, queue end */
63 70static volatile aio_req ress, rese; /* queue start, queue end */
64static void *aio_proc(void *arg);
65
66static void
67start_thread (void)
68{
69 sigset_t fullsigset, oldsigset;
70 pthread_t tid;
71 pthread_attr_t attr;
72
73 pthread_attr_init (&attr);
74 pthread_attr_setstacksize (&attr, STACKSIZE);
75 pthread_attr_setdetachstate (&attr, PTHREAD_CREATE_DETACHED);
76
77 sigfillset (&fullsigset);
78 sigprocmask (SIG_SETMASK, &fullsigset, &oldsigset);
79
80 if (pthread_create (&tid, &attr, aio_proc, 0) == 0)
81 started++;
82
83 sigprocmask (SIG_SETMASK, &oldsigset, 0);
84}
85
86static void
87send_reqs (void)
88{
89 /* this write is atomic */
90 while (qs && write (reqpipe[1], &qs, sizeof qs) == sizeof qs)
91 {
92 qs = qs->next;
93 if (!qs) qe = 0;
94 }
95}
96
97static void
98send_req (aio_req req)
99{
100 nreqs++;
101 req->next = 0;
102
103 if (qe)
104 {
105 qe->next = req;
106 qe = req;
107 }
108 else
109 qe = qs = req;
110
111 send_reqs ();
112}
113
114static void
115end_thread (void)
116{
117 aio_req req;
118 New (0, req, 1, aio_cb);
119 req->type = REQ_QUIT;
120
121 send_req (req);
122}
123
124static void
125read_write (pTHX_
126 int dowrite, int fd, off_t offset, size_t length,
127 SV *data, STRLEN dataoffset, SV *callback)
128{
129 aio_req req;
130 STRLEN svlen;
131 char *svptr = SvPV (data, svlen);
132
133 SvUPGRADE (data, SVt_PV);
134 SvPOK_on (data);
135
136 if (dataoffset < 0)
137 dataoffset += svlen;
138
139 if (dataoffset < 0 || dataoffset > svlen)
140 croak ("data offset outside of string");
141
142 if (dowrite)
143 {
144 /* write: check length and adjust. */
145 if (length < 0 || length + dataoffset > svlen)
146 length = svlen - dataoffset;
147 }
148 else
149 {
150 /* read: grow scalar as necessary */
151 svptr = SvGROW (data, length + dataoffset);
152 }
153
154 if (length < 0)
155 croak ("length must not be negative");
156
157 Newz (0, req, 1, aio_cb);
158
159 if (!req)
160 croak ("out of memory during aio_req allocation");
161
162 req->type = dowrite ? REQ_WRITE : REQ_READ;
163 req->fd = fd;
164 req->offset = offset;
165 req->length = length;
166 req->data = SvREFCNT_inc (data);
167 req->dataptr = (char *)svptr + dataoffset;
168 req->callback = SvREFCNT_inc (callback);
169
170 send_req (req);
171}
172 71
173static void 72static void
174poll_wait () 73poll_wait ()
175{ 74{
75 if (nreqs && !ress)
76 {
176 fd_set rfd; 77 fd_set rfd;
177 FD_ZERO(&rfd); 78 FD_ZERO(&rfd);
178 FD_SET(respipe[0], &rfd); 79 FD_SET(respipe [0], &rfd);
179 80
180 select (respipe[0] + 1, &rfd, 0, 0, 0); 81 select (respipe [0] + 1, &rfd, 0, 0, 0);
82 }
181} 83}
182 84
183static int 85static int
184poll_cb (pTHX) 86poll_cb ()
185{ 87{
186 dSP; 88 dSP;
187 int count = 0; 89 int count = 0;
188 aio_req req; 90 aio_req req, prv;
189 91
190 while (read (respipe[0], (void *)&req, sizeof (req)) == sizeof (req)) 92 pthread_mutex_lock (&reslock);
93
94 {
95 /* read any signals sent by the worker threads */
96 char buf [32];
97 while (read (respipe [0], buf, 32) > 0)
98 ;
99 }
100
101 req = ress;
102 ress = rese = 0;
103
104 pthread_mutex_unlock (&reslock);
105
106 while (req)
191 { 107 {
192 nreqs--; 108 nreqs--;
193 109
194 if (req->type == REQ_QUIT) 110 if (req->type == REQ_QUIT)
195 started--; 111 started--;
203 + req->result > 0 ? req->result : 0); 119 + req->result > 0 ? req->result : 0);
204 120
205 if (req->data) 121 if (req->data)
206 SvREFCNT_dec (req->data); 122 SvREFCNT_dec (req->data);
207 123
124 if (req->fh)
125 SvREFCNT_dec (req->fh);
126
208 if (req->type == REQ_STAT || req->type == REQ_LSTAT || req->type == REQ_FSTAT) 127 if (req->type == REQ_STAT || req->type == REQ_LSTAT || req->type == REQ_FSTAT)
209 { 128 {
210 PL_laststype = req->type == REQ_LSTAT ? OP_LSTAT : OP_STAT; 129 PL_laststype = req->type == REQ_LSTAT ? OP_LSTAT : OP_STAT;
211 PL_laststatval = req->result; 130 PL_laststatval = req->result;
212 PL_statcache = *(req->statdata); 131 PL_statcache = *(req->statdata);
213 132
214 Safefree (req->statdata); 133 Safefree (req->statdata);
215 } 134 }
216 135
136 ENTER;
217 PUSHMARK (SP); 137 PUSHMARK (SP);
218 XPUSHs (sv_2mortal (newSViv (req->result))); 138 XPUSHs (sv_2mortal (newSViv (req->result)));
219 139
220 if (req->type == REQ_OPEN) 140 if (req->type == REQ_OPEN)
221 { 141 {
224 144
225 PUTBACK; 145 PUTBACK;
226 call_pv ("IO::AIO::_fd2fh", G_SCALAR | G_EVAL); 146 call_pv ("IO::AIO::_fd2fh", G_SCALAR | G_EVAL);
227 SPAGAIN; 147 SPAGAIN;
228 148
229 fh = POPs; 149 fh = SvREFCNT_inc (POPs);
230 150
231 PUSHMARK (SP); 151 PUSHMARK (SP);
232 XPUSHs (fh); 152 XPUSHs (sv_2mortal (fh));
233 } 153 }
234 154
155 if (SvOK (req->callback))
156 {
235 PUTBACK; 157 PUTBACK;
236 call_sv (req->callback, G_VOID | G_EVAL); 158 call_sv (req->callback, G_VOID | G_EVAL);
237 SPAGAIN; 159 SPAGAIN;
160 }
161
162 LEAVE;
238 163
239 if (req->callback) 164 if (req->callback)
240 SvREFCNT_dec (req->callback); 165 SvREFCNT_dec (req->callback);
241 166
242 errno = errorno; 167 errno = errorno;
243 count++; 168 count++;
244 } 169 }
245 170
171 prv = req;
172 req = req->next;
246 Safefree (req); 173 Safefree (prv);
174
175 /* TODO: croak on errors? */
247 } 176 }
248 177
249 if (qs)
250 send_reqs ();
251
252 return count; 178 return count;
253} 179}
180
181static void *aio_proc(void *arg);
182
183static void
184start_thread (void)
185{
186 sigset_t fullsigset, oldsigset;
187 pthread_t tid;
188 pthread_attr_t attr;
189
190 pthread_attr_init (&attr);
191 pthread_attr_setstacksize (&attr, STACKSIZE);
192 pthread_attr_setdetachstate (&attr, PTHREAD_CREATE_DETACHED);
193
194 sigfillset (&fullsigset);
195 sigprocmask (SIG_SETMASK, &fullsigset, &oldsigset);
196
197 if (pthread_create (&tid, &attr, aio_proc, 0) == 0)
198 started++;
199
200 sigprocmask (SIG_SETMASK, &oldsigset, 0);
201}
202
203static void
204send_req (aio_req req)
205{
206 nreqs++;
207
208 pthread_mutex_lock (&reqlock);
209
210 req->next = 0;
211
212 if (reqe)
213 {
214 reqe->next = req;
215 reqe = req;
216 }
217 else
218 reqe = reqs = req;
219
220 pthread_cond_signal (&reqwait);
221 pthread_mutex_unlock (&reqlock);
222
223 while (nreqs > max_outstanding)
224 {
225 poll_wait ();
226 poll_cb ();
227 }
228}
229
230static void
231end_thread (void)
232{
233 aio_req req;
234 New (0, req, 1, aio_cb);
235 req->type = REQ_QUIT;
236
237 send_req (req);
238}
239
240/* work around various missing functions */
241
242#if !HAVE_PREADWRITE
243# define pread aio_pread
244# define pwrite aio_pwrite
245
246/*
247 * make our pread/pwrite safe against themselves, but not against
248 * normal read/write by using a mutex. slows down execution a lot,
249 * but that's your problem, not mine.
250 */
251static pthread_mutex_t iolock = PTHREAD_MUTEX_INITIALIZER;
252
253static ssize_t
254pread (int fd, void *buf, size_t count, off_t offset)
255{
256 ssize_t res;
257 off_t ooffset;
258
259 pthread_mutex_lock (&iolock);
260 ooffset = lseek (fd, 0, SEEK_CUR);
261 lseek (fd, offset, SEEK_SET);
262 res = read (fd, buf, count);
263 lseek (fd, ooffset, SEEK_SET);
264 pthread_mutex_unlock (&iolock);
265
266 return res;
267}
268
269static ssize_t
270pwrite (int fd, void *buf, size_t count, off_t offset)
271{
272 ssize_t res;
273 off_t ooffset;
274
275 pthread_mutex_lock (&iolock);
276 ooffset = lseek (fd, 0, SEEK_CUR);
277 lseek (fd, offset, SEEK_SET);
278 res = write (fd, buf, count);
279 lseek (fd, offset, SEEK_SET);
280 pthread_mutex_unlock (&iolock);
281
282 return res;
283}
284#endif
285
286#if !HAVE_FDATASYNC
287# define fdatasync fsync
288#endif
289
290#if !HAVE_READAHEAD
291# define readahead aio_readahead
292
293static char readahead_buf[4096];
294
295static ssize_t
296readahead (int fd, off_t offset, size_t count)
297{
298 while (count > 0)
299 {
300 size_t len = count < sizeof (readahead_buf) ? count : sizeof (readahead_buf);
301
302 pread (fd, readahead_buf, len, offset);
303 offset += len;
304 count -= len;
305 }
306
307 errno = 0;
308}
309#endif
254 310
255static void * 311static void *
256aio_proc (void *thr_arg) 312aio_proc (void *thr_arg)
257{ 313{
258 aio_req req; 314 aio_req req;
315 int type;
259 316
260 /* then loop */ 317 do
261 while (read (reqpipe[0], (void *)&req, sizeof (req)) == sizeof (req))
262 { 318 {
319 pthread_mutex_lock (&reqlock);
320
321 for (;;)
322 {
323 req = reqs;
324
325 if (reqs)
326 {
327 reqs = reqs->next;
328 if (!reqs) reqe = 0;
329 }
330
331 if (req)
332 break;
333
334 pthread_cond_wait (&reqwait, &reqlock);
335 }
336
337 pthread_mutex_unlock (&reqlock);
338
263 errno = 0; /* strictly unnecessary */ 339 errno = 0; /* strictly unnecessary */
264 340
341 type = req->type;
342
265 switch (req->type) 343 switch (type)
266 { 344 {
267 case REQ_READ: req->result = pread64 (req->fd, req->dataptr, req->length, req->offset); break; 345 case REQ_READ: req->result = pread (req->fd, req->dataptr, req->length, req->offset); break;
268 case REQ_WRITE: req->result = pwrite64 (req->fd, req->dataptr, req->length, req->offset); break; 346 case REQ_WRITE: req->result = pwrite (req->fd, req->dataptr, req->length, req->offset); break;
269#if SYS_readahead 347
270 case REQ_READAHEAD: req->result = readahead (req->fd, req->offset, req->length); break; 348 case REQ_READAHEAD: req->result = readahead (req->fd, req->offset, req->length); break;
271#else
272 case REQ_READAHEAD: req->result = -1; errno = ENOSYS; break;
273#endif
274 349
275 case REQ_STAT: req->result = stat (req->dataptr, req->statdata); break; 350 case REQ_STAT: req->result = stat (req->dataptr, req->statdata); break;
276 case REQ_LSTAT: req->result = lstat (req->dataptr, req->statdata); break; 351 case REQ_LSTAT: req->result = lstat (req->dataptr, req->statdata); break;
277 case REQ_FSTAT: req->result = fstat (req->fd , req->statdata); break; 352 case REQ_FSTAT: req->result = fstat (req->fd , req->statdata); break;
278 353
279 case REQ_OPEN: req->result = open (req->dataptr, req->fd, req->mode); break; 354 case REQ_OPEN: req->result = open (req->dataptr, req->fd, req->mode); break;
280 case REQ_CLOSE: req->result = close (req->fd); break; 355 case REQ_CLOSE: req->result = close (req->fd); break;
281 case REQ_UNLINK: req->result = unlink (req->dataptr); break; 356 case REQ_UNLINK: req->result = unlink (req->dataptr); break;
282 357
358 case REQ_FDATASYNC: req->result = fdatasync (req->fd); break;
283 case REQ_FSYNC: req->result = fsync (req->fd); break; 359 case REQ_FSYNC: req->result = fsync (req->fd); break;
284 case REQ_FDATASYNC: req->result = fdatasync (req->fd); break;
285 360
286 case REQ_QUIT: 361 case REQ_QUIT:
287 write (respipe[1], (void *)&req, sizeof (req)); 362 break;
288 return 0;
289 363
290 default: 364 default:
291 req->result = ENOSYS; 365 req->result = ENOSYS;
292 break; 366 break;
293 } 367 }
294 368
295 req->errorno = errno; 369 req->errorno = errno;
296 write (respipe[1], (void *)&req, sizeof (req)); 370
371 pthread_mutex_lock (&reslock);
372
373 req->next = 0;
374
375 if (rese)
376 {
377 rese->next = req;
378 rese = req;
379 }
380 else
381 {
382 rese = ress = req;
383
384 /* write a dummy byte to the pipe so fh becomes ready */
385 write (respipe [1], &respipe, 1);
386 }
387
388 pthread_mutex_unlock (&reslock);
297 } 389 }
390 while (type != REQ_QUIT);
298 391
299 return 0; 392 return 0;
300} 393}
301 394
302MODULE = IO::AIO PACKAGE = IO::AIO 395MODULE = IO::AIO PACKAGE = IO::AIO
303 396
397PROTOTYPES: ENABLE
398
304BOOT: 399BOOT:
305{ 400{
306 if (pipe (reqpipe) || pipe (respipe)) 401 if (pipe (respipe))
307 croak ("unable to initialize request or result pipe"); 402 croak ("unable to initialize result pipe");
308 403
309 if (fcntl (reqpipe[1], F_SETFL, O_NONBLOCK)) 404 if (fcntl (respipe [0], F_SETFL, O_NONBLOCK))
310 croak ("cannot set result pipe to nonblocking mode"); 405 croak ("cannot set result pipe to nonblocking mode");
311 406
312 if (fcntl (respipe[0], F_SETFL, O_NONBLOCK)) 407 if (fcntl (respipe [1], F_SETFL, O_NONBLOCK))
313 croak ("cannot set result pipe to nonblocking mode"); 408 croak ("cannot set result pipe to nonblocking mode");
314} 409}
315 410
316void 411void
317min_parallel(nthreads) 412min_parallel(nthreads)
335 } 430 }
336 431
337 while (started > nthreads) 432 while (started > nthreads)
338 { 433 {
339 poll_wait (); 434 poll_wait ();
340 poll_cb (aTHX); 435 poll_cb ();
341 } 436 }
342} 437}
343 438
439int
440max_outstanding(nreqs)
441 int nreqs
442 PROTOTYPE: $
443 CODE:
444 RETVAL = max_outstanding;
445 max_outstanding = nreqs;
446
344void 447void
345aio_open(pathname,flags,mode,callback) 448aio_open(pathname,flags,mode,callback=&PL_sv_undef)
346 SV * pathname 449 SV * pathname
347 int flags 450 int flags
348 int mode 451 int mode
349 SV * callback 452 SV * callback
350 PROTOTYPE: $$$$ 453 PROTOTYPE: $$$;$
351 CODE: 454 CODE:
352{ 455{
353 aio_req req; 456 aio_req req;
354 457
355 Newz (0, req, 1, aio_cb); 458 Newz (0, req, 1, aio_cb);
366 469
367 send_req (req); 470 send_req (req);
368} 471}
369 472
370void 473void
371aio_close(fh,callback) 474aio_close(fh,callback=&PL_sv_undef)
372 InputStream fh 475 SV * fh
373 SV * callback 476 SV * callback
374 PROTOTYPE: $$ 477 PROTOTYPE: $;$
375 ALIAS: 478 ALIAS:
376 aio_close = REQ_CLOSE 479 aio_close = REQ_CLOSE
377 aio_fsync = REQ_FSYNC 480 aio_fsync = REQ_FSYNC
378 aio_fdatasync = REQ_FDATASYNC 481 aio_fdatasync = REQ_FDATASYNC
379 CODE: 482 CODE:
384 487
385 if (!req) 488 if (!req)
386 croak ("out of memory during aio_req allocation"); 489 croak ("out of memory during aio_req allocation");
387 490
388 req->type = ix; 491 req->type = ix;
492 req->fh = newSVsv (fh);
389 req->fd = PerlIO_fileno (fh); 493 req->fd = PerlIO_fileno (IoIFP (sv_2io (fh)));
390 req->callback = SvREFCNT_inc (callback); 494 req->callback = SvREFCNT_inc (callback);
391 495
392 send_req (req); 496 send_req (req);
393} 497}
394 498
395void 499void
396aio_read(fh,offset,length,data,dataoffset,callback) 500aio_read(fh,offset,length,data,dataoffset,callback=&PL_sv_undef)
397 InputStream fh 501 SV * fh
398 UV offset 502 UV offset
399 IV length 503 IV length
400 SV * data 504 SV * data
401 IV dataoffset 505 IV dataoffset
402 SV * callback 506 SV * callback
507 ALIAS:
508 aio_read = REQ_READ
509 aio_write = REQ_WRITE
403 PROTOTYPE: $$$$$$ 510 PROTOTYPE: $$$$$;$
404 CODE: 511 CODE:
405 read_write (aTHX_ 0, PerlIO_fileno (fh), offset, length, data, dataoffset, callback);
406
407void
408aio_write(fh,offset,length,data,dataoffset,callback)
409 OutputStream fh
410 UV offset
411 IV length
412 SV * data
413 IV dataoffset
414 SV * callback
415 PROTOTYPE: $$$$$$
416 CODE:
417 read_write (aTHX_ 1, PerlIO_fileno (fh), offset, length, data, dataoffset, callback);
418
419void
420aio_readahead(fh,offset,length,callback)
421 InputStream fh
422 UV offset
423 IV length
424 SV * callback
425 PROTOTYPE: $$$$
426 CODE:
427{ 512{
428 aio_req req; 513 aio_req req;
514 STRLEN svlen;
515 char *svptr = SvPV (data, svlen);
516
517 SvUPGRADE (data, SVt_PV);
518 SvPOK_on (data);
519
520 if (dataoffset < 0)
521 dataoffset += svlen;
522
523 if (dataoffset < 0 || dataoffset > svlen)
524 croak ("data offset outside of string");
525
526 if (ix == REQ_WRITE)
527 {
528 /* write: check length and adjust. */
529 if (length < 0 || length + dataoffset > svlen)
530 length = svlen - dataoffset;
531 }
532 else
533 {
534 /* read: grow scalar as necessary */
535 svptr = SvGROW (data, length + dataoffset);
536 }
429 537
430 if (length < 0) 538 if (length < 0)
431 croak ("length must not be negative"); 539 croak ("length must not be negative");
432 540
433 Newz (0, req, 1, aio_cb); 541 Newz (0, req, 1, aio_cb);
434 542
435 if (!req) 543 if (!req)
436 croak ("out of memory during aio_req allocation"); 544 croak ("out of memory during aio_req allocation");
437 545
546 req->type = ix;
547 req->fh = newSVsv (fh);
548 req->fd = PerlIO_fileno (ix == REQ_READ ? IoIFP (sv_2io (fh))
549 : IoOFP (sv_2io (fh)));
550 req->offset = offset;
551 req->length = length;
552 req->data = SvREFCNT_inc (data);
553 req->dataptr = (char *)svptr + dataoffset;
554 req->callback = SvREFCNT_inc (callback);
555
556 send_req (req);
557}
558
559void
560aio_readahead(fh,offset,length,callback=&PL_sv_undef)
561 SV * fh
562 UV offset
563 IV length
564 SV * callback
565 PROTOTYPE: $$$;$
566 CODE:
567{
568 aio_req req;
569
570 if (length < 0)
571 croak ("length must not be negative");
572
573 Newz (0, req, 1, aio_cb);
574
575 if (!req)
576 croak ("out of memory during aio_req allocation");
577
438 req->type = REQ_READAHEAD; 578 req->type = REQ_READAHEAD;
579 req->fh = newSVsv (fh);
439 req->fd = PerlIO_fileno (fh); 580 req->fd = PerlIO_fileno (IoIFP (sv_2io (fh)));
440 req->offset = offset; 581 req->offset = offset;
441 req->length = length; 582 req->length = length;
442 req->callback = SvREFCNT_inc (callback); 583 req->callback = SvREFCNT_inc (callback);
443 584
444 send_req (req); 585 send_req (req);
445} 586}
446 587
447void 588void
448aio_stat(fh_or_path,callback) 589aio_stat(fh_or_path,callback=&PL_sv_undef)
449 SV * fh_or_path 590 SV * fh_or_path
450 SV * callback 591 SV * callback
451 PROTOTYPE: $$
452 ALIAS: 592 ALIAS:
593 aio_stat = REQ_STAT
453 aio_lstat = 1 594 aio_lstat = REQ_LSTAT
454 CODE: 595 CODE:
455{ 596{
456 aio_req req; 597 aio_req req;
457 598
458 Newz (0, req, 1, aio_cb); 599 Newz (0, req, 1, aio_cb);
465 if (!req->statdata) 606 if (!req->statdata)
466 croak ("out of memory during aio_req->statdata allocation"); 607 croak ("out of memory during aio_req->statdata allocation");
467 608
468 if (SvPOK (fh_or_path)) 609 if (SvPOK (fh_or_path))
469 { 610 {
470 req->type = ix ? REQ_LSTAT : REQ_STAT; 611 req->type = ix;
471 req->data = newSVsv (fh_or_path); 612 req->data = newSVsv (fh_or_path);
472 req->dataptr = SvPV_nolen (req->data); 613 req->dataptr = SvPV_nolen (req->data);
473 } 614 }
474 else 615 else
475 { 616 {
476 req->type = REQ_FSTAT; 617 req->type = REQ_FSTAT;
618 req->fh = newSVsv (fh_or_path);
477 req->fd = PerlIO_fileno (IoIFP (sv_2io (fh_or_path))); 619 req->fd = PerlIO_fileno (IoIFP (sv_2io (fh_or_path)));
478 } 620 }
479 621
480 req->callback = SvREFCNT_inc (callback); 622 req->callback = SvREFCNT_inc (callback);
481 623
482 send_req (req); 624 send_req (req);
483} 625}
484 626
485void 627void
486aio_unlink(pathname,callback) 628aio_unlink(pathname,callback=&PL_sv_undef)
487 SV * pathname 629 SV * pathname
488 SV * callback 630 SV * callback
489 PROTOTYPE: $$
490 CODE: 631 CODE:
491{ 632{
492 aio_req req; 633 aio_req req;
493 634
494 Newz (0, req, 1, aio_cb); 635 Newz (0, req, 1, aio_cb);
502 req->callback = SvREFCNT_inc (callback); 643 req->callback = SvREFCNT_inc (callback);
503 644
504 send_req (req); 645 send_req (req);
505} 646}
506 647
648void
649flush()
650 PROTOTYPE:
651 CODE:
652 while (nreqs)
653 {
654 poll_wait ();
655 poll_cb ();
656 }
657
658void
659poll()
660 PROTOTYPE:
661 CODE:
662 if (nreqs)
663 {
664 poll_wait ();
665 poll_cb ();
666 }
667
507int 668int
508poll_fileno() 669poll_fileno()
509 PROTOTYPE: 670 PROTOTYPE:
510 CODE: 671 CODE:
511 RETVAL = respipe[0]; 672 RETVAL = respipe [0];
512 OUTPUT: 673 OUTPUT:
513 RETVAL 674 RETVAL
514 675
515int 676int
516poll_cb(...) 677poll_cb(...)
517 PROTOTYPE: 678 PROTOTYPE:
518 CODE: 679 CODE:
519 RETVAL = poll_cb (aTHX); 680 RETVAL = poll_cb ();
520 OUTPUT: 681 OUTPUT:
521 RETVAL 682 RETVAL
522 683
523void 684void
524poll_wait() 685poll_wait()
525 PROTOTYPE: 686 PROTOTYPE:
526 CODE: 687 CODE:
688 if (nreqs)
527 poll_wait (); 689 poll_wait ();
528 690
529int 691int
530nreqs() 692nreqs()
531 PROTOTYPE: 693 PROTOTYPE:
532 CODE: 694 CODE:

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines