main.c

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <stdint.h>
#include <inttypes.h>
#include <sys/types.h>
#include <sys/queue.h>
#include <netinet/in.h>
#include <setjmp.h>
#include <stdarg.h>
#include <ctype.h>
#include <errno.h>
#include <getopt.h>
#include <assert.h>

#include <rte_common.h>
#include <rte_log.h>
#include <rte_memory.h>
#include <rte_memcpy.h>
#include <rte_memzone.h>
#include <rte_eal.h>
#include <rte_per_lcore.h>
#include <rte_launch.h>
#include <rte_atomic.h>
#include <rte_cycles.h>
#include <rte_prefetch.h>
#include <rte_lcore.h>
#include <rte_per_lcore.h>
#include <rte_branch_prediction.h>
#include <rte_interrupts.h>
#include <rte_pci.h>
#include <rte_random.h>
#include <rte_debug.h>
#include <rte_ether.h>
#include <rte_ethdev.h>
#include <rte_ring.h>
#include <rte_mempool.h>
#include <rte_mbuf.h>
#include <rte_ip.h>

#include "config.h"

#define NB_MBUF   8192

#define MAX_PKT_BURST 32
#define BURST_TX_DRAIN_US 100 /* TX drain every ~100us */

static struct rte_mempool * l2fwd_pktmbuf_pool = NULL;
/* ethernet addresses of ports */
static struct ether_addr l2fwd_ports_eth_addr[RTE_MAX_ETHPORTS];

/* mask of enabled ports */
static uint32_t l2fwd_enabled_port_mask = 0;

/* list of enabled ports */
static uint32_t l2fwd_dst_ports[RTE_MAX_ETHPORTS];

static unsigned int l2fwd_rx_queue_per_lcore = 1;

struct mbuf_table {
    unsigned len;
    struct rte_mbuf *m_table[MAX_PKT_BURST];
};

#define MAX_RX_QUEUE_PER_LCORE 16
#define MAX_TX_QUEUE_PER_PORT 16
struct lcore_queue_conf {
    unsigned n_rx_port;
    unsigned rx_port_list[MAX_RX_QUEUE_PER_LCORE];
    struct mbuf_table tx_mbufs[RTE_MAX_ETHPORTS];

} __rte_cache_aligned;
struct lcore_queue_conf lcore_queue_conf[RTE_MAX_LCORE];

static const struct rte_eth_conf port_conf = {
    .rxmode = {
    .split_hdr_size = 0,
    .header_split = 0, /**< Header Split disabled */
    .hw_ip_checksum = 0, /**< IP checksum offload disabled */
    .hw_vlan_filter = 0, /**< VLAN filtering disabled */
    .jumbo_frame = 0, /**< Jumbo Frame Support disabled */
    .hw_strip_crc = 0, /**< CRC stripped by hardware */
},
.txmode = {
    .mq_mode = ETH_MQ_TX_NONE,
},
};

/*
* Configurable number of RX/TX ring descriptors
*/
#define RTE_TEST_RX_DESC_DEFAULT 128
#define RTE_TEST_TX_DESC_DEFAULT 512
static uint16_t nb_rxd = RTE_TEST_RX_DESC_DEFAULT;
static uint16_t nb_txd = RTE_TEST_TX_DESC_DEFAULT;

/* Per-port statistics struct */
struct l2fwd_port_statistics {
    uint64_t tx;
    uint64_t rx;
    uint64_t dropped;
} __rte_cache_aligned;
struct l2fwd_port_statistics port_statistics[RTE_MAX_ETHPORTS];


int eth_init();
int eth_input(struct rte_mbuf *mbuf);

int arp_init();

static int
l2fwd_launch_one_lcore(__attribute__((unused)) void *arg)
{
	//unsigned lcore_id;
	//lcore_id = rte_lcore_id();
	//printf("hello from core %u\n", lcore_id);

    struct rte_mbuf *pkts_burst[MAX_PKT_BURST];
    struct rte_mbuf *m;
    unsigned lcore_id;
    uint64_t prev_tsc, diff_tsc, cur_tsc, timer_tsc;
    unsigned i, j, portid, nb_rx;
    struct lcore_queue_conf *qconf;
    const uint64_t drain_tsc = (rte_get_tsc_hz() + US_PER_S - 1) / US_PER_S * BURST_TX_DRAIN_US;

    prev_tsc = 0;
    timer_tsc = 0;

    lcore_id = rte_lcore_id();
    qconf = &lcore_queue_conf[lcore_id];

    if (qconf->n_rx_port == 0) {
        //RTE_LOG(INFO, L2FWD, "lcore %u has nothing to do\n", lcore_id);
        return;
    }

    printf("entering main loop on lcore %u\n", lcore_id);

    for (i = 0; i < qconf->n_rx_port; i++) {

        portid = qconf->rx_port_list[i];
        printf(" -- lcoreid=%u portid=%u\n", lcore_id, portid);
    }

    while (1) {
        cur_tsc = rte_rdtsc();

        /*
        * TX burst queue drain
        */
        diff_tsc = cur_tsc - prev_tsc;
        if (unlikely(diff_tsc > drain_tsc)) {

            for (portid = 0; portid < RTE_MAX_ETHPORTS; portid++) {
                if (qconf->tx_mbufs[portid].len == 0)
                    continue;
                //l2fwd_send_burst(&lcore_queue_conf[lcore_id],
                //    qconf->tx_mbufs[portid].len,
                //    (uint8_t)portid);
                qconf->tx_mbufs[portid].len = 0;
            }

            /* if timer is enabled */
            //if (timer_period > 0) {

            //    /* advance the timer */
            //    timer_tsc += diff_tsc;

            //    /* if timer has reached its timeout */
            //    if (unlikely(timer_tsc >= (uint64_t)timer_period)) {

            //        /* do this only on master core */
            //        if (lcore_id == rte_get_master_lcore()) {
            //            print_stats();
            //            /* reset the timer */
            //            timer_tsc = 0;
            //        }
            //    }
            //}

            prev_tsc = cur_tsc;
        }

        /*
        * Read packet from RX queues
        */
        for (i = 0; i < qconf->n_rx_port; i++) {

            portid = qconf->rx_port_list[i];
            nb_rx = rte_eth_rx_burst((uint8_t)portid, 0,
                pkts_burst, MAX_PKT_BURST);

            port_statistics[portid].rx += nb_rx;

            for (j = 0; j < nb_rx; j++) {
                m = pkts_burst[j];
                rte_prefetch0(rte_pktmbuf_mtod(m, void *));
                //l2fwd_simple_forward(m, portid);
                eth_input(m);
            }
        }
    }
    
    return 0;
}

/* Check the link status of all ports in up to 9s, and print them finally */
static void
check_all_ports_link_status(uint8_t port_num, uint32_t port_mask)
{
#define CHECK_INTERVAL 100 /* 100ms */
#define MAX_CHECK_TIME 90 /* 9s (90 * 100ms) in total */
    uint8_t portid, count, all_ports_up, print_flag = 0;
    struct rte_eth_link link;

    printf("\nChecking link status");
    fflush(stdout);
    for (count = 0; count <= MAX_CHECK_TIME; count++) {
        all_ports_up = 1;
        for (portid = 0; portid < port_num; portid++) {
            if ((port_mask & (1 << portid)) == 0)
                continue;
            memset(&link, 0, sizeof(link));
            rte_eth_link_get_nowait(portid, &link);
            /* print link status if flag set */
            if (print_flag == 1) {
                if (link.link_status)
                    printf("Port %d Link Up - speed %u "
                        "Mbps - %s\n", (uint8_t)portid,
                        (unsigned)link.link_speed,
                        (link.link_duplex == ETH_LINK_FULL_DUPLEX) ?
                        ("full-duplex") : ("half-duplex\n"));
                else
                    printf("Port %d Link Down\n",
                    (uint8_t)portid);
                continue;
            }
            /* clear all_ports_up flag if any link down */
            if (link.link_status == 0) {
                all_ports_up = 0;
                break;
            }
        }
        /* after finally printing all link status, get out */
        if (print_flag == 1)
            break;

        if (all_ports_up == 0) {
            printf(".");
            fflush(stdout);
            rte_delay_ms(CHECK_INTERVAL);
        }

        /* set the print_flag if all ports up or timeout */
        if (all_ports_up == 1 || count == (MAX_CHECK_TIME - 1)) {
            print_flag = 1;
            printf("done\n");
        }
    }
}

int
main(int argc, char **argv)
{
	int ret;
	//unsigned lcore_id;
    struct lcore_queue_conf *qconf;
    struct rte_eth_dev_info dev_info;
    uint8_t nb_ports;
    uint8_t nb_ports_available;
    uint8_t portid, last_port;
    unsigned lcore_id, rx_lcore_id;
    unsigned nb_ports_in_mask = 0;
	
    ret = rte_eal_init(argc, argv);
	if (ret < 0)
		rte_panic("Cannot init EAL\n");

    /* create the mbuf pool */
    l2fwd_pktmbuf_pool = rte_pktmbuf_pool_create("mbuf_pool", NB_MBUF, 32,
        0, RTE_MBUF_DEFAULT_BUF_SIZE, rte_socket_id());
    assert(l2fwd_pktmbuf_pool);
    ust_pktmbuf_pool = l2fwd_pktmbuf_pool;

    nb_ports = rte_eth_dev_count();
    assert(nb_ports > 0 && nb_ports <= RTE_MAX_ETHPORTS);

    portid = 0;
    rte_eth_dev_info_get(portid, &dev_info);

    rx_lcore_id = 0;
    qconf = NULL;

    /* get the lcore_id for this port */
    while (rte_lcore_is_enabled(rx_lcore_id) == 0 ||
        lcore_queue_conf[rx_lcore_id].n_rx_port ==
        l2fwd_rx_queue_per_lcore) {
        rx_lcore_id++;
        if (rx_lcore_id >= RTE_MAX_LCORE)
            rte_exit(EXIT_FAILURE, "Not enough cores\n");
    }

    if (qconf != &lcore_queue_conf[rx_lcore_id])
        /* Assigned a new logical core in the loop above. */
        qconf = &lcore_queue_conf[rx_lcore_id];

    qconf->rx_port_list[qconf->n_rx_port] = portid;
    qconf->n_rx_port++;
    printf("Lcore %u: RX port %u\n", rx_lcore_id, (unsigned)portid);

    nb_ports_available = nb_ports;

    /* init port */
    printf("Initializing port %u... ", (unsigned)portid);
    fflush(stdout);
    ret = rte_eth_dev_configure(portid, 1, 1, &port_conf);
    if (ret < 0)
        rte_exit(EXIT_FAILURE, "Cannot configure device: err=%d, port=%u\n",
            ret, (unsigned)portid);

    rte_eth_macaddr_get(portid, &l2fwd_ports_eth_addr[portid]);
    ust_eth_addr = l2fwd_ports_eth_addr[portid];

    /* init one RX queue */
    fflush(stdout);
    ret = rte_eth_rx_queue_setup(portid, 0, nb_rxd,
        rte_eth_dev_socket_id(portid),
        NULL, l2fwd_pktmbuf_pool);
    if (ret < 0)
        rte_exit(EXIT_FAILURE, "rte_eth_rx_queue_setup:err=%d, port=%u\n",
            ret, (unsigned)portid);

    /* init one TX queue on each port */
    fflush(stdout);
    ret = rte_eth_tx_queue_setup(portid, 0, nb_txd,
        rte_eth_dev_socket_id(portid),
        NULL);
    if (ret < 0)
        rte_exit(EXIT_FAILURE, "rte_eth_tx_queue_setup:err=%d, port=%u\n",
            ret, (unsigned)portid);

    /* Start device */
    ret = rte_eth_dev_start(portid);
    if (ret < 0)
        rte_exit(EXIT_FAILURE, "rte_eth_dev_start:err=%d, port=%u\n",
            ret, (unsigned)portid);

    printf("done: \n");

    rte_eth_promiscuous_disable(portid);

    printf("Port %u, MAC address: %02X:%02X:%02X:%02X:%02X:%02X\n\n",
        (unsigned)portid,
        l2fwd_ports_eth_addr[portid].addr_bytes[0],
        l2fwd_ports_eth_addr[portid].addr_bytes[1],
        l2fwd_ports_eth_addr[portid].addr_bytes[2],
        l2fwd_ports_eth_addr[portid].addr_bytes[3],
        l2fwd_ports_eth_addr[portid].addr_bytes[4],
        l2fwd_ports_eth_addr[portid].addr_bytes[5]);

    /* initialize port stats */
    //memset(&port_statistics, 0, sizeof(port_statistics));

    check_all_ports_link_status(nb_ports, 1);

    ust_ip_addr = IPv4(192, 168, 192, 22);
    ust_ip_mask = IPv4(255, 255, 255, 0);
    ust_ip_gw = IPv4(192, 168, 192, 1);

    /* reset l2fwd_dst_ports */
    //for (portid = 0; portid < RTE_MAX_ETHPORTS; portid++)
    //    l2fwd_dst_ports[portid] = 0;
    //last_port = 0;

    eth_init();
    arp_init();

	/* call lcore_hello() on every slave lcore */
	//RTE_LCORE_FOREACH_SLAVE(lcore_id) {
	//	rte_eal_remote_launch(lcore_hello, NULL, lcore_id);
	//}
    /* launch per-lcore init on every lcore */
    rte_eal_mp_remote_launch(l2fwd_launch_one_lcore, NULL, CALL_MASTER);
    RTE_LCORE_FOREACH_SLAVE(lcore_id) {
        if (rte_eal_wait_lcore(lcore_id) < 0)
            return -1;
    }

	/* call it on master lcore too */
	//lcore_hello(NULL);

	//rte_eal_mp_wait_lcore();
	return 0;
}