examples/ioat/ioatfwd.c

   1 /* SPDX-License-Identifier: BSD-3-Clause
   2  * Copyright(c) 2019 Intel Corporation
   3  */
   4
   5 #include <stdint.h>
   6 #include <getopt.h>
   7 #include <signal.h>
   8 #include <stdbool.h>
   9 #include <unistd.h>
  10
  11 #include <rte_malloc.h>
  12 #include <rte_ethdev.h>
  13 #include <rte_rawdev.h>
  14 #include <rte_ioat_rawdev.h>
  15
  16 /* size of ring used for software copying between rx and tx. */
  17 #define RTE_LOGTYPE_IOAT RTE_LOGTYPE_USER1
  18 #define MAX_PKT_BURST 32
  19 #define MEMPOOL_CACHE_SIZE 512
  20 #define MIN_POOL_SIZE 65536U
  21 #define CMD_LINE_OPT_MAC_UPDATING "mac-updating"
  22 #define CMD_LINE_OPT_NO_MAC_UPDATING "no-mac-updating"
  23 #define CMD_LINE_OPT_PORTMASK "portmask"
  24 #define CMD_LINE_OPT_NB_QUEUE "nb-queue"
  25 #define CMD_LINE_OPT_COPY_TYPE "copy-type"
  26 #define CMD_LINE_OPT_RING_SIZE "ring-size"
  27
  28 /* configurable number of RX/TX ring descriptors */
  29 #define RX_DEFAULT_RINGSIZE 1024
  30 #define TX_DEFAULT_RINGSIZE 1024
  31
  32 /* max number of RX queues per port */
  33 #define MAX_RX_QUEUES_COUNT 8
  34
  35 struct rxtx_port_config {
  36         /* common config */
  37         uint16_t rxtx_port;
  38         uint16_t nb_queues;
  39         /* for software copy mode */
  40         struct rte_ring *rx_to_tx_ring;
  41         /* for IOAT rawdev copy mode */
  42         uint16_t ioat_ids[MAX_RX_QUEUES_COUNT];
  43 };
  44
  45 struct rxtx_transmission_config {
  46         struct rxtx_port_config ports[RTE_MAX_ETHPORTS];
  47         uint16_t nb_ports;
  48         uint16_t nb_lcores;
  49 };
  50
  51 typedef enum copy_mode_t {
  52 #define COPY_MODE_SW "sw"
  53         COPY_MODE_SW_NUM,
  54 #define COPY_MODE_IOAT "hw"
  55         COPY_MODE_IOAT_NUM,
  56         COPY_MODE_INVALID_NUM,
  57         COPY_MODE_SIZE_NUM = COPY_MODE_INVALID_NUM
  58 } copy_mode_t;
  59
  60 /* mask of enabled ports */
  61 static uint32_t ioat_enabled_port_mask;
  62
  63 /* number of RX queues per port */
  64 static uint16_t nb_queues = 1;
  65
  66 /* MAC updating enabled by default. */
  67 static int mac_updating = 1;
  68
  69 /* hardare copy mode enabled by default. */
  70 static copy_mode_t copy_mode = COPY_MODE_IOAT_NUM;
  71
  72 /* size of IOAT rawdev ring for hardware copy mode or
  73  * rte_ring for software copy mode
  74  */
  75 static unsigned short ring_size = 2048;
  76
  77 /* global transmission config */
  78 struct rxtx_transmission_config cfg;
  79
  80 /* configurable number of RX/TX ring descriptors */
  81 static uint16_t nb_rxd = RX_DEFAULT_RINGSIZE;
  82 static uint16_t nb_txd = TX_DEFAULT_RINGSIZE;
  83
  84 static volatile bool force_quit;
  85
  86 /* ethernet addresses of ports */
  87 static struct rte_ether_addr ioat_ports_eth_addr[RTE_MAX_ETHPORTS];
  88
  89 static struct rte_eth_dev_tx_buffer *tx_buffer[RTE_MAX_ETHPORTS];
  90 struct rte_mempool *ioat_pktmbuf_pool;
  91
  92 static void
  93 update_mac_addrs(struct rte_mbuf *m, uint32_t dest_portid)
  94 {
  95         struct rte_ether_hdr *eth;
  96         void *tmp;
  97
  98         eth = rte_pktmbuf_mtod(m, struct rte_ether_hdr *);
  99
 100         /* 02:00:00:00:00:xx - overwriting 2 bytes of source address but
 101          * it's acceptable cause it gets overwritten by rte_ether_addr_copy
 102          */
 103         tmp = &eth->d_addr.addr_bytes[0];
 104         *((uint64_t *)tmp) = 0x000000000002 + ((uint64_t)dest_portid << 40);
 105
 106         /* src addr */
 107         rte_ether_addr_copy(&ioat_ports_eth_addr[dest_portid], &eth->s_addr);
 108 }
 109
 110 static inline void
 111 pktmbuf_sw_copy(struct rte_mbuf *src, struct rte_mbuf *dst)
 112 {
 113         /* Copy packet metadata */
 114         rte_memcpy(&dst->rearm_data,
 115                 &src->rearm_data,
 116                 offsetof(struct rte_mbuf, cacheline1)
 117                 - offsetof(struct rte_mbuf, rearm_data));
 118
 119         /* Copy packet data */
 120         rte_memcpy(rte_pktmbuf_mtod(dst, char *),
 121                 rte_pktmbuf_mtod(src, char *), src->data_len);
 122 }
 123
 124 static uint32_t
 125 ioat_enqueue_packets(struct rte_mbuf **pkts,
 126         uint32_t nb_rx, uint16_t dev_id)
 127 {
 128         int ret;
 129         uint32_t i;
 130         struct rte_mbuf *pkts_copy[MAX_PKT_BURST];
 131
 132         const uint64_t addr_offset = RTE_PTR_DIFF(pkts[0]->buf_addr,
 133                 &pkts[0]->rearm_data);
 134
 135         ret = rte_mempool_get_bulk(ioat_pktmbuf_pool,
 136                 (void *)pkts_copy, nb_rx);
 137
 138         if (unlikely(ret < 0))
 139                 rte_exit(EXIT_FAILURE, "Unable to allocate memory.\n");
 140
 141         for (i = 0; i < nb_rx; i++) {
 142                 /* Perform data copy */
 143                 ret = rte_ioat_enqueue_copy(dev_id,
 144                         pkts[i]->buf_iova
 145                         - addr_offset,
 146                         pkts_copy[i]->buf_iova
 147                         - addr_offset,
 148                         rte_pktmbuf_data_len(pkts[i])
 149                         + addr_offset,
 150                         (uintptr_t)pkts[i],
 151                         (uintptr_t)pkts_copy[i],
 152                         0 /* nofence */);
 153
 154                 if (ret != 1)
 155                         break;
 156         }
 157
 158         ret = i;
 159         /* Free any not enqueued packets. */
 160         rte_mempool_put_bulk(ioat_pktmbuf_pool, (void *)&pkts[i], nb_rx - i);
 161         rte_mempool_put_bulk(ioat_pktmbuf_pool, (void *)&pkts_copy[i],
 162                 nb_rx - i);
 163
 164         return ret;
 165 }
 166
 167 /* Receive packets on one port and enqueue to IOAT rawdev or rte_ring. */
 168 static void
 169 ioat_rx_port(struct rxtx_port_config *rx_config)
 170 {
 171         uint32_t nb_rx, nb_enq, i, j;
 172         struct rte_mbuf *pkts_burst[MAX_PKT_BURST];
 173
 174         for (i = 0; i < rx_config->nb_queues; i++) {
 175
 176                 nb_rx = rte_eth_rx_burst(rx_config->rxtx_port, i,
 177                         pkts_burst, MAX_PKT_BURST);
 178
 179                 if (nb_rx == 0)
 180                         continue;
 181
 182                 if (copy_mode == COPY_MODE_IOAT_NUM) {
 183                         /* Perform packet hardware copy */
 184                         nb_enq = ioat_enqueue_packets(pkts_burst,
 185                                 nb_rx, rx_config->ioat_ids[i]);
 186                         if (nb_enq > 0)
 187                                 rte_ioat_do_copies(rx_config->ioat_ids[i]);
 188                 } else {
 189                         /* Perform packet software copy, free source packets */
 190                         int ret;
 191                         struct rte_mbuf *pkts_burst_copy[MAX_PKT_BURST];
 192
 193                         ret = rte_mempool_get_bulk(ioat_pktmbuf_pool,
 194                                 (void *)pkts_burst_copy, nb_rx);
 195
 196                         if (unlikely(ret < 0))
 197                                 rte_exit(EXIT_FAILURE,
 198                                         "Unable to allocate memory.\n");
 199
 200                         for (j = 0; j < nb_rx; j++)
 201                                 pktmbuf_sw_copy(pkts_burst[j],
 202                                         pkts_burst_copy[j]);
 203
 204                         rte_mempool_put_bulk(ioat_pktmbuf_pool,
 205                                 (void *)pkts_burst, nb_rx);
 206
 207                         nb_enq = rte_ring_enqueue_burst(
 208                                 rx_config->rx_to_tx_ring,
 209                                 (void *)pkts_burst_copy, nb_rx, NULL);
 210
 211                         /* Free any not enqueued packets. */
 212                         rte_mempool_put_bulk(ioat_pktmbuf_pool,
 213                                 (void *)&pkts_burst_copy[nb_enq],
 214                                 nb_rx - nb_enq);
 215                 }
 216         }
 217 }
 218
 219 /* Transmit packets from IOAT rawdev/rte_ring for one port. */
 220 static void
 221 ioat_tx_port(struct rxtx_port_config *tx_config)
 222 {
 223         uint32_t i, j, nb_dq = 0;
 224         struct rte_mbuf *mbufs_src[MAX_PKT_BURST];
 225         struct rte_mbuf *mbufs_dst[MAX_PKT_BURST];
 226
 227         for (i = 0; i < tx_config->nb_queues; i++) {
 228                 if (copy_mode == COPY_MODE_IOAT_NUM) {
 229                         /* Deque the mbufs from IOAT device. */
 230                         nb_dq = rte_ioat_completed_copies(
 231                                 tx_config->ioat_ids[i], MAX_PKT_BURST,
 232                                 (void *)mbufs_src, (void *)mbufs_dst);
 233                 } else {
 234                         /* Deque the mbufs from rx_to_tx_ring. */
 235                         nb_dq = rte_ring_dequeue_burst(
 236                                 tx_config->rx_to_tx_ring, (void *)mbufs_dst,
 237                                 MAX_PKT_BURST, NULL);
 238                 }
 239
 240                 if (nb_dq == 0)
 241                         return;
 242
 243                 if (copy_mode == COPY_MODE_IOAT_NUM)
 244                         rte_mempool_put_bulk(ioat_pktmbuf_pool,
 245                                 (void *)mbufs_src, nb_dq);
 246
 247                 /* Update macs if enabled */
 248                 if (mac_updating) {
 249                         for (j = 0; j < nb_dq; j++)
 250                                 update_mac_addrs(mbufs_dst[j],
 251                                         tx_config->rxtx_port);
 252                 }
 253
 254                 const uint16_t nb_tx = rte_eth_tx_burst(
 255                         tx_config->rxtx_port, 0,
 256                         (void *)mbufs_dst, nb_dq);
 257
 258                 /* Free any unsent packets. */
 259                 if (unlikely(nb_tx < nb_dq))
 260                         rte_mempool_put_bulk(ioat_pktmbuf_pool,
 261                         (void *)&mbufs_dst[nb_tx],
 262                                 nb_dq - nb_tx);
 263         }
 264 }
 265
 266 /* Main rx and tx loop if only one slave lcore available */
 267 static void
 268 rxtx_main_loop(void)
 269 {
 270         uint16_t i;
 271         uint16_t nb_ports = cfg.nb_ports;
 272
 273         RTE_LOG(INFO, IOAT, "Entering main rx and tx loop for copy on"
 274                 " lcore %u\n", rte_lcore_id());
 275
 276         while (!force_quit)
 277                 for (i = 0; i < nb_ports; i++) {
 278                         ioat_rx_port(&cfg.ports[i]);
 279                         ioat_tx_port(&cfg.ports[i]);
 280                 }
 281 }
 282
 283 static void start_forwarding_cores(void)
 284 {
 285         uint32_t lcore_id = rte_lcore_id();
 286
 287         RTE_LOG(INFO, IOAT, "Entering %s on lcore %u\n",
 288                 __func__, rte_lcore_id());
 289
 290         lcore_id = rte_get_next_lcore(lcore_id, true, true);
 291         rte_eal_remote_launch((lcore_function_t *)rxtx_main_loop,
 292                 NULL, lcore_id);
 293 }
 294
 295 /* Display usage */
 296 static void
 297 ioat_usage(const char *prgname)
 298 {
 299         printf("%s [EAL options] -- -p PORTMASK [-q NQ]\n"
 300                 "  -p --portmask: hexadecimal bitmask of ports to configure\n"
 301                 "  -q NQ: number of RX queues per port (default is 1)\n"
 302                 "  --[no-]mac-updating: Enable or disable MAC addresses updating (enabled by default)\n"
 303                 "      When enabled:\n"
 304                 "       - The source MAC address is replaced by the TX port MAC address\n"
 305                 "       - The destination MAC address is replaced by 02:00:00:00:00:TX_PORT_ID\n"
 306                 "  -c --copy-type CT: type of copy: sw|hw\n"
 307                 "  -s --ring-size RS: size of IOAT rawdev ring for hardware copy mode or rte_ring for software copy mode\n",
 308                         prgname);
 309 }
 310
 311 static int
 312 ioat_parse_portmask(const char *portmask)
 313 {
 314         char *end = NULL;
 315         unsigned long pm;
 316
 317         /* Parse hexadecimal string */
 318         pm = strtoul(portmask, &end, 16);
 319         if ((portmask[0] == '\0') || (end == NULL) || (*end != '\0'))
 320                 return -1;
 321
 322         return pm;
 323 }
 324
 325 static copy_mode_t
 326 ioat_parse_copy_mode(const char *copy_mode)
 327 {
 328         if (strcmp(copy_mode, COPY_MODE_SW) == 0)
 329                 return COPY_MODE_SW_NUM;
 330         else if (strcmp(copy_mode, COPY_MODE_IOAT) == 0)
 331                 return COPY_MODE_IOAT_NUM;
 332
 333         return COPY_MODE_INVALID_NUM;
 334 }
 335
 336 /* Parse the argument given in the command line of the application */
 337 static int
 338 ioat_parse_args(int argc, char **argv, unsigned int nb_ports)
 339 {
 340         static const char short_options[] =
 341                 "p:"  /* portmask */
 342                 "q:"  /* number of RX queues per port */
 343                 "c:"  /* copy type (sw|hw) */
 344                 "s:"  /* ring size */
 345                 ;
 346
 347         static const struct option lgopts[] = {
 348                 {CMD_LINE_OPT_MAC_UPDATING, no_argument, &mac_updating, 1},
 349                 {CMD_LINE_OPT_NO_MAC_UPDATING, no_argument, &mac_updating, 0},
 350                 {CMD_LINE_OPT_PORTMASK, required_argument, NULL, 'p'},
 351                 {CMD_LINE_OPT_NB_QUEUE, required_argument, NULL, 'q'},
 352                 {CMD_LINE_OPT_COPY_TYPE, required_argument, NULL, 'c'},
 353                 {CMD_LINE_OPT_RING_SIZE, required_argument, NULL, 's'},
 354                 {NULL, 0, 0, 0}
 355         };
 356
 357         const unsigned int default_port_mask = (1 << nb_ports) - 1;
 358         int opt, ret;
 359         char **argvopt;
 360         int option_index;
 361         char *prgname = argv[0];
 362
 363         ioat_enabled_port_mask = default_port_mask;
 364         argvopt = argv;
 365
 366         while ((opt = getopt_long(argc, argvopt, short_options,
 367                         lgopts, &option_index)) != EOF) {
 368
 369                 switch (opt) {
 370                 /* portmask */
 371                 case 'p':
 372                         ioat_enabled_port_mask = ioat_parse_portmask(optarg);
 373                         if (ioat_enabled_port_mask & ~default_port_mask ||
 374                                         ioat_enabled_port_mask <= 0) {
 375                                 printf("Invalid portmask, %s, suggest 0x%x\n",
 376                                                 optarg, default_port_mask);
 377                                 ioat_usage(prgname);
 378                                 return -1;
 379                         }
 380                         break;
 381
 382                 case 'q':
 383                         nb_queues = atoi(optarg);
 384                         if (nb_queues == 0 || nb_queues > MAX_RX_QUEUES_COUNT) {
 385                                 printf("Invalid RX queues number %s. Max %u\n",
 386                                         optarg, MAX_RX_QUEUES_COUNT);
 387                                 ioat_usage(prgname);
 388                                 return -1;
 389                         }
 390                         break;
 391
 392                 case 'c':
 393                         copy_mode = ioat_parse_copy_mode(optarg);
 394                         if (copy_mode == COPY_MODE_INVALID_NUM) {
 395                                 printf("Invalid copy type. Use: sw, hw\n");
 396                                 ioat_usage(prgname);
 397                                 return -1;
 398                         }
 399                         break;
 400
 401                 case 's':
 402                         ring_size = atoi(optarg);
 403                         if (ring_size == 0) {
 404                                 printf("Invalid ring size, %s.\n", optarg);
 405                                 ioat_usage(prgname);
 406                                 return -1;
 407                         }
 408                         break;
 409
 410                 /* long options */
 411                 case 0:
 412                         break;
 413
 414                 default:
 415                         ioat_usage(prgname);
 416                         return -1;
 417                 }
 418         }
 419
 420         printf("MAC updating %s\n", mac_updating ? "enabled" : "disabled");
 421         if (optind >= 0)
 422                 argv[optind - 1] = prgname;
 423
 424         ret = optind - 1;
 425         optind = 1; /* reset getopt lib */
 426         return ret;
 427 }
 428
 429 /* check link status, return true if at least one port is up */
 430 static int
 431 check_link_status(uint32_t port_mask)
 432 {
 433         uint16_t portid;
 434         struct rte_eth_link link;
 435         int retval = 0;
 436
 437         printf("\nChecking link status\n");
 438         RTE_ETH_FOREACH_DEV(portid) {
 439                 if ((port_mask & (1 << portid)) == 0)
 440                         continue;
 441
 442                 memset(&link, 0, sizeof(link));
 443                 rte_eth_link_get(portid, &link);
 444
 445                 /* Print link status */
 446                 if (link.link_status) {
 447                         printf(
 448                                 "Port %d Link Up. Speed %u Mbps - %s\n",
 449                                 portid, link.link_speed,
 450                                 (link.link_duplex == ETH_LINK_FULL_DUPLEX) ?
 451                                 ("full-duplex") : ("half-duplex\n"));
 452                         retval = 1;
 453                 } else
 454                         printf("Port %d Link Down\n", portid);
 455         }
 456         return retval;
 457 }
 458
 459 static void
 460 configure_rawdev_queue(uint32_t dev_id)
 461 {
 462         struct rte_ioat_rawdev_config dev_config = { .ring_size = ring_size };
 463         struct rte_rawdev_info info = { .dev_private = &dev_config };
 464
 465         if (rte_rawdev_configure(dev_id, &info) != 0) {
 466                 rte_exit(EXIT_FAILURE,
 467                         "Error with rte_rawdev_configure()\n");
 468         }
 469         if (rte_rawdev_start(dev_id) != 0) {
 470                 rte_exit(EXIT_FAILURE,
 471                         "Error with rte_rawdev_start()\n");
 472         }
 473 }
 474
 475 static void
 476 assign_rawdevs(void)
 477 {
 478         uint16_t nb_rawdev = 0, rdev_id = 0;
 479         uint32_t i, j;
 480
 481         for (i = 0; i < cfg.nb_ports; i++) {
 482                 for (j = 0; j < cfg.ports[i].nb_queues; j++) {
 483                         struct rte_rawdev_info rdev_info = { 0 };
 484
 485                         do {
 486                                 if (rdev_id == rte_rawdev_count())
 487                                         goto end;
 488                                 rte_rawdev_info_get(rdev_id++, &rdev_info);
 489                         } while (strcmp(rdev_info.driver_name,
 490                                 IOAT_PMD_RAWDEV_NAME_STR) != 0);
 491
 492                         cfg.ports[i].ioat_ids[j] = rdev_id - 1;
 493                         configure_rawdev_queue(cfg.ports[i].ioat_ids[j]);
 494                         ++nb_rawdev;
 495                 }
 496         }
 497 end:
 498         if (nb_rawdev < cfg.nb_ports * cfg.ports[0].nb_queues)
 499                 rte_exit(EXIT_FAILURE,
 500                         "Not enough IOAT rawdevs (%u) for all queues (%u).\n",
 501                         nb_rawdev, cfg.nb_ports * cfg.ports[0].nb_queues);
 502         RTE_LOG(INFO, IOAT, "Number of used rawdevs: %u.\n", nb_rawdev);
 503 }
 504
 505 static void
 506 assign_rings(void)
 507 {
 508         uint32_t i;
 509
 510         for (i = 0; i < cfg.nb_ports; i++) {
 511                 char ring_name[RTE_RING_NAMESIZE];
 512
 513                 snprintf(ring_name, sizeof(ring_name), "rx_to_tx_ring_%u", i);
 514                 /* Create ring for inter core communication */
 515                 cfg.ports[i].rx_to_tx_ring = rte_ring_create(
 516                         ring_name, ring_size,
 517                         rte_socket_id(), RING_F_SP_ENQ | RING_F_SC_DEQ);
 518
 519                 if (cfg.ports[i].rx_to_tx_ring == NULL)
 520                         rte_exit(EXIT_FAILURE, "Ring create failed: %s\n",
 521                                 rte_strerror(rte_errno));
 522         }
 523 }
 524
 525 /*
 526  * Initializes a given port using global settings and with the RX buffers
 527  * coming from the mbuf_pool passed as a parameter.
 528  */
 529 static inline void
 530 port_init(uint16_t portid, struct rte_mempool *mbuf_pool, uint16_t nb_queues)
 531 {
 532         /* configuring port to use RSS for multiple RX queues */
 533         static const struct rte_eth_conf port_conf = {
 534                 .rxmode = {
 535                         .mq_mode = ETH_MQ_RX_RSS,
 536                         .max_rx_pkt_len = RTE_ETHER_MAX_LEN
 537                 },
 538                 .rx_adv_conf = {
 539                         .rss_conf = {
 540                                 .rss_key = NULL,
 541                                 .rss_hf = ETH_RSS_PROTO_MASK,
 542                         }
 543                 }
 544         };
 545
 546         struct rte_eth_rxconf rxq_conf;
 547         struct rte_eth_txconf txq_conf;
 548         struct rte_eth_conf local_port_conf = port_conf;
 549         struct rte_eth_dev_info dev_info;
 550         int ret, i;
 551
 552         /* Skip ports that are not enabled */
 553         if ((ioat_enabled_port_mask & (1 << portid)) == 0) {
 554                 printf("Skipping disabled port %u\n", portid);
 555                 return;
 556         }
 557
 558         /* Init port */
 559         printf("Initializing port %u... ", portid);
 560         fflush(stdout);
 561         rte_eth_dev_info_get(portid, &dev_info);
 562         local_port_conf.rx_adv_conf.rss_conf.rss_hf &=
 563                 dev_info.flow_type_rss_offloads;
 564         if (dev_info.tx_offload_capa & DEV_TX_OFFLOAD_MBUF_FAST_FREE)
 565                 local_port_conf.txmode.offloads |=
 566                         DEV_TX_OFFLOAD_MBUF_FAST_FREE;
 567         ret = rte_eth_dev_configure(portid, nb_queues, 1, &local_port_conf);
 568         if (ret < 0)
 569                 rte_exit(EXIT_FAILURE, "Cannot configure device:"
 570                         " err=%d, port=%u\n", ret, portid);
 571
 572         ret = rte_eth_dev_adjust_nb_rx_tx_desc(portid, &nb_rxd,
 573                         &nb_txd);
 574         if (ret < 0)
 575                 rte_exit(EXIT_FAILURE,
 576                         "Cannot adjust number of descriptors: err=%d, port=%u\n",
 577                         ret, portid);
 578
 579         rte_eth_macaddr_get(portid, &ioat_ports_eth_addr[portid]);
 580
 581         /* Init RX queues */
 582         rxq_conf = dev_info.default_rxconf;
 583         rxq_conf.offloads = local_port_conf.rxmode.offloads;
 584         for (i = 0; i < nb_queues; i++) {
 585                 ret = rte_eth_rx_queue_setup(portid, i, nb_rxd,
 586                         rte_eth_dev_socket_id(portid), &rxq_conf,
 587                         mbuf_pool);
 588                 if (ret < 0)
 589                         rte_exit(EXIT_FAILURE,
 590                                 "rte_eth_rx_queue_setup:err=%d,port=%u, queue_id=%u\n",
 591                                 ret, portid, i);
 592         }
 593
 594         /* Init one TX queue on each port */
 595         txq_conf = dev_info.default_txconf;
 596         txq_conf.offloads = local_port_conf.txmode.offloads;
 597         ret = rte_eth_tx_queue_setup(portid, 0, nb_txd,
 598                         rte_eth_dev_socket_id(portid),
 599                         &txq_conf);
 600         if (ret < 0)
 601                 rte_exit(EXIT_FAILURE,
 602                         "rte_eth_tx_queue_setup:err=%d,port=%u\n",
 603                         ret, portid);
 604
 605         /* Initialize TX buffers */
 606         tx_buffer[portid] = rte_zmalloc_socket("tx_buffer",
 607                         RTE_ETH_TX_BUFFER_SIZE(MAX_PKT_BURST), 0,
 608                         rte_eth_dev_socket_id(portid));
 609         if (tx_buffer[portid] == NULL)
 610                 rte_exit(EXIT_FAILURE,
 611                         "Cannot allocate buffer for tx on port %u\n",
 612                         portid);
 613
 614         rte_eth_tx_buffer_init(tx_buffer[portid], MAX_PKT_BURST);
 615
 616         /* Start device */
 617         ret = rte_eth_dev_start(portid);
 618         if (ret < 0)
 619                 rte_exit(EXIT_FAILURE,
 620                         "rte_eth_dev_start:err=%d, port=%u\n",
 621                         ret, portid);
 622
 623         rte_eth_promiscuous_enable(portid);
 624
 625         printf("Port %u, MAC address: %02X:%02X:%02X:%02X:%02X:%02X\n\n",
 626                         portid,
 627                         ioat_ports_eth_addr[portid].addr_bytes[0],
 628                         ioat_ports_eth_addr[portid].addr_bytes[1],
 629                         ioat_ports_eth_addr[portid].addr_bytes[2],
 630                         ioat_ports_eth_addr[portid].addr_bytes[3],
 631                         ioat_ports_eth_addr[portid].addr_bytes[4],
 632                         ioat_ports_eth_addr[portid].addr_bytes[5]);
 633
 634         cfg.ports[cfg.nb_ports].rxtx_port = portid;
 635         cfg.ports[cfg.nb_ports++].nb_queues = nb_queues;
 636 }
 637
 638 static void
 639 signal_handler(int signum)
 640 {
 641         if (signum == SIGINT || signum == SIGTERM) {
 642                 printf("\n\nSignal %d received, preparing to exit...\n",
 643                         signum);
 644                 force_quit = true;
 645         }
 646 }
 647
 648 int
 649 main(int argc, char **argv)
 650 {
 651         int ret;
 652         uint16_t nb_ports, portid;
 653         uint32_t i;
 654         unsigned int nb_mbufs;
 655
 656         /* Init EAL */
 657         ret = rte_eal_init(argc, argv);
 658         if (ret < 0)
 659                 rte_exit(EXIT_FAILURE, "Invalid EAL arguments\n");
 660         argc -= ret;
 661         argv += ret;
 662
 663         force_quit = false;
 664         signal(SIGINT, signal_handler);
 665         signal(SIGTERM, signal_handler);
 666
 667         nb_ports = rte_eth_dev_count_avail();
 668         if (nb_ports == 0)
 669                 rte_exit(EXIT_FAILURE, "No Ethernet ports - bye\n");
 670
 671         /* Parse application arguments (after the EAL ones) */
 672         ret = ioat_parse_args(argc, argv, nb_ports);
 673         if (ret < 0)
 674                 rte_exit(EXIT_FAILURE, "Invalid IOAT arguments\n");
 675
 676         nb_mbufs = RTE_MAX(nb_ports * (nb_queues * (nb_rxd + nb_txd +
 677                 4 * MAX_PKT_BURST) + rte_lcore_count() * MEMPOOL_CACHE_SIZE),
 678                 MIN_POOL_SIZE);
 679
 680         /* Create the mbuf pool */
 681         ioat_pktmbuf_pool = rte_pktmbuf_pool_create("mbuf_pool", nb_mbufs,
 682                 MEMPOOL_CACHE_SIZE, 0, RTE_MBUF_DEFAULT_BUF_SIZE,
 683                 rte_socket_id());
 684         if (ioat_pktmbuf_pool == NULL)
 685                 rte_exit(EXIT_FAILURE, "Cannot init mbuf pool\n");
 686
 687         /* Initialise each port */
 688         cfg.nb_ports = 0;
 689         RTE_ETH_FOREACH_DEV(portid)
 690                 port_init(portid, ioat_pktmbuf_pool, nb_queues);
 691
 692         while (!check_link_status(ioat_enabled_port_mask) && !force_quit)
 693                 sleep(1);
 694
 695         /* Check if there is enough lcores for all ports. */
 696         cfg.nb_lcores = rte_lcore_count() - 1;
 697         if (cfg.nb_lcores < 1)
 698                 rte_exit(EXIT_FAILURE,
 699                         "There should be at least one slave lcore.\n");
 700
 701         if (copy_mode == COPY_MODE_IOAT_NUM)
 702                 assign_rawdevs();
 703         else /* copy_mode == COPY_MODE_SW_NUM */
 704                 assign_rings();
 705
 706         start_forwarding_cores();
 707
 708         /* force_quit is true when we get here */
 709         rte_eal_mp_wait_lcore();
 710
 711         uint32_t j;
 712         for (i = 0; i < cfg.nb_ports; i++) {
 713                 printf("Closing port %d\n", cfg.ports[i].rxtx_port);
 714                 rte_eth_dev_stop(cfg.ports[i].rxtx_port);
 715                 rte_eth_dev_close(cfg.ports[i].rxtx_port);
 716                 if (copy_mode == COPY_MODE_IOAT_NUM) {
 717                         for (j = 0; j < cfg.ports[i].nb_queues; j++) {
 718                                 printf("Stopping rawdev %d\n",
 719                                         cfg.ports[i].ioat_ids[j]);
 720                                 rte_rawdev_stop(cfg.ports[i].ioat_ids[j]);
 721                         }
 722                 } else /* copy_mode == COPY_MODE_SW_NUM */
 723                         rte_ring_free(cfg.ports[i].rx_to_tx_ring);
 724         }
 725
 726         printf("Bye...\n");
 727         return 0;
 728 }