1 /* vim:expandtab:shiftwidth=2:tabstop=2:smarttab:
5 * Copyright (C) 2011 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.
42 #include <libtest/server.h>
43 #include <libtest/killpid.h>
46 std::ostream
& operator<<(std::ostream
& output
, const server_st
&arg
)
50 output
<< arg
.hostname();
54 output
<< arg
.hostname() << ":" << arg
.port();
56 return output
; // for multiple << operators
59 static void global_sleep(void)
61 static struct timespec global_sleep_value
= { 0, 50000 };
66 nanosleep(&global_sleep_value
, NULL
);
70 server_st::server_st(in_port_t port_arg
, test_server_getpid
*get_pid_arg
, test_server_ping
*ping_arg
) :
74 __get_pid(get_pid_arg
),
76 _hostname("localhost")
81 server_st::server_st(const std::string
&socket_file
, test_server_getpid
*get_pid_arg
, test_server_ping
*ping_arg
) :
85 __get_pid(get_pid_arg
),
87 _hostname(socket_file
)
93 server_st::~server_st()
101 bool server_st::start()
103 assert(not _command
.empty());
104 assert(not has_pid());
109 if (system(_command
.c_str()) == -1)
113 while (not ping() and --count
)
128 void server_st::reset_pid()
134 pid_t
server_st::pid()
145 bool server_st::kill()
150 if ((_pid
= get_pid()))
155 unlink(pid_file
); // If this happens we may be dealing with a dead server that left its pid file.
162 else if (pid_file
[0])
174 void server_startup_st::push_server(server_st
*arg
)
176 servers
.push_back(arg
);
179 void server_startup_st::shutdown()
181 for (std::vector
<server_st
*>::iterator iter
= servers
.begin(); iter
!= servers
.end(); iter
++)
183 if ((*iter
)->is_used())
190 server_startup_st::~server_startup_st()
192 for (std::vector
<server_st
*>::iterator iter
= servers
.begin(); iter
!= servers
.end(); iter
++)