← Back to branch summary

~drizzle-trunk/drizzle/development

« back to all changes in this revision

Viewing changes to drizzled/sql_string.h

  • Committer: Monty Taylor
  • Date: 2008-08-02 01:03:15 UTC
  • mto: (236.1.42 codestyle)
  • mto: This revision was merged to the branch mainline in revision 261.
  • Revision ID: monty@inaugust.com-20080802010315-65h5938pymg9d99z
Moved m4 macros to top-level m4 dir, per GNU standards (and where gettext wanted it :)

Show diffs side-by-side

added added

removed removed

Lines of Context:
drizzled/sql_string.h
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
 
 */
 
1
/* Copyright (C) 2000 MySQL AB
 
2
 
 
3
   This program is free software; you can redistribute it and/or modify
 
4
   it under the terms of the GNU General Public License as published by
 
5
   the Free Software Foundation; version 2 of the License.
 
6
 
 
7
   This program is distributed in the hope that it will be useful,
 
8
   but WITHOUT ANY WARRANTY; without even the implied warranty of
 
9
   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 
10
   GNU General Public License for more details.
 
11
 
 
12
   You should have received a copy of the GNU General Public License
 
13
   along with this program; if not, write to the Free Software
 
14
   Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA */
19
15
 
20
16
#ifndef DRIZZLE_SERVER_SQL_STRING_H
21
17
#define DRIZZLE_SERVER_SQL_STRING_H
22
18
 
23
19
/* This file is originally from the mysql distribution. Coded by monty */
24
20
 
 
21
#ifdef USE_PRAGMA_INTERFACE
 
22
#pragma interface                       /* gcc class implementation */
 
23
#endif
25
24
 
26
25
#ifndef NOT_FIXED_DEC
27
26
#define NOT_FIXED_DEC                   31
28
27
#endif
29
28
 
30
 
#include <drizzled/common.h>
31
 
#include <mysys/iocache.h>
32
 
#include <stdlib.h>
33
 
#include <string.h>
34
 
 
35
29
class String;
36
 
 
37
 
#if defined(__cplusplus)
38
 
extern "C" {
39
 
#endif
40
 
 
41
 
  int sortcmp(const String *a,const String *b, const CHARSET_INFO * const cs);
42
 
  int stringcmp(const String *a,const String *b);
43
 
  String *copy_if_not_alloced(String *a,String *b,uint32_t arg_length);
44
 
  uint32_t copy_and_convert(char *to, uint32_t to_length,
45
 
                            const CHARSET_INFO * const to_cs,
46
 
                            const char *from, uint32_t from_length,
47
 
                            const CHARSET_INFO * const from_cs,
48
 
                            uint32_t *errors);
49
 
  uint32_t well_formed_copy_nchars(const CHARSET_INFO * const to_cs,
50
 
                                   char *to, uint32_t to_length,
51
 
                                   const CHARSET_INFO * const from_cs,
52
 
                                   const char *from, uint32_t from_length,
53
 
                                   uint32_t nchars,
54
 
                                   const char **well_formed_error_pos,
55
 
                                   const char **cannot_convert_error_pos,
56
 
                                   const char **from_end_pos);
57
 
  size_t my_copy_with_hex_escaping(const CHARSET_INFO * const cs,
58
 
                                   char *dst, size_t dstlen,
59
 
                                   const char *src, size_t srclen);
60
 
 
61
 
#if defined(__cplusplus)
62
 
}
63
 
#endif
 
30
int sortcmp(const String *a,const String *b, CHARSET_INFO *cs);
 
31
String *copy_if_not_alloced(String *a,String *b,uint32_t arg_length);
 
32
uint32_t copy_and_convert(char *to, uint32_t to_length, CHARSET_INFO *to_cs,
 
33
                        const char *from, uint32_t from_length,
 
34
                        CHARSET_INFO *from_cs, uint *errors);
 
35
uint32_t well_formed_copy_nchars(CHARSET_INFO *to_cs,
 
36
                               char *to, uint to_length,
 
37
                               CHARSET_INFO *from_cs,
 
38
                               const char *from, uint from_length,
 
39
                               uint nchars,
 
40
                               const char **well_formed_error_pos,
 
41
                               const char **cannot_convert_error_pos,
 
42
                               const char **from_end_pos);
 
43
size_t my_copy_with_hex_escaping(CHARSET_INFO *cs,
 
44
                                 char *dst, size_t dstlen,
 
45
                                 const char *src, size_t srclen);
64
46
 
65
47
class String
66
48
{
67
49
  char *Ptr;
68
50
  uint32_t str_length,Alloced_length;
69
51
  bool alloced;
70
 
  const CHARSET_INFO *str_charset;
 
52
  CHARSET_INFO *str_charset;
71
53
public:
72
54
  String()
73
 
  {
74
 
    Ptr=0; str_length=Alloced_length=0; alloced=0;
75
 
    str_charset= &my_charset_bin;
 
55
  { 
 
56
    Ptr=0; str_length=Alloced_length=0; alloced=0; 
 
57
    str_charset= &my_charset_bin; 
76
58
  }
77
59
  String(uint32_t length_arg)
78
 
  {
79
 
    alloced=0; Alloced_length=0; (void) real_alloc(length_arg);
 
60
  { 
 
61
    alloced=0; Alloced_length=0; (void) real_alloc(length_arg); 
80
62
    str_charset= &my_charset_bin;
81
63
  }
82
 
  String(const char *str, const CHARSET_INFO * const cs)
83
 
  {
 
64
  String(const char *str, CHARSET_INFO *cs)
 
65
  { 
84
66
    Ptr=(char*) str; str_length=(uint) strlen(str); Alloced_length=0; alloced=0;
85
67
    str_charset=cs;
86
68
  }
87
 
  String(const char *str,uint32_t len, const CHARSET_INFO * const cs)
88
 
  {
 
69
  String(const char *str,uint32_t len, CHARSET_INFO *cs)
 
70
  { 
89
71
    Ptr=(char*) str; str_length=len; Alloced_length=0; alloced=0;
90
72
    str_charset=cs;
91
73
  }
92
 
  String(char *str,uint32_t len, const CHARSET_INFO * const cs)
93
 
  {
 
74
  String(char *str,uint32_t len, CHARSET_INFO *cs)
 
75
  { 
94
76
    Ptr=(char*) str; Alloced_length=str_length=len; alloced=0;
95
77
    str_charset=cs;
96
78
  }
97
79
  String(const String &str)
98
 
  {
 
80
  { 
99
81
    Ptr=str.Ptr ; str_length=str.str_length ;
100
 
    Alloced_length=str.Alloced_length; alloced=0;
 
82
    Alloced_length=str.Alloced_length; alloced=0; 
101
83
    str_charset=str.str_charset;
102
84
  }
103
85
  static void *operator new(size_t size, MEM_ROOT *mem_root)
104
86
  { return (void*) alloc_root(mem_root, (uint) size); }
105
 
  static void operator delete(void *, size_t)
 
87
  static void operator delete(void *ptr_arg __attribute__((unused)),
 
88
                              size_t size __attribute__((unused)))
106
89
  { TRASH(ptr_arg, size); }
107
 
  static void operator delete(void *, MEM_ROOT *)
 
90
  static void operator delete(void *ptr_arg __attribute__((unused)),
 
91
                              MEM_ROOT *mem_root __attribute__((unused)))
108
92
  { /* never called */ }
109
93
  ~String() { free(); }
110
94
 
111
 
  inline void set_charset(const CHARSET_INFO * const charset_arg)
 
95
  inline void set_charset(CHARSET_INFO *charset_arg)
112
96
  { str_charset= charset_arg; }
113
 
  inline const CHARSET_INFO *charset() const { return str_charset; }
 
97
  inline CHARSET_INFO *charset() const { return str_charset; }
114
98
  inline uint32_t length() const { return str_length;}
115
99
  inline uint32_t alloced_length() const { return Alloced_length;}
116
100
  inline char& operator [] (uint32_t i) const { return Ptr[i]; }
117
101
  inline void length(uint32_t len) { str_length=len ; }
118
102
  inline bool is_empty() { return (str_length == 0); }
119
103
  inline void mark_as_const() { Alloced_length= 0;}
120
 
  inline char *ptr() { return Ptr; }
121
104
  inline const char *ptr() const { return Ptr; }
122
105
  inline char *c_ptr()
123
106
  {
151
134
      Alloced_length=0;
152
135
    str_charset=str.str_charset;
153
136
  }
154
 
  inline void set(char *str,uint32_t arg_length, const CHARSET_INFO * const cs)
 
137
  inline void set(char *str,uint32_t arg_length, CHARSET_INFO *cs)
155
138
  {
156
139
    free();
157
140
    Ptr=(char*) str; str_length=Alloced_length=arg_length ; alloced=0;
158
141
    str_charset=cs;
159
142
  }
160
 
  inline void set(const char *str,uint32_t arg_length, const CHARSET_INFO * const cs)
 
143
  inline void set(const char *str,uint32_t arg_length, CHARSET_INFO *cs)
161
144
  {
162
145
    free();
163
146
    Ptr=(char*) str; str_length=arg_length; Alloced_length=0 ; alloced=0;
164
147
    str_charset=cs;
165
148
  }
166
149
  bool set_ascii(const char *str, uint32_t arg_length);
167
 
  inline void set_quick(char *str,uint32_t arg_length, const CHARSET_INFO * const cs)
 
150
  inline void set_quick(char *str,uint32_t arg_length, CHARSET_INFO *cs)
168
151
  {
169
152
    if (!alloced)
170
153
    {
172
155
    }
173
156
    str_charset=cs;
174
157
  }
175
 
  bool set_int(int64_t num, bool unsigned_flag, const CHARSET_INFO * const cs);
176
 
  bool set(int64_t num, const CHARSET_INFO * const cs)
 
158
  bool set_int(int64_t num, bool unsigned_flag, CHARSET_INFO *cs);
 
159
  bool set(int64_t num, CHARSET_INFO *cs)
177
160
  { return set_int(num, false, cs); }
178
 
  bool set(uint64_t num, const CHARSET_INFO * const cs)
 
161
  bool set(uint64_t num, CHARSET_INFO *cs)
179
162
  { return set_int((int64_t)num, true, cs); }
180
 
  bool set_real(double num,uint32_t decimals, const CHARSET_INFO * const cs);
 
163
  bool set_real(double num,uint decimals, CHARSET_INFO *cs);
181
164
 
182
165
  /*
183
166
    PMG 2004.11.12
188
171
    statement to be run on the remote server, and have a comma after each.
189
172
    When the list is complete, I "chop" off the trailing comma
190
173
 
191
 
    ex.
192
 
      String stringobj;
 
174
    ex. 
 
175
      String stringobj; 
193
176
      stringobj.append("VALUES ('foo', 'fi', 'fo',");
194
177
      stringobj.chop();
195
178
      stringobj.append(")");
199
182
    VALUES ('foo', 'fi', 'fo',
200
183
    VALUES ('foo', 'fi', 'fo'
201
184
    VALUES ('foo', 'fi', 'fo')
202
 
 
 
185
      
203
186
  */
204
187
  inline void chop()
205
188
  {
206
 
    Ptr[str_length--]= '\0';
 
189
    Ptr[str_length--]= '\0'; 
207
190
  }
208
191
 
209
192
  inline void free()
212
195
    {
213
196
      alloced=0;
214
197
      Alloced_length=0;
215
 
      ::free(Ptr);
 
198
      my_free(Ptr,MYF(0));
216
199
      Ptr=0;
217
200
      str_length=0;                             /* Safety */
218
201
    }
230
213
    if (arg_length < Alloced_length)
231
214
    {
232
215
      char *new_ptr;
233
 
      if (!(new_ptr=(char*) ::realloc(Ptr,arg_length)))
 
216
      if (!(new_ptr=(char*) my_realloc(Ptr,arg_length,MYF(0))))
234
217
      {
235
218
        Alloced_length = 0;
236
219
        real_alloc(arg_length);
248
231
    if (&s != this)
249
232
    {
250
233
      /*
251
 
        It is forbidden to do assignments like
 
234
        It is forbidden to do assignments like 
252
235
        some_string = substring_of_that_string
253
236
       */
254
237
      assert(!s.uses_buffer_owned_by(this));
261
244
 
262
245
  bool copy();                                  // Alloc string if not alloced
263
246
  bool copy(const String &s);                   // Allocate new string
264
 
  bool copy(const char *s,uint32_t arg_length, const CHARSET_INFO * const cs);  // Allocate new string
 
247
  bool copy(const char *s,uint32_t arg_length, CHARSET_INFO *cs);       // Allocate new string
265
248
  static bool needs_conversion(uint32_t arg_length,
266
 
                               const CHARSET_INFO * const cs_from, const CHARSET_INFO * const cs_to,
 
249
                               CHARSET_INFO *cs_from, CHARSET_INFO *cs_to,
267
250
                               uint32_t *offset);
268
251
  bool copy_aligned(const char *s, uint32_t arg_length, uint32_t offset,
269
 
                    const CHARSET_INFO * const cs);
270
 
  bool set_or_copy_aligned(const char *s, uint32_t arg_length, const CHARSET_INFO * const cs);
271
 
  bool copy(const char*s,uint32_t arg_length, const CHARSET_INFO * const csfrom,
272
 
            const CHARSET_INFO * const csto, uint32_t *errors);
 
252
                    CHARSET_INFO *cs);
 
253
  bool set_or_copy_aligned(const char *s, uint32_t arg_length, CHARSET_INFO *cs);
 
254
  bool copy(const char*s,uint32_t arg_length, CHARSET_INFO *csfrom,
 
255
            CHARSET_INFO *csto, uint *errors);
273
256
  bool append(const String &s);
274
257
  bool append(const char *s);
275
258
  bool append(const char *s,uint32_t arg_length);
276
 
  bool append(const char *s,uint32_t arg_length, const CHARSET_INFO * const cs);
 
259
  bool append(const char *s,uint32_t arg_length, CHARSET_INFO *cs);
277
260
  bool append(IO_CACHE* file, uint32_t arg_length);
278
 
  bool append_with_prefill(const char *s, uint32_t arg_length,
 
261
  bool append_with_prefill(const char *s, uint32_t arg_length, 
279
262
                           uint32_t full_length, char fill_char);
280
263
  int strstr(const String &search,uint32_t offset=0); // Returns offset to substring or -1
281
264
  int strrstr(const String &search,uint32_t offset=0); // Returns offset to substring or -1
296
279
    return 0;
297
280
  }
298
281
  bool fill(uint32_t max_length,char fill);
299
 
  friend int sortcmp(const String *a,const String *b, const CHARSET_INFO * const cs);
 
282
  void strip_sp();
 
283
  friend int sortcmp(const String *a,const String *b, CHARSET_INFO *cs);
300
284
  friend int stringcmp(const String *a,const String *b);
301
285
  friend String *copy_if_not_alloced(String *a,String *b,uint32_t arg_length);
302
286
  uint32_t numchars();
352
336
     str_length++;
353
337
  }
354
338
  void qs_append(int i);
355
 
  void qs_append(uint32_t i);
 
339
  void qs_append(uint i);
356
340
 
357
341
  /* Inline (general) functions used by the protocol functions */
358
342
 
389
373
  }
390
374
};
391
375
 
392
 
static inline bool check_if_only_end_space(const CHARSET_INFO * const cs, char *str,
 
376
static inline bool check_if_only_end_space(CHARSET_INFO *cs, char *str, 
393
377
                                           char *end)
394
378
{
395
379
  return str+ cs->cset->scan(cs, str, end, MY_SEQ_SPACES) == end;
396
380
}
397
381
 
398
 
extern "C++" {
399
 
bool operator==(const String &s1, const String &s2);
400
 
bool operator!=(const String &s1, const String &s2);
 
382
inline
 
383
bool operator==(const String &s1, const String &s2)
 
384
{
 
385
  return stringcmp(&s1,&s2) == 0;
 
386
}
 
387
 
 
388
inline
 
389
bool operator!=(const String &s1, const String &s2)
 
390
{
 
391
  return !(s1 == s2);
401
392
}
402
393
 
403
394
#endif /* DRIZZLE_SERVER_SQL_STRING_H */