1 /* vim:expandtab:shiftwidth=2:tabstop=2:smarttab:
5 * Copyright (C) 2011 Data Differential, http://datadifferential.com/
6 * Copyright (C) 2006-2009 Brian Aker All rights reserved.
8 * Redistribution and use in source and binary forms, with or without
9 * modification, are permitted provided that the following conditions are
12 * * Redistributions of source code must retain the above copyright
13 * notice, this list of conditions and the following disclaimer.
15 * * Redistributions in binary form must reproduce the above
16 * copyright notice, this list of conditions and the following disclaimer
17 * in the documentation and/or other materials provided with the
20 * * The names of its contributors may not be used to endorse or
21 * promote products derived from this software without specific prior
24 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
25 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
26 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
27 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
28 * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
29 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
30 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
31 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
32 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
33 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
34 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
40 Startup, and shutdown the memcached servers.
43 #define TEST_PORT_BASE MEMCACHED_DEFAULT_PORT+10
45 #include <libtest/common.h>
59 #include <libmemcached/memcached.h>
60 #include <libmemcached/util.h>
62 #include <libtest/server.h>
63 #include <libtest/killpid.h>
64 #include <libtest/wait.h>
66 #define CERR_PREFIX std::endl << __FILE__ << ":" << __LINE__ << " "
68 #define SOCKET_FILE "/tmp/memcached.socket"
70 static pid_t
__getpid(server_st
& server
)
72 memcached_return_t rc
;
73 pid_t pid
= libmemcached_util_getpid(server
.hostname(), server
.port(), &rc
);
77 static bool __ping(server_st
& server
)
79 memcached_return_t rc
;
80 bool ret
= libmemcached_util_ping(server
.hostname(), server
.port(), &rc
);
84 static bool cycle_server(server_st
*server
)
88 if (libmemcached_util_ping(server
->hostname(), server
->port(), NULL
))
90 // First we try to kill it, and on fail of that we flush it.
91 pid_t pid
= libmemcached_util_getpid(server
->hostname(), server
->port(), NULL
);
93 if (pid
> 0 and kill_pid(pid
))
95 std::cerr
<< CERR_PREFIX
<< "Killed existing server," << *server
<< " with pid:" << pid
<< std::endl
;
98 else if (libmemcached_util_flush(server
->hostname(), server
->port(), NULL
)) // If we can flush it, we will just use it
100 std::cerr
<< CERR_PREFIX
<< "Found server on port " << int(server
->port()) << ", flushed it!" << std::endl
;
103 } // No idea what is wrong here, so we need to find a different port
116 bool server_startup(server_startup_st
*construct
)
118 if (getenv(((char *)"MEMCACHED_SERVERS")))
120 construct
->server_list
= getenv(((char *)"MEMCACHED_SERVERS"));
121 printf("servers %s\n", construct
->server_list
.c_str());
126 std::string server_config_string
;
128 uint32_t port_base
= 0;
129 for (uint32_t x
= 0; x
< (construct
->count
-1); x
++)
131 server_st
*server
= NULL
;
135 char variable_buffer
[1024];
137 snprintf(variable_buffer
, sizeof(variable_buffer
), "LIBMEMCACHED_PORT_%u", x
);
139 if ((var
= getenv(variable_buffer
)))
141 server
= new server_st((in_port_t
)atoi(var
), __getpid
, __ping
);
145 server
= new server_st(in_port_t(x
+TEST_PORT_BASE
+port_base
), __getpid
, __ping
);
147 while (not cycle_server(server
))
149 std::cerr
<< CERR_PREFIX
<< "Found server " << *server
<< ", could not flush it, so trying next port." << std::endl
;
151 server
->set_port(in_port_t(x
+TEST_PORT_BASE
+port_base
));
156 if (server
->is_used())
158 std::cerr
<< std::endl
<< "Using server at : " << server
<< std::endl
;
162 char buffer
[FILENAME_MAX
];
165 snprintf(buffer
, sizeof(buffer
), "%s -d -t 1 -p %u -U %u -m 128",
166 MEMCACHED_BINARY
, server
->port(), server
->port());
170 snprintf(buffer
, sizeof(buffer
), "%s -d -t 1 -p %u -U %u",
171 MEMCACHED_BINARY
, server
->port(), server
->port());
173 server
->set_command(buffer
);
175 if (not server
->start())
177 std::cerr
<< CERR_PREFIX
<< "Failed system(" << buffer
<< ")" << std::endl
;
181 std::cerr
<< "STARTING SERVER: " << buffer
<< " pid:" << server
->pid() << std::endl
;
183 construct
->push_server(server
);
187 assert(server
->has_port());
188 set_default_port(server
->port());
191 char port_str
[NI_MAXSERV
];
192 snprintf(port_str
, sizeof(port_str
), "%u", int(server
->port()));
194 server_config_string
+= "--server=";
195 server_config_string
+= server
->hostname();
196 server_config_string
+= ":";
197 server_config_string
+= port_str
;
198 server_config_string
+= " ";
204 std::string
socket_file(SOCKET_FILE
);
207 if ((var
= getenv("LIBMEMCACHED_SOCKET")))
212 server_st
*server
= new server_st(SOCKET_FILE
, __getpid
, __ping
);
214 if (not cycle_server(server
))
216 std::cerr
<< CERR_PREFIX
<< "Found server " << server
<< ", could not flush it, failing since socket file is not available." << std::endl
;
220 if (server
->is_used())
222 std::cerr
<< std::endl
<< "Using server at : " << *server
<< std::endl
;
226 char buffer
[FILENAME_MAX
];
227 snprintf(buffer
, sizeof(buffer
), "%s -d -t 1 -s %s", MEMCACHED_BINARY
, SOCKET_FILE
);
228 server
->set_command(buffer
);
230 if (not server
->start())
232 std::cerr
<< CERR_PREFIX
<< "Failed system(" << buffer
<< ")" << std::endl
;
236 std::cerr
<< "STARTING SERVER: " << buffer
<< " pid:" << server
->pid() << std::endl
;
238 set_default_socket(server
->hostname());
239 construct
->push_server(server
);
242 server_config_string
+= "--socket=\"";
243 server_config_string
+= server
->hostname();
244 server_config_string
+= "\" ";
248 server_config_string
.resize(server_config_string
.size() -1); // Remove final space
249 construct
->server_list
= server_config_string
;
252 srandom((unsigned int)time(NULL
));
254 std::cerr
<< std::endl
;
258 void server_shutdown(server_startup_st
*construct
)
263 construct
->shutdown();