DirectoryBlob: Use DiscContent for ReadNonPartition too

ReadPartition and ReadNonPartition are now unified into
ReadInternal, with a parameter controlling whether
the read is in a partition or not.
This commit is contained in:
JosJuice 2017-06-08 18:43:52 +02:00
parent a0fa350ca6
commit 3f9eec9ae4
2 changed files with 20 additions and 63 deletions

View File

@ -172,22 +172,32 @@ DirectoryBlobReader::DirectoryBlobReader(File::IOFile dol_file, const std::strin
m_virtual_disc.emplace(APPLOADER_ADDRESS, m_apploader.size(), m_apploader.data()); m_virtual_disc.emplace(APPLOADER_ADDRESS, m_apploader.size(), m_apploader.data());
m_virtual_disc.emplace(m_dol_address, m_dol.size(), m_dol.data()); m_virtual_disc.emplace(m_dol_address, m_dol.size(), m_dol.data());
m_virtual_disc.emplace(m_fst_address, m_fst_data.size(), m_fst_data.data()); m_virtual_disc.emplace(m_fst_address, m_fst_data.size(), m_fst_data.data());
if (m_is_wii)
{
m_nonpartition_contents.emplace(DISKHEADER_ADDRESS, DISKHEADERINFO_ADDRESS,
m_disk_header.data());
m_nonpartition_contents.emplace(PARTITION_TABLE_ADDRESS, PARTITION_TABLE.size() * sizeof(u32),
reinterpret_cast<const u8*>(PARTITION_TABLE.data()));
// TODO: TMDs, tickets, more headers, the raw partition contents...
}
} }
bool DirectoryBlobReader::ReadPartition(u64 offset, u64 length, u8* buffer) bool DirectoryBlobReader::ReadInternal(u64 offset, u64 length, u8* buffer,
const std::set<DiscContent>& contents)
{ {
if (m_virtual_disc.empty()) if (contents.empty())
return true; return true;
// Determine which DiscContent the offset refers to // Determine which DiscContent the offset refers to
std::set<DiscContent>::const_iterator it = m_virtual_disc.lower_bound(DiscContent(offset)); std::set<DiscContent>::const_iterator it = contents.lower_bound(DiscContent(offset));
if (it->GetOffset() > offset && it != m_virtual_disc.begin()) if (it->GetOffset() > offset && it != contents.begin())
--it; --it;
// zero fill to start of file data // zero fill to start of file data
PadToAddress(it->GetOffset(), &offset, &length, &buffer); PadToAddress(it->GetOffset(), &offset, &length, &buffer);
while (it != m_virtual_disc.end() && length > 0) while (it != contents.end() && length > 0)
{ {
_dbg_assert_(DVDINTERFACE, it->GetOffset() <= offset); _dbg_assert_(DVDINTERFACE, it->GetOffset() <= offset);
if (!it->Read(&offset, &length, &buffer)) if (!it->Read(&offset, &length, &buffer))
@ -195,7 +205,7 @@ bool DirectoryBlobReader::ReadPartition(u64 offset, u64 length, u8* buffer)
++it; ++it;
if (it != m_virtual_disc.end()) if (it != contents.end())
{ {
_dbg_assert_(DVDINTERFACE, it->GetOffset() >= offset); _dbg_assert_(DVDINTERFACE, it->GetOffset() >= offset);
PadToAddress(it->GetOffset(), &offset, &length, &buffer); PadToAddress(it->GetOffset(), &offset, &length, &buffer);
@ -205,35 +215,9 @@ bool DirectoryBlobReader::ReadPartition(u64 offset, u64 length, u8* buffer)
return true; return true;
} }
bool DirectoryBlobReader::ReadNonPartition(u64 offset, u64 length, u8* buffer)
{
// header
if (offset < DISKHEADERINFO_ADDRESS)
{
WriteToBuffer(DISKHEADER_ADDRESS, DISKHEADERINFO_ADDRESS, m_disk_header.data(), &offset,
&length, &buffer);
}
if (offset >= 0x40000)
{
WriteToBuffer(PARTITION_TABLE_ADDRESS, PARTITION_TABLE.size() * sizeof(u32),
reinterpret_cast<const u8*>(PARTITION_TABLE.data()), &offset, &length, &buffer);
}
// TODO: TMDs, tickets, more headers, the partition contents...
if (length > 0)
{
ERROR_LOG(DISCIO, "Unsupported raw read in DirectoryBlob at 0x%" PRIx64, offset);
return false;
}
return true;
}
bool DirectoryBlobReader::Read(u64 offset, u64 length, u8* buffer) bool DirectoryBlobReader::Read(u64 offset, u64 length, u8* buffer)
{ {
return m_is_wii ? ReadNonPartition(offset, length, buffer) : return ReadInternal(offset, length, buffer, m_is_wii ? m_nonpartition_contents : m_virtual_disc);
ReadPartition(offset, length, buffer);
} }
bool DirectoryBlobReader::SupportsReadWiiDecrypted() const bool DirectoryBlobReader::SupportsReadWiiDecrypted() const
@ -246,7 +230,7 @@ bool DirectoryBlobReader::ReadWiiDecrypted(u64 offset, u64 size, u8* buffer, u64
if (!m_is_wii || partition_offset != GAME_PARTITION_ADDRESS) if (!m_is_wii || partition_offset != GAME_PARTITION_ADDRESS)
return false; return false;
return ReadPartition(offset, size, buffer); return ReadInternal(offset, size, buffer, m_virtual_disc);
} }
void DirectoryBlobReader::SetGameID(const std::string& id) void DirectoryBlobReader::SetGameID(const std::string& id)
@ -386,29 +370,6 @@ void DirectoryBlobReader::BuildFST()
Write32((u32)(m_fst_data.size() >> m_address_shift), 0x042c, &m_disk_header); Write32((u32)(m_fst_data.size() >> m_address_shift), 0x042c, &m_disk_header);
} }
void DirectoryBlobReader::WriteToBuffer(u64 source_start_address, u64 source_length,
const u8* source, u64* address, u64* length,
u8** buffer) const
{
if (*length == 0)
return;
_dbg_assert_(DVDINTERFACE, *address >= source_start_address);
u64 source_offset = *address - source_start_address;
if (source_offset < source_length)
{
size_t bytes_to_read = std::min(source_length - source_offset, *length);
memcpy(*buffer, source + source_offset, bytes_to_read);
*length -= bytes_to_read;
*buffer += bytes_to_read;
*address += bytes_to_read;
}
}
void DirectoryBlobReader::PadToAddress(u64 start_address, u64* address, u64* length, void DirectoryBlobReader::PadToAddress(u64 start_address, u64* address, u64* length,
u8** buffer) const u8** buffer) const
{ {

View File

@ -75,8 +75,7 @@ public:
private: private:
DirectoryBlobReader(File::IOFile dol_file, const std::string& root_directory); DirectoryBlobReader(File::IOFile dol_file, const std::string& root_directory);
bool ReadPartition(u64 offset, u64 length, u8* buffer); bool ReadInternal(u64 offset, u64 length, u8* buffer, const std::set<DiscContent>& contents);
bool ReadNonPartition(u64 offset, u64 length, u8* buffer);
void SetDiskTypeWii(); void SetDiskTypeWii();
void SetDiskTypeGC(); void SetDiskTypeGC();
@ -89,10 +88,6 @@ private:
void BuildFST(); void BuildFST();
// writing to read buffer
void WriteToBuffer(u64 source_start_address, u64 source_length, const u8* source, u64* address,
u64* length, u8** buffer) const;
void PadToAddress(u64 start_address, u64* address, u64* length, u8** buffer) const; void PadToAddress(u64 start_address, u64* address, u64* length, u8** buffer) const;
void Write32(u32 data, u32 offset, std::vector<u8>* const buffer); void Write32(u32 data, u32 offset, std::vector<u8>* const buffer);
@ -107,6 +102,7 @@ private:
std::string m_root_directory; std::string m_root_directory;
std::set<DiscContent> m_virtual_disc; std::set<DiscContent> m_virtual_disc;
std::set<DiscContent> m_nonpartition_contents;
bool m_is_wii = false; bool m_is_wii = false;