include/net/netfilter/nf_conntrack_act_ct.h

   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
  10 struct nf_conn_act_ct_ext {
  11         int ifindex[IP_CT_DIR_MAX];
  12 };
  13
  14 static 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
  23 static inline struct nf_conn_act_ct_ext *nf_conn_act_ct_ext_add(struct nf_conn *ct)
  24 {
  25 #if IS_ENABLED(CONFIG_NET_ACT_CT)
  26         struct nf_conn_act_ct_ext *act_ct = nf_ct_ext_find(ct, NF_CT_EXT_ACT_CT);
  27
  28         if (act_ct)
  29                 return act_ct;
  30
  31         act_ct = nf_ct_ext_add(ct, NF_CT_EXT_ACT_CT, GFP_ATOMIC);
  32         return act_ct;
  33 #else
  34         return NULL;
  35 #endif
  36 }
  37
  38 static inline void nf_conn_act_ct_ext_fill(struct sk_buff *skb, struct nf_conn *ct,
  39                                            enum ip_conntrack_info ctinfo)
  40 {
  41 #if IS_ENABLED(CONFIG_NET_ACT_CT)
  42         struct nf_conn_act_ct_ext *act_ct_ext;
  43
  44         act_ct_ext = nf_conn_act_ct_ext_find(ct);
  45         if (dev_net(skb->dev) == &init_net && act_ct_ext)
  46                 act_ct_ext->ifindex[CTINFO2DIR(ctinfo)] = skb->dev->ifindex;
  47 #endif
  48 }
  49
  50 #endif /* _NF_CONNTRACK_ACT_CT_H */