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 */ 105