/* Test of conversion to UTF-32 from legacy encodings.
   Copyright (C) 2007 Free Software Foundation, Inc.

   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; either version 3 of the License, or
   (at your option) any later version.

   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, see <http://www.gnu.org/licenses/>.  */

/* Written by Bruno Haible <bruno@clisp.org>, 2007.  */

#include <config.h>

#include "uniconv.h"

#include <stdio.h>
#include <stdlib.h>
#include <string.h>

#include "unistr.h"

#define SIZEOF(array) (sizeof (array) / sizeof (array[0]))
#define ASSERT(expr) \
  do									     \
    {									     \
      if (!(expr))							     \
        {								     \
          fprintf (stderr, "%s:%d: assertion failed\n", __FILE__, __LINE__); \
          abort ();							     \
        }								     \
    }									     \
  while (0)

/* Magic number for detecting bounds violations.  */
#define MAGIC 0x1983EFF1

static size_t *
new_offsets (size_t n)
{
  size_t *offsets = (size_t *) malloc ((n + 1) * sizeof (size_t));
  offsets[n] = MAGIC;
  return offsets;
}

int
main ()
{
  static enum iconv_ilseq_handler handlers[] =
    { iconveh_error, iconveh_question_mark, iconveh_escape_sequence };
  size_t h;
  size_t o;
  size_t i;

#if HAVE_ICONV
  /* Assume that iconv() supports at least the encodings ASCII, ISO-8859-1,
     ISO-8859-2, and UTF-8.  */

  /* Test conversion from ISO-8859-1 to UTF-16 with no errors.  */
  for (h = 0; h < SIZEOF (handlers); h++)
    {
      enum iconv_ilseq_handler handler = handlers[h];
      static const char input[] = "\304rger mit b\366sen B\374bchen ohne Augenma\337";
      static const uint32_t expected[] = /* Ärger mit bösen Bübchen ohne Augenmaß */
	{
	  0xC4, 'r', 'g', 'e', 'r', ' ', 'm', 'i', 't', ' ', 'b', 0xF6, 's',
	  'e', 'n', ' ', 'B', 0xFC, 'b', 'c', 'h', 'e', 'n', ' ', 'o', 'h',
	  'n', 'e', ' ', 'A', 'u', 'g', 'e', 'n', 'm', 'a', 0xDF
	};
      for (o = 0; o < 2; o++)
	{
	  size_t *offsets = (o ? new_offsets (strlen (input)) : NULL);
	  uint32_t *result = NULL;
	  size_t length = 0;
	  int retval = u32_conv_from_encoding ("ISO-8859-1", handler,
					       input, strlen (input),
					       offsets,
					       &result, &length);
	  ASSERT (retval == 0);
	  ASSERT (length == SIZEOF (expected));
	  ASSERT (result != NULL && u32_cmp (result, expected, SIZEOF (expected)) == 0);
	  if (o)
	    {
	      for (i = 0; i < 37; i++)
		ASSERT (offsets[i] == i);
	      ASSERT (offsets[37] == MAGIC);
	      free (offsets);
	    }
	  free (result);
	}
    }

  /* Test conversion from ISO-8859-2 to UTF-16 with no errors.  */
  for (h = 0; h < SIZEOF (handlers); h++)
    {
      enum iconv_ilseq_handler handler = handlers[h];
      static const char input[] = "Rafa\263 Maszkowski"; /* Rafał Maszkowski */
      static const uint32_t expected[] =
	{
	  'R', 'a', 'f', 'a', 0x0142, ' ', 'M', 'a', 's', 'z', 'k', 'o', 'w',
	  's', 'k', 'i'
	};
      for (o = 0; o < 2; o++)
	{
	  size_t *offsets = (o ? new_offsets (strlen (input)) : NULL);
	  uint32_t *result = NULL;
	  size_t length = 0;
	  int retval = u32_conv_from_encoding ("ISO-8859-2", handler,
					       input, strlen (input),
					       offsets,
					       &result, &length);
	  ASSERT (retval == 0);
	  ASSERT (length == SIZEOF (expected));
	  ASSERT (result != NULL && u32_cmp (result, expected, SIZEOF (expected)) == 0);
	  if (o)
	    {
	      for (i = 0; i < 16; i++)
		ASSERT (offsets[i] == i);
	      ASSERT (offsets[16] == MAGIC);
	      free (offsets);
	    }
	  free (result);
	}
    }

  /* autodetect_jp is only supported when iconv() support ISO-2022-JP-2.  */
# if defined _LIBICONV_VERSION || !(defined _AIX || defined __sgi || defined __hpux || defined __osf__)
  /* Test conversions from autodetect_jp to UTF-16.  */
  for (h = 0; h < SIZEOF (handlers); h++)
    {
      enum iconv_ilseq_handler handler = handlers[h];
      static const char input[] = "\244\263\244\363\244\313\244\301\244\317"; /* こんにちは in EUC-JP */
      static const uint32_t expected[] = /* こんにちは */
	{
	  0x3053, 0x3093, 0x306B, 0x3061, 0x306F
	};
      for (o = 0; o < 2; o++)
	{
	  size_t *offsets = (o ? new_offsets (strlen (input)) : NULL);
	  uint32_t *result = NULL;
	  size_t length = 0;
	  int retval = u32_conv_from_encoding ("autodetect_jp", handler,
					       input, strlen (input),
					       offsets,
					       &result, &length);
	  ASSERT (retval == 0);
	  ASSERT (length == SIZEOF (expected));
	  ASSERT (result != NULL && u32_cmp (result, expected, SIZEOF (expected)) == 0);
	  if (o)
	    {
	      for (i = 0; i < 10; i++)
		ASSERT (offsets[i] == ((i % 2) == 0 ? i / 2 : (size_t)(-1)));
	      ASSERT (offsets[10] == MAGIC);
	      free (offsets);
	    }
	  free (result);
	}
    }
  for (h = 0; h < SIZEOF (handlers); h++)
    {
      enum iconv_ilseq_handler handler = handlers[h];
      static const char input[] = "\202\261\202\361\202\311\202\277\202\315"; /* こんにちは in Shift_JIS */
      static const uint32_t expected[] = /* こんにちは */
	{
	  0x3053, 0x3093, 0x306B, 0x3061, 0x306F
	};
      for (o = 0; o < 2; o++)
	{
	  size_t *offsets = (o ? new_offsets (strlen (input)) : NULL);
	  uint32_t *result = NULL;
	  size_t length = 0;
	  int retval = u32_conv_from_encoding ("autodetect_jp", handler,
					       input, strlen (input),
					       offsets,
					       &result, &length);
	  ASSERT (retval == 0);
	  ASSERT (length == SIZEOF (expected));
	  ASSERT (result != NULL && u32_cmp (result, expected, SIZEOF (expected)) == 0);
	  if (o)
	    {
	      for (i = 0; i < 10; i++)
		ASSERT (offsets[i] == ((i % 2) == 0 ? i / 2 : (size_t)(-1)));
	      ASSERT (offsets[10] == MAGIC);
	      free (offsets);
	    }
	  free (result);
	}
    }
  for (h = 0; h < SIZEOF (handlers); h++)
    {
      enum iconv_ilseq_handler handler = handlers[h];
      static const char input[] = "\033$B$3$s$K$A$O\033(B"; /* こんにちは in ISO-2022-JP-2 */
      static const uint32_t expected[] = /* こんにちは */
	{
	  0x3053, 0x3093, 0x306B, 0x3061, 0x306F
	};
      for (o = 0; o < 2; o++)
	{
	  size_t *offsets = (o ? new_offsets (strlen (input)) : NULL);
	  uint32_t *result = NULL;
	  size_t length = 0;
	  int retval = u32_conv_from_encoding ("autodetect_jp", handler,
					       input, strlen (input),
					       offsets,
					       &result, &length);
	  ASSERT (retval == 0);
	  ASSERT (length == SIZEOF (expected));
	  ASSERT (result != NULL && u32_cmp (result, expected, SIZEOF (expected)) == 0);
	  if (o)
	    {
	      for (i = 0; i < 16; i++)
		ASSERT (offsets[i] == (i == 0 ? 0 :
				       i == 5 ? 1 :
				       i == 7 ? 2 :
				       i == 9 ? 3 :
				       i == 11 ? 4 :
				       i == 13 ? 5 :
				       (size_t)(-1)));
	      ASSERT (offsets[16] == MAGIC);
	      free (offsets);
	    }
	  free (result);
	}
    }
# endif

#endif

  return 0;
}