clio/src/webserver/SslHttpSession.h

#ifndef RIPPLE_REPORTING_HTTPS_SESSION_H
#define RIPPLE_REPORTING_HTTPS_SESSION_H

#include <webserver/HttpBase.h>

namespace http = boost::beast::http;
namespace net = boost::asio;
namespace ssl = boost::asio::ssl;
using tcp = boost::asio::ip::tcp;

// Handles an HTTPS server connection
class SslHttpSession : public HttpBase<SslHttpSession>,
                       public std::enable_shared_from_this<SslHttpSession>
{
    boost::beast::ssl_stream<boost::beast::tcp_stream> stream_;

public:
    // Take ownership of the socket
    explicit SslHttpSession(
        tcp::socket&& socket,
        ssl::context& ctx,
        std::shared_ptr<BackendInterface const> backend,
        std::shared_ptr<SubscriptionManager> subscriptions,
        std::shared_ptr<ETLLoadBalancer> balancer,
        DOSGuard& dosGuard,
        boost::beast::flat_buffer buffer)
        : HttpBase<SslHttpSession>(
              backend,
              subscriptions,
              balancer,
              dosGuard,
              std::move(buffer))
        , stream_(std::move(socket), ctx)
    {
    }

    boost::beast::ssl_stream<boost::beast::tcp_stream>&
    stream()
    {
        return stream_;
    }
    boost::beast::ssl_stream<boost::beast::tcp_stream>
    release_stream()
    {
        return std::move(stream_);
    }

    std::string
    ip()
    {
        return stream_.next_layer()
            .socket()
            .remote_endpoint()
            .address()
            .to_string();
    }

    // Start the asynchronous operation
    void
    run()
    {
        auto self = shared_from_this();
        // We need to be executing within a strand to perform async operations
        // on the I/O objects in this session.
        net::dispatch(stream_.get_executor(), [self]() {
            // Set the timeout.
            boost::beast::get_lowest_layer(self->stream())
                .expires_after(std::chrono::seconds(30));

            // Perform the SSL handshake
            // Note, this is the buffered version of the handshake.
            self->stream_.async_handshake(
                ssl::stream_base::server,
                self->buffer_.data(),
                boost::beast::bind_front_handler(
                    &SslHttpSession::on_handshake, self));
        });
    }

    void
    on_handshake(boost::beast::error_code ec, std::size_t bytes_used)
    {
        if (ec)
            return httpFail(ec, "handshake");

        buffer_.consume(bytes_used);

        do_read();
    }

    void
    do_close()
    {
        // Set the timeout.
        boost::beast::get_lowest_layer(stream_).expires_after(
            std::chrono::seconds(30));

        // Perform the SSL shutdown
        stream_.async_shutdown(boost::beast::bind_front_handler(
            &SslHttpSession::on_shutdown, shared_from_this()));
    }

    void
    on_shutdown(boost::beast::error_code ec)
    {
        if (ec)
            return httpFail(ec, "shutdown");

        // At this point the connection is closed gracefully
    }
};

#endif  // RIPPLE_REPORTING_HTTPS_SESSION_H