| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971 |
- /*
- * SPDX-FileCopyrightText: 2015-2022 Espressif Systems (Shanghai) CO LTD
- *
- * SPDX-License-Identifier: Apache-2.0
- */
- #include "catch.hpp"
- #include "nvs.hpp"
- #include "nvs_test_api.h"
- #include "sdkconfig.h"
- #include "spi_flash_emulation.h"
- #include "nvs_partition_manager.hpp"
- #include "nvs_partition.hpp"
- #include "mbedtls/aes.h"
- #include <sstream>
- #include <iostream>
- #include <fstream>
- #include <dirent.h>
- #include <unistd.h>
- #include <sys/wait.h>
- #include <string.h>
- #include <string>
- #include "test_fixtures.hpp"
- #define TEST_ESP_ERR(rc, res) CHECK((rc) == (res))
- #define TEST_ESP_OK(rc) CHECK((rc) == ESP_OK)
- stringstream s_perf;
- void dumpBytes(const uint8_t *data, size_t count)
- {
- for (uint32_t i = 0; i < count; ++i) {
- if (i % 32 == 0) {
- printf("%08x ", i);
- }
- printf("%02x ", data[i]);
- if ((i + 1) % 32 == 0) {
- printf("\n");
- }
- }
- }
- TEST_CASE("Page handles invalid CRC of variable length items", "[nvs][cur]")
- {
- PartitionEmulationFixture f(0, 4);
- {
- nvs::Page page;
- TEST_ESP_OK(page.load(&f.part, 0));
- char buf[128] = {0};
- TEST_ESP_OK(page.writeItem(1, nvs::ItemType::BLOB, "1", buf, sizeof(buf)));
- }
- // corrupt header of the item (64 is the offset of the first item in page)
- uint32_t overwrite_buf = 0;
- f.emu.write(64, &overwrite_buf, 4);
- // load page again
- {
- nvs::Page page;
- TEST_ESP_OK(page.load(&f.part, 0));
- }
- }
- TEST_CASE("namespace name is deep copy", "[nvs]")
- {
- char ns_name[16];
- strcpy(ns_name, "const_name");
- nvs_handle_t handle_1;
- nvs_handle_t handle_2;
- const uint32_t NVS_FLASH_SECTOR = 6;
- const uint32_t NVS_FLASH_SECTOR_COUNT_MIN = 3;
- PartitionEmulationFixture f(NVS_FLASH_SECTOR, NVS_FLASH_SECTOR_COUNT_MIN);
- f.emu.setBounds(NVS_FLASH_SECTOR, NVS_FLASH_SECTOR + NVS_FLASH_SECTOR_COUNT_MIN);
- TEST_ESP_OK( nvs::NVSPartitionManager::get_instance()->init_custom(&f.part,
- NVS_FLASH_SECTOR,
- NVS_FLASH_SECTOR_COUNT_MIN));
- TEST_ESP_OK(nvs_open("const_name", NVS_READWRITE, &handle_1));
- strcpy(ns_name, "just_kidding");
- CHECK(nvs_open("just_kidding", NVS_READONLY, &handle_2) == ESP_ERR_NVS_NOT_FOUND);
- nvs_close(handle_1);
- nvs_close(handle_2);
- nvs_flash_deinit_partition(NVS_DEFAULT_PART_NAME);
- }
- TEST_CASE("writing the identical content does not write or erase", "[nvs]")
- {
- PartitionEmulationFixture f(0, 20);
- const uint32_t NVS_FLASH_SECTOR = 5;
- const uint32_t NVS_FLASH_SECTOR_COUNT_MIN = 10;
- f.emu.setBounds(NVS_FLASH_SECTOR, NVS_FLASH_SECTOR + NVS_FLASH_SECTOR_COUNT_MIN);
- TEST_ESP_OK(nvs::NVSPartitionManager::get_instance()->init_custom(&f.part,
- NVS_FLASH_SECTOR,
- NVS_FLASH_SECTOR_COUNT_MIN));
- nvs_handle misc_handle;
- TEST_ESP_OK(nvs_open("test", NVS_READWRITE, &misc_handle));
- // Test writing a u8 twice, then changing it
- nvs_set_u8(misc_handle, "test_u8", 8);
- f.emu.clearStats();
- nvs_set_u8(misc_handle, "test_u8", 8);
- CHECK(f.emu.getWriteOps() == 0);
- CHECK(f.emu.getEraseOps() == 0);
- CHECK(f.emu.getReadOps() != 0);
- f.emu.clearStats();
- nvs_set_u8(misc_handle, "test_u8", 9);
- CHECK(f.emu.getWriteOps() != 0);
- CHECK(f.emu.getReadOps() != 0);
- // Test writing a string twice, then changing it
- static const char *test[2] = {"Hello world.", "Hello world!"};
- nvs_set_str(misc_handle, "test_str", test[0]);
- f.emu.clearStats();
- nvs_set_str(misc_handle, "test_str", test[0]);
- CHECK(f.emu.getWriteOps() == 0);
- CHECK(f.emu.getEraseOps() == 0);
- CHECK(f.emu.getReadOps() != 0);
- f.emu.clearStats();
- nvs_set_str(misc_handle, "test_str", test[1]);
- CHECK(f.emu.getWriteOps() != 0);
- CHECK(f.emu.getReadOps() != 0);
- // Test writing a multi-page blob, then changing it
- uint8_t blob[nvs::Page::CHUNK_MAX_SIZE * 3] = {0};
- memset(blob, 1, sizeof(blob));
- nvs_set_blob(misc_handle, "test_blob", blob, sizeof(blob));
- f.emu.clearStats();
- nvs_set_blob(misc_handle, "test_blob", blob, sizeof(blob));
- CHECK(f.emu.getWriteOps() == 0);
- CHECK(f.emu.getEraseOps() == 0);
- CHECK(f.emu.getReadOps() != 0);
- blob[sizeof(blob) - 1]++;
- f.emu.clearStats();
- nvs_set_blob(misc_handle, "test_blob", blob, sizeof(blob));
- CHECK(f.emu.getWriteOps() != 0);
- CHECK(f.emu.getReadOps() != 0);
- TEST_ESP_OK(nvs_flash_deinit_partition(NVS_DEFAULT_PART_NAME));
- }
- TEST_CASE("can init storage from flash with random contents", "[nvs]")
- {
- PartitionEmulationFixture f(0, 10);
- f.emu.randomize(42);
- nvs_handle_t handle;
- const uint32_t NVS_FLASH_SECTOR = 5;
- const uint32_t NVS_FLASH_SECTOR_COUNT_MIN = 3;
- f.emu.setBounds(NVS_FLASH_SECTOR, NVS_FLASH_SECTOR + NVS_FLASH_SECTOR_COUNT_MIN);
- TEST_ESP_OK(nvs::NVSPartitionManager::get_instance()->init_custom(&f.part,
- NVS_FLASH_SECTOR,
- NVS_FLASH_SECTOR_COUNT_MIN));
- TEST_ESP_OK(nvs_open("nvs.net80211", NVS_READWRITE, &handle));
- uint8_t opmode = 2;
- if (nvs_get_u8(handle, "wifi.opmode", &opmode) != ESP_OK) {
- TEST_ESP_OK(nvs_set_u8(handle, "wifi.opmode", opmode));
- }
- TEST_ESP_OK(nvs_flash_deinit_partition(NVS_DEFAULT_PART_NAME));
- }
- TEST_CASE("nvs api tests, starting with random data in flash", "[nvs][long]")
- {
- const size_t testIters = 3000;
- int lastPercent = -1;
- for (size_t count = 0; count < testIters; ++count) {
- int percentDone = (int) (count * 100 / testIters);
- if (percentDone != lastPercent) {
- lastPercent = percentDone;
- printf("%d%%\n", percentDone);
- }
- PartitionEmulationFixture f(0, 10);
- f.emu.randomize(static_cast<uint32_t>(count));
- const uint32_t NVS_FLASH_SECTOR = 6;
- const uint32_t NVS_FLASH_SECTOR_COUNT_MIN = 3;
- f.emu.setBounds(NVS_FLASH_SECTOR, NVS_FLASH_SECTOR + NVS_FLASH_SECTOR_COUNT_MIN);
- TEST_ESP_OK(nvs::NVSPartitionManager::get_instance()->init_custom(&f.part,
- NVS_FLASH_SECTOR,
- NVS_FLASH_SECTOR_COUNT_MIN));
- nvs_handle_t handle_1;
- TEST_ESP_ERR(nvs_open("namespace1", NVS_READONLY, &handle_1), ESP_ERR_NVS_NOT_FOUND);
- TEST_ESP_OK(nvs_open("namespace1", NVS_READWRITE, &handle_1));
- TEST_ESP_OK(nvs_set_i32(handle_1, "foo", 0x12345678));
- for (size_t i = 0; i < 500; ++i) {
- nvs_handle_t handle_2;
- TEST_ESP_OK(nvs_open("namespace2", NVS_READWRITE, &handle_2));
- TEST_ESP_OK(nvs_set_i32(handle_1, "foo", 0x23456789 % (i + 1)));
- TEST_ESP_OK(nvs_set_i32(handle_2, "foo", static_cast<int32_t>(i)));
- const char *str = "value 0123456789abcdef0123456789abcdef %09d";
- char str_buf[128];
- snprintf(str_buf, sizeof(str_buf), str, i + count * 1024);
- TEST_ESP_OK(nvs_set_str(handle_2, "key", str_buf));
- int32_t v1;
- TEST_ESP_OK(nvs_get_i32(handle_1, "foo", &v1));
- CHECK(0x23456789 % (i + 1) == v1);
- int32_t v2;
- TEST_ESP_OK(nvs_get_i32(handle_2, "foo", &v2));
- CHECK(static_cast<int32_t>(i) == v2);
- char buf[128];
- size_t buf_len = sizeof(buf);
- TEST_ESP_OK(nvs_get_str(handle_2, "key", buf, &buf_len));
- CHECK(0 == strcmp(buf, str_buf));
- nvs_close(handle_2);
- }
- nvs_close(handle_1);
- TEST_ESP_OK(nvs_flash_deinit_partition(f.part.get_partition_name()));
- }
- }
- extern "C" void nvs_dump(const char *partName);
- class RandomTest {
- static const size_t nKeys = 11;
- int32_t v1 = 0, v2 = 0;
- uint64_t v3 = 0, v4 = 0;
- static const size_t strBufLen = 1024;
- static const size_t smallBlobLen = nvs::Page::CHUNK_MAX_SIZE / 3;
- static const size_t largeBlobLen = nvs::Page::CHUNK_MAX_SIZE * 3;
- char v5[strBufLen], v6[strBufLen], v7[strBufLen], v8[strBufLen], v9[strBufLen];
- uint8_t v10[smallBlobLen], v11[largeBlobLen];
- bool written[nKeys];
- public:
- RandomTest()
- {
- std::fill_n(written, nKeys, false);
- }
- template<typename TGen>
- esp_err_t doRandomThings(nvs_handle_t handle, TGen gen, size_t &count)
- {
- const char *keys[] = {"foo", "bar", "longkey_0123456", "another key", "param1", "param2", "param3", "param4", "param5", "singlepage", "multipage"};
- const nvs::ItemType types[] = {nvs::ItemType::I32, nvs::ItemType::I32, nvs::ItemType::U64, nvs::ItemType::U64, nvs::ItemType::SZ, nvs::ItemType::SZ, nvs::ItemType::SZ, nvs::ItemType::SZ, nvs::ItemType::SZ, nvs::ItemType::BLOB, nvs::ItemType::BLOB};
- void *values[] = {&v1, &v2, &v3, &v4, &v5, &v6, &v7, &v8, &v9, &v10, &v11};
- const size_t nKeys = sizeof(keys) / sizeof(keys[0]);
- static_assert(nKeys == sizeof(types) / sizeof(types[0]), "");
- static_assert(nKeys == sizeof(values) / sizeof(values[0]), "");
- auto randomRead = [&](size_t index) -> esp_err_t {
- switch (types[index])
- {
- case nvs::ItemType::I32: {
- int32_t val;
- auto err = nvs_get_i32(handle, keys[index], &val);
- if (err == ESP_ERR_FLASH_OP_FAIL) {
- return err;
- }
- if (!written[index]) {
- REQUIRE(err == ESP_ERR_NVS_NOT_FOUND);
- } else {
- REQUIRE(err == ESP_OK);
- REQUIRE(val == *reinterpret_cast<int32_t *>(values[index]));
- }
- break;
- }
- case nvs::ItemType::U64: {
- uint64_t val;
- auto err = nvs_get_u64(handle, keys[index], &val);
- if (err == ESP_ERR_FLASH_OP_FAIL) {
- return err;
- }
- if (!written[index]) {
- REQUIRE(err == ESP_ERR_NVS_NOT_FOUND);
- } else {
- REQUIRE(err == ESP_OK);
- REQUIRE(val == *reinterpret_cast<uint64_t *>(values[index]));
- }
- break;
- }
- case nvs::ItemType::SZ: {
- char buf[strBufLen];
- size_t len = strBufLen;
- auto err = nvs_get_str(handle, keys[index], buf, &len);
- if (err == ESP_ERR_FLASH_OP_FAIL) {
- return err;
- }
- if (!written[index]) {
- REQUIRE(err == ESP_ERR_NVS_NOT_FOUND);
- } else {
- REQUIRE(err == ESP_OK);
- REQUIRE(strncmp(buf, reinterpret_cast<const char *>(values[index]), strBufLen) == 0);
- }
- break;
- }
- case nvs::ItemType::BLOB: {
- uint32_t blobBufLen = 0;
- if (strncmp(keys[index], "singlepage", sizeof("singlepage")) == 0) {
- blobBufLen = smallBlobLen ;
- } else {
- blobBufLen = largeBlobLen ;
- }
- uint8_t buf[blobBufLen];
- memset(buf, 0, blobBufLen);
- size_t len = blobBufLen;
- auto err = nvs_get_blob(handle, keys[index], buf, &len);
- if (err == ESP_ERR_FLASH_OP_FAIL) {
- return err;
- }
- if (!written[index]) {
- REQUIRE(err == ESP_ERR_NVS_NOT_FOUND);
- } else {
- REQUIRE(err == ESP_OK);
- REQUIRE(memcmp(buf, reinterpret_cast<const uint8_t *>(values[index]), blobBufLen) == 0);
- }
- break;
- }
- default:
- assert(0);
- }
- return ESP_OK;
- };
- auto randomWrite = [&](size_t index) -> esp_err_t {
- switch (types[index])
- {
- case nvs::ItemType::I32: {
- int32_t val = static_cast<int32_t>(gen());
- auto err = nvs_set_i32(handle, keys[index], val);
- if (err == ESP_ERR_FLASH_OP_FAIL) {
- return err;
- }
- if (err == ESP_ERR_NVS_REMOVE_FAILED) {
- written[index] = true;
- *reinterpret_cast<int32_t *>(values[index]) = val;
- return ESP_ERR_FLASH_OP_FAIL;
- }
- REQUIRE(err == ESP_OK);
- written[index] = true;
- *reinterpret_cast<int32_t *>(values[index]) = val;
- break;
- }
- case nvs::ItemType::U64: {
- uint64_t val = static_cast<uint64_t>(gen());
- auto err = nvs_set_u64(handle, keys[index], val);
- if (err == ESP_ERR_FLASH_OP_FAIL) {
- return err;
- }
- if (err == ESP_ERR_NVS_REMOVE_FAILED) {
- written[index] = true;
- *reinterpret_cast<uint64_t *>(values[index]) = val;
- return ESP_ERR_FLASH_OP_FAIL;
- }
- REQUIRE(err == ESP_OK);
- written[index] = true;
- *reinterpret_cast<uint64_t *>(values[index]) = val;
- break;
- }
- case nvs::ItemType::SZ: {
- char buf[strBufLen];
- size_t len = strBufLen;
- size_t strLen = gen() % (strBufLen - 1);
- std::generate_n(buf, strLen, [&]() -> char {
- const char c = static_cast<char>(gen() % 127);
- return (c < 32) ? 32 : c;
- });
- buf[strLen] = 0;
- auto err = nvs_set_str(handle, keys[index], buf);
- if (err == ESP_ERR_FLASH_OP_FAIL) {
- return err;
- }
- if (err == ESP_ERR_NVS_REMOVE_FAILED) {
- written[index] = true;
- strncpy(reinterpret_cast<char *>(values[index]), buf, strBufLen);
- return ESP_ERR_FLASH_OP_FAIL;
- }
- REQUIRE(err == ESP_OK);
- written[index] = true;
- strncpy(reinterpret_cast<char *>(values[index]), buf, strBufLen);
- break;
- }
- case nvs::ItemType::BLOB: {
- uint32_t blobBufLen = 0;
- if (strncmp(keys[index], "singlepage", sizeof("singlepage")) == 0) {
- blobBufLen = smallBlobLen ;
- } else {
- blobBufLen = largeBlobLen ;
- }
- uint8_t buf[blobBufLen];
- memset(buf, 0, blobBufLen);
- size_t blobLen = gen() % blobBufLen;
- std::generate_n(buf, blobLen, [&]() -> uint8_t {
- return static_cast<uint8_t>(gen() % 256);
- });
- auto err = nvs_set_blob(handle, keys[index], buf, blobLen);
- if (err == ESP_ERR_FLASH_OP_FAIL) {
- return err;
- }
- if (err == ESP_ERR_NVS_REMOVE_FAILED) {
- written[index] = true;
- memcpy(reinterpret_cast<uint8_t *>(values[index]), buf, blobBufLen);
- return ESP_ERR_FLASH_OP_FAIL;
- }
- REQUIRE(err == ESP_OK);
- written[index] = true;
- memcpy(reinterpret_cast<char *>(values[index]), buf, blobBufLen);
- break;
- }
- default:
- assert(0);
- }
- return ESP_OK;
- };
- for (; count != 0; --count) {
- size_t index = gen() % (nKeys);
- switch (gen() % 3) {
- case 0: // read, 1/3
- if (randomRead(index) == ESP_ERR_FLASH_OP_FAIL) {
- return ESP_ERR_FLASH_OP_FAIL;
- }
- break;
- default: // write, 2/3
- if (randomWrite(index) == ESP_ERR_FLASH_OP_FAIL) {
- return ESP_ERR_FLASH_OP_FAIL;
- }
- break;
- }
- }
- return ESP_OK;
- }
- esp_err_t handleExternalWriteAtIndex(uint8_t index, const void *value, const size_t len )
- {
- if (index == 9) { /* This is only done for small-page blobs for now*/
- if (len > smallBlobLen) {
- return ESP_FAIL;
- }
- memcpy(v10, value, len);
- written[index] = true;
- return ESP_OK;
- } else {
- return ESP_FAIL;
- }
- }
- };
- TEST_CASE("test recovery from sudden poweroff", "[long][nvs][recovery][monkey]")
- {
- std::random_device rd;
- std::mt19937 gen(rd());
- uint32_t seed = 3;
- gen.seed(seed);
- const size_t iter_count = 2000;
- size_t totalOps = 0;
- int lastPercent = -1;
- for (uint32_t errDelay = 0; ; ++errDelay) {
- INFO(errDelay);
- PartitionEmulationFixture f(0, 10);
- const uint32_t NVS_FLASH_SECTOR = 2;
- const uint32_t NVS_FLASH_SECTOR_COUNT_MIN = 8;
- f.emu.setBounds(NVS_FLASH_SECTOR, NVS_FLASH_SECTOR + NVS_FLASH_SECTOR_COUNT_MIN);
- f.emu.randomize(seed);
- f.emu.clearStats();
- f.emu.failAfter(errDelay);
- RandomTest test;
- if (totalOps != 0) {
- int percent = errDelay * 100 / totalOps;
- if (percent > lastPercent) {
- printf("%d/%d (%d%%)\r\n", errDelay, static_cast<int>(totalOps), percent);
- lastPercent = percent;
- }
- }
- nvs_handle_t handle;
- size_t count = iter_count;
- if (nvs::NVSPartitionManager::get_instance()->init_custom(&f.part,
- NVS_FLASH_SECTOR,
- NVS_FLASH_SECTOR_COUNT_MIN) == ESP_OK) {
- auto res = ESP_ERR_FLASH_OP_FAIL;
- if (nvs_open("namespace1", NVS_READWRITE, &handle) == ESP_OK) {
- res = test.doRandomThings(handle, gen, count);
- nvs_close(handle);
- }
- TEST_ESP_OK(nvs_flash_deinit_partition(NVS_DEFAULT_PART_NAME));
- if (res != ESP_ERR_FLASH_OP_FAIL) {
- // This means we got to the end without an error due to f.emu.failAfter(), therefore errDelay
- // is high enough that we're not triggering it any more, therefore we're done
- break;
- }
- }
- TEST_ESP_OK(nvs::NVSPartitionManager::get_instance()->init_custom(&f.part,
- NVS_FLASH_SECTOR,
- NVS_FLASH_SECTOR_COUNT_MIN));
- TEST_ESP_OK(nvs_open("namespace1", NVS_READWRITE, &handle));
- auto res = test.doRandomThings(handle, gen, count);
- if (res != ESP_OK) {
- nvs_dump(NVS_DEFAULT_PART_NAME);
- CHECK(0);
- }
- nvs_close(handle);
- totalOps = f.emu.getEraseOps() + f.emu.getWriteBytes() / 4;
- TEST_ESP_OK(nvs_flash_deinit_partition(NVS_DEFAULT_PART_NAME));
- }
- }
- TEST_CASE("duplicate items are removed", "[nvs][dupes]")
- {
- PartitionEmulationFixture f(0, 3);
- {
- // create one item
- nvs::Page p;
- p.load(&f.part, 0);
- p.writeItem<uint8_t>(1, "opmode", 3);
- }
- {
- // add another two without deleting the first one
- nvs::Item item(1, nvs::ItemType::U8, 1, "opmode");
- item.data[0] = 2;
- item.crc32 = item.calculateCrc32();
- f.emu.write(3 * 32, reinterpret_cast<const uint32_t *>(&item), sizeof(item));
- f.emu.write(4 * 32, reinterpret_cast<const uint32_t *>(&item), sizeof(item));
- uint32_t mask = 0xFFFFFFEA;
- f.emu.write(32, &mask, 4);
- }
- {
- // load page and check that second item persists
- nvs::Storage s(&f.part);
- s.init(0, 3);
- uint8_t val;
- ESP_ERROR_CHECK(s.readItem(1, "opmode", val));
- CHECK(val == 2);
- }
- {
- nvs::Page p;
- p.load(&f.part, 0);
- CHECK(p.getErasedEntryCount() == 2);
- CHECK(p.getUsedEntryCount() == 1);
- }
- }
- TEST_CASE("recovery after failure to write data", "[nvs]")
- {
- PartitionEmulationFixture f(0, 3);
- const char str[] = "value 0123456789abcdef012345678value 0123456789abcdef012345678";
- // make flash write fail exactly in nvs::Page::writeEntryData
- f.emu.failAfter(17);
- {
- nvs::Storage storage(&f.part);
- TEST_ESP_OK(storage.init(0, 3));
- TEST_ESP_ERR(storage.writeItem(1, nvs::ItemType::SZ, "key", str, strlen(str)), ESP_ERR_FLASH_OP_FAIL);
- // check that repeated operations cause an error
- TEST_ESP_ERR(storage.writeItem(1, nvs::ItemType::SZ, "key", str, strlen(str)), ESP_ERR_NVS_INVALID_STATE);
- uint8_t val;
- TEST_ESP_ERR(storage.readItem(1, nvs::ItemType::U8, "key", &val, sizeof(val)), ESP_ERR_NVS_NOT_FOUND);
- }
- {
- // load page and check that data was erased
- nvs::Page p;
- p.load(&f.part, 0);
- CHECK(p.getErasedEntryCount() == 3);
- CHECK(p.getUsedEntryCount() == 0);
- // try to write again
- TEST_ESP_OK(p.writeItem(1, nvs::ItemType::SZ, "key", str, strlen(str)));
- }
- }
- TEST_CASE("crc errors in item header are handled", "[nvs]")
- {
- PartitionEmulationFixture f(0, 3);
- nvs::Storage storage(&f.part);
- // prepare some data
- TEST_ESP_OK(storage.init(0, 3));
- TEST_ESP_OK(storage.writeItem(0, "ns1", static_cast<uint8_t>(1)));
- TEST_ESP_OK(storage.writeItem(1, "value1", static_cast<uint32_t>(1)));
- TEST_ESP_OK(storage.writeItem(1, "value2", static_cast<uint32_t>(2)));
- // corrupt item header
- uint32_t val = 0;
- f.emu.write(32 * 3, &val, 4);
- // check that storage can recover
- TEST_ESP_OK(storage.init(0, 3));
- TEST_ESP_OK(storage.readItem(1, "value2", val));
- CHECK(val == 2);
- // check that the corrupted item is no longer present
- TEST_ESP_ERR(ESP_ERR_NVS_NOT_FOUND, storage.readItem(1, "value1", val));
- // add more items to make the page full
- for (size_t i = 0; i < nvs::Page::ENTRY_COUNT; ++i) {
- char item_name[nvs::Item::MAX_KEY_LENGTH + 1];
- snprintf(item_name, sizeof(item_name), "item_%ld", (long int)i);
- TEST_ESP_OK(storage.writeItem(1, item_name, static_cast<uint32_t>(i)));
- }
- // corrupt another item on the full page
- val = 0;
- f.emu.write(32 * 4, &val, 4);
- // check that storage can recover
- TEST_ESP_OK(storage.init(0, 3));
- // check that the corrupted item is no longer present
- TEST_ESP_ERR(ESP_ERR_NVS_NOT_FOUND, storage.readItem(1, "value2", val));
- }
- TEST_CASE("crc error in variable length item is handled", "[nvs]")
- {
- PartitionEmulationFixture f(0, 3);
- const uint64_t before_val = 0xbef04e;
- const uint64_t after_val = 0xaf7e4;
- // write some data
- {
- nvs::Page p;
- p.load(&f.part, 0);
- TEST_ESP_OK(p.writeItem<uint64_t>(0, "before", before_val));
- const char *str = "foobar";
- TEST_ESP_OK(p.writeItem(0, nvs::ItemType::SZ, "key", str, strlen(str)));
- TEST_ESP_OK(p.writeItem<uint64_t>(0, "after", after_val));
- }
- // corrupt some data
- uint32_t w;
- CHECK(f.emu.read(&w, 32 * 3 + 8, sizeof(w)));
- w &= 0xf000000f;
- CHECK(f.emu.write(32 * 3 + 8, &w, sizeof(w)));
- // load and check
- {
- nvs::Page p;
- p.load(&f.part, 0);
- CHECK(p.getUsedEntryCount() == 2);
- CHECK(p.getErasedEntryCount() == 2);
- uint64_t val;
- TEST_ESP_OK(p.readItem<uint64_t>(0, "before", val));
- CHECK(val == before_val);
- TEST_ESP_ERR(p.findItem(0, nvs::ItemType::SZ, "key"), ESP_ERR_NVS_NOT_FOUND);
- TEST_ESP_OK(p.readItem<uint64_t>(0, "after", val));
- CHECK(val == after_val);
- }
- }
- TEST_CASE("multiple partitions access check", "[nvs]")
- {
- SpiFlashEmulator emu(10);
- PartitionEmulation p0(&emu, 0 * SPI_FLASH_SEC_SIZE, 5 * SPI_FLASH_SEC_SIZE, "nvs1");
- PartitionEmulation p1(&emu, 5 * SPI_FLASH_SEC_SIZE, 5 * SPI_FLASH_SEC_SIZE, "nvs2");
- TEST_ESP_OK( nvs::NVSPartitionManager::get_instance()->init_custom(&p0, 0, 5) );
- TEST_ESP_OK( nvs::NVSPartitionManager::get_instance()->init_custom(&p1, 5, 5) );
- nvs_handle_t handle1, handle2;
- TEST_ESP_OK( nvs_open_from_partition("nvs1", "test", NVS_READWRITE, &handle1) );
- TEST_ESP_OK( nvs_open_from_partition("nvs2", "test", NVS_READWRITE, &handle2) );
- TEST_ESP_OK( nvs_set_i32(handle1, "foo", 0xdeadbeef));
- TEST_ESP_OK( nvs_set_i32(handle2, "foo", 0xcafebabe));
- int32_t v1, v2;
- TEST_ESP_OK( nvs_get_i32(handle1, "foo", &v1));
- TEST_ESP_OK( nvs_get_i32(handle2, "foo", &v2));
- CHECK(v1 == 0xdeadbeef);
- CHECK(v2 == 0xcafebabe);
- TEST_ESP_OK(nvs_flash_deinit_partition(p0.get_partition_name()));
- TEST_ESP_OK(nvs_flash_deinit_partition(p1.get_partition_name()));
- }
- // leaks memory
- TEST_CASE("Recovery from power-off when the entry being erased is not on active page", "[nvs]")
- {
- const size_t blob_size = nvs::Page::CHUNK_MAX_SIZE / 2 ;
- size_t read_size = blob_size;
- uint8_t blob[blob_size] = {0x11};
- PartitionEmulationFixture f(0, 3);
- TEST_ESP_OK( nvs::NVSPartitionManager::get_instance()->init_custom(&f.part, 0, 3) );
- nvs_handle_t handle;
- TEST_ESP_OK( nvs_open("test", NVS_READWRITE, &handle) );
- f.emu.clearStats();
- f.emu.failAfter(nvs::Page::CHUNK_MAX_SIZE / 4 + 75);
- TEST_ESP_OK( nvs_set_blob(handle, "1a", blob, blob_size) );
- TEST_ESP_OK( nvs_set_blob(handle, "1b", blob, blob_size) );
- TEST_ESP_ERR( nvs_erase_key(handle, "1a"), ESP_ERR_FLASH_OP_FAIL );
- TEST_ESP_OK( nvs::NVSPartitionManager::get_instance()->init_custom(&f.part, 0, 3) );
- /* Check 1a is erased fully*/
- TEST_ESP_ERR( nvs_get_blob(handle, "1a", blob, &read_size), ESP_ERR_NVS_NOT_FOUND);
- /* Check 2b is still accessible*/
- TEST_ESP_OK( nvs_get_blob(handle, "1b", blob, &read_size));
- nvs_close(handle);
- TEST_ESP_OK(nvs_flash_deinit_partition(f.part.get_partition_name()));
- }
- // leaks memory
- TEST_CASE("Recovery from power-off when page is being freed.", "[nvs]")
- {
- const size_t blob_size = (nvs::Page::ENTRY_COUNT - 3) * nvs::Page::ENTRY_SIZE;
- size_t read_size = blob_size / 2;
- uint8_t blob[blob_size] = {0};
- PartitionEmulationFixture f(0, 3);
- TEST_ESP_OK( nvs::NVSPartitionManager::get_instance()->init_custom(&f.part, 0, 3));
- nvs_handle_t handle;
- TEST_ESP_OK(nvs_open("test", NVS_READWRITE, &handle));
- // Fill first page
- TEST_ESP_OK(nvs_set_blob(handle, "1a", blob, blob_size / 3));
- TEST_ESP_OK(nvs_set_blob(handle, "1b", blob, blob_size / 3));
- TEST_ESP_OK(nvs_set_blob(handle, "1c", blob, blob_size / 4));
- // Fill second page
- TEST_ESP_OK(nvs_set_blob(handle, "2a", blob, blob_size / 2));
- TEST_ESP_OK(nvs_set_blob(handle, "2b", blob, blob_size / 2));
- TEST_ESP_OK(nvs_erase_key(handle, "1c"));
- f.emu.clearStats();
- f.emu.failAfter(6 * nvs::Page::ENTRY_COUNT);
- TEST_ESP_ERR(nvs_set_blob(handle, "1d", blob, blob_size / 4), ESP_ERR_FLASH_OP_FAIL);
- TEST_ESP_OK( nvs::NVSPartitionManager::get_instance()->init_custom(&f.part, 0, 3));
- read_size = blob_size / 3;
- TEST_ESP_OK( nvs_get_blob(handle, "1a", blob, &read_size));
- TEST_ESP_OK( nvs_get_blob(handle, "1b", blob, &read_size));
- read_size = blob_size / 4;
- TEST_ESP_ERR( nvs_get_blob(handle, "1c", blob, &read_size), ESP_ERR_NVS_NOT_FOUND);
- TEST_ESP_ERR( nvs_get_blob(handle, "1d", blob, &read_size), ESP_ERR_NVS_NOT_FOUND);
- read_size = blob_size / 2;
- TEST_ESP_OK( nvs_get_blob(handle, "2a", blob, &read_size));
- TEST_ESP_OK( nvs_get_blob(handle, "2b", blob, &read_size));
- TEST_ESP_OK(nvs_commit(handle));
- nvs_close(handle);
- TEST_ESP_OK(nvs_flash_deinit_partition(f.part.get_partition_name()));
- }
- TEST_CASE("Check that NVS supports old blob format without blob index", "[nvs]")
- {
- SpiFlashEmulator emu("../nvs_partition_generator/part_old_blob_format.bin");
- PartitionEmulation part(&emu, 0, 2 * SPI_FLASH_SEC_SIZE, "test");
- nvs_handle_t handle;
- TEST_ESP_OK( nvs::NVSPartitionManager::get_instance()->init_custom(&part, 0, 2) );
- TEST_ESP_OK( nvs_open_from_partition("test", "dummyNamespace", NVS_READWRITE, &handle));
- char buf[64] = {0};
- size_t buflen = 64;
- uint8_t hexdata[] = {0x01, 0x02, 0x03, 0xab, 0xcd, 0xef};
- TEST_ESP_OK( nvs_get_blob(handle, "dummyHex2BinKey", buf, &buflen));
- CHECK(memcmp(buf, hexdata, buflen) == 0);
- buflen = 64;
- uint8_t base64data[] = {'1', '2', '3', 'a', 'b', 'c'};
- TEST_ESP_OK( nvs_get_blob(handle, "dummyBase64Key", buf, &buflen));
- CHECK(memcmp(buf, base64data, buflen) == 0);
- nvs::Page p;
- p.load(&part, 0);
- /* Check that item is stored in old format without blob index*/
- TEST_ESP_OK(p.findItem(1, nvs::ItemType::BLOB, "dummyHex2BinKey"));
- /* Modify the blob so that it is stored in the new format*/
- hexdata[0] = hexdata[1] = hexdata[2] = 0x99;
- TEST_ESP_OK(nvs_set_blob(handle, "dummyHex2BinKey", hexdata, sizeof(hexdata)));
- nvs::Page p2;
- p2.load(&part, 0);
- /* Check the type of the blob. Expect type mismatch since the blob is stored in new format*/
- TEST_ESP_ERR(p2.findItem(1, nvs::ItemType::BLOB, "dummyHex2BinKey"), ESP_ERR_NVS_TYPE_MISMATCH);
- /* Check that index is present for the modified blob according to new format*/
- TEST_ESP_OK(p2.findItem(1, nvs::ItemType::BLOB_IDX, "dummyHex2BinKey"));
- /* Read the blob in new format and check the contents*/
- buflen = 64;
- TEST_ESP_OK( nvs_get_blob(handle, "dummyBase64Key", buf, &buflen));
- CHECK(memcmp(buf, base64data, buflen) == 0);
- TEST_ESP_OK(nvs_flash_deinit_partition(part.get_partition_name()));
- }
- static void check_nvs_part_gen_args(SpiFlashEmulator *spi_flash_emulator,
- char const *part_name,
- int size,
- char const *filename,
- bool is_encr,
- nvs_sec_cfg_t *xts_cfg)
- {
- nvs_handle_t handle;
- esp_partition_t esp_part;
- esp_part.encrypted = false; // we're not testing generic flash encryption here, only the legacy NVS encryption
- esp_part.address = 0;
- esp_part.size = size * SPI_FLASH_SEC_SIZE;
- strncpy(esp_part.label, part_name, PART_NAME_MAX_SIZE);
- shared_ptr<nvs::Partition> part;
- if (is_encr) {
- nvs::NVSEncryptedPartition *enc_part = new nvs::NVSEncryptedPartition(&esp_part);
- TEST_ESP_OK(enc_part->init(xts_cfg));
- part.reset(enc_part);
- } else {
- part.reset(new PartitionEmulation(spi_flash_emulator, 0, size, part_name));
- }
- TEST_ESP_OK( nvs::NVSPartitionManager::get_instance()->init_custom(part.get(), 0, size) );
- TEST_ESP_OK( nvs_open_from_partition(part_name, "dummyNamespace", NVS_READONLY, &handle));
- uint8_t u8v;
- TEST_ESP_OK( nvs_get_u8(handle, "dummyU8Key", &u8v));
- CHECK(u8v == 127);
- int8_t i8v;
- TEST_ESP_OK( nvs_get_i8(handle, "dummyI8Key", &i8v));
- CHECK(i8v == -128);
- uint16_t u16v;
- TEST_ESP_OK( nvs_get_u16(handle, "dummyU16Key", &u16v));
- CHECK(u16v == 32768);
- uint32_t u32v;
- TEST_ESP_OK( nvs_get_u32(handle, "dummyU32Key", &u32v));
- CHECK(u32v == 4294967295);
- int32_t i32v;
- TEST_ESP_OK( nvs_get_i32(handle, "dummyI32Key", &i32v));
- CHECK(i32v == -2147483648);
- char string_buf[256];
- const char test_str[] = "Lorem ipsum dolor sit amet, consectetur adipiscing elit.\n"
- "Fusce quis risus justo.\n"
- "Suspendisse egestas in nisi sit amet auctor.\n"
- "Pellentesque rhoncus dictum sodales.\n"
- "In justo erat, viverra at interdum eget, interdum vel dui.";
- size_t str_len = sizeof(test_str);
- TEST_ESP_OK( nvs_get_str(handle, "dummyStringKey", string_buf, &str_len));
- CHECK(strncmp(string_buf, test_str, str_len) == 0);
- char buf[64] = {0};
- uint8_t hexdata[] = {0x01, 0x02, 0x03, 0xab, 0xcd, 0xef};
- size_t buflen = 64;
- int j;
- TEST_ESP_OK( nvs_get_blob(handle, "dummyHex2BinKey", buf, &buflen));
- CHECK(memcmp(buf, hexdata, buflen) == 0);
- uint8_t base64data[] = {'1', '2', '3', 'a', 'b', 'c'};
- TEST_ESP_OK( nvs_get_blob(handle, "dummyBase64Key", buf, &buflen));
- CHECK(memcmp(buf, base64data, buflen) == 0);
- buflen = 64;
- uint8_t hexfiledata[] = {0x01, 0x23, 0x45, 0x67, 0x89, 0xab, 0xcd, 0xef};
- TEST_ESP_OK( nvs_get_blob(handle, "hexFileKey", buf, &buflen));
- CHECK(memcmp(buf, hexfiledata, buflen) == 0);
- buflen = 64;
- uint8_t strfiledata[64] = "abcdefghijklmnopqrstuvwxyz\0";
- TEST_ESP_OK( nvs_get_str(handle, "stringFileKey", buf, &buflen));
- CHECK(memcmp(buf, strfiledata, buflen) == 0);
- char bin_data[5200];
- size_t bin_len = sizeof(bin_data);
- char binfiledata[5200];
- ifstream file;
- file.open(filename);
- file.read(binfiledata,5200);
- TEST_ESP_OK( nvs_get_blob(handle, "binFileKey", bin_data, &bin_len));
- CHECK(memcmp(bin_data, binfiledata, bin_len) == 0);
- file.close();
- nvs_close(handle);
- TEST_ESP_OK(nvs_flash_deinit_partition(part_name));
- }
- static void check_nvs_part_gen_args_mfg(SpiFlashEmulator *spi_flash_emulator,
- char const *part_name,
- int size,
- char const *filename,
- bool is_encr,
- nvs_sec_cfg_t *xts_cfg)
- {
- nvs_handle_t handle;
- esp_partition_t esp_part;
- esp_part.encrypted = false; // we're not testing generic flash encryption here, only the legacy NVS encryption
- esp_part.address = 0;
- esp_part.size = size * SPI_FLASH_SEC_SIZE;
- strncpy(esp_part.label, part_name, PART_NAME_MAX_SIZE);
- shared_ptr<nvs::Partition> part;
- if (is_encr) {
- nvs::NVSEncryptedPartition *enc_part = new nvs::NVSEncryptedPartition(&esp_part);
- TEST_ESP_OK(enc_part->init(xts_cfg));
- part.reset(enc_part);
- } else {
- part.reset(new PartitionEmulation(spi_flash_emulator, 0, size, part_name));
- }
- TEST_ESP_OK( nvs::NVSPartitionManager::get_instance()->init_custom(part.get(), 0, size) );
- TEST_ESP_OK( nvs_open_from_partition(part_name, "dummyNamespace", NVS_READONLY, &handle));
- uint8_t u8v;
- TEST_ESP_OK( nvs_get_u8(handle, "dummyU8Key", &u8v));
- CHECK(u8v == 127);
- int8_t i8v;
- TEST_ESP_OK( nvs_get_i8(handle, "dummyI8Key", &i8v));
- CHECK(i8v == -128);
- uint16_t u16v;
- TEST_ESP_OK( nvs_get_u16(handle, "dummyU16Key", &u16v));
- CHECK(u16v == 32768);
- uint32_t u32v;
- TEST_ESP_OK( nvs_get_u32(handle, "dummyU32Key", &u32v));
- CHECK(u32v == 4294967295);
- int32_t i32v;
- TEST_ESP_OK( nvs_get_i32(handle, "dummyI32Key", &i32v));
- CHECK(i32v == -2147483648);
- char buf[64] = {0};
- size_t buflen = 64;
- TEST_ESP_OK( nvs_get_str(handle, "dummyStringKey", buf, &buflen));
- CHECK(strncmp(buf, "0A:0B:0C:0D:0E:0F", buflen) == 0);
- uint8_t hexdata[] = {0x01, 0x02, 0x03, 0xab, 0xcd, 0xef};
- buflen = 64;
- int j;
- TEST_ESP_OK( nvs_get_blob(handle, "dummyHex2BinKey", buf, &buflen));
- CHECK(memcmp(buf, hexdata, buflen) == 0);
- uint8_t base64data[] = {'1', '2', '3', 'a', 'b', 'c'};
- TEST_ESP_OK( nvs_get_blob(handle, "dummyBase64Key", buf, &buflen));
- CHECK(memcmp(buf, base64data, buflen) == 0);
- buflen = 64;
- uint8_t hexfiledata[] = {0x01, 0x23, 0x45, 0x67, 0x89, 0xab, 0xcd, 0xef};
- TEST_ESP_OK( nvs_get_blob(handle, "hexFileKey", buf, &buflen));
- CHECK(memcmp(buf, hexfiledata, buflen) == 0);
- buflen = 64;
- uint8_t strfiledata[64] = "abcdefghijklmnopqrstuvwxyz\0";
- TEST_ESP_OK( nvs_get_str(handle, "stringFileKey", buf, &buflen));
- CHECK(memcmp(buf, strfiledata, buflen) == 0);
- char bin_data[5200];
- size_t bin_len = sizeof(bin_data);
- char binfiledata[5200];
- ifstream file;
- file.open(filename);
- file.read(binfiledata,5200);
- TEST_ESP_OK( nvs_get_blob(handle, "binFileKey", bin_data, &bin_len));
- CHECK(memcmp(bin_data, binfiledata, bin_len) == 0);
- file.close();
- nvs_close(handle);
- TEST_ESP_OK(nvs_flash_deinit_partition(part_name));
- }
- TEST_CASE("check and read data from partition generated via partition generation utility with multipage blob support disabled", "[nvs_part_gen]")
- {
- int status;
- int childpid = fork();
- if (childpid == 0) {
- exit(execlp("cp", " cp",
- "-rf",
- "../nvs_partition_generator/testdata",
- ".", NULL));
- } else {
- CHECK(childpid > 0);
- waitpid(childpid, &status, 0);
- CHECK(WEXITSTATUS(status) != -1);
- childpid = fork();
- if (childpid == 0) {
- exit(execlp("python", "python",
- "../nvs_partition_generator/nvs_partition_gen.py",
- "generate",
- "../nvs_partition_generator/sample_singlepage_blob.csv",
- "partition_single_page.bin",
- "0x3000",
- "--version",
- "1",
- "--outdir",
- "../nvs_partition_generator", NULL));
- } else {
- CHECK(childpid > 0);
- int status;
- waitpid(childpid, &status, 0);
- CHECK(WEXITSTATUS(status) == 0);
- }
- }
- SpiFlashEmulator emu("../nvs_partition_generator/partition_single_page.bin");
- check_nvs_part_gen_args(&emu, "test", 3, "../nvs_partition_generator/testdata/sample_singlepage_blob.bin", false, NULL);
- childpid = fork();
- if (childpid == 0) {
- exit(execlp("rm", " rm",
- "-rf",
- "testdata", NULL));
- } else {
- CHECK(childpid > 0);
- waitpid(childpid, &status, 0);
- CHECK(WEXITSTATUS(status) == 0);
- }
- }
- TEST_CASE("check and read data from partition generated via partition generation utility with multipage blob support enabled", "[nvs_part_gen]")
- {
- int status;
- int childpid = fork();
- if (childpid == 0) {
- exit(execlp("cp", " cp",
- "-rf",
- "../nvs_partition_generator/testdata",
- ".", NULL));
- } else {
- CHECK(childpid > 0);
- waitpid(childpid, &status, 0);
- CHECK(WEXITSTATUS(status) == 0);
- childpid = fork();
- if (childpid == 0) {
- exit(execlp("python", "python",
- "../nvs_partition_generator/nvs_partition_gen.py",
- "generate",
- "../nvs_partition_generator/sample_multipage_blob.csv",
- "partition_multipage_blob.bin",
- "0x4000",
- "--version",
- "2",
- "--outdir",
- "../nvs_partition_generator", NULL));
- } else {
- CHECK(childpid > 0);
- waitpid(childpid, &status, 0);
- CHECK(WEXITSTATUS(status) == 0);
- }
- }
- SpiFlashEmulator emu("../nvs_partition_generator/partition_multipage_blob.bin");
- check_nvs_part_gen_args(&emu, "test", 4, "../nvs_partition_generator/testdata/sample_multipage_blob.bin", false, NULL);
- childpid = fork();
- if (childpid == 0) {
- exit(execlp("rm", " rm",
- "-rf",
- "testdata", NULL));
- } else {
- CHECK(childpid > 0);
- waitpid(childpid, &status, 0);
- CHECK(WEXITSTATUS(status) == 0);
- }
- }
- TEST_CASE("check and read data from partition generated via manufacturing utility with multipage blob support disabled", "[mfg_gen]")
- {
- int childpid = fork();
- int status;
- if (childpid == 0) {
- exit(execlp("bash", "bash",
- "-c",
- "rm -rf ../../../tools/mass_mfg/host_test && \
- cp -rf ../../../tools/mass_mfg/testdata mfg_testdata && \
- cp -rf ../nvs_partition_generator/testdata . && \
- mkdir -p ../../../tools/mass_mfg/host_test", NULL));
- } else {
- CHECK(childpid > 0);
- waitpid(childpid, &status, 0);
- CHECK(WEXITSTATUS(status) == 0);
- childpid = fork();
- if (childpid == 0) {
- exit(execlp("python", "python",
- "../../../tools/mass_mfg/mfg_gen.py",
- "generate",
- "../../../tools/mass_mfg/samples/sample_config.csv",
- "../../../tools/mass_mfg/samples/sample_values_singlepage_blob.csv",
- "Test",
- "0x3000",
- "--outdir",
- "../../../tools/mass_mfg/host_test",
- "--version",
- "1", NULL));
- } else {
- CHECK(childpid > 0);
- waitpid(childpid, &status, 0);
- CHECK(WEXITSTATUS(status) == 0);
- childpid = fork();
- if (childpid == 0) {
- exit(execlp("python", "python",
- "../nvs_partition_generator/nvs_partition_gen.py",
- "generate",
- "../../../tools/mass_mfg/host_test/csv/Test-1.csv",
- "../nvs_partition_generator/Test-1-partition.bin",
- "0x3000",
- "--version",
- "1", NULL));
- } else {
- CHECK(childpid > 0);
- waitpid(childpid, &status, 0);
- CHECK(WEXITSTATUS(status) == 0);
- }
- }
- }
- SpiFlashEmulator emu1("../../../tools/mass_mfg/host_test/bin/Test-1.bin");
- check_nvs_part_gen_args_mfg(&emu1, "test", 3, "mfg_testdata/sample_singlepage_blob.bin", false, NULL);
- SpiFlashEmulator emu2("../nvs_partition_generator/Test-1-partition.bin");
- check_nvs_part_gen_args_mfg(&emu2, "test", 3, "testdata/sample_singlepage_blob.bin", false, NULL);
- childpid = fork();
- if (childpid == 0) {
- exit(execlp("bash", " bash",
- "-c",
- "rm -rf ../../../tools/mass_mfg/host_test | \
- rm -rf mfg_testdata | \
- rm -rf testdata", NULL));
- } else {
- CHECK(childpid > 0);
- waitpid(childpid, &status, 0);
- CHECK(WEXITSTATUS(status) == 0);
- }
- }
- TEST_CASE("check and read data from partition generated via manufacturing utility with blank lines in csv files and multipage blob support disabled", "[mfg_gen]")
- {
- int childpid = fork();
- int status;
- if (childpid == 0) {
- exit(execlp("bash", "bash",
- "-c",
- "rm -rf ../../../tools/mass_mfg/host_test && \
- cp -rf ../../../tools/mass_mfg/testdata mfg_testdata && \
- cp -rf ../nvs_partition_generator/testdata . && \
- mkdir -p ../../../tools/mass_mfg/host_test", NULL));
- } else {
- CHECK(childpid > 0);
- waitpid(childpid, &status, 0);
- CHECK(WEXITSTATUS(status) == 0);
- childpid = fork();
- if (childpid == 0) {
- exit(execlp("python", "python",
- "../../../tools/mass_mfg/mfg_gen.py",
- "generate",
- "../../../tools/mass_mfg/samples/sample_config_blank_lines.csv",
- "../../../tools/mass_mfg/samples/sample_values_singlepage_blob_blank_lines.csv",
- "Test",
- "0x3000",
- "--outdir",
- "../../../tools/mass_mfg/host_test",
- "--version",
- "1", NULL));
- } else {
- CHECK(childpid > 0);
- waitpid(childpid, &status, 0);
- CHECK(WEXITSTATUS(status) == 0);
- childpid = fork();
- if (childpid == 0) {
- exit(execlp("python", "python",
- "../nvs_partition_generator/nvs_partition_gen.py",
- "generate",
- "../../../tools/mass_mfg/host_test/csv/Test-1.csv",
- "../nvs_partition_generator/Test-1-partition.bin",
- "0x3000",
- "--version",
- "1", NULL));
- } else {
- CHECK(childpid > 0);
- waitpid(childpid, &status, 0);
- CHECK(WEXITSTATUS(status) == 0);
- }
- }
- }
- SpiFlashEmulator emu1("../../../tools/mass_mfg/host_test/bin/Test-1.bin");
- check_nvs_part_gen_args_mfg(&emu1, "test", 3, "mfg_testdata/sample_singlepage_blob.bin", false, NULL);
- SpiFlashEmulator emu2("../nvs_partition_generator/Test-1-partition.bin");
- check_nvs_part_gen_args_mfg(&emu2, "test", 3, "testdata/sample_singlepage_blob.bin", false, NULL);
- childpid = fork();
- if (childpid == 0) {
- exit(execlp("bash", " bash",
- "-c",
- "rm -rf ../../../tools/mass_mfg/host_test | \
- rm -rf mfg_testdata | \
- rm -rf testdata", NULL));
- } else {
- CHECK(childpid > 0);
- waitpid(childpid, &status, 0);
- CHECK(WEXITSTATUS(status) == 0);
- }
- }
- TEST_CASE("check and read data from partition generated via manufacturing utility with multipage blob support enabled", "[mfg_gen]")
- {
- int childpid = fork();
- int status;
- if (childpid == 0) {
- exit(execlp("bash", " bash",
- "-c",
- "rm -rf ../../../tools/mass_mfg/host_test | \
- cp -rf ../../../tools/mass_mfg/testdata mfg_testdata | \
- cp -rf ../nvs_partition_generator/testdata . | \
- mkdir -p ../../../tools/mass_mfg/host_test", NULL));
- } else {
- CHECK(childpid > 0);
- waitpid(childpid, &status, 0);
- CHECK(WEXITSTATUS(status) == 0);
- childpid = fork();
- if (childpid == 0) {
- exit(execlp("python", "python",
- "../../../tools/mass_mfg/mfg_gen.py",
- "generate",
- "../../../tools/mass_mfg/samples/sample_config.csv",
- "../../../tools/mass_mfg/samples/sample_values_multipage_blob.csv",
- "Test",
- "0x4000",
- "--outdir",
- "../../../tools/mass_mfg/host_test",
- "--version",
- "2", NULL));
- } else {
- CHECK(childpid > 0);
- waitpid(childpid, &status, 0);
- CHECK(WEXITSTATUS(status) == 0);
- childpid = fork();
- if (childpid == 0) {
- exit(execlp("python", "python",
- "../nvs_partition_generator/nvs_partition_gen.py",
- "generate",
- "../../../tools/mass_mfg/host_test/csv/Test-1.csv",
- "../nvs_partition_generator/Test-1-partition.bin",
- "0x4000",
- "--version",
- "2", NULL));
- } else {
- CHECK(childpid > 0);
- waitpid(childpid, &status, 0);
- CHECK(WEXITSTATUS(status) == 0);
- }
- }
- }
- SpiFlashEmulator emu1("../../../tools/mass_mfg/host_test/bin/Test-1.bin");
- check_nvs_part_gen_args_mfg(&emu1, "test", 4, "mfg_testdata/sample_multipage_blob.bin", false, NULL);
- SpiFlashEmulator emu2("../nvs_partition_generator/Test-1-partition.bin");
- check_nvs_part_gen_args_mfg(&emu2, "test", 4, "testdata/sample_multipage_blob.bin", false, NULL);
- childpid = fork();
- if (childpid == 0) {
- exit(execlp("bash", " bash",
- "-c",
- "rm -rf ../../../tools/mass_mfg/host_test | \
- rm -rf mfg_testdata | \
- rm -rf testdata", NULL));
- } else {
- CHECK(childpid > 0);
- waitpid(childpid, &status, 0);
- CHECK(WEXITSTATUS(status) == 0);
- }
- }
- #if CONFIG_NVS_ENCRYPTION
- TEST_CASE("check underlying xts code for 32-byte size sector encryption", "[nvs]")
- {
- auto toHex = [](char ch) {
- if(ch >= '0' && ch <= '9')
- return ch - '0';
- else if(ch >= 'a' && ch <= 'f')
- return ch - 'a' + 10;
- else if(ch >= 'A' && ch <= 'F')
- return ch - 'A' + 10;
- else
- return 0;
- };
- auto toHexByte = [toHex](char* c) {
- return 16 * toHex(c[0]) + toHex(c[1]);
- };
- auto toHexStream = [toHexByte](char* src, uint8_t* dest) {
- uint32_t cnt =0;
- char* p = src;
- while(*p != '\0' && *(p + 1) != '\0')
- {
- dest[cnt++] = toHexByte(p); p += 2;
- }
- };
- uint8_t eky_hex[2 * NVS_KEY_SIZE];
- uint8_t ptxt_hex[nvs::Page::ENTRY_SIZE], ctxt_hex[nvs::Page::ENTRY_SIZE], ba_hex[16];
- mbedtls_aes_xts_context ectx[1];
- mbedtls_aes_xts_context dctx[1];
- char eky[][2 * NVS_KEY_SIZE + 1] = {
- "0000000000000000000000000000000000000000000000000000000000000000",
- "1111111111111111111111111111111111111111111111111111111111111111"
- };
- char tky[][2 * NVS_KEY_SIZE + 1] = {
- "0000000000000000000000000000000000000000000000000000000000000000",
- "2222222222222222222222222222222222222222222222222222222222222222"
- };
- char blk_addr[][2*16 + 1] = {
- "00000000000000000000000000000000",
- "33333333330000000000000000000000"
- };
- char ptxt[][2 * nvs::Page::ENTRY_SIZE + 1] = {
- "0000000000000000000000000000000000000000000000000000000000000000",
- "4444444444444444444444444444444444444444444444444444444444444444"
- };
- char ctxt[][2 * nvs::Page::ENTRY_SIZE + 1] = {
- "d456b4fc2e620bba6ffbed27b956c9543454dd49ebd8d8ee6f94b65cbe158f73",
- "e622334f184bbce129a25b2ac76b3d92abf98e22df5bdd15af471f3db8946a85"
- };
- mbedtls_aes_xts_init(ectx);
- mbedtls_aes_xts_init(dctx);
- for(uint8_t cnt = 0; cnt < sizeof(eky)/sizeof(eky[0]); cnt++) {
- toHexStream(eky[cnt], eky_hex);
- toHexStream(tky[cnt], &eky_hex[NVS_KEY_SIZE]);
- toHexStream(ptxt[cnt], ptxt_hex);
- toHexStream(ctxt[cnt], ctxt_hex);
- toHexStream(blk_addr[cnt], ba_hex);
- CHECK(!mbedtls_aes_xts_setkey_enc(ectx, eky_hex, 2 * NVS_KEY_SIZE * 8));
- CHECK(!mbedtls_aes_xts_setkey_enc(dctx, eky_hex, 2 * NVS_KEY_SIZE * 8));
- CHECK(!mbedtls_aes_crypt_xts(ectx, MBEDTLS_AES_ENCRYPT, nvs::Page::ENTRY_SIZE, ba_hex, ptxt_hex, ptxt_hex));
- CHECK(!memcmp(ptxt_hex, ctxt_hex, nvs::Page::ENTRY_SIZE));
- }
- }
- TEST_CASE("test nvs apis with encryption enabled", "[nvs]")
- {
- nvs_handle_t handle_1;
- const uint32_t NVS_FLASH_SECTOR = 6;
- const uint32_t NVS_FLASH_SECTOR_COUNT_MIN = 3;
- nvs_sec_cfg_t xts_cfg;
- for(int count = 0; count < NVS_KEY_SIZE; count++) {
- xts_cfg.eky[count] = 0x11;
- xts_cfg.tky[count] = 0x22;
- }
- EncryptedPartitionFixture fixture(&xts_cfg, NVS_FLASH_SECTOR, NVS_FLASH_SECTOR_COUNT_MIN);
- fixture.emu.randomize(100);
- fixture.emu.setBounds(NVS_FLASH_SECTOR, NVS_FLASH_SECTOR + NVS_FLASH_SECTOR_COUNT_MIN);
- for (uint16_t i = NVS_FLASH_SECTOR; i <NVS_FLASH_SECTOR + NVS_FLASH_SECTOR_COUNT_MIN; ++i) {
- fixture.emu.erase(i);
- }
- TEST_ESP_OK( nvs::NVSPartitionManager::get_instance()->
- init_custom(&fixture.part, NVS_FLASH_SECTOR, NVS_FLASH_SECTOR_COUNT_MIN));
- TEST_ESP_ERR(nvs_open("namespace1", NVS_READONLY, &handle_1), ESP_ERR_NVS_NOT_FOUND);
- TEST_ESP_OK(nvs_open("namespace1", NVS_READWRITE, &handle_1));
- TEST_ESP_OK(nvs_set_i32(handle_1, "foo", 0x12345678));
- TEST_ESP_OK(nvs_set_i32(handle_1, "foo", 0x23456789));
- nvs_handle_t handle_2;
- TEST_ESP_OK(nvs_open("namespace2", NVS_READWRITE, &handle_2));
- TEST_ESP_OK(nvs_set_i32(handle_2, "foo", 0x3456789a));
- const char* str = "value 0123456789abcdef0123456789abcdef";
- TEST_ESP_OK(nvs_set_str(handle_2, "key", str));
- int32_t v1;
- TEST_ESP_OK(nvs_get_i32(handle_1, "foo", &v1));
- CHECK(0x23456789 == v1);
- int32_t v2;
- TEST_ESP_OK(nvs_get_i32(handle_2, "foo", &v2));
- CHECK(0x3456789a == v2);
- char buf[strlen(str) + 1];
- size_t buf_len = sizeof(buf);
- size_t buf_len_needed;
- TEST_ESP_OK(nvs_get_str(handle_2, "key", NULL, &buf_len_needed));
- CHECK(buf_len_needed == buf_len);
- size_t buf_len_short = buf_len - 1;
- TEST_ESP_ERR(ESP_ERR_NVS_INVALID_LENGTH, nvs_get_str(handle_2, "key", buf, &buf_len_short));
- CHECK(buf_len_short == buf_len);
- size_t buf_len_long = buf_len + 1;
- TEST_ESP_OK(nvs_get_str(handle_2, "key", buf, &buf_len_long));
- CHECK(buf_len_long == buf_len);
- TEST_ESP_OK(nvs_get_str(handle_2, "key", buf, &buf_len));
- CHECK(0 == strcmp(buf, str));
- nvs_close(handle_1);
- nvs_close(handle_2);
- TEST_ESP_OK(nvs_flash_deinit());
- }
- TEST_CASE("test nvs apis for nvs partition generator utility with encryption enabled", "[nvs_part_gen]")
- {
- int status;
- int childpid = fork();
- if (childpid == 0) {
- exit(execlp("cp", " cp",
- "-rf",
- "../nvs_partition_generator/testdata",
- ".", NULL));
- } else {
- CHECK(childpid > 0);
- waitpid(childpid, &status, 0);
- CHECK(WEXITSTATUS(status) == 0);
- childpid = fork();
- if (childpid == 0) {
- exit(execlp("python", "python",
- "../nvs_partition_generator/nvs_partition_gen.py",
- "encrypt",
- "../nvs_partition_generator/sample_multipage_blob.csv",
- "partition_encrypted.bin",
- "0x4000",
- "--inputkey",
- "../nvs_partition_generator/testdata/sample_encryption_keys.bin",
- "--outdir",
- "../nvs_partition_generator", NULL));
- } else {
- CHECK(childpid > 0);
- waitpid(childpid, &status, 0);
- CHECK(WEXITSTATUS(status) == 0);
- }
- }
- SpiFlashEmulator emu("../nvs_partition_generator/partition_encrypted.bin");
- nvs_sec_cfg_t cfg;
- for (int count = 0; count < NVS_KEY_SIZE; count++) {
- cfg.eky[count] = 0x11;
- cfg.tky[count] = 0x22;
- }
- check_nvs_part_gen_args(&emu, NVS_DEFAULT_PART_NAME, 4, "../nvs_partition_generator/testdata/sample_multipage_blob.bin", true, &cfg);
- childpid = fork();
- if (childpid == 0) {
- exit(execlp("rm", " rm",
- "-rf",
- "testdata", NULL));
- } else {
- CHECK(childpid > 0);
- waitpid(childpid, &status, 0);
- CHECK(WEXITSTATUS(status) == 0);
- }
- }
- TEST_CASE("test decrypt functionality for encrypted data", "[nvs_part_gen]")
- {
- //retrieving the temporary test data
- int status = system("cp -rf ../nvs_partition_generator/testdata .");
- CHECK(status == 0);
- //encoding data from sample_multipage_blob.csv
- status = system("python ../nvs_partition_generator/nvs_partition_gen.py generate ../nvs_partition_generator/sample_multipage_blob.csv partition_encoded.bin 0x5000 --outdir ../nvs_partition_generator");
- CHECK(status == 0);
- //encrypting data from sample_multipage_blob.csv
- status = system("python ../nvs_partition_generator/nvs_partition_gen.py encrypt ../nvs_partition_generator/sample_multipage_blob.csv partition_encrypted.bin 0x5000 --inputkey ../nvs_partition_generator/testdata/sample_encryption_keys.bin --outdir ../nvs_partition_generator");
- CHECK(status == 0);
- //decrypting data from partition_encrypted.bin
- status = system("python ../nvs_partition_generator/nvs_partition_gen.py decrypt ../nvs_partition_generator/partition_encrypted.bin ../nvs_partition_generator/testdata/sample_encryption_keys.bin ../nvs_partition_generator/partition_decrypted.bin");
- CHECK(status == 0);
- status = system("diff ../nvs_partition_generator/partition_decrypted.bin ../nvs_partition_generator/partition_encoded.bin");
- CHECK(status == 0);
- CHECK(WEXITSTATUS(status) == 0);
- //cleaning up the temporary test data
- status = system("rm -rf testdata");
- CHECK(status == 0);
- }
- TEST_CASE("test nvs apis for nvs partition generator utility with encryption enabled using keygen", "[nvs_part_gen]")
- {
- int childpid = fork();
- int status;
- if (childpid == 0) {
- exit(execlp("cp", " cp",
- "-rf",
- "../nvs_partition_generator/testdata",
- ".", NULL));
- } else {
- CHECK(childpid > 0);
- waitpid(childpid, &status, 0);
- CHECK(WEXITSTATUS(status) == 0);
- childpid = fork();
- if (childpid == 0) {
- exit(execlp("rm", " rm",
- "-rf",
- "../nvs_partition_generator/keys", NULL));
- } else {
- CHECK(childpid > 0);
- waitpid(childpid, &status, 0);
- CHECK(WEXITSTATUS(status) == 0);
- childpid = fork();
- if (childpid == 0) {
- exit(execlp("python", "python",
- "../nvs_partition_generator/nvs_partition_gen.py",
- "encrypt",
- "../nvs_partition_generator/sample_multipage_blob.csv",
- "partition_encrypted_using_keygen.bin",
- "0x4000",
- "--keygen",
- "--outdir",
- "../nvs_partition_generator", NULL));
- } else {
- CHECK(childpid > 0);
- waitpid(childpid, &status, 0);
- CHECK(WEXITSTATUS(status) == 0);
- }
- }
- }
- DIR *dir;
- struct dirent *file;
- char *filename;
- char *files;
- char *file_ext;
- dir = opendir("../nvs_partition_generator/keys");
- while ((file = readdir(dir)) != NULL) {
- filename = file->d_name;
- files = strrchr(filename, '.');
- if (files != NULL) {
- file_ext = files + 1;
- if (strncmp(file_ext, "bin", 3) == 0) {
- break;
- }
- }
- }
- std::string encr_file = std::string("../nvs_partition_generator/keys/") + std::string(filename);
- SpiFlashEmulator emu("../nvs_partition_generator/partition_encrypted_using_keygen.bin");
- char buffer[64];
- FILE *fp;
- fp = fopen(encr_file.c_str(), "rb");
- fread(buffer, sizeof(buffer), 1, fp);
- fclose(fp);
- nvs_sec_cfg_t cfg;
- for (int count = 0; count < NVS_KEY_SIZE; count++) {
- cfg.eky[count] = buffer[count] & 255;
- cfg.tky[count] = buffer[count + 32] & 255;
- }
- check_nvs_part_gen_args(&emu, NVS_DEFAULT_PART_NAME, 4, "../nvs_partition_generator/testdata/sample_multipage_blob.bin", true, &cfg);
- }
- TEST_CASE("test nvs apis for nvs partition generator utility with encryption enabled using inputkey", "[nvs_part_gen]")
- {
- int childpid = fork();
- int status;
- DIR *dir;
- struct dirent *file;
- char *filename;
- char *files;
- char *file_ext;
- dir = opendir("../nvs_partition_generator/keys");
- while ((file = readdir(dir)) != NULL) {
- filename = file->d_name;
- files = strrchr(filename, '.');
- if (files != NULL) {
- file_ext = files + 1;
- if (strncmp(file_ext, "bin", 3) == 0) {
- break;
- }
- }
- }
- std::string encr_file = std::string("../nvs_partition_generator/keys/") + std::string(filename);
- if (childpid == 0) {
- exit(execlp("python", "python",
- "../nvs_partition_generator/nvs_partition_gen.py",
- "encrypt",
- "../nvs_partition_generator/sample_multipage_blob.csv",
- "partition_encrypted_using_keyfile.bin",
- "0x4000",
- "--inputkey",
- encr_file.c_str(),
- "--outdir",
- "../nvs_partition_generator", NULL));
- } else {
- CHECK(childpid > 0);
- waitpid(childpid, &status, 0);
- CHECK(WEXITSTATUS(status) == 0);
- }
- SpiFlashEmulator emu("../nvs_partition_generator/partition_encrypted_using_keyfile.bin");
- char buffer[64];
- FILE *fp;
- fp = fopen(encr_file.c_str(), "rb");
- fread(buffer, sizeof(buffer), 1, fp);
- fclose(fp);
- nvs_sec_cfg_t cfg;
- for (int count = 0; count < NVS_KEY_SIZE; count++) {
- cfg.eky[count] = buffer[count] & 255;
- cfg.tky[count] = buffer[count + 32] & 255;
- }
- check_nvs_part_gen_args(&emu, NVS_DEFAULT_PART_NAME, 4, "../nvs_partition_generator/testdata/sample_multipage_blob.bin", true, &cfg);
- childpid = fork();
- if (childpid == 0) {
- exit(execlp("rm", " rm",
- "-rf",
- "../nvs_partition_generator/keys", NULL));
- } else {
- CHECK(childpid > 0);
- waitpid(childpid, &status, 0);
- CHECK(WEXITSTATUS(status) == 0);
- childpid = fork();
- if (childpid == 0) {
- exit(execlp("rm", " rm",
- "-rf",
- "testdata", NULL));
- } else {
- CHECK(childpid > 0);
- waitpid(childpid, &status, 0);
- CHECK(WEXITSTATUS(status) == 0);
- }
- }
- }
- TEST_CASE("check and read data from partition generated via manufacturing utility with encryption enabled using sample inputkey", "[mfg_gen]")
- {
- int childpid = fork();
- int status;
- if (childpid == 0) {
- exit(execlp("bash", " bash",
- "-c",
- "rm -rf ../../../tools/mass_mfg/host_test | \
- cp -rf ../../../tools/mass_mfg/testdata mfg_testdata | \
- cp -rf ../nvs_partition_generator/testdata . | \
- mkdir -p ../../../tools/mass_mfg/host_test", NULL));
- } else {
- CHECK(childpid > 0);
- waitpid(childpid, &status, 0);
- CHECK(WEXITSTATUS(status) == 0);
- childpid = fork();
- if (childpid == 0) {
- exit(execlp("python", "python",
- "../../../tools/mass_mfg/mfg_gen.py",
- "generate",
- "../../../tools/mass_mfg/samples/sample_config.csv",
- "../../../tools/mass_mfg/samples/sample_values_multipage_blob.csv",
- "Test",
- "0x4000",
- "--outdir",
- "../../../tools/mass_mfg/host_test",
- "--version",
- "2",
- "--inputkey",
- "mfg_testdata/sample_encryption_keys.bin", NULL));
- } else {
- CHECK(childpid > 0);
- waitpid(childpid, &status, 0);
- CHECK(WEXITSTATUS(status) == 0);
- childpid = fork();
- if (childpid == 0) {
- exit(execlp("python", "python",
- "../nvs_partition_generator/nvs_partition_gen.py",
- "encrypt",
- "../../../tools/mass_mfg/host_test/csv/Test-1.csv",
- "../nvs_partition_generator/Test-1-partition-encrypted.bin",
- "0x4000",
- "--version",
- "2",
- "--inputkey",
- "testdata/sample_encryption_keys.bin", NULL));
- } else {
- CHECK(childpid > 0);
- waitpid(childpid, &status, 0);
- CHECK(WEXITSTATUS(status) == 0);
- }
- }
- }
- SpiFlashEmulator emu1("../../../tools/mass_mfg/host_test/bin/Test-1.bin");
- nvs_sec_cfg_t cfg;
- for (int count = 0; count < NVS_KEY_SIZE; count++) {
- cfg.eky[count] = 0x11;
- cfg.tky[count] = 0x22;
- }
- check_nvs_part_gen_args_mfg(&emu1, NVS_DEFAULT_PART_NAME, 4, "mfg_testdata/sample_multipage_blob.bin", true, &cfg);
- SpiFlashEmulator emu2("../nvs_partition_generator/Test-1-partition-encrypted.bin");
- check_nvs_part_gen_args_mfg(&emu2, NVS_DEFAULT_PART_NAME, 4, "testdata/sample_multipage_blob.bin", true, &cfg);
- childpid = fork();
- if (childpid == 0) {
- exit(execlp("bash", " bash",
- "-c",
- "rm -rf ../../../tools/mass_mfg/host_test | \
- rm -rf mfg_testdata | \
- rm -rf testdata", NULL));
- } else {
- CHECK(childpid > 0);
- waitpid(childpid, &status, 0);
- CHECK(WEXITSTATUS(status) == 0);
- }
- }
- TEST_CASE("check and read data from partition generated via manufacturing utility with encryption enabled using new generated key", "[mfg_gen]")
- {
- int childpid = fork();
- int status;
- if (childpid == 0) {
- exit(execlp("bash", " bash",
- "-c",
- "rm -rf ../../../tools/mass_mfg/host_test | \
- cp -rf ../../../tools/mass_mfg/testdata mfg_testdata | \
- cp -rf ../nvs_partition_generator/testdata . | \
- mkdir -p ../../../tools/mass_mfg/host_test", NULL));
- } else {
- CHECK(childpid > 0);
- waitpid(childpid, &status, 0);
- CHECK(WEXITSTATUS(status) == 0);
- childpid = fork();
- if (childpid == 0) {
- exit(execlp("python", "python",
- "../../../tools/mass_mfg/mfg_gen.py",
- "generate-key",
- "--outdir",
- "../../../tools/mass_mfg/host_test",
- "--keyfile",
- "encr_keys_host_test.bin", NULL));
- } else {
- CHECK(childpid > 0);
- waitpid(childpid, &status, 0);
- CHECK(WEXITSTATUS(status) == 0);
- childpid = fork();
- if (childpid == 0) {
- exit(execlp("python", "python",
- "../../../tools/mass_mfg/mfg_gen.py",
- "generate",
- "../../../tools/mass_mfg/samples/sample_config.csv",
- "../../../tools/mass_mfg/samples/sample_values_multipage_blob.csv",
- "Test",
- "0x4000",
- "--outdir",
- "../../../tools/mass_mfg/host_test",
- "--version",
- "2",
- "--inputkey",
- "../../../tools/mass_mfg/host_test/keys/encr_keys_host_test.bin", NULL));
- } else {
- CHECK(childpid > 0);
- waitpid(childpid, &status, 0);
- CHECK(WEXITSTATUS(status) == 0);
- childpid = fork();
- if (childpid == 0) {
- exit(execlp("python", "python",
- "../nvs_partition_generator/nvs_partition_gen.py",
- "encrypt",
- "../../../tools/mass_mfg/host_test/csv/Test-1.csv",
- "../nvs_partition_generator/Test-1-partition-encrypted.bin",
- "0x4000",
- "--version",
- "2",
- "--inputkey",
- "../../../tools/mass_mfg/host_test/keys/encr_keys_host_test.bin", NULL));
- } else {
- CHECK(childpid > 0);
- waitpid(childpid, &status, 0);
- CHECK(WEXITSTATUS(status) == 0);
- }
- }
- }
- }
- SpiFlashEmulator emu1("../../../tools/mass_mfg/host_test/bin/Test-1.bin");
- char buffer[64];
- FILE *fp;
- fp = fopen("../../../tools/mass_mfg/host_test/keys/encr_keys_host_test.bin", "rb");
- fread(buffer, sizeof(buffer), 1, fp);
- fclose(fp);
- nvs_sec_cfg_t cfg;
- for (int count = 0; count < NVS_KEY_SIZE; count++) {
- cfg.eky[count] = buffer[count] & 255;
- cfg.tky[count] = buffer[count + 32] & 255;
- }
- check_nvs_part_gen_args_mfg(&emu1, NVS_DEFAULT_PART_NAME, 4, "mfg_testdata/sample_multipage_blob.bin", true, &cfg);
- SpiFlashEmulator emu2("../nvs_partition_generator/Test-1-partition-encrypted.bin");
- check_nvs_part_gen_args_mfg(&emu2, NVS_DEFAULT_PART_NAME, 4, "testdata/sample_multipage_blob.bin", true, &cfg);
- childpid = fork();
- if (childpid == 0) {
- exit(execlp("bash", " bash",
- "-c",
- "rm -rf keys | \
- rm -rf mfg_testdata | \
- rm -rf testdata | \
- rm -rf ../../../tools/mass_mfg/host_test", NULL));
- } else {
- CHECK(childpid > 0);
- waitpid(childpid, &status, 0);
- CHECK(WEXITSTATUS(status) == 0);
- }
- }
- #endif
- /* Add new tests above */
- /* This test has to be the final one */
- TEST_CASE("dump all performance data", "[nvs]")
- {
- std::cout << "====================" << std::endl << "Dumping benchmarks" << std::endl;
- std::cout << s_perf.str() << std::endl;
- std::cout << "====================" << std::endl;
- }
|