xref: /dpdk/app/pdump/main.c (revision 089e5ed727a15da2729cfee9b63533dd120bd04c)
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 	rte_eth_macaddr_get(port_id, &addr);
610 	printf("Port %u MAC: %02"PRIx8" %02"PRIx8" %02"PRIx8
611 			" %02"PRIx8" %02"PRIx8" %02"PRIx8"\n",
612 			port_id,
613 			addr.addr_bytes[0], addr.addr_bytes[1],
614 			addr.addr_bytes[2], addr.addr_bytes[3],
615 			addr.addr_bytes[4], addr.addr_bytes[5]);
616 
617 	rte_eth_promiscuous_enable(port_id);
618 
619 	return 0;
620 }
621 
622 static void
623 create_mp_ring_vdev(void)
624 {
625 	int i;
626 	uint16_t portid;
627 	struct pdump_tuples *pt = NULL;
628 	struct rte_mempool *mbuf_pool = NULL;
629 	char vdev_name[SIZE];
630 	char vdev_args[SIZE];
631 	char ring_name[SIZE];
632 	char mempool_name[SIZE];
633 
634 	for (i = 0; i < num_tuples; i++) {
635 		pt = &pdump_t[i];
636 		snprintf(mempool_name, SIZE, MP_NAME, i);
637 		mbuf_pool = rte_mempool_lookup(mempool_name);
638 		if (mbuf_pool == NULL) {
639 			/* create mempool */
640 			mbuf_pool = rte_pktmbuf_pool_create_by_ops(mempool_name,
641 					pt->total_num_mbufs,
642 					MBUF_POOL_CACHE_SIZE, 0,
643 					pt->mbuf_data_size,
644 					rte_socket_id(), "ring_mp_mc");
645 			if (mbuf_pool == NULL) {
646 				cleanup_rings();
647 				rte_exit(EXIT_FAILURE,
648 					"Mempool creation failed: %s\n",
649 					rte_strerror(rte_errno));
650 			}
651 		}
652 		pt->mp = mbuf_pool;
653 
654 		if (pt->dir == RTE_PDUMP_FLAG_RXTX) {
655 			/* if captured packets has to send to the same vdev */
656 			/* create rx_ring */
657 			snprintf(ring_name, SIZE, RX_RING, i);
658 			pt->rx_ring = rte_ring_create(ring_name, pt->ring_size,
659 					rte_socket_id(), 0);
660 			if (pt->rx_ring == NULL) {
661 				cleanup_rings();
662 				rte_exit(EXIT_FAILURE, "%s:%s:%d\n",
663 						rte_strerror(rte_errno),
664 						__func__, __LINE__);
665 			}
666 
667 			/* create tx_ring */
668 			snprintf(ring_name, SIZE, TX_RING, i);
669 			pt->tx_ring = rte_ring_create(ring_name, pt->ring_size,
670 					rte_socket_id(), 0);
671 			if (pt->tx_ring == NULL) {
672 				cleanup_rings();
673 				rte_exit(EXIT_FAILURE, "%s:%s:%d\n",
674 						rte_strerror(rte_errno),
675 						__func__, __LINE__);
676 			}
677 
678 			/* create vdevs */
679 			snprintf(vdev_name, sizeof(vdev_name),
680 				 VDEV_NAME_FMT, RX_STR, i);
681 			(pt->rx_vdev_stream_type == IFACE) ?
682 			snprintf(vdev_args, sizeof(vdev_args),
683 				 VDEV_IFACE_ARGS_FMT, pt->rx_dev) :
684 			snprintf(vdev_args, sizeof(vdev_args),
685 				 VDEV_PCAP_ARGS_FMT, pt->rx_dev);
686 			if (rte_eal_hotplug_add("vdev", vdev_name,
687 						vdev_args) < 0) {
688 				cleanup_rings();
689 				rte_exit(EXIT_FAILURE,
690 					"vdev creation failed:%s:%d\n",
691 					__func__, __LINE__);
692 			}
693 			if (rte_eth_dev_get_port_by_name(vdev_name,
694 							 &portid) != 0) {
695 				rte_eal_hotplug_remove("vdev", vdev_name);
696 				cleanup_rings();
697 				rte_exit(EXIT_FAILURE,
698 					"cannot find added vdev %s:%s:%d\n",
699 					vdev_name, __func__, __LINE__);
700 			}
701 			pt->rx_vdev_id = portid;
702 
703 			/* configure vdev */
704 			configure_vdev(pt->rx_vdev_id);
705 
706 			if (pt->single_pdump_dev)
707 				pt->tx_vdev_id = portid;
708 			else {
709 				snprintf(vdev_name, sizeof(vdev_name),
710 					 VDEV_NAME_FMT, TX_STR, i);
711 				(pt->rx_vdev_stream_type == IFACE) ?
712 				snprintf(vdev_args, sizeof(vdev_args),
713 					 VDEV_IFACE_ARGS_FMT, pt->tx_dev) :
714 				snprintf(vdev_args, sizeof(vdev_args),
715 					 VDEV_PCAP_ARGS_FMT, pt->tx_dev);
716 				if (rte_eal_hotplug_add("vdev", vdev_name,
717 							vdev_args) < 0) {
718 					cleanup_rings();
719 					rte_exit(EXIT_FAILURE,
720 						"vdev creation failed:"
721 						"%s:%d\n", __func__, __LINE__);
722 				}
723 				if (rte_eth_dev_get_port_by_name(vdev_name,
724 						&portid) != 0) {
725 					rte_eal_hotplug_remove("vdev",
726 							       vdev_name);
727 					cleanup_rings();
728 					rte_exit(EXIT_FAILURE,
729 						"cannot find added vdev %s:%s:%d\n",
730 						vdev_name, __func__, __LINE__);
731 				}
732 				pt->tx_vdev_id = portid;
733 
734 				/* configure vdev */
735 				configure_vdev(pt->tx_vdev_id);
736 			}
737 		} else if (pt->dir == RTE_PDUMP_FLAG_RX) {
738 
739 			/* create rx_ring */
740 			snprintf(ring_name, SIZE, RX_RING, i);
741 			pt->rx_ring = rte_ring_create(ring_name, pt->ring_size,
742 					rte_socket_id(), 0);
743 			if (pt->rx_ring == NULL) {
744 				cleanup_rings();
745 				rte_exit(EXIT_FAILURE, "%s\n",
746 					rte_strerror(rte_errno));
747 			}
748 
749 			snprintf(vdev_name, sizeof(vdev_name),
750 				 VDEV_NAME_FMT, RX_STR, i);
751 			(pt->rx_vdev_stream_type == IFACE) ?
752 			snprintf(vdev_args, sizeof(vdev_args),
753 				 VDEV_IFACE_ARGS_FMT, pt->rx_dev) :
754 			snprintf(vdev_args, sizeof(vdev_args),
755 				 VDEV_PCAP_ARGS_FMT, pt->rx_dev);
756 			if (rte_eal_hotplug_add("vdev", vdev_name,
757 						vdev_args) < 0) {
758 				cleanup_rings();
759 				rte_exit(EXIT_FAILURE,
760 					"vdev creation failed:%s:%d\n",
761 					__func__, __LINE__);
762 			}
763 			if (rte_eth_dev_get_port_by_name(vdev_name,
764 							 &portid) != 0) {
765 				rte_eal_hotplug_remove("vdev", vdev_name);
766 				cleanup_rings();
767 				rte_exit(EXIT_FAILURE,
768 					"cannot find added vdev %s:%s:%d\n",
769 					vdev_name, __func__, __LINE__);
770 			}
771 			pt->rx_vdev_id = portid;
772 			/* configure vdev */
773 			configure_vdev(pt->rx_vdev_id);
774 		} else if (pt->dir == RTE_PDUMP_FLAG_TX) {
775 
776 			/* create tx_ring */
777 			snprintf(ring_name, SIZE, TX_RING, i);
778 			pt->tx_ring = rte_ring_create(ring_name, pt->ring_size,
779 					rte_socket_id(), 0);
780 			if (pt->tx_ring == NULL) {
781 				cleanup_rings();
782 				rte_exit(EXIT_FAILURE, "%s\n",
783 					rte_strerror(rte_errno));
784 			}
785 
786 			snprintf(vdev_name, sizeof(vdev_name),
787 				 VDEV_NAME_FMT, TX_STR, i);
788 			(pt->tx_vdev_stream_type == IFACE) ?
789 			snprintf(vdev_args, sizeof(vdev_args),
790 				 VDEV_IFACE_ARGS_FMT, pt->tx_dev) :
791 			snprintf(vdev_args, sizeof(vdev_args),
792 				 VDEV_PCAP_ARGS_FMT, pt->tx_dev);
793 			if (rte_eal_hotplug_add("vdev", vdev_name,
794 						vdev_args) < 0) {
795 				cleanup_rings();
796 				rte_exit(EXIT_FAILURE,
797 					"vdev creation failed\n");
798 			}
799 			if (rte_eth_dev_get_port_by_name(vdev_name,
800 							 &portid) != 0) {
801 				rte_eal_hotplug_remove("vdev", vdev_name);
802 				cleanup_rings();
803 				rte_exit(EXIT_FAILURE,
804 					"cannot find added vdev %s:%s:%d\n",
805 					vdev_name, __func__, __LINE__);
806 			}
807 			pt->tx_vdev_id = portid;
808 
809 			/* configure vdev */
810 			configure_vdev(pt->tx_vdev_id);
811 		}
812 	}
813 }
814 
815 static void
816 enable_pdump(void)
817 {
818 	int i;
819 	struct pdump_tuples *pt;
820 	int ret = 0, ret1 = 0;
821 
822 	for (i = 0; i < num_tuples; i++) {
823 		pt = &pdump_t[i];
824 		if (pt->dir == RTE_PDUMP_FLAG_RXTX) {
825 			if (pt->dump_by_type == DEVICE_ID) {
826 				ret = rte_pdump_enable_by_deviceid(
827 						pt->device_id,
828 						pt->queue,
829 						RTE_PDUMP_FLAG_RX,
830 						pt->rx_ring,
831 						pt->mp, NULL);
832 				ret1 = rte_pdump_enable_by_deviceid(
833 						pt->device_id,
834 						pt->queue,
835 						RTE_PDUMP_FLAG_TX,
836 						pt->tx_ring,
837 						pt->mp, NULL);
838 			} else if (pt->dump_by_type == PORT_ID) {
839 				ret = rte_pdump_enable(pt->port, pt->queue,
840 						RTE_PDUMP_FLAG_RX,
841 						pt->rx_ring, pt->mp, NULL);
842 				ret1 = rte_pdump_enable(pt->port, pt->queue,
843 						RTE_PDUMP_FLAG_TX,
844 						pt->tx_ring, pt->mp, NULL);
845 			}
846 		} else if (pt->dir == RTE_PDUMP_FLAG_RX) {
847 			if (pt->dump_by_type == DEVICE_ID)
848 				ret = rte_pdump_enable_by_deviceid(
849 						pt->device_id,
850 						pt->queue,
851 						pt->dir, pt->rx_ring,
852 						pt->mp, NULL);
853 			else if (pt->dump_by_type == PORT_ID)
854 				ret = rte_pdump_enable(pt->port, pt->queue,
855 						pt->dir,
856 						pt->rx_ring, pt->mp, NULL);
857 		} else if (pt->dir == RTE_PDUMP_FLAG_TX) {
858 			if (pt->dump_by_type == DEVICE_ID)
859 				ret = rte_pdump_enable_by_deviceid(
860 						pt->device_id,
861 						pt->queue,
862 						pt->dir,
863 						pt->tx_ring, pt->mp, NULL);
864 			else if (pt->dump_by_type == PORT_ID)
865 				ret = rte_pdump_enable(pt->port, pt->queue,
866 						pt->dir,
867 						pt->tx_ring, pt->mp, NULL);
868 		}
869 		if (ret < 0 || ret1 < 0) {
870 			cleanup_pdump_resources();
871 			rte_exit(EXIT_FAILURE, "%s\n", rte_strerror(rte_errno));
872 		}
873 	}
874 }
875 
876 static inline void
877 pdump_packets(struct pdump_tuples *pt)
878 {
879 	if (pt->dir & RTE_PDUMP_FLAG_RX)
880 		pdump_rxtx(pt->rx_ring, pt->rx_vdev_id, &pt->stats);
881 	if (pt->dir & RTE_PDUMP_FLAG_TX)
882 		pdump_rxtx(pt->tx_ring, pt->tx_vdev_id, &pt->stats);
883 }
884 
885 static int
886 dump_packets_core(void *arg)
887 {
888 	struct pdump_tuples *pt = (struct pdump_tuples *) arg;
889 
890 	printf(" core (%u); port %u device (%s) queue %u\n",
891 			rte_lcore_id(), pt->port, pt->device_id, pt->queue);
892 	fflush(stdout);
893 
894 	while (!quit_signal)
895 		pdump_packets(pt);
896 
897 	return 0;
898 }
899 
900 static inline void
901 dump_packets(void)
902 {
903 	int i;
904 	uint32_t lcore_id = 0;
905 
906 	if (!multiple_core_capture) {
907 		printf(" core (%u), capture for (%d) tuples\n",
908 				rte_lcore_id(), num_tuples);
909 
910 		for (i = 0; i < num_tuples; i++)
911 			printf(" - port %u device (%s) queue %u\n",
912 				pdump_t[i].port,
913 				pdump_t[i].device_id,
914 				pdump_t[i].queue);
915 
916 		while (!quit_signal) {
917 			for (i = 0; i < num_tuples; i++)
918 				pdump_packets(&pdump_t[i]);
919 		}
920 
921 		return;
922 	}
923 
924 	/* check if there enough core */
925 	if ((uint32_t)num_tuples >= rte_lcore_count()) {
926 		printf("Insufficient cores to run parallel!\n");
927 		return;
928 	}
929 
930 	lcore_id = rte_get_next_lcore(lcore_id, 1, 0);
931 
932 	for (i = 0; i < num_tuples; i++) {
933 		rte_eal_remote_launch(dump_packets_core,
934 				&pdump_t[i], lcore_id);
935 		lcore_id = rte_get_next_lcore(lcore_id, 1, 0);
936 
937 		if (rte_eal_wait_lcore(lcore_id) < 0)
938 			rte_exit(EXIT_FAILURE, "failed to wait\n");
939 	}
940 
941 	/* master core */
942 	while (!quit_signal)
943 		;
944 }
945 
946 static void
947 enable_primary_monitor(void)
948 {
949 	int ret;
950 
951 	/* Once primary exits, so will pdump. */
952 	ret = rte_eal_alarm_set(MONITOR_INTERVAL, monitor_primary, NULL);
953 	if (ret < 0)
954 		printf("Fail to enable monitor:%d\n", ret);
955 }
956 
957 int
958 main(int argc, char **argv)
959 {
960 	int diag;
961 	int ret;
962 	int i;
963 
964 	char n_flag[] = "-n4";
965 	char mp_flag[] = "--proc-type=secondary";
966 	char *argp[argc + 2];
967 
968 	/* catch ctrl-c so we can print on exit */
969 	signal(SIGINT, signal_handler);
970 
971 	argp[0] = argv[0];
972 	argp[1] = n_flag;
973 	argp[2] = mp_flag;
974 
975 	for (i = 1; i < argc; i++)
976 		argp[i + 2] = argv[i];
977 
978 	argc += 2;
979 
980 	diag = rte_eal_init(argc, argp);
981 	if (diag < 0)
982 		rte_panic("Cannot init EAL\n");
983 
984 	if (rte_eth_dev_count_avail() == 0)
985 		rte_exit(EXIT_FAILURE, "No Ethernet ports - bye\n");
986 
987 	argc -= diag;
988 	argv += (diag - 2);
989 
990 	/* parse app arguments */
991 	if (argc > 1) {
992 		ret = launch_args_parse(argc, argv, argp[0]);
993 		if (ret < 0)
994 			rte_exit(EXIT_FAILURE, "Invalid argument\n");
995 	}
996 
997 	/* create mempool, ring and vdevs info */
998 	create_mp_ring_vdev();
999 	enable_pdump();
1000 	enable_primary_monitor();
1001 	dump_packets();
1002 
1003 	disable_primary_monitor();
1004 	cleanup_pdump_resources();
1005 	/* dump debug stats */
1006 	print_pdump_stats();
1007 
1008 	ret = rte_eal_cleanup();
1009 	if (ret)
1010 		printf("Error from rte_eal_cleanup(), %d\n", ret);
1011 
1012 	return 0;
1013 }
1014