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

   1 // 1999-10-11 bkoz
   2
   3 // Copyright (C) 1999-2020 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::streambuf
  28 {
  29   char          buffer[4];
  30   std::string   result;
  31
  32 public:
  33
  34   std::string&
  35   get_result()
  36   { return result; }
  37
  38   setpbuf()
  39   {
  40     char 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   std::string text = "abcdefghijklmn";
  72
  73   // 01
  74   setpbuf sp1;
  75   // Here xsputn writes over sp1.result
  76   sp1.sputn(text.c_str(), text.length());
  77
  78   // This crashes when result is accessed
  79   sp1.pubsync();
  80   VERIFY( sp1.get_result() == text );
  81
  82   // 02
  83   setpbuf sp2;
  84   for (std::string::size_type i = 0; i < text.length(); ++i)
  85     {
  86       // sputc also writes over result
  87       sp2.sputc(text[i]);
  88     }
  89
  90   // Crash here
  91   sp2.pubsync();
  92   VERIFY( sp2.get_result() == text );
  93 }
  94
  95 int main()
  96 {
  97   test04();
  98   return 0;
  99 }