1/* SPDX-License-Identifier: GPL-2.0 */
2
3#ifndef _NF_CONNTRACK_ACT_CT_H
4#define _NF_CONNTRACK_ACT_CT_H
5
6#include <net/netfilter/nf_conntrack.h>
7#include <linux/netfilter/nf_conntrack_common.h>
8#include <net/netfilter/nf_conntrack_extend.h>
9
10struct nf_conn_act_ct_ext {
11	int ifindex[IP_CT_DIR_MAX];
12};
13
14static inline struct nf_conn_act_ct_ext *nf_conn_act_ct_ext_find(const struct nf_conn *ct)
15{
16#if IS_ENABLED(CONFIG_NET_ACT_CT)
17	return nf_ct_ext_find(ct, NF_CT_EXT_ACT_CT);
18#else
19	return NULL;
20#endif
21}
22
23static inline void nf_conn_act_ct_ext_fill(struct sk_buff *skb, struct nf_conn *ct,
24					   enum ip_conntrack_info ctinfo)
25{
26#if IS_ENABLED(CONFIG_NET_ACT_CT)
27	struct nf_conn_act_ct_ext *act_ct_ext;
28
29	act_ct_ext = nf_conn_act_ct_ext_find(ct);
30	if (dev_net(skb->dev) == &init_net && act_ct_ext)
31		act_ct_ext->ifindex[CTINFO2DIR(ctinfo)] = skb->dev->ifindex;
32#endif
33}
34
35static inline struct
36nf_conn_act_ct_ext *nf_conn_act_ct_ext_add(struct sk_buff *skb,
37					   struct nf_conn *ct,
38					   enum ip_conntrack_info ctinfo)
39{
40#if IS_ENABLED(CONFIG_NET_ACT_CT)
41	struct nf_conn_act_ct_ext *act_ct = nf_ct_ext_find(ct, NF_CT_EXT_ACT_CT);
42
43	if (act_ct)
44		return act_ct;
45
46	act_ct = nf_ct_ext_add(ct, NF_CT_EXT_ACT_CT, GFP_ATOMIC);
47	nf_conn_act_ct_ext_fill(skb, ct, ctinfo);
48	return act_ct;
49#else
50	return NULL;
51#endif
52}
53
54#endif /* _NF_CONNTRACK_ACT_CT_H */
55