wcsmbs/tst-mbsrtowcs.c

   1 /* Test NUL handling of mbsrtowcs.
   2    Copyright (C) 2004 Free Software Foundation, Inc.
   3    This file is part of the GNU C Library.
   4    Contributed by Ulrich Drepper <drepper@redhat.com>, 2004.
   5
   6    The GNU C Library is free software; you can redistribute it and/or
   7    modify it under the terms of the GNU Lesser General Public
   8    License as published by the Free Software Foundation; either
   9    version 2.1 of the License, or (at your option) any later version.
  10
  11    The GNU C Library is distributed in the hope that it will be useful,
  12    but WITHOUT ANY WARRANTY; without even the implied warranty of
  13    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14    Lesser General Public License for more details.
  15
  16    You should have received a copy of the GNU Lesser General Public
  17    License along with the GNU C Library; if not, write to the Free
  18    Software Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA
  19    02111-1307 USA.  */
  20
  21 #include <stdio.h>
  22 #include <string.h>
  23 #include <wchar.h>
  24
  25 int
  26 main (void)
  27 {
  28   const unsigned char buf[] = { 'a', 'b', '\0', 'c', 'd', '\0', 'e' };
  29   wchar_t out[sizeof (buf)];
  30   memset (out, '\xff', sizeof (out));
  31
  32   mbstate_t state;
  33   memset (&state, '\0', sizeof (state));
  34
  35   const char *in = buf;
  36   size_t n = mbsrtowcs (out, &in, sizeof (out) / sizeof (wchar_t), &state);
  37
  38   int result = 0;
  39   if (n != 2)
  40     {
  41       printf ("n = %zu, expected 2\n", n);
  42       result = 1;
  43     }
  44   if (in != NULL)
  45     {
  46       printf ("in = %p, expected NULL\n", in);
  47       result = 1;
  48     }
  49   if (out[0] != L'a')
  50     {
  51       printf ("out[0] = L'%lc', expected L'a'\n", (wint_t) out[0]);
  52       result = 1;
  53     }
  54   if (out[1] != L'b')
  55     {
  56       printf ("out[1] = L'%lc', expected L'b'\n", (wint_t) out[1]);
  57       result = 1;
  58     }
  59   if (out[2] != L'\0')
  60     {
  61       printf ("out[2] = L'%lc', expected L'\\0'\n", (wint_t) out[2]);
  62       result = 1;
  63     }
  64   return result;
  65 }