mirror of
https://github.com/ddnet/ddnet.git
synced 2024-09-21 10:04:18 +00:00
74 lines
1.8 KiB
C++
74 lines
1.8 KiB
C++
#ifdef CONF_UPNP
|
|
|
|
#include <miniupnpc/miniupnpc.h>
|
|
#include <miniupnpc/upnpcommands.h>
|
|
#include <miniupnpc/upnperrors.h>
|
|
#include <engine/shared/config.h>
|
|
#include <base/system.h>
|
|
#include <game/version.h>
|
|
#include "upnp.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");
|
|
|
|
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 |