net: add rte prefix to ether structures
[dpdk.git] / lib / librte_gro / gro_tcp4.h
index 49e03b4..988cca8 100644 (file)
@@ -5,6 +5,9 @@
 #ifndef _GRO_TCP4_H_
 #define _GRO_TCP4_H_
 
+#include <rte_ip.h>
+#include <rte_tcp.h>
+
 #define INVALID_ARRAY_INDEX 0xffffffffUL
 #define GRO_TCP4_TBL_MAX_ITEM_NUM (1024UL * 1024UL)
 
  */
 #define MAX_IPV4_PKT_LENGTH UINT16_MAX
 
+/* The maximum TCP header length */
+#define MAX_TCP_HLEN 60
+#define INVALID_TCP_HDRLEN(len) \
+       (((len) < sizeof(struct tcp_hdr)) || ((len) > MAX_TCP_HLEN))
+
 /* Header fields representing a TCP/IPv4 flow */
 struct tcp4_flow_key {
-       struct ether_addr eth_saddr;
-       struct ether_addr eth_daddr;
+       struct rte_ether_addr eth_saddr;
+       struct rte_ether_addr eth_daddr;
        uint32_t ip_src_addr;
        uint32_t ip_dst_addr;
 
@@ -61,6 +69,8 @@ struct gro_tcp4_item {
        uint16_t ip_id;
        /* the number of merged packets */
        uint16_t nb_merged;
+       /* Indicate if IPv4 ID can be ignored */
+       uint8_t is_atomic;
 };
 
 /*
@@ -170,4 +180,127 @@ uint16_t gro_tcp4_tbl_timeout_flush(struct gro_tcp4_tbl *tbl,
  *  The number of packets in the table
  */
 uint32_t gro_tcp4_tbl_pkt_count(void *tbl);
+
+/*
+ * Check if two TCP/IPv4 packets belong to the same flow.
+ */
+static inline int
+is_same_tcp4_flow(struct tcp4_flow_key k1, struct tcp4_flow_key k2)
+{
+       return (is_same_ether_addr(&k1.eth_saddr, &k2.eth_saddr) &&
+                       is_same_ether_addr(&k1.eth_daddr, &k2.eth_daddr) &&
+                       (k1.ip_src_addr == k2.ip_src_addr) &&
+                       (k1.ip_dst_addr == k2.ip_dst_addr) &&
+                       (k1.recv_ack == k2.recv_ack) &&
+                       (k1.src_port == k2.src_port) &&
+                       (k1.dst_port == k2.dst_port));
+}
+
+/*
+ * Merge two TCP/IPv4 packets without updating checksums.
+ * If cmp is larger than 0, append the new packet to the
+ * original packet. Otherwise, pre-pend the new packet to
+ * the original packet.
+ */
+static inline int
+merge_two_tcp4_packets(struct gro_tcp4_item *item,
+               struct rte_mbuf *pkt,
+               int cmp,
+               uint32_t sent_seq,
+               uint16_t ip_id,
+               uint16_t l2_offset)
+{
+       struct rte_mbuf *pkt_head, *pkt_tail, *lastseg;
+       uint16_t hdr_len, l2_len;
+
+       if (cmp > 0) {
+               pkt_head = item->firstseg;
+               pkt_tail = pkt;
+       } else {
+               pkt_head = pkt;
+               pkt_tail = item->firstseg;
+       }
+
+       /* check if the IPv4 packet length is greater than the max value */
+       hdr_len = l2_offset + pkt_head->l2_len + pkt_head->l3_len +
+               pkt_head->l4_len;
+       l2_len = l2_offset > 0 ? pkt_head->outer_l2_len : pkt_head->l2_len;
+       if (unlikely(pkt_head->pkt_len - l2_len + pkt_tail->pkt_len -
+                               hdr_len > MAX_IPV4_PKT_LENGTH))
+               return 0;
+
+       /* remove the packet header for the tail packet */
+       rte_pktmbuf_adj(pkt_tail, hdr_len);
+
+       /* chain two packets together */
+       if (cmp > 0) {
+               item->lastseg->next = pkt;
+               item->lastseg = rte_pktmbuf_lastseg(pkt);
+               /* update IP ID to the larger value */
+               item->ip_id = ip_id;
+       } else {
+               lastseg = rte_pktmbuf_lastseg(pkt);
+               lastseg->next = item->firstseg;
+               item->firstseg = pkt;
+               /* update sent_seq to the smaller value */
+               item->sent_seq = sent_seq;
+               item->ip_id = ip_id;
+       }
+       item->nb_merged++;
+
+       /* update MBUF metadata for the merged packet */
+       pkt_head->nb_segs += pkt_tail->nb_segs;
+       pkt_head->pkt_len += pkt_tail->pkt_len;
+
+       return 1;
+}
+
+/*
+ * Check if two TCP/IPv4 packets are neighbors.
+ */
+static inline int
+check_seq_option(struct gro_tcp4_item *item,
+               struct tcp_hdr *tcph,
+               uint32_t sent_seq,
+               uint16_t ip_id,
+               uint16_t tcp_hl,
+               uint16_t tcp_dl,
+               uint16_t l2_offset,
+               uint8_t is_atomic)
+{
+       struct rte_mbuf *pkt_orig = item->firstseg;
+       struct ipv4_hdr *iph_orig;
+       struct tcp_hdr *tcph_orig;
+       uint16_t len, tcp_hl_orig;
+
+       iph_orig = (struct ipv4_hdr *)(rte_pktmbuf_mtod(pkt_orig, char *) +
+                       l2_offset + pkt_orig->l2_len);
+       tcph_orig = (struct tcp_hdr *)((char *)iph_orig + pkt_orig->l3_len);
+       tcp_hl_orig = pkt_orig->l4_len;
+
+       /* Check if TCP option fields equal */
+       len = RTE_MAX(tcp_hl, tcp_hl_orig) - sizeof(struct tcp_hdr);
+       if ((tcp_hl != tcp_hl_orig) || ((len > 0) &&
+                               (memcmp(tcph + 1, tcph_orig + 1,
+                                       len) != 0)))
+               return 0;
+
+       /* Don't merge packets whose DF bits are different */
+       if (unlikely(item->is_atomic ^ is_atomic))
+               return 0;
+
+       /* check if the two packets are neighbors */
+       len = pkt_orig->pkt_len - l2_offset - pkt_orig->l2_len -
+               pkt_orig->l3_len - tcp_hl_orig;
+       if ((sent_seq == item->sent_seq + len) && (is_atomic ||
+                               (ip_id == item->ip_id + 1)))
+               /* append the new packet */
+               return 1;
+       else if ((sent_seq + tcp_dl == item->sent_seq) && (is_atomic ||
+                               (ip_id + item->nb_merged == item->ip_id)))
+               /* pre-pend the new packet */
+               return -1;
+
+       return 0;
+}
 #endif