windows-server-2003/base/win32/fusion/tests/sxstest/dll1/dllmain.cpp

// Copyright (c) Microsoft Corporation
// dll1.cpp

#include "stdinc.h"
#include "resource.h"
#include "initguid.h"
#include "sxstest_idl.h"
#include "CFreeThreaded.h"
#include "CSingleThreaded.h"
#include "CApartmentThreaded.h"
#include "CBothThreaded.h"
#include "CSingleThreadedDual.h"
#include "dlldatax.h"
#include <stdio.h>
#define NUMBER_OF(x) (sizeof(x)/sizeof((x)[0]))

#ifdef _MERGE_PROXYSTUB
extern "C" HINSTANCE hProxyDll;
#endif

CComModule _Module;

#include "sxstest_trace.cpp"
#include "sxstest_formatguid.cpp"

BEGIN_OBJECT_MAP(ObjectMap)
    OBJECT_ENTRY(CLSID_CSxsTest_FreeThreaded, CFreeThreaded)
    OBJECT_ENTRY(CLSID_CSxsTest_SingleThreaded, CSingleThreaded)
    OBJECT_ENTRY(CLSID_CSxsTest_ApartmentThreaded, CApartmentThreaded)
    OBJECT_ENTRY(CLSID_CSxsTest_BothThreaded, CCBothThreaded)
    OBJECT_ENTRY(CLSID_CSxsTest_SingleThreadedDual, CSingleThreadedDual)
END_OBJECT_MAP()

/////////////////////////////////////////////////////////////////////////////
// DLL Entry Point

extern "C" BOOL WINAPI _DllMainCRTStartup(HINSTANCE hInstance, DWORD dwReason, LPVOID lpReserved);

extern "C"
BOOL WINAPI SxsTest_Dll1_DllMain(HINSTANCE hInstance, DWORD dwReason, LPVOID lpReserved)
{
    lpReserved;
#ifdef _MERGE_PROXYSTUB
    if (!PrxDllMain(hInstance, dwReason, lpReserved))
        return FALSE;
#endif
    if (dwReason == DLL_PROCESS_ATTACH)
    {
        _Module.Init(ObjectMap, hInstance /*, &LIBID_ATL_DLL1Lib */);
        DisableThreadLibraryCalls(hInstance);
    }
    else if (dwReason == DLL_PROCESS_DETACH)
        _Module.Term();
    return TRUE;    // ok
}

extern "C"
BOOL WINAPI DllEntry(HINSTANCE hInstance, DWORD dwReason, LPVOID lpReserved)
{
    DWORD Detach = DLL_PROCESS_DETACH;
    switch (dwReason)
    {
    case DLL_THREAD_ATTACH:
        Detach = DLL_THREAD_DETACH; // FALLTHROUGH
    case DLL_PROCESS_ATTACH:
        if (!_DllMainCRTStartup(hInstance, dwReason, lpReserved))
            return FALSE;
        if (!SxsTest_Dll1_DllMain(hInstance, dwReason, lpReserved))
        {
            _DllMainCRTStartup(hInstance, Detach, lpReserved);
            return FALSE;
        }
        return TRUE;

    case DLL_THREAD_DETACH:
    case DLL_PROCESS_DETACH:
        SxsTest_Dll1_DllMain(hInstance, dwReason, lpReserved);
        _DllMainCRTStartup(hInstance, dwReason, lpReserved);
        return TRUE;

    default:
        return FALSE;
    }
    return FALSE;
}

/////////////////////////////////////////////////////////////////////////////
// Used to determine whether the DLL can be unloaded by OLE

STDAPI DllCanUnloadNow(void)
{
#ifdef _MERGE_PROXYSTUB
    if (PrxDllCanUnloadNow() != S_OK)
        return S_FALSE;
#endif
    return (_Module.GetLockCount()==0) ? S_OK : S_FALSE;
}

/////////////////////////////////////////////////////////////////////////////
// Returns a class factory to create an object of the requested type

STDAPI DllGetClassObject(REFCLSID rclsid, REFIID riid, LPVOID* ppv)
{
    WCHAR ClsidBuffer[64];
    WCHAR IidBuffer[64];

    PrintComctl32Path("DllGetClassObject");
    FormatGuid(ClsidBuffer, NUMBER_OF(ClsidBuffer), rclsid);
    FormatGuid(IidBuffer, NUMBER_OF(IidBuffer), riid);
    Trace("DllGetClassObject(clsid={%ls}, iid={%ls}\n", ClsidBuffer, IidBuffer);
#ifdef _MERGE_PROXYSTUB
    if (PrxDllGetClassObject(rclsid, riid, ppv) == S_OK)
        return S_OK;
#endif
    return _Module.GetClassObject(rclsid, riid, ppv);
}

/////////////////////////////////////////////////////////////////////////////
// DllRegisterServer - Adds entries to the system registry

STDAPI DllRegisterServer(void)
{
#ifdef _MERGE_PROXYSTUB
    HRESULT hRes = PrxDllRegisterServer();
    if (FAILED(hRes))
        return hRes;
#endif
    // registers object, typelib and all interfaces in typelib
    return _Module.RegisterServer(FALSE);
}

/////////////////////////////////////////////////////////////////////////////
// DllUnregisterServer - Removes entries from the system registry

STDAPI DllUnregisterServer(void)
{
#ifdef _MERGE_PROXYSTUB
    PrxDllUnregisterServer();
#endif
    return _Module.UnregisterServer(TRUE);
}