+/*
+ * 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.gnu.org/licenses/gpl-2.0.html
+ *
+ * GPL HEADER END
+ */
+/*
+ * Copyright (c) 2002, 2010, Oracle and/or its affiliates. All rights reserved.
+ * Use is subject to license terms.
+ *
+ * Copyright (c) 2014, Intel Corporation.
+ */
+/*
+ * 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 <inttypes.h>
#include <stdio.h>
#include <string.h>
#include <unistd.h>
#include <stdlib.h>
#include <errno.h>
#include <sys/mman.h>
+#include <sys/types.h>
+#include <sys/stat.h>
+#include <linux/types.h>
-// not correctly in the headers yet!!
-//#define O_DIRECT 0
-#ifndef O_DIRECT
-#define O_DIRECT 040000 /* direct disk access hint */
-#endif
+#define READ 1
+#define WRITE 2
-#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 LPDS sizeof(uint64_t)
+int block_debug_setup(void *addr, int len, uint64_t off, uint64_t id)
+{
+ off = htole64(off);
+ id = htole64(id);
+ memcpy(addr, (char *)&off, LPDS);
+ memcpy(addr + LPDS, (char *)&id, LPDS);
-#define LPDS sizeof(__u64)
-int page_debug_setup(void *addr, int len, __u64 off, __u64 id)
+ addr += len - LPDS - LPDS;
+ memcpy(addr, (char *)&off, LPDS);
+ memcpy(addr + LPDS, (char *)&id, LPDS);
+
+ return 0;
+}
+
+int block_debug_check(char *who, void *addr, int size, uint64_t off,
+ uint64_t id)
{
- LASSERT(addr);
+ uint64_t ne_off;
+ int err = 0;
- off = HTON__u64(off);
- id = HTON__u64(id);
- memcpy(addr, (char *)&off, LPDS);
- memcpy(addr + LPDS, (char *)&id, LPDS);
+ ne_off = le64toh(off);
+ id = le64toh(id);
+ if (memcmp(addr, (char *)&ne_off, LPDS)) {
+ fprintf(stderr, "%s: for offset %"PRIu64" off: %"PRIx64" != %"PRIx64"\n",
+ who, off, *(uint64_t *)addr, ne_off);
+ err = -EINVAL;
+ }
+ if (memcmp(addr + LPDS, (char *)&id, LPDS)) {
+ fprintf(stderr, "%s: for offset %"PRIu64" id: %"PRIx64" != %"PRIx64"\n",
+ who, off, *(uint64_t *)(addr + LPDS), id);
+ err = -EINVAL;
+ }
- addr += len - LPDS - LPDS;
- memcpy(addr, (char *)&off, LPDS);
- memcpy(addr + LPDS, (char *)&id, LPDS);
+ addr += size - LPDS - LPDS;
+ if (memcmp(addr, (char *)&ne_off, LPDS)) {
+ fprintf(stderr, "%s: for offset %"PRIu64" end off: %"PRIx64" != %"PRIx64"\n",
+ who, off, *(uint64_t *)addr, ne_off);
+ err = -EINVAL;
+ }
+ if (memcmp(addr + LPDS, (char *)&id, LPDS)) {
+ fprintf(stderr, "%s: for offset %"PRIu64" end id: %"PRIx64" != %"PRIx64"\n",
+ who, off, *(uint64_t *)(addr + LPDS), id);
+ err = -EINVAL;
+ }
- return 0;
+ return err;
}
-int page_debug_check(char *who, void *addr, int end, __u64 off, __u64 id)
+#undef LPDS
+
+void usage(char *prog)
{
- __u64 ne_off;
- int err = 0;
-
- LASSERT(addr);
-
- ne_off = HTON__u64(off);
- id = HTON__u64(id);
- if (memcmp(addr, (char *)&ne_off, LPDS)) {
- CERROR("%s: for offset "LPU64" off: "LPX64" != "LPX64"\n",
- who, off, *(__u64 *)addr, ne_off);
- err = -EINVAL;
- }
- if (memcmp(addr + LPDS, (char *)&id, LPDS)) {
- CERROR("%s: for offset "LPU64" id: "LPX64" != "LPX64"\n",
- who, off, *(__u64 *)(addr + LPDS), id);
- err = -EINVAL;
- }
-
- addr += end - 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);
- err = -EINVAL;
- }
- if (memcmp(addr + LPDS, (char *)&id, LPDS)) {
- CERROR("%s: for offset "LPU64" end id: "LPX64" != "LPX64"\n",
- who, off, *(__u64 *)(addr + LPDS), id);
- err = -EINVAL;
- }
-
- return err;
+ fprintf(stderr,
+ "usage: %s file count [[d]{r|w|rw} [pages_per_vec [objid]]]\n",
+ prog);
+ exit(1);
}
-#undef LPDS
int main(int argc, char **argv)
{
- int fd;
- char *buf;
- long pg_vec, count;
- long len;
- long long end, offset;
- long objid = 3;
- int rc;
-
- if (argc < 4 || argc > 5) {
- fprintf(stderr,
- "usage: %s file pages_per_vec count [objid]\n",
+ int fd;
+ char *buf;
+ long long count, last;
+ long pg_vec, len;
+ uint64_t objid, offset;
+ struct stat st;
+ int flags = 0;
+ int cmd = 0;
+ char *end;
+ int rc;
+
+ 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]);
- return 1;
- }
+#endif
+ }
- pg_vec = strtoul(argv[2], 0, 0);
- count = strtoul(argv[3], 0, 0);
- len = pg_vec * BLOCKSIZE;
- end = (long long)count * len;
+ 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 == 5)
- objid = strtoul(argv[4], 0, 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]);
+ }
+ } else {
+ objid = 3;
+ }
- printf("directio on %s(%ld) for %ldx%ld pages \n",
+ printf("%s: %s on %s(objid %"PRIx64") 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);
- 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;
- }
+ fd = open(argv[1], flags | O_LARGEFILE);
+ if (fd == -1) {
+ fprintf(stderr, "%s: cannot open %s: %s\n", argv[0],
+ argv[1], strerror(errno));
+ return 3;
+ }
- fd = open(argv[1], O_DIRECT | O_RDWR | O_CREAT);
- if (fd == -1) {
- fprintf(stderr, "Cannot open %s: %s\n", 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;
+ }
- for (offset = 0; offset < end; offset += len) {
- int i;
+ len = pg_vec * st.st_blksize;
+ last = (long long)count * len;
- for (i = 0; i < len; i += BLOCKSIZE)
- page_debug_setup(buf + i, BLOCKSIZE, offset + i, objid);
+ 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) {
+ unsigned int i;
+
+ 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 += BLOCKSIZE) {
- if (page_debug_check("write", buf + i, BLOCKSIZE,
- offset + i, objid))
+ 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, "Write error: %s, rc %d\n",
- strerror(errno), rc);
+ 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, "Cannot seek %s\n", strerror(errno));
- return 5;
- }
+ 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) {
+ for (offset = 0; offset < last && cmd & READ; offset += len) {
int i;
rc = read(fd, buf, len);
if (rc != len) {
- fprintf(stderr, "Read error: %s, rc %d\n",
- strerror(errno), rc);
+ fprintf(stderr, "%s: read error: %s, rc %d != %ld\n",
+ argv[0], strerror(errno), rc, len);
return 6;
}
- for (i = 0; i < len; i += BLOCKSIZE) {
- if (page_debug_check("read", buf + i, BLOCKSIZE,
- offset + i, objid))
+ 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;
+ return 0;
}