Whamcloud - gitweb
LU-12275 sec: verify dir is empty when setting enc policy
[fs/lustre-release.git] / lustre / tests / test_brw.c
index 396f3b0..855ee8e 100644 (file)
@@ -1,3 +1,41 @@
+/*
+ * 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 <sys/mman.h>
 #include <sys/types.h>
 #include <sys/stat.h>
-
-// 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 HTON__u64(v) (v)
-#endif
-
-#ifndef LPU64
-#define LPU64 "%Lu"
-#define LPX64 "%#Lx"
-#endif
+#include <linux/types.h>
 
 #define READ  1
 #define WRITE 2
 
-#define LPDS sizeof(__u64)
-int page_debug_setup(void *addr, int len, __u64 off, __u64 id)
+#define LPDS sizeof(uint64_t)
+int block_debug_setup(void *addr, int len, uint64_t off, uint64_t id)
 {
-        off = HTON__u64(off);
-        id = HTON__u64(id);
-        memcpy(addr, (char *)&off, LPDS);
-        memcpy(addr + LPDS, (char *)&id, LPDS);
+       off = htole64(off);
+       id = htole64(id);
+       memcpy(addr, (char *)&off, LPDS);
+       memcpy(addr + LPDS, (char *)&id, LPDS);
 
-        addr += len - LPDS - LPDS;
-        memcpy(addr, (char *)&off, LPDS);
-        memcpy(addr + LPDS, (char *)&id, LPDS);
+       addr += len - LPDS - LPDS;
+       memcpy(addr, (char *)&off, LPDS);
+       memcpy(addr + LPDS, (char *)&id, LPDS);
 
-        return 0;
+       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, uint64_t off,
+                     uint64_t id)
 {
-        __u64 ne_off;
-        int err = 0;
-
-        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 += 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);
-                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;
+       uint64_t ne_off;
+       int err = 0;
+
+       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 += 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 err;
 }
+
 #undef LPDS
 
 void usage(char *prog)
@@ -87,21 +111,21 @@ void usage(char *prog)
 
 int main(int argc, char **argv)
 {
-        int fd;
-        char *buf;
-       long long count, last, offset;
+       int fd;
+       char *buf;
+       long long count, last;
        long pg_vec, len;
-       long long objid = 3;
+       uint64_t objid, offset;
        struct stat st;
        int flags = 0;
        int cmd = 0;
        char *end;
-        int rc;
+       int rc;
 
-        if (argc < 3 || argc > 6)
+       if (argc < 3 || argc > 6)
                usage(argv[0]);
 
-        count = strtoull(argv[2], &end, 0);
+       count = strtoull(argv[2], &end, 0);
        if (*end) {
                fprintf(stderr, "%s: invalid count '%s'\n", argv[0], argv[2]);
                usage(argv[0]);
@@ -116,13 +140,26 @@ int main(int argc, char **argv)
                        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 | O_DIRECT;
+               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) {
@@ -132,6 +169,8 @@ int main(int argc, char **argv)
                                argv[0], argv[4]);
                        usage(argv[0]);
                }
+       } else {
+               pg_vec = 16;
        }
 
        if (argc >= 6) {
@@ -141,18 +180,25 @@ int main(int argc, char **argv)
                                argv[0], argv[5]);
                        usage(argv[0]);
                }
+       } else {
+               objid = 3;
        }
 
-        printf("%s: %s on %s(objid "LPX64") for "LPU64"x%ld pages \n",
-              argv[0], flags & O_DIRECT ? "directio" : "i/o",
+       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);
 
-        fd = open(argv[1], flags | O_LARGEFILE);
-        if (fd == -1) {
-                fprintf(stderr, "%s: cannot open %s:  %s\n", argv[0],
+       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;
-        }
+               return 3;
+       }
 
        rc = fstat(fd, &st);
        if (rc < 0) {
@@ -164,25 +210,26 @@ int main(int argc, char **argv)
        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) {
-                fprintf(stderr, "%s: no buffer memory %s\n",
+       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;
-        }
+               return 2;
+       }
 
        for (offset = 0; offset < last && cmd & WRITE; offset += len) {
-               int i;
+               unsigned int i;
 
                for (i = 0; i < len; i += st.st_blksize)
-                       page_debug_setup(buf + i, st.st_blksize, offset + i,
-                                        objid);
+                       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 (page_debug_check("write", buf + i, st.st_blksize,
-                                            offset + i, objid))
+                       if (block_debug_check("write", buf + i, st.st_blksize,
+                                             offset + i, objid))
                                return 10;
                }
 
@@ -193,13 +240,13 @@ int main(int argc, char **argv)
                }
        }
 
-        if (lseek(fd, 0, SEEK_SET) != 0) {
-                fprintf(stderr, "%s: cannot seek %s\n",
+       if (lseek(fd, 0, SEEK_SET) != 0) {
+               fprintf(stderr, "%s: cannot seek %s\n",
                        argv[0], strerror(errno));
-                return 5;
-        }
+               return 5;
+       }
 
-       for (offset = 0; offset < last && cmd && READ; offset += len) {
+       for (offset = 0; offset < last && cmd & READ; offset += len) {
                int i;
 
                rc = read(fd, buf, len);
@@ -210,11 +257,11 @@ int main(int argc, char **argv)
                }
 
                for (i = 0; i < len; i += st.st_blksize) {
-                       if (page_debug_check("read", buf + i, st.st_blksize,
-                                            offset + i, objid))
+                       if (block_debug_check("read", buf + i, st.st_blksize,
+                                             offset + i, objid))
                                return 11;
                }
        }
 
-        return 0;
+       return 0;
 }