src/examples/cpp03/ssl/client.cpp | src/examples/cpp11/ssl/client.cpp |
⋮ | ⋮ |
1 | // | 1 | // |
2 | //·client.cpp | 2 | //·client.cpp |
3 | //·~~~~~~~~~~ | 3 | //·~~~~~~~~~~ |
4 | // | 4 | // |
5 | //·Copyright·(c)·2003-2022·Christopher·M.·Kohlhoff·(chris·at·kohlhoff·dot·com) | 5 | //·Copyright·(c)·2003-2022·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·<cstring> |
| 13 | #include·<functional> |
12 | #include·<iostream> | 14 | #include·<iostream> |
13 | #include·<boost/bind/bind.hpp> | |
14 | #include·"asio.hpp" | 15 | #include·"asio.hpp" |
15 | #include·"asio/ssl.hpp" | 16 | #include·"asio/ssl.hpp" |
16 | | 17 | |
| 18 | using·asio::ip::tcp; |
| 19 | using·std::placeholders::_1; |
| 20 | using·std::placeholders::_2; |
| 21 | |
17 | enum·{·max_length·=·1024·}; | 22 | enum·{·max_length·=·1024·}; |
18 | | 23 | |
19 | class·client | 24 | class·client |
20 | { | 25 | { |
21 | public: | 26 | public: |
22 | ··client(asio::io_context&·io_context, | 27 | ··client(asio::io_context&·io_context, |
23 | ······asio::ssl::context&·context, | 28 | ······asio::ssl::context&·context, |
24 | ······asio::ip::tcp::resolver::results_type·endpoints) | 29 | ······const·tcp::resolver::results_type&·endpoints) |
25 | ····:·socket_(io_context,·context) | 30 | ····:·socket_(io_context,·context) |
26 | ··{ | 31 | ··{ |
27 | ····socket_.set_verify_mode(asio::ssl::verify_peer); | 32 | ····socket_.set_verify_mode(asio::ssl::verify_peer); |
28 | ····socket_.set_verify_callback( | 33 | ····socket_.set_verify_callback( |
29 | ········boost::bind(&client::verify_certificate,·this, | 34 | ········std::bind(&client::verify_certificate,·this,·_1,·_2)); |
30 | ··········boost::placeholders::_1,·boost::placeholders::_2)); | |
31 | | 35 | |
32 | ····asio::async_connect(socket_.lowest_layer(),·endpoints, | 36 | ····connect(endpoints); |
33 | ········boost::bind(&client::handle_connect,·this, | |
34 | ··········asio::placeholders::error)); | |
35 | ··} | 37 | ··} |
36 | | 38 | |
| 39 | private: |
37 | ··bool·verify_certificate(bool·preverified, | 40 | ··bool·verify_certificate(bool·preverified, |
38 | ······asio::ssl::verify_context&·ctx) | 41 | ······asio::ssl::verify_context&·ctx) |
39 | ··{ | 42 | ··{ |
40 | ····//·The·verify·callback·can·be·used·to·check·whether·the·certificate·that·is | 43 | ····//·The·verify·callback·can·be·used·to·check·whether·the·certificate·that·is |
41 | ····//·being·presented·is·valid·for·the·peer.·For·example,·RFC·2818·describes | 44 | ····//·being·presented·is·valid·for·the·peer.·For·example,·RFC·2818·describes |
42 | ····//·the·steps·involved·in·doing·this·for·HTTPS.·Consult·the·OpenSSL | 45 | ····//·the·steps·involved·in·doing·this·for·HTTPS.·Consult·the·OpenSSL |
43 | ····//·documentation·for·more·details.·Note·that·the·callback·is·called·once | 46 | ····//·documentation·for·more·details.·Note·that·the·callback·is·called·once |
44 | ····//·for·each·certificate·in·the·certificate·chain,·starting·from·the·root | 47 | ····//·for·each·certificate·in·the·certificate·chain,·starting·from·the·root |
45 | ····//·certificate·authority. | 48 | ····//·certificate·authority. |
46 | | 49 | |
47 | ····//·In·this·example·we·will·simply·print·the·certificate's·subject·name. | 50 | ····//·In·this·example·we·will·simply·print·the·certificate's·subject·name. |
48 | ····char·subject_name[256]; | 51 | ····char·subject_name[256]; |
49 | ····X509*·cert·=·X509_STORE_CTX_get_current_cert(ctx.native_handle()); | 52 | ····X509*·cert·=·X509_STORE_CTX_get_current_cert(ctx.native_handle()); |
50 | ····X509_NAME_oneline(X509_get_subject_name(cert),·subject_name,·256); | 53 | ····X509_NAME_oneline(X509_get_subject_name(cert),·subject_name,·256); |
51 | ····std::cout·<<·"Verifying·"·<<·subject_name·<<·"\n"; | 54 | ····std::cout·<<·"Verifying·"·<<·subject_name·<<·"\n"; |
52 | | 55 | |
53 | ····return·preverified; | 56 | ····return·preverified; |
54 | ··} | 57 | ··} |
55 | | 58 | |
56 | ··void·handle_connect(const·asio::error_code&·error) | 59 | ··void·connect(const·tcp::resolver::results_type&·endpoints) |
57 | ··{ | |
58 | ····if·(!error) | |
59 | ····{ | |
60 | ······socket_.async_handshake(asio::ssl::stream_base::client, | |
61 | ··········boost::bind(&client::handle_handshake,·this, | |
62 | ············asio::placeholders::error)); | |
63 | ····} | |
64 | ····else | |
65 | ····{ | |
66 | ······std::cout·<<·"Connect·failed:·"·<<·error.message()·<<·"\n"; | |
67 | ····} | |
68 | ··} | |
69 | | |
70 | ··void·handle_handshake(const·asio::error_code&·error) | |
71 | ··{ | |
72 | ····if·(!error) | |
73 | ····{ | |
74 | ······std::cout·<<·"Enter·message:·"; | |
75 | ······std::cin.getline(request_,·max_length); | |
76 | ······size_t·request_length·=·strlen(request_); | |
77 | | |
78 | ······asio::async_write(socket_, | |
79 | ··········asio::buffer(request_,·request_length), | |
80 | ··········boost::bind(&client::handle_write,·this, | |
81 | ············asio::placeholders::error, | |
82 | ············asio::placeholders::bytes_transferred)); | |
83 | ····} | |
84 | ····else | |
85 | ····{ | |
86 | ······std::cout·<<·"Handshake·failed:·"·<<·error.message()·<<·"\n"; | |
87 | ····} | |
88 | ··} | |
89 | | |
90 | ··void·handle_write(const·asio::error_code&·error, | |
91 | ······size_t·bytes_transferred) | |
92 | ··{ | 60 | ··{ |
93 | ····if·(!error) | 61 | ····asio::async_connect(socket_.lowest_layer(),·endpoints, |
94 | ····{ | 62 | ········[this](const·std::error_code&·error, |
95 | ······asio::async_read(socket_, | 63 | ··········const·tcp::endpoint&·/*endpoint*/) |
96 | ··········asio::buffer(reply_,·bytes_transferred), | 64 | ········{ |
97 | ··········boost::bind(&client::handle_read,·this, | 65 | ··········if·(!error) |
98 | ············asio::placeholders::error, | 66 | ··········{ |
99 | ············asio::placeholders::bytes_transferred)); | 67 | ············handshake(); |
100 | ····} | 68 | ··········} |
101 | ····else | 69 | ··········else |
102 | ····{ | 70 | ··········{ |
103 | ······std::cout·<<·"Write·failed:·"·<<·error.message()·<<·"\n"; | 71 | ············std::cout·<<·"Connect·failed:·"·<<·error.message()·<<·"\n"; |
104 | ····} | 72 | ··········} |
105 | ··} | 73 | ········}); |
106 | | 74 | ··} |
107 | ··void·handle_read(const·asio::error_code&·error, | 75 | |
108 | ······size_t·bytes_transferred) | 76 | ··void·handshake() |
109 | ··{ | 77 | ··{ |
110 | ····if·(!error) | 78 | ····socket_.async_handshake(asio::ssl::stream_base::client, |
111 | ····{ | 79 | ········[this](const·std::error_code&·error) |
112 | ······std::cout·<<·"Reply:·"; | 80 | ········{ |
113 | ······std::cout.write(reply_,·bytes_transferred); | 81 | ··········if·(!error) |
114 | ······std::cout·<<·"\n"; | 82 | ··········{ |
115 | ····} | 83 | ············send_request(); |
116 | ····else | 84 | ··········} |
117 | ····{ | 85 | ··········else |
118 | ······std::cout·<<·"Read·failed:·"·<<·error.message()·<<·"\n"; | 86 | ··········{ |
119 | ····} | 87 | ············std::cout·<<·"Handshake·failed:·"·<<·error.message()·<<·"\n"; |
| 88 | ··········} |
| 89 | ········}); |
| 90 | ··} |
| 91 | |
| 92 | ··void·send_request() |
| 93 | ··{ |
| 94 | ····std::cout·<<·"Enter·message:·"; |
| 95 | ····std::cin.getline(request_,·max_length); |
| 96 | ····size_t·request_length·=·std::strlen(request_); |
| 97 | |
| 98 | ····asio::async_write(socket_, |
| 99 | ········asio::buffer(request_,·request_length), |
| 100 | ········[this](const·std::error_code&·error,·std::size_t·length) |
| 101 | ········{ |
| 102 | ··········if·(!error) |
| 103 | ··········{ |
| 104 | ············receive_response(length); |
| 105 | ··········} |
| 106 | ··········else |
| 107 | ··········{ |
| 108 | ············std::cout·<<·"Write·failed:·"·<<·error.message()·<<·"\n"; |
| 109 | ··········} |
| 110 | ········}); |
| 111 | ··} |
| 112 | |
| 113 | ··void·receive_response(std::size_t·length) |
| 114 | ··{ |
| 115 | ····asio::async_read(socket_, |
| 116 | ········asio::buffer(reply_,·length), |
| 117 | ········[this](const·std::error_code&·error,·std::size_t·length) |
| 118 | ········{ |
| 119 | ··········if·(!error) |
| 120 | ··········{ |
| 121 | ············std::cout·<<·"Reply:·"; |
| 122 | ············std::cout.write(reply_,·length); |
| 123 | ············std::cout·<<·"\n"; |
| 124 | ··········} |
| 125 | ··········else |
| 126 | ··········{ |
| 127 | ············std::cout·<<·"Read·failed:·"·<<·error.message()·<<·"\n"; |
| 128 | ··········} |
| 129 | ········}); |
120 | ··} | 130 | ··} |
121 | | 131 | |
122 | private: | 132 | ··asio::ssl::stream<tcp::socket>·socket_; |
123 | ··asio::ssl::stream<asio::ip::tcp::socket>·socket_; | |
124 | ··char·request_[max_length]; | 133 | ··char·request_[max_length]; |
125 | ··char·reply_[max_length]; | 134 | ··char·reply_[max_length]; |
126 | }; | 135 | }; |
127 | | 136 | |
128 | int·main(int·argc,·char*·argv[]) | 137 | int·main(int·argc,·char*·argv[]) |
129 | { | 138 | { |
130 | ··try | 139 | ··try |
131 | ··{ | 140 | ··{ |
132 | ····if·(argc·!=·3) | 141 | ····if·(argc·!=·3) |
133 | ····{ | 142 | ····{ |
134 | ······std::cerr·<<·"Usage:·client·<host>·<port>\n"; | 143 | ······std::cerr·<<·"Usage:·client·<host>·<port>\n"; |
135 | ······return·1; | 144 | ······return·1; |
136 | ····} | 145 | ····} |
137 | | 146 | |
138 | ····asio::io_context·io_context; | 147 | ····asio::io_context·io_context; |
139 | | 148 | |
140 | ····asio::ip::tcp::resolver·resolver(io_context); | 149 | ····tcp::resolver·resolver(io_context); |
141 | ····asio::ip::tcp::resolver::results_type·endpoints·= | 150 | ····auto·endpoints·=·resolver.resolve(argv[1],·argv[2]); |
142 | ······resolver.resolve(argv[1],·argv[2]); | |
143 | | 151 | |
144 | ····asio::ssl::context·ctx(asio::ssl::context::sslv23); | 152 | ····asio::ssl::context·ctx(asio::ssl::context::sslv23); |
145 | ····ctx.load_verify_file("ca.pem"); | 153 | ····ctx.load_verify_file("ca.pem"); |
146 | | 154 | |
147 | ····client·c(io_context,·ctx,·endpoints); | 155 | ····client·c(io_context,·ctx,·endpoints); |
148 | | 156 | |
149 | ····io_context.run(); | 157 | ····io_context.run(); |
150 | ··} | 158 | ··} |
151 | ··catch·(std::exception&·e) | 159 | ··catch·(std::exception&·e) |
152 | ··{ | 160 | ··{ |
153 | ····std::cerr·<<·"Exception:·"·<<·e.what()·<<·"\n"; | 161 | ····std::cerr·<<·"Exception:·"·<<·e.what()·<<·"\n"; |
154 | ··} | 162 | ··} |
155 | | 163 | |
156 | ··return·0; | 164 | ··return·0; |
157 | } | 165 | } |