udp_socket.h

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.
 
#pragma once
 
#include <asio.hpp>
#include <functional>
#include <memory>
#include <vector>
#include <array>
#include <system_error>
#include <mutex>
#include <atomic>
 
#include "internal/utils/common_defs.h"
 
namespace kcenon::network::internal
{
    class udp_socket : public std::enable_shared_from_this<udp_socket>
    {
    public:
        udp_socket(asio::ip::udp::socket socket);
 
        ~udp_socket() = default;
 
        auto set_receive_callback(
            std::function<void(const std::vector<uint8_t>&,
                               const asio::ip::udp::endpoint&)> callback) -> void;
 
        auto set_error_callback(std::function<void(std::error_code)> callback) -> void;
 
        auto start_receive() -> void;
 
        auto stop_receive() -> void;
 
        auto async_send_to(
            std::vector<uint8_t>&& data,
            const asio::ip::udp::endpoint& endpoint,
            std::function<void(std::error_code, std::size_t)> handler) -> void;
 
        auto socket() -> asio::ip::udp::socket& { return socket_; }
 
        auto close() -> void;
 
        [[nodiscard]] auto is_closed() const -> bool;
 
    private:
        auto do_receive() -> void;
 
    private:
        asio::ip::udp::socket socket_; 
        std::array<uint8_t, 65536> read_buffer_; 
        asio::ip::udp::endpoint sender_endpoint_; 
        std::mutex callback_mutex_; 
        std::function<void(const std::vector<uint8_t>&, const asio::ip::udp::endpoint&)>
            receive_callback_; 
        std::function<void(std::error_code)>
            error_callback_; 
        std::atomic<bool> is_receiving_{false}; 
        std::atomic<bool> is_closed_{false};    
    };
} // namespace kcenon::network::internal