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)