libstdc++-v3/testsuite/27_io/basic_filebuf/close/wchar_t/12790-3.cc

   1 // Copyright (C) 2003-2024 Free Software Foundation, Inc.
   2 //
   3 // This file is part of the GNU ISO C++ Library.  This library is free
   4 // software; you can redistribute it and/or modify it under the
   5 // terms of the GNU General Public License as published by the
   6 // Free Software Foundation; either version 3, or (at your option)
   7 // any later version.
   8
   9 // This library is distributed in the hope that it will be useful,
  10 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  11 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12 // GNU General Public License for more details.
  13
  14 // You should have received a copy of the GNU General Public License along
  15 // with this library; see the file COPYING3.  If not see
  16 // <http://www.gnu.org/licenses/>.
  17
  18 // 27.8.1.3 filebuf member functions
  19
  20 // { dg-require-fileio "" }
  21
  22 #include <locale>
  23 #include <fstream>
  24 #include <testsuite_hooks.h>
  25
  26 class Cvt : public std::codecvt<wchar_t, char, std::mbstate_t>
  27 {
  28 public:
  29   mutable bool unshift_called;
  30
  31   Cvt()
  32   : unshift_called(false)
  33   { }
  34
  35 protected:
  36   bool
  37   do_always_noconv() const throw()
  38   { return false; }
  39
  40   int
  41   do_encoding() const throw()
  42   { return -1; }
  43
  44   std::codecvt_base::result
  45   do_unshift(std::mbstate_t&, char* to, char*, char*& to_next) const
  46   {
  47     unshift_called = true;
  48     to_next = to;
  49     return codecvt_base::error;
  50   }
  51 };
  52
  53 // libstdc++/12790
  54 // basic_filebuf::close() should fail if codecvt::unshift() fails
  55 void test01()
  56 {
  57   using namespace std;
  58
  59   const char* name = "tmp_close_12790";
  60
  61   Cvt* cvt = new Cvt;
  62   locale loc(locale::classic(), cvt);
  63
  64   wfilebuf fb;
  65   fb.pubimbue(loc);
  66
  67   fb.open(name, ios_base::out);
  68   fb.sputc('a');
  69
  70   VERIFY( !cvt->unshift_called );
  71   wfilebuf* ret = fb.close();
  72   VERIFY( cvt->unshift_called );
  73   VERIFY( !ret );
  74 }
  75
  76 int main()
  77 {
  78   test01();
  79   return 0;
  80 }