diff mbox series

libmtd: avoid divide by zero

Message ID 20201126224108.2325-1-chris.packham@alliedtelesis.co.nz
State Superseded
Headers show
Series libmtd: avoid divide by zero | expand

Commit Message

Chris Packham Nov. 26, 2020, 10:41 p.m. UTC
Some mtd-ram devices report 0 for the erase block size. Avoid a divide
by zero when calculating the erase block count for such devices.

Signed-off-by: Chris Packham <chris.packham@alliedtelesis.co.nz>
---
 lib/libmtd.c | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

Comments

Richard Weinberger Nov. 26, 2020, 11:43 p.m. UTC | #1
On Thu, Nov 26, 2020 at 11:48 PM Chris Packham
<chris.packham@alliedtelesis.co.nz> wrote:
> Some mtd-ram devices report 0 for the erase block size. Avoid a divide
> by zero when calculating the erase block count for such devices.
>
> Signed-off-by: Chris Packham <chris.packham@alliedtelesis.co.nz>
> ---
>  lib/libmtd.c | 3 ++-
>  1 file changed, 2 insertions(+), 1 deletion(-)
>
> diff --git a/lib/libmtd.c b/lib/libmtd.c
> index 9d8d0e8..8aa1cb1 100644
> --- a/lib/libmtd.c
> +++ b/lib/libmtd.c
> @@ -791,7 +791,8 @@ int mtd_get_dev_info1(libmtd_t desc, int mtd_num, struct mtd_dev_info *mtd)
>                 return -1;
>         mtd->writable = !!(ret & MTD_WRITEABLE);
>
> -       mtd->eb_cnt = mtd->size / mtd->eb_size;
> +       if (mtd->eb_size)
> +               mtd->eb_cnt = mtd->size / mtd->eb_size;

I think it is better to check for MTD_NO_ERASE.
->eb_size is only allowed to be zero if MTD_NO_ERASE is set.
In that case we can configure mtd->eb_cnt to 1.
diff mbox series

Patch

diff --git a/lib/libmtd.c b/lib/libmtd.c
index 9d8d0e8..8aa1cb1 100644
--- a/lib/libmtd.c
+++ b/lib/libmtd.c
@@ -791,7 +791,8 @@  int mtd_get_dev_info1(libmtd_t desc, int mtd_num, struct mtd_dev_info *mtd)
 		return -1;
 	mtd->writable = !!(ret & MTD_WRITEABLE);
 
-	mtd->eb_cnt = mtd->size / mtd->eb_size;
+	if (mtd->eb_size)
+		mtd->eb_cnt = mtd->size / mtd->eb_size;
 	mtd->type = type_str2int(mtd->type_str);
 	mtd->bb_allowed = !!(mtd->type == MTD_NANDFLASH ||
 				mtd->type == MTD_MLCNANDFLASH);