summaryrefslogtreecommitdiff
path: root/NetCmpts/NetCmpts.cpp
blob: 56f00a2b5a93eee01ea65c579f5e544869dd9ed6 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
// NetCmpts.cpp : Implementation of DLL Exports.


// Note: Proxy/Stub Information
//      To build a separate proxy/stub DLL, 
//      run nmake -f NetCmptsps.mk in the project directory.

#include "stdafx.h"
#include "resource.h"
#include <initguid.h>
#include "NetCmpts.h"

#include "NetCmpts_i.c"
#include "../interfaces/CmptIfaces_i.c"

#include "ClearInetCache.h"

#include "..\Common\Defines.h"

CComModule _Module;

BEGIN_OBJECT_MAP(ObjectMap)
	OBJECT_ENTRY(CLSID_ClearInetCache, CClearInetCache)
END_OBJECT_MAP()

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

extern "C"
BOOL WINAPI DllMain(HINSTANCE hInstance, DWORD dwReason, LPVOID /*lpReserved*/)
{
    if (dwReason == DLL_PROCESS_ATTACH)
    {
        _Module.Init(ObjectMap, hInstance, &LIBID_NightSecNetCmpts);
        DisableThreadLibraryCalls(hInstance);
    }
    else if (dwReason == DLL_PROCESS_DETACH)
        _Module.Term();
    return TRUE;    // ok
}

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

STDAPI DllCanUnloadNow(void)
{
    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)
{
    return _Module.GetClassObject(rclsid, riid, ppv);
}

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

STDAPI DllRegisterServer(void)
{
	// <<<< Addition
	// InstallShield Doesn't call this for the current thread
	// And if we're using the _ATL_DLL (COM Based) we need this
	HRESULT hr = CoInitialize(NULL);
	// End >>>>

    // registers object, typelib and all interfaces in typelib
    return _Module.RegisterServer(TRUE);

	// <<<< Addition
	if(SUCCEEDED(hr))
		CoUninitialize();

	return hr;
	// End >>>>

}

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

STDAPI DllUnregisterServer(void)
{
	// <<<< Addition
	HRESULT hr = CoInitialize(NULL);
	// End >>>>

    return _Module.UnregisterServer(TRUE);

	// <<<< Addition
	if(SUCCEEDED(hr))
		CoUninitialize();

	return S_OK;
	// End >>>>
}