land b_md onto HEAD:

[fs/lustre-release.git] / lustre / tests / test_brw.c

diff --git a/lustre/tests/test_brw.c b/lustre/tests/test_brw.c
index 977dbeb..196f32c 100644 (file)
--- a/lustre/tests/test_brw.c
+++ b/lustre/tests/test_brw.c
@@ -14,17 +14,22 @@
 
 #define BLOCKSIZE 4096
 #define CERROR(fmt, arg...) fprintf(stderr, fmt, ## arg)
+#ifndef __u64
 #define __u64 long long
-#define LASSERT(v) do {} while(0)
 #define HTON__u64(v) (v)
-#define LPU64 "%Ld"
-#define LPX64 "%Lx"
+#endif
+
+#ifndef LPU64
+#define LPU64 "%Lu"
+#define LPX64 "%#Lx"
+#endif
+
+#define READ  1
+#define WRITE 2
 
 #define LPDS sizeof(__u64)
 int page_debug_setup(void *addr, int len, __u64 off, __u64 id)
 {
-        LASSERT(addr);
-
         off = HTON__u64(off);
         id = HTON__u64(id);
         memcpy(addr, (char *)&off, LPDS);
@@ -37,13 +42,11 @@ 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 page_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);
         if (memcmp(addr, (char *)&ne_off, LPDS)) {
@@ -57,7 +60,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);
@@ -73,48 +76,92 @@ int page_debug_check(char *who, void *addr, int end, __u64 off, __u64 id)
 }
 #undef LPDS
 
+void usage(char *prog)
+{
+       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 objid = 3;
+       long long count, last, offset;
+       long pg_vec, len;
+       long long objid = 3;
+       int flags = 0;
+       int cmd = 0;
+       char *end;
         int rc;
 
-        if (argc < 4 || argc > 5) {
-                fprintf(stderr,
-                       "usage: %s file pages_per_vec count [objid]\n",
-                       argv[0]);
-                return 1;
-        }
+        if (argc < 3 || argc > 6)
+               usage(argv[0]);
 
-        pg_vec = strtoul(argv[2], 0, 0);
-        count = strtoul(argv[3], 0, 0);
-       len = pg_vec * BLOCKSIZE;
-       end = (long long)count * len;
+        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)
-               objid = strtoul(argv[4], 0, 0);
+       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]);
+               }
+       }
+       len = pg_vec * BLOCKSIZE;
+       last = (long long)count * len;
+
+       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("directio on %s(%ld) for %ldx%ld pages \n",
+        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);
 
         buf = mmap(0, len, PROT_READ|PROT_WRITE, MAP_PRIVATE|MAP_ANON, 0, 0);
         if (!buf) {
-                fprintf(stderr, "No memory %s\n", strerror(errno));
+                fprintf(stderr, "%s: no buffer memory %s\n",
+                       argv[0], strerror(errno));
                 return 2;
         }
 
-        fd = open(argv[1], O_DIRECT | O_RDWR | O_CREAT);
+        fd = open(argv[1], flags | O_LARGEFILE);
         if (fd == -1) {
-                fprintf(stderr, "Cannot open %s:  %s\n", argv[1],
-                       strerror(errno));
+                fprintf(stderr, "%s: cannot open %s:  %s\n", argv[0],
+                       argv[1], strerror(errno));
                 return 3;
         }
 
-       for (offset = 0; offset < end; offset += len) {
+       for (offset = 0; offset < last && cmd & WRITE; offset += len) {
                int i;
 
                for (i = 0; i < len; i += BLOCKSIZE)
@@ -129,24 +176,25 @@ int main(int argc, char **argv)
                }
 
                if (rc != len) {
-                       fprintf(stderr, "Write error: %s, rc %d\n",
-                               strerror(errno), rc);
+                       fprintf(stderr, "%s: write error: %s, rc %d\n",
+                               argv[0], strerror(errno), rc);
                        return 4;
                }
        }
 
-        if ( lseek(fd, 0, SEEK_SET) != 0 ) {
-                fprintf(stderr, "Cannot seek %s\n", strerror(errno));
+        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\n",
+                               argv[0], strerror(errno), rc);
                        return 6;
                }