浏览代码

Fix: TCP & WS override AcceptProxyProtocol (#425)

Jim Han 4 年之前
父节点
当前提交
e564d9ef7e
共有 2 个文件被更改,包括 4 次插入2 次删除
  1. 2 1
      transport/internet/tcp/hub.go
  2. 2 1
      transport/internet/websocket/hub.go

+ 2 - 1
transport/internet/tcp/hub.go

@@ -38,7 +38,8 @@ func ListenTCP(ctx context.Context, address net.Address, port net.Port, streamSe
 		if streamSettings.SocketSettings == nil {
 		if streamSettings.SocketSettings == nil {
 			streamSettings.SocketSettings = &internet.SocketConfig{}
 			streamSettings.SocketSettings = &internet.SocketConfig{}
 		}
 		}
-		streamSettings.SocketSettings.AcceptProxyProtocol = l.config.AcceptProxyProtocol
+		streamSettings.SocketSettings.AcceptProxyProtocol =
+			l.config.AcceptProxyProtocol || streamSettings.SocketSettings.AcceptProxyProtocol
 	}
 	}
 	var listener net.Listener
 	var listener net.Listener
 	var err error
 	var err error

+ 2 - 1
transport/internet/websocket/hub.go

@@ -89,7 +89,8 @@ func ListenWS(ctx context.Context, address net.Address, port net.Port, streamSet
 		if streamSettings.SocketSettings == nil {
 		if streamSettings.SocketSettings == nil {
 			streamSettings.SocketSettings = &internet.SocketConfig{}
 			streamSettings.SocketSettings = &internet.SocketConfig{}
 		}
 		}
-		streamSettings.SocketSettings.AcceptProxyProtocol = l.config.AcceptProxyProtocol
+		streamSettings.SocketSettings.AcceptProxyProtocol =
+			l.config.AcceptProxyProtocol || streamSettings.SocketSettings.AcceptProxyProtocol
 	}
 	}
 	var listener net.Listener
 	var listener net.Listener
 	var err error
 	var err error