From 969d8872a5d93311fec7c28888ce438897cadbed Mon Sep 17 00:00:00 2001
From: Priit Laes <plaes@plaes.org>
Date: Tue, 30 Apr 2024 16:50:07 +0300
Subject: [PATCH] flash: Return flash capacity per-chosen device

Previous value was hardcoded to return nrf52840's flash size.
Fixes #246.
---
 nrf-softdevice/src/flash.rs | 13 ++++++++++++-
 1 file changed, 12 insertions(+), 1 deletion(-)

diff --git a/nrf-softdevice/src/flash.rs b/nrf-softdevice/src/flash.rs
index 6e7886b4..5425df48 100644
--- a/nrf-softdevice/src/flash.rs
+++ b/nrf-softdevice/src/flash.rs
@@ -85,7 +85,18 @@ impl ReadNorFlash for Flash {
     }
 
     fn capacity(&self) -> usize {
-        256 * 4096
+        #[cfg(any(feature = "nrf52805", feature = "nrf52810", feature = "nrf52811",))]
+        return 48 * 4096; // 192KB
+
+        #[cfg(feature = "nrf52820")]
+        return 64 * 4096; // 256KB
+
+        // TODO: nrf52832 has also "lite" edition (QFAB) with 256KB/32KB of flash/RAM
+        #[cfg(any(feature = "nrf52832", feature = "nrf52833",))]
+        return 128 * 4096; // 512KB
+
+        #[cfg(feature = "nrf52840")]
+        return 256 * 4096; // 1024KB
     }
 }