cUPnPNAT.h

Go to the documentation of this file.

//
//
 
#ifndef _INC_cUPnPNAT_H
#define _INC_cUPnPNAT_H
#ifndef NO_PRAGMA_ONCE
#pragma once
#endif
 
#include "cWinString.h"
#include "../Net/cNetSocket.h"
#include "../Net/cNetAddrName.h"
#include "GrayCore/include/cString.h"
#include "GrayCore/include/cArray.h"
 
namespace GrayLib
{
        UNITTEST2_PREDEF(cUPnPNAT);
 
        class GRAYLIB_LINK cUPnPNATPort
        {
 
        public:
                NET_PORT_t m_nExternalPort;     
                NET_PORT_t m_nInternalPort;     
                IPPROTO_TYPE m_nProtocolType;   
                cString m_sDescription;         
                bool m_bEnabled;
 
        public:
                cUPnPNATPort(NET_PORT_t nExternalPort = k_NET_PORT_INVALID, NET_PORT_t nInternalPort = k_NET_PORT_INVALID, IPPROTO_TYPE nProtocolType = IPPROTO_TCP, cString sDescription = "")
                : m_nExternalPort(nExternalPort)
                , m_nInternalPort(nInternalPort)
                , m_nProtocolType(nProtocolType)
                , m_sDescription(sDescription)
                , m_bEnabled(true)
                {
                }
                void CheckDefaults();
                const GChar_t* get_ProtocolName() const noexcept
                {
                        if (m_nProtocolType == IPPROTO_UDP)
                                return _GT("UDP");
                        return _GT("TCP");
                }
                cStringA getDescFull() const;   
        };
 
        class GRAYLIB_LINK cUPnPNAT     // : public cSingleton<cUPnPNAT>
        {
 
        private:
                cArrayStruct<cUPnPNATPort> m_Mappings;
                cString m_sLastError;
 
                cStringA m_sLocalAddr;          
                cNetAddress     m_LocalAddr;    
 
        private:
                bool InitLocalAddr();
                HRESULT RetError(HRESULT hRes, const char* pszDefaultString);
 
        public:
                cUPnPNAT(void);
                ~cUPnPNAT(void);
 
                HRESULT AddNATPort(cUPnPNATPort& mapping, bool bTryRandom = false);
                HRESULT RemoveNATPort(const cUPnPNATPort& mapping, bool bRemoveFromList = true);
                void RemoveMappings();
 
                cString get_LastErrorStr() const
                {
                        return m_sLastError;
                }
 
                const cNetAddress& get_LocalAddr();
                cStringA get_LocalAddrStr();
                cWinString get_LocalAddrBSTR();
 
                UNITTEST_FRIEND(cUPnPNAT);
        };
} 
 
#endif // _INC_cUPnPNAT_H