1
/* Copyright (C) 2000 MySQL AB
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.
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.
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 */
16
/* This file is originally from the mysql distribution. Coded by monty */
18
#ifdef USE_PRAGMA_IMPLEMENTATION
19
#pragma implementation // gcc: Class implementation
22
#include <my_global.h>
27
#include <floatingpoint.h>
31
The following extern declarations are ok as these are interface functions
32
required by the string function
35
extern void sql_alloc(size_t size);
36
extern void sql_element_free(void *ptr);
38
#include "sql_string.h"
40
/*****************************************************************************
42
*****************************************************************************/
44
bool String::real_alloc(uint32 arg_length)
46
arg_length=ALIGN_SIZE(arg_length+1);
48
if (Alloced_length < arg_length)
51
if (!(Ptr=(char*) my_malloc(arg_length,MYF(MY_WME))))
53
Alloced_length=arg_length;
62
** Check that string is big enough. Set string[alloc_length] to 0
66
bool String::realloc(uint32 alloc_length)
68
uint32 len=ALIGN_SIZE(alloc_length+1);
69
if (Alloced_length < len)
74
if ((new_ptr= (char*) my_realloc(Ptr,len,MYF(MY_WME))))
80
return TRUE; // Signal error
82
else if ((new_ptr= (char*) my_malloc(len,MYF(MY_WME))))
84
if (str_length) // Avoid bugs in memcpy on AIX
85
memcpy(new_ptr,Ptr,str_length);
86
new_ptr[str_length]=0;
92
return TRUE; // Signal error
94
Ptr[alloc_length]=0; // This make other funcs shorter
98
bool String::set(longlong num, CHARSET_INFO *cs)
100
uint l=20*cs->mbmaxlen+1;
104
str_length=(uint32) (cs->cset->longlong10_to_str)(cs,Ptr,l,-10,num);
109
bool String::set(ulonglong num, CHARSET_INFO *cs)
111
uint l=20*cs->mbmaxlen+1;
115
str_length=(uint32) (cs->cset->longlong10_to_str)(cs,Ptr,l,10,num);
120
bool String::set(double num,uint decimals, CHARSET_INFO *cs)
126
if (decimals >= NOT_FIXED_DEC)
128
uint32 len= my_sprintf(buff,(buff, "%.14g",num));// Enough for a DATETIME
129
return copy(buff, len, &my_charset_latin1, cs, &dummy_errors);
135
VOID(fconvert(num,(int) decimals,&decpt,&sign,buff+1));
136
if (!my_isdigit(&my_charset_latin1, buff[1]))
145
return copy(pos,(uint32) strlen(pos), &my_charset_latin1, cs, &dummy_errors);
147
if (alloc((uint32) ((uint32) decpt+3+decimals)))
160
if ((uint32) -decpt > decimals)
161
decpt= - (int) decimals;
162
decimals=(uint32) ((int) decimals+decpt);
186
str_length=(uint32) (to-Ptr);
190
buff[sizeof(buff)-1]=0; // Safety
191
snprintf(buff,sizeof(buff)-1, "%.*f",(int) decimals,num);
193
sprintf(buff,"%.*f",(int) decimals,num);
195
return copy(buff,(uint32) strlen(buff), &my_charset_latin1, cs,
205
Alloced_length=0; // Force realloc
206
return realloc(str_length);
211
bool String::copy(const String &str)
213
if (alloc(str.str_length))
215
str_length=str.str_length;
216
bmove(Ptr,str.Ptr,str_length); // May be overlapping
218
str_charset=str.str_charset;
222
bool String::copy(const char *str,uint32 arg_length, CHARSET_INFO *cs)
224
if (alloc(arg_length))
226
if ((str_length=arg_length))
227
memcpy(Ptr,str,arg_length);
235
Checks that the source string can be just copied to the destination string
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.
247
0 No conversion needed
248
1 Either character set conversion or adding leading zeros
249
(e.g. for UCS-2) must be done
252
to_cs may be NULL for "no conversion" if the system variable
253
character_set_results is NULL.
256
bool String::needs_conversion(uint32 arg_length,
257
CHARSET_INFO *from_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)))))
274
Copy a multi-byte character sets with adding leading zeros.
280
arg_length Length of string. This should NOT be dividable with
282
offset arg_length % cs->mb_minlength
283
cs Character set for 'str'
286
For real multi-byte, ascii incompatible charactser sets,
287
like UCS-2, add leading zeros if we have an incomplete character.
290
will automatically be converted into
298
bool String::copy_aligned(const char *str,uint32 arg_length, uint32 offset,
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);
305
uint32 aligned_length= arg_length + offset;
306
if (alloc(aligned_length))
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.
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;
324
bool String::set_or_copy_aligned(const char *str,uint32 arg_length,
327
/* How many bytes are in incomplete character */
328
uint32 offset= (arg_length % cs->mbminlen);
330
if (!offset) /* All characters are complete, just copy */
332
set(str, arg_length, cs);
335
return copy_aligned(str, arg_length, offset, cs);
338
/* Copy with charset convertion */
340
bool String::copy(const char *str, uint32 arg_length,
341
CHARSET_INFO *from_cs, CHARSET_INFO *to_cs, uint *errors)
344
if (!needs_conversion(arg_length, from_cs, to_cs, &offset))
347
return copy(str, arg_length, to_cs);
349
if ((from_cs == &my_charset_bin) && offset)
352
return copy_aligned(str, arg_length, offset, to_cs);
354
uint32 new_length= to_cs->mbmaxlen*arg_length;
355
if (alloc(new_length))
357
str_length=copy_and_convert((char*) Ptr, new_length, to_cs,
358
str, arg_length, from_cs, errors);
365
Set a string to the value of a latin1-string, keeping the original charset
369
str String of a simple charset (latin1)
370
arg_length Length of string
373
If string object is of a simple character set, set it to point to the
375
If not, make a copy and convert it to the new character set.
379
1 Could not allocate result buffer
383
bool String::set_ascii(const char *str, uint32 arg_length)
385
if (str_charset->mbminlen == 1)
387
set(str, arg_length, str_charset);
391
return copy(str, arg_length, &my_charset_latin1, str_charset, &dummy_errors);
395
/* This is used by mysql.cc */
397
bool String::fill(uint32 max_length,char fill_char)
399
if (str_length > max_length)
400
Ptr[str_length=max_length]=0;
403
if (realloc(max_length))
405
bfill(Ptr+str_length,max_length-str_length,fill_char);
406
str_length=max_length;
411
void String::strip_sp()
413
while (str_length && my_isspace(str_charset,Ptr[str_length-1]))
417
bool String::append(const String &s)
421
if (realloc(str_length+s.length()))
423
memcpy(Ptr+str_length,s.ptr(),s.length());
424
str_length+=s.length();
431
Append an ASCII string to the a string of the current character set
434
bool String::append(const char *s,uint32 arg_length)
440
For an ASCII incompatible string, e.g. UCS-2, we need to convert
442
if (str_charset->mbminlen > 1)
444
uint32 add_length=arg_length * str_charset->mbmaxlen;
446
if (realloc(str_length+ add_length))
448
str_length+= copy_and_convert(Ptr+str_length, add_length, str_charset,
449
s, arg_length, &my_charset_latin1,
455
For an ASCII compatinble string we can just append.
457
if (realloc(str_length+arg_length))
459
memcpy(Ptr+str_length,s,arg_length);
460
str_length+=arg_length;
466
Append a 0-terminated ASCII string
469
bool String::append(const char *s)
471
return append(s, strlen(s));
476
Append a string in the given charset to the string
477
with character set recoding
480
bool String::append(const char *s,uint32 arg_length, CHARSET_INFO *cs)
484
if (needs_conversion(arg_length, cs, str_charset, &dummy_offset))
486
uint32 add_length= arg_length / cs->mbminlen * str_charset->mbmaxlen;
488
if (realloc(str_length + add_length))
490
str_length+= copy_and_convert(Ptr+str_length, add_length, str_charset,
491
s, arg_length, cs, &dummy_errors);
495
if (realloc(str_length + arg_length))
497
memcpy(Ptr + str_length, s, arg_length);
498
str_length+= arg_length;
505
bool String::append(FILE* file, uint32 arg_length, myf my_flags)
507
if (realloc(str_length+arg_length))
509
if (my_fread(file, (uchar*) Ptr + str_length, arg_length, my_flags))
514
str_length+=arg_length;
519
bool String::append(IO_CACHE* file, uint32 arg_length)
521
if (realloc(str_length+arg_length))
523
if (my_b_read(file, (uchar*) Ptr + str_length, arg_length))
528
str_length+=arg_length;
532
bool String::append_with_prefill(const char *s,uint32 arg_length,
533
uint32 full_length, char fill_char)
535
int t_length= arg_length > full_length ? arg_length : full_length;
537
if (realloc(str_length + t_length))
539
t_length= full_length - arg_length;
542
bfill(Ptr+str_length, t_length, fill_char);
543
str_length=str_length + t_length;
545
append(s, arg_length);
549
uint32 String::numchars()
551
return str_charset->cset->numchars(str_charset, Ptr, Ptr+str_length);
554
int String::charpos(int i,uint32 offset)
558
return str_charset->cset->charpos(str_charset,Ptr+offset,Ptr+str_length,i);
561
int String::strstr(const String &s,uint32 offset)
563
if (s.length()+offset <= str_length)
566
return ((int) offset); // Empty string is always found
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();
575
if (*str++ == *search)
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;
589
** Search string from end. Offset is offset to the end of string
592
int String::strrstr(const String &s,uint32 offset)
594
if (s.length() <= offset && offset <= str_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;
601
const char *end=Ptr+s.length()-2;
602
const char *search_end=s.ptr()-1;
606
if (*str-- == *search)
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;
620
Replace substring with string
621
If wrong parameter or not enough memory, do nothing
624
bool String::replace(uint32 offset,uint32 arg_length,const String &to)
626
return replace(offset,arg_length,to.ptr(),to.length());
629
bool String::replace(uint32 offset,uint32 arg_length,
630
const char *to, uint32 to_length)
632
long diff = (long) to_length-(long) arg_length;
633
if (offset+arg_length <= str_length)
638
memcpy(Ptr+offset,to,to_length);
639
bmove(Ptr+offset+to_length,Ptr+offset+arg_length,
640
str_length-offset-arg_length);
646
if (realloc(str_length+(uint32) diff))
648
bmove_upp((uchar*) Ptr+str_length+diff, (uchar*) Ptr+str_length,
649
str_length-offset-arg_length);
652
memcpy(Ptr+offset,to,to_length);
654
str_length+=(uint32) diff;
660
// added by Holyfoot for "geometry" needs
661
int String::reserve(uint32 space_needed, uint32 grow_by)
663
if (Alloced_length < str_length + space_needed)
665
if (realloc(Alloced_length + max(space_needed, grow_by) - 1))
671
void String::qs_append(const char *str, uint32 len)
673
memcpy(Ptr + str_length, str, len + 1);
677
void String::qs_append(double d)
679
char *buff = Ptr + str_length;
680
str_length+= my_gcvt(d, MY_GCVT_ARG_DOUBLE, FLOATING_POINT_BUFFER - 1, buff, NULL);
683
void String::qs_append(double *d)
686
float8get(ld, (char*) d);
690
void String::qs_append(int i)
692
char *buff= Ptr + str_length;
693
char *end= int10_to_str(i, buff, -10);
694
str_length+= (int) (end-buff);
697
void String::qs_append(uint i)
699
char *buff= Ptr + str_length;
700
char *end= int10_to_str(i, buff, 10);
701
str_length+= (int) (end-buff);
705
Compare strings according to collation, without end space.
714
Normally this is case sensitive comparison
723
int sortcmp(const String *s,const String *t, CHARSET_INFO *cs)
725
return cs->coll->strnncollsp(cs,
726
(unsigned char *) s->ptr(),s->length(),
727
(unsigned char *) t->ptr(),t->length(), 0);
732
Compare strings byte by byte. End spaces are also compared.
740
Strings are compared as a stream of unsigned chars
749
int stringcmp(const String *s,const String *t)
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);
757
String *copy_if_not_alloced(String *to,String *from,uint32 from_length)
759
if (from->Alloced_length >= from_length)
761
if (from->alloced || !to || from == to)
763
(void) from->realloc(from_length);
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;
775
/****************************************************************************
777
****************************************************************************/
780
copy a string from one character set to another
785
to_cs Character set of result string
787
from_length Length of from string
788
from_cs From character set
791
'to' must be big enough as form_length * to_cs->mbmaxlen
794
length of bytes copied to 'to'
799
copy_and_convert(char *to, uint32 to_length, CHARSET_INFO *to_cs,
800
const char *from, uint32 from_length, CHARSET_INFO *from_cs,
805
const uchar *from_end= (const uchar*) from+from_length;
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;
814
if ((cnvres= (*mb_wc)(from_cs, &wc, (uchar*) from,
817
else if (cnvres == MY_CS_ILSEQ)
823
else if (cnvres > MY_CS_TOOSMALL)
826
A correct multibyte sequence detected
827
But it doesn't have Unicode mapping.
834
break; // Not enough characters
837
if ((cnvres= (*wc_mb)(to_cs, wc, (uchar*) to, to_end)) > 0)
839
else if (cnvres == MY_CS_ILUNI && wc != '?')
848
*errors= error_count;
849
return (uint32) (to - to_start);
853
void String::print(String *str)
855
char *st= (char*)Ptr, *end= st+str_length;
856
for (; st < end; st++)
862
str->append(STRING_WITH_LEN("\\\\"));
865
str->append(STRING_WITH_LEN("\\0"));
868
str->append(STRING_WITH_LEN("\\'"));
871
str->append(STRING_WITH_LEN("\\n"));
874
str->append(STRING_WITH_LEN("\\r"));
877
str->append(STRING_WITH_LEN("\\z"));
887
Exchange state of this object and argument.
893
Target string will contain state of this object and vice versa.
896
void String::swap(String &s)
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);