libio/tst_putwc.c

   1 /* Simple test of putwc in the C locale.
   2    Copyright (C) 2000-2014 Free Software Foundation, Inc.
   3    This file is part of the GNU C Library.
   4    Contributed by Ulrich Drepper <drepper@cygnus.com>, 2000.
   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 <errno.h>
  21 #include <error.h>
  22 #include <stdio.h>
  23 #include <stdlib.h>
  24 #include <wchar.h>
  25
  26 static const char outname[] = OBJPFX "tst_putwc.temp";
  27
  28 /* Prototype for our test function.  */
  29 int do_test (void);
  30 #define TEST_FUNCTION do_test ()
  31
  32 /* This defines the `main' function and some more.  */
  33 #include <test-skeleton.c>
  34
  35 int
  36 do_test (void)
  37 {
  38   const wchar_t str[] = L"This is a test of putwc\n";
  39   wchar_t buf[100];
  40   size_t n = 0;
  41   FILE *fp;
  42   int res = 0;
  43
  44   add_temp_file (outname);
  45
  46   fp = fopen (outname, "w+");
  47   if (fp == NULL)
  48     error (EXIT_FAILURE, errno, "cannot open temporary file");
  49
  50   for (n = 0; str[n] != L'\0'; ++n)
  51     putwc (str[n], fp);
  52
  53   /* First try reading after rewinding.  */
  54   rewind (fp);
  55
  56   wmemset (buf, L'\0', sizeof (buf) / sizeof (buf[0]));
  57   n = 0;
  58   while (! feof (fp) && n < sizeof (buf) - 1)
  59     {
  60       buf[n] = getwc (fp);
  61       if (buf[n] == WEOF)
  62         break;
  63       ++n;
  64     }
  65   buf[n] = L'\0';
  66
  67   if (wcscmp (buf, L"This is a test of putwc\n") != 0)
  68     {
  69       puts ("first comparison failed");
  70       res = 1;
  71     }
  72
  73   /* Now close the file, open it again, and read again.  */
  74   if (fclose (fp) != 0)
  75     {
  76       printf ("failure during fclose: %m\n");
  77       res = 1;
  78     }
  79
  80   fp = fopen (outname, "r");
  81   if (fp == NULL)
  82     {
  83       printf ("cannot reopen file: %m\n");
  84       return 1;
  85     }
  86
  87   /* We can remove the file now.  */
  88   remove (outname);
  89
  90   wmemset (buf, L'\0', sizeof (buf) / sizeof (buf[0]));
  91   n = 0;
  92   while (! feof (fp) && n < sizeof (buf) - 1)
  93     {
  94       buf[n] = getwc (fp);
  95       if (buf[n] == WEOF)
  96         break;
  97       ++n;
  98     }
  99   buf[n] = L'\0';
 100
 101   if (wcscmp (buf, L"This is a test of putwc\n") != 0)
 102     {
 103       puts ("second comparison failed");
 104       res = 1;
 105     }
 106
 107   if (fclose (fp) != 0)
 108     {
 109       printf ("failure during fclose: %m\n");
 110       res = 1;
 111     }
 112
 113   /* Next test: write a bit more than a few bytes.  */
 114   fp = fopen (outname, "w");
 115   if (fp == NULL)
 116     error (EXIT_FAILURE, errno, "cannot open temporary file");
 117
 118   for (n = 0; n < 4098; ++n)
 119     putwc (n & 255, fp);
 120
 121   fclose (fp);
 122
 123   return res;
 124 }