forked from mirrors/linux
		
	netfilter: Generalize ingress hook
Prepare for addition of a netfilter egress hook by generalizing the
ingress hook introduced by commit e687ad60af ("netfilter: add
netfilter ingress hook after handle_ing() under unique static key").
In particular, rename and refactor the ingress hook's static inlines
such that they can be reused for an egress hook.
No functional change intended.
Signed-off-by: Lukas Wunner <lukas@wunner.de>
Cc: Daniel Borkmann <daniel@iogearbox.net>
Signed-off-by: Pablo Neira Ayuso <pablo@netfilter.org>
			
			
This commit is contained in:
		
							parent
							
								
									b030f194ae
								
							
						
					
					
						commit
						5418d3881e
					
				
					 2 changed files with 32 additions and 15 deletions
				
			
		| 
						 | 
					@ -1,34 +1,37 @@
 | 
				
			||||||
/* SPDX-License-Identifier: GPL-2.0 */
 | 
					/* SPDX-License-Identifier: GPL-2.0 */
 | 
				
			||||||
#ifndef _NETFILTER_INGRESS_H_
 | 
					#ifndef _NETFILTER_NETDEV_H_
 | 
				
			||||||
#define _NETFILTER_INGRESS_H_
 | 
					#define _NETFILTER_NETDEV_H_
 | 
				
			||||||
 | 
					
 | 
				
			||||||
#include <linux/netfilter.h>
 | 
					#include <linux/netfilter.h>
 | 
				
			||||||
#include <linux/netdevice.h>
 | 
					#include <linux/netdevice.h>
 | 
				
			||||||
 | 
					
 | 
				
			||||||
#ifdef CONFIG_NETFILTER_INGRESS
 | 
					#ifdef CONFIG_NETFILTER
 | 
				
			||||||
static inline bool nf_hook_ingress_active(const struct sk_buff *skb)
 | 
					static __always_inline bool nf_hook_netdev_active(enum nf_dev_hooks hooknum,
 | 
				
			||||||
 | 
										  struct nf_hook_entries __rcu *hooks)
 | 
				
			||||||
{
 | 
					{
 | 
				
			||||||
#ifdef CONFIG_JUMP_LABEL
 | 
					#ifdef CONFIG_JUMP_LABEL
 | 
				
			||||||
	if (!static_key_false(&nf_hooks_needed[NFPROTO_NETDEV][NF_NETDEV_INGRESS]))
 | 
						if (!static_key_false(&nf_hooks_needed[NFPROTO_NETDEV][hooknum]))
 | 
				
			||||||
		return false;
 | 
							return false;
 | 
				
			||||||
#endif
 | 
					#endif
 | 
				
			||||||
	return rcu_access_pointer(skb->dev->nf_hooks_ingress);
 | 
						return rcu_access_pointer(hooks);
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
/* caller must hold rcu_read_lock */
 | 
					/* caller must hold rcu_read_lock */
 | 
				
			||||||
static inline int nf_hook_ingress(struct sk_buff *skb)
 | 
					static __always_inline int nf_hook_netdev(struct sk_buff *skb,
 | 
				
			||||||
 | 
										  enum nf_dev_hooks hooknum,
 | 
				
			||||||
 | 
										  struct nf_hook_entries __rcu *hooks)
 | 
				
			||||||
{
 | 
					{
 | 
				
			||||||
	struct nf_hook_entries *e = rcu_dereference(skb->dev->nf_hooks_ingress);
 | 
						struct nf_hook_entries *e = rcu_dereference(hooks);
 | 
				
			||||||
	struct nf_hook_state state;
 | 
						struct nf_hook_state state;
 | 
				
			||||||
	int ret;
 | 
						int ret;
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	/* Must recheck the ingress hook head, in the event it became NULL
 | 
						/* Must recheck the hook head, in the event it became NULL
 | 
				
			||||||
	 * after the check in nf_hook_ingress_active evaluated to true.
 | 
						 * after the check in nf_hook_netdev_active evaluated to true.
 | 
				
			||||||
	 */
 | 
						 */
 | 
				
			||||||
	if (unlikely(!e))
 | 
						if (unlikely(!e))
 | 
				
			||||||
		return 0;
 | 
							return 0;
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	nf_hook_state_init(&state, NF_NETDEV_INGRESS,
 | 
						nf_hook_state_init(&state, hooknum,
 | 
				
			||||||
			   NFPROTO_NETDEV, skb->dev, NULL, NULL,
 | 
								   NFPROTO_NETDEV, skb->dev, NULL, NULL,
 | 
				
			||||||
			   dev_net(skb->dev), NULL);
 | 
								   dev_net(skb->dev), NULL);
 | 
				
			||||||
	ret = nf_hook_slow(skb, &state, e, 0);
 | 
						ret = nf_hook_slow(skb, &state, e, 0);
 | 
				
			||||||
| 
						 | 
					@ -37,10 +40,26 @@ static inline int nf_hook_ingress(struct sk_buff *skb)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	return ret;
 | 
						return ret;
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					#endif /* CONFIG_NETFILTER */
 | 
				
			||||||
 | 
					
 | 
				
			||||||
static inline void nf_hook_ingress_init(struct net_device *dev)
 | 
					static inline void nf_hook_netdev_init(struct net_device *dev)
 | 
				
			||||||
{
 | 
					{
 | 
				
			||||||
 | 
					#ifdef CONFIG_NETFILTER_INGRESS
 | 
				
			||||||
	RCU_INIT_POINTER(dev->nf_hooks_ingress, NULL);
 | 
						RCU_INIT_POINTER(dev->nf_hooks_ingress, NULL);
 | 
				
			||||||
 | 
					#endif
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					#ifdef CONFIG_NETFILTER_INGRESS
 | 
				
			||||||
 | 
					static inline bool nf_hook_ingress_active(const struct sk_buff *skb)
 | 
				
			||||||
 | 
					{
 | 
				
			||||||
 | 
						return nf_hook_netdev_active(NF_NETDEV_INGRESS,
 | 
				
			||||||
 | 
									     skb->dev->nf_hooks_ingress);
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					static inline int nf_hook_ingress(struct sk_buff *skb)
 | 
				
			||||||
 | 
					{
 | 
				
			||||||
 | 
						return nf_hook_netdev(skb, NF_NETDEV_INGRESS,
 | 
				
			||||||
 | 
								      skb->dev->nf_hooks_ingress);
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
#else /* CONFIG_NETFILTER_INGRESS */
 | 
					#else /* CONFIG_NETFILTER_INGRESS */
 | 
				
			||||||
static inline int nf_hook_ingress_active(struct sk_buff *skb)
 | 
					static inline int nf_hook_ingress_active(struct sk_buff *skb)
 | 
				
			||||||
| 
						 | 
					@ -52,7 +71,5 @@ static inline int nf_hook_ingress(struct sk_buff *skb)
 | 
				
			||||||
{
 | 
					{
 | 
				
			||||||
	return 0;
 | 
						return 0;
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					 | 
				
			||||||
static inline void nf_hook_ingress_init(struct net_device *dev) {}
 | 
					 | 
				
			||||||
#endif /* CONFIG_NETFILTER_INGRESS */
 | 
					#endif /* CONFIG_NETFILTER_INGRESS */
 | 
				
			||||||
#endif /* _NETFILTER_INGRESS_H_ */
 | 
					#endif /* _NETFILTER_INGRESS_H_ */
 | 
				
			||||||
| 
						 | 
					
 | 
				
			||||||
| 
						 | 
					@ -9846,7 +9846,7 @@ struct net_device *alloc_netdev_mqs(int sizeof_priv, const char *name,
 | 
				
			||||||
	if (!dev->ethtool_ops)
 | 
						if (!dev->ethtool_ops)
 | 
				
			||||||
		dev->ethtool_ops = &default_ethtool_ops;
 | 
							dev->ethtool_ops = &default_ethtool_ops;
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	nf_hook_ingress_init(dev);
 | 
						nf_hook_netdev_init(dev);
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	return dev;
 | 
						return dev;
 | 
				
			||||||
 | 
					
 | 
				
			||||||
| 
						 | 
					
 | 
				
			||||||
		Loading…
	
		Reference in a new issue