~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 "m_string.h"

#include "m_ctype.h"

#ifdef USE_MB

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

{

  register uint32_t l;

  register uchar *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[(uchar)*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 uchar *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[(uchar)*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 __attribute__((unused)),

                    size_t dstlen __attribute__((unused)))

{

  register uint32_t l;

  register char *srcend= src + srclen;

  register uchar *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[(uchar) *src];

      src++;

    }

  }

  return srclen;

}

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

                    char *dst __attribute__((unused)),

                    size_t dstlen __attribute__((unused)))

{

  register uint32_t l;

  register char *srcend= src + srclen;

  register uchar *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[(uchar)*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 uchar *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[(uchar) *s++] != map[(uchar) *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) (uchar) (s)->sort_order[(uchar) (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 */

      uchar 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 __attribute__((unused)),

		      const char *pos, const char *end)

{

  register size_t count= 0;

  while (pos < end) 

  {

    uint 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 __attribute__((unused)),

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

{

  const char *start= pos;

  while (length && pos < end)

  {

    uint 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 uchar*) b, (const uchar*) e)) <= 0)

    {

      *error= b < e ? 1 : 0;

      break;

    }

    b+= mb_len;

    pos--;

  }

  return (size_t) (b - b_start);

}

uint 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, uint 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 uchar*) b, s_length, 

                                   (const uchar*) 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  cs __attribute__((unused)),

                        const uchar *s, size_t slen,

                        const uchar *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  cs __attribute__((unused)),

                          const uchar *a, size_t a_length,

                          const uchar *b, size_t b_length,

                          bool diff_if_only_endspace_difference)

{

  const uchar *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,

               uchar *dst, size_t dstlen, uint nweights,

               const uchar *src, size_t srclen, uint flags)

{

  uchar *d0= dst;

  uchar *de= dst + dstlen;

  const uchar *se= src + srclen;

  const uchar *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  cs __attribute__((unused)),

                         const char *s, const char *t)

{

  return strcmp(s,t);

}

void my_hash_sort_mb_bin(const CHARSET_INFO * const cs __attribute__((unused)),

                         const uchar *key, size_t len, uint32_t *nr1, uint32_t *nr2)

{

  const uchar *pos = key;

  /*

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

    'A ' and 'A' as identical

  */

  key= skip_trailing_space(key, len);

  for (; pos < (const uchar*) key ; pos++)

  {

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

	     ((uint)*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, (uchar*) buf,

                          (uchar*) 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)

{

  uint 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. 

      */

      *max_length= res_length;

      pad_max_char(cs, max_str, max_end);

      return 0;

    }

    if ((mb_len= my_ismbchar(cs, ptr, end)) > 1)

    {

      if (ptr+mb_len > end || min_str+mb_len > min_end)

        break;

      while (mb_len--)

       *min_str++= *max_str++= *ptr++;

    }

    else

    {

      /*

        Special case for collations with contractions.

        For example, in Chezh, 'ch' is a separate letter

        which is sorted between 'h' and 'i'.

        If the pattern 'abc%', 'c' at the end can mean:

        - letter 'c' itself,

        - beginning of the contraction 'ch'.

        If we simply return this LIKE range:

         'abc\min\min\min' and 'abc\max\max\max'

        then this query: SELECT * FROM t1 WHERE a LIKE 'abc%'

        will only find values starting from 'abc[^h]',

        but won't find values starting from 'abch'.

        We must ignore contraction heads followed by w_one or w_many.

        ('Contraction head' means any letter which can be the first

        letter in a contraction)

        For example, for Czech 'abc%', we will return LIKE range,

        which is equal to LIKE range for 'ab%':

        'ab\min\min\min\min' and 'ab\max\max\max\max'.

      */

      if (contraction_flags && ptr + 1 < end &&

          contraction_flags[(uchar) *ptr])

      {

        /* Ptr[0] is a contraction head. */

        if (ptr[1] == w_one || ptr[1] == w_many)

        {

          /* Contraction head followed by a wildcard, quit. */

          goto fill_max_and_min;

        }

        /*

          Some letters can be both contraction heads and contraction tails.

          For example, in Danish 'aa' is a separate single letter which

          is sorted after 'z'. So 'a' can be both head and tail.

          If ptr[0]+ptr[1] is a contraction,

          then put both letters together.

          If ptr[1] can be a contraction part, but ptr[0]+ptr[1]

          is not a contraction, then we put only ptr[0],

          and continue with ptr[1] on the next loop.

        */

        if (contraction_flags[(uchar) ptr[1]] &&

            cs->contractions[(*ptr-0x40)*0x40 + ptr[1] - 0x40])

        {

          /* Contraction found */

          if (maxcharlen == 1 || min_str + 1 >= min_end)

          {

            /* Both contraction parts don't fit, quit */

            goto fill_max_and_min;

          }

          /* Put contraction head */

          *min_str++= *max_str++= *ptr++;

          maxcharlen--;

        }