← Back to branch summary

~drizzle-trunk/drizzle/development

« back to all changes in this revision

Viewing changes to libdrizzle-2.0/libdrizzle/query.cc

  • Committer: Monty Taylor
  • Date: 2011-02-13 17:26:39 UTC
  • mfrom: (2157.2.2 give-in-to-pkg-config)
  • mto: This revision was merged to the branch mainline in revision 2166.
  • Revision ID: mordred@inaugust.com-20110213172639-nhy7i72sfhoq13ms
Merged in pkg-config fixes.

Show diffs side-by-side

added added

removed removed

Lines of Context:
libdrizzle-2.0/libdrizzle/query.cc
1
 
/*
2
 
 * Drizzle Client & Protocol Library
3
 
 *
4
 
 * Copyright (C) 2008 Eric Day (eday@oddments.org)
5
 
 * All rights reserved.
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
 
/**
38
 
 * @file
39
 
 * @brief Query definitions
40
 
 */
41
 
 
42
 
#include "common.h"
43
 
 
44
 
drizzle_result_st *drizzle_query(drizzle_con_st *con, drizzle_result_st *result,
45
 
                                 const char *query, size_t size,
46
 
                                 drizzle_return_t *ret_ptr)
47
 
{
48
 
  return drizzle_con_command_write(con, result, DRIZZLE_COMMAND_QUERY,
49
 
                                   (uint8_t *)query, size, size, ret_ptr);
50
 
}
51
 
 
52
 
drizzle_result_st *drizzle_query_str(drizzle_con_st *con,
53
 
                                     drizzle_result_st *result,
54
 
                                     const char *query, 
55
 
                                     drizzle_return_t *ret_ptr)
56
 
{
57
 
  size_t size;
58
 
 
59
 
  size= strlen(query);
60
 
 
61
 
  return drizzle_con_command_write(con, result, DRIZZLE_COMMAND_QUERY,
62
 
                                   (uint8_t *)query, size, size, ret_ptr);
63
 
}
64
 
 
65
 
drizzle_result_st *drizzle_query_inc(drizzle_con_st *con,
66
 
                                     drizzle_result_st *result,
67
 
                                     const char *query, size_t size,
68
 
                                     size_t total, drizzle_return_t *ret_ptr)
69
 
{
70
 
  return drizzle_con_command_write(con, result, DRIZZLE_COMMAND_QUERY,
71
 
                                   (uint8_t *)query, size, total, ret_ptr);
72
 
}
73
 
 
74
 
drizzle_query_st *drizzle_query_add(drizzle_st *drizzle,
75
 
                                    drizzle_query_st *query,
76
 
                                    drizzle_con_st *con,
77
 
                                    drizzle_result_st *result,
78
 
                                    const char *query_string, size_t size,
79
 
                                    drizzle_query_options_t options,
80
 
                                    void *context)
81
 
{
82
 
  query= drizzle_query_create(drizzle, query);
83
 
  if (query == NULL)
84
 
    return NULL;
85
 
 
86
 
  drizzle_query_set_con(query, con);
87
 
  drizzle_query_set_result(query, result);
88
 
  drizzle_query_set_string(query, query_string, size);
89
 
  drizzle_query_add_options(query, options);
90
 
  drizzle_query_set_context(query, context);
91
 
 
92
 
  return query;
93
 
}
94
 
 
95
 
drizzle_query_st *drizzle_query_create(drizzle_st *drizzle,
96
 
                                       drizzle_query_st *query)
97
 
{
98
 
  if (query == NULL)
99
 
  {
100
 
    query= new drizzle_query_st;
101
 
    if (query == NULL)
102
 
    {
103
 
      drizzle_set_error(drizzle, "drizzle_query_create", "malloc");
104
 
      return NULL;
105
 
    }
106
 
 
107
 
    query->options|= DRIZZLE_CON_ALLOCATED;
108
 
  }
109
 
  else
110
 
  {
111
 
    query->prev= NULL;
112
 
    query->options= 0;
113
 
    query->state= DRIZZLE_QUERY_STATE_INIT;
114
 
    query->con= NULL;
115
 
    query->result= NULL;
116
 
    query->string= NULL;
117
 
    query->size= 0;
118
 
    query->context= NULL;
119
 
    query->context_free_fn= NULL;
120
 
  }
121
 
 
122
 
  query->drizzle= drizzle;
123
 
 
124
 
  if (drizzle->query_list)
125
 
    drizzle->query_list->prev= query;
126
 
  query->next= drizzle->query_list;
127
 
  drizzle->query_list= query;
128
 
  drizzle->query_count++;
129
 
  drizzle->query_new++;
130
 
 
131
 
  return query;
132
 
}
133
 
 
134
 
void drizzle_query_free(drizzle_query_st *query)
135
 
{
136
 
  if (query->context != NULL && query->context_free_fn != NULL)
137
 
    query->context_free_fn(query, query->context);
138
 
 
139
 
  if (query->drizzle->query_list == query)
140
 
    query->drizzle->query_list= query->next;
141
 
  if (query->prev)
142
 
    query->prev->next= query->next;
143
 
  if (query->next)
144
 
    query->next->prev= query->prev;
145
 
  query->drizzle->query_count--;
146
 
 
147
 
  if (query->options & DRIZZLE_QUERY_ALLOCATED)
148
 
    delete query;
149
 
}
150
 
 
151
 
void drizzle_query_free_all(drizzle_st *drizzle)
152
 
{
153
 
  while (drizzle->query_list != NULL)
154
 
    drizzle_query_free(drizzle->query_list);
155
 
}
156
 
 
157
 
drizzle_con_st *drizzle_query_con(drizzle_query_st *query)
158
 
{
159
 
  return query->con;
160
 
}
161
 
 
162
 
void drizzle_query_set_con(drizzle_query_st *query, drizzle_con_st *con)
163
 
{
164
 
  query->con= con;
165
 
}
166
 
 
167
 
drizzle_result_st *drizzle_query_result(drizzle_query_st *query)
168
 
{
169
 
  return query->result;
170
 
}
171
 
 
172
 
void drizzle_query_set_result(drizzle_query_st *query,
173
 
                              drizzle_result_st *result)
174
 
{
175
 
  query->result= result;
176
 
}
177
 
 
178
 
char *drizzle_query_string(drizzle_query_st *query, size_t *size)
179
 
{
180
 
  *size= query->size;
181
 
  return (char *)(query->string);
182
 
}
183
 
 
184
 
void drizzle_query_set_string(drizzle_query_st *query, const char *string,
185
 
                              size_t size)
186
 
{
187
 
  query->string= string;
188
 
  query->size= size;
189
 
}
190
 
 
191
 
int drizzle_query_options(drizzle_query_st *query)
192
 
{
193
 
  return query->options;
194
 
}
195
 
 
196
 
void drizzle_query_set_options(drizzle_query_st *query,
197
 
                               int options)
198
 
{
199
 
  query->options= options;
200
 
}
201
 
 
202
 
void drizzle_query_add_options(drizzle_query_st *query,
203
 
                               int options)
204
 
{
205
 
  query->options|= options;
206
 
}
207
 
 
208
 
void drizzle_query_remove_options(drizzle_query_st *query,
209
 
                                  int options)
210
 
{
211
 
  query->options&= ~options;
212
 
}
213
 
 
214
 
void *drizzle_query_context(drizzle_query_st *query)
215
 
{
216
 
  return query->context;
217
 
}
218
 
 
219
 
void drizzle_query_set_context(drizzle_query_st *query, void *context)
220
 
{
221
 
  query->context= context;
222
 
}
223
 
 
224
 
void drizzle_query_set_context_free_fn(drizzle_query_st *query,
225
 
                                       drizzle_query_context_free_fn *function)
226
 
{
227
 
  query->context_free_fn= function;
228
 
}
229
 
 
230
 
static void drizzle_query_run_state(drizzle_query_st* query,
231
 
                                    drizzle_return_t* ret_ptr)
232
 
{
233
 
  switch (query->state)
234
 
  {
235
 
  case DRIZZLE_QUERY_STATE_INIT:
236
 
    query->state= DRIZZLE_QUERY_STATE_QUERY;
237
 
  case DRIZZLE_QUERY_STATE_QUERY:
238
 
    query->result= drizzle_query(query->con, query->result, query->string,
239
 
                                 query->size, ret_ptr);
240
 
    if (*ret_ptr == DRIZZLE_RETURN_IO_WAIT)
241
 
    {
242
 
      return;
243
 
    }
244
 
    else if (*ret_ptr != DRIZZLE_RETURN_OK)
245
 
    {
246
 
      query->state= DRIZZLE_QUERY_STATE_DONE;
247
 
      return;
248
 
    }
249
 
 
250
 
    query->state= DRIZZLE_QUERY_STATE_RESULT;
251
 
 
252
 
  case DRIZZLE_QUERY_STATE_RESULT:
253
 
    *ret_ptr= drizzle_result_buffer(query->result);
254
 
    if (*ret_ptr == DRIZZLE_RETURN_IO_WAIT)
255
 
    {
256
 
      return;
257
 
    }
258
 
 
259
 
    query->state= DRIZZLE_QUERY_STATE_DONE;
260
 
    return;
261
 
 
262
 
  default:
263
 
  case DRIZZLE_QUERY_STATE_DONE:
264
 
    return;
265
 
  }
266
 
}
267
 
 
268
 
drizzle_query_st *drizzle_query_run(drizzle_st *drizzle,
269
 
                                    drizzle_return_t *ret_ptr)
270
 
{
271
 
  int options;
272
 
  drizzle_query_st *query;
273
 
  drizzle_con_st *con;
274
 
 
275
 
  if (drizzle->query_new == 0 && drizzle->query_running == 0)
276
 
  {
277
 
    *ret_ptr= DRIZZLE_RETURN_OK;
278
 
    return NULL;
279
 
  }
280
 
 
281
 
  options= drizzle->options;
282
 
  drizzle->options|= DRIZZLE_NON_BLOCKING;
283
 
 
284
 
  /* Check to see if any queries need to be started. */
285
 
  if (drizzle->query_new > 0)
286
 
  {
287
 
    for (query= drizzle->query_list; query != NULL; query= query->next)
288
 
    {
289
 
      if (query->state != DRIZZLE_QUERY_STATE_INIT)
290
 
        continue;
291
 
 
292
 
      drizzle->query_new--;
293
 
      drizzle->query_running++;
294
 
      assert(query->con->query == NULL);
295
 
      query->con->query= query;
296
 
 
297
 
      drizzle_query_run_state(query, ret_ptr);
298
 
      if (*ret_ptr != DRIZZLE_RETURN_IO_WAIT)
299
 
      {
300
 
        assert(query->state == DRIZZLE_QUERY_STATE_DONE);
301
 
        drizzle->query_running--;
302
 
        drizzle->options= options;
303
 
        query->con->query= NULL;
304
 
        if (*ret_ptr == DRIZZLE_RETURN_ERROR_CODE || *ret_ptr == DRIZZLE_RETURN_OK)
305
 
        {
306
 
          return query;
307
 
        }
308
 
        return NULL;
309
 
      }
310
 
    }
311
 
    assert(drizzle->query_new == 0);
312
 
  }
313
 
 
314
 
  while (1)
315
 
  {
316
 
    /* Loop through each active connection. */
317
 
    while ((con= drizzle_con_ready(drizzle)) != NULL)
318
 
    {
319
 
      query= con->query;
320
 
      drizzle_query_run_state(query, ret_ptr);
321
 
      if (query->state == DRIZZLE_QUERY_STATE_DONE)
322
 
      {
323
 
        drizzle->query_running--;
324
 
        drizzle->options= options;
325
 
        con->query= NULL;
326
 
        return query;
327
 
      }
328
 
      assert(*ret_ptr == DRIZZLE_RETURN_IO_WAIT);
329
 
    }
330
 
 
331
 
    if (options & DRIZZLE_NON_BLOCKING)
332
 
    {
333
 
      *ret_ptr= DRIZZLE_RETURN_IO_WAIT;
334
 
      return NULL;
335
 
    }
336
 
 
337
 
    *ret_ptr= drizzle_con_wait(drizzle);
338
 
    if (*ret_ptr != DRIZZLE_RETURN_OK)
339
 
    {
340
 
      drizzle->options= options;
341
 
      return NULL;
342
 
    }
343
 
  }
344
 
}
345
 
 
346
 
drizzle_return_t drizzle_query_run_all(drizzle_st *drizzle)
347
 
{
348
 
  drizzle_return_t ret;
349
 
 
350
 
  while (drizzle->query_new > 0 || drizzle->query_running > 0)
351
 
  {
352
 
    (void)drizzle_query_run(drizzle, &ret);
353
 
    if (ret != DRIZZLE_RETURN_OK && ret != DRIZZLE_RETURN_ERROR_CODE)
354
 
      return ret;
355
 
  }
356
 
 
357
 
  return DRIZZLE_RETURN_OK;
358
 
}
359
 
 
360
 
ssize_t drizzle_safe_escape_string(char *to, size_t max_to_size, const char *from, size_t from_size)
361
 
{
362
 
  ssize_t to_size= 0;
363
 
  char newchar;
364
 
  const char *end;
365
 
 
366
 
  for (end= from + from_size; from < end; from++)
367
 
  {
368
 
    newchar= 0;
369
 
    /* All multi-byte UTF8 characters have the high bit set for all bytes. */
370
 
    if (!(*from & 0x80))
371
 
    {
372
 
      switch (*from)
373
 
      {
374
 
      case 0:
375
 
        newchar= '0';
376
 
        break;
377
 
      case '\n':
378
 
        newchar= 'n';
379
 
        break;
380
 
      case '\r':
381
 
        newchar= 'r';
382
 
        break;
383
 
      case '\032':
384
 
        newchar= 'Z';
385
 
        break;
386
 
      case '\\':
387
 
        newchar= '\\';
388
 
        break;
389
 
      case '\'':
390
 
        newchar= '\'';
391
 
        break;
392
 
      case '"':
393
 
        newchar= '"';
394
 
        break;
395
 
      default:
396
 
        break;
397
 
      }
398
 
    }
399
 
    if (newchar != '\0')
400
 
    {
401
 
      if ((size_t)to_size + 2 > max_to_size)
402
 
        return -1;
403
 
 
404
 
      *to++= '\\';
405
 
      *to++= newchar;
406
 
      to_size++;
407
 
    }
408
 
    else
409
 
    {
410
 
      if ((size_t)to_size + 1 > max_to_size)
411
 
        return -1;
412
 
 
413
 
      *to++= *from;
414
 
    }
415
 
    to_size++;
416
 
  }
417
 
 
418
 
  *to= 0;
419
 
 
420
 
  return to_size;
421
 
}
422
 
 
423
 
size_t drizzle_escape_string(char *to, const char *from, size_t from_size)
424
 
{
425
 
  return (size_t) drizzle_safe_escape_string(to, (from_size * 2), from, from_size);
426
 
}
427
 
 
428
 
size_t drizzle_hex_string(char *to, const char *from, size_t from_size)
429
 
{
430
 
  static const char hex_map[]= "0123456789ABCDEF";
431
 
  const char *from_end;
432
 
 
433
 
  for (from_end= from + from_size; from != from_end; from++)
434
 
  {
435
 
    *to++= hex_map[((unsigned char) *from) >> 4];
436
 
    *to++= hex_map[((unsigned char) *from) & 0xF];
437
 
  }
438
 
 
439
 
  *to= 0;
440
 
 
441
 
  return from_size * 2;
442
 
}
443
 
 
444
 
void drizzle_mysql_password_hash(char *to, const char *from, size_t from_size)
445
 
{
446
 
  SHA1_CTX ctx;
447
 
  uint8_t hash_tmp1[SHA1_DIGEST_LENGTH];
448
 
  uint8_t hash_tmp2[SHA1_DIGEST_LENGTH];
449
 
 
450
 
  SHA1Init(&ctx);
451
 
  SHA1Update(&ctx, (const uint8_t*)from, from_size);
452
 
  SHA1Final(hash_tmp1, &ctx);
453
 
 
454
 
  SHA1Init(&ctx);
455
 
  SHA1Update(&ctx, hash_tmp1, SHA1_DIGEST_LENGTH);
456
 
  SHA1Final(hash_tmp2, &ctx);
457
 
 
458
 
  (void)drizzle_hex_string(to, (char*)hash_tmp2, SHA1_DIGEST_LENGTH);
459
 
}