445 lines
13 KiB
C++
445 lines
13 KiB
C++
/**********************************************************************/
|
|
/** Microsoft Windows/NT **/
|
|
/** Copyright(c) Microsoft Corporation, 1997 - 1999 **/
|
|
/**********************************************************************/
|
|
|
|
/*
|
|
rtrsnap.cpp
|
|
Snapin entry points/registration functions
|
|
|
|
Note: Proxy/Stub Information
|
|
To build a separate proxy/stub DLL,
|
|
run nmake -f Snapinps.mak in the project directory.
|
|
|
|
FILE HISTORY:
|
|
|
|
*/
|
|
|
|
#include "stdafx.h"
|
|
#include <advpub.h> // For REGINSTALL
|
|
#include "dmvcomp.h"
|
|
#include "register.h"
|
|
#include "rtrguid.h"
|
|
#include "atlkcomp.h"
|
|
#include "radcfg.h" // for RouterAuthRadiusConfig
|
|
#include "qryfrm.h"
|
|
#include "ncglobal.h" // network console global defines
|
|
#include "cmptrmgr.h" // computer management snapin node types
|
|
#include "rtrutilp.h"
|
|
|
|
#include "dialog.h"
|
|
|
|
#ifdef _DEBUG
|
|
void DbgVerifyInstanceCounts();
|
|
#define DEBUG_VERIFY_INSTANCE_COUNTS DbgVerifyInstanceCounts()
|
|
#else
|
|
#define DEBUG_VERIFY_INSTANCE_COUNTS
|
|
#endif
|
|
|
|
|
|
CComModule _Module;
|
|
|
|
BEGIN_OBJECT_MAP(ObjectMap)
|
|
OBJECT_ENTRY(CLSID_ATLKAdminExtension, CATLKComponentData)
|
|
OBJECT_ENTRY(CLSID_ATLKAdminAbout, CATLKAbout)
|
|
OBJECT_ENTRY(CLSID_RouterSnapin, CDomainViewSnap)
|
|
OBJECT_ENTRY(CLSID_RouterSnapinExtension, CDomainViewSnapExtension)
|
|
OBJECT_ENTRY(CLSID_RouterSnapinAbout, CDomainViewSnapAbout)
|
|
OBJECT_ENTRY(CLSID_RouterAuthRADIUS, RouterAuthRadiusConfig)
|
|
OBJECT_ENTRY(CLSID_RouterAcctRADIUS, RouterAcctRadiusConfig)
|
|
OBJECT_ENTRY(CLSID_RRASQueryForm, CRRASQueryForm)
|
|
END_OBJECT_MAP()
|
|
|
|
#ifdef _DEBUG
|
|
#define new DEBUG_NEW
|
|
#undef THIS_FILE
|
|
static char THIS_FILE[] = __FILE__;
|
|
#endif
|
|
|
|
|
|
/*---------------------------------------------------------------------------
|
|
This is a list of nodetypes that need to be registered.
|
|
---------------------------------------------------------------------------*/
|
|
|
|
struct RegisteredNodeTypes
|
|
{
|
|
const GUID *m_pGuid;
|
|
LPCTSTR m_pszName;
|
|
};
|
|
|
|
const static RegisteredNodeTypes s_rgNodeTypes[] =
|
|
{
|
|
{ &GUID_RouterDomainNodeType, _T("Root of Router Domain Snapin") },
|
|
{ &GUID_RouterIfAdminNodeType, _T("Routing Interfaces") },
|
|
{ &GUID_RouterMachineErrorNodeType, _T("Router - Error") },
|
|
{ &GUID_RouterMachineNodeType, _T("Router Machine - General (7)") },
|
|
{ &GUID_RouterDialInNodeType, _T("Routing dial-in users") },
|
|
{ &GUID_RouterPortsNodeType, _T("Ports") },
|
|
};
|
|
|
|
|
|
|
|
class CRouterSnapinApp : public CWinApp
|
|
{
|
|
public:
|
|
virtual BOOL InitInstance();
|
|
virtual int ExitInstance();
|
|
};
|
|
|
|
CRouterSnapinApp theApp;
|
|
|
|
BOOL CRouterSnapinApp::InitInstance()
|
|
{
|
|
TCHAR tszHelpFilePath[MAX_PATH+1]={0};
|
|
|
|
_Module.Init(ObjectMap, m_hInstance);
|
|
|
|
InitializeTFSError();
|
|
CreateTFSErrorInfo(0);
|
|
|
|
// Setup the global help function
|
|
extern DWORD * MprSnapHelpMap(DWORD dwIDD);
|
|
SetGlobalHelpMapFunction(MprSnapHelpMap);
|
|
|
|
IPAddrInit(m_hInstance);
|
|
//Set the help file path
|
|
free((void*)m_pszHelpFilePath);
|
|
GetWindowsDirectory(tszHelpFilePath, MAX_PATH);
|
|
_tcscat(tszHelpFilePath, TEXT("\\help\\mprsnap.hlp"));
|
|
m_pszHelpFilePath = _tcsdup(tszHelpFilePath);
|
|
|
|
return CWinApp::InitInstance();
|
|
}
|
|
|
|
int CRouterSnapinApp::ExitInstance()
|
|
{
|
|
RemoveAllNetConnections();
|
|
|
|
_Module.Term();
|
|
|
|
DestroyTFSErrorInfo(0);
|
|
CleanupTFSError();
|
|
|
|
DEBUG_VERIFY_INSTANCE_COUNTS;
|
|
|
|
return CWinApp::ExitInstance();
|
|
}
|
|
|
|
/////////////////////////////////////////////////////////////////////////////
|
|
// Used to determine whether the DLL can be unloaded by OLE
|
|
|
|
STDAPI DllCanUnloadNow(void)
|
|
{
|
|
AFX_MANAGE_STATE(AfxGetStaticModuleState());
|
|
return (AfxDllCanUnloadNow()==S_OK && _Module.GetLockCount()==0) ? S_OK : S_FALSE;
|
|
}
|
|
|
|
const static GUID * s_pExtensionGuids[] =
|
|
{
|
|
&GUID_RouterMachineNodeType,
|
|
};
|
|
|
|
|
|
/////////////////////////////////////////////////////////////////////////////
|
|
// 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);
|
|
}
|
|
|
|
HRESULT CallRegInstall(LPSTR szSection);
|
|
|
|
/////////////////////////////////////////////////////////////////////////////
|
|
// DllRegisterServer - Adds entries to the system registry
|
|
|
|
STDAPI DllRegisterServer(void)
|
|
{
|
|
AFX_MANAGE_STATE(AfxGetStaticModuleState());
|
|
HRESULT hr = hrOK;
|
|
CString stDisplayName, stAtlkDisplayName, stNameStringIndirect;
|
|
TCHAR moduleFileName[MAX_PATH * 2];
|
|
|
|
GetModuleFileNameOnly(_Module.GetModuleInstance(), moduleFileName, MAX_PATH * 2);
|
|
|
|
int i;
|
|
|
|
// registers object, typelib and all interfaces in typelib
|
|
//
|
|
hr = _Module.RegisterServer(/* bRegTypeLib */ FALSE);
|
|
Assert(SUCCEEDED(hr));
|
|
|
|
CORg( hr );
|
|
|
|
// Load the name of the router snapins
|
|
stDisplayName.LoadString(IDS_SNAPIN_DISPLAY_NAME);
|
|
stAtlkDisplayName.LoadString(IDS_ATLK_DISPLAY_NAME);
|
|
stNameStringIndirect.Format(L"@%s,-%-d", moduleFileName, IDS_SNAPIN_DISPLAY_NAME);
|
|
|
|
// register the snapin into the console snapin list
|
|
// ~domain view snapin
|
|
CORg( RegisterSnapinGUID(&CLSID_RouterSnapin,
|
|
NULL,
|
|
&CLSID_RouterSnapinAbout,
|
|
stDisplayName,
|
|
_T("1.0"),
|
|
TRUE,
|
|
stNameStringIndirect
|
|
) );
|
|
|
|
CORg( RegisterSnapinGUID(&CLSID_RouterSnapinExtension,
|
|
NULL,
|
|
&CLSID_RouterSnapinAbout,
|
|
stDisplayName,
|
|
_T("1.0"),
|
|
FALSE,
|
|
stNameStringIndirect));
|
|
|
|
stNameStringIndirect.Format(L"@%s,-%-d", moduleFileName, IDS_ATLK_DISPLAY_NAME);
|
|
CORg( RegisterSnapinGUID(&CLSID_ATLKAdminExtension,
|
|
NULL,
|
|
&CLSID_ATLKAdminAbout,
|
|
stAtlkDisplayName,
|
|
_T("1.0"),
|
|
FALSE,
|
|
stNameStringIndirect) );
|
|
|
|
// register the snapin nodes into the console node list
|
|
//
|
|
for (i=0; i<DimensionOf(s_rgNodeTypes); i++)
|
|
{
|
|
CORg( RegisterNodeTypeGUID(&CLSID_RouterSnapin,
|
|
s_rgNodeTypes[i].m_pGuid,
|
|
s_rgNodeTypes[i].m_pszName) );
|
|
}
|
|
|
|
// register apple talk as extension of machine
|
|
for (i=0; i<DimensionOf(s_pExtensionGuids); i++)
|
|
{
|
|
CORg( RegisterAsRequiredExtensionGUID(s_pExtensionGuids[i],
|
|
&CLSID_ATLKAdminExtension,
|
|
stAtlkDisplayName,
|
|
EXTENSION_TYPE_NAMESPACE,
|
|
&CLSID_RouterSnapin) );
|
|
}
|
|
|
|
#ifdef __NETWORK_CONSOLE__
|
|
// register as extension of network console
|
|
CORg( RegisterAsRequiredExtensionGUID(&GUID_NetConsRootNodeType,
|
|
&CLSID_RouterSnapinExtension,
|
|
stDisplayName,
|
|
EXTENSION_TYPE_TASK | EXTENSION_TYPE_NAMESPACE,
|
|
&GUID_NetConsRootNodeType)); // doesn't matter what this is, just
|
|
// needs to be non-null guid
|
|
#endif
|
|
|
|
// register as extension of computer management
|
|
CORg( RegisterAsRequiredExtensionGUID(&NODETYPE_COMPUTERMANAGEMENT_SERVERAPPS,
|
|
&CLSID_RouterSnapinExtension,
|
|
stDisplayName,
|
|
EXTENSION_TYPE_TASK | EXTENSION_TYPE_NAMESPACE,
|
|
&NODETYPE_COMPUTERMANAGEMENT_SERVERAPPS));
|
|
// Register DS Query Forms -- WeiJiang 1-29-98
|
|
CORg(CallRegInstall("RegDll"));
|
|
// End of DS Query
|
|
|
|
Error:
|
|
|
|
if (!FHrSucceeded(hr))
|
|
{
|
|
// Now we need to get the error object and display it
|
|
if (!FHrSucceeded(DisplayTFSErrorMessage(NULL)))
|
|
{
|
|
TCHAR szBuffer[1024];
|
|
|
|
// Couldn't find a TFS error, bring up a general
|
|
// error message
|
|
FormatError(hr, szBuffer, DimensionOf(szBuffer));
|
|
AfxMessageBox(szBuffer);
|
|
}
|
|
}
|
|
|
|
return hr;
|
|
}
|
|
|
|
/////////////////////////////////////////////////////////////////////////////
|
|
// DllUnregisterServer - Removes entries from the system registry
|
|
|
|
STDAPI DllUnregisterServer(void)
|
|
{
|
|
int i;
|
|
HRESULT hr = hrOK;
|
|
|
|
// Initialize the error handling system
|
|
InitializeTFSError();
|
|
|
|
// Create an error object for this thread
|
|
Verify( CreateTFSErrorInfo(0) == hrOK );
|
|
|
|
|
|
hr = _Module.UnregisterServer();
|
|
Assert(SUCCEEDED(hr));
|
|
CORg( hr );
|
|
|
|
// un register the snapin
|
|
//
|
|
// We don't care about the error return for this
|
|
UnregisterSnapinGUID(&CLSID_OldRouterSnapin);
|
|
|
|
|
|
// Domain View Snapin -- weijiang 1-14-98
|
|
hr = UnregisterSnapinGUID(&CLSID_RouterSnapin);
|
|
Assert(SUCCEEDED(hr));
|
|
// ~Domain View Snapin
|
|
|
|
|
|
// Unregister the nodes that Appletalk extends
|
|
for (i=0; i<DimensionOf(s_pExtensionGuids); i++)
|
|
{
|
|
hr = UnregisterAsRequiredExtensionGUID(s_pExtensionGuids[i],
|
|
&CLSID_ATLKAdminExtension,
|
|
EXTENSION_TYPE_NAMESPACE,
|
|
&CLSID_RouterSnapin);
|
|
Assert(SUCCEEDED(hr));
|
|
}
|
|
|
|
|
|
// Unregister the appletalk extension snapin
|
|
// -----------------------------------------------------------------
|
|
hr = UnregisterSnapinGUID(&CLSID_ATLKAdminExtension);
|
|
Assert(SUCCEEDED(hr));
|
|
|
|
|
|
// Unregister the router snapin extension snapin
|
|
// -----------------------------------------------------------------
|
|
hr = UnregisterSnapinGUID(&CLSID_RouterSnapinExtension);
|
|
Assert(SUCCEEDED(hr));
|
|
|
|
|
|
// unregister the snapin nodes
|
|
// -----------------------------------------------------------------
|
|
for (i=0; i<DimensionOf(s_rgNodeTypes); i++)
|
|
{
|
|
hr = UnregisterNodeTypeGUID(s_rgNodeTypes[i].m_pGuid);
|
|
Assert(SUCCEEDED(hr));
|
|
}
|
|
// computer manangement
|
|
hr = UnregisterAsExtensionGUID(&NODETYPE_COMPUTERMANAGEMENT_SERVERAPPS,
|
|
&CLSID_RouterSnapinExtension,
|
|
EXTENSION_TYPE_TASK | EXTENSION_TYPE_NAMESPACE);
|
|
ASSERT(SUCCEEDED(hr));
|
|
|
|
|
|
// Unregister DS Query Form -- WeiJiang 1-29-98
|
|
hr = CallRegInstall("UnRegDll");
|
|
Assert(SUCCEEDED(hr));
|
|
|
|
// End of DS Query FOrm
|
|
Error:
|
|
if (!FHrSucceeded(hr))
|
|
{
|
|
// Now we need to get the error object and display it
|
|
if (!FHrSucceeded(DisplayTFSErrorMessage(NULL)))
|
|
{
|
|
TCHAR szBuffer[1024];
|
|
|
|
// Couldn't find a TFS error, bring up a general
|
|
// error message
|
|
FormatError(hr, szBuffer, DimensionOf(szBuffer));
|
|
AfxMessageBox(szBuffer);
|
|
}
|
|
}
|
|
|
|
// Destroy the TFS error information for this thread
|
|
DestroyTFSErrorInfo(0);
|
|
|
|
// Cleanup the entire error system
|
|
CleanupTFSError();
|
|
|
|
return hr;
|
|
}
|
|
|
|
/*-----------------------------------------------------------------------------
|
|
/ CallRegInstall
|
|
/ --------------
|
|
/ Call ADVPACK for the given section of our resource based INF>
|
|
/
|
|
/ In:
|
|
/ szSection = section name to invoke
|
|
/
|
|
/ Out:
|
|
/ HRESULT:
|
|
/----------------------------------------------------------------------------*/
|
|
HRESULT CallRegInstall(LPSTR szSection)
|
|
{
|
|
HRESULT hr = E_FAIL;
|
|
HINSTANCE hinstAdvPack = LoadLibrary(TEXT("ADVPACK.DLL"));
|
|
|
|
if (hinstAdvPack)
|
|
{
|
|
REGINSTALL pfnri = (REGINSTALL)GetProcAddress(hinstAdvPack, "RegInstall");
|
|
|
|
#ifdef UNICODE
|
|
if ( pfnri )
|
|
{
|
|
STRENTRY seReg[] =
|
|
{
|
|
// These two NT-specific entries must be at the end
|
|
{ "25", "%SystemRoot%" },
|
|
{ "11", "%SystemRoot%\\system32" },
|
|
};
|
|
STRTABLE stReg = { ARRAYSIZE(seReg), seReg };
|
|
|
|
hr = pfnri(_Module.m_hInst, szSection, &stReg);
|
|
}
|
|
#else
|
|
if (pfnri)
|
|
{
|
|
hr = pfnri(_Module.m_hInst, szSection, NULL);
|
|
}
|
|
|
|
#endif
|
|
FreeLibrary(hinstAdvPack);
|
|
}
|
|
|
|
return hr;
|
|
}
|
|
|
|
#ifdef _DEBUG
|
|
void DbgVerifyInstanceCounts()
|
|
{
|
|
extern void TFSCore_DbgVerifyInstanceCounts();
|
|
TFSCore_DbgVerifyInstanceCounts();
|
|
|
|
DEBUG_VERIFY_INSTANCE_COUNT(MachineNodeData);
|
|
DEBUG_VERIFY_INSTANCE_COUNT(InfoBase);
|
|
DEBUG_VERIFY_INSTANCE_COUNT(InfoBlockEnumerator);
|
|
DEBUG_VERIFY_INSTANCE_COUNT(RouterInfo);
|
|
DEBUG_VERIFY_INSTANCE_COUNT(RtrMgrInfo);
|
|
DEBUG_VERIFY_INSTANCE_COUNT(RtrMgrProtocolInfo);
|
|
DEBUG_VERIFY_INSTANCE_COUNT(InterfaceInfo);
|
|
DEBUG_VERIFY_INSTANCE_COUNT(RtrMgrInterfaceInfo);
|
|
DEBUG_VERIFY_INSTANCE_COUNT(RtrMgrProtocolInterfaceInfo);
|
|
|
|
DEBUG_VERIFY_INSTANCE_COUNT(EnumRtrMgrCB);
|
|
DEBUG_VERIFY_INSTANCE_COUNT(EnumRtrMgrProtocolCB);
|
|
DEBUG_VERIFY_INSTANCE_COUNT(EnumInterfaceCB);
|
|
DEBUG_VERIFY_INSTANCE_COUNT(EnumRtrMgrInterfaceCB);
|
|
DEBUG_VERIFY_INSTANCE_COUNT(EnumRtrMgrProtocolInterfaceCB);
|
|
|
|
DEBUG_VERIFY_INSTANCE_COUNT(InterfaceNodeHandler);
|
|
DEBUG_VERIFY_INSTANCE_COUNT(MachineHandler);
|
|
|
|
DEBUG_VERIFY_INSTANCE_COUNT(RouterInfoAggregationWrapper);
|
|
DEBUG_VERIFY_INSTANCE_COUNT(InterfaceInfoAggregationWrapper);
|
|
DEBUG_VERIFY_INSTANCE_COUNT(RtrMgrInfoAggregationWrapper);
|
|
DEBUG_VERIFY_INSTANCE_COUNT(RtrMgrProtocolInfoAggregationWrapper);
|
|
DEBUG_VERIFY_INSTANCE_COUNT(RtrMgrProtocolInterfaceInfoAggregationWrapper);
|
|
|
|
DEBUG_VERIFY_INSTANCE_COUNT(RouterRefreshObjectGroup);
|
|
DEBUG_VERIFY_INSTANCE_COUNT(RefreshItem);
|
|
|
|
}
|
|
|
|
#endif // _DEBUG
|