able/windows-nt
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity
windows-nt/Source/XPSP1/NT/net/rras/ip/wanarp2/driver.c

1805 lines
38 KiB
C
Raw Permalink Normal View History

Add source files
2020-09-26 03:20:57 -05:00
/*++
Copyright (c) 1995 Microsoft Corporation
Module Name:
net\ip\wanarp\driver.c
Abstract:
WAN ARP driver shell.
Revision History:
Gurdeep Singh Pall 8/2/95 Created
--*/
#define __FILE_SIG__ DRIVER_SIG
#include "inc.h"
#pragma hdrstop
BOOLEAN g_bExit;
BOOLEAN g_bPoolsInitialized;
NPAGED_LOOKASIDE_LIST g_llNotificationBlocks;
NPAGED_LOOKASIDE_LIST g_llConnBlocks;
#pragma alloc_text(INIT, DriverEntry)
NTSTATUS
DriverEntry(
IN PDRIVER_OBJECT DriverObject,
IN PUNICODE_STRING RegistryPath
)
/*++
Routine Description:
Installable driver initialization entry point.
This entry point is called directly by the I/O system and must be named
"Driver Entry"
The function is discardable since it is only called once
On checked builds we read some values from registry and initialize the
debugging
We create a DEVICE_OBJECT for ourselves to field the IOCTLs, create
a DOS name for the device and initialize some events and spinlocks
Locks:
None
Arguments:
DriverObject Pointer to I/O subsystem created driver object
RegistryPath Points to driver key in HKLM\System\CCS\Services...
Return Value:
STATUS_SUCCESS if everything went as planned or some status code from
ntstatus.h
--*/
{
NTSTATUS nStatus;
PDEVICE_OBJECT pDeviceObject;
UNICODE_STRING usDeviceName, usParamString, usTempString;
DWORD dwVal, i;
HANDLE hRegKey;
USHORT usRegLen;
PWCHAR pwcBuffer;
RtInitializeDebug();
usRegLen = RegistryPath->Length +
(sizeof(WCHAR) * (wcslen(L"\\Parameters") + 2));
pwcBuffer = RtAllocate(NonPagedPool,
usRegLen,
WAN_STRING_TAG);
if(pwcBuffer is NULL)
{
return STATUS_INSUFFICIENT_RESOURCES;
}
RtlZeroMemory(pwcBuffer,
usRegLen);
usParamString.MaximumLength = usRegLen;
usParamString.Buffer = pwcBuffer;
RtlCopyUnicodeString(&usParamString,
RegistryPath);
RtlInitUnicodeString(&usTempString,
L"\\Parameters");
RtlAppendUnicodeStringToString(&usParamString,
&usTempString);
nStatus = OpenRegKey(&hRegKey,
&usParamString);
RtFree(pwcBuffer);
if(nStatus is STATUS_SUCCESS)
{
#if RT_TRACE_DEBUG
nStatus = GetRegDWORDValue(hRegKey,
L"DebugLevel",
&dwVal);
if(nStatus is STATUS_SUCCESS)
{
g_byDebugLevel = (BYTE)dwVal;
}
nStatus = GetRegDWORDValue(hRegKey,
L"DebugComp",
&dwVal);
if(nStatus is STATUS_SUCCESS)
{
g_fDebugComp = dwVal;
}
#endif
#if DBG
nStatus = GetRegDWORDValue(hRegKey,
L"DebugBreak",
&dwVal);
if((nStatus is STATUS_SUCCESS) and
(dwVal is 1))
{
DbgBreakPoint();
}
#endif
ZwClose(hRegKey);
}
TraceEnter(GLOBAL, "DriverEntry");
//DbgBreakPoint();
if(g_bExit)
{
return STATUS_UNSUCCESSFUL;
}
//
// Initialize some globals (rest are all 0s)
//
g_dwDriverState = DRIVER_STOPPED;
//
// Create the device
//
RtlInitUnicodeString(&usDeviceName,
DD_WANARP_DEVICE_NAME_W);
nStatus = IoCreateDevice(DriverObject,
0,
&usDeviceName,
FILE_DEVICE_NETWORK,
FILE_DEVICE_SECURE_OPEN,
FALSE,
&pDeviceObject);
if(nStatus isnot STATUS_SUCCESS)
{
Trace(GLOBAL, ERROR,
("DriverEntry: Cant create device object %S, status %x.\n",
DD_WANARP_DEVICE_NAME_W,
nStatus));
TraceLeave(GLOBAL, "DriverEntry");
return nStatus;
}
//
// Initialize the driver object
//
DriverObject->DriverUnload = WanUnload;
DriverObject->FastIoDispatch = NULL;
for(i=0; i <= IRP_MJ_MAXIMUM_FUNCTION; i++)
{
DriverObject->MajorFunction[i] = WanDispatch;
}
//
// Initialize Events etc
//
WanpInitializeDriverStructures();
if(!WanpSetupExternalName(&usDeviceName,
WIN32_WANARP_SYMBOLIC_LINK,
TRUE))
{
Trace(GLOBAL,ERROR,
("DriverEntry: Win32 device name %S could not be created\n",
WIN32_WANARP_SYMBOLIC_LINK));
IoDeleteDevice(pDeviceObject);
TraceLeave(GLOBAL, "DriverEntry");
return STATUS_UNSUCCESSFUL;
}
//
// Register with IP
//
WanpAcquireResource(&g_wrBindMutex);
nStatus = WanpRegisterWithIp();
if(nStatus isnot STATUS_SUCCESS)
{
Trace(GLOBAL, ERROR,
("DriverEntry: RegisterWithIp failed %x\n",
nStatus));
WanpSetupExternalName(&usDeviceName,
WIN32_WANARP_SYMBOLIC_LINK,
FALSE);
IoDeleteDevice(pDeviceObject);
WanpReleaseResource(&g_wrBindMutex);
TraceLeave(GLOBAL, "InitializeDriver");
return FALSE;
}
nStatus = WanpInitializeNdis();
if(nStatus isnot STATUS_SUCCESS)
{
Trace(GLOBAL, ERROR,
("DriverEntry: IntializeNdis failed %x\n",
nStatus));
WanpDeregisterWithIp();
WanpSetupExternalName(&usDeviceName,
WIN32_WANARP_SYMBOLIC_LINK,
FALSE);
IoDeleteDevice(pDeviceObject);
WanpReleaseResource(&g_wrBindMutex);
TraceLeave(GLOBAL, "InitializeDriver");
return FALSE;
}
WanpReleaseResource(&g_wrBindMutex);
TraceLeave(GLOBAL, "DriverEntry");
return nStatus;
}
#pragma alloc_text(PAGE, WanDispatch)
NTSTATUS
WanDispatch(
IN PDEVICE_OBJECT DeviceObject,
IN PIRP Irp
)
/*++
Routine Description:
The functions which handles the IRPs sent to the driver
Locks:
This code is PAGEABLE so can not acquire locks
Arguments:
Return Value:
STATUS_SUCCESS
--*/
{
PIO_STACK_LOCATION irpStack;
ULONG ulInputBuffLen;
ULONG ulOutputBuffLen;
ULONG ioControlCode;
NTSTATUS nStatus;
KIRQL kiIrql;
LARGE_INTEGER liTimeOut;
BOOLEAN bEnter;
PAGED_CODE();
TraceEnter(GLOBAL, "WanDispatch");
Irp->IoStatus.Information = 0;
//
// Get a pointer to the current location in the Irp. This is where
// the function codes and parameters are located.
//
irpStack = IoGetCurrentIrpStackLocation(Irp);
switch (irpStack->MajorFunction)
{
case IRP_MJ_CREATE:
{
Trace(GLOBAL, TRACE,
("WanDispatch: IRP_MJ_CREATE\n"));
//
// We start the driver when the first CreateFile is done
// But we need to serialize the Creates
//
nStatus = WanpStartDriver();
if(nStatus isnot STATUS_SUCCESS)
{
//
// If pending, wait on the start event
//
if(nStatus is STATUS_PENDING)
{
}
else
{
break;
}
}
//
// Make sure the driver actually started
//
bEnter = EnterDriverCode();
if(!bEnter)
{
Trace(GLOBAL, ERROR,
("WanDispatch: Wait successful, but unable to start driver\n"));
nStatus = STATUS_UNSUCCESSFUL;
}
else
{
ExitDriverCode();
}
break;
}
case IRP_MJ_CLOSE:
{
Trace(GLOBAL, TRACE,
("WanDispatch: IRP_MJ_CLOSE\n"));
//
// We handle cleanup and not close
//
nStatus = STATUS_SUCCESS;
break;
}
case IRP_MJ_CLEANUP:
{
Trace(GLOBAL, TRACE,
("WanDispatch: IRP_MJ_CLEANUP\n"));
nStatus = STATUS_SUCCESS;
if(InterlockedDecrement(&g_ulNumCreates) is 0)
{
//
// Last handle open is now closed; lets clean up
//
WanpStopDriver();
}
break;
}
case IRP_MJ_DEVICE_CONTROL:
{
DWORD dwState;
ULONG ulControl;
//
// Get the control code and our code
//
ioControlCode = irpStack->Parameters.DeviceIoControl.IoControlCode;
//
// Get the pointer to the input/output buffer and it's length
//
ulInputBuffLen =
irpStack->Parameters.DeviceIoControl.InputBufferLength;
ulOutputBuffLen =
irpStack->Parameters.DeviceIoControl.OutputBufferLength;
//
// If the driver is stopping, dont process anything else
//
bEnter = EnterDriverCode();
if(!bEnter)
{
Trace(GLOBAL, ERROR,
("WanDispatch: Driver is not started\n"));
nStatus = STATUS_NO_SUCH_DEVICE;
break;
}
switch(ioControlCode)
{
case IOCTL_WANARP_NOTIFICATION:
{
nStatus = WanProcessNotification(Irp,
ulInputBuffLen,
ulOutputBuffLen);
break;
}
case IOCTL_WANARP_ADD_INTERFACE:
{
nStatus = WanAddUserModeInterface(Irp,
ulInputBuffLen,
ulOutputBuffLen);
break;
}
case IOCTL_WANARP_DELETE_INTERFACE:
{
nStatus = WanDeleteUserModeInterface(Irp,
ulInputBuffLen,
ulOutputBuffLen);
break;
}
case IOCTL_WANARP_CONNECT_FAILED:
{
nStatus = WanProcessConnectionFailure(Irp,
ulInputBuffLen,
ulOutputBuffLen);
break;
}
case IOCTL_WANARP_GET_IF_STATS:
{
nStatus = WanGetIfStats(Irp,
ulInputBuffLen,
ulOutputBuffLen);
break;
}
case IOCTL_WANARP_DELETE_ADAPTERS:
{
nStatus = WanDeleteAdapters(Irp,
ulInputBuffLen,
ulOutputBuffLen);
break;
}
case IOCTL_WANARP_MAP_SERVER_ADAPTER:
{
nStatus = WanMapServerAdapter(Irp,
ulInputBuffLen,
ulOutputBuffLen);
break;
}
case IOCTL_WANARP_QUEUE:
{
nStatus = WanStartStopQueuing(Irp,
ulInputBuffLen,
ulOutputBuffLen);
break;
}
default:
{
Trace(GLOBAL, ERROR,
("WanDispatch: Unknown IRP_MJ_DEVICE_CONTROL %x\n",
ioControlCode));
nStatus = STATUS_INVALID_PARAMETER;
}
}
ExitDriverCode();
break;
}
default:
{
Trace(GLOBAL, ERROR,
("WanDispatch: Unknown IRP_MJ_XX - %x\n",
irpStack->MajorFunction));
nStatus = STATUS_INVALID_PARAMETER;
break;
}
}
if(nStatus isnot STATUS_PENDING)
{
Irp->IoStatus.Status = nStatus;
IoCompleteRequest(Irp,
IO_NETWORK_INCREMENT);
}
TraceLeave(GLOBAL, "WanDispatch");
return nStatus;
}
#pragma alloc_text(PAGE, WanUnload)
VOID
WanUnload(
PDRIVER_OBJECT DriverObject
)
/*++
Routine Description:
Called by the I/O subsystem, when our driver is being unloaded
Locks:
Arguments:
Return Value:
None
--*/
{
UNICODE_STRING usDeviceName;
BOOLEAN bWait;
KIRQL kiIrql;
NDIS_STATUS nsStatus;
NTSTATUS nStatus;
TraceEnter(GLOBAL,"WanUnload");
PAGED_CODE();
//
// The driver must have stopped before it came here
//
RtAssert(g_dwDriverState is DRIVER_STOPPED);
//
// Deregister from NDIS etc
//
WanpDeinitializeNdis();
//
// Clear out IP's state. Need to do this after all adapters have
// been removed
//
WanpDeregisterWithIp();
//
// Remove ourself from NT and DOS namespace
//
RtlInitUnicodeString(&usDeviceName,
DD_WANARP_DEVICE_NAME_W);
WanpSetupExternalName(&usDeviceName,
WIN32_WANARP_SYMBOLIC_LINK,
FALSE);
//
// Free all our structures
//
if(g_puipConnTable)
{
RtFree(g_puipConnTable);
}
ExDeleteNPagedLookasideList(&g_llConnBlocks);
ExDeleteNPagedLookasideList(&g_llNotificationBlocks);
if(g_bPoolsInitialized)
{
FreeBufferPool(&g_bpHeaderBufferPool);
FreeBufferPool(&g_bpDataBufferPool);
g_bPoolsInitialized = FALSE;
}
//
// Acquire and release the resource. This lets the close adapter thread
// run if it is still around
//
WanpAcquireResource(&g_wrBindMutex);
WanpReleaseResource(&g_wrBindMutex);
//
// See if we have any free memory
//
RtAuditMemory();
//
// Delete the device object
//
IoDeleteDevice(DriverObject->DeviceObject);
TraceLeave(GLOBAL,"WanUnload");
}
#pragma alloc_text(PAGE, WanpSetupExternalName)
BOOLEAN
WanpSetupExternalName(
PUNICODE_STRING pusNtName,
PWCHAR pwcDosName,
BOOLEAN bCreate
)
/*++
Routine Description:
Setup or delete a symbolic link to DOS namespace
Locks:
Arguments:
pusNtName Name in NT space
pwcDosName Name in DOS space
bCreate Set to TRUE to create, FALSE to delete
Return Value:
TRUE if successful
FALSE otherwise
--*/
{
UNICODE_STRING usSymbolicLinkName;
WCHAR rgwcBuffer[100];
PAGED_CODE();
//
// Form the full symbolic link name we wish to create.
//
usSymbolicLinkName.Buffer = rgwcBuffer;
RtlInitUnicodeString(&usSymbolicLinkName,
pwcDosName);
if(bCreate)
{
if(!NT_SUCCESS(IoCreateSymbolicLink(&usSymbolicLinkName,
pusNtName)))
{
return FALSE;
}
}
else
{
IoDeleteSymbolicLink(&usSymbolicLinkName);
}
return TRUE;
}
NTSTATUS
WanpStartDriver(
VOID
)
/*++
Routine Description:
Main routine to start the driver. We call this when we get a CREATE irp
If the driver has started, we return success. If someone is starting the
driver, we return pending. The caller then needs to wait on g_keStartEvent
We try and start the driver. If all goes well, we set the event and
everyone parties on from there
Locks:
The function takes the g_rlStateLock to check the state and increment
the number of CREATEs it has received (open handles)
Arguments:
None
Return Value:
STATUS_SUCCESS if the driver started
STATUS_PENDING if the driver is being started by some other thread's
--*/
{
KIRQL kiOldIrql;
NTSTATUS nStatus;
DWORD dwState;
TraceEnter(GLOBAL, "StartDriver");
RtAcquireSpinLock(&g_rlStateLock,
&kiOldIrql);
g_ulNumCreates++;
if(g_ulNumCreates isnot 1)
{
if(g_dwDriverState is DRIVER_STARTING)
{
//
// Someone is trying to start the driver
//
Trace(GLOBAL, INFO,
("StartDriver: Driver is being started by someone else\n"));
nStatus = STATUS_PENDING;
}
else
{
//
// If we are not the first CreateFile, and the driver is not
// starting then the driver must already be running
//
RtAssert(g_dwDriverState is DRIVER_STARTED);
nStatus = STATUS_SUCCESS;
}
RtReleaseSpinLock(&g_rlStateLock,
kiOldIrql);
return nStatus;
}
//
// The first CreateFile
//
RtAssert(g_dwDriverState is DRIVER_STOPPED);
//
// Set the state to starting, release the lock and actually start
// the driver
//
g_dwDriverState = DRIVER_STARTING;
RtReleaseSpinLock(&g_rlStateLock,
kiOldIrql);
//
// Must be called at PASSIVE because it waits for IP to finish
//
WanpSetDemandDialCallback(TRUE);
RtAcquireSpinLock(&g_rlStateLock,
&kiOldIrql);
g_dwDriverState = DRIVER_STARTED;
//
// Someone may have been waiting for us to start
//
KeSetEvent(&g_keStartEvent,
0,
FALSE);
RtReleaseSpinLock(&g_rlStateLock,
kiOldIrql);
TraceLeave(GLOBAL, "StartDriver");
return STATUS_SUCCESS;
}
VOID
WanpStopDriver(
VOID
)
/*++
Routine Description:
Called when we get an IRP_MJ_CLEANUP. It is the inverse of StartDriver
If this is the last thread, we set the state to STOPPED and wait till
all threads of execution have exited the driver.
We then clean out resources
Locks:
The function takes the g_rlStateLock
Arguments:
None
Return Value:
None
--*/
{
KIRQL kiOldIrql;
NTSTATUS nStatus;
BOOLEAN bWait;
TraceEnter(GLOBAL, "StopDriver");
//
// Acquire the state and ref count spin lock
//
RtAcquireSpinLock(&g_rlStateLock,
&kiOldIrql);
RtAssert(g_ulNumCreates is 0);
//
// Set the state to stopping. Any reader will
// return on seeing this. So essentially we are not
// allowing any new readers in
//
g_dwDriverState = DRIVER_STOPPED;
//
// However there may already be readers. We wait
// if there are any
//
bWait = (g_ulNumThreads > 0);
RtReleaseSpinLock(&g_rlStateLock,
kiOldIrql);
//
// Now do a wait. We can do this since we are at PASSIVE
//
if(bWait)
{
Trace(GLOBAL, INFO,
("StopDriver: Need to wait for threads to exit\n"));
do
{
nStatus = KeWaitForSingleObject(&g_keStateEvent,
Executive,
KernelMode,
FALSE,
NULL);
}while((nStatus is STATUS_USER_APC) or
(nStatus is STATUS_ALERTED) or
(nStatus is STATUS_TIMEOUT));
}
//
// Cleanup all resources
//
WanpCleanOutInterfaces();
WanpClearPendingIrps();
TraceLeave(GLOBAL, "StopDriver");
}
NTSTATUS
WanpRegisterWithIp(
VOID
)
/*++
Routine Description:
Registers the ARP module with IP
Locks:
Arguments:
Return Value:
--*/
{
NDIS_STRING nsWanName;
WCHAR pwszName[] = WANARP_ARP_NAME;
TraceEnter(GLOBAL, "WanpRegisterWithIp");
nsWanName.MaximumLength = sizeof(pwszName);
nsWanName.Length = sizeof(WANARP_ARP_NAME) - sizeof(WCHAR);
nsWanName.Buffer = pwszName;
if(IPRegisterARP(&nsWanName,
IP_ARP_BIND_VERSION,
WanIpBindAdapter,
&g_pfnIpAddInterface,
&g_pfnIpDeleteInterface,
&g_pfnIpBindComplete,
&g_pfnIpAddLink,
&g_pfnIpDeleteLink,
&g_pfnIpChangeIndex,
&g_pfnIpReserveIndex,
&g_pfnIpDereserveIndex,
&g_hIpRegistration) isnot IP_SUCCESS)
{
Trace(GLOBAL, ERROR,
("WanpRegisterWithIp: Couldnt register with IP\n"));
TraceLeave(GLOBAL,
"WanpRegisterWithIp");
return STATUS_UNSUCCESSFUL;
}
TraceLeave(GLOBAL,
"WanpRegisterWithIp");
return STATUS_SUCCESS;
}
VOID
WanpDeregisterWithIp(
VOID
)
/*++
Routine Description:
DeRegisters the ARP module with IP
Locks:
Arguments:
Return Value:
--*/
{
NTSTATUS nStatus;
TraceEnter(GLOBAL, "WanpDeregisterWithIP");
nStatus = IPDeregisterARP(g_hIpRegistration);
if(nStatus isnot STATUS_SUCCESS)
{
Trace(GLOBAL, ERROR,
("WanpDeregisterWithIP: Couldnt deregister with IP. Error %x\n",
nStatus));
}
g_pfnIpAddInterface = NULL;
g_pfnIpDeleteInterface = NULL;
g_pfnIpBindComplete = NULL;
g_pfnIpRcv = NULL;
g_pfnIpRcvComplete = NULL;
g_pfnIpSendComplete = NULL;
g_pfnIpTDComplete = NULL;
g_pfnIpStatus = NULL;
g_pfnIpRcvPkt = NULL;
g_pfnIpPnp = NULL;
g_hIpRegistration = NULL;
TraceLeave(GLOBAL, "WanpDeregisterWithIP");
}
BOOLEAN
EnterDriverCode(
VOID
)
{
KIRQL irql;
BOOLEAN bEnter;
ULONG i;
//
// If we havent received any binds, sleep for 30 seconds
//
i = 0;
while((g_lBindRcvd is 0) and
(i < 3))
{
LARGE_INTEGER liTimeOut;
i++;
//
// Set this to 5,10,15 seconds
//
liTimeOut.QuadPart = (LONGLONG)((i+ 1) * 5 * 1000 * 1000 * 10 * -1);
KeDelayExecutionThread(UserMode,
FALSE,
&liTimeOut);
}
if(g_lBindRcvd is 0)
{
return FALSE;
}
RtAcquireSpinLock(&g_rlStateLock,
&irql);
if(g_dwDriverState is DRIVER_STARTED)
{
g_ulNumThreads++;
bEnter = TRUE;
}
else
{
bEnter = FALSE;
}
RtReleaseSpinLock(&g_rlStateLock,
irql);
return bEnter;
}
VOID
ExitDriverCode(
VOID
)
{
KIRQL irql;
RtAcquireSpinLock(&g_rlStateLock,
&irql);
g_ulNumThreads--;
if((g_dwDriverState is DRIVER_STOPPED) and
(g_ulNumThreads is 0))
{
KeSetEvent(&g_keStateEvent,
0,
FALSE);
}
RtReleaseSpinLock(&g_rlStateLock,
irql);
}
#pragma alloc_text(PAGE, WanpSetDemandDialCallback)
VOID
WanpSetDemandDialCallback(
BOOLEAN bSetPointer
)
/*++
Routine Description:
Sets the pointer to our demand dial request routine with the IP Stack
Locks:
None, done at INIT time
Arguments:
bSetPointer TRUE: set the pointer
FALSE: remove the pointer
Return Value:
None
--*/
{
PIRP irp;
HANDLE handle;
NTSTATUS nStatus;
KEVENT tempevent;
PFILE_OBJECT fileObject;
PDEVICE_OBJECT deviceObject;
IO_STATUS_BLOCK ioStatusBlock;
UNICODE_STRING usDeviceName;
IP_SET_MAP_ROUTE_HOOK_INFO buffer;
PAGED_CODE();
//
// Open IP driver
//
RtlInitUnicodeString(&usDeviceName,
DD_IP_DEVICE_NAME);
nStatus = IoGetDeviceObjectPointer(&usDeviceName,
SYNCHRONIZE|GENERIC_READ|GENERIC_WRITE,
&fileObject,
&deviceObject);
if(nStatus isnot STATUS_SUCCESS)
{
Trace(CONN, ERROR,
("SetDemandDialCallback: IoGetDeviceObjectPointer for %S failed with status %x\n",
DD_IP_DEVICE_NAME,
nStatus));
return;
}
//
// Reference the device object.
//
ObReferenceObject(deviceObject);
//
// IoGetDeviceObjectPointer put a reference on the file object.
//
ObDereferenceObject(fileObject);
//
// Allocate event to use for setting callback address
//
KeInitializeEvent(&tempevent,
SynchronizationEvent,
FALSE);
if(bSetPointer)
{
buffer.MapRoutePtr = WanDemandDialRequest;
}
else
{
buffer.MapRoutePtr = NULL;
}
//
// Build the IRP
//
irp = IoBuildDeviceIoControlRequest(IOCTL_IP_SET_MAP_ROUTE_POINTER,
deviceObject,
&buffer,
sizeof(IP_SET_MAP_ROUTE_HOOK_INFO),
NULL,
0,
FALSE,
&tempevent,
&ioStatusBlock);
if (irp is NULL)
{
ObDereferenceObject(deviceObject);
Trace(GLOBAL, ERROR,
("SetDemandDialCallback: Couldnt build IRP\n"));
}
else
{
nStatus = IoCallDriver(deviceObject, irp);
if(nStatus is STATUS_PENDING)
{
nStatus = KeWaitForSingleObject(&tempevent,
Executive,
KernelMode,
FALSE,
NULL);
}
//
// "close" handle to IP driver
//
ObDereferenceObject(deviceObject);
}
}
#pragma alloc_text(PAGE, WanpInitializeDriverStructures)
BOOLEAN
WanpInitializeDriverStructures(
VOID
)
/*++
Routine Description:
Initializes the internal driver structures
Locks:
None, called at init time
Arguments:
None
Return Value:
None
--*/
{
ULONG i;
PAGED_CODE();
//
// Lock and event needed to keep track of threads in the driver
//
RtInitializeSpinLock(&g_rlStateLock);
KeInitializeEvent(&g_keStateEvent,
SynchronizationEvent,
FALSE);
KeInitializeEvent(&g_keStartEvent,
NotificationEvent,
FALSE);
KeInitializeEvent(&g_keCloseEvent,
SynchronizationEvent,
FALSE);
WanpInitializeResource(&g_wrBindMutex);
InitRwLock(&g_rwlIfLock);
InitRwLock(&g_rwlAdapterLock);
InitializeListHead(&g_leIfList);
InitializeListHead(&g_lePendingNotificationList);
InitializeListHead(&g_lePendingIrpList);
InitializeListHead(&g_leMappedAdapterList);
InitializeListHead(&g_leChangeAdapterList);
InitializeListHead(&g_leAddedAdapterList);
InitializeListHead(&g_leFreeAdapterList);
//
// Initialize the connection table
//
g_puipConnTable = RtAllocate(NonPagedPool,
WAN_INIT_CONN_TABLE_SIZE * sizeof(ULONG_PTR),
WAN_CONN_TAG);
if(g_puipConnTable is NULL)
{
Trace(GLOBAL, ERROR,
("InitDriverStructures: Couldnt alloc conn table of %d bytes\n",
WAN_INIT_CONN_TABLE_SIZE * sizeof(ULONG_PTR)));
return FALSE;
}
RtlZeroMemory(g_puipConnTable,
WAN_INIT_CONN_TABLE_SIZE * sizeof(ULONG_PTR));
g_ulConnTableSize = WAN_INIT_CONN_TABLE_SIZE;
//
// The first slot is never used
//
g_puipConnTable[0] = (ULONG_PTR)-1;
g_ulNextConnIndex = 1;
RtInitializeSpinLock(&g_rlConnTableLock);
//
// Initialize the lookaside list for connection entries
//
ExInitializeNPagedLookasideList(&g_llConnBlocks,
NULL,
NULL,
0,
sizeof(CONN_ENTRY),
WAN_CONN_TAG,
WANARP_CONN_LOOKASIDE_DEPTH);
ExInitializeNPagedLookasideList(&g_llNotificationBlocks,
NULL,
NULL,
0,
sizeof(PENDING_NOTIFICATION),
WAN_NOTIFICATION_TAG,
WANARP_NOTIFICATION_LOOKASIDE_DEPTH);
//
// Create the buffer pools
//
InitBufferPool(&g_bpHeaderBufferPool,
HEADER_BUFFER_SIZE,
0,
20,
0,
TRUE,
WAN_HEADER_TAG);
InitBufferPool(&g_bpDataBufferPool,
DATA_BUFFER_SIZE,
0,
10,
0,
TRUE,
WAN_DATA_TAG);
g_bPoolsInitialized = TRUE;
return TRUE;
}
#pragma alloc_text(PAGE, WanpInitializeNdis)
NDIS_STATUS
WanpInitializeNdis(
VOID
)
{
NDIS_STATUS nsStatus;
NDIS_PROTOCOL_CHARACTERISTICS npcWanChar;
TraceEnter(GLOBAL, "InitializeNdis");
PAGED_CODE();
RtlZeroMemory(&npcWanChar,
sizeof(NDIS_PROTOCOL_CHARACTERISTICS));
npcWanChar.MajorNdisVersion = 4;
npcWanChar.MinorNdisVersion = 0;
npcWanChar.Flags = 0;
npcWanChar.OpenAdapterCompleteHandler = WanNdisOpenAdapterComplete;
npcWanChar.CloseAdapterCompleteHandler = WanNdisCloseAdapterComplete;
npcWanChar.SendCompleteHandler = WanNdisSendComplete;
npcWanChar.TransferDataCompleteHandler = WanNdisTransferDataComplete;
npcWanChar.ResetCompleteHandler = WanNdisResetComplete;
npcWanChar.RequestCompleteHandler = WanNdisRequestComplete;
npcWanChar.ReceiveHandler = WanNdisReceive;
npcWanChar.ReceiveCompleteHandler = WanNdisReceiveComplete;
npcWanChar.StatusHandler = WanNdisStatus;
npcWanChar.StatusCompleteHandler = WanNdisStatusComplete;
npcWanChar.ReceivePacketHandler = WanNdisReceivePacket;
//
// No bind handler needed
//
npcWanChar.BindAdapterHandler = WanNdisBindAdapter;
npcWanChar.UnbindAdapterHandler = WanNdisUnbindAdapter;
npcWanChar.PnPEventHandler = WanNdisPnPEvent;
npcWanChar.UnloadHandler = WanNdisUnload;
//
// Allocate the Packet Pool
//
g_nhPacketPool = (NDIS_HANDLE)WAN_PACKET_TAG;
NdisAllocatePacketPoolEx(&nsStatus,
&g_nhPacketPool,
WAN_PACKET_POOL_COUNT,
WAN_PACKET_POOL_OVERFLOW,
sizeof(struct PCCommon));
if(nsStatus isnot NDIS_STATUS_SUCCESS)
{
Trace(GLOBAL, ERROR,
("InitializeNdis: Unable to allocate packet pool. %x\n",
nsStatus));
g_nhPacketPool = NULL;
return nsStatus;
}
NdisSetPacketPoolProtocolId(g_nhPacketPool,
NDIS_PROTOCOL_ID_TCP_IP);
RtlInitUnicodeString(&npcWanChar.Name,
WANARP_NDIS_NAME);
//
// THIS MUST BE THE LAST THING DONE
//
NdisRegisterProtocol(&nsStatus,
&g_nhWanarpProtoHandle,
&npcWanChar,
sizeof(npcWanChar));
TraceLeave(GLOBAL, "InitializeNdis");
return nsStatus;
}
#pragma alloc_text(PAGE, WanpDeinitializeNdis)
VOID
WanpDeinitializeNdis(
VOID
)
{
NTSTATUS nStatus;
LARGE_INTEGER liTimeOut;
PAGED_CODE();
WanpAcquireResource(&g_wrBindMutex);
if(g_nhNdiswanBinding isnot NULL)
{
WanpCloseNdisWan(NULL);
}
else
{
WanpReleaseResource(&g_wrBindMutex);
}
//
// wait on the close event
//
nStatus = KeWaitForSingleObject(&g_keCloseEvent,
Executive,
KernelMode,
FALSE,
NULL);
//
// Wait for a 5 secs to let the ndis thread finish its stuff
//
/*liTimeOut.QuadPart = (LONGLONG)(5 * 1000 * 1000 * 10 * -1);
KeDelayExecutionThread(UserMode,
FALSE,
&liTimeOut);
*/
if(g_nhPacketPool isnot NULL)
{
NdisFreePacketPool(g_nhPacketPool);
g_nhPacketPool = NULL;
}
NdisDeregisterProtocol(&nStatus,
g_nhWanarpProtoHandle);
g_nhWanarpProtoHandle = NULL;
}
VOID
WanNdisUnload(
VOID
)
{
return;
}
#pragma alloc_text(PAGE, OpenRegKey)
NTSTATUS
OpenRegKey(
OUT PHANDLE phHandle,
IN PUNICODE_STRING pusKeyName
)
{
NTSTATUS Status;
OBJECT_ATTRIBUTES ObjectAttributes;
PAGED_CODE();
RtlZeroMemory(&ObjectAttributes,
sizeof(OBJECT_ATTRIBUTES));
InitializeObjectAttributes(&ObjectAttributes,
pusKeyName,
OBJ_CASE_INSENSITIVE,
NULL,
NULL);
Status = ZwOpenKey(phHandle,
KEY_READ,
&ObjectAttributes);
return Status;
}
#pragma alloc_text(PAGE, GetRegDWORDValue)
NTSTATUS
GetRegDWORDValue(
HANDLE KeyHandle,
PWCHAR ValueName,
PULONG ValueData
)
{
NTSTATUS status;
ULONG resultLength;
PKEY_VALUE_FULL_INFORMATION keyValueFullInformation;
UCHAR keybuf[128];
UNICODE_STRING UValueName;
PAGED_CODE();
RtlInitUnicodeString(&UValueName, ValueName);
keyValueFullInformation = (PKEY_VALUE_FULL_INFORMATION)keybuf;
RtlZeroMemory(keyValueFullInformation, sizeof(keyValueFullInformation));
status = ZwQueryValueKey(KeyHandle,
&UValueName,
KeyValueFullInformation,
keyValueFullInformation,
128,
&resultLength);
if (NT_SUCCESS(status)) {
if (keyValueFullInformation->Type != REG_DWORD) {
status = STATUS_INVALID_PARAMETER_MIX;
} else {
*ValueData = *((ULONG UNALIGNED *)((PCHAR)keyValueFullInformation +
keyValueFullInformation->DataOffset));
}
}
return status;
}
BOOLEAN
IsEntryOnList(
PLIST_ENTRY pleHead,
PLIST_ENTRY pleEntry
)
{
PLIST_ENTRY pleNode;
for(pleNode = pleHead->Flink;
pleNode isnot pleHead;
pleNode = pleNode->Flink)
{
if(pleNode is pleEntry)
{
return TRUE;
}
}
return FALSE;
}
VOID
WanpInitializeResource(
IN PWAN_RESOURCE pLock
)
{
pLock->lWaitCount = 0;
KeInitializeEvent(&(pLock->keEvent),
SynchronizationEvent,
FALSE);
}
VOID
WanpAcquireResource(
IN PWAN_RESOURCE pLock
)
{
LONG lNumWaiters;
NTSTATUS nStatus;
lNumWaiters = InterlockedIncrement(&(pLock->lWaitCount));
RtAssert(lNumWaiters >= 1);
if(lNumWaiters isnot 1)
{
nStatus = KeWaitForSingleObject(&(pLock->keEvent),
Executive,
KernelMode,
FALSE,
NULL);
}
}
VOID
WanpReleaseResource(
IN PWAN_RESOURCE pLock
)
{
LONG lNumWaiters;
lNumWaiters = InterlockedDecrement(&(pLock->lWaitCount));
RtAssert(lNumWaiters >= 0);
if(lNumWaiters isnot 0)
{
KeSetEvent(&(pLock->keEvent),
0,
FALSE);
}
}
VOID
WanpClearPendingIrps(
VOID
)
/*++
Routine Description:
Called at cleanup time to return any pending IRPs
Locks:
Acquires the IoCancelSpinLock since it controls the pending irp list
Arguments:
None
Return Value:
None
--*/
{
KIRQL irql;
TraceEnter(GLOBAL, "ClearPendingIrps");
IoAcquireCancelSpinLock (&irql);
while(!IsListEmpty(&g_lePendingIrpList))
{
PLIST_ENTRY pleNode;
PIRP pIrp;
pleNode = RemoveHeadList(&g_lePendingIrpList);
pIrp = CONTAINING_RECORD(pleNode,
IRP,
Tail.Overlay.ListEntry);
IoSetCancelRoutine(pIrp,
NULL);
pIrp->IoStatus.Status = STATUS_NO_SUCH_DEVICE;
pIrp->IoStatus.Information = 0;
//
// release lock to complete the IRP
//
IoReleaseCancelSpinLock(irql);
IoCompleteRequest(pIrp,
IO_NETWORK_INCREMENT);
//
// Reaquire the lock
//
IoAcquireCancelSpinLock(&irql);
}
IoReleaseCancelSpinLock(irql);
}
Reference in a new issue Copy permalink