piwalker/dolphin
1
0
Fork 0
mirror of https://github.com/dolphin-emu/dolphin.git synced 2025-07-24 14:49:42 -06:00
Code Issues Packages Projects Releases Wiki Activity

DiscIO/DirectoryBlob: Add ability to have an offset for a partition ContentSource.

Browse Source
This commit is contained in:
Admiral H. Curtiss
2021-09-22 02:51:10 +02:00
parent a14436fe3f
commit b7a9cc37b1
2 changed files with 32 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
Source/Core/DiscIO/DirectoryBlob.cpp
View File

@ -112,16 +112,22 @@ bool DiscContent::Read(u64* offset, u64* length, u8** buffer) const
const u8* const content_pointer = std::get<const u8*>(m_content_source) + offset_in_content; const u8* const content_pointer = std::get<const u8*>(m_content_source) + offset_in_content;
std::copy(content_pointer, content_pointer + bytes_to_read, *buffer); std::copy(content_pointer, content_pointer + bytes_to_read, *buffer);
} }
else else if (std::holds_alternative<ContentPartition>(m_content_source))
{ {
DirectoryBlobReader* blob = std::get<DirectoryBlobReader*>(m_content_source); const auto& content = std::get<ContentPartition>(m_content_source);
DirectoryBlobReader* blob = content.m_reader;
const u64 decrypted_size = m_size * VolumeWii::BLOCK_DATA_SIZE / VolumeWii::BLOCK_TOTAL_SIZE; const u64 decrypted_size = m_size * VolumeWii::BLOCK_DATA_SIZE / VolumeWii::BLOCK_TOTAL_SIZE;
if (!blob->EncryptPartitionData(offset_in_content, bytes_to_read, *buffer, m_offset, if (!blob->EncryptPartitionData(content.m_offset + offset_in_content, bytes_to_read, *buffer,
decrypted_size)) content.m_partition_data_offset, decrypted_size))
{ {
return false; return false;
} }
} }
else
{
PanicAlertFmt("DirectoryBlob: Invalid content source in DiscContent.");
return false;
}
*length -= bytes_to_read; *length -= bytes_to_read;
*buffer += bytes_to_read; *buffer += bytes_to_read;
@ -545,9 +551,10 @@ void DirectoryBlobReader::SetPartitions(std::vector<PartitionWithType>&& partiti
SetPartitionHeader(&partitions[i].partition, partition_address); SetPartitionHeader(&partitions[i].partition, partition_address);
const u64 data_size = partitions[i].partition.GetDataSize(); const u64 data_size = partitions[i].partition.GetDataSize();
m_partitions.emplace(partition_address + PARTITION_DATA_OFFSET, const u64 partition_data_offset = partition_address + PARTITION_DATA_OFFSET;
std::move(partitions[i].partition)); m_partitions.emplace(partition_data_offset, std::move(partitions[i].partition));
m_nonpartition_contents.Add(partition_address + PARTITION_DATA_OFFSET, data_size, this); m_nonpartition_contents.Add(partition_data_offset, data_size,
ContentPartition{this, 0, partition_data_offset});
const u64 unaligned_next_partition_address = VolumeWii::EncryptedPartitionOffsetToRawOffset( const u64 unaligned_next_partition_address = VolumeWii::EncryptedPartitionOffsetToRawOffset(
data_size, Partition(partition_address), PARTITION_DATA_OFFSET); data_size, Partition(partition_address), PARTITION_DATA_OFFSET);
partition_address = Common::AlignUp(unaligned_next_partition_address, 0x10000ull); partition_address = Common::AlignUp(unaligned_next_partition_address, 0x10000ull);

23
Source/Core/DiscIO/DirectoryBlob.h
View File

@ -43,11 +43,24 @@ struct ContentFile
u64 m_offset; u64 m_offset;
}; };
using ContentSource = // Content chunk that loads data from a DirectoryBlobReader.
std::variant<ContentFile, // File // Intented for representing a partition within a disc.
const u8*, // Memory struct ContentPartition
DirectoryBlobReader* // Partition (which one it is is determined by m_offset) {
>; // The reader to read data from.
DirectoryBlobReader* m_reader;
// Offset from the start of the partition for the first byte represented by this chunk.
u64 m_offset;
// The value passed as partition_data_offset to EncryptPartitionData().
u64 m_partition_data_offset;
};
using ContentSource = std::variant<ContentFile, // File
const u8*, // Memory
ContentPartition // Partition
>;
class DiscContent class DiscContent
{ {