X-Git-Url: https://git.whamcloud.com/?a=blobdiff_plain;f=lustre%2Ftests%2Ftest_brw.c;h=0e5605ec15a9f20f72c8de1fade4f40c66765fc5;hb=2403e112d69cb58a15cec0018979a89ae389b3a4;hp=7f97a7733ce6bcc164b5042c4430600a789bab64;hpb=3a7236fc0944f6c1c685c4f8fdae5bf7a5ff4e5c;p=fs%2Flustre-release.git diff --git a/lustre/tests/test_brw.c b/lustre/tests/test_brw.c index 7f97a77..0e5605e 100644 --- a/lustre/tests/test_brw.c +++ b/lustre/tests/test_brw.c @@ -1,3 +1,10 @@ +/* -*- mode: c; c-basic-offset: 8; indent-tabs-mode: nil; -*- + * vim:expandtab:shiftwidth=8:tabstop=8: + */ + +/* for O_DIRECT */ +#define _GNU_SOURCE + #include #include #include @@ -5,28 +12,19 @@ #include #include #include +#include +#include -// not correctly in the headers yet!! -//#define O_DIRECT 0 -#ifndef O_DIRECT -#define O_DIRECT 040000 /* direct disk access hint */ -#endif +#include -#define BLOCKSIZE 4096 -#define CERROR(fmt, arg...) fprintf(stderr, fmt, ## arg) -#define __u64 long long -#define LASSERT(v) do {} while(0) -#define HTON__u64(v) (v) -#define LPU64 "%Ld" -#define LPX64 "%Lx" +#define READ 1 +#define WRITE 2 #define LPDS sizeof(__u64) -int page_debug_setup(void *addr, int len, __u64 off, __u64 id) +int block_debug_setup(void *addr, int len, __u64 off, __u64 id) { - LASSERT(addr); - - off = HTON__u64(off); - id = HTON__u64(id); + off = cpu_to_le64(off); + id = cpu_to_le64(id); memcpy(addr, (char *)&off, LPDS); memcpy(addr + LPDS, (char *)&id, LPDS); @@ -37,15 +35,13 @@ int page_debug_setup(void *addr, int len, __u64 off, __u64 id) return 0; } -int page_debug_check(char *who, void *addr, int end, __u64 off, __u64 id) +int block_debug_check(char *who, void *addr, int size, __u64 off, __u64 id) { __u64 ne_off; int err = 0; - LASSERT(addr); - - ne_off = HTON__u64(off); - id = HTON__u64(id); + ne_off = le64_to_cpu(off); + id = le64_to_cpu(id); if (memcmp(addr, (char *)&ne_off, LPDS)) { CERROR("%s: for offset "LPU64" off: "LPX64" != "LPX64"\n", who, off, *(__u64 *)addr, ne_off); @@ -57,7 +53,7 @@ int page_debug_check(char *who, void *addr, int end, __u64 off, __u64 id) err = -EINVAL; } - addr += end - LPDS - LPDS; + addr += size - LPDS - LPDS; if (memcmp(addr, (char *)&ne_off, LPDS)) { CERROR("%s: for offset "LPU64" end off: "LPX64" != "LPX64"\n", who, off, *(__u64 *)addr, ne_off); @@ -73,84 +69,166 @@ int page_debug_check(char *who, void *addr, int end, __u64 off, __u64 id) } #undef LPDS +void usage(char *prog) +{ + fprintf(stderr, + "usage: %s file count [[d]{r|w|rw} [pages_per_vec [objid]]]\n", + prog); + exit(1); +} + int main(int argc, char **argv) { int fd; char *buf; - long pg_vec, count; - long len; - long long end, offset; - long objid = 3; + long long count, last, offset; + long pg_vec, len; + __u64 objid; + struct stat st; + int flags = 0; + int cmd = 0; + char *end; int rc; - if (argc < 4 || argc > 5) { - fprintf(stderr, - "usage: %s file pages_per_vec count [objid]\n", - argv[0]); - return 1; + if (argc < 3 || argc > 6) + usage(argv[0]); + + count = strtoull(argv[2], &end, 0); + if (*end) { + fprintf(stderr, "%s: invalid count '%s'\n", argv[0], argv[2]); + usage(argv[0]); + } + if (argc >= 4) { + if (strchr(argv[3], 'r')) { + cmd = READ; + flags = O_RDONLY; + } + if (strchr(argv[3], 'w')) { + cmd |= WRITE; + flags = O_RDWR | O_CREAT; + } + if (strchr(argv[3], 'd')) { +#ifdef O_DIRECT + flags |= O_DIRECT; +#else + fprintf(stderr, + "%s: O_DIRECT not supported in this build\n", + argv[0]); + exit(1); +#endif + } + if (!cmd) + usage(argv[0]); + } else { + cmd = READ | WRITE; + flags = O_RDWR | O_CREAT; +#ifdef O_DIRECT + flags |= O_DIRECT; +#else + fprintf(stderr, "%s: warning: not setting O_DIRECT\n", + argv[0]); +#endif } - pg_vec = strtoul(argv[2], 0, 0); - count = strtoul(argv[3], 0, 0); - len = pg_vec * BLOCKSIZE; - end = (long long)count * len; - printf("directio on %s for %ldx%ld pages \n", argv[1], count, pg_vec); + if (argc >= 5) { + pg_vec = strtoul(argv[4], &end, 0); + if (*end) { + fprintf(stderr, "%s: invalid pages_per_vec '%s'\n", + argv[0], argv[4]); + usage(argv[0]); + } + } else { + pg_vec = 16; + } - buf = mmap(0, len, PROT_READ|PROT_WRITE, MAP_PRIVATE|MAP_ANON, 0, 0); - if (!buf) { - fprintf(stderr, "No memory %s\n", strerror(errno)); - return 2; + if (argc >= 6) { + objid = strtoull(argv[5], &end, 0); + if (*end) { + fprintf(stderr, "%s: invalid objid '%s'\n", + argv[0], argv[5]); + usage(argv[0]); + } + } else { + objid = 3; } - fd = open(argv[1], O_DIRECT | O_RDWR | O_CREAT); + printf("%s: %s on %s(objid "LPX64") for %llux%ld pages \n", + argv[0], +#ifdef O_DIRECT + flags & O_DIRECT ? "directio" : "i/o", +#else + "i/o", +#endif + argv[1], objid, count, pg_vec); + + fd = open(argv[1], flags | O_LARGEFILE); if (fd == -1) { - fprintf(stderr, "Cannot open %s: %s\n", argv[1], - strerror(errno)); + fprintf(stderr, "%s: cannot open %s: %s\n", argv[0], + argv[1], strerror(errno)); return 3; } - for (offset = 0; offset < end; offset += len) { - int i; + rc = fstat(fd, &st); + if (rc < 0) { + fprintf(stderr, "%s: cannot stat %s: %s\n", argv[0], + argv[1], strerror(errno)); + return 4; + } - for (i = 0; i < len; i += BLOCKSIZE) - page_debug_setup(buf + i, BLOCKSIZE, offset + i, objid); + len = pg_vec * st.st_blksize; + last = (long long)count * len; - rc = write(fd, buf, len); + buf = mmap(0, len, PROT_READ|PROT_WRITE, MAP_PRIVATE|MAP_ANON, 0, 0); + if (buf == MAP_FAILED) { + fprintf(stderr, "%s: no buffer memory %s\n", + argv[0], strerror(errno)); + return 2; + } + + for (offset = 0; offset < last && cmd & WRITE; offset += len) { + int i; + + for (i = 0; i < len; i += st.st_blksize) + block_debug_setup(buf + i, st.st_blksize, + offset + i, objid); - for (i = 0; i < len; i += BLOCKSIZE) { - if (page_debug_check("write", buf + i, BLOCKSIZE, - offset + i, objid)) - return 10; - } + rc = write(fd, buf, len); - if (rc != len) { - fprintf(stderr, "Write error: %s, rc %d\n", - strerror(errno), rc); - return 4; - } - } + for (i = 0; i < len; i += st.st_blksize) { + if (block_debug_check("write", buf + i, st.st_blksize, + offset + i, objid)) + return 10; + } - if ( lseek(fd, 0, SEEK_SET) != 0 ) { - fprintf(stderr, "Cannot seek %s\n", strerror(errno)); + if (rc != len) { + fprintf(stderr, "%s: write error: %s, rc %d != %ld\n", + argv[0], strerror(errno), rc, len); + return 4; + } + } + + if (lseek(fd, 0, SEEK_SET) != 0) { + fprintf(stderr, "%s: cannot seek %s\n", + argv[0], strerror(errno)); return 5; } - for (offset = 0; offset < end; offset += len) { - int i; - - rc = read(fd, buf, len); - if (rc != len) { - fprintf(stderr, "Read error: %s, rc %d\n", - strerror(errno), rc); - return 6; - } - - for (i = 0; i < len; i += BLOCKSIZE) { - if (page_debug_check("read", buf + i, BLOCKSIZE, - offset + i, objid)) - return 11; - } - } + for (offset = 0; offset < last && cmd & READ; offset += len) { + int i; + + rc = read(fd, buf, len); + if (rc != len) { + fprintf(stderr, "%s: read error: %s, rc %d != %ld\n", + argv[0], strerror(errno), rc, len); + return 6; + } + + for (i = 0; i < len; i += st.st_blksize) { + if (block_debug_check("read", buf + i, st.st_blksize, + offset + i, objid)) + return 11; + } + } return 0; }