Fixes for scanner/parser from OSX.
[awesomized/libmemcached] / libtest / gearmand.cc
1 /* vim:expandtab:shiftwidth=2:tabstop=2:smarttab:
2 *
3 * Data Differential YATL (i.e. libtest) library
4 *
5 * Copyright (C) 2012 Data Differential, http://datadifferential.com/
6 *
7 * Redistribution and use in source and binary forms, with or without
8 * modification, are permitted provided that the following conditions are
9 * met:
10 *
11 * * Redistributions of source code must retain the above copyright
12 * notice, this list of conditions and the following disclaimer.
13 *
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
17 * distribution.
18 *
19 * * The names of its contributors may not be used to endorse or
20 * promote products derived from this software without specific prior
21 * written permission.
22 *
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.
34 *
35 */
36
37 #include "libtest/yatlcon.h"
38 #include <libtest/common.h>
39
40 #include <libtest/gearmand.h>
41
42 using namespace libtest;
43
44 #include <cassert>
45 #include <cerrno>
46 #include <cstdio>
47 #include <cstdlib>
48 #include <cstring>
49 #include <iostream>
50 #include <signal.h>
51 #include <sstream>
52 #include <sys/types.h>
53 #include <sys/wait.h>
54 #include <unistd.h>
55
56 #ifndef __INTEL_COMPILER
57 #pragma GCC diagnostic ignored "-Wold-style-cast"
58 #endif
59
60 using namespace libtest;
61
62 class Gearmand : public libtest::Server
63 {
64 private:
65 public:
66 Gearmand(const std::string& host_arg, in_port_t port_arg, bool libtool_, const char* binary);
67
68 bool ping()
69 {
70 reset_error();
71
72 if (out_of_ban_killed())
73 {
74 return false;
75 }
76
77 SimpleClient client(_hostname, _port);
78
79 std::string response;
80 bool ret= client.send_message("version", response);
81
82 if (client.is_error())
83 {
84 error(client.error());
85 }
86
87 return ret;
88 }
89
90 const char *name()
91 {
92 return "gearmand";
93 };
94
95 void log_file_option(Application& app, const std::string& arg)
96 {
97 if (arg.empty() == false)
98 {
99 std::string buffer("--log-file=");
100 buffer+= arg;
101 app.add_option("--verbose=DEBUG");
102 app.add_option(buffer);
103 }
104 }
105
106 bool has_log_file_option() const
107 {
108 return true;
109 }
110
111 bool is_libtool()
112 {
113 return true;
114 }
115
116 bool has_syslog() const
117 {
118 return false; // --syslog.errmsg-enable
119 }
120
121 bool has_port_option() const
122 {
123 return true;
124 }
125
126 bool build(size_t argc, const char *argv[]);
127 };
128
129 Gearmand::Gearmand(const std::string& host_arg, in_port_t port_arg, bool libtool_, const char* binary_arg) :
130 libtest::Server(host_arg, port_arg, binary_arg, libtool_)
131 {
132 set_pid_file();
133 }
134
135 bool Gearmand::build(size_t argc, const char *argv[])
136 {
137 if (getuid() == 0 or geteuid() == 0)
138 {
139 add_option("-u", "root");
140 }
141
142 add_option("--listen=localhost");
143
144 for (size_t x= 0 ; x < argc ; x++)
145 {
146 if (argv[x] == NULL)
147 {
148 break;
149 }
150
151 add_option(argv[x]);
152 }
153
154 return true;
155 }
156
157 namespace libtest {
158
159 libtest::Server *build_gearmand(const char *hostname, in_port_t try_port, const char* binary)
160 {
161 if (binary == NULL)
162 {
163 #if defined(HAVE_GEARMAND_BINARY)
164 # if defined(GEARMAND_BINARY)
165 if (HAVE_GEARMAND_BINARY)
166 {
167 binary= GEARMAND_BINARY;
168 }
169 # endif
170 #endif
171 }
172
173 if (binary == NULL)
174 {
175 return NULL;
176 }
177
178 bool is_libtool_script= true;
179
180 if (binary[0] == '/')
181 {
182 is_libtool_script= false;
183 }
184
185 return new Gearmand(hostname, try_port, is_libtool_script, binary);
186 }
187
188 }