123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311 |
- // (C) Copyright 2008 CodeRage, LLC (turkanis at coderage dot com)
- // (C) Copyright 2003-2007 Jonathan Turkanis
- // Distributed under the Boost Software License, Version 1.0. (See accompanying
- // file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt.)
- // See http://www.boost.org/libs/iostreams for documentation.
- #ifndef BOOST_IOSTREAMS_DETAIL_DIRECT_STREAMBUF_HPP_INCLUDED
- #define BOOST_IOSTREAMS_DETAIL_DIRECT_STREAMBUF_HPP_INCLUDED
- #if defined(_MSC_VER)
- # pragma once
- #endif
- #include <boost/assert.hpp>
- #include <cstddef>
- #include <utility> // pair.
- #include <boost/config.hpp> // BOOST_DEDUCED_TYPENAME,
- #include <boost/core/typeinfo.hpp>
- #include <boost/iostreams/detail/char_traits.hpp> // member template friends.
- #include <boost/iostreams/detail/config/wide_streams.hpp>
- #include <boost/iostreams/detail/error.hpp>
- #include <boost/iostreams/detail/execute.hpp>
- #include <boost/iostreams/detail/functional.hpp>
- #include <boost/iostreams/detail/ios.hpp>
- #include <boost/iostreams/detail/optional.hpp>
- #include <boost/iostreams/detail/streambuf.hpp>
- #include <boost/iostreams/detail/streambuf/linked_streambuf.hpp>
- #include <boost/iostreams/operations.hpp>
- #include <boost/iostreams/positioning.hpp>
- #include <boost/iostreams/traits.hpp>
- #include <boost/throw_exception.hpp>
- // Must come last.
- #include <boost/iostreams/detail/config/disable_warnings.hpp> // MSVC.
- namespace boost { namespace iostreams {
-
- namespace detail {
- template< typename T,
- typename Tr =
- BOOST_IOSTREAMS_CHAR_TRAITS(
- BOOST_DEDUCED_TYPENAME char_type_of<T>::type
- ) >
- class direct_streambuf
- : public linked_streambuf<BOOST_DEDUCED_TYPENAME char_type_of<T>::type, Tr>
- {
- public:
- typedef typename char_type_of<T>::type char_type;
- BOOST_IOSTREAMS_STREAMBUF_TYPEDEFS(Tr)
- private:
- typedef linked_streambuf<char_type, traits_type> base_type;
- typedef typename category_of<T>::type category;
- typedef BOOST_IOSTREAMS_BASIC_STREAMBUF(
- char_type, traits_type
- ) streambuf_type;
- public: // stream needs access.
- void open(const T& t, std::streamsize buffer_size,
- std::streamsize pback_size);
- bool is_open() const;
- void close();
- bool auto_close() const { return auto_close_; }
- void set_auto_close(bool close) { auto_close_ = close; }
- bool strict_sync() { return true; }
- // Declared in linked_streambuf.
- T* component() { return storage_.get(); }
- protected:
- BOOST_IOSTREAMS_USING_PROTECTED_STREAMBUF_MEMBERS(base_type)
- direct_streambuf();
- //--------------Virtual functions-----------------------------------------//
- // Declared in linked_streambuf.
- void close_impl(BOOST_IOS::openmode m);
- const boost::core::typeinfo& component_type() const { return BOOST_CORE_TYPEID(T); }
- void* component_impl() { return component(); }
- #ifdef BOOST_IOSTREAMS_NO_STREAM_TEMPLATES
- public:
- #endif
- // Declared in basic_streambuf.
- int_type underflow();
- int_type pbackfail(int_type c);
- int_type overflow(int_type c);
- pos_type seekoff( off_type off, BOOST_IOS::seekdir way,
- BOOST_IOS::openmode which );
- pos_type seekpos(pos_type sp, BOOST_IOS::openmode which);
- private:
- pos_type seek_impl( stream_offset off, BOOST_IOS::seekdir way,
- BOOST_IOS::openmode which );
- void init_input(any_tag) { }
- void init_input(input);
- void init_output(any_tag) { }
- void init_output(output);
- void init_get_area();
- void init_put_area();
- bool one_head() const;
- bool two_head() const;
- optional<T> storage_;
- char_type *ibeg_, *iend_, *obeg_, *oend_;
- bool auto_close_;
- };
-
- //------------------Implementation of direct_streambuf------------------------//
- template<typename T, typename Tr>
- direct_streambuf<T, Tr>::direct_streambuf()
- : ibeg_(0), iend_(0), obeg_(0), oend_(0), auto_close_(true)
- { this->set_true_eof(true); }
- template<typename T, typename Tr>
- void direct_streambuf<T, Tr>::open
- (const T& t, std::streamsize, std::streamsize)
- {
- storage_.reset(t);
- init_input(category());
- init_output(category());
- setg(0, 0, 0);
- setp(0, 0);
- this->set_needs_close();
- }
- template<typename T, typename Tr>
- bool direct_streambuf<T, Tr>::is_open() const
- { return ibeg_ != 0 || obeg_ != 0; }
- template<typename T, typename Tr>
- void direct_streambuf<T, Tr>::close()
- {
- base_type* self = this;
- detail::execute_all( detail::call_member_close(*self, BOOST_IOS::in),
- detail::call_member_close(*self, BOOST_IOS::out),
- detail::call_reset(storage_) );
- }
- template<typename T, typename Tr>
- typename direct_streambuf<T, Tr>::int_type
- direct_streambuf<T, Tr>::underflow()
- {
- if (!ibeg_)
- boost::throw_exception(cant_read());
- if (!gptr())
- init_get_area();
- return gptr() != iend_ ?
- traits_type::to_int_type(*gptr()) :
- traits_type::eof();
- }
- template<typename T, typename Tr>
- typename direct_streambuf<T, Tr>::int_type
- direct_streambuf<T, Tr>::pbackfail(int_type c)
- {
- using namespace std;
- if (!ibeg_)
- boost::throw_exception(cant_read());
- if (gptr() != 0 && gptr() != ibeg_) {
- gbump(-1);
- if (!traits_type::eq_int_type(c, traits_type::eof()))
- *gptr() = traits_type::to_char_type(c);
- return traits_type::not_eof(c);
- }
- boost::throw_exception(bad_putback());
- }
- template<typename T, typename Tr>
- typename direct_streambuf<T, Tr>::int_type
- direct_streambuf<T, Tr>::overflow(int_type c)
- {
- using namespace std;
- if (!obeg_)
- boost::throw_exception(BOOST_IOSTREAMS_FAILURE("no write access"));
- if (!pptr()) init_put_area();
- if (!traits_type::eq_int_type(c, traits_type::eof())) {
- if (pptr() == oend_)
- boost::throw_exception(
- BOOST_IOSTREAMS_FAILURE("write area exhausted")
- );
- *pptr() = traits_type::to_char_type(c);
- pbump(1);
- return c;
- }
- return traits_type::not_eof(c);
- }
- template<typename T, typename Tr>
- inline typename direct_streambuf<T, Tr>::pos_type
- direct_streambuf<T, Tr>::seekoff
- (off_type off, BOOST_IOS::seekdir way, BOOST_IOS::openmode which)
- { return seek_impl(off, way, which); }
- template<typename T, typename Tr>
- inline typename direct_streambuf<T, Tr>::pos_type
- direct_streambuf<T, Tr>::seekpos
- (pos_type sp, BOOST_IOS::openmode which)
- {
- return seek_impl(position_to_offset(sp), BOOST_IOS::beg, which);
- }
- template<typename T, typename Tr>
- void direct_streambuf<T, Tr>::close_impl(BOOST_IOS::openmode which)
- {
- if (which == BOOST_IOS::in && ibeg_ != 0) {
- setg(0, 0, 0);
- ibeg_ = iend_ = 0;
- }
- if (which == BOOST_IOS::out && obeg_ != 0) {
- sync();
- setp(0, 0);
- obeg_ = oend_ = 0;
- }
- boost::iostreams::close(*storage_, which);
- }
- template<typename T, typename Tr>
- typename direct_streambuf<T, Tr>::pos_type direct_streambuf<T, Tr>::seek_impl
- (stream_offset off, BOOST_IOS::seekdir way, BOOST_IOS::openmode which)
- {
- using namespace std;
- BOOST_IOS::openmode both = BOOST_IOS::in | BOOST_IOS::out;
- if (two_head() && (which & both) == both)
- boost::throw_exception(bad_seek());
- stream_offset result = -1;
- bool one = one_head();
- if (one && (pptr() != 0 || gptr()== 0))
- init_get_area(); // Switch to input mode, for code reuse.
- if (one || ((which & BOOST_IOS::in) != 0 && ibeg_ != 0)) {
- if (!gptr()) setg(ibeg_, ibeg_, iend_);
- ptrdiff_t next = 0;
- switch (way) {
- case BOOST_IOS::beg: next = off; break;
- case BOOST_IOS::cur: next = (gptr() - ibeg_) + off; break;
- case BOOST_IOS::end: next = (iend_ - ibeg_) + off; break;
- default: BOOST_ASSERT(0);
- }
- if (next < 0 || next > (iend_ - ibeg_))
- boost::throw_exception(bad_seek());
- setg(ibeg_, ibeg_ + next, iend_);
- result = next;
- }
- if (!one && (which & BOOST_IOS::out) != 0 && obeg_ != 0) {
- if (!pptr()) setp(obeg_, oend_);
- ptrdiff_t next = 0;
- switch (way) {
- case BOOST_IOS::beg: next = off; break;
- case BOOST_IOS::cur: next = (pptr() - obeg_) + off; break;
- case BOOST_IOS::end: next = (oend_ - obeg_) + off; break;
- default: BOOST_ASSERT(0);
- }
- if (next < 0 || next > (oend_ - obeg_))
- boost::throw_exception(bad_seek());
- pbump(static_cast<int>(next - (pptr() - obeg_)));
- result = next;
- }
- return offset_to_position(result);
- }
- template<typename T, typename Tr>
- void direct_streambuf<T, Tr>::init_input(input)
- {
- std::pair<char_type*, char_type*> p = input_sequence(*storage_);
- ibeg_ = p.first;
- iend_ = p.second;
- }
- template<typename T, typename Tr>
- void direct_streambuf<T, Tr>::init_output(output)
- {
- std::pair<char_type*, char_type*> p = output_sequence(*storage_);
- obeg_ = p.first;
- oend_ = p.second;
- }
- template<typename T, typename Tr>
- void direct_streambuf<T, Tr>::init_get_area()
- {
- setg(ibeg_, ibeg_, iend_);
- if (one_head() && pptr()) {
- gbump(static_cast<int>(pptr() - obeg_));
- setp(0, 0);
- }
- }
- template<typename T, typename Tr>
- void direct_streambuf<T, Tr>::init_put_area()
- {
- setp(obeg_, oend_);
- if (one_head() && gptr()) {
- pbump(static_cast<int>(gptr() - ibeg_));
- setg(0, 0, 0);
- }
- }
- template<typename T, typename Tr>
- inline bool direct_streambuf<T, Tr>::one_head() const
- { return ibeg_ && obeg_ && ibeg_ == obeg_; }
- template<typename T, typename Tr>
- inline bool direct_streambuf<T, Tr>::two_head() const
- { return ibeg_ && obeg_ && ibeg_ != obeg_; }
- //----------------------------------------------------------------------------//
- } // End namespace detail.
- } } // End namespaces iostreams, boost.
- #include <boost/iostreams/detail/config/enable_warnings.hpp> // MSVC
- #endif // #ifndef BOOST_IOSTREAMS_DETAIL_DIRECT_STREAMBUF_HPP_INCLUDED
|