1218 lines
29 KiB
C++
1218 lines
29 KiB
C++
|
//////////////////////////////////////////////////////////////////////////////
|
||
|
//
|
||
|
// Copyright (c) 1999-2001 Microsoft Corporation
|
||
|
//
|
||
|
// Module Name:
|
||
|
// ObjectManager.cpp
|
||
|
//
|
||
|
// Description:
|
||
|
// Object Manager implementation.
|
||
|
//
|
||
|
// Documentation:
|
||
|
// Yes.
|
||
|
//
|
||
|
// Maintained By:
|
||
|
// Galen Barbee (GalenB) 22-NOV-1999
|
||
|
//
|
||
|
//////////////////////////////////////////////////////////////////////////////
|
||
|
|
||
|
#include "pch.h"
|
||
|
#include "ObjectManager.h"
|
||
|
#include "ConnectionInfo.h"
|
||
|
#include "StandardInfo.h"
|
||
|
#include "EnumCookies.h"
|
||
|
|
||
|
DEFINE_THISCLASS("CObjectManager")
|
||
|
|
||
|
#define COOKIE_BUFFER_GROW_SIZE 100
|
||
|
|
||
|
// ************************************************************************
|
||
|
//
|
||
|
// Constructor / Destructor
|
||
|
//
|
||
|
// ************************************************************************
|
||
|
|
||
|
//////////////////////////////////////////////////////////////////////////////
|
||
|
//
|
||
|
// LPUNKNOWN
|
||
|
// CObjectManager::S_HrCreateInstance(
|
||
|
// IUnknown ** ppunkOut
|
||
|
// )
|
||
|
//
|
||
|
//////////////////////////////////////////////////////////////////////////////
|
||
|
HRESULT
|
||
|
CObjectManager::S_HrCreateInstance(
|
||
|
IUnknown ** ppunkOut
|
||
|
)
|
||
|
{
|
||
|
TraceFunc( "" );
|
||
|
|
||
|
Assert( ppunkOut != NULL );
|
||
|
|
||
|
HRESULT hr;
|
||
|
IServiceProvider * psp;
|
||
|
|
||
|
// Don't wrap - this can fail with E_POINTER.
|
||
|
hr = CServiceManager::S_HrGetManagerPointer( &psp );
|
||
|
|
||
|
if ( SUCCEEDED( hr ) )
|
||
|
{
|
||
|
hr = THR( psp->TypeSafeQS( CLSID_ObjectManager,
|
||
|
IUnknown,
|
||
|
ppunkOut
|
||
|
) );
|
||
|
psp->Release( );
|
||
|
|
||
|
} // if: service manager
|
||
|
else if ( hr == E_POINTER )
|
||
|
{
|
||
|
//
|
||
|
// This happens when the Service Manager is first started.
|
||
|
//
|
||
|
CObjectManager * pom = new CObjectManager( );
|
||
|
if ( pom != NULL )
|
||
|
{
|
||
|
hr = THR( pom->Init( ) );
|
||
|
if ( SUCCEEDED( hr ) )
|
||
|
{
|
||
|
hr = THR( pom->TypeSafeQI( IUnknown, ppunkOut ) );
|
||
|
} // if: success
|
||
|
|
||
|
pom->Release( );
|
||
|
|
||
|
} // if: got object
|
||
|
else
|
||
|
{
|
||
|
hr = E_OUTOFMEMORY;
|
||
|
}
|
||
|
|
||
|
} // if: service manager doesn't exists
|
||
|
else
|
||
|
{
|
||
|
THR( hr );
|
||
|
} // else:
|
||
|
|
||
|
HRETURN( hr );
|
||
|
|
||
|
} // S_HrCreateInstance( )
|
||
|
|
||
|
//////////////////////////////////////////////////////////////////////////////
|
||
|
//
|
||
|
// CObjectManager::CObjectManager( void )
|
||
|
//
|
||
|
//////////////////////////////////////////////////////////////////////////////
|
||
|
CObjectManager::CObjectManager( void )
|
||
|
{
|
||
|
TraceFunc( "" );
|
||
|
|
||
|
InterlockedIncrement( &g_cObjects );
|
||
|
|
||
|
TraceFuncExit();
|
||
|
} // CObjectManager( )
|
||
|
|
||
|
//////////////////////////////////////////////////////////////////////////////
|
||
|
//
|
||
|
// STDMETHODIMP
|
||
|
// CObjectManager::Init( void )
|
||
|
//
|
||
|
//////////////////////////////////////////////////////////////////////////////
|
||
|
STDMETHODIMP
|
||
|
CObjectManager::Init( void )
|
||
|
{
|
||
|
TraceFunc( "" );
|
||
|
|
||
|
HRESULT hr = S_OK;
|
||
|
|
||
|
// IUnknown stuff
|
||
|
Assert( m_cRef == 0 );
|
||
|
AddRef( ); // Add one count
|
||
|
|
||
|
// IObjectManager
|
||
|
Assert( m_cAllocSize == 0 );
|
||
|
Assert( m_cCurrentUsed == 0 );
|
||
|
Assert( m_pCookies == NULL );
|
||
|
|
||
|
HRETURN( hr );
|
||
|
|
||
|
} // Init( )
|
||
|
|
||
|
//////////////////////////////////////////////////////////////////////////////
|
||
|
//
|
||
|
// CObjectManager::~CObjectManager( )
|
||
|
//
|
||
|
//////////////////////////////////////////////////////////////////////////////
|
||
|
CObjectManager::~CObjectManager( )
|
||
|
{
|
||
|
TraceFunc( "" );
|
||
|
|
||
|
if ( m_pCookies != NULL )
|
||
|
{
|
||
|
while ( m_cCurrentUsed != 0 )
|
||
|
{
|
||
|
m_cCurrentUsed --;
|
||
|
|
||
|
if ( m_pCookies[ m_cCurrentUsed ] != NULL )
|
||
|
{
|
||
|
m_pCookies[ m_cCurrentUsed ]->Release( );
|
||
|
}
|
||
|
}
|
||
|
|
||
|
TraceFree( m_pCookies );
|
||
|
}
|
||
|
|
||
|
InterlockedDecrement( &g_cObjects );
|
||
|
|
||
|
TraceFuncExit();
|
||
|
} // ~CObjectManager( )
|
||
|
|
||
|
|
||
|
// ************************************************************************
|
||
|
//
|
||
|
// IUnknown
|
||
|
//
|
||
|
// ************************************************************************
|
||
|
|
||
|
|
||
|
//////////////////////////////////////////////////////////////////////////////
|
||
|
//
|
||
|
// STDMETHODIMP
|
||
|
// CObjectManager::QueryInterface(
|
||
|
// REFIID riid,
|
||
|
// LPVOID *ppv
|
||
|
// )
|
||
|
//
|
||
|
//////////////////////////////////////////////////////////////////////////////
|
||
|
STDMETHODIMP
|
||
|
CObjectManager::QueryInterface(
|
||
|
REFIID riid,
|
||
|
LPVOID *ppv
|
||
|
)
|
||
|
{
|
||
|
TraceQIFunc( riid, ppv );
|
||
|
|
||
|
HRESULT hr = E_NOINTERFACE;
|
||
|
|
||
|
if ( IsEqualIID( riid, IID_IUnknown ) )
|
||
|
{
|
||
|
*ppv = static_cast< LPUNKNOWN >( this );
|
||
|
hr = S_OK;
|
||
|
} // if: IUnknown
|
||
|
else if ( IsEqualIID( riid, IID_IObjectManager ) )
|
||
|
{
|
||
|
*ppv = TraceInterface( __THISCLASS__, IObjectManager, this, 0 );
|
||
|
hr = S_OK;
|
||
|
} // else if: IObjectManager
|
||
|
|
||
|
if ( SUCCEEDED( hr ) )
|
||
|
{
|
||
|
((IUnknown*) *ppv)->AddRef( );
|
||
|
} // if: success
|
||
|
|
||
|
QIRETURN_IGNORESTDMARSHALLING( hr, riid );
|
||
|
}
|
||
|
|
||
|
//////////////////////////////////////////////////////////////////////////////
|
||
|
//
|
||
|
// STDMETHODIMP_(ULONG)
|
||
|
// CObjectManager::AddRef( void )
|
||
|
//
|
||
|
//////////////////////////////////////////////////////////////////////////////
|
||
|
STDMETHODIMP_(ULONG)
|
||
|
CObjectManager::AddRef( void )
|
||
|
{
|
||
|
TraceFunc( "[IUnknown]" );
|
||
|
|
||
|
InterlockedIncrement( &m_cRef );
|
||
|
|
||
|
RETURN( m_cRef );
|
||
|
|
||
|
} // AddRef( )
|
||
|
|
||
|
//////////////////////////////////////////////////////////////////////////////
|
||
|
//
|
||
|
// STDMETHODIMP_(ULONG)
|
||
|
// CObjectManager::Release( void )
|
||
|
//
|
||
|
//////////////////////////////////////////////////////////////////////////////
|
||
|
STDMETHODIMP_(ULONG)
|
||
|
CObjectManager::Release( void )
|
||
|
{
|
||
|
TraceFunc( "[IUnknown]" );
|
||
|
|
||
|
InterlockedDecrement( &m_cRef );
|
||
|
|
||
|
if ( m_cRef )
|
||
|
RETURN( m_cRef );
|
||
|
|
||
|
TraceDo( delete this );
|
||
|
|
||
|
RETURN(0);
|
||
|
|
||
|
} // Release( )
|
||
|
|
||
|
|
||
|
// ************************************************************************
|
||
|
//
|
||
|
// IObjectManager
|
||
|
//
|
||
|
// ************************************************************************
|
||
|
|
||
|
|
||
|
//////////////////////////////////////////////////////////////////////////////
|
||
|
//
|
||
|
// STDMETHODIMP
|
||
|
// CObjectManager::FindObject(
|
||
|
// REFCLSID rclsidTypeIn,
|
||
|
// OBJECTCOOKIE cookieParentIn,
|
||
|
// LPCWSTR pcszNameIn,
|
||
|
// REFCLSID rclsidFormatIn,
|
||
|
// OBJECTCOOKIE * cookieOut,
|
||
|
// LPUNKNOWN * punkOut
|
||
|
// )
|
||
|
//
|
||
|
//////////////////////////////////////////////////////////////////////////////
|
||
|
STDMETHODIMP
|
||
|
CObjectManager::FindObject(
|
||
|
REFCLSID rclsidTypeIn,
|
||
|
OBJECTCOOKIE cookieParentIn,
|
||
|
LPCWSTR pcszNameIn,
|
||
|
REFCLSID rclsidFormatIn,
|
||
|
OBJECTCOOKIE * pcookieOut,
|
||
|
LPUNKNOWN * ppunkOut
|
||
|
)
|
||
|
{
|
||
|
TraceFunc( "[IObjectManager]" );
|
||
|
|
||
|
ExtObjectEntry * pentry;
|
||
|
|
||
|
HRESULT hr = E_UNEXPECTED;
|
||
|
|
||
|
OBJECTCOOKIE cookie = 0;
|
||
|
|
||
|
CStandardInfo * pcsi = NULL; // don't free
|
||
|
|
||
|
BOOL fTempCookie = FALSE;
|
||
|
|
||
|
CEnumCookies * pcec = NULL;
|
||
|
IUnknown * punk = NULL;
|
||
|
IExtendObjectManager * peom = NULL;
|
||
|
|
||
|
//
|
||
|
// Check to see if we already have an object.
|
||
|
//
|
||
|
|
||
|
if ( pcszNameIn != NULL )
|
||
|
{
|
||
|
hr = STHR( HrSearchForExistingCookie( rclsidTypeIn, cookieParentIn, pcszNameIn, &cookie ) );
|
||
|
if ( FAILED( hr ) )
|
||
|
goto Cleanup;
|
||
|
|
||
|
if ( hr == S_FALSE )
|
||
|
{
|
||
|
hr = THR( HrCreateNewCookie( rclsidTypeIn, cookieParentIn, pcszNameIn, &cookie ) );
|
||
|
if ( FAILED( hr ) )
|
||
|
goto Cleanup;
|
||
|
|
||
|
pcsi = m_pCookies[ cookie ];
|
||
|
Assert( pcsi != NULL );
|
||
|
}
|
||
|
else if ( hr == S_OK )
|
||
|
{
|
||
|
//
|
||
|
// Found an existing cookie.
|
||
|
//
|
||
|
|
||
|
if ( pcookieOut != NULL )
|
||
|
{
|
||
|
*pcookieOut = cookie;
|
||
|
}
|
||
|
|
||
|
if ( ppunkOut != NULL )
|
||
|
{
|
||
|
pcsi = m_pCookies[ cookie ];
|
||
|
|
||
|
//
|
||
|
// Is the object still in an failed state or still pending?
|
||
|
//
|
||
|
|
||
|
if ( FAILED( pcsi->m_hrStatus ) )
|
||
|
{
|
||
|
hr = pcsi->m_hrStatus;
|
||
|
goto Cleanup;
|
||
|
}
|
||
|
|
||
|
//
|
||
|
// Retrieve the requested format.
|
||
|
//
|
||
|
|
||
|
hr = THR( GetObject( rclsidFormatIn, cookie, ppunkOut ) );
|
||
|
// we always jump to cleanup. No need to check hr here.
|
||
|
|
||
|
goto Cleanup;
|
||
|
}
|
||
|
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
//
|
||
|
// Unexpected error_success - now what?
|
||
|
//
|
||
|
Assert( !hr );
|
||
|
goto Cleanup;
|
||
|
}
|
||
|
|
||
|
} // if: named object
|
||
|
else
|
||
|
{
|
||
|
Assert( pcsi == NULL );
|
||
|
}
|
||
|
|
||
|
//
|
||
|
// Create a new object.
|
||
|
//
|
||
|
|
||
|
if ( IsEqualIID( rclsidFormatIn, IID_NULL )
|
||
|
|| ppunkOut == NULL
|
||
|
)
|
||
|
{
|
||
|
//
|
||
|
// No-op.
|
||
|
//
|
||
|
hr = S_OK;
|
||
|
}
|
||
|
else if ( IsEqualIID( rclsidFormatIn, DFGUID_StandardInfo ) )
|
||
|
{
|
||
|
hr = THR( pcsi->QueryInterface( DFGUID_StandardInfo,
|
||
|
reinterpret_cast< void ** >( ppunkOut )
|
||
|
) );
|
||
|
if ( FAILED( hr ) )
|
||
|
goto Cleanup;
|
||
|
}
|
||
|
else if ( IsEqualIID( rclsidFormatIn, DFGUID_ConnectionInfoFormat ) )
|
||
|
{
|
||
|
if ( pcsi->m_pci != NULL )
|
||
|
{
|
||
|
*ppunkOut = TraceInterface( L"CConnectionInfo!ObjectManager", IConnectionInfo, pcsi->m_pci, 0 );
|
||
|
(*ppunkOut)->AddRef( );
|
||
|
hr = S_OK;
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
hr = THR( CConnectionInfo::S_HrCreateInstance( &punk,
|
||
|
pcsi->m_cookieParent
|
||
|
) );
|
||
|
if ( FAILED( hr ) )
|
||
|
goto Cleanup;
|
||
|
|
||
|
hr = THR( punk->TypeSafeQI( IConnectionInfo,
|
||
|
&pcsi->m_pci
|
||
|
) );
|
||
|
if ( FAILED( hr ) )
|
||
|
goto Cleanup;
|
||
|
|
||
|
hr = THR( punk->QueryInterface( IID_IConnectionInfo,
|
||
|
reinterpret_cast< void ** >( ppunkOut )
|
||
|
) );
|
||
|
if ( FAILED( hr ) )
|
||
|
goto Cleanup;
|
||
|
}
|
||
|
}
|
||
|
else if ( IsEqualIID( rclsidFormatIn, DFGUID_EnumCookies ) )
|
||
|
{
|
||
|
ULONG cIter;
|
||
|
|
||
|
//
|
||
|
// Create a new cookie enumerator.
|
||
|
//
|
||
|
|
||
|
pcec = new CEnumCookies;
|
||
|
if ( pcec == NULL )
|
||
|
goto OutOfMemory;
|
||
|
|
||
|
//
|
||
|
// Initialize the enumerator. This also cause an AddRef( ).
|
||
|
//
|
||
|
|
||
|
hr = THR( pcec->Init( ) );
|
||
|
if ( FAILED( hr ) )
|
||
|
goto Cleanup;
|
||
|
|
||
|
//
|
||
|
// See who matches our citeria.
|
||
|
//
|
||
|
|
||
|
pcec->m_cIter = 0;
|
||
|
|
||
|
for( cIter = 1; cIter < m_cCurrentUsed; cIter ++ )
|
||
|
{
|
||
|
pcsi = m_pCookies[ cIter ];
|
||
|
|
||
|
if ( pcsi != NULL )
|
||
|
{
|
||
|
if ( rclsidTypeIn == IID_NULL
|
||
|
|| pcsi->m_clsidType == rclsidTypeIn
|
||
|
)
|
||
|
{
|
||
|
if ( cookieParentIn == NULL
|
||
|
|| pcsi->m_cookieParent == cookieParentIn
|
||
|
)
|
||
|
{
|
||
|
if ( ( pcszNameIn == NULL )
|
||
|
|| ( ( pcsi->m_bstrName != NULL )
|
||
|
&& ( StrCmpI( pcsi->m_bstrName, pcszNameIn ) == 0 )
|
||
|
)
|
||
|
)
|
||
|
{
|
||
|
//
|
||
|
// Match!
|
||
|
//
|
||
|
pcec->m_cIter ++;
|
||
|
|
||
|
} // if: names match
|
||
|
|
||
|
} // if: parents match
|
||
|
|
||
|
} // if: match parent and type
|
||
|
|
||
|
} // if: valid element
|
||
|
|
||
|
} // for: cIter
|
||
|
|
||
|
if ( pcec->m_cIter == 0 )
|
||
|
goto ErrorNotFound;
|
||
|
|
||
|
//
|
||
|
// Alloc an array to hold the cookies.
|
||
|
//
|
||
|
|
||
|
pcec->m_pList = (OBJECTCOOKIE*) TraceAlloc( HEAP_ZERO_MEMORY, pcec->m_cIter * sizeof(OBJECTCOOKIE) );
|
||
|
if ( pcec->m_pList == NULL )
|
||
|
goto OutOfMemory;
|
||
|
|
||
|
pcec->m_cAlloced = pcec->m_cIter;
|
||
|
pcec->m_cIter = 0;
|
||
|
|
||
|
for( cIter = 1; cIter < m_cCurrentUsed; cIter ++ )
|
||
|
{
|
||
|
pcsi = m_pCookies[ cIter ];
|
||
|
|
||
|
if ( pcsi != NULL )
|
||
|
{
|
||
|
if ( rclsidTypeIn == IID_NULL
|
||
|
|| pcsi->m_clsidType == rclsidTypeIn
|
||
|
)
|
||
|
{
|
||
|
if ( cookieParentIn == NULL
|
||
|
|| pcsi->m_cookieParent == cookieParentIn
|
||
|
)
|
||
|
{
|
||
|
if ( ( pcszNameIn == NULL )
|
||
|
|| ( ( pcsi->m_bstrName != NULL )
|
||
|
&& ( StrCmpI( pcsi->m_bstrName, pcszNameIn ) == 0 )
|
||
|
)
|
||
|
)
|
||
|
{
|
||
|
//
|
||
|
// Match!
|
||
|
//
|
||
|
|
||
|
pcec->m_pList[ pcec->m_cIter ] = cIter;
|
||
|
|
||
|
pcec->m_cIter ++;
|
||
|
|
||
|
} // if: names match
|
||
|
|
||
|
} // if: parents match
|
||
|
|
||
|
} // if: match parent and type
|
||
|
|
||
|
} // if: valid element
|
||
|
|
||
|
} // for: cIter
|
||
|
|
||
|
Assert( pcec->m_cIter != 0 );
|
||
|
pcec->m_cCookies = pcec->m_cIter;
|
||
|
pcec->m_cIter = 0;
|
||
|
|
||
|
//
|
||
|
// Grab the inteface on the way out.
|
||
|
//
|
||
|
|
||
|
hr = THR( pcec->QueryInterface( IID_IEnumCookies,
|
||
|
reinterpret_cast< void ** >( ppunkOut )
|
||
|
) );
|
||
|
if ( FAILED( hr ) )
|
||
|
goto Cleanup;
|
||
|
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
//
|
||
|
// Check for extension formats.
|
||
|
//
|
||
|
|
||
|
//
|
||
|
// See if the format already exists for this cookie.
|
||
|
//
|
||
|
|
||
|
if ( punk != NULL )
|
||
|
{
|
||
|
punk->Release( );
|
||
|
punk = NULL;
|
||
|
}
|
||
|
|
||
|
if ( pcsi != NULL )
|
||
|
{
|
||
|
for( pentry = pcsi->m_pExtObjList; pentry != NULL; pentry = pentry->pNext )
|
||
|
{
|
||
|
if ( pentry->iid == rclsidFormatIn )
|
||
|
{
|
||
|
hr = THR( pentry->punk->QueryInterface( rclsidFormatIn,
|
||
|
reinterpret_cast< void ** >( &punk )
|
||
|
) );
|
||
|
if ( FAILED( hr ) )
|
||
|
goto Cleanup;
|
||
|
|
||
|
break; // exit loop
|
||
|
}
|
||
|
|
||
|
} // for: pentry
|
||
|
|
||
|
} // if: have cookie
|
||
|
else
|
||
|
{
|
||
|
//
|
||
|
// Create a temporary cookie.
|
||
|
//
|
||
|
|
||
|
Assert( pcszNameIn == NULL );
|
||
|
|
||
|
hr = THR( HrCreateNewCookie( IID_NULL, cookieParentIn, NULL, &cookie ) );
|
||
|
if ( FAILED( hr ) )
|
||
|
goto Cleanup;
|
||
|
|
||
|
fTempCookie = TRUE;
|
||
|
|
||
|
Assert( pcsi == NULL );
|
||
|
|
||
|
} // else: need a temporary cookie
|
||
|
|
||
|
if ( punk == NULL )
|
||
|
{
|
||
|
//
|
||
|
// Possibly a new or externally object, try creating it and querying.
|
||
|
//
|
||
|
|
||
|
hr = THR( HrCoCreateInternalInstance( rclsidFormatIn,
|
||
|
NULL,
|
||
|
CLSCTX_ALL,
|
||
|
TypeSafeParams( IExtendObjectManager, &peom )
|
||
|
) );
|
||
|
if ( FAILED( hr ) )
|
||
|
goto Cleanup;
|
||
|
|
||
|
Assert( punk == NULL );
|
||
|
// Can't wrap with THR because it can return E_PENDING.
|
||
|
hr = peom->FindObject( cookie,
|
||
|
rclsidTypeIn,
|
||
|
pcszNameIn,
|
||
|
&punk
|
||
|
);
|
||
|
if ( hr == E_PENDING )
|
||
|
{
|
||
|
// ignore
|
||
|
}
|
||
|
else if ( FAILED( hr ) )
|
||
|
{
|
||
|
THR( hr );
|
||
|
goto Cleanup;
|
||
|
}
|
||
|
|
||
|
if ( fTempCookie )
|
||
|
{
|
||
|
(m_pCookies[ cookie ])->Release( );
|
||
|
m_pCookies[ cookie ] = NULL;
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
//
|
||
|
// Keep track of the format (if extension wants)
|
||
|
//
|
||
|
|
||
|
if ( ( ( SUCCEEDED( hr )
|
||
|
&& hr != S_FALSE
|
||
|
)
|
||
|
|| hr == E_PENDING
|
||
|
)
|
||
|
&& punk != NULL
|
||
|
&& pcsi != NULL
|
||
|
)
|
||
|
{
|
||
|
pentry = (ExtObjectEntry *) TraceAlloc( 0, sizeof(ExtObjectEntry) );
|
||
|
if ( pentry == NULL )
|
||
|
goto OutOfMemory;
|
||
|
|
||
|
pentry->iid = rclsidFormatIn;
|
||
|
pentry->pNext = pcsi->m_pExtObjList;
|
||
|
pentry->punk = punk;
|
||
|
pentry->punk->AddRef( );
|
||
|
|
||
|
pcsi->m_pExtObjList = pentry; // update header of list (LIFO)
|
||
|
pcsi->m_hrStatus = hr; // update status
|
||
|
}
|
||
|
|
||
|
} // else: persistent cookie
|
||
|
|
||
|
if ( SUCCEEDED( hr ) )
|
||
|
{
|
||
|
// Give up ownership
|
||
|
*ppunkOut = punk;
|
||
|
punk = NULL;
|
||
|
}
|
||
|
|
||
|
} // if: creating new object
|
||
|
|
||
|
} // else: possible extension format
|
||
|
|
||
|
//
|
||
|
// Save stuff for the caller.
|
||
|
//
|
||
|
|
||
|
if ( pcookieOut != NULL )
|
||
|
{
|
||
|
*pcookieOut = cookie;
|
||
|
}
|
||
|
|
||
|
Cleanup:
|
||
|
if ( punk != NULL )
|
||
|
{
|
||
|
punk->Release( );
|
||
|
}
|
||
|
if ( peom != NULL )
|
||
|
{
|
||
|
peom->Release( );
|
||
|
}
|
||
|
if ( pcec != NULL )
|
||
|
{
|
||
|
pcec->Release( );
|
||
|
}
|
||
|
|
||
|
HRETURN( hr );
|
||
|
|
||
|
OutOfMemory:
|
||
|
hr = E_OUTOFMEMORY;
|
||
|
goto Cleanup;
|
||
|
|
||
|
ErrorNotFound:
|
||
|
// The error text is better than the coding value.
|
||
|
hr = THR( HRESULT_FROM_WIN32( ERROR_NOT_FOUND ) );
|
||
|
goto Cleanup;
|
||
|
|
||
|
} // FindObject( )
|
||
|
|
||
|
//////////////////////////////////////////////////////////////////////////////
|
||
|
//
|
||
|
// STDMETHODIMP
|
||
|
// CObjectManager::GetObject(
|
||
|
// REFCLSID rclsidFormatIn,
|
||
|
// OBJECTCOOKIE cookieIn,
|
||
|
// LPUNKNOWN * ppunkOut
|
||
|
// )
|
||
|
//
|
||
|
//////////////////////////////////////////////////////////////////////////////
|
||
|
STDMETHODIMP
|
||
|
CObjectManager::GetObject(
|
||
|
REFCLSID rclsidFormatIn,
|
||
|
OBJECTCOOKIE cookieIn,
|
||
|
LPUNKNOWN * ppunkOut
|
||
|
)
|
||
|
{
|
||
|
TraceFunc( "[IObjectManager]" );
|
||
|
|
||
|
CStandardInfo * pcsi;
|
||
|
ExtObjectEntry * pentry;
|
||
|
|
||
|
HRESULT hr = E_UNEXPECTED;
|
||
|
|
||
|
IUnknown * punk = NULL;
|
||
|
IExtendObjectManager * peom = NULL;
|
||
|
|
||
|
//
|
||
|
// Check parameters
|
||
|
//
|
||
|
if ( cookieIn == 0 || cookieIn >= m_cCurrentUsed )
|
||
|
goto InvalidArg;
|
||
|
|
||
|
pcsi = m_pCookies[ cookieIn ];
|
||
|
if ( pcsi == NULL )
|
||
|
goto ErrorNotFound;
|
||
|
|
||
|
//
|
||
|
// Create the request format object.
|
||
|
//
|
||
|
|
||
|
if ( IsEqualIID( rclsidFormatIn, IID_NULL )
|
||
|
|| ppunkOut == NULL
|
||
|
)
|
||
|
{
|
||
|
//
|
||
|
// No-op.
|
||
|
//
|
||
|
hr = S_OK;
|
||
|
}
|
||
|
else if ( IsEqualIID( rclsidFormatIn, DFGUID_StandardInfo ) )
|
||
|
{
|
||
|
hr = THR( pcsi->QueryInterface( DFGUID_StandardInfo,
|
||
|
reinterpret_cast< void ** >( ppunkOut )
|
||
|
) );
|
||
|
if ( FAILED( hr ) )
|
||
|
goto Cleanup;
|
||
|
}
|
||
|
else if ( IsEqualIID( rclsidFormatIn, DFGUID_ConnectionInfoFormat ) )
|
||
|
{
|
||
|
if ( pcsi->m_pci != NULL )
|
||
|
{
|
||
|
*ppunkOut = pcsi->m_pci;
|
||
|
(*ppunkOut)->AddRef( );
|
||
|
hr = S_OK;
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
hr = THR( CConnectionInfo::S_HrCreateInstance( &punk,
|
||
|
pcsi->m_cookieParent
|
||
|
) );
|
||
|
if ( FAILED( hr ) )
|
||
|
goto Cleanup;
|
||
|
|
||
|
hr = THR( punk->TypeSafeQI( IConnectionInfo,
|
||
|
&pcsi->m_pci
|
||
|
) );
|
||
|
if ( FAILED( hr ) )
|
||
|
goto Cleanup;
|
||
|
|
||
|
hr = THR( punk->QueryInterface( IID_IConnectionInfo,
|
||
|
reinterpret_cast< void ** >( ppunkOut )
|
||
|
) );
|
||
|
if ( FAILED( hr ) )
|
||
|
goto Cleanup;
|
||
|
}
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
//
|
||
|
// See if the format already exists for this cookie.
|
||
|
//
|
||
|
|
||
|
if ( punk != NULL )
|
||
|
{
|
||
|
punk->Release( );
|
||
|
punk = NULL;
|
||
|
}
|
||
|
|
||
|
for( pentry = pcsi->m_pExtObjList; pentry != NULL; pentry = pentry->pNext )
|
||
|
{
|
||
|
if ( pentry->iid == rclsidFormatIn )
|
||
|
{
|
||
|
hr = THR( pentry->punk->QueryInterface( rclsidFormatIn,
|
||
|
reinterpret_cast< void ** >( &punk )
|
||
|
) );
|
||
|
if ( FAILED( hr ) )
|
||
|
goto Cleanup;
|
||
|
|
||
|
break; // exit loop
|
||
|
}
|
||
|
|
||
|
} // for: pentry
|
||
|
|
||
|
if ( punk == NULL )
|
||
|
goto ErrorNotFound;
|
||
|
|
||
|
// Give up ownership
|
||
|
*ppunkOut = punk;
|
||
|
punk = NULL;
|
||
|
|
||
|
} // else: external?
|
||
|
|
||
|
Cleanup:
|
||
|
if ( punk != NULL )
|
||
|
{
|
||
|
punk->Release( );
|
||
|
}
|
||
|
if ( peom != NULL )
|
||
|
{
|
||
|
peom->Release( );
|
||
|
}
|
||
|
|
||
|
HRETURN( hr );
|
||
|
|
||
|
InvalidArg:
|
||
|
hr = THR( E_INVALIDARG );
|
||
|
goto Cleanup;
|
||
|
|
||
|
ErrorNotFound:
|
||
|
hr = THR( HRESULT_FROM_WIN32( ERROR_NOT_FOUND ) );
|
||
|
goto Cleanup;
|
||
|
#if 0
|
||
|
OutOfMemory:
|
||
|
hr = E_OUTOFMEMORY;
|
||
|
goto Cleanup;
|
||
|
#endif
|
||
|
|
||
|
} // GetObject( )
|
||
|
|
||
|
//////////////////////////////////////////////////////////////////////////////
|
||
|
//
|
||
|
// STDMETHODIMP
|
||
|
// CObjectManager::RemoveObject(
|
||
|
// OBJECTCOOKIE cookieIn
|
||
|
// )
|
||
|
//
|
||
|
//////////////////////////////////////////////////////////////////////////////
|
||
|
STDMETHODIMP
|
||
|
CObjectManager::RemoveObject(
|
||
|
OBJECTCOOKIE cookieIn
|
||
|
)
|
||
|
{
|
||
|
TraceFunc( "[IObjectManager]" );
|
||
|
|
||
|
HRESULT hr = E_UNEXPECTED;
|
||
|
CStandardInfo * pcsi;
|
||
|
|
||
|
BOOL fLocked = FALSE;
|
||
|
|
||
|
//
|
||
|
// Check parameters
|
||
|
//
|
||
|
if ( cookieIn == 0 || cookieIn >= m_cCurrentUsed )
|
||
|
goto InvalidArg;
|
||
|
|
||
|
pcsi = m_pCookies[ cookieIn ];
|
||
|
if ( pcsi == NULL )
|
||
|
goto ErrorNotFound;
|
||
|
|
||
|
hr = THR( HrDeleteInstanceAndChildren( cookieIn ) );
|
||
|
if ( FAILED( hr ) )
|
||
|
goto Cleanup;
|
||
|
|
||
|
Cleanup:
|
||
|
HRETURN( hr );
|
||
|
|
||
|
InvalidArg:
|
||
|
hr = THR( E_INVALIDARG );
|
||
|
goto Cleanup;
|
||
|
|
||
|
ErrorNotFound:
|
||
|
hr = THR( HRESULT_FROM_WIN32( ERROR_NOT_FOUND ) );
|
||
|
goto Cleanup;
|
||
|
|
||
|
} // RemoveObject( )
|
||
|
|
||
|
//////////////////////////////////////////////////////////////////////////////
|
||
|
//
|
||
|
// STDMETHODIMP
|
||
|
// CObjectManager::SetObjectStatus(
|
||
|
// OBJECTCOOKIE cookieIn,
|
||
|
// HRESULT hrIn
|
||
|
// )
|
||
|
//
|
||
|
//////////////////////////////////////////////////////////////////////////////
|
||
|
STDMETHODIMP
|
||
|
CObjectManager::SetObjectStatus(
|
||
|
OBJECTCOOKIE cookieIn,
|
||
|
HRESULT hrIn
|
||
|
)
|
||
|
{
|
||
|
TraceFunc( "[IObjectManager]" );
|
||
|
|
||
|
HRESULT hr = S_OK;
|
||
|
CStandardInfo * pcsi;
|
||
|
|
||
|
//
|
||
|
// Check parameters
|
||
|
//
|
||
|
|
||
|
if ( cookieIn == 0 || cookieIn >= m_cCurrentUsed )
|
||
|
goto InvalidArg;
|
||
|
|
||
|
pcsi = m_pCookies[ cookieIn ];
|
||
|
if ( pcsi == NULL )
|
||
|
goto ErrorNotFound;
|
||
|
|
||
|
//
|
||
|
// Update the status.
|
||
|
//
|
||
|
|
||
|
pcsi->m_hrStatus = hrIn;
|
||
|
|
||
|
Cleanup:
|
||
|
HRETURN( hr );
|
||
|
|
||
|
InvalidArg:
|
||
|
hr = THR( E_INVALIDARG );
|
||
|
goto Cleanup;
|
||
|
|
||
|
ErrorNotFound:
|
||
|
hr = THR( HRESULT_FROM_WIN32( ERROR_NOT_FOUND ) );
|
||
|
goto Cleanup;
|
||
|
|
||
|
} // SetObjectStatus( )
|
||
|
|
||
|
|
||
|
//****************************************************************************
|
||
|
//
|
||
|
// Privates
|
||
|
//
|
||
|
//****************************************************************************
|
||
|
|
||
|
|
||
|
//////////////////////////////////////////////////////////////////////////////
|
||
|
//
|
||
|
// HRESULT
|
||
|
// CObjectManager::HrDeleteCookie(
|
||
|
// OBJECTCOOKIE cookieIn
|
||
|
// )
|
||
|
//
|
||
|
//////////////////////////////////////////////////////////////////////////////
|
||
|
HRESULT
|
||
|
CObjectManager::HrDeleteCookie(
|
||
|
OBJECTCOOKIE cookieIn
|
||
|
)
|
||
|
{
|
||
|
TraceFunc1( "cookieIn = %#X", cookieIn );
|
||
|
|
||
|
HRESULT hr = S_OK;
|
||
|
|
||
|
CStandardInfo * pcsi;
|
||
|
|
||
|
Assert( cookieIn != 0 && cookieIn < m_cCurrentUsed );
|
||
|
|
||
|
pcsi = m_pCookies[ cookieIn ];
|
||
|
Assert( pcsi != NULL );
|
||
|
pcsi->Release( );
|
||
|
m_pCookies[ cookieIn ] = NULL;
|
||
|
|
||
|
HRETURN( hr );
|
||
|
|
||
|
} // HrDeleteCookie( )
|
||
|
|
||
|
//////////////////////////////////////////////////////////////////////////////
|
||
|
//
|
||
|
// HRESULT
|
||
|
// CObjectManager::HrSearchForExistingCookie(
|
||
|
// OBJECTCOOKIE cookieIn,
|
||
|
// LPUNKNOWN ppunkOut
|
||
|
// )
|
||
|
//
|
||
|
//////////////////////////////////////////////////////////////////////////////
|
||
|
HRESULT
|
||
|
CObjectManager::HrSearchForExistingCookie(
|
||
|
REFCLSID rclsidTypeIn,
|
||
|
OBJECTCOOKIE cookieParentIn,
|
||
|
LPCWSTR pcszNameIn,
|
||
|
OBJECTCOOKIE * pcookieOut
|
||
|
)
|
||
|
{
|
||
|
TraceFunc( "" );
|
||
|
|
||
|
Assert( pcszNameIn != NULL );
|
||
|
Assert( pcookieOut != NULL );
|
||
|
|
||
|
HRESULT hr = S_FALSE;
|
||
|
ULONG idx;
|
||
|
|
||
|
CStandardInfo * pcsi;
|
||
|
|
||
|
//
|
||
|
// Search the list.
|
||
|
//
|
||
|
for( idx = 1; idx < m_cCurrentUsed; idx ++ )
|
||
|
{
|
||
|
pcsi = m_pCookies[ idx ];
|
||
|
|
||
|
if ( pcsi != NULL )
|
||
|
{
|
||
|
if ( pcsi->m_cookieParent == cookieParentIn // matching parents
|
||
|
&& IsEqualIID( pcsi->m_clsidType, rclsidTypeIn ) // matching types
|
||
|
&& StrCmpI( pcsi->m_bstrName, pcszNameIn ) == 0 // matching names
|
||
|
)
|
||
|
{
|
||
|
//
|
||
|
// Found a match.
|
||
|
//
|
||
|
|
||
|
*pcookieOut = idx;
|
||
|
hr = S_OK;
|
||
|
|
||
|
break; // exit loop
|
||
|
|
||
|
} // if: match
|
||
|
|
||
|
} // if: cookie exists
|
||
|
|
||
|
} // while: pcsi
|
||
|
|
||
|
HRETURN( hr );
|
||
|
|
||
|
} // HrSearchForExistingCookie( )
|
||
|
|
||
|
|
||
|
//////////////////////////////////////////////////////////////////////////////
|
||
|
//
|
||
|
// HRESULT
|
||
|
// CObjectManager::HrDeleteInstanceAndChildren(
|
||
|
// OBJECTCOOKIE pcsiIn
|
||
|
// )
|
||
|
//
|
||
|
// Notes:
|
||
|
// This should be called while the ListLock is held!
|
||
|
//
|
||
|
//////////////////////////////////////////////////////////////////////////////
|
||
|
HRESULT
|
||
|
CObjectManager::HrDeleteInstanceAndChildren(
|
||
|
OBJECTCOOKIE cookieIn
|
||
|
)
|
||
|
{
|
||
|
TraceFunc1( "cookieIn = %#X", cookieIn );
|
||
|
|
||
|
ULONG idx;
|
||
|
CStandardInfo * pcsi;
|
||
|
|
||
|
HRESULT hr = S_OK;
|
||
|
|
||
|
hr = THR( HrDeleteCookie( cookieIn ) );
|
||
|
if ( FAILED( hr ) )
|
||
|
goto Cleanup;
|
||
|
|
||
|
for ( idx = 1; idx < m_cCurrentUsed; idx ++ )
|
||
|
{
|
||
|
pcsi = m_pCookies[ idx ];
|
||
|
|
||
|
if ( pcsi != NULL
|
||
|
&& pcsi->m_cookieParent == cookieIn )
|
||
|
{
|
||
|
hr = THR( HrDeleteInstanceAndChildren( idx ) );
|
||
|
if ( FAILED( hr ) )
|
||
|
goto Cleanup;
|
||
|
|
||
|
} // if:
|
||
|
|
||
|
} // while:
|
||
|
|
||
|
Cleanup:
|
||
|
HRETURN( hr );
|
||
|
|
||
|
} // HrDeleteInstanceAndChildren( )
|
||
|
|
||
|
//////////////////////////////////////////////////////////////////////////////
|
||
|
//
|
||
|
// HRESULT
|
||
|
// CObjectManager::HrCreateNewCookie(
|
||
|
// REFCLSID rclsidTypeIn
|
||
|
// OBJECTCOOKIE cookieParentIn,
|
||
|
// BSTR pcszNameIn,
|
||
|
// OBJECTCOOKIE * pcookieOut
|
||
|
// )
|
||
|
//
|
||
|
//////////////////////////////////////////////////////////////////////////////
|
||
|
HRESULT
|
||
|
CObjectManager::HrCreateNewCookie(
|
||
|
REFCLSID rclsidTypeIn,
|
||
|
OBJECTCOOKIE cookieParentIn,
|
||
|
LPCWSTR pcszNameIn,
|
||
|
OBJECTCOOKIE * pcookieOut
|
||
|
)
|
||
|
{
|
||
|
TraceFunc( "" );
|
||
|
|
||
|
HRESULT hr = E_UNEXPECTED;
|
||
|
|
||
|
CStandardInfo * pcsi = NULL;
|
||
|
|
||
|
Assert( pcookieOut != NULL );
|
||
|
|
||
|
*pcookieOut = 0;
|
||
|
|
||
|
//
|
||
|
// Create some space for it.
|
||
|
//
|
||
|
|
||
|
if ( m_cCurrentUsed == m_cAllocSize )
|
||
|
{
|
||
|
CStandardInfo ** pnew = (CStandardInfo **) TraceAlloc( HEAP_ZERO_MEMORY, sizeof(CStandardInfo *) * ( m_cAllocSize + COOKIE_BUFFER_GROW_SIZE ) );
|
||
|
if ( pnew == NULL )
|
||
|
goto OutOfMemory;
|
||
|
|
||
|
if ( m_pCookies != NULL )
|
||
|
{
|
||
|
CopyMemory( pnew, m_pCookies, sizeof(CStandardInfo *) * m_cCurrentUsed );
|
||
|
TraceFree( m_pCookies );
|
||
|
}
|
||
|
|
||
|
m_pCookies = pnew;
|
||
|
|
||
|
m_cAllocSize += COOKIE_BUFFER_GROW_SIZE;
|
||
|
|
||
|
if ( m_cCurrentUsed == 0 )
|
||
|
{
|
||
|
//
|
||
|
// Always skip zero.
|
||
|
//
|
||
|
m_cCurrentUsed = 1;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
pcsi = new CStandardInfo( );
|
||
|
if ( pcsi == NULL )
|
||
|
goto OutOfMemory;
|
||
|
|
||
|
hr = THR( pcsi->Init( ) );
|
||
|
if ( FAILED( hr ) )
|
||
|
goto Cleanup;
|
||
|
|
||
|
m_pCookies[ m_cCurrentUsed ] = pcsi;
|
||
|
|
||
|
//
|
||
|
// Initialize the rest of the structure.
|
||
|
//
|
||
|
|
||
|
pcsi->m_cookieParent = cookieParentIn;
|
||
|
pcsi->m_hrStatus = E_PENDING;
|
||
|
|
||
|
CopyMemory( &pcsi->m_clsidType, &rclsidTypeIn, sizeof( pcsi->m_clsidType ) );
|
||
|
|
||
|
if ( pcszNameIn != NULL )
|
||
|
{
|
||
|
pcsi->m_bstrName = TraceSysAllocString( pcszNameIn );
|
||
|
if ( pcsi->m_bstrName == NULL )
|
||
|
{
|
||
|
m_cCurrentUsed --;
|
||
|
goto OutOfMemory;
|
||
|
} // if: out of memory
|
||
|
}
|
||
|
|
||
|
Assert( pcsi->m_pci == NULL );
|
||
|
Assert( pcsi->m_pExtObjList == NULL );
|
||
|
|
||
|
//
|
||
|
// Keep it around and return SUCCESS!
|
||
|
//
|
||
|
|
||
|
pcsi = NULL;
|
||
|
*pcookieOut = m_cCurrentUsed;
|
||
|
m_cCurrentUsed ++;
|
||
|
hr = S_OK;
|
||
|
|
||
|
Cleanup:
|
||
|
if ( pcsi != NULL )
|
||
|
{
|
||
|
pcsi->Release( );
|
||
|
}
|
||
|
|
||
|
HRETURN( hr );
|
||
|
|
||
|
OutOfMemory:
|
||
|
hr = E_OUTOFMEMORY;
|
||
|
goto Cleanup;
|
||
|
|
||
|
} // HrCreateNewCookie( )
|