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