3
0
Fork 0
forked from suyu/suyu

fssystem: reduce overalignment of unbuffered storage operations

This commit is contained in:
Liam 2023-08-10 21:49:19 -04:00
parent 86f6b6b7b2
commit 0398b34370
5 changed files with 28 additions and 54 deletions

View file

@ -17,8 +17,6 @@ Result HierarchicalIntegrityVerificationStorage::Initialize(
const HierarchicalIntegrityVerificationInformation& info, const HierarchicalIntegrityVerificationInformation& info,
HierarchicalStorageInformation storage, int max_data_cache_entries, int max_hash_cache_entries, HierarchicalStorageInformation storage, int max_data_cache_entries, int max_hash_cache_entries,
s8 buffer_level) { s8 buffer_level) {
using AlignedStorage = AlignmentMatchingStoragePooledBuffer<1>;
// Validate preconditions. // Validate preconditions.
ASSERT(IntegrityMinLayerCount <= info.max_layers && info.max_layers <= IntegrityMaxLayerCount); ASSERT(IntegrityMinLayerCount <= info.max_layers && info.max_layers <= IntegrityMaxLayerCount);
@ -38,8 +36,7 @@ Result HierarchicalIntegrityVerificationStorage::Initialize(
}; };
// Initialize the top level buffer storage. // Initialize the top level buffer storage.
m_buffer_storages[0] = std::make_shared<AlignedStorage>( m_buffer_storages[0] = m_verify_storages[0];
m_verify_storages[0], static_cast<s64>(1) << info.info[0].block_order);
R_UNLESS(m_buffer_storages[0] != nullptr, ResultAllocationMemoryFailedAllocateShared); R_UNLESS(m_buffer_storages[0] != nullptr, ResultAllocationMemoryFailedAllocateShared);
// Prepare to initialize the level storages. // Prepare to initialize the level storages.
@ -65,8 +62,7 @@ Result HierarchicalIntegrityVerificationStorage::Initialize(
static_cast<s64>(1) << info.info[level].block_order, false); static_cast<s64>(1) << info.info[level].block_order, false);
// Initialize the buffer storage. // Initialize the buffer storage.
m_buffer_storages[level + 1] = std::make_shared<AlignedStorage>( m_buffer_storages[level + 1] = m_verify_storages[level + 1];
m_verify_storages[level + 1], static_cast<s64>(1) << info.info[level + 1].block_order);
R_UNLESS(m_buffer_storages[level + 1] != nullptr, R_UNLESS(m_buffer_storages[level + 1] != nullptr,
ResultAllocationMemoryFailedAllocateShared); ResultAllocationMemoryFailedAllocateShared);
} }
@ -82,8 +78,7 @@ Result HierarchicalIntegrityVerificationStorage::Initialize(
static_cast<s64>(1) << info.info[level].block_order, true); static_cast<s64>(1) << info.info[level].block_order, true);
// Initialize the buffer storage. // Initialize the buffer storage.
m_buffer_storages[level + 1] = std::make_shared<AlignedStorage>( m_buffer_storages[level + 1] = m_verify_storages[level + 1];
m_verify_storages[level + 1], static_cast<s64>(1) << info.info[level + 1].block_order);
R_UNLESS(m_buffer_storages[level + 1] != nullptr, R_UNLESS(m_buffer_storages[level + 1] != nullptr,
ResultAllocationMemoryFailedAllocateShared); ResultAllocationMemoryFailedAllocateShared);
} }

View file

@ -123,7 +123,7 @@ private:
private: private:
std::shared_ptr<IntegrityVerificationStorage> m_verify_storages[MaxLayers - 1]; std::shared_ptr<IntegrityVerificationStorage> m_verify_storages[MaxLayers - 1];
std::shared_ptr<AlignmentMatchingStoragePooledBuffer<1>> m_buffer_storages[MaxLayers - 1]; VirtualFile m_buffer_storages[MaxLayers - 1];
s64 m_data_size; s64 m_data_size;
s32 m_max_layers; s32 m_max_layers;

View file

@ -73,31 +73,8 @@ size_t HierarchicalSha256Storage::Read(u8* buffer, size_t size, size_t offset) c
// Validate that we have a buffer to read into. // Validate that we have a buffer to read into.
ASSERT(buffer != nullptr); ASSERT(buffer != nullptr);
// Validate preconditions.
ASSERT(Common::IsAligned(offset, m_hash_target_block_size));
ASSERT(Common::IsAligned(size, m_hash_target_block_size));
// Read the data. // Read the data.
const size_t reduced_size = static_cast<size_t>( return m_base_storage->Read(buffer, size, offset);
std::min<s64>(m_base_storage_size,
Common::AlignUp(offset + size, m_hash_target_block_size)) -
offset);
m_base_storage->Read(buffer, reduced_size, offset);
// Setup tracking variables.
auto cur_offset = offset;
auto remaining_size = reduced_size;
while (remaining_size > 0) {
const auto cur_size =
static_cast<size_t>(std::min<s64>(m_hash_target_block_size, remaining_size));
ASSERT(static_cast<size_t>(cur_offset >> m_log_size_ratio) < m_hash_buffer_size);
// Advance.
cur_offset += cur_size;
remaining_size -= cur_size;
}
return size;
} }
} // namespace FileSys } // namespace FileSys

View file

@ -48,10 +48,6 @@ void IntegrityVerificationStorage::Finalize() {
} }
size_t IntegrityVerificationStorage::Read(u8* buffer, size_t size, size_t offset) const { size_t IntegrityVerificationStorage::Read(u8* buffer, size_t size, size_t offset) const {
// Validate preconditions.
ASSERT(Common::IsAligned(offset, static_cast<size_t>(m_verification_block_size)));
ASSERT(Common::IsAligned(size, static_cast<size_t>(m_verification_block_size)));
// Succeed if zero size. // Succeed if zero size.
if (size == 0) { if (size == 0) {
return size; return size;

View file

@ -508,12 +508,15 @@ Result NcaFileSystemDriver::CreateSparseStorageMetaStorage(VirtualFile* out,
offset + meta_offset, sparse_info.MakeAesCtrUpperIv(upper_iv), offset + meta_offset, sparse_info.MakeAesCtrUpperIv(upper_iv),
AlignmentStorageRequirement::None)); AlignmentStorageRequirement::None));
// Create meta storage. // Create buffered storage.
auto meta_storage = std::make_shared<OffsetVfsFile>(decrypted_storage, meta_size, 0); std::vector<u8> meta_data(meta_size);
R_UNLESS(meta_storage != nullptr, ResultAllocationMemoryFailedAllocateShared); decrypted_storage->Read(meta_data.data(), meta_size, 0);
auto buffered_storage = std::make_shared<VectorVfsFile>(std::move(meta_data));
R_UNLESS(buffered_storage != nullptr, ResultAllocationMemoryFailedAllocateShared);
// Set the output. // Set the output.
*out = std::move(meta_storage); *out = std::move(buffered_storage);
R_SUCCEED(); R_SUCCEED();
} }
@ -817,13 +820,15 @@ Result NcaFileSystemDriver::CreateAesCtrExStorageMetaStorage(
auto meta_storage = std::make_shared<OffsetVfsFile>(decrypted_storage, meta_size, 0); auto meta_storage = std::make_shared<OffsetVfsFile>(decrypted_storage, meta_size, 0);
R_UNLESS(meta_storage != nullptr, ResultAllocationMemoryFailedAllocateShared); R_UNLESS(meta_storage != nullptr, ResultAllocationMemoryFailedAllocateShared);
// Create an alignment-matching storage. // Create buffered storage.
using AlignedStorage = AlignmentMatchingStorage<NcaHeader::CtrBlockSize, 1>; std::vector<u8> meta_data(meta_size);
auto aligned_storage = std::make_shared<AlignedStorage>(std::move(meta_storage)); meta_storage->Read(meta_data.data(), meta_size, 0);
R_UNLESS(aligned_storage != nullptr, ResultAllocationMemoryFailedAllocateShared);
auto buffered_storage = std::make_shared<VectorVfsFile>(std::move(meta_data));
R_UNLESS(buffered_storage != nullptr, ResultAllocationMemoryFailedAllocateShared);
// Set the output. // Set the output.
*out = std::move(aligned_storage); *out = std::move(buffered_storage);
R_SUCCEED(); R_SUCCEED();
} }
@ -937,8 +942,15 @@ Result NcaFileSystemDriver::CreateIndirectStorageMetaStorage(VirtualFile* out,
patch_info.indirect_offset); patch_info.indirect_offset);
R_UNLESS(meta_storage != nullptr, ResultAllocationMemoryFailedAllocateShared); R_UNLESS(meta_storage != nullptr, ResultAllocationMemoryFailedAllocateShared);
// Create buffered storage.
std::vector<u8> meta_data(patch_info.indirect_size);
meta_storage->Read(meta_data.data(), patch_info.indirect_size, 0);
auto buffered_storage = std::make_shared<VectorVfsFile>(std::move(meta_data));
R_UNLESS(buffered_storage != nullptr, ResultAllocationMemoryFailedAllocateShared);
// Set the output. // Set the output.
*out = std::move(meta_storage); *out = std::move(buffered_storage);
R_SUCCEED(); R_SUCCEED();
} }
@ -1090,7 +1102,6 @@ Result NcaFileSystemDriver::CreateSha256Storage(
// Define storage types. // Define storage types.
using VerificationStorage = HierarchicalSha256Storage; using VerificationStorage = HierarchicalSha256Storage;
using AlignedStorage = AlignmentMatchingStoragePooledBuffer<1>;
// Validate the hash data. // Validate the hash data.
R_UNLESS(Common::IsPowerOfTwo(hash_data.hash_block_size), R_UNLESS(Common::IsPowerOfTwo(hash_data.hash_block_size),
@ -1141,13 +1152,8 @@ Result NcaFileSystemDriver::CreateSha256Storage(
hash_data.hash_block_size, hash_data.hash_block_size,
buffer_hold_storage->GetBuffer(), hash_buffer_size)); buffer_hold_storage->GetBuffer(), hash_buffer_size));
// Make the aligned storage.
auto aligned_storage = std::make_shared<AlignedStorage>(std::move(verification_storage),
hash_data.hash_block_size);
R_UNLESS(aligned_storage != nullptr, ResultAllocationMemoryFailedAllocateShared);
// Set the output. // Set the output.
*out = std::move(aligned_storage); *out = std::move(verification_storage);
R_SUCCEED(); R_SUCCEED();
} }