소스 검색

nvdrv: Unregister already registered events

Chloe Marcec 5 년 전
부모
커밋
3b0458a7a5
1개의 변경된 파일8개의 추가작업 그리고 1개의 파일을 삭제
  1. 8 1
      src/core/hle/service/nvdrv/devices/nvhost_ctrl.cpp

+ 8 - 1
src/core/hle/service/nvdrv/devices/nvhost_ctrl.cpp

@@ -155,7 +155,14 @@ NvResult nvhost_ctrl::IocCtrlEventRegister(const std::vector<u8>& input, std::ve
         return NvResult::BadParameter;
     }
     if (events_interface.registered[event_id]) {
-        return NvResult::BadParameter;
+        const auto event_state = events_interface.status[event_id];
+        if (event_state == EventState::Registered || event_state == EventState::Waiting ||
+            event_state == EventState::Busy) {
+            LOG_WARNING(Service_NVDRV, "Event already registered! Unregistering previous event");
+            events_interface.UnregisterEvent(event_id);
+        } else {
+            return NvResult::BadParameter;
+        }
     }
     events_interface.RegisterEvent(event_id);
     return NvResult::Success;