Deepak Kodihalli | 393821d | 2017-04-28 04:44:38 -0500 | [diff] [blame] | 1 | #include "pnor_partition_table.hpp" |
| 2 | #include "common.h" |
| 3 | #include "config.h" |
| 4 | #include <syslog.h> |
| 5 | #include <endian.h> |
| 6 | #include <regex> |
| 7 | #include <fstream> |
| 8 | #include <algorithm> |
| 9 | |
| 10 | namespace openpower |
| 11 | { |
| 12 | namespace virtual_pnor |
| 13 | { |
| 14 | |
| 15 | namespace partition |
| 16 | { |
Deepak Kodihalli | 393821d | 2017-04-28 04:44:38 -0500 | [diff] [blame] | 17 | |
Deepak Kodihalli | d1d7930 | 2017-07-11 23:01:39 -0500 | [diff] [blame^] | 18 | Table::Table(size_t blockSize, size_t pnorSize): |
| 19 | Table(fs::path(PARTITION_FILES_RO_LOC), blockSize, pnorSize) |
Deepak Kodihalli | 393821d | 2017-04-28 04:44:38 -0500 | [diff] [blame] | 20 | { |
| 21 | } |
| 22 | |
Deepak Kodihalli | d1d7930 | 2017-07-11 23:01:39 -0500 | [diff] [blame^] | 23 | Table::Table(fs::path&& directory, |
| 24 | size_t blockSize, size_t pnorSize): |
Deepak Kodihalli | 393821d | 2017-04-28 04:44:38 -0500 | [diff] [blame] | 25 | szBlocks(0), |
| 26 | imgBlocks(0), |
| 27 | directory(std::move(directory)), |
Deepak Kodihalli | d1d7930 | 2017-07-11 23:01:39 -0500 | [diff] [blame^] | 28 | numParts(0), |
| 29 | blockSize(blockSize), |
| 30 | pnorSize(pnorSize) |
Deepak Kodihalli | 393821d | 2017-04-28 04:44:38 -0500 | [diff] [blame] | 31 | { |
| 32 | preparePartitions(); |
| 33 | prepareHeader(); |
| 34 | hostTbl = endianFixup(tbl); |
| 35 | } |
| 36 | |
| 37 | void Table::prepareHeader() |
| 38 | { |
| 39 | decltype(auto) table = getNativeTable(); |
| 40 | table.data.magic = PARTITION_HEADER_MAGIC; |
| 41 | table.data.version = PARTITION_VERSION_1; |
| 42 | table.data.size = szBlocks; |
| 43 | table.data.entry_size = sizeof(pnor_partition); |
| 44 | table.data.entry_count = numParts; |
Deepak Kodihalli | d1d7930 | 2017-07-11 23:01:39 -0500 | [diff] [blame^] | 45 | table.data.block_size = blockSize; |
| 46 | table.data.block_count = pnorSize / blockSize; |
Deepak Kodihalli | 393821d | 2017-04-28 04:44:38 -0500 | [diff] [blame] | 47 | table.checksum = details::checksum(table.data); |
| 48 | } |
| 49 | |
| 50 | inline void Table::allocateMemory(const fs::path& tocFile) |
| 51 | { |
| 52 | size_t num = 0; |
| 53 | std::string line; |
| 54 | std::ifstream file(tocFile.c_str()); |
| 55 | |
| 56 | // Find number of lines in partition file - this will help |
| 57 | // determine the number of partitions and hence also how much |
| 58 | // memory to allocate for the partitions array. |
| 59 | // The actual number of partitions may turn out to be lesser than this, |
| 60 | // in case of errors. |
| 61 | while (std::getline(file, line)) |
| 62 | { |
| 63 | // Check if line starts with "partition" |
| 64 | if (std::string::npos != line.find("partition", 0)) |
| 65 | { |
| 66 | ++num; |
| 67 | } |
| 68 | } |
| 69 | |
| 70 | size_t totalSizeBytes = sizeof(pnor_partition_table) + |
| 71 | (num * sizeof(pnor_partition)); |
Deepak Kodihalli | d1d7930 | 2017-07-11 23:01:39 -0500 | [diff] [blame^] | 72 | size_t totalSizeAligned = align_up(totalSizeBytes, blockSize); |
| 73 | szBlocks = totalSizeAligned / blockSize; |
Deepak Kodihalli | 393821d | 2017-04-28 04:44:38 -0500 | [diff] [blame] | 74 | imgBlocks = szBlocks; |
| 75 | tbl.resize(totalSizeAligned); |
| 76 | } |
| 77 | |
| 78 | inline void Table::writeSizes(pnor_partition& part, size_t start, size_t end) |
| 79 | { |
| 80 | size_t size = end - start; |
| 81 | part.data.base = imgBlocks; |
Deepak Kodihalli | d1d7930 | 2017-07-11 23:01:39 -0500 | [diff] [blame^] | 82 | size_t sizeInBlocks = align_up(size, blockSize) / blockSize; |
Deepak Kodihalli | 393821d | 2017-04-28 04:44:38 -0500 | [diff] [blame] | 83 | imgBlocks += sizeInBlocks; |
| 84 | part.data.size = sizeInBlocks; |
| 85 | part.data.actual = size; |
| 86 | } |
| 87 | |
| 88 | inline void Table::writeUserdata(pnor_partition& part, const std::string& data) |
| 89 | { |
| 90 | if (std::string::npos != data.find("ECC")) |
| 91 | { |
| 92 | part.data.user.data[0] = PARTITION_ECC_PROTECTED; |
| 93 | } |
| 94 | auto perms = 0; |
| 95 | if (std::string::npos != data.find("READONLY")) |
| 96 | { |
| 97 | perms |= PARTITION_READONLY; |
| 98 | } |
| 99 | if (std::string::npos != data.find("PRESERVED")) |
| 100 | { |
| 101 | perms |= PARTITION_PRESERVED; |
| 102 | } |
| 103 | part.data.user.data[1] = perms; |
| 104 | } |
| 105 | |
| 106 | inline void Table::writeDefaults(pnor_partition& part) |
| 107 | { |
| 108 | part.data.pid = PARENT_PATITION_ID; |
| 109 | part.data.type = PARTITION_TYPE_DATA; |
| 110 | part.data.flags = 0; // flags unused |
| 111 | } |
| 112 | |
| 113 | inline void Table::writeNameAndId(pnor_partition& part, std::string&& name, |
| 114 | const std::string& id) |
| 115 | { |
| 116 | name.resize(PARTITION_NAME_MAX); |
| 117 | memcpy(part.data.name, |
| 118 | name.c_str(), |
| 119 | sizeof(part.data.name)); |
| 120 | part.data.id = std::stoul(id); |
| 121 | } |
| 122 | |
| 123 | void Table::preparePartitions() |
| 124 | { |
| 125 | fs::path tocFile = directory; |
| 126 | tocFile /= PARTITION_TOC_FILE; |
| 127 | allocateMemory(tocFile); |
| 128 | |
| 129 | std::ifstream file(tocFile.c_str()); |
| 130 | static constexpr auto ID_MATCH = 1; |
| 131 | static constexpr auto NAME_MATCH = 2; |
| 132 | static constexpr auto START_ADDR_MATCH = 3; |
| 133 | static constexpr auto END_ADDR_MATCH = 4; |
| 134 | // Parse PNOR toc (table of contents) file, which has lines like : |
| 135 | // partition01=HBB,00010000,000a0000,ECC,PRESERVED, to indicate partitions |
| 136 | std::regex regex |
| 137 | { |
| 138 | "^partition([0-9]+)=([A-Za-z0-9_]+)," |
| 139 | "([0-9a-fA-F]+),([0-9a-fA-F]+)", |
| 140 | std::regex::extended |
| 141 | }; |
| 142 | std::smatch match; |
| 143 | std::string line; |
| 144 | |
| 145 | decltype(auto) table = getNativeTable(); |
| 146 | |
| 147 | while (std::getline(file, line)) |
| 148 | { |
| 149 | if (std::regex_search(line, match, regex)) |
| 150 | { |
| 151 | fs::path partitionFile = directory; |
| 152 | partitionFile /= match[NAME_MATCH].str(); |
| 153 | if (!fs::exists(partitionFile)) |
| 154 | { |
| 155 | MSG_ERR("Partition file %s does not exist", |
| 156 | partitionFile.c_str()); |
| 157 | continue; |
| 158 | } |
| 159 | |
| 160 | writeNameAndId(table.partitions[numParts], |
| 161 | match[NAME_MATCH].str(), |
| 162 | match[ID_MATCH].str()); |
| 163 | writeDefaults(table.partitions[numParts]); |
| 164 | writeSizes(table.partitions[numParts], |
| 165 | std::stoul(match[START_ADDR_MATCH].str(), nullptr, 16), |
| 166 | std::stoul(match[END_ADDR_MATCH].str(), nullptr, 16)); |
| 167 | writeUserdata(table.partitions[numParts], match.suffix().str()); |
| 168 | table.partitions[numParts].checksum = |
| 169 | details::checksum(table.partitions[numParts].data); |
| 170 | |
| 171 | ++numParts; |
| 172 | } |
| 173 | } |
| 174 | } |
| 175 | |
| 176 | const pnor_partition& Table::partition(size_t offset) const |
| 177 | { |
| 178 | const decltype(auto) table = getNativeTable(); |
Deepak Kodihalli | d1d7930 | 2017-07-11 23:01:39 -0500 | [diff] [blame^] | 179 | size_t offt = offset / blockSize; |
Deepak Kodihalli | 393821d | 2017-04-28 04:44:38 -0500 | [diff] [blame] | 180 | |
| 181 | for (decltype(numParts) i{}; i < numParts; ++i) |
| 182 | { |
| 183 | if ((offt >= table.partitions[i].data.base) && |
| 184 | (offt < (table.partitions[i].data.base + |
| 185 | table.partitions[i].data.size))) |
| 186 | { |
| 187 | return table.partitions[i]; |
| 188 | } |
| 189 | } |
| 190 | |
| 191 | static pnor_partition p{}; |
| 192 | return p; |
| 193 | } |
| 194 | |
| 195 | } // namespace partition |
| 196 | |
| 197 | PartitionTable endianFixup(const PartitionTable& in) |
| 198 | { |
| 199 | PartitionTable out; |
| 200 | out.resize(in.size()); |
| 201 | auto src = reinterpret_cast<const pnor_partition_table*>(in.data()); |
| 202 | auto dst = reinterpret_cast<pnor_partition_table*>(out.data()); |
| 203 | |
| 204 | dst->data.magic = htobe32(src->data.magic); |
| 205 | dst->data.version = htobe32(src->data.version); |
| 206 | dst->data.size = htobe32(src->data.size); |
| 207 | dst->data.entry_size = htobe32(src->data.entry_size); |
| 208 | dst->data.entry_count = htobe32(src->data.entry_count); |
| 209 | dst->data.block_size = htobe32(src->data.block_size); |
| 210 | dst->data.block_count = htobe32(src->data.block_count); |
Deepak Kodihalli | b9cdcd2 | 2017-07-12 08:14:48 -0500 | [diff] [blame] | 211 | dst->checksum = details::checksum(dst->data); |
Deepak Kodihalli | 393821d | 2017-04-28 04:44:38 -0500 | [diff] [blame] | 212 | |
| 213 | for (decltype(src->data.entry_count) i{}; i < src->data.entry_count; ++i) |
| 214 | { |
| 215 | auto psrc = &src->partitions[i]; |
| 216 | auto pdst = &dst->partitions[i]; |
| 217 | strncpy(pdst->data.name, psrc->data.name, PARTITION_NAME_MAX); |
| 218 | // Just to be safe |
| 219 | pdst->data.name[PARTITION_NAME_MAX] = '\0'; |
| 220 | pdst->data.base = htobe32(psrc->data.base); |
| 221 | pdst->data.size = htobe32(psrc->data.size); |
| 222 | pdst->data.pid = htobe32(psrc->data.pid); |
| 223 | pdst->data.id = htobe32(psrc->data.id); |
| 224 | pdst->data.type = htobe32(psrc->data.type); |
| 225 | pdst->data.flags = htobe32(psrc->data.flags); |
| 226 | pdst->data.actual = htobe32(psrc->data.actual); |
| 227 | for (size_t j = 0; j < PARTITION_USER_WORDS; ++j) |
| 228 | { |
| 229 | pdst->data.user.data[j] = htobe32(psrc->data.user.data[j]); |
| 230 | } |
Deepak Kodihalli | b9cdcd2 | 2017-07-12 08:14:48 -0500 | [diff] [blame] | 231 | pdst->checksum = details::checksum(pdst->data); |
Deepak Kodihalli | 393821d | 2017-04-28 04:44:38 -0500 | [diff] [blame] | 232 | } |
| 233 | |
| 234 | return out; |
| 235 | } |
| 236 | |
| 237 | } // namespace virtual_pnor |
| 238 | } // namespace openpower |