Use flashprog_flash_getsize() where possible
Change-Id: I4eab12dc9dcf0c7fdffc5c662081868c6f32e36c
Signed-off-by: Nico Huber <nico.h@gmx.de>
Reviewed-on: https://review.sourcearcade.org/c/flashprog/+/493
diff --git a/sst_fwhub.c b/sst_fwhub.c
index 3e53080..cd1465d 100644
--- a/sst_fwhub.c
+++ b/sst_fwhub.c
@@ -69,7 +69,7 @@
{
unsigned int i;
- for (i = 0; i < flash->chip->total_size * 1024; i += flash->chip->page_size)
+ for (i = 0; i < flashprog_flash_getsize(flash); i += flash->chip->page_size)
check_sst_fwhub_block_lock(flash, i);
return 0;
@@ -80,7 +80,7 @@
unsigned int i;
int ret = 0;
- for (i = 0; i < flash->chip->total_size * 1024; i += flash->chip->page_size)
+ for (i = 0; i < flashprog_flash_getsize(flash); i += flash->chip->page_size)
{
if (clear_sst_fwhub_block_lock(flash, i))
{