summaryrefslogtreecommitdiff
path: root/Source/Core/DiscIO/VolumeDirectory.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'Source/Core/DiscIO/VolumeDirectory.cpp')
-rw-r--r--Source/Core/DiscIO/VolumeDirectory.cpp278
1 files changed, 138 insertions, 140 deletions
diff --git a/Source/Core/DiscIO/VolumeDirectory.cpp b/Source/Core/DiscIO/VolumeDirectory.cpp
index 7ec043786f..504d17c20f 100644
--- a/Source/Core/DiscIO/VolumeDirectory.cpp
+++ b/Source/Core/DiscIO/VolumeDirectory.cpp
@@ -24,30 +24,30 @@
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_ID_LENGTH;
-CVolumeDirectory::CVolumeDirectory(const std::string& _rDirectory, bool _bIsWii,
- const std::string& _rApploader, const std::string& _rDOL)
- : m_dataStartAddress(-1), m_diskHeader(DISKHEADERINFO_ADDRESS),
- m_diskHeaderInfo(std::make_unique<SDiskHeaderInfo>()), m_fst_address(0), m_dol_address(0)
+CVolumeDirectory::CVolumeDirectory(const std::string& directory, bool is_wii,
+ const std::string& apploader, const std::string& dol)
+ : m_data_start_address(-1), m_disk_header(DISKHEADERINFO_ADDRESS),
+ 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
SetGameID("AGBJ01");
SetName("Default name");
- if (_bIsWii)
+ if (is_wii)
SetDiskTypeWii();
else
SetDiskTypeGC();
// Don't load the DOL if we don't have an apploader
- if (SetApploader(_rApploader))
- SetDOL(_rDOL);
+ if (SetApploader(apploader))
+ SetDOL(dol);
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.
// 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");
// header
- if (_Offset < DISKHEADERINFO_ADDRESS)
+ if (offset < DISKHEADERINFO_ADDRESS)
{
- WriteToBuffer(DISKHEADER_ADDRESS, DISKHEADERINFO_ADDRESS, m_diskHeader.data(), _Offset, _Length,
- _pBuffer);
+ WriteToBuffer(DISKHEADER_ADDRESS, DISKHEADERINFO_ADDRESS, m_disk_header.data(), &offset,
+ &length, &buffer);
}
// 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(),
- _Offset, _Length, _pBuffer);
+ WriteToBuffer(DISKHEADERINFO_ADDRESS, sizeof(m_disk_header_info), (u8*)m_disk_header_info.get(),
+ &offset, &length, &buffer);
}
// 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,
- _pBuffer);
+ WriteToBuffer(APPLOADER_ADDRESS, m_apploader.size(), m_apploader.data(), &offset, &length,
+ &buffer);
}
// 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
- 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;
// Determine which file the offset refers to
- std::map<u64, std::string>::const_iterator fileIter = m_virtualDisk.lower_bound(_Offset);
- if (fileIter->first > _Offset && fileIter != m_virtualDisk.begin())
+ std::map<u64, std::string>::const_iterator fileIter = m_virtual_disk.lower_bound(offset);
+ if (fileIter->first > offset && fileIter != m_virtual_disk.begin())
--fileIter;
// 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);
- u64 fileOffset = _Offset - fileIter->first;
+ _dbg_assert_(DVDINTERFACE, fileIter->first <= offset);
+ u64 fileOffset = offset - fileIter->first;
const std::string fileName = fileIter->second;
File::IOFile file(fileName, "rb");
@@ -132,24 +132,24 @@ bool CVolumeDirectory::Read(u64 _Offset, u64 _Length, u8* _pBuffer, bool decrypt
if (fileOffset < fileSize)
{
- u64 fileBytes = std::min(fileSize - fileOffset, _Length);
+ u64 fileBytes = std::min(fileSize - fileOffset, length);
if (!file.Seek(fileOffset, SEEK_SET))
return false;
- if (!file.ReadBytes(_pBuffer, fileBytes))
+ if (!file.ReadBytes(buffer, fileBytes))
return false;
- _Length -= fileBytes;
- _pBuffer += fileBytes;
- _Offset += fileBytes;
+ length -= fileBytes;
+ buffer += fileBytes;
+ offset += fileBytes;
}
++fileIter;
- if (fileIter != m_virtualDisk.end())
+ if (fileIter != m_virtual_disk.end())
{
- _dbg_assert_(DVDINTERFACE, fileIter->first >= _Offset);
- PadToAddress(fileIter->first, _Offset, _Length, _pBuffer);
+ _dbg_assert_(DVDINTERFACE, fileIter->first >= offset);
+ 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
{
- 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)
{
- 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
{
- return CountrySwitch(m_diskHeader[3]);
+ return CountrySwitch(m_disk_header[3]);
}
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)
{
size_t length = std::min(name.length(), MAX_NAME_LENGTH);
- memcpy(&m_diskHeader[0x20], name.c_str(), length);
- m_diskHeader[length + 0x20] = 0;
+ memcpy(&m_disk_header[0x20], name.c_str(), length);
+ m_disk_header[length + 0x20] = 0;
}
u64 CVolumeDirectory::GetFSTSize() const
@@ -246,67 +246,67 @@ u64 CVolumeDirectory::GetRawSize() const
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
// and directory names never will; both assumptions are often
// right but in general wrong.
- size_t extensionStart = directoryName.find_last_of('.');
- if (extensionStart != std::string::npos && extensionStart > lastSep)
+ size_t extension_start = result.find_last_of('.');
+ if (extension_start != std::string::npos && extension_start > last_separator)
{
- directoryName.resize(lastSep);
+ result.resize(last_separator);
}
}
else
{
- directoryName.resize(lastSep);
+ result.resize(last_separator);
}
- return directoryName;
+ return result;
}
void CVolumeDirectory::SetDiskTypeWii()
{
- Write32(0x5d1c9ea3, 0x18, &m_diskHeader);
- memset(&m_diskHeader[0x1c], 0, 4);
+ Write32(0x5d1c9ea3, 0x18, &m_disk_header);
+ memset(&m_disk_header[0x1c], 0, 4);
m_is_wii = true;
- m_addressShift = 2;
+ m_address_shift = 2;
}
void CVolumeDirectory::SetDiskTypeGC()
{
- memset(&m_diskHeader[0x18], 0, 4);
- Write32(0xc2339f3d, 0x1c, &m_diskHeader);
+ memset(&m_disk_header[0x18], 0, 4);
+ Write32(0xc2339f3d, 0x1c, &m_disk_header);
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;
- if (!File::ReadFileToString(_rApploader, data))
+ if (!File::ReadFileToString(apploader, data))
{
PanicAlertT("Apploader unable to load from file");
return false;
}
- size_t apploaderSize = 0x20 + Common::swap32(*(u32*)&data.data()[0x14]) +
- Common::swap32(*(u32*)&data.data()[0x18]);
- if (apploaderSize != data.size())
+ size_t apploader_size = 0x20 + Common::swap32(*(u32*)&data.data()[0x14]) +
+ Common::swap32(*(u32*)&data.data()[0x18]);
+ if (apploader_size != data.size())
{
PanicAlertT("Apploader is the wrong size...is it really an apploader?");
return false;
}
- m_apploader.resize(apploaderSize);
+ m_apploader.resize(apploader_size);
std::copy(data.begin(), data.end(), m_apploader.begin());
// 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;
- File::ReadFileToString(rDOL, data);
- m_DOL.resize(data.size());
- std::copy(data.begin(), data.end(), m_DOL.begin());
+ File::ReadFileToString(dol, data);
+ m_dol.resize(data.size());
+ 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)
- 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()
{
- 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);
- m_fstNameOffset = rootEntry.size * ENTRY_SIZE; // offset of name table in FST
- m_FSTData.resize(m_fstNameOffset + name_table_size);
+ m_fst_name_offset = rootEntry.size * ENTRY_SIZE; // offset of name table in FST
+ 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 (m_fst_address == 0)
m_fst_address = APPLOADER_ADDRESS + 0x2000;
// 4 byte aligned start of data on disk
- m_dataStartAddress = Common::AlignUp(m_fst_address + m_FSTData.size(), 0x8000ull);
- u64 curDataAddress = m_dataStartAddress;
+ m_data_start_address = Common::AlignUp(m_fst_address + m_fst_data.size(), 0x8000ull);
+ u64 current_data_address = m_data_start_address;
- u32 fstOffset = 0; // Offset within FST data
- u32 nameOffset = 0; // Offset within name table
- u32 rootOffset = 0; // Offset of root of FST
+ u32 fst_offset = 0; // Offset within FST data
+ u32 name_offset = 0; // Offset within name table
+ u32 root_offset = 0; // Offset of root of FST
// 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
- _dbg_assert_(DVDINTERFACE, nameOffset == name_table_size);
+ _dbg_assert_(DVDINTERFACE, name_offset == name_table_size);
// write FST size and location
- Write32((u32)(m_fst_address >> m_addressShift), 0x0424, &m_diskHeader);
- Write32((u32)(m_FSTData.size() >> m_addressShift), 0x0428, &m_diskHeader);
- Write32((u32)(m_FSTData.size() >> m_addressShift), 0x042c, &m_diskHeader);
+ Write32((u32)(m_fst_address >> m_address_shift), 0x0424, &m_disk_header);
+ Write32((u32)(m_fst_data.size() >> m_address_shift), 0x0428, &m_disk_header);
+ Write32((u32)(m_fst_data.size() >> m_address_shift), 0x042c, &m_disk_header);
}
-void CVolumeDirectory::WriteToBuffer(u64 _SrcStartAddress, u64 _SrcLength, const u8* _Src,
- u64& _Address, u64& _Length, u8*& _pBuffer) const
+void CVolumeDirectory::WriteToBuffer(u64 source_start_address, u64 source_length, const u8* source,
+ u64* address, u64* length, u8** buffer) const
{
- if (_Length == 0)
+ if (*length == 0)
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;
- _pBuffer += srcBytes;
- _Address += srcBytes;
+ *length -= bytes_to_read;
+ *buffer += bytes_to_read;
+ *address += bytes_to_read;
}
}
-void CVolumeDirectory::PadToAddress(u64 _StartAddress, u64& _Address, u64& _Length,
- u8*& _pBuffer) const
+void CVolumeDirectory::PadToAddress(u64 start_address, u64* address, u64* length, u8** buffer) const
{
- if (_StartAddress > _Address && _Length > 0)
+ if (start_address > *address && *length > 0)
{
- u64 padBytes = std::min(_StartAddress - _Address, _Length);
- memset(_pBuffer, 0, (size_t)padBytes);
- _Length -= padBytes;
- _pBuffer += padBytes;
- _Address += padBytes;
+ u64 padBytes = std::min(start_address - *address, *length);
+ memset(*buffer, 0, (size_t)padBytes);
+ *length -= padBytes;
+ *buffer += padBytes;
+ *address += padBytes;
}
}
@@ -417,31 +416,31 @@ void CVolumeDirectory::Write32(u32 data, u32 offset, std::vector<u8>* const buff
(*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)
{
- m_FSTData[entryOffset++] = type;
+ m_fst_data[(*entry_offset)++] = type;
- m_FSTData[entryOffset++] = (nameOffset >> 16) & 0xff;
- m_FSTData[entryOffset++] = (nameOffset >> 8) & 0xff;
- m_FSTData[entryOffset++] = (nameOffset)&0xff;
+ m_fst_data[(*entry_offset)++] = (name_offset >> 16) & 0xff;
+ m_fst_data[(*entry_offset)++] = (name_offset >> 8) & 0xff;
+ m_fst_data[(*entry_offset)++] = (name_offset)&0xff;
- Write32((u32)(dataOffset >> m_addressShift), entryOffset, &m_FSTData);
- entryOffset += 4;
+ Write32((u32)(data_offset >> m_address_shift), *entry_offset, &m_fst_data);
+ *entry_offset += 4;
- Write32((u32)length, entryOffset, &m_FSTData);
- entryOffset += 4;
+ Write32((u32)length, *entry_offset, &m_fst_data);
+ *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,
- u32& nameOffset, u64& dataOffset, u32 parentEntryNum)
+void CVolumeDirectory::WriteDirectory(const File::FSTEntry& parent_entry, u32* fst_offset,
+ u32* name_offset, u64* data_offset, u32 parent_entry_index)
{
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)
{
- u32 myOffset = fstOffset;
- u32 myEntryNum = myOffset / ENTRY_SIZE;
- WriteEntryData(fstOffset, DIRECTORY_ENTRY, nameOffset, parentEntryNum,
- myEntryNum + entry.size + 1);
- WriteEntryName(nameOffset, entry.virtualName);
+ u32 entry_index = *fst_offset / ENTRY_SIZE;
+ WriteEntryData(fst_offset, DIRECTORY_ENTRY, *name_offset, parent_entry_index,
+ entry_index + entry.size + 1);
+ WriteEntryName(name_offset, entry.virtualName);
- WriteDirectory(entry, fstOffset, nameOffset, dataOffset, myEntryNum);
+ WriteDirectory(entry, fst_offset, name_offset, data_offset, entry_index);
}
else
{
// put entry in FST
- WriteEntryData(fstOffset, FILE_ENTRY, nameOffset, dataOffset, entry.size);
- WriteEntryName(nameOffset, entry.virtualName);
+ WriteEntryData(fst_offset, FILE_ENTRY, *name_offset, *data_offset, entry.size);
+ WriteEntryName(name_offset, entry.virtualName);
// write entry to virtual disk
- _dbg_assert_(DVDINTERFACE, m_virtualDisk.find(dataOffset) == m_virtualDisk.end());
- m_virtualDisk.emplace(dataOffset, entry.physicalName);
+ _dbg_assert_(DVDINTERFACE, m_virtual_disk.find(*data_offset) == m_virtual_disk.end());
+ m_virtual_disk.emplace(*data_offset, entry.physicalName);
// 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;
- for (const File::FSTEntry& entry : parentEntry.children)
+ u32 name_size = 0;
+ for (const File::FSTEntry& entry : parent_entry.children)
{
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