1790 lines
59 KiB
C
1790 lines
59 KiB
C
//================================================================================
|
|
// Copyright (C) 1997 Microsoft Corporation
|
|
// Author: RameshV
|
|
// Description: simple registry utilities
|
|
//================================================================================
|
|
|
|
#include <mmregpch.h>
|
|
|
|
//BeginExport(typedef)
|
|
typedef struct _REG_HANDLE {
|
|
HKEY Key;
|
|
HKEY SubKey;
|
|
LPWSTR SubKeyLocation;
|
|
} REG_HANDLE, *PREG_HANDLE, *LPREG_HANDLE;
|
|
//EndExport(typedef)
|
|
|
|
//BeginExport(constants)
|
|
#define REG_THIS_SERVER L"Software\\Microsoft\\DHCPServer\\Configuration"
|
|
#define REG_THIS_SERVER_DS L"Software\\Microsoft\\DHCPServer\\Config_DS"
|
|
#define REG_THIS_SERVER_DS_VALUE L"Config_DS"
|
|
#define REG_THIS_SERVER_DS_PARENT L"Software\\Microsoft\\DHCPServer"
|
|
|
|
#define REG_SERVER_GLOBAL_OPTIONS L"GlobalOptionValues"
|
|
#define REG_SERVER_OPTDEFS L"OptionInfo"
|
|
#define REG_SERVER_SUBNETS L"Subnets"
|
|
#define REG_SERVER_SSCOPES L"SuperScope"
|
|
#define REG_SERVER_CLASSDEFS L"ClassDefs"
|
|
#define REG_SERVER_MSCOPES L"MulticastScopes"
|
|
|
|
#define REG_SUBNET_SERVERS L"DHCPServers"
|
|
#define REG_SUBNET_RANGES L"IpRanges"
|
|
#define REG_SUBNET_RESERVATIONS L"ReservedIps"
|
|
#define REG_SUBNET_OPTIONS L"SubnetOptions"
|
|
|
|
#define REG_SUBNET_EXCL L"ExcludedIpRanges"
|
|
#define REG_SUBNET_ADDRESS L"SubnetAddress"
|
|
#define REG_SUBNET_NAME L"SubnetName"
|
|
#define REG_SUBNET_COMMENT L"SubnetComment"
|
|
#define REG_SUBNET_MASK L"SubnetMask"
|
|
#define REG_SUBNET_STATE L"SubnetState"
|
|
#define REG_SUBNET_SWITCHED_FLAG L"SwitchedNetworkFlag"
|
|
|
|
#define REG_MSCOPE_NAME L"MScopeName"
|
|
#define REG_MSCOPE_COMMENT L"MScopeComment"
|
|
//
|
|
// Win2K Beta2 and Beta3 went out with scope id param value MScopeId.
|
|
// Since their meaning is being changed, to avoid any costly upgrade
|
|
// code, this value is being changed to MScopeIdValue: to automatically
|
|
// chose a good scope id, TTL values. Note that the default value of
|
|
// zero is treated specially for this scope id param. It implies that
|
|
// this was probably a pre-RC1 upgrade. In this case, the Scope ID
|
|
// defaults to first value in the range.
|
|
//
|
|
#define REG_MSCOPE_SCOPEID L"MScopeIdValue"
|
|
#define REG_MSCOPE_STATE L"MScopeState"
|
|
#define REG_MSCOPE_ADDR_POLICY L"MScopeAddressPolicy"
|
|
#define REG_MSCOPE_TTL L"MScopeTTL"
|
|
#define REG_MSCOPE_LANG_TAG L"MScopeLangTag"
|
|
#define REG_MSCOPE_EXPIRY_TIME L"MScopeExpiryTime"
|
|
|
|
#define REG_SUB_SERVER_NAME L"ServerHostName"
|
|
#define REG_SUB_SERVER_COMMENT L"ServerComment"
|
|
#define REG_SUB_SERVER_ADDRESS L"ServerAddress"
|
|
#define REG_SUB_SERVER_ROLE L"Role"
|
|
|
|
#define REG_RANGE_NAME L"RangeName"
|
|
#define REG_RANGE_COMMENT L"RangeComment"
|
|
#define REG_RANGE_START_ADDRESS L"StartAddress"
|
|
#define REG_RANGE_END_ADDRESS L"EndAddress"
|
|
#define REG_RANGE_INUSE_CLUSTERS L"InUseClusters"
|
|
#define REG_RANGE_USED_CLUSTERS L"UsedClusters"
|
|
#define REG_RANGE_BITS_PREFIX L"Bits "
|
|
#define REG_RANGE_BITS_PREFIX_WCHAR_COUNT (5)
|
|
#define REG_RANGE_FLAGS L"RangeFlags"
|
|
#define REG_RANGE_ALLOC L"RangeBootpAllocated"
|
|
#define REG_RANGE_MAX_ALLOC L"RangeBootpMaxAllowed"
|
|
|
|
#define REG_OPTION_NAME L"OptionName"
|
|
#define REG_OPTION_COMMENT L"OptionComment"
|
|
#define REG_OPTION_TYPE L"OptionType"
|
|
#define REG_OPTION_VALUE L"OptionValue"
|
|
#define REG_OPTION_ID L"OptionId"
|
|
#define REG_OPTION_CLASSNAME L"OptionClassName"
|
|
#define REG_OPTION_VENDORNAME L"OptionVendorName"
|
|
|
|
#define REG_CLASSDEF_NAME L"ClassName"
|
|
#define REG_CLASSDEF_COMMENT L"ClassComment"
|
|
#define REG_CLASSDEF_TYPE L"ClassType"
|
|
#define REG_CLASSDEF_VALUE L"ClassValue"
|
|
|
|
#define REG_RESERVATION_ADDRESS L"IpAddress"
|
|
#define REG_RESERVATION_UID L"ClientUID"
|
|
#define REG_RESERVATION_TYPE L"AllowedClientTypes"
|
|
#define REG_RESERVATION_NAME L"ReservationName"
|
|
#define REG_RESERVATION_COMMENT L"ReservationComment"
|
|
|
|
#define REG_FLAGS L"Flags"
|
|
|
|
#define REG_ACCESS KEY_ALL_ACCESS
|
|
#define REG_DEFAULT_SUBNET_STATE 0
|
|
#define REG_DEFAULT_SUBNET_MASK 0xFFFFFFFF
|
|
#define REG_DEFAULT_SWITCHED_FLAG FALSE
|
|
|
|
#define REG_CLASS L"DhcpClass"
|
|
|
|
#define DHCP_LAST_DOWNLOAD_TIME_VALUE L"LastDownloadTime"
|
|
#define DHCP_LAST_DOWNLOAD_TIME_TYPE REG_BINARY
|
|
|
|
#define DEF_RANGE_ALLOC 0
|
|
#define DEF_RANGE_MAX_ALLOC (~(ULONG)0)
|
|
|
|
//EndExport(constants)
|
|
|
|
const DWORD ZeroReserved = 0;
|
|
const LPVOID NullReserved = 0;
|
|
#define MAX_KEY_SIZE 512
|
|
#define DEF_RANGE_FLAG_VAL (MM_FLAG_ALLOW_DHCP)
|
|
#define DEF_RESERVATION_TYPE (MM_FLAG_ALLOW_DHCP|MM_FLAG_ALLOW_BOOTP)
|
|
|
|
//BeginExport(comment)
|
|
//================================================================================
|
|
// The basic open/traverse/close functions are here
|
|
//================================================================================
|
|
//EndExport(comment)
|
|
HKEY CurrentServerKey = NULL;
|
|
|
|
//BeginExport(function)
|
|
DWORD
|
|
DhcpRegSetCurrentServer(
|
|
IN OUT PREG_HANDLE Hdl
|
|
) //EndExport(function)
|
|
{
|
|
CurrentServerKey = Hdl? Hdl->Key : NULL;
|
|
return ERROR_SUCCESS;
|
|
}
|
|
|
|
//BeginExport(function)
|
|
DWORD
|
|
DhcpRegGetThisServer(
|
|
IN OUT PREG_HANDLE Hdl
|
|
) //EndExport(function)
|
|
{
|
|
DWORD Disposition;
|
|
|
|
if( NULL != CurrentServerKey ) {
|
|
return RegOpenKeyEx( // duplicate key
|
|
CurrentServerKey,
|
|
NULL,
|
|
ZeroReserved,
|
|
REG_ACCESS,
|
|
&Hdl->Key
|
|
);
|
|
}
|
|
return RegCreateKeyEx(
|
|
HKEY_LOCAL_MACHINE,
|
|
REG_THIS_SERVER,
|
|
ZeroReserved,
|
|
REG_CLASS,
|
|
REG_OPTION_NON_VOLATILE,
|
|
REG_ACCESS,
|
|
NULL,
|
|
&Hdl->Key,
|
|
&Disposition
|
|
);
|
|
}
|
|
|
|
//BeginExport(function)
|
|
DWORD
|
|
DhcpRegGetNextHdl(
|
|
IN PREG_HANDLE Hdl,
|
|
IN LPWSTR NextLoc,
|
|
OUT PREG_HANDLE OutHdl
|
|
) //EndExport(function)
|
|
{
|
|
DWORD Disposition;
|
|
DWORD Error;
|
|
|
|
Error = RegCreateKeyEx(
|
|
Hdl->Key,
|
|
NextLoc,
|
|
ZeroReserved,
|
|
REG_CLASS,
|
|
REG_OPTION_NON_VOLATILE,
|
|
REG_ACCESS,
|
|
NULL,
|
|
&OutHdl->Key,
|
|
&Disposition
|
|
);
|
|
return Error;
|
|
}
|
|
|
|
//BeginExport(function)
|
|
DWORD
|
|
DhcpRegCloseHdl(
|
|
IN OUT PREG_HANDLE Hdl
|
|
) //EndExport(function)
|
|
{
|
|
DWORD Error;
|
|
|
|
Error = RegCloseKey(Hdl->Key);
|
|
Hdl->Key = NULL;
|
|
return Error;
|
|
}
|
|
|
|
//BeginExport(comment)
|
|
//================================================================================
|
|
// MISC utilities for registry manipulation
|
|
//================================================================================
|
|
//EndExport(comment)
|
|
//BeginExport(function)
|
|
DWORD
|
|
DhcpRegFillSubKeys(
|
|
IN PREG_HANDLE Hdl,
|
|
IN OUT PARRAY Array // fill in a list of key names
|
|
) //EndExport(function)
|
|
{
|
|
DWORD Error;
|
|
DWORD Index;
|
|
DWORD Size;
|
|
WCHAR KeyName[MAX_KEY_SIZE];
|
|
LPWSTR ThisKeyName;
|
|
|
|
Index = 0;
|
|
do {
|
|
Size = sizeof(KeyName)/sizeof(KeyName[0]);
|
|
Error = RegEnumKeyEx(
|
|
Hdl->Key,
|
|
Index++,
|
|
KeyName,
|
|
&Size,
|
|
NullReserved,
|
|
NULL,
|
|
NULL,
|
|
NULL
|
|
);
|
|
if( ERROR_NO_MORE_ITEMS == Error ) {
|
|
Error = ERROR_SUCCESS;
|
|
break;
|
|
}
|
|
if( ERROR_SUCCESS != Error ) break;
|
|
Require(0 != Size);
|
|
Size += 1; // for the terminating L'\0' char
|
|
Size *= sizeof(WCHAR); // looks like the units are WCHAR!!
|
|
|
|
ThisKeyName = MemAlloc(Size);
|
|
if( NULL == ThisKeyName ) return ERROR_NOT_ENOUGH_MEMORY;
|
|
|
|
wcscpy(ThisKeyName, KeyName);
|
|
|
|
Error = MemArrayAddElement(Array, (LPVOID)ThisKeyName);
|
|
if( ERROR_SUCCESS != Error ) {
|
|
MemFree(ThisKeyName);
|
|
}
|
|
} while( ERROR_SUCCESS == Error );
|
|
|
|
Require(ERROR_MORE_DATA != Error);
|
|
return Index? ERROR_SUCCESS : Error; // if we added something, dont bother about reporting error
|
|
}
|
|
|
|
//BeginExport(function)
|
|
LPVOID // DWORD or LPWSTR or LPBYTE
|
|
DhcpRegRead( // read differnt values from registry and allocate if not DWORD
|
|
IN PREG_HANDLE Hdl,
|
|
IN DWORD Type, // if DWORD dont allocate memory
|
|
IN LPWSTR ValueName,
|
|
IN LPVOID RetValue // value to use if nothing found
|
|
) //EndExport(function)
|
|
{
|
|
DWORD Error;
|
|
DWORD Size;
|
|
DWORD Dword;
|
|
LPVOID Ret;
|
|
|
|
if( REG_DWORD == Type ) {
|
|
Size = sizeof(DWORD);
|
|
Ret = (LPVOID)&Dword;
|
|
} else {
|
|
Size = 0;
|
|
Error = RegQueryValueEx(
|
|
Hdl->Key,
|
|
ValueName,
|
|
NullReserved,
|
|
NULL,
|
|
NULL,
|
|
&Size
|
|
);
|
|
if( ERROR_SUCCESS != Error ) return RetValue;
|
|
if (Size == 0) return RetValue; // MemAlloc does not check the size!
|
|
Ret = MemAlloc(Size);
|
|
if( NULL == Ret ) return RetValue; // should not really happen
|
|
}
|
|
|
|
Error = RegQueryValueEx(
|
|
Hdl->Key,
|
|
ValueName,
|
|
NullReserved,
|
|
NULL,
|
|
Ret,
|
|
&Size
|
|
);
|
|
if( ERROR_SUCCESS != Error && Ret != (LPVOID)&Dword ) {
|
|
MemFree(Ret);
|
|
Ret = NULL;
|
|
}
|
|
|
|
if( ERROR_SUCCESS != Error) return RetValue;
|
|
|
|
if( Ret == (LPVOID)&Dword ) {
|
|
return ULongToPtr(Dword);
|
|
} else {
|
|
return Ret;
|
|
}
|
|
}
|
|
|
|
//BeginExport(function)
|
|
DWORD
|
|
DhcpRegReadBinary( // read binary type
|
|
IN PREG_HANDLE Hdl,
|
|
IN LPWSTR ValueName,
|
|
OUT LPBYTE *RetVal,
|
|
OUT DWORD *RetValSize
|
|
) //EndExport(function)
|
|
{
|
|
DWORD Error;
|
|
DWORD Size;
|
|
LPVOID Ret;
|
|
|
|
*RetVal = NULL;
|
|
*RetValSize = 0;
|
|
|
|
Size = 0;
|
|
Error = RegQueryValueEx(
|
|
Hdl->Key,
|
|
ValueName,
|
|
NullReserved,
|
|
NULL,
|
|
NULL,
|
|
&Size
|
|
);
|
|
if( ERROR_SUCCESS != Error ) return Error;
|
|
if( 0 == Size ) return ERROR_SUCCESS;
|
|
Ret = MemAlloc(Size);
|
|
if( NULL == Ret ) return ERROR_NOT_ENOUGH_MEMORY;
|
|
|
|
Error = RegQueryValueEx(
|
|
Hdl->Key,
|
|
ValueName,
|
|
NullReserved,
|
|
NULL,
|
|
Ret,
|
|
&Size
|
|
);
|
|
if( ERROR_SUCCESS != Error ) {
|
|
MemFree(Ret);
|
|
return Error;
|
|
}
|
|
|
|
*RetVal = Ret;
|
|
*RetValSize = Size;
|
|
return ERROR_SUCCESS;
|
|
}
|
|
|
|
//BeginExport(function)
|
|
LPWSTR
|
|
DhcpRegCombineClassAndOption( // create string based on class name and option id
|
|
IN LPWSTR ClassName,
|
|
IN LPWSTR VendorName,
|
|
IN DWORD OptionId
|
|
) //EndExport(function)
|
|
{
|
|
DWORD Size;
|
|
LPWSTR Ptr;
|
|
|
|
if( NULL == VendorName && NULL == ClassName ) {
|
|
|
|
//
|
|
// Special case usual options for downward compatability with older
|
|
// options.. (NT4 options in registry don't have any "!" in them.
|
|
//
|
|
|
|
Ptr = MemAlloc( 4 * sizeof(WCHAR) );
|
|
if( NULL == Ptr ) return NULL;
|
|
Ptr [2] = L'0' + (BYTE)(OptionId %10); OptionId /= 10;
|
|
Ptr [1] = L'0' + (BYTE)(OptionId %10); OptionId /= 10;
|
|
Ptr [0] = L'0' + (BYTE)(OptionId %10);
|
|
Ptr [3] = L'\0';
|
|
return Ptr;
|
|
}
|
|
|
|
if( NULL == VendorName ) VendorName = L"";
|
|
if( NULL == ClassName ) ClassName = L"";
|
|
Size = (wcslen(ClassName) + 1 + 5)*sizeof(WCHAR);
|
|
Size += wcslen(VendorName)*sizeof(WCHAR);
|
|
|
|
Ptr = MemAlloc(Size);
|
|
if( NULL == Ptr ) return NULL;
|
|
|
|
Size = 0;
|
|
|
|
Ptr[2+Size] = L'0' + (BYTE)(OptionId % 10); OptionId /= 10;
|
|
Ptr[1+Size] = L'0' + (BYTE)(OptionId % 10); OptionId /= 10;
|
|
Ptr[0+Size] = L'0' + (BYTE)(OptionId % 10);
|
|
Ptr[3+Size] = L'\0';
|
|
wcscat(Ptr, L"!");
|
|
wcscat(Ptr, VendorName);
|
|
wcscat(Ptr, L"!");
|
|
wcscat(Ptr, ClassName);
|
|
return Ptr;
|
|
}
|
|
|
|
//BeginExport(function)
|
|
LPWSTR
|
|
ConvertAddressToLPWSTR(
|
|
IN DWORD Address,
|
|
IN OUT LPWSTR BufferStr // input buffer to fill with dotted notation
|
|
) //EndExport(function)
|
|
{
|
|
LPSTR AddressStr;
|
|
DWORD Count;
|
|
|
|
Address = ntohl(Address);
|
|
AddressStr = inet_ntoa(*(struct in_addr *)&Address);
|
|
Count = mbstowcs(BufferStr, AddressStr, sizeof("000.000.000.000"));
|
|
if( -1 == Count ) return NULL;
|
|
return BufferStr;
|
|
}
|
|
|
|
//BeginExport(comment)
|
|
//================================================================================
|
|
// the following functions help traversing the registry
|
|
//================================================================================
|
|
//EndExport(comment)
|
|
|
|
DWORD
|
|
DhcpRegGetNextNextHdl(
|
|
IN PREG_HANDLE Hdl,
|
|
IN LPWSTR Loc1,
|
|
IN LPWSTR Loc2,
|
|
OUT PREG_HANDLE Hdl2
|
|
)
|
|
{
|
|
WCHAR Loc[MAX_KEY_SIZE*2];
|
|
Loc[ 0 ] = L'\0';
|
|
|
|
if ( ( wcslen(Loc1) + wcslen(Loc2) + 1 ) < ( MAX_KEY_SIZE * 2 ) )
|
|
{
|
|
wcscpy(Loc,Loc1);
|
|
wcscat(Loc, L"\\");
|
|
wcscat(Loc,Loc2);
|
|
}
|
|
|
|
return DhcpRegGetNextHdl(Hdl, Loc, Hdl2);
|
|
}
|
|
|
|
//BeginExport(function)
|
|
DWORD
|
|
DhcpRegServerGetSubnetHdl(
|
|
IN PREG_HANDLE Hdl,
|
|
IN LPWSTR Subnet,
|
|
OUT PREG_HANDLE Hdl2
|
|
) //EndExport(function)
|
|
{
|
|
return DhcpRegGetNextNextHdl(Hdl, REG_SERVER_SUBNETS, Subnet, Hdl2);
|
|
}
|
|
|
|
//BeginExport(function)
|
|
DWORD
|
|
DhcpRegServerGetSScopeHdl(
|
|
IN PREG_HANDLE Hdl,
|
|
IN LPWSTR SScope,
|
|
OUT PREG_HANDLE Hdl2
|
|
) //EndExport(function)
|
|
{
|
|
return DhcpRegGetNextNextHdl(Hdl, REG_SERVER_SSCOPES, SScope, Hdl2);
|
|
}
|
|
|
|
//BeginExport(function)
|
|
DWORD
|
|
DhcpRegServerGetOptDefHdl(
|
|
IN PREG_HANDLE Hdl,
|
|
IN LPWSTR OptDef,
|
|
OUT PREG_HANDLE Hdl2
|
|
) //EndExport(function)
|
|
{
|
|
return DhcpRegGetNextNextHdl(Hdl, REG_SERVER_OPTDEFS, OptDef, Hdl2);
|
|
}
|
|
|
|
//BeginExport(function)
|
|
DWORD
|
|
DhcpRegServerGetOptHdl(
|
|
IN PREG_HANDLE Hdl,
|
|
IN LPWSTR Opt,
|
|
OUT PREG_HANDLE Hdl2
|
|
) //EndExport(function)
|
|
{
|
|
return DhcpRegGetNextNextHdl(Hdl, REG_SERVER_GLOBAL_OPTIONS, Opt, Hdl2);
|
|
}
|
|
|
|
//BeginExport(function)
|
|
DWORD
|
|
DhcpRegServerGetMScopeHdl(
|
|
IN PREG_HANDLE Hdl,
|
|
IN LPWSTR MScope,
|
|
OUT PREG_HANDLE Hdl2
|
|
) //EndExport(function)
|
|
{
|
|
return DhcpRegGetNextNextHdl(Hdl, REG_SERVER_MSCOPES, MScope, Hdl2);
|
|
}
|
|
|
|
//BeginExport(function)
|
|
DWORD
|
|
DhcpRegServerGetClassDefHdl(
|
|
IN PREG_HANDLE Hdl,
|
|
IN LPWSTR ClassDef,
|
|
OUT PREG_HANDLE Hdl2
|
|
) //EndExport(function)
|
|
{
|
|
return DhcpRegGetNextNextHdl(Hdl, REG_SERVER_CLASSDEFS, ClassDef, Hdl2);
|
|
}
|
|
|
|
//BeginExport(function)
|
|
DWORD
|
|
DhcpRegSubnetGetOptHdl(
|
|
IN PREG_HANDLE Hdl,
|
|
IN LPWSTR Opt,
|
|
OUT PREG_HANDLE Hdl2
|
|
) //EndExport(function)
|
|
{
|
|
return DhcpRegGetNextNextHdl(Hdl, REG_SUBNET_OPTIONS, Opt, Hdl2);
|
|
}
|
|
|
|
//BeginExport(function)
|
|
DWORD
|
|
DhcpRegSubnetGetRangeHdl(
|
|
IN PREG_HANDLE Hdl,
|
|
IN LPWSTR Range,
|
|
OUT PREG_HANDLE Hdl2
|
|
) //EndExport(function)
|
|
{
|
|
return DhcpRegGetNextNextHdl(Hdl, REG_SUBNET_RANGES, Range, Hdl2);
|
|
}
|
|
|
|
//BeginExport(function)
|
|
DWORD
|
|
DhcpRegSubnetGetReservationHdl(
|
|
IN PREG_HANDLE Hdl,
|
|
IN LPWSTR Reservation,
|
|
OUT PREG_HANDLE Hdl2
|
|
) //EndExport(function)
|
|
{
|
|
return DhcpRegGetNextNextHdl(Hdl, REG_SUBNET_RESERVATIONS, Reservation, Hdl2);
|
|
}
|
|
|
|
//BeginExport(function)
|
|
DWORD
|
|
DhcpRegSubnetGetServerHdl(
|
|
IN PREG_HANDLE Hdl,
|
|
IN LPWSTR Server,
|
|
OUT PREG_HANDLE Hdl2
|
|
) //EndExport(function)
|
|
{
|
|
return DhcpRegGetNextNextHdl(Hdl, REG_SUBNET_SERVERS, Server, Hdl2);
|
|
}
|
|
|
|
//BeginExport(function)
|
|
DWORD
|
|
DhcpRegReservationGetOptHdl(
|
|
IN PREG_HANDLE Hdl,
|
|
IN LPWSTR OptionName,
|
|
OUT PREG_HANDLE Hdl2
|
|
) //EndExport(function)
|
|
{
|
|
return DhcpRegGetNextHdl(Hdl, OptionName, Hdl2);
|
|
}
|
|
|
|
//BeginExport(comment)
|
|
//================================================================================
|
|
// List retrieval functions.. for servers, subnets, ranges etc.
|
|
//================================================================================
|
|
//EndExport(comment)
|
|
|
|
//BeginExport(function)
|
|
DWORD
|
|
DhcpRegServerGetList(
|
|
IN PREG_HANDLE Hdl, // ptr to server location
|
|
IN OUT PARRAY OptList, // list of LPWSTR options
|
|
IN OUT PARRAY OptDefList, // list of LPWSTR optdefs
|
|
IN OUT PARRAY Subnets, // list of LPWSTR subnets
|
|
IN OUT PARRAY SScopes, // list of LPWSTR sscopes
|
|
IN OUT PARRAY ClassDefs, // list of LPWSTR classes
|
|
IN OUT PARRAY MScopes // list of LPWSTR mscopes
|
|
) //EndExport(function)
|
|
{
|
|
DWORD Error;
|
|
DWORD LocalError;
|
|
DWORD Index;
|
|
REG_HANDLE Hdl2;
|
|
struct {
|
|
PARRAY Array;
|
|
LPWSTR Location;
|
|
} Table[] = {
|
|
OptList, REG_SERVER_GLOBAL_OPTIONS,
|
|
OptDefList, REG_SERVER_OPTDEFS,
|
|
Subnets, REG_SERVER_SUBNETS,
|
|
SScopes, REG_SERVER_SSCOPES,
|
|
ClassDefs, REG_SERVER_CLASSDEFS,
|
|
MScopes, REG_SERVER_MSCOPES
|
|
};
|
|
|
|
for( Index = 0; Index < sizeof(Table)/sizeof(Table[0]); Index ++ ) {
|
|
if( NULL == Table[Index].Array ) continue;
|
|
|
|
Error = DhcpRegGetNextHdl(Hdl, Table[Index].Location, &Hdl2);
|
|
if( ERROR_SUCCESS != Error ) return Error;
|
|
|
|
Error = DhcpRegFillSubKeys(&Hdl2, Table[Index].Array);
|
|
LocalError = DhcpRegCloseHdl(&Hdl2);
|
|
Require(ERROR_SUCCESS == LocalError);
|
|
|
|
if( ERROR_SUCCESS != Error ) return Error;
|
|
}
|
|
|
|
return Error;
|
|
}
|
|
|
|
//BeginExport(function)
|
|
DWORD
|
|
DhcpRegSubnetGetExclusions(
|
|
IN PREG_HANDLE Hdl,
|
|
OUT LPBYTE *Excl,
|
|
OUT DWORD *ExclSize
|
|
) //EndExport(function)
|
|
{
|
|
DWORD Error;
|
|
DWORD Size;
|
|
DWORD Type;
|
|
|
|
if( NULL == Excl ) return ERROR_SUCCESS;
|
|
|
|
Size = 0;
|
|
Error = RegQueryValueEx(
|
|
Hdl->Key,
|
|
REG_SUBNET_EXCL,
|
|
NullReserved,
|
|
&Type,
|
|
NULL,
|
|
&Size
|
|
);
|
|
if( ERROR_SUCCESS != Error ) return Error;
|
|
|
|
*Excl = NULL;
|
|
*ExclSize = 0;
|
|
if( 0 == Size ) return ERROR_SUCCESS;
|
|
|
|
*Excl = MemAlloc(Size);
|
|
if( NULL == *Excl ) return ERROR_NOT_ENOUGH_MEMORY;
|
|
*ExclSize = Size;
|
|
Error = RegQueryValueEx(
|
|
Hdl->Key,
|
|
REG_SUBNET_EXCL,
|
|
NullReserved,
|
|
&Type,
|
|
*Excl,
|
|
ExclSize
|
|
);
|
|
if( ERROR_SUCCESS != Error ) {
|
|
MemFree(*Excl);
|
|
*Excl = NULL;
|
|
}
|
|
return Error;
|
|
}
|
|
|
|
//BeginExport(function)
|
|
DWORD
|
|
DhcpRegSubnetGetList(
|
|
IN PREG_HANDLE Hdl,
|
|
IN OUT PARRAY Servers,
|
|
IN OUT PARRAY IpRanges,
|
|
IN OUT PARRAY Reservations,
|
|
IN OUT PARRAY Options,
|
|
OUT LPBYTE *Excl,
|
|
OUT DWORD *ExclSizeInBytes
|
|
) //EndExport(function)
|
|
{
|
|
DWORD Error;
|
|
DWORD LocalError;
|
|
DWORD Index;
|
|
REG_HANDLE Hdl2;
|
|
struct {
|
|
PARRAY Array;
|
|
LPWSTR Location;
|
|
} Table[] = {
|
|
Servers, REG_SUBNET_SERVERS,
|
|
IpRanges, REG_SUBNET_RANGES,
|
|
Reservations, REG_SUBNET_RESERVATIONS,
|
|
Options, REG_SUBNET_OPTIONS,
|
|
// Exclusions are to be handled a bit differently
|
|
};
|
|
|
|
for( Index = 0; Index < sizeof(Table)/sizeof(Table[0]); Index ++ ) {
|
|
if( NULL == Table[Index].Array ) continue;
|
|
|
|
Error = DhcpRegGetNextHdl(Hdl, Table[Index].Location, &Hdl2);
|
|
if( ERROR_SUCCESS != Error ) return Error;
|
|
|
|
Error = DhcpRegFillSubKeys(&Hdl2, Table[Index].Array);
|
|
LocalError = DhcpRegCloseHdl(&Hdl2);
|
|
Require(ERROR_SUCCESS == LocalError);
|
|
|
|
if( ERROR_SUCCESS != Error ) return Error;
|
|
}
|
|
|
|
// Now read the exclusions from off here
|
|
return DhcpRegSubnetGetExclusions(Hdl, Excl, ExclSizeInBytes );
|
|
}
|
|
|
|
//BeginExport(function)
|
|
DWORD
|
|
DhcpRegSScopeGetList(
|
|
IN PREG_HANDLE Hdl,
|
|
IN OUT PARRAY Subnets
|
|
) //EndExport(function)
|
|
{
|
|
return DhcpRegFillSubKeys(Hdl, Subnets);
|
|
}
|
|
|
|
//BeginExport(function)
|
|
DWORD
|
|
DhcpRegReservationGetList(
|
|
IN PREG_HANDLE Hdl,
|
|
IN OUT PARRAY Options
|
|
) //EndExport(function)
|
|
{
|
|
return DhcpRegFillSubKeys(Hdl, Options);
|
|
}
|
|
|
|
//BeginExport(comment)
|
|
//================================================================================
|
|
// the separate stuff are here -- these are not list stuff, but just simple
|
|
// single valued attributes
|
|
// some of these actually, dont even go to the registry, but that's fine alright?
|
|
//================================================================================
|
|
//EndExport(comment)
|
|
|
|
//BeginExport(function)
|
|
DWORD
|
|
DhcpRegServerGetAttributes(
|
|
IN PREG_HANDLE Hdl,
|
|
OUT LPWSTR *Name,
|
|
OUT LPWSTR *Comment,
|
|
OUT DWORD *Flags
|
|
// more attributes will come here soon?
|
|
) //EndExport(function)
|
|
{
|
|
if( Name ) *Name = NULL;
|
|
if( Comment ) *Comment = NULL;
|
|
if( Flags ) *Flags = 0;
|
|
|
|
return ERROR_SUCCESS;
|
|
}
|
|
|
|
//BeginExport(function)
|
|
DWORD
|
|
DhcpRegSubnetGetAttributes(
|
|
IN PREG_HANDLE Hdl,
|
|
OUT LPWSTR *Name,
|
|
OUT LPWSTR *Comment,
|
|
OUT DWORD *Flags,
|
|
OUT DWORD *Address,
|
|
OUT DWORD *Mask
|
|
) //EndExport(function)
|
|
{
|
|
DWORD fSwitched;
|
|
|
|
if( Name ) *Name = DhcpRegRead(Hdl, REG_SZ, REG_SUBNET_NAME, NULL);
|
|
if( Comment ) *Comment = DhcpRegRead(Hdl, REG_SZ, REG_SUBNET_COMMENT, NULL);
|
|
if( Flags ) {
|
|
*Flags = PtrToUlong(DhcpRegRead(Hdl, REG_DWORD, REG_SUBNET_STATE, (LPVOID)REG_DEFAULT_SUBNET_STATE));
|
|
fSwitched = PtrToUlong(DhcpRegRead(Hdl, REG_DWORD, REG_SUBNET_SWITCHED_FLAG, (LPVOID)REG_DEFAULT_SWITCHED_FLAG));
|
|
if(fSwitched) SWITCHED(*Flags);
|
|
}
|
|
if( Address )
|
|
*Address = PtrToUlong(DhcpRegRead(Hdl, REG_DWORD, REG_SUBNET_ADDRESS, (LPVOID)0));
|
|
if( Mask ) *Mask = PtrToUlong(DhcpRegRead(Hdl, REG_DWORD, REG_SUBNET_MASK, ULongToPtr(REG_DEFAULT_SUBNET_MASK)));
|
|
|
|
return ERROR_SUCCESS;
|
|
}
|
|
|
|
typedef struct {
|
|
LPVOID RetPtr;
|
|
LPWSTR ValueName;
|
|
DWORD ValueType;
|
|
LPVOID Defaults;
|
|
} ATTRIB_TBL, *PATTRIB_TBL, *LPATTRIB_TBL;
|
|
|
|
VOID
|
|
DhcpRegFillAttribTable(
|
|
IN PREG_HANDLE Hdl,
|
|
IN PATTRIB_TBL Table,
|
|
IN DWORD TableSize
|
|
) {
|
|
DWORD i;
|
|
PVOID Tmp;
|
|
|
|
for( i = 0; i < TableSize ; i ++ ) {
|
|
if( NULL == Table[i].RetPtr) continue;
|
|
Tmp = DhcpRegRead(
|
|
Hdl,
|
|
Table[i].ValueType,
|
|
Table[i].ValueName,
|
|
Table[i].Defaults
|
|
);
|
|
if( REG_DWORD == Table[i].ValueType ) {
|
|
*((DWORD *)Table[i].RetPtr) = PtrToUlong(Tmp);
|
|
} else {
|
|
*((LPVOID *)Table[i].RetPtr) = Tmp;
|
|
}
|
|
}
|
|
}
|
|
|
|
//
|
|
// Hack O Rama -- This routine returns ERROR_INVALID_DATA if
|
|
// the registry has been upgraded from pre-win2k build to win2k.
|
|
// So that defaults can be chosen for ScopeId etc.
|
|
//
|
|
//BeginExport(function)
|
|
DWORD
|
|
DhcpRegMScopeGetAttributes(
|
|
IN PREG_HANDLE Hdl,
|
|
OUT LPWSTR *Comments,
|
|
OUT DWORD *State,
|
|
OUT DWORD *ScopeId,
|
|
OUT DWORD *Policy,
|
|
OUT DWORD *TTL,
|
|
OUT LPWSTR *LangTag,
|
|
OUT PDATE_TIME *ExpiryTime
|
|
) //EndExport(function)
|
|
{
|
|
DATE_TIME DefaultExpiryTime = {DHCP_DATE_TIME_INFINIT_LOW, DHCP_DATE_TIME_INFINIT_HIGH};
|
|
LPWSTR DefaultLangTag = L"en-US";
|
|
DWORD ScopeIdFake1, ScopeIdFake2;
|
|
|
|
ATTRIB_TBL Table[] = {
|
|
Comments, REG_MSCOPE_COMMENT, REG_SZ, NULL,
|
|
State, REG_MSCOPE_STATE, REG_DWORD, (LPVOID)0,
|
|
ScopeId, REG_MSCOPE_SCOPEID, REG_DWORD, (LPVOID)0,
|
|
&ScopeIdFake1, REG_MSCOPE_SCOPEID, REG_DWORD, (LPVOID)1,
|
|
&ScopeIdFake2, REG_MSCOPE_SCOPEID, REG_DWORD, (LPVOID)2,
|
|
Policy, REG_MSCOPE_ADDR_POLICY, REG_DWORD, (LPVOID)0,
|
|
TTL, REG_MSCOPE_TTL, REG_DWORD, (LPVOID)DEFAULT_MCAST_TTL,
|
|
LangTag, REG_MSCOPE_LANG_TAG, REG_SZ, (LPVOID)0,
|
|
ExpiryTime, REG_MSCOPE_EXPIRY_TIME, REG_BINARY, (LPVOID)0
|
|
};
|
|
DhcpRegFillAttribTable(Hdl, Table, sizeof(Table)/sizeof(Table[0]));
|
|
if (*LangTag == 0) {
|
|
*LangTag = MemAlloc(wcslen((DefaultLangTag)+1)*sizeof(WCHAR));
|
|
if (*LangTag) {
|
|
wcscpy(*LangTag, DefaultLangTag);
|
|
} else {
|
|
return ERROR_NOT_ENOUGH_MEMORY;
|
|
}
|
|
}
|
|
if (*ExpiryTime == 0) {
|
|
*ExpiryTime = MemAlloc(sizeof (DefaultExpiryTime));
|
|
if (*ExpiryTime) {
|
|
**ExpiryTime = DefaultExpiryTime;
|
|
} else {
|
|
return ERROR_NOT_ENOUGH_MEMORY;
|
|
}
|
|
|
|
}
|
|
|
|
if( ScopeIdFake1 != ScopeIdFake2 ) {
|
|
Require(ScopeIdFake1 == 1 && ScopeIdFake2 == 2);
|
|
//
|
|
// Basically no value for ScopeId in the registry. return a warning.
|
|
// such as ERROR_INVALID_DATA.
|
|
//
|
|
return ERROR_INVALID_DATA;
|
|
}
|
|
|
|
return ERROR_SUCCESS;
|
|
}
|
|
|
|
//BeginExport(function)
|
|
DWORD
|
|
DhcpRegOptDefGetAttributes(
|
|
IN PREG_HANDLE Hdl,
|
|
OUT LPWSTR *Name,
|
|
OUT LPWSTR *Comments,
|
|
OUT DWORD *Flags,
|
|
OUT DWORD *OptionId,
|
|
OUT LPWSTR *ClassName,
|
|
OUT LPWSTR *VendorName,
|
|
OUT LPBYTE *Value,
|
|
OUT DWORD *ValueSize
|
|
) //EndExport(function)
|
|
{
|
|
DWORD Error;
|
|
ATTRIB_TBL Table[] = {
|
|
Name, REG_OPTION_NAME, REG_SZ, NULL,
|
|
Comments, REG_OPTION_COMMENT, REG_SZ, NULL,
|
|
ClassName, REG_OPTION_CLASSNAME, REG_SZ, NULL,
|
|
VendorName, REG_OPTION_VENDORNAME, REG_SZ, NULL,
|
|
Flags, REG_OPTION_TYPE, REG_DWORD, (LPVOID)0,
|
|
OptionId, REG_OPTION_ID, REG_DWORD, (LPVOID)0
|
|
};
|
|
|
|
DhcpRegFillAttribTable(Hdl, Table, sizeof(Table)/sizeof(Table[0]));
|
|
if( Value ) {
|
|
Error = DhcpRegReadBinary(Hdl, REG_OPTION_VALUE, Value, ValueSize);
|
|
Require(*Value);
|
|
}
|
|
return ERROR_SUCCESS;
|
|
}
|
|
|
|
//BeginExport(function)
|
|
DWORD
|
|
DhcpRegSScopeGetAttributes( // superscopes dont have any information stored.. dont use this
|
|
IN PREG_HANDLE Hdl,
|
|
OUT LPWSTR *Name,
|
|
OUT LPWSTR *Comment,
|
|
OUT DWORD *Flags
|
|
) //EndExport(function)
|
|
{
|
|
Require(FALSE);
|
|
return ERROR_INVALID_PARAMETER;
|
|
}
|
|
|
|
//BeginExport(function)
|
|
DWORD
|
|
DhcpRegClassDefGetAttributes(
|
|
IN PREG_HANDLE Hdl,
|
|
OUT LPWSTR *Name,
|
|
OUT LPWSTR *Comment,
|
|
OUT DWORD *Flags,
|
|
OUT LPBYTE *Value,
|
|
OUT DWORD *ValueSize
|
|
) //EndExport(function)
|
|
{
|
|
DWORD Error;
|
|
ATTRIB_TBL Table[] = {
|
|
Name, REG_CLASSDEF_NAME, REG_SZ, NULL,
|
|
Comment, REG_CLASSDEF_COMMENT, REG_SZ, NULL,
|
|
Flags, REG_CLASSDEF_TYPE, REG_DWORD, (LPVOID)0
|
|
};
|
|
|
|
DhcpRegFillAttribTable(Hdl, Table, sizeof(Table)/sizeof(Table[0]));
|
|
if( Value ) {
|
|
Error = DhcpRegReadBinary(Hdl, REG_CLASSDEF_VALUE, Value, ValueSize);
|
|
Require(*Value);
|
|
}
|
|
return ERROR_SUCCESS;
|
|
}
|
|
|
|
//BeginExport(function)
|
|
DWORD
|
|
DhcpRegSubnetServerGetAttributes(
|
|
IN PREG_HANDLE Hdl,
|
|
OUT LPWSTR *Name,
|
|
OUT LPWSTR *Comment,
|
|
OUT DWORD *Flags,
|
|
OUT DWORD *Address,
|
|
OUT DWORD *Role
|
|
) //EndExport(function)
|
|
{
|
|
ATTRIB_TBL Table[] = {
|
|
Name, REG_SUB_SERVER_NAME, REG_SZ, NULL,
|
|
Comment, REG_SUB_SERVER_COMMENT,REG_SZ, NULL,
|
|
Flags, REG_FLAGS, REG_DWORD, (LPVOID)0,
|
|
Address, REG_SUB_SERVER_ADDRESS,REG_DWORD, (LPVOID)0,
|
|
Role, REG_SUB_SERVER_ROLE, REG_DWORD, (LPVOID)0
|
|
};
|
|
|
|
DhcpRegFillAttribTable(Hdl, Table, sizeof(Table)/sizeof(Table[0]));
|
|
return ERROR_SUCCESS;
|
|
}
|
|
|
|
//BeginExport(function)
|
|
DWORD
|
|
DhcpRegRangeGetAttributes(
|
|
IN PREG_HANDLE Hdl,
|
|
OUT LPWSTR *Name,
|
|
OUT LPWSTR *Comment,
|
|
OUT DWORD *Flags,
|
|
OUT ULONG *AllocCount,
|
|
OUT ULONG *MaxAllocCount,
|
|
OUT DWORD *StartAddress,
|
|
OUT DWORD *EndAddress,
|
|
OUT LPBYTE *InUseClusters,
|
|
OUT DWORD *InUseClusterSize,
|
|
OUT LPBYTE *UsedClusters,
|
|
OUT DWORD *UsedClustersSize
|
|
) //EndExport(function)
|
|
{
|
|
DWORD Error;
|
|
ATTRIB_TBL Table[] = {
|
|
Name, REG_RANGE_NAME, REG_SZ, NULL,
|
|
Comment, REG_RANGE_COMMENT, REG_SZ, NULL,
|
|
Flags, REG_RANGE_FLAGS, REG_DWORD, (LPVOID)(DEF_RANGE_FLAG_VAL),
|
|
AllocCount, REG_RANGE_ALLOC, REG_DWORD, (LPVOID)(DEF_RANGE_ALLOC),
|
|
MaxAllocCount, REG_RANGE_MAX_ALLOC, REG_DWORD, (LPVOID)(ULONG_PTR)(DEF_RANGE_MAX_ALLOC),
|
|
StartAddress, REG_RANGE_START_ADDRESS, REG_DWORD, (LPVOID)0,
|
|
EndAddress, REG_RANGE_END_ADDRESS, REG_DWORD, (LPVOID)0
|
|
};
|
|
|
|
DhcpRegFillAttribTable(Hdl, Table, sizeof(Table)/sizeof(Table[0]));
|
|
if( InUseClusters ) {
|
|
Error = DhcpRegReadBinary(Hdl, REG_RANGE_INUSE_CLUSTERS, InUseClusters, InUseClusterSize);
|
|
//Require(ERROR_SUCCESS == Error); //-- after registry changed, NO_SUCH_FILE could come up here as well.
|
|
}
|
|
if( UsedClusters ) {
|
|
Error = DhcpRegReadBinary(Hdl, REG_RANGE_USED_CLUSTERS, UsedClusters, UsedClustersSize);
|
|
//Require(ERROR_SUCCESS == Error); //-- after registry changed, NO_SUCH_FILE could come up here as well.
|
|
}
|
|
return ERROR_SUCCESS;
|
|
}
|
|
|
|
//BeginExport(function)
|
|
DWORD
|
|
DhcpRegReservationGetAttributes(
|
|
IN PREG_HANDLE Hdl,
|
|
OUT LPWSTR *Name,
|
|
OUT LPWSTR *Comment,
|
|
OUT DWORD *Flags,
|
|
OUT DWORD *Address,
|
|
OUT LPBYTE *ClientUID,
|
|
OUT DWORD *ClientUIDSize
|
|
) //EndExport(function)
|
|
{
|
|
DWORD Error;
|
|
DWORD deftype = DEF_RESERVATION_TYPE;
|
|
ATTRIB_TBL Table[] = {
|
|
Name, REG_RESERVATION_NAME, REG_SZ, NULL,
|
|
Comment, REG_RESERVATION_COMMENT, REG_SZ, NULL,
|
|
Flags, REG_RESERVATION_TYPE, REG_DWORD, ULongToPtr(deftype),
|
|
Address, REG_RESERVATION_ADDRESS, REG_DWORD, (LPVOID)0,
|
|
};
|
|
|
|
DhcpRegFillAttribTable(Hdl, Table, sizeof(Table)/sizeof(Table[0]));
|
|
if( ClientUID ) {
|
|
Error = DhcpRegReadBinary(Hdl, REG_RESERVATION_UID, ClientUID, ClientUIDSize);
|
|
Require(ERROR_SUCCESS == Error);
|
|
}
|
|
return ERROR_SUCCESS;
|
|
}
|
|
|
|
//BeginExport(function)
|
|
DWORD
|
|
DhcpRegOptGetAttributes(
|
|
IN PREG_HANDLE Hdl,
|
|
OUT DWORD *OptionId,
|
|
OUT LPWSTR *ClassName,
|
|
OUT LPWSTR *VendorName,
|
|
OUT DWORD *Flags,
|
|
OUT LPBYTE *Value,
|
|
OUT DWORD *ValueSize
|
|
) //EndExport(function)
|
|
{
|
|
DWORD Error;
|
|
ATTRIB_TBL Table[] = {
|
|
OptionId, REG_OPTION_ID, REG_DWORD, (LPVOID)0,
|
|
ClassName, REG_OPTION_CLASSNAME, REG_SZ, NULL,
|
|
VendorName, REG_OPTION_VENDORNAME, REG_SZ, NULL,
|
|
Flags, REG_OPTION_TYPE, REG_DWORD, (LPVOID)0,
|
|
};
|
|
|
|
DhcpRegFillAttribTable(Hdl, Table, sizeof(Table)/sizeof(Table[0]));
|
|
if( Value ) {
|
|
Error = DhcpRegReadBinary(Hdl, REG_OPTION_VALUE, Value, ValueSize);
|
|
}
|
|
return ERROR_SUCCESS;
|
|
}
|
|
|
|
//BeginExport(comment)
|
|
//================================================================================
|
|
// the following functiosn help in writing to the registry
|
|
//================================================================================
|
|
//EndExport(comment)
|
|
|
|
typedef struct {
|
|
LPVOID Value;
|
|
DWORD Size;
|
|
DWORD Type;
|
|
LPWSTR ValueName;
|
|
} WATTRIB_TBL, *PWATTRIB_TBL, *LPWATTRIB_TBL;
|
|
|
|
DWORD
|
|
DhcpRegSaveAttribTable(
|
|
IN PREG_HANDLE Hdl,
|
|
IN PWATTRIB_TBL Table,
|
|
IN DWORD Size
|
|
)
|
|
{
|
|
DWORD i;
|
|
DWORD Error;
|
|
DWORD PtrSize;
|
|
LPBYTE Ptr;
|
|
|
|
for(i = 0; i < Size; i ++ ) {
|
|
if( NULL == Table[i].Value ) continue;
|
|
PtrSize = Table[i].Size;
|
|
Ptr = *(LPBYTE *)Table[i].Value;
|
|
switch(Table[i].Type) {
|
|
case REG_SZ:
|
|
if( NULL == *(LPWSTR *)Table[i].Value) { PtrSize = sizeof(WCHAR); Ptr = (LPBYTE)L""; break; }
|
|
PtrSize = sizeof(WCHAR)*(wcslen(*((LPWSTR *)Table[i].Value))+1);
|
|
Ptr = *(LPBYTE *)Table[i].Value;
|
|
break;
|
|
case REG_DWORD:
|
|
PtrSize = sizeof(DWORD);
|
|
Ptr = Table[i].Value; // This is because we deref this ptr down below..
|
|
break;
|
|
}
|
|
|
|
Error = RegSetValueEx(
|
|
Hdl->Key,
|
|
Table[i].ValueName,
|
|
ZeroReserved,
|
|
Table[i].Type,
|
|
Ptr,
|
|
PtrSize
|
|
);
|
|
if( ERROR_SUCCESS != Error ) {
|
|
return Error;
|
|
}
|
|
}
|
|
return ERROR_SUCCESS;
|
|
}
|
|
|
|
//BeginExport(functions)
|
|
DWORD
|
|
DhcpRegSaveSubKeys(
|
|
IN PREG_HANDLE Hdl,
|
|
IN OUT PARRAY Array
|
|
) //EndExport(function)
|
|
{
|
|
ARRAY_LOCATION Loc;
|
|
DWORD Error;
|
|
REG_HANDLE Hdl2;
|
|
LPWSTR KeyName;
|
|
|
|
Error = MemArrayInitLoc(Array, &Loc);
|
|
while(ERROR_FILE_NOT_FOUND != Error ) {
|
|
Require(ERROR_SUCCESS == Error);
|
|
|
|
Error = MemArrayGetElement(Array, &Loc, (LPVOID *)&KeyName);
|
|
Require(ERROR_SUCCESS == Error && NULL != KeyName);
|
|
|
|
Error = DhcpRegGetNextHdl(Hdl, KeyName, &Hdl2);
|
|
if( ERROR_SUCCESS != Error ) return Error;
|
|
|
|
Error = DhcpRegCloseHdl(&Hdl2);
|
|
Require(ERROR_SUCCESS == Error);
|
|
|
|
Error = MemArrayNextLoc(Array, &Loc);
|
|
}
|
|
return ERROR_SUCCESS;
|
|
}
|
|
|
|
//BeginExport(functions)
|
|
DWORD
|
|
DhcpRegSaveSubKeysPrefixed(
|
|
IN PREG_HANDLE Hdl,
|
|
IN OUT PARRAY Array,
|
|
IN LPWSTR CommonPrefix
|
|
) //EndExport(function)
|
|
{
|
|
ARRAY_LOCATION Loc;
|
|
DWORD Error;
|
|
REG_HANDLE Hdl2;
|
|
LPWSTR KeyName;
|
|
|
|
Error = MemArrayInitLoc(Array, &Loc);
|
|
while(ERROR_FILE_NOT_FOUND != Error ) {
|
|
Require(ERROR_SUCCESS == Error);
|
|
|
|
Error = MemArrayGetElement(Array, &Loc, (LPVOID *)&KeyName);
|
|
Require(ERROR_SUCCESS == Error && NULL != KeyName);
|
|
|
|
Error = DhcpRegGetNextNextHdl(Hdl, CommonPrefix, KeyName, &Hdl2);
|
|
if( ERROR_SUCCESS != Error ) return Error;
|
|
|
|
Error = DhcpRegCloseHdl(&Hdl2);
|
|
Require(ERROR_SUCCESS == Error);
|
|
|
|
Error = MemArrayNextLoc(Array, &Loc);
|
|
}
|
|
return ERROR_SUCCESS;
|
|
}
|
|
|
|
//BeginExport(function)
|
|
DWORD
|
|
DhcpRegServerSetList(
|
|
IN PREG_HANDLE Hdl,
|
|
IN PARRAY OptList, // list of LPWSTR options
|
|
IN PARRAY OptDefList, // list of LPWSTR optdefs
|
|
IN PARRAY Subnets, // list of LPWSTR subnets
|
|
IN PARRAY SScopes, // list of LPWSTR sscopes
|
|
IN PARRAY ClassDefs, // list of LPWSTR classes
|
|
IN PARRAY MScopes // list of LPWSTR mscopes
|
|
) //EndExport(function)
|
|
{
|
|
DWORD Error;
|
|
DWORD LocalError;
|
|
DWORD Index;
|
|
REG_HANDLE Hdl2;
|
|
struct {
|
|
PARRAY Array;
|
|
LPWSTR Location;
|
|
} Table[] = {
|
|
OptList, REG_SERVER_GLOBAL_OPTIONS,
|
|
OptDefList, REG_SERVER_OPTDEFS,
|
|
Subnets, REG_SERVER_SUBNETS,
|
|
SScopes, REG_SERVER_SSCOPES,
|
|
ClassDefs, REG_SERVER_CLASSDEFS,
|
|
MScopes, REG_SERVER_MSCOPES
|
|
};
|
|
|
|
for( Index = 0; Index < sizeof(Table)/sizeof(Table[0]); Index ++ ) {
|
|
if( NULL == Table[Index].Array ) continue;
|
|
|
|
Error = DhcpRegGetNextHdl(Hdl, Table[Index].Location, &Hdl2);
|
|
if( ERROR_SUCCESS != Error ) return Error;
|
|
|
|
Error = DhcpRegSaveSubKeys(&Hdl2, Table[Index].Array);
|
|
LocalError = DhcpRegCloseHdl(&Hdl2);
|
|
Require(ERROR_SUCCESS == LocalError);
|
|
|
|
if( ERROR_SUCCESS != Error ) return Error;
|
|
}
|
|
|
|
return Error;
|
|
}
|
|
|
|
//BeginExport(function)
|
|
DWORD
|
|
DhcpRegSubnetSetExclusions(
|
|
IN PREG_HANDLE Hdl,
|
|
IN LPBYTE *Excl,
|
|
IN DWORD ExclSize
|
|
) //EndExport(function)
|
|
{
|
|
DWORD Error;
|
|
WATTRIB_TBL Table[] = {
|
|
(LPVOID*)Excl, ExclSize, REG_BINARY, REG_SUBNET_EXCL,
|
|
};
|
|
|
|
return DhcpRegSaveAttribTable(Hdl, Table, sizeof(Table)/sizeof(Table[0]));
|
|
}
|
|
|
|
//BeginExport(function)
|
|
DWORD
|
|
DhcpRegSubnetSetList(
|
|
IN PREG_HANDLE Hdl,
|
|
IN PARRAY Servers,
|
|
IN PARRAY IpRanges,
|
|
IN PARRAY Reservations,
|
|
IN PARRAY Options,
|
|
IN LPBYTE *Excl,
|
|
IN DWORD ExclSizeInBytes
|
|
) //EndExport(function)
|
|
{
|
|
DWORD Error;
|
|
DWORD LocalError;
|
|
DWORD Index;
|
|
REG_HANDLE Hdl2;
|
|
struct {
|
|
PARRAY Array;
|
|
LPWSTR Location;
|
|
} Table[] = {
|
|
Servers, REG_SUBNET_SERVERS,
|
|
IpRanges, REG_SUBNET_RANGES,
|
|
Reservations, REG_SUBNET_RESERVATIONS,
|
|
Options, REG_SUBNET_OPTIONS,
|
|
// Exclusions are to be handled a bit differently
|
|
};
|
|
|
|
for( Index = 0; Index < sizeof(Table)/sizeof(Table[0]); Index ++ ) {
|
|
if( NULL == Table[Index].Array ) continue;
|
|
|
|
Error = DhcpRegGetNextHdl(Hdl, Table[Index].Location, &Hdl2);
|
|
if( ERROR_SUCCESS != Error ) return Error;
|
|
|
|
Error = DhcpRegSaveSubKeys(&Hdl2, Table[Index].Array);
|
|
LocalError = DhcpRegCloseHdl(&Hdl2);
|
|
Require(ERROR_SUCCESS == LocalError);
|
|
|
|
if( ERROR_SUCCESS != Error ) return Error;
|
|
}
|
|
|
|
// Now read the exclusions from off here
|
|
return DhcpRegSubnetSetExclusions(Hdl, Excl, ExclSizeInBytes );
|
|
}
|
|
|
|
//BeginExport(function)
|
|
DWORD
|
|
DhcpRegSScopeSetList(
|
|
IN PREG_HANDLE Hdl,
|
|
IN OUT PARRAY Subnets
|
|
) //EndExport(function)
|
|
{
|
|
return DhcpRegSaveSubKeys(Hdl, Subnets);
|
|
}
|
|
|
|
//BeginExport(function)
|
|
DWORD
|
|
DhcpRegReservationSetList(
|
|
IN PREG_HANDLE Hdl,
|
|
IN PARRAY Subnets
|
|
) //EndExport(function)
|
|
{
|
|
return DhcpRegSaveSubKeys(Hdl, Subnets);
|
|
}
|
|
|
|
//BeginExport(comment)
|
|
//================================================================================
|
|
// the single stuff are here -- these are not list stuff, but just simple
|
|
// single valued attributes
|
|
// some of these actually, dont even go to the registry, but that's fine alright?
|
|
//================================================================================
|
|
//EndExport(comment)
|
|
|
|
//BeginExport(function)
|
|
DWORD
|
|
DhcpRegServerSetAttributes(
|
|
IN PREG_HANDLE Hdl,
|
|
IN LPWSTR *Name,
|
|
IN LPWSTR *Comment,
|
|
IN DWORD *Flags
|
|
// more attributes will come here soon?
|
|
) //EndExport(function)
|
|
{
|
|
#if 0
|
|
WATTRIB_TBL Table[] = {
|
|
Name, REG_SERVER_NAME
|
|
}
|
|
if( Name ) *Name = NULL;
|
|
if( Comment ) *Comment = NULL;
|
|
if( Flags ) *Flags = 0;
|
|
#endif
|
|
|
|
return ERROR_SUCCESS;
|
|
}
|
|
|
|
DWORD
|
|
DhcpRegSubnetSetAttributesInternal(
|
|
IN PREG_HANDLE Hdl,
|
|
IN LPWSTR *Name,
|
|
IN LPWSTR *Comment,
|
|
IN DWORD *Flags,
|
|
IN DWORD *Address,
|
|
IN DWORD *Mask,
|
|
IN DWORD *SwitchedNetwork
|
|
)
|
|
{
|
|
DWORD Error, SrvIpAddress, SrvRole;
|
|
LPWSTR EmptyString;
|
|
REG_HANDLE Hdl2;
|
|
WATTRIB_TBL Table[] = {
|
|
Name, 0, REG_SZ, REG_SUBNET_NAME,
|
|
Comment, 0, REG_SZ, REG_SUBNET_COMMENT,
|
|
Flags, 0, REG_DWORD, REG_SUBNET_STATE,
|
|
Address, 0, REG_DWORD, REG_SUBNET_ADDRESS,
|
|
Mask, 0, REG_DWORD, REG_SUBNET_MASK,
|
|
SwitchedNetwork, 0, REG_DWORD, REG_SUBNET_SWITCHED_FLAG
|
|
};
|
|
|
|
Error = DhcpRegSaveAttribTable(Hdl, Table, sizeof(Table)/sizeof(Table[0]));
|
|
if( NO_ERROR != Error ) return Error;
|
|
|
|
//
|
|
// The following lines are for backward compat with NT4.
|
|
//
|
|
|
|
//
|
|
// Create the reservation key in any case
|
|
//
|
|
|
|
Error = DhcpRegGetNextHdl(
|
|
Hdl, REG_SUBNET_RESERVATIONS, &Hdl2 );
|
|
if( NO_ERROR != Error ) return Error;
|
|
|
|
DhcpRegCloseHdl( &Hdl2 );
|
|
|
|
//
|
|
// Create the servers key
|
|
//
|
|
|
|
Error = DhcpRegGetNextHdl(
|
|
Hdl, L"DHCPServers", &Hdl2 );
|
|
if( NO_ERROR != Error ) return Error;
|
|
DhcpRegCloseHdl( &Hdl2 );
|
|
|
|
Error = DhcpRegGetNextNextHdl(
|
|
Hdl, L"DHCPServers", L"127.0.0.1", &Hdl2 );
|
|
if( NO_ERROR != Error ) return Error;
|
|
|
|
//
|
|
// Now set the role of the newly created server as primary
|
|
//
|
|
SrvIpAddress = INADDR_LOOPBACK;
|
|
SrvRole = 1; // primary
|
|
EmptyString = L"";
|
|
{
|
|
WATTRIB_TBL SrvTable[] = {
|
|
&SrvRole, 0, REG_DWORD, L"Role",
|
|
&SrvIpAddress, 0, REG_DWORD, L"ServerIpAddress",
|
|
&EmptyString, 0, REG_SZ, L"ServerHostName",
|
|
&EmptyString, 0, REG_SZ, L"ServerNetBiosName"
|
|
};
|
|
|
|
Error = DhcpRegSaveAttribTable(
|
|
&Hdl2, SrvTable, sizeof(SrvTable)/sizeof(SrvTable[0]));
|
|
}
|
|
|
|
DhcpRegCloseHdl(&Hdl2);
|
|
return Error;
|
|
}
|
|
|
|
//BeginExport(function)
|
|
DWORD
|
|
DhcpRegSubnetSetAttributes(
|
|
IN PREG_HANDLE Hdl,
|
|
IN LPWSTR *Name,
|
|
IN LPWSTR *Comment,
|
|
IN DWORD *Flags,
|
|
IN DWORD *Address,
|
|
IN DWORD *Mask
|
|
) //EndExport(function)
|
|
{
|
|
DWORD xFlags, SwitchedNetwork = FALSE;
|
|
OSVERSIONINFO Ver;
|
|
|
|
Ver.dwOSVersionInfoSize = sizeof(Ver);
|
|
if( FALSE == GetVersionEx(&Ver) ) return GetLastError();
|
|
|
|
if( Flags && Ver.dwMajorVersion < 5 ) {
|
|
SwitchedNetwork = IS_SWITCHED(*Flags);
|
|
xFlags = IS_DISABLED(*Flags);
|
|
Flags = &xFlags;
|
|
}
|
|
|
|
return DhcpRegSubnetSetAttributesInternal(
|
|
Hdl, Name, Comment, Flags, Address, Mask,
|
|
Flags ? &SwitchedNetwork : NULL );
|
|
}
|
|
|
|
//BeginExport(function)
|
|
DWORD
|
|
DhcpRegMScopeSetAttributes(
|
|
IN PREG_HANDLE Hdl,
|
|
IN LPWSTR *Comments,
|
|
IN DWORD *State,
|
|
IN DWORD *ScopeId,
|
|
IN DWORD *Policy,
|
|
IN DWORD *TTL,
|
|
IN LPWSTR *LangTag,
|
|
IN PDATE_TIME *ExpiryTime
|
|
) //EndExport(function)
|
|
{
|
|
WATTRIB_TBL Table[] = {
|
|
Comments, 0, REG_SZ, REG_MSCOPE_COMMENT,
|
|
State, 0, REG_DWORD, REG_MSCOPE_STATE,
|
|
ScopeId, 0, REG_DWORD, REG_MSCOPE_SCOPEID,
|
|
Policy, 0, REG_DWORD, REG_MSCOPE_ADDR_POLICY,
|
|
TTL, 0, REG_DWORD, REG_MSCOPE_TTL,
|
|
LangTag, 0, REG_SZ, REG_MSCOPE_LANG_TAG,
|
|
ExpiryTime, sizeof(**ExpiryTime), REG_BINARY, REG_MSCOPE_EXPIRY_TIME
|
|
};
|
|
return DhcpRegSaveAttribTable(Hdl, Table, sizeof(Table)/sizeof(Table[0]));
|
|
}
|
|
|
|
//BeginExport(function)
|
|
DWORD
|
|
DhcpRegOptDefSetAttributes(
|
|
IN PREG_HANDLE Hdl,
|
|
IN LPWSTR *Name,
|
|
IN LPWSTR *Comments,
|
|
IN DWORD *Flags,
|
|
IN DWORD *OptionId,
|
|
IN LPWSTR *ClassName,
|
|
IN LPWSTR *VendorName,
|
|
IN LPBYTE *Value,
|
|
IN DWORD ValueSize
|
|
) //EndExport(function)
|
|
{
|
|
DWORD Error;
|
|
WATTRIB_TBL Table[] = {
|
|
Name, 0, REG_SZ, REG_OPTION_NAME,
|
|
Comments, 0, REG_SZ, REG_OPTION_COMMENT,
|
|
ClassName, 0, REG_SZ, REG_OPTION_CLASSNAME,
|
|
VendorName, 0, REG_SZ, REG_OPTION_VENDORNAME,
|
|
Flags, 0, REG_DWORD, REG_OPTION_TYPE,
|
|
OptionId, 0, REG_DWORD, REG_OPTION_ID,
|
|
Value, ValueSize,REG_BINARY, REG_OPTION_VALUE
|
|
};
|
|
|
|
return DhcpRegSaveAttribTable(Hdl, Table, sizeof(Table)/sizeof(Table[0]));
|
|
}
|
|
|
|
//BeginExport(function)
|
|
DWORD
|
|
DhcpRegSScopeSetAttributes( // superscopes dont have any information stored.. dont use this
|
|
IN PREG_HANDLE Hdl,
|
|
IN LPWSTR *Name,
|
|
IN LPWSTR *Comment,
|
|
IN DWORD *Flags
|
|
) //EndExport(function)
|
|
{
|
|
Require(FALSE);
|
|
return ERROR_INVALID_PARAMETER;
|
|
}
|
|
|
|
//BeginExport(function)
|
|
DWORD
|
|
DhcpRegClassDefSetAttributes(
|
|
IN PREG_HANDLE Hdl,
|
|
IN LPWSTR *Name,
|
|
IN LPWSTR *Comment,
|
|
IN DWORD *Flags,
|
|
IN LPBYTE *Value,
|
|
IN DWORD ValueSize
|
|
) //EndExport(function)
|
|
{
|
|
DWORD Error;
|
|
WATTRIB_TBL Table[] = {
|
|
Name, 0, REG_SZ, REG_CLASSDEF_NAME,
|
|
Comment, 0, REG_SZ, REG_CLASSDEF_COMMENT,
|
|
Flags, 0, REG_DWORD, REG_CLASSDEF_TYPE,
|
|
Value, ValueSize,REG_BINARY, REG_CLASSDEF_VALUE
|
|
};
|
|
|
|
return DhcpRegSaveAttribTable(Hdl, Table, sizeof(Table)/sizeof(Table[0]));
|
|
}
|
|
|
|
//BeginExport(function)
|
|
DWORD
|
|
DhcpRegSubnetServerSetAttributes(
|
|
IN PREG_HANDLE Hdl,
|
|
IN LPWSTR *Name,
|
|
IN LPWSTR *Comment,
|
|
IN DWORD *Flags,
|
|
IN DWORD *Address,
|
|
IN DWORD *Role
|
|
) //EndExport(function)
|
|
{
|
|
WATTRIB_TBL Table[] = {
|
|
Name, 0, REG_SZ, REG_SUB_SERVER_NAME,
|
|
Comment, 0, REG_SZ, REG_SUB_SERVER_COMMENT,
|
|
Flags, 0, REG_DWORD, REG_FLAGS,
|
|
Address, 0, REG_DWORD, REG_SUB_SERVER_ADDRESS,
|
|
Role, 0, REG_DWORD, REG_SUB_SERVER_ROLE,
|
|
};
|
|
|
|
return DhcpRegSaveAttribTable(Hdl, Table, sizeof(Table)/sizeof(Table[0]));
|
|
}
|
|
|
|
//BeginExport(function)
|
|
DWORD
|
|
DhcpRegRangeSetAttributes(
|
|
IN PREG_HANDLE Hdl,
|
|
IN LPWSTR *Name,
|
|
IN LPWSTR *Comment,
|
|
IN DWORD *Flags,
|
|
IN ULONG *AllocCount,
|
|
IN ULONG *MaxAllocCount,
|
|
IN DWORD *StartAddress,
|
|
IN DWORD *EndAddress,
|
|
IN LPBYTE *InUseClusters,
|
|
IN DWORD InUseClusterSize,
|
|
IN LPBYTE *UsedClusters,
|
|
IN DWORD UsedClustersSize
|
|
) //EndExport(function)
|
|
{
|
|
DWORD Error;
|
|
WATTRIB_TBL Table[] = {
|
|
Name, 0, REG_SZ, REG_RANGE_NAME,
|
|
Comment, 0, REG_SZ, REG_RANGE_COMMENT,
|
|
Flags, 0, REG_DWORD, REG_RANGE_FLAGS,
|
|
AllocCount, 0, REG_DWORD, REG_RANGE_ALLOC,
|
|
MaxAllocCount, 0, REG_DWORD, REG_RANGE_MAX_ALLOC,
|
|
StartAddress, 0, REG_DWORD, REG_RANGE_START_ADDRESS,
|
|
EndAddress, 0, REG_DWORD, REG_RANGE_END_ADDRESS,
|
|
InUseClusters, InUseClusterSize, REG_BINARY, REG_RANGE_INUSE_CLUSTERS,
|
|
UsedClusters, UsedClustersSize, REG_BINARY, REG_RANGE_USED_CLUSTERS,
|
|
};
|
|
|
|
return DhcpRegSaveAttribTable(Hdl, Table, sizeof(Table)/sizeof(Table[0]));
|
|
}
|
|
|
|
//BeginExport(function)
|
|
DWORD
|
|
DhcpRegReservationSetAttributes(
|
|
IN PREG_HANDLE Hdl,
|
|
IN LPWSTR *Name,
|
|
IN LPWSTR *Comment,
|
|
IN DWORD *Flags,
|
|
IN DWORD *Address,
|
|
IN LPBYTE *ClientUID,
|
|
IN DWORD ClientUIDSize
|
|
) //EndExport(function)
|
|
{
|
|
DWORD Error;
|
|
WATTRIB_TBL Table[] = {
|
|
Name, 0, REG_SZ, REG_RESERVATION_NAME,
|
|
Comment, 0, REG_SZ, REG_RESERVATION_COMMENT,
|
|
Flags, 0, REG_DWORD, REG_RESERVATION_TYPE,
|
|
Address, 0, REG_DWORD, REG_RESERVATION_ADDRESS,
|
|
ClientUID, ClientUIDSize, REG_BINARY, REG_RESERVATION_UID,
|
|
};
|
|
|
|
return DhcpRegSaveAttribTable(Hdl, Table, sizeof(Table)/sizeof(Table[0]));
|
|
}
|
|
|
|
//BeginExport(function)
|
|
DWORD
|
|
DhcpRegOptSetAttributes(
|
|
IN PREG_HANDLE Hdl,
|
|
IN DWORD *OptionId,
|
|
IN LPWSTR *ClassName,
|
|
IN LPWSTR *VendorName,
|
|
IN DWORD *Flags,
|
|
IN LPBYTE *Value,
|
|
IN DWORD ValueSize
|
|
) //EndExport(function)
|
|
{
|
|
DWORD Error;
|
|
WATTRIB_TBL Table[] = {
|
|
OptionId, 0, REG_DWORD, REG_OPTION_ID,
|
|
ClassName, 0, REG_SZ, REG_OPTION_CLASSNAME,
|
|
VendorName, 0, REG_SZ, REG_OPTION_VENDORNAME,
|
|
Flags, 0, REG_DWORD, REG_OPTION_TYPE,
|
|
Value, ValueSize,REG_BINARY, REG_OPTION_VALUE,
|
|
};
|
|
|
|
return DhcpRegSaveAttribTable(Hdl, Table, sizeof(Table)/sizeof(Table[0]));
|
|
}
|
|
|
|
//================================================================================
|
|
// recursive deleting of keys...
|
|
//================================================================================
|
|
|
|
//BeginExport(function)
|
|
DWORD
|
|
DhcpRegRecurseDelete(
|
|
IN PREG_HANDLE Hdl,
|
|
IN LPWSTR KeyName
|
|
) //EndExport(function)
|
|
{
|
|
REG_HANDLE Hdl2;
|
|
DWORD Error;
|
|
DWORD LocalError, RetError;
|
|
ARRAY Array; // sub keys
|
|
ARRAY_LOCATION Loc;
|
|
LPWSTR SubKey;
|
|
|
|
RetError = ERROR_SUCCESS;
|
|
|
|
Error = DhcpRegGetNextHdl(Hdl, KeyName, &Hdl2);
|
|
if( ERROR_SUCCESS != Error ) return Error;
|
|
|
|
Error = MemArrayInit(&Array);
|
|
if( ERROR_SUCCESS != Error ) {
|
|
LocalError = DhcpRegCloseHdl(&Hdl2);
|
|
Require(ERROR_SUCCESS == LocalError);
|
|
return Error;
|
|
}
|
|
|
|
Error = DhcpRegFillSubKeys(&Hdl2, &Array);
|
|
Require( ERROR_SUCCESS == Error );
|
|
|
|
Error = MemArrayInitLoc(&Array, &Loc);
|
|
while(ERROR_FILE_NOT_FOUND != Error) {
|
|
Require(ERROR_SUCCESS == Error);
|
|
|
|
Error = MemArrayGetElement(&Array, &Loc, (LPVOID *)&SubKey);
|
|
Require(ERROR_SUCCESS == Error && SubKey);
|
|
|
|
Error = DhcpRegRecurseDelete(&Hdl2, SubKey);
|
|
if( ERROR_SUCCESS != Error ) RetError = Error;
|
|
|
|
if( SubKey ) MemFree(SubKey);
|
|
|
|
Error = MemArrayNextLoc(&Array, &Loc);
|
|
}
|
|
|
|
Error = MemArrayCleanup(&Array);
|
|
Require(ERROR_SUCCESS == Error);
|
|
|
|
Error = DhcpRegCloseHdl(&Hdl2);
|
|
Require(ERROR_SUCCESS == Error);
|
|
|
|
Error = RegDeleteKey(Hdl->Key, KeyName);
|
|
if( ERROR_SUCCESS != Error ) RetError = Error;
|
|
|
|
return RetError;
|
|
}
|
|
|
|
//BeginExport(function)
|
|
DWORD
|
|
DhcpRegRecurseDeleteBunch(
|
|
IN PREG_HANDLE Hdl,
|
|
IN PARRAY KeysArray
|
|
) //EndExport(function)
|
|
{
|
|
ARRAY_LOCATION Loc;
|
|
LPWSTR ThisKeyName;
|
|
DWORD Error;
|
|
|
|
Error = MemArrayInitLoc(KeysArray, &Loc);
|
|
while( ERROR_FILE_NOT_FOUND != Error ) {
|
|
Require(ERROR_SUCCESS == Error);
|
|
|
|
Error = MemArrayGetElement(KeysArray, &Loc, &ThisKeyName);
|
|
Require(ERROR_SUCCESS == Error && NULL != ThisKeyName);
|
|
|
|
Error = DhcpRegRecurseDelete(Hdl, ThisKeyName);
|
|
if( ERROR_SUCCESS != Error ) return Error;
|
|
|
|
Error = MemArrayNextLoc(KeysArray, &Loc);
|
|
}
|
|
|
|
return ERROR_SUCCESS;
|
|
}
|
|
|
|
static
|
|
VOID
|
|
GetLocalFileTime( // fill in filetime struct w/ current local time
|
|
IN OUT LPFILETIME Time // struct to fill in
|
|
)
|
|
{
|
|
BOOL Status;
|
|
SYSTEMTIME SysTime;
|
|
|
|
GetSystemTime(&SysTime); // get sys time as UTC time.
|
|
Status = SystemTimeToFileTime(&SysTime,Time); // conver system time to file time
|
|
if( FALSE == Status ) { // convert failed?
|
|
Time->dwLowDateTime = 0xFFFFFFFF; // set time to weird value in case of failiure
|
|
Time->dwHighDateTime = 0xFFFFFFFF;
|
|
}
|
|
}
|
|
|
|
//BeginExport(function)
|
|
DWORD
|
|
DhcpRegUpdateTime( // update the last modified time
|
|
VOID
|
|
) //EndExport(function)
|
|
{
|
|
FILETIME Time;
|
|
DWORD Err, Size;
|
|
HKEY hKey;
|
|
|
|
GetLocalFileTime(&Time); // first get current time
|
|
(*(LONGLONG *)&Time) += 10*1000*60*2; // 2 minutes (Filetime is in 100-nano seconds)
|
|
// HACK! the previous line is there as the DS takes a little while to update the
|
|
// last changed time..
|
|
Time.dwLowDateTime =Time.dwHighDateTime =0; // set time to "long back" initially
|
|
Err = RegOpenKeyEx // try to open the config key.
|
|
(
|
|
/* hKey */ HKEY_LOCAL_MACHINE,
|
|
/* lpSubKey */ REG_THIS_SERVER,
|
|
/* ulOptions */ 0 /* Reserved */ ,
|
|
/* samDesired */ KEY_ALL_ACCESS,
|
|
/* phkResult */ &hKey
|
|
);
|
|
if( ERROR_SUCCESS != Err ) return Err; // time is still set to ages back
|
|
|
|
Err = RegSetValueEx // now save the time value
|
|
(
|
|
/* hKey */ hKey,
|
|
/* lpValueName */ DHCP_LAST_DOWNLOAD_TIME_VALUE,
|
|
/* lpReserved */ 0,
|
|
/* lpType */ REG_BINARY,
|
|
/* lpData */ (LPBYTE)&Time,
|
|
/* lpcData */ sizeof(Time)
|
|
);
|
|
RegCloseKey(hKey); // close key before we forget
|
|
|
|
return Err;
|
|
}
|
|
|
|
|
|
//================================================================================
|
|
// end of file
|
|
//================================================================================
|
|
|
|
|
|
|