drivers: flash: stm32 flash drivers supports the get_size API function

Implement the get_size flash device API function for the various stm32
This is just CONFIG_FLASH_SIZE expressed in Bytes or the FLASH_SIZE
given by the LL driver

Signed-off-by: Francois Ramu <francois.ramu@st.com>

Author: FRASTM

drivers/flash/flash_stm32.c

@@ -19,6 +19,7 @@
 #include <soc.h>
 #include <stm32_ll_bus.h>
 #include <stm32_ll_rcc.h>
+#include <stm32_ll_utils.h>
 #include <zephyr/logging/log.h>
 
 #include "flash_stm32.h"
@@ -354,6 +355,16 @@ flash_stm32_get_parameters(const struct device *dev)
 	return &flash_stm32_parameters;
 }
 
+/* Gives the total logical device size in bytes and return 0. */
+static int flash_stm32_get_size(const struct device *dev, uint64_t *size)
+{
+	ARG_UNUSED(dev);
+
+	*size = (uint64_t)LL_GetFlashSize() * 1024U;
+
+	return 0;
+}
+
 static struct flash_stm32_priv flash_data = {
 	.regs = (FLASH_TypeDef *) DT_INST_REG_ADDR(0),
 	/* Getting clocks information from device tree description depending
@@ -372,6 +383,7 @@ static DEVICE_API(flash, flash_stm32_api) = {
 	.write = flash_stm32_write,
 	.read = flash_stm32_read,
 	.get_parameters = flash_stm32_get_parameters,
+	.get_size = flash_stm32_get_size,
 #ifdef CONFIG_FLASH_PAGE_LAYOUT
 	.page_layout = flash_stm32_page_layout,
 #endif

drivers/flash/flash_stm32h7x.c

@@ -836,6 +836,16 @@ static const struct flash_parameters *flash_stm32h7_get_parameters(const struct
 	return &flash_stm32h7_parameters;
 }
 
+/* Gives the total logical device size in bytes and return 0. */
+static int flash_stm32h7_get_size(const struct device *dev, uint64_t *size)
+{
+	ARG_UNUSED(dev);
+
+	*size = (uint64_t)LL_GetFlashSize() * 1024U;
+
+	return 0;
+}
+
 void flash_stm32_page_layout(const struct device *dev, const struct flash_pages_layout **layout,
 			     size_t *layout_size)
 {
@@ -893,6 +903,7 @@ static DEVICE_API(flash, flash_stm32h7_api) = {
 	.write = flash_stm32h7_write,
 	.read = flash_stm32h7_read,
 	.get_parameters = flash_stm32h7_get_parameters,
+	.get_size = flash_stm32h7_get_size,
 #ifdef CONFIG_FLASH_PAGE_LAYOUT
 	.page_layout = flash_stm32_page_layout,
 #endif

drivers/flash/flash_stm32wb0x.c

@@ -22,6 +22,7 @@
 #include <stm32_ll_bus.h>
 #include <stm32_ll_rcc.h>
 #include <stm32_ll_system.h>
+#include <stm32_ll_utils.h>
 
 #include <zephyr/logging/log.h>
 LOG_MODULE_REGISTER(flash_stm32wb0x, CONFIG_FLASH_LOG_LEVEL);
@@ -400,6 +401,8 @@ int flash_wb0x_erase(const struct device *dev, off_t offset, size_t size)
 const struct flash_parameters *flash_wb0x_get_parameters(
 					const struct device *dev)
 {
+	ARG_UNUSED(dev);
+
 	static const struct flash_parameters fp = {
 		.write_block_size = WRITE_BLOCK_SIZE,
 		.erase_value = 0xff,
@@ -408,6 +411,16 @@ const struct flash_parameters *flash_wb0x_get_parameters(
 	return &fp;
 }
 
+/* Gives the total logical device size in bytes and return 0. */
+static int flash_wb0x_get_size(const struct device *dev, uint64_t *size)
+{
+	ARG_UNUSED(dev);
+
+	*size = (uint64_t)LL_GetFlashSize() * 1024U;
+
+	return 0;
+}
+
 #if defined(CONFIG_FLASH_PAGE_LAYOUT)
 void flash_wb0x_pages_layout(const struct device *dev,
 				const struct flash_pages_layout **layout,
@@ -432,6 +445,7 @@ static DEVICE_API(flash, flash_wb0x_api) = {
 	.write = flash_wb0x_write,
 	.read = flash_wb0x_read,
 	.get_parameters = flash_wb0x_get_parameters,
+	.get_size = flash_wb0x_get_size,
 #ifdef CONFIG_FLASH_PAGE_LAYOUT
 	.page_layout = flash_wb0x_pages_layout,
 #endif

drivers/flash/flash_stm32wba_fm.c

@@ -18,6 +18,8 @@ LOG_MODULE_REGISTER(flash_stm32wba, CONFIG_FLASH_LOG_LEVEL);
 #include "flash_manager.h"
 #include "flash_driver.h"
 
+#include <stm32_ll_utils.h>
+
 /* Let's wait for double the max erase time to be sure that the operation is
  * completed.
  */
@@ -163,6 +165,16 @@ static const struct flash_parameters *
 	return &flash_stm32_parameters;
 }
 
+/* Gives the total logical device size in bytes and return 0. */
+static int flash_stm32h7_get_size(const struct device *dev, uint64_t *size)
+{
+	ARG_UNUSED(dev);
+
+	*size = (uint64_t)LL_GetFlashSize() * 1024U;
+
+	return 0;
+}
+
 static struct flash_stm32_priv flash_data = {
 	.regs = (FLASH_TypeDef *) DT_INST_REG_ADDR(0),
 };
@@ -192,6 +204,7 @@ static DEVICE_API(flash, flash_stm32_api) = {
 	.write = flash_stm32_write,
 	.read = flash_stm32_read,
 	.get_parameters = flash_stm32_get_parameters,
+	.get_size = flash_stm32_get_size,
 #ifdef CONFIG_FLASH_PAGE_LAYOUT
 	.page_layout = flash_stm32wba_page_layout,
 #endif