[ViewVC] Diff of: cvs/libeio/eio.c

Comparing libeio/eio.c (file contents):
Revision 1.53 by root, Sat Jan 9 10:03:09 2010 UTC vs.
Revision 1.84 by root, Thu Jul 7 22:39:56 2011 UTC

 /*
  * libeio implementation
  *
- * Copyright (c) 2007,2008,2009,2010 Marc Alexander Lehmann <libeio@schmorp.de>
+ * Copyright (c) 2007,2008,2009,2010,2011 Marc Alexander Lehmann <libeio@schmorp.de>
  * All rights reserved.
  *
  * Redistribution and use in source and binary forms, with or without modifica-
  * tion, are permitted provided that the following conditions are met:
  *
  * and other provisions required by the GPL. If you do not delete the
  * provisions above, a recipient may use your version of this file under
  * either the BSD or the GPL.
  */
+#ifndef _WIN32
+# include "config.h"
+#endif
 #include "eio.h"
+#include "ecb.h"
 #ifdef EIO_STACKSIZE
 # define XTHREAD_STACKSIZE EIO_STACKSIZE
 #endif
 #include "xthread.h"
 #include <sys/statvfs.h>
 #include <limits.h>
 #include <fcntl.h>
 #include <assert.h>
+/* intptr_t comes from unistd.h, says POSIX/UNIX/tradition */
+/* intptr_t only comes from stdint.h, says idiot openbsd coder */
+#if HAVE_STDINT_H
+# include <stdint.h>
+#endif
+#ifndef ECANCELED
+# define ECANCELED EDOM
+#endif
+static void eio_destroy (eio_req *req);
 #ifndef EIO_FINISH
 # define EIO_FINISH(req)  ((req)->finish) && !EIO_CANCELLED (req) ? (req)->finish (req) : 0
 #endif
 #ifndef EIO_DESTROY
 #ifdef _WIN32
   /*doh*/
 #else
-# include "config.h"
 # include <sys/time.h>
 # include <sys/select.h>
-# include <sys/mman.h>
 # include <unistd.h>
 # include <utime.h>
 # include <signal.h>
 # include <dirent.h>
+#if _POSIX_MEMLOCK || _POSIX_MEMLOCK_RANGE || _POSIX_MAPPED_FILES
+# include <sys/mman.h>
+#endif
 /* POSIX_SOURCE is useless on bsd's, and XOPEN_SOURCE is unreliable there, too */
 # if __FreeBSD__ || defined __NetBSD__ || defined __OpenBSD__
 #  define _DIRENT_HAVE_D_TYPE /* sigh */
 #  define D_INO(de) (de)->d_fileno
 #endif
 #ifndef D_NAMLEN
 # define D_NAMLEN(de) strlen ((de)->d_name)
 #endif
-/* number of seconds after which an idle threads exit */
-#define IDLE_TIMEOUT 10
 /* used for struct dirent, AIX doesn't provide it */
 #ifndef NAME_MAX
 # define NAME_MAX 4096
+#endif
+/* used for readlink etc. */
+#ifndef PATH_MAX
+# define PATH_MAX 4096
 #endif
 /* buffer size for various temporary buffers */
 #define EIO_BUFSIZE 65536
   errno = ENOMEM;				\
   if (!eio_buf)					\
     return -1;
 #define EIO_TICKS ((1000000 + 1023) >> 10)
-/*****************************************************************************/
-#if __GNUC__ >= 3
-# define expect(expr,value) __builtin_expect ((expr),(value))
-#else
-# define expect(expr,value) (expr)
-#endif
-#define expect_false(expr) expect ((expr) != 0, 0)
-#define expect_true(expr)  expect ((expr) != 0, 1)
-/*****************************************************************************/
 #define ETP_PRI_MIN EIO_PRI_MIN
 #define ETP_PRI_MAX EIO_PRI_MAX
 struct etp_worker;
 /*****************************************************************************/
 #define ETP_NUM_PRI (ETP_PRI_MAX - ETP_PRI_MIN + 1)
 /* calculate time difference in ~1/EIO_TICKS of a second */
+ecb_inline int
-static int tvdiff (struct timeval *tv1, struct timeval *tv2)
+tvdiff (struct timeval *tv1, struct timeval *tv2)
 {
   return  (tv2->tv_sec  - tv1->tv_sec ) * EIO_TICKS
        + ((tv2->tv_usec - tv1->tv_usec) >> 10);
 }
 static unsigned int max_poll_reqs;     /* reslock */
 static volatile unsigned int nreqs;    /* reqlock */
 static volatile unsigned int nready;   /* reqlock */
 static volatile unsigned int npending; /* reqlock */
-static volatile unsigned int max_idle = 4;
+static volatile unsigned int max_idle = 4;      /* maximum number of threads that can idle indefinitely */
+static volatile unsigned int idle_timeout = 10; /* number of seconds after which an idle threads exit */
-static mutex_t wrklock = X_MUTEX_INIT;
+static xmutex_t wrklock;
-static mutex_t reslock = X_MUTEX_INIT;
+static xmutex_t reslock;
-static mutex_t reqlock = X_MUTEX_INIT;
+static xmutex_t reqlock;
-static cond_t  reqwait = X_COND_INIT;
+static xcond_t  reqwait;
 #if !HAVE_PREADWRITE
 /*
  * make our pread/pwrite emulation safe against themselves, but not against
  * normal read/write by using a mutex. slows down execution a lot,
  * but that's your problem, not mine.
  */
-static mutex_t preadwritelock = X_MUTEX_INIT;
+static xmutex_t preadwritelock = X_MUTEX_INIT;
 #endif
 typedef struct etp_worker
 {
   /* locked by wrklock */
   struct etp_worker *prev, *next;
-  thread_t tid;
+  xthread_t tid;
   /* locked by reslock, reqlock or wrklock */
   ETP_REQ *req; /* currently processed request */
   ETP_WORKER_COMMON
 #define ETP_WORKER_LOCK(wrk)   X_LOCK   (wrklock)
 #define ETP_WORKER_UNLOCK(wrk) X_UNLOCK (wrklock)
 /* worker threads management */
+static void ecb_cold
-static void etp_worker_clear (etp_worker *wrk)
+etp_worker_clear (etp_worker *wrk)
 {
   ETP_WORKER_CLEAR (wrk);
 }
+static void ecb_cold
-static void etp_worker_free (etp_worker *wrk)
+etp_worker_free (etp_worker *wrk)
 {
   wrk->next->prev = wrk->prev;
   wrk->prev->next = wrk->next;
   free (wrk);
 }
-static unsigned int etp_nreqs (void)
+static unsigned int
+etp_nreqs (void)
 {
   int retval;
   if (WORDACCESS_UNSAFE) X_LOCK   (reqlock);
   retval = nreqs;
   if (WORDACCESS_UNSAFE) X_UNLOCK (reqlock);
   return retval;
 }
-static unsigned int etp_nready (void)
+static unsigned int
+etp_nready (void)
 {
   unsigned int retval;
   if (WORDACCESS_UNSAFE) X_LOCK   (reqlock);
   retval = nready;
   if (WORDACCESS_UNSAFE) X_UNLOCK (reqlock);
   return retval;
 }
-static unsigned int etp_npending (void)
+static unsigned int
+etp_npending (void)
 {
   unsigned int retval;
   if (WORDACCESS_UNSAFE) X_LOCK   (reqlock);
   retval = npending;
   if (WORDACCESS_UNSAFE) X_UNLOCK (reqlock);
   return retval;
 }
-static unsigned int etp_nthreads (void)
+static unsigned int
+etp_nthreads (void)
 {
   unsigned int retval;
   if (WORDACCESS_UNSAFE) X_LOCK   (reqlock);
   retval = started;
 } etp_reqq;
 static etp_reqq req_queue;
 static etp_reqq res_queue;
+static int ecb_noinline
-static int reqq_push (etp_reqq *q, ETP_REQ *req)
+reqq_push (etp_reqq *q, ETP_REQ *req)
 {
   int pri = req->pri;
   req->next = 0;
   if (q->qe[pri])
     q->qe[pri] = q->qs[pri] = req;
   return q->size++;
 }
+static ETP_REQ * ecb_noinline
-static ETP_REQ *reqq_shift (etp_reqq *q)
+reqq_shift (etp_reqq *q)
 {
   int pri;
   if (!q->size)
     return 0;
     }
   abort ();
 }
+static void ecb_cold
+etp_thread_init (void)
+{
+  X_MUTEX_CREATE (wrklock);
+  X_MUTEX_CREATE (reslock);
+  X_MUTEX_CREATE (reqlock);
+  X_COND_CREATE  (reqwait);
+}
+static void ecb_cold
-static void etp_atfork_prepare (void)
+etp_atfork_prepare (void)
 {
   X_LOCK (wrklock);
   X_LOCK (reqlock);
   X_LOCK (reslock);
 #if !HAVE_PREADWRITE
   X_LOCK (preadwritelock);
 #endif
 }
+static void ecb_cold
-static void etp_atfork_parent (void)
+etp_atfork_parent (void)
 {
 #if !HAVE_PREADWRITE
   X_UNLOCK (preadwritelock);
 #endif
   X_UNLOCK (reslock);
   X_UNLOCK (reqlock);
   X_UNLOCK (wrklock);
 }
+static void ecb_cold
-static void etp_atfork_child (void)
+etp_atfork_child (void)
 {
   ETP_REQ *prv;
   while ((prv = reqq_shift (&req_queue)))
     ETP_DESTROY (prv);
   idle     = 0;
   nreqs    = 0;
   nready   = 0;
   npending = 0;
-  etp_atfork_parent ();
+  etp_thread_init ();
 }
-static void
+static void ecb_cold
 etp_once_init (void)
 {
+  etp_thread_init ();
   X_THREAD_ATFORK (etp_atfork_prepare, etp_atfork_parent, etp_atfork_child);
 }
-static int
+static int ecb_cold
 etp_init (void (*want_poll)(void), void (*done_poll)(void))
 {
   static pthread_once_t doinit = PTHREAD_ONCE_INIT;
   pthread_once (&doinit, etp_once_init);
   return 0;
 }
 X_THREAD_PROC (etp_proc);
+static void ecb_cold
-static void etp_start_thread (void)
+etp_start_thread (void)
 {
   etp_worker *wrk = calloc (1, sizeof (etp_worker));
   /*TODO*/
   assert (("unable to allocate worker thread data", wrk));
     free (wrk);
   X_UNLOCK (wrklock);
 }
+static void
-static void etp_maybe_start_thread (void)
+etp_maybe_start_thread (void)
 {
-  if (expect_true (etp_nthreads () >= wanted))
+  if (ecb_expect_true (etp_nthreads () >= wanted))
     return;
   /* todo: maybe use idle here, but might be less exact */
-  if (expect_true (0 <= (int)etp_nthreads () + (int)etp_npending () - (int)etp_nreqs ()))
+  if (ecb_expect_true (0 <= (int)etp_nthreads () + (int)etp_npending () - (int)etp_nreqs ()))
     return;
   etp_start_thread ();
 }
+static void ecb_cold
-static void etp_end_thread (void)
+etp_end_thread (void)
 {
   eio_req *req = calloc (1, sizeof (eio_req));
   req->type = -1;
   req->pri  = ETP_PRI_MAX - ETP_PRI_MIN;
   X_LOCK (wrklock);
   --started;
   X_UNLOCK (wrklock);
 }
-static int etp_poll (void)
+static int
+etp_poll (void)
 {
   unsigned int maxreqs;
   unsigned int maxtime;
   struct timeval tv_start, tv_now;
       X_LOCK (reqlock);
       --nreqs;
       X_UNLOCK (reqlock);
-      if (expect_false (req->type == EIO_GROUP && req->size))
+      if (ecb_expect_false (req->type == EIO_GROUP && req->size))
         {
           req->int1 = 1; /* mark request as delayed */
           continue;
         }
       else
         {
           int res = ETP_FINISH (req);
-          if (expect_false (res))
+          if (ecb_expect_false (res))
             return res;
         }
-      if (expect_false (maxreqs && !--maxreqs))
+      if (ecb_expect_false (maxreqs && !--maxreqs))
         break;
       if (maxtime)
         {
           gettimeofday (&tv_now, 0);
   errno = EAGAIN;
   return -1;
 }
+static void
-static void etp_cancel (ETP_REQ *req)
+etp_cancel (ETP_REQ *req)
 {
-  X_LOCK   (wrklock);
+  req->cancelled = 1;
-  req->flags |= EIO_FLAG_CANCELLED;
-  X_UNLOCK (wrklock);
   eio_grp_cancel (req);
 }
+static void
-static void etp_submit (ETP_REQ *req)
+etp_submit (ETP_REQ *req)
 {
   req->pri -= ETP_PRI_MIN;
-  if (expect_false (req->pri < ETP_PRI_MIN - ETP_PRI_MIN)) req->pri = ETP_PRI_MIN - ETP_PRI_MIN;
+  if (ecb_expect_false (req->pri < ETP_PRI_MIN - ETP_PRI_MIN)) req->pri = ETP_PRI_MIN - ETP_PRI_MIN;
-  if (expect_false (req->pri > ETP_PRI_MAX - ETP_PRI_MIN)) req->pri = ETP_PRI_MAX - ETP_PRI_MIN;
+  if (ecb_expect_false (req->pri > ETP_PRI_MAX - ETP_PRI_MIN)) req->pri = ETP_PRI_MAX - ETP_PRI_MIN;
-  if (expect_false (req->type == EIO_GROUP))
+  if (ecb_expect_false (req->type == EIO_GROUP))
     {
       /* I hope this is worth it :/ */
       X_LOCK (reqlock);
       ++nreqs;
       X_UNLOCK (reqlock);
       etp_maybe_start_thread ();
     }
 }
+static void ecb_cold
-static void etp_set_max_poll_time (double nseconds)
+etp_set_max_poll_time (double nseconds)
 {
   if (WORDACCESS_UNSAFE) X_LOCK   (reslock);
   max_poll_time = nseconds * EIO_TICKS;
   if (WORDACCESS_UNSAFE) X_UNLOCK (reslock);
 }
+static void ecb_cold
-static void etp_set_max_poll_reqs (unsigned int maxreqs)
+etp_set_max_poll_reqs (unsigned int maxreqs)
 {
   if (WORDACCESS_UNSAFE) X_LOCK   (reslock);
   max_poll_reqs = maxreqs;
   if (WORDACCESS_UNSAFE) X_UNLOCK (reslock);
 }
+static void ecb_cold
-static void etp_set_max_idle (unsigned int nthreads)
+etp_set_max_idle (unsigned int nthreads)
 {
   if (WORDACCESS_UNSAFE) X_LOCK   (reqlock);
-  max_idle = nthreads <= 0 ? 1 : nthreads;
+  max_idle = nthreads;
   if (WORDACCESS_UNSAFE) X_UNLOCK (reqlock);
 }
+static void ecb_cold
+etp_set_idle_timeout (unsigned int seconds)
+{
+  if (WORDACCESS_UNSAFE) X_LOCK   (reqlock);
+  idle_timeout = seconds;
+  if (WORDACCESS_UNSAFE) X_UNLOCK (reqlock);
+}
+static void ecb_cold
-static void etp_set_min_parallel (unsigned int nthreads)
+etp_set_min_parallel (unsigned int nthreads)
 {
   if (wanted < nthreads)
     wanted = nthreads;
 }
+static void ecb_cold
-static void etp_set_max_parallel (unsigned int nthreads)
+etp_set_max_parallel (unsigned int nthreads)
 {
   if (wanted > nthreads)
     wanted = nthreads;
   while (started > wanted)
     etp_end_thread ();
 }
 /*****************************************************************************/
+static void
-static void grp_try_feed (eio_req *grp)
+grp_try_feed (eio_req *grp)
 {
   while (grp->size < grp->int2 && !EIO_CANCELLED (grp))
     {
       grp->flags &= ~EIO_FLAG_GROUPADD;
           break;
         }
     }
 }
+static int
-static int grp_dec (eio_req *grp)
+grp_dec (eio_req *grp)
 {
   --grp->size;
   /* call feeder, if applicable */
   grp_try_feed (grp);
     return eio_finish (grp);
   else
     return 0;
 }
+static void
-void eio_destroy (eio_req *req)
+eio_destroy (eio_req *req)
 {
   if ((req)->flags & EIO_FLAG_PTR1_FREE) free (req->ptr1);
   if ((req)->flags & EIO_FLAG_PTR2_FREE) free (req->ptr2);
   EIO_DESTROY (req);
 }
+static int
-static int eio_finish (eio_req *req)
+eio_finish (eio_req *req)
 {
   int res = EIO_FINISH (req);
   if (req->grp)
     {
       if (grp->grp_first == req)
         grp->grp_first = req->grp_next;
       res2 = grp_dec (grp);
-      if (!res && res2)
+      if (!res)
         res = res2;
     }
   eio_destroy (req);
   return res;
 }
+void
-void eio_grp_cancel (eio_req *grp)
+eio_grp_cancel (eio_req *grp)
 {
   for (grp = grp->grp_first; grp; grp = grp->grp_next)
     eio_cancel (grp);
 }
+void
-void eio_cancel (eio_req *req)
+eio_cancel (eio_req *req)
 {
   etp_cancel (req);
 }
+void
-void eio_submit (eio_req *req)
+eio_submit (eio_req *req)
 {
   etp_submit (req);
 }
-unsigned int eio_nreqs (void)
+unsigned int
+eio_nreqs (void)
 {
   return etp_nreqs ();
 }
-unsigned int eio_nready (void)
+unsigned int
+eio_nready (void)
 {
   return etp_nready ();
 }
-unsigned int eio_npending (void)
+unsigned int
+eio_npending (void)
 {
   return etp_npending ();
 }
-unsigned int eio_nthreads (void)
+unsigned int ecb_cold
+eio_nthreads (void)
 {
   return etp_nthreads ();
 }
+void ecb_cold
-void eio_set_max_poll_time (double nseconds)
+eio_set_max_poll_time (double nseconds)
 {
   etp_set_max_poll_time (nseconds);
 }
+void ecb_cold
-void eio_set_max_poll_reqs (unsigned int maxreqs)
+eio_set_max_poll_reqs (unsigned int maxreqs)
 {
   etp_set_max_poll_reqs (maxreqs);
 }
+void ecb_cold
-void eio_set_max_idle (unsigned int nthreads)
+eio_set_max_idle (unsigned int nthreads)
 {
   etp_set_max_idle (nthreads);
 }
+void ecb_cold
+eio_set_idle_timeout (unsigned int seconds)
+{
+  etp_set_idle_timeout (seconds);
+}
+void ecb_cold
-void eio_set_min_parallel (unsigned int nthreads)
+eio_set_min_parallel (unsigned int nthreads)
 {
   etp_set_min_parallel (nthreads);
 }
+void ecb_cold
-void eio_set_max_parallel (unsigned int nthreads)
+eio_set_max_parallel (unsigned int nthreads)
 {
   etp_set_max_parallel (nthreads);
 }
 int eio_poll (void)
   return res;
 }
 #endif
-#ifndef HAVE_FUTIMES
+#ifndef HAVE_UTIMES
 # undef utimes
-# undef futimes
-# define utimes(path,times)  eio__utimes  (path, times)
+# define utimes(path,times)  eio__utimes (path, times)
-# define futimes(fd,times)   eio__futimes (fd, times)
 static int
 eio__utimes (const char *filename, const struct timeval times[2])
 {
   if (times)
     }
   else
     return utime (filename, 0);
 }
+#endif
+#ifndef HAVE_FUTIMES
+# undef futimes
+# define futimes(fd,times) eio__futimes (fd, times)
+static int
-static int eio__futimes (int fd, const struct timeval tv[2])
+eio__futimes (int fd, const struct timeval tv[2])
 {
   errno = ENOSYS;
   return -1;
 }
 # undef fdatasync
 # define fdatasync(fd) fsync (fd)
 #endif
 /* sync_file_range always needs emulation */
-int
+static int
 eio__sync_file_range (int fd, off_t offset, size_t nbytes, unsigned int flags)
 {
 #if HAVE_SYNC_FILE_RANGE
   int res;
   /* even though we could play tricks with the flags, it's better to always
    * call fdatasync, as that matches the expectation of its users best */
   return fdatasync (fd);
 }
+static int
+eio__fallocate (int fd, int mode, off_t offset, size_t len)
+{
+#if HAVE_FALLOCATE
+  return fallocate (fd, mode, offset, len);
+#else
+  errno = ENOSYS;
+  return -1;
+#endif
+}
 #if !HAVE_READAHEAD
 # undef readahead
 # define readahead(fd,offset,count) eio__readahead (fd, offset, count, self)
 static ssize_t
 /* sendfile always needs emulation */
 static ssize_t
 eio__sendfile (int ofd, int ifd, off_t offset, size_t count, etp_worker *self)
 {
+  ssize_t written = 0;
   ssize_t res;
   if (!count)
     return 0;
+  for (;;)
+    {
+#ifdef __APPLE__
+# undef HAVE_SENDFILE /* broken, as everything on os x */
+#endif
 #if HAVE_SENDFILE
 # if __linux
+      off_t soffset = offset;
-  res = sendfile (ofd, ifd, &offset, count);
+      res = sendfile (ofd, ifd, &soffset, count);
 # elif __FreeBSD__
-  /*
+      /*
-   * Of course, the freebsd sendfile is a dire hack with no thoughts
+       * Of course, the freebsd sendfile is a dire hack with no thoughts
-   * wasted on making it similar to other I/O functions.
+       * wasted on making it similar to other I/O functions.
-   */
+       */
-  {
-    off_t sbytes;
+      off_t sbytes;
-    res = sendfile (ifd, ofd, offset, count, 0, &sbytes, 0);
+      res = sendfile (ifd, ofd, offset, count, 0, &sbytes, 0);
-    #if 0 /* according to the manpage, this is correct, but broken behaviour */
+      #if 0 /* according to the manpage, this is correct, but broken behaviour */
-    /* freebsd' sendfile will return 0 on success */
+      /* freebsd' sendfile will return 0 on success */
-    /* freebsd 8 documents it as only setting *sbytes on EINTR and EAGAIN, but */
+      /* freebsd 8 documents it as only setting *sbytes on EINTR and EAGAIN, but */
-    /* not on e.g. EIO or EPIPE - sounds broken */
+      /* not on e.g. EIO or EPIPE - sounds broken */
-    if ((res < 0 && (errno == EAGAIN || errno == EINTR) && sbytes) || res == 0)
+      if ((res < 0 && (errno == EAGAIN || errno == EINTR) && sbytes) || res == 0)
-      res = sbytes;
+        res = sbytes;
-    #endif
+      #endif
-    /* according to source inspection, this is correct, and useful behaviour */
+      /* according to source inspection, this is correct, and useful behaviour */
-    if (sbytes)
+      if (sbytes)
-      res = sbytes;
+        res = sbytes;
-  }
 # elif defined (__APPLE__)
-  {
-    off_t sbytes = count;
+      off_t sbytes = count;
-    res = sendfile (ifd, ofd, offset, &sbytes, 0, 0);
+      res = sendfile (ifd, ofd, offset, &sbytes, 0, 0);
-    /* according to the manpage, sbytes is always valid */
+      /* according to the manpage, sbytes is always valid */
-    if (sbytes)
+      if (sbytes)
-      res = sbytes;
+        res = sbytes;
-  }
 # elif __hpux
-  res = sendfile (ofd, ifd, offset, count, 0, 0);
+      res = sendfile (ofd, ifd, offset, count, 0, 0);
 # elif __solaris
-  {
-    struct sendfilevec vec;
+      struct sendfilevec vec;
-    size_t sbytes;
+      size_t sbytes;
-    vec.sfv_fd   = ifd;
+      vec.sfv_fd   = ifd;
-    vec.sfv_flag = 0;
+      vec.sfv_flag = 0;
-    vec.sfv_off  = offset;
+      vec.sfv_off  = offset;
-    vec.sfv_len  = count;
+      vec.sfv_len  = count;
-    res = sendfilev (ofd, &vec, 1, &sbytes);
+      res = sendfilev (ofd, &vec, 1, &sbytes);
-    if (res < 0 && sbytes)
+      if (res < 0 && sbytes)
-      res = sbytes;
+        res = sbytes;
-  }
 # endif
 #elif defined (_WIN32)
-  /* does not work, just for documentation of what would need to be done */
+      /* does not work, just for documentation of what would need to be done */
-  {
+      /* actually, cannot be done like this, as TransmitFile changes the file offset, */
+      /* libeio guarantees that the file offset does not change, and windows */
+      /* has no way to get an independent handle to the same file description */
-    HANDLE h = TO_SOCKET (ifd);
+      HANDLE h = TO_SOCKET (ifd);
-    SetFilePointer (h, offset, 0, FILE_BEGIN);
+      SetFilePointer (h, offset, 0, FILE_BEGIN);
-    res = TransmitFile (TO_SOCKET (ofd), h, count, 0, 0, 0, 0);
+      res = TransmitFile (TO_SOCKET (ofd), h, count, 0, 0, 0, 0);
-  }
 #else
-  res = -1;
+      res = -1;
-  errno = ENOSYS;
+      errno = ENOSYS;
 #endif
+      /* we assume sendfile can copy at least 128mb in one go */
+      if (res <= 128 * 1024 * 1024)
+        {
+          if (res > 0)
+            written += res;
+          if (written)
+            return written;
+          break;
+        }
+      else
+        {
+          /* if we requested more, then probably the kernel was lazy */
+          written += res;
+          offset  += res;
+          count   -= res;
+          if (!count)
+            return written;
+        }
+    }
-  if (res <  0
+  if (res < 0
       && (errno == ENOSYS || errno == EINVAL || errno == ENOTSOCK
           /* BSDs */
 #ifdef ENOTSUP /* sigh, if the steenking pile called openbsd would only try to at least compile posix code... */
           || errno == ENOTSUP
 #endif
     }
   return res;
 }
+#ifdef PAGESIZE
+# define eio_pagesize() PAGESIZE
+#else
+static intptr_t
+eio_pagesize (void)
+{
+  static intptr_t page;
+  if (!page)
+    page = sysconf (_SC_PAGESIZE);
+  return page;
+}
+#endif
+static void
+eio_page_align (void **addr, size_t *length)
+{
+  intptr_t mask = eio_pagesize () - 1;
+  /* round down addr */
+  intptr_t adj = mask & (intptr_t)*addr;
+  *addr   = (void *)((intptr_t)*addr - adj);
+  *length += adj;
+  /* round up length */
+  *length = (*length + mask) & ~mask;
+}
+#if !_POSIX_MEMLOCK
+# define eio__mlockall(a) ((errno = ENOSYS), -1)
+#else
+static int
+eio__mlockall (int flags)
+{
+  #if __GLIBC__ == 2 && __GLIBC_MINOR__ <= 7
+    extern int mallopt (int, int);
+    mallopt (-6, 238); /* http://bugs.debian.org/cgi-bin/bugreport.cgi?bug=473812 */
+  #endif
+  if (EIO_MCL_CURRENT   != MCL_CURRENT
+      || EIO_MCL_FUTURE != MCL_FUTURE)
+    {
+      flags = 0
+         | (flags & EIO_MCL_CURRENT ? MCL_CURRENT : 0)
+         | (flags & EIO_MCL_FUTURE  ? MCL_FUTURE : 0);
+    }
+  return mlockall (flags);
+}
+#endif
+#if !_POSIX_MEMLOCK_RANGE
+# define eio__mlock(a,b) ((errno = ENOSYS), -1)
+#else
+static int
+eio__mlock (void *addr, size_t length)
+{
+  eio_page_align (&addr, &length);
+  return mlock (addr, length);
+}
+#endif
+#if !(_POSIX_MAPPED_FILES && _POSIX_SYNCHRONIZED_IO)
+# define eio__msync(a,b,c) ((errno = ENOSYS), -1)
+#else
+static int
+eio__msync (void *mem, size_t len, int flags)
+{
+  eio_page_align (&mem, &len);
+  if (EIO_MS_ASYNC         != MS_SYNC
+      || EIO_MS_INVALIDATE != MS_INVALIDATE
+      || EIO_MS_SYNC       != MS_SYNC)
+    {
+      flags = 0
+         | (flags & EIO_MS_ASYNC      ? MS_ASYNC : 0)
+         | (flags & EIO_MS_INVALIDATE ? MS_INVALIDATE : 0)
+         | (flags & EIO_MS_SYNC       ? MS_SYNC : 0);
+    }
+  return msync (mem, len, flags);
+}
+#endif
+static int
+eio__mtouch (eio_req *req)
+{
+  void *mem  = req->ptr2;
+  size_t len = req->size;
+  int flags  = req->int1;
+  eio_page_align (&mem, &len);
+  {
+    intptr_t addr = (intptr_t)mem;
+    intptr_t end = addr + len;
+    intptr_t page = eio_pagesize ();
+    if (addr < end)
+      if (flags & EIO_MT_MODIFY) /* modify */
+        do { *((volatile sig_atomic_t *)addr) |= 0; } while ((addr += page) < len && !EIO_CANCELLED (req));
+      else
+        do { *((volatile sig_atomic_t *)addr)     ; } while ((addr += page) < len && !EIO_CANCELLED (req));
+  }
+  return 0;
+}
+/*****************************************************************************/
+/* requests implemented outside eio_execute, because they are so large */
+static void
+eio__realpath (eio_req *req, etp_worker *self)
+{
+  char *rel = req->ptr1;
+  char *res;
+  char *tmp1, *tmp2;
+#if SYMLOOP_MAX > 32
+  int symlinks = SYMLOOP_MAX;
+#else
+  int symlinks = 32;
+#endif
+  req->result = -1;
+  errno = EINVAL;
+  if (!rel)
+    return;
+  errno = ENOENT;
+  if (!*rel)
+    return;
+  if (!req->ptr2)
+    {
+      X_LOCK (wrklock);
+      req->flags |= EIO_FLAG_PTR2_FREE;
+      X_UNLOCK (wrklock);
+      req->ptr2 = malloc (PATH_MAX * 3);
+      errno = ENOMEM;
+      if (!req->ptr2)
+        return;
+    }
+  res  = req->ptr2;
+  tmp1 = res  + PATH_MAX;
+  tmp2 = tmp1 + PATH_MAX;
+#if 0 /* disabled, the musl way to do things is just too racy */
+#if __linux && defined(O_NONBLOCK) && defined(O_NOATIME)
+  /* on linux we may be able to ask the kernel */
+  {
+    int fd = open (rel, O_RDONLY | O_NONBLOCK | O_NOCTTY | O_NOATIME);
+    if (fd >= 0)
+      {
+        sprintf (tmp1, "/proc/self/fd/%d", fd);
+        req->result = readlink (tmp1, res, PATH_MAX);
+        close (fd);
+        /* here we should probably stat the open file and the disk file, to make sure they still match */
+        if (req->result > 0)
+          goto done;
+      }
+    else if (errno == ELOOP || errno == ENAMETOOLONG || errno == ENOENT || errno == ENOTDIR || errno == EIO)
+      return;
+  }
+#endif
+#endif
+  if (*rel != '/')
+    {
+      if (!getcwd (res, PATH_MAX))
+        return;
+      if (res [1]) /* only use if not / */
+        res += strlen (res);
+    }
+  while (*rel)
+    {
+      ssize_t len, linklen;
+      char *beg = rel;
+      while (*rel && *rel != '/')
+        ++rel;
+      len = rel - beg;
+      if (!len) /* skip slashes */
+        {
+          ++rel;
+          continue;
+        }
+      if (beg [0] == '.')
+        {
+          if (len == 1)
+            continue; /* . - nop */
+          if (beg [1] == '.' && len == 2)
+            {
+              /* .. - back up one component, if possible */
+              while (res != req->ptr2)
+                if (*--res == '/')
+                  break;
+              continue;
+            }
+        }
+        errno = ENAMETOOLONG;
+        if (res + 1 + len + 1 >= tmp1)
+          return;
+        /* copy one component */
+        *res = '/';
+        memcpy (res + 1, beg, len);
+        /* zero-terminate, for readlink */
+        res [len + 1] = 0;
+        /* now check if it's a symlink */
+        linklen = readlink (req->ptr2, tmp1, PATH_MAX);
+        if (linklen < 0)
+          {
+            if (errno != EINVAL)
+              return;
+            /* it's a normal directory. hopefully */
+            res += len + 1;
+          }
+        else
+          {
+            /* yay, it was a symlink - build new path in tmp2 */
+            int rellen = strlen (rel);
+            errno = ENAMETOOLONG;
+            if (linklen + 1 + rellen >= PATH_MAX)
+              return;
+            errno = ELOOP;
+            if (!--symlinks)
+              return;
+            if (*tmp1 == '/')
+              res = req->ptr2; /* symlink resolves to an absolute path */
+            /* we need to be careful, as rel might point into tmp2 already */
+            memmove (tmp2 + linklen + 1, rel, rellen + 1);
+            tmp2 [linklen] = '/';
+            memcpy (tmp2, tmp1, linklen);
+            rel = tmp2;
+          }
+    }
+  /* special case for the lone root path */
+  if (res == req->ptr2)
+    *res++ = '/';
+  req->result = res - (char *)req->ptr2;
+done:
+  req->ptr2 = realloc (req->ptr2, req->result); /* trade time for space savings */
+}
 static signed char
 eio_dent_cmp (const eio_dirent *a, const eio_dirent *b)
 {
-    return a->score - b->score ? a->score - b->score /* works because our signed char is always 0..100 */
+  return a->score - b->score ? a->score - b->score /* works because our signed char is always 0..100 */
-              : a->inode < b->inode ? -1 : a->inode > b->inode ? 1 : 0;
+       : a->inode < b->inode ? -1
+       : a->inode > b->inode ?  1
+       :                        0;
 }
 #define EIO_DENT_CMP(i,op,j) eio_dent_cmp (&i, &j) op 0
 #define EIO_SORT_CUTOFF 30 /* quite high, but performs well on many filesystems */
   unsigned char bits [9 + sizeof (ino_t) * 8];
   unsigned char *bit = bits;
   assert (CHAR_BIT == 8);
   assert (sizeof (eio_dirent) * 8 < 256);
-  assert (offsetof (eio_dirent, inode)); /* we use 0 as sentinel */
+  assert (offsetof (eio_dirent, inode)); /* we use bit #0 as sentinel */
-  assert (offsetof (eio_dirent, score)); /* we use 0 as sentinel */
+  assert (offsetof (eio_dirent, score)); /* we use bit #0 as sentinel */
   if (size <= EIO_SORT_FAST)
     return;
   /* first prepare an array of bits to test in our radix sort */
     flags &= ~(EIO_READDIR_DIRS_FIRST | EIO_READDIR_STAT_ORDER);
   X_LOCK (wrklock);
   /* the corresponding closedir is in ETP_WORKER_CLEAR */
   self->dirp = dirp = opendir (req->ptr1);
+  if (req->flags & EIO_FLAG_PTR1_FREE)
+    free (req->ptr1);
   req->flags |= EIO_FLAG_PTR1_FREE | EIO_FLAG_PTR2_FREE;
   req->ptr1 = dents = flags ? malloc (dentalloc * sizeof (eio_dirent)) : 0;
   req->ptr2 = names = malloc (namesalloc);
   X_UNLOCK (wrklock);
             /* sort etc. */
             req->int1   = flags;
             req->result = dentoffs;
             if (flags & EIO_READDIR_STAT_ORDER)
-              eio_dent_sort (dents, dentoffs, 0, inode_bits); /* sort by inode exclusively */
+              eio_dent_sort (dents, dentoffs, flags & EIO_READDIR_DIRS_FIRST ? 7 : 0, inode_bits);
             else if (flags & EIO_READDIR_DIRS_FIRST)
               if (flags & EIO_READDIR_FOUND_UNKNOWN)
                 eio_dent_sort (dents, dentoffs, 7, inode_bits); /* sort by score and inode */
               else
                 {
                   eio_dirent *oth = dents + dentoffs;
                   eio_dirent *dir = dents;
                   /* now partition dirs to the front, and non-dirs to the back */
                   /* by walking from both sides and swapping if necessary */
-                  /* also clear score, so it doesn't influence sorting */
                   while (oth > dir)
                     {
                       if (dir->type == EIO_DT_DIR)
                         ++dir;
                       else if ((--oth)->type == EIO_DT_DIR)
                           ++dir;
                         }
                     }
-                  /* now sort the dirs only */
+                  /* now sort the dirs only (dirs all have the same score) */
                   eio_dent_sort (dents, dir - dents, 0, inode_bits);
                 }
             break;
           }
         /* skip . and .. entries */
         if (name [0] != '.' || (name [1] && (name [1] != '.' || name [2])))
           {
             int len = D_NAMLEN (entp) + 1;
-            while (expect_false (namesoffs + len > namesalloc))
+            while (ecb_expect_false (namesoffs + len > namesalloc))
               {
                 namesalloc *= 2;
                 X_LOCK (wrklock);
                 req->ptr2 = names = realloc (names, namesalloc);
                 X_UNLOCK (wrklock);
             if (dents)
               {
                 struct eio_dirent *ent;
-                if (expect_false (dentoffs == dentalloc))
+                if (ecb_expect_false (dentoffs == dentalloc))
                   {
                     dentalloc *= 2;
                     X_LOCK (wrklock);
                     req->ptr1 = dents = realloc (dents, dentalloc * sizeof (eio_dirent));
                     X_UNLOCK (wrklock);
             break;
           }
       }
 }
-#if !(_POSIX_MAPPED_FILES && _POSIX_SYNCHRONIZED_IO)
-# define eio__msync(a,b,c) ((errno = ENOSYS), -1)
-#else
-int
-eio__msync (void *mem, size_t len, int flags)
-{
-  if (EIO_MS_ASYNC         != MS_SYNC
-      || EIO_MS_INVALIDATE != MS_INVALIDATE
-      || EIO_MS_SYNC       != MS_SYNC)
-    {
-      flags = 0
-         | (flags & EIO_MS_ASYNC      ? MS_ASYNC : 0)
-         | (flags & EIO_MS_INVALIDATE ? MS_INVALIDATE : 0)
-         | (flags & EIO_MS_SYNC       ? MS_SYNC : 0);
-    }
-  return msync (mem, len, flags);
-}
-#endif
-int
-eio__mtouch (void *mem, size_t len, int flags)
-{
-  intptr_t addr = (intptr_t)mem;
-  intptr_t end = addr + len;
-#ifdef PAGESIZE
-  const intptr_t page = PAGESIZE;
-#else
-  static intptr_t page;
-  if (!page)
-    page = sysconf (_SC_PAGESIZE);
-#endif
-  /* round down to start of page, although this is probably useless */
-  addr &= ~(page - 1); /* assume page size is always a power of two */
-  if (addr < end)
-    if (flags & EIO_MT_MODIFY) /* modify */
-      do { *((volatile sig_atomic_t *)addr) |= 0; } while ((addr += page) < len);
-    else
-      do { *((volatile sig_atomic_t *)addr)     ; } while ((addr += page) < len);
-  return 0;
-}
 /*****************************************************************************/
 #define ALLOC(len)				\
   if (!req->ptr2)				\
     {						\
           if (req)
             break;
           ++idle;
-          ts.tv_sec = time (0) + IDLE_TIMEOUT;
+          ts.tv_sec = time (0) + idle_timeout;
           if (X_COND_TIMEDWAIT (reqwait, reqlock, ts) == ETIMEDOUT)
             {
               if (idle > max_idle)
                 {
                   --idle;
       X_UNLOCK (reqlock);
       if (req->type < 0)
         goto quit;
-      if (!EIO_CANCELLED (req))
-        ETP_EXECUTE (self, req);
+      ETP_EXECUTE (self, req);
       X_LOCK (reslock);
       ++npending;
   return 0;
 }
 /*****************************************************************************/
+int ecb_cold
-int eio_init (void (*want_poll)(void), void (*done_poll)(void))
+eio_init (void (*want_poll)(void), void (*done_poll)(void))
 {
   return etp_init (want_poll, done_poll);
 }
+ecb_inline void
-static void eio_api_destroy (eio_req *req)
+eio_api_destroy (eio_req *req)
 {
   free (req);
 }
 #define REQ(rtype)                                            	\
     {								\
       eio_api_destroy (req);					\
       return 0;							\
     }
+static void
-static void eio_execute (etp_worker *self, eio_req *req)
+eio_execute (etp_worker *self, eio_req *req)
 {
-  errno = 0;
+  if (ecb_expect_false (EIO_CANCELLED (req)))
+    {
+      req->result  = -1;
+      req->errorno = ECANCELED;
+      return;
+    }
   switch (req->type)
     {
       case EIO_READ:      ALLOC (req->size);
                           req->result = req->offs >= 0
       case EIO_RMDIR:     req->result = rmdir     (req->ptr1); break;
       case EIO_MKDIR:     req->result = mkdir     (req->ptr1, (mode_t)req->int2); break;
       case EIO_RENAME:    req->result = rename    (req->ptr1, req->ptr2); break;
       case EIO_LINK:      req->result = link      (req->ptr1, req->ptr2); break;
       case EIO_SYMLINK:   req->result = symlink   (req->ptr1, req->ptr2); break;
-      case EIO_MKNOD:     req->result = mknod     (req->ptr1, (mode_t)req->int2, (dev_t)req->int3); break;
+      case EIO_MKNOD:     req->result = mknod     (req->ptr1, (mode_t)req->int2, (dev_t)req->offs); break;
+      case EIO_REALPATH:  eio__realpath (req, self); break;
-      case EIO_READLINK:  ALLOC (NAME_MAX);
+      case EIO_READLINK:  ALLOC (PATH_MAX);
-                          req->result = readlink  (req->ptr1, req->ptr2, NAME_MAX); break;
+                          req->result = readlink  (req->ptr1, req->ptr2, PATH_MAX); break;
       case EIO_SYNC:      req->result = 0; sync (); break;
       case EIO_FSYNC:     req->result = fsync     (req->int1); break;
       case EIO_FDATASYNC: req->result = fdatasync (req->int1); break;
       case EIO_MSYNC:     req->result = eio__msync (req->ptr2, req->size, req->int1); break;
+      case EIO_MTOUCH:    req->result = eio__mtouch (req); break;
-      case EIO_MTOUCH:    req->result = eio__mtouch (req->ptr2, req->size, req->int1); break;
+      case EIO_MLOCK:     req->result = eio__mlock (req->ptr2, req->size); break;
+      case EIO_MLOCKALL:  req->result = eio__mlockall (req->int1); break;
       case EIO_SYNC_FILE_RANGE: req->result = eio__sync_file_range (req->int1, req->offs, req->size, req->int2); break;
+      case EIO_FALLOCATE: req->result = eio__fallocate (req->int1, req->int2, req->offs, req->size); break;
       case EIO_READDIR:   eio__scandir (req, self); break;
       case EIO_BUSY:
 #ifdef _WIN32
-	Sleep (req->nv1 * 1000.);
+	Sleep (req->nv1 * 1e3);
 #else
         {
           struct timeval tv;
           tv.tv_sec  = req->nv1;
-          tv.tv_usec = (req->nv1 - tv.tv_sec) * 1000000.;
+          tv.tv_usec = (req->nv1 - tv.tv_sec) * 1e6;
           req->result = select (0, 0, 0, 0, &tv);
         }
 #endif
         break;
               times = tv;
             }
           else
             times = 0;
           req->result = req->type == EIO_FUTIME
                         ? futimes (req->int1, times)
                         : utimes  (req->ptr1, times);
         }
         break;
       case EIO_NOP:
         req->result = 0;
         break;
       case EIO_CUSTOM:
-        ((void (*)(eio_req *))req->feed) (req);
+        req->feed (req);
         break;
       default:
+        errno = ENOSYS;
         req->result = -1;
         break;
     }
   req->errorno = errno;
 eio_req *eio_mtouch (void *addr, size_t length, int flags, int pri, eio_cb cb, void *data)
 {
   REQ (EIO_MTOUCH); req->ptr2 = addr; req->size = length; req->int1 = flags; SEND;
 }
+eio_req *eio_mlock (void *addr, size_t length, int pri, eio_cb cb, void *data)
+{
+  REQ (EIO_MLOCK); req->ptr2 = addr; req->size = length; SEND;
+}
+eio_req *eio_mlockall (int flags, int pri, eio_cb cb, void *data)
+{
+  REQ (EIO_MLOCKALL); req->int1 = flags; SEND;
+}
 eio_req *eio_sync_file_range (int fd, off_t offset, size_t nbytes, unsigned int flags, int pri, eio_cb cb, void *data)
 {
   REQ (EIO_SYNC_FILE_RANGE); req->int1 = fd; req->offs = offset; req->size = nbytes; req->int2 = flags; SEND;
+}
+eio_req *eio_fallocate (int fd, int mode, off_t offset, size_t len, int pri, eio_cb cb, void *data)
+{
+  REQ (EIO_FALLOCATE); req->int1 = fd; req->int2 = mode; req->offs = offset; req->size = len; SEND;
 }
 eio_req *eio_fdatasync (int fd, int pri, eio_cb cb, void *data)
 {
   REQ (EIO_FDATASYNC); req->int1 = fd; SEND;
 eio_req *eio_readlink (const char *path, int pri, eio_cb cb, void *data)
 {
   return eio__1path (EIO_READLINK, path, pri, cb, data);
 }
+eio_req *eio_realpath (const char *path, int pri, eio_cb cb, void *data)
+{
+  return eio__1path (EIO_REALPATH, path, pri, cb, data);
+}
 eio_req *eio_stat (const char *path, int pri, eio_cb cb, void *data)
 {
   return eio__1path (EIO_STAT, path, pri, cb, data);
 }
   REQ (EIO_READDIR); PATH; req->int1 = flags; SEND;
 }
 eio_req *eio_mknod (const char *path, mode_t mode, dev_t dev, int pri, eio_cb cb, void *data)
 {
-  REQ (EIO_MKNOD); PATH; req->int2 = (long)mode; req->int3 = (long)dev; SEND;
+  REQ (EIO_MKNOD); PATH; req->int2 = (long)mode; req->offs = (off_t)dev; SEND;
 }
 static eio_req *
 eio__2path (int type, const char *path, const char *new_path, int pri, eio_cb cb, void *data)
 {
 eio_req *eio_rename (const char *path, const char *new_path, int pri, eio_cb cb, void *data)
 {
   return eio__2path (EIO_RENAME, path, new_path, pri, cb, data);
 }
-eio_req *eio_custom (eio_cb execute, int pri, eio_cb cb, void *data)
+eio_req *eio_custom (void (*execute)(eio_req *), int pri, eio_cb cb, void *data)
 {
-  REQ (EIO_CUSTOM); req->feed = (void (*)(eio_req *))execute; SEND;
+  REQ (EIO_CUSTOM); req->feed = execute; SEND;
 }
 #endif
 eio_req *eio_grp (eio_cb cb, void *data)
 #undef SEND
 /*****************************************************************************/
 /* grp functions */
+void
-void eio_grp_feed (eio_req *grp, void (*feed)(eio_req *req), int limit)
+eio_grp_feed (eio_req *grp, void (*feed)(eio_req *req), int limit)
 {
   grp->int2 = limit;
   grp->feed = feed;
   grp_try_feed (grp);
 }
+void
-void eio_grp_limit (eio_req *grp, int limit)
+eio_grp_limit (eio_req *grp, int limit)
 {
   grp->int2 = limit;
   grp_try_feed (grp);
 }
+void
-void eio_grp_add (eio_req *grp, eio_req *req)
+eio_grp_add (eio_req *grp, eio_req *req)
 {
   assert (("cannot add requests to IO::AIO::GRP after the group finished", grp->int1 != 2));
   grp->flags |= EIO_FLAG_GROUPADD;
 }
 /*****************************************************************************/
 /* misc garbage */
+ssize_t
-ssize_t eio_sendfile_sync (int ofd, int ifd, off_t offset, size_t count)
+eio_sendfile_sync (int ofd, int ifd, off_t offset, size_t count)
 {
   etp_worker wrk;
   ssize_t ret;
   wrk.dbuf = 0;

Diff Legend

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

Comparing libeio/eio.c (file contents): Revision 1.53 by root, Sat Jan 9 10:03:09 2010 UTC vs. Revision 1.84 by root, Thu Jul 7 22:39:56 2011 UTC

Diff Legend

Comparing libeio/eio.c (file contents):
Revision 1.53 by root, Sat Jan 9 10:03:09 2010 UTC vs.
Revision 1.84 by root, Thu Jul 7 22:39:56 2011 UTC