Whamcloud - gitweb
LU-15103 tests: clean up busy cifs mount
[fs/lustre-release.git] / lustre / tests / mpi / createmany-mpi.c
1 /*
2  * GPL HEADER START
3  *
4  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
5  *
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.
9  *
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).
15  *
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.gnu.org/licenses/gpl-2.0.html
19  *
20  * GPL HEADER END
21  */
22 /*
23  * Copyright (c) 2003, 2010, Oracle and/or its affiliates. All rights reserved.
24  * Use is subject to license terms.
25  */
26 /*
27  * This file is part of Lustre, http://www.lustre.org/
28  */
29
30 #include <stdio.h>
31 #include <sys/types.h>
32 #include <sys/stat.h>
33 #include <time.h>
34 #include <errno.h>
35 #include <string.h>
36 #include <fcntl.h>
37 #include <unistd.h>
38 #include <stdlib.h>
39 #include <stdarg.h>
40
41 #include "mpi.h"
42
43 void usage(char *prog)
44 {
45         printf("usage: %s {-o|-m|-l<tgt>} filenamefmt count\n", prog);
46         printf("       %s {-o|-m|-l<tgt>} filenamefmt -seconds\n", prog);
47         printf("       %s {-o|-m|-l<tgt>} filenamefmt start count\n", prog);
48 }
49
50 /* Print process rank, loop count, message, and exit (i.e. a fatal error) */
51 int rprintf(int rank, int loop, const char *fmt, ...)
52 {
53         va_list ap;
54
55         printf("rank %d, loop %d: ", rank, loop);
56
57         va_start(ap, fmt);
58
59         printf(fmt, ap);
60
61         MPI_Finalize();
62         exit(1);
63 }
64
65 int main(int argc, char **argv)
66 {
67         int i, rc = 0, do_open = 0, do_link = 0, rank;
68         char format[4096], *fmt, *tgt = NULL;
69         char filename[4096];
70         long start, last, end;
71         long begin = 0, count;
72
73         rc = MPI_Init(&argc, &argv);
74         if (rc != MPI_SUCCESS)
75                 rprintf(-1, -1, "MPI_Init failed: %d\n", rc);
76
77         if (argc < 4 || argc > 5) {
78                 usage(argv[0]);
79                 return 1;
80         }
81
82         if (strcmp(argv[1], "-o") == 0) {
83                 do_open = 1;
84                 tgt = NULL;
85         } else if (strncmp(argv[1], "-l", 2) == 0 && argv[1][2]) {
86                 tgt = argv[1] + 2;
87                 do_link = 1;
88         } else if (strcmp(argv[1], "-m") != 0) {
89                 usage(argv[0]);
90                 return 1;
91         }
92
93         if (strlen(argv[2]) > 4080) {
94                 printf("name too long\n");
95                 return 1;
96         }
97
98         rc = MPI_Comm_rank(MPI_COMM_WORLD, &rank);
99         if (rc != MPI_SUCCESS)
100                 rprintf(-1, -1, "MPI_Comm_rank failed: %d\n", rc);
101
102         rc = MPI_Barrier(MPI_COMM_WORLD);
103         if (rc != MPI_SUCCESS)
104                 rprintf(rank, -1, "prep MPI_Barrier failed: %d\n", rc);
105
106         start = last = time(0);
107
108         if (argc == 4) {
109                 end = strtol(argv[3], NULL, 0);
110         } else {
111                 begin = strtol(argv[3], NULL, 0);
112                 end = strtol(argv[4], NULL, 0);
113         }
114         if (end > 0) {
115                 count = end;
116                 end = -1UL >> 1;
117         } else {
118                 end = start - end;
119                 count = -1UL >> 1;
120         }
121
122         if (strchr(argv[2], '%')) {
123                 fmt = argv[2];
124         } else {
125                 sprintf(format, "%s%%d", argv[2]);
126                 fmt = format;
127         }
128         printf("starting at %s", ctime(&start));
129         for (i = 0; i < count && time(0) < end; i++, begin++) {
130                 sprintf(filename, fmt, begin);
131                 if (do_open) {
132                         int fd = open(filename, O_CREAT | O_RDWR, 0644);
133
134                         if (fd < 0) {
135                                 printf("open(%s) error: %s\n", filename,
136                                        strerror(errno));
137                                 rc = errno;
138                                 break;
139                         }
140                         close(fd);
141                 } else if (do_link) {
142                         rc = link(tgt, filename);
143                         if (rc) {
144                                 printf("link(%s, %s) error: %s\n", tgt,
145                                        filename, strerror(errno));
146                                 rc = errno;
147                                 break;
148                         }
149                 } else {
150                         rc = mknod(filename, S_IFREG | 0444, 0);
151                         if (rc) {
152                                 printf("mknod(%s) error: %s\n", filename,
153                                        strerror(errno));
154                                 rc = errno;
155                                 break;
156                         }
157                 }
158                 if ((i % 10000) == 0) {
159                         printf(" - created %d (time %ld total %ld last %ld)\n",
160                                i, time(0), time(0) - start, time(0) - last);
161                         last = time(0);
162                 }
163         }
164         printf("total: %d creates in %ld seconds: %f creates/second\n", i,
165                time(0) - start, ((float)i / (time(0) - start)));
166         start = time(0);
167         printf("finish at %s", ctime(&start));
168
169         return rc;
170 }