mirror of
https://github.com/Atmosphere-NX/Atmosphere.git
synced 2025-12-19 23:45:13 +01:00
134 lines
4.5 KiB
C++
134 lines
4.5 KiB
C++
/*
|
|
* Copyright (c) 2019-2020 Adubbz, Atmosphere-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 <switch.h>
|
|
#include <stratosphere.hpp>
|
|
|
|
namespace ams::lr::impl {
|
|
|
|
template<typename Key, typename Value, size_t NumEntries>
|
|
class RegisteredData {
|
|
NON_COPYABLE(RegisteredData);
|
|
NON_MOVEABLE(RegisteredData);
|
|
private:
|
|
struct Entry {
|
|
Value value;
|
|
ncm::ProgramId owner_id;
|
|
Key key;
|
|
bool is_valid;
|
|
};
|
|
private:
|
|
Entry entries[NumEntries];
|
|
size_t soft_entry_limit;
|
|
public:
|
|
RegisteredData(size_t soft_entry_limit = NumEntries) : soft_entry_limit(soft_entry_limit) {
|
|
this->Clear();
|
|
}
|
|
|
|
bool Register(const Key &key, const Value &value, const ncm::ProgramId owner_id) {
|
|
/* Try to find an existing value. */
|
|
for (size_t i = 0; i < this->GetSoftEntryLimit(); i++) {
|
|
Entry& entry = this->entries[i];
|
|
if (entry.is_valid && entry.key == key) {
|
|
entry.value = value;
|
|
return true;
|
|
}
|
|
}
|
|
|
|
for (size_t i = 0; i < this->GetSoftEntryLimit(); i++) {
|
|
Entry& entry = this->entries[i];
|
|
if (!entry.is_valid) {
|
|
entry.key = key;
|
|
entry.value = value;
|
|
entry.owner_id = owner_id;
|
|
entry.is_valid = true;
|
|
return true;
|
|
}
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
void Unregister(const Key &key) {
|
|
for (size_t i = 0; i < this->GetSoftEntryLimit(); i++) {
|
|
Entry& entry = this->entries[i];
|
|
if (entry.is_valid && entry.key == key) {
|
|
entry.is_valid = false;
|
|
}
|
|
}
|
|
}
|
|
|
|
void UnregisterOwnerProgram(ncm::ProgramId owner_id) {
|
|
for (size_t i = 0; i < this->GetSoftEntryLimit(); i++) {
|
|
Entry& entry = this->entries[i];
|
|
if (entry.owner_id == owner_id) {
|
|
entry.is_valid = false;
|
|
}
|
|
}
|
|
}
|
|
|
|
bool Find(Value *out, const Key &key) {
|
|
for (size_t i = 0; i < this->GetSoftEntryLimit(); i++) {
|
|
Entry& entry = this->entries[i];
|
|
if (entry.is_valid && entry.key == key) {
|
|
*out = entry.value;
|
|
return true;
|
|
}
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
void Clear() {
|
|
for (size_t i = 0; i < this->GetSoftEntryLimit(); i++) {
|
|
this->entries[i].is_valid = false;
|
|
}
|
|
}
|
|
|
|
void ClearExcluding(const ncm::ProgramId* ids, size_t num_ids) {
|
|
for (size_t i = 0; i < this->GetSoftEntryLimit(); i++) {
|
|
Entry& entry = this->entries[i];
|
|
bool found = false;
|
|
|
|
for (size_t j = 0; j < num_ids; j++) {
|
|
ncm::ProgramId id = ids[j];
|
|
|
|
if (entry.owner_id == id) {
|
|
found = true;
|
|
break;
|
|
}
|
|
}
|
|
|
|
if (!found) {
|
|
entry.is_valid = false;
|
|
}
|
|
}
|
|
}
|
|
|
|
size_t GetSoftEntryLimit() const {
|
|
return this->soft_entry_limit;
|
|
}
|
|
};
|
|
|
|
template<typename Key, size_t NumEntries>
|
|
using RegisteredLocations = RegisteredData<Key, lr::Path, NumEntries>;
|
|
|
|
template<typename Key, size_t NumEntries>
|
|
using RegisteredStorages = RegisteredData<Key, ncm::StorageId, NumEntries>;
|
|
|
|
}
|