Intravision/reactos
1
0
Fork 0
mirror of https://github.com/reactos/reactos.git synced 2024-11-20 06:15:26 +00:00
Code Issues Projects Releases Packages Wiki Activity

[ATL]

Browse source 
- Implement CComClassFactorySingleton

svn path=/trunk/; revision=68203
This commit is contained in:
Thomas Faber 2015-06-20 12:08:13 +00:00
parent dddbfc03a0
commit cf07db224f
1 changed files with 94 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

94
reactos/lib/atl/atlcom.h
View file

@ -29,6 +29,7 @@ class CComEnum;
#define DECLARE_CLASSFACTORY_EX(cf) typedef ATL::CComCreator<ATL::CComObjectCached<cf> > _ClassFactoryCreatorClass;
#define DECLARE_CLASSFACTORY() DECLARE_CLASSFACTORY_EX(ATL::CComClassFactory)
#define DECLARE_CLASSFACTORY_SINGLETON(obj) DECLARE_CLASSFACTORY_EX(ATL::CComClassFactorySingleton<obj>)
class CComObjectRootBase
{
@ -470,6 +471,11 @@ public:
{
}
virtual ~CComObjectCached()
{
this->FinalRelease();
}
STDMETHOD_(ULONG, AddRef)()
{
ULONG newRefCount;
@ -496,6 +502,38 @@ public:
{
return this->_InternalQueryInterface(iid, ppvObject);
}
static HRESULT WINAPI CreateInstance(CComObjectCached<Base> **pp)
{
CComObjectCached<Base> *newInstance;
HRESULT hResult;
ATLASSERT(pp != NULL);
if (pp == NULL)
return E_POINTER;
hResult = E_OUTOFMEMORY;
newInstance = NULL;
ATLTRY(newInstance = new CComObjectCached<Base>())
if (newInstance != NULL)
{
newInstance->SetVoid(NULL);
newInstance->InternalFinalConstructAddRef();
hResult = newInstance->_AtlInitialConstruct();
if (SUCCEEDED(hResult))
hResult = newInstance->FinalConstruct();
if (SUCCEEDED(hResult))
hResult = newInstance->_AtlFinalConstruct();
newInstance->InternalFinalConstructRelease();
if (hResult != S_OK)
{
delete newInstance;
newInstance = NULL;
}
}
*pp = newInstance;
return hResult;
}
};
#define BEGIN_COM_MAP(x) \
@ -670,6 +708,62 @@ public:
END_COM_MAP()
};
template <class T>
class CComClassFactorySingleton :
public CComClassFactory
{
public:
HRESULT m_hrCreate;
IUnknown *m_spObj;
public:
CComClassFactorySingleton() :
m_hrCreate(S_OK),
m_spObj(NULL)
{
}
STDMETHOD(CreateInstance)(LPUNKNOWN pUnkOuter, REFIID riid, void **ppvObj)
{
HRESULT hResult;
if (ppvObj == NULL)
return E_POINTER;
*ppvObj = NULL;
if (pUnkOuter != NULL)
hResult = CLASS_E_NOAGGREGATION;
else if (m_hrCreate == S_OK && m_spObj == NULL)
{
_SEH2_TRY
{
Lock();
if (m_hrCreate == S_OK && m_spObj == NULL)
{
CComObjectCached<T> *pObj;
m_hrCreate = CComObjectCached<T>::CreateInstance(&pObj);
if (SUCCEEDED(m_hrCreate))
{
m_hrCreate = pObj->QueryInterface(IID_IUnknown, reinterpret_cast<PVOID *>(&m_spObj));
if (FAILED(m_hrCreate))
delete pObj;
}
}
}
_SEH2_FINALLY
{
Unlock();
}
_SEH2_END;
}
if (m_hrCreate == S_OK)
hResult = m_spObj->QueryInterface(riid, ppvObj);
else
hResult = m_hrCreate;
return hResult;
}
};
template <class T, const CLSID *pclsid = &CLSID_NULL>
class CComCoClass
{