net/core/net_namespace.c

   1 #include <linux/workqueue.h>
   2 #include <linux/rtnetlink.h>
   3 #include <linux/cache.h>
   4 #include <linux/slab.h>
   5 #include <linux/list.h>
   6 #include <linux/delay.h>
   7 #include <linux/sched.h>
   8 #include <linux/idr.h>
   9 #include <net/net_namespace.h>
  10 #include <net/netns/generic.h>
  11
  12 /*
  13  *      Our network namespace constructor/destructor lists
  14  */
  15
  16 static LIST_HEAD(pernet_list);
  17 static struct list_head *first_device = &pernet_list;
  18 static DEFINE_MUTEX(net_mutex);
  19
  20 LIST_HEAD(net_namespace_list);
  21 EXPORT_SYMBOL_GPL(net_namespace_list);
  22
  23 struct net init_net;
  24 EXPORT_SYMBOL(init_net);
  25
  26 #define INITIAL_NET_GEN_PTRS    13 /* +1 for len +2 for rcu_head */
  27
  28 /*
  29  * setup_net runs the initializers for the network namespace object.
  30  */
  31 static __net_init int setup_net(struct net *net)
  32 {
  33         /* Must be called with net_mutex held */
  34         struct pernet_operations *ops;
  35         int error;
  36         struct net_generic *ng;
  37
  38         atomic_set(&net->count, 1);
  39 #ifdef NETNS_REFCNT_DEBUG
  40         atomic_set(&net->use_count, 0);
  41 #endif
  42
  43         error = -ENOMEM;
  44         ng = kzalloc(sizeof(struct net_generic) +
  45                         INITIAL_NET_GEN_PTRS * sizeof(void *), GFP_KERNEL);
  46         if (ng == NULL)
  47                 goto out;
  48
  49         ng->len = INITIAL_NET_GEN_PTRS;
  50         rcu_assign_pointer(net->gen, ng);
  51
  52         error = 0;
  53         list_for_each_entry(ops, &pernet_list, list) {
  54                 if (ops->init) {
  55                         error = ops->init(net);
  56                         if (error < 0)
  57                                 goto out_undo;
  58                 }
  59         }
  60 out:
  61         return error;
  62
  63 out_undo:
  64         /* Walk through the list backwards calling the exit functions
  65          * for the pernet modules whose init functions did not fail.
  66          */
  67         list_for_each_entry_continue_reverse(ops, &pernet_list, list) {
  68                 if (ops->exit)
  69                         ops->exit(net);
  70         }
  71
  72         rcu_barrier();
  73         kfree(ng);
  74         goto out;
  75 }
  76
  77 #ifdef CONFIG_NET_NS
  78 static struct kmem_cache *net_cachep;
  79 static struct workqueue_struct *netns_wq;
  80
  81 static struct net *net_alloc(void)
  82 {
  83         return kmem_cache_zalloc(net_cachep, GFP_KERNEL);
  84 }
  85
  86 static void net_free(struct net *net)
  87 {
  88         if (!net)
  89                 return;
  90
  91 #ifdef NETNS_REFCNT_DEBUG
  92         if (unlikely(atomic_read(&net->use_count) != 0)) {
  93                 printk(KERN_EMERG "network namespace not free! Usage: %d\n",
  94                         atomic_read(&net->use_count));
  95                 return;
  96         }
  97 #endif
  98         kfree(net->gen);
  99         kmem_cache_free(net_cachep, net);
 100 }
 101
 102 struct net *copy_net_ns(unsigned long flags, struct net *old_net)
 103 {
 104         struct net *new_net = NULL;
 105         int err;
 106
 107         get_net(old_net);
 108
 109         if (!(flags & CLONE_NEWNET))
 110                 return old_net;
 111
 112         err = -ENOMEM;
 113         new_net = net_alloc();
 114         if (!new_net)
 115                 goto out;
 116
 117         mutex_lock(&net_mutex);
 118         err = setup_net(new_net);
 119         if (err)
 120                 goto out_unlock;
 121
 122         rtnl_lock();
 123         list_add_tail(&new_net->list, &net_namespace_list);
 124         rtnl_unlock();
 125
 126
 127 out_unlock:
 128         mutex_unlock(&net_mutex);
 129 out:
 130         put_net(old_net);
 131         if (err) {
 132                 net_free(new_net);
 133                 new_net = ERR_PTR(err);
 134         }
 135         return new_net;
 136 }
 137
 138 static void cleanup_net(struct work_struct *work)
 139 {
 140         struct pernet_operations *ops;
 141         struct net *net;
 142
 143         /* Be very certain incoming network packets will not find us */
 144         rcu_barrier();
 145
 146         net = container_of(work, struct net, work);
 147
 148         mutex_lock(&net_mutex);
 149
 150         /* Don't let anyone else find us. */
 151         rtnl_lock();
 152         list_del(&net->list);
 153         rtnl_unlock();
 154
 155         /* Run all of the network namespace exit methods */
 156         list_for_each_entry_reverse(ops, &pernet_list, list) {
 157                 if (ops->exit)
 158                         ops->exit(net);
 159         }
 160
 161         mutex_unlock(&net_mutex);
 162
 163         /* Ensure there are no outstanding rcu callbacks using this
 164          * network namespace.
 165          */
 166         rcu_barrier();
 167
 168         /* Finally it is safe to free my network namespace structure */
 169         net_free(net);
 170 }
 171
 172 void __put_net(struct net *net)
 173 {
 174         /* Cleanup the network namespace in process context */
 175         INIT_WORK(&net->work, cleanup_net);
 176         queue_work(netns_wq, &net->work);
 177 }
 178 EXPORT_SYMBOL_GPL(__put_net);
 179
 180 #else
 181 struct net *copy_net_ns(unsigned long flags, struct net *old_net)
 182 {
 183         if (flags & CLONE_NEWNET)
 184                 return ERR_PTR(-EINVAL);
 185         return old_net;
 186 }
 187 #endif
 188
 189 static int __init net_ns_init(void)
 190 {
 191         int err;
 192
 193         printk(KERN_INFO "net_namespace: %zd bytes\n", sizeof(struct net));
 194 #ifdef CONFIG_NET_NS
 195         net_cachep = kmem_cache_create("net_namespace", sizeof(struct net),
 196                                         SMP_CACHE_BYTES,
 197                                         SLAB_PANIC, NULL);
 198
 199         /* Create workqueue for cleanup */
 200         netns_wq = create_singlethread_workqueue("netns");
 201         if (!netns_wq)
 202                 panic("Could not create netns workq");
 203 #endif
 204
 205         mutex_lock(&net_mutex);
 206         err = setup_net(&init_net);
 207
 208         rtnl_lock();
 209         list_add_tail(&init_net.list, &net_namespace_list);
 210         rtnl_unlock();
 211
 212         mutex_unlock(&net_mutex);
 213         if (err)
 214                 panic("Could not setup the initial network namespace");
 215
 216         return 0;
 217 }
 218
 219 pure_initcall(net_ns_init);
 220
 221 #ifdef CONFIG_NET_NS
 222 static int register_pernet_operations(struct list_head *list,
 223                                       struct pernet_operations *ops)
 224 {
 225         struct net *net, *undo_net;
 226         int error;
 227
 228         list_add_tail(&ops->list, list);
 229         if (ops->init) {
 230                 for_each_net(net) {
 231                         error = ops->init(net);
 232                         if (error)
 233                                 goto out_undo;
 234                 }
 235         }
 236         return 0;
 237
 238 out_undo:
 239         /* If I have an error cleanup all namespaces I initialized */
 240         list_del(&ops->list);
 241         if (ops->exit) {
 242                 for_each_net(undo_net) {
 243                         if (undo_net == net)
 244                                 goto undone;
 245                         ops->exit(undo_net);
 246                 }
 247         }
 248 undone:
 249         return error;
 250 }
 251
 252 static void unregister_pernet_operations(struct pernet_operations *ops)
 253 {
 254         struct net *net;
 255
 256         list_del(&ops->list);
 257         if (ops->exit)
 258                 for_each_net(net)
 259                         ops->exit(net);
 260 }
 261
 262 #else
 263
 264 static int register_pernet_operations(struct list_head *list,
 265                                       struct pernet_operations *ops)
 266 {
 267         if (ops->init == NULL)
 268                 return 0;
 269         return ops->init(&init_net);
 270 }
 271
 272 static void unregister_pernet_operations(struct pernet_operations *ops)
 273 {
 274         if (ops->exit)
 275                 ops->exit(&init_net);
 276 }
 277 #endif
 278
 279 static DEFINE_IDA(net_generic_ids);
 280
 281 /**
 282  *      register_pernet_subsys - register a network namespace subsystem
 283  *      @ops:  pernet operations structure for the subsystem
 284  *
 285  *      Register a subsystem which has init and exit functions
 286  *      that are called when network namespaces are created and
 287  *      destroyed respectively.
 288  *
 289  *      When registered all network namespace init functions are
 290  *      called for every existing network namespace.  Allowing kernel
 291  *      modules to have a race free view of the set of network namespaces.
 292  *
 293  *      When a new network namespace is created all of the init
 294  *      methods are called in the order in which they were registered.
 295  *
 296  *      When a network namespace is destroyed all of the exit methods
 297  *      are called in the reverse of the order with which they were
 298  *      registered.
 299  */
 300 int register_pernet_subsys(struct pernet_operations *ops)
 301 {
 302         int error;
 303         mutex_lock(&net_mutex);
 304         error =  register_pernet_operations(first_device, ops);
 305         mutex_unlock(&net_mutex);
 306         return error;
 307 }
 308 EXPORT_SYMBOL_GPL(register_pernet_subsys);
 309
 310 /**
 311  *      unregister_pernet_subsys - unregister a network namespace subsystem
 312  *      @ops: pernet operations structure to manipulate
 313  *
 314  *      Remove the pernet operations structure from the list to be
 315  *      used when network namespaces are created or destroyed.  In
 316  *      addition run the exit method for all existing network
 317  *      namespaces.
 318  */
 319 void unregister_pernet_subsys(struct pernet_operations *module)
 320 {
 321         mutex_lock(&net_mutex);
 322         unregister_pernet_operations(module);
 323         mutex_unlock(&net_mutex);
 324 }
 325 EXPORT_SYMBOL_GPL(unregister_pernet_subsys);
 326
 327 int register_pernet_gen_subsys(int *id, struct pernet_operations *ops)
 328 {
 329         int rv;
 330
 331         mutex_lock(&net_mutex);
 332 again:
 333         rv = ida_get_new_above(&net_generic_ids, 1, id);
 334         if (rv < 0) {
 335                 if (rv == -EAGAIN) {
 336                         ida_pre_get(&net_generic_ids, GFP_KERNEL);
 337                         goto again;
 338                 }
 339                 goto out;
 340         }
 341         rv = register_pernet_operations(first_device, ops);
 342         if (rv < 0)
 343                 ida_remove(&net_generic_ids, *id);
 344         mutex_unlock(&net_mutex);
 345 out:
 346         return rv;
 347 }
 348 EXPORT_SYMBOL_GPL(register_pernet_gen_subsys);
 349
 350 void unregister_pernet_gen_subsys(int id, struct pernet_operations *ops)
 351 {
 352         mutex_lock(&net_mutex);
 353         unregister_pernet_operations(ops);
 354         ida_remove(&net_generic_ids, id);
 355         mutex_unlock(&net_mutex);
 356 }
 357 EXPORT_SYMBOL_GPL(unregister_pernet_gen_subsys);
 358
 359 /**
 360  *      register_pernet_device - register a network namespace device
 361  *      @ops:  pernet operations structure for the subsystem
 362  *
 363  *      Register a device which has init and exit functions
 364  *      that are called when network namespaces are created and
 365  *      destroyed respectively.
 366  *
 367  *      When registered all network namespace init functions are
 368  *      called for every existing network namespace.  Allowing kernel
 369  *      modules to have a race free view of the set of network namespaces.
 370  *
 371  *      When a new network namespace is created all of the init
 372  *      methods are called in the order in which they were registered.
 373  *
 374  *      When a network namespace is destroyed all of the exit methods
 375  *      are called in the reverse of the order with which they were
 376  *      registered.
 377  */
 378 int register_pernet_device(struct pernet_operations *ops)
 379 {
 380         int error;
 381         mutex_lock(&net_mutex);
 382         error = register_pernet_operations(&pernet_list, ops);
 383         if (!error && (first_device == &pernet_list))
 384                 first_device = &ops->list;
 385         mutex_unlock(&net_mutex);
 386         return error;
 387 }
 388 EXPORT_SYMBOL_GPL(register_pernet_device);
 389
 390 int register_pernet_gen_device(int *id, struct pernet_operations *ops)
 391 {
 392         int error;
 393         mutex_lock(&net_mutex);
 394 again:
 395         error = ida_get_new_above(&net_generic_ids, 1, id);
 396         if (error) {
 397                 if (error == -EAGAIN) {
 398                         ida_pre_get(&net_generic_ids, GFP_KERNEL);
 399                         goto again;
 400                 }
 401                 goto out;
 402         }
 403         error = register_pernet_operations(&pernet_list, ops);
 404         if (error)
 405                 ida_remove(&net_generic_ids, *id);
 406         else if (first_device == &pernet_list)
 407                 first_device = &ops->list;
 408 out:
 409         mutex_unlock(&net_mutex);
 410         return error;
 411 }
 412 EXPORT_SYMBOL_GPL(register_pernet_gen_device);
 413
 414 /**
 415  *      unregister_pernet_device - unregister a network namespace netdevice
 416  *      @ops: pernet operations structure to manipulate
 417  *
 418  *      Remove the pernet operations structure from the list to be
 419  *      used when network namespaces are created or destroyed.  In
 420  *      addition run the exit method for all existing network
 421  *      namespaces.
 422  */
 423 void unregister_pernet_device(struct pernet_operations *ops)
 424 {
 425         mutex_lock(&net_mutex);
 426         if (&ops->list == first_device)
 427                 first_device = first_device->next;
 428         unregister_pernet_operations(ops);
 429         mutex_unlock(&net_mutex);
 430 }
 431 EXPORT_SYMBOL_GPL(unregister_pernet_device);
 432
 433 void unregister_pernet_gen_device(int id, struct pernet_operations *ops)
 434 {
 435         mutex_lock(&net_mutex);
 436         if (&ops->list == first_device)
 437                 first_device = first_device->next;
 438         unregister_pernet_operations(ops);
 439         ida_remove(&net_generic_ids, id);
 440         mutex_unlock(&net_mutex);
 441 }
 442 EXPORT_SYMBOL_GPL(unregister_pernet_gen_device);
 443
 444 static void net_generic_release(struct rcu_head *rcu)
 445 {
 446         struct net_generic *ng;
 447
 448         ng = container_of(rcu, struct net_generic, rcu);
 449         kfree(ng);
 450 }
 451
 452 int net_assign_generic(struct net *net, int id, void *data)
 453 {
 454         struct net_generic *ng, *old_ng;
 455
 456         BUG_ON(!mutex_is_locked(&net_mutex));
 457         BUG_ON(id == 0);
 458
 459         ng = old_ng = net->gen;
 460         if (old_ng->len >= id)
 461                 goto assign;
 462
 463         ng = kzalloc(sizeof(struct net_generic) +
 464                         id * sizeof(void *), GFP_KERNEL);
 465         if (ng == NULL)
 466                 return -ENOMEM;
 467
 468         /*
 469          * Some synchronisation notes:
 470          *
 471          * The net_generic explores the net->gen array inside rcu
 472          * read section. Besides once set the net->gen->ptr[x]
 473          * pointer never changes (see rules in netns/generic.h).
 474          *
 475          * That said, we simply duplicate this array and schedule
 476          * the old copy for kfree after a grace period.
 477          */
 478
 479         ng->len = id;
 480         memcpy(&ng->ptr, &old_ng->ptr, old_ng->len);
 481
 482         rcu_assign_pointer(net->gen, ng);
 483         call_rcu(&old_ng->rcu, net_generic_release);
 484 assign:
 485         ng->ptr[id - 1] = data;
 486         return 0;
 487 }
 488 EXPORT_SYMBOL_GPL(net_assign_generic);