X-Git-Url: https://git.whamcloud.com/?p=fs%2Flustre-release.git;a=blobdiff_plain;f=lustre%2Ftests%2Ftest_brw.c;h=e0b39932af172870d3be76c2121cd10a5137e834;hp=7da1221dfdbe2c98b899d3073d5418a9d608b5d0;hb=4f8b7393063e2e14dbd46f899dfbf20c7b646af8;hpb=22e9915b30aa80a2d768100ccbb54825eaf90b6d diff --git a/lustre/tests/test_brw.c b/lustre/tests/test_brw.c index 7da1221..e0b3993 100644 --- a/lustre/tests/test_brw.c +++ b/lustre/tests/test_brw.c @@ -1,3 +1,44 @@ +/* -*- mode: c; c-basic-offset: 8; indent-tabs-mode: nil; -*- + * vim:expandtab:shiftwidth=8:tabstop=8: + * + * GPL HEADER START + * + * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. + * + * This program is free software; you can redistribute it and/or modify + * it under the terms of the GNU General Public License version 2 only, + * as published by the Free Software Foundation. + * + * This program is distributed in the hope that it will be useful, but + * WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU + * General Public License version 2 for more details (a copy is included + * in the LICENSE file that accompanied this code). + * + * You should have received a copy of the GNU General Public License + * version 2 along with this program; If not, see + * http://www.sun.com/software/products/lustre/docs/GPLv2.pdf + * + * Please contact Sun Microsystems, Inc., 4150 Network Circle, Santa Clara, + * CA 95054 USA or visit www.sun.com if you need additional information or + * have any questions. + * + * GPL HEADER END + */ +/* + * Copyright (c) 2002, 2010, Oracle and/or its affiliates. All rights reserved. + * Use is subject to license terms. + */ +/* + * This file is part of Lustre, http://www.lustre.org/ + * Lustre is a trademark of Sun Microsystems, Inc. + */ + +/* for O_DIRECT */ +#ifndef _GNU_SOURCE +#define _GNU_SOURCE +#endif + #include #include #include @@ -5,31 +46,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 - -#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 "%Lu" -#define LPX64 "%Lx" +#include #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); @@ -40,15 +69,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); @@ -60,7 +87,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); @@ -78,116 +105,164 @@ int page_debug_check(char *who, void *addr, int end, __u64 off, __u64 id) void usage(char *prog) { - fprintf(stderr, "usage: %s file count [[d]{r|w|rw} [pages_per_vec [objid]]]\n", - prog); - exit(1); + 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 long objid = 3; - int flags = O_RDWR | O_CREAT; - int cmd = 0; + 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 < 3 || argc > 6) - usage(argv[0]); - - count = strtoul(argv[2], 0, 0); - if (argc >= 4) { - if (strchr(argv[3], 'r')) { - cmd |= READ; - printf("reading\n"); - } - if (strchr(argv[3], 'w')) { - cmd |= WRITE; - printf("writing\n"); - } - if (strchr(argv[3], 'd')) { - flags |= O_DIRECT; - printf("directing\n"); - } - } - if (!cmd) - usage(argv[0]); - printf("cmd = %x, flags = %x\n", cmd, flags); - - if (argc >= 5) - pg_vec = strtoul(argv[4], 0, 0); - len = pg_vec * BLOCKSIZE; - end = (long long)count * len; - - if (argc >= 6) { - objid = strtoull(argv[5], 0, 0); - printf("objid %s = 0x%Lx\n", argv[5], objid); - } - - printf("%s: %s on %s(objid 0x"LPX64") for %ldx%ld pages \n", - argv[0], flags & O_DIRECT ? "directio" : "i/o", - argv[1], objid, count, pg_vec); + usage(argv[0]); - buf = mmap(0, len, PROT_READ|PROT_WRITE, MAP_PRIVATE|MAP_ANON, 0, 0); - if (!buf) { - fprintf(stderr, "%s: no buffer memory %s\n", - argv[0], strerror(errno)); - return 2; + 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 + } + + 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; } - fd = open(argv[1], flags); + 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; + } + + 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, "%s: cannot open %s: %s\n", argv[1], - argv[0], strerror(errno)); + fprintf(stderr, "%s: cannot open %s: %s\n", argv[0], + argv[1], strerror(errno)); return 3; } - for (offset = 0; offset < end && cmd & WRITE; 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 (i = 0; i < len; i += BLOCKSIZE) { - if (page_debug_check("write", buf + i, BLOCKSIZE, - offset + i, objid)) - return 10; - } + for (offset = 0; offset < last && cmd & WRITE; offset += len) { + int i; - if (rc != len) { - fprintf(stderr, "%s: write error: %s, rc %d\n", - argv[0], strerror(errno), rc); - return 4; - } - } + for (i = 0; i < len; i += st.st_blksize) + block_debug_setup(buf + i, st.st_blksize, + offset + i, objid); + + rc = write(fd, buf, len); + + 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 (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)); + argv[0], strerror(errno)); return 5; } - for (offset = 0; offset < end && cmd && READ; offset += len) { - int i; - - rc = read(fd, buf, len); - if (rc != len) { - fprintf(stderr, "%s: read error: %s, rc %d\n", - argv[0], 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; }