120 lines
3.2 KiB
C++
120 lines
3.2 KiB
C++
// DPrtMapC.cpp : Implementation of CDynamicPortMappingCollection
|
|
#include "stdafx.h"
|
|
#pragma hdrstop
|
|
|
|
#include "NATUPnP.h"
|
|
#include "DPrtMapC.h"
|
|
#include "DPortMap.h"
|
|
|
|
/////////////////////////////////////////////////////////////////////////////
|
|
// CDynamicPortMappingCollection
|
|
|
|
STDMETHODIMP CDynamicPortMappingCollection::get_Item (BSTR bstrRemoteHost, long lExternalPort, BSTR bstrProtocol, IDynamicPortMapping ** ppDPM)
|
|
{
|
|
NAT_API_ENTER
|
|
|
|
if (!ppDPM)
|
|
return E_POINTER;
|
|
*ppDPM = NULL;
|
|
|
|
return CDynamicPortMapping::CreateInstance (m_spUPS, bstrRemoteHost, lExternalPort, bstrProtocol, ppDPM);
|
|
|
|
NAT_API_LEAVE
|
|
}
|
|
|
|
STDMETHODIMP CDynamicPortMappingCollection::get_Count(long *pVal)
|
|
{
|
|
NAT_API_ENTER
|
|
|
|
if (!pVal)
|
|
return E_POINTER;
|
|
*pVal = 0;
|
|
|
|
ULONG ul = 0;
|
|
HRESULT hr = GetNumberOfEntries (m_spUPS, &ul);
|
|
if (SUCCEEDED(hr))
|
|
*pVal = (long)ul;
|
|
return hr;
|
|
|
|
NAT_API_LEAVE
|
|
}
|
|
|
|
STDMETHODIMP CDynamicPortMappingCollection::Remove(BSTR bstrRemoteHost, long lExternalPort, BSTR bstrProtocol)
|
|
{
|
|
NAT_API_ENTER
|
|
|
|
return DeletePortMapping (m_spUPS, bstrRemoteHost, lExternalPort, bstrProtocol);
|
|
|
|
NAT_API_LEAVE
|
|
}
|
|
|
|
STDMETHODIMP CDynamicPortMappingCollection::Add(BSTR bstrRemoteHost, long lExternalPort, BSTR bstrProtocol, long lInternalPort, BSTR bstrInternalClient, VARIANT_BOOL bEnabled, BSTR bstrDescription, long lLeaseDuration, IDynamicPortMapping ** ppDPM)
|
|
{
|
|
NAT_API_ENTER
|
|
|
|
if (!ppDPM)
|
|
return E_POINTER;
|
|
*ppDPM = NULL;
|
|
|
|
if (!bstrRemoteHost)
|
|
return E_INVALIDARG;
|
|
if ((lExternalPort < 0) || (lExternalPort > 65535))
|
|
return E_INVALIDARG;
|
|
if (!bstrProtocol)
|
|
return E_INVALIDARG;
|
|
if (wcscmp (bstrProtocol, L"TCP") && wcscmp (bstrProtocol, L"UDP"))
|
|
return E_INVALIDARG;
|
|
if ((lInternalPort < 0) || (lInternalPort > 65535))
|
|
return E_INVALIDARG;
|
|
if (!bstrInternalClient)
|
|
return E_INVALIDARG;
|
|
if (!bstrDescription)
|
|
return E_INVALIDARG;
|
|
if ((lLeaseDuration < 0) || (lLeaseDuration > 65535))
|
|
return E_INVALIDARG;
|
|
|
|
HRESULT hr = AddPortMapping (m_spUPS,
|
|
bstrRemoteHost,
|
|
lExternalPort,
|
|
bstrProtocol,
|
|
lInternalPort,
|
|
bstrInternalClient,
|
|
bEnabled,
|
|
bstrDescription,
|
|
lLeaseDuration);
|
|
if (SUCCEEDED(hr)) {
|
|
hr = CDynamicPortMapping::CreateInstance (
|
|
m_spUPS, bstrRemoteHost,lExternalPort,
|
|
bstrProtocol, ppDPM);
|
|
}
|
|
return hr;
|
|
|
|
NAT_API_LEAVE
|
|
}
|
|
|
|
STDMETHODIMP CDynamicPortMappingCollection::get__NewEnum(IUnknown **ppVal)
|
|
{
|
|
NAT_API_ENTER
|
|
|
|
if (!ppVal)
|
|
return E_POINTER;
|
|
*ppVal = NULL;
|
|
|
|
CComPtr<IEnumVARIANT> spEV =
|
|
CEnumDynamicPortMappingCollection::CreateInstance (m_spUPS);
|
|
if (!spEV)
|
|
return E_OUTOFMEMORY;
|
|
return spEV->QueryInterface (__uuidof(IUnknown), (void**)ppVal);
|
|
|
|
NAT_API_LEAVE
|
|
}
|
|
|
|
HRESULT CDynamicPortMappingCollection::Initialize (IUPnPService * pUPS)
|
|
{
|
|
_ASSERT (pUPS);
|
|
_ASSERT (m_spUPS == NULL);
|
|
|
|
m_spUPS = pUPS;
|
|
return S_OK;
|
|
}
|