UnknownObject/WebCC
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

extract a new class SslClient

Browse Source
This commit is contained in:
Chunting Gu
2021-06-23 10:03:11 +08:00
parent 326a3ea215
commit d294cda3c1
13 changed files with 410 additions and 281 deletions
Download Patch File Download Diff File Expand all files Collapse all files
webcc/socket.h
+1 -79
View File
@@ -1,56 +1,10 @@
#ifndef WEBCC_SOCKET_H_
#define WEBCC_SOCKET_H_
#include <vector>
#include "boost/asio/ip/tcp.hpp"
#include "webcc/config.h"
#include "webcc/request.h"
#if WEBCC_ENABLE_SSL
#include "boost/asio/ssl.hpp"
#endif // WEBCC_ENABLE_SSL
#include "webcc/socket_base.h"
namespace webcc {
// -----------------------------------------------------------------------------
class SocketBase {
public:
using Endpoints = boost::asio::ip::tcp::resolver::results_type;
using ConnectHandler = std::function<void(boost::system::error_code,
boost::asio::ip::tcp::endpoint)>;
using WriteHandler =
std::function<void(boost::system::error_code, std::size_t)>;
using ReadHandler =
std::function<void(boost::system::error_code, std::size_t)>;
SocketBase() = default;
SocketBase(const SocketBase&) = delete;
SocketBase& operator=(const SocketBase&) = delete;
virtual ~SocketBase() = default;
virtual void AsyncConnect(const std::string& host, const Endpoints& endpoints,
ConnectHandler&& handler) = 0;
virtual void AsyncWrite(const Payload& payload, WriteHandler&& handler) = 0;
virtual void AsyncReadSome(ReadHandler&& handler,
std::vector<char>* buffer) = 0;
virtual bool Shutdown() = 0;
virtual bool Close() = 0;
};
// -----------------------------------------------------------------------------
class Socket : public SocketBase {
public:
explicit Socket(boost::asio::io_context& io_context);
@@ -70,38 +24,6 @@ private:
boost::asio::ip::tcp::socket socket_;
};
// -----------------------------------------------------------------------------
#if WEBCC_ENABLE_SSL
class SslSocket : public SocketBase {
public:
SslSocket(boost::asio::io_context& io_context,
boost::asio::ssl::context& ssl_context);
void AsyncConnect(const std::string& host, const Endpoints& endpoints,
ConnectHandler&& handler) override;
void AsyncWrite(const Payload& payload, WriteHandler&& handler) override;
void AsyncReadSome(ReadHandler&& handler, std::vector<char>* buffer) override;
bool Shutdown() override;
bool Close() override;
private:
void OnConnect(boost::system::error_code ec,
boost::asio::ip::tcp::endpoint endpoint);
ConnectHandler connect_handler_;
boost::asio::ip::tcp::endpoint endpoint_;
boost::asio::ssl::stream<boost::asio::ip::tcp::socket> ssl_stream_;
};
#endif // WEBCC_ENABLE_SSL
} // namespace webcc
#endif // WEBCC_SOCKET_H_