Ensure IPv6 compliant host

Co-Authored-By: vnxme <46669194+vnxme@users.noreply.github.com>
This commit is contained in:
mhsanaei
2024-05-22 17:51:55 +02:00
parent 68c9b55447
commit e7c59adc59
3 changed files with 6 additions and 6 deletions

View File

@@ -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)