You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
1961 lines
65 KiB
1961 lines
65 KiB
/**********************************************************************/
|
|
/** Microsoft Windows/NT **/
|
|
/** Copyright(c) Microsoft Corporation, 1997 - 1999 **/
|
|
/**********************************************************************/
|
|
|
|
/*
|
|
rtrwiz.cpp
|
|
Router & Remote access configuration wizard
|
|
|
|
FILE HISTORY:
|
|
|
|
*/
|
|
|
|
#include "stdafx.h"
|
|
#include "rtrwiz.h"
|
|
#include "rtrutilp.h"
|
|
#include "rtrcomn.h" // CoCreateRouterConfig
|
|
#include "rrasutil.h"
|
|
#include "rtutils.h" // Tracing functions
|
|
#include "helper.h" // HrIsStandaloneServer
|
|
#include "infoi.h" // SRtrMgrProtocolCBList
|
|
#include "routprot.h" // MS_IP_XXXX
|
|
#include "snaputil.h"
|
|
#include "globals.h"
|
|
#include "rraswiz.h"
|
|
#include "iprtrmib.h" // MIB_IPFORWARDROW
|
|
|
|
|
|
// Include headers for IP-specific stuff
|
|
extern "C"
|
|
{
|
|
#include <ipnat.h>
|
|
#include <ipnathlp.h>
|
|
#include <sainfo.h>
|
|
};
|
|
|
|
|
|
|
|
#include "igmprm.h"
|
|
#include "ipbootp.h"
|
|
|
|
|
|
|
|
#ifdef _DEBUG
|
|
#define new DEBUG_NEW
|
|
#undef THIS_FILE
|
|
static char THIS_FILE[] = __FILE__;
|
|
#endif
|
|
|
|
|
|
|
|
HRESULT AddIGMPToInterface(IInterfaceInfo *pIf, BOOL fRouter);
|
|
HRESULT AddIPBOOTPToInterface(IInterfaceInfo *pIf);
|
|
HRESULT AddNATSimpleServers(NewRtrWizData *pNewRtrWizData,
|
|
RtrConfigData *pRtrConfigData);
|
|
HRESULT AddNATToInterfaces(NewRtrWizData *pNewRtrWizData,
|
|
RtrConfigData *pRtrConfigData,
|
|
IRouterInfo *pRouter,
|
|
BOOL fCreateDD);
|
|
HRESULT AddNATToInterface(MPR_SERVER_HANDLE hMprServer,
|
|
HANDLE hMprConfig,
|
|
NewRtrWizData *pNewRtrWizData,
|
|
RtrConfigData *pRtrConfigData,
|
|
LPCTSTR pszInterface,
|
|
BOOL fDemandDial,
|
|
BOOL fPublic);
|
|
HRESULT AddDhcpServerToBOOTPGlobalInfo(LPCTSTR pszServerName,
|
|
DWORD netDhcpServer);
|
|
|
|
|
|
/*---------------------------------------------------------------------------
|
|
Defaults
|
|
---------------------------------------------------------------------------*/
|
|
|
|
|
|
//
|
|
// Default values for LAN-interface IGMP configuration
|
|
//
|
|
// NOTE: Any changes made here should also be made to ipsnap\globals.cpp
|
|
//
|
|
IGMP_MIB_IF_CONFIG g_IGMPLanDefault = {
|
|
IGMP_VERSION_3, //Version
|
|
0, //IfIndex (readOnly)
|
|
0, //IpAddr (readOnly)
|
|
IGMP_IF_NOT_RAS, //IfType;
|
|
IGMP_INTERFACE_ENABLED_IN_CONFIG, //Flags
|
|
IGMP_ROUTER_V3, //IgmpProtocolType;
|
|
2, //RobustnessVariable;
|
|
31, //StartupQueryInterval;
|
|
2, //StartupQueryCount;
|
|
125, //GenQueryInterval;
|
|
10, //GenQueryMaxResponseTime;
|
|
1000, //LastMemQueryInterval; (msec)
|
|
2, //LastMemQueryCount;
|
|
255, //OtherQuerierPresentInterval;
|
|
260, //GroupMembershipTimeout;
|
|
0 //NumStaticGroups
|
|
};
|
|
|
|
|
|
//----------------------------------------------------------------------------
|
|
// DHCP Relay-agent default configuration
|
|
//
|
|
//----------------------------------------------------------------------------
|
|
//
|
|
// Default values for LAN-interface DHCP Relay-agent configuration
|
|
//
|
|
|
|
IPBOOTP_IF_CONFIG
|
|
g_relayLanDefault = {
|
|
0, // State (read-only)
|
|
IPBOOTP_RELAY_ENABLED, // Relay-mode
|
|
4, // Max hop-count
|
|
4 // Min seconds-since-boot
|
|
};
|
|
|
|
|
|
|
|
/*!--------------------------------------------------------------------------
|
|
RtrWizFinish
|
|
-
|
|
Author: KennT
|
|
---------------------------------------------------------------------------*/
|
|
DWORD RtrWizFinish(RtrConfigData* pRtrConfigData, IRouterInfo *pRouter)
|
|
{
|
|
HRESULT hr = hrOK;
|
|
LPCTSTR pszServerFlagsKey = NULL;
|
|
LPCTSTR pszRouterTypeKey = NULL;
|
|
RegKey rkey;
|
|
RouterVersionInfo routerVersion;
|
|
HKEY hkeyMachine = NULL;
|
|
CString stMachine, stPhonebookPath;
|
|
|
|
|
|
// Connect to the remote machine's registry
|
|
// ----------------------------------------------------------------
|
|
CWRg( ConnectRegistry(pRtrConfigData->m_stServerName, &hkeyMachine) );
|
|
|
|
|
|
QueryRouterVersionInfo(hkeyMachine, &routerVersion);
|
|
|
|
if (routerVersion.dwOsBuildNo < RASMAN_PPP_KEY_LAST_VERSION)
|
|
pszRouterTypeKey = c_szRegKeyRasProtocols;
|
|
else
|
|
pszRouterTypeKey = c_szRegKeyRemoteAccessParameters;
|
|
|
|
// enable routing & RAS page
|
|
// ----------------------------------------------------------------
|
|
if ( ERROR_SUCCESS == rkey.Open(hkeyMachine, pszRouterTypeKey) )
|
|
{
|
|
rkey.SetValue( c_szRouterType,pRtrConfigData->m_dwRouterType);
|
|
}
|
|
|
|
// protocols page
|
|
// ----------------------------------------------------------------
|
|
|
|
// RAS routing
|
|
// ----------------------------------------------------------------
|
|
|
|
if ((pRtrConfigData->m_fUseIp) &&
|
|
(pRtrConfigData->m_dwRouterType & (ROUTER_TYPE_RAS | ROUTER_TYPE_WAN)))
|
|
pRtrConfigData->m_ipData.SaveToReg(pRouter, routerVersion);
|
|
|
|
if ( pRtrConfigData->m_dwRouterType & ROUTER_TYPE_RAS )
|
|
{
|
|
if (pRtrConfigData->m_fUseIpx)
|
|
{
|
|
pRtrConfigData->m_ipxData.SaveToReg(pRouter);
|
|
}
|
|
|
|
if (pRtrConfigData->m_fUseNbf)
|
|
pRtrConfigData->m_nbfData.SaveToReg();
|
|
|
|
if (pRtrConfigData->m_fUseArap)
|
|
pRtrConfigData->m_arapData.SaveToReg();
|
|
}
|
|
|
|
// Save the err log data
|
|
pRtrConfigData->m_errlogData.SaveToReg();
|
|
|
|
// Save the auth data
|
|
pRtrConfigData->m_authData.SaveToReg(NULL);
|
|
|
|
// Set some global registry settings (that need to get set,
|
|
// independent of the router type).
|
|
// ----------------------------------------------------------------
|
|
InstallGlobalSettings(pRtrConfigData->m_stServerName, pRouter);
|
|
|
|
|
|
// router only will start service and return
|
|
// ----------------------------------------------------------------
|
|
if ( !(pRtrConfigData->m_dwRouterType & (ROUTER_TYPE_RAS | ROUTER_TYPE_WAN)) )
|
|
{
|
|
// implies LAN routing; so start router & return
|
|
// ------------------------------------------------------------
|
|
goto EndConfig;
|
|
}
|
|
|
|
|
|
// security page
|
|
// ----------------------------------------------------------------
|
|
|
|
// Depending on the version depends on where we look for the
|
|
// key.
|
|
// ----------------------------------------------------------------
|
|
if (routerVersion.dwOsBuildNo < RASMAN_PPP_KEY_LAST_VERSION)
|
|
pszServerFlagsKey = c_szRasmanPPPKey;
|
|
else
|
|
pszServerFlagsKey = c_szRegKeyRemoteAccessParameters;
|
|
|
|
if ( ERROR_SUCCESS == rkey.Open(HKEY_LOCAL_MACHINE,
|
|
pszServerFlagsKey,
|
|
KEY_ALL_ACCESS, pRtrConfigData->m_stServerName) )
|
|
{
|
|
DWORD dwServerFlags = 0;
|
|
|
|
// Windows NT Bug : 299456
|
|
// Query for the server flags before overwriting it. This way
|
|
// we don't overwrite the old values.
|
|
// ------------------------------------------------------------
|
|
rkey.QueryValue( c_szServerFlags, dwServerFlags );
|
|
|
|
// Add in the defaults (minus MSCHAP) for the PPP Settings.
|
|
// ------------------------------------------------------------
|
|
dwServerFlags |= (PPPCFG_UseSwCompression |
|
|
PPPCFG_UseLcpExtensions |
|
|
PPPCFG_NegotiateMultilink |
|
|
PPPCFG_NegotiateBacp);
|
|
|
|
// Set the value
|
|
// ------------------------------------------------------------
|
|
rkey.SetValue( c_szServerFlags, dwServerFlags );
|
|
}
|
|
|
|
// Delete the router.pbk
|
|
// ----------------------------------------------------------------
|
|
DeleteRouterPhonebook( pRtrConfigData->m_stServerName );
|
|
|
|
|
|
EndConfig:
|
|
|
|
WriteRouterConfiguredReg(pRtrConfigData->m_stServerName, TRUE);
|
|
|
|
WriteRRASExtendsComputerManagementKey(pRtrConfigData->m_stServerName, TRUE);
|
|
|
|
WriteErasePSKReg ( pRtrConfigData->m_stServerName, TRUE );
|
|
Error:
|
|
|
|
if (hkeyMachine)
|
|
DisconnectRegistry(hkeyMachine);
|
|
|
|
return hr;
|
|
}
|
|
|
|
|
|
|
|
extern "C"
|
|
HRESULT APIENTRY MprConfigServerInstallPrivate( VOID )
|
|
{
|
|
AFX_MANAGE_STATE(AfxGetStaticModuleState());
|
|
DWORD dwErr = ERROR_SUCCESS;
|
|
RtrConfigData wizData;
|
|
|
|
//
|
|
// We removed this smart pointer definition as it was causing
|
|
// a warning during build on alpha 32 bit. Looks like the extern "C"
|
|
// of the function definition is conflicting with the smart pointer.
|
|
//
|
|
// SPIRemoteNetworkConfig spNetwork;
|
|
IRemoteNetworkConfig * pNetwork = NULL;
|
|
IUnknown * punk = NULL;
|
|
CWaitCursor wait;
|
|
HRESULT hr = hrOK;
|
|
|
|
// Create the remote config object
|
|
// ----------------------------------------------------------------
|
|
CORg( CoCreateRouterConfig(NULL,
|
|
NULL,
|
|
NULL,
|
|
IID_IRemoteNetworkConfig,
|
|
&punk) );
|
|
|
|
pNetwork = (IRemoteNetworkConfig *) punk;
|
|
punk = NULL;
|
|
|
|
// Upgrade the configuration (ensure that the registry keys
|
|
// are populated correctly).
|
|
// ------------------------------------------------------------
|
|
CORg( pNetwork->UpgradeRouterConfig() );
|
|
|
|
|
|
wizData.m_stServerName.Empty();
|
|
wizData.m_dwRouterType = (ROUTER_TYPE_RAS|ROUTER_TYPE_LAN|ROUTER_TYPE_WAN);
|
|
|
|
// Need to get version information and initialize
|
|
// the RAS structures (IP only)
|
|
// Assume that this is on NT5
|
|
wizData.m_fUseIp = TRUE;
|
|
wizData.m_ipData.UseDefaults(_T(""), FALSE);
|
|
wizData.m_ipData.m_dwAllowNetworkAccess = TRUE;
|
|
|
|
wizData.m_authData.UseDefaults(NULL, FALSE);
|
|
wizData.m_errlogData.UseDefaults(NULL, FALSE);
|
|
|
|
dwErr = RtrWizFinish(&wizData, NULL);
|
|
hr = HResultFromWin32(dwErr);
|
|
|
|
SetDeviceType(NULL, wizData.m_dwRouterType, 10);
|
|
|
|
RegisterRouterInDomain(NULL, TRUE);
|
|
|
|
Error:
|
|
if (pNetwork)
|
|
pNetwork->Release();
|
|
|
|
return hr;
|
|
}
|
|
|
|
|
|
|
|
/*!--------------------------------------------------------------------------
|
|
MprConfigServerUnattendedInstall
|
|
Call this function to setup the registry entries for the server.
|
|
This works only with the local machine for now.
|
|
|
|
pswzServer - name of the server
|
|
fInstall - TRUE if we are installing, FALSE if we are removing
|
|
|
|
Author: KennT
|
|
---------------------------------------------------------------------------*/
|
|
|
|
extern "C"
|
|
HRESULT APIENTRY MprConfigServerUnattendedInstall(LPCWSTR pswzServer, BOOL fInstall)
|
|
{
|
|
HRESULT hr = hrOK;
|
|
|
|
#if 0 // remove this restriciton and see what happens
|
|
// We only the local machine (for now).
|
|
// ----------------------------------------------------------------
|
|
if (pswzServer)
|
|
{
|
|
return HResultFromWin32( ERROR_INVALID_PARAMETER );
|
|
}
|
|
#endif
|
|
|
|
// Write out the various registry settings
|
|
// ----------------------------------------------------------------
|
|
// CORg( SetRouterInstallRegistrySettings(pswzServer, fInstall, TRUE) );
|
|
|
|
|
|
// Write the "router is configured" flag
|
|
// ----------------------------------------------------------------
|
|
CORg( WriteRouterConfiguredReg(pswzServer, fInstall) );
|
|
|
|
// Write out the RRAS extend Computer Management key
|
|
CORg( WriteRRASExtendsComputerManagementKey(pswzServer, fInstall) );
|
|
|
|
|
|
if (fInstall)
|
|
{
|
|
// Set the state of the router to Autostart
|
|
// ------------------------------------------------------------
|
|
SetRouterServiceStartType(pswzServer,
|
|
SERVICE_AUTO_START);
|
|
}
|
|
|
|
Error:
|
|
|
|
return hr;
|
|
}
|
|
|
|
|
|
|
|
/*!--------------------------------------------------------------------------
|
|
AddIGMPToRasServer
|
|
-
|
|
Author: KennT
|
|
---------------------------------------------------------------------------*/
|
|
HRESULT AddIGMPToRasServer(RtrConfigData *pRtrConfigData,
|
|
IRouterInfo *pRouter)
|
|
{
|
|
HRESULT hr = hrOK;
|
|
SPIRouterProtocolConfig spRouterConfig;
|
|
SRtrMgrProtocolCBList SRmProtCBList;
|
|
POSITION pos;
|
|
SRtrMgrProtocolCB * pSRmProtCB;
|
|
BOOL fFoundDedicatedInterface;
|
|
GUID guidConfig = GUID_RouterNull;
|
|
SPIInterfaceInfo spIf;
|
|
SPIRtrMgrInterfaceInfo spRmIf;
|
|
DWORD dwIfType;
|
|
SPIEnumInterfaceInfo spEnumInterface;
|
|
|
|
// Check to see if IP is enabled
|
|
// ------------------------------------------------------------
|
|
if (pRtrConfigData->m_ipData.m_dwEnableIn &&
|
|
pRtrConfigData->m_fIpSetup)
|
|
{
|
|
// If so, then we can add IGMP.
|
|
// --------------------------------------------------------
|
|
|
|
// Find the GUID for the IGMP Configuration.
|
|
// We get the list directly (rather than from the pRouter)
|
|
// because the data for the RtrMgrProtocols has not been
|
|
// loaded yet. The IRouterInfo only has information on the
|
|
// interfaces and not for the protocols (since the router
|
|
// is not yet configured).
|
|
// --------------------------------------------------------
|
|
RouterInfo::LoadInstalledRtrMgrProtocolList(pRtrConfigData->m_stServerName,
|
|
PID_IP, &SRmProtCBList, pRouter);
|
|
|
|
|
|
// Now iterate through this list looking for the igmp entry.
|
|
// ------------------------------------------------------------
|
|
pos = SRmProtCBList.GetHeadPosition();
|
|
while (pos)
|
|
{
|
|
pSRmProtCB = SRmProtCBList.GetNext(pos);
|
|
if ((pSRmProtCB->dwTransportId == PID_IP) &&
|
|
(pSRmProtCB->dwProtocolId == MS_IP_IGMP))
|
|
{
|
|
guidConfig = pSRmProtCB->guidConfig;
|
|
break;
|
|
}
|
|
}
|
|
|
|
if (guidConfig == GUID_RouterNull)
|
|
goto Error;
|
|
|
|
// Now add IGMP.
|
|
// --------------------------------------------------------
|
|
CORg( CoCreateProtocolConfig(guidConfig,
|
|
pRouter,
|
|
PID_IP,
|
|
MS_IP_IGMP,
|
|
&spRouterConfig) );
|
|
|
|
if (spRouterConfig)
|
|
hr = spRouterConfig->AddProtocol(pRtrConfigData->m_stServerName,
|
|
PID_IP,
|
|
MS_IP_IGMP,
|
|
NULL,
|
|
0,
|
|
pRouter,
|
|
0);
|
|
CORg( hr );
|
|
|
|
// In addition, we will also need to add IGMP router to the
|
|
// internal interface and IGMP proxy to one of the LAN
|
|
// interfaces.
|
|
// ------------------------------------------------------------
|
|
|
|
|
|
// Do we have the router managers for the interfaces?
|
|
// ------------------------------------------------------------
|
|
|
|
pRouter->EnumInterface(&spEnumInterface);
|
|
fFoundDedicatedInterface = FALSE;
|
|
|
|
for (spEnumInterface->Reset();
|
|
spEnumInterface->Next(1, &spIf, NULL) == hrOK;
|
|
spIf.Release())
|
|
{
|
|
dwIfType = spIf->GetInterfaceType();
|
|
|
|
// Add IGMP if this is an internal interface or
|
|
// if this is the first dedicated interface.
|
|
// --------------------------------------------------------
|
|
if ((dwIfType == ROUTER_IF_TYPE_INTERNAL) ||
|
|
(!fFoundDedicatedInterface &&
|
|
(dwIfType == ROUTER_IF_TYPE_DEDICATED)))
|
|
{
|
|
// Ok, add IGMP to this interface
|
|
// ----------------------------------------------------
|
|
|
|
// If this is a dedicated interface and a private network
|
|
// is specified, use that.
|
|
// ----------------------------------------------------
|
|
if ((dwIfType == ROUTER_IF_TYPE_DEDICATED) &&
|
|
!pRtrConfigData->m_ipData.m_stNetworkAdapterGUID.IsEmpty() &&
|
|
(pRtrConfigData->m_ipData.m_stNetworkAdapterGUID != spIf->GetId()))
|
|
continue;
|
|
|
|
// Get the IP Router Manager
|
|
// ----------------------------------------------------
|
|
spRmIf.Release();
|
|
CORg( spIf->FindRtrMgrInterface(PID_IP, &spRmIf) );
|
|
|
|
if (!spRmIf)
|
|
break;
|
|
|
|
if (dwIfType == ROUTER_IF_TYPE_DEDICATED)
|
|
fFoundDedicatedInterface = TRUE;
|
|
|
|
if (dwIfType == ROUTER_IF_TYPE_INTERNAL)
|
|
AddIGMPToInterface(spIf, TRUE /* fRouter */);
|
|
else
|
|
AddIGMPToInterface(spIf, FALSE /* fRouter */);
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
Error:
|
|
while (!SRmProtCBList.IsEmpty())
|
|
delete SRmProtCBList.RemoveHead();
|
|
|
|
return hr;
|
|
}
|
|
|
|
|
|
/*!--------------------------------------------------------------------------
|
|
AddIGMPToNATServer
|
|
-
|
|
Author: KennT
|
|
---------------------------------------------------------------------------*/
|
|
HRESULT AddIGMPToNATServer(NewRtrWizData *pNewRtrWizData,
|
|
RtrConfigData *pRtrConfigData,
|
|
IRouterInfo *pRouter,
|
|
BOOL fAddInternal)
|
|
{
|
|
HRESULT hr = hrOK;
|
|
SPIRouterProtocolConfig spRouterConfig;
|
|
SRtrMgrProtocolCBList SRmProtCBList;
|
|
POSITION pos;
|
|
SRtrMgrProtocolCB * pSRmProtCB;
|
|
GUID guidConfig = GUID_RouterNull;
|
|
SPIInterfaceInfo spIf;
|
|
SPIEnumInterfaceInfo spEnumInterface;
|
|
DWORD dwIfType;
|
|
CString stTempGUID;
|
|
|
|
Assert(pNewRtrWizData);
|
|
|
|
// Check to see if IP is enabled
|
|
// ------------------------------------------------------------
|
|
if (pRtrConfigData->m_ipData.m_dwEnableIn &&
|
|
pRtrConfigData->m_fIpSetup)
|
|
{
|
|
// If so, then we can add IGMP.
|
|
// --------------------------------------------------------
|
|
|
|
// Find the GUID for the IGMP Configuration.
|
|
// We get the list directly (rather than from the pRouter)
|
|
// because the data for the RtrMgrProtocols has not been
|
|
// loaded yet. The IRouterInfo only has information on the
|
|
// interfaces and not for the protocols (since the router
|
|
// is not yet configured).
|
|
// --------------------------------------------------------
|
|
RouterInfo::LoadInstalledRtrMgrProtocolList(pRtrConfigData->m_stServerName,
|
|
PID_IP, &SRmProtCBList, pRouter);
|
|
|
|
|
|
// Now iterate through this list looking for the igmp entry.
|
|
// ------------------------------------------------------------
|
|
pos = SRmProtCBList.GetHeadPosition();
|
|
while (pos)
|
|
{
|
|
pSRmProtCB = SRmProtCBList.GetNext(pos);
|
|
if ((pSRmProtCB->dwTransportId == PID_IP) &&
|
|
(pSRmProtCB->dwProtocolId == MS_IP_IGMP))
|
|
{
|
|
guidConfig = pSRmProtCB->guidConfig;
|
|
break;
|
|
}
|
|
}
|
|
|
|
if (guidConfig == GUID_RouterNull)
|
|
goto Error;
|
|
|
|
// Now add IGMP.
|
|
// --------------------------------------------------------
|
|
CORg( CoCreateProtocolConfig(guidConfig,
|
|
pRouter,
|
|
PID_IP,
|
|
MS_IP_IGMP,
|
|
&spRouterConfig) );
|
|
|
|
if (spRouterConfig)
|
|
hr = spRouterConfig->AddProtocol(pRtrConfigData->m_stServerName,
|
|
PID_IP,
|
|
MS_IP_IGMP,
|
|
NULL,
|
|
0,
|
|
pRouter,
|
|
0);
|
|
CORg( hr );
|
|
|
|
if (pNewRtrWizData->m_wizType == NewRtrWizData::NewWizardRouterType_VPNandNAT)
|
|
{
|
|
stTempGUID = pNewRtrWizData->m_stNATPrivateInterfaceId;
|
|
}
|
|
else
|
|
{
|
|
stTempGUID = pNewRtrWizData->m_stPrivateInterfaceId;
|
|
}
|
|
|
|
pRouter->EnumInterface(&spEnumInterface);
|
|
|
|
for (spEnumInterface->Reset();
|
|
spEnumInterface->Next(1, &spIf, NULL) == hrOK;
|
|
spIf.Release())
|
|
{
|
|
if (pRtrConfigData->m_ipData.m_stPublicAdapterGUID.CompareNoCase(spIf->GetId()) == 0)
|
|
{
|
|
// Ok, add the public interface as the proxy
|
|
AddIGMPToInterface(spIf, FALSE /* fRouter */);
|
|
}
|
|
|
|
if (stTempGUID.CompareNoCase(spIf->GetId()) == 0)
|
|
{
|
|
// Ok, add the private interface as the router
|
|
AddIGMPToInterface(spIf, TRUE /* fRouter */);
|
|
}
|
|
|
|
dwIfType = spIf->GetInterfaceType();
|
|
if ( fAddInternal && dwIfType == ROUTER_IF_TYPE_INTERNAL)
|
|
{
|
|
//
|
|
//If we have been instructed to add internal interface
|
|
//then add it.
|
|
//
|
|
AddIGMPToInterface(spIf, TRUE /* fRouter */);
|
|
}
|
|
|
|
|
|
}
|
|
|
|
}
|
|
|
|
Error:
|
|
while (!SRmProtCBList.IsEmpty())
|
|
delete SRmProtCBList.RemoveHead();
|
|
|
|
return hr;
|
|
}
|
|
|
|
|
|
HRESULT AddIGMPToInterface(IInterfaceInfo *pIf, BOOL fRouter)
|
|
{
|
|
HRESULT hr = hrOK;
|
|
SPIRtrMgrInterfaceInfo spRmIf;
|
|
SPIInfoBase spInfoBase;
|
|
IGMP_MIB_IF_CONFIG igmpConfig;
|
|
BOOL bVersion2=TRUE;
|
|
CString str;
|
|
|
|
// Get the IP Router Manager
|
|
// ----------------------------------------------------
|
|
CORg( pIf->FindRtrMgrInterface(PID_IP, &spRmIf) );
|
|
if (spRmIf == NULL)
|
|
CORg( E_FAIL );
|
|
|
|
CORg( spRmIf->GetInfoBase(NULL, NULL, NULL, &spInfoBase) );
|
|
|
|
igmpConfig = g_IGMPLanDefault;
|
|
{
|
|
if(pIf)
|
|
{
|
|
CString szMachineName;
|
|
LPWSTR lpcwMachine = NULL;
|
|
|
|
WKSTA_INFO_100* pWkstaInfo100 = NULL;
|
|
|
|
szMachineName = pIf->GetMachineName();
|
|
|
|
if(!szMachineName.IsEmpty() && szMachineName[0] != L'\\')
|
|
// append \\ prefix the machine name before call NetWks
|
|
{
|
|
str = L"\\\\";
|
|
str += szMachineName;
|
|
lpcwMachine = (LPWSTR)(LPCWSTR)str;
|
|
|
|
}
|
|
|
|
if( NERR_Success == NetWkstaGetInfo(lpcwMachine, 100, (LPBYTE*)&pWkstaInfo100))
|
|
{
|
|
Assert(pWkstaInfo100);
|
|
|
|
if(pWkstaInfo100->wki100_ver_major > 5 || ( pWkstaInfo100->wki100_ver_major == 5 && pWkstaInfo100->wki100_ver_minor > 0))
|
|
// dont support IGMPv3
|
|
bVersion2 = FALSE;
|
|
NetApiBufferFree(pWkstaInfo100);
|
|
}
|
|
}
|
|
}
|
|
|
|
if (fRouter)
|
|
{
|
|
igmpConfig.IgmpProtocolType = bVersion2? IGMP_ROUTER_V2: IGMP_ROUTER_V3;
|
|
}
|
|
else
|
|
igmpConfig.IgmpProtocolType = IGMP_PROXY;
|
|
|
|
if (bVersion2)
|
|
igmpConfig.Version = IGMP_VERSION_1_2;
|
|
|
|
|
|
CORg( spInfoBase->AddBlock(MS_IP_IGMP,
|
|
sizeof(IGMP_MIB_IF_CONFIG),
|
|
(LPBYTE) &igmpConfig,
|
|
1,
|
|
TRUE) );
|
|
|
|
CORg( spRmIf->Save(pIf->GetMachineName(),
|
|
NULL, NULL, NULL, spInfoBase, 0) );
|
|
|
|
Error:
|
|
return hr;
|
|
}
|
|
|
|
/*!--------------------------------------------------------------------------
|
|
RtrConfigData::Init
|
|
-
|
|
Author: KennT
|
|
---------------------------------------------------------------------------*/
|
|
HRESULT RtrConfigData::Init(LPCTSTR pszServerName, IRouterInfo *pRouter)
|
|
{
|
|
RouterVersionInfo routerVersion;
|
|
BOOL fNt4 = FALSE;
|
|
BOOL fRemote;
|
|
|
|
pRouter->GetRouterVersionInfo(&routerVersion);
|
|
fNt4 = (routerVersion.dwRouterVersion == 4);
|
|
|
|
m_fRemote = !IsLocalMachine(pszServerName);
|
|
|
|
m_stServerName = pszServerName;
|
|
|
|
// Load data from the registry
|
|
m_ipData.UseDefaults(pszServerName, fNt4);
|
|
m_ipxData.UseDefaults(pszServerName, fNt4);
|
|
m_nbfData.UseDefaults(pszServerName, fNt4);
|
|
m_arapData.UseDefaults(pszServerName, fNt4);
|
|
m_errlogData.UseDefaults(pszServerName, fNt4);
|
|
m_authData.UseDefaults(pszServerName, fNt4);
|
|
|
|
|
|
// Determine what protocols are installed
|
|
// ----------------------------------------------------------------
|
|
m_fUseIp = (HrIsProtocolSupported(pszServerName,
|
|
c_szRegKeyTcpip,
|
|
c_szRegKeyRasIp,
|
|
c_szRegKeyRasIpRtrMgr) == hrOK);
|
|
#if (WINVER >= 0x0501)
|
|
m_fUseIpx = FALSE;
|
|
#else
|
|
m_fUseIpx = (HrIsProtocolSupported(pszServerName,
|
|
c_szRegKeyNwlnkIpx,
|
|
c_szRegKeyRasIpx,
|
|
NULL) == hrOK);
|
|
#endif
|
|
m_fUseNbf = (HrIsProtocolSupported(pszServerName,
|
|
c_szRegKeyNbf,
|
|
c_szRegKeyRasNbf,
|
|
NULL) == hrOK);
|
|
|
|
m_fUseArap = (HrIsProtocolSupported(pszServerName,
|
|
c_szRegKeyAppletalk,
|
|
c_szRegKeyRasAppletalk,
|
|
NULL) == hrOK);
|
|
return hrOK;
|
|
}
|
|
|
|
|
|
#ifdef KSL_IPINIP
|
|
/*!--------------------------------------------------------------------------
|
|
CleanupTunnelFriendlyNames
|
|
Removes the list of Ip-in-Ip tunnel friendly names. This should
|
|
be used ONLY if we have already removed the interfaces.
|
|
Author: KennT
|
|
---------------------------------------------------------------------------*/
|
|
DWORD CleanupTunnelFriendlyNames(IRouterInfo *pRouter)
|
|
{
|
|
LPBYTE pBuffer = NULL;
|
|
DWORD dwErr = ERROR_SUCCESS;
|
|
DWORD dwEntriesRead = 0;
|
|
MPR_IPINIP_INTERFACE_0 *pTunnel0 = NULL;
|
|
|
|
// Get the list of Ip-in-Ip tunnels
|
|
// ----------------------------------------------------------------
|
|
dwErr = MprSetupIpInIpInterfaceFriendlyNameEnum((LPWSTR) pRouter->GetMachineName(),
|
|
&pBuffer,
|
|
&dwEntriesRead);
|
|
pTunnel0 = (MPR_IPINIP_INTERFACE_0 *) pBuffer;
|
|
|
|
if (dwErr == ERROR_SUCCESS)
|
|
{
|
|
// Now go through the tunnels and delete all of them
|
|
// ------------------------------------------------------------
|
|
for (DWORD i=0; i<dwEntriesRead; i++, pTunnel0++)
|
|
{
|
|
MprSetupIpInIpInterfaceFriendlyNameDelete(
|
|
(LPWSTR) pRouter->GetMachineName(),
|
|
&(pTunnel0->Guid));
|
|
}
|
|
}
|
|
|
|
// Free up the buffer returned from the enum
|
|
// ----------------------------------------------------------------
|
|
if (pBuffer)
|
|
MprSetupIpInIpInterfaceFriendlyNameFree(pBuffer);
|
|
|
|
return dwErr;
|
|
}
|
|
#endif //KSL_IPINIP
|
|
|
|
|
|
|
|
/*!--------------------------------------------------------------------------
|
|
AddNATToServer
|
|
-
|
|
Author: KennT
|
|
---------------------------------------------------------------------------*/
|
|
HRESULT AddNATToServer(NewRtrWizData *pNewRtrWizData,
|
|
RtrConfigData *pRtrConfigData,
|
|
IRouterInfo *pRouter,
|
|
BOOL fCreateDD,
|
|
BOOL fAddProtocolOnly //Do not add interfaces. Just add the protocol and nothing else
|
|
)
|
|
{
|
|
HRESULT hr = hrOK;
|
|
SPIRouterProtocolConfig spRouterConfig;
|
|
SRtrMgrProtocolCBList SRmProtCBList;
|
|
POSITION pos;
|
|
SRtrMgrProtocolCB * pSRmProtCB;
|
|
GUID guidConfig = GUID_RouterNull;
|
|
SPIInfoBase spInfoBase;
|
|
BOOL fSave = FALSE;
|
|
|
|
Assert(pNewRtrWizData);
|
|
Assert(pRtrConfigData);
|
|
|
|
Assert(pRtrConfigData->m_dwConfigFlags & RTRCONFIG_SETUP_NAT);
|
|
|
|
// Check to see if IP is enabled
|
|
// ------------------------------------------------------------
|
|
if (!pRtrConfigData->m_ipData.m_dwEnableIn ||
|
|
!pRtrConfigData->m_fIpSetup)
|
|
return hrOK;
|
|
|
|
// If so, then we can add NAT.
|
|
// --------------------------------------------------------
|
|
|
|
// Find the GUID for the NAT Configuration.
|
|
// Manually load this since the IRouterInfo has not yet
|
|
// loaded the RtrMgrProtocol info since the router is
|
|
// still in an unconfigured state.
|
|
// --------------------------------------------------------
|
|
RouterInfo::LoadInstalledRtrMgrProtocolList(pRtrConfigData->m_stServerName,
|
|
PID_IP, &SRmProtCBList, pRouter);
|
|
|
|
|
|
// Now iterate through this list looking for the nat entry.
|
|
// ------------------------------------------------------------
|
|
pos = SRmProtCBList.GetHeadPosition();
|
|
while (pos)
|
|
{
|
|
pSRmProtCB = SRmProtCBList.GetNext(pos);
|
|
if ((pSRmProtCB->dwTransportId == PID_IP) &&
|
|
(pSRmProtCB->dwProtocolId == MS_IP_NAT))
|
|
{
|
|
guidConfig = pSRmProtCB->guidConfig;
|
|
break;
|
|
}
|
|
}
|
|
|
|
if (guidConfig == GUID_RouterNull)
|
|
goto Error;
|
|
|
|
// Now add NAT.
|
|
// --------------------------------------------------------
|
|
CORg( CoCreateProtocolConfig(guidConfig,
|
|
pRouter,
|
|
PID_IP,
|
|
MS_IP_NAT,
|
|
&spRouterConfig) );
|
|
|
|
if (spRouterConfig)
|
|
hr = spRouterConfig->AddProtocol(pRtrConfigData->m_stServerName,
|
|
PID_IP,
|
|
MS_IP_NAT,
|
|
NULL,
|
|
0,
|
|
pRouter,
|
|
0);
|
|
CORg( hr );
|
|
|
|
if ( !fAddProtocolOnly )
|
|
{
|
|
// Check the flags to see if we have to add the DNS proxy
|
|
// and the DHCP allocator.
|
|
// ------------------------------------------------------------
|
|
|
|
// Get the router manager for IP
|
|
// We have to do this manually since the IRouterInfo will not
|
|
// have RtrMgr or RtrMgrProtocol information since the router
|
|
// is still in the unconfigured state.
|
|
// ------------------------------------------------------------
|
|
if (FHrSucceeded(hr))
|
|
CORg( AddNATSimpleServers(pNewRtrWizData, pRtrConfigData) );
|
|
|
|
|
|
// Now that we've added the DNS proxy/DHCP allocator, add
|
|
// NAT to the specific interfaces involved.
|
|
// ----------------------------------------------------------------
|
|
if (FHrSucceeded(hr))
|
|
CORg( AddNATToInterfaces(pNewRtrWizData, pRtrConfigData, pRouter, fCreateDD) );
|
|
}
|
|
|
|
|
|
Error:
|
|
while (!SRmProtCBList.IsEmpty())
|
|
delete SRmProtCBList.RemoveHead();
|
|
|
|
return hr;
|
|
}
|
|
|
|
|
|
|
|
/*!--------------------------------------------------------------------------
|
|
AddNATSimpleServers
|
|
-
|
|
Author: KennT
|
|
---------------------------------------------------------------------------*/
|
|
HRESULT AddNATSimpleServers(NewRtrWizData *pNewRtrWizData,
|
|
RtrConfigData *pRtrConfigData)
|
|
{
|
|
HRESULT hr = hrOK;
|
|
DWORD dwErr = ERROR_SUCCESS;
|
|
DWORD dwErrT = ERROR_SUCCESS;
|
|
MPR_SERVER_HANDLE hMprServer = NULL;
|
|
MPR_CONFIG_HANDLE hMprConfig = NULL;
|
|
LPBYTE pByte = NULL;
|
|
DWORD dwSize = 0;
|
|
SPIInfoBase spInfoBase;
|
|
HANDLE hTransport = NULL;
|
|
BOOL fSave = FALSE;
|
|
|
|
CORg( CreateInfoBase(&spInfoBase) );
|
|
|
|
// Connect to the server
|
|
// ----------------------------------------------------------------
|
|
dwErr = MprAdminServerConnect((LPWSTR) (LPCTSTR) pRtrConfigData->m_stServerName, &hMprServer);
|
|
if (dwErr == ERROR_SUCCESS)
|
|
{
|
|
// Ok, get the infobase from the server
|
|
dwErr = MprAdminTransportGetInfo(hMprServer,
|
|
PID_IP,
|
|
&pByte,
|
|
&dwSize,
|
|
NULL,
|
|
NULL);
|
|
|
|
if (dwErr == ERROR_SUCCESS)
|
|
{
|
|
spInfoBase->LoadFrom(dwSize, pByte);
|
|
|
|
MprAdminBufferFree(pByte);
|
|
pByte = NULL;
|
|
dwSize = 0;
|
|
}
|
|
}
|
|
|
|
// We also have to open the hMprConfig, but we can ignore the error
|
|
dwErrT = MprConfigServerConnect((LPWSTR) (LPCTSTR) pRtrConfigData->m_stServerName, &hMprConfig);
|
|
if (dwErrT == ERROR_SUCCESS)
|
|
{
|
|
dwErrT = MprConfigTransportGetHandle(hMprConfig, PID_IP, &hTransport);
|
|
}
|
|
|
|
if (dwErr != ERROR_SUCCESS)
|
|
{
|
|
// Ok, try to use the MprConfig calls.
|
|
CWRg( MprConfigTransportGetInfo(hMprConfig,
|
|
hTransport,
|
|
&pByte,
|
|
&dwSize,
|
|
NULL,
|
|
NULL,
|
|
NULL) );
|
|
|
|
|
|
spInfoBase->LoadFrom(dwSize, pByte);
|
|
|
|
MprConfigBufferFree(pByte);
|
|
pByte = NULL;
|
|
dwSize = 0;
|
|
}
|
|
|
|
Assert(spInfoBase);
|
|
|
|
// deonb - add H323 & Directplay support
|
|
if (pRtrConfigData->m_dwConfigFlags & RTRCONFIG_SETUP_H323)
|
|
{
|
|
IP_H323_GLOBAL_INFO globalInfo;
|
|
|
|
globalInfo = *( (IP_H323_GLOBAL_INFO *)g_pH323GlobalDefault);
|
|
|
|
CORg( spInfoBase->AddBlock(MS_IP_H323,
|
|
sizeof(IP_H323_GLOBAL_INFO),
|
|
(LPBYTE) &globalInfo, 1, TRUE));
|
|
fSave = TRUE;
|
|
}
|
|
// deonb - add H323 & Directplay support <end>
|
|
|
|
// savasg - add ALG support
|
|
if (pRtrConfigData->m_dwConfigFlags & RTRCONFIG_SETUP_ALG)
|
|
{
|
|
IP_ALG_GLOBAL_INFO globalInfo;
|
|
|
|
globalInfo = *( (IP_ALG_GLOBAL_INFO *)g_pAlgGlobalDefault);
|
|
|
|
CORg( spInfoBase->AddBlock(MS_IP_ALG,
|
|
sizeof(IP_ALG_GLOBAL_INFO),
|
|
(LPBYTE) &globalInfo, 1, TRUE));
|
|
fSave = TRUE;
|
|
} // savasg - end
|
|
|
|
if (pRtrConfigData->m_dwConfigFlags & RTRCONFIG_SETUP_DNS_PROXY)
|
|
{
|
|
IP_DNS_PROXY_GLOBAL_INFO globalInfo;
|
|
|
|
globalInfo = *( (IP_DNS_PROXY_GLOBAL_INFO *)g_pDnsProxyGlobalDefault);
|
|
|
|
// Windows NT Bug : 393749
|
|
// Remove the WINS flag
|
|
globalInfo.Flags &= ~IP_DNS_PROXY_FLAG_ENABLE_WINS;
|
|
|
|
CORg( spInfoBase->AddBlock(MS_IP_DNS_PROXY,
|
|
sizeof(IP_DNS_PROXY_GLOBAL_INFO),
|
|
(LPBYTE) &globalInfo, 1, TRUE));
|
|
fSave = TRUE;
|
|
}
|
|
|
|
if (pRtrConfigData->m_dwConfigFlags & RTRCONFIG_SETUP_DHCP_ALLOCATOR)
|
|
{
|
|
IP_AUTO_DHCP_GLOBAL_INFO dhcpGlobalInfo;
|
|
RtrWizInterface * pRtrWizIf = NULL;
|
|
CString stTempInterfaceId;
|
|
|
|
dhcpGlobalInfo = * ( (IP_AUTO_DHCP_GLOBAL_INFO *) g_pAutoDhcpGlobalDefault);
|
|
|
|
// Windows NT Bug : 385112
|
|
// Due to the problems with changing the IP Address of the
|
|
// adapter, let's just set the DHCP scope to be the scope of the
|
|
// underlying subnet.
|
|
|
|
if ( pNewRtrWizData->m_wizType == NewRtrWizData::NewWizardRouterType_VPNandNAT )
|
|
{
|
|
stTempInterfaceId = pNewRtrWizData->m_stNATPrivateInterfaceId;
|
|
}
|
|
else
|
|
{
|
|
stTempInterfaceId = pNewRtrWizData->m_stPrivateInterfaceId;
|
|
}
|
|
|
|
// Need to get the IP address of the private interface
|
|
pNewRtrWizData->m_ifMap.Lookup(stTempInterfaceId,
|
|
pRtrWizIf);
|
|
|
|
// If we cannot find this interface, go with the default values
|
|
// for the subnet/mask. Otherwise use the IP address of the private
|
|
// interface.
|
|
if (pRtrWizIf && !pRtrWizIf->m_stIpAddress.IsEmpty())
|
|
{
|
|
CString stFirstIp;
|
|
CString stFirstMask;
|
|
int iPos;
|
|
|
|
// Just take the first IP Address
|
|
stFirstIp = pRtrWizIf->m_stIpAddress;
|
|
iPos = pRtrWizIf->m_stIpAddress.Find(_T(','));
|
|
if (iPos >= 0)
|
|
stFirstIp = pRtrWizIf->m_stIpAddress.Left(iPos);
|
|
else
|
|
stFirstIp = pRtrWizIf->m_stIpAddress;
|
|
|
|
stFirstMask = pRtrWizIf->m_stMask;
|
|
iPos = pRtrWizIf->m_stMask.Find(_T(','));
|
|
if (iPos >= 0)
|
|
stFirstMask = pRtrWizIf->m_stMask.Left(iPos);
|
|
else
|
|
stFirstMask = pRtrWizIf->m_stMask;
|
|
|
|
// Now convert this into a net address
|
|
dhcpGlobalInfo.ScopeMask = INET_ADDR(stFirstMask);
|
|
dhcpGlobalInfo.ScopeNetwork = INET_ADDR(stFirstIp) & dhcpGlobalInfo.ScopeMask;
|
|
}
|
|
|
|
CORg( spInfoBase->AddBlock(MS_IP_DHCP_ALLOCATOR,
|
|
sizeof(dhcpGlobalInfo),
|
|
(PBYTE) &dhcpGlobalInfo, 1, TRUE) );
|
|
fSave = TRUE;
|
|
}
|
|
|
|
if (fSave)
|
|
{
|
|
spInfoBase->WriteTo(&pByte, &dwSize);
|
|
|
|
if (hMprServer)
|
|
{
|
|
MprAdminTransportSetInfo(hMprServer,
|
|
PID_IP,
|
|
pByte,
|
|
dwSize,
|
|
NULL,
|
|
0);
|
|
}
|
|
|
|
if (hMprConfig && hTransport)
|
|
{
|
|
MprConfigTransportSetInfo(hMprConfig,
|
|
hTransport,
|
|
pByte,
|
|
dwSize,
|
|
NULL,
|
|
NULL,
|
|
NULL);
|
|
}
|
|
|
|
if (pByte)
|
|
CoTaskMemFree(pByte);
|
|
}
|
|
|
|
Error:
|
|
if (hMprConfig)
|
|
MprConfigServerDisconnect(hMprConfig);
|
|
|
|
if (hMprServer)
|
|
MprAdminServerDisconnect(hMprServer);
|
|
|
|
return hr;
|
|
}
|
|
|
|
|
|
/*!--------------------------------------------------------------------------
|
|
AddNATToInterfaces
|
|
-
|
|
Author: KennT
|
|
---------------------------------------------------------------------------*/
|
|
HRESULT AddNATToInterfaces(NewRtrWizData *pNewRtrWizData,
|
|
RtrConfigData *pRtrConfigData,
|
|
IRouterInfo *pRouter,
|
|
BOOL fCreateDD)
|
|
{
|
|
HRESULT hr = hrOK;
|
|
DWORD dwErr = ERROR_SUCCESS;
|
|
DWORD dwErrT = ERROR_SUCCESS;
|
|
MPR_SERVER_HANDLE hMprServer = NULL;
|
|
MPR_CONFIG_HANDLE hMprConfig = NULL;
|
|
LPBYTE pByte = NULL;
|
|
DWORD dwSize = 0;
|
|
SPIInfoBase spInfoBase;
|
|
HANDLE hTransport = NULL;
|
|
CString stIfName;
|
|
CString stTempGUID;
|
|
DWORD dwIfType;
|
|
SPIEnumInterfaceInfo spEnumInterface;
|
|
SPIInterfaceInfo spIf;
|
|
|
|
|
|
|
|
CORg( CreateInfoBase(&spInfoBase) );
|
|
|
|
// Connect to the server
|
|
// ----------------------------------------------------------------
|
|
MprAdminServerConnect((LPWSTR) (LPCTSTR) pRtrConfigData->m_stServerName, &hMprServer);
|
|
|
|
MprConfigServerConnect((LPWSTR) (LPCTSTR) pRtrConfigData->m_stServerName, &hMprConfig);
|
|
|
|
// Install public NAT on public interface
|
|
AddNATToInterface(hMprServer,
|
|
hMprConfig,
|
|
pNewRtrWizData,
|
|
pRtrConfigData,
|
|
pRtrConfigData->m_ipData.m_stPublicAdapterGUID,
|
|
fCreateDD,
|
|
TRUE);
|
|
|
|
if ( (pNewRtrWizData->m_wizType == NewRtrWizData::NewWizardRouterType_NAT) ||
|
|
(pNewRtrWizData->m_wizType == NewRtrWizData::NewWizardRouterType_VPNandNAT) )
|
|
{
|
|
// Install private NAT on private interface
|
|
if (pNewRtrWizData->m_wizType == NewRtrWizData::NewWizardRouterType_VPNandNAT)
|
|
{
|
|
stTempGUID = pNewRtrWizData->m_stNATPrivateInterfaceId;
|
|
}
|
|
else
|
|
{
|
|
stTempGUID = pNewRtrWizData->m_stPrivateInterfaceId;
|
|
}
|
|
|
|
AddNATToInterface(hMprServer,
|
|
hMprConfig,
|
|
pNewRtrWizData,
|
|
pRtrConfigData,
|
|
stTempGUID,
|
|
FALSE,
|
|
FALSE);
|
|
}
|
|
|
|
//
|
|
//Need to add NAT on internal interface also in case this is a VPN/NAT
|
|
//path.
|
|
//
|
|
pRouter->EnumInterface(&spEnumInterface);
|
|
|
|
|
|
for (spEnumInterface->Reset();
|
|
spEnumInterface->Next(1, &spIf, NULL) == hrOK;
|
|
spIf.Release())
|
|
{
|
|
dwIfType = spIf->GetInterfaceType();
|
|
if (dwIfType == ROUTER_IF_TYPE_INTERNAL)
|
|
{
|
|
AddNATToInterface( hMprServer,
|
|
hMprConfig,
|
|
pNewRtrWizData,
|
|
pRtrConfigData,
|
|
spIf->GetId(),
|
|
FALSE,
|
|
FALSE);
|
|
break;
|
|
}
|
|
|
|
}
|
|
|
|
|
|
Error:
|
|
if (hMprConfig)
|
|
MprConfigServerDisconnect(hMprConfig);
|
|
|
|
if (hMprServer)
|
|
MprAdminServerDisconnect(hMprServer);
|
|
|
|
return hr;
|
|
}
|
|
|
|
//
|
|
// Default values for LAN-interface NAT configuration
|
|
//
|
|
IP_NAT_INTERFACE_INFO
|
|
g_ipnatLanDefault = {
|
|
0, // Index (unused)
|
|
0, // Flags
|
|
{ IP_NAT_VERSION, sizeof(RTR_INFO_BLOCK_HEADER), 0,
|
|
{ 0, 0, 0, 0}} // Header
|
|
};
|
|
|
|
BYTE* g_pIpnatLanDefault = (BYTE*)&g_ipnatLanDefault;
|
|
|
|
|
|
//
|
|
// Default values for WAN-interface NAT configuration
|
|
//
|
|
IP_NAT_INTERFACE_INFO
|
|
g_ipnatWanDefault = {
|
|
0, // Index (unused)
|
|
IP_NAT_INTERFACE_FLAGS_BOUNDARY|
|
|
IP_NAT_INTERFACE_FLAGS_NAPT, // Flags
|
|
{ IP_NAT_VERSION, sizeof(RTR_INFO_BLOCK_HEADER), 0,
|
|
{ 0, 0, 0, 0}} // Header
|
|
};
|
|
|
|
BYTE* g_pIpnatWanDefault = (BYTE*)&g_ipnatWanDefault;
|
|
|
|
|
|
DWORD CreatePortMappingsForVPNFilters(
|
|
NewRtrWizData *pNewRtrWizData,
|
|
IP_NAT_PORT_MAPPING **ppPortMappingsForVPNFilters,
|
|
DWORD *dwNumPortMappings)
|
|
{
|
|
DWORD i, j, dwSize, dwNumMappingsPerAddress;
|
|
DWORD dwIpAddress = 0;
|
|
CString singleAddr;
|
|
CString tempAddrList;
|
|
CDWordArray arrIpAddr;
|
|
RtrWizInterface *pIf = NULL;
|
|
IP_NAT_PORT_MAPPING *pMappings = NULL;
|
|
USES_CONVERSION;
|
|
|
|
//
|
|
// The set of generic Port Mappings corresponding to
|
|
// VPN server specific filters
|
|
//
|
|
IP_NAT_PORT_MAPPING
|
|
GenericPortMappingsArray[] =
|
|
{
|
|
{
|
|
NAT_PROTOCOL_TCP,
|
|
htons(1723),
|
|
htonl(IP_NAT_ADDRESS_UNSPECIFIED),
|
|
htons(1723),
|
|
htonl(INADDR_LOOPBACK)
|
|
},
|
|
|
|
{
|
|
NAT_PROTOCOL_UDP,
|
|
htons(500),
|
|
htonl(IP_NAT_ADDRESS_UNSPECIFIED),
|
|
htons(500),
|
|
htonl(INADDR_LOOPBACK)
|
|
},
|
|
|
|
{
|
|
NAT_PROTOCOL_UDP,
|
|
htons(1701),
|
|
htonl(IP_NAT_ADDRESS_UNSPECIFIED),
|
|
htons(1701),
|
|
htonl(INADDR_LOOPBACK)
|
|
},
|
|
|
|
{
|
|
NAT_PROTOCOL_UDP,
|
|
htons(4500),
|
|
htonl(IP_NAT_ADDRESS_UNSPECIFIED),
|
|
htons(4500),
|
|
htonl(INADDR_LOOPBACK)
|
|
}
|
|
|
|
};
|
|
|
|
dwSize = sizeof(GenericPortMappingsArray);
|
|
|
|
pMappings = (PIP_NAT_PORT_MAPPING) new BYTE[dwSize];
|
|
|
|
if ( pMappings == NULL )
|
|
{
|
|
*ppPortMappingsForVPNFilters = NULL;
|
|
*dwNumPortMappings = 0;
|
|
return ERROR_NOT_ENOUGH_MEMORY;
|
|
}
|
|
|
|
memcpy(
|
|
(LPVOID) pMappings,
|
|
(LPVOID) &(GenericPortMappingsArray[0]),
|
|
sizeof(GenericPortMappingsArray));
|
|
|
|
*ppPortMappingsForVPNFilters = pMappings;
|
|
*dwNumPortMappings =
|
|
sizeof(GenericPortMappingsArray)/sizeof(IP_NAT_PORT_MAPPING);
|
|
|
|
return ERROR_SUCCESS;
|
|
}
|
|
|
|
|
|
|
|
/*!--------------------------------------------------------------------------
|
|
AddNATToInterface
|
|
-
|
|
Author: KennT
|
|
---------------------------------------------------------------------------*/
|
|
HRESULT AddNATToInterface(MPR_SERVER_HANDLE hMprServer,
|
|
HANDLE hMprConfig,
|
|
NewRtrWizData *pNewRtrWizData,
|
|
RtrConfigData *pRtrConfigData,
|
|
LPCTSTR pszInterface,
|
|
BOOL fDemandDial,
|
|
BOOL fPublic)
|
|
{
|
|
HRESULT hr = hrOK;
|
|
HANDLE hInterface = NULL;
|
|
HANDLE hIfTransport = NULL;
|
|
LPBYTE pByte = NULL;
|
|
DWORD dwSize = 0, dwIfBlkSize = 0;
|
|
LPBYTE pDefault = NULL;
|
|
IP_NAT_INTERFACE_INFO ipnat;
|
|
PIP_NAT_INTERFACE_INFO pipnat = NULL;
|
|
DWORD dwErr = ERROR_SUCCESS;
|
|
DWORD dwNumPortMappings = 0;
|
|
PIP_NAT_PORT_MAPPING pPortMappingsForVPNFilters = NULL;
|
|
|
|
SPIInfoBase spInfoBase;
|
|
|
|
IP_DNS_PROXY_INTERFACE_INFO dnsIfInfo;
|
|
MIB_IPFORWARDROW row;
|
|
|
|
|
|
::ZeroMemory(&dnsIfInfo, sizeof(dnsIfInfo));
|
|
|
|
if ((pszInterface == NULL) || (*pszInterface == 0))
|
|
return hrOK;
|
|
|
|
// Setup the data structures
|
|
// ----------------------------------------------------------------
|
|
if (pNewRtrWizData->m_fSetVPNFilter && fPublic )
|
|
{
|
|
SPIInfoBase spIB;
|
|
|
|
CORg( CreateInfoBase( &spIB ) );
|
|
|
|
dwErr = CreatePortMappingsForVPNFilters(
|
|
pNewRtrWizData,
|
|
&pPortMappingsForVPNFilters,
|
|
&dwNumPortMappings
|
|
);
|
|
|
|
if ( dwErr == ERROR_SUCCESS && pPortMappingsForVPNFilters)
|
|
{
|
|
|
|
spIB->AddBlock(
|
|
IP_NAT_PORT_MAPPING_TYPE,
|
|
sizeof( IP_NAT_PORT_MAPPING ),
|
|
(PBYTE)pPortMappingsForVPNFilters,
|
|
dwNumPortMappings,
|
|
TRUE
|
|
);
|
|
|
|
spIB->WriteTo(&pByte, &dwIfBlkSize);
|
|
|
|
pipnat = (PIP_NAT_INTERFACE_INFO)
|
|
new BYTE[
|
|
FIELD_OFFSET(IP_NAT_INTERFACE_INFO, Header) +
|
|
dwIfBlkSize ];
|
|
|
|
memcpy(&(pipnat->Header), pByte, dwIfBlkSize);
|
|
|
|
if ( pNewRtrWizData->m_wizType != NewRtrWizData::NewWizardRouterType_VPNandNAT )
|
|
{
|
|
//
|
|
//If this is not a VPN NAT combo, then setup as firewall only
|
|
//
|
|
pipnat->Flags = IP_NAT_INTERFACE_FLAGS_FW;
|
|
}
|
|
else
|
|
{
|
|
//
|
|
//Otherwise this is a boundary also and firewall
|
|
//
|
|
pipnat->Flags = IP_NAT_INTERFACE_FLAGS_BOUNDARY |
|
|
IP_NAT_INTERFACE_FLAGS_NAPT |
|
|
IP_NAT_INTERFACE_FLAGS_FW;
|
|
}
|
|
|
|
pipnat->Index = 0;
|
|
pDefault = (LPBYTE) pipnat;
|
|
dwIfBlkSize += FIELD_OFFSET(IP_NAT_INTERFACE_INFO, Header);
|
|
|
|
CoTaskMemFree( pByte );
|
|
spIB.Release();
|
|
pByte = NULL;
|
|
}
|
|
else
|
|
{
|
|
ipnat = g_ipnatLanDefault;
|
|
pDefault = (LPBYTE) &ipnat;
|
|
dwIfBlkSize = sizeof(IP_NAT_INTERFACE_INFO);
|
|
}
|
|
}
|
|
|
|
else
|
|
{
|
|
if (fDemandDial)
|
|
ipnat = g_ipnatWanDefault;
|
|
else
|
|
ipnat = g_ipnatLanDefault;
|
|
|
|
if (fPublic)
|
|
{
|
|
ipnat.Flags |= IP_NAT_INTERFACE_FLAGS_BOUNDARY;
|
|
if ( pNewRtrWizData->m_fNATEnableFireWall == TRUE )
|
|
ipnat.Flags |= IP_NAT_INTERFACE_FLAGS_FW;
|
|
|
|
// Windows NT Bug : 393731
|
|
// This will enable the "Translate TCP/UDP headers in the UI"
|
|
// ------------------------------------------------------------
|
|
ipnat.Flags |= IP_NAT_INTERFACE_FLAGS_NAPT;
|
|
|
|
// Windows NT Bug : 393809
|
|
// Enable the DNS resolution
|
|
// ------------------------------------------------------------
|
|
if (fDemandDial)
|
|
dnsIfInfo.Flags |= IP_DNS_PROXY_INTERFACE_FLAG_DEFAULT;
|
|
}
|
|
|
|
pDefault = (LPBYTE) &ipnat;
|
|
dwIfBlkSize = sizeof(IP_NAT_INTERFACE_INFO);
|
|
}
|
|
|
|
|
|
::ZeroMemory(&row, sizeof(row));
|
|
|
|
// Windows Nt Bug : 389441
|
|
// If this is a demand-dial interface, we will have to add
|
|
// a static route to the interface
|
|
// ----------------------------------------------------------------
|
|
if (fDemandDial && fPublic)
|
|
{
|
|
// Note: this is a new interface so there should not be
|
|
// any blocks.
|
|
// ------------------------------------------------------------
|
|
// What is the index of the demand-dial interface?
|
|
row.dwForwardMetric1 = 1;
|
|
row.dwForwardProto = PROTO_IP_NT_STATIC;
|
|
}
|
|
|
|
|
|
|
|
CORg( CreateInfoBase( &spInfoBase ) );
|
|
|
|
// ok, we need to get the RmIf
|
|
if (hMprServer)
|
|
{
|
|
dwErr = MprAdminInterfaceGetHandle(hMprServer,
|
|
(LPWSTR) pszInterface,
|
|
&hInterface,
|
|
FALSE);
|
|
|
|
if (dwErr == ERROR_SUCCESS)
|
|
dwErr = MprAdminInterfaceTransportGetInfo(hMprServer,
|
|
hInterface,
|
|
PID_IP,
|
|
&pByte,
|
|
&dwSize);
|
|
|
|
if (dwErr == ERROR_SUCCESS)
|
|
{
|
|
spInfoBase->LoadFrom(dwSize, pByte);
|
|
MprAdminBufferFree(pByte);
|
|
}
|
|
|
|
pByte = NULL;
|
|
dwSize = 0;
|
|
|
|
if (dwErr == ERROR_SUCCESS)
|
|
{
|
|
// Manipulate the infobase
|
|
spInfoBase->AddBlock(MS_IP_NAT,
|
|
dwIfBlkSize,
|
|
pDefault,
|
|
1,
|
|
TRUE);
|
|
|
|
if (pRtrConfigData->m_dwConfigFlags & RTRCONFIG_SETUP_DNS_PROXY)
|
|
{
|
|
spInfoBase->AddBlock(MS_IP_DNS_PROXY,
|
|
sizeof(dnsIfInfo),
|
|
(LPBYTE) &dnsIfInfo,
|
|
1,
|
|
TRUE);
|
|
}
|
|
|
|
// Windows NT Bug : 389441
|
|
// Add the default route to the internet.
|
|
// --------------------------------------------------------
|
|
if (fDemandDial && fPublic)
|
|
{
|
|
// Note: this assumes that there are no routes
|
|
// already defined for this interface
|
|
// ----------------------------------------------------
|
|
if ( spInfoBase->BlockExists(IP_ROUTE_INFO) == S_FALSE)
|
|
{
|
|
//
|
|
//Add a route default route iff there are no
|
|
//routes already present for this dd interface
|
|
//
|
|
spInfoBase->AddBlock(IP_ROUTE_INFO,
|
|
sizeof(row),
|
|
(PBYTE) &row,
|
|
1, TRUE);
|
|
}
|
|
}
|
|
|
|
spInfoBase->WriteTo(&pByte, &dwSize);
|
|
}
|
|
|
|
if (dwErr == ERROR_SUCCESS)
|
|
{
|
|
MprAdminInterfaceTransportSetInfo(hMprServer,
|
|
hInterface,
|
|
PID_IP,
|
|
pByte,
|
|
dwSize);
|
|
}
|
|
|
|
if (pByte)
|
|
CoTaskMemFree(pByte);
|
|
pByte = NULL;
|
|
dwSize = 0;
|
|
}
|
|
|
|
hInterface = NULL;
|
|
|
|
if (hMprConfig)
|
|
{
|
|
dwErr = MprConfigInterfaceGetHandle(hMprConfig,
|
|
(LPWSTR) pszInterface,
|
|
&hInterface);
|
|
|
|
if (dwErr == ERROR_SUCCESS)
|
|
dwErr = MprConfigInterfaceTransportGetHandle(hMprConfig,
|
|
hInterface,
|
|
PID_IP,
|
|
&hIfTransport);
|
|
|
|
if (dwErr == ERROR_SUCCESS)
|
|
dwErr = MprConfigInterfaceTransportGetInfo(hMprConfig,
|
|
hInterface,
|
|
hIfTransport,
|
|
&pByte,
|
|
&dwSize);
|
|
|
|
if (dwErr == ERROR_SUCCESS)
|
|
{
|
|
spInfoBase->LoadFrom(dwSize, pByte);
|
|
MprConfigBufferFree(pByte);
|
|
}
|
|
|
|
pByte = NULL;
|
|
dwSize = 0;
|
|
|
|
if (dwErr == ERROR_SUCCESS)
|
|
{
|
|
// Manipulate the infobase
|
|
spInfoBase->AddBlock(MS_IP_NAT,
|
|
dwIfBlkSize,
|
|
pDefault,
|
|
1,
|
|
TRUE);
|
|
|
|
if (pRtrConfigData->m_dwConfigFlags & RTRCONFIG_SETUP_DNS_PROXY)
|
|
{
|
|
spInfoBase->AddBlock(MS_IP_DNS_PROXY,
|
|
sizeof(dnsIfInfo),
|
|
(LPBYTE) &dnsIfInfo,
|
|
1,
|
|
TRUE);
|
|
}
|
|
|
|
// Windows NT Bug : 389441
|
|
// Add the default route to the internet.
|
|
// --------------------------------------------------------
|
|
if (fDemandDial && fPublic)
|
|
{
|
|
// Note: this assumes that there are no routes
|
|
// already defined for this interface
|
|
// ----------------------------------------------------
|
|
if (spInfoBase->BlockExists(IP_ROUTE_INFO) == S_FALSE)
|
|
{
|
|
spInfoBase->AddBlock(IP_ROUTE_INFO,
|
|
sizeof(row),
|
|
(PBYTE) &row,
|
|
1, TRUE);
|
|
}
|
|
}
|
|
|
|
spInfoBase->WriteTo(&pByte, &dwSize);
|
|
}
|
|
|
|
if (dwErr == ERROR_SUCCESS)
|
|
{
|
|
MprConfigInterfaceTransportSetInfo(hMprConfig,
|
|
hInterface,
|
|
hIfTransport,
|
|
pByte,
|
|
dwSize);
|
|
}
|
|
|
|
if (pByte)
|
|
CoTaskMemFree(pByte);
|
|
pByte = NULL;
|
|
dwSize = 0;
|
|
|
|
spInfoBase.Release();
|
|
}
|
|
|
|
Error:
|
|
if (pipnat != NULL) { delete [] pipnat; }
|
|
return HResultFromWin32(dwErr);
|
|
}
|
|
|
|
|
|
|
|
/*!--------------------------------------------------------------------------
|
|
AddIPBOOTPToServer
|
|
If dwDhcpServer is 0, then we do not set it in the global list.
|
|
|
|
dwDhcpServer is the IP address of the DHCP Server in network order.
|
|
Author: KennT
|
|
---------------------------------------------------------------------------*/
|
|
HRESULT AddIPBOOTPToServer(RtrConfigData *pRtrConfigData,
|
|
IRouterInfo *pRouter,
|
|
DWORD dwDhcpServer)
|
|
{
|
|
HRESULT hr = hrOK;
|
|
SPIRouterProtocolConfig spRouterConfig;
|
|
SRtrMgrProtocolCBList SRmProtCBList;
|
|
POSITION pos;
|
|
SRtrMgrProtocolCB * pSRmProtCB;
|
|
GUID guidConfig = GUID_RouterNull;
|
|
SPIInterfaceInfo spIf;
|
|
SPIEnumInterfaceInfo spEnumInterface;
|
|
|
|
// Check to see if IP is enabled
|
|
// ------------------------------------------------------------
|
|
if (pRtrConfigData->m_ipData.m_dwEnableIn &&
|
|
pRtrConfigData->m_fIpSetup)
|
|
{
|
|
// If so, then we can add IPBOOTP.
|
|
// --------------------------------------------------------
|
|
|
|
// Find the GUID for the IPBOOTP Configuration.
|
|
// We get the list directly (rather than from the pRouter)
|
|
// because the data for the RtrMgrProtocols has not been
|
|
// loaded yet. The IRouterInfo only has information on the
|
|
// interfaces and not for the protocols (since the router
|
|
// is not yet configured).
|
|
// --------------------------------------------------------
|
|
RouterInfo::LoadInstalledRtrMgrProtocolList(pRtrConfigData->m_stServerName,
|
|
PID_IP, &SRmProtCBList, pRouter);
|
|
|
|
|
|
// Now iterate through this list looking for the igmp entry.
|
|
// ------------------------------------------------------------
|
|
pos = SRmProtCBList.GetHeadPosition();
|
|
while (pos)
|
|
{
|
|
pSRmProtCB = SRmProtCBList.GetNext(pos);
|
|
if ((pSRmProtCB->dwTransportId == PID_IP) &&
|
|
(pSRmProtCB->dwProtocolId == MS_IP_BOOTP))
|
|
{
|
|
guidConfig = pSRmProtCB->guidConfig;
|
|
break;
|
|
}
|
|
}
|
|
|
|
if (guidConfig == GUID_RouterNull)
|
|
goto Error;
|
|
|
|
// Now add IGMP.
|
|
// --------------------------------------------------------
|
|
CORg( CoCreateProtocolConfig(guidConfig,
|
|
pRouter,
|
|
PID_IP,
|
|
MS_IP_BOOTP,
|
|
&spRouterConfig) );
|
|
|
|
if (spRouterConfig)
|
|
hr = spRouterConfig->AddProtocol(pRtrConfigData->m_stServerName,
|
|
PID_IP,
|
|
MS_IP_BOOTP,
|
|
NULL,
|
|
0,
|
|
pRouter,
|
|
0);
|
|
CORg( hr );
|
|
|
|
// In order to do this, we'll have to get the IPBOOTP global
|
|
// info and add the server to the list.
|
|
// ------------------------------------------------------------
|
|
if ((dwDhcpServer != 0) &&
|
|
(dwDhcpServer != MAKEIPADDRESS(255,255,255,255)))
|
|
{
|
|
AddDhcpServerToBOOTPGlobalInfo(pRtrConfigData->m_stServerName,
|
|
dwDhcpServer);
|
|
}
|
|
|
|
pRouter->EnumInterface(&spEnumInterface);
|
|
|
|
for (spEnumInterface->Reset();
|
|
spEnumInterface->Next(1, &spIf, NULL) == hrOK;
|
|
spIf.Release())
|
|
{
|
|
|
|
// Look for the internal interface
|
|
if (spIf->GetInterfaceType() == ROUTER_IF_TYPE_INTERNAL)
|
|
{
|
|
AddIPBOOTPToInterface(spIf);
|
|
break;
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
Error:
|
|
while (!SRmProtCBList.IsEmpty())
|
|
delete SRmProtCBList.RemoveHead();
|
|
|
|
return hr;
|
|
}
|
|
|
|
|
|
|
|
/*!--------------------------------------------------------------------------
|
|
AddIPBOOTPToInterface
|
|
-
|
|
Author: KennT
|
|
---------------------------------------------------------------------------*/
|
|
HRESULT AddIPBOOTPToInterface(IInterfaceInfo *pIf)
|
|
{
|
|
HRESULT hr = hrOK;
|
|
SPIRtrMgrInterfaceInfo spRmIf;
|
|
SPIInfoBase spInfoBase;
|
|
|
|
// Get the IP Router Manager
|
|
// ----------------------------------------------------
|
|
CORg( pIf->FindRtrMgrInterface(PID_IP, &spRmIf) );
|
|
if (spRmIf == NULL)
|
|
CORg( E_FAIL );
|
|
|
|
CORg( spRmIf->GetInfoBase(NULL, NULL, NULL, &spInfoBase) );
|
|
|
|
CORg( spInfoBase->AddBlock(MS_IP_BOOTP,
|
|
sizeof(IPBOOTP_IF_CONFIG),
|
|
(LPBYTE) &g_relayLanDefault,
|
|
1,
|
|
TRUE) );
|
|
|
|
CORg( spRmIf->Save(pIf->GetMachineName(),
|
|
NULL, NULL, NULL, spInfoBase, 0) );
|
|
|
|
Error:
|
|
return hr;
|
|
}
|
|
|
|
|
|
HRESULT AddDhcpServerToBOOTPGlobalInfo(LPCTSTR pszServerName,
|
|
DWORD netDhcpServer)
|
|
{
|
|
HRESULT hr = hrOK;
|
|
DWORD dwErr = ERROR_SUCCESS;
|
|
DWORD dwErrT = ERROR_SUCCESS;
|
|
MPR_SERVER_HANDLE hMprServer = NULL;
|
|
MPR_CONFIG_HANDLE hMprConfig = NULL;
|
|
LPBYTE pByte = NULL;
|
|
DWORD dwSize = 0;
|
|
SPIInfoBase spInfoBase;
|
|
HANDLE hTransport = NULL;
|
|
BOOL fSave = FALSE;
|
|
IPBOOTP_GLOBAL_CONFIG * pgc = NULL;
|
|
IPBOOTP_GLOBAL_CONFIG * pgcNew = NULL;
|
|
|
|
CORg( CreateInfoBase(&spInfoBase) );
|
|
|
|
// Connect to the server
|
|
// ----------------------------------------------------------------
|
|
dwErr = MprAdminServerConnect((LPWSTR) pszServerName, &hMprServer);
|
|
if (dwErr == ERROR_SUCCESS)
|
|
{
|
|
// Ok, get the infobase from the server
|
|
dwErr = MprAdminTransportGetInfo(hMprServer,
|
|
PID_IP,
|
|
&pByte,
|
|
&dwSize,
|
|
NULL,
|
|
NULL);
|
|
|
|
if (dwErr == ERROR_SUCCESS)
|
|
{
|
|
spInfoBase->LoadFrom(dwSize, pByte);
|
|
|
|
MprAdminBufferFree(pByte);
|
|
pByte = NULL;
|
|
dwSize = 0;
|
|
}
|
|
}
|
|
|
|
// We also have to open the hMprConfig, but we can ignore the error
|
|
// ----------------------------------------------------------------
|
|
dwErrT = MprConfigServerConnect((LPWSTR) pszServerName, &hMprConfig);
|
|
if (dwErrT == ERROR_SUCCESS)
|
|
{
|
|
dwErrT = MprConfigTransportGetHandle(hMprConfig, PID_IP, &hTransport);
|
|
}
|
|
|
|
if (dwErr != ERROR_SUCCESS)
|
|
{
|
|
// No errors from the MprConfig calls
|
|
// ------------------------------------------------------------
|
|
CWRg( dwErrT );
|
|
|
|
// Ok, try to use the MprConfig calls.
|
|
// ------------------------------------------------------------
|
|
CWRg( MprConfigTransportGetInfo(hMprConfig,
|
|
hTransport,
|
|
&pByte,
|
|
&dwSize,
|
|
NULL,
|
|
NULL,
|
|
NULL) );
|
|
|
|
|
|
spInfoBase->LoadFrom(dwSize, pByte);
|
|
|
|
MprConfigBufferFree(pByte);
|
|
pByte = NULL;
|
|
dwSize = 0;
|
|
}
|
|
|
|
|
|
Assert(spInfoBase);
|
|
|
|
// Ok, get the current global config and add on this particular
|
|
// DHCP server
|
|
// ----------------------------------------------------------------
|
|
spInfoBase->GetData(MS_IP_BOOTP, 0, (PBYTE *) &pgc);
|
|
|
|
|
|
// Resize the struct for the increased address
|
|
// ----------------------------------------------------------------
|
|
dwSize = sizeof(IPBOOTP_GLOBAL_CONFIG) +
|
|
((pgc->GC_ServerCount + 1) * sizeof(DWORD));
|
|
pgcNew = (IPBOOTP_GLOBAL_CONFIG *) new BYTE[dwSize];
|
|
|
|
|
|
// Copy over the original information
|
|
// ----------------------------------------------------------------
|
|
CopyMemory(pgcNew, pgc, IPBOOTP_GLOBAL_CONFIG_SIZE(pgc));
|
|
|
|
|
|
// Add in the new DHCP server
|
|
// ----------------------------------------------------------------
|
|
IPBOOTP_GLOBAL_SERVER_TABLE(pgcNew)[pgc->GC_ServerCount] = netDhcpServer;
|
|
pgcNew->GC_ServerCount++;
|
|
|
|
spInfoBase->AddBlock(MS_IP_BOOTP,
|
|
dwSize,
|
|
(LPBYTE) pgcNew,
|
|
1,
|
|
TRUE);
|
|
|
|
|
|
spInfoBase->WriteTo(&pByte, &dwSize);
|
|
|
|
if (hMprServer)
|
|
{
|
|
MprAdminTransportSetInfo(hMprServer,
|
|
PID_IP,
|
|
pByte,
|
|
dwSize,
|
|
NULL,
|
|
0);
|
|
}
|
|
|
|
if (hMprConfig && hTransport)
|
|
{
|
|
MprConfigTransportSetInfo(hMprConfig,
|
|
hTransport,
|
|
pByte,
|
|
dwSize,
|
|
NULL,
|
|
NULL,
|
|
NULL);
|
|
}
|
|
|
|
if (pByte)
|
|
CoTaskMemFree(pByte);
|
|
|
|
Error:
|
|
delete [] pgcNew;
|
|
|
|
if (hMprConfig)
|
|
MprConfigServerDisconnect(hMprConfig);
|
|
|
|
if (hMprServer)
|
|
MprAdminServerDisconnect(hMprServer);
|
|
|
|
return hr;
|
|
|
|
}
|