memory leak fix

NiiightmareXD · Feb 10, 2024 · 0a0113c · 0a0113c
1 parent f819156
commit 0a0113c
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/graphics_capture_api.rs b/src/graphics_capture_api.rs
@@ -367,11 +367,14 @@ impl GraphicsCaptureApi {
 impl Drop for GraphicsCaptureApi {
     fn drop(&mut self) {
         if let Some(frame_pool) = self.frame_pool.take() {
+            frame_pool.RemoveFrameArrived(self.frame_arrived_event_token).expect("Failed to remove Frame Arrived event handler");
             frame_pool.Close().expect("Failed to Close Frame Pool");
         }
 
         if let Some(session) = self.session.take() {
             session.Close().expect("Failed to Close Capture Session");
         }
+
+        self.item.RemoveClosed(self.capture_closed_event_token).expect("Failed to remove Capture Session Closed event handler");
     }
 }