]> git.proxmox.com Git - ceph.git/blob - ceph/src/dpdk/app/pdump/main.c
add subtree-ish sources for 12.0.3
[ceph.git] / ceph / src / dpdk / app / pdump / main.c
1 /*
2 * BSD LICENSE
3 *
4 * Copyright(c) 2016 Intel Corporation. All rights reserved.
5 * All rights reserved.
6 *
7 * Redistribution and use in source and binary forms, with or without
8 * modification, are permitted provided that the following conditions
9 * are met:
10 *
11 * * Redistributions of source code must retain the above copyright
12 * notice, this list of conditions and the following disclaimer.
13 * * Redistributions in binary form must reproduce the above copyright
14 * notice, this list of conditions and the following disclaimer in
15 * the documentation and/or other materials provided with the
16 * distribution.
17 * * Neither the name of Intel Corporation nor the names of its
18 * contributors may be used to endorse or promote products derived
19 * from this software without specific prior written permission.
20 *
21 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
22 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
23 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
24 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
25 * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
26 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
27 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
28 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
29 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
30 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
31 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
32 */
33
34 #include <stdio.h>
35 #include <string.h>
36 #include <stdint.h>
37 #include <inttypes.h>
38 #include <stdlib.h>
39 #include <getopt.h>
40 #include <signal.h>
41 #include <stdbool.h>
42 #include <net/if.h>
43
44 #include <rte_eal.h>
45 #include <rte_common.h>
46 #include <rte_debug.h>
47 #include <rte_ethdev.h>
48 #include <rte_memory.h>
49 #include <rte_lcore.h>
50 #include <rte_branch_prediction.h>
51 #include <rte_errno.h>
52 #include <rte_dev.h>
53 #include <rte_kvargs.h>
54 #include <rte_mempool.h>
55 #include <rte_ring.h>
56 #include <rte_pdump.h>
57
58 #define CMD_LINE_OPT_PDUMP "pdump"
59 #define PDUMP_PORT_ARG "port"
60 #define PDUMP_PCI_ARG "device_id"
61 #define PDUMP_QUEUE_ARG "queue"
62 #define PDUMP_DIR_ARG "dir"
63 #define PDUMP_RX_DEV_ARG "rx-dev"
64 #define PDUMP_TX_DEV_ARG "tx-dev"
65 #define PDUMP_RING_SIZE_ARG "ring-size"
66 #define PDUMP_MSIZE_ARG "mbuf-size"
67 #define PDUMP_NUM_MBUFS_ARG "total-num-mbufs"
68 #define CMD_LINE_OPT_SER_SOCK_PATH "server-socket-path"
69 #define CMD_LINE_OPT_CLI_SOCK_PATH "client-socket-path"
70
71 #define VDEV_PCAP "net_pcap_%s_%d,tx_pcap=%s"
72 #define VDEV_IFACE "net_pcap_%s_%d,tx_iface=%s"
73 #define TX_STREAM_SIZE 64
74
75 #define MP_NAME "pdump_pool_%d"
76
77 #define RX_RING "rx_ring_%d"
78 #define TX_RING "tx_ring_%d"
79
80 #define RX_STR "rx"
81 #define TX_STR "tx"
82
83 /* Maximum long option length for option parsing. */
84 #define APP_ARG_TCPDUMP_MAX_TUPLES 54
85 #define MBUF_POOL_CACHE_SIZE 250
86 #define TX_DESC_PER_QUEUE 512
87 #define RX_DESC_PER_QUEUE 128
88 #define MBUFS_PER_POOL 65535
89 #define MAX_LONG_OPT_SZ 64
90 #define RING_SIZE 16384
91 #define SIZE 256
92 #define BURST_SIZE 32
93 #define NUM_VDEVS 2
94
95 #define RTE_RING_SZ_MASK (unsigned)(0x0fffffff) /**< Ring size mask */
96 /* true if x is a power of 2 */
97 #define POWEROF2(x) ((((x)-1) & (x)) == 0)
98
99 enum pdump_en_dis {
100 DISABLE = 1,
101 ENABLE = 2
102 };
103
104 enum pcap_stream {
105 IFACE = 1,
106 PCAP = 2
107 };
108
109 enum pdump_by {
110 PORT_ID = 1,
111 DEVICE_ID = 2
112 };
113
114 const char *valid_pdump_arguments[] = {
115 PDUMP_PORT_ARG,
116 PDUMP_PCI_ARG,
117 PDUMP_QUEUE_ARG,
118 PDUMP_DIR_ARG,
119 PDUMP_RX_DEV_ARG,
120 PDUMP_TX_DEV_ARG,
121 PDUMP_RING_SIZE_ARG,
122 PDUMP_MSIZE_ARG,
123 PDUMP_NUM_MBUFS_ARG,
124 NULL
125 };
126
127 struct pdump_stats {
128 uint64_t dequeue_pkts;
129 uint64_t tx_pkts;
130 uint64_t freed_pkts;
131 };
132
133 struct pdump_tuples {
134 /* cli params */
135 uint8_t port;
136 char *device_id;
137 uint16_t queue;
138 char rx_dev[TX_STREAM_SIZE];
139 char tx_dev[TX_STREAM_SIZE];
140 uint32_t ring_size;
141 uint16_t mbuf_data_size;
142 uint32_t total_num_mbufs;
143
144 /* params for library API call */
145 uint32_t dir;
146 struct rte_mempool *mp;
147 struct rte_ring *rx_ring;
148 struct rte_ring *tx_ring;
149
150 /* params for packet dumping */
151 enum pdump_by dump_by_type;
152 int rx_vdev_id;
153 int tx_vdev_id;
154 enum pcap_stream rx_vdev_stream_type;
155 enum pcap_stream tx_vdev_stream_type;
156 bool single_pdump_dev;
157
158 /* stats */
159 struct pdump_stats stats;
160 } __rte_cache_aligned;
161 static struct pdump_tuples pdump_t[APP_ARG_TCPDUMP_MAX_TUPLES];
162
163 struct parse_val {
164 uint64_t min;
165 uint64_t max;
166 uint64_t val;
167 };
168
169 int num_tuples;
170 static struct rte_eth_conf port_conf_default;
171 volatile uint8_t quit_signal;
172 static char server_socket_path[PATH_MAX];
173 static char client_socket_path[PATH_MAX];
174
175 /**< display usage */
176 static void
177 pdump_usage(const char *prgname)
178 {
179 printf("usage: %s [EAL options] -- --pdump "
180 "'(port=<port id> | device_id=<pci id or vdev name>),"
181 "(queue=<queue_id>),"
182 "(rx-dev=<iface or pcap file> |"
183 " tx-dev=<iface or pcap file>,"
184 "[ring-size=<ring size>default:16384],"
185 "[mbuf-size=<mbuf data size>default:2176],"
186 "[total-num-mbufs=<number of mbufs>default:65535]'\n"
187 "[--server-socket-path=<server socket dir>"
188 "default:/var/run/.dpdk/ (or) ~/.dpdk/]\n"
189 "[--client-socket-path=<client socket dir>"
190 "default:/var/run/.dpdk/ (or) ~/.dpdk/]\n",
191 prgname);
192 }
193
194 static int
195 parse_device_id(const char *key __rte_unused, const char *value,
196 void *extra_args)
197 {
198 struct pdump_tuples *pt = extra_args;
199
200 pt->device_id = strdup(value);
201 pt->dump_by_type = DEVICE_ID;
202
203 return 0;
204 }
205
206 static int
207 parse_queue(const char *key __rte_unused, const char *value, void *extra_args)
208 {
209 unsigned long n;
210 struct pdump_tuples *pt = extra_args;
211
212 if (!strcmp(value, "*"))
213 pt->queue = RTE_PDUMP_ALL_QUEUES;
214 else {
215 n = strtoul(value, NULL, 10);
216 pt->queue = (uint16_t) n;
217 }
218 return 0;
219 }
220
221 static int
222 parse_rxtxdev(const char *key, const char *value, void *extra_args)
223 {
224
225 struct pdump_tuples *pt = extra_args;
226
227 if (!strcmp(key, PDUMP_RX_DEV_ARG)) {
228 snprintf(pt->rx_dev, sizeof(pt->rx_dev), "%s", value);
229 /* identify the tx stream type for pcap vdev */
230 if (if_nametoindex(pt->rx_dev))
231 pt->rx_vdev_stream_type = IFACE;
232 } else if (!strcmp(key, PDUMP_TX_DEV_ARG)) {
233 snprintf(pt->tx_dev, sizeof(pt->tx_dev), "%s", value);
234 /* identify the tx stream type for pcap vdev */
235 if (if_nametoindex(pt->tx_dev))
236 pt->tx_vdev_stream_type = IFACE;
237 }
238
239 return 0;
240 }
241
242 static int
243 parse_uint_value(const char *key, const char *value, void *extra_args)
244 {
245 struct parse_val *v;
246 unsigned long t;
247 char *end;
248 int ret = 0;
249
250 errno = 0;
251 v = extra_args;
252 t = strtoul(value, &end, 10);
253
254 if (errno != 0 || end[0] != 0 || t < v->min || t > v->max) {
255 printf("invalid value:\"%s\" for key:\"%s\", "
256 "value must be >= %"PRIu64" and <= %"PRIu64"\n",
257 value, key, v->min, v->max);
258 ret = -EINVAL;
259 }
260 if (!strcmp(key, PDUMP_RING_SIZE_ARG) && !POWEROF2(t)) {
261 printf("invalid value:\"%s\" for key:\"%s\", "
262 "value must be power of 2\n", value, key);
263 ret = -EINVAL;
264 }
265
266 if (ret != 0)
267 return ret;
268
269 v->val = t;
270 return 0;
271 }
272
273 static int
274 parse_pdump(const char *optarg)
275 {
276 struct rte_kvargs *kvlist;
277 int ret = 0, cnt1, cnt2;
278 struct pdump_tuples *pt;
279 struct parse_val v = {0};
280
281 pt = &pdump_t[num_tuples];
282
283 /* initial check for invalid arguments */
284 kvlist = rte_kvargs_parse(optarg, valid_pdump_arguments);
285 if (kvlist == NULL) {
286 printf("--pdump=\"%s\": invalid argument passed\n", optarg);
287 return -1;
288 }
289
290 /* port/device_id parsing and validation */
291 cnt1 = rte_kvargs_count(kvlist, PDUMP_PORT_ARG);
292 cnt2 = rte_kvargs_count(kvlist, PDUMP_PCI_ARG);
293 if (!((cnt1 == 1 && cnt2 == 0) || (cnt1 == 0 && cnt2 == 1))) {
294 printf("--pdump=\"%s\": must have either port or "
295 "device_id argument\n", optarg);
296 ret = -1;
297 goto free_kvlist;
298 } else if (cnt1 == 1) {
299 v.min = 0;
300 v.max = RTE_MAX_ETHPORTS-1;
301 ret = rte_kvargs_process(kvlist, PDUMP_PORT_ARG,
302 &parse_uint_value, &v);
303 if (ret < 0)
304 goto free_kvlist;
305 pt->port = (uint8_t) v.val;
306 pt->dump_by_type = PORT_ID;
307 } else if (cnt2 == 1) {
308 ret = rte_kvargs_process(kvlist, PDUMP_PCI_ARG,
309 &parse_device_id, pt);
310 if (ret < 0)
311 goto free_kvlist;
312 }
313
314 /* queue parsing and validation */
315 cnt1 = rte_kvargs_count(kvlist, PDUMP_QUEUE_ARG);
316 if (cnt1 != 1) {
317 printf("--pdump=\"%s\": must have queue argument\n", optarg);
318 ret = -1;
319 goto free_kvlist;
320 }
321 ret = rte_kvargs_process(kvlist, PDUMP_QUEUE_ARG, &parse_queue, pt);
322 if (ret < 0)
323 goto free_kvlist;
324
325 /* rx-dev and tx-dev parsing and validation */
326 cnt1 = rte_kvargs_count(kvlist, PDUMP_RX_DEV_ARG);
327 cnt2 = rte_kvargs_count(kvlist, PDUMP_TX_DEV_ARG);
328 if (cnt1 == 0 && cnt2 == 0) {
329 printf("--pdump=\"%s\": must have either rx-dev or "
330 "tx-dev argument\n", optarg);
331 ret = -1;
332 goto free_kvlist;
333 } else if (cnt1 == 1 && cnt2 == 1) {
334 ret = rte_kvargs_process(kvlist, PDUMP_RX_DEV_ARG,
335 &parse_rxtxdev, pt);
336 if (ret < 0)
337 goto free_kvlist;
338 ret = rte_kvargs_process(kvlist, PDUMP_TX_DEV_ARG,
339 &parse_rxtxdev, pt);
340 if (ret < 0)
341 goto free_kvlist;
342 /* if captured packets has to send to the same vdev */
343 if (!strcmp(pt->rx_dev, pt->tx_dev))
344 pt->single_pdump_dev = true;
345 pt->dir = RTE_PDUMP_FLAG_RXTX;
346 } else if (cnt1 == 1) {
347 ret = rte_kvargs_process(kvlist, PDUMP_RX_DEV_ARG,
348 &parse_rxtxdev, pt);
349 if (ret < 0)
350 goto free_kvlist;
351 pt->dir = RTE_PDUMP_FLAG_RX;
352 } else if (cnt2 == 1) {
353 ret = rte_kvargs_process(kvlist, PDUMP_TX_DEV_ARG,
354 &parse_rxtxdev, pt);
355 if (ret < 0)
356 goto free_kvlist;
357 pt->dir = RTE_PDUMP_FLAG_TX;
358 }
359
360 /* optional */
361 /* ring_size parsing and validation */
362 cnt1 = rte_kvargs_count(kvlist, PDUMP_RING_SIZE_ARG);
363 if (cnt1 == 1) {
364 v.min = 2;
365 v.max = RTE_RING_SZ_MASK-1;
366 ret = rte_kvargs_process(kvlist, PDUMP_RING_SIZE_ARG,
367 &parse_uint_value, &v);
368 if (ret < 0)
369 goto free_kvlist;
370 pt->ring_size = (uint32_t) v.val;
371 } else
372 pt->ring_size = RING_SIZE;
373
374 /* mbuf_data_size parsing and validation */
375 cnt1 = rte_kvargs_count(kvlist, PDUMP_MSIZE_ARG);
376 if (cnt1 == 1) {
377 v.min = 1;
378 v.max = UINT16_MAX;
379 ret = rte_kvargs_process(kvlist, PDUMP_MSIZE_ARG,
380 &parse_uint_value, &v);
381 if (ret < 0)
382 goto free_kvlist;
383 pt->mbuf_data_size = (uint16_t) v.val;
384 } else
385 pt->mbuf_data_size = RTE_MBUF_DEFAULT_BUF_SIZE;
386
387 /* total_num_mbufs parsing and validation */
388 cnt1 = rte_kvargs_count(kvlist, PDUMP_NUM_MBUFS_ARG);
389 if (cnt1 == 1) {
390 v.min = 1025;
391 v.max = UINT16_MAX;
392 ret = rte_kvargs_process(kvlist, PDUMP_NUM_MBUFS_ARG,
393 &parse_uint_value, &v);
394 if (ret < 0)
395 goto free_kvlist;
396 pt->total_num_mbufs = (uint16_t) v.val;
397 } else
398 pt->total_num_mbufs = MBUFS_PER_POOL;
399
400 num_tuples++;
401
402 free_kvlist:
403 rte_kvargs_free(kvlist);
404 return ret;
405 }
406
407 /* Parse the argument given in the command line of the application */
408 static int
409 launch_args_parse(int argc, char **argv, char *prgname)
410 {
411 int opt, ret;
412 int option_index;
413 static struct option long_option[] = {
414 {"pdump", 1, 0, 0},
415 {"server-socket-path", 1, 0, 0},
416 {"client-socket-path", 1, 0, 0},
417 {NULL, 0, 0, 0}
418 };
419
420 if (argc == 1)
421 pdump_usage(prgname);
422
423 /* Parse command line */
424 while ((opt = getopt_long(argc, argv, " ",
425 long_option, &option_index)) != EOF) {
426 switch (opt) {
427 case 0:
428 if (!strncmp(long_option[option_index].name,
429 CMD_LINE_OPT_PDUMP,
430 sizeof(CMD_LINE_OPT_PDUMP))) {
431 ret = parse_pdump(optarg);
432 if (ret) {
433 pdump_usage(prgname);
434 return -1;
435 }
436 }
437
438 if (!strncmp(long_option[option_index].name,
439 CMD_LINE_OPT_SER_SOCK_PATH,
440 sizeof(CMD_LINE_OPT_SER_SOCK_PATH))) {
441 snprintf(server_socket_path,
442 sizeof(server_socket_path), "%s",
443 optarg);
444 }
445
446 if (!strncmp(long_option[option_index].name,
447 CMD_LINE_OPT_CLI_SOCK_PATH,
448 sizeof(CMD_LINE_OPT_CLI_SOCK_PATH))) {
449 snprintf(client_socket_path,
450 sizeof(client_socket_path), "%s",
451 optarg);
452 }
453
454 break;
455 default:
456 pdump_usage(prgname);
457 return -1;
458 }
459 }
460
461 return 0;
462 }
463
464 static void
465 print_pdump_stats(void)
466 {
467 int i;
468 struct pdump_tuples *pt;
469
470 for (i = 0; i < num_tuples; i++) {
471 printf("##### PDUMP DEBUG STATS #####\n");
472 pt = &pdump_t[i];
473 printf(" -packets dequeued: %"PRIu64"\n",
474 pt->stats.dequeue_pkts);
475 printf(" -packets transmitted to vdev: %"PRIu64"\n",
476 pt->stats.tx_pkts);
477 printf(" -packets freed: %"PRIu64"\n",
478 pt->stats.freed_pkts);
479 }
480 }
481
482 static inline void
483 disable_pdump(struct pdump_tuples *pt)
484 {
485 if (pt->dump_by_type == DEVICE_ID)
486 rte_pdump_disable_by_deviceid(pt->device_id, pt->queue,
487 pt->dir);
488 else if (pt->dump_by_type == PORT_ID)
489 rte_pdump_disable(pt->port, pt->queue, pt->dir);
490 }
491
492 static inline void
493 pdump_rxtx(struct rte_ring *ring, uint8_t vdev_id, struct pdump_stats *stats)
494 {
495 /* write input packets of port to vdev for pdump */
496 struct rte_mbuf *rxtx_bufs[BURST_SIZE];
497
498 /* first dequeue packets from ring of primary process */
499 const uint16_t nb_in_deq = rte_ring_dequeue_burst(ring,
500 (void *)rxtx_bufs, BURST_SIZE);
501 stats->dequeue_pkts += nb_in_deq;
502
503 if (nb_in_deq) {
504 /* then sent on vdev */
505 uint16_t nb_in_txd = rte_eth_tx_burst(
506 vdev_id,
507 0, rxtx_bufs, nb_in_deq);
508 stats->tx_pkts += nb_in_txd;
509
510 if (unlikely(nb_in_txd < nb_in_deq)) {
511 do {
512 rte_pktmbuf_free(rxtx_bufs[nb_in_txd]);
513 stats->freed_pkts++;
514 } while (++nb_in_txd < nb_in_deq);
515 }
516 }
517 }
518
519 static void
520 free_ring_data(struct rte_ring *ring, uint8_t vdev_id,
521 struct pdump_stats *stats)
522 {
523 while (rte_ring_count(ring))
524 pdump_rxtx(ring, vdev_id, stats);
525 }
526
527 static void
528 cleanup_rings(void)
529 {
530 int i;
531 struct pdump_tuples *pt;
532
533 for (i = 0; i < num_tuples; i++) {
534 pt = &pdump_t[i];
535
536 if (pt->device_id)
537 free(pt->device_id);
538
539 /* free the rings */
540 if (pt->rx_ring)
541 rte_ring_free(pt->rx_ring);
542 if (pt->tx_ring)
543 rte_ring_free(pt->tx_ring);
544 }
545 }
546
547 static void
548 cleanup_pdump_resources(void)
549 {
550 int i;
551 struct pdump_tuples *pt;
552
553 /* disable pdump and free the pdump_tuple resources */
554 for (i = 0; i < num_tuples; i++) {
555 pt = &pdump_t[i];
556
557 /* remove callbacks */
558 disable_pdump(pt);
559
560 /*
561 * transmit rest of the enqueued packets of the rings on to
562 * the vdev, in order to release mbufs to the mepool.
563 **/
564 if (pt->dir & RTE_PDUMP_FLAG_RX)
565 free_ring_data(pt->rx_ring, pt->rx_vdev_id, &pt->stats);
566 if (pt->dir & RTE_PDUMP_FLAG_TX)
567 free_ring_data(pt->tx_ring, pt->tx_vdev_id, &pt->stats);
568 }
569 cleanup_rings();
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(uint8_t port_id)
584 {
585 struct ether_addr addr;
586 const uint16_t rxRings = 0, txRings = 1;
587 const uint8_t nb_ports = rte_eth_dev_count();
588 int ret;
589 uint16_t q;
590
591 if (port_id > nb_ports)
592 return -1;
593
594 ret = rte_eth_dev_configure(port_id, rxRings, txRings,
595 &port_conf_default);
596 if (ret != 0)
597 rte_exit(EXIT_FAILURE, "dev config failed\n");
598
599 for (q = 0; q < txRings; q++) {
600 ret = rte_eth_tx_queue_setup(port_id, q, TX_DESC_PER_QUEUE,
601 rte_eth_dev_socket_id(port_id), NULL);
602 if (ret < 0)
603 rte_exit(EXIT_FAILURE, "queue setup failed\n");
604 }
605
606 ret = rte_eth_dev_start(port_id);
607 if (ret < 0)
608 rte_exit(EXIT_FAILURE, "dev start failed\n");
609
610 rte_eth_macaddr_get(port_id, &addr);
611 printf("Port %u MAC: %02"PRIx8" %02"PRIx8" %02"PRIx8
612 " %02"PRIx8" %02"PRIx8" %02"PRIx8"\n",
613 (unsigned)port_id,
614 addr.addr_bytes[0], addr.addr_bytes[1],
615 addr.addr_bytes[2], addr.addr_bytes[3],
616 addr.addr_bytes[4], addr.addr_bytes[5]);
617
618 rte_eth_promiscuous_enable(port_id);
619
620 return 0;
621 }
622
623 static void
624 create_mp_ring_vdev(void)
625 {
626 int i;
627 uint8_t portid;
628 struct pdump_tuples *pt = NULL;
629 struct rte_mempool *mbuf_pool = NULL;
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(mempool_name,
641 pt->total_num_mbufs,
642 MBUF_POOL_CACHE_SIZE, 0,
643 pt->mbuf_data_size,
644 rte_socket_id());
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 (pt->rx_vdev_stream_type == IFACE) ?
680 snprintf(vdev_args, SIZE, VDEV_IFACE, RX_STR, i,
681 pt->rx_dev) :
682 snprintf(vdev_args, SIZE, VDEV_PCAP, RX_STR, i,
683 pt->rx_dev);
684 if (rte_eth_dev_attach(vdev_args, &portid) < 0) {
685 cleanup_rings();
686 rte_exit(EXIT_FAILURE,
687 "vdev creation failed:%s:%d\n",
688 __func__, __LINE__);
689 }
690 pt->rx_vdev_id = portid;
691
692 /* configure vdev */
693 configure_vdev(pt->rx_vdev_id);
694
695 if (pt->single_pdump_dev)
696 pt->tx_vdev_id = portid;
697 else {
698 (pt->tx_vdev_stream_type == IFACE) ?
699 snprintf(vdev_args, SIZE, VDEV_IFACE, TX_STR, i,
700 pt->tx_dev) :
701 snprintf(vdev_args, SIZE, VDEV_PCAP, TX_STR, i,
702 pt->tx_dev);
703 if (rte_eth_dev_attach(vdev_args,
704 &portid) < 0) {
705 cleanup_rings();
706 rte_exit(EXIT_FAILURE,
707 "vdev creation failed:"
708 "%s:%d\n", __func__, __LINE__);
709 }
710 pt->tx_vdev_id = portid;
711
712 /* configure vdev */
713 configure_vdev(pt->tx_vdev_id);
714 }
715 } else if (pt->dir == RTE_PDUMP_FLAG_RX) {
716
717 /* create rx_ring */
718 snprintf(ring_name, SIZE, RX_RING, i);
719 pt->rx_ring = rte_ring_create(ring_name, pt->ring_size,
720 rte_socket_id(), 0);
721 if (pt->rx_ring == NULL) {
722 cleanup_rings();
723 rte_exit(EXIT_FAILURE, "%s\n",
724 rte_strerror(rte_errno));
725 }
726
727 (pt->rx_vdev_stream_type == IFACE) ?
728 snprintf(vdev_args, SIZE, VDEV_IFACE, RX_STR, i,
729 pt->rx_dev) :
730 snprintf(vdev_args, SIZE, VDEV_PCAP, RX_STR, i,
731 pt->rx_dev);
732 if (rte_eth_dev_attach(vdev_args, &portid) < 0) {
733 cleanup_rings();
734 rte_exit(EXIT_FAILURE,
735 "vdev creation failed:%s:%d\n",
736 __func__, __LINE__);
737 }
738 pt->rx_vdev_id = portid;
739 /* configure vdev */
740 configure_vdev(pt->rx_vdev_id);
741 } else if (pt->dir == RTE_PDUMP_FLAG_TX) {
742
743 /* create tx_ring */
744 snprintf(ring_name, SIZE, TX_RING, i);
745 pt->tx_ring = rte_ring_create(ring_name, pt->ring_size,
746 rte_socket_id(), 0);
747 if (pt->tx_ring == NULL) {
748 cleanup_rings();
749 rte_exit(EXIT_FAILURE, "%s\n",
750 rte_strerror(rte_errno));
751 }
752
753 (pt->tx_vdev_stream_type == IFACE) ?
754 snprintf(vdev_args, SIZE, VDEV_IFACE, TX_STR, i,
755 pt->tx_dev) :
756 snprintf(vdev_args, SIZE, VDEV_PCAP, TX_STR, i,
757 pt->tx_dev);
758 if (rte_eth_dev_attach(vdev_args, &portid) < 0) {
759 cleanup_rings();
760 rte_exit(EXIT_FAILURE,
761 "vdev creation failed\n");
762 }
763 pt->tx_vdev_id = portid;
764
765 /* configure vdev */
766 configure_vdev(pt->tx_vdev_id);
767 }
768 }
769 }
770
771 static void
772 enable_pdump(void)
773 {
774 int i;
775 struct pdump_tuples *pt;
776 int ret = 0, ret1 = 0;
777
778 if (server_socket_path[0] != 0)
779 ret = rte_pdump_set_socket_dir(server_socket_path,
780 RTE_PDUMP_SOCKET_SERVER);
781 if (ret == 0 && client_socket_path[0] != 0) {
782 ret = rte_pdump_set_socket_dir(client_socket_path,
783 RTE_PDUMP_SOCKET_CLIENT);
784 }
785 if (ret < 0) {
786 cleanup_pdump_resources();
787 rte_exit(EXIT_FAILURE,
788 "failed to set socket paths of server:%s, "
789 "client:%s\n",
790 server_socket_path,
791 client_socket_path);
792 }
793
794 for (i = 0; i < num_tuples; i++) {
795 pt = &pdump_t[i];
796 if (pt->dir == RTE_PDUMP_FLAG_RXTX) {
797 if (pt->dump_by_type == DEVICE_ID) {
798 ret = rte_pdump_enable_by_deviceid(
799 pt->device_id,
800 pt->queue,
801 RTE_PDUMP_FLAG_RX,
802 pt->rx_ring,
803 pt->mp, NULL);
804 ret1 = rte_pdump_enable_by_deviceid(
805 pt->device_id,
806 pt->queue,
807 RTE_PDUMP_FLAG_TX,
808 pt->tx_ring,
809 pt->mp, NULL);
810 } else if (pt->dump_by_type == PORT_ID) {
811 ret = rte_pdump_enable(pt->port, pt->queue,
812 RTE_PDUMP_FLAG_RX,
813 pt->rx_ring, pt->mp, NULL);
814 ret1 = rte_pdump_enable(pt->port, pt->queue,
815 RTE_PDUMP_FLAG_TX,
816 pt->tx_ring, pt->mp, NULL);
817 }
818 } else if (pt->dir == RTE_PDUMP_FLAG_RX) {
819 if (pt->dump_by_type == DEVICE_ID)
820 ret = rte_pdump_enable_by_deviceid(
821 pt->device_id,
822 pt->queue,
823 pt->dir, pt->rx_ring,
824 pt->mp, NULL);
825 else if (pt->dump_by_type == PORT_ID)
826 ret = rte_pdump_enable(pt->port, pt->queue,
827 pt->dir,
828 pt->rx_ring, pt->mp, NULL);
829 } else if (pt->dir == RTE_PDUMP_FLAG_TX) {
830 if (pt->dump_by_type == DEVICE_ID)
831 ret = rte_pdump_enable_by_deviceid(
832 pt->device_id,
833 pt->queue,
834 pt->dir,
835 pt->tx_ring, pt->mp, NULL);
836 else if (pt->dump_by_type == PORT_ID)
837 ret = rte_pdump_enable(pt->port, pt->queue,
838 pt->dir,
839 pt->tx_ring, pt->mp, NULL);
840 }
841 if (ret < 0 || ret1 < 0) {
842 cleanup_pdump_resources();
843 rte_exit(EXIT_FAILURE, "%s\n", rte_strerror(rte_errno));
844 }
845 }
846 }
847
848 static inline void
849 dump_packets(void)
850 {
851 int i;
852 struct pdump_tuples *pt;
853
854 while (!quit_signal) {
855 for (i = 0; i < num_tuples; i++) {
856 pt = &pdump_t[i];
857 if (pt->dir & RTE_PDUMP_FLAG_RX)
858 pdump_rxtx(pt->rx_ring, pt->rx_vdev_id,
859 &pt->stats);
860 if (pt->dir & RTE_PDUMP_FLAG_TX)
861 pdump_rxtx(pt->tx_ring, pt->tx_vdev_id,
862 &pt->stats);
863 }
864 }
865 }
866
867 int
868 main(int argc, char **argv)
869 {
870 int diag;
871 int ret;
872 int i;
873
874 char c_flag[] = "-c1";
875 char n_flag[] = "-n4";
876 char mp_flag[] = "--proc-type=secondary";
877 char *argp[argc + 3];
878
879 /* catch ctrl-c so we can print on exit */
880 signal(SIGINT, signal_handler);
881
882 argp[0] = argv[0];
883 argp[1] = c_flag;
884 argp[2] = n_flag;
885 argp[3] = mp_flag;
886
887 for (i = 1; i < argc; i++)
888 argp[i + 3] = argv[i];
889
890 argc += 3;
891
892 diag = rte_eal_init(argc, argp);
893 if (diag < 0)
894 rte_panic("Cannot init EAL\n");
895
896 argc -= diag;
897 argv += (diag - 3);
898
899 /* parse app arguments */
900 if (argc > 1) {
901 ret = launch_args_parse(argc, argv, argp[0]);
902 if (ret < 0)
903 rte_exit(EXIT_FAILURE, "Invalid argument\n");
904 }
905
906 /* create mempool, ring and vdevs info */
907 create_mp_ring_vdev();
908 enable_pdump();
909 dump_packets();
910
911 cleanup_pdump_resources();
912 /* dump debug stats */
913 print_pdump_stats();
914
915 return 0;
916 }