examples/flow_classify/flow_classify.c

   1 /* SPDX-License-Identifier: BSD-3-Clause
   2  * Copyright(c) 2017 Intel Corporation
   3  */
   4
   5 #include <stdint.h>
   6 #include <inttypes.h>
   7 #include <getopt.h>
   8
   9 #include <rte_eal.h>
  10 #include <rte_ethdev.h>
  11 #include <rte_cycles.h>
  12 #include <rte_lcore.h>
  13 #include <rte_mbuf.h>
  14 #include <rte_flow.h>
  15 #include <rte_flow_classify.h>
  16 #include <rte_table_acl.h>
  17
  18 #define RX_RING_SIZE 128
  19 #define TX_RING_SIZE 512
  20
  21 #define NUM_MBUFS 8191
  22 #define MBUF_CACHE_SIZE 250
  23 #define BURST_SIZE 32
  24
  25 #define MAX_NUM_CLASSIFY 30
  26 #define FLOW_CLASSIFY_MAX_RULE_NUM 91
  27 #define FLOW_CLASSIFY_MAX_PRIORITY 8
  28 #define FLOW_CLASSIFIER_NAME_SIZE 64
  29
  30 #define COMMENT_LEAD_CHAR       ('#')
  31 #define OPTION_RULE_IPV4        "rule_ipv4"
  32 #define RTE_LOGTYPE_FLOW_CLASSIFY       RTE_LOGTYPE_USER3
  33 #define flow_classify_log(format, ...) \
  34                 RTE_LOG(ERR, FLOW_CLASSIFY, format, ##__VA_ARGS__)
  35
  36 #define uint32_t_to_char(ip, a, b, c, d) do {\
  37                 *a = (unsigned char)(ip >> 24 & 0xff);\
  38                 *b = (unsigned char)(ip >> 16 & 0xff);\
  39                 *c = (unsigned char)(ip >> 8 & 0xff);\
  40                 *d = (unsigned char)(ip & 0xff);\
  41         } while (0)
  42
  43 enum {
  44         CB_FLD_SRC_ADDR,
  45         CB_FLD_DST_ADDR,
  46         CB_FLD_SRC_PORT,
  47         CB_FLD_SRC_PORT_DLM,
  48         CB_FLD_SRC_PORT_MASK,
  49         CB_FLD_DST_PORT,
  50         CB_FLD_DST_PORT_DLM,
  51         CB_FLD_DST_PORT_MASK,
  52         CB_FLD_PROTO,
  53         CB_FLD_PRIORITY,
  54         CB_FLD_NUM,
  55 };
  56
  57 static struct{
  58         const char *rule_ipv4_name;
  59 } parm_config;
  60 const char cb_port_delim[] = ":";
  61
  62 static const struct rte_eth_conf port_conf_default = {
  63         .rxmode = { .max_rx_pkt_len = ETHER_MAX_LEN }
  64 };
  65
  66 struct flow_classifier {
  67         struct rte_flow_classifier *cls;
  68         uint32_t table_id[RTE_FLOW_CLASSIFY_TABLE_MAX];
  69 };
  70
  71 struct flow_classifier_acl {
  72         struct flow_classifier cls;
  73 } __rte_cache_aligned;
  74
  75 /* ACL field definitions for IPv4 5 tuple rule */
  76
  77 enum {
  78         PROTO_FIELD_IPV4,
  79         SRC_FIELD_IPV4,
  80         DST_FIELD_IPV4,
  81         SRCP_FIELD_IPV4,
  82         DSTP_FIELD_IPV4,
  83         NUM_FIELDS_IPV4
  84 };
  85
  86 enum {
  87         PROTO_INPUT_IPV4,
  88         SRC_INPUT_IPV4,
  89         DST_INPUT_IPV4,
  90         SRCP_DESTP_INPUT_IPV4
  91 };
  92
  93 static struct rte_acl_field_def ipv4_defs[NUM_FIELDS_IPV4] = {
  94         /* first input field - always one byte long. */
  95         {
  96                 .type = RTE_ACL_FIELD_TYPE_BITMASK,
  97                 .size = sizeof(uint8_t),
  98                 .field_index = PROTO_FIELD_IPV4,
  99                 .input_index = PROTO_INPUT_IPV4,
 100                 .offset = sizeof(struct ether_hdr) +
 101                         offsetof(struct ipv4_hdr, next_proto_id),
 102         },
 103         /* next input field (IPv4 source address) - 4 consecutive bytes. */
 104         {
 105                 /* rte_flow uses a bit mask for IPv4 addresses */
 106                 .type = RTE_ACL_FIELD_TYPE_BITMASK,
 107                 .size = sizeof(uint32_t),
 108                 .field_index = SRC_FIELD_IPV4,
 109                 .input_index = SRC_INPUT_IPV4,
 110                 .offset = sizeof(struct ether_hdr) +
 111                         offsetof(struct ipv4_hdr, src_addr),
 112         },
 113         /* next input field (IPv4 destination address) - 4 consecutive bytes. */
 114         {
 115                 /* rte_flow uses a bit mask for IPv4 addresses */
 116                 .type = RTE_ACL_FIELD_TYPE_BITMASK,
 117                 .size = sizeof(uint32_t),
 118                 .field_index = DST_FIELD_IPV4,
 119                 .input_index = DST_INPUT_IPV4,
 120                 .offset = sizeof(struct ether_hdr) +
 121                         offsetof(struct ipv4_hdr, dst_addr),
 122         },
 123         /*
 124          * Next 2 fields (src & dst ports) form 4 consecutive bytes.
 125          * They share the same input index.
 126          */
 127         {
 128                 /* rte_flow uses a bit mask for protocol ports */
 129                 .type = RTE_ACL_FIELD_TYPE_BITMASK,
 130                 .size = sizeof(uint16_t),
 131                 .field_index = SRCP_FIELD_IPV4,
 132                 .input_index = SRCP_DESTP_INPUT_IPV4,
 133                 .offset = sizeof(struct ether_hdr) +
 134                         sizeof(struct ipv4_hdr) +
 135                         offsetof(struct tcp_hdr, src_port),
 136         },
 137         {
 138                 /* rte_flow uses a bit mask for protocol ports */
 139                 .type = RTE_ACL_FIELD_TYPE_BITMASK,
 140                 .size = sizeof(uint16_t),
 141                 .field_index = DSTP_FIELD_IPV4,
 142                 .input_index = SRCP_DESTP_INPUT_IPV4,
 143                 .offset = sizeof(struct ether_hdr) +
 144                         sizeof(struct ipv4_hdr) +
 145                         offsetof(struct tcp_hdr, dst_port),
 146         },
 147 };
 148
 149 /* flow classify data */
 150 static int num_classify_rules;
 151 static struct rte_flow_classify_rule *rules[MAX_NUM_CLASSIFY];
 152 static struct rte_flow_classify_ipv4_5tuple_stats ntuple_stats;
 153 static struct rte_flow_classify_stats classify_stats = {
 154                 .stats = (void **)&ntuple_stats
 155 };
 156
 157 /* parameters for rte_flow_classify_validate and
 158  * rte_flow_classify_table_entry_add functions
 159  */
 160
 161 static struct rte_flow_item  eth_item = { RTE_FLOW_ITEM_TYPE_ETH,
 162         0, 0, 0 };
 163 static struct rte_flow_item  end_item = { RTE_FLOW_ITEM_TYPE_END,
 164         0, 0, 0 };
 165
 166 /* sample actions:
 167  * "actions count / end"
 168  */
 169 static struct rte_flow_action count_action = { RTE_FLOW_ACTION_TYPE_COUNT, 0};
 170 static struct rte_flow_action end_action = { RTE_FLOW_ACTION_TYPE_END, 0};
 171 static struct rte_flow_action actions[2];
 172
 173 /* sample attributes */
 174 static struct rte_flow_attr attr;
 175
 176 /* flow_classify.c: * Based on DPDK skeleton forwarding example. */
 177
 178 /*
 179  * Initializes a given port using global settings and with the RX buffers
 180  * coming from the mbuf_pool passed as a parameter.
 181  */
 182 static inline int
 183 port_init(uint8_t port, struct rte_mempool *mbuf_pool)
 184 {
 185         struct rte_eth_conf port_conf = port_conf_default;
 186         struct ether_addr addr;
 187         const uint16_t rx_rings = 1, tx_rings = 1;
 188         int retval;
 189         uint16_t q;
 190
 191         if (port >= rte_eth_dev_count())
 192                 return -1;
 193
 194         /* Configure the Ethernet device. */
 195         retval = rte_eth_dev_configure(port, rx_rings, tx_rings, &port_conf);
 196         if (retval != 0)
 197                 return retval;
 198
 199         /* Allocate and set up 1 RX queue per Ethernet port. */
 200         for (q = 0; q < rx_rings; q++) {
 201                 retval = rte_eth_rx_queue_setup(port, q, RX_RING_SIZE,
 202                                 rte_eth_dev_socket_id(port), NULL, mbuf_pool);
 203                 if (retval < 0)
 204                         return retval;
 205         }
 206
 207         /* Allocate and set up 1 TX queue per Ethernet port. */
 208         for (q = 0; q < tx_rings; q++) {
 209                 retval = rte_eth_tx_queue_setup(port, q, TX_RING_SIZE,
 210                                 rte_eth_dev_socket_id(port), NULL);
 211                 if (retval < 0)
 212                         return retval;
 213         }
 214
 215         /* Start the Ethernet port. */
 216         retval = rte_eth_dev_start(port);
 217         if (retval < 0)
 218                 return retval;
 219
 220         /* Display the port MAC address. */
 221         rte_eth_macaddr_get(port, &addr);
 222         printf("Port %u MAC: %02" PRIx8 " %02" PRIx8 " %02" PRIx8
 223                            " %02" PRIx8 " %02" PRIx8 " %02" PRIx8 "\n",
 224                         port,
 225                         addr.addr_bytes[0], addr.addr_bytes[1],
 226                         addr.addr_bytes[2], addr.addr_bytes[3],
 227                         addr.addr_bytes[4], addr.addr_bytes[5]);
 228
 229         /* Enable RX in promiscuous mode for the Ethernet device. */
 230         rte_eth_promiscuous_enable(port);
 231
 232         return 0;
 233 }
 234
 235 /*
 236  * The lcore main. This is the main thread that does the work, reading from
 237  * an input port classifying the packets and writing to an output port.
 238  */
 239 static __attribute__((noreturn)) void
 240 lcore_main(struct flow_classifier *cls_app)
 241 {
 242         const uint8_t nb_ports = rte_eth_dev_count();
 243         uint8_t port;
 244         int ret;
 245         int i = 0;
 246
 247         ret = rte_flow_classify_table_entry_delete(cls_app->cls,
 248                         cls_app->table_id[0], rules[7]);
 249         if (ret)
 250                 printf("table_entry_delete failed [7] %d\n\n", ret);
 251         else
 252                 printf("table_entry_delete succeeded [7]\n\n");
 253
 254         /*
 255          * Check that the port is on the same NUMA node as the polling thread
 256          * for best performance.
 257          */
 258         for (port = 0; port < nb_ports; port++)
 259                 if (rte_eth_dev_socket_id(port) > 0 &&
 260                         rte_eth_dev_socket_id(port) != (int)rte_socket_id()) {
 261                         printf("\n\n");
 262                         printf("WARNING: port %u is on remote NUMA node\n",
 263                                port);
 264                         printf("to polling thread.\n");
 265                         printf("Performance will not be optimal.\n");
 266
 267                         printf("\nCore %u forwarding packets. ",
 268                                rte_lcore_id());
 269                         printf("[Ctrl+C to quit]\n");
 270                 }
 271         /* Run until the application is quit or killed. */
 272         for (;;) {
 273                 /*
 274                  * Receive packets on a port, classify them and forward them
 275                  * on the paired port.
 276                  * The mapping is 0 -> 1, 1 -> 0, 2 -> 3, 3 -> 2, etc.
 277                  */
 278                 for (port = 0; port < nb_ports; port++) {
 279                         /* Get burst of RX packets, from first port of pair. */
 280                         struct rte_mbuf *bufs[BURST_SIZE];
 281                         const uint16_t nb_rx = rte_eth_rx_burst(port, 0,
 282                                         bufs, BURST_SIZE);
 283
 284                         if (unlikely(nb_rx == 0))
 285                                 continue;
 286
 287                         for (i = 0; i < MAX_NUM_CLASSIFY; i++) {
 288                                 if (rules[i]) {
 289                                         ret = rte_flow_classifier_query(
 290                                                 cls_app->cls,
 291                                                 cls_app->table_id[0],
 292                                                 bufs, nb_rx, rules[i],
 293                                                 &classify_stats);
 294                                         if (ret)
 295                                                 printf(
 296                                                         "rule [%d] query failed ret [%d]\n\n",
 297                                                         i, ret);
 298                                         else {
 299                                                 printf(
 300                                                 "rule[%d] count=%"PRIu64"\n",
 301                                                 i, ntuple_stats.counter1);
 302
 303                                                 printf("proto = %d\n",
 304                                                 ntuple_stats.ipv4_5tuple.proto);
 305                                         }
 306                                 }
 307                         }
 308
 309                         /* Send burst of TX packets, to second port of pair. */
 310                         const uint16_t nb_tx = rte_eth_tx_burst(port ^ 1, 0,
 311                                         bufs, nb_rx);
 312
 313                         /* Free any unsent packets. */
 314                         if (unlikely(nb_tx < nb_rx)) {
 315                                 uint16_t buf;
 316
 317                                 for (buf = nb_tx; buf < nb_rx; buf++)
 318                                         rte_pktmbuf_free(bufs[buf]);
 319                         }
 320                 }
 321         }
 322 }
 323
 324 /*
 325  * Parse IPv4 5 tuple rules file, ipv4_rules_file.txt.
 326  * Expected format:
 327  * <src_ipv4_addr>'/'<masklen> <space> \
 328  * <dst_ipv4_addr>'/'<masklen> <space> \
 329  * <src_port> <space> ":" <src_port_mask> <space> \
 330  * <dst_port> <space> ":" <dst_port_mask> <space> \
 331  * <proto>'/'<proto_mask> <space> \
 332  * <priority>
 333  */
 334
 335 static int
 336 get_cb_field(char **in, uint32_t *fd, int base, unsigned long lim,
 337                 char dlm)
 338 {
 339         unsigned long val;
 340         char *end;
 341
 342         errno = 0;
 343         val = strtoul(*in, &end, base);
 344         if (errno != 0 || end[0] != dlm || val > lim)
 345                 return -EINVAL;
 346         *fd = (uint32_t)val;
 347         *in = end + 1;
 348         return 0;
 349 }
 350
 351 static int
 352 parse_ipv4_net(char *in, uint32_t *addr, uint32_t *mask_len)
 353 {
 354         uint32_t a, b, c, d, m;
 355
 356         if (get_cb_field(&in, &a, 0, UINT8_MAX, '.'))
 357                 return -EINVAL;
 358         if (get_cb_field(&in, &b, 0, UINT8_MAX, '.'))
 359                 return -EINVAL;
 360         if (get_cb_field(&in, &c, 0, UINT8_MAX, '.'))
 361                 return -EINVAL;
 362         if (get_cb_field(&in, &d, 0, UINT8_MAX, '/'))
 363                 return -EINVAL;
 364         if (get_cb_field(&in, &m, 0, sizeof(uint32_t) * CHAR_BIT, 0))
 365                 return -EINVAL;
 366
 367         addr[0] = IPv4(a, b, c, d);
 368         mask_len[0] = m;
 369         return 0;
 370 }
 371
 372 static int
 373 parse_ipv4_5tuple_rule(char *str, struct rte_eth_ntuple_filter *ntuple_filter)
 374 {
 375         int i, ret;
 376         char *s, *sp, *in[CB_FLD_NUM];
 377         static const char *dlm = " \t\n";
 378         int dim = CB_FLD_NUM;
 379         uint32_t temp;
 380
 381         s = str;
 382         for (i = 0; i != dim; i++, s = NULL) {
 383                 in[i] = strtok_r(s, dlm, &sp);
 384                 if (in[i] == NULL)
 385                         return -EINVAL;
 386         }
 387
 388         ret = parse_ipv4_net(in[CB_FLD_SRC_ADDR],
 389                         &ntuple_filter->src_ip,
 390                         &ntuple_filter->src_ip_mask);
 391         if (ret != 0) {
 392                 flow_classify_log("failed to read source address/mask: %s\n",
 393                         in[CB_FLD_SRC_ADDR]);
 394                 return ret;
 395         }
 396
 397         ret = parse_ipv4_net(in[CB_FLD_DST_ADDR],
 398                         &ntuple_filter->dst_ip,
 399                         &ntuple_filter->dst_ip_mask);
 400         if (ret != 0) {
 401                 flow_classify_log("failed to read source address/mask: %s\n",
 402                         in[CB_FLD_DST_ADDR]);
 403                 return ret;
 404         }
 405
 406         if (get_cb_field(&in[CB_FLD_SRC_PORT], &temp, 0, UINT16_MAX, 0))
 407                 return -EINVAL;
 408         ntuple_filter->src_port = (uint16_t)temp;
 409
 410         if (strncmp(in[CB_FLD_SRC_PORT_DLM], cb_port_delim,
 411                         sizeof(cb_port_delim)) != 0)
 412                 return -EINVAL;
 413
 414         if (get_cb_field(&in[CB_FLD_SRC_PORT_MASK], &temp, 0, UINT16_MAX, 0))
 415                 return -EINVAL;
 416         ntuple_filter->src_port_mask = (uint16_t)temp;
 417
 418         if (get_cb_field(&in[CB_FLD_DST_PORT], &temp, 0, UINT16_MAX, 0))
 419                 return -EINVAL;
 420         ntuple_filter->dst_port = (uint16_t)temp;
 421
 422         if (strncmp(in[CB_FLD_DST_PORT_DLM], cb_port_delim,
 423                         sizeof(cb_port_delim)) != 0)
 424                 return -EINVAL;
 425
 426         if (get_cb_field(&in[CB_FLD_DST_PORT_MASK], &temp, 0, UINT16_MAX, 0))
 427                 return -EINVAL;
 428         ntuple_filter->dst_port_mask = (uint16_t)temp;
 429
 430         if (get_cb_field(&in[CB_FLD_PROTO], &temp, 0, UINT8_MAX, '/'))
 431                 return -EINVAL;
 432         ntuple_filter->proto = (uint8_t)temp;
 433
 434         if (get_cb_field(&in[CB_FLD_PROTO], &temp, 0, UINT8_MAX, 0))
 435                 return -EINVAL;
 436         ntuple_filter->proto_mask = (uint8_t)temp;
 437
 438         if (get_cb_field(&in[CB_FLD_PRIORITY], &temp, 0, UINT16_MAX, 0))
 439                 return -EINVAL;
 440         ntuple_filter->priority = (uint16_t)temp;
 441         if (ntuple_filter->priority > FLOW_CLASSIFY_MAX_PRIORITY)
 442                 ret = -EINVAL;
 443
 444         return ret;
 445 }
 446
 447 /* Bypass comment and empty lines */
 448 static inline int
 449 is_bypass_line(char *buff)
 450 {
 451         int i = 0;
 452
 453         /* comment line */
 454         if (buff[0] == COMMENT_LEAD_CHAR)
 455                 return 1;
 456         /* empty line */
 457         while (buff[i] != '\0') {
 458                 if (!isspace(buff[i]))
 459                         return 0;
 460                 i++;
 461         }
 462         return 1;
 463 }
 464
 465 static uint32_t
 466 convert_depth_to_bitmask(uint32_t depth_val)
 467 {
 468         uint32_t bitmask = 0;
 469         int i, j;
 470
 471         for (i = depth_val, j = 0; i > 0; i--, j++)
 472                 bitmask |= (1 << (31 - j));
 473         return bitmask;
 474 }
 475
 476 static int
 477 add_classify_rule(struct rte_eth_ntuple_filter *ntuple_filter,
 478                 struct flow_classifier *cls_app)
 479 {
 480         int ret = -1;
 481         int key_found;
 482         struct rte_flow_error error;
 483         struct rte_flow_item_ipv4 ipv4_spec;
 484         struct rte_flow_item_ipv4 ipv4_mask;
 485         struct rte_flow_item ipv4_udp_item;
 486         struct rte_flow_item ipv4_tcp_item;
 487         struct rte_flow_item ipv4_sctp_item;
 488         struct rte_flow_item_udp udp_spec;
 489         struct rte_flow_item_udp udp_mask;
 490         struct rte_flow_item udp_item;
 491         struct rte_flow_item_tcp tcp_spec;
 492         struct rte_flow_item_tcp tcp_mask;
 493         struct rte_flow_item tcp_item;
 494         struct rte_flow_item_sctp sctp_spec;
 495         struct rte_flow_item_sctp sctp_mask;
 496         struct rte_flow_item sctp_item;
 497         struct rte_flow_item pattern_ipv4_5tuple[4];
 498         struct rte_flow_classify_rule *rule;
 499         uint8_t ipv4_proto;
 500
 501         if (num_classify_rules >= MAX_NUM_CLASSIFY) {
 502                 printf(
 503                         "\nINFO:  classify rule capacity %d reached\n",
 504                         num_classify_rules);
 505                 return ret;
 506         }
 507
 508         /* set up parameters for validate and add */
 509         memset(&ipv4_spec, 0, sizeof(ipv4_spec));
 510         ipv4_spec.hdr.next_proto_id = ntuple_filter->proto;
 511         ipv4_spec.hdr.src_addr = ntuple_filter->src_ip;
 512         ipv4_spec.hdr.dst_addr = ntuple_filter->dst_ip;
 513         ipv4_proto = ipv4_spec.hdr.next_proto_id;
 514
 515         memset(&ipv4_mask, 0, sizeof(ipv4_mask));
 516         ipv4_mask.hdr.next_proto_id = ntuple_filter->proto_mask;
 517         ipv4_mask.hdr.src_addr = ntuple_filter->src_ip_mask;
 518         ipv4_mask.hdr.src_addr =
 519                 convert_depth_to_bitmask(ipv4_mask.hdr.src_addr);
 520         ipv4_mask.hdr.dst_addr = ntuple_filter->dst_ip_mask;
 521         ipv4_mask.hdr.dst_addr =
 522                 convert_depth_to_bitmask(ipv4_mask.hdr.dst_addr);
 523
 524         switch (ipv4_proto) {
 525         case IPPROTO_UDP:
 526                 ipv4_udp_item.type = RTE_FLOW_ITEM_TYPE_IPV4;
 527                 ipv4_udp_item.spec = &ipv4_spec;
 528                 ipv4_udp_item.mask = &ipv4_mask;
 529                 ipv4_udp_item.last = NULL;
 530
 531                 udp_spec.hdr.src_port = ntuple_filter->src_port;
 532                 udp_spec.hdr.dst_port = ntuple_filter->dst_port;
 533                 udp_spec.hdr.dgram_len = 0;
 534                 udp_spec.hdr.dgram_cksum = 0;
 535
 536                 udp_mask.hdr.src_port = ntuple_filter->src_port_mask;
 537                 udp_mask.hdr.dst_port = ntuple_filter->dst_port_mask;
 538                 udp_mask.hdr.dgram_len = 0;
 539                 udp_mask.hdr.dgram_cksum = 0;
 540
 541                 udp_item.type = RTE_FLOW_ITEM_TYPE_UDP;
 542                 udp_item.spec = &udp_spec;
 543                 udp_item.mask = &udp_mask;
 544                 udp_item.last = NULL;
 545
 546                 attr.priority = ntuple_filter->priority;
 547                 pattern_ipv4_5tuple[1] = ipv4_udp_item;
 548                 pattern_ipv4_5tuple[2] = udp_item;
 549                 break;
 550         case IPPROTO_TCP:
 551                 ipv4_tcp_item.type = RTE_FLOW_ITEM_TYPE_IPV4;
 552                 ipv4_tcp_item.spec = &ipv4_spec;
 553                 ipv4_tcp_item.mask = &ipv4_mask;
 554                 ipv4_tcp_item.last = NULL;
 555
 556                 memset(&tcp_spec, 0, sizeof(tcp_spec));
 557                 tcp_spec.hdr.src_port = ntuple_filter->src_port;
 558                 tcp_spec.hdr.dst_port = ntuple_filter->dst_port;
 559
 560                 memset(&tcp_mask, 0, sizeof(tcp_mask));
 561                 tcp_mask.hdr.src_port = ntuple_filter->src_port_mask;
 562                 tcp_mask.hdr.dst_port = ntuple_filter->dst_port_mask;
 563
 564                 tcp_item.type = RTE_FLOW_ITEM_TYPE_TCP;
 565                 tcp_item.spec = &tcp_spec;
 566                 tcp_item.mask = &tcp_mask;
 567                 tcp_item.last = NULL;
 568
 569                 attr.priority = ntuple_filter->priority;
 570                 pattern_ipv4_5tuple[1] = ipv4_tcp_item;
 571                 pattern_ipv4_5tuple[2] = tcp_item;
 572                 break;
 573         case IPPROTO_SCTP:
 574                 ipv4_sctp_item.type = RTE_FLOW_ITEM_TYPE_IPV4;
 575                 ipv4_sctp_item.spec = &ipv4_spec;
 576                 ipv4_sctp_item.mask = &ipv4_mask;
 577                 ipv4_sctp_item.last = NULL;
 578
 579                 sctp_spec.hdr.src_port = ntuple_filter->src_port;
 580                 sctp_spec.hdr.dst_port = ntuple_filter->dst_port;
 581                 sctp_spec.hdr.cksum = 0;
 582                 sctp_spec.hdr.tag = 0;
 583
 584                 sctp_mask.hdr.src_port = ntuple_filter->src_port_mask;
 585                 sctp_mask.hdr.dst_port = ntuple_filter->dst_port_mask;
 586                 sctp_mask.hdr.cksum = 0;
 587                 sctp_mask.hdr.tag = 0;
 588
 589                 sctp_item.type = RTE_FLOW_ITEM_TYPE_SCTP;
 590                 sctp_item.spec = &sctp_spec;
 591                 sctp_item.mask = &sctp_mask;
 592                 sctp_item.last = NULL;
 593
 594                 attr.priority = ntuple_filter->priority;
 595                 pattern_ipv4_5tuple[1] = ipv4_sctp_item;
 596                 pattern_ipv4_5tuple[2] = sctp_item;
 597                 break;
 598         default:
 599                 return ret;
 600         }
 601
 602         attr.ingress = 1;
 603         pattern_ipv4_5tuple[0] = eth_item;
 604         pattern_ipv4_5tuple[3] = end_item;
 605         actions[0] = count_action;
 606         actions[1] = end_action;
 607
 608         rule = rte_flow_classify_table_entry_add(
 609                         cls_app->cls, cls_app->table_id[0], &key_found,
 610                         &attr, pattern_ipv4_5tuple, actions, &error);
 611         if (rule == NULL) {
 612                 printf("table entry add failed ipv4_proto = %u\n",
 613                         ipv4_proto);
 614                 ret = -1;
 615                 return ret;
 616         }
 617
 618         rules[num_classify_rules] = rule;
 619         num_classify_rules++;
 620         return 0;
 621 }
 622
 623 static int
 624 add_rules(const char *rule_path, struct flow_classifier *cls_app)
 625 {
 626         FILE *fh;
 627         char buff[LINE_MAX];
 628         unsigned int i = 0;
 629         unsigned int total_num = 0;
 630         struct rte_eth_ntuple_filter ntuple_filter;
 631         int ret;
 632
 633         fh = fopen(rule_path, "rb");
 634         if (fh == NULL)
 635                 rte_exit(EXIT_FAILURE, "%s: fopen %s failed\n", __func__,
 636                         rule_path);
 637
 638         ret = fseek(fh, 0, SEEK_SET);
 639         if (ret)
 640                 rte_exit(EXIT_FAILURE, "%s: fseek %d failed\n", __func__,
 641                         ret);
 642
 643         i = 0;
 644         while (fgets(buff, LINE_MAX, fh) != NULL) {
 645                 i++;
 646
 647                 if (is_bypass_line(buff))
 648                         continue;
 649
 650                 if (total_num >= FLOW_CLASSIFY_MAX_RULE_NUM - 1) {
 651                         printf("\nINFO: classify rule capacity %d reached\n",
 652                                 total_num);
 653                         break;
 654                 }
 655
 656                 if (parse_ipv4_5tuple_rule(buff, &ntuple_filter) != 0)
 657                         rte_exit(EXIT_FAILURE,
 658                                 "%s Line %u: parse rules error\n",
 659                                 rule_path, i);
 660
 661                 if (add_classify_rule(&ntuple_filter, cls_app) != 0)
 662                         rte_exit(EXIT_FAILURE, "add rule error\n");
 663
 664                 total_num++;
 665         }
 666
 667         fclose(fh);
 668         return 0;
 669 }
 670
 671 /* display usage */
 672 static void
 673 print_usage(const char *prgname)
 674 {
 675         printf("%s usage:\n", prgname);
 676         printf("[EAL options] --  --"OPTION_RULE_IPV4"=FILE: ");
 677         printf("specify the ipv4 rules file.\n");
 678         printf("Each rule occupies one line in the file.\n");
 679 }
 680
 681 /* Parse the argument given in the command line of the application */
 682 static int
 683 parse_args(int argc, char **argv)
 684 {
 685         int opt, ret;
 686         char **argvopt;
 687         int option_index;
 688         char *prgname = argv[0];
 689         static struct option lgopts[] = {
 690                 {OPTION_RULE_IPV4, 1, 0, 0},
 691                 {NULL, 0, 0, 0}
 692         };
 693
 694         argvopt = argv;
 695
 696         while ((opt = getopt_long(argc, argvopt, "",
 697                                 lgopts, &option_index)) != EOF) {
 698
 699                 switch (opt) {
 700                 /* long options */
 701                 case 0:
 702                         if (!strncmp(lgopts[option_index].name,
 703                                         OPTION_RULE_IPV4,
 704                                         sizeof(OPTION_RULE_IPV4)))
 705                                 parm_config.rule_ipv4_name = optarg;
 706                         break;
 707                 default:
 708                         print_usage(prgname);
 709                         return -1;
 710                 }
 711         }
 712
 713         if (optind >= 0)
 714                 argv[optind-1] = prgname;
 715
 716         ret = optind-1;
 717         optind = 1; /* reset getopt lib */
 718         return ret;
 719 }
 720
 721 /*
 722  * The main function, which does initialization and calls the lcore_main
 723  * function.
 724  */
 725 int
 726 main(int argc, char *argv[])
 727 {
 728         struct rte_mempool *mbuf_pool;
 729         uint8_t nb_ports;
 730         uint8_t portid;
 731         int ret;
 732         int socket_id;
 733         struct rte_table_acl_params table_acl_params;
 734         struct rte_flow_classify_table_params cls_table_params;
 735         struct flow_classifier *cls_app;
 736         struct rte_flow_classifier_params cls_params;
 737         uint32_t size;
 738
 739         /* Initialize the Environment Abstraction Layer (EAL). */
 740         ret = rte_eal_init(argc, argv);
 741         if (ret < 0)
 742                 rte_exit(EXIT_FAILURE, "Error with EAL initialization\n");
 743
 744         argc -= ret;
 745         argv += ret;
 746
 747         /* parse application arguments (after the EAL ones) */
 748         ret = parse_args(argc, argv);
 749         if (ret < 0)
 750                 rte_exit(EXIT_FAILURE, "Invalid flow_classify parameters\n");
 751
 752         /* Check that there is an even number of ports to send/receive on. */
 753         nb_ports = rte_eth_dev_count();
 754         if (nb_ports < 2 || (nb_ports & 1))
 755                 rte_exit(EXIT_FAILURE, "Error: number of ports must be even\n");
 756
 757         /* Creates a new mempool in memory to hold the mbufs. */
 758         mbuf_pool = rte_pktmbuf_pool_create("MBUF_POOL", NUM_MBUFS * nb_ports,
 759                 MBUF_CACHE_SIZE, 0, RTE_MBUF_DEFAULT_BUF_SIZE, rte_socket_id());
 760
 761         if (mbuf_pool == NULL)
 762                 rte_exit(EXIT_FAILURE, "Cannot create mbuf pool\n");
 763
 764         /* Initialize all ports. */
 765         for (portid = 0; portid < nb_ports; portid++)
 766                 if (port_init(portid, mbuf_pool) != 0)
 767                         rte_exit(EXIT_FAILURE, "Cannot init port %"PRIu8 "\n",
 768                                         portid);
 769
 770         if (rte_lcore_count() > 1)
 771                 printf("\nWARNING: Too many lcores enabled. Only 1 used.\n");
 772
 773         socket_id = rte_eth_dev_socket_id(0);
 774
 775         /* Memory allocation */
 776         size = RTE_CACHE_LINE_ROUNDUP(sizeof(struct flow_classifier_acl));
 777         cls_app = rte_zmalloc(NULL, size, RTE_CACHE_LINE_SIZE);
 778         if (cls_app == NULL)
 779                 rte_exit(EXIT_FAILURE, "Cannot allocate classifier memory\n");
 780
 781         cls_params.name = "flow_classifier";
 782         cls_params.socket_id = socket_id;
 783         cls_params.type = RTE_FLOW_CLASSIFY_TABLE_TYPE_ACL;
 784
 785         cls_app->cls = rte_flow_classifier_create(&cls_params);
 786         if (cls_app->cls == NULL) {
 787                 rte_free(cls_app);
 788                 rte_exit(EXIT_FAILURE, "Cannot create classifier\n");
 789         }
 790
 791         /* initialise ACL table params */
 792         table_acl_params.name = "table_acl_ipv4_5tuple";
 793         table_acl_params.n_rules = FLOW_CLASSIFY_MAX_RULE_NUM;
 794         table_acl_params.n_rule_fields = RTE_DIM(ipv4_defs);
 795         memcpy(table_acl_params.field_format, ipv4_defs, sizeof(ipv4_defs));
 796
 797         /* initialise table create params */
 798         cls_table_params.ops = &rte_table_acl_ops,
 799         cls_table_params.arg_create = &table_acl_params,
 800
 801         ret = rte_flow_classify_table_create(cls_app->cls, &cls_table_params,
 802                         &cls_app->table_id[0]);
 803         if (ret) {
 804                 rte_flow_classifier_free(cls_app->cls);
 805                 rte_free(cls_app);
 806                 rte_exit(EXIT_FAILURE, "Failed to create classifier table\n");
 807         }
 808
 809         /* read file of IPv4 5 tuple rules and initialize parameters
 810          * for rte_flow_classify_validate and rte_flow_classify_table_entry_add
 811          * API's.
 812          */
 813         if (add_rules(parm_config.rule_ipv4_name, cls_app)) {
 814                 rte_flow_classifier_free(cls_app->cls);
 815                 rte_free(cls_app);
 816                 rte_exit(EXIT_FAILURE, "Failed to add rules\n");
 817         }
 818
 819         /* Call lcore_main on the master core only. */
 820         lcore_main(cls_app);
 821
 822         return 0;
 823 }