dolphin/Source/Core/Common/UPnP.cpp

Ignoring revisions in .git-blame-ignore-revs. Click here to bypass and see the normal blame view.

183 lines
4.5 KiB
C++
Raw Normal View History

// Copyright 2017 Dolphin Emulator Project
// SPDX-License-Identifier: GPL-2.0-or-later
#ifdef USE_UPNP
#include "Common/UPnP.h"
#include "Common/Logging/Log.h"
2017-08-07 08:41:48 +02:00
#include <array>
2017-08-07 08:45:39 +02:00
#include <cstdlib>
#include <cstring>
#include <miniupnpc.h>
#include <miniwget.h>
#include <string>
#include <thread>
#include <upnpcommands.h>
#include <upnperrors.h>
2017-08-07 08:45:39 +02:00
#include <vector>
2017-08-07 08:37:17 +02:00
static UPNPUrls s_urls;
static IGDdatas s_data;
2017-08-07 08:41:48 +02:00
static std::array<char, 20> s_our_ip;
2017-08-07 08:37:17 +02:00
static u16 s_mapped = 0;
static std::thread s_thread;
// called from ---UPnP--- thread
// discovers the IGD
2017-08-07 08:43:24 +02:00
static bool InitUPnP()
{
static bool s_inited = false;
static bool s_error = false;
// Don't init if already inited
if (s_inited)
return true;
// Don't init if it failed before
if (s_error)
return false;
s_urls = {};
s_data = {};
// Find all UPnP devices
int upnperror = 0;
std::unique_ptr<UPNPDev, decltype(&freeUPNPDevlist)> devlist(nullptr, freeUPNPDevlist);
#if MINIUPNPC_API_VERSION >= 14
devlist.reset(upnpDiscover(2000, nullptr, nullptr, 0, 0, 2, &upnperror));
#else
devlist.reset(upnpDiscover(2000, nullptr, nullptr, 0, 0, &upnperror));
#endif
if (!devlist)
{
if (upnperror == UPNPDISCOVER_SUCCESS)
{
WARN_LOG_FMT(NETPLAY, "No UPnP devices could be found.");
}
else
{
WARN_LOG_FMT(NETPLAY, "An error occurred trying to discover UPnP devices: {}",
strupnperror(upnperror));
}
s_error = true;
return false;
}
// Look for the IGD
bool found_valid_igd = false;
for (UPNPDev* dev = devlist.get(); dev; dev = dev->pNext)
{
if (!std::strstr(dev->st, "InternetGatewayDevice"))
continue;
2017-08-07 08:58:00 +02:00
int desc_xml_size = 0;
std::unique_ptr<char, decltype(&std::free)> desc_xml(nullptr, std::free);
int statusCode = 200;
#if MINIUPNPC_API_VERSION >= 16
2017-08-07 08:58:00 +02:00
desc_xml.reset(
static_cast<char*>(miniwget_getaddr(dev->descURL, &desc_xml_size, s_our_ip.data(),
2017-08-07 08:41:48 +02:00
static_cast<int>(s_our_ip.size()), 0, &statusCode)));
#else
2017-08-07 08:58:00 +02:00
desc_xml.reset(static_cast<char*>(miniwget_getaddr(
dev->descURL, &desc_xml_size, s_our_ip.data(), static_cast<int>(s_our_ip.size()), 0)));
#endif
2017-08-07 08:58:00 +02:00
if (desc_xml && statusCode == 200)
{
2017-08-07 08:58:00 +02:00
parserootdesc(desc_xml.get(), desc_xml_size, &s_data);
2017-08-07 08:37:17 +02:00
GetUPNPUrls(&s_urls, &s_data, dev->descURL, 0);
found_valid_igd = true;
NOTICE_LOG_FMT(NETPLAY, "Got info from IGD at {}.", dev->descURL);
break;
}
else
{
WARN_LOG_FMT(NETPLAY, "Error getting info from IGD at {}.", dev->descURL);
}
}
if (!found_valid_igd)
WARN_LOG_FMT(NETPLAY, "Could not find a valid IGD in the discovered UPnP devices.");
s_inited = true;
return true;
}
// called from ---UPnP--- thread
// Attempt to stop portforwarding.
// --
// NOTE: It is important that this happens! A few very crappy routers
// apparently do not delete UPnP mappings on their own, so if you leave them
// hanging, the NVRAM will fill with portmappings, and eventually all UPnP
// requests will fail silently, with the only recourse being a factory reset.
// --
2017-08-07 08:43:24 +02:00
static bool UnmapPort(const u16 port)
{
2017-08-07 08:45:39 +02:00
std::string port_str = std::to_string(port);
2017-08-07 08:37:17 +02:00
UPNP_DeletePortMapping(s_urls.controlURL, s_data.first.servicetype, port_str.c_str(), "UDP",
nullptr);
return true;
}
// called from ---UPnP--- thread
// Attempt to portforward!
2017-08-07 08:43:24 +02:00
static bool MapPort(const char* addr, const u16 port)
{
2017-08-07 08:37:17 +02:00
if (s_mapped > 0)
2017-08-07 08:43:24 +02:00
UnmapPort(s_mapped);
2017-08-07 08:45:39 +02:00
std::string port_str = std::to_string(port);
int result = UPNP_AddPortMapping(
2017-08-07 08:41:48 +02:00
s_urls.controlURL, s_data.first.servicetype, port_str.c_str(), port_str.c_str(), addr,
2017-08-07 08:37:17 +02:00
(std::string("dolphin-emu UDP on ") + addr).c_str(), "UDP", nullptr, nullptr);
if (result != 0)
return false;
2017-08-07 08:37:17 +02:00
s_mapped = port;
return true;
}
// UPnP thread: try to map a port
2017-08-07 08:43:24 +02:00
static void MapPortThread(const u16 port)
{
2017-08-07 08:43:24 +02:00
if (InitUPnP() && MapPort(s_our_ip.data(), port))
{
NOTICE_LOG_FMT(NETPLAY, "Successfully mapped port {} to {}.", port, s_our_ip.data());
return;
}
WARN_LOG_FMT(NETPLAY, "Failed to map port {} to {}.", port, s_our_ip.data());
}
// UPnP thread: try to unmap a port
2017-08-07 08:43:24 +02:00
static void UnmapPortThread()
{
2017-08-07 08:37:17 +02:00
if (s_mapped > 0)
2017-08-07 08:43:24 +02:00
UnmapPort(s_mapped);
}
void Common::UPnP::TryPortmapping(u16 port)
{
2017-08-07 08:37:17 +02:00
if (s_thread.joinable())
s_thread.join();
2017-08-07 08:43:24 +02:00
s_thread = std::thread(&MapPortThread, port);
}
void Common::UPnP::StopPortmapping()
{
2017-08-07 08:37:17 +02:00
if (s_thread.joinable())
s_thread.join();
2017-08-07 08:43:24 +02:00
s_thread = std::thread(&UnmapPortThread);
2017-08-07 08:37:17 +02:00
s_thread.join();
}
#endif