wp-go/middleware/validateservername.go

package middleware

import (
	"github.com/gin-gonic/gin"
	"github/fthvgb1/wp-go/config"
	"github/fthvgb1/wp-go/safety"
	"net/http"
	"strings"
)

func ValidateServerNames() (func(ctx *gin.Context), func()) {
	var serverName safety.Map[string, struct{}]
	fn := func() {
		r := config.Conf.Load().TrustServerNames
		if len(r) > 0 {
			for _, name := range r {
				serverName.Store(name, struct{}{})
			}
		} else {
			serverName.Flush()
		}

	}
	fn()
	return func(c *gin.Context) {
		if serverName.Len() > 0 {
			if _, ok := serverName.Load(strings.Split(c.Request.Host, ":")[0]); !ok {
				c.Status(http.StatusForbidden)
				c.Abort()
				return
			}
		}
		c.Next()
	}, fn
}