1712.1.1
by Monty Taylor
Merged libdrizzle directly into tree. |
1 |
/*
|
2 |
* Drizzle Client & Protocol Library
|
|
3 |
*
|
|
4 |
* Copyright (C) 2008 Eric Day (eday@oddments.org)
|
|
5 |
* All rights reserved.
|
|
6 |
*
|
|
1971.2.1
by kalebral at gmail
update files that did not have license or had incorrect license structure |
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 |
*
|
|
1712.1.1
by Monty Taylor
Merged libdrizzle directly into tree. |
35 |
*/
|
36 |
||
1971.2.1
by kalebral at gmail
update files that did not have license or had incorrect license structure |
37 |
|
2173.2.1
by Monty Taylor
Fixes incorrect usage of include |
38 |
#include <config.h> |
2164.1.1
by Monty Taylor
Merged in outstanding changes from the win32 work. |
39 |
|
1712.1.1
by Monty Taylor
Merged libdrizzle directly into tree. |
40 |
#include <errno.h> |
41 |
#include <stdio.h> |
|
42 |
#include <stdlib.h> |
|
43 |
#include <unistd.h> |
|
44 |
||
45 |
#include <libdrizzle/drizzle_server.h> |
|
46 |
||
47 |
#define DRIZZLE_FIELD_MAX 32
|
|
48 |
#define DRIZZLE_RESULT_ROWS 20
|
|
49 |
||
50 |
#define DRIZZLE_RETURN_CHECK(__ret, __function, __drizzle) \
|
|
51 |
{ \
|
|
52 |
if ((__ret) != DRIZZLE_RETURN_OK) \
|
|
53 |
DRIZZLE_RETURN_ERROR(__function, __drizzle) \
|
|
54 |
}
|
|
55 |
||
56 |
#define DRIZZLE_RETURN_ERROR(__function, __drizzle) \
|
|
57 |
{ \
|
|
58 |
printf(__function ":%s\n", drizzle_error(__drizzle)); \
|
|
59 |
return; \
|
|
60 |
}
|
|
61 |
||
62 |
static void server(drizzle_st *drizzle, drizzle_con_st *con, |
|
63 |
drizzle_result_st *result, drizzle_column_st *column); |
|
64 |
||
65 |
int main(int argc, char *argv[]) |
|
66 |
{
|
|
67 |
int c; |
|
68 |
uint32_t count= 0; |
|
69 |
const char *host= NULL; |
|
70 |
bool mysql= false; |
|
71 |
in_port_t port= 0; |
|
72 |
drizzle_verbose_t verbose= DRIZZLE_VERBOSE_NEVER; |
|
73 |
drizzle_return_t ret; |
|
74 |
drizzle_st drizzle; |
|
1929.1.29
by Stewart Smith
fix examples/server.c stack usage: |
75 |
drizzle_con_st *con_listen= (drizzle_con_st*)malloc(sizeof(drizzle_con_st)); |
76 |
drizzle_con_st *con= (drizzle_con_st*)malloc(sizeof(drizzle_con_st)); |
|
1712.1.1
by Monty Taylor
Merged libdrizzle directly into tree. |
77 |
drizzle_result_st result; |
78 |
drizzle_column_st column; |
|
79 |
||
80 |
while((c = getopt(argc, argv, "c:h:mp:v")) != -1) |
|
81 |
{
|
|
82 |
switch(c) |
|
83 |
{
|
|
84 |
case 'c': |
|
85 |
count= (uint32_t)atoi(optarg); |
|
86 |
break; |
|
87 |
||
88 |
case 'h': |
|
89 |
host= optarg; |
|
90 |
break; |
|
91 |
||
92 |
case 'm': |
|
93 |
mysql= true; |
|
94 |
break; |
|
95 |
||
96 |
case 'p': |
|
97 |
port= (in_port_t)atoi(optarg); |
|
98 |
break; |
|
99 |
||
100 |
case 'v': |
|
1992.6.2
by Monty Taylor
Cleaned up for additional gcc 4.5 warnings. |
101 |
switch(verbose) |
102 |
{
|
|
103 |
case DRIZZLE_VERBOSE_NEVER: |
|
104 |
verbose= DRIZZLE_VERBOSE_FATAL; |
|
105 |
break; |
|
106 |
case DRIZZLE_VERBOSE_FATAL: |
|
107 |
verbose= DRIZZLE_VERBOSE_ERROR; |
|
108 |
break; |
|
109 |
case DRIZZLE_VERBOSE_ERROR: |
|
110 |
verbose= DRIZZLE_VERBOSE_INFO; |
|
111 |
break; |
|
112 |
case DRIZZLE_VERBOSE_INFO: |
|
113 |
verbose= DRIZZLE_VERBOSE_DEBUG; |
|
114 |
break; |
|
115 |
case DRIZZLE_VERBOSE_DEBUG: |
|
116 |
verbose= DRIZZLE_VERBOSE_CRAZY; |
|
117 |
break; |
|
118 |
case DRIZZLE_VERBOSE_CRAZY: |
|
119 |
case DRIZZLE_VERBOSE_MAX: |
|
120 |
break; |
|
121 |
}
|
|
1712.1.1
by Monty Taylor
Merged libdrizzle directly into tree. |
122 |
break; |
123 |
||
124 |
default: |
|
125 |
printf("\nusage: %s [-c <count>] [-h <host>] [-m] [-p <port>] [-v]\n", |
|
126 |
argv[0]); |
|
127 |
printf("\t-c <count> - Number of connections to accept before exiting\n"); |
|
128 |
printf("\t-h <host> - Host to listen on\n"); |
|
129 |
printf("\t-m - Use the MySQL protocol\n"); |
|
130 |
printf("\t-p <port> - Port to listen on\n"); |
|
131 |
printf("\t-v - Increase verbosity level\n"); |
|
132 |
return 1; |
|
133 |
}
|
|
134 |
}
|
|
135 |
||
136 |
if (drizzle_create(&drizzle) == NULL) |
|
137 |
{
|
|
138 |
printf("drizzle_create:NULL\n"); |
|
139 |
return 1; |
|
140 |
}
|
|
141 |
||
142 |
drizzle_add_options(&drizzle, DRIZZLE_FREE_OBJECTS); |
|
143 |
drizzle_set_verbose(&drizzle, verbose); |
|
144 |
||
1929.1.29
by Stewart Smith
fix examples/server.c stack usage: |
145 |
if (drizzle_con_create(&drizzle, con_listen) == NULL) |
1712.1.1
by Monty Taylor
Merged libdrizzle directly into tree. |
146 |
{
|
147 |
printf("drizzle_con_create:NULL\n"); |
|
148 |
return 1; |
|
149 |
}
|
|
150 |
||
1929.1.29
by Stewart Smith
fix examples/server.c stack usage: |
151 |
drizzle_con_add_options(con_listen, DRIZZLE_CON_LISTEN); |
152 |
drizzle_con_set_tcp(con_listen, host, port); |
|
1712.1.1
by Monty Taylor
Merged libdrizzle directly into tree. |
153 |
|
154 |
if (mysql) |
|
1929.1.29
by Stewart Smith
fix examples/server.c stack usage: |
155 |
drizzle_con_add_options(con_listen, DRIZZLE_CON_MYSQL); |
1712.1.1
by Monty Taylor
Merged libdrizzle directly into tree. |
156 |
|
1929.1.29
by Stewart Smith
fix examples/server.c stack usage: |
157 |
if (drizzle_con_listen(con_listen) != DRIZZLE_RETURN_OK) |
1712.1.1
by Monty Taylor
Merged libdrizzle directly into tree. |
158 |
{
|
159 |
printf("drizzle_con_listen:%s\n", drizzle_error(&drizzle)); |
|
160 |
return 1; |
|
161 |
}
|
|
162 |
||
163 |
while (1) |
|
164 |
{
|
|
1929.1.29
by Stewart Smith
fix examples/server.c stack usage: |
165 |
(void)drizzle_con_accept(&drizzle, con, &ret); |
1712.1.1
by Monty Taylor
Merged libdrizzle directly into tree. |
166 |
if (ret != DRIZZLE_RETURN_OK) |
167 |
{
|
|
168 |
printf("drizzle_con_accept:%s\n", drizzle_error(&drizzle)); |
|
169 |
return 1; |
|
170 |
}
|
|
171 |
||
1929.1.29
by Stewart Smith
fix examples/server.c stack usage: |
172 |
server(&drizzle, con, &result, &column); |
1712.1.1
by Monty Taylor
Merged libdrizzle directly into tree. |
173 |
|
1929.1.29
by Stewart Smith
fix examples/server.c stack usage: |
174 |
drizzle_con_free(con); |
1712.1.1
by Monty Taylor
Merged libdrizzle directly into tree. |
175 |
|
176 |
if (count > 0) |
|
177 |
{
|
|
178 |
count--; |
|
179 |
||
180 |
if (count == 0) |
|
181 |
break; |
|
182 |
}
|
|
183 |
}
|
|
184 |
||
1929.1.29
by Stewart Smith
fix examples/server.c stack usage: |
185 |
drizzle_con_free(con_listen); |
1712.1.1
by Monty Taylor
Merged libdrizzle directly into tree. |
186 |
drizzle_free(&drizzle); |
187 |
||
1929.1.29
by Stewart Smith
fix examples/server.c stack usage: |
188 |
free(con); |
189 |
free(con_listen); |
|
190 |
||
1712.1.1
by Monty Taylor
Merged libdrizzle directly into tree. |
191 |
return 0; |
192 |
}
|
|
193 |
||
194 |
static void server(drizzle_st *drizzle, drizzle_con_st *con, |
|
195 |
drizzle_result_st *result, drizzle_column_st *column) |
|
196 |
{
|
|
197 |
drizzle_return_t ret; |
|
198 |
drizzle_command_t command; |
|
199 |
uint8_t *data= NULL; |
|
200 |
size_t total; |
|
201 |
char *field[2]; |
|
202 |
char field1[DRIZZLE_FIELD_MAX]; |
|
203 |
char field2[DRIZZLE_FIELD_MAX]; |
|
204 |
size_t size[2]; |
|
205 |
uint64_t x; |
|
206 |
||
207 |
field[0]= field1; |
|
208 |
field[1]= field2; |
|
209 |
||
210 |
/* Handshake packets. */
|
|
211 |
drizzle_con_set_protocol_version(con, 10); |
|
212 |
drizzle_con_set_server_version(con, "libdrizzle example 1.2.3"); |
|
213 |
drizzle_con_set_thread_id(con, 1); |
|
214 |
drizzle_con_set_scramble(con, (const uint8_t *)"ABCDEFGHIJKLMNOPQRST"); |
|
215 |
drizzle_con_set_capabilities(con, DRIZZLE_CAPABILITIES_NONE); |
|
216 |
drizzle_con_set_charset(con, 8); |
|
217 |
drizzle_con_set_status(con, DRIZZLE_CON_STATUS_NONE); |
|
218 |
drizzle_con_set_max_packet_size(con, DRIZZLE_MAX_PACKET_SIZE); |
|
219 |
||
220 |
ret= drizzle_handshake_server_write(con); |
|
221 |
DRIZZLE_RETURN_CHECK(ret, "drizzle_handshake_server_write", drizzle) |
|
222 |
||
223 |
ret= drizzle_handshake_client_read(con); |
|
224 |
DRIZZLE_RETURN_CHECK(ret, "drizzle_handshake_client_read", drizzle) |
|
225 |
||
226 |
if (drizzle_result_create(con, result) == NULL) |
|
227 |
DRIZZLE_RETURN_ERROR("drizzle_result_create", drizzle) |
|
228 |
||
229 |
ret= drizzle_result_write(con, result, true); |
|
230 |
DRIZZLE_RETURN_CHECK(ret, "drizzle_result_write", drizzle) |
|
231 |
||
232 |
/* Command loop. */
|
|
233 |
while (1) |
|
234 |
{
|
|
235 |
drizzle_result_free(result); |
|
236 |
if (data != NULL) |
|
237 |
free(data); |
|
238 |
||
1992.6.2
by Monty Taylor
Cleaned up for additional gcc 4.5 warnings. |
239 |
data= (uint8_t *)drizzle_con_command_buffer(con, &command, &total, &ret); |
1712.1.1
by Monty Taylor
Merged libdrizzle directly into tree. |
240 |
if (ret == DRIZZLE_RETURN_LOST_CONNECTION || |
241 |
(ret == DRIZZLE_RETURN_OK && command == DRIZZLE_COMMAND_QUIT)) |
|
242 |
{
|
|
243 |
if (data != NULL) |
|
244 |
free(data); |
|
245 |
return; |
|
246 |
}
|
|
247 |
DRIZZLE_RETURN_CHECK(ret, "drizzle_con_command_buffer", drizzle) |
|
248 |
||
249 |
if (drizzle_result_create(con, result) == NULL) |
|
250 |
DRIZZLE_RETURN_ERROR("drizzle_result_create", drizzle) |
|
251 |
||
252 |
if (command != DRIZZLE_COMMAND_QUERY) |
|
253 |
{
|
|
254 |
ret= drizzle_result_write(con, result, true); |
|
255 |
DRIZZLE_RETURN_CHECK(ret, "drizzle_result_write", drizzle) |
|
256 |
continue; |
|
257 |
}
|
|
258 |
||
259 |
drizzle_result_set_column_count(result, 2); |
|
260 |
||
261 |
ret= drizzle_result_write(con, result, false); |
|
262 |
DRIZZLE_RETURN_CHECK(ret, "drizzle_result_write", drizzle) |
|
263 |
||
264 |
/* Columns. */
|
|
265 |
if (drizzle_column_create(result, column) == NULL) |
|
266 |
DRIZZLE_RETURN_ERROR("drizzle_column_create", drizzle) |
|
267 |
||
268 |
drizzle_column_set_catalog(column, "default"); |
|
269 |
drizzle_column_set_db(column, "drizzle_test_db"); |
|
270 |
drizzle_column_set_table(column, "drizzle_test_table"); |
|
271 |
drizzle_column_set_orig_table(column, "drizzle_test_table"); |
|
272 |
drizzle_column_set_name(column, "test_column_1"); |
|
273 |
drizzle_column_set_orig_name(column, "test_column_1"); |
|
274 |
drizzle_column_set_charset(column, 8); |
|
275 |
drizzle_column_set_size(column, DRIZZLE_FIELD_MAX); |
|
276 |
drizzle_column_set_type(column, DRIZZLE_COLUMN_TYPE_VARCHAR); |
|
277 |
||
278 |
ret= drizzle_column_write(result, column); |
|
279 |
DRIZZLE_RETURN_CHECK(ret, "drizzle_column_write", drizzle) |
|
280 |
||
281 |
drizzle_column_set_name(column, "test_column_2"); |
|
282 |
drizzle_column_set_orig_name(column, "test_column_2"); |
|
283 |
||
284 |
ret= drizzle_column_write(result, column); |
|
285 |
DRIZZLE_RETURN_CHECK(ret, "drizzle_column_write", drizzle) |
|
286 |
||
287 |
drizzle_column_free(column); |
|
288 |
||
289 |
drizzle_result_set_eof(result, true); |
|
290 |
||
291 |
ret= drizzle_result_write(con, result, false); |
|
292 |
DRIZZLE_RETURN_CHECK(ret, "drizzle_result_write", drizzle) |
|
293 |
||
294 |
/* Rows. */
|
|
295 |
for (x= 0; x < DRIZZLE_RESULT_ROWS; x++) |
|
296 |
{
|
|
297 |
size[0]= (size_t)snprintf(field[0], DRIZZLE_FIELD_MAX, |
|
298 |
"field %" PRIu64 "-1", x); |
|
299 |
if (size[0] >= DRIZZLE_FIELD_MAX) |
|
300 |
size[0]= DRIZZLE_FIELD_MAX - 1; |
|
301 |
||
302 |
size[1]= (size_t)snprintf(field[1], DRIZZLE_FIELD_MAX, |
|
303 |
"field %" PRIu64 "-2", x); |
|
304 |
if (size[1] >= DRIZZLE_FIELD_MAX) |
|
305 |
size[1]= DRIZZLE_FIELD_MAX - 1; |
|
306 |
||
307 |
/* This is needed for MySQL and old Drizzle protocol. */
|
|
308 |
drizzle_result_calc_row_size(result, (drizzle_field_t *)field, size); |
|
309 |
||
310 |
ret= drizzle_row_write(result); |
|
311 |
DRIZZLE_RETURN_CHECK(ret, "drizzle_row_write", drizzle) |
|
312 |
||
313 |
/* Fields. */
|
|
314 |
ret= drizzle_field_write(result, (drizzle_field_t)field[0], size[0], |
|
315 |
size[0]); |
|
316 |
DRIZZLE_RETURN_CHECK(ret, "drizzle_field_write", drizzle) |
|
317 |
||
318 |
ret= drizzle_field_write(result, (drizzle_field_t)field[1], size[1], |
|
319 |
size[1]); |
|
320 |
DRIZZLE_RETURN_CHECK(ret, "drizzle_field_write", drizzle) |
|
321 |
}
|
|
322 |
||
323 |
ret= drizzle_result_write(con, result, true); |
|
324 |
DRIZZLE_RETURN_CHECK(ret, "drizzle_result_write", drizzle) |
|
325 |
}
|
|
326 |
}
|