1  /* SPDX-License-Identifier: GPL-2.0 */
2  #ifndef __LINUX_NETLINK_H
3  #define __LINUX_NETLINK_H
4  
5  
6  #include <linux/capability.h>
7  #include <linux/skbuff.h>
8  #include <linux/export.h>
9  #include <net/scm.h>
10  #include <uapi/linux/netlink.h>
11  
12  struct net;
13  
14  void do_trace_netlink_extack(const char *msg);
15  
nlmsg_hdr(const struct sk_buff * skb)16  static inline struct nlmsghdr *nlmsg_hdr(const struct sk_buff *skb)
17  {
18  	return (struct nlmsghdr *)skb->data;
19  }
20  
21  enum netlink_skb_flags {
22  	NETLINK_SKB_DST		= 0x8,	/* Dst set in sendto or sendmsg */
23  };
24  
25  struct netlink_skb_parms {
26  	struct scm_creds	creds;		/* Skb credentials	*/
27  	__u32			portid;
28  	__u32			dst_group;
29  	__u32			flags;
30  	struct sock		*sk;
31  	bool			nsid_is_set;
32  	int			nsid;
33  };
34  
35  #define NETLINK_CB(skb)		(*(struct netlink_skb_parms*)&((skb)->cb))
36  #define NETLINK_CREDS(skb)	(&NETLINK_CB((skb)).creds)
37  
38  
39  void netlink_table_grab(void);
40  void netlink_table_ungrab(void);
41  
42  #define NL_CFG_F_NONROOT_RECV	(1 << 0)
43  #define NL_CFG_F_NONROOT_SEND	(1 << 1)
44  
45  /* optional Netlink kernel configuration parameters */
46  struct netlink_kernel_cfg {
47  	unsigned int	groups;
48  	unsigned int	flags;
49  	void		(*input)(struct sk_buff *skb);
50  	int		(*bind)(struct net *net, int group);
51  	void		(*unbind)(struct net *net, int group);
52  	void            (*release) (struct sock *sk, unsigned long *groups);
53  };
54  
55  struct sock *__netlink_kernel_create(struct net *net, int unit,
56  					    struct module *module,
57  					    struct netlink_kernel_cfg *cfg);
58  static inline struct sock *
netlink_kernel_create(struct net * net,int unit,struct netlink_kernel_cfg * cfg)59  netlink_kernel_create(struct net *net, int unit, struct netlink_kernel_cfg *cfg)
60  {
61  	return __netlink_kernel_create(net, unit, THIS_MODULE, cfg);
62  }
63  
64  /* this can be increased when necessary - don't expose to userland */
65  #define NETLINK_MAX_COOKIE_LEN	20
66  #define NETLINK_MAX_FMTMSG_LEN	80
67  
68  /**
69   * struct netlink_ext_ack - netlink extended ACK report struct
70   * @_msg: message string to report - don't access directly, use
71   *	%NL_SET_ERR_MSG
72   * @bad_attr: attribute with error
73   * @policy: policy for a bad attribute
74   * @miss_type: attribute type which was missing
75   * @miss_nest: nest missing an attribute (%NULL if missing top level attr)
76   * @cookie: cookie data to return to userspace (for success)
77   * @cookie_len: actual cookie data length
78   * @_msg_buf: output buffer for formatted message strings - don't access
79   *	directly, use %NL_SET_ERR_MSG_FMT
80   */
81  struct netlink_ext_ack {
82  	const char *_msg;
83  	const struct nlattr *bad_attr;
84  	const struct nla_policy *policy;
85  	const struct nlattr *miss_nest;
86  	u16 miss_type;
87  	u8 cookie[NETLINK_MAX_COOKIE_LEN];
88  	u8 cookie_len;
89  	char _msg_buf[NETLINK_MAX_FMTMSG_LEN];
90  };
91  
92  /* Always use this macro, this allows later putting the
93   * message into a separate section or such for things
94   * like translation or listing all possible messages.
95   * If string formatting is needed use NL_SET_ERR_MSG_FMT.
96   */
97  #define NL_SET_ERR_MSG(extack, msg) do {		\
98  	static const char __msg[] = msg;		\
99  	struct netlink_ext_ack *__extack = (extack);	\
100  							\
101  	do_trace_netlink_extack(__msg);			\
102  							\
103  	if (__extack)					\
104  		__extack->_msg = __msg;			\
105  } while (0)
106  
107  /* We splice fmt with %s at each end even in the snprintf so that both calls
108   * can use the same string constant, avoiding its duplication in .ro
109   */
110  #define NL_SET_ERR_MSG_FMT(extack, fmt, args...) do {			       \
111  	struct netlink_ext_ack *__extack = (extack);			       \
112  									       \
113  	if (!__extack)							       \
114  		break;							       \
115  	if (snprintf(__extack->_msg_buf, NETLINK_MAX_FMTMSG_LEN,	       \
116  		     "%s" fmt "%s", "", ##args, "") >=			       \
117  	    NETLINK_MAX_FMTMSG_LEN)					       \
118  		net_warn_ratelimited("%s" fmt "%s", "truncated extack: ",      \
119  				     ##args, "\n");			       \
120  									       \
121  	do_trace_netlink_extack(__extack->_msg_buf);			       \
122  									       \
123  	__extack->_msg = __extack->_msg_buf;				       \
124  } while (0)
125  
126  #define NL_SET_ERR_MSG_MOD(extack, msg)			\
127  	NL_SET_ERR_MSG((extack), KBUILD_MODNAME ": " msg)
128  
129  #define NL_SET_ERR_MSG_FMT_MOD(extack, fmt, args...)	\
130  	NL_SET_ERR_MSG_FMT((extack), KBUILD_MODNAME ": " fmt, ##args)
131  
132  #define NL_SET_ERR_MSG_WEAK(extack, msg) do {		\
133  	if ((extack) && !(extack)->_msg)		\
134  		NL_SET_ERR_MSG((extack), msg);		\
135  } while (0)
136  
137  #define NL_SET_ERR_MSG_WEAK_MOD(extack, msg) do {	\
138  	if ((extack) && !(extack)->_msg)		\
139  		NL_SET_ERR_MSG_MOD((extack), msg);	\
140  } while (0)
141  
142  #define NL_SET_BAD_ATTR_POLICY(extack, attr, pol) do {	\
143  	if ((extack)) {					\
144  		(extack)->bad_attr = (attr);		\
145  		(extack)->policy = (pol);		\
146  	}						\
147  } while (0)
148  
149  #define NL_SET_BAD_ATTR(extack, attr) NL_SET_BAD_ATTR_POLICY(extack, attr, NULL)
150  
151  #define NL_SET_ERR_MSG_ATTR_POL(extack, attr, pol, msg) do {	\
152  	static const char __msg[] = msg;			\
153  	struct netlink_ext_ack *__extack = (extack);		\
154  								\
155  	do_trace_netlink_extack(__msg);				\
156  								\
157  	if (__extack) {						\
158  		__extack->_msg = __msg;				\
159  		__extack->bad_attr = (attr);			\
160  		__extack->policy = (pol);			\
161  	}							\
162  } while (0)
163  
164  #define NL_SET_ERR_MSG_ATTR_POL_FMT(extack, attr, pol, fmt, args...) do {	\
165  	struct netlink_ext_ack *__extack = (extack);				\
166  										\
167  	if (!__extack)								\
168  		break;								\
169  										\
170  	if (snprintf(__extack->_msg_buf, NETLINK_MAX_FMTMSG_LEN,		\
171  		     "%s" fmt "%s", "", ##args, "") >=				\
172  	    NETLINK_MAX_FMTMSG_LEN)						\
173  		net_warn_ratelimited("%s" fmt "%s", "truncated extack: ",       \
174  				     ##args, "\n");				\
175  										\
176  	do_trace_netlink_extack(__extack->_msg_buf);				\
177  										\
178  	__extack->_msg = __extack->_msg_buf;					\
179  	__extack->bad_attr = (attr);						\
180  	__extack->policy = (pol);						\
181  } while (0)
182  
183  #define NL_SET_ERR_MSG_ATTR(extack, attr, msg)		\
184  	NL_SET_ERR_MSG_ATTR_POL(extack, attr, NULL, msg)
185  
186  #define NL_SET_ERR_MSG_ATTR_FMT(extack, attr, msg, args...) \
187  	NL_SET_ERR_MSG_ATTR_POL_FMT(extack, attr, NULL, msg, ##args)
188  
189  #define NL_SET_ERR_ATTR_MISS(extack, nest, type)  do {	\
190  	struct netlink_ext_ack *__extack = (extack);	\
191  							\
192  	if (__extack) {					\
193  		__extack->miss_nest = (nest);		\
194  		__extack->miss_type = (type);		\
195  	}						\
196  } while (0)
197  
198  #define NL_REQ_ATTR_CHECK(extack, nest, tb, type) ({		\
199  	struct nlattr **__tb = (tb);				\
200  	u32 __attr = (type);					\
201  	int __retval;						\
202  								\
203  	__retval = !__tb[__attr];				\
204  	if (__retval)						\
205  		NL_SET_ERR_ATTR_MISS((extack), (nest), __attr);	\
206  	__retval;						\
207  })
208  
nl_set_extack_cookie_u64(struct netlink_ext_ack * extack,u64 cookie)209  static inline void nl_set_extack_cookie_u64(struct netlink_ext_ack *extack,
210  					    u64 cookie)
211  {
212  	if (!extack)
213  		return;
214  	memcpy(extack->cookie, &cookie, sizeof(cookie));
215  	extack->cookie_len = sizeof(cookie);
216  }
217  
218  void netlink_kernel_release(struct sock *sk);
219  int __netlink_change_ngroups(struct sock *sk, unsigned int groups);
220  int netlink_change_ngroups(struct sock *sk, unsigned int groups);
221  void __netlink_clear_multicast_users(struct sock *sk, unsigned int group);
222  void netlink_ack(struct sk_buff *in_skb, struct nlmsghdr *nlh, int err,
223  		 const struct netlink_ext_ack *extack);
224  int netlink_has_listeners(struct sock *sk, unsigned int group);
225  bool netlink_strict_get_check(struct sk_buff *skb);
226  
227  int netlink_unicast(struct sock *ssk, struct sk_buff *skb, __u32 portid, int nonblock);
228  int netlink_broadcast(struct sock *ssk, struct sk_buff *skb, __u32 portid,
229  		      __u32 group, gfp_t allocation);
230  
231  typedef int (*netlink_filter_fn)(struct sock *dsk, struct sk_buff *skb, void *data);
232  
233  int netlink_broadcast_filtered(struct sock *ssk, struct sk_buff *skb,
234  			       __u32 portid, __u32 group, gfp_t allocation,
235  			       netlink_filter_fn filter,
236  			       void *filter_data);
237  int netlink_set_err(struct sock *ssk, __u32 portid, __u32 group, int code);
238  int netlink_register_notifier(struct notifier_block *nb);
239  int netlink_unregister_notifier(struct notifier_block *nb);
240  
241  /* finegrained unicast helpers: */
242  struct sock *netlink_getsockbyfilp(struct file *filp);
243  int netlink_attachskb(struct sock *sk, struct sk_buff *skb,
244  		      long *timeo, struct sock *ssk);
245  void netlink_detachskb(struct sock *sk, struct sk_buff *skb);
246  int netlink_sendskb(struct sock *sk, struct sk_buff *skb);
247  
248  static inline struct sk_buff *
netlink_skb_clone(struct sk_buff * skb,gfp_t gfp_mask)249  netlink_skb_clone(struct sk_buff *skb, gfp_t gfp_mask)
250  {
251  	struct sk_buff *nskb;
252  
253  	nskb = skb_clone(skb, gfp_mask);
254  	if (!nskb)
255  		return NULL;
256  
257  	/* This is a large skb, set destructor callback to release head */
258  	if (is_vmalloc_addr(skb->head))
259  		nskb->destructor = skb->destructor;
260  
261  	return nskb;
262  }
263  
264  /*
265   *	skb should fit one page. This choice is good for headerless malloc.
266   *	But we should limit to 8K so that userspace does not have to
267   *	use enormous buffer sizes on recvmsg() calls just to avoid
268   *	MSG_TRUNC when PAGE_SIZE is very large.
269   */
270  #if PAGE_SIZE < 8192UL
271  #define NLMSG_GOODSIZE	SKB_WITH_OVERHEAD(PAGE_SIZE)
272  #else
273  #define NLMSG_GOODSIZE	SKB_WITH_OVERHEAD(8192UL)
274  #endif
275  
276  #define NLMSG_DEFAULT_SIZE (NLMSG_GOODSIZE - NLMSG_HDRLEN)
277  
278  
279  struct netlink_callback {
280  	struct sk_buff		*skb;
281  	const struct nlmsghdr	*nlh;
282  	int			(*dump)(struct sk_buff * skb,
283  					struct netlink_callback *cb);
284  	int			(*done)(struct netlink_callback *cb);
285  	void			*data;
286  	/* the module that dump function belong to */
287  	struct module		*module;
288  	struct netlink_ext_ack	*extack;
289  	u16			family;
290  	u16			answer_flags;
291  	u32			min_dump_alloc;
292  	unsigned int		prev_seq, seq;
293  	int			flags;
294  	bool			strict_check;
295  	union {
296  		u8		ctx[48];
297  
298  		/* args is deprecated. Cast a struct over ctx instead
299  		 * for proper type safety.
300  		 */
301  		long		args[6];
302  	};
303  };
304  
305  #define NL_ASSERT_DUMP_CTX_FITS(type_name)				\
306  	BUILD_BUG_ON(sizeof(type_name) >				\
307  		     sizeof_field(struct netlink_callback, ctx))
308  
309  struct netlink_notify {
310  	struct net *net;
311  	u32 portid;
312  	int protocol;
313  };
314  
315  struct nlmsghdr *
316  __nlmsg_put(struct sk_buff *skb, u32 portid, u32 seq, int type, int len, int flags);
317  
318  struct netlink_dump_control {
319  	int (*start)(struct netlink_callback *);
320  	int (*dump)(struct sk_buff *skb, struct netlink_callback *);
321  	int (*done)(struct netlink_callback *);
322  	struct netlink_ext_ack *extack;
323  	void *data;
324  	struct module *module;
325  	u32 min_dump_alloc;
326  	int flags;
327  };
328  
329  int __netlink_dump_start(struct sock *ssk, struct sk_buff *skb,
330  				const struct nlmsghdr *nlh,
331  				struct netlink_dump_control *control);
netlink_dump_start(struct sock * ssk,struct sk_buff * skb,const struct nlmsghdr * nlh,struct netlink_dump_control * control)332  static inline int netlink_dump_start(struct sock *ssk, struct sk_buff *skb,
333  				     const struct nlmsghdr *nlh,
334  				     struct netlink_dump_control *control)
335  {
336  	if (!control->module)
337  		control->module = THIS_MODULE;
338  
339  	return __netlink_dump_start(ssk, skb, nlh, control);
340  }
341  
342  struct netlink_tap {
343  	struct net_device *dev;
344  	struct module *module;
345  	struct list_head list;
346  };
347  
348  int netlink_add_tap(struct netlink_tap *nt);
349  int netlink_remove_tap(struct netlink_tap *nt);
350  
351  bool __netlink_ns_capable(const struct netlink_skb_parms *nsp,
352  			  struct user_namespace *ns, int cap);
353  bool netlink_ns_capable(const struct sk_buff *skb,
354  			struct user_namespace *ns, int cap);
355  bool netlink_capable(const struct sk_buff *skb, int cap);
356  bool netlink_net_capable(const struct sk_buff *skb, int cap);
357  struct sk_buff *netlink_alloc_large_skb(unsigned int size, int broadcast);
358  
359  #endif	/* __LINUX_NETLINK_H */
360