1 files changed, 5 insertions, 0 deletions

diff --git a/drivers/nvme/host/core.c b/drivers/nvme/host/core.c
index a3771c5729f5..2cbc378bc0d6 100644
--- a/drivers/nvme/host/core.c
+++ b/drivers/nvme/host/core.c
@@ -99,6 +99,7 @@ static struct class *nvme_subsys_class;
 
 static void nvme_ns_remove(struct nvme_ns *ns);
 static int nvme_revalidate_disk(struct gendisk *disk);
+static void nvme_put_subsystem(struct nvme_subsystem *subsys);
 
 int nvme_reset_ctrl(struct nvme_ctrl *ctrl)
 {
@@ -350,6 +351,7 @@ static void nvme_free_ns_head(struct kref *ref)
         ida_simple_remove(&head->subsys->ns_ida, head->instance);
         list_del_init(&head->entry);
         cleanup_srcu_struct(&head->srcu);
+        nvme_put_subsystem(head->subsys);
         kfree(head);
 }
 
@@ -2861,6 +2863,9 @@ static struct nvme_ns_head *nvme_alloc_ns_head(struct nvme_ctrl *ctrl,
                 goto out_cleanup_srcu;
 
         list_add_tail(&head->entry, &ctrl->subsys->nsheads);
+
+        kref_get(&ctrl->subsys->ref);
+
         return head;
 out_cleanup_srcu:
         cleanup_srcu_struct(&head->srcu);