Leaked source code of windows server 2003
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 
 

795 lines
17 KiB

// binding.cpp : Implementation of CNntpServerBinding & CNntpServerBindings.
#include "stdafx.h"
#include "nntpcmn.h"
#include "cmultisz.h"
#include "binding.h"
#include "oleutil.h"
// #include <stdio.h>
HRESULT CBinding::SetProperties (
BSTR strIpAddress,
long dwTcpPort,
long dwSslPort
)
{
_ASSERT ( IS_VALID_STRING ( strIpAddress ) );
m_strIpAddress = strIpAddress;
m_dwTcpPort = dwTcpPort;
m_dwSslPort = dwSslPort;
if ( !m_strIpAddress ) {
return E_OUTOFMEMORY;
}
return NOERROR;
}
// Must define THIS_FILE_* macros to use NntpCreateException()
#define THIS_FILE_HELP_CONTEXT 0
#define THIS_FILE_PROG_ID _T("Nntpadm.VirtualServer.1")
#define THIS_FILE_IID IID_INntpServerBinding
/////////////////////////////////////////////////////////////////////////////
//
STDMETHODIMP CNntpServerBinding::InterfaceSupportsErrorInfo(REFIID riid)
{
static const IID* arr[] =
{
&IID_INntpServerBinding,
};
for (int i=0;i<sizeof(arr)/sizeof(arr[0]);i++)
{
if (InlineIsEqualGUID(*arr[i],riid))
return S_OK;
}
return S_FALSE;
}
CNntpServerBinding::CNntpServerBinding ()
// CComBSTR's are initialized to NULL by default.
{
}
CNntpServerBinding::~CNntpServerBinding ()
{
// All CComBSTR's are freed automatically.
}
//////////////////////////////////////////////////////////////////////
// Properties:
//////////////////////////////////////////////////////////////////////
STDMETHODIMP CNntpServerBinding::get_IpAddress ( BSTR * pstrIpAddress )
{
return StdPropertyGet ( m_binding.m_strIpAddress, pstrIpAddress );
}
STDMETHODIMP CNntpServerBinding::put_IpAddress ( BSTR strIpAddress )
{
return StdPropertyPut ( &m_binding.m_strIpAddress, strIpAddress );
}
STDMETHODIMP CNntpServerBinding::get_TcpPort ( long * pdwTcpPort )
{
return StdPropertyGet ( m_binding.m_dwTcpPort, pdwTcpPort );
}
STDMETHODIMP CNntpServerBinding::put_TcpPort ( long dwTcpPort )
{
return StdPropertyPut ( &m_binding.m_dwTcpPort, dwTcpPort );
}
STDMETHODIMP CNntpServerBinding::get_SslPort ( long * plSslPort )
{
return StdPropertyGet ( m_binding.m_dwSslPort, plSslPort );
}
STDMETHODIMP CNntpServerBinding::put_SslPort ( long lSslPort )
{
return StdPropertyPut ( &m_binding.m_dwSslPort, lSslPort );
}
//
// Must define THIS_FILE_* macros to use NntpCreateException()
//
#undef THIS_FILE_HELP_CONTEXT
#undef THIS_FILE_PROG_ID
#undef THIS_FILE_IID
#define THIS_FILE_HELP_CONTEXT 0
#define THIS_FILE_PROG_ID _T("Nntpadm.VirtualServer.1")
#define THIS_FILE_IID IID_INntpServerBindings
/////////////////////////////////////////////////////////////////////////////
//
STDMETHODIMP CNntpServerBindings::InterfaceSupportsErrorInfo(REFIID riid)
{
static const IID* arr[] =
{
&IID_INntpServerBindings,
};
for (int i=0;i<sizeof(arr)/sizeof(arr[0]);i++)
{
if (InlineIsEqualGUID(*arr[i],riid))
return S_OK;
}
return S_FALSE;
}
CNntpServerBindings::CNntpServerBindings () :
m_dwCount ( 0 ),
m_rgBindings ( NULL )
// CComBSTR's are initialized to NULL by default.
{
}
CNntpServerBindings::~CNntpServerBindings ()
{
// All CComBSTR's are freed automatically.
delete [] m_rgBindings;
}
//////////////////////////////////////////////////////////////////////
// Properties:
//////////////////////////////////////////////////////////////////////
STDMETHODIMP CNntpServerBindings::get_Count ( long * pdwCount )
{
return StdPropertyGet ( m_dwCount, pdwCount );
}
//////////////////////////////////////////////////////////////////////
// Methods:
//////////////////////////////////////////////////////////////////////
STDMETHODIMP CNntpServerBindings::Item (
long index,
INntpServerBinding ** ppBinding
)
{
TraceFunctEnter ( "CNntpServerBindings::Item" );
_ASSERT ( IS_VALID_OUT_PARAM ( ppBinding ) );
*ppBinding = NULL;
HRESULT hr = NOERROR;
CComObject<CNntpServerBinding> * pBinding = NULL;
if ( index < 0 || index >= m_dwCount ) {
hr = NntpCreateException ( IDS_NNTPEXCEPTION_INVALID_INDEX );
goto Exit;
}
hr = CComObject<CNntpServerBinding>::CreateInstance ( &pBinding );
if ( FAILED(hr) ) {
goto Exit;
}
_ASSERT ( pBinding );
hr = pBinding->SetProperties ( m_rgBindings[index] );
if ( FAILED(hr) ) {
goto Exit;
}
hr = pBinding->QueryInterface ( IID_INntpServerBinding, (void **) ppBinding );
_ASSERT ( SUCCEEDED(hr) );
Exit:
if ( FAILED(hr) ) {
delete pBinding;
}
TRACE_HRESULT(hr);
TraceFunctLeave ();
return hr;
}
STDMETHODIMP CNntpServerBindings::ItemDispatch ( long index, IDispatch ** ppDispatch )
{
HRESULT hr;
CComPtr<INntpServerBinding> pBinding;
hr = Item ( index, &pBinding );
BAIL_ON_FAILURE ( hr );
hr = pBinding->QueryInterface ( IID_IDispatch, (void **) ppDispatch );
BAIL_ON_FAILURE ( hr );
Exit:
return hr;
}
STDMETHODIMP CNntpServerBindings::Add (
BSTR strIpAddress,
long dwTcpPort,
long dwSslPort
)
{
TraceFunctEnter ( "CNntpServerBindings::Add" );
_ASSERT ( IS_VALID_STRING ( strIpAddress ) );
HRESULT hr = NOERROR;
CBinding * rgNewBindings = NULL;
long i;
//
// Validate the new binding:
//
//
// See if we can merge this binding with an existing one:
//
if ( dwTcpPort == 0 || dwSslPort == 0 ) {
for ( i = 0; i < m_dwCount; i++ ) {
if ( (dwTcpPort == 0 && m_rgBindings[i].m_dwSslPort == 0) ||
(dwSslPort == 0 && m_rgBindings[i].m_dwTcpPort == 0) ) {
if ( lstrcmpi ( m_rgBindings[i].m_strIpAddress, strIpAddress ) == 0 ) {
if ( m_rgBindings[i].m_dwSslPort == 0 ) {
m_rgBindings[i].m_dwSslPort = dwSslPort;
}
else {
m_rgBindings[i].m_dwTcpPort = dwTcpPort;
}
hr = NOERROR;
goto Exit;
}
}
}
}
// Allocate the new binding array:
rgNewBindings = new CBinding [ m_dwCount + 1 ];
if ( !rgNewBindings ) {
hr = E_OUTOFMEMORY;
goto Exit;
}
// Copy the old bindings to the new array:
for ( i = 0; i < m_dwCount; i++ ) {
hr = rgNewBindings[i].SetProperties ( m_rgBindings[i] );
if ( FAILED (hr) ) {
goto Exit;
}
}
// Add the new binding to the end of the array:
hr = rgNewBindings[m_dwCount].SetProperties ( strIpAddress, dwTcpPort, dwSslPort );
if ( FAILED(hr) ) {
goto Exit;
}
_ASSERT ( SUCCEEDED(hr) );
delete [] m_rgBindings;
m_rgBindings = rgNewBindings;
rgNewBindings = NULL;
m_dwCount++;
Exit:
if (rgNewBindings) {
delete [] rgNewBindings;
}
TraceFunctLeave ();
return hr;
}
STDMETHODIMP CNntpServerBindings::ChangeBinding (
long index,
INntpServerBinding * pBinding
)
{
TraceFunctEnter ( "CNntpServerBindings::ChangeBinding" );
HRESULT hr = NOERROR;
CComBSTR strIpAddress;
long dwTcpPort;
long dwSslPort;
if ( index < 0 || index >= m_dwCount ) {
hr = NntpCreateException ( IDS_NNTPEXCEPTION_INVALID_INDEX );
goto Exit;
}
hr = pBinding->get_IpAddress ( &strIpAddress );
if ( FAILED(hr) ) {
goto Exit;
}
hr = pBinding->get_TcpPort ( &dwTcpPort );
if ( FAILED(hr) ) {
goto Exit;
}
hr = pBinding->get_SslPort ( &dwSslPort );
if ( FAILED(hr) ) {
goto Exit;
}
hr = m_rgBindings[index].SetProperties ( strIpAddress, dwTcpPort, dwSslPort );
if ( FAILED(hr) ) {
goto Exit;
}
Exit:
TraceFunctLeave ();
return hr;
}
STDMETHODIMP CNntpServerBindings::ChangeBindingDispatch ( long index, IDispatch * pDispatch )
{
HRESULT hr;
CComPtr<INntpServerBinding> pBinding;
hr = pDispatch->QueryInterface ( IID_INntpServerBinding, (void **) &pBinding );
BAIL_ON_FAILURE ( hr );
hr = ChangeBinding ( index, pBinding );
BAIL_ON_FAILURE ( hr );
Exit:
return hr;
}
STDMETHODIMP CNntpServerBindings::Remove ( long index )
{
TraceFunctEnter ( "CNntpServerBindings::Remove" );
HRESULT hr = NOERROR;
CBinding temp;
long cPositionsToSlide;
if ( index < 0 || index >= m_dwCount ) {
hr = NntpCreateException ( IDS_NNTPEXCEPTION_INVALID_INDEX );
goto Exit;
}
// Slide the array down by one position:
_ASSERT ( m_rgBindings );
cPositionsToSlide = (m_dwCount - 1) - index;
_ASSERT ( cPositionsToSlide < m_dwCount );
if ( cPositionsToSlide > 0 ) {
// Save the deleted binding in temp:
CopyMemory ( &temp, &m_rgBindings[index], sizeof ( CBinding ) );
// Move the array down one:
MoveMemory ( &m_rgBindings[index], &m_rgBindings[index + 1], sizeof ( CBinding ) * cPositionsToSlide );
// Put the deleted binding on the end (so it gets destructed):
CopyMemory ( &m_rgBindings[m_dwCount - 1], &temp, sizeof ( CBinding ) );
// Zero out the temp binding:
ZeroMemory ( &temp, sizeof ( CBinding ) );
}
m_dwCount--;
Exit:
TraceFunctLeave ();
return hr;
}
STDMETHODIMP CNntpServerBindings::Clear ( )
{
delete [] m_rgBindings;
m_rgBindings = NULL;
m_dwCount = 0;
return NOERROR;
}
//////////////////////////////////////////////////////////////////////
//
// Useful routines to go from INntpServerBindings to
// Metabase data types.
//
//////////////////////////////////////////////////////////////////////
static DWORD CountBindingChars ( LPCWSTR strIpAddress, DWORD dwPort )
{
_ASSERT ( IS_VALID_STRING ( strIpAddress ) );
DWORD cchResult = 0;
WCHAR wszPort [256];
wsprintf ( wszPort, _T("%u"), dwPort );
cchResult += lstrlen ( strIpAddress ); // <IPADDRESS>
cchResult += 1; // :
cchResult += lstrlen ( wszPort ); // <PORT>
cchResult += 1; // :
// cchResult += lstrlen ( strPathHeader ); // <PATHHEADER>
cchResult += 1; // For the terminating NULL
return cchResult;
}
static void ToBindingString ( LPCWSTR strIpAddress, DWORD dwPort, LPWSTR wszBinding )
{
_ASSERT ( IS_VALID_STRING ( strIpAddress ) );
_ASSERT ( dwPort != 0 );
_ASSERT ( !IsBadWritePtr ( wszBinding, CountBindingChars ( strIpAddress, dwPort ) ) );
wsprintf ( wszBinding, _T("%s:%u:"), strIpAddress, dwPort );
}
static HRESULT FromBindingString ( LPCWSTR wszBinding, LPWSTR wszIpAddressOut, DWORD * pdwPort )
{
HRESULT hr = NOERROR;
LPWSTR pchFirstColon;
LPWSTR pchSecondColon;
WCHAR wszIpAddress [ 256 ];
WCHAR wszPort [ 256 ];
long dwPort;
LPWSTR pchColon;
wszIpAddress[0] = NULL;
wszPort[0] = NULL;
pchFirstColon = wcschr ( wszBinding, _T(':') );
if ( pchFirstColon ) {
pchSecondColon = wcschr ( pchFirstColon + 1, _T(':') );
}
if ( !pchFirstColon || !pchSecondColon ) {
hr = HRESULT_FROM_WIN32 ( ERROR_INVALID_DATA );
goto Exit;
}
lstrcpyn ( wszIpAddress, wszBinding, 250 );
lstrcpyn ( wszPort, pchFirstColon + 1, 250 );
// Get the Port:
dwPort = _wtoi ( wszPort );
// Cutoff the IpAddress at the colon:
pchColon = wcschr ( wszIpAddress, _T(':') );
if ( pchColon ) {
*pchColon = NULL;
}
lstrcpy ( wszIpAddressOut, wszIpAddress );
*pdwPort = dwPort;
Exit:
return hr;
}
HRESULT
MDBindingsToIBindings (
CMultiSz * pmsz,
BOOL fTcpBindings,
INntpServerBindings * pBindings
)
{
HRESULT hr = NOERROR;
DWORD cBindings;
DWORD i;
LPCWSTR pchCurrent;
CBinding binding;
cBindings = pmsz->Count ();
for (
i = 0, pchCurrent = *pmsz;
i < cBindings;
i++, pchCurrent += lstrlen ( pchCurrent ) + 1
) {
WCHAR wszIpAddress[512];
DWORD dwPort;
hr = FromBindingString ( pchCurrent, wszIpAddress, &dwPort );
BAIL_ON_FAILURE(hr);
if ( fTcpBindings ) {
hr = pBindings->Add ( wszIpAddress, dwPort, 0 );
}
else {
hr = pBindings->Add ( wszIpAddress, 0, dwPort );
}
BAIL_ON_FAILURE(hr);
}
Exit:
return hr;
}
HRESULT IBindingsToMDBindings (
INntpServerBindings * pBindings,
BOOL fTcpBindings,
CMultiSz * pmsz
)
{
HRESULT hr = NOERROR;
long cBindings;
long i;
DWORD cbCount = 0;
LPWSTR wszBindings = NULL;
// Count the characters of the regular bindings list:
cbCount = 0;
pBindings->get_Count ( &cBindings );
for ( i = 0; i < cBindings; i++ ) {
CComPtr<INntpServerBinding> pBinding;
CComBSTR strIpAddress;
long lTcpPort;
long lSslPort;
hr = pBindings->Item ( i, &pBinding );
BAIL_ON_FAILURE(hr);
pBinding->get_IpAddress ( &strIpAddress );
pBinding->get_TcpPort ( &lTcpPort );
pBinding->get_SslPort ( &lSslPort );
if ( fTcpBindings ) {
if ( lTcpPort != 0 ) {
cbCount += CountBindingChars ( strIpAddress, lTcpPort );
}
}
else {
if ( lSslPort != 0 ) {
cbCount += CountBindingChars ( strIpAddress, lSslPort );
}
}
}
if ( cbCount == 0 ) {
cbCount = 2;
wszBindings = new WCHAR [ cbCount ];
if ( !wszBindings ) {
hr = E_OUTOFMEMORY;
goto Exit;
}
wszBindings[0] = NULL;
wszBindings[1] = NULL;
}
else {
cbCount++; // For double null terminator
wszBindings = new WCHAR [ cbCount ];
if ( !wszBindings ) {
BAIL_WITH_FAILURE(hr, E_OUTOFMEMORY);
}
LPWSTR pchCurrent = wszBindings;
for ( i = 0; i < cBindings; i++ ) {
CComPtr<INntpServerBinding> pBinding;
CComBSTR strIpAddress;
long lTcpPort;
long lSslPort;
hr = pBindings->Item ( i, &pBinding );
BAIL_ON_FAILURE(hr);
pBinding->get_IpAddress ( &strIpAddress );
pBinding->get_TcpPort ( &lTcpPort );
pBinding->get_SslPort ( &lSslPort );
if ( fTcpBindings ) {
if ( lTcpPort != 0 ) {
ToBindingString ( strIpAddress, lTcpPort, pchCurrent );
pchCurrent += lstrlen ( pchCurrent ) + 1;
}
}
else {
if ( lSslPort != 0 ) {
ToBindingString ( strIpAddress, lSslPort, pchCurrent );
pchCurrent += lstrlen ( pchCurrent ) + 1;
}
}
}
*pchCurrent = NULL;
}
_ASSERT ( wszBindings[cbCount - 1] == NULL );
_ASSERT ( wszBindings[cbCount - 2] == NULL );
pmsz->Attach ( wszBindings );
Exit:
return hr;
}
#if 0
DWORD CBinding::SizeInChars ( )
{
DWORD cchResult = 0;
WCHAR wszTcpPort [256];
wsprintf ( wszTcpPort, _T("%d"), m_dwTcpPort );
cchResult += lstrlen ( m_strIpAddress ); // <IPADDRESS>
cchResult += 1; // :
cchResult += lstrlen ( wszTcpPort ); // <TCPPORT>
cchResult += 1; // :
// cchResult += lstrlen ( m_strPathHeader ); // <PATHHEADER>
return cchResult;
}
void CBinding::ToString ( LPWSTR wszBinding )
{
wsprintf ( wszBinding, _T("%s:%d:"), m_strIpAddress, m_dwTcpPort );
}
HRESULT CBinding::FromString ( LPCWSTR wszBinding )
{
HRESULT hr = NOERROR;
LPWSTR pchFirstColon;
LPWSTR pchSecondColon;
WCHAR wszIpAddress [ 256 ];
WCHAR wszTcpPort [ 256 ];
long dwTcpPort;
LPWSTR pchColon;
wszIpAddress[0] = NULL;
wszTcpPort[0] = NULL;
pchFirstColon = wcschr ( wszBinding, _T(':') );
if ( pchFirstColon ) {
pchSecondColon = wcschr ( pchFirstColon + 1, _T(':') );
}
if ( !pchFirstColon || !pchSecondColon ) {
hr = E_FAIL;
goto Exit;
}
lstrcpyn ( wszIpAddress, wszBinding, 250 );
lstrcpyn ( wszTcpPort, pchFirstColon + 1, 250 );
// Get the TcpPort:
dwTcpPort = _wtoi ( wszTcpPort );
// Cutoff the IpAddress at the colon:
pchColon = wcschr ( wszIpAddress, _T(':') );
if ( pchColon ) {
*pchColon = NULL;
}
m_strIpAddress = wszIpAddress;
m_dwTcpPort = dwTcpPort;
if ( !m_strIpAddress ) {
hr = E_OUTOFMEMORY;
goto Exit;
}
Exit:
return hr;
}
HRESULT CNntpServerBindings::FromMultiSz ( CMultiSz * pmsz )
{
HRESULT hr;
DWORD cBindings;
DWORD i;
LPCWSTR pchCurrent;
CBinding binding;
hr = Clear ();
_ASSERT ( SUCCEEDED(hr) );
cBindings = pmsz->Count ();
for (
i = 0, pchCurrent = *pmsz;
i < cBindings;
i++, pchCurrent += lstrlen ( pchCurrent ) + 1
) {
hr = binding.FromString ( pchCurrent );
if ( FAILED(hr) ) {
if ( hr == E_FAIL ) {
// Skip the bad binding strings.
continue;
}
else {
goto Exit;
}
}
hr = Add ( binding.m_strIpAddress, binding.m_dwTcpPort );
if ( FAILED(hr) ) {
goto Exit;
}
}
Exit:
return hr;
}
HRESULT CNntpServerBindings::ToMultiSz ( CMultiSz * pmsz )
{
HRESULT hr = NOERROR;
DWORD cchSize;
long i;
LPWSTR wszBindings;
LPWSTR pchCurrent;
// Special case - the empty binding list:
if ( m_dwCount == 0 ) {
cchSize = 2;
wszBindings = new WCHAR [ cchSize ];
if ( !wszBindings ) {
hr = E_OUTOFMEMORY;
goto Exit;
}
wszBindings[0] = NULL;
wszBindings[1] = NULL;
}
else {
cchSize = 0;
for ( i = 0; i < m_dwCount; i++ ) {
cchSize += m_rgBindings[i].SizeInChars ( ) + 1;
}
// Add the size of the final terminator:
cchSize += 1;
wszBindings = new WCHAR [ cchSize ];
if ( !wszBindings ) {
hr = E_OUTOFMEMORY;
goto Exit;
}
for ( i = 0, pchCurrent = wszBindings; i < m_dwCount; i++ ) {
m_rgBindings[i].ToString ( pchCurrent );
pchCurrent += lstrlen ( pchCurrent ) + 1;
}
// Add the final NULL terminator:
*pchCurrent = NULL;
}
_ASSERT ( wszBindings[cchSize - 1] == NULL );
_ASSERT ( wszBindings[cchSize - 2] == NULL );
pmsz->Attach ( wszBindings );
_ASSERT ( pmsz->Count () == (DWORD) m_dwCount );
Exit:
return hr;
}
#endif