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

Comparing libeio/eio.c (file contents):
Revision 1.106 by root, Mon Sep 26 20:19:08 2011 UTC vs.
Revision 1.115 by root, Tue Mar 27 18:54:45 2012 UTC

58#include <sys/stat.h> 58#include <sys/stat.h>
59#include <limits.h> 59#include <limits.h>
60#include <fcntl.h> 60#include <fcntl.h>
61#include <assert.h> 61#include <assert.h>
62 62
63#if _POSIX_VERSION >= 200809L
64# define HAVE_AT 1
65#else
66# define HAVE_AT 0
67#endif
68
69/* intptr_t comes from unistd.h, says POSIX/UNIX/tradition */ 63/* intptr_t comes from unistd.h, says POSIX/UNIX/tradition */
70/* intptr_t only comes from stdint.h, says idiot openbsd coder */ 64/* intptr_t only comes from stdint.h, says idiot openbsd coder */
71#if HAVE_STDINT_H 65#if HAVE_STDINT_H
72# include <stdint.h> 66# include <stdint.h>
73#endif 67#endif
318 } 312 }
319 313
320 return buf->ptr; 314 return buf->ptr;
321} 315}
322 316
317struct tmpbuf;
318
319#if _POSIX_VERSION >= 200809L
320 #define HAVE_AT 1
321 #define WD2FD(wd) ((wd) ? (wd)->fd : AT_FDCWD)
322 #ifndef O_SEARCH
323 #define O_SEARCH O_RDONLY
324 #endif
325#else
326 #define HAVE_AT 0
327 static const char *wd_expand (struct tmpbuf *tmpbuf, eio_wd wd, const char *path);
328#endif
329
330struct eio_pwd
331{
332#if HAVE_AT
333 int fd;
334#endif
335 int len;
336 char str[1]; /* actually, a 0-terminated canonical path */
337};
338
323/*****************************************************************************/ 339/*****************************************************************************/
324 340
325#define ETP_PRI_MIN EIO_PRI_MIN 341#define ETP_PRI_MIN EIO_PRI_MIN
326#define ETP_PRI_MAX EIO_PRI_MAX 342#define ETP_PRI_MAX EIO_PRI_MAX
327 343
393#define ETP_WORKER_LOCK(wrk) X_LOCK (wrklock) 409#define ETP_WORKER_LOCK(wrk) X_LOCK (wrklock)
394#define ETP_WORKER_UNLOCK(wrk) X_UNLOCK (wrklock) 410#define ETP_WORKER_UNLOCK(wrk) X_UNLOCK (wrklock)
395 411
396/* worker threads management */ 412/* worker threads management */
397 413
398static void ecb_cold 414static void
399etp_worker_clear (etp_worker *wrk) 415etp_worker_clear (etp_worker *wrk)
400{ 416{
401} 417}
402 418
403static void ecb_cold 419static void ecb_cold
1078 dBUF; 1094 dBUF;
1079 1095
1080 while (todo > 0) 1096 while (todo > 0)
1081 { 1097 {
1082 size_t len = todo < EIO_BUFSIZE ? todo : EIO_BUFSIZE; 1098 size_t len = todo < EIO_BUFSIZE ? todo : EIO_BUFSIZE;
1083
1084 pread (fd, eio_buf, len, offset);
1085 offset += len; 1099 offset += len;
1086 todo -= len; 1100 todo -= len;
1087 } 1101 }
1088 1102
1089 FUBd; 1103 FUBd;
1090 1104
1091 errno = 0; 1105 /* linux's readahead basically only fails for EBADF or EINVAL (not mmappable) */
1106 /* but not for e.g. EIO or eof, so we also never fail */
1092 return count; 1107 return 0;
1093} 1108}
1094 1109
1095#endif 1110#endif
1096 1111
1097/* sendfile always needs emulation */ 1112/* sendfile always needs emulation */
1366} 1381}
1367 1382
1368/*****************************************************************************/ 1383/*****************************************************************************/
1369/* requests implemented outside eio_execute, because they are so large */ 1384/* requests implemented outside eio_execute, because they are so large */
1370 1385
1371/* copies some absolute path to tmpbuf */
1372static char *
1373eio__getwd (struct tmpbuf *tmpbuf, eio_wd wd)
1374{
1375 if (wd == EIO_CWD)
1376 return getcwd (tmpbuf->ptr, PATH_MAX);
1377
1378#if HAVE_AT
1379 abort (); /*TODO*/
1380#else
1381 strcpy (tmpbuf->ptr, wd);
1382#endif
1383 return tmpbuf->ptr;
1384}
1385
1386/* result will always end up in tmpbuf, there is always space for adding a 0-byte */ 1386/* result will always end up in tmpbuf, there is always space for adding a 0-byte */
1387static int 1387static int
1388eio__realpath (struct tmpbuf *tmpbuf, eio_wd wd, const char *path) 1388eio__realpath (struct tmpbuf *tmpbuf, eio_wd wd, const char *path)
1389{ 1389{
1390 const char *rel = path; 1390 const char *rel = path;
1394 int symlinks = SYMLOOP_MAX; 1394 int symlinks = SYMLOOP_MAX;
1395#else 1395#else
1396 int symlinks = 32; 1396 int symlinks = 32;
1397#endif 1397#endif
1398 1398
1399 /*D*/ /*TODO: wd ignored */
1400
1401 errno = EINVAL; 1399 errno = EINVAL;
1402 if (!rel) 1400 if (!rel)
1403 return -1; 1401 return -1;
1404 1402
1405 errno = ENOENT; 1403 errno = ENOENT;
1433#endif 1431#endif
1434#endif 1432#endif
1435 1433
1436 if (*rel != '/') 1434 if (*rel != '/')
1437 { 1435 {
1438 if (!eio__getwd (tmpbuf, wd)) 1436 int len;
1437
1438 errno = ENOENT;
1439 if (wd == EIO_INVALID_WD)
1439 return -1; 1440 return -1;
1441
1442 if (wd == EIO_CWD)
1443 {
1444 if (!getcwd (res, PATH_MAX))
1445 return -1;
1446
1447 len = strlen (res);
1448 }
1449 else
1450 memcpy (res, wd->str, len = wd->len);
1440 1451
1441 if (res [1]) /* only use if not / */ 1452 if (res [1]) /* only use if not / */
1442 res += strlen (res); 1453 res += len;
1443 } 1454 }
1444 1455
1445 while (*rel) 1456 while (*rel)
1446 { 1457 {
1447 eio_ssize_t len, linklen; 1458 eio_ssize_t len, linklen;
1475 } 1486 }
1476 } 1487 }
1477 1488
1478 errno = ENAMETOOLONG; 1489 errno = ENAMETOOLONG;
1479 if (res + 1 + len + 1 >= tmp1) 1490 if (res + 1 + len + 1 >= tmp1)
1480 return; 1491 return -1;
1481 1492
1482 /* copy one component */ 1493 /* copy one component */
1483 *res = '/'; 1494 *res = '/';
1484 memcpy (res + 1, beg, len); 1495 memcpy (res + 1, beg, len);
1485 1496
1694 eio_dent_insertion_sort (dents, size); 1705 eio_dent_insertion_sort (dents, size);
1695} 1706}
1696 1707
1697/* read a full directory */ 1708/* read a full directory */
1698static void 1709static void
1699eio__scandir (eio_req *req) 1710eio__scandir (eio_req *req, etp_worker *self)
1700{ 1711{
1701 char *name, *names; 1712 char *name, *names;
1702 int namesalloc = 4096 - sizeof (void *) * 4; 1713 int namesalloc = 4096 - sizeof (void *) * 4;
1703 int namesoffs = 0; 1714 int namesoffs = 0;
1704 int flags = req->int1; 1715 int flags = req->int1;
1722#ifdef _WIN32 1733#ifdef _WIN32
1723 { 1734 {
1724 int len = strlen ((const char *)req->ptr1); 1735 int len = strlen ((const char *)req->ptr1);
1725 char *path = malloc (MAX_PATH); 1736 char *path = malloc (MAX_PATH);
1726 const char *fmt; 1737 const char *fmt;
1738 const char *reqpath = wd_expand (&self->tmpbuf, req->wd, req->ptr1);
1727 1739
1728 if (!len) 1740 if (!len)
1729 fmt = "./*"; 1741 fmt = "./*";
1730 else if (((const char *)req->ptr1)[len - 1] == '/' || ((const char *)req->ptr1)[len - 1] == '\\') 1742 else if (reqpath[len - 1] == '/' || reqpath[len - 1] == '\\')
1731 fmt = "%s*"; 1743 fmt = "%s*";
1732 else 1744 else
1733 fmt = "%s/*"; 1745 fmt = "%s/*";
1734 1746
1735 _snprintf (path, MAX_PATH, fmt, (const char *)req->ptr1); 1747 _snprintf (path, MAX_PATH, fmt, reqpath);
1736 dirp = FindFirstFile (path, &entp); 1748 dirp = FindFirstFile (path, &entp);
1737 free (path); 1749 free (path);
1738 1750
1739 if (dirp == INVALID_HANDLE_VALUE) 1751 if (dirp == INVALID_HANDLE_VALUE)
1740 { 1752 {
1762 1774
1763 return; 1775 return;
1764 } 1776 }
1765 } 1777 }
1766#else 1778#else
1779 #if HAVE_AT
1780 if (req->wd)
1781 {
1782 int fd = openat (WD2FD (req->wd), req->ptr1, O_CLOEXEC | O_SEARCH | O_DIRECTORY);
1783
1784 if (fd < 0)
1785 return;
1786
1787 dirp = fdopendir (fd);
1788
1789 if (!dirp)
1790 close (fd);
1791 }
1792 else
1767 dirp = opendir (req->ptr1); 1793 dirp = opendir (req->ptr1);
1794 #else
1795 dirp = opendir (wd_expand (&self->tmpbuf, req->wd, req->ptr1));
1796 #endif
1768 1797
1769 if (!dirp) 1798 if (!dirp)
1770 return; 1799 return;
1771#endif 1800#endif
1772 1801
1936 { 1965 {
1937 if (ent->type == EIO_DT_UNKNOWN) 1966 if (ent->type == EIO_DT_UNKNOWN)
1938 { 1967 {
1939 if (*name == '.') /* leading dots are likely directories, and, in any case, rare */ 1968 if (*name == '.') /* leading dots are likely directories, and, in any case, rare */
1940 ent->score = 1; 1969 ent->score = 1;
1941 else if (!strchr (name, '.')) /* absense of dots indicate likely dirs */ 1970 else if (!strchr (name, '.')) /* absence of dots indicate likely dirs */
1942 ent->score = len <= 2 ? 4 - len : len <= 4 ? 4 : len <= 7 ? 5 : 6; /* shorter == more likely dir, but avoid too many classes */ 1971 ent->score = len <= 2 ? 4 - len : len <= 4 ? 4 : len <= 7 ? 5 : 6; /* shorter == more likely dir, but avoid too many classes */
1943 } 1972 }
1944 else if (ent->type == EIO_DT_DIR) 1973 else if (ent->type == EIO_DT_DIR)
1945 ent->score = 0; 1974 ent->score = 0;
1946 } 1975 }
1966 } 1995 }
1967} 1996}
1968 1997
1969/*****************************************************************************/ 1998/*****************************************************************************/
1970/* working directory stuff */ 1999/* working directory stuff */
2000/* various deficiencies in the posix 2008 api force us to */
2001/* keep the absolute path in string form at all times */
2002/* fuck yeah. */
2003
2004#if !HAVE_AT
2005
2006/* a bit like realpath, but usually faster because it doesn'T have to return */
2007/* an absolute or canonical path */
2008static const char *
2009wd_expand (struct tmpbuf *tmpbuf, eio_wd wd, const char *path)
2010{
2011 if (!wd || *path == '/')
2012 return path;
2013
2014 if (path [0] == '.' && !path [1])
2015 return wd->str;
2016
2017 {
2018 int l1 = wd->len;
2019 int l2 = strlen (path);
2020
2021 char *res = tmpbuf_get (tmpbuf, l1 + l2 + 2);
2022
2023 memcpy (res, wd->str, l1);
2024 res [l1] = '/';
2025 memcpy (res + l1 + 1, path, l2 + 1);
2026
2027 return res;
2028 }
2029}
2030
2031#endif
2032
2033static eio_wd
2034eio__wd_open_sync (struct tmpbuf *tmpbuf, eio_wd wd, const char *path)
2035{
2036 int fd;
2037 eio_wd res;
2038 int len = eio__realpath (tmpbuf, wd, path);
2039
2040 if (len < 0)
2041 return EIO_INVALID_WD;
1971 2042
1972#if HAVE_AT 2043#if HAVE_AT
2044 fd = openat (WD2FD (wd), path, O_CLOEXEC | O_SEARCH | O_DIRECTORY);
1973 2045
1974#define WD2FD(wd) ((wd) ? ((int)(long)(wd)) - 1 : AT_FDCWD) 2046 if (fd < 0)
1975 2047 return EIO_INVALID_WD;
1976#ifndef O_SEARCH
1977# define O_SEARCH O_RDONLY
1978#endif 2048#endif
2049
2050 res = malloc (sizeof (*res) + len); /* one extra 0-byte */
2051
2052#if HAVE_AT
2053 res->fd = fd;
2054#endif
2055
2056 res->len = len;
2057 memcpy (res->str, tmpbuf->ptr, len);
2058 res->str [len] = 0;
2059
2060 return res;
2061}
1979 2062
1980eio_wd 2063eio_wd
1981eio_wd_open_sync (eio_wd wd, const char *path) 2064eio_wd_open_sync (eio_wd wd, const char *path)
1982{ 2065{
1983 int fd = openat (WD2FD (wd), path, O_CLOEXEC | O_SEARCH | O_DIRECTORY); 2066 struct tmpbuf tmpbuf = { 0 };
1984
1985 return fd >= 0 ? (eio_wd)(long)(fd + 1) : EIO_INVALID_WD;
1986}
1987
1988static eio_wd
1989eio__wd_open_sync (struct tmpbuf *tmpbuf, eio_wd wd, const char *path)
1990{
1991 return eio_wd_open_sync (wd, path); 2067 wd = eio__wd_open_sync (&tmpbuf, wd, path);
2068 free (tmpbuf.ptr);
2069
2070 return wd;
1992} 2071}
1993 2072
1994void 2073void
1995eio_wd_close_sync (eio_wd wd) 2074eio_wd_close_sync (eio_wd wd)
1996{ 2075{
1997 int fd = WD2FD (wd); 2076 if (wd != EIO_INVALID_WD && wd != EIO_CWD)
1998 2077 {
1999 if (fd >= 0) 2078 #if HAVE_AT
2000 close (fd); 2079 close (wd->fd);
2080 #endif
2081 free (wd);
2082 }
2001} 2083}
2084
2085#if HAVE_AT
2086
2087/* they forgot these */
2002 2088
2003static int 2089static int
2004eio__truncateat (int dirfd, const char *path, off_t length) 2090eio__truncateat (int dirfd, const char *path, off_t length)
2005{ 2091{
2006 int fd = openat (dirfd, path, O_WRONLY | O_CLOEXEC); 2092 int fd = openat (dirfd, path, O_WRONLY | O_CLOEXEC);
2025 2111
2026 res = fstatvfs (fd, buf); 2112 res = fstatvfs (fd, buf);
2027 close (fd); 2113 close (fd);
2028 return res; 2114 return res;
2029 2115
2030}
2031
2032#else
2033
2034/* on legacy systems, we represent the working directories simply by their path strings */
2035
2036static const char *
2037wd_expand (struct tmpbuf *tmpbuf, eio_wd wd, const char *path)
2038{
2039 if (!wd || *path == '/')
2040 return path;
2041
2042 {
2043 int l1 = strlen ((const char *)wd);
2044 int l2 = strlen (path);
2045
2046 char *res = tmpbuf_get (tmpbuf, l1 + l2 + 2);
2047
2048 memcpy (res, wd, l1);
2049 res [l1] = '/';
2050 memcpy (res + l1 + 1, path, l2 + 1);
2051
2052 return res;
2053 }
2054}
2055
2056static eio_wd
2057eio__wd_open_sync (struct tmpbuf *tmpbuf, eio_wd wd, const char *path)
2058{
2059 if (*path == '/') /* absolute paths ignore wd */
2060 path = strdup (path);
2061 else if (path [0] == '.' && !path [1]) /* special case '.', as it is common */
2062 return wd;
2063 else
2064 {
2065 int len = eio__realpath (tmpbuf, wd, path);
2066
2067 path = EIO_INVALID_WD;
2068
2069 if (len >= 0)
2070 {
2071 ((char *)tmpbuf->ptr)[len] = 0;
2072 path = strdup (tmpbuf->ptr);
2073 }
2074 }
2075
2076 if (!path)
2077 path = EIO_INVALID_WD;
2078
2079 return (eio_wd)path;
2080}
2081
2082eio_wd
2083eio_wd_open_sync (eio_wd wd, const char *path)
2084{
2085 struct tmpbuf tmpbuf = { 0 };
2086 wd = eio__wd_open_sync (&tmpbuf, wd, path);
2087 free (tmpbuf.ptr);
2088
2089 return wd;
2090}
2091
2092void
2093eio_wd_close_sync (eio_wd wd)
2094{
2095 if (wd != EIO_INVALID_WD)
2096 free (wd);
2097} 2116}
2098 2117
2099#endif 2118#endif
2100 2119
2101/*****************************************************************************/ 2120/*****************************************************************************/
2113 req->result = -1; \ 2132 req->result = -1; \
2114 break; \ 2133 break; \
2115 } \ 2134 } \
2116 } 2135 }
2117 2136
2137static void ecb_noinline ecb_cold
2138etp_proc_init (void)
2139{
2140#if HAVE_PRCTL_SET_NAME
2141 /* provide a more sensible "thread name" */
2142 char name[16 + 1];
2143 const int namelen = sizeof (name) - 1;
2144 int len;
2145
2146 prctl (PR_GET_NAME, (unsigned long)name, 0, 0, 0);
2147 name [namelen] = 0;
2148 len = strlen (name);
2149 strcpy (name + (len <= namelen - 4 ? len : namelen - 4), "/eio");
2150 prctl (PR_SET_NAME, (unsigned long)name, 0, 0, 0);
2151#endif
2152}
2153
2118X_THREAD_PROC (etp_proc) 2154X_THREAD_PROC (etp_proc)
2119{ 2155{
2120 ETP_REQ *req; 2156 ETP_REQ *req;
2121 struct timespec ts; 2157 struct timespec ts;
2122 etp_worker *self = (etp_worker *)thr_arg; 2158 etp_worker *self = (etp_worker *)thr_arg;
2123 2159
2124#if HAVE_PRCTL_SET_NAME 2160 etp_proc_init ();
2125 prctl (PR_SET_NAME, (unsigned long)"eio_thread", 0, 0, 0);
2126#endif
2127 2161
2128 /* try to distribute timeouts somewhat evenly */ 2162 /* try to distribute timeouts somewhat evenly */
2129 ts.tv_nsec = ((unsigned long)self & 1023UL) * (1000000000UL / 1024UL); 2163 ts.tv_nsec = ((unsigned long)self & 1023UL) * (1000000000UL / 1024UL);
2130 2164
2131 for (;;) 2165 for (;;)
2304 case EIO_OPEN: req->result = openat (dirfd, req->ptr1, req->int1, (mode_t)req->int2); break; 2338 case EIO_OPEN: req->result = openat (dirfd, req->ptr1, req->int1, (mode_t)req->int2); break;
2305 2339
2306 case EIO_UNLINK: req->result = unlinkat (dirfd, req->ptr1, 0); break; 2340 case EIO_UNLINK: req->result = unlinkat (dirfd, req->ptr1, 0); break;
2307 case EIO_RMDIR: req->result = unlinkat (dirfd, req->ptr1, AT_REMOVEDIR); break; 2341 case EIO_RMDIR: req->result = unlinkat (dirfd, req->ptr1, AT_REMOVEDIR); break;
2308 case EIO_MKDIR: req->result = mkdirat (dirfd, req->ptr1, (mode_t)req->int2); break; 2342 case EIO_MKDIR: req->result = mkdirat (dirfd, req->ptr1, (mode_t)req->int2); break;
2309 case EIO_RENAME: req->result = renameat (dirfd, req->ptr1, WD2FD (req->int3), req->ptr2); break; 2343 case EIO_RENAME: req->result = renameat (dirfd, req->ptr1, WD2FD ((eio_wd)req->int3), req->ptr2); break;
2310 case EIO_LINK: req->result = linkat (dirfd, req->ptr1, WD2FD (req->int3), req->ptr2, 0); break; 2344 case EIO_LINK: req->result = linkat (dirfd, req->ptr1, WD2FD ((eio_wd)req->int3), req->ptr2, 0); break;
2311 case EIO_SYMLINK: req->result = symlinkat (req->ptr1, dirfd, req->ptr2); break; 2345 case EIO_SYMLINK: req->result = symlinkat (req->ptr1, dirfd, req->ptr2); break;
2312 case EIO_MKNOD: req->result = mknodat (dirfd, req->ptr1, (mode_t)req->int2, (dev_t)req->offs); break; 2346 case EIO_MKNOD: req->result = mknodat (dirfd, req->ptr1, (mode_t)req->int2, (dev_t)req->offs); break;
2313 case EIO_READLINK: ALLOC (PATH_MAX); 2347 case EIO_READLINK: ALLOC (PATH_MAX);
2314 req->result = readlinkat (dirfd, req->ptr1, req->ptr2, PATH_MAX); break; 2348 req->result = readlinkat (dirfd, req->ptr1, req->ptr2, PATH_MAX); break;
2315 case EIO_STATVFS: ALLOC (sizeof (EIO_STRUCT_STATVFS)); 2349 case EIO_STATVFS: ALLOC (sizeof (EIO_STRUCT_STATVFS));
2415 case EIO_MTOUCH: req->result = eio__mtouch (req); break; 2449 case EIO_MTOUCH: req->result = eio__mtouch (req); break;
2416 case EIO_MLOCK: req->result = eio__mlock (req->ptr2, req->size); break; 2450 case EIO_MLOCK: req->result = eio__mlock (req->ptr2, req->size); break;
2417 case EIO_MLOCKALL: req->result = eio__mlockall (req->int1); break; 2451 case EIO_MLOCKALL: req->result = eio__mlockall (req->int1); break;
2418 case EIO_FALLOCATE: req->result = eio__fallocate (req->int1, req->int2, req->offs, req->size); break; 2452 case EIO_FALLOCATE: req->result = eio__fallocate (req->int1, req->int2, req->offs, req->size); break;
2419 2453
2420 case EIO_READDIR: eio__scandir (req); break; 2454 case EIO_READDIR: eio__scandir (req, self); break;
2421 2455
2422 case EIO_BUSY: 2456 case EIO_BUSY:
2423#ifdef _WIN32 2457#ifdef _WIN32
2424 Sleep (req->nv1 * 1e3); 2458 Sleep (req->nv1 * 1e3);
2425#else 2459#else
2453 2487
2454 req->errorno = errno; 2488 req->errorno = errno;
2455} 2489}
2456 2490
2457#ifndef EIO_NO_WRAPPERS 2491#ifndef EIO_NO_WRAPPERS
2492
2493eio_req *eio_wd_open (const char *path, int pri, eio_cb cb, void *data)
2494{
2495 REQ (EIO_WD_OPEN); PATH; SEND;
2496}
2497
2498eio_req *eio_wd_close (eio_wd wd, int pri, eio_cb cb, void *data)
2499{
2500 REQ (EIO_WD_CLOSE); req->wd = wd; SEND;
2501}
2458 2502
2459eio_req *eio_nop (int pri, eio_cb cb, void *data) 2503eio_req *eio_nop (int pri, eio_cb cb, void *data)
2460{ 2504{
2461 REQ (EIO_NOP); SEND; 2505 REQ (EIO_NOP); SEND;
2462} 2506}

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines