acceptor.h

Go to the documentation of this file.

 
#ifndef QB_IO_ASYNC_TCP_ACCEPTOR_H
#define QB_IO_ASYNC_TCP_ACCEPTOR_H
 
#include <filesystem>
#ifdef QB_IO_WITH_SSL
#include "../../tcp/ssl/listener.h"
#endif
#include "../../protocol/accept.h"
#include "../io.h"
 
namespace qb::io::async::tcp {

template <typename _Derived, typename _Prot>
class acceptor
    : public input<acceptor<_Derived, _Prot>>
    , public _Prot {
    friend class input<acceptor<_Derived, _Prot>>;
    using base_t = input<acceptor<_Derived, _Prot>>;
    using Protocol =
        protocol::accept<acceptor,
                         typename _Prot::socket_type>; 
 
public:
    void
    on(event::disconnected &&e) {
        if constexpr (has_method_on<_Derived, void, event::disconnected>::value)
            static_cast<_Derived &>(*this).on(std::forward<event::disconnected>(e));
        else
            throw std::runtime_error("Acceptor has been disconnected");
    }
 
public:
    using accepted_socket_type = typename _Prot::socket_type;
 
public:
    acceptor() noexcept
        : base_t(new Protocol(*this)) {}

    ~acceptor() = default;

    void
    on(typename Protocol::message &&new_socket) {
        static_cast<_Derived &>(*this).on(
            std::forward<typename Protocol::message>(new_socket));
    }

    bool listen(qb::io::uri uri,
                std::filesystem::path cert_file = {},
                std::filesystem::path key_file = {},
                std::vector<std::string> alpn_protocols = {}) {
#ifdef QB_IO_WITH_SSL
        using tpt = std::decay_t<decltype(this->transport())>;
        if constexpr (tpt::is_secure()) {
            this->transport().init(qb::io::ssl::create_server_context(TLS_server_method(), cert_file, key_file));
            if (!this->transport().ssl_handle()) {
                LOG_CRIT("Failed to initialize SSL/TLS server context.");
                return false;
            }
            this->transport().set_supported_alpn_protocols(std::move(alpn_protocols));
        }
#endif
        return !this->transport().listen(std::move(uri));
    }
};
 
} // namespace qb::io::async::tcp
 
#endif // QB_IO_ASYNC_TCP_ACCEPTOR_H