Hesham/wintun
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Optimize ActiveNBLCount management in TunSendNetBufferLists

Browse Source 
Signed-off-by: Simon Rozman <simon@rozman.si>
This commit is contained in:
Simon Rozman 2019-03-22 09:17:14 +01:00 committed by Jason A. Donenfeld
parent a9afa3a692
commit 999cfaabcc
1 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
wintun.c
View File

@ -1071,7 +1071,7 @@ _Use_decl_annotations_
static void TunSendNetBufferLists(NDIS_HANDLE MiniportAdapterContext, NET_BUFFER_LIST *NetBufferLists, NDIS_PORT_NUMBER PortNumber, ULONG SendFlags) static void TunSendNetBufferLists(NDIS_HANDLE MiniportAdapterContext, NET_BUFFER_LIST *NetBufferLists, NDIS_PORT_NUMBER PortNumber, ULONG SendFlags)
{ {
TUN_CTX *ctx = (TUN_CTX *)MiniportAdapterContext; TUN_CTX *ctx = (TUN_CTX *)MiniportAdapterContext;
ULONG nbl_count; ULONG nbl_count = 0;
InterlockedIncrement64(&ctx->ActiveNBLCount); InterlockedIncrement64(&ctx->ActiveNBLCount);
@ -1090,7 +1090,7 @@ static void TunSendNetBufferLists(NDIS_HANDLE MiniportAdapterContext, NET_BUFFER
ctx->PacketQueue.Head = NetBufferLists; ctx->PacketQueue.Head = NetBufferLists;
ctx->PacketQueue.Buffer = NET_BUFFER_LIST_FIRST_NB(NetBufferLists); ctx->PacketQueue.Buffer = NET_BUFFER_LIST_FIRST_NB(NetBufferLists);
} }
for (nbl_count = 0; NetBufferLists; NetBufferLists = NET_BUFFER_LIST_NEXT_NBL(NetBufferLists), nbl_count++) for (; NetBufferLists; NetBufferLists = NET_BUFFER_LIST_NEXT_NBL(NetBufferLists), nbl_count++)
ctx->PacketQueue.Tail = NetBufferLists; ctx->PacketQueue.Tail = NetBufferLists;
ctx->PacketQueue.Count += nbl_count; ctx->PacketQueue.Count += nbl_count;
NdisReleaseSpinLock(&ctx->PacketQueue.Lock); NdisReleaseSpinLock(&ctx->PacketQueue.Lock);
@ -1110,13 +1110,12 @@ static void TunSendNetBufferLists(NDIS_HANDLE MiniportAdapterContext, NET_BUFFER
NetBufferLists = nbl; NetBufferLists = nbl;
} }
NdisReleaseSpinLock(&ctx->PacketQueue.Lock); NdisReleaseSpinLock(&ctx->PacketQueue.Lock);
InterlockedSubtract64(&ctx->ActiveNBLCount, nbl_count);
cleanup_NdisMSendNetBufferListsComplete: cleanup_NdisMSendNetBufferListsComplete:
if (NetBufferLists) if (NetBufferLists)
NdisMSendNetBufferListsComplete(ctx->MiniportAdapterHandle, NetBufferLists, SendFlags & NDIS_SEND_FLAGS_DISPATCH_LEVEL ? NDIS_SEND_COMPLETE_FLAGS_DISPATCH_LEVEL : 0); NdisMSendNetBufferListsComplete(ctx->MiniportAdapterHandle, NetBufferLists, SendFlags & NDIS_SEND_FLAGS_DISPATCH_LEVEL ? NDIS_SEND_COMPLETE_FLAGS_DISPATCH_LEVEL : 0);
TunCompletePausing(ctx, 1); TunCompletePausing(ctx, 1i64 + nbl_count);
} }
DRIVER_INITIALIZE DriverEntry; DRIVER_INITIALIZE DriverEntry;