xref: /openbmc/linux/net/netlink/af_netlink.c (revision bdd91001)
1 /*
2  * NETLINK      Kernel-user communication protocol.
3  *
4  * 		Authors:	Alan Cox <alan@lxorguk.ukuu.org.uk>
5  * 				Alexey Kuznetsov <kuznet@ms2.inr.ac.ru>
6  * 				Patrick McHardy <kaber@trash.net>
7  *
8  *		This program is free software; you can redistribute it and/or
9  *		modify it under the terms of the GNU General Public License
10  *		as published by the Free Software Foundation; either version
11  *		2 of the License, or (at your option) any later version.
12  *
13  * Tue Jun 26 14:36:48 MEST 2001 Herbert "herp" Rosmanith
14  *                               added netlink_proto_exit
15  * Tue Jan 22 18:32:44 BRST 2002 Arnaldo C. de Melo <acme@conectiva.com.br>
16  * 				 use nlk_sk, as sk->protinfo is on a diet 8)
17  * Fri Jul 22 19:51:12 MEST 2005 Harald Welte <laforge@gnumonks.org>
18  * 				 - inc module use count of module that owns
19  * 				   the kernel socket in case userspace opens
20  * 				   socket of same protocol
21  * 				 - remove all module support, since netlink is
22  * 				   mandatory if CONFIG_NET=y these days
23  */
24 
25 #include <linux/module.h>
26 
27 #include <linux/capability.h>
28 #include <linux/kernel.h>
29 #include <linux/init.h>
30 #include <linux/signal.h>
31 #include <linux/sched.h>
32 #include <linux/errno.h>
33 #include <linux/string.h>
34 #include <linux/stat.h>
35 #include <linux/socket.h>
36 #include <linux/un.h>
37 #include <linux/fcntl.h>
38 #include <linux/termios.h>
39 #include <linux/sockios.h>
40 #include <linux/net.h>
41 #include <linux/fs.h>
42 #include <linux/slab.h>
43 #include <asm/uaccess.h>
44 #include <linux/skbuff.h>
45 #include <linux/netdevice.h>
46 #include <linux/rtnetlink.h>
47 #include <linux/proc_fs.h>
48 #include <linux/seq_file.h>
49 #include <linux/notifier.h>
50 #include <linux/security.h>
51 #include <linux/jhash.h>
52 #include <linux/jiffies.h>
53 #include <linux/random.h>
54 #include <linux/bitops.h>
55 #include <linux/mm.h>
56 #include <linux/types.h>
57 #include <linux/audit.h>
58 #include <linux/mutex.h>
59 #include <linux/vmalloc.h>
60 #include <linux/if_arp.h>
61 #include <linux/rhashtable.h>
62 #include <asm/cacheflush.h>
63 #include <linux/hash.h>
64 #include <linux/genetlink.h>
65 
66 #include <net/net_namespace.h>
67 #include <net/sock.h>
68 #include <net/scm.h>
69 #include <net/netlink.h>
70 
71 #include "af_netlink.h"
72 
73 struct listeners {
74 	struct rcu_head		rcu;
75 	unsigned long		masks[0];
76 };
77 
78 /* state bits */
79 #define NETLINK_S_CONGESTED		0x0
80 
81 /* flags */
82 #define NETLINK_F_KERNEL_SOCKET		0x1
83 #define NETLINK_F_RECV_PKTINFO		0x2
84 #define NETLINK_F_BROADCAST_SEND_ERROR	0x4
85 #define NETLINK_F_RECV_NO_ENOBUFS	0x8
86 #define NETLINK_F_LISTEN_ALL_NSID	0x10
87 #define NETLINK_F_CAP_ACK		0x20
88 
89 static inline int netlink_is_kernel(struct sock *sk)
90 {
91 	return nlk_sk(sk)->flags & NETLINK_F_KERNEL_SOCKET;
92 }
93 
94 struct netlink_table *nl_table __read_mostly;
95 EXPORT_SYMBOL_GPL(nl_table);
96 
97 static DECLARE_WAIT_QUEUE_HEAD(nl_table_wait);
98 
99 static int netlink_dump(struct sock *sk);
100 static void netlink_skb_destructor(struct sk_buff *skb);
101 
102 /* nl_table locking explained:
103  * Lookup and traversal are protected with an RCU read-side lock. Insertion
104  * and removal are protected with per bucket lock while using RCU list
105  * modification primitives and may run in parallel to RCU protected lookups.
106  * Destruction of the Netlink socket may only occur *after* nl_table_lock has
107  * been acquired * either during or after the socket has been removed from
108  * the list and after an RCU grace period.
109  */
110 DEFINE_RWLOCK(nl_table_lock);
111 EXPORT_SYMBOL_GPL(nl_table_lock);
112 static atomic_t nl_table_users = ATOMIC_INIT(0);
113 
114 #define nl_deref_protected(X) rcu_dereference_protected(X, lockdep_is_held(&nl_table_lock));
115 
116 static ATOMIC_NOTIFIER_HEAD(netlink_chain);
117 
118 static DEFINE_SPINLOCK(netlink_tap_lock);
119 static struct list_head netlink_tap_all __read_mostly;
120 
121 static const struct rhashtable_params netlink_rhashtable_params;
122 
123 static inline u32 netlink_group_mask(u32 group)
124 {
125 	return group ? 1 << (group - 1) : 0;
126 }
127 
128 static struct sk_buff *netlink_to_full_skb(const struct sk_buff *skb,
129 					   gfp_t gfp_mask)
130 {
131 	unsigned int len = skb_end_offset(skb);
132 	struct sk_buff *new;
133 
134 	new = alloc_skb(len, gfp_mask);
135 	if (new == NULL)
136 		return NULL;
137 
138 	NETLINK_CB(new).portid = NETLINK_CB(skb).portid;
139 	NETLINK_CB(new).dst_group = NETLINK_CB(skb).dst_group;
140 	NETLINK_CB(new).creds = NETLINK_CB(skb).creds;
141 
142 	memcpy(skb_put(new, len), skb->data, len);
143 	return new;
144 }
145 
146 int netlink_add_tap(struct netlink_tap *nt)
147 {
148 	if (unlikely(nt->dev->type != ARPHRD_NETLINK))
149 		return -EINVAL;
150 
151 	spin_lock(&netlink_tap_lock);
152 	list_add_rcu(&nt->list, &netlink_tap_all);
153 	spin_unlock(&netlink_tap_lock);
154 
155 	__module_get(nt->module);
156 
157 	return 0;
158 }
159 EXPORT_SYMBOL_GPL(netlink_add_tap);
160 
161 static int __netlink_remove_tap(struct netlink_tap *nt)
162 {
163 	bool found = false;
164 	struct netlink_tap *tmp;
165 
166 	spin_lock(&netlink_tap_lock);
167 
168 	list_for_each_entry(tmp, &netlink_tap_all, list) {
169 		if (nt == tmp) {
170 			list_del_rcu(&nt->list);
171 			found = true;
172 			goto out;
173 		}
174 	}
175 
176 	pr_warn("__netlink_remove_tap: %p not found\n", nt);
177 out:
178 	spin_unlock(&netlink_tap_lock);
179 
180 	if (found)
181 		module_put(nt->module);
182 
183 	return found ? 0 : -ENODEV;
184 }
185 
186 int netlink_remove_tap(struct netlink_tap *nt)
187 {
188 	int ret;
189 
190 	ret = __netlink_remove_tap(nt);
191 	synchronize_net();
192 
193 	return ret;
194 }
195 EXPORT_SYMBOL_GPL(netlink_remove_tap);
196 
197 static bool netlink_filter_tap(const struct sk_buff *skb)
198 {
199 	struct sock *sk = skb->sk;
200 
201 	/* We take the more conservative approach and
202 	 * whitelist socket protocols that may pass.
203 	 */
204 	switch (sk->sk_protocol) {
205 	case NETLINK_ROUTE:
206 	case NETLINK_USERSOCK:
207 	case NETLINK_SOCK_DIAG:
208 	case NETLINK_NFLOG:
209 	case NETLINK_XFRM:
210 	case NETLINK_FIB_LOOKUP:
211 	case NETLINK_NETFILTER:
212 	case NETLINK_GENERIC:
213 		return true;
214 	}
215 
216 	return false;
217 }
218 
219 static int __netlink_deliver_tap_skb(struct sk_buff *skb,
220 				     struct net_device *dev)
221 {
222 	struct sk_buff *nskb;
223 	struct sock *sk = skb->sk;
224 	int ret = -ENOMEM;
225 
226 	dev_hold(dev);
227 
228 	if (is_vmalloc_addr(skb->head))
229 		nskb = netlink_to_full_skb(skb, GFP_ATOMIC);
230 	else
231 		nskb = skb_clone(skb, GFP_ATOMIC);
232 	if (nskb) {
233 		nskb->dev = dev;
234 		nskb->protocol = htons((u16) sk->sk_protocol);
235 		nskb->pkt_type = netlink_is_kernel(sk) ?
236 				 PACKET_KERNEL : PACKET_USER;
237 		skb_reset_network_header(nskb);
238 		ret = dev_queue_xmit(nskb);
239 		if (unlikely(ret > 0))
240 			ret = net_xmit_errno(ret);
241 	}
242 
243 	dev_put(dev);
244 	return ret;
245 }
246 
247 static void __netlink_deliver_tap(struct sk_buff *skb)
248 {
249 	int ret;
250 	struct netlink_tap *tmp;
251 
252 	if (!netlink_filter_tap(skb))
253 		return;
254 
255 	list_for_each_entry_rcu(tmp, &netlink_tap_all, list) {
256 		ret = __netlink_deliver_tap_skb(skb, tmp->dev);
257 		if (unlikely(ret))
258 			break;
259 	}
260 }
261 
262 static void netlink_deliver_tap(struct sk_buff *skb)
263 {
264 	rcu_read_lock();
265 
266 	if (unlikely(!list_empty(&netlink_tap_all)))
267 		__netlink_deliver_tap(skb);
268 
269 	rcu_read_unlock();
270 }
271 
272 static void netlink_deliver_tap_kernel(struct sock *dst, struct sock *src,
273 				       struct sk_buff *skb)
274 {
275 	if (!(netlink_is_kernel(dst) && netlink_is_kernel(src)))
276 		netlink_deliver_tap(skb);
277 }
278 
279 static void netlink_overrun(struct sock *sk)
280 {
281 	struct netlink_sock *nlk = nlk_sk(sk);
282 
283 	if (!(nlk->flags & NETLINK_F_RECV_NO_ENOBUFS)) {
284 		if (!test_and_set_bit(NETLINK_S_CONGESTED,
285 				      &nlk_sk(sk)->state)) {
286 			sk->sk_err = ENOBUFS;
287 			sk->sk_error_report(sk);
288 		}
289 	}
290 	atomic_inc(&sk->sk_drops);
291 }
292 
293 static void netlink_rcv_wake(struct sock *sk)
294 {
295 	struct netlink_sock *nlk = nlk_sk(sk);
296 
297 	if (skb_queue_empty(&sk->sk_receive_queue))
298 		clear_bit(NETLINK_S_CONGESTED, &nlk->state);
299 	if (!test_bit(NETLINK_S_CONGESTED, &nlk->state))
300 		wake_up_interruptible(&nlk->wait);
301 }
302 
303 static void netlink_skb_destructor(struct sk_buff *skb)
304 {
305 	if (is_vmalloc_addr(skb->head)) {
306 		if (!skb->cloned ||
307 		    !atomic_dec_return(&(skb_shinfo(skb)->dataref)))
308 			vfree(skb->head);
309 
310 		skb->head = NULL;
311 	}
312 	if (skb->sk != NULL)
313 		sock_rfree(skb);
314 }
315 
316 static void netlink_skb_set_owner_r(struct sk_buff *skb, struct sock *sk)
317 {
318 	WARN_ON(skb->sk != NULL);
319 	skb->sk = sk;
320 	skb->destructor = netlink_skb_destructor;
321 	atomic_add(skb->truesize, &sk->sk_rmem_alloc);
322 	sk_mem_charge(sk, skb->truesize);
323 }
324 
325 static void netlink_sock_destruct(struct sock *sk)
326 {
327 	struct netlink_sock *nlk = nlk_sk(sk);
328 
329 	if (nlk->cb_running) {
330 		if (nlk->cb.done)
331 			nlk->cb.done(&nlk->cb);
332 
333 		module_put(nlk->cb.module);
334 		kfree_skb(nlk->cb.skb);
335 	}
336 
337 	skb_queue_purge(&sk->sk_receive_queue);
338 
339 	if (!sock_flag(sk, SOCK_DEAD)) {
340 		printk(KERN_ERR "Freeing alive netlink socket %p\n", sk);
341 		return;
342 	}
343 
344 	WARN_ON(atomic_read(&sk->sk_rmem_alloc));
345 	WARN_ON(atomic_read(&sk->sk_wmem_alloc));
346 	WARN_ON(nlk_sk(sk)->groups);
347 }
348 
349 /* This lock without WQ_FLAG_EXCLUSIVE is good on UP and it is _very_ bad on
350  * SMP. Look, when several writers sleep and reader wakes them up, all but one
351  * immediately hit write lock and grab all the cpus. Exclusive sleep solves
352  * this, _but_ remember, it adds useless work on UP machines.
353  */
354 
355 void netlink_table_grab(void)
356 	__acquires(nl_table_lock)
357 {
358 	might_sleep();
359 
360 	write_lock_irq(&nl_table_lock);
361 
362 	if (atomic_read(&nl_table_users)) {
363 		DECLARE_WAITQUEUE(wait, current);
364 
365 		add_wait_queue_exclusive(&nl_table_wait, &wait);
366 		for (;;) {
367 			set_current_state(TASK_UNINTERRUPTIBLE);
368 			if (atomic_read(&nl_table_users) == 0)
369 				break;
370 			write_unlock_irq(&nl_table_lock);
371 			schedule();
372 			write_lock_irq(&nl_table_lock);
373 		}
374 
375 		__set_current_state(TASK_RUNNING);
376 		remove_wait_queue(&nl_table_wait, &wait);
377 	}
378 }
379 
380 void netlink_table_ungrab(void)
381 	__releases(nl_table_lock)
382 {
383 	write_unlock_irq(&nl_table_lock);
384 	wake_up(&nl_table_wait);
385 }
386 
387 static inline void
388 netlink_lock_table(void)
389 {
390 	/* read_lock() synchronizes us to netlink_table_grab */
391 
392 	read_lock(&nl_table_lock);
393 	atomic_inc(&nl_table_users);
394 	read_unlock(&nl_table_lock);
395 }
396 
397 static inline void
398 netlink_unlock_table(void)
399 {
400 	if (atomic_dec_and_test(&nl_table_users))
401 		wake_up(&nl_table_wait);
402 }
403 
404 struct netlink_compare_arg
405 {
406 	possible_net_t pnet;
407 	u32 portid;
408 };
409 
410 /* Doing sizeof directly may yield 4 extra bytes on 64-bit. */
411 #define netlink_compare_arg_len \
412 	(offsetof(struct netlink_compare_arg, portid) + sizeof(u32))
413 
414 static inline int netlink_compare(struct rhashtable_compare_arg *arg,
415 				  const void *ptr)
416 {
417 	const struct netlink_compare_arg *x = arg->key;
418 	const struct netlink_sock *nlk = ptr;
419 
420 	return nlk->portid != x->portid ||
421 	       !net_eq(sock_net(&nlk->sk), read_pnet(&x->pnet));
422 }
423 
424 static void netlink_compare_arg_init(struct netlink_compare_arg *arg,
425 				     struct net *net, u32 portid)
426 {
427 	memset(arg, 0, sizeof(*arg));
428 	write_pnet(&arg->pnet, net);
429 	arg->portid = portid;
430 }
431 
432 static struct sock *__netlink_lookup(struct netlink_table *table, u32 portid,
433 				     struct net *net)
434 {
435 	struct netlink_compare_arg arg;
436 
437 	netlink_compare_arg_init(&arg, net, portid);
438 	return rhashtable_lookup_fast(&table->hash, &arg,
439 				      netlink_rhashtable_params);
440 }
441 
442 static int __netlink_insert(struct netlink_table *table, struct sock *sk)
443 {
444 	struct netlink_compare_arg arg;
445 
446 	netlink_compare_arg_init(&arg, sock_net(sk), nlk_sk(sk)->portid);
447 	return rhashtable_lookup_insert_key(&table->hash, &arg,
448 					    &nlk_sk(sk)->node,
449 					    netlink_rhashtable_params);
450 }
451 
452 static struct sock *netlink_lookup(struct net *net, int protocol, u32 portid)
453 {
454 	struct netlink_table *table = &nl_table[protocol];
455 	struct sock *sk;
456 
457 	rcu_read_lock();
458 	sk = __netlink_lookup(table, portid, net);
459 	if (sk)
460 		sock_hold(sk);
461 	rcu_read_unlock();
462 
463 	return sk;
464 }
465 
466 static const struct proto_ops netlink_ops;
467 
468 static void
469 netlink_update_listeners(struct sock *sk)
470 {
471 	struct netlink_table *tbl = &nl_table[sk->sk_protocol];
472 	unsigned long mask;
473 	unsigned int i;
474 	struct listeners *listeners;
475 
476 	listeners = nl_deref_protected(tbl->listeners);
477 	if (!listeners)
478 		return;
479 
480 	for (i = 0; i < NLGRPLONGS(tbl->groups); i++) {
481 		mask = 0;
482 		sk_for_each_bound(sk, &tbl->mc_list) {
483 			if (i < NLGRPLONGS(nlk_sk(sk)->ngroups))
484 				mask |= nlk_sk(sk)->groups[i];
485 		}
486 		listeners->masks[i] = mask;
487 	}
488 	/* this function is only called with the netlink table "grabbed", which
489 	 * makes sure updates are visible before bind or setsockopt return. */
490 }
491 
492 static int netlink_insert(struct sock *sk, u32 portid)
493 {
494 	struct netlink_table *table = &nl_table[sk->sk_protocol];
495 	int err;
496 
497 	lock_sock(sk);
498 
499 	err = nlk_sk(sk)->portid == portid ? 0 : -EBUSY;
500 	if (nlk_sk(sk)->bound)
501 		goto err;
502 
503 	err = -ENOMEM;
504 	if (BITS_PER_LONG > 32 &&
505 	    unlikely(atomic_read(&table->hash.nelems) >= UINT_MAX))
506 		goto err;
507 
508 	nlk_sk(sk)->portid = portid;
509 	sock_hold(sk);
510 
511 	err = __netlink_insert(table, sk);
512 	if (err) {
513 		/* In case the hashtable backend returns with -EBUSY
514 		 * from here, it must not escape to the caller.
515 		 */
516 		if (unlikely(err == -EBUSY))
517 			err = -EOVERFLOW;
518 		if (err == -EEXIST)
519 			err = -EADDRINUSE;
520 		sock_put(sk);
521 		goto err;
522 	}
523 
524 	/* We need to ensure that the socket is hashed and visible. */
525 	smp_wmb();
526 	nlk_sk(sk)->bound = portid;
527 
528 err:
529 	release_sock(sk);
530 	return err;
531 }
532 
533 static void netlink_remove(struct sock *sk)
534 {
535 	struct netlink_table *table;
536 
537 	table = &nl_table[sk->sk_protocol];
538 	if (!rhashtable_remove_fast(&table->hash, &nlk_sk(sk)->node,
539 				    netlink_rhashtable_params)) {
540 		WARN_ON(atomic_read(&sk->sk_refcnt) == 1);
541 		__sock_put(sk);
542 	}
543 
544 	netlink_table_grab();
545 	if (nlk_sk(sk)->subscriptions) {
546 		__sk_del_bind_node(sk);
547 		netlink_update_listeners(sk);
548 	}
549 	if (sk->sk_protocol == NETLINK_GENERIC)
550 		atomic_inc(&genl_sk_destructing_cnt);
551 	netlink_table_ungrab();
552 }
553 
554 static struct proto netlink_proto = {
555 	.name	  = "NETLINK",
556 	.owner	  = THIS_MODULE,
557 	.obj_size = sizeof(struct netlink_sock),
558 };
559 
560 static int __netlink_create(struct net *net, struct socket *sock,
561 			    struct mutex *cb_mutex, int protocol,
562 			    int kern)
563 {
564 	struct sock *sk;
565 	struct netlink_sock *nlk;
566 
567 	sock->ops = &netlink_ops;
568 
569 	sk = sk_alloc(net, PF_NETLINK, GFP_KERNEL, &netlink_proto, kern);
570 	if (!sk)
571 		return -ENOMEM;
572 
573 	sock_init_data(sock, sk);
574 
575 	nlk = nlk_sk(sk);
576 	if (cb_mutex) {
577 		nlk->cb_mutex = cb_mutex;
578 	} else {
579 		nlk->cb_mutex = &nlk->cb_def_mutex;
580 		mutex_init(nlk->cb_mutex);
581 	}
582 	init_waitqueue_head(&nlk->wait);
583 
584 	sk->sk_destruct = netlink_sock_destruct;
585 	sk->sk_protocol = protocol;
586 	return 0;
587 }
588 
589 static int netlink_create(struct net *net, struct socket *sock, int protocol,
590 			  int kern)
591 {
592 	struct module *module = NULL;
593 	struct mutex *cb_mutex;
594 	struct netlink_sock *nlk;
595 	int (*bind)(struct net *net, int group);
596 	void (*unbind)(struct net *net, int group);
597 	int err = 0;
598 
599 	sock->state = SS_UNCONNECTED;
600 
601 	if (sock->type != SOCK_RAW && sock->type != SOCK_DGRAM)
602 		return -ESOCKTNOSUPPORT;
603 
604 	if (protocol < 0 || protocol >= MAX_LINKS)
605 		return -EPROTONOSUPPORT;
606 
607 	netlink_lock_table();
608 #ifdef CONFIG_MODULES
609 	if (!nl_table[protocol].registered) {
610 		netlink_unlock_table();
611 		request_module("net-pf-%d-proto-%d", PF_NETLINK, protocol);
612 		netlink_lock_table();
613 	}
614 #endif
615 	if (nl_table[protocol].registered &&
616 	    try_module_get(nl_table[protocol].module))
617 		module = nl_table[protocol].module;
618 	else
619 		err = -EPROTONOSUPPORT;
620 	cb_mutex = nl_table[protocol].cb_mutex;
621 	bind = nl_table[protocol].bind;
622 	unbind = nl_table[protocol].unbind;
623 	netlink_unlock_table();
624 
625 	if (err < 0)
626 		goto out;
627 
628 	err = __netlink_create(net, sock, cb_mutex, protocol, kern);
629 	if (err < 0)
630 		goto out_module;
631 
632 	local_bh_disable();
633 	sock_prot_inuse_add(net, &netlink_proto, 1);
634 	local_bh_enable();
635 
636 	nlk = nlk_sk(sock->sk);
637 	nlk->module = module;
638 	nlk->netlink_bind = bind;
639 	nlk->netlink_unbind = unbind;
640 out:
641 	return err;
642 
643 out_module:
644 	module_put(module);
645 	goto out;
646 }
647 
648 static void deferred_put_nlk_sk(struct rcu_head *head)
649 {
650 	struct netlink_sock *nlk = container_of(head, struct netlink_sock, rcu);
651 
652 	sock_put(&nlk->sk);
653 }
654 
655 static int netlink_release(struct socket *sock)
656 {
657 	struct sock *sk = sock->sk;
658 	struct netlink_sock *nlk;
659 
660 	if (!sk)
661 		return 0;
662 
663 	netlink_remove(sk);
664 	sock_orphan(sk);
665 	nlk = nlk_sk(sk);
666 
667 	/*
668 	 * OK. Socket is unlinked, any packets that arrive now
669 	 * will be purged.
670 	 */
671 
672 	/* must not acquire netlink_table_lock in any way again before unbind
673 	 * and notifying genetlink is done as otherwise it might deadlock
674 	 */
675 	if (nlk->netlink_unbind) {
676 		int i;
677 
678 		for (i = 0; i < nlk->ngroups; i++)
679 			if (test_bit(i, nlk->groups))
680 				nlk->netlink_unbind(sock_net(sk), i + 1);
681 	}
682 	if (sk->sk_protocol == NETLINK_GENERIC &&
683 	    atomic_dec_return(&genl_sk_destructing_cnt) == 0)
684 		wake_up(&genl_sk_destructing_waitq);
685 
686 	sock->sk = NULL;
687 	wake_up_interruptible_all(&nlk->wait);
688 
689 	skb_queue_purge(&sk->sk_write_queue);
690 
691 	if (nlk->portid && nlk->bound) {
692 		struct netlink_notify n = {
693 						.net = sock_net(sk),
694 						.protocol = sk->sk_protocol,
695 						.portid = nlk->portid,
696 					  };
697 		atomic_notifier_call_chain(&netlink_chain,
698 				NETLINK_URELEASE, &n);
699 	}
700 
701 	module_put(nlk->module);
702 
703 	if (netlink_is_kernel(sk)) {
704 		netlink_table_grab();
705 		BUG_ON(nl_table[sk->sk_protocol].registered == 0);
706 		if (--nl_table[sk->sk_protocol].registered == 0) {
707 			struct listeners *old;
708 
709 			old = nl_deref_protected(nl_table[sk->sk_protocol].listeners);
710 			RCU_INIT_POINTER(nl_table[sk->sk_protocol].listeners, NULL);
711 			kfree_rcu(old, rcu);
712 			nl_table[sk->sk_protocol].module = NULL;
713 			nl_table[sk->sk_protocol].bind = NULL;
714 			nl_table[sk->sk_protocol].unbind = NULL;
715 			nl_table[sk->sk_protocol].flags = 0;
716 			nl_table[sk->sk_protocol].registered = 0;
717 		}
718 		netlink_table_ungrab();
719 	}
720 
721 	kfree(nlk->groups);
722 	nlk->groups = NULL;
723 
724 	local_bh_disable();
725 	sock_prot_inuse_add(sock_net(sk), &netlink_proto, -1);
726 	local_bh_enable();
727 	call_rcu(&nlk->rcu, deferred_put_nlk_sk);
728 	return 0;
729 }
730 
731 static int netlink_autobind(struct socket *sock)
732 {
733 	struct sock *sk = sock->sk;
734 	struct net *net = sock_net(sk);
735 	struct netlink_table *table = &nl_table[sk->sk_protocol];
736 	s32 portid = task_tgid_vnr(current);
737 	int err;
738 	s32 rover = -4096;
739 	bool ok;
740 
741 retry:
742 	cond_resched();
743 	rcu_read_lock();
744 	ok = !__netlink_lookup(table, portid, net);
745 	rcu_read_unlock();
746 	if (!ok) {
747 		/* Bind collision, search negative portid values. */
748 		if (rover == -4096)
749 			/* rover will be in range [S32_MIN, -4097] */
750 			rover = S32_MIN + prandom_u32_max(-4096 - S32_MIN);
751 		else if (rover >= -4096)
752 			rover = -4097;
753 		portid = rover--;
754 		goto retry;
755 	}
756 
757 	err = netlink_insert(sk, portid);
758 	if (err == -EADDRINUSE)
759 		goto retry;
760 
761 	/* If 2 threads race to autobind, that is fine.  */
762 	if (err == -EBUSY)
763 		err = 0;
764 
765 	return err;
766 }
767 
768 /**
769  * __netlink_ns_capable - General netlink message capability test
770  * @nsp: NETLINK_CB of the socket buffer holding a netlink command from userspace.
771  * @user_ns: The user namespace of the capability to use
772  * @cap: The capability to use
773  *
774  * Test to see if the opener of the socket we received the message
775  * from had when the netlink socket was created and the sender of the
776  * message has has the capability @cap in the user namespace @user_ns.
777  */
778 bool __netlink_ns_capable(const struct netlink_skb_parms *nsp,
779 			struct user_namespace *user_ns, int cap)
780 {
781 	return ((nsp->flags & NETLINK_SKB_DST) ||
782 		file_ns_capable(nsp->sk->sk_socket->file, user_ns, cap)) &&
783 		ns_capable(user_ns, cap);
784 }
785 EXPORT_SYMBOL(__netlink_ns_capable);
786 
787 /**
788  * netlink_ns_capable - General netlink message capability test
789  * @skb: socket buffer holding a netlink command from userspace
790  * @user_ns: The user namespace of the capability to use
791  * @cap: The capability to use
792  *
793  * Test to see if the opener of the socket we received the message
794  * from had when the netlink socket was created and the sender of the
795  * message has has the capability @cap in the user namespace @user_ns.
796  */
797 bool netlink_ns_capable(const struct sk_buff *skb,
798 			struct user_namespace *user_ns, int cap)
799 {
800 	return __netlink_ns_capable(&NETLINK_CB(skb), user_ns, cap);
801 }
802 EXPORT_SYMBOL(netlink_ns_capable);
803 
804 /**
805  * netlink_capable - Netlink global message capability test
806  * @skb: socket buffer holding a netlink command from userspace
807  * @cap: The capability to use
808  *
809  * Test to see if the opener of the socket we received the message
810  * from had when the netlink socket was created and the sender of the
811  * message has has the capability @cap in all user namespaces.
812  */
813 bool netlink_capable(const struct sk_buff *skb, int cap)
814 {
815 	return netlink_ns_capable(skb, &init_user_ns, cap);
816 }
817 EXPORT_SYMBOL(netlink_capable);
818 
819 /**
820  * netlink_net_capable - Netlink network namespace message capability test
821  * @skb: socket buffer holding a netlink command from userspace
822  * @cap: The capability to use
823  *
824  * Test to see if the opener of the socket we received the message
825  * from had when the netlink socket was created and the sender of the
826  * message has has the capability @cap over the network namespace of
827  * the socket we received the message from.
828  */
829 bool netlink_net_capable(const struct sk_buff *skb, int cap)
830 {
831 	return netlink_ns_capable(skb, sock_net(skb->sk)->user_ns, cap);
832 }
833 EXPORT_SYMBOL(netlink_net_capable);
834 
835 static inline int netlink_allowed(const struct socket *sock, unsigned int flag)
836 {
837 	return (nl_table[sock->sk->sk_protocol].flags & flag) ||
838 		ns_capable(sock_net(sock->sk)->user_ns, CAP_NET_ADMIN);
839 }
840 
841 static void
842 netlink_update_subscriptions(struct sock *sk, unsigned int subscriptions)
843 {
844 	struct netlink_sock *nlk = nlk_sk(sk);
845 
846 	if (nlk->subscriptions && !subscriptions)
847 		__sk_del_bind_node(sk);
848 	else if (!nlk->subscriptions && subscriptions)
849 		sk_add_bind_node(sk, &nl_table[sk->sk_protocol].mc_list);
850 	nlk->subscriptions = subscriptions;
851 }
852 
853 static int netlink_realloc_groups(struct sock *sk)
854 {
855 	struct netlink_sock *nlk = nlk_sk(sk);
856 	unsigned int groups;
857 	unsigned long *new_groups;
858 	int err = 0;
859 
860 	netlink_table_grab();
861 
862 	groups = nl_table[sk->sk_protocol].groups;
863 	if (!nl_table[sk->sk_protocol].registered) {
864 		err = -ENOENT;
865 		goto out_unlock;
866 	}
867 
868 	if (nlk->ngroups >= groups)
869 		goto out_unlock;
870 
871 	new_groups = krealloc(nlk->groups, NLGRPSZ(groups), GFP_ATOMIC);
872 	if (new_groups == NULL) {
873 		err = -ENOMEM;
874 		goto out_unlock;
875 	}
876 	memset((char *)new_groups + NLGRPSZ(nlk->ngroups), 0,
877 	       NLGRPSZ(groups) - NLGRPSZ(nlk->ngroups));
878 
879 	nlk->groups = new_groups;
880 	nlk->ngroups = groups;
881  out_unlock:
882 	netlink_table_ungrab();
883 	return err;
884 }
885 
886 static void netlink_undo_bind(int group, long unsigned int groups,
887 			      struct sock *sk)
888 {
889 	struct netlink_sock *nlk = nlk_sk(sk);
890 	int undo;
891 
892 	if (!nlk->netlink_unbind)
893 		return;
894 
895 	for (undo = 0; undo < group; undo++)
896 		if (test_bit(undo, &groups))
897 			nlk->netlink_unbind(sock_net(sk), undo + 1);
898 }
899 
900 static int netlink_bind(struct socket *sock, struct sockaddr *addr,
901 			int addr_len)
902 {
903 	struct sock *sk = sock->sk;
904 	struct net *net = sock_net(sk);
905 	struct netlink_sock *nlk = nlk_sk(sk);
906 	struct sockaddr_nl *nladdr = (struct sockaddr_nl *)addr;
907 	int err;
908 	long unsigned int groups = nladdr->nl_groups;
909 	bool bound;
910 
911 	if (addr_len < sizeof(struct sockaddr_nl))
912 		return -EINVAL;
913 
914 	if (nladdr->nl_family != AF_NETLINK)
915 		return -EINVAL;
916 
917 	/* Only superuser is allowed to listen multicasts */
918 	if (groups) {
919 		if (!netlink_allowed(sock, NL_CFG_F_NONROOT_RECV))
920 			return -EPERM;
921 		err = netlink_realloc_groups(sk);
922 		if (err)
923 			return err;
924 	}
925 
926 	bound = nlk->bound;
927 	if (bound) {
928 		/* Ensure nlk->portid is up-to-date. */
929 		smp_rmb();
930 
931 		if (nladdr->nl_pid != nlk->portid)
932 			return -EINVAL;
933 	}
934 
935 	if (nlk->netlink_bind && groups) {
936 		int group;
937 
938 		for (group = 0; group < nlk->ngroups; group++) {
939 			if (!test_bit(group, &groups))
940 				continue;
941 			err = nlk->netlink_bind(net, group + 1);
942 			if (!err)
943 				continue;
944 			netlink_undo_bind(group, groups, sk);
945 			return err;
946 		}
947 	}
948 
949 	/* No need for barriers here as we return to user-space without
950 	 * using any of the bound attributes.
951 	 */
952 	if (!bound) {
953 		err = nladdr->nl_pid ?
954 			netlink_insert(sk, nladdr->nl_pid) :
955 			netlink_autobind(sock);
956 		if (err) {
957 			netlink_undo_bind(nlk->ngroups, groups, sk);
958 			return err;
959 		}
960 	}
961 
962 	if (!groups && (nlk->groups == NULL || !(u32)nlk->groups[0]))
963 		return 0;
964 
965 	netlink_table_grab();
966 	netlink_update_subscriptions(sk, nlk->subscriptions +
967 					 hweight32(groups) -
968 					 hweight32(nlk->groups[0]));
969 	nlk->groups[0] = (nlk->groups[0] & ~0xffffffffUL) | groups;
970 	netlink_update_listeners(sk);
971 	netlink_table_ungrab();
972 
973 	return 0;
974 }
975 
976 static int netlink_connect(struct socket *sock, struct sockaddr *addr,
977 			   int alen, int flags)
978 {
979 	int err = 0;
980 	struct sock *sk = sock->sk;
981 	struct netlink_sock *nlk = nlk_sk(sk);
982 	struct sockaddr_nl *nladdr = (struct sockaddr_nl *)addr;
983 
984 	if (alen < sizeof(addr->sa_family))
985 		return -EINVAL;
986 
987 	if (addr->sa_family == AF_UNSPEC) {
988 		sk->sk_state	= NETLINK_UNCONNECTED;
989 		nlk->dst_portid	= 0;
990 		nlk->dst_group  = 0;
991 		return 0;
992 	}
993 	if (addr->sa_family != AF_NETLINK)
994 		return -EINVAL;
995 
996 	if ((nladdr->nl_groups || nladdr->nl_pid) &&
997 	    !netlink_allowed(sock, NL_CFG_F_NONROOT_SEND))
998 		return -EPERM;
999 
1000 	/* No need for barriers here as we return to user-space without
1001 	 * using any of the bound attributes.
1002 	 */
1003 	if (!nlk->bound)
1004 		err = netlink_autobind(sock);
1005 
1006 	if (err == 0) {
1007 		sk->sk_state	= NETLINK_CONNECTED;
1008 		nlk->dst_portid = nladdr->nl_pid;
1009 		nlk->dst_group  = ffs(nladdr->nl_groups);
1010 	}
1011 
1012 	return err;
1013 }
1014 
1015 static int netlink_getname(struct socket *sock, struct sockaddr *addr,
1016 			   int *addr_len, int peer)
1017 {
1018 	struct sock *sk = sock->sk;
1019 	struct netlink_sock *nlk = nlk_sk(sk);
1020 	DECLARE_SOCKADDR(struct sockaddr_nl *, nladdr, addr);
1021 
1022 	nladdr->nl_family = AF_NETLINK;
1023 	nladdr->nl_pad = 0;
1024 	*addr_len = sizeof(*nladdr);
1025 
1026 	if (peer) {
1027 		nladdr->nl_pid = nlk->dst_portid;
1028 		nladdr->nl_groups = netlink_group_mask(nlk->dst_group);
1029 	} else {
1030 		nladdr->nl_pid = nlk->portid;
1031 		nladdr->nl_groups = nlk->groups ? nlk->groups[0] : 0;
1032 	}
1033 	return 0;
1034 }
1035 
1036 static int netlink_ioctl(struct socket *sock, unsigned int cmd,
1037 			 unsigned long arg)
1038 {
1039 	/* try to hand this ioctl down to the NIC drivers.
1040 	 */
1041 	return -ENOIOCTLCMD;
1042 }
1043 
1044 static struct sock *netlink_getsockbyportid(struct sock *ssk, u32 portid)
1045 {
1046 	struct sock *sock;
1047 	struct netlink_sock *nlk;
1048 
1049 	sock = netlink_lookup(sock_net(ssk), ssk->sk_protocol, portid);
1050 	if (!sock)
1051 		return ERR_PTR(-ECONNREFUSED);
1052 
1053 	/* Don't bother queuing skb if kernel socket has no input function */
1054 	nlk = nlk_sk(sock);
1055 	if (sock->sk_state == NETLINK_CONNECTED &&
1056 	    nlk->dst_portid != nlk_sk(ssk)->portid) {
1057 		sock_put(sock);
1058 		return ERR_PTR(-ECONNREFUSED);
1059 	}
1060 	return sock;
1061 }
1062 
1063 struct sock *netlink_getsockbyfilp(struct file *filp)
1064 {
1065 	struct inode *inode = file_inode(filp);
1066 	struct sock *sock;
1067 
1068 	if (!S_ISSOCK(inode->i_mode))
1069 		return ERR_PTR(-ENOTSOCK);
1070 
1071 	sock = SOCKET_I(inode)->sk;
1072 	if (sock->sk_family != AF_NETLINK)
1073 		return ERR_PTR(-EINVAL);
1074 
1075 	sock_hold(sock);
1076 	return sock;
1077 }
1078 
1079 static struct sk_buff *netlink_alloc_large_skb(unsigned int size,
1080 					       int broadcast)
1081 {
1082 	struct sk_buff *skb;
1083 	void *data;
1084 
1085 	if (size <= NLMSG_GOODSIZE || broadcast)
1086 		return alloc_skb(size, GFP_KERNEL);
1087 
1088 	size = SKB_DATA_ALIGN(size) +
1089 	       SKB_DATA_ALIGN(sizeof(struct skb_shared_info));
1090 
1091 	data = vmalloc(size);
1092 	if (data == NULL)
1093 		return NULL;
1094 
1095 	skb = __build_skb(data, size);
1096 	if (skb == NULL)
1097 		vfree(data);
1098 	else
1099 		skb->destructor = netlink_skb_destructor;
1100 
1101 	return skb;
1102 }
1103 
1104 /*
1105  * Attach a skb to a netlink socket.
1106  * The caller must hold a reference to the destination socket. On error, the
1107  * reference is dropped. The skb is not send to the destination, just all
1108  * all error checks are performed and memory in the queue is reserved.
1109  * Return values:
1110  * < 0: error. skb freed, reference to sock dropped.
1111  * 0: continue
1112  * 1: repeat lookup - reference dropped while waiting for socket memory.
1113  */
1114 int netlink_attachskb(struct sock *sk, struct sk_buff *skb,
1115 		      long *timeo, struct sock *ssk)
1116 {
1117 	struct netlink_sock *nlk;
1118 
1119 	nlk = nlk_sk(sk);
1120 
1121 	if ((atomic_read(&sk->sk_rmem_alloc) > sk->sk_rcvbuf ||
1122 	     test_bit(NETLINK_S_CONGESTED, &nlk->state))) {
1123 		DECLARE_WAITQUEUE(wait, current);
1124 		if (!*timeo) {
1125 			if (!ssk || netlink_is_kernel(ssk))
1126 				netlink_overrun(sk);
1127 			sock_put(sk);
1128 			kfree_skb(skb);
1129 			return -EAGAIN;
1130 		}
1131 
1132 		__set_current_state(TASK_INTERRUPTIBLE);
1133 		add_wait_queue(&nlk->wait, &wait);
1134 
1135 		if ((atomic_read(&sk->sk_rmem_alloc) > sk->sk_rcvbuf ||
1136 		     test_bit(NETLINK_S_CONGESTED, &nlk->state)) &&
1137 		    !sock_flag(sk, SOCK_DEAD))
1138 			*timeo = schedule_timeout(*timeo);
1139 
1140 		__set_current_state(TASK_RUNNING);
1141 		remove_wait_queue(&nlk->wait, &wait);
1142 		sock_put(sk);
1143 
1144 		if (signal_pending(current)) {
1145 			kfree_skb(skb);
1146 			return sock_intr_errno(*timeo);
1147 		}
1148 		return 1;
1149 	}
1150 	netlink_skb_set_owner_r(skb, sk);
1151 	return 0;
1152 }
1153 
1154 static int __netlink_sendskb(struct sock *sk, struct sk_buff *skb)
1155 {
1156 	int len = skb->len;
1157 
1158 	netlink_deliver_tap(skb);
1159 
1160 	skb_queue_tail(&sk->sk_receive_queue, skb);
1161 	sk->sk_data_ready(sk);
1162 	return len;
1163 }
1164 
1165 int netlink_sendskb(struct sock *sk, struct sk_buff *skb)
1166 {
1167 	int len = __netlink_sendskb(sk, skb);
1168 
1169 	sock_put(sk);
1170 	return len;
1171 }
1172 
1173 void netlink_detachskb(struct sock *sk, struct sk_buff *skb)
1174 {
1175 	kfree_skb(skb);
1176 	sock_put(sk);
1177 }
1178 
1179 static struct sk_buff *netlink_trim(struct sk_buff *skb, gfp_t allocation)
1180 {
1181 	int delta;
1182 
1183 	WARN_ON(skb->sk != NULL);
1184 	delta = skb->end - skb->tail;
1185 	if (is_vmalloc_addr(skb->head) || delta * 2 < skb->truesize)
1186 		return skb;
1187 
1188 	if (skb_shared(skb)) {
1189 		struct sk_buff *nskb = skb_clone(skb, allocation);
1190 		if (!nskb)
1191 			return skb;
1192 		consume_skb(skb);
1193 		skb = nskb;
1194 	}
1195 
1196 	if (!pskb_expand_head(skb, 0, -delta, allocation))
1197 		skb->truesize -= delta;
1198 
1199 	return skb;
1200 }
1201 
1202 static int netlink_unicast_kernel(struct sock *sk, struct sk_buff *skb,
1203 				  struct sock *ssk)
1204 {
1205 	int ret;
1206 	struct netlink_sock *nlk = nlk_sk(sk);
1207 
1208 	ret = -ECONNREFUSED;
1209 	if (nlk->netlink_rcv != NULL) {
1210 		ret = skb->len;
1211 		netlink_skb_set_owner_r(skb, sk);
1212 		NETLINK_CB(skb).sk = ssk;
1213 		netlink_deliver_tap_kernel(sk, ssk, skb);
1214 		nlk->netlink_rcv(skb);
1215 		consume_skb(skb);
1216 	} else {
1217 		kfree_skb(skb);
1218 	}
1219 	sock_put(sk);
1220 	return ret;
1221 }
1222 
1223 int netlink_unicast(struct sock *ssk, struct sk_buff *skb,
1224 		    u32 portid, int nonblock)
1225 {
1226 	struct sock *sk;
1227 	int err;
1228 	long timeo;
1229 
1230 	skb = netlink_trim(skb, gfp_any());
1231 
1232 	timeo = sock_sndtimeo(ssk, nonblock);
1233 retry:
1234 	sk = netlink_getsockbyportid(ssk, portid);
1235 	if (IS_ERR(sk)) {
1236 		kfree_skb(skb);
1237 		return PTR_ERR(sk);
1238 	}
1239 	if (netlink_is_kernel(sk))
1240 		return netlink_unicast_kernel(sk, skb, ssk);
1241 
1242 	if (sk_filter(sk, skb)) {
1243 		err = skb->len;
1244 		kfree_skb(skb);
1245 		sock_put(sk);
1246 		return err;
1247 	}
1248 
1249 	err = netlink_attachskb(sk, skb, &timeo, ssk);
1250 	if (err == 1)
1251 		goto retry;
1252 	if (err)
1253 		return err;
1254 
1255 	return netlink_sendskb(sk, skb);
1256 }
1257 EXPORT_SYMBOL(netlink_unicast);
1258 
1259 int netlink_has_listeners(struct sock *sk, unsigned int group)
1260 {
1261 	int res = 0;
1262 	struct listeners *listeners;
1263 
1264 	BUG_ON(!netlink_is_kernel(sk));
1265 
1266 	rcu_read_lock();
1267 	listeners = rcu_dereference(nl_table[sk->sk_protocol].listeners);
1268 
1269 	if (listeners && group - 1 < nl_table[sk->sk_protocol].groups)
1270 		res = test_bit(group - 1, listeners->masks);
1271 
1272 	rcu_read_unlock();
1273 
1274 	return res;
1275 }
1276 EXPORT_SYMBOL_GPL(netlink_has_listeners);
1277 
1278 static int netlink_broadcast_deliver(struct sock *sk, struct sk_buff *skb)
1279 {
1280 	struct netlink_sock *nlk = nlk_sk(sk);
1281 
1282 	if (atomic_read(&sk->sk_rmem_alloc) <= sk->sk_rcvbuf &&
1283 	    !test_bit(NETLINK_S_CONGESTED, &nlk->state)) {
1284 		netlink_skb_set_owner_r(skb, sk);
1285 		__netlink_sendskb(sk, skb);
1286 		return atomic_read(&sk->sk_rmem_alloc) > (sk->sk_rcvbuf >> 1);
1287 	}
1288 	return -1;
1289 }
1290 
1291 struct netlink_broadcast_data {
1292 	struct sock *exclude_sk;
1293 	struct net *net;
1294 	u32 portid;
1295 	u32 group;
1296 	int failure;
1297 	int delivery_failure;
1298 	int congested;
1299 	int delivered;
1300 	gfp_t allocation;
1301 	struct sk_buff *skb, *skb2;
1302 	int (*tx_filter)(struct sock *dsk, struct sk_buff *skb, void *data);
1303 	void *tx_data;
1304 };
1305 
1306 static void do_one_broadcast(struct sock *sk,
1307 				    struct netlink_broadcast_data *p)
1308 {
1309 	struct netlink_sock *nlk = nlk_sk(sk);
1310 	int val;
1311 
1312 	if (p->exclude_sk == sk)
1313 		return;
1314 
1315 	if (nlk->portid == p->portid || p->group - 1 >= nlk->ngroups ||
1316 	    !test_bit(p->group - 1, nlk->groups))
1317 		return;
1318 
1319 	if (!net_eq(sock_net(sk), p->net)) {
1320 		if (!(nlk->flags & NETLINK_F_LISTEN_ALL_NSID))
1321 			return;
1322 
1323 		if (!peernet_has_id(sock_net(sk), p->net))
1324 			return;
1325 
1326 		if (!file_ns_capable(sk->sk_socket->file, p->net->user_ns,
1327 				     CAP_NET_BROADCAST))
1328 			return;
1329 	}
1330 
1331 	if (p->failure) {
1332 		netlink_overrun(sk);
1333 		return;
1334 	}
1335 
1336 	sock_hold(sk);
1337 	if (p->skb2 == NULL) {
1338 		if (skb_shared(p->skb)) {
1339 			p->skb2 = skb_clone(p->skb, p->allocation);
1340 		} else {
1341 			p->skb2 = skb_get(p->skb);
1342 			/*
1343 			 * skb ownership may have been set when
1344 			 * delivered to a previous socket.
1345 			 */
1346 			skb_orphan(p->skb2);
1347 		}
1348 	}
1349 	if (p->skb2 == NULL) {
1350 		netlink_overrun(sk);
1351 		/* Clone failed. Notify ALL listeners. */
1352 		p->failure = 1;
1353 		if (nlk->flags & NETLINK_F_BROADCAST_SEND_ERROR)
1354 			p->delivery_failure = 1;
1355 		goto out;
1356 	}
1357 	if (p->tx_filter && p->tx_filter(sk, p->skb2, p->tx_data)) {
1358 		kfree_skb(p->skb2);
1359 		p->skb2 = NULL;
1360 		goto out;
1361 	}
1362 	if (sk_filter(sk, p->skb2)) {
1363 		kfree_skb(p->skb2);
1364 		p->skb2 = NULL;
1365 		goto out;
1366 	}
1367 	NETLINK_CB(p->skb2).nsid = peernet2id(sock_net(sk), p->net);
1368 	NETLINK_CB(p->skb2).nsid_is_set = true;
1369 	val = netlink_broadcast_deliver(sk, p->skb2);
1370 	if (val < 0) {
1371 		netlink_overrun(sk);
1372 		if (nlk->flags & NETLINK_F_BROADCAST_SEND_ERROR)
1373 			p->delivery_failure = 1;
1374 	} else {
1375 		p->congested |= val;
1376 		p->delivered = 1;
1377 		p->skb2 = NULL;
1378 	}
1379 out:
1380 	sock_put(sk);
1381 }
1382 
1383 int netlink_broadcast_filtered(struct sock *ssk, struct sk_buff *skb, u32 portid,
1384 	u32 group, gfp_t allocation,
1385 	int (*filter)(struct sock *dsk, struct sk_buff *skb, void *data),
1386 	void *filter_data)
1387 {
1388 	struct net *net = sock_net(ssk);
1389 	struct netlink_broadcast_data info;
1390 	struct sock *sk;
1391 
1392 	skb = netlink_trim(skb, allocation);
1393 
1394 	info.exclude_sk = ssk;
1395 	info.net = net;
1396 	info.portid = portid;
1397 	info.group = group;
1398 	info.failure = 0;
1399 	info.delivery_failure = 0;
1400 	info.congested = 0;
1401 	info.delivered = 0;
1402 	info.allocation = allocation;
1403 	info.skb = skb;
1404 	info.skb2 = NULL;
1405 	info.tx_filter = filter;
1406 	info.tx_data = filter_data;
1407 
1408 	/* While we sleep in clone, do not allow to change socket list */
1409 
1410 	netlink_lock_table();
1411 
1412 	sk_for_each_bound(sk, &nl_table[ssk->sk_protocol].mc_list)
1413 		do_one_broadcast(sk, &info);
1414 
1415 	consume_skb(skb);
1416 
1417 	netlink_unlock_table();
1418 
1419 	if (info.delivery_failure) {
1420 		kfree_skb(info.skb2);
1421 		return -ENOBUFS;
1422 	}
1423 	consume_skb(info.skb2);
1424 
1425 	if (info.delivered) {
1426 		if (info.congested && gfpflags_allow_blocking(allocation))
1427 			yield();
1428 		return 0;
1429 	}
1430 	return -ESRCH;
1431 }
1432 EXPORT_SYMBOL(netlink_broadcast_filtered);
1433 
1434 int netlink_broadcast(struct sock *ssk, struct sk_buff *skb, u32 portid,
1435 		      u32 group, gfp_t allocation)
1436 {
1437 	return netlink_broadcast_filtered(ssk, skb, portid, group, allocation,
1438 		NULL, NULL);
1439 }
1440 EXPORT_SYMBOL(netlink_broadcast);
1441 
1442 struct netlink_set_err_data {
1443 	struct sock *exclude_sk;
1444 	u32 portid;
1445 	u32 group;
1446 	int code;
1447 };
1448 
1449 static int do_one_set_err(struct sock *sk, struct netlink_set_err_data *p)
1450 {
1451 	struct netlink_sock *nlk = nlk_sk(sk);
1452 	int ret = 0;
1453 
1454 	if (sk == p->exclude_sk)
1455 		goto out;
1456 
1457 	if (!net_eq(sock_net(sk), sock_net(p->exclude_sk)))
1458 		goto out;
1459 
1460 	if (nlk->portid == p->portid || p->group - 1 >= nlk->ngroups ||
1461 	    !test_bit(p->group - 1, nlk->groups))
1462 		goto out;
1463 
1464 	if (p->code == ENOBUFS && nlk->flags & NETLINK_F_RECV_NO_ENOBUFS) {
1465 		ret = 1;
1466 		goto out;
1467 	}
1468 
1469 	sk->sk_err = p->code;
1470 	sk->sk_error_report(sk);
1471 out:
1472 	return ret;
1473 }
1474 
1475 /**
1476  * netlink_set_err - report error to broadcast listeners
1477  * @ssk: the kernel netlink socket, as returned by netlink_kernel_create()
1478  * @portid: the PORTID of a process that we want to skip (if any)
1479  * @group: the broadcast group that will notice the error
1480  * @code: error code, must be negative (as usual in kernelspace)
1481  *
1482  * This function returns the number of broadcast listeners that have set the
1483  * NETLINK_NO_ENOBUFS socket option.
1484  */
1485 int netlink_set_err(struct sock *ssk, u32 portid, u32 group, int code)
1486 {
1487 	struct netlink_set_err_data info;
1488 	struct sock *sk;
1489 	int ret = 0;
1490 
1491 	info.exclude_sk = ssk;
1492 	info.portid = portid;
1493 	info.group = group;
1494 	/* sk->sk_err wants a positive error value */
1495 	info.code = -code;
1496 
1497 	read_lock(&nl_table_lock);
1498 
1499 	sk_for_each_bound(sk, &nl_table[ssk->sk_protocol].mc_list)
1500 		ret += do_one_set_err(sk, &info);
1501 
1502 	read_unlock(&nl_table_lock);
1503 	return ret;
1504 }
1505 EXPORT_SYMBOL(netlink_set_err);
1506 
1507 /* must be called with netlink table grabbed */
1508 static void netlink_update_socket_mc(struct netlink_sock *nlk,
1509 				     unsigned int group,
1510 				     int is_new)
1511 {
1512 	int old, new = !!is_new, subscriptions;
1513 
1514 	old = test_bit(group - 1, nlk->groups);
1515 	subscriptions = nlk->subscriptions - old + new;
1516 	if (new)
1517 		__set_bit(group - 1, nlk->groups);
1518 	else
1519 		__clear_bit(group - 1, nlk->groups);
1520 	netlink_update_subscriptions(&nlk->sk, subscriptions);
1521 	netlink_update_listeners(&nlk->sk);
1522 }
1523 
1524 static int netlink_setsockopt(struct socket *sock, int level, int optname,
1525 			      char __user *optval, unsigned int optlen)
1526 {
1527 	struct sock *sk = sock->sk;
1528 	struct netlink_sock *nlk = nlk_sk(sk);
1529 	unsigned int val = 0;
1530 	int err;
1531 
1532 	if (level != SOL_NETLINK)
1533 		return -ENOPROTOOPT;
1534 
1535 	if (optlen >= sizeof(int) &&
1536 	    get_user(val, (unsigned int __user *)optval))
1537 		return -EFAULT;
1538 
1539 	switch (optname) {
1540 	case NETLINK_PKTINFO:
1541 		if (val)
1542 			nlk->flags |= NETLINK_F_RECV_PKTINFO;
1543 		else
1544 			nlk->flags &= ~NETLINK_F_RECV_PKTINFO;
1545 		err = 0;
1546 		break;
1547 	case NETLINK_ADD_MEMBERSHIP:
1548 	case NETLINK_DROP_MEMBERSHIP: {
1549 		if (!netlink_allowed(sock, NL_CFG_F_NONROOT_RECV))
1550 			return -EPERM;
1551 		err = netlink_realloc_groups(sk);
1552 		if (err)
1553 			return err;
1554 		if (!val || val - 1 >= nlk->ngroups)
1555 			return -EINVAL;
1556 		if (optname == NETLINK_ADD_MEMBERSHIP && nlk->netlink_bind) {
1557 			err = nlk->netlink_bind(sock_net(sk), val);
1558 			if (err)
1559 				return err;
1560 		}
1561 		netlink_table_grab();
1562 		netlink_update_socket_mc(nlk, val,
1563 					 optname == NETLINK_ADD_MEMBERSHIP);
1564 		netlink_table_ungrab();
1565 		if (optname == NETLINK_DROP_MEMBERSHIP && nlk->netlink_unbind)
1566 			nlk->netlink_unbind(sock_net(sk), val);
1567 
1568 		err = 0;
1569 		break;
1570 	}
1571 	case NETLINK_BROADCAST_ERROR:
1572 		if (val)
1573 			nlk->flags |= NETLINK_F_BROADCAST_SEND_ERROR;
1574 		else
1575 			nlk->flags &= ~NETLINK_F_BROADCAST_SEND_ERROR;
1576 		err = 0;
1577 		break;
1578 	case NETLINK_NO_ENOBUFS:
1579 		if (val) {
1580 			nlk->flags |= NETLINK_F_RECV_NO_ENOBUFS;
1581 			clear_bit(NETLINK_S_CONGESTED, &nlk->state);
1582 			wake_up_interruptible(&nlk->wait);
1583 		} else {
1584 			nlk->flags &= ~NETLINK_F_RECV_NO_ENOBUFS;
1585 		}
1586 		err = 0;
1587 		break;
1588 	case NETLINK_LISTEN_ALL_NSID:
1589 		if (!ns_capable(sock_net(sk)->user_ns, CAP_NET_BROADCAST))
1590 			return -EPERM;
1591 
1592 		if (val)
1593 			nlk->flags |= NETLINK_F_LISTEN_ALL_NSID;
1594 		else
1595 			nlk->flags &= ~NETLINK_F_LISTEN_ALL_NSID;
1596 		err = 0;
1597 		break;
1598 	case NETLINK_CAP_ACK:
1599 		if (val)
1600 			nlk->flags |= NETLINK_F_CAP_ACK;
1601 		else
1602 			nlk->flags &= ~NETLINK_F_CAP_ACK;
1603 		err = 0;
1604 		break;
1605 	default:
1606 		err = -ENOPROTOOPT;
1607 	}
1608 	return err;
1609 }
1610 
1611 static int netlink_getsockopt(struct socket *sock, int level, int optname,
1612 			      char __user *optval, int __user *optlen)
1613 {
1614 	struct sock *sk = sock->sk;
1615 	struct netlink_sock *nlk = nlk_sk(sk);
1616 	int len, val, err;
1617 
1618 	if (level != SOL_NETLINK)
1619 		return -ENOPROTOOPT;
1620 
1621 	if (get_user(len, optlen))
1622 		return -EFAULT;
1623 	if (len < 0)
1624 		return -EINVAL;
1625 
1626 	switch (optname) {
1627 	case NETLINK_PKTINFO:
1628 		if (len < sizeof(int))
1629 			return -EINVAL;
1630 		len = sizeof(int);
1631 		val = nlk->flags & NETLINK_F_RECV_PKTINFO ? 1 : 0;
1632 		if (put_user(len, optlen) ||
1633 		    put_user(val, optval))
1634 			return -EFAULT;
1635 		err = 0;
1636 		break;
1637 	case NETLINK_BROADCAST_ERROR:
1638 		if (len < sizeof(int))
1639 			return -EINVAL;
1640 		len = sizeof(int);
1641 		val = nlk->flags & NETLINK_F_BROADCAST_SEND_ERROR ? 1 : 0;
1642 		if (put_user(len, optlen) ||
1643 		    put_user(val, optval))
1644 			return -EFAULT;
1645 		err = 0;
1646 		break;
1647 	case NETLINK_NO_ENOBUFS:
1648 		if (len < sizeof(int))
1649 			return -EINVAL;
1650 		len = sizeof(int);
1651 		val = nlk->flags & NETLINK_F_RECV_NO_ENOBUFS ? 1 : 0;
1652 		if (put_user(len, optlen) ||
1653 		    put_user(val, optval))
1654 			return -EFAULT;
1655 		err = 0;
1656 		break;
1657 	case NETLINK_LIST_MEMBERSHIPS: {
1658 		int pos, idx, shift;
1659 
1660 		err = 0;
1661 		netlink_lock_table();
1662 		for (pos = 0; pos * 8 < nlk->ngroups; pos += sizeof(u32)) {
1663 			if (len - pos < sizeof(u32))
1664 				break;
1665 
1666 			idx = pos / sizeof(unsigned long);
1667 			shift = (pos % sizeof(unsigned long)) * 8;
1668 			if (put_user((u32)(nlk->groups[idx] >> shift),
1669 				     (u32 __user *)(optval + pos))) {
1670 				err = -EFAULT;
1671 				break;
1672 			}
1673 		}
1674 		if (put_user(ALIGN(nlk->ngroups / 8, sizeof(u32)), optlen))
1675 			err = -EFAULT;
1676 		netlink_unlock_table();
1677 		break;
1678 	}
1679 	case NETLINK_CAP_ACK:
1680 		if (len < sizeof(int))
1681 			return -EINVAL;
1682 		len = sizeof(int);
1683 		val = nlk->flags & NETLINK_F_CAP_ACK ? 1 : 0;
1684 		if (put_user(len, optlen) ||
1685 		    put_user(val, optval))
1686 			return -EFAULT;
1687 		err = 0;
1688 		break;
1689 	default:
1690 		err = -ENOPROTOOPT;
1691 	}
1692 	return err;
1693 }
1694 
1695 static void netlink_cmsg_recv_pktinfo(struct msghdr *msg, struct sk_buff *skb)
1696 {
1697 	struct nl_pktinfo info;
1698 
1699 	info.group = NETLINK_CB(skb).dst_group;
1700 	put_cmsg(msg, SOL_NETLINK, NETLINK_PKTINFO, sizeof(info), &info);
1701 }
1702 
1703 static void netlink_cmsg_listen_all_nsid(struct sock *sk, struct msghdr *msg,
1704 					 struct sk_buff *skb)
1705 {
1706 	if (!NETLINK_CB(skb).nsid_is_set)
1707 		return;
1708 
1709 	put_cmsg(msg, SOL_NETLINK, NETLINK_LISTEN_ALL_NSID, sizeof(int),
1710 		 &NETLINK_CB(skb).nsid);
1711 }
1712 
1713 static int netlink_sendmsg(struct socket *sock, struct msghdr *msg, size_t len)
1714 {
1715 	struct sock *sk = sock->sk;
1716 	struct netlink_sock *nlk = nlk_sk(sk);
1717 	DECLARE_SOCKADDR(struct sockaddr_nl *, addr, msg->msg_name);
1718 	u32 dst_portid;
1719 	u32 dst_group;
1720 	struct sk_buff *skb;
1721 	int err;
1722 	struct scm_cookie scm;
1723 	u32 netlink_skb_flags = 0;
1724 
1725 	if (msg->msg_flags&MSG_OOB)
1726 		return -EOPNOTSUPP;
1727 
1728 	err = scm_send(sock, msg, &scm, true);
1729 	if (err < 0)
1730 		return err;
1731 
1732 	if (msg->msg_namelen) {
1733 		err = -EINVAL;
1734 		if (addr->nl_family != AF_NETLINK)
1735 			goto out;
1736 		dst_portid = addr->nl_pid;
1737 		dst_group = ffs(addr->nl_groups);
1738 		err =  -EPERM;
1739 		if ((dst_group || dst_portid) &&
1740 		    !netlink_allowed(sock, NL_CFG_F_NONROOT_SEND))
1741 			goto out;
1742 		netlink_skb_flags |= NETLINK_SKB_DST;
1743 	} else {
1744 		dst_portid = nlk->dst_portid;
1745 		dst_group = nlk->dst_group;
1746 	}
1747 
1748 	if (!nlk->bound) {
1749 		err = netlink_autobind(sock);
1750 		if (err)
1751 			goto out;
1752 	} else {
1753 		/* Ensure nlk is hashed and visible. */
1754 		smp_rmb();
1755 	}
1756 
1757 	err = -EMSGSIZE;
1758 	if (len > sk->sk_sndbuf - 32)
1759 		goto out;
1760 	err = -ENOBUFS;
1761 	skb = netlink_alloc_large_skb(len, dst_group);
1762 	if (skb == NULL)
1763 		goto out;
1764 
1765 	NETLINK_CB(skb).portid	= nlk->portid;
1766 	NETLINK_CB(skb).dst_group = dst_group;
1767 	NETLINK_CB(skb).creds	= scm.creds;
1768 	NETLINK_CB(skb).flags	= netlink_skb_flags;
1769 
1770 	err = -EFAULT;
1771 	if (memcpy_from_msg(skb_put(skb, len), msg, len)) {
1772 		kfree_skb(skb);
1773 		goto out;
1774 	}
1775 
1776 	err = security_netlink_send(sk, skb);
1777 	if (err) {
1778 		kfree_skb(skb);
1779 		goto out;
1780 	}
1781 
1782 	if (dst_group) {
1783 		atomic_inc(&skb->users);
1784 		netlink_broadcast(sk, skb, dst_portid, dst_group, GFP_KERNEL);
1785 	}
1786 	err = netlink_unicast(sk, skb, dst_portid, msg->msg_flags&MSG_DONTWAIT);
1787 
1788 out:
1789 	scm_destroy(&scm);
1790 	return err;
1791 }
1792 
1793 static int netlink_recvmsg(struct socket *sock, struct msghdr *msg, size_t len,
1794 			   int flags)
1795 {
1796 	struct scm_cookie scm;
1797 	struct sock *sk = sock->sk;
1798 	struct netlink_sock *nlk = nlk_sk(sk);
1799 	int noblock = flags&MSG_DONTWAIT;
1800 	size_t copied;
1801 	struct sk_buff *skb, *data_skb;
1802 	int err, ret;
1803 
1804 	if (flags&MSG_OOB)
1805 		return -EOPNOTSUPP;
1806 
1807 	copied = 0;
1808 
1809 	skb = skb_recv_datagram(sk, flags, noblock, &err);
1810 	if (skb == NULL)
1811 		goto out;
1812 
1813 	data_skb = skb;
1814 
1815 #ifdef CONFIG_COMPAT_NETLINK_MESSAGES
1816 	if (unlikely(skb_shinfo(skb)->frag_list)) {
1817 		/*
1818 		 * If this skb has a frag_list, then here that means that we
1819 		 * will have to use the frag_list skb's data for compat tasks
1820 		 * and the regular skb's data for normal (non-compat) tasks.
1821 		 *
1822 		 * If we need to send the compat skb, assign it to the
1823 		 * 'data_skb' variable so that it will be used below for data
1824 		 * copying. We keep 'skb' for everything else, including
1825 		 * freeing both later.
1826 		 */
1827 		if (flags & MSG_CMSG_COMPAT)
1828 			data_skb = skb_shinfo(skb)->frag_list;
1829 	}
1830 #endif
1831 
1832 	/* Record the max length of recvmsg() calls for future allocations */
1833 	nlk->max_recvmsg_len = max(nlk->max_recvmsg_len, len);
1834 	nlk->max_recvmsg_len = min_t(size_t, nlk->max_recvmsg_len,
1835 				     16384);
1836 
1837 	copied = data_skb->len;
1838 	if (len < copied) {
1839 		msg->msg_flags |= MSG_TRUNC;
1840 		copied = len;
1841 	}
1842 
1843 	skb_reset_transport_header(data_skb);
1844 	err = skb_copy_datagram_msg(data_skb, 0, msg, copied);
1845 
1846 	if (msg->msg_name) {
1847 		DECLARE_SOCKADDR(struct sockaddr_nl *, addr, msg->msg_name);
1848 		addr->nl_family = AF_NETLINK;
1849 		addr->nl_pad    = 0;
1850 		addr->nl_pid	= NETLINK_CB(skb).portid;
1851 		addr->nl_groups	= netlink_group_mask(NETLINK_CB(skb).dst_group);
1852 		msg->msg_namelen = sizeof(*addr);
1853 	}
1854 
1855 	if (nlk->flags & NETLINK_F_RECV_PKTINFO)
1856 		netlink_cmsg_recv_pktinfo(msg, skb);
1857 	if (nlk->flags & NETLINK_F_LISTEN_ALL_NSID)
1858 		netlink_cmsg_listen_all_nsid(sk, msg, skb);
1859 
1860 	memset(&scm, 0, sizeof(scm));
1861 	scm.creds = *NETLINK_CREDS(skb);
1862 	if (flags & MSG_TRUNC)
1863 		copied = data_skb->len;
1864 
1865 	skb_free_datagram(sk, skb);
1866 
1867 	if (nlk->cb_running &&
1868 	    atomic_read(&sk->sk_rmem_alloc) <= sk->sk_rcvbuf / 2) {
1869 		ret = netlink_dump(sk);
1870 		if (ret) {
1871 			sk->sk_err = -ret;
1872 			sk->sk_error_report(sk);
1873 		}
1874 	}
1875 
1876 	scm_recv(sock, msg, &scm, flags);
1877 out:
1878 	netlink_rcv_wake(sk);
1879 	return err ? : copied;
1880 }
1881 
1882 static void netlink_data_ready(struct sock *sk)
1883 {
1884 	BUG();
1885 }
1886 
1887 /*
1888  *	We export these functions to other modules. They provide a
1889  *	complete set of kernel non-blocking support for message
1890  *	queueing.
1891  */
1892 
1893 struct sock *
1894 __netlink_kernel_create(struct net *net, int unit, struct module *module,
1895 			struct netlink_kernel_cfg *cfg)
1896 {
1897 	struct socket *sock;
1898 	struct sock *sk;
1899 	struct netlink_sock *nlk;
1900 	struct listeners *listeners = NULL;
1901 	struct mutex *cb_mutex = cfg ? cfg->cb_mutex : NULL;
1902 	unsigned int groups;
1903 
1904 	BUG_ON(!nl_table);
1905 
1906 	if (unit < 0 || unit >= MAX_LINKS)
1907 		return NULL;
1908 
1909 	if (sock_create_lite(PF_NETLINK, SOCK_DGRAM, unit, &sock))
1910 		return NULL;
1911 
1912 	if (__netlink_create(net, sock, cb_mutex, unit, 1) < 0)
1913 		goto out_sock_release_nosk;
1914 
1915 	sk = sock->sk;
1916 
1917 	if (!cfg || cfg->groups < 32)
1918 		groups = 32;
1919 	else
1920 		groups = cfg->groups;
1921 
1922 	listeners = kzalloc(sizeof(*listeners) + NLGRPSZ(groups), GFP_KERNEL);
1923 	if (!listeners)
1924 		goto out_sock_release;
1925 
1926 	sk->sk_data_ready = netlink_data_ready;
1927 	if (cfg && cfg->input)
1928 		nlk_sk(sk)->netlink_rcv = cfg->input;
1929 
1930 	if (netlink_insert(sk, 0))
1931 		goto out_sock_release;
1932 
1933 	nlk = nlk_sk(sk);
1934 	nlk->flags |= NETLINK_F_KERNEL_SOCKET;
1935 
1936 	netlink_table_grab();
1937 	if (!nl_table[unit].registered) {
1938 		nl_table[unit].groups = groups;
1939 		rcu_assign_pointer(nl_table[unit].listeners, listeners);
1940 		nl_table[unit].cb_mutex = cb_mutex;
1941 		nl_table[unit].module = module;
1942 		if (cfg) {
1943 			nl_table[unit].bind = cfg->bind;
1944 			nl_table[unit].unbind = cfg->unbind;
1945 			nl_table[unit].flags = cfg->flags;
1946 			if (cfg->compare)
1947 				nl_table[unit].compare = cfg->compare;
1948 		}
1949 		nl_table[unit].registered = 1;
1950 	} else {
1951 		kfree(listeners);
1952 		nl_table[unit].registered++;
1953 	}
1954 	netlink_table_ungrab();
1955 	return sk;
1956 
1957 out_sock_release:
1958 	kfree(listeners);
1959 	netlink_kernel_release(sk);
1960 	return NULL;
1961 
1962 out_sock_release_nosk:
1963 	sock_release(sock);
1964 	return NULL;
1965 }
1966 EXPORT_SYMBOL(__netlink_kernel_create);
1967 
1968 void
1969 netlink_kernel_release(struct sock *sk)
1970 {
1971 	if (sk == NULL || sk->sk_socket == NULL)
1972 		return;
1973 
1974 	sock_release(sk->sk_socket);
1975 }
1976 EXPORT_SYMBOL(netlink_kernel_release);
1977 
1978 int __netlink_change_ngroups(struct sock *sk, unsigned int groups)
1979 {
1980 	struct listeners *new, *old;
1981 	struct netlink_table *tbl = &nl_table[sk->sk_protocol];
1982 
1983 	if (groups < 32)
1984 		groups = 32;
1985 
1986 	if (NLGRPSZ(tbl->groups) < NLGRPSZ(groups)) {
1987 		new = kzalloc(sizeof(*new) + NLGRPSZ(groups), GFP_ATOMIC);
1988 		if (!new)
1989 			return -ENOMEM;
1990 		old = nl_deref_protected(tbl->listeners);
1991 		memcpy(new->masks, old->masks, NLGRPSZ(tbl->groups));
1992 		rcu_assign_pointer(tbl->listeners, new);
1993 
1994 		kfree_rcu(old, rcu);
1995 	}
1996 	tbl->groups = groups;
1997 
1998 	return 0;
1999 }
2000 
2001 /**
2002  * netlink_change_ngroups - change number of multicast groups
2003  *
2004  * This changes the number of multicast groups that are available
2005  * on a certain netlink family. Note that it is not possible to
2006  * change the number of groups to below 32. Also note that it does
2007  * not implicitly call netlink_clear_multicast_users() when the
2008  * number of groups is reduced.
2009  *
2010  * @sk: The kernel netlink socket, as returned by netlink_kernel_create().
2011  * @groups: The new number of groups.
2012  */
2013 int netlink_change_ngroups(struct sock *sk, unsigned int groups)
2014 {
2015 	int err;
2016 
2017 	netlink_table_grab();
2018 	err = __netlink_change_ngroups(sk, groups);
2019 	netlink_table_ungrab();
2020 
2021 	return err;
2022 }
2023 
2024 void __netlink_clear_multicast_users(struct sock *ksk, unsigned int group)
2025 {
2026 	struct sock *sk;
2027 	struct netlink_table *tbl = &nl_table[ksk->sk_protocol];
2028 
2029 	sk_for_each_bound(sk, &tbl->mc_list)
2030 		netlink_update_socket_mc(nlk_sk(sk), group, 0);
2031 }
2032 
2033 struct nlmsghdr *
2034 __nlmsg_put(struct sk_buff *skb, u32 portid, u32 seq, int type, int len, int flags)
2035 {
2036 	struct nlmsghdr *nlh;
2037 	int size = nlmsg_msg_size(len);
2038 
2039 	nlh = (struct nlmsghdr *)skb_put(skb, NLMSG_ALIGN(size));
2040 	nlh->nlmsg_type = type;
2041 	nlh->nlmsg_len = size;
2042 	nlh->nlmsg_flags = flags;
2043 	nlh->nlmsg_pid = portid;
2044 	nlh->nlmsg_seq = seq;
2045 	if (!__builtin_constant_p(size) || NLMSG_ALIGN(size) - size != 0)
2046 		memset(nlmsg_data(nlh) + len, 0, NLMSG_ALIGN(size) - size);
2047 	return nlh;
2048 }
2049 EXPORT_SYMBOL(__nlmsg_put);
2050 
2051 /*
2052  * It looks a bit ugly.
2053  * It would be better to create kernel thread.
2054  */
2055 
2056 static int netlink_dump(struct sock *sk)
2057 {
2058 	struct netlink_sock *nlk = nlk_sk(sk);
2059 	struct netlink_callback *cb;
2060 	struct sk_buff *skb = NULL;
2061 	struct nlmsghdr *nlh;
2062 	struct module *module;
2063 	int len, err = -ENOBUFS;
2064 	int alloc_min_size;
2065 	int alloc_size;
2066 
2067 	mutex_lock(nlk->cb_mutex);
2068 	if (!nlk->cb_running) {
2069 		err = -EINVAL;
2070 		goto errout_skb;
2071 	}
2072 
2073 	if (atomic_read(&sk->sk_rmem_alloc) >= sk->sk_rcvbuf)
2074 		goto errout_skb;
2075 
2076 	/* NLMSG_GOODSIZE is small to avoid high order allocations being
2077 	 * required, but it makes sense to _attempt_ a 16K bytes allocation
2078 	 * to reduce number of system calls on dump operations, if user
2079 	 * ever provided a big enough buffer.
2080 	 */
2081 	cb = &nlk->cb;
2082 	alloc_min_size = max_t(int, cb->min_dump_alloc, NLMSG_GOODSIZE);
2083 
2084 	if (alloc_min_size < nlk->max_recvmsg_len) {
2085 		alloc_size = nlk->max_recvmsg_len;
2086 		skb = alloc_skb(alloc_size, GFP_KERNEL |
2087 					    __GFP_NOWARN | __GFP_NORETRY);
2088 	}
2089 	if (!skb) {
2090 		alloc_size = alloc_min_size;
2091 		skb = alloc_skb(alloc_size, GFP_KERNEL);
2092 	}
2093 	if (!skb)
2094 		goto errout_skb;
2095 
2096 	/* Trim skb to allocated size. User is expected to provide buffer as
2097 	 * large as max(min_dump_alloc, 16KiB (mac_recvmsg_len capped at
2098 	 * netlink_recvmsg())). dump will pack as many smaller messages as
2099 	 * could fit within the allocated skb. skb is typically allocated
2100 	 * with larger space than required (could be as much as near 2x the
2101 	 * requested size with align to next power of 2 approach). Allowing
2102 	 * dump to use the excess space makes it difficult for a user to have a
2103 	 * reasonable static buffer based on the expected largest dump of a
2104 	 * single netdev. The outcome is MSG_TRUNC error.
2105 	 */
2106 	skb_reserve(skb, skb_tailroom(skb) - alloc_size);
2107 	netlink_skb_set_owner_r(skb, sk);
2108 
2109 	len = cb->dump(skb, cb);
2110 
2111 	if (len > 0) {
2112 		mutex_unlock(nlk->cb_mutex);
2113 
2114 		if (sk_filter(sk, skb))
2115 			kfree_skb(skb);
2116 		else
2117 			__netlink_sendskb(sk, skb);
2118 		return 0;
2119 	}
2120 
2121 	nlh = nlmsg_put_answer(skb, cb, NLMSG_DONE, sizeof(len), NLM_F_MULTI);
2122 	if (!nlh)
2123 		goto errout_skb;
2124 
2125 	nl_dump_check_consistent(cb, nlh);
2126 
2127 	memcpy(nlmsg_data(nlh), &len, sizeof(len));
2128 
2129 	if (sk_filter(sk, skb))
2130 		kfree_skb(skb);
2131 	else
2132 		__netlink_sendskb(sk, skb);
2133 
2134 	if (cb->done)
2135 		cb->done(cb);
2136 
2137 	nlk->cb_running = false;
2138 	module = cb->module;
2139 	skb = cb->skb;
2140 	mutex_unlock(nlk->cb_mutex);
2141 	module_put(module);
2142 	consume_skb(skb);
2143 	return 0;
2144 
2145 errout_skb:
2146 	mutex_unlock(nlk->cb_mutex);
2147 	kfree_skb(skb);
2148 	return err;
2149 }
2150 
2151 int __netlink_dump_start(struct sock *ssk, struct sk_buff *skb,
2152 			 const struct nlmsghdr *nlh,
2153 			 struct netlink_dump_control *control)
2154 {
2155 	struct netlink_callback *cb;
2156 	struct sock *sk;
2157 	struct netlink_sock *nlk;
2158 	int ret;
2159 
2160 	atomic_inc(&skb->users);
2161 
2162 	sk = netlink_lookup(sock_net(ssk), ssk->sk_protocol, NETLINK_CB(skb).portid);
2163 	if (sk == NULL) {
2164 		ret = -ECONNREFUSED;
2165 		goto error_free;
2166 	}
2167 
2168 	nlk = nlk_sk(sk);
2169 	mutex_lock(nlk->cb_mutex);
2170 	/* A dump is in progress... */
2171 	if (nlk->cb_running) {
2172 		ret = -EBUSY;
2173 		goto error_unlock;
2174 	}
2175 	/* add reference of module which cb->dump belongs to */
2176 	if (!try_module_get(control->module)) {
2177 		ret = -EPROTONOSUPPORT;
2178 		goto error_unlock;
2179 	}
2180 
2181 	cb = &nlk->cb;
2182 	memset(cb, 0, sizeof(*cb));
2183 	cb->start = control->start;
2184 	cb->dump = control->dump;
2185 	cb->done = control->done;
2186 	cb->nlh = nlh;
2187 	cb->data = control->data;
2188 	cb->module = control->module;
2189 	cb->min_dump_alloc = control->min_dump_alloc;
2190 	cb->skb = skb;
2191 
2192 	nlk->cb_running = true;
2193 
2194 	mutex_unlock(nlk->cb_mutex);
2195 
2196 	if (cb->start)
2197 		cb->start(cb);
2198 
2199 	ret = netlink_dump(sk);
2200 	sock_put(sk);
2201 
2202 	if (ret)
2203 		return ret;
2204 
2205 	/* We successfully started a dump, by returning -EINTR we
2206 	 * signal not to send ACK even if it was requested.
2207 	 */
2208 	return -EINTR;
2209 
2210 error_unlock:
2211 	sock_put(sk);
2212 	mutex_unlock(nlk->cb_mutex);
2213 error_free:
2214 	kfree_skb(skb);
2215 	return ret;
2216 }
2217 EXPORT_SYMBOL(__netlink_dump_start);
2218 
2219 void netlink_ack(struct sk_buff *in_skb, struct nlmsghdr *nlh, int err)
2220 {
2221 	struct sk_buff *skb;
2222 	struct nlmsghdr *rep;
2223 	struct nlmsgerr *errmsg;
2224 	size_t payload = sizeof(*errmsg);
2225 	struct netlink_sock *nlk = nlk_sk(NETLINK_CB(in_skb).sk);
2226 
2227 	/* Error messages get the original request appened, unless the user
2228 	 * requests to cap the error message.
2229 	 */
2230 	if (!(nlk->flags & NETLINK_F_CAP_ACK) && err)
2231 		payload += nlmsg_len(nlh);
2232 
2233 	skb = nlmsg_new(payload, GFP_KERNEL);
2234 	if (!skb) {
2235 		struct sock *sk;
2236 
2237 		sk = netlink_lookup(sock_net(in_skb->sk),
2238 				    in_skb->sk->sk_protocol,
2239 				    NETLINK_CB(in_skb).portid);
2240 		if (sk) {
2241 			sk->sk_err = ENOBUFS;
2242 			sk->sk_error_report(sk);
2243 			sock_put(sk);
2244 		}
2245 		return;
2246 	}
2247 
2248 	rep = __nlmsg_put(skb, NETLINK_CB(in_skb).portid, nlh->nlmsg_seq,
2249 			  NLMSG_ERROR, payload, 0);
2250 	errmsg = nlmsg_data(rep);
2251 	errmsg->error = err;
2252 	memcpy(&errmsg->msg, nlh, payload > sizeof(*errmsg) ? nlh->nlmsg_len : sizeof(*nlh));
2253 	netlink_unicast(in_skb->sk, skb, NETLINK_CB(in_skb).portid, MSG_DONTWAIT);
2254 }
2255 EXPORT_SYMBOL(netlink_ack);
2256 
2257 int netlink_rcv_skb(struct sk_buff *skb, int (*cb)(struct sk_buff *,
2258 						     struct nlmsghdr *))
2259 {
2260 	struct nlmsghdr *nlh;
2261 	int err;
2262 
2263 	while (skb->len >= nlmsg_total_size(0)) {
2264 		int msglen;
2265 
2266 		nlh = nlmsg_hdr(skb);
2267 		err = 0;
2268 
2269 		if (nlh->nlmsg_len < NLMSG_HDRLEN || skb->len < nlh->nlmsg_len)
2270 			return 0;
2271 
2272 		/* Only requests are handled by the kernel */
2273 		if (!(nlh->nlmsg_flags & NLM_F_REQUEST))
2274 			goto ack;
2275 
2276 		/* Skip control messages */
2277 		if (nlh->nlmsg_type < NLMSG_MIN_TYPE)
2278 			goto ack;
2279 
2280 		err = cb(skb, nlh);
2281 		if (err == -EINTR)
2282 			goto skip;
2283 
2284 ack:
2285 		if (nlh->nlmsg_flags & NLM_F_ACK || err)
2286 			netlink_ack(skb, nlh, err);
2287 
2288 skip:
2289 		msglen = NLMSG_ALIGN(nlh->nlmsg_len);
2290 		if (msglen > skb->len)
2291 			msglen = skb->len;
2292 		skb_pull(skb, msglen);
2293 	}
2294 
2295 	return 0;
2296 }
2297 EXPORT_SYMBOL(netlink_rcv_skb);
2298 
2299 /**
2300  * nlmsg_notify - send a notification netlink message
2301  * @sk: netlink socket to use
2302  * @skb: notification message
2303  * @portid: destination netlink portid for reports or 0
2304  * @group: destination multicast group or 0
2305  * @report: 1 to report back, 0 to disable
2306  * @flags: allocation flags
2307  */
2308 int nlmsg_notify(struct sock *sk, struct sk_buff *skb, u32 portid,
2309 		 unsigned int group, int report, gfp_t flags)
2310 {
2311 	int err = 0;
2312 
2313 	if (group) {
2314 		int exclude_portid = 0;
2315 
2316 		if (report) {
2317 			atomic_inc(&skb->users);
2318 			exclude_portid = portid;
2319 		}
2320 
2321 		/* errors reported via destination sk->sk_err, but propagate
2322 		 * delivery errors if NETLINK_BROADCAST_ERROR flag is set */
2323 		err = nlmsg_multicast(sk, skb, exclude_portid, group, flags);
2324 	}
2325 
2326 	if (report) {
2327 		int err2;
2328 
2329 		err2 = nlmsg_unicast(sk, skb, portid);
2330 		if (!err || err == -ESRCH)
2331 			err = err2;
2332 	}
2333 
2334 	return err;
2335 }
2336 EXPORT_SYMBOL(nlmsg_notify);
2337 
2338 #ifdef CONFIG_PROC_FS
2339 struct nl_seq_iter {
2340 	struct seq_net_private p;
2341 	struct rhashtable_iter hti;
2342 	int link;
2343 };
2344 
2345 static int netlink_walk_start(struct nl_seq_iter *iter)
2346 {
2347 	int err;
2348 
2349 	err = rhashtable_walk_init(&nl_table[iter->link].hash, &iter->hti,
2350 				   GFP_KERNEL);
2351 	if (err) {
2352 		iter->link = MAX_LINKS;
2353 		return err;
2354 	}
2355 
2356 	err = rhashtable_walk_start(&iter->hti);
2357 	return err == -EAGAIN ? 0 : err;
2358 }
2359 
2360 static void netlink_walk_stop(struct nl_seq_iter *iter)
2361 {
2362 	rhashtable_walk_stop(&iter->hti);
2363 	rhashtable_walk_exit(&iter->hti);
2364 }
2365 
2366 static void *__netlink_seq_next(struct seq_file *seq)
2367 {
2368 	struct nl_seq_iter *iter = seq->private;
2369 	struct netlink_sock *nlk;
2370 
2371 	do {
2372 		for (;;) {
2373 			int err;
2374 
2375 			nlk = rhashtable_walk_next(&iter->hti);
2376 
2377 			if (IS_ERR(nlk)) {
2378 				if (PTR_ERR(nlk) == -EAGAIN)
2379 					continue;
2380 
2381 				return nlk;
2382 			}
2383 
2384 			if (nlk)
2385 				break;
2386 
2387 			netlink_walk_stop(iter);
2388 			if (++iter->link >= MAX_LINKS)
2389 				return NULL;
2390 
2391 			err = netlink_walk_start(iter);
2392 			if (err)
2393 				return ERR_PTR(err);
2394 		}
2395 	} while (sock_net(&nlk->sk) != seq_file_net(seq));
2396 
2397 	return nlk;
2398 }
2399 
2400 static void *netlink_seq_start(struct seq_file *seq, loff_t *posp)
2401 {
2402 	struct nl_seq_iter *iter = seq->private;
2403 	void *obj = SEQ_START_TOKEN;
2404 	loff_t pos;
2405 	int err;
2406 
2407 	iter->link = 0;
2408 
2409 	err = netlink_walk_start(iter);
2410 	if (err)
2411 		return ERR_PTR(err);
2412 
2413 	for (pos = *posp; pos && obj && !IS_ERR(obj); pos--)
2414 		obj = __netlink_seq_next(seq);
2415 
2416 	return obj;
2417 }
2418 
2419 static void *netlink_seq_next(struct seq_file *seq, void *v, loff_t *pos)
2420 {
2421 	++*pos;
2422 	return __netlink_seq_next(seq);
2423 }
2424 
2425 static void netlink_seq_stop(struct seq_file *seq, void *v)
2426 {
2427 	struct nl_seq_iter *iter = seq->private;
2428 
2429 	if (iter->link >= MAX_LINKS)
2430 		return;
2431 
2432 	netlink_walk_stop(iter);
2433 }
2434 
2435 
2436 static int netlink_seq_show(struct seq_file *seq, void *v)
2437 {
2438 	if (v == SEQ_START_TOKEN) {
2439 		seq_puts(seq,
2440 			 "sk       Eth Pid    Groups   "
2441 			 "Rmem     Wmem     Dump     Locks     Drops     Inode\n");
2442 	} else {
2443 		struct sock *s = v;
2444 		struct netlink_sock *nlk = nlk_sk(s);
2445 
2446 		seq_printf(seq, "%pK %-3d %-6u %08x %-8d %-8d %d %-8d %-8d %-8lu\n",
2447 			   s,
2448 			   s->sk_protocol,
2449 			   nlk->portid,
2450 			   nlk->groups ? (u32)nlk->groups[0] : 0,
2451 			   sk_rmem_alloc_get(s),
2452 			   sk_wmem_alloc_get(s),
2453 			   nlk->cb_running,
2454 			   atomic_read(&s->sk_refcnt),
2455 			   atomic_read(&s->sk_drops),
2456 			   sock_i_ino(s)
2457 			);
2458 
2459 	}
2460 	return 0;
2461 }
2462 
2463 static const struct seq_operations netlink_seq_ops = {
2464 	.start  = netlink_seq_start,
2465 	.next   = netlink_seq_next,
2466 	.stop   = netlink_seq_stop,
2467 	.show   = netlink_seq_show,
2468 };
2469 
2470 
2471 static int netlink_seq_open(struct inode *inode, struct file *file)
2472 {
2473 	return seq_open_net(inode, file, &netlink_seq_ops,
2474 				sizeof(struct nl_seq_iter));
2475 }
2476 
2477 static const struct file_operations netlink_seq_fops = {
2478 	.owner		= THIS_MODULE,
2479 	.open		= netlink_seq_open,
2480 	.read		= seq_read,
2481 	.llseek		= seq_lseek,
2482 	.release	= seq_release_net,
2483 };
2484 
2485 #endif
2486 
2487 int netlink_register_notifier(struct notifier_block *nb)
2488 {
2489 	return atomic_notifier_chain_register(&netlink_chain, nb);
2490 }
2491 EXPORT_SYMBOL(netlink_register_notifier);
2492 
2493 int netlink_unregister_notifier(struct notifier_block *nb)
2494 {
2495 	return atomic_notifier_chain_unregister(&netlink_chain, nb);
2496 }
2497 EXPORT_SYMBOL(netlink_unregister_notifier);
2498 
2499 static const struct proto_ops netlink_ops = {
2500 	.family =	PF_NETLINK,
2501 	.owner =	THIS_MODULE,
2502 	.release =	netlink_release,
2503 	.bind =		netlink_bind,
2504 	.connect =	netlink_connect,
2505 	.socketpair =	sock_no_socketpair,
2506 	.accept =	sock_no_accept,
2507 	.getname =	netlink_getname,
2508 	.poll =		datagram_poll,
2509 	.ioctl =	netlink_ioctl,
2510 	.listen =	sock_no_listen,
2511 	.shutdown =	sock_no_shutdown,
2512 	.setsockopt =	netlink_setsockopt,
2513 	.getsockopt =	netlink_getsockopt,
2514 	.sendmsg =	netlink_sendmsg,
2515 	.recvmsg =	netlink_recvmsg,
2516 	.mmap =		sock_no_mmap,
2517 	.sendpage =	sock_no_sendpage,
2518 };
2519 
2520 static const struct net_proto_family netlink_family_ops = {
2521 	.family = PF_NETLINK,
2522 	.create = netlink_create,
2523 	.owner	= THIS_MODULE,	/* for consistency 8) */
2524 };
2525 
2526 static int __net_init netlink_net_init(struct net *net)
2527 {
2528 #ifdef CONFIG_PROC_FS
2529 	if (!proc_create("netlink", 0, net->proc_net, &netlink_seq_fops))
2530 		return -ENOMEM;
2531 #endif
2532 	return 0;
2533 }
2534 
2535 static void __net_exit netlink_net_exit(struct net *net)
2536 {
2537 #ifdef CONFIG_PROC_FS
2538 	remove_proc_entry("netlink", net->proc_net);
2539 #endif
2540 }
2541 
2542 static void __init netlink_add_usersock_entry(void)
2543 {
2544 	struct listeners *listeners;
2545 	int groups = 32;
2546 
2547 	listeners = kzalloc(sizeof(*listeners) + NLGRPSZ(groups), GFP_KERNEL);
2548 	if (!listeners)
2549 		panic("netlink_add_usersock_entry: Cannot allocate listeners\n");
2550 
2551 	netlink_table_grab();
2552 
2553 	nl_table[NETLINK_USERSOCK].groups = groups;
2554 	rcu_assign_pointer(nl_table[NETLINK_USERSOCK].listeners, listeners);
2555 	nl_table[NETLINK_USERSOCK].module = THIS_MODULE;
2556 	nl_table[NETLINK_USERSOCK].registered = 1;
2557 	nl_table[NETLINK_USERSOCK].flags = NL_CFG_F_NONROOT_SEND;
2558 
2559 	netlink_table_ungrab();
2560 }
2561 
2562 static struct pernet_operations __net_initdata netlink_net_ops = {
2563 	.init = netlink_net_init,
2564 	.exit = netlink_net_exit,
2565 };
2566 
2567 static inline u32 netlink_hash(const void *data, u32 len, u32 seed)
2568 {
2569 	const struct netlink_sock *nlk = data;
2570 	struct netlink_compare_arg arg;
2571 
2572 	netlink_compare_arg_init(&arg, sock_net(&nlk->sk), nlk->portid);
2573 	return jhash2((u32 *)&arg, netlink_compare_arg_len / sizeof(u32), seed);
2574 }
2575 
2576 static const struct rhashtable_params netlink_rhashtable_params = {
2577 	.head_offset = offsetof(struct netlink_sock, node),
2578 	.key_len = netlink_compare_arg_len,
2579 	.obj_hashfn = netlink_hash,
2580 	.obj_cmpfn = netlink_compare,
2581 	.automatic_shrinking = true,
2582 };
2583 
2584 static int __init netlink_proto_init(void)
2585 {
2586 	int i;
2587 	int err = proto_register(&netlink_proto, 0);
2588 
2589 	if (err != 0)
2590 		goto out;
2591 
2592 	BUILD_BUG_ON(sizeof(struct netlink_skb_parms) > FIELD_SIZEOF(struct sk_buff, cb));
2593 
2594 	nl_table = kcalloc(MAX_LINKS, sizeof(*nl_table), GFP_KERNEL);
2595 	if (!nl_table)
2596 		goto panic;
2597 
2598 	for (i = 0; i < MAX_LINKS; i++) {
2599 		if (rhashtable_init(&nl_table[i].hash,
2600 				    &netlink_rhashtable_params) < 0) {
2601 			while (--i > 0)
2602 				rhashtable_destroy(&nl_table[i].hash);
2603 			kfree(nl_table);
2604 			goto panic;
2605 		}
2606 	}
2607 
2608 	INIT_LIST_HEAD(&netlink_tap_all);
2609 
2610 	netlink_add_usersock_entry();
2611 
2612 	sock_register(&netlink_family_ops);
2613 	register_pernet_subsys(&netlink_net_ops);
2614 	/* The netlink device handler may be needed early. */
2615 	rtnetlink_init();
2616 out:
2617 	return err;
2618 panic:
2619 	panic("netlink_init: Cannot allocate nl_table\n");
2620 }
2621 
2622 core_initcall(netlink_proto_init);
2623