From 8d31a457b84e65170d2c34a6921b92b52e6693b0 Mon Sep 17 00:00:00 2001
From: Ramana Reddy <ramanaredy.manda@gmail.com>
Date: Thu, 30 May 2024 16:52:12 +0530
Subject: [PATCH] fix edge cases

---
 common/httpx/csp.go | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

diff --git a/common/httpx/csp.go b/common/httpx/csp.go
index 1992796..6cd2103 100644
--- a/common/httpx/csp.go
+++ b/common/httpx/csp.go
@@ -91,9 +91,18 @@ func extractDomain(str string) string {
 }
 
 func removeWildcards(domain string) string {
+	if stringsutil.HasPrefixAny(domain, "'", "\"") {
+		domain = domain[1:]
+	}
+	if stringsutil.HasSuffixAny(domain, "'", "\"") {
+		domain = domain[:len(domain)-1]
+	}
+	if strings.Contains(domain, "://") {
+		domain = strings.Split(domain, "://")[1]
+	}
 	parts := []string{}
 	for _, part := range strings.Split(domain, ".") {
-		if part != "*" {
+		if !strings.Contains(part, "*") {
 			parts = append(parts, part)
 		}
 	}