2015-01-03 20:38:48 +00:00
|
|
|
#ifndef __UPNP_H__
|
|
|
|
#define __UPNP_H__
|
|
|
|
|
|
|
|
#ifdef USE_UPNP
|
|
|
|
#include <string>
|
|
|
|
#include <thread>
|
2016-07-31 14:22:41 +00:00
|
|
|
#include <condition_variable>
|
|
|
|
#include <mutex>
|
2016-07-18 18:01:58 +00:00
|
|
|
#include <memory>
|
2015-01-03 20:38:48 +00:00
|
|
|
|
|
|
|
#include <miniupnpc/miniwget.h>
|
|
|
|
#include <miniupnpc/miniupnpc.h>
|
|
|
|
#include <miniupnpc/upnpcommands.h>
|
|
|
|
#include <miniupnpc/upnperrors.h>
|
2014-02-09 20:15:47 +00:00
|
|
|
|
|
|
|
#include <boost/asio.hpp>
|
|
|
|
|
|
|
|
namespace i2p
|
|
|
|
{
|
2015-05-06 16:19:20 +00:00
|
|
|
namespace transport
|
2015-01-03 20:38:48 +00:00
|
|
|
{
|
2019-09-05 18:58:28 +00:00
|
|
|
const int UPNP_RESPONSE_TIMEOUT = 2000; // in milliseconds
|
|
|
|
|
|
|
|
enum
|
|
|
|
{
|
|
|
|
UPNP_IGD_NONE = 0,
|
|
|
|
UPNP_IGD_VALID_CONNECTED = 1,
|
|
|
|
UPNP_IGD_VALID_NOT_CONNECTED = 2,
|
|
|
|
UPNP_IGD_INVALID = 3
|
|
|
|
};
|
|
|
|
|
2014-02-09 20:15:47 +00:00
|
|
|
class UPnP
|
|
|
|
{
|
2019-09-05 18:58:28 +00:00
|
|
|
public:
|
2015-01-03 20:38:48 +00:00
|
|
|
|
|
|
|
UPnP ();
|
|
|
|
~UPnP ();
|
2019-09-05 18:58:28 +00:00
|
|
|
void Close ();
|
2015-01-03 20:38:48 +00:00
|
|
|
|
2019-09-05 18:58:28 +00:00
|
|
|
void Start ();
|
|
|
|
void Stop ();
|
2015-01-03 20:38:48 +00:00
|
|
|
|
2019-09-05 18:58:28 +00:00
|
|
|
private:
|
2016-07-18 18:01:58 +00:00
|
|
|
|
|
|
|
void Discover ();
|
2019-09-05 18:58:28 +00:00
|
|
|
int CheckMapping (const char* port, const char* type);
|
2016-07-18 18:01:58 +00:00
|
|
|
void PortMapping ();
|
|
|
|
void TryPortMapping (std::shared_ptr<i2p::data::RouterInfo::Address> address);
|
|
|
|
void CloseMapping ();
|
|
|
|
void CloseMapping (std::shared_ptr<i2p::data::RouterInfo::Address> address);
|
|
|
|
|
2015-01-03 20:38:48 +00:00
|
|
|
void Run ();
|
2016-07-18 18:01:58 +00:00
|
|
|
std::string GetProto (std::shared_ptr<i2p::data::RouterInfo::Address> address);
|
2015-01-03 20:38:48 +00:00
|
|
|
|
2019-09-05 18:58:28 +00:00
|
|
|
private:
|
2018-01-06 03:48:51 +00:00
|
|
|
|
2016-07-18 18:01:58 +00:00
|
|
|
bool m_IsRunning;
|
2019-09-05 18:58:28 +00:00
|
|
|
std::unique_ptr<std::thread> m_Thread;
|
2018-01-06 03:48:51 +00:00
|
|
|
std::condition_variable m_Started;
|
|
|
|
std::mutex m_StartedMutex;
|
2016-07-18 18:01:58 +00:00
|
|
|
boost::asio::io_service m_Service;
|
2018-01-06 03:48:51 +00:00
|
|
|
boost::asio::deadline_timer m_Timer;
|
2019-09-05 19:21:26 +00:00
|
|
|
bool m_upnpUrlsInitialized=false;
|
2019-09-05 18:58:28 +00:00
|
|
|
struct UPNPUrls m_upnpUrls;
|
|
|
|
struct IGDdatas m_upnpData;
|
|
|
|
|
|
|
|
// For miniupnpc
|
|
|
|
struct UPNPDev * m_Devlist = 0;
|
|
|
|
char m_NetworkAddr[64];
|
|
|
|
char m_externalIPAddress[40];
|
2015-01-03 20:38:48 +00:00
|
|
|
};
|
2014-02-09 20:15:47 +00:00
|
|
|
}
|
2015-01-03 20:38:48 +00:00
|
|
|
}
|
|
|
|
|
2016-07-12 00:00:00 +00:00
|
|
|
#else // USE_UPNP
|
|
|
|
namespace i2p {
|
|
|
|
namespace transport {
|
2019-09-05 18:58:28 +00:00
|
|
|
/* class stub */
|
|
|
|
class UPnP {
|
|
|
|
public:
|
|
|
|
|
|
|
|
UPnP () {};
|
|
|
|
~UPnP () {};
|
|
|
|
void Start () { LogPrint(eLogWarning, "UPnP: this module was disabled at compile-time"); }
|
|
|
|
void Stop () {};
|
|
|
|
};
|
2016-07-12 00:00:00 +00:00
|
|
|
}
|
|
|
|
}
|
2016-03-24 00:00:00 +00:00
|
|
|
#endif // USE_UPNP
|
|
|
|
#endif // __UPNP_H__
|