← Back to branch summary

~drizzle-trunk/drizzle/development

« back to all changes in this revision

Viewing changes to drizzled/functions/str/quote.cc

  • Committer: Brian Aker
  • Date: 2008-11-13 02:56:15 UTC
  • mfrom: (575.4.10 devel)
  • Revision ID: brian@tangent.org-20081113025615-snhsi52yb2ivmx6f
Merging Monty's code.

Show diffs side-by-side

added added

removed removed

Lines of Context:
drizzled/functions/str/quote.cc
 
1
/* -*- mode: c++; c-basic-offset: 2; indent-tabs-mode: nil; -*-
 
2
 *  vim:expandtab:shiftwidth=2:tabstop=2:smarttab:
 
3
 *
 
4
 *  Copyright (C) 2008 Sun Microsystems
 
5
 *
 
6
 *  This program is free software; you can redistribute it and/or modify
 
7
 *  it under the terms of the GNU General Public License as published by
 
8
 *  the Free Software Foundation; version 2 of the License.
 
9
 *
 
10
 *  This program is distributed in the hope that it will be useful,
 
11
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 
12
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 
13
 *  GNU General Public License for more details.
 
14
 *
 
15
 *  You should have received a copy of the GNU General Public License
 
16
 *  along with this program; if not, write to the Free Software
 
17
 *  Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
 
18
 */
 
19
 
 
20
#include <drizzled/server_includes.h>
 
21
#include CSTDINT_H
 
22
#include <drizzled/functions/str/quote.h>
 
23
 
 
24
 
 
25
#define get_esc_bit(mask, num) (1 & (*((mask) + ((num) >> 3))) >> ((num) & 7))
 
26
 
 
27
/**
 
28
  QUOTE() function returns argument string in single quotes suitable for
 
29
  using in a SQL statement.
 
30
 
 
31
  Adds a \\ before all characters that needs to be escaped in a SQL string.
 
32
  We also escape '^Z' (END-OF-FILE in windows) to avoid probelms when
 
33
  running commands from a file in windows.
 
34
 
 
35
  This function is very useful when you want to generate SQL statements.
 
36
 
 
37
  @note
 
38
    QUOTE(NULL) returns the string 'NULL' (4 letters, without quotes).
 
39
 
 
40
  @retval
 
41
    str    Quoted string
 
42
  @retval
 
43
    NULL           Out of memory.
 
44
*/
 
45
 
 
46
String *Item_func_quote::val_str(String *str)
 
47
{
 
48
  assert(fixed == 1);
 
49
  /*
 
50
    Bit mask that has 1 for set for the position of the following characters:
 
51
    0, \, ' and ^Z
 
52
  */
 
53
 
 
54
  static unsigned char escmask[32]=
 
55
  {
 
56
    0x01, 0x00, 0x00, 0x04, 0x80, 0x00, 0x00, 0x00,
 
57
    0x00, 0x00, 0x00, 0x10, 0x00, 0x00, 0x00, 0x00,
 
58
    0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
 
59
    0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00
 
60
  };
 
61
 
 
62
  char *from, *to, *end, *start;
 
63
  String *arg= args[0]->val_str(str);
 
64
  uint32_t arg_length, new_length;
 
65
  if (!arg)                                     // Null argument
 
66
  {
 
67
    /* Return the string 'NULL' */
 
68
    str->copy(STRING_WITH_LEN("NULL"), collation.collation);
 
69
    null_value= 0;
 
70
    return str;
 
71
  }
 
72
 
 
73
  arg_length= arg->length();
 
74
  new_length= arg_length+2; /* for beginning and ending ' signs */
 
75
 
 
76
  for (from= (char*) arg->ptr(), end= from + arg_length; from < end; from++)
 
77
    new_length+= get_esc_bit(escmask, (unsigned char) *from);
 
78
 
 
79
  if (tmp_value.alloc(new_length))
 
80
    goto null;
 
81
 
 
82
  /*
 
83
    We replace characters from the end to the beginning
 
84
  */
 
85
  to= (char*) tmp_value.ptr() + new_length - 1;
 
86
  *to--= '\'';
 
87
  for (start= (char*) arg->ptr(),end= start + arg_length; end-- != start; to--)
 
88
  {
 
89
    /*
 
90
      We can't use the bitmask here as we want to replace \O and ^Z with 0
 
91
      and Z
 
92
    */
 
93
    switch (*end)  {
 
94
    case 0:
 
95
      *to--= '0';
 
96
      *to=   '\\';
 
97
      break;
 
98
    case '\032':
 
99
      *to--= 'Z';
 
100
      *to=   '\\';
 
101
      break;
 
102
    case '\'':
 
103
    case '\\':
 
104
      *to--= *end;
 
105
      *to=   '\\';
 
106
      break;
 
107
    default:
 
108
      *to= *end;
 
109
      break;
 
110
    }
 
111
  }
 
112
  *to= '\'';
 
113
  tmp_value.length(new_length);
 
114
  tmp_value.set_charset(collation.collation);
 
115
  null_value= 0;
 
116
  return &tmp_value;
 
117
 
 
118
null:
 
119
  null_value= 1;
 
120
  return 0;
 
121
}