1 /* SPDX-License-Identifier: BSD-3-Clause 2 * Copyright(c) 2016 Intel Corporation 3 */ 4 5 #include <stdio.h> 6 #include <string.h> 7 #include <stdint.h> 8 #include <inttypes.h> 9 #include <stdlib.h> 10 #include <getopt.h> 11 #include <signal.h> 12 #include <stdbool.h> 13 #include <net/if.h> 14 15 #include <rte_eal.h> 16 #include <rte_alarm.h> 17 #include <rte_common.h> 18 #include <rte_debug.h> 19 #include <rte_ethdev.h> 20 #include <rte_memory.h> 21 #include <rte_lcore.h> 22 #include <rte_branch_prediction.h> 23 #include <rte_errno.h> 24 #include <rte_dev.h> 25 #include <rte_kvargs.h> 26 #include <rte_mempool.h> 27 #include <rte_ring.h> 28 #include <rte_string_fns.h> 29 #include <rte_pdump.h> 30 31 #define CMD_LINE_OPT_PDUMP "pdump" 32 #define CMD_LINE_OPT_PDUMP_NUM 256 33 #define CMD_LINE_OPT_MULTI "multi" 34 #define CMD_LINE_OPT_MULTI_NUM 257 35 #define PDUMP_PORT_ARG "port" 36 #define PDUMP_PCI_ARG "device_id" 37 #define PDUMP_QUEUE_ARG "queue" 38 #define PDUMP_DIR_ARG "dir" 39 #define PDUMP_RX_DEV_ARG "rx-dev" 40 #define PDUMP_TX_DEV_ARG "tx-dev" 41 #define PDUMP_RING_SIZE_ARG "ring-size" 42 #define PDUMP_MSIZE_ARG "mbuf-size" 43 #define PDUMP_NUM_MBUFS_ARG "total-num-mbufs" 44 45 #define VDEV_NAME_FMT "net_pcap_%s_%d" 46 #define VDEV_PCAP_ARGS_FMT "tx_pcap=%s" 47 #define VDEV_IFACE_ARGS_FMT "tx_iface=%s" 48 #define TX_STREAM_SIZE 64 49 50 #define MP_NAME "pdump_pool_%d" 51 52 #define RX_RING "rx_ring_%d" 53 #define TX_RING "tx_ring_%d" 54 55 #define RX_STR "rx" 56 #define TX_STR "tx" 57 58 /* Maximum long option length for option parsing. */ 59 #define APP_ARG_TCPDUMP_MAX_TUPLES 54 60 #define MBUF_POOL_CACHE_SIZE 250 61 #define TX_DESC_PER_QUEUE 512 62 #define RX_DESC_PER_QUEUE 128 63 #define MBUFS_PER_POOL 65535 64 #define MAX_LONG_OPT_SZ 64 65 #define RING_SIZE 16384 66 #define SIZE 256 67 #define BURST_SIZE 32 68 #define NUM_VDEVS 2 69 /* Maximum delay for exiting after primary process. */ 70 #define MONITOR_INTERVAL (500 * 1000) 71 72 /* true if x is a power of 2 */ 73 #define POWEROF2(x) ((((x)-1) & (x)) == 0) 74 75 enum pdump_en_dis { 76 DISABLE = 1, 77 ENABLE = 2 78 }; 79 80 enum pcap_stream { 81 IFACE = 1, 82 PCAP = 2 83 }; 84 85 enum pdump_by { 86 PORT_ID = 1, 87 DEVICE_ID = 2 88 }; 89 90 static const char * const valid_pdump_arguments[] = { 91 PDUMP_PORT_ARG, 92 PDUMP_PCI_ARG, 93 PDUMP_QUEUE_ARG, 94 PDUMP_DIR_ARG, 95 PDUMP_RX_DEV_ARG, 96 PDUMP_TX_DEV_ARG, 97 PDUMP_RING_SIZE_ARG, 98 PDUMP_MSIZE_ARG, 99 PDUMP_NUM_MBUFS_ARG, 100 NULL 101 }; 102 103 struct pdump_stats { 104 uint64_t dequeue_pkts; 105 uint64_t tx_pkts; 106 uint64_t freed_pkts; 107 }; 108 109 struct pdump_tuples { 110 /* cli params */ 111 uint16_t port; 112 char *device_id; 113 uint16_t queue; 114 char rx_dev[TX_STREAM_SIZE]; 115 char tx_dev[TX_STREAM_SIZE]; 116 uint32_t ring_size; 117 uint16_t mbuf_data_size; 118 uint32_t total_num_mbufs; 119 120 /* params for library API call */ 121 uint32_t dir; 122 struct rte_mempool *mp; 123 struct rte_ring *rx_ring; 124 struct rte_ring *tx_ring; 125 126 /* params for packet dumping */ 127 enum pdump_by dump_by_type; 128 uint16_t rx_vdev_id; 129 uint16_t tx_vdev_id; 130 enum pcap_stream rx_vdev_stream_type; 131 enum pcap_stream tx_vdev_stream_type; 132 bool single_pdump_dev; 133 134 /* stats */ 135 struct pdump_stats stats; 136 } __rte_cache_aligned; 137 static struct pdump_tuples pdump_t[APP_ARG_TCPDUMP_MAX_TUPLES]; 138 139 struct parse_val { 140 uint64_t min; 141 uint64_t max; 142 uint64_t val; 143 }; 144 145 static int num_tuples; 146 static struct rte_eth_conf port_conf_default; 147 static volatile uint8_t quit_signal; 148 static uint8_t multiple_core_capture; 149 150 /**< display usage */ 151 static void 152 pdump_usage(const char *prgname) 153 { 154 printf("usage: %s [EAL options]" 155 " --["CMD_LINE_OPT_MULTI"]\n" 156 " --"CMD_LINE_OPT_PDUMP" " 157 "'(port=<port id> | device_id=<pci id or vdev name>)," 158 "(queue=<queue_id>)," 159 "(rx-dev=<iface or pcap file> |" 160 " tx-dev=<iface or pcap file>," 161 "[ring-size=<ring size>default:16384]," 162 "[mbuf-size=<mbuf data size>default:2176]," 163 "[total-num-mbufs=<number of mbufs>default:65535]'\n", 164 prgname); 165 } 166 167 static int 168 parse_device_id(const char *key __rte_unused, const char *value, 169 void *extra_args) 170 { 171 struct pdump_tuples *pt = extra_args; 172 173 pt->device_id = strdup(value); 174 pt->dump_by_type = DEVICE_ID; 175 176 return 0; 177 } 178 179 static int 180 parse_queue(const char *key __rte_unused, const char *value, void *extra_args) 181 { 182 unsigned long n; 183 struct pdump_tuples *pt = extra_args; 184 185 if (!strcmp(value, "*")) 186 pt->queue = RTE_PDUMP_ALL_QUEUES; 187 else { 188 n = strtoul(value, NULL, 10); 189 pt->queue = (uint16_t) n; 190 } 191 return 0; 192 } 193 194 static int 195 parse_rxtxdev(const char *key, const char *value, void *extra_args) 196 { 197 198 struct pdump_tuples *pt = extra_args; 199 200 if (!strcmp(key, PDUMP_RX_DEV_ARG)) { 201 strlcpy(pt->rx_dev, value, sizeof(pt->rx_dev)); 202 /* identify the tx stream type for pcap vdev */ 203 if (if_nametoindex(pt->rx_dev)) 204 pt->rx_vdev_stream_type = IFACE; 205 } else if (!strcmp(key, PDUMP_TX_DEV_ARG)) { 206 strlcpy(pt->tx_dev, value, sizeof(pt->tx_dev)); 207 /* identify the tx stream type for pcap vdev */ 208 if (if_nametoindex(pt->tx_dev)) 209 pt->tx_vdev_stream_type = IFACE; 210 } 211 212 return 0; 213 } 214 215 static int 216 parse_uint_value(const char *key, const char *value, void *extra_args) 217 { 218 struct parse_val *v; 219 unsigned long t; 220 char *end; 221 int ret = 0; 222 223 errno = 0; 224 v = extra_args; 225 t = strtoul(value, &end, 10); 226 227 if (errno != 0 || end[0] != 0 || t < v->min || t > v->max) { 228 printf("invalid value:\"%s\" for key:\"%s\", " 229 "value must be >= %"PRIu64" and <= %"PRIu64"\n", 230 value, key, v->min, v->max); 231 ret = -EINVAL; 232 } 233 if (!strcmp(key, PDUMP_RING_SIZE_ARG) && !POWEROF2(t)) { 234 printf("invalid value:\"%s\" for key:\"%s\", " 235 "value must be power of 2\n", value, key); 236 ret = -EINVAL; 237 } 238 239 if (ret != 0) 240 return ret; 241 242 v->val = t; 243 return 0; 244 } 245 246 static int 247 parse_pdump(const char *optarg) 248 { 249 struct rte_kvargs *kvlist; 250 int ret = 0, cnt1, cnt2; 251 struct pdump_tuples *pt; 252 struct parse_val v = {0}; 253 254 pt = &pdump_t[num_tuples]; 255 256 /* initial check for invalid arguments */ 257 kvlist = rte_kvargs_parse(optarg, valid_pdump_arguments); 258 if (kvlist == NULL) { 259 printf("--pdump=\"%s\": invalid argument passed\n", optarg); 260 return -1; 261 } 262 263 /* port/device_id parsing and validation */ 264 cnt1 = rte_kvargs_count(kvlist, PDUMP_PORT_ARG); 265 cnt2 = rte_kvargs_count(kvlist, PDUMP_PCI_ARG); 266 if (!((cnt1 == 1 && cnt2 == 0) || (cnt1 == 0 && cnt2 == 1))) { 267 printf("--pdump=\"%s\": must have either port or " 268 "device_id argument\n", optarg); 269 ret = -1; 270 goto free_kvlist; 271 } else if (cnt1 == 1) { 272 v.min = 0; 273 v.max = RTE_MAX_ETHPORTS-1; 274 ret = rte_kvargs_process(kvlist, PDUMP_PORT_ARG, 275 &parse_uint_value, &v); 276 if (ret < 0) 277 goto free_kvlist; 278 pt->port = (uint16_t) v.val; 279 pt->dump_by_type = PORT_ID; 280 } else if (cnt2 == 1) { 281 ret = rte_kvargs_process(kvlist, PDUMP_PCI_ARG, 282 &parse_device_id, pt); 283 if (ret < 0) 284 goto free_kvlist; 285 } 286 287 /* queue parsing and validation */ 288 cnt1 = rte_kvargs_count(kvlist, PDUMP_QUEUE_ARG); 289 if (cnt1 != 1) { 290 printf("--pdump=\"%s\": must have queue argument\n", optarg); 291 ret = -1; 292 goto free_kvlist; 293 } 294 ret = rte_kvargs_process(kvlist, PDUMP_QUEUE_ARG, &parse_queue, pt); 295 if (ret < 0) 296 goto free_kvlist; 297 298 /* rx-dev and tx-dev parsing and validation */ 299 cnt1 = rte_kvargs_count(kvlist, PDUMP_RX_DEV_ARG); 300 cnt2 = rte_kvargs_count(kvlist, PDUMP_TX_DEV_ARG); 301 if (cnt1 == 0 && cnt2 == 0) { 302 printf("--pdump=\"%s\": must have either rx-dev or " 303 "tx-dev argument\n", optarg); 304 ret = -1; 305 goto free_kvlist; 306 } else if (cnt1 == 1 && cnt2 == 1) { 307 ret = rte_kvargs_process(kvlist, PDUMP_RX_DEV_ARG, 308 &parse_rxtxdev, pt); 309 if (ret < 0) 310 goto free_kvlist; 311 ret = rte_kvargs_process(kvlist, PDUMP_TX_DEV_ARG, 312 &parse_rxtxdev, pt); 313 if (ret < 0) 314 goto free_kvlist; 315 /* if captured packets has to send to the same vdev */ 316 if (!strcmp(pt->rx_dev, pt->tx_dev)) 317 pt->single_pdump_dev = true; 318 pt->dir = RTE_PDUMP_FLAG_RXTX; 319 } else if (cnt1 == 1) { 320 ret = rte_kvargs_process(kvlist, PDUMP_RX_DEV_ARG, 321 &parse_rxtxdev, pt); 322 if (ret < 0) 323 goto free_kvlist; 324 pt->dir = RTE_PDUMP_FLAG_RX; 325 } else if (cnt2 == 1) { 326 ret = rte_kvargs_process(kvlist, PDUMP_TX_DEV_ARG, 327 &parse_rxtxdev, pt); 328 if (ret < 0) 329 goto free_kvlist; 330 pt->dir = RTE_PDUMP_FLAG_TX; 331 } 332 333 /* optional */ 334 /* ring_size parsing and validation */ 335 cnt1 = rte_kvargs_count(kvlist, PDUMP_RING_SIZE_ARG); 336 if (cnt1 == 1) { 337 v.min = 2; 338 v.max = RTE_RING_SZ_MASK-1; 339 ret = rte_kvargs_process(kvlist, PDUMP_RING_SIZE_ARG, 340 &parse_uint_value, &v); 341 if (ret < 0) 342 goto free_kvlist; 343 pt->ring_size = (uint32_t) v.val; 344 } else 345 pt->ring_size = RING_SIZE; 346 347 /* mbuf_data_size parsing and validation */ 348 cnt1 = rte_kvargs_count(kvlist, PDUMP_MSIZE_ARG); 349 if (cnt1 == 1) { 350 v.min = 1; 351 v.max = UINT16_MAX; 352 ret = rte_kvargs_process(kvlist, PDUMP_MSIZE_ARG, 353 &parse_uint_value, &v); 354 if (ret < 0) 355 goto free_kvlist; 356 pt->mbuf_data_size = (uint16_t) v.val; 357 } else 358 pt->mbuf_data_size = RTE_MBUF_DEFAULT_BUF_SIZE; 359 360 /* total_num_mbufs parsing and validation */ 361 cnt1 = rte_kvargs_count(kvlist, PDUMP_NUM_MBUFS_ARG); 362 if (cnt1 == 1) { 363 v.min = 1025; 364 v.max = UINT16_MAX; 365 ret = rte_kvargs_process(kvlist, PDUMP_NUM_MBUFS_ARG, 366 &parse_uint_value, &v); 367 if (ret < 0) 368 goto free_kvlist; 369 pt->total_num_mbufs = (uint16_t) v.val; 370 } else 371 pt->total_num_mbufs = MBUFS_PER_POOL; 372 373 num_tuples++; 374 375 free_kvlist: 376 rte_kvargs_free(kvlist); 377 return ret; 378 } 379 380 /* Parse the argument given in the command line of the application */ 381 static int 382 launch_args_parse(int argc, char **argv, char *prgname) 383 { 384 int opt, ret; 385 int option_index; 386 static struct option long_option[] = { 387 {CMD_LINE_OPT_PDUMP, 1, 0, CMD_LINE_OPT_PDUMP_NUM}, 388 {CMD_LINE_OPT_MULTI, 0, 0, CMD_LINE_OPT_MULTI_NUM}, 389 {NULL, 0, 0, 0} 390 }; 391 392 if (argc == 1) 393 pdump_usage(prgname); 394 395 /* Parse command line */ 396 while ((opt = getopt_long(argc, argv, " ", 397 long_option, &option_index)) != EOF) { 398 switch (opt) { 399 case CMD_LINE_OPT_PDUMP_NUM: 400 ret = parse_pdump(optarg); 401 if (ret) { 402 pdump_usage(prgname); 403 return -1; 404 } 405 break; 406 case CMD_LINE_OPT_MULTI_NUM: 407 multiple_core_capture = 1; 408 break; 409 default: 410 pdump_usage(prgname); 411 return -1; 412 } 413 } 414 415 return 0; 416 } 417 418 static void 419 monitor_primary(void *arg __rte_unused) 420 { 421 if (quit_signal) 422 return; 423 424 if (rte_eal_primary_proc_alive(NULL)) { 425 rte_eal_alarm_set(MONITOR_INTERVAL, monitor_primary, NULL); 426 return; 427 } 428 429 printf("Primary process is no longer active, exiting...\n"); 430 quit_signal = 1; 431 } 432 433 static void 434 print_pdump_stats(void) 435 { 436 int i; 437 struct pdump_tuples *pt; 438 439 for (i = 0; i < num_tuples; i++) { 440 printf("##### PDUMP DEBUG STATS #####\n"); 441 pt = &pdump_t[i]; 442 printf(" -packets dequeued: %"PRIu64"\n", 443 pt->stats.dequeue_pkts); 444 printf(" -packets transmitted to vdev: %"PRIu64"\n", 445 pt->stats.tx_pkts); 446 printf(" -packets freed: %"PRIu64"\n", 447 pt->stats.freed_pkts); 448 } 449 } 450 451 static inline void 452 disable_pdump(struct pdump_tuples *pt) 453 { 454 if (pt->dump_by_type == DEVICE_ID) 455 rte_pdump_disable_by_deviceid(pt->device_id, pt->queue, 456 pt->dir); 457 else if (pt->dump_by_type == PORT_ID) 458 rte_pdump_disable(pt->port, pt->queue, pt->dir); 459 } 460 461 static inline void 462 pdump_rxtx(struct rte_ring *ring, uint16_t vdev_id, struct pdump_stats *stats) 463 { 464 /* write input packets of port to vdev for pdump */ 465 struct rte_mbuf *rxtx_bufs[BURST_SIZE]; 466 467 /* first dequeue packets from ring of primary process */ 468 const uint16_t nb_in_deq = rte_ring_dequeue_burst(ring, 469 (void *)rxtx_bufs, BURST_SIZE, NULL); 470 stats->dequeue_pkts += nb_in_deq; 471 472 if (nb_in_deq) { 473 /* then sent on vdev */ 474 uint16_t nb_in_txd = rte_eth_tx_burst( 475 vdev_id, 476 0, rxtx_bufs, nb_in_deq); 477 stats->tx_pkts += nb_in_txd; 478 479 if (unlikely(nb_in_txd < nb_in_deq)) { 480 do { 481 rte_pktmbuf_free(rxtx_bufs[nb_in_txd]); 482 stats->freed_pkts++; 483 } while (++nb_in_txd < nb_in_deq); 484 } 485 } 486 } 487 488 static void 489 free_ring_data(struct rte_ring *ring, uint16_t vdev_id, 490 struct pdump_stats *stats) 491 { 492 while (rte_ring_count(ring)) 493 pdump_rxtx(ring, vdev_id, stats); 494 } 495 496 static void 497 cleanup_rings(void) 498 { 499 int i; 500 struct pdump_tuples *pt; 501 502 for (i = 0; i < num_tuples; i++) { 503 pt = &pdump_t[i]; 504 505 if (pt->device_id) 506 free(pt->device_id); 507 508 /* free the rings */ 509 if (pt->rx_ring) 510 rte_ring_free(pt->rx_ring); 511 if (pt->tx_ring) 512 rte_ring_free(pt->tx_ring); 513 } 514 } 515 516 static void 517 cleanup_pdump_resources(void) 518 { 519 int i; 520 struct pdump_tuples *pt; 521 char name[RTE_ETH_NAME_MAX_LEN]; 522 523 /* disable pdump and free the pdump_tuple resources */ 524 for (i = 0; i < num_tuples; i++) { 525 pt = &pdump_t[i]; 526 527 /* remove callbacks */ 528 disable_pdump(pt); 529 530 /* 531 * transmit rest of the enqueued packets of the rings on to 532 * the vdev, in order to release mbufs to the mepool. 533 **/ 534 if (pt->dir & RTE_PDUMP_FLAG_RX) 535 free_ring_data(pt->rx_ring, pt->rx_vdev_id, &pt->stats); 536 if (pt->dir & RTE_PDUMP_FLAG_TX) 537 free_ring_data(pt->tx_ring, pt->tx_vdev_id, &pt->stats); 538 539 /* Remove the vdev(s) created */ 540 if (pt->dir & RTE_PDUMP_FLAG_RX) { 541 rte_eth_dev_get_name_by_port(pt->rx_vdev_id, name); 542 rte_eal_hotplug_remove("vdev", name); 543 } 544 545 if (pt->single_pdump_dev) 546 continue; 547 548 if (pt->dir & RTE_PDUMP_FLAG_TX) { 549 rte_eth_dev_get_name_by_port(pt->tx_vdev_id, name); 550 rte_eal_hotplug_remove("vdev", name); 551 } 552 553 } 554 cleanup_rings(); 555 } 556 557 static void 558 disable_primary_monitor(void) 559 { 560 int ret; 561 562 /* 563 * Cancel monitoring of primary process. 564 * There will be no error if no alarm is set 565 * (in case primary process kill was detected earlier). 566 */ 567 ret = rte_eal_alarm_cancel(monitor_primary, NULL); 568 if (ret < 0) 569 printf("Fail to disable monitor:%d\n", ret); 570 } 571 572 static void 573 signal_handler(int sig_num) 574 { 575 if (sig_num == SIGINT) { 576 printf("\n\nSignal %d received, preparing to exit...\n", 577 sig_num); 578 quit_signal = 1; 579 } 580 } 581 582 static inline int 583 configure_vdev(uint16_t port_id) 584 { 585 struct rte_ether_addr addr; 586 const uint16_t rxRings = 0, txRings = 1; 587 int ret; 588 uint16_t q; 589 590 if (!rte_eth_dev_is_valid_port(port_id)) 591 return -1; 592 593 ret = rte_eth_dev_configure(port_id, rxRings, txRings, 594 &port_conf_default); 595 if (ret != 0) 596 rte_exit(EXIT_FAILURE, "dev config failed\n"); 597 598 for (q = 0; q < txRings; q++) { 599 ret = rte_eth_tx_queue_setup(port_id, q, TX_DESC_PER_QUEUE, 600 rte_eth_dev_socket_id(port_id), NULL); 601 if (ret < 0) 602 rte_exit(EXIT_FAILURE, "queue setup failed\n"); 603 } 604 605 ret = rte_eth_dev_start(port_id); 606 if (ret < 0) 607 rte_exit(EXIT_FAILURE, "dev start failed\n"); 608 609 ret = rte_eth_macaddr_get(port_id, &addr); 610 if (ret != 0) 611 rte_exit(EXIT_FAILURE, "macaddr get failed\n"); 612 613 printf("Port %u MAC: %02"PRIx8" %02"PRIx8" %02"PRIx8 614 " %02"PRIx8" %02"PRIx8" %02"PRIx8"\n", 615 port_id, 616 addr.addr_bytes[0], addr.addr_bytes[1], 617 addr.addr_bytes[2], addr.addr_bytes[3], 618 addr.addr_bytes[4], addr.addr_bytes[5]); 619 620 ret = rte_eth_promiscuous_enable(port_id); 621 if (ret != 0) { 622 rte_exit(EXIT_FAILURE, 623 "promiscuous mode enable failed: %s\n", 624 rte_strerror(-ret)); 625 return ret; 626 } 627 628 return 0; 629 } 630 631 static void 632 create_mp_ring_vdev(void) 633 { 634 int i; 635 uint16_t portid; 636 struct pdump_tuples *pt = NULL; 637 struct rte_mempool *mbuf_pool = NULL; 638 char vdev_name[SIZE]; 639 char vdev_args[SIZE]; 640 char ring_name[SIZE]; 641 char mempool_name[SIZE]; 642 643 for (i = 0; i < num_tuples; i++) { 644 pt = &pdump_t[i]; 645 snprintf(mempool_name, SIZE, MP_NAME, i); 646 mbuf_pool = rte_mempool_lookup(mempool_name); 647 if (mbuf_pool == NULL) { 648 /* create mempool */ 649 mbuf_pool = rte_pktmbuf_pool_create_by_ops(mempool_name, 650 pt->total_num_mbufs, 651 MBUF_POOL_CACHE_SIZE, 0, 652 pt->mbuf_data_size, 653 rte_socket_id(), "ring_mp_mc"); 654 if (mbuf_pool == NULL) { 655 cleanup_rings(); 656 rte_exit(EXIT_FAILURE, 657 "Mempool creation failed: %s\n", 658 rte_strerror(rte_errno)); 659 } 660 } 661 pt->mp = mbuf_pool; 662 663 if (pt->dir == RTE_PDUMP_FLAG_RXTX) { 664 /* if captured packets has to send to the same vdev */ 665 /* create rx_ring */ 666 snprintf(ring_name, SIZE, RX_RING, i); 667 pt->rx_ring = rte_ring_create(ring_name, pt->ring_size, 668 rte_socket_id(), 0); 669 if (pt->rx_ring == NULL) { 670 cleanup_rings(); 671 rte_exit(EXIT_FAILURE, "%s:%s:%d\n", 672 rte_strerror(rte_errno), 673 __func__, __LINE__); 674 } 675 676 /* create tx_ring */ 677 snprintf(ring_name, SIZE, TX_RING, i); 678 pt->tx_ring = rte_ring_create(ring_name, pt->ring_size, 679 rte_socket_id(), 0); 680 if (pt->tx_ring == NULL) { 681 cleanup_rings(); 682 rte_exit(EXIT_FAILURE, "%s:%s:%d\n", 683 rte_strerror(rte_errno), 684 __func__, __LINE__); 685 } 686 687 /* create vdevs */ 688 snprintf(vdev_name, sizeof(vdev_name), 689 VDEV_NAME_FMT, RX_STR, i); 690 (pt->rx_vdev_stream_type == IFACE) ? 691 snprintf(vdev_args, sizeof(vdev_args), 692 VDEV_IFACE_ARGS_FMT, pt->rx_dev) : 693 snprintf(vdev_args, sizeof(vdev_args), 694 VDEV_PCAP_ARGS_FMT, pt->rx_dev); 695 if (rte_eal_hotplug_add("vdev", vdev_name, 696 vdev_args) < 0) { 697 cleanup_rings(); 698 rte_exit(EXIT_FAILURE, 699 "vdev creation failed:%s:%d\n", 700 __func__, __LINE__); 701 } 702 if (rte_eth_dev_get_port_by_name(vdev_name, 703 &portid) != 0) { 704 rte_eal_hotplug_remove("vdev", vdev_name); 705 cleanup_rings(); 706 rte_exit(EXIT_FAILURE, 707 "cannot find added vdev %s:%s:%d\n", 708 vdev_name, __func__, __LINE__); 709 } 710 pt->rx_vdev_id = portid; 711 712 /* configure vdev */ 713 configure_vdev(pt->rx_vdev_id); 714 715 if (pt->single_pdump_dev) 716 pt->tx_vdev_id = portid; 717 else { 718 snprintf(vdev_name, sizeof(vdev_name), 719 VDEV_NAME_FMT, TX_STR, i); 720 (pt->rx_vdev_stream_type == IFACE) ? 721 snprintf(vdev_args, sizeof(vdev_args), 722 VDEV_IFACE_ARGS_FMT, pt->tx_dev) : 723 snprintf(vdev_args, sizeof(vdev_args), 724 VDEV_PCAP_ARGS_FMT, pt->tx_dev); 725 if (rte_eal_hotplug_add("vdev", vdev_name, 726 vdev_args) < 0) { 727 cleanup_rings(); 728 rte_exit(EXIT_FAILURE, 729 "vdev creation failed:" 730 "%s:%d\n", __func__, __LINE__); 731 } 732 if (rte_eth_dev_get_port_by_name(vdev_name, 733 &portid) != 0) { 734 rte_eal_hotplug_remove("vdev", 735 vdev_name); 736 cleanup_rings(); 737 rte_exit(EXIT_FAILURE, 738 "cannot find added vdev %s:%s:%d\n", 739 vdev_name, __func__, __LINE__); 740 } 741 pt->tx_vdev_id = portid; 742 743 /* configure vdev */ 744 configure_vdev(pt->tx_vdev_id); 745 } 746 } else if (pt->dir == RTE_PDUMP_FLAG_RX) { 747 748 /* create rx_ring */ 749 snprintf(ring_name, SIZE, RX_RING, i); 750 pt->rx_ring = rte_ring_create(ring_name, pt->ring_size, 751 rte_socket_id(), 0); 752 if (pt->rx_ring == NULL) { 753 cleanup_rings(); 754 rte_exit(EXIT_FAILURE, "%s\n", 755 rte_strerror(rte_errno)); 756 } 757 758 snprintf(vdev_name, sizeof(vdev_name), 759 VDEV_NAME_FMT, RX_STR, i); 760 (pt->rx_vdev_stream_type == IFACE) ? 761 snprintf(vdev_args, sizeof(vdev_args), 762 VDEV_IFACE_ARGS_FMT, pt->rx_dev) : 763 snprintf(vdev_args, sizeof(vdev_args), 764 VDEV_PCAP_ARGS_FMT, pt->rx_dev); 765 if (rte_eal_hotplug_add("vdev", vdev_name, 766 vdev_args) < 0) { 767 cleanup_rings(); 768 rte_exit(EXIT_FAILURE, 769 "vdev creation failed:%s:%d\n", 770 __func__, __LINE__); 771 } 772 if (rte_eth_dev_get_port_by_name(vdev_name, 773 &portid) != 0) { 774 rte_eal_hotplug_remove("vdev", vdev_name); 775 cleanup_rings(); 776 rte_exit(EXIT_FAILURE, 777 "cannot find added vdev %s:%s:%d\n", 778 vdev_name, __func__, __LINE__); 779 } 780 pt->rx_vdev_id = portid; 781 /* configure vdev */ 782 configure_vdev(pt->rx_vdev_id); 783 } else if (pt->dir == RTE_PDUMP_FLAG_TX) { 784 785 /* create tx_ring */ 786 snprintf(ring_name, SIZE, TX_RING, i); 787 pt->tx_ring = rte_ring_create(ring_name, pt->ring_size, 788 rte_socket_id(), 0); 789 if (pt->tx_ring == NULL) { 790 cleanup_rings(); 791 rte_exit(EXIT_FAILURE, "%s\n", 792 rte_strerror(rte_errno)); 793 } 794 795 snprintf(vdev_name, sizeof(vdev_name), 796 VDEV_NAME_FMT, TX_STR, i); 797 (pt->tx_vdev_stream_type == IFACE) ? 798 snprintf(vdev_args, sizeof(vdev_args), 799 VDEV_IFACE_ARGS_FMT, pt->tx_dev) : 800 snprintf(vdev_args, sizeof(vdev_args), 801 VDEV_PCAP_ARGS_FMT, pt->tx_dev); 802 if (rte_eal_hotplug_add("vdev", vdev_name, 803 vdev_args) < 0) { 804 cleanup_rings(); 805 rte_exit(EXIT_FAILURE, 806 "vdev creation failed\n"); 807 } 808 if (rte_eth_dev_get_port_by_name(vdev_name, 809 &portid) != 0) { 810 rte_eal_hotplug_remove("vdev", vdev_name); 811 cleanup_rings(); 812 rte_exit(EXIT_FAILURE, 813 "cannot find added vdev %s:%s:%d\n", 814 vdev_name, __func__, __LINE__); 815 } 816 pt->tx_vdev_id = portid; 817 818 /* configure vdev */ 819 configure_vdev(pt->tx_vdev_id); 820 } 821 } 822 } 823 824 static void 825 enable_pdump(void) 826 { 827 int i; 828 struct pdump_tuples *pt; 829 int ret = 0, ret1 = 0; 830 831 for (i = 0; i < num_tuples; i++) { 832 pt = &pdump_t[i]; 833 if (pt->dir == RTE_PDUMP_FLAG_RXTX) { 834 if (pt->dump_by_type == DEVICE_ID) { 835 ret = rte_pdump_enable_by_deviceid( 836 pt->device_id, 837 pt->queue, 838 RTE_PDUMP_FLAG_RX, 839 pt->rx_ring, 840 pt->mp, NULL); 841 ret1 = rte_pdump_enable_by_deviceid( 842 pt->device_id, 843 pt->queue, 844 RTE_PDUMP_FLAG_TX, 845 pt->tx_ring, 846 pt->mp, NULL); 847 } else if (pt->dump_by_type == PORT_ID) { 848 ret = rte_pdump_enable(pt->port, pt->queue, 849 RTE_PDUMP_FLAG_RX, 850 pt->rx_ring, pt->mp, NULL); 851 ret1 = rte_pdump_enable(pt->port, pt->queue, 852 RTE_PDUMP_FLAG_TX, 853 pt->tx_ring, pt->mp, NULL); 854 } 855 } else if (pt->dir == RTE_PDUMP_FLAG_RX) { 856 if (pt->dump_by_type == DEVICE_ID) 857 ret = rte_pdump_enable_by_deviceid( 858 pt->device_id, 859 pt->queue, 860 pt->dir, pt->rx_ring, 861 pt->mp, NULL); 862 else if (pt->dump_by_type == PORT_ID) 863 ret = rte_pdump_enable(pt->port, pt->queue, 864 pt->dir, 865 pt->rx_ring, pt->mp, NULL); 866 } else if (pt->dir == RTE_PDUMP_FLAG_TX) { 867 if (pt->dump_by_type == DEVICE_ID) 868 ret = rte_pdump_enable_by_deviceid( 869 pt->device_id, 870 pt->queue, 871 pt->dir, 872 pt->tx_ring, pt->mp, NULL); 873 else if (pt->dump_by_type == PORT_ID) 874 ret = rte_pdump_enable(pt->port, pt->queue, 875 pt->dir, 876 pt->tx_ring, pt->mp, NULL); 877 } 878 if (ret < 0 || ret1 < 0) { 879 cleanup_pdump_resources(); 880 rte_exit(EXIT_FAILURE, "%s\n", rte_strerror(rte_errno)); 881 } 882 } 883 } 884 885 static inline void 886 pdump_packets(struct pdump_tuples *pt) 887 { 888 if (pt->dir & RTE_PDUMP_FLAG_RX) 889 pdump_rxtx(pt->rx_ring, pt->rx_vdev_id, &pt->stats); 890 if (pt->dir & RTE_PDUMP_FLAG_TX) 891 pdump_rxtx(pt->tx_ring, pt->tx_vdev_id, &pt->stats); 892 } 893 894 static int 895 dump_packets_core(void *arg) 896 { 897 struct pdump_tuples *pt = (struct pdump_tuples *) arg; 898 899 printf(" core (%u); port %u device (%s) queue %u\n", 900 rte_lcore_id(), pt->port, pt->device_id, pt->queue); 901 fflush(stdout); 902 903 while (!quit_signal) 904 pdump_packets(pt); 905 906 return 0; 907 } 908 909 static inline void 910 dump_packets(void) 911 { 912 int i; 913 uint32_t lcore_id = 0; 914 915 if (!multiple_core_capture) { 916 printf(" core (%u), capture for (%d) tuples\n", 917 rte_lcore_id(), num_tuples); 918 919 for (i = 0; i < num_tuples; i++) 920 printf(" - port %u device (%s) queue %u\n", 921 pdump_t[i].port, 922 pdump_t[i].device_id, 923 pdump_t[i].queue); 924 925 while (!quit_signal) { 926 for (i = 0; i < num_tuples; i++) 927 pdump_packets(&pdump_t[i]); 928 } 929 930 return; 931 } 932 933 /* check if there enough core */ 934 if ((uint32_t)num_tuples >= rte_lcore_count()) { 935 printf("Insufficient cores to run parallel!\n"); 936 return; 937 } 938 939 lcore_id = rte_get_next_lcore(lcore_id, 1, 0); 940 941 for (i = 0; i < num_tuples; i++) { 942 rte_eal_remote_launch(dump_packets_core, 943 &pdump_t[i], lcore_id); 944 lcore_id = rte_get_next_lcore(lcore_id, 1, 0); 945 946 if (rte_eal_wait_lcore(lcore_id) < 0) 947 rte_exit(EXIT_FAILURE, "failed to wait\n"); 948 } 949 950 /* master core */ 951 while (!quit_signal) 952 ; 953 } 954 955 static void 956 enable_primary_monitor(void) 957 { 958 int ret; 959 960 /* Once primary exits, so will pdump. */ 961 ret = rte_eal_alarm_set(MONITOR_INTERVAL, monitor_primary, NULL); 962 if (ret < 0) 963 printf("Fail to enable monitor:%d\n", ret); 964 } 965 966 int 967 main(int argc, char **argv) 968 { 969 int diag; 970 int ret; 971 int i; 972 973 char n_flag[] = "-n4"; 974 char mp_flag[] = "--proc-type=secondary"; 975 char *argp[argc + 2]; 976 977 /* catch ctrl-c so we can print on exit */ 978 signal(SIGINT, signal_handler); 979 980 argp[0] = argv[0]; 981 argp[1] = n_flag; 982 argp[2] = mp_flag; 983 984 for (i = 1; i < argc; i++) 985 argp[i + 2] = argv[i]; 986 987 argc += 2; 988 989 diag = rte_eal_init(argc, argp); 990 if (diag < 0) 991 rte_panic("Cannot init EAL\n"); 992 993 if (rte_eth_dev_count_avail() == 0) 994 rte_exit(EXIT_FAILURE, "No Ethernet ports - bye\n"); 995 996 argc -= diag; 997 argv += (diag - 2); 998 999 /* parse app arguments */ 1000 if (argc > 1) { 1001 ret = launch_args_parse(argc, argv, argp[0]); 1002 if (ret < 0) 1003 rte_exit(EXIT_FAILURE, "Invalid argument\n"); 1004 } 1005 1006 /* create mempool, ring and vdevs info */ 1007 create_mp_ring_vdev(); 1008 enable_pdump(); 1009 enable_primary_monitor(); 1010 dump_packets(); 1011 1012 disable_primary_monitor(); 1013 cleanup_pdump_resources(); 1014 /* dump debug stats */ 1015 print_pdump_stats(); 1016 1017 ret = rte_eal_cleanup(); 1018 if (ret) 1019 printf("Error from rte_eal_cleanup(), %d\n", ret); 1020 1021 return 0; 1022 } 1023