Refactor FXIOS-7772 [v123] Added url check for server side redirection

mozilla-mobile · Jan 16, 2024 · ee053a0 · ee053a0
1 parent ceb87c9
commit ee053a0
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/focus-ios/Blockzilla/Modules/WebView/WebViewController.swift b/focus-ios/Blockzilla/Modules/WebView/WebViewController.swift
@@ -379,6 +379,16 @@ extension WebViewController: UIScrollViewDelegate {
 }
 
 extension WebViewController: WKNavigationDelegate {
+    func webView(_ webView: WKWebView, didReceiveServerRedirectForProvisionalNavigation navigation: WKNavigation!) {
+        // validate the URL using URIFixup
+        guard let urlString = webView.url?.absoluteString,
+              URIFixup.getURL(entry: urlString) != nil else {
+            // URL failed validation, prevent loading
+            stop()
+            return
+        }
+    }
+
     func webView(_ webView: WKWebView, didCommit navigation: WKNavigation!) {
         delegate?.webControllerDidStartNavigation(self)
         trackingProtectionManager.trackingProtectionStatus.trackingInformation = TPPageStats()