sql_string.cc 20 KB
Newer Older
1 2 3
/*
   Copyright (c) 2000-2007 MySQL AB, 2009 Sun Microsystems, Inc.
   Use is subject to license terms.
unknown's avatar
unknown committed
4 5 6

   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
7
   the Free Software Foundation; version 2 of the License.
unknown's avatar
unknown committed
8 9

   This program is distributed in the hope that it will be useful,
unknown's avatar
unknown committed
10
   but WITHOUT ANY WARRANTY; without even the implied warranty of
unknown's avatar
unknown committed
11 12 13 14 15
   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
16 17
   Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301  USA
*/
unknown's avatar
unknown committed
18 19 20

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

21
#ifdef USE_PRAGMA_IMPLEMENTATION
unknown's avatar
unknown committed
22 23 24
#pragma implementation				// gcc: Class implementation
#endif

25
#include <my_global.h>
unknown's avatar
unknown committed
26 27 28 29 30 31 32
#include <my_sys.h>
#include <m_string.h>
#include <m_ctype.h>
#ifdef HAVE_FCONVERT
#include <floatingpoint.h>
#endif

unknown's avatar
unknown committed
33 34 35 36 37
/*
  The following extern declarations are ok as these are interface functions
  required by the string function
*/

unknown's avatar
unknown committed
38 39 40 41 42 43 44 45 46 47 48 49
extern gptr sql_alloc(unsigned size);
extern void sql_element_free(void *ptr);

#include "sql_string.h"

/*****************************************************************************
** String functions
*****************************************************************************/

bool String::real_alloc(uint32 arg_length)
{
  arg_length=ALIGN_SIZE(arg_length+1);
50
  str_length=0;
unknown's avatar
unknown committed
51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76
  if (Alloced_length < arg_length)
  {
    free();
    if (!(Ptr=(char*) my_malloc(arg_length,MYF(MY_WME))))
      return TRUE;
    Alloced_length=arg_length;
    alloced=1;
  }
  Ptr[0]=0;
  return FALSE;
}


/*
** Check that string is big enough. Set string[alloc_length] to 0
** (for C functions)
*/

bool String::realloc(uint32 alloc_length)
{
  uint32 len=ALIGN_SIZE(alloc_length+1);
  if (Alloced_length < len)
  {
    char *new_ptr;
    if (alloced)
    {
77 78
      if (!(new_ptr= (char*) my_realloc(Ptr,len,MYF(MY_WME))))
        return TRUE;				// Signal error
unknown's avatar
unknown committed
79 80 81
    }
    else if ((new_ptr= (char*) my_malloc(len,MYF(MY_WME))))
    {
82 83
      if (str_length > len - 1)
        str_length= 0;
unknown's avatar
unknown committed
84 85
      if (str_length)				// Avoid bugs in memcpy on AIX
	memcpy(new_ptr,Ptr,str_length);
unknown's avatar
unknown committed
86 87 88 89 90
      new_ptr[str_length]=0;
      alloced=1;
    }
    else
      return TRUE;			// Signal error
91 92
    Ptr= new_ptr;
    Alloced_length= len;
unknown's avatar
unknown committed
93 94 95 96 97
  }
  Ptr[alloc_length]=0;			// This make other funcs shorter
  return FALSE;
}

98
bool String::set(longlong num, CHARSET_INFO *cs)
unknown's avatar
unknown committed
99
{
unknown's avatar
unknown committed
100 101 102
  uint l=20*cs->mbmaxlen+1;

  if (alloc(l))
unknown's avatar
unknown committed
103
    return TRUE;
unknown's avatar
unknown committed
104
  str_length=(uint32) (cs->cset->longlong10_to_str)(cs,Ptr,l,-10,num);
105
  str_charset=cs;
unknown's avatar
unknown committed
106 107 108
  return FALSE;
}

109
bool String::set(ulonglong num, CHARSET_INFO *cs)
unknown's avatar
unknown committed
110
{
unknown's avatar
unknown committed
111 112 113
  uint l=20*cs->mbmaxlen+1;

  if (alloc(l))
unknown's avatar
unknown committed
114
    return TRUE;
unknown's avatar
unknown committed
115
  str_length=(uint32) (cs->cset->longlong10_to_str)(cs,Ptr,l,10,num);
116
  str_charset=cs;
unknown's avatar
unknown committed
117 118 119
  return FALSE;
}

120
bool String::set(double num,uint decimals, CHARSET_INFO *cs)
unknown's avatar
unknown committed
121
{
122
  char buff[FLOATING_POINT_BUFFER];
unknown's avatar
unknown committed
123
  uint dummy_errors;
124 125

  str_charset=cs;
unknown's avatar
unknown committed
126 127
  if (decimals >= NOT_FIXED_DEC)
  {
128
    uint32 len= my_sprintf(buff,(buff, "%.15g",num));// Enough for a DATETIME
unknown's avatar
unknown committed
129
    return copy(buff, len, &my_charset_latin1, cs, &dummy_errors);
unknown's avatar
unknown committed
130 131 132 133 134 135
  }
#ifdef HAVE_FCONVERT
  int decpt,sign;
  char *pos,*to;

  VOID(fconvert(num,(int) decimals,&decpt,&sign,buff+1));
unknown's avatar
unknown committed
136
  if (!my_isdigit(&my_charset_latin1, buff[1]))
unknown's avatar
unknown committed
137 138 139 140 141 142 143
  {						// Nan or Inf
    pos=buff+1;
    if (sign)
    {
      buff[0]='-';
      pos=buff;
    }
unknown's avatar
unknown committed
144 145
    uint dummy_errors;
    return copy(pos,(uint32) strlen(pos), &my_charset_latin1, cs, &dummy_errors);
unknown's avatar
unknown committed
146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188
  }
  if (alloc((uint32) ((uint32) decpt+3+decimals)))
    return TRUE;
  to=Ptr;
  if (sign)
    *to++='-';

  pos=buff+1;
  if (decpt < 0)
  {					/* value is < 0 */
    *to++='0';
    if (!decimals)
      goto end;
    *to++='.';
    if ((uint32) -decpt > decimals)
      decpt= - (int) decimals;
    decimals=(uint32) ((int) decimals+decpt);
    while (decpt++ < 0)
      *to++='0';
  }
  else if (decpt == 0)
  {
    *to++= '0';
    if (!decimals)
      goto end;
    *to++='.';
  }
  else
  {
    while (decpt-- > 0)
      *to++= *pos++;
    if (!decimals)
      goto end;
    *to++='.';
  }
  while (decimals--)
    *to++= *pos++;

end:
  *to=0;
  str_length=(uint32) (to-Ptr);
  return FALSE;
#else
unknown's avatar
unknown committed
189 190
#ifdef HAVE_SNPRINTF
  buff[sizeof(buff)-1]=0;			// Safety
191 192 193
  int num_chars= snprintf(buff, sizeof(buff)-1, "%.*f",(int) decimals, num);
  DBUG_ASSERT(num_chars > 0);
  DBUG_ASSERT(num_chars < (int) sizeof(buff));
unknown's avatar
unknown committed
194 195 196
#else
  sprintf(buff,"%.*f",(int) decimals,num);
#endif
unknown's avatar
unknown committed
197 198
  return copy(buff,(uint32) strlen(buff), &my_charset_latin1, cs,
              &dummy_errors);
unknown's avatar
unknown committed
199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219
#endif
}


bool String::copy()
{
  if (!alloced)
  {
    Alloced_length=0;				// Force realloc
    return realloc(str_length);
  }
  return FALSE;
}

bool String::copy(const String &str)
{
  if (alloc(str.str_length))
    return TRUE;
  str_length=str.str_length;
  bmove(Ptr,str.Ptr,str_length);		// May be overlapping
  Ptr[str_length]=0;
220
  str_charset=str.str_charset;
unknown's avatar
unknown committed
221 222 223
  return FALSE;
}

224
bool String::copy(const char *str,uint32 arg_length, CHARSET_INFO *cs)
unknown's avatar
unknown committed
225 226 227
{
  if (alloc(arg_length))
    return TRUE;
unknown's avatar
unknown committed
228 229
  if ((str_length=arg_length))
    memcpy(Ptr,str,arg_length);
unknown's avatar
unknown committed
230
  Ptr[arg_length]=0;
231
  str_charset=cs;
unknown's avatar
unknown committed
232 233 234
  return FALSE;
}

unknown's avatar
unknown committed
235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251

/*
  Checks that the source string can be just copied to the destination string
  without conversion.

  SYNPOSIS

  needs_conversion()
  arg_length		Length of string to copy.
  from_cs		Character set to copy from
  to_cs			Character set to copy to
  uint32 *offset	Returns number of unaligned characters.

  RETURN
   0  No conversion needed
   1  Either character set conversion or adding leading  zeros
      (e.g. for UCS-2) must be done
252 253 254 255

  NOTE
  to_cs may be NULL for "no conversion" if the system variable
  character_set_results is NULL.
unknown's avatar
unknown committed
256 257 258 259 260 261 262 263
*/

bool String::needs_conversion(uint32 arg_length,
			      CHARSET_INFO *from_cs,
			      CHARSET_INFO *to_cs,
			      uint32 *offset)
{
  *offset= 0;
264 265
  if (!to_cs ||
      (to_cs == &my_charset_bin) || 
unknown's avatar
unknown committed
266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296 297 298 299 300 301
      (to_cs == from_cs) ||
      my_charset_same(from_cs, to_cs) ||
      ((from_cs == &my_charset_bin) &&
       (!(*offset=(arg_length % to_cs->mbminlen)))))
    return FALSE;
  return TRUE;
}


/*
  Copy a multi-byte character sets with adding leading zeros.

  SYNOPSIS

  copy_aligned()
  str			String to copy
  arg_length		Length of string. This should NOT be dividable with
			cs->mbminlen.
  offset		arg_length % cs->mb_minlength
  cs			Character set for 'str'

  NOTES
    For real multi-byte, ascii incompatible charactser sets,
    like UCS-2, add leading zeros if we have an incomplete character.
    Thus, 
      SELECT _ucs2 0xAA 
    will automatically be converted into
      SELECT _ucs2 0x00AA

  RETURN
    0  ok
    1  error
*/

bool String::copy_aligned(const char *str,uint32 arg_length, uint32 offset,
			  CHARSET_INFO *cs)
unknown's avatar
unknown committed
302
{
unknown's avatar
unknown committed
303 304 305
  /* How many bytes are in incomplete character */
  offset= cs->mbmaxlen - offset; /* How many zeros we should prepend */
  DBUG_ASSERT(offset && offset != cs->mbmaxlen);
unknown's avatar
unknown committed
306

unknown's avatar
unknown committed
307 308
  uint32 aligned_length= arg_length + offset;
  if (alloc(aligned_length))
unknown's avatar
unknown committed
309
    return TRUE;
unknown's avatar
unknown committed
310 311 312 313 314 315 316 317 318 319 320 321 322 323 324
  
  /*
    Note, this is only safe for little-endian UCS-2.
    If we add big-endian UCS-2 sometimes, this code
    will be more complicated. But it's OK for now.
  */
  bzero((char*) Ptr, offset);
  memcpy(Ptr + offset, str, arg_length);
  Ptr[aligned_length]=0;
  /* str_length is always >= 0 as arg_length is != 0 */
  str_length= aligned_length;
  str_charset= cs;
  return FALSE;
}

unknown's avatar
unknown committed
325

unknown's avatar
unknown committed
326 327 328 329 330
bool String::set_or_copy_aligned(const char *str,uint32 arg_length,
				 CHARSET_INFO *cs)
{
  /* How many bytes are in incomplete character */
  uint32 offset= (arg_length % cs->mbminlen); 
unknown's avatar
unknown committed
331
  
unknown's avatar
unknown committed
332
  if (!offset) /* All characters are complete, just copy */
unknown's avatar
unknown committed
333
  {
unknown's avatar
unknown committed
334 335 336 337 338
    set(str, arg_length, cs);
    return FALSE;
  }
  return copy_aligned(str, arg_length, offset, cs);
}
unknown's avatar
unknown committed
339

unknown's avatar
unknown committed
340 341 342 343 344 345 346 347 348 349 350 351 352 353 354
	/* Copy with charset convertion */

bool String::copy(const char *str, uint32 arg_length,
		  CHARSET_INFO *from_cs, CHARSET_INFO *to_cs, uint *errors)
{
  uint32 offset;
  if (!needs_conversion(arg_length, from_cs, to_cs, &offset))
  {
    *errors= 0;
    return copy(str, arg_length, to_cs);
  }
  if ((from_cs == &my_charset_bin) && offset)
  {
    *errors= 0;
    return copy_aligned(str, arg_length, offset, to_cs);
unknown's avatar
unknown committed
355
  }
unknown's avatar
unknown committed
356 357 358 359 360 361
  uint32 new_length= to_cs->mbmaxlen*arg_length;
  if (alloc(new_length))
    return TRUE;
  str_length=copy_and_convert((char*) Ptr, new_length, to_cs,
                              str, arg_length, from_cs, errors);
  str_charset=to_cs;
unknown's avatar
unknown committed
362 363 364
  return FALSE;
}

unknown's avatar
unknown committed
365 366 367 368 369 370 371 372 373 374 375 376 377 378 379 380 381 382 383 384 385 386 387 388 389 390 391 392 393 394 395 396

/*
  Set a string to the value of a latin1-string, keeping the original charset
  
  SYNOPSIS
    copy_or_set()
    str			String of a simple charset (latin1)
    arg_length		Length of string

  IMPLEMENTATION
    If string object is of a simple character set, set it to point to the
    given string.
    If not, make a copy and convert it to the new character set.

  RETURN
    0	ok
    1	Could not allocate result buffer

*/

bool String::set_ascii(const char *str, uint32 arg_length)
{
  if (str_charset->mbminlen == 1)
  {
    set(str, arg_length, str_charset);
    return 0;
  }
  uint dummy_errors;
  return copy(str, arg_length, &my_charset_latin1, str_charset, &dummy_errors);
}


unknown's avatar
unknown committed
397 398 399 400 401 402 403 404 405 406 407 408 409 410 411 412 413 414
/* This is used by mysql.cc */

bool String::fill(uint32 max_length,char fill_char)
{
  if (str_length > max_length)
    Ptr[str_length=max_length]=0;
  else
  {
    if (realloc(max_length))
      return TRUE;
    bfill(Ptr+str_length,max_length-str_length,fill_char);
    str_length=max_length;
  }
  return FALSE;
}

void String::strip_sp()
{
415
   while (str_length && my_isspace(str_charset,Ptr[str_length-1]))
unknown's avatar
unknown committed
416 417 418 419 420
    str_length--;
}

bool String::append(const String &s)
{
unknown's avatar
unknown committed
421 422 423 424 425 426 427
  if (s.length())
  {
    if (realloc(str_length+s.length()))
      return TRUE;
    memcpy(Ptr+str_length,s.ptr(),s.length());
    str_length+=s.length();
  }
unknown's avatar
unknown committed
428 429 430
  return FALSE;
}

unknown's avatar
unknown committed
431 432 433 434 435

/*
  Append an ASCII string to the a string of the current character set
*/

unknown's avatar
unknown committed
436 437
bool String::append(const char *s,uint32 arg_length)
{
unknown's avatar
unknown committed
438 439 440 441 442 443 444 445 446 447 448 449 450 451 452 453 454 455 456 457 458
  if (!arg_length)
    return FALSE;

  /*
    For an ASCII incompatible string, e.g. UCS-2, we need to convert
  */
  if (str_charset->mbminlen > 1)
  {
    uint32 add_length=arg_length * str_charset->mbmaxlen;
    uint dummy_errors;
    if (realloc(str_length+ add_length))
      return TRUE;
    str_length+= copy_and_convert(Ptr+str_length, add_length, str_charset,
				  s, arg_length, &my_charset_latin1,
                                  &dummy_errors);
    return FALSE;
  }

  /*
    For an ASCII compatinble string we can just append.
  */
unknown's avatar
unknown committed
459 460 461 462 463 464 465
  if (realloc(str_length+arg_length))
    return TRUE;
  memcpy(Ptr+str_length,s,arg_length);
  str_length+=arg_length;
  return FALSE;
}

unknown's avatar
unknown committed
466 467 468 469 470 471 472

/*
  Append a 0-terminated ASCII string
*/

bool String::append(const char *s)
{
473
  return append(s, (uint) strlen(s));
unknown's avatar
unknown committed
474 475 476 477 478 479 480 481 482 483 484 485 486 487 488 489 490 491 492 493 494 495 496 497 498 499 500 501 502 503 504 505
}


/*
  Append a string in the given charset to the string
  with character set recoding
*/

bool String::append(const char *s,uint32 arg_length, CHARSET_INFO *cs)
{
  uint32 dummy_offset;
  
  if (needs_conversion(arg_length, cs, str_charset, &dummy_offset))
  {
    uint32 add_length= arg_length / cs->mbminlen * str_charset->mbmaxlen;
    uint dummy_errors;
    if (realloc(str_length + add_length)) 
      return TRUE;
    str_length+= copy_and_convert(Ptr+str_length, add_length, str_charset,
				  s, arg_length, cs, &dummy_errors);
  }
  else
  {
    if (realloc(str_length + arg_length)) 
      return TRUE;
    memcpy(Ptr + str_length, s, arg_length);
    str_length+= arg_length;
  }
  return FALSE;
}


506
#ifdef TO_BE_REMOVED
unknown's avatar
unknown committed
507 508 509 510 511 512 513 514 515 516 517 518
bool String::append(FILE* file, uint32 arg_length, myf my_flags)
{
  if (realloc(str_length+arg_length))
    return TRUE;
  if (my_fread(file, (byte*) Ptr + str_length, arg_length, my_flags))
  {
    shrink(str_length);
    return TRUE;
  }
  str_length+=arg_length;
  return FALSE;
}
519 520 521 522 523 524 525 526 527 528 529 530 531 532
#endif

bool String::append(IO_CACHE* file, uint32 arg_length)
{
  if (realloc(str_length+arg_length))
    return TRUE;
  if (my_b_read(file, (byte*) Ptr + str_length, arg_length))
  {
    shrink(str_length);
    return TRUE;
  }
  str_length+=arg_length;
  return FALSE;
}
unknown's avatar
unknown committed
533

unknown's avatar
unknown committed
534 535
bool String::append_with_prefill(const char *s,uint32 arg_length,
		 uint32 full_length, char fill_char)
unknown's avatar
unknown committed
536
{
unknown's avatar
unknown committed
537 538 539 540 541 542
  int t_length= arg_length > full_length ? arg_length : full_length;

  if (realloc(str_length + t_length))
    return TRUE;
  t_length= full_length - arg_length;
  if (t_length > 0)
unknown's avatar
unknown committed
543
  {
unknown's avatar
unknown committed
544 545
    bfill(Ptr+str_length, t_length, fill_char);
    str_length=str_length + t_length;
unknown's avatar
unknown committed
546
  }
unknown's avatar
unknown committed
547 548 549 550 551 552 553
  append(s, arg_length);
  return FALSE;
}

uint32 String::numchars()
{
  return str_charset->cset->numchars(str_charset, Ptr, Ptr+str_length);
unknown's avatar
unknown committed
554 555 556 557
}

int String::charpos(int i,uint32 offset)
{
unknown's avatar
unknown committed
558
  if (i <= 0)
unknown's avatar
unknown committed
559
    return i;
unknown's avatar
unknown committed
560
  return str_charset->cset->charpos(str_charset,Ptr+offset,Ptr+str_length,i);
unknown's avatar
unknown committed
561 562 563 564 565 566 567
}

int String::strstr(const String &s,uint32 offset)
{
  if (s.length()+offset <= str_length)
  {
    if (!s.length())
unknown's avatar
unknown committed
568
      return ((int) offset);	// Empty string is always found
unknown's avatar
unknown committed
569 570 571 572 573

    register const char *str = Ptr+offset;
    register const char *search=s.ptr();
    const char *end=Ptr+str_length-s.length()+1;
    const char *search_end=s.ptr()+s.length();
574
skip:
unknown's avatar
unknown committed
575 576 577 578 579 580 581
    while (str != end)
    {
      if (*str++ == *search)
      {
	register char *i,*j;
	i=(char*) str; j=(char*) search+1;
	while (j != search_end)
582
	  if (*i++ != *j++) goto skip;
unknown's avatar
unknown committed
583 584 585 586 587 588 589 590 591 592 593 594 595 596 597 598 599 600 601 602 603 604
	return (int) (str-Ptr) -1;
      }
    }
  }
  return -1;
}

/*
** Search string from end. Offset is offset to the end of string
*/

int String::strrstr(const String &s,uint32 offset)
{
  if (s.length() <= offset && offset <= str_length)
  {
    if (!s.length())
      return offset;				// Empty string is always found
    register const char *str = Ptr+offset-1;
    register const char *search=s.ptr()+s.length()-1;

    const char *end=Ptr+s.length()-2;
    const char *search_end=s.ptr()-1;
605
skip:
unknown's avatar
unknown committed
606 607 608 609 610 611 612
    while (str != end)
    {
      if (*str-- == *search)
      {
	register char *i,*j;
	i=(char*) str; j=(char*) search-1;
	while (j != search_end)
613
	  if (*i-- != *j--) goto skip;
unknown's avatar
unknown committed
614 615 616 617 618 619 620 621
	return (int) (i-Ptr) +1;
      }
    }
  }
  return -1;
}

/*
622 623
  Replace substring with string
  If wrong parameter or not enough memory, do nothing
unknown's avatar
unknown committed
624 625 626 627
*/

bool String::replace(uint32 offset,uint32 arg_length,const String &to)
{
unknown's avatar
unknown committed
628 629 630 631
  return replace(offset,arg_length,to.ptr(),to.length());
}

bool String::replace(uint32 offset,uint32 arg_length,
632
                     const char *to, uint32 to_length)
unknown's avatar
unknown committed
633
{
634
  long diff = (long) to_length-(long) arg_length;
unknown's avatar
unknown committed
635 636 637 638
  if (offset+arg_length <= str_length)
  {
    if (diff < 0)
    {
639 640 641
      if (to_length)
	memcpy(Ptr+offset,to,to_length);
      bmove(Ptr+offset+to_length,Ptr+offset+arg_length,
unknown's avatar
unknown committed
642 643 644 645 646 647 648 649 650 651 652
	    str_length-offset-arg_length);
    }
    else
    {
      if (diff)
      {
	if (realloc(str_length+(uint32) diff))
	  return TRUE;
	bmove_upp(Ptr+str_length+diff,Ptr+str_length,
		  str_length-offset-arg_length);
      }
653 654
      if (to_length)
	memcpy(Ptr+offset,to,to_length);
unknown's avatar
unknown committed
655 656 657 658 659 660
    }
    str_length+=(uint32) diff;
  }
  return FALSE;
}

unknown's avatar
unknown committed
661

662 663 664 665 666 667 668 669 670 671 672
// added by Holyfoot for "geometry" needs
int String::reserve(uint32 space_needed, uint32 grow_by)
{
  if (Alloced_length < str_length + space_needed)
  {
    if (realloc(Alloced_length + max(space_needed, grow_by) - 1))
      return TRUE;
  }
  return FALSE;
}

unknown's avatar
unknown committed
673
void String::qs_append(const char *str, uint32 len)
674 675 676 677 678 679 680 681
{
  memcpy(Ptr + str_length, str, len + 1);
  str_length += len;
}

void String::qs_append(double d)
{
  char *buff = Ptr + str_length;
682
  str_length+= my_sprintf(buff, (buff, "%.15g", d));
683 684 685 686 687
}

void String::qs_append(double *d)
{
  double ld;
688
  float8get(ld, (char*) d);
689 690 691
  qs_append(ld);
}

unknown's avatar
unknown committed
692
void String::qs_append(int i)
693
{
unknown's avatar
unknown committed
694 695 696
  char *buff= Ptr + str_length;
  char *end= int10_to_str(i, buff, -10);
  str_length+= (int) (end-buff);
697 698
}

unknown's avatar
unknown committed
699
void String::qs_append(uint i)
unknown's avatar
unknown committed
700
{
unknown's avatar
unknown committed
701 702 703
  char *buff= Ptr + str_length;
  char *end= int10_to_str(i, buff, 10);
  str_length+= (int) (end-buff);
unknown's avatar
unknown committed
704 705
}

unknown's avatar
unknown committed
706 707 708 709 710 711 712 713 714 715 716
/*
  Compare strings according to collation, without end space.

  SYNOPSIS
    sortcmp()
    s		First string
    t		Second string
    cs		Collation

  NOTE:
    Normally this is case sensitive comparison
unknown's avatar
unknown committed
717

unknown's avatar
unknown committed
718 719 720 721 722 723 724 725
  RETURN
  < 0	s < t
  0	s == t
  > 0	s > t
*/


int sortcmp(const String *s,const String *t, CHARSET_INFO *cs)
unknown's avatar
unknown committed
726
{
unknown's avatar
unknown committed
727 728 729 730
 return cs->coll->strnncollsp(cs,
                              (unsigned char *) s->ptr(),s->length(),
                              (unsigned char *) t->ptr(),t->length(), 0);
}
unknown's avatar
unknown committed
731

unknown's avatar
unknown committed
732 733 734 735 736 737 738 739 740 741 742 743 744 745 746 747 748 749 750 751 752 753 754 755

/*
  Compare strings byte by byte. End spaces are also compared.

  SYNOPSIS
    stringcmp()
    s		First string
    t		Second string

  NOTE:
    Strings are compared as a stream of unsigned chars

  RETURN
  < 0	s < t
  0	s == t
  > 0	s > t
*/


int stringcmp(const String *s,const String *t)
{
  uint32 s_len=s->length(),t_len=t->length(),len=min(s_len,t_len);
  int cmp= memcmp(s->ptr(), t->ptr(), len);
  return (cmp) ? cmp : (int) (s_len - t_len);
unknown's avatar
unknown committed
756 757 758 759 760 761 762 763 764 765 766 767 768 769
}


String *copy_if_not_alloced(String *to,String *from,uint32 from_length)
{
  if (from->Alloced_length >= from_length)
    return from;
  if (from->alloced || !to || from == to)
  {
    (void) from->realloc(from_length);
    return from;
  }
  if (to->realloc(from_length))
    return from;				// Actually an error
unknown's avatar
unknown committed
770 771
  if ((to->str_length=min(from->str_length,from_length)))
    memcpy(to->Ptr,from->Ptr,to->str_length);
772
  to->str_charset=from->str_charset;
unknown's avatar
unknown committed
773 774 775
  return to;
}

776

unknown's avatar
unknown committed
777 778 779
/****************************************************************************
  Help functions
****************************************************************************/
780

unknown's avatar
unknown committed
781 782 783 784 785 786 787 788 789 790 791 792 793 794 795 796 797 798 799 800 801 802 803 804 805 806 807 808 809 810 811 812 813 814 815 816 817 818 819 820 821 822 823 824 825 826
/*
  copy a string from one character set to another
  
  SYNOPSIS
    copy_and_convert()
    to			Store result here
    to_cs		Character set of result string
    from		Copy from here
    from_length		Length of from string
    from_cs		From character set

  NOTES
    'to' must be big enough as form_length * to_cs->mbmaxlen

  RETURN
    length of bytes copied to 'to'
*/


uint32
copy_and_convert(char *to, uint32 to_length, CHARSET_INFO *to_cs, 
                 const char *from, uint32 from_length, CHARSET_INFO *from_cs,
                 uint *errors)
{
  int         cnvres;
  my_wc_t     wc;
  const uchar *from_end= (const uchar*) from+from_length;
  char *to_start= to;
  uchar *to_end= (uchar*) to+to_length;
  int (*mb_wc)(struct charset_info_st *, my_wc_t *, const uchar *,
	       const uchar *) = from_cs->cset->mb_wc;
  int (*wc_mb)(struct charset_info_st *, my_wc_t, uchar *s, uchar *e)=
    to_cs->cset->wc_mb;
  uint error_count= 0;

  while (1)
  {
    if ((cnvres= (*mb_wc)(from_cs, &wc, (uchar*) from,
				      from_end)) > 0)
      from+= cnvres;
    else if (cnvres == MY_CS_ILSEQ)
    {
      error_count++;
      from++;
      wc= '?';
    }
827 828 829 830 831 832 833 834 835 836
    else if (cnvres > MY_CS_TOOSMALL)
    {
      /*
        A correct multibyte sequence detected
        But it doesn't have Unicode mapping.
      */
      error_count++;
      from+= (-cnvres);
      wc= '?';
    }
unknown's avatar
unknown committed
837
    else
838
      break;  // Not enough characters
unknown's avatar
unknown committed
839 840 841 842 843 844 845 846 847 848 849 850 851 852 853 854 855 856 857 858 859 860 861 862 863 864 865

outp:
    if ((cnvres= (*wc_mb)(to_cs, wc, (uchar*) to, to_end)) > 0)
      to+= cnvres;
    else if (cnvres == MY_CS_ILUNI && wc != '?')
    {
      error_count++;
      wc= '?';
      goto outp;
    }
    else
      break;
  }
  *errors= error_count;
  return (uint32) (to - to_start);
}


void String::print(String *str)
{
  char *st= (char*)Ptr, *end= st+str_length;
  for (; st < end; st++)
  {
    uchar c= *st;
    switch (c)
    {
    case '\\':
866
      str->append(STRING_WITH_LEN("\\\\"));
unknown's avatar
unknown committed
867 868
      break;
    case '\0':
869
      str->append(STRING_WITH_LEN("\\0"));
unknown's avatar
unknown committed
870 871
      break;
    case '\'':
872
      str->append(STRING_WITH_LEN("\\'"));
unknown's avatar
unknown committed
873 874
      break;
    case '\n':
875
      str->append(STRING_WITH_LEN("\\n"));
unknown's avatar
unknown committed
876 877
      break;
    case '\r':
878
      str->append(STRING_WITH_LEN("\\r"));
unknown's avatar
unknown committed
879 880
      break;
    case 26: //Ctrl-Z
881
      str->append(STRING_WITH_LEN("\\z"));
unknown's avatar
unknown committed
882 883 884 885 886 887 888 889 890 891 892 893 894 895 896 897 898 899 900 901 902 903 904 905 906 907
      break;
    default:
      str->append(c);
    }
  }
}


/*
  Exchange state of this object and argument.

  SYNOPSIS
    String::swap()

  RETURN
    Target string will contain state of this object and vice versa.
*/

void String::swap(String &s)
{
  swap_variables(char *, Ptr, s.Ptr);
  swap_variables(uint32, str_length, s.str_length);
  swap_variables(uint32, Alloced_length, s.Alloced_length);
  swap_variables(bool, alloced, s.alloced);
  swap_variables(CHARSET_INFO*, str_charset, s.str_charset);
}