1 /* SPDX-License-Identifier: BSD-3-Clause
2 * Copyright 2014-2020 Mellanox Technologies, Ltd
13 #include <sys/queue.h>
16 #include <rte_common.h>
17 #include <rte_byteorder.h>
19 #include <rte_debug.h>
20 #include <rte_cycles.h>
21 #include <rte_memory.h>
22 #include <rte_memcpy.h>
23 #include <rte_launch.h>
25 #include <rte_per_lcore.h>
26 #include <rte_lcore.h>
27 #include <rte_branch_prediction.h>
28 #include <rte_mempool.h>
30 #include <rte_interrupts.h>
32 #include <rte_ether.h>
33 #include <rte_ethdev.h>
37 #include <rte_string_fns.h>
42 static uint32_t cfg_ip_src = RTE_IPV4(10, 254, 0, 0);
43 static uint32_t cfg_ip_dst = RTE_IPV4(10, 253, 0, 0);
44 static uint16_t cfg_udp_src = 1000;
45 static uint16_t cfg_udp_dst = 1001;
46 static struct rte_ether_addr cfg_ether_src =
47 {{ 0x00, 0x01, 0x02, 0x03, 0x04, 0x00 }};
48 static struct rte_ether_addr cfg_ether_dst =
49 {{ 0x00, 0x01, 0x02, 0x03, 0x04, 0x01 }};
51 #define IP_DEFTTL 64 /* from RFC 1340. */
53 RTE_DEFINE_PER_LCORE(int, _next_flow);
56 * Multi-flow generation mode.
58 * We originate a bunch of flows (varying destination IP addresses), and
59 * terminate receive traffic. Received traffic is simply discarded, but we
60 * still do so in order to maintain traffic statistics.
63 pkt_burst_flow_gen(struct fwd_stream *fs)
65 unsigned pkt_size = tx_pkt_length - 4; /* Adjust FCS */
66 struct rte_mbuf *pkts_burst[MAX_PKT_BURST];
67 struct rte_mempool *mbp;
68 struct rte_mbuf *pkt = NULL;
69 struct rte_ether_hdr *eth_hdr;
70 struct rte_ipv4_hdr *ip_hdr;
71 struct rte_udp_hdr *udp_hdr;
72 uint16_t vlan_tci, vlan_tci_outer;
73 uint64_t ol_flags = 0;
78 uint16_t nb_clones = nb_pkt_flowgen_clones;
82 uint64_t start_tsc = 0;
83 int next_flow = RTE_PER_LCORE(_next_flow);
85 get_start_cycles(&start_tsc);
87 /* Receive a burst of packets and discard them. */
88 nb_rx = rte_eth_rx_burst(fs->rx_port, fs->rx_queue, pkts_burst,
90 inc_rx_burst_stats(fs, nb_rx);
91 fs->rx_packets += nb_rx;
93 for (i = 0; i < nb_rx; i++)
94 rte_pktmbuf_free(pkts_burst[i]);
96 mbp = current_fwd_lcore()->mbp;
97 vlan_tci = ports[fs->tx_port].tx_vlan_id;
98 vlan_tci_outer = ports[fs->tx_port].tx_vlan_id_outer;
100 tx_offloads = ports[fs->tx_port].dev_conf.txmode.offloads;
101 if (tx_offloads & RTE_ETH_TX_OFFLOAD_VLAN_INSERT)
102 ol_flags |= RTE_MBUF_F_TX_VLAN;
103 if (tx_offloads & RTE_ETH_TX_OFFLOAD_QINQ_INSERT)
104 ol_flags |= RTE_MBUF_F_TX_QINQ;
105 if (tx_offloads & RTE_ETH_TX_OFFLOAD_MACSEC_INSERT)
106 ol_flags |= RTE_MBUF_F_TX_MACSEC;
108 for (nb_pkt = 0; nb_pkt < nb_pkt_per_burst; nb_pkt++) {
109 if (!nb_pkt || !nb_clones) {
110 nb_clones = nb_pkt_flowgen_clones;
111 /* Logic limitation */
112 if (nb_clones > nb_pkt_per_burst)
113 nb_clones = nb_pkt_per_burst;
115 pkt = rte_mbuf_raw_alloc(mbp);
119 pkt->data_len = pkt_size;
122 /* Initialize Ethernet header. */
123 eth_hdr = rte_pktmbuf_mtod(pkt, struct rte_ether_hdr *);
124 rte_ether_addr_copy(&cfg_ether_dst, ð_hdr->dst_addr);
125 rte_ether_addr_copy(&cfg_ether_src, ð_hdr->src_addr);
126 eth_hdr->ether_type = rte_cpu_to_be_16(RTE_ETHER_TYPE_IPV4);
128 /* Initialize IP header. */
129 ip_hdr = (struct rte_ipv4_hdr *)(eth_hdr + 1);
130 memset(ip_hdr, 0, sizeof(*ip_hdr));
131 ip_hdr->version_ihl = RTE_IPV4_VHL_DEF;
132 ip_hdr->type_of_service = 0;
133 ip_hdr->fragment_offset = 0;
134 ip_hdr->time_to_live = IP_DEFTTL;
135 ip_hdr->next_proto_id = IPPROTO_UDP;
136 ip_hdr->packet_id = 0;
137 ip_hdr->src_addr = rte_cpu_to_be_32(cfg_ip_src);
138 ip_hdr->dst_addr = rte_cpu_to_be_32(cfg_ip_dst +
140 ip_hdr->total_length = RTE_CPU_TO_BE_16(pkt_size -
142 ip_hdr->hdr_checksum = rte_ipv4_cksum(ip_hdr);
144 /* Initialize UDP header. */
145 udp_hdr = (struct rte_udp_hdr *)(ip_hdr + 1);
146 udp_hdr->src_port = rte_cpu_to_be_16(cfg_udp_src);
147 udp_hdr->dst_port = rte_cpu_to_be_16(cfg_udp_dst);
148 udp_hdr->dgram_cksum = 0; /* No UDP checksum. */
149 udp_hdr->dgram_len = RTE_CPU_TO_BE_16(pkt_size -
153 pkt->pkt_len = pkt_size;
154 pkt->ol_flags &= RTE_MBUF_F_EXTERNAL;
155 pkt->ol_flags |= ol_flags;
156 pkt->vlan_tci = vlan_tci;
157 pkt->vlan_tci_outer = vlan_tci_outer;
158 pkt->l2_len = sizeof(struct rte_ether_hdr);
159 pkt->l3_len = sizeof(struct rte_ipv4_hdr);
162 rte_mbuf_refcnt_update(pkt, 1);
164 pkts_burst[nb_pkt] = pkt;
166 if (++next_flow >= nb_flows_flowgen)
170 nb_tx = rte_eth_tx_burst(fs->tx_port, fs->tx_queue, pkts_burst, nb_pkt);
174 if (unlikely(nb_tx < nb_pkt) && fs->retry_enabled) {
176 while (nb_tx < nb_pkt && retry++ < burst_tx_retry_num) {
177 rte_delay_us(burst_tx_delay_time);
178 nb_tx += rte_eth_tx_burst(fs->tx_port, fs->tx_queue,
179 &pkts_burst[nb_tx], nb_pkt - nb_tx);
182 fs->tx_packets += nb_tx;
184 inc_tx_burst_stats(fs, nb_tx);
185 nb_dropped = nb_pkt - nb_tx;
186 if (unlikely(nb_dropped > 0)) {
187 /* Back out the flow counter. */
188 next_flow -= nb_dropped;
189 while (next_flow < 0)
190 next_flow += nb_flows_flowgen;
192 fs->fwd_dropped += nb_dropped;
194 rte_pktmbuf_free(pkts_burst[nb_tx]);
195 } while (++nb_tx < nb_pkt);
198 RTE_PER_LCORE(_next_flow) = next_flow;
200 get_end_cycles(fs, start_tsc);
204 flowgen_begin(portid_t pi)
206 printf(" number of flows for port %u: %d\n", pi, nb_flows_flowgen);
210 struct fwd_engine flow_gen_engine = {
211 .fwd_mode_name = "flowgen",
212 .port_fwd_begin = flowgen_begin,
213 .port_fwd_end = NULL,
214 .packet_fwd = pkt_burst_flow_gen,