[PATCH v5 2/5] mtd: ubi: use 'max_bad_blocks' to compute bad_peb_limit if available

[Zach Brown]

From: Jeff Westfahl <jeff.westf...@ni.com>

Use the MTD function 'max_bad_blocks' to compute the UBI bad_peb_limit,
if the function is implemented for an MTD and doesn't return an error.

Signed-off-by: Jeff Westfahl <jeff.westf...@ni.com>
Signed-off-by: Zach Brown <zach.br...@ni.com>
Acked-by: Boris Brezillon <boris.brezil...@free-electron.com>
---
 drivers/mtd/ubi/build.c | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/drivers/mtd/ubi/build.c b/drivers/mtd/ubi/build.c
index 85d54f3..e9940a9 100644
--- a/drivers/mtd/ubi/build.c
+++ b/drivers/mtd/ubi/build.c
@@ -584,6 +584,10 @@ static int get_bad_peb_limit(const struct ubi_device *ubi, 
int max_beb_per1024)
        int limit, device_pebs;
        uint64_t device_size;
 
+       limit = mtd_max_bad_blocks(ubi->mtd, 0, ubi->mtd->size);
+       if (limit > 0)
+               return limit;
+
        if (!max_beb_per1024)
                return 0;
 
-- 
2.7.4