ceph/src/boost/libs/asio/include/boost/asio/ssl/detail/buffered_handshake_op.hpp

   1 //
   2 // ssl/detail/buffered_handshake_op.hpp
   3 // ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
   4 //
   5 // Copyright (c) 2003-2016 Christopher M. Kohlhoff (chris at kohlhoff dot com)
   6 //
   7 // Distributed under the Boost Software License, Version 1.0. (See accompanying
   8 // file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
   9 //
  10
  11 #ifndef BOOST_ASIO_SSL_DETAIL_BUFFERED_HANDSHAKE_OP_HPP
  12 #define BOOST_ASIO_SSL_DETAIL_BUFFERED_HANDSHAKE_OP_HPP
  13
  14 #if defined(_MSC_VER) && (_MSC_VER >= 1200)
  15 # pragma once
  16 #endif // defined(_MSC_VER) && (_MSC_VER >= 1200)
  17
  18 #include <boost/asio/detail/config.hpp>
  19
  20 #if !defined(BOOST_ASIO_ENABLE_OLD_SSL)
  21 # include <boost/asio/ssl/detail/engine.hpp>
  22 #endif // !defined(BOOST_ASIO_ENABLE_OLD_SSL)
  23
  24 #include <boost/asio/detail/push_options.hpp>
  25
  26 namespace boost {
  27 namespace asio {
  28 namespace ssl {
  29 namespace detail {
  30
  31 #if !defined(BOOST_ASIO_ENABLE_OLD_SSL)
  32
  33 template <typename ConstBufferSequence>
  34 class buffered_handshake_op
  35 {
  36 public:
  37   buffered_handshake_op(stream_base::handshake_type type,
  38       const ConstBufferSequence& buffers)
  39     : type_(type),
  40       buffers_(buffers),
  41       total_buffer_size_(boost::asio::buffer_size(buffers_))
  42   {
  43   }
  44
  45   engine::want operator()(engine& eng,
  46       boost::system::error_code& ec,
  47       std::size_t& bytes_transferred) const
  48   {
  49     typename ConstBufferSequence::const_iterator iter = buffers_.begin();
  50     typename ConstBufferSequence::const_iterator end = buffers_.end();
  51     std::size_t accumulated_size = 0;
  52
  53     for (;;)
  54     {
  55       engine::want want = eng.handshake(type_, ec);
  56       if (want != engine::want_input_and_retry
  57           || bytes_transferred == total_buffer_size_)
  58         return want;
  59
  60       // Find the next buffer piece to be fed to the engine.
  61       while (iter != end)
  62       {
  63         const_buffer buffer(*iter);
  64
  65         // Skip over any buffers which have already been consumed by the engine.
  66         if (bytes_transferred >= accumulated_size + buffer_size(buffer))
  67         {
  68           accumulated_size += buffer_size(buffer);
  69           ++iter;
  70           continue;
  71         }
  72
  73         // The current buffer may have been partially consumed by the engine on
  74         // a previous iteration. If so, adjust the buffer to point to the
  75         // unused portion.
  76         if (bytes_transferred > accumulated_size)
  77           buffer = buffer + (bytes_transferred - accumulated_size);
  78
  79         // Pass the buffer to the engine, and update the bytes transferred to
  80         // reflect the total number of bytes consumed so far.
  81         bytes_transferred += buffer_size(buffer);
  82         buffer = eng.put_input(buffer);
  83         bytes_transferred -= buffer_size(buffer);
  84         break;
  85       }
  86     }
  87   }
  88
  89   template <typename Handler>
  90   void call_handler(Handler& handler,
  91       const boost::system::error_code& ec,
  92       const std::size_t& bytes_transferred) const
  93   {
  94     handler(ec, bytes_transferred);
  95   }
  96
  97 private:
  98   stream_base::handshake_type type_;
  99   ConstBufferSequence buffers_;
 100   std::size_t total_buffer_size_;
 101 };
 102
 103 #endif // !defined(BOOST_ASIO_ENABLE_OLD_SSL)
 104
 105 } // namespace detail
 106 } // namespace ssl
 107 } // namespace asio
 108 } // namespace boost
 109
 110 #include <boost/asio/detail/pop_options.hpp>
 111
 112 #endif // BOOST_ASIO_SSL_DETAIL_BUFFERED_HANDSHAKE_OP_HPP