From 4d9c68dd7beab26620c90952f66bfc5b825ef30a Mon Sep 17 00:00:00 2001 From: Alireza Ahmadi Date: Sun, 14 Jul 2024 03:36:05 +0200 Subject: [PATCH] fix domain validator #1329 --- web/middleware/domainValidator.go | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/web/middleware/domainValidator.go b/web/middleware/domainValidator.go index 2beecfdb..c94130c0 100644 --- a/web/middleware/domainValidator.go +++ b/web/middleware/domainValidator.go @@ -3,13 +3,17 @@ package middleware import ( "net" "net/http" + "strings" "github.com/gin-gonic/gin" ) func DomainValidatorMiddleware(domain string) gin.HandlerFunc { return func(c *gin.Context) { - host, _, _ := net.SplitHostPort(c.Request.Host) + host := c.Request.Host + if colonIndex := strings.LastIndex(host, ":"); colonIndex != -1 { + host, _, _ = net.SplitHostPort(c.Request.Host) + } if host != domain { c.AbortWithStatus(http.StatusForbidden)