Howard Hinnant | bc8d3f9 | 2010-05-11 19:42:16 +0000 | [diff] [blame] | 1 | //===----------------------------------------------------------------------===// |
| 2 | // |
Howard Hinnant | f5256e1 | 2010-05-11 21:36:01 +0000 | [diff] [blame] | 3 | // The LLVM Compiler Infrastructure |
Howard Hinnant | bc8d3f9 | 2010-05-11 19:42:16 +0000 | [diff] [blame] | 4 | // |
Howard Hinnant | b64f8b0 | 2010-11-16 22:09:02 +0000 | [diff] [blame] | 5 | // This file is dual licensed under the MIT and the University of Illinois Open |
| 6 | // Source Licenses. See LICENSE.TXT for details. |
Howard Hinnant | bc8d3f9 | 2010-05-11 19:42:16 +0000 | [diff] [blame] | 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: |
Dan Albert | 1d4a1ed | 2016-05-25 22:36:09 -0700 | [diff] [blame^] | 39 | int_type overflow(int_type c = traits_type::eof()) |
Howard Hinnant | bc8d3f9 | 2010-05-11 19:42:16 +0000 | [diff] [blame] | 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 | } |