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
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
|
// NSCmpts.cpp : Implementation of DLL Exports.
// Note: Proxy/Stub Information
// To build a separate proxy/stub DLL,
// run nmake -f NSCmptsps.mk in the project directory.
#include "stdafx.h"
#include "resource.h"
#include <initguid.h>
#include "NSCmpts.h"
#include "NSCmpts_i.c"
#include "../common/interfaces.cpp"
#include "EmptyRecycleBin.h"
#include "EmptyTempFolder.h"
#include "DeleteSwapFile.h"
#include "RunScanDisk.h"
#include "WipefreeSpace.h"
#include "Backup.h"
#include "..\Common\Defines.h"
#include "BackupSourceProp.h"
#include "BackupDestProp.h"
#include "BackupAdvancedProp.h"
#include "NightSecError.h"
#include "Encrypt.h"
#include "EncryptSourceProp.h"
#include "EncryptAdvancedProp.h"
CComModule _Module;
BEGIN_OBJECT_MAP(ObjectMap)
OBJECT_ENTRY(CLSID_EmptyRecycleBin, CEmptyRecycleBin)
OBJECT_ENTRY(CLSID_EmptyTempFolder, CEmptyTempFolder)
OBJECT_ENTRY(CLSID_DeleteSwapFile, CDeleteSwapFile)
OBJECT_ENTRY(CLSID_RunScanDisk, CRunScanDisk)
OBJECT_ENTRY(CLSID_WipefreeSpace, CWipefreeSpace)
OBJECT_ENTRY(CLSID_Backup, CBackup)
OBJECT_ENTRY(CLSID_BackupSourceProp, CBackupSourceProp)
OBJECT_ENTRY(CLSID_BackupDestProp, CBackupDestProp)
OBJECT_ENTRY(CLSID_BackupAdvancedProp, CBackupAdvancedProp)
OBJECT_ENTRY(CLSID_Encrypt, CEncrypt)
OBJECT_ENTRY(CLSID_EncryptSourceProp, CEncryptSourceProp)
OBJECT_ENTRY(CLSID_EncryptAdvancedProp, CEncryptAdvancedProp)
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_NightSecCmpts);
DisableThreadLibraryCalls(hInstance);
// Not mission critical (used only for drag drop)
OleInitialize(NULL);
}
else if (dwReason == DLL_PROCESS_DETACH)
{
_Module.Term();
// Not mission critical (used only for drag drop)
OleUninitialize();
}
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 >>>>
}
|
