- int fd;
- char *buf;
- long long count, last, offset;
- long pg_vec, len;
- long long objid;
- 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')) {
- 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]);
- }
- } 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 "LPU64"x%ld pages \n",
- argv[0], flags & O_DIRECT ? "directio" : "i/o",
- 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));
- 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;
- }
-
- 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 == 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) {
- int i;
-
- for (i = 0; i < len; i += st.st_blksize)
- page_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))
- return 10;
- }
-
- 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));
- 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;
- }
- }
-
- return 0;
+ 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]);
+#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 %"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],
+ 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;
+ }
+
+ 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 == 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 += 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 (lseek(fd, 0, SEEK_SET) != 0) {
+ fprintf(stderr, "%s: cannot seek %s\n",
+ 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 (block_debug_check("read", buf + i, st.st_blksize,
+ offset + i, objid))
+ return 11;
+ }
+ }
+
+ return 0;