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

Comparing libeio/eio.c (file contents):
Revision 1.27 by root, Wed Oct 22 18:15:36 2008 UTC vs.
Revision 1.90 by root, Sat Jul 16 16:46:10 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 *
35 * and other provisions required by the GPL. If you do not delete the 35 * and other provisions required by the GPL. If you do not delete the
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#ifndef _WIN32
41# include "config.h"
42#endif
43
40#include "eio.h" 44#include "eio.h"
45#include "ecb.h"
46
47#ifdef EIO_STACKSIZE
48# define XTHREAD_STACKSIZE EIO_STACKSIZE
49#endif
41#include "xthread.h" 50#include "xthread.h"
42 51
43#include <errno.h> 52#include <errno.h>
44#include <stddef.h> 53#include <stddef.h>
45#include <stdlib.h> 54#include <stdlib.h>
49#include <sys/stat.h> 58#include <sys/stat.h>
50#include <limits.h> 59#include <limits.h>
51#include <fcntl.h> 60#include <fcntl.h>
52#include <assert.h> 61#include <assert.h>
53 62
63/* intptr_t comes from unistd.h, says POSIX/UNIX/tradition */
64/* intptr_t only comes from stdint.h, says idiot openbsd coder */
65#if HAVE_STDINT_H
66# include <stdint.h>
67#endif
68
69#ifndef ECANCELED
70# define ECANCELED EDOM
71#endif
72#ifndef ELOOP
73# define ELOOP EDOM
74#endif
75
76static void eio_destroy (eio_req *req);
77
54#ifndef EIO_FINISH 78#ifndef EIO_FINISH
55# 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
56#endif 80#endif
57 81
58#ifndef EIO_DESTROY 82#ifndef EIO_DESTROY
61 85
62#ifndef EIO_FEED 86#ifndef EIO_FEED
63# 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)
64#endif 88#endif
65 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
66#ifdef _WIN32 101#ifdef _WIN32
67 102
68 /*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 fchmod(fd,mode) EIO_ENOSYS ()
117 #define chown(path,uid,gid) EIO_ENOSYS ()
118 #define fchown(fd,uid,gid) EIO_ENOSYS ()
119 #define truncate(path,offs) EIO_ENOSYS () /* far-miss: SetEndOfFile */
120 #define ftruncate(fd,offs) EIO_ENOSYS () /* near-miss: SetEndOfFile */
121 #define mknod(path,mode,dev) EIO_ENOSYS ()
122 #define sync() EIO_ENOSYS ()
123 #define readlink(path,buf,s) EIO_ENOSYS ()
124 #define statvfs(path,buf) EIO_ENOSYS ()
125 #define fstatvfs(fd,buf) EIO_ENOSYS ()
126
127 /* we could even stat and see if it exists */
128 static int
129 symlink (const char *old, const char *neu)
130 {
131 #if WINVER >= 0x0600
132 if (CreateSymbolicLink (neu, old, 1))
133 return 0;
134
135 if (CreateSymbolicLink (neu, old, 0))
136 return 0;
137 #endif
138
139 return EIO_ERRNO (ENOENT, -1);
140 }
141
142 /* POSIX API only */
143 #define CreateHardLink(neu,old) 0
144 #define CreateSymbolicLink(neu,old,flags) 0
145
146 struct statvfs
147 {
148 int dummy;
149 };
69 150
70#else 151#else
71 152
72# include "config.h"
73# include <sys/time.h> 153 #include <sys/time.h>
74# include <sys/select.h> 154 #include <sys/select.h>
155 #include <sys/statvfs.h>
75# include <unistd.h> 156 #include <unistd.h>
76# include <utime.h> 157 #include <utime.h>
77# include <signal.h> 158 #include <signal.h>
78# include <dirent.h> 159 #include <dirent.h>
79 160
161 #if _POSIX_MEMLOCK || _POSIX_MEMLOCK_RANGE || _POSIX_MAPPED_FILES
162 #include <sys/mman.h>
163 #endif
164
165 /* POSIX_SOURCE is useless on bsd's, and XOPEN_SOURCE is unreliable there, too */
166 #if __FreeBSD__ || defined __NetBSD__ || defined __OpenBSD__
167 #define _DIRENT_HAVE_D_TYPE /* sigh */
168 #define D_INO(de) (de)->d_fileno
169 #define D_NAMLEN(de) (de)->d_namlen
170 #elif __linux || defined d_ino || _XOPEN_SOURCE >= 600
171 #define D_INO(de) (de)->d_ino
172 #endif
173
174 #ifdef _D_EXACT_NAMLEN
175 #undef D_NAMLEN
176 #define D_NAMLEN(de) _D_EXACT_NAMLEN (de)
177 #endif
178
179 #ifdef _DIRENT_HAVE_D_TYPE
180 #define D_TYPE(de) (de)->d_type
181 #endif
182
80# ifndef EIO_STRUCT_DIRENT 183 #ifndef EIO_STRUCT_DIRENT
81# define EIO_STRUCT_DIRENT struct dirent 184 #define EIO_STRUCT_DIRENT struct dirent
82# endif 185 #endif
83 186
84#endif 187#endif
85 188
86#if HAVE_SENDFILE 189#if HAVE_SENDFILE
87# if __linux 190# if __linux
88# include <sys/sendfile.h> 191# include <sys/sendfile.h>
89# elif __freebsd 192# elif __FreeBSD__ || defined __APPLE__
90# include <sys/socket.h> 193# include <sys/socket.h>
91# include <sys/uio.h> 194# include <sys/uio.h>
92# elif __hpux 195# elif __hpux
93# include <sys/socket.h> 196# include <sys/socket.h>
94# elif __solaris /* not yet */ 197# elif __solaris
95# include <sys/sendfile.h> 198# include <sys/sendfile.h>
96# else 199# else
97# error sendfile support requested but not available 200# error sendfile support requested but not available
98# endif 201# endif
99#endif 202#endif
100 203
101/* number of seconds after which an idle threads exit */ 204#ifndef D_TYPE
102#define IDLE_TIMEOUT 10 205# define D_TYPE(de) 0
206#endif
207#ifndef D_INO
208# define D_INO(de) 0
209#endif
210#ifndef D_NAMLEN
211# define D_NAMLEN(de) strlen ((de)->d_name)
212#endif
103 213
104/* used for struct dirent, AIX doesn't provide it */ 214/* used for struct dirent, AIX doesn't provide it */
105#ifndef NAME_MAX 215#ifndef NAME_MAX
106# define NAME_MAX 4096 216# define NAME_MAX 4096
217#endif
218
219/* used for readlink etc. */
220#ifndef PATH_MAX
221# define PATH_MAX 4096
107#endif 222#endif
108 223
109/* buffer size for various temporary buffers */ 224/* buffer size for various temporary buffers */
110#define EIO_BUFSIZE 65536 225#define EIO_BUFSIZE 65536
111 226
117 errno = ENOMEM; \ 232 errno = ENOMEM; \
118 if (!eio_buf) \ 233 if (!eio_buf) \
119 return -1; 234 return -1;
120 235
121#define EIO_TICKS ((1000000 + 1023) >> 10) 236#define EIO_TICKS ((1000000 + 1023) >> 10)
122
123/*****************************************************************************/
124
125#if __GNUC__ >= 3
126# define expect(expr,value) __builtin_expect ((expr),(value))
127#else
128# define expect(expr,value) (expr)
129#endif
130
131#define expect_false(expr) expect ((expr) != 0, 0)
132#define expect_true(expr) expect ((expr) != 0, 1)
133
134/*****************************************************************************/
135 237
136#define ETP_PRI_MIN EIO_PRI_MIN 238#define ETP_PRI_MIN EIO_PRI_MIN
137#define ETP_PRI_MAX EIO_PRI_MAX 239#define ETP_PRI_MAX EIO_PRI_MAX
138 240
139struct etp_worker; 241struct etp_worker;
155 if (wrk->dirp) \ 257 if (wrk->dirp) \
156 { \ 258 { \
157 closedir (wrk->dirp); \ 259 closedir (wrk->dirp); \
158 wrk->dirp = 0; \ 260 wrk->dirp = 0; \
159 } 261 }
262
160#define ETP_WORKER_COMMON \ 263#define ETP_WORKER_COMMON \
161 void *dbuf; \ 264 void *dbuf; \
162 DIR *dirp; 265 DIR *dirp;
163 266
164/*****************************************************************************/ 267/*****************************************************************************/
165 268
166#define ETP_NUM_PRI (ETP_PRI_MAX - ETP_PRI_MIN + 1) 269#define ETP_NUM_PRI (ETP_PRI_MAX - ETP_PRI_MIN + 1)
167 270
168/* calculcate time difference in ~1/EIO_TICKS of a second */ 271/* calculate time difference in ~1/EIO_TICKS of a second */
272ecb_inline int
169static int tvdiff (struct timeval *tv1, struct timeval *tv2) 273tvdiff (struct timeval *tv1, struct timeval *tv2)
170{ 274{
171 return (tv2->tv_sec - tv1->tv_sec ) * EIO_TICKS 275 return (tv2->tv_sec - tv1->tv_sec ) * EIO_TICKS
172 + ((tv2->tv_usec - tv1->tv_usec) >> 10); 276 + ((tv2->tv_usec - tv1->tv_usec) >> 10);
173} 277}
174 278
181static unsigned int max_poll_reqs; /* reslock */ 285static unsigned int max_poll_reqs; /* reslock */
182 286
183static volatile unsigned int nreqs; /* reqlock */ 287static volatile unsigned int nreqs; /* reqlock */
184static volatile unsigned int nready; /* reqlock */ 288static volatile unsigned int nready; /* reqlock */
185static volatile unsigned int npending; /* reqlock */ 289static volatile unsigned int npending; /* reqlock */
186static volatile unsigned int max_idle = 4; 290static volatile unsigned int max_idle = 4; /* maximum number of threads that can idle indefinitely */
291static volatile unsigned int idle_timeout = 10; /* number of seconds after which an idle threads exit */
187 292
188static mutex_t wrklock = X_MUTEX_INIT; 293static xmutex_t wrklock;
189static mutex_t reslock = X_MUTEX_INIT; 294static xmutex_t reslock;
190static mutex_t reqlock = X_MUTEX_INIT; 295static xmutex_t reqlock;
191static cond_t reqwait = X_COND_INIT; 296static xcond_t reqwait;
192 297
193#if !HAVE_PREADWRITE 298#if !HAVE_PREADWRITE
194/* 299/*
195 * make our pread/pwrite emulation safe against themselves, but not against 300 * make our pread/pwrite emulation safe against themselves, but not against
196 * normal read/write by using a mutex. slows down execution a lot, 301 * normal read/write by using a mutex. slows down execution a lot,
197 * but that's your problem, not mine. 302 * but that's your problem, not mine.
198 */ 303 */
199static mutex_t preadwritelock = X_MUTEX_INIT; 304static xmutex_t preadwritelock = X_MUTEX_INIT;
200#endif 305#endif
201 306
202typedef struct etp_worker 307typedef struct etp_worker
203{ 308{
204 /* locked by wrklock */ 309 /* locked by wrklock */
205 struct etp_worker *prev, *next; 310 struct etp_worker *prev, *next;
206 311
207 thread_t tid; 312 xthread_t tid;
208 313
209 /* locked by reslock, reqlock or wrklock */ 314 /* locked by reslock, reqlock or wrklock */
210 ETP_REQ *req; /* currently processed request */ 315 ETP_REQ *req; /* currently processed request */
211 316
212 ETP_WORKER_COMMON 317 ETP_WORKER_COMMON
217#define ETP_WORKER_LOCK(wrk) X_LOCK (wrklock) 322#define ETP_WORKER_LOCK(wrk) X_LOCK (wrklock)
218#define ETP_WORKER_UNLOCK(wrk) X_UNLOCK (wrklock) 323#define ETP_WORKER_UNLOCK(wrk) X_UNLOCK (wrklock)
219 324
220/* worker threads management */ 325/* worker threads management */
221 326
327static void ecb_cold
222static void etp_worker_clear (etp_worker *wrk) 328etp_worker_clear (etp_worker *wrk)
223{ 329{
224 ETP_WORKER_CLEAR (wrk); 330 ETP_WORKER_CLEAR (wrk);
225} 331}
226 332
333static void ecb_cold
227static void etp_worker_free (etp_worker *wrk) 334etp_worker_free (etp_worker *wrk)
228{ 335{
229 wrk->next->prev = wrk->prev; 336 wrk->next->prev = wrk->prev;
230 wrk->prev->next = wrk->next; 337 wrk->prev->next = wrk->next;
231 338
232 free (wrk); 339 free (wrk);
233} 340}
234 341
235static unsigned int etp_nreqs (void) 342static unsigned int
343etp_nreqs (void)
236{ 344{
237 int retval; 345 int retval;
238 if (WORDACCESS_UNSAFE) X_LOCK (reqlock); 346 if (WORDACCESS_UNSAFE) X_LOCK (reqlock);
239 retval = nreqs; 347 retval = nreqs;
240 if (WORDACCESS_UNSAFE) X_UNLOCK (reqlock); 348 if (WORDACCESS_UNSAFE) X_UNLOCK (reqlock);
241 return retval; 349 return retval;
242} 350}
243 351
244static unsigned int etp_nready (void) 352static unsigned int
353etp_nready (void)
245{ 354{
246 unsigned int retval; 355 unsigned int retval;
247 356
248 if (WORDACCESS_UNSAFE) X_LOCK (reqlock); 357 if (WORDACCESS_UNSAFE) X_LOCK (reqlock);
249 retval = nready; 358 retval = nready;
250 if (WORDACCESS_UNSAFE) X_UNLOCK (reqlock); 359 if (WORDACCESS_UNSAFE) X_UNLOCK (reqlock);
251 360
252 return retval; 361 return retval;
253} 362}
254 363
255static unsigned int etp_npending (void) 364static unsigned int
365etp_npending (void)
256{ 366{
257 unsigned int retval; 367 unsigned int retval;
258 368
259 if (WORDACCESS_UNSAFE) X_LOCK (reqlock); 369 if (WORDACCESS_UNSAFE) X_LOCK (reqlock);
260 retval = npending; 370 retval = npending;
261 if (WORDACCESS_UNSAFE) X_UNLOCK (reqlock); 371 if (WORDACCESS_UNSAFE) X_UNLOCK (reqlock);
262 372
263 return retval; 373 return retval;
264} 374}
265 375
266static unsigned int etp_nthreads (void) 376static unsigned int
377etp_nthreads (void)
267{ 378{
268 unsigned int retval; 379 unsigned int retval;
269 380
270 if (WORDACCESS_UNSAFE) X_LOCK (reqlock); 381 if (WORDACCESS_UNSAFE) X_LOCK (reqlock);
271 retval = started; 382 retval = started;
285} etp_reqq; 396} etp_reqq;
286 397
287static etp_reqq req_queue; 398static etp_reqq req_queue;
288static etp_reqq res_queue; 399static etp_reqq res_queue;
289 400
401static int ecb_noinline
290static int reqq_push (etp_reqq *q, ETP_REQ *req) 402reqq_push (etp_reqq *q, ETP_REQ *req)
291{ 403{
292 int pri = req->pri; 404 int pri = req->pri;
293 req->next = 0; 405 req->next = 0;
294 406
295 if (q->qe[pri]) 407 if (q->qe[pri])
301 q->qe[pri] = q->qs[pri] = req; 413 q->qe[pri] = q->qs[pri] = req;
302 414
303 return q->size++; 415 return q->size++;
304} 416}
305 417
418static ETP_REQ * ecb_noinline
306static ETP_REQ *reqq_shift (etp_reqq *q) 419reqq_shift (etp_reqq *q)
307{ 420{
308 int pri; 421 int pri;
309 422
310 if (!q->size) 423 if (!q->size)
311 return 0; 424 return 0;
326 } 439 }
327 440
328 abort (); 441 abort ();
329} 442}
330 443
331static void etp_atfork_prepare (void) 444static void ecb_cold
445etp_thread_init (void)
332{ 446{
333 X_LOCK (wrklock);
334 X_LOCK (reqlock);
335 X_LOCK (reslock);
336#if !HAVE_PREADWRITE 447#if !HAVE_PREADWRITE
337 X_LOCK (preadwritelock); 448 X_MUTEX_CREATE (preadwritelock);
338#endif 449#endif
450 X_MUTEX_CREATE (wrklock);
451 X_MUTEX_CREATE (reslock);
452 X_MUTEX_CREATE (reqlock);
453 X_COND_CREATE (reqwait);
339} 454}
340 455
456static void ecb_cold
457etp_atfork_prepare (void)
458{
459}
460
461static void ecb_cold
341static void etp_atfork_parent (void) 462etp_atfork_parent (void)
342{ 463{
343#if !HAVE_PREADWRITE
344 X_UNLOCK (preadwritelock);
345#endif
346 X_UNLOCK (reslock);
347 X_UNLOCK (reqlock);
348 X_UNLOCK (wrklock);
349} 464}
350 465
466static void ecb_cold
351static void etp_atfork_child (void) 467etp_atfork_child (void)
352{ 468{
353 ETP_REQ *prv; 469 ETP_REQ *prv;
354 470
355 while ((prv = reqq_shift (&req_queue))) 471 while ((prv = reqq_shift (&req_queue)))
356 ETP_DESTROY (prv); 472 ETP_DESTROY (prv);
373 idle = 0; 489 idle = 0;
374 nreqs = 0; 490 nreqs = 0;
375 nready = 0; 491 nready = 0;
376 npending = 0; 492 npending = 0;
377 493
378 etp_atfork_parent (); 494 etp_thread_init ();
379} 495}
380 496
381static void 497static void ecb_cold
382etp_once_init (void) 498etp_once_init (void)
383{ 499{
500 etp_thread_init ();
384 X_THREAD_ATFORK (etp_atfork_prepare, etp_atfork_parent, etp_atfork_child); 501 X_THREAD_ATFORK (etp_atfork_prepare, etp_atfork_parent, etp_atfork_child);
385} 502}
386 503
387static int 504static int ecb_cold
388etp_init (void (*want_poll)(void), void (*done_poll)(void)) 505etp_init (void (*want_poll)(void), void (*done_poll)(void))
389{ 506{
390 static pthread_once_t doinit = PTHREAD_ONCE_INIT; 507 static pthread_once_t doinit = PTHREAD_ONCE_INIT;
391 508
392 pthread_once (&doinit, etp_once_init); 509 pthread_once (&doinit, etp_once_init);
397 return 0; 514 return 0;
398} 515}
399 516
400X_THREAD_PROC (etp_proc); 517X_THREAD_PROC (etp_proc);
401 518
519static void ecb_cold
402static void etp_start_thread (void) 520etp_start_thread (void)
403{ 521{
404 etp_worker *wrk = calloc (1, sizeof (etp_worker)); 522 etp_worker *wrk = calloc (1, sizeof (etp_worker));
405 523
406 /*TODO*/ 524 /*TODO*/
407 assert (("unable to allocate worker thread data", wrk)); 525 assert (("unable to allocate worker thread data", wrk));
420 free (wrk); 538 free (wrk);
421 539
422 X_UNLOCK (wrklock); 540 X_UNLOCK (wrklock);
423} 541}
424 542
543static void
425static void etp_maybe_start_thread (void) 544etp_maybe_start_thread (void)
426{ 545{
427 if (expect_true (etp_nthreads () >= wanted)) 546 if (ecb_expect_true (etp_nthreads () >= wanted))
428 return; 547 return;
429 548
430 /* todo: maybe use idle here, but might be less exact */ 549 /* todo: maybe use idle here, but might be less exact */
431 if (expect_true (0 <= (int)etp_nthreads () + (int)etp_npending () - (int)etp_nreqs ())) 550 if (ecb_expect_true (0 <= (int)etp_nthreads () + (int)etp_npending () - (int)etp_nreqs ()))
432 return; 551 return;
433 552
434 etp_start_thread (); 553 etp_start_thread ();
435} 554}
436 555
556static void ecb_cold
437static void etp_end_thread (void) 557etp_end_thread (void)
438{ 558{
439 eio_req *req = calloc (1, sizeof (eio_req)); 559 eio_req *req = calloc (1, sizeof (eio_req));
440 560
441 req->type = -1; 561 req->type = -1;
442 req->pri = ETP_PRI_MAX - ETP_PRI_MIN; 562 req->pri = ETP_PRI_MAX - ETP_PRI_MIN;
449 X_LOCK (wrklock); 569 X_LOCK (wrklock);
450 --started; 570 --started;
451 X_UNLOCK (wrklock); 571 X_UNLOCK (wrklock);
452} 572}
453 573
454static int etp_poll (void) 574static int
575etp_poll (void)
455{ 576{
456 unsigned int maxreqs; 577 unsigned int maxreqs;
457 unsigned int maxtime; 578 unsigned int maxtime;
458 struct timeval tv_start, tv_now; 579 struct timeval tv_start, tv_now;
459 580
489 610
490 X_LOCK (reqlock); 611 X_LOCK (reqlock);
491 --nreqs; 612 --nreqs;
492 X_UNLOCK (reqlock); 613 X_UNLOCK (reqlock);
493 614
494 if (expect_false (req->type == EIO_GROUP && req->size)) 615 if (ecb_expect_false (req->type == EIO_GROUP && req->size))
495 { 616 {
496 req->int1 = 1; /* mark request as delayed */ 617 req->int1 = 1; /* mark request as delayed */
497 continue; 618 continue;
498 } 619 }
499 else 620 else
500 { 621 {
501 int res = ETP_FINISH (req); 622 int res = ETP_FINISH (req);
502 if (expect_false (res)) 623 if (ecb_expect_false (res))
503 return res; 624 return res;
504 } 625 }
505 626
506 if (expect_false (maxreqs && !--maxreqs)) 627 if (ecb_expect_false (maxreqs && !--maxreqs))
507 break; 628 break;
508 629
509 if (maxtime) 630 if (maxtime)
510 { 631 {
511 gettimeofday (&tv_now, 0); 632 gettimeofday (&tv_now, 0);
517 638
518 errno = EAGAIN; 639 errno = EAGAIN;
519 return -1; 640 return -1;
520} 641}
521 642
643static void
522static void etp_cancel (ETP_REQ *req) 644etp_cancel (ETP_REQ *req)
523{ 645{
524 X_LOCK (wrklock); 646 req->cancelled = 1;
525 req->flags |= EIO_FLAG_CANCELLED;
526 X_UNLOCK (wrklock);
527 647
528 eio_grp_cancel (req); 648 eio_grp_cancel (req);
529} 649}
530 650
651static void
531static void etp_submit (ETP_REQ *req) 652etp_submit (ETP_REQ *req)
532{ 653{
533 req->pri -= ETP_PRI_MIN; 654 req->pri -= ETP_PRI_MIN;
534 655
535 if (expect_false (req->pri < ETP_PRI_MIN - ETP_PRI_MIN)) req->pri = ETP_PRI_MIN - ETP_PRI_MIN; 656 if (ecb_expect_false (req->pri < ETP_PRI_MIN - ETP_PRI_MIN)) req->pri = ETP_PRI_MIN - ETP_PRI_MIN;
536 if (expect_false (req->pri > ETP_PRI_MAX - ETP_PRI_MIN)) req->pri = ETP_PRI_MAX - ETP_PRI_MIN; 657 if (ecb_expect_false (req->pri > ETP_PRI_MAX - ETP_PRI_MIN)) req->pri = ETP_PRI_MAX - ETP_PRI_MIN;
537 658
538 if (expect_false (req->type == EIO_GROUP)) 659 if (ecb_expect_false (req->type == EIO_GROUP))
539 { 660 {
540 /* I hope this is worth it :/ */ 661 /* I hope this is worth it :/ */
541 X_LOCK (reqlock); 662 X_LOCK (reqlock);
542 ++nreqs; 663 ++nreqs;
543 X_UNLOCK (reqlock); 664 X_UNLOCK (reqlock);
562 683
563 etp_maybe_start_thread (); 684 etp_maybe_start_thread ();
564 } 685 }
565} 686}
566 687
688static void ecb_cold
567static void etp_set_max_poll_time (double nseconds) 689etp_set_max_poll_time (double nseconds)
568{ 690{
569 if (WORDACCESS_UNSAFE) X_LOCK (reslock); 691 if (WORDACCESS_UNSAFE) X_LOCK (reslock);
570 max_poll_time = nseconds; 692 max_poll_time = nseconds * EIO_TICKS;
571 if (WORDACCESS_UNSAFE) X_UNLOCK (reslock); 693 if (WORDACCESS_UNSAFE) X_UNLOCK (reslock);
572} 694}
573 695
696static void ecb_cold
574static void etp_set_max_poll_reqs (unsigned int maxreqs) 697etp_set_max_poll_reqs (unsigned int maxreqs)
575{ 698{
576 if (WORDACCESS_UNSAFE) X_LOCK (reslock); 699 if (WORDACCESS_UNSAFE) X_LOCK (reslock);
577 max_poll_reqs = maxreqs; 700 max_poll_reqs = maxreqs;
578 if (WORDACCESS_UNSAFE) X_UNLOCK (reslock); 701 if (WORDACCESS_UNSAFE) X_UNLOCK (reslock);
579} 702}
580 703
704static void ecb_cold
581static void etp_set_max_idle (unsigned int nthreads) 705etp_set_max_idle (unsigned int nthreads)
582{ 706{
583 if (WORDACCESS_UNSAFE) X_LOCK (reqlock); 707 if (WORDACCESS_UNSAFE) X_LOCK (reqlock);
584 max_idle = nthreads <= 0 ? 1 : nthreads; 708 max_idle = nthreads;
585 if (WORDACCESS_UNSAFE) X_UNLOCK (reqlock); 709 if (WORDACCESS_UNSAFE) X_UNLOCK (reqlock);
586} 710}
587 711
712static void ecb_cold
713etp_set_idle_timeout (unsigned int seconds)
714{
715 if (WORDACCESS_UNSAFE) X_LOCK (reqlock);
716 idle_timeout = seconds;
717 if (WORDACCESS_UNSAFE) X_UNLOCK (reqlock);
718}
719
720static void ecb_cold
588static void etp_set_min_parallel (unsigned int nthreads) 721etp_set_min_parallel (unsigned int nthreads)
589{ 722{
590 if (wanted < nthreads) 723 if (wanted < nthreads)
591 wanted = nthreads; 724 wanted = nthreads;
592} 725}
593 726
727static void ecb_cold
594static void etp_set_max_parallel (unsigned int nthreads) 728etp_set_max_parallel (unsigned int nthreads)
595{ 729{
596 if (wanted > nthreads) 730 if (wanted > nthreads)
597 wanted = nthreads; 731 wanted = nthreads;
598 732
599 while (started > wanted) 733 while (started > wanted)
600 etp_end_thread (); 734 etp_end_thread ();
601} 735}
602 736
603/*****************************************************************************/ 737/*****************************************************************************/
604 738
739static void
605static void grp_try_feed (eio_req *grp) 740grp_try_feed (eio_req *grp)
606{ 741{
607 while (grp->size < grp->int2 && !EIO_CANCELLED (grp)) 742 while (grp->size < grp->int2 && !EIO_CANCELLED (grp))
608 { 743 {
609 grp->flags &= ~EIO_FLAG_GROUPADD; 744 grp->flags &= ~EIO_FLAG_GROUPADD;
610 745
617 break; 752 break;
618 } 753 }
619 } 754 }
620} 755}
621 756
757static int
622static int grp_dec (eio_req *grp) 758grp_dec (eio_req *grp)
623{ 759{
624 --grp->size; 760 --grp->size;
625 761
626 /* call feeder, if applicable */ 762 /* call feeder, if applicable */
627 grp_try_feed (grp); 763 grp_try_feed (grp);
631 return eio_finish (grp); 767 return eio_finish (grp);
632 else 768 else
633 return 0; 769 return 0;
634} 770}
635 771
772static void
636void eio_destroy (eio_req *req) 773eio_destroy (eio_req *req)
637{ 774{
638 if ((req)->flags & EIO_FLAG_PTR1_FREE) free (req->ptr1); 775 if ((req)->flags & EIO_FLAG_PTR1_FREE) free (req->ptr1);
639 if ((req)->flags & EIO_FLAG_PTR2_FREE) free (req->ptr2); 776 if ((req)->flags & EIO_FLAG_PTR2_FREE) free (req->ptr2);
640 777
641 EIO_DESTROY (req); 778 EIO_DESTROY (req);
642} 779}
643 780
781static int
644static int eio_finish (eio_req *req) 782eio_finish (eio_req *req)
645{ 783{
646 int res = EIO_FINISH (req); 784 int res = EIO_FINISH (req);
647 785
648 if (req->grp) 786 if (req->grp)
649 { 787 {
657 if (grp->grp_first == req) 795 if (grp->grp_first == req)
658 grp->grp_first = req->grp_next; 796 grp->grp_first = req->grp_next;
659 797
660 res2 = grp_dec (grp); 798 res2 = grp_dec (grp);
661 799
662 if (!res && res2) 800 if (!res)
663 res = res2; 801 res = res2;
664 } 802 }
665 803
666 eio_destroy (req); 804 eio_destroy (req);
667 805
668 return res; 806 return res;
669} 807}
670 808
809void
671void eio_grp_cancel (eio_req *grp) 810eio_grp_cancel (eio_req *grp)
672{ 811{
673 for (grp = grp->grp_first; grp; grp = grp->grp_next) 812 for (grp = grp->grp_first; grp; grp = grp->grp_next)
674 eio_cancel (grp); 813 eio_cancel (grp);
675} 814}
676 815
816void
677void eio_cancel (eio_req *req) 817eio_cancel (eio_req *req)
678{ 818{
679 etp_cancel (req); 819 etp_cancel (req);
680} 820}
681 821
822void
682void eio_submit (eio_req *req) 823eio_submit (eio_req *req)
683{ 824{
684 etp_submit (req); 825 etp_submit (req);
685} 826}
686 827
687unsigned int eio_nreqs (void) 828unsigned int
829eio_nreqs (void)
688{ 830{
689 return etp_nreqs (); 831 return etp_nreqs ();
690} 832}
691 833
692unsigned int eio_nready (void) 834unsigned int
835eio_nready (void)
693{ 836{
694 return etp_nready (); 837 return etp_nready ();
695} 838}
696 839
697unsigned int eio_npending (void) 840unsigned int
841eio_npending (void)
698{ 842{
699 return etp_npending (); 843 return etp_npending ();
700} 844}
701 845
702unsigned int eio_nthreads (void) 846unsigned int ecb_cold
847eio_nthreads (void)
703{ 848{
704 return etp_nthreads (); 849 return etp_nthreads ();
705} 850}
706 851
852void ecb_cold
707void eio_set_max_poll_time (double nseconds) 853eio_set_max_poll_time (double nseconds)
708{ 854{
709 etp_set_max_poll_time (nseconds); 855 etp_set_max_poll_time (nseconds);
710} 856}
711 857
858void ecb_cold
712void eio_set_max_poll_reqs (unsigned int maxreqs) 859eio_set_max_poll_reqs (unsigned int maxreqs)
713{ 860{
714 etp_set_max_poll_reqs (maxreqs); 861 etp_set_max_poll_reqs (maxreqs);
715} 862}
716 863
864void ecb_cold
717void eio_set_max_idle (unsigned int nthreads) 865eio_set_max_idle (unsigned int nthreads)
718{ 866{
719 etp_set_max_idle (nthreads); 867 etp_set_max_idle (nthreads);
720} 868}
721 869
870void ecb_cold
871eio_set_idle_timeout (unsigned int seconds)
872{
873 etp_set_idle_timeout (seconds);
874}
875
876void ecb_cold
722void eio_set_min_parallel (unsigned int nthreads) 877eio_set_min_parallel (unsigned int nthreads)
723{ 878{
724 etp_set_min_parallel (nthreads); 879 etp_set_min_parallel (nthreads);
725} 880}
726 881
882void ecb_cold
727void eio_set_max_parallel (unsigned int nthreads) 883eio_set_max_parallel (unsigned int nthreads)
728{ 884{
729 etp_set_max_parallel (nthreads); 885 etp_set_max_parallel (nthreads);
730} 886}
731 887
732int eio_poll (void) 888int eio_poll (void)
741# undef pread 897# undef pread
742# undef pwrite 898# undef pwrite
743# define pread eio__pread 899# define pread eio__pread
744# define pwrite eio__pwrite 900# define pwrite eio__pwrite
745 901
746static ssize_t 902static eio_ssize_t
747eio__pread (int fd, void *buf, size_t count, off_t offset) 903eio__pread (int fd, void *buf, size_t count, off_t offset)
748{ 904{
749 ssize_t res; 905 eio_ssize_t res;
750 off_t ooffset; 906 off_t ooffset;
751 907
752 X_LOCK (preadwritelock); 908 X_LOCK (preadwritelock);
753 ooffset = lseek (fd, 0, SEEK_CUR); 909 ooffset = lseek (fd, 0, SEEK_CUR);
754 lseek (fd, offset, SEEK_SET); 910 lseek (fd, offset, SEEK_SET);
757 X_UNLOCK (preadwritelock); 913 X_UNLOCK (preadwritelock);
758 914
759 return res; 915 return res;
760} 916}
761 917
762static ssize_t 918static eio_ssize_t
763eio__pwrite (int fd, void *buf, size_t count, off_t offset) 919eio__pwrite (int fd, void *buf, size_t count, off_t offset)
764{ 920{
765 ssize_t res; 921 eio_ssize_t res;
766 off_t ooffset; 922 off_t ooffset;
767 923
768 X_LOCK (preadwritelock); 924 X_LOCK (preadwritelock);
769 ooffset = lseek (fd, 0, SEEK_CUR); 925 ooffset = lseek (fd, 0, SEEK_CUR);
770 lseek (fd, offset, SEEK_SET); 926 lseek (fd, offset, SEEK_SET);
771 res = write (fd, buf, count); 927 res = write (fd, buf, count);
772 lseek (fd, offset, SEEK_SET); 928 lseek (fd, ooffset, SEEK_SET);
773 X_UNLOCK (preadwritelock); 929 X_UNLOCK (preadwritelock);
774 930
775 return res; 931 return res;
776} 932}
777#endif 933#endif
778 934
779#ifndef HAVE_FUTIMES 935#ifndef HAVE_UTIMES
780 936
781# undef utimes 937# undef utimes
782# undef futimes
783# define utimes(path,times) eio__utimes (path, times) 938# define utimes(path,times) eio__utimes (path, times)
784# define futimes(fd,times) eio__futimes (fd, times)
785 939
786static int 940static int
787eio__utimes (const char *filename, const struct timeval times[2]) 941eio__utimes (const char *filename, const struct timeval times[2])
788{ 942{
789 if (times) 943 if (times)
797 } 951 }
798 else 952 else
799 return utime (filename, 0); 953 return utime (filename, 0);
800} 954}
801 955
956#endif
957
958#ifndef HAVE_FUTIMES
959
960# undef futimes
961# define futimes(fd,times) eio__futimes (fd, times)
962
963static int
802static int eio__futimes (int fd, const struct timeval tv[2]) 964eio__futimes (int fd, const struct timeval tv[2])
803{ 965{
804 errno = ENOSYS; 966 errno = ENOSYS;
805 return -1; 967 return -1;
806} 968}
807 969
811# undef fdatasync 973# undef fdatasync
812# define fdatasync(fd) fsync (fd) 974# define fdatasync(fd) fsync (fd)
813#endif 975#endif
814 976
815/* sync_file_range always needs emulation */ 977/* sync_file_range always needs emulation */
816int 978static int
817eio__sync_file_range (int fd, off_t offset, size_t nbytes, unsigned int flags) 979eio__sync_file_range (int fd, off_t offset, size_t nbytes, unsigned int flags)
818{ 980{
819#if HAVE_SYNC_FILE_RANGE 981#if HAVE_SYNC_FILE_RANGE
820 int res; 982 int res;
821 983
829 | (flags & EIO_SYNC_FILE_RANGE_WAIT_AFTER ? SYNC_FILE_RANGE_WAIT_AFTER : 0); 991 | (flags & EIO_SYNC_FILE_RANGE_WAIT_AFTER ? SYNC_FILE_RANGE_WAIT_AFTER : 0);
830 } 992 }
831 993
832 res = sync_file_range (fd, offset, nbytes, flags); 994 res = sync_file_range (fd, offset, nbytes, flags);
833 995
834 if (res != ENOSYS) 996 if (!res || errno != ENOSYS)
835 return res; 997 return res;
836#endif 998#endif
837 999
838 /* even though we could play tricks with the flags, it's better to always 1000 /* even though we could play tricks with the flags, it's better to always
839 * call fdatasync, as thta matches the expectation of it's users best */ 1001 * call fdatasync, as that matches the expectation of its users best */
840 return fdatasync (fd); 1002 return fdatasync (fd);
1003}
1004
1005static int
1006eio__fallocate (int fd, int mode, off_t offset, size_t len)
1007{
1008#if HAVE_FALLOCATE
1009 return fallocate (fd, mode, offset, len);
1010#else
1011 errno = ENOSYS;
1012 return -1;
1013#endif
841} 1014}
842 1015
843#if !HAVE_READAHEAD 1016#if !HAVE_READAHEAD
844# undef readahead 1017# undef readahead
845# define readahead(fd,offset,count) eio__readahead (fd, offset, count, self) 1018# define readahead(fd,offset,count) eio__readahead (fd, offset, count, self)
846 1019
847static ssize_t 1020static eio_ssize_t
848eio__readahead (int fd, off_t offset, size_t count, etp_worker *self) 1021eio__readahead (int fd, off_t offset, size_t count, etp_worker *self)
849{ 1022{
850 size_t todo = count; 1023 size_t todo = count;
851 dBUF; 1024 dBUF;
852 1025
864} 1037}
865 1038
866#endif 1039#endif
867 1040
868/* sendfile always needs emulation */ 1041/* sendfile always needs emulation */
869static ssize_t 1042static eio_ssize_t
870eio__sendfile (int ofd, int ifd, off_t offset, size_t count, etp_worker *self) 1043eio__sendfile (int ofd, int ifd, off_t offset, size_t count, etp_worker *self)
871{ 1044{
1045 eio_ssize_t written = 0;
872 ssize_t res; 1046 eio_ssize_t res;
873 1047
874 if (!count) 1048 if (!count)
875 return 0; 1049 return 0;
876 1050
1051 for (;;)
1052 {
1053#ifdef __APPLE__
1054# undef HAVE_SENDFILE /* broken, as everything on os x */
1055#endif
877#if HAVE_SENDFILE 1056#if HAVE_SENDFILE
878# if __linux 1057# if __linux
1058 off_t soffset = offset;
879 res = sendfile (ofd, ifd, &offset, count); 1059 res = sendfile (ofd, ifd, &soffset, count);
880 1060
881# elif __freebsd 1061# elif __FreeBSD__
882 /* 1062 /*
883 * Of course, the freebsd sendfile is a dire hack with no thoughts 1063 * Of course, the freebsd sendfile is a dire hack with no thoughts
884 * wasted on making it similar to other I/O functions. 1064 * wasted on making it similar to other I/O functions.
885 */ 1065 */
886 {
887 off_t sbytes; 1066 off_t sbytes;
888 res = sendfile (ifd, ofd, offset, count, 0, &sbytes, 0); 1067 res = sendfile (ifd, ofd, offset, count, 0, &sbytes, 0);
889 1068
890 if (res < 0 && sbytes) 1069 #if 0 /* according to the manpage, this is correct, but broken behaviour */
891 /* maybe only on EAGAIN: as usual, the manpage leaves you guessing */ 1070 /* freebsd' sendfile will return 0 on success */
1071 /* freebsd 8 documents it as only setting *sbytes on EINTR and EAGAIN, but */
1072 /* not on e.g. EIO or EPIPE - sounds broken */
1073 if ((res < 0 && (errno == EAGAIN || errno == EINTR) && sbytes) || res == 0)
892 res = sbytes; 1074 res = sbytes;
893 } 1075 #endif
1076
1077 /* according to source inspection, this is correct, and useful behaviour */
1078 if (sbytes)
1079 res = sbytes;
1080
1081# elif defined (__APPLE__)
1082 off_t sbytes = count;
1083 res = sendfile (ifd, ofd, offset, &sbytes, 0, 0);
1084
1085 /* according to the manpage, sbytes is always valid */
1086 if (sbytes)
1087 res = sbytes;
894 1088
895# elif __hpux 1089# elif __hpux
896 res = sendfile (ofd, ifd, offset, count, 0, 0); 1090 res = sendfile (ofd, ifd, offset, count, 0, 0);
897 1091
898# elif __solaris 1092# elif __solaris
899 {
900 struct sendfilevec vec; 1093 struct sendfilevec vec;
901 size_t sbytes; 1094 size_t sbytes;
902 1095
903 vec.sfv_fd = ifd; 1096 vec.sfv_fd = ifd;
904 vec.sfv_flag = 0; 1097 vec.sfv_flag = 0;
905 vec.sfv_off = offset; 1098 vec.sfv_off = offset;
906 vec.sfv_len = count; 1099 vec.sfv_len = count;
907 1100
908 res = sendfilev (ofd, &vec, 1, &sbytes); 1101 res = sendfilev (ofd, &vec, 1, &sbytes);
909 1102
910 if (res < 0 && sbytes) 1103 if (res < 0 && sbytes)
911 res = sbytes; 1104 res = sbytes;
912 }
913 1105
914# endif 1106# endif
1107
1108#elif defined (_WIN32) && 0
1109 /* does not work, just for documentation of what would need to be done */
1110 /* actually, cannot be done like this, as TransmitFile changes the file offset, */
1111 /* libeio guarantees that the file offset does not change, and windows */
1112 /* has no way to get an independent handle to the same file description */
1113 HANDLE h = TO_SOCKET (ifd);
1114 SetFilePointer (h, offset, 0, FILE_BEGIN);
1115 res = TransmitFile (TO_SOCKET (ofd), h, count, 0, 0, 0, 0);
1116
915#else 1117#else
916 res = -1; 1118 res = -1;
917 errno = ENOSYS; 1119 errno = ENOSYS;
918#endif 1120#endif
919 1121
1122 /* we assume sendfile can copy at least 128mb in one go */
1123 if (res <= 128 * 1024 * 1024)
1124 {
1125 if (res > 0)
1126 written += res;
1127
1128 if (written)
1129 return written;
1130
1131 break;
1132 }
1133 else
1134 {
1135 /* if we requested more, then probably the kernel was lazy */
1136 written += res;
1137 offset += res;
1138 count -= res;
1139
1140 if (!count)
1141 return written;
1142 }
1143 }
1144
920 if (res < 0 1145 if (res < 0
921 && (errno == ENOSYS || errno == EINVAL || errno == ENOTSOCK 1146 && (errno == ENOSYS || errno == EINVAL || errno == ENOTSOCK
1147 /* BSDs */
1148#ifdef ENOTSUP /* sigh, if the steenking pile called openbsd would only try to at least compile posix code... */
1149 || errno == ENOTSUP
1150#endif
1151#ifdef EOPNOTSUPP /* windows */
1152 || errno == EOPNOTSUPP /* BSDs */
1153#endif
922#if __solaris 1154#if __solaris
923 || errno == EAFNOSUPPORT || errno == EPROTOTYPE 1155 || errno == EAFNOSUPPORT || errno == EPROTOTYPE
924#endif 1156#endif
925 ) 1157 )
926 ) 1158 )
930 1162
931 res = 0; 1163 res = 0;
932 1164
933 while (count) 1165 while (count)
934 { 1166 {
935 ssize_t cnt; 1167 eio_ssize_t cnt;
936 1168
937 cnt = pread (ifd, eio_buf, count > EIO_BUFSIZE ? EIO_BUFSIZE : count, offset); 1169 cnt = pread (ifd, eio_buf, count > EIO_BUFSIZE ? EIO_BUFSIZE : count, offset);
938 1170
939 if (cnt <= 0) 1171 if (cnt <= 0)
940 { 1172 {
957 } 1189 }
958 1190
959 return res; 1191 return res;
960} 1192}
961 1193
1194#ifdef PAGESIZE
1195# define eio_pagesize() PAGESIZE
1196#else
1197static intptr_t
1198eio_pagesize (void)
1199{
1200 static intptr_t page;
1201
1202 if (!page)
1203 page = sysconf (_SC_PAGESIZE);
1204
1205 return page;
1206}
1207#endif
1208
1209static void
1210eio_page_align (void **addr, size_t *length)
1211{
1212 intptr_t mask = eio_pagesize () - 1;
1213
1214 /* round down addr */
1215 intptr_t adj = mask & (intptr_t)*addr;
1216
1217 *addr = (void *)((intptr_t)*addr - adj);
1218 *length += adj;
1219
1220 /* round up length */
1221 *length = (*length + mask) & ~mask;
1222}
1223
1224#if !_POSIX_MEMLOCK
1225# define eio__mlockall(a) EIO_ENOSYS ()
1226#else
1227
1228static int
1229eio__mlockall (int flags)
1230{
1231 #if __GLIBC__ == 2 && __GLIBC_MINOR__ <= 7
1232 extern int mallopt (int, int);
1233 mallopt (-6, 238); /* http://bugs.debian.org/cgi-bin/bugreport.cgi?bug=473812 */
1234 #endif
1235
1236 if (EIO_MCL_CURRENT != MCL_CURRENT
1237 || EIO_MCL_FUTURE != MCL_FUTURE)
1238 {
1239 flags = 0
1240 | (flags & EIO_MCL_CURRENT ? MCL_CURRENT : 0)
1241 | (flags & EIO_MCL_FUTURE ? MCL_FUTURE : 0);
1242 }
1243
1244 return mlockall (flags);
1245}
1246#endif
1247
1248#if !_POSIX_MEMLOCK_RANGE
1249# define eio__mlock(a,b) EIO_ENOSYS ()
1250#else
1251
1252static int
1253eio__mlock (void *addr, size_t length)
1254{
1255 eio_page_align (&addr, &length);
1256
1257 return mlock (addr, length);
1258}
1259
1260#endif
1261
1262#if !(_POSIX_MAPPED_FILES && _POSIX_SYNCHRONIZED_IO)
1263# define eio__msync(a,b,c) EIO_ENOSYS ()
1264#else
1265
1266static int
1267eio__msync (void *mem, size_t len, int flags)
1268{
1269 eio_page_align (&mem, &len);
1270
1271 if (EIO_MS_ASYNC != MS_SYNC
1272 || EIO_MS_INVALIDATE != MS_INVALIDATE
1273 || EIO_MS_SYNC != MS_SYNC)
1274 {
1275 flags = 0
1276 | (flags & EIO_MS_ASYNC ? MS_ASYNC : 0)
1277 | (flags & EIO_MS_INVALIDATE ? MS_INVALIDATE : 0)
1278 | (flags & EIO_MS_SYNC ? MS_SYNC : 0);
1279 }
1280
1281 return msync (mem, len, flags);
1282}
1283
1284#endif
1285
1286static int
1287eio__mtouch (eio_req *req)
1288{
1289 void *mem = req->ptr2;
1290 size_t len = req->size;
1291 int flags = req->int1;
1292
1293 eio_page_align (&mem, &len);
1294
1295 {
1296 intptr_t addr = (intptr_t)mem;
1297 intptr_t end = addr + len;
1298 intptr_t page = eio_pagesize ();
1299
1300 if (addr < end)
1301 if (flags & EIO_MT_MODIFY) /* modify */
1302 do { *((volatile sig_atomic_t *)addr) |= 0; } while ((addr += page) < len && !EIO_CANCELLED (req));
1303 else
1304 do { *((volatile sig_atomic_t *)addr) ; } while ((addr += page) < len && !EIO_CANCELLED (req));
1305 }
1306
1307 return 0;
1308}
1309
1310/*****************************************************************************/
1311/* requests implemented outside eio_execute, because they are so large */
1312
1313static void
1314eio__realpath (eio_req *req, etp_worker *self)
1315{
1316 char *rel = req->ptr1;
1317 char *res;
1318 char *tmp1, *tmp2;
1319#if SYMLOOP_MAX > 32
1320 int symlinks = SYMLOOP_MAX;
1321#else
1322 int symlinks = 32;
1323#endif
1324
1325 req->result = -1;
1326
1327 errno = EINVAL;
1328 if (!rel)
1329 return;
1330
1331 errno = ENOENT;
1332 if (!*rel)
1333 return;
1334
1335 if (!req->ptr2)
1336 {
1337 X_LOCK (wrklock);
1338 req->flags |= EIO_FLAG_PTR2_FREE;
1339 X_UNLOCK (wrklock);
1340 req->ptr2 = malloc (PATH_MAX * 3);
1341
1342 errno = ENOMEM;
1343 if (!req->ptr2)
1344 return;
1345 }
1346
1347 res = req->ptr2;
1348 tmp1 = res + PATH_MAX;
1349 tmp2 = tmp1 + PATH_MAX;
1350
1351#if 0 /* disabled, the musl way to do things is just too racy */
1352#if __linux && defined(O_NONBLOCK) && defined(O_NOATIME)
1353 /* on linux we may be able to ask the kernel */
1354 {
1355 int fd = open (rel, O_RDONLY | O_NONBLOCK | O_NOCTTY | O_NOATIME);
1356
1357 if (fd >= 0)
1358 {
1359 sprintf (tmp1, "/proc/self/fd/%d", fd);
1360 req->result = readlink (tmp1, res, PATH_MAX);
1361 close (fd);
1362
1363 /* here we should probably stat the open file and the disk file, to make sure they still match */
1364
1365 if (req->result > 0)
1366 goto done;
1367 }
1368 else if (errno == ELOOP || errno == ENAMETOOLONG || errno == ENOENT || errno == ENOTDIR || errno == EIO)
1369 return;
1370 }
1371#endif
1372#endif
1373
1374 if (*rel != '/')
1375 {
1376 if (!getcwd (res, PATH_MAX))
1377 return;
1378
1379 if (res [1]) /* only use if not / */
1380 res += strlen (res);
1381 }
1382
1383 while (*rel)
1384 {
1385 eio_ssize_t len, linklen;
1386 char *beg = rel;
1387
1388 while (*rel && *rel != '/')
1389 ++rel;
1390
1391 len = rel - beg;
1392
1393 if (!len) /* skip slashes */
1394 {
1395 ++rel;
1396 continue;
1397 }
1398
1399 if (beg [0] == '.')
1400 {
1401 if (len == 1)
1402 continue; /* . - nop */
1403
1404 if (beg [1] == '.' && len == 2)
1405 {
1406 /* .. - back up one component, if possible */
1407
1408 while (res != req->ptr2)
1409 if (*--res == '/')
1410 break;
1411
1412 continue;
1413 }
1414 }
1415
1416 errno = ENAMETOOLONG;
1417 if (res + 1 + len + 1 >= tmp1)
1418 return;
1419
1420 /* copy one component */
1421 *res = '/';
1422 memcpy (res + 1, beg, len);
1423
1424 /* zero-terminate, for readlink */
1425 res [len + 1] = 0;
1426
1427 /* now check if it's a symlink */
1428 linklen = readlink (req->ptr2, tmp1, PATH_MAX);
1429
1430 if (linklen < 0)
1431 {
1432 if (errno != EINVAL)
1433 return;
1434
1435 /* it's a normal directory. hopefully */
1436 res += len + 1;
1437 }
1438 else
1439 {
1440 /* yay, it was a symlink - build new path in tmp2 */
1441 int rellen = strlen (rel);
1442
1443 errno = ENAMETOOLONG;
1444 if (linklen + 1 + rellen >= PATH_MAX)
1445 return;
1446
1447 errno = ELOOP;
1448 if (!--symlinks)
1449 return;
1450
1451 if (*tmp1 == '/')
1452 res = req->ptr2; /* symlink resolves to an absolute path */
1453
1454 /* we need to be careful, as rel might point into tmp2 already */
1455 memmove (tmp2 + linklen + 1, rel, rellen + 1);
1456 tmp2 [linklen] = '/';
1457 memcpy (tmp2, tmp1, linklen);
1458
1459 rel = tmp2;
1460 }
1461 }
1462
1463 /* special case for the lone root path */
1464 if (res == req->ptr2)
1465 *res++ = '/';
1466
1467 req->result = res - (char *)req->ptr2;
1468
1469done:
1470 req->ptr2 = realloc (req->ptr2, req->result); /* trade time for space savings */
1471}
1472
1473static signed char
1474eio_dent_cmp (const eio_dirent *a, const eio_dirent *b)
1475{
1476 return a->score - b->score ? a->score - b->score /* works because our signed char is always 0..100 */
1477 : a->inode < b->inode ? -1
1478 : a->inode > b->inode ? 1
1479 : 0;
1480}
1481
1482#define EIO_DENT_CMP(i,op,j) eio_dent_cmp (&i, &j) op 0
1483
1484#define EIO_SORT_CUTOFF 30 /* quite high, but performs well on many filesystems */
1485#define EIO_SORT_FAST 60 /* when to only use insertion sort */
1486
1487static void
1488eio_dent_radix_sort (eio_dirent *dents, int size, signed char score_bits, ino_t inode_bits)
1489{
1490 unsigned char bits [9 + sizeof (ino_t) * 8];
1491 unsigned char *bit = bits;
1492
1493 assert (CHAR_BIT == 8);
1494 assert (sizeof (eio_dirent) * 8 < 256);
1495 assert (offsetof (eio_dirent, inode)); /* we use bit #0 as sentinel */
1496 assert (offsetof (eio_dirent, score)); /* we use bit #0 as sentinel */
1497
1498 if (size <= EIO_SORT_FAST)
1499 return;
1500
1501 /* first prepare an array of bits to test in our radix sort */
1502 /* try to take endianness into account, as well as differences in ino_t sizes */
1503 /* inode_bits must contain all inodes ORed together */
1504 /* which is used to skip bits that are 0 everywhere, which is very common */
1505 {
1506 ino_t endianness;
1507 int i, j;
1508
1509 /* we store the byte offset of byte n into byte n of "endianness" */
1510 for (i = 0; i < sizeof (ino_t); ++i)
1511 ((unsigned char *)&endianness)[i] = i;
1512
1513 *bit++ = 0;
1514
1515 for (i = 0; i < sizeof (ino_t); ++i)
1516 {
1517 /* shifting off the byte offsets out of "endianness" */
1518 int offs = (offsetof (eio_dirent, inode) + (endianness & 0xff)) * 8;
1519 endianness >>= 8;
1520
1521 for (j = 0; j < 8; ++j)
1522 if (inode_bits & (((ino_t)1) << (i * 8 + j)))
1523 *bit++ = offs + j;
1524 }
1525
1526 for (j = 0; j < 8; ++j)
1527 if (score_bits & (1 << j))
1528 *bit++ = offsetof (eio_dirent, score) * 8 + j;
1529 }
1530
1531 /* now actually do the sorting (a variant of MSD radix sort) */
1532 {
1533 eio_dirent *base_stk [9 + sizeof (ino_t) * 8], *base;
1534 eio_dirent *end_stk [9 + sizeof (ino_t) * 8], *end;
1535 unsigned char *bit_stk [9 + sizeof (ino_t) * 8];
1536 int stk_idx = 0;
1537
1538 base_stk [stk_idx] = dents;
1539 end_stk [stk_idx] = dents + size;
1540 bit_stk [stk_idx] = bit - 1;
1541
1542 do
1543 {
1544 base = base_stk [stk_idx];
1545 end = end_stk [stk_idx];
1546 bit = bit_stk [stk_idx];
1547
1548 for (;;)
1549 {
1550 unsigned char O = *bit >> 3;
1551 unsigned char M = 1 << (*bit & 7);
1552
1553 eio_dirent *a = base;
1554 eio_dirent *b = end;
1555
1556 if (b - a < EIO_SORT_CUTOFF)
1557 break;
1558
1559 /* now bit-partition the array on the bit */
1560 /* this ugly asymmetric loop seems to perform much better than typical */
1561 /* partition algos found in the literature */
1562 do
1563 if (!(((unsigned char *)a)[O] & M))
1564 ++a;
1565 else if (!(((unsigned char *)--b)[O] & M))
1566 {
1567 eio_dirent tmp = *a; *a = *b; *b = tmp;
1568 ++a;
1569 }
1570 while (b > a);
1571
1572 /* next bit, or stop, if no bits left in this path */
1573 if (!*--bit)
1574 break;
1575
1576 base_stk [stk_idx] = a;
1577 end_stk [stk_idx] = end;
1578 bit_stk [stk_idx] = bit;
1579 ++stk_idx;
1580
1581 end = a;
1582 }
1583 }
1584 while (stk_idx--);
1585 }
1586}
1587
1588static void
1589eio_dent_insertion_sort (eio_dirent *dents, int size)
1590{
1591 /* first move the smallest element to the front, to act as a sentinel */
1592 {
1593 int i;
1594 eio_dirent *min = dents;
1595
1596 /* the radix pre-pass ensures that the minimum element is in the first EIO_SORT_CUTOFF + 1 elements */
1597 for (i = size > EIO_SORT_FAST ? EIO_SORT_CUTOFF + 1 : size; --i; )
1598 if (EIO_DENT_CMP (dents [i], <, *min))
1599 min = &dents [i];
1600
1601 /* swap elements 0 and j (minimum) */
1602 {
1603 eio_dirent tmp = *dents; *dents = *min; *min = tmp;
1604 }
1605 }
1606
1607 /* then do standard insertion sort, assuming that all elements are >= dents [0] */
1608 {
1609 eio_dirent *i, *j;
1610
1611 for (i = dents + 1; i < dents + size; ++i)
1612 {
1613 eio_dirent value = *i;
1614
1615 for (j = i - 1; EIO_DENT_CMP (*j, >, value); --j)
1616 j [1] = j [0];
1617
1618 j [1] = value;
1619 }
1620 }
1621}
1622
1623static void
1624eio_dent_sort (eio_dirent *dents, int size, signed char score_bits, ino_t inode_bits)
1625{
1626 if (size <= 1)
1627 return; /* our insertion sort relies on size > 0 */
1628
1629 /* first we use a radix sort, but only for dirs >= EIO_SORT_FAST */
1630 /* and stop sorting when the partitions are <= EIO_SORT_CUTOFF */
1631 eio_dent_radix_sort (dents, size, score_bits, inode_bits);
1632
1633 /* use an insertion sort at the end, or for small arrays, */
1634 /* as insertion sort is more efficient for small partitions */
1635 eio_dent_insertion_sort (dents, size);
1636}
1637
962/* read a full directory */ 1638/* read a full directory */
963static void 1639static void
964eio__scandir (eio_req *req, etp_worker *self) 1640eio__scandir (eio_req *req, etp_worker *self)
965{ 1641{
966 DIR *dirp; 1642 DIR *dirp;
967 EIO_STRUCT_DIRENT *entp; 1643 EIO_STRUCT_DIRENT *entp;
968 char *name, *names; 1644 char *name, *names;
969 int memlen = 4096; 1645 int namesalloc = 4096;
970 int memofs = 0; 1646 int namesoffs = 0;
1647 int flags = req->int1;
1648 eio_dirent *dents = 0;
1649 int dentalloc = 128;
971 int res = 0; 1650 int dentoffs = 0;
1651 ino_t inode_bits = 0;
1652
1653 req->result = -1;
1654
1655 if (!(flags & EIO_READDIR_DENTS))
1656 flags &= ~(EIO_READDIR_DIRS_FIRST | EIO_READDIR_STAT_ORDER);
972 1657
973 X_LOCK (wrklock); 1658 X_LOCK (wrklock);
974 /* the corresponding closedir is in ETP_WORKER_CLEAR */ 1659 /* the corresponding closedir is in ETP_WORKER_CLEAR */
975 self->dirp = dirp = opendir (req->ptr1); 1660 self->dirp = dirp = opendir (req->ptr1);
1661
976 req->flags |= EIO_FLAG_PTR2_FREE; 1662 if (req->flags & EIO_FLAG_PTR1_FREE)
1663 free (req->ptr1);
1664
1665 req->flags |= EIO_FLAG_PTR1_FREE | EIO_FLAG_PTR2_FREE;
1666 req->ptr1 = dents = flags ? malloc (dentalloc * sizeof (eio_dirent)) : 0;
977 req->ptr2 = names = malloc (memlen); 1667 req->ptr2 = names = malloc (namesalloc);
978 X_UNLOCK (wrklock); 1668 X_UNLOCK (wrklock);
979 1669
980 if (dirp && names) 1670 if (dirp && names && (!flags || dents))
981 for (;;) 1671 for (;;)
982 { 1672 {
983 errno = 0; 1673 errno = 0;
984 entp = readdir (dirp); 1674 entp = readdir (dirp);
985 1675
986 if (!entp) 1676 if (!entp)
1677 {
1678 if (errno)
1679 break;
1680
1681 /* sort etc. */
1682 req->int1 = flags;
1683 req->result = dentoffs;
1684
1685 if (flags & EIO_READDIR_STAT_ORDER)
1686 eio_dent_sort (dents, dentoffs, flags & EIO_READDIR_DIRS_FIRST ? 7 : 0, inode_bits);
1687 else if (flags & EIO_READDIR_DIRS_FIRST)
1688 if (flags & EIO_READDIR_FOUND_UNKNOWN)
1689 eio_dent_sort (dents, dentoffs, 7, inode_bits); /* sort by score and inode */
1690 else
1691 {
1692 /* in this case, all is known, and we just put dirs first and sort them */
1693 eio_dirent *oth = dents + dentoffs;
1694 eio_dirent *dir = dents;
1695
1696 /* now partition dirs to the front, and non-dirs to the back */
1697 /* by walking from both sides and swapping if necessary */
1698 while (oth > dir)
1699 {
1700 if (dir->type == EIO_DT_DIR)
1701 ++dir;
1702 else if ((--oth)->type == EIO_DT_DIR)
1703 {
1704 eio_dirent tmp = *dir; *dir = *oth; *oth = tmp;
1705
1706 ++dir;
1707 }
1708 }
1709
1710 /* now sort the dirs only (dirs all have the same score) */
1711 eio_dent_sort (dents, dir - dents, 0, inode_bits);
1712 }
1713
987 break; 1714 break;
1715 }
988 1716
1717 /* now add the entry to our list(s) */
989 name = entp->d_name; 1718 name = entp->d_name;
990 1719
1720 /* skip . and .. entries */
991 if (name [0] != '.' || (name [1] && (name [1] != '.' || name [2]))) 1721 if (name [0] != '.' || (name [1] && (name [1] != '.' || name [2])))
992 { 1722 {
993 int len = strlen (name) + 1; 1723 int len = D_NAMLEN (entp) + 1;
994 1724
995 res++; 1725 while (ecb_expect_false (namesoffs + len > namesalloc))
996
997 while (memofs + len > memlen)
998 { 1726 {
999 memlen *= 2; 1727 namesalloc *= 2;
1000 X_LOCK (wrklock); 1728 X_LOCK (wrklock);
1001 req->ptr2 = names = realloc (names, memlen); 1729 req->ptr2 = names = realloc (names, namesalloc);
1002 X_UNLOCK (wrklock); 1730 X_UNLOCK (wrklock);
1003 1731
1004 if (!names) 1732 if (!names)
1005 break; 1733 break;
1006 } 1734 }
1007 1735
1008 memcpy (names + memofs, name, len); 1736 memcpy (names + namesoffs, name, len);
1737
1738 if (dents)
1739 {
1740 struct eio_dirent *ent;
1741
1742 if (ecb_expect_false (dentoffs == dentalloc))
1743 {
1744 dentalloc *= 2;
1745 X_LOCK (wrklock);
1746 req->ptr1 = dents = realloc (dents, dentalloc * sizeof (eio_dirent));
1747 X_UNLOCK (wrklock);
1748
1749 if (!dents)
1750 break;
1751 }
1752
1753 ent = dents + dentoffs;
1754
1755 ent->nameofs = namesoffs; /* rather dirtily we store the offset in the pointer */
1756 ent->namelen = len - 1;
1757 ent->inode = D_INO (entp);
1758
1759 inode_bits |= ent->inode;
1760
1761 switch (D_TYPE (entp))
1762 {
1763 default:
1764 ent->type = EIO_DT_UNKNOWN;
1765 flags |= EIO_READDIR_FOUND_UNKNOWN;
1766 break;
1767
1768 #ifdef DT_FIFO
1769 case DT_FIFO: ent->type = EIO_DT_FIFO; break;
1770 #endif
1771 #ifdef DT_CHR
1772 case DT_CHR: ent->type = EIO_DT_CHR; break;
1773 #endif
1774 #ifdef DT_MPC
1775 case DT_MPC: ent->type = EIO_DT_MPC; break;
1776 #endif
1777 #ifdef DT_DIR
1778 case DT_DIR: ent->type = EIO_DT_DIR; break;
1779 #endif
1780 #ifdef DT_NAM
1781 case DT_NAM: ent->type = EIO_DT_NAM; break;
1782 #endif
1783 #ifdef DT_BLK
1784 case DT_BLK: ent->type = EIO_DT_BLK; break;
1785 #endif
1786 #ifdef DT_MPB
1787 case DT_MPB: ent->type = EIO_DT_MPB; break;
1788 #endif
1789 #ifdef DT_REG
1790 case DT_REG: ent->type = EIO_DT_REG; break;
1791 #endif
1792 #ifdef DT_NWK
1793 case DT_NWK: ent->type = EIO_DT_NWK; break;
1794 #endif
1795 #ifdef DT_CMP
1796 case DT_CMP: ent->type = EIO_DT_CMP; break;
1797 #endif
1798 #ifdef DT_LNK
1799 case DT_LNK: ent->type = EIO_DT_LNK; break;
1800 #endif
1801 #ifdef DT_SOCK
1802 case DT_SOCK: ent->type = EIO_DT_SOCK; break;
1803 #endif
1804 #ifdef DT_DOOR
1805 case DT_DOOR: ent->type = EIO_DT_DOOR; break;
1806 #endif
1807 #ifdef DT_WHT
1808 case DT_WHT: ent->type = EIO_DT_WHT; break;
1809 #endif
1810 }
1811
1812 ent->score = 7;
1813
1814 if (flags & EIO_READDIR_DIRS_FIRST)
1815 {
1816 if (ent->type == EIO_DT_UNKNOWN)
1817 {
1818 if (*name == '.') /* leading dots are likely directories, and, in any case, rare */
1819 ent->score = 1;
1820 else if (!strchr (name, '.')) /* absense of dots indicate likely dirs */
1821 ent->score = len <= 2 ? 4 - len : len <= 4 ? 4 : len <= 7 ? 5 : 6; /* shorter == more likely dir, but avoid too many classes */
1822 }
1823 else if (ent->type == EIO_DT_DIR)
1824 ent->score = 0;
1825 }
1826 }
1827
1009 memofs += len; 1828 namesoffs += len;
1829 ++dentoffs;
1830 }
1831
1832 if (EIO_CANCELLED (req))
1833 {
1834 errno = ECANCELED;
1835 break;
1010 } 1836 }
1011 } 1837 }
1012
1013 if (errno)
1014 res = -1;
1015
1016 req->result = res;
1017}
1018
1019#if !(_POSIX_MAPPED_FILES && _POSIX_SYNCHRONIZED_IO)
1020# undef msync
1021# define msync(a,b,c) ENOSYS
1022#endif
1023
1024int
1025eio__mtouch (void *mem, size_t len, int flags)
1026{
1027 intptr_t addr = (intptr_t)mem;
1028 intptr_t end = addr + len;
1029#ifdef PAGESIZE
1030 const intptr_t page = PAGESIZE;
1031#else
1032 static intptr_t page;
1033
1034 if (!page)
1035 page = sysconf (_SC_PAGESIZE);
1036#endif
1037
1038 addr &= ~(page - 1); /* assume page size is always a power of two */
1039
1040 if (addr < end)
1041 if (flags) /* modify */
1042 do { *((volatile sig_atomic_t *)addr) |= 0; } while ((addr += page) < len);
1043 else
1044 do { *((volatile sig_atomic_t *)addr) ; } while ((addr += page) < len);
1045
1046 return 0;
1047} 1838}
1048 1839
1049/*****************************************************************************/ 1840/*****************************************************************************/
1050 1841
1051#define ALLOC(len) \ 1842#define ALLOC(len) \
1066X_THREAD_PROC (etp_proc) 1857X_THREAD_PROC (etp_proc)
1067{ 1858{
1068 ETP_REQ *req; 1859 ETP_REQ *req;
1069 struct timespec ts; 1860 struct timespec ts;
1070 etp_worker *self = (etp_worker *)thr_arg; 1861 etp_worker *self = (etp_worker *)thr_arg;
1862 int timeout;
1071 1863
1072 /* try to distribute timeouts somewhat randomly */ 1864 /* try to distribute timeouts somewhat evenly */
1073 ts.tv_nsec = ((unsigned long)self & 1023UL) * (1000000000UL / 1024UL); 1865 ts.tv_nsec = ((unsigned long)self & 1023UL) * (1000000000UL / 1024UL);
1074 1866
1075 for (;;) 1867 for (;;)
1076 { 1868 {
1869 ts.tv_sec = 0;
1870
1077 X_LOCK (reqlock); 1871 X_LOCK (reqlock);
1078 1872
1079 for (;;) 1873 for (;;)
1080 { 1874 {
1081 self->req = req = reqq_shift (&req_queue); 1875 self->req = req = reqq_shift (&req_queue);
1082 1876
1083 if (req) 1877 if (req)
1084 break; 1878 break;
1085 1879
1880 if (ts.tv_sec == 1) /* no request, but timeout detected, let's quit */
1881 {
1882 X_UNLOCK (reqlock);
1883 X_LOCK (wrklock);
1884 --started;
1885 X_UNLOCK (wrklock);
1886 goto quit;
1887 }
1888
1086 ++idle; 1889 ++idle;
1087 1890
1088 ts.tv_sec = time (0) + IDLE_TIMEOUT; 1891 if (idle <= max_idle)
1089 if (X_COND_TIMEDWAIT (reqwait, reqlock, ts) == ETIMEDOUT) 1892 /* we are allowed to idle, so do so without any timeout */
1893 X_COND_WAIT (reqwait, reqlock);
1894 else
1090 { 1895 {
1091 if (idle > max_idle) 1896 /* initialise timeout once */
1092 { 1897 if (!ts.tv_sec)
1093 --idle; 1898 ts.tv_sec = time (0) + idle_timeout;
1094 X_UNLOCK (reqlock);
1095 X_LOCK (wrklock);
1096 --started;
1097 X_UNLOCK (wrklock);
1098 goto quit;
1099 }
1100 1899
1101 /* we are allowed to idle, so do so without any timeout */
1102 X_COND_WAIT (reqwait, reqlock); 1900 if (X_COND_TIMEDWAIT (reqwait, reqlock, ts) == ETIMEDOUT)
1901 ts.tv_sec = 1; /* assuming this is not a value computed above.,.. */
1103 } 1902 }
1104 1903
1105 --idle; 1904 --idle;
1106 } 1905 }
1107 1906
1110 X_UNLOCK (reqlock); 1909 X_UNLOCK (reqlock);
1111 1910
1112 if (req->type < 0) 1911 if (req->type < 0)
1113 goto quit; 1912 goto quit;
1114 1913
1115 if (!EIO_CANCELLED (req))
1116 ETP_EXECUTE (self, req); 1914 ETP_EXECUTE (self, req);
1117 1915
1118 X_LOCK (reslock); 1916 X_LOCK (reslock);
1119 1917
1120 ++npending; 1918 ++npending;
1121 1919
1136 return 0; 1934 return 0;
1137} 1935}
1138 1936
1139/*****************************************************************************/ 1937/*****************************************************************************/
1140 1938
1939int ecb_cold
1141int eio_init (void (*want_poll)(void), void (*done_poll)(void)) 1940eio_init (void (*want_poll)(void), void (*done_poll)(void))
1142{ 1941{
1143 return etp_init (want_poll, done_poll); 1942 return etp_init (want_poll, done_poll);
1144} 1943}
1145 1944
1945ecb_inline void
1146static void eio_api_destroy (eio_req *req) 1946eio_api_destroy (eio_req *req)
1147{ 1947{
1148 free (req); 1948 free (req);
1149} 1949}
1150 1950
1151#define REQ(rtype) \ 1951#define REQ(rtype) \
1170 { \ 1970 { \
1171 eio_api_destroy (req); \ 1971 eio_api_destroy (req); \
1172 return 0; \ 1972 return 0; \
1173 } 1973 }
1174 1974
1975static void
1175static void eio_execute (etp_worker *self, eio_req *req) 1976eio_execute (etp_worker *self, eio_req *req)
1176{ 1977{
1177 errno = 0; 1978 if (ecb_expect_false (EIO_CANCELLED (req)))
1979 {
1980 req->result = -1;
1981 req->errorno = ECANCELED;
1982 return;
1983 }
1178 1984
1179 switch (req->type) 1985 switch (req->type)
1180 { 1986 {
1181 case EIO_READ: ALLOC (req->size); 1987 case EIO_READ: ALLOC (req->size);
1182 req->result = req->offs >= 0 1988 req->result = req->offs >= 0
1194 case EIO_LSTAT: ALLOC (sizeof (EIO_STRUCT_STAT)); 2000 case EIO_LSTAT: ALLOC (sizeof (EIO_STRUCT_STAT));
1195 req->result = lstat (req->ptr1, (EIO_STRUCT_STAT *)req->ptr2); break; 2001 req->result = lstat (req->ptr1, (EIO_STRUCT_STAT *)req->ptr2); break;
1196 case EIO_FSTAT: ALLOC (sizeof (EIO_STRUCT_STAT)); 2002 case EIO_FSTAT: ALLOC (sizeof (EIO_STRUCT_STAT));
1197 req->result = fstat (req->int1, (EIO_STRUCT_STAT *)req->ptr2); break; 2003 req->result = fstat (req->int1, (EIO_STRUCT_STAT *)req->ptr2); break;
1198 2004
2005 case EIO_STATVFS: ALLOC (sizeof (EIO_STRUCT_STATVFS));
2006 req->result = statvfs (req->ptr1, (EIO_STRUCT_STATVFS *)req->ptr2); break;
2007 case EIO_FSTATVFS: ALLOC (sizeof (EIO_STRUCT_STATVFS));
2008 req->result = fstatvfs (req->int1, (EIO_STRUCT_STATVFS *)req->ptr2); break;
2009
1199 case EIO_CHOWN: req->result = chown (req->ptr1, req->int2, req->int3); break; 2010 case EIO_CHOWN: req->result = chown (req->ptr1, req->int2, req->int3); break;
1200 case EIO_FCHOWN: req->result = fchown (req->int1, req->int2, req->int3); break; 2011 case EIO_FCHOWN: req->result = fchown (req->int1, req->int2, req->int3); break;
1201 case EIO_CHMOD: req->result = chmod (req->ptr1, (mode_t)req->int2); break; 2012 case EIO_CHMOD: req->result = chmod (req->ptr1, (mode_t)req->int2); break;
1202 case EIO_FCHMOD: req->result = fchmod (req->int1, (mode_t)req->int2); break; 2013 case EIO_FCHMOD: req->result = fchmod (req->int1, (mode_t)req->int2); break;
1203 case EIO_TRUNCATE: req->result = truncate (req->ptr1, req->offs); break; 2014 case EIO_TRUNCATE: req->result = truncate (req->ptr1, req->offs); break;
1210 case EIO_RMDIR: req->result = rmdir (req->ptr1); break; 2021 case EIO_RMDIR: req->result = rmdir (req->ptr1); break;
1211 case EIO_MKDIR: req->result = mkdir (req->ptr1, (mode_t)req->int2); break; 2022 case EIO_MKDIR: req->result = mkdir (req->ptr1, (mode_t)req->int2); break;
1212 case EIO_RENAME: req->result = rename (req->ptr1, req->ptr2); break; 2023 case EIO_RENAME: req->result = rename (req->ptr1, req->ptr2); break;
1213 case EIO_LINK: req->result = link (req->ptr1, req->ptr2); break; 2024 case EIO_LINK: req->result = link (req->ptr1, req->ptr2); break;
1214 case EIO_SYMLINK: req->result = symlink (req->ptr1, req->ptr2); break; 2025 case EIO_SYMLINK: req->result = symlink (req->ptr1, req->ptr2); break;
1215 case EIO_MKNOD: req->result = mknod (req->ptr1, (mode_t)req->int2, (dev_t)req->int3); break; 2026 case EIO_MKNOD: req->result = mknod (req->ptr1, (mode_t)req->int2, (dev_t)req->offs); break;
1216 2027
2028 case EIO_REALPATH: eio__realpath (req, self); break;
2029
1217 case EIO_READLINK: ALLOC (NAME_MAX); 2030 case EIO_READLINK: ALLOC (PATH_MAX);
1218 req->result = readlink (req->ptr1, req->ptr2, NAME_MAX); break; 2031 req->result = readlink (req->ptr1, req->ptr2, PATH_MAX); break;
1219 2032
1220 case EIO_SYNC: req->result = 0; sync (); break; 2033 case EIO_SYNC: req->result = 0; sync (); break;
1221 case EIO_FSYNC: req->result = fsync (req->int1); break; 2034 case EIO_FSYNC: req->result = fsync (req->int1); break;
1222 case EIO_FDATASYNC: req->result = fdatasync (req->int1); break; 2035 case EIO_FDATASYNC: req->result = fdatasync (req->int1); break;
1223 case EIO_MSYNC: req->result = msync (req->ptr2, req->size, req->int1); break; 2036 case EIO_MSYNC: req->result = eio__msync (req->ptr2, req->size, req->int1); break;
2037 case EIO_MTOUCH: req->result = eio__mtouch (req); break;
1224 case EIO_MTOUCH: req->result = eio__mtouch (req->ptr2, req->size, req->int1); break; 2038 case EIO_MLOCK: req->result = eio__mlock (req->ptr2, req->size); break;
2039 case EIO_MLOCKALL: req->result = eio__mlockall (req->int1); break;
1225 case EIO_SYNC_FILE_RANGE: req->result = eio__sync_file_range (req->int1, req->offs, req->size, req->int2); break; 2040 case EIO_SYNC_FILE_RANGE: req->result = eio__sync_file_range (req->int1, req->offs, req->size, req->int2); break;
2041 case EIO_FALLOCATE: req->result = eio__fallocate (req->int1, req->int2, req->offs, req->size); break;
1226 2042
1227 case EIO_READDIR: eio__scandir (req, self); break; 2043 case EIO_READDIR: eio__scandir (req, self); break;
1228 2044
1229 case EIO_BUSY: 2045 case EIO_BUSY:
1230#ifdef _WIN32 2046#ifdef _WIN32
1231 Sleep (req->nv1 * 1000.); 2047 Sleep (req->nv1 * 1e3);
1232#else 2048#else
1233 { 2049 {
1234 struct timeval tv; 2050 struct timeval tv;
1235 2051
1236 tv.tv_sec = req->nv1; 2052 tv.tv_sec = req->nv1;
1237 tv.tv_usec = (req->nv1 - tv.tv_sec) * 1000000.; 2053 tv.tv_usec = (req->nv1 - tv.tv_sec) * 1e6;
1238 2054
1239 req->result = select (0, 0, 0, 0, &tv); 2055 req->result = select (0, 0, 0, 0, &tv);
1240 } 2056 }
1241#endif 2057#endif
1242 break; 2058 break;
1257 times = tv; 2073 times = tv;
1258 } 2074 }
1259 else 2075 else
1260 times = 0; 2076 times = 0;
1261 2077
1262
1263 req->result = req->type == EIO_FUTIME 2078 req->result = req->type == EIO_FUTIME
1264 ? futimes (req->int1, times) 2079 ? futimes (req->int1, times)
1265 : utimes (req->ptr1, times); 2080 : utimes (req->ptr1, times);
1266 } 2081 }
1267 break; 2082 break;
1272 case EIO_NOP: 2087 case EIO_NOP:
1273 req->result = 0; 2088 req->result = 0;
1274 break; 2089 break;
1275 2090
1276 case EIO_CUSTOM: 2091 case EIO_CUSTOM:
1277 ((void (*)(eio_req *))req->feed) (req); 2092 req->feed (req);
1278 break; 2093 break;
1279 2094
1280 default: 2095 default:
2096 errno = ENOSYS;
1281 req->result = -1; 2097 req->result = -1;
1282 break; 2098 break;
1283 } 2099 }
1284 2100
1285 req->errorno = errno; 2101 req->errorno = errno;
1315eio_req *eio_mtouch (void *addr, size_t length, int flags, int pri, eio_cb cb, void *data) 2131eio_req *eio_mtouch (void *addr, size_t length, int flags, int pri, eio_cb cb, void *data)
1316{ 2132{
1317 REQ (EIO_MTOUCH); req->ptr2 = addr; req->size = length; req->int1 = flags; SEND; 2133 REQ (EIO_MTOUCH); req->ptr2 = addr; req->size = length; req->int1 = flags; SEND;
1318} 2134}
1319 2135
2136eio_req *eio_mlock (void *addr, size_t length, int pri, eio_cb cb, void *data)
2137{
2138 REQ (EIO_MLOCK); req->ptr2 = addr; req->size = length; SEND;
2139}
2140
2141eio_req *eio_mlockall (int flags, int pri, eio_cb cb, void *data)
2142{
2143 REQ (EIO_MLOCKALL); req->int1 = flags; SEND;
2144}
2145
1320eio_req *eio_sync_file_range (int fd, off_t offset, size_t nbytes, unsigned int flags, int pri, eio_cb cb, void *data) 2146eio_req *eio_sync_file_range (int fd, off_t offset, size_t nbytes, unsigned int flags, int pri, eio_cb cb, void *data)
1321{ 2147{
1322 REQ (EIO_SYNC_FILE_RANGE); req->int1 = fd; req->offs = offset; req->size = nbytes; req->int2 = flags; SEND; 2148 REQ (EIO_SYNC_FILE_RANGE); req->int1 = fd; req->offs = offset; req->size = nbytes; req->int2 = flags; SEND;
1323} 2149}
1324 2150
2151eio_req *eio_fallocate (int fd, int mode, off_t offset, size_t len, int pri, eio_cb cb, void *data)
2152{
2153 REQ (EIO_FALLOCATE); req->int1 = fd; req->int2 = mode; req->offs = offset; req->size = len; SEND;
2154}
2155
1325eio_req *eio_fdatasync (int fd, int pri, eio_cb cb, void *data) 2156eio_req *eio_fdatasync (int fd, int pri, eio_cb cb, void *data)
1326{ 2157{
1327 REQ (EIO_FDATASYNC); req->int1 = fd; SEND; 2158 REQ (EIO_FDATASYNC); req->int1 = fd; SEND;
1328} 2159}
1329 2160
1350eio_req *eio_fstat (int fd, int pri, eio_cb cb, void *data) 2181eio_req *eio_fstat (int fd, int pri, eio_cb cb, void *data)
1351{ 2182{
1352 REQ (EIO_FSTAT); req->int1 = fd; SEND; 2183 REQ (EIO_FSTAT); req->int1 = fd; SEND;
1353} 2184}
1354 2185
2186eio_req *eio_fstatvfs (int fd, int pri, eio_cb cb, void *data)
2187{
2188 REQ (EIO_FSTATVFS); req->int1 = fd; SEND;
2189}
2190
1355eio_req *eio_futime (int fd, double atime, double mtime, int pri, eio_cb cb, void *data) 2191eio_req *eio_futime (int fd, double atime, double mtime, int pri, eio_cb cb, void *data)
1356{ 2192{
1357 REQ (EIO_FUTIME); req->int1 = fd; req->nv1 = atime; req->nv2 = mtime; SEND; 2193 REQ (EIO_FUTIME); req->int1 = fd; req->nv1 = atime; req->nv2 = mtime; SEND;
1358} 2194}
1359 2195
1365eio_req *eio_fchmod (int fd, mode_t mode, int pri, eio_cb cb, void *data) 2201eio_req *eio_fchmod (int fd, mode_t mode, int pri, eio_cb cb, void *data)
1366{ 2202{
1367 REQ (EIO_FCHMOD); req->int1 = fd; req->int2 = (long)mode; SEND; 2203 REQ (EIO_FCHMOD); req->int1 = fd; req->int2 = (long)mode; SEND;
1368} 2204}
1369 2205
1370eio_req *eio_fchown (int fd, uid_t uid, gid_t gid, int pri, eio_cb cb, void *data) 2206eio_req *eio_fchown (int fd, eio_uid_t uid, eio_gid_t gid, int pri, eio_cb cb, void *data)
1371{ 2207{
1372 REQ (EIO_FCHOWN); req->int1 = fd; req->int2 = (long)uid; req->int3 = (long)gid; SEND; 2208 REQ (EIO_FCHOWN); req->int1 = fd; req->int2 = (long)uid; req->int3 = (long)gid; SEND;
1373} 2209}
1374 2210
1375eio_req *eio_dup2 (int fd, int fd2, int pri, eio_cb cb, void *data) 2211eio_req *eio_dup2 (int fd, int fd2, int pri, eio_cb cb, void *data)
1395eio_req *eio_truncate (const char *path, off_t offset, int pri, eio_cb cb, void *data) 2231eio_req *eio_truncate (const char *path, off_t offset, int pri, eio_cb cb, void *data)
1396{ 2232{
1397 REQ (EIO_TRUNCATE); PATH; req->offs = offset; SEND; 2233 REQ (EIO_TRUNCATE); PATH; req->offs = offset; SEND;
1398} 2234}
1399 2235
1400eio_req *eio_chown (const char *path, uid_t uid, gid_t gid, int pri, eio_cb cb, void *data) 2236eio_req *eio_chown (const char *path, eio_uid_t uid, eio_gid_t gid, int pri, eio_cb cb, void *data)
1401{ 2237{
1402 REQ (EIO_CHOWN); PATH; req->int2 = (long)uid; req->int3 = (long)gid; SEND; 2238 REQ (EIO_CHOWN); PATH; req->int2 = (long)uid; req->int3 = (long)gid; SEND;
1403} 2239}
1404 2240
1405eio_req *eio_chmod (const char *path, mode_t mode, int pri, eio_cb cb, void *data) 2241eio_req *eio_chmod (const char *path, mode_t mode, int pri, eio_cb cb, void *data)
1421eio_req *eio_readlink (const char *path, int pri, eio_cb cb, void *data) 2257eio_req *eio_readlink (const char *path, int pri, eio_cb cb, void *data)
1422{ 2258{
1423 return eio__1path (EIO_READLINK, path, pri, cb, data); 2259 return eio__1path (EIO_READLINK, path, pri, cb, data);
1424} 2260}
1425 2261
2262eio_req *eio_realpath (const char *path, int pri, eio_cb cb, void *data)
2263{
2264 return eio__1path (EIO_REALPATH, path, pri, cb, data);
2265}
2266
1426eio_req *eio_stat (const char *path, int pri, eio_cb cb, void *data) 2267eio_req *eio_stat (const char *path, int pri, eio_cb cb, void *data)
1427{ 2268{
1428 return eio__1path (EIO_STAT, path, pri, cb, data); 2269 return eio__1path (EIO_STAT, path, pri, cb, data);
1429} 2270}
1430 2271
1431eio_req *eio_lstat (const char *path, int pri, eio_cb cb, void *data) 2272eio_req *eio_lstat (const char *path, int pri, eio_cb cb, void *data)
1432{ 2273{
1433 return eio__1path (EIO_LSTAT, path, pri, cb, data); 2274 return eio__1path (EIO_LSTAT, path, pri, cb, data);
1434} 2275}
1435 2276
2277eio_req *eio_statvfs (const char *path, int pri, eio_cb cb, void *data)
2278{
2279 return eio__1path (EIO_STATVFS, path, pri, cb, data);
2280}
2281
1436eio_req *eio_unlink (const char *path, int pri, eio_cb cb, void *data) 2282eio_req *eio_unlink (const char *path, int pri, eio_cb cb, void *data)
1437{ 2283{
1438 return eio__1path (EIO_UNLINK, path, pri, cb, data); 2284 return eio__1path (EIO_UNLINK, path, pri, cb, data);
1439} 2285}
1440 2286
1441eio_req *eio_rmdir (const char *path, int pri, eio_cb cb, void *data) 2287eio_req *eio_rmdir (const char *path, int pri, eio_cb cb, void *data)
1442{ 2288{
1443 return eio__1path (EIO_RMDIR, path, pri, cb, data); 2289 return eio__1path (EIO_RMDIR, path, pri, cb, data);
1444} 2290}
1445 2291
1446eio_req *eio_readdir (const char *path, int pri, eio_cb cb, void *data) 2292eio_req *eio_readdir (const char *path, int flags, int pri, eio_cb cb, void *data)
1447{ 2293{
1448 return eio__1path (EIO_READDIR, path, pri, cb, data); 2294 REQ (EIO_READDIR); PATH; req->int1 = flags; SEND;
1449} 2295}
1450 2296
1451eio_req *eio_mknod (const char *path, mode_t mode, dev_t dev, int pri, eio_cb cb, void *data) 2297eio_req *eio_mknod (const char *path, mode_t mode, dev_t dev, int pri, eio_cb cb, void *data)
1452{ 2298{
1453 REQ (EIO_MKNOD); PATH; req->int2 = (long)mode; req->int3 = (long)dev; SEND; 2299 REQ (EIO_MKNOD); PATH; req->int2 = (long)mode; req->offs = (off_t)dev; SEND;
1454} 2300}
1455 2301
1456static eio_req * 2302static eio_req *
1457eio__2path (int type, const char *path, const char *new_path, int pri, eio_cb cb, void *data) 2303eio__2path (int type, const char *path, const char *new_path, int pri, eio_cb cb, void *data)
1458{ 2304{
1482eio_req *eio_rename (const char *path, const char *new_path, int pri, eio_cb cb, void *data) 2328eio_req *eio_rename (const char *path, const char *new_path, int pri, eio_cb cb, void *data)
1483{ 2329{
1484 return eio__2path (EIO_RENAME, path, new_path, pri, cb, data); 2330 return eio__2path (EIO_RENAME, path, new_path, pri, cb, data);
1485} 2331}
1486 2332
1487eio_req *eio_custom (eio_cb execute, int pri, eio_cb cb, void *data) 2333eio_req *eio_custom (void (*execute)(eio_req *), int pri, eio_cb cb, void *data)
1488{ 2334{
1489 REQ (EIO_CUSTOM); req->feed = (void (*)(eio_req *))execute; SEND; 2335 REQ (EIO_CUSTOM); req->feed = execute; SEND;
1490} 2336}
1491 2337
1492#endif 2338#endif
1493 2339
1494eio_req *eio_grp (eio_cb cb, void *data) 2340eio_req *eio_grp (eio_cb cb, void *data)
1503#undef SEND 2349#undef SEND
1504 2350
1505/*****************************************************************************/ 2351/*****************************************************************************/
1506/* grp functions */ 2352/* grp functions */
1507 2353
2354void
1508void eio_grp_feed (eio_req *grp, void (*feed)(eio_req *req), int limit) 2355eio_grp_feed (eio_req *grp, void (*feed)(eio_req *req), int limit)
1509{ 2356{
1510 grp->int2 = limit; 2357 grp->int2 = limit;
1511 grp->feed = feed; 2358 grp->feed = feed;
1512 2359
1513 grp_try_feed (grp); 2360 grp_try_feed (grp);
1514} 2361}
1515 2362
2363void
1516void eio_grp_limit (eio_req *grp, int limit) 2364eio_grp_limit (eio_req *grp, int limit)
1517{ 2365{
1518 grp->int2 = limit; 2366 grp->int2 = limit;
1519 2367
1520 grp_try_feed (grp); 2368 grp_try_feed (grp);
1521} 2369}
1522 2370
2371void
1523void eio_grp_add (eio_req *grp, eio_req *req) 2372eio_grp_add (eio_req *grp, eio_req *req)
1524{ 2373{
1525 assert (("cannot add requests to IO::AIO::GRP after the group finished", grp->int1 != 2)); 2374 assert (("cannot add requests to IO::AIO::GRP after the group finished", grp->int1 != 2));
1526 2375
1527 grp->flags |= EIO_FLAG_GROUPADD; 2376 grp->flags |= EIO_FLAG_GROUPADD;
1528 2377
1539} 2388}
1540 2389
1541/*****************************************************************************/ 2390/*****************************************************************************/
1542/* misc garbage */ 2391/* misc garbage */
1543 2392
2393eio_ssize_t
1544ssize_t eio_sendfile_sync (int ofd, int ifd, off_t offset, size_t count) 2394eio_sendfile_sync (int ofd, int ifd, off_t offset, size_t count)
1545{ 2395{
1546 etp_worker wrk; 2396 etp_worker wrk;
2397 eio_ssize_t ret;
1547 2398
1548 wrk.dbuf = 0; 2399 wrk.dbuf = 0;
1549 2400
1550 eio__sendfile (ofd, ifd, offset, count, &wrk); 2401 ret = eio__sendfile (ofd, ifd, offset, count, &wrk);
1551 2402
1552 if (wrk.dbuf) 2403 if (wrk.dbuf)
1553 free (wrk.dbuf); 2404 free (wrk.dbuf);
1554}
1555 2405
2406 return ret;
2407}
2408

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines