include/net/tc_act/tc_ct.h

   1 /* SPDX-License-Identifier: GPL-2.0 */
   2 #ifndef __NET_TC_CT_H
   3 #define __NET_TC_CT_H
   4
   5 #include <net/act_api.h>
   6 #include <uapi/linux/tc_act/tc_ct.h>
   7
   8 #if IS_ENABLED(CONFIG_NF_CONNTRACK)
   9 #include <net/netfilter/nf_nat.h>
  10 #include <net/netfilter/nf_conntrack_labels.h>
  11
  12 struct tcf_ct_params {
  13         struct nf_conntrack_helper *helper;
  14         struct nf_conn *tmpl;
  15         u16 zone;
  16
  17         u32 mark;
  18         u32 mark_mask;
  19
  20         u32 labels[NF_CT_LABELS_MAX_SIZE / sizeof(u32)];
  21         u32 labels_mask[NF_CT_LABELS_MAX_SIZE / sizeof(u32)];
  22
  23         struct nf_nat_range2 range;
  24         bool ipv4_range;
  25         bool put_labels;
  26
  27         u16 ct_action;
  28
  29         struct rcu_head rcu;
  30
  31         struct tcf_ct_flow_table *ct_ft;
  32         struct nf_flowtable *nf_ft;
  33 };
  34
  35 struct tcf_ct {
  36         struct tc_action common;
  37         struct tcf_ct_params __rcu *params;
  38 };
  39
  40 #define to_ct(a) ((struct tcf_ct *)a)
  41 #define to_ct_params(a)                                                 \
  42         ((struct tcf_ct_params *)                                       \
  43          rcu_dereference_protected(to_ct(a)->params,                    \
  44                                    lockdep_is_held(&a->tcfa_lock)))
  45
  46 static inline uint16_t tcf_ct_zone(const struct tc_action *a)
  47 {
  48         return to_ct_params(a)->zone;
  49 }
  50
  51 static inline int tcf_ct_action(const struct tc_action *a)
  52 {
  53         return to_ct_params(a)->ct_action;
  54 }
  55
  56 static inline struct nf_flowtable *tcf_ct_ft(const struct tc_action *a)
  57 {
  58         return to_ct_params(a)->nf_ft;
  59 }
  60
  61 static inline struct nf_conntrack_helper *tcf_ct_helper(const struct tc_action *a)
  62 {
  63         return to_ct_params(a)->helper;
  64 }
  65
  66 #else
  67 static inline uint16_t tcf_ct_zone(const struct tc_action *a) { return 0; }
  68 static inline int tcf_ct_action(const struct tc_action *a) { return 0; }
  69 static inline struct nf_flowtable *tcf_ct_ft(const struct tc_action *a)
  70 {
  71         return NULL;
  72 }
  73 static inline struct nf_conntrack_helper *tcf_ct_helper(const struct tc_action *a)
  74 {
  75         return NULL;
  76 }
  77 #endif /* CONFIG_NF_CONNTRACK */
  78
  79 #if IS_ENABLED(CONFIG_NET_ACT_CT)
  80 static inline void
  81 tcf_ct_flow_table_restore_skb(struct sk_buff *skb, unsigned long cookie)
  82 {
  83         enum ip_conntrack_info ctinfo = cookie & NFCT_INFOMASK;
  84         struct nf_conn *ct;
  85
  86         ct = (struct nf_conn *)(cookie & NFCT_PTRMASK);
  87         nf_conntrack_get(&ct->ct_general);
  88         nf_ct_set(skb, ct, ctinfo);
  89 }
  90 #else
  91 static inline void
  92 tcf_ct_flow_table_restore_skb(struct sk_buff *skb, unsigned long cookie) { }
  93 #endif
  94
  95 static inline bool is_tcf_ct(const struct tc_action *a)
  96 {
  97 #if defined(CONFIG_NET_CLS_ACT) && IS_ENABLED(CONFIG_NF_CONNTRACK)
  98         if (a->ops && a->ops->id == TCA_ID_CT)
  99                 return true;
 100 #endif
 101         return false;
 102 }
 103
 104 #endif /* __NET_TC_CT_H */