jenkins-promote-staging-trunk-libmemcached-14
[awesomized/libmemcached] / libtest / drizzled.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 <config.h>
38 #include <libtest/common.h>
39
40 #include <libtest/drizzled.h>
41
42 #include "util/instance.hpp"
43 #include "util/operation.hpp"
44
45 using namespace datadifferential;
46 using namespace libtest;
47
48 #include <cassert>
49 #include <cerrno>
50 #include <cstdio>
51 #include <cstdlib>
52 #include <cstring>
53 #include <iostream>
54 #include <signal.h>
55 #include <sstream>
56 #include <sys/types.h>
57 #include <sys/wait.h>
58 #include <unistd.h>
59
60 #ifndef __INTEL_COMPILER
61 #pragma GCC diagnostic ignored "-Wold-style-cast"
62 #endif
63
64 #if defined(HAVE_LIBDRIZZLE) && HAVE_LIBDRIZZLE
65 #include <libdrizzle-1.0/drizzle_client.h>
66 #endif
67
68 using namespace libtest;
69
70 namespace libtest {
71 bool ping_drizzled(const in_port_t _port)
72 {
73 #if defined(HAVE_LIBDRIZZLE) && HAVE_LIBDRIZZLE
74 {
75 drizzle_st *drizzle= drizzle_create(NULL);
76
77 if (drizzle == NULL)
78 {
79 return false;
80 }
81
82 drizzle_con_st *con;
83
84 if ((con= drizzle_con_create(drizzle, NULL)) == NULL)
85 {
86 drizzle_free(drizzle);
87 return false;
88 }
89
90 drizzle_con_set_tcp(con, "localhost", _port);
91 drizzle_con_set_auth(con, "root", 0);
92
93 bool success= false;
94
95 drizzle_return_t rc;
96 if ((rc= drizzle_con_connect(con)) == DRIZZLE_RETURN_OK)
97 {
98 drizzle_result_st *result= drizzle_ping(con, NULL, &rc);
99 success= bool(result);
100 drizzle_result_free(result);
101 }
102
103 if (success == true)
104 { }
105 else if (rc != DRIZZLE_RETURN_OK)
106 {
107 Error << drizzle_error(drizzle) << " localhost:" << _port;
108 }
109
110 drizzle_con_free(con);
111 drizzle_free(drizzle);
112
113 return success;
114 }
115 #endif
116
117 return false;
118 }
119 } // namespace libtest
120
121 class Drizzle : public libtest::Server
122 {
123 private:
124 public:
125 Drizzle(const std::string& host_arg, in_port_t port_arg) :
126 libtest::Server(host_arg, port_arg, DRIZZLED_BINARY, false)
127 {
128 set_pid_file();
129 }
130
131 bool ping()
132 {
133 size_t limit= 5;
134 while (_app.check() and --limit)
135 {
136 if (ping_drizzled(_port))
137 {
138 return true;
139 }
140 libtest::dream(1, 0);
141 }
142
143 return false;
144 }
145
146 const char *name()
147 {
148 return "drizzled";
149 };
150
151 void log_file_option(Application& app, const std::string& arg)
152 {
153 }
154
155 bool has_log_file_option() const
156 {
157 return true;
158 }
159
160 bool broken_pid_file()
161 {
162 return true;
163 }
164
165 bool is_libtool()
166 {
167 return false;
168 }
169
170 bool has_syslog() const
171 {
172 return true;
173 }
174
175 bool has_port_option() const
176 {
177 return true;
178 }
179
180 void port_option(Application& app, in_port_t arg)
181 {
182 if (arg > 0)
183 {
184 char buffer[1024];
185 snprintf(buffer, sizeof(buffer), "--drizzle-protocol.port=%d", int(arg));
186 app.add_option(buffer);
187 }
188 }
189
190 bool build(size_t argc, const char *argv[]);
191 };
192
193 bool Drizzle::build(size_t argc, const char *argv[])
194 {
195 if (getuid() == 0 or geteuid() == 0)
196 {
197 add_option("--user=root");
198 }
199
200 add_option("--verbose=INSPECT");
201 #if 0
202 add_option("--datadir=var/drizzle");
203 #endif
204
205 for (size_t x= 0 ; x < argc ; x++)
206 {
207 add_option(argv[x]);
208 }
209
210 return true;
211 }
212
213 namespace libtest {
214
215 libtest::Server *build_drizzled(const char *hostname, in_port_t try_port)
216 {
217 return new Drizzle(hostname, try_port);
218 }
219
220 }