blk-mq: Free tags in blk_mq_init_tags() upon error

[linux-2.6-microblaze.git] / block / blk-mq-tag.c

diff --git a/block/blk-mq-tag.c b/block/blk-mq-tag.c
index 32d82e2..c2c9f36 100644 (file)
--- a/block/blk-mq-tag.c
+++ b/block/blk-mq-tag.c
@@ -429,24 +429,22 @@ static int bt_alloc(struct sbitmap_queue *bt, unsigned int depth,
                                       node);
 }
 
-static struct blk_mq_tags *blk_mq_init_bitmap_tags(struct blk_mq_tags *tags,
-                                                  int node, int alloc_policy)
+static int blk_mq_init_bitmap_tags(struct blk_mq_tags *tags,
+                                  int node, int alloc_policy)
 {
        unsigned int depth = tags->nr_tags - tags->nr_reserved_tags;
        bool round_robin = alloc_policy == BLK_TAG_ALLOC_RR;
 
        if (bt_alloc(&tags->bitmap_tags, depth, round_robin, node))
-               goto free_tags;
+               return -ENOMEM;
        if (bt_alloc(&tags->breserved_tags, tags->nr_reserved_tags, round_robin,
                     node))
                goto free_bitmap_tags;
 
-       return tags;
+       return 0;
 free_bitmap_tags:
        sbitmap_queue_free(&tags->bitmap_tags);
-free_tags:
-       kfree(tags);
-       return NULL;
+       return -ENOMEM;
 }
 
 struct blk_mq_tags *blk_mq_init_tags(unsigned int total_tags,
@@ -467,7 +465,11 @@ struct blk_mq_tags *blk_mq_init_tags(unsigned int total_tags,
        tags->nr_tags = total_tags;
        tags->nr_reserved_tags = reserved_tags;
 
-       return blk_mq_init_bitmap_tags(tags, node, alloc_policy);
+       if (blk_mq_init_bitmap_tags(tags, node, alloc_policy) < 0) {
+               kfree(tags);
+               return NULL;
+       }
+       return tags;
 }
 
 void blk_mq_free_tags(struct blk_mq_tags *tags)