proxy protocol: fix detect method for IPV4 and IPV6

6 years ago · fdc6902a90
parent d8d587fd93
commit fdc6902a90
1 changed files with 1 additions and 1 deletions
--- a/client/proxy/proxy.go
+++ b/client/proxy/proxy.go
@ -521,7 +521,7 @@ func HandleTcpWorkConnection(localInfo *config.LocalSvrConf, proxyPlugin plugin.
 				DestinationPort:    m.DstPort,
 			}

-			if h.SourceAddress.To16() == nil {
+			if strings.Contains(m.SrcAddr, ".") {
 				h.TransportProtocol = pp.TCPv4
 			} else {
 				h.TransportProtocol = pp.TCPv6