libio/tst-memstream4.c

   1 /* Test for open_memstream BZ #21037.
   2    Copyright (C) 2018-2020 Free Software Foundation, Inc.
   3    This file is part of the GNU C Library.
   4
   5    The GNU C Library is free software; you can redistribute it and/or
   6    modify it under the terms of the GNU Lesser General Public
   7    License as published by the Free Software Foundation; either
   8    version 2.1 of the License, or (at your option) any later version.
   9
  10    The GNU C Library is distributed in the hope that it will be useful,
  11    but WITHOUT ANY WARRANTY; without even the implied warranty of
  12    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13    Lesser General Public License for more details.
  14
  15    You should have received a copy of the GNU Lesser General Public
  16    License along with the GNU C Library; if not, see
  17    <https://www.gnu.org/licenses/>.  */
  18
  19 #include "tst-memstream.h"
  20
  21 static void
  22 mcheck_abort (enum mcheck_status ev)
  23 {
  24   printf ("mecheck failed with status %d\n", (int) ev);
  25   exit (1);
  26 }
  27
  28 static int
  29 do_test (void)
  30 {
  31   mcheck_pedantic (mcheck_abort);
  32
  33   /* Check if freopen proper fflush the stream.  */
  34   {
  35     CHAR_T old[] = W("old");
  36     CHAR_T *buf = old;
  37     size_t size;
  38
  39     FILE *fp = OPEN_MEMSTREAM (&buf, &size);
  40     TEST_VERIFY_EXIT (fp != NULL);
  41
  42     FPUTS (W("new"), fp);
  43     /* The stream buffer pointer should be updated with only a fflush or
  44        fclose.  */
  45     TEST_COMPARE (STRCMP (buf, old), 0);
  46
  47     /* The old stream should be fflush the stream, even for an invalid
  48        streams.  */
  49     FILE *nfp = freopen ("invalid-file", "r", fp);
  50     TEST_VERIFY_EXIT (nfp == NULL);
  51
  52     TEST_VERIFY (STRCMP (buf, W("new")) == 0);
  53     TEST_COMPARE_BLOB (buf, STRLEN (buf) * sizeof (CHAR_T),
  54                        W("new"), STRLEN (W("new")) * sizeof (CHAR_T));
  55
  56     TEST_COMPARE (fclose (fp), 0);
  57
  58     free (buf);
  59   }
  60
  61   return 0;
  62 }
  63
  64 #include <support/test-driver.c>