350 lines
8.4 KiB
C++
350 lines
8.4 KiB
C++
|
//////////////////////////////////////////////////////////////////////////////
|
||
|
|
/*++
|
||
|
|
|
||
|
|
Copyright (C) Microsoft Corporation, 1997 - 1999
|
||
|
|
|
||
|
|
Module Name:
|
||
|
|
|
||
|
|
NAPMMCDLL.cpp
|
||
|
|
|
||
|
|
Abstract:
|
||
|
|
|
||
|
|
Implementation of DLL Exports.
|
||
|
|
|
||
|
|
Proxy/Stub Information:
|
||
|
|
|
||
|
|
To build a separate proxy/stub DLL,
|
||
|
|
run nmake -f NAPMMCps.mk in the project directory.
|
||
|
|
|
||
|
|
Revision History:
|
||
|
|
mmaguire 11/6/97 - created using MMC snap-in wizard
|
||
|
|
|
||
|
|
|
||
|
|
--*/
|
||
|
|
//////////////////////////////////////////////////////////////////////////////
|
||
|
|
|
||
|
|
//////////////////////////////////////////////////////////////////////////////
|
||
|
|
// BEGIN INCLUDES
|
||
|
|
|
||
|
|
|
||
|
|
//
|
||
|
|
// standard includes:
|
||
|
|
//
|
||
|
|
#include "Precompiled.h"
|
||
|
|
//
|
||
|
|
// where we can find declaration for main class in this file:
|
||
|
|
//
|
||
|
|
|
||
|
|
//
|
||
|
|
// where we can find declarations needed in this file:
|
||
|
|
//
|
||
|
|
#include "initguid.h"
|
||
|
|
#include "NAPMMC_i.c"
|
||
|
|
|
||
|
|
// For IComponent, IComponentData, and ISnapinAbout COM classes:
|
||
|
|
|
||
|
|
#include "Component.h"
|
||
|
|
#include "About.h"
|
||
|
|
#include "LogComp.h"
|
||
|
|
#include "LogAbout.H"
|
||
|
|
|
||
|
|
// For AttributeInfo COM classes:
|
||
|
|
|
||
|
|
#include "IASAttributeInfo.h"
|
||
|
|
#include "IASEnumerableAttributeInfo.h"
|
||
|
|
|
||
|
|
// SDO declarations
|
||
|
|
//TODO: change it when TODD's CLSIDs are ready
|
||
|
|
|
||
|
|
//
|
||
|
|
// Need to include this at least once to compile it in from the common directory:
|
||
|
|
#include "mmcutility.cpp"
|
||
|
|
|
||
|
|
// We are hosting a few extra COM objects in this dll:
|
||
|
|
|
||
|
|
|
||
|
|
// For AttributeEditor COM classes:
|
||
|
|
#include "IASIPAttributeEditor.h"
|
||
|
|
#include "IASMultivaluedAttributeEditor.h"
|
||
|
|
#include "IASVendorSpecificAttributeEditor.h"
|
||
|
|
#include "IASEnumerableAttributeEditor.h"
|
||
|
|
#include "IASStringAttributeEditor.h"
|
||
|
|
#include "IASBooleanAttributeEditor.h"
|
||
|
|
#include "NTGroups.h"
|
||
|
|
|
||
|
|
// For NASVendors COM object:
|
||
|
|
#include "Vendors.h"
|
||
|
|
|
||
|
|
#include <proxyext.h>
|
||
|
|
#include <proxyres.h>
|
||
|
|
unsigned int CF_MMC_NodeID = RegisterClipboardFormatW(CCF_NODEID2);
|
||
|
|
|
||
|
|
|
||
|
|
//
|
||
|
|
// END INCLUDES
|
||
|
|
//////////////////////////////////////////////////////////////////////////////
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
CComModule _Module;
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
BEGIN_OBJECT_MAP(ObjectMap)
|
||
|
|
OBJECT_ENTRY(CLSID_NAPSnapin, CComponentData)
|
||
|
|
OBJECT_ENTRY(CLSID_NAPSnapinAbout, CSnapinAbout)
|
||
|
|
OBJECT_ENTRY(CLSID_LoggingSnapin, CLoggingComponentData)
|
||
|
|
OBJECT_ENTRY(CLSID_LoggingSnapinAbout, CLoggingSnapinAbout)
|
||
|
|
OBJECT_ENTRY(CLSID_IASAttributeInfo, CAttributeInfo)
|
||
|
|
OBJECT_ENTRY(CLSID_IASEnumerableAttributeInfo, CEnumerableAttributeInfo)
|
||
|
|
OBJECT_ENTRY(CLSID_IASIPAttributeEditor, CIASIPAttributeEditor)
|
||
|
|
OBJECT_ENTRY(CLSID_IASMultivaluedAttributeEditor, CIASMultivaluedAttributeEditor)
|
||
|
|
OBJECT_ENTRY(CLSID_IASVendorSpecificAttributeEditor, CIASVendorSpecificAttributeEditor)
|
||
|
|
OBJECT_ENTRY(CLSID_IASEnumerableAttributeEditor, CIASEnumerableAttributeEditor)
|
||
|
|
OBJECT_ENTRY(CLSID_IASStringAttributeEditor, CIASStringAttributeEditor)
|
||
|
|
OBJECT_ENTRY(CLSID_IASBooleanAttributeEditor, CIASBooleanAttributeEditor)
|
||
|
|
OBJECT_ENTRY(CLSID_IASGroupsAttributeEditor, CIASGroupsAttributeEditor)
|
||
|
|
OBJECT_ENTRY(CLSID_IASNASVendors, CIASNASVendors)
|
||
|
|
OBJECT_ENTRY(__uuidof(ProxyExtension), ProxyExtension)
|
||
|
|
END_OBJECT_MAP()
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
#if 1 // Use CWinApp for MFC support -- some of the COM objects in this dll use MFC.
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
class CNAPMMCApp : public CWinApp
|
||
|
|
{
|
||
|
|
public:
|
||
|
|
virtual BOOL InitInstance();
|
||
|
|
virtual int ExitInstance();
|
||
|
|
};
|
||
|
|
|
||
|
|
CNAPMMCApp theApp;
|
||
|
|
|
||
|
|
DWORD g_dwTraceHandle = 0;
|
||
|
|
|
||
|
|
//////////////////////////////////////////////////////////////////////////////
|
||
|
|
/*++
|
||
|
|
|
||
|
|
CNAPMMCApp::InitInstance
|
||
|
|
|
||
|
|
MFC's dll entry point.
|
||
|
|
|
||
|
|
--*/
|
||
|
|
//////////////////////////////////////////////////////////////////////////////
|
||
|
|
BOOL CNAPMMCApp::InitInstance()
|
||
|
|
{
|
||
|
|
_Module.Init(ObjectMap, m_hInstance);
|
||
|
|
|
||
|
|
g_dwTraceHandle = TraceRegister(_T("NapMMC"));
|
||
|
|
TracePrintf(g_dwTraceHandle, _T("DLL Init"));
|
||
|
|
|
||
|
|
// Initialize static class variables of CSnapInItem.
|
||
|
|
CSnapInItem::Init();
|
||
|
|
|
||
|
|
// Initialize any other static class variables.
|
||
|
|
CMachineNode::InitClipboardFormat();
|
||
|
|
CLoggingMachineNode::InitClipboardFormat();
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
return CWinApp::InitInstance();
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
//////////////////////////////////////////////////////////////////////////////
|
||
|
|
/*++
|
||
|
|
|
||
|
|
CNAPMMCApp::ExitInstance
|
||
|
|
|
||
|
|
MFC's dll exit point.
|
||
|
|
|
||
|
|
--*/
|
||
|
|
//////////////////////////////////////////////////////////////////////////////
|
||
|
|
int CNAPMMCApp::ExitInstance()
|
||
|
|
{
|
||
|
|
TracePrintf(g_dwTraceHandle, _T("DLL Exit"));
|
||
|
|
TraceDeregister(g_dwTraceHandle);
|
||
|
|
|
||
|
|
_Module.Term();
|
||
|
|
|
||
|
|
return CWinApp::ExitInstance();
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
#else // Use CWinApp
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
//////////////////////////////////////////////////////////////////////////////
|
||
|
|
/*++
|
||
|
|
|
||
|
|
DllMain
|
||
|
|
|
||
|
|
|
||
|
|
Remarks
|
||
|
|
|
||
|
|
DLL Entry Point
|
||
|
|
|
||
|
|
--*/
|
||
|
|
//////////////////////////////////////////////////////////////////////////////
|
||
|
|
extern "C"
|
||
|
|
BOOL WINAPI DllMain(HINSTANCE hInstance, DWORD dwReason, LPVOID /*lpReserved*/)
|
||
|
|
{
|
||
|
|
if (dwReason == DLL_PROCESS_ATTACH)
|
||
|
|
{
|
||
|
|
_Module.Init(ObjectMap, hInstance);
|
||
|
|
// Initialize static class variables of CSnapInItem.
|
||
|
|
CSnapInItem::Init();
|
||
|
|
|
||
|
|
// Initialize any other static class variables.
|
||
|
|
CMachineNode::InitClipboardFormat();
|
||
|
|
|
||
|
|
DisableThreadLibraryCalls(hInstance);
|
||
|
|
}
|
||
|
|
else if (dwReason == DLL_PROCESS_DETACH)
|
||
|
|
_Module.Term();
|
||
|
|
return TRUE; // ok
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
#endif // Use CWinApp
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
//////////////////////////////////////////////////////////////////////////////
|
||
|
|
/*++
|
||
|
|
|
||
|
|
DllCanUnloadNow
|
||
|
|
|
||
|
|
|
||
|
|
Remarks
|
||
|
|
|
||
|
|
Used to determine whether the DLL can be unloaded by OLE
|
||
|
|
|
||
|
|
--*/
|
||
|
|
//////////////////////////////////////////////////////////////////////////////
|
||
|
|
STDAPI DllCanUnloadNow(void)
|
||
|
|
{
|
||
|
|
return (_Module.GetLockCount()==0) ? S_OK : S_FALSE;
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
//////////////////////////////////////////////////////////////////////////////
|
||
|
|
/*++
|
||
|
|
|
||
|
|
DllGetClassObject
|
||
|
|
|
||
|
|
|
||
|
|
Remarks
|
||
|
|
|
||
|
|
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
|
||
|
|
|
||
|
|
|
||
|
|
Remarks
|
||
|
|
|
||
|
|
Adds entries to the system registry
|
||
|
|
|
||
|
|
--*/
|
||
|
|
//////////////////////////////////////////////////////////////////////////////
|
||
|
|
|
||
|
|
STDAPI DllRegisterServer(void)
|
||
|
|
{
|
||
|
|
// Set the protocol.
|
||
|
|
TCHAR NapName[IAS_MAX_STRING];
|
||
|
|
TCHAR NapName_Indirect[IAS_MAX_STRING];
|
||
|
|
TCHAR ModuleName[MAX_PATH];
|
||
|
|
TCHAR LoggingName[IAS_MAX_STRING];
|
||
|
|
TCHAR LoggingName_Indirect[IAS_MAX_STRING];
|
||
|
|
|
||
|
|
GetModuleFileNameOnly(_Module.GetModuleInstance(), ModuleName, MAX_PATH);
|
||
|
|
|
||
|
|
int iLoadStringResult = LoadString( _Module.GetResourceInstance(), IDS_SNAPINNAME_NAP, NapName, IAS_MAX_STRING );
|
||
|
|
swprintf(NapName_Indirect, L"@%s,-%-d", ModuleName, IDS_SNAPINNAME_NAP);
|
||
|
|
|
||
|
|
iLoadStringResult = LoadString( _Module.GetResourceInstance(), IDS_SNAPINNAME_LOGGING, LoggingName, IAS_MAX_STRING );
|
||
|
|
swprintf(LoggingName_Indirect, L"@%s,-%-d", ModuleName, IDS_SNAPINNAME_LOGGING);
|
||
|
|
|
||
|
|
struct _ATL_REGMAP_ENTRY regMap[] = {
|
||
|
|
{OLESTR("NAPSNAPIN"), NapName}, // subsitute %NAPSNAPIN% for registry
|
||
|
|
{OLESTR("NAPSNAPIN_INDIRECT"), NapName_Indirect}, // subsitute %IASSNAPIN% for registry
|
||
|
|
{OLESTR("LOGGINGSNAPIN"), LoggingName}, // subsitute %LOGGINGSNAPIN% for registry
|
||
|
|
{OLESTR("LOGGINGSNAPIN_INDIRECT"), LoggingName_Indirect}, // subsitute %IASSNAPIN% for registry
|
||
|
|
{0, 0}
|
||
|
|
};
|
||
|
|
|
||
|
|
HRESULT hr = _Module.UpdateRegistryFromResource(IDR_NAPSNAPIN, TRUE, regMap);
|
||
|
|
if (SUCCEEDED(hr))
|
||
|
|
{
|
||
|
|
ResourceString proxyName(IDS_PROXY_EXTENSION);
|
||
|
|
_ATL_REGMAP_ENTRY entries[] =
|
||
|
|
{
|
||
|
|
{ L"PROXY_EXTENSION", proxyName },
|
||
|
|
{ NULL, NULL }
|
||
|
|
};
|
||
|
|
|
||
|
|
hr = _Module.UpdateRegistryFromResource(
|
||
|
|
IDR_PROXY_REGISTRY,
|
||
|
|
TRUE,
|
||
|
|
entries
|
||
|
|
);
|
||
|
|
}
|
||
|
|
|
||
|
|
return hr;
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
//////////////////////////////////////////////////////////////////////////////
|
||
|
|
/*++
|
||
|
|
|
||
|
|
DllUnregisterServer
|
||
|
|
|
||
|
|
|
||
|
|
Remarks
|
||
|
|
|
||
|
|
Removes entries from the system registry
|
||
|
|
|
||
|
|
--*/
|
||
|
|
//////////////////////////////////////////////////////////////////////////////
|
||
|
|
STDAPI DllUnregisterServer(void)
|
||
|
|
{
|
||
|
|
// Set the protocol.
|
||
|
|
TCHAR NapName[IAS_MAX_STRING];
|
||
|
|
TCHAR LoggingName[IAS_MAX_STRING];
|
||
|
|
int iLoadStringResult = LoadString( _Module.GetResourceInstance(), IDS_SNAPINNAME_NAP, NapName, IAS_MAX_STRING );
|
||
|
|
iLoadStringResult = LoadString( _Module.GetResourceInstance(), IDS_SNAPINNAME_LOGGING, LoggingName, IAS_MAX_STRING );
|
||
|
|
|
||
|
|
struct _ATL_REGMAP_ENTRY regMap[] = {
|
||
|
|
{OLESTR("NAPSNAPIN"), NapName}, // subsitute %NAPSNAPIN% for registry
|
||
|
|
{OLESTR("LOGGINGSNAPIN"), LoggingName}, // subsitute %LOGGINGSNAPIN% for registry
|
||
|
|
{0, 0}
|
||
|
|
};
|
||
|
|
|
||
|
|
_Module.UpdateRegistryFromResource(IDR_NAPSNAPIN, FALSE, regMap);
|
||
|
|
_Module.UpdateRegistryFromResource(IDR_PROXY_REGISTRY, FALSE, NULL);
|
||
|
|
|
||
|
|
return S_OK;
|
||
|
|
}
|
||
|
|
|
||
|
|
|