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.160 by root, Thu Jan 7 18:25:51 2010 UTC vs.
Revision 1.171 by root, Wed Aug 4 17:16:19 2010 UTC

15#include <sys/stat.h> 15#include <sys/stat.h>
16#include <sys/statvfs.h> 16#include <sys/statvfs.h>
17#include <limits.h> 17#include <limits.h>
18#include <fcntl.h> 18#include <fcntl.h>
19#include <sched.h> 19#include <sched.h>
20
21#if _POSIX_MEMLOCK || _POSIX_MAPPED_FILES
22# include <sys/mman.h>
23#endif
20 24
21/* perl namespace pollution */ 25/* perl namespace pollution */
22#undef VERSION 26#undef VERSION
23 27
24#ifdef _WIN32 28#ifdef _WIN32
120 124
121#ifndef POSIX_FADV_NORMAL 125#ifndef POSIX_FADV_NORMAL
122# define POSIX_FADV_NORMAL 0 126# define POSIX_FADV_NORMAL 0
123# define NO_FADVISE 1 127# define NO_FADVISE 1
124#endif 128#endif
125
126#ifndef POSIX_FADV_SEQUENTIAL 129#ifndef POSIX_FADV_SEQUENTIAL
127# define POSIX_FADV_SEQUENTIAL 0 130# define POSIX_FADV_SEQUENTIAL 0
128#endif 131#endif
129
130#ifndef POSIX_FADV_RANDOM 132#ifndef POSIX_FADV_RANDOM
131# define POSIX_FADV_RANDOM 0 133# define POSIX_FADV_RANDOM 0
132#endif 134#endif
133
134#ifndef POSIX_FADV_NOREUSE 135#ifndef POSIX_FADV_NOREUSE
135# define POSIX_FADV_NOREUSE 0 136# define POSIX_FADV_NOREUSE 0
136#endif 137#endif
137
138#ifndef POSIX_FADV_WILLNEED 138#ifndef POSIX_FADV_WILLNEED
139# define POSIX_FADV_WILLNEED 0 139# define POSIX_FADV_WILLNEED 0
140#endif 140#endif
141
142#ifndef POSIX_FADV_DONTNEED 141#ifndef POSIX_FADV_DONTNEED
143# define POSIX_FADV_DONTNEED 0 142# define POSIX_FADV_DONTNEED 0
144#endif 143#endif
145 144
145#if _XOPEN_SOURCE < 600 || NO_FADVISE
146# define posix_fadvise(a,b,c,d) errno = ENOSYS /* also return ENOSYS */
147#endif
148
149#ifndef POSIX_MADV_NORMAL
150# define POSIX_MADV_NORMAL 0
151# define NO_MADVISE 1
152#endif
153#ifndef POSIX_MADV_SEQUENTIAL
154# define POSIX_MADV_SEQUENTIAL 0
155#endif
156#ifndef POSIX_MADV_RANDOM
157# define POSIX_MADV_RANDOM 0
158#endif
159#ifndef POSIX_MADV_WILLNEED
160# define POSIX_MADV_WILLNEED 0
161#endif
162#ifndef POSIX_MADV_DONTNEED
163# define POSIX_MADV_DONTNEED 0
164#endif
165
166#if _XOPEN_SOURCE < 600 || NO_MADVISE
167# define posix_madvise(a,b,c) errno = ENOSYS /* also return ENOSYS */
168#endif
169
170#ifndef PROT_NONE
171# define PROT_NONE 0
172#endif
173#ifndef PROT_READ
174# define PROT_READ 0
175#endif
176#ifndef PROT_WRITE
177# define PROT_READ 0
178#endif
179#ifndef PROT_EXEC
180# define PROT_EXEC 0
181#endif
182
146#ifndef ST_NODEV 183#ifndef ST_NODEV
147# define ST_NODEV 0 184# define ST_NODEV 0
148#endif 185#endif
149
150#ifndef ST_NOEXEC 186#ifndef ST_NOEXEC
151# define ST_NOEXEC 0 187# define ST_NOEXEC 0
152#endif 188#endif
153
154#ifndef ST_SYNCHRONOUS 189#ifndef ST_SYNCHRONOUS
155# define ST_SYNCHRONOUS 0 190# define ST_SYNCHRONOUS 0
156#endif 191#endif
157
158#ifndef ST_MANDLOCK 192#ifndef ST_MANDLOCK
159# define ST_MANDLOCK 0 193# define ST_MANDLOCK 0
160#endif 194#endif
161
162#ifndef ST_WRITE 195#ifndef ST_WRITE
163# define ST_WRITE 0 196# define ST_WRITE 0
164#endif 197#endif
165
166#ifndef ST_APPEND 198#ifndef ST_APPEND
167# define ST_APPEND 0 199# define ST_APPEND 0
168#endif 200#endif
169
170#ifndef ST_IMMUTABLE 201#ifndef ST_IMMUTABLE
171# define ST_IMMUTABLE 0 202# define ST_IMMUTABLE 0
172#endif 203#endif
173
174#ifndef ST_NOATIME 204#ifndef ST_NOATIME
175# define ST_NOATIME 0 205# define ST_NOATIME 0
176#endif 206#endif
177
178#ifndef ST_NODIRATIME 207#ifndef ST_NODIRATIME
179# define ST_NODIRATIME 0 208# define ST_NODIRATIME 0
180#endif 209#endif
181
182#ifndef ST_RELATIME 210#ifndef ST_RELATIME
183# define ST_RELATIME 0 211# define ST_RELATIME 0
212#endif
213
214#ifndef MCL_CURRENT
215# define MCL_CURRENT 0
216#endif
217#ifndef MCL_FUTURE
218# define MCL_FUTURE 0
219#endif
220
221#ifndef MAP_ANONYMOUS
222# ifdef MAP_ANON
223# define MAP_ANONYMOUS MAP_ANON
224# else
225# define MAP_ANONYMOUS MAP_FIXED /* and hope this fails */
226# endif
227#endif
228#ifndef MAP_HUGETLB
229# define MAP_HUGETLB 0
230#endif
231#ifndef MAP_LOCKED
232# define MAP_LOCKED 0
233#endif
234#ifndef MAP_NORESERVE
235# define MAP_NORESERVE 0
236#endif
237#ifndef MAP_POPULATE
238# define MAP_POPULATE 0
239#endif
240#ifndef MAP_NONBLOCK
241# define MAP_NONBLOCK 0
242#endif
243
244#ifndef PAGESIZE
245# define PAGESIZE sysconf (_SC_PAGESIZE)
184#endif 246#endif
185 247
186static int req_invoke (eio_req *req); 248static int req_invoke (eio_req *req);
187#define EIO_FINISH(req) req_invoke (req) 249#define EIO_FINISH(req) req_invoke (req)
188static void req_destroy (eio_req *grp); 250static void req_destroy (eio_req *grp);
280 } 342 }
281} 343}
282 344
283static int req_invoke (eio_req *req) 345static int req_invoke (eio_req *req)
284{ 346{
285 dSP;
286
287 if (req->flags & FLAG_SV2_RO_OFF) 347 if (req->flags & FLAG_SV2_RO_OFF)
288 SvREADONLY_off (req->sv2); 348 SvREADONLY_off (req->sv2);
289 349
290 if (!EIO_CANCELLED (req) && req->callback) 350 if (!EIO_CANCELLED (req) && req->callback)
291 { 351 {
352 dSP;
353 static SV *sv_result_cache; /* caches the result integer SV */
354 SV *sv_result;
355
292 ENTER; 356 ENTER;
293 SAVETMPS; 357 SAVETMPS;
294 PUSHMARK (SP); 358 PUSHMARK (SP);
295 EXTEND (SP, 1); 359 EXTEND (SP, 1);
360
361 /* do not recreate the result IV from scratch each time */
362 if (expect_true (sv_result_cache))
363 {
364 sv_result = sv_result_cache; sv_result_cache = 0;
365 SvIV_set (sv_result, req->result);
366 }
367 else
368 {
369 sv_result = newSViv (req->result);
370 SvREADONLY_on (sv_result);
371 }
296 372
297 switch (req->type) 373 switch (req->type)
298 { 374 {
299 case EIO_READDIR: 375 case EIO_READDIR:
300 { 376 {
445 case EIO_LSTAT: 521 case EIO_LSTAT:
446 case EIO_FSTAT: 522 case EIO_FSTAT:
447 PL_laststype = req->type == EIO_LSTAT ? OP_LSTAT : OP_STAT; 523 PL_laststype = req->type == EIO_LSTAT ? OP_LSTAT : OP_STAT;
448 PL_laststatval = req->result; 524 PL_laststatval = req->result;
449 PL_statcache = *(EIO_STRUCT_STAT *)(req->ptr2); 525 PL_statcache = *(EIO_STRUCT_STAT *)(req->ptr2);
450 PUSHs (sv_2mortal (newSViv (req->result))); 526 PUSHs (sv_result);
451 break; 527 break;
452 528
453 case EIO_READ: 529 case EIO_READ:
454 { 530 {
455 SvCUR_set (req->sv2, req->stroffset + (req->result > 0 ? req->result : 0)); 531 SvCUR_set (req->sv2, req->stroffset + (req->result > 0 ? req->result : 0));
456 *SvEND (req->sv2) = 0; 532 *SvEND (req->sv2) = 0;
457 SvPOK_only (req->sv2); 533 SvPOK_only (req->sv2);
458 SvSETMAGIC (req->sv2); 534 SvSETMAGIC (req->sv2);
459 PUSHs (sv_2mortal (newSViv (req->result))); 535 PUSHs (sv_result);
460 } 536 }
461 break; 537 break;
462 538
463 case EIO_DUP2: 539 case EIO_DUP2: /* EIO_DUP2 actually means aio_close(), su fudge result value */
464 if (req->result > 0) 540 if (req->result > 0)
465 req->result = 0; 541 SvIV_set (sv_result, 0);
466 /* FALLTHROUGH */ 542 /* FALLTHROUGH */
467 543
468 default: 544 default:
469 PUSHs (sv_2mortal (newSViv (req->result))); 545 PUSHs (sv_result);
470 break; 546 break;
471 } 547 }
472 548
473 errno = req->errorno; 549 errno = req->errorno;
474 550
475 PUTBACK; 551 PUTBACK;
476 call_sv (req->callback, G_VOID | G_EVAL | G_DISCARD); 552 call_sv (req->callback, G_VOID | G_EVAL | G_DISCARD);
477 SPAGAIN; 553 SPAGAIN;
554
555 if (expect_false (SvREFCNT (sv_result) != 1 || sv_result_cache))
556 SvREFCNT_dec (sv_result);
557 else
558 sv_result_cache = sv_result;
478 559
479 FREETMPS; 560 FREETMPS;
480 LEAVE; 561 LEAVE;
481 562
482 PUTBACK; 563 PUTBACK;
557 638
558static void atfork_child (void) 639static void atfork_child (void)
559{ 640{
560 create_respipe (); 641 create_respipe ();
561} 642}
643
644/*****************************************************************************/
645
646#if !_POSIX_MAPPED_FILES
647# define mmap(addr,length,prot,flags,fd,offs) (errno = ENOSYS, -1)
648# define munmap(addr,length) (errno = ENOSYS, -1)
649# define mprotect(addr,len,prot) (errno = ENOSYS, -1)
650#endif
651
652#define MMAP_MAGIC PERL_MAGIC_ext
653
654static int
655mmap_free (pTHX_ SV *sv, MAGIC *mg)
656{
657 int old_errno = errno;
658 munmap (mg->mg_ptr, (size_t)mg->mg_obj);
659 errno = old_errno;
660
661 mg->mg_obj = 0; /* just in case */
662
663 SvREADONLY_off (sv);
664
665 if (SvPVX (sv) != mg->mg_ptr)
666 croak ("ERROR: IO::AIO::mmap-mapped scalar changed location, detected");
667
668 SvCUR_set (sv, 0);
669 SvPVX (sv) = 0;
670 SvOK_off (sv);
671
672 return 0;
673}
674
675static MGVTBL mmap_vtbl = {
676 0, 0, 0, 0, mmap_free
677};
678
679/*****************************************************************************/
562 680
563static SV * 681static SV *
564get_cb (SV *cb_sv) 682get_cb (SV *cb_sv)
565{ 683{
566 SvGETMAGIC (cb_sv); 684 SvGETMAGIC (cb_sv);
598{ 716{
599 static const struct { 717 static const struct {
600 const char *name; 718 const char *name;
601 IV iv; 719 IV iv;
602 } *civ, const_iv[] = { 720 } *civ, const_iv[] = {
603# define const_iv(name, value) { # name, (IV) value }, 721# define const_niv(name, value) { # name, (IV) value },
722# define const_iv(name) { # name, (IV) name },
604# define const_eio(name) { # name, (IV) EIO_ ## name }, 723# define const_eio(name) { # name, (IV) EIO_ ## name },
605 const_iv (EXDEV , EXDEV) 724 const_iv (EXDEV)
606 const_iv (ENOSYS , ENOSYS) 725 const_iv (ENOSYS)
607 const_iv (O_RDONLY, O_RDONLY) 726 const_iv (O_RDONLY)
608 const_iv (O_WRONLY, O_WRONLY) 727 const_iv (O_WRONLY)
728 const_iv (O_RDWR)
609 const_iv (O_CREAT , O_CREAT) 729 const_iv (O_CREAT)
610 const_iv (O_TRUNC , O_TRUNC) 730 const_iv (O_TRUNC)
731 const_iv (O_EXCL)
732 const_iv (O_APPEND)
611#ifndef _WIN32 733#ifndef _WIN32
612 const_iv (S_IFIFO , S_IFIFO) 734 const_iv (S_IFIFO)
613#endif 735#endif
614 const_iv (FADV_NORMAL , POSIX_FADV_NORMAL) 736 const_niv (FADV_NORMAL , POSIX_FADV_NORMAL)
615 const_iv (FADV_SEQUENTIAL, POSIX_FADV_SEQUENTIAL) 737 const_niv (FADV_SEQUENTIAL, POSIX_FADV_SEQUENTIAL)
616 const_iv (FADV_RANDOM , POSIX_FADV_RANDOM) 738 const_niv (FADV_RANDOM , POSIX_FADV_RANDOM)
617 const_iv (FADV_NOREUSE , POSIX_FADV_NOREUSE) 739 const_niv (FADV_NOREUSE , POSIX_FADV_NOREUSE)
618 const_iv (FADV_WILLNEED , POSIX_FADV_WILLNEED) 740 const_niv (FADV_WILLNEED , POSIX_FADV_WILLNEED)
619 const_iv (FADV_DONTNEED , POSIX_FADV_DONTNEED) 741 const_niv (FADV_DONTNEED , POSIX_FADV_DONTNEED)
620 742
621 const_iv (ST_RDONLY , ST_RDONLY) 743 const_niv (MADV_NORMAL , POSIX_MADV_NORMAL)
622 const_iv (ST_NOSUID , ST_NOSUID) 744 const_niv (MADV_SEQUENTIAL, POSIX_MADV_SEQUENTIAL)
623 const_iv (ST_NODEV , ST_NODEV) 745 const_niv (MADV_RANDOM , POSIX_MADV_RANDOM)
624 const_iv (ST_NOEXEC , ST_NOEXEC) 746 const_niv (MADV_WILLNEED , POSIX_MADV_WILLNEED)
747 const_niv (MADV_DONTNEED , POSIX_MADV_DONTNEED)
748
749 const_iv (ST_RDONLY)
750 const_iv (ST_NOSUID)
751 const_iv (ST_NODEV)
752 const_iv (ST_NOEXEC)
625 const_iv (ST_SYNCHRONOUS , ST_SYNCHRONOUS) 753 const_iv (ST_SYNCHRONOUS)
626 const_iv (ST_MANDLOCK , ST_MANDLOCK) 754 const_iv (ST_MANDLOCK)
627 const_iv (ST_WRITE , ST_WRITE) 755 const_iv (ST_WRITE)
628 const_iv (ST_APPEND , ST_APPEND) 756 const_iv (ST_APPEND)
629 const_iv (ST_IMMUTABLE , ST_IMMUTABLE) 757 const_iv (ST_IMMUTABLE)
630 const_iv (ST_NOATIME , ST_NOATIME) 758 const_iv (ST_NOATIME)
631 const_iv (ST_NODIRATIME , ST_NODIRATIME) 759 const_iv (ST_NODIRATIME)
632 const_iv (ST_RELATIME , ST_RELATIME) 760 const_iv (ST_RELATIME)
761
762 const_iv (PROT_NONE)
763 const_iv (PROT_EXEC)
764 const_iv (PROT_READ)
765 const_iv (PROT_WRITE)
766
767 /*const_iv (MAP_FIXED)*/
768 const_iv (MAP_PRIVATE)
769 const_iv (MAP_SHARED)
770 const_iv (MAP_ANONYMOUS)
771
772 /* linuxish */
773 const_iv (MAP_HUGETLB)
774 const_iv (MAP_LOCKED)
775 const_iv (MAP_NORESERVE)
776 const_iv (MAP_POPULATE)
777 const_iv (MAP_NONBLOCK)
778
779 const_iv (MCL_FUTURE)
780 const_iv (MCL_CURRENT)
633 781
634 const_eio (MS_ASYNC) 782 const_eio (MS_ASYNC)
635 const_eio (MS_INVALIDATE) 783 const_eio (MS_INVALIDATE)
636 const_eio (MS_SYNC) 784 const_eio (MS_SYNC)
637 785
661 aio_req_stash = gv_stashpv ("IO::AIO::REQ", 1); 809 aio_req_stash = gv_stashpv ("IO::AIO::REQ", 1);
662 aio_grp_stash = gv_stashpv ("IO::AIO::GRP", 1); 810 aio_grp_stash = gv_stashpv ("IO::AIO::GRP", 1);
663 811
664 for (civ = const_iv + sizeof (const_iv) / sizeof (const_iv [0]); civ-- > const_iv; ) 812 for (civ = const_iv + sizeof (const_iv) / sizeof (const_iv [0]); civ-- > const_iv; )
665 newCONSTSUB (aio_stash, (char *)civ->name, newSViv (civ->iv)); 813 newCONSTSUB (aio_stash, (char *)civ->name, newSViv (civ->iv));
814
815 newCONSTSUB (aio_stash, "PAGESIZE", newSViv (PAGESIZE));
666 816
667 create_respipe (); 817 create_respipe ();
668 818
669 if (eio_init (want_poll, done_poll) < 0) 819 if (eio_init (want_poll, done_poll) < 0)
670 croak ("IO::AIO: unable to initialise eio library"); 820 croak ("IO::AIO: unable to initialise eio library");
1257 1407
1258int 1408int
1259fadvise (aio_rfd fh, off_t offset, off_t length, IV advice) 1409fadvise (aio_rfd fh, off_t offset, off_t length, IV advice)
1260 PROTOTYPE: $$$$ 1410 PROTOTYPE: $$$$
1261 CODE: 1411 CODE:
1262#if _XOPEN_SOURCE >= 600 && !NO_FADVISE
1263 RETVAL = posix_fadvise (fh, offset, length, advice); 1412 RETVAL = posix_fadvise (fh, offset, length, advice);
1264#else
1265 RETVAL = errno = ENOSYS;
1266#endif
1267 OUTPUT: 1413 OUTPUT:
1268 RETVAL 1414 RETVAL
1269 1415
1270ssize_t 1416ssize_t
1271sendfile (aio_wfd ofh, aio_rfd ifh, off_t offset, size_t count) 1417sendfile (aio_wfd ofh, aio_rfd ifh, off_t offset, size_t count)
1273 CODE: 1419 CODE:
1274 RETVAL = eio_sendfile_sync (ofh, ifh, offset, count); 1420 RETVAL = eio_sendfile_sync (ofh, ifh, offset, count);
1275 OUTPUT: 1421 OUTPUT:
1276 RETVAL 1422 RETVAL
1277 1423
1424void
1425mmap (SV *scalar, size_t length, int prot, int flags, SV *fh, off_t offset = 0)
1426 PROTOTYPE: $$$$$;$
1427 PPCODE:
1428 sv_unmagic (scalar, MMAP_MAGIC);
1429{
1430 int fd = SvOK (fh) ? s_fileno_croak (fh, flags & PROT_WRITE) : -1;
1431 void *addr = (void *)mmap (0, length, prot, flags, fd, offset);
1432 if (addr == (void *)-1)
1433 XSRETURN_NO;
1434
1435 sv_force_normal (scalar);
1436
1437 /* we store the length in mg_obj, as namlen is I32 :/ */
1438 sv_magicext (scalar, 0, MMAP_MAGIC, &mmap_vtbl, (char *)addr, 0)
1439 ->mg_obj = (SV *)length;
1440
1441 SvUPGRADE (scalar, SVt_PV); /* nop... */
1442
1443 if (!(prot & PROT_WRITE))
1444 SvREADONLY_on (scalar);
1445
1446 if (SvLEN (scalar))
1447 Safefree (SvPVX (scalar));
1448
1449 SvPVX (scalar) = (char *)addr;
1450 SvCUR_set (scalar, length);
1451 SvLEN_set (scalar, 0);
1452 SvPOK_only (scalar);
1453
1454 XSRETURN_YES;
1455}
1456
1457void
1458munmap (SV *scalar)
1459 PROTOTYPE: $
1460 CODE:
1461 sv_unmagic (scalar, MMAP_MAGIC);
1462
1463int
1464madvise (SV *scalar, off_t offset, off_t length, IV advice_or_prot)
1465 ALIAS:
1466 mprotect = 1
1467 PROTOTYPE: $$$$
1468 CODE:
1469{
1470 STRLEN cur;
1471 char *addr = SvPVbyte (scalar, cur);
1472
1473 if (offset > cur)
1474 RETVAL = errno = EFAULT;
1475 else
1476 {
1477 if (!SvOK (ST (2)))
1478 length = cur - offset;
1479
1480 if (offset + length > cur)
1481 RETVAL = errno = EFAULT;
1482 else
1483 switch (ix)
1484 {
1485 case 0: RETVAL = posix_madvise (addr + offset, length, advice_or_prot); break;
1486 case 1: RETVAL = mprotect (addr + offset, length, advice_or_prot); break;
1487 }
1488 }
1489}
1490 OUTPUT:
1491 RETVAL
1492
1493int
1494mlockall (int flags)
1495 PROTOTYPE: $
1496 CODE:
1497#if _POSIX_MEMLOCK
1498#if __GLIBC__ == 2 && __GLIBC_MINOR__ <= 7
1499 extern int mallopt (int, int);
1500 mallopt (-6, 238); /* http://bugs.debian.org/cgi-bin/bugreport.cgi?bug=473812 */
1501#endif
1502 mlockall (flags);
1503#else
1504 RETVAL = -1;
1505 errno = ENOSYS;
1506#endif
1507 OUTPUT:
1508 RETVAL
1509
1510int
1511munlockall ()
1512 PROTOTYPE:
1513 CODE:
1514#if _POSIX_MEMLOCK
1515 munlockall ();
1516#else
1517 RETVAL = -1;
1518 errno = ENOSYS;
1519#endif
1520 OUTPUT:
1521 RETVAL
1522
1278void _on_next_submit (SV *cb) 1523void _on_next_submit (SV *cb)
1279 CODE: 1524 CODE:
1280 SvREFCNT_dec (on_next_submit); 1525 SvREFCNT_dec (on_next_submit);
1281 on_next_submit = SvOK (cb) ? newSVsv (cb) : 0; 1526 on_next_submit = SvOK (cb) ? newSVsv (cb) : 0;
1282 1527

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines