context.c

#include "context.h"

#include <rte_ether.h>
#include <rte_ethdev.h>
#include <rte_malloc.h>

#include <netinet/ether.h>

#include <assert.h>

#include "config.h"

static int dpdk_port = 0;

static uint8_t local_mac[RTE_ETHER_ADDR_LEN];

static struct rte_mempool* mbuf_pool = NULL;
static struct inout_ring* ring_ins = NULL;

// static uint8_t dst_mac[RTE_ETHER_ADDR_LEN] = { 0x88, 0x66, 0x5a, 0x53, 0x3a, 0xd0 };

void init_server_context(void) {
    rte_eth_macaddr_get(get_dpdk_port(), (struct rte_ether_addr*)local_mac);
    printf("local mac: %s\n", ether_ntoa((struct ether_addr*)local_mac));

    mbuf_pool = rte_pktmbuf_pool_create(
        "mbuf pool", NUM_MBUFS, 0, 0, RTE_MBUF_DEFAULT_BUF_SIZE, rte_socket_id()
    );
    if (mbuf_pool == NULL) {
        rte_exit(EXIT_FAILURE, "Could not create mbuf pool\n");
    }
}

inline int get_dpdk_port(void) {
    assert(dpdk_port == 0);
    return dpdk_port;
}

inline uint8_t* get_local_mac(void) {
    assert(local_mac != NULL);
    return local_mac;
}

inline uint32_t get_local_ip(void) {
    assert(local_ip);
    return local_ip;
}

inline struct rte_mempool* get_server_mempool(void) {
    assert(mbuf_pool);
    return mbuf_pool;
}

struct inout_ring* get_server_ring(void) {
    if (ring_ins == NULL) {
        ring_ins = rte_malloc("in/out ring", sizeof(struct inout_ring), 0);
        memset(ring_ins, 0, sizeof(struct inout_ring));
    }

    if (ring_ins == NULL) {
        rte_exit(EXIT_FAILURE, "ring buffer init failed\n");
    }

    if (ring_ins->in == NULL) {
        ring_ins->in = rte_ring_create(
            "sys in ring", RING_SIZE, rte_socket_id(), RING_F_SP_ENQ | RING_F_SC_DEQ
        );
    }

    if (ring_ins->out == NULL) {
        ring_ins->out = rte_ring_create(
            "sys out ring", RING_SIZE, rte_socket_id(), RING_F_SP_ENQ | RING_F_SC_DEQ
        );
    }

    return ring_ins;
}