test_nvs.c 19 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519
  1. #include <stdio.h>
  2. #include <ctype.h>
  3. #include <errno.h>
  4. #include <stdlib.h>
  5. #include <time.h>
  6. #include "unity.h"
  7. #include "nvs.h"
  8. #include "nvs_flash.h"
  9. #include "esp_partition.h"
  10. #include "esp_flash_encrypt.h"
  11. #include "esp_log.h"
  12. #include <string.h>
  13. #include "esp_system.h"
  14. #ifdef CONFIG_NVS_ENCRYPTION
  15. #include "mbedtls/aes.h"
  16. #endif
  17. static const char* TAG = "test_nvs";
  18. TEST_CASE("flash erase deinitializes initialized partition", "[nvs]")
  19. {
  20. nvs_handle_t handle;
  21. esp_err_t err = nvs_flash_init();
  22. if (err == ESP_ERR_NVS_NO_FREE_PAGES || err == ESP_ERR_NVS_NEW_VERSION_FOUND) {
  23. nvs_flash_erase();
  24. err = nvs_flash_init();
  25. }
  26. ESP_ERROR_CHECK( err );
  27. TEST_ESP_OK(nvs_flash_init());
  28. TEST_ESP_OK(nvs_open("uninit_ns", NVS_READWRITE, &handle));
  29. nvs_close(handle);
  30. TEST_ESP_OK(nvs_flash_erase());
  31. // exptected: no partition is initialized since nvs_flash_erase() deinitialized the partition again
  32. TEST_ESP_ERR(ESP_ERR_NVS_NOT_INITIALIZED, nvs_open("uninit_ns", NVS_READWRITE, &handle));
  33. // just to be sure it's deinitialized in case of error and not affecting other tests
  34. nvs_flash_deinit();
  35. }
  36. TEST_CASE("various nvs tests", "[nvs]")
  37. {
  38. nvs_handle_t handle_1;
  39. esp_err_t err = nvs_flash_init();
  40. if (err == ESP_ERR_NVS_NO_FREE_PAGES || err == ESP_ERR_NVS_NEW_VERSION_FOUND) {
  41. ESP_LOGW(TAG, "nvs_flash_init failed (0x%x), erasing partition and retrying", err);
  42. ESP_ERROR_CHECK(nvs_flash_erase());
  43. err = nvs_flash_init();
  44. }
  45. ESP_ERROR_CHECK( err );
  46. TEST_ESP_ERR(ESP_ERR_NVS_NOT_FOUND, nvs_open("test_namespace1", NVS_READONLY, &handle_1));
  47. TEST_ESP_ERR(ESP_ERR_NVS_INVALID_HANDLE, nvs_set_i32(handle_1, "foo", 0x12345678));
  48. nvs_close(handle_1);
  49. TEST_ESP_OK(nvs_open("test_namespace2", NVS_READWRITE, &handle_1));
  50. TEST_ESP_OK(nvs_erase_all(handle_1));
  51. TEST_ESP_OK(nvs_set_i32(handle_1, "foo", 0x12345678));
  52. TEST_ESP_OK(nvs_set_i32(handle_1, "foo", 0x23456789));
  53. nvs_handle_t handle_2;
  54. TEST_ESP_OK(nvs_open("test_namespace3", NVS_READWRITE, &handle_2));
  55. TEST_ESP_OK(nvs_erase_all(handle_2));
  56. TEST_ESP_OK(nvs_set_i32(handle_2, "foo", 0x3456789a));
  57. const char* str = "value 0123456789abcdef0123456789abcdef";
  58. TEST_ESP_OK(nvs_set_str(handle_2, "key", str));
  59. int32_t v1;
  60. TEST_ESP_OK(nvs_get_i32(handle_1, "foo", &v1));
  61. TEST_ASSERT_EQUAL_INT32(0x23456789, v1);
  62. int32_t v2;
  63. TEST_ESP_OK(nvs_get_i32(handle_2, "foo", &v2));
  64. TEST_ASSERT_EQUAL_INT32(0x3456789a, v2);
  65. char buf[strlen(str) + 1];
  66. size_t buf_len = sizeof(buf);
  67. TEST_ESP_OK(nvs_get_str(handle_2, "key", buf, &buf_len));
  68. TEST_ASSERT_EQUAL_INT32(0, strcmp(buf, str));
  69. nvs_close(handle_1);
  70. // check that deinit does not leak memory if some handles are still open
  71. nvs_flash_deinit();
  72. nvs_close(handle_2);
  73. }
  74. TEST_CASE("calculate used and free space", "[nvs]")
  75. {
  76. TEST_ESP_ERR(ESP_ERR_INVALID_ARG, nvs_get_stats(NULL, NULL));
  77. nvs_stats_t stat1;
  78. nvs_stats_t stat2;
  79. TEST_ESP_ERR(ESP_ERR_NVS_NOT_INITIALIZED, nvs_get_stats(NULL, &stat1));
  80. TEST_ASSERT_TRUE(stat1.free_entries == 0);
  81. TEST_ASSERT_TRUE(stat1.namespace_count == 0);
  82. TEST_ASSERT_TRUE(stat1.total_entries == 0);
  83. TEST_ASSERT_TRUE(stat1.used_entries == 0);
  84. nvs_handle_t handle = 0;
  85. size_t h_count_entries;
  86. TEST_ESP_ERR(ESP_ERR_NVS_INVALID_HANDLE, nvs_get_used_entry_count(handle, &h_count_entries));
  87. TEST_ASSERT_TRUE(h_count_entries == 0);
  88. esp_err_t err = nvs_flash_init();
  89. if (err == ESP_ERR_NVS_NO_FREE_PAGES || err == ESP_ERR_NVS_NEW_VERSION_FOUND) {
  90. ESP_LOGW(TAG, "nvs_flash_init failed (0x%x), erasing partition and retrying", err);
  91. const esp_partition_t* nvs_partition = esp_partition_find_first(
  92. ESP_PARTITION_TYPE_DATA, ESP_PARTITION_SUBTYPE_DATA_NVS, NULL);
  93. assert(nvs_partition && "partition table must have an NVS partition");
  94. ESP_ERROR_CHECK(nvs_flash_erase());
  95. err = nvs_flash_init();
  96. }
  97. ESP_ERROR_CHECK( err );
  98. // erase if have any namespace
  99. TEST_ESP_OK(nvs_get_stats(NULL, &stat1));
  100. if(stat1.namespace_count != 0) {
  101. TEST_ESP_OK(nvs_flash_deinit());
  102. TEST_ESP_OK(nvs_flash_erase());
  103. TEST_ESP_OK(nvs_flash_init());
  104. }
  105. // after erase. empty partition
  106. TEST_ESP_OK(nvs_get_stats(NULL, &stat1));
  107. TEST_ASSERT_TRUE(stat1.free_entries != 0);
  108. TEST_ASSERT_TRUE(stat1.namespace_count == 0);
  109. TEST_ASSERT_TRUE(stat1.total_entries != 0);
  110. TEST_ASSERT_TRUE(stat1.used_entries == 0);
  111. // create namespace test_k1
  112. nvs_handle_t handle_1;
  113. TEST_ESP_OK(nvs_open("test_k1", NVS_READWRITE, &handle_1));
  114. TEST_ESP_OK(nvs_get_stats(NULL, &stat2));
  115. TEST_ASSERT_TRUE(stat2.free_entries + 1 == stat1.free_entries);
  116. TEST_ASSERT_TRUE(stat2.namespace_count == 1);
  117. TEST_ASSERT_TRUE(stat2.total_entries == stat1.total_entries);
  118. TEST_ASSERT_TRUE(stat2.used_entries == 1);
  119. // create pair key-value com
  120. TEST_ESP_OK(nvs_set_i32(handle_1, "com", 0x12345678));
  121. TEST_ESP_OK(nvs_get_stats(NULL, &stat1));
  122. TEST_ASSERT_TRUE(stat1.free_entries + 1 == stat2.free_entries);
  123. TEST_ASSERT_TRUE(stat1.namespace_count == 1);
  124. TEST_ASSERT_TRUE(stat1.total_entries == stat2.total_entries);
  125. TEST_ASSERT_TRUE(stat1.used_entries == 2);
  126. // change value in com
  127. TEST_ESP_OK(nvs_set_i32(handle_1, "com", 0x01234567));
  128. TEST_ESP_OK(nvs_get_stats(NULL, &stat2));
  129. TEST_ASSERT_TRUE(stat2.free_entries == stat1.free_entries);
  130. TEST_ASSERT_TRUE(stat2.namespace_count == 1);
  131. TEST_ASSERT_TRUE(stat2.total_entries != 0);
  132. TEST_ASSERT_TRUE(stat2.used_entries == 2);
  133. // create pair key-value ru
  134. TEST_ESP_OK(nvs_set_i32(handle_1, "ru", 0x00FF00FF));
  135. TEST_ESP_OK(nvs_get_stats(NULL, &stat1));
  136. TEST_ASSERT_TRUE(stat1.free_entries + 1 == stat2.free_entries);
  137. TEST_ASSERT_TRUE(stat1.namespace_count == 1);
  138. TEST_ASSERT_TRUE(stat1.total_entries != 0);
  139. TEST_ASSERT_TRUE(stat1.used_entries == 3);
  140. // amount valid pair in namespace 1
  141. size_t h1_count_entries;
  142. TEST_ESP_OK(nvs_get_used_entry_count(handle_1, &h1_count_entries));
  143. TEST_ASSERT_TRUE(h1_count_entries == 2);
  144. nvs_handle_t handle_2;
  145. // create namespace test_k2
  146. TEST_ESP_OK(nvs_open("test_k2", NVS_READWRITE, &handle_2));
  147. TEST_ESP_OK(nvs_get_stats(NULL, &stat2));
  148. TEST_ASSERT_TRUE(stat2.free_entries + 1 == stat1.free_entries);
  149. TEST_ASSERT_TRUE(stat2.namespace_count == 2);
  150. TEST_ASSERT_TRUE(stat2.total_entries == stat1.total_entries);
  151. TEST_ASSERT_TRUE(stat2.used_entries == 4);
  152. // create pair key-value
  153. TEST_ESP_OK(nvs_set_i32(handle_2, "su1", 0x00000001));
  154. TEST_ESP_OK(nvs_set_i32(handle_2, "su2", 0x00000002));
  155. TEST_ESP_OK(nvs_set_i32(handle_2, "sus", 0x00000003));
  156. TEST_ESP_OK(nvs_get_stats(NULL, &stat1));
  157. TEST_ASSERT_TRUE(stat1.free_entries + 3 == stat2.free_entries);
  158. TEST_ASSERT_TRUE(stat1.namespace_count == 2);
  159. TEST_ASSERT_TRUE(stat1.total_entries == stat2.total_entries);
  160. TEST_ASSERT_TRUE(stat1.used_entries == 7);
  161. TEST_ASSERT_TRUE(stat1.total_entries == (stat1.used_entries + stat1.free_entries));
  162. // amount valid pair in namespace 2
  163. size_t h2_count_entries;
  164. TEST_ESP_OK(nvs_get_used_entry_count(handle_2, &h2_count_entries));
  165. TEST_ASSERT_TRUE(h2_count_entries == 3);
  166. TEST_ASSERT_TRUE(stat1.used_entries == (h1_count_entries + h2_count_entries + stat1.namespace_count));
  167. nvs_close(handle_1);
  168. nvs_close(handle_2);
  169. size_t temp = h2_count_entries;
  170. TEST_ESP_ERR(ESP_ERR_NVS_INVALID_HANDLE, nvs_get_used_entry_count(handle_1, &h2_count_entries));
  171. TEST_ASSERT_TRUE(h2_count_entries == 0);
  172. h2_count_entries = temp;
  173. TEST_ESP_ERR(ESP_ERR_INVALID_ARG, nvs_get_used_entry_count(handle_1, NULL));
  174. nvs_handle_t handle_3;
  175. // create namespace test_k3
  176. TEST_ESP_OK(nvs_open("test_k3", NVS_READWRITE, &handle_3));
  177. TEST_ESP_OK(nvs_get_stats(NULL, &stat2));
  178. TEST_ASSERT_TRUE(stat2.free_entries + 1 == stat1.free_entries);
  179. TEST_ASSERT_TRUE(stat2.namespace_count == 3);
  180. TEST_ASSERT_TRUE(stat2.total_entries == stat1.total_entries);
  181. TEST_ASSERT_TRUE(stat2.used_entries == 8);
  182. // create pair blobs
  183. uint32_t blob[12];
  184. TEST_ESP_OK(nvs_set_blob(handle_3, "bl1", &blob, sizeof(blob)));
  185. TEST_ESP_OK(nvs_get_stats(NULL, &stat1));
  186. TEST_ASSERT_TRUE(stat1.free_entries + 4 == stat2.free_entries);
  187. TEST_ASSERT_TRUE(stat1.namespace_count == 3);
  188. TEST_ASSERT_TRUE(stat1.total_entries == stat2.total_entries);
  189. TEST_ASSERT_TRUE(stat1.used_entries == 12);
  190. // amount valid pair in namespace 2
  191. size_t h3_count_entries;
  192. TEST_ESP_OK(nvs_get_used_entry_count(handle_3, &h3_count_entries));
  193. TEST_ASSERT_TRUE(h3_count_entries == 4);
  194. TEST_ASSERT_TRUE(stat1.used_entries == (h1_count_entries + h2_count_entries + h3_count_entries + stat1.namespace_count));
  195. nvs_close(handle_3);
  196. TEST_ESP_OK(nvs_flash_deinit());
  197. TEST_ESP_OK(nvs_flash_erase());
  198. }
  199. TEST_CASE("check for memory leaks in nvs_set_blob", "[nvs]")
  200. {
  201. esp_err_t err = nvs_flash_init();
  202. if (err == ESP_ERR_NVS_NO_FREE_PAGES || err == ESP_ERR_NVS_NEW_VERSION_FOUND) {
  203. ESP_ERROR_CHECK(nvs_flash_erase());
  204. err = nvs_flash_init();
  205. }
  206. TEST_ESP_OK( err );
  207. for (int i = 0; i < 500; ++i) {
  208. nvs_handle_t my_handle;
  209. uint8_t key[20] = {0};
  210. TEST_ESP_OK( nvs_open("leak_check_ns", NVS_READWRITE, &my_handle) );
  211. TEST_ESP_OK( nvs_set_blob(my_handle, "key", key, sizeof(key)) );
  212. TEST_ESP_OK( nvs_commit(my_handle) );
  213. nvs_close(my_handle);
  214. printf("%d\n", esp_get_free_heap_size());
  215. }
  216. nvs_flash_deinit();
  217. printf("%d\n", esp_get_free_heap_size());
  218. /* heap leaks will be checked in unity_platform.c */
  219. }
  220. #ifdef CONFIG_NVS_ENCRYPTION
  221. TEST_CASE("check underlying xts code for 32-byte size sector encryption", "[nvs]")
  222. {
  223. uint8_t eky_hex[2 * NVS_KEY_SIZE] = { 0x11,0x11,0x11,0x11,0x11,0x11,0x11,0x11,
  224. 0x11,0x11,0x11,0x11,0x11,0x11,0x11,0x11,
  225. 0x11,0x11,0x11,0x11,0x11,0x11,0x11,0x11,
  226. 0x11,0x11,0x11,0x11,0x11,0x11,0x11,0x11,
  227. /* Tweak key below*/
  228. 0x22,0x22,0x22,0x22,0x22,0x22,0x22,0x22,
  229. 0x22,0x22,0x22,0x22,0x22,0x22,0x22,0x22,
  230. 0x22,0x22,0x22,0x22,0x22,0x22,0x22,0x22,
  231. 0x22,0x22,0x22,0x22,0x22,0x22,0x22,0x22 };
  232. uint8_t ba_hex[16] = { 0x33,0x33,0x33,0x33,0x33,0x00,0x00,0x00,
  233. 0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00 };
  234. uint8_t ptxt_hex[32] = { 0x44,0x44,0x44,0x44,0x44,0x44,0x44,0x44,
  235. 0x44,0x44,0x44,0x44,0x44,0x44,0x44,0x44,
  236. 0x44,0x44,0x44,0x44,0x44,0x44,0x44,0x44,
  237. 0x44,0x44,0x44,0x44,0x44,0x44,0x44,0x44 };
  238. uint8_t ctxt_hex[32] = { 0xe6,0x22,0x33,0x4f,0x18,0x4b,0xbc,0xe1,
  239. 0x29,0xa2,0x5b,0x2a,0xc7,0x6b,0x3d,0x92,
  240. 0xab,0xf9,0x8e,0x22,0xdf,0x5b,0xdd,0x15,
  241. 0xaf,0x47,0x1f,0x3d,0xb8,0x94,0x6a,0x85 };
  242. mbedtls_aes_xts_context ectx[1];
  243. mbedtls_aes_xts_context dctx[1];
  244. mbedtls_aes_xts_init(ectx);
  245. mbedtls_aes_xts_init(dctx);
  246. TEST_ASSERT_TRUE(!mbedtls_aes_xts_setkey_enc(ectx, eky_hex, 2 * NVS_KEY_SIZE * 8));
  247. TEST_ASSERT_TRUE(!mbedtls_aes_xts_setkey_enc(dctx, eky_hex, 2 * NVS_KEY_SIZE * 8));
  248. TEST_ASSERT_TRUE(!mbedtls_aes_crypt_xts(ectx, MBEDTLS_AES_ENCRYPT, 32, ba_hex, ptxt_hex, ptxt_hex));
  249. TEST_ASSERT_TRUE(!memcmp(ptxt_hex, ctxt_hex, 32));
  250. }
  251. TEST_CASE("Check nvs key partition APIs (read and generate keys)", "[nvs]")
  252. {
  253. nvs_sec_cfg_t cfg, cfg2;
  254. const esp_partition_t* key_part = esp_partition_find_first(
  255. ESP_PARTITION_TYPE_DATA, ESP_PARTITION_SUBTYPE_DATA_NVS_KEYS, NULL);
  256. if (!esp_flash_encryption_enabled()) {
  257. TEST_IGNORE_MESSAGE("flash encryption disabled, skipping nvs_key partition related tests");
  258. }
  259. TEST_ESP_OK(esp_partition_erase_range(key_part, 0, key_part->size));
  260. TEST_ESP_ERR(ESP_ERR_NVS_KEYS_NOT_INITIALIZED, nvs_flash_read_security_cfg(key_part, &cfg));
  261. TEST_ESP_OK(nvs_flash_generate_keys(key_part, &cfg));
  262. TEST_ESP_OK(nvs_flash_read_security_cfg(key_part, &cfg2));
  263. TEST_ASSERT_TRUE(!memcmp(&cfg, &cfg2, sizeof(nvs_sec_cfg_t)));
  264. }
  265. TEST_CASE("test nvs apis with encryption enabled", "[nvs]")
  266. {
  267. if (!esp_flash_encryption_enabled()) {
  268. TEST_IGNORE_MESSAGE("flash encryption disabled, skipping nvs_api tests with encryption enabled");
  269. }
  270. const esp_partition_t* key_part = esp_partition_find_first(
  271. ESP_PARTITION_TYPE_DATA, ESP_PARTITION_SUBTYPE_DATA_NVS_KEYS, NULL);
  272. assert(key_part && "partition table must have an NVS Key partition");
  273. const esp_partition_t* nvs_partition = esp_partition_find_first(
  274. ESP_PARTITION_TYPE_DATA, ESP_PARTITION_SUBTYPE_DATA_NVS, NULL);
  275. assert(nvs_partition && "partition table must have an NVS partition");
  276. ESP_ERROR_CHECK( esp_partition_erase_range(key_part, 0, key_part->size) );
  277. bool done = false;
  278. do {
  279. ESP_ERROR_CHECK( esp_partition_erase_range(nvs_partition, 0, nvs_partition->size) );
  280. nvs_sec_cfg_t cfg;
  281. esp_err_t err = nvs_flash_read_security_cfg(key_part, &cfg);
  282. if(err == ESP_ERR_NVS_KEYS_NOT_INITIALIZED) {
  283. uint8_t value[4096] = {[0 ... 4095] = 0xff};
  284. TEST_ESP_OK(esp_partition_write(key_part, 0, value, sizeof(value)));
  285. TEST_ESP_ERR(ESP_ERR_NVS_KEYS_NOT_INITIALIZED, nvs_flash_read_security_cfg(key_part, &cfg));
  286. TEST_ESP_OK(nvs_flash_generate_keys(key_part, &cfg));
  287. } else {
  288. /* Second time key_partition exists already*/
  289. ESP_ERROR_CHECK(err);
  290. done = true;
  291. }
  292. TEST_ESP_OK(nvs_flash_secure_init(&cfg));
  293. nvs_handle_t handle_1;
  294. TEST_ESP_ERR(ESP_ERR_NVS_NOT_FOUND, nvs_open("namespace1", NVS_READONLY, &handle_1));
  295. TEST_ESP_OK(nvs_open("namespace1", NVS_READWRITE, &handle_1));
  296. TEST_ESP_OK(nvs_set_i32(handle_1, "foo", 0x12345678));
  297. TEST_ESP_OK(nvs_set_i32(handle_1, "foo", 0x23456789));
  298. nvs_handle_t handle_2;
  299. TEST_ESP_OK(nvs_open("namespace2", NVS_READWRITE, &handle_2));
  300. TEST_ESP_OK(nvs_set_i32(handle_2, "foo", 0x3456789a));
  301. const char* str = "value 0123456789abcdef0123456789abcdef";
  302. TEST_ESP_OK(nvs_set_str(handle_2, "key", str));
  303. int32_t v1;
  304. TEST_ESP_OK(nvs_get_i32(handle_1, "foo", &v1));
  305. TEST_ASSERT_TRUE(0x23456789 == v1);
  306. int32_t v2;
  307. TEST_ESP_OK(nvs_get_i32(handle_2, "foo", &v2));
  308. TEST_ASSERT_TRUE(0x3456789a == v2);
  309. char buf[strlen(str) + 1];
  310. size_t buf_len = sizeof(buf);
  311. size_t buf_len_needed;
  312. TEST_ESP_OK(nvs_get_str(handle_2, "key", NULL, &buf_len_needed));
  313. TEST_ASSERT_TRUE(buf_len_needed == buf_len);
  314. size_t buf_len_short = buf_len - 1;
  315. TEST_ESP_ERR(ESP_ERR_NVS_INVALID_LENGTH, nvs_get_str(handle_2, "key", buf, &buf_len_short));
  316. TEST_ASSERT_TRUE(buf_len_short == buf_len);
  317. size_t buf_len_long = buf_len + 1;
  318. TEST_ESP_OK(nvs_get_str(handle_2, "key", buf, &buf_len_long));
  319. TEST_ASSERT_TRUE(buf_len_long == buf_len);
  320. TEST_ESP_OK(nvs_get_str(handle_2, "key", buf, &buf_len));
  321. TEST_ASSERT_TRUE(0 == strcmp(buf, str));
  322. nvs_close(handle_1);
  323. nvs_close(handle_2);
  324. TEST_ESP_OK(nvs_flash_deinit());
  325. } while(!done);
  326. }
  327. TEST_CASE("test nvs apis for nvs partition generator utility with encryption enabled", "[nvs_part_gen]")
  328. {
  329. if (!esp_flash_encryption_enabled()) {
  330. TEST_IGNORE_MESSAGE("flash encryption disabled, skipping nvs_api tests with encryption enabled");
  331. }
  332. nvs_handle_t handle;
  333. nvs_sec_cfg_t xts_cfg;
  334. extern const char nvs_key_start[] asm("_binary_encryption_keys_bin_start");
  335. extern const char nvs_key_end[] asm("_binary_encryption_keys_bin_end");
  336. extern const char nvs_data_start[] asm("_binary_partition_encrypted_bin_start");
  337. extern const char sample_bin_start[] asm("_binary_sample_bin_start");
  338. const esp_partition_t* key_part = esp_partition_find_first(
  339. ESP_PARTITION_TYPE_DATA, ESP_PARTITION_SUBTYPE_DATA_NVS_KEYS, NULL);
  340. const esp_partition_t* nvs_part = esp_partition_find_first(
  341. ESP_PARTITION_TYPE_DATA, ESP_PARTITION_SUBTYPE_DATA_NVS, NULL);
  342. assert(key_part && "partition table must have a KEY partition");
  343. TEST_ASSERT_TRUE((nvs_key_end - nvs_key_start - 1) == SPI_FLASH_SEC_SIZE);
  344. assert(nvs_part && "partition table must have an NVS partition");
  345. printf("\n nvs_part size:%d\n", nvs_part->size);
  346. ESP_ERROR_CHECK(esp_partition_erase_range(key_part, 0, key_part->size));
  347. ESP_ERROR_CHECK( esp_partition_erase_range(nvs_part, 0, nvs_part->size) );
  348. for (int i = 0; i < key_part->size; i+= SPI_FLASH_SEC_SIZE) {
  349. ESP_ERROR_CHECK( esp_partition_write(key_part, i, nvs_key_start + i, SPI_FLASH_SEC_SIZE) );
  350. }
  351. for (int i = 0; i < nvs_part->size; i+= SPI_FLASH_SEC_SIZE) {
  352. ESP_ERROR_CHECK( spi_flash_write(nvs_part->address + i, nvs_data_start + i, SPI_FLASH_SEC_SIZE) );
  353. }
  354. esp_err_t err = nvs_flash_read_security_cfg(key_part, &xts_cfg);
  355. ESP_ERROR_CHECK(err);
  356. TEST_ESP_OK(nvs_flash_secure_init(&xts_cfg));
  357. TEST_ESP_OK(nvs_open("dummyNamespace", NVS_READONLY, &handle));
  358. uint8_t u8v;
  359. TEST_ESP_OK( nvs_get_u8(handle, "dummyU8Key", &u8v));
  360. TEST_ASSERT_TRUE(u8v == 127);
  361. int8_t i8v;
  362. TEST_ESP_OK( nvs_get_i8(handle, "dummyI8Key", &i8v));
  363. TEST_ASSERT_TRUE(i8v == -128);
  364. uint16_t u16v;
  365. TEST_ESP_OK( nvs_get_u16(handle, "dummyU16Key", &u16v));
  366. TEST_ASSERT_TRUE(u16v == 32768);
  367. uint32_t u32v;
  368. TEST_ESP_OK( nvs_get_u32(handle, "dummyU32Key", &u32v));
  369. TEST_ASSERT_TRUE(u32v == 4294967295);
  370. int32_t i32v;
  371. TEST_ESP_OK( nvs_get_i32(handle, "dummyI32Key", &i32v));
  372. TEST_ASSERT_TRUE(i32v == -2147483648);
  373. char buf[64] = {0};
  374. size_t buflen = 64;
  375. TEST_ESP_OK( nvs_get_str(handle, "dummyStringKey", buf, &buflen));
  376. TEST_ASSERT_TRUE(strncmp(buf, "0A:0B:0C:0D:0E:0F", buflen) == 0);
  377. uint8_t hexdata[] = {0x01, 0x02, 0x03, 0xab, 0xcd, 0xef};
  378. buflen = 64;
  379. TEST_ESP_OK( nvs_get_blob(handle, "dummyHex2BinKey", buf, &buflen));
  380. TEST_ASSERT_TRUE(memcmp(buf, hexdata, buflen) == 0);
  381. uint8_t base64data[] = {'1', '2', '3', 'a', 'b', 'c'};
  382. buflen = 64;
  383. TEST_ESP_OK( nvs_get_blob(handle, "dummyBase64Key", buf, &buflen));
  384. TEST_ASSERT_TRUE(memcmp(buf, base64data, buflen) == 0);
  385. uint8_t hexfiledata[] = {0x01, 0x23, 0x45, 0x67, 0x89, 0xab, 0xcd, 0xef};
  386. buflen = 64;
  387. TEST_ESP_OK( nvs_get_blob(handle, "hexFileKey", buf, &buflen));
  388. TEST_ASSERT_TRUE(memcmp(buf, hexfiledata, buflen) == 0);
  389. uint8_t base64filedata[] = {0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, 0x08, 0x09, 0xab, 0xcd, 0xef};
  390. buflen = 64;
  391. TEST_ESP_OK( nvs_get_blob(handle, "base64FileKey", buf, &buflen));
  392. TEST_ASSERT_TRUE(memcmp(buf, base64filedata, buflen) == 0);
  393. uint8_t strfiledata[64] = "abcdefghijklmnopqrstuvwxyz\0";
  394. buflen = 64;
  395. TEST_ESP_OK( nvs_get_str(handle, "stringFileKey", buf, &buflen));
  396. TEST_ASSERT_TRUE(memcmp(buf, strfiledata, buflen) == 0);
  397. char bin_data[5120];
  398. size_t bin_len = sizeof(bin_data);
  399. TEST_ESP_OK( nvs_get_blob(handle, "binFileKey", bin_data, &bin_len));
  400. TEST_ASSERT_TRUE(memcmp(bin_data, sample_bin_start, bin_len) == 0);
  401. nvs_close(handle);
  402. TEST_ESP_OK(nvs_flash_deinit());
  403. }
  404. #endif