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 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);
166 rte_free(pkt_len_aligns);
167 rte_free(port->pkt_len);
168 rte_free(port->pkts);
169 rte_free(port->pkt_buff);
174 #define PCAP_SOURCE_LOAD(port, file_name, n_bytes, socket_id) \
175 pcap_source_load(port, file_name, n_bytes, socket_id)
177 #else /* RTE_PORT_PCAP */
179 #define PCAP_SOURCE_LOAD(port, file_name, n_bytes, socket_id) \
184 RTE_LOG(ERR, PORT, "Source port field " \
185 "\"file_name\" is not NULL.\n"); \
192 #endif /* RTE_PORT_PCAP */
195 rte_port_source_create(void *params, int socket_id)
197 struct rte_port_source_params *p =
199 struct rte_port_source *port;
201 /* Check input arguments*/
202 if ((p == NULL) || (p->mempool == NULL)) {
203 RTE_LOG(ERR, PORT, "%s: Invalid params\n", __func__);
207 /* Memory allocation */
208 port = rte_zmalloc_socket("PORT", sizeof(*port),
209 RTE_CACHE_LINE_SIZE, socket_id);
211 RTE_LOG(ERR, PORT, "%s: Failed to allocate port\n", __func__);
216 port->mempool = (struct rte_mempool *) p->mempool;
219 int status = PCAP_SOURCE_LOAD(port, p->file_name,
220 p->n_bytes_per_pkt, socket_id);
232 rte_port_source_free(void *port)
234 struct rte_port_source *p =
237 /* Check input parameters */
241 rte_free(p->pkt_len);
243 rte_free(p->pkt_buff);
251 rte_port_source_rx(void *port, struct rte_mbuf **pkts, uint32_t n_pkts)
253 struct rte_port_source *p = port;
256 if (rte_pktmbuf_alloc_bulk(p->mempool, pkts, n_pkts) != 0)
259 if (p->pkt_buff != NULL) {
260 for (i = 0; i < n_pkts; i++) {
261 uint8_t *pkt_data = rte_pktmbuf_mtod(pkts[i],
264 rte_memcpy(pkt_data, p->pkts[p->pkt_index],
265 p->pkt_len[p->pkt_index]);
266 pkts[i]->data_len = p->pkt_len[p->pkt_index];
267 pkts[i]->pkt_len = pkts[i]->data_len;
270 if (p->pkt_index >= p->n_pkts)
275 RTE_PORT_SOURCE_STATS_PKTS_IN_ADD(p, n_pkts);
281 rte_port_source_stats_read(void *port,
282 struct rte_port_in_stats *stats, int clear)
284 struct rte_port_source *p =
288 memcpy(stats, &p->stats, sizeof(p->stats));
291 memset(&p->stats, 0, sizeof(p->stats));
299 #ifdef RTE_PORT_STATS_COLLECT
301 #define RTE_PORT_SINK_STATS_PKTS_IN_ADD(port, val) \
302 (port->stats.n_pkts_in += val)
303 #define RTE_PORT_SINK_STATS_PKTS_DROP_ADD(port, val) \
304 (port->stats.n_pkts_drop += val)
308 #define RTE_PORT_SINK_STATS_PKTS_IN_ADD(port, val)
309 #define RTE_PORT_SINK_STATS_PKTS_DROP_ADD(port, val)
313 struct rte_port_sink {
314 struct rte_port_out_stats stats;
316 /* PCAP dumper handle and pkts number */
320 uint32_t dump_finish;
326 pcap_sink_open(struct rte_port_sink *port,
327 const char *file_name,
331 pcap_dumper_t *pcap_dumper;
333 /** Open a dead pcap handler for opening dumper file */
334 tx_pcap = pcap_open_dead(DLT_EN10MB, 65535);
335 if (tx_pcap == NULL) {
336 RTE_LOG(ERR, PORT, "Cannot open pcap dead handler\n");
340 /* The dumper is created using the previous pcap_t reference */
341 pcap_dumper = pcap_dump_open(tx_pcap, file_name);
342 if (pcap_dumper == NULL) {
343 RTE_LOG(ERR, PORT, "Failed to open pcap file "
344 "\"%s\" for writing\n", file_name);
348 port->dumper = pcap_dumper;
349 port->max_pkts = max_n_pkts;
351 port->dump_finish = 0;
353 RTE_LOG(INFO, PORT, "Ready to dump packets to file \"%s\"\n",
360 pcap_sink_write_pkt(struct rte_port_sink *port, struct rte_mbuf *mbuf)
362 uint8_t *pcap_dumper = (port->dumper);
363 struct pcap_pkthdr pcap_hdr;
364 uint8_t jumbo_pkt_buf[RTE_ETHER_MAX_JUMBO_FRAME_LEN];
367 /* Maximum num packets already reached */
368 if (port->dump_finish)
371 pkt = rte_pktmbuf_mtod(mbuf, uint8_t *);
373 pcap_hdr.len = mbuf->pkt_len;
374 pcap_hdr.caplen = pcap_hdr.len;
375 gettimeofday(&(pcap_hdr.ts), NULL);
377 if (mbuf->nb_segs > 1) {
378 struct rte_mbuf *jumbo_mbuf;
379 uint32_t pkt_index = 0;
381 /* if packet size longer than RTE_ETHER_MAX_JUMBO_FRAME_LEN,
384 if (mbuf->pkt_len > RTE_ETHER_MAX_JUMBO_FRAME_LEN)
387 for (jumbo_mbuf = mbuf; jumbo_mbuf != NULL;
388 jumbo_mbuf = jumbo_mbuf->next) {
389 rte_memcpy(&jumbo_pkt_buf[pkt_index],
390 rte_pktmbuf_mtod(jumbo_mbuf, uint8_t *),
391 jumbo_mbuf->data_len);
392 pkt_index += jumbo_mbuf->data_len;
395 jumbo_pkt_buf[pkt_index] = '\0';
400 pcap_dump(pcap_dumper, &pcap_hdr, pkt);
404 if ((port->max_pkts != 0) && (port->pkt_index >= port->max_pkts)) {
405 port->dump_finish = 1;
406 RTE_LOG(INFO, PORT, "Dumped %u packets to file\n",
412 #define PCAP_SINK_OPEN(port, file_name, max_n_pkts) \
413 pcap_sink_open(port, file_name, max_n_pkts)
415 #define PCAP_SINK_WRITE_PKT(port, mbuf) \
416 pcap_sink_write_pkt(port, mbuf)
418 #define PCAP_SINK_FLUSH_PKT(dumper) \
421 pcap_dump_flush((pcap_dumper_t *)dumper); \
424 #define PCAP_SINK_CLOSE(dumper) \
427 pcap_dump_close((pcap_dumper_t *)dumper); \
432 #define PCAP_SINK_OPEN(port, file_name, max_n_pkts) \
437 RTE_LOG(ERR, PORT, "Sink port field " \
438 "\"file_name\" is not NULL.\n"); \
445 #define PCAP_SINK_WRITE_PKT(port, mbuf) {}
447 #define PCAP_SINK_FLUSH_PKT(dumper)
449 #define PCAP_SINK_CLOSE(dumper)
454 rte_port_sink_create(void *params, int socket_id)
456 struct rte_port_sink *port;
457 struct rte_port_sink_params *p = params;
459 /* Memory allocation */
460 port = rte_zmalloc_socket("PORT", sizeof(*port),
461 RTE_CACHE_LINE_SIZE, socket_id);
463 RTE_LOG(ERR, PORT, "%s: Failed to allocate port\n", __func__);
471 int status = PCAP_SINK_OPEN(port, p->file_name,
484 rte_port_sink_tx(void *port, struct rte_mbuf *pkt)
486 struct rte_port_sink *p = port;
488 RTE_PORT_SINK_STATS_PKTS_IN_ADD(p, 1);
489 if (p->dumper != NULL)
490 PCAP_SINK_WRITE_PKT(p, pkt);
491 rte_pktmbuf_free(pkt);
492 RTE_PORT_SINK_STATS_PKTS_DROP_ADD(p, 1);
498 rte_port_sink_tx_bulk(void *port, struct rte_mbuf **pkts,
501 struct rte_port_sink *p = port;
503 if ((pkts_mask & (pkts_mask + 1)) == 0) {
504 uint64_t n_pkts = __builtin_popcountll(pkts_mask);
507 RTE_PORT_SINK_STATS_PKTS_IN_ADD(p, n_pkts);
508 RTE_PORT_SINK_STATS_PKTS_DROP_ADD(p, n_pkts);
511 for (i = 0; i < n_pkts; i++)
512 PCAP_SINK_WRITE_PKT(p, pkts[i]);
515 for (i = 0; i < n_pkts; i++) {
516 struct rte_mbuf *pkt = pkts[i];
518 rte_pktmbuf_free(pkt);
523 uint64_t dump_pkts_mask = pkts_mask;
526 for ( ; dump_pkts_mask; ) {
527 pkt_index = __builtin_ctzll(
529 PCAP_SINK_WRITE_PKT(p, pkts[pkt_index]);
530 dump_pkts_mask &= ~(1LLU << pkt_index);
534 for ( ; pkts_mask; ) {
535 uint32_t pkt_index = __builtin_ctzll(pkts_mask);
536 uint64_t pkt_mask = 1LLU << pkt_index;
537 struct rte_mbuf *pkt = pkts[pkt_index];
539 RTE_PORT_SINK_STATS_PKTS_IN_ADD(p, 1);
540 RTE_PORT_SINK_STATS_PKTS_DROP_ADD(p, 1);
541 rte_pktmbuf_free(pkt);
542 pkts_mask &= ~pkt_mask;
550 rte_port_sink_flush(void *port)
552 struct rte_port_sink *p =
558 PCAP_SINK_FLUSH_PKT(p->dumper);
564 rte_port_sink_free(void *port)
566 struct rte_port_sink *p =
572 PCAP_SINK_CLOSE(p->dumper);
580 rte_port_sink_stats_read(void *port, struct rte_port_out_stats *stats,
583 struct rte_port_sink *p =
587 memcpy(stats, &p->stats, sizeof(p->stats));
590 memset(&p->stats, 0, sizeof(p->stats));
596 * Summary of port operations
598 struct rte_port_in_ops rte_port_source_ops = {
599 .f_create = rte_port_source_create,
600 .f_free = rte_port_source_free,
601 .f_rx = rte_port_source_rx,
602 .f_stats = rte_port_source_stats_read,
605 struct rte_port_out_ops rte_port_sink_ops = {
606 .f_create = rte_port_sink_create,
607 .f_free = rte_port_sink_free,
608 .f_tx = rte_port_sink_tx,
609 .f_tx_bulk = rte_port_sink_tx_bulk,
610 .f_flush = rte_port_sink_flush,
611 .f_stats = rte_port_sink_stats_read,