mirror of
				https://github.com/torvalds/linux.git
				synced 2025-11-04 10:40:15 +02:00 
			
		
		
		
	refcount_t type should be used instead of int when fib_treeref is used as a reference counter,and avoid use-after-free risks. Signed-off-by: Yajun Deng <yajun.deng@linux.dev> Reviewed-by: David Ahern <dsahern@kernel.org> Link: https://lore.kernel.org/r/20210729071350.28919-1-yajun.deng@linux.dev Signed-off-by: Jakub Kicinski <kuba@kernel.org>
		
			
				
	
	
		
			167 lines
		
	
	
	
		
			4 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			167 lines
		
	
	
	
		
			4 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
/* SPDX-License-Identifier: GPL-2.0 */
 | 
						|
#ifndef _NET_DN_FIB_H
 | 
						|
#define _NET_DN_FIB_H
 | 
						|
 | 
						|
#include <linux/netlink.h>
 | 
						|
#include <linux/refcount.h>
 | 
						|
 | 
						|
extern const struct nla_policy rtm_dn_policy[];
 | 
						|
 | 
						|
struct dn_fib_res {
 | 
						|
	struct fib_rule *r;
 | 
						|
	struct dn_fib_info *fi;
 | 
						|
	unsigned char prefixlen;
 | 
						|
	unsigned char nh_sel;
 | 
						|
	unsigned char type;
 | 
						|
	unsigned char scope;
 | 
						|
};
 | 
						|
 | 
						|
struct dn_fib_nh {
 | 
						|
	struct net_device	*nh_dev;
 | 
						|
	unsigned int		nh_flags;
 | 
						|
	unsigned char		nh_scope;
 | 
						|
	int			nh_weight;
 | 
						|
	int			nh_power;
 | 
						|
	int			nh_oif;
 | 
						|
	__le16			nh_gw;
 | 
						|
};
 | 
						|
 | 
						|
struct dn_fib_info {
 | 
						|
	struct dn_fib_info	*fib_next;
 | 
						|
	struct dn_fib_info	*fib_prev;
 | 
						|
	refcount_t		fib_treeref;
 | 
						|
	refcount_t		fib_clntref;
 | 
						|
	int			fib_dead;
 | 
						|
	unsigned int		fib_flags;
 | 
						|
	int			fib_protocol;
 | 
						|
	__le16			fib_prefsrc;
 | 
						|
	__u32			fib_priority;
 | 
						|
	__u32			fib_metrics[RTAX_MAX];
 | 
						|
	int			fib_nhs;
 | 
						|
	int			fib_power;
 | 
						|
	struct dn_fib_nh	fib_nh[0];
 | 
						|
#define dn_fib_dev		fib_nh[0].nh_dev
 | 
						|
};
 | 
						|
 | 
						|
 | 
						|
#define DN_FIB_RES_RESET(res)	((res).nh_sel = 0)
 | 
						|
#define DN_FIB_RES_NH(res)	((res).fi->fib_nh[(res).nh_sel])
 | 
						|
 | 
						|
#define DN_FIB_RES_PREFSRC(res)	((res).fi->fib_prefsrc ? : __dn_fib_res_prefsrc(&res))
 | 
						|
#define DN_FIB_RES_GW(res)	(DN_FIB_RES_NH(res).nh_gw)
 | 
						|
#define DN_FIB_RES_DEV(res)	(DN_FIB_RES_NH(res).nh_dev)
 | 
						|
#define DN_FIB_RES_OIF(res)	(DN_FIB_RES_NH(res).nh_oif)
 | 
						|
 | 
						|
typedef struct {
 | 
						|
	__le16	datum;
 | 
						|
} dn_fib_key_t;
 | 
						|
 | 
						|
typedef struct {
 | 
						|
	__le16	datum;
 | 
						|
} dn_fib_hash_t;
 | 
						|
 | 
						|
typedef struct {
 | 
						|
	__u16	datum;
 | 
						|
} dn_fib_idx_t;
 | 
						|
 | 
						|
struct dn_fib_node {
 | 
						|
	struct dn_fib_node *fn_next;
 | 
						|
	struct dn_fib_info *fn_info;
 | 
						|
#define DN_FIB_INFO(f) ((f)->fn_info)
 | 
						|
	dn_fib_key_t	fn_key;
 | 
						|
	u8		fn_type;
 | 
						|
	u8		fn_scope;
 | 
						|
	u8		fn_state;
 | 
						|
};
 | 
						|
 | 
						|
 | 
						|
struct dn_fib_table {
 | 
						|
	struct hlist_node hlist;
 | 
						|
	u32 n;
 | 
						|
 | 
						|
	int (*insert)(struct dn_fib_table *t, struct rtmsg *r, 
 | 
						|
			struct nlattr *attrs[], struct nlmsghdr *n,
 | 
						|
			struct netlink_skb_parms *req);
 | 
						|
	int (*delete)(struct dn_fib_table *t, struct rtmsg *r,
 | 
						|
			struct nlattr *attrs[], struct nlmsghdr *n,
 | 
						|
			struct netlink_skb_parms *req);
 | 
						|
	int (*lookup)(struct dn_fib_table *t, const struct flowidn *fld,
 | 
						|
			struct dn_fib_res *res);
 | 
						|
	int (*flush)(struct dn_fib_table *t);
 | 
						|
	int (*dump)(struct dn_fib_table *t, struct sk_buff *skb, struct netlink_callback *cb);
 | 
						|
 | 
						|
	unsigned char data[];
 | 
						|
};
 | 
						|
 | 
						|
#ifdef CONFIG_DECNET_ROUTER
 | 
						|
/*
 | 
						|
 * dn_fib.c
 | 
						|
 */
 | 
						|
void dn_fib_init(void);
 | 
						|
void dn_fib_cleanup(void);
 | 
						|
 | 
						|
int dn_fib_ioctl(struct socket *sock, unsigned int cmd, unsigned long arg);
 | 
						|
struct dn_fib_info *dn_fib_create_info(const struct rtmsg *r,
 | 
						|
				       struct nlattr *attrs[],
 | 
						|
				       const struct nlmsghdr *nlh, int *errp);
 | 
						|
int dn_fib_semantic_match(int type, struct dn_fib_info *fi,
 | 
						|
			  const struct flowidn *fld, struct dn_fib_res *res);
 | 
						|
void dn_fib_release_info(struct dn_fib_info *fi);
 | 
						|
void dn_fib_flush(void);
 | 
						|
void dn_fib_select_multipath(const struct flowidn *fld, struct dn_fib_res *res);
 | 
						|
 | 
						|
/*
 | 
						|
 * dn_tables.c
 | 
						|
 */
 | 
						|
struct dn_fib_table *dn_fib_get_table(u32 n, int creat);
 | 
						|
struct dn_fib_table *dn_fib_empty_table(void);
 | 
						|
void dn_fib_table_init(void);
 | 
						|
void dn_fib_table_cleanup(void);
 | 
						|
 | 
						|
/*
 | 
						|
 * dn_rules.c
 | 
						|
 */
 | 
						|
void dn_fib_rules_init(void);
 | 
						|
void dn_fib_rules_cleanup(void);
 | 
						|
unsigned int dnet_addr_type(__le16 addr);
 | 
						|
int dn_fib_lookup(struct flowidn *fld, struct dn_fib_res *res);
 | 
						|
 | 
						|
int dn_fib_dump(struct sk_buff *skb, struct netlink_callback *cb);
 | 
						|
 | 
						|
void dn_fib_free_info(struct dn_fib_info *fi);
 | 
						|
 | 
						|
static inline void dn_fib_info_put(struct dn_fib_info *fi)
 | 
						|
{
 | 
						|
	if (refcount_dec_and_test(&fi->fib_clntref))
 | 
						|
		dn_fib_free_info(fi);
 | 
						|
}
 | 
						|
 | 
						|
static inline void dn_fib_res_put(struct dn_fib_res *res)
 | 
						|
{
 | 
						|
	if (res->fi)
 | 
						|
		dn_fib_info_put(res->fi);
 | 
						|
	if (res->r)
 | 
						|
		fib_rule_put(res->r);
 | 
						|
}
 | 
						|
 | 
						|
#else /* Endnode */
 | 
						|
 | 
						|
#define dn_fib_init()  do { } while(0)
 | 
						|
#define dn_fib_cleanup() do { } while(0)
 | 
						|
 | 
						|
#define dn_fib_lookup(fl, res) (-ESRCH)
 | 
						|
#define dn_fib_info_put(fi) do { } while(0)
 | 
						|
#define dn_fib_select_multipath(fl, res) do { } while(0)
 | 
						|
#define dn_fib_rules_policy(saddr,res,flags) (0)
 | 
						|
#define dn_fib_res_put(res) do { } while(0)
 | 
						|
 | 
						|
#endif /* CONFIG_DECNET_ROUTER */
 | 
						|
 | 
						|
static inline __le16 dnet_make_mask(int n)
 | 
						|
{
 | 
						|
	if (n)
 | 
						|
		return cpu_to_le16(~((1 << (16 - n)) - 1));
 | 
						|
	return cpu_to_le16(0);
 | 
						|
}
 | 
						|
 | 
						|
#endif /* _NET_DN_FIB_H */
 |