#include <linux/netdevice.h>
 #include <linux/if_vlan.h>
 #include <net/net_namespace.h>
+#include <net/netns/generic.h>
 #include "vlanproc.h"
 #include "vlan.h"
 
  * Proc filesystem derectory entries.
  */
 
-/*
- *     /proc/net/vlan
- */
-
-static struct proc_dir_entry *proc_vlan_dir;
-
-/*
- *     /proc/net/vlan/config
- */
-
-static struct proc_dir_entry *proc_vlan_conf;
-
 /* Strings */
 static const char *vlan_name_type_str[VLAN_NAME_TYPE_HIGHEST] = {
     [VLAN_NAME_TYPE_RAW_PLUS_VID]        = "VLAN_NAME_TYPE_RAW_PLUS_VID",
 
 void vlan_proc_cleanup(struct net *net)
 {
-       if (net != &init_net)
-               return;
+       struct vlan_net *vn = net_generic(net, vlan_net_id);
 
-       if (proc_vlan_conf)
-               remove_proc_entry(name_conf, proc_vlan_dir);
+       if (vn->proc_vlan_conf)
+               remove_proc_entry(name_conf, vn->proc_vlan_dir);
 
-       if (proc_vlan_dir)
-               proc_net_remove(&init_net, name_root);
+       if (vn->proc_vlan_dir)
+               proc_net_remove(net, name_root);
 
        /* Dynamically added entries should be cleaned up as their vlan_device
         * is removed, so we should not have to take care of it here...
 
 int vlan_proc_init(struct net *net)
 {
-       if (net != &init_net)
-               return 0;
+       struct vlan_net *vn = net_generic(net, vlan_net_id);
 
-       proc_vlan_dir = proc_mkdir(name_root, init_net.proc_net);
-       if (!proc_vlan_dir)
+       vn->proc_vlan_dir = proc_net_mkdir(net, name_root, net->proc_net);
+       if (!vn->proc_vlan_dir)
                goto err;
 
-       proc_vlan_conf = proc_create(name_conf, S_IFREG|S_IRUSR|S_IWUSR,
-                                    proc_vlan_dir, &vlan_fops);
-       if (!proc_vlan_conf)
+       vn->proc_vlan_conf = proc_create(name_conf, S_IFREG|S_IRUSR|S_IWUSR,
+                                    vn->proc_vlan_dir, &vlan_fops);
+       if (!vn->proc_vlan_conf)
                goto err;
        return 0;
 
 int vlan_proc_add_dev(struct net_device *vlandev)
 {
        struct vlan_dev_info *dev_info = vlan_dev_info(vlandev);
+       struct vlan_net *vn = net_generic(dev_net(vlandev), vlan_net_id);
 
        dev_info->dent = proc_create(vlandev->name, S_IFREG|S_IRUSR|S_IWUSR,
-                                    proc_vlan_dir, &vlandev_fops);
+                                    vn->proc_vlan_dir, &vlandev_fops);
        if (!dev_info->dent)
                return -ENOBUFS;
 
  */
 int vlan_proc_rem_dev(struct net_device *vlandev)
 {
+       struct vlan_net *vn = net_generic(dev_net(vlandev), vlan_net_id);
+
        /** NOTE:  This will consume the memory pointed to by dent, it seems. */
        if (vlan_dev_info(vlandev)->dent) {
                remove_proc_entry(vlan_dev_info(vlandev)->dent->name,
-                                 proc_vlan_dir);
+                                 vn->proc_vlan_dir);
                vlan_dev_info(vlandev)->dent = NULL;
        }
        return 0;