able/windows-nt
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity
windows-nt/Source/XPSP1/NT/base/ntsetup/oobe/msobcomm/inets.cpp

783 lines
29 KiB
C++
Raw Normal View History

Add source files
2020-09-26 03:20:57 -05:00
// inets.cpp: implementation of the CInetSetup class.
//
//////////////////////////////////////////////////////////////////////
#include "inets.h"
#include <ras.h>
#include <tapi.h>
#include <stdio.h>
#include <string.h>
#include <appdefs.h>
#define MAXNAME 200
typedef DWORD (WINAPI *LPFNDLL_RASGETENTRYPROPERTIES)(LPCWSTR, LPCWSTR, LPRASENTRY, LPDWORD, LPBYTE, LPDWORD);
typedef DWORD (WINAPI *LPFNDLL_RASSETENTRYPROPERTIES)(LPCWSTR, LPCWSTR, LPRASENTRY, DWORD , LPBYTE, DWORD );
/////////////////////////////////////////////////////////////////////////
// Registry Values //
/////////////////////////////////////////////////////////////////////////
// For CInetSetup::InetSSetLanConnection
//static const WCHAR cszRegEnumPciKey[] = L"Enum\\PCI";
//static const WCHAR cszRegEnumNetworkTcpKey[] = L"Enum\\Network\\MSTCP";
//static const WCHAR cszRegClassKey[] = L"System\\CurrentControlSet\\Services\\Class";
static const WCHAR cszRegBindings[] = L"Bindings";
//static const WCHAR cszRegEnumDriverKey[] = L"Driver";
static const WCHAR cszRegTcpIp[] = L"MSTCP";
// Global TcpIp Reg Location
static const WCHAR cszRegFixedTcpInfoKey[] = L"System\\CurrentControlSet\\Services\\VxD\\MSTCP";
//
// IP Address
//
static const WCHAR cszRegIPAddress[] = L"IPAddress";
static const WCHAR cszRegIPMask[] = L"IPMask";
//
// WINS
//
static const WCHAR cszRegWINS[] = L"NameServer";
//
// Gateway
//
static const WCHAR cszRegDefaultGateway[] = L"DefaultGateway";
//
// DNS
//
static const WCHAR cszRegDomainName[] = L"Domain";
static const WCHAR cszRegNameServer[] = L"NameServer";
static const WCHAR cszRegHostName[] = L"HostName";
static const WCHAR cszRegEnableDNS[] = L"EnableDNS";
static const WCHAR cszRegSuffixSearchList[] = L"SearchList";
static const WCHAR cszNullIP[] = L"0.0.0.0";
static const WCHAR cszAdapterClass[] = L"Net";
static const WCHAR cszProtocolClass[] = L"NetTrans";
static const WCHAR cszNodeType[] = L"NodeType";
static const WCHAR cszScopeID[] = L"ScopeID";
// Not in use by OOBE, but included in case this becomes
// a seperate module. API does *NOT* work with PPPOA.
DWORD WINAPI InetSSetRasConnection ( RASINFO& RasEntry )
{
DWORD nRetVal = ERROR_SUCCESS;
LPFNDLL_RASSETENTRYPROPERTIES rsep = NULL;
LPFNDLL_RASGETENTRYPROPERTIES rgep = NULL;
LPBYTE lpDeviceBuf = NULL;
DWORD dwDeviceBufSize = 0;
HMODULE hRasApi = LoadLibrary (L"RasApi32.dll");
if (!hRasApi) return GetLastError();
if (!(rsep = (LPFNDLL_RASSETENTRYPROPERTIES) GetProcAddress (hRasApi, "RasSetEntryProperties"))) {
nRetVal = GetLastError();
goto end;
}
if (!(rgep = (LPFNDLL_RASGETENTRYPROPERTIES) GetProcAddress (hRasApi, "RasGetEntryProperties"))) {
nRetVal = GetLastError();
goto end;
}
if ( (nRetVal = RasValidateEntryName ( ((!lstrlen(RasEntry.szPhoneBook)) ? NULL : RasEntry.szPhoneBook), RasEntry.szEntryName )) != ERROR_SUCCESS &&
nRetVal != ERROR_ALREADY_EXISTS ) {
nRetVal = ERROR_INVALID_NAME;
goto end;
}
// we place the RASENTRY structure first. the lpDeviceInfo information is only considered for
// ATM.
if ( (nRetVal = rsep (((!lstrlen(RasEntry.szPhoneBook)) ? NULL : RasEntry.szPhoneBook), RasEntry.szEntryName, &RasEntry.RasEntry, sizeof (RasEntry.RasEntry), NULL, 0)) != ERROR_SUCCESS ) {
goto end;
}
// unless the device is ATM, no further action is necessary.
//if ( lstrcmpi (RasEntry.RasEntry.szDeviceType, RASDT_Atm) )
{
nRetVal = ERROR_SUCCESS;
goto end;
}
if ( RasEntry.dwDeviceInfoSize != sizeof (ATMPBCONFIG) ) {
nRetVal = ERROR_INVALID_PARAMETER;
goto end;
}
if ( (nRetVal = rgep (((!lstrlen(RasEntry.szPhoneBook)) ? NULL : RasEntry.szPhoneBook), RasEntry.szEntryName, &(RasEntry.RasEntry), &(RasEntry.RasEntry.dwSize), NULL, &dwDeviceBufSize)) != ERROR_SUCCESS ) {
goto end;
}
if ( !(lpDeviceBuf = (LPBYTE) malloc (dwDeviceBufSize)) ) {
nRetVal = ERROR_NOT_ENOUGH_MEMORY;
goto end;
}
if ( (nRetVal = rgep (((!lstrlen(RasEntry.szPhoneBook)) ? NULL : RasEntry.szPhoneBook), RasEntry.szEntryName, &(RasEntry.RasEntry), &(RasEntry.RasEntry.dwSize), lpDeviceBuf, &dwDeviceBufSize)) != ERROR_SUCCESS ) {
goto end;
}
// ** BUGBUG: WARNING: THIS IS NOT STABLE CODE: THERE IS NO DOCUMENTATION ON THE CORRECT
// ** -------- USE OF THE RASSETENTRYPROPERTIES FOR THE LPDEVICEINFO BUFFER.
memcpy (lpDeviceBuf+66, &RasEntry.lpDeviceInfo, RasEntry.dwDeviceInfoSize); // HACK!
if ( (nRetVal = rsep (NULL, RasEntry.szEntryName,
&RasEntry.RasEntry, sizeof (RasEntry.RasEntry),
lpDeviceBuf, dwDeviceBufSize)) != ERROR_SUCCESS) {
goto end;
}
end:
free (lpDeviceBuf);
FreeLibrary(hRasApi);
return nRetVal;
}
// this function sets a PPPOE connection. Presently, it merely updates
// the device's registry location with the parameters in the INS file.
// in the future, InetSSetPppoeConnection () will have native support.
DWORD WINAPI InetSSetPppoeConnection ( PPPOEINFO& PppoeInfo )
{
// settings:
// ---------------------------------------------------
// Format: "RegKey=RegVal" e.g. "Pvc1=10"
// LPBYTE lpbNdiBuf = PppoeInfo.PppoeModule.lpbRegNdiParamBuf;
LPWSTR pwchSetBuf = (LPWSTR)PppoeInfo.PppoeModule.lpbRegSettingsBuf;
LPWSTR eq = 0;
DWORD cwchValue = 0;
HKEY hkeyAdapterClass = NULL;
// BUGBUG: error checking is ignored. BUG-BUG
DWORD nRetVal = 0;
if ( (nRetVal = InetSGetAdapterKey ( cszAdapterClass, PppoeInfo.TcpIpInfo.szPnPId, INETS_ADAPTER_HARDWAREID, DIREG_DRV, hkeyAdapterClass )) != ERROR_SUCCESS )
{
return nRetVal;
}
while ( *pwchSetBuf )
{
if ( !(eq = wcschr ( pwchSetBuf, L'=' )) )
{
return ERROR_INVALID_PARAMETER;
}
// we also disallow the following: "Vci="
if ( !(*(eq+1)) )
{
return ERROR_INVALID_PARAMETER;
}
// flush out the '=' so that we have two token strings.
// we simply move each string directly into the registry.
*eq = L'\0';
cwchValue = lstrlen(eq + 1) + 1; // account for trailing 0
if ( RegSetValueEx ( hkeyAdapterClass, pwchSetBuf, 0, REG_SZ, (LPBYTE)(eq + 1), cwchValue * sizeof(WCHAR)) != ERROR_SUCCESS )
{
*eq = L'=';
return E_FAIL;
}
// restore the '=' and move to the next pair "name=value"
*eq = L'=';
pwchSetBuf = eq + 1 + cwchValue; // include '='
}
if ( InetSSetLanConnection ( PppoeInfo.TcpIpInfo ) != ERROR_SUCCESS )
{
return E_FAIL;
}
return ERROR_SUCCESS;
}
DWORD WINAPI InetSSetRfc1483Connection ( RFC1483INFO &Rfc1483Info )
{
// settings:
// ---------------------------------------------------
// Format: "RegKey=RegVal" e.g. "Pvc1=10"
// LPBYTE lpbNdiBuf = Rfc1483Info.Rfc1483Module.lpbRegNdiParamBuf;
// BUGBUG: What does lpbSetBuf contain??
LPBYTE lpbSetBuf = Rfc1483Info.Rfc1483Module.lpbRegSettingsBuf;
WCHAR *eq = 0;
DWORD_PTR dwNameSize = 0;
DWORD dwValueSize = 0;
HKEY hkeyAdapterClass = NULL;
// BUGBUG: error checking is ignored. BUG-BUG
DWORD nRetVal = 0;
if ( (nRetVal = InetSGetAdapterKey ( cszAdapterClass, Rfc1483Info.TcpIpInfo.szPnPId, INETS_ADAPTER_HARDWAREID, DIREG_DRV, hkeyAdapterClass )) != ERROR_SUCCESS )
{
return nRetVal;
}
while ( *lpbSetBuf )
{
if ( !(eq = wcschr ( (WCHAR*)lpbSetBuf, L'=' )) )
{
return ERROR_INVALID_PARAMETER;
}
// we also disallow the following: "Vci="
if ( !(*(eq+1)) )
{
return ERROR_INVALID_PARAMETER;
}
// flush out the '=' so that we have two token strings.
// we simply move each string directly into the registry.
*eq = L'\0';
dwNameSize = eq-(WCHAR*)lpbSetBuf;
dwValueSize = BYTES_REQUIRED_BY_SZ(eq+1);
if ( RegSetValueEx ( hkeyAdapterClass, (WCHAR*)lpbSetBuf, 0, REG_SZ, (LPBYTE)eq+1, dwValueSize+1) != ERROR_SUCCESS )
{
*eq = L'=';
return E_FAIL;
}
// restore the '=' and move to the next pair "name=value"
*eq = L'=';
lpbSetBuf += dwNameSize+dwValueSize+2; // for '=' and '\0'
}
if ( InetSSetLanConnection ( Rfc1483Info.TcpIpInfo ) != ERROR_SUCCESS )
{
return E_FAIL;
}
return ERROR_SUCCESS;
}
// -*-*-*-*-*-*-*-*-*-*-*-*-*-*-* InetSGetAdapterKey -*-*-*-*-*-*-*-*-*-*-*-*-*-*-*-* //
//
// Description:
// This function returns a Driver Registry Key for a Device.
//
// Arguments:
// cszDeviceClass - Any Device Class, e.g. "Net", "NetTrans", etc.
// cszDeviceParam - Value that we're using to identify the Device.
// dwEnumType - Can be either INETS_ADAPTER_HARDWAREID or
// INETS_ADAPTER_INSTANCEID.
// dwRequiredKeyType - This corresponds to KeyType in SetupDiOpenDevRegKey
// in SetupAPI.
//
// hkeyDevKey - Registry Key Handle provided by the caller.
//
// Return Values:
// ERROR_SUCCESS - Function returns successfully.
// Other - use GetLastError(). Note: hkeyDevKey = INVALID_HANDLE_VALUE in this case.
//
// Remarks:
// Use this function to browse the Device Manager for Network Devices and Protocol Drivers.
//
DWORD WINAPI InetSGetAdapterKey ( LPCWSTR cszDeviceClass, LPCWSTR cszDeviceParam, DWORD dwEnumType, DWORD dwRequiredKeyType, HKEY &hkeyDevKey ) {
// initialization of parameter
hkeyDevKey = (HKEY) INVALID_HANDLE_VALUE;
DWORD nRetVal = ERROR_SUCCESS;
// We find out the network adapter's TCP/IP Binding first.
HINSTANCE hSetupLib = LoadLibrary (L"SetupApi.Dll");
if (!hSetupLib)
{
return GetLastError();
}
// Get procedures we need from the DLL.
LPFNDLL_SETUPDICLASSGUIDSFROMNAME lpfndll_SetupDiClassGuidsFromName = NULL;
LPFNDLL_SETUPDIGETCLASSDEVS lpfndll_SetupDiGetClassDevs = NULL;
LPFNDLL_SETUPDIENUMDEVICEINFO lpfndll_SetupDiEnumDeviceInfo = NULL;
LPFNDLL_SETUPDIGETDEVICEREGISTRYPROPERTY lpfndll_SetupDiGetDeviceRegistryProperty = NULL;
LPFNDLL_SETUPDIOPENDEVREGKEY lpfndll_SetupDiOpenDevRegKey = NULL;
LPFNDLL_SETUPDIGETDEVICEINSTANCEID lpfndll_SetupDiGetDeviceInstanceId = NULL;
if ( !(lpfndll_SetupDiClassGuidsFromName = (LPFNDLL_SETUPDICLASSGUIDSFROMNAME) GetProcAddress ( hSetupLib, cszSetupDiClassGuidsFromName )) )
{
nRetVal = GetLastError();
FreeLibrary ( hSetupLib );
return nRetVal;
}
if ( !(lpfndll_SetupDiGetClassDevs = (LPFNDLL_SETUPDIGETCLASSDEVS) GetProcAddress ( hSetupLib, cszSetupDiGetClassDevs )) )
{
nRetVal = GetLastError();
FreeLibrary ( hSetupLib );
return nRetVal;
}
if ( !(lpfndll_SetupDiEnumDeviceInfo = (LPFNDLL_SETUPDIENUMDEVICEINFO) GetProcAddress ( hSetupLib, "SetupDiEnumDeviceInfo" )) )
{
nRetVal = GetLastError();
FreeLibrary ( hSetupLib );
return nRetVal;
}
if ( !(lpfndll_SetupDiGetDeviceRegistryProperty = (LPFNDLL_SETUPDIGETDEVICEREGISTRYPROPERTY) GetProcAddress ( hSetupLib, cszSetupDiGetDeviceRegistryProperty )) )
{
nRetVal = GetLastError();
FreeLibrary ( hSetupLib );
return nRetVal;
}
if ( !(lpfndll_SetupDiOpenDevRegKey = (LPFNDLL_SETUPDIOPENDEVREGKEY) GetProcAddress ( hSetupLib, "SetupDiOpenDevRegKey" )) )
{
nRetVal = GetLastError();
FreeLibrary ( hSetupLib );
return nRetVal;
}
if ( !(lpfndll_SetupDiGetDeviceInstanceId = (LPFNDLL_SETUPDIGETDEVICEINSTANCEID) GetProcAddress ( hSetupLib, cszSetupDiGetDeviceInstanceId ) ) )
{
nRetVal = GetLastError();
FreeLibrary ( hSetupLib );
return nRetVal;
}
// fantastic. we have the functions. now, on to business.
// Get Class Guid.
BOOLEAN bRet = FALSE;
DWORD dwArraySize = 0;
LPGUID lpguidArray = NULL;
bRet = lpfndll_SetupDiClassGuidsFromName ( cszDeviceClass, NULL, NULL, &dwArraySize );
// We depend on SetupDiClassGuidsFromName() to provide us with the Guid, and we need to
// allocate space to accomodate the Guid. If this cannot be done, we are crippled!
if ( !dwArraySize )
{
FreeLibrary ( hSetupLib );
return ERROR_INVALID_DATA;
}
if ( !(lpguidArray = (LPGUID) malloc (dwArraySize*sizeof(GUID))) )
{
FreeLibrary ( hSetupLib );
return ERROR_NOT_ENOUGH_MEMORY;
}
if ( !(bRet = (lpfndll_SetupDiClassGuidsFromName ( cszDeviceClass, lpguidArray, dwArraySize, &dwArraySize )) ) )
{
FreeLibrary ( hSetupLib );
free ( lpguidArray );
return ERROR_INVALID_FUNCTION;
}
// we retrieve the list of devices.
HDEVINFO hdevNetDeviceList = lpfndll_SetupDiGetClassDevs ( lpguidArray, NULL, NULL, DIGCF_PRESENT );
if ( !hdevNetDeviceList )
{
FreeLibrary ( hSetupLib );
free ( lpguidArray );
return ERROR_INVALID_FUNCTION;
}
free ( lpguidArray ); // where shall we do this garbage collection ?
// we will now enumerate through the list of Net devices.
SP_DEVINFO_DATA DevInfoStruct;
memset ( &DevInfoStruct, 0, sizeof (DevInfoStruct) );
DevInfoStruct.cbSize = sizeof (DevInfoStruct);
int i = 0;
LPBYTE lpbHardwareIdBuf = NULL;
DWORD dwHardwareIdBufSize = 0;
DWORD dwRequiredSize = 0;
BOOL bFound = FALSE;
const DWORD cdwIncrement = 500; // BUGBUG: What's magic about 500??
if ( !(lpbHardwareIdBuf = (LPBYTE) malloc (500)) )
{
FreeLibrary ( hSetupLib );
return ERROR_NOT_ENOUGH_MEMORY;
}
dwHardwareIdBufSize = 500;
while ( bRet = ( lpfndll_SetupDiEnumDeviceInfo (hdevNetDeviceList, i, &DevInfoStruct )) )
{
// for each Net device, we will compare its hardware ID to the one
// provided in the parameter.
switch ( dwEnumType )
{
case INETS_ADAPTER_HARDWAREID:
while ( !(bRet = lpfndll_SetupDiGetDeviceRegistryProperty ( hdevNetDeviceList, &DevInfoStruct, SPDRP_HARDWAREID, NULL, lpbHardwareIdBuf, dwHardwareIdBufSize, &dwRequiredSize )) && ((nRetVal = GetLastError()) == ERROR_INSUFFICIENT_BUFFER ))
{ // we need to reallocate the buffer size.
if ( !dwRequiredSize ) dwHardwareIdBufSize += cdwIncrement;
else dwHardwareIdBufSize += dwRequiredSize;
if ( !(lpbHardwareIdBuf = (LPBYTE) realloc ( (void*) lpbHardwareIdBuf, dwHardwareIdBufSize )) )
{
// not enough memory!
free (lpbHardwareIdBuf);
FreeLibrary ( hSetupLib );
return ERROR_NOT_ENOUGH_MEMORY;
}
}
break;
case INETS_ADAPTER_INSTANCEID:
while ( !(bRet = lpfndll_SetupDiGetDeviceInstanceId ( hdevNetDeviceList, &DevInfoStruct, (PCWSTR) lpbHardwareIdBuf, dwHardwareIdBufSize, &dwRequiredSize )) && ((nRetVal = GetLastError()) == ERROR_INSUFFICIENT_BUFFER ))
{
// we need to reallocate the buffer size.
if ( !dwRequiredSize ) dwHardwareIdBufSize += cdwIncrement;
else dwHardwareIdBufSize += dwRequiredSize;
if ( !(lpbHardwareIdBuf = (LPBYTE) realloc ( (void*) lpbHardwareIdBuf, dwHardwareIdBufSize )) )
{
// not enough memory!
free (lpbHardwareIdBuf);
FreeLibrary ( hSetupLib );
return ERROR_NOT_ENOUGH_MEMORY;
}
}
break;
default:
free (lpbHardwareIdBuf);
FreeLibrary ( hSetupLib );
return ERROR_INVALID_PARAMETER;
}
if ( bRet )
{
// we should have the hardware ID, at this stage. we compare it with
// the device's plug-and-play ID.
// BUGBUG: Is lpbHardwareIdBuf ANSI or Unicode?
if ( wcsstr( (const WCHAR *)lpbHardwareIdBuf, cszDeviceParam) )
{
// found!
bFound = TRUE;
// we get the device's registry key.
if ( (hkeyDevKey = lpfndll_SetupDiOpenDevRegKey ( hdevNetDeviceList, &DevInfoStruct, DICS_FLAG_GLOBAL, 0, dwRequiredKeyType, KEY_ALL_ACCESS )) == INVALID_HANDLE_VALUE )
{
free (lpbHardwareIdBuf);
FreeLibrary ( hSetupLib );
return ERROR_BADKEY;
}
free (lpbHardwareIdBuf);
FreeLibrary ( hSetupLib );
return ERROR_SUCCESS;
}
}
i++;
}
// the while loop enumerated unsuccessfully.
free (lpbHardwareIdBuf);
FreeLibrary ( hSetupLib );
return ERROR_NOT_FOUND;
}
DWORD WINAPI InetSSetLanConnection ( LANINFO& LANINFO )
{
HKEY hkeyAdapter = NULL;
DWORD nRetVal = 0;
HKEY hkeyGlobalTcp = NULL;
LPBYTE lpbBufPtr = NULL;
DWORD dwValueBufSize = 0;
WCHAR *Token = NULL, *PlaceHolder = NULL;
WCHAR *WINSListPtr = NULL;
// TCP/IP InstanceID ==> Class Key
HKEY hkeyClassTcp = NULL;
HKEY hkeyAdapterBinding = NULL;
__try
{
// PnPId ==> Device Configuration Key
if ( (nRetVal = InetSGetAdapterKey ( cszAdapterClass, LANINFO.szPnPId, INETS_ADAPTER_HARDWAREID, DIREG_DEV, hkeyAdapter)) != ERROR_SUCCESS )
{
__leave;
}
// Open the Bindings subkey to look for TCP/IP Binding.
if ( RegOpenKeyEx ( hkeyAdapter, cszRegBindings, 0, KEY_ALL_ACCESS, &hkeyAdapterBinding ) != ERROR_SUCCESS )
{
nRetVal = GetLastError();
__leave;
}
// Find the TCP/IP binding.
WCHAR szBindingValueName [GEN_MAX_STRING_LENGTH];
DWORD dwBindingValueNameSize = sizeof (szBindingValueName) / sizeof(WCHAR);
int index = 0;
while ( RegEnumValue ( hkeyAdapterBinding, index, szBindingValueName, &dwBindingValueNameSize, 0, 0, 0, 0 ) == ERROR_SUCCESS )
{
if ( !wcsncmp ( szBindingValueName, cszRegTcpIp, sizeof (cszRegTcpIp)-1 ) )
{
// we found a binding!
break;
}
index++;
}
if ( (nRetVal = InetSGetAdapterKey ( cszProtocolClass, szBindingValueName, INETS_ADAPTER_INSTANCEID, DIREG_DRV, hkeyClassTcp )) != ERROR_SUCCESS )
{
nRetVal = GetLastError();
__leave;
}
// Got it. we will now start the update.
//
// IP Address
//
lpbBufPtr = 0;
dwValueBufSize = 0;
if ( LANINFO.TcpIpInfo.EnableIP )
{
lpbBufPtr = (LPBYTE) LANINFO.TcpIpInfo.szIPAddress;
dwValueBufSize = BYTES_REQUIRED_BY_SZ(LANINFO.TcpIpInfo.szIPAddress);
if (RegSetValueEx ( hkeyClassTcp, cszRegIPAddress, 0, REG_SZ, lpbBufPtr, dwValueBufSize) != ERROR_SUCCESS)
{
// close handles also!
nRetVal = E_FAIL;
__leave;
}
lpbBufPtr = (LPBYTE) LANINFO.TcpIpInfo.szIPMask;
dwValueBufSize = BYTES_REQUIRED_BY_SZ(LANINFO.TcpIpInfo.szIPMask);
if (RegSetValueEx ( hkeyClassTcp, cszRegIPMask, 0, REG_SZ, lpbBufPtr, dwValueBufSize) != ERROR_SUCCESS)
{
// close handles also!
nRetVal = E_FAIL;
__leave;
}
}
else
{
lpbBufPtr = (LPBYTE) cszNullIP;
dwValueBufSize = BYTES_REQUIRED_BY_SZ(cszNullIP);
if (RegSetValueEx ( hkeyClassTcp, cszRegIPAddress, 0, REG_SZ, lpbBufPtr, dwValueBufSize) != ERROR_SUCCESS)
{
// close handles also!
nRetVal = E_FAIL;
__leave;
}
if (RegSetValueEx ( hkeyClassTcp, cszRegIPMask, 0, REG_SZ, lpbBufPtr, dwValueBufSize) != ERROR_SUCCESS)
{
// close handles also!
nRetVal = E_FAIL;
__leave;
}
}
//
// WINS
//
lpbBufPtr = 0;
dwValueBufSize = 0;
index = 1;
WCHAR szWINSEntry [GEN_MAX_STRING_LENGTH];
WCHAR szWINSListCopy [GEN_MAX_STRING_LENGTH];
WINSListPtr = szWINSListCopy;
// BUGBUG: Is LANINFO.TcpIpInfo.szWINSList ANSI or Unicode?
lstrcpy (WINSListPtr, LANINFO.TcpIpInfo.szWINSList);
wsprintf (szWINSEntry, L"%s%d", cszRegWINS, index);
PlaceHolder = szWINSEntry+lstrlen(cszRegWINS);
if ( LANINFO.TcpIpInfo.EnableWINS )
{
while ( Token = wcstok ((index > 1) ? NULL : WINSListPtr, L", " )) { // WARNING. wcstok uses static data! Also the whitespace in ", " is necessary!
if (!Token)
{
nRetVal = E_FAIL;
__leave;
}
lpbBufPtr = (LPBYTE) Token;
dwValueBufSize = BYTES_REQUIRED_BY_SZ(Token);
if (RegSetValueEx ( hkeyClassTcp, szWINSEntry, 0, REG_SZ, lpbBufPtr, dwValueBufSize) != ERROR_SUCCESS)
{
// close handles also!
nRetVal = E_FAIL;
__leave;
}
wsprintf (PlaceHolder, L"%d", ++index);
}
if (RegSetValueEx ( hkeyClassTcp, cszNodeType, 0, REG_SZ, (LPBYTE) L"8", sizeof (L"8") ) != ERROR_SUCCESS)
{
nRetVal = E_FAIL;
__leave;
}
}
else
{
// TODO: Remove all instances of NameServerX <== IMPORTANT
index = 0;
WCHAR szEnumValueBuffer[GEN_MAX_STRING_LENGTH];
DWORD dwEnumValueBufferSize;
while ( RegEnumValue ( hkeyClassTcp, index, szEnumValueBuffer, &(dwEnumValueBufferSize=sizeof(szEnumValueBuffer)/sizeof(WCHAR)), 0, 0, 0, 0 ) != ERROR_NO_MORE_ITEMS )
{
if ( !wcsncmp (szEnumValueBuffer, cszRegWINS, sizeof (cszRegWINS)-1) )
{
if ( RegDeleteValue ( hkeyClassTcp, szEnumValueBuffer ) != ERROR_SUCCESS )
{
nRetVal = E_FAIL;
__leave;
}
continue;
}
index++;
}
if (RegSetValueEx ( hkeyClassTcp, cszNodeType, 0, REG_SZ, (LPBYTE) L"1", sizeof (L"1") ) != ERROR_SUCCESS)
{
nRetVal = E_FAIL;
__leave;
}
}
//
// Default Gateway
//
lpbBufPtr = (LPBYTE) LANINFO.TcpIpInfo.szDefaultGatewayList;
dwValueBufSize = BYTES_REQUIRED_BY_SZ (LANINFO.TcpIpInfo.szDefaultGatewayList);
if (RegSetValueEx ( hkeyClassTcp, cszRegDefaultGateway, 0, REG_SZ, lpbBufPtr, dwValueBufSize) != ERROR_SUCCESS)
{
// close handles also!
nRetVal = E_FAIL;
__leave;
}
// Step 4: Update global TCPIP entries (DNS)
if ( RegOpenKeyEx ( HKEY_LOCAL_MACHINE, cszRegFixedTcpInfoKey, 0, KEY_ALL_ACCESS, &hkeyGlobalTcp) != ERROR_SUCCESS )
{
// close keys
nRetVal = E_FAIL;
__leave;
}
if ( LANINFO.TcpIpInfo.EnableDNS )
{
lpbBufPtr = (LPBYTE) LANINFO.TcpIpInfo.szHostName;
dwValueBufSize = BYTES_REQUIRED_BY_SZ (LANINFO.TcpIpInfo.szHostName);
if (RegSetValueEx ( hkeyGlobalTcp, cszRegHostName, 0, REG_SZ, lpbBufPtr, dwValueBufSize) != ERROR_SUCCESS)
{
// close handles also!
nRetVal = E_FAIL;
__leave;
}
lpbBufPtr = (LPBYTE) LANINFO.TcpIpInfo.szDomainName;
dwValueBufSize = BYTES_REQUIRED_BY_SZ (LANINFO.TcpIpInfo.szDomainName);
if (RegSetValueEx ( hkeyGlobalTcp, cszRegDomainName, 0, REG_SZ, lpbBufPtr, dwValueBufSize) != ERROR_SUCCESS)
{
// close handles also!
nRetVal = E_FAIL;
__leave;
}
lpbBufPtr = (LPBYTE) LANINFO.TcpIpInfo.szDNSList;
dwValueBufSize = BYTES_REQUIRED_BY_SZ (LANINFO.TcpIpInfo.szDNSList);
if (RegSetValueEx ( hkeyGlobalTcp, cszRegNameServer, 0, REG_SZ, lpbBufPtr, dwValueBufSize) != ERROR_SUCCESS)
{
// close handles also!
nRetVal = E_FAIL;
__leave;
}
lpbBufPtr = (LPBYTE) LANINFO.TcpIpInfo.szSuffixSearchList;
dwValueBufSize = BYTES_REQUIRED_BY_SZ (LANINFO.TcpIpInfo.szSuffixSearchList);
if (RegSetValueEx ( hkeyGlobalTcp, cszRegSuffixSearchList, 0, REG_SZ, lpbBufPtr, dwValueBufSize) != ERROR_SUCCESS)
{
// close handles also!
nRetVal = E_FAIL;
__leave;
}
if (RegSetValueEx ( hkeyGlobalTcp, cszRegEnableDNS, 0, REG_SZ, (LPBYTE)L"1", sizeof(L"1")) != ERROR_SUCCESS)
{
// close handles also!
nRetVal = E_FAIL;
__leave;
}
}
else
{
if (RegSetValueEx ( hkeyGlobalTcp, cszRegEnableDNS, 0, REG_SZ, (LPBYTE)L"0", sizeof(L"0")) != ERROR_SUCCESS)
{
// close handles also!
nRetVal = E_FAIL;
__leave;
}
}
WCHAR szScopeID[GEN_MAX_STRING_LENGTH];
if ( LANINFO.TcpIpInfo.EnableWINS )
{
if (LANINFO.TcpIpInfo.uiScopeID == (UINT)~0x0) // this line implies that no ScopeID is given.
{
if ( RegSetValueEx ( hkeyGlobalTcp, cszScopeID, 0, REG_SZ, (LPBYTE)L"", sizeof(L"") ) )
{
nRetVal = E_FAIL;
__leave;
}
}
else if (RegSetValueEx(hkeyGlobalTcp,
cszScopeID,
0,
REG_SZ,
(LPBYTE)_itow( LANINFO.TcpIpInfo.uiScopeID, szScopeID, 10 ),
BYTES_REQUIRED_BY_SZ(szScopeID)
) != ERROR_SUCCESS )
{
nRetVal = E_FAIL;
__leave;
}
}
else
{
if ( RegDeleteValue ( hkeyGlobalTcp, cszScopeID ) != ERROR_SUCCESS )
{
nRetVal = E_FAIL;
__leave;
}
}
}
// end.
__finally
{
if ( hkeyAdapter )
{
RegCloseKey ( hkeyAdapter );
}
if ( hkeyGlobalTcp )
{
RegCloseKey ( hkeyGlobalTcp );
}
if ( hkeyAdapterBinding )
{
RegCloseKey ( hkeyAdapterBinding );
}
if ( hkeyClassTcp )
{
RegCloseKey ( hkeyClassTcp );
}
}
return nRetVal;
}
/*
int main() {
CInetSetup inetSetup;
const WCHAR cszINS[] = L"C:\\test.ins";
LANINFO LanInfo;
RASINFO RasInfo;
memset ( &RasInfo, 0, sizeof(RASINFO) );
memset ( &LanInfo, 0, sizeof(LANINFO) );
RasInfo.dwDeviceInfoSize = sizeof(ATMPBCONFIG);
RasInfo.lpDeviceInfo = (LPBYTE) malloc (sizeof (ATMPBCONFIG));
RasInfo.RasEntry.dwSize = sizeof (RASENTRY);
RasInfo.RasEntry.dwfNetProtocols = RASNP_Ip;
RasInfo.RasEntry.dwFramingProtocol = RASFP_Ppp;
lstrcpy ( RasInfo.RasEntry.szDeviceType, RASDT_Modem );
lstrcpy ( RasInfo.RasEntry.szDeviceName, L"Standard 56000 bps V90 Modem" );
lstrcpy ( RasInfo.RasEntry.szLocalPhoneNumber, L"5551212" );
lstrcpy ( RasInfo.szEntryName, L"Test1" );
lstrcpy ( RasInfo.szPhoneBook, L"" );
inetSetup.InetSImportLanConnection ( LanInfo, cszINS );
inetSetup.InetSSetLanConnection ( LanInfo );
inetSetup.InetSSetRasConnection ( RasInfo );
free ( RasInfo.lpDeviceInfo );
return 0;
}
*/
Reference in a new issue Copy permalink