4 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
6 * This program is free software; you can redistribute it and/or modify
7 * it under the terms of the GNU General Public License version 2 only,
8 * as published by the Free Software Foundation.
10 * This program is distributed in the hope that it will be useful, but
11 * WITHOUT ANY WARRANTY; without even the implied warranty of
12 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
13 * General Public License version 2 for more details (a copy is included
14 * in the LICENSE file that accompanied this code).
16 * You should have received a copy of the GNU General Public License
17 * version 2 along with this program; If not, see
18 * http://www.sun.com/software/products/lustre/docs/GPLv2.pdf
20 * Please contact Sun Microsystems, Inc., 4150 Network Circle, Santa Clara,
21 * CA 95054 USA or visit www.sun.com if you need additional information or
27 * Copyright (c) 2003, 2010, Oracle and/or its affiliates. All rights reserved.
28 * Use is subject to license terms.
30 * Copyright (c) 2012, Intel Corporation.
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>
48 #include <sys/ioctl.h>
52 #include <semaphore.h>
54 #include <lustre/lustreapi.h>
56 #define T1 "write data before unlink\n"
57 #define T2 "write data after unlink\n"
58 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";
59 char *buf, *buf_align;
65 "Usage: %s filename command-sequence\n"
66 " command-sequence items:\n"
68 " C[num] create with optional stripes\n"
70 " D open(O_DIRECTORY)\n"
72 " G gid get grouplock\n"
73 " g gid put grouplock\n"
77 " M rw mmap to EOF (must open and stat prior)\n"
80 " O open(O_CREAT|O_RDWR)\n"
81 " r[num] read [optional length]\n"
82 " R reference entire mmap-ed region\n"
86 " T[num] ftruncate [optional position, default 0]\n"
90 " w[num] write optional length\n"
91 " W write entire mmap-ed region\n"
94 " z[num] seek [optional position, default 0]\n"
95 " _ wait for signal\n";
97 void usr1_handler(int unused)
99 int saved_errno = errno;
102 * signal(7): POSIX.1-2004 ...requires an implementation to guarantee
103 * that the following functions can be safely called inside a signal
113 pop_arg(int argc, char *argv[])
115 static int cur_arg = 3;
120 return argv[cur_arg++];
123 struct flag_mapping {
127 {"O_RDONLY", O_RDONLY},
128 {"O_WRONLY", O_WRONLY},
130 {"O_CREAT", O_CREAT},
132 {"O_NOCTTY", O_NOCTTY},
133 {"O_TRUNC", O_TRUNC},
134 {"O_APPEND", O_APPEND},
135 {"O_NONBLOCK", O_NONBLOCK},
136 {"O_NDELAY", O_NDELAY},
139 {"O_DIRECT", O_DIRECT},
141 {"O_LARGEFILE", O_LARGEFILE},
142 {"O_DIRECTORY", O_DIRECTORY},
143 {"O_NOFOLLOW", O_NOFOLLOW},
147 int get_flags(char *data, int *rflags)
155 cloned_flags = strdup(data);
156 if (cloned_flags == NULL) {
157 fprintf(stderr, "Insufficient memory.\n");
161 for (tmp = strtok(cloned_flags, ":"); tmp;
162 tmp = strtok(NULL, ":")) {
165 size = tmp - cloned_flags;
166 for (i = 0; flag_table[i].flag != -1; i++) {
167 if (!strcmp(tmp, flag_table[i].string)){
168 flags |= flag_table[i].flag;
169 size += strlen(flag_table[i].string);
186 #define POP_ARG() (pop_arg(argc, argv))
188 int main(int argc, char **argv)
190 char *fname, *commands;
194 size_t mmap_len = 0, i;
195 unsigned char *mmap_ptr = NULL, junk = 0;
196 int rc, len, fd = -1;
203 fprintf(stderr, usage, argv[0]);
207 memset(&st, 0, sizeof(st));
208 sem_init(&sem, 0, 0);
209 /* use sigaction instead of signal to avoid SA_ONESHOT semantics */
210 sigaction(SIGUSR1, &(const struct sigaction){.sa_handler = &usr1_handler},
215 for (commands = argv[2]; *commands; commands++) {
222 while (sem_wait(&sem) == -1 && errno == EINTR);
225 if (close(fd) == -1) {
233 len = atoi(commands+1);
234 fd = llapi_file_open(fname, O_CREAT | O_WRONLY, 0644,
238 perror("create stripe file");
243 if (mkdir(fname, 0755) == -1) {
245 perror("mkdir(0755)");
250 fd = open(fname, O_DIRECTORY);
253 perror("open(O_DIRECTORY)");
258 if (statfs(fname, &stfs) == -1) {
265 gid = atoi(commands+1);
266 if (ioctl(fd, LL_IOC_GROUP_LOCK, gid) == -1) {
268 perror("ioctl(GROUP_LOCK)");
273 gid = atoi(commands+1);
274 if (ioctl(fd, LL_IOC_GROUP_UNLOCK, gid) == -1) {
276 perror("ioctl(GROUP_UNLOCK)");
284 if (symlink(fname, newfile)) {
294 if (link(fname, newfile)) {
301 if (mknod(fname, S_IFREG | 0644, 0) == -1) {
303 perror("mknod(S_IFREG|0644, 0)");
308 if (st.st_size == 0) {
309 fprintf(stderr, "mmap without preceeding stat, or on"
310 " zero length file.\n");
313 mmap_len = st.st_size;
314 mmap_ptr = mmap(NULL, mmap_len, PROT_WRITE | PROT_READ,
316 if (mmap_ptr == MAP_FAILED) {
326 if (rename (fname, newfile)) {
333 fd = open(fname, O_CREAT|O_RDWR, 0644);
336 perror("open(O_RDWR|O_CREAT)");
341 len = get_flags(commands+1, &flags);
344 fd = open(fname, flags, 0666);
346 fd = open(fname, flags);
354 len = atoi(commands+1);
358 buf = realloc(buf, len + ALIGN);
361 perror("allocating buf for read\n");
365 buf_align = (char *)((long)(buf + ALIGN) &
369 rc = read(fd, buf_align, len);
376 fprintf(stderr, "short read: %u/%u\n",
383 printf("%.*s\n", rc, buf_align);
387 for (i = 0; i < mmap_len && mmap_ptr; i += 4096)
391 if (stat(fname, &st) == -1) {
398 if (fstat(fd, &st) == -1) {
405 if (fchmod(fd, 0) == -1) {
412 len = atoi(commands+1);
413 if (ftruncate(fd, len) == -1) {
415 printf("ftruncate (%d,%d)\n", fd, len);
421 if (unlink(fname) == -1) {
428 if (munmap(mmap_ptr, mmap_len)) {
438 len = atoi(commands+1);
442 buf = realloc(buf, len + ALIGN);
445 perror("allocating buf for write\n");
449 buf_align = (char *)((long)(buf + ALIGN) &
451 strncpy(buf_align, msg, bufsize);
454 rc = write(fd, buf_align, len);
461 fprintf(stderr, "short write: %u/%u\n",
467 for (i = 0; i < mmap_len && mmap_ptr; i += 4096)
468 mmap_ptr[i] += junk++;
471 if (fsync(fd) == -1) {
478 if (fdatasync(fd) == -1) {
484 len = atoi(commands+1);
485 if (lseek(fd, len, SEEK_SET) == -1) {
503 fprintf(stderr, "unknown command \"%c\"\n", *commands);
504 fprintf(stderr, usage, argv[0]);