/* Test of conversion of string to wide 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[])
{
  mbstate_t state;
  wchar_t wc;
  size_t ret;

  /* configure should already have checked that the locale is supported.  */
  if (setlocale (LC_ALL, "") == NULL)
    return 1;

  /* Test NUL byte input.  */
  {
    const char *src;

    memset (&state, '\0', sizeof (mbstate_t));

    src = "";
    ret = mbsrtowcs (NULL, &src, 0, &state);
    ASSERT (ret == 0);
    ASSERT (mbsinit (&state));

    src = "";
    ret = mbsrtowcs (NULL, &src, 1, &state);
    ASSERT (ret == 0);
    ASSERT (mbsinit (&state));

    wc = (wchar_t) 0xBADFACE;
    src = "";
    ret = mbsrtowcs (&wc, &src, 0, &state);
    ASSERT (ret == 0);
    ASSERT (wc == (wchar_t) 0xBADFACE);
    ASSERT (mbsinit (&state));

    wc = (wchar_t) 0xBADFACE;
    src = "";
    ret = mbsrtowcs (&wc, &src, 1, &state);
    ASSERT (ret == 0);
    ASSERT (wc == 0);
    ASSERT (mbsinit (&state));
  }

  if (argc > 1)
    {
      int unlimited;

      for (unlimited = 0; unlimited < 2; unlimited++)
	{
	  #define BUFSIZE 10
	  wchar_t buf[BUFSIZE];
	  const char *src;

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

	  switch (argv[1][0])
	    {
	    case '1':
	      /* Locale encoding is ISO-8859-1 or ISO-8859-15.  */
	      {
		char input[] = "B\374\337er"; /* "Büßer" */
		memset (&state, '\0', sizeof (mbstate_t));

		wc = (wchar_t) 0xBADFACE;
		ret = mbrtowc (&wc, input, 1, &state);
		ASSERT (ret == 1);
		ASSERT (wc == 'B');
		ASSERT (mbsinit (&state));
		input[0] = '\0';

		wc = (wchar_t) 0xBADFACE;
		ret = mbrtowc (&wc, input + 1, 1, &state);
		ASSERT (ret == 1);
		ASSERT (wctob (wc) == (unsigned char) '\374');
		ASSERT (mbsinit (&state));
		input[1] = '\0';

		src = input + 2;
		ret = mbsrtowcs (NULL, &src, unlimited ? BUFSIZE : 1, &state);
		ASSERT (ret == 3);
		ASSERT (src == input + 2);
		ASSERT (mbsinit (&state));

		src = input + 2;
		ret = mbsrtowcs (buf, &src, unlimited ? BUFSIZE : 1, &state);
		ASSERT (ret == (unlimited ? 3 : 1));
		ASSERT (src == (unlimited ? NULL : input + 3));
		ASSERT (wctob (buf[0]) == (unsigned char) '\337');
		if (unlimited)
		  {
		    ASSERT (buf[1] == 'e');
		    ASSERT (buf[2] == 'r');
		    ASSERT (buf[3] == 0);
		    ASSERT (buf[4] == (wchar_t) 0xBADFACE);
		  }
		else
		  ASSERT (buf[1] == (wchar_t) 0xBADFACE);
		ASSERT (mbsinit (&state));
	      }
	      return 0;

	    case '2':
	      /* Locale encoding is UTF-8.  */
	      {
		char input[] = "B\303\274\303\237er"; /* "Büßer" */
		memset (&state, '\0', sizeof (mbstate_t));

		wc = (wchar_t) 0xBADFACE;
		ret = mbrtowc (&wc, input, 1, &state);
		ASSERT (ret == 1);
		ASSERT (wc == 'B');
		ASSERT (mbsinit (&state));
		input[0] = '\0';

		wc = (wchar_t) 0xBADFACE;
		ret = mbrtowc (&wc, input + 1, 1, &state);
		ASSERT (ret == (size_t)(-2));
		ASSERT (wc == (wchar_t) 0xBADFACE);
		ASSERT (!mbsinit (&state));
		input[1] = '\0';

		src = input + 2;
		ret = mbsrtowcs (NULL, &src, unlimited ? BUFSIZE : 2, &state);
		ASSERT (ret == 4);
		ASSERT (src == input + 2);
		ASSERT (!mbsinit (&state));

		src = input + 2;
		ret = mbsrtowcs (buf, &src, unlimited ? BUFSIZE : 2, &state);
		ASSERT (ret == (unlimited ? 4 : 2));
		ASSERT (src == (unlimited ? NULL : input + 5));
		ASSERT (wctob (buf[0]) == EOF);
		ASSERT (wctob (buf[1]) == EOF);
		if (unlimited)
		  {
		    ASSERT (buf[2] == 'e');
		    ASSERT (buf[3] == 'r');
		    ASSERT (buf[4] == 0);
		    ASSERT (buf[5] == (wchar_t) 0xBADFACE);
		  }
		else
		  ASSERT (buf[2] == (wchar_t) 0xBADFACE);
		ASSERT (mbsinit (&state));
	      }
	      return 0;

	    case '3':
	      /* Locale encoding is EUC-JP.  */
	      {
		char input[] = "B\217\253\344\217\251\316er"; /* "Büßer" */
		memset (&state, '\0', sizeof (mbstate_t));

		wc = (wchar_t) 0xBADFACE;
		ret = mbrtowc (&wc, input, 1, &state);
		ASSERT (ret == 1);
		ASSERT (wc == 'B');
		ASSERT (mbsinit (&state));
		input[0] = '\0';

		wc = (wchar_t) 0xBADFACE;
		ret = mbrtowc (&wc, input + 1, 1, &state);
		ASSERT (ret == (size_t)(-2));
		ASSERT (wc == (wchar_t) 0xBADFACE);
		ASSERT (!mbsinit (&state));
		input[1] = '\0';

		src = input + 2;
		ret = mbsrtowcs (NULL, &src, unlimited ? BUFSIZE : 2, &state);
		ASSERT (ret == 4);
		ASSERT (src == input + 2);
		ASSERT (!mbsinit (&state));

		src = input + 2;
		ret = mbsrtowcs (buf, &src, unlimited ? BUFSIZE : 2, &state);
		ASSERT (ret == (unlimited ? 4 : 2));
		ASSERT (src == (unlimited ? NULL : input + 7));
		ASSERT (wctob (buf[0]) == EOF);
		ASSERT (wctob (buf[1]) == EOF);
		if (unlimited)
		  {
		    ASSERT (buf[2] == 'e');
		    ASSERT (buf[3] == 'r');
		    ASSERT (buf[4] == 0);
		    ASSERT (buf[5] == (wchar_t) 0xBADFACE);
		  }
		else
		  ASSERT (buf[2] == (wchar_t) 0xBADFACE);
		ASSERT (mbsinit (&state));
	      }
	      return 0;

	    case '4':
	      /* Locale encoding is GB18030.  */
	      {
		char input[] = "B\250\271\201\060\211\070er"; /* "Büßer" */
		memset (&state, '\0', sizeof (mbstate_t));

		wc = (wchar_t) 0xBADFACE;
		ret = mbrtowc (&wc, input, 1, &state);
		ASSERT (ret == 1);
		ASSERT (wc == 'B');
		ASSERT (mbsinit (&state));
		input[0] = '\0';

		wc = (wchar_t) 0xBADFACE;
		ret = mbrtowc (&wc, input + 1, 1, &state);
		ASSERT (ret == (size_t)(-2));
		ASSERT (wc == (wchar_t) 0xBADFACE);
		ASSERT (!mbsinit (&state));
		input[1] = '\0';

		src = input + 2;
		ret = mbsrtowcs (NULL, &src, unlimited ? BUFSIZE : 2, &state);
		ASSERT (ret == 4);
		ASSERT (src == input + 2);
		ASSERT (!mbsinit (&state));

		src = input + 2;
		ret = mbsrtowcs (buf, &src, unlimited ? BUFSIZE : 2, &state);
		ASSERT (ret == (unlimited ? 4 : 2));
		ASSERT (src == (unlimited ? NULL : input + 7));
		ASSERT (wctob (buf[0]) == EOF);
		ASSERT (wctob (buf[1]) == EOF);
		if (unlimited)
		  {
		    ASSERT (buf[2] == 'e');
		    ASSERT (buf[3] == 'r');
		    ASSERT (buf[4] == 0);
		    ASSERT (buf[5] == (wchar_t) 0xBADFACE);
		  }
		else
		  ASSERT (buf[2] == (wchar_t) 0xBADFACE);
		ASSERT (mbsinit (&state));
	      }
	      return 0;

	    default:
	      return 1;
	    }
	}

      return 0;
    }

  return 1;
}