1 /* vim:expandtab:shiftwidth=2:tabstop=2:smarttab:
3 * Data Differential YATL (i.e. libtest) library
5 * Copyright (C) 2012 Data Differential, http://datadifferential.com/
7 * Redistribution and use in source and binary forms, with or without
8 * modification, are permitted provided that the following conditions are
11 * * Redistributions of source code must retain the above copyright
12 * notice, this list of conditions and the following disclaimer.
14 * * Redistributions in binary form must reproduce the above
15 * copyright notice, this list of conditions and the following disclaimer
16 * in the documentation and/or other materials provided with the
19 * * The names of its contributors may not be used to endorse or
20 * promote products derived from this software without specific prior
23 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
24 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
25 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
26 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
27 * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
28 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
29 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
30 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
31 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
32 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
33 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
37 #include "mem_config.h"
38 #include <libtest/common.h>
45 #include <sys/types.h>
46 #include <sys/types.h>
50 #include <libtest/killpid.h>
51 #include <libtest/stream.h>
53 using namespace libtest
;
55 bool kill_pid(pid_t pid_arg
)
60 Error
<< "Invalid pid:" << pid_arg
;
64 if ((::kill(pid_arg
, SIGTERM
) == -1))
69 Error
<< "Does someone else have a process running locally for " << int(pid_arg
) << "?";
73 Error
<< "Process " << int(pid_arg
) << " not found.";
78 Error
<< "kill() " << strerror(errno
);
84 uint32_t this_wait
= 0;
85 uint32_t timeout
= 20; // This number should be high enough for valgrind startup (which is slow)
89 for (waited
= 0, retry
= 4; ; retry
++, waited
+= this_wait
)
92 if (waitpid(pid_arg
, &status
, WNOHANG
) == 0)
96 else if (errno
== ECHILD
)
98 // Server has already gone away
101 else if (waited
>= timeout
)
104 kill(pid_arg
, SIGKILL
);
108 this_wait
= retry
* retry
/ 3 + 1;
109 libtest::dream(this_wait
, 0);
116 bool check_pid(const std::string
&filename
)
118 if (filename
.empty())
124 if ((fp
= fopen(filename
.c_str(), "r")))
126 char pid_buffer
[1024];
128 char *ptr
= fgets(pid_buffer
, sizeof(pid_buffer
), fp
);
133 pid_t pid
= (pid_t
)atoi(pid_buffer
);
136 return (::kill(pid
, 0) == 0);
145 bool kill_file(const std::string
&filename
)
147 if (filename
.empty())
153 if ((fp
= fopen(filename
.c_str(), "r")))
155 char pid_buffer
[1024];
157 char *ptr
= fgets(pid_buffer
, sizeof(pid_buffer
), fp
);
162 pid_t pid
= (pid_t
)atoi(pid_buffer
);
165 bool ret
= kill_pid(pid
);
166 unlink(filename
.c_str()); // If this happens we may be dealing with a dead server that left its pid file.
176 #define STRINGIFY(x) #x
177 #define TOSTRING(x) STRINGIFY(x)
178 #define LIBTEST_AT __FILE__ ":" TOSTRING(__LINE__)
180 pid_t
get_pid_from_file(const std::string
&filename
, std::stringstream
& error_message
)
185 if (filename
.empty())
187 error_message
<< LIBTEST_AT
<< " empty pid file";
191 if ((fp
= fopen(filename
.c_str(), "r")))
193 char pid_buffer
[1024];
195 char *ptr
= fgets(pid_buffer
, sizeof(pid_buffer
), fp
);
200 ret
= (pid_t
)atoi(pid_buffer
);
203 error_message
<< LIBTEST_AT
<< " Invalid pid was read from file " << filename
;
208 error_message
<< LIBTEST_AT
<< " File " << filename
<< " was empty ";
216 char *current_directory
= getcwd(buffer
, sizeof(buffer
));
217 error_message
<< "Error while opening " << current_directory
<< "/" << filename
<< " " << strerror(errno
);