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

Comparing libeio/eio.c (file contents):
Revision 1.30 by root, Wed Jun 3 12:24:49 2009 UTC vs.
Revision 1.61 by root, Fri Feb 11 00:53:24 2011 UTC

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,2010,2011 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 *
… …
49#include <stdlib.h> 49#include <stdlib.h>
50#include <string.h> 50#include <string.h>
51#include <errno.h> 51#include <errno.h>
52#include <sys/types.h> 52#include <sys/types.h>
53#include <sys/stat.h> 53#include <sys/stat.h>
54#include <sys/statvfs.h>
54#include <limits.h> 55#include <limits.h>
55#include <fcntl.h> 56#include <fcntl.h>
56#include <assert.h> 57#include <assert.h>
57 58
58#ifndef EIO_FINISH 59#ifndef EIO_FINISH
… …
68#endif 69#endif
69 70
70#ifdef _WIN32 71#ifdef _WIN32
71 72
72 /*doh*/ 73 /*doh*/
73
74#else 74#else
75 75
76# include "config.h" 76# include "config.h"
77# include <sys/time.h> 77# include <sys/time.h>
78# include <sys/select.h> 78# include <sys/select.h>
79# include <unistd.h> 79# include <unistd.h>
80# include <utime.h> 80# include <utime.h>
81# include <signal.h> 81# include <signal.h>
82# include <dirent.h> 82# include <dirent.h>
83 83
84#if _POSIX_MEMLOCK || _POSIX_MEMLOCK_RANGE || _POSIX_MAPPED_FILES
85# include <sys/mman.h>
86#endif
87
88/* POSIX_SOURCE is useless on bsd's, and XOPEN_SOURCE is unreliable there, too */
89# if __FreeBSD__ || defined __NetBSD__ || defined __OpenBSD__
90# define _DIRENT_HAVE_D_TYPE /* sigh */
91# define D_INO(de) (de)->d_fileno
92# define D_NAMLEN(de) (de)->d_namlen
93# elif __linux || defined d_ino || _XOPEN_SOURCE >= 600
94# define D_INO(de) (de)->d_ino
95# endif
96
97#ifdef _D_EXACT_NAMLEN
98# undef D_NAMLEN
99# define D_NAMLEN(de) _D_EXACT_NAMLEN (de)
100#endif
101
102# ifdef _DIRENT_HAVE_D_TYPE
103# define D_TYPE(de) (de)->d_type
104# endif
105
84# ifndef EIO_STRUCT_DIRENT 106# ifndef EIO_STRUCT_DIRENT
85# define EIO_STRUCT_DIRENT struct dirent 107# define EIO_STRUCT_DIRENT struct dirent
86# endif 108# endif
87 109
88#endif 110#endif
89 111
90#if HAVE_SENDFILE 112#if HAVE_SENDFILE
91# if __linux 113# if __linux
92# include <sys/sendfile.h> 114# include <sys/sendfile.h>
93# elif __freebsd 115# elif __FreeBSD__ || defined __APPLE__
94# include <sys/socket.h> 116# include <sys/socket.h>
95# include <sys/uio.h> 117# include <sys/uio.h>
96# elif __hpux 118# elif __hpux
97# include <sys/socket.h> 119# include <sys/socket.h>
98# elif __solaris /* not yet */ 120# elif __solaris
99# include <sys/sendfile.h> 121# include <sys/sendfile.h>
100# else 122# else
101# error sendfile support requested but not available 123# error sendfile support requested but not available
102# endif 124# endif
103#endif 125#endif
104 126
127#ifndef D_TYPE
128# define D_TYPE(de) 0
129#endif
130#ifndef D_INO
131# define D_INO(de) 0
132#endif
133#ifndef D_NAMLEN
134# define D_NAMLEN(de) strlen ((de)->d_name)
135#endif
136
105/* number of seconds after which an idle threads exit */ 137/* number of seconds after which an idle threads exit */
106#define IDLE_TIMEOUT 10 138#define IDLE_TIMEOUT 10
107 139
108/* used for struct dirent, AIX doesn't provide it */ 140/* used for struct dirent, AIX doesn't provide it */
109#ifndef NAME_MAX 141#ifndef NAME_MAX
110# define NAME_MAX 4096 142# define NAME_MAX 4096
143#endif
144
145/* used for readlink etc. */
146#ifndef PATH_MAX
147# define PATH_MAX 4096
111#endif 148#endif
112 149
113/* buffer size for various temporary buffers */ 150/* buffer size for various temporary buffers */
114#define EIO_BUFSIZE 65536 151#define EIO_BUFSIZE 65536
115 152
… …
159 if (wrk->dirp) \ 196 if (wrk->dirp) \
160 { \ 197 { \
161 closedir (wrk->dirp); \ 198 closedir (wrk->dirp); \
162 wrk->dirp = 0; \ 199 wrk->dirp = 0; \
163 } 200 }
201
164#define ETP_WORKER_COMMON \ 202#define ETP_WORKER_COMMON \
165 void *dbuf; \ 203 void *dbuf; \
166 DIR *dirp; 204 DIR *dirp;
167 205
168/*****************************************************************************/ 206/*****************************************************************************/
169 207
170#define ETP_NUM_PRI (ETP_PRI_MAX - ETP_PRI_MIN + 1) 208#define ETP_NUM_PRI (ETP_PRI_MAX - ETP_PRI_MIN + 1)
171 209
172/* calculcate time difference in ~1/EIO_TICKS of a second */ 210/* calculate time difference in ~1/EIO_TICKS of a second */
173static int tvdiff (struct timeval *tv1, struct timeval *tv2) 211static int tvdiff (struct timeval *tv1, struct timeval *tv2)
174{ 212{
175 return (tv2->tv_sec - tv1->tv_sec ) * EIO_TICKS 213 return (tv2->tv_sec - tv1->tv_sec ) * EIO_TICKS
176 + ((tv2->tv_usec - tv1->tv_usec) >> 10); 214 + ((tv2->tv_usec - tv1->tv_usec) >> 10);
177} 215}
… …
187static volatile unsigned int nreqs; /* reqlock */ 225static volatile unsigned int nreqs; /* reqlock */
188static volatile unsigned int nready; /* reqlock */ 226static volatile unsigned int nready; /* reqlock */
189static volatile unsigned int npending; /* reqlock */ 227static volatile unsigned int npending; /* reqlock */
190static volatile unsigned int max_idle = 4; 228static volatile unsigned int max_idle = 4;
191 229
192static mutex_t wrklock = X_MUTEX_INIT; 230static xmutex_t wrklock = X_MUTEX_INIT;
193static mutex_t reslock = X_MUTEX_INIT; 231static xmutex_t reslock = X_MUTEX_INIT;
194static mutex_t reqlock = X_MUTEX_INIT; 232static xmutex_t reqlock = X_MUTEX_INIT;
195static cond_t reqwait = X_COND_INIT; 233static xcond_t reqwait = X_COND_INIT;
196 234
197#if !HAVE_PREADWRITE 235#if !HAVE_PREADWRITE
198/* 236/*
199 * make our pread/pwrite emulation safe against themselves, but not against 237 * make our pread/pwrite emulation safe against themselves, but not against
200 * normal read/write by using a mutex. slows down execution a lot, 238 * normal read/write by using a mutex. slows down execution a lot,
201 * but that's your problem, not mine. 239 * but that's your problem, not mine.
202 */ 240 */
203static mutex_t preadwritelock = X_MUTEX_INIT; 241static xmutex_t preadwritelock = X_MUTEX_INIT;
204#endif 242#endif
205 243
206typedef struct etp_worker 244typedef struct etp_worker
207{ 245{
208 /* locked by wrklock */ 246 /* locked by wrklock */
209 struct etp_worker *prev, *next; 247 struct etp_worker *prev, *next;
210 248
211 thread_t tid; 249 xthread_t tid;
212 250
213 /* locked by reslock, reqlock or wrklock */ 251 /* locked by reslock, reqlock or wrklock */
214 ETP_REQ *req; /* currently processed request */ 252 ETP_REQ *req; /* currently processed request */
215 253
216 ETP_WORKER_COMMON 254 ETP_WORKER_COMMON
… …
569} 607}
570 608
571static void etp_set_max_poll_time (double nseconds) 609static void etp_set_max_poll_time (double nseconds)
572{ 610{
573 if (WORDACCESS_UNSAFE) X_LOCK (reslock); 611 if (WORDACCESS_UNSAFE) X_LOCK (reslock);
574 max_poll_time = nseconds; 612 max_poll_time = nseconds * EIO_TICKS;
575 if (WORDACCESS_UNSAFE) X_UNLOCK (reslock); 613 if (WORDACCESS_UNSAFE) X_UNLOCK (reslock);
576} 614}
577 615
578static void etp_set_max_poll_reqs (unsigned int maxreqs) 616static void etp_set_max_poll_reqs (unsigned int maxreqs)
579{ 617{
… …
778 816
779 return res; 817 return res;
780} 818}
781#endif 819#endif
782 820
783#ifndef HAVE_FUTIMES 821#ifndef HAVE_UTIMES
784 822
785# undef utimes 823# undef utimes
786# undef futimes
787# define utimes(path,times) eio__utimes (path, times) 824# define utimes(path,times) eio__utimes (path, times)
788# define futimes(fd,times) eio__futimes (fd, times)
789 825
790static int 826static int
791eio__utimes (const char *filename, const struct timeval times[2]) 827eio__utimes (const char *filename, const struct timeval times[2])
792{ 828{
793 if (times) 829 if (times)
… …
800 return utime (filename, &buf); 836 return utime (filename, &buf);
801 } 837 }
802 else 838 else
803 return utime (filename, 0); 839 return utime (filename, 0);
804} 840}
841
842#endif
843
844#ifndef HAVE_FUTIMES
845
846# undef futimes
847# define futimes(fd,times) eio__futimes (fd, times)
805 848
806static int eio__futimes (int fd, const struct timeval tv[2]) 849static int eio__futimes (int fd, const struct timeval tv[2])
807{ 850{
808 errno = ENOSYS; 851 errno = ENOSYS;
809 return -1; 852 return -1;
… …
838 if (!res || errno != ENOSYS) 881 if (!res || errno != ENOSYS)
839 return res; 882 return res;
840#endif 883#endif
841 884
842 /* even though we could play tricks with the flags, it's better to always 885 /* even though we could play tricks with the flags, it's better to always
843 * call fdatasync, as thta matches the expectation of it's users best */ 886 * call fdatasync, as that matches the expectation of its users best */
844 return fdatasync (fd); 887 return fdatasync (fd);
845} 888}
846 889
847#if !HAVE_READAHEAD 890#if !HAVE_READAHEAD
848# undef readahead 891# undef readahead
… …
880 923
881#if HAVE_SENDFILE 924#if HAVE_SENDFILE
882# if __linux 925# if __linux
883 res = sendfile (ofd, ifd, &offset, count); 926 res = sendfile (ofd, ifd, &offset, count);
884 927
885# elif __freebsd 928# elif __FreeBSD__
886 /* 929 /*
887 * Of course, the freebsd sendfile is a dire hack with no thoughts 930 * Of course, the freebsd sendfile is a dire hack with no thoughts
888 * wasted on making it similar to other I/O functions. 931 * wasted on making it similar to other I/O functions.
889 */ 932 */
890 { 933 {
891 off_t sbytes; 934 off_t sbytes;
892 res = sendfile (ifd, ofd, offset, count, 0, &sbytes, 0); 935 res = sendfile (ifd, ofd, offset, count, 0, &sbytes, 0);
893 936
894 if (res < 0 && sbytes) 937 #if 0 /* according to the manpage, this is correct, but broken behaviour */
895 /* maybe only on EAGAIN: as usual, the manpage leaves you guessing */ 938 /* freebsd' sendfile will return 0 on success */
939 /* freebsd 8 documents it as only setting *sbytes on EINTR and EAGAIN, but */
940 /* not on e.g. EIO or EPIPE - sounds broken */
941 if ((res < 0 && (errno == EAGAIN || errno == EINTR) && sbytes) || res == 0)
942 res = sbytes;
943 #endif
944
945 /* according to source inspection, this is correct, and useful behaviour */
946 if (sbytes)
947 res = sbytes;
948 }
949
950# elif defined (__APPLE__)
951
952 {
953 off_t sbytes = count;
954 res = sendfile (ifd, ofd, offset, &sbytes, 0, 0);
955
956 /* according to the manpage, sbytes is always valid */
957 if (sbytes)
896 res = sbytes; 958 res = sbytes;
897 } 959 }
898 960
899# elif __hpux 961# elif __hpux
900 res = sendfile (ofd, ifd, offset, count, 0, 0); 962 res = sendfile (ofd, ifd, offset, count, 0, 0);
… …
914 if (res < 0 && sbytes) 976 if (res < 0 && sbytes)
915 res = sbytes; 977 res = sbytes;
916 } 978 }
917 979
918# endif 980# endif
981
982#elif defined (_WIN32)
983
984 /* does not work, just for documentation of what would need to be done */
985 {
986 HANDLE h = TO_SOCKET (ifd);
987 SetFilePointer (h, offset, 0, FILE_BEGIN);
988 res = TransmitFile (TO_SOCKET (ofd), h, count, 0, 0, 0, 0);
989 }
990
919#else 991#else
920 res = -1; 992 res = -1;
921 errno = ENOSYS; 993 errno = ENOSYS;
922#endif 994#endif
923 995
924 if (res < 0 996 if (res < 0
925 && (errno == ENOSYS || errno == EINVAL || errno == ENOTSOCK 997 && (errno == ENOSYS || errno == EINVAL || errno == ENOTSOCK
998 /* BSDs */
999#ifdef ENOTSUP /* sigh, if the steenking pile called openbsd would only try to at least compile posix code... */
1000 || errno == ENOTSUP
1001#endif
1002 || errno == EOPNOTSUPP /* BSDs */
926#if __solaris 1003#if __solaris
927 || errno == EAFNOSUPPORT || errno == EPROTOTYPE 1004 || errno == EAFNOSUPPORT || errno == EPROTOTYPE
928#endif 1005#endif
929 ) 1006 )
930 ) 1007 )
… …
961 } 1038 }
962 1039
963 return res; 1040 return res;
964} 1041}
965 1042
1043static signed char
1044eio_dent_cmp (const eio_dirent *a, const eio_dirent *b)
1045{
1046 return a->score - b->score ? a->score - b->score /* works because our signed char is always 0..100 */
1047 : a->inode < b->inode ? -1 : a->inode > b->inode ? 1 : 0;
1048}
1049
1050#define EIO_DENT_CMP(i,op,j) eio_dent_cmp (&i, &j) op 0
1051
1052#define EIO_SORT_CUTOFF 30 /* quite high, but performs well on many filesystems */
1053#define EIO_SORT_FAST 60 /* when to only use insertion sort */
1054
1055static void
1056eio_dent_radix_sort (eio_dirent *dents, int size, signed char score_bits, ino_t inode_bits)
1057{
1058 unsigned char bits [9 + sizeof (ino_t) * 8];
1059 unsigned char *bit = bits;
1060
1061 assert (CHAR_BIT == 8);
1062 assert (sizeof (eio_dirent) * 8 < 256);
1063 assert (offsetof (eio_dirent, inode)); /* we use 0 as sentinel */
1064 assert (offsetof (eio_dirent, score)); /* we use 0 as sentinel */
1065
1066 if (size <= EIO_SORT_FAST)
1067 return;
1068
1069 /* first prepare an array of bits to test in our radix sort */
1070 /* try to take endianness into account, as well as differences in ino_t sizes */
1071 /* inode_bits must contain all inodes ORed together */
1072 /* which is used to skip bits that are 0 everywhere, which is very common */
1073 {
1074 ino_t endianness;
1075 int i, j;
1076
1077 /* we store the byte offset of byte n into byte n of "endianness" */
1078 for (i = 0; i < sizeof (ino_t); ++i)
1079 ((unsigned char *)&endianness)[i] = i;
1080
1081 *bit++ = 0;
1082
1083 for (i = 0; i < sizeof (ino_t); ++i)
1084 {
1085 /* shifting off the byte offsets out of "endianness" */
1086 int offs = (offsetof (eio_dirent, inode) + (endianness & 0xff)) * 8;
1087 endianness >>= 8;
1088
1089 for (j = 0; j < 8; ++j)
1090 if (inode_bits & (((ino_t)1) << (i * 8 + j)))
1091 *bit++ = offs + j;
1092 }
1093
1094 for (j = 0; j < 8; ++j)
1095 if (score_bits & (1 << j))
1096 *bit++ = offsetof (eio_dirent, score) * 8 + j;
1097 }
1098
1099 /* now actually do the sorting (a variant of MSD radix sort) */
1100 {
1101 eio_dirent *base_stk [9 + sizeof (ino_t) * 8], *base;
1102 eio_dirent *end_stk [9 + sizeof (ino_t) * 8], *end;
1103 unsigned char *bit_stk [9 + sizeof (ino_t) * 8];
1104 int stk_idx = 0;
1105
1106 base_stk [stk_idx] = dents;
1107 end_stk [stk_idx] = dents + size;
1108 bit_stk [stk_idx] = bit - 1;
1109
1110 do
1111 {
1112 base = base_stk [stk_idx];
1113 end = end_stk [stk_idx];
1114 bit = bit_stk [stk_idx];
1115
1116 for (;;)
1117 {
1118 unsigned char O = *bit >> 3;
1119 unsigned char M = 1 << (*bit & 7);
1120
1121 eio_dirent *a = base;
1122 eio_dirent *b = end;
1123
1124 if (b - a < EIO_SORT_CUTOFF)
1125 break;
1126
1127 /* now bit-partition the array on the bit */
1128 /* this ugly asymmetric loop seems to perform much better than typical */
1129 /* partition algos found in the literature */
1130 do
1131 if (!(((unsigned char *)a)[O] & M))
1132 ++a;
1133 else if (!(((unsigned char *)--b)[O] & M))
1134 {
1135 eio_dirent tmp = *a; *a = *b; *b = tmp;
1136 ++a;
1137 }
1138 while (b > a);
1139
1140 /* next bit, or stop, if no bits left in this path */
1141 if (!*--bit)
1142 break;
1143
1144 base_stk [stk_idx] = a;
1145 end_stk [stk_idx] = end;
1146 bit_stk [stk_idx] = bit;
1147 ++stk_idx;
1148
1149 end = a;
1150 }
1151 }
1152 while (stk_idx--);
1153 }
1154}
1155
1156static void
1157eio_dent_insertion_sort (eio_dirent *dents, int size)
1158{
1159 /* first move the smallest element to the front, to act as a sentinel */
1160 {
1161 int i;
1162 eio_dirent *min = dents;
1163
1164 /* the radix pre-pass ensures that the minimum element is in the first EIO_SORT_CUTOFF + 1 elements */
1165 for (i = size > EIO_SORT_FAST ? EIO_SORT_CUTOFF + 1 : size; --i; )
1166 if (EIO_DENT_CMP (dents [i], <, *min))
1167 min = &dents [i];
1168
1169 /* swap elements 0 and j (minimum) */
1170 {
1171 eio_dirent tmp = *dents; *dents = *min; *min = tmp;
1172 }
1173 }
1174
1175 /* then do standard insertion sort, assuming that all elements are >= dents [0] */
1176 {
1177 eio_dirent *i, *j;
1178
1179 for (i = dents + 1; i < dents + size; ++i)
1180 {
1181 eio_dirent value = *i;
1182
1183 for (j = i - 1; EIO_DENT_CMP (*j, >, value); --j)
1184 j [1] = j [0];
1185
1186 j [1] = value;
1187 }
1188 }
1189}
1190
1191static void
1192eio_dent_sort (eio_dirent *dents, int size, signed char score_bits, ino_t inode_bits)
1193{
1194 if (size <= 1)
1195 return; /* our insertion sort relies on size > 0 */
1196
1197 /* first we use a radix sort, but only for dirs >= EIO_SORT_FAST */
1198 /* and stop sorting when the partitions are <= EIO_SORT_CUTOFF */
1199 eio_dent_radix_sort (dents, size, score_bits, inode_bits);
1200
1201 /* use an insertion sort at the end, or for small arrays, */
1202 /* as insertion sort is more efficient for small partitions */
1203 eio_dent_insertion_sort (dents, size);
1204}
1205
966/* read a full directory */ 1206/* read a full directory */
967static void 1207static void
968eio__scandir (eio_req *req, etp_worker *self) 1208eio__scandir (eio_req *req, etp_worker *self)
969{ 1209{
970 DIR *dirp; 1210 DIR *dirp;
971 EIO_STRUCT_DIRENT *entp; 1211 EIO_STRUCT_DIRENT *entp;
972 char *name, *names; 1212 char *name, *names;
973 int memlen = 4096; 1213 int namesalloc = 4096;
974 int memofs = 0; 1214 int namesoffs = 0;
1215 int flags = req->int1;
1216 eio_dirent *dents = 0;
1217 int dentalloc = 128;
975 int res = 0; 1218 int dentoffs = 0;
1219 ino_t inode_bits = 0;
1220
1221 req->result = -1;
1222
1223 if (!(flags & EIO_READDIR_DENTS))
1224 flags &= ~(EIO_READDIR_DIRS_FIRST | EIO_READDIR_STAT_ORDER);
976 1225
977 X_LOCK (wrklock); 1226 X_LOCK (wrklock);
978 /* the corresponding closedir is in ETP_WORKER_CLEAR */ 1227 /* the corresponding closedir is in ETP_WORKER_CLEAR */
979 self->dirp = dirp = opendir (req->ptr1); 1228 self->dirp = dirp = opendir (req->ptr1);
980 req->flags |= EIO_FLAG_PTR2_FREE; 1229 req->flags |= EIO_FLAG_PTR1_FREE | EIO_FLAG_PTR2_FREE;
1230 req->ptr1 = dents = flags ? malloc (dentalloc * sizeof (eio_dirent)) : 0;
981 req->ptr2 = names = malloc (memlen); 1231 req->ptr2 = names = malloc (namesalloc);
982 X_UNLOCK (wrklock); 1232 X_UNLOCK (wrklock);
983 1233
984 if (dirp && names) 1234 if (dirp && names && (!flags || dents))
985 for (;;) 1235 for (;;)
986 { 1236 {
987 errno = 0; 1237 errno = 0;
988 entp = readdir (dirp); 1238 entp = readdir (dirp);
989 1239
990 if (!entp) 1240 if (!entp)
1241 {
1242 if (errno)
1243 break;
1244
1245 /* sort etc. */
1246 req->int1 = flags;
1247 req->result = dentoffs;
1248
1249 if (flags & EIO_READDIR_STAT_ORDER)
1250 eio_dent_sort (dents, dentoffs, 0, inode_bits); /* sort by inode exclusively */
1251 else if (flags & EIO_READDIR_DIRS_FIRST)
1252 if (flags & EIO_READDIR_FOUND_UNKNOWN)
1253 eio_dent_sort (dents, dentoffs, 7, inode_bits); /* sort by score and inode */
1254 else
1255 {
1256 /* in this case, all is known, and we just put dirs first and sort them */
1257 eio_dirent *oth = dents + dentoffs;
1258 eio_dirent *dir = dents;
1259
1260 /* now partition dirs to the front, and non-dirs to the back */
1261 /* by walking from both sides and swapping if necessary */
1262 /* also clear score, so it doesn't influence sorting */
1263 while (oth > dir)
1264 {
1265 if (dir->type == EIO_DT_DIR)
1266 ++dir;
1267 else if ((--oth)->type == EIO_DT_DIR)
1268 {
1269 eio_dirent tmp = *dir; *dir = *oth; *oth = tmp;
1270
1271 ++dir;
1272 }
1273 }
1274
1275 /* now sort the dirs only */
1276 eio_dent_sort (dents, dir - dents, 0, inode_bits);
1277 }
1278
991 break; 1279 break;
1280 }
992 1281
1282 /* now add the entry to our list(s) */
993 name = entp->d_name; 1283 name = entp->d_name;
994 1284
1285 /* skip . and .. entries */
995 if (name [0] != '.' || (name [1] && (name [1] != '.' || name [2]))) 1286 if (name [0] != '.' || (name [1] && (name [1] != '.' || name [2])))
996 { 1287 {
997 int len = strlen (name) + 1; 1288 int len = D_NAMLEN (entp) + 1;
998 1289
999 res++; 1290 while (expect_false (namesoffs + len > namesalloc))
1000
1001 while (memofs + len > memlen)
1002 { 1291 {
1003 memlen *= 2; 1292 namesalloc *= 2;
1004 X_LOCK (wrklock); 1293 X_LOCK (wrklock);
1005 req->ptr2 = names = realloc (names, memlen); 1294 req->ptr2 = names = realloc (names, namesalloc);
1006 X_UNLOCK (wrklock); 1295 X_UNLOCK (wrklock);
1007 1296
1008 if (!names) 1297 if (!names)
1009 break; 1298 break;
1010 } 1299 }
1011 1300
1012 memcpy (names + memofs, name, len); 1301 memcpy (names + namesoffs, name, len);
1302
1303 if (dents)
1304 {
1305 struct eio_dirent *ent;
1306
1307 if (expect_false (dentoffs == dentalloc))
1308 {
1309 dentalloc *= 2;
1310 X_LOCK (wrklock);
1311 req->ptr1 = dents = realloc (dents, dentalloc * sizeof (eio_dirent));
1312 X_UNLOCK (wrklock);
1313
1314 if (!dents)
1315 break;
1316 }
1317
1318 ent = dents + dentoffs;
1319
1320 ent->nameofs = namesoffs; /* rather dirtily we store the offset in the pointer */
1321 ent->namelen = len - 1;
1322 ent->inode = D_INO (entp);
1323
1324 inode_bits |= ent->inode;
1325
1326 switch (D_TYPE (entp))
1327 {
1328 default:
1329 ent->type = EIO_DT_UNKNOWN;
1330 flags |= EIO_READDIR_FOUND_UNKNOWN;
1331 break;
1332
1333 #ifdef DT_FIFO
1334 case DT_FIFO: ent->type = EIO_DT_FIFO; break;
1335 #endif
1336 #ifdef DT_CHR
1337 case DT_CHR: ent->type = EIO_DT_CHR; break;
1338 #endif
1339 #ifdef DT_MPC
1340 case DT_MPC: ent->type = EIO_DT_MPC; break;
1341 #endif
1342 #ifdef DT_DIR
1343 case DT_DIR: ent->type = EIO_DT_DIR; break;
1344 #endif
1345 #ifdef DT_NAM
1346 case DT_NAM: ent->type = EIO_DT_NAM; break;
1347 #endif
1348 #ifdef DT_BLK
1349 case DT_BLK: ent->type = EIO_DT_BLK; break;
1350 #endif
1351 #ifdef DT_MPB
1352 case DT_MPB: ent->type = EIO_DT_MPB; break;
1353 #endif
1354 #ifdef DT_REG
1355 case DT_REG: ent->type = EIO_DT_REG; break;
1356 #endif
1357 #ifdef DT_NWK
1358 case DT_NWK: ent->type = EIO_DT_NWK; break;
1359 #endif
1360 #ifdef DT_CMP
1361 case DT_CMP: ent->type = EIO_DT_CMP; break;
1362 #endif
1363 #ifdef DT_LNK
1364 case DT_LNK: ent->type = EIO_DT_LNK; break;
1365 #endif
1366 #ifdef DT_SOCK
1367 case DT_SOCK: ent->type = EIO_DT_SOCK; break;
1368 #endif
1369 #ifdef DT_DOOR
1370 case DT_DOOR: ent->type = EIO_DT_DOOR; break;
1371 #endif
1372 #ifdef DT_WHT
1373 case DT_WHT: ent->type = EIO_DT_WHT; break;
1374 #endif
1375 }
1376
1377 ent->score = 7;
1378
1379 if (flags & EIO_READDIR_DIRS_FIRST)
1380 {
1381 if (ent->type == EIO_DT_UNKNOWN)
1382 {
1383 if (*name == '.') /* leading dots are likely directories, and, in any case, rare */
1384 ent->score = 1;
1385 else if (!strchr (name, '.')) /* absense of dots indicate likely dirs */
1386 ent->score = len <= 2 ? 4 - len : len <= 4 ? 4 : len <= 7 ? 5 : 6; /* shorter == more likely dir, but avoid too many classes */
1387 }
1388 else if (ent->type == EIO_DT_DIR)
1389 ent->score = 0;
1390 }
1391 }
1392
1013 memofs += len; 1393 namesoffs += len;
1394 ++dentoffs;
1395 }
1396
1397 if (EIO_CANCELLED (req))
1398 {
1399 errno = ECANCELED;
1400 break;
1014 } 1401 }
1015 } 1402 }
1016
1017 if (errno)
1018 res = -1;
1019
1020 req->result = res;
1021} 1403}
1404
1405#ifdef PAGESIZE
1406# define eio_pagesize() PAGESIZE
1407#else
1408static intptr_t
1409eio_pagesize (void)
1410{
1411 static intptr_t page;
1412
1413 if (!page)
1414 page = sysconf (_SC_PAGESIZE);
1415
1416 return page;
1417}
1418#endif
1419
1420static void
1421eio_page_align (void **addr, size_t *length)
1422{
1423 intptr_t mask = eio_pagesize () - 1;
1424
1425 /* round down addr */
1426 intptr_t adj = mask & (intptr_t)*addr;
1427
1428 *addr = (void *)((intptr_t)*addr - adj);
1429 *length += adj;
1430
1431 /* round up length */
1432 *length = (*length + mask) & ~mask;
1433}
1434
1435#if !_POSIX_MEMLOCK
1436# define eio__mlockall(a) ((errno = ENOSYS), -1)
1437#else
1438
1439static int
1440eio__mlockall (int flags)
1441{
1442 #if __GLIBC__ == 2 && __GLIBC_MINOR__ <= 7
1443 extern int mallopt (int, int);
1444 mallopt (-6, 238); /* http://bugs.debian.org/cgi-bin/bugreport.cgi?bug=473812 */
1445 #endif
1446
1447 if (EIO_MCL_CURRENT != MCL_CURRENT
1448 || EIO_MCL_FUTURE != MCL_FUTURE)
1449 {
1450 flags = 0
1451 | (flags & EIO_MCL_CURRENT ? MCL_CURRENT : 0)
1452 | (flags & EIO_MCL_FUTURE ? MCL_FUTURE : 0);
1453 }
1454
1455 return mlockall (flags);
1456}
1457#endif
1458
1459#if !_POSIX_MEMLOCK_RANGE
1460# define eio__mlock(a,b) ((errno = ENOSYS), -1)
1461#else
1462
1463static int
1464eio__mlock (void *addr, size_t length)
1465{
1466 eio_page_align (&addr, &length);
1467
1468 return mlock (addr, length);
1469}
1470
1471#endif
1022 1472
1023#if !(_POSIX_MAPPED_FILES && _POSIX_SYNCHRONIZED_IO) 1473#if !(_POSIX_MAPPED_FILES && _POSIX_SYNCHRONIZED_IO)
1024# undef msync
1025# define msync(a,b,c) ((errno = ENOSYS), -1) 1474# define eio__msync(a,b,c) ((errno = ENOSYS), -1)
1475#else
1476
1477int
1478eio__msync (void *mem, size_t len, int flags)
1479{
1480 eio_page_align (&mem, &len);
1481
1482 if (EIO_MS_ASYNC != MS_SYNC
1483 || EIO_MS_INVALIDATE != MS_INVALIDATE
1484 || EIO_MS_SYNC != MS_SYNC)
1485 {
1486 flags = 0
1487 | (flags & EIO_MS_ASYNC ? MS_ASYNC : 0)
1488 | (flags & EIO_MS_INVALIDATE ? MS_INVALIDATE : 0)
1489 | (flags & EIO_MS_SYNC ? MS_SYNC : 0);
1490 }
1491
1492 return msync (mem, len, flags);
1493}
1494
1026#endif 1495#endif
1027 1496
1028int 1497int
1029eio__mtouch (void *mem, size_t len, int flags) 1498eio__mtouch (void *mem, size_t len, int flags)
1030{ 1499{
1500 eio_page_align (&mem, &len);
1501
1502 {
1031 intptr_t addr = (intptr_t)mem; 1503 intptr_t addr = (intptr_t)mem;
1032 intptr_t end = addr + len; 1504 intptr_t end = addr + len;
1033#ifdef PAGESIZE 1505 intptr_t page = eio_pagesize ();
1034 const intptr_t page = PAGESIZE;
1035#else
1036 static intptr_t page;
1037 1506
1038 if (!page)
1039 page = sysconf (_SC_PAGESIZE);
1040#endif
1041
1042 addr &= ~(page - 1); /* assume page size is always a power of two */
1043
1044 if (addr < end) 1507 if (addr < end)
1045 if (flags) /* modify */ 1508 if (flags & EIO_MT_MODIFY) /* modify */
1046 do { *((volatile sig_atomic_t *)addr) |= 0; } while ((addr += page) < len); 1509 do { *((volatile sig_atomic_t *)addr) |= 0; } while ((addr += page) < len);
1047 else 1510 else
1048 do { *((volatile sig_atomic_t *)addr) ; } while ((addr += page) < len); 1511 do { *((volatile sig_atomic_t *)addr) ; } while ((addr += page) < len);
1512 }
1049 1513
1050 return 0; 1514 return 0;
1051} 1515}
1052 1516
1053/*****************************************************************************/ 1517/*****************************************************************************/
… …
1176 return 0; \ 1640 return 0; \
1177 } 1641 }
1178 1642
1179static void eio_execute (etp_worker *self, eio_req *req) 1643static void eio_execute (etp_worker *self, eio_req *req)
1180{ 1644{
1181 errno = 0;
1182
1183 switch (req->type) 1645 switch (req->type)
1184 { 1646 {
1185 case EIO_READ: ALLOC (req->size); 1647 case EIO_READ: ALLOC (req->size);
1186 req->result = req->offs >= 0 1648 req->result = req->offs >= 0
1187 ? pread (req->int1, req->ptr2, req->size, req->offs) 1649 ? pread (req->int1, req->ptr2, req->size, req->offs)
… …
1198 case EIO_LSTAT: ALLOC (sizeof (EIO_STRUCT_STAT)); 1660 case EIO_LSTAT: ALLOC (sizeof (EIO_STRUCT_STAT));
1199 req->result = lstat (req->ptr1, (EIO_STRUCT_STAT *)req->ptr2); break; 1661 req->result = lstat (req->ptr1, (EIO_STRUCT_STAT *)req->ptr2); break;
1200 case EIO_FSTAT: ALLOC (sizeof (EIO_STRUCT_STAT)); 1662 case EIO_FSTAT: ALLOC (sizeof (EIO_STRUCT_STAT));
1201 req->result = fstat (req->int1, (EIO_STRUCT_STAT *)req->ptr2); break; 1663 req->result = fstat (req->int1, (EIO_STRUCT_STAT *)req->ptr2); break;
1202 1664
1665 case EIO_STATVFS: ALLOC (sizeof (EIO_STRUCT_STATVFS));
1666 req->result = statvfs (req->ptr1, (EIO_STRUCT_STATVFS *)req->ptr2); break;
1667 case EIO_FSTATVFS: ALLOC (sizeof (EIO_STRUCT_STATVFS));
1668 req->result = fstatvfs (req->int1, (EIO_STRUCT_STATVFS *)req->ptr2); break;
1669
1203 case EIO_CHOWN: req->result = chown (req->ptr1, req->int2, req->int3); break; 1670 case EIO_CHOWN: req->result = chown (req->ptr1, req->int2, req->int3); break;
1204 case EIO_FCHOWN: req->result = fchown (req->int1, req->int2, req->int3); break; 1671 case EIO_FCHOWN: req->result = fchown (req->int1, req->int2, req->int3); break;
1205 case EIO_CHMOD: req->result = chmod (req->ptr1, (mode_t)req->int2); break; 1672 case EIO_CHMOD: req->result = chmod (req->ptr1, (mode_t)req->int2); break;
1206 case EIO_FCHMOD: req->result = fchmod (req->int1, (mode_t)req->int2); break; 1673 case EIO_FCHMOD: req->result = fchmod (req->int1, (mode_t)req->int2); break;
1207 case EIO_TRUNCATE: req->result = truncate (req->ptr1, req->offs); break; 1674 case EIO_TRUNCATE: req->result = truncate (req->ptr1, req->offs); break;
… …
1214 case EIO_RMDIR: req->result = rmdir (req->ptr1); break; 1681 case EIO_RMDIR: req->result = rmdir (req->ptr1); break;
1215 case EIO_MKDIR: req->result = mkdir (req->ptr1, (mode_t)req->int2); break; 1682 case EIO_MKDIR: req->result = mkdir (req->ptr1, (mode_t)req->int2); break;
1216 case EIO_RENAME: req->result = rename (req->ptr1, req->ptr2); break; 1683 case EIO_RENAME: req->result = rename (req->ptr1, req->ptr2); break;
1217 case EIO_LINK: req->result = link (req->ptr1, req->ptr2); break; 1684 case EIO_LINK: req->result = link (req->ptr1, req->ptr2); break;
1218 case EIO_SYMLINK: req->result = symlink (req->ptr1, req->ptr2); break; 1685 case EIO_SYMLINK: req->result = symlink (req->ptr1, req->ptr2); break;
1219 case EIO_MKNOD: req->result = mknod (req->ptr1, (mode_t)req->int2, (dev_t)req->int3); break; 1686 case EIO_MKNOD: req->result = mknod (req->ptr1, (mode_t)req->int2, (dev_t)req->offs); break;
1220 1687
1221 case EIO_READLINK: ALLOC (NAME_MAX); 1688 case EIO_READLINK: ALLOC (PATH_MAX);
1222 req->result = readlink (req->ptr1, req->ptr2, NAME_MAX); break; 1689 req->result = readlink (req->ptr1, req->ptr2, PATH_MAX); break;
1223 1690
1224 case EIO_SYNC: req->result = 0; sync (); break; 1691 case EIO_SYNC: req->result = 0; sync (); break;
1225 case EIO_FSYNC: req->result = fsync (req->int1); break; 1692 case EIO_FSYNC: req->result = fsync (req->int1); break;
1226 case EIO_FDATASYNC: req->result = fdatasync (req->int1); break; 1693 case EIO_FDATASYNC: req->result = fdatasync (req->int1); break;
1227 case EIO_MSYNC: req->result = msync (req->ptr2, req->size, req->int1); break; 1694 case EIO_MSYNC: req->result = eio__msync (req->ptr2, req->size, req->int1); break;
1228 case EIO_MTOUCH: req->result = eio__mtouch (req->ptr2, req->size, req->int1); break; 1695 case EIO_MTOUCH: req->result = eio__mtouch (req->ptr2, req->size, req->int1); break;
1696 case EIO_MLOCK: req->result = eio__mlock (req->ptr2, req->size); break;
1697 case EIO_MLOCKALL: req->result = eio__mlockall (req->int1); break;
1229 case EIO_SYNC_FILE_RANGE: req->result = eio__sync_file_range (req->int1, req->offs, req->size, req->int2); break; 1698 case EIO_SYNC_FILE_RANGE: req->result = eio__sync_file_range (req->int1, req->offs, req->size, req->int2); break;
1230 1699
1231 case EIO_READDIR: eio__scandir (req, self); break; 1700 case EIO_READDIR: eio__scandir (req, self); break;
1232 1701
1233 case EIO_BUSY: 1702 case EIO_BUSY:
1234#ifdef _WIN32 1703#ifdef _WIN32
1235 Sleep (req->nv1 * 1000.); 1704 Sleep (req->nv1 * 1e3);
1236#else 1705#else
1237 { 1706 {
1238 struct timeval tv; 1707 struct timeval tv;
1239 1708
1240 tv.tv_sec = req->nv1; 1709 tv.tv_sec = req->nv1;
1241 tv.tv_usec = (req->nv1 - tv.tv_sec) * 1000000.; 1710 tv.tv_usec = (req->nv1 - tv.tv_sec) * 1e6;
1242 1711
1243 req->result = select (0, 0, 0, 0, &tv); 1712 req->result = select (0, 0, 0, 0, &tv);
1244 } 1713 }
1245#endif 1714#endif
1246 break; 1715 break;
… …
1261 times = tv; 1730 times = tv;
1262 } 1731 }
1263 else 1732 else
1264 times = 0; 1733 times = 0;
1265 1734
1266
1267 req->result = req->type == EIO_FUTIME 1735 req->result = req->type == EIO_FUTIME
1268 ? futimes (req->int1, times) 1736 ? futimes (req->int1, times)
1269 : utimes (req->ptr1, times); 1737 : utimes (req->ptr1, times);
1270 } 1738 }
1271 break; 1739 break;
… …
1280 case EIO_CUSTOM: 1748 case EIO_CUSTOM:
1281 ((void (*)(eio_req *))req->feed) (req); 1749 ((void (*)(eio_req *))req->feed) (req);
1282 break; 1750 break;
1283 1751
1284 default: 1752 default:
1753 errno = ENOSYS;
1285 req->result = -1; 1754 req->result = -1;
1286 break; 1755 break;
1287 } 1756 }
1288 1757
1289 req->errorno = errno; 1758 req->errorno = errno;
… …
1319eio_req *eio_mtouch (void *addr, size_t length, int flags, int pri, eio_cb cb, void *data) 1788eio_req *eio_mtouch (void *addr, size_t length, int flags, int pri, eio_cb cb, void *data)
1320{ 1789{
1321 REQ (EIO_MTOUCH); req->ptr2 = addr; req->size = length; req->int1 = flags; SEND; 1790 REQ (EIO_MTOUCH); req->ptr2 = addr; req->size = length; req->int1 = flags; SEND;
1322} 1791}
1323 1792
1793eio_req *eio_mlock (void *addr, size_t length, int pri, eio_cb cb, void *data)
1794{
1795 REQ (EIO_MLOCK); req->ptr2 = addr; req->size = length; SEND;
1796}
1797
1798eio_req *eio_mlockall (int flags, int pri, eio_cb cb, void *data)
1799{
1800 REQ (EIO_MLOCKALL); req->int1 = flags; SEND;
1801}
1802
1324eio_req *eio_sync_file_range (int fd, off_t offset, size_t nbytes, unsigned int flags, int pri, eio_cb cb, void *data) 1803eio_req *eio_sync_file_range (int fd, off_t offset, size_t nbytes, unsigned int flags, int pri, eio_cb cb, void *data)
1325{ 1804{
1326 REQ (EIO_SYNC_FILE_RANGE); req->int1 = fd; req->offs = offset; req->size = nbytes; req->int2 = flags; SEND; 1805 REQ (EIO_SYNC_FILE_RANGE); req->int1 = fd; req->offs = offset; req->size = nbytes; req->int2 = flags; SEND;
1327} 1806}
1328 1807
… …
1352} 1831}
1353 1832
1354eio_req *eio_fstat (int fd, int pri, eio_cb cb, void *data) 1833eio_req *eio_fstat (int fd, int pri, eio_cb cb, void *data)
1355{ 1834{
1356 REQ (EIO_FSTAT); req->int1 = fd; SEND; 1835 REQ (EIO_FSTAT); req->int1 = fd; SEND;
1836}
1837
1838eio_req *eio_fstatvfs (int fd, int pri, eio_cb cb, void *data)
1839{
1840 REQ (EIO_FSTATVFS); req->int1 = fd; SEND;
1357} 1841}
1358 1842
1359eio_req *eio_futime (int fd, double atime, double mtime, int pri, eio_cb cb, void *data) 1843eio_req *eio_futime (int fd, double atime, double mtime, int pri, eio_cb cb, void *data)
1360{ 1844{
1361 REQ (EIO_FUTIME); req->int1 = fd; req->nv1 = atime; req->nv2 = mtime; SEND; 1845 REQ (EIO_FUTIME); req->int1 = fd; req->nv1 = atime; req->nv2 = mtime; SEND;
… …
1435eio_req *eio_lstat (const char *path, int pri, eio_cb cb, void *data) 1919eio_req *eio_lstat (const char *path, int pri, eio_cb cb, void *data)
1436{ 1920{
1437 return eio__1path (EIO_LSTAT, path, pri, cb, data); 1921 return eio__1path (EIO_LSTAT, path, pri, cb, data);
1438} 1922}
1439 1923
1924eio_req *eio_statvfs (const char *path, int pri, eio_cb cb, void *data)
1925{
1926 return eio__1path (EIO_STATVFS, path, pri, cb, data);
1927}
1928
1440eio_req *eio_unlink (const char *path, int pri, eio_cb cb, void *data) 1929eio_req *eio_unlink (const char *path, int pri, eio_cb cb, void *data)
1441{ 1930{
1442 return eio__1path (EIO_UNLINK, path, pri, cb, data); 1931 return eio__1path (EIO_UNLINK, path, pri, cb, data);
1443} 1932}
1444 1933
1445eio_req *eio_rmdir (const char *path, int pri, eio_cb cb, void *data) 1934eio_req *eio_rmdir (const char *path, int pri, eio_cb cb, void *data)
1446{ 1935{
1447 return eio__1path (EIO_RMDIR, path, pri, cb, data); 1936 return eio__1path (EIO_RMDIR, path, pri, cb, data);
1448} 1937}
1449 1938
1450eio_req *eio_readdir (const char *path, int pri, eio_cb cb, void *data) 1939eio_req *eio_readdir (const char *path, int flags, int pri, eio_cb cb, void *data)
1451{ 1940{
1452 return eio__1path (EIO_READDIR, path, pri, cb, data); 1941 REQ (EIO_READDIR); PATH; req->int1 = flags; SEND;
1453} 1942}
1454 1943
1455eio_req *eio_mknod (const char *path, mode_t mode, dev_t dev, int pri, eio_cb cb, void *data) 1944eio_req *eio_mknod (const char *path, mode_t mode, dev_t dev, int pri, eio_cb cb, void *data)
1456{ 1945{
1457 REQ (EIO_MKNOD); PATH; req->int2 = (long)mode; req->int3 = (long)dev; SEND; 1946 REQ (EIO_MKNOD); PATH; req->int2 = (long)mode; req->offs = (off_t)dev; SEND;
1458} 1947}
1459 1948
1460static eio_req * 1949static eio_req *
1461eio__2path (int type, const char *path, const char *new_path, int pri, eio_cb cb, void *data) 1950eio__2path (int type, const char *path, const char *new_path, int pri, eio_cb cb, void *data)
1462{ 1951{
… …
1546/* misc garbage */ 2035/* misc garbage */
1547 2036
1548ssize_t eio_sendfile_sync (int ofd, int ifd, off_t offset, size_t count) 2037ssize_t eio_sendfile_sync (int ofd, int ifd, off_t offset, size_t count)
1549{ 2038{
1550 etp_worker wrk; 2039 etp_worker wrk;
2040 ssize_t ret;
1551 2041
1552 wrk.dbuf = 0; 2042 wrk.dbuf = 0;
1553 2043
1554 eio__sendfile (ofd, ifd, offset, count, &wrk); 2044 ret = eio__sendfile (ofd, ifd, offset, count, &wrk);
1555 2045
1556 if (wrk.dbuf) 2046 if (wrk.dbuf)
1557 free (wrk.dbuf); 2047 free (wrk.dbuf);
1558}
1559 2048
2049 return ret;
2050}
2051

Diff Legend

– Removed lines
+ Added lines
< Changed lines
> Changed lines