NTSTATUS KrnlHlprNDISPoolDataPopulate(_Inout_ NDIS_POOL_DATA* pNDISPoolData,
_In_opt_ UINT32 memoryTag) /* WFPSAMPLER_NDIS_POOL_TAG */
{
#if DBG
DbgPrintEx(DPFLTR_IHVNETWORK_ID,
DPFLTR_INFO_LEVEL,
" ---> KrnlHlprNDISPoolDataPopulate()\n");
#endif /// DBG
NT_ASSERT(pNDISPoolData);
NTSTATUS status = STATUS_SUCCESS;
NET_BUFFER_LIST_POOL_PARAMETERS nblPoolParameters = {0};
NET_BUFFER_POOL_PARAMETERS nbPoolParameters = {0};
pNDISPoolData->ndisHandle = NdisAllocateGenericObject(0,
memoryTag,
0);
if(pNDISPoolData->ndisHandle == 0)
{
status = STATUS_INVALID_HANDLE;
DbgPrintEx(DPFLTR_IHVNETWORK_ID,
DPFLTR_ERROR_LEVEL,
" !!!! KrnlHlprNDISPoolDataPopulate : NdisAllocateGenericObject() [status: %#x]\n",
status);
HLPR_BAIL;
}
nblPoolParameters.Header.Type = NDIS_OBJECT_TYPE_DEFAULT;
nblPoolParameters.Header.Revision = NET_BUFFER_LIST_POOL_PARAMETERS_REVISION_1;
nblPoolParameters.Header.Size = NDIS_SIZEOF_NET_BUFFER_LIST_POOL_PARAMETERS_REVISION_1;
nblPoolParameters.fAllocateNetBuffer = TRUE;
nblPoolParameters.DataSize = 0;
nblPoolParameters.PoolTag = memoryTag;
pNDISPoolData->nblPoolHandle = NdisAllocateNetBufferListPool(pNDISPoolData->ndisHandle,
&nblPoolParameters);
if(pNDISPoolData->nblPoolHandle == 0)
{
status = STATUS_INVALID_HANDLE;
DbgPrintEx(DPFLTR_IHVNETWORK_ID,
DPFLTR_ERROR_LEVEL,
" !!!! KrnlHlprNDISPoolDataPopulate : NdisAllocateNetBufferListPool() [status: %#x]\n",
status);
HLPR_BAIL;
}
nbPoolParameters.Header.Type = NDIS_OBJECT_TYPE_DEFAULT;
nbPoolParameters.Header.Revision = NET_BUFFER_POOL_PARAMETERS_REVISION_1;
nbPoolParameters.Header.Size = NDIS_SIZEOF_NET_BUFFER_POOL_PARAMETERS_REVISION_1;
nbPoolParameters.PoolTag = memoryTag;
nbPoolParameters.DataSize = 0;
pNDISPoolData->nbPoolHandle = NdisAllocateNetBufferPool(pNDISPoolData->ndisHandle,
&nbPoolParameters);
if(pNDISPoolData->nbPoolHandle == 0)
{
status = STATUS_INVALID_HANDLE;
DbgPrintEx(DPFLTR_IHVNETWORK_ID,
DPFLTR_ERROR_LEVEL,
" !!!! KrnlHlprNDISPoolDataPopulate : NdisAllocateNetBufferPool() [status: %#x]\n",
status);
HLPR_BAIL;
}
HLPR_BAIL_LABEL:
if(status != STATUS_SUCCESS)
KrnlHlprNDISPoolDataPurge(pNDISPoolData);
#if DBG
DbgPrintEx(DPFLTR_IHVNETWORK_ID,
DPFLTR_INFO_LEVEL,
" <--- KrnlHlprNDISPoolDataPopulate()\n");
#endif /// DBG
return status;
}
/*
* --------------------------------------------------------------------------
* OvsInitBufferPool --
*
* Allocate NBL and NB pool
*
* XXX: more optimization may be done for buffer management include local cache
* of NBL, NB, data, context, MDL.
* --------------------------------------------------------------------------
*/
NDIS_STATUS
OvsInitBufferPool(PVOID ovsContext)
{
POVS_NBL_POOL ovsPool;
POVS_SWITCH_CONTEXT context = (POVS_SWITCH_CONTEXT)ovsContext;
NET_BUFFER_LIST_POOL_PARAMETERS nblParam;
NET_BUFFER_POOL_PARAMETERS nbParam;
C_ASSERT(MEMORY_ALLOCATION_ALIGNMENT >= 8);
OVS_LOG_TRACE("Enter: context: %p", context);
ovsPool = &context->ovsPool;
RtlZeroMemory(ovsPool, sizeof (OVS_NBL_POOL));
ovsPool->ndisHandle = context->NdisFilterHandle;
ovsPool->ndisContext = context->NdisSwitchContext;
/*
* fix size NBL pool includes
* NBL + NB + MDL + DATA + Context
* This is mainly used for Packet execute or slow path when copy is
* required and size is less than OVS_DEFAULT_DATA_SIZE. We expect
* Most of packet from user space will use this Pool. (This is
* true for all bfd and cfm packet.
*/
RtlZeroMemory(&nblParam, sizeof (nblParam));
OVS_INIT_OBJECT_HEADER(&nblParam.Header,
NDIS_OBJECT_TYPE_DEFAULT,
NET_BUFFER_LIST_POOL_PARAMETERS_REVISION_1,
NDIS_SIZEOF_NET_BUFFER_LIST_POOL_PARAMETERS_REVISION_1);
nblParam.ContextSize = OVS_DEFAULT_NBL_CONTEXT_SIZE;
nblParam.PoolTag = OVS_FIX_SIZE_NBL_POOL_TAG;
nblParam.fAllocateNetBuffer = TRUE;
nblParam.DataSize = OVS_DEFAULT_DATA_SIZE + OVS_DEFAULT_HEADROOM_SIZE;
ovsPool->fixSizePool =
NdisAllocateNetBufferListPool(context->NdisSwitchContext, &nblParam);
if (ovsPool->fixSizePool == NULL) {
goto pool_cleanup;
}
/*
* Zero Size NBL Pool includes
* NBL + NB + Context
* This is mainly for packet with large data Size, in this case MDL and
* Data will be allocate separately.
*/
RtlZeroMemory(&nblParam, sizeof (nblParam));
OVS_INIT_OBJECT_HEADER(&nblParam.Header,
NDIS_OBJECT_TYPE_DEFAULT,
NET_BUFFER_LIST_POOL_PARAMETERS_REVISION_1,
NDIS_SIZEOF_NET_BUFFER_LIST_POOL_PARAMETERS_REVISION_1);
nblParam.ContextSize = OVS_DEFAULT_NBL_CONTEXT_SIZE;
nblParam.PoolTag = OVS_VARIABLE_SIZE_NBL_POOL_TAG;
nblParam.fAllocateNetBuffer = TRUE;
nblParam.DataSize = 0;
ovsPool->zeroSizePool =
NdisAllocateNetBufferListPool(context->NdisSwitchContext, &nblParam);
if (ovsPool->zeroSizePool == NULL) {
goto pool_cleanup;
}
/*
* NBL only pool just includes
* NBL (+ context)
* This is mainly used for clone and partial copy
*/
RtlZeroMemory(&nblParam, sizeof (nblParam));
OVS_INIT_OBJECT_HEADER(&nblParam.Header,
NDIS_OBJECT_TYPE_DEFAULT,
NET_BUFFER_LIST_POOL_PARAMETERS_REVISION_1,
NDIS_SIZEOF_NET_BUFFER_LIST_POOL_PARAMETERS_REVISION_1);
nblParam.ContextSize = OVS_DEFAULT_NBL_CONTEXT_SIZE;
nblParam.PoolTag = OVS_NBL_ONLY_POOL_TAG;
nblParam.fAllocateNetBuffer = FALSE;
nblParam.DataSize = 0;
ovsPool->nblOnlyPool =
NdisAllocateNetBufferListPool(context->NdisSwitchContext, &nblParam);
if (ovsPool->nblOnlyPool == NULL) {
goto pool_cleanup;
}
/* nb Pool
* NB only pool, used for copy
*/
OVS_INIT_OBJECT_HEADER(&nbParam.Header,
NDIS_OBJECT_TYPE_DEFAULT,
NET_BUFFER_POOL_PARAMETERS_REVISION_1,
NDIS_SIZEOF_NET_BUFFER_POOL_PARAMETERS_REVISION_1);
nbParam.PoolTag = OVS_NET_BUFFER_POOL_TAG;
nbParam.DataSize = 0;
ovsPool->nbPool =
NdisAllocateNetBufferPool(context->NdisSwitchContext, &nbParam);
if (ovsPool->nbPool == NULL) {
goto pool_cleanup;
}
OVS_LOG_TRACE("Exit: fixSizePool: %p zeroSizePool: %p nblOnlyPool: %p"
"nbPool: %p", ovsPool->fixSizePool, ovsPool->zeroSizePool,
ovsPool->nblOnlyPool, ovsPool->nbPool);
return NDIS_STATUS_SUCCESS;
pool_cleanup:
OvsCleanupBufferPool(context);
OVS_LOG_TRACE("Exit: Fail to initialize ovs buffer pool");
return NDIS_STATUS_RESOURCES;
}
