wcsmbs/tst-mbsrtowcs.c

   1 /* Test NUL handling of mbsrtowcs.
   2    Copyright (C) 2004-2014 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, see
  18    <http://www.gnu.org/licenses/>.  */
  19
  20 #include <stdio.h>
  21 #include <string.h>
  22 #include <wchar.h>
  23
  24 int
  25 main (void)
  26 {
  27   const unsigned char buf[] = { 'a', 'b', '\0', 'c', 'd', '\0', 'e' };
  28   wchar_t out[sizeof (buf)];
  29   memset (out, '\xff', sizeof (out));
  30
  31   mbstate_t state;
  32   memset (&state, '\0', sizeof (state));
  33
  34   const char *in = (const char *) buf;
  35   size_t n = mbsrtowcs (out, &in, sizeof (out) / sizeof (wchar_t), &state);
  36
  37   int result = 0;
  38   if (n != 2)
  39     {
  40       printf ("n = %zu, expected 2\n", n);
  41       result = 1;
  42     }
  43   if (in != NULL)
  44     {
  45       printf ("in = %p, expected NULL\n", in);
  46       result = 1;
  47     }
  48   if (out[0] != L'a')
  49     {
  50       printf ("out[0] = L'%lc', expected L'a'\n", (wint_t) out[0]);
  51       result = 1;
  52     }
  53   if (out[1] != L'b')
  54     {
  55       printf ("out[1] = L'%lc', expected L'b'\n", (wint_t) out[1]);
  56       result = 1;
  57     }
  58   if (out[2] != L'\0')
  59     {
  60       printf ("out[2] = L'%lc', expected L'\\0'\n", (wint_t) out[2]);
  61       result = 1;
  62     }
  63   return result;
  64 }