GlobalDirectoryServer.hpp

Go to the documentation of this file.

 
#pragma once
 
#include "network/discovery/GlobalDiscoveryProtocol.hpp"
#include "network/transport/UdpEndpoint.hpp"
 
#include <SDL3/SDL_stdinc.h>
 
#include <atomic>
#include <cstdint>
#include <string>
#include <unordered_map>
 
class GlobalDirectoryServer
{
public:
    bool init(const char* bindHost, Uint16 tcpPort, Uint16 udpPort);
    void run();
    void stop();
 
private:
    struct ServerRecord
    {
        net::discovery::ServerInfo info;
        Uint64 lastSeenMs = 0;
        net::UdpEndpointAddr endpoint;
    };
 
    struct ClientEndpoint
    {
        net::UdpEndpointAddr endpoint;
        net::RelayToken relayToken;
        Uint64 lastSeenMs = 0;
        bool relayAuthorized = false;
    };
 
    void pollUdp();
    void pruneExpired();
    void handleDirectoryControl(const net::UdpReceivedMessage& msg);
    void handleRegistration(const net::UdpReceivedMessage& msg,
                            net::discovery::DirectoryMessage kind,
                            const std::uint8_t* data,
                            std::size_t len);
    void handleListRequest(const net::UdpReceivedMessage& msg);
    void handlePunchRequest(const net::UdpReceivedMessage& msg, const std::uint8_t* data, std::size_t len);
    void handleRelayPayload(const net::UdpReceivedMessage& msg);
    void sendDirectory(const net::UdpEndpointAddr& dest, const std::vector<std::uint8_t>& payload);
    void sendRelay(const net::UdpEndpointAddr& dest, const std::uint8_t* payload, std::size_t len);
    net::RelayToken makeRelayToken(std::uint32_t serverId, std::uint32_t clientNonce, Uint64 nowMs) const;
    bool validateRelayToken(const net::RelayToken& token,
                            std::uint32_t serverId,
                            std::uint32_t clientNonce,
                            Uint64 nowMs) const;
 
    net::UdpEndpoint udpEndpoint_;
    std::vector<std::uint8_t> relaySecret_;
    std::unordered_map<std::uint32_t, ServerRecord> servers_;
    std::unordered_map<std::uint64_t, ClientEndpoint> clientsByRelaySession_;
    std::uint32_t nextServerId_ = 1;
    std::atomic<bool> shouldStop_{false};
};