4 * Copyright 2009 Andrew Hill <ash77@reactos.org>
6 * This library is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU Lesser General Public
8 * License as published by the Free Software Foundation; either
9 * version 2.1 of the License, or (at your option) any later version.
11 * This library is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 * Lesser General Public License for more details.
16 * You should have received a copy of the GNU Lesser General Public
17 * License along with this library; if not, write to the Free Software
18 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
27 // It is common to use this in ATL constructors. They only store this for later use, so the usage is safe.
28 #pragma warning(disable:4355)
32 #define _ATL_PACKING 8
35 #ifndef _ATL_FREE_THREADED
36 #ifndef _ATL_APARTMENT_THREADED
37 #ifndef _ATL_SINGLE_THREADED
38 #define _ATL_FREE_THREADED
47 #ifdef _ATL_DISABLE_NO_VTABLE
50 #define ATL_NO_VTABLE __declspec(novtable)
53 #define offsetofclass(base, derived) (reinterpret_cast<DWORD_PTR>(static_cast<base *>(reinterpret_cast<derived *>(_ATL_PACKING))) - _ATL_PACKING)
61 __declspec(selectany
) CAtlModule
*_pAtlModule
= NULL
;
62 __declspec(selectany
) CComModule
*_pModule
= NULL
;
63 extern CAtlComModule _AtlComModule
;
65 typedef HRESULT (WINAPI _ATL_CREATORFUNC
)(void *pv
, REFIID riid
, LPVOID
*ppv
);
66 typedef LPCTSTR (WINAPI _ATL_DESCRIPTIONFUNC
)();
67 typedef const struct _ATL_CATMAP_ENTRY
* (_ATL_CATMAPFUNC
)();
69 struct _ATL_OBJMAP_ENTRY30
72 HRESULT (WINAPI
*pfnUpdateRegistry
)(BOOL bRegister
);
73 _ATL_CREATORFUNC
*pfnGetClassObject
;
74 _ATL_CREATORFUNC
*pfnCreateInstance
;
77 _ATL_DESCRIPTIONFUNC
*pfnGetObjectDescription
;
78 _ATL_CATMAPFUNC
*pfnGetCategoryMap
;
79 void (WINAPI
*pfnObjectMain
)(bool bStarting
);
81 HRESULT WINAPI
RevokeClassObject()
85 return CoRevokeClassObject(dwRegister
);
88 HRESULT WINAPI
RegisterClassObject(DWORD dwClsContext
, DWORD dwFlags
)
94 if (pfnGetClassObject
== NULL
)
97 hResult
= pfnGetClassObject(reinterpret_cast<LPVOID
*>(pfnCreateInstance
), IID_IUnknown
, reinterpret_cast<LPVOID
*>(&p
));
98 if (SUCCEEDED(hResult
))
99 hResult
= CoRegisterClassObject(*pclsid
, p
, dwClsContext
, dwFlags
, &dwRegister
);
108 typedef _ATL_OBJMAP_ENTRY30 _ATL_OBJMAP_ENTRY
;
110 typedef void (__stdcall _ATL_TERMFUNC
)(DWORD_PTR dw
);
112 struct _ATL_TERMFUNC_ELEM
114 _ATL_TERMFUNC
*pFunc
;
116 _ATL_TERMFUNC_ELEM
*pNext
;
123 _ATL_TERMFUNC_ELEM
*m_pTermFuncs
;
124 CComCriticalSection m_csStaticDataInitAndTypeInfo
;
126 typedef _ATL_MODULE70 _ATL_MODULE
;
128 typedef HRESULT (WINAPI _ATL_CREATORARGFUNC
)(void *pv
, REFIID riid
, LPVOID
*ppv
, DWORD_PTR dw
);
130 #define _ATL_SIMPLEMAPENTRY ((ATL::_ATL_CREATORARGFUNC *)1)
132 struct _ATL_INTMAP_ENTRY
136 _ATL_CREATORARGFUNC
*pFunc
;
139 struct _AtlCreateWndData
143 _AtlCreateWndData
*m_pNext
;
146 struct _ATL_COM_MODULE70
149 HINSTANCE m_hInstTypeLib
;
150 _ATL_OBJMAP_ENTRY
**m_ppAutoObjMapFirst
;
151 _ATL_OBJMAP_ENTRY
**m_ppAutoObjMapLast
;
152 CComCriticalSection m_csObjMap
;
154 typedef _ATL_COM_MODULE70 _ATL_COM_MODULE
;
156 struct _ATL_WIN_MODULE70
159 CComCriticalSection m_csWindowCreate
;
160 _AtlCreateWndData
*m_pCreateWndList
;
162 CSimpleArray
<ATOM
> m_rgWindowClassAtoms
;
165 typedef _ATL_WIN_MODULE70 _ATL_WIN_MODULE
;
167 struct _ATL_REGMAP_ENTRY
173 HRESULT __stdcall
AtlWinModuleInit(_ATL_WIN_MODULE
*pWinModule
);
174 HRESULT __stdcall
AtlWinModuleTerm(_ATL_WIN_MODULE
*pWinModule
, HINSTANCE hInst
);
175 HRESULT __stdcall
AtlInternalQueryInterface(void *pThis
, const _ATL_INTMAP_ENTRY
*pEntries
, REFIID iid
, void **ppvObject
);
176 void __stdcall
AtlWinModuleAddCreateWndData(_ATL_WIN_MODULE
*pWinModule
, _AtlCreateWndData
*pData
, void *pObject
);
177 void *__stdcall
AtlWinModuleExtractCreateWndData(_ATL_WIN_MODULE
*pWinModule
);
178 HRESULT __stdcall
AtlComModuleGetClassObject(_ATL_COM_MODULE
*pComModule
, REFCLSID rclsid
, REFIID riid
, LPVOID
*ppv
);
180 template<class TLock
>
181 class CComCritSecLock
187 CComCritSecLock(TLock
&cs
, bool bInitialLock
= true) : m_cs(cs
)
212 ATLASSERT(!m_bLocked
);
213 hResult
= m_cs
.Lock();
225 ATLASSERT(m_bLocked
);
226 hResult
= m_cs
.Unlock();
235 inline BOOL WINAPI
InlineIsEqualUnknown(REFGUID rguid1
)
238 ((unsigned long *)&rguid1
)[0] == 0 &&
239 ((unsigned long *)&rguid1
)[1] == 0 &&
240 ((unsigned long *)&rguid1
)[2] == 0x000000C0 &&
241 ((unsigned long *)&rguid1
)[3] == 0x46000000);
244 class CComMultiThreadModelNoCS
247 typedef CComFakeCriticalSection AutoCriticalSection
;
248 typedef CComFakeCriticalSection CriticalSection
;
249 typedef CComMultiThreadModelNoCS ThreadModelNoCS
;
250 typedef CComFakeCriticalSection AutoDeleteCriticalSection
;
252 static ULONG WINAPI
Increment(LPLONG p
)
254 return InterlockedIncrement(p
);
257 static ULONG WINAPI
Decrement(LPLONG p
)
259 return InterlockedDecrement(p
);
263 class CComMultiThreadModel
266 typedef CComAutoCriticalSection AutoCriticalSection
;
267 typedef CComCriticalSection CriticalSection
;
268 typedef CComMultiThreadModelNoCS ThreadModelNoCS
;
269 typedef CComAutoDeleteCriticalSection AutoDeleteCriticalSection
;
271 static ULONG WINAPI
Increment(LPLONG p
)
273 return InterlockedIncrement(p
);
276 static ULONG WINAPI
Decrement(LPLONG p
)
278 return InterlockedDecrement(p
);
282 class CComSingleThreadModel
285 typedef CComFakeCriticalSection AutoCriticalSection
;
286 typedef CComFakeCriticalSection CriticalSection
;
287 typedef CComSingleThreadModel ThreadModelNoCS
;
288 typedef CComFakeCriticalSection AutoDeleteCriticalSection
;
290 static ULONG WINAPI
Increment(LPLONG p
)
295 static ULONG WINAPI
Decrement(LPLONG p
)
301 #if defined(_ATL_FREE_THREADED)
303 typedef CComMultiThreadModel CComObjectThreadModel
;
304 typedef CComMultiThreadModel CComGlobalsThreadModel
;
306 #elif defined(_ATL_APARTMENT_THREADED)
308 typedef CComSingleThreadModel CComObjectThreadModel
;
309 typedef CComMultiThreadModel CComGlobalsThreadModel
;
311 #elif defined(_ATL_SINGLE_THREADED)
313 typedef CComSingleThreadModel CComObjectThreadModel
;
314 typedef CComSingleThreadModel CComGlobalsThreadModel
;
317 #error No threading model
320 class CAtlModule
: public _ATL_MODULE
327 ATLASSERT(_pAtlModule
== NULL
);
329 cbSize
= sizeof(_ATL_MODULE
);
333 virtual LONG
GetLockCount()
340 return CComGlobalsThreadModel::Increment(&m_nLockCnt
);
343 virtual LONG
Unlock()
345 return CComGlobalsThreadModel::Decrement(&m_nLockCnt
);
348 virtual HRESULT
AddCommonRGSReplacements(IRegistrarBase
* /*pRegistrar*/) = 0;
350 HRESULT WINAPI
UpdateRegistryFromResource(LPCTSTR lpszRes
, BOOL bRegister
, struct _ATL_REGMAP_ENTRY
*pMapEntries
= NULL
)
352 CRegObject registrar
;
353 TCHAR modulePath
[MAX_PATH
];
356 hResult
= CommonInitRegistrar(registrar
, modulePath
, sizeof(modulePath
) / sizeof(modulePath
[0]), pMapEntries
);
360 if (bRegister
!= FALSE
)
361 hResult
= registrar
.ResourceRegisterSz(modulePath
, lpszRes
, _T("REGISTRY"));
363 hResult
= registrar
.ResourceUnregisterSz(modulePath
, lpszRes
, _T("REGISTRY"));
368 HRESULT WINAPI
UpdateRegistryFromResource(UINT nResID
, BOOL bRegister
, struct _ATL_REGMAP_ENTRY
*pMapEntries
= NULL
)
370 CRegObject registrar
;
371 TCHAR modulePath
[MAX_PATH
];
374 hResult
= CommonInitRegistrar(registrar
, modulePath
, sizeof(modulePath
) / sizeof(modulePath
[0]), pMapEntries
);
378 if (bRegister
!= FALSE
)
379 hResult
= registrar
.ResourceRegister(modulePath
, nResID
, _T("REGISTRY"));
381 hResult
= registrar
.ResourceRegister(modulePath
, nResID
, _T("REGISTRY"));
387 HRESULT
CommonInitRegistrar(CRegObject
®istrar
, TCHAR
*modulePath
, DWORD modulePathCount
, struct _ATL_REGMAP_ENTRY
*pMapEntries
)
393 hInstance
= _AtlBaseModule
.GetModuleInstance();
394 dwFLen
= GetModuleFileName(hInstance
, modulePath
, modulePathCount
);
395 if (dwFLen
== modulePathCount
)
396 return HRESULT_FROM_WIN32(ERROR_INSUFFICIENT_BUFFER
);
397 else if (dwFLen
== 0)
398 return HRESULT_FROM_WIN32(GetLastError());
400 if (pMapEntries
!= NULL
)
402 while (pMapEntries
->szKey
!= NULL
)
404 ATLASSERT(pMapEntries
->szData
!= NULL
);
405 hResult
= registrar
.AddReplacement(pMapEntries
->szKey
, pMapEntries
->szData
);
412 hResult
= AddCommonRGSReplacements(®istrar
);
416 hResult
= registrar
.AddReplacement(_T("Module"), modulePath
);
420 hResult
= registrar
.AddReplacement(_T("Module_Raw"), modulePath
);
428 __declspec(selectany
) GUID
CAtlModule::m_libid
= {0x0, 0x0, 0x0, {0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0} };
431 class CAtlModuleT
: public CAtlModule
435 virtual HRESULT
AddCommonRGSReplacements(IRegistrarBase
*pRegistrar
)
437 return pRegistrar
->AddReplacement(L
"APPID", T::GetAppId());
440 static LPCOLESTR
GetAppId()
446 class CAtlComModule
: public _ATL_COM_MODULE
451 GetModuleHandleExW(GET_MODULE_HANDLE_EX_FLAG_FROM_ADDRESS
, (LPCWSTR
)this, &m_hInstTypeLib
);
452 m_ppAutoObjMapFirst
= NULL
;
453 m_ppAutoObjMapLast
= NULL
;
454 if (FAILED(m_csObjMap
.Init()))
457 CAtlBaseModule::m_bInitFailed
= true;
460 cbSize
= sizeof(_ATL_COM_MODULE
);
472 ATLASSERT(m_ppAutoObjMapFirst
== NULL
);
473 ATLASSERT(m_ppAutoObjMapLast
== NULL
);
481 class CAtlDllModuleT
: public CAtlModuleT
<T
>
488 HRESULT
DllCanUnloadNow()
492 pThis
= static_cast<T
*>(this);
493 if (pThis
->GetLockCount() == 0)
498 HRESULT
DllGetClassObject(REFCLSID rclsid
, REFIID riid
, LPVOID
*ppv
)
502 pThis
= static_cast<T
*>(this);
503 return pThis
->GetClassObject(rclsid
, riid
, ppv
);
506 HRESULT
DllRegisterServer(BOOL bRegTypeLib
= TRUE
)
511 pThis
= static_cast<T
*>(this);
512 hResult
= pThis
->RegisterServer(bRegTypeLib
);
516 HRESULT
DllUnregisterServer(BOOL bUnRegTypeLib
= TRUE
)
521 pThis
= static_cast<T
*>(this);
522 hResult
= pThis
->UnregisterServer(bUnRegTypeLib
);
526 HRESULT
GetClassObject(REFCLSID rclsid
, REFIID riid
, LPVOID
*ppv
)
528 return AtlComModuleGetClassObject(&_AtlComModule
, rclsid
, riid
, ppv
);
532 class CComModule
: public CAtlModuleT
<CComModule
>
535 _ATL_OBJMAP_ENTRY
*m_pObjMap
;
539 ATLASSERT(_pModule
== NULL
);
541 _pModule
->m_pObjMap
= NULL
;
549 HRESULT
Init(_ATL_OBJMAP_ENTRY
*p
, HINSTANCE
/* h */, const GUID
*plibid
)
551 _ATL_OBJMAP_ENTRY
*objectMapEntry
;
556 if (p
!= reinterpret_cast<_ATL_OBJMAP_ENTRY
*>(-1))
562 while (objectMapEntry
->pclsid
!= NULL
)
564 objectMapEntry
->pfnObjectMain(true);
574 _ATL_OBJMAP_ENTRY
*objectMapEntry
;
576 if (m_pObjMap
!= NULL
)
578 objectMapEntry
= m_pObjMap
;
579 while (objectMapEntry
->pclsid
!= NULL
)
581 if (objectMapEntry
->pCF
!= NULL
)
582 objectMapEntry
->pCF
->Release();
583 objectMapEntry
->pCF
= NULL
;
584 objectMapEntry
->pfnObjectMain(false);
590 HRESULT
GetClassObject(REFCLSID rclsid
, REFIID riid
, LPVOID
*ppv
)
592 _ATL_OBJMAP_ENTRY
*objectMapEntry
;
595 ATLASSERT(ppv
!= NULL
);
599 if (m_pObjMap
!= NULL
)
601 objectMapEntry
= m_pObjMap
;
602 while (objectMapEntry
->pclsid
!= NULL
)
604 if (objectMapEntry
->pfnGetClassObject
!= NULL
&& InlineIsEqualGUID(rclsid
, *objectMapEntry
->pclsid
) != FALSE
)
606 if (objectMapEntry
->pCF
== NULL
)
608 CComCritSecLock
<CComCriticalSection
> lock(_AtlComModule
.m_csObjMap
, true);
610 if (objectMapEntry
->pCF
== NULL
)
611 hResult
= objectMapEntry
->pfnGetClassObject(reinterpret_cast<void *>(objectMapEntry
->pfnCreateInstance
), IID_IUnknown
, reinterpret_cast<LPVOID
*>(&objectMapEntry
->pCF
));
613 if (objectMapEntry
->pCF
!= NULL
)
614 hResult
= objectMapEntry
->pCF
->QueryInterface(riid
, ppv
);
623 HRESULT
RegisterServer(BOOL bRegTypeLib
= FALSE
, const CLSID
*pCLSID
= NULL
)
625 _ATL_OBJMAP_ENTRY
*objectMapEntry
;
629 objectMapEntry
= m_pObjMap
;
630 if (objectMapEntry
!= NULL
)
632 while (objectMapEntry
->pclsid
!= NULL
)
634 if (pCLSID
== NULL
|| IsEqualGUID(*pCLSID
, *objectMapEntry
->pclsid
) != FALSE
)
636 hResult
= objectMapEntry
->pfnUpdateRegistry(TRUE
);
646 HRESULT
UnregisterServer(BOOL bUnRegTypeLib
, const CLSID
*pCLSID
= NULL
)
648 _ATL_OBJMAP_ENTRY
*objectMapEntry
;
652 objectMapEntry
= m_pObjMap
;
653 if (objectMapEntry
!= NULL
)
655 while (objectMapEntry
->pclsid
!= NULL
)
657 if (pCLSID
== NULL
|| IsEqualGUID(*pCLSID
, *objectMapEntry
->pclsid
) != FALSE
)
659 hResult
= objectMapEntry
->pfnUpdateRegistry(FALSE
); //unregister
669 HRESULT
DllCanUnloadNow()
671 if (GetLockCount() == 0)
676 HRESULT
DllGetClassObject(REFCLSID rclsid
, REFIID riid
, LPVOID
*ppv
)
678 return GetClassObject(rclsid
, riid
, ppv
);
681 HRESULT
DllRegisterServer(BOOL bRegTypeLib
= TRUE
)
683 return RegisterServer(bRegTypeLib
);
686 HRESULT
DllUnregisterServer(BOOL bUnRegTypeLib
= TRUE
)
688 return UnregisterServer(bUnRegTypeLib
);
693 class CAtlWinModule
: public _ATL_WIN_MODULE
700 hResult
= AtlWinModuleInit(this);
703 CAtlBaseModule::m_bInitFailed
= true;
715 AtlWinModuleTerm(this, _AtlBaseModule
.GetModuleInstance());
718 void AddCreateWndData(_AtlCreateWndData
*pData
, void *pObject
)
720 AtlWinModuleAddCreateWndData(this, pData
, pObject
);
723 void *ExtractCreateWndData()
725 return AtlWinModuleExtractCreateWndData(this);
729 extern CAtlWinModule _AtlWinModule
;
749 CComPtr(const CComPtr
<T
> &lp
)
762 T
*operator = (T
*lp
)
772 T
*operator = (const CComPtr
<T
> &lp
)
809 ATLASSERT(p
== NULL
);
820 ATLASSERT(p
!= NULL
);
830 CComBSTR(LPCOLESTR pSrc
)
835 m_str
= ::SysAllocString(pSrc
);
839 ::SysFreeString(m_str
);
844 class CComVariant
: public tagVARIANT
859 return ::VariantClear(this);
863 inline HRESULT __stdcall
AtlAdvise(IUnknown
*pUnkCP
, IUnknown
*pUnk
, const IID
&iid
, LPDWORD pdw
)
865 CComPtr
<IConnectionPointContainer
> container
;
866 CComPtr
<IConnectionPoint
> connectionPoint
;
871 hResult
= pUnkCP
->QueryInterface(IID_IConnectionPointContainer
, (void **)&container
);
874 hResult
= container
->FindConnectionPoint(iid
, &connectionPoint
);
877 return connectionPoint
->Advise(pUnk
, pdw
);
880 inline HRESULT __stdcall
AtlUnadvise(IUnknown
*pUnkCP
, const IID
&iid
, DWORD dw
)
882 CComPtr
<IConnectionPointContainer
> container
;
883 CComPtr
<IConnectionPoint
> connectionPoint
;
888 hResult
= pUnkCP
->QueryInterface(IID_IConnectionPointContainer
, (void **)&container
);
891 hResult
= container
->FindConnectionPoint(iid
, &connectionPoint
);
894 return connectionPoint
->Unadvise(dw
);
897 inline HRESULT __stdcall
AtlInternalQueryInterface(void *pThis
, const _ATL_INTMAP_ENTRY
*pEntries
, REFIID iid
, void **ppvObject
)
900 IUnknown
*resultInterface
;
903 ATLASSERT(pThis
!= NULL
&& pEntries
!= NULL
);
904 if (pThis
== NULL
|| pEntries
== NULL
)
906 ATLASSERT(ppvObject
!= NULL
);
907 if (ppvObject
== NULL
)
910 if (InlineIsEqualUnknown(iid
))
912 resultInterface
= reinterpret_cast<IUnknown
*>(reinterpret_cast<char *>(pThis
) + pEntries
[0].dw
);
913 *ppvObject
= resultInterface
;
914 resultInterface
->AddRef();
919 while (pEntries
[i
].pFunc
!= 0)
921 if (pEntries
[i
].piid
== NULL
|| InlineIsEqualGUID(iid
, *pEntries
[i
].piid
))
923 if (pEntries
[i
].pFunc
== reinterpret_cast<_ATL_CREATORARGFUNC
*>(1))
925 ATLASSERT(pEntries
[i
].piid
!= NULL
);
926 resultInterface
= reinterpret_cast<IUnknown
*>(reinterpret_cast<char *>(pThis
) + pEntries
[i
].dw
);
927 *ppvObject
= resultInterface
;
928 resultInterface
->AddRef();
933 hResult
= pEntries
[i
].pFunc(pThis
, iid
, ppvObject
, 0);
934 if (hResult
== S_OK
|| (FAILED(hResult
) && pEntries
[i
].piid
!= NULL
))
942 return E_NOINTERFACE
;
945 inline HRESULT __stdcall
AtlWinModuleInit(_ATL_WIN_MODULE
*pWinModule
)
947 if (pWinModule
== NULL
)
949 pWinModule
->m_pCreateWndList
= NULL
;
950 return pWinModule
->m_csWindowCreate
.Init();
953 inline HRESULT __stdcall
AtlWinModuleTerm(_ATL_WIN_MODULE
*pWinModule
, HINSTANCE hInst
)
955 if (pWinModule
== NULL
)
957 pWinModule
->m_csWindowCreate
.Term();
961 inline void __stdcall
AtlWinModuleAddCreateWndData(_ATL_WIN_MODULE
*pWinModule
, _AtlCreateWndData
*pData
, void *pObject
)
963 CComCritSecLock
<CComCriticalSection
> lock(pWinModule
->m_csWindowCreate
, true);
965 ATLASSERT(pWinModule
!= NULL
);
966 ATLASSERT(pObject
!= NULL
);
968 pData
->m_pThis
= pObject
;
969 pData
->m_dwThreadID
= ::GetCurrentThreadId();
970 pData
->m_pNext
= pWinModule
->m_pCreateWndList
;
971 pWinModule
->m_pCreateWndList
= pData
;
974 inline void *__stdcall
AtlWinModuleExtractCreateWndData(_ATL_WIN_MODULE
*pWinModule
)
976 CComCritSecLock
<CComCriticalSection
> lock(pWinModule
->m_csWindowCreate
, true);
978 _AtlCreateWndData
*currentEntry
;
979 _AtlCreateWndData
**previousLink
;
982 ATLASSERT(pWinModule
!= NULL
);
985 threadID
= GetCurrentThreadId();
986 currentEntry
= pWinModule
->m_pCreateWndList
;
987 previousLink
= &pWinModule
->m_pCreateWndList
;
988 while (currentEntry
!= NULL
)
990 if (currentEntry
->m_dwThreadID
== threadID
)
992 *previousLink
= currentEntry
->m_pNext
;
993 result
= currentEntry
->m_pThis
;
996 previousLink
= ¤tEntry
->m_pNext
;
997 currentEntry
= currentEntry
->m_pNext
;
1004 #ifndef _ATL_NO_AUTOMATIC_NAMESPACE
1005 using namespace ATL
;
1006 #endif //!_ATL_NO_AUTOMATIC_NAMESPACE