4 * Copyright(c) 2010-2016 Intel Corporation. All rights reserved.
7 * Redistribution and use in source and binary forms, with or without
8 * modification, are permitted provided that the following conditions
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
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.
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.
37 #include <rte_mempool.h>
38 #include <rte_malloc.h>
39 #include <rte_memcpy.h>
42 #include <rte_ether.h>
46 #include "rte_port_source_sink.h"
51 #ifdef RTE_PORT_STATS_COLLECT
53 #define RTE_PORT_SOURCE_STATS_PKTS_IN_ADD(port, val) \
54 port->stats.n_pkts_in += val
55 #define RTE_PORT_SOURCE_STATS_PKTS_DROP_ADD(port, val) \
56 port->stats.n_pkts_drop += val
60 #define RTE_PORT_SOURCE_STATS_PKTS_IN_ADD(port, val)
61 #define RTE_PORT_SOURCE_STATS_PKTS_DROP_ADD(port, val)
65 struct rte_port_source {
66 struct rte_port_in_stats stats;
68 struct rte_mempool *mempool;
70 /* PCAP buffers and indices */
81 pcap_source_load(struct rte_port_source *port,
82 const char *file_name,
83 uint32_t n_bytes_per_pkt,
88 uint32_t *pkt_len_aligns = NULL;
89 size_t total_buff_len = 0;
91 char pcap_errbuf[PCAP_ERRBUF_SIZE];
93 struct pcap_pkthdr pcap_hdr;
96 uint32_t pktmbuf_maxlen = (uint32_t)
97 (rte_pktmbuf_data_room_size(port->mempool) -
98 RTE_PKTMBUF_HEADROOM);
100 if (n_bytes_per_pkt == 0)
101 max_len = pktmbuf_maxlen;
103 max_len = RTE_MIN(n_bytes_per_pkt, pktmbuf_maxlen);
105 /* first time open, get packet number */
106 pcap_handle = pcap_open_offline(file_name, pcap_errbuf);
107 if (pcap_handle == NULL) {
108 RTE_LOG(ERR, PORT, "Failed to open pcap file "
109 "'%s' for reading\n", file_name);
113 while ((pkt = pcap_next(pcap_handle, &pcap_hdr)) != NULL)
116 pcap_close(pcap_handle);
118 port->pkt_len = rte_zmalloc_socket("PCAP",
119 (sizeof(*port->pkt_len) * n_pkts), 0, socket_id);
120 if (port->pkt_len == NULL) {
121 RTE_LOG(ERR, PORT, "No enough memory\n");
125 pkt_len_aligns = rte_malloc("PCAP",
126 (sizeof(*pkt_len_aligns) * n_pkts), 0);
127 if (pkt_len_aligns == NULL) {
128 RTE_LOG(ERR, PORT, "No enough memory\n");
132 port->pkts = rte_zmalloc_socket("PCAP",
133 (sizeof(*port->pkts) * n_pkts), 0, socket_id);
134 if (port->pkts == NULL) {
135 RTE_LOG(ERR, PORT, "No enough memory\n");
139 /* open 2nd time, get pkt_len */
140 pcap_handle = pcap_open_offline(file_name, pcap_errbuf);
141 if (pcap_handle == NULL) {
142 RTE_LOG(ERR, PORT, "Failed to open pcap file "
143 "'%s' for reading\n", file_name);
147 for (i = 0; i < n_pkts; i++) {
148 pkt = pcap_next(pcap_handle, &pcap_hdr);
149 port->pkt_len[i] = RTE_MIN(max_len, pcap_hdr.len);
150 pkt_len_aligns[i] = RTE_CACHE_LINE_ROUNDUP(
152 total_buff_len += pkt_len_aligns[i];
155 pcap_close(pcap_handle);
157 /* allocate a big trunk of data for pcap file load */
158 buff = rte_zmalloc_socket("PCAP",
159 total_buff_len, 0, socket_id);
161 RTE_LOG(ERR, PORT, "No enough memory\n");
165 port->pkt_buff = buff;
167 /* open file one last time to copy the pkt content */
168 pcap_handle = pcap_open_offline(file_name, pcap_errbuf);
169 if (pcap_handle == NULL) {
170 RTE_LOG(ERR, PORT, "Failed to open pcap file "
171 "'%s' for reading\n", file_name);
175 for (i = 0; i < n_pkts; i++) {
176 pkt = pcap_next(pcap_handle, &pcap_hdr);
177 rte_memcpy(buff, pkt, port->pkt_len[i]);
178 port->pkts[i] = buff;
179 buff += pkt_len_aligns[i];
182 pcap_close(pcap_handle);
184 port->n_pkts = n_pkts;
186 rte_free(pkt_len_aligns);
188 RTE_LOG(INFO, PORT, "Successfully load pcap file "
189 "'%s' with %u pkts\n",
190 file_name, port->n_pkts);
196 rte_free(pkt_len_aligns);
198 rte_free(port->pkt_len);
200 rte_free(port->pkts);
202 rte_free(port->pkt_buff);
207 #define PCAP_SOURCE_LOAD(port, file_name, n_bytes, socket_id) \
208 pcap_source_load(port, file_name, n_bytes, socket_id)
210 #else /* RTE_PORT_PCAP */
212 #define PCAP_SOURCE_LOAD(port, file_name, n_bytes, socket_id) \
217 RTE_LOG(ERR, PORT, "Source port field " \
218 "\"file_name\" is not NULL.\n"); \
225 #endif /* RTE_PORT_PCAP */
228 rte_port_source_create(void *params, int socket_id)
230 struct rte_port_source_params *p =
232 struct rte_port_source *port;
234 /* Check input arguments*/
235 if ((p == NULL) || (p->mempool == NULL)) {
236 RTE_LOG(ERR, PORT, "%s: Invalid params\n", __func__);
240 /* Memory allocation */
241 port = rte_zmalloc_socket("PORT", sizeof(*port),
242 RTE_CACHE_LINE_SIZE, socket_id);
244 RTE_LOG(ERR, PORT, "%s: Failed to allocate port\n", __func__);
249 port->mempool = (struct rte_mempool *) p->mempool;
252 int status = PCAP_SOURCE_LOAD(port, p->file_name,
253 p->n_bytes_per_pkt, socket_id);
265 rte_port_source_free(void *port)
267 struct rte_port_source *p =
270 /* Check input parameters */
275 rte_free(p->pkt_len);
279 rte_free(p->pkt_buff);
287 rte_port_source_rx(void *port, struct rte_mbuf **pkts, uint32_t n_pkts)
289 struct rte_port_source *p = port;
292 if (rte_pktmbuf_alloc_bulk(p->mempool, pkts, n_pkts) != 0)
295 if (p->pkt_buff != NULL) {
296 for (i = 0; i < n_pkts; i++) {
297 uint8_t *pkt_data = rte_pktmbuf_mtod(pkts[i],
300 rte_memcpy(pkt_data, p->pkts[p->pkt_index],
301 p->pkt_len[p->pkt_index]);
302 pkts[i]->data_len = p->pkt_len[p->pkt_index];
303 pkts[i]->pkt_len = pkts[i]->data_len;
306 if (p->pkt_index >= p->n_pkts)
311 RTE_PORT_SOURCE_STATS_PKTS_IN_ADD(p, n_pkts);
317 rte_port_source_stats_read(void *port,
318 struct rte_port_in_stats *stats, int clear)
320 struct rte_port_source *p =
324 memcpy(stats, &p->stats, sizeof(p->stats));
327 memset(&p->stats, 0, sizeof(p->stats));
335 #ifdef RTE_PORT_STATS_COLLECT
337 #define RTE_PORT_SINK_STATS_PKTS_IN_ADD(port, val) \
338 (port->stats.n_pkts_in += val)
339 #define RTE_PORT_SINK_STATS_PKTS_DROP_ADD(port, val) \
340 (port->stats.n_pkts_drop += val)
344 #define RTE_PORT_SINK_STATS_PKTS_IN_ADD(port, val)
345 #define RTE_PORT_SINK_STATS_PKTS_DROP_ADD(port, val)
349 struct rte_port_sink {
350 struct rte_port_out_stats stats;
352 /* PCAP dumper handle and pkts number */
356 uint32_t dump_finish;
362 pcap_sink_open(struct rte_port_sink *port,
363 const char *file_name,
367 pcap_dumper_t *pcap_dumper;
369 /** Open a dead pcap handler for opening dumper file */
370 tx_pcap = pcap_open_dead(DLT_EN10MB, 65535);
371 if (tx_pcap == NULL) {
372 RTE_LOG(ERR, PORT, "Cannot open pcap dead handler\n");
376 /* The dumper is created using the previous pcap_t reference */
377 pcap_dumper = pcap_dump_open(tx_pcap, file_name);
378 if (pcap_dumper == NULL) {
379 RTE_LOG(ERR, PORT, "Failed to open pcap file "
380 "\"%s\" for writing\n", file_name);
384 port->dumper = pcap_dumper;
385 port->max_pkts = max_n_pkts;
387 port->dump_finish = 0;
389 RTE_LOG(INFO, PORT, "Ready to dump packets to file \"%s\"\n",
396 pcap_sink_write_pkt(struct rte_port_sink *port, struct rte_mbuf *mbuf)
398 uint8_t *pcap_dumper = (port->dumper);
399 struct pcap_pkthdr pcap_hdr;
400 uint8_t jumbo_pkt_buf[ETHER_MAX_JUMBO_FRAME_LEN];
403 /* Maximum num packets already reached */
404 if (port->dump_finish)
407 pkt = rte_pktmbuf_mtod(mbuf, uint8_t *);
409 pcap_hdr.len = mbuf->pkt_len;
410 pcap_hdr.caplen = pcap_hdr.len;
411 gettimeofday(&(pcap_hdr.ts), NULL);
413 if (mbuf->nb_segs > 1) {
414 struct rte_mbuf *jumbo_mbuf;
415 uint32_t pkt_index = 0;
417 /* if packet size longer than ETHER_MAX_JUMBO_FRAME_LEN,
420 if (mbuf->pkt_len > ETHER_MAX_JUMBO_FRAME_LEN)
423 for (jumbo_mbuf = mbuf; jumbo_mbuf != NULL;
424 jumbo_mbuf = jumbo_mbuf->next) {
425 rte_memcpy(&jumbo_pkt_buf[pkt_index],
426 rte_pktmbuf_mtod(jumbo_mbuf, uint8_t *),
427 jumbo_mbuf->data_len);
428 pkt_index += jumbo_mbuf->data_len;
431 jumbo_pkt_buf[pkt_index] = '\0';
436 pcap_dump(pcap_dumper, &pcap_hdr, pkt);
440 if ((port->max_pkts != 0) && (port->pkt_index >= port->max_pkts)) {
441 port->dump_finish = 1;
442 RTE_LOG(INFO, PORT, "Dumped %u packets to file\n",
448 #define PCAP_SINK_OPEN(port, file_name, max_n_pkts) \
449 pcap_sink_open(port, file_name, max_n_pkts)
451 #define PCAP_SINK_WRITE_PKT(port, mbuf) \
452 pcap_sink_write_pkt(port, mbuf)
454 #define PCAP_SINK_FLUSH_PKT(dumper) \
457 pcap_dump_flush((pcap_dumper_t *)dumper); \
460 #define PCAP_SINK_CLOSE(dumper) \
463 pcap_dump_close((pcap_dumper_t *)dumper); \
468 #define PCAP_SINK_OPEN(port, file_name, max_n_pkts) \
473 RTE_LOG(ERR, PORT, "Sink port field " \
474 "\"file_name\" is not NULL.\n"); \
481 #define PCAP_SINK_WRITE_PKT(port, mbuf) {}
483 #define PCAP_SINK_FLUSH_PKT(dumper)
485 #define PCAP_SINK_CLOSE(dumper)
490 rte_port_sink_create(void *params, int socket_id)
492 struct rte_port_sink *port;
493 struct rte_port_sink_params *p = params;
495 /* Memory allocation */
496 port = rte_zmalloc_socket("PORT", sizeof(*port),
497 RTE_CACHE_LINE_SIZE, socket_id);
499 RTE_LOG(ERR, PORT, "%s: Failed to allocate port\n", __func__);
507 int status = PCAP_SINK_OPEN(port, p->file_name,
520 rte_port_sink_tx(void *port, struct rte_mbuf *pkt)
522 struct rte_port_sink *p = port;
524 RTE_PORT_SINK_STATS_PKTS_IN_ADD(p, 1);
525 if (p->dumper != NULL)
526 PCAP_SINK_WRITE_PKT(p, pkt);
527 rte_pktmbuf_free(pkt);
528 RTE_PORT_SINK_STATS_PKTS_DROP_ADD(p, 1);
534 rte_port_sink_tx_bulk(void *port, struct rte_mbuf **pkts,
537 struct rte_port_sink *p = port;
539 if ((pkts_mask & (pkts_mask + 1)) == 0) {
540 uint64_t n_pkts = __builtin_popcountll(pkts_mask);
543 RTE_PORT_SINK_STATS_PKTS_IN_ADD(p, n_pkts);
544 RTE_PORT_SINK_STATS_PKTS_DROP_ADD(p, n_pkts);
547 for (i = 0; i < n_pkts; i++)
548 PCAP_SINK_WRITE_PKT(p, pkts[i]);
551 for (i = 0; i < n_pkts; i++) {
552 struct rte_mbuf *pkt = pkts[i];
554 rte_pktmbuf_free(pkt);
559 uint64_t dump_pkts_mask = pkts_mask;
562 for ( ; dump_pkts_mask; ) {
563 pkt_index = __builtin_ctzll(
565 PCAP_SINK_WRITE_PKT(p, pkts[pkt_index]);
566 dump_pkts_mask &= ~(1LLU << pkt_index);
570 for ( ; pkts_mask; ) {
571 uint32_t pkt_index = __builtin_ctzll(pkts_mask);
572 uint64_t pkt_mask = 1LLU << pkt_index;
573 struct rte_mbuf *pkt = pkts[pkt_index];
575 RTE_PORT_SINK_STATS_PKTS_IN_ADD(p, 1);
576 RTE_PORT_SINK_STATS_PKTS_DROP_ADD(p, 1);
577 rte_pktmbuf_free(pkt);
578 pkts_mask &= ~pkt_mask;
586 rte_port_sink_flush(void *port)
588 struct rte_port_sink *p =
594 PCAP_SINK_FLUSH_PKT(p->dumper);
600 rte_port_sink_free(void *port)
602 struct rte_port_sink *p =
608 PCAP_SINK_CLOSE(p->dumper);
616 rte_port_sink_stats_read(void *port, struct rte_port_out_stats *stats,
619 struct rte_port_sink *p =
623 memcpy(stats, &p->stats, sizeof(p->stats));
626 memset(&p->stats, 0, sizeof(p->stats));
632 * Summary of port operations
634 struct rte_port_in_ops rte_port_source_ops = {
635 .f_create = rte_port_source_create,
636 .f_free = rte_port_source_free,
637 .f_rx = rte_port_source_rx,
638 .f_stats = rte_port_source_stats_read,
641 struct rte_port_out_ops rte_port_sink_ops = {
642 .f_create = rte_port_sink_create,
643 .f_free = rte_port_sink_free,
644 .f_tx = rte_port_sink_tx,
645 .f_tx_bulk = rte_port_sink_tx_bulk,
646 .f_flush = rte_port_sink_flush,
647 .f_stats = rte_port_sink_stats_read,