You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
99 lines
2.6 KiB
C++
99 lines
2.6 KiB
C++
#ifndef WEBCC_HTTP_SOCKET_H_
|
|
#define WEBCC_HTTP_SOCKET_H_
|
|
|
|
#include <vector>
|
|
|
|
#include "boost/asio/ip/tcp.hpp"
|
|
|
|
#include "webcc/config.h"
|
|
#include "webcc/http_request.h"
|
|
|
|
#if WEBCC_ENABLE_SSL
|
|
#include "boost/asio/ssl.hpp"
|
|
#endif // WEBCC_ENABLE_SSL
|
|
|
|
namespace webcc {
|
|
|
|
// -----------------------------------------------------------------------------
|
|
|
|
class HttpSocketBase {
|
|
public:
|
|
virtual ~HttpSocketBase() = default;
|
|
|
|
typedef boost::asio::ip::tcp::resolver::results_type Endpoints;
|
|
|
|
typedef std::function<void(boost::system::error_code, std::size_t)>
|
|
ReadHandler;
|
|
|
|
// TODO: Remove |host|
|
|
virtual void Connect(const std::string& host,
|
|
const Endpoints& endpoints,
|
|
boost::system::error_code* ec) = 0;
|
|
|
|
virtual void Write(const HttpRequest& request,
|
|
boost::system::error_code* ec) = 0;
|
|
|
|
virtual void AsyncReadSome(ReadHandler&& handler,
|
|
std::vector<char>* buffer) = 0;
|
|
|
|
virtual void Close(boost::system::error_code* ec) = 0;
|
|
};
|
|
|
|
// -----------------------------------------------------------------------------
|
|
|
|
class HttpSocket : public HttpSocketBase {
|
|
public:
|
|
explicit HttpSocket(boost::asio::io_context& io_context);
|
|
|
|
void Connect(const std::string& host,
|
|
const Endpoints& endpoints,
|
|
boost::system::error_code* ec) final;
|
|
|
|
void Write(const HttpRequest& request,
|
|
boost::system::error_code* ec) final;
|
|
|
|
void AsyncReadSome(ReadHandler&& handler, std::vector<char>* buffer) final;
|
|
|
|
void Close(boost::system::error_code* ec) final;
|
|
|
|
private:
|
|
boost::asio::ip::tcp::socket socket_;
|
|
};
|
|
|
|
// -----------------------------------------------------------------------------
|
|
|
|
#if WEBCC_ENABLE_SSL
|
|
|
|
class HttpSslSocket : public HttpSocketBase {
|
|
public:
|
|
explicit HttpSslSocket(boost::asio::io_context& io_context,
|
|
bool ssl_verify = true);
|
|
|
|
void Connect(const std::string& host,
|
|
const Endpoints& endpoints,
|
|
boost::system::error_code* ec) final;
|
|
|
|
void Write(const HttpRequest& request,
|
|
boost::system::error_code* ec) final;
|
|
|
|
void AsyncReadSome(ReadHandler&& handler, std::vector<char>* buffer) final;
|
|
|
|
void Close(boost::system::error_code* ec) final;
|
|
|
|
private:
|
|
void Handshake(const std::string& host, boost::system::error_code* ec);
|
|
|
|
boost::asio::ssl::context ssl_context_;
|
|
|
|
boost::asio::ssl::stream<boost::asio::ip::tcp::socket> ssl_socket_;
|
|
|
|
// Verify the certificate of the peer (remote server) or not.
|
|
bool ssl_verify_;
|
|
};
|
|
|
|
#endif // WEBCC_ENABLE_SSL
|
|
|
|
} // namespace webcc
|
|
|
|
#endif // WEBCC_HTTP_SOCKET_H_
|