From b2d013456758f9171d7cf07cc8b8aea24074115a Mon Sep 17 00:00:00 2001 From: Alireza Ahmadi Date: Thu, 18 Jul 2024 22:44:59 +0200 Subject: [PATCH] fix domain validator --- backend/middleware/domainValidator.go | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/backend/middleware/domainValidator.go b/backend/middleware/domainValidator.go index e99d35f..4ef71eb 100644 --- a/backend/middleware/domainValidator.go +++ b/backend/middleware/domainValidator.go @@ -1,6 +1,7 @@ package middleware import ( + "net" "net/http" "strings" @@ -9,7 +10,10 @@ import ( func DomainValidator(domain string) gin.HandlerFunc { return func(c *gin.Context) { - host := strings.Split(c.Request.Host, ":")[0] + host := c.Request.Host + if colonIndex := strings.LastIndex(host, ":"); colonIndex != -1 { + host, _, _ = net.SplitHostPort(c.Request.Host) + } if host != domain { c.AbortWithStatus(http.StatusForbidden)