troposphere: add haze MTP server (#2095)

* 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
2023-04-17 17:08:05 -04:00
parent e9b28ab4b1
commit 3b662122f9
26 changed files with 3674 additions and 1 deletions
--- a/troposphere/haze/source/ptp_object_heap.cpp
+++ b/troposphere/haze/source/ptp_object_heap.cpp
@@ -0,0 +1,69 @@
+/*
+ * 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 {
+
+    namespace {
+
+        /* Allow 20MiB for use by libnx. */
+        static constexpr size_t LibnxReservedMemorySize = 20_MB;
+
+    }
+
+    void PtpObjectHeap::Initialize() {
+        /* If we're already initialized, skip re-initialization. */
+        if (m_heap_block_size != 0) {
+            return;
+        }
+
+        /* Estimate how much memory we can reserve. */
+        size_t mem_used = 0;
+        HAZE_R_ABORT_UNLESS(svcGetInfo(std::addressof(mem_used), InfoType_UsedMemorySize, svc::CurrentProcess, 0));
+        HAZE_ASSERT(mem_used > LibnxReservedMemorySize);
+        mem_used -= LibnxReservedMemorySize;
+
+        /* Calculate size of blocks. */
+        m_heap_block_size = mem_used / NumHeapBlocks;
+        HAZE_ASSERT(m_heap_block_size > 0);
+
+        /* Allocate the memory. */
+        for (size_t i = 0; i < NumHeapBlocks; i++) {
+            m_heap_blocks[i] = std::malloc(m_heap_block_size);
+            HAZE_ASSERT(m_heap_blocks[i] != nullptr);
+        }
+
+        /* Set the address to allocate from. */
+        m_next_address = m_heap_blocks[0];
+    }
+
+    void PtpObjectHeap::Finalize() {
+        if (m_heap_block_size == 0) {
+            return;
+        }
+
+        /* Tear down the heap, allowing a subsequent call to Initialize() if desired. */
+        for (size_t i = 0; i < NumHeapBlocks; i++) {
+            std::free(m_heap_blocks[i]);
+            m_heap_blocks[i] = nullptr;
+        }
+
+        m_next_address       = nullptr;
+        m_heap_block_size    = 0;
+        m_current_heap_block = 0;
+    }
+
+}