able/windows-nt
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity
windows-nt/Source/XPSP1/NT/net/netbeui/sys/info.c

3518 lines
94 KiB
C
Raw Normal View History

Add source files
2020-09-26 03:20:57 -05:00
/*++
Copyright (c) 1989, 1990, 1991 Microsoft Corporation
Module Name:
info.c
Abstract:
This module contains code which performs the following TDI services:
o TdiQueryInformation
o TdiSetInformation
Author:
David Beaver (dbeaver) 1-July-1991
Environment:
Kernel mode
Revision History:
--*/
#include "precomp.h"
#pragma hdrstop
//
// Only the following routine is active in this module. All is is commented
// out waiting for the definition of Get/Set info in TDI version 2.
//
//
// Useful macro to obtain the total length of an MDL chain.
//
#define NbfGetMdlChainLength(Mdl, Length) { \
PMDL _Mdl = (Mdl); \
*(Length) = 0; \
while (_Mdl) { \
*(Length) += MmGetMdlByteCount(_Mdl); \
_Mdl = _Mdl->Next; \
} \
}
//
// Local functions used to satisfy various requests.
//
VOID
NbfStoreProviderStatistics(
IN PDEVICE_CONTEXT DeviceContext,
IN PTDI_PROVIDER_STATISTICS ProviderStatistics
);
VOID
NbfStoreAdapterStatus(
IN PDEVICE_CONTEXT DeviceContext,
IN PUCHAR SourceRouting,
IN UINT SourceRoutingLength,
IN PVOID StatusBuffer
);
VOID
NbfStoreNameBuffers(
IN PDEVICE_CONTEXT DeviceContext,
IN PVOID Buffer,
IN ULONG BufferLength,
IN ULONG NamesToSkip,
OUT PULONG NamesWritten,
OUT PULONG TotalNameCount OPTIONAL,
OUT PBOOLEAN Truncated
);
NTSTATUS
NbfTdiQueryInformation(
IN PDEVICE_CONTEXT DeviceContext,
IN PIRP Irp
)
/*++
Routine Description:
This routine performs the TdiQueryInformation request for the transport
provider.
Arguments:
Irp - the Irp for the requested operation.
Return Value:
NTSTATUS - status of operation.
--*/
{
NTSTATUS status;
PIO_STACK_LOCATION irpSp;
PVOID adapterStatus;
PTDI_REQUEST_KERNEL_QUERY_INFORMATION query;
PTA_NETBIOS_ADDRESS broadcastAddress;
PTDI_PROVIDER_STATISTICS ProviderStatistics;
PTDI_CONNECTION_INFO ConnectionInfo;
ULONG TargetBufferLength;
PFIND_NAME_HEADER FindNameHeader;
LARGE_INTEGER timeout = {0,0};
PTP_REQUEST tpRequest;
PTP_CONNECTION Connection;
PTP_ADDRESS_FILE AddressFile;
PTP_ADDRESS Address;
ULONG NamesWritten, TotalNameCount, BytesWritten;
BOOLEAN Truncated;
BOOLEAN RemoteAdapterStatus;
TDI_ADDRESS_NETBIOS * RemoteAddress;
struct {
ULONG ActivityCount;
TA_NETBIOS_ADDRESS TaAddressBuffer;
} AddressInfo;
PTRANSPORT_ADDRESS TaAddress;
TDI_DATAGRAM_INFO DatagramInfo;
BOOLEAN UsedConnection;
PLIST_ENTRY p;
KIRQL oldirql;
ULONG BytesCopied;
//
// what type of status do we want?
//
irpSp = IoGetCurrentIrpStackLocation (Irp);
query = (PTDI_REQUEST_KERNEL_QUERY_INFORMATION)&irpSp->Parameters;
switch (query->QueryType) {
#if 0
case 0x12345678:
{
typedef struct _NBF_CONNECTION_STATUS {
UCHAR LocalName[16];
UCHAR RemoteName[16];
BOOLEAN SendActive;
BOOLEAN ReceiveQueued;
BOOLEAN ReceiveActive;
BOOLEAN ReceiveWakeUp;
ULONG Flags;
ULONG Flags2;
} NBF_CONNECTION_STATUS, *PNBF_CONNECTION_STATUS;
PNBF_CONNECTION_STATUS CurStatus;
ULONG TotalStatus;
ULONG AllowedStatus;
PLIST_ENRY q;
CurStatus = MmGetSystemAddressForMdl (Irp->MdlAddress);
TotalStatus = 0;
AllowedStatus = MmGetMdlByteCount (Irp->MdlAddress) / sizeof(NBF_CONNECTION_STATUS);
for (p = DeviceContext->AddressDatabase.Flink;
p != &DeviceContext->AddressDatabase;
p = p->Flink) {
Address = CONTAINING_RECORD (p, TP_ADDRESS, Linkage);
if ((Address->Flags & ADDRESS_FLAGS_STOPPING) != 0) {
continue;
}
for (q = Address->ConnectionDatabase.Flink;
q != &Address->ConnectionDatabase;
q = q->Flink) {
Connection = CONTAINING_RECORD (q, TP_CONNECTION, AddressList);
if ((Connection->Flags & CONNECTION_FLAGS_READY) == 0) {
continue;
}
if (TotalStatus >= AllowedStatus) {
continue;
}
RtlMoveMemory (CurStatus->LocalName, Address->NetworkName->NetbiosName, 16);
RtlMoveMemory (CurStatus->RemoteName, Connection->RemoteName, 16);
CurStatus->Flags = Connection->Flags;
CurStatus->Flags2 = Connection->Flags2;
CurStatus->SendActive = (BOOLEAN)(!IsListEmpty(&Connection->SendQueue));
CurStatus->ReceiveQueued = (BOOLEAN)(!IsListEmpty(&Connection->ReceiveQueue));
CurStatus->ReceiveActive = (BOOLEAN)((Connection->Flags & CONNECTION_FLAGS_ACTIVE_RECEIVE) != 0);
CurStatus->ReceiveWakeUp = (BOOLEAN)((Connection->Flags & CONNECTION_FLAGS_RECEIVE_WAKEUP) != 0);
++CurStatus;
++TotalStatus;
}
}
Irp->IoStatus.Information = TotalStatus * sizeof(NBF_CONNECTION_STATUS);
status = STATUS_SUCCESS;
}
break;
#endif
case TDI_QUERY_CONNECTION_INFO:
//
// Connection info is queried on a connection,
// verify this.
//
if (irpSp->FileObject->FsContext2 != (PVOID) TDI_CONNECTION_FILE) {
return STATUS_INVALID_CONNECTION;
}
Connection = irpSp->FileObject->FsContext;
status = NbfVerifyConnectionObject (Connection);
if (!NT_SUCCESS (status)) {
#if DBG
NbfPrint2 ("TdiQueryInfo: Invalid Connection %lx Irp %lx\n", Connection, Irp);
#endif
return status;
}
ConnectionInfo = ExAllocatePoolWithTag (
NonPagedPool,
sizeof (TDI_CONNECTION_INFO),
NBF_MEM_TAG_TDI_CONNECTION_INFO);
if (ConnectionInfo == NULL) {
PANIC ("NbfQueryInfo: Cannot allocate connection info!\n");
NbfWriteResourceErrorLog(
DeviceContext,
EVENT_TRANSPORT_RESOURCE_POOL,
6,
sizeof(TDI_CONNECTION_INFO),
0);
status = STATUS_INSUFFICIENT_RESOURCES;
} else if ((Connection->Flags & CONNECTION_FLAGS_READY) == 0) {
status = STATUS_INVALID_CONNECTION;
ExFreePool (ConnectionInfo);
} else {
PTP_LINK Link = Connection->Link;
RtlZeroMemory ((PVOID)ConnectionInfo, sizeof(TDI_CONNECTION_INFO));
//
// Get link delay and throughput.
//
if (Link->Delay == 0xffffffff) {
//
// If delay is not known, assume 0.
//
ConnectionInfo->Delay.HighPart = 0;
ConnectionInfo->Delay.LowPart = 0;
} else {
//
// Copy the delay as an NT relative time.
//
ConnectionInfo->Delay.HighPart = -1L;
ConnectionInfo->Delay.LowPart = (ULONG)-((LONG)(Link->Delay));
}
if (DeviceContext->MacInfo.MediumAsync) {
ULONG PacketsSent;
ULONG PacketsResent;
ULONG MultiplyFactor;
//
// Calculate the packets sent and resent since the
// last time the throughput was queried.
//
PacketsSent = Link->PacketsSent - Connection->LastPacketsSent;
PacketsResent = Link->PacketsResent - Connection->LastPacketsResent;
//
// Save these for next time.
//
Connection->LastPacketsSent = Link->PacketsSent;
Connection->LastPacketsResent = Link->PacketsResent;
//
// To convert exactly from 100 bits-per-second to
// bytes-per-second, we need to multiply by 12.5.
// Using lower numbers will give worse throughput.
// If there have been no errors we use 12, if there
// have been 20% or more errors we use 1, and in
// between we subtract 11 * (error%/20%) from 12
// and use that.
//
if (PacketsResent == 0 || PacketsSent <= 10) {
MultiplyFactor = 12;
} else if ((PacketsSent / PacketsResent) <= 5) {
MultiplyFactor = 1;
} else {
//
// error%/20% is error%/(1/5), which is 5*error%,
// which is 5 * (resent/send).
//
ASSERT (((11 * 5 * PacketsResent) / PacketsSent) <= 11);
MultiplyFactor = 12 - ((11 * 5 * PacketsResent) / PacketsSent);
}
ConnectionInfo->Throughput.QuadPart =
UInt32x32To64(DeviceContext->MediumSpeed, MultiplyFactor);
} else if (!Link->ThroughputAccurate) {
//
// If throughput is not known, then guess. We
// have MediumSpeed in units of 100 bps; we
// return four times that number as the throughput,
// which corresponds to about 1/3 of the
// maximum bandwidth expressed in bytes/sec.
//
ConnectionInfo->Throughput.QuadPart =
UInt32x32To64(DeviceContext->MediumSpeed, 4);
} else {
//
// Throughput is accurate, return it.
//
ConnectionInfo->Throughput = Link->Throughput;
}
//
// Calculate reliability using the sent/resent ratio,
// if there has been enough activity to make it
// worthwhile. >10% resent is unreliable.
//
if ((Link->PacketsResent > 0) &&
(Link->PacketsSent > 20)) {
ConnectionInfo->Unreliable =
((Link->PacketsSent / Link->PacketsResent) < 10);
} else {
ConnectionInfo->Unreliable = FALSE;
}
ConnectionInfo->TransmittedTsdus = Connection->TransmittedTsdus;
ConnectionInfo->ReceivedTsdus = Connection->ReceivedTsdus;
ConnectionInfo->TransmissionErrors = Connection->TransmissionErrors;
ConnectionInfo->ReceiveErrors = Connection->ReceiveErrors;
status = TdiCopyBufferToMdl (
(PVOID)ConnectionInfo,
0L,
sizeof(TDI_CONNECTION_INFO),
Irp->MdlAddress,
0,
&BytesCopied);
Irp->IoStatus.Information = BytesCopied;
ExFreePool (ConnectionInfo);
}
NbfDereferenceConnection ("query connection info", Connection, CREF_BY_ID);
break;
case TDI_QUERY_ADDRESS_INFO:
if (irpSp->FileObject->FsContext2 == (PVOID)TDI_TRANSPORT_ADDRESS_FILE) {
AddressFile = irpSp->FileObject->FsContext;
status = NbfVerifyAddressObject(AddressFile);
if (!NT_SUCCESS (status)) {
#if DBG
NbfPrint2 ("TdiQueryInfo: Invalid AddressFile %lx Irp %lx\n", AddressFile, Irp);
#endif
return status;
}
UsedConnection = FALSE;
} else if (irpSp->FileObject->FsContext2 == (PVOID)TDI_CONNECTION_FILE) {
Connection = irpSp->FileObject->FsContext;
status = NbfVerifyConnectionObject (Connection);
if (!NT_SUCCESS (status)) {
#if DBG
NbfPrint2 ("TdiQueryInfo: Invalid Connection %lx Irp %lx\n", Connection, Irp);
#endif
return status;
}
AddressFile = Connection->AddressFile;
UsedConnection = TRUE;
} else {
return STATUS_INVALID_ADDRESS;
}
Address = AddressFile->Address;
TdiBuildNetbiosAddress(
Address->NetworkName->NetbiosName,
(BOOLEAN)(Address->Flags & ADDRESS_FLAGS_GROUP ? TRUE : FALSE),
&AddressInfo.TaAddressBuffer);
//
// Count the active addresses.
//
AddressInfo.ActivityCount = 0;
ACQUIRE_SPIN_LOCK (&Address->SpinLock, &oldirql);
for (p = Address->AddressFileDatabase.Flink;
p != &Address->AddressFileDatabase;
p = p->Flink) {
++AddressInfo.ActivityCount;
}
RELEASE_SPIN_LOCK (&Address->SpinLock, oldirql);
status = TdiCopyBufferToMdl (
&AddressInfo,
0,
sizeof(ULONG) + sizeof(TA_NETBIOS_ADDRESS),
Irp->MdlAddress,
0,
&BytesCopied);
Irp->IoStatus.Information = BytesCopied;
if (UsedConnection) {
NbfDereferenceConnection ("query address info", Connection, CREF_BY_ID);
} else {
NbfDereferenceAddress ("query address info", Address, AREF_VERIFY);
}
break;
case TDI_QUERY_BROADCAST_ADDRESS:
//
// for this provider, the broadcast address is a zero byte name,
// contained in a Transport address structure.
//
broadcastAddress = ExAllocatePoolWithTag (
NonPagedPool,
sizeof (TA_NETBIOS_ADDRESS),
NBF_MEM_TAG_TDI_QUERY_BUFFER);
if (broadcastAddress == NULL) {
PANIC ("NbfQueryInfo: Cannot allocate broadcast address!\n");
NbfWriteResourceErrorLog(
DeviceContext,
EVENT_TRANSPORT_RESOURCE_POOL,
2,
sizeof(TA_NETBIOS_ADDRESS),
0);
status = STATUS_INSUFFICIENT_RESOURCES;
} else {
broadcastAddress->TAAddressCount = 1;
broadcastAddress->Address[0].AddressType = TDI_ADDRESS_TYPE_NETBIOS;
broadcastAddress->Address[0].AddressLength = 0;
Irp->IoStatus.Information =
sizeof (broadcastAddress->TAAddressCount) +
sizeof (broadcastAddress->Address[0].AddressType) +
sizeof (broadcastAddress->Address[0].AddressLength);
BytesCopied = (ULONG)Irp->IoStatus.Information;
status = TdiCopyBufferToMdl (
(PVOID)broadcastAddress,
0L,
BytesCopied,
Irp->MdlAddress,
0,
&BytesCopied);
Irp->IoStatus.Information = BytesCopied;
ExFreePool (broadcastAddress);
}
break;
case TDI_QUERY_PROVIDER_INFO:
status = TdiCopyBufferToMdl (
&(DeviceContext->Information),
0,
sizeof (TDI_PROVIDER_INFO),
Irp->MdlAddress,
0,
&BytesCopied);
Irp->IoStatus.Information = BytesCopied;
break;
case TDI_QUERY_PROVIDER_STATISTICS:
//
// This information is probablt available somewhere else.
//
NbfGetMdlChainLength (Irp->MdlAddress, &TargetBufferLength);
if (TargetBufferLength < sizeof(TDI_PROVIDER_STATISTICS) + ((NBF_TDI_RESOURCES-1) * sizeof(TDI_PROVIDER_RESOURCE_STATS))) {
Irp->IoStatus.Information = 0;
status = STATUS_BUFFER_OVERFLOW;
} else {
ProviderStatistics = ExAllocatePoolWithTag(
NonPagedPool,
sizeof(TDI_PROVIDER_STATISTICS) +
((NBF_TDI_RESOURCES-1) * sizeof(TDI_PROVIDER_RESOURCE_STATS)),
NBF_MEM_TAG_TDI_PROVIDER_STATS);
if (ProviderStatistics == NULL) {
PANIC ("NbfQueryInfo: Cannot allocate provider statistics!\n");
NbfWriteResourceErrorLog(
DeviceContext,
EVENT_TRANSPORT_RESOURCE_POOL,
7,
sizeof(TDI_PROVIDER_STATISTICS),
0);
status = STATUS_INSUFFICIENT_RESOURCES;
} else {
NbfStoreProviderStatistics (DeviceContext, ProviderStatistics);
status = TdiCopyBufferToMdl (
(PVOID)ProviderStatistics,
0L,
sizeof(TDI_PROVIDER_STATISTICS) +
((NBF_TDI_RESOURCES-1) * sizeof(TDI_PROVIDER_RESOURCE_STATS)),
Irp->MdlAddress,
0,
&BytesCopied);
Irp->IoStatus.Information = BytesCopied;
ExFreePool (ProviderStatistics);
}
}
break;
case TDI_QUERY_SESSION_STATUS:
status = STATUS_NOT_IMPLEMENTED;
break;
case TDI_QUERY_ADAPTER_STATUS:
NbfGetMdlChainLength (Irp->MdlAddress, &TargetBufferLength);
//
// Determine if this is a local or remote query. It is
// local if there is no remote address specific at all,
// or if it is equal to our reserved address.
//
RemoteAdapterStatus = FALSE;
if (query->RequestConnectionInformation != NULL) {
if (!NbfValidateTdiAddress(
query->RequestConnectionInformation->RemoteAddress,
query->RequestConnectionInformation->RemoteAddressLength)) {
return STATUS_BAD_NETWORK_PATH;
}
RemoteAddress = NbfParseTdiAddress(query->RequestConnectionInformation->RemoteAddress, FALSE);
if (!RemoteAddress) {
return STATUS_BAD_NETWORK_PATH;
}
if (!RtlEqualMemory(
RemoteAddress->NetbiosName,
DeviceContext->ReservedNetBIOSAddress,
NETBIOS_NAME_LENGTH)) {
RemoteAdapterStatus = TRUE;
}
}
if (RemoteAdapterStatus) {
//
// We need a request object to keep track of this TDI request.
// Attach this request to the device context.
//
status = NbfCreateRequest (
Irp, // IRP for this request.
DeviceContext, // context.
REQUEST_FLAGS_DC, // partial flags.
Irp->MdlAddress, // the data to be received.
TargetBufferLength, // length of the data.
timeout, // do this ourselves here.
&tpRequest);
if (NT_SUCCESS (status)) {
NbfReferenceDeviceContext ("Remote status", DeviceContext, DCREF_REQUEST);
tpRequest->Owner = DeviceContextType;
//
// Allocate a temp buffer to hold our results.
//
tpRequest->ResponseBuffer = ExAllocatePoolWithTag(
NonPagedPool,
TargetBufferLength,
NBF_MEM_TAG_TDI_QUERY_BUFFER);
if (tpRequest->ResponseBuffer == NULL) {
NbfWriteResourceErrorLog(
DeviceContext,
EVENT_TRANSPORT_RESOURCE_POOL,
12,
TargetBufferLength,
0);
NbfCompleteRequest (tpRequest, STATUS_INSUFFICIENT_RESOURCES, 0);
} else {
ACQUIRE_SPIN_LOCK (&DeviceContext->SpinLock,&oldirql);
if (DeviceContext->State != DEVICECONTEXT_STATE_OPEN) {
RELEASE_SPIN_LOCK (&DeviceContext->SpinLock,oldirql);
NbfCompleteRequest (tpRequest, STATUS_DEVICE_NOT_READY, 0);
} else {
PUCHAR SingleSR;
UINT SingleSRLength;
InsertTailList (
&DeviceContext->StatusQueryQueue,
&tpRequest->Linkage);
tpRequest->FrameContext = DeviceContext->UniqueIdentifier | 0x8000;
++DeviceContext->UniqueIdentifier;
if (DeviceContext->UniqueIdentifier == 0x8000) {
DeviceContext->UniqueIdentifier = 1;
}
RELEASE_SPIN_LOCK (&DeviceContext->SpinLock, oldirql);
//
// The request is queued. Now send out the first packet and
// start the timer.
//
tpRequest->Retries = DeviceContext->GeneralRetries;
tpRequest->BytesWritten = 0;
//
// STATUS_QUERY frames go out as
// single-route source routing.
//
MacReturnSingleRouteSR(
&DeviceContext->MacInfo,
&SingleSR,
&SingleSRLength);
NbfSendStatusQuery(
DeviceContext,
tpRequest,
&DeviceContext->NetBIOSAddress,
SingleSR,
SingleSRLength);
}
}
//
// As long as the request is created, pend here.
// The IRP will complete when the request completes.
//
status = STATUS_PENDING;
}
} else {
//
// Local.
//
adapterStatus = ExAllocatePoolWithTag (
NonPagedPool,
TargetBufferLength,
NBF_MEM_TAG_TDI_QUERY_BUFFER);
if (adapterStatus == NULL) {
PANIC("NbfQueryInfo: PANIC! Could not allocate adapter status buffer\n");
NbfWriteResourceErrorLog(
DeviceContext,
EVENT_TRANSPORT_RESOURCE_POOL,
3,
TargetBufferLength,
0);
return STATUS_INSUFFICIENT_RESOURCES;
}
NbfStoreAdapterStatus (
DeviceContext,
NULL,
0,
adapterStatus);
NbfStoreNameBuffers (
DeviceContext,
(PUCHAR)adapterStatus + sizeof(ADAPTER_STATUS),
TargetBufferLength - sizeof(ADAPTER_STATUS),
0,
&NamesWritten,
&TotalNameCount,
&Truncated);
((PADAPTER_STATUS)adapterStatus)->name_count = (WORD)TotalNameCount;
BytesWritten = sizeof(ADAPTER_STATUS) + (NamesWritten * sizeof(NAME_BUFFER));
status = TdiCopyBufferToMdl (
adapterStatus,
0,
BytesWritten,
Irp->MdlAddress,
0,
&BytesCopied);
Irp->IoStatus.Information = BytesCopied;
if (Truncated) {
status = STATUS_BUFFER_OVERFLOW;
}
ExFreePool (adapterStatus);
}
break;
case TDI_QUERY_FIND_NAME:
NbfGetMdlChainLength (Irp->MdlAddress, &TargetBufferLength);
//
// Check that there is a valid Netbios remote address.
//
if (!NbfValidateTdiAddress(
query->RequestConnectionInformation->RemoteAddress,
query->RequestConnectionInformation->RemoteAddressLength)) {
return STATUS_BAD_NETWORK_PATH;
}
RemoteAddress = NbfParseTdiAddress(query->RequestConnectionInformation->RemoteAddress, FALSE);
if (!RemoteAddress) {
return STATUS_BAD_NETWORK_PATH;
}
//
// We need a request object to keep track of this TDI request.
// Attach this request to the device context.
//
status = NbfCreateRequest (
Irp, // IRP for this request.
DeviceContext, // context.
REQUEST_FLAGS_DC, // partial flags.
Irp->MdlAddress, // the data to be received.
TargetBufferLength, // length of the data.
timeout, // do this ourselves here.
&tpRequest);
if (NT_SUCCESS (status)) {
NbfReferenceDeviceContext ("Find name", DeviceContext, DCREF_REQUEST);
tpRequest->Owner = DeviceContextType;
//
// Allocate a temp buffer to hold our results.
//
tpRequest->ResponseBuffer = ExAllocatePoolWithTag(
NonPagedPool,
TargetBufferLength,
NBF_MEM_TAG_TDI_QUERY_BUFFER);
if (tpRequest->ResponseBuffer == NULL) {
NbfWriteResourceErrorLog(
DeviceContext,
EVENT_TRANSPORT_RESOURCE_POOL,
4,
TargetBufferLength,
0);
NbfCompleteRequest (tpRequest, STATUS_INSUFFICIENT_RESOURCES, 0);
} else {
ACQUIRE_SPIN_LOCK (&DeviceContext->SpinLock,&oldirql);
if (DeviceContext->State != DEVICECONTEXT_STATE_OPEN) {
RELEASE_SPIN_LOCK (&DeviceContext->SpinLock,oldirql);
NbfCompleteRequest (tpRequest, STATUS_DEVICE_NOT_READY, 0);
} else {
InsertTailList (
&DeviceContext->FindNameQueue,
&tpRequest->Linkage);
tpRequest->FrameContext = DeviceContext->UniqueIdentifier | 0x8000;
++DeviceContext->UniqueIdentifier;
if (DeviceContext->UniqueIdentifier == 0x8000) {
DeviceContext->UniqueIdentifier = 1;
}
RELEASE_SPIN_LOCK (&DeviceContext->SpinLock, oldirql);
//
// The request is queued. Now send out the first packet and
// start the timer.
//
// We fill in the FIND_NAME_HEADER in the buffer, but
// set BytesWritten to 0; we don't include the header
// in BytesWritten until we get a response, so that
// a BytesWritten of 0 means "no response".
//
tpRequest->Retries = DeviceContext->GeneralRetries;
tpRequest->BytesWritten = 0;
FindNameHeader = (PFIND_NAME_HEADER)tpRequest->ResponseBuffer;
FindNameHeader->node_count = 0;
FindNameHeader->unique_group = NETBIOS_NAME_TYPE_UNIQUE;
NbfSendQueryFindName (DeviceContext, tpRequest);
}
}
//
// As long as the request is created, pend here.
// The IRP will complete when the request completes.
//
status = STATUS_PENDING;
}
break;
case TDI_QUERY_DATA_LINK_ADDRESS:
case TDI_QUERY_NETWORK_ADDRESS:
TaAddress = (PTRANSPORT_ADDRESS)&AddressInfo.TaAddressBuffer;
TaAddress->TAAddressCount = 1;
TaAddress->Address[0].AddressLength = 6;
if (query->QueryType == TDI_QUERY_DATA_LINK_ADDRESS) {
TaAddress->Address[0].AddressType =
DeviceContext->MacInfo.MediumAsync ?
NdisMediumWan : DeviceContext->MacInfo.MediumType;
} else {
TaAddress->Address[0].AddressType = TDI_ADDRESS_TYPE_UNSPEC;
}
RtlCopyMemory (TaAddress->Address[0].Address, DeviceContext->LocalAddress.Address, 6);
status = TdiCopyBufferToMdl (
&AddressInfo.TaAddressBuffer,
0,
sizeof(TRANSPORT_ADDRESS)+5,
Irp->MdlAddress,
0,
&BytesCopied);
Irp->IoStatus.Information = BytesCopied;
break;
case TDI_QUERY_DATAGRAM_INFO:
DatagramInfo.MaximumDatagramBytes = 0;
DatagramInfo.MaximumDatagramCount = 0;
status = TdiCopyBufferToMdl (
&DatagramInfo,
0,
sizeof(DatagramInfo),
Irp->MdlAddress,
0,
&BytesCopied);
Irp->IoStatus.Information = BytesCopied;
break;
default:
status = STATUS_INVALID_DEVICE_REQUEST;
break;
}
return status;
} /* NbfTdiQueryInformation */
//
// Quick macros, assumes DeviceContext and ProviderStatistics exist.
//
#define STORE_RESOURCE_STATS_1(_ResourceNum,_ResourceId,_ResourceName) \
{ \
PTDI_PROVIDER_RESOURCE_STATS RStats = &ProviderStatistics->ResourceStats[_ResourceNum]; \
RStats->ResourceId = (_ResourceId); \
RStats->MaximumResourceUsed = DeviceContext->_ResourceName ## MaxInUse; \
if (DeviceContext->_ResourceName ## Samples > 0) { \
RStats->AverageResourceUsed = DeviceContext->_ResourceName ## Total / DeviceContext->_ResourceName ## Samples; \
} else { \
RStats->AverageResourceUsed = 0; \
} \
RStats->ResourceExhausted = DeviceContext->_ResourceName ## Exhausted; \
}
#define STORE_RESOURCE_STATS_2(_ResourceNum,_ResourceId,_ResourceName) \
{ \
PTDI_PROVIDER_RESOURCE_STATS RStats = &ProviderStatistics->ResourceStats[_ResourceNum]; \
RStats->ResourceId = (_ResourceId); \
RStats->MaximumResourceUsed = DeviceContext->_ResourceName ## Allocated; \
RStats->AverageResourceUsed = DeviceContext->_ResourceName ## Allocated; \
RStats->ResourceExhausted = DeviceContext->_ResourceName ## Exhausted; \
}
VOID
NbfStoreProviderStatistics(
IN PDEVICE_CONTEXT DeviceContext,
IN PTDI_PROVIDER_STATISTICS ProviderStatistics
)
/*++
Routine Description:
This routine writes the TDI_PROVIDER_STATISTICS structure
from the device context into ProviderStatistics.
Arguments:
DeviceContext - a pointer to the device context.
ProviderStatistics - The buffer that holds the result. It is assumed
that it is long enough.
Return Value:
None.
--*/
{
//
// Copy all the statistics up to NumberOfResources
// in one move.
//
RtlCopyMemory(
ProviderStatistics,
&DeviceContext->Statistics,
FIELD_OFFSET (TDI_PROVIDER_STATISTICS, NumberOfResources));
//
// Calculate AverageSendWindow.
//
if (DeviceContext->SendWindowSamples > 0) {
ProviderStatistics->AverageSendWindow =
DeviceContext->SendWindowTotal / DeviceContext->SendWindowSamples;
} else {
ProviderStatistics->AverageSendWindow = 1;
}
//
// Copy the resource statistics.
//
ProviderStatistics->NumberOfResources = NBF_TDI_RESOURCES;
STORE_RESOURCE_STATS_1 (0, LINK_RESOURCE_ID, Link);
STORE_RESOURCE_STATS_1 (1, ADDRESS_RESOURCE_ID, Address);
STORE_RESOURCE_STATS_1 (2, ADDRESS_FILE_RESOURCE_ID, AddressFile);
STORE_RESOURCE_STATS_1 (3, CONNECTION_RESOURCE_ID, Connection);
STORE_RESOURCE_STATS_1 (4, REQUEST_RESOURCE_ID, Request);
STORE_RESOURCE_STATS_2 (5, UI_FRAME_RESOURCE_ID, UIFrame);
STORE_RESOURCE_STATS_2 (6, PACKET_RESOURCE_ID, Packet);
STORE_RESOURCE_STATS_2 (7, RECEIVE_PACKET_RESOURCE_ID, ReceivePacket);
STORE_RESOURCE_STATS_2 (8, RECEIVE_BUFFER_RESOURCE_ID, ReceiveBuffer);
} /* NbfStoreProviderStatistics */
VOID
NbfStoreAdapterStatus(
IN PDEVICE_CONTEXT DeviceContext,
IN PUCHAR SourceRouting,
IN UINT SourceRoutingLength,
IN PVOID StatusBuffer
)
/*++
Routine Description:
This routine writes the ADAPTER_STATUS structure for the
device context into StatusBuffer. The name_count field is
initialized to zero; NbfStoreNameBuffers is used to write
name buffers.
Arguments:
DeviceContext - a pointer to the device context.
SourceRouting - If this is a remote request, the source
routing information from the frame.
SourceRoutingLength - The length of SourceRouting.
StatusBuffer - The buffer that holds the result. It is assumed
that it is at least sizeof(ADAPTER_STATUS) bytes long.
Return Value:
None.
--*/
{
PADAPTER_STATUS AdapterStatus = (PADAPTER_STATUS)StatusBuffer;
UINT MaxUserData;
RtlZeroMemory ((PVOID)AdapterStatus, sizeof(ADAPTER_STATUS));
RtlCopyMemory (AdapterStatus->adapter_address, DeviceContext->LocalAddress.Address, 6);
AdapterStatus->rev_major = 0x03;
switch (DeviceContext->MacInfo.MediumType) {
case NdisMedium802_5: AdapterStatus->adapter_type = 0xff; break;
default: AdapterStatus->adapter_type = 0xfe; break;
}
AdapterStatus->frmr_recv = (WORD)DeviceContext->FrmrReceived;
AdapterStatus->frmr_xmit = (WORD)DeviceContext->FrmrTransmitted;
AdapterStatus->recv_buff_unavail = (WORD)(DeviceContext->ReceivePacketExhausted + DeviceContext->ReceiveBufferExhausted);
AdapterStatus->xmit_buf_unavail = (WORD)DeviceContext->PacketExhausted;
AdapterStatus->xmit_success = (WORD)(DeviceContext->Statistics.DataFramesSent - DeviceContext->Statistics.DataFramesResent);
AdapterStatus->recv_success = (WORD)DeviceContext->Statistics.DataFramesReceived;
AdapterStatus->iframe_recv_err = (WORD)DeviceContext->Statistics.DataFramesRejected;
AdapterStatus->iframe_xmit_err = (WORD)DeviceContext->Statistics.DataFramesResent;
AdapterStatus->t1_timeouts = (WORD)DeviceContext->Statistics.ResponseTimerExpirations;
AdapterStatus->ti_timeouts = (WORD)DeviceContext->TiExpirations;
AdapterStatus->xmit_aborts = (WORD)0;
AdapterStatus->free_ncbs = (WORD)0xffff;
AdapterStatus->max_cfg_ncbs = (WORD)0xffff;
AdapterStatus->max_ncbs = (WORD)0xffff;
AdapterStatus->pending_sess = (WORD)DeviceContext->Statistics.OpenConnections;
AdapterStatus->max_cfg_sess = (WORD)0xffff;
AdapterStatus->max_sess = (WORD)0xffff;
MacReturnMaxDataSize(
&DeviceContext->MacInfo,
NULL,
0,
DeviceContext->MaxSendPacketSize,
TRUE,
&MaxUserData);
AdapterStatus->max_dgram_size = (WORD)(MaxUserData - (sizeof(DLC_FRAME) + sizeof(NBF_HDR_CONNECTIONLESS)));
MacReturnMaxDataSize(
&DeviceContext->MacInfo,
SourceRouting,
SourceRoutingLength,
DeviceContext->MaxSendPacketSize,
FALSE,
&MaxUserData);
AdapterStatus->max_sess_pkt_size = (WORD)(MaxUserData - (sizeof(DLC_I_FRAME) + sizeof(NBF_HDR_CONNECTION)));
return;
} /* NbfStoreAdapterStatus */
VOID
NbfStoreNameBuffers(
IN PDEVICE_CONTEXT DeviceContext,
IN PVOID Buffer,
IN ULONG BufferLength,
IN ULONG NamesToSkip,
OUT PULONG NamesWritten,
OUT PULONG TotalNameCount OPTIONAL,
OUT PBOOLEAN Truncated
)
/*++
Routine Description:
This routine writes NAME_BUFFER structures for the
device context into NameBuffer. It can skip a specified
number of names at the beginning, and returns the number
of names written into NameBuffer. If a name will only
partially fit, it is not written.
Arguments:
DeviceContext - a pointer to the device context.
NameBuffer - The buffer to write the names into.
NameBufferLength - The length of NameBuffer.
NamesToSkip - The number of names to skip.
NamesWritten - Returns the number of names written.
TotalNameCount - Returns the total number of names available,
if specified.
Truncated - More names are available than were written.
Return Value:
None.
--*/
{
ULONG NameCount = 0;
ULONG BytesWritten = 0;
KIRQL oldirql;
PLIST_ENTRY p;
PNAME_BUFFER NameBuffer = (PNAME_BUFFER)Buffer;
PTP_ADDRESS address;
//
// Spin through the address list for this device context.
//
ACQUIRE_SPIN_LOCK (&DeviceContext->SpinLock, &oldirql);
p = DeviceContext->AddressDatabase.Flink;
for (p = DeviceContext->AddressDatabase.Flink;
p != &DeviceContext->AddressDatabase;
p = p->Flink) {
address = CONTAINING_RECORD (p, TP_ADDRESS, Linkage);
//
// Ignore addresses that are shutting down.
//
if ((address->Flags & ADDRESS_FLAGS_STOPPING) != 0) {
continue;
}
//
// Ignore the broadcast address.
//
if (address->NetworkName == NULL) {
continue;
}
//
// Ignore the reserved address.
//
if ((address->NetworkName->NetbiosName[0] == 0) &&
(RtlEqualMemory(
address->NetworkName->NetbiosName,
DeviceContext->ReservedNetBIOSAddress,
NETBIOS_NAME_LENGTH))) {
continue;
}
//
// Check if we are still skipping.
//
if (NameCount < NamesToSkip) {
++NameCount;
continue;
}
//
// Make sure we still have room.
//
if (BytesWritten + sizeof(NAME_BUFFER) > BufferLength) {
break;
}
RtlCopyMemory(
NameBuffer->name,
address->NetworkName->NetbiosName,
NETBIOS_NAME_LENGTH);
++NameCount;
NameBuffer->name_num = (UCHAR)NameCount;
NameBuffer->name_flags = REGISTERED;
if (address->Flags & ADDRESS_FLAGS_GROUP) {
NameBuffer->name_flags |= GROUP_NAME;
}
// name_flags should be done more accurately.
BytesWritten += sizeof(NAME_BUFFER);
++NameBuffer;
}
*NamesWritten = (ULONG)(NameBuffer - (PNAME_BUFFER)Buffer);
if (p == &DeviceContext->AddressDatabase) {
*Truncated = FALSE;
if (ARGUMENT_PRESENT(TotalNameCount)) {
*TotalNameCount = NameCount;
}
} else {
*Truncated = TRUE;
//
// If requested, continue through the list and count
// all the addresses.
//
if (ARGUMENT_PRESENT(TotalNameCount)) {
for ( ;
p != &DeviceContext->AddressDatabase;
p = p->Flink) {
address = CONTAINING_RECORD (p, TP_ADDRESS, Linkage);
//
// Ignore addresses that are shutting down.
//
if ((address->Flags & ADDRESS_FLAGS_STOPPING) != 0) {
continue;
}
//
// Ignore the broadcast address.
//
if (address->NetworkName == NULL) {
continue;
}
//
// Ignore the reserved address, since we count it no matter what.
//
if ((address->NetworkName->NetbiosName[0] == 0) &&
(RtlEqualMemory(
address->NetworkName->NetbiosName,
DeviceContext->ReservedNetBIOSAddress,
NETBIOS_NAME_LENGTH))) {
continue;
}
++NameCount;
}
*TotalNameCount = NameCount;
}
}
RELEASE_SPIN_LOCK (&DeviceContext->SpinLock, oldirql);
return;
} /* NbfStoreNameBuffers */
NTSTATUS
NbfProcessStatusQuery(
IN PDEVICE_CONTEXT DeviceContext,
IN PTP_ADDRESS Address OPTIONAL,
IN PNBF_HDR_CONNECTIONLESS UiFrame,
IN PHARDWARE_ADDRESS SourceAddress,
IN PUCHAR SourceRouting,
IN UINT SourceRoutingLength
)
/*++
Routine Description:
This routine processes a STATUS.QUERY packet.
Arguments:
DeviceContext - a pointer to the device context the frame was received on.
Address - The address we are responding from, or NULL if the STATUS.QUERY
was sent to the reserved address.
UiFrame - The packet in question, starting at the Netbios header.
SourceAddress - The source hardware address of the packet.
SourceRouting - Source routing data in the query.
SourceRoutingLength - The length of SourceRouting.
Return Value:
NTSTATUS - status of operation.
--*/
{
NTSTATUS Status;
NDIS_STATUS NdisStatus;
PTP_UI_FRAME RawFrame;
PVOID ResponseBuffer;
UINT ResponseBufferLength;
ULONG NamesWritten, TotalNameCount;
ULONG BytesWritten;
UCHAR RequestType;
BOOLEAN Truncated, UsersBufferTooShort;
USHORT UsersBufferLength;
UINT HeaderLength;
UCHAR TempSR[MAX_SOURCE_ROUTING];
PUCHAR ResponseSR;
PNDIS_BUFFER NdisBuffer;
//
// Allocate a buffer to hold the status.
//
MacReturnMaxDataSize(
&DeviceContext->MacInfo,
SourceRouting,
SourceRoutingLength,
DeviceContext->CurSendPacketSize,
FALSE,
&ResponseBufferLength);
ResponseBufferLength -= (sizeof(DLC_FRAME) + sizeof(NBF_HDR_CONNECTIONLESS));
UsersBufferLength = (UiFrame->Data2High * 256) + UiFrame->Data2Low;
//
// See how big to make our buffer; if the amount remaining in the user's
// buffer is less than our max size, chop it down.
//
if (UiFrame->Data1 <= 1) {
//
// This is the initial request.
//
if (ResponseBufferLength > (UINT)UsersBufferLength) {
ResponseBufferLength = UsersBufferLength;
}
} else {
//
// Subsequent request; compensate for already-sent data.
//
UsersBufferLength -= (sizeof(ADAPTER_STATUS) + (UiFrame->Data1 * sizeof(NAME_BUFFER)));
if (ResponseBufferLength > (UINT)UsersBufferLength) {
ResponseBufferLength = UsersBufferLength;
}
}
//
// If the remote station is asking for no data, ignore this request.
// This prevents us from trying to allocate 0 bytes of pool.
//
if ( (LONG)ResponseBufferLength <= 0 ) {
return STATUS_ABANDONED;
}
ResponseBuffer = ExAllocatePoolWithTag(
NonPagedPool,
ResponseBufferLength,
NBF_MEM_TAG_TDI_QUERY_BUFFER);
if (ResponseBuffer == NULL) {
NbfWriteResourceErrorLog(
DeviceContext,
EVENT_TRANSPORT_RESOURCE_POOL,
5,
ResponseBufferLength,
0);
return STATUS_ABANDONED;
}
//
// Fill in the response buffer.
//
if (UiFrame->Data1 <= 1) {
//
// First request.
//
NbfStoreAdapterStatus (
DeviceContext,
SourceRouting,
SourceRoutingLength,
ResponseBuffer);
NbfStoreNameBuffers (
DeviceContext,
(PUCHAR)ResponseBuffer + sizeof(ADAPTER_STATUS),
ResponseBufferLength - sizeof(ADAPTER_STATUS),
0,
&NamesWritten,
&TotalNameCount,
&Truncated);
BytesWritten = sizeof(ADAPTER_STATUS) + (NamesWritten * sizeof(NAME_BUFFER));
//
// If the data was truncated, but we are returning the maximum
// that the user requested, report that as "user's buffer
// too short" instead of "truncated".
//
if (Truncated && (ResponseBufferLength >= (UINT)UsersBufferLength)) {
Truncated = FALSE;
UsersBufferTooShort = TRUE;
} else {
UsersBufferTooShort = FALSE;
}
((PADAPTER_STATUS)ResponseBuffer)->name_count = (WORD)TotalNameCount;
} else {
NbfStoreNameBuffers (
DeviceContext,
ResponseBuffer,
ResponseBufferLength,
UiFrame->Data1,
&NamesWritten,
NULL,
&Truncated);
BytesWritten = NamesWritten * sizeof(NAME_BUFFER);
if (Truncated && (ResponseBufferLength >= (UINT)UsersBufferLength)) {
Truncated = FALSE;
UsersBufferTooShort = TRUE;
} else {
UsersBufferTooShort = FALSE;
}
}
//
// Allocate a UI frame from the pool.
//
Status = NbfCreateConnectionlessFrame (DeviceContext, &RawFrame);
if (!NT_SUCCESS (Status)) { // couldn't make frame.
ExFreePool (ResponseBuffer);
return STATUS_ABANDONED;
}
IF_NBFDBG (NBF_DEBUG_DEVCTX) {
NbfPrint2 ("NbfProcessStatusQuery: Sending Frame: %lx, NdisPacket: %lx\n",
RawFrame, RawFrame->NdisPacket);
}
//
// Build the MAC header. STATUS_RESPONSE frames go out as
// non-broadcast source routing.
//
if (SourceRouting != NULL) {
RtlCopyMemory(
TempSR,
SourceRouting,
SourceRoutingLength);
MacCreateNonBroadcastReplySR(
&DeviceContext->MacInfo,
TempSR,
SourceRoutingLength,
&ResponseSR);
} else {
ResponseSR = NULL;
}
MacConstructHeader (
&DeviceContext->MacInfo,
RawFrame->Header,
SourceAddress->Address,
DeviceContext->LocalAddress.Address,
sizeof (DLC_FRAME) + sizeof (NBF_HDR_CONNECTIONLESS) + BytesWritten,
ResponseSR,
SourceRoutingLength,
&HeaderLength);
//
// Build the DLC UI frame header.
//
NbfBuildUIFrameHeader(&RawFrame->Header[HeaderLength]);
HeaderLength += sizeof(DLC_FRAME);
//
// Build the Netbios header.
//
switch (UiFrame->Data1) {
case 0: // pre 2.1 request
RequestType = (UCHAR)0;
break;
case 1: // 2.1, first request
RequestType = (UCHAR)NamesWritten;
break;
default: // 2.1, subsequent request
RequestType = (UCHAR)(UiFrame->Data1 + NamesWritten);
break;
}
ConstructStatusResponse (
(PNBF_HDR_CONNECTIONLESS)&(RawFrame->Header[HeaderLength]),
RequestType, // request type.
Truncated, // more data.
UsersBufferTooShort, // user's buffer too small
(USHORT)BytesWritten, // bytes in response
RESPONSE_CORR(UiFrame), // correlator
UiFrame->SourceName, // receiver permanent name
(ARGUMENT_PRESENT(Address)) ?
Address->NetworkName->NetbiosName :
DeviceContext->ReservedNetBIOSAddress); // source name
HeaderLength += sizeof(NBF_HDR_CONNECTIONLESS);
//
// Munge the packet length (now, before we append the second
// buffer).
//
NbfSetNdisPacketLength(RawFrame->NdisPacket, HeaderLength);
//
// Now, if we have any name data, attach our buffer onto the frame.
// Note that it's possible at the end of the user's buffer for us
// to not have room for any names, and thus we'll have no data to
// send.
//
if ( BytesWritten != 0 ) {
RawFrame->DataBuffer = ResponseBuffer;
NdisAllocateBuffer(
&NdisStatus,
&NdisBuffer,
DeviceContext->NdisBufferPool,
ResponseBuffer,
BytesWritten);
if (NdisStatus != NDIS_STATUS_SUCCESS) {
PANIC ("ConstructStatusResponse: NdisAllocateBuffer failed.\n");
NbfDestroyConnectionlessFrame (DeviceContext, RawFrame);
return STATUS_ABANDONED;
}
NdisChainBufferAtBack (RawFrame->NdisPacket, NdisBuffer);
} else {
RawFrame->DataBuffer = NULL;
}
NbfSendUIFrame (
DeviceContext,
RawFrame,
FALSE); // no loopback (MC frame)
return STATUS_ABANDONED;
} /* NbfProcessStatusQuery */
VOID
NbfSendQueryFindName(
IN PDEVICE_CONTEXT DeviceContext,
IN PTP_REQUEST Request
)
/*++
Routine Description:
This routine will send a FIND.NAME packet for the specified
find name request, and start the request timer.
Arguments:
DeviceContext - a pointer to the device context to send the find name on.
Request - The find name request.
Return Value:
None.
--*/
{
TDI_ADDRESS_NETBIOS * remoteAddress;
PIO_STACK_LOCATION irpSp;
NTSTATUS Status;
PTP_UI_FRAME RawFrame;
PUCHAR SingleSR;
UINT SingleSRLength;
UINT HeaderLength;
LARGE_INTEGER Timeout;
irpSp = IoGetCurrentIrpStackLocation (Request->IoRequestPacket);
remoteAddress = NbfParseTdiAddress(
((PTDI_REQUEST_KERNEL_QUERY_INFORMATION)(&irpSp->Parameters))->
RequestConnectionInformation->RemoteAddress, FALSE);
//
// Start the timer for this request.
//
Request->Flags |= REQUEST_FLAGS_TIMER; // there is a timeout on this request.
KeInitializeTimer (&Request->Timer); // set to not-signaled state.
NbfReferenceRequest ("Find Name: timer", Request, RREF_TIMER); // one for the timer
Timeout.LowPart = (ULONG)(-(LONG)DeviceContext->GeneralTimeout);
Timeout.HighPart = -1;
KeSetTimer (&Request->Timer, Timeout, &Request->Dpc);
//
// Allocate a UI frame from the pool.
//
Status = NbfCreateConnectionlessFrame (DeviceContext, &RawFrame);
if (!NT_SUCCESS (Status)) { // couldn't make frame.
return;
}
IF_NBFDBG (NBF_DEBUG_DEVCTX) {
NbfPrint2 ("NbfSendFindNames: Sending Frame: %lx, NdisPacket: %lx\n",
RawFrame, RawFrame->NdisPacket);
}
//
// Build the MAC header. NAME_QUERY frames go out as
// single-route source routing.
//
MacReturnSingleRouteSR(
&DeviceContext->MacInfo,
&SingleSR,
&SingleSRLength);
MacConstructHeader (
&DeviceContext->MacInfo,
RawFrame->Header,
DeviceContext->NetBIOSAddress.Address,
DeviceContext->LocalAddress.Address,
sizeof (DLC_FRAME) + sizeof (NBF_HDR_CONNECTIONLESS),
SingleSR,
SingleSRLength,
&HeaderLength);
//
// Build the DLC UI frame header.
//
NbfBuildUIFrameHeader(&RawFrame->Header[HeaderLength]);
HeaderLength += sizeof(DLC_FRAME);
//
// Build the Netbios header.
//
ConstructNameQuery (
(PNBF_HDR_CONNECTIONLESS)&(RawFrame->Header[HeaderLength]),
NETBIOS_NAME_TYPE_UNIQUE, // call from a unique name.
NAME_QUERY_LSN_FIND_NAME, // LSN
Request->FrameContext, // corr. in 1st NAME_RECOGNIZED.
DeviceContext->ReservedNetBIOSAddress,
(PNAME)remoteAddress->NetbiosName);
HeaderLength += sizeof(NBF_HDR_CONNECTIONLESS);
//
// Munge the packet length.
//
NbfSetNdisPacketLength(RawFrame->NdisPacket, HeaderLength);
NbfSendUIFrame (
DeviceContext,
RawFrame,
FALSE); // no loopback (MC frame)
} /* NbfSendQueryFindName */
NTSTATUS
NbfProcessQueryNameRecognized(
IN PDEVICE_CONTEXT DeviceContext,
IN PUCHAR Packet,
PNBF_HDR_CONNECTIONLESS UiFrame
)
/*++
Routine Description:
This routine processes a NAME.RECOGNIZED request with a
correlator of 0, indicating it was a response to a previous
FIND.NAME packet.
Arguments:
DeviceContext - a pointer to the device context the frame was received on.
Packet - The packet in question, starting at the MAC header.
UiFrame - The packet, starting at the Netbios header.
Return Value:
NTSTATUS - status of operation.
--*/
{
KIRQL oldirql;
PTP_REQUEST Request;
PFIND_NAME_BUFFER FindNameBuffer;
PFIND_NAME_HEADER FindNameHeader;
PUCHAR DestinationAddress;
HARDWARE_ADDRESS SourceAddressBuffer;
PHARDWARE_ADDRESS SourceAddress;
PUCHAR SourceRouting;
UINT SourceRoutingLength;
PUCHAR TargetBuffer;
USHORT FrameContext;
PLIST_ENTRY p;
MacReturnDestinationAddress(
&DeviceContext->MacInfo,
Packet,
&DestinationAddress);
MacReturnSourceAddress(
&DeviceContext->MacInfo,
Packet,
&SourceAddressBuffer,
&SourceAddress,
NULL);
MacReturnSourceRouting(
&DeviceContext->MacInfo,
Packet,
&SourceRouting,
&SourceRoutingLength);
//
// Find the request that this is for, using the frame context.
//
FrameContext = TRANSMIT_CORR(UiFrame);
ACQUIRE_SPIN_LOCK (&DeviceContext->SpinLock, &oldirql);
for (p=DeviceContext->FindNameQueue.Flink;
p != &DeviceContext->FindNameQueue;
p=p->Flink) {
Request = CONTAINING_RECORD (p, TP_REQUEST, Linkage);
if (Request->FrameContext == FrameContext) {
break;
}
}
if (p == &DeviceContext->FindNameQueue) {
RELEASE_SPIN_LOCK (&DeviceContext->SpinLock, oldirql);
return STATUS_SUCCESS;
}
NbfReferenceRequest ("Name Recognized", Request, RREF_FIND_NAME);
RELEASE_SPIN_LOCK (&DeviceContext->SpinLock, oldirql);
//
// Make sure that this physical address has not
// responded yet.
//
ACQUIRE_SPIN_LOCK (&Request->SpinLock, &oldirql);
//
// Make sure this request is not stopping.
//
if ((Request->Flags & REQUEST_FLAGS_STOPPING) != 0) {
RELEASE_SPIN_LOCK (&Request->SpinLock, oldirql);
NbfDereferenceRequest ("Stopping", Request, RREF_STATUS);
return STATUS_SUCCESS;
}
//
// If this is the first response, update BytesWritten to include
// the header that is already written in ResponseBuffer.
//
if (Request->BytesWritten == 0) {
Request->BytesWritten = sizeof(FIND_NAME_HEADER);
}
TargetBuffer = Request->ResponseBuffer;
FindNameBuffer = (PFIND_NAME_BUFFER)(TargetBuffer + sizeof(FIND_NAME_HEADER));
for ( ; FindNameBuffer < (PFIND_NAME_BUFFER)(TargetBuffer + Request->BytesWritten); FindNameBuffer++) {
if (RtlEqualMemory (FindNameBuffer->source_addr, SourceAddress->Address, 6)) {
RELEASE_SPIN_LOCK (&Request->SpinLock, oldirql);
NbfDereferenceRequest ("Duplicate NR", Request, RREF_FIND_NAME);
return STATUS_SUCCESS;
}
}
//
// This is a new address, update if there is room.
//
if ((Request->BytesWritten + sizeof(FIND_NAME_BUFFER)) >
Request->Buffer2Length) {
RELEASE_SPIN_LOCK (&Request->SpinLock, oldirql);
ACQUIRE_SPIN_LOCK (&DeviceContext->SpinLock,&oldirql);
RemoveEntryList (&Request->Linkage);
RELEASE_SPIN_LOCK (&DeviceContext->SpinLock, oldirql);
NbfCompleteRequest (Request, STATUS_SUCCESS, Request->BytesWritten);
NbfDereferenceRequest ("No Buffer", Request, RREF_FIND_NAME);
return STATUS_SUCCESS;
}
FindNameHeader = (PFIND_NAME_HEADER)TargetBuffer;
FindNameHeader->unique_group = UiFrame->Data2High;
Request->BytesWritten += sizeof(FIND_NAME_BUFFER);
++FindNameHeader->node_count;
RtlCopyMemory(FindNameBuffer->source_addr, SourceAddress->Address, 6);
RELEASE_SPIN_LOCK (&Request->SpinLock, oldirql);
RtlCopyMemory(FindNameBuffer->destination_addr, DestinationAddress, 6);
FindNameBuffer->length = 14;
if (DeviceContext->MacInfo.MediumType == NdisMedium802_5) {
//
// token-ring, copy the correct fields.
//
FindNameBuffer->access_control = Packet[0];
FindNameBuffer->frame_control = Packet[1];
if (SourceRouting != NULL) {
RtlCopyMemory (FindNameBuffer->routing_info, SourceRouting, SourceRoutingLength);
FindNameBuffer->length += (UCHAR) SourceRoutingLength;
}
} else {
//
// non-token-ring, nothing else is significant.
//
FindNameBuffer->access_control = 0x0;
FindNameBuffer->frame_control = 0x0;
}
//
// If this is a unique name, complete the request now.
//
if (UiFrame->Data2High == NETBIOS_NAME_TYPE_UNIQUE) {
ACQUIRE_SPIN_LOCK (&DeviceContext->SpinLock,&oldirql);
RemoveEntryList (&Request->Linkage);
RELEASE_SPIN_LOCK (&DeviceContext->SpinLock, oldirql);
NbfCompleteRequest(Request, STATUS_SUCCESS, Request->BytesWritten);
}
NbfDereferenceRequest ("NR processed", Request, RREF_FIND_NAME);
return STATUS_SUCCESS;
} /* NbfProcessQueryNameRecognized */
VOID
NbfSendStatusQuery(
IN PDEVICE_CONTEXT DeviceContext,
IN PTP_REQUEST Request,
IN PHARDWARE_ADDRESS DestinationAddress,
IN PUCHAR SourceRouting,
IN UINT SourceRoutingLength
)
/*++
Routine Description:
This routine will send a STATUS.NAME packet for the specified
find name request, and start the request timer.
Arguments:
DeviceContext - a pointer to the device context to send the status query on.
Request - The find name request.
DestinationAddress - The hardware destination address of the frame.
SourceRouting - Optional source routing information in the frame.
SourceRoutingLength - The length of SourceRouting.
Return Value:
NTSTATUS - status of operation.
--*/
{
TDI_ADDRESS_NETBIOS * remoteAddress;
PIO_STACK_LOCATION irpSp;
NTSTATUS Status;
PTP_UI_FRAME RawFrame;
PUCHAR SingleSR;
UINT SingleSRLength;
UINT HeaderLength;
LARGE_INTEGER Timeout;
UCHAR RequestType;
irpSp = IoGetCurrentIrpStackLocation (Request->IoRequestPacket);
remoteAddress = NbfParseTdiAddress(
((PTDI_REQUEST_KERNEL_QUERY_INFORMATION)(&irpSp->Parameters))->
RequestConnectionInformation->RemoteAddress, FALSE);
//
// Start the timer for this request.
//
Request->Flags |= REQUEST_FLAGS_TIMER; // there is a timeout on this request.
KeInitializeTimer (&Request->Timer); // set to not-signaled state.
NbfReferenceRequest ("Find Name: timer", Request, RREF_TIMER); // one for the timer
Timeout.LowPart = (ULONG)(-(LONG)DeviceContext->GeneralTimeout);
Timeout.HighPart = -1;
KeSetTimer (&Request->Timer, Timeout, &Request->Dpc);
//
// Allocate a UI frame from the pool.
//
Status = NbfCreateConnectionlessFrame (DeviceContext, &RawFrame);
if (!NT_SUCCESS (Status)) { // couldn't make frame.
return;
}
IF_NBFDBG (NBF_DEBUG_DEVCTX) {
NbfPrint2 ("NbfSendFindNames: Sending Frame: %lx, NdisPacket: %lx\n",
RawFrame, RawFrame->NdisPacket);
}
//
// Build the MAC header. STATUS_QUERY frames go out as
// single-route source routing.
//
MacReturnSingleRouteSR(
&DeviceContext->MacInfo,
&SingleSR,
&SingleSRLength);
MacConstructHeader (
&DeviceContext->MacInfo,
RawFrame->Header,
DeviceContext->NetBIOSAddress.Address,
DeviceContext->LocalAddress.Address,
sizeof (DLC_FRAME) + sizeof (NBF_HDR_CONNECTIONLESS),
SingleSR,
SingleSRLength,
&HeaderLength);
//
// Build the DLC UI frame header.
//
NbfBuildUIFrameHeader(&RawFrame->Header[HeaderLength]);
HeaderLength += sizeof(DLC_FRAME);
//
// Build the Netbios header.
//
//
// Determine what RequestType should be.
//
if (Request->BytesWritten == 0) {
//
// No way to know if he is 2.1 or not, so we put a 1 here
// instead of 0.
//
RequestType = 1;
} else {
RequestType = (UCHAR)((Request->BytesWritten - sizeof(ADAPTER_STATUS)) / sizeof(NAME_BUFFER));
}
ConstructStatusQuery (
(PNBF_HDR_CONNECTIONLESS)&(RawFrame->Header[HeaderLength]),
RequestType, // request status type.
(USHORT)Request->Buffer2Length, // user's buffer length
Request->FrameContext, // corr. in 1st NAME_RECOGNIZED.
(PNAME)remoteAddress->NetbiosName,
DeviceContext->ReservedNetBIOSAddress);
HeaderLength += sizeof(NBF_HDR_CONNECTIONLESS);
//
// Munge the packet length.
//
NbfSetNdisPacketLength(RawFrame->NdisPacket, HeaderLength);
NbfSendUIFrame (
DeviceContext,
RawFrame,
FALSE); // no loopback (MC frame)
} /* NbfSendStatusQuery */
NTSTATUS
NbfProcessStatusResponse(
IN PDEVICE_CONTEXT DeviceContext,
IN NDIS_HANDLE ReceiveContext,
IN PNBF_HDR_CONNECTIONLESS UiFrame,
IN PHARDWARE_ADDRESS SourceAddress,
IN PUCHAR SourceRouting,
IN UINT SourceRoutingLength
)
/*++
Routine Description:
This routine processes a STATUS.RESPONSE packet.
Arguments:
DeviceContext - a pointer to the device context the frame was received on.
ReceiveContext - The context for calling NdisTransferData.
UiFrame - The packet in question, starting at the Netbios header.
SourceAddress - The source hardware address of the packet.
SourceRouting - Source routing data in the query.
SourceRoutingLength - The length of SourceRouting.
Return Value:
NTSTATUS - status of operation.
--*/
{
KIRQL oldirql;
PTP_REQUEST Request;
PUCHAR TargetBuffer;
USHORT FrameContext;
USHORT NamesReceived;
USHORT ResponseLength, ResponseBytesToCopy;
PLIST_ENTRY p;
PSINGLE_LIST_ENTRY linkage;
NDIS_STATUS ndisStatus;
PNDIS_BUFFER NdisBuffer;
PNDIS_PACKET ndisPacket;
ULONG ndisBytesTransferred;
PRECEIVE_PACKET_TAG receiveTag;
NDIS_STATUS NdisStatus;
//
// Find the request that this is for, using the frame context.
//
FrameContext = TRANSMIT_CORR(UiFrame);
ACQUIRE_SPIN_LOCK (&DeviceContext->SpinLock, &oldirql);
for (p=DeviceContext->StatusQueryQueue.Flink;
p != &DeviceContext->StatusQueryQueue;
p=p->Flink) {
Request = CONTAINING_RECORD (p, TP_REQUEST, Linkage);
if (Request->FrameContext == FrameContext) {
break;
}
}
if (p == &DeviceContext->StatusQueryQueue) {
RELEASE_SPIN_LOCK (&DeviceContext->SpinLock, oldirql);
return STATUS_SUCCESS;
}
NbfReferenceRequest ("Status Response", Request, RREF_STATUS);
RELEASE_SPIN_LOCK (&DeviceContext->SpinLock, oldirql);
ACQUIRE_SPIN_LOCK (&Request->SpinLock, &oldirql);
//
// Make sure this request is not stopping.
//
if ((Request->Flags & REQUEST_FLAGS_STOPPING) != 0) {
RELEASE_SPIN_LOCK (&Request->SpinLock, oldirql);
NbfDereferenceRequest ("Stopping", Request, RREF_STATUS);
return STATUS_SUCCESS;
}
//
// See if this is packet has new data.
//
if (Request->BytesWritten == 0) {
NamesReceived = 0;
} else {
NamesReceived = (USHORT)(Request->BytesWritten - sizeof(ADAPTER_STATUS)) / sizeof(NAME_BUFFER);
}
if ((UiFrame->Data1 > 0) && (UiFrame->Data1 <= NamesReceived)) {
//
// If it is a post-2.1 response, but we already got
// this data, ignore it.
//
RELEASE_SPIN_LOCK (&Request->SpinLock, oldirql);
NbfDereferenceRequest ("Duplicate SR", Request, RREF_STATUS);
return STATUS_SUCCESS;
}
//
// This is new data, append if there is room.
//
ResponseLength = ((UiFrame->Data2High & 0x3f) * 256) + UiFrame->Data2Low;
if ((ULONG)(Request->BytesWritten + ResponseLength) >
Request->Buffer2Length) {
ResponseBytesToCopy = (USHORT)(Request->Buffer2Length - Request->BytesWritten);
} else {
ResponseBytesToCopy = ResponseLength;
}
//
// Allocate a receive packer for this operation.
//
linkage = ExInterlockedPopEntryList(
&DeviceContext->ReceivePacketPool,
&DeviceContext->Interlock);
if (linkage != NULL) {
ndisPacket = CONTAINING_RECORD( linkage, NDIS_PACKET, ProtocolReserved[0] );
} else {
//
// Could not get a packet, oh well, it is connectionless.
//
DeviceContext->ReceivePacketExhausted++;
RELEASE_SPIN_LOCK (&Request->SpinLock, oldirql);
return STATUS_SUCCESS;
}
receiveTag = (PRECEIVE_PACKET_TAG)(ndisPacket->ProtocolReserved);
receiveTag->PacketType = TYPE_STATUS_RESPONSE;
receiveTag->Connection = (PTP_CONNECTION)Request;
TargetBuffer = (PUCHAR)Request->ResponseBuffer + Request->BytesWritten;
//
// Allocate an MDL to describe the part of the buffer we
// want transferred.
//
NdisAllocateBuffer(
&NdisStatus,
&NdisBuffer,
DeviceContext->NdisBufferPool,
TargetBuffer,
ResponseBytesToCopy);
if (NdisStatus != NDIS_STATUS_SUCCESS) {
ExInterlockedPushEntryList(
&DeviceContext->ReceivePacketPool,
linkage,
&DeviceContext->Interlock);
RELEASE_SPIN_LOCK (&Request->SpinLock, oldirql);
return STATUS_SUCCESS;
}
//
// Assume success, if not we fail the request.
//
Request->BytesWritten += ResponseBytesToCopy;
RELEASE_SPIN_LOCK (&Request->SpinLock, oldirql);
NdisChainBufferAtFront(ndisPacket, NdisBuffer);
//
// See if the response was too big (we can complete the
// request here since we still reference it).
//
if ((ResponseLength > ResponseBytesToCopy) ||
(UiFrame->Data2High & 0x40)) {
ACQUIRE_SPIN_LOCK (&DeviceContext->SpinLock,&oldirql);
RemoveEntryList (&Request->Linkage);
RELEASE_SPIN_LOCK (&DeviceContext->SpinLock, oldirql);
receiveTag->CompleteReceive = TRUE;
receiveTag->EndOfMessage = FALSE;
} else {
//
// If we are done, complete the packet, otherwise send off
// the next request (unless it is a pre-2.1 response).
//
if ((UiFrame->Data1 > 0) && (UiFrame->Data2High & 0x80)) {
UCHAR TempSR[MAX_SOURCE_ROUTING];
PUCHAR ResponseSR;
receiveTag->CompleteReceive = FALSE;
//
// Try to cancel the timer, no harm if we fail.
//
ACQUIRE_SPIN_LOCK (&Request->SpinLock, &oldirql);
if ((Request->Flags & REQUEST_FLAGS_TIMER) != 0) {
Request->Flags &= ~REQUEST_FLAGS_TIMER;
RELEASE_SPIN_LOCK (&Request->SpinLock, oldirql);
if (KeCancelTimer (&Request->Timer)) {
NbfDereferenceRequest ("Status Response: stop timer", Request, RREF_TIMER);
}
} else {
RELEASE_SPIN_LOCK (&Request->SpinLock, oldirql);
}
Request->Retries = DeviceContext->GeneralRetries;
//
// Send a STATUS_QUERY directed.
//
if (SourceRouting != NULL) {
RtlCopyMemory(
TempSR,
SourceRouting,
SourceRoutingLength);
MacCreateNonBroadcastReplySR(
&DeviceContext->MacInfo,
TempSR,
SourceRoutingLength,
&ResponseSR);
} else {
ResponseSR = NULL;
}
NbfSendStatusQuery(
DeviceContext,
Request,
SourceAddress,
ResponseSR,
SourceRoutingLength);
} else {
ACQUIRE_SPIN_LOCK (&DeviceContext->SpinLock,&oldirql);
RemoveEntryList (&Request->Linkage);
RELEASE_SPIN_LOCK (&DeviceContext->SpinLock, oldirql);
receiveTag->CompleteReceive = TRUE;
receiveTag->EndOfMessage = TRUE;
}
}
//
// Now do the actual data transfer.
//
if (DeviceContext->NdisBindingHandle) {
NdisTransferData (
&ndisStatus,
DeviceContext->NdisBindingHandle,
ReceiveContext,
DeviceContext->MacInfo.TransferDataOffset +
3 + sizeof(NBF_HDR_CONNECTIONLESS),
ResponseBytesToCopy,
ndisPacket,
(PUINT)&ndisBytesTransferred);
}
else {
ndisStatus = STATUS_INVALID_DEVICE_STATE;
}
if (ndisStatus != NDIS_STATUS_PENDING) {
NbfTransferDataComplete(
(NDIS_HANDLE)DeviceContext,
ndisPacket,
ndisStatus,
ndisBytesTransferred);
}
return STATUS_SUCCESS;
} /* NbfProcessStatusResponse */
NTSTATUS
NbfTdiSetInformation(
IN PIRP Irp
)
/*++
Routine Description:
This routine performs the TdiSetInformation request for the transport
provider.
Arguments:
Irp - the Irp for the requested operation.
Return Value:
NTSTATUS - status of operation.
--*/
{
UNREFERENCED_PARAMETER (Irp); // prevent compiler warnings
return STATUS_NOT_IMPLEMENTED;
} /* NbfTdiQueryInformation */
#if 0
NTSTATUS
NbfQueryInfoEndpoint(
IN PTP_ENDPOINT Endpoint,
IN PTDI_REQ_QUERY_INFORMATION TdiRequest,
IN ULONG TdiRequestLength,
OUT PTDI_ENDPOINT_INFO InfoBuffer,
IN ULONG InfoBufferLength,
OUT PULONG InformationSize
)
/*++
Routine Description:
This routine returns information for the specified endpoint.
Arguments:
Endpoint - Pointer to transport endpoint context.
TdiRequest - Pointer to request buffer.
TdiRequestLength - Length of request buffer.
InfoBuffer - Pointer to output buffer to return information into.
InfoBufferLength - Length of output buffer.
InformationSize - Pointer to ulong where actual size of returned
information is to be stored.
Return Value:
NTSTATUS - status of operation.
--*/
{
KIRQL oldirql;
TdiRequest, TdiRequestLength; // prevent compiler warnings
if (InfoBufferLength < sizeof (TDI_ENDPOINT_INFO)) {
return STATUS_BUFFER_TOO_SMALL;
}
ACQUIRE_SPIN_LOCK (&Endpoint->SpinLock, &oldirql);
*InfoBuffer = Endpoint->Information; // structure copy.
RELEASE_SPIN_LOCK (&Endpoint->SpinLock, oldirql);
*InformationSize = sizeof (Endpoint->Information);
return STATUS_SUCCESS;
} /* NbfQueryInfoEndpoint */
NTSTATUS
NbfQueryInfoConnection(
IN PTP_CONNECTION Connection,
IN PTDI_REQUEST_KERNEL TdiRequest,
IN ULONG TdiRequestLength,
OUT PTDI_CONNECTION_INFO InfoBuffer,
IN ULONG InfoBufferLength,
OUT PULONG InformationSize
)
/*++
Routine Description:
This routine returns information for the specified connection.
Arguments:
Connection - Pointer to transport connection object.
TdiRequest - Pointer to request buffer.
TdiRequestLength - Length of request buffer.
InfoBuffer - Pointer to output buffer to return information into.
InfoBufferLength - Length of output buffer.
InformationSize - Pointer to ulong where actual size of returned
information is to be stored.
Return Value:
NTSTATUS - status of operation.
--*/
{
KIRQL oldirql;
TdiRequest, TdiRequestLength; // prevent compiler warnings
if (InfoBufferLength < sizeof (TDI_CONNECTION_INFO)) {
return STATUS_BUFFER_TOO_SMALL;
}
ACQUIRE_C_SPIN_LOCK (&Connection->SpinLock, &oldirql);
*InfoBuffer = Connection->Information; // structure copy.
RELEASE_C_SPIN_LOCK (&Connection->SpinLock, oldirql);
*InformationSize = sizeof (Connection->Information);
return STATUS_SUCCESS;
} /* NbfQueryInfoConnection */
NTSTATUS
NbfQueryInfoAddress(
IN PTP_ADDRESS Address,
IN PTDI_REQUEST_KERNEL TdiRequest,
IN ULONG TdiRequestLength,
OUT PTDI_ADDRESS_INFO InfoBuffer,
IN ULONG InfoBufferLength,
OUT PULONG InformationSize
)
/*++
Routine Description:
This routine returns information for the specified address. We
don't acquire a spinlock in this routine because there are no statistics
which must be read atomically.
Arguments:
Address - Pointer to transport address object.
TdiRequest - Pointer to request buffer.
TdiRequestLength - Length of request buffer.
InfoBuffer - Pointer to output buffer to return information into.
InfoBufferLength - Length of output buffer.
InformationSize - Pointer to ulong where actual size of returned
information is to be stored.
Return Value:
NTSTATUS - status of operation.
--*/
{
SHORT i;
PSZ p, q;
TdiRequest, TdiRequestLength; // prevent compiler warnings
//
// Calculate whether his buffer is big enough to return the entire
// information. The total size of the address information is the
// size of the fixed part, plus the size of the variable-length flat
// string in the NETWORK_NAME component of the TRANSPORT_ADDRESS
// component.
//
if (InfoBufferLength <
sizeof (TDI_ADDRESS_INFO) +
Address->NetworkName.Length)
{
return STATUS_BUFFER_TOO_SMALL;
}
//
// Copy both the fixed part of the address information, and the variable
// part. The variable part comes from the NETWORK_NAME component of the
// TRANSPORT_ADDRESS structure. This component contains a FLAT_STRING,
// which is of variable length.
//
InfoBuffer->Address.AddressComponents = Address->AddressComponents;
InfoBuffer->Address.Tsap = Address->Tsap;
InfoBuffer->Address.NetworkName.Name.Length =
Address->NetworkName.Length;
p = Address->NetworkName.Buffer; // p = ptr, source string.
q = InfoBuffer->Address.NetworkName.Name.Buffer; // q = ptr, dest string.
for (i=0; i<InfoBuffer->Address.NetworkName.Name.Length; i++) {
*(q++) = *(p++);
}
*InformationSize = sizeof (TDI_ADDRESS_INFO) +
Address->NetworkName.Length;
return STATUS_SUCCESS;
} /* NbfQueryInfoAddress */
NTSTATUS
NbfQueryInfoProvider(
IN PDEVICE_CONTEXT Provider,
IN PTDI_REQUEST_KERNEL TdiRequest,
IN ULONG TdiRequestLength,
OUT PTDI_PROVIDER_INFO InfoBuffer,
IN ULONG InfoBufferLength,
OUT PULONG InformationSize
)
/*++
Routine Description:
This routine returns information for the transport provider.
Arguments:
Provider - Pointer to device context for provider.
TdiRequest - Pointer to request buffer.
TdiRequestLength - Length of request buffer.
InfoBuffer - Pointer to output buffer to return information into.
InfoBufferLength - Length of output buffer.
InformationSize - Pointer to ulong where actual size of returned
information is to be stored.
Return Value:
NTSTATUS - status of operation.
--*/
{
KIRQL oldirql;
TdiRequest, TdiRequestLength; // prevent compiler warnings
if (InfoBufferLength < sizeof (TDI_PROVIDER_INFO)) {
return STATUS_BUFFER_TOO_SMALL;
}
ACQUIRE_SPIN_LOCK (&Provider->SpinLock, &oldirql);
*InfoBuffer = Provider->Information; // structure copy.
RELEASE_SPIN_LOCK (&Provider->SpinLock, oldirql);
*InformationSize = sizeof (Provider->Information);
return STATUS_SUCCESS;
} /* NbfQueryInfoProvider */
NTSTATUS
NbfQueryInfoNetman(
IN PDEVICE_CONTEXT Provider,
IN PTDI_REQUEST_KERNEL TdiRequest,
IN ULONG TdiRequestLength,
OUT PTDI_NETMAN_INFO InfoBuffer,
IN ULONG InfoBufferLength,
OUT PULONG InformationSize
)
/*++
Routine Description:
This routine returns information for the specified network-managable
variable managed by the transport provider.
Arguments:
Provider - Pointer to device context for provider.
TdiRequest - Pointer to request buffer.
TdiRequestLength - Length of request buffer.
InfoBuffer - Pointer to output buffer to return information into.
InfoBufferLength - Length of output buffer.
InformationSize - Pointer to ulong where actual size of returned
information is to be stored.
Return Value:
NTSTATUS - status of operation.
--*/
{
KIRQL oldirql;
PFLAT_STRING p;
PTP_VARIABLE v;
PTDI_NETMAN_VARIABLE n;
USHORT i;
ULONG NameOffset, ValueOffset;
TdiRequest, TdiRequestLength; // prevent compiler warnings
InfoBufferLength, InformationSize;
//
// check param lengths here.
//
ACQUIRE_SPIN_LOCK (&Provider->SpinLock, &oldirql);
NbfReferenceDeviceContext ("Query InfoNetMan", Provider, DCREF_QUERY_INFO);
for (v=Provider->NetmanVariables; v != NULL; v=v->Fwdlink) {
if (TdiRequest->Identification == v->VariableSerialNumber) {
//
// Return the variable information here.
//
NameOffset = sizeof (TDI_NETMAN_INFO);
ValueOffset = NameOffset + (sizeof (FLAT_STRING)-1) +
v->VariableName.Length;
InfoBuffer->VariableName = NameOffset;
InfoBuffer->VariableValue = ValueOffset;
//
// Copy the variable name to the user's buffer.
//
p = (PFLAT_STRING)((PUCHAR)InfoBuffer + NameOffset);
p->MaximumLength = v->VariableName.Length;
p->Length = v->VariableName.Length;
for (i=0; i<v->VariableName.Length; i++) {
p->Buffer [i] = v->VariableName.Buffer [i];
}
//
// Now copy the variable's contents to the user's buffer.
//
n = (PTDI_NETMAN_VARIABLE)((PUCHAR)InfoBuffer + ValueOffset);
n->VariableType = v->VariableType;
switch (v->VariableType) {
case NETMAN_VARTYPE_ULONG:
n->Value.LongValue = v->Value.LongValue;
break;
case NETMAN_VARTYPE_HARDWARE_ADDRESS:
n->Value.HardwareAddressValue =
v->Value.HardwareAddressValue;
break;
case NETMAN_VARTYPE_STRING:
p = &n->Value.StringValue;
p->MaximumLength = v->Value.StringValue.Length;
p->Length = v->Value.StringValue.Length;
for (i=0; i<v->Value.StringValue.Length; i++) {
p->Buffer [i] = v->Value.StringValue.Buffer [i];
}
} /* switch */
RELEASE_SPIN_LOCK (&Provider->SpinLock, oldirql);
NbfDereferenceDeviceContext ("Query InfoNetMan success", Provider, DCREF_QUERY_INFO);
return STATUS_SUCCESS;
} /* if */
} /* for */
RELEASE_SPIN_LOCK (&Provider->SpinLock, oldirql);
NbfDereferenceDeviceContext ("Query InfoNetMan no exist", Provider, DCREF_QUERY_INFO);
return STATUS_INVALID_INFO_CLASS; // variable does not exist.
} /* NbfQueryInfoNetman */
NTSTATUS
NbfSetInfoEndpoint(
IN PTP_ENDPOINT Endpoint,
IN PTDI_REQUEST_KERNEL TdiRequest,
IN ULONG TdiRequestLength
)
/*++
Routine Description:
This routine sets information for the specified endpoint.
Arguments:
Endpoint - Pointer to transport endpoint context.
TdiRequest - Pointer to request buffer.
TdiRequestLength - Length of request buffer.
Return Value:
NTSTATUS - status of operation.
--*/
{
KIRQL oldirql;
PTDI_ENDPOINT_INFO InfoBuffer;
if (TdiRequestLength !=
sizeof (TDI_ENDPOINT_INFO) + sizeof (TDI_REQ_SET_INFORMATION) -
sizeof (TDI_INFO_BUFFER)) {
return STATUS_BUFFER_TOO_SMALL; // buffer sizes must match.
}
InfoBuffer = (PTDI_ENDPOINT_INFO)&TdiRequest->InfoBuffer;
if ((InfoBuffer->MinimumLookaheadData <= NBF_MAX_LOOKAHEAD_DATA) ||
(InfoBuffer->MaximumLookaheadData <= NBF_MAX_LOOKAHEAD_DATA) ||
(InfoBuffer->MinimumLookaheadData > InfoBuffer->MaximumLookaheadData)) {
return STATUS_INVALID_PARAMETER;
}
ACQUIRE_SPIN_LOCK (&Endpoint->SpinLock, &oldirql);
//
// Set minimum lookahead data size. This is the number of bytes of
// contiguous data that will be supplied to TDI_IND_RECEIVE and
// TDI_IND_RECEIVE_DATAGRAM event handlers at indication time.
//
Endpoint->Information.MinimumLookaheadData = InfoBuffer->MinimumLookaheadData;
//
// Set maximum lookahead data size. This is the number of bytes of
// contiguous data that will be supplied to TDI_IND_RECEIVE and
// TDI_IND_RECEIVE_DATAGRAM event handlers at indication time.
//
Endpoint->Information.MaximumLookaheadData = InfoBuffer->MaximumLookaheadData;
//
// Reset all the statistics to his new values.
//
Endpoint->Information.TransmittedTsdus = InfoBuffer->TransmittedTsdus;
Endpoint->Information.ReceivedTsdus = InfoBuffer->ReceivedTsdus;
Endpoint->Information.TransmissionErrors = InfoBuffer->TransmissionErrors;
Endpoint->Information.ReceiveErrors = InfoBuffer->ReceiveErrors;
Endpoint->Information.PriorityLevel = InfoBuffer->PriorityLevel;
Endpoint->Information.SecurityLevel = InfoBuffer->SecurityLevel;
Endpoint->Information.SecurityCompartment = InfoBuffer->SecurityCompartment;
//
// The State and Event fields are read-only, so we DON'T set them here.
//
RELEASE_SPIN_LOCK (&Endpoint->SpinLock, oldirql);
return STATUS_SUCCESS;
} /* NbfSetInfoEndpoint */
NTSTATUS
NbfSetInfoAddress(
IN PTP_ADDRESS Address,
IN PTDI_REQUEST_KERNEL TdiRequest,
IN ULONG TdiRequestLength
)
/*++
Routine Description:
This routine sets information for the specified address. Currently,
all the user-visible fields in the transport address object are read-only.
Arguments:
Address - Pointer to transport address object.
TdiRequest - Pointer to request buffer.
TdiRequestLength - Length of request buffer.
Return Value:
NTSTATUS - status of operation.
--*/
{
Address, TdiRequest, TdiRequestLength; // prevent compiler warnings
return STATUS_SUCCESS;
} /* NbfSetInfoAddress */
NTSTATUS
NbfSetInfoConnection(
IN PTP_CONNECTION Connection,
IN PTDI_REQUEST_KERNEL TdiRequest,
IN ULONG TdiRequestLength
)
/*++
Routine Description:
This routine sets information for the specified connection.
Arguments:
Connection - Pointer to transport connection object.
TdiRequest - Pointer to request buffer.
TdiRequestLength - Length of request buffer.
Return Value:
NTSTATUS - status of operation.
--*/
{
KIRQL oldirql;
PTDI_CONNECTION_INFO InfoBuffer;
if (TdiRequestLength !=
sizeof (TDI_CONNECTION_INFO) + sizeof (TDI_REQ_SET_INFORMATION) -
sizeof (TDI_INFO_BUFFER)) {
return STATUS_BUFFER_TOO_SMALL; // buffer sizes must match.
}
InfoBuffer = (PTDI_CONNECTION_INFO)&TdiRequest->InfoBuffer;
ACQUIRE_C_SPIN_LOCK (&Connection->SpinLock, &oldirql);
//
// Reset all the statistics to his new values.
//
Connection->Information.TransmittedTsdus = InfoBuffer->TransmittedTsdus;
Connection->Information.ReceivedTsdus = InfoBuffer->ReceivedTsdus;
Connection->Information.TransmissionErrors = InfoBuffer->TransmissionErrors;
Connection->Information.ReceiveErrors = InfoBuffer->ReceiveErrors;
//
// The State and Event fields are read-only, so we DON'T set them here.
//
RELEASE_C_SPIN_LOCK (&Connection->SpinLock, oldirql);
return STATUS_SUCCESS;
} /* NbfSetInfoConnection */
NTSTATUS
NbfSetInfoProvider(
IN PDEVICE_CONTEXT Provider,
IN PTDI_REQUEST_KERNEL TdiRequest,
IN ULONG TdiRequestLength
)
/*++
Routine Description:
This routine sets information for the specified transport provider.
Arguments:
Provider - Pointer to device context.
TdiRequest - Pointer to request buffer.
TdiRequestLength - Length of request buffer.
Return Value:
NTSTATUS - status of operation.
--*/
{
KIRQL oldirql;
PTDI_PROVIDER_INFO InfoBuffer;
if (TdiRequestLength !=
sizeof (TDI_PROVIDER_INFO) + sizeof (TDI_REQ_SET_INFORMATION) -
sizeof (TDI_INFO_BUFFER)) {
return STATUS_BUFFER_TOO_SMALL; // buffer sizes must match.
}
InfoBuffer = (PTDI_PROVIDER_INFO)&TdiRequest->InfoBuffer;
//
// By changing the service flags the caller can request additional
// or fewer services on the fly. Make sure that he is requesting
// services we can provide, or else fail the request.
//
if (InfoBuffer->ServiceFlags & ~NBF_SERVICE_FLAGS) {
return STATUS_NOT_SUPPORTED;
}
ACQUIRE_SPIN_LOCK (&Provider->SpinLock, &oldirql);
//
// Reset all the statistics to his new values.
//
Provider->Information.TransmittedTsdus = InfoBuffer->TransmittedTsdus;
Provider->Information.ReceivedTsdus = InfoBuffer->ReceivedTsdus;
Provider->Information.TransmissionErrors = InfoBuffer->TransmissionErrors;
Provider->Information.ReceiveErrors = InfoBuffer->ReceiveErrors;
Provider->Information.DiscardedFrames = InfoBuffer->DiscardedFrames;
Provider->Information.ReceiveErrors = InfoBuffer->ReceiveErrors;
Provider->Information.OversizeTsdusReceived = InfoBuffer->OversizeTsdusReceived;
Provider->Information.UndersizeTsdusReceived = InfoBuffer->UndersizeTsdusReceived;
Provider->Information.MulticastTsdusReceived = InfoBuffer->MulticastTsdusReceived;
Provider->Information.BroadcastTsdusReceived = InfoBuffer->BroadcastTsdusReceived;
Provider->Information.MulticastTsdusTransmitted = InfoBuffer->MulticastTsdusTransmitted;
Provider->Information.BroadcastTsdusTransmitted = InfoBuffer->BroadcastTsdusTransmitted;
Provider->Information.SendTimeouts = InfoBuffer->SendTimeouts;
Provider->Information.ReceiveTimeouts = InfoBuffer->ReceiveTimeouts;
Provider->Information.ConnectionIndicationsReceived = InfoBuffer->ConnectionIndicationsReceived;
Provider->Information.ConnectionIndicationsAccepted = InfoBuffer->ConnectionIndicationsAccepted;
Provider->Information.ConnectionsInitiated = InfoBuffer->ConnectionsInitiated;
Provider->Information.ConnectionsAccepted = InfoBuffer->ConnectionsAccepted;
//
// The following fields are read-only, so we DON'T set them here:
// Version, MaxTsduSize, MaxConnectionUserData, MinimumLookaheadData,
// MaximumLookaheadData.
//
RELEASE_SPIN_LOCK (&Provider->SpinLock, oldirql);
return STATUS_SUCCESS;
} /* NbfSetInfoProvider */
NTSTATUS
NbfSetInfoNetman(
IN PDEVICE_CONTEXT Provider,
IN PTDI_REQ_SET_INFORMATION TdiRequest,
IN ULONG TdiRequestLength
)
/*++
Routine Description:
This routine sets information for the specified transport provider's
network-managable variable.
Arguments:
Provider - Pointer to device context.
TdiRequest - Pointer to request buffer.
TdiRequestLength - Length of request buffer.
Return Value:
NTSTATUS - status of operation.
--*/
{
PTDI_NETMAN_INFO InfoBuffer;
Provider; // prevent compiler warnings
if (TdiRequestLength !=
sizeof (TDI_NETMAN_INFO) + sizeof (TDI_REQ_SET_INFORMATION) -
sizeof (TDI_INFO_BUFFER)) {
return STATUS_BUFFER_TOO_SMALL; // buffer sizes must match.
}
InfoBuffer = (PTDI_NETMAN_INFO)&TdiRequest->InfoBuffer;
//
// set the network-managable variable here.
//
return STATUS_SUCCESS;
} /* NbfSetInfoNetman */
NTSTATUS
NbfTdiQueryInformation(
IN PTP_ENDPOINT Endpoint,
IN PTDI_REQ_QUERY_INFORMATION TdiRequest,
IN ULONG TdiRequestLength,
OUT PTDI_INFO_BUFFER InfoBuffer,
IN ULONG InfoBufferLength,
OUT PULONG InformationSize
)
/*++
Routine Description:
This routine performs the TdiQueryInformation request for the transport
provider.
Arguments:
Endpoint - Pointer to transport endpoint context.
TdiRequest - Pointer to request buffer.
TdiRequestLength - Length of request buffer.
InfoBuffer - Pointer to output buffer to return information into.
InfoBufferLength - Length of output buffer.
InformationSize - Pointer to ulong where actual size of returned
information is to be stored.
Return Value:
NTSTATUS - status of operation.
--*/
{
NTSTATUS Status;
PTP_CONNECTION Connection;
switch (TdiRequest->InformationClass) {
//
// ENDPOINT information: return information about the endpoint
// to which this request was submitted.
//
case TDI_INFO_CLASS_ENDPOINT:
Status = NbfQueryInfoEndpoint (
Endpoint,
TdiRequest,
TdiRequestLength,
(PTDI_ENDPOINT_INFO)InfoBuffer,
InfoBufferLength,
InformationSize);
break;
//
// CONNECTION information: return information about a connection
// that is associated with the endpoint on which this request was
// submitted.
//
case TDI_INFO_CLASS_CONNECTION:
// This causes a connection reference which is removed below.
Connection = NbfLookupConnectionById (
Endpoint,
TdiRequest->Identification);
if (Connection == NULL) {
Status = STATUS_INVALID_HANDLE;
break;
}
Status = NbfQueryInfoConnection (
Connection,
TdiRequest,
TdiRequestLength,
(PTDI_CONNECTION_INFO)InfoBuffer,
InfoBufferLength,
InformationSize);
NbfDereferenceConnection("Query Connection Info", Connection, CREF_BY_ID);
break;
//
// ADDRESS information: return information about the address object
// that is associated with the endpoint on which this request was
// submitted.
//
case TDI_INFO_CLASS_ADDRESS:
Status = NbfQueryInfoAddress (
Endpoint->BoundAddress,
TdiRequest,
TdiRequestLength,
(PTDI_ADDRESS_INFO)InfoBuffer,
InfoBufferLength,
InformationSize);
break;
//
// PROVIDER information: return information about the transport
// provider itself.
//
case TDI_INFO_CLASS_PROVIDER:
Status = NbfQueryInfoProvider (
Endpoint->BoundAddress->Provider,
TdiRequest,
TdiRequestLength,
(PTDI_PROVIDER_INFO)InfoBuffer,
InfoBufferLength,
InformationSize);
break;
//
// NETMAN information: return information about the network-managable
// variables managed by the provider itself.
//
case TDI_INFO_CLASS_NETMAN:
Status = NbfQueryInfoNetman (
Endpoint->BoundAddress->Provider,
TdiRequest,
TdiRequestLength,
(PTDI_NETMAN_INFO)InfoBuffer,
InfoBufferLength,
InformationSize);
break;
default:
Status = STATUS_INVALID_INFO_CLASS;
} /* switch */
return Status;
} /* TdiQueryInformation */
NTSTATUS
TdiSetInformation(
IN PTP_ENDPOINT Endpoint,
IN PTDI_REQ_SET_INFORMATION TdiRequest,
IN ULONG TdiRequestLength
)
/*++
Routine Description:
This routine performs the TdiSetInformation request for the transport
provider.
Arguments:
Endpoint - Pointer to transport endpoint context.
TdiRequest - Pointer to request buffer.
TdiRequestLength - Length of request buffer.
Return Value:
NTSTATUS - status of operation.
--*/
{
NTSTATUS Status;
PTP_CONNECTION Connection;
switch (TdiRequest->InformationClass) {
//
// ENDPOINT information: set information on the endpoint
// to which this request was submitted.
//
case TDI_INFO_CLASS_ENDPOINT:
Status = NbfSetInfoEndpoint (
Endpoint,
TdiRequest,
TdiRequestLength);
break;
//
// CONNECTION information: set information for a connection
// that is associated with the endpoint on which this request
// was submitted.
//
case TDI_INFO_CLASS_CONNECTION:
// This causes a connection reference which is removed below.
Connection = NbfLookupConnectionById (
Endpoint,
TdiRequest->Identification);
if (Connection == NULL) {
Status = STATUS_INVALID_HANDLE;
break;
}
Status = NbfSetInfoConnection (
Connection,
TdiRequest,
TdiRequestLength);
NbfDereferenceConnection("Set Connection Info", Connection, CREF_BY_ID);
break;
//
// ADDRESS information: set information for the address object
// that is associated with the endpoint on which this request
// was submitted.
//
case TDI_INFO_CLASS_ADDRESS:
Status = NbfSetInfoAddress (
Endpoint->BoundAddress,
TdiRequest,
TdiRequestLength);
break;
//
// PROVIDER information: set information for the transport
// provider itself.
//
case TDI_INFO_CLASS_PROVIDER:
Status = NbfSetInfoProvider (
Endpoint->BoundAddress->Provider,
TdiRequest,
TdiRequestLength);
break;
//
// NETMAN information: set information for the network-managable
// variables managed by the provider itself.
//
case TDI_INFO_CLASS_NETMAN:
Status = NbfSetInfoNetman (
Endpoint->BoundAddress->Provider,
TdiRequest,
TdiRequestLength);
break;
default:
Status = STATUS_INVALID_INFO_CLASS;
} /* switch */
return Status;
} /* TdiSetInformation */
#endif
Reference in a new issue Copy permalink