/**********************************************************************/
/** 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