mirror of
https://github.com/RetroShare/RetroShare.git
synced 2024-10-01 02:35:48 -04:00
ae4ee9df00
* Fixed #includes in upnp * switched p3disc over to the new data types. * added new source to .pro file * minor changes to other code. git-svn-id: http://svn.code.sf.net/p/retroshare/code/trunk@3249 b45a01b8-16f6-495d-af2f-9b41ad6348cc
198 lines
4.2 KiB
C++
198 lines
4.2 KiB
C++
//Linux only...
|
|
#if !defined(WINDOWS_SYS) && !defined(__APPLE__)
|
|
|
|
#ifndef _RS_UPNP_IFACE_H
|
|
#define _RS_UPNP_IFACE_H
|
|
|
|
#include <string.h>
|
|
|
|
/* platform independent networking... */
|
|
#include "pqi/pqinetwork.h"
|
|
#include "pqi/pqiassist.h"
|
|
|
|
#include "util/rsthreads.h"
|
|
|
|
#include <upnp/upnp.h>
|
|
#include "upnp/UPnPBase.h"
|
|
|
|
#define RS_UPNP_S_UNINITIALISED 0
|
|
#define RS_UPNP_S_UNAVAILABLE 1
|
|
#define RS_UPNP_S_READY 2
|
|
#define RS_UPNP_S_TCP_AND_FAILED 3
|
|
//#define RS_UPNP_S_UDP_FAILED 4
|
|
#define RS_UPNP_S_ACTIVE 5
|
|
|
|
class upnphandler: public pqiNetAssistFirewall
|
|
{
|
|
public:
|
|
|
|
upnphandler();
|
|
virtual ~upnphandler();
|
|
|
|
/* External Interface (pqiNetAssistFirewall) */
|
|
virtual void enable(bool active);
|
|
virtual void shutdown();
|
|
virtual void restart();
|
|
|
|
virtual bool getEnabled();
|
|
virtual bool getActive();
|
|
|
|
virtual void setInternalPort(unsigned short iport_in);
|
|
virtual void setExternalPort(unsigned short eport_in);
|
|
virtual bool getInternalAddress(struct sockaddr_in &addr);
|
|
virtual bool getExternalAddress(struct sockaddr_in &addr);
|
|
|
|
/* Public functions - for background thread operation,
|
|
* but effectively private from rest of RS, as in derived class
|
|
*/
|
|
unsigned int upnpState;
|
|
|
|
bool start_upnp();
|
|
bool shutdown_upnp();
|
|
|
|
bool initUPnPState();
|
|
|
|
/* Mutex for data below */
|
|
RsMutex dataMtx;
|
|
|
|
private:
|
|
|
|
CUPnPControlPoint *cUPnPControlPoint;
|
|
|
|
bool background_setup_upnp(bool, bool);
|
|
|
|
|
|
bool toEnable; /* overall on/off switch */
|
|
bool toStart; /* if set start forwarding */
|
|
bool toStop; /* if set stop forwarding */
|
|
|
|
unsigned short iport;
|
|
unsigned short eport; /* config */
|
|
unsigned short eport_curr; /* current forwarded */
|
|
|
|
/* info from upnp */
|
|
struct sockaddr_in upnp_iaddr;
|
|
struct sockaddr_in upnp_eaddr;
|
|
};
|
|
|
|
/* info from upnp */
|
|
int CtrlPointCallbackEventHandler(Upnp_EventType ,void* , void*);
|
|
|
|
#endif /* _RS_UPNP_IFACE_H */
|
|
|
|
#endif
|
|
|
|
|
|
|
|
|
|
|
|
#if defined(WINDOWS_SYS) || defined(__APPLE__)
|
|
|
|
//windows implementation
|
|
#ifndef _RS_UPNP_IFACE_H
|
|
#define _RS_UPNP_IFACE_H
|
|
|
|
#include <string.h>
|
|
|
|
#include <string>
|
|
#include <map>
|
|
|
|
/* platform independent networking... */
|
|
#include "pqi/pqinetwork.h"
|
|
#include "pqi/pqiassist.h"
|
|
|
|
#include "util/rsthreads.h"
|
|
|
|
class upnpentry
|
|
{
|
|
public:
|
|
std::string name;
|
|
std::string id;
|
|
struct sockaddr_in addr;
|
|
unsigned int flags;
|
|
int status;
|
|
int lastTs;
|
|
};
|
|
|
|
class upnpforward
|
|
{
|
|
public:
|
|
std::string name;
|
|
unsigned int flags;
|
|
struct sockaddr_in iaddr;
|
|
struct sockaddr_in eaddr;
|
|
int status;
|
|
int lastTs;
|
|
};
|
|
|
|
|
|
#define RS_UPNP_S_UNINITIALISED 0
|
|
#define RS_UPNP_S_UNAVAILABLE 1
|
|
#define RS_UPNP_S_READY 2
|
|
#define RS_UPNP_S_TCP_FAILED 3
|
|
#define RS_UPNP_S_UDP_FAILED 4
|
|
#define RS_UPNP_S_ACTIVE 5
|
|
|
|
class uPnPConfigData;
|
|
|
|
class upnphandler: public pqiNetAssistFirewall
|
|
{
|
|
public:
|
|
|
|
upnphandler();
|
|
virtual ~upnphandler();
|
|
|
|
/* External Interface (pqiNetAssistFirewall) */
|
|
virtual void enable(bool active);
|
|
virtual void shutdown();
|
|
virtual void restart();
|
|
|
|
virtual bool getEnabled();
|
|
virtual bool getActive();
|
|
|
|
virtual void setInternalPort(unsigned short iport_in);
|
|
virtual void setExternalPort(unsigned short eport_in);
|
|
virtual bool getInternalAddress(struct sockaddr_in &addr);
|
|
virtual bool getExternalAddress(struct sockaddr_in &addr);
|
|
|
|
/* Public functions - for background thread operation,
|
|
* but effectively private from rest of RS, as in derived class
|
|
*/
|
|
|
|
bool start_upnp();
|
|
bool shutdown_upnp();
|
|
|
|
bool initUPnPState();
|
|
bool printUPnPState();
|
|
|
|
private:
|
|
|
|
bool background_setup_upnp(bool, bool);
|
|
bool checkUPnPActive();
|
|
|
|
/* Mutex for data below */
|
|
RsMutex dataMtx;
|
|
|
|
bool toEnable; /* overall on/off switch */
|
|
bool toStart; /* if set start forwarding */
|
|
bool toStop; /* if set stop forwarding */
|
|
|
|
unsigned short iport;
|
|
unsigned short eport; /* config */
|
|
unsigned short eport_curr; /* current forwarded */
|
|
|
|
/* info from upnp */
|
|
unsigned int upnpState;
|
|
uPnPConfigData *upnpConfig;
|
|
|
|
struct sockaddr_in upnp_iaddr;
|
|
struct sockaddr_in upnp_eaddr;
|
|
|
|
/* active port forwarding */
|
|
std::list<upnpforward> activeForwards;
|
|
|
|
};
|
|
|
|
#endif /* _RS_UPNP_IFACE_H */
|
|
#endif
|