1bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant//===----------------------------------------------------------------------===// 2bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant// 3f5256e16dfc425c1d466f6308d4026d529ce9e0bHoward Hinnant// The LLVM Compiler Infrastructure 4bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant// 5b64f8b07c104c6cc986570ac8ee0ed16a9f23976Howard Hinnant// This file is dual licensed under the MIT and the University of Illinois Open 6b64f8b07c104c6cc986570ac8ee0ed16a9f23976Howard Hinnant// Source Licenses. See LICENSE.TXT for details. 7bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant// 8bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant//===----------------------------------------------------------------------===// 9bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant 10bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant// <istream> 11bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant 12bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant// basic_istream<charT,traits>& get(char_type* s, streamsize n); 13bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant 14bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant#include <istream> 15bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant#include <cassert> 16bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant 17bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnanttemplate <class CharT> 18bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnantstruct testbuf 19bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant : public std::basic_streambuf<CharT> 20bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant{ 21bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant typedef std::basic_string<CharT> string_type; 22bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant typedef std::basic_streambuf<CharT> base; 23bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnantprivate: 24bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant string_type str_; 25bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnantpublic: 26bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant 27bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant testbuf() {} 28bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant testbuf(const string_type& str) 29bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant : str_(str) 30bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant { 31bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant base::setg(const_cast<CharT*>(str_.data()), 32bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant const_cast<CharT*>(str_.data()), 33bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant const_cast<CharT*>(str_.data()) + str_.size()); 34bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant } 35bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant 36bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant CharT* eback() const {return base::eback();} 37bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant CharT* gptr() const {return base::gptr();} 38bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant CharT* egptr() const {return base::egptr();} 39bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant}; 40bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant 41bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnantint main() 42bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant{ 43bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant { 44bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant testbuf<char> sb(" \n \n "); 45bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant std::istream is(&sb); 46bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant char s[5]; 47bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant is.get(s, 5); 48bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant assert(!is.eof()); 49bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant assert(!is.fail()); 50bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant assert(std::string(s) == " "); 51bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant assert(is.gcount() == 2); 52bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant is.get(s, 5); 53bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant assert(!is.eof()); 54bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant assert( is.fail()); 55bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant assert(std::string(s) == ""); 56bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant assert(is.gcount() == 0); 57bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant is.clear(); 58bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant assert(is.get() == '\n'); 59bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant is.get(s, 5); 60bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant assert(!is.eof()); 61bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant assert(!is.fail()); 62bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant assert(std::string(s) == " "); 63bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant assert(is.gcount() == 4); 64bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant assert(is.get() == '\n'); 65bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant is.get(s, 5); 66bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant assert( is.eof()); 67bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant assert(!is.fail()); 68bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant assert(std::string(s) == " "); 69bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant assert(is.gcount() == 1); 70bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant } 71bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant { 72bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant testbuf<wchar_t> sb(L" \n \n "); 73bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant std::wistream is(&sb); 74bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant wchar_t s[5]; 75bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant is.get(s, 5); 76bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant assert(!is.eof()); 77bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant assert(!is.fail()); 78bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant assert(std::wstring(s) == L" "); 79bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant assert(is.gcount() == 2); 80bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant is.get(s, 5); 81bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant assert(!is.eof()); 82bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant assert( is.fail()); 83bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant assert(std::wstring(s) == L""); 84bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant assert(is.gcount() == 0); 85bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant is.clear(); 86bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant assert(is.get() == L'\n'); 87bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant is.get(s, 5); 88bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant assert(!is.eof()); 89bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant assert(!is.fail()); 90bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant assert(std::wstring(s) == L" "); 91bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant assert(is.gcount() == 4); 92bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant assert(is.get() == L'\n'); 93bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant is.get(s, 5); 94bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant assert( is.eof()); 95bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant assert(!is.fail()); 96bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant assert(std::wstring(s) == L" "); 97bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant assert(is.gcount() == 1); 98bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant } 99bc8d3f97eb5c958007f2713238472e0c1c8fe02Howard Hinnant} 100