1  /* Test that assigning to stdout redirects puts, putchar, etc (BZ#24051)
       2     Copyright (C) 2019-2023 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  
      20  /* Prevent putchar -> _IO_putc inline expansion.  */
      21  #define __NO_INLINE__
      22  #pragma GCC optimize("O0")
      23  
      24  #include <stdio.h>
      25  #include <string.h>
      26  #include <wchar.h>
      27  
      28  #include <array_length.h>
      29  #include <support/check.h>
      30  #include <support/temp_file.h>
      31  #include <support/test-driver.h>
      32  
      33  #undef putchar
      34  #undef putwchar
      35  
      36  static int
      37  do_test_narrow (void)
      38  {
      39    char buf[100];
      40    int fd = create_temp_file ("tst-bz24051", NULL);
      41    stdout = fdopen (fd, "w+");
      42    TEST_VERIFY_EXIT (stdout != NULL);
      43  
      44    printf ("ab%s", "cd");
      45    putchar ('e');
      46    putchar_unlocked ('f');
      47    puts ("ghi");
      48  
      49    rewind (stdout);
      50    TEST_VERIFY_EXIT (fgets (buf, sizeof (buf), stdout) != NULL);
      51    TEST_VERIFY (strcmp (buf, "abcdefghi\n") == 0);
      52  
      53    return 0;
      54  }
      55  
      56  static int
      57  do_test_wide (void)
      58  {
      59    wchar_t buf[100];
      60    int fd = create_temp_file ("tst-bz24051w", NULL);
      61    stdout = fdopen (fd, "w+");
      62    TEST_VERIFY_EXIT (stdout != NULL);
      63  
      64    wprintf (L"ab%ls", L"cd");
      65    putwchar (L'e');
      66    putwchar_unlocked (L'f');
      67  
      68    rewind (stdout);
      69    TEST_VERIFY_EXIT (fgetws (buf, array_length (buf), stdout) != NULL);
      70    TEST_VERIFY (wcscmp (buf, L"abcdef") == 0);
      71  
      72    return 0;
      73  }
      74  
      75  static int
      76  do_test (void)
      77  {
      78    return do_test_narrow () + do_test_wide ();
      79  }
      80  
      81  #include <support/test-driver.c>