[focus-ios] Refactor FXIOS-7772 [v123] Added url check for server sid…

…e redirection (mozilla-mobile/focus-ios#3989)
isabelrios · Jan 17, 2024 · 9d19795 · 9d19795
1 parent 6c9f949
commit 9d19795
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/focus-ios/focus-ios/Blockzilla/Modules/WebView/WebViewController.swift b/focus-ios/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()