Improve NvDevice Registration + Access

StigStiggi · Sep 20, 2020 · 20253a9 · 20253a9
1 parent 4cc3a3b
commit 20253a9
Show file tree

Hide file tree

Showing 18 changed files with 149 additions and 166 deletions.
diff --git a/.gitmodules b/.gitmodules
@@ -14,3 +14,6 @@
 [submodule "app/libraries/lz4"]
 	path = app/libraries/lz4
 	url = https://github.com/lz4/lz4.git
+[submodule "app/libraries/frozen"]
+	path = app/libraries/frozen
+	url = https://github.com/serge-sans-paille/frozen
diff --git a/.idea/codeStyles/Project.xml b/.idea/codeStyles/Project.xml
diff --git a/.idea/vcs.xml b/.idea/vcs.xml
diff --git a/app/CMakeLists.txt b/app/CMakeLists.txt
@@ -22,6 +22,7 @@ add_subdirectory("libraries/lz4/contrib/cmake_unofficial")
 include_directories("libraries/lz4/lib")
 include_directories("libraries/oboe/include")
 include_directories("libraries/vkhpp/include")
+include_directories("libraries/frozen/include")
 set(CMAKE_POLICY_DEFAULT_CMP0048 NEW)
 
 find_package(mbedtls REQUIRED CONFIG)
@@ -114,6 +115,7 @@ add_library(skyline SHARED
         ${source_DIR}/skyline/services/fssrv/IStorage.cpp
         ${source_DIR}/skyline/services/nvdrv/INvDrvServices.cpp
         ${source_DIR}/skyline/services/nvdrv/driver.cpp
+        ${source_DIR}/skyline/services/nvdrv/devices/nvdevice.cpp
         ${source_DIR}/skyline/services/nvdrv/devices/nvmap.cpp
         ${source_DIR}/skyline/services/nvdrv/devices/nvhost_ctrl_gpu.cpp
         ${source_DIR}/skyline/services/nvdrv/devices/nvhost_ctrl.cpp

diff --git a/app/build.gradle b/app/build.gradle
@@ -4,11 +4,11 @@ apply plugin: 'kotlin-android-extensions'
 
 android {
     compileSdkVersion 30
-    buildToolsVersion '30.0.0'
+    buildToolsVersion '30.0.2'
     defaultConfig {
         applicationId "skyline.emu"
         minSdkVersion 26
-        targetSdkVersion 29
+        targetSdkVersion 30
         versionCode 3
         versionName "0.3"
         ndk {
@@ -59,7 +59,7 @@ android {
 dependencies {
     implementation fileTree(dir: 'libs', include: ['*.jar'])
     implementation 'androidx.appcompat:appcompat:1.2.0'
-    implementation 'androidx.constraintlayout:constraintlayout:1.1.3'
+    implementation 'androidx.constraintlayout:constraintlayout:2.0.1'
     implementation 'androidx.preference:preference:1.1.1'
     implementation 'com.google.android.material:material:1.3.0-alpha02'
     implementation "androidx.core:core-ktx:1.3.1"

diff --git a/app/libraries/frozen b/app/libraries/frozen
diff --git a/app/src/main/cpp/skyline/common.h b/app/src/main/cpp/skyline/common.h
@@ -8,21 +8,24 @@
 #include <span>
 #include <vector>
 #include <fstream>
-#include <syslog.h>
 #include <mutex>
 #include <thread>
 #include <string>
-#include <sstream>
-#include <memory>
-#include <fmt/format.h>
-#include <sys/mman.h>
 #include <cstdint>
 #include <stdexcept>
 #include <string>
+#include <sstream>
+#include <memory>
+#include <syslog.h>
+#include <sys/mman.h>
+#include <fmt/format.h>
+#include <frozen/unordered_map.h>
+#include <frozen/string.h>
 #include <jni.h>
 #include "nce/guest_common.h"
 
 namespace skyline {
+    namespace frz = frozen;
     using KHandle = u32; //!< The type of a kernel handle
 
     /**
@@ -194,6 +197,10 @@ namespace skyline {
             }
             return result;
         }
+
+        constexpr std::size_t Hash(const std::string_view& view) {
+            return frz::elsa<frz::string>{}(frz::string(view.data(), view.size()), 0);
+        }
     }
 
     /**

diff --git a/app/src/main/cpp/skyline/services/hosbinder/GraphicBufferProducer.cpp b/app/src/main/cpp/skyline/services/hosbinder/GraphicBufferProducer.cpp
@@ -124,7 +124,7 @@ namespace skyline::service::hosbinder {
         std::shared_ptr<nvdrv::device::NvMap::NvMapObject> nvBuffer{};
 
         auto driver = nvdrv::driver.lock();
-        auto nvmap = driver->GetDevice<nvdrv::device::NvMap>(nvdrv::device::NvDeviceType::nvmap);
+        auto nvmap = driver->nvMap.lock();
 
         if (gbpBuffer.nvmapHandle) {
             nvBuffer = nvmap->handleTable.at(gbpBuffer.nvmapHandle);

diff --git a/app/src/main/cpp/skyline/services/nvdrv/devices/nvdevice.cpp b/app/src/main/cpp/skyline/services/nvdrv/devices/nvdevice.cpp
@@ -0,0 +1,34 @@
+// SPDX-License-Identifier: MPL-2.0
+// Copyright © 2020 Skyline Team and Contributors (https://github.com/skyline-emu/)
+
+#include "nvdevice.h"
+
+namespace skyline::service::nvdrv::device {
+    NvDevice::NvDevice(const DeviceState &state, std::unordered_map <u32, std::function<void(IoctlData &)>> vTable) : state(state), vTable(vTable) {}
+
+    std::string NvDevice::GetName() {
+        int status{};
+        size_t length{};
+        auto mangledName{typeid(*this).name()};
+
+        std::unique_ptr<char, decltype(&std::free)> demangled{ abi::__cxa_demangle(mangledName, nullptr, &length, &status), std::free};
+
+        return (status == 0) ? std::string(demangled.get() + std::char_traits<char>::length("skyline::service::nvdrv::device")) : mangledName;
+    }
+
+    void NvDevice::HandleIoctl(u32 cmd, IoctlData &input) {
+        std::function<void(IoctlData &)> function;
+        try {
+            function = vTable.at(cmd);
+        } catch (std::out_of_range &) {
+            state.logger->Warn("Cannot find IOCTL for device '{}': 0x{:X}", GetName(), cmd);
+            input.status = NvStatus::NotImplemented;
+            return;
+        }
+        try {
+            function(input);
+        } catch (std::exception &e) {
+            throw exception("{} (Device: {})", e.what(), GetName());
+        }
+    }
+}
diff --git a/app/src/main/cpp/skyline/services/nvdrv/devices/nvdevice.h b/app/src/main/cpp/skyline/services/nvdrv/devices/nvdevice.h
@@ -3,6 +3,7 @@
 
 #pragma once
 
+#include <cxxabi.h>
 #include <common.h>
 #include <kernel/ipc.h>
 #include <kernel/types/KEvent.h>
@@ -12,54 +13,6 @@
 namespace skyline::service::nvdrv::device {
     using namespace kernel;
 
-    /**
-     * @brief An enumeration of all the devices that can be opened by nvdrv
-     */
-    enum class NvDeviceType {
-        nvhost_ctrl, //!< https://switchbrew.org/wiki/NV_services#.2Fdev.2Fnvhost-ctrl
-        nvhost_gpu, //!< https://switchbrew.org/wiki/NV_services#Channels
-        nvhost_nvdec, //!< https://switchbrew.org/wiki/NV_services#Channels
-        nvhost_vic, //!< https://switchbrew.org/wiki/NV_services#Channels
-        nvmap, //!< https://switchbrew.org/wiki/NV_services#.2Fdev.2Fnvmap
-        nvdisp_ctrl, //!< https://switchbrew.org/wiki/NV_services#.2Fdev.2Fnvdisp-ctrl
-        nvdisp_disp0, //!< https://switchbrew.org/wiki/NV_services#.2Fdev.2Fnvdisp-disp0.2C_.2Fdev.2Fnvdisp-disp1
-        nvdisp_disp1, //!< https://switchbrew.org/wiki/NV_services#.2Fdev.2Fnvdisp-disp0.2C_.2Fdev.2Fnvdisp-disp1
-        nvcec_ctrl, //!< https://switchbrew.org/wiki/NV_services#.2Fdev.2Fnvcec-ctrl
-        nvhdcp_up_ctrl, //!< https://switchbrew.org/wiki/NV_services#.2Fdev.2Fnvhdcp_up-ctrl
-        nvdcutil_disp0, //!< https://switchbrew.org/wiki/NV_services#.2Fdev.2Fnvdcutil-disp0.2C_.2Fdev.2Fnvdcutil-disp1
-        nvdcutil_disp1, //!< https://switchbrew.org/wiki/NV_services#.2Fdev.2Fnvdcutil-disp0.2C_.2Fdev.2Fnvdcutil-disp1
-        nvsched_ctrl, //!< https://switchbrew.org/wiki/NV_services#.2Fdev.2Fnvsched-ctrl
-        nverpt_ctrl, //!< https://switchbrew.org/wiki/NV_services#.2Fdev.2Fnverpt-ctrl
-        nvhost_as_gpu, //!< https://switchbrew.org/wiki/NV_services#.2Fdev.2Fnvhost-as-gpu
-        nvhost_dbg_gpu, //!< https://switchbrew.org/wiki/NV_services#.2Fdev.2Fnvhost-dbg-gpu
-        nvhost_prof_gpu, //!< https://switchbrew.org/wiki/NV_services#.2Fdev.2Fnvhost-prof-gpu
-        nvhost_ctrl_gpu //!< https://switchbrew.org/wiki/NV_services#.2Fdev.2Fnvhost-ctrl-gpu
-    };
-
-    /**
-     * @brief A mapping from a device's path to it's nvDevice entry
-     */
-    const static std::unordered_map<std::string, NvDeviceType> nvDeviceMap{
-        {"/dev/nvhost-ctrl", NvDeviceType::nvhost_ctrl},
-        {"/dev/nvhost-gpu", NvDeviceType::nvhost_gpu},
-        {"/dev/nvhost-nvdec", NvDeviceType::nvhost_nvdec},
-        {"/dev/nvhost-vic", NvDeviceType::nvhost_vic},
-        {"/dev/nvmap", NvDeviceType::nvmap},
-        {"/dev/nvdisp-ctrl", NvDeviceType::nvdisp_ctrl},
-        {"/dev/nvdisp-disp0", NvDeviceType::nvdisp_disp0},
-        {"/dev/nvdisp-disp1", NvDeviceType::nvdisp_disp1},
-        {"/dev/nvcec-ctrl", NvDeviceType::nvcec_ctrl},
-        {"/dev/nvhdcp_up-ctrl", NvDeviceType::nvhdcp_up_ctrl},
-        {"/dev/nvdcutil-disp0", NvDeviceType::nvdcutil_disp0},
-        {"/dev/nvdcutil-disp1", NvDeviceType::nvdcutil_disp1},
-        {"/dev/nvsched-ctrl", NvDeviceType::nvsched_ctrl},
-        {"/dev/nverpt-ctrl", NvDeviceType::nverpt_ctrl},
-        {"/dev/nvhost-as-gpu", NvDeviceType::nvhost_as_gpu},
-        {"/dev/nvhost-dbg-gpu", NvDeviceType::nvhost_dbg_gpu},
-        {"/dev/nvhost-prof-gpu", NvDeviceType::nvhost_prof_gpu},
-        {"/dev/nvhost-ctrl-gpu", NvDeviceType::nvhost_ctrl_gpu},
-    };
-
     /**
      * @brief This enumerates all the possible error codes returned by the Nvidia driver (https://switchbrew.org/wiki/NV_services#Errors)
      */
@@ -134,61 +87,35 @@ namespace skyline::service::nvdrv::device {
     };
 
     /**
-     * @brief NvDevice is the base class all /dev/nv* devices inherit from
+     * @brief NvDevice is the base class that all /dev/nv* devices inherit from
      */
     class NvDevice {
       protected:
         const DeviceState &state; //!< The state of the device
         std::unordered_map<u32, std::function<void(IoctlData &)>> vTable; //!< This holds the mapping from an Ioctl to the actual function
 
       public:
-        u16 refCount{1}; //!< The amount of handles to the device
-        NvDeviceType deviceType; //!< The type of the device
-
         /**
          * @param state The state of the device
-         * @param deviceType The type of the device
          * @param vTable The functions in this device
          */
-        NvDevice(const DeviceState &state, NvDeviceType deviceType, std::unordered_map<u32, std::function<void(IoctlData &)>> vTable) : state(state), deviceType(deviceType), vTable(vTable) {}
+        NvDevice(const DeviceState &state, std::unordered_map<u32, std::function<void(IoctlData & )>> vTable);
 
         virtual ~NvDevice() = default;
 
         /**
-         * @brief This returns the name of the current service
-         * @note It may not return the exact name the service was initialized with if there are multiple entries in ServiceString
-         * @return The name of the service
+         * @return The name of the class
          */
-        std::string getName() {
-            std::string serviceName;
-            for (const auto&[name, type] : nvDeviceMap)
-                if (type == deviceType)
-                    serviceName = name;
-            return serviceName;
-        }
+        std::string GetName();
 
         /**
          * @brief This handles IOCTL calls for devices
          * @param cmd The IOCTL command that was called
          * @param input The input to the IOCTL call
          */
-        void HandleIoctl(u32 cmd, IoctlData &input) {
-            std::function<void(IoctlData &)> function;
-            try {
-                function = vTable.at(cmd);
-            } catch (std::out_of_range &) {
-                state.logger->Warn("Cannot find IOCTL for device '{}': 0x{:X}", getName(), cmd);
-                input.status = NvStatus::NotImplemented;
-                return;
-            }
-            try {
-                function(input);
-            } catch (std::exception &e) {
-                throw exception("{} (Device: {})", e.what(), getName());
-            }
-        }
+        void HandleIoctl(u32 cmd, IoctlData &input);
 
-        virtual std::shared_ptr<kernel::type::KEvent> QueryEvent(u32 eventId) {
+        inline virtual std::shared_ptr<kernel::type::KEvent> QueryEvent(u32 eventId) {
             return nullptr;
         }
     };

diff --git a/app/src/main/cpp/skyline/services/nvdrv/devices/nvhost_as_gpu.cpp b/app/src/main/cpp/skyline/services/nvdrv/devices/nvhost_as_gpu.cpp
@@ -9,7 +9,7 @@
 #include "nvhost_as_gpu.h"
 
 namespace skyline::service::nvdrv::device {
-    NvHostAsGpu::NvHostAsGpu(const DeviceState &state) : NvDevice(state, NvDeviceType::nvhost_as_gpu, {
+    NvHostAsGpu::NvHostAsGpu(const DeviceState &state) : NvDevice(state, {
         {0x4101, NFUNC(NvHostAsGpu::BindChannel)},
         {0x4102, NFUNC(NvHostAsGpu::AllocSpace)},
         {0x4105, NFUNC(NvHostAsGpu::UnmapBuffer)},
@@ -71,10 +71,11 @@ namespace skyline::service::nvdrv::device {
             return;
 
         auto driver = nvdrv::driver.lock();
-        auto nvmap = driver->GetDevice<nvdrv::device::NvMap>(nvdrv::device::NvDeviceType::nvmap)->handleTable.at(region.nvmapHandle);
+        auto nvmap = driver->nvMap.lock();
+        auto mapping = nvmap->handleTable.at(region.nvmapHandle);
 
-        u64 mapPhysicalAddress = region.bufferOffset + nvmap->address;
-        u64 mapSize = region.mappingSize ? region.mappingSize : nvmap->size;
+        u64 mapPhysicalAddress = region.bufferOffset + mapping->address;
+        u64 mapSize = region.mappingSize ? region.mappingSize : mapping->size;
 
         if (region.flags & 1)
             region.offset = state.gpu->memoryManager.MapFixed(region.offset, mapPhysicalAddress, mapSize);
@@ -135,10 +136,11 @@ namespace skyline::service::nvdrv::device {
         for (auto entry : entries) {
             try {
                 auto driver = nvdrv::driver.lock();
-                auto nvmap = driver->GetDevice<nvdrv::device::NvMap>(nvdrv::device::NvDeviceType::nvmap)->handleTable.at(entry.nvmapHandle);
+                auto nvmap = driver->nvMap.lock();
+                auto mapping = nvmap->handleTable.at(entry.nvmapHandle);
 
                 u64 mapAddress = static_cast<u64>(entry.gpuOffset) << MinAlignmentShift;
-                u64 mapPhysicalAddress = nvmap->address + (static_cast<u64>(entry.mapOffset) << MinAlignmentShift);
+                u64 mapPhysicalAddress = mapping->address + (static_cast<u64>(entry.mapOffset) << MinAlignmentShift);
                 u64 mapSize = static_cast<u64>(entry.pages) << MinAlignmentShift;
 
                 state.gpu->memoryManager.MapFixed(mapAddress, mapPhysicalAddress, mapSize);

diff --git a/app/src/main/cpp/skyline/services/nvdrv/devices/nvhost_channel.cpp b/app/src/main/cpp/skyline/services/nvdrv/devices/nvhost_channel.cpp
@@ -8,7 +8,7 @@
 #include "nvhost_channel.h"
 
 namespace skyline::service::nvdrv::device {
-    NvHostChannel::NvHostChannel(const DeviceState &state, NvDeviceType type) : smExceptionBreakpointIntReportEvent(std::make_shared<type::KEvent>(state)), smExceptionBreakpointPauseReportEvent(std::make_shared<type::KEvent>(state)), errorNotifierEvent(std::make_shared<type::KEvent>(state)), NvDevice(state, type, {
+    NvHostChannel::NvHostChannel(const DeviceState &state) : smExceptionBreakpointIntReportEvent(std::make_shared<type::KEvent>(state)), smExceptionBreakpointPauseReportEvent(std::make_shared<type::KEvent>(state)), errorNotifierEvent(std::make_shared<type::KEvent>(state)), NvDevice(state, {
         {0x4801, NFUNC(NvHostChannel::SetNvmapFd)},
         {0x4803, NFUNC(NvHostChannel::SetSubmitTimeout)},
         {0x4808, NFUNC(NvHostChannel::SubmitGpfifo)},

diff --git a/app/src/main/cpp/skyline/services/nvdrv/devices/nvhost_channel.h b/app/src/main/cpp/skyline/services/nvdrv/devices/nvhost_channel.h
@@ -25,7 +25,7 @@ namespace skyline::service::nvdrv::device {
         std::shared_ptr<type::KEvent> errorNotifierEvent;
 
       public:
-        NvHostChannel(const DeviceState &state, NvDeviceType type);
+        NvHostChannel(const DeviceState &state);
 
         /**
          * @brief This sets the nvmap file descriptor (https://switchbrew.org/wiki/NV_services#NVGPU_IOCTL_CHANNEL_SET_NVMAP_FD)

diff --git a/app/src/main/cpp/skyline/services/nvdrv/devices/nvhost_ctrl.cpp b/app/src/main/cpp/skyline/services/nvdrv/devices/nvhost_ctrl.cpp
@@ -37,7 +37,7 @@ namespace skyline::service::nvdrv::device {
         }
     }
 
-    NvHostCtrl::NvHostCtrl(const DeviceState &state) : NvDevice(state, NvDeviceType::nvhost_ctrl, {
+    NvHostCtrl::NvHostCtrl(const DeviceState &state) : NvDevice(state, {
         {0x001B, NFUNC(NvHostCtrl::GetConfig)},
         {0x001C, NFUNC(NvHostCtrl::EventSignal)},
         {0x001D, NFUNC(NvHostCtrl::EventWait)},

diff --git a/app/src/main/cpp/skyline/services/nvdrv/devices/nvhost_ctrl_gpu.cpp b/app/src/main/cpp/skyline/services/nvdrv/devices/nvhost_ctrl_gpu.cpp
@@ -5,7 +5,7 @@
 #include "nvhost_ctrl_gpu.h"
 
 namespace skyline::service::nvdrv::device {
-    NvHostCtrlGpu::NvHostCtrlGpu(const DeviceState &state) : errorNotifierEvent(std::make_shared<type::KEvent>(state)), unknownEvent(std::make_shared<type::KEvent>(state)), NvDevice(state, NvDeviceType::nvhost_ctrl_gpu, {
+    NvHostCtrlGpu::NvHostCtrlGpu(const DeviceState &state) : errorNotifierEvent(std::make_shared<type::KEvent>(state)), unknownEvent(std::make_shared<type::KEvent>(state)), NvDevice(state, {
         {0x4701, NFUNC(NvHostCtrlGpu::ZCullGetCtxSize)},
         {0x4702, NFUNC(NvHostCtrlGpu::ZCullGetInfo)},
         {0x4706, NFUNC(NvHostCtrlGpu::GetTpcMasks)},

diff --git a/app/src/main/cpp/skyline/services/nvdrv/devices/nvmap.cpp b/app/src/main/cpp/skyline/services/nvdrv/devices/nvmap.cpp
@@ -7,7 +7,7 @@
 namespace skyline::service::nvdrv::device {
     NvMap::NvMapObject::NvMapObject(u32 id, u32 size) : id(id), size(size) {}
 
-    NvMap::NvMap(const DeviceState &state) : NvDevice(state, NvDeviceType::nvmap, {
+    NvMap::NvMap(const DeviceState &state) : NvDevice(state, {
         {0x0101, NFUNC(NvMap::Create)},
         {0x0103, NFUNC(NvMap::FromId)},
         {0x0104, NFUNC(NvMap::Alloc)},