mirror of
https://github.com/dolphin-emu/dolphin.git
synced 2025-05-03 07:18:04 +03:00
DiscIO: Move parts of Filesystem to the new file DiscExtractor
This commit is contained in:
parent
23bb029250
commit
d06b532150
12 changed files with 255 additions and 191 deletions
|
@ -3,6 +3,7 @@ set(SRCS
|
||||||
CISOBlob.cpp
|
CISOBlob.cpp
|
||||||
WbfsBlob.cpp
|
WbfsBlob.cpp
|
||||||
CompressedBlob.cpp
|
CompressedBlob.cpp
|
||||||
|
DiscExtractor.cpp
|
||||||
DiscScrubber.cpp
|
DiscScrubber.cpp
|
||||||
DriveBlob.cpp
|
DriveBlob.cpp
|
||||||
Enums.cpp
|
Enums.cpp
|
||||||
|
|
174
Source/Core/DiscIO/DiscExtractor.cpp
Normal file
174
Source/Core/DiscIO/DiscExtractor.cpp
Normal file
|
@ -0,0 +1,174 @@
|
||||||
|
// Copyright 2017 Dolphin Emulator Project
|
||||||
|
// Licensed under GPLv2+
|
||||||
|
// Refer to the license.txt file included.
|
||||||
|
|
||||||
|
#include "DiscIO/DiscExtractor.h"
|
||||||
|
|
||||||
|
#include <algorithm>
|
||||||
|
#include <cinttypes>
|
||||||
|
#include <optional>
|
||||||
|
|
||||||
|
#include "Common/CommonTypes.h"
|
||||||
|
#include "DiscIO/Enums.h"
|
||||||
|
#include "DiscIO/Filesystem.h"
|
||||||
|
#include "DiscIO/Volume.h"
|
||||||
|
|
||||||
|
namespace DiscIO
|
||||||
|
{
|
||||||
|
u64 ReadFile(const Volume& volume, const Partition& partition, const FileInfo* file_info,
|
||||||
|
u8* buffer, u64 max_buffer_size, u64 offset_in_file)
|
||||||
|
{
|
||||||
|
if (!file_info || file_info->IsDirectory())
|
||||||
|
return 0;
|
||||||
|
|
||||||
|
if (offset_in_file >= file_info->GetSize())
|
||||||
|
return 0;
|
||||||
|
|
||||||
|
u64 read_length = std::min(max_buffer_size, file_info->GetSize() - offset_in_file);
|
||||||
|
|
||||||
|
DEBUG_LOG(DISCIO, "Reading %" PRIx64 " bytes at %" PRIx64 " from file %s. Offset: %" PRIx64
|
||||||
|
" Size: %" PRIx32,
|
||||||
|
read_length, offset_in_file, file_info->GetPath().c_str(), file_info->GetOffset(),
|
||||||
|
file_info->GetSize());
|
||||||
|
|
||||||
|
volume.Read(file_info->GetOffset() + offset_in_file, read_length, buffer, partition);
|
||||||
|
|
||||||
|
return read_length;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool ExportFile(const Volume& volume, const Partition& partition, const FileInfo* file_info,
|
||||||
|
const std::string& export_filename)
|
||||||
|
{
|
||||||
|
if (!file_info || file_info->IsDirectory())
|
||||||
|
return false;
|
||||||
|
|
||||||
|
u64 remaining_size = file_info->GetSize();
|
||||||
|
u64 file_offset = file_info->GetOffset();
|
||||||
|
|
||||||
|
File::IOFile f(export_filename, "wb");
|
||||||
|
if (!f)
|
||||||
|
return false;
|
||||||
|
|
||||||
|
bool result = true;
|
||||||
|
|
||||||
|
while (remaining_size)
|
||||||
|
{
|
||||||
|
// Limit read size to 128 MB
|
||||||
|
size_t read_size = (size_t)std::min(remaining_size, (u64)0x08000000);
|
||||||
|
|
||||||
|
std::vector<u8> buffer(read_size);
|
||||||
|
|
||||||
|
result = volume.Read(file_offset, read_size, &buffer[0], partition);
|
||||||
|
|
||||||
|
if (!result)
|
||||||
|
break;
|
||||||
|
|
||||||
|
f.WriteBytes(&buffer[0], read_size);
|
||||||
|
|
||||||
|
remaining_size -= read_size;
|
||||||
|
file_offset += read_size;
|
||||||
|
}
|
||||||
|
|
||||||
|
return result;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool ExportApploader(const Volume& volume, const Partition& partition,
|
||||||
|
const std::string& export_folder)
|
||||||
|
{
|
||||||
|
if (!IsDisc(volume.GetVolumeType()))
|
||||||
|
return false;
|
||||||
|
|
||||||
|
std::optional<u32> apploader_size = volume.ReadSwapped<u32>(0x2440 + 0x14, partition);
|
||||||
|
const std::optional<u32> trailer_size = volume.ReadSwapped<u32>(0x2440 + 0x18, partition);
|
||||||
|
constexpr u32 header_size = 0x20;
|
||||||
|
if (!apploader_size || !trailer_size)
|
||||||
|
return false;
|
||||||
|
*apploader_size += *trailer_size + header_size;
|
||||||
|
DEBUG_LOG(DISCIO, "Apploader size -> %x", *apploader_size);
|
||||||
|
|
||||||
|
std::vector<u8> buffer(*apploader_size);
|
||||||
|
if (volume.Read(0x2440, *apploader_size, buffer.data(), partition))
|
||||||
|
{
|
||||||
|
std::string export_name(export_folder + "/apploader.img");
|
||||||
|
|
||||||
|
File::IOFile apploader_file(export_name, "wb");
|
||||||
|
if (apploader_file)
|
||||||
|
{
|
||||||
|
apploader_file.WriteBytes(buffer.data(), *apploader_size);
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
std::optional<u64> GetBootDOLOffset(const Volume& volume, const Partition& partition)
|
||||||
|
{
|
||||||
|
const Platform volume_type = volume.GetVolumeType();
|
||||||
|
if (!IsDisc(volume_type))
|
||||||
|
return {};
|
||||||
|
|
||||||
|
std::optional<u32> offset = volume.ReadSwapped<u32>(0x420, partition);
|
||||||
|
const u8 offset_shift = volume_type == Platform::WII_DISC ? 2 : 0;
|
||||||
|
return offset ? static_cast<u64>(*offset) << offset_shift : std::optional<u64>();
|
||||||
|
}
|
||||||
|
|
||||||
|
std::optional<u32> GetBootDOLSize(const Volume& volume, const Partition& partition, u64 dol_offset)
|
||||||
|
{
|
||||||
|
if (!IsDisc(volume.GetVolumeType()))
|
||||||
|
return {};
|
||||||
|
|
||||||
|
u32 dol_size = 0;
|
||||||
|
|
||||||
|
// Iterate through the 7 code segments
|
||||||
|
for (u8 i = 0; i < 7; i++)
|
||||||
|
{
|
||||||
|
const std::optional<u32> offset = volume.ReadSwapped<u32>(dol_offset + 0x00 + i * 4, partition);
|
||||||
|
const std::optional<u32> size = volume.ReadSwapped<u32>(dol_offset + 0x90 + i * 4, partition);
|
||||||
|
if (!offset || !size)
|
||||||
|
return {};
|
||||||
|
dol_size = std::max(*offset + *size, dol_size);
|
||||||
|
}
|
||||||
|
|
||||||
|
// Iterate through the 11 data segments
|
||||||
|
for (u8 i = 0; i < 11; i++)
|
||||||
|
{
|
||||||
|
const std::optional<u32> offset = volume.ReadSwapped<u32>(dol_offset + 0x1c + i * 4, partition);
|
||||||
|
const std::optional<u32> size = volume.ReadSwapped<u32>(dol_offset + 0xac + i * 4, partition);
|
||||||
|
if (!offset || !size)
|
||||||
|
return {};
|
||||||
|
dol_size = std::max(*offset + *size, dol_size);
|
||||||
|
}
|
||||||
|
|
||||||
|
return dol_size;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool ExportDOL(const Volume& volume, const Partition& partition, const std::string& export_folder)
|
||||||
|
{
|
||||||
|
if (!IsDisc(volume.GetVolumeType()))
|
||||||
|
return false;
|
||||||
|
|
||||||
|
std::optional<u64> dol_offset = GetBootDOLOffset(volume, partition);
|
||||||
|
if (!dol_offset)
|
||||||
|
return false;
|
||||||
|
std::optional<u32> dol_size = GetBootDOLSize(volume, partition, *dol_offset);
|
||||||
|
if (!dol_size)
|
||||||
|
return false;
|
||||||
|
|
||||||
|
std::vector<u8> buffer(*dol_size);
|
||||||
|
if (volume.Read(*dol_offset, *dol_size, buffer.data(), partition))
|
||||||
|
{
|
||||||
|
std::string export_name(export_folder + "/boot.dol");
|
||||||
|
|
||||||
|
File::IOFile dol_file(export_name, "wb");
|
||||||
|
if (dol_file)
|
||||||
|
{
|
||||||
|
dol_file.WriteBytes(&buffer[0], *dol_size);
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
} // namespace DiscIO
|
27
Source/Core/DiscIO/DiscExtractor.h
Normal file
27
Source/Core/DiscIO/DiscExtractor.h
Normal file
|
@ -0,0 +1,27 @@
|
||||||
|
// Copyright 2017 Dolphin Emulator Project
|
||||||
|
// Licensed under GPLv2+
|
||||||
|
// Refer to the license.txt file included.
|
||||||
|
|
||||||
|
#pragma once
|
||||||
|
|
||||||
|
#include <optional>
|
||||||
|
|
||||||
|
#include "Common/CommonTypes.h"
|
||||||
|
|
||||||
|
namespace DiscIO
|
||||||
|
{
|
||||||
|
class FileInfo;
|
||||||
|
struct Partition;
|
||||||
|
class Volume;
|
||||||
|
|
||||||
|
u64 ReadFile(const Volume& volume, const Partition& partition, const FileInfo* file_info,
|
||||||
|
u8* buffer, u64 max_buffer_size, u64 offset_in_file = 0);
|
||||||
|
bool ExportFile(const Volume& volume, const Partition& partition, const FileInfo* file_info,
|
||||||
|
const std::string& export_filename);
|
||||||
|
bool ExportApploader(const Volume& volume, const Partition& partition,
|
||||||
|
const std::string& export_folder);
|
||||||
|
std::optional<u64> GetBootDOLOffset(const Volume& volume, const Partition& partition);
|
||||||
|
std::optional<u32> GetBootDOLSize(const Volume& volume, const Partition& partition, u64 dol_offset);
|
||||||
|
bool ExportDOL(const Volume& volume, const Partition& partition, const std::string& export_folder);
|
||||||
|
|
||||||
|
} // namespace DiscIO
|
|
@ -39,6 +39,7 @@
|
||||||
<ClCompile Include="Blob.cpp" />
|
<ClCompile Include="Blob.cpp" />
|
||||||
<ClCompile Include="CISOBlob.cpp" />
|
<ClCompile Include="CISOBlob.cpp" />
|
||||||
<ClCompile Include="CompressedBlob.cpp" />
|
<ClCompile Include="CompressedBlob.cpp" />
|
||||||
|
<ClCompile Include="DiscExtractor.cpp" />
|
||||||
<ClCompile Include="DiscScrubber.cpp" />
|
<ClCompile Include="DiscScrubber.cpp" />
|
||||||
<ClCompile Include="DriveBlob.cpp" />
|
<ClCompile Include="DriveBlob.cpp" />
|
||||||
<ClCompile Include="Enums.cpp" />
|
<ClCompile Include="Enums.cpp" />
|
||||||
|
@ -60,6 +61,7 @@
|
||||||
<ClInclude Include="Blob.h" />
|
<ClInclude Include="Blob.h" />
|
||||||
<ClInclude Include="CISOBlob.h" />
|
<ClInclude Include="CISOBlob.h" />
|
||||||
<ClInclude Include="CompressedBlob.h" />
|
<ClInclude Include="CompressedBlob.h" />
|
||||||
|
<ClInclude Include="DiscExtractor.h" />
|
||||||
<ClInclude Include="DiscScrubber.h" />
|
<ClInclude Include="DiscScrubber.h" />
|
||||||
<ClInclude Include="DriveBlob.h" />
|
<ClInclude Include="DriveBlob.h" />
|
||||||
<ClInclude Include="Enums.h" />
|
<ClInclude Include="Enums.h" />
|
||||||
|
|
|
@ -1,6 +1,9 @@
|
||||||
<?xml version="1.0" encoding="utf-8"?>
|
<?xml version="1.0" encoding="utf-8"?>
|
||||||
<Project ToolsVersion="15.0" xmlns="http://schemas.microsoft.com/developer/msbuild/2003">
|
<Project ToolsVersion="15.0" xmlns="http://schemas.microsoft.com/developer/msbuild/2003">
|
||||||
<ItemGroup>
|
<ItemGroup>
|
||||||
|
<Filter Include="DiscExtractor">
|
||||||
|
<UniqueIdentifier>{51cdf366-d3fe-464a-9f89-c9f1592a6f1c}</UniqueIdentifier>
|
||||||
|
</Filter>
|
||||||
<Filter Include="DiscScrubber">
|
<Filter Include="DiscScrubber">
|
||||||
<UniqueIdentifier>{3873659a-9a30-4a58-af9e-8dad7d7eb627}</UniqueIdentifier>
|
<UniqueIdentifier>{3873659a-9a30-4a58-af9e-8dad7d7eb627}</UniqueIdentifier>
|
||||||
</Filter>
|
</Filter>
|
||||||
|
@ -75,6 +78,9 @@
|
||||||
<ClCompile Include="TGCBlob.cpp">
|
<ClCompile Include="TGCBlob.cpp">
|
||||||
<Filter>Volume\Blob</Filter>
|
<Filter>Volume\Blob</Filter>
|
||||||
</ClCompile>
|
</ClCompile>
|
||||||
|
<ClCompile Include="DiscExtractor.cpp">
|
||||||
|
<Filter>DiscExtractor</Filter>
|
||||||
|
</ClCompile>
|
||||||
</ItemGroup>
|
</ItemGroup>
|
||||||
<ItemGroup>
|
<ItemGroup>
|
||||||
<ClInclude Include="DiscScrubber.h">
|
<ClInclude Include="DiscScrubber.h">
|
||||||
|
@ -134,6 +140,9 @@
|
||||||
<ClInclude Include="TGCBlob.h">
|
<ClInclude Include="TGCBlob.h">
|
||||||
<Filter>Volume\Blob</Filter>
|
<Filter>Volume\Blob</Filter>
|
||||||
</ClInclude>
|
</ClInclude>
|
||||||
|
<ClInclude Include="DiscExtractor.h">
|
||||||
|
<Filter>DiscExtractor</Filter>
|
||||||
|
</ClInclude>
|
||||||
</ItemGroup>
|
</ItemGroup>
|
||||||
<ItemGroup>
|
<ItemGroup>
|
||||||
<Text Include="CMakeLists.txt" />
|
<Text Include="CMakeLists.txt" />
|
||||||
|
|
|
@ -2,6 +2,8 @@
|
||||||
// Licensed under GPLv2+
|
// Licensed under GPLv2+
|
||||||
// Refer to the license.txt file included.
|
// Refer to the license.txt file included.
|
||||||
|
|
||||||
|
#include "DiscIO/DiscScrubber.h"
|
||||||
|
|
||||||
#include <algorithm>
|
#include <algorithm>
|
||||||
#include <cinttypes>
|
#include <cinttypes>
|
||||||
#include <cstddef>
|
#include <cstddef>
|
||||||
|
@ -14,7 +16,8 @@
|
||||||
#include "Common/CommonTypes.h"
|
#include "Common/CommonTypes.h"
|
||||||
#include "Common/File.h"
|
#include "Common/File.h"
|
||||||
#include "Common/Logging/Log.h"
|
#include "Common/Logging/Log.h"
|
||||||
#include "DiscIO/DiscScrubber.h"
|
|
||||||
|
#include "DiscIO/DiscExtractor.h"
|
||||||
#include "DiscIO/Filesystem.h"
|
#include "DiscIO/Filesystem.h"
|
||||||
#include "DiscIO/Volume.h"
|
#include "DiscIO/Volume.h"
|
||||||
|
|
||||||
|
@ -200,10 +203,10 @@ bool DiscScrubber::ParsePartitionData(const Partition& partition, PartitionHeade
|
||||||
0x2440 + header->apploader_size + header->apploader_trailer_size);
|
0x2440 + header->apploader_size + header->apploader_trailer_size);
|
||||||
|
|
||||||
// DOL
|
// DOL
|
||||||
const std::optional<u64> dol_offset = filesystem->GetBootDOLOffset();
|
const std::optional<u64> dol_offset = GetBootDOLOffset(*m_disc, partition);
|
||||||
if (!dol_offset)
|
if (!dol_offset)
|
||||||
return false;
|
return false;
|
||||||
const std::optional<u64> dol_size = filesystem->GetBootDOLSize(*dol_offset);
|
const std::optional<u64> dol_size = GetBootDOLSize(*m_disc, partition, *dol_offset);
|
||||||
if (!dol_size)
|
if (!dol_size)
|
||||||
return false;
|
return false;
|
||||||
header->dol_offset = *dol_offset;
|
header->dol_offset = *dol_offset;
|
||||||
|
|
|
@ -330,148 +330,4 @@ std::unique_ptr<FileInfo> FileSystemGCWii::FindFileInfo(u64 disc_offset) const
|
||||||
return nullptr;
|
return nullptr;
|
||||||
}
|
}
|
||||||
|
|
||||||
u64 FileSystemGCWii::ReadFile(const FileInfo* file_info, u8* buffer, u64 max_buffer_size,
|
|
||||||
u64 offset_in_file) const
|
|
||||||
{
|
|
||||||
if (!file_info || file_info->IsDirectory())
|
|
||||||
return 0;
|
|
||||||
|
|
||||||
if (offset_in_file >= file_info->GetSize())
|
|
||||||
return 0;
|
|
||||||
|
|
||||||
u64 read_length = std::min(max_buffer_size, file_info->GetSize() - offset_in_file);
|
|
||||||
|
|
||||||
DEBUG_LOG(DISCIO, "Reading %" PRIx64 " bytes at %" PRIx64 " from file %s. Offset: %" PRIx64
|
|
||||||
" Size: %" PRIx32,
|
|
||||||
read_length, offset_in_file, file_info->GetPath().c_str(), file_info->GetOffset(),
|
|
||||||
file_info->GetSize());
|
|
||||||
|
|
||||||
m_volume->Read(file_info->GetOffset() + offset_in_file, read_length, buffer, m_partition);
|
|
||||||
return read_length;
|
|
||||||
}
|
|
||||||
|
|
||||||
bool FileSystemGCWii::ExportFile(const FileInfo* file_info,
|
|
||||||
const std::string& export_filename) const
|
|
||||||
{
|
|
||||||
if (!file_info || file_info->IsDirectory())
|
|
||||||
return false;
|
|
||||||
|
|
||||||
u64 remaining_size = file_info->GetSize();
|
|
||||||
u64 file_offset = file_info->GetOffset();
|
|
||||||
|
|
||||||
File::IOFile f(export_filename, "wb");
|
|
||||||
if (!f)
|
|
||||||
return false;
|
|
||||||
|
|
||||||
bool result = true;
|
|
||||||
|
|
||||||
while (remaining_size)
|
|
||||||
{
|
|
||||||
// Limit read size to 128 MB
|
|
||||||
size_t read_size = (size_t)std::min(remaining_size, (u64)0x08000000);
|
|
||||||
|
|
||||||
std::vector<u8> buffer(read_size);
|
|
||||||
|
|
||||||
result = m_volume->Read(file_offset, read_size, &buffer[0], m_partition);
|
|
||||||
|
|
||||||
if (!result)
|
|
||||||
break;
|
|
||||||
|
|
||||||
f.WriteBytes(&buffer[0], read_size);
|
|
||||||
|
|
||||||
remaining_size -= read_size;
|
|
||||||
file_offset += read_size;
|
|
||||||
}
|
|
||||||
|
|
||||||
return result;
|
|
||||||
}
|
|
||||||
|
|
||||||
bool FileSystemGCWii::ExportApploader(const std::string& export_folder) const
|
|
||||||
{
|
|
||||||
std::optional<u32> apploader_size = m_volume->ReadSwapped<u32>(0x2440 + 0x14, m_partition);
|
|
||||||
const std::optional<u32> trailer_size = m_volume->ReadSwapped<u32>(0x2440 + 0x18, m_partition);
|
|
||||||
constexpr u32 header_size = 0x20;
|
|
||||||
if (!apploader_size || !trailer_size)
|
|
||||||
return false;
|
|
||||||
*apploader_size += *trailer_size + header_size;
|
|
||||||
DEBUG_LOG(DISCIO, "Apploader size -> %x", *apploader_size);
|
|
||||||
|
|
||||||
std::vector<u8> buffer(*apploader_size);
|
|
||||||
if (m_volume->Read(0x2440, *apploader_size, buffer.data(), m_partition))
|
|
||||||
{
|
|
||||||
std::string export_name(export_folder + "/apploader.img");
|
|
||||||
|
|
||||||
File::IOFile apploader_file(export_name, "wb");
|
|
||||||
if (apploader_file)
|
|
||||||
{
|
|
||||||
apploader_file.WriteBytes(buffer.data(), *apploader_size);
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
std::optional<u64> FileSystemGCWii::GetBootDOLOffset() const
|
|
||||||
{
|
|
||||||
std::optional<u32> offset = m_volume->ReadSwapped<u32>(0x420, m_partition);
|
|
||||||
return offset ? static_cast<u64>(*offset) << m_offset_shift : std::optional<u64>();
|
|
||||||
}
|
|
||||||
|
|
||||||
std::optional<u32> FileSystemGCWii::GetBootDOLSize(u64 dol_offset) const
|
|
||||||
{
|
|
||||||
u32 dol_size = 0;
|
|
||||||
|
|
||||||
// Iterate through the 7 code segments
|
|
||||||
for (u8 i = 0; i < 7; i++)
|
|
||||||
{
|
|
||||||
const std::optional<u32> offset =
|
|
||||||
m_volume->ReadSwapped<u32>(dol_offset + 0x00 + i * 4, m_partition);
|
|
||||||
const std::optional<u32> size =
|
|
||||||
m_volume->ReadSwapped<u32>(dol_offset + 0x90 + i * 4, m_partition);
|
|
||||||
if (!offset || !size)
|
|
||||||
return {};
|
|
||||||
dol_size = std::max(*offset + *size, dol_size);
|
|
||||||
}
|
|
||||||
|
|
||||||
// Iterate through the 11 data segments
|
|
||||||
for (u8 i = 0; i < 11; i++)
|
|
||||||
{
|
|
||||||
const std::optional<u32> offset =
|
|
||||||
m_volume->ReadSwapped<u32>(dol_offset + 0x1c + i * 4, m_partition);
|
|
||||||
const std::optional<u32> size =
|
|
||||||
m_volume->ReadSwapped<u32>(dol_offset + 0xac + i * 4, m_partition);
|
|
||||||
if (!offset || !size)
|
|
||||||
return {};
|
|
||||||
dol_size = std::max(*offset + *size, dol_size);
|
|
||||||
}
|
|
||||||
|
|
||||||
return dol_size;
|
|
||||||
}
|
|
||||||
|
|
||||||
bool FileSystemGCWii::ExportDOL(const std::string& export_folder) const
|
|
||||||
{
|
|
||||||
std::optional<u64> dol_offset = GetBootDOLOffset();
|
|
||||||
if (!dol_offset)
|
|
||||||
return false;
|
|
||||||
std::optional<u32> dol_size = GetBootDOLSize(*dol_offset);
|
|
||||||
if (!dol_size)
|
|
||||||
return false;
|
|
||||||
|
|
||||||
std::vector<u8> buffer(*dol_size);
|
|
||||||
if (m_volume->Read(*dol_offset, *dol_size, buffer.data(), m_partition))
|
|
||||||
{
|
|
||||||
std::string export_name(export_folder + "/boot.dol");
|
|
||||||
|
|
||||||
File::IOFile dol_file(export_name, "wb");
|
|
||||||
if (dol_file)
|
|
||||||
{
|
|
||||||
dol_file.WriteBytes(&buffer[0], *dol_size);
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
} // namespace
|
} // namespace
|
||||||
|
|
|
@ -93,14 +93,6 @@ public:
|
||||||
std::unique_ptr<FileInfo> FindFileInfo(const std::string& path) const override;
|
std::unique_ptr<FileInfo> FindFileInfo(const std::string& path) const override;
|
||||||
std::unique_ptr<FileInfo> FindFileInfo(u64 disc_offset) const override;
|
std::unique_ptr<FileInfo> FindFileInfo(u64 disc_offset) const override;
|
||||||
|
|
||||||
u64 ReadFile(const FileInfo* file_info, u8* buffer, u64 max_buffer_size,
|
|
||||||
u64 offset_in_file) const override;
|
|
||||||
bool ExportFile(const FileInfo* file_info, const std::string& export_filename) const override;
|
|
||||||
bool ExportApploader(const std::string& export_folder) const override;
|
|
||||||
bool ExportDOL(const std::string& export_folder) const override;
|
|
||||||
std::optional<u64> GetBootDOLOffset() const override;
|
|
||||||
std::optional<u32> GetBootDOLSize(u64 dol_offset) const override;
|
|
||||||
|
|
||||||
private:
|
private:
|
||||||
bool m_valid;
|
bool m_valid;
|
||||||
u32 m_offset_shift;
|
u32 m_offset_shift;
|
||||||
|
|
|
@ -120,14 +120,6 @@ public:
|
||||||
// Returns nullptr if not found
|
// Returns nullptr if not found
|
||||||
virtual std::unique_ptr<FileInfo> FindFileInfo(u64 disc_offset) const = 0;
|
virtual std::unique_ptr<FileInfo> FindFileInfo(u64 disc_offset) const = 0;
|
||||||
|
|
||||||
virtual u64 ReadFile(const FileInfo* file_info, u8* buffer, u64 max_buffer_size,
|
|
||||||
u64 offset_in_file = 0) const = 0;
|
|
||||||
virtual bool ExportFile(const FileInfo* file_info, const std::string& export_filename) const = 0;
|
|
||||||
virtual bool ExportApploader(const std::string& export_folder) const = 0;
|
|
||||||
virtual bool ExportDOL(const std::string& export_folder) const = 0;
|
|
||||||
virtual std::optional<u64> GetBootDOLOffset() const = 0;
|
|
||||||
virtual std::optional<u32> GetBootDOLSize(u64 dol_offset) const = 0;
|
|
||||||
|
|
||||||
virtual const Partition GetPartition() const { return m_partition; }
|
virtual const Partition GetPartition() const { return m_partition; }
|
||||||
protected:
|
protected:
|
||||||
const Volume* const m_volume;
|
const Volume* const m_volume;
|
||||||
|
|
|
@ -16,7 +16,9 @@
|
||||||
#include "Common/Logging/Log.h"
|
#include "Common/Logging/Log.h"
|
||||||
#include "Common/MsgHandler.h"
|
#include "Common/MsgHandler.h"
|
||||||
#include "Common/StringUtil.h"
|
#include "Common/StringUtil.h"
|
||||||
|
|
||||||
#include "DiscIO/Blob.h"
|
#include "DiscIO/Blob.h"
|
||||||
|
#include "DiscIO/DiscExtractor.h"
|
||||||
#include "DiscIO/Enums.h"
|
#include "DiscIO/Enums.h"
|
||||||
#include "DiscIO/Filesystem.h"
|
#include "DiscIO/Filesystem.h"
|
||||||
#include "DiscIO/Volume.h"
|
#include "DiscIO/Volume.h"
|
||||||
|
@ -190,8 +192,8 @@ void VolumeGC::LoadBannerFile() const
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
if (file_size !=
|
if (file_size != ReadFile(*this, PARTITION_NONE, file_info.get(),
|
||||||
file_system->ReadFile(file_info.get(), reinterpret_cast<u8*>(&banner_file), file_size))
|
reinterpret_cast<u8*>(&banner_file), file_size))
|
||||||
{
|
{
|
||||||
WARN_LOG(DISCIO, "Could not read opening.bnr.");
|
WARN_LOG(DISCIO, "Could not read opening.bnr.");
|
||||||
return;
|
return;
|
||||||
|
|
|
@ -24,6 +24,7 @@
|
||||||
#include "Common/Swap.h"
|
#include "Common/Swap.h"
|
||||||
|
|
||||||
#include "DiscIO/Blob.h"
|
#include "DiscIO/Blob.h"
|
||||||
|
#include "DiscIO/DiscExtractor.h"
|
||||||
#include "DiscIO/Enums.h"
|
#include "DiscIO/Enums.h"
|
||||||
#include "DiscIO/Filesystem.h"
|
#include "DiscIO/Filesystem.h"
|
||||||
#include "DiscIO/Volume.h"
|
#include "DiscIO/Volume.h"
|
||||||
|
@ -274,8 +275,8 @@ std::map<Language, std::string> VolumeWii::GetLongNames() const
|
||||||
|
|
||||||
std::vector<u8> opening_bnr(NAMES_TOTAL_BYTES);
|
std::vector<u8> opening_bnr(NAMES_TOTAL_BYTES);
|
||||||
std::unique_ptr<FileInfo> file_info = file_system->FindFileInfo("opening.bnr");
|
std::unique_ptr<FileInfo> file_info = file_system->FindFileInfo("opening.bnr");
|
||||||
opening_bnr.resize(
|
opening_bnr.resize(ReadFile(*this, GetGamePartition(), file_info.get(), opening_bnr.data(),
|
||||||
file_system->ReadFile(file_info.get(), opening_bnr.data(), opening_bnr.size(), 0x5C));
|
opening_bnr.size(), 0x5C));
|
||||||
return ReadWiiNames(opening_bnr);
|
return ReadWiiNames(opening_bnr);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
@ -25,6 +25,7 @@
|
||||||
#include "Common/CommonPaths.h"
|
#include "Common/CommonPaths.h"
|
||||||
#include "Common/FileUtil.h"
|
#include "Common/FileUtil.h"
|
||||||
#include "Common/Logging/Log.h"
|
#include "Common/Logging/Log.h"
|
||||||
|
#include "DiscIO/DiscExtractor.h"
|
||||||
#include "DiscIO/Enums.h"
|
#include "DiscIO/Enums.h"
|
||||||
#include "DiscIO/Filesystem.h"
|
#include "DiscIO/Filesystem.h"
|
||||||
#include "DiscIO/Volume.h"
|
#include "DiscIO/Volume.h"
|
||||||
|
@ -261,32 +262,32 @@ void FilesystemPanel::OnExtractDirectories(wxCommandEvent& event)
|
||||||
|
|
||||||
void FilesystemPanel::OnExtractHeaderData(wxCommandEvent& event)
|
void FilesystemPanel::OnExtractHeaderData(wxCommandEvent& event)
|
||||||
{
|
{
|
||||||
DiscIO::FileSystem* file_system = nullptr;
|
|
||||||
const wxString path = wxDirSelector(_("Choose the folder to extract to"));
|
const wxString path = wxDirSelector(_("Choose the folder to extract to"));
|
||||||
|
|
||||||
if (path.empty())
|
if (path.empty())
|
||||||
return;
|
return;
|
||||||
|
|
||||||
|
DiscIO::Partition partition;
|
||||||
if (m_has_partitions)
|
if (m_has_partitions)
|
||||||
{
|
{
|
||||||
const auto* const selection_data = m_tree_ctrl->GetItemData(m_tree_ctrl->GetSelection());
|
const auto* const selection_data = m_tree_ctrl->GetItemData(m_tree_ctrl->GetSelection());
|
||||||
const auto* const partition = static_cast<const WiiPartition*>(selection_data);
|
const auto* const wii_partition = static_cast<const WiiPartition*>(selection_data);
|
||||||
|
|
||||||
file_system = partition->filesystem.get();
|
partition = wii_partition->filesystem->GetPartition();
|
||||||
}
|
}
|
||||||
else
|
else
|
||||||
{
|
{
|
||||||
file_system = m_filesystem.get();
|
partition = DiscIO::PARTITION_NONE;
|
||||||
}
|
}
|
||||||
|
|
||||||
bool ret = false;
|
bool ret = false;
|
||||||
if (event.GetId() == ID_EXTRACT_APPLOADER)
|
if (event.GetId() == ID_EXTRACT_APPLOADER)
|
||||||
{
|
{
|
||||||
ret = file_system->ExportApploader(WxStrToStr(path));
|
ret = DiscIO::ExportApploader(*m_opened_iso, partition, WxStrToStr(path));
|
||||||
}
|
}
|
||||||
else if (event.GetId() == ID_EXTRACT_DOL)
|
else if (event.GetId() == ID_EXTRACT_DOL)
|
||||||
{
|
{
|
||||||
ret = file_system->ExportDOL(WxStrToStr(path));
|
ret = DiscIO::ExportDOL(*m_opened_iso, partition, WxStrToStr(path));
|
||||||
}
|
}
|
||||||
|
|
||||||
if (!ret)
|
if (!ret)
|
||||||
|
@ -366,14 +367,15 @@ void FilesystemPanel::ExtractSingleFile(const wxString& output_file_path) const
|
||||||
// Remove "Partition x/"
|
// Remove "Partition x/"
|
||||||
selection_file_path.erase(0, slash_index + 1);
|
selection_file_path.erase(0, slash_index + 1);
|
||||||
|
|
||||||
partition->filesystem->ExportFile(
|
DiscIO::ExportFile(*m_opened_iso, partition->filesystem->GetPartition(),
|
||||||
partition->filesystem->FindFileInfo(WxStrToStr(selection_file_path)).get(),
|
partition->filesystem->FindFileInfo(WxStrToStr(selection_file_path)).get(),
|
||||||
WxStrToStr(output_file_path));
|
WxStrToStr(output_file_path));
|
||||||
}
|
}
|
||||||
else
|
else
|
||||||
{
|
{
|
||||||
m_filesystem->ExportFile(m_filesystem->FindFileInfo(WxStrToStr(selection_file_path)).get(),
|
DiscIO::ExportFile(*m_opened_iso, DiscIO::PARTITION_NONE,
|
||||||
WxStrToStr(output_file_path));
|
m_filesystem->FindFileInfo(WxStrToStr(selection_file_path)).get(),
|
||||||
|
WxStrToStr(output_file_path));
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -400,7 +402,8 @@ void FilesystemPanel::ExtractSingleDirectory(const wxString& output_folder)
|
||||||
}
|
}
|
||||||
|
|
||||||
static void ExtractDir(const std::string& full_path, const std::string& output_folder,
|
static void ExtractDir(const std::string& full_path, const std::string& output_folder,
|
||||||
const DiscIO::FileSystem& file_system, const DiscIO::FileInfo& directory,
|
const DiscIO::Volume& volume, const DiscIO::Partition partition,
|
||||||
|
const DiscIO::FileInfo& directory,
|
||||||
const std::function<bool(const std::string& path)>& update_progress)
|
const std::function<bool(const std::string& path)>& update_progress)
|
||||||
{
|
{
|
||||||
for (const DiscIO::FileInfo& file_info : directory)
|
for (const DiscIO::FileInfo& file_info : directory)
|
||||||
|
@ -416,13 +419,13 @@ static void ExtractDir(const std::string& full_path, const std::string& output_f
|
||||||
if (file_info.IsDirectory())
|
if (file_info.IsDirectory())
|
||||||
{
|
{
|
||||||
File::CreateFullPath(output_path);
|
File::CreateFullPath(output_path);
|
||||||
ExtractDir(path, output_folder, file_system, file_info, update_progress);
|
ExtractDir(path, output_folder, volume, partition, file_info, update_progress);
|
||||||
}
|
}
|
||||||
else
|
else
|
||||||
{
|
{
|
||||||
if (File::Exists(output_path))
|
if (File::Exists(output_path))
|
||||||
NOTICE_LOG(DISCIO, "%s already exists", output_path.c_str());
|
NOTICE_LOG(DISCIO, "%s already exists", output_path.c_str());
|
||||||
else if (!file_system.ExportFile(&file_info, output_path))
|
else if (!DiscIO::ExportFile(volume, partition, &file_info, output_path))
|
||||||
ERROR_LOG(DISCIO, "Could not export %s", output_path.c_str());
|
ERROR_LOG(DISCIO, "Could not export %s", output_path.c_str());
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -434,8 +437,8 @@ void FilesystemPanel::ExtractDirectories(const std::string& full_path,
|
||||||
{
|
{
|
||||||
if (full_path.empty()) // Root
|
if (full_path.empty()) // Root
|
||||||
{
|
{
|
||||||
filesystem.ExportApploader(output_folder);
|
DiscIO::ExportApploader(*m_opened_iso, filesystem.GetPartition(), output_folder);
|
||||||
filesystem.ExportDOL(output_folder);
|
DiscIO::ExportDOL(*m_opened_iso, filesystem.GetPartition(), output_folder);
|
||||||
}
|
}
|
||||||
|
|
||||||
std::unique_ptr<DiscIO::FileInfo> file_info = filesystem.FindFileInfo(full_path);
|
std::unique_ptr<DiscIO::FileInfo> file_info = filesystem.FindFileInfo(full_path);
|
||||||
|
@ -448,14 +451,16 @@ void FilesystemPanel::ExtractDirectories(const std::string& full_path,
|
||||||
wxPD_ESTIMATED_TIME | wxPD_REMAINING_TIME | wxPD_SMOOTH);
|
wxPD_ESTIMATED_TIME | wxPD_REMAINING_TIME | wxPD_SMOOTH);
|
||||||
|
|
||||||
File::CreateFullPath(output_folder + "/" + full_path);
|
File::CreateFullPath(output_folder + "/" + full_path);
|
||||||
ExtractDir(full_path, output_folder, filesystem, *file_info, [&](const std::string& path) {
|
ExtractDir(
|
||||||
dialog.SetTitle(wxString::Format(
|
full_path, output_folder, *m_opened_iso, filesystem.GetPartition(), *file_info,
|
||||||
"%s : %d%%", dialog_title.c_str(),
|
[&](const std::string& path) {
|
||||||
static_cast<u32>((static_cast<float>(progress) / static_cast<float>(size)) * 100)));
|
dialog.SetTitle(wxString::Format(
|
||||||
dialog.Update(progress, wxString::Format(_("Extracting %s"), StrToWxStr(path)));
|
"%s : %d%%", dialog_title.c_str(),
|
||||||
++progress;
|
static_cast<u32>((static_cast<float>(progress) / static_cast<float>(size)) * 100)));
|
||||||
return dialog.WasCancelled();
|
dialog.Update(progress, wxString::Format(_("Extracting %s"), StrToWxStr(path)));
|
||||||
});
|
++progress;
|
||||||
|
return dialog.WasCancelled();
|
||||||
|
});
|
||||||
}
|
}
|
||||||
|
|
||||||
wxString FilesystemPanel::BuildFilePathFromSelection() const
|
wxString FilesystemPanel::BuildFilePathFromSelection() const
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue