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 unsigned int drops = nb_in_deq - nb_in_txd; 481 482 rte_pktmbuf_free_bulk(&rxtx_bufs[nb_in_txd], drops); 483 stats->freed_pkts += drops; 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 free(pt->device_id); 506 507 /* free the rings */ 508 rte_ring_free(pt->rx_ring); 509 rte_ring_free(pt->tx_ring); 510 } 511 } 512 513 static void 514 cleanup_pdump_resources(void) 515 { 516 int i; 517 struct pdump_tuples *pt; 518 char name[RTE_ETH_NAME_MAX_LEN]; 519 520 /* disable pdump and free the pdump_tuple resources */ 521 for (i = 0; i < num_tuples; i++) { 522 pt = &pdump_t[i]; 523 524 /* remove callbacks */ 525 disable_pdump(pt); 526 527 /* 528 * transmit rest of the enqueued packets of the rings on to 529 * the vdev, in order to release mbufs to the mepool. 530 **/ 531 if (pt->dir & RTE_PDUMP_FLAG_RX) 532 free_ring_data(pt->rx_ring, pt->rx_vdev_id, &pt->stats); 533 if (pt->dir & RTE_PDUMP_FLAG_TX) 534 free_ring_data(pt->tx_ring, pt->tx_vdev_id, &pt->stats); 535 536 /* Remove the vdev(s) created */ 537 if (pt->dir & RTE_PDUMP_FLAG_RX) { 538 rte_eth_dev_get_name_by_port(pt->rx_vdev_id, name); 539 rte_eal_hotplug_remove("vdev", name); 540 } 541 542 if (pt->single_pdump_dev) 543 continue; 544 545 if (pt->dir & RTE_PDUMP_FLAG_TX) { 546 rte_eth_dev_get_name_by_port(pt->tx_vdev_id, name); 547 rte_eal_hotplug_remove("vdev", name); 548 } 549 550 } 551 cleanup_rings(); 552 } 553 554 static void 555 disable_primary_monitor(void) 556 { 557 int ret; 558 559 /* 560 * Cancel monitoring of primary process. 561 * There will be no error if no alarm is set 562 * (in case primary process kill was detected earlier). 563 */ 564 ret = rte_eal_alarm_cancel(monitor_primary, NULL); 565 if (ret < 0) 566 printf("Fail to disable monitor:%d\n", ret); 567 } 568 569 static void 570 signal_handler(int sig_num) 571 { 572 if (sig_num == SIGINT) { 573 printf("\n\nSignal %d received, preparing to exit...\n", 574 sig_num); 575 quit_signal = 1; 576 } 577 } 578 579 static inline int 580 configure_vdev(uint16_t port_id) 581 { 582 struct rte_ether_addr addr; 583 const uint16_t rxRings = 0, txRings = 1; 584 int ret; 585 uint16_t q; 586 587 if (!rte_eth_dev_is_valid_port(port_id)) 588 return -1; 589 590 ret = rte_eth_dev_configure(port_id, rxRings, txRings, 591 &port_conf_default); 592 if (ret != 0) 593 rte_exit(EXIT_FAILURE, "dev config failed\n"); 594 595 for (q = 0; q < txRings; q++) { 596 ret = rte_eth_tx_queue_setup(port_id, q, TX_DESC_PER_QUEUE, 597 rte_eth_dev_socket_id(port_id), NULL); 598 if (ret < 0) 599 rte_exit(EXIT_FAILURE, "queue setup failed\n"); 600 } 601 602 ret = rte_eth_dev_start(port_id); 603 if (ret < 0) 604 rte_exit(EXIT_FAILURE, "dev start failed\n"); 605 606 ret = rte_eth_macaddr_get(port_id, &addr); 607 if (ret != 0) 608 rte_exit(EXIT_FAILURE, "macaddr get failed\n"); 609 610 printf("Port %u MAC: %02"PRIx8" %02"PRIx8" %02"PRIx8 611 " %02"PRIx8" %02"PRIx8" %02"PRIx8"\n", 612 port_id, RTE_ETHER_ADDR_BYTES(&addr)); 613 614 ret = rte_eth_promiscuous_enable(port_id); 615 if (ret != 0) { 616 rte_exit(EXIT_FAILURE, 617 "promiscuous mode enable failed: %s\n", 618 rte_strerror(-ret)); 619 return ret; 620 } 621 622 return 0; 623 } 624 625 static void 626 create_mp_ring_vdev(void) 627 { 628 int i; 629 uint16_t portid; 630 struct pdump_tuples *pt = NULL; 631 struct rte_mempool *mbuf_pool = NULL; 632 char vdev_name[SIZE]; 633 char vdev_args[SIZE]; 634 char ring_name[SIZE]; 635 char mempool_name[SIZE]; 636 637 for (i = 0; i < num_tuples; i++) { 638 pt = &pdump_t[i]; 639 snprintf(mempool_name, SIZE, MP_NAME, i); 640 mbuf_pool = rte_mempool_lookup(mempool_name); 641 if (mbuf_pool == NULL) { 642 /* create mempool */ 643 mbuf_pool = rte_pktmbuf_pool_create_by_ops(mempool_name, 644 pt->total_num_mbufs, 645 MBUF_POOL_CACHE_SIZE, 0, 646 pt->mbuf_data_size, 647 rte_socket_id(), "ring_mp_mc"); 648 if (mbuf_pool == NULL) { 649 cleanup_rings(); 650 rte_exit(EXIT_FAILURE, 651 "Mempool creation failed: %s\n", 652 rte_strerror(rte_errno)); 653 } 654 } 655 pt->mp = mbuf_pool; 656 657 if (pt->dir == RTE_PDUMP_FLAG_RXTX) { 658 /* if captured packets has to send to the same vdev */ 659 /* create rx_ring */ 660 snprintf(ring_name, SIZE, RX_RING, i); 661 pt->rx_ring = rte_ring_create(ring_name, pt->ring_size, 662 rte_socket_id(), 0); 663 if (pt->rx_ring == NULL) { 664 cleanup_rings(); 665 rte_exit(EXIT_FAILURE, "%s:%s:%d\n", 666 rte_strerror(rte_errno), 667 __func__, __LINE__); 668 } 669 670 /* create tx_ring */ 671 snprintf(ring_name, SIZE, TX_RING, i); 672 pt->tx_ring = rte_ring_create(ring_name, pt->ring_size, 673 rte_socket_id(), 0); 674 if (pt->tx_ring == NULL) { 675 cleanup_rings(); 676 rte_exit(EXIT_FAILURE, "%s:%s:%d\n", 677 rte_strerror(rte_errno), 678 __func__, __LINE__); 679 } 680 681 /* create vdevs */ 682 snprintf(vdev_name, sizeof(vdev_name), 683 VDEV_NAME_FMT, RX_STR, i); 684 (pt->rx_vdev_stream_type == IFACE) ? 685 snprintf(vdev_args, sizeof(vdev_args), 686 VDEV_IFACE_ARGS_FMT, pt->rx_dev) : 687 snprintf(vdev_args, sizeof(vdev_args), 688 VDEV_PCAP_ARGS_FMT, pt->rx_dev); 689 if (rte_eal_hotplug_add("vdev", vdev_name, 690 vdev_args) < 0) { 691 cleanup_rings(); 692 rte_exit(EXIT_FAILURE, 693 "vdev creation failed:%s:%d\n", 694 __func__, __LINE__); 695 } 696 if (rte_eth_dev_get_port_by_name(vdev_name, 697 &portid) != 0) { 698 rte_eal_hotplug_remove("vdev", vdev_name); 699 cleanup_rings(); 700 rte_exit(EXIT_FAILURE, 701 "cannot find added vdev %s:%s:%d\n", 702 vdev_name, __func__, __LINE__); 703 } 704 pt->rx_vdev_id = portid; 705 706 /* configure vdev */ 707 configure_vdev(pt->rx_vdev_id); 708 709 if (pt->single_pdump_dev) 710 pt->tx_vdev_id = portid; 711 else { 712 snprintf(vdev_name, sizeof(vdev_name), 713 VDEV_NAME_FMT, TX_STR, i); 714 (pt->rx_vdev_stream_type == IFACE) ? 715 snprintf(vdev_args, sizeof(vdev_args), 716 VDEV_IFACE_ARGS_FMT, pt->tx_dev) : 717 snprintf(vdev_args, sizeof(vdev_args), 718 VDEV_PCAP_ARGS_FMT, pt->tx_dev); 719 if (rte_eal_hotplug_add("vdev", vdev_name, 720 vdev_args) < 0) { 721 cleanup_rings(); 722 rte_exit(EXIT_FAILURE, 723 "vdev creation failed:" 724 "%s:%d\n", __func__, __LINE__); 725 } 726 if (rte_eth_dev_get_port_by_name(vdev_name, 727 &portid) != 0) { 728 rte_eal_hotplug_remove("vdev", 729 vdev_name); 730 cleanup_rings(); 731 rte_exit(EXIT_FAILURE, 732 "cannot find added vdev %s:%s:%d\n", 733 vdev_name, __func__, __LINE__); 734 } 735 pt->tx_vdev_id = portid; 736 737 /* configure vdev */ 738 configure_vdev(pt->tx_vdev_id); 739 } 740 } else if (pt->dir == RTE_PDUMP_FLAG_RX) { 741 742 /* create rx_ring */ 743 snprintf(ring_name, SIZE, RX_RING, i); 744 pt->rx_ring = rte_ring_create(ring_name, pt->ring_size, 745 rte_socket_id(), 0); 746 if (pt->rx_ring == NULL) { 747 cleanup_rings(); 748 rte_exit(EXIT_FAILURE, "%s\n", 749 rte_strerror(rte_errno)); 750 } 751 752 snprintf(vdev_name, sizeof(vdev_name), 753 VDEV_NAME_FMT, RX_STR, i); 754 (pt->rx_vdev_stream_type == IFACE) ? 755 snprintf(vdev_args, sizeof(vdev_args), 756 VDEV_IFACE_ARGS_FMT, pt->rx_dev) : 757 snprintf(vdev_args, sizeof(vdev_args), 758 VDEV_PCAP_ARGS_FMT, pt->rx_dev); 759 if (rte_eal_hotplug_add("vdev", vdev_name, 760 vdev_args) < 0) { 761 cleanup_rings(); 762 rte_exit(EXIT_FAILURE, 763 "vdev creation failed:%s:%d\n", 764 __func__, __LINE__); 765 } 766 if (rte_eth_dev_get_port_by_name(vdev_name, 767 &portid) != 0) { 768 rte_eal_hotplug_remove("vdev", vdev_name); 769 cleanup_rings(); 770 rte_exit(EXIT_FAILURE, 771 "cannot find added vdev %s:%s:%d\n", 772 vdev_name, __func__, __LINE__); 773 } 774 pt->rx_vdev_id = portid; 775 /* configure vdev */ 776 configure_vdev(pt->rx_vdev_id); 777 } else if (pt->dir == RTE_PDUMP_FLAG_TX) { 778 779 /* create tx_ring */ 780 snprintf(ring_name, SIZE, TX_RING, i); 781 pt->tx_ring = rte_ring_create(ring_name, pt->ring_size, 782 rte_socket_id(), 0); 783 if (pt->tx_ring == NULL) { 784 cleanup_rings(); 785 rte_exit(EXIT_FAILURE, "%s\n", 786 rte_strerror(rte_errno)); 787 } 788 789 snprintf(vdev_name, sizeof(vdev_name), 790 VDEV_NAME_FMT, TX_STR, i); 791 (pt->tx_vdev_stream_type == IFACE) ? 792 snprintf(vdev_args, sizeof(vdev_args), 793 VDEV_IFACE_ARGS_FMT, pt->tx_dev) : 794 snprintf(vdev_args, sizeof(vdev_args), 795 VDEV_PCAP_ARGS_FMT, pt->tx_dev); 796 if (rte_eal_hotplug_add("vdev", vdev_name, 797 vdev_args) < 0) { 798 cleanup_rings(); 799 rte_exit(EXIT_FAILURE, 800 "vdev creation failed\n"); 801 } 802 if (rte_eth_dev_get_port_by_name(vdev_name, 803 &portid) != 0) { 804 rte_eal_hotplug_remove("vdev", vdev_name); 805 cleanup_rings(); 806 rte_exit(EXIT_FAILURE, 807 "cannot find added vdev %s:%s:%d\n", 808 vdev_name, __func__, __LINE__); 809 } 810 pt->tx_vdev_id = portid; 811 812 /* configure vdev */ 813 configure_vdev(pt->tx_vdev_id); 814 } 815 } 816 } 817 818 static void 819 enable_pdump(void) 820 { 821 int i; 822 struct pdump_tuples *pt; 823 int ret = 0, ret1 = 0; 824 825 for (i = 0; i < num_tuples; i++) { 826 pt = &pdump_t[i]; 827 if (pt->dir == RTE_PDUMP_FLAG_RXTX) { 828 if (pt->dump_by_type == DEVICE_ID) { 829 ret = rte_pdump_enable_by_deviceid( 830 pt->device_id, 831 pt->queue, 832 RTE_PDUMP_FLAG_RX, 833 pt->rx_ring, 834 pt->mp, NULL); 835 ret1 = rte_pdump_enable_by_deviceid( 836 pt->device_id, 837 pt->queue, 838 RTE_PDUMP_FLAG_TX, 839 pt->tx_ring, 840 pt->mp, NULL); 841 } else if (pt->dump_by_type == PORT_ID) { 842 ret = rte_pdump_enable(pt->port, pt->queue, 843 RTE_PDUMP_FLAG_RX, 844 pt->rx_ring, pt->mp, NULL); 845 ret1 = rte_pdump_enable(pt->port, pt->queue, 846 RTE_PDUMP_FLAG_TX, 847 pt->tx_ring, pt->mp, NULL); 848 } 849 } else if (pt->dir == RTE_PDUMP_FLAG_RX) { 850 if (pt->dump_by_type == DEVICE_ID) 851 ret = rte_pdump_enable_by_deviceid( 852 pt->device_id, 853 pt->queue, 854 pt->dir, pt->rx_ring, 855 pt->mp, NULL); 856 else if (pt->dump_by_type == PORT_ID) 857 ret = rte_pdump_enable(pt->port, pt->queue, 858 pt->dir, 859 pt->rx_ring, pt->mp, NULL); 860 } else if (pt->dir == RTE_PDUMP_FLAG_TX) { 861 if (pt->dump_by_type == DEVICE_ID) 862 ret = rte_pdump_enable_by_deviceid( 863 pt->device_id, 864 pt->queue, 865 pt->dir, 866 pt->tx_ring, pt->mp, NULL); 867 else if (pt->dump_by_type == PORT_ID) 868 ret = rte_pdump_enable(pt->port, pt->queue, 869 pt->dir, 870 pt->tx_ring, pt->mp, NULL); 871 } 872 if (ret < 0 || ret1 < 0) { 873 cleanup_pdump_resources(); 874 rte_exit(EXIT_FAILURE, "%s\n", rte_strerror(rte_errno)); 875 } 876 } 877 } 878 879 static inline void 880 pdump_packets(struct pdump_tuples *pt) 881 { 882 if (pt->dir & RTE_PDUMP_FLAG_RX) 883 pdump_rxtx(pt->rx_ring, pt->rx_vdev_id, &pt->stats); 884 if (pt->dir & RTE_PDUMP_FLAG_TX) 885 pdump_rxtx(pt->tx_ring, pt->tx_vdev_id, &pt->stats); 886 } 887 888 static int 889 dump_packets_core(void *arg) 890 { 891 struct pdump_tuples *pt = (struct pdump_tuples *) arg; 892 893 printf(" core (%u); port %u device (%s) queue %u\n", 894 rte_lcore_id(), pt->port, pt->device_id, pt->queue); 895 fflush(stdout); 896 897 while (!quit_signal) 898 pdump_packets(pt); 899 900 return 0; 901 } 902 903 static inline void 904 dump_packets(void) 905 { 906 int i; 907 uint32_t lcore_id = 0; 908 909 if (!multiple_core_capture) { 910 printf(" core (%u), capture for (%d) tuples\n", 911 rte_lcore_id(), num_tuples); 912 913 for (i = 0; i < num_tuples; i++) 914 printf(" - port %u device (%s) queue %u\n", 915 pdump_t[i].port, 916 pdump_t[i].device_id, 917 pdump_t[i].queue); 918 919 while (!quit_signal) { 920 for (i = 0; i < num_tuples; i++) 921 pdump_packets(&pdump_t[i]); 922 } 923 924 return; 925 } 926 927 /* check if there enough core */ 928 if ((uint32_t)num_tuples >= rte_lcore_count()) { 929 printf("Insufficient cores to run parallel!\n"); 930 return; 931 } 932 933 lcore_id = rte_get_next_lcore(lcore_id, 1, 0); 934 935 for (i = 0; i < num_tuples; i++) { 936 rte_eal_remote_launch(dump_packets_core, 937 &pdump_t[i], lcore_id); 938 lcore_id = rte_get_next_lcore(lcore_id, 1, 0); 939 940 if (rte_eal_wait_lcore(lcore_id) < 0) 941 rte_exit(EXIT_FAILURE, "failed to wait\n"); 942 } 943 944 /* main core */ 945 while (!quit_signal) 946 ; 947 } 948 949 static void 950 enable_primary_monitor(void) 951 { 952 int ret; 953 954 /* Once primary exits, so will pdump. */ 955 ret = rte_eal_alarm_set(MONITOR_INTERVAL, monitor_primary, NULL); 956 if (ret < 0) 957 printf("Fail to enable monitor:%d\n", ret); 958 } 959 960 int 961 main(int argc, char **argv) 962 { 963 int diag; 964 int ret; 965 int i; 966 967 char n_flag[] = "-n4"; 968 char mp_flag[] = "--proc-type=secondary"; 969 char *argp[argc + 2]; 970 971 /* catch ctrl-c so we can print on exit */ 972 signal(SIGINT, signal_handler); 973 974 argp[0] = argv[0]; 975 argp[1] = n_flag; 976 argp[2] = mp_flag; 977 978 for (i = 1; i < argc; i++) 979 argp[i + 2] = argv[i]; 980 981 argc += 2; 982 983 diag = rte_eal_init(argc, argp); 984 if (diag < 0) 985 rte_panic("Cannot init EAL\n"); 986 987 if (rte_eth_dev_count_avail() == 0) 988 rte_exit(EXIT_FAILURE, "No Ethernet ports - bye\n"); 989 990 argc -= diag; 991 argv += (diag - 2); 992 993 /* parse app arguments */ 994 if (argc > 1) { 995 ret = launch_args_parse(argc, argv, argp[0]); 996 if (ret < 0) 997 rte_exit(EXIT_FAILURE, "Invalid argument\n"); 998 } 999 1000 /* create mempool, ring and vdevs info */ 1001 create_mp_ring_vdev(); 1002 enable_pdump(); 1003 enable_primary_monitor(); 1004 dump_packets(); 1005 1006 disable_primary_monitor(); 1007 cleanup_pdump_resources(); 1008 /* dump debug stats */ 1009 print_pdump_stats(); 1010 1011 ret = rte_eal_cleanup(); 1012 if (ret) 1013 printf("Error from rte_eal_cleanup(), %d\n", ret); 1014 1015 return 0; 1016 } 1017