From a899f35562dfc277bcb666d167967ce1e047251c Mon Sep 17 00:00:00 2001
From: "S.J.R. van Schaik" <stephan@whiteboxsystems.nl>
Date: Fri, 19 May 2017 15:09:25 +0200
Subject: [PATCH] flash: add flash_get_capacity()

---
 include/flash.h     | 2 ++
 source/core/flash.c | 8 ++++++++
 2 files changed, 10 insertions(+)

diff --git a/include/flash.h b/include/flash.h
index 093813e..f06e66e 100644
--- a/include/flash.h
+++ b/include/flash.h
@@ -7,6 +7,7 @@ struct flash_dev;
 
 struct flash_ops {
 	size_t (* get_size)(struct flash_dev *dev);
+	size_t (* get_capacity)(struct flash_dev *dev);
 	int (* read)(struct flash_dev *dev, uint32_t addr, void *data, size_t len);
 	int (* write)(struct flash_dev *dev, uint32_t addr, const void *data,
 		size_t len);
@@ -22,6 +23,7 @@ struct flash_dev {
 struct flash_dev *flash_probe(void);
 void flash_release(struct flash_dev *dev);
 size_t flash_get_size(struct flash_dev *dev);
+size_t flash_get_capacity(struct flash_dev *dev);
 int flash_read(struct flash_dev *dev, uint32_t addr, void *data, size_t len);
 int flash_write(struct flash_dev *dev, uint32_t addr, const void *data,
 	size_t len);
diff --git a/source/core/flash.c b/source/core/flash.c
index bee69a0..e6f7273 100644
--- a/source/core/flash.c
+++ b/source/core/flash.c
@@ -11,6 +11,14 @@ size_t flash_get_size(struct flash_dev *dev)
 	return dev->ops->get_size(dev);
 }
 
+size_t flash_get_capacity(struct flash_dev *dev)
+{
+	if (!dev)
+		return 0;
+
+	return dev->ops->get_capacity(dev);
+}
+
 int flash_read(struct flash_dev *dev, uint32_t addr, void *data, size_t len)
 {
 	if (!dev)