+/*
+ * 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.
+ */
+/*
+ * This file is part of Lustre, http://www.lustre.org/
+ * Lustre is a trademark of Sun Microsystems, Inc.
+ */
+
+#ifndef _GNU_SOURCE
+#define _GNU_SOURCE
+#endif
#include <stdio.h>
#include <string.h>
#include <unistd.h>
#include <sys/stat.h>
#include <sys/mman.h>
-// not correctly in the headers yet!!
-#ifndef O_DIRECT
-#define O_DIRECT 040000 /* direct disk access hint */
-#endif
+/*
+ * return index of the first byte not matching given byte
+ * or buffer size if all bytes are matching
+ */
+static size_t check_bytes(const char *buf, int byte, size_t len)
+{
+ const char *p;
+
+ for (p = buf; p < buf + len; p++)
+ if (*p != byte)
+ break;
+ return p - buf;
+}
int main(int argc, char **argv)
{
- int fd;
- char *buf;
- int blocks, seek_blocks;
- long len;
- off64_t seek;
- struct stat64 st;
- int rc;
-
- if (argc != 4) {
- printf("Usage: %s file seek nr_blocks\n", argv[0]);
- return 1;
- }
-
- seek_blocks = strtoul(argv[2], 0, 0);
- blocks = strtoul(argv[3], 0, 0);
- fd = open(argv[1], O_LARGEFILE | O_DIRECT | O_RDWR | O_CREAT, 0644);
- if (fd == -1) {
- printf("Cannot open %s: %s\n", argv[1], strerror(errno));
- return 1;
- }
-
- if (fstat64(fd, &st) < 0) {
- printf("Cannot stat %s: %s\n", argv[1], strerror(errno));
- return 1;
- }
-
- printf("directio on %s for %dx%lu bytes \n", argv[1], blocks,
- st.st_blksize);
-
- seek = (off64_t)seek_blocks * (off64_t)st.st_blksize;
- if (lseek64(fd, seek, SEEK_SET) < 0) {
- printf("lseek64 failed: %s\n", strerror(errno));
- return 1;
- }
-
- len = blocks * st.st_blksize;
- buf = mmap(0, len, PROT_READ|PROT_WRITE, MAP_PRIVATE|MAP_ANON, 0, 0);
- if (!buf) {
- printf("No memory %s\n", strerror(errno));
- return 1;
- }
-
- memset(buf, 0xba, len);
- rc = write(fd, buf, len);
- if (rc != len) {
- printf("Write error %s (rc = %d)\n", strerror(errno), rc);
- return 1;
- }
-
- if (lseek64(fd, seek, SEEK_SET) < 0) {
- printf("Cannot seek %s\n", strerror(errno));
- return 1;
- }
-
- rc = read(fd, buf, len);
- if (rc != len) {
- printf("Read error: %s (rc = %d)\n", strerror(errno), rc);
- return 1;
- }
+#ifdef O_DIRECT
+ int fd;
+ char *buf, *fname;
+ int blocks, seek_blocks;
+ long len;
+ off64_t seek;
+ struct stat64 st;
+ char pad = 0xba;
+ int action;
+ int rc;
+
+ if (argc < 5 || argc > 6) {
+ printf("Usage: %s <read/write/rdwr/readhole> file seek nr_blocks [blocksize]\n",
+ argv[0]);
+ return 1;
+ }
+
+ if (!strcmp(argv[1], "read")) {
+ action = O_RDONLY;
+ } else if (!strcmp(argv[1], "write")) {
+ action = O_WRONLY;
+ } else if (!strcmp(argv[1], "rdwr")) {
+ action = O_RDWR;
+ } else if (!strcmp(argv[1], "readhole")) {
+ action = O_RDONLY;
+ pad = 0;
+ } else {
+ printf("Usage: %s <read/write/rdwr> file seek nr_blocks [blocksize]\n",
+ argv[0]);
+ return 1;
+ }
+
+ fname = argv[2];
+ seek_blocks = strtoul(argv[3], 0, 0);
+ blocks = strtoul(argv[4], 0, 0);
+ if (!blocks) {
+ printf("Usage: %s <read/write/rdwr> file seek nr_blocks [blocksize]\n",
+ argv[0]);
+ return 1;
+ }
+
+ fd = open(fname, O_LARGEFILE | O_DIRECT | O_RDWR | O_CREAT, 0644);
+ if (fd == -1) {
+ printf("Cannot open %s: %s\n", fname, strerror(errno));
+ return 1;
+ }
+
+ if (argc >= 6) {
+ st.st_blksize = strtoul(argv[5], 0, 0);
+ } else if (fstat64(fd, &st) < 0) {
+ printf("Cannot stat %s: %s\n", fname, strerror(errno));
+ return 1;
+ }
+
+ printf("directio on %s for %dx%lu bytes\n", fname, blocks,
+ (unsigned long)st.st_blksize);
+
+ seek = (off64_t)seek_blocks * (off64_t)st.st_blksize;
+ len = blocks * st.st_blksize;
+
+ buf = mmap(0, len,
+ PROT_READ | PROT_WRITE, MAP_PRIVATE | MAP_ANON, 0, 0);
+ if (buf == MAP_FAILED) {
+ printf("No memory %s\n", strerror(errno));
+ return 1;
+ }
+ memset(buf, pad, len);
+
+ if (action == O_WRONLY || action == O_RDWR) {
+ if (lseek64(fd, seek, SEEK_SET) < 0) {
+ printf("lseek64 failed: %s\n", strerror(errno));
+ return 1;
+ }
+
+ rc = write(fd, buf, len);
+ if (rc != len) {
+ printf("Write error %s (rc = %d, len = %ld)\n",
+ strerror(errno), rc, len);
+ return 1;
+ }
+ }
+
+ if (action == O_RDONLY || action == O_RDWR) {
+ if (lseek64(fd, seek, SEEK_SET) < 0) {
+ printf("Cannot seek %s\n", strerror(errno));
+ return 1;
+ }
+ /* reset all bytes to something nor 0x0 neither 0xab */
+ memset(buf, 0x5e, len);
+ rc = read(fd, buf, len);
+ if (rc != len) {
+ printf("Read error: %s rc = %d\n", strerror(errno), rc);
+ return 1;
+ }
+
+ if (check_bytes(buf, pad, len) != len) {
+ printf("Data mismatch\n");
+ return 1;
+ }
+ }
printf("PASS\n");
- return 0;
+ return 0;
+#else /* !O_DIRECT */
+#warning O_DIRECT not defined, directio test will fail
+ printf("O_DIRECT not defined\n");
+ return 1;
+#endif /* !O_DIRECT */
}