2286 lines
50 KiB
C++
2286 lines
50 KiB
C++
|
/*++
|
|||
|
|
|||
|
Copyright (c) 1998, Microsoft Corporation
|
|||
|
|
|||
|
Module Name:
|
|||
|
|
|||
|
dnsif.c
|
|||
|
|
|||
|
Abstract:
|
|||
|
|
|||
|
This module contains code for the DNS proxy's interface management.
|
|||
|
|
|||
|
Author:
|
|||
|
|
|||
|
Abolade Gbadegesin (aboladeg) 9-Mar-1998
|
|||
|
|
|||
|
Revision History:
|
|||
|
|
|||
|
--*/
|
|||
|
|
|||
|
#include "precomp.h"
|
|||
|
#pragma hdrstop
|
|||
|
|
|||
|
//
|
|||
|
// LOCAL TYPE DECLARATIONS
|
|||
|
//
|
|||
|
|
|||
|
typedef struct _DNS_DEFER_READ_CONTEXT {
|
|||
|
ULONG Index;
|
|||
|
SOCKET Socket;
|
|||
|
ULONG DeferralCount;
|
|||
|
} DNS_DEFER_READ_CONTEXT, *PDNS_DEFER_READ_CONTEXT;
|
|||
|
|
|||
|
#define DNS_DEFER_READ_INITIAL_TIMEOUT (1 * 1000)
|
|||
|
#define DNS_DEFER_READ_TIMEOUT (5 * 1000)
|
|||
|
#define DNS_CONNECT_TIMEOUT (60 * 1000)
|
|||
|
|
|||
|
//
|
|||
|
// GLOBAL DATA DEFINITIONS
|
|||
|
//
|
|||
|
|
|||
|
LIST_ENTRY DnsInterfaceList;
|
|||
|
CRITICAL_SECTION DnsInterfaceLock;
|
|||
|
ULONG DnspLastConnectAttemptTickCount;
|
|||
|
|
|||
|
//
|
|||
|
// Forward declarations
|
|||
|
//
|
|||
|
|
|||
|
VOID NTAPI
|
|||
|
DnspDeferReadCallbackRoutine(
|
|||
|
PVOID Context,
|
|||
|
BOOLEAN TimedOut
|
|||
|
);
|
|||
|
|
|||
|
VOID APIENTRY
|
|||
|
DnspDeferReadWorkerRoutine(
|
|||
|
PVOID Context
|
|||
|
);
|
|||
|
|
|||
|
ULONG NTAPI
|
|||
|
DnspSaveFileWorkerRoutine(
|
|||
|
PVOID Context
|
|||
|
);
|
|||
|
|
|||
|
|
|||
|
ULONG
|
|||
|
DnsActivateInterface(
|
|||
|
PDNS_INTERFACE Interfacep
|
|||
|
)
|
|||
|
|
|||
|
/*++
|
|||
|
|
|||
|
Routine Description:
|
|||
|
|
|||
|
This routine is called to activate an interface, when the interface
|
|||
|
becomes both enabled and bound.
|
|||
|
Activation involves
|
|||
|
(a) creating sockets for each binding of the interface
|
|||
|
(b) initiating datagram-reads on each created socket
|
|||
|
|
|||
|
Arguments:
|
|||
|
|
|||
|
Interfacep - the interface to be activated
|
|||
|
|
|||
|
Return Value:
|
|||
|
|
|||
|
ULONG - Win32 status code indicating success or failure.
|
|||
|
|
|||
|
Environment:
|
|||
|
|
|||
|
Always invoked locally, with 'Interfacep' referenced by caller and/or
|
|||
|
'DnsInterfaceLock' held by caller.
|
|||
|
|
|||
|
--*/
|
|||
|
|
|||
|
{
|
|||
|
BOOLEAN EnableDns;
|
|||
|
BOOLEAN EnableWins = FALSE;
|
|||
|
ULONG Error;
|
|||
|
ULONG i;
|
|||
|
BOOLEAN IsNatInterface;
|
|||
|
|
|||
|
PROFILE("DnsActivateInterface");
|
|||
|
|
|||
|
EnterCriticalSection(&DnsGlobalInfoLock);
|
|||
|
EnableDns =
|
|||
|
(DnsGlobalInfo->Flags & IP_DNS_PROXY_FLAG_ENABLE_DNS) ? TRUE : FALSE;
|
|||
|
LeaveCriticalSection(&DnsGlobalInfoLock);
|
|||
|
|
|||
|
//
|
|||
|
// (re)take the interface lock for the duration of the routine
|
|||
|
//
|
|||
|
|
|||
|
EnterCriticalSection(&DnsInterfaceLock);
|
|||
|
if (!(EnableDns || EnableWins) ||
|
|||
|
DNS_INTERFACE_ADMIN_DISABLED(Interfacep)) {
|
|||
|
LeaveCriticalSection(&DnsInterfaceLock);
|
|||
|
return NO_ERROR;
|
|||
|
}
|
|||
|
|
|||
|
if (NhIsBoundaryInterface(Interfacep->Index, &IsNatInterface)) {
|
|||
|
NhTrace(
|
|||
|
TRACE_FLAG_DNS,
|
|||
|
"DnsActivateInterface: ignoring NAT interface %d",
|
|||
|
Interfacep->Index
|
|||
|
);
|
|||
|
NhWarningLog(
|
|||
|
IP_DNS_PROXY_LOG_NAT_INTERFACE_IGNORED,
|
|||
|
0,
|
|||
|
"%d",
|
|||
|
Interfacep->Index
|
|||
|
);
|
|||
|
LeaveCriticalSection(&DnsInterfaceLock);
|
|||
|
return NO_ERROR;
|
|||
|
}
|
|||
|
|
|||
|
if (!IsNatInterface) {
|
|||
|
NhTrace(
|
|||
|
TRACE_FLAG_DNS,
|
|||
|
"DnsActivateInterface: ignoring non-NAT interface %d",
|
|||
|
Interfacep->Index
|
|||
|
);
|
|||
|
LeaveCriticalSection(&DnsInterfaceLock);
|
|||
|
return NO_ERROR;
|
|||
|
}
|
|||
|
|
|||
|
//
|
|||
|
// Create datagram sockets for receiving data on each logical network
|
|||
|
//
|
|||
|
|
|||
|
Error = NO_ERROR;
|
|||
|
|
|||
|
ACQUIRE_LOCK(Interfacep);
|
|||
|
|
|||
|
for (i = 0; i < Interfacep->BindingCount; i++) {
|
|||
|
|
|||
|
if (EnableDns) {
|
|||
|
Error =
|
|||
|
NhCreateDatagramSocket(
|
|||
|
Interfacep->BindingArray[i].Address,
|
|||
|
DNS_PORT_SERVER,
|
|||
|
&Interfacep->BindingArray[i].Socket[DnsProxyDns]
|
|||
|
);
|
|||
|
if (Error) { break; }
|
|||
|
}
|
|||
|
|
|||
|
if (EnableWins) {
|
|||
|
Error =
|
|||
|
NhCreateDatagramSocket(
|
|||
|
Interfacep->BindingArray[i].Address,
|
|||
|
WINS_PORT_SERVER,
|
|||
|
&Interfacep->BindingArray[i].Socket[DnsProxyWins]
|
|||
|
);
|
|||
|
if (Error) { break; }
|
|||
|
}
|
|||
|
}
|
|||
|
|
|||
|
//
|
|||
|
// If an error occurred, roll back all work done so far and fail.
|
|||
|
//
|
|||
|
|
|||
|
if (Error) {
|
|||
|
ULONG FailedAddress = i;
|
|||
|
for (; (LONG)i >= 0; i--) {
|
|||
|
NhDeleteDatagramSocket(
|
|||
|
Interfacep->BindingArray[i].Socket[DnsProxyDns]
|
|||
|
);
|
|||
|
Interfacep->BindingArray[i].Socket[DnsProxyDns] = INVALID_SOCKET;
|
|||
|
NhDeleteDatagramSocket(
|
|||
|
Interfacep->BindingArray[i].Socket[DnsProxyWins]
|
|||
|
);
|
|||
|
Interfacep->BindingArray[i].Socket[DnsProxyWins] = INVALID_SOCKET;
|
|||
|
}
|
|||
|
NhErrorLog(
|
|||
|
IP_DNS_PROXY_LOG_ACTIVATE_FAILED,
|
|||
|
Error,
|
|||
|
"%I",
|
|||
|
Interfacep->BindingArray[FailedAddress].Address
|
|||
|
);
|
|||
|
RELEASE_LOCK(Interfacep);
|
|||
|
LeaveCriticalSection(&DnsInterfaceLock);
|
|||
|
return Error;
|
|||
|
}
|
|||
|
|
|||
|
if (EnableWins && DNS_REFERENCE_INTERFACE(Interfacep)) {
|
|||
|
Error =
|
|||
|
NhReadDatagramSocket(
|
|||
|
&DnsComponentReference,
|
|||
|
DnsGlobalSocket,
|
|||
|
NULL,
|
|||
|
DnsReadCompletionRoutine,
|
|||
|
Interfacep,
|
|||
|
NULL
|
|||
|
);
|
|||
|
if (Error) { DNS_DEREFERENCE_INTERFACE(Interfacep); }
|
|||
|
}
|
|||
|
|
|||
|
//
|
|||
|
// Initiate read-operations on each socket
|
|||
|
//
|
|||
|
|
|||
|
for (i = 0; i < Interfacep->BindingCount; i++) {
|
|||
|
|
|||
|
if (EnableDns) {
|
|||
|
|
|||
|
//
|
|||
|
// Make a reference to the interface;
|
|||
|
// this reference is released in the completion routine
|
|||
|
//
|
|||
|
|
|||
|
if (!DNS_REFERENCE_INTERFACE(Interfacep)) { continue; }
|
|||
|
|
|||
|
//
|
|||
|
// Initiate the read-operation
|
|||
|
//
|
|||
|
|
|||
|
Error =
|
|||
|
NhReadDatagramSocket(
|
|||
|
&DnsComponentReference,
|
|||
|
Interfacep->BindingArray[i].Socket[DnsProxyDns],
|
|||
|
NULL,
|
|||
|
DnsReadCompletionRoutine,
|
|||
|
Interfacep,
|
|||
|
UlongToPtr(Interfacep->BindingArray[i].Address)
|
|||
|
);
|
|||
|
|
|||
|
//
|
|||
|
// Drop the reference if a failure occurred
|
|||
|
//
|
|||
|
|
|||
|
if (Error) {
|
|||
|
|
|||
|
NhErrorLog(
|
|||
|
IP_DNS_PROXY_LOG_RECEIVE_FAILED,
|
|||
|
Error,
|
|||
|
"%I",
|
|||
|
Interfacep->BindingArray[i].Address
|
|||
|
);
|
|||
|
|
|||
|
DNS_DEREFERENCE_INTERFACE(Interfacep);
|
|||
|
|
|||
|
//
|
|||
|
// Reissue the read-operation later
|
|||
|
//
|
|||
|
|
|||
|
DnsDeferReadInterface(
|
|||
|
Interfacep,
|
|||
|
Interfacep->BindingArray[i].Socket[DnsProxyDns]
|
|||
|
);
|
|||
|
|
|||
|
Error = NO_ERROR;
|
|||
|
}
|
|||
|
}
|
|||
|
|
|||
|
if (EnableWins) {
|
|||
|
|
|||
|
//
|
|||
|
// Reference the interface for the WINS socket receive
|
|||
|
//
|
|||
|
|
|||
|
if (!DNS_REFERENCE_INTERFACE(Interfacep)) { continue; }
|
|||
|
|
|||
|
//
|
|||
|
// Initiate the read-operation
|
|||
|
//
|
|||
|
|
|||
|
Error =
|
|||
|
NhReadDatagramSocket(
|
|||
|
&DnsComponentReference,
|
|||
|
Interfacep->BindingArray[i].Socket[DnsProxyWins],
|
|||
|
NULL,
|
|||
|
DnsReadCompletionRoutine,
|
|||
|
Interfacep,
|
|||
|
UlongToPtr(Interfacep->BindingArray[i].Address)
|
|||
|
);
|
|||
|
|
|||
|
//
|
|||
|
// Drop the reference if a failure occurred
|
|||
|
//
|
|||
|
|
|||
|
if (Error) {
|
|||
|
|
|||
|
NhErrorLog(
|
|||
|
IP_DNS_PROXY_LOG_RECEIVE_FAILED,
|
|||
|
Error,
|
|||
|
"%I",
|
|||
|
Interfacep->BindingArray[i].Address
|
|||
|
);
|
|||
|
|
|||
|
DNS_DEREFERENCE_INTERFACE(Interfacep);
|
|||
|
|
|||
|
//
|
|||
|
// Reissue the read-operation later
|
|||
|
//
|
|||
|
|
|||
|
DnsDeferReadInterface(
|
|||
|
Interfacep,
|
|||
|
Interfacep->BindingArray[i].Socket[DnsProxyWins]
|
|||
|
);
|
|||
|
|
|||
|
Error = NO_ERROR;
|
|||
|
}
|
|||
|
}
|
|||
|
}
|
|||
|
|
|||
|
RELEASE_LOCK(Interfacep);
|
|||
|
LeaveCriticalSection(&DnsInterfaceLock);
|
|||
|
|
|||
|
//
|
|||
|
// queue a write to disk (ip address may have changed)
|
|||
|
// (necessary to do this to prevent possible deadlock)
|
|||
|
//
|
|||
|
if (REFERENCE_DNS())
|
|||
|
{
|
|||
|
if (!QueueUserWorkItem(DnspSaveFileWorkerRoutine, NULL, WT_EXECUTEDEFAULT))
|
|||
|
{
|
|||
|
Error = GetLastError();
|
|||
|
NhTrace(
|
|||
|
TRACE_FLAG_DNS,
|
|||
|
"DnsActivateInterface: QueueUserWorkItem failed with error %d (0x%08x)",
|
|||
|
Error,
|
|||
|
Error
|
|||
|
);
|
|||
|
DEREFERENCE_DNS();
|
|||
|
}
|
|||
|
else
|
|||
|
{
|
|||
|
NhTrace(
|
|||
|
TRACE_FLAG_DNS,
|
|||
|
"DnsActivateInterface: queued a write of %s file",
|
|||
|
HOSTSICSFILE
|
|||
|
);
|
|||
|
}
|
|||
|
}
|
|||
|
|
|||
|
return NO_ERROR;
|
|||
|
|
|||
|
} // DnsActivateInterface
|
|||
|
|
|||
|
|
|||
|
ULONG
|
|||
|
DnsBindInterface(
|
|||
|
ULONG Index,
|
|||
|
PIP_ADAPTER_BINDING_INFO BindingInfo
|
|||
|
)
|
|||
|
|
|||
|
/*++
|
|||
|
|
|||
|
Routine Description:
|
|||
|
|
|||
|
This routine is invoked to supply the binding for an interface.
|
|||
|
It records the binding information received, and if necessary,
|
|||
|
it activates the interface.
|
|||
|
|
|||
|
Arguments:
|
|||
|
|
|||
|
Index - the index of the interface to be bound
|
|||
|
|
|||
|
BindingInfo - the binding-information for the interface
|
|||
|
|
|||
|
Return Value:
|
|||
|
|
|||
|
ULONG - Win32 status code.
|
|||
|
|
|||
|
Environment:
|
|||
|
|
|||
|
Invoked internally in the context of an IP router-manager thread.
|
|||
|
(See 'RMDNS.C').
|
|||
|
|
|||
|
--*/
|
|||
|
|
|||
|
{
|
|||
|
ULONG Error = NO_ERROR;
|
|||
|
ULONG i;
|
|||
|
PDNS_INTERFACE Interfacep;
|
|||
|
|
|||
|
PROFILE("DnsBindInterface");
|
|||
|
|
|||
|
EnterCriticalSection(&DnsInterfaceLock);
|
|||
|
|
|||
|
//
|
|||
|
// Retrieve the interface to be bound
|
|||
|
//
|
|||
|
|
|||
|
if (!(Interfacep = DnsLookupInterface(Index, NULL))) {
|
|||
|
LeaveCriticalSection(&DnsInterfaceLock);
|
|||
|
NhTrace(
|
|||
|
TRACE_FLAG_IF,
|
|||
|
"DnsBindInterface: interface %d not found",
|
|||
|
Index
|
|||
|
);
|
|||
|
return ERROR_NO_SUCH_INTERFACE;
|
|||
|
}
|
|||
|
|
|||
|
//
|
|||
|
// Make sure the interface isn't already bound
|
|||
|
//
|
|||
|
|
|||
|
if (DNS_INTERFACE_BOUND(Interfacep)) {
|
|||
|
LeaveCriticalSection(&DnsInterfaceLock);
|
|||
|
NhTrace(
|
|||
|
TRACE_FLAG_IF,
|
|||
|
"DnsBindInterface: interface %d is already bound",
|
|||
|
Index
|
|||
|
);
|
|||
|
return ERROR_ADDRESS_ALREADY_ASSOCIATED;
|
|||
|
}
|
|||
|
|
|||
|
//
|
|||
|
// Reference the interface
|
|||
|
//
|
|||
|
|
|||
|
if (!DNS_REFERENCE_INTERFACE(Interfacep)) {
|
|||
|
LeaveCriticalSection(&DnsInterfaceLock);
|
|||
|
NhTrace(
|
|||
|
TRACE_FLAG_IF,
|
|||
|
"DnsBindInterface: interface %d cannot be referenced",
|
|||
|
Index
|
|||
|
);
|
|||
|
return ERROR_INTERFACE_DISABLED;
|
|||
|
}
|
|||
|
|
|||
|
//
|
|||
|
// Update the interface's flags
|
|||
|
//
|
|||
|
|
|||
|
Interfacep->Flags |= DNS_INTERFACE_FLAG_BOUND;
|
|||
|
|
|||
|
LeaveCriticalSection(&DnsInterfaceLock);
|
|||
|
|
|||
|
ACQUIRE_LOCK(Interfacep);
|
|||
|
|
|||
|
//
|
|||
|
// Allocate space for the binding
|
|||
|
//
|
|||
|
|
|||
|
if (!BindingInfo->AddressCount) {
|
|||
|
Interfacep->BindingCount = 0;
|
|||
|
Interfacep->BindingArray = NULL;
|
|||
|
}
|
|||
|
else {
|
|||
|
Interfacep->BindingArray =
|
|||
|
reinterpret_cast<PDNS_BINDING>(
|
|||
|
NH_ALLOCATE(BindingInfo->AddressCount * sizeof(DNS_BINDING))
|
|||
|
);
|
|||
|
if (!Interfacep->BindingArray) {
|
|||
|
RELEASE_LOCK(Interfacep);
|
|||
|
DNS_DEREFERENCE_INTERFACE(Interfacep);
|
|||
|
NhTrace(
|
|||
|
TRACE_FLAG_IF,
|
|||
|
"DnsBindInterface: allocation failed for interface %d binding",
|
|||
|
Index
|
|||
|
);
|
|||
|
NhErrorLog(
|
|||
|
IP_DNS_PROXY_LOG_ALLOCATION_FAILED,
|
|||
|
0,
|
|||
|
"%d",
|
|||
|
BindingInfo->AddressCount * sizeof(DNS_BINDING)
|
|||
|
);
|
|||
|
return ERROR_NOT_ENOUGH_MEMORY;
|
|||
|
}
|
|||
|
Interfacep->BindingCount = BindingInfo->AddressCount;
|
|||
|
}
|
|||
|
|
|||
|
//
|
|||
|
// Copy the binding
|
|||
|
//
|
|||
|
|
|||
|
for (i = 0; i < BindingInfo->AddressCount; i++) {
|
|||
|
Interfacep->BindingArray[i].Address = BindingInfo->Address[i].Address;
|
|||
|
Interfacep->BindingArray[i].Mask = BindingInfo->Address[i].Mask;
|
|||
|
Interfacep->BindingArray[i].Socket[DnsProxyDns] = INVALID_SOCKET;
|
|||
|
Interfacep->BindingArray[i].TimerPending[DnsProxyDns] = FALSE;
|
|||
|
Interfacep->BindingArray[i].Socket[DnsProxyWins] = INVALID_SOCKET;
|
|||
|
Interfacep->BindingArray[i].TimerPending[DnsProxyWins] = FALSE;
|
|||
|
}
|
|||
|
|
|||
|
RELEASE_LOCK(Interfacep);
|
|||
|
|
|||
|
//
|
|||
|
// Activate the interface if necessary
|
|||
|
//
|
|||
|
|
|||
|
if (DNS_INTERFACE_ACTIVE(Interfacep)) {
|
|||
|
Error = DnsActivateInterface(Interfacep);
|
|||
|
}
|
|||
|
|
|||
|
DNS_DEREFERENCE_INTERFACE(Interfacep);
|
|||
|
|
|||
|
return Error;
|
|||
|
|
|||
|
} // DnsBindInterface
|
|||
|
|
|||
|
|
|||
|
VOID
|
|||
|
DnsCleanupInterface(
|
|||
|
PDNS_INTERFACE Interfacep
|
|||
|
)
|
|||
|
|
|||
|
/*++
|
|||
|
|
|||
|
Routine Description:
|
|||
|
|
|||
|
This routine is invoked when the very last reference to an interface
|
|||
|
is released, and the interface must be destroyed.
|
|||
|
|
|||
|
Arguments:
|
|||
|
|
|||
|
Interfacep - the interface to be destroyed
|
|||
|
|
|||
|
Return Value:
|
|||
|
|
|||
|
none.
|
|||
|
|
|||
|
Environment:
|
|||
|
|
|||
|
Invoked internally from an arbitrary context.
|
|||
|
|
|||
|
--*/
|
|||
|
|
|||
|
{
|
|||
|
PLIST_ENTRY Link;
|
|||
|
PDNS_QUERY Queryp;
|
|||
|
|
|||
|
PROFILE("DnsCleanupInterface");
|
|||
|
|
|||
|
if (Interfacep->BindingArray) {
|
|||
|
NH_FREE(Interfacep->BindingArray);
|
|||
|
Interfacep->BindingArray = NULL;
|
|||
|
}
|
|||
|
|
|||
|
while (!IsListEmpty(&Interfacep->QueryList)) {
|
|||
|
Link = Interfacep->QueryList.Flink;
|
|||
|
Queryp = CONTAINING_RECORD(Link, DNS_QUERY, Link);
|
|||
|
DnsDeleteQuery(Interfacep, Queryp);
|
|||
|
}
|
|||
|
|
|||
|
DeleteCriticalSection(&Interfacep->Lock);
|
|||
|
|
|||
|
NH_FREE(Interfacep);
|
|||
|
|
|||
|
} // DnsCleanupInterface
|
|||
|
|
|||
|
|
|||
|
VOID
|
|||
|
DnsConnectDefaultInterface(
|
|||
|
PVOID Unused
|
|||
|
)
|
|||
|
|
|||
|
/*++
|
|||
|
|
|||
|
Routine Description:
|
|||
|
|
|||
|
This routine is invoked to attempt to initiate a demand-dial connection
|
|||
|
on the interface marked as 'default' for DNS requests.
|
|||
|
If no such interface is found, no operation is performed.
|
|||
|
|
|||
|
Arguments:
|
|||
|
|
|||
|
none used.
|
|||
|
|
|||
|
Return Value:
|
|||
|
|
|||
|
ULONG - Win32 status code.
|
|||
|
|
|||
|
Environment:
|
|||
|
|
|||
|
The routine is invoked from the context of an RTUTILS work item.
|
|||
|
|
|||
|
--*/
|
|||
|
|
|||
|
{
|
|||
|
ULONG Error;
|
|||
|
ULONG Index;
|
|||
|
PDNS_INTERFACE Interfacep;
|
|||
|
PLIST_ENTRY Link;
|
|||
|
ULONG TickCount;
|
|||
|
ROUTER_INTERFACE_TYPE Type;
|
|||
|
|
|||
|
PROFILE("DnsConnectDefaultInterface");
|
|||
|
|
|||
|
//
|
|||
|
// To avoid repeated autodial dialogs, we record the last time
|
|||
|
// we attempted to connect the default interface.
|
|||
|
// If we did so recently, return silently.
|
|||
|
// N.B. If the tick-count wrapped, we reset the last-attempt counter.
|
|||
|
//
|
|||
|
|
|||
|
EnterCriticalSection(&DnsGlobalInfoLock);
|
|||
|
TickCount = NtGetTickCount();
|
|||
|
if (TickCount > DnspLastConnectAttemptTickCount &&
|
|||
|
TickCount <= (DnspLastConnectAttemptTickCount + DNS_CONNECT_TIMEOUT)
|
|||
|
) {
|
|||
|
LeaveCriticalSection(&DnsGlobalInfoLock);
|
|||
|
DEREFERENCE_DNS();
|
|||
|
return;
|
|||
|
}
|
|||
|
DnspLastConnectAttemptTickCount = TickCount;
|
|||
|
LeaveCriticalSection(&DnsGlobalInfoLock);
|
|||
|
|
|||
|
//
|
|||
|
// Look through the interface list for one which is marked as default
|
|||
|
//
|
|||
|
|
|||
|
EnterCriticalSection(&DnsInterfaceLock);
|
|||
|
|
|||
|
for (Link = DnsInterfaceList.Flink;
|
|||
|
Link != &DnsInterfaceList;
|
|||
|
Link = Link->Flink
|
|||
|
) {
|
|||
|
|
|||
|
Interfacep = CONTAINING_RECORD(Link, DNS_INTERFACE, Link);
|
|||
|
|
|||
|
if (!DNS_INTERFACE_ADMIN_DEFAULT(Interfacep)) { continue; }
|
|||
|
|
|||
|
//
|
|||
|
// We've found the default interface.
|
|||
|
//
|
|||
|
|
|||
|
Index = Interfacep->Index;
|
|||
|
|
|||
|
LeaveCriticalSection(&DnsInterfaceLock);
|
|||
|
|
|||
|
//
|
|||
|
// Attempt to connect it.
|
|||
|
//
|
|||
|
|
|||
|
EnterCriticalSection(&DnsGlobalInfoLock);
|
|||
|
Error =
|
|||
|
DnsSupportFunctions.DemandDialRequest(
|
|||
|
MS_IP_DNS_PROXY,
|
|||
|
Index
|
|||
|
);
|
|||
|
LeaveCriticalSection(&DnsGlobalInfoLock);
|
|||
|
DEREFERENCE_DNS();
|
|||
|
return;
|
|||
|
}
|
|||
|
|
|||
|
//
|
|||
|
// No interface is marked as the default.
|
|||
|
//
|
|||
|
|
|||
|
LeaveCriticalSection(&DnsInterfaceLock);
|
|||
|
NhDialSharedConnection();
|
|||
|
NhWarningLog(
|
|||
|
IP_DNS_PROXY_LOG_NO_DEFAULT_INTERFACE,
|
|||
|
0,
|
|||
|
""
|
|||
|
);
|
|||
|
DEREFERENCE_DNS();
|
|||
|
|
|||
|
} // DnsConnectDefaultInterface
|
|||
|
|
|||
|
|
|||
|
ULONG
|
|||
|
DnsConfigureInterface(
|
|||
|
ULONG Index,
|
|||
|
PIP_DNS_PROXY_INTERFACE_INFO InterfaceInfo
|
|||
|
)
|
|||
|
|
|||
|
/*++
|
|||
|
|
|||
|
Routine Description:
|
|||
|
|
|||
|
This routine is called to set the configuration for an interface.
|
|||
|
|
|||
|
Arguments:
|
|||
|
|
|||
|
Index - the interface to be configured
|
|||
|
|
|||
|
InterfaceInfo - the new configuration
|
|||
|
|
|||
|
Return Value:
|
|||
|
|
|||
|
ULONG - Win32 status code
|
|||
|
|
|||
|
Environment:
|
|||
|
|
|||
|
Invoked internally in the context of a IP router-manager thread.
|
|||
|
(See 'RMDNS.C').
|
|||
|
|
|||
|
--*/
|
|||
|
|
|||
|
{
|
|||
|
ULONG Error;
|
|||
|
PDNS_INTERFACE Interfacep;
|
|||
|
ULONG NewFlags;
|
|||
|
ULONG OldFlags;
|
|||
|
|
|||
|
PROFILE("DnsConfigureInterface");
|
|||
|
|
|||
|
//
|
|||
|
// Retrieve the interface to be configured
|
|||
|
//
|
|||
|
|
|||
|
EnterCriticalSection(&DnsInterfaceLock);
|
|||
|
|
|||
|
if (!(Interfacep = DnsLookupInterface(Index, NULL))) {
|
|||
|
LeaveCriticalSection(&DnsInterfaceLock);
|
|||
|
NhTrace(
|
|||
|
TRACE_FLAG_IF,
|
|||
|
"DnsConfigureInterface: interface %d not found",
|
|||
|
Index
|
|||
|
);
|
|||
|
return ERROR_NO_SUCH_INTERFACE;
|
|||
|
}
|
|||
|
|
|||
|
//
|
|||
|
// Reference the interface
|
|||
|
//
|
|||
|
|
|||
|
if (!DNS_REFERENCE_INTERFACE(Interfacep)) {
|
|||
|
LeaveCriticalSection(&DnsInterfaceLock);
|
|||
|
NhTrace(
|
|||
|
TRACE_FLAG_IF,
|
|||
|
"DnsConfigureInterface: interface %d cannot be referenced",
|
|||
|
Index
|
|||
|
);
|
|||
|
return ERROR_INTERFACE_DISABLED;
|
|||
|
}
|
|||
|
|
|||
|
LeaveCriticalSection(&DnsInterfaceLock);
|
|||
|
|
|||
|
Error = NO_ERROR;
|
|||
|
|
|||
|
ACQUIRE_LOCK(Interfacep);
|
|||
|
|
|||
|
//
|
|||
|
// Compare the interface's current and new configuration
|
|||
|
//
|
|||
|
|
|||
|
OldFlags = Interfacep->Info.Flags;
|
|||
|
NewFlags =
|
|||
|
(InterfaceInfo
|
|||
|
? (InterfaceInfo->Flags|DNS_INTERFACE_FLAG_CONFIGURED) : 0);
|
|||
|
|
|||
|
Interfacep->Flags &= ~OldFlags;
|
|||
|
Interfacep->Flags |= NewFlags;
|
|||
|
|
|||
|
if (!InterfaceInfo) {
|
|||
|
|
|||
|
ZeroMemory(&Interfacep->Info, sizeof(*InterfaceInfo));
|
|||
|
|
|||
|
//
|
|||
|
// The interface no longer has any information;
|
|||
|
// default to being enabled.
|
|||
|
//
|
|||
|
|
|||
|
if (OldFlags & IP_DNS_PROXY_INTERFACE_FLAG_DISABLED) {
|
|||
|
|
|||
|
//
|
|||
|
// Activate the interface if necessary
|
|||
|
//
|
|||
|
|
|||
|
if (DNS_INTERFACE_ACTIVE(Interfacep)) {
|
|||
|
RELEASE_LOCK(Interfacep);
|
|||
|
Error = DnsActivateInterface(Interfacep);
|
|||
|
ACQUIRE_LOCK(Interfacep);
|
|||
|
}
|
|||
|
}
|
|||
|
}
|
|||
|
else {
|
|||
|
|
|||
|
CopyMemory(&Interfacep->Info, InterfaceInfo, sizeof(*InterfaceInfo));
|
|||
|
|
|||
|
//
|
|||
|
// Activate or deactivate the interface if its status changed
|
|||
|
//
|
|||
|
|
|||
|
if ((OldFlags & IP_DNS_PROXY_INTERFACE_FLAG_DISABLED) &&
|
|||
|
!(NewFlags & IP_DNS_PROXY_INTERFACE_FLAG_DISABLED)) {
|
|||
|
|
|||
|
//
|
|||
|
// Activate the interface
|
|||
|
//
|
|||
|
|
|||
|
if (DNS_INTERFACE_ACTIVE(Interfacep)) {
|
|||
|
RELEASE_LOCK(Interfacep);
|
|||
|
Error = DnsActivateInterface(Interfacep);
|
|||
|
ACQUIRE_LOCK(Interfacep);
|
|||
|
}
|
|||
|
}
|
|||
|
else
|
|||
|
if (!(OldFlags & IP_DNS_PROXY_INTERFACE_FLAG_DISABLED) &&
|
|||
|
(NewFlags & IP_DNS_PROXY_INTERFACE_FLAG_DISABLED)) {
|
|||
|
|
|||
|
//
|
|||
|
// Deactivate the interface if necessary
|
|||
|
//
|
|||
|
|
|||
|
if (DNS_INTERFACE_ACTIVE(Interfacep)) {
|
|||
|
RELEASE_LOCK(Interfacep);
|
|||
|
DnsDeactivateInterface(Interfacep);
|
|||
|
ACQUIRE_LOCK(Interfacep);
|
|||
|
}
|
|||
|
}
|
|||
|
}
|
|||
|
|
|||
|
RELEASE_LOCK(Interfacep);
|
|||
|
DNS_DEREFERENCE_INTERFACE(Interfacep);
|
|||
|
|
|||
|
return Error;
|
|||
|
|
|||
|
} // DnsConfigureInterface
|
|||
|
|
|||
|
|
|||
|
ULONG
|
|||
|
DnsCreateInterface(
|
|||
|
ULONG Index,
|
|||
|
NET_INTERFACE_TYPE Type,
|
|||
|
PIP_DNS_PROXY_INTERFACE_INFO InterfaceInfo,
|
|||
|
OUT PDNS_INTERFACE* InterfaceCreated
|
|||
|
)
|
|||
|
|
|||
|
/*++
|
|||
|
|
|||
|
Routine Description:
|
|||
|
|
|||
|
This routine is invoked by the router-manager to add a new interface
|
|||
|
to the DNS proxy.
|
|||
|
|
|||
|
Arguments:
|
|||
|
|
|||
|
Index - the index of the new interface
|
|||
|
|
|||
|
Type - the media type of the new interface
|
|||
|
|
|||
|
InterfaceInfo - the interface's configuration
|
|||
|
|
|||
|
Interfacep - receives the interface created
|
|||
|
|
|||
|
Return Value:
|
|||
|
|
|||
|
ULONG - Win32 error code
|
|||
|
|
|||
|
Environment:
|
|||
|
|
|||
|
Invoked internally in the context of an IP router-manager thread.
|
|||
|
(See 'RMDNS.C').
|
|||
|
|
|||
|
--*/
|
|||
|
|
|||
|
{
|
|||
|
PLIST_ENTRY InsertionPoint;
|
|||
|
PDNS_INTERFACE Interfacep;
|
|||
|
|
|||
|
PROFILE("DnsCreateInterface");
|
|||
|
|
|||
|
EnterCriticalSection(&DnsInterfaceLock);
|
|||
|
|
|||
|
//
|
|||
|
// See if the interface already exists;
|
|||
|
// If not, this obtains the insertion point
|
|||
|
//
|
|||
|
|
|||
|
if (DnsLookupInterface(Index, &InsertionPoint)) {
|
|||
|
LeaveCriticalSection(&DnsInterfaceLock);
|
|||
|
NhTrace(
|
|||
|
TRACE_FLAG_IF,
|
|||
|
"DnsCreateInterface: duplicate index found for %d",
|
|||
|
Index
|
|||
|
);
|
|||
|
return ERROR_INTERFACE_ALREADY_EXISTS;
|
|||
|
}
|
|||
|
|
|||
|
//
|
|||
|
// Allocate a new interface
|
|||
|
//
|
|||
|
|
|||
|
Interfacep = reinterpret_cast<PDNS_INTERFACE>(
|
|||
|
NH_ALLOCATE(sizeof(DNS_INTERFACE))
|
|||
|
);
|
|||
|
|
|||
|
if (!Interfacep) {
|
|||
|
LeaveCriticalSection(&DnsInterfaceLock);
|
|||
|
NhTrace(
|
|||
|
TRACE_FLAG_IF, "DnsCreateInterface: error allocating interface"
|
|||
|
);
|
|||
|
NhErrorLog(
|
|||
|
IP_DNS_PROXY_LOG_ALLOCATION_FAILED,
|
|||
|
0,
|
|||
|
"%d",
|
|||
|
sizeof(DNS_INTERFACE)
|
|||
|
);
|
|||
|
return ERROR_NOT_ENOUGH_MEMORY;
|
|||
|
}
|
|||
|
|
|||
|
//
|
|||
|
// Initialize the new interface
|
|||
|
//
|
|||
|
|
|||
|
ZeroMemory(Interfacep, sizeof(*Interfacep));
|
|||
|
|
|||
|
__try {
|
|||
|
InitializeCriticalSection(&Interfacep->Lock);
|
|||
|
}
|
|||
|
__except(EXCEPTION_EXECUTE_HANDLER) {
|
|||
|
LeaveCriticalSection(&DnsInterfaceLock);
|
|||
|
NH_FREE(Interfacep);
|
|||
|
return GetExceptionCode();
|
|||
|
}
|
|||
|
|
|||
|
Interfacep->Index = Index;
|
|||
|
Interfacep->Type = Type;
|
|||
|
if (InterfaceInfo) {
|
|||
|
Interfacep->Flags = InterfaceInfo->Flags|DNS_INTERFACE_FLAG_CONFIGURED;
|
|||
|
CopyMemory(&Interfacep->Info, InterfaceInfo, sizeof(*InterfaceInfo));
|
|||
|
}
|
|||
|
Interfacep->ReferenceCount = 1;
|
|||
|
InitializeListHead(&Interfacep->QueryList);
|
|||
|
InsertTailList(InsertionPoint, &Interfacep->Link);
|
|||
|
|
|||
|
LeaveCriticalSection(&DnsInterfaceLock);
|
|||
|
|
|||
|
if (InterfaceCreated) { *InterfaceCreated = Interfacep; }
|
|||
|
|
|||
|
return NO_ERROR;
|
|||
|
|
|||
|
} // DnsCreateInterface
|
|||
|
|
|||
|
|
|||
|
VOID
|
|||
|
DnsDeactivateInterface(
|
|||
|
PDNS_INTERFACE Interfacep
|
|||
|
)
|
|||
|
|
|||
|
/*++
|
|||
|
|
|||
|
Routine Description:
|
|||
|
|
|||
|
This routine is called to deactivate an interface.
|
|||
|
It closes all sockets on the interface's bindings (if any).
|
|||
|
|
|||
|
Arguments:
|
|||
|
|
|||
|
Interfacep - the interface to be deactivated
|
|||
|
|
|||
|
Return Value:
|
|||
|
|
|||
|
none.
|
|||
|
|
|||
|
Environment:
|
|||
|
|
|||
|
Always invoked locally, with 'Interfacep' referenced by caller and/or
|
|||
|
'DnsInterfaceLock' held by caller.
|
|||
|
|
|||
|
--*/
|
|||
|
|
|||
|
{
|
|||
|
ULONG i;
|
|||
|
PLIST_ENTRY Link;
|
|||
|
PDNS_QUERY Queryp;
|
|||
|
|
|||
|
PROFILE("DnsDeactivateInterface");
|
|||
|
|
|||
|
ACQUIRE_LOCK(Interfacep);
|
|||
|
|
|||
|
//
|
|||
|
// Stop all network I/O on the interface's logical networks
|
|||
|
//
|
|||
|
|
|||
|
for (i = 0; i < Interfacep->BindingCount; i++) {
|
|||
|
NhDeleteDatagramSocket(
|
|||
|
Interfacep->BindingArray[i].Socket[DnsProxyDns]
|
|||
|
);
|
|||
|
Interfacep->BindingArray[i].Socket[DnsProxyDns] = INVALID_SOCKET;
|
|||
|
NhDeleteDatagramSocket(
|
|||
|
Interfacep->BindingArray[i].Socket[DnsProxyWins]
|
|||
|
);
|
|||
|
Interfacep->BindingArray[i].Socket[DnsProxyWins] = INVALID_SOCKET;
|
|||
|
}
|
|||
|
|
|||
|
//
|
|||
|
// Eliminate all pending queries
|
|||
|
//
|
|||
|
|
|||
|
while (!IsListEmpty(&Interfacep->QueryList)) {
|
|||
|
Link = RemoveHeadList(&Interfacep->QueryList);
|
|||
|
Queryp = CONTAINING_RECORD(Link, DNS_QUERY, Link);
|
|||
|
NH_FREE(Queryp);
|
|||
|
}
|
|||
|
|
|||
|
RELEASE_LOCK(Interfacep);
|
|||
|
|
|||
|
} // DnsDeactivateInterface
|
|||
|
|
|||
|
|
|||
|
VOID NTAPI
|
|||
|
DnspDeferReadCallbackRoutine(
|
|||
|
PVOID Context,
|
|||
|
BOOLEAN TimedOut
|
|||
|
)
|
|||
|
|
|||
|
/*++
|
|||
|
|
|||
|
Routine Description:
|
|||
|
|
|||
|
This routine is invoked to re-issue a deferred read when the countdown
|
|||
|
for the deferral completes.
|
|||
|
|
|||
|
Arguments:
|
|||
|
|
|||
|
Context - holds information identifying the interface and socket
|
|||
|
|
|||
|
TimedOut - indicates whether the countdown completed
|
|||
|
|
|||
|
Return Value:
|
|||
|
|
|||
|
none.
|
|||
|
|
|||
|
Environment:
|
|||
|
|
|||
|
Invoked with an outstanding reference to the component on our behalf.
|
|||
|
|
|||
|
--*/
|
|||
|
|
|||
|
{
|
|||
|
PDNS_DEFER_READ_CONTEXT Contextp;
|
|||
|
ULONG Error;
|
|||
|
ULONG i;
|
|||
|
PDNS_INTERFACE Interfacep;
|
|||
|
NTSTATUS status;
|
|||
|
DNS_PROXY_TYPE Type;
|
|||
|
|
|||
|
PROFILE("DnspDeferReadCallbackRoutine");
|
|||
|
|
|||
|
Contextp = (PDNS_DEFER_READ_CONTEXT)Context;
|
|||
|
|
|||
|
//
|
|||
|
// Find the interface on which the read was deferred
|
|||
|
//
|
|||
|
|
|||
|
EnterCriticalSection(&DnsInterfaceLock);
|
|||
|
Interfacep = DnsLookupInterface(Contextp->Index, NULL);
|
|||
|
if (!Interfacep ||
|
|||
|
!DNS_INTERFACE_ACTIVE(Interfacep) ||
|
|||
|
!DNS_REFERENCE_INTERFACE(Interfacep)) {
|
|||
|
LeaveCriticalSection(&DnsInterfaceLock);
|
|||
|
NH_FREE(Contextp);
|
|||
|
DEREFERENCE_DNS();
|
|||
|
return;
|
|||
|
}
|
|||
|
LeaveCriticalSection(&DnsInterfaceLock);
|
|||
|
|
|||
|
ACQUIRE_LOCK(Interfacep);
|
|||
|
|
|||
|
//
|
|||
|
// Search for the socket on which to reissue the read
|
|||
|
//
|
|||
|
|
|||
|
for (i = 0; i < Interfacep->BindingCount; i++) {
|
|||
|
|
|||
|
if (Interfacep->BindingArray[i].Socket[Type = DnsProxyDns]
|
|||
|
!= Contextp->Socket &&
|
|||
|
Interfacep->BindingArray[i].Socket[Type = DnsProxyWins]
|
|||
|
!= Contextp->Socket) {
|
|||
|
continue;
|
|||
|
}
|
|||
|
|
|||
|
//
|
|||
|
// This is the binding on which to reissue the read.
|
|||
|
// If no pending timer is recorded, assume a rebind occurred, and quit.
|
|||
|
//
|
|||
|
|
|||
|
if (!Interfacep->BindingArray[i].TimerPending[Type]) { break; }
|
|||
|
|
|||
|
Interfacep->BindingArray[i].TimerPending[Type] = FALSE;
|
|||
|
|
|||
|
Error =
|
|||
|
NhReadDatagramSocket(
|
|||
|
&DnsComponentReference,
|
|||
|
Contextp->Socket,
|
|||
|
NULL,
|
|||
|
DnsReadCompletionRoutine,
|
|||
|
Interfacep,
|
|||
|
UlongToPtr(Interfacep->BindingArray[i].Mask)
|
|||
|
);
|
|||
|
|
|||
|
RELEASE_LOCK(Interfacep);
|
|||
|
|
|||
|
if (!Error) {
|
|||
|
NH_FREE(Contextp);
|
|||
|
DEREFERENCE_DNS();
|
|||
|
return;
|
|||
|
}
|
|||
|
|
|||
|
//
|
|||
|
// An error occurred; we'll have to retry later.
|
|||
|
// we queue a work item which sets the timer.
|
|||
|
//
|
|||
|
|
|||
|
NhTrace(
|
|||
|
TRACE_FLAG_DNS,
|
|||
|
"DnspDeferReadCallbackRoutine: error %d reading interface %d",
|
|||
|
Error,
|
|||
|
Interfacep->Index
|
|||
|
);
|
|||
|
|
|||
|
//
|
|||
|
// Reference the component on behalf of the work-item
|
|||
|
//
|
|||
|
|
|||
|
if (REFERENCE_DNS()) {
|
|||
|
|
|||
|
//
|
|||
|
// Queue a work-item, reusing the deferral context
|
|||
|
//
|
|||
|
|
|||
|
status =
|
|||
|
RtlQueueWorkItem(
|
|||
|
DnspDeferReadWorkerRoutine,
|
|||
|
Contextp,
|
|||
|
WT_EXECUTEINIOTHREAD
|
|||
|
);
|
|||
|
|
|||
|
if (NT_SUCCESS(status)) {
|
|||
|
Contextp = NULL;
|
|||
|
}
|
|||
|
else {
|
|||
|
NH_FREE(Contextp);
|
|||
|
NhTrace(
|
|||
|
TRACE_FLAG_DNS,
|
|||
|
"DnspDeferReadCallbackRoutine: error %d deferring %d",
|
|||
|
Error,
|
|||
|
Interfacep->Index
|
|||
|
);
|
|||
|
DEREFERENCE_DNS();
|
|||
|
}
|
|||
|
}
|
|||
|
|
|||
|
DNS_DEREFERENCE_INTERFACE(Interfacep);
|
|||
|
DEREFERENCE_DNS();
|
|||
|
return;
|
|||
|
}
|
|||
|
|
|||
|
//
|
|||
|
// The interface was not found; never mind.
|
|||
|
//
|
|||
|
|
|||
|
RELEASE_LOCK(Interfacep);
|
|||
|
DNS_DEREFERENCE_INTERFACE(Interfacep);
|
|||
|
NH_FREE(Contextp);
|
|||
|
DEREFERENCE_DNS();
|
|||
|
|
|||
|
} // DnspDeferReadCallbackRoutine
|
|||
|
|
|||
|
|
|||
|
VOID
|
|||
|
DnsDeferReadInterface(
|
|||
|
PDNS_INTERFACE Interfacep,
|
|||
|
SOCKET Socket
|
|||
|
)
|
|||
|
|
|||
|
/*++
|
|||
|
|
|||
|
Routine Description:
|
|||
|
|
|||
|
This routine is invoked to defer a read-request on an interface,
|
|||
|
typically if an attempt to post a read failed.
|
|||
|
|
|||
|
Arguments:
|
|||
|
|
|||
|
Interfacep - the interface on which to defer the request
|
|||
|
|
|||
|
Socket - the socket on which to defer the request
|
|||
|
|
|||
|
Return Value:
|
|||
|
|
|||
|
none.
|
|||
|
|
|||
|
Environment:
|
|||
|
|
|||
|
Invoked with 'Interfacep' referenced and locked by the caller.
|
|||
|
The caller may release the reference upon return.
|
|||
|
|
|||
|
--*/
|
|||
|
|
|||
|
{
|
|||
|
PDNS_DEFER_READ_CONTEXT Contextp;
|
|||
|
ULONG i;
|
|||
|
NTSTATUS status;
|
|||
|
DNS_PROXY_TYPE Type;
|
|||
|
|
|||
|
PROFILE("DnsDeferReadInterface");
|
|||
|
|
|||
|
//
|
|||
|
// Find the binding for the given socket.
|
|||
|
//
|
|||
|
|
|||
|
status = STATUS_SUCCESS;
|
|||
|
|
|||
|
for (i = 0; i < Interfacep->BindingCount; i++) {
|
|||
|
|
|||
|
if (Interfacep->BindingArray[i].Socket[Type = DnsProxyDns] != Socket &&
|
|||
|
Interfacep->BindingArray[i].Socket[Type = DnsProxyWins] != Socket) {
|
|||
|
continue;
|
|||
|
}
|
|||
|
|
|||
|
//
|
|||
|
// This is the binding. If there is already a timer for it,
|
|||
|
// then just return silently.
|
|||
|
//
|
|||
|
|
|||
|
if (Interfacep->BindingArray[i].TimerPending[Type]) {
|
|||
|
status = STATUS_UNSUCCESSFUL;
|
|||
|
break;
|
|||
|
}
|
|||
|
|
|||
|
//
|
|||
|
// Allocate a context block for the deferral.
|
|||
|
//
|
|||
|
|
|||
|
Contextp =
|
|||
|
(PDNS_DEFER_READ_CONTEXT)
|
|||
|
NH_ALLOCATE(sizeof(DNS_DEFER_READ_CONTEXT));
|
|||
|
|
|||
|
if (!Contextp) {
|
|||
|
NhTrace(
|
|||
|
TRACE_FLAG_DNS,
|
|||
|
"DnsDeferReadInterface: cannot allocate deferral context"
|
|||
|
);
|
|||
|
status = STATUS_NO_MEMORY;
|
|||
|
break;
|
|||
|
}
|
|||
|
|
|||
|
Contextp->Index = Interfacep->Index;
|
|||
|
Contextp->Socket = Socket;
|
|||
|
Contextp->DeferralCount = 1;
|
|||
|
|
|||
|
//
|
|||
|
// Install a timer to re-issue the read request
|
|||
|
//
|
|||
|
|
|||
|
status =
|
|||
|
NhSetTimer(
|
|||
|
&DnsComponentReference,
|
|||
|
NULL,
|
|||
|
DnspDeferReadCallbackRoutine,
|
|||
|
Contextp,
|
|||
|
DNS_DEFER_READ_INITIAL_TIMEOUT
|
|||
|
);
|
|||
|
|
|||
|
if (NT_SUCCESS(status)) {
|
|||
|
Interfacep->BindingArray[i].TimerPending[Type] = TRUE;
|
|||
|
}
|
|||
|
else {
|
|||
|
NH_FREE(Contextp);
|
|||
|
NhTrace(
|
|||
|
TRACE_FLAG_DNS,
|
|||
|
"DnsDeferReadInterface: status %08x setting deferral timer",
|
|||
|
status
|
|||
|
);
|
|||
|
}
|
|||
|
|
|||
|
break;
|
|||
|
}
|
|||
|
|
|||
|
if (i >= Interfacep->BindingCount) { status = STATUS_UNSUCCESSFUL; }
|
|||
|
|
|||
|
} // DnsDeferReadInterface
|
|||
|
|
|||
|
|
|||
|
VOID APIENTRY
|
|||
|
DnspDeferReadWorkerRoutine(
|
|||
|
PVOID Context
|
|||
|
)
|
|||
|
|
|||
|
/*++
|
|||
|
|
|||
|
Routine Description:
|
|||
|
|
|||
|
This routine is invoked to set a timer for reissuing a deferred read.
|
|||
|
|
|||
|
Arguments:
|
|||
|
|
|||
|
Context - contains the context for the timer.
|
|||
|
|
|||
|
Return Value:
|
|||
|
|
|||
|
none.
|
|||
|
|
|||
|
Environment:
|
|||
|
|
|||
|
Invoked with an outstanding reference to the module made on our behalf.
|
|||
|
|
|||
|
--*/
|
|||
|
|
|||
|
{
|
|||
|
PDNS_DEFER_READ_CONTEXT Contextp;
|
|||
|
ULONG i;
|
|||
|
PDNS_INTERFACE Interfacep;
|
|||
|
NTSTATUS status;
|
|||
|
DNS_PROXY_TYPE Type;
|
|||
|
|
|||
|
PROFILE("DnspDeferReadWorkerRoutine");
|
|||
|
|
|||
|
Contextp = (PDNS_DEFER_READ_CONTEXT)Context;
|
|||
|
++Contextp->DeferralCount;
|
|||
|
|
|||
|
//
|
|||
|
// Find the interface on which the read was deferred
|
|||
|
//
|
|||
|
|
|||
|
EnterCriticalSection(&DnsInterfaceLock);
|
|||
|
Interfacep = DnsLookupInterface(Contextp->Index, NULL);
|
|||
|
if (!Interfacep ||
|
|||
|
!DNS_INTERFACE_ACTIVE(Interfacep) ||
|
|||
|
!DNS_REFERENCE_INTERFACE(Interfacep)) {
|
|||
|
LeaveCriticalSection(&DnsInterfaceLock);
|
|||
|
NH_FREE(Contextp);
|
|||
|
DEREFERENCE_DNS();
|
|||
|
return;
|
|||
|
}
|
|||
|
LeaveCriticalSection(&DnsInterfaceLock);
|
|||
|
|
|||
|
ACQUIRE_LOCK(Interfacep);
|
|||
|
|
|||
|
//
|
|||
|
// Search for the binding on which to set the timer
|
|||
|
//
|
|||
|
|
|||
|
for (i = 0; i < Interfacep->BindingCount; i++) {
|
|||
|
|
|||
|
if (Interfacep->BindingArray[i].Socket[Type = DnsProxyDns]
|
|||
|
!= Contextp->Socket &&
|
|||
|
Interfacep->BindingArray[i].Socket[Type = DnsProxyWins]
|
|||
|
!= Contextp->Socket) {
|
|||
|
continue;
|
|||
|
}
|
|||
|
|
|||
|
//
|
|||
|
// This is the binding on which to reissue the read.
|
|||
|
// If a timer is already pending, assume a rebind occurred, and quit.
|
|||
|
//
|
|||
|
|
|||
|
if (Interfacep->BindingArray[i].TimerPending[Type]) { break; }
|
|||
|
|
|||
|
//
|
|||
|
// Install a timer to re-issue the read request,
|
|||
|
// reusing the deferral context.
|
|||
|
//
|
|||
|
|
|||
|
status =
|
|||
|
NhSetTimer(
|
|||
|
&DnsComponentReference,
|
|||
|
NULL,
|
|||
|
DnspDeferReadCallbackRoutine,
|
|||
|
Contextp,
|
|||
|
DNS_DEFER_READ_TIMEOUT
|
|||
|
);
|
|||
|
|
|||
|
if (NT_SUCCESS(status)) {
|
|||
|
Contextp = NULL;
|
|||
|
Interfacep->BindingArray[i].TimerPending[Type] = TRUE;
|
|||
|
}
|
|||
|
else {
|
|||
|
NhTrace(
|
|||
|
TRACE_FLAG_DNS,
|
|||
|
"DnspDeferReadWorkerRoutine: status %08x setting timer",
|
|||
|
status
|
|||
|
);
|
|||
|
}
|
|||
|
}
|
|||
|
|
|||
|
RELEASE_LOCK(Interfacep);
|
|||
|
DNS_DEREFERENCE_INTERFACE(Interfacep);
|
|||
|
if (Contextp) { NH_FREE(Contextp); }
|
|||
|
DEREFERENCE_DNS();
|
|||
|
|
|||
|
} // DnspDeferReadWorkerRoutine
|
|||
|
|
|||
|
|
|||
|
ULONG
|
|||
|
DnsDeleteInterface(
|
|||
|
ULONG Index
|
|||
|
)
|
|||
|
|
|||
|
/*++
|
|||
|
|
|||
|
Routine Description:
|
|||
|
|
|||
|
This routine is called to delete an interface.
|
|||
|
It drops the reference count on the interface so that the last
|
|||
|
dereferencer will delete the interface, and sets the 'deleted' flag
|
|||
|
so that further references to the interface will fail.
|
|||
|
|
|||
|
Arguments:
|
|||
|
|
|||
|
Index - the index of the interface to be deleted
|
|||
|
|
|||
|
Return Value:
|
|||
|
|
|||
|
ULONG - Win32 status code.
|
|||
|
|
|||
|
Environment:
|
|||
|
|
|||
|
Invoked internally in the context of an IP router-manager thread.
|
|||
|
(See 'RMDNS.C').
|
|||
|
|
|||
|
--*/
|
|||
|
|
|||
|
{
|
|||
|
PDNS_INTERFACE Interfacep;
|
|||
|
|
|||
|
PROFILE("DnsDeleteInterface");
|
|||
|
|
|||
|
//
|
|||
|
// Retrieve the interface to be deleted
|
|||
|
//
|
|||
|
|
|||
|
EnterCriticalSection(&DnsInterfaceLock);
|
|||
|
|
|||
|
if (!(Interfacep = DnsLookupInterface(Index, NULL))) {
|
|||
|
LeaveCriticalSection(&DnsInterfaceLock);
|
|||
|
NhTrace(
|
|||
|
TRACE_FLAG_IF,
|
|||
|
"DnsDeleteInterface: interface %d not found",
|
|||
|
Index
|
|||
|
);
|
|||
|
return ERROR_NO_SUCH_INTERFACE;
|
|||
|
}
|
|||
|
|
|||
|
//
|
|||
|
// Deactivate the interface
|
|||
|
//
|
|||
|
|
|||
|
DnsDeactivateInterface(Interfacep);
|
|||
|
|
|||
|
//
|
|||
|
// Mark the interface as deleted and take it off the interface list
|
|||
|
//
|
|||
|
|
|||
|
Interfacep->Flags |= DNS_INTERFACE_FLAG_DELETED;
|
|||
|
Interfacep->Flags &= ~DNS_INTERFACE_FLAG_ENABLED;
|
|||
|
RemoveEntryList(&Interfacep->Link);
|
|||
|
|
|||
|
//
|
|||
|
// Drop the reference count; if it is non-zero,
|
|||
|
// the deletion will complete later.
|
|||
|
//
|
|||
|
|
|||
|
if (--Interfacep->ReferenceCount) {
|
|||
|
LeaveCriticalSection(&DnsInterfaceLock);
|
|||
|
NhTrace(
|
|||
|
TRACE_FLAG_IF,
|
|||
|
"DnsDeleteInterface: interface %d deletion pending",
|
|||
|
Index
|
|||
|
);
|
|||
|
return NO_ERROR;
|
|||
|
}
|
|||
|
|
|||
|
//
|
|||
|
// The reference count is zero, so perform final cleanup
|
|||
|
//
|
|||
|
|
|||
|
DnsCleanupInterface(Interfacep);
|
|||
|
|
|||
|
LeaveCriticalSection(&DnsInterfaceLock);
|
|||
|
|
|||
|
return NO_ERROR;
|
|||
|
|
|||
|
} // DnsDeleteInterface
|
|||
|
|
|||
|
|
|||
|
ULONG
|
|||
|
DnsDisableInterface(
|
|||
|
ULONG Index
|
|||
|
)
|
|||
|
|
|||
|
/*++
|
|||
|
|
|||
|
Routine Description:
|
|||
|
|
|||
|
This routine is called to disable I/O on an interface.
|
|||
|
If the interface is active, it is deactivated.
|
|||
|
|
|||
|
Arguments:
|
|||
|
|
|||
|
Index - the index of the interface to be disabled.
|
|||
|
|
|||
|
Return Value:
|
|||
|
|
|||
|
none.
|
|||
|
|
|||
|
Environment:
|
|||
|
|
|||
|
Invoked internally in the context of an IP router-manager thread.
|
|||
|
(See 'RMDNS.C').
|
|||
|
|
|||
|
--*/
|
|||
|
|
|||
|
{
|
|||
|
PDNS_INTERFACE Interfacep;
|
|||
|
|
|||
|
PROFILE("DnsDisableInterface");
|
|||
|
|
|||
|
//
|
|||
|
// Retrieve the interface to be disabled
|
|||
|
//
|
|||
|
|
|||
|
EnterCriticalSection(&DnsInterfaceLock);
|
|||
|
|
|||
|
if (!(Interfacep = DnsLookupInterface(Index, NULL))) {
|
|||
|
LeaveCriticalSection(&DnsInterfaceLock);
|
|||
|
NhTrace(
|
|||
|
TRACE_FLAG_IF,
|
|||
|
"DnsDisableInterface: interface %d not found",
|
|||
|
Index
|
|||
|
);
|
|||
|
return ERROR_NO_SUCH_INTERFACE;
|
|||
|
}
|
|||
|
|
|||
|
//
|
|||
|
// Make sure the interface is not already disabled
|
|||
|
//
|
|||
|
|
|||
|
if (!DNS_INTERFACE_ENABLED(Interfacep)) {
|
|||
|
LeaveCriticalSection(&DnsInterfaceLock);
|
|||
|
NhTrace(
|
|||
|
TRACE_FLAG_IF,
|
|||
|
"DnsDisableInterface: interface %d already disabled",
|
|||
|
Index
|
|||
|
);
|
|||
|
return ERROR_INTERFACE_DISABLED;
|
|||
|
}
|
|||
|
|
|||
|
//
|
|||
|
// Reference the interface
|
|||
|
//
|
|||
|
|
|||
|
if (!DNS_REFERENCE_INTERFACE(Interfacep)) {
|
|||
|
LeaveCriticalSection(&DnsInterfaceLock);
|
|||
|
NhTrace(
|
|||
|
TRACE_FLAG_IF,
|
|||
|
"DnsDisableInterface: interface %d cannot be referenced",
|
|||
|
Index
|
|||
|
);
|
|||
|
return ERROR_INTERFACE_DISABLED;
|
|||
|
}
|
|||
|
|
|||
|
//
|
|||
|
// Clear the 'enabled' flag
|
|||
|
//
|
|||
|
|
|||
|
Interfacep->Flags &= ~DNS_INTERFACE_FLAG_ENABLED;
|
|||
|
|
|||
|
//
|
|||
|
// Deactivate the interface, if necessary
|
|||
|
//
|
|||
|
|
|||
|
if (DNS_INTERFACE_BOUND(Interfacep)) {
|
|||
|
DnsDeactivateInterface(Interfacep);
|
|||
|
}
|
|||
|
|
|||
|
LeaveCriticalSection(&DnsInterfaceLock);
|
|||
|
|
|||
|
DNS_DEREFERENCE_INTERFACE(Interfacep);
|
|||
|
|
|||
|
return NO_ERROR;
|
|||
|
|
|||
|
} // DnsDisableInterface
|
|||
|
|
|||
|
|
|||
|
ULONG
|
|||
|
DnsEnableInterface(
|
|||
|
ULONG Index
|
|||
|
)
|
|||
|
|
|||
|
/*++
|
|||
|
|
|||
|
Routine Description:
|
|||
|
|
|||
|
This routine is called to enable I/O on an interface.
|
|||
|
If the interface is already bound, this enabling activates it.
|
|||
|
|
|||
|
Arguments:
|
|||
|
|
|||
|
Index - the index of the interfaec to be enabled
|
|||
|
|
|||
|
Return Value:
|
|||
|
|
|||
|
ULONG - Win32 status code.
|
|||
|
|
|||
|
Environment:
|
|||
|
|
|||
|
Invoked internally in the context of an IP router-manager thread.
|
|||
|
(See 'RMDNS.C').
|
|||
|
|
|||
|
--*/
|
|||
|
|
|||
|
{
|
|||
|
ULONG Error = NO_ERROR;
|
|||
|
PDNS_INTERFACE Interfacep;
|
|||
|
|
|||
|
PROFILE("DnsEnableInterface");
|
|||
|
|
|||
|
//
|
|||
|
// Retrieve the interface to be enabled
|
|||
|
//
|
|||
|
|
|||
|
EnterCriticalSection(&DnsInterfaceLock);
|
|||
|
|
|||
|
if (!(Interfacep = DnsLookupInterface(Index, NULL))) {
|
|||
|
LeaveCriticalSection(&DnsInterfaceLock);
|
|||
|
NhTrace(
|
|||
|
TRACE_FLAG_IF,
|
|||
|
"DnsEnableInterface: interface %d not found",
|
|||
|
Index
|
|||
|
);
|
|||
|
return ERROR_NO_SUCH_INTERFACE;
|
|||
|
}
|
|||
|
|
|||
|
//
|
|||
|
// Make sure the interface is not already enabled
|
|||
|
//
|
|||
|
|
|||
|
if (DNS_INTERFACE_ENABLED(Interfacep)) {
|
|||
|
LeaveCriticalSection(&DnsInterfaceLock);
|
|||
|
NhTrace(
|
|||
|
TRACE_FLAG_IF,
|
|||
|
"DnsEnableInterface: interface %d already enabled",
|
|||
|
Index
|
|||
|
);
|
|||
|
return ERROR_INTERFACE_ALREADY_EXISTS;
|
|||
|
}
|
|||
|
|
|||
|
//
|
|||
|
// Reference the interface
|
|||
|
//
|
|||
|
|
|||
|
if (!DNS_REFERENCE_INTERFACE(Interfacep)) {
|
|||
|
LeaveCriticalSection(&DnsInterfaceLock);
|
|||
|
NhTrace(
|
|||
|
TRACE_FLAG_IF,
|
|||
|
"DnsEnableInterface: interface %d cannot be referenced",
|
|||
|
Index
|
|||
|
);
|
|||
|
return ERROR_INTERFACE_DISABLED;
|
|||
|
}
|
|||
|
|
|||
|
//
|
|||
|
// Set the 'enabled' flag
|
|||
|
//
|
|||
|
|
|||
|
Interfacep->Flags |= DNS_INTERFACE_FLAG_ENABLED;
|
|||
|
|
|||
|
//
|
|||
|
// Activate the interface, if necessary
|
|||
|
//
|
|||
|
|
|||
|
if (DNS_INTERFACE_ACTIVE(Interfacep)) {
|
|||
|
Error = DnsActivateInterface(Interfacep);
|
|||
|
}
|
|||
|
|
|||
|
LeaveCriticalSection(&DnsInterfaceLock);
|
|||
|
|
|||
|
DNS_DEREFERENCE_INTERFACE(Interfacep);
|
|||
|
|
|||
|
return Error;
|
|||
|
|
|||
|
} // DnsEnableInterface
|
|||
|
|
|||
|
|
|||
|
ULONG
|
|||
|
DnsInitializeInterfaceManagement(
|
|||
|
VOID
|
|||
|
)
|
|||
|
|
|||
|
/*++
|
|||
|
|
|||
|
Routine Description:
|
|||
|
|
|||
|
This routine is called to initialize the interface-management module.
|
|||
|
|
|||
|
Arguments:
|
|||
|
|
|||
|
none.
|
|||
|
|
|||
|
Return Value:
|
|||
|
|
|||
|
ULONG - Win32 status code.
|
|||
|
|
|||
|
Environment:
|
|||
|
|
|||
|
Invoked internally in the context of an IP router-manager thread.
|
|||
|
(See 'RMDNS.C').
|
|||
|
|
|||
|
--*/
|
|||
|
|
|||
|
{
|
|||
|
ULONG Error = NO_ERROR;
|
|||
|
PROFILE("DnsInitializeInterfaceManagement");
|
|||
|
|
|||
|
InitializeListHead(&DnsInterfaceList);
|
|||
|
__try {
|
|||
|
InitializeCriticalSection(&DnsInterfaceLock);
|
|||
|
}
|
|||
|
__except(EXCEPTION_EXECUTE_HANDLER) {
|
|||
|
NhTrace(
|
|||
|
TRACE_FLAG_IF,
|
|||
|
"DnsInitializeInterfaceManagement: exception %d creating lock",
|
|||
|
Error = GetExceptionCode()
|
|||
|
);
|
|||
|
}
|
|||
|
|
|||
|
DnspLastConnectAttemptTickCount = NtGetTickCount();
|
|||
|
|
|||
|
return Error;
|
|||
|
|
|||
|
} // DnsInitializeInterfaceManagement
|
|||
|
|
|||
|
|
|||
|
PDNS_INTERFACE
|
|||
|
DnsLookupInterface(
|
|||
|
ULONG Index,
|
|||
|
OUT PLIST_ENTRY* InsertionPoint OPTIONAL
|
|||
|
)
|
|||
|
|
|||
|
/*++
|
|||
|
|
|||
|
Routine Description:
|
|||
|
|
|||
|
This routine is called to retrieve an interface given its index.
|
|||
|
|
|||
|
Arguments:
|
|||
|
|
|||
|
Index - the index of the interface to be retrieved
|
|||
|
|
|||
|
InsertionPoint - if the interface is not found, optionally receives
|
|||
|
the point where the interface would be inserted in the interface list
|
|||
|
|
|||
|
Return Value:
|
|||
|
|
|||
|
PDNS_INTERFACE - the interface, if found; otherwise, NULL.
|
|||
|
|
|||
|
Environment:
|
|||
|
|
|||
|
Invoked internally from an arbitrary context, with 'DnsInterfaceLock'
|
|||
|
held by caller.
|
|||
|
|
|||
|
--*/
|
|||
|
|
|||
|
{
|
|||
|
PDNS_INTERFACE Interfacep;
|
|||
|
PLIST_ENTRY Link;
|
|||
|
|
|||
|
PROFILE("DnsLookupInterface");
|
|||
|
|
|||
|
for (Link = DnsInterfaceList.Flink;
|
|||
|
Link != &DnsInterfaceList;
|
|||
|
Link = Link->Flink
|
|||
|
) {
|
|||
|
|
|||
|
Interfacep = CONTAINING_RECORD(Link, DNS_INTERFACE, Link);
|
|||
|
|
|||
|
if (Index > Interfacep->Index) { continue; }
|
|||
|
else
|
|||
|
if (Index < Interfacep->Index) { break; }
|
|||
|
|
|||
|
return Interfacep;
|
|||
|
}
|
|||
|
|
|||
|
if (InsertionPoint) { *InsertionPoint = Link; }
|
|||
|
|
|||
|
return NULL;
|
|||
|
|
|||
|
} // DnsLookupInterface
|
|||
|
|
|||
|
|
|||
|
ULONG
|
|||
|
DnsQueryInterface(
|
|||
|
ULONG Index,
|
|||
|
PVOID InterfaceInfo,
|
|||
|
PULONG InterfaceInfoSize
|
|||
|
)
|
|||
|
|
|||
|
/*++
|
|||
|
|
|||
|
Routine Description:
|
|||
|
|
|||
|
This routine is invoked to retrieve the configuration for an interface.
|
|||
|
|
|||
|
Arguments:
|
|||
|
|
|||
|
Index - the interface to be queried
|
|||
|
|
|||
|
InterfaceInfo - receives the retrieved information
|
|||
|
|
|||
|
InterfaceInfoSize - receives the (required) size of the information
|
|||
|
|
|||
|
Return Value:
|
|||
|
|
|||
|
ULONG - Win32 status code.
|
|||
|
|
|||
|
--*/
|
|||
|
|
|||
|
{
|
|||
|
PDNS_INTERFACE Interfacep;
|
|||
|
|
|||
|
PROFILE("DnsQueryInterface");
|
|||
|
|
|||
|
//
|
|||
|
// Check the caller's buffer size
|
|||
|
//
|
|||
|
|
|||
|
if (!InterfaceInfoSize) { return ERROR_INVALID_PARAMETER; }
|
|||
|
|
|||
|
//
|
|||
|
// Retrieve the interface to be configured
|
|||
|
//
|
|||
|
|
|||
|
EnterCriticalSection(&DnsInterfaceLock);
|
|||
|
|
|||
|
if (!(Interfacep = DnsLookupInterface(Index, NULL))) {
|
|||
|
LeaveCriticalSection(&DnsInterfaceLock);
|
|||
|
NhTrace(
|
|||
|
TRACE_FLAG_IF,
|
|||
|
"DnsQueryInterface: interface %d not found",
|
|||
|
Index
|
|||
|
);
|
|||
|
return ERROR_NO_SUCH_INTERFACE;
|
|||
|
}
|
|||
|
|
|||
|
//
|
|||
|
// Reference the interface
|
|||
|
//
|
|||
|
|
|||
|
if (!DNS_REFERENCE_INTERFACE(Interfacep)) {
|
|||
|
LeaveCriticalSection(&DnsInterfaceLock);
|
|||
|
NhTrace(
|
|||
|
TRACE_FLAG_IF,
|
|||
|
"DnsQueryInterface: interface %d cannot be referenced",
|
|||
|
Index
|
|||
|
);
|
|||
|
return ERROR_INTERFACE_DISABLED;
|
|||
|
}
|
|||
|
|
|||
|
//
|
|||
|
// See if there is any explicit config on this interface
|
|||
|
//
|
|||
|
|
|||
|
if (!DNS_INTERFACE_CONFIGURED(Interfacep)) {
|
|||
|
LeaveCriticalSection(&DnsInterfaceLock);
|
|||
|
DNS_DEREFERENCE_INTERFACE(Interfacep);
|
|||
|
NhTrace(
|
|||
|
TRACE_FLAG_IF,
|
|||
|
"DnsQueryInterface: interface %d has no configuration",
|
|||
|
Index
|
|||
|
);
|
|||
|
*InterfaceInfoSize = 0;
|
|||
|
return NO_ERROR;
|
|||
|
}
|
|||
|
|
|||
|
//
|
|||
|
// See if there is enough buffer space
|
|||
|
//
|
|||
|
|
|||
|
if (*InterfaceInfoSize < sizeof(IP_DNS_PROXY_INTERFACE_INFO)) {
|
|||
|
LeaveCriticalSection(&DnsInterfaceLock);
|
|||
|
DNS_DEREFERENCE_INTERFACE(Interfacep);
|
|||
|
*InterfaceInfoSize = sizeof(IP_DNS_PROXY_INTERFACE_INFO);
|
|||
|
return ERROR_INSUFFICIENT_BUFFER;
|
|||
|
}
|
|||
|
|
|||
|
//
|
|||
|
// Copy the requested data
|
|||
|
//
|
|||
|
|
|||
|
CopyMemory(
|
|||
|
InterfaceInfo,
|
|||
|
&Interfacep->Info,
|
|||
|
sizeof(IP_DNS_PROXY_INTERFACE_INFO)
|
|||
|
);
|
|||
|
*InterfaceInfoSize = sizeof(IP_DNS_PROXY_INTERFACE_INFO);
|
|||
|
|
|||
|
LeaveCriticalSection(&DnsInterfaceLock);
|
|||
|
|
|||
|
DNS_DEREFERENCE_INTERFACE(Interfacep);
|
|||
|
|
|||
|
return NO_ERROR;
|
|||
|
|
|||
|
} // DnsQueryInterface
|
|||
|
|
|||
|
|
|||
|
VOID
|
|||
|
DnsReactivateEveryInterface(
|
|||
|
VOID
|
|||
|
)
|
|||
|
|
|||
|
/*++
|
|||
|
|
|||
|
Routine Description:
|
|||
|
|
|||
|
This routine is called to reactivate all activate interfaces
|
|||
|
when a change occurs to the global DNS or WINS proxy setting.
|
|||
|
Thus if, for instance, WINS proxy is disabled, during deactivation
|
|||
|
all such sockets are closed, and during reactivation they are
|
|||
|
not reopened.
|
|||
|
|
|||
|
Arguments:
|
|||
|
|
|||
|
none.
|
|||
|
|
|||
|
Return Value:
|
|||
|
|
|||
|
none.
|
|||
|
|
|||
|
Environment:
|
|||
|
|
|||
|
Invoked from a router-manager thread with no locks held.
|
|||
|
|
|||
|
--*/
|
|||
|
|
|||
|
{
|
|||
|
PDNS_INTERFACE Interfacep;
|
|||
|
PLIST_ENTRY Link;
|
|||
|
|
|||
|
PROFILE("DnsReactivateEveryInterface");
|
|||
|
|
|||
|
EnterCriticalSection(&DnsInterfaceLock);
|
|||
|
|
|||
|
for (Link = DnsInterfaceList.Flink;
|
|||
|
Link != &DnsInterfaceList;
|
|||
|
Link = Link->Flink
|
|||
|
) {
|
|||
|
|
|||
|
Interfacep = CONTAINING_RECORD(Link, DNS_INTERFACE, Link);
|
|||
|
|
|||
|
if (!DNS_REFERENCE_INTERFACE(Interfacep)) { continue; }
|
|||
|
|
|||
|
if (DNS_INTERFACE_ACTIVE(Interfacep)) {
|
|||
|
DnsDeactivateInterface(Interfacep);
|
|||
|
DnsActivateInterface(Interfacep);
|
|||
|
}
|
|||
|
|
|||
|
DNS_DEREFERENCE_INTERFACE(Interfacep);
|
|||
|
}
|
|||
|
|
|||
|
LeaveCriticalSection(&DnsInterfaceLock);
|
|||
|
|
|||
|
} // DnsReactivateEveryInterface
|
|||
|
|
|||
|
|
|||
|
ULONG NTAPI
|
|||
|
DnspSaveFileWorkerRoutine(
|
|||
|
PVOID Context
|
|||
|
)
|
|||
|
{
|
|||
|
//
|
|||
|
// Context unused
|
|||
|
//
|
|||
|
|
|||
|
PROFILE("DnspSaveFileWorkerRoutine");
|
|||
|
|
|||
|
SaveHostsIcsFile(FALSE);
|
|||
|
|
|||
|
DEREFERENCE_DNS();
|
|||
|
return NO_ERROR;
|
|||
|
} // DnspSaveFileWorkerRoutine
|
|||
|
|
|||
|
|
|||
|
VOID
|
|||
|
DnsShutdownInterfaceManagement(
|
|||
|
VOID
|
|||
|
)
|
|||
|
|
|||
|
/*++
|
|||
|
|
|||
|
Routine Description:
|
|||
|
|
|||
|
This routine is called to shutdown the interface-management module.
|
|||
|
|
|||
|
Arguments:
|
|||
|
|
|||
|
none.
|
|||
|
|
|||
|
Return Value:
|
|||
|
|
|||
|
none.
|
|||
|
|
|||
|
Environment:
|
|||
|
|
|||
|
Invoked in an arbitrary thread context, after all references
|
|||
|
to all interfaces have been released.
|
|||
|
|
|||
|
--*/
|
|||
|
|
|||
|
{
|
|||
|
PDNS_INTERFACE Interfacep;
|
|||
|
PLIST_ENTRY Link;
|
|||
|
|
|||
|
PROFILE("DnsShutdownInterfaceManagement");
|
|||
|
|
|||
|
while (!IsListEmpty(&DnsInterfaceList)) {
|
|||
|
Link = RemoveHeadList(&DnsInterfaceList);
|
|||
|
Interfacep = CONTAINING_RECORD(Link, DNS_INTERFACE, Link);
|
|||
|
if (DNS_INTERFACE_ACTIVE(Interfacep)) {
|
|||
|
DnsDeactivateInterface(Interfacep);
|
|||
|
}
|
|||
|
DnsCleanupInterface(Interfacep);
|
|||
|
}
|
|||
|
|
|||
|
DeleteCriticalSection(&DnsInterfaceLock);
|
|||
|
|
|||
|
} // DnsShutdownInterfaceManagement
|
|||
|
|
|||
|
|
|||
|
VOID
|
|||
|
DnsSignalNatInterface(
|
|||
|
ULONG Index,
|
|||
|
BOOLEAN Boundary
|
|||
|
)
|
|||
|
|
|||
|
/*++
|
|||
|
|
|||
|
Routine Description:
|
|||
|
|
|||
|
This routine is invoked upon reconfiguration of a NAT interface.
|
|||
|
Note that this routine may be invoked even when the DNS proxy
|
|||
|
is neither installed nor running; it operates as expected,
|
|||
|
since the interface list and lock are always initialized.
|
|||
|
|
|||
|
Upon invocation, the routine activates or deactivates the interface
|
|||
|
depending on whether the NAT is not or is running on the interface,
|
|||
|
respectively.
|
|||
|
|
|||
|
Arguments:
|
|||
|
|
|||
|
Index - the reconfigured interface
|
|||
|
|
|||
|
Boundary - indicates whether the interface is now a boundary interface
|
|||
|
|
|||
|
Return Value:
|
|||
|
|
|||
|
none.
|
|||
|
|
|||
|
Environment:
|
|||
|
|
|||
|
Invoked from an arbitrary context.
|
|||
|
|
|||
|
--*/
|
|||
|
|
|||
|
{
|
|||
|
PDNS_INTERFACE Interfacep;
|
|||
|
|
|||
|
PROFILE("DnsSignalNatInterface");
|
|||
|
|
|||
|
EnterCriticalSection(&DnsGlobalInfoLock);
|
|||
|
if (!DnsGlobalInfo) {
|
|||
|
LeaveCriticalSection(&DnsGlobalInfoLock);
|
|||
|
return;
|
|||
|
}
|
|||
|
LeaveCriticalSection(&DnsGlobalInfoLock);
|
|||
|
EnterCriticalSection(&DnsInterfaceLock);
|
|||
|
if (!(Interfacep = DnsLookupInterface(Index, NULL))) {
|
|||
|
LeaveCriticalSection(&DnsInterfaceLock);
|
|||
|
return;
|
|||
|
}
|
|||
|
DnsDeactivateInterface(Interfacep);
|
|||
|
if (DNS_INTERFACE_ACTIVE(Interfacep)) {
|
|||
|
DnsActivateInterface(Interfacep);
|
|||
|
}
|
|||
|
LeaveCriticalSection(&DnsInterfaceLock);
|
|||
|
|
|||
|
} // DnsSignalNatInterface
|
|||
|
|
|||
|
|
|||
|
ULONG
|
|||
|
DnsUnbindInterface(
|
|||
|
ULONG Index
|
|||
|
)
|
|||
|
|
|||
|
/*++
|
|||
|
|
|||
|
Routine Description:
|
|||
|
|
|||
|
This routine is invoked to revoke the binding on an interface.
|
|||
|
This involves deactivating the interface if it is active.
|
|||
|
|
|||
|
Arguments:
|
|||
|
|
|||
|
Index - the index of the interface to be unbound
|
|||
|
|
|||
|
Return Value:
|
|||
|
|
|||
|
none.
|
|||
|
|
|||
|
Environment:
|
|||
|
|
|||
|
Invoked internally in the context of an IP router-manager thread.
|
|||
|
(See 'RMDNS.C').
|
|||
|
|
|||
|
--*/
|
|||
|
|
|||
|
{
|
|||
|
PDNS_INTERFACE Interfacep;
|
|||
|
|
|||
|
PROFILE("DnsUnbindInterface");
|
|||
|
|
|||
|
//
|
|||
|
// Retrieve the interface to be unbound
|
|||
|
//
|
|||
|
|
|||
|
EnterCriticalSection(&DnsInterfaceLock);
|
|||
|
|
|||
|
if (!(Interfacep = DnsLookupInterface(Index, NULL))) {
|
|||
|
LeaveCriticalSection(&DnsInterfaceLock);
|
|||
|
NhTrace(
|
|||
|
TRACE_FLAG_IF,
|
|||
|
"DnsUnbindInterface: interface %d not found",
|
|||
|
Index
|
|||
|
);
|
|||
|
return ERROR_NO_SUCH_INTERFACE;
|
|||
|
}
|
|||
|
|
|||
|
//
|
|||
|
// Make sure the interface is not already unbound
|
|||
|
//
|
|||
|
|
|||
|
if (!DNS_INTERFACE_BOUND(Interfacep)) {
|
|||
|
LeaveCriticalSection(&DnsInterfaceLock);
|
|||
|
NhTrace(
|
|||
|
TRACE_FLAG_IF,
|
|||
|
"DnsUnbindInterface: interface %d already unbound",
|
|||
|
Index
|
|||
|
);
|
|||
|
return ERROR_ADDRESS_NOT_ASSOCIATED;
|
|||
|
}
|
|||
|
|
|||
|
//
|
|||
|
// Reference the interface
|
|||
|
//
|
|||
|
|
|||
|
if (!DNS_REFERENCE_INTERFACE(Interfacep)) {
|
|||
|
LeaveCriticalSection(&DnsInterfaceLock);
|
|||
|
NhTrace(
|
|||
|
TRACE_FLAG_IF,
|
|||
|
"DnsUnbindInterface: interface %d cannot be referenced",
|
|||
|
Index
|
|||
|
);
|
|||
|
return ERROR_INTERFACE_DISABLED;
|
|||
|
}
|
|||
|
|
|||
|
//
|
|||
|
// Clear the 'bound' flag
|
|||
|
//
|
|||
|
|
|||
|
Interfacep->Flags &= ~DNS_INTERFACE_FLAG_BOUND;
|
|||
|
|
|||
|
//
|
|||
|
// Deactivate the interface, if necessary
|
|||
|
//
|
|||
|
|
|||
|
if (DNS_INTERFACE_ENABLED(Interfacep)) {
|
|||
|
DnsDeactivateInterface(Interfacep);
|
|||
|
}
|
|||
|
|
|||
|
LeaveCriticalSection(&DnsInterfaceLock);
|
|||
|
|
|||
|
//
|
|||
|
// Destroy the interface's binding
|
|||
|
//
|
|||
|
|
|||
|
ACQUIRE_LOCK(Interfacep);
|
|||
|
NH_FREE(Interfacep->BindingArray);
|
|||
|
Interfacep->BindingArray = NULL;
|
|||
|
Interfacep->BindingCount = 0;
|
|||
|
RELEASE_LOCK(Interfacep);
|
|||
|
|
|||
|
DNS_DEREFERENCE_INTERFACE(Interfacep);
|
|||
|
return NO_ERROR;
|
|||
|
|
|||
|
} // DnsUnbindInterface
|
|||
|
|
|||
|
|
|||
|
ULONG
|
|||
|
DnsGetPrivateInterfaceAddress(
|
|||
|
VOID
|
|||
|
)
|
|||
|
/*++
|
|||
|
|
|||
|
Routine Description:
|
|||
|
|
|||
|
This routine is invoked to return the IP address on which DNS
|
|||
|
has been enabled.
|
|||
|
|
|||
|
Arguments:
|
|||
|
|
|||
|
none.
|
|||
|
|
|||
|
Return Value:
|
|||
|
|
|||
|
Bound IP address if an address is found (else 0).
|
|||
|
|
|||
|
Environment:
|
|||
|
|
|||
|
Invoked from an arbitrary context.
|
|||
|
|
|||
|
--*/
|
|||
|
{
|
|||
|
PROFILE("DnsGetPrivateInterfaceAddress");
|
|||
|
|
|||
|
ULONG ipAddr = 0;
|
|||
|
ULONG ulRet = NO_ERROR;
|
|||
|
|
|||
|
//
|
|||
|
// Find out the first available interface on which we are enabled and
|
|||
|
// return the primary IP address to which we are bound.
|
|||
|
//
|
|||
|
|
|||
|
PDNS_INTERFACE Interfacep = NULL;
|
|||
|
PLIST_ENTRY Link;
|
|||
|
ULONG i;
|
|||
|
|
|||
|
EnterCriticalSection(&DnsInterfaceLock);
|
|||
|
|
|||
|
for (Link = DnsInterfaceList.Flink;
|
|||
|
Link != &DnsInterfaceList;
|
|||
|
Link = Link->Flink
|
|||
|
)
|
|||
|
{
|
|||
|
Interfacep = CONTAINING_RECORD(Link, DNS_INTERFACE, Link);
|
|||
|
|
|||
|
ACQUIRE_LOCK(Interfacep);
|
|||
|
|
|||
|
for (i = 0; i < Interfacep->BindingCount; i++)
|
|||
|
{
|
|||
|
NhTrace(
|
|||
|
TRACE_FLAG_DNS,
|
|||
|
"DnsGetPrivateInterfaceAddress: IP address %s (Index %d)",
|
|||
|
INET_NTOA(Interfacep->BindingArray[i].Address),
|
|||
|
Interfacep->Index
|
|||
|
);
|
|||
|
|
|||
|
if (Interfacep->BindingArray[i].Address &
|
|||
|
Interfacep->BindingArray[i].Mask)
|
|||
|
{
|
|||
|
ipAddr = Interfacep->BindingArray[i].Address;
|
|||
|
break;
|
|||
|
}
|
|||
|
}
|
|||
|
|
|||
|
RELEASE_LOCK(Interfacep);
|
|||
|
|
|||
|
if (ipAddr)
|
|||
|
{
|
|||
|
LeaveCriticalSection(&DnsInterfaceLock);
|
|||
|
|
|||
|
NhTrace(
|
|||
|
TRACE_FLAG_DNS,
|
|||
|
"DnsGetPrivateInterfaceAddress: Dns private interface IP address %s (Index %d)",
|
|||
|
INET_NTOA(ipAddr),
|
|||
|
Interfacep->Index
|
|||
|
);
|
|||
|
|
|||
|
return ipAddr;
|
|||
|
}
|
|||
|
}
|
|||
|
|
|||
|
LeaveCriticalSection(&DnsInterfaceLock);
|
|||
|
|
|||
|
return ipAddr;
|
|||
|
} // DnsGetPrivateInterfaceAddress
|
|||
|
|