| src/examples/cpp03/ssl/server.cpp | src/examples/cpp11/ssl/server.cpp |
| ⋮ | ⋮ |
| 1 | // | 1 | // |
| 2 | //·server.cpp | 2 | //·server.cpp |
| 3 | //·~~~~~~~~~~ | 3 | //·~~~~~~~~~~ |
| 4 | // | 4 | // |
| 5 | //·Copyright·(c)·2003-2018·Christopher·M.·Kohlhoff·(chris·at·kohlhoff·dot·com) | 5 | //·Copyright·(c)·2003-2018·Christopher·M.·Kohlhoff·(chris·at·kohlhoff·dot·com) |
| 6 | // | 6 | // |
| 7 | //·Distributed·under·the·Boost·Software·License,·Version·1.0.·(See·accompanying | 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) | 8 | //·file·LICENSE_1_0.txt·or·copy·at·http://www.boost.org/LICENSE_1_0.txt) |
| 9 | // | 9 | // |
| 10 | | 10 | |
| 11 | #include·<cstdlib> | 11 | #include·<cstdlib> |
| | 12 | #include·<functional> |
| 12 | #include·<iostream> | 13 | #include·<iostream> |
| 13 | #include·<boost/bind.hpp> | |
| 14 | #include·"asio.hpp" | 14 | #include·"asio.hpp" |
| 15 | #include·"asio/ssl.hpp" | 15 | #include·"asio/ssl.hpp" |
| 16 | | 16 | |
| 17 | typedef·asio::ssl::stream<asio::ip::tcp::socket>·ssl_socket; | 17 | using·asio::ip::tcp; |
| 18 | | 18 | |
| 19 | class·session | 19 | class·session·:·public·std::enable_shared_from_this<session> |
| 20 | { | 20 | { |
| 21 | public: | 21 | public: |
| 22 | ··session(asio::io_context&·io_context, | 22 | ··session(tcp::socket·socket,·asio::ssl::context&·context) |
| 23 | ······asio::ssl::context&·context) | 23 | ····:·socket_(std::move(socket),·context) |
| 24 | ····:·socket_(io_context,·context) | |
| 25 | ··{ | 24 | ··{ |
| 26 | ··} | 25 | ··} |
| 27 | | 26 | |
| 28 | ··ssl_socket::lowest_layer_type&·socket() | |
| 29 | ··{ | |
| 30 | ····return·socket_.lowest_layer(); | |
| 31 | ··} | |
| 32 | | |
| 33 | ··void·start() | 27 | ··void·start() |
| 34 | ··{ | 28 | ··{ |
| 35 | ····socket_.async_handshake(asio::ssl::stream_base::server, | 29 | ····do_handshake(); |
| 36 | ········boost::bind(&session::handle_handshake,·this, | |
| 37 | ··········asio::placeholders::error)); | |
| 38 | ··} | |
| 39 | | |
| 40 | ··void·handle_handshake(const·asio::error_code&·error) | |
| 41 | ··{ | |
| 42 | ····if·(!error) | |
| 43 | ····{ | |
| 44 | ······socket_.async_read_some(asio::buffer(data_,·max_length), | |
| 45 | ··········boost::bind(&session::handle_read,·this, | |
| 46 | ············asio::placeholders::error, | |
| 47 | ············asio::placeholders::bytes_transferred)); | |
| 48 | ····} | |
| 49 | ····else | |
| 50 | ····{ | |
| 51 | ······delete·this; | |
| 52 | ····} | |
| 53 | ··} | |
| 54 | | |
| 55 | ··void·handle_read(const·asio::error_code&·error, | |
| 56 | ······size_t·bytes_transferred) | |
| 57 | ··{ | |
| 58 | ····if·(!error) | |
| 59 | ····{ | |
| 60 | ······asio::async_write(socket_, | |
| 61 | ··········asio::buffer(data_,·bytes_transferred), | |
| 62 | ··········boost::bind(&session::handle_write,·this, | |
| 63 | ············asio::placeholders::error)); | |
| 64 | ····} | |
| 65 | ····else | |
| 66 | ····{ | |
| 67 | ······delete·this; | |
| 68 | ····} | |
| 69 | ··} | 30 | ··} |
| 70 | | 31 | |
| 71 | ··void·handle_write(const·asio::error_code&·error) | 32 | private: |
| | 33 | ··void·do_handshake() |
| 72 | ··{ | 34 | ··{ |
| 73 | ····if·(!error) | 35 | ····auto·self(shared_from_this()); |
| 74 | ····{ | 36 | ····socket_.async_handshake(asio::ssl::stream_base::server,· |
| 75 | ······socket_.async_read_some(asio::buffer(data_,·max_length), | 37 | ········[this,·self](const·std::error_code&·error) |
| 76 | ··········boost::bind(&session::handle_read,·this, | 38 | ········{ |
| 77 | ············asio::placeholders::error, | 39 | ··········if·(!error) |
| 78 | ············asio::placeholders::bytes_transferred)); | 40 | ··········{ |
| 79 | ····} | 41 | ············do_read(); |
| 80 | ····else | 42 | ··········} |
| 81 | ····{ | 43 | ········}); |
| 82 | ······delete·this; | 44 | ··} |
| 83 | ····} | 45 | |
| | 46 | ··void·do_read() |
| | 47 | ··{ |
| | 48 | ····auto·self(shared_from_this()); |
| | 49 | ····socket_.async_read_some(asio::buffer(data_), |
| | 50 | ········[this,·self](const·std::error_code&·ec,·std::size_t·length) |
| | 51 | ········{ |
| | 52 | ··········if·(!ec) |
| | 53 | ··········{ |
| | 54 | ············do_write(length); |
| | 55 | ··········} |
| | 56 | ········}); |
| | 57 | ··} |
| | 58 | |
| | 59 | ··void·do_write(std::size_t·length) |
| | 60 | ··{ |
| | 61 | ····auto·self(shared_from_this()); |
| | 62 | ····asio::async_write(socket_,·asio::buffer(data_,·length), |
| | 63 | ········[this,·self](const·std::error_code&·ec, |
| | 64 | ··········std::size_t·/*length*/) |
| | 65 | ········{ |
| | 66 | ··········if·(!ec) |
| | 67 | ··········{ |
| | 68 | ············do_read(); |
| | 69 | ··········} |
| | 70 | ········}); |
| 84 | ··} | 71 | ··} |
| 85 | | 72 | |
| 86 | private: | 73 | ··asio::ssl::stream<tcp::socket>·socket_; |
| 87 | ··ssl_socket·socket_; | 74 | ··char·data_[1024]; |
| 88 | ··enum·{·max_length·=·1024·}; | |
| 89 | ··char·data_[max_length]; | |
| 90 | }; | 75 | }; |
| 91 | | 76 | |
| 92 | class·server | 77 | class·server |
| 93 | { | 78 | { |
| 94 | public: | 79 | public: |
| 95 | ··server(asio::io_context&·io_context,·unsigned·short·port) | 80 | ··server(asio::io_context&·io_context,·unsigned·short·port) |
| 96 | ····:·io_context_(io_context), | 81 | ····:·acceptor_(io_context,·tcp::endpoint(tcp::v4(),·port)), |
| 97 | ······acceptor_(io_context, | |
| 98 | ··········asio::ip::tcp::endpoint(asio::ip::tcp::v4(),·port)), | |
| 99 | ······context_(asio::ssl::context::sslv23) | 82 | ······context_(asio::ssl::context::sslv23) |
| 100 | ··{ | 83 | ··{ |
| 101 | ····context_.set_options( | 84 | ····context_.set_options( |
| 102 | ········asio::ssl::context::default_workarounds | 85 | ········asio::ssl::context::default_workarounds |
| 103 | ········|·asio::ssl::context::no_sslv2 | 86 | ········|·asio::ssl::context::no_sslv2 |
| 104 | ········|·asio::ssl::context::single_dh_use); | 87 | ········|·asio::ssl::context::single_dh_use); |
| 105 | ····context_.set_password_callback(boost::bind(&server::get_password,·this)); | 88 | ····context_.set_password_callback(std::bind(&server::get_password,·this)); |
| 106 | ····context_.use_certificate_chain_file("server.pem"); | 89 | ····context_.use_certificate_chain_file("server.pem"); |
| 107 | ····context_.use_private_key_file("server.pem",·asio::ssl::context::pem); | 90 | ····context_.use_private_key_file("server.pem",·asio::ssl::context::pem); |
| 108 | ····context_.use_tmp_dh_file("dh2048.pem"); | 91 | ····context_.use_tmp_dh_file("dh2048.pem"); |
| 109 | | 92 | |
| 110 | ····start_accept(); | 93 | ····do_accept(); |
| 111 | ··} | 94 | ··} |
| 112 | | 95 | |
| | 96 | private: |
| 113 | ··std::string·get_password()·const | 97 | ··std::string·get_password()·const |
| 114 | ··{ | 98 | ··{ |
| 115 | ····return·"test"; | 99 | ····return·"test"; |
| 116 | ··} | 100 | ··} |
| 117 | | 101 | |
| 118 | ··void·start_accept() | 102 | ··void·do_accept() |
| 119 | ··{ | 103 | ··{ |
| 120 | ····session*·new_session·=·new·session(io_context_,·context_); | 104 | ····acceptor_.async_accept( |
| 121 | ····acceptor_.async_accept(new_session->socket(), | 105 | ········[this](const·std::error_code&·error,·tcp::socket·socket) |
| 122 | ········boost::bind(&server::handle_accept,·this,·new_session, | 106 | ········{ |
| 123 | ··········asio::placeholders::error)); | 107 | ··········if·(!error) |
| 124 | ··} | 108 | ··········{ |
| 125 | | 109 | ············std::make_shared<session>(std::move(socket),·context_)->start(); |
| 126 | ··void·handle_accept(session*·new_session, | 110 | ··········} |
| 127 | ······const·asio::error_code&·error) | |
| 128 | ··{ | |
| 129 | ····if·(!error) | |
| 130 | ····{ | |
| 131 | ······new_session->start(); | |
| 132 | ····} | |
| 133 | ····else | |
| 134 | ····{ | |
| 135 | ······delete·new_session; | |
| 136 | ····} | |
| 137 | | 111 | |
| 138 | ····start_accept(); | 112 | ··········do_accept(); |
| | 113 | ········}); |
| 139 | ··} | 114 | ··} |
| 140 | | 115 | |
| 141 | private: | 116 | ··tcp::acceptor·acceptor_; |
| 142 | ··asio::io_context&·io_context_; | |
| 143 | ··asio::ip::tcp::acceptor·acceptor_; | |
| 144 | ··asio::ssl::context·context_; | 117 | ··asio::ssl::context·context_; |
| 145 | }; | 118 | }; |
| 146 | | 119 | |
| 147 | int·main(int·argc,·char*·argv[]) | 120 | int·main(int·argc,·char*·argv[]) |
| 148 | { | 121 | { |
| 149 | ··try | 122 | ··try |
| 150 | ··{ | 123 | ··{ |
| 151 | ····if·(argc·!=·2) | 124 | ····if·(argc·!=·2) |
| 152 | ····{ | 125 | ····{ |
| 153 | ······std::cerr·<<·"Usage:·server·<port>\n"; | 126 | ······std::cerr·<<·"Usage:·server·<port>\n"; |
| 154 | ······return·1; | 127 | ······return·1; |
| 155 | ····} | 128 | ····} |
| 156 | | 129 | |
| 157 | ····asio::io_context·io_context; | 130 | ····asio::io_context·io_context; |
| 158 | | 131 | |
| 159 | ····using·namespace·std;·//·For·atoi. | 132 | ····using·namespace·std;·//·For·atoi. |
| 160 | ····server·s(io_context,·atoi(argv[1])); | 133 | ····server·s(io_context,·atoi(argv[1])); |
| 161 | | 134 | |
| 162 | ····io_context.run(); | 135 | ····io_context.run(); |
| 163 | ··} | 136 | ··} |
| 164 | ··catch·(std::exception&·e) | 137 | ··catch·(std::exception&·e) |
| 165 | ··{ | 138 | ··{ |
| 166 | ····std::cerr·<<·"Exception:·"·<<·e.what()·<<·"\n"; | 139 | ····std::cerr·<<·"Exception:·"·<<·e.what()·<<·"\n"; |
| 167 | ··} | 140 | ··} |
| 168 | | 141 | |
| 169 | ··return·0; | 142 | ··return·0; |
| 170 | } | 143 | } |