Latest of libtest.
[awesomized/libmemcached] / libtest / memcached.cc
1 /* vim:expandtab:shiftwidth=2:tabstop=2:smarttab:
2 *
3 * libtest
4 *
5 * Copyright (C) 2011 Data Differential, http://datadifferential.com/
6 *
7 * This library is free software; you can redistribute it and/or
8 * modify it under the terms of the GNU Lesser General Public
9 * License as published by the Free Software Foundation; either
10 * version 3 of the License, or (at your option) any later version.
11 *
12 * This library is distributed in the hope that it will be useful,
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15 * Lesser General Public License for more details.
16 *
17 * You should have received a copy of the GNU Lesser General Public
18 * License along with this library; if not, write to the Free Software
19 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
20 */
21
22
23 #include <config.h>
24 #include <libtest/common.h>
25
26 #include <libmemcached-1.0/memcached.h>
27 #include <libmemcachedutil-1.0/util.h>
28
29 using namespace libtest;
30
31 #include <cassert>
32 #include <cerrno>
33 #include <cstdio>
34 #include <cstdlib>
35 #include <cstring>
36 #include <iostream>
37 #include <signal.h>
38 #include <sys/types.h>
39 #include <sys/wait.h>
40 #include <unistd.h>
41
42 #include <libtest/server.h>
43 #include <libtest/wait.h>
44
45 #include <libtest/memcached.h>
46
47 #ifndef __INTEL_COMPILER
48 #pragma GCC diagnostic ignored "-Wold-style-cast"
49 #endif
50
51 using namespace libtest;
52
53 namespace {
54 bool is_memcached_libtool()
55 {
56 if (MEMCACHED_BINARY and strcmp(MEMCACHED_BINARY, "memcached/memcached") == 0)
57 {
58 return true;
59 }
60
61 return false;
62 }
63 }
64
65 class Memcached : public libtest::Server
66 {
67 std::string _username;
68 std::string _password;
69
70 public:
71 Memcached(const std::string& host_arg,
72 const in_port_t port_arg,
73 const bool is_socket_arg,
74 const std::string& username_arg,
75 const std::string& password_arg) :
76 libtest::Server(host_arg, port_arg,
77 MEMCACHED_BINARY, is_memcached_libtool(), is_socket_arg),
78 _username(username_arg),
79 _password(password_arg)
80 { }
81
82 Memcached(const std::string& host_arg, const in_port_t port_arg, const bool is_socket_arg) :
83 libtest::Server(host_arg, port_arg,
84 MEMCACHED_BINARY, is_memcached_libtool(), is_socket_arg)
85 {
86 set_pid_file();
87 }
88
89 virtual const char *sasl() const
90 {
91 return NULL;
92 }
93
94 const std::string& password() const
95 {
96 return _password;
97 }
98
99 const std::string& username() const
100 {
101 return _username;
102 }
103
104 bool wait_for_pidfile() const
105 {
106 Wait wait(pid(), 4);
107
108 return wait.successful();
109 }
110
111 bool ping()
112 {
113 #if 0
114 // Memcached is slow to start, so we need to do this
115 if (pid_file().empty() == false)
116 {
117 if (wait_for_pidfile() == false)
118 {
119 Error << "Pidfile was not found:" << pid_file() << " :" << running();
120 return -1;
121 }
122 }
123 #endif
124
125 memcached_return_t rc;
126 bool ret;
127
128 if (has_socket())
129 {
130 ret= libmemcached_util_ping(socket().c_str(), 0, &rc);
131 }
132 else
133 {
134 ret= libmemcached_util_ping(hostname().c_str(), port(), &rc);
135 }
136
137 if (memcached_failed(rc) or not ret)
138 {
139 Error << "libmemcached_util_ping(" << hostname() << ", " << port() << ") error: " << memcached_strerror(NULL, rc);
140 }
141
142 return ret;
143 }
144
145 const char *name()
146 {
147 return "memcached";
148 };
149
150 const char *executable()
151 {
152 return MEMCACHED_BINARY;
153 }
154
155 bool is_libtool()
156 {
157 return is_memcached_libtool();
158 }
159
160 virtual void pid_file_option(Application& app, const std::string& arg)
161 {
162 if (arg.empty() == false)
163 {
164 app.add_option("-P", arg);
165 }
166 }
167
168 const char *socket_file_option() const
169 {
170 return "-s ";
171 }
172
173 virtual void port_option(Application& app, in_port_t arg)
174 {
175 char buffer[30];
176 snprintf(buffer, sizeof(buffer), "%d", int(arg));
177 app.add_option("-p", buffer);
178 }
179
180 bool has_port_option() const
181 {
182 return true;
183 }
184
185 bool has_socket_file_option() const
186 {
187 return has_socket();
188 }
189
190 void socket_file_option(Application& app, const std::string& socket_arg)
191 {
192 if (socket_arg.empty() == false)
193 {
194 app.add_option("-s", socket_arg);
195 }
196 }
197
198 bool broken_socket_cleanup()
199 {
200 return true;
201 }
202
203 // Memcached's pidfile is broken
204 bool broken_pid_file()
205 {
206 return true;
207 }
208
209 bool build(size_t argc, const char *argv[]);
210 };
211
212 class MemcachedLight : public libtest::Server
213 {
214
215 public:
216 MemcachedLight(const std::string& host_arg, const in_port_t port_arg) :
217 libtest::Server(host_arg, port_arg, MEMCACHED_LIGHT_BINARY, true)
218 {
219 set_pid_file();
220 }
221
222 bool ping()
223 {
224 // Memcached is slow to start, so we need to do this
225 if (not pid_file().empty())
226 {
227 if (not wait_for_pidfile())
228 {
229 Error << "Pidfile was not found:" << pid_file();
230 return false;
231 }
232 }
233
234 std::stringstream error_message;
235 pid_t local_pid= get_pid_from_file(pid_file(), error_message);
236 if (local_pid > 0)
237 {
238 if (::kill(local_pid, 0) == 0)
239 {
240 return true;
241 }
242 }
243
244 return false;
245 }
246
247 const char *name()
248 {
249 return "memcached_light";
250 };
251
252 const char *executable()
253 {
254 return MEMCACHED_LIGHT_BINARY;
255 }
256
257 virtual void port_option(Application& app, in_port_t arg)
258 {
259 char buffer[1024];
260 snprintf(buffer, sizeof(buffer), "--port=%d", int(arg));
261 app.add_option(buffer);
262 }
263
264 bool has_port_option() const
265 {
266 return true;
267 }
268
269 bool is_libtool()
270 {
271 return true;
272 }
273
274 void log_file_option(Application& app, const std::string& arg)
275 {
276 if (arg.empty() == false)
277 {
278 std::string buffer("--log-file=");
279 buffer+= arg;
280 app.add_option("--verbose");
281 app.add_option(buffer);
282 }
283 }
284
285 bool has_log_file_option() const
286 {
287 return true;
288 }
289
290 bool build(size_t argc, const char *argv[]);
291 };
292
293 class MemcachedSaSL : public Memcached
294 {
295 public:
296 MemcachedSaSL(const std::string& host_arg,
297 const in_port_t port_arg,
298 const bool is_socket_arg,
299 const std::string& username_arg,
300 const std::string &password_arg) :
301 Memcached(host_arg, port_arg, is_socket_arg, username_arg, password_arg)
302 { }
303
304 const char *name()
305 {
306 return "memcached-sasl";
307 };
308
309 const char *sasl() const
310 {
311 return " -S -B binary ";
312 }
313
314 const char *executable()
315 {
316 return MEMCACHED_SASL_BINARY;
317 }
318
319 bool ping()
320 {
321 // Memcached is slow to start, so we need to do this
322 if (pid_file().empty() == false)
323 {
324 if (wait_for_pidfile() == false)
325 {
326 Error << "Pidfile was not found:" << pid_file();
327 return -1;
328 }
329 }
330
331 memcached_return_t rc;
332 bool ret;
333
334 if (has_socket())
335 {
336 ret= libmemcached_util_ping2(socket().c_str(), 0, username().c_str(), password().c_str(), &rc);
337 }
338 else
339 {
340 ret= libmemcached_util_ping2(hostname().c_str(), port(), username().c_str(), password().c_str(), &rc);
341 }
342
343 if (memcached_failed(rc) or ret == false)
344 {
345 Error << "libmemcached_util_ping2(" << hostname() << ", " << port() << ", " << username() << ", " << password() << ") error: " << memcached_strerror(NULL, rc);
346 }
347
348 return ret;
349 }
350
351 };
352
353
354 #include <sstream>
355
356 bool Memcached::build(size_t argc, const char *argv[])
357 {
358 std::stringstream arg_buffer;
359
360 if (getuid() == 0 or geteuid() == 0)
361 {
362 add_option("-u", "root");
363 }
364
365 add_option("-l", "localhost");
366 add_option("-m", "128");
367 add_option("-M");
368
369 if (sasl())
370 {
371 add_option(sasl());
372 }
373
374 for (int x= 0 ; x < argc ; x++)
375 {
376 add_option(argv[x]);
377 }
378
379 return true;
380 }
381
382 bool MemcachedLight::build(size_t argc, const char *argv[])
383 {
384 for (size_t x= 0 ; x < argc ; x++)
385 {
386 add_option(argv[x]);
387 }
388
389 return true;
390 }
391
392 namespace libtest {
393
394 libtest::Server *build_memcached(const std::string& hostname, const in_port_t try_port)
395 {
396 return new Memcached(hostname, try_port, false);
397 }
398
399 libtest::Server *build_memcached_socket(const std::string& socket_file, const in_port_t try_port)
400 {
401 return new Memcached(socket_file, try_port, true);
402 }
403
404 libtest::Server *build_memcached_light(const std::string& hostname, const in_port_t try_port)
405 {
406 return new MemcachedLight(hostname, try_port);
407 }
408
409
410 libtest::Server *build_memcached_sasl(const std::string& hostname, const in_port_t try_port, const std::string& username, const std::string &password)
411 {
412 if (username.empty())
413 {
414 return new MemcachedSaSL(hostname, try_port, false, "memcached", "memcached");
415 }
416
417 return new MemcachedSaSL(hostname, try_port, false, username, password);
418 }
419
420 libtest::Server *build_memcached_sasl_socket(const std::string& socket_file, const in_port_t try_port, const std::string& username, const std::string &password)
421 {
422 if (username.empty())
423 {
424 return new MemcachedSaSL(socket_file, try_port, true, "memcached", "memcached");
425 }
426
427 return new MemcachedSaSL(socket_file, try_port, true, username, password);
428 }
429
430 }
431