fix: 调整获取远程目的的位置

2022-05-28 20:00:02 +08:00
parent d0268bb9a2
commit 067c02aba1
2 changed files with 8 additions and 11 deletions
--- a/tunnel/statistic/tracker.go
+++ b/tunnel/statistic/tracker.go
@ -59,7 +59,13 @@ func (tt *tcpTracker) Close() error {

 func NewTCPTracker(conn C.Conn, manager *Manager, metadata *C.Metadata, rule C.Rule) *tcpTracker {
 	uuid, _ := uuid.NewV4()
-	metadata.RemoteDst = conn.RemoteDestination()
+	if conn != nil {
+		if tcpAddr, ok := conn.RemoteAddr().(*net.TCPAddr); ok {
+			metadata.RemoteDst = tcpAddr.IP.String()
+		} else {
+			metadata.RemoteDst = conn.RemoteDestination()
+		}
+	}

 	t := &tcpTracker{
 		Conn:    conn,