examples/ip_pipeline/mempool.c

/* SPDX-License-Identifier: BSD-3-Clause
 * Copyright(c) 2010-2018 Intel Corporation
 */
 
#include <stdlib.h>
#include <string.h>
 
#include <rte_mbuf.h>
#include <rte_string_fns.h>
 
#include "mempool.h"
 
#define BUFFER_SIZE_MIN        (sizeof(struct rte_mbuf) + RTE_PKTMBUF_HEADROOM)
 
static struct mempool_list mempool_list;
 
int
mempool_init(void)
{
    TAILQ_INIT(&mempool_list);
 
    return 0;
}
 
struct mempool *
mempool_find(const char *name)
{
    struct mempool *mempool;
 
    if (name == NULL)
        return NULL;
 
    TAILQ_FOREACH(mempool, &mempool_list, node)
        if (strcmp(mempool->name, name) == 0)
            return mempool;
 
    return NULL;
}
 
struct mempool *
mempool_create(const char *name, struct mempool_params *params)
{
    struct mempool *mempool;
    struct rte_mempool *m;
 
    /* Check input params */
    if ((name == NULL) ||
        mempool_find(name) ||
        (params == NULL) ||
        (params->buffer_size < BUFFER_SIZE_MIN) ||
        (params->pool_size == 0))
        return NULL;
 
    /* Resource create */
    m = rte_pktmbuf_pool_create(
        name,
        params->pool_size,
        params->cache_size,
        0,
        params->buffer_size - sizeof(struct rte_mbuf),
        params->cpu_id);
 
    if (m == NULL)
        return NULL;
 
    /* Node allocation */
    mempool = calloc(1, sizeof(struct mempool));
    if (mempool == NULL) {
        rte_mempool_free(m);
        return NULL;
    }
 
    /* Node fill in */
    strlcpy(mempool->name, name, sizeof(mempool->name));
    mempool->m = m;
    mempool->buffer_size = params->buffer_size;
 
    /* Node add to list */
    TAILQ_INSERT_TAIL(&mempool_list, mempool, node);
 
    return mempool;
}