raw_sequential.cpp

#include "raw_sequential.hpp"
 
#include <cstddef>
#include <cstdint>
 
#include "libxr_def.hpp"
#include "libxr_mem.hpp"
 
using namespace LibXR;
 
DatabaseRawSequential::DatabaseRawSequential(Flash& flash, size_t max_buffer_size)
    : flash_(flash), max_buffer_size_(max_buffer_size)
{
  ASSERT(flash.MinEraseSize() * 2 <= flash.Size());
  ASSERT(max_buffer_size <= flash.Size() / 2);
  auto block_num = static_cast<size_t>(flash.Size() / flash.MinEraseSize());
  block_size_ = block_num / 2 * flash.MinEraseSize();
  buffer_ = new uint8_t[max_buffer_size];
 
  Init();
}
 
DatabaseRawSequential::KeyInfo::KeyInfo() : raw_data(0xFFFFFFFF) {}
 
DatabaseRawSequential::KeyInfo::KeyInfo(bool nextKey, uint8_t nameLength,
                                        uint32_t dataSize)
    : raw_data(0)
{
  SetNextKeyExist(nextKey);
  SetNameLength(nameLength);
  SetDataSize(dataSize);
}
 
void DatabaseRawSequential::KeyInfo::SetNextKeyExist(bool value)
{
  raw_data = (raw_data & 0x7FFFFFFF) | (static_cast<uint32_t>(value & 0x1) << 31);
}
 
bool DatabaseRawSequential::KeyInfo::GetNextKeyExist() const
{
  return (raw_data >> 31) & 0x1;
}
 
void DatabaseRawSequential::KeyInfo::SetNameLength(uint8_t len)
{
  raw_data = (raw_data & 0x80FFFFFF) | (static_cast<uint32_t>(len & 0x7F) << 24);
}
 
uint8_t DatabaseRawSequential::KeyInfo::GetNameLength() const
{
  return (raw_data >> 24) & 0x7F;
}
 
void DatabaseRawSequential::KeyInfo::SetDataSize(uint32_t size)
{
  raw_data = (raw_data & 0xFF000000) | (size & 0x00FFFFFF);
}
 
uint32_t DatabaseRawSequential::KeyInfo::GetDataSize() const
{
  return raw_data & 0x00FFFFFF;
}
 
void DatabaseRawSequential::ReadFlashOrExit(size_t offset, RawData data)
{
  REQUIRE(flash_.Read(offset, data) == ErrorCode::OK);
}
 
void DatabaseRawSequential::WriteFlashOrExit(size_t offset, ConstRawData data)
{
  REQUIRE(flash_.Write(offset, data) == ErrorCode::OK);
}
 
void DatabaseRawSequential::EraseFlashOrExit(size_t offset, size_t size)
{
  REQUIRE(flash_.Erase(offset, size) == ErrorCode::OK);
}
 
void DatabaseRawSequential::Init()
{
  Memory::FastSet(buffer_, 0xFF, max_buffer_size_);
  FlashInfo* flash_data_ = reinterpret_cast<FlashInfo*>(buffer_);
  flash_data_->header = FLASH_HEADER;
  flash_data_->key = {0, 0, 0};
  buffer_[max_buffer_size_ - 1] = CHECKSUM_BYTE;
 
  if (!IsBlockInited(BlockType::BACKUP) || IsBlockError(BlockType::BACKUP))
  {
    InitBlock(BlockType::BACKUP);
  }
 
  if (IsBlockError(BlockType::MAIN))
  {
    if (IsBlockEmpty(BlockType::BACKUP))
    {
      InitBlock(BlockType::MAIN);
    }
    else
    {
      ReadFlashOrExit(block_size_, {buffer_, max_buffer_size_});
      EraseFlashOrExit(0, block_size_);
      WriteFlashOrExit(0, {buffer_, max_buffer_size_});
    }
  }
 
  Load();
}
 
void DatabaseRawSequential::Save()
{
  EraseFlashOrExit(block_size_, block_size_);
  WriteFlashOrExit(block_size_, {buffer_, max_buffer_size_});
 
  EraseFlashOrExit(0, block_size_);
  WriteFlashOrExit(0, {buffer_, max_buffer_size_});
}
 
void DatabaseRawSequential::Load() { ReadFlashOrExit(0, {buffer_, max_buffer_size_}); }
 
void DatabaseRawSequential::Restore()
{
  Memory::FastSet(buffer_, 0xFF, max_buffer_size_);
  FlashInfo* flash_data_ = reinterpret_cast<FlashInfo*>(buffer_);
  flash_data_->header = FLASH_HEADER;
  flash_data_->key = {0, 0, 0};
  buffer_[max_buffer_size_ - 1] = CHECKSUM_BYTE;
  EraseFlashOrExit(0, block_size_);
  EraseFlashOrExit(block_size_, block_size_);
  WriteFlashOrExit(0, {buffer_, max_buffer_size_});
  WriteFlashOrExit(block_size_, {buffer_, max_buffer_size_});
}
 
ErrorCode DatabaseRawSequential::Get(Database::KeyBase& key)
{
  auto ans = SearchKey(key.name_);
  if (!ans)
  {
    return ErrorCode::NOT_FOUND;
  }
 
  KeyInfo key_buffer;
  ReadFlashOrExit(ans, key_buffer);
 
  if (key.raw_data_.size_ != key_buffer.GetDataSize())
  {
    return ErrorCode::FAILED;
  }
 
  GetKeyData(ans, key.raw_data_);
 
  return ErrorCode::OK;
}
 
void DatabaseRawSequential::InitBlock(BlockType block)
{
  size_t offset = 0;
  if (block == BlockType::BACKUP)
  {
    offset = block_size_;
  }
 
  EraseFlashOrExit(offset, block_size_);
  WriteFlashOrExit(offset, {buffer_, max_buffer_size_});
}
 
bool DatabaseRawSequential::IsBlockInited(BlockType block)
{
  size_t offset = 0;
  if (block == BlockType::BACKUP)
  {
    offset = block_size_;
  }
  FlashInfo flash_data_;
  ReadFlashOrExit(offset, flash_data_);
  return flash_data_.header == FLASH_HEADER;
}
 
bool DatabaseRawSequential::IsBlockEmpty(BlockType block)
{
  size_t offset = 0;
  if (block == BlockType::BACKUP)
  {
    offset = block_size_;
  }
  FlashInfo flash_data_;
  ReadFlashOrExit(offset, flash_data_);
  return flash_data_.key.GetNameLength() == 0;
}
 
bool DatabaseRawSequential::IsBlockError(BlockType block)
{
  size_t offset = 0;
  if (block == BlockType::BACKUP)
  {
    offset = block_size_;
  }
  uint8_t checksum_byte = 0;
  ReadFlashOrExit(offset + max_buffer_size_ / sizeof(CHECKSUM_BYTE) - 1, checksum_byte);
  return checksum_byte != CHECKSUM_BYTE;
}
 
bool DatabaseRawSequential::HasLastKey(size_t offset)
{
  KeyInfo key;
  ReadFlashOrExit(offset, key);
  return key.GetNextKeyExist();
}
 
size_t DatabaseRawSequential::GetKeySize(size_t offset)
{
  KeyInfo key;
  ReadFlashOrExit(offset, key);
  return sizeof(KeyInfo) + key.GetNameLength() + key.GetDataSize();
}
 
size_t DatabaseRawSequential::GetNextKey(size_t offset)
{
  return offset + GetKeySize(offset);
}
 
size_t DatabaseRawSequential::GetLastKey(BlockType block)
{
  if (IsBlockEmpty(block))
  {
    return 0;
  }
 
  size_t offset = LibXR::OffsetOf(&FlashInfo::key);
  while (HasLastKey(offset))
  {
    offset = GetNextKey(offset);
  }
  return offset;
}
 
void DatabaseRawSequential::SetNestKeyExist(size_t offset, bool exist)
{
  KeyInfo key;
  ReadFlashOrExit(offset, key);
  key.SetNextKeyExist(exist);
  LibXR::Memory::FastCopy(buffer_ + offset, &key, sizeof(KeyInfo));
}
 
bool DatabaseRawSequential::KeyDataCompare(size_t offset, const void* data, size_t size)
{
  KeyInfo key;
  ReadFlashOrExit(offset, key);
  size_t key_data_offset = offset + sizeof(KeyInfo) + key.GetNameLength();
  uint8_t data_buffer;
  for (size_t i = 0; i < size; i++)
  {
    ReadFlashOrExit(key_data_offset + i, data_buffer);
    if (data_buffer != ((uint8_t*)data)[i])
    {
      return true;
    }
  }
  return false;
}
 
bool DatabaseRawSequential::KeyNameCompare(size_t offset, const char* name)
{
  KeyInfo key;
  ReadFlashOrExit(offset, key);
  for (size_t i = 0; i < key.GetNameLength(); i++)
  {
    uint8_t data_buffer;
    ReadFlashOrExit(offset + sizeof(KeyInfo) + i, data_buffer);
    if (data_buffer != name[i])
    {
      return true;
    }
  }
  return false;
}
 
ErrorCode DatabaseRawSequential::AddKey(const char* name, const void* data, size_t size)
{
  if (auto ans = SearchKey(name))
  {
    return SetKey(ans, data, size);
  }
 
  const uint32_t NAME_LEN = strlen(name) + 1;
  size_t last_key_offset = GetLastKey(BlockType::MAIN);
  size_t key_buf_offset =
      last_key_offset ? GetNextKey(last_key_offset) : LibXR::OffsetOf(&FlashInfo::key);
 
  size_t end_pos_offset = key_buf_offset + sizeof(KeyInfo) + NAME_LEN + size;
  if (end_pos_offset > max_buffer_size_ - 1)
  {
    return ErrorCode::FULL;
  }
 
  size_t data_ptr_offset = key_buf_offset + sizeof(KeyInfo);
  LibXR::Memory::FastCopy(buffer_ + data_ptr_offset, name, NAME_LEN);
  LibXR::Memory::FastCopy(buffer_ + data_ptr_offset + NAME_LEN, data, size);
 
  KeyInfo key_buf = {0, static_cast<uint8_t>(NAME_LEN), static_cast<uint32_t>(size)};
  LibXR::Memory::FastCopy(buffer_ + key_buf_offset, &key_buf, sizeof(KeyInfo));
  if (last_key_offset)
  {
    SetNestKeyExist(last_key_offset, 1);
  }
 
  Save();
  return ErrorCode::OK;
}
 
ErrorCode DatabaseRawSequential::SetKey(const char* name, const void* data, size_t size)
{
  if (size_t key_offset = SearchKey(name))
  {
    return SetKey(key_offset, data, size);
  }
  return ErrorCode::FAILED;
}
 
ErrorCode DatabaseRawSequential::SetKey(size_t offset, const void* data, size_t size)
{
  ASSERT(offset != 0);
 
  KeyInfo key;
  ReadFlashOrExit(offset, key);
 
  if (key.GetDataSize() == size)
  {
    if (KeyDataCompare(offset, data, size))
    {
      LibXR::Memory::FastCopy(buffer_ + offset + sizeof(KeyInfo) + key.GetNameLength(),
                              data, size);
      Save();
    }
    return ErrorCode::OK;
  }
 
  return ErrorCode::FAILED;
}
 
ErrorCode DatabaseRawSequential::GetKeyData(size_t offset, RawData data)
{
  KeyInfo key;
  ReadFlashOrExit(offset, key);
  if (key.GetDataSize() > data.size_)
  {
    return ErrorCode::FAILED;
  }
  auto data_offset = offset + sizeof(KeyInfo) + key.GetNameLength();
  ReadFlashOrExit(data_offset, {data.addr_, key.GetDataSize()});
  return ErrorCode::OK;
}
 
size_t DatabaseRawSequential::SearchKey(const char* name)
{
  if (IsBlockEmpty(BlockType::MAIN))
  {
    return 0;
  }
 
  size_t key_offset = LibXR::OffsetOf(&FlashInfo::key);
  while (true)
  {
    if (KeyNameCompare(key_offset, name) == 0)
    {
      return key_offset;
    }
    if (!HasLastKey(key_offset))
    {
      break;
    }
    key_offset = GetNextKey(key_offset);
  }
  return 0;
}