116 lines
2.9 KiB
C++
116 lines
2.9 KiB
C++
// SLNKDWF.cpp : Implementation of DLL Exports.
|
|
|
|
#include "stdafx.h"
|
|
|
|
#include "resource.h"
|
|
#include "SLNKDWF.h"
|
|
#include "dlldatax.h"
|
|
|
|
// Detect Memory Leaks
|
|
#ifdef _DEBUG
|
|
//CMemLeakDetect memLeakDetect;
|
|
#endif
|
|
|
|
HINSTANCE g_hInstance;
|
|
extern CString buildDATE;
|
|
extern CString buildTIME;
|
|
|
|
class CSLNKDWFModule : public CAtlDllModuleT< CSLNKDWFModule >
|
|
{
|
|
public :
|
|
DECLARE_LIBID(LIBID_SLNKDWFLib)
|
|
DECLARE_REGISTRY_APPID_RESOURCEID(IDR_SLNKDWF, "{71B32202-1B68-4E92-9C86-8F53FB298011}")
|
|
};
|
|
|
|
CSLNKDWFModule _AtlModule;
|
|
|
|
|
|
// DLL Entry Point
|
|
extern "C" BOOL WINAPI DllMain(HINSTANCE hInstance, DWORD dwReason, LPVOID lpReserved)
|
|
{
|
|
g_hInstance = hInstance;
|
|
char module[_MAX_PATH]; // HMODULE
|
|
|
|
GetModuleFileName(g_hInstance, module, sizeof(module));
|
|
|
|
char *txt;
|
|
switch (dwReason)
|
|
{
|
|
case DLL_PROCESS_ATTACH: txt = "DLL_PROCESS_ATTACH"; break;
|
|
case DLL_PROCESS_DETACH: txt = "DLL_PROCESS_DETACH"; break;
|
|
case DLL_THREAD_ATTACH: txt = "DLL_THREAD_ATTACH"; break;
|
|
case DLL_THREAD_DETACH: txt = "DLL_THREAD_DETACH"; break;
|
|
default: txt = "UNKNOWN reason";
|
|
}
|
|
|
|
DWORD pid = GetCurrentProcessId();
|
|
if (dwReason == DLL_PROCESS_ATTACH)
|
|
myDoTRACE("\nLoading SLNKDWF: built on %s, %s"
|
|
"\n DLL: %s"
|
|
"\n PID: %d"
|
|
// "\n Handle: %x\n"
|
|
, buildDATE, buildTIME, module, pid, hInstance);
|
|
// else
|
|
// myDoTRACE("\nUn/Loading SLNKDWF handle %x, reason %s", hInstance, txt);
|
|
|
|
#ifdef _MERGE_PROXYSTUB
|
|
if (!PrxDllMain(hInstance, dwReason, lpReserved))
|
|
return FALSE;
|
|
#endif
|
|
hInstance;
|
|
return _AtlModule.DllMain(dwReason, lpReserved);
|
|
}
|
|
|
|
|
|
// Used to determine whether the DLL can be unloaded by OLE
|
|
STDAPI DllCanUnloadNow(void)
|
|
{
|
|
#ifdef _MERGE_PROXYSTUB
|
|
HRESULT hr = PrxDllCanUnloadNow();
|
|
if (FAILED(hr))
|
|
return hr;
|
|
#endif
|
|
return _AtlModule.DllCanUnloadNow();
|
|
}
|
|
|
|
|
|
// Returns a class factory to create an object of the requested type
|
|
STDAPI DllGetClassObject(REFCLSID rclsid, REFIID riid, LPVOID* ppv)
|
|
{
|
|
#ifdef _MERGE_PROXYSTUB
|
|
if (PrxDllGetClassObject(rclsid, riid, ppv) == S_OK)
|
|
return S_OK;
|
|
#endif
|
|
return _AtlModule.DllGetClassObject(rclsid, riid, ppv);
|
|
}
|
|
|
|
|
|
// DllRegisterServer - Adds entries to the system registry
|
|
STDAPI DllRegisterServer(void)
|
|
{
|
|
// registers object, typelib and all interfaces in typelib
|
|
HRESULT hr = _AtlModule.DllRegisterServer();
|
|
#ifdef _MERGE_PROXYSTUB
|
|
if (FAILED(hr))
|
|
return hr;
|
|
hr = PrxDllRegisterServer();
|
|
#endif
|
|
return hr;
|
|
}
|
|
|
|
|
|
// DllUnregisterServer - Removes entries from the system registry
|
|
STDAPI DllUnregisterServer(void)
|
|
{
|
|
HRESULT hr = _AtlModule.DllUnregisterServer();
|
|
#ifdef _MERGE_PROXYSTUB
|
|
if (FAILED(hr))
|
|
return hr;
|
|
hr = PrxDllRegisterServer();
|
|
if (FAILED(hr))
|
|
return hr;
|
|
hr = PrxDllUnregisterServer();
|
|
#endif
|
|
return hr;
|
|
}
|