port_mapping_service.h

Go to the documentation of this file.

#pragma once

 
#include "librats/util/rats_export.h"
#include "librats/node/peer_network.h"
#include "librats/nat/port_mapping.h"   // PortMappingConfig, PortMapResult, PortMapCallback
 
#include <cstdint>
#include <memory>
#include <mutex>
#include <optional>
#include <string>
#include <utility>
 
namespace librats {
 
class EventBus;
class UpnpClient;
class NatPmpClient;

class RATS_API PortMappingService final : public Subsystem {
public:
    explicit PortMappingService(PortMappingConfig config = {});
    ~PortMappingService() override;
 
    void attach(NodeContext& ctx) override;
    void start() override;
    void stop() override;

    void on_result(PortMapCallback cb) { user_callback_ = std::move(cb); }

    std::optional<std::pair<std::string, uint16_t>> mapped_public_address(
        PortMapProtocol protocol = PortMapProtocol::TCP) const;
 
private:
    void handle_result(const PortMapResult& result);
    uint16_t& external_port_slot(PortMapProtocol protocol);
 
    PortMappingConfig config_;
    PeerNetwork*      network_ = nullptr;
    EventBus*         events_ = nullptr;  
    PortMapCallback   user_callback_;
 
    mutable std::mutex            mutex_;  
    std::unique_ptr<UpnpClient>   upnp_;
    std::unique_ptr<NatPmpClient> natpmp_;
    std::string                   mapped_external_ip_;
    uint16_t                      mapped_external_tcp_port_ = 0;
    uint16_t                      mapped_external_udp_port_ = 0;
    bool                          double_nat_warned_ = false;
};
 
} // namespace librats