1 /* -*- mode: c; c-basic-offset: 8; indent-tabs-mode: nil; -*-
2 * vim:expandtab:shiftwidth=8:tabstop=8:
6 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
8 * This program is free software; you can redistribute it and/or modify
9 * it under the terms of the GNU General Public License version 2 only,
10 * as published by the Free Software Foundation.
12 * This program is distributed in the hope that it will be useful, but
13 * WITHOUT ANY WARRANTY; without even the implied warranty of
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15 * General Public License version 2 for more details (a copy is included
16 * in the LICENSE file that accompanied this code).
18 * You should have received a copy of the GNU General Public License
19 * version 2 along with this program; If not, see
20 * http://www.sun.com/software/products/lustre/docs/GPLv2.pdf
22 * Please contact Sun Microsystems, Inc., 4150 Network Circle, Santa Clara,
23 * CA 95054 USA or visit www.sun.com if you need additional information or
29 * Copyright 2008 Sun Microsystems, Inc. All rights reserved
30 * Use is subject to license terms.
33 * This file is part of Lustre, http://www.lustre.org/
34 * Lustre is a trademark of Sun Microsystems, Inc.
38 #define _GNU_SOURCE /* pull in O_DIRECTORY in bits/fcntl.h */
44 #include <sys/types.h>
52 #include <semaphore.h>
53 #include <lustre/liblustreapi.h>
55 #define T1 "write data before unlink\n"
56 #define T2 "write data after unlink\n"
57 char msg[] = "yabba dabba doo, I'm coming for you, I live in a shoe, I don't know what to do.\n'Bigger, bigger,and bigger yet!' cried the Creator. 'You are not yet substantial enough for my boundless intents!' And ever greater and greater the object became, until all was lost 'neath its momentus bulk.\n";
58 char *buf, *buf_align;
62 #define MAX_SHIFT 4096
65 "Usage: %s filename command-sequence\n"
66 " command-sequence items:\n"
67 " b[num] write optional length from random in-memory offset\n"
69 " C[num] create with optional stripes\n"
71 " D open(O_DIRECTORY)\n"
76 " M rw mmap to EOF (must open and stat prior)\n"
79 " O open(O_CREAT|O_RDWR)\n"
80 " r[num] read [optional length]\n"
81 " R reference entire mmap-ed region\n"
85 " T[num] ftruncate [optional position, default 0]\n"
88 " w[num] write optional length\n"
89 " W write entire mmap-ed region\n"
92 " z[num] seek [optional position, default 0]\n"
93 " _ wait for signal\n";
95 void usr1_handler(int unused)
97 int saved_errno = errno;
100 * signal(7): POSIX.1-2004 ...requires an implementation to guarantee
101 * that the following functions can be safely called inside a signal
111 pop_arg(int argc, char *argv[])
113 static int cur_arg = 3;
118 return argv[cur_arg++];
121 struct flag_mapping {
125 {"O_RDONLY", O_RDONLY},
126 {"O_WRONLY", O_WRONLY},
128 {"O_CREAT", O_CREAT},
130 {"O_NOCTTY", O_NOCTTY},
131 {"O_TRUNC", O_TRUNC},
132 {"O_APPEND", O_APPEND},
133 {"O_NONBLOCK", O_NONBLOCK},
134 {"O_NDELAY", O_NDELAY},
137 {"O_DIRECT", O_DIRECT},
139 {"O_LARGEFILE", O_LARGEFILE},
140 {"O_DIRECTORY", O_DIRECTORY},
141 {"O_NOFOLLOW", O_NOFOLLOW},
145 int get_flags(char *data, int *rflags)
153 cloned_flags = strdup(data);
154 if (cloned_flags == NULL) {
155 fprintf(stderr, "Insufficient memory.\n");
159 for (tmp = strtok(cloned_flags, ":"); tmp;
160 tmp = strtok(NULL, ":")) {
163 size = tmp - cloned_flags;
164 for (i = 0; flag_table[i].flag != -1; i++) {
165 if (!strcmp(tmp, flag_table[i].string)){
166 flags |= flag_table[i].flag;
167 size += strlen(flag_table[i].string);
184 #define POP_ARG() (pop_arg(argc, argv))
185 #define min(a,b) ((a)>(b)?(b):(a))
187 int main(int argc, char **argv)
189 char *fname, *commands;
193 size_t mmap_len = 0, i;
194 unsigned char *mmap_ptr = NULL, junk = 0;
195 int rc, len, fd = -1;
201 fprintf(stderr, usage, argv[0]);
205 memset(&st, 0, sizeof(st));
206 sem_init(&sem, 0, 0);
207 /* use sigaction instead of signal to avoid SA_ONESHOT semantics */
208 sigaction(SIGUSR1, &(const struct sigaction){.sa_handler = &usr1_handler},
214 for (commands = argv[2]; *commands; commands++) {
221 while (sem_wait(&sem) == -1 && errno == EINTR);
224 if (close(fd) == -1) {
232 len = atoi(commands+1);
233 fd = llapi_file_open(fname, O_CREAT | O_WRONLY, 0644,
237 perror("create stripe file");
242 if (mkdir(fname, 0755) == -1) {
244 perror("mkdir(0755)");
249 fd = open(fname, O_DIRECTORY);
252 perror("open(O_DIRECTORY)");
257 if (statfs(fname, &stfs) == -1) {
267 if (symlink(fname, newfile)) {
277 if (link(fname, newfile)) {
284 if (mknod(fname, S_IFREG | 0644, 0) == -1) {
286 perror("mknod(S_IFREG|0644, 0)");
291 if (st.st_size == 0) {
292 fprintf(stderr, "mmap without preceeding stat, or on"
293 " zero length file.\n");
296 mmap_len = st.st_size;
297 mmap_ptr = mmap(NULL, mmap_len, PROT_WRITE | PROT_READ,
299 if (mmap_ptr == MAP_FAILED) {
309 if (rename (fname, newfile)) {
316 fd = open(fname, O_CREAT|O_RDWR, 0644);
319 perror("open(O_RDWR|O_CREAT)");
324 len = get_flags(commands+1, &flags);
327 fd = open(fname, flags, 0666);
329 fd = open(fname, flags);
337 len = atoi(commands+1);
341 buf = realloc(buf, len + ALIGN);
344 perror("allocating buf for read\n");
348 buf_align = (char *)((long)(buf + ALIGN) &
352 rc = read(fd, buf_align, len);
359 fprintf(stderr, "short read: %u/%u\n",
365 for (i = 0; i < mmap_len && mmap_ptr; i += 4096)
369 if (stat(fname, &st) == -1) {
376 if (fstat(fd, &st) == -1) {
383 if (fchmod(fd, 0) == -1) {
390 len = atoi(commands+1);
391 if (ftruncate(fd, len) == -1) {
393 printf("ftruncate (%d,%d)\n", fd, len);
399 if (unlink(fname) == -1) {
406 if (munmap(mmap_ptr, mmap_len)) {
417 len = atoi(commands+1);
421 int shift = (*commands == 'b') ?
422 (random() % MAX_SHIFT) : 0;
423 buf = realloc(buf, len + ALIGN + shift);
426 perror("allocating buf for write\n");
430 buf_align = (char *)((long)(buf + ALIGN) &
432 strncpy(buf_align, msg, bufsize);
435 rc = write(fd, buf_align, len);
442 fprintf(stderr, "short write: %u/%u\n",
448 for (i = 0; i < mmap_len && mmap_ptr; i += 4096)
449 mmap_ptr[i] += junk++;
452 if (fsync(fd) == -1) {
459 if (fdatasync(fd) == -1) {
465 len = atoi(commands+1);
466 if (lseek(fd, len, SEEK_SET) == -1) {
484 fprintf(stderr, "unknown command \"%c\"\n", *commands);
485 fprintf(stderr, usage, argv[0]);