X-Git-Url: https://git.whamcloud.com/?p=fs%2Flustre-release.git;a=blobdiff_plain;f=lustre%2Ftests%2Ftest_brw.c;h=3d7a3447755ea9c84e0396b5069933fcc3738e26;hp=6cbfcb5002957a8c70486097dd33026de4d56f7e;hb=c52d48bf7838c8d9a886a8865017e150e294d91c;hpb=96ec6856f91f7f9031cfce4273c714d72cfe59ae diff --git a/lustre/tests/test_brw.c b/lustre/tests/test_brw.c index 6cbfcb5..3d7a344 100644 --- a/lustre/tests/test_brw.c +++ b/lustre/tests/test_brw.c @@ -1,3 +1,42 @@ +/* + * 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 @@ -8,29 +47,13 @@ #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 CERROR(fmt, arg...) fprintf(stderr, fmt, ## arg) -#ifndef __u64 -#define __u64 long long -#define cpu_to_le64(v) (v) -#define le64_to_cpu(v) (v) -#endif - -#ifndef LPU64 -#define LPU64 "%Lu" -#define LPX64 "%#Lx" -#endif +#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) { off = cpu_to_le64(off); id = cpu_to_le64(id); @@ -44,7 +67,7 @@ int page_debug_setup(void *addr, int len, __u64 off, __u64 id) return 0; } -int page_debug_check(char *who, void *addr, int size, __u64 off, __u64 id) +int block_debug_check(char *who, void *addr, int size, __u64 off, __u64 id) { __u64 ne_off; int err = 0; @@ -80,142 +103,164 @@ int page_debug_check(char *who, void *addr, int size, __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 long count, last, offset; - long pg_vec, len; - long long objid = 3; - struct stat st; - int flags = 0; - int cmd = 0; - char *end; + 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]); + 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')) { - flags |= O_DIRECT; - } - if (!cmd) - usage(argv[0]); - } else { - cmd = READ | WRITE; - flags = O_RDWR | O_CREAT | O_DIRECT; - } - - 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]); - } - } - - 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]); - } - } - - printf("%s: %s on %s(objid "LPX64") for "LPU64"x%ld pages \n", - argv[0], flags & O_DIRECT ? "directio" : "i/o", - argv[1], objid, count, pg_vec); + 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; + } + + 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[0], - argv[1], strerror(errno)); + argv[1], strerror(errno)); return 3; } - rc = fstat(fd, &st); - if (rc < 0) { - fprintf(stderr, "%s: cannot stat %s: %s\n", argv[0], - argv[1], strerror(errno)); - return 4; - } + rc = fstat(fd, &st); + if (rc < 0) { + fprintf(stderr, "%s: cannot stat %s: %s\n", argv[0], + argv[1], strerror(errno)); + return 4; + } - len = pg_vec * st.st_blksize; - last = (long long)count * len; + len = pg_vec * st.st_blksize; + last = (long long)count * len; buf = mmap(0, len, PROT_READ|PROT_WRITE, MAP_PRIVATE|MAP_ANON, 0, 0); - if (!buf) { + if (buf == MAP_FAILED) { fprintf(stderr, "%s: no buffer memory %s\n", - argv[0], strerror(errno)); + argv[0], strerror(errno)); return 2; } - for (offset = 0; offset < last && cmd & WRITE; offset += len) { - int i; + for (offset = 0; offset < last && cmd & WRITE; offset += len) { + int i; - for (i = 0; i < len; i += st.st_blksize) - page_debug_setup(buf + i, st.st_blksize, offset + i, - objid); + 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); + rc = write(fd, buf, len); - for (i = 0; i < len; i += st.st_blksize) { - if (page_debug_check("write", buf + i, st.st_blksize, - offset + i, objid)) - return 10; - } + 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 (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 < 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 (page_debug_check("read", buf + i, st.st_blksize, - 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; }