examples/flow_filtering/main.c

   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 ether_addr *eth_addr)
  49 {
  50         char buf[ETHER_ADDR_FMT_SIZE];
  51         ether_format_addr(buf, 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 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 ether_hdr *);
  75                                         print_ether_addr("src=",
  76                                                         &eth_hdr->s_addr);
  77                                         print_ether_addr(" - dst=",
  78                                                         &eth_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                         .offloads = DEV_RX_OFFLOAD_CRC_STRIP,
 125                 },
 126                 .txmode = {
 127                         .offloads =
 128                                 DEV_TX_OFFLOAD_VLAN_INSERT |
 129                                 DEV_TX_OFFLOAD_IPV4_CKSUM  |
 130                                 DEV_TX_OFFLOAD_UDP_CKSUM   |
 131                                 DEV_TX_OFFLOAD_TCP_CKSUM   |
 132                                 DEV_TX_OFFLOAD_SCTP_CKSUM  |
 133                                 DEV_TX_OFFLOAD_TCP_TSO,
 134                 },
 135                 /*
 136                  * Initialize fdir_conf of rte_eth_conf.
 137                  * Fdir is used in flow filtering for I40e,
 138                  * so rte_flow rules involve some fdir
 139                  * configurations. In long term it's better
 140                  * that drivers don't require any fdir
 141                  * configuration for rte_flow, but we need to
 142                  * get this workaround so that sample app can
 143                  * run on I40e.
 144                  */
 145                 .fdir_conf = {
 146                         .mode = RTE_FDIR_MODE_PERFECT,
 147                         .pballoc = RTE_FDIR_PBALLOC_64K,
 148                         .status = RTE_FDIR_REPORT_STATUS,
 149                         .drop_queue = 127,
 150                 },
 151         };
 152         struct rte_eth_txconf txq_conf;
 153         struct rte_eth_rxconf rxq_conf;
 154         struct rte_eth_dev_info dev_info;
 155
 156         printf(":: initializing port: %d\n", port_id);
 157         ret = rte_eth_dev_configure(port_id,
 158                                 nr_queues, nr_queues, &port_conf);
 159         if (ret < 0) {
 160                 rte_exit(EXIT_FAILURE,
 161                         ":: cannot configure device: err=%d, port=%u\n",
 162                         ret, port_id);
 163         }
 164
 165         rte_eth_dev_info_get(port_id, &dev_info);
 166         rxq_conf = dev_info.default_rxconf;
 167         rxq_conf.offloads = port_conf.rxmode.offloads;
 168         /* only set Rx queues: something we care only so far */
 169         for (i = 0; i < nr_queues; i++) {
 170                 ret = rte_eth_rx_queue_setup(port_id, i, 512,
 171                                      rte_eth_dev_socket_id(port_id),
 172                                      &rxq_conf,
 173                                      mbuf_pool);
 174                 if (ret < 0) {
 175                         rte_exit(EXIT_FAILURE,
 176                                 ":: Rx queue setup failed: err=%d, port=%u\n",
 177                                 ret, port_id);
 178                 }
 179         }
 180
 181         txq_conf = dev_info.default_txconf;
 182         txq_conf.offloads = port_conf.txmode.offloads;
 183
 184         for (i = 0; i < nr_queues; i++) {
 185                 ret = rte_eth_tx_queue_setup(port_id, i, 512,
 186                                 rte_eth_dev_socket_id(port_id),
 187                                 &txq_conf);
 188                 if (ret < 0) {
 189                         rte_exit(EXIT_FAILURE,
 190                                 ":: Tx queue setup failed: err=%d, port=%u\n",
 191                                 ret, port_id);
 192                 }
 193         }
 194
 195         rte_eth_promiscuous_enable(port_id);
 196         ret = rte_eth_dev_start(port_id);
 197         if (ret < 0) {
 198                 rte_exit(EXIT_FAILURE,
 199                         "rte_eth_dev_start:err=%d, port=%u\n",
 200                         ret, port_id);
 201         }
 202
 203         assert_link_status();
 204
 205         printf(":: initializing port: %d done\n", port_id);
 206 }
 207
 208 static void
 209 signal_handler(int signum)
 210 {
 211         if (signum == SIGINT || signum == SIGTERM) {
 212                 printf("\n\nSignal %d received, preparing to exit...\n",
 213                                 signum);
 214                 force_quit = true;
 215         }
 216 }
 217
 218 int
 219 main(int argc, char **argv)
 220 {
 221         int ret;
 222         uint16_t nr_ports;
 223         struct rte_flow_error error;
 224
 225         ret = rte_eal_init(argc, argv);
 226         if (ret < 0)
 227                 rte_exit(EXIT_FAILURE, ":: invalid EAL arguments\n");
 228
 229         force_quit = false;
 230         signal(SIGINT, signal_handler);
 231         signal(SIGTERM, signal_handler);
 232
 233         nr_ports = rte_eth_dev_count_avail();
 234         if (nr_ports == 0)
 235                 rte_exit(EXIT_FAILURE, ":: no Ethernet ports found\n");
 236         port_id = 0;
 237         if (nr_ports != 1) {
 238                 printf(":: warn: %d ports detected, but we use only one: port %u\n",
 239                         nr_ports, port_id);
 240         }
 241         mbuf_pool = rte_pktmbuf_pool_create("mbuf_pool", 4096, 128, 0,
 242                                             RTE_MBUF_DEFAULT_BUF_SIZE,
 243                                             rte_socket_id());
 244         if (mbuf_pool == NULL)
 245                 rte_exit(EXIT_FAILURE, "Cannot init mbuf pool\n");
 246
 247         init_port();
 248
 249         /* create flow for send packet with */
 250         flow = generate_ipv4_flow(port_id, selected_queue,
 251                                 SRC_IP, EMPTY_MASK,
 252                                 DEST_IP, FULL_MASK, &error);
 253         if (!flow) {
 254                 printf("Flow can't be created %d message: %s\n",
 255                         error.type,
 256                         error.message ? error.message : "(no stated reason)");
 257                 rte_exit(EXIT_FAILURE, "error in creating flow");
 258         }
 259
 260         main_loop();
 261
 262         return 0;
 263 }