Whamcloud - gitweb
LU-11388 test: disable replay-single test_131b
[fs/lustre-release.git] / lustre / tests / write_time_limit.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) 2007, 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  * Lustre is a trademark of Sun Microsystems, Inc.
29  */
30
31 /* for O_DIRECT */
32 #ifndef _GNU_SOURCE
33 #define _GNU_SOURCE
34 #endif
35
36 #include<stdio.h>
37 #include<errno.h>
38 #include<fcntl.h>
39 #include<sys/types.h>
40 #include<sys/stat.h>
41 #include<unistd.h>
42 #include<signal.h>
43 #include<stdlib.h>
44 #include<string.h>
45
46 #define BUFFERSIZE 4096
47
48 /* This flag controls termination of the main loop. */
49 volatile sig_atomic_t keep_going = 1;
50
51 /* The signal handler just clears the flag and re-enables itself. */
52 void catch_alarm(int sig)
53 {
54         keep_going = 0;
55         signal(sig, catch_alarm);
56 }
57
58 int main(int argc, char **argv)
59 {
60         char *file;
61         unsigned char buf[BUFFERSIZE];
62         int fd, i, rc;
63         unsigned int test_time;
64         mode_t mode = S_IRUSR | S_IWUSR | S_IRGRP | S_IWGRP | S_IROTH;
65
66         if (argc != 3) {
67                 printf("Invalid number of arguments.\n");
68                 printf("Usage %s <file> <test_time_in_seconds>\n", argv[0]);
69                 return -1;
70         }
71
72         file = argv[1];
73         test_time = atoi(argv[2]);
74
75         /* Establish a handler for SIGALRM signals. */
76         signal(SIGALRM, catch_alarm);
77
78         /* Set an alarm to go off in a little while. */
79         alarm(test_time);
80
81         fd = open(file, O_RDWR|O_TRUNC|O_CREAT|O_SYNC|O_LARGEFILE, mode);
82         if (fd < 0) {
83                 fprintf(stderr, "Error: Cannot open file named ");
84                 perror(file);
85                 return -1;
86         }
87
88         memset(buf, 1, BUFFERSIZE);
89         while (keep_going) {
90                 for (i = 0; i < 1024; i++) {
91                         rc = write(fd, buf, BUFFERSIZE);
92                         if (rc < 0) {
93                                 fprintf(stderr, "Error: Write error ");
94                                 perror(file);
95                                 exit(-1);
96                         } else if (rc != BUFFERSIZE) {
97                                 fprintf(stderr, "Error: Ddidn't write all data \n");
98                         }
99                 }
100
101                 if (ftruncate(fd, 0) < 0) {
102                         fprintf(stderr, "Error: Truncate error ");
103                         perror(file);
104                         exit(-1);
105                 }
106         }
107
108         if (close(fd) < 0) {
109                 fprintf(stderr, "Error: Cannot close ");
110                 perror(file);
111                 return -1;
112         }
113
114         return 0;
115 }