DPDK  16.04.0
l2fwd-ivshmem/guest/guest.c
/*-
* BSD LICENSE
*
* Copyright(c) 2010-2014 Intel Corporation. All rights reserved.
* All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions
* are met:
*
* * Redistributions of source code must retain the above copyright
* notice, this list of conditions and the following disclaimer.
* * Redistributions in binary form must reproduce the above copyright
* notice, this list of conditions and the following disclaimer in
* the documentation and/or other materials provided with the
* distribution.
* * Neither the name of Intel Corporation nor the names of its
* contributors may be used to endorse or promote products derived
* from this software without specific prior written permission.
*
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
* "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
* LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
* A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
* OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
* SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
* LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
* DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
* THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
* OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
*/
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <stdint.h>
#include <unistd.h>
#include <getopt.h>
#include <signal.h>
#include <sys/mman.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <sys/queue.h>
#include <sys/file.h>
#include <unistd.h>
#include <limits.h>
#include <errno.h>
#include <sys/ioctl.h>
#include <sys/time.h>
#include <rte_common.h>
#include <rte_eal_memconfig.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_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_ivshmem.h>
#include "../include/common.h"
#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 rx_mbufs[RTE_MAX_ETHPORTS];
struct vm_port_param * port_param[MAX_RX_QUEUE_PER_LCORE];
static struct lcore_queue_conf lcore_queue_conf[RTE_MAX_LCORE];
/* Print out statistics on packets dropped */
static void
print_stats(void)
{
uint64_t total_packets_dropped, total_packets_tx, total_packets_rx;
unsigned portid;
total_packets_dropped = 0;
total_packets_tx = 0;
total_packets_rx = 0;
const char clr[] = { 27, '[', '2', 'J', '\0' };
const char topLeft[] = { 27, '[', '1', ';', '1', 'H','\0' };
/* Clear screen and move to top left */
printf("%s%s", clr, topLeft);
printf("\nPort statistics ====================================");
for (portid = 0; portid < ctrl->nb_ports; portid++) {
/* skip ports that are not enabled */
printf("\nStatistics for port %u ------------------------------"
"\nPackets sent: %24"PRIu64
"\nPackets received: %20"PRIu64
"\nPackets dropped: %21"PRIu64,
portid,
ctrl->vm_ports[portid].stats.tx,
ctrl->vm_ports[portid].stats.rx,
ctrl->vm_ports[portid].stats.dropped);
total_packets_dropped += ctrl->vm_ports[portid].stats.dropped;
total_packets_tx += ctrl->vm_ports[portid].stats.tx;
total_packets_rx += ctrl->vm_ports[portid].stats.rx;
}
printf("\nAggregate statistics ==============================="
"\nTotal packets sent: %18"PRIu64
"\nTotal packets received: %14"PRIu64
"\nTotal packets dropped: %15"PRIu64,
total_packets_tx,
total_packets_rx,
total_packets_dropped);
printf("\n====================================================\n");
}
/* display usage */
static void
l2fwd_ivshmem_usage(const char *prgname)
{
printf("%s [EAL options] -- [-q NQ -T PERIOD]\n"
" -q NQ: number of queue (=ports) per lcore (default is 1)\n"
" -T PERIOD: statistics will be refreshed each PERIOD seconds (0 to disable, 10 default, 86400 maximum)\n",
prgname);
}
static unsigned int
l2fwd_ivshmem_parse_nqueue(const char *q_arg)
{
char *end = NULL;
unsigned long n;
/* parse hexadecimal string */
n = strtoul(q_arg, &end, 10);
if ((q_arg[0] == '\0') || (end == NULL) || (*end != '\0'))
return 0;
if (n == 0)
return 0;
if (n >= MAX_RX_QUEUE_PER_LCORE)
return 0;
return n;
}
static int
l2fwd_ivshmem_parse_timer_period(const char *q_arg)
{
char *end = NULL;
int n;
/* parse number string */
n = strtol(q_arg, &end, 10);
if ((q_arg[0] == '\0') || (end == NULL) || (*end != '\0'))
return -1;
if (n >= MAX_TIMER_PERIOD)
return -1;
return n;
}
/* Parse the argument given in the command line of the application */
static int
l2fwd_ivshmem_parse_args(int argc, char **argv)
{
int opt, ret;
char **argvopt;
int option_index;
char *prgname = argv[0];
static struct option lgopts[] = {
{NULL, 0, 0, 0}
};
argvopt = argv;
while ((opt = getopt_long(argc, argvopt, "q:p:T:",
lgopts, &option_index)) != EOF) {
switch (opt) {
/* nqueue */
case 'q':
l2fwd_ivshmem_rx_queue_per_lcore = l2fwd_ivshmem_parse_nqueue(optarg);
if (l2fwd_ivshmem_rx_queue_per_lcore == 0) {
printf("invalid queue number\n");
l2fwd_ivshmem_usage(prgname);
return -1;
}
break;
/* timer period */
case 'T':
timer_period = l2fwd_ivshmem_parse_timer_period(optarg) * 1000 * TIMER_MILLISECOND;
if (timer_period < 0) {
printf("invalid timer period\n");
l2fwd_ivshmem_usage(prgname);
return -1;
}
break;
/* long options */
case 0:
l2fwd_ivshmem_usage(prgname);
return -1;
default:
l2fwd_ivshmem_usage(prgname);
return -1;
}
}
if (optind >= 0)
argv[optind-1] = prgname;
ret = optind-1;
optind = 0; /* reset getopt lib */
return ret;
}
/*
* this loop is getting packets from RX rings of each port, and puts them
* into TX rings of destination ports.
*/
static void
fwd_loop(void)
{
struct rte_mbuf *pkts_burst[MAX_PKT_BURST];
struct rte_mbuf **m_table;
struct rte_mbuf *m;
struct rte_ring *rx, *tx;
unsigned lcore_id, len;
uint64_t prev_tsc, diff_tsc, cur_tsc, timer_tsc;
unsigned i, j, portid, nb_rx;
struct lcore_queue_conf *qconf;
struct ether_hdr *eth;
void *tmp;
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_IVSHMEM, "lcore %u has nothing to do\n", lcore_id);
return;
}
RTE_LOG(INFO, L2FWD_IVSHMEM, "entering main loop on lcore %u\n", lcore_id);
for (i = 0; i < qconf->n_rx_port; i++) {
portid = qconf->rx_port_list[i];
RTE_LOG(INFO, L2FWD_IVSHMEM, " -- lcoreid=%u portid=%u\n", lcore_id,
portid);
}
while (ctrl->state == STATE_FWD) {
cur_tsc = rte_rdtsc();
diff_tsc = cur_tsc - prev_tsc;
/*
* Read packet from RX queues and send it to TX queues
*/
for (i = 0; i < qconf->n_rx_port; i++) {
portid = qconf->rx_port_list[i];
len = qconf->rx_mbufs[portid].len;
rx = ctrl->vm_ports[portid].rx_ring;
tx = ctrl->vm_ports[portid].dst->tx_ring;
m_table = qconf->rx_mbufs[portid].m_table;
/* if we have something in the queue, try and transmit it down */
if (len != 0) {
/* if we succeed in sending the packets down, mark queue as free */
if (rte_ring_enqueue_bulk(tx, (void**) m_table, len) == 0) {
ctrl->vm_ports[portid].stats.tx += len;
qconf->rx_mbufs[portid].len = 0;
len = 0;
}
}
nb_rx = rte_ring_count(rx);
nb_rx = RTE_MIN(nb_rx, (unsigned) MAX_PKT_BURST);
if (nb_rx == 0)
continue;
/* if we can get packets into the m_table */
if (nb_rx < (RTE_DIM(qconf->rx_mbufs[portid].m_table) - len)) {
/* this situation cannot exist, so if we fail to dequeue, that
* means something went horribly wrong, hence the failure. */
if (rte_ring_dequeue_bulk(rx, (void**) pkts_burst, nb_rx) < 0) {
ctrl->state = STATE_FAIL;
return;
}
ctrl->vm_ports[portid].stats.rx += nb_rx;
/* put packets into the queue */
for (j = 0; j < nb_rx; j++) {
m = pkts_burst[j];
m_table[len + j] = m;
eth = rte_pktmbuf_mtod(m, struct ether_hdr *);
/* 02:00:00:00:00:xx */
tmp = &eth->d_addr.addr_bytes[0];
*((uint64_t *)tmp) = 0x000000000002 + ((uint64_t)portid << 40);
/* src addr */
ether_addr_copy(&ctrl->vm_ports[portid].dst->ethaddr,
&eth->s_addr);
}
qconf->rx_mbufs[portid].len += nb_rx;
}
}
/* 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;
}
}
static int
l2fwd_ivshmem_launch_one_lcore(__attribute__((unused)) void *dummy)
{
fwd_loop();
return 0;
}
int
main(int argc, char **argv)
{
struct lcore_queue_conf *qconf;
const struct rte_memzone * mz;
int ret;
uint8_t portid;
unsigned rx_lcore_id, lcore_id;
/* init EAL */
ret = rte_eal_init(argc, argv);
if (ret < 0)
rte_exit(EXIT_FAILURE, "Invalid EAL arguments\n");
argc -= ret;
argv += ret;
/* parse application arguments (after the EAL ones) */
ret = l2fwd_ivshmem_parse_args(argc, argv);
if (ret < 0)
rte_exit(EXIT_FAILURE, "Invalid l2fwd-ivshmem arguments\n");
/* find control structure */
mz = rte_memzone_lookup(CTRL_MZ_NAME);
if (mz == NULL)
rte_exit(EXIT_FAILURE, "Cannot find control memzone\n");
ctrl = (struct ivshmem_ctrl*) mz->addr;
/* lock the ctrl so that we don't have conflicts with anything else */
rte_spinlock_lock(&ctrl->lock);
if (ctrl->state == STATE_FWD)
rte_exit(EXIT_FAILURE, "Forwarding already started!\n");
rx_lcore_id = 0;
qconf = NULL;
/* Initialize the port/queue configuration of each logical core */
for (portid = 0; portid < ctrl->nb_ports; portid++) {
/* 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_ivshmem_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->port_param[qconf->n_rx_port] = &ctrl->vm_ports[portid];
qconf->n_rx_port++;
printf("Lcore %u: RX port %u\n", rx_lcore_id, (unsigned) portid);
}
sigsetup();
/* indicate that we are ready to forward */
ctrl->state = STATE_FWD;
/* unlock */
rte_spinlock_unlock(&ctrl->lock);
/* launch per-lcore init on every lcore */
rte_eal_mp_remote_launch(l2fwd_ivshmem_launch_one_lcore, NULL, CALL_MASTER);
if (rte_eal_wait_lcore(lcore_id) < 0)
return -1;
}
return 0;
}