2 * Implementation of Active Template Library (atl.dll)
4 * Copyright 2004 Aric Stewart for CodeWeavers
5 * Copyright 2005 Jacek Caban
7 * This library is free software; you can redistribute it and/or
8 * modify it under the terms of the GNU Lesser General Public
9 * License as published by the Free Software Foundation; either
10 * version 2.1 of the License, or (at your option) any later version.
12 * This library is distributed in the hope that it will be useful,
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15 * Lesser General Public License for more details.
17 * You should have received a copy of the GNU Lesser General Public
18 * License along with this library; if not, write to the Free Software
19 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
31 #include "wine/debug.h"
32 #include "wine/unicode.h"
34 WINE_DEFAULT_DEBUG_CHANNEL(atl);
36 static HINSTANCE hInst;
38 BOOL WINAPI DllMain(HINSTANCE hinstDLL, DWORD fdwReason, LPVOID lpvReserved)
40 TRACE("(0x%p, %d, %p)\n",hinstDLL,fdwReason,lpvReserved);
42 if (fdwReason == DLL_PROCESS_ATTACH) {
43 DisableThreadLibraryCalls(hinstDLL);
49 #define ATLVer1Size FIELD_OFFSET(_ATL_MODULEW, dwAtlBuildVer)
51 HRESULT WINAPI AtlModuleInit(_ATL_MODULEW* pM, _ATL_OBJMAP_ENTRYW* p, HINSTANCE h)
56 TRACE("(%p %p %p)\n", pM, p, h);
62 case sizeof(_ATL_MODULEW):
64 case sizeof(_ATL_MODULEW) + sizeof(void *):
68 WARN("Unknown structure version (size %i)\n",size);
72 memset(pM,0,pM->cbSize);
75 pM->m_hInstResource = h;
76 pM->m_hInstTypeLib = h;
78 pM->m_hHeap = GetProcessHeap();
80 InitializeCriticalSection(&pM->u.m_csTypeInfoHolder);
81 InitializeCriticalSection(&pM->m_csWindowCreate);
82 InitializeCriticalSection(&pM->m_csObjMap);
86 if (pM->m_pObjMap != NULL && size > ATLVer1Size)
88 while (pM->m_pObjMap[i].pclsid != NULL)
90 TRACE("Initializing object %i %p\n",i,p[i].pfnObjectMain);
91 if (p[i].pfnObjectMain)
92 p[i].pfnObjectMain(TRUE);
100 static _ATL_OBJMAP_ENTRYW_V1 *get_objmap_entry( _ATL_MODULEW *mod, unsigned int index )
102 _ATL_OBJMAP_ENTRYW_V1 *ret;
104 if (mod->cbSize == ATLVer1Size)
105 ret = (_ATL_OBJMAP_ENTRYW_V1 *)mod->m_pObjMap + index;
107 ret = (_ATL_OBJMAP_ENTRYW_V1 *)(mod->m_pObjMap + index);
109 if (!ret->pclsid) ret = NULL;
113 HRESULT WINAPI AtlModuleLoadTypeLib(_ATL_MODULEW *pM, LPCOLESTR lpszIndex,
114 BSTR *pbstrPath, ITypeLib **ppTypeLib)
116 TRACE("(%p, %s, %p, %p)\n", pM, debugstr_w(lpszIndex), pbstrPath, ppTypeLib);
121 return AtlLoadTypeLib(pM->m_hInstTypeLib, lpszIndex, pbstrPath, ppTypeLib);
124 HRESULT WINAPI AtlModuleTerm(_ATL_MODULE *pM)
126 _ATL_TERMFUNC_ELEM *iter = pM->m_pTermFuncs, *tmp;
131 iter->pFunc(iter->dw);
134 HeapFree(GetProcessHeap(), 0, tmp);
140 HRESULT WINAPI AtlModuleAddTermFunc(_ATL_MODULEW *pM, _ATL_TERMFUNC *pFunc, DWORD_PTR dw)
142 _ATL_TERMFUNC_ELEM *termfunc_elem;
144 TRACE("(%p %p %ld)\n", pM, pFunc, dw);
146 termfunc_elem = HeapAlloc(GetProcessHeap(), 0, sizeof(_ATL_TERMFUNC_ELEM));
147 termfunc_elem->pFunc = pFunc;
148 termfunc_elem->dw = dw;
149 termfunc_elem->pNext = pM->m_pTermFuncs;
151 pM->m_pTermFuncs = termfunc_elem;
156 HRESULT WINAPI AtlModuleRegisterClassObjects(_ATL_MODULEW *pM, DWORD dwClsContext,
159 _ATL_OBJMAP_ENTRYW_V1 *obj;
162 TRACE("(%p %i %i)\n",pM, dwClsContext, dwFlags);
167 while ((obj = get_objmap_entry( pM, i++ )))
172 TRACE("Registering object %i\n",i);
173 if (obj->pfnGetClassObject)
175 rc = obj->pfnGetClassObject(obj->pfnCreateInstance, &IID_IUnknown,
179 rc = CoRegisterClassObject(obj->pclsid, pUnknown, dwClsContext,
180 dwFlags, &obj->dwRegister);
183 WARN("Failed to register object %i: 0x%08x\n", i, rc);
186 IUnknown_Release(pUnknown);
194 HRESULT WINAPI AtlModuleUnregisterServerEx(_ATL_MODULEW* pM, BOOL bUnRegTypeLib, const CLSID* pCLSID)
196 FIXME("(%p, %i, %p) stub\n", pM, bUnRegTypeLib, pCLSID);
200 /***********************************************************************
201 * AtlModuleRegisterServer [ATL.@]
204 HRESULT WINAPI AtlModuleRegisterServer(_ATL_MODULEW* pM, BOOL bRegTypeLib, const CLSID* clsid)
206 const _ATL_OBJMAP_ENTRYW_V1 *obj;
210 TRACE("%p %d %s\n", pM, bRegTypeLib, debugstr_guid(clsid));
215 for (i = 0; (obj = get_objmap_entry( pM, i )) != NULL; i++) /* register CLSIDs */
217 if (!clsid || IsEqualCLSID(obj->pclsid, clsid))
219 TRACE("Registering clsid %s\n", debugstr_guid(obj->pclsid));
220 hRes = obj->pfnUpdateRegistry(TRUE); /* register */
224 if(pM->cbSize > ATLVer1Size) {
225 const struct _ATL_CATMAP_ENTRY *catmap;
227 catmap = ((const _ATL_OBJMAP_ENTRYW*)obj)->pfnGetCategoryMap();
229 hRes = AtlRegisterClassCategoriesHelper(obj->pclsid, catmap, TRUE);
239 hRes = AtlRegisterTypeLib(pM->m_hInstTypeLib, NULL);
247 /***********************************************************************
248 * AtlModuleGetClassObject [ATL.@]
250 HRESULT WINAPI AtlModuleGetClassObject(_ATL_MODULEW *pm, REFCLSID rclsid,
251 REFIID riid, LPVOID *ppv)
253 _ATL_OBJMAP_ENTRYW_V1 *obj;
255 HRESULT hres = CLASS_E_CLASSNOTAVAILABLE;
257 TRACE("%p %s %s %p\n", pm, debugstr_guid(rclsid), debugstr_guid(riid), ppv);
262 for (i = 0; (obj = get_objmap_entry( pm, i )) != NULL; i++)
264 if (IsEqualCLSID(obj->pclsid, rclsid))
266 TRACE("found object %i\n", i);
267 if (obj->pfnGetClassObject)
270 hres = obj->pfnGetClassObject(obj->pfnCreateInstance,
274 hres = IUnknown_QueryInterface(obj->pCF, riid, ppv);
280 WARN("no class object found for %s\n", debugstr_guid(rclsid));
285 /***********************************************************************
286 * AtlModuleGetClassObject [ATL.@]
288 HRESULT WINAPI AtlModuleRegisterTypeLib(_ATL_MODULEW *pm, LPCOLESTR lpszIndex)
290 TRACE("%p %s\n", pm, debugstr_w(lpszIndex));
295 return AtlRegisterTypeLib(pm->m_hInstTypeLib, lpszIndex);
298 /***********************************************************************
299 * AtlModuleRevokeClassObjects [ATL.@]
301 HRESULT WINAPI AtlModuleRevokeClassObjects(_ATL_MODULEW *pm)
307 /***********************************************************************
308 * AtlModuleUnregisterServer [ATL.@]
310 HRESULT WINAPI AtlModuleUnregisterServer(_ATL_MODULEW *pm, const CLSID *clsid)
312 FIXME("%p %s\n", pm, debugstr_guid(clsid));
316 /***********************************************************************
317 * AtlModuleRegisterWndClassInfoA [ATL.@]
319 * See AtlModuleRegisterWndClassInfoW.
321 ATOM WINAPI AtlModuleRegisterWndClassInfoA(_ATL_MODULEA *pm, _ATL_WNDCLASSINFOA *wci, WNDPROC *pProc)
325 FIXME("%p %p %p semi-stub\n", pm, wci, pProc);
332 TRACE("wci->m_wc.lpszClassName = %s\n", wci->m_wc.lpszClassName);
334 if (wci->m_lpszOrigName)
335 FIXME( "subclassing %s not implemented\n", debugstr_a(wci->m_lpszOrigName));
337 if (!wci->m_wc.lpszClassName)
339 snprintf(wci->m_szAutoName, sizeof(wci->m_szAutoName), "ATL%08lx", (UINT_PTR)wci);
340 TRACE("auto-generated class name %s\n", wci->m_szAutoName);
341 wci->m_wc.lpszClassName = wci->m_szAutoName;
344 atom = GetClassInfoExA(pm->m_hInst, wci->m_wc.lpszClassName, &wc);
347 wci->m_wc.hInstance = pm->m_hInst;
348 wci->m_wc.hCursor = LoadCursorA( wci->m_bSystemCursor ? NULL : pm->m_hInst,
349 wci->m_lpszCursorID );
350 atom = RegisterClassExA(&wci->m_wc);
352 wci->pWndProc = wci->m_wc.lpfnWndProc;
356 if (wci->m_lpszOrigName) *pProc = wci->pWndProc;
358 TRACE("returning 0x%04x\n", atom);
362 /***********************************************************************
363 * AtlModuleRegisterWndClassInfoW [ATL.@]
366 * pm [IO] Information about the module registering the window.
367 * wci [IO] Information about the window being registered.
368 * pProc [O] Window procedure of the registered class.
371 * Atom representing the registered class.
374 * Can be called multiple times without error, unlike RegisterClassEx().
376 * If the class name is NULL, then a class with a name of "ATLxxxxxxxx" is
377 * registered, where the 'x's represent a unique value.
380 ATOM WINAPI AtlModuleRegisterWndClassInfoW(_ATL_MODULEW *pm, _ATL_WNDCLASSINFOW *wci, WNDPROC *pProc)
384 FIXME("%p %p %p semi-stub\n", pm, wci, pProc);
391 TRACE("wci->m_wc.lpszClassName = %s\n", debugstr_w(wci->m_wc.lpszClassName));
393 if (wci->m_lpszOrigName)
394 FIXME( "subclassing %s not implemented\n", debugstr_w(wci->m_lpszOrigName));
396 if (!wci->m_wc.lpszClassName)
398 static const WCHAR szFormat[] = {'A','T','L','%','0','8','l','x',0};
399 snprintfW(wci->m_szAutoName, sizeof(wci->m_szAutoName)/sizeof(WCHAR), szFormat, (UINT_PTR)wci);
400 TRACE("auto-generated class name %s\n", debugstr_w(wci->m_szAutoName));
401 wci->m_wc.lpszClassName = wci->m_szAutoName;
404 atom = GetClassInfoExW(pm->m_hInst, wci->m_wc.lpszClassName, &wc);
407 wci->m_wc.hInstance = pm->m_hInst;
408 wci->m_wc.hCursor = LoadCursorW( wci->m_bSystemCursor ? NULL : pm->m_hInst,
409 wci->m_lpszCursorID );
410 atom = RegisterClassExW(&wci->m_wc);
412 wci->pWndProc = wci->m_wc.lpfnWndProc;
416 if (wci->m_lpszOrigName) *pProc = wci->pWndProc;
418 TRACE("returning 0x%04x\n", atom);
422 /***********************************************************************
423 * AtlModuleAddCreateWndData [ATL.@]
425 void WINAPI AtlModuleAddCreateWndData(_ATL_MODULEW *pM, _AtlCreateWndData *pData, void* pvObject)
427 TRACE("(%p, %p, %p)\n", pM, pData, pvObject);
429 pData->m_pThis = pvObject;
430 pData->m_dwThreadID = GetCurrentThreadId();
432 EnterCriticalSection(&pM->m_csWindowCreate);
433 pData->m_pNext = pM->m_pCreateWndList;
434 pM->m_pCreateWndList = pData;
435 LeaveCriticalSection(&pM->m_csWindowCreate);
438 /***********************************************************************
439 * AtlModuleExtractCreateWndData [ATL.@]
441 * NOTE: Tests show that this function extracts one of _AtlCreateWndData
442 * records from the current thread from a list
445 void* WINAPI AtlModuleExtractCreateWndData(_ATL_MODULEW *pM)
447 _AtlCreateWndData **ppData;
452 EnterCriticalSection(&pM->m_csWindowCreate);
454 for(ppData = &pM->m_pCreateWndList; *ppData!=NULL; ppData = &(*ppData)->m_pNext)
456 if ((*ppData)->m_dwThreadID == GetCurrentThreadId())
458 _AtlCreateWndData *pData = *ppData;
459 *ppData = pData->m_pNext;
460 ret = pData->m_pThis;
465 LeaveCriticalSection(&pM->m_csWindowCreate);
469 /***********************************************************************
470 * AtlModuleUpdateRegistryFromResourceD [ATL.@]
473 HRESULT WINAPI AtlModuleUpdateRegistryFromResourceD(_ATL_MODULEW* pM, LPCOLESTR lpszRes,
474 BOOL bRegister, struct _ATL_REGMAP_ENTRY* pMapEntries, IRegistrar* pReg)
476 TRACE("(%p %s %d %p %p)\n", pM, debugstr_w(lpszRes), bRegister, pMapEntries, pReg);
478 return AtlUpdateRegistryFromResourceD(pM->m_hInst, lpszRes, bRegister, pMapEntries, pReg);
481 static HRESULT WINAPI RegistrarCF_QueryInterface(IClassFactory *iface, REFIID riid, void **ppvObject)
483 TRACE("(%p)->(%s %p)\n", iface, debugstr_guid(riid), ppvObject);
485 if(IsEqualGUID(&IID_IUnknown, riid) || IsEqualGUID(&IID_IClassFactory, riid)) {
487 IClassFactory_AddRef( iface );
491 return E_NOINTERFACE;
494 static ULONG WINAPI RegistrarCF_AddRef(IClassFactory *iface)
499 static ULONG WINAPI RegistrarCF_Release(IClassFactory *iface)
504 static HRESULT WINAPI RegistrarCF_CreateInstance(IClassFactory *iface, LPUNKNOWN pUnkOuter,
505 REFIID riid, void **ppv)
507 IRegistrar *registrar;
510 TRACE("(%p)->(%s %p)\n", iface, debugstr_guid(riid), ppv);
514 return CLASS_E_NOAGGREGATION;
517 hres = AtlCreateRegistrar(®istrar);
521 hres = IRegistrar_QueryInterface(registrar, riid, ppv);
522 IRegistrar_Release(registrar);
526 static HRESULT WINAPI RegistrarCF_LockServer(IClassFactory *iface, BOOL lock)
528 TRACE("(%p)->(%x)\n", iface, lock);
532 static const IClassFactoryVtbl IRegistrarCFVtbl = {
533 RegistrarCF_QueryInterface,
536 RegistrarCF_CreateInstance,
537 RegistrarCF_LockServer
540 static IClassFactory RegistrarCF = { &IRegistrarCFVtbl };
542 /**************************************************************
543 * DllGetClassObject (ATL.2)
545 HRESULT WINAPI DllGetClassObject(REFCLSID clsid, REFIID riid, LPVOID *ppvObject)
547 TRACE("(%s %s %p)\n", debugstr_guid(clsid), debugstr_guid(riid), ppvObject);
549 if(IsEqualGUID(&CLSID_Registrar, clsid))
550 return IClassFactory_QueryInterface( &RegistrarCF, riid, ppvObject );
552 FIXME("Not supported class %s\n", debugstr_guid(clsid));
553 return CLASS_E_CLASSNOTAVAILABLE;
556 /***********************************************************************
557 * DllRegisterServer (ATL.@)
559 HRESULT WINAPI DllRegisterServer(void)
561 return __wine_register_resources( hInst );
564 /***********************************************************************
565 * DllUnRegisterServer (ATL.@)
567 HRESULT WINAPI DllUnregisterServer(void)
569 return __wine_unregister_resources( hInst );
572 /***********************************************************************
573 * DllCanUnloadNow (ATL.@)
575 HRESULT WINAPI DllCanUnloadNow(void)
580 /***********************************************************************
581 * AtlGetVersion [ATL.@]
583 DWORD WINAPI AtlGetVersion(void *pReserved)