struct mutex                    group_lock; /* locks group_list */
        struct ida                      group_ida;
        dev_t                           group_devt;
+       struct class                    *device_class;
+       struct ida                      device_ida;
 } vfio;
 
 struct vfio_iommu_driver {
  * VFIO driver API
  */
 /* Release helper called by vfio_put_device() */
-void vfio_device_release(struct kref *kref)
+static void vfio_device_release(struct device *dev)
 {
        struct vfio_device *device =
-                       container_of(kref, struct vfio_device, kref);
+                       container_of(dev, struct vfio_device, device);
 
        vfio_release_device_set(device);
+       ida_free(&vfio.device_ida, device->index);
 
        /*
         * kvfree() cannot be done here due to a life cycle mess in
         */
        device->ops->release(device);
 }
-EXPORT_SYMBOL_GPL(vfio_device_release);
 
 /*
  * Allocate and initialize vfio_device so it can be registered to vfio
 {
        int ret;
 
+       ret = ida_alloc_max(&vfio.device_ida, MINORMASK, GFP_KERNEL);
+       if (ret < 0) {
+               dev_dbg(dev, "Error to alloc index\n");
+               return ret;
+       }
+
+       device->index = ret;
        init_completion(&device->comp);
        device->dev = dev;
        device->ops = ops;
                        goto out_uninit;
        }
 
-       kref_init(&device->kref);
+       device_initialize(&device->device);
+       device->device.release = vfio_device_release;
+       device->device.class = vfio.device_class;
+       device->device.parent = device->dev;
        return 0;
 
 out_uninit:
        vfio_release_device_set(device);
+       ida_free(&vfio.device_ida, device->index);
        return ret;
 }
 EXPORT_SYMBOL_GPL(vfio_init_device);
                struct vfio_group *group)
 {
        struct vfio_device *existing_device;
+       int ret;
 
        if (IS_ERR(group))
                return PTR_ERR(group);
                dev_WARN(device->dev, "Device already exists on group %d\n",
                         iommu_group_id(group->iommu_group));
                vfio_device_put_registration(existing_device);
-               if (group->type == VFIO_NO_IOMMU ||
-                   group->type == VFIO_EMULATED_IOMMU)
-                       iommu_group_remove_device(device->dev);
-               vfio_group_put(group);
-               return -EBUSY;
+               ret = -EBUSY;
+               goto err_out;
        }
 
        /* Our reference on group is moved to the device */
        device->group = group;
 
+       ret = dev_set_name(&device->device, "vfio%d", device->index);
+       if (ret)
+               goto err_out;
+
+       ret = device_add(&device->device);
+       if (ret)
+               goto err_out;
+
        /* Refcounting can't start until the driver calls register */
        refcount_set(&device->refcount, 1);
 
        mutex_unlock(&group->device_lock);
 
        return 0;
+err_out:
+       if (group->type == VFIO_NO_IOMMU ||
+           group->type == VFIO_EMULATED_IOMMU)
+               iommu_group_remove_device(device->dev);
+       vfio_group_put(group);
+       return ret;
 }
 
 int vfio_register_group_dev(struct vfio_device *device)
        list_del(&device->group_next);
        mutex_unlock(&group->device_lock);
 
+       /* Balances device_add in register path */
+       device_del(&device->device);
+
        if (group->type == VFIO_NO_IOMMU || group->type == VFIO_EMULATED_IOMMU)
                iommu_group_remove_device(device->dev);
 
        int ret;
 
        ida_init(&vfio.group_ida);
+       ida_init(&vfio.device_ida);
        mutex_init(&vfio.group_lock);
        mutex_init(&vfio.iommu_drivers_lock);
        INIT_LIST_HEAD(&vfio.group_list);
        vfio.class = class_create(THIS_MODULE, "vfio");
        if (IS_ERR(vfio.class)) {
                ret = PTR_ERR(vfio.class);
-               goto err_class;
+               goto err_group_class;
        }
 
        vfio.class->devnode = vfio_devnode;
 
+       /* /sys/class/vfio-dev/vfioX */
+       vfio.device_class = class_create(THIS_MODULE, "vfio-dev");
+       if (IS_ERR(vfio.device_class)) {
+               ret = PTR_ERR(vfio.device_class);
+               goto err_dev_class;
+       }
+
        ret = alloc_chrdev_region(&vfio.group_devt, 0, MINORMASK + 1, "vfio");
        if (ret)
                goto err_alloc_chrdev;
 err_driver_register:
        unregister_chrdev_region(vfio.group_devt, MINORMASK + 1);
 err_alloc_chrdev:
+       class_destroy(vfio.device_class);
+       vfio.device_class = NULL;
+err_dev_class:
        class_destroy(vfio.class);
        vfio.class = NULL;
-err_class:
+err_group_class:
        misc_deregister(&vfio_dev);
        return ret;
 }
 #ifdef CONFIG_VFIO_NOIOMMU
        vfio_unregister_iommu_driver(&vfio_noiommu_ops);
 #endif
+       ida_destroy(&vfio.device_ida);
        ida_destroy(&vfio.group_ida);
        unregister_chrdev_region(vfio.group_devt, MINORMASK + 1);
+       class_destroy(vfio.device_class);
+       vfio.device_class = NULL;
        class_destroy(vfio.class);
        vfio.class = NULL;
        misc_deregister(&vfio_dev);