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.117 by root, Sat May 10 20:02:09 2008 UTC vs.
Revision 1.157 by root, Sat Jan 2 12:58:37 2010 UTC

3#include <errno.h> 3#include <errno.h>
4 4
5#include "EXTERN.h" 5#include "EXTERN.h"
6#include "perl.h" 6#include "perl.h"
7#include "XSUB.h" 7#include "XSUB.h"
8
9#include "schmorp.h"
8 10
9#include <stddef.h> 11#include <stddef.h>
10#include <stdlib.h> 12#include <stdlib.h>
11#include <errno.h> 13#include <errno.h>
12#include <sys/types.h> 14#include <sys/types.h>
13#include <sys/stat.h> 15#include <sys/stat.h>
14#include <limits.h> 16#include <limits.h>
15#include <fcntl.h> 17#include <fcntl.h>
16#include <sched.h> 18#include <sched.h>
19
20/* perl namespace pollution */
21#undef VERSION
17 22
18#ifdef _WIN32 23#ifdef _WIN32
19 24
20# define EIO_STRUCT_DIRENT Direntry_t 25# define EIO_STRUCT_DIRENT Direntry_t
21# undef malloc 26# undef malloc
62# include <signal.h> 67# include <signal.h>
63# define EIO_STRUCT_DIRENT struct dirent 68# define EIO_STRUCT_DIRENT struct dirent
64 69
65#endif 70#endif
66 71
72/* perl stupidly overrides readdir and maybe others */
73/* with thread-unsafe versions, imagine that :( */
74#undef readdir
75#undef opendir
76#undef closedir
77
67#define EIO_STRUCT_STAT Stat_t 78#define EIO_STRUCT_STAT Stat_t
68 79
69/* use NV for 32 bit perls as it allows larger offsets */ 80/* use NV for 32 bit perls as it allows larger offsets */
70#if IVSIZE >= 8 81#if IVSIZE >= 8
82# define VAL64 IV
71# define SvVAL64 SvIV 83# define SvVAL64 SvIV
84# define newSVval64 newSViv
72#else 85#else
86# define VAL64 NV
73# define SvVAL64 SvNV 87# define SvVAL64 SvNV
88# define newSVval64 newSVnv
74#endif 89#endif
75 90
76static HV *stash; 91/*****************************************************************************/
92
93#if __GNUC__ >= 3
94# define expect(expr,value) __builtin_expect ((expr),(value))
95#else
96# define expect(expr,value) (expr)
97#endif
98
99#define expect_false(expr) expect ((expr) != 0, 0)
100#define expect_true(expr) expect ((expr) != 0, 1)
101
102/*****************************************************************************/
103
77typedef SV SV8; /* byte-sv, used for argument-checking */ 104typedef SV SV8; /* byte-sv, used for argument-checking */
105typedef int aio_rfd; /* read file desriptor */
106typedef int aio_wfd; /* write file descriptor */
78 107
79#define AIO_REQ_KLASS "IO::AIO::REQ" 108static HV *aio_stash, *aio_req_stash, *aio_grp_stash;
80#define AIO_GRP_KLASS "IO::AIO::GRP"
81 109
82#define EIO_COMMON \ 110#define EIO_REQ_MEMBERS \
83 SV *callback; \ 111 SV *callback; \
84 SV *sv1, *sv2; \ 112 SV *sv1, *sv2; \
85 STRLEN stroffset; \ 113 STRLEN stroffset; \
86 SV *self 114 SV *self;
115
116#define EIO_NO_WRAPPERS 1
87 117
88#include "libeio/eio.h" 118#include "libeio/eio.h"
119
120#ifndef POSIX_FADV_NORMAL
121# define POSIX_FADV_NORMAL 0
122# define NO_FADVISE 1
123#endif
124
125#ifndef POSIX_FADV_SEQUENTIAL
126# define POSIX_FADV_SEQUENTIAL 0
127#endif
128
129#ifndef POSIX_FADV_RANDOM
130# define POSIX_FADV_RANDOM 0
131#endif
132
133#ifndef POSIX_FADV_NOREUSE
134# define POSIX_FADV_NOREUSE 0
135#endif
136
137#ifndef POSIX_FADV_WILLNEED
138# define POSIX_FADV_WILLNEED 0
139#endif
140
141#ifndef POSIX_FADV_DONTNEED
142# define POSIX_FADV_DONTNEED 0
143#endif
89 144
90static int req_invoke (eio_req *req); 145static int req_invoke (eio_req *req);
91#define EIO_FINISH(req) req_invoke (req) 146#define EIO_FINISH(req) req_invoke (req)
92static void aio_grp_feed (eio_req *grp);
93#define EIO_FEED(req) aio_grp_feed (req)
94static void req_destroy (eio_req *grp); 147static void req_destroy (eio_req *grp);
95#define EIO_DESTROY(req) req_destroy (req) 148#define EIO_DESTROY(req) req_destroy (req)
96 149
97enum { 150enum {
98 FLAG_SV2_RO_OFF = 0x40, /* data was set readonly */ 151 FLAG_SV2_RO_OFF = 0x40, /* data was set readonly */
102 155
103typedef eio_req *aio_req; 156typedef eio_req *aio_req;
104typedef eio_req *aio_req_ornot; 157typedef eio_req *aio_req_ornot;
105 158
106static SV *on_next_submit; 159static SV *on_next_submit;
107static int next_pri = EIO_DEFAULT_PRI + EIO_PRI_BIAS; 160static int next_pri = EIO_PRI_DEFAULT;
108static int max_outstanding; 161static int max_outstanding;
109 162
110static int respipe_osf [2], respipe [2] = { -1, -1 }; 163static s_epipe respipe;
111 164
112static void req_destroy (aio_req req); 165static void req_destroy (aio_req req);
113static void req_cancel (aio_req req); 166static void req_cancel (aio_req req);
114 167
115static void want_poll (void) 168static void want_poll (void)
116{ 169{
117 /* write a dummy byte to the pipe so fh becomes ready */ 170 /* write a dummy byte to the pipe so fh becomes ready */
118 respipe_write (respipe_osf [1], (const void *)&respipe_osf, 1); 171 s_epipe_signal (&respipe);
119} 172}
120 173
121static void done_poll (void) 174static void done_poll (void)
122{ 175{
123 /* read any signals sent by the worker threads */ 176 /* read any signals sent by the worker threads */
124 char buf [4]; 177 s_epipe_drain (&respipe);
125 while (respipe_read (respipe [0], buf, 4) == 4)
126 ;
127} 178}
128 179
129/* must be called at most once */ 180/* must be called at most once */
130static SV *req_sv (aio_req req, const char *klass) 181static SV *req_sv (aio_req req, HV *stash)
131{ 182{
132 if (!req->self) 183 if (!req->self)
133 { 184 {
134 req->self = (SV *)newHV (); 185 req->self = (SV *)newHV ();
135 sv_magic (req->self, 0, PERL_MAGIC_ext, (char *)req, 0); 186 sv_magic (req->self, 0, PERL_MAGIC_ext, (char *)req, 0);
136 } 187 }
137 188
138 return sv_2mortal (sv_bless (newRV_inc (req->self), gv_stashpv (klass, 1))); 189 return sv_2mortal (sv_bless (newRV_inc (req->self), stash));
139} 190}
140 191
141static aio_req SvAIO_REQ (SV *sv) 192static aio_req SvAIO_REQ (SV *sv)
142{ 193{
143 MAGIC *mg; 194 MAGIC *mg;
144 195
145 if (!sv_derived_from (sv, AIO_REQ_KLASS) || !SvROK (sv)) 196 if (!SvROK (sv)
197 || (SvSTASH (SvRV (sv)) != aio_grp_stash
198 && SvSTASH (SvRV (sv)) != aio_req_stash
199 && !sv_derived_from (sv, "IO::AIO::REQ")))
146 croak ("object of class " AIO_REQ_KLASS " expected"); 200 croak ("object of class IO::AIO::REQ expected");
147 201
148 mg = mg_find (SvRV (sv), PERL_MAGIC_ext); 202 mg = mg_find (SvRV (sv), PERL_MAGIC_ext);
149 203
150 return mg ? (aio_req)mg->mg_ptr : 0; 204 return mg ? (aio_req)mg->mg_ptr : 0;
151} 205}
157 dSP; 211 dSP;
158 212
159 ENTER; 213 ENTER;
160 SAVETMPS; 214 SAVETMPS;
161 PUSHMARK (SP); 215 PUSHMARK (SP);
162 XPUSHs (req_sv (grp, AIO_GRP_KLASS)); 216 XPUSHs (req_sv (grp, aio_grp_stash));
163 PUTBACK; 217 PUTBACK;
164 call_sv (grp->sv2, G_VOID | G_EVAL | G_KEEPERR); 218 call_sv (grp->sv2, G_VOID | G_EVAL | G_KEEPERR);
165 SPAGAIN; 219 SPAGAIN;
166 FREETMPS; 220 FREETMPS;
167 LEAVE; 221 LEAVE;
170 224
171static void req_submit (eio_req *req) 225static void req_submit (eio_req *req)
172{ 226{
173 eio_submit (req); 227 eio_submit (req);
174 228
175 if (on_next_submit) 229 if (expect_false (on_next_submit))
176 { 230 {
177 dSP; 231 dSP;
178 SV *cb = sv_2mortal (on_next_submit); 232 SV *cb = sv_2mortal (on_next_submit);
179 233
180 on_next_submit = 0; 234 on_next_submit = 0;
190 dSP; 244 dSP;
191 245
192 if (req->flags & FLAG_SV2_RO_OFF) 246 if (req->flags & FLAG_SV2_RO_OFF)
193 SvREADONLY_off (req->sv2); 247 SvREADONLY_off (req->sv2);
194 248
195 if (!EIO_CANCELLED (req) && SvOK (req->callback)) 249 if (!EIO_CANCELLED (req) && req->callback)
196 { 250 {
197 ENTER; 251 ENTER;
198 SAVETMPS; 252 SAVETMPS;
199 PUSHMARK (SP); 253 PUSHMARK (SP);
200 EXTEND (SP, 1); 254 EXTEND (SP, 1);
206 SV *rv = &PL_sv_undef; 260 SV *rv = &PL_sv_undef;
207 261
208 if (req->result >= 0) 262 if (req->result >= 0)
209 { 263 {
210 int i; 264 int i;
211 char *buf = req->ptr2; 265 char *names = (char *)req->ptr2;
266 eio_dirent *ent = (eio_dirent *)req->ptr1; /* might be 0 */
212 AV *av = newAV (); 267 AV *av = newAV ();
213 268
214 av_extend (av, req->result - 1); 269 av_extend (av, req->result - 1);
215 270
216 for (i = 0; i < req->result; ++i) 271 for (i = 0; i < req->result; ++i)
217 { 272 {
218 SV *sv = newSVpv (buf, 0); 273 if (req->int1 & EIO_READDIR_DENTS)
274 {
275 SV *namesv = newSVpvn (names + ent->nameofs, ent->namelen);
219 276
277 if (req->int1 & EIO_READDIR_CUSTOM2)
278 {
279 static SV *sv_type [EIO_DT_MAX + 1]; /* type sv cache */
280 AV *avent = newAV ();
281
282 av_extend (avent, 2);
283
284 if (!sv_type [ent->type])
285 {
286 sv_type [ent->type] = newSViv (ent->type);
287 SvREADONLY_on (sv_type [ent->type]);
288 }
289
290 av_store (avent, 0, namesv);
291 av_store (avent, 1, SvREFCNT_inc (sv_type [ent->type]));
292 av_store (avent, 2, IVSIZE >= 8 ? newSVuv (ent->inode) : newSVnv (ent->inode));
293
294 av_store (av, i, newRV_noinc ((SV *)avent));
295 }
296 else
220 av_store (av, i, sv); 297 av_store (av, i, namesv);
298
299 ++ent;
300 }
301 else
302 {
303 SV *name = newSVpv (names, 0);
304 av_store (av, i, name);
221 buf += SvCUR (sv) + 1; 305 names += SvCUR (name) + 1;
306 }
222 } 307 }
223 308
224 rv = sv_2mortal (newRV_noinc ((SV *)av)); 309 rv = sv_2mortal (newRV_noinc ((SV *)av));
225 } 310 }
226 311
227 PUSHs (rv); 312 PUSHs (rv);
313
314 if (req->int1 & EIO_READDIR_CUSTOM1)
315 XPUSHs (sv_2mortal (newSViv (req->int1 & ~(EIO_READDIR_CUSTOM1 | EIO_READDIR_CUSTOM2))));
228 } 316 }
229 break; 317 break;
230 318
231 case EIO_OPEN: 319 case EIO_OPEN:
232 { 320 {
238 GV *gv = (GV *)sv_newmortal (); 326 GV *gv = (GV *)sv_newmortal ();
239 int flags = req->int1 & (O_RDONLY | O_WRONLY | O_RDWR); 327 int flags = req->int1 & (O_RDONLY | O_WRONLY | O_RDWR);
240 char sym [64]; 328 char sym [64];
241 int symlen; 329 int symlen;
242 330
243 symlen = snprintf (sym, sizeof (sym), "fd#%d", req->result); 331 symlen = snprintf (sym, sizeof (sym), "fd#%d", (int)req->result);
244 gv_init (gv, stash, sym, symlen, 0); 332 gv_init (gv, aio_stash, sym, symlen, 0);
245 333
246 symlen = snprintf ( 334 symlen = snprintf (
247 sym, 335 sym,
248 sizeof (sym), 336 sizeof (sym),
249 "%s&=%d", 337 "%s&=%d",
250 flags == O_RDONLY ? "<" : flags == O_WRONLY ? ">" : "+<", 338 flags == O_RDONLY ? "<" : flags == O_WRONLY ? ">" : "+<",
251 req->result 339 (int)req->result
252 ); 340 );
253 341
254 if (do_open (gv, sym, symlen, 0, 0, 0, 0)) 342 if (do_open (gv, sym, symlen, 0, 0, 0, 0))
255 fh = (SV *)gv; 343 fh = (SV *)gv;
256 } 344 }
277 case EIO_BUSY: 365 case EIO_BUSY:
278 break; 366 break;
279 367
280 case EIO_READLINK: 368 case EIO_READLINK:
281 if (req->result > 0) 369 if (req->result > 0)
282 { 370 PUSHs (sv_2mortal (newSVpvn (req->ptr2, req->result)));
283 SvCUR_set (req->sv2, req->result);
284 *SvEND (req->sv2) = 0;
285 PUSHs (req->sv2);
286 }
287 break; 371 break;
288 372
289 case EIO_STAT: 373 case EIO_STAT:
290 case EIO_LSTAT: 374 case EIO_LSTAT:
291 case EIO_FSTAT: 375 case EIO_FSTAT:
294 PL_statcache = *(EIO_STRUCT_STAT *)(req->ptr2); 378 PL_statcache = *(EIO_STRUCT_STAT *)(req->ptr2);
295 PUSHs (sv_2mortal (newSViv (req->result))); 379 PUSHs (sv_2mortal (newSViv (req->result)));
296 break; 380 break;
297 381
298 case EIO_READ: 382 case EIO_READ:
383 {
299 SvCUR_set (req->sv2, req->stroffset + (req->result > 0 ? req->result : 0)); 384 SvCUR_set (req->sv2, req->stroffset + (req->result > 0 ? req->result : 0));
300 *SvEND (req->sv2) = 0; 385 *SvEND (req->sv2) = 0;
386 SvPOK_only (req->sv2);
387 SvSETMAGIC (req->sv2);
301 PUSHs (sv_2mortal (newSViv (req->result))); 388 PUSHs (sv_2mortal (newSViv (req->result)));
389 }
302 break; 390 break;
303 391
304 case EIO_DUP2: 392 case EIO_DUP2:
305 if (req->result > 0) 393 if (req->result > 0)
306 req->result = 0; 394 req->result = 0;
352 grp->sv2 = 0; 440 grp->sv2 = 0;
353 441
354 eio_grp_cancel (grp); 442 eio_grp_cancel (grp);
355} 443}
356 444
357#ifdef USE_SOCKETS_AS_HANDLES
358# define TO_SOCKET(x) (win32_get_osfhandle (x))
359#else
360# define TO_SOCKET(x) (x)
361#endif
362
363static void 445static void
364create_respipe (void) 446create_respipe (void)
365{ 447{
366 int old_readfd = respipe [0];
367
368 if (respipe [1] >= 0)
369 respipe_close (TO_SOCKET (respipe [1]));
370
371#ifdef _WIN32
372 if (PerlSock_socketpair (AF_UNIX, SOCK_STREAM, 0, respipe))
373#else
374 if (pipe (respipe)) 448 if (s_epipe_renew (&respipe))
375#endif
376 croak ("unable to initialize result pipe"); 449 croak ("IO::AIO: unable to initialize result pipe");
377
378 if (old_readfd >= 0)
379 {
380 if (dup2 (TO_SOCKET (respipe [0]), TO_SOCKET (old_readfd)) < 0)
381 croak ("unable to initialize result pipe(2)");
382
383 respipe_close (respipe [0]);
384 respipe [0] = old_readfd;
385 }
386
387#ifdef _WIN32
388 int arg = 1;
389 if (ioctlsocket (TO_SOCKET (respipe [0]), FIONBIO, &arg)
390 || ioctlsocket (TO_SOCKET (respipe [1]), FIONBIO, &arg))
391#else
392 if (fcntl (respipe [0], F_SETFL, O_NONBLOCK)
393 || fcntl (respipe [1], F_SETFL, O_NONBLOCK))
394#endif
395 croak ("unable to initialize result pipe(3)");
396
397 respipe_osf [0] = TO_SOCKET (respipe [0]);
398 respipe_osf [1] = TO_SOCKET (respipe [1]);
399} 450}
400 451
401static void poll_wait (void) 452static void poll_wait (void)
402{ 453{
403 fd_set rfd;
404
405 while (eio_nreqs ()) 454 while (eio_nreqs ())
406 { 455 {
407 int size; 456 int size;
408 457
409 X_LOCK (reslock); 458 X_LOCK (reslock);
411 X_UNLOCK (reslock); 460 X_UNLOCK (reslock);
412 461
413 if (size) 462 if (size)
414 return; 463 return;
415 464
416 maybe_start_thread (); 465 etp_maybe_start_thread ();
417 466
418 FD_ZERO (&rfd); 467 s_epipe_wait (&respipe);
419 FD_SET (respipe [0], &rfd);
420
421 PerlSock_select (respipe [0] + 1, &rfd, 0, 0, 0);
422 } 468 }
423} 469}
424 470
425static int poll_cb (void) 471static int poll_cb (void)
426{ 472{
427 int res; 473 for (;;)
428
429 do
430 { 474 {
431 res = eio_poll (); 475 int res = eio_poll ();
432 476
433 if (res > 0) 477 if (res > 0)
434 croak (0); 478 croak (0);
479
480 if (!max_outstanding || max_outstanding > eio_nreqs ())
481 return res;
482
483 poll_wait ();
435 } 484 }
436 while (max_outstanding && max_outstanding <= eio_nreqs ());
437
438 return res;
439} 485}
440 486
441static void atfork_child (void) 487static void atfork_child (void)
442{ 488{
443 create_respipe (); 489 create_respipe ();
444} 490}
445 491
492static SV *
493get_cb (SV *cb_sv)
494{
495 SvGETMAGIC (cb_sv);
496 return SvOK (cb_sv) ? s_get_cv_croak (cb_sv) : 0;
497}
498
446#define dREQ \ 499#define dREQ \
500 SV *cb_cv; \
447 aio_req req; \ 501 aio_req req; \
448 int req_pri = next_pri; \ 502 int req_pri = next_pri; \
449 next_pri = EIO_DEFAULT_PRI + EIO_PRI_BIAS; \ 503 next_pri = EIO_PRI_DEFAULT; \
450 \ 504 \
451 if (SvOK (callback) && !SvROK (callback)) \ 505 cb_cv = get_cb (callback); \
452 croak ("callback must be undef or of reference type"); \
453 \ 506 \
454 Newz (0, req, 1, eio_req); \ 507 Newz (0, req, 1, eio_req); \
455 if (!req) \ 508 if (!req) \
456 croak ("out of memory during eio_req allocation"); \ 509 croak ("out of memory during eio_req allocation"); \
457 \ 510 \
458 req->callback = newSVsv (callback); \ 511 req->callback = SvREFCNT_inc (cb_cv); \
459 req->pri = req_pri 512 req->pri = req_pri
460 513
461#define REQ_SEND \ 514#define REQ_SEND \
515 PUTBACK; \
462 req_submit (req); \ 516 req_submit (req); \
517 SPAGAIN; \
463 \ 518 \
464 if (GIMME_V != G_VOID) \ 519 if (GIMME_V != G_VOID) \
465 XPUSHs (req_sv (req, AIO_REQ_KLASS)); 520 XPUSHs (req_sv (req, aio_req_stash));
466 521
467MODULE = IO::AIO PACKAGE = IO::AIO 522MODULE = IO::AIO PACKAGE = IO::AIO
468 523
469PROTOTYPES: ENABLE 524PROTOTYPES: ENABLE
470 525
471BOOT: 526BOOT:
472{ 527{
473 stash = gv_stashpv ("IO::AIO", 1); 528 static const struct {
474 529 const char *name;
475 newCONSTSUB (stash, "EXDEV", newSViv (EXDEV)); 530 IV iv;
476 newCONSTSUB (stash, "O_RDONLY", newSViv (O_RDONLY)); 531 } *civ, const_iv[] = {
477 newCONSTSUB (stash, "O_WRONLY", newSViv (O_WRONLY)); 532# define const_iv(name, value) { # name, (IV) value },
478 newCONSTSUB (stash, "O_CREAT", newSViv (O_CREAT)); 533# define const_eio(name) { # name, (IV) EIO_ ## name },
479 newCONSTSUB (stash, "O_TRUNC", newSViv (O_TRUNC)); 534 const_iv (EXDEV , EXDEV)
535 const_iv (ENOSYS , ENOSYS)
536 const_iv (O_RDONLY, O_RDONLY)
537 const_iv (O_WRONLY, O_WRONLY)
538 const_iv (O_CREAT , O_CREAT)
539 const_iv (O_TRUNC , O_TRUNC)
480#ifdef _WIN32 540#ifndef _WIN32
481 X_MUTEX_CHECK (wrklock); 541 const_iv (S_IFIFO , S_IFIFO)
482 X_MUTEX_CHECK (reslock);
483 X_MUTEX_CHECK (reqlock);
484 X_MUTEX_CHECK (reqwait);
485 X_MUTEX_CHECK (preadwritelock);
486 X_MUTEX_CHECK (readdirlock);
487
488 X_COND_CHECK (reqwait);
489#else
490 newCONSTSUB (stash, "S_IFIFO", newSViv (S_IFIFO));
491#endif 542#endif
543 const_iv (FADV_NORMAL , POSIX_FADV_NORMAL)
544 const_iv (FADV_SEQUENTIAL, POSIX_FADV_SEQUENTIAL)
545 const_iv (FADV_RANDOM , POSIX_FADV_RANDOM)
546 const_iv (FADV_NOREUSE , POSIX_FADV_NOREUSE)
547 const_iv (FADV_WILLNEED , POSIX_FADV_WILLNEED)
548 const_iv (FADV_DONTNEED , POSIX_FADV_DONTNEED)
492 549
550 const_eio (SYNC_FILE_RANGE_WAIT_BEFORE)
551 const_eio (SYNC_FILE_RANGE_WRITE)
552 const_eio (SYNC_FILE_RANGE_WAIT_AFTER)
553
554 const_eio (READDIR_DENTS)
555 const_eio (READDIR_DIRS_FIRST)
556 const_eio (READDIR_STAT_ORDER)
557 const_eio (READDIR_FOUND_UNKNOWN)
558
559 const_eio (DT_UNKNOWN)
560 const_eio (DT_FIFO)
561 const_eio (DT_CHR)
562 const_eio (DT_DIR)
563 const_eio (DT_BLK)
564 const_eio (DT_REG)
565 const_eio (DT_LNK)
566 const_eio (DT_SOCK)
567 const_eio (DT_WHT)
568 };
569
570 aio_stash = gv_stashpv ("IO::AIO" , 1);
571 aio_req_stash = gv_stashpv ("IO::AIO::REQ", 1);
572 aio_grp_stash = gv_stashpv ("IO::AIO::GRP", 1);
573
574 for (civ = const_iv + sizeof (const_iv) / sizeof (const_iv [0]); civ-- > const_iv; )
575 newCONSTSUB (aio_stash, (char *)civ->name, newSViv (civ->iv));
576
493 create_respipe (); 577 create_respipe ();
494 578
495 if (eio_init (want_poll, done_poll) < 0) 579 if (eio_init (want_poll, done_poll) < 0)
496 croak ("IO::AIO: unable to initialise eio library"); 580 croak ("IO::AIO: unable to initialise eio library");
497 581
498 /* atfork child called in fifo order, so before eio's handler */ 582 /* atfork child called in fifo order, so before eio's handler */
499 X_THREAD_ATFORK (0, 0, atfork_child); 583 X_THREAD_ATFORK (0, 0, atfork_child);
500} 584}
501 585
502void 586void
503max_poll_reqs (int nreqs) 587max_poll_reqs (int nreqs)
504 PROTOTYPE: $ 588 PROTOTYPE: $
557 ALIAS: 641 ALIAS:
558 aio_fsync = EIO_FSYNC 642 aio_fsync = EIO_FSYNC
559 aio_fdatasync = EIO_FDATASYNC 643 aio_fdatasync = EIO_FDATASYNC
560 PPCODE: 644 PPCODE:
561{ 645{
646 int fd = s_fileno_croak (fh, 0);
562 dREQ; 647 dREQ;
563 648
564 req->type = ix; 649 req->type = ix;
565 req->sv1 = newSVsv (fh); 650 req->sv1 = newSVsv (fh);
566 req->int1 = PerlIO_fileno (IoIFP (sv_2io (fh))); 651 req->int1 = fd;
652
653 REQ_SEND (req);
654}
655
656void
657aio_sync_file_range (SV *fh, off_t offset, size_t nbytes, UV flags, SV *callback=&PL_sv_undef)
658 PROTOTYPE: $$$$;$
659 PPCODE:
660{
661 int fd = s_fileno_croak (fh, 0);
662 dREQ;
663
664 req->type = EIO_SYNC_FILE_RANGE;
665 req->sv1 = newSVsv (fh);
666 req->int1 = fd;
667 req->offs = offset;
668 req->size = nbytes;
669 req->int2 = flags;
567 670
568 REQ_SEND (req); 671 REQ_SEND (req);
569} 672}
570 673
571void 674void
572aio_close (SV *fh, SV *callback=&PL_sv_undef) 675aio_close (SV *fh, SV *callback=&PL_sv_undef)
573 PROTOTYPE: $;$ 676 PROTOTYPE: $;$
574 PPCODE: 677 PPCODE:
575{ 678{
576 static int close_pipe = -1; /* dummy fd to close fds via dup2 */ 679 static int close_pipe = -1; /* dummy fd to close fds via dup2 */
680 int fd = s_fileno_croak (fh, 0);
577 dREQ; 681 dREQ;
578 682
579 if (close_pipe < 0) 683 if (close_pipe < 0)
580 { 684 {
581 int pipefd [2]; 685 int pipefd [2];
589 } 693 }
590 694
591 req->type = EIO_DUP2; 695 req->type = EIO_DUP2;
592 req->int1 = close_pipe; 696 req->int1 = close_pipe;
593 req->sv2 = newSVsv (fh); 697 req->sv2 = newSVsv (fh);
594 req->int2 = PerlIO_fileno (IoIFP (sv_2io (fh))); 698 req->int2 = fd;
595 699
596 REQ_SEND (req); 700 REQ_SEND (req);
597} 701}
598 702
599void 703void
603 aio_write = EIO_WRITE 707 aio_write = EIO_WRITE
604 PROTOTYPE: $$$$$;$ 708 PROTOTYPE: $$$$$;$
605 PPCODE: 709 PPCODE:
606{ 710{
607 STRLEN svlen; 711 STRLEN svlen;
712 int fd = s_fileno_croak (fh, ix == EIO_WRITE);
608 char *svptr = SvPVbyte (data, svlen); 713 char *svptr = SvPVbyte (data, svlen);
609 UV len = SvUV (length); 714 UV len = SvUV (length);
610
611 SvUPGRADE (data, SVt_PV);
612 SvPOK_on (data);
613 715
614 if (dataoffset < 0) 716 if (dataoffset < 0)
615 dataoffset += svlen; 717 dataoffset += svlen;
616 718
617 if (dataoffset < 0 || dataoffset > svlen) 719 if (dataoffset < 0 || dataoffset > svlen)
623 if (!SvOK (length) || len + dataoffset > svlen) 725 if (!SvOK (length) || len + dataoffset > svlen)
624 len = svlen - dataoffset; 726 len = svlen - dataoffset;
625 } 727 }
626 else 728 else
627 { 729 {
628 /* read: grow scalar as necessary */ 730 /* read: check type and grow scalar as necessary */
731 SvUPGRADE (data, SVt_PV);
629 svptr = SvGROW (data, len + dataoffset + 1); 732 svptr = SvGROW (data, len + dataoffset + 1);
630 } 733 }
631
632 if (len < 0)
633 croak ("length must not be negative");
634 734
635 { 735 {
636 dREQ; 736 dREQ;
637 737
638 req->type = ix; 738 req->type = ix;
639 req->sv1 = newSVsv (fh); 739 req->sv1 = newSVsv (fh);
640 req->int1 = PerlIO_fileno (ix == EIO_READ ? IoIFP (sv_2io (fh)) 740 req->int1 = fd;
641 : IoOFP (sv_2io (fh)));
642 req->offs = SvOK (offset) ? SvVAL64 (offset) : -1; 741 req->offs = SvOK (offset) ? SvVAL64 (offset) : -1;
643 req->size = len; 742 req->size = len;
644 req->sv2 = SvREFCNT_inc (data); 743 req->sv2 = SvREFCNT_inc (data);
645 req->ptr2 = (char *)svptr + dataoffset; 744 req->ptr2 = (char *)svptr + dataoffset;
646 req->stroffset = dataoffset; 745 req->stroffset = dataoffset;
661 PPCODE: 760 PPCODE:
662{ 761{
663 SV *data; 762 SV *data;
664 dREQ; 763 dREQ;
665 764
666 data = newSV (NAME_MAX);
667 SvPOK_on (data);
668
669 req->type = EIO_READLINK; 765 req->type = EIO_READLINK;
670 req->sv1 = newSVsv (path); 766 req->sv1 = newSVsv (path);
671 req->ptr1 = SvPVbyte_nolen (req->sv1); 767 req->ptr1 = SvPVbyte_nolen (req->sv1);
672 req->sv2 = data;
673 req->ptr2 = SvPVbyte_nolen (data);
674 768
675 REQ_SEND; 769 REQ_SEND;
676} 770}
677 771
678void 772void
679aio_sendfile (SV *out_fh, SV *in_fh, SV *in_offset, UV length, SV *callback=&PL_sv_undef) 773aio_sendfile (SV *out_fh, SV *in_fh, off_t in_offset, size_t length, SV *callback=&PL_sv_undef)
680 PROTOTYPE: $$$$;$ 774 PROTOTYPE: $$$$;$
681 PPCODE: 775 PPCODE:
682{ 776{
777 int ifd = s_fileno_croak (in_fh , 0);
778 int ofd = s_fileno_croak (out_fh, 1);
683 dREQ; 779 dREQ;
684 780
685 req->type = EIO_SENDFILE; 781 req->type = EIO_SENDFILE;
686 req->sv1 = newSVsv (out_fh); 782 req->sv1 = newSVsv (out_fh);
687 req->int1 = PerlIO_fileno (IoIFP (sv_2io (out_fh))); 783 req->int1 = ofd;
688 req->sv2 = newSVsv (in_fh); 784 req->sv2 = newSVsv (in_fh);
689 req->int2 = PerlIO_fileno (IoIFP (sv_2io (in_fh))); 785 req->int2 = ifd;
690 req->offs = SvVAL64 (in_offset); 786 req->offs = in_offset;
691 req->size = length; 787 req->size = length;
692 788
693 REQ_SEND; 789 REQ_SEND;
694} 790}
695 791
696void 792void
697aio_readahead (SV *fh, SV *offset, IV length, SV *callback=&PL_sv_undef) 793aio_readahead (SV *fh, off_t offset, size_t length, SV *callback=&PL_sv_undef)
698 PROTOTYPE: $$$;$ 794 PROTOTYPE: $$$;$
699 PPCODE: 795 PPCODE:
700{ 796{
797 int fd = s_fileno_croak (fh, 0);
701 dREQ; 798 dREQ;
702 799
703 req->type = EIO_READAHEAD; 800 req->type = EIO_READAHEAD;
704 req->sv1 = newSVsv (fh); 801 req->sv1 = newSVsv (fh);
705 req->int1 = PerlIO_fileno (IoIFP (sv_2io (fh))); 802 req->int1 = fd;
706 req->offs = SvVAL64 (offset); 803 req->offs = offset;
707 req->size = length; 804 req->size = length;
708 805
709 REQ_SEND; 806 REQ_SEND;
710} 807}
711 808
716 aio_lstat = EIO_LSTAT 813 aio_lstat = EIO_LSTAT
717 PPCODE: 814 PPCODE:
718{ 815{
719 dREQ; 816 dREQ;
720 817
721 req->ptr2 = malloc (sizeof (EIO_STRUCT_STAT));
722 if (!req->ptr2)
723 {
724 req_destroy (req);
725 croak ("out of memory during aio_stat statdata allocation");
726 }
727
728 req->flags |= EIO_FLAG_PTR2_FREE;
729 req->sv1 = newSVsv (fh_or_path); 818 req->sv1 = newSVsv (fh_or_path);
730 819
731 if (SvPOK (fh_or_path)) 820 if (SvPOK (req->sv1))
732 { 821 {
733 req->type = ix; 822 req->type = ix;
734 req->ptr1 = SvPVbyte_nolen (req->sv1); 823 req->ptr1 = SvPVbyte_nolen (req->sv1);
735 } 824 }
736 else 825 else
750 839
751 req->nv1 = SvOK (atime) ? SvNV (atime) : -1.; 840 req->nv1 = SvOK (atime) ? SvNV (atime) : -1.;
752 req->nv2 = SvOK (mtime) ? SvNV (mtime) : -1.; 841 req->nv2 = SvOK (mtime) ? SvNV (mtime) : -1.;
753 req->sv1 = newSVsv (fh_or_path); 842 req->sv1 = newSVsv (fh_or_path);
754 843
755 if (SvPOK (fh_or_path)) 844 if (SvPOK (req->sv1))
756 { 845 {
757 req->type = EIO_UTIME; 846 req->type = EIO_UTIME;
758 req->ptr1 = SvPVbyte_nolen (req->sv1); 847 req->ptr1 = SvPVbyte_nolen (req->sv1);
759 } 848 }
760 else 849 else
773 dREQ; 862 dREQ;
774 863
775 req->sv1 = newSVsv (fh_or_path); 864 req->sv1 = newSVsv (fh_or_path);
776 req->offs = SvOK (offset) ? SvVAL64 (offset) : -1; 865 req->offs = SvOK (offset) ? SvVAL64 (offset) : -1;
777 866
778 if (SvPOK (fh_or_path)) 867 if (SvPOK (req->sv1))
779 { 868 {
780 req->type = EIO_TRUNCATE; 869 req->type = EIO_TRUNCATE;
781 req->ptr1 = SvPVbyte_nolen (req->sv1); 870 req->ptr1 = SvPVbyte_nolen (req->sv1);
782 } 871 }
783 else 872 else
791 880
792void 881void
793aio_chmod (SV8 *fh_or_path, int mode, SV *callback=&PL_sv_undef) 882aio_chmod (SV8 *fh_or_path, int mode, SV *callback=&PL_sv_undef)
794 ALIAS: 883 ALIAS:
795 aio_chmod = EIO_CHMOD 884 aio_chmod = EIO_CHMOD
796 aio_fchmod = EIO_FCHMOD
797 aio_mkdir = EIO_MKDIR 885 aio_mkdir = EIO_MKDIR
798 PPCODE: 886 PPCODE:
799{ 887{
800 dREQ; 888 dREQ;
801 889
802 req->type = ix;
803 req->int2 = mode; 890 req->int2 = mode;
804 req->sv1 = newSVsv (fh_or_path); 891 req->sv1 = newSVsv (fh_or_path);
805 892
806 if (ix == EIO_FCHMOD) 893 if (SvPOK (req->sv1))
807 req->int1 = PerlIO_fileno (IoIFP (sv_2io (fh_or_path))); 894 {
895 req->type = ix;
896 req->ptr1 = SvPVbyte_nolen (req->sv1);
897 }
808 else 898 else
809 req->ptr1 = SvPVbyte_nolen (req->sv1); 899 {
900 req->type = EIO_FCHMOD;
901 req->int1 = PerlIO_fileno (IoIFP (sv_2io (fh_or_path)));
902 }
810 903
811 REQ_SEND; 904 REQ_SEND;
812} 905}
813 906
814void 907void
819 912
820 req->int2 = SvOK (uid) ? SvIV (uid) : -1; 913 req->int2 = SvOK (uid) ? SvIV (uid) : -1;
821 req->int3 = SvOK (gid) ? SvIV (gid) : -1; 914 req->int3 = SvOK (gid) ? SvIV (gid) : -1;
822 req->sv1 = newSVsv (fh_or_path); 915 req->sv1 = newSVsv (fh_or_path);
823 916
824 if (SvPOK (fh_or_path)) 917 if (SvPOK (req->sv1))
825 { 918 {
826 req->type = EIO_CHOWN; 919 req->type = EIO_CHOWN;
827 req->ptr1 = SvPVbyte_nolen (req->sv1); 920 req->ptr1 = SvPVbyte_nolen (req->sv1);
828 } 921 }
829 else 922 else
831 req->type = EIO_FCHOWN; 924 req->type = EIO_FCHOWN;
832 req->int1 = PerlIO_fileno (IoIFP (sv_2io (fh_or_path))); 925 req->int1 = PerlIO_fileno (IoIFP (sv_2io (fh_or_path)));
833 } 926 }
834 927
835 REQ_SEND; 928 REQ_SEND;
929}
930
931void
932aio_readdirx (SV8 *pathname, IV flags, SV *callback=&PL_sv_undef)
933 PPCODE:
934{
935 dREQ;
936
937 req->type = EIO_READDIR;
938 req->sv1 = newSVsv (pathname);
939 req->ptr1 = SvPVbyte_nolen (req->sv1);
940 req->int1 = flags | EIO_READDIR_DENTS | EIO_READDIR_CUSTOM1;
941
942 if (flags & EIO_READDIR_DENTS)
943 req->int1 |= EIO_READDIR_CUSTOM2;
944
945 REQ_SEND;
836} 946}
837 947
838void 948void
839aio_unlink (SV8 *pathname, SV *callback=&PL_sv_undef) 949aio_unlink (SV8 *pathname, SV *callback=&PL_sv_undef)
840 ALIAS: 950 ALIAS:
905{ 1015{
906 dREQ; 1016 dREQ;
907 1017
908 req->type = EIO_GROUP; 1018 req->type = EIO_GROUP;
909 1019
910 eio_submit (req); 1020 req_submit (req);
911 XPUSHs (req_sv (req, AIO_GRP_KLASS)); 1021 XPUSHs (req_sv (req, aio_grp_stash));
912} 1022}
913 1023
914void 1024void
915aio_nop (SV *callback=&PL_sv_undef) 1025aio_nop (SV *callback=&PL_sv_undef)
916 ALIAS: 1026 ALIAS:
927 1037
928int 1038int
929aioreq_pri (int pri = 0) 1039aioreq_pri (int pri = 0)
930 PROTOTYPE: ;$ 1040 PROTOTYPE: ;$
931 CODE: 1041 CODE:
932 RETVAL = next_pri - EIO_PRI_BIAS; 1042 RETVAL = next_pri;
933 if (items > 0) 1043 if (items > 0)
934 { 1044 {
935 if (pri < EIO_PRI_MIN) pri = EIO_PRI_MIN; 1045 if (pri < EIO_PRI_MIN) pri = EIO_PRI_MIN;
936 if (pri > EIO_PRI_MAX) pri = EIO_PRI_MAX; 1046 if (pri > EIO_PRI_MAX) pri = EIO_PRI_MAX;
937 next_pri = pri + EIO_PRI_BIAS; 1047 next_pri = pri;
938 } 1048 }
939 OUTPUT: 1049 OUTPUT:
940 RETVAL 1050 RETVAL
941 1051
942void 1052void
943aioreq_nice (int nice = 0) 1053aioreq_nice (int nice = 0)
944 CODE: 1054 CODE:
945 nice = next_pri - nice; 1055 nice = next_pri - nice;
946 if (nice < EIO_PRI_MIN) nice = EIO_PRI_MIN; 1056 if (nice < EIO_PRI_MIN) nice = EIO_PRI_MIN;
947 if (nice > EIO_PRI_MAX) nice = EIO_PRI_MAX; 1057 if (nice > EIO_PRI_MAX) nice = EIO_PRI_MAX;
948 next_pri = nice + EIO_PRI_BIAS; 1058 next_pri = nice;
949 1059
950void 1060void
951flush () 1061flush ()
952 PROTOTYPE: 1062 PROTOTYPE:
953 CODE: 1063 CODE:
968 1078
969int 1079int
970poll_fileno() 1080poll_fileno()
971 PROTOTYPE: 1081 PROTOTYPE:
972 CODE: 1082 CODE:
973 RETVAL = respipe [0]; 1083 RETVAL = s_epipe_fd (&respipe);
974 OUTPUT: 1084 OUTPUT:
975 RETVAL 1085 RETVAL
976 1086
977int 1087int
978poll_cb(...) 1088poll_cb(...)
1014 1124
1015int 1125int
1016nthreads() 1126nthreads()
1017 PROTOTYPE: 1127 PROTOTYPE:
1018 CODE: 1128 CODE:
1019 if (WORDACCESS_UNSAFE) X_LOCK (wrklock); 1129 RETVAL = eio_nthreads ();
1020 RETVAL = started;
1021 if (WORDACCESS_UNSAFE) X_UNLOCK (wrklock);
1022 OUTPUT: 1130 OUTPUT:
1023 RETVAL 1131 RETVAL
1024 1132
1133int
1134fadvise (aio_rfd fh, off_t offset, off_t length, IV advice)
1135 PROTOTYPE: $$$$
1136 CODE:
1137#if _XOPEN_SOURCE >= 600 && !NO_FADVISE
1138 RETVAL = posix_fadvise (fh, offset, length, advice);
1139#else
1140 RETVAL = errno = ENOSYS;
1141#endif
1142 OUTPUT:
1143 RETVAL
1144
1145ssize_t
1146sendfile (aio_wfd ofh, aio_rfd ifh, off_t offset, size_t count)
1147 PROTOTYPE: $$$$
1148 CODE:
1149 RETVAL = eio_sendfile_sync (ofh, ifh, offset, count);
1150 OUTPUT:
1151 RETVAL
1152
1025void _on_next_submit (SV *cb) 1153void _on_next_submit (SV *cb)
1026 CODE: 1154 CODE:
1027 SvREFCNT_dec (on_next_submit); 1155 SvREFCNT_dec (on_next_submit);
1028 on_next_submit = SvOK (cb) ? newSVsv (cb) : 0; 1156 on_next_submit = SvOK (cb) ? newSVsv (cb) : 0;
1029 1157
1036 CODE: 1164 CODE:
1037 eio_cancel (req); 1165 eio_cancel (req);
1038 1166
1039void 1167void
1040cb (aio_req_ornot req, SV *callback=&PL_sv_undef) 1168cb (aio_req_ornot req, SV *callback=&PL_sv_undef)
1041 CODE: 1169 PPCODE:
1170{
1171 if (GIMME_V != G_VOID)
1172 XPUSHs (req->callback ? sv_2mortal (newRV_inc (req->callback)) : &PL_sv_undef);
1173
1174 if (items > 1)
1175 {
1176 SV *cb_cv =get_cb (callback);
1177
1042 SvREFCNT_dec (req->callback); 1178 SvREFCNT_dec (req->callback);
1043 req->callback = newSVsv (callback); 1179 req->callback = SvREFCNT_inc (cb_cv);
1180 }
1181}
1044 1182
1045MODULE = IO::AIO PACKAGE = IO::AIO::GRP 1183MODULE = IO::AIO PACKAGE = IO::AIO::GRP
1046 1184
1047void 1185void
1048add (aio_req grp, ...) 1186add (aio_req grp, ...)
1080 AV *av; 1218 AV *av;
1081 1219
1082 grp->errorno = errno; 1220 grp->errorno = errno;
1083 1221
1084 av = newAV (); 1222 av = newAV ();
1223 av_extend (av, items - 1);
1085 1224
1086 for (i = 1; i < items; ++i ) 1225 for (i = 1; i < items; ++i )
1087 av_push (av, newSVsv (ST (i))); 1226 av_push (av, newSVsv (ST (i)));
1088 1227
1089 SvREFCNT_dec (grp->sv1); 1228 SvREFCNT_dec (grp->sv1);
1103void 1242void
1104feed (aio_req grp, SV *callback=&PL_sv_undef) 1243feed (aio_req grp, SV *callback=&PL_sv_undef)
1105 CODE: 1244 CODE:
1106{ 1245{
1107 SvREFCNT_dec (grp->sv2); 1246 SvREFCNT_dec (grp->sv2);
1108 grp->sv2 = newSVsv (callback); 1247 grp->sv2 = newSVsv (callback);
1248 grp->feed = aio_grp_feed;
1109 1249
1110 if (grp->int2 <= 0) 1250 if (grp->int2 <= 0)
1111 grp->int2 = 2; 1251 grp->int2 = 2;
1112 1252
1113 eio_grp_limit (grp, grp->int2); 1253 eio_grp_limit (grp, grp->int2);

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines