HTTPClientSSLContext.h

#pragma once
 
#include <xrpl/basics/Log.h>
#include <xrpl/basics/contract.h>
#include <xrpl/beast/utility/Journal.h>
#include <xrpl/net/RegisterSSLCerts.h>
 
#include <boost/asio.hpp>
#include <boost/asio/ip/tcp.hpp>
#include <boost/asio/ssl.hpp>
#include <boost/format.hpp>
 
namespace xrpl {
 
class HTTPClientSSLContext
{
public:
    explicit HTTPClientSSLContext(
        std::string const& sslVerifyDir,
        std::string const& sslVerifyFile,
        bool sslVerify,
        beast::Journal j,
        boost::asio::ssl::context_base::method method = boost::asio::ssl::context::sslv23)
        : ssl_context_{method}, j_(j), verify_{sslVerify}
    {
        boost::system::error_code ec;
 
        if (sslVerifyFile.empty())
        {
            registerSSLCerts(ssl_context_, ec, j_);
 
            if (ec && sslVerifyDir.empty())
                Throw<std::runtime_error>(boost::str(
                    boost::format("Failed to set_default_verify_paths: %s") % ec.message()));
        }
        else
        {
            ssl_context_.load_verify_file(sslVerifyFile);
        }
 
        if (!sslVerifyDir.empty())
        {
            ssl_context_.add_verify_path(sslVerifyDir, ec);
 
            if (ec)
                Throw<std::runtime_error>(
                    boost::str(boost::format("Failed to add verify path: %s") % ec.message()));
        }
    }
 
    boost::asio::ssl::context&
    context()
    {
        return ssl_context_;
    }
 
    bool
    sslVerify() const
    {
        return verify_;
    }
 
    template <
        class T,
        class = std::enable_if_t<
            std::is_same<T, boost::asio::ssl::stream<boost::asio::ip::tcp::socket>>::value ||
            std::is_same<T, boost::asio::ssl::stream<boost::asio::ip::tcp::socket&>>::value>>
    boost::system::error_code
    preConnectVerify(T& strm, std::string const& host)
    {
        boost::system::error_code ec;
        if (!SSL_set_tlsext_host_name(strm.native_handle(), host.c_str()))
        {
            ec.assign(static_cast<int>(::ERR_get_error()), boost::asio::error::get_ssl_category());
        }
        else if (!sslVerify())
        {
            strm.set_verify_mode(boost::asio::ssl::verify_none, ec);
        }
        return ec;
    }
 
    template <
        class T,
        class = std::enable_if_t<
            std::is_same<T, boost::asio::ssl::stream<boost::asio::ip::tcp::socket>>::value ||
            std::is_same<T, boost::asio::ssl::stream<boost::asio::ip::tcp::socket&>>::value>>
    boost::system::error_code
    postConnectVerify(T& strm, std::string const& host)
    {
        boost::system::error_code ec;
 
        if (sslVerify())
        {
            strm.set_verify_mode(boost::asio::ssl::verify_peer, ec);
            if (!ec)
            {
                strm.set_verify_callback(
                    std::bind(
                        &rfc6125_verify, host, std::placeholders::_1, std::placeholders::_2, j_),
                    ec);
            }
        }
 
        return ec;
    }
 
    static bool
    rfc6125_verify(
        std::string const& domain,
        bool preverified,
        boost::asio::ssl::verify_context& ctx,
        beast::Journal j)
    {
        if (boost::asio::ssl::host_name_verification(domain)(preverified, ctx))
            return true;
 
        JLOG(j.warn()) << "Outbound SSL connection to " << domain
                       << " fails certificate verification";
        return false;
    }
 
private:
    boost::asio::ssl::context ssl_context_;
    beast::Journal const j_;
    bool const verify_;
};
 
}  // namespace xrpl