~drizzle-trunk/drizzle/development

/* Copyright (C) 2000 MySQL AB

   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

   the Free Software Foundation; version 2 of the License.

   This program is distributed in the hope that it will be useful,

   but WITHOUT ANY WARRANTY; without even the implied warranty of

   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

   Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA */

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

#include "config.h"

#include "drizzled/internal/my_sys.h"

#include "drizzled/internal/m_string.h"

#include "drizzled/charset.h"

#include "drizzled/global_charset_info.h"

#include <algorithm>

#include "drizzled/sql_string.h"

using namespace std;

namespace drizzled

{

// Converstion functions to and from std::string.

std::string String_to_std_string(String const& s)

{

   return std::string(s.ptr(), s.length());

}

String* set_String_from_std_string(String* s, std::string const& cs)

{

   s->set_ascii(cs.c_str(), cs.length());

   s->copy();

   return s;

}

/*****************************************************************************

** String functions

*****************************************************************************/

String::String()

  : Ptr(NULL),

    str_length(0),

    Alloced_length(0),

    alloced(false),

    str_charset(&my_charset_bin)

{ }

String::String(uint32_t length_arg)

  : Ptr(NULL),

    str_length(0),

    Alloced_length(0),

    alloced(false),

    str_charset(&my_charset_bin)

{

  (void) real_alloc(length_arg);

}

String::String(const char *str, const CHARSET_INFO * const cs)

  : Ptr(const_cast<char *>(str)),

    str_length(static_cast<uint32_t>(strlen(str))),

    Alloced_length(0),

    alloced(false),

    str_charset(cs)

{ }

String::String(const char *str, uint32_t len, const CHARSET_INFO * const cs)

  : Ptr(const_cast<char *>(str)),

    str_length(len),

    Alloced_length(0),

    alloced(false),

    str_charset(cs)

{ }

String::String(char *str, uint32_t len, const CHARSET_INFO * const cs)

  : Ptr(str),

    str_length(len),

    Alloced_length(len),

    alloced(false),

    str_charset(cs)

{ }

String::String(const String &str)

  : Ptr(str.Ptr),

    str_length(str.str_length),

    Alloced_length(str.Alloced_length),

    alloced(false),

    str_charset(str.str_charset)

{ }

void *String::operator new(size_t size, memory::Root *mem_root)

{

  return mem_root->alloc_root(static_cast<uint32_t>(size));

}

String::~String() { free(); }

bool String::real_alloc(uint32_t arg_length)

{

  arg_length=ALIGN_SIZE(arg_length+1);

  str_length=0;

  if (Alloced_length < arg_length)

  {

    free();

    if (!(Ptr=(char*) malloc(arg_length)))

      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_t alloc_length)

{

  uint32_t len=ALIGN_SIZE(alloc_length+1);

  if (Alloced_length < len)

  {

    char *new_ptr;

    if (alloced)

    {

      if ((new_ptr= (char*) ::realloc(Ptr,len)))

      {

	Ptr=new_ptr;

	Alloced_length=len;

      }

      else

	return true;				// Signal error

    }

    else if ((new_ptr= (char*) malloc(len)))

    {

      if (str_length)				// Avoid bugs in memcpy on AIX

	memcpy(new_ptr,Ptr,str_length);

      new_ptr[str_length]=0;

      Ptr=new_ptr;

      Alloced_length=len;

      alloced=1;

    }

    else

      return true;			// Signal error

  }

  Ptr[alloc_length]=0;			// This make other funcs shorter

  return false;

}

bool String::set_int(int64_t num, bool unsigned_flag, const CHARSET_INFO * const cs)

{

  uint32_t l=20*cs->mbmaxlen+1;

  int base= unsigned_flag ? 10 : -10;

  if (alloc(l))

    return true;

  str_length=(uint32_t) (cs->cset->int64_t10_to_str)(cs,Ptr,l,base,num);

  str_charset=cs;

  return false;

}

bool String::set_real(double num,uint32_t decimals, const CHARSET_INFO * const cs)

{

  char buff[FLOATING_POINT_BUFFER];

  uint32_t dummy_errors;

  size_t len;

  str_charset=cs;

  if (decimals >= NOT_FIXED_DEC)

  {

    len= internal::my_gcvt(num,

                           internal::MY_GCVT_ARG_DOUBLE,

                           sizeof(buff) - 1, buff, NULL);

    return copy(buff, len, &my_charset_utf8_general_ci, cs, &dummy_errors);

  }

  len= internal::my_fcvt(num, decimals, buff, NULL);

  return copy(buff, (uint32_t) len, &my_charset_utf8_general_ci, cs,

              &dummy_errors);

}

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;

  memmove(Ptr, str.Ptr, str_length);		// May be overlapping

  Ptr[str_length]=0;

  str_charset=str.str_charset;

  return false;

}

bool String::copy(const char *str,uint32_t arg_length, const CHARSET_INFO * const cs)

{

  if (alloc(arg_length))

    return true;

  if ((str_length=arg_length))

    memcpy(Ptr,str,arg_length);

  Ptr[arg_length]=0;

  str_charset=cs;

  return false;

}

/*

  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_t *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

  NOTE

  to_cs may be NULL for "no conversion" if the system variable

  character_set_results is NULL.

*/

bool String::needs_conversion(uint32_t arg_length,

			      const CHARSET_INFO * const from_cs,

			      const CHARSET_INFO * const to_cs,

			      uint32_t *offset)

{

  *offset= 0;

  if (!to_cs ||

      (to_cs == &my_charset_bin) ||

      (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;

}

bool String::set_or_copy_aligned(const char *str,uint32_t arg_length,

                                 const CHARSET_INFO * const cs)

{

  /* How many bytes are in incomplete character */

  uint32_t offset= (arg_length % cs->mbminlen);

  assert(!offset); /* All characters are complete, just copy */

  set(str, arg_length, cs);

  return false;

}

	/* Copy with charset conversion */

bool String::copy(const char *str, uint32_t arg_length,

		          const CHARSET_INFO * const,

				  const CHARSET_INFO * const to_cs, uint32_t *errors)

{

  *errors= 0;

  return copy(str, arg_length, to_cs);

}

/*

  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_t arg_length)

{

  if (str_charset->mbminlen == 1)

  {

    set(str, arg_length, str_charset);

    return 0;

  }

  uint32_t dummy_errors;

  return copy(str, arg_length, &my_charset_utf8_general_ci, str_charset, &dummy_errors);

}

bool String::append(const String &s)

{

  if (s.length())

  {

    if (realloc(str_length+s.length()))

      return true;

    memcpy(Ptr+str_length,s.ptr(),s.length());

    str_length+=s.length();

  }

  return false;

}

/*

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

*/

bool String::append(const char *s,uint32_t arg_length)

{

  if (!arg_length)

    return false;

  /*

    For an ASCII compatinble string we can just append.

  */

  if (realloc(str_length+arg_length))

    return true;

  memcpy(Ptr+str_length,s,arg_length);

  str_length+=arg_length;

  return false;

}

/*

  Append a 0-terminated ASCII string

*/

bool String::append(const char *s)

{

  return append(s, strlen(s));

}

/*

  Append a string in the given charset to the string

  with character set recoding

*/

bool String::append(const char *s,uint32_t arg_length, const CHARSET_INFO * const)

{

  if (realloc(str_length + arg_length))

    return true;

  memcpy(Ptr + str_length, s, arg_length);

  str_length+= arg_length;

  return false;

}

bool String::append_with_prefill(const char *s,uint32_t arg_length,

		 uint32_t full_length, char fill_char)

{

  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)

  {

    memset(Ptr+str_length, fill_char, t_length);

    str_length=str_length + t_length;

  }

  append(s, arg_length);

  return false;

}

uint32_t String::numchars()

{

  return str_charset->cset->numchars(str_charset, Ptr, Ptr+str_length);

}

int String::charpos(int i,uint32_t offset)

{

  if (i <= 0)

    return i;

  return str_charset->cset->charpos(str_charset,Ptr+offset,Ptr+str_length,i);

}

int String::strstr(const String &s,uint32_t offset)

{

  if (s.length()+offset <= str_length)

  {

    if (!s.length())

      return ((int) offset);	// Empty string is always found

    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();

skip:

    while (str != end)

    {

      if (*str++ == *search)

      {

	register char *i,*j;

	i=(char*) str; j=(char*) search+1;

	while (j != search_end)

	  if (*i++ != *j++) goto skip;

	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_t 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;

skip:

    while (str != end)

    {

      if (*str-- == *search)

      {

	register char *i,*j;

	i=(char*) str; j=(char*) search-1;

	while (j != search_end)

	  if (*i-- != *j--) goto skip;

	return (int) (i-Ptr) +1;

      }

    }

  }

  return -1;

}

/*

  Replace substring with string

  If wrong parameter or not enough memory, do nothing

*/

bool String::replace(uint32_t offset,uint32_t arg_length,const String &to)

{

  return replace(offset,arg_length,to.ptr(),to.length());

}

bool String::replace(uint32_t offset,uint32_t arg_length,

                     const char *to, uint32_t to_length)

{

  long diff = (long) to_length-(long) arg_length;

  if (offset+arg_length <= str_length)

  {

    if (diff < 0)

    {

      if (to_length)

	memcpy(Ptr+offset,to,to_length);

      memmove(Ptr+offset+to_length, Ptr+offset+arg_length,

              str_length-offset-arg_length);

    }

    else

    {

      if (diff)

      {

	if (realloc(str_length+(uint32_t) diff))

	  return true;

	internal::bmove_upp((unsigned char*) Ptr+str_length+diff,

                            (unsigned char*) Ptr+str_length,

                            str_length-offset-arg_length);

      }

      if (to_length)

	memcpy(Ptr+offset,to,to_length);

    }

    str_length+=(uint32_t) diff;

  }

  return false;

}

/*

  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

  RETURN

  < 0	s < t

  0	s == t

  > 0	s > t

*/

int sortcmp(const String *s,const String *t, const CHARSET_INFO * const cs)

{

 return cs->coll->strnncollsp(cs,

                              (unsigned char *) s->ptr(),s->length(),

                              (unsigned char *) t->ptr(),t->length(), 0);

}

/*

  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_t 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);

}

String *copy_if_not_alloced(String *to,String *from,uint32_t 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

  if ((to->str_length= min(from->str_length,from_length)))

    memcpy(to->Ptr,from->Ptr,to->str_length);

  to->str_charset=from->str_charset;

  return to;

}

/****************************************************************************

  Help functions

****************************************************************************/

/*

  copy a string,

  with optional character set conversion,

  with optional left padding (for binary -> UCS2 conversion)

  SYNOPSIS

    well_formed_copy_nchars()

    to			     Store result here

    to_length                Maxinum length of "to" string

    to_cs		     Character set of "to" string

    from		     Copy from here

    from_length		     Length of from string

    from_cs		     From character set

    nchars                   Copy not more that nchars characters

    well_formed_error_pos    Return position when "from" is not well formed

                             or NULL otherwise.

    cannot_convert_error_pos Return position where a not convertable

                             character met, or NULL otherwise.

    from_end_pos             Return position where scanning of "from"

                             string stopped.

  NOTES

  RETURN

    length of bytes copied to 'to'

*/

uint32_t

well_formed_copy_nchars(const CHARSET_INFO * const to_cs,

                        char *to, uint32_t to_length,

                        const CHARSET_INFO * const from_cs,

                        const char *from, uint32_t from_length,

                        uint32_t nchars,

                        const char **well_formed_error_pos,

                        const char **cannot_convert_error_pos,

                        const char **from_end_pos)

{

  uint32_t res;

  assert((to_cs == &my_charset_bin) ||

         (from_cs == &my_charset_bin) ||

         (to_cs == from_cs) ||

         my_charset_same(from_cs, to_cs));

  if (to_length < to_cs->mbminlen || !nchars)

  {

    *from_end_pos= from;

    *cannot_convert_error_pos= NULL;

    *well_formed_error_pos= NULL;

    return 0;

  }

  if (to_cs == &my_charset_bin)

  {

    res= min(min(nchars, to_length), from_length);

    memmove(to, from, res);

    *from_end_pos= from + res;

    *well_formed_error_pos= NULL;

    *cannot_convert_error_pos= NULL;

  }

  else

  {

    int well_formed_error;

    uint32_t from_offset;

    if ((from_offset= (from_length % to_cs->mbminlen)) &&

        (from_cs == &my_charset_bin))

    {

      /*

        Copying from BINARY to UCS2 needs to prepend zeros sometimes:

        INSERT INTO t1 (ucs2_column) VALUES (0x01);

        0x01 -> 0x0001

      */

      uint32_t pad_length= to_cs->mbminlen - from_offset;

      memset(to, 0, pad_length);

      memmove(to + pad_length, from, from_offset);

      nchars--;

      from+= from_offset;

      from_length-= from_offset;

      to+= to_cs->mbminlen;

      to_length-= to_cs->mbminlen;

    }

    set_if_smaller(from_length, to_length);

    res= to_cs->cset->well_formed_len(to_cs, from, from + from_length,

                                      nchars, &well_formed_error);

    memmove(to, from, res);

    *from_end_pos= from + res;

    *well_formed_error_pos= well_formed_error ? from + res : NULL;

    *cannot_convert_error_pos= NULL;

    if (from_offset)

      res+= to_cs->mbminlen;

  }

  return res;

}

void String::print(String *str)

{

  char *st= (char*)Ptr, *end= st+str_length;

  for (; st < end; st++)

  {

    unsigned char c= *st;

    switch (c)

    {

    case '\\':

      str->append("\\\\", sizeof("\\\\")-1);

      break;

    case '\0':

      str->append("\\0", sizeof("\\0")-1);

      break;

    case '\'':