sql_string.h 19.8 KB
Newer Older
1 2 3
#ifndef SQL_STRING_INCLUDED
#define SQL_STRING_INCLUDED

4
/*
Sergei Golubchik's avatar
Sergei Golubchik committed
5
   Copyright (c) 2000, 2013, Oracle and/or its affiliates.
6
   Copyright (c) 2008, 2020, MariaDB Corporation.
unknown's avatar
unknown committed
7 8 9

   This program is free software; you can redistribute it and/or modify
   it under the terms of the GNU General Public License as published by
unknown's avatar
unknown committed
10
   the Free Software Foundation; version 2 of the License.
unknown's avatar
unknown committed
11 12

   This program is distributed in the hope that it will be useful,
unknown's avatar
unknown committed
13
   but WITHOUT ANY WARRANTY; without even the implied warranty of
unknown's avatar
unknown committed
14 15 16 17 18
   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
   GNU General Public License for more details.

   You should have received a copy of the GNU General Public License
   along with this program; if not, write to the Free Software
Vicențiu Ciorbaru's avatar
Vicențiu Ciorbaru committed
19
   Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1335  USA */
unknown's avatar
unknown committed
20 21 22

/* This file is originally from the mysql distribution. Coded by monty */

23
#ifdef USE_PRAGMA_INTERFACE
unknown's avatar
unknown committed
24 25 26
#pragma interface			/* gcc class implementation */
#endif

27 28 29 30
#include "m_ctype.h"                            /* my_charset_bin */
#include "my_sys.h"              /* alloc_root, my_free, my_realloc */
#include "m_string.h"                           /* TRASH */

31
class String;
32 33 34
typedef struct st_io_cache IO_CACHE;
typedef struct st_mem_root MEM_ROOT;

35
int sortcmp(const String *a,const String *b, CHARSET_INFO *cs);
36
String *copy_if_not_alloced(String *a,String *b,uint32 arg_length);
37
inline uint32 copy_and_convert(char *to, uint32 to_length,
38
                               CHARSET_INFO *to_cs,
39
                               const char *from, uint32 from_length,
40
                               CHARSET_INFO *from_cs, uint *errors)
41 42 43
{
  return my_convert(to, to_length, to_cs, from, from_length, from_cs, errors);
}
44 45


46
class String_copier: private MY_STRCONV_STATUS
47 48 49
{
public:
  const char *source_end_pos() const
50
  { return m_native_copy_status.m_source_end_pos; }
51
  const char *well_formed_error_pos() const
52
  { return m_native_copy_status.m_well_formed_error_pos; }
53 54 55 56 57 58 59
  const char *cannot_convert_error_pos() const
  { return m_cannot_convert_error_pos; }
  const char *most_important_error_pos() const
  {
    return well_formed_error_pos() ? well_formed_error_pos() :
                                     cannot_convert_error_pos();
  }
60 61 62 63 64 65 66 67 68 69 70
  /*
    Convert a string between character sets.
    "dstcs" and "srccs" cannot be &my_charset_bin.
  */
  uint convert_fix(CHARSET_INFO *dstcs, char *dst, uint dst_length,
                   CHARSET_INFO *srccs, const char *src, uint src_length,
                   uint nchars)
  {
    return my_convert_fix(dstcs, dst, dst_length,
                          srccs, src, src_length, nchars, this);
  }
71
  /*
72
     Copy a string. Fix bad bytes/characters to '?'.
73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88
  */
  uint well_formed_copy(CHARSET_INFO *to_cs, char *to, uint to_length,
                        CHARSET_INFO *from_cs, const char *from,
                        uint from_length, uint nchars);
  // Same as above, but without the "nchars" limit.
  uint well_formed_copy(CHARSET_INFO *to_cs, char *to, uint to_length,
                        CHARSET_INFO *from_cs, const char *from,
                        uint from_length)
  {
    return well_formed_copy(to_cs, to, to_length,
                            from_cs, from, from_length,
                            from_length /* No limit on "nchars"*/);
  }
};


89 90 91
size_t my_copy_with_hex_escaping(CHARSET_INFO *cs,
                                 char *dst, size_t dstlen,
                                 const char *src, size_t srclen);
92 93 94
uint convert_to_printable(char *to, size_t to_len,
                          const char *from, size_t from_len,
                          CHARSET_INFO *from_cs, size_t nbytes= 0);
95

unknown's avatar
unknown committed
96 97 98
class String
{
  char *Ptr;
99
  uint32 str_length,Alloced_length, extra_alloc;
100
  bool alloced,thread_specific;
101
  CHARSET_INFO *str_charset;
unknown's avatar
unknown committed
102 103
public:
  String()
104
  { 
105 106
    Ptr=0; str_length=Alloced_length=extra_alloc=0;
    alloced= thread_specific= 0; 
unknown's avatar
unknown committed
107
    str_charset= &my_charset_bin; 
108
  }
unknown's avatar
unknown committed
109
  String(uint32 length_arg)
110
  { 
111 112
    alloced= thread_specific= 0;
    Alloced_length= extra_alloc= 0; (void) real_alloc(length_arg); 
unknown's avatar
unknown committed
113
    str_charset= &my_charset_bin;
114
  }
115
  String(const char *str, CHARSET_INFO *cs)
116
  { 
117
    Ptr=(char*) str; str_length= (uint32) strlen(str);
118 119
    Alloced_length= extra_alloc= 0;
    alloced= thread_specific= 0;
120
    str_charset=cs;
121
  }
122 123 124 125 126
  /*
    NOTE: If one intend to use the c_ptr() method, the following two
    contructors need the size of memory for STR to be at least LEN+1 (to make
    room for zero termination).
  */
127
  String(const char *str,uint32 len, CHARSET_INFO *cs)
128
  { 
129 130
    Ptr=(char*) str; str_length=len; Alloced_length= extra_alloc=0;
    alloced= thread_specific= 0;
131
    str_charset=cs;
132
  }
133
  String(char *str,uint32 len, CHARSET_INFO *cs)
134
  { 
135 136
    Ptr=(char*) str; Alloced_length=str_length=len; extra_alloc= 0;
    alloced= thread_specific= 0;
137
    str_charset=cs;
138
  }
unknown's avatar
unknown committed
139
  String(const String &str)
140 141
  { 
    Ptr=str.Ptr ; str_length=str.str_length ;
142 143
    Alloced_length=str.Alloced_length; extra_alloc= 0;
    alloced= thread_specific= 0;
144 145
    str_charset=str.str_charset;
  }
146
  static void *operator new(size_t size, MEM_ROOT *mem_root) throw ()
147
  { return (void*) alloc_root(mem_root, (uint) size); }
148 149 150 151
  static void operator delete(void *ptr_arg, size_t size)
  {
    (void) ptr_arg;
    (void) size;
152
    TRASH_FREE(ptr_arg, size);
153 154
  }
  static void operator delete(void *, MEM_ROOT *)
unknown's avatar
unknown committed
155
  { /* never called */ }
unknown's avatar
unknown committed
156 157
  ~String() { free(); }

158 159 160 161 162 163
  /* Mark variable thread specific it it's not allocated already */
  inline void set_thread_specific()
  {
    if (!alloced)
      thread_specific= 1;
  }
164 165
  inline void set_charset(CHARSET_INFO *charset_arg)
  { str_charset= charset_arg; }
166
  inline CHARSET_INFO *charset() const { return str_charset; }
unknown's avatar
unknown committed
167 168
  inline uint32 length() const { return str_length;}
  inline uint32 alloced_length() const { return Alloced_length;}
169
  inline uint32 extra_allocation() const { return extra_alloc;}
unknown's avatar
unknown committed
170 171
  inline char& operator [] (uint32 i) const { return Ptr[i]; }
  inline void length(uint32 len) { str_length=len ; }
172
  inline void extra_allocation(uint32 len) { extra_alloc= len; }
173
  inline bool is_empty() const { return (str_length == 0); }
unknown's avatar
unknown committed
174
  inline void mark_as_const() { Alloced_length= 0;}
unknown's avatar
unknown committed
175
  inline const char *ptr() const { return Ptr; }
176
  inline const char *end() const { return Ptr + str_length; }
unknown's avatar
unknown committed
177 178
  inline char *c_ptr()
  {
179 180 181
    DBUG_ASSERT(!alloced || !Ptr || !Alloced_length || 
                (Alloced_length >= (str_length + 1)));

unknown's avatar
unknown committed
182 183 184 185 186 187 188 189 190 191
    if (!Ptr || Ptr[str_length])		/* Should be safe */
      (void) realloc(str_length);
    return Ptr;
  }
  inline char *c_ptr_quick()
  {
    if (Ptr && str_length < Alloced_length)
      Ptr[str_length]=0;
    return Ptr;
  }
unknown's avatar
unknown committed
192 193 194 195 196 197 198 199
  inline char *c_ptr_safe()
  {
    if (Ptr && str_length < Alloced_length)
      Ptr[str_length]=0;
    else
      (void) realloc(str_length);
    return Ptr;
  }
200 201
  LEX_STRING lex_string() const
  {
202 203
    LEX_STRING str = { (char*) ptr(), length() };
    return str;
204
  }
205 206
  LEX_CSTRING lex_cstring() const
  {
207 208
    LEX_CSTRING skr = { ptr(), length() };
    return skr;
209
  }
unknown's avatar
unknown committed
210 211 212

  void set(String &str,uint32 offset,uint32 arg_length)
  {
213
    DBUG_ASSERT(&str != this);
unknown's avatar
unknown committed
214
    free();
215
    Ptr=(char*) str.ptr()+offset; str_length=arg_length;
unknown's avatar
unknown committed
216 217
    if (str.Alloced_length)
      Alloced_length=str.Alloced_length-offset;
218
    str_charset=str.str_charset;
unknown's avatar
unknown committed
219
  }
220 221 222 223 224 225 226 227 228 229


  /**
     Points the internal buffer to the supplied one. The old buffer is freed.
     @param str Pointer to the new buffer.
     @param arg_length Length of the new buffer in characters, excluding any 
            null character.
     @param cs Character set to use for interpreting string data.
     @note The new buffer will not be null terminated.
  */
230
  inline void set(char *str,uint32 arg_length, CHARSET_INFO *cs)
unknown's avatar
unknown committed
231 232
  {
    free();
233
    Ptr=(char*) str; str_length=Alloced_length=arg_length;
234
    str_charset=cs;
unknown's avatar
unknown committed
235
  }
236
  inline void set(const char *str,uint32 arg_length, CHARSET_INFO *cs)
unknown's avatar
unknown committed
237 238
  {
    free();
239
    Ptr=(char*) str; str_length=arg_length;
240
    str_charset=cs;
unknown's avatar
unknown committed
241
  }
242
  bool set_ascii(const char *str, uint32 arg_length);
243
  inline void set_quick(char *str,uint32 arg_length, CHARSET_INFO *cs)
unknown's avatar
unknown committed
244 245 246 247 248
  {
    if (!alloced)
    {
      Ptr=(char*) str; str_length=Alloced_length=arg_length;
    }
249
    str_charset=cs;
unknown's avatar
unknown committed
250
  }
251
  bool set_int(longlong num, bool unsigned_flag, CHARSET_INFO *cs);
252 253 254 255 256 257
  bool set(int num, CHARSET_INFO *cs) { return set_int(num, false, cs); }
  bool set(uint num, CHARSET_INFO *cs) { return set_int(num, true, cs); }
  bool set(long num, CHARSET_INFO *cs) { return set_int(num, false, cs); }
  bool set(ulong num, CHARSET_INFO *cs) { return set_int(num, true, cs); }
  bool set(longlong num, CHARSET_INFO *cs) { return set_int(num, false, cs); }
  bool set(ulonglong num, CHARSET_INFO *cs) { return set_int((longlong)num, true, cs); }
258
  bool set_real(double num,uint decimals, CHARSET_INFO *cs);
259

260 261 262
  /* Take over handling of buffer from some other object */
  void reset(char *ptr_arg, uint32 length_arg, uint32 alloced_length_arg,
             CHARSET_INFO *cs)
263 264
  { 
    free();
265 266 267
    Ptr= ptr_arg;
    str_length= length_arg;
    Alloced_length= alloced_length_arg;
268
    str_charset= cs;
269
    alloced= ptr_arg != 0;
270 271
  }

272 273 274 275 276 277 278 279 280
  /* Forget about the buffer, let some other object handle it */
  char *release()
  {
    char *old= Ptr;
    Ptr=0; str_length= Alloced_length= extra_alloc= 0;
    alloced= thread_specific= 0;
    return old;
  }

281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296 297 298 299 300 301 302 303 304
  /*
    PMG 2004.11.12
    This is a method that works the same as perl's "chop". It simply
    drops the last character of a string. This is useful in the case
    of the federated storage handler where I'm building a unknown
    number, list of values and fields to be used in a sql insert
    statement to be run on the remote server, and have a comma after each.
    When the list is complete, I "chop" off the trailing comma

    ex. 
      String stringobj; 
      stringobj.append("VALUES ('foo', 'fi', 'fo',");
      stringobj.chop();
      stringobj.append(")");

    In this case, the value of string was:

    VALUES ('foo', 'fi', 'fo',
    VALUES ('foo', 'fi', 'fo'
    VALUES ('foo', 'fi', 'fo')
      
  */
  inline void chop()
  {
305 306 307
    str_length--;
    Ptr[str_length]= '\0';
    DBUG_ASSERT(strlen(Ptr) == str_length);
308 309
  }

unknown's avatar
unknown committed
310
  inline void free()
unknown's avatar
unknown committed
311 312
  {
    if (alloced)
unknown's avatar
unknown committed
313
    {
unknown's avatar
unknown committed
314
      alloced=0;
315
      my_free(Ptr);
unknown's avatar
unknown committed
316
    }
317 318 319
    Alloced_length= extra_alloc= 0;
    Ptr=0;
    str_length=0;				/* Safety */
unknown's avatar
unknown committed
320
  }
unknown's avatar
unknown committed
321 322 323 324 325 326 327
  inline bool alloc(uint32 arg_length)
  {
    if (arg_length < Alloced_length)
      return 0;
    return real_alloc(arg_length);
  }
  bool real_alloc(uint32 arg_length);			// Empties old string
328 329 330 331 332 333 334 335
  bool realloc_raw(uint32 arg_length);
  bool realloc(uint32 arg_length)
  {
    if (realloc_raw(arg_length))
      return TRUE;
    Ptr[arg_length]=0;        // This make other funcs shorter
    return FALSE;
  }
336 337 338 339
  bool realloc_with_extra(uint32 arg_length)
  {
    if (extra_alloc < 4096)
      extra_alloc= extra_alloc*2+128;
340 341 342 343
    if (realloc_raw(arg_length + extra_alloc))
      return TRUE;
    Ptr[arg_length]=0;        // This make other funcs shorter
    return FALSE;
344 345
  }
  bool realloc_with_extra_if_needed(uint32 arg_length)
unknown's avatar
unknown committed
346 347
  {
    if (arg_length < Alloced_length)
348 349
    {
      Ptr[arg_length]=0; // behave as if realloc was called.
350
      return 0;
351
    }
352 353
    return realloc_with_extra(arg_length);
  }
354 355
  // Shrink the buffer, but only if it is allocated on the heap.
  inline void shrink(uint32 arg_length)
356
  {
357 358
    if (!is_alloced())
      return;
359
    if (ALIGN_SIZE(arg_length+1) < Alloced_length)
unknown's avatar
unknown committed
360 361 362 363
    {
      char *new_ptr;
      if (!(new_ptr=(char*) my_realloc(Ptr,arg_length,MYF(0))))
      {
unknown's avatar
unknown committed
364
	Alloced_length = 0;
unknown's avatar
unknown committed
365 366 367 368 369 370 371 372 373
	real_alloc(arg_length);
      }
      else
      {
	Ptr=new_ptr;
	Alloced_length=arg_length;
      }
    }
  }
374
  bool is_alloced() const { return alloced; }
unknown's avatar
unknown committed
375 376 377 378
  inline String& operator = (const String &s)
  {
    if (&s != this)
    {
379 380 381 382 383
      /*
        It is forbidden to do assignments like 
        some_string = substring_of_that_string
       */
      DBUG_ASSERT(!s.uses_buffer_owned_by(this));
unknown's avatar
unknown committed
384 385
      free();
      Ptr=s.Ptr ; str_length=s.str_length ; Alloced_length=s.Alloced_length;
386
      str_charset=s.str_charset;
unknown's avatar
unknown committed
387 388 389 390 391 392
    }
    return *this;
  }

  bool copy();					// Alloc string if not alloced
  bool copy(const String &s);			// Allocate new string
393
  bool copy(const char *s,uint32 arg_length, CHARSET_INFO *cs);	// Allocate new string
unknown's avatar
unknown committed
394 395 396
  static bool needs_conversion(uint32 arg_length,
  			       CHARSET_INFO *cs_from, CHARSET_INFO *cs_to,
			       uint32 *offset);
397 398 399
  static bool needs_conversion_on_storage(uint32 arg_length,
                                          CHARSET_INFO *cs_from,
                                          CHARSET_INFO *cs_to);
unknown's avatar
unknown committed
400 401
  bool copy_aligned(const char *s, uint32 arg_length, uint32 offset,
		    CHARSET_INFO *cs);
unknown's avatar
unknown committed
402
  bool set_or_copy_aligned(const char *s, uint32 arg_length, CHARSET_INFO *cs);
unknown's avatar
unknown committed
403
  bool copy(const char*s,uint32 arg_length, CHARSET_INFO *csfrom,
404
	    CHARSET_INFO *csto, uint *errors);
405 406 407 408
  bool copy(const String *str, CHARSET_INFO *tocs, uint *errors)
  {
    return copy(str->ptr(), str->length(), str->charset(), tocs, errors);
  }
409 410 411 412 413 414 415 416 417 418 419
  bool copy(CHARSET_INFO *tocs,
            CHARSET_INFO *fromcs, const char *src, uint32 src_length,
            uint32 nchars, String_copier *copier)
  {
    if (alloc(tocs->mbmaxlen * src_length))
      return true;
    str_length= copier->well_formed_copy(tocs, Ptr, Alloced_length,
                                         fromcs, src, src_length, nchars);
    str_charset= tocs;
    return false;
  }
420 421 422 423 424 425
  void move(String &s)
  {
    free();
    Ptr=s.Ptr ; str_length=s.str_length ; Alloced_length=s.Alloced_length;
    extra_alloc= s.extra_alloc;
    alloced= s.alloced;
426
    thread_specific= s.thread_specific;
427 428
    s.alloced= 0;
  }
unknown's avatar
unknown committed
429
  bool append(const String &s);
430
  bool append(const char *s);
431
  bool append(const LEX_STRING *ls)
432 433 434 435 436
  {
    return append(ls->str, ls->length);
  }
  bool append(const char *s, uint32 arg_length);
  bool append(const char *s, uint32 arg_length, CHARSET_INFO *cs);
437
  bool append_ulonglong(ulonglong val);
438
  bool append(IO_CACHE* file, uint32 arg_length);
unknown's avatar
unknown committed
439 440
  bool append_with_prefill(const char *s, uint32 arg_length, 
			   uint32 full_length, char fill_char);
441
  bool append_parenthesized(long nr, int radix= 10);
unknown's avatar
unknown committed
442 443
  int strstr(const String &search,uint32 offset=0); // Returns offset to substring or -1
  int strrstr(const String &search,uint32 offset=0); // Returns offset to substring or -1
444
  bool replace(uint32 offset,uint32 arg_length,const char *to,uint32 length);
unknown's avatar
unknown committed
445 446 447 448 449 450 451 452 453
  bool replace(uint32 offset,uint32 arg_length,const String &to);
  inline bool append(char chr)
  {
    if (str_length < Alloced_length)
    {
      Ptr[str_length++]=chr;
    }
    else
    {
454
      if (realloc_with_extra(str_length + 1))
unknown's avatar
unknown committed
455 456 457 458 459
	return 1;
      Ptr[str_length++]=chr;
    }
    return 0;
  }
Alexander Barkov's avatar
Alexander Barkov committed
460 461
  bool append_hex(const char *src, uint32 srclen)
  {
462
    for (const char *src_end= src + srclen ; src != src_end ; src++)
Alexander Barkov's avatar
Alexander Barkov committed
463 464 465 466 467 468 469
    {
      if (append(_dig_vec_lower[((uchar) *src) >> 4]) ||
          append(_dig_vec_lower[((uchar) *src) & 0x0F]))
        return true;
    }
    return false;
  }
470 471 472 473
  bool append_hex(const uchar *src, uint32 srclen)
  {
    return append_hex((const char*)src, srclen);
  }
unknown's avatar
unknown committed
474 475
  bool fill(uint32 max_length,char fill);
  void strip_sp();
476
  friend int sortcmp(const String *a,const String *b, CHARSET_INFO *cs);
unknown's avatar
unknown committed
477
  friend int stringcmp(const String *a,const String *b);
unknown's avatar
unknown committed
478
  friend String *copy_if_not_alloced(String *a,String *b,uint32 arg_length);
479
  friend class Field;
480
  uint32 numchars() const;
Sergei Golubchik's avatar
Sergei Golubchik committed
481
  int charpos(longlong i,uint32 offset=0);
unknown's avatar
unknown committed
482 483 484 485 486 487 488

  int reserve(uint32 space_needed)
  {
    return realloc(str_length + space_needed);
  }
  int reserve(uint32 space_needed, uint32 grow_by);

unknown's avatar
unknown committed
489 490 491 492 493
  /*
    The following append operations do NOT check alloced memory
    q_*** methods writes values of parameters itself
    qs_*** methods writes string representation of value
  */
unknown's avatar
unknown committed
494
  void q_append(const char c)
unknown's avatar
unknown committed
495 496 497
  {
    Ptr[str_length++] = c;
  }
unknown's avatar
unknown committed
498
  void q_append(const uint32 n)
unknown's avatar
unknown committed
499 500 501 502 503 504 505 506 507 508 509 510 511 512 513 514
  {
    int4store(Ptr + str_length, n);
    str_length += 4;
  }
  void q_append(double d)
  {
    float8store(Ptr + str_length, d);
    str_length += 8;
  }
  void q_append(double *d)
  {
    float8store(Ptr + str_length, *d);
    str_length += 8;
  }
  void q_append(const char *data, uint32 data_len)
  {
515 516
    if (data_len)
      memcpy(Ptr + str_length, data, data_len);
unknown's avatar
unknown committed
517 518 519
    str_length += data_len;
  }

unknown's avatar
unknown committed
520
  void write_at_position(int position, uint32 value)
unknown's avatar
unknown committed
521 522 523 524
  {
    int4store(Ptr + position,value);
  }

525 526
  void qs_append(const char *str)
  {
527
    qs_append(str, (uint32)strlen(str));
528
  }
unknown's avatar
unknown committed
529
  void qs_append(const char *str, uint32 len);
unknown's avatar
unknown committed
530 531
  void qs_append(double d);
  void qs_append(double *d);
unknown's avatar
unknown committed
532 533 534 535 536
  inline void qs_append(const char c)
  {
     Ptr[str_length]= c;
     str_length++;
  }
537
  void qs_append(int i);
538 539 540 541 542 543 544 545 546
  void qs_append(uint i)
  {
    qs_append((ulonglong)i);
  }
  void qs_append(ulong i)
  {
    qs_append((ulonglong)i);
  }
  void qs_append(ulonglong i);
547 548 549 550 551 552 553 554 555 556 557 558 559 560 561 562 563 564 565 566 567 568 569 570 571

  /* Inline (general) functions used by the protocol functions */

  inline char *prep_append(uint32 arg_length, uint32 step_alloc)
  {
    uint32 new_length= arg_length + str_length;
    if (new_length > Alloced_length)
    {
      if (realloc(new_length + step_alloc))
        return 0;
    }
    uint32 old_length= str_length;
    str_length+= arg_length;
    return Ptr+ old_length;			/* Area to use */
  }

  inline bool append(const char *s, uint32 arg_length, uint32 step_alloc)
  {
    uint32 new_length= arg_length + str_length;
    if (new_length > Alloced_length && realloc(new_length + step_alloc))
      return TRUE;
    memcpy(Ptr+str_length, s, arg_length);
    str_length+= arg_length;
    return FALSE;
  }
572 573 574 575 576 577 578 579 580
  void print(String *to) const;
  void print_with_conversion(String *to, CHARSET_INFO *cs) const;
  void print(String *to, CHARSET_INFO *cs) const
  {
    if (my_charset_same(charset(), cs))
      print(to);
    else
      print_with_conversion(to, cs);
  }
581

582
  bool append_for_single_quote(const char *st, uint len);
583 584 585 586 587 588 589 590
  bool append_for_single_quote(const String *s)
  {
    return append_for_single_quote(s->ptr(), s->length());
  }
  bool append_for_single_quote(const char *st)
  {
    return append_for_single_quote(st, strlen(st));
  }
591 592 593

  /* Swap two string objects. Efficient way to exchange data without memcpy. */
  void swap(String &s);
594 595 596 597 598

  inline bool uses_buffer_owned_by(const String *s) const
  {
    return (s->alloced && Ptr >= s->Ptr && Ptr < s->Ptr + s->str_length);
  }
599 600 601 602 603 604
  uint well_formed_length() const
  {
    int dummy_error;
    return charset()->cset->well_formed_len(charset(), ptr(), ptr() + length(),
                                            length(), &dummy_error);
  }
605 606 607 608 609 610
  bool is_ascii() const
  {
    if (length() == 0)
      return TRUE;
    if (charset()->mbminlen > 1)
      return FALSE;
611
    for (const char *c= ptr(), *c_end= c + length(); c < c_end; c++)
612 613 614 615 616 617
    {
      if (!my_isascii(*c))
        return FALSE;
    }
    return TRUE;
  }
618 619 620 621 622 623 624 625 626
  bool bin_eq(const String *other) const
  {
    return length() == other->length() &&
           !memcmp(ptr(), other->ptr(), length());
  }
  bool eq(const String *other, CHARSET_INFO *cs) const
  {
    return !sortcmp(this, other, cs);
  }
unknown's avatar
unknown committed
627
};
628

Sergey Petrunya's avatar
Sergey Petrunya committed
629 630 631 632 633 634 635 636 637 638 639 640 641 642 643 644 645 646 647 648

// The following class is a backport from MySQL 5.6:
/**
  String class wrapper with a preallocated buffer of size buff_sz

  This class allows to replace sequences of:
     char buff[12345];
     String str(buff, sizeof(buff));
     str.length(0);
  with a simple equivalent declaration:
     StringBuffer<12345> str;
*/

template<size_t buff_sz>
class StringBuffer : public String
{
  char buff[buff_sz];

public:
  StringBuffer() : String(buff, buff_sz, &my_charset_bin) { length(0); }
649
  explicit StringBuffer(CHARSET_INFO *cs) : String(buff, buff_sz, cs)
Sergey Petrunya's avatar
Sergey Petrunya committed
650 651 652 653 654 655
  {
    length(0);
  }
};


656 657 658
static inline bool check_if_only_end_space(CHARSET_INFO *cs,
                                           const char *str, 
                                           const char *end)
659 660 661
{
  return str+ cs->cset->scan(cs, str, end, MY_SEQ_SPACES) == end;
}
662

663 664 665
int append_query_string(CHARSET_INFO *csinfo, String *to,
                        const char *str, size_t len, bool no_backslash);

666
#endif /* SQL_STRING_INCLUDED */