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 #else 62 static inline uint16_t tcf_ct_zone(const struct tc_action *a) { return 0; } 63 static inline int tcf_ct_action(const struct tc_action *a) { return 0; } 64 static inline struct nf_flowtable *tcf_ct_ft(const struct tc_action *a) 65 { 66 return NULL; 67 } 68 #endif /* CONFIG_NF_CONNTRACK */ 69 70 #if IS_ENABLED(CONFIG_NET_ACT_CT) 71 static inline void 72 tcf_ct_flow_table_restore_skb(struct sk_buff *skb, unsigned long cookie) 73 { 74 enum ip_conntrack_info ctinfo = cookie & NFCT_INFOMASK; 75 struct nf_conn *ct; 76 77 ct = (struct nf_conn *)(cookie & NFCT_PTRMASK); 78 nf_conntrack_get(&ct->ct_general); 79 nf_ct_set(skb, ct, ctinfo); 80 } 81 #else 82 static inline void 83 tcf_ct_flow_table_restore_skb(struct sk_buff *skb, unsigned long cookie) { } 84 #endif 85 86 static inline bool is_tcf_ct(const struct tc_action *a) 87 { 88 #if defined(CONFIG_NET_CLS_ACT) && IS_ENABLED(CONFIG_NF_CONNTRACK) 89 if (a->ops && a->ops->id == TCA_ID_CT) 90 return true; 91 #endif 92 return false; 93 } 94 95 #endif /* __NET_TC_CT_H */ 96