port_mapping_service.h

Go to the documentation of this file.

#pragma once
 
#include "util/rats_export.h"
#include "node/peer_network.h"
#include "nat/port_mapping.h"   // PortMappingConfig, PortMapResult, PortMapCallback
 
#include <cstdint>
#include <memory>
#include <mutex>
#include <optional>
#include <string>
#include <utility>
 
namespace librats {
 
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() const;
 
private:
    void handle_result(const PortMapResult& result);
 
    PortMappingConfig config_;
    PeerNetwork*      network_ = 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;
    bool                          double_nat_warned_ = false;
};
 
} // namespace librats