atl30.c File Reference

#include <stdio.h>
#include "objidl.h"
#include "rpcproxy.h"
#include "wine/atlbase.h"
#include "wine/atlwin.h"
#include "wine/debug.h"

Include dependency graph for atl30.c:

Go to the source code of this file.

Macros
#define	COBJMACROS
#define	ATLVer1Size   FIELD_OFFSET(_ATL_MODULEW, dwAtlBuildVer)

Functions
	WINE_DEFAULT_DEBUG_CHANNEL (atl)
HRESULT WINAPI	AtlModuleInit (_ATL_MODULEW *pM, _ATL_OBJMAP_ENTRYW *p, HINSTANCE h)
static _ATL_OBJMAP_ENTRYW_V1 *	get_objmap_entry (_ATL_MODULEW *mod, unsigned int index)
HRESULT WINAPI	AtlModuleLoadTypeLib (_ATL_MODULEW *pM, LPCOLESTR lpszIndex, BSTR *pbstrPath, ITypeLib **ppTypeLib)
HRESULT WINAPI	AtlModuleTerm (_ATL_MODULE *pM)
HRESULT WINAPI	AtlModuleRegisterClassObjects (_ATL_MODULEW *pM, DWORD dwClsContext, DWORD dwFlags)
HRESULT WINAPI	AtlModuleUnregisterServerEx (_ATL_MODULEW *pM, BOOL bUnRegTypeLib, const CLSID *pCLSID)
HRESULT WINAPI	AtlModuleRegisterServer (_ATL_MODULEW *pM, BOOL bRegTypeLib, const CLSID *clsid)
HRESULT WINAPI	AtlModuleGetClassObject (_ATL_MODULEW *pm, REFCLSID rclsid, REFIID riid, LPVOID *ppv)
HRESULT WINAPI	AtlModuleRegisterTypeLib (_ATL_MODULEW *pm, LPCOLESTR lpszIndex)
HRESULT WINAPI	AtlModuleRevokeClassObjects (_ATL_MODULEW *pm)
HRESULT WINAPI	AtlModuleUnregisterServer (_ATL_MODULEW *pm, const CLSID *clsid)
ATOM WINAPI	AtlModuleRegisterWndClassInfoA (_ATL_MODULEA *pm, _ATL_WNDCLASSINFOA *wci, WNDPROC *pProc)
ATOM WINAPI	AtlModuleRegisterWndClassInfoW (_ATL_MODULEW *pm, _ATL_WNDCLASSINFOW *wci, WNDPROC *pProc)
void WINAPI	AtlModuleAddCreateWndData (_ATL_MODULEW *pM, _AtlCreateWndData *pData, void *pvObject)
void *WINAPI	AtlModuleExtractCreateWndData (_ATL_MODULEW *pM)
HRESULT WINAPI	AtlModuleUpdateRegistryFromResourceD (_ATL_MODULEW *pM, LPCOLESTR lpszRes, BOOL bRegister, struct _ATL_REGMAP_ENTRY *pMapEntries, IRegistrar *pReg)
static HRESULT WINAPI	RegistrarCF_QueryInterface (IClassFactory *iface, REFIID riid, void **ppvObject)
static ULONG WINAPI	RegistrarCF_AddRef (IClassFactory *iface)
static ULONG WINAPI	RegistrarCF_Release (IClassFactory *iface)
static HRESULT WINAPI	RegistrarCF_CreateInstance (IClassFactory *iface, LPUNKNOWN pUnkOuter, REFIID riid, void **ppv)
static HRESULT WINAPI	RegistrarCF_LockServer (IClassFactory *iface, BOOL lock)
HRESULT WINAPI	DllGetClassObject (REFCLSID clsid, REFIID riid, LPVOID *ppvObject)
HRESULT WINAPI	DllRegisterServer (void)
HRESULT WINAPI	DllUnregisterServer (void)
HRESULT WINAPI	DllCanUnloadNow (void)

Variables
HINSTANCE	atl_instance
static const IClassFactoryVtbl	IRegistrarCFVtbl
static IClassFactory	RegistrarCF = { &IRegistrarCFVtbl }

Macro Definition Documentation

ATLVer1Size

#define ATLVer1Size   FIELD_OFFSET(_ATL_MODULEW, dwAtlBuildVer)

Definition at line 37 of file atl30.c.

COBJMACROS

#define COBJMACROS

Definition at line 24 of file atl30.c.

Function Documentation

AtlModuleAddCreateWndData()

void WINAPI AtlModuleAddCreateWndData	(	_ATL_MODULEW *	pM,
		_AtlCreateWndData *	pData,
		void *	pvObject
	)

Definition at line 405 of file atl30.c.

{
    TRACE("(%p, %p, %p)\n", pM, pData, pvObject);

    pData->m_pThis = pvObject;
    pData->m_dwThreadID = GetCurrentThreadId();

    EnterCriticalSection(&pM->m_csWindowCreate);
    pData->m_pNext = pM->m_pCreateWndList;
    pM->m_pCreateWndList = pData;
    LeaveCriticalSection(&pM->m_csWindowCreate);
}

Referenced by test_winmodule().

AtlModuleExtractCreateWndData()

void* WINAPI AtlModuleExtractCreateWndData

(

_ATL_MODULEW *

pM

)

Definition at line 425 of file atl30.c.

{
    _AtlCreateWndData **ppData;
    void *ret = NULL;

    TRACE("(%p)\n", pM);

    EnterCriticalSection(&pM->m_csWindowCreate);

    for(ppData = &pM->m_pCreateWndList; *ppData!=NULL; ppData = &(*ppData)->m_pNext)
    {
        if ((*ppData)->m_dwThreadID == GetCurrentThreadId())
        {
            _AtlCreateWndData *pData = *ppData;
            *ppData = pData->m_pNext;
            ret = pData->m_pThis;
            break;
        }
    }

    LeaveCriticalSection(&pM->m_csWindowCreate);
    return ret;
}

Referenced by test_winmodule().

AtlModuleGetClassObject()

HRESULT WINAPI AtlModuleGetClassObject	(	_ATL_MODULEW *	pm,
		REFCLSID	rclsid,
		REFIID	riid,
		LPVOID *	ppv
	)

Definition at line 227 of file atl30.c.

{
    _ATL_OBJMAP_ENTRYW_V1 *obj;
    int i;
    HRESULT hres = CLASS_E_CLASSNOTAVAILABLE;

    TRACE("%p %s %s %p\n", pm, debugstr_guid(rclsid), debugstr_guid(riid), ppv);

    if (pm == NULL)
        return E_INVALIDARG;

    for (i = 0; (obj = get_objmap_entry( pm, i )) != NULL; i++)
    {
        if (IsEqualCLSID(obj->pclsid, rclsid))
        {
            TRACE("found object %i\n", i);
            if (obj->pfnGetClassObject)
            {
                if (!obj->pCF)
                    hres = obj->pfnGetClassObject(obj->pfnCreateInstance,
                                                  &IID_IUnknown,
                                                  (void **)&obj->pCF);
                if (obj->pCF)
                    hres = IUnknown_QueryInterface(obj->pCF, riid, ppv);
                break;
            }
        }
    }

    WARN("no class object found for %s\n", debugstr_guid(rclsid));

    return hres;
}

AtlModuleInit()

HRESULT WINAPI AtlModuleInit	(	_ATL_MODULEW *	pM,
		_ATL_OBJMAP_ENTRYW *	p,
		HINSTANCE	h
	)

Definition at line 39 of file atl30.c.

{
    INT i;
    UINT size;

    TRACE("(%p %p %p)\n", pM, p, h);

    size = pM->cbSize;
    switch (size)
    {
    case ATLVer1Size:
    case sizeof(_ATL_MODULEW):
#ifdef _WIN64
    case sizeof(_ATL_MODULEW) + sizeof(void *):
#endif
        break;
    default:
        WARN("Unknown structure version (size %i)\n",size);
        return E_INVALIDARG;
    }

    memset(pM,0,pM->cbSize);
    pM->cbSize = size;
    pM->m_hInst = h;
    pM->m_hInstResource = h;
    pM->m_hInstTypeLib = h;
    pM->m_pObjMap = p;
    pM->m_hHeap = GetProcessHeap();

    InitializeCriticalSection(&pM->u.m_csTypeInfoHolder);
    InitializeCriticalSection(&pM->m_csWindowCreate);
    InitializeCriticalSection(&pM->m_csObjMap);

    /* call mains */
    i = 0;
    if (pM->m_pObjMap != NULL  && size > ATLVer1Size)
    {
        while (pM->m_pObjMap[i].pclsid != NULL)
        {
            TRACE("Initializing object %i %p\n",i,p[i].pfnObjectMain);
            if (p[i].pfnObjectMain)
                p[i].pfnObjectMain(TRUE);
            i++;
        }
    }

    return S_OK;
}

Referenced by test_StructSize(), test_term(), test_winclassinfo(), and test_winmodule().

AtlModuleLoadTypeLib()

HRESULT WINAPI AtlModuleLoadTypeLib	(	_ATL_MODULEW *	pM,
		LPCOLESTR	lpszIndex,
		BSTR *	pbstrPath,
		ITypeLib **	ppTypeLib
	)

Definition at line 101 of file atl30.c.

{
    TRACE("(%p, %s, %p, %p)\n", pM, debugstr_w(lpszIndex), pbstrPath, ppTypeLib);

    if (!pM)
        return E_INVALIDARG;

    return AtlLoadTypeLib(pM->m_hInstTypeLib, lpszIndex, pbstrPath, ppTypeLib);
}

AtlModuleRegisterClassObjects()

HRESULT WINAPI AtlModuleRegisterClassObjects	(	_ATL_MODULEW *	pM,
		DWORD	dwClsContext,
		DWORD	dwFlags
	)

Definition at line 133 of file atl30.c.

{
    _ATL_OBJMAP_ENTRYW_V1 *obj;
    int i=0;

    TRACE("(%p %i %i)\n",pM, dwClsContext, dwFlags);

    if (pM == NULL)
        return E_INVALIDARG;

    while ((obj = get_objmap_entry( pM, i++ )))
    {
        IUnknown* pUnknown;
        HRESULT rc;

        TRACE("Registering object %i\n",i);
        if (obj->pfnGetClassObject)
        {
            rc = obj->pfnGetClassObject(obj->pfnCreateInstance, &IID_IUnknown,
                                   (LPVOID*)&pUnknown);
            if (SUCCEEDED (rc) )
            {
                rc = CoRegisterClassObject(obj->pclsid, pUnknown, dwClsContext,
                                           dwFlags, &obj->dwRegister);

                if (FAILED (rc) )
                    WARN("Failed to register object %i: 0x%08x\n", i, rc);

                if (pUnknown)
                    IUnknown_Release(pUnknown);
            }
        }
    }

   return S_OK;
}

AtlModuleRegisterServer()

HRESULT WINAPI AtlModuleRegisterServer	(	_ATL_MODULEW *	pM,
		BOOL	bRegTypeLib,
		const CLSID *	clsid
	)

Definition at line 181 of file atl30.c.

{
    const _ATL_OBJMAP_ENTRYW_V1 *obj;
    int i;
    HRESULT hRes;

    TRACE("%p %d %s\n", pM, bRegTypeLib, debugstr_guid(clsid));

    if (pM == NULL)
        return E_INVALIDARG;

    for (i = 0; (obj = get_objmap_entry( pM, i )) != NULL; i++) /* register CLSIDs */
    {
        if (!clsid || IsEqualCLSID(obj->pclsid, clsid))
        {
            TRACE("Registering clsid %s\n", debugstr_guid(obj->pclsid));
            hRes = obj->pfnUpdateRegistry(TRUE); /* register */
            if (FAILED(hRes))
                return hRes;

            if(pM->cbSize > ATLVer1Size) {
                const struct _ATL_CATMAP_ENTRY *catmap;

                catmap = ((const _ATL_OBJMAP_ENTRYW*)obj)->pfnGetCategoryMap();
                if(catmap) {
                    hRes = AtlRegisterClassCategoriesHelper(obj->pclsid, catmap, TRUE);
                    if(FAILED(hRes))
                        return hRes;
                }
            }
        }
    }

    if (bRegTypeLib)
    {
        hRes = AtlRegisterTypeLib(pM->m_hInstTypeLib, NULL);
        if (FAILED(hRes))
            return hRes;
    }

    return S_OK;
}

AtlModuleRegisterTypeLib()

HRESULT WINAPI AtlModuleRegisterTypeLib	(	_ATL_MODULEW *	pm,
		LPCOLESTR	lpszIndex
	)

Definition at line 265 of file atl30.c.

{
    TRACE("%p %s\n", pm, debugstr_w(lpszIndex));

    if (!pm)
        return E_INVALIDARG;

    return AtlRegisterTypeLib(pm->m_hInstTypeLib, lpszIndex);
}

AtlModuleRegisterWndClassInfoA()

ATOM WINAPI AtlModuleRegisterWndClassInfoA	(	_ATL_MODULEA *	pm,
		_ATL_WNDCLASSINFOA *	wci,
		WNDPROC *	pProc
	)

Definition at line 298 of file atl30.c.

{
    ATOM atom;

    FIXME("%p %p %p semi-stub\n", pm, wci, pProc);

    atom = wci->m_atom;
    if (!atom)
    {
        WNDCLASSEXA wc;

        TRACE("wci->m_wc.lpszClassName = %s\n", wci->m_wc.lpszClassName);

        if (wci->m_lpszOrigName)
            FIXME( "subclassing %s not implemented\n", debugstr_a(wci->m_lpszOrigName));

        if (!wci->m_wc.lpszClassName)
        {
            sprintf(wci->m_szAutoName, "ATL:%p", wci);
            TRACE("auto-generated class name %s\n", wci->m_szAutoName);
            wci->m_wc.lpszClassName = wci->m_szAutoName;
        }

        atom = GetClassInfoExA(pm->m_hInst, wci->m_wc.lpszClassName, &wc);
        if (!atom)
        {
            wci->m_wc.hInstance = pm->m_hInst;
            wci->m_wc.hCursor   = LoadCursorA( wci->m_bSystemCursor ? NULL : pm->m_hInst,
                                               wci->m_lpszCursorID );
            atom = RegisterClassExA(&wci->m_wc);
        }
        wci->pWndProc = wci->m_wc.lpfnWndProc;
        wci->m_atom = atom;
    }

    if (wci->m_lpszOrigName) *pProc = wci->pWndProc;

    TRACE("returning 0x%04x\n", atom);
    return atom;
}

AtlModuleRegisterWndClassInfoW()

ATOM WINAPI AtlModuleRegisterWndClassInfoW	(	_ATL_MODULEW *	pm,
		_ATL_WNDCLASSINFOW *	wci,
		WNDPROC *	pProc
	)

Definition at line 357 of file atl30.c.

{
    ATOM atom;

    FIXME("%p %p %p semi-stub\n", pm, wci, pProc);

    atom = wci->m_atom;
    if (!atom)
    {
        WNDCLASSEXW wc;

        TRACE("wci->m_wc.lpszClassName = %s\n", debugstr_w(wci->m_wc.lpszClassName));

        if (wci->m_lpszOrigName)
            FIXME( "subclassing %s not implemented\n", debugstr_w(wci->m_lpszOrigName));

        if (!wci->m_wc.lpszClassName)
        {
#ifndef __REACTOS__
            swprintf(wci->m_szAutoName, ARRAY_SIZE(wci->m_szAutoName), L"ATL:%p", wci);
#else
            swprintf(wci->m_szAutoName, L"ATL:%p", wci);
#endif
            TRACE("auto-generated class name %s\n", debugstr_w(wci->m_szAutoName));
            wci->m_wc.lpszClassName = wci->m_szAutoName;
        }

        atom = GetClassInfoExW(pm->m_hInst, wci->m_wc.lpszClassName, &wc);
        if (!atom)
        {
            wci->m_wc.hInstance = pm->m_hInst;
            wci->m_wc.hCursor   = LoadCursorW( wci->m_bSystemCursor ? NULL : pm->m_hInst,
                                               wci->m_lpszCursorID );
            atom = RegisterClassExW(&wci->m_wc);
        }
        wci->pWndProc = wci->m_wc.lpfnWndProc;
        wci->m_atom = atom;
    }

    if (wci->m_lpszOrigName) *pProc = wci->pWndProc;

    TRACE("returning 0x%04x\n", atom);
    return atom;
}

Referenced by test_winclassinfo().

AtlModuleRevokeClassObjects()

HRESULT WINAPI AtlModuleRevokeClassObjects

(

_ATL_MODULEW *

pm

)

Definition at line 278 of file atl30.c.

{
    FIXME("%p\n", pm);
    return E_FAIL;
}

AtlModuleTerm()

HRESULT WINAPI AtlModuleTerm

(

_ATL_MODULE *

pM

)

Definition at line 112 of file atl30.c.

{
    _ATL_TERMFUNC_ELEM *iter, *tmp;

    TRACE("(%p)\n", pM);

    if (pM->cbSize > ATLVer1Size)
    {
        iter = pM->m_pTermFuncs;

        while(iter) {
            iter->pFunc(iter->dw);
            tmp = iter;
            iter = iter->pNext;
            HeapFree(GetProcessHeap(), 0, tmp);
        }
    }

    return S_OK;
}

Referenced by test_term().

AtlModuleUnregisterServer()

HRESULT WINAPI AtlModuleUnregisterServer	(	_ATL_MODULEW *	pm,
		const CLSID *	clsid
	)

Definition at line 287 of file atl30.c.

{
    FIXME("%p %s\n", pm, debugstr_guid(clsid));
    return E_FAIL;
}

AtlModuleUnregisterServerEx()

HRESULT WINAPI AtlModuleUnregisterServerEx	(	_ATL_MODULEW *	pM,
		BOOL	bUnRegTypeLib,
		const CLSID *	pCLSID
	)

Definition at line 171 of file atl30.c.

{
    FIXME("(%p, %i, %p) stub\n", pM, bUnRegTypeLib, pCLSID);
    return S_OK;
}

AtlModuleUpdateRegistryFromResourceD()

HRESULT WINAPI AtlModuleUpdateRegistryFromResourceD	(	_ATL_MODULEW *	pM,
		LPCOLESTR	lpszRes,
		BOOL	bRegister,
		struct _ATL_REGMAP_ENTRY *	pMapEntries,
		IRegistrar *	pReg
	)

Definition at line 453 of file atl30.c.

{
    TRACE("(%p %s %d %p %p)\n", pM, debugstr_w(lpszRes), bRegister, pMapEntries, pReg);

    return AtlUpdateRegistryFromResourceD(pM->m_hInst, lpszRes, bRegister, pMapEntries, pReg);
}

DllCanUnloadNow()

HRESULT WINAPI DllCanUnloadNow

(

void

)

Definition at line 613 of file atl30.c.

{
    return S_FALSE;
}

DllGetClassObject()

HRESULT WINAPI DllGetClassObject	(	REFCLSID	clsid,
		REFIID	riid,
		LPVOID *	ppvObject
	)

Definition at line 574 of file atl30.c.

{
    TRACE("(%s %s %p)\n", debugstr_guid(clsid), debugstr_guid(riid), ppvObject);

    if(IsEqualGUID(&CLSID_Registrar, clsid))
        return IClassFactory_QueryInterface( &RegistrarCF, riid, ppvObject );

    FIXME("Not supported class %s\n", debugstr_guid(clsid));
    return CLASS_E_CLASSNOTAVAILABLE;
}

DllRegisterServer()

HRESULT WINAPI DllRegisterServer

(

void

)

Definition at line 588 of file atl30.c.

{
#ifdef __REACTOS__
    /* Note: we can't use __wine_register_server here because it uses CLSID_Registrar which isn't registred yet */
    return do_register_server(TRUE);
#else
    return __wine_register_resources( atl_instance );
#endif
}

DllUnregisterServer()

HRESULT WINAPI DllUnregisterServer

(

void

)

Definition at line 601 of file atl30.c.

{
#ifdef __REACTOS__
    return do_register_server(FALSE);
#else
    return __wine_unregister_resources( atl_instance );
#endif
}

get_objmap_entry()

static _ATL_OBJMAP_ENTRYW_V1* get_objmap_entry ( _ATL_MODULEW *  mod, unsigned int  index  )

static

Definition at line 88 of file atl30.c.

{
    _ATL_OBJMAP_ENTRYW_V1 *ret;

    if (mod->cbSize == ATLVer1Size)
        ret = (_ATL_OBJMAP_ENTRYW_V1 *)mod->m_pObjMap + index;
    else
        ret = (_ATL_OBJMAP_ENTRYW_V1 *)(mod->m_pObjMap + index);

    if (!ret->pclsid) ret = NULL;
    return ret;
}

Referenced by AtlModuleGetClassObject(), AtlModuleRegisterClassObjects(), and AtlModuleRegisterServer().

RegistrarCF_AddRef()

static ULONG WINAPI RegistrarCF_AddRef ( IClassFactory *  iface )

static

Definition at line 474 of file atl30.c.

{
    return 2;
}

RegistrarCF_CreateInstance()

static HRESULT WINAPI RegistrarCF_CreateInstance ( IClassFactory *  iface, LPUNKNOWN  pUnkOuter, REFIID  riid, void **  ppv  )

static

Definition at line 484 of file atl30.c.

{
    IRegistrar *registrar;
    HRESULT hres;

    TRACE("(%p)->(%s %p)\n", iface, debugstr_guid(riid), ppv);

    if(pUnkOuter) {
        *ppv = NULL;
        return CLASS_E_NOAGGREGATION;
    }

    hres = AtlCreateRegistrar(&registrar);
    if(FAILED(hres))
        return hres;

    hres = IRegistrar_QueryInterface(registrar, riid, ppv);
    IRegistrar_Release(registrar);
    return hres;
}

RegistrarCF_LockServer()

static HRESULT WINAPI RegistrarCF_LockServer ( IClassFactory *  iface, BOOL  lock  )

static

Definition at line 506 of file atl30.c.

{
    TRACE("(%p)->(%x)\n", iface, lock);
    return S_OK;
}

RegistrarCF_QueryInterface()

static HRESULT WINAPI RegistrarCF_QueryInterface ( IClassFactory *  iface, REFIID  riid, void **  ppvObject  )

static

Definition at line 461 of file atl30.c.

{
    TRACE("(%p)->(%s %p)\n", iface, debugstr_guid(riid), ppvObject);

    if(IsEqualGUID(&IID_IUnknown, riid) || IsEqualGUID(&IID_IClassFactory, riid)) {
        *ppvObject = iface;
        IClassFactory_AddRef( iface );
        return S_OK;
    }

    return E_NOINTERFACE;
}

RegistrarCF_Release()

static ULONG WINAPI RegistrarCF_Release ( IClassFactory *  iface )

static

Definition at line 479 of file atl30.c.

{
    return 1;
}

WINE_DEFAULT_DEBUG_CHANNEL()

WINE_DEFAULT_DEBUG_CHANNEL

(

atl

)

Variable Documentation

atl_instance

HINSTANCE atl_instance

Definition at line 36 of file atl.c.

Referenced by DllMain(), DllRegisterServer(), and DllUnregisterServer().

IRegistrarCFVtbl

const IClassFactoryVtbl IRegistrarCFVtbl

static

Initial value:

= {
    RegistrarCF_QueryInterface,
    RegistrarCF_AddRef,
    RegistrarCF_Release,
    RegistrarCF_CreateInstance,
    RegistrarCF_LockServer
}

Definition at line 512 of file atl30.c.

RegistrarCF

IClassFactory RegistrarCF = { &IRegistrarCFVtbl }

static

Definition at line 520 of file atl30.c.

Referenced by DllGetClassObject().