1 | /* |
1 | /* |
2 | * libeio implementation |
2 | * libeio implementation |
3 | * |
3 | * |
4 | * Copyright (c) 2007,2008 Marc Alexander Lehmann <libeio@schmorp.de> |
4 | * Copyright (c) 2007,2008,2009 Marc Alexander Lehmann <libeio@schmorp.de> |
5 | * All rights reserved. |
5 | * All rights reserved. |
6 | * |
6 | * |
7 | * Redistribution and use in source and binary forms, with or without modifica- |
7 | * Redistribution and use in source and binary forms, with or without modifica- |
8 | * tion, are permitted provided that the following conditions are met: |
8 | * tion, are permitted provided that the following conditions are met: |
9 | * |
9 | * |
… | |
… | |
36 | * provisions above, a recipient may use your version of this file under |
36 | * provisions above, a recipient may use your version of this file under |
37 | * either the BSD or the GPL. |
37 | * either the BSD or the GPL. |
38 | */ |
38 | */ |
39 | |
39 | |
40 | #include "eio.h" |
40 | #include "eio.h" |
|
|
41 | |
|
|
42 | #ifdef EIO_STACKSIZE |
|
|
43 | # define XTHREAD_STACKSIZE EIO_STACKSIZE |
|
|
44 | #endif |
41 | #include "xthread.h" |
45 | #include "xthread.h" |
42 | |
46 | |
43 | #include <errno.h> |
47 | #include <errno.h> |
44 | #include <stddef.h> |
48 | #include <stddef.h> |
45 | #include <stdlib.h> |
49 | #include <stdlib.h> |
… | |
… | |
64 | #endif |
68 | #endif |
65 | |
69 | |
66 | #ifdef _WIN32 |
70 | #ifdef _WIN32 |
67 | |
71 | |
68 | /*doh*/ |
72 | /*doh*/ |
69 | |
|
|
70 | #else |
73 | #else |
71 | |
74 | |
72 | # include "config.h" |
75 | # include "config.h" |
73 | # include <sys/time.h> |
76 | # include <sys/time.h> |
74 | # include <sys/select.h> |
77 | # include <sys/select.h> |
75 | # include <unistd.h> |
78 | # include <unistd.h> |
76 | # include <utime.h> |
79 | # include <utime.h> |
77 | # include <signal.h> |
80 | # include <signal.h> |
78 | # include <dirent.h> |
81 | # include <dirent.h> |
|
|
82 | |
|
|
83 | /* POSIX_SOURCE is useless on bsd's, and XOPEN_SOURCE is unreliable there, too */ |
|
|
84 | # if defined(__FreeBSD__) || defined(__NetBSD__) || defined(__OpenBSD__) |
|
|
85 | # define D_INO(de) (de)->d_fileno |
|
|
86 | # define _DIRENT_HAVE_D_TYPE /* sigh */ |
|
|
87 | # elif defined(__linux) || defined(d_ino) || _XOPEN_SOURCE >= 600 |
|
|
88 | # define D_INO(de) (de)->d_ino |
|
|
89 | # endif |
|
|
90 | |
|
|
91 | # ifdef _DIRENT_HAVE_D_TYPE |
|
|
92 | # define D_TYPE(de) (de)->d_type |
|
|
93 | # endif |
79 | |
94 | |
80 | # ifndef EIO_STRUCT_DIRENT |
95 | # ifndef EIO_STRUCT_DIRENT |
81 | # define EIO_STRUCT_DIRENT struct dirent |
96 | # define EIO_STRUCT_DIRENT struct dirent |
82 | # endif |
97 | # endif |
83 | |
98 | |
… | |
… | |
96 | # else |
111 | # else |
97 | # error sendfile support requested but not available |
112 | # error sendfile support requested but not available |
98 | # endif |
113 | # endif |
99 | #endif |
114 | #endif |
100 | |
115 | |
|
|
116 | #ifndef D_TYPE |
|
|
117 | # define D_TYPE(de) 0 |
|
|
118 | #endif |
|
|
119 | |
|
|
120 | #ifndef D_INO |
|
|
121 | # define D_INO(de) 0 |
|
|
122 | #endif |
|
|
123 | |
101 | /* number of seconds after which an idle threads exit */ |
124 | /* number of seconds after which an idle threads exit */ |
102 | #define IDLE_TIMEOUT 10 |
125 | #define IDLE_TIMEOUT 10 |
103 | |
126 | |
104 | /* used for struct dirent, AIX doesn't provide it */ |
127 | /* used for struct dirent, AIX doesn't provide it */ |
105 | #ifndef NAME_MAX |
128 | #ifndef NAME_MAX |
… | |
… | |
117 | errno = ENOMEM; \ |
140 | errno = ENOMEM; \ |
118 | if (!eio_buf) \ |
141 | if (!eio_buf) \ |
119 | return -1; |
142 | return -1; |
120 | |
143 | |
121 | #define EIO_TICKS ((1000000 + 1023) >> 10) |
144 | #define EIO_TICKS ((1000000 + 1023) >> 10) |
|
|
145 | |
|
|
146 | /*****************************************************************************/ |
|
|
147 | |
|
|
148 | #if __GNUC__ >= 3 |
|
|
149 | # define expect(expr,value) __builtin_expect ((expr),(value)) |
|
|
150 | #else |
|
|
151 | # define expect(expr,value) (expr) |
|
|
152 | #endif |
|
|
153 | |
|
|
154 | #define expect_false(expr) expect ((expr) != 0, 0) |
|
|
155 | #define expect_true(expr) expect ((expr) != 0, 1) |
122 | |
156 | |
123 | /*****************************************************************************/ |
157 | /*****************************************************************************/ |
124 | |
158 | |
125 | #define ETP_PRI_MIN EIO_PRI_MIN |
159 | #define ETP_PRI_MIN EIO_PRI_MIN |
126 | #define ETP_PRI_MAX EIO_PRI_MAX |
160 | #define ETP_PRI_MAX EIO_PRI_MAX |
… | |
… | |
144 | if (wrk->dirp) \ |
178 | if (wrk->dirp) \ |
145 | { \ |
179 | { \ |
146 | closedir (wrk->dirp); \ |
180 | closedir (wrk->dirp); \ |
147 | wrk->dirp = 0; \ |
181 | wrk->dirp = 0; \ |
148 | } |
182 | } |
|
|
183 | |
149 | #define ETP_WORKER_COMMON \ |
184 | #define ETP_WORKER_COMMON \ |
150 | void *dbuf; \ |
185 | void *dbuf; \ |
151 | DIR *dirp; |
186 | DIR *dirp; |
152 | |
187 | |
153 | /*****************************************************************************/ |
188 | /*****************************************************************************/ |
… | |
… | |
176 | |
211 | |
177 | static mutex_t wrklock = X_MUTEX_INIT; |
212 | static mutex_t wrklock = X_MUTEX_INIT; |
178 | static mutex_t reslock = X_MUTEX_INIT; |
213 | static mutex_t reslock = X_MUTEX_INIT; |
179 | static mutex_t reqlock = X_MUTEX_INIT; |
214 | static mutex_t reqlock = X_MUTEX_INIT; |
180 | static cond_t reqwait = X_COND_INIT; |
215 | static cond_t reqwait = X_COND_INIT; |
|
|
216 | |
|
|
217 | #if !HAVE_PREADWRITE |
|
|
218 | /* |
|
|
219 | * make our pread/pwrite emulation safe against themselves, but not against |
|
|
220 | * normal read/write by using a mutex. slows down execution a lot, |
|
|
221 | * but that's your problem, not mine. |
|
|
222 | */ |
|
|
223 | static mutex_t preadwritelock = X_MUTEX_INIT; |
|
|
224 | #endif |
181 | |
225 | |
182 | typedef struct etp_worker |
226 | typedef struct etp_worker |
183 | { |
227 | { |
184 | /* locked by wrklock */ |
228 | /* locked by wrklock */ |
185 | struct etp_worker *prev, *next; |
229 | struct etp_worker *prev, *next; |
… | |
… | |
330 | |
374 | |
331 | static void etp_atfork_child (void) |
375 | static void etp_atfork_child (void) |
332 | { |
376 | { |
333 | ETP_REQ *prv; |
377 | ETP_REQ *prv; |
334 | |
378 | |
335 | while (prv = reqq_shift (&req_queue)) |
379 | while ((prv = reqq_shift (&req_queue))) |
336 | ETP_DESTROY (prv); |
380 | ETP_DESTROY (prv); |
337 | |
381 | |
338 | while (prv = reqq_shift (&res_queue)) |
382 | while ((prv = reqq_shift (&res_queue))) |
339 | ETP_DESTROY (prv); |
383 | ETP_DESTROY (prv); |
340 | |
384 | |
341 | while (wrk_first.next != &wrk_first) |
385 | while (wrk_first.next != &wrk_first) |
342 | { |
386 | { |
343 | etp_worker *wrk = wrk_first.next; |
387 | etp_worker *wrk = wrk_first.next; |
… | |
… | |
402 | X_UNLOCK (wrklock); |
446 | X_UNLOCK (wrklock); |
403 | } |
447 | } |
404 | |
448 | |
405 | static void etp_maybe_start_thread (void) |
449 | static void etp_maybe_start_thread (void) |
406 | { |
450 | { |
407 | if (etp_nthreads () >= wanted) |
451 | if (expect_true (etp_nthreads () >= wanted)) |
408 | return; |
452 | return; |
409 | |
453 | |
410 | /* todo: maybe use idle here, but might be less exact */ |
454 | /* todo: maybe use idle here, but might be less exact */ |
411 | if (0 <= (int)etp_nthreads () + (int)etp_npending () - (int)etp_nreqs ()) |
455 | if (expect_true (0 <= (int)etp_nthreads () + (int)etp_npending () - (int)etp_nreqs ())) |
412 | return; |
456 | return; |
413 | |
457 | |
414 | etp_start_thread (); |
458 | etp_start_thread (); |
415 | } |
459 | } |
416 | |
460 | |
… | |
… | |
469 | |
513 | |
470 | X_LOCK (reqlock); |
514 | X_LOCK (reqlock); |
471 | --nreqs; |
515 | --nreqs; |
472 | X_UNLOCK (reqlock); |
516 | X_UNLOCK (reqlock); |
473 | |
517 | |
474 | if (req->type == EIO_GROUP && req->size) |
518 | if (expect_false (req->type == EIO_GROUP && req->size)) |
475 | { |
519 | { |
476 | req->int1 = 1; /* mark request as delayed */ |
520 | req->int1 = 1; /* mark request as delayed */ |
477 | continue; |
521 | continue; |
478 | } |
522 | } |
479 | else |
523 | else |
480 | { |
524 | { |
481 | int res = ETP_FINISH (req); |
525 | int res = ETP_FINISH (req); |
482 | if (res) |
526 | if (expect_false (res)) |
483 | return res; |
527 | return res; |
484 | } |
528 | } |
485 | |
529 | |
486 | if (maxreqs && !--maxreqs) |
530 | if (expect_false (maxreqs && !--maxreqs)) |
487 | break; |
531 | break; |
488 | |
532 | |
489 | if (maxtime) |
533 | if (maxtime) |
490 | { |
534 | { |
491 | gettimeofday (&tv_now, 0); |
535 | gettimeofday (&tv_now, 0); |
… | |
… | |
510 | |
554 | |
511 | static void etp_submit (ETP_REQ *req) |
555 | static void etp_submit (ETP_REQ *req) |
512 | { |
556 | { |
513 | req->pri -= ETP_PRI_MIN; |
557 | req->pri -= ETP_PRI_MIN; |
514 | |
558 | |
515 | if (req->pri < ETP_PRI_MIN - ETP_PRI_MIN) req->pri = ETP_PRI_MIN - ETP_PRI_MIN; |
559 | if (expect_false (req->pri < ETP_PRI_MIN - ETP_PRI_MIN)) req->pri = ETP_PRI_MIN - ETP_PRI_MIN; |
516 | if (req->pri > ETP_PRI_MAX - ETP_PRI_MIN) req->pri = ETP_PRI_MAX - ETP_PRI_MIN; |
560 | if (expect_false (req->pri > ETP_PRI_MAX - ETP_PRI_MIN)) req->pri = ETP_PRI_MAX - ETP_PRI_MIN; |
517 | |
561 | |
|
|
562 | if (expect_false (req->type == EIO_GROUP)) |
|
|
563 | { |
|
|
564 | /* I hope this is worth it :/ */ |
518 | X_LOCK (reqlock); |
565 | X_LOCK (reqlock); |
519 | ++nreqs; |
566 | ++nreqs; |
|
|
567 | X_UNLOCK (reqlock); |
|
|
568 | |
|
|
569 | X_LOCK (reslock); |
|
|
570 | |
|
|
571 | ++npending; |
|
|
572 | |
|
|
573 | if (!reqq_push (&res_queue, req) && want_poll_cb) |
|
|
574 | want_poll_cb (); |
|
|
575 | |
|
|
576 | X_UNLOCK (reslock); |
|
|
577 | } |
|
|
578 | else |
|
|
579 | { |
|
|
580 | X_LOCK (reqlock); |
|
|
581 | ++nreqs; |
520 | ++nready; |
582 | ++nready; |
521 | reqq_push (&req_queue, req); |
583 | reqq_push (&req_queue, req); |
522 | X_COND_SIGNAL (reqwait); |
584 | X_COND_SIGNAL (reqwait); |
523 | X_UNLOCK (reqlock); |
585 | X_UNLOCK (reqlock); |
524 | |
586 | |
525 | etp_maybe_start_thread (); |
587 | etp_maybe_start_thread (); |
|
|
588 | } |
526 | } |
589 | } |
527 | |
590 | |
528 | static void etp_set_max_poll_time (double nseconds) |
591 | static void etp_set_max_poll_time (double nseconds) |
529 | { |
592 | { |
530 | if (WORDACCESS_UNSAFE) X_LOCK (reslock); |
593 | if (WORDACCESS_UNSAFE) X_LOCK (reslock); |
… | |
… | |
565 | |
628 | |
566 | static void grp_try_feed (eio_req *grp) |
629 | static void grp_try_feed (eio_req *grp) |
567 | { |
630 | { |
568 | while (grp->size < grp->int2 && !EIO_CANCELLED (grp)) |
631 | while (grp->size < grp->int2 && !EIO_CANCELLED (grp)) |
569 | { |
632 | { |
570 | int old_len = grp->size; |
633 | grp->flags &= ~EIO_FLAG_GROUPADD; |
571 | |
634 | |
572 | EIO_FEED (grp); |
635 | EIO_FEED (grp); |
573 | |
636 | |
574 | /* stop if no progress has been made */ |
637 | /* stop if no progress has been made */ |
575 | if (old_len == grp->size) |
638 | if (!(grp->flags & EIO_FLAG_GROUPADD)) |
576 | { |
639 | { |
577 | grp->feed = 0; |
640 | grp->feed = 0; |
578 | break; |
641 | break; |
579 | } |
642 | } |
580 | } |
643 | } |
… | |
… | |
697 | |
760 | |
698 | /*****************************************************************************/ |
761 | /*****************************************************************************/ |
699 | /* work around various missing functions */ |
762 | /* work around various missing functions */ |
700 | |
763 | |
701 | #if !HAVE_PREADWRITE |
764 | #if !HAVE_PREADWRITE |
|
|
765 | # undef pread |
|
|
766 | # undef pwrite |
702 | # define pread eio__pread |
767 | # define pread eio__pread |
703 | # define pwrite eio__pwrite |
768 | # define pwrite eio__pwrite |
704 | |
|
|
705 | /* |
|
|
706 | * make our pread/pwrite safe against themselves, but not against |
|
|
707 | * normal read/write by using a mutex. slows down execution a lot, |
|
|
708 | * but that's your problem, not mine. |
|
|
709 | */ |
|
|
710 | static mutex_t preadwritelock = X_MUTEX_INIT; |
|
|
711 | |
769 | |
712 | static ssize_t |
770 | static ssize_t |
713 | eio__pread (int fd, void *buf, size_t count, off_t offset) |
771 | eio__pread (int fd, void *buf, size_t count, off_t offset) |
714 | { |
772 | { |
715 | ssize_t res; |
773 | ssize_t res; |
… | |
… | |
733 | |
791 | |
734 | X_LOCK (preadwritelock); |
792 | X_LOCK (preadwritelock); |
735 | ooffset = lseek (fd, 0, SEEK_CUR); |
793 | ooffset = lseek (fd, 0, SEEK_CUR); |
736 | lseek (fd, offset, SEEK_SET); |
794 | lseek (fd, offset, SEEK_SET); |
737 | res = write (fd, buf, count); |
795 | res = write (fd, buf, count); |
738 | lseek (fd, offset, SEEK_SET); |
796 | lseek (fd, ooffset, SEEK_SET); |
739 | X_UNLOCK (preadwritelock); |
797 | X_UNLOCK (preadwritelock); |
740 | |
798 | |
741 | return res; |
799 | return res; |
742 | } |
800 | } |
743 | #endif |
801 | #endif |
744 | |
802 | |
745 | #ifndef HAVE_FUTIMES |
803 | #ifndef HAVE_FUTIMES |
746 | |
804 | |
|
|
805 | # undef utimes |
|
|
806 | # undef futimes |
747 | # define utimes(path,times) eio__utimes (path, times) |
807 | # define utimes(path,times) eio__utimes (path, times) |
748 | # define futimes(fd,times) eio__futimes (fd, times) |
808 | # define futimes(fd,times) eio__futimes (fd, times) |
749 | |
809 | |
750 | static int |
810 | static int |
751 | eio__utimes (const char *filename, const struct timeval times[2]) |
811 | eio__utimes (const char *filename, const struct timeval times[2]) |
… | |
… | |
770 | } |
830 | } |
771 | |
831 | |
772 | #endif |
832 | #endif |
773 | |
833 | |
774 | #if !HAVE_FDATASYNC |
834 | #if !HAVE_FDATASYNC |
|
|
835 | # undef fdatasync |
775 | # define fdatasync fsync |
836 | # define fdatasync(fd) fsync (fd) |
776 | #endif |
837 | #endif |
|
|
838 | |
|
|
839 | /* sync_file_range always needs emulation */ |
|
|
840 | int |
|
|
841 | eio__sync_file_range (int fd, off_t offset, size_t nbytes, unsigned int flags) |
|
|
842 | { |
|
|
843 | #if HAVE_SYNC_FILE_RANGE |
|
|
844 | int res; |
|
|
845 | |
|
|
846 | if (EIO_SYNC_FILE_RANGE_WAIT_BEFORE != SYNC_FILE_RANGE_WAIT_BEFORE |
|
|
847 | || EIO_SYNC_FILE_RANGE_WRITE != SYNC_FILE_RANGE_WRITE |
|
|
848 | || EIO_SYNC_FILE_RANGE_WAIT_AFTER != SYNC_FILE_RANGE_WAIT_AFTER) |
|
|
849 | { |
|
|
850 | flags = 0 |
|
|
851 | | (flags & EIO_SYNC_FILE_RANGE_WAIT_BEFORE ? SYNC_FILE_RANGE_WAIT_BEFORE : 0) |
|
|
852 | | (flags & EIO_SYNC_FILE_RANGE_WRITE ? SYNC_FILE_RANGE_WRITE : 0) |
|
|
853 | | (flags & EIO_SYNC_FILE_RANGE_WAIT_AFTER ? SYNC_FILE_RANGE_WAIT_AFTER : 0); |
|
|
854 | } |
|
|
855 | |
|
|
856 | res = sync_file_range (fd, offset, nbytes, flags); |
|
|
857 | |
|
|
858 | if (!res || errno != ENOSYS) |
|
|
859 | return res; |
|
|
860 | #endif |
|
|
861 | |
|
|
862 | /* even though we could play tricks with the flags, it's better to always |
|
|
863 | * call fdatasync, as thta matches the expectation of it's users best */ |
|
|
864 | return fdatasync (fd); |
|
|
865 | } |
777 | |
866 | |
778 | #if !HAVE_READAHEAD |
867 | #if !HAVE_READAHEAD |
|
|
868 | # undef readahead |
779 | # define readahead(fd,offset,count) eio__readahead (fd, offset, count, self) |
869 | # define readahead(fd,offset,count) eio__readahead (fd, offset, count, self) |
780 | |
870 | |
781 | static ssize_t |
871 | static ssize_t |
782 | eio__readahead (int fd, off_t offset, size_t count, etp_worker *self) |
872 | eio__readahead (int fd, off_t offset, size_t count, etp_worker *self) |
783 | { |
873 | { |
… | |
… | |
891 | } |
981 | } |
892 | |
982 | |
893 | return res; |
983 | return res; |
894 | } |
984 | } |
895 | |
985 | |
|
|
986 | static int |
|
|
987 | eio_dent_cmp (const void *a_, const void *b_) |
|
|
988 | { |
|
|
989 | const eio_dirent *a = (const eio_dirent *)a_; |
|
|
990 | const eio_dirent *b = (const eio_dirent *)b_; |
|
|
991 | |
|
|
992 | return (int)b->score - (int)a->score ? (int)b->score - (int)a->score |
|
|
993 | : a->inode < b->inode ? -1 : a->inode > b->inode ? 1 : 0; /* int might be < ino_t */ |
|
|
994 | } |
|
|
995 | |
896 | /* read a full directory */ |
996 | /* read a full directory */ |
897 | static void |
997 | static void |
898 | eio__scandir (eio_req *req, etp_worker *self) |
998 | eio__scandir (eio_req *req, etp_worker *self) |
899 | { |
999 | { |
900 | DIR *dirp; |
1000 | DIR *dirp; |
901 | EIO_STRUCT_DIRENT *entp; |
1001 | EIO_STRUCT_DIRENT *entp; |
902 | char *name, *names; |
1002 | unsigned char *name, *names; |
903 | int memlen = 4096; |
1003 | int namesalloc = 4096; |
904 | int memofs = 0; |
1004 | int namesoffs = 0; |
|
|
1005 | int flags = req->int1; |
|
|
1006 | eio_dirent *dents = 0; |
|
|
1007 | int dentalloc = 128; |
905 | int res = 0; |
1008 | int dentoffs = 0; |
|
|
1009 | |
|
|
1010 | req->result = -1; |
|
|
1011 | |
|
|
1012 | if (!(flags & EIO_READDIR_DENTS)) |
|
|
1013 | flags &= ~(EIO_READDIR_DIRS_FIRST | EIO_READDIR_STAT_ORDER); |
906 | |
1014 | |
907 | X_LOCK (wrklock); |
1015 | X_LOCK (wrklock); |
|
|
1016 | /* the corresponding closedir is in ETP_WORKER_CLEAR */ |
908 | self->dirp = dirp = opendir (req->ptr1); |
1017 | self->dirp = dirp = opendir (req->ptr1); |
909 | req->flags |= EIO_FLAG_PTR2_FREE; |
1018 | req->flags |= EIO_FLAG_PTR1_FREE | EIO_FLAG_PTR2_FREE; |
910 | req->ptr2 = names = malloc (memlen); |
1019 | req->ptr1 = names = malloc (namesalloc); |
|
|
1020 | req->ptr2 = dents = flags ? malloc (dentalloc * sizeof (eio_dirent)) : 0; |
911 | X_UNLOCK (wrklock); |
1021 | X_UNLOCK (wrklock); |
912 | |
1022 | |
913 | if (dirp && names) |
1023 | if (dirp && names && (!flags || dents)) |
914 | for (;;) |
1024 | for (;;) |
915 | { |
1025 | { |
916 | errno = 0; |
1026 | errno = 0; |
917 | entp = readdir (dirp); |
1027 | entp = readdir (dirp); |
918 | |
1028 | |
919 | if (!entp) |
1029 | if (!entp) |
|
|
1030 | { |
|
|
1031 | if (errno) |
|
|
1032 | break; |
|
|
1033 | |
|
|
1034 | /* sort etc. */ |
|
|
1035 | req->int1 = flags; |
|
|
1036 | req->result = dentoffs; |
|
|
1037 | |
|
|
1038 | if (dents) |
|
|
1039 | { |
|
|
1040 | eio_dirent *ent = dents + dentoffs; |
|
|
1041 | |
|
|
1042 | while (ent > dents) |
|
|
1043 | (--ent)->name = names + (size_t)ent->name; |
|
|
1044 | } |
|
|
1045 | |
|
|
1046 | if (flags & EIO_READDIR_STAT_ORDER |
|
|
1047 | || !(~flags & (EIO_READDIR_DIRS_FIRST | EIO_READDIR_FOUND_UNKNOWN))) |
|
|
1048 | { |
|
|
1049 | /* pray your qsort doesn't use quicksort */ |
|
|
1050 | qsort (dents, dentoffs, sizeof (*dents), eio_dent_cmp); /* score depends of DIRS_FIRST */ |
|
|
1051 | } |
|
|
1052 | else if (flags & EIO_READDIR_DIRS_FIRST) |
|
|
1053 | { |
|
|
1054 | /* in this case, all is known, and we just put dirs first and sort them */ |
|
|
1055 | eio_dirent *ent = dents + dentoffs; |
|
|
1056 | eio_dirent *dir = dents; |
|
|
1057 | |
|
|
1058 | while (ent > dir) |
|
|
1059 | { |
|
|
1060 | if (dir->type == DT_DIR) |
|
|
1061 | ++dir; |
|
|
1062 | else |
|
|
1063 | { |
|
|
1064 | --ent; |
|
|
1065 | |
|
|
1066 | if (ent->type == DT_DIR) |
|
|
1067 | { |
|
|
1068 | eio_dirent tmp = *dir; |
|
|
1069 | *dir = *ent; |
|
|
1070 | *ent = tmp; |
|
|
1071 | |
|
|
1072 | ++dir; |
|
|
1073 | } |
|
|
1074 | } |
|
|
1075 | } |
|
|
1076 | |
|
|
1077 | /* now sort the dirs only */ |
|
|
1078 | qsort (dents, dir - dents, sizeof (*dents), eio_dent_cmp); |
|
|
1079 | } |
|
|
1080 | |
|
|
1081 | /* only provide the names array unless DENTS is specified */ |
|
|
1082 | if (!(flags & EIO_READDIR_DENTS)) |
|
|
1083 | { |
|
|
1084 | X_LOCK (wrklock); |
|
|
1085 | assert (!dents); |
|
|
1086 | req->ptr1 = 0; |
|
|
1087 | req->ptr2 = names; |
|
|
1088 | X_UNLOCK (wrklock); |
|
|
1089 | } |
|
|
1090 | |
920 | break; |
1091 | break; |
|
|
1092 | } |
921 | |
1093 | |
|
|
1094 | /* now add the entry to our list(s) */ |
922 | name = entp->d_name; |
1095 | name = entp->d_name; |
923 | |
1096 | |
|
|
1097 | /* skip . and .. entries */ |
924 | if (name [0] != '.' || (name [1] && (name [1] != '.' || name [2]))) |
1098 | if (name [0] != '.' || (name [1] && (name [1] != '.' || name [2]))) |
925 | { |
1099 | { |
926 | int len = strlen (name) + 1; |
1100 | int len = strlen (name) + 1; |
927 | |
1101 | |
928 | res++; |
1102 | while (expect_false (namesoffs + len > namesalloc)) |
929 | |
|
|
930 | while (memofs + len > memlen) |
|
|
931 | { |
1103 | { |
932 | memlen *= 2; |
1104 | namesalloc *= 2; |
933 | X_LOCK (wrklock); |
1105 | X_LOCK (wrklock); |
934 | req->ptr2 = names = realloc (names, memlen); |
1106 | req->ptr1 = names = realloc (names, namesalloc); |
935 | X_UNLOCK (wrklock); |
1107 | X_UNLOCK (wrklock); |
936 | |
1108 | |
937 | if (!names) |
1109 | if (!names) |
938 | break; |
1110 | break; |
939 | } |
1111 | } |
940 | |
1112 | |
941 | memcpy (names + memofs, name, len); |
1113 | memcpy (names + namesoffs, name, len); |
|
|
1114 | |
|
|
1115 | if (dents) |
|
|
1116 | { |
|
|
1117 | struct eio_dirent *ent; |
|
|
1118 | |
|
|
1119 | if (expect_false (dentoffs == dentalloc)) |
|
|
1120 | { |
|
|
1121 | dentalloc *= 2; |
|
|
1122 | X_LOCK (wrklock); |
|
|
1123 | req->ptr2 = dents = realloc (dents, dentalloc * sizeof (eio_dirent)); |
|
|
1124 | X_UNLOCK (wrklock); |
|
|
1125 | |
|
|
1126 | if (!dents) |
|
|
1127 | break; |
|
|
1128 | } |
|
|
1129 | |
|
|
1130 | ent = dents + dentoffs; |
|
|
1131 | |
|
|
1132 | ent->name = (char *)(size_t)namesoffs; /* rather dirtily we store the offset in the pointer */ |
|
|
1133 | ent->namelen = len - 1; |
|
|
1134 | ent->inode = D_INO (entp); |
|
|
1135 | |
|
|
1136 | switch (D_TYPE (entp)) |
|
|
1137 | { |
|
|
1138 | default: |
|
|
1139 | ent->type = EIO_DT_UNKNOWN; |
|
|
1140 | flags |= EIO_READDIR_FOUND_UNKNOWN; |
|
|
1141 | break; |
|
|
1142 | |
|
|
1143 | #ifdef DT_FIFO |
|
|
1144 | case DT_FIFO: ent->type = EIO_DT_FIFO; break; |
|
|
1145 | #endif |
|
|
1146 | #ifdef DT_CHR |
|
|
1147 | case DT_CHR: ent->type = EIO_DT_CHR; break; |
|
|
1148 | #endif |
|
|
1149 | #ifdef DT_DIR |
|
|
1150 | case DT_DIR: ent->type = EIO_DT_DIR; break; |
|
|
1151 | #endif |
|
|
1152 | #ifdef DT_BLK |
|
|
1153 | case DT_BLK: ent->type = EIO_DT_BLK; break; |
|
|
1154 | #endif |
|
|
1155 | #ifdef DT_REG |
|
|
1156 | case DT_REG: ent->type = EIO_DT_REG; break; |
|
|
1157 | #endif |
|
|
1158 | #ifdef DT_LNK |
|
|
1159 | case DT_LNK: ent->type = EIO_DT_LNK; break; |
|
|
1160 | #endif |
|
|
1161 | #ifdef DT_SOCK |
|
|
1162 | case DT_SOCK: ent->type = EIO_DT_SOCK; break; |
|
|
1163 | #endif |
|
|
1164 | #ifdef DT_WHT |
|
|
1165 | case DT_WHT: ent->type = EIO_DT_WHT; break; |
|
|
1166 | #endif |
|
|
1167 | } |
|
|
1168 | |
|
|
1169 | ent->score = 0; |
|
|
1170 | |
|
|
1171 | if (flags & EIO_READDIR_DIRS_FIRST) |
|
|
1172 | { |
|
|
1173 | if (ent->type == EIO_DT_UNKNOWN) |
|
|
1174 | { |
|
|
1175 | if (*name == '.') /* leading dots are likely directories, and, in any case, rare */ |
|
|
1176 | ent->score = 98; |
|
|
1177 | else if (!strchr (name, '.')) /* absense of dots indicate likely dirs */ |
|
|
1178 | ent->score = len <= 4 ? 5 : len <= 7 ? 4 : 1; /* shorter == more likely dir, but avoid too many classes */ |
|
|
1179 | } |
|
|
1180 | else if (ent->type == DT_DIR) |
|
|
1181 | ent->score = 100; |
|
|
1182 | } |
|
|
1183 | } |
|
|
1184 | |
942 | memofs += len; |
1185 | namesoffs += len; |
|
|
1186 | ++dentoffs; |
943 | } |
1187 | } |
944 | } |
1188 | } |
945 | |
1189 | else |
946 | if (errno) |
|
|
947 | res = -1; |
|
|
948 | |
|
|
949 | req->result = res; |
1190 | req->result = -1; |
|
|
1191 | } |
|
|
1192 | |
|
|
1193 | #if !(_POSIX_MAPPED_FILES && _POSIX_SYNCHRONIZED_IO) |
|
|
1194 | # undef msync |
|
|
1195 | # define msync(a,b,c) ((errno = ENOSYS), -1) |
|
|
1196 | #endif |
|
|
1197 | |
|
|
1198 | int |
|
|
1199 | eio__mtouch (void *mem, size_t len, int flags) |
|
|
1200 | { |
|
|
1201 | intptr_t addr = (intptr_t)mem; |
|
|
1202 | intptr_t end = addr + len; |
|
|
1203 | #ifdef PAGESIZE |
|
|
1204 | const intptr_t page = PAGESIZE; |
|
|
1205 | #else |
|
|
1206 | static intptr_t page; |
|
|
1207 | |
|
|
1208 | if (!page) |
|
|
1209 | page = sysconf (_SC_PAGESIZE); |
|
|
1210 | #endif |
|
|
1211 | |
|
|
1212 | addr &= ~(page - 1); /* assume page size is always a power of two */ |
|
|
1213 | |
|
|
1214 | if (addr < end) |
|
|
1215 | if (flags) /* modify */ |
|
|
1216 | do { *((volatile sig_atomic_t *)addr) |= 0; } while ((addr += page) < len); |
|
|
1217 | else |
|
|
1218 | do { *((volatile sig_atomic_t *)addr) ; } while ((addr += page) < len); |
|
|
1219 | |
|
|
1220 | return 0; |
950 | } |
1221 | } |
951 | |
1222 | |
952 | /*****************************************************************************/ |
1223 | /*****************************************************************************/ |
953 | |
1224 | |
954 | #define ALLOC(len) \ |
1225 | #define ALLOC(len) \ |
… | |
… | |
1121 | req->result = readlink (req->ptr1, req->ptr2, NAME_MAX); break; |
1392 | req->result = readlink (req->ptr1, req->ptr2, NAME_MAX); break; |
1122 | |
1393 | |
1123 | case EIO_SYNC: req->result = 0; sync (); break; |
1394 | case EIO_SYNC: req->result = 0; sync (); break; |
1124 | case EIO_FSYNC: req->result = fsync (req->int1); break; |
1395 | case EIO_FSYNC: req->result = fsync (req->int1); break; |
1125 | case EIO_FDATASYNC: req->result = fdatasync (req->int1); break; |
1396 | case EIO_FDATASYNC: req->result = fdatasync (req->int1); break; |
|
|
1397 | case EIO_MSYNC: req->result = msync (req->ptr2, req->size, req->int1); break; |
|
|
1398 | case EIO_MTOUCH: req->result = eio__mtouch (req->ptr2, req->size, req->int1); break; |
|
|
1399 | case EIO_SYNC_FILE_RANGE: req->result = eio__sync_file_range (req->int1, req->offs, req->size, req->int2); break; |
1126 | |
1400 | |
1127 | case EIO_READDIR: eio__scandir (req, self); break; |
1401 | case EIO_READDIR: eio__scandir (req, self); break; |
1128 | |
1402 | |
1129 | case EIO_BUSY: |
1403 | case EIO_BUSY: |
1130 | #ifdef _WIN32 |
1404 | #ifdef _WIN32 |
… | |
… | |
1162 | |
1436 | |
1163 | req->result = req->type == EIO_FUTIME |
1437 | req->result = req->type == EIO_FUTIME |
1164 | ? futimes (req->int1, times) |
1438 | ? futimes (req->int1, times) |
1165 | : utimes (req->ptr1, times); |
1439 | : utimes (req->ptr1, times); |
1166 | } |
1440 | } |
|
|
1441 | break; |
1167 | |
1442 | |
1168 | case EIO_GROUP: |
1443 | case EIO_GROUP: |
|
|
1444 | abort (); /* handled in eio_request */ |
|
|
1445 | |
1169 | case EIO_NOP: |
1446 | case EIO_NOP: |
1170 | req->result = 0; |
1447 | req->result = 0; |
1171 | break; |
1448 | break; |
1172 | |
1449 | |
1173 | case EIO_CUSTOM: |
1450 | case EIO_CUSTOM: |
1174 | req->feed (req); |
1451 | ((void (*)(eio_req *))req->feed) (req); |
1175 | break; |
1452 | break; |
1176 | |
1453 | |
1177 | default: |
1454 | default: |
1178 | req->result = -1; |
1455 | req->result = -1; |
1179 | break; |
1456 | break; |
… | |
… | |
1202 | eio_req *eio_fsync (int fd, int pri, eio_cb cb, void *data) |
1479 | eio_req *eio_fsync (int fd, int pri, eio_cb cb, void *data) |
1203 | { |
1480 | { |
1204 | REQ (EIO_FSYNC); req->int1 = fd; SEND; |
1481 | REQ (EIO_FSYNC); req->int1 = fd; SEND; |
1205 | } |
1482 | } |
1206 | |
1483 | |
|
|
1484 | eio_req *eio_msync (void *addr, size_t length, int flags, int pri, eio_cb cb, void *data) |
|
|
1485 | { |
|
|
1486 | REQ (EIO_MSYNC); req->ptr2 = addr; req->size = length; req->int1 = flags; SEND; |
|
|
1487 | } |
|
|
1488 | |
|
|
1489 | eio_req *eio_mtouch (void *addr, size_t length, int flags, int pri, eio_cb cb, void *data) |
|
|
1490 | { |
|
|
1491 | REQ (EIO_MTOUCH); req->ptr2 = addr; req->size = length; req->int1 = flags; SEND; |
|
|
1492 | } |
|
|
1493 | |
|
|
1494 | eio_req *eio_sync_file_range (int fd, off_t offset, size_t nbytes, unsigned int flags, int pri, eio_cb cb, void *data) |
|
|
1495 | { |
|
|
1496 | REQ (EIO_SYNC_FILE_RANGE); req->int1 = fd; req->offs = offset; req->size = nbytes; req->int2 = flags; SEND; |
|
|
1497 | } |
|
|
1498 | |
1207 | eio_req *eio_fdatasync (int fd, int pri, eio_cb cb, void *data) |
1499 | eio_req *eio_fdatasync (int fd, int pri, eio_cb cb, void *data) |
1208 | { |
1500 | { |
1209 | REQ (EIO_FDATASYNC); req->int1 = fd; SEND; |
1501 | REQ (EIO_FDATASYNC); req->int1 = fd; SEND; |
1210 | } |
1502 | } |
1211 | |
1503 | |
… | |
… | |
1323 | eio_req *eio_rmdir (const char *path, int pri, eio_cb cb, void *data) |
1615 | eio_req *eio_rmdir (const char *path, int pri, eio_cb cb, void *data) |
1324 | { |
1616 | { |
1325 | return eio__1path (EIO_RMDIR, path, pri, cb, data); |
1617 | return eio__1path (EIO_RMDIR, path, pri, cb, data); |
1326 | } |
1618 | } |
1327 | |
1619 | |
1328 | eio_req *eio_readdir (const char *path, int pri, eio_cb cb, void *data) |
1620 | eio_req *eio_readdir (const char *path, int flags, int pri, eio_cb cb, void *data) |
1329 | { |
1621 | { |
1330 | return eio__1path (EIO_READDIR, path, pri, cb, data); |
1622 | REQ (EIO_READDIR); PATH; req->int1 = flags; SEND; |
1331 | } |
1623 | } |
1332 | |
1624 | |
1333 | eio_req *eio_mknod (const char *path, mode_t mode, dev_t dev, int pri, eio_cb cb, void *data) |
1625 | eio_req *eio_mknod (const char *path, mode_t mode, dev_t dev, int pri, eio_cb cb, void *data) |
1334 | { |
1626 | { |
1335 | REQ (EIO_MKNOD); PATH; req->int2 = (long)mode; req->int3 = (long)dev; SEND; |
1627 | REQ (EIO_MKNOD); PATH; req->int2 = (long)mode; req->int3 = (long)dev; SEND; |
… | |
… | |
1366 | return eio__2path (EIO_RENAME, path, new_path, pri, cb, data); |
1658 | return eio__2path (EIO_RENAME, path, new_path, pri, cb, data); |
1367 | } |
1659 | } |
1368 | |
1660 | |
1369 | eio_req *eio_custom (eio_cb execute, int pri, eio_cb cb, void *data) |
1661 | eio_req *eio_custom (eio_cb execute, int pri, eio_cb cb, void *data) |
1370 | { |
1662 | { |
1371 | REQ (EIO_CUSTOM); req->feed = execute; SEND; |
1663 | REQ (EIO_CUSTOM); req->feed = (void (*)(eio_req *))execute; SEND; |
1372 | } |
1664 | } |
1373 | |
1665 | |
1374 | #endif |
1666 | #endif |
1375 | |
1667 | |
1376 | eio_req *eio_grp (eio_cb cb, void *data) |
1668 | eio_req *eio_grp (eio_cb cb, void *data) |
… | |
… | |
1404 | |
1696 | |
1405 | void eio_grp_add (eio_req *grp, eio_req *req) |
1697 | void eio_grp_add (eio_req *grp, eio_req *req) |
1406 | { |
1698 | { |
1407 | assert (("cannot add requests to IO::AIO::GRP after the group finished", grp->int1 != 2)); |
1699 | assert (("cannot add requests to IO::AIO::GRP after the group finished", grp->int1 != 2)); |
1408 | |
1700 | |
|
|
1701 | grp->flags |= EIO_FLAG_GROUPADD; |
|
|
1702 | |
1409 | ++grp->size; |
1703 | ++grp->size; |
1410 | req->grp = grp; |
1704 | req->grp = grp; |
1411 | |
1705 | |
1412 | req->grp_prev = 0; |
1706 | req->grp_prev = 0; |
1413 | req->grp_next = grp->grp_first; |
1707 | req->grp_next = grp->grp_first; |