158 lines
4.6 KiB
C++
158 lines
4.6 KiB
C++
#include "pch.hxx"
|
|
#define DEFINE_STRING_CONSTANTS
|
|
#include <commctrl.h>
|
|
#include "dllmain.h"
|
|
#include "shared.h"
|
|
#include "strconst.h"
|
|
#include "demand.h"
|
|
|
|
#define ICC_FLAGS (ICC_LISTVIEW_CLASSES|ICC_PROGRESS_CLASS|ICC_NATIVEFNTCTL_CLASS)
|
|
|
|
// --------------------------------------------------------------------------------
|
|
// Globals - Object count and lock count
|
|
// --------------------------------------------------------------------------------
|
|
CRITICAL_SECTION g_csDllMain = {0};
|
|
LONG g_cRef = 0;
|
|
LONG g_cLock = 0;
|
|
HINSTANCE g_hInstImp = NULL;
|
|
LPMALLOC g_pMalloc = NULL;
|
|
|
|
SYSTEM_INFO g_SystemInfo={0};
|
|
OSVERSIONINFO g_OSInfo={0};
|
|
BOOL g_fAttached = FALSE;
|
|
|
|
inline BOOL fIsNT5() { return((g_OSInfo.dwPlatformId == VER_PLATFORM_WIN32_NT) && (g_OSInfo.dwMajorVersion >= 5)); }
|
|
|
|
// --------------------------------------------------------------------------------
|
|
// Debug Globals
|
|
// --------------------------------------------------------------------------------
|
|
#ifdef DEBUG
|
|
DWORD dwDOUTLevel=0;
|
|
DWORD dwDOUTLMod=0;
|
|
DWORD dwDOUTLModLevel=0;
|
|
#endif
|
|
|
|
static HINSTANCE s_hInst = NULL;
|
|
|
|
// --------------------------------------------------------------------------------
|
|
// InitGlobalVars
|
|
// --------------------------------------------------------------------------------
|
|
void InitGlobalVars(void)
|
|
{
|
|
INITCOMMONCONTROLSEX icex = { sizeof(icex), ICC_FLAGS };
|
|
|
|
// Initialize Global Critical Sections
|
|
InitializeCriticalSection(&g_csDllMain);
|
|
g_fAttached = TRUE;
|
|
|
|
// Create OLE Task Memory Allocator
|
|
CoGetMalloc(1, &g_pMalloc);
|
|
Assert(g_pMalloc);
|
|
|
|
// Initialize Demand-loaded Libs
|
|
InitDemandLoadedLibs();
|
|
|
|
InitCommonControlsEx(&icex);
|
|
}
|
|
|
|
// --------------------------------------------------------------------------------
|
|
// FreeGlobalVars
|
|
// --------------------------------------------------------------------------------
|
|
void FreeGlobalVars(void)
|
|
{
|
|
// Free libraries that demand.cpp loaded
|
|
FreeDemandLoadedLibs();
|
|
|
|
// Release Global Memory allocator
|
|
SafeRelease(g_pMalloc);
|
|
|
|
// Delete Global Critical Sections
|
|
g_fAttached = FALSE;
|
|
DeleteCriticalSection(&g_csDllMain);
|
|
}
|
|
|
|
// --------------------------------------------------------------------------------
|
|
// Dll Entry Point
|
|
// --------------------------------------------------------------------------------
|
|
int APIENTRY DllMain(HINSTANCE hInst, DWORD dwReason, LPVOID lpReserved)
|
|
{
|
|
// Handle Attach - detach reason
|
|
switch (dwReason)
|
|
{
|
|
case DLL_PROCESS_ATTACH:
|
|
// Set global instance handle
|
|
|
|
s_hInst = hInst;
|
|
|
|
// Initialize Global Variables
|
|
InitGlobalVars();
|
|
|
|
g_hInstImp = LoadLangDll(s_hInst, c_szOEResDll, fIsNT5());
|
|
|
|
// No Thread Attach Stuff
|
|
// SideAssert(DisableThreadLibraryCalls(hInst));
|
|
|
|
// Done
|
|
break;
|
|
|
|
case DLL_PROCESS_DETACH:
|
|
// Free Global Variables
|
|
FreeGlobalVars();
|
|
|
|
// Done
|
|
break;
|
|
}
|
|
|
|
// Done
|
|
return TRUE;
|
|
}
|
|
|
|
// --------------------------------------------------------------------------------
|
|
// DllAddRef
|
|
// --------------------------------------------------------------------------------
|
|
ULONG DllAddRef(void)
|
|
{
|
|
InterlockedIncrement(&g_cRef);
|
|
return g_cRef;
|
|
}
|
|
|
|
// --------------------------------------------------------------------------------
|
|
// DllRelease
|
|
// --------------------------------------------------------------------------------
|
|
ULONG DllRelease(void)
|
|
{
|
|
InterlockedDecrement(&g_cRef);
|
|
return g_cRef;
|
|
}
|
|
|
|
// --------------------------------------------------------------------------------
|
|
// DllCanUnloadNow
|
|
// --------------------------------------------------------------------------------
|
|
STDAPI DllCanUnloadNow(void)
|
|
{
|
|
HRESULT hr;
|
|
|
|
if(!g_fAttached) // critacal sections was deleted (or not created): we defently can be unloaded
|
|
return S_OK;
|
|
|
|
EnterCriticalSection(&g_csDllMain);
|
|
DebugTrace("DllCanUnloadNow: %s - Reference Count: %d, LockServer Count: %d\n", __FILE__, g_cRef, g_cLock);
|
|
hr = (0 == g_cRef && 0 == g_cLock) ? S_OK : S_FALSE;
|
|
LeaveCriticalSection(&g_csDllMain);
|
|
return hr;
|
|
}
|
|
|
|
STDAPI DllRegisterServer(void)
|
|
{
|
|
HRESULT hr;
|
|
hr = CallRegInstall(s_hInst, s_hInst, c_szReg, NULL);
|
|
return(hr);
|
|
}
|
|
|
|
STDAPI DllUnregisterServer(void)
|
|
{
|
|
HRESULT hr;
|
|
hr = CallRegInstall(s_hInst, s_hInst, c_szUnReg, NULL);
|
|
return(hr);
|
|
}
|