blob: 3e3f1d1881c8237dd8c4e8e7b07ea4cb557ba1d2 [file] [log] [blame]
#include "pnor_partition_table.hpp"
#include "common.h"
#include "config.h"
#include "xyz/openbmc_project/Common/error.hpp"
#include <phosphor-logging/elog-errors.hpp>
#include <syslog.h>
#include <endian.h>
#include <regex>
#include <fstream>
#include <algorithm>
namespace openpower
{
namespace virtual_pnor
{
using namespace phosphor::logging;
using namespace sdbusplus::xyz::openbmc_project::Common::Error;
namespace partition
{
Table::Table(size_t blockSize, size_t pnorSize):
Table(fs::path(PARTITION_FILES_RO_LOC), blockSize, pnorSize)
{
}
Table::Table(fs::path&& directory,
size_t blockSize, size_t pnorSize):
szBlocks(0),
imgBlocks(0),
directory(std::move(directory)),
numParts(0),
blockSize(blockSize),
pnorSize(pnorSize)
{
preparePartitions();
prepareHeader();
hostTbl = endianFixup(tbl);
}
void Table::prepareHeader()
{
decltype(auto) table = getNativeTable();
table.data.magic = PARTITION_HEADER_MAGIC;
table.data.version = PARTITION_VERSION_1;
table.data.size = szBlocks;
table.data.entry_size = sizeof(pnor_partition);
table.data.entry_count = numParts;
table.data.block_size = blockSize;
table.data.block_count = pnorSize / blockSize;
table.checksum = details::checksum(table.data);
}
inline void Table::allocateMemory(const fs::path& tocFile)
{
size_t num = 0;
std::string line;
std::ifstream file(tocFile.c_str());
// Find number of lines in partition file - this will help
// determine the number of partitions and hence also how much
// memory to allocate for the partitions array.
// The actual number of partitions may turn out to be lesser than this,
// in case of errors.
while (std::getline(file, line))
{
// Check if line starts with "partition"
if (std::string::npos != line.find("partition", 0))
{
++num;
}
}
size_t totalSizeBytes = sizeof(pnor_partition_table) +
(num * sizeof(pnor_partition));
size_t totalSizeAligned = align_up(totalSizeBytes, blockSize);
szBlocks = totalSizeAligned / blockSize;
imgBlocks = szBlocks;
tbl.resize(totalSizeAligned);
}
inline void Table::writeSizes(pnor_partition& part, size_t start, size_t end)
{
size_t size = end - start;
part.data.base = imgBlocks;
size_t sizeInBlocks = align_up(size, blockSize) / blockSize;
imgBlocks += sizeInBlocks;
part.data.size = sizeInBlocks;
// If a a patch partition file exists, populate actual size with its file
// size if it is smaller than the total size.
fs::path patchFile(PARTITION_FILES_PATCH_LOC);
patchFile /= part.data.name;
if (fs::is_regular_file(patchFile))
{
part.data.actual = std::min(
size, static_cast<size_t>(fs::file_size(patchFile)));
}
else
{
part.data.actual = size;
}
}
inline void Table::writeUserdata(pnor_partition& part,
uint32_t version,
const std::string& data)
{
if (std::string::npos != data.find("ECC"))
{
part.data.user.data[0] = PARTITION_ECC_PROTECTED;
}
auto perms = 0;
if (std::string::npos != data.find("READONLY"))
{
perms |= PARTITION_READONLY;
}
if (std::string::npos != data.find("PRESERVED"))
{
perms |= PARTITION_PRESERVED;
}
part.data.user.data[1] = perms;
part.data.user.data[1] |= version;
}
inline void Table::writeDefaults(pnor_partition& part)
{
part.data.pid = PARENT_PATITION_ID;
part.data.type = PARTITION_TYPE_DATA;
part.data.flags = 0; // flags unused
}
inline void Table::writeNameAndId(pnor_partition& part, std::string&& name,
const std::string& id)
{
name.resize(PARTITION_NAME_MAX);
memcpy(part.data.name,
name.c_str(),
sizeof(part.data.name));
part.data.id = std::stoul(id);
}
void Table::preparePartitions()
{
fs::path tocFile = directory;
tocFile /= PARTITION_TOC_FILE;
allocateMemory(tocFile);
std::ifstream file(tocFile.c_str());
static constexpr auto ID_MATCH = 1;
static constexpr auto NAME_MATCH = 2;
static constexpr auto START_ADDR_MATCH = 4;
static constexpr auto END_ADDR_MATCH = 6;
static constexpr auto VERSION_MATCH = 8;
// Parse PNOR toc (table of contents) file, which has lines like :
// partition01=HBB,0x00010000,0x000a0000,0x80,ECC,PRESERVED, to indicate
// partition information
std::regex regex
{
"^partition([0-9]+)=([A-Za-z0-9_]+),"
"(0x)?([0-9a-fA-F]+),(0x)?([0-9a-fA-F]+),(0x)?([A-Fa-f0-9]{2})",
std::regex::extended
};
std::smatch match;
std::string line;
constexpr auto versionShift = 24;
decltype(auto) table = getNativeTable();
while (std::getline(file, line))
{
if (std::regex_search(line, match, regex))
{
fs::path partitionFile = directory;
partitionFile /= match[NAME_MATCH].str();
if (!fs::exists(partitionFile))
{
MSG_ERR("Partition file %s does not exist",
partitionFile.c_str());
continue;
}
writeNameAndId(table.partitions[numParts],
match[NAME_MATCH].str(),
match[ID_MATCH].str());
writeDefaults(table.partitions[numParts]);
writeSizes(table.partitions[numParts],
std::stoul(match[START_ADDR_MATCH].str(), nullptr, 16),
std::stoul(match[END_ADDR_MATCH].str(), nullptr, 16));
writeUserdata(
table.partitions[numParts],
std::stoul(match[VERSION_MATCH].str(), nullptr, 16) <<
versionShift, // For eg, convert "80" to 0x80000000
match.suffix().str());
table.partitions[numParts].checksum =
details::checksum(table.partitions[numParts].data);
++numParts;
}
}
}
const pnor_partition& Table::partition(size_t offset) const
{
const decltype(auto) table = getNativeTable();
size_t offt = offset / blockSize;
for (decltype(numParts) i{}; i < numParts; ++i)
{
if ((offt >= table.partitions[i].data.base) &&
(offt < (table.partitions[i].data.base +
table.partitions[i].data.size)))
{
return table.partitions[i];
}
}
MSG_ERR("Partition corresponding to offset %x not found", offset);
elog<InternalFailure>();
static pnor_partition p{};
return p;
}
const pnor_partition& Table::partition(const std::string& name) const
{
const decltype(auto) table = getNativeTable();
for (decltype(numParts) i{}; i < numParts; ++i)
{
if (name == table.partitions[i].data.name)
{
return table.partitions[i];
}
}
MSG_ERR("Partition %s not found", name.c_str());
log<level::ERR>("Table::partition partition not found ",
entry("PARTITION_NAME=%s", name.c_str()));
elog<InternalFailure>();
static pnor_partition p{};
return p;
}
} // namespace partition
PartitionTable endianFixup(const PartitionTable& in)
{
PartitionTable out;
out.resize(in.size());
auto src = reinterpret_cast<const pnor_partition_table*>(in.data());
auto dst = reinterpret_cast<pnor_partition_table*>(out.data());
dst->data.magic = htobe32(src->data.magic);
dst->data.version = htobe32(src->data.version);
dst->data.size = htobe32(src->data.size);
dst->data.entry_size = htobe32(src->data.entry_size);
dst->data.entry_count = htobe32(src->data.entry_count);
dst->data.block_size = htobe32(src->data.block_size);
dst->data.block_count = htobe32(src->data.block_count);
dst->checksum = details::checksum(dst->data);
for (decltype(src->data.entry_count) i{}; i < src->data.entry_count; ++i)
{
auto psrc = &src->partitions[i];
auto pdst = &dst->partitions[i];
strncpy(pdst->data.name, psrc->data.name, PARTITION_NAME_MAX);
// Just to be safe
pdst->data.name[PARTITION_NAME_MAX] = '\0';
pdst->data.base = htobe32(psrc->data.base);
pdst->data.size = htobe32(psrc->data.size);
pdst->data.pid = htobe32(psrc->data.pid);
pdst->data.id = htobe32(psrc->data.id);
pdst->data.type = htobe32(psrc->data.type);
pdst->data.flags = htobe32(psrc->data.flags);
pdst->data.actual = htobe32(psrc->data.actual);
for (size_t j = 0; j < PARTITION_USER_WORDS; ++j)
{
pdst->data.user.data[j] = htobe32(psrc->data.user.data[j]);
}
pdst->checksum = details::checksum(pdst->data);
}
return out;
}
} // namespace virtual_pnor
} // namespace openpower