quic_client.cpp

Go to the documentation of this file.

// BSD 3-Clause License
// Copyright (c) 2024, 🍀☀🌕🌥 🌊
// See the LICENSE file in the project root for full license information.
 
#define NETWORK_USE_EXPERIMENTAL
#include "internal/experimental/quic_client.h"
#include "internal/integration/logger_integration.h"
#include "kcenon/network/detail/tracing/span.h"
#include "kcenon/network/detail/tracing/trace_context.h"
#include "kcenon/network/detail/tracing/tracing_config.h"
#include "internal/quic_socket.h"
 
namespace kcenon::network::core
{
 
messaging_quic_client::messaging_quic_client(std::string_view client_id)
    : client_id_(client_id)
{
}
 
messaging_quic_client::~messaging_quic_client() noexcept
{
    if (lifecycle_.is_running())
    {
        (void)stop_client(); // Ignore result in destructor
    }
}
 
auto messaging_quic_client::client_id() const -> const std::string&
{
    return client_id_;
}
 
auto messaging_quic_client::is_running() const -> bool
{
    return lifecycle_.is_running();
}
 
auto messaging_quic_client::wait_for_stop() -> void
{
    lifecycle_.wait_for_stop();
}
 
auto messaging_quic_client::start_client(std::string_view host,
                                         unsigned short port) -> VoidResult
{
    // Use default config
    return start_client(host, port, quic_client_config{});
}
 
auto messaging_quic_client::start_client(std::string_view host,
                                         unsigned short port,
                                         const quic_client_config& config) -> VoidResult
{
    // Create tracing span for client start operation
    auto span = tracing::is_tracing_enabled()
        ? std::make_optional(tracing::trace_context::create_span("quic.client.start"))
        : std::nullopt;
    if (span)
    {
        span->set_attribute("net.peer.name", host)
             .set_attribute("net.peer.port", static_cast<int64_t>(port))
             .set_attribute("net.transport", "quic")
             .set_attribute("client.id", client_id_);
    }
 
    if (lifecycle_.is_running())
    {
        if (span)
        {
            span->set_error("QUIC client is already running");
        }
        return error_void(
            error_codes::common_errors::already_exists,
            "QUIC client is already running",
            "messaging_quic_client::start_client");
    }
 
    if (host.empty())
    {
        if (span)
        {
            span->set_error("Host cannot be empty");
        }
        return error_void(
            error_codes::common_errors::invalid_argument,
            "Host cannot be empty",
            "messaging_quic_client::start_client");
    }
 
    config_ = config;
    lifecycle_.set_running();
    is_connected_.store(false);
 
    auto result = do_start_impl(host, port);
    if (result.is_err())
    {
        lifecycle_.mark_stopped();
        if (span)
        {
            span->set_error(result.error().message);
        }
    }
    else
    {
        if (span)
        {
            span->set_status(tracing::span_status::ok);
        }
    }
 
    return result;
}
 
auto messaging_quic_client::stop_client() -> VoidResult
{
    if (!lifecycle_.prepare_stop())
    {
        return ok(); // Already stopped or not running
    }
 
    is_connected_.store(false);
    auto result = do_stop_impl();
    lifecycle_.mark_stopped();
 
    return result;
}
 
auto messaging_quic_client::do_start_impl(std::string_view host,
                                          unsigned short port) -> VoidResult
{
    try
    {
        {
            std::lock_guard<std::mutex> lock(socket_mutex_);
            socket_.reset();
        }
 
        io_context_ = std::make_unique<asio::io_context>();
        work_guard_ = std::make_unique<
            asio::executor_work_guard<asio::io_context::executor_type>>(
            asio::make_work_guard(*io_context_));
 
        thread_pool_ = network_context::instance().get_thread_pool();
        if (!thread_pool_)
        {
            thread_pool_ = std::make_shared<integration::basic_thread_pool>(2);
        }
 
        io_context_future_ = thread_pool_->submit([this]() {
            try
            {
                NETWORK_LOG_INFO(
                    "[messaging_quic_client] Starting io_context on thread pool");
                io_context_->run();
                NETWORK_LOG_INFO("[messaging_quic_client] io_context stopped");
            }
            catch (const std::exception& e)
            {
                NETWORK_LOG_ERROR("[messaging_quic_client] Exception in io_context: " +
                                  std::string(e.what()));
            }
        });
 
        do_connect(host, port);
 
        NETWORK_LOG_INFO("[messaging_quic_client] started. ID=" + client_id_ +
                         " target=" + std::string(host) + ":" +
                         std::to_string(port));
 
        return ok();
    }
    catch (const std::exception& e)
    {
        return error_void(
            error_codes::common_errors::internal_error,
            "Failed to start client: " + std::string(e.what()),
            "messaging_quic_client::do_start_impl",
            "Client ID: " + client_id_ + ", Host: " + std::string(host));
    }
}
 
auto messaging_quic_client::do_stop_impl() -> VoidResult
{
    try
    {
        std::shared_ptr<internal::quic_socket> local_socket;
        {
            std::lock_guard<std::mutex> lock(socket_mutex_);
            local_socket = std::move(socket_);
        }
        if (local_socket)
        {
            local_socket->stop_receive();
            auto close_result = local_socket->close();
            if (close_result.is_err())
            {
                NETWORK_LOG_WARN("[messaging_quic_client] Close error: " +
                                 close_result.error().message);
            }
        }
 
        if (work_guard_)
        {
            work_guard_.reset();
        }
        if (io_context_)
        {
            io_context_->stop();
        }
        if (io_context_future_.valid())
        {
            io_context_future_.wait();
        }
        thread_pool_.reset();
        io_context_.reset();
 
        handshake_complete_.store(false);
 
        NETWORK_LOG_INFO("[messaging_quic_client] stopped.");
        return ok();
    }
    catch (const std::exception& e)
    {
        return error_void(
            error_codes::common_errors::internal_error,
            "Failed to stop client: " + std::string(e.what()),
            "messaging_quic_client::do_stop_impl",
            "Client ID: " + client_id_);
    }
}
 
auto messaging_quic_client::is_connected() const -> bool
{
    return is_connected_.load(std::memory_order_acquire);
}
 
auto messaging_quic_client::is_handshake_complete() const -> bool
{
    auto local_socket = get_socket();
    if (!local_socket)
    {
        return false;
    }
    return local_socket->is_handshake_complete();
}
 
auto messaging_quic_client::send_packet(std::vector<uint8_t>&& data) -> VoidResult
{
    // Create tracing span for send operation
    auto span = tracing::is_tracing_enabled()
        ? std::make_optional(tracing::trace_context::create_span("quic.client.send"))
        : std::nullopt;
    if (span)
    {
        span->set_attribute("net.transport", "quic")
             .set_attribute("message.size", static_cast<int64_t>(data.size()))
             .set_attribute("quic.stream_id", static_cast<int64_t>(default_stream_id_))
             .set_attribute("client.id", client_id_);
    }
 
    if (!is_running())
    {
        if (span)
        {
            span->set_error("Client is not running");
        }
        return error_void(
            error_codes::network_system::connection_closed,
            "Client is not running",
            "messaging_quic_client::send_packet",
            "Client ID: " + client_id_);
    }
 
    if (data.empty())
    {
        if (span)
        {
            span->set_error("Data cannot be empty");
        }
        return error_void(
            error_codes::common_errors::invalid_argument,
            "Data cannot be empty",
            "messaging_quic_client::send_packet",
            "Client ID: " + client_id_);
    }
 
    auto local_socket = get_socket();
    if (!is_connected() || !local_socket)
    {
        if (span)
        {
            span->set_error("Client is not connected");
        }
        return error_void(
            error_codes::network_system::connection_closed,
            "Client is not connected",
            "messaging_quic_client::send_packet",
            "Client ID: " + client_id_);
    }
 
    auto result = local_socket->send_stream_data(default_stream_id_, std::move(data));
    if (span)
    {
        if (result.is_err())
        {
            span->set_error(result.error().message);
        }
        else
        {
            span->set_status(tracing::span_status::ok);
        }
    }
    return result;
}
 
auto messaging_quic_client::send_packet(std::string_view data) -> VoidResult
{
    if (data.empty())
    {
        return error_void(
            error_codes::common_errors::invalid_argument,
            "Data cannot be empty",
            "messaging_quic_client::send_packet",
            "Client ID: " + client_id_);
    }
 
    std::vector<uint8_t> byte_data(data.begin(), data.end());
    return send_packet(std::move(byte_data));
}
 
auto messaging_quic_client::start(std::string_view host, uint16_t port) -> VoidResult
{
    return start_client(host, port);
}
 
auto messaging_quic_client::stop() -> VoidResult
{
    return stop_client();
}
 
auto messaging_quic_client::send(std::vector<uint8_t>&& data) -> VoidResult
{
    return send_packet(std::move(data));
}
 
auto messaging_quic_client::create_stream() -> Result<uint64_t>
{
    auto local_socket = get_socket();
    if (!local_socket)
    {
        return error<uint64_t>(
            error_codes::network_system::connection_closed,
            "Client is not connected",
            "messaging_quic_client::create_stream",
            "Client ID: " + client_id_);
    }
 
    return local_socket->create_stream(false);
}
 
auto messaging_quic_client::create_unidirectional_stream() -> Result<uint64_t>
{
    auto local_socket = get_socket();
    if (!local_socket)
    {
        return error<uint64_t>(
            error_codes::network_system::connection_closed,
            "Client is not connected",
            "messaging_quic_client::create_unidirectional_stream",
            "Client ID: " + client_id_);
    }
 
    return local_socket->create_stream(true);
}
 
auto messaging_quic_client::send_on_stream(uint64_t stream_id,
                                           std::vector<uint8_t>&& data,
                                           bool fin) -> VoidResult
{
    // Create tracing span for stream send operation
    auto span = tracing::is_tracing_enabled()
        ? std::make_optional(tracing::trace_context::create_span("quic.stream.send"))
        : std::nullopt;
    if (span)
    {
        span->set_attribute("net.transport", "quic")
             .set_attribute("message.size", static_cast<int64_t>(data.size()))
             .set_attribute("quic.stream_id", static_cast<int64_t>(stream_id))
             .set_attribute("quic.fin", fin)
             .set_attribute("client.id", client_id_);
    }
 
    if (data.empty())
    {
        if (span)
        {
            span->set_error("Data cannot be empty");
        }
        return error_void(
            error_codes::common_errors::invalid_argument,
            "Data cannot be empty",
            "messaging_quic_client::send_on_stream",
            "Client ID: " + client_id_);
    }
 
    auto local_socket = get_socket();
    if (!local_socket)
    {
        if (span)
        {
            span->set_error("Client is not connected");
        }
        return error_void(
            error_codes::network_system::connection_closed,
            "Client is not connected",
            "messaging_quic_client::send_on_stream",
            "Client ID: " + client_id_);
    }
 
    auto result = local_socket->send_stream_data(stream_id, std::move(data), fin);
    if (span)
    {
        if (result.is_err())
        {
            span->set_error(result.error().message);
        }
        else
        {
            span->set_status(tracing::span_status::ok);
        }
    }
    return result;
}
 
auto messaging_quic_client::close_stream(uint64_t stream_id) -> VoidResult
{
    auto local_socket = get_socket();
    if (!local_socket)
    {
        return error_void(
            error_codes::network_system::connection_closed,
            "Client is not connected",
            "messaging_quic_client::close_stream",
            "Client ID: " + client_id_);
    }
 
    return local_socket->close_stream(stream_id);
}
 
auto messaging_quic_client::set_alpn_protocols(
    const std::vector<std::string>& protocols) -> void
{
    config_.alpn_protocols = protocols;
}
 
auto messaging_quic_client::alpn_protocol() const -> std::optional<std::string>
{
    // TODO: Implement ALPN negotiation result retrieval
    return std::nullopt;
}
 
auto messaging_quic_client::is_early_data_accepted() const -> bool
{
    return early_data_accepted_.load();
}
 
auto messaging_quic_client::stats() const -> quic_connection_stats
{
    // TODO: Implement connection statistics retrieval
    return quic_connection_stats{};
}
 
auto messaging_quic_client::do_connect(std::string_view host,
                                       unsigned short port) -> void
{
    auto resolver = std::make_shared<asio::ip::udp::resolver>(*io_context_);
    auto self = shared_from_this();
 
    NETWORK_LOG_INFO("[messaging_quic_client] Starting async resolve for " +
                     std::string(host) + ":" + std::to_string(port));
 
    resolver->async_resolve(
        std::string(host),
        std::to_string(port),
        [this, self, resolver, host_str = std::string(host)](
            std::error_code ec,
            asio::ip::udp::resolver::results_type results) {
            NETWORK_LOG_INFO("[messaging_quic_client] Resolve callback invoked");
            if (ec)
            {
                NETWORK_LOG_ERROR("[messaging_quic_client] Resolve error: " +
                                  ec.message());
                on_error(ec);
                return;
            }
 
            if (results.empty())
            {
                NETWORK_LOG_ERROR("[messaging_quic_client] No endpoints resolved");
                on_error(asio::error::host_not_found);
                return;
            }
 
            NETWORK_LOG_INFO("[messaging_quic_client] Resolve successful, creating socket");
 
            try
            {
                auto endpoint = *results.begin();
 
                // Create UDP socket
                asio::ip::udp::socket udp_socket(*io_context_, asio::ip::udp::v4());
 
                // Create QUIC socket as client
                auto quic_sock = std::make_shared<internal::quic_socket>(
                    std::move(udp_socket),
                    internal::quic_role::client);
 
                // Set callbacks
                quic_sock->set_stream_data_callback(
                    [this, self](uint64_t stream_id,
                                 std::span<const uint8_t> data,
                                 bool fin) {
                        on_stream_data(stream_id, data, fin);
                    });
 
                quic_sock->set_connected_callback([this, self]() {
                    on_connect();
                });
 
                quic_sock->set_error_callback([this, self](std::error_code err) {
                    on_error(err);
                });
 
                quic_sock->set_close_callback(
                    [this, self](uint64_t error_code, const std::string& reason) {
                        on_close(error_code, reason);
                    });
 
                // Store socket
                {
                    std::lock_guard<std::mutex> lock(socket_mutex_);
                    socket_ = quic_sock;
                }
 
                // Connect to server
                auto connect_result = quic_sock->connect(endpoint, host_str);
                if (connect_result.is_err())
                {
                    NETWORK_LOG_ERROR("[messaging_quic_client] Connect failed: " +
                                      connect_result.error().message);
                    on_error(std::make_error_code(std::errc::connection_refused));
                    return;
                }
 
                // Start receive loop
                quic_sock->start_receive();
 
                NETWORK_LOG_INFO("[messaging_quic_client] Connection initiated");
            }
            catch (const std::exception& e)
            {
                NETWORK_LOG_ERROR("[messaging_quic_client] Exception during connect: " +
                                  std::string(e.what()));
                on_error(std::make_error_code(std::errc::connection_refused));
            }
        });
}
 
auto messaging_quic_client::on_connect() -> void
{
    NETWORK_LOG_INFO("[messaging_quic_client] Connected successfully.");
    is_connected_.store(true, std::memory_order_release);
    handshake_complete_.store(true);
 
    // Create default stream for send_packet()
    auto local_socket = get_socket();
    if (local_socket)
    {
        auto stream_result = local_socket->create_stream(false);
        if (stream_result.is_ok())
        {
            default_stream_id_ = stream_result.value();
            NETWORK_LOG_DEBUG("[messaging_quic_client] Default stream created: " +
                              std::to_string(default_stream_id_));
        }
    }
 
    // Invoke connected callback
    invoke_connected_callback();
}
 
auto messaging_quic_client::on_stream_data(uint64_t stream_id,
                                           std::span<const uint8_t> data,
                                           bool fin) -> void
{
    if (!is_connected())
    {
        return;
    }
 
    NETWORK_LOG_DEBUG("[messaging_quic_client] Received " +
                      std::to_string(data.size()) +
                      " bytes on stream " + std::to_string(stream_id));
 
    std::vector<uint8_t> data_copy(data.begin(), data.end());
 
    // Invoke stream receive callback
    invoke_stream_receive_callback(stream_id, data_copy, fin);
 
    // Also invoke default receive callback for default stream
    if (stream_id == default_stream_id_)
    {
        invoke_receive_callback(data_copy);
    }
}
 
auto messaging_quic_client::on_error(std::error_code ec) -> void
{
    NETWORK_LOG_ERROR("[messaging_quic_client] Error: " + ec.message());
 
    // Invoke error callback
    invoke_error_callback(ec);
 
    if (is_connected())
    {
        // Invoke disconnected callback
        invoke_disconnected_callback();
    }
 
    is_connected_.store(false, std::memory_order_release);
}
 
auto messaging_quic_client::on_close(uint64_t error_code,
                                     const std::string& reason) -> void
{
    NETWORK_LOG_INFO("[messaging_quic_client] Connection closed. Error code: " +
                     std::to_string(error_code) + ", reason: " + reason);
 
    if (is_connected())
    {
        // Invoke disconnected callback
        invoke_disconnected_callback();
    }
 
    is_connected_.store(false, std::memory_order_release);
}
 
auto messaging_quic_client::get_socket() const
    -> std::shared_ptr<internal::quic_socket>
{
    std::lock_guard<std::mutex> lock(socket_mutex_);
    return socket_;
}
 
// =============================================================================
// Callback invocation helpers
// =============================================================================
 
auto messaging_quic_client::invoke_receive_callback(const std::vector<uint8_t>& data) -> void
{
    callbacks_.invoke<to_index(callback_index::receive)>(data);
}
 
auto messaging_quic_client::invoke_stream_receive_callback(uint64_t stream_id,
                                                           const std::vector<uint8_t>& data,
                                                           bool fin) -> void
{
    callbacks_.invoke<to_index(callback_index::stream_receive)>(stream_id, data, fin);
}
 
auto messaging_quic_client::invoke_connected_callback() -> void
{
    callbacks_.invoke<to_index(callback_index::connected)>();
}
 
auto messaging_quic_client::invoke_disconnected_callback() -> void
{
    callbacks_.invoke<to_index(callback_index::disconnected)>();
}
 
auto messaging_quic_client::invoke_error_callback(std::error_code ec) -> void
{
    callbacks_.invoke<to_index(callback_index::error)>(ec);
}
 
// =============================================================================
// Legacy callback setters
// =============================================================================
 
auto messaging_quic_client::set_stream_receive_callback(stream_receive_callback_t callback) -> void
{
    callbacks_.set<to_index(callback_index::stream_receive)>(std::move(callback));
}
 
// =============================================================================
// i_quic_client interface callback implementations
// =============================================================================
 
auto messaging_quic_client::set_receive_callback(
    interfaces::i_quic_client::receive_callback_t callback) -> void
{
    callbacks_.set<to_index(callback_index::receive)>(std::move(callback));
}
 
auto messaging_quic_client::set_stream_callback(
    interfaces::i_quic_client::stream_callback_t callback) -> void
{
    callbacks_.set<to_index(callback_index::stream_receive)>(std::move(callback));
}
 
auto messaging_quic_client::set_connected_callback(
    interfaces::i_quic_client::connected_callback_t callback) -> void
{
    callbacks_.set<to_index(callback_index::connected)>(std::move(callback));
}
 
auto messaging_quic_client::set_disconnected_callback(
    interfaces::i_quic_client::disconnected_callback_t callback) -> void
{
    callbacks_.set<to_index(callback_index::disconnected)>(std::move(callback));
}
 
auto messaging_quic_client::set_error_callback(
    interfaces::i_quic_client::error_callback_t callback) -> void
{
    callbacks_.set<to_index(callback_index::error)>(std::move(callback));
}
 
auto messaging_quic_client::set_session_ticket_callback(
    interfaces::i_quic_client::session_ticket_callback_t callback) -> void
{
    session_ticket_cb_ = std::move(callback);
}
 
auto messaging_quic_client::set_early_data_callback(
    interfaces::i_quic_client::early_data_callback_t callback) -> void
{
    early_data_cb_ = std::move(callback);
}
 
auto messaging_quic_client::set_early_data_accepted_callback(
    interfaces::i_quic_client::early_data_accepted_callback_t callback) -> void
{
    early_data_accepted_cb_ = std::move(callback);
}
 
} // namespace kcenon::network::core