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
12struct 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
35struct 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
46static inline uint16_t tcf_ct_zone(const struct tc_action *a)
47{
48	return to_ct_params(a)->zone;
49}
50
51static inline int tcf_ct_action(const struct tc_action *a)
52{
53	return to_ct_params(a)->ct_action;
54}
55
56static inline struct nf_flowtable *tcf_ct_ft(const struct tc_action *a)
57{
58	return to_ct_params(a)->nf_ft;
59}
60
61static 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
67static inline uint16_t tcf_ct_zone(const struct tc_action *a) { return 0; }
68static inline int tcf_ct_action(const struct tc_action *a) { return 0; }
69static inline struct nf_flowtable *tcf_ct_ft(const struct tc_action *a)
70{
71	return NULL;
72}
73static 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)
80static inline void
81tcf_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
91static inline void
92tcf_ct_flow_table_restore_skb(struct sk_buff *skb, unsigned long cookie) { }
93#endif
94
95static 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