espressif/esp-protocols
1
0
Fork 1
mirror of https://github.com/espressif/esp-protocols.git synced 2025-07-29 10:17:30 +02:00
Code Issues Packages Projects Releases Wiki Activity

asio: make the example code conform to Espressif C++ standards

Browse Source 
* Original commit: espressif/esp-idf@b2150f86a5
This commit is contained in:
David Čermák
2020-06-18 16:16:41 +08:00
committed by gabsuren
parent c05558ba28
commit f00c3be139
1 changed files with 73 additions and 97 deletions
Download Patch File Download Diff File Expand all files Collapse all files

96
examples/protocols/asio/ssl_client_server/main/asio_ssl_main.cpp
View File

@ -27,19 +27,19 @@ extern const unsigned char cacert_pem_end[] asm("_binary_ca_crt_end");
extern const unsigned char prvtkey_pem_start[] asm("_binary_server_key_start");
extern const unsigned char prvtkey_pem_end[] asm("_binary_server_key_end");
const asio::const_buffer cert_chain(cacert_pem_start, cacert_pem_end - cacert_pem_start);
const asio::const_buffer privkey(prvtkey_pem_start, prvtkey_pem_end - prvtkey_pem_start);
const asio::const_buffer server_cert(server_pem_start, server_pem_end - server_pem_start);
static const asio::const_buffer cert_chain(cacert_pem_start, cacert_pem_end - cacert_pem_start);
static const asio::const_buffer privkey(prvtkey_pem_start, prvtkey_pem_end - prvtkey_pem_start);
static const asio::const_buffer server_cert(server_pem_start, server_pem_end - server_pem_start);
using asio::ip::tcp;
enum { max_length = 1024 };
static const std::size_t max_length = 1024;
class Client {
public:
Client(asio::io_context& io_context,
asio::ssl::context& context,
const tcp::resolver::results_type& endpoints)
Client(asio::io_context &io_context,
asio::ssl::context &context,
const tcp::resolver::results_type &endpoints)
: socket_(io_context, context)
{
@ -53,18 +53,14 @@ public:
}
private:
void connect(const tcp::resolver::results_type& endpoints)
void connect(const tcp::resolver::results_type &endpoints)
{
asio::async_connect(socket_.lowest_layer(), endpoints,
[this](const std::error_code& error,
const tcp::endpoint& /*endpoint*/)
{
if (!error)
{
[this](const std::error_code & error,
const tcp::endpoint & /*endpoint*/) {
if (!error) {
handshake();
}
else
{
} else {
std::cout << "Connect failed: " << error.message() << "\n";
}
});
@ -73,14 +69,10 @@ private:
void handshake()
{
socket_.async_handshake(asio::ssl::stream_base::client,
[this](const std::error_code& error)
{
if (!error)
{
[this](const std::error_code & error) {
if (!error) {
send_request();
}
else
{
} else {
std::cout << "Handshake failed: " << error.message() << "\n";
}
});
@ -92,14 +84,10 @@ private:
asio::async_write(socket_,
asio::buffer(request_, request_length),
[this](const std::error_code& error, std::size_t length)
{
if (!error)
{
[this](const std::error_code & error, std::size_t length) {
if (!error) {
receive_response(length);
}
else
{
} else {
std::cout << "Write failed: " << error.message() << "\n";
}
});
@ -109,20 +97,15 @@ private:
{
asio::async_read(socket_,
asio::buffer(reply_, length),
[this](const std::error_code& error, std::size_t length)
{
if (!error)
{
[this](const std::error_code & error, std::size_t length) {
if (!error) {
std::cout << "Reply: ";
std::cout.write(reply_, length);
std::cout << "\n";
}
else
{
} else {
std::cout << "Read failed: " << error.message() << "\n";
}
});
}
asio::ssl::stream<tcp::socket> socket_;
@ -132,7 +115,7 @@ private:
class Session : public std::enable_shared_from_this<Session> {
public:
Session(tcp::socket socket, asio::ssl::context& context)
Session(tcp::socket socket, asio::ssl::context &context)
: socket_(std::move(socket), context)
{
}
@ -147,10 +130,8 @@ private:
{
auto self(shared_from_this());
socket_.async_handshake(asio::ssl::stream_base::server,
[this, self](const std::error_code& error)
{
if (!error)
{
[this, self](const std::error_code & error) {
if (!error) {
do_read();
}
});
@ -160,12 +141,11 @@ private:
{
auto self(shared_from_this());
socket_.async_read_some(asio::buffer(data_),
[this, self](const std::error_code& ec, std::size_t length)
{
if (!ec)
{
data_[length] = 0;
std::cout << "Server received: " << data_ << std::endl;
[this, self](const std::error_code & ec, std::size_t length) {
if (!ec) {
std::cout << "Server received: ";
std::cout.write(data_, length);
std::cout << std::endl;
do_write(length);
}
});
@ -175,11 +155,9 @@ private:
{
auto self(shared_from_this());
asio::async_write(socket_, asio::buffer(data_, length),
[this, self](const std::error_code& ec,
std::size_t /*length*/)
{
if (!ec)
{
[this, self](const std::error_code & ec,
std::size_t /*length*/) {
if (!ec) {
do_read();
}
});
@ -191,7 +169,7 @@ private:
class Server {
public:
Server(asio::io_context& io_context, unsigned short port)
Server(asio::io_context &io_context, unsigned short port)
: acceptor_(io_context, tcp::endpoint(tcp::v4(), port)),
context_(asio::ssl::context::tls_server)
{
@ -208,10 +186,8 @@ private:
void do_accept()
{
acceptor_.async_accept(
[this](const std::error_code& error, tcp::socket socket)
{
if (!error)
{
[this](const std::error_code & error, tcp::socket socket) {
if (!error) {
std::make_shared<Session>(std::move(socket), context_)->start();
}
@ -289,7 +265,7 @@ extern "C" void app_main(void)
work_threads.emplace_back(ssl_client_thread);
#endif // CONFIG_EXAMPLE_CLIENT
for (auto & t : work_threads) {
for (auto &t : work_threads) {
t.join();
}