ViewVC Help
View File | Revision Log | Show Annotations | Download File
/cvs/libeio/eio.c
(Generate patch)

Comparing libeio/eio.c (file contents):
Revision 1.80 by root, Tue Jul 5 18:59:28 2011 UTC vs.
Revision 1.93 by root, Mon Jul 18 01:27:03 2011 UTC

54#include <stdlib.h> 54#include <stdlib.h>
55#include <string.h> 55#include <string.h>
56#include <errno.h> 56#include <errno.h>
57#include <sys/types.h> 57#include <sys/types.h>
58#include <sys/stat.h> 58#include <sys/stat.h>
59#include <sys/statvfs.h>
60#include <limits.h> 59#include <limits.h>
61#include <fcntl.h> 60#include <fcntl.h>
62#include <assert.h> 61#include <assert.h>
63 62
64/* intptr_t comes from unistd.h, says POSIX/UNIX/tradition */ 63/* intptr_t comes from unistd.h, says POSIX/UNIX/tradition */
68#endif 67#endif
69 68
70#ifndef ECANCELED 69#ifndef ECANCELED
71# define ECANCELED EDOM 70# define ECANCELED EDOM
72#endif 71#endif
72#ifndef ELOOP
73# define ELOOP EDOM
74#endif
73 75
74static void eio_destroy (eio_req *req); 76static void eio_destroy (eio_req *req);
75 77
76#ifndef EIO_FINISH 78#ifndef EIO_FINISH
77# define EIO_FINISH(req) ((req)->finish) && !EIO_CANCELLED (req) ? (req)->finish (req) : 0 79# define EIO_FINISH(req) ((req)->finish) && !EIO_CANCELLED (req) ? (req)->finish (req) : 0
83 85
84#ifndef EIO_FEED 86#ifndef EIO_FEED
85# define EIO_FEED(req) do { if ((req)->feed ) (req)->feed (req); } while (0) 87# define EIO_FEED(req) do { if ((req)->feed ) (req)->feed (req); } while (0)
86#endif 88#endif
87 89
90#ifndef EIO_FD_TO_WIN32_HANDLE
91# define EIO_FD_TO_WIN32_HANDLE(fd) _get_osfhandle (fd)
92#endif
93#ifndef EIO_WIN32_HANDLE_TO_FD
94# define EIO_WIN32_HANDLE_TO_FD(handle) _open_osfhandle (handle, 0)
95#endif
96
97#define EIO_ERRNO(errval,retval) ((errno = errval), retval)
98
99#define EIO_ENOSYS() EIO_ERRNO (ENOSYS, -1)
100
88#ifdef _WIN32 101#ifdef _WIN32
89 102
90 /*doh*/ 103 #undef PAGESIZE
104 #define PAGESIZE 4096 /* GetSystemInfo? */
105
106 #ifdef EIO_STRUCT_STATI64
107 #define stat(path,buf) _stati64 (path,buf)
108 #define fstat(fd,buf) _fstati64 (path,buf)
109 #endif
110 #define lstat(path,buf) stat (path,buf)
111 #define fsync(fd) (FlushFileBuffers ((HANDLE)EIO_FD_TO_WIN32_HANDLE (fd)) ? 0 : EIO_ERRNO (EBADF, -1))
112 #define mkdir(path,mode) _mkdir (path)
113 #define link(old,neu) (CreateHardLink (neu, old, 0) ? 0 : EIO_ERRNO (ENOENT, -1))
114
115 #define chmod(path,mode) _chmod (path, mode)
116 #define dup(fd) _dup (fd)
117 #define dup2(fd1,fd2) _dup2 (fd1, fd2)
118
119 #define fchmod(fd,mode) EIO_ENOSYS ()
120 #define chown(path,uid,gid) EIO_ENOSYS ()
121 #define fchown(fd,uid,gid) EIO_ENOSYS ()
122 #define truncate(path,offs) EIO_ENOSYS () /* far-miss: SetEndOfFile */
123 #define ftruncate(fd,offs) EIO_ENOSYS () /* near-miss: SetEndOfFile */
124 #define mknod(path,mode,dev) EIO_ENOSYS ()
125 #define sync() EIO_ENOSYS ()
126 #define readlink(path,buf,s) EIO_ENOSYS ()
127 #define statvfs(path,buf) EIO_ENOSYS ()
128 #define fstatvfs(fd,buf) EIO_ENOSYS ()
129
130 /* we could even stat and see if it exists */
131 static int
132 symlink (const char *old, const char *neu)
133 {
134 #if WINVER >= 0x0600
135 if (CreateSymbolicLink (neu, old, 1))
136 return 0;
137
138 if (CreateSymbolicLink (neu, old, 0))
139 return 0;
140 #endif
141
142 return EIO_ERRNO (ENOENT, -1);
143 }
144
145 /* POSIX API only */
146 #define CreateHardLink(neu,old,flags) 0
147 #define CreateSymbolicLink(neu,old,flags) 0
148
149 struct statvfs
150 {
151 int dummy;
152 };
153
154 #define DT_DIR EIO_DT_DIR
155 #define DT_REG EIO_DT_REG
156 #define D_NAME(entp) entp.cFileName
157 #define D_TYPE(entp) (entp.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY ? DT_DIR : DT_REG)
158
91#else 159#else
92 160
93# include <sys/time.h> 161 #include <sys/time.h>
94# include <sys/select.h> 162 #include <sys/select.h>
163 #include <sys/statvfs.h>
95# include <unistd.h> 164 #include <unistd.h>
96# include <utime.h> 165 #include <utime.h>
97# include <signal.h> 166 #include <signal.h>
98# include <dirent.h> 167 #include <dirent.h>
99 168
100#if _POSIX_MEMLOCK || _POSIX_MEMLOCK_RANGE || _POSIX_MAPPED_FILES 169 #if _POSIX_MEMLOCK || _POSIX_MEMLOCK_RANGE || _POSIX_MAPPED_FILES
101# include <sys/mman.h> 170 #include <sys/mman.h>
102#endif 171 #endif
103 172
173 #define D_NAME(entp) entp->d_name
174
104/* POSIX_SOURCE is useless on bsd's, and XOPEN_SOURCE is unreliable there, too */ 175 /* POSIX_SOURCE is useless on bsd's, and XOPEN_SOURCE is unreliable there, too */
105# if __FreeBSD__ || defined __NetBSD__ || defined __OpenBSD__ 176 #if __FreeBSD__ || defined __NetBSD__ || defined __OpenBSD__
106# define _DIRENT_HAVE_D_TYPE /* sigh */ 177 #define _DIRENT_HAVE_D_TYPE /* sigh */
107# define D_INO(de) (de)->d_fileno 178 #define D_INO(de) (de)->d_fileno
108# define D_NAMLEN(de) (de)->d_namlen 179 #define D_NAMLEN(de) (de)->d_namlen
109# elif __linux || defined d_ino || _XOPEN_SOURCE >= 600 180 #elif __linux || defined d_ino || _XOPEN_SOURCE >= 600
110# define D_INO(de) (de)->d_ino 181 #define D_INO(de) (de)->d_ino
111# endif 182 #endif
112 183
113#ifdef _D_EXACT_NAMLEN 184 #ifdef _D_EXACT_NAMLEN
114# undef D_NAMLEN 185 #undef D_NAMLEN
115# define D_NAMLEN(de) _D_EXACT_NAMLEN (de) 186 #define D_NAMLEN(de) _D_EXACT_NAMLEN (de)
116#endif 187 #endif
117 188
118# ifdef _DIRENT_HAVE_D_TYPE 189 #ifdef _DIRENT_HAVE_D_TYPE
119# define D_TYPE(de) (de)->d_type 190 #define D_TYPE(de) (de)->d_type
120# endif 191 #endif
121 192
122# ifndef EIO_STRUCT_DIRENT 193 #ifndef EIO_STRUCT_DIRENT
123# define EIO_STRUCT_DIRENT struct dirent 194 #define EIO_STRUCT_DIRENT struct dirent
124# endif 195 #endif
125 196
126#endif 197#endif
127 198
128#if HAVE_SENDFILE 199#if HAVE_SENDFILE
129# if __linux 200# if __linux
145#endif 216#endif
146#ifndef D_INO 217#ifndef D_INO
147# define D_INO(de) 0 218# define D_INO(de) 0
148#endif 219#endif
149#ifndef D_NAMLEN 220#ifndef D_NAMLEN
150# define D_NAMLEN(de) strlen ((de)->d_name) 221# define D_NAMLEN(entp) strlen (D_NAME (entp))
151#endif 222#endif
152 223
153/* used for struct dirent, AIX doesn't provide it */ 224/* used for struct dirent, AIX doesn't provide it */
154#ifndef NAME_MAX 225#ifndef NAME_MAX
155# define NAME_MAX 4096 226# define NAME_MAX 4096
189#define ETP_WORKER_CLEAR(req) \ 260#define ETP_WORKER_CLEAR(req) \
190 if (wrk->dbuf) \ 261 if (wrk->dbuf) \
191 { \ 262 { \
192 free (wrk->dbuf); \ 263 free (wrk->dbuf); \
193 wrk->dbuf = 0; \ 264 wrk->dbuf = 0; \
194 } \
195 \
196 if (wrk->dirp) \
197 { \
198 closedir (wrk->dirp); \
199 wrk->dirp = 0; \
200 } 265 }
201 266
202#define ETP_WORKER_COMMON \ 267#define ETP_WORKER_COMMON \
203 void *dbuf; \ 268 void *dbuf;
204 DIR *dirp;
205 269
206/*****************************************************************************/ 270/*****************************************************************************/
207 271
208#define ETP_NUM_PRI (ETP_PRI_MAX - ETP_PRI_MIN + 1) 272#define ETP_NUM_PRI (ETP_PRI_MAX - ETP_PRI_MIN + 1)
209 273
238/* 302/*
239 * make our pread/pwrite emulation safe against themselves, but not against 303 * make our pread/pwrite emulation safe against themselves, but not against
240 * normal read/write by using a mutex. slows down execution a lot, 304 * normal read/write by using a mutex. slows down execution a lot,
241 * but that's your problem, not mine. 305 * but that's your problem, not mine.
242 */ 306 */
243static xmutex_t preadwritelock = X_MUTEX_INIT; 307static xmutex_t preadwritelock;
244#endif 308#endif
245 309
246typedef struct etp_worker 310typedef struct etp_worker
247{ 311{
248 /* locked by wrklock */ 312 /* locked by wrklock */
254 ETP_REQ *req; /* currently processed request */ 318 ETP_REQ *req; /* currently processed request */
255 319
256 ETP_WORKER_COMMON 320 ETP_WORKER_COMMON
257} etp_worker; 321} etp_worker;
258 322
259static etp_worker wrk_first = { &wrk_first, &wrk_first, 0 }; /* NOT etp */ 323static etp_worker wrk_first; /* NOT etp */
260 324
261#define ETP_WORKER_LOCK(wrk) X_LOCK (wrklock) 325#define ETP_WORKER_LOCK(wrk) X_LOCK (wrklock)
262#define ETP_WORKER_UNLOCK(wrk) X_UNLOCK (wrklock) 326#define ETP_WORKER_UNLOCK(wrk) X_UNLOCK (wrklock)
263 327
264/* worker threads management */ 328/* worker threads management */
335} etp_reqq; 399} etp_reqq;
336 400
337static etp_reqq req_queue; 401static etp_reqq req_queue;
338static etp_reqq res_queue; 402static etp_reqq res_queue;
339 403
404static void ecb_noinline ecb_cold
405reqq_init (etp_reqq *q)
406{
407 int pri;
408
409 for (pri = 0; pri < ETP_NUM_PRI; ++pri)
410 q->qs[pri] = q->qe[pri] = 0;
411
412 q->size = 0;
413}
414
340static int ecb_noinline 415static int ecb_noinline
341reqq_push (etp_reqq *q, ETP_REQ *req) 416reqq_push (etp_reqq *q, ETP_REQ *req)
342{ 417{
343 int pri = req->pri; 418 int pri = req->pri;
344 req->next = 0; 419 req->next = 0;
378 } 453 }
379 454
380 abort (); 455 abort ();
381} 456}
382 457
383static void ecb_cold 458static int ecb_cold
384etp_thread_init (void) 459etp_init (void (*want_poll)(void), void (*done_poll)(void))
385{ 460{
386 X_MUTEX_CREATE (wrklock); 461 X_MUTEX_CREATE (wrklock);
387 X_MUTEX_CREATE (reslock); 462 X_MUTEX_CREATE (reslock);
388 X_MUTEX_CREATE (reqlock); 463 X_MUTEX_CREATE (reqlock);
389 X_COND_CREATE (reqwait); 464 X_COND_CREATE (reqwait);
390}
391 465
392static void ecb_cold 466 reqq_init (&req_queue);
393etp_atfork_prepare (void) 467 reqq_init (&res_queue);
394{
395 X_LOCK (wrklock);
396 X_LOCK (reqlock);
397 X_LOCK (reslock);
398#if !HAVE_PREADWRITE
399 X_LOCK (preadwritelock);
400#endif
401}
402 468
403static void ecb_cold 469 wrk_first.next =
404etp_atfork_parent (void) 470 wrk_first.prev = &wrk_first;
405{
406#if !HAVE_PREADWRITE
407 X_UNLOCK (preadwritelock);
408#endif
409 X_UNLOCK (reslock);
410 X_UNLOCK (reqlock);
411 X_UNLOCK (wrklock);
412}
413
414static void ecb_cold
415etp_atfork_child (void)
416{
417 ETP_REQ *prv;
418
419 while ((prv = reqq_shift (&req_queue)))
420 ETP_DESTROY (prv);
421
422 while ((prv = reqq_shift (&res_queue)))
423 ETP_DESTROY (prv);
424
425 while (wrk_first.next != &wrk_first)
426 {
427 etp_worker *wrk = wrk_first.next;
428
429 if (wrk->req)
430 ETP_DESTROY (wrk->req);
431
432 etp_worker_clear (wrk);
433 etp_worker_free (wrk);
434 }
435 471
436 started = 0; 472 started = 0;
437 idle = 0; 473 idle = 0;
438 nreqs = 0; 474 nreqs = 0;
439 nready = 0; 475 nready = 0;
440 npending = 0; 476 npending = 0;
441
442 etp_thread_init ();
443}
444
445static void ecb_cold
446etp_once_init (void)
447{
448 etp_thread_init ();
449 X_THREAD_ATFORK (etp_atfork_prepare, etp_atfork_parent, etp_atfork_child);
450}
451
452static int ecb_cold
453etp_init (void (*want_poll)(void), void (*done_poll)(void))
454{
455 static pthread_once_t doinit = PTHREAD_ONCE_INIT;
456
457 pthread_once (&doinit, etp_once_init);
458 477
459 want_poll_cb = want_poll; 478 want_poll_cb = want_poll;
460 done_poll_cb = done_poll; 479 done_poll_cb = done_poll;
461 480
462 return 0; 481 return 0;
589} 608}
590 609
591static void 610static void
592etp_cancel (ETP_REQ *req) 611etp_cancel (ETP_REQ *req)
593{ 612{
594 X_LOCK (wrklock); 613 req->cancelled = 1;
595 req->flags |= EIO_FLAG_CANCELLED;
596 X_UNLOCK (wrklock);
597 614
598 eio_grp_cancel (req); 615 eio_grp_cancel (req);
599} 616}
600 617
601static void 618static void
847# undef pread 864# undef pread
848# undef pwrite 865# undef pwrite
849# define pread eio__pread 866# define pread eio__pread
850# define pwrite eio__pwrite 867# define pwrite eio__pwrite
851 868
852static ssize_t 869static eio_ssize_t
853eio__pread (int fd, void *buf, size_t count, off_t offset) 870eio__pread (int fd, void *buf, size_t count, off_t offset)
854{ 871{
855 ssize_t res; 872 eio_ssize_t res;
856 off_t ooffset; 873 off_t ooffset;
857 874
858 X_LOCK (preadwritelock); 875 X_LOCK (preadwritelock);
859 ooffset = lseek (fd, 0, SEEK_CUR); 876 ooffset = lseek (fd, 0, SEEK_CUR);
860 lseek (fd, offset, SEEK_SET); 877 lseek (fd, offset, SEEK_SET);
863 X_UNLOCK (preadwritelock); 880 X_UNLOCK (preadwritelock);
864 881
865 return res; 882 return res;
866} 883}
867 884
868static ssize_t 885static eio_ssize_t
869eio__pwrite (int fd, void *buf, size_t count, off_t offset) 886eio__pwrite (int fd, void *buf, size_t count, off_t offset)
870{ 887{
871 ssize_t res; 888 eio_ssize_t res;
872 off_t ooffset; 889 off_t ooffset;
873 890
874 X_LOCK (preadwritelock); 891 X_LOCK (preadwritelock);
875 ooffset = lseek (fd, 0, SEEK_CUR); 892 ooffset = lseek (fd, 0, SEEK_CUR);
876 lseek (fd, offset, SEEK_SET); 893 lseek (fd, offset, SEEK_SET);
950 /* even though we could play tricks with the flags, it's better to always 967 /* even though we could play tricks with the flags, it's better to always
951 * call fdatasync, as that matches the expectation of its users best */ 968 * call fdatasync, as that matches the expectation of its users best */
952 return fdatasync (fd); 969 return fdatasync (fd);
953} 970}
954 971
972static int
973eio__fallocate (int fd, int mode, off_t offset, size_t len)
974{
975#if HAVE_FALLOCATE
976 return fallocate (fd, mode, offset, len);
977#else
978 errno = ENOSYS;
979 return -1;
980#endif
981}
982
955#if !HAVE_READAHEAD 983#if !HAVE_READAHEAD
956# undef readahead 984# undef readahead
957# define readahead(fd,offset,count) eio__readahead (fd, offset, count, self) 985# define readahead(fd,offset,count) eio__readahead (fd, offset, count, self)
958 986
959static ssize_t 987static eio_ssize_t
960eio__readahead (int fd, off_t offset, size_t count, etp_worker *self) 988eio__readahead (int fd, off_t offset, size_t count, etp_worker *self)
961{ 989{
962 size_t todo = count; 990 size_t todo = count;
963 dBUF; 991 dBUF;
964 992
976} 1004}
977 1005
978#endif 1006#endif
979 1007
980/* sendfile always needs emulation */ 1008/* sendfile always needs emulation */
981static ssize_t 1009static eio_ssize_t
982eio__sendfile (int ofd, int ifd, off_t offset, size_t count, etp_worker *self) 1010eio__sendfile (int ofd, int ifd, off_t offset, size_t count, etp_worker *self)
983{ 1011{
984 ssize_t written = 0; 1012 eio_ssize_t written = 0;
985 ssize_t res; 1013 eio_ssize_t res;
986 1014
987 if (!count) 1015 if (!count)
988 return 0; 1016 return 0;
989 1017
990 for (;;) 1018 for (;;)
991 { 1019 {
1020#ifdef __APPLE__
1021# undef HAVE_SENDFILE /* broken, as everything on os x */
1022#endif
992#if HAVE_SENDFILE 1023#if HAVE_SENDFILE
993# if __linux 1024# if __linux
994 off_t soffset = offset; 1025 off_t soffset = offset;
995 res = sendfile (ofd, ifd, &soffset, count); 1026 res = sendfile (ofd, ifd, &soffset, count);
996 1027
1012 1043
1013 /* according to source inspection, this is correct, and useful behaviour */ 1044 /* according to source inspection, this is correct, and useful behaviour */
1014 if (sbytes) 1045 if (sbytes)
1015 res = sbytes; 1046 res = sbytes;
1016 1047
1017# elif defined (__APPLE__) && 0 /* broken, as everything on os x */ 1048# elif defined (__APPLE__)
1018 off_t sbytes = count; 1049 off_t sbytes = count;
1019 res = sendfile (ifd, ofd, offset, &sbytes, 0, 0); 1050 res = sendfile (ifd, ofd, offset, &sbytes, 0, 0);
1020 1051
1021 /* according to the manpage, sbytes is always valid */ 1052 /* according to the manpage, sbytes is always valid */
1022 if (sbytes) 1053 if (sbytes)
1039 if (res < 0 && sbytes) 1070 if (res < 0 && sbytes)
1040 res = sbytes; 1071 res = sbytes;
1041 1072
1042# endif 1073# endif
1043 1074
1044#elif defined (_WIN32) 1075#elif defined (_WIN32) && 0
1045 /* does not work, just for documentation of what would need to be done */ 1076 /* does not work, just for documentation of what would need to be done */
1046 /* actually, cannot be done like this, as TransmitFile changes the file offset, */ 1077 /* actually, cannot be done like this, as TransmitFile changes the file offset, */
1047 /* libeio guarantees that the file offset does not change, and windows */ 1078 /* libeio guarantees that the file offset does not change, and windows */
1048 /* has no way to get an independent handle to the same file description */ 1079 /* has no way to get an independent handle to the same file description */
1049 HANDLE h = TO_SOCKET (ifd); 1080 HANDLE h = TO_SOCKET (ifd);
1082 && (errno == ENOSYS || errno == EINVAL || errno == ENOTSOCK 1113 && (errno == ENOSYS || errno == EINVAL || errno == ENOTSOCK
1083 /* BSDs */ 1114 /* BSDs */
1084#ifdef ENOTSUP /* sigh, if the steenking pile called openbsd would only try to at least compile posix code... */ 1115#ifdef ENOTSUP /* sigh, if the steenking pile called openbsd would only try to at least compile posix code... */
1085 || errno == ENOTSUP 1116 || errno == ENOTSUP
1086#endif 1117#endif
1118#ifdef EOPNOTSUPP /* windows */
1087 || errno == EOPNOTSUPP /* BSDs */ 1119 || errno == EOPNOTSUPP /* BSDs */
1120#endif
1088#if __solaris 1121#if __solaris
1089 || errno == EAFNOSUPPORT || errno == EPROTOTYPE 1122 || errno == EAFNOSUPPORT || errno == EPROTOTYPE
1090#endif 1123#endif
1091 ) 1124 )
1092 ) 1125 )
1096 1129
1097 res = 0; 1130 res = 0;
1098 1131
1099 while (count) 1132 while (count)
1100 { 1133 {
1101 ssize_t cnt; 1134 eio_ssize_t cnt;
1102 1135
1103 cnt = pread (ifd, eio_buf, count > EIO_BUFSIZE ? EIO_BUFSIZE : count, offset); 1136 cnt = pread (ifd, eio_buf, count > EIO_BUFSIZE ? EIO_BUFSIZE : count, offset);
1104 1137
1105 if (cnt <= 0) 1138 if (cnt <= 0)
1106 { 1139 {
1154 /* round up length */ 1187 /* round up length */
1155 *length = (*length + mask) & ~mask; 1188 *length = (*length + mask) & ~mask;
1156} 1189}
1157 1190
1158#if !_POSIX_MEMLOCK 1191#if !_POSIX_MEMLOCK
1159# define eio__mlockall(a) ((errno = ENOSYS), -1) 1192# define eio__mlockall(a) EIO_ENOSYS ()
1160#else 1193#else
1161 1194
1162static int 1195static int
1163eio__mlockall (int flags) 1196eio__mlockall (int flags)
1164{ 1197{
1178 return mlockall (flags); 1211 return mlockall (flags);
1179} 1212}
1180#endif 1213#endif
1181 1214
1182#if !_POSIX_MEMLOCK_RANGE 1215#if !_POSIX_MEMLOCK_RANGE
1183# define eio__mlock(a,b) ((errno = ENOSYS), -1) 1216# define eio__mlock(a,b) EIO_ENOSYS ()
1184#else 1217#else
1185 1218
1186static int 1219static int
1187eio__mlock (void *addr, size_t length) 1220eio__mlock (void *addr, size_t length)
1188{ 1221{
1192} 1225}
1193 1226
1194#endif 1227#endif
1195 1228
1196#if !(_POSIX_MAPPED_FILES && _POSIX_SYNCHRONIZED_IO) 1229#if !(_POSIX_MAPPED_FILES && _POSIX_SYNCHRONIZED_IO)
1197# define eio__msync(a,b,c) ((errno = ENOSYS), -1) 1230# define eio__msync(a,b,c) EIO_ENOSYS ()
1198#else 1231#else
1199 1232
1200static int 1233static int
1201eio__msync (void *mem, size_t len, int flags) 1234eio__msync (void *mem, size_t len, int flags)
1202{ 1235{
1314 res += strlen (res); 1347 res += strlen (res);
1315 } 1348 }
1316 1349
1317 while (*rel) 1350 while (*rel)
1318 { 1351 {
1319 ssize_t len, linklen; 1352 eio_ssize_t len, linklen;
1320 char *beg = rel; 1353 char *beg = rel;
1321 1354
1322 while (*rel && *rel != '/') 1355 while (*rel && *rel != '/')
1323 ++rel; 1356 ++rel;
1324 1357
1417 1450
1418#define EIO_SORT_CUTOFF 30 /* quite high, but performs well on many filesystems */ 1451#define EIO_SORT_CUTOFF 30 /* quite high, but performs well on many filesystems */
1419#define EIO_SORT_FAST 60 /* when to only use insertion sort */ 1452#define EIO_SORT_FAST 60 /* when to only use insertion sort */
1420 1453
1421static void 1454static void
1422eio_dent_radix_sort (eio_dirent *dents, int size, signed char score_bits, ino_t inode_bits) 1455eio_dent_radix_sort (eio_dirent *dents, int size, signed char score_bits, eio_ino_t inode_bits)
1423{ 1456{
1424 unsigned char bits [9 + sizeof (ino_t) * 8]; 1457 unsigned char bits [9 + sizeof (eio_ino_t) * 8];
1425 unsigned char *bit = bits; 1458 unsigned char *bit = bits;
1426 1459
1427 assert (CHAR_BIT == 8); 1460 assert (CHAR_BIT == 8);
1428 assert (sizeof (eio_dirent) * 8 < 256); 1461 assert (sizeof (eio_dirent) * 8 < 256);
1429 assert (offsetof (eio_dirent, inode)); /* we use bit #0 as sentinel */ 1462 assert (offsetof (eio_dirent, inode)); /* we use bit #0 as sentinel */
1431 1464
1432 if (size <= EIO_SORT_FAST) 1465 if (size <= EIO_SORT_FAST)
1433 return; 1466 return;
1434 1467
1435 /* first prepare an array of bits to test in our radix sort */ 1468 /* first prepare an array of bits to test in our radix sort */
1436 /* try to take endianness into account, as well as differences in ino_t sizes */ 1469 /* try to take endianness into account, as well as differences in eio_ino_t sizes */
1437 /* inode_bits must contain all inodes ORed together */ 1470 /* inode_bits must contain all inodes ORed together */
1438 /* which is used to skip bits that are 0 everywhere, which is very common */ 1471 /* which is used to skip bits that are 0 everywhere, which is very common */
1439 { 1472 {
1440 ino_t endianness; 1473 eio_ino_t endianness;
1441 int i, j; 1474 int i, j;
1442 1475
1443 /* we store the byte offset of byte n into byte n of "endianness" */ 1476 /* we store the byte offset of byte n into byte n of "endianness" */
1444 for (i = 0; i < sizeof (ino_t); ++i) 1477 for (i = 0; i < sizeof (eio_ino_t); ++i)
1445 ((unsigned char *)&endianness)[i] = i; 1478 ((unsigned char *)&endianness)[i] = i;
1446 1479
1447 *bit++ = 0; 1480 *bit++ = 0;
1448 1481
1449 for (i = 0; i < sizeof (ino_t); ++i) 1482 for (i = 0; i < sizeof (eio_ino_t); ++i)
1450 { 1483 {
1451 /* shifting off the byte offsets out of "endianness" */ 1484 /* shifting off the byte offsets out of "endianness" */
1452 int offs = (offsetof (eio_dirent, inode) + (endianness & 0xff)) * 8; 1485 int offs = (offsetof (eio_dirent, inode) + (endianness & 0xff)) * 8;
1453 endianness >>= 8; 1486 endianness >>= 8;
1454 1487
1455 for (j = 0; j < 8; ++j) 1488 for (j = 0; j < 8; ++j)
1456 if (inode_bits & (((ino_t)1) << (i * 8 + j))) 1489 if (inode_bits & (((eio_ino_t)1) << (i * 8 + j)))
1457 *bit++ = offs + j; 1490 *bit++ = offs + j;
1458 } 1491 }
1459 1492
1460 for (j = 0; j < 8; ++j) 1493 for (j = 0; j < 8; ++j)
1461 if (score_bits & (1 << j)) 1494 if (score_bits & (1 << j))
1462 *bit++ = offsetof (eio_dirent, score) * 8 + j; 1495 *bit++ = offsetof (eio_dirent, score) * 8 + j;
1463 } 1496 }
1464 1497
1465 /* now actually do the sorting (a variant of MSD radix sort) */ 1498 /* now actually do the sorting (a variant of MSD radix sort) */
1466 { 1499 {
1467 eio_dirent *base_stk [9 + sizeof (ino_t) * 8], *base; 1500 eio_dirent *base_stk [9 + sizeof (eio_ino_t) * 8], *base;
1468 eio_dirent *end_stk [9 + sizeof (ino_t) * 8], *end; 1501 eio_dirent *end_stk [9 + sizeof (eio_ino_t) * 8], *end;
1469 unsigned char *bit_stk [9 + sizeof (ino_t) * 8]; 1502 unsigned char *bit_stk [9 + sizeof (eio_ino_t) * 8];
1470 int stk_idx = 0; 1503 int stk_idx = 0;
1471 1504
1472 base_stk [stk_idx] = dents; 1505 base_stk [stk_idx] = dents;
1473 end_stk [stk_idx] = dents + size; 1506 end_stk [stk_idx] = dents + size;
1474 bit_stk [stk_idx] = bit - 1; 1507 bit_stk [stk_idx] = bit - 1;
1553 } 1586 }
1554 } 1587 }
1555} 1588}
1556 1589
1557static void 1590static void
1558eio_dent_sort (eio_dirent *dents, int size, signed char score_bits, ino_t inode_bits) 1591eio_dent_sort (eio_dirent *dents, int size, signed char score_bits, eio_ino_t inode_bits)
1559{ 1592{
1560 if (size <= 1) 1593 if (size <= 1)
1561 return; /* our insertion sort relies on size > 0 */ 1594 return; /* our insertion sort relies on size > 0 */
1562 1595
1563 /* first we use a radix sort, but only for dirs >= EIO_SORT_FAST */ 1596 /* first we use a radix sort, but only for dirs >= EIO_SORT_FAST */
1571 1604
1572/* read a full directory */ 1605/* read a full directory */
1573static void 1606static void
1574eio__scandir (eio_req *req, etp_worker *self) 1607eio__scandir (eio_req *req, etp_worker *self)
1575{ 1608{
1576 DIR *dirp;
1577 EIO_STRUCT_DIRENT *entp;
1578 char *name, *names; 1609 char *name, *names;
1579 int namesalloc = 4096; 1610 int namesalloc = 4096 - sizeof (void *) * 4;
1580 int namesoffs = 0; 1611 int namesoffs = 0;
1581 int flags = req->int1; 1612 int flags = req->int1;
1582 eio_dirent *dents = 0; 1613 eio_dirent *dents = 0;
1583 int dentalloc = 128; 1614 int dentalloc = 128;
1584 int dentoffs = 0; 1615 int dentoffs = 0;
1585 ino_t inode_bits = 0; 1616 eio_ino_t inode_bits = 0;
1617#ifdef _WIN32
1618 HANDLE dirp;
1619 WIN32_FIND_DATA entp;
1620#else
1621 DIR *dirp;
1622 EIO_STRUCT_DIRENT *entp;
1623#endif
1586 1624
1587 req->result = -1; 1625 req->result = -1;
1588 1626
1589 if (!(flags & EIO_READDIR_DENTS)) 1627 if (!(flags & EIO_READDIR_DENTS))
1590 flags &= ~(EIO_READDIR_DIRS_FIRST | EIO_READDIR_STAT_ORDER); 1628 flags &= ~(EIO_READDIR_DIRS_FIRST | EIO_READDIR_STAT_ORDER);
1591 1629
1592 X_LOCK (wrklock); 1630#ifdef _WIN32
1593 /* the corresponding closedir is in ETP_WORKER_CLEAR */ 1631 {
1632 int len = strlen ((const char *)req->ptr1);
1633 char *path = malloc (MAX_PATH);
1634 const char *fmt;
1635
1636 if (!len)
1637 fmt = "./*";
1638 else if (((const char *)req->ptr1)[len - 1] == '/' || ((const char *)req->ptr1)[len - 1] == '\\')
1639 fmt = "%s*";
1640 else
1641 fmt = "%s/*";
1642
1643 _snprintf (path, MAX_PATH, fmt, (const char *)req->ptr1);
1644 dirp = FindFirstFile (path, &entp);
1645 free (path);
1646
1647 if (dirp == INVALID_HANDLE_VALUE)
1648 {
1649 dirp = 0;
1650
1651 switch (GetLastError ())
1652 {
1653 case ERROR_FILE_NOT_FOUND:
1654 req->result = 0;
1655 break;
1656
1657 case ERROR_INVALID_NAME:
1658 case ERROR_PATH_NOT_FOUND:
1659 case ERROR_NO_MORE_FILES:
1660 errno = ENOENT;
1661 break;
1662
1663 case ERROR_NOT_ENOUGH_MEMORY:
1664 errno = ENOMEM;
1665 break;
1666
1667 default:
1668 errno = EINVAL;
1669 break;
1670 }
1671 }
1672 }
1673#else
1594 self->dirp = dirp = opendir (req->ptr1); 1674 dirp = opendir (req->ptr1);
1675#endif
1595 1676
1596 if (req->flags & EIO_FLAG_PTR1_FREE) 1677 if (req->flags & EIO_FLAG_PTR1_FREE)
1597 free (req->ptr1); 1678 free (req->ptr1);
1598 1679
1599 req->flags |= EIO_FLAG_PTR1_FREE | EIO_FLAG_PTR2_FREE; 1680 req->flags |= EIO_FLAG_PTR1_FREE | EIO_FLAG_PTR2_FREE;
1600 req->ptr1 = dents = flags ? malloc (dentalloc * sizeof (eio_dirent)) : 0; 1681 req->ptr1 = dents = flags ? malloc (dentalloc * sizeof (eio_dirent)) : 0;
1601 req->ptr2 = names = malloc (namesalloc); 1682 req->ptr2 = names = malloc (namesalloc);
1602 X_UNLOCK (wrklock);
1603 1683
1604 if (dirp && names && (!flags || dents)) 1684 if (dirp && names && (!flags || dents))
1605 for (;;) 1685 for (;;)
1606 { 1686 {
1687 int done;
1688
1689#ifdef _WIN32
1690 done = !dirp;
1691#else
1607 errno = 0; 1692 errno = 0;
1608 entp = readdir (dirp); 1693 entp = readdir (dirp);
1694 done = !entp;
1695#endif
1609 1696
1610 if (!entp) 1697 if (done)
1611 { 1698 {
1699#ifndef _WIN32
1700 int old_errno = errno;
1701 closedir (dirp);
1702 errno = old_errno;
1703
1612 if (errno) 1704 if (errno)
1613 break; 1705 break;
1706#endif
1614 1707
1615 /* sort etc. */ 1708 /* sort etc. */
1616 req->int1 = flags; 1709 req->int1 = flags;
1617 req->result = dentoffs; 1710 req->result = dentoffs;
1618 1711
1647 1740
1648 break; 1741 break;
1649 } 1742 }
1650 1743
1651 /* now add the entry to our list(s) */ 1744 /* now add the entry to our list(s) */
1652 name = entp->d_name; 1745 name = D_NAME (entp);
1653 1746
1654 /* skip . and .. entries */ 1747 /* skip . and .. entries */
1655 if (name [0] != '.' || (name [1] && (name [1] != '.' || name [2]))) 1748 if (name [0] != '.' || (name [1] && (name [1] != '.' || name [2])))
1656 { 1749 {
1657 int len = D_NAMLEN (entp) + 1; 1750 int len = D_NAMLEN (entp) + 1;
1658 1751
1659 while (ecb_expect_false (namesoffs + len > namesalloc)) 1752 while (ecb_expect_false (namesoffs + len > namesalloc))
1660 { 1753 {
1661 namesalloc *= 2; 1754 namesalloc *= 2;
1662 X_LOCK (wrklock);
1663 req->ptr2 = names = realloc (names, namesalloc); 1755 req->ptr2 = names = realloc (names, namesalloc);
1664 X_UNLOCK (wrklock);
1665 1756
1666 if (!names) 1757 if (!names)
1667 break; 1758 break;
1668 } 1759 }
1669 1760
1674 struct eio_dirent *ent; 1765 struct eio_dirent *ent;
1675 1766
1676 if (ecb_expect_false (dentoffs == dentalloc)) 1767 if (ecb_expect_false (dentoffs == dentalloc))
1677 { 1768 {
1678 dentalloc *= 2; 1769 dentalloc *= 2;
1679 X_LOCK (wrklock);
1680 req->ptr1 = dents = realloc (dents, dentalloc * sizeof (eio_dirent)); 1770 req->ptr1 = dents = realloc (dents, dentalloc * sizeof (eio_dirent));
1681 X_UNLOCK (wrklock);
1682 1771
1683 if (!dents) 1772 if (!dents)
1684 break; 1773 break;
1685 } 1774 }
1686 1775
1766 if (EIO_CANCELLED (req)) 1855 if (EIO_CANCELLED (req))
1767 { 1856 {
1768 errno = ECANCELED; 1857 errno = ECANCELED;
1769 break; 1858 break;
1770 } 1859 }
1860
1861#ifdef _WIN32
1862 if (!FindNextFile (dirp, &entp))
1863 {
1864 FindClose (dirp);
1865 dirp = 0;
1866 }
1867#endif
1771 } 1868 }
1772} 1869}
1773 1870
1774/*****************************************************************************/ 1871/*****************************************************************************/
1775 1872
1792{ 1889{
1793 ETP_REQ *req; 1890 ETP_REQ *req;
1794 struct timespec ts; 1891 struct timespec ts;
1795 etp_worker *self = (etp_worker *)thr_arg; 1892 etp_worker *self = (etp_worker *)thr_arg;
1796 1893
1797 /* try to distribute timeouts somewhat randomly */ 1894 /* try to distribute timeouts somewhat evenly */
1798 ts.tv_nsec = ((unsigned long)self & 1023UL) * (1000000000UL / 1024UL); 1895 ts.tv_nsec = ((unsigned long)self & 1023UL) * (1000000000UL / 1024UL);
1799 1896
1800 for (;;) 1897 for (;;)
1801 { 1898 {
1899 ts.tv_sec = 0;
1900
1802 X_LOCK (reqlock); 1901 X_LOCK (reqlock);
1803 1902
1804 for (;;) 1903 for (;;)
1805 { 1904 {
1806 self->req = req = reqq_shift (&req_queue); 1905 self->req = req = reqq_shift (&req_queue);
1807 1906
1808 if (req) 1907 if (req)
1809 break; 1908 break;
1810 1909
1910 if (ts.tv_sec == 1) /* no request, but timeout detected, let's quit */
1911 {
1912 X_UNLOCK (reqlock);
1913 X_LOCK (wrklock);
1914 --started;
1915 X_UNLOCK (wrklock);
1916 goto quit;
1917 }
1918
1811 ++idle; 1919 ++idle;
1812 1920
1813 ts.tv_sec = time (0) + idle_timeout; 1921 if (idle <= max_idle)
1814 if (X_COND_TIMEDWAIT (reqwait, reqlock, ts) == ETIMEDOUT) 1922 /* we are allowed to idle, so do so without any timeout */
1923 X_COND_WAIT (reqwait, reqlock);
1924 else
1815 { 1925 {
1816 if (idle > max_idle) 1926 /* initialise timeout once */
1817 { 1927 if (!ts.tv_sec)
1818 --idle; 1928 ts.tv_sec = time (0) + idle_timeout;
1819 X_UNLOCK (reqlock);
1820 X_LOCK (wrklock);
1821 --started;
1822 X_UNLOCK (wrklock);
1823 goto quit;
1824 }
1825 1929
1826 /* we are allowed to idle, so do so without any timeout */
1827 X_COND_WAIT (reqwait, reqlock); 1930 if (X_COND_TIMEDWAIT (reqwait, reqlock, ts) == ETIMEDOUT)
1931 ts.tv_sec = 1; /* assuming this is not a value computed above.,.. */
1828 } 1932 }
1829 1933
1830 --idle; 1934 --idle;
1831 } 1935 }
1832 1936
1863/*****************************************************************************/ 1967/*****************************************************************************/
1864 1968
1865int ecb_cold 1969int ecb_cold
1866eio_init (void (*want_poll)(void), void (*done_poll)(void)) 1970eio_init (void (*want_poll)(void), void (*done_poll)(void))
1867{ 1971{
1972#if !HAVE_PREADWRITE
1973 X_MUTEX_CREATE (preadwritelock);
1974#endif
1975
1868 return etp_init (want_poll, done_poll); 1976 return etp_init (want_poll, done_poll);
1869} 1977}
1870 1978
1871ecb_inline void 1979ecb_inline void
1872eio_api_destroy (eio_req *req) 1980eio_api_destroy (eio_req *req)
1962 case EIO_MSYNC: req->result = eio__msync (req->ptr2, req->size, req->int1); break; 2070 case EIO_MSYNC: req->result = eio__msync (req->ptr2, req->size, req->int1); break;
1963 case EIO_MTOUCH: req->result = eio__mtouch (req); break; 2071 case EIO_MTOUCH: req->result = eio__mtouch (req); break;
1964 case EIO_MLOCK: req->result = eio__mlock (req->ptr2, req->size); break; 2072 case EIO_MLOCK: req->result = eio__mlock (req->ptr2, req->size); break;
1965 case EIO_MLOCKALL: req->result = eio__mlockall (req->int1); break; 2073 case EIO_MLOCKALL: req->result = eio__mlockall (req->int1); break;
1966 case EIO_SYNC_FILE_RANGE: req->result = eio__sync_file_range (req->int1, req->offs, req->size, req->int2); break; 2074 case EIO_SYNC_FILE_RANGE: req->result = eio__sync_file_range (req->int1, req->offs, req->size, req->int2); break;
2075 case EIO_FALLOCATE: req->result = eio__fallocate (req->int1, req->int2, req->offs, req->size); break;
1967 2076
1968 case EIO_READDIR: eio__scandir (req, self); break; 2077 case EIO_READDIR: eio__scandir (req, self); break;
1969 2078
1970 case EIO_BUSY: 2079 case EIO_BUSY:
1971#ifdef _WIN32 2080#ifdef _WIN32
2071eio_req *eio_sync_file_range (int fd, off_t offset, size_t nbytes, unsigned int flags, int pri, eio_cb cb, void *data) 2180eio_req *eio_sync_file_range (int fd, off_t offset, size_t nbytes, unsigned int flags, int pri, eio_cb cb, void *data)
2072{ 2181{
2073 REQ (EIO_SYNC_FILE_RANGE); req->int1 = fd; req->offs = offset; req->size = nbytes; req->int2 = flags; SEND; 2182 REQ (EIO_SYNC_FILE_RANGE); req->int1 = fd; req->offs = offset; req->size = nbytes; req->int2 = flags; SEND;
2074} 2183}
2075 2184
2185eio_req *eio_fallocate (int fd, int mode, off_t offset, size_t len, int pri, eio_cb cb, void *data)
2186{
2187 REQ (EIO_FALLOCATE); req->int1 = fd; req->int2 = mode; req->offs = offset; req->size = len; SEND;
2188}
2189
2076eio_req *eio_fdatasync (int fd, int pri, eio_cb cb, void *data) 2190eio_req *eio_fdatasync (int fd, int pri, eio_cb cb, void *data)
2077{ 2191{
2078 REQ (EIO_FDATASYNC); req->int1 = fd; SEND; 2192 REQ (EIO_FDATASYNC); req->int1 = fd; SEND;
2079} 2193}
2080 2194
2121eio_req *eio_fchmod (int fd, mode_t mode, int pri, eio_cb cb, void *data) 2235eio_req *eio_fchmod (int fd, mode_t mode, int pri, eio_cb cb, void *data)
2122{ 2236{
2123 REQ (EIO_FCHMOD); req->int1 = fd; req->int2 = (long)mode; SEND; 2237 REQ (EIO_FCHMOD); req->int1 = fd; req->int2 = (long)mode; SEND;
2124} 2238}
2125 2239
2126eio_req *eio_fchown (int fd, uid_t uid, gid_t gid, int pri, eio_cb cb, void *data) 2240eio_req *eio_fchown (int fd, eio_uid_t uid, eio_gid_t gid, int pri, eio_cb cb, void *data)
2127{ 2241{
2128 REQ (EIO_FCHOWN); req->int1 = fd; req->int2 = (long)uid; req->int3 = (long)gid; SEND; 2242 REQ (EIO_FCHOWN); req->int1 = fd; req->int2 = (long)uid; req->int3 = (long)gid; SEND;
2129} 2243}
2130 2244
2131eio_req *eio_dup2 (int fd, int fd2, int pri, eio_cb cb, void *data) 2245eio_req *eio_dup2 (int fd, int fd2, int pri, eio_cb cb, void *data)
2151eio_req *eio_truncate (const char *path, off_t offset, int pri, eio_cb cb, void *data) 2265eio_req *eio_truncate (const char *path, off_t offset, int pri, eio_cb cb, void *data)
2152{ 2266{
2153 REQ (EIO_TRUNCATE); PATH; req->offs = offset; SEND; 2267 REQ (EIO_TRUNCATE); PATH; req->offs = offset; SEND;
2154} 2268}
2155 2269
2156eio_req *eio_chown (const char *path, uid_t uid, gid_t gid, int pri, eio_cb cb, void *data) 2270eio_req *eio_chown (const char *path, eio_uid_t uid, eio_gid_t gid, int pri, eio_cb cb, void *data)
2157{ 2271{
2158 REQ (EIO_CHOWN); PATH; req->int2 = (long)uid; req->int3 = (long)gid; SEND; 2272 REQ (EIO_CHOWN); PATH; req->int2 = (long)uid; req->int3 = (long)gid; SEND;
2159} 2273}
2160 2274
2161eio_req *eio_chmod (const char *path, mode_t mode, int pri, eio_cb cb, void *data) 2275eio_req *eio_chmod (const char *path, mode_t mode, int pri, eio_cb cb, void *data)
2308} 2422}
2309 2423
2310/*****************************************************************************/ 2424/*****************************************************************************/
2311/* misc garbage */ 2425/* misc garbage */
2312 2426
2313ssize_t 2427eio_ssize_t
2314eio_sendfile_sync (int ofd, int ifd, off_t offset, size_t count) 2428eio_sendfile_sync (int ofd, int ifd, off_t offset, size_t count)
2315{ 2429{
2316 etp_worker wrk; 2430 etp_worker wrk;
2317 ssize_t ret; 2431 eio_ssize_t ret;
2318 2432
2319 wrk.dbuf = 0; 2433 wrk.dbuf = 0;
2320 2434
2321 ret = eio__sendfile (ofd, ifd, offset, count, &wrk); 2435 ret = eio__sendfile (ofd, ifd, offset, count, &wrk);
2322 2436

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines