able/windows-nt
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity
windows-nt/Source/XPSP1/NT/net/rras/ip/nathlp/dhcp/rmdhcp.cpp

1177 lines
21 KiB
C++
Raw Normal View History

Add source files
2020-09-26 03:20:57 -05:00
/*++
Copyright (c) 1998, Microsoft Corporation
Module Name:
rmdhcp.c
Abstract:
This module contains routines for the DHCP allocator module's interface
to the IP router-manager. (See ROUTPROT.H for details).
Author:
Abolade Gbadegesin (aboladeg) 4-Mar-1998
Revision History:
--*/
#include "precomp.h"
#pragma hdrstop
COMPONENT_REFERENCE DhcpComponentReference;
PCHAR DhcpDomainName = NULL;
PIP_AUTO_DHCP_GLOBAL_INFO DhcpGlobalInfo;
CRITICAL_SECTION DhcpGlobalInfoLock;
HANDLE DhcpNotificationEvent;
ULONG DhcpProtocolStopped = 0;
const MPR_ROUTING_CHARACTERISTICS DhcpRoutingCharacteristics =
{
MS_ROUTER_VERSION,
MS_IP_DHCP_ALLOCATOR,
RF_ROUTING|RF_ADD_ALL_INTERFACES,
DhcpRmStartProtocol,
DhcpRmStartComplete,
DhcpRmStopProtocol,
DhcpRmGetGlobalInfo,
DhcpRmSetGlobalInfo,
NULL,
NULL,
DhcpRmAddInterface,
DhcpRmDeleteInterface,
DhcpRmInterfaceStatus,
DhcpRmGetInterfaceInfo,
DhcpRmSetInterfaceInfo,
DhcpRmGetEventMessage,
NULL,
NULL,
NULL,
NULL,
NULL,
DhcpRmMibCreate,
DhcpRmMibDelete,
DhcpRmMibGet,
DhcpRmMibSet,
DhcpRmMibGetFirst,
DhcpRmMibGetNext,
NULL,
NULL
};
IP_AUTO_DHCP_STATISTICS DhcpStatistics;
SUPPORT_FUNCTIONS DhcpSupportFunctions;
extern "C"
LPSTR WINAPI
DnsGetPrimaryDomainName_A(
VOID
);
VOID
DhcpCleanupModule(
VOID
)
/*++
Routine Description:
This routine is invoked to cleanup the DHCP module.
Arguments:
none.
Return Value:
none.
Environment:
Invoked from within a 'DllMain' routine on 'DLL_PROCESS_DETACH'.
--*/
{
DeleteCriticalSection(&DhcpGlobalInfoLock);
DhcpShutdownInterfaceManagement();
DeleteComponentReference(&DhcpComponentReference);
} // DhcpCleanupModule
VOID
DhcpCleanupProtocol(
VOID
)
/*++
Routine Description:
This routine is invoked to cleanup the DHCP protocol-component
after a 'StopProtocol'. It runs when the last reference to the
DHCP component is released. (See 'COMPREF.H').
Arguments:
none.
Return Value:
none.
Environment:
Invoked from within an arbitrary context with no locks held.
--*/
{
PROFILE("DhcpCleanupProtocol");
#if 1
if (DhcpDomainName) {
LocalFree(DhcpDomainName);
DhcpDomainName = NULL;
}
#else
if (DhcpDomainName) {
NH_FREE(DhcpDomainName);
DhcpDomainName = NULL;
}
#endif
if (DhcpGlobalInfo) { NH_FREE(DhcpGlobalInfo); DhcpGlobalInfo = NULL; }
InterlockedExchange(reinterpret_cast<LPLONG>(&DhcpProtocolStopped), 1);
SetEvent(DhcpNotificationEvent);
ResetComponentReference(&DhcpComponentReference);
} // DhcpCleanupProtocol
BOOLEAN
DhcpInitializeModule(
VOID
)
/*++
Routine Description:
This routine is invoked to initialize the DHCP module.
Arguments:
none.
Return Value:
BOOLEAN - TRUE if initialization succeeded, FALSE otherwise
Environment:
Invoked in the context of a 'DllMain' routine on 'DLL_PROCESS_ATTACH'.
--*/
{
if (InitializeComponentReference(
&DhcpComponentReference, DhcpCleanupProtocol
)) {
return FALSE;
} else if (DhcpInitializeInterfaceManagement()) {
DeleteComponentReference(&DhcpComponentReference);
return FALSE;
} else {
__try {
InitializeCriticalSection(&DhcpGlobalInfoLock);
} __except(EXCEPTION_EXECUTE_HANDLER) {
DeleteComponentReference(&DhcpComponentReference);
return FALSE;
}
}
return TRUE;
} // DhcpInitializeModule
ULONG
APIENTRY
DhcpRmStartProtocol(
HANDLE NotificationEvent,
PSUPPORT_FUNCTIONS SupportFunctions,
PVOID GlobalInfo,
ULONG StructureVersion,
ULONG StructureSize,
ULONG StructureCount
)
/*++
Routine Description:
This routine is invoked to indicate the component's operation should begin.
Arguments:
NotificationEvent - event on which we notify the router-manager
about asynchronous occurrences
SupportFunctions - functions for initiating router-related operations
GlobalInfo - configuration for the component
Return Value:
ULONG - Win32 status code.
Environment:
The routine runs in the context of an IP router-manager thread.
--*/
{
ULONG Error = NO_ERROR;
ULONG Size;
PROFILE("DhcpRmStartProtocol");
REFERENCE_DHCP_OR_RETURN(ERROR_CAN_NOT_COMPLETE);
if (!GlobalInfo) { DEREFERENCE_DHCP_AND_RETURN(ERROR_INVALID_PARAMETER); }
do {
//
// Create a copy of the global configuration
//
Size =
sizeof(*DhcpGlobalInfo) +
((PIP_AUTO_DHCP_GLOBAL_INFO)GlobalInfo)->ExclusionCount *
sizeof(ULONG);
DhcpGlobalInfo =
reinterpret_cast<PIP_AUTO_DHCP_GLOBAL_INFO>(NH_ALLOCATE(Size));
if (!DhcpGlobalInfo) {
NhTrace(
TRACE_FLAG_INIT,
"DhcpRmStartProtocol: cannot allocate global info"
);
NhErrorLog(
IP_AUTO_DHCP_LOG_ALLOCATION_FAILED,
0,
"%d",
Size
);
Error = ERROR_NOT_ENOUGH_MEMORY;
break;
}
CopyMemory(DhcpGlobalInfo, GlobalInfo, Size);
//
// Save the notification event
//
DhcpNotificationEvent = NotificationEvent;
//
// Save the support functions
//
if (!SupportFunctions) {
ZeroMemory(&DhcpSupportFunctions, sizeof(DhcpSupportFunctions));
}
else {
CopyMemory(
&DhcpSupportFunctions,
SupportFunctions,
sizeof(*SupportFunctions)
);
}
InterlockedExchange(reinterpret_cast<LPLONG>(&DhcpProtocolStopped), 0);
} while (FALSE);
DEREFERENCE_DHCP_AND_RETURN(Error);
} // DhcpRmStartProtocol
ULONG
APIENTRY
DhcpRmStartComplete(
VOID
)
/*++
Routine Description:
This routine is invoked when the router has finished adding the initial
configuration
Arguments:
none.
Return Value:
ULONG - Win32 status code
Environment:
The routine runs in the context of an IP router-manager thread.
--*/
{
return NO_ERROR;
} // DhcpRmStartComplete
ULONG
APIENTRY
DhcpRmStopProtocol(
VOID
)
/*++
Routine Description:
This routine is invoked to stop the protocol.
Arguments:
none.
Return Value:
ULONG - Win32 status code
Environment:
The routine runs in the context of an IP router-manager thread.
--*/
{
PROFILE("DhcpStopProtocol");
//
// Reference the module to make sure it's running
//
REFERENCE_DHCP_OR_RETURN(ERROR_CAN_NOT_COMPLETE);
//
// Drop the initial reference to cause a cleanup
//
ReleaseInitialComponentReference(&DhcpComponentReference);
return DEREFERENCE_DHCP() ? NO_ERROR : ERROR_PROTOCOL_STOP_PENDING;
} // DhcpRmStopProtocol
ULONG
APIENTRY
DhcpRmAddInterface(
PWCHAR Name,
ULONG Index,
NET_INTERFACE_TYPE Type,
ULONG MediaType,
USHORT AccessType,
USHORT ConnectionType,
PVOID InterfaceInfo,
ULONG StructureVersion,
ULONG StructureSize,
ULONG StructureCount
)
/*++
Routine Description:
This routine is invoked to add an interface to the component.
Arguments:
Name - the name of the interface (unused)
Index - the index of the interface
Type - the type of the interface
InterfaceInfo - the configuration information for the interface
Return Value:
ULONG - Win32 status code.
Environment:
The routine runs in the context of an IP router-manager thread.
--*/
{
ULONG Error;
PROFILE("DhcpRmAddInterface");
if (Type != PERMANENT) { return NO_ERROR; }
REFERENCE_DHCP_OR_RETURN(ERROR_CAN_NOT_COMPLETE);
Error =
DhcpCreateInterface(
Index,
Type,
(PIP_AUTO_DHCP_INTERFACE_INFO)InterfaceInfo,
NULL
);
DEREFERENCE_DHCP_AND_RETURN(Error);
} // DhcpRmAddInterface
ULONG
APIENTRY
DhcpRmDeleteInterface(
ULONG Index
)
/*++
Routine Description:
This routine is invoked to delete an interface from the component.
Arguments:
Index - the index of the interface
Return Value:
ULONG - Win32 status code
Environment:
The routine runs in the context of an IP router-manager thread.
--*/
{
ULONG Error;
PROFILE("DhcpRmDeleteInterface");
REFERENCE_DHCP_OR_RETURN(ERROR_CAN_NOT_COMPLETE);
Error =
DhcpDeleteInterface(
Index
);
DEREFERENCE_DHCP_AND_RETURN(Error);
} // DhcpRmDeleteInterface
ULONG
APIENTRY
DhcpRmGetEventMessage(
OUT ROUTING_PROTOCOL_EVENTS* Event,
OUT MESSAGE* Result
)
/*++
Routine Description:
This routine is invoked to retrieve an event message from the component.
The only event message we generate is the 'ROUTER_STOPPED' message.
Arguments:
Event - receives the generated event
Result - receives the associated result
Return Value:
ULONG - Win32 status code.
--*/
{
PROFILE("DhcpRmGetEventMessage");
if (InterlockedExchange(reinterpret_cast<LPLONG>(&DhcpProtocolStopped), 0)) {
*Event = ROUTER_STOPPED;
return NO_ERROR;
}
return ERROR_NO_MORE_ITEMS;
} // DhcpRmGetEventMessage
ULONG
APIENTRY
DhcpRmGetInterfaceInfo(
ULONG Index,
PVOID InterfaceInfo,
IN OUT PULONG InterfaceInfoSize,
IN OUT PULONG StructureVersion,
IN OUT PULONG StructureSize,
IN OUT PULONG StructureCount
)
/*++
Routine Description:
This routine is invoked to retrieve the component's per-interface
configuration.
Arguments:
Index - the index of the interface to be queried
InterfaceInfo - receives the query results
InterfaceInfoSize - receives the amount of data retrieved
Return Value:
ULONG - Win32 status code.
--*/
{
ULONG Error;
PROFILE("DhcpRmGetInterfaceInfo");
REFERENCE_DHCP_OR_RETURN(ERROR_CAN_NOT_COMPLETE);
Error =
DhcpQueryInterface(
Index,
(PIP_AUTO_DHCP_INTERFACE_INFO)InterfaceInfo,
InterfaceInfoSize
);
*StructureSize = *InterfaceInfoSize;
if (StructureCount) {*StructureCount = 1;}
DEREFERENCE_DHCP_AND_RETURN(Error);
} // DhcpRmGetInterfaceInfo
ULONG
APIENTRY
DhcpRmSetInterfaceInfo(
ULONG Index,
PVOID InterfaceInfo,
ULONG StructureVersion,
ULONG StructureSize,
ULONG StructureCount
)
/*++
Routine Description:
This routine is invoked to change the component's per-interface
configuration.
Arguments:
Index - the index of the interface to be updated
InterfaceInfo - supplies the new configuration
Return Value:
ULONG - Win32 status code.
--*/
{
ULONG Error;
PROFILE("DhcpRmSetInterfaceInfo");
REFERENCE_DHCP_OR_RETURN(ERROR_CAN_NOT_COMPLETE);
Error =
DhcpConfigureInterface(
Index,
(PIP_AUTO_DHCP_INTERFACE_INFO)InterfaceInfo
);
DEREFERENCE_DHCP_AND_RETURN(Error);
} // DhcpRmSetInterfaceInfo
ULONG
APIENTRY
DhcpRmInterfaceStatus(
ULONG Index,
BOOL InterfaceActive,
ULONG StatusType,
PVOID StatusInfo
)
/*++
Routine Description:
This routine is invoked to bind/unbind, enable/disable an interface
Arguments:
Index - the interface to be bound
InterfaceActive - whether the interface is active
StatusType - type of status being changed (bind or enabled)
StatusInfo - Info pertaining to the state being changed
Return Value:
ULONG - Win32 Status code
Environment:
The routine runs in the context of an IP router-manager thread.
--*/
{
ULONG Error = NO_ERROR;
switch(StatusType) {
case RIS_INTERFACE_ADDRESS_CHANGE: {
PIP_ADAPTER_BINDING_INFO BindInfo =
(PIP_ADAPTER_BINDING_INFO)StatusInfo;
if (BindInfo->AddressCount) {
Error = DhcpRmBindInterface(Index, StatusInfo);
} else {
Error = DhcpRmUnbindInterface(Index);
}
break;
}
case RIS_INTERFACE_ENABLED: {
Error = DhcpRmEnableInterface(Index);
break;
}
case RIS_INTERFACE_DISABLED: {
Error = DhcpRmDisableInterface(Index);
break;
}
}
return Error;
} // DhcpRmInterfaceStatus
ULONG
DhcpRmBindInterface(
ULONG Index,
PVOID BindingInfo
)
/*++
Routine Description:
This routine is invoked to bind an interface to its IP address(es).
Arguments:
Index - the interface to be bound
BindingInfo - the addressing information
Return Value:
ULONG - Win32 status code.
Environment:
The routine runs in the context of an IP router-manager thread.
--*/
{
PCHAR DomainName;
ULONG Error;
NTSTATUS status;
PROFILE("DhcpRmBindInterface");
REFERENCE_DHCP_OR_RETURN(ERROR_CAN_NOT_COMPLETE);
Error =
DhcpBindInterface(
Index,
(PIP_ADAPTER_BINDING_INFO)BindingInfo
);
//
// Re-read the domain name in case it changed
//
EnterCriticalSection(&DhcpGlobalInfoLock);
#if 1
DomainName = DnsGetPrimaryDomainName_A();
if (DomainName) {
if (DhcpDomainName && lstrcmpiA(DomainName, DhcpDomainName) == 0) {
LocalFree(DomainName);
} else {
if (DhcpDomainName) { LocalFree(DhcpDomainName); }
DhcpDomainName = DomainName;
}
}
#else
status = NhQueryDomainName(&DomainName);
if (NT_SUCCESS(status)) {
if (DhcpDomainName && lstrcmpiA(DomainName, DhcpDomainName) == 0) {
NH_FREE(DomainName);
} else {
NH_FREE(DhcpDomainName);
if (DhcpDomainName) { NH_FREE(DhcpDomainName); }
DhcpDomainName = DomainName;
}
}
#endif
LeaveCriticalSection(&DhcpGlobalInfoLock);
DEREFERENCE_DHCP_AND_RETURN(Error);
} // DhcpRmBindInterface
ULONG
DhcpRmUnbindInterface(
ULONG Index
)
/*++
Routine Description:
This routine is invoked to unbind an interface from its IP address(es).
Arguments:
Index - the interface to be unbound
Return Value:
ULONG - Win32 status code.
Environment:
The routine runs in the context of an IP router-manager thread.
--*/
{
ULONG Error;
PROFILE("DhcpRmUnbindInterface");
REFERENCE_DHCP_OR_RETURN(ERROR_CAN_NOT_COMPLETE);
Error =
DhcpUnbindInterface(
Index
);
DEREFERENCE_DHCP_AND_RETURN(Error);
} // DhcpRmUnbindInterface
ULONG
DhcpRmEnableInterface(
ULONG Index
)
/*++
Routine Description:
This routine is invoked to enable operation on an interface.
Arguments:
Index - the interface to be enabled.
Return Value:
ULONG - Win32 status code.
Environment:
The routine runs in the context of an IP router-manager thread.
--*/
{
ULONG Error;
PROFILE("DhcpRmEnableInterface");
REFERENCE_DHCP_OR_RETURN(ERROR_CAN_NOT_COMPLETE);
Error =
DhcpEnableInterface(
Index
);
DEREFERENCE_DHCP_AND_RETURN(Error);
} // DhcpRmEnableInterface
ULONG
DhcpRmDisableInterface(
ULONG Index
)
/*++
Routine Description:
This routine is invoked to disable operation on an interface.
Arguments:
Index - the interface to be disabled.
Return Value:
ULONG - Win32 status code.
Environment:
The routine runs in the context of an IP router-manager thread.
--*/
{
ULONG Error;
PROFILE("DhcpRmDisableInterface");
REFERENCE_DHCP_OR_RETURN(ERROR_CAN_NOT_COMPLETE);
Error =
DhcpDisableInterface(
Index
);
DEREFERENCE_DHCP_AND_RETURN(Error);
} // DhcpRmDisableInterface
ULONG
APIENTRY
DhcpRmGetGlobalInfo(
PVOID GlobalInfo,
IN OUT PULONG GlobalInfoSize,
PULONG StructureVersion,
IN OUT PULONG StructureSize,
IN OUT PULONG StructureCount
)
/*++
Routine Description:
This routine is invoked to retrieve the configuration for the component.
Arguments:
GlobalInfo - receives the configuration
GlobalInfoSize - receives the size of the configuration
Return Value:
ULONG - Win32 status code
Environment:
The routine runs in the context of an IP router-manager thread.
--*/
{
ULONG Size;
PROFILE("DhcpRmGetGlobalInfo");
REFERENCE_DHCP_OR_RETURN(ERROR_CAN_NOT_COMPLETE);
if (!GlobalInfoSize || (*GlobalInfoSize && !GlobalInfo)) {
DEREFERENCE_DHCP_AND_RETURN(ERROR_INVALID_PARAMETER);
}
EnterCriticalSection(&DhcpGlobalInfoLock);
Size =
sizeof(*DhcpGlobalInfo) +
DhcpGlobalInfo->ExclusionCount * sizeof(ULONG);
if (*GlobalInfoSize < Size) {
LeaveCriticalSection(&DhcpGlobalInfoLock);
*StructureSize = *GlobalInfoSize = Size;
if (StructureCount) {*StructureCount = 1;}
DEREFERENCE_DHCP_AND_RETURN(ERROR_INSUFFICIENT_BUFFER);
}
CopyMemory(GlobalInfo, DhcpGlobalInfo, Size);
LeaveCriticalSection(&DhcpGlobalInfoLock);
*StructureSize = *GlobalInfoSize = Size;
if (StructureCount) {*StructureCount = 1;}
DEREFERENCE_DHCP_AND_RETURN(NO_ERROR);
} // DhcpRmGetGlobalInfo
ULONG
APIENTRY
DhcpRmSetGlobalInfo(
PVOID GlobalInfo,
ULONG StructureVersion,
ULONG StructureSize,
ULONG StructureCount
)
/*++
Routine Description:
This routine is invoked to change the configuration for the component.
Arguments:
GlobalInfo - the new configuration
Return Value:
ULONG - Win32 status code
Environment:
The routine runs in the context of an IP router-manager thread.
--*/
{
PIP_AUTO_DHCP_GLOBAL_INFO NewInfo;
ULONG NewScope;
ULONG OldScope;
ULONG Size;
PROFILE("DhcpRmSetGlobalInfo");
REFERENCE_DHCP_OR_RETURN(ERROR_CAN_NOT_COMPLETE);
if (!GlobalInfo) { DEREFERENCE_DHCP_AND_RETURN(ERROR_INVALID_PARAMETER); }
Size =
sizeof(*DhcpGlobalInfo) +
((PIP_AUTO_DHCP_GLOBAL_INFO)GlobalInfo)->ExclusionCount * sizeof(ULONG);
NewInfo = reinterpret_cast<PIP_AUTO_DHCP_GLOBAL_INFO>(NH_ALLOCATE(Size));
if (!NewInfo) {
NhTrace(
TRACE_FLAG_INIT,
"DhcpRmSetGlobalInfo: error reallocating global info"
);
NhErrorLog(
IP_AUTO_DHCP_LOG_ALLOCATION_FAILED,
0,
"%d",
Size
);
DEREFERENCE_DHCP_AND_RETURN(ERROR_NOT_ENOUGH_MEMORY);
}
CopyMemory(NewInfo, GlobalInfo, Size);
EnterCriticalSection(&DhcpGlobalInfoLock);
OldScope = DhcpGlobalInfo->ScopeNetwork & DhcpGlobalInfo->ScopeMask;
NH_FREE(DhcpGlobalInfo);
DhcpGlobalInfo = NewInfo;
NewScope = DhcpGlobalInfo->ScopeNetwork & DhcpGlobalInfo->ScopeMask;
LeaveCriticalSection(&DhcpGlobalInfoLock);
if (OldScope != NewScope) {
DhcpReactivateEveryInterface();
}
DEREFERENCE_DHCP_AND_RETURN(NO_ERROR);
} // DhcpRmSetGlobalInfo
ULONG
APIENTRY
DhcpRmMibCreate(
ULONG InputDataSize,
PVOID InputData
)
{
return ERROR_NOT_SUPPORTED;
}
ULONG
APIENTRY
DhcpRmMibDelete(
ULONG InputDataSize,
PVOID InputData
)
{
return ERROR_NOT_SUPPORTED;
}
ULONG
APIENTRY
DhcpRmMibGet(
ULONG InputDataSize,
PVOID InputData,
OUT PULONG OutputDataSize,
OUT PVOID OutputData
)
/*++
Routine Description:
The DHCP allocator only exposes one item to the MIB; its statistics.
Arguments:
InputDataSize - the MIB query data size
InputData - specifies the MIB object to be retrieved
OutputDataSize - the MIB response data size
OutputData - receives the MIB object retrieved
Return Value:
ULONG - Win32 status code.
--*/
{
ULONG Error;
PIP_AUTO_DHCP_MIB_QUERY Oidp;
PROFILE("DhcpRmMibGet");
REFERENCE_DHCP_OR_RETURN(ERROR_CAN_NOT_COMPLETE);
if (InputDataSize < sizeof(*Oidp) || !OutputDataSize) {
Error = ERROR_INVALID_PARAMETER;
}
else {
Oidp = (PIP_AUTO_DHCP_MIB_QUERY)InputData;
switch(Oidp->Oid) {
case IP_AUTO_DHCP_STATISTICS_OID: {
if (*OutputDataSize < sizeof(*Oidp) + sizeof(DhcpStatistics)) {
*OutputDataSize = sizeof(*Oidp) + sizeof(DhcpStatistics);
Error = ERROR_INSUFFICIENT_BUFFER;
}
else {
*OutputDataSize = sizeof(*Oidp) + sizeof(DhcpStatistics);
Oidp = (PIP_AUTO_DHCP_MIB_QUERY)OutputData;
Oidp->Oid = IP_AUTO_DHCP_STATISTICS_OID;
CopyMemory(
Oidp->Data,
&DhcpStatistics,
sizeof(DhcpStatistics)
);
Error = NO_ERROR;
}
break;
}
default: {
NhTrace(
TRACE_FLAG_DHCP,
"DhcpRmMibGet: oid %d invalid",
Oidp->Oid
);
Error = ERROR_INVALID_PARAMETER;
break;
}
}
}
DEREFERENCE_DHCP_AND_RETURN(Error);
}
ULONG
APIENTRY
DhcpRmMibSet(
ULONG InputDataSize,
PVOID InputData
)
{
return ERROR_NOT_SUPPORTED;
}
ULONG
APIENTRY
DhcpRmMibGetFirst(
ULONG InputDataSize,
PVOID InputData,
OUT PULONG OutputDataSize,
OUT PVOID OutputData
)
{
return ERROR_NOT_SUPPORTED;
}
ULONG
APIENTRY
DhcpRmMibGetNext(
ULONG InputDataSize,
PVOID InputData,
OUT PULONG OutputDataSize,
OUT PVOID OutputData
)
{
return ERROR_NOT_SUPPORTED;
}
Reference in a new issue Copy permalink