};
 
 struct tun_struct {
-       struct list_head        list;
        unsigned int            flags;
        int                     attached;
        uid_t                   owner;
 
 /* Network device part of the driver */
 
-static int tun_net_id;
-struct tun_net {
-       struct list_head dev_list;
-};
-
 static const struct ethtool_ops tun_ethtool_ops;
 
 /* Net device open. */
        dev->features |= NETIF_F_NETNS_LOCAL;
 }
 
-static struct tun_struct *tun_get_by_name(struct tun_net *tn, const char *name)
-{
-       struct tun_struct *tun;
-
-       ASSERT_RTNL();
-       list_for_each_entry(tun, &tn->dev_list, list) {
-               if (!strncmp(tun->dev->name, name, IFNAMSIZ))
-                   return tun;
-       }
-
-       return NULL;
-}
-
 static int tun_set_iff(struct net *net, struct file *file, struct ifreq *ifr)
 {
-       struct tun_net *tn;
        struct tun_struct *tun;
        struct net_device *dev;
        const struct cred *cred = current_cred();
        int err;
 
-       tn = net_generic(net, tun_net_id);
-       tun = tun_get_by_name(tn, ifr->ifr_name);
-       if (tun) {
+       dev = __dev_get_by_name(net, ifr->ifr_name);
+       if (dev) {
+               if ((ifr->ifr_flags & IFF_TUN) && dev->netdev_ops == &tun_netdev_ops)
+                       tun = netdev_priv(dev);
+               else if ((ifr->ifr_flags & IFF_TAP) && dev->netdev_ops == &tap_netdev_ops)
+                       tun = netdev_priv(dev);
+               else
+                       return -EINVAL;
+
                if (tun->attached)
                        return -EBUSY;
 
                        return -EPERM;
                }
        }
-       else if (__dev_get_by_name(net, ifr->ifr_name))
-               return -EINVAL;
        else {
                char *name;
                unsigned long flags = 0;
                err = register_netdevice(tun->dev);
                if (err < 0)
                        goto err_free_dev;
-
-               list_add(&tun->list, &tn->dev_list);
        }
 
        DBG(KERN_INFO "%s: tun_set_iff\n", tun->dev->name);
        /* Drop read queue */
        skb_queue_purge(&tun->readq);
 
-       if (!(tun->flags & TUN_PERSIST)) {
-               list_del(&tun->list);
+       if (!(tun->flags & TUN_PERSIST))
                unregister_netdevice(tun->dev);
-       }
 
        rtnl_unlock();
 
 
 static int tun_init_net(struct net *net)
 {
-       struct tun_net *tn;
-
-       tn = kmalloc(sizeof(*tn), GFP_KERNEL);
-       if (tn == NULL)
-               return -ENOMEM;
-
-       INIT_LIST_HEAD(&tn->dev_list);
-
-       if (net_assign_generic(net, tun_net_id, tn)) {
-               kfree(tn);
-               return -ENOMEM;
-       }
-
        return 0;
 }
 
 static void tun_exit_net(struct net *net)
 {
-       struct tun_net *tn;
-       struct tun_struct *tun, *nxt;
-
-       tn = net_generic(net, tun_net_id);
+       struct net_device *dev, *next;
 
        rtnl_lock();
-       list_for_each_entry_safe(tun, nxt, &tn->dev_list, list) {
-               DBG(KERN_INFO "%s cleaned up\n", tun->dev->name);
-               unregister_netdevice(tun->dev);
+       for_each_netdev_safe(net, dev, next) {
+               if (dev->ethtool_ops != &tun_ethtool_ops)
+                       continue;
+               DBG(KERN_INFO "%s cleaned up\n", dev->name);
+               unregister_netdevice(dev);
        }
        rtnl_unlock();
-
-       kfree(tn);
 }
 
 static struct pernet_operations tun_net_ops = {
        printk(KERN_INFO "tun: %s, %s\n", DRV_DESCRIPTION, DRV_VERSION);
        printk(KERN_INFO "tun: %s\n", DRV_COPYRIGHT);
 
-       ret = register_pernet_gen_device(&tun_net_id, &tun_net_ops);
+       ret = register_pernet_device(&tun_net_ops);
        if (ret) {
                printk(KERN_ERR "tun: Can't register pernet ops\n");
                goto err_pernet;
        return 0;
 
 err_misc:
-       unregister_pernet_gen_device(tun_net_id, &tun_net_ops);
+       unregister_pernet_device(&tun_net_ops);
 err_pernet:
        return ret;
 }
 static void tun_cleanup(void)
 {
        misc_deregister(&tun_miscdev);
-       unregister_pernet_gen_device(tun_net_id, &tun_net_ops);
+       unregister_pernet_device(&tun_net_ops);
 }
 
 module_init(tun_init);