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.118 by root, Sun May 11 00:01:03 2008 UTC vs.
Revision 1.158 by root, Sat Jan 2 14:11:32 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>
65# include <signal.h> 67# include <signal.h>
66# define EIO_STRUCT_DIRENT struct dirent 68# define EIO_STRUCT_DIRENT struct dirent
67 69
68#endif 70#endif
69 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
70#define EIO_STRUCT_STAT Stat_t 78#define EIO_STRUCT_STAT Stat_t
71 79
72/* use NV for 32 bit perls as it allows larger offsets */ 80/* use NV for 32 bit perls as it allows larger offsets */
73#if IVSIZE >= 8 81#if IVSIZE >= 8
82# define VAL64 IV
74# define SvVAL64 SvIV 83# define SvVAL64 SvIV
84# define newSVval64 newSViv
75#else 85#else
86# define VAL64 NV
76# define SvVAL64 SvNV 87# define SvVAL64 SvNV
88# define newSVval64 newSVnv
77#endif 89#endif
78 90
79static 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
80typedef 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 */
81 107
82#define AIO_REQ_KLASS "IO::AIO::REQ" 108static HV *aio_stash, *aio_req_stash, *aio_grp_stash;
83#define AIO_GRP_KLASS "IO::AIO::GRP"
84 109
85#define EIO_COMMON \ 110#define EIO_REQ_MEMBERS \
86 SV *callback; \ 111 SV *callback; \
87 SV *sv1, *sv2; \ 112 SV *sv1, *sv2; \
88 STRLEN stroffset; \ 113 STRLEN stroffset; \
89 SV *self 114 SV *self;
115
116#define EIO_NO_WRAPPERS 1
90 117
91#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
92 144
93static int req_invoke (eio_req *req); 145static int req_invoke (eio_req *req);
94#define EIO_FINISH(req) req_invoke (req) 146#define EIO_FINISH(req) req_invoke (req)
95static void aio_grp_feed (eio_req *grp);
96#define EIO_FEED(req) aio_grp_feed (req)
97static void req_destroy (eio_req *grp); 147static void req_destroy (eio_req *grp);
98#define EIO_DESTROY(req) req_destroy (req) 148#define EIO_DESTROY(req) req_destroy (req)
99 149
100enum { 150enum {
101 FLAG_SV2_RO_OFF = 0x40, /* data was set readonly */ 151 FLAG_SV2_RO_OFF = 0x40, /* data was set readonly */
105 155
106typedef eio_req *aio_req; 156typedef eio_req *aio_req;
107typedef eio_req *aio_req_ornot; 157typedef eio_req *aio_req_ornot;
108 158
109static SV *on_next_submit; 159static SV *on_next_submit;
110static int next_pri = EIO_DEFAULT_PRI + EIO_PRI_BIAS; 160static int next_pri = EIO_PRI_DEFAULT;
111static int max_outstanding; 161static int max_outstanding;
112 162
113static int respipe_osf [2], respipe [2] = { -1, -1 }; 163static s_epipe respipe;
114 164
115static void req_destroy (aio_req req); 165static void req_destroy (aio_req req);
116static void req_cancel (aio_req req); 166static void req_cancel (aio_req req);
117 167
118static void want_poll (void) 168static void want_poll (void)
119{ 169{
120 /* write a dummy byte to the pipe so fh becomes ready */ 170 /* write a dummy byte to the pipe so fh becomes ready */
121 respipe_write (respipe_osf [1], (const void *)&respipe_osf, 1); 171 s_epipe_signal (&respipe);
122} 172}
123 173
124static void done_poll (void) 174static void done_poll (void)
125{ 175{
126 /* read any signals sent by the worker threads */ 176 /* read any signals sent by the worker threads */
127 char buf [4]; 177 s_epipe_drain (&respipe);
128 while (respipe_read (respipe [0], buf, 4) == 4)
129 ;
130} 178}
131 179
132/* must be called at most once */ 180/* must be called at most once */
133static SV *req_sv (aio_req req, const char *klass) 181static SV *req_sv (aio_req req, HV *stash)
134{ 182{
135 if (!req->self) 183 if (!req->self)
136 { 184 {
137 req->self = (SV *)newHV (); 185 req->self = (SV *)newHV ();
138 sv_magic (req->self, 0, PERL_MAGIC_ext, (char *)req, 0); 186 sv_magic (req->self, 0, PERL_MAGIC_ext, (char *)req, 0);
139 } 187 }
140 188
141 return sv_2mortal (sv_bless (newRV_inc (req->self), gv_stashpv (klass, 1))); 189 return sv_2mortal (sv_bless (newRV_inc (req->self), stash));
142} 190}
143 191
144static aio_req SvAIO_REQ (SV *sv) 192static aio_req SvAIO_REQ (SV *sv)
145{ 193{
146 MAGIC *mg; 194 MAGIC *mg;
147 195
148 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")))
149 croak ("object of class " AIO_REQ_KLASS " expected"); 200 croak ("object of class IO::AIO::REQ expected");
150 201
151 mg = mg_find (SvRV (sv), PERL_MAGIC_ext); 202 mg = mg_find (SvRV (sv), PERL_MAGIC_ext);
152 203
153 return mg ? (aio_req)mg->mg_ptr : 0; 204 return mg ? (aio_req)mg->mg_ptr : 0;
154} 205}
160 dSP; 211 dSP;
161 212
162 ENTER; 213 ENTER;
163 SAVETMPS; 214 SAVETMPS;
164 PUSHMARK (SP); 215 PUSHMARK (SP);
165 XPUSHs (req_sv (grp, AIO_GRP_KLASS)); 216 XPUSHs (req_sv (grp, aio_grp_stash));
166 PUTBACK; 217 PUTBACK;
167 call_sv (grp->sv2, G_VOID | G_EVAL | G_KEEPERR); 218 call_sv (grp->sv2, G_VOID | G_EVAL | G_KEEPERR);
168 SPAGAIN; 219 SPAGAIN;
169 FREETMPS; 220 FREETMPS;
170 LEAVE; 221 LEAVE;
173 224
174static void req_submit (eio_req *req) 225static void req_submit (eio_req *req)
175{ 226{
176 eio_submit (req); 227 eio_submit (req);
177 228
178 if (on_next_submit) 229 if (expect_false (on_next_submit))
179 { 230 {
180 dSP; 231 dSP;
181 SV *cb = sv_2mortal (on_next_submit); 232 SV *cb = sv_2mortal (on_next_submit);
182 233
183 on_next_submit = 0; 234 on_next_submit = 0;
193 dSP; 244 dSP;
194 245
195 if (req->flags & FLAG_SV2_RO_OFF) 246 if (req->flags & FLAG_SV2_RO_OFF)
196 SvREADONLY_off (req->sv2); 247 SvREADONLY_off (req->sv2);
197 248
198 if (!EIO_CANCELLED (req) && SvOK (req->callback)) 249 if (!EIO_CANCELLED (req) && req->callback)
199 { 250 {
200 ENTER; 251 ENTER;
201 SAVETMPS; 252 SAVETMPS;
202 PUSHMARK (SP); 253 PUSHMARK (SP);
203 EXTEND (SP, 1); 254 EXTEND (SP, 1);
209 SV *rv = &PL_sv_undef; 260 SV *rv = &PL_sv_undef;
210 261
211 if (req->result >= 0) 262 if (req->result >= 0)
212 { 263 {
213 int i; 264 int i;
214 char *buf = req->ptr2; 265 char *names = (char *)req->ptr2;
266 eio_dirent *ent = (eio_dirent *)req->ptr1; /* might be 0 */
215 AV *av = newAV (); 267 AV *av = newAV ();
216 268
217 av_extend (av, req->result - 1); 269 av_extend (av, req->result - 1);
218 270
219 for (i = 0; i < req->result; ++i) 271 for (i = 0; i < req->result; ++i)
220 { 272 {
221 SV *sv = newSVpv (buf, 0); 273 if (req->int1 & EIO_READDIR_DENTS)
274 {
275 SV *namesv = newSVpvn (names + ent->nameofs, ent->namelen);
222 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
223 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);
224 buf += SvCUR (sv) + 1; 305 names += SvCUR (name) + 1;
306 }
225 } 307 }
226 308
227 rv = sv_2mortal (newRV_noinc ((SV *)av)); 309 rv = sv_2mortal (newRV_noinc ((SV *)av));
228 } 310 }
229 311
230 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))));
231 } 316 }
232 break; 317 break;
233 318
234 case EIO_OPEN: 319 case EIO_OPEN:
235 { 320 {
241 GV *gv = (GV *)sv_newmortal (); 326 GV *gv = (GV *)sv_newmortal ();
242 int flags = req->int1 & (O_RDONLY | O_WRONLY | O_RDWR); 327 int flags = req->int1 & (O_RDONLY | O_WRONLY | O_RDWR);
243 char sym [64]; 328 char sym [64];
244 int symlen; 329 int symlen;
245 330
246 symlen = snprintf (sym, sizeof (sym), "fd#%d", req->result); 331 symlen = snprintf (sym, sizeof (sym), "fd#%d", (int)req->result);
247 gv_init (gv, stash, sym, symlen, 0); 332 gv_init (gv, aio_stash, sym, symlen, 0);
248 333
249 symlen = snprintf ( 334 symlen = snprintf (
250 sym, 335 sym,
251 sizeof (sym), 336 sizeof (sym),
252 "%s&=%d", 337 "%s&=%d",
253 flags == O_RDONLY ? "<" : flags == O_WRONLY ? ">" : "+<", 338 flags == O_RDONLY ? "<" : flags == O_WRONLY ? ">" : "+<",
254 req->result 339 (int)req->result
255 ); 340 );
256 341
257 if (do_open (gv, sym, symlen, 0, 0, 0, 0)) 342 if (do_open (gv, sym, symlen, 0, 0, 0, 0))
258 fh = (SV *)gv; 343 fh = (SV *)gv;
259 } 344 }
280 case EIO_BUSY: 365 case EIO_BUSY:
281 break; 366 break;
282 367
283 case EIO_READLINK: 368 case EIO_READLINK:
284 if (req->result > 0) 369 if (req->result > 0)
285 { 370 PUSHs (sv_2mortal (newSVpvn (req->ptr2, req->result)));
286 SvCUR_set (req->sv2, req->result);
287 *SvEND (req->sv2) = 0;
288 PUSHs (req->sv2);
289 }
290 break; 371 break;
291 372
292 case EIO_STAT: 373 case EIO_STAT:
293 case EIO_LSTAT: 374 case EIO_LSTAT:
294 case EIO_FSTAT: 375 case EIO_FSTAT:
297 PL_statcache = *(EIO_STRUCT_STAT *)(req->ptr2); 378 PL_statcache = *(EIO_STRUCT_STAT *)(req->ptr2);
298 PUSHs (sv_2mortal (newSViv (req->result))); 379 PUSHs (sv_2mortal (newSViv (req->result)));
299 break; 380 break;
300 381
301 case EIO_READ: 382 case EIO_READ:
383 {
302 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));
303 *SvEND (req->sv2) = 0; 385 *SvEND (req->sv2) = 0;
386 SvPOK_only (req->sv2);
387 SvSETMAGIC (req->sv2);
304 PUSHs (sv_2mortal (newSViv (req->result))); 388 PUSHs (sv_2mortal (newSViv (req->result)));
389 }
305 break; 390 break;
306 391
307 case EIO_DUP2: 392 case EIO_DUP2:
308 if (req->result > 0) 393 if (req->result > 0)
309 req->result = 0; 394 req->result = 0;
355 grp->sv2 = 0; 440 grp->sv2 = 0;
356 441
357 eio_grp_cancel (grp); 442 eio_grp_cancel (grp);
358} 443}
359 444
360#ifdef USE_SOCKETS_AS_HANDLES
361# define TO_SOCKET(x) (win32_get_osfhandle (x))
362#else
363# define TO_SOCKET(x) (x)
364#endif
365
366static void 445static void
367create_respipe (void) 446create_respipe (void)
368{ 447{
369 int old_readfd = respipe [0];
370
371 if (respipe [1] >= 0)
372 respipe_close (TO_SOCKET (respipe [1]));
373
374#ifdef _WIN32
375 if (PerlSock_socketpair (AF_UNIX, SOCK_STREAM, 0, respipe))
376#else
377 if (pipe (respipe)) 448 if (s_epipe_renew (&respipe))
378#endif
379 croak ("unable to initialize result pipe"); 449 croak ("IO::AIO: unable to initialize result pipe");
380
381 if (old_readfd >= 0)
382 {
383 if (dup2 (TO_SOCKET (respipe [0]), TO_SOCKET (old_readfd)) < 0)
384 croak ("unable to initialize result pipe(2)");
385
386 respipe_close (respipe [0]);
387 respipe [0] = old_readfd;
388 }
389
390#ifdef _WIN32
391 int arg = 1;
392 if (ioctlsocket (TO_SOCKET (respipe [0]), FIONBIO, &arg)
393 || ioctlsocket (TO_SOCKET (respipe [1]), FIONBIO, &arg))
394#else
395 if (fcntl (respipe [0], F_SETFL, O_NONBLOCK)
396 || fcntl (respipe [1], F_SETFL, O_NONBLOCK))
397#endif
398 croak ("unable to initialize result pipe(3)");
399
400 respipe_osf [0] = TO_SOCKET (respipe [0]);
401 respipe_osf [1] = TO_SOCKET (respipe [1]);
402} 450}
403 451
404static void poll_wait (void) 452static void poll_wait (void)
405{ 453{
406 fd_set rfd;
407
408 while (eio_nreqs ()) 454 while (eio_nreqs ())
409 { 455 {
410 int size; 456 int size;
411 457
412 X_LOCK (reslock); 458 X_LOCK (reslock);
414 X_UNLOCK (reslock); 460 X_UNLOCK (reslock);
415 461
416 if (size) 462 if (size)
417 return; 463 return;
418 464
419 maybe_start_thread (); 465 etp_maybe_start_thread ();
420 466
421 FD_ZERO (&rfd); 467 s_epipe_wait (&respipe);
422 FD_SET (respipe [0], &rfd);
423
424 PerlSock_select (respipe [0] + 1, &rfd, 0, 0, 0);
425 } 468 }
426} 469}
427 470
428static int poll_cb (void) 471static int poll_cb (void)
429{ 472{
430 int res; 473 for (;;)
431
432 do
433 { 474 {
434 res = eio_poll (); 475 int res = eio_poll ();
435 476
436 if (res > 0) 477 if (res > 0)
437 croak (0); 478 croak (0);
479
480 if (!max_outstanding || max_outstanding > eio_nreqs ())
481 return res;
482
483 poll_wait ();
438 } 484 }
439 while (max_outstanding && max_outstanding <= eio_nreqs ());
440
441 return res;
442} 485}
443 486
444static void atfork_child (void) 487static void atfork_child (void)
445{ 488{
446 create_respipe (); 489 create_respipe ();
447} 490}
448 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
449#define dREQ \ 499#define dREQ \
500 SV *cb_cv; \
450 aio_req req; \ 501 aio_req req; \
451 int req_pri = next_pri; \ 502 int req_pri = next_pri; \
452 next_pri = EIO_DEFAULT_PRI + EIO_PRI_BIAS; \ 503 next_pri = EIO_PRI_DEFAULT; \
453 \ 504 \
454 if (SvOK (callback) && !SvROK (callback)) \ 505 cb_cv = get_cb (callback); \
455 croak ("callback must be undef or of reference type"); \
456 \ 506 \
457 Newz (0, req, 1, eio_req); \ 507 Newz (0, req, 1, eio_req); \
458 if (!req) \ 508 if (!req) \
459 croak ("out of memory during eio_req allocation"); \ 509 croak ("out of memory during eio_req allocation"); \
460 \ 510 \
461 req->callback = newSVsv (callback); \ 511 req->callback = SvREFCNT_inc (cb_cv); \
462 req->pri = req_pri 512 req->pri = req_pri
463 513
464#define REQ_SEND \ 514#define REQ_SEND \
515 PUTBACK; \
465 req_submit (req); \ 516 req_submit (req); \
517 SPAGAIN; \
466 \ 518 \
467 if (GIMME_V != G_VOID) \ 519 if (GIMME_V != G_VOID) \
468 XPUSHs (req_sv (req, AIO_REQ_KLASS)); 520 XPUSHs (req_sv (req, aio_req_stash));
469 521
470MODULE = IO::AIO PACKAGE = IO::AIO 522MODULE = IO::AIO PACKAGE = IO::AIO
471 523
472PROTOTYPES: ENABLE 524PROTOTYPES: ENABLE
473 525
474BOOT: 526BOOT:
475{ 527{
476 stash = gv_stashpv ("IO::AIO", 1); 528 static const struct {
477 529 const char *name;
478 newCONSTSUB (stash, "EXDEV", newSViv (EXDEV)); 530 IV iv;
479 newCONSTSUB (stash, "O_RDONLY", newSViv (O_RDONLY)); 531 } *civ, const_iv[] = {
480 newCONSTSUB (stash, "O_WRONLY", newSViv (O_WRONLY)); 532# define const_iv(name, value) { # name, (IV) value },
481 newCONSTSUB (stash, "O_CREAT", newSViv (O_CREAT)); 533# define const_eio(name) { # name, (IV) EIO_ ## name },
482 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)
483#ifdef _WIN32 540#ifndef _WIN32
484 X_MUTEX_CHECK (wrklock); 541 const_iv (S_IFIFO , S_IFIFO)
485 X_MUTEX_CHECK (reslock);
486 X_MUTEX_CHECK (reqlock);
487 X_MUTEX_CHECK (reqwait);
488 X_MUTEX_CHECK (preadwritelock);
489 X_MUTEX_CHECK (readdirlock);
490
491 X_COND_CHECK (reqwait);
492#else
493 newCONSTSUB (stash, "S_IFIFO", newSViv (S_IFIFO));
494#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)
495 549
550 const_eio (MS_ASYNC)
551 const_eio (MS_INVALIDATE)
552 const_eio (MS_SYNC)
553
554 const_eio (MT_MODIFY)
555
556 const_eio (SYNC_FILE_RANGE_WAIT_BEFORE)
557 const_eio (SYNC_FILE_RANGE_WRITE)
558 const_eio (SYNC_FILE_RANGE_WAIT_AFTER)
559
560 const_eio (READDIR_DENTS)
561 const_eio (READDIR_DIRS_FIRST)
562 const_eio (READDIR_STAT_ORDER)
563 const_eio (READDIR_FOUND_UNKNOWN)
564
565 const_eio (DT_UNKNOWN)
566 const_eio (DT_FIFO)
567 const_eio (DT_CHR)
568 const_eio (DT_DIR)
569 const_eio (DT_BLK)
570 const_eio (DT_REG)
571 const_eio (DT_LNK)
572 const_eio (DT_SOCK)
573 const_eio (DT_WHT)
574 };
575
576 aio_stash = gv_stashpv ("IO::AIO" , 1);
577 aio_req_stash = gv_stashpv ("IO::AIO::REQ", 1);
578 aio_grp_stash = gv_stashpv ("IO::AIO::GRP", 1);
579
580 for (civ = const_iv + sizeof (const_iv) / sizeof (const_iv [0]); civ-- > const_iv; )
581 newCONSTSUB (aio_stash, (char *)civ->name, newSViv (civ->iv));
582
496 create_respipe (); 583 create_respipe ();
497 584
498 if (eio_init (want_poll, done_poll) < 0) 585 if (eio_init (want_poll, done_poll) < 0)
499 croak ("IO::AIO: unable to initialise eio library"); 586 croak ("IO::AIO: unable to initialise eio library");
500 587
501 /* atfork child called in fifo order, so before eio's handler */ 588 /* atfork child called in fifo order, so before eio's handler */
502 X_THREAD_ATFORK (0, 0, atfork_child); 589 X_THREAD_ATFORK (0, 0, atfork_child);
503} 590}
504 591
505void 592void
506max_poll_reqs (int nreqs) 593max_poll_reqs (int nreqs)
507 PROTOTYPE: $ 594 PROTOTYPE: $
560 ALIAS: 647 ALIAS:
561 aio_fsync = EIO_FSYNC 648 aio_fsync = EIO_FSYNC
562 aio_fdatasync = EIO_FDATASYNC 649 aio_fdatasync = EIO_FDATASYNC
563 PPCODE: 650 PPCODE:
564{ 651{
652 int fd = s_fileno_croak (fh, 0);
565 dREQ; 653 dREQ;
566 654
567 req->type = ix; 655 req->type = ix;
568 req->sv1 = newSVsv (fh); 656 req->sv1 = newSVsv (fh);
569 req->int1 = PerlIO_fileno (IoIFP (sv_2io (fh))); 657 req->int1 = fd;
658
659 REQ_SEND (req);
660}
661
662void
663aio_sync_file_range (SV *fh, off_t offset, size_t nbytes, UV flags, SV *callback=&PL_sv_undef)
664 PROTOTYPE: $$$$;$
665 PPCODE:
666{
667 int fd = s_fileno_croak (fh, 0);
668 dREQ;
669
670 req->type = EIO_SYNC_FILE_RANGE;
671 req->sv1 = newSVsv (fh);
672 req->int1 = fd;
673 req->offs = offset;
674 req->size = nbytes;
675 req->int2 = flags;
570 676
571 REQ_SEND (req); 677 REQ_SEND (req);
572} 678}
573 679
574void 680void
575aio_close (SV *fh, SV *callback=&PL_sv_undef) 681aio_close (SV *fh, SV *callback=&PL_sv_undef)
576 PROTOTYPE: $;$ 682 PROTOTYPE: $;$
577 PPCODE: 683 PPCODE:
578{ 684{
579 static int close_pipe = -1; /* dummy fd to close fds via dup2 */ 685 static int close_pipe = -1; /* dummy fd to close fds via dup2 */
686 int fd = s_fileno_croak (fh, 0);
580 dREQ; 687 dREQ;
581 688
582 if (close_pipe < 0) 689 if (close_pipe < 0)
583 { 690 {
584 int pipefd [2]; 691 int pipefd [2];
592 } 699 }
593 700
594 req->type = EIO_DUP2; 701 req->type = EIO_DUP2;
595 req->int1 = close_pipe; 702 req->int1 = close_pipe;
596 req->sv2 = newSVsv (fh); 703 req->sv2 = newSVsv (fh);
597 req->int2 = PerlIO_fileno (IoIFP (sv_2io (fh))); 704 req->int2 = fd;
598 705
599 REQ_SEND (req); 706 REQ_SEND (req);
600} 707}
601 708
602void 709void
606 aio_write = EIO_WRITE 713 aio_write = EIO_WRITE
607 PROTOTYPE: $$$$$;$ 714 PROTOTYPE: $$$$$;$
608 PPCODE: 715 PPCODE:
609{ 716{
610 STRLEN svlen; 717 STRLEN svlen;
718 int fd = s_fileno_croak (fh, ix == EIO_WRITE);
611 char *svptr = SvPVbyte (data, svlen); 719 char *svptr = SvPVbyte (data, svlen);
612 UV len = SvUV (length); 720 UV len = SvUV (length);
613
614 SvUPGRADE (data, SVt_PV);
615 SvPOK_on (data);
616 721
617 if (dataoffset < 0) 722 if (dataoffset < 0)
618 dataoffset += svlen; 723 dataoffset += svlen;
619 724
620 if (dataoffset < 0 || dataoffset > svlen) 725 if (dataoffset < 0 || dataoffset > svlen)
626 if (!SvOK (length) || len + dataoffset > svlen) 731 if (!SvOK (length) || len + dataoffset > svlen)
627 len = svlen - dataoffset; 732 len = svlen - dataoffset;
628 } 733 }
629 else 734 else
630 { 735 {
631 /* read: grow scalar as necessary */ 736 /* read: check type and grow scalar as necessary */
737 SvUPGRADE (data, SVt_PV);
632 svptr = SvGROW (data, len + dataoffset + 1); 738 svptr = SvGROW (data, len + dataoffset + 1);
633 } 739 }
634
635 if (len < 0)
636 croak ("length must not be negative");
637 740
638 { 741 {
639 dREQ; 742 dREQ;
640 743
641 req->type = ix; 744 req->type = ix;
642 req->sv1 = newSVsv (fh); 745 req->sv1 = newSVsv (fh);
643 req->int1 = PerlIO_fileno (ix == EIO_READ ? IoIFP (sv_2io (fh)) 746 req->int1 = fd;
644 : IoOFP (sv_2io (fh)));
645 req->offs = SvOK (offset) ? SvVAL64 (offset) : -1; 747 req->offs = SvOK (offset) ? SvVAL64 (offset) : -1;
646 req->size = len; 748 req->size = len;
647 req->sv2 = SvREFCNT_inc (data); 749 req->sv2 = SvREFCNT_inc (data);
648 req->ptr2 = (char *)svptr + dataoffset; 750 req->ptr2 = (char *)svptr + dataoffset;
649 req->stroffset = dataoffset; 751 req->stroffset = dataoffset;
664 PPCODE: 766 PPCODE:
665{ 767{
666 SV *data; 768 SV *data;
667 dREQ; 769 dREQ;
668 770
669 data = newSV (NAME_MAX);
670 SvPOK_on (data);
671
672 req->type = EIO_READLINK; 771 req->type = EIO_READLINK;
673 req->sv1 = newSVsv (path); 772 req->sv1 = newSVsv (path);
674 req->ptr1 = SvPVbyte_nolen (req->sv1); 773 req->ptr1 = SvPVbyte_nolen (req->sv1);
675 req->sv2 = data;
676 req->ptr2 = SvPVbyte_nolen (data);
677 774
678 REQ_SEND; 775 REQ_SEND;
679} 776}
680 777
681void 778void
682aio_sendfile (SV *out_fh, SV *in_fh, SV *in_offset, UV length, SV *callback=&PL_sv_undef) 779aio_sendfile (SV *out_fh, SV *in_fh, off_t in_offset, size_t length, SV *callback=&PL_sv_undef)
683 PROTOTYPE: $$$$;$ 780 PROTOTYPE: $$$$;$
684 PPCODE: 781 PPCODE:
685{ 782{
783 int ifd = s_fileno_croak (in_fh , 0);
784 int ofd = s_fileno_croak (out_fh, 1);
686 dREQ; 785 dREQ;
687 786
688 req->type = EIO_SENDFILE; 787 req->type = EIO_SENDFILE;
689 req->sv1 = newSVsv (out_fh); 788 req->sv1 = newSVsv (out_fh);
690 req->int1 = PerlIO_fileno (IoIFP (sv_2io (out_fh))); 789 req->int1 = ofd;
691 req->sv2 = newSVsv (in_fh); 790 req->sv2 = newSVsv (in_fh);
692 req->int2 = PerlIO_fileno (IoIFP (sv_2io (in_fh))); 791 req->int2 = ifd;
693 req->offs = SvVAL64 (in_offset); 792 req->offs = in_offset;
694 req->size = length; 793 req->size = length;
695 794
696 REQ_SEND; 795 REQ_SEND;
697} 796}
698 797
699void 798void
700aio_readahead (SV *fh, SV *offset, IV length, SV *callback=&PL_sv_undef) 799aio_readahead (SV *fh, off_t offset, size_t length, SV *callback=&PL_sv_undef)
701 PROTOTYPE: $$$;$ 800 PROTOTYPE: $$$;$
702 PPCODE: 801 PPCODE:
703{ 802{
803 int fd = s_fileno_croak (fh, 0);
704 dREQ; 804 dREQ;
705 805
706 req->type = EIO_READAHEAD; 806 req->type = EIO_READAHEAD;
707 req->sv1 = newSVsv (fh); 807 req->sv1 = newSVsv (fh);
708 req->int1 = PerlIO_fileno (IoIFP (sv_2io (fh))); 808 req->int1 = fd;
709 req->offs = SvVAL64 (offset); 809 req->offs = offset;
710 req->size = length; 810 req->size = length;
711 811
712 REQ_SEND; 812 REQ_SEND;
713} 813}
714 814
719 aio_lstat = EIO_LSTAT 819 aio_lstat = EIO_LSTAT
720 PPCODE: 820 PPCODE:
721{ 821{
722 dREQ; 822 dREQ;
723 823
724 req->ptr2 = malloc (sizeof (EIO_STRUCT_STAT));
725 if (!req->ptr2)
726 {
727 req_destroy (req);
728 croak ("out of memory during aio_stat statdata allocation");
729 }
730
731 req->flags |= EIO_FLAG_PTR2_FREE;
732 req->sv1 = newSVsv (fh_or_path); 824 req->sv1 = newSVsv (fh_or_path);
733 825
734 if (SvPOK (fh_or_path)) 826 if (SvPOK (req->sv1))
735 { 827 {
736 req->type = ix; 828 req->type = ix;
737 req->ptr1 = SvPVbyte_nolen (req->sv1); 829 req->ptr1 = SvPVbyte_nolen (req->sv1);
738 } 830 }
739 else 831 else
753 845
754 req->nv1 = SvOK (atime) ? SvNV (atime) : -1.; 846 req->nv1 = SvOK (atime) ? SvNV (atime) : -1.;
755 req->nv2 = SvOK (mtime) ? SvNV (mtime) : -1.; 847 req->nv2 = SvOK (mtime) ? SvNV (mtime) : -1.;
756 req->sv1 = newSVsv (fh_or_path); 848 req->sv1 = newSVsv (fh_or_path);
757 849
758 if (SvPOK (fh_or_path)) 850 if (SvPOK (req->sv1))
759 { 851 {
760 req->type = EIO_UTIME; 852 req->type = EIO_UTIME;
761 req->ptr1 = SvPVbyte_nolen (req->sv1); 853 req->ptr1 = SvPVbyte_nolen (req->sv1);
762 } 854 }
763 else 855 else
776 dREQ; 868 dREQ;
777 869
778 req->sv1 = newSVsv (fh_or_path); 870 req->sv1 = newSVsv (fh_or_path);
779 req->offs = SvOK (offset) ? SvVAL64 (offset) : -1; 871 req->offs = SvOK (offset) ? SvVAL64 (offset) : -1;
780 872
781 if (SvPOK (fh_or_path)) 873 if (SvPOK (req->sv1))
782 { 874 {
783 req->type = EIO_TRUNCATE; 875 req->type = EIO_TRUNCATE;
784 req->ptr1 = SvPVbyte_nolen (req->sv1); 876 req->ptr1 = SvPVbyte_nolen (req->sv1);
785 } 877 }
786 else 878 else
794 886
795void 887void
796aio_chmod (SV8 *fh_or_path, int mode, SV *callback=&PL_sv_undef) 888aio_chmod (SV8 *fh_or_path, int mode, SV *callback=&PL_sv_undef)
797 ALIAS: 889 ALIAS:
798 aio_chmod = EIO_CHMOD 890 aio_chmod = EIO_CHMOD
799 aio_fchmod = EIO_FCHMOD
800 aio_mkdir = EIO_MKDIR 891 aio_mkdir = EIO_MKDIR
801 PPCODE: 892 PPCODE:
802{ 893{
803 dREQ; 894 dREQ;
804 895
805 req->type = ix;
806 req->int2 = mode; 896 req->int2 = mode;
807 req->sv1 = newSVsv (fh_or_path); 897 req->sv1 = newSVsv (fh_or_path);
808 898
809 if (ix == EIO_FCHMOD) 899 if (SvPOK (req->sv1))
810 req->int1 = PerlIO_fileno (IoIFP (sv_2io (fh_or_path))); 900 {
901 req->type = ix;
902 req->ptr1 = SvPVbyte_nolen (req->sv1);
903 }
811 else 904 else
812 req->ptr1 = SvPVbyte_nolen (req->sv1); 905 {
906 req->type = EIO_FCHMOD;
907 req->int1 = PerlIO_fileno (IoIFP (sv_2io (fh_or_path)));
908 }
813 909
814 REQ_SEND; 910 REQ_SEND;
815} 911}
816 912
817void 913void
822 918
823 req->int2 = SvOK (uid) ? SvIV (uid) : -1; 919 req->int2 = SvOK (uid) ? SvIV (uid) : -1;
824 req->int3 = SvOK (gid) ? SvIV (gid) : -1; 920 req->int3 = SvOK (gid) ? SvIV (gid) : -1;
825 req->sv1 = newSVsv (fh_or_path); 921 req->sv1 = newSVsv (fh_or_path);
826 922
827 if (SvPOK (fh_or_path)) 923 if (SvPOK (req->sv1))
828 { 924 {
829 req->type = EIO_CHOWN; 925 req->type = EIO_CHOWN;
830 req->ptr1 = SvPVbyte_nolen (req->sv1); 926 req->ptr1 = SvPVbyte_nolen (req->sv1);
831 } 927 }
832 else 928 else
834 req->type = EIO_FCHOWN; 930 req->type = EIO_FCHOWN;
835 req->int1 = PerlIO_fileno (IoIFP (sv_2io (fh_or_path))); 931 req->int1 = PerlIO_fileno (IoIFP (sv_2io (fh_or_path)));
836 } 932 }
837 933
838 REQ_SEND; 934 REQ_SEND;
935}
936
937void
938aio_readdirx (SV8 *pathname, IV flags, SV *callback=&PL_sv_undef)
939 PPCODE:
940{
941 dREQ;
942
943 req->type = EIO_READDIR;
944 req->sv1 = newSVsv (pathname);
945 req->ptr1 = SvPVbyte_nolen (req->sv1);
946 req->int1 = flags | EIO_READDIR_DENTS | EIO_READDIR_CUSTOM1;
947
948 if (flags & EIO_READDIR_DENTS)
949 req->int1 |= EIO_READDIR_CUSTOM2;
950
951 REQ_SEND;
839} 952}
840 953
841void 954void
842aio_unlink (SV8 *pathname, SV *callback=&PL_sv_undef) 955aio_unlink (SV8 *pathname, SV *callback=&PL_sv_undef)
843 ALIAS: 956 ALIAS:
888 1001
889 REQ_SEND; 1002 REQ_SEND;
890} 1003}
891 1004
892void 1005void
1006aio_mtouch (SV8 *data, IV offset = 0, SV *length = &PL_sv_undef, int flags = 0, SV *callback=&PL_sv_undef)
1007 ALIAS:
1008 aio_mtouch = EIO_MTOUCH
1009 aio_msync = EIO_MSYNC
1010 PROTOTYPE: $$$$;$
1011 PPCODE:
1012{
1013 STRLEN svlen;
1014 UV len = SvUV (length);
1015 char *svptr = SvPVbyte (data, svlen);
1016
1017 if (offset < 0)
1018 offset += svlen;
1019
1020 if (offset < 0 || offset > svlen)
1021 croak ("offset outside of scalar");
1022
1023 if (!SvOK (length) || len + offset > svlen)
1024 len = svlen - offset;
1025
1026 {
1027 dREQ;
1028
1029 req->type = ix;
1030 req->size = len;
1031 req->sv2 = SvREFCNT_inc (data);
1032 req->ptr2 = (char *)svptr + offset;
1033 req->int1 = flags;
1034
1035 REQ_SEND;
1036 }
1037}
1038
1039void
893aio_busy (double delay, SV *callback=&PL_sv_undef) 1040aio_busy (double delay, SV *callback=&PL_sv_undef)
894 PPCODE: 1041 PPCODE:
895{ 1042{
896 dREQ; 1043 dREQ;
897 1044
908{ 1055{
909 dREQ; 1056 dREQ;
910 1057
911 req->type = EIO_GROUP; 1058 req->type = EIO_GROUP;
912 1059
913 eio_submit (req); 1060 req_submit (req);
914 XPUSHs (req_sv (req, AIO_GRP_KLASS)); 1061 XPUSHs (req_sv (req, aio_grp_stash));
915} 1062}
916 1063
917void 1064void
918aio_nop (SV *callback=&PL_sv_undef) 1065aio_nop (SV *callback=&PL_sv_undef)
919 ALIAS: 1066 ALIAS:
930 1077
931int 1078int
932aioreq_pri (int pri = 0) 1079aioreq_pri (int pri = 0)
933 PROTOTYPE: ;$ 1080 PROTOTYPE: ;$
934 CODE: 1081 CODE:
935 RETVAL = next_pri - EIO_PRI_BIAS; 1082 RETVAL = next_pri;
936 if (items > 0) 1083 if (items > 0)
937 { 1084 {
938 if (pri < EIO_PRI_MIN) pri = EIO_PRI_MIN; 1085 if (pri < EIO_PRI_MIN) pri = EIO_PRI_MIN;
939 if (pri > EIO_PRI_MAX) pri = EIO_PRI_MAX; 1086 if (pri > EIO_PRI_MAX) pri = EIO_PRI_MAX;
940 next_pri = pri + EIO_PRI_BIAS; 1087 next_pri = pri;
941 } 1088 }
942 OUTPUT: 1089 OUTPUT:
943 RETVAL 1090 RETVAL
944 1091
945void 1092void
946aioreq_nice (int nice = 0) 1093aioreq_nice (int nice = 0)
947 CODE: 1094 CODE:
948 nice = next_pri - nice; 1095 nice = next_pri - nice;
949 if (nice < EIO_PRI_MIN) nice = EIO_PRI_MIN; 1096 if (nice < EIO_PRI_MIN) nice = EIO_PRI_MIN;
950 if (nice > EIO_PRI_MAX) nice = EIO_PRI_MAX; 1097 if (nice > EIO_PRI_MAX) nice = EIO_PRI_MAX;
951 next_pri = nice + EIO_PRI_BIAS; 1098 next_pri = nice;
952 1099
953void 1100void
954flush () 1101flush ()
955 PROTOTYPE: 1102 PROTOTYPE:
956 CODE: 1103 CODE:
971 1118
972int 1119int
973poll_fileno() 1120poll_fileno()
974 PROTOTYPE: 1121 PROTOTYPE:
975 CODE: 1122 CODE:
976 RETVAL = respipe [0]; 1123 RETVAL = s_epipe_fd (&respipe);
977 OUTPUT: 1124 OUTPUT:
978 RETVAL 1125 RETVAL
979 1126
980int 1127int
981poll_cb(...) 1128poll_cb(...)
1017 1164
1018int 1165int
1019nthreads() 1166nthreads()
1020 PROTOTYPE: 1167 PROTOTYPE:
1021 CODE: 1168 CODE:
1022 if (WORDACCESS_UNSAFE) X_LOCK (wrklock); 1169 RETVAL = eio_nthreads ();
1023 RETVAL = started;
1024 if (WORDACCESS_UNSAFE) X_UNLOCK (wrklock);
1025 OUTPUT: 1170 OUTPUT:
1026 RETVAL 1171 RETVAL
1027 1172
1173int
1174fadvise (aio_rfd fh, off_t offset, off_t length, IV advice)
1175 PROTOTYPE: $$$$
1176 CODE:
1177#if _XOPEN_SOURCE >= 600 && !NO_FADVISE
1178 RETVAL = posix_fadvise (fh, offset, length, advice);
1179#else
1180 RETVAL = errno = ENOSYS;
1181#endif
1182 OUTPUT:
1183 RETVAL
1184
1185ssize_t
1186sendfile (aio_wfd ofh, aio_rfd ifh, off_t offset, size_t count)
1187 PROTOTYPE: $$$$
1188 CODE:
1189 RETVAL = eio_sendfile_sync (ofh, ifh, offset, count);
1190 OUTPUT:
1191 RETVAL
1192
1028void _on_next_submit (SV *cb) 1193void _on_next_submit (SV *cb)
1029 CODE: 1194 CODE:
1030 SvREFCNT_dec (on_next_submit); 1195 SvREFCNT_dec (on_next_submit);
1031 on_next_submit = SvOK (cb) ? newSVsv (cb) : 0; 1196 on_next_submit = SvOK (cb) ? newSVsv (cb) : 0;
1032 1197
1039 CODE: 1204 CODE:
1040 eio_cancel (req); 1205 eio_cancel (req);
1041 1206
1042void 1207void
1043cb (aio_req_ornot req, SV *callback=&PL_sv_undef) 1208cb (aio_req_ornot req, SV *callback=&PL_sv_undef)
1044 CODE: 1209 PPCODE:
1210{
1211 if (GIMME_V != G_VOID)
1212 XPUSHs (req->callback ? sv_2mortal (newRV_inc (req->callback)) : &PL_sv_undef);
1213
1214 if (items > 1)
1215 {
1216 SV *cb_cv =get_cb (callback);
1217
1045 SvREFCNT_dec (req->callback); 1218 SvREFCNT_dec (req->callback);
1046 req->callback = newSVsv (callback); 1219 req->callback = SvREFCNT_inc (cb_cv);
1220 }
1221}
1047 1222
1048MODULE = IO::AIO PACKAGE = IO::AIO::GRP 1223MODULE = IO::AIO PACKAGE = IO::AIO::GRP
1049 1224
1050void 1225void
1051add (aio_req grp, ...) 1226add (aio_req grp, ...)
1083 AV *av; 1258 AV *av;
1084 1259
1085 grp->errorno = errno; 1260 grp->errorno = errno;
1086 1261
1087 av = newAV (); 1262 av = newAV ();
1263 av_extend (av, items - 1);
1088 1264
1089 for (i = 1; i < items; ++i ) 1265 for (i = 1; i < items; ++i )
1090 av_push (av, newSVsv (ST (i))); 1266 av_push (av, newSVsv (ST (i)));
1091 1267
1092 SvREFCNT_dec (grp->sv1); 1268 SvREFCNT_dec (grp->sv1);
1106void 1282void
1107feed (aio_req grp, SV *callback=&PL_sv_undef) 1283feed (aio_req grp, SV *callback=&PL_sv_undef)
1108 CODE: 1284 CODE:
1109{ 1285{
1110 SvREFCNT_dec (grp->sv2); 1286 SvREFCNT_dec (grp->sv2);
1111 grp->sv2 = newSVsv (callback); 1287 grp->sv2 = newSVsv (callback);
1288 grp->feed = aio_grp_feed;
1112 1289
1113 if (grp->int2 <= 0) 1290 if (grp->int2 <= 0)
1114 grp->int2 = 2; 1291 grp->int2 = 2;
1115 1292
1116 eio_grp_limit (grp, grp->int2); 1293 eio_grp_limit (grp, grp->int2);

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines