Generalize KVStore phase 1/2 test BlockDevice sizes

Author: Matthew Macovsky

features/storage/TESTS/kvstore/general_tests_phase_1/main.cpp

@@ -66,9 +66,6 @@ static const char *kv_prefix[] = {"TDB_", "FS_", "SEC_"};
 
 static int kv_setup = TDBStoreSet;
 
-static const size_t ul_bd_size = 16 * 4096;
-static const size_t rbp_bd_size = 8 * 4096;
-
 static const int heap_alloc_threshold_size = 4096;
 
 /*----------------initialization------------------*/
@@ -77,6 +74,8 @@ static const int heap_alloc_threshold_size = 4096;
 static void kvstore_init()
 {
     int res;
+    size_t erase_size, ul_bd_size, rbp_bd_size;
+    BlockDevice *sec_bd;
 
     res = bd->init();
     TEST_ASSERT_EQUAL_ERROR_CODE(0, res);
@@ -105,14 +104,19 @@ static void kvstore_init()
 
 #if SECURESTORE_ENABLED
     if (kv_setup == SecStoreSet) {
+        sec_bd = bd;
         if (erase_val == -1) {
             flash_bd = new FlashSimBlockDevice(bd);
-            ul_bd = new SlicingBlockDevice(flash_bd, 0, ul_bd_size);
-            rbp_bd = new SlicingBlockDevice(flash_bd, ul_bd_size, ul_bd_size + rbp_bd_size);
-        } else {
-            ul_bd = new SlicingBlockDevice(bd, 0, ul_bd_size);
-            rbp_bd = new SlicingBlockDevice(bd, ul_bd_size, ul_bd_size + rbp_bd_size);
+            sec_bd = flash_bd;
         }
+
+        erase_size  = sec_bd->get_erase_size();
+        ul_bd_size  = erase_size * 4;
+        rbp_bd_size = erase_size * 2;
+
+        ul_bd = new SlicingBlockDevice(sec_bd, 0, ul_bd_size);
+        rbp_bd = new SlicingBlockDevice(sec_bd, ul_bd_size, ul_bd_size + rbp_bd_size);
+
         TDBStore *ul_kv = new TDBStore(ul_bd);
         TDBStore *rbp_kv = new TDBStore(rbp_bd);
         kvstore = new SecureStore(ul_kv, rbp_kv);

features/storage/TESTS/kvstore/general_tests_phase_2/main.cpp

@@ -66,9 +66,6 @@ static const char *kv_prefix[] = {"TDB_", "FS_", "SEC_"};
 
 static int kv_setup = TDBStoreSet;
 
-static const size_t ul_bd_size = 16 * 4096;
-static const size_t rbp_bd_size = 8 * 4096;
-
 static const int heap_alloc_threshold_size = 4096;
 
 /*----------------initialization------------------*/
@@ -77,6 +74,8 @@ static const int heap_alloc_threshold_size = 4096;
 static void kvstore_init()
 {
     int res;
+    size_t erase_size, ul_bd_size, rbp_bd_size;
+    BlockDevice *sec_bd;
 
     res = bd->init();
     TEST_ASSERT_EQUAL_ERROR_CODE(0, res);
@@ -105,14 +104,19 @@ static void kvstore_init()
 
 #if SECURESTORE_ENABLED
     if (kv_setup == SecStoreSet) {
+        sec_bd = bd;
         if (erase_val == -1) {
             flash_bd = new FlashSimBlockDevice(bd);
-            ul_bd = new SlicingBlockDevice(flash_bd, 0, ul_bd_size);
-            rbp_bd = new SlicingBlockDevice(flash_bd, ul_bd_size, ul_bd_size + rbp_bd_size);
-        } else {
-            ul_bd = new SlicingBlockDevice(bd, 0, ul_bd_size);
-            rbp_bd = new SlicingBlockDevice(bd, ul_bd_size, ul_bd_size + rbp_bd_size);
+            sec_bd = flash_bd;
         }
+
+        erase_size  = sec_bd->get_erase_size();
+        ul_bd_size  = erase_size * 4;
+        rbp_bd_size = erase_size * 2;
+
+        ul_bd = new SlicingBlockDevice(sec_bd, 0, ul_bd_size);
+        rbp_bd = new SlicingBlockDevice(sec_bd, ul_bd_size, ul_bd_size + rbp_bd_size);
+
         TDBStore *ul_kv = new TDBStore(ul_bd);
         TDBStore *rbp_kv = new TDBStore(rbp_bd);
         kvstore = new SecureStore(ul_kv, rbp_kv);