UdpEndpoint.hpp

Go to the documentation of this file.

 
#pragma once
 
#include "PacketHeader.hpp"
 
#include <SDL3/SDL_stdinc.h>
 
#include <SDL3_net/SDL_net.h>
#include <cstdint>
#include <vector>
 
namespace net
{

struct UdpEndpointAddr
{
    NET_Address* addr = nullptr; 
    Uint16 port = 0;             
 
    UdpEndpointAddr() = default;
    UdpEndpointAddr(NET_Address* address, Uint16 p) : addr(address ? NET_RefAddress(address) : nullptr), port(p) {}
    UdpEndpointAddr(const UdpEndpointAddr& other)
        : addr(other.addr ? NET_RefAddress(other.addr) : nullptr), port(other.port)
    {}
    UdpEndpointAddr& operator=(const UdpEndpointAddr& other)
    {
        if (this == &other)
            return *this;
        release();
        addr = other.addr ? NET_RefAddress(other.addr) : nullptr;
        port = other.port;
        return *this;
    }
    UdpEndpointAddr(UdpEndpointAddr&& other) noexcept : addr(other.addr), port(other.port)
    {
        other.addr = nullptr;
        other.port = 0;
    }
    UdpEndpointAddr& operator=(UdpEndpointAddr&& other) noexcept
    {
        if (this == &other)
            return *this;
        release();
        addr = other.addr;
        port = other.port;
        other.addr = nullptr;
        other.port = 0;
        return *this;
    }
    ~UdpEndpointAddr() { release(); }

    void release() noexcept
    {
        if (addr) {
            NET_UnrefAddress(addr);
            addr = nullptr;
        }
    }
};

struct UdpReceivedMessage
{
    UdpEndpointAddr from;         
    PacketHeader header{};        
    std::vector<uint8_t> payload; 
};

class UdpEndpoint
{
public:
    UdpEndpoint() = default;
    UdpEndpoint(const UdpEndpoint&) = delete;
    UdpEndpoint& operator=(const UdpEndpoint&) = delete;
    UdpEndpoint(UdpEndpoint&&) = delete;
    UdpEndpoint& operator=(UdpEndpoint&&) = delete;
    ~UdpEndpoint() { close(); }

    bool open(const char* bindAddr, Uint16 port);

    void close() noexcept;

    bool send(const UdpEndpointAddr& dest, PacketHeader hdr, const void* payload, int payloadLen);

    bool sendDatagramBytes(const UdpEndpointAddr& dest, const void* bytes, int len);

    bool
    sendFragmented(const UdpEndpointAddr& dest, PacketHeader hdr, const void* data, int dataLen, int redundancy = 1);

    bool tryReceive(UdpReceivedMessage& out);

    [[nodiscard]] bool isOpen() const noexcept { return socket_ != nullptr; }

    [[nodiscard]] uint16_t localPort() const noexcept;
 
private:
    NET_DatagramSocket* socket_ = nullptr;
};
 
} // namespace net