Merge pull request #4559 from JosJuice/volumedirectory-totalnamesize-local

VolumeDirectory: Make m_totalNameSize a local variable
This commit is contained in:
Mat M 2016-12-25 17:30:21 -05:00 committed by GitHub
commit 8da0396fc4
2 changed files with 11 additions and 27 deletions

View File

@ -24,12 +24,14 @@
namespace DiscIO namespace DiscIO
{ {
static u32 ComputeNameSize(const File::FSTEntry& parentEntry);
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& _rDirectory, bool _bIsWii,
const std::string& _rApploader, const std::string& _rDOL) const std::string& _rApploader, const std::string& _rDOL)
: m_totalNameSize(0), m_dataStartAddress(-1), m_diskHeader(DISKHEADERINFO_ADDRESS), : m_dataStartAddress(-1), m_diskHeader(DISKHEADERINFO_ADDRESS),
m_diskHeaderInfo(std::make_unique<SDiskHeaderInfo>()), m_fst_address(0), m_dol_address(0) m_diskHeaderInfo(std::make_unique<SDiskHeaderInfo>()), m_fst_address(0), m_dol_address(0)
{ {
m_rootDirectory = ExtractDirectoryName(_rDirectory); m_rootDirectory = ExtractDirectoryName(_rDirectory);
@ -340,13 +342,11 @@ void CVolumeDirectory::BuildFST()
{ {
m_FSTData.clear(); m_FSTData.clear();
File::FSTEntry rootEntry; File::FSTEntry rootEntry = File::ScanDirectoryTree(m_rootDirectory, true);
u32 name_table_size = ComputeNameSize(rootEntry);
// read data from physical disk to rootEntry m_fstNameOffset = rootEntry.size * ENTRY_SIZE; // offset of name table in FST
u64 totalEntries = AddDirectoryEntries(m_rootDirectory, rootEntry) + 1; m_FSTData.resize(m_fstNameOffset + name_table_size);
m_fstNameOffset = totalEntries * ENTRY_SIZE; // offset in FST nameTable
m_FSTData.resize(m_fstNameOffset + m_totalNameSize);
// 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)
@ -361,7 +361,7 @@ void CVolumeDirectory::BuildFST()
u32 rootOffset = 0; // Offset of root of FST u32 rootOffset = 0; // Offset of root of FST
// write root entry // write root entry
WriteEntryData(fstOffset, DIRECTORY_ENTRY, 0, 0, totalEntries); WriteEntryData(fstOffset, DIRECTORY_ENTRY, 0, 0, rootEntry.size);
for (auto& entry : rootEntry.children) for (auto& entry : rootEntry.children)
{ {
@ -369,7 +369,7 @@ void CVolumeDirectory::BuildFST()
} }
// overflow check // overflow check
_dbg_assert_(DVDINTERFACE, nameOffset == m_totalNameSize); _dbg_assert_(DVDINTERFACE, nameOffset == 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_addressShift), 0x0424, &m_diskHeader);
@ -477,25 +477,14 @@ void CVolumeDirectory::WriteEntry(const File::FSTEntry& entry, u32& fstOffset, u
static u32 ComputeNameSize(const File::FSTEntry& parentEntry) static u32 ComputeNameSize(const File::FSTEntry& parentEntry)
{ {
u32 nameSize = 0; u32 nameSize = 0;
const std::vector<File::FSTEntry>& children = parentEntry.children; for (const File::FSTEntry& entry : parentEntry.children)
for (auto it = children.cbegin(); it != children.cend(); ++it)
{ {
const File::FSTEntry& entry = *it;
if (entry.isDirectory) if (entry.isDirectory)
{
nameSize += ComputeNameSize(entry); nameSize += ComputeNameSize(entry);
}
nameSize += (u32)entry.virtualName.length() + 1; nameSize += (u32)entry.virtualName.length() + 1;
} }
return nameSize; return nameSize;
} }
u64 CVolumeDirectory::AddDirectoryEntries(const std::string& _Directory,
File::FSTEntry& parentEntry)
{
parentEntry = File::ScanDirectoryTree(_Directory, true);
m_totalNameSize += ComputeNameSize(parentEntry);
return parentEntry.size;
}
} // namespace } // namespace

View File

@ -88,15 +88,10 @@ private:
void WriteEntry(const File::FSTEntry& entry, u32& fstOffset, u32& nameOffset, u64& dataOffset, void WriteEntry(const File::FSTEntry& entry, u32& fstOffset, u32& nameOffset, u64& dataOffset,
u32 parentEntryNum); u32 parentEntryNum);
// returns number of entries found in _Directory
u64 AddDirectoryEntries(const std::string& _Directory, File::FSTEntry& parentEntry);
std::string m_rootDirectory; std::string m_rootDirectory;
std::map<u64, std::string> m_virtualDisk; std::map<u64, std::string> m_virtualDisk;
u32 m_totalNameSize;
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