ddnet/src/engine/server/upnp.cpp

73 lines
1.8 KiB
C++
Raw Normal View History

2020-04-19 13:14:21 +00:00
#ifdef CONF_UPNP
#include "upnp.h"
#include <base/system.h>
#include <engine/shared/config.h>
#include <game/version.h>
2020-04-19 13:14:21 +00:00
#include <miniupnpc/miniupnpc.h>
#include <miniupnpc/upnpcommands.h>
#include <miniupnpc/upnperrors.h>
void CUPnP::Open(NETADDR Address)
{
if(g_Config.m_SvUseUPnP)
{
m_Enabled = false;
m_Addr = Address;
m_UPnPUrls = (struct UPNPUrls *)malloc(sizeof(struct UPNPUrls));
m_UPnPData = (struct IGDdatas *)malloc(sizeof(struct IGDdatas));
char aLanAddr[64];
char aPort[6];
int Error;
m_UPnPDevice = upnpDiscover(2000, NULL, NULL, 0, 0, 2, &Error);
int Status = UPNP_GetValidIGD(m_UPnPDevice, m_UPnPUrls, m_UPnPData, aLanAddr, sizeof(aLanAddr));
dbg_msg("upnp", "status=%d, lan_addr=%s", Status, aLanAddr);
if(Status == 1)
{
m_Enabled = true;
dbg_msg("upnp", "found valid IGD: %s", m_UPnPUrls->controlURL);
str_format(aPort, sizeof(aPort), "%d", m_Addr.port);
Error = UPNP_AddPortMapping(m_UPnPUrls->controlURL, m_UPnPData->first.servicetype,
aPort, aPort, aLanAddr,
"DDNet Server " GAME_RELEASE_VERSION,
"UDP", NULL, "0");
2020-04-19 13:14:21 +00:00
if(Error)
dbg_msg("upnp", "failed to map port, error: %s", strupnperror(Error));
else
dbg_msg("upnp", "successfully mapped port");
}
else
dbg_msg("upnp", "no valid IGD found, disabled");
}
}
void CUPnP::Shutdown()
{
if(g_Config.m_SvUseUPnP)
{
if(m_Enabled)
{
char aPort[6];
str_format(aPort, sizeof(aPort), "%d", m_Addr.port);
int Error = UPNP_DeletePortMapping(m_UPnPUrls->controlURL, m_UPnPData->first.servicetype, aPort, "UDP", NULL);
if(Error != 0)
{
dbg_msg("upnp", "failed to delete port mapping on shutdown: %s", strupnperror(Error));
}
FreeUPNPUrls(m_UPnPUrls);
freeUPNPDevlist(m_UPnPDevice);
}
free(m_UPnPUrls);
free(m_UPnPData);
m_UPnPUrls = NULL;
m_UPnPData = NULL;
}
}
#endif