feat: Add hostname access

layou233 · Jan 27, 2023 · bb8c73d · bb8c73d
1 parent cc52dd8
commit bb8c73d
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/config/type.go b/config/type.go
@@ -33,6 +33,9 @@ type minecraft struct {
 	EnableHostnameRewrite bool
 	RewrittenHostname     string `json:",omitempty"`
 
+	EnableHostnameAccess bool
+	HostnameAccess       string `json:",omitempty"`
+
 	OnlineCount onlineCount
 
 	IgnoreFMLSuffix bool `json:",omitempty"`

diff --git a/service/minecraft/handler.go b/service/minecraft/handler.go
@@ -63,6 +63,12 @@ func NewConnHandler(s *config.ConfigProxyService,
 	if err != nil {
 		return nil, err
 	}
+	if s.Minecraft.EnableHostnameAccess {
+		if !strings.Contains(hostname, s.Minecraft.HostnameAccess) {
+			c.(*net.TCPConn).SetLinger(0)
+			return nil, errors.New("hostname not allowed")
+		}
+	}
 	if nextState == 1 { // status
 		if s.Minecraft.MotdDescription == "" && s.Minecraft.MotdFavicon == "" {
 			// directly proxy MOTD from server