libstdc++-v3/testsuite/27_io/basic_stringbuf/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 <sstream>
  24 #include <testsuite_hooks.h>
  25
  26 class setpbuf : public std::stringbuf
  27 {
  28   char          buffer[4];
  29   std::string   result;
  30
  31 public:
  32
  33   std::string&
  34   get_result()
  35   { return result; }
  36
  37   setpbuf()
  38   {
  39     char foo [32];
  40     setp(foo, foo + 32);
  41     setp(buffer, buffer + 4);
  42   }
  43
  44   ~setpbuf()
  45   { sync(); }
  46
  47   virtual int_type
  48   overflow(int_type n)
  49   {
  50     if (sync() != 0)
  51       return traits_type::eof();
  52
  53     result += traits_type::to_char_type(n);
  54
  55     return n;
  56   }
  57
  58   virtual int
  59   sync()
  60   {
  61     result.append(pbase(), pptr());
  62     setp(buffer, buffer + 4);
  63     return 0;
  64   }
  65 };
  66
  67 // libstdc++/1057
  68 void test04()
  69 {
  70   std::string text = "abcdefghijklmn";
  71
  72   // 01
  73   setpbuf sp1;
  74   // Here xsputn writes over sp1.result
  75   sp1.sputn(text.c_str(), text.length());
  76
  77   // This crashes when result is accessed
  78   sp1.pubsync();
  79   VERIFY( sp1.get_result() == text );
  80
  81   // 02
  82   setpbuf sp2;
  83   for (std::string::size_type i = 0; i < text.length(); ++i)
  84     {
  85       // sputc also writes over result
  86       sp2.sputc(text[i]);
  87     }
  88
  89   // Crash here
  90   sp2.pubsync();
  91   VERIFY( sp2.get_result() == text );
  92 }
  93
  94 int main()
  95 {
  96   test04();
  97   return 0;
  98 }