]> asedeno.scripts.mit.edu Git - linux.git/blob - net/core/net_namespace.c
Merge tag 'xfs-5.6-merge-8' of git://git.kernel.org/pub/scm/fs/xfs/xfs-linux
[linux.git] / net / core / net_namespace.c
1 // SPDX-License-Identifier: GPL-2.0-only
2 #define pr_fmt(fmt) KBUILD_MODNAME ": " fmt
3
4 #include <linux/workqueue.h>
5 #include <linux/rtnetlink.h>
6 #include <linux/cache.h>
7 #include <linux/slab.h>
8 #include <linux/list.h>
9 #include <linux/delay.h>
10 #include <linux/sched.h>
11 #include <linux/idr.h>
12 #include <linux/rculist.h>
13 #include <linux/nsproxy.h>
14 #include <linux/fs.h>
15 #include <linux/proc_ns.h>
16 #include <linux/file.h>
17 #include <linux/export.h>
18 #include <linux/user_namespace.h>
19 #include <linux/net_namespace.h>
20 #include <linux/sched/task.h>
21 #include <linux/uidgid.h>
22
23 #include <net/sock.h>
24 #include <net/netlink.h>
25 #include <net/net_namespace.h>
26 #include <net/netns/generic.h>
27
28 /*
29  *      Our network namespace constructor/destructor lists
30  */
31
32 static LIST_HEAD(pernet_list);
33 static struct list_head *first_device = &pernet_list;
34
35 LIST_HEAD(net_namespace_list);
36 EXPORT_SYMBOL_GPL(net_namespace_list);
37
38 /* Protects net_namespace_list. Nests iside rtnl_lock() */
39 DECLARE_RWSEM(net_rwsem);
40 EXPORT_SYMBOL_GPL(net_rwsem);
41
42 #ifdef CONFIG_KEYS
43 static struct key_tag init_net_key_domain = { .usage = REFCOUNT_INIT(1) };
44 #endif
45
46 struct net init_net = {
47         .count          = REFCOUNT_INIT(1),
48         .dev_base_head  = LIST_HEAD_INIT(init_net.dev_base_head),
49 #ifdef CONFIG_KEYS
50         .key_domain     = &init_net_key_domain,
51 #endif
52 };
53 EXPORT_SYMBOL(init_net);
54
55 static bool init_net_initialized;
56 /*
57  * pernet_ops_rwsem: protects: pernet_list, net_generic_ids,
58  * init_net_initialized and first_device pointer.
59  * This is internal net namespace object. Please, don't use it
60  * outside.
61  */
62 DECLARE_RWSEM(pernet_ops_rwsem);
63 EXPORT_SYMBOL_GPL(pernet_ops_rwsem);
64
65 #define MIN_PERNET_OPS_ID       \
66         ((sizeof(struct net_generic) + sizeof(void *) - 1) / sizeof(void *))
67
68 #define INITIAL_NET_GEN_PTRS    13 /* +1 for len +2 for rcu_head */
69
70 static unsigned int max_gen_ptrs = INITIAL_NET_GEN_PTRS;
71
72 static struct net_generic *net_alloc_generic(void)
73 {
74         struct net_generic *ng;
75         unsigned int generic_size = offsetof(struct net_generic, ptr[max_gen_ptrs]);
76
77         ng = kzalloc(generic_size, GFP_KERNEL);
78         if (ng)
79                 ng->s.len = max_gen_ptrs;
80
81         return ng;
82 }
83
84 static int net_assign_generic(struct net *net, unsigned int id, void *data)
85 {
86         struct net_generic *ng, *old_ng;
87
88         BUG_ON(id < MIN_PERNET_OPS_ID);
89
90         old_ng = rcu_dereference_protected(net->gen,
91                                            lockdep_is_held(&pernet_ops_rwsem));
92         if (old_ng->s.len > id) {
93                 old_ng->ptr[id] = data;
94                 return 0;
95         }
96
97         ng = net_alloc_generic();
98         if (ng == NULL)
99                 return -ENOMEM;
100
101         /*
102          * Some synchronisation notes:
103          *
104          * The net_generic explores the net->gen array inside rcu
105          * read section. Besides once set the net->gen->ptr[x]
106          * pointer never changes (see rules in netns/generic.h).
107          *
108          * That said, we simply duplicate this array and schedule
109          * the old copy for kfree after a grace period.
110          */
111
112         memcpy(&ng->ptr[MIN_PERNET_OPS_ID], &old_ng->ptr[MIN_PERNET_OPS_ID],
113                (old_ng->s.len - MIN_PERNET_OPS_ID) * sizeof(void *));
114         ng->ptr[id] = data;
115
116         rcu_assign_pointer(net->gen, ng);
117         kfree_rcu(old_ng, s.rcu);
118         return 0;
119 }
120
121 static int ops_init(const struct pernet_operations *ops, struct net *net)
122 {
123         int err = -ENOMEM;
124         void *data = NULL;
125
126         if (ops->id && ops->size) {
127                 data = kzalloc(ops->size, GFP_KERNEL);
128                 if (!data)
129                         goto out;
130
131                 err = net_assign_generic(net, *ops->id, data);
132                 if (err)
133                         goto cleanup;
134         }
135         err = 0;
136         if (ops->init)
137                 err = ops->init(net);
138         if (!err)
139                 return 0;
140
141 cleanup:
142         kfree(data);
143
144 out:
145         return err;
146 }
147
148 static void ops_free(const struct pernet_operations *ops, struct net *net)
149 {
150         if (ops->id && ops->size) {
151                 kfree(net_generic(net, *ops->id));
152         }
153 }
154
155 static void ops_pre_exit_list(const struct pernet_operations *ops,
156                               struct list_head *net_exit_list)
157 {
158         struct net *net;
159
160         if (ops->pre_exit) {
161                 list_for_each_entry(net, net_exit_list, exit_list)
162                         ops->pre_exit(net);
163         }
164 }
165
166 static void ops_exit_list(const struct pernet_operations *ops,
167                           struct list_head *net_exit_list)
168 {
169         struct net *net;
170         if (ops->exit) {
171                 list_for_each_entry(net, net_exit_list, exit_list)
172                         ops->exit(net);
173         }
174         if (ops->exit_batch)
175                 ops->exit_batch(net_exit_list);
176 }
177
178 static void ops_free_list(const struct pernet_operations *ops,
179                           struct list_head *net_exit_list)
180 {
181         struct net *net;
182         if (ops->size && ops->id) {
183                 list_for_each_entry(net, net_exit_list, exit_list)
184                         ops_free(ops, net);
185         }
186 }
187
188 /* should be called with nsid_lock held */
189 static int alloc_netid(struct net *net, struct net *peer, int reqid)
190 {
191         int min = 0, max = 0;
192
193         if (reqid >= 0) {
194                 min = reqid;
195                 max = reqid + 1;
196         }
197
198         return idr_alloc(&net->netns_ids, peer, min, max, GFP_ATOMIC);
199 }
200
201 /* This function is used by idr_for_each(). If net is equal to peer, the
202  * function returns the id so that idr_for_each() stops. Because we cannot
203  * returns the id 0 (idr_for_each() will not stop), we return the magic value
204  * NET_ID_ZERO (-1) for it.
205  */
206 #define NET_ID_ZERO -1
207 static int net_eq_idr(int id, void *net, void *peer)
208 {
209         if (net_eq(net, peer))
210                 return id ? : NET_ID_ZERO;
211         return 0;
212 }
213
214 /* Must be called from RCU-critical section or with nsid_lock held */
215 static int __peernet2id(const struct net *net, struct net *peer)
216 {
217         int id = idr_for_each(&net->netns_ids, net_eq_idr, peer);
218
219         /* Magic value for id 0. */
220         if (id == NET_ID_ZERO)
221                 return 0;
222         if (id > 0)
223                 return id;
224
225         return NETNSA_NSID_NOT_ASSIGNED;
226 }
227
228 static void rtnl_net_notifyid(struct net *net, int cmd, int id, u32 portid,
229                               struct nlmsghdr *nlh, gfp_t gfp);
230 /* This function returns the id of a peer netns. If no id is assigned, one will
231  * be allocated and returned.
232  */
233 int peernet2id_alloc(struct net *net, struct net *peer, gfp_t gfp)
234 {
235         int id;
236
237         if (refcount_read(&net->count) == 0)
238                 return NETNSA_NSID_NOT_ASSIGNED;
239
240         spin_lock(&net->nsid_lock);
241         id = __peernet2id(net, peer);
242         if (id >= 0) {
243                 spin_unlock(&net->nsid_lock);
244                 return id;
245         }
246
247         /* When peer is obtained from RCU lists, we may race with
248          * its cleanup. Check whether it's alive, and this guarantees
249          * we never hash a peer back to net->netns_ids, after it has
250          * just been idr_remove()'d from there in cleanup_net().
251          */
252         if (!maybe_get_net(peer)) {
253                 spin_unlock(&net->nsid_lock);
254                 return NETNSA_NSID_NOT_ASSIGNED;
255         }
256
257         id = alloc_netid(net, peer, -1);
258         spin_unlock(&net->nsid_lock);
259
260         put_net(peer);
261         if (id < 0)
262                 return NETNSA_NSID_NOT_ASSIGNED;
263
264         rtnl_net_notifyid(net, RTM_NEWNSID, id, 0, NULL, gfp);
265
266         return id;
267 }
268 EXPORT_SYMBOL_GPL(peernet2id_alloc);
269
270 /* This function returns, if assigned, the id of a peer netns. */
271 int peernet2id(const struct net *net, struct net *peer)
272 {
273         int id;
274
275         rcu_read_lock();
276         id = __peernet2id(net, peer);
277         rcu_read_unlock();
278
279         return id;
280 }
281 EXPORT_SYMBOL(peernet2id);
282
283 /* This function returns true is the peer netns has an id assigned into the
284  * current netns.
285  */
286 bool peernet_has_id(const struct net *net, struct net *peer)
287 {
288         return peernet2id(net, peer) >= 0;
289 }
290
291 struct net *get_net_ns_by_id(const struct net *net, int id)
292 {
293         struct net *peer;
294
295         if (id < 0)
296                 return NULL;
297
298         rcu_read_lock();
299         peer = idr_find(&net->netns_ids, id);
300         if (peer)
301                 peer = maybe_get_net(peer);
302         rcu_read_unlock();
303
304         return peer;
305 }
306
307 /*
308  * setup_net runs the initializers for the network namespace object.
309  */
310 static __net_init int setup_net(struct net *net, struct user_namespace *user_ns)
311 {
312         /* Must be called with pernet_ops_rwsem held */
313         const struct pernet_operations *ops, *saved_ops;
314         int error = 0;
315         LIST_HEAD(net_exit_list);
316
317         refcount_set(&net->count, 1);
318         refcount_set(&net->passive, 1);
319         get_random_bytes(&net->hash_mix, sizeof(u32));
320         net->dev_base_seq = 1;
321         net->user_ns = user_ns;
322         idr_init(&net->netns_ids);
323         spin_lock_init(&net->nsid_lock);
324         mutex_init(&net->ipv4.ra_mutex);
325
326         list_for_each_entry(ops, &pernet_list, list) {
327                 error = ops_init(ops, net);
328                 if (error < 0)
329                         goto out_undo;
330         }
331         down_write(&net_rwsem);
332         list_add_tail_rcu(&net->list, &net_namespace_list);
333         up_write(&net_rwsem);
334 out:
335         return error;
336
337 out_undo:
338         /* Walk through the list backwards calling the exit functions
339          * for the pernet modules whose init functions did not fail.
340          */
341         list_add(&net->exit_list, &net_exit_list);
342         saved_ops = ops;
343         list_for_each_entry_continue_reverse(ops, &pernet_list, list)
344                 ops_pre_exit_list(ops, &net_exit_list);
345
346         synchronize_rcu();
347
348         ops = saved_ops;
349         list_for_each_entry_continue_reverse(ops, &pernet_list, list)
350                 ops_exit_list(ops, &net_exit_list);
351
352         ops = saved_ops;
353         list_for_each_entry_continue_reverse(ops, &pernet_list, list)
354                 ops_free_list(ops, &net_exit_list);
355
356         rcu_barrier();
357         goto out;
358 }
359
360 static int __net_init net_defaults_init_net(struct net *net)
361 {
362         net->core.sysctl_somaxconn = SOMAXCONN;
363         return 0;
364 }
365
366 static struct pernet_operations net_defaults_ops = {
367         .init = net_defaults_init_net,
368 };
369
370 static __init int net_defaults_init(void)
371 {
372         if (register_pernet_subsys(&net_defaults_ops))
373                 panic("Cannot initialize net default settings");
374
375         return 0;
376 }
377
378 core_initcall(net_defaults_init);
379
380 #ifdef CONFIG_NET_NS
381 static struct ucounts *inc_net_namespaces(struct user_namespace *ns)
382 {
383         return inc_ucount(ns, current_euid(), UCOUNT_NET_NAMESPACES);
384 }
385
386 static void dec_net_namespaces(struct ucounts *ucounts)
387 {
388         dec_ucount(ucounts, UCOUNT_NET_NAMESPACES);
389 }
390
391 static struct kmem_cache *net_cachep __ro_after_init;
392 static struct workqueue_struct *netns_wq;
393
394 static struct net *net_alloc(void)
395 {
396         struct net *net = NULL;
397         struct net_generic *ng;
398
399         ng = net_alloc_generic();
400         if (!ng)
401                 goto out;
402
403         net = kmem_cache_zalloc(net_cachep, GFP_KERNEL);
404         if (!net)
405                 goto out_free;
406
407 #ifdef CONFIG_KEYS
408         net->key_domain = kzalloc(sizeof(struct key_tag), GFP_KERNEL);
409         if (!net->key_domain)
410                 goto out_free_2;
411         refcount_set(&net->key_domain->usage, 1);
412 #endif
413
414         rcu_assign_pointer(net->gen, ng);
415 out:
416         return net;
417
418 #ifdef CONFIG_KEYS
419 out_free_2:
420         kmem_cache_free(net_cachep, net);
421         net = NULL;
422 #endif
423 out_free:
424         kfree(ng);
425         goto out;
426 }
427
428 static void net_free(struct net *net)
429 {
430         kfree(rcu_access_pointer(net->gen));
431         kmem_cache_free(net_cachep, net);
432 }
433
434 void net_drop_ns(void *p)
435 {
436         struct net *ns = p;
437         if (ns && refcount_dec_and_test(&ns->passive))
438                 net_free(ns);
439 }
440
441 struct net *copy_net_ns(unsigned long flags,
442                         struct user_namespace *user_ns, struct net *old_net)
443 {
444         struct ucounts *ucounts;
445         struct net *net;
446         int rv;
447
448         if (!(flags & CLONE_NEWNET))
449                 return get_net(old_net);
450
451         ucounts = inc_net_namespaces(user_ns);
452         if (!ucounts)
453                 return ERR_PTR(-ENOSPC);
454
455         net = net_alloc();
456         if (!net) {
457                 rv = -ENOMEM;
458                 goto dec_ucounts;
459         }
460         refcount_set(&net->passive, 1);
461         net->ucounts = ucounts;
462         get_user_ns(user_ns);
463
464         rv = down_read_killable(&pernet_ops_rwsem);
465         if (rv < 0)
466                 goto put_userns;
467
468         rv = setup_net(net, user_ns);
469
470         up_read(&pernet_ops_rwsem);
471
472         if (rv < 0) {
473 put_userns:
474                 key_remove_domain(net->key_domain);
475                 put_user_ns(user_ns);
476                 net_drop_ns(net);
477 dec_ucounts:
478                 dec_net_namespaces(ucounts);
479                 return ERR_PTR(rv);
480         }
481         return net;
482 }
483
484 /**
485  * net_ns_get_ownership - get sysfs ownership data for @net
486  * @net: network namespace in question (can be NULL)
487  * @uid: kernel user ID for sysfs objects
488  * @gid: kernel group ID for sysfs objects
489  *
490  * Returns the uid/gid pair of root in the user namespace associated with the
491  * given network namespace.
492  */
493 void net_ns_get_ownership(const struct net *net, kuid_t *uid, kgid_t *gid)
494 {
495         if (net) {
496                 kuid_t ns_root_uid = make_kuid(net->user_ns, 0);
497                 kgid_t ns_root_gid = make_kgid(net->user_ns, 0);
498
499                 if (uid_valid(ns_root_uid))
500                         *uid = ns_root_uid;
501
502                 if (gid_valid(ns_root_gid))
503                         *gid = ns_root_gid;
504         } else {
505                 *uid = GLOBAL_ROOT_UID;
506                 *gid = GLOBAL_ROOT_GID;
507         }
508 }
509 EXPORT_SYMBOL_GPL(net_ns_get_ownership);
510
511 static void unhash_nsid(struct net *net, struct net *last)
512 {
513         struct net *tmp;
514         /* This function is only called from cleanup_net() work,
515          * and this work is the only process, that may delete
516          * a net from net_namespace_list. So, when the below
517          * is executing, the list may only grow. Thus, we do not
518          * use for_each_net_rcu() or net_rwsem.
519          */
520         for_each_net(tmp) {
521                 int id;
522
523                 spin_lock(&tmp->nsid_lock);
524                 id = __peernet2id(tmp, net);
525                 if (id >= 0)
526                         idr_remove(&tmp->netns_ids, id);
527                 spin_unlock(&tmp->nsid_lock);
528                 if (id >= 0)
529                         rtnl_net_notifyid(tmp, RTM_DELNSID, id, 0, NULL,
530                                           GFP_KERNEL);
531                 if (tmp == last)
532                         break;
533         }
534         spin_lock(&net->nsid_lock);
535         idr_destroy(&net->netns_ids);
536         spin_unlock(&net->nsid_lock);
537 }
538
539 static LLIST_HEAD(cleanup_list);
540
541 static void cleanup_net(struct work_struct *work)
542 {
543         const struct pernet_operations *ops;
544         struct net *net, *tmp, *last;
545         struct llist_node *net_kill_list;
546         LIST_HEAD(net_exit_list);
547
548         /* Atomically snapshot the list of namespaces to cleanup */
549         net_kill_list = llist_del_all(&cleanup_list);
550
551         down_read(&pernet_ops_rwsem);
552
553         /* Don't let anyone else find us. */
554         down_write(&net_rwsem);
555         llist_for_each_entry(net, net_kill_list, cleanup_list)
556                 list_del_rcu(&net->list);
557         /* Cache last net. After we unlock rtnl, no one new net
558          * added to net_namespace_list can assign nsid pointer
559          * to a net from net_kill_list (see peernet2id_alloc()).
560          * So, we skip them in unhash_nsid().
561          *
562          * Note, that unhash_nsid() does not delete nsid links
563          * between net_kill_list's nets, as they've already
564          * deleted from net_namespace_list. But, this would be
565          * useless anyway, as netns_ids are destroyed there.
566          */
567         last = list_last_entry(&net_namespace_list, struct net, list);
568         up_write(&net_rwsem);
569
570         llist_for_each_entry(net, net_kill_list, cleanup_list) {
571                 unhash_nsid(net, last);
572                 list_add_tail(&net->exit_list, &net_exit_list);
573         }
574
575         /* Run all of the network namespace pre_exit methods */
576         list_for_each_entry_reverse(ops, &pernet_list, list)
577                 ops_pre_exit_list(ops, &net_exit_list);
578
579         /*
580          * Another CPU might be rcu-iterating the list, wait for it.
581          * This needs to be before calling the exit() notifiers, so
582          * the rcu_barrier() below isn't sufficient alone.
583          * Also the pre_exit() and exit() methods need this barrier.
584          */
585         synchronize_rcu();
586
587         /* Run all of the network namespace exit methods */
588         list_for_each_entry_reverse(ops, &pernet_list, list)
589                 ops_exit_list(ops, &net_exit_list);
590
591         /* Free the net generic variables */
592         list_for_each_entry_reverse(ops, &pernet_list, list)
593                 ops_free_list(ops, &net_exit_list);
594
595         up_read(&pernet_ops_rwsem);
596
597         /* Ensure there are no outstanding rcu callbacks using this
598          * network namespace.
599          */
600         rcu_barrier();
601
602         /* Finally it is safe to free my network namespace structure */
603         list_for_each_entry_safe(net, tmp, &net_exit_list, exit_list) {
604                 list_del_init(&net->exit_list);
605                 dec_net_namespaces(net->ucounts);
606                 key_remove_domain(net->key_domain);
607                 put_user_ns(net->user_ns);
608                 net_drop_ns(net);
609         }
610 }
611
612 /**
613  * net_ns_barrier - wait until concurrent net_cleanup_work is done
614  *
615  * cleanup_net runs from work queue and will first remove namespaces
616  * from the global list, then run net exit functions.
617  *
618  * Call this in module exit path to make sure that all netns
619  * ->exit ops have been invoked before the function is removed.
620  */
621 void net_ns_barrier(void)
622 {
623         down_write(&pernet_ops_rwsem);
624         up_write(&pernet_ops_rwsem);
625 }
626 EXPORT_SYMBOL(net_ns_barrier);
627
628 static DECLARE_WORK(net_cleanup_work, cleanup_net);
629
630 void __put_net(struct net *net)
631 {
632         /* Cleanup the network namespace in process context */
633         if (llist_add(&net->cleanup_list, &cleanup_list))
634                 queue_work(netns_wq, &net_cleanup_work);
635 }
636 EXPORT_SYMBOL_GPL(__put_net);
637
638 struct net *get_net_ns_by_fd(int fd)
639 {
640         struct file *file;
641         struct ns_common *ns;
642         struct net *net;
643
644         file = proc_ns_fget(fd);
645         if (IS_ERR(file))
646                 return ERR_CAST(file);
647
648         ns = get_proc_ns(file_inode(file));
649         if (ns->ops == &netns_operations)
650                 net = get_net(container_of(ns, struct net, ns));
651         else
652                 net = ERR_PTR(-EINVAL);
653
654         fput(file);
655         return net;
656 }
657
658 #else
659 struct net *get_net_ns_by_fd(int fd)
660 {
661         return ERR_PTR(-EINVAL);
662 }
663 #endif
664 EXPORT_SYMBOL_GPL(get_net_ns_by_fd);
665
666 struct net *get_net_ns_by_pid(pid_t pid)
667 {
668         struct task_struct *tsk;
669         struct net *net;
670
671         /* Lookup the network namespace */
672         net = ERR_PTR(-ESRCH);
673         rcu_read_lock();
674         tsk = find_task_by_vpid(pid);
675         if (tsk) {
676                 struct nsproxy *nsproxy;
677                 task_lock(tsk);
678                 nsproxy = tsk->nsproxy;
679                 if (nsproxy)
680                         net = get_net(nsproxy->net_ns);
681                 task_unlock(tsk);
682         }
683         rcu_read_unlock();
684         return net;
685 }
686 EXPORT_SYMBOL_GPL(get_net_ns_by_pid);
687
688 static __net_init int net_ns_net_init(struct net *net)
689 {
690 #ifdef CONFIG_NET_NS
691         net->ns.ops = &netns_operations;
692 #endif
693         return ns_alloc_inum(&net->ns);
694 }
695
696 static __net_exit void net_ns_net_exit(struct net *net)
697 {
698         ns_free_inum(&net->ns);
699 }
700
701 static struct pernet_operations __net_initdata net_ns_ops = {
702         .init = net_ns_net_init,
703         .exit = net_ns_net_exit,
704 };
705
706 static const struct nla_policy rtnl_net_policy[NETNSA_MAX + 1] = {
707         [NETNSA_NONE]           = { .type = NLA_UNSPEC },
708         [NETNSA_NSID]           = { .type = NLA_S32 },
709         [NETNSA_PID]            = { .type = NLA_U32 },
710         [NETNSA_FD]             = { .type = NLA_U32 },
711         [NETNSA_TARGET_NSID]    = { .type = NLA_S32 },
712 };
713
714 static int rtnl_net_newid(struct sk_buff *skb, struct nlmsghdr *nlh,
715                           struct netlink_ext_ack *extack)
716 {
717         struct net *net = sock_net(skb->sk);
718         struct nlattr *tb[NETNSA_MAX + 1];
719         struct nlattr *nla;
720         struct net *peer;
721         int nsid, err;
722
723         err = nlmsg_parse_deprecated(nlh, sizeof(struct rtgenmsg), tb,
724                                      NETNSA_MAX, rtnl_net_policy, extack);
725         if (err < 0)
726                 return err;
727         if (!tb[NETNSA_NSID]) {
728                 NL_SET_ERR_MSG(extack, "nsid is missing");
729                 return -EINVAL;
730         }
731         nsid = nla_get_s32(tb[NETNSA_NSID]);
732
733         if (tb[NETNSA_PID]) {
734                 peer = get_net_ns_by_pid(nla_get_u32(tb[NETNSA_PID]));
735                 nla = tb[NETNSA_PID];
736         } else if (tb[NETNSA_FD]) {
737                 peer = get_net_ns_by_fd(nla_get_u32(tb[NETNSA_FD]));
738                 nla = tb[NETNSA_FD];
739         } else {
740                 NL_SET_ERR_MSG(extack, "Peer netns reference is missing");
741                 return -EINVAL;
742         }
743         if (IS_ERR(peer)) {
744                 NL_SET_BAD_ATTR(extack, nla);
745                 NL_SET_ERR_MSG(extack, "Peer netns reference is invalid");
746                 return PTR_ERR(peer);
747         }
748
749         spin_lock(&net->nsid_lock);
750         if (__peernet2id(net, peer) >= 0) {
751                 spin_unlock(&net->nsid_lock);
752                 err = -EEXIST;
753                 NL_SET_BAD_ATTR(extack, nla);
754                 NL_SET_ERR_MSG(extack,
755                                "Peer netns already has a nsid assigned");
756                 goto out;
757         }
758
759         err = alloc_netid(net, peer, nsid);
760         spin_unlock(&net->nsid_lock);
761         if (err >= 0) {
762                 rtnl_net_notifyid(net, RTM_NEWNSID, err, NETLINK_CB(skb).portid,
763                                   nlh, GFP_KERNEL);
764                 err = 0;
765         } else if (err == -ENOSPC && nsid >= 0) {
766                 err = -EEXIST;
767                 NL_SET_BAD_ATTR(extack, tb[NETNSA_NSID]);
768                 NL_SET_ERR_MSG(extack, "The specified nsid is already used");
769         }
770 out:
771         put_net(peer);
772         return err;
773 }
774
775 static int rtnl_net_get_size(void)
776 {
777         return NLMSG_ALIGN(sizeof(struct rtgenmsg))
778                + nla_total_size(sizeof(s32)) /* NETNSA_NSID */
779                + nla_total_size(sizeof(s32)) /* NETNSA_CURRENT_NSID */
780                ;
781 }
782
783 struct net_fill_args {
784         u32 portid;
785         u32 seq;
786         int flags;
787         int cmd;
788         int nsid;
789         bool add_ref;
790         int ref_nsid;
791 };
792
793 static int rtnl_net_fill(struct sk_buff *skb, struct net_fill_args *args)
794 {
795         struct nlmsghdr *nlh;
796         struct rtgenmsg *rth;
797
798         nlh = nlmsg_put(skb, args->portid, args->seq, args->cmd, sizeof(*rth),
799                         args->flags);
800         if (!nlh)
801                 return -EMSGSIZE;
802
803         rth = nlmsg_data(nlh);
804         rth->rtgen_family = AF_UNSPEC;
805
806         if (nla_put_s32(skb, NETNSA_NSID, args->nsid))
807                 goto nla_put_failure;
808
809         if (args->add_ref &&
810             nla_put_s32(skb, NETNSA_CURRENT_NSID, args->ref_nsid))
811                 goto nla_put_failure;
812
813         nlmsg_end(skb, nlh);
814         return 0;
815
816 nla_put_failure:
817         nlmsg_cancel(skb, nlh);
818         return -EMSGSIZE;
819 }
820
821 static int rtnl_net_valid_getid_req(struct sk_buff *skb,
822                                     const struct nlmsghdr *nlh,
823                                     struct nlattr **tb,
824                                     struct netlink_ext_ack *extack)
825 {
826         int i, err;
827
828         if (!netlink_strict_get_check(skb))
829                 return nlmsg_parse_deprecated(nlh, sizeof(struct rtgenmsg),
830                                               tb, NETNSA_MAX, rtnl_net_policy,
831                                               extack);
832
833         err = nlmsg_parse_deprecated_strict(nlh, sizeof(struct rtgenmsg), tb,
834                                             NETNSA_MAX, rtnl_net_policy,
835                                             extack);
836         if (err)
837                 return err;
838
839         for (i = 0; i <= NETNSA_MAX; i++) {
840                 if (!tb[i])
841                         continue;
842
843                 switch (i) {
844                 case NETNSA_PID:
845                 case NETNSA_FD:
846                 case NETNSA_NSID:
847                 case NETNSA_TARGET_NSID:
848                         break;
849                 default:
850                         NL_SET_ERR_MSG(extack, "Unsupported attribute in peer netns getid request");
851                         return -EINVAL;
852                 }
853         }
854
855         return 0;
856 }
857
858 static int rtnl_net_getid(struct sk_buff *skb, struct nlmsghdr *nlh,
859                           struct netlink_ext_ack *extack)
860 {
861         struct net *net = sock_net(skb->sk);
862         struct nlattr *tb[NETNSA_MAX + 1];
863         struct net_fill_args fillargs = {
864                 .portid = NETLINK_CB(skb).portid,
865                 .seq = nlh->nlmsg_seq,
866                 .cmd = RTM_NEWNSID,
867         };
868         struct net *peer, *target = net;
869         struct nlattr *nla;
870         struct sk_buff *msg;
871         int err;
872
873         err = rtnl_net_valid_getid_req(skb, nlh, tb, extack);
874         if (err < 0)
875                 return err;
876         if (tb[NETNSA_PID]) {
877                 peer = get_net_ns_by_pid(nla_get_u32(tb[NETNSA_PID]));
878                 nla = tb[NETNSA_PID];
879         } else if (tb[NETNSA_FD]) {
880                 peer = get_net_ns_by_fd(nla_get_u32(tb[NETNSA_FD]));
881                 nla = tb[NETNSA_FD];
882         } else if (tb[NETNSA_NSID]) {
883                 peer = get_net_ns_by_id(net, nla_get_s32(tb[NETNSA_NSID]));
884                 if (!peer)
885                         peer = ERR_PTR(-ENOENT);
886                 nla = tb[NETNSA_NSID];
887         } else {
888                 NL_SET_ERR_MSG(extack, "Peer netns reference is missing");
889                 return -EINVAL;
890         }
891
892         if (IS_ERR(peer)) {
893                 NL_SET_BAD_ATTR(extack, nla);
894                 NL_SET_ERR_MSG(extack, "Peer netns reference is invalid");
895                 return PTR_ERR(peer);
896         }
897
898         if (tb[NETNSA_TARGET_NSID]) {
899                 int id = nla_get_s32(tb[NETNSA_TARGET_NSID]);
900
901                 target = rtnl_get_net_ns_capable(NETLINK_CB(skb).sk, id);
902                 if (IS_ERR(target)) {
903                         NL_SET_BAD_ATTR(extack, tb[NETNSA_TARGET_NSID]);
904                         NL_SET_ERR_MSG(extack,
905                                        "Target netns reference is invalid");
906                         err = PTR_ERR(target);
907                         goto out;
908                 }
909                 fillargs.add_ref = true;
910                 fillargs.ref_nsid = peernet2id(net, peer);
911         }
912
913         msg = nlmsg_new(rtnl_net_get_size(), GFP_KERNEL);
914         if (!msg) {
915                 err = -ENOMEM;
916                 goto out;
917         }
918
919         fillargs.nsid = peernet2id(target, peer);
920         err = rtnl_net_fill(msg, &fillargs);
921         if (err < 0)
922                 goto err_out;
923
924         err = rtnl_unicast(msg, net, NETLINK_CB(skb).portid);
925         goto out;
926
927 err_out:
928         nlmsg_free(msg);
929 out:
930         if (fillargs.add_ref)
931                 put_net(target);
932         put_net(peer);
933         return err;
934 }
935
936 struct rtnl_net_dump_cb {
937         struct net *tgt_net;
938         struct net *ref_net;
939         struct sk_buff *skb;
940         struct net_fill_args fillargs;
941         int idx;
942         int s_idx;
943 };
944
945 /* Runs in RCU-critical section. */
946 static int rtnl_net_dumpid_one(int id, void *peer, void *data)
947 {
948         struct rtnl_net_dump_cb *net_cb = (struct rtnl_net_dump_cb *)data;
949         int ret;
950
951         if (net_cb->idx < net_cb->s_idx)
952                 goto cont;
953
954         net_cb->fillargs.nsid = id;
955         if (net_cb->fillargs.add_ref)
956                 net_cb->fillargs.ref_nsid = __peernet2id(net_cb->ref_net, peer);
957         ret = rtnl_net_fill(net_cb->skb, &net_cb->fillargs);
958         if (ret < 0)
959                 return ret;
960
961 cont:
962         net_cb->idx++;
963         return 0;
964 }
965
966 static int rtnl_valid_dump_net_req(const struct nlmsghdr *nlh, struct sock *sk,
967                                    struct rtnl_net_dump_cb *net_cb,
968                                    struct netlink_callback *cb)
969 {
970         struct netlink_ext_ack *extack = cb->extack;
971         struct nlattr *tb[NETNSA_MAX + 1];
972         int err, i;
973
974         err = nlmsg_parse_deprecated_strict(nlh, sizeof(struct rtgenmsg), tb,
975                                             NETNSA_MAX, rtnl_net_policy,
976                                             extack);
977         if (err < 0)
978                 return err;
979
980         for (i = 0; i <= NETNSA_MAX; i++) {
981                 if (!tb[i])
982                         continue;
983
984                 if (i == NETNSA_TARGET_NSID) {
985                         struct net *net;
986
987                         net = rtnl_get_net_ns_capable(sk, nla_get_s32(tb[i]));
988                         if (IS_ERR(net)) {
989                                 NL_SET_BAD_ATTR(extack, tb[i]);
990                                 NL_SET_ERR_MSG(extack,
991                                                "Invalid target network namespace id");
992                                 return PTR_ERR(net);
993                         }
994                         net_cb->fillargs.add_ref = true;
995                         net_cb->ref_net = net_cb->tgt_net;
996                         net_cb->tgt_net = net;
997                 } else {
998                         NL_SET_BAD_ATTR(extack, tb[i]);
999                         NL_SET_ERR_MSG(extack,
1000                                        "Unsupported attribute in dump request");
1001                         return -EINVAL;
1002                 }
1003         }
1004
1005         return 0;
1006 }
1007
1008 static int rtnl_net_dumpid(struct sk_buff *skb, struct netlink_callback *cb)
1009 {
1010         struct rtnl_net_dump_cb net_cb = {
1011                 .tgt_net = sock_net(skb->sk),
1012                 .skb = skb,
1013                 .fillargs = {
1014                         .portid = NETLINK_CB(cb->skb).portid,
1015                         .seq = cb->nlh->nlmsg_seq,
1016                         .flags = NLM_F_MULTI,
1017                         .cmd = RTM_NEWNSID,
1018                 },
1019                 .idx = 0,
1020                 .s_idx = cb->args[0],
1021         };
1022         int err = 0;
1023
1024         if (cb->strict_check) {
1025                 err = rtnl_valid_dump_net_req(cb->nlh, skb->sk, &net_cb, cb);
1026                 if (err < 0)
1027                         goto end;
1028         }
1029
1030         rcu_read_lock();
1031         idr_for_each(&net_cb.tgt_net->netns_ids, rtnl_net_dumpid_one, &net_cb);
1032         rcu_read_unlock();
1033
1034         cb->args[0] = net_cb.idx;
1035 end:
1036         if (net_cb.fillargs.add_ref)
1037                 put_net(net_cb.tgt_net);
1038         return err < 0 ? err : skb->len;
1039 }
1040
1041 static void rtnl_net_notifyid(struct net *net, int cmd, int id, u32 portid,
1042                               struct nlmsghdr *nlh, gfp_t gfp)
1043 {
1044         struct net_fill_args fillargs = {
1045                 .portid = portid,
1046                 .seq = nlh ? nlh->nlmsg_seq : 0,
1047                 .cmd = cmd,
1048                 .nsid = id,
1049         };
1050         struct sk_buff *msg;
1051         int err = -ENOMEM;
1052
1053         msg = nlmsg_new(rtnl_net_get_size(), gfp);
1054         if (!msg)
1055                 goto out;
1056
1057         err = rtnl_net_fill(msg, &fillargs);
1058         if (err < 0)
1059                 goto err_out;
1060
1061         rtnl_notify(msg, net, portid, RTNLGRP_NSID, nlh, gfp);
1062         return;
1063
1064 err_out:
1065         nlmsg_free(msg);
1066 out:
1067         rtnl_set_sk_err(net, RTNLGRP_NSID, err);
1068 }
1069
1070 static int __init net_ns_init(void)
1071 {
1072         struct net_generic *ng;
1073
1074 #ifdef CONFIG_NET_NS
1075         net_cachep = kmem_cache_create("net_namespace", sizeof(struct net),
1076                                         SMP_CACHE_BYTES,
1077                                         SLAB_PANIC|SLAB_ACCOUNT, NULL);
1078
1079         /* Create workqueue for cleanup */
1080         netns_wq = create_singlethread_workqueue("netns");
1081         if (!netns_wq)
1082                 panic("Could not create netns workq");
1083 #endif
1084
1085         ng = net_alloc_generic();
1086         if (!ng)
1087                 panic("Could not allocate generic netns");
1088
1089         rcu_assign_pointer(init_net.gen, ng);
1090
1091         down_write(&pernet_ops_rwsem);
1092         if (setup_net(&init_net, &init_user_ns))
1093                 panic("Could not setup the initial network namespace");
1094
1095         init_net_initialized = true;
1096         up_write(&pernet_ops_rwsem);
1097
1098         if (register_pernet_subsys(&net_ns_ops))
1099                 panic("Could not register network namespace subsystems");
1100
1101         rtnl_register(PF_UNSPEC, RTM_NEWNSID, rtnl_net_newid, NULL,
1102                       RTNL_FLAG_DOIT_UNLOCKED);
1103         rtnl_register(PF_UNSPEC, RTM_GETNSID, rtnl_net_getid, rtnl_net_dumpid,
1104                       RTNL_FLAG_DOIT_UNLOCKED);
1105
1106         return 0;
1107 }
1108
1109 pure_initcall(net_ns_init);
1110
1111 #ifdef CONFIG_NET_NS
1112 static int __register_pernet_operations(struct list_head *list,
1113                                         struct pernet_operations *ops)
1114 {
1115         struct net *net;
1116         int error;
1117         LIST_HEAD(net_exit_list);
1118
1119         list_add_tail(&ops->list, list);
1120         if (ops->init || (ops->id && ops->size)) {
1121                 /* We held write locked pernet_ops_rwsem, and parallel
1122                  * setup_net() and cleanup_net() are not possible.
1123                  */
1124                 for_each_net(net) {
1125                         error = ops_init(ops, net);
1126                         if (error)
1127                                 goto out_undo;
1128                         list_add_tail(&net->exit_list, &net_exit_list);
1129                 }
1130         }
1131         return 0;
1132
1133 out_undo:
1134         /* If I have an error cleanup all namespaces I initialized */
1135         list_del(&ops->list);
1136         ops_pre_exit_list(ops, &net_exit_list);
1137         synchronize_rcu();
1138         ops_exit_list(ops, &net_exit_list);
1139         ops_free_list(ops, &net_exit_list);
1140         return error;
1141 }
1142
1143 static void __unregister_pernet_operations(struct pernet_operations *ops)
1144 {
1145         struct net *net;
1146         LIST_HEAD(net_exit_list);
1147
1148         list_del(&ops->list);
1149         /* See comment in __register_pernet_operations() */
1150         for_each_net(net)
1151                 list_add_tail(&net->exit_list, &net_exit_list);
1152         ops_pre_exit_list(ops, &net_exit_list);
1153         synchronize_rcu();
1154         ops_exit_list(ops, &net_exit_list);
1155         ops_free_list(ops, &net_exit_list);
1156 }
1157
1158 #else
1159
1160 static int __register_pernet_operations(struct list_head *list,
1161                                         struct pernet_operations *ops)
1162 {
1163         if (!init_net_initialized) {
1164                 list_add_tail(&ops->list, list);
1165                 return 0;
1166         }
1167
1168         return ops_init(ops, &init_net);
1169 }
1170
1171 static void __unregister_pernet_operations(struct pernet_operations *ops)
1172 {
1173         if (!init_net_initialized) {
1174                 list_del(&ops->list);
1175         } else {
1176                 LIST_HEAD(net_exit_list);
1177                 list_add(&init_net.exit_list, &net_exit_list);
1178                 ops_pre_exit_list(ops, &net_exit_list);
1179                 synchronize_rcu();
1180                 ops_exit_list(ops, &net_exit_list);
1181                 ops_free_list(ops, &net_exit_list);
1182         }
1183 }
1184
1185 #endif /* CONFIG_NET_NS */
1186
1187 static DEFINE_IDA(net_generic_ids);
1188
1189 static int register_pernet_operations(struct list_head *list,
1190                                       struct pernet_operations *ops)
1191 {
1192         int error;
1193
1194         if (ops->id) {
1195                 error = ida_alloc_min(&net_generic_ids, MIN_PERNET_OPS_ID,
1196                                 GFP_KERNEL);
1197                 if (error < 0)
1198                         return error;
1199                 *ops->id = error;
1200                 max_gen_ptrs = max(max_gen_ptrs, *ops->id + 1);
1201         }
1202         error = __register_pernet_operations(list, ops);
1203         if (error) {
1204                 rcu_barrier();
1205                 if (ops->id)
1206                         ida_free(&net_generic_ids, *ops->id);
1207         }
1208
1209         return error;
1210 }
1211
1212 static void unregister_pernet_operations(struct pernet_operations *ops)
1213 {
1214         __unregister_pernet_operations(ops);
1215         rcu_barrier();
1216         if (ops->id)
1217                 ida_free(&net_generic_ids, *ops->id);
1218 }
1219
1220 /**
1221  *      register_pernet_subsys - register a network namespace subsystem
1222  *      @ops:  pernet operations structure for the subsystem
1223  *
1224  *      Register a subsystem which has init and exit functions
1225  *      that are called when network namespaces are created and
1226  *      destroyed respectively.
1227  *
1228  *      When registered all network namespace init functions are
1229  *      called for every existing network namespace.  Allowing kernel
1230  *      modules to have a race free view of the set of network namespaces.
1231  *
1232  *      When a new network namespace is created all of the init
1233  *      methods are called in the order in which they were registered.
1234  *
1235  *      When a network namespace is destroyed all of the exit methods
1236  *      are called in the reverse of the order with which they were
1237  *      registered.
1238  */
1239 int register_pernet_subsys(struct pernet_operations *ops)
1240 {
1241         int error;
1242         down_write(&pernet_ops_rwsem);
1243         error =  register_pernet_operations(first_device, ops);
1244         up_write(&pernet_ops_rwsem);
1245         return error;
1246 }
1247 EXPORT_SYMBOL_GPL(register_pernet_subsys);
1248
1249 /**
1250  *      unregister_pernet_subsys - unregister a network namespace subsystem
1251  *      @ops: pernet operations structure to manipulate
1252  *
1253  *      Remove the pernet operations structure from the list to be
1254  *      used when network namespaces are created or destroyed.  In
1255  *      addition run the exit method for all existing network
1256  *      namespaces.
1257  */
1258 void unregister_pernet_subsys(struct pernet_operations *ops)
1259 {
1260         down_write(&pernet_ops_rwsem);
1261         unregister_pernet_operations(ops);
1262         up_write(&pernet_ops_rwsem);
1263 }
1264 EXPORT_SYMBOL_GPL(unregister_pernet_subsys);
1265
1266 /**
1267  *      register_pernet_device - register a network namespace device
1268  *      @ops:  pernet operations structure for the subsystem
1269  *
1270  *      Register a device which has init and exit functions
1271  *      that are called when network namespaces are created and
1272  *      destroyed respectively.
1273  *
1274  *      When registered all network namespace init functions are
1275  *      called for every existing network namespace.  Allowing kernel
1276  *      modules to have a race free view of the set of network namespaces.
1277  *
1278  *      When a new network namespace is created all of the init
1279  *      methods are called in the order in which they were registered.
1280  *
1281  *      When a network namespace is destroyed all of the exit methods
1282  *      are called in the reverse of the order with which they were
1283  *      registered.
1284  */
1285 int register_pernet_device(struct pernet_operations *ops)
1286 {
1287         int error;
1288         down_write(&pernet_ops_rwsem);
1289         error = register_pernet_operations(&pernet_list, ops);
1290         if (!error && (first_device == &pernet_list))
1291                 first_device = &ops->list;
1292         up_write(&pernet_ops_rwsem);
1293         return error;
1294 }
1295 EXPORT_SYMBOL_GPL(register_pernet_device);
1296
1297 /**
1298  *      unregister_pernet_device - unregister a network namespace netdevice
1299  *      @ops: pernet operations structure to manipulate
1300  *
1301  *      Remove the pernet operations structure from the list to be
1302  *      used when network namespaces are created or destroyed.  In
1303  *      addition run the exit method for all existing network
1304  *      namespaces.
1305  */
1306 void unregister_pernet_device(struct pernet_operations *ops)
1307 {
1308         down_write(&pernet_ops_rwsem);
1309         if (&ops->list == first_device)
1310                 first_device = first_device->next;
1311         unregister_pernet_operations(ops);
1312         up_write(&pernet_ops_rwsem);
1313 }
1314 EXPORT_SYMBOL_GPL(unregister_pernet_device);
1315
1316 #ifdef CONFIG_NET_NS
1317 static struct ns_common *netns_get(struct task_struct *task)
1318 {
1319         struct net *net = NULL;
1320         struct nsproxy *nsproxy;
1321
1322         task_lock(task);
1323         nsproxy = task->nsproxy;
1324         if (nsproxy)
1325                 net = get_net(nsproxy->net_ns);
1326         task_unlock(task);
1327
1328         return net ? &net->ns : NULL;
1329 }
1330
1331 static inline struct net *to_net_ns(struct ns_common *ns)
1332 {
1333         return container_of(ns, struct net, ns);
1334 }
1335
1336 static void netns_put(struct ns_common *ns)
1337 {
1338         put_net(to_net_ns(ns));
1339 }
1340
1341 static int netns_install(struct nsproxy *nsproxy, struct ns_common *ns)
1342 {
1343         struct net *net = to_net_ns(ns);
1344
1345         if (!ns_capable(net->user_ns, CAP_SYS_ADMIN) ||
1346             !ns_capable(current_user_ns(), CAP_SYS_ADMIN))
1347                 return -EPERM;
1348
1349         put_net(nsproxy->net_ns);
1350         nsproxy->net_ns = get_net(net);
1351         return 0;
1352 }
1353
1354 static struct user_namespace *netns_owner(struct ns_common *ns)
1355 {
1356         return to_net_ns(ns)->user_ns;
1357 }
1358
1359 const struct proc_ns_operations netns_operations = {
1360         .name           = "net",
1361         .type           = CLONE_NEWNET,
1362         .get            = netns_get,
1363         .put            = netns_put,
1364         .install        = netns_install,
1365         .owner          = netns_owner,
1366 };
1367 #endif