我正在编写Win98/Win95下的防火墙程序,用98DDK中的NDIS3,有些疑问?
jmsly 2001-04-12 08:25:00 我认为应该在READ.C中写防火墙的一些过滤程序,可是我现在无法确定
具体的位置,98DDK提供的READ.C文件如下:
/*++
Copyright (c) 1996 Microsoft Corporation
Module Name:
packet.c
Abstract:
Author:
William Ingle
Environment:
Ndis 3.1 Windows 95 VxD
Notes:
Copyright (c) 1996, Microsoft Corporation, all rights reserve
This document is provided for informational purposes only and Microsoft
Corporation makes no warranties, either expressed or implied, in this document.
Information in this document may be substantially changed without notice in
subsequent versions of windows and does not represent a commitment on the
part of Microsoft Corporation.
--*/
#include <basedef.h>
#include <vmm.h>
#include <vwin32.h>
#include <winerror.h>
#include <ndis.h>
#include "debug.h"
#include "packet.h"
#include "ntddpack.h"
#pragma VxD_LOCKED_CODE_SEG
#pragma VxD_LOCKED_DATA_SEG
DWORD _stdcall MyPageLock(DWORD, DWORD);
void _stdcall MyPageUnlock(DWORD, DWORD);
VOID
PacketAllocatePacketBuffer( PNDIS_STATUS pStatus,
POPEN_INSTANCE pOpen,
PNDIS_PACKET *lplpPacket,
PDIOCPARAMETERS pDiocParms,
DWORD FunctionCode )
/*++
Routine Description:
Arguments:
Return Value:
Status is returned.
--*/
{
PNDIS_BUFFER pNdisBuffer;
PPACKET_RESERVED pReserved;
TRACE_ENTER( "PacketAllocatePacket" );
//
// Try to get a packet from our list of free ones
//
NdisAllocatePacket( pStatus, lplpPacket, pOpen->PacketPool );
if ( *pStatus != NDIS_STATUS_SUCCESS )
{
IF_VERY_LOUD( "Read- No free packets" );
*(DWORD *)(pDiocParms->lpcbBytesReturned) = 0;
return;
}
//
// Initialize Reserved LIST_ELEMENT
//
InitializeListHead( &(RESERVED(*lplpPacket)->ListElement) );
pReserved = RESERVED(*lplpPacket);
//
// Buffers used asynchronously must be page locked
//
switch ( FunctionCode )
{
case IOCTL_PROTOCOL_READ:
pReserved->lpBuffer = (PVOID)PacketPageLock( (PVOID)pDiocParms->lpvOutBuffer,
pDiocParms->cbOutBuffer );
pReserved->cbBuffer = pDiocParms->cbOutBuffer;
break;
case IOCTL_PROTOCOL_WRITE:
pReserved->lpBuffer = (PVOID)PacketPageLock( pDiocParms->lpvInBuffer,
pDiocParms->cbInBuffer );
pReserved->cbBuffer = pDiocParms->cbInBuffer;
break;
default:
IF_TRACE_MSG( "Allocate- Invalid FunctionCode %x", FunctionCode );
// recycle the packet
NdisReinitializePacket( *lplpPacket );
// Put the packet on the free queue
NdisFreePacket( *lplpPacket );
*(DWORD *)(pDiocParms->lpcbBytesReturned) = 0;
*pStatus = NDIS_STATUS_NOT_ACCEPTED;
return;
}
//
// Buffers used by pended I/O operations must be page locked
//
pReserved->lpcbBytesReturned =
(PVOID)PacketPageLock( (PVOID)pDiocParms->lpcbBytesReturned, sizeof(DWORD) );
pReserved->lpoOverlapped =
(PVOID)PacketPageLock( (PVOID)pDiocParms->lpoOverlapped, sizeof(OVERLAPPED) );
pReserved->hDevice = pDiocParms->hDevice;
pReserved->tagProcess = pDiocParms->tagProcess;
//
// Here we allocate a buffer descriptor for our page locked buffer given us by the
// client application.
//
NdisAllocateBuffer( pStatus,
&pNdisBuffer,
pOpen->BufferPool,
(PVOID)pReserved->lpBuffer,
pDiocParms->cbOutBuffer );
if ( *pStatus != NDIS_STATUS_SUCCESS )
{
IF_TRACE( "Read- No free buffers" );
// recycle the packet
NdisReinitializePacket(*lplpPacket);
// Put the packet on the free queue
NdisFreePacket(*lplpPacket);
*(DWORD *)(pDiocParms->lpcbBytesReturned) = 0;
return;
}
// Attach buffer to Packet
NdisChainBufferAtFront( *lplpPacket, pNdisBuffer );
// Let's only print this when PACKET_DEBUG_VERY_LOUD is set
IF_PACKETDEBUG( PACKET_DEBUG_VERY_LOUD )
{
IF_TRACE_MSG( " lplpPacket : %lx", lplpPacket );
IF_TRACE_MSG( " lpPacket : %lx", *lplpPacket );
IF_TRACE_MSG3( "pNdisBuffer : %lx %lx %lx", pNdisBuffer, (*lplpPacket)->Private.Head, (*lplpPacket)->Private.Tail );
IF_TRACE_MSG( " Reserved : %lx", pReserved );
IF_TRACE_MSG4( " lpBuffer : %lx %lx %lx %lx", pReserved->lpBuffer, pNdisBuffer->VirtualAddress, pDiocParms->lpvOutBuffer, pDiocParms->lpvInBuffer );
IF_TRACE_MSG3( " cbBuffer : %lx %lx %lx", pReserved->cbBuffer, pDiocParms->cbOutBuffer, pDiocParms->cbInBuffer );
IF_TRACE_MSG2( " lpcbBytes : %lx %lx", pReserved->lpcbBytesReturned, pDiocParms->lpcbBytesReturned );
IF_TRACE_MSG2( " lpoOverlap : %lx %lx", pReserved->lpoOverlapped, pDiocParms->lpoOverlapped );
IF_TRACE_MSG2( " hDevice : %lx %lx", pReserved->hDevice, pDiocParms->hDevice );
IF_TRACE_MSG2( " tagProcess : %lx %lx", pReserved->tagProcess, pDiocParms->tagProcess );
}
PACKETASSERT( pReserved->lpBuffer );
PACKETASSERT( pReserved->cbBuffer );
PACKETASSERT( pReserved->lpcbBytesReturned );
PACKETASSERT( pReserved->lpoOverlapped );
PACKETASSERT( pReserved->hDevice == pDiocParms->hDevice );
PACKETASSERT( pReserved->tagProcess == pDiocParms->tagProcess );
PACKETASSERT( pNdisBuffer == (*lplpPacket)->Private.Head );
PACKETASSERT( pNdisBuffer->VirtualAddress == pReserved->lpBuffer );
TRACE_LEAVE( "PacketAllocatePacket" );
return;
}
DWORD
PacketRead( POPEN_INSTANCE Open,
DWORD dwDDB,
DWORD hDevice,
PDIOCPARAMETERS pDiocParms
)
/*++
Routine Description:
This is the dispatch routine for create/open and close requests.
These requests complete successfully.
Arguments:
Return Value:
Status is returned.
--*/
{
NDIS_STATUS Status;
PNDIS_PACKET pPacket;
TRACE_ENTER( "PacketRead" );
//
// See if the buffer is at least big enough to hold the
// ethernet header
//
if ( pDiocParms->cbOutBuffer < ETHERNET_HEADER_LENGTH )
{
// Need bigger buffer
*(DWORD *)(pDiocParms->lpcbBytesReturned) = 0;
IF_VERY_LOUD( "Read- Buffer too small" );
TRACE_LEAVE( "ReadPacket" );
return NDIS_STATUS_SUCCESS;
}
PacketAllocatePacketBuffer( &Status, Open, &pPacket, pDiocParms, IOCTL_PROTOCOL_READ );
if ( Status == NDIS_STATUS_SUCCESS )
{
//
// Put this packet in a list of pending reads.
// The receive indication handler will attemp to remove packets
// from this list for use in transfer data calls
//
PACKETASSERT( Open != NULL );
PACKETASSERT( pPacket != NULL );
NdisAcquireSpinLock( &Open->RcvQSpinLock );
InsertTailList( &Open->RcvList, &RESERVED(pPacket)->ListElement );
NdisReleaseSpinLock( &Open->RcvQSpinLock );
IF_TRACE_MSG2( "RcvList Link : %lx %lx", Open->RcvList.Blink, &RESERVED(pPacket)->ListElement );
PACKETASSERT( Open->RcvList.Blink == &RESERVED(pPacket)->ListElement );
PACKETASSERT( &(Open->RcvList) == RESERVED(pPacket)->ListElement.Flink );
}
TRACE_LEAVE( "PacketRead" );
return(-1); // This will make DeviceIOControl return ERROR_IO_PENDING
}
NDIS_STATUS NDIS_API
PacketReceiveIndicate ( IN NDIS_HANDLE ProtocolBindingContext,
IN NDIS_HANDLE MacReceiveContext,
IN PVOID pvHeaderBuffer,
IN UINT uiHeaderBufferSize,
IN PVOID pvLookAheadBuffer,
IN UINT uiLookaheadBufferSize,
IN UINT uiPacketSize
)
/*++
Routine Description:
Arguments:
Return Value:
Status is returned.
--*/
#define pOpen ((POPEN_INSTANCE)ProtocolBindingContext)
{
PLIST_ENTRY PacketListEntry;
PNDIS_PACKET pPacket;
ULONG ulSizeToTransfer;
NDIS_STATUS Status;
UINT uiBytesTransferred;
PPACKET_RESERVED pReserved;
PNDIS_BUFFER pNdisBuffer;
PVOID pvActualVirtualAddress;
UINT uiActualLength;
// we wait to print the trace to see if we even have a buffer
PACKETASSERT( (pOpen != NULL) );
//
// See if there are any pending read that we can satisfy
//
NdisAcquireSpinLock( &pOpen->RcvQSpinLock );
PacketListEntry = PacketRemoveHeadList( &pOpen->RcvList );
NdisReleaseSpinLock( &pOpen->RcvQSpinLock );
if ( PacketListEntry == NULL )
{
return NDIS_STATUS_SUCCESS;
}
TRACE_ENTER( "IndicateReceive" );
pReserved = CONTAINING_RECORD( PacketListEntry, PACKET_RESERVED, ListElement );
pPacket = CONTAINING_RECORD( pReserved, NDIS_PACKET, ProtocolReserved );
// Let's only print this when PACKET_DEBUG_VERY_LOUD is set
IF_PACKETDEBUG( PACKET_DEBUG_VERY_LOUD )
{
IF_TRACE_MSG( " Reserved : %lx", pReserved );
IF_TRACE_MSG( " pPacket : %lx", pPacket );
IF_TRACE_MSG2( " Header : %lx %lx", pvHeaderBuffer, uiHeaderBufferSize );
IF_TRACE_MSG2( " LookAhead : %lx %lx", pvLookAheadBuffer, uiLookaheadBufferSize );
IF_TRACE_MSG( " PacketSize : %lx", uiPacketSize );
}
PACKETASSERT( (pReserved != NULL) );
PACKETASSERT( (pPacket != NULL) );
// Clear total
uiBytesTransferred = 0;
// Get pointer to private buffer
pNdisBuffer = pPacket->Private.Head;
// Save private buffer's start address
pvActualVirtualAddress = pNdisBuffer->VirtualAddress;
uiActualLength = pNdisBuffer->Length;
// Copy header buffer into client buffer
if ( uiHeaderBufferSize > 0 )
{
if ( uiHeaderBufferSize > pNdisBuffer->Length )
uiHeaderBufferSize = pNdisBuffer->Length;
// copy the header buffer
NdisMoveMemory( pNdisBuffer->VirtualAddress, pvHeaderBuffer, uiHeaderBufferSize );
// update bytes transferred
uiBytesTransferred += uiHeaderBufferSize;
// Advance the NDIS_BUFFER address to the end of the header
(BYTE *)(pNdisBuffer->VirtualAddress) += uiHeaderBufferSize;
// Adjust the NDIS_BUFFER Length
pNdisBuffer->Length -= uiHeaderBufferSize;
}
// Copy look ahead buffer into client buffer
if ( uiLookaheadBufferSize > 0 )
{
if ( uiLookaheadBufferSize > pNdisBuffer->Length )
uiLookaheadBufferSize = pNdisBuffer->Length;
NdisMoveMemory( pNdisBuffer->VirtualAddress, pvLookAheadBuffer, uiLookaheadBufferSize );
// update bytes transferred
uiBytesTransferred += uiLookaheadBufferSize;
// Advance the NDIS_BUFFER address to the end of the Lookahead data
(BYTE *)(pNdisBuffer->VirtualAddress) += uiLookaheadBufferSize;
// Adjust the NDIS_BUFFER Length
pNdisBuffer->Length -= uiLookaheadBufferSize;
}
// Save bytes transferred to client buffer so far
*(pReserved->lpcbBytesReturned) = uiBytesTransferred;
// Copy any remaining bytes in packet
if ( uiLookaheadBufferSize < uiPacketSize )
{
if ( uiPacketSize >= 1500 )
{
_asm { int 3 }
}
// Adjust packet size by the look ahead buffer
ulSizeToTransfer = uiPacketSize - uiLookaheadBufferSize;
// Make sure we only copy as many bytes as the client buffer can handler
if ( ulSizeToTransfer > pNdisBuffer->Length )
ulSizeToTransfer = pNdisBuffer->Length;
// Call the Mac to transfer the packet
NdisTransferData( &Status, // ndis status
pOpen->AdapterHandle, // from NdisOpenAdapter
MacReceiveContext, // handle from NIC
uiLookaheadBufferSize, // offset from start of buffer
ulSizeToTransfer, // number of bytes to copy
pPacket, // packet descriptor pointer
&uiBytesTransferred ); // number of bytes actually copied
// Restore the private buffer's address pointer
pNdisBuffer->VirtualAddress = pvActualVirtualAddress;
pNdisBuffer->Length = uiActualLength;
if ( Status != NDIS_STATUS_PENDING )
{
// If it didn't pend, call the completeion routine now
PacketTransferDataComplete( pOpen, // protocol context
pPacket, // packet descriptor pointer
Status, // ndis status
uiBytesTransferred );// number of bytes actually copied
}
}
else
{
// The entire packet was in the look ahead buffer
Status = NDIS_STATUS_SUCCESS;
// Restore the private buffer's address pointer
pNdisBuffer->VirtualAddress = pvActualVirtualAddress;
pNdisBuffer->Length = uiActualLength;
PacketTransferDataComplete( pOpen, // protocol context
pPacket, // packet descriptor pointer
Status, // ndis status
0 ); // number of bytes copied by NdisTransferData
}
TRACE_LEAVE( "IndicateReceive" );
return NDIS_STATUS_SUCCESS;
}
VOID NDIS_API
PacketTransferDataComplete ( IN NDIS_HANDLE ProtocolBindingContext,
IN PNDIS_PACKET pPacket,
IN NDIS_STATUS Status,
IN UINT uiBytesTransferred
)
/*++
Routine Description:
Arguments:
Return Value:
Status is returned.
--*/
{
PPACKET_RESERVED pReserved;
OVERLAPPED* pOverlap;
PNDIS_BUFFER pNdisBuffer;
TRACE_ENTER( "TransferDataComplete" );
pReserved = (PPACKET_RESERVED) pPacket->ProtocolReserved;
pOverlap = (OVERLAPPED *) pReserved->lpoOverlapped;
PACKETASSERT( (pOpen != NULL) );
PACKETASSERT( (pReserved != NULL) );
PACKETASSERT( (pOverlap != NULL) );
// Let's only print this when PACKET_DEBUG_VERY_LOUD is set
IF_PACKETDEBUG( PACKET_DEBUG_VERY_LOUD )
{
IF_TRACE_MSG( " Status : %lx", Status );
IF_TRACE_MSG( "BytesXfered : %lx", uiBytesTransferred );
IF_TRACE_MSG( "Byte Offset : %lx", *(pReserved->lpcbBytesReturned) );
}
// free buffer descriptor
NdisUnchainBufferAtFront( pPacket, &pNdisBuffer );
PACKETASSERT( (pNdisBuffer != NULL) );
if ( pNdisBuffer )
NdisFreeBuffer( pNdisBuffer );
// increment total bytes returned and set in overlapped structure
*(pReserved->lpcbBytesReturned) += uiBytesTransferred;
pOverlap->O_InternalHigh = *(pReserved->lpcbBytesReturned);
// The internal member of overlapped structure contains
// a pointer to the event structure that will be signalled,
// resuming the execution of the waitng GetOverlappedResult
// call.
VWIN32_DIOCCompletionRoutine( pOverlap->O_Internal );
// Unlock buffers
PacketPageUnlock( pReserved->lpBuffer, pReserved->cbBuffer );
PacketPageUnlock( pReserved->lpcbBytesReturned, sizeof(DWORD) );
PacketPageUnlock( pReserved->lpoOverlapped, sizeof(OVERLAPPED) );
// recycle the packet
NdisReinitializePacket( pPacket );
// Put the packet on the free queue
NdisFreePacket( pPacket );
TRACE_LEAVE( "TransferDataComplete" );
return;
}
VOID NDIS_API
PacketReceiveComplete( IN NDIS_HANDLE ProtocolBindingContext )
/*++
Routine Description:
Arguments:
Return Value:
Status is returned.
--*/
{
IF_PACKETDEBUG( PACKET_DEBUG_VERY_LOUD )
{
TRACE_ENTER( "ReceiveComplete" );
TRACE_LEAVE( "ReceiveComplete" );
}
return;
}