1 /* SPDX-License-Identifier: BSD-3-Clause 2 * Copyright 2017 Mellanox Technologies, Ltd 3 */ 4 5 #include <stdio.h> 6 #include <stdlib.h> 7 #include <string.h> 8 #include <stdint.h> 9 #include <inttypes.h> 10 #include <sys/types.h> 11 #include <sys/queue.h> 12 #include <netinet/in.h> 13 #include <setjmp.h> 14 #include <stdarg.h> 15 #include <ctype.h> 16 #include <errno.h> 17 #include <getopt.h> 18 #include <signal.h> 19 #include <stdbool.h> 20 21 #include <rte_eal.h> 22 #include <rte_common.h> 23 #include <rte_malloc.h> 24 #include <rte_ether.h> 25 #include <rte_ethdev.h> 26 #include <rte_mempool.h> 27 #include <rte_mbuf.h> 28 #include <rte_net.h> 29 #include <rte_flow.h> 30 #include <rte_cycles.h> 31 32 static volatile bool force_quit; 33 34 static uint16_t port_id; 35 static uint16_t nr_queues = 5; 36 static uint8_t selected_queue = 1; 37 struct rte_mempool *mbuf_pool; 38 struct rte_flow *flow; 39 40 #define SRC_IP ((0<<24) + (0<<16) + (0<<8) + 0) /* src ip = 0.0.0.0 */ 41 #define DEST_IP ((192<<24) + (168<<16) + (1<<8) + 1) /* dest ip = 192.168.1.1 */ 42 #define FULL_MASK 0xffffffff /* full mask */ 43 #define EMPTY_MASK 0x0 /* empty mask */ 44 45 #include "flow_blocks.c" 46 47 static inline void 48 print_ether_addr(const char *what, struct rte_ether_addr *eth_addr) 49 { 50 char buf[RTE_ETHER_ADDR_FMT_SIZE]; 51 rte_ether_format_addr(buf, RTE_ETHER_ADDR_FMT_SIZE, eth_addr); 52 printf("%s%s", what, buf); 53 } 54 55 static void 56 main_loop(void) 57 { 58 struct rte_mbuf *mbufs[32]; 59 struct rte_ether_hdr *eth_hdr; 60 struct rte_flow_error error; 61 uint16_t nb_rx; 62 uint16_t i; 63 uint16_t j; 64 65 while (!force_quit) { 66 for (i = 0; i < nr_queues; i++) { 67 nb_rx = rte_eth_rx_burst(port_id, 68 i, mbufs, 32); 69 if (nb_rx) { 70 for (j = 0; j < nb_rx; j++) { 71 struct rte_mbuf *m = mbufs[j]; 72 73 eth_hdr = rte_pktmbuf_mtod(m, 74 struct rte_ether_hdr *); 75 print_ether_addr("src=", 76 ð_hdr->s_addr); 77 print_ether_addr(" - dst=", 78 ð_hdr->d_addr); 79 printf(" - queue=0x%x", 80 (unsigned int)i); 81 printf("\n"); 82 83 rte_pktmbuf_free(m); 84 } 85 } 86 } 87 } 88 89 /* closing and releasing resources */ 90 rte_flow_flush(port_id, &error); 91 rte_eth_dev_stop(port_id); 92 rte_eth_dev_close(port_id); 93 } 94 95 #define CHECK_INTERVAL 1000 /* 100ms */ 96 #define MAX_REPEAT_TIMES 90 /* 9s (90 * 100ms) in total */ 97 98 static void 99 assert_link_status(void) 100 { 101 struct rte_eth_link link; 102 uint8_t rep_cnt = MAX_REPEAT_TIMES; 103 104 memset(&link, 0, sizeof(link)); 105 do { 106 rte_eth_link_get(port_id, &link); 107 if (link.link_status == ETH_LINK_UP) 108 break; 109 rte_delay_ms(CHECK_INTERVAL); 110 } while (--rep_cnt); 111 112 if (link.link_status == ETH_LINK_DOWN) 113 rte_exit(EXIT_FAILURE, ":: error: link is still down\n"); 114 } 115 116 static void 117 init_port(void) 118 { 119 int ret; 120 uint16_t i; 121 struct rte_eth_conf port_conf = { 122 .rxmode = { 123 .split_hdr_size = 0, 124 }, 125 .txmode = { 126 .offloads = 127 DEV_TX_OFFLOAD_VLAN_INSERT | 128 DEV_TX_OFFLOAD_IPV4_CKSUM | 129 DEV_TX_OFFLOAD_UDP_CKSUM | 130 DEV_TX_OFFLOAD_TCP_CKSUM | 131 DEV_TX_OFFLOAD_SCTP_CKSUM | 132 DEV_TX_OFFLOAD_TCP_TSO, 133 }, 134 }; 135 struct rte_eth_txconf txq_conf; 136 struct rte_eth_rxconf rxq_conf; 137 struct rte_eth_dev_info dev_info; 138 139 rte_eth_dev_info_get(port_id, &dev_info); 140 port_conf.txmode.offloads &= dev_info.tx_offload_capa; 141 printf(":: initializing port: %d\n", port_id); 142 ret = rte_eth_dev_configure(port_id, 143 nr_queues, nr_queues, &port_conf); 144 if (ret < 0) { 145 rte_exit(EXIT_FAILURE, 146 ":: cannot configure device: err=%d, port=%u\n", 147 ret, port_id); 148 } 149 150 rxq_conf = dev_info.default_rxconf; 151 rxq_conf.offloads = port_conf.rxmode.offloads; 152 for (i = 0; i < nr_queues; i++) { 153 ret = rte_eth_rx_queue_setup(port_id, i, 512, 154 rte_eth_dev_socket_id(port_id), 155 &rxq_conf, 156 mbuf_pool); 157 if (ret < 0) { 158 rte_exit(EXIT_FAILURE, 159 ":: Rx queue setup failed: err=%d, port=%u\n", 160 ret, port_id); 161 } 162 } 163 164 txq_conf = dev_info.default_txconf; 165 txq_conf.offloads = port_conf.txmode.offloads; 166 167 for (i = 0; i < nr_queues; i++) { 168 ret = rte_eth_tx_queue_setup(port_id, i, 512, 169 rte_eth_dev_socket_id(port_id), 170 &txq_conf); 171 if (ret < 0) { 172 rte_exit(EXIT_FAILURE, 173 ":: Tx queue setup failed: err=%d, port=%u\n", 174 ret, port_id); 175 } 176 } 177 178 rte_eth_promiscuous_enable(port_id); 179 ret = rte_eth_dev_start(port_id); 180 if (ret < 0) { 181 rte_exit(EXIT_FAILURE, 182 "rte_eth_dev_start:err=%d, port=%u\n", 183 ret, port_id); 184 } 185 186 assert_link_status(); 187 188 printf(":: initializing port: %d done\n", port_id); 189 } 190 191 static void 192 signal_handler(int signum) 193 { 194 if (signum == SIGINT || signum == SIGTERM) { 195 printf("\n\nSignal %d received, preparing to exit...\n", 196 signum); 197 force_quit = true; 198 } 199 } 200 201 int 202 main(int argc, char **argv) 203 { 204 int ret; 205 uint16_t nr_ports; 206 struct rte_flow_error error; 207 208 ret = rte_eal_init(argc, argv); 209 if (ret < 0) 210 rte_exit(EXIT_FAILURE, ":: invalid EAL arguments\n"); 211 212 force_quit = false; 213 signal(SIGINT, signal_handler); 214 signal(SIGTERM, signal_handler); 215 216 nr_ports = rte_eth_dev_count_avail(); 217 if (nr_ports == 0) 218 rte_exit(EXIT_FAILURE, ":: no Ethernet ports found\n"); 219 port_id = 0; 220 if (nr_ports != 1) { 221 printf(":: warn: %d ports detected, but we use only one: port %u\n", 222 nr_ports, port_id); 223 } 224 mbuf_pool = rte_pktmbuf_pool_create("mbuf_pool", 4096, 128, 0, 225 RTE_MBUF_DEFAULT_BUF_SIZE, 226 rte_socket_id()); 227 if (mbuf_pool == NULL) 228 rte_exit(EXIT_FAILURE, "Cannot init mbuf pool\n"); 229 230 init_port(); 231 232 /* create flow for send packet with */ 233 flow = generate_ipv4_flow(port_id, selected_queue, 234 SRC_IP, EMPTY_MASK, 235 DEST_IP, FULL_MASK, &error); 236 if (!flow) { 237 printf("Flow can't be created %d message: %s\n", 238 error.type, 239 error.message ? error.message : "(no stated reason)"); 240 rte_exit(EXIT_FAILURE, "error in creating flow"); 241 } 242 243 main_loop(); 244 245 return 0; 246 } 247