| // |
| // ssl/detail/buffered_handshake_op.hpp |
| // ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ |
| // |
| // Copyright (c) 2003-2015 Christopher M. Kohlhoff (chris at kohlhoff dot com) |
| // |
| // 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) |
| // |
| |
| #ifndef BOOST_ASIO_SSL_DETAIL_BUFFERED_HANDSHAKE_OP_HPP |
| #define BOOST_ASIO_SSL_DETAIL_BUFFERED_HANDSHAKE_OP_HPP |
| |
| #if defined(_MSC_VER) && (_MSC_VER >= 1200) |
| # pragma once |
| #endif // defined(_MSC_VER) && (_MSC_VER >= 1200) |
| |
| #include <boost/asio/detail/config.hpp> |
| |
| #if !defined(BOOST_ASIO_ENABLE_OLD_SSL) |
| # include <boost/asio/ssl/detail/engine.hpp> |
| #endif // !defined(BOOST_ASIO_ENABLE_OLD_SSL) |
| |
| #include <boost/asio/detail/push_options.hpp> |
| |
| namespace boost { |
| namespace asio { |
| namespace ssl { |
| namespace detail { |
| |
| #if !defined(BOOST_ASIO_ENABLE_OLD_SSL) |
| |
| template <typename ConstBufferSequence> |
| class buffered_handshake_op |
| { |
| public: |
| buffered_handshake_op(stream_base::handshake_type type, |
| const ConstBufferSequence& buffers) |
| : type_(type), |
| buffers_(buffers), |
| total_buffer_size_(boost::asio::buffer_size(buffers_)) |
| { |
| } |
| |
| engine::want operator()(engine& eng, |
| boost::system::error_code& ec, |
| std::size_t& bytes_transferred) const |
| { |
| typename ConstBufferSequence::const_iterator iter = buffers_.begin(); |
| typename ConstBufferSequence::const_iterator end = buffers_.end(); |
| std::size_t accumulated_size = 0; |
| |
| for (;;) |
| { |
| engine::want want = eng.handshake(type_, ec); |
| if (want != engine::want_input_and_retry |
| || bytes_transferred == total_buffer_size_) |
| return want; |
| |
| // Find the next buffer piece to be fed to the engine. |
| while (iter != end) |
| { |
| const_buffer buffer(*iter); |
| |
| // Skip over any buffers which have already been consumed by the engine. |
| if (bytes_transferred >= accumulated_size + buffer_size(buffer)) |
| { |
| accumulated_size += buffer_size(buffer); |
| ++iter; |
| continue; |
| } |
| |
| // The current buffer may have been partially consumed by the engine on |
| // a previous iteration. If so, adjust the buffer to point to the |
| // unused portion. |
| if (bytes_transferred > accumulated_size) |
| buffer = buffer + (bytes_transferred - accumulated_size); |
| |
| // Pass the buffer to the engine, and update the bytes transferred to |
| // reflect the total number of bytes consumed so far. |
| bytes_transferred += buffer_size(buffer); |
| buffer = eng.put_input(buffer); |
| bytes_transferred -= buffer_size(buffer); |
| break; |
| } |
| } |
| } |
| |
| template <typename Handler> |
| void call_handler(Handler& handler, |
| const boost::system::error_code& ec, |
| const std::size_t& bytes_transferred) const |
| { |
| handler(ec, bytes_transferred); |
| } |
| |
| private: |
| stream_base::handshake_type type_; |
| ConstBufferSequence buffers_; |
| std::size_t total_buffer_size_; |
| }; |
| |
| #endif // !defined(BOOST_ASIO_ENABLE_OLD_SSL) |
| |
| } // namespace detail |
| } // namespace ssl |
| } // namespace asio |
| } // namespace boost |
| |
| #include <boost/asio/detail/pop_options.hpp> |
| |
| #endif // BOOST_ASIO_SSL_DETAIL_BUFFERED_HANDSHAKE_OP_HPP |