Merge pull request #4564 from JosJuice/volumedirectory-references

VolumeDirectory: Don't use references inappropriately
This commit is contained in:
Markus Wick 2016-12-26 11:48:38 +01:00 committed by GitHub
commit 4ebd369218
2 changed files with 164 additions and 166 deletions

View File

@ -24,30 +24,30 @@
namespace DiscIO namespace DiscIO
{ {
static u32 ComputeNameSize(const File::FSTEntry& parentEntry); static u32 ComputeNameSize(const File::FSTEntry& parent_entry);
const size_t CVolumeDirectory::MAX_NAME_LENGTH; const size_t CVolumeDirectory::MAX_NAME_LENGTH;
const size_t CVolumeDirectory::MAX_ID_LENGTH; const size_t CVolumeDirectory::MAX_ID_LENGTH;
CVolumeDirectory::CVolumeDirectory(const std::string& _rDirectory, bool _bIsWii, CVolumeDirectory::CVolumeDirectory(const std::string& directory, bool is_wii,
const std::string& _rApploader, const std::string& _rDOL) const std::string& apploader, const std::string& dol)
: m_dataStartAddress(-1), m_diskHeader(DISKHEADERINFO_ADDRESS), : m_data_start_address(-1), m_disk_header(DISKHEADERINFO_ADDRESS),
m_diskHeaderInfo(std::make_unique<SDiskHeaderInfo>()), m_fst_address(0), m_dol_address(0) m_disk_header_info(std::make_unique<SDiskHeaderInfo>()), m_fst_address(0), m_dol_address(0)
{ {
m_rootDirectory = ExtractDirectoryName(_rDirectory); m_root_directory = ExtractDirectoryName(directory);
// create the default disk header // create the default disk header
SetGameID("AGBJ01"); SetGameID("AGBJ01");
SetName("Default name"); SetName("Default name");
if (_bIsWii) if (is_wii)
SetDiskTypeWii(); SetDiskTypeWii();
else else
SetDiskTypeGC(); SetDiskTypeGC();
// Don't load the DOL if we don't have an apploader // Don't load the DOL if we don't have an apploader
if (SetApploader(_rApploader)) if (SetApploader(apploader))
SetDOL(_rDOL); SetDOL(dol);
BuildFST(); BuildFST();
} }
@ -56,14 +56,14 @@ CVolumeDirectory::~CVolumeDirectory()
{ {
} }
bool CVolumeDirectory::IsValidDirectory(const std::string& _rDirectory) bool CVolumeDirectory::IsValidDirectory(const std::string& directory)
{ {
return File::IsDirectory(ExtractDirectoryName(_rDirectory)); return File::IsDirectory(ExtractDirectoryName(directory));
} }
bool CVolumeDirectory::Read(u64 _Offset, u64 _Length, u8* _pBuffer, bool decrypt) const bool CVolumeDirectory::Read(u64 offset, u64 length, u8* buffer, bool decrypt) const
{ {
if (!decrypt && (_Offset + _Length >= 0x400) && m_is_wii) if (!decrypt && (offset + length >= 0x400) && m_is_wii)
{ {
// Fully supporting this would require re-encrypting every file that's read. // Fully supporting this would require re-encrypting every file that's read.
// Only supporting the areas that IOS allows software to read could be more feasible. // Only supporting the areas that IOS allows software to read could be more feasible.
@ -77,49 +77,49 @@ bool CVolumeDirectory::Read(u64 _Offset, u64 _Length, u8* _pBuffer, bool decrypt
PanicAlertT("Tried to decrypt data from a non-Wii volume"); PanicAlertT("Tried to decrypt data from a non-Wii volume");
// header // header
if (_Offset < DISKHEADERINFO_ADDRESS) if (offset < DISKHEADERINFO_ADDRESS)
{ {
WriteToBuffer(DISKHEADER_ADDRESS, DISKHEADERINFO_ADDRESS, m_diskHeader.data(), _Offset, _Length, WriteToBuffer(DISKHEADER_ADDRESS, DISKHEADERINFO_ADDRESS, m_disk_header.data(), &offset,
_pBuffer); &length, &buffer);
} }
// header info // header info
if (_Offset >= DISKHEADERINFO_ADDRESS && _Offset < APPLOADER_ADDRESS) if (offset >= DISKHEADERINFO_ADDRESS && offset < APPLOADER_ADDRESS)
{ {
WriteToBuffer(DISKHEADERINFO_ADDRESS, sizeof(m_diskHeaderInfo), (u8*)m_diskHeaderInfo.get(), WriteToBuffer(DISKHEADERINFO_ADDRESS, sizeof(m_disk_header_info), (u8*)m_disk_header_info.get(),
_Offset, _Length, _pBuffer); &offset, &length, &buffer);
} }
// apploader // apploader
if (_Offset >= APPLOADER_ADDRESS && _Offset < APPLOADER_ADDRESS + m_apploader.size()) if (offset >= APPLOADER_ADDRESS && offset < APPLOADER_ADDRESS + m_apploader.size())
{ {
WriteToBuffer(APPLOADER_ADDRESS, m_apploader.size(), m_apploader.data(), _Offset, _Length, WriteToBuffer(APPLOADER_ADDRESS, m_apploader.size(), m_apploader.data(), &offset, &length,
_pBuffer); &buffer);
} }
// dol // dol
if (_Offset >= m_dol_address && _Offset < m_dol_address + m_DOL.size()) if (offset >= m_dol_address && offset < m_dol_address + m_dol.size())
{ {
WriteToBuffer(m_dol_address, m_DOL.size(), m_DOL.data(), _Offset, _Length, _pBuffer); WriteToBuffer(m_dol_address, m_dol.size(), m_dol.data(), &offset, &length, &buffer);
} }
// fst // fst
if (_Offset >= m_fst_address && _Offset < m_dataStartAddress) if (offset >= m_fst_address && offset < m_data_start_address)
{ {
WriteToBuffer(m_fst_address, m_FSTData.size(), m_FSTData.data(), _Offset, _Length, _pBuffer); WriteToBuffer(m_fst_address, m_fst_data.size(), m_fst_data.data(), &offset, &length, &buffer);
} }
if (m_virtualDisk.empty()) if (m_virtual_disk.empty())
return true; return true;
// Determine which file the offset refers to // Determine which file the offset refers to
std::map<u64, std::string>::const_iterator fileIter = m_virtualDisk.lower_bound(_Offset); std::map<u64, std::string>::const_iterator fileIter = m_virtual_disk.lower_bound(offset);
if (fileIter->first > _Offset && fileIter != m_virtualDisk.begin()) if (fileIter->first > offset && fileIter != m_virtual_disk.begin())
--fileIter; --fileIter;
// zero fill to start of file data // zero fill to start of file data
PadToAddress(fileIter->first, _Offset, _Length, _pBuffer); PadToAddress(fileIter->first, &offset, &length, &buffer);
while (fileIter != m_virtualDisk.end() && _Length > 0) while (fileIter != m_virtual_disk.end() && length > 0)
{ {
_dbg_assert_(DVDINTERFACE, fileIter->first <= _Offset); _dbg_assert_(DVDINTERFACE, fileIter->first <= offset);
u64 fileOffset = _Offset - fileIter->first; u64 fileOffset = offset - fileIter->first;
const std::string fileName = fileIter->second; const std::string fileName = fileIter->second;
File::IOFile file(fileName, "rb"); File::IOFile file(fileName, "rb");
@ -132,24 +132,24 @@ bool CVolumeDirectory::Read(u64 _Offset, u64 _Length, u8* _pBuffer, bool decrypt
if (fileOffset < fileSize) if (fileOffset < fileSize)
{ {
u64 fileBytes = std::min(fileSize - fileOffset, _Length); u64 fileBytes = std::min(fileSize - fileOffset, length);
if (!file.Seek(fileOffset, SEEK_SET)) if (!file.Seek(fileOffset, SEEK_SET))
return false; return false;
if (!file.ReadBytes(_pBuffer, fileBytes)) if (!file.ReadBytes(buffer, fileBytes))
return false; return false;
_Length -= fileBytes; length -= fileBytes;
_pBuffer += fileBytes; buffer += fileBytes;
_Offset += fileBytes; offset += fileBytes;
} }
++fileIter; ++fileIter;
if (fileIter != m_virtualDisk.end()) if (fileIter != m_virtual_disk.end())
{ {
_dbg_assert_(DVDINTERFACE, fileIter->first >= _Offset); _dbg_assert_(DVDINTERFACE, fileIter->first >= offset);
PadToAddress(fileIter->first, _Offset, _Length, _pBuffer); PadToAddress(fileIter->first, &offset, &length, &buffer);
} }
} }
@ -158,17 +158,17 @@ bool CVolumeDirectory::Read(u64 _Offset, u64 _Length, u8* _pBuffer, bool decrypt
std::string CVolumeDirectory::GetGameID() const std::string CVolumeDirectory::GetGameID() const
{ {
return std::string(m_diskHeader.begin(), m_diskHeader.begin() + MAX_ID_LENGTH); return std::string(m_disk_header.begin(), m_disk_header.begin() + MAX_ID_LENGTH);
} }
void CVolumeDirectory::SetGameID(const std::string& id) void CVolumeDirectory::SetGameID(const std::string& id)
{ {
memcpy(m_diskHeader.data(), id.c_str(), std::min(id.length(), MAX_ID_LENGTH)); memcpy(m_disk_header.data(), id.c_str(), std::min(id.length(), MAX_ID_LENGTH));
} }
Country CVolumeDirectory::GetCountry() const Country CVolumeDirectory::GetCountry() const
{ {
return CountrySwitch(m_diskHeader[3]); return CountrySwitch(m_disk_header[3]);
} }
std::string CVolumeDirectory::GetMakerID() const std::string CVolumeDirectory::GetMakerID() const
@ -205,8 +205,8 @@ std::vector<u32> CVolumeDirectory::GetBanner(int* width, int* height) const
void CVolumeDirectory::SetName(const std::string& name) void CVolumeDirectory::SetName(const std::string& name)
{ {
size_t length = std::min(name.length(), MAX_NAME_LENGTH); size_t length = std::min(name.length(), MAX_NAME_LENGTH);
memcpy(&m_diskHeader[0x20], name.c_str(), length); memcpy(&m_disk_header[0x20], name.c_str(), length);
m_diskHeader[length + 0x20] = 0; m_disk_header[length + 0x20] = 0;
} }
u64 CVolumeDirectory::GetFSTSize() const u64 CVolumeDirectory::GetFSTSize() const
@ -246,67 +246,67 @@ u64 CVolumeDirectory::GetRawSize() const
return 0; return 0;
} }
std::string CVolumeDirectory::ExtractDirectoryName(const std::string& _rDirectory) std::string CVolumeDirectory::ExtractDirectoryName(const std::string& directory)
{ {
std::string directoryName = _rDirectory; std::string result = directory;
size_t lastSep = directoryName.find_last_of(DIR_SEP_CHR); size_t last_separator = result.find_last_of(DIR_SEP_CHR);
if (lastSep != directoryName.size() - 1) if (last_separator != result.size() - 1)
{ {
// TODO: This assumes that file names will always have a dot in them // TODO: This assumes that file names will always have a dot in them
// and directory names never will; both assumptions are often // and directory names never will; both assumptions are often
// right but in general wrong. // right but in general wrong.
size_t extensionStart = directoryName.find_last_of('.'); size_t extension_start = result.find_last_of('.');
if (extensionStart != std::string::npos && extensionStart > lastSep) if (extension_start != std::string::npos && extension_start > last_separator)
{ {
directoryName.resize(lastSep); result.resize(last_separator);
} }
} }
else else
{ {
directoryName.resize(lastSep); result.resize(last_separator);
} }
return directoryName; return result;
} }
void CVolumeDirectory::SetDiskTypeWii() void CVolumeDirectory::SetDiskTypeWii()
{ {
Write32(0x5d1c9ea3, 0x18, &m_diskHeader); Write32(0x5d1c9ea3, 0x18, &m_disk_header);
memset(&m_diskHeader[0x1c], 0, 4); memset(&m_disk_header[0x1c], 0, 4);
m_is_wii = true; m_is_wii = true;
m_addressShift = 2; m_address_shift = 2;
} }
void CVolumeDirectory::SetDiskTypeGC() void CVolumeDirectory::SetDiskTypeGC()
{ {
memset(&m_diskHeader[0x18], 0, 4); memset(&m_disk_header[0x18], 0, 4);
Write32(0xc2339f3d, 0x1c, &m_diskHeader); Write32(0xc2339f3d, 0x1c, &m_disk_header);
m_is_wii = false; m_is_wii = false;
m_addressShift = 0; m_address_shift = 0;
} }
bool CVolumeDirectory::SetApploader(const std::string& _rApploader) bool CVolumeDirectory::SetApploader(const std::string& apploader)
{ {
if (!_rApploader.empty()) if (!apploader.empty())
{ {
std::string data; std::string data;
if (!File::ReadFileToString(_rApploader, data)) if (!File::ReadFileToString(apploader, data))
{ {
PanicAlertT("Apploader unable to load from file"); PanicAlertT("Apploader unable to load from file");
return false; return false;
} }
size_t apploaderSize = 0x20 + Common::swap32(*(u32*)&data.data()[0x14]) + size_t apploader_size = 0x20 + Common::swap32(*(u32*)&data.data()[0x14]) +
Common::swap32(*(u32*)&data.data()[0x18]); Common::swap32(*(u32*)&data.data()[0x18]);
if (apploaderSize != data.size()) if (apploader_size != data.size())
{ {
PanicAlertT("Apploader is the wrong size...is it really an apploader?"); PanicAlertT("Apploader is the wrong size...is it really an apploader?");
return false; return false;
} }
m_apploader.resize(apploaderSize); m_apploader.resize(apploader_size);
std::copy(data.begin(), data.end(), m_apploader.begin()); std::copy(data.begin(), data.end(), m_apploader.begin());
// 32byte aligned (plus 0x20 padding) // 32byte aligned (plus 0x20 padding)
@ -322,90 +322,89 @@ bool CVolumeDirectory::SetApploader(const std::string& _rApploader)
} }
} }
void CVolumeDirectory::SetDOL(const std::string& rDOL) void CVolumeDirectory::SetDOL(const std::string& dol)
{ {
if (!rDOL.empty()) if (!dol.empty())
{ {
std::string data; std::string data;
File::ReadFileToString(rDOL, data); File::ReadFileToString(dol, data);
m_DOL.resize(data.size()); m_dol.resize(data.size());
std::copy(data.begin(), data.end(), m_DOL.begin()); std::copy(data.begin(), data.end(), m_dol.begin());
Write32((u32)(m_dol_address >> m_addressShift), 0x0420, &m_diskHeader); Write32((u32)(m_dol_address >> m_address_shift), 0x0420, &m_disk_header);
// 32byte aligned (plus 0x20 padding) // 32byte aligned (plus 0x20 padding)
m_fst_address = Common::AlignUp(m_dol_address + m_DOL.size() + 0x20, 0x20ull); m_fst_address = Common::AlignUp(m_dol_address + m_dol.size() + 0x20, 0x20ull);
} }
} }
void CVolumeDirectory::BuildFST() void CVolumeDirectory::BuildFST()
{ {
m_FSTData.clear(); m_fst_data.clear();
File::FSTEntry rootEntry = File::ScanDirectoryTree(m_rootDirectory, true); File::FSTEntry rootEntry = File::ScanDirectoryTree(m_root_directory, true);
u32 name_table_size = ComputeNameSize(rootEntry); u32 name_table_size = ComputeNameSize(rootEntry);
m_fstNameOffset = rootEntry.size * ENTRY_SIZE; // offset of name table in FST m_fst_name_offset = rootEntry.size * ENTRY_SIZE; // offset of name table in FST
m_FSTData.resize(m_fstNameOffset + name_table_size); m_fst_data.resize(m_fst_name_offset + name_table_size);
// if FST hasn't been assigned (ie no apploader/dol setup), set to default // if FST hasn't been assigned (ie no apploader/dol setup), set to default
if (m_fst_address == 0) if (m_fst_address == 0)
m_fst_address = APPLOADER_ADDRESS + 0x2000; m_fst_address = APPLOADER_ADDRESS + 0x2000;
// 4 byte aligned start of data on disk // 4 byte aligned start of data on disk
m_dataStartAddress = Common::AlignUp(m_fst_address + m_FSTData.size(), 0x8000ull); m_data_start_address = Common::AlignUp(m_fst_address + m_fst_data.size(), 0x8000ull);
u64 curDataAddress = m_dataStartAddress; u64 current_data_address = m_data_start_address;
u32 fstOffset = 0; // Offset within FST data u32 fst_offset = 0; // Offset within FST data
u32 nameOffset = 0; // Offset within name table u32 name_offset = 0; // Offset within name table
u32 rootOffset = 0; // Offset of root of FST u32 root_offset = 0; // Offset of root of FST
// write root entry // write root entry
WriteEntryData(fstOffset, DIRECTORY_ENTRY, 0, 0, rootEntry.size); WriteEntryData(&fst_offset, DIRECTORY_ENTRY, 0, 0, rootEntry.size);
WriteDirectory(rootEntry, fstOffset, nameOffset, curDataAddress, rootOffset); WriteDirectory(rootEntry, &fst_offset, &name_offset, &current_data_address, root_offset);
// overflow check // overflow check
_dbg_assert_(DVDINTERFACE, nameOffset == name_table_size); _dbg_assert_(DVDINTERFACE, name_offset == name_table_size);
// write FST size and location // write FST size and location
Write32((u32)(m_fst_address >> m_addressShift), 0x0424, &m_diskHeader); Write32((u32)(m_fst_address >> m_address_shift), 0x0424, &m_disk_header);
Write32((u32)(m_FSTData.size() >> m_addressShift), 0x0428, &m_diskHeader); Write32((u32)(m_fst_data.size() >> m_address_shift), 0x0428, &m_disk_header);
Write32((u32)(m_FSTData.size() >> m_addressShift), 0x042c, &m_diskHeader); Write32((u32)(m_fst_data.size() >> m_address_shift), 0x042c, &m_disk_header);
} }
void CVolumeDirectory::WriteToBuffer(u64 _SrcStartAddress, u64 _SrcLength, const u8* _Src, void CVolumeDirectory::WriteToBuffer(u64 source_start_address, u64 source_length, const u8* source,
u64& _Address, u64& _Length, u8*& _pBuffer) const u64* address, u64* length, u8** buffer) const
{ {
if (_Length == 0) if (*length == 0)
return; return;
_dbg_assert_(DVDINTERFACE, _Address >= _SrcStartAddress); _dbg_assert_(DVDINTERFACE, *address >= source_start_address);
u64 srcOffset = _Address - _SrcStartAddress; u64 source_offset = *address - source_start_address;
if (srcOffset < _SrcLength) if (source_offset < source_length)
{ {
u64 srcBytes = std::min(_SrcLength - srcOffset, _Length); size_t bytes_to_read = std::min(source_length - source_offset, *length);
memcpy(_pBuffer, _Src + srcOffset, (size_t)srcBytes); memcpy(*buffer, source + source_offset, bytes_to_read);
_Length -= srcBytes; *length -= bytes_to_read;
_pBuffer += srcBytes; *buffer += bytes_to_read;
_Address += srcBytes; *address += bytes_to_read;
} }
} }
void CVolumeDirectory::PadToAddress(u64 _StartAddress, u64& _Address, u64& _Length, void CVolumeDirectory::PadToAddress(u64 start_address, u64* address, u64* length, u8** buffer) const
u8*& _pBuffer) const
{ {
if (_StartAddress > _Address && _Length > 0) if (start_address > *address && *length > 0)
{ {
u64 padBytes = std::min(_StartAddress - _Address, _Length); u64 padBytes = std::min(start_address - *address, *length);
memset(_pBuffer, 0, (size_t)padBytes); memset(*buffer, 0, (size_t)padBytes);
_Length -= padBytes; *length -= padBytes;
_pBuffer += padBytes; *buffer += padBytes;
_Address += padBytes; *address += padBytes;
} }
} }
@ -417,31 +416,31 @@ void CVolumeDirectory::Write32(u32 data, u32 offset, std::vector<u8>* const buff
(*buffer)[offset] = (data)&0xff; (*buffer)[offset] = (data)&0xff;
} }
void CVolumeDirectory::WriteEntryData(u32& entryOffset, u8 type, u32 nameOffset, u64 dataOffset, void CVolumeDirectory::WriteEntryData(u32* entry_offset, u8 type, u32 name_offset, u64 data_offset,
u64 length) u64 length)
{ {
m_FSTData[entryOffset++] = type; m_fst_data[(*entry_offset)++] = type;
m_FSTData[entryOffset++] = (nameOffset >> 16) & 0xff; m_fst_data[(*entry_offset)++] = (name_offset >> 16) & 0xff;
m_FSTData[entryOffset++] = (nameOffset >> 8) & 0xff; m_fst_data[(*entry_offset)++] = (name_offset >> 8) & 0xff;
m_FSTData[entryOffset++] = (nameOffset)&0xff; m_fst_data[(*entry_offset)++] = (name_offset)&0xff;
Write32((u32)(dataOffset >> m_addressShift), entryOffset, &m_FSTData); Write32((u32)(data_offset >> m_address_shift), *entry_offset, &m_fst_data);
entryOffset += 4; *entry_offset += 4;
Write32((u32)length, entryOffset, &m_FSTData); Write32((u32)length, *entry_offset, &m_fst_data);
entryOffset += 4; *entry_offset += 4;
} }
void CVolumeDirectory::WriteEntryName(u32& nameOffset, const std::string& name) void CVolumeDirectory::WriteEntryName(u32* name_offset, const std::string& name)
{ {
strncpy((char*)&m_FSTData[nameOffset + m_fstNameOffset], name.c_str(), name.length() + 1); strncpy((char*)&m_fst_data[*name_offset + m_fst_name_offset], name.c_str(), name.length() + 1);
nameOffset += (u32)(name.length() + 1); *name_offset += (u32)(name.length() + 1);
} }
void CVolumeDirectory::WriteDirectory(const File::FSTEntry& parent_entry, u32& fstOffset, void CVolumeDirectory::WriteDirectory(const File::FSTEntry& parent_entry, u32* fst_offset,
u32& nameOffset, u64& dataOffset, u32 parentEntryNum) u32* name_offset, u64* data_offset, u32 parent_entry_index)
{ {
std::vector<File::FSTEntry> sorted_entries = parent_entry.children; std::vector<File::FSTEntry> sorted_entries = parent_entry.children;
@ -454,41 +453,40 @@ void CVolumeDirectory::WriteDirectory(const File::FSTEntry& parent_entry, u32& f
{ {
if (entry.isDirectory) if (entry.isDirectory)
{ {
u32 myOffset = fstOffset; u32 entry_index = *fst_offset / ENTRY_SIZE;
u32 myEntryNum = myOffset / ENTRY_SIZE; WriteEntryData(fst_offset, DIRECTORY_ENTRY, *name_offset, parent_entry_index,
WriteEntryData(fstOffset, DIRECTORY_ENTRY, nameOffset, parentEntryNum, entry_index + entry.size + 1);
myEntryNum + entry.size + 1); WriteEntryName(name_offset, entry.virtualName);
WriteEntryName(nameOffset, entry.virtualName);
WriteDirectory(entry, fstOffset, nameOffset, dataOffset, myEntryNum); WriteDirectory(entry, fst_offset, name_offset, data_offset, entry_index);
} }
else else
{ {
// put entry in FST // put entry in FST
WriteEntryData(fstOffset, FILE_ENTRY, nameOffset, dataOffset, entry.size); WriteEntryData(fst_offset, FILE_ENTRY, *name_offset, *data_offset, entry.size);
WriteEntryName(nameOffset, entry.virtualName); WriteEntryName(name_offset, entry.virtualName);
// write entry to virtual disk // write entry to virtual disk
_dbg_assert_(DVDINTERFACE, m_virtualDisk.find(dataOffset) == m_virtualDisk.end()); _dbg_assert_(DVDINTERFACE, m_virtual_disk.find(*data_offset) == m_virtual_disk.end());
m_virtualDisk.emplace(dataOffset, entry.physicalName); m_virtual_disk.emplace(*data_offset, entry.physicalName);
// 4 byte aligned // 4 byte aligned
dataOffset = Common::AlignUp(dataOffset + std::max<u64>(entry.size, 1ull), 0x8000ull); *data_offset = Common::AlignUp(*data_offset + std::max<u64>(entry.size, 1ull), 0x8000ull);
} }
} }
} }
static u32 ComputeNameSize(const File::FSTEntry& parentEntry) static u32 ComputeNameSize(const File::FSTEntry& parent_entry)
{ {
u32 nameSize = 0; u32 name_size = 0;
for (const File::FSTEntry& entry : parentEntry.children) for (const File::FSTEntry& entry : parent_entry.children)
{ {
if (entry.isDirectory) if (entry.isDirectory)
nameSize += ComputeNameSize(entry); name_size += ComputeNameSize(entry);
nameSize += (u32)entry.virtualName.length() + 1; name_size += (u32)entry.virtualName.length() + 1;
} }
return nameSize; return name_size;
} }
} // namespace } // namespace

View File

@ -31,17 +31,17 @@ enum class Platform;
class CVolumeDirectory : public IVolume class CVolumeDirectory : public IVolume
{ {
public: public:
CVolumeDirectory(const std::string& _rDirectory, bool _bIsWii, CVolumeDirectory(const std::string& directory, bool is_wii, const std::string& apploader = "",
const std::string& _rApploader = "", const std::string& _rDOL = ""); const std::string& dol = "");
~CVolumeDirectory(); ~CVolumeDirectory();
static bool IsValidDirectory(const std::string& _rDirectory); static bool IsValidDirectory(const std::string& directory);
bool Read(u64 _Offset, u64 _Length, u8* _pBuffer, bool decrypt) const override; bool Read(u64 offset, u64 length, u8* buffer, bool decrypt) const override;
std::string GetGameID() const override; std::string GetGameID() const override;
void SetGameID(const std::string& _ID); void SetGameID(const std::string& id);
std::string GetMakerID() const override; std::string GetMakerID() const override;
@ -65,50 +65,50 @@ public:
void BuildFST(); void BuildFST();
private: private:
static std::string ExtractDirectoryName(const std::string& _rDirectory); static std::string ExtractDirectoryName(const std::string& directory);
void SetDiskTypeWii(); void SetDiskTypeWii();
void SetDiskTypeGC(); void SetDiskTypeGC();
bool SetApploader(const std::string& _rApploader); bool SetApploader(const std::string& apploader);
void SetDOL(const std::string& _rDOL); void SetDOL(const std::string& dol);
// writing to read buffer // writing to read buffer
void WriteToBuffer(u64 _SrcStartAddress, u64 _SrcLength, const u8* _Src, u64& _Address, void WriteToBuffer(u64 source_start_address, u64 source_length, const u8* source, u64* address,
u64& _Length, u8*& _pBuffer) const; u64* length, u8** buffer) const;
void PadToAddress(u64 _StartAddress, u64& _Address, u64& _Length, u8*& _pBuffer) 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);
// FST creation // FST creation
void WriteEntryData(u32& entryOffset, u8 type, u32 nameOffset, u64 dataOffset, u64 length); void WriteEntryData(u32* entry_offset, u8 type, u32 name_offset, u64 data_offset, u64 length);
void WriteEntryName(u32& nameOffset, const std::string& name); void WriteEntryName(u32* name_offset, const std::string& name);
void WriteDirectory(const File::FSTEntry& parent_entry, u32& fstOffset, u32& nameOffset, void WriteDirectory(const File::FSTEntry& parent_entry, u32* fst_offset, u32* name_offset,
u64& dataOffset, u32 parentEntryNum); u64* data_offset, u32 parent_entry_index);
std::string m_rootDirectory; std::string m_root_directory;
std::map<u64, std::string> m_virtualDisk; std::map<u64, std::string> m_virtual_disk;
bool m_is_wii; bool m_is_wii;
// GameCube has no shift, Wii has 2 bit shift // GameCube has no shift, Wii has 2 bit shift
u32 m_addressShift; u32 m_address_shift;
// first address on disk containing file data // first address on disk containing file data
u64 m_dataStartAddress; u64 m_data_start_address;
u64 m_fstNameOffset; u64 m_fst_name_offset;
std::vector<u8> m_FSTData; std::vector<u8> m_fst_data;
std::vector<u8> m_diskHeader; std::vector<u8> m_disk_header;
#pragma pack(push, 1) #pragma pack(push, 1)
struct SDiskHeaderInfo struct SDiskHeaderInfo
{ {
u32 debug_mntr_size; u32 debug_monitor_size;
u32 simulated_mem_size; u32 simulated_mem_size;
u32 arg_offset; u32 arg_offset;
u32 debug_flag; u32 debug_flag;
@ -121,7 +121,7 @@ private:
// All the data is byteswapped // All the data is byteswapped
SDiskHeaderInfo() SDiskHeaderInfo()
{ {
debug_mntr_size = 0; debug_monitor_size = 0;
simulated_mem_size = 0; simulated_mem_size = 0;
arg_offset = 0; arg_offset = 0;
debug_flag = 0; debug_flag = 0;
@ -133,10 +133,10 @@ private:
} }
}; };
#pragma pack(pop) #pragma pack(pop)
std::unique_ptr<SDiskHeaderInfo> m_diskHeaderInfo; std::unique_ptr<SDiskHeaderInfo> m_disk_header_info;
std::vector<u8> m_apploader; std::vector<u8> m_apploader;
std::vector<u8> m_DOL; std::vector<u8> m_dol;
u64 m_fst_address; u64 m_fst_address;
u64 m_dol_address; u64 m_dol_address;