Преглед изворни кода

Merge pull request #1478 from ogniK5377/remap-invalidhandle-remap

Passing an invalid nmap handle to Remap should throw an error
bunnei пре 7 година
родитељ
комит
0f7ab3e21a
1 измењених фајлова са 10 додато и 3 уклоњено
  1. 10 3
      src/core/hle/service/nvdrv/devices/nvhost_as_gpu.cpp

+ 10 - 3
src/core/hle/service/nvdrv/devices/nvhost_as_gpu.cpp

@@ -15,6 +15,11 @@
 #include "video_core/renderer_base.h"
 #include "video_core/renderer_base.h"
 
 
 namespace Service::Nvidia::Devices {
 namespace Service::Nvidia::Devices {
+namespace NvErrCodes {
+enum {
+    InvalidNmapHandle = -22,
+};
+}
 
 
 nvhost_as_gpu::nvhost_as_gpu(std::shared_ptr<nvmap> nvmap_dev) : nvmap_dev(std::move(nvmap_dev)) {}
 nvhost_as_gpu::nvhost_as_gpu(std::shared_ptr<nvmap> nvmap_dev) : nvmap_dev(std::move(nvmap_dev)) {}
 nvhost_as_gpu::~nvhost_as_gpu() = default;
 nvhost_as_gpu::~nvhost_as_gpu() = default;
@@ -79,14 +84,16 @@ u32 nvhost_as_gpu::Remap(const std::vector<u8>& input, std::vector<u8>& output)
     std::memcpy(entries.data(), input.data(), input.size());
     std::memcpy(entries.data(), input.data(), input.size());
 
 
     auto& gpu = Core::System::GetInstance().GPU();
     auto& gpu = Core::System::GetInstance().GPU();
-
     for (const auto& entry : entries) {
     for (const auto& entry : entries) {
         LOG_WARNING(Service_NVDRV, "remap entry, offset=0x{:X} handle=0x{:X} pages=0x{:X}",
         LOG_WARNING(Service_NVDRV, "remap entry, offset=0x{:X} handle=0x{:X} pages=0x{:X}",
                     entry.offset, entry.nvmap_handle, entry.pages);
                     entry.offset, entry.nvmap_handle, entry.pages);
         Tegra::GPUVAddr offset = static_cast<Tegra::GPUVAddr>(entry.offset) << 0x10;
         Tegra::GPUVAddr offset = static_cast<Tegra::GPUVAddr>(entry.offset) << 0x10;
-
         auto object = nvmap_dev->GetObject(entry.nvmap_handle);
         auto object = nvmap_dev->GetObject(entry.nvmap_handle);
-        ASSERT(object);
+        if (!object) {
+            LOG_CRITICAL(Service_NVDRV, "nvmap {} is an invalid handle!", entry.nvmap_handle);
+            std::memcpy(output.data(), entries.data(), output.size());
+            return static_cast<u32>(NvErrCodes::InvalidNmapHandle);
+        }
 
 
         ASSERT(object->status == nvmap::Object::Status::Allocated);
         ASSERT(object->status == nvmap::Object::Status::Allocated);