[docs] add whitelist

sh-akira · Aug 7, 2024 · 2572b3a · 2572b3a
1 parent 822f4c0
commit 2572b3a
Showing 1 changed file with 12 additions and 2 deletions.
diff --git a/docs/redirect/redirect.js b/docs/redirect/redirect.js
@@ -1,2 +1,12 @@
-const r = new URLSearchParams(window.location.search).get("r")
-if(r) window.location.replace(r)
+const allowedURLs = [
+    'ms-settings:*'
+];
+
+const r = new URLSearchParams(window.location.search).get('r');
+
+const isAllowed = allowedURLs.some((url) => {
+    const regex = new RegExp('^' + url.replace(/\*/g, '.*') + '$');
+    return regex.test(r);
+});
+
+if(r && isAllowed) window.location.replace(r);