From e7c59adc5928d9d9ef96480297f7618e3d57837c Mon Sep 17 00:00:00 2001 From: mhsanaei Date: Wed, 22 May 2024 17:51:55 +0200 Subject: Ensure IPv6 compliant host Co-Authored-By: vnxme <46669194+vnxme@users.noreply.github.com> --- web/middleware/domainValidator.go | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'web/middleware') diff --git a/web/middleware/domainValidator.go b/web/middleware/domainValidator.go index 3adb0f0f..2beecfdb 100644 --- a/web/middleware/domainValidator.go +++ b/web/middleware/domainValidator.go @@ -1,15 +1,15 @@ package middleware import ( + "net" "net/http" - "strings" "github.com/gin-gonic/gin" ) func DomainValidatorMiddleware(domain string) gin.HandlerFunc { return func(c *gin.Context) { - host := strings.Split(c.Request.Host, ":")[0] + host, _, _ := net.SplitHostPort(c.Request.Host) if host != domain { c.AbortWithStatus(http.StatusForbidden) -- cgit v1.2.3