mirror of
https://github.com/Atmosphere-NX/Atmosphere.git
synced 2024-12-12 21:42:04 +00:00
3b662122f9
* troposphere: add haze MTP server * haze: adjust banner, new lines between class sections, single-statement if * haze: remove additional newlines between sections * haze: avoid use of reference out parameter * haze: console_main_loop: style * haze: event_reactor, file_system_proxy, ptp: style * haze: ptp_data_builder, ptp_object_database, ptp_object_heap, results, usb_session: style * haze: event_reactor, ptp_data_parser, async_usb_server, ptp_object_database, ptp_object_heap: style * haze: ptp_responder: style * haze: usb_session: style * haze: use svc defs from vapours * haze: misc comments * haze: fix pointer overflow check * haze: fix copy paste error * haze: use alignment, invalidate cached use values in console * haze: fix stray hex constant * haze: misc style * haze: fixes for windows * haze: add GetObjectPropsSupported, GetObjectPropDesc, GetObjectPropValue * haze: add SetObjectPropValue * haze: improve object database API * haze: improve WriteVariableLengthData readability * haze: fix directory renames on windows * haze: ptp_object_database: fix collation * haze: event_reactor: fix size validation * haze: ptp_responder: improve documentation * haze: ptp_responder: avoid unnecessary fs interaction in GetObjectPropValue * haze: ptp_responder: fix object deletion on windows * haze: tear down sessions on suspension * haze: prefer more specific data types * haze: misc * haze: fix usb 3.0 support * haze: improve host-to-device file transfer performance * haze: report device serial * haze: move static_assert to requires, fix alignment
152 lines
5.3 KiB
C++
152 lines
5.3 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/>.
|
|
*/
|
|
#include <haze.hpp>
|
|
|
|
namespace haze {
|
|
|
|
void PtpObjectDatabase::Initialize(PtpObjectHeap *object_heap) {
|
|
m_object_heap = object_heap;
|
|
m_object_heap->Initialize();
|
|
|
|
std::construct_at(std::addressof(m_name_tree));
|
|
std::construct_at(std::addressof(m_object_id_tree));
|
|
|
|
m_next_object_id = 1;
|
|
}
|
|
|
|
void PtpObjectDatabase::Finalize() {
|
|
std::destroy_at(std::addressof(m_object_id_tree));
|
|
std::destroy_at(std::addressof(m_name_tree));
|
|
|
|
m_next_object_id = 0;
|
|
|
|
m_object_heap->Finalize();
|
|
m_object_heap = nullptr;
|
|
}
|
|
|
|
Result PtpObjectDatabase::CreateOrFindObject(const char *parent_name, const char *name, u32 parent_id, PtpObject **out_object) {
|
|
constexpr auto separator = "/";
|
|
|
|
/* Calculate length of the new name with null terminator. */
|
|
const size_t parent_name_len = util::Strlen(parent_name);
|
|
const size_t separator_len = util::Strlen(separator);
|
|
const size_t name_len = util::Strlen(name);
|
|
const size_t terminator_len = 1;
|
|
const size_t alloc_len = sizeof(PtpObject) + parent_name_len + separator_len + name_len + terminator_len;
|
|
|
|
/* Allocate memory for the object. */
|
|
PtpObject * const object = m_object_heap->Allocate<PtpObject>(alloc_len);
|
|
R_UNLESS(object != nullptr, haze::ResultOutOfMemory());
|
|
|
|
/* Build the object name. */
|
|
std::strncpy(object->m_name, parent_name, parent_name_len + terminator_len);
|
|
std::strncpy(object->m_name + parent_name_len, separator, separator_len + terminator_len);
|
|
std::strncpy(object->m_name + parent_name_len + separator_len, name, name_len + terminator_len);
|
|
|
|
{
|
|
/* Ensure we maintain a clean state on failure. */
|
|
auto guard = SCOPE_GUARD { m_object_heap->Deallocate(object, alloc_len); };
|
|
|
|
/* Check if an object with this name already exists. If it does, we can just return it here. */
|
|
if (auto * const existing = this->GetObjectByName(object->GetName()); existing != nullptr) {
|
|
*out_object = existing;
|
|
R_SUCCEED();
|
|
}
|
|
|
|
/* Persist the reference to the object. */
|
|
guard.Cancel();
|
|
}
|
|
|
|
/* Set object properties. */
|
|
object->m_parent_id = parent_id;
|
|
object->m_object_id = 0;
|
|
|
|
/* Set output. */
|
|
*out_object = object;
|
|
|
|
/* We succeeded. */
|
|
R_SUCCEED();
|
|
}
|
|
|
|
void PtpObjectDatabase::RegisterObject(PtpObject *object, u32 desired_id) {
|
|
/* If the object is already registered, skip registration. */
|
|
if (object->GetIsRegistered()) {
|
|
return;
|
|
}
|
|
|
|
/* Set desired object ID. */
|
|
if (desired_id == 0) {
|
|
desired_id = m_next_object_id++;
|
|
}
|
|
|
|
/* Insert object into trees. */
|
|
object->Register(desired_id);
|
|
m_object_id_tree.insert(*object);
|
|
m_name_tree.insert(*object);
|
|
}
|
|
|
|
void PtpObjectDatabase::UnregisterObject(PtpObject *object) {
|
|
/* If the object is not registered, skip trying to unregister. */
|
|
if (!object->GetIsRegistered()) {
|
|
return;
|
|
}
|
|
|
|
/* Remove object from trees. */
|
|
m_object_id_tree.erase(m_object_id_tree.iterator_to(*object));
|
|
m_name_tree.erase(m_name_tree.iterator_to(*object));
|
|
object->Unregister();
|
|
}
|
|
|
|
void PtpObjectDatabase::DeleteObject(PtpObject *object) {
|
|
/* Unregister the object as required. */
|
|
this->UnregisterObject(object);
|
|
|
|
/* Free the object. */
|
|
m_object_heap->Deallocate(object, sizeof(PtpObject) + std::strlen(object->GetName()) + 1);
|
|
}
|
|
|
|
Result PtpObjectDatabase::CreateAndRegisterObjectId(const char *parent_name, const char *name, u32 parent_id, u32 *out_object_id) {
|
|
/* Try to create the object. */
|
|
PtpObject *object;
|
|
R_TRY(this->CreateOrFindObject(parent_name, name, parent_id, std::addressof(object)));
|
|
|
|
/* We succeeded, so register it. */
|
|
this->RegisterObject(object);
|
|
|
|
/* Set the output ID. */
|
|
*out_object_id = object->GetObjectId();
|
|
|
|
R_SUCCEED();
|
|
}
|
|
|
|
PtpObject *PtpObjectDatabase::GetObjectById(u32 object_id) {
|
|
/* Find in ID mapping. */
|
|
if (auto it = m_object_id_tree.find_key(object_id); it != m_object_id_tree.end()) {
|
|
return std::addressof(*it);
|
|
} else {
|
|
return nullptr;
|
|
}
|
|
}
|
|
|
|
PtpObject *PtpObjectDatabase::GetObjectByName(const char *name) {
|
|
/* Find in name mapping. */
|
|
if (auto it = m_name_tree.find_key(name); it != m_name_tree.end()) {
|
|
return std::addressof(*it);
|
|
} else {
|
|
return nullptr;
|
|
}
|
|
}
|
|
}
|