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