From f1837d84c4829bf929051bf38cbc0620c68ba33d Mon Sep 17 00:00:00 2001
From: wangyanwen <wangyanwen@nucleisys.com>
Date: Mon, 2 Dec 2024 14:48:31 +0800
Subject: [PATCH] src/flash/nor: fix custom flash driver when bank->size is 0

Change-Id: I1ef6cd6dad81a62902cf72372bc11bb0437bfeb6
Signed-off-by: wangyanwen <wangyanwen@nucleisys.com>
---
 src/flash/nor/custom.c | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/flash/nor/custom.c b/src/flash/nor/custom.c
index 0da76ce83..da0e5f83f 100644
--- a/src/flash/nor/custom.c
+++ b/src/flash/nor/custom.c
@@ -388,6 +388,8 @@ static int custom_probe(struct flash_bank *bank)
 			break;
 		}
 	}
+	if (bank->size == 0)
+		bank->size = bank_msg->dev->size_in_bytes;
 	/* if no sectors, treat whole bank as single sector */
 	if (0 == bank_msg->sectorsize) {
 		bank_msg->sectorsize = bank_msg->dev->sectorsize ?