ceph/src/spdk/dpdk/app/test-pmd/txonly.c

   1 /* SPDX-License-Identifier: BSD-3-Clause
   2  * Copyright(c) 2010-2014 Intel Corporation
   3  */
   4
   5 #include <stdarg.h>
   6 #include <string.h>
   7 #include <stdio.h>
   8 #include <errno.h>
   9 #include <stdint.h>
  10 #include <unistd.h>
  11 #include <inttypes.h>
  12
  13 #include <sys/queue.h>
  14 #include <sys/stat.h>
  15
  16 #include <rte_common.h>
  17 #include <rte_byteorder.h>
  18 #include <rte_log.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>
  24 #include <rte_eal.h>
  25 #include <rte_per_lcore.h>
  26 #include <rte_lcore.h>
  27 #include <rte_atomic.h>
  28 #include <rte_branch_prediction.h>
  29 #include <rte_mempool.h>
  30 #include <rte_mbuf.h>
  31 #include <rte_interrupts.h>
  32 #include <rte_pci.h>
  33 #include <rte_ether.h>
  34 #include <rte_ethdev.h>
  35 #include <rte_ip.h>
  36 #include <rte_tcp.h>
  37 #include <rte_udp.h>
  38 #include <rte_string_fns.h>
  39 #include <rte_flow.h>
  40
  41 #include "testpmd.h"
  42
  43 #define UDP_SRC_PORT 1024
  44 #define UDP_DST_PORT 1024
  45
  46 #define IP_SRC_ADDR ((192U << 24) | (168 << 16) | (0 << 8) | 1)
  47 #define IP_DST_ADDR ((192U << 24) | (168 << 16) | (0 << 8) | 2)
  48
  49 #define IP_DEFTTL  64   /* from RFC 1340. */
  50 #define IP_VERSION 0x40
  51 #define IP_HDRLEN  0x05 /* default IP header length == five 32-bits words. */
  52 #define IP_VHL_DEF (IP_VERSION | IP_HDRLEN)
  53
  54 static struct ipv4_hdr  pkt_ip_hdr;  /**< IP header of transmitted packets. */
  55 static struct udp_hdr pkt_udp_hdr; /**< UDP header of transmitted packets. */
  56
  57 static void
  58 copy_buf_to_pkt_segs(void* buf, unsigned len, struct rte_mbuf *pkt,
  59                      unsigned offset)
  60 {
  61         struct rte_mbuf *seg;
  62         void *seg_buf;
  63         unsigned copy_len;
  64
  65         seg = pkt;
  66         while (offset >= seg->data_len) {
  67                 offset -= seg->data_len;
  68                 seg = seg->next;
  69         }
  70         copy_len = seg->data_len - offset;
  71         seg_buf = rte_pktmbuf_mtod_offset(seg, char *, offset);
  72         while (len > copy_len) {
  73                 rte_memcpy(seg_buf, buf, (size_t) copy_len);
  74                 len -= copy_len;
  75                 buf = ((char*) buf + copy_len);
  76                 seg = seg->next;
  77                 seg_buf = rte_pktmbuf_mtod(seg, char *);
  78                 copy_len = seg->data_len;
  79         }
  80         rte_memcpy(seg_buf, buf, (size_t) len);
  81 }
  82
  83 static inline void
  84 copy_buf_to_pkt(void* buf, unsigned len, struct rte_mbuf *pkt, unsigned offset)
  85 {
  86         if (offset + len <= pkt->data_len) {
  87                 rte_memcpy(rte_pktmbuf_mtod_offset(pkt, char *, offset),
  88                         buf, (size_t) len);
  89                 return;
  90         }
  91         copy_buf_to_pkt_segs(buf, len, pkt, offset);
  92 }
  93
  94 static void
  95 setup_pkt_udp_ip_headers(struct ipv4_hdr *ip_hdr,
  96                          struct udp_hdr *udp_hdr,
  97                          uint16_t pkt_data_len)
  98 {
  99         uint16_t *ptr16;
 100         uint32_t ip_cksum;
 101         uint16_t pkt_len;
 102
 103         /*
 104          * Initialize UDP header.
 105          */
 106         pkt_len = (uint16_t) (pkt_data_len + sizeof(struct udp_hdr));
 107         udp_hdr->src_port = rte_cpu_to_be_16(UDP_SRC_PORT);
 108         udp_hdr->dst_port = rte_cpu_to_be_16(UDP_DST_PORT);
 109         udp_hdr->dgram_len      = RTE_CPU_TO_BE_16(pkt_len);
 110         udp_hdr->dgram_cksum    = 0; /* No UDP checksum. */
 111
 112         /*
 113          * Initialize IP header.
 114          */
 115         pkt_len = (uint16_t) (pkt_len + sizeof(struct ipv4_hdr));
 116         ip_hdr->version_ihl   = IP_VHL_DEF;
 117         ip_hdr->type_of_service   = 0;
 118         ip_hdr->fragment_offset = 0;
 119         ip_hdr->time_to_live   = IP_DEFTTL;
 120         ip_hdr->next_proto_id = IPPROTO_UDP;
 121         ip_hdr->packet_id = 0;
 122         ip_hdr->total_length   = RTE_CPU_TO_BE_16(pkt_len);
 123         ip_hdr->src_addr = rte_cpu_to_be_32(IP_SRC_ADDR);
 124         ip_hdr->dst_addr = rte_cpu_to_be_32(IP_DST_ADDR);
 125
 126         /*
 127          * Compute IP header checksum.
 128          */
 129         ptr16 = (unaligned_uint16_t*) ip_hdr;
 130         ip_cksum = 0;
 131         ip_cksum += ptr16[0]; ip_cksum += ptr16[1];
 132         ip_cksum += ptr16[2]; ip_cksum += ptr16[3];
 133         ip_cksum += ptr16[4];
 134         ip_cksum += ptr16[6]; ip_cksum += ptr16[7];
 135         ip_cksum += ptr16[8]; ip_cksum += ptr16[9];
 136
 137         /*
 138          * Reduce 32 bit checksum to 16 bits and complement it.
 139          */
 140         ip_cksum = ((ip_cksum & 0xFFFF0000) >> 16) +
 141                 (ip_cksum & 0x0000FFFF);
 142         if (ip_cksum > 65535)
 143                 ip_cksum -= 65535;
 144         ip_cksum = (~ip_cksum) & 0x0000FFFF;
 145         if (ip_cksum == 0)
 146                 ip_cksum = 0xFFFF;
 147         ip_hdr->hdr_checksum = (uint16_t) ip_cksum;
 148 }
 149
 150 /*
 151  * Transmit a burst of multi-segments packets.
 152  */
 153 static void
 154 pkt_burst_transmit(struct fwd_stream *fs)
 155 {
 156         struct rte_mbuf *pkts_burst[MAX_PKT_BURST];
 157         struct rte_port *txp;
 158         struct rte_mbuf *pkt;
 159         struct rte_mbuf *pkt_seg;
 160         struct rte_mempool *mbp;
 161         struct ether_hdr eth_hdr;
 162         uint16_t nb_tx;
 163         uint16_t nb_pkt;
 164         uint16_t vlan_tci, vlan_tci_outer;
 165         uint32_t retry;
 166         uint64_t ol_flags = 0;
 167         uint8_t  i;
 168         uint64_t tx_offloads;
 169 #ifdef RTE_TEST_PMD_RECORD_CORE_CYCLES
 170         uint64_t start_tsc;
 171         uint64_t end_tsc;
 172         uint64_t core_cycles;
 173 #endif
 174         uint32_t nb_segs, pkt_len;
 175
 176 #ifdef RTE_TEST_PMD_RECORD_CORE_CYCLES
 177         start_tsc = rte_rdtsc();
 178 #endif
 179
 180         mbp = current_fwd_lcore()->mbp;
 181         txp = &ports[fs->tx_port];
 182         tx_offloads = txp->dev_conf.txmode.offloads;
 183         vlan_tci = txp->tx_vlan_id;
 184         vlan_tci_outer = txp->tx_vlan_id_outer;
 185         if (tx_offloads & DEV_TX_OFFLOAD_VLAN_INSERT)
 186                 ol_flags = PKT_TX_VLAN_PKT;
 187         if (tx_offloads & DEV_TX_OFFLOAD_QINQ_INSERT)
 188                 ol_flags |= PKT_TX_QINQ_PKT;
 189         if (tx_offloads & DEV_TX_OFFLOAD_MACSEC_INSERT)
 190                 ol_flags |= PKT_TX_MACSEC;
 191         for (nb_pkt = 0; nb_pkt < nb_pkt_per_burst; nb_pkt++) {
 192                 pkt = rte_mbuf_raw_alloc(mbp);
 193                 if (pkt == NULL) {
 194                 nomore_mbuf:
 195                         if (nb_pkt == 0)
 196                                 return;
 197                         break;
 198                 }
 199
 200                 /*
 201                  * Using raw alloc is good to improve performance,
 202                  * but some consumers may use the headroom and so
 203                  * decrement data_off. We need to make sure it is
 204                  * reset to default value.
 205                  */
 206                 rte_pktmbuf_reset_headroom(pkt);
 207                 pkt->data_len = tx_pkt_seg_lengths[0];
 208                 pkt_seg = pkt;
 209                 if (tx_pkt_split == TX_PKT_SPLIT_RND)
 210                         nb_segs = random() % tx_pkt_nb_segs + 1;
 211                 else
 212                         nb_segs = tx_pkt_nb_segs;
 213                 pkt_len = pkt->data_len;
 214                 for (i = 1; i < nb_segs; i++) {
 215                         pkt_seg->next = rte_mbuf_raw_alloc(mbp);
 216                         if (pkt_seg->next == NULL) {
 217                                 pkt->nb_segs = i;
 218                                 rte_pktmbuf_free(pkt);
 219                                 goto nomore_mbuf;
 220                         }
 221                         pkt_seg = pkt_seg->next;
 222                         pkt_seg->data_len = tx_pkt_seg_lengths[i];
 223                         pkt_len += pkt_seg->data_len;
 224                 }
 225                 pkt_seg->next = NULL; /* Last segment of packet. */
 226
 227                 /*
 228                  * Initialize Ethernet header.
 229                  */
 230                 ether_addr_copy(&peer_eth_addrs[fs->peer_addr],&eth_hdr.d_addr);
 231                 ether_addr_copy(&ports[fs->tx_port].eth_addr, &eth_hdr.s_addr);
 232                 eth_hdr.ether_type = rte_cpu_to_be_16(ETHER_TYPE_IPv4);
 233
 234                 /*
 235                  * Copy headers in first packet segment(s).
 236                  */
 237                 copy_buf_to_pkt(&eth_hdr, sizeof(eth_hdr), pkt, 0);
 238                 copy_buf_to_pkt(&pkt_ip_hdr, sizeof(pkt_ip_hdr), pkt,
 239                                 sizeof(struct ether_hdr));
 240                 copy_buf_to_pkt(&pkt_udp_hdr, sizeof(pkt_udp_hdr), pkt,
 241                                 sizeof(struct ether_hdr) +
 242                                 sizeof(struct ipv4_hdr));
 243
 244                 /*
 245                  * Complete first mbuf of packet and append it to the
 246                  * burst of packets to be transmitted.
 247                  */
 248                 pkt->nb_segs = nb_segs;
 249                 pkt->pkt_len = pkt_len;
 250                 pkt->ol_flags = ol_flags;
 251                 pkt->vlan_tci = vlan_tci;
 252                 pkt->vlan_tci_outer = vlan_tci_outer;
 253                 pkt->l2_len = sizeof(struct ether_hdr);
 254                 pkt->l3_len = sizeof(struct ipv4_hdr);
 255                 pkts_burst[nb_pkt] = pkt;
 256         }
 257         nb_tx = rte_eth_tx_burst(fs->tx_port, fs->tx_queue, pkts_burst, nb_pkt);
 258         /*
 259          * Retry if necessary
 260          */
 261         if (unlikely(nb_tx < nb_pkt) && fs->retry_enabled) {
 262                 retry = 0;
 263                 while (nb_tx < nb_pkt && retry++ < burst_tx_retry_num) {
 264                         rte_delay_us(burst_tx_delay_time);
 265                         nb_tx += rte_eth_tx_burst(fs->tx_port, fs->tx_queue,
 266                                         &pkts_burst[nb_tx], nb_pkt - nb_tx);
 267                 }
 268         }
 269         fs->tx_packets += nb_tx;
 270
 271 #ifdef RTE_TEST_PMD_RECORD_BURST_STATS
 272         fs->tx_burst_stats.pkt_burst_spread[nb_tx]++;
 273 #endif
 274         if (unlikely(nb_tx < nb_pkt)) {
 275                 if (verbose_level > 0 && fs->fwd_dropped == 0)
 276                         printf("port %d tx_queue %d - drop "
 277                                "(nb_pkt:%u - nb_tx:%u)=%u packets\n",
 278                                fs->tx_port, fs->tx_queue,
 279                                (unsigned) nb_pkt, (unsigned) nb_tx,
 280                                (unsigned) (nb_pkt - nb_tx));
 281                 fs->fwd_dropped += (nb_pkt - nb_tx);
 282                 do {
 283                         rte_pktmbuf_free(pkts_burst[nb_tx]);
 284                 } while (++nb_tx < nb_pkt);
 285         }
 286
 287 #ifdef RTE_TEST_PMD_RECORD_CORE_CYCLES
 288         end_tsc = rte_rdtsc();
 289         core_cycles = (end_tsc - start_tsc);
 290         fs->core_cycles = (uint64_t) (fs->core_cycles + core_cycles);
 291 #endif
 292 }
 293
 294 static void
 295 tx_only_begin(__attribute__((unused)) portid_t pi)
 296 {
 297         uint16_t pkt_data_len;
 298
 299         pkt_data_len = (uint16_t) (tx_pkt_length - (sizeof(struct ether_hdr) +
 300                                                     sizeof(struct ipv4_hdr) +
 301                                                     sizeof(struct udp_hdr)));
 302         setup_pkt_udp_ip_headers(&pkt_ip_hdr, &pkt_udp_hdr, pkt_data_len);
 303 }
 304
 305 struct fwd_engine tx_only_engine = {
 306         .fwd_mode_name  = "txonly",
 307         .port_fwd_begin = tx_only_begin,
 308         .port_fwd_end   = NULL,
 309         .packet_fwd     = pkt_burst_transmit,
 310 };