libstdc++-v3/testsuite/27_io/basic_streambuf/sync/wchar_t/1057.cc

   1 // 1999-10-11 bkoz
   2
   3 // Copyright (C) 1999-2016 Free Software Foundation, Inc.
   4 //
   5 // This file is part of the GNU ISO C++ Library.  This library is free
   6 // software; you can redistribute it and/or modify it under the
   7 // terms of the GNU General Public License as published by the
   8 // Free Software Foundation; either version 3, or (at your option)
   9 // any later version.
  10
  11 // This 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
  14 // GNU General Public License for more details.
  15
  16 // You should have received a copy of the GNU General Public License along
  17 // with this library; see the file COPYING3.  If not see
  18 // <http://www.gnu.org/licenses/>.
  19
  20
  21 // 27.5.2 template class basic_streambuf
  22
  23 #include <streambuf>
  24 #include <string>
  25 #include <testsuite_hooks.h>
  26
  27 class setpbuf : public std::wstreambuf
  28 {
  29   wchar_t       buffer[4];
  30   std::wstring  result;
  31
  32 public:
  33
  34   std::wstring&
  35   get_result()
  36   { return result; }
  37
  38   setpbuf()
  39   {
  40     wchar_t foo[32];
  41     setp(foo, foo + 32);
  42     setp(buffer, buffer + 4);
  43   }
  44
  45   ~setpbuf()
  46   { sync(); }
  47
  48   virtual int_type
  49   overflow(int_type n)
  50   {
  51     if (sync() != 0)
  52       return traits_type::eof();
  53
  54     result += traits_type::to_char_type(n);
  55
  56     return n;
  57   }
  58
  59   virtual int
  60   sync()
  61   {
  62     result.append(pbase(), pptr());
  63     setp(buffer, buffer + 4);
  64     return 0;
  65   }
  66 };
  67
  68 // libstdc++/1057
  69 void test04()
  70 {
  71   bool test __attribute__((unused)) = true;
  72   std::wstring text = L"abcdefghijklmn";
  73
  74   // 01
  75   setpbuf sp1;
  76   // Here xsputn writes over sp1.result
  77   sp1.sputn(text.c_str(), text.length());
  78
  79   // This crashes when result is accessed
  80   sp1.pubsync();
  81   VERIFY( sp1.get_result() == text );
  82
  83   // 02
  84   setpbuf sp2;
  85   for (std::wstring::size_type i = 0; i < text.length(); ++i)
  86     {
  87       // sputc also writes over result
  88       sp2.sputc(text[i]);
  89     }
  90
  91   // Crash here
  92   sp2.pubsync();
  93   VERIFY( sp2.get_result() == text );
  94 }
  95
  96 int main()
  97 {
  98   test04();
  99   return 0;
 100 }