windows-nt/Source/XPSP1/NT/net/ias/mmc/common/sdohelperfuncs.cpp

746 lines
15 KiB
C++
Raw Normal View History

2020-09-26 03:20:57 -05:00
/////////////////////////////////////////////////////////////////////////////
//
// Copyright(C) Microsoft Corporation, 1997 - 1999 all rights reserved.
//
// Module: sdohelperfuncs.cpp
//
// Project: Everest
//
// Description: Helper Functions
//
// Log:
//
// When Who What
// ---- --- ----
// 6/08/98 TLP Initial Version
// 7/03/98 MAM Adapted from \ias\sdo\sdoias to use in UI
// 11/03/98 MAM Moved GetSdo/PutSdo routines here from mmcutility.cpp
//
/////////////////////////////////////////////////////////////////////////////
#include "sdohelperfuncs.h"
#include "comdef.h"
//////////////////////////////////////////////////////////////////////////////
// CORE HELPER FUNCTIONS
//////////////////////////////////////////////////////////////////////////////
///////////////////////////////////////////////////////////////////
HRESULT SDOGetCollectionEnumerator(
ISdo* pSdo,
LONG lPropertyId,
IEnumVARIANT** ppEnum
)
{
HRESULT hr;
CComPtr<IUnknown> pUnknown;
CComPtr<ISdoCollection> pSdoCollection;
_variant_t vtDispatch;
_ASSERT( NULL != pSdo && NULL == *ppEnum );
hr = pSdo->GetProperty(lPropertyId, &vtDispatch);
_ASSERT( VT_DISPATCH == V_VT(&vtDispatch) );
if ( SUCCEEDED(hr) )
{
hr = vtDispatch.pdispVal->QueryInterface(IID_ISdoCollection, (void**)&pSdoCollection);
if ( SUCCEEDED(hr) )
{
hr = pSdoCollection->get__NewEnum(&pUnknown);
if ( SUCCEEDED(hr) )
{
hr = pUnknown->QueryInterface(IID_IEnumVARIANT, (void**)ppEnum);
}
}
}
return hr;
}
///////////////////////////////////////////////////////////////////
HRESULT SDONextObjectFromCollection(
IEnumVARIANT* pEnum,
ISdo** ppSdo
)
{
HRESULT hr;
DWORD dwRetrieved = 1;
_variant_t vtDispatch;
_ASSERT( NULL != pEnum && NULL == *ppSdo );
hr = pEnum->Next(1, &vtDispatch, &dwRetrieved);
_ASSERT( S_OK == hr || S_FALSE == hr );
if ( S_OK == hr )
{
hr = vtDispatch.pdispVal->QueryInterface(IID_ISdo, (void**)ppSdo);
}
return hr;
}
///////////////////////////////////////////////////////////////////
HRESULT SDOGetComponentIdFromObject(
ISdo* pSdo,
LONG lPropertyId,
PLONG pComponentId
)
{
HRESULT hr;
_variant_t vtProperty;
_ASSERT( NULL != pSdo && NULL != pComponentId );
hr = pSdo->GetProperty(lPropertyId, &vtProperty);
if ( SUCCEEDED(hr) )
{
_ASSERT( VT_I4 == V_VT(&vtProperty) );
*pComponentId = V_I4(&vtProperty);
}
return hr;
}
///////////////////////////////////////////////////////////////////
HRESULT SDOGetSdoFromCollection(
ISdo* pSdoServer,
LONG lCollectionPropertyId,
LONG lComponentPropertyId,
LONG lComponentId,
ISdo** ppSdo
)
{
HRESULT hr;
CComPtr<IEnumVARIANT> pEnum;
CComPtr<ISdo> pSdo;
LONG ComponentId;
do
{
hr = SDOGetCollectionEnumerator(
pSdoServer,
lCollectionPropertyId,
&pEnum
);
if ( FAILED(hr) )
break;
hr = SDONextObjectFromCollection(pEnum,&pSdo);
while( S_OK == hr )
{
hr = SDOGetComponentIdFromObject(
pSdo,
lComponentPropertyId,
&ComponentId
);
if ( FAILED(hr) )
break;
if ( ComponentId == lComponentId )
{
pSdo->AddRef();
*ppSdo = pSdo;
break;
}
pSdo.Release();
hr = SDONextObjectFromCollection(pEnum,&pSdo);
}
if ( S_OK != hr )
hr = E_FAIL;
} while ( FALSE );
return hr;
}
//////////////////////////////////////////////////////////////////////////////
/*++
GetSdoVariant
Gets a Variant from the SDO's and handles any error checking.
--*/
//////////////////////////////////////////////////////////////////////////////
HRESULT GetSdoVariant(
ISdo *pSdo
, LONG lPropertyID
, VARIANT * pVariant
, UINT uiErrorID
, HWND hWnd
, IConsole *pConsole
)
{
ATLTRACE(_T("# GetSdoVariant\n"));
// Check for preconditions:
_ASSERTE( pSdo != NULL );
_ASSERTE( pVariant != NULL );
HRESULT hr;
hr = pSdo->GetProperty( lPropertyID, pVariant );
if( FAILED( hr ) )
{
CComPtr<IUnknown> spUnknown(pSdo);
if( spUnknown == NULL )
{
// Just put up an error dialog with the error string ID passed to us.
ShowErrorDialog( hWnd, uiErrorID );
}
else
{
CComBSTR bstrError;
HRESULT hrTemp = GetLastOLEErrorDescription( spUnknown, IID_ISdo, (BSTR *) &bstrError );
if( SUCCEEDED( hr ) )
{
ShowErrorDialog( hWnd, 1, bstrError );
}
else
{
// Just put up an error dialog with the error string ID passed to us.
ShowErrorDialog( hWnd, uiErrorID );
}
}
}
else
{
if( pVariant->vt == VT_EMPTY )
{
// This is not a real error -- we just need to
// know that this item has not yet been initialized.
return OLE_E_BLANK;
}
}
return hr;
}
//////////////////////////////////////////////////////////////////////////////
/*++
GetSdoBSTR
Gets a BSTR from the SDO's and handles any error checking.
--*/
//////////////////////////////////////////////////////////////////////////////
HRESULT GetSdoBSTR(
ISdo *pSdo
, LONG lPropertyID
, BSTR * pBSTR
, UINT uiErrorID
, HWND hWnd
, IConsole *pConsole
)
{
ATLTRACE(_T("# GetSdoBSTR\n"));
// Check for preconditions:
_ASSERTE( pSdo != NULL );
_ASSERTE( pBSTR != NULL );
HRESULT hr;
CComVariant spVariant;
hr = GetSdoVariant( pSdo, lPropertyID, &spVariant, uiErrorID, hWnd, pConsole );
if( SUCCEEDED( hr ) )
{
_ASSERTE( spVariant.vt == VT_BSTR );
// Copy the string from the variant to the BSTR pointer passed in.
if( SysReAllocString( pBSTR, spVariant.bstrVal ) == FALSE )
{
ShowErrorDialog( hWnd, uiErrorID );
}
}
return hr;
}
//////////////////////////////////////////////////////////////////////////////
/*++
GetSdoBOOL
Gets a BOOL from the SDO's and handles any error checking.
--*/
//////////////////////////////////////////////////////////////////////////////
HRESULT GetSdoBOOL(
ISdo *pSdo
, LONG lPropertyID
, BOOL * pBOOL
, UINT uiErrorID
, HWND hWnd
, IConsole *pConsole
)
{
ATLTRACE(_T("# GetSdoBOOL\n"));
// Check for preconditions:
_ASSERTE( pSdo != NULL );
_ASSERTE( pBOOL != NULL );
HRESULT hr;
CComVariant spVariant;
hr = GetSdoVariant( pSdo, lPropertyID, &spVariant, uiErrorID, hWnd, pConsole );
if( SUCCEEDED( hr ) )
{
_ASSERTE( spVariant.vt == VT_BOOL );
// Copy the value from the variant to the BOOL pointer passed in.
// We must do a quick and dirty conversion here because of the way OLE
// does BOOL values.
*pBOOL = ( spVariant.boolVal == VARIANT_TRUE ? TRUE : FALSE );
}
return hr;
}
//////////////////////////////////////////////////////////////////////////////
/*++
GetSdoI4
Gets an I4 (LONG) from the SDO's and handles any error checking.
--*/
//////////////////////////////////////////////////////////////////////////////
HRESULT GetSdoI4(
ISdo *pSdo
, LONG lPropertyID
, LONG * pI4
, UINT uiErrorID
, HWND hWnd
, IConsole *pConsole
)
{
ATLTRACE(_T("# GetSdoI4\n"));
// Check for preconditions:
_ASSERTE( pSdo != NULL );
_ASSERTE( pI4 != NULL );
HRESULT hr;
CComVariant spVariant;
hr = GetSdoVariant( pSdo, lPropertyID, &spVariant, uiErrorID, hWnd, pConsole );
if( SUCCEEDED( hr ) )
{
_ASSERTE( spVariant.vt == VT_I4 );
// Copy the value from the variant to the BOOL pointer passed in.
*pI4 = spVariant.lVal;
}
return hr;
}
//////////////////////////////////////////////////////////////////////////////
/*++
PutSdoVariant
Writes a Variant to the SDO's and handles any error checking.
--*/
//////////////////////////////////////////////////////////////////////////////
HRESULT PutSdoVariant(
ISdo *pSdo
, LONG lPropertyID
, VARIANT * pVariant
, UINT uiErrorID
, HWND hWnd
, IConsole *pConsole
)
{
ATLTRACE(_T("# PutSdoVariant\n"));
// Check for preconditions:
_ASSERTE( pSdo != NULL );
_ASSERTE( pVariant != NULL );
HRESULT hr;
hr = pSdo->PutProperty( lPropertyID, pVariant );
if( FAILED( hr ) )
{
CComPtr<IUnknown> spUnknown(pSdo);
if( spUnknown == NULL )
{
// Just put up an error dialog with the error string ID passed to us.
ShowErrorDialog( hWnd, uiErrorID );
}
else
{
CComBSTR bstrError;
HRESULT hrTemp = GetLastOLEErrorDescription( spUnknown, IID_ISdo, (BSTR *) &bstrError );
if( SUCCEEDED( hr ) )
{
ShowErrorDialog( hWnd, 1, bstrError );
}
else
{
// Just put up an error dialog with the error string ID passed to us.
ShowErrorDialog( hWnd, uiErrorID );
}
}
}
return hr;
}
//////////////////////////////////////////////////////////////////////////////
/*++
PutSdoBSTR
Writes a BSTR to the SDO's and handles any error checking.
--*/
//////////////////////////////////////////////////////////////////////////////
HRESULT PutSdoBSTR(
ISdo *pSdo
, LONG lPropertyID
, BSTR *pBSTR
, UINT uiErrorID
, HWND hWnd
, IConsole *pConsole
)
{
ATLTRACE(_T("# PutSdoBSTR\n"));
// Check for preconditions:
_ASSERTE( pSdo != NULL );
_ASSERTE( pBSTR != NULL );
HRESULT hr;
CComVariant spVariant;
// Load the Variant with the required info.
spVariant.vt = VT_BSTR;
spVariant.bstrVal = SysAllocString( *pBSTR );
if( spVariant.bstrVal == NULL )
{
ShowErrorDialog( hWnd, uiErrorID );
}
hr = PutSdoVariant( pSdo, lPropertyID, &spVariant, uiErrorID, hWnd, pConsole );
return hr;
}
//////////////////////////////////////////////////////////////////////////////
/*++
PutSdoBOOL
Writes a BOOL to the SDO's and handles any error checking.
--*/
//////////////////////////////////////////////////////////////////////////////
HRESULT PutSdoBOOL(
ISdo *pSdo
, LONG lPropertyID
, BOOL bValue
, UINT uiErrorID
, HWND hWnd
, IConsole *pConsole
)
{
ATLTRACE(_T("# PutSdoBOOL\n"));
// Check for preconditions:
_ASSERTE( pSdo != NULL );
HRESULT hr;
CComVariant spVariant;
// Load the Variant with the required info.
// We have to do a little mapping here because of the way Automation does BOOL's.
spVariant.vt = VT_BOOL;
spVariant.boolVal = ( bValue ? VARIANT_TRUE : VARIANT_FALSE );
hr = PutSdoVariant( pSdo, lPropertyID, &spVariant, uiErrorID, hWnd, pConsole );
return hr;
}
//////////////////////////////////////////////////////////////////////////////
/*++
PutSdoI4
Writes an I4 (LONG) to the SDO's and handles any error checking.
--*/
//////////////////////////////////////////////////////////////////////////////
HRESULT PutSdoI4(
ISdo *pSdo
, LONG lPropertyID
, LONG lValue
, UINT uiErrorID
, HWND hWnd
, IConsole *pConsole
)
{
ATLTRACE(_T("# PutSdoI4\n"));
// Check for preconditions:
_ASSERTE( pSdo != NULL );
HRESULT hr;
CComVariant spVariant;
// Load the Variant with the required info.
spVariant.vt = VT_I4;
spVariant.lVal = lValue;
hr = PutSdoVariant( pSdo, lPropertyID, &spVariant, uiErrorID, hWnd, pConsole );
return hr;
}
//////////////////////////////////////////////////////////////////////////////
/*++
GetLastOLEErrorDescription
Gets an error string from an interface.
--*/
//////////////////////////////////////////////////////////////////////////////
HRESULT GetLastOLEErrorDescription(
IUnknown *pUnknown
, REFIID riid
, BSTR *pbstrError
)
{
ATLTRACE(_T("# GetLastOLEErrorDescription\n"));
// Check for preconditions:
_ASSERTE( pUnknown != NULL );
HRESULT hr;
CComQIPtr<ISupportErrorInfo, &IID_ISupportErrorInfo> spSupportErrorInfo(pUnknown);
if( spSupportErrorInfo == NULL )
{
return E_NOINTERFACE;
}
hr = spSupportErrorInfo->InterfaceSupportsErrorInfo(riid);
if( S_OK != hr )
{
return E_FAIL;
}
CComPtr<IErrorInfo> spErrorInfo;
hr = GetErrorInfo( /* reserved */ 0, (IErrorInfo **) &spErrorInfo );
if( hr != S_OK || spErrorInfo == NULL )
{
return E_FAIL;
}
hr = spErrorInfo->GetDescription( pbstrError );
return hr;
}
//////////////////////////////////////////////////////////////////////////////
/*++
VendorsVector::VendorsVector
Constructor for STL vector wrapper for the SDO's vendor list.
Can throw E_FAIL or exceptions from std::vector::push_back.
--*/
//////////////////////////////////////////////////////////////////////////////
VendorsVector::VendorsVector( ISdoCollection * pSdoVendors )
{
HRESULT hr = S_OK;
CComPtr<IUnknown> spUnknown;
CComPtr<IEnumVARIANT> spEnumVariant;
CComVariant spVariant;
long ulCount;
ULONG ulCountReceived;
if( ! pSdoVendors )
{
throw E_FAIL; // Is there a better error to return here?
}
// We check the count of items in our collection and don't bother getting the
// enumerator if the count is zero.
// This saves time and also helps us to avoid a bug in the the enumerator which
// causes it to fail if we call next when it is empty.
pSdoVendors->get_Count( & ulCount );
if( ulCount > 0 )
{
// Get the enumerator for the Clients collection.
hr = pSdoVendors->get__NewEnum( (IUnknown **) & spUnknown );
if( FAILED( hr ) || ! spUnknown )
{
throw E_FAIL;
}
hr = spUnknown->QueryInterface( IID_IEnumVARIANT, (void **) &spEnumVariant );
spUnknown.Release();
if( FAILED( hr ) || ! spEnumVariant )
{
throw E_FAIL;
}
// Get the first item.
hr = spEnumVariant->Next( 1, & spVariant, &ulCountReceived );
while( SUCCEEDED( hr ) && ulCountReceived == 1 )
{
// Get an sdo pointer from the variant we received.
if( spVariant.vt != VT_DISPATCH || ! spVariant.pdispVal )
{
_ASSERTE( FALSE );
continue;
}
CComPtr<ISdo> spSdo;
hr = spVariant.pdispVal->QueryInterface( IID_ISdo, (void **) &spSdo );
spVariant.Clear();
if( FAILED( hr ) )
{
_ASSERTE( FALSE );
continue;
}
// Get Vendor Name.
hr = spSdo->GetProperty( PROPERTY_SDO_NAME, &spVariant );
if( FAILED( hr ) )
{
_ASSERTE( FALSE );
continue;
}
_ASSERTE( spVariant.vt == VT_BSTR );
CComBSTR bstrVendorName = spVariant.bstrVal;
spVariant.Clear();
// Get Vendor ID.
hr = spSdo->GetProperty( PROPERTY_NAS_VENDOR_ID, &spVariant );
if( FAILED( hr ) )
{
_ASSERTE( FALSE );
continue;
}
_ASSERTE( spVariant.vt == VT_I4 );
LONG lVendorID = spVariant.lVal;
spVariant.Clear();
// Add the vendor infor to the list of vendors.
push_back( std::make_pair(bstrVendorName, lVendorID) );
// Get the next item.
hr = spEnumVariant->Next( 1, & spVariant, &ulCountReceived );
}
}
else
{
// There are no items in the enumeration
// Do nothing.
}
}
//////////////////////////////////////////////////////////////////////////////
/*++
VendorsVector::VendorIDToOrdinal
Given a RADIUS vendor ID, tells you the position of that vendor in the vector.
--*/
//////////////////////////////////////////////////////////////////////////////
int VendorsVector::VendorIDToOrdinal( LONG lVendorID )
{
for (int i = 0; i < size() ; ++i)
{
if( lVendorID == operator[](i).second )
{
return i;
}
}
return 0;
}