1 //===----------------------------------------------------------------------===// 2 // 3 // The LLVM Compiler Infrastructure 4 // 5 // This file is dual licensed under the MIT and the University of Illinois Open 6 // Source Licenses. See LICENSE.TXT for details. 7 // 8 //===----------------------------------------------------------------------===// 9 10 // <streambuf> 11 12 // template <class charT, class traits = char_traits<charT> > 13 // class basic_streambuf; 14 15 // int_type sputc(char_type c); 16 17 #include <streambuf> 18 #include <cassert> 19 20 int overflow_called = 0; 21 22 struct test 23 : public std::basic_streambuf<char> 24 { 25 typedef std::basic_streambuf<char> base; 26 27 test() {} 28 29 void setg(char* gbeg, char* gnext, char* gend) 30 { 31 base::setg(gbeg, gnext, gend); 32 } 33 void setp(char* pbeg, char* pend) 34 { 35 base::setp(pbeg, pend); 36 } 37 38 protected: 39 int_type overflow(int_type c = traits_type::eof()) 40 { 41 ++overflow_called; 42 return 'a'; 43 } 44 }; 45 46 int main() 47 { 48 { 49 test t; 50 assert(overflow_called == 0); 51 assert(t.sputc('A') == 'a'); 52 assert(overflow_called == 1); 53 char out[3] = {0}; 54 t.setp(out, out+sizeof(out)); 55 assert(t.sputc('A') == 'A'); 56 assert(overflow_called == 1); 57 assert(out[0] == 'A'); 58 assert(t.sputc('B') == 'B'); 59 assert(overflow_called == 1); 60 assert(out[0] == 'A'); 61 assert(out[1] == 'B'); 62 } 63 } 64
