1 /* SPDX-License-Identifier: BSD-3-Clause
2 * Copyright(c) 2010-2016 Intel Corporation
8 #include <rte_mempool.h>
9 #include <rte_malloc.h>
10 #include <rte_memcpy.h>
13 #include <rte_ether.h>
17 #include "rte_port_source_sink.h"
22 #ifdef RTE_PORT_STATS_COLLECT
24 #define RTE_PORT_SOURCE_STATS_PKTS_IN_ADD(port, val) \
25 port->stats.n_pkts_in += val
26 #define RTE_PORT_SOURCE_STATS_PKTS_DROP_ADD(port, val) \
27 port->stats.n_pkts_drop += val
31 #define RTE_PORT_SOURCE_STATS_PKTS_IN_ADD(port, val)
32 #define RTE_PORT_SOURCE_STATS_PKTS_DROP_ADD(port, val)
36 struct rte_port_source
{
37 struct rte_port_in_stats stats
;
39 struct rte_mempool
*mempool
;
41 /* PCAP buffers and indices */
52 pcap_source_load(struct rte_port_source
*port
,
53 const char *file_name
,
54 uint32_t n_bytes_per_pkt
,
59 uint32_t *pkt_len_aligns
= NULL
;
60 size_t total_buff_len
= 0;
62 char pcap_errbuf
[PCAP_ERRBUF_SIZE
];
64 struct pcap_pkthdr pcap_hdr
;
67 uint32_t pktmbuf_maxlen
= (uint32_t)
68 (rte_pktmbuf_data_room_size(port
->mempool
) -
69 RTE_PKTMBUF_HEADROOM
);
71 if (n_bytes_per_pkt
== 0)
72 max_len
= pktmbuf_maxlen
;
74 max_len
= RTE_MIN(n_bytes_per_pkt
, pktmbuf_maxlen
);
76 /* first time open, get packet number */
77 pcap_handle
= pcap_open_offline(file_name
, pcap_errbuf
);
78 if (pcap_handle
== NULL
) {
79 RTE_LOG(ERR
, PORT
, "Failed to open pcap file "
80 "'%s' for reading\n", file_name
);
84 while ((pkt
= pcap_next(pcap_handle
, &pcap_hdr
)) != NULL
)
87 pcap_close(pcap_handle
);
89 port
->pkt_len
= rte_zmalloc_socket("PCAP",
90 (sizeof(*port
->pkt_len
) * n_pkts
), 0, socket_id
);
91 if (port
->pkt_len
== NULL
) {
92 RTE_LOG(ERR
, PORT
, "No enough memory\n");
96 pkt_len_aligns
= rte_malloc("PCAP",
97 (sizeof(*pkt_len_aligns
) * n_pkts
), 0);
98 if (pkt_len_aligns
== NULL
) {
99 RTE_LOG(ERR
, PORT
, "No enough memory\n");
103 port
->pkts
= rte_zmalloc_socket("PCAP",
104 (sizeof(*port
->pkts
) * n_pkts
), 0, socket_id
);
105 if (port
->pkts
== NULL
) {
106 RTE_LOG(ERR
, PORT
, "No enough memory\n");
110 /* open 2nd time, get pkt_len */
111 pcap_handle
= pcap_open_offline(file_name
, pcap_errbuf
);
112 if (pcap_handle
== NULL
) {
113 RTE_LOG(ERR
, PORT
, "Failed to open pcap file "
114 "'%s' for reading\n", file_name
);
118 for (i
= 0; i
< n_pkts
; i
++) {
119 pkt
= pcap_next(pcap_handle
, &pcap_hdr
);
120 port
->pkt_len
[i
] = RTE_MIN(max_len
, pcap_hdr
.len
);
121 pkt_len_aligns
[i
] = RTE_CACHE_LINE_ROUNDUP(
123 total_buff_len
+= pkt_len_aligns
[i
];
126 pcap_close(pcap_handle
);
128 /* allocate a big trunk of data for pcap file load */
129 buff
= rte_zmalloc_socket("PCAP",
130 total_buff_len
, 0, socket_id
);
132 RTE_LOG(ERR
, PORT
, "No enough memory\n");
136 port
->pkt_buff
= buff
;
138 /* open file one last time to copy the pkt content */
139 pcap_handle
= pcap_open_offline(file_name
, pcap_errbuf
);
140 if (pcap_handle
== NULL
) {
141 RTE_LOG(ERR
, PORT
, "Failed to open pcap file "
142 "'%s' for reading\n", file_name
);
146 for (i
= 0; i
< n_pkts
; i
++) {
147 pkt
= pcap_next(pcap_handle
, &pcap_hdr
);
148 rte_memcpy(buff
, pkt
, port
->pkt_len
[i
]);
149 port
->pkts
[i
] = buff
;
150 buff
+= pkt_len_aligns
[i
];
153 pcap_close(pcap_handle
);
155 port
->n_pkts
= n_pkts
;
157 rte_free(pkt_len_aligns
);
159 RTE_LOG(INFO
, PORT
, "Successfully load pcap file "
160 "'%s' with %u pkts\n",
161 file_name
, port
->n_pkts
);
167 rte_free(pkt_len_aligns
);
169 rte_free(port
->pkt_len
);
171 rte_free(port
->pkts
);
173 rte_free(port
->pkt_buff
);
178 #define PCAP_SOURCE_LOAD(port, file_name, n_bytes, socket_id) \
179 pcap_source_load(port, file_name, n_bytes, socket_id)
181 #else /* RTE_PORT_PCAP */
183 #define PCAP_SOURCE_LOAD(port, file_name, n_bytes, socket_id) \
188 RTE_LOG(ERR, PORT, "Source port field " \
189 "\"file_name\" is not NULL.\n"); \
196 #endif /* RTE_PORT_PCAP */
199 rte_port_source_create(void *params
, int socket_id
)
201 struct rte_port_source_params
*p
=
203 struct rte_port_source
*port
;
205 /* Check input arguments*/
206 if ((p
== NULL
) || (p
->mempool
== NULL
)) {
207 RTE_LOG(ERR
, PORT
, "%s: Invalid params\n", __func__
);
211 /* Memory allocation */
212 port
= rte_zmalloc_socket("PORT", sizeof(*port
),
213 RTE_CACHE_LINE_SIZE
, socket_id
);
215 RTE_LOG(ERR
, PORT
, "%s: Failed to allocate port\n", __func__
);
220 port
->mempool
= (struct rte_mempool
*) p
->mempool
;
223 int status
= PCAP_SOURCE_LOAD(port
, p
->file_name
,
224 p
->n_bytes_per_pkt
, socket_id
);
236 rte_port_source_free(void *port
)
238 struct rte_port_source
*p
=
241 /* Check input parameters */
246 rte_free(p
->pkt_len
);
250 rte_free(p
->pkt_buff
);
258 rte_port_source_rx(void *port
, struct rte_mbuf
**pkts
, uint32_t n_pkts
)
260 struct rte_port_source
*p
= port
;
263 if (rte_pktmbuf_alloc_bulk(p
->mempool
, pkts
, n_pkts
) != 0)
266 if (p
->pkt_buff
!= NULL
) {
267 for (i
= 0; i
< n_pkts
; i
++) {
268 uint8_t *pkt_data
= rte_pktmbuf_mtod(pkts
[i
],
271 rte_memcpy(pkt_data
, p
->pkts
[p
->pkt_index
],
272 p
->pkt_len
[p
->pkt_index
]);
273 pkts
[i
]->data_len
= p
->pkt_len
[p
->pkt_index
];
274 pkts
[i
]->pkt_len
= pkts
[i
]->data_len
;
277 if (p
->pkt_index
>= p
->n_pkts
)
282 RTE_PORT_SOURCE_STATS_PKTS_IN_ADD(p
, n_pkts
);
288 rte_port_source_stats_read(void *port
,
289 struct rte_port_in_stats
*stats
, int clear
)
291 struct rte_port_source
*p
=
295 memcpy(stats
, &p
->stats
, sizeof(p
->stats
));
298 memset(&p
->stats
, 0, sizeof(p
->stats
));
306 #ifdef RTE_PORT_STATS_COLLECT
308 #define RTE_PORT_SINK_STATS_PKTS_IN_ADD(port, val) \
309 (port->stats.n_pkts_in += val)
310 #define RTE_PORT_SINK_STATS_PKTS_DROP_ADD(port, val) \
311 (port->stats.n_pkts_drop += val)
315 #define RTE_PORT_SINK_STATS_PKTS_IN_ADD(port, val)
316 #define RTE_PORT_SINK_STATS_PKTS_DROP_ADD(port, val)
320 struct rte_port_sink
{
321 struct rte_port_out_stats stats
;
323 /* PCAP dumper handle and pkts number */
327 uint32_t dump_finish
;
333 pcap_sink_open(struct rte_port_sink
*port
,
334 const char *file_name
,
338 pcap_dumper_t
*pcap_dumper
;
340 /** Open a dead pcap handler for opening dumper file */
341 tx_pcap
= pcap_open_dead(DLT_EN10MB
, 65535);
342 if (tx_pcap
== NULL
) {
343 RTE_LOG(ERR
, PORT
, "Cannot open pcap dead handler\n");
347 /* The dumper is created using the previous pcap_t reference */
348 pcap_dumper
= pcap_dump_open(tx_pcap
, file_name
);
349 if (pcap_dumper
== NULL
) {
350 RTE_LOG(ERR
, PORT
, "Failed to open pcap file "
351 "\"%s\" for writing\n", file_name
);
355 port
->dumper
= pcap_dumper
;
356 port
->max_pkts
= max_n_pkts
;
358 port
->dump_finish
= 0;
360 RTE_LOG(INFO
, PORT
, "Ready to dump packets to file \"%s\"\n",
367 pcap_sink_write_pkt(struct rte_port_sink
*port
, struct rte_mbuf
*mbuf
)
369 uint8_t *pcap_dumper
= (port
->dumper
);
370 struct pcap_pkthdr pcap_hdr
;
371 uint8_t jumbo_pkt_buf
[ETHER_MAX_JUMBO_FRAME_LEN
];
374 /* Maximum num packets already reached */
375 if (port
->dump_finish
)
378 pkt
= rte_pktmbuf_mtod(mbuf
, uint8_t *);
380 pcap_hdr
.len
= mbuf
->pkt_len
;
381 pcap_hdr
.caplen
= pcap_hdr
.len
;
382 gettimeofday(&(pcap_hdr
.ts
), NULL
);
384 if (mbuf
->nb_segs
> 1) {
385 struct rte_mbuf
*jumbo_mbuf
;
386 uint32_t pkt_index
= 0;
388 /* if packet size longer than ETHER_MAX_JUMBO_FRAME_LEN,
391 if (mbuf
->pkt_len
> ETHER_MAX_JUMBO_FRAME_LEN
)
394 for (jumbo_mbuf
= mbuf
; jumbo_mbuf
!= NULL
;
395 jumbo_mbuf
= jumbo_mbuf
->next
) {
396 rte_memcpy(&jumbo_pkt_buf
[pkt_index
],
397 rte_pktmbuf_mtod(jumbo_mbuf
, uint8_t *),
398 jumbo_mbuf
->data_len
);
399 pkt_index
+= jumbo_mbuf
->data_len
;
402 jumbo_pkt_buf
[pkt_index
] = '\0';
407 pcap_dump(pcap_dumper
, &pcap_hdr
, pkt
);
411 if ((port
->max_pkts
!= 0) && (port
->pkt_index
>= port
->max_pkts
)) {
412 port
->dump_finish
= 1;
413 RTE_LOG(INFO
, PORT
, "Dumped %u packets to file\n",
419 #define PCAP_SINK_OPEN(port, file_name, max_n_pkts) \
420 pcap_sink_open(port, file_name, max_n_pkts)
422 #define PCAP_SINK_WRITE_PKT(port, mbuf) \
423 pcap_sink_write_pkt(port, mbuf)
425 #define PCAP_SINK_FLUSH_PKT(dumper) \
428 pcap_dump_flush((pcap_dumper_t *)dumper); \
431 #define PCAP_SINK_CLOSE(dumper) \
434 pcap_dump_close((pcap_dumper_t *)dumper); \
439 #define PCAP_SINK_OPEN(port, file_name, max_n_pkts) \
444 RTE_LOG(ERR, PORT, "Sink port field " \
445 "\"file_name\" is not NULL.\n"); \
452 #define PCAP_SINK_WRITE_PKT(port, mbuf) {}
454 #define PCAP_SINK_FLUSH_PKT(dumper)
456 #define PCAP_SINK_CLOSE(dumper)
461 rte_port_sink_create(void *params
, int socket_id
)
463 struct rte_port_sink
*port
;
464 struct rte_port_sink_params
*p
= params
;
466 /* Memory allocation */
467 port
= rte_zmalloc_socket("PORT", sizeof(*port
),
468 RTE_CACHE_LINE_SIZE
, socket_id
);
470 RTE_LOG(ERR
, PORT
, "%s: Failed to allocate port\n", __func__
);
478 int status
= PCAP_SINK_OPEN(port
, p
->file_name
,
491 rte_port_sink_tx(void *port
, struct rte_mbuf
*pkt
)
493 struct rte_port_sink
*p
= port
;
495 RTE_PORT_SINK_STATS_PKTS_IN_ADD(p
, 1);
496 if (p
->dumper
!= NULL
)
497 PCAP_SINK_WRITE_PKT(p
, pkt
);
498 rte_pktmbuf_free(pkt
);
499 RTE_PORT_SINK_STATS_PKTS_DROP_ADD(p
, 1);
505 rte_port_sink_tx_bulk(void *port
, struct rte_mbuf
**pkts
,
508 struct rte_port_sink
*p
= port
;
510 if ((pkts_mask
& (pkts_mask
+ 1)) == 0) {
511 uint64_t n_pkts
= __builtin_popcountll(pkts_mask
);
514 RTE_PORT_SINK_STATS_PKTS_IN_ADD(p
, n_pkts
);
515 RTE_PORT_SINK_STATS_PKTS_DROP_ADD(p
, n_pkts
);
518 for (i
= 0; i
< n_pkts
; i
++)
519 PCAP_SINK_WRITE_PKT(p
, pkts
[i
]);
522 for (i
= 0; i
< n_pkts
; i
++) {
523 struct rte_mbuf
*pkt
= pkts
[i
];
525 rte_pktmbuf_free(pkt
);
530 uint64_t dump_pkts_mask
= pkts_mask
;
533 for ( ; dump_pkts_mask
; ) {
534 pkt_index
= __builtin_ctzll(
536 PCAP_SINK_WRITE_PKT(p
, pkts
[pkt_index
]);
537 dump_pkts_mask
&= ~(1LLU << pkt_index
);
541 for ( ; pkts_mask
; ) {
542 uint32_t pkt_index
= __builtin_ctzll(pkts_mask
);
543 uint64_t pkt_mask
= 1LLU << pkt_index
;
544 struct rte_mbuf
*pkt
= pkts
[pkt_index
];
546 RTE_PORT_SINK_STATS_PKTS_IN_ADD(p
, 1);
547 RTE_PORT_SINK_STATS_PKTS_DROP_ADD(p
, 1);
548 rte_pktmbuf_free(pkt
);
549 pkts_mask
&= ~pkt_mask
;
557 rte_port_sink_flush(void *port
)
559 struct rte_port_sink
*p
=
565 PCAP_SINK_FLUSH_PKT(p
->dumper
);
571 rte_port_sink_free(void *port
)
573 struct rte_port_sink
*p
=
579 PCAP_SINK_CLOSE(p
->dumper
);
587 rte_port_sink_stats_read(void *port
, struct rte_port_out_stats
*stats
,
590 struct rte_port_sink
*p
=
594 memcpy(stats
, &p
->stats
, sizeof(p
->stats
));
597 memset(&p
->stats
, 0, sizeof(p
->stats
));
603 * Summary of port operations
605 struct rte_port_in_ops rte_port_source_ops
= {
606 .f_create
= rte_port_source_create
,
607 .f_free
= rte_port_source_free
,
608 .f_rx
= rte_port_source_rx
,
609 .f_stats
= rte_port_source_stats_read
,
612 struct rte_port_out_ops rte_port_sink_ops
= {
613 .f_create
= rte_port_sink_create
,
614 .f_free
= rte_port_sink_free
,
615 .f_tx
= rte_port_sink_tx
,
616 .f_tx_bulk
= rte_port_sink_tx_bulk
,
617 .f_flush
= rte_port_sink_flush
,
618 .f_stats
= rte_port_sink_stats_read
,