mockbuffer.c File Reference

#include "precomp.h"

Include dependency graph for mockbuffer.c:

Go to the source code of this file.

Functions
VOID	XNdisGetFirstBufferFromPacket (PNDIS_PACKET Packet, PNDIS_BUFFER *FirstBuffer, PVOID *FirstBufferVA, PUINT FirstBufferLength, PUINT TotalBufferLength)
VOID	XNdisQueryBuffer (PNDIS_BUFFER Buffer, PVOID *VirtualAddress, PUINT Length)
__inline INT	SkipToOffset (PNDIS_BUFFER Buffer, UINT Offset, PCHAR *Data, PUINT Size)
UINT	CopyBufferToBufferChain (PNDIS_BUFFER DstBuffer, UINT DstOffset, PCHAR SrcData, UINT Length)
UINT	CopyBufferChainToBuffer (PCHAR DstData, PNDIS_BUFFER SrcBuffer, UINT SrcOffset, UINT Length)
UINT	CopyPacketToBuffer (PCHAR DstData, PNDIS_PACKET SrcPacket, UINT SrcOffset, UINT Length)
UINT	CopyPacketToBufferChain (PNDIS_BUFFER DstBuffer, UINT DstOffset, PNDIS_PACKET SrcPacket, UINT SrcOffset, UINT Length)
PVOID	AdjustPacket (PNDIS_PACKET Packet, UINT Available, UINT Needed)
UINT	ResizePacket (PNDIS_PACKET Packet, UINT Size)
NDIS_STATUS	PrependPacket (PNDIS_PACKET Packet, PCHAR Data, UINT Length, BOOLEAN Copy)
void	GetDataPtr (PNDIS_PACKET Packet, UINT Offset, PCHAR *DataOut, PUINT Size)
NDIS_STATUS	AllocatePacketWithBufferX (PNDIS_PACKET *NdisPacket, PCHAR Data, UINT Len, PCHAR File, UINT Line)
VOID	FreeNdisPacketX (PNDIS_PACKET Packet, PCHAR File, UINT Line)

Function Documentation

AdjustPacket()

PVOID AdjustPacket	(	PNDIS_PACKET	Packet,
		UINT	Available,
		UINT	Needed
	)

Definition at line 334 of file mockbuffer.c.

{
    PNDIS_BUFFER NdisBuffer;
    INT Adjust;
 
    TI_DbgPrint(DEBUG_PBUFFER, ("Available = %d, Needed = %d.\n", Available, Needed));
 
    Adjust = Available - Needed;
 
    NdisQueryPacket(Packet, NULL, NULL, &NdisBuffer, NULL);
 
    /* If Adjust is zero there is no need to adjust this packet as
       there is no additional space at start the of first buffer */
    if (Adjust != 0) {
        NdisBuffer->MappedSystemVa  = (PVOID) ((ULONG_PTR)(NdisBuffer->MappedSystemVa) + Adjust);
        NdisBuffer->ByteOffset     += Adjust;
        NdisBuffer->ByteCount      -= Adjust;
    }
 
    return NdisBuffer->MappedSystemVa;
}

AllocatePacketWithBufferX()

NDIS_STATUS AllocatePacketWithBufferX	(	PNDIS_PACKET *	NdisPacket,
		PCHAR	Data,
		UINT	Len,
		PCHAR	File,
		UINT	Line
	)

Definition at line 425 of file mockbuffer.c.

                                               {
    PNDIS_PACKET Packet;
    PNDIS_BUFFER Buffer;
    NDIS_STATUS Status;
    PCHAR NewData;
 
    NewData = exAllocatePool( NonPagedPool, Len );
    if( !NewData ) return NDIS_STATUS_NOT_ACCEPTED; // XXX
 
    if( Data )
    RtlCopyMemory(NewData, Data, Len);
 
    NdisAllocatePacket( &Status, &Packet, GlobalPacketPool );
    if( Status != NDIS_STATUS_SUCCESS ) {
    exFreePool( NewData );
    return Status;
    }
    TrackWithTag(NDIS_PACKET_TAG, Packet, File, Line);
 
    NdisAllocateBuffer( &Status, &Buffer, GlobalBufferPool, NewData, Len );
    if( Status != NDIS_STATUS_SUCCESS ) {
    exFreePool( NewData );
    FreeNdisPacket( Packet );
    }
    TrackWithTag(NDIS_BUFFER_TAG, Buffer, File, Line);
 
    NdisChainBufferAtFront( Packet, Buffer );
    *NdisPacket = Packet;
 
    return NDIS_STATUS_SUCCESS;
}

CopyBufferChainToBuffer()

UINT CopyBufferChainToBuffer	(	PCHAR	DstData,
		PNDIS_BUFFER	SrcBuffer,
		UINT	SrcOffset,
		UINT	Length
	)

Definition at line 157 of file mockbuffer.c.

{
    UINT BytesCopied, BytesToCopy, SrcSize;
    PCHAR SrcData;
 
    TI_DbgPrint(DEBUG_PBUFFER, ("DstData 0x%X  SrcBuffer 0x%X  SrcOffset 0x%X  Length %d\n",DstData,SrcBuffer, SrcOffset, Length));
 
    /* Skip SrcOffset bytes in the source buffer chain */
    if (SkipToOffset(SrcBuffer, SrcOffset, &SrcData, &SrcSize) == -1)
        return 0;
 
    /* Start copying the data */
    BytesCopied = 0;
    for (;;) {
        BytesToCopy = MIN(SrcSize, Length);
 
        TI_DbgPrint(DEBUG_PBUFFER, ("Copying (%d) bytes from 0x%X to 0x%X\n", BytesToCopy, SrcData, DstData));
 
        RtlCopyMemory((PVOID)DstData, (PVOID)SrcData, BytesToCopy);
        BytesCopied += BytesToCopy;
        DstData      = (PCHAR)((ULONG_PTR)DstData + BytesToCopy);
 
        Length -= BytesToCopy;
        if (Length == 0)
            break;
 
        SrcSize -= BytesToCopy;
        if (SrcSize == 0) {
            /* No more bytes in source buffer. Proceed to
               the next buffer in the source buffer chain */
            NdisGetNextBuffer(SrcBuffer, &SrcBuffer);
            if (!SrcBuffer)
                break;
 
            XNdisQueryBuffer(SrcBuffer, (PVOID)&SrcData, &SrcSize);
        }
    }
 
    return BytesCopied;
}

Referenced by CopyPacketToBuffer().

CopyBufferToBufferChain()

UINT CopyBufferToBufferChain	(	PNDIS_BUFFER	DstBuffer,
		UINT	DstOffset,
		PCHAR	SrcData,
		UINT	Length
	)

Definition at line 100 of file mockbuffer.c.

{
    UINT BytesCopied, BytesToCopy, DstSize;
    PCHAR DstData;
 
    TI_DbgPrint(DEBUG_PBUFFER, ("DstBuffer (0x%X)  DstOffset (0x%X)  SrcData (0x%X)  Length (%d)\n", DstBuffer, DstOffset, SrcData, Length));
 
    /* Skip DstOffset bytes in the destination buffer chain */
    if (SkipToOffset(DstBuffer, DstOffset, &DstData, &DstSize) == -1)
        return 0;
 
    /* Start copying the data */
    BytesCopied = 0;
    for (;;) {
        BytesToCopy = MIN(DstSize, Length);
 
        RtlCopyMemory((PVOID)DstData, (PVOID)SrcData, BytesToCopy);
        BytesCopied += BytesToCopy;
        SrcData      = (PCHAR)((ULONG_PTR)SrcData + BytesToCopy);
 
        Length -= BytesToCopy;
        if (Length == 0)
            break;
 
        DstSize -= BytesToCopy;
        if (DstSize == 0) {
            /* No more bytes in destination buffer. Proceed to
               the next buffer in the destination buffer chain */
            NdisGetNextBuffer(DstBuffer, &DstBuffer);
            if (!DstBuffer)
                break;
 
            XNdisQueryBuffer(DstBuffer, (PVOID)&DstData, &DstSize);
        }
    }
 
    return BytesCopied;
}

CopyPacketToBuffer()

UINT CopyPacketToBuffer	(	PCHAR	DstData,
		PNDIS_PACKET	SrcPacket,
		UINT	SrcOffset,
		UINT	Length
	)

Definition at line 216 of file mockbuffer.c.

{
    PNDIS_BUFFER FirstBuffer;
    PVOID Address;
    UINT FirstLength;
    UINT TotalLength;
 
    TI_DbgPrint(DEBUG_PBUFFER, ("DstData (0x%X)  SrcPacket (0x%X)  SrcOffset (0x%X)  Length (%d)\n", DstData, SrcPacket, SrcOffset, Length));
 
    XNdisGetFirstBufferFromPacket(SrcPacket,
                  &FirstBuffer,
                  &Address,
                  &FirstLength,
                  &TotalLength);
 
    return CopyBufferChainToBuffer(DstData, FirstBuffer, SrcOffset, Length);
}

CopyPacketToBufferChain()

UINT CopyPacketToBufferChain	(	PNDIS_BUFFER	DstBuffer,
		UINT	DstOffset,
		PNDIS_PACKET	SrcPacket,
		UINT	SrcOffset,
		UINT	Length
	)

Definition at line 252 of file mockbuffer.c.

{
    PNDIS_BUFFER SrcBuffer;
    PCHAR DstData, SrcData;
    UINT DstSize, SrcSize;
    UINT Count, Total;
 
    TI_DbgPrint(DEBUG_PBUFFER, ("DstBuffer (0x%X)  DstOffset (0x%X)  SrcPacket (0x%X)  SrcOffset (0x%X)  Length (%d)\n", DstBuffer, DstOffset, SrcPacket, SrcOffset, Length));
 
    /* Skip DstOffset bytes in the destination buffer chain */
    XNdisQueryBuffer(DstBuffer, (PVOID)&DstData, &DstSize);
    if (SkipToOffset(DstBuffer, DstOffset, &DstData, &DstSize) == -1)
        return 0;
 
    /* Skip SrcOffset bytes in the source packet */
    XNdisGetFirstBufferFromPacket(SrcPacket, &SrcBuffer, (PVOID *)&SrcData, &SrcSize, &Total);
    if (SkipToOffset(SrcBuffer, SrcOffset, &SrcData, &SrcSize) == -1)
        return 0;
 
    /* Copy the data */
    for (Total = 0;;) {
        /* Find out how many bytes we can copy at one time */
        if (Length < SrcSize)
            Count = Length;
        else
            Count = SrcSize;
        if (DstSize < Count)
            Count = DstSize;
 
        RtlCopyMemory((PVOID)DstData, (PVOID)SrcData, Count);
 
        Total  += Count;
        Length -= Count;
        if (Length == 0)
            break;
 
        DstSize -= Count;
        if (DstSize == 0) {
            /* No more bytes in destination buffer. Proceed to
               the next buffer in the destination buffer chain */
            NdisGetNextBuffer(DstBuffer, &DstBuffer);
            if (!DstBuffer)
                break;
 
            XNdisQueryBuffer(DstBuffer, (PVOID)&DstData, &DstSize);
        }
 
        SrcSize -= Count;
        if (SrcSize == 0) {
            /* No more bytes in source buffer. Proceed to
               the next buffer in the source buffer chain */
            NdisGetNextBuffer(SrcBuffer, &SrcBuffer);
            if (!SrcBuffer)
                break;
 
            XNdisQueryBuffer(SrcBuffer, (PVOID)&SrcData, &SrcSize);
        }
    }
 
    return Total;
}

FreeNdisPacketX()

VOID FreeNdisPacketX	(	PNDIS_PACKET	Packet,
		PCHAR	File,
		UINT	Line
	)

Definition at line 460 of file mockbuffer.c.

{
    PNDIS_BUFFER Buffer, NextBuffer;
 
    TI_DbgPrint(DEBUG_PBUFFER, ("Packet (0x%X)\n", Packet));
 
    /* Free all the buffers in the packet first */
    NdisQueryPacket(Packet, NULL, NULL, &Buffer, NULL);
    for (; Buffer != NULL; Buffer = NextBuffer) {
        PVOID Data;
        UINT Length;
 
        NdisGetNextBuffer(Buffer, &NextBuffer);
        XNdisQueryBuffer(Buffer, &Data, &Length);
    UntrackFL(File,Line,Buffer);
        NdisFreeBuffer(Buffer);
        exFreePool(Data);
    }
 
    /* Finally free the NDIS packet descriptor */
    UntrackFL(File,Line,Packet);
    NdisFreePacket(Packet);
}

GetDataPtr()

void GetDataPtr	(	PNDIS_PACKET	Packet,
		UINT	Offset,
		PCHAR *	DataOut,
		PUINT	Size
	)

Definition at line 414 of file mockbuffer.c.

                      {
    PNDIS_BUFFER Buffer;
 
    NdisQueryPacket(Packet, NULL, NULL, &Buffer, NULL);
    if( !Buffer ) return;
    SkipToOffset( Buffer, Offset, DataOut, Size );
}

Referenced by AddGenericHeaderIPv4(), LANTransmit(), LANTransmitInternal(), LoopTransmit(), PrepareARPPacket(), ProtocolReceive(), SendFragments(), SendPacket(), and TCPSendDataCallback().

PrependPacket()

NDIS_STATUS PrependPacket	(	PNDIS_PACKET	Packet,
		PCHAR	Data,
		UINT	Length,
		BOOLEAN	Copy
	)

Definition at line 394 of file mockbuffer.c.

                              {
    PNDIS_BUFFER Buffer;
    NDIS_STATUS Status;
    PCHAR NewBuf;
 
    if( Copy ) {
    NewBuf = PoolAllocateBuffer( Length );
    if( !NewBuf ) return STATUS_NO_MEMORY;
    RtlCopyMemory( NewBuf, Data, Length );
    } else NewBuf = Data;
 
    NdisAllocateBuffer( &Status, &Buffer, GlobalBufferPool, Data, Length );
    if( Status != NDIS_STATUS_SUCCESS ) return Status;
 
    NdisChainBufferAtFront( Packet, Buffer );
 
    return STATUS_SUCCESS;
}

ResizePacket()

UINT ResizePacket	(	PNDIS_PACKET	Packet,
		UINT	Size
	)

Definition at line 369 of file mockbuffer.c.

{
    PNDIS_BUFFER NdisBuffer;
    UINT OldSize;
 
    NdisQueryPacket(Packet, NULL, NULL, &NdisBuffer, NULL);
 
    OldSize = NdisBuffer->ByteCount;
 
    if (Size != OldSize)
        NdisBuffer->ByteCount = Size;
 
    return OldSize;
}

SkipToOffset()

__inline INT SkipToOffset	(	PNDIS_BUFFER	Buffer,
		UINT	Offset,
		PCHAR *	Data,
		PUINT	Size
	)

Definition at line 58 of file mockbuffer.c.

{
    for (;;) {
 
        if (!Buffer)
            return -1;
 
        XNdisQueryBuffer(Buffer, (PVOID)Data, Size);
 
        if (Offset < *Size) {
            *Data = (PCHAR)((ULONG_PTR) *Data + Offset);
            *Size              -= Offset;
            break;
        }
 
        Offset -= *Size;
 
        NdisGetNextBuffer(Buffer, &Buffer);
    }
 
    return Offset;
}

Referenced by CopyBufferChainToBuffer(), CopyBufferToBufferChain(), CopyPacketToBufferChain(), and GetDataPtr().

XNdisGetFirstBufferFromPacket()

VOID XNdisGetFirstBufferFromPacket	(	PNDIS_PACKET	Packet,
		PNDIS_BUFFER *	FirstBuffer,
		PVOID *	FirstBufferVA,
		PUINT	FirstBufferLength,
		PUINT	TotalBufferLength
	)

Definition at line 13 of file mockbuffer.c.

{
    PNDIS_BUFFER _Buffer;
 
    _Buffer              = (Packet)->Private.Head;
    *(FirstBuffer)       = _Buffer;
    *(FirstBufferVA)     = MmGetMdlVirtualAddress(_Buffer);
    if (_Buffer != NULL) {
        *(FirstBufferLength) = MmGetMdlByteCount(_Buffer);
        _Buffer = _Buffer->Next;
    } else
        *(FirstBufferLength) = 0;
    *(TotalBufferLength) = *(FirstBufferLength);
    while (_Buffer != NULL) {
        *(TotalBufferLength) += MmGetMdlByteCount(_Buffer);
        _Buffer = _Buffer->Next;
    }
}

Referenced by CopyPacketToBuffer(), and CopyPacketToBufferChain().

XNdisQueryBuffer()

VOID XNdisQueryBuffer	(	PNDIS_BUFFER	Buffer,
		PVOID *	VirtualAddress,
		PUINT	Length
	)

Definition at line 39 of file mockbuffer.c.

{
    if (VirtualAddress != NULL)
        *(PVOID*)VirtualAddress = Buffer->MappedSystemVa;
 
    *Length = ((PMDL)Buffer)->ByteCount;
}

Referenced by CopyBufferChainToBuffer(), CopyBufferToBufferChain(), CopyPacketToBufferChain(), FreeNdisPacketX(), and SkipToOffset().