lib/librte_ip_frag/ip_frag_common.h

   1 /* SPDX-License-Identifier: BSD-3-Clause
   2  * Copyright(c) 2010-2014 Intel Corporation
   3  */
   4
   5 #ifndef _IP_FRAG_COMMON_H_
   6 #define _IP_FRAG_COMMON_H_
   7
   8 #include "rte_ip_frag.h"
   9
  10 /* logging macros. */
  11 #ifdef RTE_LIBRTE_IP_FRAG_DEBUG
  12 #define IP_FRAG_LOG(lvl, fmt, args...)  RTE_LOG(lvl, USER1, fmt, ##args)
  13 #else
  14 #define IP_FRAG_LOG(lvl, fmt, args...)  do {} while(0)
  15 #endif /* IP_FRAG_DEBUG */
  16
  17 #define IPV4_KEYLEN 1
  18 #define IPV6_KEYLEN 4
  19
  20 /* helper macros */
  21 #define IP_FRAG_MBUF2DR(dr, mb) ((dr)->row[(dr)->cnt++] = (mb))
  22
  23 #define IPv6_KEY_BYTES(key) \
  24         (key)[0], (key)[1], (key)[2], (key)[3]
  25 #define IPv6_KEY_BYTES_FMT \
  26         "%08" PRIx64 "%08" PRIx64 "%08" PRIx64 "%08" PRIx64
  27
  28 #ifdef RTE_LIBRTE_IP_FRAG_TBL_STAT
  29 #define IP_FRAG_TBL_STAT_UPDATE(s, f, v)        ((s)->f += (v))
  30 #else
  31 #define IP_FRAG_TBL_STAT_UPDATE(s, f, v)        do {} while (0)
  32 #endif /* IP_FRAG_TBL_STAT */
  33
  34 /* internal functions declarations */
  35 struct rte_mbuf * ip_frag_process(struct ip_frag_pkt *fp,
  36                 struct rte_ip_frag_death_row *dr, struct rte_mbuf *mb,
  37                 uint16_t ofs, uint16_t len, uint16_t more_frags);
  38
  39 struct ip_frag_pkt * ip_frag_find(struct rte_ip_frag_tbl *tbl,
  40                 struct rte_ip_frag_death_row *dr,
  41                 const struct ip_frag_key *key, uint64_t tms);
  42
  43 struct ip_frag_pkt * ip_frag_lookup(struct rte_ip_frag_tbl *tbl,
  44         const struct ip_frag_key *key, uint64_t tms,
  45         struct ip_frag_pkt **free, struct ip_frag_pkt **stale);
  46
  47 /* these functions need to be declared here as ip_frag_process relies on them */
  48 struct rte_mbuf *ipv4_frag_reassemble(struct ip_frag_pkt *fp);
  49 struct rte_mbuf *ipv6_frag_reassemble(struct ip_frag_pkt *fp);
  50
  51
  52
  53 /*
  54  * misc frag key functions
  55  */
  56
  57 /* check if key is empty */
  58 static inline int
  59 ip_frag_key_is_empty(const struct ip_frag_key * key)
  60 {
  61         uint32_t i;
  62         for (i = 0; i < RTE_MIN(key->key_len, RTE_DIM(key->src_dst)); i++)
  63                 if (key->src_dst[i] != 0)
  64                         return 0;
  65         return 1;
  66 }
  67
  68 /* empty the key */
  69 static inline void
  70 ip_frag_key_invalidate(struct ip_frag_key * key)
  71 {
  72         uint32_t i;
  73         for (i = 0; i < key->key_len; i++)
  74                 key->src_dst[i] = 0;
  75 }
  76
  77 /* compare two keys */
  78 static inline uint64_t
  79 ip_frag_key_cmp(const struct ip_frag_key * k1, const struct ip_frag_key * k2)
  80 {
  81         uint32_t i;
  82         uint64_t val;
  83         val = k1->id ^ k2->id;
  84         for (i = 0; i < k1->key_len; i++)
  85                 val |= k1->src_dst[i] ^ k2->src_dst[i];
  86         return val;
  87 }
  88
  89 /*
  90  * misc fragment functions
  91  */
  92
  93 /* put fragment on death row */
  94 static inline void
  95 ip_frag_free(struct ip_frag_pkt *fp, struct rte_ip_frag_death_row *dr)
  96 {
  97         uint32_t i, k;
  98
  99         k = dr->cnt;
 100         for (i = 0; i != fp->last_idx; i++) {
 101                 if (fp->frags[i].mb != NULL) {
 102                         dr->row[k++] = fp->frags[i].mb;
 103                         fp->frags[i].mb = NULL;
 104                 }
 105         }
 106
 107         fp->last_idx = 0;
 108         dr->cnt = k;
 109 }
 110
 111 /* delete fragment's mbufs immediately instead of using death row */
 112 static inline void
 113 ip_frag_free_immediate(struct ip_frag_pkt *fp)
 114 {
 115         uint32_t i;
 116
 117         for (i = 0; i < fp->last_idx; i++) {
 118                 if (fp->frags[i].mb != NULL) {
 119                         IP_FRAG_LOG(DEBUG, "%s:%d\n"
 120                             "mbuf: %p, tms: %" PRIu64", key: <%" PRIx64 ", %#x>\n",
 121                             __func__, __LINE__, fp->frags[i].mb, fp->start,
 122                             fp->key.src_dst[0], fp->key.id);
 123                         rte_pktmbuf_free(fp->frags[i].mb);
 124                         fp->frags[i].mb = NULL;
 125                 }
 126         }
 127
 128         fp->last_idx = 0;
 129 }
 130
 131 /* if key is empty, mark key as in use */
 132 static inline void
 133 ip_frag_inuse(struct rte_ip_frag_tbl *tbl, const struct  ip_frag_pkt *fp)
 134 {
 135         if (ip_frag_key_is_empty(&fp->key)) {
 136                 TAILQ_REMOVE(&tbl->lru, fp, lru);
 137                 tbl->use_entries--;
 138         }
 139 }
 140
 141 /* reset the fragment */
 142 static inline void
 143 ip_frag_reset(struct ip_frag_pkt *fp, uint64_t tms)
 144 {
 145         static const struct ip_frag zero_frag = {
 146                 .ofs = 0,
 147                 .len = 0,
 148                 .mb = NULL,
 149         };
 150
 151         fp->start = tms;
 152         fp->total_size = UINT32_MAX;
 153         fp->frag_size = 0;
 154         fp->last_idx = IP_MIN_FRAG_NUM;
 155         fp->frags[IP_LAST_FRAG_IDX] = zero_frag;
 156         fp->frags[IP_FIRST_FRAG_IDX] = zero_frag;
 157 }
 158
 159 /* local frag table helper functions */
 160 static inline void
 161 ip_frag_tbl_del(struct rte_ip_frag_tbl *tbl, struct rte_ip_frag_death_row *dr,
 162         struct ip_frag_pkt *fp)
 163 {
 164         ip_frag_free(fp, dr);
 165         ip_frag_key_invalidate(&fp->key);
 166         TAILQ_REMOVE(&tbl->lru, fp, lru);
 167         tbl->use_entries--;
 168         IP_FRAG_TBL_STAT_UPDATE(&tbl->stat, del_num, 1);
 169 }
 170
 171 #endif /* _IP_FRAG_COMMON_H_ */