1005 lines
22 KiB
C
1005 lines
22 KiB
C
|
/*++
|
|||
|
|
|||
|
Copyright (c) 1994 Microsoft Corporation
|
|||
|
|
|||
|
Module Name:
|
|||
|
|
|||
|
dhcpcli.c
|
|||
|
|
|||
|
Abstract:
|
|||
|
|
|||
|
This is the API tester for the DHCP client.
|
|||
|
|
|||
|
To build, 'nmake UMTEST=dhcpcli'
|
|||
|
|
|||
|
Author:
|
|||
|
|
|||
|
Manny Weiser (mannyw) 1-Dec-1992
|
|||
|
|
|||
|
Environment:
|
|||
|
|
|||
|
User Mode - Win32
|
|||
|
|
|||
|
Revision History:
|
|||
|
|
|||
|
Madan Appiah (madana) 21-Oct-1993
|
|||
|
|
|||
|
--*/
|
|||
|
|
|||
|
|
|||
|
#include "precomp.h"
|
|||
|
#include "dhcploc.h"
|
|||
|
#include "dhcppro.h"
|
|||
|
#include "dhcpcapi.h"
|
|||
|
#include "lmcons.h"
|
|||
|
#include "conio.h"
|
|||
|
|
|||
|
DWORD // win32 status
|
|||
|
DhcpAcquireParametersByBroadcast( // acquire/renew a lease
|
|||
|
IN LPWSTR AdapterName // adapter to acquire lease on
|
|||
|
);
|
|||
|
|
|||
|
|
|||
|
#define IP_ADDRESS_STRING L"IPAddress"
|
|||
|
#define IP_ADDRESS_STRING_TYPE REG_MULTI_SZ
|
|||
|
|
|||
|
#define SUBNET_MASK_STRING L"SubnetMask"
|
|||
|
#define SUBNET_MASK_STRING_TYPE REG_MULTI_SZ
|
|||
|
|
|||
|
#define USAGE_MESSAGE0 \
|
|||
|
"Usage: dhcpcli { " \
|
|||
|
"renew,release,set,enabledhcp,disabledhcp,timestamp," \
|
|||
|
"leasetest,fbrefresh, reinit } parameters\n"
|
|||
|
#define USAGE_MESSAGE1 "Usage: dhcpcli renew adapter\n"
|
|||
|
#define USAGE_MESSAGE2 "Usage: dhcpcli release adapter\n"
|
|||
|
#define USAGE_MESSAGE3 "Usage: dhcpcli set ip-address subnet-mask adapter index \n"
|
|||
|
#define USAGE_MESSAGE4 "Usage: dhcpcli enabledhcp adapter\n"
|
|||
|
#define USAGE_MESSAGE5 "Usage: dhcpcli disabledhcp adapter ip-address subnet-mask \n"
|
|||
|
#define USAGE_MESSAGE6 "Usage: dhcpcli timestamp Value\n"
|
|||
|
#define USAGE_MESSAGE7 "Usage: dhcpcli leasetest adapter-ip-addres hardware-address\n"
|
|||
|
#define USAGE_MESSAGE8 "Usage: dhcpcli fbrefresh adapter\n"
|
|||
|
#define USAGE_MESSAGE9 "Usage: dhcpcli reinit adapter\n"
|
|||
|
|
|||
|
DWORD
|
|||
|
GetRegistryString(
|
|||
|
HKEY Key,
|
|||
|
LPWSTR ValueStringName,
|
|||
|
LPWSTR *String,
|
|||
|
LPDWORD StringSize
|
|||
|
)
|
|||
|
/*++
|
|||
|
|
|||
|
Routine Description:
|
|||
|
|
|||
|
This function retrieves the specified string value from the
|
|||
|
registry. It allocates local memory for the returned string.
|
|||
|
|
|||
|
Arguments:
|
|||
|
|
|||
|
Key : registry handle to the key where the value is.
|
|||
|
|
|||
|
ValueStringName : name of the value string.
|
|||
|
|
|||
|
String : pointer to a location where the string pointer is returned.
|
|||
|
|
|||
|
StringSize : size of the string data returned. Optional
|
|||
|
|
|||
|
Return Value:
|
|||
|
|
|||
|
The status of the operation.
|
|||
|
|
|||
|
--*/
|
|||
|
{
|
|||
|
DWORD Error;
|
|||
|
DWORD LocalValueType;
|
|||
|
DWORD ValueSize;
|
|||
|
LPWSTR LocalString;
|
|||
|
|
|||
|
DhcpAssert( *String == NULL );
|
|||
|
|
|||
|
//
|
|||
|
// Query DataType and BufferSize.
|
|||
|
//
|
|||
|
|
|||
|
Error = RegQueryValueEx(
|
|||
|
Key,
|
|||
|
ValueStringName,
|
|||
|
0,
|
|||
|
&LocalValueType,
|
|||
|
NULL,
|
|||
|
&ValueSize );
|
|||
|
|
|||
|
if( Error != ERROR_SUCCESS ) {
|
|||
|
return(Error);
|
|||
|
}
|
|||
|
|
|||
|
DhcpAssert( (LocalValueType == REG_SZ) ||
|
|||
|
(LocalValueType == REG_MULTI_SZ) );
|
|||
|
|
|||
|
if( ValueSize == 0 ) {
|
|||
|
*String = NULL;
|
|||
|
*StringSize = 0;
|
|||
|
return( ERROR_SUCCESS );
|
|||
|
}
|
|||
|
|
|||
|
//
|
|||
|
// now allocate memory for string data.
|
|||
|
//
|
|||
|
|
|||
|
LocalString = DhcpAllocateMemory( ValueSize );
|
|||
|
|
|||
|
if(LocalString == NULL) {
|
|||
|
return( ERROR_NOT_ENOUGH_MEMORY );
|
|||
|
}
|
|||
|
|
|||
|
//
|
|||
|
// Now query the string data.
|
|||
|
//
|
|||
|
|
|||
|
Error = RegQueryValueEx(
|
|||
|
Key,
|
|||
|
ValueStringName,
|
|||
|
0,
|
|||
|
&LocalValueType,
|
|||
|
(LPBYTE)(LocalString),
|
|||
|
&ValueSize );
|
|||
|
|
|||
|
if( Error != ERROR_SUCCESS ) {
|
|||
|
DhcpFreeMemory(LocalString);
|
|||
|
return(Error);
|
|||
|
}
|
|||
|
|
|||
|
*String = LocalString;
|
|||
|
|
|||
|
if( StringSize != NULL ) {
|
|||
|
*StringSize = ValueSize;
|
|||
|
}
|
|||
|
|
|||
|
return( ERROR_SUCCESS );
|
|||
|
}
|
|||
|
|
|||
|
|
|||
|
DWORD
|
|||
|
RegSetIpAddress(
|
|||
|
HKEY KeyHandle,
|
|||
|
LPWSTR ValueName,
|
|||
|
DWORD ValueType,
|
|||
|
DHCP_IP_ADDRESS IpAddress
|
|||
|
)
|
|||
|
/*++
|
|||
|
|
|||
|
Routine Description:
|
|||
|
|
|||
|
This function sets IpAddress Value in the registry.
|
|||
|
|
|||
|
Arguments:
|
|||
|
|
|||
|
KeyHandle - handle to the key.
|
|||
|
|
|||
|
ValueName - name of the value field.
|
|||
|
|
|||
|
ValueType - Type of the value field.
|
|||
|
|
|||
|
IpAddress - Ipaddress to be set.
|
|||
|
|
|||
|
Return Value:
|
|||
|
|
|||
|
Registry Error.
|
|||
|
|
|||
|
--*/
|
|||
|
{
|
|||
|
DWORD Error;
|
|||
|
|
|||
|
LPSTR AnsiAddressString;
|
|||
|
WCHAR UnicodeAddressBuf[DOT_IP_ADDR_SIZE];
|
|||
|
LPWSTR UnicodeAddressString;
|
|||
|
|
|||
|
LPWSTR MultiIpAddressString = NULL;
|
|||
|
LPWSTR NewMultiIpAddressString = NULL;
|
|||
|
DWORD MultiIpAddressStringSize;
|
|||
|
DWORD NewMultiIpAddressStringSize;
|
|||
|
DWORD FirstOldIpAddressSize;
|
|||
|
|
|||
|
AnsiAddressString = inet_ntoa( *(struct in_addr *)&IpAddress );
|
|||
|
|
|||
|
UnicodeAddressString = DhcpOemToUnicode(
|
|||
|
AnsiAddressString,
|
|||
|
UnicodeAddressBuf );
|
|||
|
|
|||
|
DhcpAssert( UnicodeAddressString != NULL );
|
|||
|
|
|||
|
if( ValueType == REG_SZ ) {
|
|||
|
Error = RegSetValueEx(
|
|||
|
KeyHandle,
|
|||
|
ValueName,
|
|||
|
0,
|
|||
|
ValueType,
|
|||
|
(LPBYTE)UnicodeAddressString,
|
|||
|
(wcslen(UnicodeAddressString) + 1) * sizeof(WCHAR) );
|
|||
|
|
|||
|
goto Cleanup;
|
|||
|
}
|
|||
|
|
|||
|
DhcpAssert( ValueType == REG_MULTI_SZ );
|
|||
|
|
|||
|
//
|
|||
|
// replace the first IpAddress.
|
|||
|
//
|
|||
|
|
|||
|
//
|
|||
|
// query current multi-IpAddress string.
|
|||
|
//
|
|||
|
|
|||
|
Error = GetRegistryString(
|
|||
|
KeyHandle,
|
|||
|
ValueName,
|
|||
|
&MultiIpAddressString,
|
|||
|
&MultiIpAddressStringSize );
|
|||
|
|
|||
|
if( Error != ERROR_SUCCESS ) {
|
|||
|
goto Cleanup;
|
|||
|
}
|
|||
|
|
|||
|
//
|
|||
|
// allocate new address string.
|
|||
|
//
|
|||
|
|
|||
|
DhcpAssert(MultiIpAddressString != NULL);
|
|||
|
|
|||
|
FirstOldIpAddressSize =
|
|||
|
(wcslen(MultiIpAddressString) + 1) * sizeof(WCHAR);
|
|||
|
|
|||
|
NewMultiIpAddressStringSize =
|
|||
|
MultiIpAddressStringSize - FirstOldIpAddressSize +
|
|||
|
(wcslen(UnicodeAddressString) + 1) * sizeof(WCHAR);
|
|||
|
|
|||
|
NewMultiIpAddressString = DhcpAllocateMemory( NewMultiIpAddressStringSize );
|
|||
|
|
|||
|
if( NewMultiIpAddressString == NULL ) {
|
|||
|
Error = ERROR_NOT_ENOUGH_MEMORY;
|
|||
|
goto Cleanup;
|
|||
|
}
|
|||
|
|
|||
|
//
|
|||
|
// make new address string first.
|
|||
|
//
|
|||
|
|
|||
|
wcscpy( NewMultiIpAddressString, UnicodeAddressString );
|
|||
|
|
|||
|
//
|
|||
|
// copy rest of the old addresses
|
|||
|
//
|
|||
|
|
|||
|
RtlCopyMemory(
|
|||
|
(LPBYTE)NewMultiIpAddressString +
|
|||
|
(wcslen(UnicodeAddressString) + 1) * sizeof(WCHAR),
|
|||
|
(LPBYTE)MultiIpAddressString + FirstOldIpAddressSize,
|
|||
|
MultiIpAddressStringSize - FirstOldIpAddressSize );
|
|||
|
|
|||
|
Error = RegSetValueEx(
|
|||
|
KeyHandle,
|
|||
|
ValueName,
|
|||
|
0,
|
|||
|
ValueType,
|
|||
|
(LPBYTE)NewMultiIpAddressString,
|
|||
|
NewMultiIpAddressStringSize );
|
|||
|
|
|||
|
Cleanup:
|
|||
|
|
|||
|
if( MultiIpAddressString != NULL) {
|
|||
|
DhcpFreeMemory( MultiIpAddressString );
|
|||
|
}
|
|||
|
|
|||
|
if( NewMultiIpAddressString != NULL) {
|
|||
|
DhcpFreeMemory( NewMultiIpAddressString );
|
|||
|
}
|
|||
|
|
|||
|
return( Error );
|
|||
|
}
|
|||
|
|
|||
|
EnableDhcp(
|
|||
|
char **argv
|
|||
|
)
|
|||
|
{
|
|||
|
DWORD Error;
|
|||
|
LPWSTR RegKey = NULL;
|
|||
|
HKEY KeyHandle = NULL;
|
|||
|
WCHAR AdapterNameBuffer[PATHLEN];
|
|||
|
LPWSTR AdapterName;
|
|||
|
|
|||
|
DWORD EnableDhcp;
|
|||
|
|
|||
|
AdapterName = DhcpOemToUnicode( argv[0], AdapterNameBuffer );
|
|||
|
|
|||
|
if( AdapterName == NULL ) {
|
|||
|
Error = ERROR_NOT_ENOUGH_MEMORY;
|
|||
|
goto Cleanup;
|
|||
|
}
|
|||
|
|
|||
|
//
|
|||
|
// make NIC IP parameter key.
|
|||
|
//
|
|||
|
|
|||
|
RegKey = DhcpAllocateMemory(
|
|||
|
(wcslen(DHCP_SERVICES_KEY) +
|
|||
|
wcslen(REGISTRY_CONNECT_STRING) +
|
|||
|
wcslen(AdapterName) +
|
|||
|
wcslen(DHCP_ADAPTER_PARAMETERS_KEY) + 1) *
|
|||
|
sizeof(WCHAR) ); // termination char.
|
|||
|
|
|||
|
if( RegKey == NULL ) {
|
|||
|
Error = ERROR_NOT_ENOUGH_MEMORY;
|
|||
|
goto Cleanup;
|
|||
|
}
|
|||
|
|
|||
|
#if defined(_PNP_POWER_)
|
|||
|
wcscpy( RegKey, DHCP_SERVICES_KEY );
|
|||
|
wcscat( RegKey, DHCP_ADAPTER_PARAMETERS_KEY );
|
|||
|
wcscat( RegKey, REGISTRY_CONNECT_STRING );
|
|||
|
wcscat( RegKey, AdapterName );
|
|||
|
#else
|
|||
|
wcscpy( RegKey, DHCP_SERVICES_KEY );
|
|||
|
wcscat( RegKey, REGISTRY_CONNECT_STRING );
|
|||
|
wcscat( RegKey, AdapterName );
|
|||
|
wcscat( RegKey, DHCP_ADAPTER_PARAMETERS_KEY );
|
|||
|
#endif _PNP_POWER_
|
|||
|
|
|||
|
|
|||
|
//
|
|||
|
// open this key.
|
|||
|
//
|
|||
|
|
|||
|
Error = RegOpenKeyEx(
|
|||
|
HKEY_LOCAL_MACHINE,
|
|||
|
RegKey,
|
|||
|
0, // Reserved field
|
|||
|
DHCP_CLIENT_KEY_ACCESS,
|
|||
|
&KeyHandle
|
|||
|
);
|
|||
|
|
|||
|
if( Error != ERROR_SUCCESS ) {
|
|||
|
goto Cleanup;
|
|||
|
}
|
|||
|
|
|||
|
//
|
|||
|
// Set DHCP switch.
|
|||
|
//
|
|||
|
|
|||
|
EnableDhcp = TRUE;
|
|||
|
Error = RegSetValueEx(
|
|||
|
KeyHandle,
|
|||
|
DHCP_ENABLE_STRING,
|
|||
|
0,
|
|||
|
DHCP_ENABLE_STRING_TYPE,
|
|||
|
(LPBYTE)&EnableDhcp,
|
|||
|
sizeof(EnableDhcp) );
|
|||
|
|
|||
|
if( Error != ERROR_SUCCESS ) {
|
|||
|
goto Cleanup;
|
|||
|
}
|
|||
|
|
|||
|
//
|
|||
|
// set Ipaddress and SubnetMask to zero.
|
|||
|
//
|
|||
|
|
|||
|
Error = RegSetIpAddress(
|
|||
|
KeyHandle,
|
|||
|
IP_ADDRESS_STRING,
|
|||
|
IP_ADDRESS_STRING_TYPE,
|
|||
|
0 );
|
|||
|
|
|||
|
if( Error != ERROR_SUCCESS ) {
|
|||
|
goto Cleanup;
|
|||
|
}
|
|||
|
|
|||
|
Error = RegSetIpAddress(
|
|||
|
KeyHandle,
|
|||
|
DHCP_IP_ADDRESS_STRING,
|
|||
|
DHCP_IP_ADDRESS_STRING_TYPE,
|
|||
|
0 );
|
|||
|
|
|||
|
if( Error != ERROR_SUCCESS ) {
|
|||
|
goto Cleanup;
|
|||
|
}
|
|||
|
|
|||
|
Error = RegSetIpAddress(
|
|||
|
KeyHandle,
|
|||
|
SUBNET_MASK_STRING,
|
|||
|
SUBNET_MASK_STRING_TYPE,
|
|||
|
0xff );
|
|||
|
|
|||
|
if( Error != ERROR_SUCCESS ) {
|
|||
|
goto Cleanup;
|
|||
|
}
|
|||
|
|
|||
|
Error = RegSetIpAddress(
|
|||
|
KeyHandle,
|
|||
|
DHCP_SUBNET_MASK_STRING,
|
|||
|
DHCP_SUBNET_MASK_STRING_TYPE,
|
|||
|
0xff );
|
|||
|
|
|||
|
if( Error != ERROR_SUCCESS ) {
|
|||
|
goto Cleanup;
|
|||
|
}
|
|||
|
|
|||
|
//
|
|||
|
// Call config API to indicate to DHCP service.
|
|||
|
//
|
|||
|
|
|||
|
Error = DhcpNotifyConfigChange(
|
|||
|
L"",
|
|||
|
AdapterName,
|
|||
|
TRUE,
|
|||
|
0, // AdapterIndex
|
|||
|
0, // IpAddress
|
|||
|
0, // SubnetMask
|
|||
|
DhcpEnable );
|
|||
|
|
|||
|
Cleanup:
|
|||
|
|
|||
|
if( RegKey != NULL ) {
|
|||
|
DhcpFreeMemory( RegKey );
|
|||
|
}
|
|||
|
|
|||
|
if( KeyHandle != NULL ) {
|
|||
|
RegCloseKey( KeyHandle );
|
|||
|
}
|
|||
|
|
|||
|
return(Error);
|
|||
|
}
|
|||
|
|
|||
|
DisableDhcp(
|
|||
|
char **argv
|
|||
|
)
|
|||
|
{
|
|||
|
DWORD Error;
|
|||
|
LPWSTR RegKey = NULL;
|
|||
|
HKEY KeyHandle = NULL;
|
|||
|
|
|||
|
DWORD IpAddress;
|
|||
|
DWORD SubnetMask;
|
|||
|
WCHAR AdapterNameBuffer[PATHLEN];
|
|||
|
LPWSTR AdapterName;
|
|||
|
|
|||
|
DWORD EnableDhcp;
|
|||
|
|
|||
|
AdapterName = DhcpOemToUnicode( argv[0], AdapterNameBuffer );
|
|||
|
|
|||
|
if( AdapterName == NULL ) {
|
|||
|
Error = ERROR_NOT_ENOUGH_MEMORY;
|
|||
|
goto Cleanup;
|
|||
|
}
|
|||
|
|
|||
|
IpAddress = inet_addr( argv[1] );
|
|||
|
SubnetMask = inet_addr( argv[2] );
|
|||
|
|
|||
|
//
|
|||
|
// make NIC IP parameter key.
|
|||
|
//
|
|||
|
|
|||
|
RegKey = DhcpAllocateMemory(
|
|||
|
(wcslen(DHCP_SERVICES_KEY) +
|
|||
|
wcslen(REGISTRY_CONNECT_STRING) +
|
|||
|
wcslen(AdapterName) +
|
|||
|
wcslen(DHCP_ADAPTER_PARAMETERS_KEY) + 1) *
|
|||
|
sizeof(WCHAR) ); // termination char.
|
|||
|
|
|||
|
if( RegKey == NULL ) {
|
|||
|
Error = ERROR_NOT_ENOUGH_MEMORY;
|
|||
|
goto Cleanup;
|
|||
|
}
|
|||
|
|
|||
|
#if defined(_PNP_POWER_)
|
|||
|
wcscpy( RegKey, DHCP_SERVICES_KEY );
|
|||
|
wcscat( RegKey, DHCP_ADAPTER_PARAMETERS_KEY );
|
|||
|
wcscat( RegKey, REGISTRY_CONNECT_STRING );
|
|||
|
wcscat( RegKey, AdapterName );
|
|||
|
#else
|
|||
|
wcscpy( RegKey, DHCP_SERVICES_KEY );
|
|||
|
wcscat( RegKey, REGISTRY_CONNECT_STRING );
|
|||
|
wcscat( RegKey, AdapterName );
|
|||
|
wcscat( RegKey, DHCP_ADAPTER_PARAMETERS_KEY );
|
|||
|
#endif _PNP_POWER_
|
|||
|
|
|||
|
|
|||
|
//
|
|||
|
// open this key.
|
|||
|
//
|
|||
|
|
|||
|
Error = RegOpenKeyEx(
|
|||
|
HKEY_LOCAL_MACHINE,
|
|||
|
RegKey,
|
|||
|
0, // Reserved field
|
|||
|
DHCP_CLIENT_KEY_ACCESS,
|
|||
|
&KeyHandle
|
|||
|
);
|
|||
|
|
|||
|
if( Error != ERROR_SUCCESS ) {
|
|||
|
goto Cleanup;
|
|||
|
}
|
|||
|
|
|||
|
//
|
|||
|
// Set DHCP switch.
|
|||
|
//
|
|||
|
|
|||
|
EnableDhcp = FALSE;
|
|||
|
Error = RegSetValueEx(
|
|||
|
KeyHandle,
|
|||
|
DHCP_ENABLE_STRING,
|
|||
|
0,
|
|||
|
DHCP_ENABLE_STRING_TYPE,
|
|||
|
(LPBYTE)&EnableDhcp,
|
|||
|
sizeof(EnableDhcp) );
|
|||
|
|
|||
|
if( Error != ERROR_SUCCESS ) {
|
|||
|
goto Cleanup;
|
|||
|
}
|
|||
|
|
|||
|
//
|
|||
|
// set Ipaddress and SubnetMask to zero.
|
|||
|
//
|
|||
|
|
|||
|
Error = RegSetIpAddress(
|
|||
|
KeyHandle,
|
|||
|
IP_ADDRESS_STRING,
|
|||
|
IP_ADDRESS_STRING_TYPE,
|
|||
|
IpAddress );
|
|||
|
|
|||
|
if( Error != ERROR_SUCCESS ) {
|
|||
|
goto Cleanup;
|
|||
|
}
|
|||
|
|
|||
|
Error = RegSetIpAddress(
|
|||
|
KeyHandle,
|
|||
|
DHCP_IP_ADDRESS_STRING,
|
|||
|
DHCP_IP_ADDRESS_STRING_TYPE,
|
|||
|
0 );
|
|||
|
|
|||
|
if( Error != ERROR_SUCCESS ) {
|
|||
|
goto Cleanup;
|
|||
|
}
|
|||
|
|
|||
|
Error = RegSetIpAddress(
|
|||
|
KeyHandle,
|
|||
|
SUBNET_MASK_STRING,
|
|||
|
SUBNET_MASK_STRING_TYPE,
|
|||
|
SubnetMask );
|
|||
|
|
|||
|
if( Error != ERROR_SUCCESS ) {
|
|||
|
goto Cleanup;
|
|||
|
}
|
|||
|
|
|||
|
Error = RegSetIpAddress(
|
|||
|
KeyHandle,
|
|||
|
DHCP_SUBNET_MASK_STRING,
|
|||
|
DHCP_SUBNET_MASK_STRING_TYPE,
|
|||
|
0xff );
|
|||
|
|
|||
|
if( Error != ERROR_SUCCESS ) {
|
|||
|
goto Cleanup;
|
|||
|
}
|
|||
|
|
|||
|
//
|
|||
|
// Call config API to indicate to DHCP service.
|
|||
|
//
|
|||
|
|
|||
|
Error = DhcpNotifyConfigChange(
|
|||
|
L"",
|
|||
|
AdapterName,
|
|||
|
TRUE,
|
|||
|
0, // AdapterIndex
|
|||
|
IpAddress,
|
|||
|
SubnetMask,
|
|||
|
DhcpDisable );
|
|||
|
|
|||
|
Cleanup:
|
|||
|
|
|||
|
if( RegKey != NULL ) {
|
|||
|
DhcpFreeMemory( RegKey );
|
|||
|
}
|
|||
|
|
|||
|
if( KeyHandle != NULL ) {
|
|||
|
RegCloseKey( KeyHandle );
|
|||
|
}
|
|||
|
|
|||
|
return(Error);
|
|||
|
}
|
|||
|
|
|||
|
SetIpAddress(
|
|||
|
char **argv
|
|||
|
)
|
|||
|
{
|
|||
|
DWORD Error;
|
|||
|
DWORD IpAddress;
|
|||
|
DWORD SubnetMask;
|
|||
|
WCHAR AdapterNameBuffer[PATHLEN];
|
|||
|
LPWSTR AdapterName;
|
|||
|
DWORD AdapterIndex;
|
|||
|
|
|||
|
IpAddress = inet_addr( argv[0] );
|
|||
|
SubnetMask = inet_addr( argv[1] );
|
|||
|
|
|||
|
AdapterName = DhcpOemToUnicode( argv[2], AdapterNameBuffer );
|
|||
|
|
|||
|
if( AdapterName == NULL ) {
|
|||
|
return( ERROR_NOT_ENOUGH_MEMORY );
|
|||
|
}
|
|||
|
|
|||
|
AdapterIndex = atoi( argv[3] );
|
|||
|
|
|||
|
Error = DhcpNotifyConfigChange(
|
|||
|
L"",
|
|||
|
AdapterName,
|
|||
|
TRUE,
|
|||
|
AdapterIndex,
|
|||
|
IpAddress,
|
|||
|
SubnetMask,
|
|||
|
IgnoreFlag );
|
|||
|
|
|||
|
return( Error );
|
|||
|
}
|
|||
|
|
|||
|
VOID
|
|||
|
PrintClientLeaseInfo(
|
|||
|
LPDHCP_LEASE_INFO LeaseInfo
|
|||
|
)
|
|||
|
{
|
|||
|
DWORD i;
|
|||
|
DWORD DataLength;
|
|||
|
LPBYTE Data;
|
|||
|
|
|||
|
if( LeaseInfo == NULL ) {
|
|||
|
printf( "LeaseInfo is NULL .\n" );
|
|||
|
}
|
|||
|
|
|||
|
printf("ClientInfo :\n");
|
|||
|
|
|||
|
printf("\tClient Hardware Address = ");
|
|||
|
|
|||
|
DataLength = LeaseInfo->ClientUID.ClientUIDLength;
|
|||
|
Data = LeaseInfo->ClientUID.ClientUID;
|
|||
|
for( i = 0; i < DataLength; i++ ) {
|
|||
|
printf("%.2lx", (DWORD)Data[i]);
|
|||
|
}
|
|||
|
printf( "\n" );
|
|||
|
|
|||
|
printf("\tIP Address = %s.\n",
|
|||
|
DhcpIpAddressToDottedString(LeaseInfo->IpAddress));
|
|||
|
printf("\tSubnetMask = %s.\n",
|
|||
|
DhcpIpAddressToDottedString(LeaseInfo->SubnetMask));
|
|||
|
printf("\tDhcpServerAddress = %s.\n",
|
|||
|
DhcpIpAddressToDottedString(LeaseInfo->DhcpServerAddress));
|
|||
|
printf("\tLease = %ld secs.\n", LeaseInfo->Lease );
|
|||
|
printf("\tLease Obtained at %s", ctime(&LeaseInfo->LeaseObtained) );
|
|||
|
printf("\tT1 time is %s", ctime(&LeaseInfo->T1Time) );
|
|||
|
printf("\tT2 time is %s", ctime(&LeaseInfo->T2Time) );
|
|||
|
printf("\tLease Expires at %s", ctime(&LeaseInfo->LeaseExpires) );
|
|||
|
|
|||
|
return;
|
|||
|
}
|
|||
|
|
|||
|
DWORD
|
|||
|
DhcpTestLease(
|
|||
|
LPSTR AdapterIpAddressString,
|
|||
|
LPSTR HardwareAddressString
|
|||
|
)
|
|||
|
{
|
|||
|
#define MAX_ADDRESS_LENGTH 64 // 64 bytes
|
|||
|
|
|||
|
DWORD Error;
|
|||
|
DWORD AdapterIpAddress;
|
|||
|
DHCP_CLIENT_UID ClientUID;
|
|||
|
BYTE Address[MAX_ADDRESS_LENGTH];
|
|||
|
DWORD i;
|
|||
|
LPDHCP_LEASE_INFO LeaseInfo = NULL;
|
|||
|
LPDHCP_OPTION_INFO OptionInfo;
|
|||
|
DHCP_OPTION_LIST OptionList;
|
|||
|
CHAR ch;
|
|||
|
|
|||
|
AdapterIpAddress = ntohl( inet_addr( AdapterIpAddressString ) );
|
|||
|
ClientUID.ClientUIDLength = strlen(HardwareAddressString);
|
|||
|
if( ClientUID.ClientUIDLength % 2 != 0 ) {
|
|||
|
|
|||
|
//
|
|||
|
// address must be even length.
|
|||
|
//
|
|||
|
|
|||
|
printf("DhcpTestLease: Hardware address must be even length.\n");
|
|||
|
return( ERROR_INVALID_PARAMETER );
|
|||
|
}
|
|||
|
|
|||
|
ClientUID.ClientUIDLength /= 2;
|
|||
|
|
|||
|
if( ClientUID.ClientUIDLength > MAX_ADDRESS_LENGTH ) {
|
|||
|
|
|||
|
printf("DhcpTestLease: hardware address is too long.\n");
|
|||
|
return( ERROR_INVALID_PARAMETER );
|
|||
|
}
|
|||
|
|
|||
|
i = DhcpStringToHwAddress( (LPSTR)Address, HardwareAddressString );
|
|||
|
DhcpAssert( i == ClientUID.ClientUIDLength );
|
|||
|
|
|||
|
ClientUID.ClientUID = Address;
|
|||
|
|
|||
|
OptionList.NumOptions = 0;
|
|||
|
OptionList.OptionIDArray = NULL;
|
|||
|
|
|||
|
Error = DhcpLeaseIpAddress(
|
|||
|
AdapterIpAddress, // any subnet.
|
|||
|
&ClientUID,
|
|||
|
0, // desired address.
|
|||
|
&OptionList, // option list - not supported
|
|||
|
&LeaseInfo, // lease info returned.
|
|||
|
&OptionInfo); // option data returned.
|
|||
|
|
|||
|
if( Error != ERROR_SUCCESS ) {
|
|||
|
printf("DhcpLeaseIpAddress failed, %ld.\n", Error);
|
|||
|
goto Cleanup;
|
|||
|
}
|
|||
|
|
|||
|
PrintClientLeaseInfo( LeaseInfo );
|
|||
|
|
|||
|
printf("Renew the lease ? (Y/N) ");
|
|||
|
|
|||
|
do {
|
|||
|
ch = (CHAR)getchar();
|
|||
|
} while ( (ch != 'Y') && (ch != 'y') && (ch != 'N') && (ch != 'n') );
|
|||
|
|
|||
|
printf("%c\n", ch);
|
|||
|
|
|||
|
if( (ch == 'N') || (ch == 'n') ) {
|
|||
|
|
|||
|
printf( "NOTE: YOU HAVE CONSUMED AN IP ADDRESS FROM "
|
|||
|
"THE DHCP SERVER. \n");
|
|||
|
|
|||
|
Error = ERROR_SUCCESS;
|
|||
|
goto Cleanup;
|
|||
|
}
|
|||
|
|
|||
|
Error = DhcpRenewIpAddressLease(
|
|||
|
AdapterIpAddress,
|
|||
|
LeaseInfo,
|
|||
|
&OptionList, // option list - not supported
|
|||
|
&OptionInfo); // option data returned.
|
|||
|
|
|||
|
if( Error != ERROR_SUCCESS ) {
|
|||
|
printf("DhcpRenewIpAddressLease failed, %ld.\n", Error);
|
|||
|
goto Cleanup;
|
|||
|
}
|
|||
|
|
|||
|
PrintClientLeaseInfo( LeaseInfo );
|
|||
|
|
|||
|
printf("Release the lease ? (Y/N) ");
|
|||
|
|
|||
|
do {
|
|||
|
ch = (CHAR)getchar();
|
|||
|
} while ( (ch != 'Y') && (ch != 'y') && (ch != 'N') && (ch != 'n') );
|
|||
|
|
|||
|
printf("%c\n", ch);
|
|||
|
|
|||
|
if( (ch == 'N') || (ch == 'n') ) {
|
|||
|
|
|||
|
printf( "NOTE: YOU HAVE CONSUMED AN IP ADDRESS FROM "
|
|||
|
"THE DHCP SERVER. \n");
|
|||
|
|
|||
|
Error = ERROR_SUCCESS;
|
|||
|
goto Cleanup;
|
|||
|
}
|
|||
|
|
|||
|
Error = DhcpReleaseIpAddressLease(
|
|||
|
AdapterIpAddress,
|
|||
|
LeaseInfo );
|
|||
|
|
|||
|
if( Error != ERROR_SUCCESS ) {
|
|||
|
printf("DhcpReleaseIpAddressLease failed, %ld.\n", Error);
|
|||
|
goto Cleanup;
|
|||
|
}
|
|||
|
|
|||
|
Error = ERROR_SUCCESS;
|
|||
|
|
|||
|
Cleanup:
|
|||
|
|
|||
|
if( LeaseInfo != NULL ) {
|
|||
|
DhcpFreeMemory( LeaseInfo );
|
|||
|
}
|
|||
|
|
|||
|
return( Error );
|
|||
|
}
|
|||
|
|
|||
|
|
|||
|
DWORD __cdecl
|
|||
|
main(
|
|||
|
int argc,
|
|||
|
char **argv
|
|||
|
)
|
|||
|
{
|
|||
|
DWORD error;
|
|||
|
WCHAR AdapterNameBuffer[PATHLEN];
|
|||
|
LPWSTR AdapterName;
|
|||
|
|
|||
|
//
|
|||
|
// seed random generator.
|
|||
|
//
|
|||
|
|
|||
|
srand( GetCurrentTime() );
|
|||
|
|
|||
|
if ( argc < 2) {
|
|||
|
printf( USAGE_MESSAGE0 );
|
|||
|
return( 1 );
|
|||
|
}
|
|||
|
|
|||
|
if ( _stricmp( argv[1], "renew" ) == 0 ) {
|
|||
|
|
|||
|
if ( argc < 3) {
|
|||
|
printf( USAGE_MESSAGE1 );
|
|||
|
return( 1 );
|
|||
|
}
|
|||
|
|
|||
|
AdapterName = DhcpOemToUnicode( argv[2], AdapterNameBuffer );
|
|||
|
|
|||
|
if( AdapterName == NULL ) {
|
|||
|
error = ERROR_NOT_ENOUGH_MEMORY;
|
|||
|
}
|
|||
|
else {
|
|||
|
error = DhcpAcquireParameters( AdapterName );
|
|||
|
}
|
|||
|
|
|||
|
} else if ( _stricmp( argv[1], "fbrefresh" ) == 0 ) {
|
|||
|
|
|||
|
if ( argc < 3) {
|
|||
|
printf( USAGE_MESSAGE8 );
|
|||
|
return( 1 );
|
|||
|
}
|
|||
|
|
|||
|
AdapterName = DhcpOemToUnicode( argv[2], AdapterNameBuffer );
|
|||
|
|
|||
|
if( AdapterName == NULL ) {
|
|||
|
error = ERROR_NOT_ENOUGH_MEMORY;
|
|||
|
}
|
|||
|
else {
|
|||
|
error = DhcpFallbackRefreshParams( AdapterName );
|
|||
|
}
|
|||
|
|
|||
|
} else if ( _stricmp( argv[1], "release" ) == 0 ) {
|
|||
|
|
|||
|
if ( argc < 3) {
|
|||
|
printf( USAGE_MESSAGE2 );
|
|||
|
return( 1 );
|
|||
|
}
|
|||
|
|
|||
|
AdapterName = DhcpOemToUnicode( argv[2], AdapterNameBuffer );
|
|||
|
|
|||
|
if( AdapterName == NULL ) {
|
|||
|
error = ERROR_NOT_ENOUGH_MEMORY;
|
|||
|
}
|
|||
|
else {
|
|||
|
error = DhcpReleaseParameters( AdapterName );
|
|||
|
}
|
|||
|
|
|||
|
} else if ( _stricmp( argv[1], "set" ) == 0 ) {
|
|||
|
|
|||
|
if ( argc < 6) {
|
|||
|
printf( USAGE_MESSAGE3 );
|
|||
|
return( 1 );
|
|||
|
}
|
|||
|
|
|||
|
error = SetIpAddress( &argv[2] );
|
|||
|
|
|||
|
|
|||
|
} else if ( _stricmp( argv[1], "enabledhcp" ) == 0 ) {
|
|||
|
|
|||
|
if ( argc < 3) {
|
|||
|
printf( USAGE_MESSAGE4 );
|
|||
|
return( 1 );
|
|||
|
}
|
|||
|
|
|||
|
error = EnableDhcp( &argv[2] );
|
|||
|
|
|||
|
|
|||
|
} else if ( _stricmp( argv[1], "disabledhcp" ) == 0 ) {
|
|||
|
|
|||
|
if ( argc < 5) {
|
|||
|
printf( USAGE_MESSAGE5 );
|
|||
|
return( 1 );
|
|||
|
}
|
|||
|
|
|||
|
error = DisableDhcp( &argv[2] );
|
|||
|
|
|||
|
|
|||
|
} else if ( _stricmp( argv[1], "timestamp" ) == 0 ) {
|
|||
|
|
|||
|
time_t Time;
|
|||
|
char *endptr;
|
|||
|
|
|||
|
if ( argc < 3) {
|
|||
|
printf( USAGE_MESSAGE6 );
|
|||
|
return( 1 );
|
|||
|
}
|
|||
|
|
|||
|
Time = strtol( argv[2], &endptr, 0 );
|
|||
|
|
|||
|
if( (endptr != NULL) && (*endptr != '\0') ) {
|
|||
|
printf( "Invalid Input, %s\n", endptr );
|
|||
|
return( 1 );
|
|||
|
}
|
|||
|
|
|||
|
printf("TimeStamp = %s\n", ctime(&Time) );
|
|||
|
error = ERROR_SUCCESS;
|
|||
|
|
|||
|
|
|||
|
} else if ( _stricmp( argv[1], "leasetest" ) == 0 ) {
|
|||
|
|
|||
|
if ( argc < 3) {
|
|||
|
printf( USAGE_MESSAGE7 );
|
|||
|
return( 1 );
|
|||
|
}
|
|||
|
|
|||
|
error = DhcpTestLease( argv[2], argv[3] );
|
|||
|
|
|||
|
} else if ( _stricmp( argv[1], "reinit" ) == 0 ) {
|
|||
|
|
|||
|
if ( argc < 3) {
|
|||
|
printf( USAGE_MESSAGE9 );
|
|||
|
return( 1 );
|
|||
|
}
|
|||
|
|
|||
|
AdapterName = DhcpOemToUnicode( argv[2], AdapterNameBuffer );
|
|||
|
|
|||
|
if( AdapterName == NULL ) {
|
|||
|
error = ERROR_NOT_ENOUGH_MEMORY;
|
|||
|
}
|
|||
|
else {
|
|||
|
error = DhcpAcquireParametersByBroadcast( AdapterName );
|
|||
|
}
|
|||
|
|
|||
|
} else {
|
|||
|
printf("Unknown function %s\n", argv[1] );
|
|||
|
error = ERROR_INVALID_FUNCTION;
|
|||
|
}
|
|||
|
|
|||
|
if( error != ERROR_SUCCESS ) {
|
|||
|
printf("Result = %d\n", error );
|
|||
|
}
|
|||
|
else {
|
|||
|
printf("Command completed successfully.\n");
|
|||
|
}
|
|||
|
|
|||
|
return(0);
|
|||
|
}
|
|||
|
|
|||
|
#if DBG
|
|||
|
|
|||
|
VOID
|
|||
|
DhcpPrintRoutine(
|
|||
|
IN DWORD DebugFlag,
|
|||
|
IN LPSTR Format,
|
|||
|
...
|
|||
|
)
|
|||
|
|
|||
|
{
|
|||
|
|
|||
|
#define MAX_PRINTF_LEN 1024 // Arbitrary.
|
|||
|
|
|||
|
va_list arglist;
|
|||
|
char OutputBuffer[MAX_PRINTF_LEN];
|
|||
|
ULONG length = 0;
|
|||
|
|
|||
|
//
|
|||
|
// Put a the information requested by the caller onto the line
|
|||
|
//
|
|||
|
|
|||
|
va_start(arglist, Format);
|
|||
|
length += (ULONG) vsprintf(&OutputBuffer[length], Format, arglist);
|
|||
|
va_end(arglist);
|
|||
|
|
|||
|
DhcpAssert(length <= MAX_PRINTF_LEN);
|
|||
|
|
|||
|
//
|
|||
|
// Output to the debug terminal,
|
|||
|
//
|
|||
|
|
|||
|
printf( "%s", OutputBuffer);
|
|||
|
}
|
|||
|
|
|||
|
#endif // DBG
|