1 /*
2 * Copyright (c) 2004, Bull S.A.. All rights reserved.
3 * Created by: Sebastien Decugis
4
5 * This program is free software; you can redistribute it and/or modify it
6 * under the terms of version 2 of the GNU General Public License as
7 * published by the Free Software Foundation.
8 *
9 * This program is distributed in the hope that it would be useful, but
10 * WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
12 *
13 * You should have received a copy of the GNU General Public License along
14 * with this program; if not, write the Free Software Foundation, Inc.,
15 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
16
17 */
18
19 /*
20 * The purpose of this file is to provide a monitor process
21 * for the stress/threads/ * / *.c testcases in the OPTS.
22 *
23 * The goal is:
24 * -> if the testcase returns, the monitor returns the error code.
25 * -> after a specified timeout, the monitor let the stress test terminate
26
27 * This allows for the stress tests to be run in an automatic maneer
28 * with a script such as:
29 #!/bin/sh
30
31 #monitor the system
32 vmstat -n 120 180 >monitor.txt 2>&1 &
33
34 #run the tests
35 for TS in `ls -1 *.c`;
36 do <compile $TS>;
37 if [ $? -eq 0 ];
38 then <run in background:>
39 helper 6 $TS.exe &
40 fi
41 done
42
43 #wait for the end
44 while [ "`ps -e | grep helper`" ];
45 do sleep 30;
46 done
47
48 */
49
50 #include <pthread.h>
51 #include <stdio.h>
52 #include <stdlib.h>
53 #include <unistd.h>
54 #include <signal.h>
55 #include <sys/wait.h>
56 #include <assert.h>
57 #include <time.h>
58
59 static pid_t child;
60 static int timeout;
61
62 /* Note that there could be a race between
63 the moment the stress test terminates and
64 when the timeout expires. As this is highly
65 improbable, we don't care... */
66
timer(void * arg)67 static void *timer(void *arg)
68 {
69 int ret = 0;
70
71 unsigned remaining = timeout * 3600;
72 do {
73 remaining = sleep(remaining);
74 } while (remaining);
75 ret = kill(child, SIGUSR1);
76 if (ret != 0) {
77 perror("Failed to kill the stress test");
78 exit(2);
79 }
80
81 return NULL;
82 }
83
main(int argc,char * argv[])84 int main(int argc, char *argv[])
85 {
86 int ret;
87 pthread_t th;
88 pid_t chk;
89 int status;
90 char *ts = "[??:??:??]";
91 struct tm *now;
92 time_t nw;
93
94 /* check args */
95 if (argc < 3) {
96 printf("\nUsage: \n");
97 printf(" $ %s n exe arglist\n", argv[0]);
98 printf("\nWhere:\n");
99 printf(" n is the timeout duration in hours,\n");
100 printf(" exe is the stress test executable to monitor,\n");
101 printf
102 (" arglist is the arguments to be passed to executable.\n\n");
103 return 2;
104 }
105
106 timeout = atoi(argv[1]);
107 if (timeout < 1) {
108 fprintf(stderr,
109 "Invalid timeout value \"%s\". Timeout must be a positive integer.\n",
110 argv[1]);
111 return 2;
112 }
113
114 /* create the timer thread */
115 ret = pthread_create(&th, NULL, timer, NULL);
116 if (ret != 0) {
117 perror("Failed to create the timeout thread\n");
118 return 2;
119 }
120
121 /* Create the new process for the stress test */
122 child = fork();
123
124 if (child == (pid_t) - 1) {
125 perror("Failed to create a new process");
126 exit(2);
127 }
128
129 /* The child process executes the test */
130 if (child == (pid_t) 0) {
131
132 /* Execute the command */
133 ret = execvp(argv[2], &argv[2]);
134 if (ret == -1) {
135 /* Application was not launched */
136 perror("Unable to run child application");
137 return 2;
138 }
139 assert(0);
140 perror("Should not see me");
141 return 2;
142 }
143
144 /* The parent: */
145
146 /* wait for the child process to terminate */
147 chk = waitpid(child, &status, 0);
148 if (chk != child) {
149 perror("Got the wrong process image status");
150 return 2;
151 }
152
153 /* Cancel the timer thread in case the process returned by itself */
154 (void)pthread_cancel(th);
155
156 ret = pthread_join(th, NULL);
157 if (ret != 0) {
158 perror("Unable to join the timer thread");
159 return 2;
160 }
161
162 /* return */
163 nw = time(NULL);
164 now = localtime(&nw);
165 if (now == NULL)
166 printf(ts);
167 else
168 printf("[%2.2d:%2.2d:%2.2d]\n", now->tm_hour, now->tm_min,
169 now->tm_sec);
170 if (!WIFEXITED(status)) {
171 printf("The stress sample did not exit\n");
172 if (WIFSIGNALED(status)) {
173 printf("It was killed with signal %i\n",
174 WTERMSIG(status));
175 } else {
176 printf("and it was not killed...\n");
177 }
178 exit(1);
179 }
180 if (WEXITSTATUS(status) == 0) {
181 printf("Test %s PASSED\n", argv[2]);
182 } else {
183 printf("Test %s: returned %d\n", argv[2], WEXITSTATUS(status));
184 }
185 exit(WEXITSTATUS(status));
186 }
187