mirror of
https://github.com/Atmosphere-NX/Atmosphere
synced 2024-11-10 15:16:34 +00:00
157 lines
7.6 KiB
C++
157 lines
7.6 KiB
C++
/*
|
|
* Copyright (c) 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/>.
|
|
*/
|
|
#pragma once
|
|
#include <stratosphere/os.hpp>
|
|
#include <stratosphere/fs/fs_mount.hpp>
|
|
#include <stratosphere/fs/fs_bis.hpp>
|
|
#include <stratosphere/fs/fs_content_storage.hpp>
|
|
#include <stratosphere/fs/fs_system_save_data.hpp>
|
|
#include <stratosphere/ncm/ncm_i_content_manager.hpp>
|
|
#include <stratosphere/ncm/ncm_content_manager_config.hpp>
|
|
#include <stratosphere/ncm/ncm_content_meta_database.hpp>
|
|
#include <stratosphere/ncm/ncm_bounded_map.hpp>
|
|
#include <stratosphere/ncm/ncm_rights_id_cache.hpp>
|
|
#include <stratosphere/ncm/ncm_content_management_utils.hpp>
|
|
#include <stratosphere/ncm/ncm_content_meta_utils.hpp>
|
|
#include <stratosphere/ncm/ncm_registered_host_content.hpp>
|
|
#include <stratosphere/kvdb/kvdb_memory_key_value_store.hpp>
|
|
|
|
namespace ams::ncm {
|
|
|
|
class ContentMetaMemoryResource : public MemoryResource {
|
|
private:
|
|
mem::StandardAllocator m_allocator;
|
|
size_t m_peak_total_alloc_size;
|
|
size_t m_peak_alloc_size;
|
|
public:
|
|
explicit ContentMetaMemoryResource(void *heap, size_t heap_size) : m_allocator(heap, heap_size), m_peak_total_alloc_size(0), m_peak_alloc_size(0) { /* ... */ }
|
|
|
|
mem::StandardAllocator *GetAllocator() { return std::addressof(m_allocator); }
|
|
size_t GetPeakTotalAllocationSize() const { return m_peak_total_alloc_size; }
|
|
size_t GetPeakAllocationSize() const { return m_peak_alloc_size; }
|
|
private:
|
|
virtual void *AllocateImpl(size_t size, size_t alignment) override {
|
|
void *mem = m_allocator.Allocate(size, alignment);
|
|
m_peak_total_alloc_size = std::max(m_allocator.Hash().allocated_size, m_peak_total_alloc_size);
|
|
m_peak_alloc_size = std::max(size, m_peak_alloc_size);
|
|
return mem;
|
|
}
|
|
|
|
virtual void DeallocateImpl(void *buffer, size_t size, size_t alignment) override {
|
|
AMS_UNUSED(size, alignment);
|
|
return m_allocator.Free(buffer);
|
|
}
|
|
|
|
virtual bool IsEqualImpl(const MemoryResource &resource) const override {
|
|
return this == std::addressof(resource);
|
|
}
|
|
};
|
|
|
|
struct SystemSaveDataInfo {
|
|
u64 id;
|
|
u64 size;
|
|
u64 journal_size;
|
|
u32 flags;
|
|
fs::SaveDataSpaceId space_id;
|
|
};
|
|
static_assert(util::is_pod<SystemSaveDataInfo>::value);
|
|
|
|
class ContentManagerImpl {
|
|
private:
|
|
constexpr static size_t MaxContentStorageRoots = 8;
|
|
constexpr static size_t MaxContentMetaDatabaseRoots = 8;
|
|
private:
|
|
struct ContentStorageRoot {
|
|
NON_COPYABLE(ContentStorageRoot);
|
|
NON_MOVEABLE(ContentStorageRoot);
|
|
|
|
char mount_name[fs::MountNameLengthMax + 1];
|
|
char path[128];
|
|
StorageId storage_id;
|
|
fs::ContentStorageId content_storage_id;
|
|
sf::SharedPointer<IContentStorage> content_storage;
|
|
|
|
ContentStorageRoot() : mount_name(), path(), storage_id(), content_storage_id(), content_storage() { /* ... */ }
|
|
};
|
|
|
|
struct ContentMetaDatabaseRoot {
|
|
NON_COPYABLE(ContentMetaDatabaseRoot);
|
|
NON_MOVEABLE(ContentMetaDatabaseRoot);
|
|
|
|
char mount_name[fs::MountNameLengthMax + 1];
|
|
char path[128];
|
|
StorageId storage_id;
|
|
SystemSaveDataInfo info;
|
|
sf::SharedPointer<IContentMetaDatabase> content_meta_database;
|
|
util::optional<kvdb::MemoryKeyValueStore<ContentMetaKey>> kvs;
|
|
ContentMetaMemoryResource *memory_resource;
|
|
u32 max_content_metas;
|
|
|
|
ContentMetaDatabaseRoot() : mount_name(), path(), storage_id(), info(), content_meta_database(), kvs(util::nullopt), memory_resource(), max_content_metas() { /* ... */ }
|
|
};
|
|
private:
|
|
os::SdkRecursiveMutex m_mutex;
|
|
bool m_initialized;
|
|
ContentStorageRoot m_content_storage_roots[MaxContentStorageRoots];
|
|
ContentMetaDatabaseRoot m_content_meta_database_roots[MaxContentMetaDatabaseRoots];
|
|
u32 m_num_content_storage_entries;
|
|
u32 m_num_content_meta_entries;
|
|
RightsIdCache m_rights_id_cache;
|
|
RegisteredHostContent m_registered_host_content;
|
|
public:
|
|
ContentManagerImpl() : m_mutex(), m_initialized(false), m_content_storage_roots(), m_content_meta_database_roots(), m_num_content_storage_entries(0), m_num_content_meta_entries(0), m_rights_id_cache(), m_registered_host_content() {
|
|
/* ... */
|
|
};
|
|
~ContentManagerImpl();
|
|
public:
|
|
Result Initialize(const ContentManagerConfig &config);
|
|
private:
|
|
/* Helpers. */
|
|
Result GetContentStorageRoot(ContentStorageRoot **out, StorageId id);
|
|
Result GetContentMetaDatabaseRoot(ContentMetaDatabaseRoot **out, StorageId id);
|
|
|
|
Result InitializeContentStorageRoot(ContentStorageRoot *out, StorageId storage_id, fs::ContentStorageId content_storage_id);
|
|
Result InitializeGameCardContentStorageRoot(ContentStorageRoot *out);
|
|
|
|
Result InitializeContentMetaDatabaseRoot(ContentMetaDatabaseRoot *out, StorageId storage_id, const SystemSaveDataInfo &info, size_t max_content_metas, ContentMetaMemoryResource *mr);
|
|
Result InitializeGameCardContentMetaDatabaseRoot(ContentMetaDatabaseRoot *out, size_t max_content_metas, ContentMetaMemoryResource *mr);
|
|
|
|
Result BuildContentMetaDatabase(StorageId storage_id);
|
|
Result ImportContentMetaDatabase(StorageId storage_id, bool from_signed_partition);
|
|
Result ImportContentMetaDatabaseImpl(StorageId storage_id, const char *import_mount_name, const char *path);
|
|
|
|
Result EnsureAndMountSystemSaveData(const char *mount, const SystemSaveDataInfo &info) const;
|
|
public:
|
|
/* Actual commands. */
|
|
Result CreateContentStorage(StorageId storage_id);
|
|
Result CreateContentMetaDatabase(StorageId storage_id);
|
|
Result VerifyContentStorage(StorageId storage_id);
|
|
Result VerifyContentMetaDatabase(StorageId storage_id);
|
|
Result OpenContentStorage(sf::Out<sf::SharedPointer<IContentStorage>> out, StorageId storage_id);
|
|
Result OpenContentMetaDatabase(sf::Out<sf::SharedPointer<IContentMetaDatabase>> out, StorageId storage_id);
|
|
Result CloseContentStorageForcibly(StorageId storage_id);
|
|
Result CloseContentMetaDatabaseForcibly(StorageId storage_id);
|
|
Result CleanupContentMetaDatabase(StorageId storage_id);
|
|
Result ActivateContentStorage(StorageId storage_id);
|
|
Result InactivateContentStorage(StorageId storage_id);
|
|
Result ActivateContentMetaDatabase(StorageId storage_id);
|
|
Result InactivateContentMetaDatabase(StorageId storage_id);
|
|
Result InvalidateRightsIdCache();
|
|
Result GetMemoryReport(sf::Out<MemoryReport> out);
|
|
};
|
|
static_assert(IsIContentManager<ContentManagerImpl>);
|
|
|
|
}
|