diff options
author | Robin Gareus <robin@gareus.org> | 2015-03-12 13:15:12 +0100 |
---|---|---|
committer | Robin Gareus <robin@gareus.org> | 2015-03-12 13:15:12 +0100 |
commit | b0405c3c2ce0038cf8e4fac3fd17df80a475f1d1 (patch) | |
tree | eb2cab6d0de27a8fcb08512a0b27d5f781a3c4e0 /tools/readtest.c | |
parent | 421d4aae2e44baa73c03e5b5201583143fd38215 (diff) |
pimp readtest.
Diffstat (limited to 'tools/readtest.c')
-rw-r--r-- | tools/readtest.c | 86 |
1 files changed, 80 insertions, 6 deletions
diff --git a/tools/readtest.c b/tools/readtest.c index 4e04629583..95907f530e 100644 --- a/tools/readtest.c +++ b/tools/readtest.c @@ -1,5 +1,9 @@ /* gcc -o readtest readtest.c `pkg-config --cflags --libs glib-2.0` -lm */ +#ifndef _WIN32 +# define HAVE_MMAP +#endif + #include <stdlib.h> #include <errno.h> #include <stdio.h> @@ -10,6 +14,11 @@ #include <fcntl.h> #include <math.h> +#ifdef HAVE_MMAP +# include <sys/stat.h> +# include <sys/mman.h> +#endif + #include <glib.h> char* data = 0; @@ -17,22 +26,30 @@ char* data = 0; void usage () { - fprintf (stderr, "readtest [ -b BLOCKSIZE ] [-l FILELIMIT] [ -D ] filename-template\n"); + fprintf (stderr, "readtest [ -b BLOCKSIZE ] [-l FILELIMIT] [ -D ] [ -R ] [ -M ] filename-template\n"); } int main (int argc, char* argv[]) { int* files; - char optstring[] = "b:Dl:q"; + char optstring[] = "b:DRMl:q"; uint32_t block_size = 64 * 1024 * 4; int max_files = -1; #ifdef __APPLE__ int direct = 0; + int noreadahead = 0; +#endif +#ifdef HAVE_MMAP + int use_mmap = 0; + void **addr; + size_t *flen; #endif const struct option longopts[] = { { "blocksize", 1, 0, 'b' }, { "direct", 0, 0, 'D' }, + { "mmap", 0, 0, 'M' }, + { "noreadahead", 0, 0, 'R' }, { "limit", 1, 0, 'l' }, { 0, 0, 0, 0 } }; @@ -62,6 +79,16 @@ main (int argc, char* argv[]) direct = 1; #endif break; + case 'M': +#ifdef HAVE_MMAP + use_mmap = 1; +#endif + break; + case 'R': +#ifdef __APPLE__ + noreadahead = 1; +#endif + break; case 'q': quiet = 1; break; @@ -105,6 +132,15 @@ main (int argc, char* argv[]) nfiles = n; files = (int *) malloc (sizeof (int) * nfiles); +#ifdef HAVE_MMAP + if (use_mmap) { + if (!quiet) { + printf ("# Using mmap().\n"); + } + addr = malloc (sizeof (void*) * nfiles); + flen = (size_t*) malloc (sizeof (size_t) * nfiles); + } +#endif for (n = 0; n < nfiles; ++n) { @@ -128,9 +164,35 @@ main (int argc, char* argv[]) fprintf (stderr, "Cannot set F_NOCACHE on file #%d\n", n); } } + + if (noreadahead) { + if (fcntl (fd, F_RDAHEAD, 0) == -1) { + fprintf (stderr, "Cannot set F_READAHED on file #%d\n", n); + } + } #endif files[n] = fd; + +#ifdef HAVE_MMAP + if (use_mmap) { + struct stat s; + if (fstat (fd, & s)) { + fprintf (stderr, "Could not stat fd #%d @ %s\n", n, path); + return 1; + } + if (s.st_size < block_size) { + fprintf (stderr, "file is shorter than blocksize #%d @ %s\n", n, path); + return 1; + } + flen[n] = s.st_size; + addr[n] = mmap (0, s.st_size, PROT_READ, MAP_PRIVATE, fd, 0); + if (addr[n] == MAP_FAILED) { + fprintf (stderr, "Could not mmap file #%d @ %s (%s)\n", n, path, strerror (errno)); + return 1; + } + } +#endif } data = (char*) malloc (sizeof (char) * block_size); @@ -145,10 +207,22 @@ main (int argc, char* argv[]) gint64 before; before = g_get_monotonic_time(); - for (n = 0; n < nfiles; ++n) { - - if (read (files[n], (char*) data, block_size) != block_size) { - goto out; +#ifdef HAVE_MMAP + if (use_mmap) { + for (n = 0; n < nfiles; ++n) { + if (_read + n + block_size > flen[n]) { + goto out; + } + memmove(data, &addr[n][_read], block_size); + } + } + else +#endif + { + for (n = 0; n < nfiles; ++n) { + if (read (files[n], (char*) data, block_size) != block_size) { + goto out; + } } } |