Fix bad if path
[m6w6/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 "libtest/yatlcon.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-5.1/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 (void)(_port);
74 #if defined(HAVE_LIBDRIZZLE) && HAVE_LIBDRIZZLE
75 if (HAVE_LIBDRIZZLE)
76 {
77 drizzle_st *drizzle= drizzle_create(getenv("MYSQL_SERVER"),
78 getenv("MYSQL_PORT") ? atoi("MYSQL_PORT") : DRIZZLE_DEFAULT_TCP_PORT,
79 getenv("MYSQL_USER"),
80 getenv("MYSQL_PASSWORD"),
81 getenv("MYSQL_SCHEMA"), 0);
82
83 if (drizzle == NULL)
84 {
85 return false;
86 }
87
88 bool success= false;
89
90 drizzle_return_t rc;
91 if ((rc= drizzle_connect(drizzle)) == DRIZZLE_RETURN_OK)
92 {
93 drizzle_result_st *result= drizzle_ping(drizzle, &rc);
94 success= bool(result);
95 drizzle_result_free(result);
96 }
97
98 if (success == true)
99 { }
100 else if (rc != DRIZZLE_RETURN_OK)
101 {
102 Error << drizzle_error(drizzle) << " localhost:" << _port;
103 }
104
105 drizzle_quit(drizzle);
106
107 return success;
108 }
109 #endif
110
111 return false;
112 }
113 } // namespace libtest
114
115 class Drizzle : public libtest::Server
116 {
117 private:
118 public:
119 Drizzle(const std::string& host_arg, in_port_t port_arg) :
120 libtest::Server(host_arg, port_arg, DRIZZLED_BINARY, false)
121 {
122 set_pid_file();
123 }
124
125 bool ping()
126 {
127 size_t limit= 5;
128 while (_app.check() and --limit)
129 {
130 if (ping_drizzled(_port))
131 {
132 return true;
133 }
134 libtest::dream(1, 0);
135 }
136
137 return false;
138 }
139
140 const char *name()
141 {
142 return "drizzled";
143 };
144
145 void log_file_option(Application&, const std::string&)
146 {
147 }
148
149 bool has_log_file_option() const
150 {
151 return true;
152 }
153
154 bool broken_pid_file()
155 {
156 return true;
157 }
158
159 bool is_libtool()
160 {
161 return false;
162 }
163
164 bool has_syslog() const
165 {
166 return true;
167 }
168
169 bool has_port_option() const
170 {
171 return true;
172 }
173
174 void port_option(Application& app, in_port_t arg)
175 {
176 if (arg > 0)
177 {
178 libtest::vchar_t buffer;
179 buffer.resize(1024);
180 snprintf(&buffer[1024], buffer.size(), "--drizzle-protocol.port=%d", int(arg));
181 app.add_option(&buffer[1024]);
182 }
183 }
184
185 bool build();
186 };
187
188 bool Drizzle::build()
189 {
190 if (getuid() == 0 or geteuid() == 0)
191 {
192 add_option("--user=root");
193 }
194
195 add_option("--verbose=INSPECT");
196 #if 0
197 add_option("--datadir=var/drizzle");
198 #endif
199
200 return true;
201 }
202
203 namespace libtest {
204
205 libtest::Server *build_drizzled(const char *hostname, in_port_t try_port)
206 {
207 return new Drizzle(hostname, try_port);
208 }
209
210 }