Skip to content
Snippets Groups Projects
Select Git revision
  • 9c66dbd373f70a27aca1ab67aaf2a97cffd7aa07
  • master default protected
  • release/202005
  • release/202001
  • release/201912
  • release/201911
  • release/releaseWindowsTestOne
  • release/windowsReleaseTest
  • release/releaseTest
  • release/releaseWindowsTest
  • release/201910
  • release/qt/201910
  • release/windows-test/201910
  • release/201908
  • release/201906
  • release/201905
  • release/201904
  • release/201903
  • release/201902
  • release/201901
  • release/201812
  • 4.0.0
  • 2.2.0
  • 2.1.0
  • 2.0.1
  • 2.0.0
  • 1.4.1
  • 1.4.0
  • 1.3.0
  • 1.2.0
  • 1.1.0
31 results

map_utils.h

Blame
  • Code owners
    Assign users and groups as approvers for specific file changes. Learn more.
    upnp_context.cpp 40.35 KiB
    /*
     *  Copyright (C) 2004-2023 Savoir-faire Linux Inc.
     *
     *  This program is free software: you can redistribute it and/or modify
     *  it under the terms of the GNU General Public License as published by
     *  the Free Software Foundation, either version 3 of the License, or
     *  (at your option) any later version.
     *
     *  This program is distributed in the hope that it will be useful,
     *  but WITHOUT ANY WARRANTY; without even the implied warranty of
     *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
     *  GNU General Public License for more details.
     *
     *  You should have received a copy of the GNU General Public License
     *  along with this program. If not, see <https://www.gnu.org/licenses/>.
     */
    #include "upnp/upnp_context.h"
    #include "protocol/upnp_protocol.h"
    
    #if HAVE_LIBNATPMP
    #include "protocol/natpmp/nat_pmp.h"
    #endif
    #if HAVE_LIBUPNP
    #include "protocol/pupnp/pupnp.h"
    #endif
    
    #include <asio/steady_timer.hpp>
    #if __has_include(<fmt/std.h>)
    #include <fmt/std.h>
    #else
    #include <fmt/ostream.h>
    #endif
    
    namespace dhtnet {
    namespace upnp {
    
    constexpr static auto MAP_UPDATE_INTERVAL = std::chrono::seconds(30);
    constexpr static int MAX_REQUEST_RETRIES = 20;
    constexpr static int MAX_REQUEST_REMOVE_COUNT = 5;
    
    constexpr static uint16_t UPNP_TCP_PORT_MIN {10000};
    constexpr static uint16_t UPNP_TCP_PORT_MAX {UPNP_TCP_PORT_MIN + 5000};
    constexpr static uint16_t UPNP_UDP_PORT_MIN {20000};
    constexpr static uint16_t UPNP_UDP_PORT_MAX {UPNP_UDP_PORT_MIN + 5000};
    
    UPnPContext::UPnPContext(const std::shared_ptr<asio::io_context>& ioContext, const std::shared_ptr<dht::log::Logger>& logger)
     : ctx(createIoContext(ioContext, logger))
     , mappingListUpdateTimer_(*ioContext)
     , connectivityChangedTimer_(*ioContext)
     , logger_(logger)
    {
        if (logger_) logger_->debug("Creating UPnPContext instance [{}]", fmt::ptr(this));
    
        // Set port ranges
        portRange_.emplace(PortType::TCP, std::make_pair(UPNP_TCP_PORT_MIN, UPNP_TCP_PORT_MAX));
        portRange_.emplace(PortType::UDP, std::make_pair(UPNP_UDP_PORT_MIN, UPNP_UDP_PORT_MAX));
    
        ctx->post([this] { init(); });
    }
    
    std::shared_ptr<asio::io_context>
    UPnPContext::createIoContext(const std::shared_ptr<asio::io_context>& ctx, const std::shared_ptr<dht::log::Logger>& logger) {
        if (ctx) {
            return ctx;
        } else {
            if (logger) logger->debug("UPnPContext: starting dedicated io_context thread");
            auto ioCtx = std::make_shared<asio::io_context>();
            ioContextRunner_ = std::make_unique<std::thread>([ioCtx, l=logger]() {
                try {
                    auto work = asio::make_work_guard(*ioCtx);