/* Test of conversion of wide string to string.
   Copyright (C) 2008 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>, 2008.  */

#include <config.h>

#include <wchar.h>

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

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

int
main (int argc, char *argv[])
{
  /* configure should already have checked that the locale is supported.  */
  if (setlocale (LC_ALL, "") == NULL)
    return 1;

  if (argc > 1)
    {
      wchar_t input[10];
      size_t n;
      const wchar_t *src;
      #define BUFSIZE 20
      char buf[BUFSIZE];
      size_t ret;

      {
	size_t i;
	for (i = 0; i < BUFSIZE; i++)
	  buf[i] = '_';
      }

      switch (argv[1][0])
	{
	case '1':
	  /* Locale encoding is ISO-8859-1 or ISO-8859-15.  */
	  {
	    const char original[] = "B\374\337er"; /* "Büßer" */

	    ret = mbstowcs (input, original, 10);
	    ASSERT (ret == 5);

	    for (n = 0; n < 10; n++)
	      {
		src = input;
		ret = wcsrtombs (NULL, &src, n, NULL);
		ASSERT (ret == 5);
		ASSERT (src == input);

		src = input;
		ret = wcsrtombs (buf, &src, n, NULL);
		ASSERT (ret == (n <= 5 ? n : 5));
		ASSERT (src == (n <= 5 ? input + n : NULL));
		ASSERT (memcmp (buf, original, ret) == 0);
		if (src == NULL)
		  ASSERT (buf[ret] == '\0');
		ASSERT (buf[ret + (src == NULL) + 0] == '_');
		ASSERT (buf[ret + (src == NULL) + 1] == '_');
		ASSERT (buf[ret + (src == NULL) + 2] == '_');
	      }
	  }
	  break;

	case '2':
	  /* Locale encoding is UTF-8.  */
	  {
	    const char original[] = "B\303\274\303\237er"; /* "Büßer" */

	    ret = mbstowcs (input, original, 10);
	    ASSERT (ret == 5);

	    for (n = 0; n < 10; n++)
	      {
		src = input;
		ret = wcsrtombs (NULL, &src, n, NULL);
		ASSERT (ret == 7);
		ASSERT (src == input);

		src = input;
		ret = wcsrtombs (buf, &src, n, NULL);
		ASSERT (ret == (n < 1 ? n :
				n < 3 ? 1 :
				n < 5 ? 3 :
				n <= 7 ? n : 7));
		ASSERT (src == (n < 1 ? input + n :
				n < 3 ? input + 1 :
				n < 5 ? input + 2 :
				n <= 7 ? input + (n - 2) : NULL));
		ASSERT (memcmp (buf, original, ret) == 0);
		if (src == NULL)
		  ASSERT (buf[ret] == '\0');
		ASSERT (buf[ret + (src == NULL) + 0] == '_');
		ASSERT (buf[ret + (src == NULL) + 1] == '_');
		ASSERT (buf[ret + (src == NULL) + 2] == '_');
	      }
	  }
	  break;

	case '3':
	  /* Locale encoding is EUC-JP.  */
	  {
	    const char original[] = "<\306\374\313\334\270\354>"; /* "<日本語>" */

	    ret = mbstowcs (input, original, 10);
	    ASSERT (ret == 5);

	    for (n = 0; n < 10; n++)
	      {
		src = input;
		ret = wcsrtombs (NULL, &src, n, NULL);
		ASSERT (ret == 8);
		ASSERT (src == input);

		src = input;
		ret = wcsrtombs (buf, &src, n, NULL);
		ASSERT (ret == (n < 1 ? n :
				n < 3 ? 1 :
				n < 5 ? 3 :
				n < 7 ? 5 :
				n <= 8 ? n : 8));
		ASSERT (src == (n < 1 ? input + n :
				n < 3 ? input + 1 :
				n < 5 ? input + 2 :
				n < 7 ? input + 3 :
				n <= 8 ? input + (n - 3) : NULL));
		ASSERT (memcmp (buf, original, ret) == 0);
		if (src == NULL)
		  ASSERT (buf[ret] == '\0');
		ASSERT (buf[ret + (src == NULL) + 0] == '_');
		ASSERT (buf[ret + (src == NULL) + 1] == '_');
		ASSERT (buf[ret + (src == NULL) + 2] == '_');
	      }
	  }
	  break;


	case '4':
	  /* Locale encoding is GB18030.  */
	  {
	    const char original[] = "B\250\271\201\060\211\070er"; /* "Büßer" */

	    ret = mbstowcs (input, original, 10);
	    ASSERT (ret == 5);

	    for (n = 0; n < 10; n++)
	      {
		src = input;
		ret = wcsrtombs (NULL, &src, n, NULL);
		ASSERT (ret == 9);
		ASSERT (src == input);

		src = input;
		ret = wcsrtombs (buf, &src, n, NULL);
		ASSERT (ret == (n < 1 ? n :
				n < 3 ? 1 :
				n < 7 ? 3 :
				n <= 9 ? n : 9));
		ASSERT (src == (n < 1 ? input + n :
				n < 3 ? input + 1 :
				n < 7 ? input + 2 :
				n <= 9 ? input + (n - 4) : NULL));
		ASSERT (memcmp (buf, original, ret) == 0);
		if (src == NULL)
		  ASSERT (buf[ret] == '\0');
		ASSERT (buf[ret + (src == NULL) + 0] == '_');
		ASSERT (buf[ret + (src == NULL) + 1] == '_');
		ASSERT (buf[ret + (src == NULL) + 2] == '_');
	      }
	  }
	  break;

	default:
	  return 1;
	}

      return 0;
    }

  return 1;
}