Atmosphere/libraries/libstratosphere/source/fssystem/save/fssystem_buffered_storage.cpp

1085 lines
43 KiB
C++
Raw Normal View History

/*
* Copyright (c) 2018-2020 Atmosphère-NX
*
* This program is free software; you can redistribute it and/or modify it
* under the terms and conditions of the GNU General Public License,
* version 2, as published by the Free Software Foundation.
*
* This program is distributed in the hope it will be useful, but WITHOUT
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
* FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
* more details.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
#include <stratosphere.hpp>
namespace ams::fssystem::save {
namespace {
constexpr inline uintptr_t InvalidAddress = 0;
constexpr inline s64 InvalidOffset = std::numeric_limits<s64>::max();
}
class BufferedStorage::Cache : public ::ams::fs::impl::Newable {
private:
struct FetchParameter {
s64 offset;
void *buffer;
size_t size;
};
2020-05-14 09:22:24 +00:00
static_assert(util::is_pod<FetchParameter>::value);
private:
BufferedStorage *buffered_storage;
std::pair<uintptr_t, size_t> memory_range;
IBufferManager::CacheHandle cache_handle;
s64 offset;
std::atomic<bool> is_valid;
std::atomic<bool> is_dirty;
u8 reserved[2];
s32 reference_count;
Cache *next;
Cache *prev;
public:
Cache() : buffered_storage(nullptr), memory_range(InvalidAddress, 0), cache_handle(), offset(InvalidOffset), is_valid(false), is_dirty(false), reference_count(1), next(nullptr), prev(nullptr) {
/* ... */
}
~Cache() {
this->Finalize();
}
void Initialize(BufferedStorage *bs) {
AMS_ASSERT(bs != nullptr);
AMS_ASSERT(this->buffered_storage == nullptr);
this->buffered_storage = bs;
this->Link();
}
void Finalize() {
AMS_ASSERT(this->buffered_storage != nullptr);
AMS_ASSERT(this->buffered_storage->buffer_manager != nullptr);
AMS_ASSERT(this->reference_count == 0);
/* If we're valid, acquire our cache handle and free our buffer. */
if (this->IsValid()) {
const auto buffer_manager = this->buffered_storage->buffer_manager;
if (!this->is_dirty) {
AMS_ASSERT(this->memory_range.first == InvalidAddress);
this->memory_range = buffer_manager->AcquireCache(this->cache_handle);
}
if (this->memory_range.first != InvalidAddress) {
buffer_manager->DeallocateBuffer(this->memory_range.first, this->memory_range.second);
this->memory_range.first = InvalidAddress;
this->memory_range.second = 0;
}
}
/* Clear all our members. */
this->buffered_storage = nullptr;
this->offset = InvalidOffset;
this->is_valid = false;
this->is_dirty = false;
this->next = nullptr;
this->prev = nullptr;
}
void Link() {
AMS_ASSERT(this->buffered_storage != nullptr);
AMS_ASSERT(this->buffered_storage->buffer_manager != nullptr);
AMS_ASSERT(this->reference_count > 0);
if ((--this->reference_count) == 0) {
AMS_ASSERT(this->next == nullptr);
AMS_ASSERT(this->prev == nullptr);
if (this->buffered_storage->next_fetch_cache == nullptr) {
this->buffered_storage->next_fetch_cache = this;
this->next = this;
this->prev = this;
} else {
/* Check against a cache being registered twice. */
{
auto cache = this->buffered_storage->next_fetch_cache;
do {
if (cache->IsValid() && this->Hits(cache->offset, this->buffered_storage->block_size)) {
this->is_valid = false;
break;
}
cache = cache->next;
} while (cache != this->buffered_storage->next_fetch_cache);
}
/* Link into the fetch list. */
{
AMS_ASSERT(this->buffered_storage->next_fetch_cache->prev != nullptr);
AMS_ASSERT(this->buffered_storage->next_fetch_cache->prev->next == this->buffered_storage->next_fetch_cache);
this->next = this->buffered_storage->next_fetch_cache;
this->prev = this->buffered_storage->next_fetch_cache->prev;
this->next->prev = this;
this->prev->next = this;
}
/* Insert invalid caches at the start of the list. */
if (!this->IsValid()) {
this->buffered_storage->next_fetch_cache = this;
}
}
/* If we're not valid, clear our offset. */
if (!this->IsValid()) {
this->offset = InvalidOffset;
this->is_dirty = false;
}
/* Ensure our buffer state is coherent. */
if (this->memory_range.first != InvalidAddress && !this->is_dirty) {
if (this->IsValid()) {
this->cache_handle = this->buffered_storage->buffer_manager->RegisterCache(this->memory_range.first, this->memory_range.second, IBufferManager::BufferAttribute());
} else {
this->buffered_storage->buffer_manager->DeallocateBuffer(this->memory_range.first, this->memory_range.second);
}
this->memory_range.first = InvalidAddress;
this->memory_range.second = 0;
}
}
}
void Unlink() {
AMS_ASSERT(this->buffered_storage != nullptr);
AMS_ASSERT(this->reference_count >= 0);
if ((++this->reference_count) == 1) {
AMS_ASSERT(this->next != nullptr);
AMS_ASSERT(this->prev != nullptr);
AMS_ASSERT(this->next->prev == this);
AMS_ASSERT(this->prev->next == this);
if (this->buffered_storage->next_fetch_cache == this) {
if (this->next != this) {
this->buffered_storage->next_fetch_cache = this->next;
} else {
this->buffered_storage->next_fetch_cache = nullptr;
}
}
this->buffered_storage->next_acquire_cache = this;
this->next->prev = this->prev;
this->prev->next = this->next;
this->next = nullptr;
this->prev = nullptr;
} else {
AMS_ASSERT(this->next == nullptr);
AMS_ASSERT(this->prev == nullptr);
}
}
void Read(s64 offset, void *buffer, size_t size) const {
AMS_ASSERT(this->buffered_storage != nullptr);
AMS_ASSERT(this->next == nullptr);
AMS_ASSERT(this->prev == nullptr);
AMS_ASSERT(this->IsValid());
AMS_ASSERT(this->Hits(offset, 1));
AMS_ASSERT(this->memory_range.first != InvalidAddress);
const auto read_offset = offset - this->offset;
const auto readable_offset_max = this->buffered_storage->block_size - size;
const auto cache_buffer = reinterpret_cast<u8 *>(this->memory_range.first) + read_offset;
AMS_ASSERT(read_offset >= 0);
AMS_ASSERT(static_cast<size_t>(read_offset) <= readable_offset_max);
AMS_UNUSED(readable_offset_max);
std::memcpy(buffer, cache_buffer, size);
}
void Write(s64 offset, const void *buffer, size_t size) {
AMS_ASSERT(this->buffered_storage != nullptr);
AMS_ASSERT(this->next == nullptr);
AMS_ASSERT(this->prev == nullptr);
AMS_ASSERT(this->IsValid());
AMS_ASSERT(this->Hits(offset, 1));
AMS_ASSERT(this->memory_range.first != InvalidAddress);
const auto write_offset = offset - this->offset;
const auto writable_offset_max = this->buffered_storage->block_size - size;
const auto cache_buffer = reinterpret_cast<u8 *>(this->memory_range.first) + write_offset;
AMS_ASSERT(write_offset >= 0);
AMS_ASSERT(static_cast<size_t>(write_offset) <= writable_offset_max);
AMS_UNUSED(writable_offset_max);
std::memcpy(cache_buffer, buffer, size);
this->is_dirty = true;
}
Result Flush() {
AMS_ASSERT(this->buffered_storage != nullptr);
AMS_ASSERT(this->next == nullptr);
AMS_ASSERT(this->prev == nullptr);
AMS_ASSERT(this->IsValid());
if (this->is_dirty) {
AMS_ASSERT(this->memory_range.first != InvalidAddress);
const auto base_size = this->buffered_storage->base_storage_size;
const auto block_size = static_cast<s64>(this->buffered_storage->block_size);
const auto flush_size = static_cast<size_t>(std::min(block_size, base_size - this->offset));
auto &base_storage = this->buffered_storage->base_storage;
const auto cache_buffer = reinterpret_cast<void *>(this->memory_range.first);
R_TRY(base_storage.Write(this->offset, cache_buffer, flush_size));
this->is_dirty = false;
buffers::EnableBlockingBufferManagerAllocation();
}
return ResultSuccess();
}
const std::pair<Result, bool> PrepareFetch() {
AMS_ASSERT(this->buffered_storage != nullptr);
AMS_ASSERT(this->buffered_storage->buffer_manager != nullptr);
AMS_ASSERT(this->next == nullptr);
AMS_ASSERT(this->prev == nullptr);
AMS_ASSERT(this->IsValid());
AMS_ASSERT(this->buffered_storage->mutex.IsLockedByCurrentThread());
std::pair<Result, bool> result(ResultSuccess(), false);
if (this->reference_count == 1) {
result.first = this->Flush();
if (R_SUCCEEDED(result.first)) {
this->is_valid = false;
this->reference_count = 0;
result.second = true;
}
}
return result;
}
void UnprepareFetch() {
AMS_ASSERT(this->buffered_storage != nullptr);
AMS_ASSERT(this->buffered_storage->buffer_manager != nullptr);
AMS_ASSERT(this->next == nullptr);
AMS_ASSERT(this->prev == nullptr);
AMS_ASSERT(!this->IsValid());
AMS_ASSERT(!this->is_dirty);
AMS_ASSERT(this->buffered_storage->mutex.IsLockedByCurrentThread());
this->is_valid = true;
this->reference_count = 1;
}
Result Fetch(s64 offset) {
AMS_ASSERT(this->buffered_storage != nullptr);
AMS_ASSERT(this->buffered_storage->buffer_manager != nullptr);
AMS_ASSERT(this->next == nullptr);
AMS_ASSERT(this->prev == nullptr);
AMS_ASSERT(!this->IsValid());
AMS_ASSERT(!this->is_dirty);
if (this->memory_range.first == InvalidAddress) {
R_TRY(this->AllocateFetchBuffer());
}
FetchParameter fetch_param = {};
this->CalcFetchParameter(std::addressof(fetch_param), offset);
auto &base_storage = this->buffered_storage->base_storage;
R_TRY(base_storage.Read(fetch_param.offset, fetch_param.buffer, fetch_param.size));
this->offset = fetch_param.offset;
AMS_ASSERT(this->Hits(offset, 1));
return ResultSuccess();
}
Result FetchFromBuffer(s64 offset, const void *buffer, size_t buffer_size) {
AMS_ASSERT(this->buffered_storage != nullptr);
AMS_ASSERT(this->buffered_storage->buffer_manager != nullptr);
AMS_ASSERT(this->next == nullptr);
AMS_ASSERT(this->prev == nullptr);
AMS_ASSERT(!this->IsValid());
AMS_ASSERT(!this->is_dirty);
AMS_ASSERT(util::IsAligned(offset, this->buffered_storage->block_size));
if (this->memory_range.first == InvalidAddress) {
R_TRY(this->AllocateFetchBuffer());
}
FetchParameter fetch_param = {};
this->CalcFetchParameter(std::addressof(fetch_param), offset);
AMS_ASSERT(fetch_param.offset == offset);
AMS_ASSERT(fetch_param.size <= buffer_size);
std::memcpy(fetch_param.buffer, buffer, fetch_param.size);
this->offset = fetch_param.offset;
AMS_ASSERT(this->Hits(offset, 1));
return ResultSuccess();
}
bool TryAcquireCache() {
AMS_ASSERT(this->buffered_storage != nullptr);
AMS_ASSERT(this->buffered_storage->buffer_manager != nullptr);
AMS_ASSERT(this->IsValid());
if (this->memory_range.first != InvalidAddress) {
return true;
} else {
this->memory_range = this->buffered_storage->buffer_manager->AcquireCache(this->cache_handle);
this->is_valid = this->memory_range.first != InvalidAddress;
return this->is_valid;
}
}
void Invalidate() {
AMS_ASSERT(this->buffered_storage != nullptr);
this->is_valid = false;
}
bool IsValid() const {
AMS_ASSERT(this->buffered_storage != nullptr);
return this->is_valid || this->reference_count > 0;
}
bool IsDirty() const {
AMS_ASSERT(this->buffered_storage != nullptr);
return this->is_dirty;
}
bool Hits(s64 offset, s64 size) const {
AMS_ASSERT(this->buffered_storage != nullptr);
const auto block_size = static_cast<s64>(this->buffered_storage->block_size);
return (offset < this->offset + block_size) && (this->offset < offset + size);
}
private:
Result AllocateFetchBuffer() {
IBufferManager *buffer_manager = this->buffered_storage->buffer_manager;
AMS_ASSERT(buffer_manager->AcquireCache(this->cache_handle).first == InvalidAddress);
auto range_guard = SCOPE_GUARD { this->memory_range.first = InvalidAddress; };
R_TRY(buffers::AllocateBufferUsingBufferManagerContext(std::addressof(this->memory_range), buffer_manager, this->buffered_storage->block_size, IBufferManager::BufferAttribute(), [](const std::pair<uintptr_t, size_t> &buffer) {
return buffer.first != 0;
}, AMS_CURRENT_FUNCTION_NAME));
range_guard.Cancel();
return ResultSuccess();
}
void CalcFetchParameter(FetchParameter *out, s64 offset) const {
AMS_ASSERT(out != nullptr);
const auto block_size = static_cast<s64>(this->buffered_storage->block_size);
const auto cache_offset = util::AlignDown(offset, this->buffered_storage->block_size);
const auto base_size = this->buffered_storage->base_storage_size;
const auto cache_size = static_cast<size_t>(std::min(block_size, base_size - cache_offset));
const auto cache_buffer = reinterpret_cast<void *>(this->memory_range.first);
AMS_ASSERT(offset >= 0);
AMS_ASSERT(offset < base_size);
out->offset = cache_offset;
out->buffer = cache_buffer;
out->size = cache_size;
}
};
class BufferedStorage::SharedCache {
NON_COPYABLE(SharedCache);
NON_MOVEABLE(SharedCache);
friend class UniqueCache;
private:
Cache *cache;
Cache *start_cache;
BufferedStorage *buffered_storage;
public:
explicit SharedCache(BufferedStorage *bs) : cache(nullptr), start_cache(bs->next_acquire_cache), buffered_storage(bs) {
AMS_ASSERT(this->buffered_storage != nullptr);
}
~SharedCache() {
std::scoped_lock lk(buffered_storage->mutex);
this->Release();
}
bool AcquireNextOverlappedCache(s64 offset, s64 size) {
AMS_ASSERT(this->buffered_storage != nullptr);
auto is_first = this->cache == nullptr;
const auto start = is_first ? this->start_cache : this->cache + 1;
AMS_ASSERT(start >= this->buffered_storage->caches.get());
AMS_ASSERT(start <= this->buffered_storage->caches.get() + this->buffered_storage->cache_count);
std::scoped_lock lk(this->buffered_storage->mutex);
this->Release();
AMS_ASSERT(this->cache == nullptr);
for (auto cache = start; true; ++cache) {
if (this->buffered_storage->caches.get() + this->buffered_storage->cache_count <= cache) {
cache = this->buffered_storage->caches.get();
}
if (!is_first && cache == this->start_cache) {
break;
}
if (cache->IsValid() && cache->Hits(offset, size) && cache->TryAcquireCache()) {
cache->Unlink();
this->cache = cache;
return true;
}
is_first = false;
}
this->cache = nullptr;
return false;
}
bool AcquireNextDirtyCache() {
AMS_ASSERT(this->buffered_storage != nullptr);
const auto start = this->cache != nullptr ? this->cache + 1 : this->buffered_storage->caches.get();
const auto end = this->buffered_storage->caches.get() + this->buffered_storage->cache_count;
AMS_ASSERT(start >= this->buffered_storage->caches.get());
AMS_ASSERT(start <= end);
this->Release();
AMS_ASSERT(this->cache == nullptr);
for (auto cache = start; cache < end; ++cache) {
if (cache->IsValid() && cache->IsDirty() && cache->TryAcquireCache()) {
cache->Unlink();
this->cache = cache;
return true;
}
}
this->cache = nullptr;
return false;
}
bool AcquireNextValidCache() {
AMS_ASSERT(this->buffered_storage != nullptr);
const auto start = this->cache != nullptr ? this->cache + 1 : this->buffered_storage->caches.get();
const auto end = this->buffered_storage->caches.get() + this->buffered_storage->cache_count;
AMS_ASSERT(start >= this->buffered_storage->caches.get());
AMS_ASSERT(start <= end);
this->Release();
AMS_ASSERT(this->cache == nullptr);
for (auto cache = start; cache < end; ++cache) {
if (cache->IsValid() && cache->TryAcquireCache()) {
cache->Unlink();
this->cache = cache;
return true;
}
}
this->cache = nullptr;
return false;
}
bool AcquireFetchableCache() {
AMS_ASSERT(this->buffered_storage != nullptr);
std::scoped_lock lk(this->buffered_storage->mutex);
this->Release();
AMS_ASSERT(this->cache == nullptr);
this->cache = this->buffered_storage->next_fetch_cache;
if (this->cache != nullptr) {
if (this->cache->IsValid()) {
this->cache->TryAcquireCache();
}
this->cache->Unlink();
}
return this->cache != nullptr;
}
void Read(s64 offset, void *buffer, size_t size) {
AMS_ASSERT(this->cache != nullptr);
this->cache->Read(offset, buffer, size);
}
void Write(s64 offset, const void *buffer, size_t size) {
AMS_ASSERT(this->cache != nullptr);
this->cache->Write(offset, buffer, size);
}
Result Flush() {
AMS_ASSERT(this->cache != nullptr);
return this->cache->Flush();
}
void Invalidate() {
AMS_ASSERT(this->cache != nullptr);
return this->cache->Invalidate();
}
bool Hits(s64 offset, s64 size) const {
AMS_ASSERT(this->cache != nullptr);
return this->cache->Hits(offset, size);
}
private:
void Release() {
if (this->cache != nullptr) {
AMS_ASSERT(this->buffered_storage->caches.get() <= this->cache);
AMS_ASSERT(this->cache <= this->buffered_storage->caches.get() + this->buffered_storage->cache_count);
this->cache->Link();
this->cache = nullptr;
}
}
};
class BufferedStorage::UniqueCache {
NON_COPYABLE(UniqueCache);
NON_MOVEABLE(UniqueCache);
private:
Cache *cache;
BufferedStorage *buffered_storage;
public:
explicit UniqueCache(BufferedStorage *bs) : cache(nullptr), buffered_storage(bs) {
AMS_ASSERT(this->buffered_storage != nullptr);
}
~UniqueCache() {
if (this->cache != nullptr) {
std::scoped_lock lk(this->buffered_storage->mutex);
this->cache->UnprepareFetch();
}
}
const std::pair<Result, bool> Upgrade(const SharedCache &shared_cache) {
AMS_ASSERT(this->buffered_storage == shared_cache.buffered_storage);
AMS_ASSERT(shared_cache.cache != nullptr);
std::scoped_lock lk(this->buffered_storage->mutex);
const auto result = shared_cache.cache->PrepareFetch();
if (R_SUCCEEDED(result.first) && result.second) {
this->cache = shared_cache.cache;
}
return result;
}
Result Fetch(s64 offset) {
AMS_ASSERT(this->cache != nullptr);
return this->cache->Fetch(offset);
}
Result FetchFromBuffer(s64 offset, const void *buffer, size_t buffer_size) {
AMS_ASSERT(this->cache != nullptr);
R_TRY(this->cache->FetchFromBuffer(offset, buffer, buffer_size));
return ResultSuccess();
}
};
BufferedStorage::BufferedStorage() : base_storage(), buffer_manager(), block_size(), base_storage_size(), caches(), cache_count(), next_acquire_cache(), next_fetch_cache(), mutex(), bulk_read_enabled() {
/* ... */
}
BufferedStorage::~BufferedStorage() {
this->Finalize();
}
Result BufferedStorage::Initialize(fs::SubStorage base_storage, IBufferManager *buffer_manager, size_t block_size, s32 buffer_count) {
AMS_ASSERT(buffer_manager != nullptr);
AMS_ASSERT(block_size > 0);
AMS_ASSERT(util::IsPowerOfTwo(block_size));
AMS_ASSERT(buffer_count > 0);
/* Get the base storage size. */
R_TRY(base_storage.GetSize(std::addressof(this->base_storage_size)));
/* Set members. */
this->base_storage = base_storage;
this->buffer_manager = buffer_manager;
this->block_size = block_size;
2020-12-02 14:48:42 +00:00
this->cache_count = buffer_count;
/* Allocate the caches. */
this->caches.reset(new Cache[buffer_count]);
R_UNLESS(this->caches != nullptr, fs::ResultAllocationFailureInBufferedStorageA());
/* Initialize the caches. */
for (auto i = 0; i < buffer_count; i++) {
this->caches[i].Initialize(this);
}
this->next_acquire_cache = std::addressof(this->caches[0]);
return ResultSuccess();
}
void BufferedStorage::Finalize() {
this->base_storage = fs::SubStorage();
this->base_storage_size = 0;
this->caches.reset();
this->cache_count = 0;
this->next_fetch_cache = nullptr;
}
Result BufferedStorage::Read(s64 offset, void *buffer, size_t size) {
AMS_ASSERT(this->IsInitialized());
/* Succeed if zero size. */
R_SUCCEED_IF(size == 0);
/* Validate arguments. */
R_UNLESS(buffer != nullptr, fs::ResultNullptrArgument());
/* Do the read. */
R_TRY(this->ReadCore(offset, buffer, size));
return ResultSuccess();
}
Result BufferedStorage::Write(s64 offset, const void *buffer, size_t size) {
AMS_ASSERT(this->IsInitialized());
/* Succeed if zero size. */
R_SUCCEED_IF(size == 0);
/* Validate arguments. */
R_UNLESS(buffer != nullptr, fs::ResultNullptrArgument());
/* Do the write. */
R_TRY(this->WriteCore(offset, buffer, size));
return ResultSuccess();
}
Result BufferedStorage::GetSize(s64 *out) {
AMS_ASSERT(out != nullptr);
AMS_ASSERT(this->IsInitialized());
*out = this->base_storage_size;
return ResultSuccess();
}
Result BufferedStorage::SetSize(s64 size) {
AMS_ASSERT(this->IsInitialized());
const s64 prev_size = this->base_storage_size;
if (prev_size < size) {
/* Prepare to expand. */
if (!util::IsAligned(prev_size, this->block_size)) {
SharedCache cache(this);
const auto invalidate_offset = prev_size;
const auto invalidate_size = size - prev_size;
if (cache.AcquireNextOverlappedCache(invalidate_offset, invalidate_size)) {
R_TRY(cache.Flush());
cache.Invalidate();
}
2021-09-30 05:02:58 +00:00
AMS_ASSERT(!cache.AcquireNextOverlappedCache(invalidate_offset, invalidate_size));
}
} else if (size < prev_size) {
/* Prepare to do a shrink. */
SharedCache cache(this);
const auto invalidate_offset = prev_size;
const auto invalidate_size = size - prev_size;
const auto is_fragment = util::IsAligned(size, this->block_size);
while (cache.AcquireNextOverlappedCache(invalidate_offset, invalidate_size)) {
if (is_fragment && cache.Hits(invalidate_offset, 1)) {
R_TRY(cache.Flush());
}
cache.Invalidate();
}
}
/* Set the size. */
R_TRY(this->base_storage.SetSize(size));
/* Get our new size. */
s64 new_size = 0;
R_TRY(this->base_storage.GetSize(std::addressof(new_size)));
this->base_storage_size = new_size;
return ResultSuccess();
}
Result BufferedStorage::Flush() {
AMS_ASSERT(this->IsInitialized());
/* Flush caches. */
SharedCache cache(this);
while (cache.AcquireNextDirtyCache()) {
R_TRY(cache.Flush());
}
/* Flush the base storage. */
R_TRY(this->base_storage.Flush());
return ResultSuccess();
}
Result BufferedStorage::OperateRange(void *dst, size_t dst_size, fs::OperationId op_id, s64 offset, s64 size, const void *src, size_t src_size) {
AMS_ASSERT(this->IsInitialized());
/* Invalidate caches, if we should. */
if (op_id == fs::OperationId::Invalidate) {
SharedCache cache(this);
while (cache.AcquireNextOverlappedCache(offset, size)) {
cache.Invalidate();
}
}
return this->base_storage.OperateRange(dst, dst_size, op_id, offset, size, src, src_size);
}
void BufferedStorage::InvalidateCaches() {
AMS_ASSERT(this->IsInitialized());
SharedCache cache(this);
while (cache.AcquireNextValidCache()) {
cache.Invalidate();
}
}
Result BufferedStorage::PrepareAllocation() {
const auto flush_threshold = this->buffer_manager->GetTotalSize() / 8;
if (this->buffer_manager->GetTotalAllocatableSize() < flush_threshold) {
R_TRY(this->Flush());
}
return ResultSuccess();
}
Result BufferedStorage::ControlDirtiness() {
const auto flush_threshold = this->buffer_manager->GetTotalSize() / 4;
if (this->buffer_manager->GetTotalAllocatableSize() < flush_threshold) {
s32 dirty_count = 0;
SharedCache cache(this);
while (cache.AcquireNextDirtyCache()) {
if ((++dirty_count) > 1) {
R_TRY(cache.Flush());
cache.Invalidate();
}
}
}
return ResultSuccess();
}
Result BufferedStorage::ReadCore(s64 offset, void *buffer, size_t size) {
AMS_ASSERT(this->caches != nullptr);
AMS_ASSERT(buffer != nullptr);
/* Validate the offset. */
const auto base_storage_size = this->base_storage_size;
R_UNLESS(offset >= 0, fs::ResultInvalidOffset());
R_UNLESS(offset <= base_storage_size, fs::ResultInvalidOffset());
/* Setup tracking variables. */
size_t remaining_size = static_cast<size_t>(std::min<s64>(size, base_storage_size - offset));
s64 cur_offset = offset;
s64 buf_offset = 0;
/* Determine what caches are needed, if we have bulk read set. */
if (this->bulk_read_enabled) {
/* Check head cache. */
const auto head_cache_needed = this->ReadHeadCache(std::addressof(cur_offset), buffer, std::addressof(remaining_size), std::addressof(buf_offset));
R_SUCCEED_IF(remaining_size == 0);
/* Check tail cache. */
const auto tail_cache_needed = this->ReadTailCache(cur_offset, buffer, std::addressof(remaining_size), buf_offset);
R_SUCCEED_IF(remaining_size == 0);
/* Perform bulk reads. */
constexpr size_t BulkReadSizeMax = 2_MB;
if (remaining_size <= BulkReadSizeMax) {
do {
/* Try to do a bulk read. */
R_TRY_CATCH(this->BulkRead(cur_offset, static_cast<u8 *>(buffer) + buf_offset, remaining_size, head_cache_needed, tail_cache_needed)) {
R_CATCH(fs::ResultAllocationFailurePooledBufferNotEnoughSize) {
/* If the read fails due to insufficient pooled buffer size, */
/* then we want to fall back to the normal read path. */
break;
}
} R_END_TRY_CATCH;
return ResultSuccess();
} while(0);
}
}
/* Repeatedly read until we're done. */
while (remaining_size > 0) {
/* Determine how much to read this iteration. */
auto *cur_dst = static_cast<u8 *>(buffer) + buf_offset;
size_t cur_size = 0;
if (!util::IsAligned(cur_offset, this->block_size)) {
const size_t aligned_size = this->block_size - (cur_offset & (this->block_size - 1));
cur_size = std::min(aligned_size, remaining_size);
} else if (remaining_size < this->block_size) {
cur_size = remaining_size;
} else {
cur_size = util::AlignDown(remaining_size, this->block_size);
}
if (cur_size <= this->block_size) {
SharedCache cache(this);
if (!cache.AcquireNextOverlappedCache(cur_offset, cur_size)) {
R_TRY(this->PrepareAllocation());
while (true) {
R_UNLESS(cache.AcquireFetchableCache(), fs::ResultOutOfResource());
UniqueCache fetch_cache(this);
const auto upgrade_result = fetch_cache.Upgrade(cache);
R_TRY(upgrade_result.first);
if (upgrade_result.second) {
R_TRY(fetch_cache.Fetch(cur_offset));
break;
}
}
R_TRY(this->ControlDirtiness());
}
cache.Read(cur_offset, cur_dst, cur_size);
} else {
{
SharedCache cache(this);
while (cache.AcquireNextOverlappedCache(cur_offset, cur_size)) {
R_TRY(cache.Flush());
cache.Invalidate();
}
}
R_TRY(this->base_storage.Read(cur_offset, cur_dst, cur_size));
}
remaining_size -= cur_size;
cur_offset += cur_size;
buf_offset += cur_size;
}
return ResultSuccess();
}
bool BufferedStorage::ReadHeadCache(s64 *offset, void *buffer, size_t *size, s64 *buffer_offset) {
AMS_ASSERT(offset != nullptr);
AMS_ASSERT(buffer != nullptr);
AMS_ASSERT(size != nullptr);
AMS_ASSERT(buffer_offset != nullptr);
bool is_cache_needed = !util::IsAligned(*offset, this->block_size);
while (*size > 0) {
size_t cur_size = 0;
if (!util::IsAligned(*offset, this->block_size)) {
const s64 aligned_size = util::AlignUp(*offset, this->block_size) - *offset;
cur_size = std::min(aligned_size, static_cast<s64>(*size));
} else if (*size < this->block_size) {
cur_size = *size;
} else {
cur_size = this->block_size;
}
SharedCache cache(this);
if (!cache.AcquireNextOverlappedCache(*offset, cur_size)) {
break;
}
cache.Read(*offset, static_cast<u8 *>(buffer) + *buffer_offset, cur_size);
*offset += cur_size;
*buffer_offset += cur_size;
*size -= cur_size;
is_cache_needed = false;
}
return is_cache_needed;
}
bool BufferedStorage::ReadTailCache(s64 offset, void *buffer, size_t *size, s64 buffer_offset) {
AMS_ASSERT(buffer != nullptr);
AMS_ASSERT(size != nullptr);
bool is_cache_needed = !util::IsAligned(offset + *size, this->block_size);
while (*size > 0) {
const s64 cur_offset_end = offset + *size;
size_t cur_size = 0;
2021-01-28 22:28:45 +00:00
if (!util::IsAligned(cur_offset_end, this->block_size)) {
const s64 aligned_size = cur_offset_end - util::AlignDown(cur_offset_end, this->block_size);
cur_size = std::min(aligned_size, static_cast<s64>(*size));
} else if (*size < this->block_size) {
cur_size = *size;
} else {
cur_size = this->block_size;
}
const s64 cur_offset = cur_offset_end - static_cast<s64>(cur_size);
AMS_ASSERT(cur_offset >= 0);
SharedCache cache(this);
if (!cache.AcquireNextOverlappedCache(cur_offset, cur_size)) {
break;
}
cache.Read(cur_offset, static_cast<u8 *>(buffer) + buffer_offset + cur_offset - offset, cur_size);
*size -= cur_size;
is_cache_needed = false;
}
return is_cache_needed;
}
Result BufferedStorage::BulkRead(s64 offset, void *buffer, size_t size, bool head_cache_needed, bool tail_cache_needed) {
/* Determine aligned extents. */
const s64 aligned_offset = util::AlignDown(offset, this->block_size);
const s64 aligned_offset_end = std::min(util::AlignUp(offset + static_cast<s64>(size), this->block_size), this->base_storage_size);
const s64 aligned_size = aligned_offset_end - aligned_offset;
/* Allocate a work buffer. */
char *work_buffer = nullptr;
PooledBuffer pooled_buffer;
if (offset == aligned_offset && size == static_cast<size_t>(aligned_size)) {
work_buffer = static_cast<char *>(buffer);
} else {
pooled_buffer.AllocateParticularlyLarge(static_cast<size_t>(aligned_size), 1);
R_UNLESS(static_cast<s64>(pooled_buffer.GetSize()) >= aligned_size, fs::ResultAllocationFailurePooledBufferNotEnoughSize());
work_buffer = pooled_buffer.GetBuffer();
}
/* Ensure cache is coherent. */
{
SharedCache cache(this);
while (cache.AcquireNextOverlappedCache(aligned_offset, aligned_size)) {
R_TRY(cache.Flush());
cache.Invalidate();
}
}
/* Read from the base storage. */
R_TRY(this->base_storage.Read(aligned_offset, work_buffer, static_cast<size_t>(aligned_size)));
if (work_buffer != static_cast<char *>(buffer)) {
std::memcpy(buffer, work_buffer + offset - aligned_offset, size);
}
bool cached = false;
/* Handle head cache if needed. */
if (head_cache_needed) {
R_TRY(this->PrepareAllocation());
SharedCache cache(this);
while (true) {
R_UNLESS(cache.AcquireFetchableCache(), fs::ResultOutOfResource());
UniqueCache fetch_cache(this);
const auto upgrade_result = fetch_cache.Upgrade(cache);
R_TRY(upgrade_result.first);
if (upgrade_result.second) {
R_TRY(fetch_cache.FetchFromBuffer(aligned_offset, work_buffer, static_cast<size_t>(aligned_size)));
break;
}
}
cached = true;
}
/* Handle tail cache if needed. */
if (tail_cache_needed && (!head_cache_needed || aligned_size > static_cast<s64>(this->block_size))) {
if (!cached) {
R_TRY(this->PrepareAllocation());
}
SharedCache cache(this);
while (true) {
R_UNLESS(cache.AcquireFetchableCache(), fs::ResultOutOfResource());
UniqueCache fetch_cache(this);
const auto upgrade_result = fetch_cache.Upgrade(cache);
R_TRY(upgrade_result.first);
if (upgrade_result.second) {
const s64 tail_cache_offset = util::AlignDown(offset + static_cast<s64>(size), this->block_size);
const size_t tail_cache_size = static_cast<size_t>(aligned_size - tail_cache_offset + aligned_offset);
R_TRY(fetch_cache.FetchFromBuffer(tail_cache_offset, work_buffer + tail_cache_offset - aligned_offset, tail_cache_size));
break;
}
}
}
if (cached) {
R_TRY(this->ControlDirtiness());
}
return ResultSuccess();
}
Result BufferedStorage::WriteCore(s64 offset, const void *buffer, size_t size) {
AMS_ASSERT(this->caches != nullptr);
AMS_ASSERT(buffer != nullptr);
/* Validate the offset. */
const auto base_storage_size = this->base_storage_size;
R_UNLESS(offset >= 0, fs::ResultInvalidOffset());
R_UNLESS(offset <= base_storage_size, fs::ResultInvalidOffset());
/* Setup tracking variables. */
size_t remaining_size = static_cast<size_t>(std::min<s64>(size, base_storage_size - offset));
s64 cur_offset = offset;
s64 buf_offset = 0;
/* Repeatedly read until we're done. */
while (remaining_size > 0) {
/* Determine how much to read this iteration. */
const auto *cur_src = static_cast<const u8 *>(buffer) + buf_offset;
size_t cur_size = 0;
if (!util::IsAligned(cur_offset, this->block_size)) {
const size_t aligned_size = this->block_size - (cur_offset & (this->block_size - 1));
cur_size = std::min(aligned_size, remaining_size);
} else if (remaining_size < this->block_size) {
cur_size = remaining_size;
} else {
cur_size = util::AlignDown(remaining_size, this->block_size);
}
if (cur_size <= this->block_size) {
SharedCache cache(this);
if (!cache.AcquireNextOverlappedCache(cur_offset, cur_size)) {
R_TRY(this->PrepareAllocation());
while (true) {
R_UNLESS(cache.AcquireFetchableCache(), fs::ResultOutOfResource());
UniqueCache fetch_cache(this);
const auto upgrade_result = fetch_cache.Upgrade(cache);
R_TRY(upgrade_result.first);
if (upgrade_result.second) {
R_TRY(fetch_cache.Fetch(cur_offset));
break;
}
}
}
cache.Write(cur_offset, cur_src, cur_size);
buffers::EnableBlockingBufferManagerAllocation();
R_TRY(this->ControlDirtiness());
} else {
{
SharedCache cache(this);
while (cache.AcquireNextOverlappedCache(cur_offset, cur_size)) {
R_TRY(cache.Flush());
cache.Invalidate();
}
}
R_TRY(this->base_storage.Write(cur_offset, cur_src, cur_size));
buffers::EnableBlockingBufferManagerAllocation();
}
remaining_size -= cur_size;
cur_offset += cur_size;
buf_offset += cur_size;
}
return ResultSuccess();
}
}