~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 */

#include "config.h"

#include "drizzled/internal/m_string.h"

#include "drizzled/charset_info.h"

#include <algorithm>

using namespace std;

namespace drizzled

{

size_t my_caseup_str_mb(const CHARSET_INFO * const  cs, char *str)

{

  register uint32_t l;

  register unsigned char *map= cs->to_upper;

  char *str_orig= str;

  while (*str)

  {

    /* Pointing after the '\0' is safe here. */

    if ((l= my_ismbchar(cs, str, str + cs->mbmaxlen)))

      str+= l;

    else

    {

      *str= (char) map[(unsigned char)*str];

      str++;

    }

  }

  return (size_t) (str - str_orig);

}

size_t my_casedn_str_mb(const CHARSET_INFO * const  cs, char *str)

{

  register uint32_t l;

  register unsigned char *map= cs->to_lower;

  char *str_orig= str;

  while (*str)

  {

    /* Pointing after the '\0' is safe here. */

    if ((l= my_ismbchar(cs, str, str + cs->mbmaxlen)))

      str+= l;

    else

    {

      *str= (char) map[(unsigned char)*str];

      str++;

    }

  }

  return (size_t) (str - str_orig);

}

size_t my_caseup_mb(const CHARSET_INFO * const  cs, char *src, size_t srclen,

                    char *dst, size_t dstlen)

{

#ifdef NDEBUG

  (void)dst;

  (void)dstlen;

#endif

  register uint32_t l;

  register char *srcend= src + srclen;

  register unsigned char *map= cs->to_upper;

  assert(src == dst && srclen == dstlen);

  while (src < srcend)

  {

    if ((l=my_ismbchar(cs, src, srcend)))

      src+= l;

    else

    {

      *src=(char) map[(unsigned char) *src];

      src++;

    }

  }

  return srclen;

}

size_t my_casedn_mb(const CHARSET_INFO * const  cs, char *src, size_t srclen,

                    char *dst, size_t dstlen)

{

#ifdef NDEBUG

  (void)dst;

  (void)dstlen;

#endif

  register uint32_t l;

  register char *srcend= src + srclen;

  register unsigned char *map=cs->to_lower;

  assert(src == dst && srclen == dstlen);

  while (src < srcend)

  {

    if ((l= my_ismbchar(cs, src, srcend)))

      src+= l;

    else

    {

      *src= (char) map[(unsigned char)*src];

      src++;

    }

  }

  return srclen;

}

/*

  my_strcasecmp_mb() returns 0 if strings are equal, non-zero otherwise.

 */

int my_strcasecmp_mb(const CHARSET_INFO * const  cs,const char *s, const char *t)

{

  register uint32_t l;

  register unsigned char *map=cs->to_upper;

  while (*s && *t)

  {

    /* Pointing after the '\0' is safe here. */

    if ((l=my_ismbchar(cs, s, s + cs->mbmaxlen)))

    {

      while (l--)

        if (*s++ != *t++)

          return 1;

    }

    else if (my_mbcharlen(cs, *t) > 1)

      return 1;

    else if (map[(unsigned char) *s++] != map[(unsigned char) *t++])

      return 1;

  }

  /* At least one of '*s' and '*t' is zero here. */

  return (*t != *s);

}

/*

** Compare string against string with wildcard

**	0 if matched

**	-1 if not matched with wildcard

**	 1 if matched with wildcard

*/

#define INC_PTR(cs,A,B) A+=(my_ismbchar(cs,A,B) ? my_ismbchar(cs,A,B) : 1)

#define likeconv(s,A) (unsigned char) (s)->sort_order[(unsigned char) (A)]

int my_wildcmp_mb(const CHARSET_INFO * const cs,

		  const char *str,const char *str_end,

		  const char *wildstr,const char *wildend,

		  int escape, int w_one, int w_many)

{

  int result= -1;				/* Not found, using wildcards */

  while (wildstr != wildend)

  {

    while (*wildstr != w_many && *wildstr != w_one)

    {

      int l;

      if (*wildstr == escape && wildstr+1 != wildend)

	wildstr++;

      if ((l = my_ismbchar(cs, wildstr, wildend)))

      {

	  if (str+l > str_end || memcmp(str, wildstr, l) != 0)

	      return 1;

	  str += l;

	  wildstr += l;

      }

      else

      if (str == str_end || likeconv(cs,*wildstr++) != likeconv(cs,*str++))

	return(1);				/* No match */

      if (wildstr == wildend)

	return (str != str_end);		/* Match if both are at end */

      result=1;					/* Found an anchor char */

    }

    if (*wildstr == w_one)

    {

      do

      {

	if (str == str_end)			/* Skip one char if possible */

	  return (result);

	INC_PTR(cs,str,str_end);

      } while (++wildstr < wildend && *wildstr == w_one);

      if (wildstr == wildend)

	break;

    }

    if (*wildstr == w_many)

    {						/* Found w_many */

      unsigned char cmp;

      const char* mb = wildstr;

      int mb_len=0;

      wildstr++;

      /* Remove any '%' and '_' from the wild search string */

      for (; wildstr != wildend ; wildstr++)

      {

	if (*wildstr == w_many)

	  continue;

	if (*wildstr == w_one)

	{

	  if (str == str_end)

	    return (-1);

	  INC_PTR(cs,str,str_end);

	  continue;

	}

	break;					/* Not a wild character */

      }

      if (wildstr == wildend)

	return(0);				/* Ok if w_many is last */

      if (str == str_end)

	return -1;

      if ((cmp= *wildstr) == escape && wildstr+1 != wildend)

	cmp= *++wildstr;

      mb=wildstr;

      mb_len= my_ismbchar(cs, wildstr, wildend);

      INC_PTR(cs,wildstr,wildend);		/* This is compared trough cmp */

      cmp=likeconv(cs,cmp);

      do

      {

        for (;;)

        {

          if (str >= str_end)

            return -1;

          if (mb_len)

          {

            if (str+mb_len <= str_end && memcmp(str, mb, mb_len) == 0)

            {

              str += mb_len;

              break;

            }

          }

          else if (!my_ismbchar(cs, str, str_end) &&

                   likeconv(cs,*str) == cmp)

          {

            str++;

            break;

          }

          INC_PTR(cs,str, str_end);

        }

	{

	  int tmp=my_wildcmp_mb(cs,str,str_end,wildstr,wildend,escape,w_one,

                                w_many);

	  if (tmp <= 0)

	    return (tmp);

	}

      } while (str != str_end && wildstr[0] != w_many);

      return(-1);

    }

  }

  return (str != str_end ? 1 : 0);

}

size_t my_numchars_mb(const CHARSET_INFO * const cs,

		      const char *pos, const char *end)

{

  register size_t count= 0;

  while (pos < end)

  {

    uint32_t mb_len;

    pos+= (mb_len= my_ismbchar(cs,pos,end)) ? mb_len : 1;

    count++;

  }

  return count;

}

size_t my_charpos_mb(const CHARSET_INFO * const cs,

		     const char *pos, const char *end, size_t length)

{

  const char *start= pos;

  while (length && pos < end)

  {

    uint32_t mb_len;

    pos+= (mb_len= my_ismbchar(cs, pos, end)) ? mb_len : 1;

    length--;

  }

  return (size_t) (length ? end+2-start : pos-start);

}

size_t my_well_formed_len_mb(const CHARSET_INFO * const cs, const char *b, const char *e,

                             size_t pos, int *error)

{

  const char *b_start= b;

  *error= 0;

  while (pos)

  {

    my_wc_t wc;

    int mb_len;

    if ((mb_len= cs->cset->mb_wc(cs, &wc, (const unsigned char*) b, (const unsigned char*) e)) <= 0)

    {

      *error= b < e ? 1 : 0;

      break;

    }

    b+= mb_len;

    pos--;

  }

  return (size_t) (b - b_start);

}

uint32_t my_instr_mb(const CHARSET_INFO * const cs,

                 const char *b, size_t b_length,

                 const char *s, size_t s_length,

                 my_match_t *match, uint32_t nmatch)

{

  register const char *end, *b0;

  int res= 0;

  if (s_length <= b_length)

  {

    if (!s_length)

    {

      if (nmatch)

      {

        match->beg= 0;

        match->end= 0;

        match->mb_len= 0;

      }

      return 1;		/* Empty string is always found */

    }

    b0= b;

    end= b+b_length-s_length+1;

    while (b < end)

    {

      int mb_len;

      if (!cs->coll->strnncoll(cs, (const unsigned char*) b, s_length,

                                   (const unsigned char*) s, s_length, 0))

      {

        if (nmatch)

        {

          match[0].beg= 0;

          match[0].end= (size_t) (b-b0);

          match[0].mb_len= res;

          if (nmatch > 1)

          {

            match[1].beg= match[0].end;

            match[1].end= match[0].end+s_length;

            match[1].mb_len= 0;	/* Not computed */

          }

        }

        return 2;

      }

      mb_len= (mb_len= my_ismbchar(cs, b, end)) ? mb_len : 1;

      b+= mb_len;

      b_length-= mb_len;

      res++;

    }

  }

  return 0;

}

/* BINARY collations handlers for MB charsets */

int my_strnncoll_mb_bin(const CHARSET_INFO * const,

                        const unsigned char *s, size_t slen,

                        const unsigned char *t, size_t tlen,

                        bool t_is_prefix)

{

  size_t len= min(slen,tlen);

  int cmp= memcmp(s,t,len);

  return cmp ? cmp : (int) ((t_is_prefix ? len : slen) - tlen);

}

/*

  Compare two strings.

  SYNOPSIS

    my_strnncollsp_mb_bin()

    cs			Chararacter set

    s			String to compare

    slen		Length of 's'

    t			String to compare

    tlen		Length of 't'

    diff_if_only_endspace_difference

		        Set to 1 if the strings should be regarded as different

                        if they only difference in end space

  NOTE

   This function is used for character strings with binary collations.

   The shorter string is extended with end space to be as long as the longer

   one.

  RETURN

    A negative number if s < t

    A positive number if s > t

    0 if strings are equal

*/

int my_strnncollsp_mb_bin(const CHARSET_INFO * const,

                          const unsigned char *a, size_t a_length,

                          const unsigned char *b, size_t b_length,

                          bool diff_if_only_endspace_difference)

{

  const unsigned char *end;

  size_t length;

  int res;

#ifndef VARCHAR_WITH_DIFF_ENDSPACE_ARE_DIFFERENT_FOR_UNIQUE

  diff_if_only_endspace_difference= 0;

#endif

  end= a + (length= min(a_length, b_length));

  while (a < end)

  {

    if (*a++ != *b++)

      return ((int) a[-1] - (int) b[-1]);

  }

  res= 0;

  if (a_length != b_length)

  {

    int swap= 1;

    if (diff_if_only_endspace_difference)

      res= 1;                                   /* Assume 'a' is bigger */

    /*

      Check the next not space character of the longer key. If it's < ' ',

      then it's smaller than the other key.

    */

    if (a_length < b_length)

    {

      /* put shorter key in s */

      a_length= b_length;

      a= b;

      swap= -1;					/* swap sign of result */

      res= -res;

    }

    for (end= a + a_length-length; a < end ; a++)

    {

      if (*a != ' ')

	return (*a < ' ') ? -swap : swap;

    }

  }

  return res;

}

/*

  Copy one non-ascii character.

  "dst" must have enough room for the character.

  Note, we don't use sort_order[] in this macros.

  This is correct even for case insensitive collations:

  - basic Latin letters are processed outside this macros;

  - for other characters sort_order[x] is equal to x.

*/

#define my_strnxfrm_mb_non_ascii_char(cs, dst, src, se)                  \

{                                                                        \

  switch (cs->cset->ismbchar(cs, (const char*) src, (const char*) se)) { \

  case 4:                                                                \

    *dst++= *src++;                                                      \

    /* fall through */                                                   \

  case 3:                                                                \

    *dst++= *src++;                                                      \

    /* fall through */                                                   \

  case 2:                                                                \

    *dst++= *src++;                                                      \

    /* fall through */                                                   \

  case 0:                                                                \

    *dst++= *src++; /* byte in range 0x80..0xFF which is not MB head */  \

  }                                                                      \

}

/*

  For character sets with two or three byte multi-byte

  characters having multibyte weights *equal* to their codes:

  cp932, euckr, gb2312, sjis, eucjpms, ujis.

*/

size_t

my_strnxfrm_mb(const CHARSET_INFO * const cs,

               unsigned char *dst, size_t dstlen, uint32_t nweights,

               const unsigned char *src, size_t srclen, uint32_t flags)

{

  unsigned char *d0= dst;

  unsigned char *de= dst + dstlen;

  const unsigned char *se= src + srclen;

  const unsigned char *sort_order= cs->sort_order;

  assert(cs->mbmaxlen <= 4);

  /*

    If "srclen" is smaller than both "dstlen" and "nweights"

    then we can run a simplified loop -

    without checking "nweights" and "de".

  */

  if (dstlen >= srclen && nweights >= srclen)

  {

    if (sort_order)

    {

      /* Optimized version for a case insensitive collation */

      for (; src < se; nweights--)

      {

        if (*src < 128) /* quickly catch ASCII characters */

          *dst++= sort_order[*src++];

        else

          my_strnxfrm_mb_non_ascii_char(cs, dst, src, se);

      }

    }

    else

    {

      /* Optimized version for a case sensitive collation (no sort_order) */

      for (; src < se; nweights--)

      {

        if (*src < 128) /* quickly catch ASCII characters */

          *dst++= *src++;

        else

          my_strnxfrm_mb_non_ascii_char(cs, dst, src, se);

      }

    }

    goto pad;

  }

  /*

    A thourough loop, checking all possible limits:

    "se", "nweights" and "de".

  */

  for (; src < se && nweights; nweights--)

  {

    int chlen;

    if (*src < 128 ||

        !(chlen= cs->cset->ismbchar(cs, (const char*) src, (const char*) se)))

    {

      /* Single byte character */

      if (dst >= de)

        break;

      *dst++= sort_order ? sort_order[*src++] : *src++;

    }

    else

    {

      /* Multi-byte character */

      if (dst + chlen > de)

        break;

      *dst++= *src++;

      *dst++= *src++;

      if (chlen == 3)

        *dst++= *src++;

    }

  }

pad:

  return my_strxfrm_pad_desc_and_reverse(cs, d0, dst, de, nweights, flags, 0);

}

int my_strcasecmp_mb_bin(const CHARSET_INFO * const,

                         const char *s, const char *t)

{

  return strcmp(s,t);

}

void my_hash_sort_mb_bin(const CHARSET_INFO * const,

                         const unsigned char *key, size_t len,

                         uint32_t *nr1, uint32_t *nr2)

{

  const unsigned char *pos = key;

  /*

     Remove trailing spaces. We have to do this to be able to compare

    'A ' and 'A' as identical

  */

  key= internal::skip_trailing_space(key, len);

  for (; pos < (const unsigned char*) key ; pos++)

  {

    nr1[0]^=(ulong) ((((uint32_t) nr1[0] & 63)+nr2[0]) *

	     ((uint32_t)*pos)) + (nr1[0] << 8);

    nr2[0]+=3;

  }

}

/*

  Fill the given buffer with 'maximum character' for given charset

  SYNOPSIS

      pad_max_char()

      cs   Character set

      str  Start of buffer to fill

      end  End of buffer to fill

  DESCRIPTION

      Write max key:

      - for non-Unicode character sets:

        just set to 255.

      - for Unicode character set (utf-8):

        create a buffer with multibyte representation of the max_sort_char

        character, and copy it into max_str in a loop.

*/

static void pad_max_char(const CHARSET_INFO * const cs, char *str, char *end)

{

  char buf[10];

  char buflen;

  if (!(cs->state & MY_CS_UNICODE))

  {

    memset(str, 255, end - str);

    return;

  }

  buflen= cs->cset->wc_mb(cs, cs->max_sort_char, (unsigned char*) buf,

                          (unsigned char*) buf + sizeof(buf));

  assert(buflen > 0);

  do

  {

    if ((str + buflen) < end)

    {

      /* Enough space for the characer */

      memcpy(str, buf, buflen);

      str+= buflen;

    }

    else

    {

      /*

        There is no space for whole multibyte

        character, then add trailing spaces.

      */

      *str++= ' ';

    }

  } while (str < end);

}

/*

** Calculate min_str and max_str that ranges a LIKE string.

** Arguments:

** ptr		Pointer to LIKE string.

** ptr_length	Length of LIKE string.

** escape	Escape character in LIKE.  (Normally '\').

**		All escape characters should be removed from

**              min_str and max_str

** w_one        Single char matching char in LIKE (Normally '_')

** w_many       Multiple char matching char in LIKE (Normally '%')

** res_length	Length of min_str and max_str.

** min_str	Smallest case sensitive string that ranges LIKE.

**		Should be space padded to res_length.

** max_str	Largest case sensitive string that ranges LIKE.

**		Normally padded with the biggest character sort value.

**

** The function should return 0 if ok and 1 if the LIKE string can't be

** optimized !

*/

bool my_like_range_mb(const CHARSET_INFO * const cs,

                         const char *ptr,size_t ptr_length,

                         char escape, char w_one, char w_many,

                         size_t res_length,

                         char *min_str,char *max_str,

                         size_t *min_length,size_t *max_length)

{

  uint32_t mb_len;

  const char *end= ptr + ptr_length;

  char *min_org= min_str;

  char *min_end= min_str + res_length;

  char *max_end= max_str + res_length;

  size_t maxcharlen= res_length / cs->mbmaxlen;

  const char *contraction_flags= cs->contractions ?

              ((const char*) cs->contractions) + 0x40*0x40 : NULL;

  for (; ptr != end && min_str != min_end && maxcharlen ; maxcharlen--)

  {

    /* We assume here that escape, w_any, w_namy are one-byte characters */

    if (*ptr == escape && ptr+1 != end)

      ptr++;                                    /* Skip escape */

    else if (*ptr == w_one || *ptr == w_many)   /* '_' and '%' in SQL */

    {

fill_max_and_min:

      /*

        Calculate length of keys:

        'a\0\0... is the smallest possible string when we have space expand

        a\ff\ff... is the biggest possible string

      */

      *min_length= ((cs->state & MY_CS_BINSORT) ? (size_t) (min_str - min_org) :

                    res_length);

      *max_length= res_length;

      /* Create min key  */

      do

      {

	*min_str++= (char) cs->min_sort_char;

      } while (min_str != min_end);

      /*

        Write max key: create a buffer with multibyte

        representation of the max_sort_char character,

        and copy it into max_str in a loop.

      */