disk_part_iter_exit(&piter);
 }
 
+void put_disk_devt(struct disk_devt *disk_devt)
+{
+       if (disk_devt && atomic_dec_and_test(&disk_devt->count))
+               disk_devt->release(disk_devt);
+}
+EXPORT_SYMBOL(put_disk_devt);
+
+void get_disk_devt(struct disk_devt *disk_devt)
+{
+       if (disk_devt)
+               atomic_inc(&disk_devt->count);
+}
+EXPORT_SYMBOL(get_disk_devt);
+
 /**
  * device_add_disk - add partitioning information to kernel list
  * @parent: parent device for the disk
 
        disk_alloc_events(disk);
 
+       /*
+        * Take a reference on the devt and assign it to queue since it
+        * must not be reallocated while the bdi is registered
+        */
+       disk->queue->disk_devt = disk->disk_devt;
+       get_disk_devt(disk->disk_devt);
+
        /* Register BDI before referencing it from bdev */
        bdi = disk->queue->backing_dev_info;
        bdi_register_owner(bdi, disk_to_dev(disk));
 
        put_device(&sdkp->dev);
 }
 
+struct sd_devt {
+       int idx;
+       struct disk_devt disk_devt;
+};
+
+void sd_devt_release(struct disk_devt *disk_devt)
+{
+       struct sd_devt *sd_devt = container_of(disk_devt, struct sd_devt,
+                       disk_devt);
+
+       spin_lock(&sd_index_lock);
+       ida_remove(&sd_index_ida, sd_devt->idx);
+       spin_unlock(&sd_index_lock);
+
+       kfree(sd_devt);
+}
+
 /**
  *     sd_probe - called during driver initialization and whenever a
  *     new scsi device is attached to the system. It is called once
 static int sd_probe(struct device *dev)
 {
        struct scsi_device *sdp = to_scsi_device(dev);
+       struct sd_devt *sd_devt;
        struct scsi_disk *sdkp;
        struct gendisk *gd;
        int index;
        if (!sdkp)
                goto out;
 
+       sd_devt = kzalloc(sizeof(*sd_devt), GFP_KERNEL);
+       if (!sd_devt)
+               goto out_free;
+
        gd = alloc_disk(SD_MINORS);
        if (!gd)
-               goto out_free;
+               goto out_free_devt;
 
        do {
                if (!ida_pre_get(&sd_index_ida, GFP_KERNEL))
                goto out_put;
        }
 
+       atomic_set(&sd_devt->disk_devt.count, 1);
+       sd_devt->disk_devt.release = sd_devt_release;
+       sd_devt->idx = index;
+       gd->disk_devt = &sd_devt->disk_devt;
+
        error = sd_format_disk_name("sd", index, gd->disk_name, DISK_NAME_LEN);
        if (error) {
                sdev_printk(KERN_WARNING, sdp, "SCSI disk (sd) name length exceeded.\n");
        return 0;
 
  out_free_index:
-       spin_lock(&sd_index_lock);
-       ida_remove(&sd_index_ida, index);
-       spin_unlock(&sd_index_lock);
+       put_disk_devt(&sd_devt->disk_devt);
+       sd_devt = NULL;
  out_put:
        put_disk(gd);
  out_free:
        kfree(sdkp);
+ out_free_devt:
+       kfree(sd_devt);
  out:
        scsi_autopm_put_device(sdp);
        return error;
        struct scsi_disk *sdkp = to_scsi_disk(dev);
        struct gendisk *disk = sdkp->disk;
        
-       spin_lock(&sd_index_lock);
-       ida_remove(&sd_index_ida, sdkp->index);
-       spin_unlock(&sd_index_lock);
-
+       put_disk_devt(disk->disk_devt);
        disk->private_data = NULL;
        put_disk(disk);
        put_device(&sdkp->device->sdev_gendev);
 
 };
 
 #endif /* CONFIG_BLK_DEV_INTEGRITY */
+struct disk_devt {
+       atomic_t count;
+       void (*release)(struct disk_devt *disk_devt);
+};
+
+void put_disk_devt(struct disk_devt *disk_devt);
+void get_disk_devt(struct disk_devt *disk_devt);
 
 struct gendisk {
        /* major, first_minor and minors are input parameters only,
        int first_minor;
        int minors;                     /* maximum number of minors, =1 for
                                          * disks that can't be partitioned. */
+       struct disk_devt *disk_devt;
 
        char disk_name[DISK_NAME_LEN];  /* name of major driver */
        char *(*devnode)(struct gendisk *gd, umode_t *mode);