| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772 |
- // Copyright 2015-2016 Espressif Systems (Shanghai) PTE LTD
- //
- // Licensed under the Apache License, Version 2.0 (the "License");
- // you may not use this file except in compliance with the License.
- // You may obtain a copy of the License at
- // http://www.apache.org/licenses/LICENSE-2.0
- //
- // Unless required by applicable law or agreed to in writing, software
- // distributed under the License is distributed on an "AS IS" BASIS,
- // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- // See the License for the specific language governing permissions and
- // limitations under the License.
- #include "nvs_storage.hpp"
- #ifndef ESP_PLATFORM
- #include <map>
- #include <sstream>
- #endif
- namespace nvs
- {
- Storage::~Storage()
- {
- clearNamespaces();
- }
- void Storage::clearNamespaces()
- {
- mNamespaces.clearAndFreeNodes();
- }
- void Storage::populateBlobIndices(TBlobIndexList& blobIdxList)
- {
- for (auto it = mPageManager.begin(); it != mPageManager.end(); ++it) {
- Page& p = *it;
- size_t itemIndex = 0;
- Item item;
- /* If the power went off just after writing a blob index, the duplicate detection
- * logic in pagemanager will remove the earlier index. So we should never find a
- * duplicate index at this point */
- while (p.findItem(Page::NS_ANY, ItemType::BLOB_IDX, nullptr, itemIndex, item) == ESP_OK) {
- BlobIndexNode* entry = new BlobIndexNode;
- item.getKey(entry->key, sizeof(entry->key));
- entry->nsIndex = item.nsIndex;
- entry->chunkStart = item.blobIndex.chunkStart;
- entry->chunkCount = item.blobIndex.chunkCount;
- blobIdxList.push_back(entry);
- itemIndex += item.span;
- }
- }
- }
- void Storage::eraseOrphanDataBlobs(TBlobIndexList& blobIdxList)
- {
- for (auto it = mPageManager.begin(); it != mPageManager.end(); ++it) {
- Page& p = *it;
- size_t itemIndex = 0;
- Item item;
- /* Chunks with same <ns,key> and with chunkIndex in the following ranges
- * belong to same family.
- * 1) VER_0_OFFSET <= chunkIndex < VER_1_OFFSET-1 => Version0 chunks
- * 2) VER_1_OFFSET <= chunkIndex < VER_ANY => Version1 chunks
- */
- while (p.findItem(Page::NS_ANY, ItemType::BLOB_DATA, nullptr, itemIndex, item) == ESP_OK) {
- auto iter = std::find_if(blobIdxList.begin(),
- blobIdxList.end(),
- [=] (const BlobIndexNode& e) -> bool
- {return (strncmp(item.key, e.key, sizeof(e.key) - 1) == 0)
- && (item.nsIndex == e.nsIndex)
- && (item.chunkIndex >= static_cast<uint8_t> (e.chunkStart))
- && (item.chunkIndex < static_cast<uint8_t> (e.chunkStart) + e.chunkCount);});
- if (iter == std::end(blobIdxList)) {
- p.eraseItem(item.nsIndex, item.datatype, item.key, item.chunkIndex);
- }
- itemIndex += item.span;
- }
- }
- }
- esp_err_t Storage::init(uint32_t baseSector, uint32_t sectorCount)
- {
- auto err = mPageManager.load(baseSector, sectorCount);
- if (err != ESP_OK) {
- mState = StorageState::INVALID;
- return err;
- }
- // load namespaces list
- clearNamespaces();
- std::fill_n(mNamespaceUsage.data(), mNamespaceUsage.byteSize() / 4, 0);
- for (auto it = mPageManager.begin(); it != mPageManager.end(); ++it) {
- Page& p = *it;
- size_t itemIndex = 0;
- Item item;
- while (p.findItem(Page::NS_INDEX, ItemType::U8, nullptr, itemIndex, item) == ESP_OK) {
- NamespaceEntry* entry = new NamespaceEntry;
- item.getKey(entry->mName, sizeof(entry->mName));
- item.getValue(entry->mIndex);
- mNamespaces.push_back(entry);
- mNamespaceUsage.set(entry->mIndex, true);
- itemIndex += item.span;
- }
- }
- mNamespaceUsage.set(0, true);
- mNamespaceUsage.set(255, true);
- mState = StorageState::ACTIVE;
- // Populate list of multi-page index entries.
- TBlobIndexList blobIdxList;
- populateBlobIndices(blobIdxList);
- // Remove the entries for which there is no parent multi-page index.
- eraseOrphanDataBlobs(blobIdxList);
- // Purge the blob index list
- blobIdxList.clearAndFreeNodes();
- #ifndef ESP_PLATFORM
- debugCheck();
- #endif
- return ESP_OK;
- }
- bool Storage::isValid() const
- {
- return mState == StorageState::ACTIVE;
- }
- esp_err_t Storage::findItem(uint8_t nsIndex, ItemType datatype, const char* key, Page* &page, Item& item, uint8_t chunkIdx, VerOffset chunkStart)
- {
- for (auto it = std::begin(mPageManager); it != std::end(mPageManager); ++it) {
- size_t itemIndex = 0;
- auto err = it->findItem(nsIndex, datatype, key, itemIndex, item, chunkIdx, chunkStart);
- if (err == ESP_OK) {
- page = it;
- return ESP_OK;
- }
- }
- return ESP_ERR_NVS_NOT_FOUND;
- }
- esp_err_t Storage::writeMultiPageBlob(uint8_t nsIndex, const char* key, const void* data, size_t dataSize, VerOffset chunkStart)
- {
- uint8_t chunkCount = 0;
- TUsedPageList usedPages;
- size_t remainingSize = dataSize;
- size_t offset=0;
- esp_err_t err = ESP_OK;
- /* Check how much maximum data can be accommodated**/
- uint32_t max_pages = mPageManager.getPageCount() - 1;
- if(max_pages > (Page::CHUNK_ANY-1)/2) {
- max_pages = (Page::CHUNK_ANY-1)/2;
- }
- if (dataSize > max_pages * Page::CHUNK_MAX_SIZE) {
- return ESP_ERR_NVS_VALUE_TOO_LONG;
- }
- do {
- Page& page = getCurrentPage();
- size_t tailroom = page.getVarDataTailroom();
- size_t chunkSize =0;
- if (!chunkCount && tailroom < dataSize && tailroom < Page::CHUNK_MAX_SIZE/10) {
- /** This is the first chunk and tailroom is too small ***/
- if (page.state() != Page::PageState::FULL) {
- err = page.markFull();
- if (err != ESP_OK) {
- return err;
- }
- }
- err = mPageManager.requestNewPage();
- if (err != ESP_OK) {
- return err;
- } else if(getCurrentPage().getVarDataTailroom() == tailroom) {
- /* We got the same page or we are not improving.*/
- return ESP_ERR_NVS_NOT_ENOUGH_SPACE;
- } else {
- continue;
- }
- } else if (!tailroom) {
- err = ESP_ERR_NVS_NOT_ENOUGH_SPACE;
- break;
- }
- /* Split the blob into two and store the chunk of available size onto the current page */
- assert(tailroom!=0);
- chunkSize = (remainingSize > tailroom)? tailroom : remainingSize;
- remainingSize -= chunkSize;
- err = page.writeItem(nsIndex, ItemType::BLOB_DATA, key,
- static_cast<const uint8_t*> (data) + offset, chunkSize, static_cast<uint8_t> (chunkStart) + chunkCount);
- chunkCount++;
- assert(err != ESP_ERR_NVS_PAGE_FULL);
- if (err != ESP_OK) {
- break;
- } else {
- UsedPageNode* node = new UsedPageNode();
- node->mPage = &page;
- usedPages.push_back(node);
- if (remainingSize || (tailroom - chunkSize) < Page::ENTRY_SIZE) {
- if (page.state() != Page::PageState::FULL) {
- err = page.markFull();
- if (err != ESP_OK) {
- break;
- }
- }
- err = mPageManager.requestNewPage();
- if (err != ESP_OK) {
- break;
- }
- }
- }
- offset += chunkSize;
- if (!remainingSize) {
- /* All pages are stored. Now store the index.*/
- Item item;
- std::fill_n(item.data, sizeof(item.data), 0xff);
- item.blobIndex.dataSize = dataSize;
- item.blobIndex.chunkCount = chunkCount;
- item.blobIndex.chunkStart = chunkStart;
- err = getCurrentPage().writeItem(nsIndex, ItemType::BLOB_IDX, key, item.data, sizeof(item.data));
- assert(err != ESP_ERR_NVS_PAGE_FULL);
- break;
- }
- } while (1);
- if (err != ESP_OK) {
- /* Anything failed, then we should erase all the written chunks*/
- int ii=0;
- for (auto it = std::begin(usedPages); it != std::end(usedPages); it++) {
- it->mPage->eraseItem(nsIndex, ItemType::BLOB_DATA, key, ii++);
- }
- }
- usedPages.clearAndFreeNodes();
- return err;
- }
- esp_err_t Storage::writeItem(uint8_t nsIndex, ItemType datatype, const char* key, const void* data, size_t dataSize)
- {
- if (mState != StorageState::ACTIVE) {
- return ESP_ERR_NVS_NOT_INITIALIZED;
- }
- Page* findPage = nullptr;
- Item item;
- esp_err_t err;
- if (datatype == ItemType::BLOB) {
- err = findItem(nsIndex, ItemType::BLOB_IDX, key, findPage, item);
- } else {
- err = findItem(nsIndex, datatype, key, findPage, item);
- }
- if (err != ESP_OK && err != ESP_ERR_NVS_NOT_FOUND) {
- return err;
- }
- if (datatype == ItemType::BLOB) {
- VerOffset prevStart, nextStart;
- prevStart = nextStart = VerOffset::VER_0_OFFSET;
- if (findPage) {
- // Do a sanity check that the item in question is actually being modified.
- // If it isn't, it is cheaper to purposefully not write out new data.
- // since it may invoke an erasure of flash.
- if (cmpMultiPageBlob(nsIndex, key, data, dataSize) == ESP_OK) {
- return ESP_OK;
- }
- if (findPage->state() == Page::PageState::UNINITIALIZED ||
- findPage->state() == Page::PageState::INVALID) {
- ESP_ERROR_CHECK(findItem(nsIndex, datatype, key, findPage, item));
- }
- /* Get the version of the previous index with same <ns,key> */
- prevStart = item.blobIndex.chunkStart;
- assert(prevStart == VerOffset::VER_0_OFFSET || prevStart == VerOffset::VER_1_OFFSET);
- /* Toggle the version by changing the offset */
- nextStart
- = (prevStart == VerOffset::VER_1_OFFSET) ? VerOffset::VER_0_OFFSET : VerOffset::VER_1_OFFSET;
- }
- /* Write the blob with new version*/
- err = writeMultiPageBlob(nsIndex, key, data, dataSize, nextStart);
- if (err == ESP_ERR_NVS_PAGE_FULL) {
- return ESP_ERR_NVS_NOT_ENOUGH_SPACE;
- }
- if (err != ESP_OK) {
- return err;
- }
- if (findPage) {
- /* Erase the blob with earlier version*/
- err = eraseMultiPageBlob(nsIndex, key, prevStart);
- if (err == ESP_ERR_FLASH_OP_FAIL) {
- return ESP_ERR_NVS_REMOVE_FAILED;
- }
- if (err != ESP_OK) {
- return err;
- }
- findPage = nullptr;
- } else {
- /* Support for earlier versions where BLOBS were stored without index */
- err = findItem(nsIndex, datatype, key, findPage, item);
- if (err != ESP_OK && err != ESP_ERR_NVS_NOT_FOUND) {
- return err;
- }
- }
- } else {
- // Do a sanity check that the item in question is actually being modified.
- // If it isn't, it is cheaper to purposefully not write out new data.
- // since it may invoke an erasure of flash.
- if (findPage != nullptr &&
- findPage->cmpItem(nsIndex, datatype, key, data, dataSize) == ESP_OK) {
- return ESP_OK;
- }
- Page& page = getCurrentPage();
- err = page.writeItem(nsIndex, datatype, key, data, dataSize);
- if (err == ESP_ERR_NVS_PAGE_FULL) {
- if (page.state() != Page::PageState::FULL) {
- err = page.markFull();
- if (err != ESP_OK) {
- return err;
- }
- }
- err = mPageManager.requestNewPage();
- if (err != ESP_OK) {
- return err;
- }
- err = getCurrentPage().writeItem(nsIndex, datatype, key, data, dataSize);
- if (err == ESP_ERR_NVS_PAGE_FULL) {
- return ESP_ERR_NVS_NOT_ENOUGH_SPACE;
- }
- if (err != ESP_OK) {
- return err;
- }
- } else if (err != ESP_OK) {
- return err;
- }
- }
- if (findPage) {
- if (findPage->state() == Page::PageState::UNINITIALIZED ||
- findPage->state() == Page::PageState::INVALID) {
- ESP_ERROR_CHECK(findItem(nsIndex, datatype, key, findPage, item));
- }
- err = findPage->eraseItem(nsIndex, datatype, key);
- if (err == ESP_ERR_FLASH_OP_FAIL) {
- return ESP_ERR_NVS_REMOVE_FAILED;
- }
- if (err != ESP_OK) {
- return err;
- }
- }
- #ifndef ESP_PLATFORM
- debugCheck();
- #endif
- return ESP_OK;
- }
- esp_err_t Storage::createOrOpenNamespace(const char* nsName, bool canCreate, uint8_t& nsIndex)
- {
- if (mState != StorageState::ACTIVE) {
- return ESP_ERR_NVS_NOT_INITIALIZED;
- }
- auto it = std::find_if(mNamespaces.begin(), mNamespaces.end(), [=] (const NamespaceEntry& e) -> bool {
- return strncmp(nsName, e.mName, sizeof(e.mName) - 1) == 0;
- });
- if (it == std::end(mNamespaces)) {
- if (!canCreate) {
- return ESP_ERR_NVS_NOT_FOUND;
- }
- uint8_t ns;
- for (ns = 1; ns < 255; ++ns) {
- if (mNamespaceUsage.get(ns) == false) {
- break;
- }
- }
- if (ns == 255) {
- return ESP_ERR_NVS_NOT_ENOUGH_SPACE;
- }
- auto err = writeItem(Page::NS_INDEX, ItemType::U8, nsName, &ns, sizeof(ns));
- if (err != ESP_OK) {
- return err;
- }
- mNamespaceUsage.set(ns, true);
- nsIndex = ns;
- NamespaceEntry* entry = new NamespaceEntry;
- entry->mIndex = ns;
- strncpy(entry->mName, nsName, sizeof(entry->mName) - 1);
- entry->mName[sizeof(entry->mName) - 1] = 0;
- mNamespaces.push_back(entry);
- } else {
- nsIndex = it->mIndex;
- }
- return ESP_OK;
- }
- esp_err_t Storage::readMultiPageBlob(uint8_t nsIndex, const char* key, void* data, size_t dataSize)
- {
- Item item;
- Page* findPage = nullptr;
- /* First read the blob index */
- auto err = findItem(nsIndex, ItemType::BLOB_IDX, key, findPage, item);
- if (err != ESP_OK) {
- return err;
- }
- uint8_t chunkCount = item.blobIndex.chunkCount;
- VerOffset chunkStart = item.blobIndex.chunkStart;
- size_t readSize = item.blobIndex.dataSize;
- size_t offset = 0;
- assert(dataSize == readSize);
- /* Now read corresponding chunks */
- for (uint8_t chunkNum = 0; chunkNum < chunkCount; chunkNum++) {
- err = findItem(nsIndex, ItemType::BLOB_DATA, key, findPage, item, static_cast<uint8_t> (chunkStart) + chunkNum);
- if (err != ESP_OK) {
- if (err == ESP_ERR_NVS_NOT_FOUND) {
- break;
- }
- return err;
- }
- err = findPage->readItem(nsIndex, ItemType::BLOB_DATA, key, static_cast<uint8_t*>(data) + offset, item.varLength.dataSize, static_cast<uint8_t> (chunkStart) + chunkNum);
- if (err != ESP_OK) {
- return err;
- }
- assert(static_cast<uint8_t> (chunkStart) + chunkNum == item.chunkIndex);
- offset += item.varLength.dataSize;
- }
- if (err == ESP_OK) {
- assert(offset == dataSize);
- }
- if (err == ESP_ERR_NVS_NOT_FOUND) {
- eraseMultiPageBlob(nsIndex, key); // cleanup if a chunk is not found
- }
- return err;
- }
- esp_err_t Storage::cmpMultiPageBlob(uint8_t nsIndex, const char* key, const void* data, size_t dataSize)
- {
- Item item;
- Page* findPage = nullptr;
- /* First read the blob index */
- auto err = findItem(nsIndex, ItemType::BLOB_IDX, key, findPage, item);
- if (err != ESP_OK) {
- return err;
- }
- uint8_t chunkCount = item.blobIndex.chunkCount;
- VerOffset chunkStart = item.blobIndex.chunkStart;
- size_t readSize = item.blobIndex.dataSize;
- size_t offset = 0;
- if (dataSize != readSize) {
- return ESP_ERR_NVS_CONTENT_DIFFERS;
- }
- /* Now read corresponding chunks */
- for (uint8_t chunkNum = 0; chunkNum < chunkCount; chunkNum++) {
- err = findItem(nsIndex, ItemType::BLOB_DATA, key, findPage, item, static_cast<uint8_t> (chunkStart) + chunkNum);
- if (err != ESP_OK) {
- if (err == ESP_ERR_NVS_NOT_FOUND) {
- break;
- }
- return err;
- }
- err = findPage->cmpItem(nsIndex, ItemType::BLOB_DATA, key, static_cast<const uint8_t*>(data) + offset, item.varLength.dataSize, static_cast<uint8_t> (chunkStart) + chunkNum);
- if (err != ESP_OK) {
- return err;
- }
- assert(static_cast<uint8_t> (chunkStart) + chunkNum == item.chunkIndex);
- offset += item.varLength.dataSize;
- }
- if (err == ESP_OK) {
- assert(offset == dataSize);
- }
- return err;
- }
- esp_err_t Storage::readItem(uint8_t nsIndex, ItemType datatype, const char* key, void* data, size_t dataSize)
- {
- if (mState != StorageState::ACTIVE) {
- return ESP_ERR_NVS_NOT_INITIALIZED;
- }
- Item item;
- Page* findPage = nullptr;
- if (datatype == ItemType::BLOB) {
- auto err = readMultiPageBlob(nsIndex, key, data, dataSize);
- if (err != ESP_ERR_NVS_NOT_FOUND) {
- return err;
- } // else check if the blob is stored with earlier version format without index
- }
- auto err = findItem(nsIndex, datatype, key, findPage, item);
- if (err != ESP_OK) {
- return err;
- }
- return findPage->readItem(nsIndex, datatype, key, data, dataSize);
- }
- esp_err_t Storage::eraseMultiPageBlob(uint8_t nsIndex, const char* key, VerOffset chunkStart)
- {
- if (mState != StorageState::ACTIVE) {
- return ESP_ERR_NVS_NOT_INITIALIZED;
- }
- Item item;
- Page* findPage = nullptr;
- auto err = findItem(nsIndex, ItemType::BLOB_IDX, key, findPage, item, Page::CHUNK_ANY, chunkStart);
- if (err != ESP_OK) {
- return err;
- }
- /* Erase the index first and make children blobs orphan*/
- err = findPage->eraseItem(nsIndex, ItemType::BLOB_IDX, key, Page::CHUNK_ANY, chunkStart);
- if (err != ESP_OK) {
- return err;
- }
- uint8_t chunkCount = item.blobIndex.chunkCount;
- if (chunkStart == VerOffset::VER_ANY) {
- chunkStart = item.blobIndex.chunkStart;
- } else {
- assert(chunkStart == item.blobIndex.chunkStart);
- }
- /* Now erase corresponding chunks*/
- for (uint8_t chunkNum = 0; chunkNum < chunkCount; chunkNum++) {
- err = findItem(nsIndex, ItemType::BLOB_DATA, key, findPage, item, static_cast<uint8_t> (chunkStart) + chunkNum);
- if (err != ESP_OK && err != ESP_ERR_NVS_NOT_FOUND) {
- return err;
- } else if (err == ESP_ERR_NVS_NOT_FOUND) {
- continue; // Keep erasing other chunks
- }
- err = findPage->eraseItem(nsIndex, ItemType::BLOB_DATA, key, static_cast<uint8_t> (chunkStart) + chunkNum);
- if (err != ESP_OK) {
- return err;
- }
- }
- return ESP_OK;
- }
- esp_err_t Storage::eraseItem(uint8_t nsIndex, ItemType datatype, const char* key)
- {
- if (mState != StorageState::ACTIVE) {
- return ESP_ERR_NVS_NOT_INITIALIZED;
- }
- if (datatype == ItemType::BLOB) {
- return eraseMultiPageBlob(nsIndex, key);
- }
- Item item;
- Page* findPage = nullptr;
- auto err = findItem(nsIndex, datatype, key, findPage, item);
- if (err != ESP_OK) {
- return err;
- }
- if (item.datatype == ItemType::BLOB_DATA || item.datatype == ItemType::BLOB_IDX) {
- return eraseMultiPageBlob(nsIndex, key);
- }
- return findPage->eraseItem(nsIndex, datatype, key);
- }
- esp_err_t Storage::eraseNamespace(uint8_t nsIndex)
- {
- if (mState != StorageState::ACTIVE) {
- return ESP_ERR_NVS_NOT_INITIALIZED;
- }
- for (auto it = std::begin(mPageManager); it != std::end(mPageManager); ++it) {
- while (true) {
- auto err = it->eraseItem(nsIndex, ItemType::ANY, nullptr);
- if (err == ESP_ERR_NVS_NOT_FOUND) {
- break;
- }
- else if (err != ESP_OK) {
- return err;
- }
- }
- }
- return ESP_OK;
- }
- esp_err_t Storage::getItemDataSize(uint8_t nsIndex, ItemType datatype, const char* key, size_t& dataSize)
- {
- if (mState != StorageState::ACTIVE) {
- return ESP_ERR_NVS_NOT_INITIALIZED;
- }
- Item item;
- Page* findPage = nullptr;
- auto err = findItem(nsIndex, datatype, key, findPage, item);
- if (err != ESP_OK) {
- if (datatype != ItemType::BLOB) {
- return err;
- }
- err = findItem(nsIndex, ItemType::BLOB_IDX, key, findPage, item);
- if (err != ESP_OK) {
- return err;
- }
- dataSize = item.blobIndex.dataSize;
- return ESP_OK;
- }
- dataSize = item.varLength.dataSize;
- return ESP_OK;
- }
- void Storage::debugDump()
- {
- for (auto p = mPageManager.begin(); p != mPageManager.end(); ++p) {
- p->debugDump();
- }
- }
- #ifndef ESP_PLATFORM
- void Storage::debugCheck()
- {
- std::map<std::string, Page*> keys;
- for (auto p = mPageManager.begin(); p != mPageManager.end(); ++p) {
- size_t itemIndex = 0;
- size_t usedCount = 0;
- Item item;
- while (p->findItem(Page::NS_ANY, ItemType::ANY, nullptr, itemIndex, item) == ESP_OK) {
- std::stringstream keyrepr;
- keyrepr << static_cast<unsigned>(item.nsIndex) << "_" << static_cast<unsigned>(item.datatype) << "_" << item.key <<"_"<<static_cast<unsigned>(item.chunkIndex);
- std::string keystr = keyrepr.str();
- if (keys.find(keystr) != std::end(keys)) {
- printf("Duplicate key: %s\n", keystr.c_str());
- debugDump();
- assert(0);
- }
- keys.insert(std::make_pair(keystr, static_cast<Page*>(p)));
- itemIndex += item.span;
- usedCount += item.span;
- }
- assert(usedCount == p->getUsedEntryCount());
- }
- }
- #endif //ESP_PLATFORM
- esp_err_t Storage::fillStats(nvs_stats_t& nvsStats)
- {
- nvsStats.namespace_count = mNamespaces.size();
- return mPageManager.fillStats(nvsStats);
- }
- esp_err_t Storage::calcEntriesInNamespace(uint8_t nsIndex, size_t& usedEntries)
- {
- usedEntries = 0;
- if (mState != StorageState::ACTIVE) {
- return ESP_ERR_NVS_NOT_INITIALIZED;
- }
- for (auto it = std::begin(mPageManager); it != std::end(mPageManager); ++it) {
- size_t itemIndex = 0;
- Item item;
- while (true) {
- auto err = it->findItem(nsIndex, ItemType::ANY, nullptr, itemIndex, item);
- if (err == ESP_ERR_NVS_NOT_FOUND) {
- break;
- }
- else if (err != ESP_OK) {
- return err;
- }
- usedEntries += item.span;
- itemIndex += item.span;
- if (itemIndex >= it->ENTRY_COUNT) break;
- }
- }
- return ESP_OK;
- }
- void Storage::fillEntryInfo(Item &item, nvs_entry_info_t &info)
- {
- info.type = static_cast<nvs_type_t>(item.datatype);
- strncpy(info.key, item.key, sizeof(info.key));
- for (auto &name : mNamespaces) {
- if(item.nsIndex == name.mIndex) {
- strncpy(info.namespace_name, name.mName, sizeof(info.namespace_name));
- break;
- }
- }
- }
- bool Storage::findEntry(nvs_opaque_iterator_t* it, const char* namespace_name)
- {
- it->entryIndex = 0;
- it->nsIndex = Page::NS_ANY;
- it->page = mPageManager.begin();
- if (namespace_name != nullptr) {
- if(createOrOpenNamespace(namespace_name, false, it->nsIndex) != ESP_OK) {
- return false;
- }
- }
- return nextEntry(it);
- }
- inline bool isIterableItem(Item& item)
- {
- return (item.nsIndex != 0 &&
- item.datatype != ItemType::BLOB &&
- item.datatype != ItemType::BLOB_IDX);
- }
- inline bool isMultipageBlob(Item& item)
- {
- return (item.datatype == ItemType::BLOB_DATA &&
- !(item.chunkIndex == static_cast<uint8_t>(VerOffset::VER_0_OFFSET)
- || item.chunkIndex == static_cast<uint8_t>(VerOffset::VER_1_OFFSET)));
- }
- bool Storage::nextEntry(nvs_opaque_iterator_t* it)
- {
- Item item;
- esp_err_t err;
- for (auto page = it->page; page != mPageManager.end(); ++page) {
- do {
- err = page->findItem(it->nsIndex, (ItemType)it->type, nullptr, it->entryIndex, item);
- it->entryIndex += item.span;
- if(err == ESP_OK && isIterableItem(item) && !isMultipageBlob(item)) {
- fillEntryInfo(item, it->entry_info);
- it->page = page;
- return true;
- }
- } while (err != ESP_ERR_NVS_NOT_FOUND);
- it->entryIndex = 0;
- }
- return false;
- }
- }
|