1 /* SPDX-License-Identifier: BSD-3-Clause
2 * Copyright(c) 2015 Intel Corporation
9 #include <rte_common.h>
10 #include <rte_spinlock.h>
12 #include <rte_ethdev.h>
13 #include <rte_ether.h>
15 #include <rte_memory.h>
16 #include <rte_mempool.h>
21 #define MAX_PORTS RTE_MAX_ETHPORTS
22 #define MAX_BURST_LENGTH 32
23 #define PORT_RX_QUEUE_SIZE 1024
24 #define PORT_TX_QUEUE_SIZE 1024
25 #define PKTPOOL_EXTRA_SIZE 512
26 #define PKTPOOL_CACHE 32
31 struct rte_mbuf *buf_frames[MAX_BURST_LENGTH];
35 struct rte_ether_addr mac_addr;
41 struct rte_mempool *pkt_pool;
45 struct app_port ports[MAX_PORTS];
51 struct app_config app_cfg;
54 void lock_port(int idx_port)
56 struct app_port *ptr_port = &app_cfg.ports[idx_port];
58 rte_spinlock_lock(&ptr_port->lock);
61 void unlock_port(int idx_port)
63 struct app_port *ptr_port = &app_cfg.ports[idx_port];
65 rte_spinlock_unlock(&ptr_port->lock);
68 void mark_port_active(int idx_port)
70 struct app_port *ptr_port = &app_cfg.ports[idx_port];
72 ptr_port->port_active = 1;
75 void mark_port_inactive(int idx_port)
77 struct app_port *ptr_port = &app_cfg.ports[idx_port];
79 ptr_port->port_active = 0;
82 void mark_port_newmac(int idx_port)
84 struct app_port *ptr_port = &app_cfg.ports[idx_port];
86 ptr_port->port_dirty = 1;
89 static void setup_ports(struct app_config *app_cfg, int cnt_ports)
93 struct rte_eth_conf cfg_port;
94 struct rte_eth_dev_info dev_info;
96 uint16_t nb_rxd = PORT_RX_QUEUE_SIZE;
97 uint16_t nb_txd = PORT_TX_QUEUE_SIZE;
100 memset(&cfg_port, 0, sizeof(cfg_port));
101 cfg_port.txmode.mq_mode = RTE_ETH_MQ_TX_NONE;
103 for (idx_port = 0; idx_port < cnt_ports; idx_port++) {
104 struct app_port *ptr_port = &app_cfg->ports[idx_port];
106 ret = rte_eth_dev_info_get(idx_port, &dev_info);
108 rte_exit(EXIT_FAILURE,
109 "Error during getting device (port %u) info: %s\n",
110 idx_port, strerror(-ret));
112 size_pktpool = dev_info.rx_desc_lim.nb_max +
113 dev_info.tx_desc_lim.nb_max + PKTPOOL_EXTRA_SIZE;
115 snprintf(str_name, 16, "pkt_pool%i", idx_port);
116 ptr_port->pkt_pool = rte_pktmbuf_pool_create(
118 size_pktpool, PKTPOOL_CACHE,
120 RTE_MBUF_DEFAULT_BUF_SIZE,
123 if (ptr_port->pkt_pool == NULL)
124 rte_exit(EXIT_FAILURE,
125 "rte_pktmbuf_pool_create failed"
128 printf("Init port %i..\n", idx_port);
129 ptr_port->port_active = 1;
130 ptr_port->port_dirty = 0;
131 ptr_port->idx_port = idx_port;
133 if (rte_eth_dev_configure(idx_port, 1, 1, &cfg_port) < 0)
134 rte_exit(EXIT_FAILURE,
135 "rte_eth_dev_configure failed");
136 if (rte_eth_dev_adjust_nb_rx_tx_desc(idx_port, &nb_rxd,
138 rte_exit(EXIT_FAILURE,
139 "rte_eth_dev_adjust_nb_rx_tx_desc failed");
141 if (rte_eth_rx_queue_setup(
143 rte_eth_dev_socket_id(idx_port), NULL,
144 ptr_port->pkt_pool) < 0)
145 rte_exit(EXIT_FAILURE,
146 "rte_eth_rx_queue_setup failed"
148 if (rte_eth_tx_queue_setup(
150 rte_eth_dev_socket_id(idx_port), NULL) < 0)
151 rte_exit(EXIT_FAILURE,
152 "rte_eth_tx_queue_setup failed"
154 if (rte_eth_dev_start(idx_port) < 0)
155 rte_exit(EXIT_FAILURE,
156 "%s:%i: rte_eth_dev_start failed",
159 ret = rte_eth_macaddr_get(idx_port, &ptr_port->mac_addr);
161 rte_exit(EXIT_FAILURE,
162 "rte_eth_macaddr_get failed (port %u): %s\n",
163 idx_port, rte_strerror(-ret));
165 rte_spinlock_init(&ptr_port->lock);
169 static void process_frame(struct app_port *ptr_port,
170 struct rte_mbuf *ptr_frame)
172 struct rte_ether_hdr *ptr_mac_hdr;
174 ptr_mac_hdr = rte_pktmbuf_mtod(ptr_frame, struct rte_ether_hdr *);
175 rte_ether_addr_copy(&ptr_mac_hdr->src_addr, &ptr_mac_hdr->dst_addr);
176 rte_ether_addr_copy(&ptr_port->mac_addr, &ptr_mac_hdr->src_addr);
179 static int worker_main(__rte_unused void *ptr_data)
181 struct app_port *ptr_port;
182 struct rte_mbuf *ptr_frame;
183 struct txq_port *txq;
185 uint16_t cnt_recv_frames;
189 uint16_t lock_result;
192 while (app_cfg.exit_now == 0) {
193 for (idx_port = 0; idx_port < app_cfg.cnt_ports; idx_port++) {
194 /* Check that port is active and unlocked */
195 ptr_port = &app_cfg.ports[idx_port];
196 lock_result = rte_spinlock_trylock(&ptr_port->lock);
197 if (lock_result == 0)
199 if (ptr_port->port_active == 0) {
200 rte_spinlock_unlock(&ptr_port->lock);
203 txq = &ptr_port->txq;
205 /* MAC address was updated */
206 if (ptr_port->port_dirty == 1) {
207 ret = rte_eth_macaddr_get(ptr_port->idx_port,
208 &ptr_port->mac_addr);
210 rte_spinlock_unlock(&ptr_port->lock);
211 printf("Failed to get MAC address (port %u): %s",
217 ptr_port->port_dirty = 0;
220 /* Incoming frames */
221 cnt_recv_frames = rte_eth_rx_burst(
222 ptr_port->idx_port, 0,
223 &txq->buf_frames[txq->cnt_unsent],
224 RTE_DIM(txq->buf_frames) - txq->cnt_unsent
226 if (cnt_recv_frames > 0) {
228 idx_frame < cnt_recv_frames;
230 ptr_frame = txq->buf_frames[
231 idx_frame + txq->cnt_unsent];
232 process_frame(ptr_port, ptr_frame);
234 txq->cnt_unsent += cnt_recv_frames;
237 /* Outgoing frames */
238 if (txq->cnt_unsent > 0) {
239 cnt_sent = rte_eth_tx_burst(
240 ptr_port->idx_port, 0,
244 /* Shuffle up unsent frame pointers */
245 for (idx_frame = cnt_sent;
246 idx_frame < txq->cnt_unsent;
248 txq->buf_frames[idx_frame - cnt_sent] =
249 txq->buf_frames[idx_frame];
250 txq->cnt_unsent -= cnt_sent;
252 rte_spinlock_unlock(&ptr_port->lock);
253 } /* end for( idx_port ) */
259 static void close_ports(void)
264 for (portid = 0; portid < app_cfg.cnt_ports; portid++) {
265 printf("Closing port %d...", portid);
266 ret = rte_eth_dev_stop(portid);
268 rte_exit(EXIT_FAILURE, "rte_eth_dev_stop: err=%s, port=%u\n",
269 strerror(-ret), portid);
270 rte_eth_dev_close(portid);
275 int main(int argc, char **argv)
281 /* Init runtime environment */
282 cnt_args_parsed = rte_eal_init(argc, argv);
283 if (cnt_args_parsed < 0)
284 rte_exit(EXIT_FAILURE, "rte_eal_init(): Failed");
286 cnt_ports = rte_eth_dev_count_avail();
287 printf("Number of NICs: %i\n", cnt_ports);
289 rte_exit(EXIT_FAILURE, "No available NIC ports!\n");
290 if (cnt_ports > MAX_PORTS) {
291 printf("Info: Using only %i of %i ports\n",
294 cnt_ports = MAX_PORTS;
297 setup_ports(&app_cfg, cnt_ports);
299 app_cfg.exit_now = 0;
300 app_cfg.cnt_ports = cnt_ports;
302 if (rte_lcore_count() < 2)
303 rte_exit(EXIT_FAILURE, "No available worker core!\n");
305 /* Assume there is an available worker.. */
306 id_core = rte_lcore_id();
307 id_core = rte_get_next_lcore(id_core, 1, 1);
308 rte_eal_remote_launch(worker_main, NULL, id_core);
312 app_cfg.exit_now = 1;
313 RTE_LCORE_FOREACH_WORKER(id_core) {
314 if (rte_eal_wait_lcore(id_core) < 0)
320 /* clean up the EAL */