gfp_t gfp_mask)
 {
        struct nfs4_deviceid_node *node;
-       unsigned long start, end;
+       int err = -ENODEV;
 
 retry:
        node = nfs4_find_get_deviceid(server, id, cred, gfp_mask);
        if (!node)
                return ERR_PTR(-ENODEV);
 
-       if (test_bit(NFS_DEVICEID_UNAVAILABLE, &node->flags) == 0)
-               return node;
+       if (test_bit(NFS_DEVICEID_UNAVAILABLE, &node->flags)) {
+               unsigned long end = jiffies;
+               unsigned long start = end - PNFS_DEVICE_RETRY_TIMEOUT;
 
-       end = jiffies;
-       start = end - PNFS_DEVICE_RETRY_TIMEOUT;
-       if (!time_in_range(node->timestamp_unavailable, start, end)) {
-               nfs4_delete_deviceid(node->ld, node->nfs_client, id);
-               goto retry;
+               if (!time_in_range(node->timestamp_unavailable, start, end)) {
+                       nfs4_delete_deviceid(node->ld, node->nfs_client, id);
+                       goto retry;
+               }
+               goto out_put;
        }
 
+       if (!bl_register_dev(container_of(node, struct pnfs_block_dev, node)))
+               goto out_put;
+
+       return node;
+
+out_put:
        nfs4_put_deviceid_node(node);
-       return ERR_PTR(-ENODEV);
+       return ERR_PTR(err);
 }
 
 static int
 
        u64                             start;
        u64                             len;
 
+       enum pnfs_block_volume_type     type;
        u32                             nr_children;
        struct pnfs_block_dev           *children;
        u64                             chunk_size;
 
        struct file                     *bdev_file;
        u64                             disk_offset;
+       unsigned long                   flags;
 
        u64                             pr_key;
-       bool                            pr_registered;
 
        bool (*map)(struct pnfs_block_dev *dev, u64 offset,
                        struct pnfs_block_dev_map *map);
 };
 
+/* pnfs_block_dev flag bits */
+enum {
+       PNFS_BDEV_REGISTERED = 0,
+};
+
 /* sector_t fields are all in 512-byte sectors */
 struct pnfs_block_extent {
        union {
 #define BL_DEVICE_REQUEST_ERR          0x2 /* User level process fails */
 
 /* dev.c */
+bool bl_register_dev(struct pnfs_block_dev *d);
 struct nfs4_deviceid_node *bl_alloc_deviceid_node(struct nfs_server *server,
                struct pnfs_device *pdev, gfp_t gfp_mask);
 void bl_free_deviceid_node(struct nfs4_deviceid_node *d);
 
 
 #define NFSDBG_FACILITY                NFSDBG_PNFS_LD
 
+static void bl_unregister_scsi(struct pnfs_block_dev *dev)
+{
+       struct block_device *bdev = file_bdev(dev->bdev_file);
+       const struct pr_ops *ops = bdev->bd_disk->fops->pr_ops;
+
+       if (!test_and_clear_bit(PNFS_BDEV_REGISTERED, &dev->flags))
+               return;
+
+       if (ops->pr_register(bdev, dev->pr_key, 0, false))
+               pr_err("failed to unregister PR key.\n");
+}
+
+static bool bl_register_scsi(struct pnfs_block_dev *dev)
+{
+       struct block_device *bdev = file_bdev(dev->bdev_file);
+       const struct pr_ops *ops = bdev->bd_disk->fops->pr_ops;
+       int status;
+
+       if (test_and_set_bit(PNFS_BDEV_REGISTERED, &dev->flags))
+               return true;
+
+       status = ops->pr_register(bdev, 0, dev->pr_key, true);
+       if (status) {
+               pr_err("pNFS: failed to register key for block device %s.",
+                      bdev->bd_disk->disk_name);
+               return false;
+       }
+       return true;
+}
+
+static void bl_unregister_dev(struct pnfs_block_dev *dev)
+{
+       u32 i;
+
+       if (dev->nr_children) {
+               for (i = 0; i < dev->nr_children; i++)
+                       bl_unregister_dev(&dev->children[i]);
+               return;
+       }
+
+       if (dev->type == PNFS_BLOCK_VOLUME_SCSI)
+               bl_unregister_scsi(dev);
+}
+
+bool bl_register_dev(struct pnfs_block_dev *dev)
+{
+       u32 i;
+
+       if (dev->nr_children) {
+               for (i = 0; i < dev->nr_children; i++) {
+                       if (!bl_register_dev(&dev->children[i])) {
+                               while (i > 0)
+                                       bl_unregister_dev(&dev->children[--i]);
+                               return false;
+                       }
+               }
+               return true;
+       }
+
+       if (dev->type == PNFS_BLOCK_VOLUME_SCSI)
+               return bl_register_scsi(dev);
+       return true;
+}
+
 static void
 bl_free_device(struct pnfs_block_dev *dev)
 {
+       bl_unregister_dev(dev);
+
        if (dev->nr_children) {
                int i;
 
                        bl_free_device(&dev->children[i]);
                kfree(dev->children);
        } else {
-               if (dev->pr_registered) {
-                       const struct pr_ops *ops =
-                               file_bdev(dev->bdev_file)->bd_disk->fops->pr_ops;
-                       int error;
-
-                       error = ops->pr_register(file_bdev(dev->bdev_file),
-                               dev->pr_key, 0, false);
-                       if (error)
-                               pr_err("failed to unregister PR key.\n");
-               }
-
                if (dev->bdev_file)
                        fput(dev->bdev_file);
        }
                goto out_blkdev_put;
        }
 
-       error = ops->pr_register(file_bdev(d->bdev_file), 0, d->pr_key, true);
-       if (error) {
-               pr_err("pNFS: failed to register key for block device %s.",
-                               file_bdev(d->bdev_file)->bd_disk->disk_name);
-               goto out_blkdev_put;
-       }
-
-       d->pr_registered = true;
        return 0;
 
 out_blkdev_put:
 bl_parse_deviceid(struct nfs_server *server, struct pnfs_block_dev *d,
                struct pnfs_block_volume *volumes, int idx, gfp_t gfp_mask)
 {
-       switch (volumes[idx].type) {
+       d->type = volumes[idx].type;
+
+       switch (d->type) {
        case PNFS_BLOCK_VOLUME_SIMPLE:
                return bl_parse_simple(server, d, volumes, idx, gfp_mask);
        case PNFS_BLOCK_VOLUME_SLICE:
        case PNFS_BLOCK_VOLUME_SCSI:
                return bl_parse_scsi(server, d, volumes, idx, gfp_mask);
        default:
-               dprintk("unsupported volume type: %d\n", volumes[idx].type);
+               dprintk("unsupported volume type: %d\n", d->type);
                return -EIO;
        }
 }