fix: tried to access uninitialized instance variable

tauri-apps · Nov 23, 2024 · 48e31fa · 48e31fa
1 parent eb0c816
commit 48e31fa
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/src/wkwebview/class/url_scheme_handler.rs b/src/wkwebview/class/url_scheme_handler.rs
@@ -261,8 +261,15 @@ extern "C" fn start_task(
                 }))
                 .unwrap();
 
-                webview.remove_custom_task_key(task_key);
-                Ok(())
+                {
+                  let ids = WEBVIEW_IDS.lock().unwrap();
+                  if ids.contains(webview_id) {
+                    webview.remove_custom_task_key(task_key);
+                    Ok(())
+                  } else {
+                    Err(crate::Error::CustomProtocolTaskInvalid)
+                  }
+                }
               }
 
               let _ = response(