#include <stdio.h>
#include <stdlib.h>
#include <stdint.h>
#include <inttypes.h>
#include <string.h>
#include <sys/queue.h>
#include <stdarg.h>
#include <errno.h>
#include <getopt.h>
#include <netinet/in.h>
#include <linux/if.h>
#include <linux/if_tun.h>
#include <fcntl.h>
#include <sys/ioctl.h>
#include <unistd.h>
#include <signal.h>
#include <rte_bus_pci.h>
#define RTE_LOGTYPE_APP RTE_LOGTYPE_USER1
#define MAX_PACKET_SZ 2048
#define MBUF_DATA_SZ (MAX_PACKET_SZ + RTE_PKTMBUF_HEADROOM)
#define NB_MBUF (8192 * 16)
#define PKT_BURST_SZ 32
#define MEMPOOL_CACHE_SZ PKT_BURST_SZ
#define NB_RXD 1024
#define NB_TXD 1024
#define KNI_ENET_HEADER_SIZE 14
#define KNI_ENET_FCS_SIZE 4
#define KNI_US_PER_SECOND 1000000
#define KNI_SECOND_PER_DAY 86400
#define KNI_MAX_KTHREAD 32
struct kni_port_params {
uint16_t port_id;
unsigned lcore_rx;
unsigned lcore_tx;
uint32_t nb_lcore_k;
uint32_t nb_kni;
unsigned lcore_k[KNI_MAX_KTHREAD];
struct rte_kni *kni[KNI_MAX_KTHREAD];
static struct kni_port_params *kni_port_params_array[RTE_MAX_ETHPORTS];
},
};
static uint32_t ports_mask = 0;
static int promiscuous_on = 0;
static int monitor_links;
struct kni_interface_stats {
uint64_t rx_packets;
uint64_t rx_dropped;
uint64_t tx_packets;
uint64_t tx_dropped;
};
static struct kni_interface_stats kni_stats[RTE_MAX_ETHPORTS];
static int kni_change_mtu(uint16_t port_id, unsigned int new_mtu);
static int kni_config_network_interface(uint16_t port_id, uint8_t if_up);
static int kni_config_mac_address(uint16_t port_id, uint8_t mac_addr[]);
static void
print_stats(void)
{
uint16_t i;
printf("\n**KNI example application statistics**\n"
"====== ============== ============ ============ ============ ============\n"
" Port Lcore(RX/TX) rx_packets rx_dropped tx_packets tx_dropped\n"
"------ -------------- ------------ ------------ ------------ ------------\n");
for (i = 0; i < RTE_MAX_ETHPORTS; i++) {
if (!kni_port_params_array[i])
continue;
printf("%7d %10u/%2u %13"PRIu64" %13"PRIu64" %13"PRIu64" "
"%13"PRIu64"\n", i,
kni_port_params_array[i]->lcore_rx,
kni_port_params_array[i]->lcore_tx,
kni_stats[i].rx_packets,
kni_stats[i].rx_dropped,
kni_stats[i].tx_packets,
kni_stats[i].tx_dropped);
}
printf("====== ============== ============ ============ ============ ============\n");
}
static void
signal_handler(int signum)
{
if (signum == SIGUSR1) {
print_stats();
}
if (signum == SIGUSR2) {
memset(&kni_stats, 0, sizeof(kni_stats));
printf("\n** Statistics have been reset **\n");
return;
}
if (signum == SIGRTMIN || signum == SIGINT || signum == SIGTERM) {
printf("\nSIGRTMIN/SIGINT/SIGTERM received. "
"KNI processing stopping.\n");
return;
}
}
static void
kni_burst_free_mbufs(
struct rte_mbuf **pkts,
unsigned num)
{
unsigned i;
if (pkts == NULL)
return;
for (i = 0; i < num; i++) {
pkts[i] = NULL;
}
}
static void
kni_ingress(struct kni_port_params *p)
{
uint8_t i;
uint16_t port_id;
unsigned nb_rx, num;
uint32_t nb_kni;
struct rte_mbuf *pkts_burst[PKT_BURST_SZ];
if (p == NULL)
return;
nb_kni = p->nb_kni;
port_id = p->port_id;
for (i = 0; i < nb_kni; i++) {
RTE_LOG(ERR, APP,
"Error receiving from eth\n");
return;
}
if (num)
kni_stats[port_id].rx_packets += num;
kni_burst_free_mbufs(&pkts_burst[num], nb_rx - num);
kni_stats[port_id].rx_dropped += nb_rx - num;
}
}
}
static void
kni_egress(struct kni_port_params *p)
{
uint8_t i;
uint16_t port_id;
unsigned nb_tx, num;
uint32_t nb_kni;
struct rte_mbuf *pkts_burst[PKT_BURST_SZ];
if (p == NULL)
return;
nb_kni = p->nb_kni;
port_id = p->port_id;
for (i = 0; i < nb_kni; i++) {
RTE_LOG(ERR, APP,
"Error receiving from KNI\n");
return;
}
if (nb_tx)
kni_stats[port_id].tx_packets += nb_tx;
kni_burst_free_mbufs(&pkts_burst[nb_tx], num - nb_tx);
kni_stats[port_id].tx_dropped += num - nb_tx;
}
}
}
static int
{
uint16_t i;
int32_t f_stop;
int32_t f_pause;
enum lcore_rxtx {
LCORE_NONE,
LCORE_RX,
LCORE_TX,
LCORE_MAX
};
enum lcore_rxtx flag = LCORE_NONE;
if (!kni_port_params_array[i])
continue;
if (kni_port_params_array[i]->lcore_rx == (uint8_t)lcore_id) {
flag = LCORE_RX;
break;
} else if (kni_port_params_array[i]->lcore_tx ==
(uint8_t)lcore_id) {
flag = LCORE_TX;
break;
}
}
if (flag == LCORE_RX) {
RTE_LOG(INFO, APP,
"Lcore %u is reading from port %d\n",
kni_port_params_array[i]->lcore_rx,
kni_port_params_array[i]->port_id);
while (1) {
if (f_stop)
break;
if (f_pause)
continue;
kni_ingress(kni_port_params_array[i]);
}
} else if (flag == LCORE_TX) {
RTE_LOG(INFO, APP,
"Lcore %u is writing to port %d\n",
kni_port_params_array[i]->lcore_tx,
kni_port_params_array[i]->port_id);
while (1) {
if (f_stop)
break;
if (f_pause)
continue;
kni_egress(kni_port_params_array[i]);
}
} else
RTE_LOG(INFO, APP,
"Lcore %u has nothing to do\n", lcore_id);
return 0;
}
static void
print_usage(const char *prgname)
{
RTE_LOG(INFO, APP,
"\nUsage: %s [EAL options] -- -p PORTMASK -P -m "
"[--config (port,lcore_rx,lcore_tx,lcore_kthread...)"
"[,(port,lcore_rx,lcore_tx,lcore_kthread...)]]\n"
" -p PORTMASK: hex bitmask of ports to use\n"
" -P : enable promiscuous mode\n"
" -m : enable monitoring of port carrier state\n"
" --config (port,lcore_rx,lcore_tx,lcore_kthread...): "
"port and lcore configurations\n",
prgname);
}
static uint32_t
parse_unsigned(const char *portmask)
{
char *end = NULL;
unsigned long num;
num = strtoul(portmask, &end, 16);
if ((portmask[0] == '\0') || (end == NULL) || (*end != '\0'))
return 0;
return (uint32_t)num;
}
static void
print_config(void)
{
uint32_t i, j;
struct kni_port_params **p = kni_port_params_array;
for (i = 0; i < RTE_MAX_ETHPORTS; i++) {
if (!p[i])
continue;
RTE_LOG(DEBUG, APP,
"Port ID: %d\n", p[i]->port_id);
RTE_LOG(DEBUG, APP,
"Rx lcore ID: %u, Tx lcore ID: %u\n",
p[i]->lcore_rx, p[i]->lcore_tx);
for (j = 0; j < p[i]->nb_lcore_k; j++)
RTE_LOG(DEBUG, APP,
"Kernel thread lcore ID: %u\n",
p[i]->lcore_k[j]);
}
}
static int
parse_config(const char *arg)
{
const char *p, *p0 = arg;
char s[256], *end;
unsigned size;
enum fieldnames {
FLD_PORT = 0,
FLD_LCORE_RX,
FLD_LCORE_TX,
_NUM_FLD = KNI_MAX_KTHREAD + 3,
};
int i, j, nb_token;
char *str_fld[_NUM_FLD];
unsigned long int_fld[_NUM_FLD];
uint16_t port_id, nb_kni_port_params = 0;
memset(&kni_port_params_array, 0, sizeof(kni_port_params_array));
while (((p = strchr(p0, '(')) != NULL) &&
nb_kni_port_params < RTE_MAX_ETHPORTS) {
p++;
if ((p0 = strchr(p, ')')) == NULL)
goto fail;
size = p0 - p;
if (size >= sizeof(s)) {
printf("Invalid config parameters\n");
goto fail;
}
snprintf(s, sizeof(s), "%.*s", size, p);
nb_token =
rte_strsplit(s,
sizeof(s), str_fld, _NUM_FLD,
',');
if (nb_token <= FLD_LCORE_TX) {
printf("Invalid config parameters\n");
goto fail;
}
for (i = 0; i < nb_token; i++) {
errno = 0;
int_fld[i] = strtoul(str_fld[i], &end, 0);
if (errno != 0 || end == str_fld[i]) {
printf("Invalid config parameters\n");
goto fail;
}
}
i = 0;
port_id = int_fld[i++];
if (port_id >= RTE_MAX_ETHPORTS) {
printf("Port ID %d could not exceed the maximum %d\n",
port_id, RTE_MAX_ETHPORTS);
goto fail;
}
if (kni_port_params_array[port_id]) {
printf("Port %d has been configured\n", port_id);
goto fail;
}
kni_port_params_array[port_id] =
sizeof(struct kni_port_params), RTE_CACHE_LINE_SIZE);
kni_port_params_array[port_id]->port_id = port_id;
kni_port_params_array[port_id]->lcore_rx =
(uint8_t)int_fld[i++];
kni_port_params_array[port_id]->lcore_tx =
(uint8_t)int_fld[i++];
if (kni_port_params_array[port_id]->lcore_rx >= RTE_MAX_LCORE ||
kni_port_params_array[port_id]->lcore_tx >= RTE_MAX_LCORE) {
printf("lcore_rx %u or lcore_tx %u ID could not "
"exceed the maximum %u\n",
kni_port_params_array[port_id]->lcore_rx,
kni_port_params_array[port_id]->lcore_tx,
(unsigned)RTE_MAX_LCORE);
goto fail;
}
for (j = 0; i < nb_token && j < KNI_MAX_KTHREAD; i++, j++)
kni_port_params_array[port_id]->lcore_k[j] =
(uint8_t)int_fld[i];
kni_port_params_array[port_id]->nb_lcore_k = j;
}
print_config();
return 0;
fail:
for (i = 0; i < RTE_MAX_ETHPORTS; i++) {
if (kni_port_params_array[i]) {
kni_port_params_array[i] = NULL;
}
}
return -1;
}
static int
validate_parameters(uint32_t portmask)
{
uint32_t i;
if (!portmask) {
printf("No port configured in port mask\n");
return -1;
}
for (i = 0; i < RTE_MAX_ETHPORTS; i++) {
if (((portmask & (1 << i)) && !kni_port_params_array[i]) ||
(!(portmask & (1 << i)) && kni_port_params_array[i]))
rte_exit(EXIT_FAILURE,
"portmask is not consistent "
"to port ids specified in --config\n");
(unsigned)(kni_port_params_array[i]->lcore_rx)))
rte_exit(EXIT_FAILURE,
"lcore id %u for "
"port %d receiving not enabled\n",
kni_port_params_array[i]->lcore_rx,
kni_port_params_array[i]->port_id);
(unsigned)(kni_port_params_array[i]->lcore_tx)))
rte_exit(EXIT_FAILURE,
"lcore id %u for "
"port %d transmitting not enabled\n",
kni_port_params_array[i]->lcore_tx,
kni_port_params_array[i]->port_id);
}
return 0;
}
#define CMDLINE_OPT_CONFIG "config"
static int
parse_args(int argc, char **argv)
{
int opt, longindex, ret = 0;
const char *prgname = argv[0];
static struct option longopts[] = {
{CMDLINE_OPT_CONFIG, required_argument, NULL, 0},
{NULL, 0, NULL, 0}
};
opterr = 0;
while ((opt = getopt_long(argc, argv, "p:Pm", longopts,
&longindex)) != EOF) {
switch (opt) {
case 'p':
ports_mask = parse_unsigned(optarg);
break;
case 'P':
promiscuous_on = 1;
break;
case 'm':
monitor_links = 1;
break;
case 0:
if (!strncmp(longopts[longindex].name,
CMDLINE_OPT_CONFIG,
sizeof(CMDLINE_OPT_CONFIG))) {
ret = parse_config(optarg);
if (ret) {
printf("Invalid config\n");
print_usage(prgname);
return -1;
}
}
break;
default:
print_usage(prgname);
rte_exit(EXIT_FAILURE,
"Invalid option specified\n");
}
}
if (validate_parameters(ports_mask) < 0) {
print_usage(prgname);
rte_exit(EXIT_FAILURE,
"Invalid parameters\n");
}
return ret;
}
static void
init_kni(void)
{
unsigned int num_of_kni_ports = 0, i;
struct kni_port_params **params = kni_port_params_array;
for (i = 0; i < RTE_MAX_ETHPORTS; i++) {
if (kni_port_params_array[i]) {
num_of_kni_ports += (params[i]->nb_lcore_k ?
params[i]->nb_lcore_k : 1);
}
}
}
static void
init_port(uint16_t port)
{
int ret;
uint16_t nb_rxd = NB_RXD;
uint16_t nb_txd = NB_TXD;
RTE_LOG(INFO, APP,
"Initialising port %u ...\n", (
unsigned)port);
fflush(stdout);
if (ret != 0)
"Error during getting device (port %u) info: %s\n",
port, strerror(-ret));
if (ret < 0)
rte_exit(EXIT_FAILURE,
"Could not configure port%u (%d)\n",
(unsigned)port, ret);
if (ret < 0)
rte_exit(EXIT_FAILURE,
"Could not adjust number of descriptors "
"for port%u (%d)\n", (unsigned)port, ret);
rxq_conf = dev_info.default_rxconf;
if (ret < 0)
rte_exit(EXIT_FAILURE,
"Could not setup up RX queue for "
"port%u (%d)\n", (unsigned)port, ret);
txq_conf = dev_info.default_txconf;
if (ret < 0)
rte_exit(EXIT_FAILURE,
"Could not setup up TX queue for "
"port%u (%d)\n", (unsigned)port, ret);
if (ret < 0)
rte_exit(EXIT_FAILURE,
"Could not start port%u (%d)\n",
(unsigned)port, ret);
if (promiscuous_on) {
if (ret != 0)
"Could not enable promiscuous mode for port%u: %s\n",
}
}
static void
check_all_ports_link_status(uint32_t port_mask)
{
#define CHECK_INTERVAL 100
#define MAX_CHECK_TIME 90
uint16_t portid;
uint8_t count, all_ports_up, print_flag = 0;
int ret;
printf("\nChecking link status\n");
fflush(stdout);
for (count = 0; count <= MAX_CHECK_TIME; count++) {
all_ports_up = 1;
if ((port_mask & (1 << portid)) == 0)
continue;
memset(&link, 0, sizeof(link));
if (ret < 0) {
all_ports_up = 0;
if (print_flag == 1)
printf("Port %u link get failed: %s\n",
continue;
}
if (print_flag == 1) {
if (link.link_status)
printf(
"Port%d Link Up - speed %uMbps - %s\n",
portid, link.link_speed,
("full-duplex") : ("half-duplex"));
else
printf("Port %d Link Down\n", portid);
continue;
}
all_ports_up = 0;
break;
}
}
if (print_flag == 1)
break;
if (all_ports_up == 0) {
printf(".");
fflush(stdout);
}
if (all_ports_up == 1 || count == (MAX_CHECK_TIME - 1)) {
print_flag = 1;
printf("done\n");
}
}
}
static void
log_link_state(
struct rte_kni *kni,
int prev,
struct rte_eth_link *link)
{
if (kni == NULL || link == NULL)
return;
RTE_LOG(INFO, APP,
"%s NIC Link is Up %d Mbps %s %s.\n",
RTE_LOG(INFO, APP,
"%s NIC Link is Down.\n",
}
}
static void *
monitor_all_ports_link_status(void *arg)
{
uint16_t portid;
unsigned int i;
struct kni_port_params **p = kni_port_params_array;
int prev;
(void) arg;
int ret;
while (monitor_links) {
if ((ports_mask & (1 << portid)) == 0)
continue;
memset(&link, 0, sizeof(link));
if (ret < 0) {
"Get link failed (port %u): %s\n",
continue;
}
for (i = 0; i < p[portid]->nb_kni; i++) {
log_link_state(p[portid]->kni[i], prev, &link);
}
}
}
return NULL;
}
static int
kni_change_mtu_(uint16_t port_id, unsigned int new_mtu)
{
int ret;
uint16_t nb_rxd = NB_RXD;
uint16_t nb_txd = NB_TXD;
RTE_LOG(ERR, APP,
"Invalid port id %d\n", port_id);
return -EINVAL;
}
RTE_LOG(INFO, APP,
"Change MTU of port %d to %u\n", port_id, new_mtu);
memcpy(&conf, &port_conf, sizeof(conf));
conf.rxmode.offloads |= DEV_RX_OFFLOAD_JUMBO_FRAME;
else
conf.rxmode.offloads &= ~DEV_RX_OFFLOAD_JUMBO_FRAME;
conf.rxmode.max_rx_pkt_len = new_mtu + KNI_ENET_HEADER_SIZE +
KNI_ENET_FCS_SIZE;
if (ret < 0) {
RTE_LOG(ERR, APP,
"Fail to reconfigure port %d\n", port_id);
return ret;
}
if (ret < 0)
rte_exit(EXIT_FAILURE,
"Could not adjust number of descriptors "
"for port%u (%d)\n", (unsigned int)port_id,
ret);
if (ret != 0) {
"Error during getting device (port %u) info: %s\n",
port_id, strerror(-ret));
return ret;
}
rxq_conf = dev_info.default_rxconf;
rxq_conf.offloads = conf.rxmode.offloads;
if (ret < 0) {
RTE_LOG(ERR, APP,
"Fail to setup Rx queue of port %d\n",
port_id);
return ret;
}
txq_conf = dev_info.default_txconf;
txq_conf.offloads = conf.txmode.offloads;
if (ret < 0) {
RTE_LOG(ERR, APP,
"Fail to setup Tx queue of port %d\n",
port_id);
return ret;
}
if (ret < 0) {
RTE_LOG(ERR, APP,
"Fail to restart port %d\n", port_id);
return ret;
}
return 0;
}
static int
kni_change_mtu(uint16_t port_id, unsigned int new_mtu)
{
int ret;
ret = kni_change_mtu_(port_id, new_mtu);
return ret;
}
static int
kni_config_network_interface(uint16_t port_id, uint8_t if_up)
{
int ret = 0;
RTE_LOG(ERR, APP,
"Invalid port id %d\n", port_id);
return -EINVAL;
}
RTE_LOG(INFO, APP,
"Configure network interface of %d %s\n",
port_id, if_up ? "up" : "down");
if (if_up != 0) {
} else
if (ret < 0)
RTE_LOG(ERR, APP,
"Failed to start port %d\n", port_id);
return ret;
}
static void
{
char buf[RTE_ETHER_ADDR_FMT_SIZE];
RTE_LOG(INFO, APP,
"\t%s%s\n", name, buf);
}
static int
kni_config_mac_address(uint16_t port_id, uint8_t mac_addr[])
{
int ret = 0;
RTE_LOG(ERR, APP,
"Invalid port id %d\n", port_id);
return -EINVAL;
}
RTE_LOG(INFO, APP,
"Configure mac address of %d\n", port_id);
if (ret < 0)
RTE_LOG(ERR, APP,
"Failed to config mac_addr for port %d\n",
port_id);
return ret;
}
static int
kni_alloc(uint16_t port_id)
{
uint8_t i;
struct rte_kni *kni;
struct kni_port_params **params = kni_port_params_array;
int ret;
if (port_id >= RTE_MAX_ETHPORTS || !params[port_id])
return -1;
params[port_id]->nb_kni = params[port_id]->nb_lcore_k ?
params[port_id]->nb_lcore_k : 1;
for (i = 0; i < params[port_id]->nb_kni; i++) {
memset(&conf, 0, sizeof(conf));
if (params[port_id]->nb_lcore_k) {
snprintf(conf.name, RTE_KNI_NAMESIZE,
"vEth%u_%u", port_id, i);
conf.core_id = params[port_id]->lcore_k[i];
conf.force_bind = 1;
} else
snprintf(conf.name, RTE_KNI_NAMESIZE,
"vEth%u", port_id);
conf.group_id = port_id;
conf.mbuf_size = MAX_PACKET_SZ;
if (i == 0) {
if (ret != 0)
"Error during getting device (port %u) info: %s\n",
port_id, strerror(-ret));
if (ret != 0)
"Failed to get MAC address (port %u): %s\n",
conf.min_mtu = dev_info.min_mtu;
conf.max_mtu = dev_info.max_mtu;
memset(&ops, 0, sizeof(ops));
ops.port_id = port_id;
ops.change_mtu = kni_change_mtu;
ops.config_network_if = kni_config_network_interface;
ops.config_mac_address = kni_config_mac_address;
} else
if (!kni)
rte_exit(EXIT_FAILURE,
"Fail to create kni for "
"port: %d\n", port_id);
params[port_id]->kni[i] = kni;
}
return 0;
}
static int
kni_free_kni(uint16_t port_id)
{
uint8_t i;
struct kni_port_params **p = kni_port_params_array;
if (port_id >= RTE_MAX_ETHPORTS || !p[port_id])
return -1;
for (i = 0; i < p[port_id]->nb_kni; i++) {
printf("Fail to release kni\n");
p[port_id]->kni[i] = NULL;
}
return 0;
}
int
main(int argc, char** argv)
{
int ret;
uint16_t nb_sys_ports, port;
unsigned i;
void *retval;
pthread_t kni_link_tid;
int pid;
signal(SIGUSR1, signal_handler);
signal(SIGUSR2, signal_handler);
signal(SIGRTMIN, signal_handler);
signal(SIGINT, signal_handler);
signal(SIGTERM, signal_handler);
if (ret < 0)
rte_exit(EXIT_FAILURE,
"Could not initialise EAL (%d)\n", ret);
argc -= ret;
argv += ret;
ret = parse_args(argc, argv);
if (ret < 0)
rte_exit(EXIT_FAILURE,
"Could not parse input parameters\n");
if (pktmbuf_pool == NULL) {
rte_exit(EXIT_FAILURE,
"Could not initialise mbuf pool\n");
return -1;
}
if (nb_sys_ports == 0)
rte_exit(EXIT_FAILURE,
"No supported Ethernet device found\n");
for (i = 0; i < RTE_MAX_ETHPORTS; i++)
rte_exit(EXIT_FAILURE,
"Configured invalid "
"port ID %u\n", i);
init_kni();
if (!(ports_mask & (1 << port)))
continue;
init_port(port);
if (port >= RTE_MAX_ETHPORTS)
rte_exit(EXIT_FAILURE,
"Can not use more than "
"%d ports for kni\n", RTE_MAX_ETHPORTS);
kni_alloc(port);
}
check_all_ports_link_status(ports_mask);
pid = getpid();
RTE_LOG(INFO, APP,
"========================\n");
RTE_LOG(INFO, APP,
"KNI Running\n");
RTE_LOG(INFO, APP,
"kill -SIGUSR1 %d\n", pid);
RTE_LOG(INFO, APP,
" Show KNI Statistics.\n");
RTE_LOG(INFO, APP,
"kill -SIGUSR2 %d\n", pid);
RTE_LOG(INFO, APP,
" Zero KNI Statistics.\n");
RTE_LOG(INFO, APP,
"========================\n");
fflush(stdout);
"KNI link status check", NULL,
monitor_all_ports_link_status, NULL);
if (ret < 0)
"Could not create link status thread!\n");
return -1;
}
monitor_links = 0;
pthread_join(kni_link_tid, &retval);
if (!(ports_mask & (1 << port)))
continue;
kni_free_kni(port);
}
for (i = 0; i < RTE_MAX_ETHPORTS; i++)
if (kni_port_params_array[i]) {
kni_port_params_array[i] = NULL;
}
return 0;
}