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 #include <boost/lexical_cast.hpp>
69 #define CERR_PREFIX std::endl << __FILE__ << ":" << __LINE__ << " "
71 static void global_sleep(void)
73 static struct timespec global_sleep_value
= { 0, 50000 };
78 nanosleep(&global_sleep_value
, NULL
);
82 #define SOCKET_FILE "/tmp/memcached.socket"
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
= construct
->server
[x
];
135 char variable_buffer
[1024];
137 snprintf(variable_buffer
, sizeof(variable_buffer
), "LIBMEMCACHED_PORT_%u", x
);
139 if ((var
= getenv(variable_buffer
)))
141 server
.set_port((in_port_t
)atoi(var
));
145 server
.set_port(in_port_t(x
+ TEST_PORT_BASE
+ port_base
));
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());
174 int status
= system(buffer
);
177 std::cerr
<< CERR_PREFIX
<< "Failed system(" << buffer
<< ")" << std::endl
;
180 fprintf(stderr
, "STARTING SERVER: %s\n", buffer
);
183 memcached_return_t rc
;
184 while (not libmemcached_util_ping(server
.hostname
, server
.port(), &rc
) and --count
)
189 if (memcached_failed(rc
))
191 std::cerr
<< CERR_PREFIX
<< "libmemcached_util_ping() failed:" << memcached_strerror(NULL
, rc
) << " Connection:" << server
<< std::endl
;
195 server
.set_pid(libmemcached_util_getpid(server
.hostname
, server
.port(), &rc
));
196 if (not server
.has_pid())
198 std::cerr
<< CERR_PREFIX
<< "libmemcached_util_getpid() failed" << memcached_strerror(NULL
, rc
) << " Connection: " << server
<< std::endl
;
203 server_config_string
+= "--server=";
204 server_config_string
+= server
.hostname
;
205 server_config_string
+= ":";
206 server_config_string
+= boost::lexical_cast
<std::string
>(server
.port());
207 server_config_string
+= " ";
208 fprintf(stderr
, " Port %d\n", server
.port());
212 server_st
&server
= construct
->server
[construct
->count
-1];
215 std::string socket_file
;
218 server
.set_hostname(SOCKET_FILE
);
220 if ((var
= getenv("LIBMEMCACHED_SOCKET")))
226 if (not cycle_server(server
))
228 std::cerr
<< CERR_PREFIX
<< "Found server " << server
<< ", could not flush it, so trying next port." << std::endl
;
234 if (server
.is_used())
236 std::cerr
<< std::endl
<< "Using server at : " << server
<< std::endl
;
240 char buffer
[FILENAME_MAX
];
241 snprintf(buffer
, sizeof(buffer
), "%s -d -t 1 -s %s", MEMCACHED_BINARY
, SOCKET_FILE
);
243 int status
= system(buffer
);
246 std::cerr
<< CERR_PREFIX
<< "Failed system(" << buffer
<< ")" << std::endl
;
249 fprintf(stderr
, "STARTING SERVER: %s\n", buffer
);
252 memcached_return_t rc
;
253 while (not libmemcached_util_ping(server
.hostname
, server
.port(), &rc
) and --count
)
258 if (memcached_failed(rc
))
260 std::cerr
<< CERR_PREFIX
<< "libmemcached_util_ping() failed:" << memcached_strerror(NULL
, rc
) << " Connection:" << server
<< std::endl
;
264 server
.set_pid(libmemcached_util_getpid(server
.hostname
, server
.port(), &rc
));
265 if (not server
.has_pid())
267 std::cerr
<< CERR_PREFIX
<< "libmemcached_util_getpid() failed" << memcached_strerror(NULL
, rc
) << " Connection: " << server
<< std::endl
;
273 set_default_socket(server
.hostname
);
274 server_config_string
+= "--socket=\"";
275 server_config_string
+= server
.hostname
;
276 server_config_string
+= "\" ";
280 server_config_string
.resize(server_config_string
.size() -1); // Remove final space
281 construct
->server_list
= server_config_string
;
284 srandom((unsigned int)time(NULL
));
286 std::cerr
<< std::endl
;
290 void server_shutdown(server_startup_st
*construct
)
292 for (uint32_t x
= 0; x
< construct
->count
; x
++)
294 if (construct
->server
[x
].is_used())
297 construct
->server
[x
].kill();