block/partitions: optimize memory allocation in check_partition()

[firefly-linux-kernel-4.4.55.git] / block / partition-generic.c

diff --git a/block/partition-generic.c b/block/partition-generic.c
index 1cb4deca1324a0afd1d8239e6d564d5d3c8be1ec..789cdea05893bb8e3420ede5d1aa65e7af408e1b 100644 (file)
--- a/block/partition-generic.c
+++ b/block/partition-generic.c
@@ -418,7 +418,7 @@ int rescan_partitions(struct gendisk *disk, struct block_device *bdev)
        int p, highest, res;
 rescan:
        if (state && !IS_ERR(state)) {
-               kfree(state);
+               free_partitions(state);
                state = NULL;
        }
 
@@ -525,7 +525,7 @@ rescan:
                        md_autodetect_dev(part_to_dev(part)->devt);
 #endif
        }
-       kfree(state);
+       free_partitions(state);
        return 0;
 }