/*

* Copyright (c) 2000 - 2001 Nokia Corporation and/or its subsidiary(-ies).

* All rights reserved.

* This component and the accompanying materials are made available

* under the terms of the License "Eclipse Public License v1.0"

* which accompanies this distribution, and is available

* at the URL "http://www.eclipse.org/legal/epl-v10.html".

*

* Initial Contributors:

* Nokia Corporation - initial contribution.

*

* Contributors:

*

* Description: 

*

*/

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

**  File: character.c

**  Description:

* 

* Note: all of the char functions assume that the input buffer points

* to an array of characters which contains a null somewhere and is

* correctly encoded for the particular encoding. Bad things can happen

* if this is not the case. In order to avoid having to check these 

* conditions on every operation, a set of validate functions is provided

* to pre-test a string where the caller is not sure these conditions

* are met.  It is especially important, when calling character operations

* on bytecode, to make sure to validate all strings.

* 

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

#include "cxml_internal.h"

#include <xml/cxml/nw_string_char.h>

/*  

* TODO:  Note that there is some duplication between the Validate*

* calls and the string length function.  The Validate*S could return

* the length too.

*/

/*

* Check that storage points to a valid UTF8 string no longer

* than length bytes.

*/

static NW_Int32

StringValidUTF8 (NW_Byte * storage, NW_Uint32 length)

{

  NW_Uint32 i;

  NW_Byte bits;

  NW_ASSERT(storage != NULL);

  NW_ASSERT(length != 0);

  for (i = 0; i < length;)

  {

    if (storage[i] == 0)

      {

        return 1;

      }

    bits = (NW_Byte) (storage[i] >> 4);

    if (bits < 8)

    {

          i++;

    }

    else if ((bits == 12) || (bits == 13))

    { 

  	  i+=2;

    }

    else if (bits == 14)

    {

	    i += 3;

    }

    else if (bits == 15)

    {

	    i += 4;

    }

    else

    {

          return 0;

    }

  }    

  return 0;

}

/*  

* Check validity of UCS2 string storage 

*/

static NW_Int32

StringValidUCS2 (NW_Byte * storage, NW_Uint32 length)

{

  NW_Uint32 i;

  NW_ASSERT(storage != NULL);

  NW_ASSERT(length != 0);

  for (i = 0; i < (length - 1); i += 2)

  {

    if (((storage[i] << 8) | storage[i + 1]) == 0)

    {

      return 1;

    }

  }

  return 0;

}

/* 

* Check validity of ISO8859 string storage

*/

static NW_Int32

StringValidISO88591 (NW_Byte * storage, NW_Uint32 length)

{

  NW_Uint32 i;

  NW_ASSERT(storage != NULL);

  NW_ASSERT(length != 0);

  for (i = 0; i < length; i++)

  {

    if (storage[i] == 0)

    {

	return 1;

    }

  }

  return 0;

}

/* 

* Check validity of ASCII string storage

*/

static NW_Int32

StringValidUSASCII (NW_Byte * storage, NW_Uint32 length)

{

  NW_Uint32 i;

  NW_ASSERT(storage != NULL);

  NW_ASSERT(length != 0);

  for (i = 0; i < length; i++)

  {

    if (storage[i] == 0)

    {

	return 1;

    }

  }

  return 0;

}

/*

* Check the given charset encoding (MIBENUM) and if it

*              is supported.

*/

NW_Status_t

NW_String_charsetValid (NW_Uint32 encoding)

{

  switch (encoding)

  {

  case HTTP_iso_10646_ucs_2:

  case HTTP_iso_8859_1:

  case HTTP_utf_8:

  case HTTP_us_ascii:

    return NW_STAT_SUCCESS;

  default:

    return NW_STAT_WBXML_ERROR_CHARSET_UNSUPPORTED;

  }

}

/*

* RETURN -1 if the encoding is not supported

*/

NW_Int32

NW_String_valid(NW_Byte * storage, NW_Uint32 length, NW_Uint32 encoding)

{

  if (encoding == HTTP_iso_10646_ucs_2)

  {

	  return StringValidUCS2 (storage, length);

  }

  else if (encoding == HTTP_utf_8)

  {

	  return StringValidUTF8 (storage, length);

  }

  else if (encoding == HTTP_iso_8859_1)

  {

	  return StringValidISO88591 (storage, length);

  }

  else if (encoding == HTTP_us_ascii)

  {

	  return StringValidUSASCII (storage, length);

  }

  return -1;

}

/* 

* TODO:  The following routines are taken from Rainbow.

* They should be revisited for better efficiency, etc.

*/

/*

* Read one UTF8 character from a buffer and store it as a NW_Ucs2.

* Returns number of input bytes read.

*/

static NW_Int32

ReadUTF8Char (NW_Byte * buff, NW_Ucs2 * c)

{

  switch ((buff[0] >> 4) & 0xf)

  {

  case 0:

  case 1:

  case 2:

  case 3:

  case 4:

  case 5:

  case 6:

  case 7:

    /* 1 NW_Byte */

    *c = (NW_Ucs2) buff[0];

    return 1;

  case 12:  

  case 13:

    /* 2 bytes */

    if ((buff[1] & 0xC0) != 0x80)

    {

	     return -1;

    }

    *c = (NW_Ucs2) (((buff[0] & 0x1F) << 6) | (buff[1] & 0x3F));

    return 2;

  case 14:

    /* 3 bytes */

    if (((buff[1] & 0xC0) != 0x80) && ((buff[2] & 0xC0) != 0x80))

    {

	     return -1;

    }

    *c = (NW_Ucs2) (((buff[0] & 0x0F) << 12) |

      ((buff[1] & 0x3F) << 6) | ((buff[2] & 0x3F) << 0));

    return 3;

  //we used not to handle 4-bytes UTF-8 case (only 16 bits handled), the case 15 is newly added, it may cause

  //problem if in an application the a 4-byte character would convert to ucs2 encoding. 

  case 15:  

    /* 4 bytes */

    if (((buff[1] & 0xC0) != 0x80) && ((buff[2] & 0xC0) != 0x80) && ((buff[3] & 0xC0) != 0x80))

    {

	     return -1;

    }

    *c = (((buff[0] & 0x07) << 18) |

          ((buff[1] & 0x3F) << 12) |

          ((buff[2] & 0x3F) << 6) |

           (buff[3] & 0x3F));

    return 4;

  default:

    return -1;    /* Bad format */

  }

}

/* 

* Write a NW_Ucs2 into a buffer as UTF8. Returns number of bytes written 

*/

NW_Uint32

NW_String_writeUTF8Char (NW_Ucs2 c, NW_Byte * buff)

{

  if (c <= 0x007F)

  {

    /* 0x0000 - 0x007F: 1 NW_Byte UTF-8 encoding. */

    buff[0] = (NW_Byte) c;

    return 1;

  }

  else if (c > 0x07FF)

  {

    /* 0x0800 - 0xFFFF: 3 NW_Byte UTF-8 encoding. */

    buff[0] = (NW_Byte) (0xE0 | ((c >> 12) & 0x0F));

    buff[1] = (NW_Byte) (0x80 | ((c >> 6) & 0x3F));

    buff[2] = (NW_Byte) (0x80 | ((c >> 0) & 0x3F));

    return 3;

  }

  else

  {

    /* 0x0080 - 0x07ff: 2 NW_Byte UTF-8 encoding. */

    buff[0] = (NW_Byte) (0xC0 | ((c >> 6) & 0x1F));

    buff[1] = (NW_Byte) (0x80 | ((c >> 0) & 0x3F));

    return 2;

  }

}

static NW_Int32

ReadInt16Char (NW_Byte * buff, NW_Ucs2 * c)

{

  /* read unaligned native-endian to aligned native-endian */

  (void) NW_Mem_memcpy(c, buff, sizeof(NW_Ucs2));

  return sizeof(NW_Ucs2);

}

static NW_Int32

ReadISO88591Char (NW_Byte * buff, NW_Ucs2 * c)

{

  *c = buff[0];

  return 1;

}

static NW_Int32

ReadUSASCIIChar (NW_Byte * buff, NW_Ucs2 * c)

{

  *c = buff[0];

  return 1;

}

/*

* Read one character of some encoding, returning the NW_Ucs2 

* equivalent and the count of raw characters read

*

* RETURN -1 if encoding is not supported

*/

EXPORT_C NW_Int32

NW_String_readChar (NW_Byte * buff, NW_Ucs2 * c, NW_Uint32 encoding)

{

  NW_Int32 nbytes = 0;

  if (encoding == HTTP_iso_10646_ucs_2)

    return ReadInt16Char (&buff[nbytes], c);

  else if (encoding == HTTP_utf_8)

    return ReadUTF8Char (&buff[nbytes], c);

  else if (encoding == HTTP_iso_8859_1)

    return ReadISO88591Char (&buff[nbytes], c);

  else if (encoding == HTTP_us_ascii)

    return ReadUSASCIIChar (&buff[nbytes], c);

  return -1;

}

/* 

* Get the length of a character string in some encoding. Returns the number

* of characters (less the terminating char). The out param byte_count returns

* the number of bytes of storage scanned (including the terminating char).

* Note that there is NO validity check here. This should be done first if

* needed.  TODO:  Also note that the validity check could return the length 

* directly, thus eliminating the need for call to this function when 

* doint32 validity checkint32.

*/

EXPORT_C NW_Int32

NW_String_charBuffGetLength (void *buffer, NW_Uint32 encoding, NW_Uint32 * byte_count)

{

  NW_Int32 chars = 0;

  NW_Ucs2 c = 1;

  NW_Int32 retval = 0;

  *byte_count = 0;

  while (c)

  {

    c = 0; /* partial protection against an infinite loop */

    retval = NW_String_readChar ((NW_Byte *) buffer + *byte_count, &c, encoding);

    if(retval < 0){

      return -1;

    }

    (*byte_count) += (NW_Uint32) retval;

    chars++;

  }

  return chars - 1;

}

/* 

* Conversions among character strings of various types and ucs2.

* These functions assume that the length in characters of the 

* input buffer has been pre-calculated, so that this operation

* doesn't have to be performed for every conversion. This works well

* for String_t which store the character count.

*

* RETURN NULL if malloc fails

*/

NW_String_UCS2Buff_t *

NW_String_charToUCS2Buff (NW_Byte * s, NW_Uint32 encoding)

{

  NW_String_UCS2Buff_t *storage;

  NW_Ucs2 c;

  NW_Int32 i;

  NW_Int32 count = 0;

  NW_Int32 length = 0;

  NW_Uint32 byteCount = 0;

  NW_Int32 retval = 0;

  if (!NW_String_charsetValid(encoding))

  {

    return NULL;

  }

  length = NW_String_charBuffGetLength(s, encoding, &byteCount);

  if(length < 0){

    return NULL;

  }

  storage =

    (NW_String_UCS2Buff_t*) 

    NW_Mem_Malloc(((NW_Uint32)length + 1) * sizeof (NW_String_UCS2Buff_t));

  if (storage == NULL)

  {

    return NULL;

  }

  for (i = 0; i < length; i++)

  {

    retval = NW_String_readChar (s + count, &c, encoding);

    if(retval < 0){

      NW_Mem_Free(storage);

      return NULL;

    }

    count += retval;

    storage[i].bytes[0] = (NW_Byte) ((c & 0xff00) >> 8);

    storage[i].bytes[1] = (NW_Byte) (c & 0xff);

  }

  storage[length].bytes[0] = 0;

  storage[length].bytes[1] = 0;

  return storage;

}

/*

* TODO: is this a public or private function ???

*/

NW_String_UCS2Buff_t *

NW_String_UTF8ToUCS2Buff (NW_Byte * s)

{

  return NW_String_charToUCS2Buff (s, HTTP_utf_8);

}

/*

* TODO: is this a public or private function ???

*/

NW_String_UCS2Buff_t *

NW_String_ISO88591ToUCS2Buff (NW_Byte * s)

{

  return NW_String_charToUCS2Buff (s, HTTP_iso_8859_1);

}

/*

* RETURN NULL if malloc fails

*/

NW_Byte *

NW_String_UCS2ToUTF8 (NW_String_UCS2Buff_t * s, NW_Uint32 length)

{

  NW_Byte *tstore;

  NW_Byte *storage;

  NW_Ucs2 c;

  NW_Uint32 i;

  NW_Int32 count = 0;

  NW_Ucs2 *src = (NW_Ucs2 *)s; /*WMS we should use UCS2 pointer, 

                                 because s is a structure and the size of a structure is not fixed

					             in ARM processor, the size of NW_String_UCS2Buff_t is 4 byte 

                                 (address alignment issue) */

  tstore = (NW_Byte *) NW_Mem_Malloc ((length + 1) * 3);

  if (tstore == NULL)

  {

    return NULL;

  }

  for (i = 0; i < length; i++)

    {

      ReadInt16Char ((NW_Byte *) (src + i), &c);

      count += NW_String_writeUTF8Char (c, tstore + count);

    }

    *(tstore + count) = 0;

    storage = (NW_Byte *) NW_Mem_Malloc (count + 1);

    if (storage)

    {

      NW_Mem_memcpy (storage, tstore, count + 1);

    }

    NW_Mem_Free (tstore);

    return storage;

}

/*

* RETURN NULL if malloc fails

* byteCount is total allocation size of s as far as conversion is concerned

*/

NW_Byte *

NW_String_UCS2ToISO88591 (NW_String_UCS2Buff_t * s, NW_Uint32 byteCount)

{

  NW_Byte *storage = NULL;

  NW_Ucs2 c;

  NW_Uint32 i;

  NW_Ucs2 *src = (NW_Ucs2 *)s; /*WMS we should use UCS2 pointer, 

                                 because s is a structure and the size of a structure is not fixed

					             in ARM processor, the size of NW_String_UCS2Buff_t is 4 byte 

                                 (address alignment issue) */

  storage = (NW_Byte *) NW_Mem_Malloc (byteCount + 1);

  if (storage == NULL)

  {

    return NULL;

  }

  for (i = 0; i < byteCount; i++)

  {

    ReadInt16Char ((NW_Byte *) (src + i), &c);

    storage[i] = (NW_Byte) (c & 0xff);

  }

  storage[byteCount] = 0;

  return storage;

}

/* Ordered comparison of ucs2 strings */

NW_Int32

NW_String_UCS2BuffCmp (NW_String_UCS2Buff_t * s1, 

                       NW_String_UCS2Buff_t * s2,

                       NW_Bool matchCase)

{