1578 lines
37 KiB
C++
1578 lines
37 KiB
C++
|
//////////////////////////////////////////////////////////////////////////////
|
||
|
//
|
||
|
// Copyright (c) 2000-2001 Microsoft Corporation
|
||
|
//
|
||
|
// Module Name:
|
||
|
// CEnumClusCfgIPAddresses.cpp
|
||
|
//
|
||
|
// Description:
|
||
|
// This file contains the definition of the CEnumClusCfgIPAddresses
|
||
|
// class.
|
||
|
//
|
||
|
// The class CEnumClusCfgIPAddresses is the enumeration of IP addresses.
|
||
|
// It implements the IEnumClusCfgIPAddresses interface.
|
||
|
//
|
||
|
// Documentation:
|
||
|
//
|
||
|
// Header File:
|
||
|
// CEnumClusCfgIPAddresses.h
|
||
|
//
|
||
|
// Maintained By:
|
||
|
// Galen Barbee (GalenB) 23-MAR-2000
|
||
|
//
|
||
|
//////////////////////////////////////////////////////////////////////////////
|
||
|
|
||
|
|
||
|
//////////////////////////////////////////////////////////////////////////////
|
||
|
// Include Files
|
||
|
//////////////////////////////////////////////////////////////////////////////
|
||
|
#include "pch.h"
|
||
|
#include "CEnumClusCfgIPAddresses.h"
|
||
|
#include "CClusCfgIPAddressInfo.h"
|
||
|
#include <ClusRtl.h>
|
||
|
#include <commctrl.h>
|
||
|
|
||
|
|
||
|
//////////////////////////////////////////////////////////////////////////////
|
||
|
// Constant Definitions
|
||
|
//////////////////////////////////////////////////////////////////////////////
|
||
|
|
||
|
DEFINE_THISCLASS( "CEnumClusCfgIPAddresses" );
|
||
|
|
||
|
|
||
|
//*************************************************************************//
|
||
|
|
||
|
|
||
|
/////////////////////////////////////////////////////////////////////////////
|
||
|
// CEnumClusCfgIPAddresses class
|
||
|
/////////////////////////////////////////////////////////////////////////////
|
||
|
|
||
|
|
||
|
//////////////////////////////////////////////////////////////////////////////
|
||
|
//++
|
||
|
//
|
||
|
// CEnumClusCfgIPAddresses::S_HrCreateInstance
|
||
|
//
|
||
|
// Description:
|
||
|
// Create a CEnumClusCfgIPAddresses instance.
|
||
|
//
|
||
|
// Arguments:
|
||
|
// None.
|
||
|
//
|
||
|
// Return Values:
|
||
|
// Pointer to CEnumClusCfgIPAddresses instance.
|
||
|
//
|
||
|
//--
|
||
|
//////////////////////////////////////////////////////////////////////////////
|
||
|
HRESULT
|
||
|
CEnumClusCfgIPAddresses::S_HrCreateInstance( IUnknown ** ppunkOut )
|
||
|
{
|
||
|
TraceFunc( "" );
|
||
|
|
||
|
HRESULT hr;
|
||
|
CEnumClusCfgIPAddresses * pccsd = NULL;
|
||
|
|
||
|
if ( ppunkOut == NULL )
|
||
|
{
|
||
|
hr = THR( E_POINTER );
|
||
|
goto Cleanup;
|
||
|
} // if:
|
||
|
|
||
|
pccsd = new CEnumClusCfgIPAddresses();
|
||
|
if ( pccsd == NULL )
|
||
|
{
|
||
|
hr = THR( E_OUTOFMEMORY );
|
||
|
goto Cleanup;
|
||
|
} // if: error allocating object
|
||
|
|
||
|
hr = THR( pccsd->HrInit() );
|
||
|
if ( FAILED( hr ) )
|
||
|
{
|
||
|
goto Cleanup;
|
||
|
} // if: HrInit() failed
|
||
|
|
||
|
hr = THR( pccsd->TypeSafeQI( IUnknown, ppunkOut ) );
|
||
|
|
||
|
Cleanup:
|
||
|
|
||
|
if ( FAILED( hr ) )
|
||
|
{
|
||
|
LogMsg( L"[SRV] CEnumClusCfgIPAddresses::S_HrCreateInstance() failed. (hr = %#08x)", hr );
|
||
|
} // if:
|
||
|
|
||
|
if ( pccsd != NULL )
|
||
|
{
|
||
|
pccsd->Release();
|
||
|
} // if:
|
||
|
|
||
|
HRETURN( hr );
|
||
|
|
||
|
} //*** CEnumClusCfgIPAddresses::S_HrCreateInstance
|
||
|
|
||
|
|
||
|
//////////////////////////////////////////////////////////////////////////////
|
||
|
//++
|
||
|
//
|
||
|
// CEnumClusCfgIPAddresses::S_HrCreateInstance
|
||
|
//
|
||
|
// Description:
|
||
|
// Create a CEnumClusCfgIPAddresses instance.
|
||
|
//
|
||
|
// Arguments:
|
||
|
// None.
|
||
|
//
|
||
|
// Return Values:
|
||
|
// Pointer to CEnumClusCfgIPAddresses instance.
|
||
|
//
|
||
|
//--
|
||
|
//////////////////////////////////////////////////////////////////////////////
|
||
|
HRESULT
|
||
|
CEnumClusCfgIPAddresses::S_HrCreateInstance(
|
||
|
ULONG ulIPAddressIn
|
||
|
, ULONG ulIPSubnetIn
|
||
|
, IUnknown * punkCallbackIn
|
||
|
, LCID lcidIn
|
||
|
, IWbemServices * pIWbemServicesIn
|
||
|
, IUnknown ** ppunkOut
|
||
|
)
|
||
|
{
|
||
|
TraceFunc( "" );
|
||
|
Assert( ppunkOut != NULL );
|
||
|
Assert( ulIPAddressIn != 0 );
|
||
|
Assert( ulIPSubnetIn != 0 );
|
||
|
|
||
|
HRESULT hr;
|
||
|
CEnumClusCfgIPAddresses * pccsd = NULL;
|
||
|
|
||
|
if ( ppunkOut == NULL )
|
||
|
{
|
||
|
hr = THR( E_POINTER );
|
||
|
goto Cleanup;
|
||
|
} // if:
|
||
|
|
||
|
pccsd = new CEnumClusCfgIPAddresses();
|
||
|
if ( pccsd == NULL )
|
||
|
{
|
||
|
hr = THR( E_OUTOFMEMORY );
|
||
|
goto Cleanup;
|
||
|
} // if: error allocating object
|
||
|
|
||
|
hr = THR( pccsd->Initialize( punkCallbackIn, lcidIn ) );
|
||
|
if ( FAILED( hr ) )
|
||
|
{
|
||
|
goto Cleanup;
|
||
|
} // if:
|
||
|
|
||
|
hr = THR( pccsd->SetWbemServices( pIWbemServicesIn ) );
|
||
|
if ( FAILED( hr ) )
|
||
|
{
|
||
|
goto Cleanup;
|
||
|
} // if:
|
||
|
|
||
|
hr = THR( pccsd->HrInit( ulIPAddressIn, ulIPSubnetIn ) );
|
||
|
if ( FAILED( hr ) )
|
||
|
{
|
||
|
goto Cleanup;
|
||
|
} // if: HrInit() failed
|
||
|
|
||
|
hr = THR( pccsd->TypeSafeQI( IUnknown, ppunkOut ) );
|
||
|
|
||
|
Cleanup:
|
||
|
|
||
|
if ( FAILED( hr ) )
|
||
|
{
|
||
|
LogMsg( L"[SRV] CEnumClusCfgIPAddresses::S_HrCreateInstance( ULONG, ULONG ) failed. (hr = %#08x)", hr );
|
||
|
} // if:
|
||
|
|
||
|
if ( pccsd != NULL )
|
||
|
{
|
||
|
pccsd->Release();
|
||
|
} // if:
|
||
|
|
||
|
HRETURN( hr );
|
||
|
|
||
|
} //*** CEnumClusCfgIPAddresses::S_HrCreateInstance
|
||
|
|
||
|
|
||
|
//////////////////////////////////////////////////////////////////////////////
|
||
|
//++
|
||
|
//
|
||
|
// CEnumClusCfgIPAddresses::CEnumClusCfgIPAddresses
|
||
|
//
|
||
|
// Description:
|
||
|
// Constructor of the CEnumClusCfgIPAddresses class. This initializes
|
||
|
// the m_cRef variable to 1 instead of 0 to account of possible
|
||
|
// QueryInterface failure in DllGetClassObject.
|
||
|
//
|
||
|
// Arguments:
|
||
|
// None.
|
||
|
//
|
||
|
// Return Value:
|
||
|
// None.
|
||
|
//
|
||
|
// Remarks:
|
||
|
// None.
|
||
|
//
|
||
|
//--
|
||
|
//////////////////////////////////////////////////////////////////////////////
|
||
|
CEnumClusCfgIPAddresses::CEnumClusCfgIPAddresses( void )
|
||
|
: m_cRef( 1 )
|
||
|
, m_lcid( LOCALE_NEUTRAL )
|
||
|
{
|
||
|
TraceFunc( "" );
|
||
|
|
||
|
Assert( m_idxNext == 0 );
|
||
|
Assert( m_idxEnumNext == 0 );
|
||
|
Assert( m_picccCallback == NULL );
|
||
|
Assert( m_pIWbemServices == NULL );
|
||
|
Assert( m_prgAddresses == NULL );
|
||
|
Assert( m_cAddresses == 0 );
|
||
|
|
||
|
// Increment the count of components in memory so the DLL hosting this
|
||
|
// object cannot be unloaded.
|
||
|
InterlockedIncrement( &g_cObjects );
|
||
|
|
||
|
TraceFuncExit();
|
||
|
|
||
|
} //*** CEnumClusCfgIPAddresses::CEnumClusCfgIPAddresses
|
||
|
|
||
|
|
||
|
//////////////////////////////////////////////////////////////////////////////
|
||
|
//++
|
||
|
//
|
||
|
// CEnumClusCfgIPAddresses::~CEnumClusCfgIPAddresses
|
||
|
//
|
||
|
// Description:
|
||
|
// Desstructor of the CEnumClusCfgIPAddresses class.
|
||
|
//
|
||
|
// Arguments:
|
||
|
// None.
|
||
|
//
|
||
|
// Return Value:
|
||
|
// None.
|
||
|
//
|
||
|
// Remarks:
|
||
|
// None.
|
||
|
//
|
||
|
//--
|
||
|
//////////////////////////////////////////////////////////////////////////////
|
||
|
CEnumClusCfgIPAddresses::~CEnumClusCfgIPAddresses( void )
|
||
|
{
|
||
|
TraceFunc( "" );
|
||
|
|
||
|
ULONG idx;
|
||
|
|
||
|
if ( m_pIWbemServices != NULL )
|
||
|
{
|
||
|
m_pIWbemServices->Release();
|
||
|
} // if:
|
||
|
|
||
|
if ( m_picccCallback != NULL )
|
||
|
{
|
||
|
m_picccCallback->Release();
|
||
|
} // if:
|
||
|
|
||
|
for ( idx = 0; idx < m_idxNext; idx++ )
|
||
|
{
|
||
|
((*m_prgAddresses)[ idx ])->Release();
|
||
|
} // for:
|
||
|
|
||
|
TraceFree( m_prgAddresses );
|
||
|
|
||
|
// There's going to be one less component in memory. Decrement component count.
|
||
|
InterlockedDecrement( &g_cObjects );
|
||
|
|
||
|
TraceFuncExit();
|
||
|
|
||
|
} //*** CEnumClusCfgIPAddresses::~CEnumClusCfgIPAddresses
|
||
|
|
||
|
|
||
|
//*************************************************************************//
|
||
|
|
||
|
|
||
|
/////////////////////////////////////////////////////////////////////////////
|
||
|
// CEnumClusCfgIPAddresses -- IUknkown interface.
|
||
|
/////////////////////////////////////////////////////////////////////////////
|
||
|
|
||
|
|
||
|
//////////////////////////////////////////////////////////////////////////////
|
||
|
//++
|
||
|
//
|
||
|
// STDMETHODIMP_( ULONG )
|
||
|
// CEnumClusCfgIPAddresses:: [IUNKNOWN] AddRef
|
||
|
//
|
||
|
// Description:
|
||
|
// Increment the reference count of this object by one.
|
||
|
//
|
||
|
// Arguments:
|
||
|
// None.
|
||
|
//
|
||
|
// Return Value:
|
||
|
// The new reference count.
|
||
|
//
|
||
|
// Remarks:
|
||
|
// None.
|
||
|
//
|
||
|
//--
|
||
|
//////////////////////////////////////////////////////////////////////////////
|
||
|
STDMETHODIMP_( ULONG )
|
||
|
CEnumClusCfgIPAddresses::AddRef( void )
|
||
|
{
|
||
|
TraceFunc( "[IUnknown]" );
|
||
|
|
||
|
InterlockedIncrement( & m_cRef );
|
||
|
|
||
|
RETURN( m_cRef );
|
||
|
|
||
|
} //*** CEnumClusCfgIPAddresses::AddRef
|
||
|
|
||
|
|
||
|
//////////////////////////////////////////////////////////////////////////////
|
||
|
//++
|
||
|
//
|
||
|
// STDMETHODIMP_( ULONG )
|
||
|
// CEnumClusCfgIPAddresses:: [IUNKNOWN] Release
|
||
|
//
|
||
|
// Description:
|
||
|
// Decrement the reference count of this object by one.
|
||
|
//
|
||
|
// Arguments:
|
||
|
// None.
|
||
|
//
|
||
|
// Return Value:
|
||
|
// The new reference count.
|
||
|
//
|
||
|
// Remarks:
|
||
|
// None.
|
||
|
//
|
||
|
//--
|
||
|
//////////////////////////////////////////////////////////////////////////////
|
||
|
STDMETHODIMP_( ULONG )
|
||
|
CEnumClusCfgIPAddresses::Release( void )
|
||
|
{
|
||
|
TraceFunc( "[IUnknown]" );
|
||
|
|
||
|
LONG cRef;
|
||
|
|
||
|
cRef = InterlockedDecrement( &m_cRef );
|
||
|
if ( cRef == 0 )
|
||
|
{
|
||
|
TraceDo( delete this );
|
||
|
} // if: reference count equal to zero
|
||
|
|
||
|
RETURN( cRef );
|
||
|
|
||
|
} //*** CEnumClusCfgIPAddresses::Release
|
||
|
|
||
|
|
||
|
//////////////////////////////////////////////////////////////////////////////
|
||
|
//++
|
||
|
//
|
||
|
// CEnumClusCfgIPAddresses:: [INKNOWN] QueryInterface
|
||
|
//
|
||
|
// Description:
|
||
|
// Query this object for the passed in interface.
|
||
|
//
|
||
|
// Arguments:
|
||
|
// IN REFIID riid,
|
||
|
// Id of interface requested.
|
||
|
//
|
||
|
// OUT void ** ppv
|
||
|
// Pointer to the requested interface.
|
||
|
//
|
||
|
// Return Value:
|
||
|
// S_OK
|
||
|
// If the interface is available on this object.
|
||
|
//
|
||
|
// E_NOINTERFACE
|
||
|
// If the interface is not available.
|
||
|
//
|
||
|
// Remarks:
|
||
|
// None.
|
||
|
//
|
||
|
//--
|
||
|
//////////////////////////////////////////////////////////////////////////////
|
||
|
STDMETHODIMP
|
||
|
CEnumClusCfgIPAddresses::QueryInterface( REFIID riid, void ** ppv )
|
||
|
{
|
||
|
TraceQIFunc( riid, ppv );
|
||
|
|
||
|
HRESULT hr = E_NOINTERFACE;
|
||
|
|
||
|
if ( IsEqualIID( riid, IID_IUnknown ) )
|
||
|
{
|
||
|
*ppv = static_cast< IEnumClusCfgIPAddresses * >( this );
|
||
|
hr = S_OK;
|
||
|
} // if: IUnknown
|
||
|
else if ( IsEqualIID( riid, IID_IEnumClusCfgIPAddresses ) )
|
||
|
{
|
||
|
*ppv = TraceInterface( __THISCLASS__, IEnumClusCfgIPAddresses, this, 0 );
|
||
|
hr = S_OK;
|
||
|
} // else if:
|
||
|
else if ( IsEqualIID( riid, IID_IClusCfgWbemServices ) )
|
||
|
{
|
||
|
*ppv = TraceInterface( __THISCLASS__, IClusCfgWbemServices, this, 0 );
|
||
|
hr = S_OK;
|
||
|
} // else if:
|
||
|
else if ( IsEqualIID( riid, IID_IClusCfgInitialize ) )
|
||
|
{
|
||
|
*ppv = TraceInterface( __THISCLASS__, IClusCfgInitialize, this, 0 );
|
||
|
hr = S_OK;
|
||
|
} // else if:
|
||
|
else if ( IsEqualIID( riid, IID_IClusCfgSetWbemObject ) )
|
||
|
{
|
||
|
*ppv = TraceInterface( __THISCLASS__, IClusCfgSetWbemObject, this, 0 );
|
||
|
hr = S_OK;
|
||
|
} // else if:
|
||
|
|
||
|
if ( SUCCEEDED( hr ) )
|
||
|
{
|
||
|
((IUnknown *) *ppv)->AddRef( );
|
||
|
} // if: success
|
||
|
|
||
|
QIRETURN( hr, riid );
|
||
|
|
||
|
} //*** CEnumClusCfgIPAddresses::QueryInterface
|
||
|
|
||
|
|
||
|
//*************************************************************************//
|
||
|
|
||
|
|
||
|
/////////////////////////////////////////////////////////////////////////////
|
||
|
// CEnumClusCfgIPAddresses -- IClusCfgWbemServices interface.
|
||
|
/////////////////////////////////////////////////////////////////////////////
|
||
|
|
||
|
|
||
|
//////////////////////////////////////////////////////////////////////////////
|
||
|
//++
|
||
|
//
|
||
|
// CEnumClusCfgIPAddresses::SetWbemServices
|
||
|
//
|
||
|
// Description:
|
||
|
// Set the WBEM services provider.
|
||
|
//
|
||
|
// Arguments:
|
||
|
// IN IWbemServices pIWbemServicesIn
|
||
|
//
|
||
|
// Return Value:
|
||
|
// S_OK
|
||
|
// Success
|
||
|
//
|
||
|
// E_POINTER
|
||
|
// The pIWbemServicesIn param is NULL.
|
||
|
//
|
||
|
// Remarks:
|
||
|
// None.
|
||
|
//
|
||
|
//--
|
||
|
//////////////////////////////////////////////////////////////////////////////
|
||
|
STDMETHODIMP
|
||
|
CEnumClusCfgIPAddresses::SetWbemServices( IWbemServices * pIWbemServicesIn )
|
||
|
{
|
||
|
TraceFunc( "[IClusCfgWbemServices]" );
|
||
|
|
||
|
HRESULT hr = S_OK;
|
||
|
|
||
|
if ( pIWbemServicesIn == NULL )
|
||
|
{
|
||
|
hr = THR( E_POINTER );
|
||
|
STATUS_REPORT( TASKID_Major_Find_Devices, TASKID_Minor_SetWbemServices_Enum_IPAddresses, IDS_ERROR_NULL_POINTER, hr );
|
||
|
goto Cleanup;
|
||
|
} // if:
|
||
|
|
||
|
m_pIWbemServices = pIWbemServicesIn;
|
||
|
m_pIWbemServices->AddRef();
|
||
|
|
||
|
Cleanup:
|
||
|
|
||
|
HRETURN( hr );
|
||
|
|
||
|
} //*** CEnumClusCfgIPAddresses::SetWbemServices
|
||
|
|
||
|
|
||
|
//*************************************************************************//
|
||
|
|
||
|
|
||
|
/////////////////////////////////////////////////////////////////////////////
|
||
|
// CEnumClusCfgIPAddresses -- IClusCfgInitialize interface.
|
||
|
/////////////////////////////////////////////////////////////////////////////
|
||
|
|
||
|
|
||
|
//////////////////////////////////////////////////////////////////////////////
|
||
|
//++
|
||
|
//
|
||
|
// CEnumClusCfgIPAddresses::Initialize
|
||
|
//
|
||
|
// Description:
|
||
|
// Initialize this component.
|
||
|
//
|
||
|
// Arguments:
|
||
|
// IN IUknown * punkCallbackIn
|
||
|
//
|
||
|
// IN LCID lcidIn
|
||
|
//
|
||
|
// Return Value:
|
||
|
// S_OK
|
||
|
// Success
|
||
|
//
|
||
|
// Remarks:
|
||
|
// None.
|
||
|
//
|
||
|
//--
|
||
|
//////////////////////////////////////////////////////////////////////////////
|
||
|
STDMETHODIMP
|
||
|
CEnumClusCfgIPAddresses::Initialize(
|
||
|
IUnknown * punkCallbackIn,
|
||
|
LCID lcidIn
|
||
|
)
|
||
|
{
|
||
|
TraceFunc( "[IClusCfgInitialize]" );
|
||
|
Assert( m_picccCallback == NULL );
|
||
|
|
||
|
HRESULT hr = S_OK;
|
||
|
|
||
|
m_lcid = lcidIn;
|
||
|
|
||
|
if ( punkCallbackIn == NULL )
|
||
|
{
|
||
|
hr = THR( E_POINTER );
|
||
|
goto Cleanup;
|
||
|
} // if:
|
||
|
|
||
|
hr = THR( punkCallbackIn->TypeSafeQI( IClusCfgCallback, &m_picccCallback ) );
|
||
|
|
||
|
Cleanup:
|
||
|
|
||
|
HRETURN( hr );
|
||
|
|
||
|
} //*** CEnumClusCfgIPAddresses::Initialize
|
||
|
|
||
|
|
||
|
//*************************************************************************//
|
||
|
|
||
|
|
||
|
/////////////////////////////////////////////////////////////////////////////
|
||
|
// CEnumClusCfgIPAddresses -- IEnumClusCfgIPAddresses interface.
|
||
|
/////////////////////////////////////////////////////////////////////////////
|
||
|
|
||
|
|
||
|
//////////////////////////////////////////////////////////////////////////////
|
||
|
//++
|
||
|
//
|
||
|
// CEnumClusCfgIPAddresses::Next
|
||
|
//
|
||
|
// Description:
|
||
|
//
|
||
|
// Arguments:
|
||
|
//
|
||
|
// Return Value:
|
||
|
//
|
||
|
// Remarks:
|
||
|
// None.
|
||
|
//
|
||
|
//--
|
||
|
//////////////////////////////////////////////////////////////////////////////
|
||
|
STDMETHODIMP
|
||
|
CEnumClusCfgIPAddresses::Next(
|
||
|
ULONG cNumberRequestedIn,
|
||
|
IClusCfgIPAddressInfo ** rgpIPAddressInfoOut,
|
||
|
ULONG * pcNumberFetchedOut
|
||
|
)
|
||
|
{
|
||
|
TraceFunc( "[IEnumClusCfgIPAddresses]" );
|
||
|
|
||
|
HRESULT hr = S_FALSE;
|
||
|
ULONG cFetched = 0;
|
||
|
ULONG idx;
|
||
|
IClusCfgIPAddressInfo * pccipai;
|
||
|
|
||
|
if ( rgpIPAddressInfoOut == NULL )
|
||
|
{
|
||
|
hr = THR( E_POINTER );
|
||
|
STATUS_REPORT( TASKID_Major_Find_Devices, TASKID_Minor_Next_Enum_IPAddresses, IDS_ERROR_NULL_POINTER, hr );
|
||
|
goto Cleanup;
|
||
|
} // if:
|
||
|
|
||
|
cFetched = min( cNumberRequestedIn, ( m_idxNext - m_idxEnumNext ) );
|
||
|
|
||
|
for ( idx = 0; idx < cFetched; idx++, m_idxEnumNext++ )
|
||
|
{
|
||
|
hr = THR( ((*m_prgAddresses)[ m_idxEnumNext ])->TypeSafeQI( IClusCfgIPAddressInfo, &pccipai ) );
|
||
|
if ( FAILED( hr ) )
|
||
|
{
|
||
|
break;
|
||
|
} // if:
|
||
|
|
||
|
rgpIPAddressInfoOut[ idx ] = pccipai;
|
||
|
} // for:
|
||
|
|
||
|
if ( FAILED( hr ) )
|
||
|
{
|
||
|
ULONG idxStop = idx;
|
||
|
|
||
|
m_idxEnumNext -= idx;
|
||
|
|
||
|
for ( idx = 0; idx < idxStop; idx++ )
|
||
|
{
|
||
|
(rgpIPAddressInfoOut[ idx ])->Release();
|
||
|
} // for:
|
||
|
|
||
|
cFetched = 0;
|
||
|
goto Cleanup;
|
||
|
} // if:
|
||
|
|
||
|
if ( cFetched < cNumberRequestedIn )
|
||
|
{
|
||
|
hr = S_FALSE;
|
||
|
} // if:
|
||
|
|
||
|
Cleanup:
|
||
|
|
||
|
if ( pcNumberFetchedOut != NULL )
|
||
|
{
|
||
|
*pcNumberFetchedOut = cFetched;
|
||
|
} // if:
|
||
|
|
||
|
HRETURN( hr );
|
||
|
|
||
|
} //*** CEnumClusCfgIPAddresses::Next
|
||
|
|
||
|
|
||
|
//////////////////////////////////////////////////////////////////////////////
|
||
|
//++
|
||
|
//
|
||
|
// CEnumClusCfgIPAddresses::Skip
|
||
|
//
|
||
|
// Description:
|
||
|
//
|
||
|
// Arguments:
|
||
|
//
|
||
|
// Return Value:
|
||
|
//
|
||
|
// Remarks:
|
||
|
// None.
|
||
|
//
|
||
|
//--
|
||
|
//////////////////////////////////////////////////////////////////////////////
|
||
|
STDMETHODIMP
|
||
|
CEnumClusCfgIPAddresses::Skip( ULONG cNumberToSkipIn )
|
||
|
{
|
||
|
TraceFunc( "[IEnumClusCfgIPAddresses]" );
|
||
|
|
||
|
HRESULT hr = S_OK;
|
||
|
|
||
|
m_idxEnumNext += cNumberToSkipIn;
|
||
|
if ( m_idxEnumNext >= m_idxNext )
|
||
|
{
|
||
|
m_idxEnumNext = m_idxNext;
|
||
|
hr = STHR( S_FALSE );
|
||
|
} // if:
|
||
|
|
||
|
HRETURN( hr );
|
||
|
|
||
|
} //*** CEnumClusCfgIPAddresses::Skip
|
||
|
|
||
|
|
||
|
//////////////////////////////////////////////////////////////////////////////
|
||
|
//++
|
||
|
//
|
||
|
// CEnumClusCfgIPAddresses::Reset
|
||
|
//
|
||
|
// Description:
|
||
|
//
|
||
|
// Arguments:
|
||
|
//
|
||
|
// Return Value:
|
||
|
//
|
||
|
// Remarks:
|
||
|
// None.
|
||
|
//
|
||
|
//--
|
||
|
//////////////////////////////////////////////////////////////////////////////
|
||
|
STDMETHODIMP
|
||
|
CEnumClusCfgIPAddresses::Reset( void )
|
||
|
{
|
||
|
TraceFunc( "[IEnumClusCfgIPAddresses]" );
|
||
|
|
||
|
HRESULT hr S_OK;
|
||
|
|
||
|
m_idxEnumNext = 0;
|
||
|
|
||
|
HRETURN( hr );
|
||
|
|
||
|
} //*** CEnumClusCfgIPAddresses::Reset
|
||
|
|
||
|
|
||
|
//////////////////////////////////////////////////////////////////////////////
|
||
|
//++
|
||
|
//
|
||
|
// CEnumClusCfgIPAddresses::Clone
|
||
|
//
|
||
|
// Description:
|
||
|
//
|
||
|
// Arguments:
|
||
|
//
|
||
|
// Return Value:
|
||
|
//
|
||
|
// Remarks:
|
||
|
// None.
|
||
|
//
|
||
|
//--
|
||
|
//////////////////////////////////////////////////////////////////////////////
|
||
|
STDMETHODIMP
|
||
|
CEnumClusCfgIPAddresses::Clone(
|
||
|
IEnumClusCfgIPAddresses ** ppEnumClusCfgIPAddressesOut
|
||
|
)
|
||
|
{
|
||
|
TraceFunc( "[IEnumClusCfgIPAddresses]" );
|
||
|
|
||
|
HRESULT hr = S_OK;
|
||
|
|
||
|
if ( ppEnumClusCfgIPAddressesOut == NULL )
|
||
|
{
|
||
|
hr = THR( E_POINTER );
|
||
|
STATUS_REPORT( TASKID_Major_Find_Devices, TASKID_Minor_Clone_Enum_IPAddresses, IDS_ERROR_NULL_POINTER, hr );
|
||
|
goto Cleanup;
|
||
|
} // if:
|
||
|
|
||
|
hr = THR( E_NOTIMPL );
|
||
|
|
||
|
Cleanup:
|
||
|
|
||
|
HRETURN( hr );
|
||
|
|
||
|
} //*** CEnumClusCfgIPAddresses::Clone
|
||
|
|
||
|
|
||
|
//////////////////////////////////////////////////////////////////////////////
|
||
|
//++
|
||
|
//
|
||
|
// CEnumClusCfgIPAddresses::Count
|
||
|
//
|
||
|
// Description:
|
||
|
//
|
||
|
// Arguments:
|
||
|
//
|
||
|
// Return Value:
|
||
|
//
|
||
|
// Remarks:
|
||
|
// None.
|
||
|
//
|
||
|
//--
|
||
|
//////////////////////////////////////////////////////////////////////////////
|
||
|
STDMETHODIMP
|
||
|
CEnumClusCfgIPAddresses::Count( DWORD * pnCountOut )
|
||
|
{
|
||
|
TraceFunc( "[IEnumClusCfgIPAddresses]" );
|
||
|
|
||
|
HRESULT hr = S_OK;
|
||
|
|
||
|
if ( pnCountOut == NULL )
|
||
|
{
|
||
|
hr = THR( E_POINTER );
|
||
|
goto Cleanup;
|
||
|
} // if:
|
||
|
|
||
|
*pnCountOut = m_cAddresses;
|
||
|
|
||
|
Cleanup:
|
||
|
|
||
|
HRETURN( hr );
|
||
|
|
||
|
} //*** CEnumClusCfgIPAddress::Count
|
||
|
|
||
|
|
||
|
//*************************************************************************//
|
||
|
|
||
|
|
||
|
/////////////////////////////////////////////////////////////////////////////
|
||
|
// CEnumClusCfgIPAddresses class -- IClusCfgNetworkAdapterInfo Interface
|
||
|
/////////////////////////////////////////////////////////////////////////////
|
||
|
|
||
|
|
||
|
//////////////////////////////////////////////////////////////////////////////
|
||
|
//++
|
||
|
//
|
||
|
// CEnumClusCfgIPAddresses::SetWbemObject
|
||
|
//
|
||
|
// Description:
|
||
|
// Get the configuration from the passed in adapter and load this
|
||
|
// enumerator.
|
||
|
//
|
||
|
// Arguments:
|
||
|
// None.
|
||
|
//
|
||
|
// Return Value:
|
||
|
//
|
||
|
//
|
||
|
// Remarks:
|
||
|
// None.
|
||
|
//
|
||
|
//--
|
||
|
//////////////////////////////////////////////////////////////////////////////
|
||
|
STDMETHODIMP
|
||
|
CEnumClusCfgIPAddresses::SetWbemObject(
|
||
|
IWbemClassObject * pNetworkAdapterIn
|
||
|
, bool * pfRetainObjectOut
|
||
|
)
|
||
|
{
|
||
|
TraceFunc( "[IClusCfgNetworkAdapterInfo]" );
|
||
|
Assert( pfRetainObjectOut != NULL );
|
||
|
|
||
|
HRESULT hr = E_INVALIDARG;
|
||
|
|
||
|
if ( pNetworkAdapterIn != NULL )
|
||
|
{
|
||
|
hr = STHR( HrGetAdapterConfiguration( pNetworkAdapterIn ) );
|
||
|
*pfRetainObjectOut = true;
|
||
|
} // if:
|
||
|
|
||
|
HRETURN( hr );
|
||
|
|
||
|
} //*** CEnumClusCfgIPAddresses::SetWbemObject
|
||
|
|
||
|
|
||
|
//*************************************************************************//
|
||
|
|
||
|
|
||
|
/////////////////////////////////////////////////////////////////////////////
|
||
|
// CEnumClusCfgIPAddresses class -- Private Methods.
|
||
|
/////////////////////////////////////////////////////////////////////////////
|
||
|
|
||
|
|
||
|
//////////////////////////////////////////////////////////////////////////////
|
||
|
//++
|
||
|
//
|
||
|
// CEnumClusCfgIPAddresses::HrInit
|
||
|
//
|
||
|
// Description:
|
||
|
// Initialize this component.
|
||
|
//
|
||
|
// Arguments:
|
||
|
// None.
|
||
|
//
|
||
|
// Return Value:
|
||
|
//
|
||
|
//
|
||
|
// Remarks:
|
||
|
// None.
|
||
|
//
|
||
|
//--
|
||
|
//////////////////////////////////////////////////////////////////////////////
|
||
|
HRESULT
|
||
|
CEnumClusCfgIPAddresses::HrInit( void )
|
||
|
{
|
||
|
TraceFunc( "" );
|
||
|
|
||
|
HRESULT hr = S_OK;
|
||
|
|
||
|
HRETURN( hr );
|
||
|
|
||
|
} //*** CEnumClusCfgIPAddresses::HrInit
|
||
|
|
||
|
|
||
|
//////////////////////////////////////////////////////////////////////////////
|
||
|
//++
|
||
|
//
|
||
|
// CEnumClusCfgIPAddresses::HrInit
|
||
|
//
|
||
|
// Description:
|
||
|
// Initialize this component.
|
||
|
//
|
||
|
// Arguments:
|
||
|
// None.
|
||
|
//
|
||
|
// Return Value:
|
||
|
//
|
||
|
//
|
||
|
// Remarks:
|
||
|
// None.
|
||
|
//
|
||
|
//--
|
||
|
//////////////////////////////////////////////////////////////////////////////
|
||
|
HRESULT
|
||
|
CEnumClusCfgIPAddresses::HrInit( ULONG ulIPAddressIn, ULONG ulIPSubnetIn )
|
||
|
{
|
||
|
TraceFunc( "" );
|
||
|
|
||
|
HRESULT hr;
|
||
|
IUnknown * punk = NULL;
|
||
|
|
||
|
hr = THR( HrCreateIPAddress( ulIPAddressIn, ulIPSubnetIn, &punk ) );
|
||
|
if ( FAILED( hr ) )
|
||
|
{
|
||
|
goto Cleanup;
|
||
|
} // if:
|
||
|
|
||
|
hr = THR( HrAddIPAddressToArray( punk ) );
|
||
|
|
||
|
Cleanup:
|
||
|
|
||
|
if ( punk != NULL )
|
||
|
{
|
||
|
punk->Release();
|
||
|
} // if:
|
||
|
|
||
|
HRETURN( hr );
|
||
|
|
||
|
} //*** CEnumClusCfgIPAddresses::HrInit
|
||
|
|
||
|
|
||
|
//////////////////////////////////////////////////////////////////////////////
|
||
|
//++
|
||
|
//
|
||
|
// CEnumClusCfgIPAddresses::HrGetAdapterConfiguration
|
||
|
//
|
||
|
// Description:
|
||
|
// Get the configuration of the passed in adapter.
|
||
|
//
|
||
|
// Arguments:
|
||
|
// None.
|
||
|
//
|
||
|
// Return Value:
|
||
|
//
|
||
|
//
|
||
|
// Remarks:
|
||
|
// None.
|
||
|
//
|
||
|
//--
|
||
|
//////////////////////////////////////////////////////////////////////////////
|
||
|
HRESULT
|
||
|
CEnumClusCfgIPAddresses::HrGetAdapterConfiguration(
|
||
|
IWbemClassObject * pNetworkAdapterIn
|
||
|
)
|
||
|
{
|
||
|
TraceFunc( "" );
|
||
|
|
||
|
HRESULT hr = S_OK;
|
||
|
BSTR bstrQuery = NULL;
|
||
|
BSTR bstrWQL = NULL;
|
||
|
VARIANT var;
|
||
|
WCHAR sz[ 256 ];
|
||
|
IEnumWbemClassObject * pConfigurations = NULL;
|
||
|
ULONG ulReturned;
|
||
|
IWbemClassObject * pConfiguration = NULL;
|
||
|
int cFound = 0;
|
||
|
BSTR bstrAdapterName = NULL;
|
||
|
int idx;
|
||
|
|
||
|
VariantInit( &var );
|
||
|
|
||
|
bstrWQL = TraceSysAllocString( L"WQL" );
|
||
|
if ( bstrWQL == NULL )
|
||
|
{
|
||
|
goto OutOfMemory;
|
||
|
} // if:
|
||
|
|
||
|
hr = THR( HrGetWMIProperty( pNetworkAdapterIn, L"DeviceID", VT_BSTR, &var ) );
|
||
|
if ( FAILED( hr ) )
|
||
|
{
|
||
|
goto Cleanup;
|
||
|
} // if:
|
||
|
|
||
|
_snwprintf( sz, ARRAYSIZE( sz ), L"Associators of {Win32_NetworkAdapter.DeviceID='%s'} where AssocClass=Win32_NetworkAdapterSetting", var.bstrVal );
|
||
|
|
||
|
bstrQuery = TraceSysAllocString( sz );
|
||
|
if ( bstrQuery == NULL )
|
||
|
{
|
||
|
goto OutOfMemory;
|
||
|
} // if:
|
||
|
|
||
|
VariantClear( &var );
|
||
|
|
||
|
hr = THR( HrGetWMIProperty( pNetworkAdapterIn, L"NetConnectionID", VT_BSTR, &var ) );
|
||
|
if ( FAILED( hr ) )
|
||
|
{
|
||
|
goto Cleanup;
|
||
|
} // if:
|
||
|
|
||
|
bstrAdapterName = TraceSysAllocString( var.bstrVal );
|
||
|
if ( bstrAdapterName == NULL )
|
||
|
{
|
||
|
goto OutOfMemory;
|
||
|
} // if:
|
||
|
|
||
|
hr = THR( m_pIWbemServices->ExecQuery( bstrWQL, bstrQuery, WBEM_FLAG_FORWARD_ONLY, NULL, &pConfigurations ) );
|
||
|
if ( FAILED( hr ) )
|
||
|
{
|
||
|
STATUS_REPORT_STRING(
|
||
|
TASKID_Major_Find_Devices,
|
||
|
TASKID_Minor_WMI_NetworkAdapterSetting_Qry_Failed,
|
||
|
IDS_ERROR_WMI_NETWORKADAPTERSETTINGS_QRY_FAILED,
|
||
|
bstrAdapterName,
|
||
|
hr
|
||
|
);
|
||
|
goto Cleanup;
|
||
|
} // if:
|
||
|
|
||
|
for ( idx = 0; ; idx++ )
|
||
|
{
|
||
|
hr = pConfigurations->Next( WBEM_INFINITE, 1, &pConfiguration, &ulReturned );
|
||
|
if ( ( hr == S_OK ) && ( ulReturned == 1 ) )
|
||
|
{
|
||
|
//
|
||
|
// KB: 25-AUG-2000 GalenB
|
||
|
//
|
||
|
// WMI only supports one configuration per adapter!
|
||
|
//
|
||
|
Assert( idx < 1 );
|
||
|
|
||
|
VariantClear( &var );
|
||
|
|
||
|
hr = THR( HrGetWMIProperty( pConfiguration, L"IPEnabled", VT_BOOL, &var ) );
|
||
|
if ( FAILED( hr ) )
|
||
|
{
|
||
|
goto Cleanup;
|
||
|
} // if:
|
||
|
|
||
|
//
|
||
|
// If this configuration is not for TCP/IP then skip it.
|
||
|
//
|
||
|
if ( ( var.vt != VT_BOOL ) || ( var.boolVal != VARIANT_TRUE ) )
|
||
|
{
|
||
|
hr = S_FALSE;
|
||
|
STATUS_REPORT_STRING( TASKID_Major_Find_Devices, TASKID_Minor_Non_Tcp_Config, IDS_WARNING__NON_TCP_CONFIG, bstrAdapterName, hr );
|
||
|
continue;
|
||
|
} // if:
|
||
|
|
||
|
hr = STHR( HrSaveIPAddresses( bstrAdapterName, pConfiguration ) );
|
||
|
if ( FAILED( hr ) )
|
||
|
{
|
||
|
goto Cleanup;
|
||
|
} // if:
|
||
|
|
||
|
//
|
||
|
// KB: 24-AUG-2000 GalenB
|
||
|
//
|
||
|
// If any configuration returns S_FALSE then we skip.
|
||
|
//
|
||
|
if ( hr == S_FALSE )
|
||
|
{
|
||
|
pConfiguration->Release();
|
||
|
pConfiguration = NULL;
|
||
|
continue;
|
||
|
} // if:
|
||
|
|
||
|
cFound++;
|
||
|
pConfiguration->Release();
|
||
|
pConfiguration = NULL;
|
||
|
} // if:
|
||
|
else if ( ( hr == S_FALSE ) && ( ulReturned == 0 ) )
|
||
|
{
|
||
|
hr = S_OK;
|
||
|
break;
|
||
|
} // else if:
|
||
|
else
|
||
|
{
|
||
|
STATUS_REPORT_STRING( TASKID_Major_Find_Devices, TASKID_Minor_WQL_Qry_Next_Failed, IDS_ERROR_WQL_QRY_NEXT_FAILED, bstrQuery, hr );
|
||
|
goto Cleanup;
|
||
|
} // else:
|
||
|
} // for:
|
||
|
|
||
|
//
|
||
|
// If we didn't find any valid configurations then we should return S_FALSE
|
||
|
// to tell the caller to ingore that adpater.
|
||
|
//
|
||
|
if ( cFound == 0 )
|
||
|
{
|
||
|
hr = S_FALSE;
|
||
|
STATUS_REPORT_STRING( TASKID_Major_Find_Devices, TASKID_Minor_No_Valid_TCP_Configs, IDS_WARNING_NO_VALID_TCP_CONFIGS, bstrAdapterName, hr );
|
||
|
} // if:
|
||
|
|
||
|
goto Cleanup;
|
||
|
|
||
|
OutOfMemory:
|
||
|
|
||
|
hr = THR( E_OUTOFMEMORY );
|
||
|
STATUS_REPORT( TASKID_Major_Find_Devices, TASKID_Minor_HrGetAdapterConfiguration, IDS_ERROR_OUTOFMEMORY, hr );
|
||
|
|
||
|
Cleanup:
|
||
|
|
||
|
VariantClear( &var );
|
||
|
|
||
|
TraceSysFreeString( bstrQuery );
|
||
|
TraceSysFreeString( bstrWQL );
|
||
|
TraceSysFreeString( bstrAdapterName );
|
||
|
|
||
|
if ( pConfiguration != NULL )
|
||
|
{
|
||
|
pConfiguration->Release();
|
||
|
} // if:
|
||
|
|
||
|
if ( pConfigurations != NULL )
|
||
|
{
|
||
|
pConfigurations->Release();
|
||
|
} // if:
|
||
|
|
||
|
HRETURN( hr );
|
||
|
|
||
|
} //*** CEnumClusCfgIPAddresses::HrGetAdapterConfiguration
|
||
|
|
||
|
|
||
|
/////////////////////////////////////////////////////////////////////////////
|
||
|
//++
|
||
|
//
|
||
|
// CEnumClusCfgIPAddresses:HrAddIPAddressToArray
|
||
|
//
|
||
|
// Description:
|
||
|
// Add the passed in address to the array of address.
|
||
|
//
|
||
|
// Arguments:
|
||
|
//
|
||
|
//
|
||
|
// Return Value:
|
||
|
// S_OK
|
||
|
// Success
|
||
|
//
|
||
|
// E_OUTOFMEMORY
|
||
|
// Couldn't allocate memeory.
|
||
|
//
|
||
|
// Remarks:
|
||
|
// None.
|
||
|
//
|
||
|
//--
|
||
|
//////////////////////////////////////////////////////////////////////////////
|
||
|
HRESULT
|
||
|
CEnumClusCfgIPAddresses::HrAddIPAddressToArray( IUnknown * punkIn )
|
||
|
{
|
||
|
TraceFunc( "" );
|
||
|
|
||
|
HRESULT hr = S_OK;
|
||
|
IUnknown * ((*prgpunks)[]) = NULL;
|
||
|
|
||
|
prgpunks = (IUnknown *((*)[])) TraceReAlloc( m_prgAddresses, sizeof( IUnknown * ) * ( m_idxNext + 1 ), HEAP_ZERO_MEMORY );
|
||
|
if ( prgpunks == NULL )
|
||
|
{
|
||
|
hr = THR( E_OUTOFMEMORY );
|
||
|
STATUS_REPORT( TASKID_Major_Find_Devices, TASKID_Minor_HrAddIPAddressToArray, IDS_ERROR_OUTOFMEMORY, hr );
|
||
|
goto Cleanup;
|
||
|
} // if:
|
||
|
|
||
|
m_prgAddresses = prgpunks;
|
||
|
|
||
|
(*m_prgAddresses)[ m_idxNext++ ] = punkIn;
|
||
|
punkIn->AddRef();
|
||
|
m_cAddresses += 1;
|
||
|
|
||
|
Cleanup:
|
||
|
|
||
|
HRETURN( hr );
|
||
|
|
||
|
} //*** CEnumClusCfgIPAddresses::HrAddIPAddressToArray
|
||
|
|
||
|
//////////////////////////////////////////////////////////////////////////////
|
||
|
//++
|
||
|
//
|
||
|
// CEnumClusCfgIPAddresses::HrMakeDottedQuad
|
||
|
//
|
||
|
// Description:
|
||
|
// Take the passed in IP address and convert it into a dotted quad.
|
||
|
//
|
||
|
// Arguments:
|
||
|
// None.
|
||
|
//
|
||
|
// Return Value:
|
||
|
//
|
||
|
//
|
||
|
// Remarks:
|
||
|
// Internet Addresses
|
||
|
// Values specified using the ".'' notation take one of the following forms:
|
||
|
//
|
||
|
// a.b.c.d a.b.c a.b a
|
||
|
//
|
||
|
// When four parts are specified, each is interpreted as a byte of data and
|
||
|
// assigned, from left to right, to the 4 bytes of an Internet address. When
|
||
|
// an Internet address is viewed as a 32-bit integer quantity on the
|
||
|
// Intel architecture, the bytes referred to above appear as "d.c.b.a''.
|
||
|
// That is, the bytes on an Intel processor are ordered from right to left.
|
||
|
//
|
||
|
// The parts that make up an address in "." notation can be decimal, octal
|
||
|
// or hexadecimal as specified in the C language. Numbers that start
|
||
|
// with "0x" or "0X" imply hexadecimal. Numbers that start with "0" imply
|
||
|
// octal. All other numbers are interpreted as decimal.
|
||
|
//
|
||
|
//--
|
||
|
//////////////////////////////////////////////////////////////////////////////
|
||
|
HRESULT
|
||
|
CEnumClusCfgIPAddresses::HrMakeDottedQuad(
|
||
|
BSTR bstrDottedQuadIn,
|
||
|
ULONG * pulDottedQuadOut
|
||
|
)
|
||
|
{
|
||
|
TraceFunc( "" );
|
||
|
|
||
|
HRESULT hr = S_OK;
|
||
|
DWORD sc;
|
||
|
|
||
|
if ( pulDottedQuadOut == NULL )
|
||
|
{
|
||
|
hr = THR( E_POINTER );
|
||
|
STATUS_REPORT( TASKID_Major_Find_Devices, TASKID_Minor_HrMakeDottedQuad, IDS_ERROR_NULL_POINTER, hr );
|
||
|
goto Cleanup;
|
||
|
} // if:
|
||
|
|
||
|
if ( bstrDottedQuadIn == NULL )
|
||
|
{
|
||
|
hr = THR( E_INVALIDARG );
|
||
|
LOG_STATUS_REPORT( L"CEnumClusCfgIPAddresses::HrMakeDottedQuad() was given an invalid argument.", hr );
|
||
|
goto Cleanup;
|
||
|
} // if:
|
||
|
|
||
|
sc = TW32( ClRtlTcpipStringToAddress( bstrDottedQuadIn, pulDottedQuadOut ) );
|
||
|
if ( sc != ERROR_SUCCESS )
|
||
|
{
|
||
|
hr = HRESULT_FROM_WIN32( sc );
|
||
|
STATUS_REPORT_STRING( TASKID_Major_Find_Devices, TASKID_Minor_Dotted_Quad_Failed, IDS_ERROR_CONVERT_TO_DOTTED_QUAD_FAILED, bstrDottedQuadIn, hr );
|
||
|
goto Cleanup;
|
||
|
} // if:
|
||
|
|
||
|
Cleanup:
|
||
|
|
||
|
RETURN( hr );
|
||
|
|
||
|
} //*** CEnumClusCfgIPAddresses::HrMakeDottedQuad
|
||
|
|
||
|
|
||
|
//////////////////////////////////////////////////////////////////////////////
|
||
|
//++
|
||
|
//
|
||
|
// CEnumClusCfgIPAddresses::HrSaveIPAddresses
|
||
|
//
|
||
|
// Description:
|
||
|
// Add the IP addresses to the array.
|
||
|
//
|
||
|
// Arguments:
|
||
|
//
|
||
|
//
|
||
|
// Return Value:
|
||
|
//
|
||
|
//
|
||
|
// Remarks:
|
||
|
// None.
|
||
|
//
|
||
|
//--
|
||
|
//////////////////////////////////////////////////////////////////////////////
|
||
|
HRESULT
|
||
|
CEnumClusCfgIPAddresses::HrSaveIPAddresses(
|
||
|
BSTR bstrAdapterNameIn,
|
||
|
IWbemClassObject * pConfigurationIn
|
||
|
)
|
||
|
{
|
||
|
TraceFunc( "" );
|
||
|
Assert( bstrAdapterNameIn != NULL );
|
||
|
Assert( pConfigurationIn != NULL );
|
||
|
|
||
|
HRESULT hr;
|
||
|
VARIANT varIPAddress;
|
||
|
VARIANT varIPSubnet;
|
||
|
long lIPAddressesUpperBound;
|
||
|
long lIPAddressesLowerBound;
|
||
|
long lIPSubnetsUpperBound;
|
||
|
long lIPSubnetsLowerBound;
|
||
|
long idx;
|
||
|
ULONG ulIPAddress;
|
||
|
ULONG ulIPSubnet;
|
||
|
BSTR bstrIPAddress = NULL;
|
||
|
BSTR bstrIPSubnet = NULL;
|
||
|
IUnknown * punk = NULL;
|
||
|
IClusCfgIPAddressInfo * piccipai = NULL;
|
||
|
|
||
|
VariantInit( &varIPAddress );
|
||
|
VariantInit( &varIPSubnet );
|
||
|
|
||
|
hr = THR( HrGetWMIProperty( pConfigurationIn, L"IPAddress", ( VT_ARRAY | VT_BSTR ), &varIPAddress ) );
|
||
|
if ( FAILED( hr ) )
|
||
|
{
|
||
|
goto Cleanup;
|
||
|
} // if:
|
||
|
|
||
|
hr = THR( HrGetWMIProperty( pConfigurationIn, L"IPSubnet", ( VT_ARRAY | VT_BSTR ), &varIPSubnet ) );
|
||
|
if ( FAILED( hr ) )
|
||
|
{
|
||
|
goto Cleanup;
|
||
|
} // if:
|
||
|
|
||
|
hr = THR( SafeArrayGetUBound( varIPAddress.parray, 1, &lIPAddressesUpperBound ) );
|
||
|
if ( FAILED( hr ) )
|
||
|
{
|
||
|
goto Cleanup;
|
||
|
} // if:
|
||
|
|
||
|
hr = THR( SafeArrayGetUBound( varIPSubnet.parray, 1, &lIPSubnetsUpperBound ) );
|
||
|
if ( FAILED( hr ) )
|
||
|
{
|
||
|
goto Cleanup;
|
||
|
} // if:
|
||
|
|
||
|
Assert( lIPAddressesUpperBound == lIPSubnetsUpperBound );
|
||
|
if ( lIPAddressesUpperBound != lIPSubnetsUpperBound )
|
||
|
{
|
||
|
hr = S_FALSE;
|
||
|
STATUS_REPORT( TASKID_Major_Find_Devices, TASKID_Minor_IP_Address_Subnet_Count_Unequal, IDS_ERROR_IP_ADDRESS_SUBNET_COUNT_UNEQUAL, hr );
|
||
|
goto Cleanup;
|
||
|
} // if:
|
||
|
|
||
|
hr = THR( SafeArrayGetLBound( varIPAddress.parray, 1, &lIPAddressesLowerBound ) );
|
||
|
if ( FAILED( hr ) )
|
||
|
{
|
||
|
goto Cleanup;
|
||
|
} // if:
|
||
|
|
||
|
hr = THR( SafeArrayGetLBound( varIPSubnet.parray, 1, &lIPSubnetsLowerBound ) );
|
||
|
if ( FAILED( hr ) )
|
||
|
{
|
||
|
goto Cleanup;
|
||
|
} // if:
|
||
|
|
||
|
Assert( lIPAddressesLowerBound == lIPSubnetsLowerBound );
|
||
|
if ( lIPAddressesLowerBound != lIPSubnetsLowerBound )
|
||
|
{
|
||
|
hr = S_FALSE;
|
||
|
STATUS_REPORT( TASKID_Major_Find_Devices, TASKID_Minor_IP_Address_Subnet_Count_Unequal, IDS_ERROR_IP_ADDRESS_SUBNET_COUNT_UNEQUAL, hr );
|
||
|
goto Cleanup;
|
||
|
} // if:
|
||
|
|
||
|
for ( idx = lIPAddressesLowerBound; idx <= lIPAddressesUpperBound; idx++ )
|
||
|
{
|
||
|
hr = THR( SafeArrayGetElement( varIPAddress.parray, &idx, &bstrIPAddress ) );
|
||
|
if ( FAILED( hr ) )
|
||
|
{
|
||
|
goto Cleanup;
|
||
|
} // if:
|
||
|
|
||
|
TraceMemoryAddBSTR( bstrIPAddress );
|
||
|
|
||
|
if ( wcscmp( bstrIPAddress, L"" ) == 0 )
|
||
|
{
|
||
|
STATUS_REPORT_STRING( TASKID_Major_Find_Devices, TASKID_Minor_Warning_No_IP_Addresses, IDS_WARNING_NO_IP_ADDRESSES, bstrAdapterNameIn, hr );
|
||
|
goto Cleanup;
|
||
|
} // end if:
|
||
|
|
||
|
hr = THR( SafeArrayGetElement( varIPSubnet.parray, &idx, &bstrIPSubnet ) );
|
||
|
if ( FAILED( hr ) )
|
||
|
{
|
||
|
goto Cleanup;
|
||
|
} // if:
|
||
|
|
||
|
Assert( bstrIPAddress != NULL );
|
||
|
Assert( wcslen( bstrIPAddress ) > 0 );
|
||
|
|
||
|
TraceMemoryAddBSTR( bstrIPSubnet );
|
||
|
|
||
|
LOG_STATUS_REPORT_STRING2( L"Found IP Address '%1!ws!' with subnet mask '%2!ws!'." , bstrIPAddress, bstrIPSubnet, hr );
|
||
|
|
||
|
hr = THR( HrMakeDottedQuad( bstrIPAddress, &ulIPAddress ) );
|
||
|
if ( FAILED( hr ) )
|
||
|
{
|
||
|
goto Cleanup;
|
||
|
} // if:
|
||
|
|
||
|
hr = THR( HrMakeDottedQuad( bstrIPSubnet, &ulIPSubnet ) );
|
||
|
if ( FAILED( hr ) )
|
||
|
{
|
||
|
goto Cleanup;
|
||
|
} // if:
|
||
|
|
||
|
TraceSysFreeString( bstrIPAddress );
|
||
|
bstrIPAddress = NULL;
|
||
|
|
||
|
TraceSysFreeString( bstrIPSubnet );
|
||
|
bstrIPSubnet = NULL;
|
||
|
|
||
|
hr = THR( HrCreateIPAddress( &punk ) );
|
||
|
if ( FAILED( hr ) )
|
||
|
{
|
||
|
goto Cleanup;
|
||
|
} // if:
|
||
|
|
||
|
hr = THR( punk->TypeSafeQI( IClusCfgIPAddressInfo, &piccipai ) );
|
||
|
if ( FAILED( hr ) )
|
||
|
{
|
||
|
goto Cleanup;
|
||
|
} // if:
|
||
|
|
||
|
hr = THR( piccipai->SetIPAddress( ulIPAddress ) );
|
||
|
if ( FAILED( hr ) )
|
||
|
{
|
||
|
goto Cleanup;
|
||
|
} // if:
|
||
|
|
||
|
hr = THR( piccipai->SetSubnetMask( ulIPSubnet ) );
|
||
|
if ( FAILED( hr ) )
|
||
|
{
|
||
|
goto Cleanup;
|
||
|
} // if:
|
||
|
|
||
|
piccipai->Release();
|
||
|
piccipai = NULL;
|
||
|
|
||
|
hr = THR( HrAddIPAddressToArray( punk ) );
|
||
|
if ( FAILED( hr ) )
|
||
|
{
|
||
|
goto Cleanup;
|
||
|
} // if:
|
||
|
|
||
|
punk->Release();
|
||
|
punk = NULL;
|
||
|
} // for:
|
||
|
|
||
|
Cleanup:
|
||
|
|
||
|
if ( piccipai != NULL )
|
||
|
{
|
||
|
piccipai->Release();
|
||
|
} // if:
|
||
|
|
||
|
if ( punk != NULL )
|
||
|
{
|
||
|
punk->Release();
|
||
|
} // if:
|
||
|
|
||
|
TraceSysFreeString( bstrIPAddress );
|
||
|
TraceSysFreeString( bstrIPSubnet );
|
||
|
|
||
|
VariantClear( &varIPAddress );
|
||
|
VariantClear( &varIPSubnet );
|
||
|
|
||
|
HRETURN( hr );
|
||
|
|
||
|
} //*** CEnumClusCfgIPAddresses::HrSaveIPAddresses
|
||
|
|
||
|
|
||
|
/////////////////////////////////////////////////////////////////////////////
|
||
|
//++
|
||
|
//
|
||
|
// CEnumClusCfgIPAddresses:HrCreateIPAddress
|
||
|
//
|
||
|
// Description:
|
||
|
//
|
||
|
//
|
||
|
// Arguments:
|
||
|
//
|
||
|
//
|
||
|
// Return Value:
|
||
|
// S_OK
|
||
|
// Success
|
||
|
//
|
||
|
// E_OUTOFMEMORY
|
||
|
// Couldn't allocate memeory.
|
||
|
//
|
||
|
// Remarks:
|
||
|
// None.
|
||
|
//
|
||
|
//--
|
||
|
//////////////////////////////////////////////////////////////////////////////
|
||
|
HRESULT
|
||
|
CEnumClusCfgIPAddresses::HrCreateIPAddress( IUnknown ** ppunkOut )
|
||
|
{
|
||
|
TraceFunc( "" );
|
||
|
Assert( ppunkOut != NULL );
|
||
|
|
||
|
HRESULT hr;
|
||
|
IUnknown * punk = NULL;
|
||
|
|
||
|
hr = THR( CClusCfgIPAddressInfo::S_HrCreateInstance( &punk ) );
|
||
|
if ( FAILED( hr ) )
|
||
|
{
|
||
|
goto Cleanup;
|
||
|
} // if:
|
||
|
|
||
|
punk = TraceInterface( L"CClusCfgIPAddressInfo", IUnknown, punk, 1 );
|
||
|
|
||
|
hr = THR( HrSetInitialize( punk, m_picccCallback, m_lcid ) );
|
||
|
if ( FAILED( hr ) )
|
||
|
{
|
||
|
goto Cleanup;
|
||
|
} // if:
|
||
|
|
||
|
hr = THR( HrSetWbemServices( punk, m_pIWbemServices ) );
|
||
|
if ( FAILED( hr ) )
|
||
|
{
|
||
|
goto Cleanup;
|
||
|
} // if:
|
||
|
|
||
|
*ppunkOut = punk;
|
||
|
(*ppunkOut)->AddRef();
|
||
|
|
||
|
Cleanup:
|
||
|
|
||
|
if ( punk != NULL )
|
||
|
{
|
||
|
punk->Release();
|
||
|
} // if:
|
||
|
|
||
|
HRETURN( hr );
|
||
|
|
||
|
} //*** CEnumClusCfgIPAddresses::HrCreateIPAddress
|
||
|
|
||
|
|
||
|
/////////////////////////////////////////////////////////////////////////////
|
||
|
//++
|
||
|
//
|
||
|
// CEnumClusCfgIPAddresses:HrCreateIPAddress
|
||
|
//
|
||
|
// Description:
|
||
|
//
|
||
|
//
|
||
|
// Arguments:
|
||
|
//
|
||
|
//
|
||
|
// Return Value:
|
||
|
// S_OK
|
||
|
// Success
|
||
|
//
|
||
|
// E_OUTOFMEMORY
|
||
|
// Couldn't allocate memeory.
|
||
|
//
|
||
|
// Remarks:
|
||
|
// None.
|
||
|
//
|
||
|
//--
|
||
|
//////////////////////////////////////////////////////////////////////////////
|
||
|
HRESULT
|
||
|
CEnumClusCfgIPAddresses::HrCreateIPAddress(
|
||
|
ULONG ulIPAddressIn
|
||
|
, ULONG ulIPSubnetIn
|
||
|
, IUnknown ** ppunkOut
|
||
|
)
|
||
|
{
|
||
|
TraceFunc( "" );
|
||
|
Assert( ppunkOut != NULL );
|
||
|
|
||
|
HRESULT hr;
|
||
|
IUnknown * punk = NULL;
|
||
|
|
||
|
hr = THR( CClusCfgIPAddressInfo::S_HrCreateInstance( ulIPAddressIn, ulIPSubnetIn, &punk ) );
|
||
|
if ( FAILED( hr ) )
|
||
|
{
|
||
|
goto Cleanup;
|
||
|
} // if:
|
||
|
|
||
|
punk = TraceInterface( L"CClusCfgIPAddressInfo", IUnknown, punk, 1 );
|
||
|
|
||
|
hr = THR( HrSetInitialize( punk, m_picccCallback, m_lcid ) );
|
||
|
if ( FAILED( hr ) )
|
||
|
{
|
||
|
goto Cleanup;
|
||
|
} // if:
|
||
|
|
||
|
hr = THR( HrSetWbemServices( punk, m_pIWbemServices ) );
|
||
|
if ( FAILED( hr ) )
|
||
|
{
|
||
|
goto Cleanup;
|
||
|
} // if:
|
||
|
|
||
|
*ppunkOut = punk;
|
||
|
(*ppunkOut)->AddRef();
|
||
|
|
||
|
Cleanup:
|
||
|
|
||
|
if ( punk != NULL )
|
||
|
{
|
||
|
punk->Release();
|
||
|
} // if:
|
||
|
|
||
|
HRETURN( hr );
|
||
|
|
||
|
} //*** CEnumClusCfgIPAddresses::HrCreateIPAddress
|