+/* -*- mode: c; c-basic-offset: 8; indent-tabs-mode: nil; -*-
+ * vim:expandtab:shiftwidth=8:tabstop=8:
+ */
+
+/* for O_DIRECT */
+#define _GNU_SOURCE
+
#include <stdlib.h>
#include <stdio.h>
#include <unistd.h>
#include <errno.h>
#include <string.h>
#include <sys/wait.h>
+#include <sys/ioctl.h>
-#ifndef O_DIRECT
-#define O_DIRECT 040000 /* direct disk access hint */
-#endif
+#include <lustre/lustre_user.h>
int main(int argc, char *argv[])
{
char filename[1024];
unsigned long count, i;
- int thread = 0;
- int threads = 0;
- int rc;
- int fd;
+ int thread = 0;
+ int threads = 0;
+ int rc = 0;
+ int fd, ioctl_flags = 0;
if (argc < 3 || argc > 4) {
fprintf(stderr, "usage: %s <filename> <iterations> [threads]\n",
- argv[0]);
+ argv[0]);
exit(1);
}
count = strtoul(argv[2], NULL, 0);
- if (argc == 4)
- threads = strtoul(argv[3], NULL, 0);
+ if (argc == 4)
+ threads = strtoul(argv[3], NULL, 0);
for (i = 1; i <= threads; i++) {
rc = fork();
rc = 0;
}
- if (threads && thread == 0) { /* parent process */
+ if (threads && thread == 0) { /* parent process */
int live_threads = threads;
while (live_threads > 0) {
pid_t ret;
ret = waitpid(0, &status, 0);
- if (ret == 0) {
+ if (ret == 0)
continue;
- }
if (ret < 0) {
fprintf(stderr, "error: %s: wait - %s\n",
argv[0], ret, err);
if (!rc)
rc = err;
-
- live_threads--;
}
+ live_threads--;
}
- } else {
- if (threads)
- sprintf(filename, "%s-%d", argv[1], thread);
- else
- strcpy(filename, argv[1]);
+ } else {
+ if (threads)
+ sprintf(filename, "%s-%d", argv[1], thread);
+ else
+ strcpy(filename, argv[1]);
- fd = open(filename, O_RDWR|O_CREAT, 0644);
- if (fd < 0) {
- fprintf(stderr, "open(%s, O_CREAT): %s\n", filename,
- strerror(errno));
- exit(errno);
- }
- if (close(fd) < 0) {
- fprintf(stderr, "close(): %s\n", strerror(errno));
- rc = errno;
- goto unlink;
- }
+ fd = open(filename, O_RDWR|O_CREAT, 0644);
+ if (fd < 0) {
+ fprintf(stderr, "open(%s, O_CREAT): %s\n", filename,
+ strerror(errno));
+ exit(errno);
+ }
+ if (close(fd) < 0) {
+ fprintf(stderr, "close(): %s\n", strerror(errno));
+ rc = errno;
+ goto unlink;
+ }
- for (i = 0; i < count; i++) {
- fd = open(filename, O_RDWR|O_LARGEFILE|O_DIRECT);
- if (fd < 0) {
- fprintf(stderr, "open(%s, O_RDWR): %s\n",
- filename, strerror(errno));
- rc = errno;
- break;
- }
- if (close(fd) < 0) {
- fprintf(stderr, "close(): %s\n",
- strerror(errno));
- rc = errno;
- break;
- }
- }
- unlink:
- if (unlink(filename) < 0) {
- fprintf(stderr, "unlink(%s): %s\n", filename,
- strerror(errno));
- rc = errno;
- }
- if (threads)
- printf("Thread %d done: rc = %d\n", thread, rc);
- else
- printf("Done: rc = %d\n", rc);
- }
+ for (i = 0; i < count; i++) {
+ fd = open(filename, O_RDWR|O_LARGEFILE|O_DIRECT);
+ if (fd < 0) {
+ fprintf(stderr, "open(%s, O_RDWR): %s\n",
+ filename, strerror(errno));
+ rc = errno;
+ break;
+ }
+ if (ioctl(fd, LL_IOC_SETFLAGS, &ioctl_flags) < 0 &&
+ errno != ENOTTY) {
+ fprintf(stderr, "ioctl(): %s\n",
+ strerror(errno));
+ rc = errno;
+ break;
+ }
+ if (close(fd) < 0) {
+ fprintf(stderr, "close(): %s\n",
+ strerror(errno));
+ rc = errno;
+ break;
+ }
+ }
+ unlink:
+ if (unlink(filename) < 0) {
+ fprintf(stderr, "unlink(%s): %s\n", filename,
+ strerror(errno));
+ rc = errno;
+ }
+ if (threads)
+ printf("Thread %d done: rc = %d\n", thread, rc);
+ else
+ printf("Done: rc = %d\n", rc);
+ }
return rc;
}