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>
51 #include <semaphore.h>
52 #include <libcfs/libcfs.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;
64 "Usage: %s filename command-sequence\n"
65 " command-sequence items:\n"
67 " C[num] create with optional stripes\n"
69 " D open(O_DIRECTORY)\n"
74 " M rw mmap to EOF (must open and stat prior)\n"
77 " O open(O_CREAT|O_RDWR)\n"
78 " r[num] read [optional length]\n"
79 " R reference entire mmap-ed region\n"
83 " T[num] ftruncate [optional position, default 0]\n"
87 " w[num] write optional length\n"
88 " W write entire mmap-ed region\n"
91 " z[num] seek [optional position, default 0]\n"
92 " _ wait for signal\n";
94 void usr1_handler(int unused)
96 int saved_errno = errno;
99 * signal(7): POSIX.1-2004 ...requires an implementation to guarantee
100 * that the following functions can be safely called inside a signal
110 pop_arg(int argc, char *argv[])
112 static int cur_arg = 3;
117 return argv[cur_arg++];
120 struct flag_mapping {
124 {"O_RDONLY", O_RDONLY},
125 {"O_WRONLY", O_WRONLY},
127 {"O_CREAT", O_CREAT},
129 {"O_NOCTTY", O_NOCTTY},
130 {"O_TRUNC", O_TRUNC},
131 {"O_APPEND", O_APPEND},
132 {"O_NONBLOCK", O_NONBLOCK},
133 {"O_NDELAY", O_NDELAY},
136 {"O_DIRECT", O_DIRECT},
138 {"O_LARGEFILE", O_LARGEFILE},
139 {"O_DIRECTORY", O_DIRECTORY},
140 {"O_NOFOLLOW", O_NOFOLLOW},
144 int get_flags(char *data, int *rflags)
152 cloned_flags = strdup(data);
153 if (cloned_flags == NULL) {
154 fprintf(stderr, "Insufficient memory.\n");
158 for (tmp = strtok(cloned_flags, ":"); tmp;
159 tmp = strtok(NULL, ":")) {
162 size = tmp - cloned_flags;
163 for (i = 0; flag_table[i].flag != -1; i++) {
164 if (!strcmp(tmp, flag_table[i].string)){
165 flags |= flag_table[i].flag;
166 size += strlen(flag_table[i].string);
183 #define POP_ARG() (pop_arg(argc, argv))
184 #define min(a,b) ((a)>(b)?(b):(a))
186 int main(int argc, char **argv)
188 char *fname, *commands;
192 size_t mmap_len = 0, i;
193 unsigned char *mmap_ptr = NULL, junk = 0;
194 int rc, len, fd = -1;
200 fprintf(stderr, usage, argv[0]);
204 memset(&st, 0, sizeof(st));
205 sem_init(&sem, 0, 0);
206 /* use sigaction instead of signal to avoid SA_ONESHOT semantics */
207 sigaction(SIGUSR1, &(const struct sigaction){.sa_handler = &usr1_handler},
212 for (commands = argv[2]; *commands; commands++) {
219 while (sem_wait(&sem) == -1 && errno == EINTR);
222 if (close(fd) == -1) {
230 len = atoi(commands+1);
231 fd = llapi_file_open(fname, O_CREAT | O_WRONLY, 0644,
235 perror("create stripe file");
240 if (mkdir(fname, 0755) == -1) {
242 perror("mkdir(0755)");
247 fd = open(fname, O_DIRECTORY);
250 perror("open(O_DIRECTORY)");
255 if (statfs(fname, &stfs) == -1) {
265 if (symlink(fname, newfile)) {
275 if (link(fname, newfile)) {
282 if (mknod(fname, S_IFREG | 0644, 0) == -1) {
284 perror("mknod(S_IFREG|0644, 0)");
289 if (st.st_size == 0) {
290 fprintf(stderr, "mmap without preceeding stat, or on"
291 " zero length file.\n");
294 mmap_len = st.st_size;
295 mmap_ptr = mmap(NULL, mmap_len, PROT_WRITE | PROT_READ,
297 if (mmap_ptr == MAP_FAILED) {
307 if (rename (fname, newfile)) {
314 fd = open(fname, O_CREAT|O_RDWR, 0644);
317 perror("open(O_RDWR|O_CREAT)");
322 len = get_flags(commands+1, &flags);
325 fd = open(fname, flags, 0666);
327 fd = open(fname, flags);
335 len = atoi(commands+1);
339 buf = realloc(buf, len + ALIGN);
342 perror("allocating buf for read\n");
346 buf_align = (char *)((long)(buf + ALIGN) &
350 rc = read(fd, buf_align, len);
357 fprintf(stderr, "short read: %u/%u\n",
361 printf("%.*s\n", rc, buf_align);
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)) {
416 len = atoi(commands+1);
420 buf = realloc(buf, len + ALIGN);
423 perror("allocating buf for write\n");
427 buf_align = (char *)((long)(buf + ALIGN) &
429 strncpy(buf_align, msg, bufsize);
432 rc = write(fd, buf_align, len);
439 fprintf(stderr, "short write: %u/%u\n",
445 for (i = 0; i < mmap_len && mmap_ptr; i += 4096)
446 mmap_ptr[i] += junk++;
449 if (fsync(fd) == -1) {
456 if (fdatasync(fd) == -1) {
462 len = atoi(commands+1);
463 if (lseek(fd, len, SEEK_SET) == -1) {
481 fprintf(stderr, "unknown command \"%c\"\n", *commands);
482 fprintf(stderr, usage, argv[0]);