~drizzle-trunk/drizzle/development

« back to all changes in this revision

Viewing changes to client/sql_string.cc

  • Committer: brian
  • Date: 2008-06-25 05:29:13 UTC
  • Revision ID: brian@localhost.localdomain-20080625052913-6upwo0jsrl4lnapl
clean slate

Show diffs side-by-side

added added

removed removed

Lines of Context:
 
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 */
 
15
 
 
16
/* This file is originally from the mysql distribution. Coded by monty */
 
17
 
 
18
#ifdef USE_PRAGMA_IMPLEMENTATION
 
19
#pragma implementation                          // gcc: Class implementation
 
20
#endif
 
21
 
 
22
#include <my_global.h>
 
23
#include <my_sys.h>
 
24
#include <m_string.h>
 
25
#include <m_ctype.h>
 
26
#ifdef HAVE_FCONVERT
 
27
#include <floatingpoint.h>
 
28
#endif
 
29
 
 
30
/*
 
31
  The following extern declarations are ok as these are interface functions
 
32
  required by the string function
 
33
*/
 
34
 
 
35
extern void sql_alloc(size_t size);
 
36
extern void sql_element_free(void *ptr);
 
37
 
 
38
#include "sql_string.h"
 
39
 
 
40
/*****************************************************************************
 
41
** String functions
 
42
*****************************************************************************/
 
43
 
 
44
bool String::real_alloc(uint32 arg_length)
 
45
{
 
46
  arg_length=ALIGN_SIZE(arg_length+1);
 
47
  str_length=0;
 
48
  if (Alloced_length < arg_length)
 
49
  {
 
50
    free();
 
51
    if (!(Ptr=(char*) my_malloc(arg_length,MYF(MY_WME))))
 
52
      return TRUE;
 
53
    Alloced_length=arg_length;
 
54
    alloced=1;
 
55
  }
 
56
  Ptr[0]=0;
 
57
  return FALSE;
 
58
}
 
59
 
 
60
 
 
61
/*
 
62
** Check that string is big enough. Set string[alloc_length] to 0
 
63
** (for C functions)
 
64
*/
 
65
 
 
66
bool String::realloc(uint32 alloc_length)
 
67
{
 
68
  uint32 len=ALIGN_SIZE(alloc_length+1);
 
69
  if (Alloced_length < len)
 
70
  {
 
71
    char *new_ptr;
 
72
    if (alloced)
 
73
    {
 
74
      if ((new_ptr= (char*) my_realloc(Ptr,len,MYF(MY_WME))))
 
75
      {
 
76
        Ptr=new_ptr;
 
77
        Alloced_length=len;
 
78
      }
 
79
      else
 
80
        return TRUE;                            // Signal error
 
81
    }
 
82
    else if ((new_ptr= (char*) my_malloc(len,MYF(MY_WME))))
 
83
    {
 
84
      if (str_length)                           // Avoid bugs in memcpy on AIX
 
85
        memcpy(new_ptr,Ptr,str_length);
 
86
      new_ptr[str_length]=0;
 
87
      Ptr=new_ptr;
 
88
      Alloced_length=len;
 
89
      alloced=1;
 
90
    }
 
91
    else
 
92
      return TRUE;                      // Signal error
 
93
  }
 
94
  Ptr[alloc_length]=0;                  // This make other funcs shorter
 
95
  return FALSE;
 
96
}
 
97
 
 
98
bool String::set(longlong num, CHARSET_INFO *cs)
 
99
{
 
100
  uint l=20*cs->mbmaxlen+1;
 
101
 
 
102
  if (alloc(l))
 
103
    return TRUE;
 
104
  str_length=(uint32) (cs->cset->longlong10_to_str)(cs,Ptr,l,-10,num);
 
105
  str_charset=cs;
 
106
  return FALSE;
 
107
}
 
108
 
 
109
bool String::set(ulonglong num, CHARSET_INFO *cs)
 
110
{
 
111
  uint l=20*cs->mbmaxlen+1;
 
112
 
 
113
  if (alloc(l))
 
114
    return TRUE;
 
115
  str_length=(uint32) (cs->cset->longlong10_to_str)(cs,Ptr,l,10,num);
 
116
  str_charset=cs;
 
117
  return FALSE;
 
118
}
 
119
 
 
120
bool String::set(double num,uint decimals, CHARSET_INFO *cs)
 
121
{
 
122
  char buff[331];
 
123
  uint dummy_errors;
 
124
 
 
125
  str_charset=cs;
 
126
  if (decimals >= NOT_FIXED_DEC)
 
127
  {
 
128
    uint32 len= my_sprintf(buff,(buff, "%.14g",num));// Enough for a DATETIME
 
129
    return copy(buff, len, &my_charset_latin1, cs, &dummy_errors);
 
130
  }
 
131
#ifdef HAVE_FCONVERT
 
132
  int decpt,sign;
 
133
  char *pos,*to;
 
134
 
 
135
  VOID(fconvert(num,(int) decimals,&decpt,&sign,buff+1));
 
136
  if (!my_isdigit(&my_charset_latin1, buff[1]))
 
137
  {                                             // Nan or Inf
 
138
    pos=buff+1;
 
139
    if (sign)
 
140
    {
 
141
      buff[0]='-';
 
142
      pos=buff;
 
143
    }
 
144
    uint dummy_errors;
 
145
    return copy(pos,(uint32) strlen(pos), &my_charset_latin1, cs, &dummy_errors);
 
146
  }
 
147
  if (alloc((uint32) ((uint32) decpt+3+decimals)))
 
148
    return TRUE;
 
149
  to=Ptr;
 
150
  if (sign)
 
151
    *to++='-';
 
152
 
 
153
  pos=buff+1;
 
154
  if (decpt < 0)
 
155
  {                                     /* value is < 0 */
 
156
    *to++='0';
 
157
    if (!decimals)
 
158
      goto end;
 
159
    *to++='.';
 
160
    if ((uint32) -decpt > decimals)
 
161
      decpt= - (int) decimals;
 
162
    decimals=(uint32) ((int) decimals+decpt);
 
163
    while (decpt++ < 0)
 
164
      *to++='0';
 
165
  }
 
166
  else if (decpt == 0)
 
167
  {
 
168
    *to++= '0';
 
169
    if (!decimals)
 
170
      goto end;
 
171
    *to++='.';
 
172
  }
 
173
  else
 
174
  {
 
175
    while (decpt-- > 0)
 
176
      *to++= *pos++;
 
177
    if (!decimals)
 
178
      goto end;
 
179
    *to++='.';
 
180
  }
 
181
  while (decimals--)
 
182
    *to++= *pos++;
 
183
 
 
184
end:
 
185
  *to=0;
 
186
  str_length=(uint32) (to-Ptr);
 
187
  return FALSE;
 
188
#else
 
189
#ifdef HAVE_SNPRINTF
 
190
  buff[sizeof(buff)-1]=0;                       // Safety
 
191
  snprintf(buff,sizeof(buff)-1, "%.*f",(int) decimals,num);
 
192
#else
 
193
  sprintf(buff,"%.*f",(int) decimals,num);
 
194
#endif
 
195
  return copy(buff,(uint32) strlen(buff), &my_charset_latin1, cs,
 
196
              &dummy_errors);
 
197
#endif
 
198
}
 
199
 
 
200
 
 
201
bool String::copy()
 
202
{
 
203
  if (!alloced)
 
204
  {
 
205
    Alloced_length=0;                           // Force realloc
 
206
    return realloc(str_length);
 
207
  }
 
208
  return FALSE;
 
209
}
 
210
 
 
211
bool String::copy(const String &str)
 
212
{
 
213
  if (alloc(str.str_length))
 
214
    return TRUE;
 
215
  str_length=str.str_length;
 
216
  bmove(Ptr,str.Ptr,str_length);                // May be overlapping
 
217
  Ptr[str_length]=0;
 
218
  str_charset=str.str_charset;
 
219
  return FALSE;
 
220
}
 
221
 
 
222
bool String::copy(const char *str,uint32 arg_length, CHARSET_INFO *cs)
 
223
{
 
224
  if (alloc(arg_length))
 
225
    return TRUE;
 
226
  if ((str_length=arg_length))
 
227
    memcpy(Ptr,str,arg_length);
 
228
  Ptr[arg_length]=0;
 
229
  str_charset=cs;
 
230
  return FALSE;
 
231
}
 
232
 
 
233
 
 
234
/*
 
235
  Checks that the source string can be just copied to the destination string
 
236
  without conversion.
 
237
 
 
238
  SYNPOSIS
 
239
 
 
240
  needs_conversion()
 
241
  arg_length            Length of string to copy.
 
242
  from_cs               Character set to copy from
 
243
  to_cs                 Character set to copy to
 
244
  uint32 *offset        Returns number of unaligned characters.
 
245
 
 
246
  RETURN
 
247
   0  No conversion needed
 
248
   1  Either character set conversion or adding leading  zeros
 
249
      (e.g. for UCS-2) must be done
 
250
 
 
251
  NOTE
 
252
  to_cs may be NULL for "no conversion" if the system variable
 
253
  character_set_results is NULL.
 
254
*/
 
255
 
 
256
bool String::needs_conversion(uint32 arg_length,
 
257
                              CHARSET_INFO *from_cs,
 
258
                              CHARSET_INFO *to_cs,
 
259
                              uint32 *offset)
 
260
{
 
261
  *offset= 0;
 
262
  if (!to_cs ||
 
263
      (to_cs == &my_charset_bin) || 
 
264
      (to_cs == from_cs) ||
 
265
      my_charset_same(from_cs, to_cs) ||
 
266
      ((from_cs == &my_charset_bin) &&
 
267
       (!(*offset=(arg_length % to_cs->mbminlen)))))
 
268
    return FALSE;
 
269
  return TRUE;
 
270
}
 
271
 
 
272
 
 
273
/*
 
274
  Copy a multi-byte character sets with adding leading zeros.
 
275
 
 
276
  SYNOPSIS
 
277
 
 
278
  copy_aligned()
 
279
  str                   String to copy
 
280
  arg_length            Length of string. This should NOT be dividable with
 
281
                        cs->mbminlen.
 
282
  offset                arg_length % cs->mb_minlength
 
283
  cs                    Character set for 'str'
 
284
 
 
285
  NOTES
 
286
    For real multi-byte, ascii incompatible charactser sets,
 
287
    like UCS-2, add leading zeros if we have an incomplete character.
 
288
    Thus, 
 
289
      SELECT _ucs2 0xAA 
 
290
    will automatically be converted into
 
291
      SELECT _ucs2 0x00AA
 
292
 
 
293
  RETURN
 
294
    0  ok
 
295
    1  error
 
296
*/
 
297
 
 
298
bool String::copy_aligned(const char *str,uint32 arg_length, uint32 offset,
 
299
                          CHARSET_INFO *cs)
 
300
{
 
301
  /* How many bytes are in incomplete character */
 
302
  offset= cs->mbmaxlen - offset; /* How many zeros we should prepend */
 
303
  DBUG_ASSERT(offset && offset != cs->mbmaxlen);
 
304
 
 
305
  uint32 aligned_length= arg_length + offset;
 
306
  if (alloc(aligned_length))
 
307
    return TRUE;
 
308
  
 
309
  /*
 
310
    Note, this is only safe for little-endian UCS-2.
 
311
    If we add big-endian UCS-2 sometimes, this code
 
312
    will be more complicated. But it's OK for now.
 
313
  */
 
314
  bzero((char*) Ptr, offset);
 
315
  memcpy(Ptr + offset, str, arg_length);
 
316
  Ptr[aligned_length]=0;
 
317
  /* str_length is always >= 0 as arg_length is != 0 */
 
318
  str_length= aligned_length;
 
319
  str_charset= cs;
 
320
  return FALSE;
 
321
}
 
322
 
 
323
 
 
324
bool String::set_or_copy_aligned(const char *str,uint32 arg_length,
 
325
                                 CHARSET_INFO *cs)
 
326
{
 
327
  /* How many bytes are in incomplete character */
 
328
  uint32 offset= (arg_length % cs->mbminlen); 
 
329
  
 
330
  if (!offset) /* All characters are complete, just copy */
 
331
  {
 
332
    set(str, arg_length, cs);
 
333
    return FALSE;
 
334
  }
 
335
  return copy_aligned(str, arg_length, offset, cs);
 
336
}
 
337
 
 
338
        /* Copy with charset convertion */
 
339
 
 
340
bool String::copy(const char *str, uint32 arg_length,
 
341
                  CHARSET_INFO *from_cs, CHARSET_INFO *to_cs, uint *errors)
 
342
{
 
343
  uint32 offset;
 
344
  if (!needs_conversion(arg_length, from_cs, to_cs, &offset))
 
345
  {
 
346
    *errors= 0;
 
347
    return copy(str, arg_length, to_cs);
 
348
  }
 
349
  if ((from_cs == &my_charset_bin) && offset)
 
350
  {
 
351
    *errors= 0;
 
352
    return copy_aligned(str, arg_length, offset, to_cs);
 
353
  }
 
354
  uint32 new_length= to_cs->mbmaxlen*arg_length;
 
355
  if (alloc(new_length))
 
356
    return TRUE;
 
357
  str_length=copy_and_convert((char*) Ptr, new_length, to_cs,
 
358
                              str, arg_length, from_cs, errors);
 
359
  str_charset=to_cs;
 
360
  return FALSE;
 
361
}
 
362
 
 
363
 
 
364
/*
 
365
  Set a string to the value of a latin1-string, keeping the original charset
 
366
  
 
367
  SYNOPSIS
 
368
    copy_or_set()
 
369
    str                 String of a simple charset (latin1)
 
370
    arg_length          Length of string
 
371
 
 
372
  IMPLEMENTATION
 
373
    If string object is of a simple character set, set it to point to the
 
374
    given string.
 
375
    If not, make a copy and convert it to the new character set.
 
376
 
 
377
  RETURN
 
378
    0   ok
 
379
    1   Could not allocate result buffer
 
380
 
 
381
*/
 
382
 
 
383
bool String::set_ascii(const char *str, uint32 arg_length)
 
384
{
 
385
  if (str_charset->mbminlen == 1)
 
386
  {
 
387
    set(str, arg_length, str_charset);
 
388
    return 0;
 
389
  }
 
390
  uint dummy_errors;
 
391
  return copy(str, arg_length, &my_charset_latin1, str_charset, &dummy_errors);
 
392
}
 
393
 
 
394
 
 
395
/* This is used by mysql.cc */
 
396
 
 
397
bool String::fill(uint32 max_length,char fill_char)
 
398
{
 
399
  if (str_length > max_length)
 
400
    Ptr[str_length=max_length]=0;
 
401
  else
 
402
  {
 
403
    if (realloc(max_length))
 
404
      return TRUE;
 
405
    bfill(Ptr+str_length,max_length-str_length,fill_char);
 
406
    str_length=max_length;
 
407
  }
 
408
  return FALSE;
 
409
}
 
410
 
 
411
void String::strip_sp()
 
412
{
 
413
   while (str_length && my_isspace(str_charset,Ptr[str_length-1]))
 
414
    str_length--;
 
415
}
 
416
 
 
417
bool String::append(const String &s)
 
418
{
 
419
  if (s.length())
 
420
  {
 
421
    if (realloc(str_length+s.length()))
 
422
      return TRUE;
 
423
    memcpy(Ptr+str_length,s.ptr(),s.length());
 
424
    str_length+=s.length();
 
425
  }
 
426
  return FALSE;
 
427
}
 
428
 
 
429
 
 
430
/*
 
431
  Append an ASCII string to the a string of the current character set
 
432
*/
 
433
 
 
434
bool String::append(const char *s,uint32 arg_length)
 
435
{
 
436
  if (!arg_length)
 
437
    return FALSE;
 
438
 
 
439
  /*
 
440
    For an ASCII incompatible string, e.g. UCS-2, we need to convert
 
441
  */
 
442
  if (str_charset->mbminlen > 1)
 
443
  {
 
444
    uint32 add_length=arg_length * str_charset->mbmaxlen;
 
445
    uint dummy_errors;
 
446
    if (realloc(str_length+ add_length))
 
447
      return TRUE;
 
448
    str_length+= copy_and_convert(Ptr+str_length, add_length, str_charset,
 
449
                                  s, arg_length, &my_charset_latin1,
 
450
                                  &dummy_errors);
 
451
    return FALSE;
 
452
  }
 
453
 
 
454
  /*
 
455
    For an ASCII compatinble string we can just append.
 
456
  */
 
457
  if (realloc(str_length+arg_length))
 
458
    return TRUE;
 
459
  memcpy(Ptr+str_length,s,arg_length);
 
460
  str_length+=arg_length;
 
461
  return FALSE;
 
462
}
 
463
 
 
464
 
 
465
/*
 
466
  Append a 0-terminated ASCII string
 
467
*/
 
468
 
 
469
bool String::append(const char *s)
 
470
{
 
471
  return append(s, strlen(s));
 
472
}
 
473
 
 
474
 
 
475
/*
 
476
  Append a string in the given charset to the string
 
477
  with character set recoding
 
478
*/
 
479
 
 
480
bool String::append(const char *s,uint32 arg_length, CHARSET_INFO *cs)
 
481
{
 
482
  uint32 dummy_offset;
 
483
  
 
484
  if (needs_conversion(arg_length, cs, str_charset, &dummy_offset))
 
485
  {
 
486
    uint32 add_length= arg_length / cs->mbminlen * str_charset->mbmaxlen;
 
487
    uint dummy_errors;
 
488
    if (realloc(str_length + add_length)) 
 
489
      return TRUE;
 
490
    str_length+= copy_and_convert(Ptr+str_length, add_length, str_charset,
 
491
                                  s, arg_length, cs, &dummy_errors);
 
492
  }
 
493
  else
 
494
  {
 
495
    if (realloc(str_length + arg_length)) 
 
496
      return TRUE;
 
497
    memcpy(Ptr + str_length, s, arg_length);
 
498
    str_length+= arg_length;
 
499
  }
 
500
  return FALSE;
 
501
}
 
502
 
 
503
 
 
504
#ifdef TO_BE_REMOVED
 
505
bool String::append(FILE* file, uint32 arg_length, myf my_flags)
 
506
{
 
507
  if (realloc(str_length+arg_length))
 
508
    return TRUE;
 
509
  if (my_fread(file, (uchar*) Ptr + str_length, arg_length, my_flags))
 
510
  {
 
511
    shrink(str_length);
 
512
    return TRUE;
 
513
  }
 
514
  str_length+=arg_length;
 
515
  return FALSE;
 
516
}
 
517
#endif
 
518
 
 
519
bool String::append(IO_CACHE* file, uint32 arg_length)
 
520
{
 
521
  if (realloc(str_length+arg_length))
 
522
    return TRUE;
 
523
  if (my_b_read(file, (uchar*) Ptr + str_length, arg_length))
 
524
  {
 
525
    shrink(str_length);
 
526
    return TRUE;
 
527
  }
 
528
  str_length+=arg_length;
 
529
  return FALSE;
 
530
}
 
531
 
 
532
bool String::append_with_prefill(const char *s,uint32 arg_length,
 
533
                 uint32 full_length, char fill_char)
 
534
{
 
535
  int t_length= arg_length > full_length ? arg_length : full_length;
 
536
 
 
537
  if (realloc(str_length + t_length))
 
538
    return TRUE;
 
539
  t_length= full_length - arg_length;
 
540
  if (t_length > 0)
 
541
  {
 
542
    bfill(Ptr+str_length, t_length, fill_char);
 
543
    str_length=str_length + t_length;
 
544
  }
 
545
  append(s, arg_length);
 
546
  return FALSE;
 
547
}
 
548
 
 
549
uint32 String::numchars()
 
550
{
 
551
  return str_charset->cset->numchars(str_charset, Ptr, Ptr+str_length);
 
552
}
 
553
 
 
554
int String::charpos(int i,uint32 offset)
 
555
{
 
556
  if (i <= 0)
 
557
    return i;
 
558
  return str_charset->cset->charpos(str_charset,Ptr+offset,Ptr+str_length,i);
 
559
}
 
560
 
 
561
int String::strstr(const String &s,uint32 offset)
 
562
{
 
563
  if (s.length()+offset <= str_length)
 
564
  {
 
565
    if (!s.length())
 
566
      return ((int) offset);    // Empty string is always found
 
567
 
 
568
    register const char *str = Ptr+offset;
 
569
    register const char *search=s.ptr();
 
570
    const char *end=Ptr+str_length-s.length()+1;
 
571
    const char *search_end=s.ptr()+s.length();
 
572
skip:
 
573
    while (str != end)
 
574
    {
 
575
      if (*str++ == *search)
 
576
      {
 
577
        register char *i,*j;
 
578
        i=(char*) str; j=(char*) search+1;
 
579
        while (j != search_end)
 
580
          if (*i++ != *j++) goto skip;
 
581
        return (int) (str-Ptr) -1;
 
582
      }
 
583
    }
 
584
  }
 
585
  return -1;
 
586
}
 
587
 
 
588
/*
 
589
** Search string from end. Offset is offset to the end of string
 
590
*/
 
591
 
 
592
int String::strrstr(const String &s,uint32 offset)
 
593
{
 
594
  if (s.length() <= offset && offset <= str_length)
 
595
  {
 
596
    if (!s.length())
 
597
      return offset;                            // Empty string is always found
 
598
    register const char *str = Ptr+offset-1;
 
599
    register const char *search=s.ptr()+s.length()-1;
 
600
 
 
601
    const char *end=Ptr+s.length()-2;
 
602
    const char *search_end=s.ptr()-1;
 
603
skip:
 
604
    while (str != end)
 
605
    {
 
606
      if (*str-- == *search)
 
607
      {
 
608
        register char *i,*j;
 
609
        i=(char*) str; j=(char*) search-1;
 
610
        while (j != search_end)
 
611
          if (*i-- != *j--) goto skip;
 
612
        return (int) (i-Ptr) +1;
 
613
      }
 
614
    }
 
615
  }
 
616
  return -1;
 
617
}
 
618
 
 
619
/*
 
620
  Replace substring with string
 
621
  If wrong parameter or not enough memory, do nothing
 
622
*/
 
623
 
 
624
bool String::replace(uint32 offset,uint32 arg_length,const String &to)
 
625
{
 
626
  return replace(offset,arg_length,to.ptr(),to.length());
 
627
}
 
628
 
 
629
bool String::replace(uint32 offset,uint32 arg_length,
 
630
                     const char *to, uint32 to_length)
 
631
{
 
632
  long diff = (long) to_length-(long) arg_length;
 
633
  if (offset+arg_length <= str_length)
 
634
  {
 
635
    if (diff < 0)
 
636
    {
 
637
      if (to_length)
 
638
        memcpy(Ptr+offset,to,to_length);
 
639
      bmove(Ptr+offset+to_length,Ptr+offset+arg_length,
 
640
            str_length-offset-arg_length);
 
641
    }
 
642
    else
 
643
    {
 
644
      if (diff)
 
645
      {
 
646
        if (realloc(str_length+(uint32) diff))
 
647
          return TRUE;
 
648
        bmove_upp((uchar*) Ptr+str_length+diff, (uchar*) Ptr+str_length,
 
649
                  str_length-offset-arg_length);
 
650
      }
 
651
      if (to_length)
 
652
        memcpy(Ptr+offset,to,to_length);
 
653
    }
 
654
    str_length+=(uint32) diff;
 
655
  }
 
656
  return FALSE;
 
657
}
 
658
 
 
659
 
 
660
// added by Holyfoot for "geometry" needs
 
661
int String::reserve(uint32 space_needed, uint32 grow_by)
 
662
{
 
663
  if (Alloced_length < str_length + space_needed)
 
664
  {
 
665
    if (realloc(Alloced_length + max(space_needed, grow_by) - 1))
 
666
      return TRUE;
 
667
  }
 
668
  return FALSE;
 
669
}
 
670
 
 
671
void String::qs_append(const char *str, uint32 len)
 
672
{
 
673
  memcpy(Ptr + str_length, str, len + 1);
 
674
  str_length += len;
 
675
}
 
676
 
 
677
void String::qs_append(double d)
 
678
{
 
679
  char *buff = Ptr + str_length;
 
680
  str_length+= my_gcvt(d, MY_GCVT_ARG_DOUBLE, FLOATING_POINT_BUFFER - 1, buff, NULL);
 
681
}
 
682
 
 
683
void String::qs_append(double *d)
 
684
{
 
685
  double ld;
 
686
  float8get(ld, (char*) d);
 
687
  qs_append(ld);
 
688
}
 
689
 
 
690
void String::qs_append(int i)
 
691
{
 
692
  char *buff= Ptr + str_length;
 
693
  char *end= int10_to_str(i, buff, -10);
 
694
  str_length+= (int) (end-buff);
 
695
}
 
696
 
 
697
void String::qs_append(uint i)
 
698
{
 
699
  char *buff= Ptr + str_length;
 
700
  char *end= int10_to_str(i, buff, 10);
 
701
  str_length+= (int) (end-buff);
 
702
}
 
703
 
 
704
/*
 
705
  Compare strings according to collation, without end space.
 
706
 
 
707
  SYNOPSIS
 
708
    sortcmp()
 
709
    s           First string
 
710
    t           Second string
 
711
    cs          Collation
 
712
 
 
713
  NOTE:
 
714
    Normally this is case sensitive comparison
 
715
 
 
716
  RETURN
 
717
  < 0   s < t
 
718
  0     s == t
 
719
  > 0   s > t
 
720
*/
 
721
 
 
722
 
 
723
int sortcmp(const String *s,const String *t, CHARSET_INFO *cs)
 
724
{
 
725
 return cs->coll->strnncollsp(cs,
 
726
                              (unsigned char *) s->ptr(),s->length(),
 
727
                              (unsigned char *) t->ptr(),t->length(), 0);
 
728
}
 
729
 
 
730
 
 
731
/*
 
732
  Compare strings byte by byte. End spaces are also compared.
 
733
 
 
734
  SYNOPSIS
 
735
    stringcmp()
 
736
    s           First string
 
737
    t           Second string
 
738
 
 
739
  NOTE:
 
740
    Strings are compared as a stream of unsigned chars
 
741
 
 
742
  RETURN
 
743
  < 0   s < t
 
744
  0     s == t
 
745
  > 0   s > t
 
746
*/
 
747
 
 
748
 
 
749
int stringcmp(const String *s,const String *t)
 
750
{
 
751
  uint32 s_len=s->length(),t_len=t->length(),len=min(s_len,t_len);
 
752
  int cmp= memcmp(s->ptr(), t->ptr(), len);
 
753
  return (cmp) ? cmp : (int) (s_len - t_len);
 
754
}
 
755
 
 
756
 
 
757
String *copy_if_not_alloced(String *to,String *from,uint32 from_length)
 
758
{
 
759
  if (from->Alloced_length >= from_length)
 
760
    return from;
 
761
  if (from->alloced || !to || from == to)
 
762
  {
 
763
    (void) from->realloc(from_length);
 
764
    return from;
 
765
  }
 
766
  if (to->realloc(from_length))
 
767
    return from;                                // Actually an error
 
768
  if ((to->str_length=min(from->str_length,from_length)))
 
769
    memcpy(to->Ptr,from->Ptr,to->str_length);
 
770
  to->str_charset=from->str_charset;
 
771
  return to;
 
772
}
 
773
 
 
774
 
 
775
/****************************************************************************
 
776
  Help functions
 
777
****************************************************************************/
 
778
 
 
779
/*
 
780
  copy a string from one character set to another
 
781
  
 
782
  SYNOPSIS
 
783
    copy_and_convert()
 
784
    to                  Store result here
 
785
    to_cs               Character set of result string
 
786
    from                Copy from here
 
787
    from_length         Length of from string
 
788
    from_cs             From character set
 
789
 
 
790
  NOTES
 
791
    'to' must be big enough as form_length * to_cs->mbmaxlen
 
792
 
 
793
  RETURN
 
794
    length of bytes copied to 'to'
 
795
*/
 
796
 
 
797
 
 
798
uint32
 
799
copy_and_convert(char *to, uint32 to_length, CHARSET_INFO *to_cs, 
 
800
                 const char *from, uint32 from_length, CHARSET_INFO *from_cs,
 
801
                 uint *errors)
 
802
{
 
803
  int         cnvres;
 
804
  my_wc_t     wc;
 
805
  const uchar *from_end= (const uchar*) from+from_length;
 
806
  char *to_start= to;
 
807
  uchar *to_end= (uchar*) to+to_length;
 
808
  my_charset_conv_mb_wc mb_wc= from_cs->cset->mb_wc;
 
809
  my_charset_conv_wc_mb wc_mb= to_cs->cset->wc_mb;
 
810
  uint error_count= 0;
 
811
 
 
812
  while (1)
 
813
  {
 
814
    if ((cnvres= (*mb_wc)(from_cs, &wc, (uchar*) from,
 
815
                                      from_end)) > 0)
 
816
      from+= cnvres;
 
817
    else if (cnvres == MY_CS_ILSEQ)
 
818
    {
 
819
      error_count++;
 
820
      from++;
 
821
      wc= '?';
 
822
    }
 
823
    else if (cnvres > MY_CS_TOOSMALL)
 
824
    {
 
825
      /*
 
826
        A correct multibyte sequence detected
 
827
        But it doesn't have Unicode mapping.
 
828
      */
 
829
      error_count++;
 
830
      from+= (-cnvres);
 
831
      wc= '?';
 
832
    }
 
833
    else
 
834
      break;  // Not enough characters
 
835
 
 
836
outp:
 
837
    if ((cnvres= (*wc_mb)(to_cs, wc, (uchar*) to, to_end)) > 0)
 
838
      to+= cnvres;
 
839
    else if (cnvres == MY_CS_ILUNI && wc != '?')
 
840
    {
 
841
      error_count++;
 
842
      wc= '?';
 
843
      goto outp;
 
844
    }
 
845
    else
 
846
      break;
 
847
  }
 
848
  *errors= error_count;
 
849
  return (uint32) (to - to_start);
 
850
}
 
851
 
 
852
 
 
853
void String::print(String *str)
 
854
{
 
855
  char *st= (char*)Ptr, *end= st+str_length;
 
856
  for (; st < end; st++)
 
857
  {
 
858
    uchar c= *st;
 
859
    switch (c)
 
860
    {
 
861
    case '\\':
 
862
      str->append(STRING_WITH_LEN("\\\\"));
 
863
      break;
 
864
    case '\0':
 
865
      str->append(STRING_WITH_LEN("\\0"));
 
866
      break;
 
867
    case '\'':
 
868
      str->append(STRING_WITH_LEN("\\'"));
 
869
      break;
 
870
    case '\n':
 
871
      str->append(STRING_WITH_LEN("\\n"));
 
872
      break;
 
873
    case '\r':
 
874
      str->append(STRING_WITH_LEN("\\r"));
 
875
      break;
 
876
    case 26: //Ctrl-Z
 
877
      str->append(STRING_WITH_LEN("\\z"));
 
878
      break;
 
879
    default:
 
880
      str->append(c);
 
881
    }
 
882
  }
 
883
}
 
884
 
 
885
 
 
886
/*
 
887
  Exchange state of this object and argument.
 
888
 
 
889
  SYNOPSIS
 
890
    String::swap()
 
891
 
 
892
  RETURN
 
893
    Target string will contain state of this object and vice versa.
 
894
*/
 
895
 
 
896
void String::swap(String &s)
 
897
{
 
898
  swap_variables(char *, Ptr, s.Ptr);
 
899
  swap_variables(uint32, str_length, s.str_length);
 
900
  swap_variables(uint32, Alloced_length, s.Alloced_length);
 
901
  swap_variables(bool, alloced, s.alloced);
 
902
  swap_variables(CHARSET_INFO*, str_charset, s.str_charset);
 
903
}