forked from ShuriZma/suyu
fssystem: reduce overalignment of unbuffered storage operations
This commit is contained in:
parent
86f6b6b7b2
commit
0398b34370
|
@ -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);
|
||||||
}
|
}
|
||||||
|
|
|
@ -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;
|
||||||
|
|
||||||
|
|
|
@ -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
|
||||||
|
|
|
@ -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;
|
||||||
|
|
|
@ -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();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue