flycast/core/network/miniupnp.cpp

100 lines
3.0 KiB
C++

/*
Copyright 2020 flyinghead
This file is part of Flycast.
Flycast 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 2 of the License, or
(at your option) any later version.
Flycast 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 Flycast. If not, see <https://www.gnu.org/licenses/>.
*/
#include "build.h"
#ifndef FEAT_NO_MINIUPNPC
#include <miniupnpc.h>
#include <upnpcommands.h>
#include "log/Log.h"
#include "miniupnp.h"
#include <string>
#ifndef UPNP_LOCAL_PORT_ANY
#define UPNP_LOCAL_PORT_ANY 0
#endif
bool MiniUPnP::Init()
{
DEBUG_LOG(NETWORK, "MiniUPnP::Init");
int error = 0;
#if MINIUPNPC_API_VERSION >= 14
UPNPDev *devlist = upnpDiscover(2000, nullptr, nullptr, UPNP_LOCAL_PORT_ANY, 0, 2, &error);
#else
UPNPDev *devlist = upnpDiscover(2000, nullptr, nullptr, UPNP_LOCAL_PORT_ANY, 0, &error);
#endif
if (devlist == nullptr)
{
WARN_LOG(NETWORK, "UPnP discover failed: error %d", error);
return false;
}
#if MINIUPNPC_API_VERSION >= 18
error = UPNP_GetValidIGD(devlist, &urls, &data, lanAddress, sizeof(lanAddress), nullptr, 0);
#else
error = UPNP_GetValidIGD(devlist, &urls, &data, lanAddress, sizeof(lanAddress));
#endif
freeUPNPDevlist(devlist);
if (error != 1)
{
WARN_LOG(NETWORK, "Internet Gateway not found: error %d", error);
return false;
}
wanAddress[0] = 0;
initialized = true;
if (UPNP_GetExternalIPAddress(urls.controlURL, data.first.servicetype, wanAddress) != 0)
WARN_LOG(NETWORK, "Cannot determine external IP address");
DEBUG_LOG(NETWORK, "MiniUPnP: public IP is %s", wanAddress);
return true;
}
void MiniUPnP::Term()
{
if (!initialized)
return;
DEBUG_LOG(NETWORK, "MiniUPnP::Term");
for (const auto& port : mappedPorts)
UPNP_DeletePortMapping(urls.controlURL, data.first.servicetype, port.first.c_str(),
port.second ? "TCP" : "UDP", nullptr);
mappedPorts.clear();
FreeUPNPUrls(&urls);
initialized = false;
DEBUG_LOG(NETWORK, "MiniUPnP: terminated");
}
bool MiniUPnP::AddPortMapping(int port, bool tcp)
{
std::string portStr(std::to_string(port));
int error = UPNP_AddPortMapping(urls.controlURL, data.first.servicetype,
portStr.c_str(), // WAN port
portStr.c_str(), // LAN port
lanAddress,
"Flycast",
tcp ? "TCP" : "UDP",
nullptr, // remote (peer) host address or nullptr for no restriction
"86400"); // port map lease duration (in seconds) or zero for "as long as possible"
if (error != 0)
{
WARN_LOG(NETWORK, "Port %d redirection failed: error %d", port, error);
return false;
}
mappedPorts.emplace_back(portStr, tcp);
DEBUG_LOG(NETWORK, "MiniUPnP: forwarding %s port %d", tcp ? "TCP" : "UDP", port);
return true;
}
#endif