net: add rte prefix to ether structures
[dpdk.git] / app / test / packet_burst_generator.c
index db7f023..a05e3c8 100644 (file)
@@ -1,34 +1,5 @@
-/*-
- *   BSD LICENSE
- *
- *   Copyright(c) 2010-2014 Intel Corporation. All rights reserved.
- *   All rights reserved.
- *
- *   Redistribution and use in source and binary forms, with or without
- *   modification, are permitted provided that the following conditions
- *   are met:
- *
- *     * Redistributions of source code must retain the above copyright
- *       notice, this list of conditions and the following disclaimer.
- *     * Redistributions in binary form must reproduce the above copyright
- *       notice, this list of conditions and the following disclaimer in
- *       the documentation and/or other materials provided with the
- *       distribution.
- *     * Neither the name of Intel Corporation nor the names of its
- *       contributors may be used to endorse or promote products derived
- *       from this software without specific prior written permission.
- *
- *   THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
- *   "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
- *   LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
- *   A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
- *   OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
- *   SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
- *   LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
- *   DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
- *   THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
- *   (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
- *   OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+/* SPDX-License-Identifier: BSD-3-Clause
+ * Copyright(c) 2010-2014 Intel Corporation
  */
 
 #include <rte_byteorder.h>
@@ -59,13 +30,13 @@ copy_buf_to_pkt_segs(void *buf, unsigned len, struct rte_mbuf *pkt,
                seg = seg->next;
        }
        copy_len = seg->data_len - offset;
-       seg_buf = ((char *) seg->data + offset);
+       seg_buf = rte_pktmbuf_mtod_offset(seg, char *, offset);
        while (len > copy_len) {
                rte_memcpy(seg_buf, buf, (size_t) copy_len);
                len -= copy_len;
                buf = ((char *) buf + copy_len);
                seg = seg->next;
-               seg_buf = seg->data;
+               seg_buf = rte_pktmbuf_mtod(seg, void *);
        }
        rte_memcpy(seg_buf, buf, (size_t) len);
 }
@@ -74,32 +45,51 @@ static inline void
 copy_buf_to_pkt(void *buf, unsigned len, struct rte_mbuf *pkt, unsigned offset)
 {
        if (offset + len <= pkt->data_len) {
-               rte_memcpy(((char *) pkt->data + offset), buf, (size_t) len);
+               rte_memcpy(rte_pktmbuf_mtod_offset(pkt, char *, offset), buf,
+                          (size_t) len);
                return;
        }
        copy_buf_to_pkt_segs(buf, len, pkt, offset);
 }
 
-
 void
-initialize_eth_header(struct ether_hdr *eth_hdr, struct ether_addr *src_mac,
-               struct ether_addr *dst_mac, uint8_t vlan_enabled, uint16_t van_id)
+initialize_eth_header(struct rte_ether_hdr *eth_hdr,
+               struct rte_ether_addr *src_mac,
+               struct rte_ether_addr *dst_mac, uint16_t ether_type,
+               uint8_t vlan_enabled, uint16_t van_id)
 {
        ether_addr_copy(dst_mac, &eth_hdr->d_addr);
        ether_addr_copy(src_mac, &eth_hdr->s_addr);
 
        if (vlan_enabled) {
-               struct vlan_hdr *vhdr = (struct vlan_hdr *)((uint8_t *)eth_hdr +
-                               sizeof(struct ether_hdr));
+               struct rte_vlan_hdr *vhdr = (struct rte_vlan_hdr *)(
+                       (uint8_t *)eth_hdr + sizeof(struct rte_ether_hdr));
 
                eth_hdr->ether_type = rte_cpu_to_be_16(ETHER_TYPE_VLAN);
 
-               vhdr->eth_proto =  rte_cpu_to_be_16(ETHER_TYPE_IPv4);
+               vhdr->eth_proto =  rte_cpu_to_be_16(ether_type);
                vhdr->vlan_tci = van_id;
        } else {
-               eth_hdr->ether_type = rte_cpu_to_be_16(ETHER_TYPE_VLAN);
+               eth_hdr->ether_type = rte_cpu_to_be_16(ether_type);
        }
+}
 
+void
+initialize_arp_header(struct rte_arp_hdr *arp_hdr,
+               struct rte_ether_addr *src_mac,
+               struct rte_ether_addr *dst_mac,
+               uint32_t src_ip, uint32_t dst_ip,
+               uint32_t opcode)
+{
+       arp_hdr->arp_hardware = rte_cpu_to_be_16(RTE_ARP_HRD_ETHER);
+       arp_hdr->arp_protocol = rte_cpu_to_be_16(ETHER_TYPE_IPv4);
+       arp_hdr->arp_hlen = ETHER_ADDR_LEN;
+       arp_hdr->arp_plen = sizeof(uint32_t);
+       arp_hdr->arp_opcode = rte_cpu_to_be_16(opcode);
+       ether_addr_copy(src_mac, &arp_hdr->arp_data.arp_sha);
+       arp_hdr->arp_data.arp_sip = src_ip;
+       ether_addr_copy(dst_mac, &arp_hdr->arp_data.arp_tha);
+       arp_hdr->arp_data.arp_tip = dst_ip;
 }
 
 uint16_t
@@ -118,6 +108,36 @@ initialize_udp_header(struct udp_hdr *udp_hdr, uint16_t src_port,
        return pkt_len;
 }
 
+uint16_t
+initialize_tcp_header(struct tcp_hdr *tcp_hdr, uint16_t src_port,
+               uint16_t dst_port, uint16_t pkt_data_len)
+{
+       uint16_t pkt_len;
+
+       pkt_len = (uint16_t) (pkt_data_len + sizeof(struct tcp_hdr));
+
+       memset(tcp_hdr, 0, sizeof(struct tcp_hdr));
+       tcp_hdr->src_port = rte_cpu_to_be_16(src_port);
+       tcp_hdr->dst_port = rte_cpu_to_be_16(dst_port);
+
+       return pkt_len;
+}
+
+uint16_t
+initialize_sctp_header(struct sctp_hdr *sctp_hdr, uint16_t src_port,
+               uint16_t dst_port, uint16_t pkt_data_len)
+{
+       uint16_t pkt_len;
+
+       pkt_len = (uint16_t) (pkt_data_len + sizeof(struct udp_hdr));
+
+       sctp_hdr->src_port = rte_cpu_to_be_16(src_port);
+       sctp_hdr->dst_port = rte_cpu_to_be_16(dst_port);
+       sctp_hdr->tag = 0;
+       sctp_hdr->cksum = 0; /* No SCTP checksum. */
+
+       return pkt_len;
+}
 
 uint16_t
 initialize_ipv6_header(struct ipv6_hdr *ip_hdr, uint8_t *src_addr,
@@ -139,7 +159,7 @@ initialize_ipv4_header(struct ipv4_hdr *ip_hdr, uint32_t src_addr,
                uint32_t dst_addr, uint16_t pkt_data_len)
 {
        uint16_t pkt_len;
-       uint16_t *ptr16;
+       unaligned_uint16_t *ptr16;
        uint32_t ip_cksum;
 
        /*
@@ -160,7 +180,7 @@ initialize_ipv4_header(struct ipv4_hdr *ip_hdr, uint32_t src_addr,
        /*
         * Compute IP header checksum.
         */
-       ptr16 = (uint16_t *)ip_hdr;
+       ptr16 = (unaligned_uint16_t *)ip_hdr;
        ip_cksum = 0;
        ip_cksum += ptr16[0]; ip_cksum += ptr16[1];
        ip_cksum += ptr16[2]; ip_cksum += ptr16[3];
@@ -182,7 +202,53 @@ initialize_ipv4_header(struct ipv4_hdr *ip_hdr, uint32_t src_addr,
        return pkt_len;
 }
 
+uint16_t
+initialize_ipv4_header_proto(struct ipv4_hdr *ip_hdr, uint32_t src_addr,
+               uint32_t dst_addr, uint16_t pkt_data_len, uint8_t proto)
+{
+       uint16_t pkt_len;
+       unaligned_uint16_t *ptr16;
+       uint32_t ip_cksum;
 
+       /*
+        * Initialize IP header.
+        */
+       pkt_len = (uint16_t) (pkt_data_len + sizeof(struct ipv4_hdr));
+
+       ip_hdr->version_ihl   = IP_VHL_DEF;
+       ip_hdr->type_of_service   = 0;
+       ip_hdr->fragment_offset = 0;
+       ip_hdr->time_to_live   = IP_DEFTTL;
+       ip_hdr->next_proto_id = proto;
+       ip_hdr->packet_id = 0;
+       ip_hdr->total_length   = rte_cpu_to_be_16(pkt_len);
+       ip_hdr->src_addr = rte_cpu_to_be_32(src_addr);
+       ip_hdr->dst_addr = rte_cpu_to_be_32(dst_addr);
+
+       /*
+        * Compute IP header checksum.
+        */
+       ptr16 = (unaligned_uint16_t *)ip_hdr;
+       ip_cksum = 0;
+       ip_cksum += ptr16[0]; ip_cksum += ptr16[1];
+       ip_cksum += ptr16[2]; ip_cksum += ptr16[3];
+       ip_cksum += ptr16[4];
+       ip_cksum += ptr16[6]; ip_cksum += ptr16[7];
+       ip_cksum += ptr16[8]; ip_cksum += ptr16[9];
+
+       /*
+        * Reduce 32 bit checksum to 16 bits and complement it.
+        */
+       ip_cksum = ((ip_cksum & 0xFFFF0000) >> 16) +
+               (ip_cksum & 0x0000FFFF);
+       ip_cksum %= 65536;
+       ip_cksum = (~ip_cksum) & 0x0000FFFF;
+       if (ip_cksum == 0)
+               ip_cksum = 0xFFFF;
+       ip_hdr->hdr_checksum = (uint16_t) ip_cksum;
+
+       return pkt_len;
+}
 
 /*
  * The maximum number of segments per packet is used when creating
@@ -190,20 +256,12 @@ initialize_ipv4_header(struct ipv4_hdr *ip_hdr, uint32_t src_addr,
  */
 #define RTE_MAX_SEGS_PER_PKT 255 /**< pkt.nb_segs is a 8-bit unsigned char. */
 
-#define TXONLY_DEF_PACKET_LEN 64
-#define TXONLY_DEF_PACKET_LEN_128 128
-
-uint16_t tx_pkt_length = TXONLY_DEF_PACKET_LEN;
-uint16_t tx_pkt_seg_lengths[RTE_MAX_SEGS_PER_PKT] = {
-               TXONLY_DEF_PACKET_LEN_128,
-};
-
-uint8_t  tx_pkt_nb_segs = 1;
 
 int
 generate_packet_burst(struct rte_mempool *mp, struct rte_mbuf **pkts_burst,
-               struct ether_hdr *eth_hdr, uint8_t vlan_enabled, void *ip_hdr,
-               uint8_t ipv4, struct udp_hdr *udp_hdr, int nb_pkt_per_burst)
+               struct rte_ether_hdr *eth_hdr, uint8_t vlan_enabled,
+               void *ip_hdr, uint8_t ipv4, struct udp_hdr *udp_hdr,
+               int nb_pkt_per_burst, uint8_t pkt_len, uint8_t nb_pkt_segs)
 {
        int i, nb_pkt = 0;
        size_t eth_hdr_size;
@@ -220,9 +278,9 @@ nomore_mbuf:
                        break;
                }
 
-               pkt->data_len = tx_pkt_seg_lengths[0];
+               pkt->data_len = pkt_len;
                pkt_seg = pkt;
-               for (i = 1; i < tx_pkt_nb_segs; i++) {
+               for (i = 1; i < nb_pkt_segs; i++) {
                        pkt_seg->next = rte_pktmbuf_alloc(mp);
                        if (pkt_seg->next == NULL) {
                                pkt->nb_segs = i;
@@ -230,7 +288,7 @@ nomore_mbuf:
                                goto nomore_mbuf;
                        }
                        pkt_seg = pkt_seg->next;
-                       pkt_seg->data_len = tx_pkt_seg_lengths[i];
+                       pkt_seg->data_len = pkt_len;
                }
                pkt_seg->next = NULL; /* Last segment of packet. */
 
@@ -238,9 +296,10 @@ nomore_mbuf:
                 * Copy headers in first packet segment(s).
                 */
                if (vlan_enabled)
-                       eth_hdr_size = sizeof(struct ether_hdr) + sizeof(struct vlan_hdr);
+                       eth_hdr_size = sizeof(struct rte_ether_hdr) +
+                               sizeof(struct rte_vlan_hdr);
                else
-                       eth_hdr_size = sizeof(struct ether_hdr);
+                       eth_hdr_size = sizeof(struct rte_ether_hdr);
 
                copy_buf_to_pkt(eth_hdr, eth_hdr_size, pkt, 0);
 
@@ -258,26 +317,131 @@ nomore_mbuf:
                 * Complete first mbuf of packet and append it to the
                 * burst of packets to be transmitted.
                 */
-               pkt->nb_segs = tx_pkt_nb_segs;
-               pkt->pkt_len = tx_pkt_length;
+               pkt->nb_segs = nb_pkt_segs;
+               pkt->pkt_len = pkt_len;
                pkt->l2_len = eth_hdr_size;
 
                if (ipv4) {
                        pkt->vlan_tci  = ETHER_TYPE_IPv4;
                        pkt->l3_len = sizeof(struct ipv4_hdr);
-
-                       if (vlan_enabled)
-                               pkt->ol_flags = PKT_RX_IPV4_HDR | PKT_RX_VLAN_PKT;
-                       else
-                               pkt->ol_flags = PKT_RX_IPV4_HDR;
                } else {
                        pkt->vlan_tci  = ETHER_TYPE_IPv6;
                        pkt->l3_len = sizeof(struct ipv6_hdr);
+               }
 
-                       if (vlan_enabled)
-                               pkt->ol_flags = PKT_RX_IPV6_HDR | PKT_RX_VLAN_PKT;
-                       else
-                               pkt->ol_flags = PKT_RX_IPV6_HDR;
+               pkts_burst[nb_pkt] = pkt;
+       }
+
+       return nb_pkt;
+}
+
+int
+generate_packet_burst_proto(struct rte_mempool *mp,
+               struct rte_mbuf **pkts_burst, struct rte_ether_hdr *eth_hdr,
+               uint8_t vlan_enabled, void *ip_hdr,
+               uint8_t ipv4, uint8_t proto, void *proto_hdr,
+               int nb_pkt_per_burst, uint8_t pkt_len, uint8_t nb_pkt_segs)
+{
+       int i, nb_pkt = 0;
+       size_t eth_hdr_size;
+
+       struct rte_mbuf *pkt_seg;
+       struct rte_mbuf *pkt;
+
+       for (nb_pkt = 0; nb_pkt < nb_pkt_per_burst; nb_pkt++) {
+               pkt = rte_pktmbuf_alloc(mp);
+               if (pkt == NULL) {
+nomore_mbuf:
+                       if (nb_pkt == 0)
+                               return -1;
+                       break;
+               }
+
+               pkt->data_len = pkt_len;
+               pkt_seg = pkt;
+               for (i = 1; i < nb_pkt_segs; i++) {
+                       pkt_seg->next = rte_pktmbuf_alloc(mp);
+                       if (pkt_seg->next == NULL) {
+                               pkt->nb_segs = i;
+                               rte_pktmbuf_free(pkt);
+                               goto nomore_mbuf;
+                       }
+                       pkt_seg = pkt_seg->next;
+                       pkt_seg->data_len = pkt_len;
+               }
+               pkt_seg->next = NULL; /* Last segment of packet. */
+
+               /*
+                * Copy headers in first packet segment(s).
+                */
+               if (vlan_enabled)
+                       eth_hdr_size = sizeof(struct rte_ether_hdr) +
+                               sizeof(struct rte_vlan_hdr);
+               else
+                       eth_hdr_size = sizeof(struct rte_ether_hdr);
+
+               copy_buf_to_pkt(eth_hdr, eth_hdr_size, pkt, 0);
+
+               if (ipv4) {
+                       copy_buf_to_pkt(ip_hdr, sizeof(struct ipv4_hdr), pkt,
+                               eth_hdr_size);
+                       switch (proto) {
+                       case IPPROTO_UDP:
+                               copy_buf_to_pkt(proto_hdr,
+                                       sizeof(struct udp_hdr), pkt,
+                                       eth_hdr_size + sizeof(struct ipv4_hdr));
+                               break;
+                       case IPPROTO_TCP:
+                               copy_buf_to_pkt(proto_hdr,
+                                       sizeof(struct tcp_hdr), pkt,
+                                       eth_hdr_size + sizeof(struct ipv4_hdr));
+                               break;
+                       case IPPROTO_SCTP:
+                               copy_buf_to_pkt(proto_hdr,
+                                       sizeof(struct sctp_hdr), pkt,
+                                       eth_hdr_size + sizeof(struct ipv4_hdr));
+                               break;
+                       default:
+                               break;
+                       }
+               } else {
+                       copy_buf_to_pkt(ip_hdr, sizeof(struct ipv6_hdr), pkt,
+                               eth_hdr_size);
+                       switch (proto) {
+                       case IPPROTO_UDP:
+                               copy_buf_to_pkt(proto_hdr,
+                                       sizeof(struct udp_hdr), pkt,
+                                       eth_hdr_size + sizeof(struct ipv6_hdr));
+                               break;
+                       case IPPROTO_TCP:
+                               copy_buf_to_pkt(proto_hdr,
+                                       sizeof(struct tcp_hdr), pkt,
+                                       eth_hdr_size + sizeof(struct ipv6_hdr));
+                               break;
+                       case IPPROTO_SCTP:
+                               copy_buf_to_pkt(proto_hdr,
+                                       sizeof(struct sctp_hdr), pkt,
+                                       eth_hdr_size + sizeof(struct ipv6_hdr));
+                               break;
+                       default:
+                               break;
+                       }
+               }
+
+               /*
+                * Complete first mbuf of packet and append it to the
+                * burst of packets to be transmitted.
+                */
+               pkt->nb_segs = nb_pkt_segs;
+               pkt->pkt_len = pkt_len;
+               pkt->l2_len = eth_hdr_size;
+
+               if (ipv4) {
+                       pkt->vlan_tci  = ETHER_TYPE_IPv4;
+                       pkt->l3_len = sizeof(struct ipv4_hdr);
+               } else {
+                       pkt->vlan_tci  = ETHER_TYPE_IPv6;
+                       pkt->l3_len = sizeof(struct ipv6_hdr);
                }
 
                pkts_burst[nb_pkt] = pkt;