kern/svc: implement IoPool/Region svc support

libraries/libmesosphere/source/kern_k_io_pool.cpp

@@ -0,0 +1,129 @@
+/*
+ * 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 <mesosphere.hpp>
+
+namespace ams::kern {
+
+    namespace {
+
+        constinit KLightLock g_io_pool_lock;
+        constinit bool g_pool_used[ams::svc::IoPoolType_Count];
+
+        struct IoRegionExtents {
+            KPhysicalAddress address;
+            size_t size;
+        };
+
+        #if defined(ATMOSPHERE_BOARD_NINTENDO_NX)
+
+            #include "board/nintendo/nx/kern_k_io_pool.board.nintendo_nx.inc"
+
+        #elif defined(AMS_SVC_IO_POOL_NOT_SUPPORTED)
+
+            #include "kern_k_io_pool.unsupported.inc"
+
+        #else
+
+            #error "Unknown context for IoPoolType!"
+
+        #endif
+
+        constexpr bool IsValidIoRegionImpl(ams::svc::IoPoolType pool_type, KPhysicalAddress address, size_t size) {
+            /* NOTE: It seems likely this depends on pool type, but this isn't confirmable as of now. */
+            MESOSPHERE_UNUSED(pool_type);
+
+            /* Check if the address/size falls within any allowable extents. */
+            for (const auto &extents : g_io_region_extents) {
+                if (extents.address <= address && address + size - 1 <= extents.address + extents.size - 1) {
+                    return true;
+                }
+            }
+
+            return false;
+        }
+
+    }
+
+    bool KIoPool::IsValidIoPoolType(ams::svc::IoPoolType pool_type) {
+        return IsValidIoPoolTypeImpl(pool_type);
+    }
+
+    Result KIoPool::Initialize(ams::svc::IoPoolType pool_type) {
+        MESOSPHERE_ASSERT_THIS();
+
+        /* Register the pool type. */
+        {
+            /* Lock the pool used table. */
+            KScopedLightLock lk(g_io_pool_lock);
+
+            /* Check that the pool isn't already used. */
+            R_UNLESS(!g_pool_used[pool_type], svc::ResultBusy());
+
+            /* Set the pool as used. */
+            g_pool_used[pool_type] = true;
+        }
+
+        /* Set our fields. */
+        m_pool_type      = pool_type;
+        m_is_initialized = true;
+
+        return ResultSuccess();
+    }
+
+    void KIoPool::Finalize() {
+        MESOSPHERE_ASSERT_THIS();
+
+        /* Lock the pool used table. */
+        KScopedLightLock lk(g_io_pool_lock);
+
+        /* Check that the pool is used. */
+        MESOSPHERE_ASSERT(g_pool_used[m_pool_type]);
+
+        /* Set the pool as unused. */
+        g_pool_used[m_pool_type] = false;
+    }
+
+    Result KIoPool::AddIoRegion(KIoRegion *new_region) {
+        MESOSPHERE_ASSERT_THIS();
+
+        /* Check that the region is allowed. */
+        R_UNLESS(IsValidIoRegionImpl(m_pool_type, new_region->GetAddress(), new_region->GetSize()), svc::ResultInvalidMemoryRegion());
+
+        /* Lock ourselves. */
+        KScopedLightLock lk(m_lock);
+
+        /* Check that the desired range isn't already in our pool. */
+        for (const auto &region : m_io_region_list) {
+            R_UNLESS(!region.Overlaps(new_region->GetAddress(), new_region->GetSize()), svc::ResultBusy());
+        }
+
+        /* Add the region to our pool. */
+        m_io_region_list.push_back(*new_region);
+
+        return ResultSuccess();
+    }
+
+    void KIoPool::RemoveIoRegion(KIoRegion *region) {
+        MESOSPHERE_ASSERT_THIS();
+
+        /* Lock ourselves. */
+        KScopedLightLock lk(m_lock);
+
+        /* Remove the region from our list. */
+        m_io_region_list.erase(m_io_region_list.iterator_to(*region));
+    }
+
+}