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

Comparing cvsroot/libeio/eio.c (file contents):
Revision 1.52 by root, Fri Jan 8 09:29:56 2010 UTC vs.
Revision 1.81 by root, Tue Jul 5 20:34:42 2011 UTC

1/* 1/*
2 * libeio implementation 2 * libeio implementation
3 * 3 *
4 * Copyright (c) 2007,2008,2009,2010 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"
41 46
42#ifdef EIO_STACKSIZE 47#ifdef EIO_STACKSIZE
43# define XTHREAD_STACKSIZE EIO_STACKSIZE 48# define XTHREAD_STACKSIZE EIO_STACKSIZE
44#endif 49#endif
45#include "xthread.h" 50#include "xthread.h"
54#include <sys/statvfs.h> 59#include <sys/statvfs.h>
55#include <limits.h> 60#include <limits.h>
56#include <fcntl.h> 61#include <fcntl.h>
57#include <assert.h> 62#include <assert.h>
58 63
64/* intptr_t comes from unistd.h, says POSIX/UNIX/tradition */
65/* intptr_t only comes from stdint.h, says idiot openbsd coder */
66#if HAVE_STDINT_H
67# include <stdint.h>
68#endif
69
70#ifndef ECANCELED
71# define ECANCELED EDOM
72#endif
73
74static void eio_destroy (eio_req *req);
75
59#ifndef EIO_FINISH 76#ifndef EIO_FINISH
60# define EIO_FINISH(req) ((req)->finish) && !EIO_CANCELLED (req) ? (req)->finish (req) : 0 77# define EIO_FINISH(req) ((req)->finish) && !EIO_CANCELLED (req) ? (req)->finish (req) : 0
61#endif 78#endif
62 79
63#ifndef EIO_DESTROY 80#ifndef EIO_DESTROY
71#ifdef _WIN32 88#ifdef _WIN32
72 89
73 /*doh*/ 90 /*doh*/
74#else 91#else
75 92
76# include "config.h"
77# include <sys/time.h> 93# include <sys/time.h>
78# include <sys/select.h> 94# include <sys/select.h>
79# include <sys/mman.h>
80# include <unistd.h> 95# include <unistd.h>
81# include <utime.h> 96# include <utime.h>
82# include <signal.h> 97# include <signal.h>
83# include <dirent.h> 98# include <dirent.h>
99
100#if _POSIX_MEMLOCK || _POSIX_MEMLOCK_RANGE || _POSIX_MAPPED_FILES
101# include <sys/mman.h>
102#endif
84 103
85/* POSIX_SOURCE is useless on bsd's, and XOPEN_SOURCE is unreliable there, too */ 104/* POSIX_SOURCE is useless on bsd's, and XOPEN_SOURCE is unreliable there, too */
86# if __FreeBSD__ || defined __NetBSD__ || defined __OpenBSD__ 105# if __FreeBSD__ || defined __NetBSD__ || defined __OpenBSD__
87# define _DIRENT_HAVE_D_TYPE /* sigh */ 106# define _DIRENT_HAVE_D_TYPE /* sigh */
88# define D_INO(de) (de)->d_fileno 107# define D_INO(de) (de)->d_fileno
129#endif 148#endif
130#ifndef D_NAMLEN 149#ifndef D_NAMLEN
131# define D_NAMLEN(de) strlen ((de)->d_name) 150# define D_NAMLEN(de) strlen ((de)->d_name)
132#endif 151#endif
133 152
134/* number of seconds after which an idle threads exit */
135#define IDLE_TIMEOUT 10
136
137/* used for struct dirent, AIX doesn't provide it */ 153/* used for struct dirent, AIX doesn't provide it */
138#ifndef NAME_MAX 154#ifndef NAME_MAX
139# define NAME_MAX 4096 155# define NAME_MAX 4096
156#endif
157
158/* used for readlink etc. */
159#ifndef PATH_MAX
160# define PATH_MAX 4096
140#endif 161#endif
141 162
142/* buffer size for various temporary buffers */ 163/* buffer size for various temporary buffers */
143#define EIO_BUFSIZE 65536 164#define EIO_BUFSIZE 65536
144 165
150 errno = ENOMEM; \ 171 errno = ENOMEM; \
151 if (!eio_buf) \ 172 if (!eio_buf) \
152 return -1; 173 return -1;
153 174
154#define EIO_TICKS ((1000000 + 1023) >> 10) 175#define EIO_TICKS ((1000000 + 1023) >> 10)
155
156/*****************************************************************************/
157
158#if __GNUC__ >= 3
159# define expect(expr,value) __builtin_expect ((expr),(value))
160#else
161# define expect(expr,value) (expr)
162#endif
163
164#define expect_false(expr) expect ((expr) != 0, 0)
165#define expect_true(expr) expect ((expr) != 0, 1)
166
167/*****************************************************************************/
168 176
169#define ETP_PRI_MIN EIO_PRI_MIN 177#define ETP_PRI_MIN EIO_PRI_MIN
170#define ETP_PRI_MAX EIO_PRI_MAX 178#define ETP_PRI_MAX EIO_PRI_MAX
171 179
172struct etp_worker; 180struct etp_worker;
198/*****************************************************************************/ 206/*****************************************************************************/
199 207
200#define ETP_NUM_PRI (ETP_PRI_MAX - ETP_PRI_MIN + 1) 208#define ETP_NUM_PRI (ETP_PRI_MAX - ETP_PRI_MIN + 1)
201 209
202/* calculate time difference in ~1/EIO_TICKS of a second */ 210/* calculate time difference in ~1/EIO_TICKS of a second */
211ecb_inline int
203static int tvdiff (struct timeval *tv1, struct timeval *tv2) 212tvdiff (struct timeval *tv1, struct timeval *tv2)
204{ 213{
205 return (tv2->tv_sec - tv1->tv_sec ) * EIO_TICKS 214 return (tv2->tv_sec - tv1->tv_sec ) * EIO_TICKS
206 + ((tv2->tv_usec - tv1->tv_usec) >> 10); 215 + ((tv2->tv_usec - tv1->tv_usec) >> 10);
207} 216}
208 217
215static unsigned int max_poll_reqs; /* reslock */ 224static unsigned int max_poll_reqs; /* reslock */
216 225
217static volatile unsigned int nreqs; /* reqlock */ 226static volatile unsigned int nreqs; /* reqlock */
218static volatile unsigned int nready; /* reqlock */ 227static volatile unsigned int nready; /* reqlock */
219static volatile unsigned int npending; /* reqlock */ 228static volatile unsigned int npending; /* reqlock */
220static volatile unsigned int max_idle = 4; 229static volatile unsigned int max_idle = 4; /* maximum number of threads that can idle indefinitely */
230static volatile unsigned int idle_timeout = 10; /* number of seconds after which an idle threads exit */
221 231
222static mutex_t wrklock = X_MUTEX_INIT; 232static xmutex_t wrklock;
223static mutex_t reslock = X_MUTEX_INIT; 233static xmutex_t reslock;
224static mutex_t reqlock = X_MUTEX_INIT; 234static xmutex_t reqlock;
225static cond_t reqwait = X_COND_INIT; 235static xcond_t reqwait;
226 236
227#if !HAVE_PREADWRITE 237#if !HAVE_PREADWRITE
228/* 238/*
229 * make our pread/pwrite emulation safe against themselves, but not against 239 * make our pread/pwrite emulation safe against themselves, but not against
230 * normal read/write by using a mutex. slows down execution a lot, 240 * normal read/write by using a mutex. slows down execution a lot,
231 * but that's your problem, not mine. 241 * but that's your problem, not mine.
232 */ 242 */
233static mutex_t preadwritelock = X_MUTEX_INIT; 243static xmutex_t preadwritelock = X_MUTEX_INIT;
234#endif 244#endif
235 245
236typedef struct etp_worker 246typedef struct etp_worker
237{ 247{
238 /* locked by wrklock */ 248 /* locked by wrklock */
239 struct etp_worker *prev, *next; 249 struct etp_worker *prev, *next;
240 250
241 thread_t tid; 251 xthread_t tid;
242 252
243 /* locked by reslock, reqlock or wrklock */ 253 /* locked by reslock, reqlock or wrklock */
244 ETP_REQ *req; /* currently processed request */ 254 ETP_REQ *req; /* currently processed request */
245 255
246 ETP_WORKER_COMMON 256 ETP_WORKER_COMMON
251#define ETP_WORKER_LOCK(wrk) X_LOCK (wrklock) 261#define ETP_WORKER_LOCK(wrk) X_LOCK (wrklock)
252#define ETP_WORKER_UNLOCK(wrk) X_UNLOCK (wrklock) 262#define ETP_WORKER_UNLOCK(wrk) X_UNLOCK (wrklock)
253 263
254/* worker threads management */ 264/* worker threads management */
255 265
266static void ecb_cold
256static void etp_worker_clear (etp_worker *wrk) 267etp_worker_clear (etp_worker *wrk)
257{ 268{
258 ETP_WORKER_CLEAR (wrk); 269 ETP_WORKER_CLEAR (wrk);
259} 270}
260 271
272static void ecb_cold
261static void etp_worker_free (etp_worker *wrk) 273etp_worker_free (etp_worker *wrk)
262{ 274{
263 wrk->next->prev = wrk->prev; 275 wrk->next->prev = wrk->prev;
264 wrk->prev->next = wrk->next; 276 wrk->prev->next = wrk->next;
265 277
266 free (wrk); 278 free (wrk);
267} 279}
268 280
269static unsigned int etp_nreqs (void) 281static unsigned int
282etp_nreqs (void)
270{ 283{
271 int retval; 284 int retval;
272 if (WORDACCESS_UNSAFE) X_LOCK (reqlock); 285 if (WORDACCESS_UNSAFE) X_LOCK (reqlock);
273 retval = nreqs; 286 retval = nreqs;
274 if (WORDACCESS_UNSAFE) X_UNLOCK (reqlock); 287 if (WORDACCESS_UNSAFE) X_UNLOCK (reqlock);
275 return retval; 288 return retval;
276} 289}
277 290
278static unsigned int etp_nready (void) 291static unsigned int
292etp_nready (void)
279{ 293{
280 unsigned int retval; 294 unsigned int retval;
281 295
282 if (WORDACCESS_UNSAFE) X_LOCK (reqlock); 296 if (WORDACCESS_UNSAFE) X_LOCK (reqlock);
283 retval = nready; 297 retval = nready;
284 if (WORDACCESS_UNSAFE) X_UNLOCK (reqlock); 298 if (WORDACCESS_UNSAFE) X_UNLOCK (reqlock);
285 299
286 return retval; 300 return retval;
287} 301}
288 302
289static unsigned int etp_npending (void) 303static unsigned int
304etp_npending (void)
290{ 305{
291 unsigned int retval; 306 unsigned int retval;
292 307
293 if (WORDACCESS_UNSAFE) X_LOCK (reqlock); 308 if (WORDACCESS_UNSAFE) X_LOCK (reqlock);
294 retval = npending; 309 retval = npending;
295 if (WORDACCESS_UNSAFE) X_UNLOCK (reqlock); 310 if (WORDACCESS_UNSAFE) X_UNLOCK (reqlock);
296 311
297 return retval; 312 return retval;
298} 313}
299 314
300static unsigned int etp_nthreads (void) 315static unsigned int
316etp_nthreads (void)
301{ 317{
302 unsigned int retval; 318 unsigned int retval;
303 319
304 if (WORDACCESS_UNSAFE) X_LOCK (reqlock); 320 if (WORDACCESS_UNSAFE) X_LOCK (reqlock);
305 retval = started; 321 retval = started;
319} etp_reqq; 335} etp_reqq;
320 336
321static etp_reqq req_queue; 337static etp_reqq req_queue;
322static etp_reqq res_queue; 338static etp_reqq res_queue;
323 339
340static int ecb_noinline
324static int reqq_push (etp_reqq *q, ETP_REQ *req) 341reqq_push (etp_reqq *q, ETP_REQ *req)
325{ 342{
326 int pri = req->pri; 343 int pri = req->pri;
327 req->next = 0; 344 req->next = 0;
328 345
329 if (q->qe[pri]) 346 if (q->qe[pri])
335 q->qe[pri] = q->qs[pri] = req; 352 q->qe[pri] = q->qs[pri] = req;
336 353
337 return q->size++; 354 return q->size++;
338} 355}
339 356
357static ETP_REQ * ecb_noinline
340static ETP_REQ *reqq_shift (etp_reqq *q) 358reqq_shift (etp_reqq *q)
341{ 359{
342 int pri; 360 int pri;
343 361
344 if (!q->size) 362 if (!q->size)
345 return 0; 363 return 0;
360 } 378 }
361 379
362 abort (); 380 abort ();
363} 381}
364 382
383static void ecb_cold
384etp_thread_init (void)
385{
386 X_MUTEX_CREATE (wrklock);
387 X_MUTEX_CREATE (reslock);
388 X_MUTEX_CREATE (reqlock);
389 X_COND_CREATE (reqwait);
390}
391
392static void ecb_cold
365static void etp_atfork_prepare (void) 393etp_atfork_prepare (void)
366{ 394{
367 X_LOCK (wrklock); 395 X_LOCK (wrklock);
368 X_LOCK (reqlock); 396 X_LOCK (reqlock);
369 X_LOCK (reslock); 397 X_LOCK (reslock);
370#if !HAVE_PREADWRITE 398#if !HAVE_PREADWRITE
371 X_LOCK (preadwritelock); 399 X_LOCK (preadwritelock);
372#endif 400#endif
373} 401}
374 402
403static void ecb_cold
375static void etp_atfork_parent (void) 404etp_atfork_parent (void)
376{ 405{
377#if !HAVE_PREADWRITE 406#if !HAVE_PREADWRITE
378 X_UNLOCK (preadwritelock); 407 X_UNLOCK (preadwritelock);
379#endif 408#endif
380 X_UNLOCK (reslock); 409 X_UNLOCK (reslock);
381 X_UNLOCK (reqlock); 410 X_UNLOCK (reqlock);
382 X_UNLOCK (wrklock); 411 X_UNLOCK (wrklock);
383} 412}
384 413
414static void ecb_cold
385static void etp_atfork_child (void) 415etp_atfork_child (void)
386{ 416{
387 ETP_REQ *prv; 417 ETP_REQ *prv;
388 418
389 while ((prv = reqq_shift (&req_queue))) 419 while ((prv = reqq_shift (&req_queue)))
390 ETP_DESTROY (prv); 420 ETP_DESTROY (prv);
407 idle = 0; 437 idle = 0;
408 nreqs = 0; 438 nreqs = 0;
409 nready = 0; 439 nready = 0;
410 npending = 0; 440 npending = 0;
411 441
412 etp_atfork_parent (); 442 etp_thread_init ();
413} 443}
414 444
415static void 445static void ecb_cold
416etp_once_init (void) 446etp_once_init (void)
417{ 447{
448 etp_thread_init ();
418 X_THREAD_ATFORK (etp_atfork_prepare, etp_atfork_parent, etp_atfork_child); 449 X_THREAD_ATFORK (etp_atfork_prepare, etp_atfork_parent, etp_atfork_child);
419} 450}
420 451
421static int 452static int ecb_cold
422etp_init (void (*want_poll)(void), void (*done_poll)(void)) 453etp_init (void (*want_poll)(void), void (*done_poll)(void))
423{ 454{
424 static pthread_once_t doinit = PTHREAD_ONCE_INIT; 455 static pthread_once_t doinit = PTHREAD_ONCE_INIT;
425 456
426 pthread_once (&doinit, etp_once_init); 457 pthread_once (&doinit, etp_once_init);
431 return 0; 462 return 0;
432} 463}
433 464
434X_THREAD_PROC (etp_proc); 465X_THREAD_PROC (etp_proc);
435 466
467static void ecb_cold
436static void etp_start_thread (void) 468etp_start_thread (void)
437{ 469{
438 etp_worker *wrk = calloc (1, sizeof (etp_worker)); 470 etp_worker *wrk = calloc (1, sizeof (etp_worker));
439 471
440 /*TODO*/ 472 /*TODO*/
441 assert (("unable to allocate worker thread data", wrk)); 473 assert (("unable to allocate worker thread data", wrk));
454 free (wrk); 486 free (wrk);
455 487
456 X_UNLOCK (wrklock); 488 X_UNLOCK (wrklock);
457} 489}
458 490
491static void
459static void etp_maybe_start_thread (void) 492etp_maybe_start_thread (void)
460{ 493{
461 if (expect_true (etp_nthreads () >= wanted)) 494 if (ecb_expect_true (etp_nthreads () >= wanted))
462 return; 495 return;
463 496
464 /* todo: maybe use idle here, but might be less exact */ 497 /* todo: maybe use idle here, but might be less exact */
465 if (expect_true (0 <= (int)etp_nthreads () + (int)etp_npending () - (int)etp_nreqs ())) 498 if (ecb_expect_true (0 <= (int)etp_nthreads () + (int)etp_npending () - (int)etp_nreqs ()))
466 return; 499 return;
467 500
468 etp_start_thread (); 501 etp_start_thread ();
469} 502}
470 503
504static void ecb_cold
471static void etp_end_thread (void) 505etp_end_thread (void)
472{ 506{
473 eio_req *req = calloc (1, sizeof (eio_req)); 507 eio_req *req = calloc (1, sizeof (eio_req));
474 508
475 req->type = -1; 509 req->type = -1;
476 req->pri = ETP_PRI_MAX - ETP_PRI_MIN; 510 req->pri = ETP_PRI_MAX - ETP_PRI_MIN;
483 X_LOCK (wrklock); 517 X_LOCK (wrklock);
484 --started; 518 --started;
485 X_UNLOCK (wrklock); 519 X_UNLOCK (wrklock);
486} 520}
487 521
488static int etp_poll (void) 522static int
523etp_poll (void)
489{ 524{
490 unsigned int maxreqs; 525 unsigned int maxreqs;
491 unsigned int maxtime; 526 unsigned int maxtime;
492 struct timeval tv_start, tv_now; 527 struct timeval tv_start, tv_now;
493 528
523 558
524 X_LOCK (reqlock); 559 X_LOCK (reqlock);
525 --nreqs; 560 --nreqs;
526 X_UNLOCK (reqlock); 561 X_UNLOCK (reqlock);
527 562
528 if (expect_false (req->type == EIO_GROUP && req->size)) 563 if (ecb_expect_false (req->type == EIO_GROUP && req->size))
529 { 564 {
530 req->int1 = 1; /* mark request as delayed */ 565 req->int1 = 1; /* mark request as delayed */
531 continue; 566 continue;
532 } 567 }
533 else 568 else
534 { 569 {
535 int res = ETP_FINISH (req); 570 int res = ETP_FINISH (req);
536 if (expect_false (res)) 571 if (ecb_expect_false (res))
537 return res; 572 return res;
538 } 573 }
539 574
540 if (expect_false (maxreqs && !--maxreqs)) 575 if (ecb_expect_false (maxreqs && !--maxreqs))
541 break; 576 break;
542 577
543 if (maxtime) 578 if (maxtime)
544 { 579 {
545 gettimeofday (&tv_now, 0); 580 gettimeofday (&tv_now, 0);
551 586
552 errno = EAGAIN; 587 errno = EAGAIN;
553 return -1; 588 return -1;
554} 589}
555 590
591static void
556static void etp_cancel (ETP_REQ *req) 592etp_cancel (ETP_REQ *req)
557{ 593{
558 X_LOCK (wrklock); 594 req->cancelled = 1;
559 req->flags |= EIO_FLAG_CANCELLED;
560 X_UNLOCK (wrklock);
561 595
562 eio_grp_cancel (req); 596 eio_grp_cancel (req);
563} 597}
564 598
599static void
565static void etp_submit (ETP_REQ *req) 600etp_submit (ETP_REQ *req)
566{ 601{
567 req->pri -= ETP_PRI_MIN; 602 req->pri -= ETP_PRI_MIN;
568 603
569 if (expect_false (req->pri < ETP_PRI_MIN - ETP_PRI_MIN)) req->pri = ETP_PRI_MIN - ETP_PRI_MIN; 604 if (ecb_expect_false (req->pri < ETP_PRI_MIN - ETP_PRI_MIN)) req->pri = ETP_PRI_MIN - ETP_PRI_MIN;
570 if (expect_false (req->pri > ETP_PRI_MAX - ETP_PRI_MIN)) req->pri = ETP_PRI_MAX - ETP_PRI_MIN; 605 if (ecb_expect_false (req->pri > ETP_PRI_MAX - ETP_PRI_MIN)) req->pri = ETP_PRI_MAX - ETP_PRI_MIN;
571 606
572 if (expect_false (req->type == EIO_GROUP)) 607 if (ecb_expect_false (req->type == EIO_GROUP))
573 { 608 {
574 /* I hope this is worth it :/ */ 609 /* I hope this is worth it :/ */
575 X_LOCK (reqlock); 610 X_LOCK (reqlock);
576 ++nreqs; 611 ++nreqs;
577 X_UNLOCK (reqlock); 612 X_UNLOCK (reqlock);
596 631
597 etp_maybe_start_thread (); 632 etp_maybe_start_thread ();
598 } 633 }
599} 634}
600 635
636static void ecb_cold
601static void etp_set_max_poll_time (double nseconds) 637etp_set_max_poll_time (double nseconds)
602{ 638{
603 if (WORDACCESS_UNSAFE) X_LOCK (reslock); 639 if (WORDACCESS_UNSAFE) X_LOCK (reslock);
604 max_poll_time = nseconds * EIO_TICKS; 640 max_poll_time = nseconds * EIO_TICKS;
605 if (WORDACCESS_UNSAFE) X_UNLOCK (reslock); 641 if (WORDACCESS_UNSAFE) X_UNLOCK (reslock);
606} 642}
607 643
644static void ecb_cold
608static void etp_set_max_poll_reqs (unsigned int maxreqs) 645etp_set_max_poll_reqs (unsigned int maxreqs)
609{ 646{
610 if (WORDACCESS_UNSAFE) X_LOCK (reslock); 647 if (WORDACCESS_UNSAFE) X_LOCK (reslock);
611 max_poll_reqs = maxreqs; 648 max_poll_reqs = maxreqs;
612 if (WORDACCESS_UNSAFE) X_UNLOCK (reslock); 649 if (WORDACCESS_UNSAFE) X_UNLOCK (reslock);
613} 650}
614 651
652static void ecb_cold
615static void etp_set_max_idle (unsigned int nthreads) 653etp_set_max_idle (unsigned int nthreads)
616{ 654{
617 if (WORDACCESS_UNSAFE) X_LOCK (reqlock); 655 if (WORDACCESS_UNSAFE) X_LOCK (reqlock);
618 max_idle = nthreads <= 0 ? 1 : nthreads; 656 max_idle = nthreads;
619 if (WORDACCESS_UNSAFE) X_UNLOCK (reqlock); 657 if (WORDACCESS_UNSAFE) X_UNLOCK (reqlock);
620} 658}
621 659
660static void ecb_cold
661etp_set_idle_timeout (unsigned int seconds)
662{
663 if (WORDACCESS_UNSAFE) X_LOCK (reqlock);
664 idle_timeout = seconds;
665 if (WORDACCESS_UNSAFE) X_UNLOCK (reqlock);
666}
667
668static void ecb_cold
622static void etp_set_min_parallel (unsigned int nthreads) 669etp_set_min_parallel (unsigned int nthreads)
623{ 670{
624 if (wanted < nthreads) 671 if (wanted < nthreads)
625 wanted = nthreads; 672 wanted = nthreads;
626} 673}
627 674
675static void ecb_cold
628static void etp_set_max_parallel (unsigned int nthreads) 676etp_set_max_parallel (unsigned int nthreads)
629{ 677{
630 if (wanted > nthreads) 678 if (wanted > nthreads)
631 wanted = nthreads; 679 wanted = nthreads;
632 680
633 while (started > wanted) 681 while (started > wanted)
634 etp_end_thread (); 682 etp_end_thread ();
635} 683}
636 684
637/*****************************************************************************/ 685/*****************************************************************************/
638 686
687static void
639static void grp_try_feed (eio_req *grp) 688grp_try_feed (eio_req *grp)
640{ 689{
641 while (grp->size < grp->int2 && !EIO_CANCELLED (grp)) 690 while (grp->size < grp->int2 && !EIO_CANCELLED (grp))
642 { 691 {
643 grp->flags &= ~EIO_FLAG_GROUPADD; 692 grp->flags &= ~EIO_FLAG_GROUPADD;
644 693
651 break; 700 break;
652 } 701 }
653 } 702 }
654} 703}
655 704
705static int
656static int grp_dec (eio_req *grp) 706grp_dec (eio_req *grp)
657{ 707{
658 --grp->size; 708 --grp->size;
659 709
660 /* call feeder, if applicable */ 710 /* call feeder, if applicable */
661 grp_try_feed (grp); 711 grp_try_feed (grp);
665 return eio_finish (grp); 715 return eio_finish (grp);
666 else 716 else
667 return 0; 717 return 0;
668} 718}
669 719
720static void
670void eio_destroy (eio_req *req) 721eio_destroy (eio_req *req)
671{ 722{
672 if ((req)->flags & EIO_FLAG_PTR1_FREE) free (req->ptr1); 723 if ((req)->flags & EIO_FLAG_PTR1_FREE) free (req->ptr1);
673 if ((req)->flags & EIO_FLAG_PTR2_FREE) free (req->ptr2); 724 if ((req)->flags & EIO_FLAG_PTR2_FREE) free (req->ptr2);
674 725
675 EIO_DESTROY (req); 726 EIO_DESTROY (req);
676} 727}
677 728
729static int
678static int eio_finish (eio_req *req) 730eio_finish (eio_req *req)
679{ 731{
680 int res = EIO_FINISH (req); 732 int res = EIO_FINISH (req);
681 733
682 if (req->grp) 734 if (req->grp)
683 { 735 {
691 if (grp->grp_first == req) 743 if (grp->grp_first == req)
692 grp->grp_first = req->grp_next; 744 grp->grp_first = req->grp_next;
693 745
694 res2 = grp_dec (grp); 746 res2 = grp_dec (grp);
695 747
696 if (!res && res2) 748 if (!res)
697 res = res2; 749 res = res2;
698 } 750 }
699 751
700 eio_destroy (req); 752 eio_destroy (req);
701 753
702 return res; 754 return res;
703} 755}
704 756
757void
705void eio_grp_cancel (eio_req *grp) 758eio_grp_cancel (eio_req *grp)
706{ 759{
707 for (grp = grp->grp_first; grp; grp = grp->grp_next) 760 for (grp = grp->grp_first; grp; grp = grp->grp_next)
708 eio_cancel (grp); 761 eio_cancel (grp);
709} 762}
710 763
764void
711void eio_cancel (eio_req *req) 765eio_cancel (eio_req *req)
712{ 766{
713 etp_cancel (req); 767 etp_cancel (req);
714} 768}
715 769
770void
716void eio_submit (eio_req *req) 771eio_submit (eio_req *req)
717{ 772{
718 etp_submit (req); 773 etp_submit (req);
719} 774}
720 775
721unsigned int eio_nreqs (void) 776unsigned int
777eio_nreqs (void)
722{ 778{
723 return etp_nreqs (); 779 return etp_nreqs ();
724} 780}
725 781
726unsigned int eio_nready (void) 782unsigned int
783eio_nready (void)
727{ 784{
728 return etp_nready (); 785 return etp_nready ();
729} 786}
730 787
731unsigned int eio_npending (void) 788unsigned int
789eio_npending (void)
732{ 790{
733 return etp_npending (); 791 return etp_npending ();
734} 792}
735 793
736unsigned int eio_nthreads (void) 794unsigned int ecb_cold
795eio_nthreads (void)
737{ 796{
738 return etp_nthreads (); 797 return etp_nthreads ();
739} 798}
740 799
800void ecb_cold
741void eio_set_max_poll_time (double nseconds) 801eio_set_max_poll_time (double nseconds)
742{ 802{
743 etp_set_max_poll_time (nseconds); 803 etp_set_max_poll_time (nseconds);
744} 804}
745 805
806void ecb_cold
746void eio_set_max_poll_reqs (unsigned int maxreqs) 807eio_set_max_poll_reqs (unsigned int maxreqs)
747{ 808{
748 etp_set_max_poll_reqs (maxreqs); 809 etp_set_max_poll_reqs (maxreqs);
749} 810}
750 811
812void ecb_cold
751void eio_set_max_idle (unsigned int nthreads) 813eio_set_max_idle (unsigned int nthreads)
752{ 814{
753 etp_set_max_idle (nthreads); 815 etp_set_max_idle (nthreads);
754} 816}
755 817
818void ecb_cold
819eio_set_idle_timeout (unsigned int seconds)
820{
821 etp_set_idle_timeout (seconds);
822}
823
824void ecb_cold
756void eio_set_min_parallel (unsigned int nthreads) 825eio_set_min_parallel (unsigned int nthreads)
757{ 826{
758 etp_set_min_parallel (nthreads); 827 etp_set_min_parallel (nthreads);
759} 828}
760 829
830void ecb_cold
761void eio_set_max_parallel (unsigned int nthreads) 831eio_set_max_parallel (unsigned int nthreads)
762{ 832{
763 etp_set_max_parallel (nthreads); 833 etp_set_max_parallel (nthreads);
764} 834}
765 835
766int eio_poll (void) 836int eio_poll (void)
808 878
809 return res; 879 return res;
810} 880}
811#endif 881#endif
812 882
813#ifndef HAVE_FUTIMES 883#ifndef HAVE_UTIMES
814 884
815# undef utimes 885# undef utimes
816# undef futimes
817# define utimes(path,times) eio__utimes (path, times) 886# define utimes(path,times) eio__utimes (path, times)
818# define futimes(fd,times) eio__futimes (fd, times)
819 887
820static int 888static int
821eio__utimes (const char *filename, const struct timeval times[2]) 889eio__utimes (const char *filename, const struct timeval times[2])
822{ 890{
823 if (times) 891 if (times)
831 } 899 }
832 else 900 else
833 return utime (filename, 0); 901 return utime (filename, 0);
834} 902}
835 903
904#endif
905
906#ifndef HAVE_FUTIMES
907
908# undef futimes
909# define futimes(fd,times) eio__futimes (fd, times)
910
911static int
836static int eio__futimes (int fd, const struct timeval tv[2]) 912eio__futimes (int fd, const struct timeval tv[2])
837{ 913{
838 errno = ENOSYS; 914 errno = ENOSYS;
839 return -1; 915 return -1;
840} 916}
841 917
845# undef fdatasync 921# undef fdatasync
846# define fdatasync(fd) fsync (fd) 922# define fdatasync(fd) fsync (fd)
847#endif 923#endif
848 924
849/* sync_file_range always needs emulation */ 925/* sync_file_range always needs emulation */
850int 926static int
851eio__sync_file_range (int fd, off_t offset, size_t nbytes, unsigned int flags) 927eio__sync_file_range (int fd, off_t offset, size_t nbytes, unsigned int flags)
852{ 928{
853#if HAVE_SYNC_FILE_RANGE 929#if HAVE_SYNC_FILE_RANGE
854 int res; 930 int res;
855 931
901 977
902/* sendfile always needs emulation */ 978/* sendfile always needs emulation */
903static ssize_t 979static ssize_t
904eio__sendfile (int ofd, int ifd, off_t offset, size_t count, etp_worker *self) 980eio__sendfile (int ofd, int ifd, off_t offset, size_t count, etp_worker *self)
905{ 981{
982 ssize_t written = 0;
906 ssize_t res; 983 ssize_t res;
907 984
908 if (!count) 985 if (!count)
909 return 0; 986 return 0;
910 987
988 for (;;)
989 {
911#if HAVE_SENDFILE 990#if HAVE_SENDFILE
912# if __linux 991# if __linux
992 off_t soffset = offset;
913 res = sendfile (ofd, ifd, &offset, count); 993 res = sendfile (ofd, ifd, &soffset, count);
914 994
915# elif __FreeBSD__ 995# elif __FreeBSD__
916 /* 996 /*
917 * Of course, the freebsd sendfile is a dire hack with no thoughts 997 * Of course, the freebsd sendfile is a dire hack with no thoughts
918 * wasted on making it similar to other I/O functions. 998 * wasted on making it similar to other I/O functions.
919 */ 999 */
920 {
921 off_t sbytes; 1000 off_t sbytes;
922 res = sendfile (ifd, ofd, offset, count, 0, &sbytes, 0); 1001 res = sendfile (ifd, ofd, offset, count, 0, &sbytes, 0);
923 1002
1003 #if 0 /* according to the manpage, this is correct, but broken behaviour */
924 /* freebsd' sendfile will return 0 on success */ 1004 /* freebsd' sendfile will return 0 on success */
925 /* freebsd 8 documents it as only setting *sbytes on EINTR and EAGAIN, but */ 1005 /* freebsd 8 documents it as only setting *sbytes on EINTR and EAGAIN, but */
926 /* not on e.g. EIO or EPIPE - sounds broken */ 1006 /* not on e.g. EIO or EPIPE - sounds broken */
927 if ((res < 0 && (errno == EAGAIN || errno == EINTR) && sbytes) || res == 0) 1007 if ((res < 0 && (errno == EAGAIN || errno == EINTR) && sbytes) || res == 0)
928 res = sbytes; 1008 res = sbytes;
929 } 1009 #endif
930 1010
931# elif defined (__APPLE__) 1011 /* according to source inspection, this is correct, and useful behaviour */
1012 if (sbytes)
1013 res = sbytes;
932 1014
933 { 1015# elif defined (__APPLE__) && 0 /* broken, as everything on os x */
934 off_t sbytes = count; 1016 off_t sbytes = count;
935 res = sendfile (ifd, ofd, offset, &sbytes, 0, 0); 1017 res = sendfile (ifd, ofd, offset, &sbytes, 0, 0);
936 1018
937 /* according to the manpage, sbytes is always valid */ 1019 /* according to the manpage, sbytes is always valid */
938 if (sbytes) 1020 if (sbytes)
939 res = sbytes; 1021 res = sbytes;
940 }
941 1022
942# elif __hpux 1023# elif __hpux
943 res = sendfile (ofd, ifd, offset, count, 0, 0); 1024 res = sendfile (ofd, ifd, offset, count, 0, 0);
944 1025
945# elif __solaris 1026# elif __solaris
946 {
947 struct sendfilevec vec; 1027 struct sendfilevec vec;
948 size_t sbytes; 1028 size_t sbytes;
949 1029
950 vec.sfv_fd = ifd; 1030 vec.sfv_fd = ifd;
951 vec.sfv_flag = 0; 1031 vec.sfv_flag = 0;
952 vec.sfv_off = offset; 1032 vec.sfv_off = offset;
953 vec.sfv_len = count; 1033 vec.sfv_len = count;
954 1034
955 res = sendfilev (ofd, &vec, 1, &sbytes); 1035 res = sendfilev (ofd, &vec, 1, &sbytes);
956 1036
957 if (res < 0 && sbytes) 1037 if (res < 0 && sbytes)
958 res = sbytes; 1038 res = sbytes;
959 }
960 1039
961# endif 1040# endif
962 1041
963#elif defined (_WIN32) 1042#elif defined (_WIN32)
964
965 /* does not work, just for documentation of what would need to be done */ 1043 /* does not work, just for documentation of what would need to be done */
966 { 1044 /* actually, cannot be done like this, as TransmitFile changes the file offset, */
1045 /* libeio guarantees that the file offset does not change, and windows */
1046 /* has no way to get an independent handle to the same file description */
967 HANDLE h = TO_SOCKET (ifd); 1047 HANDLE h = TO_SOCKET (ifd);
968 SetFilePointer (h, offset, 0, FILE_BEGIN); 1048 SetFilePointer (h, offset, 0, FILE_BEGIN);
969 res = TransmitFile (TO_SOCKET (ofd), h, count, 0, 0, 0, 0); 1049 res = TransmitFile (TO_SOCKET (ofd), h, count, 0, 0, 0, 0);
970 }
971 1050
972#else 1051#else
973 res = -1; 1052 res = -1;
974 errno = ENOSYS; 1053 errno = ENOSYS;
975#endif 1054#endif
976 1055
1056 /* we assume sendfile can copy at least 128mb in one go */
1057 if (res <= 128 * 1024 * 1024)
1058 {
1059 if (res > 0)
1060 written += res;
1061
1062 if (written)
1063 return written;
1064
1065 break;
1066 }
1067 else
1068 {
1069 /* if we requested more, then probably the kernel was lazy */
1070 written += res;
1071 offset += res;
1072 count -= res;
1073
1074 if (!count)
1075 return written;
1076 }
1077 }
1078
977 if (res < 0 1079 if (res < 0
978 && (errno == ENOSYS || errno == EINVAL || errno == ENOTSOCK 1080 && (errno == ENOSYS || errno == EINVAL || errno == ENOTSOCK
979 /* BSDs */ 1081 /* BSDs */
980#ifdef ENOTSUP /* sigh, if the steenking pile called openbsd would only try to at least compile posix code... */ 1082#ifdef ENOTSUP /* sigh, if the steenking pile called openbsd would only try to at least compile posix code... */
981 || errno == ENOTSUP 1083 || errno == ENOTSUP
982#endif 1084#endif
1019 } 1121 }
1020 1122
1021 return res; 1123 return res;
1022} 1124}
1023 1125
1126#ifdef PAGESIZE
1127# define eio_pagesize() PAGESIZE
1128#else
1129static intptr_t
1130eio_pagesize (void)
1131{
1132 static intptr_t page;
1133
1134 if (!page)
1135 page = sysconf (_SC_PAGESIZE);
1136
1137 return page;
1138}
1139#endif
1140
1141static void
1142eio_page_align (void **addr, size_t *length)
1143{
1144 intptr_t mask = eio_pagesize () - 1;
1145
1146 /* round down addr */
1147 intptr_t adj = mask & (intptr_t)*addr;
1148
1149 *addr = (void *)((intptr_t)*addr - adj);
1150 *length += adj;
1151
1152 /* round up length */
1153 *length = (*length + mask) & ~mask;
1154}
1155
1156#if !_POSIX_MEMLOCK
1157# define eio__mlockall(a) ((errno = ENOSYS), -1)
1158#else
1159
1160static int
1161eio__mlockall (int flags)
1162{
1163 #if __GLIBC__ == 2 && __GLIBC_MINOR__ <= 7
1164 extern int mallopt (int, int);
1165 mallopt (-6, 238); /* http://bugs.debian.org/cgi-bin/bugreport.cgi?bug=473812 */
1166 #endif
1167
1168 if (EIO_MCL_CURRENT != MCL_CURRENT
1169 || EIO_MCL_FUTURE != MCL_FUTURE)
1170 {
1171 flags = 0
1172 | (flags & EIO_MCL_CURRENT ? MCL_CURRENT : 0)
1173 | (flags & EIO_MCL_FUTURE ? MCL_FUTURE : 0);
1174 }
1175
1176 return mlockall (flags);
1177}
1178#endif
1179
1180#if !_POSIX_MEMLOCK_RANGE
1181# define eio__mlock(a,b) ((errno = ENOSYS), -1)
1182#else
1183
1184static int
1185eio__mlock (void *addr, size_t length)
1186{
1187 eio_page_align (&addr, &length);
1188
1189 return mlock (addr, length);
1190}
1191
1192#endif
1193
1194#if !(_POSIX_MAPPED_FILES && _POSIX_SYNCHRONIZED_IO)
1195# define eio__msync(a,b,c) ((errno = ENOSYS), -1)
1196#else
1197
1198static int
1199eio__msync (void *mem, size_t len, int flags)
1200{
1201 eio_page_align (&mem, &len);
1202
1203 if (EIO_MS_ASYNC != MS_SYNC
1204 || EIO_MS_INVALIDATE != MS_INVALIDATE
1205 || EIO_MS_SYNC != MS_SYNC)
1206 {
1207 flags = 0
1208 | (flags & EIO_MS_ASYNC ? MS_ASYNC : 0)
1209 | (flags & EIO_MS_INVALIDATE ? MS_INVALIDATE : 0)
1210 | (flags & EIO_MS_SYNC ? MS_SYNC : 0);
1211 }
1212
1213 return msync (mem, len, flags);
1214}
1215
1216#endif
1217
1218static int
1219eio__mtouch (eio_req *req)
1220{
1221 void *mem = req->ptr2;
1222 size_t len = req->size;
1223 int flags = req->int1;
1224
1225 eio_page_align (&mem, &len);
1226
1227 {
1228 intptr_t addr = (intptr_t)mem;
1229 intptr_t end = addr + len;
1230 intptr_t page = eio_pagesize ();
1231
1232 if (addr < end)
1233 if (flags & EIO_MT_MODIFY) /* modify */
1234 do { *((volatile sig_atomic_t *)addr) |= 0; } while ((addr += page) < len && !EIO_CANCELLED (req));
1235 else
1236 do { *((volatile sig_atomic_t *)addr) ; } while ((addr += page) < len && !EIO_CANCELLED (req));
1237 }
1238
1239 return 0;
1240}
1241
1242/*****************************************************************************/
1243/* requests implemented outside eio_execute, because they are so large */
1244
1245static void
1246eio__realpath (eio_req *req, etp_worker *self)
1247{
1248 char *rel = req->ptr1;
1249 char *res;
1250 char *tmp1, *tmp2;
1251#if SYMLOOP_MAX > 32
1252 int symlinks = SYMLOOP_MAX;
1253#else
1254 int symlinks = 32;
1255#endif
1256
1257 req->result = -1;
1258
1259 errno = EINVAL;
1260 if (!rel)
1261 return;
1262
1263 errno = ENOENT;
1264 if (!*rel)
1265 return;
1266
1267 if (!req->ptr2)
1268 {
1269 X_LOCK (wrklock);
1270 req->flags |= EIO_FLAG_PTR2_FREE;
1271 X_UNLOCK (wrklock);
1272 req->ptr2 = malloc (PATH_MAX * 3);
1273
1274 errno = ENOMEM;
1275 if (!req->ptr2)
1276 return;
1277 }
1278
1279 res = req->ptr2;
1280 tmp1 = res + PATH_MAX;
1281 tmp2 = tmp1 + PATH_MAX;
1282
1283#if 0 /* disabled, the musl way to do things is just too racy */
1284#if __linux && defined(O_NONBLOCK) && defined(O_NOATIME)
1285 /* on linux we may be able to ask the kernel */
1286 {
1287 int fd = open (rel, O_RDONLY | O_NONBLOCK | O_NOCTTY | O_NOATIME);
1288
1289 if (fd >= 0)
1290 {
1291 sprintf (tmp1, "/proc/self/fd/%d", fd);
1292 req->result = readlink (tmp1, res, PATH_MAX);
1293 close (fd);
1294
1295 /* here we should probably stat the open file and the disk file, to make sure they still match */
1296
1297 if (req->result > 0)
1298 goto done;
1299 }
1300 else if (errno == ELOOP || errno == ENAMETOOLONG || errno == ENOENT || errno == ENOTDIR || errno == EIO)
1301 return;
1302 }
1303#endif
1304#endif
1305
1306 if (*rel != '/')
1307 {
1308 if (!getcwd (res, PATH_MAX))
1309 return;
1310
1311 if (res [1]) /* only use if not / */
1312 res += strlen (res);
1313 }
1314
1315 while (*rel)
1316 {
1317 ssize_t len, linklen;
1318 char *beg = rel;
1319
1320 while (*rel && *rel != '/')
1321 ++rel;
1322
1323 len = rel - beg;
1324
1325 if (!len) /* skip slashes */
1326 {
1327 ++rel;
1328 continue;
1329 }
1330
1331 if (beg [0] == '.')
1332 {
1333 if (len == 1)
1334 continue; /* . - nop */
1335
1336 if (beg [1] == '.' && len == 2)
1337 {
1338 /* .. - back up one component, if possible */
1339
1340 while (res != req->ptr2)
1341 if (*--res == '/')
1342 break;
1343
1344 continue;
1345 }
1346 }
1347
1348 errno = ENAMETOOLONG;
1349 if (res + 1 + len + 1 >= tmp1)
1350 return;
1351
1352 /* copy one component */
1353 *res = '/';
1354 memcpy (res + 1, beg, len);
1355
1356 /* zero-terminate, for readlink */
1357 res [len + 1] = 0;
1358
1359 /* now check if it's a symlink */
1360 linklen = readlink (req->ptr2, tmp1, PATH_MAX);
1361
1362 if (linklen < 0)
1363 {
1364 if (errno != EINVAL)
1365 return;
1366
1367 /* it's a normal directory. hopefully */
1368 res += len + 1;
1369 }
1370 else
1371 {
1372 /* yay, it was a symlink - build new path in tmp2 */
1373 int rellen = strlen (rel);
1374
1375 errno = ENAMETOOLONG;
1376 if (linklen + 1 + rellen >= PATH_MAX)
1377 return;
1378
1379 errno = ELOOP;
1380 if (!--symlinks)
1381 return;
1382
1383 if (*tmp1 == '/')
1384 res = req->ptr2; /* symlink resolves to an absolute path */
1385
1386 /* we need to be careful, as rel might point into tmp2 already */
1387 memmove (tmp2 + linklen + 1, rel, rellen + 1);
1388 tmp2 [linklen] = '/';
1389 memcpy (tmp2, tmp1, linklen);
1390
1391 rel = tmp2;
1392 }
1393 }
1394
1395 /* special case for the lone root path */
1396 if (res == req->ptr2)
1397 *res++ = '/';
1398
1399 req->result = res - (char *)req->ptr2;
1400
1401done:
1402 req->ptr2 = realloc (req->ptr2, req->result); /* trade time for space savings */
1403}
1404
1024static signed char 1405static signed char
1025eio_dent_cmp (const eio_dirent *a, const eio_dirent *b) 1406eio_dent_cmp (const eio_dirent *a, const eio_dirent *b)
1026{ 1407{
1027 return a->score - b->score ? a->score - b->score /* works because our signed char is always 0..100 */ 1408 return a->score - b->score ? a->score - b->score /* works because our signed char is always 0..100 */
1028 : a->inode < b->inode ? -1 : a->inode > b->inode ? 1 : 0; 1409 : a->inode < b->inode ? -1
1410 : a->inode > b->inode ? 1
1411 : 0;
1029} 1412}
1030 1413
1031#define EIO_DENT_CMP(i,op,j) eio_dent_cmp (&i, &j) op 0 1414#define EIO_DENT_CMP(i,op,j) eio_dent_cmp (&i, &j) op 0
1032 1415
1033#define EIO_SORT_CUTOFF 30 /* quite high, but performs well on many filesystems */ 1416#define EIO_SORT_CUTOFF 30 /* quite high, but performs well on many filesystems */
1039 unsigned char bits [9 + sizeof (ino_t) * 8]; 1422 unsigned char bits [9 + sizeof (ino_t) * 8];
1040 unsigned char *bit = bits; 1423 unsigned char *bit = bits;
1041 1424
1042 assert (CHAR_BIT == 8); 1425 assert (CHAR_BIT == 8);
1043 assert (sizeof (eio_dirent) * 8 < 256); 1426 assert (sizeof (eio_dirent) * 8 < 256);
1044 assert (offsetof (eio_dirent, inode)); /* we use 0 as sentinel */ 1427 assert (offsetof (eio_dirent, inode)); /* we use bit #0 as sentinel */
1045 assert (offsetof (eio_dirent, score)); /* we use 0 as sentinel */ 1428 assert (offsetof (eio_dirent, score)); /* we use bit #0 as sentinel */
1046 1429
1047 if (size <= EIO_SORT_FAST) 1430 if (size <= EIO_SORT_FAST)
1048 return; 1431 return;
1049 1432
1050 /* first prepare an array of bits to test in our radix sort */ 1433 /* first prepare an array of bits to test in our radix sort */
1205 flags &= ~(EIO_READDIR_DIRS_FIRST | EIO_READDIR_STAT_ORDER); 1588 flags &= ~(EIO_READDIR_DIRS_FIRST | EIO_READDIR_STAT_ORDER);
1206 1589
1207 X_LOCK (wrklock); 1590 X_LOCK (wrklock);
1208 /* the corresponding closedir is in ETP_WORKER_CLEAR */ 1591 /* the corresponding closedir is in ETP_WORKER_CLEAR */
1209 self->dirp = dirp = opendir (req->ptr1); 1592 self->dirp = dirp = opendir (req->ptr1);
1593
1594 if (req->flags & EIO_FLAG_PTR1_FREE)
1595 free (req->ptr1);
1596
1210 req->flags |= EIO_FLAG_PTR1_FREE | EIO_FLAG_PTR2_FREE; 1597 req->flags |= EIO_FLAG_PTR1_FREE | EIO_FLAG_PTR2_FREE;
1211 req->ptr1 = dents = flags ? malloc (dentalloc * sizeof (eio_dirent)) : 0; 1598 req->ptr1 = dents = flags ? malloc (dentalloc * sizeof (eio_dirent)) : 0;
1212 req->ptr2 = names = malloc (namesalloc); 1599 req->ptr2 = names = malloc (namesalloc);
1213 X_UNLOCK (wrklock); 1600 X_UNLOCK (wrklock);
1214 1601
1226 /* sort etc. */ 1613 /* sort etc. */
1227 req->int1 = flags; 1614 req->int1 = flags;
1228 req->result = dentoffs; 1615 req->result = dentoffs;
1229 1616
1230 if (flags & EIO_READDIR_STAT_ORDER) 1617 if (flags & EIO_READDIR_STAT_ORDER)
1231 eio_dent_sort (dents, dentoffs, 0, inode_bits); /* sort by inode exclusively */ 1618 eio_dent_sort (dents, dentoffs, flags & EIO_READDIR_DIRS_FIRST ? 7 : 0, inode_bits);
1232 else if (flags & EIO_READDIR_DIRS_FIRST) 1619 else if (flags & EIO_READDIR_DIRS_FIRST)
1233 if (flags & EIO_READDIR_FOUND_UNKNOWN) 1620 if (flags & EIO_READDIR_FOUND_UNKNOWN)
1234 eio_dent_sort (dents, dentoffs, 7, inode_bits); /* sort by score and inode */ 1621 eio_dent_sort (dents, dentoffs, 7, inode_bits); /* sort by score and inode */
1235 else 1622 else
1236 { 1623 {
1238 eio_dirent *oth = dents + dentoffs; 1625 eio_dirent *oth = dents + dentoffs;
1239 eio_dirent *dir = dents; 1626 eio_dirent *dir = dents;
1240 1627
1241 /* now partition dirs to the front, and non-dirs to the back */ 1628 /* now partition dirs to the front, and non-dirs to the back */
1242 /* by walking from both sides and swapping if necessary */ 1629 /* by walking from both sides and swapping if necessary */
1243 /* also clear score, so it doesn't influence sorting */
1244 while (oth > dir) 1630 while (oth > dir)
1245 { 1631 {
1246 if (dir->type == EIO_DT_DIR) 1632 if (dir->type == EIO_DT_DIR)
1247 ++dir; 1633 ++dir;
1248 else if ((--oth)->type == EIO_DT_DIR) 1634 else if ((--oth)->type == EIO_DT_DIR)
1251 1637
1252 ++dir; 1638 ++dir;
1253 } 1639 }
1254 } 1640 }
1255 1641
1256 /* now sort the dirs only */ 1642 /* now sort the dirs only (dirs all have the same score) */
1257 eio_dent_sort (dents, dir - dents, 0, inode_bits); 1643 eio_dent_sort (dents, dir - dents, 0, inode_bits);
1258 } 1644 }
1259 1645
1260 break; 1646 break;
1261 } 1647 }
1266 /* skip . and .. entries */ 1652 /* skip . and .. entries */
1267 if (name [0] != '.' || (name [1] && (name [1] != '.' || name [2]))) 1653 if (name [0] != '.' || (name [1] && (name [1] != '.' || name [2])))
1268 { 1654 {
1269 int len = D_NAMLEN (entp) + 1; 1655 int len = D_NAMLEN (entp) + 1;
1270 1656
1271 while (expect_false (namesoffs + len > namesalloc)) 1657 while (ecb_expect_false (namesoffs + len > namesalloc))
1272 { 1658 {
1273 namesalloc *= 2; 1659 namesalloc *= 2;
1274 X_LOCK (wrklock); 1660 X_LOCK (wrklock);
1275 req->ptr2 = names = realloc (names, namesalloc); 1661 req->ptr2 = names = realloc (names, namesalloc);
1276 X_UNLOCK (wrklock); 1662 X_UNLOCK (wrklock);
1283 1669
1284 if (dents) 1670 if (dents)
1285 { 1671 {
1286 struct eio_dirent *ent; 1672 struct eio_dirent *ent;
1287 1673
1288 if (expect_false (dentoffs == dentalloc)) 1674 if (ecb_expect_false (dentoffs == dentalloc))
1289 { 1675 {
1290 dentalloc *= 2; 1676 dentalloc *= 2;
1291 X_LOCK (wrklock); 1677 X_LOCK (wrklock);
1292 req->ptr1 = dents = realloc (dents, dentalloc * sizeof (eio_dirent)); 1678 req->ptr1 = dents = realloc (dents, dentalloc * sizeof (eio_dirent));
1293 X_UNLOCK (wrklock); 1679 X_UNLOCK (wrklock);
1381 break; 1767 break;
1382 } 1768 }
1383 } 1769 }
1384} 1770}
1385 1771
1386#if !(_POSIX_MAPPED_FILES && _POSIX_SYNCHRONIZED_IO)
1387# define eio__msync(a,b,c) ((errno = ENOSYS), -1)
1388#else
1389
1390int
1391eio__msync (void *mem, size_t len, int flags)
1392{
1393 if (EIO_MS_ASYNC != MS_SYNC
1394 || EIO_MS_INVALIDATE != MS_INVALIDATE
1395 || EIO_MS_SYNC != MS_SYNC)
1396 {
1397 flags = 0
1398 | (flags & EIO_MS_ASYNC ? MS_ASYNC : 0)
1399 | (flags & EIO_MS_INVALIDATE ? MS_INVALIDATE : 0)
1400 | (flags & EIO_MS_SYNC ? MS_SYNC : 0);
1401 }
1402
1403 return msync (mem, len, flags);
1404}
1405
1406#endif
1407
1408int
1409eio__mtouch (void *mem, size_t len, int flags)
1410{
1411 intptr_t addr = (intptr_t)mem;
1412 intptr_t end = addr + len;
1413#ifdef PAGESIZE
1414 const intptr_t page = PAGESIZE;
1415#else
1416 static intptr_t page;
1417
1418 if (!page)
1419 page = sysconf (_SC_PAGESIZE);
1420#endif
1421
1422 /* round down to start of page, although this is probably useless */
1423 addr &= ~(page - 1); /* assume page size is always a power of two */
1424
1425 if (addr < end)
1426 if (flags & EIO_MT_MODIFY) /* modify */
1427 do { *((volatile sig_atomic_t *)addr) |= 0; } while ((addr += page) < len);
1428 else
1429 do { *((volatile sig_atomic_t *)addr) ; } while ((addr += page) < len);
1430
1431 return 0;
1432}
1433
1434/*****************************************************************************/ 1772/*****************************************************************************/
1435 1773
1436#define ALLOC(len) \ 1774#define ALLOC(len) \
1437 if (!req->ptr2) \ 1775 if (!req->ptr2) \
1438 { \ 1776 { \
1468 if (req) 1806 if (req)
1469 break; 1807 break;
1470 1808
1471 ++idle; 1809 ++idle;
1472 1810
1473 ts.tv_sec = time (0) + IDLE_TIMEOUT; 1811 ts.tv_sec = time (0) + idle_timeout;
1474 if (X_COND_TIMEDWAIT (reqwait, reqlock, ts) == ETIMEDOUT) 1812 if (X_COND_TIMEDWAIT (reqwait, reqlock, ts) == ETIMEDOUT)
1475 { 1813 {
1476 if (idle > max_idle) 1814 if (idle > max_idle)
1477 { 1815 {
1478 --idle; 1816 --idle;
1495 X_UNLOCK (reqlock); 1833 X_UNLOCK (reqlock);
1496 1834
1497 if (req->type < 0) 1835 if (req->type < 0)
1498 goto quit; 1836 goto quit;
1499 1837
1500 if (!EIO_CANCELLED (req))
1501 ETP_EXECUTE (self, req); 1838 ETP_EXECUTE (self, req);
1502 1839
1503 X_LOCK (reslock); 1840 X_LOCK (reslock);
1504 1841
1505 ++npending; 1842 ++npending;
1506 1843
1521 return 0; 1858 return 0;
1522} 1859}
1523 1860
1524/*****************************************************************************/ 1861/*****************************************************************************/
1525 1862
1863int ecb_cold
1526int eio_init (void (*want_poll)(void), void (*done_poll)(void)) 1864eio_init (void (*want_poll)(void), void (*done_poll)(void))
1527{ 1865{
1528 return etp_init (want_poll, done_poll); 1866 return etp_init (want_poll, done_poll);
1529} 1867}
1530 1868
1869ecb_inline void
1531static void eio_api_destroy (eio_req *req) 1870eio_api_destroy (eio_req *req)
1532{ 1871{
1533 free (req); 1872 free (req);
1534} 1873}
1535 1874
1536#define REQ(rtype) \ 1875#define REQ(rtype) \
1555 { \ 1894 { \
1556 eio_api_destroy (req); \ 1895 eio_api_destroy (req); \
1557 return 0; \ 1896 return 0; \
1558 } 1897 }
1559 1898
1899static void
1560static void eio_execute (etp_worker *self, eio_req *req) 1900eio_execute (etp_worker *self, eio_req *req)
1561{ 1901{
1562 errno = 0; 1902 if (ecb_expect_false (EIO_CANCELLED (req)))
1903 {
1904 req->result = -1;
1905 req->errorno = ECANCELED;
1906 return;
1907 }
1563 1908
1564 switch (req->type) 1909 switch (req->type)
1565 { 1910 {
1566 case EIO_READ: ALLOC (req->size); 1911 case EIO_READ: ALLOC (req->size);
1567 req->result = req->offs >= 0 1912 req->result = req->offs >= 0
1600 case EIO_RMDIR: req->result = rmdir (req->ptr1); break; 1945 case EIO_RMDIR: req->result = rmdir (req->ptr1); break;
1601 case EIO_MKDIR: req->result = mkdir (req->ptr1, (mode_t)req->int2); break; 1946 case EIO_MKDIR: req->result = mkdir (req->ptr1, (mode_t)req->int2); break;
1602 case EIO_RENAME: req->result = rename (req->ptr1, req->ptr2); break; 1947 case EIO_RENAME: req->result = rename (req->ptr1, req->ptr2); break;
1603 case EIO_LINK: req->result = link (req->ptr1, req->ptr2); break; 1948 case EIO_LINK: req->result = link (req->ptr1, req->ptr2); break;
1604 case EIO_SYMLINK: req->result = symlink (req->ptr1, req->ptr2); break; 1949 case EIO_SYMLINK: req->result = symlink (req->ptr1, req->ptr2); break;
1605 case EIO_MKNOD: req->result = mknod (req->ptr1, (mode_t)req->int2, (dev_t)req->int3); break; 1950 case EIO_MKNOD: req->result = mknod (req->ptr1, (mode_t)req->int2, (dev_t)req->offs); break;
1606 1951
1952 case EIO_REALPATH: eio__realpath (req, self); break;
1953
1607 case EIO_READLINK: ALLOC (NAME_MAX); 1954 case EIO_READLINK: ALLOC (PATH_MAX);
1608 req->result = readlink (req->ptr1, req->ptr2, NAME_MAX); break; 1955 req->result = readlink (req->ptr1, req->ptr2, PATH_MAX); break;
1609 1956
1610 case EIO_SYNC: req->result = 0; sync (); break; 1957 case EIO_SYNC: req->result = 0; sync (); break;
1611 case EIO_FSYNC: req->result = fsync (req->int1); break; 1958 case EIO_FSYNC: req->result = fsync (req->int1); break;
1612 case EIO_FDATASYNC: req->result = fdatasync (req->int1); break; 1959 case EIO_FDATASYNC: req->result = fdatasync (req->int1); break;
1613 case EIO_MSYNC: req->result = eio__msync (req->ptr2, req->size, req->int1); break; 1960 case EIO_MSYNC: req->result = eio__msync (req->ptr2, req->size, req->int1); break;
1961 case EIO_MTOUCH: req->result = eio__mtouch (req); break;
1614 case EIO_MTOUCH: req->result = eio__mtouch (req->ptr2, req->size, req->int1); break; 1962 case EIO_MLOCK: req->result = eio__mlock (req->ptr2, req->size); break;
1963 case EIO_MLOCKALL: req->result = eio__mlockall (req->int1); break;
1615 case EIO_SYNC_FILE_RANGE: req->result = eio__sync_file_range (req->int1, req->offs, req->size, req->int2); break; 1964 case EIO_SYNC_FILE_RANGE: req->result = eio__sync_file_range (req->int1, req->offs, req->size, req->int2); break;
1616 1965
1617 case EIO_READDIR: eio__scandir (req, self); break; 1966 case EIO_READDIR: eio__scandir (req, self); break;
1618 1967
1619 case EIO_BUSY: 1968 case EIO_BUSY:
1620#ifdef _WIN32 1969#ifdef _WIN32
1621 Sleep (req->nv1 * 1000.); 1970 Sleep (req->nv1 * 1e3);
1622#else 1971#else
1623 { 1972 {
1624 struct timeval tv; 1973 struct timeval tv;
1625 1974
1626 tv.tv_sec = req->nv1; 1975 tv.tv_sec = req->nv1;
1627 tv.tv_usec = (req->nv1 - tv.tv_sec) * 1000000.; 1976 tv.tv_usec = (req->nv1 - tv.tv_sec) * 1e6;
1628 1977
1629 req->result = select (0, 0, 0, 0, &tv); 1978 req->result = select (0, 0, 0, 0, &tv);
1630 } 1979 }
1631#endif 1980#endif
1632 break; 1981 break;
1647 times = tv; 1996 times = tv;
1648 } 1997 }
1649 else 1998 else
1650 times = 0; 1999 times = 0;
1651 2000
1652
1653 req->result = req->type == EIO_FUTIME 2001 req->result = req->type == EIO_FUTIME
1654 ? futimes (req->int1, times) 2002 ? futimes (req->int1, times)
1655 : utimes (req->ptr1, times); 2003 : utimes (req->ptr1, times);
1656 } 2004 }
1657 break; 2005 break;
1662 case EIO_NOP: 2010 case EIO_NOP:
1663 req->result = 0; 2011 req->result = 0;
1664 break; 2012 break;
1665 2013
1666 case EIO_CUSTOM: 2014 case EIO_CUSTOM:
1667 ((void (*)(eio_req *))req->feed) (req); 2015 req->feed (req);
1668 break; 2016 break;
1669 2017
1670 default: 2018 default:
2019 errno = ENOSYS;
1671 req->result = -1; 2020 req->result = -1;
1672 break; 2021 break;
1673 } 2022 }
1674 2023
1675 req->errorno = errno; 2024 req->errorno = errno;
1705eio_req *eio_mtouch (void *addr, size_t length, int flags, int pri, eio_cb cb, void *data) 2054eio_req *eio_mtouch (void *addr, size_t length, int flags, int pri, eio_cb cb, void *data)
1706{ 2055{
1707 REQ (EIO_MTOUCH); req->ptr2 = addr; req->size = length; req->int1 = flags; SEND; 2056 REQ (EIO_MTOUCH); req->ptr2 = addr; req->size = length; req->int1 = flags; SEND;
1708} 2057}
1709 2058
2059eio_req *eio_mlock (void *addr, size_t length, int pri, eio_cb cb, void *data)
2060{
2061 REQ (EIO_MLOCK); req->ptr2 = addr; req->size = length; SEND;
2062}
2063
2064eio_req *eio_mlockall (int flags, int pri, eio_cb cb, void *data)
2065{
2066 REQ (EIO_MLOCKALL); req->int1 = flags; SEND;
2067}
2068
1710eio_req *eio_sync_file_range (int fd, off_t offset, size_t nbytes, unsigned int flags, int pri, eio_cb cb, void *data) 2069eio_req *eio_sync_file_range (int fd, off_t offset, size_t nbytes, unsigned int flags, int pri, eio_cb cb, void *data)
1711{ 2070{
1712 REQ (EIO_SYNC_FILE_RANGE); req->int1 = fd; req->offs = offset; req->size = nbytes; req->int2 = flags; SEND; 2071 REQ (EIO_SYNC_FILE_RANGE); req->int1 = fd; req->offs = offset; req->size = nbytes; req->int2 = flags; SEND;
1713} 2072}
1714 2073
1816eio_req *eio_readlink (const char *path, int pri, eio_cb cb, void *data) 2175eio_req *eio_readlink (const char *path, int pri, eio_cb cb, void *data)
1817{ 2176{
1818 return eio__1path (EIO_READLINK, path, pri, cb, data); 2177 return eio__1path (EIO_READLINK, path, pri, cb, data);
1819} 2178}
1820 2179
2180eio_req *eio_realpath (const char *path, int pri, eio_cb cb, void *data)
2181{
2182 return eio__1path (EIO_REALPATH, path, pri, cb, data);
2183}
2184
1821eio_req *eio_stat (const char *path, int pri, eio_cb cb, void *data) 2185eio_req *eio_stat (const char *path, int pri, eio_cb cb, void *data)
1822{ 2186{
1823 return eio__1path (EIO_STAT, path, pri, cb, data); 2187 return eio__1path (EIO_STAT, path, pri, cb, data);
1824} 2188}
1825 2189
1848 REQ (EIO_READDIR); PATH; req->int1 = flags; SEND; 2212 REQ (EIO_READDIR); PATH; req->int1 = flags; SEND;
1849} 2213}
1850 2214
1851eio_req *eio_mknod (const char *path, mode_t mode, dev_t dev, int pri, eio_cb cb, void *data) 2215eio_req *eio_mknod (const char *path, mode_t mode, dev_t dev, int pri, eio_cb cb, void *data)
1852{ 2216{
1853 REQ (EIO_MKNOD); PATH; req->int2 = (long)mode; req->int3 = (long)dev; SEND; 2217 REQ (EIO_MKNOD); PATH; req->int2 = (long)mode; req->offs = (off_t)dev; SEND;
1854} 2218}
1855 2219
1856static eio_req * 2220static eio_req *
1857eio__2path (int type, const char *path, const char *new_path, int pri, eio_cb cb, void *data) 2221eio__2path (int type, const char *path, const char *new_path, int pri, eio_cb cb, void *data)
1858{ 2222{
1882eio_req *eio_rename (const char *path, const char *new_path, int pri, eio_cb cb, void *data) 2246eio_req *eio_rename (const char *path, const char *new_path, int pri, eio_cb cb, void *data)
1883{ 2247{
1884 return eio__2path (EIO_RENAME, path, new_path, pri, cb, data); 2248 return eio__2path (EIO_RENAME, path, new_path, pri, cb, data);
1885} 2249}
1886 2250
1887eio_req *eio_custom (eio_cb execute, int pri, eio_cb cb, void *data) 2251eio_req *eio_custom (void (*execute)(eio_req *), int pri, eio_cb cb, void *data)
1888{ 2252{
1889 REQ (EIO_CUSTOM); req->feed = (void (*)(eio_req *))execute; SEND; 2253 REQ (EIO_CUSTOM); req->feed = execute; SEND;
1890} 2254}
1891 2255
1892#endif 2256#endif
1893 2257
1894eio_req *eio_grp (eio_cb cb, void *data) 2258eio_req *eio_grp (eio_cb cb, void *data)
1903#undef SEND 2267#undef SEND
1904 2268
1905/*****************************************************************************/ 2269/*****************************************************************************/
1906/* grp functions */ 2270/* grp functions */
1907 2271
2272void
1908void eio_grp_feed (eio_req *grp, void (*feed)(eio_req *req), int limit) 2273eio_grp_feed (eio_req *grp, void (*feed)(eio_req *req), int limit)
1909{ 2274{
1910 grp->int2 = limit; 2275 grp->int2 = limit;
1911 grp->feed = feed; 2276 grp->feed = feed;
1912 2277
1913 grp_try_feed (grp); 2278 grp_try_feed (grp);
1914} 2279}
1915 2280
2281void
1916void eio_grp_limit (eio_req *grp, int limit) 2282eio_grp_limit (eio_req *grp, int limit)
1917{ 2283{
1918 grp->int2 = limit; 2284 grp->int2 = limit;
1919 2285
1920 grp_try_feed (grp); 2286 grp_try_feed (grp);
1921} 2287}
1922 2288
2289void
1923void eio_grp_add (eio_req *grp, eio_req *req) 2290eio_grp_add (eio_req *grp, eio_req *req)
1924{ 2291{
1925 assert (("cannot add requests to IO::AIO::GRP after the group finished", grp->int1 != 2)); 2292 assert (("cannot add requests to IO::AIO::GRP after the group finished", grp->int1 != 2));
1926 2293
1927 grp->flags |= EIO_FLAG_GROUPADD; 2294 grp->flags |= EIO_FLAG_GROUPADD;
1928 2295
1939} 2306}
1940 2307
1941/*****************************************************************************/ 2308/*****************************************************************************/
1942/* misc garbage */ 2309/* misc garbage */
1943 2310
2311ssize_t
1944ssize_t eio_sendfile_sync (int ofd, int ifd, off_t offset, size_t count) 2312eio_sendfile_sync (int ofd, int ifd, off_t offset, size_t count)
1945{ 2313{
1946 etp_worker wrk; 2314 etp_worker wrk;
1947 ssize_t ret; 2315 ssize_t ret;
1948 2316
1949 wrk.dbuf = 0; 2317 wrk.dbuf = 0;

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines