partition_data_manager: Reserve and insert data within output vector in DecryptPackage2()

We can just reserve the memory then perform successive insertions
instead of needing to use memcpy. This also avoids the need to zero out
the output vector's memory before performing the insertions.

We can also std::move the output std::vector into the destination so
that we don't need to make a completely new copy of the vector, getting
rid of an unnecessary allocation.

Additionally, we can use iterators to determine the beginning and end
ranges of the std::vector instances that comprise the output vector, as
the end of one range just becomes the beginning for the next successive
range, and since std::vector's iterator constructor copies data within
the range [begin, end), this is more straightforward and gets rid of the
need to have an offset variable that keeps getting incremented to
determine where to do the next std::memcpy.
This commit is contained in:
Lioncash 2018-10-13 09:30:09 -04:00
parent 781fd7983c
commit 6467b01de2
1 changed files with 16 additions and 20 deletions

View File

@ -499,33 +499,29 @@ void PartitionDataManager::DecryptPackage2(const std::array<Key128, 0x20>& packa
continue; continue;
} }
std::vector<u8> text(kip.sections[0].size_compressed); const u64 initial_offset = sizeof(KIPHeader) + offset;
std::vector<u8> rodata(kip.sections[1].size_compressed); const auto text_begin = c.cbegin() + initial_offset;
std::vector<u8> data(kip.sections[2].size_compressed); const auto text_end = text_begin + kip.sections[0].size_compressed;
const std::vector<u8> text = DecompressBLZ({text_begin, text_end});
u64 offset_sec = sizeof(KIPHeader) + offset; const auto rodata_end = text_end + kip.sections[1].size_compressed;
std::memcpy(text.data(), c.data() + offset_sec, text.size()); const std::vector<u8> rodata = DecompressBLZ({text_end, rodata_end});
offset_sec += text.size();
std::memcpy(rodata.data(), c.data() + offset_sec, rodata.size());
offset_sec += rodata.size();
std::memcpy(data.data(), c.data() + offset_sec, data.size());
offset += sizeof(KIPHeader) + kip.sections[0].size_compressed + const auto data_end = rodata_end + kip.sections[2].size_compressed;
kip.sections[1].size_compressed + kip.sections[2].size_compressed; const std::vector<u8> data = DecompressBLZ({rodata_end, data_end});
text = DecompressBLZ(text); std::vector<u8> out;
rodata = DecompressBLZ(rodata); out.reserve(text.size() + rodata.size() + data.size());
data = DecompressBLZ(data); out.insert(out.end(), text.begin(), text.end());
out.insert(out.end(), rodata.begin(), rodata.end());
out.insert(out.end(), data.begin(), data.end());
std::vector<u8> out(text.size() + rodata.size() + data.size()); offset += sizeof(KIPHeader) + out.size();
std::memcpy(out.data(), text.data(), text.size());
std::memcpy(out.data() + text.size(), rodata.data(), rodata.size());
std::memcpy(out.data() + text.size() + rodata.size(), data.data(), data.size());
if (name == "FS") if (name == "FS")
package2_fs[static_cast<size_t>(type)] = out; package2_fs[static_cast<size_t>(type)] = std::move(out);
else if (name == "spl") else if (name == "spl")
package2_spl[static_cast<size_t>(type)] = out; package2_spl[static_cast<size_t>(type)] = std::move(out);
} }
} }