#include <stdint.h>
#include <getopt.h>
#include <signal.h>
#include <stdbool.h>
#include <unistd.h>
#include <rte_ioat_rawdev.h>
#define RTE_LOGTYPE_IOAT RTE_LOGTYPE_USER1
#define MAX_PKT_BURST 32
#define MEMPOOL_CACHE_SIZE 512
#define MIN_POOL_SIZE 65536U
#define CMD_LINE_OPT_MAC_UPDATING "mac-updating"
#define CMD_LINE_OPT_NO_MAC_UPDATING "no-mac-updating"
#define CMD_LINE_OPT_PORTMASK "portmask"
#define CMD_LINE_OPT_NB_QUEUE "nb-queue"
#define CMD_LINE_OPT_COPY_TYPE "copy-type"
#define CMD_LINE_OPT_RING_SIZE "ring-size"
#define RX_DEFAULT_RINGSIZE 1024
#define TX_DEFAULT_RINGSIZE 1024
#define MAX_RX_QUEUES_COUNT 8
struct rxtx_port_config {
uint16_t rxtx_port;
uint16_t nb_queues;
uint16_t ioat_ids[MAX_RX_QUEUES_COUNT];
};
struct rxtx_transmission_config {
struct rxtx_port_config ports[RTE_MAX_ETHPORTS];
uint16_t nb_ports;
uint16_t nb_lcores;
};
struct ioat_port_statistics {
uint64_t rx[RTE_MAX_ETHPORTS];
uint64_t tx[RTE_MAX_ETHPORTS];
uint64_t tx_dropped[RTE_MAX_ETHPORTS];
uint64_t copy_dropped[RTE_MAX_ETHPORTS];
};
struct ioat_port_statistics port_statistics;
struct total_statistics {
uint64_t total_packets_dropped;
uint64_t total_packets_tx;
uint64_t total_packets_rx;
uint64_t total_successful_enqueues;
uint64_t total_failed_enqueues;
};
typedef enum copy_mode_t {
#define COPY_MODE_SW "sw"
COPY_MODE_SW_NUM,
#define COPY_MODE_IOAT "hw"
COPY_MODE_IOAT_NUM,
COPY_MODE_INVALID_NUM,
COPY_MODE_SIZE_NUM = COPY_MODE_INVALID_NUM
} copy_mode_t;
static uint32_t ioat_enabled_port_mask;
static uint16_t nb_queues = 1;
static int mac_updating = 1;
static copy_mode_t copy_mode = COPY_MODE_IOAT_NUM;
static unsigned short ring_size = 2048;
struct rxtx_transmission_config cfg;
static uint16_t nb_rxd = RX_DEFAULT_RINGSIZE;
static uint16_t nb_txd = TX_DEFAULT_RINGSIZE;
static volatile bool force_quit;
static void
print_port_stats(uint16_t port_id)
{
printf("\nStatistics for port %u ------------------------------"
"\nPackets sent: %34"PRIu64
"\nPackets received: %30"PRIu64
"\nPackets dropped on tx: %25"PRIu64
"\nPackets dropped on copy: %23"PRIu64,
port_id,
port_statistics.tx[port_id],
port_statistics.rx[port_id],
port_statistics.tx_dropped[port_id],
port_statistics.copy_dropped[port_id]);
}
static void
print_rawdev_stats(uint32_t dev_id, uint64_t *xstats,
unsigned int *ids_xstats, uint16_t nb_xstats,
{
uint16_t i;
printf("\nIOAT channel %u", dev_id);
for (i = 0; i < nb_xstats; i++)
printf("\n\t %s: %*"PRIu64,
names_xstats[ids_xstats[i]].
name,
(int)(37 - strlen(names_xstats[ids_xstats[i]].name)),
xstats[i]);
}
static void
print_total_stats(struct total_statistics *ts)
{
printf("\nAggregate statistics ==============================="
"\nTotal packets Tx: %24"PRIu64" [pps]"
"\nTotal packets Rx: %24"PRIu64" [pps]"
"\nTotal packets dropped: %19"PRIu64" [pps]",
ts->total_packets_tx,
ts->total_packets_rx,
ts->total_packets_dropped);
if (copy_mode == COPY_MODE_IOAT_NUM) {
printf("\nTotal IOAT successful enqueues: %8"PRIu64" [enq/s]"
"\nTotal IOAT failed enqueues: %12"PRIu64" [enq/s]",
ts->total_successful_enqueues,
ts->total_failed_enqueues);
}
printf("\n====================================================\n");
}
static void
print_stats(char *prgname)
{
struct total_statistics ts, delta_ts;
uint32_t i, port_id, dev_id;
uint64_t *xstats;
unsigned int *ids_xstats, nb_xstats;
char status_string[255];
int status_strlen;
int ret;
const char clr[] = { 27, '[', '2', 'J', '\0' };
const char topLeft[] = { 27, '[', '1', ';', '1', 'H', '\0' };
status_strlen = snprintf(status_string, sizeof(status_string),
"%s, ", prgname);
status_strlen += snprintf(status_string + status_strlen,
sizeof(status_string) - status_strlen,
"Worker Threads = %d, ",
status_strlen += snprintf(status_string + status_strlen,
sizeof(status_string) - status_strlen,
"Copy Mode = %s,\n", copy_mode == COPY_MODE_SW_NUM ?
COPY_MODE_SW : COPY_MODE_IOAT);
status_strlen += snprintf(status_string + status_strlen,
sizeof(status_string) - status_strlen,
"Updating MAC = %s, ", mac_updating ?
"enabled" : "disabled");
status_strlen += snprintf(status_string + status_strlen,
sizeof(status_string) - status_strlen,
"Rx Queues = %d, ", nb_queues);
status_strlen += snprintf(status_string + status_strlen,
sizeof(status_string) - status_strlen,
"Ring Size = %d", ring_size);
cfg.ports[0].ioat_ids[0], NULL, 0);
if (ret < 0)
return;
nb_xstats = (unsigned int)ret;
names_xstats = malloc(sizeof(*names_xstats) * nb_xstats);
if (names_xstats == NULL) {
"Error allocating xstat names memory\n");
}
names_xstats, nb_xstats);
ids_xstats = malloc(sizeof(*ids_xstats) * 2);
if (ids_xstats == NULL) {
"Error allocating xstat ids_xstats memory\n");
}
xstats = malloc(sizeof(*xstats) * 2);
if (xstats == NULL) {
"Error allocating xstat memory\n");
}
ids_xstats[0] = ids_xstats[1] = nb_xstats;
for (i = 0; i < nb_xstats; i++) {
if (!strcmp(names_xstats[i].name, "failed_enqueues"))
ids_xstats[0] = i;
else if (!strcmp(names_xstats[i].name, "successful_enqueues"))
ids_xstats[1] = i;
if (ids_xstats[0] < nb_xstats && ids_xstats[1] < nb_xstats)
break;
}
if (ids_xstats[0] == nb_xstats || ids_xstats[1] == nb_xstats) {
"Error getting failed/successful enqueues stats index\n");
}
memset(&ts, 0, sizeof(struct total_statistics));
while (!force_quit) {
sleep(1);
printf("%s%s", clr, topLeft);
memset(&delta_ts, 0, sizeof(struct total_statistics));
printf("%s\n", status_string);
for (i = 0; i < cfg.nb_ports; i++) {
port_id = cfg.ports[i].rxtx_port;
print_port_stats(port_id);
delta_ts.total_packets_dropped +=
port_statistics.tx_dropped[port_id]
+ port_statistics.copy_dropped[port_id];
delta_ts.total_packets_tx +=
port_statistics.tx[port_id];
delta_ts.total_packets_rx +=
port_statistics.rx[port_id];
if (copy_mode == COPY_MODE_IOAT_NUM) {
uint32_t j;
for (j = 0; j < cfg.ports[i].nb_queues; j++) {
dev_id = cfg.ports[i].ioat_ids[j];
ids_xstats, xstats, 2);
print_rawdev_stats(dev_id, xstats,
ids_xstats, 2, names_xstats);
delta_ts.total_failed_enqueues +=
xstats[ids_xstats[0]];
delta_ts.total_successful_enqueues +=
xstats[ids_xstats[1]];
}
}
}
delta_ts.total_packets_tx -= ts.total_packets_tx;
delta_ts.total_packets_rx -= ts.total_packets_rx;
delta_ts.total_packets_dropped -= ts.total_packets_dropped;
delta_ts.total_failed_enqueues -= ts.total_failed_enqueues;
delta_ts.total_successful_enqueues -=
ts.total_successful_enqueues;
printf("\n");
print_total_stats(&delta_ts);
fflush(stdout);
ts.total_packets_tx += delta_ts.total_packets_tx;
ts.total_packets_rx += delta_ts.total_packets_rx;
ts.total_packets_dropped += delta_ts.total_packets_dropped;
ts.total_failed_enqueues += delta_ts.total_failed_enqueues;
ts.total_successful_enqueues +=
delta_ts.total_successful_enqueues;
}
free(names_xstats);
free(xstats);
free(ids_xstats);
}
static void
update_mac_addrs(
struct rte_mbuf *m, uint32_t dest_portid)
{
void *tmp;
*((uint64_t *)tmp) = 0x000000000002 + ((uint64_t)dest_portid << 40);
}
static inline void
{
&src->rearm_data,
}
static uint32_t
ioat_enqueue_packets(
struct rte_mbuf **pkts,
uint32_t nb_rx, uint16_t dev_id)
{
int ret;
uint32_t i;
struct rte_mbuf *pkts_copy[MAX_PKT_BURST];
&pkts[0]->rearm_data);
(void *)pkts_copy, nb_rx);
rte_exit(EXIT_FAILURE,
"Unable to allocate memory.\n");
for (i = 0; i < nb_rx; i++) {
ret = rte_ioat_enqueue_copy(dev_id,
pkts[i]->buf_iova
- addr_offset,
pkts_copy[i]->buf_iova
- addr_offset,
+ addr_offset,
(uintptr_t)pkts[i],
(uintptr_t)pkts_copy[i],
0 );
if (ret != 1)
break;
}
ret = i;
nb_rx - i);
return ret;
}
static void
ioat_rx_port(struct rxtx_port_config *rx_config)
{
uint32_t nb_rx, nb_enq, i, j;
struct rte_mbuf *pkts_burst[MAX_PKT_BURST];
for (i = 0; i < rx_config->nb_queues; i++) {
pkts_burst, MAX_PKT_BURST);
if (nb_rx == 0)
continue;
port_statistics.rx[rx_config->rxtx_port] += nb_rx;
if (copy_mode == COPY_MODE_IOAT_NUM) {
nb_enq = ioat_enqueue_packets(pkts_burst,
nb_rx, rx_config->ioat_ids[i]);
if (nb_enq > 0)
rte_ioat_do_copies(rx_config->ioat_ids[i]);
} else {
int ret;
struct rte_mbuf *pkts_burst_copy[MAX_PKT_BURST];
(void *)pkts_burst_copy, nb_rx);
"Unable to allocate memory.\n");
for (j = 0; j < nb_rx; j++)
pktmbuf_sw_copy(pkts_burst[j],
pkts_burst_copy[j]);
(void *)pkts_burst, nb_rx);
rx_config->rx_to_tx_ring,
(void *)pkts_burst_copy, nb_rx, NULL);
(void *)&pkts_burst_copy[nb_enq],
nb_rx - nb_enq);
}
port_statistics.copy_dropped[rx_config->rxtx_port] +=
(nb_rx - nb_enq);
}
}
static void
ioat_tx_port(struct rxtx_port_config *tx_config)
{
uint32_t i, j, nb_dq = 0;
struct rte_mbuf *mbufs_src[MAX_PKT_BURST];
struct rte_mbuf *mbufs_dst[MAX_PKT_BURST];
for (i = 0; i < tx_config->nb_queues; i++) {
if (copy_mode == COPY_MODE_IOAT_NUM) {
nb_dq = rte_ioat_completed_copies(
tx_config->ioat_ids[i], MAX_PKT_BURST,
(void *)mbufs_src, (void *)mbufs_dst);
} else {
tx_config->rx_to_tx_ring, (void *)mbufs_dst,
MAX_PKT_BURST, NULL);
}
if ((int32_t) nb_dq <= 0)
return;
if (copy_mode == COPY_MODE_IOAT_NUM)
(void *)mbufs_src, nb_dq);
if (mac_updating) {
for (j = 0; j < nb_dq; j++)
update_mac_addrs(mbufs_dst[j],
tx_config->rxtx_port);
}
tx_config->rxtx_port, 0,
(void *)mbufs_dst, nb_dq);
port_statistics.tx[tx_config->rxtx_port] += nb_tx;
(void *)&mbufs_dst[nb_tx],
nb_dq - nb_tx);
}
}
static void
rx_main_loop(void)
{
uint16_t i;
uint16_t nb_ports = cfg.nb_ports;
RTE_LOG(INFO, IOAT,
"Entering main rx loop for copy on lcore %u\n",
while (!force_quit)
for (i = 0; i < nb_ports; i++)
ioat_rx_port(&cfg.ports[i]);
}
static void
tx_main_loop(void)
{
uint16_t i;
uint16_t nb_ports = cfg.nb_ports;
RTE_LOG(INFO, IOAT,
"Entering main tx loop for copy on lcore %u\n",
while (!force_quit)
for (i = 0; i < nb_ports; i++)
ioat_tx_port(&cfg.ports[i]);
}
static void
rxtx_main_loop(void)
{
uint16_t i;
uint16_t nb_ports = cfg.nb_ports;
RTE_LOG(INFO, IOAT,
"Entering main rx and tx loop for copy on" while (!force_quit)
for (i = 0; i < nb_ports; i++) {
ioat_rx_port(&cfg.ports[i]);
ioat_tx_port(&cfg.ports[i]);
}
}
static void start_forwarding_cores(void)
{
RTE_LOG(INFO, IOAT,
"Entering %s on lcore %u\n",
if (cfg.nb_lcores == 1) {
NULL, lcore_id);
} else if (cfg.nb_lcores > 1) {
NULL, lcore_id);
lcore_id);
}
}
static void
ioat_usage(const char *prgname)
{
printf("%s [EAL options] -- -p PORTMASK [-q NQ]\n"
" -p --portmask: hexadecimal bitmask of ports to configure\n"
" -q NQ: number of RX queues per port (default is 1)\n"
" --[no-]mac-updating: Enable or disable MAC addresses updating (enabled by default)\n"
" When enabled:\n"
" - The source MAC address is replaced by the TX port MAC address\n"
" - The destination MAC address is replaced by 02:00:00:00:00:TX_PORT_ID\n"
" -c --copy-type CT: type of copy: sw|hw\n"
" -s --ring-size RS: size of IOAT rawdev ring for hardware copy mode or rte_ring for software copy mode\n",
prgname);
}
static int
ioat_parse_portmask(const char *portmask)
{
char *end = NULL;
unsigned long pm;
pm = strtoul(portmask, &end, 16);
if ((portmask[0] == '\0') || (end == NULL) || (*end != '\0'))
return -1;
return pm;
}
static copy_mode_t
ioat_parse_copy_mode(const char *copy_mode)
{
if (strcmp(copy_mode, COPY_MODE_SW) == 0)
return COPY_MODE_SW_NUM;
else if (strcmp(copy_mode, COPY_MODE_IOAT) == 0)
return COPY_MODE_IOAT_NUM;
return COPY_MODE_INVALID_NUM;
}
static int
ioat_parse_args(int argc, char **argv, unsigned int nb_ports)
{
static const char short_options[] =
"p:"
"q:"
"c:"
"s:"
;
static const struct option lgopts[] = {
{CMD_LINE_OPT_MAC_UPDATING, no_argument, &mac_updating, 1},
{CMD_LINE_OPT_NO_MAC_UPDATING, no_argument, &mac_updating, 0},
{CMD_LINE_OPT_PORTMASK, required_argument, NULL, 'p'},
{CMD_LINE_OPT_NB_QUEUE, required_argument, NULL, 'q'},
{CMD_LINE_OPT_COPY_TYPE, required_argument, NULL, 'c'},
{CMD_LINE_OPT_RING_SIZE, required_argument, NULL, 's'},
{NULL, 0, 0, 0}
};
const unsigned int default_port_mask = (1 << nb_ports) - 1;
int opt, ret;
char **argvopt;
int option_index;
char *prgname = argv[0];
ioat_enabled_port_mask = default_port_mask;
argvopt = argv;
while ((opt = getopt_long(argc, argvopt, short_options,
lgopts, &option_index)) != EOF) {
switch (opt) {
case 'p':
ioat_enabled_port_mask = ioat_parse_portmask(optarg);
if (ioat_enabled_port_mask & ~default_port_mask ||
ioat_enabled_port_mask <= 0) {
printf("Invalid portmask, %s, suggest 0x%x\n",
optarg, default_port_mask);
ioat_usage(prgname);
return -1;
}
break;
case 'q':
nb_queues = atoi(optarg);
if (nb_queues == 0 || nb_queues > MAX_RX_QUEUES_COUNT) {
printf("Invalid RX queues number %s. Max %u\n",
optarg, MAX_RX_QUEUES_COUNT);
ioat_usage(prgname);
return -1;
}
break;
case 'c':
copy_mode = ioat_parse_copy_mode(optarg);
if (copy_mode == COPY_MODE_INVALID_NUM) {
printf("Invalid copy type. Use: sw, hw\n");
ioat_usage(prgname);
return -1;
}
break;
case 's':
ring_size = atoi(optarg);
if (ring_size == 0) {
printf("Invalid ring size, %s.\n", optarg);
ioat_usage(prgname);
return -1;
}
break;
case 0:
break;
default:
ioat_usage(prgname);
return -1;
}
}
printf("MAC updating %s\n", mac_updating ? "enabled" : "disabled");
if (optind >= 0)
argv[optind - 1] = prgname;
ret = optind - 1;
optind = 1;
return ret;
}
static int
check_link_status(uint32_t port_mask)
{
uint16_t portid;
printf("\nChecking link status\n");
if ((port_mask & (1 << portid)) == 0)
continue;
memset(&link, 0, sizeof(link));
if (ret < 0) {
printf("Port %u link get failed: err=%d\n",
portid, ret);
continue;
}
printf(
"Port %d Link Up. Speed %u Mbps - %s\n",
("full-duplex") : ("half-duplex"));
link_status = 1;
} else
printf("Port %d Link Down\n", portid);
}
}
static void
configure_rawdev_queue(uint32_t dev_id)
{
struct rte_ioat_rawdev_config dev_config = { .ring_size = ring_size };
struct rte_rawdev_info info = { .dev_private = &dev_config };
"Error with rte_rawdev_configure()\n");
}
"Error with rte_rawdev_start()\n");
}
}
static void
assign_rawdevs(void)
{
uint16_t nb_rawdev = 0, rdev_id = 0;
uint32_t i, j;
for (i = 0; i < cfg.nb_ports; i++) {
for (j = 0; j < cfg.ports[i].nb_queues; j++) {
struct rte_rawdev_info rdev_info = { 0 };
do {
goto end;
} while (rdev_info.driver_name == NULL ||
strcmp(rdev_info.driver_name,
IOAT_PMD_RAWDEV_NAME_STR) != 0);
cfg.ports[i].ioat_ids[j] = rdev_id - 1;
configure_rawdev_queue(cfg.ports[i].ioat_ids[j]);
++nb_rawdev;
}
}
end:
if (nb_rawdev < cfg.nb_ports * cfg.ports[0].nb_queues)
"Not enough IOAT rawdevs (%u) for all queues (%u).\n",
nb_rawdev, cfg.nb_ports * cfg.ports[0].nb_queues);
RTE_LOG(INFO, IOAT,
"Number of used rawdevs: %u.\n", nb_rawdev);
}
static void
assign_rings(void)
{
uint32_t i;
for (i = 0; i < cfg.nb_ports; i++) {
snprintf(ring_name, sizeof(ring_name), "rx_to_tx_ring_%u", i);
ring_name, ring_size,
if (cfg.ports[i].rx_to_tx_ring == NULL)
rte_exit(EXIT_FAILURE,
"Ring create failed: %s\n",
}
}
static inline void
port_init(uint16_t portid,
struct rte_mempool *mbuf_pool, uint16_t nb_queues)
{
},
.rx_adv_conf = {
.rss_conf = {
.rss_key = NULL,
.rss_hf = ETH_RSS_PROTO_MASK,
}
}
};
int ret, i;
if ((ioat_enabled_port_mask & (1 << portid)) == 0) {
printf("Skipping disabled port %u\n", portid);
return;
}
printf("Initializing port %u... ", portid);
fflush(stdout);
if (ret < 0)
rte_exit(EXIT_FAILURE,
"Cannot get device info: %s, port=%u\n",
if (ret < 0)
rte_exit(EXIT_FAILURE,
"Cannot configure device:" " err=%d, port=%u\n", ret, portid);
&nb_txd);
if (ret < 0)
"Cannot adjust number of descriptors: err=%d, port=%u\n",
ret, portid);
for (i = 0; i < nb_queues; i++) {
mbuf_pool);
if (ret < 0)
"rte_eth_rx_queue_setup:err=%d,port=%u, queue_id=%u\n",
ret, portid, i);
}
&txq_conf);
if (ret < 0)
"rte_eth_tx_queue_setup:err=%d,port=%u\n",
ret, portid);
if (tx_buffer[portid] == NULL)
"Cannot allocate buffer for tx on port %u\n",
portid);
&port_statistics.tx_dropped[portid]);
if (ret < 0)
"Cannot set error callback for tx buffer on port %u\n",
portid);
if (ret < 0)
"rte_eth_dev_start:err=%d, port=%u\n",
ret, portid);
printf("Port %u, MAC address: %02X:%02X:%02X:%02X:%02X:%02X\n\n",
portid,
ioat_ports_eth_addr[portid].addr_bytes[0],
ioat_ports_eth_addr[portid].addr_bytes[1],
ioat_ports_eth_addr[portid].addr_bytes[2],
ioat_ports_eth_addr[portid].addr_bytes[3],
ioat_ports_eth_addr[portid].addr_bytes[4],
ioat_ports_eth_addr[portid].addr_bytes[5]);
cfg.ports[cfg.nb_ports].rxtx_port = portid;
cfg.ports[cfg.nb_ports++].nb_queues = nb_queues;
}
static void
signal_handler(int signum)
{
if (signum == SIGINT || signum == SIGTERM) {
printf("\n\nSignal %d received, preparing to exit...\n",
signum);
force_quit = true;
}
}
int
main(int argc, char **argv)
{
int ret;
uint16_t nb_ports, portid;
uint32_t i;
unsigned int nb_mbufs;
if (ret < 0)
rte_exit(EXIT_FAILURE,
"Invalid EAL arguments\n");
argc -= ret;
argv += ret;
force_quit = false;
signal(SIGINT, signal_handler);
signal(SIGTERM, signal_handler);
if (nb_ports == 0)
rte_exit(EXIT_FAILURE,
"No Ethernet ports - bye\n");
ret = ioat_parse_args(argc, argv, nb_ports);
if (ret < 0)
rte_exit(EXIT_FAILURE,
"Invalid IOAT arguments\n");
nb_mbufs =
RTE_MAX(nb_ports * (nb_queues * (nb_rxd + nb_txd +
MIN_POOL_SIZE);
MEMPOOL_CACHE_SIZE, 0, RTE_MBUF_DEFAULT_BUF_SIZE,
if (ioat_pktmbuf_pool == NULL)
rte_exit(EXIT_FAILURE,
"Cannot init mbuf pool\n");
cfg.nb_ports = 0;
port_init(portid, ioat_pktmbuf_pool, nb_queues);
memset(&port_statistics, 0, sizeof(port_statistics));
while (!check_link_status(ioat_enabled_port_mask) && !force_quit)
sleep(1);
if (cfg.nb_lcores < 1)
"There should be at least one slave lcore.\n");
if (copy_mode == COPY_MODE_IOAT_NUM)
assign_rawdevs();
else
assign_rings();
start_forwarding_cores();
print_stats(argv[0]);
uint32_t j;
for (i = 0; i < cfg.nb_ports; i++) {
printf("Closing port %d\n", cfg.ports[i].rxtx_port);
if (copy_mode == COPY_MODE_IOAT_NUM) {
for (j = 0; j < cfg.ports[i].nb_queues; j++) {
printf("Stopping rawdev %d\n",
cfg.ports[i].ioat_ids[j]);
}
} else
}
printf("Bye...\n");
return 0;
}