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) 2007, 2010, Oracle and/or its affiliates. All rights reserved.
28 * Use is subject to license terms.
31 * This file is part of Lustre, http://www.lustre.org/
32 * Lustre is a trademark of Sun Microsystems, Inc.
38 #include <sys/types.h>
49 #define difftime(a, b) \
50 ((double)(a)->tv_sec - (b)->tv_sec + \
51 ((double)((a)->tv_usec - (b)->tv_usec) / 1000000))
59 void usage(char *name)
61 fprintf(stderr, "usage: %s [opts] <dirname> <seconds> <threads>\n",
63 fprintf(stderr, " -q quiet\n");
64 fprintf(stderr, " -a abort other children on first err\n");
71 struct kid_list_t *next;
74 struct kid_list_t *head = NULL;
76 int push_kid(pid_t kid)
78 struct kid_list_t *new;
79 new = (struct kid_list_t *)malloc(sizeof(struct kid_list_t));
92 kill(head->kid, SIGTERM);
97 static int usr1_received;
98 void usr1_handler(int unused)
104 int wait_for_threads(int live_threads)
108 while (live_threads > 0) {
112 ret = waitpid(0, &status, 0);
118 fprintf(stderr, "%s: error: wait - %s\n",
119 cmdname, strerror(errno));
124 * This is a hack. We _should_ be able to use
125 * WIFEXITED(status) to see if there was an
126 * error, but it appears to be broken and it
127 * always returns 1 (OK). See wait(2).
129 int err = WEXITSTATUS(status);
132 "%s: error: PID %d had rc=%d\n",
134 /* Record first error */
138 /* Give up on first error */
148 printf("%s done, rc = %d\n", cmdname, rc);
152 void print_err(char *op, char *filename, struct timeval *time, int err)
154 fprintf(stderr, "%s: %d.%.06d error: %s(%s): %s\n",
155 cmdname, (int)(time->tv_sec), (int)(time->tv_usec), op,
156 filename, strerror(errno));
159 int run_one_child(char *file, int thread, int seconds)
161 struct timeval start, cur;
165 int fd, rc = 0, rand, maxrand, len;
166 long nfiles = 0, nbytes = 0;
169 printf("%s: running thread #%d\n", cmdname, thread);
172 /* Higher thread numbers will produce bigger random files.
173 Thread 1 will produce only 0-len files. */
174 maxrand = 1; rand = thread;
178 gettimeofday(&start, NULL);
185 gettimeofday(&cur, NULL);
187 if (cur.tv_sec > (start.tv_sec + seconds))
191 snprintf(filename, sizeof(filename), "%s-%d-%ld",
192 file, thread, nfiles);
194 fd = open(filename, O_RDWR | O_CREAT, 0666);
196 print_err("open", filename, &cur, errno);
201 sprintf(buf, "%s %010ld %.19s.%012d\n", cmdname,
202 nfiles++, ctime(&cur.tv_sec), (int)cur.tv_usec);
205 rand = random() % maxrand;
207 if (write(fd, buf, len) != len) {
208 print_err("write", filename, &cur, errno);
216 print_err("close", filename, &cur, errno);
220 if (unlink(filename) < 0) {
221 print_err("unlink", filename, &cur, errno);
222 if (errno == ENOENT) {
223 printf("Ignoring known bug 6082\n");
231 diff = difftime(&cur, &start);
233 printf("%s: %7ld files, %4ld MB in %.2fs (%7.2f files/s, "
234 "%5.2f MB/s): rc = %d\n",
235 cmdname, nfiles, nbytes >> 20, diff,
236 diff == 0 ? (double)0 : (double)nfiles / diff,
237 diff == 0 ? (double)0 : (double)nbytes/1024/1024 / diff,
243 int main(int argc, char *argv[])
245 unsigned long duration;
251 snprintf(cmdname, sizeof(cmdname), "%s", argv[0]);
253 while((i < argc) && (argv[i][0] == '-')) {
254 switch (argv[i][1]) {
269 duration = strtoul(argv[++i], &end, 0);
271 fprintf(stderr, "%s: error: bad number of seconds '%s'\n",
276 threads = strtoul(argv[++i], &end, 0);
278 fprintf(stderr, "%s: error: bad thread count '%s'\n",
283 signal(SIGUSR1, usr1_handler);
285 for (i = 1; i <= threads; i++) {
289 fprintf(stderr, "%s: error: #%d - %s\n",
290 cmdname, i, strerror(rc = errno));
295 snprintf(cmdname, sizeof(cmdname), "%s-%d", argv[0], i);
296 return (run_one_child(directory, i, duration));
308 printf("%s will run for %ld minutes\n", cmdname, duration/60);
309 return (wait_for_threads(threads));