fix: RawSrcAddr and RawSrcAddr in metadata

MetaCubeX · Sep 10, 2024 · 32caa6f · 32caa6f
1 parent f8557f5
commit 32caa6f
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 2 deletions.
diff --git a/adapter/inbound/http.go b/adapter/inbound/http.go
@@ -12,8 +12,8 @@ func NewHTTP(target socks5.Addr, srcConn net.Conn, conn net.Conn, additions ...A
 	metadata := parseSocksAddr(target)
 	metadata.NetWork = C.TCP
 	metadata.Type = C.HTTP
-	metadata.RawSrcAddr = srcConn.RemoteAddr()
-	metadata.RawDstAddr = srcConn.LocalAddr()
+	metadata.RawSrcAddr = srcConn.LocalAddr()
+	metadata.RawDstAddr = srcConn.RemoteAddr()
 	ApplyAdditions(metadata, WithSrcAddr(srcConn.RemoteAddr()), WithInAddr(srcConn.LocalAddr()))
 	ApplyAdditions(metadata, additions...)
 	return conn, metadata

diff --git a/adapter/inbound/https.go b/adapter/inbound/https.go
@@ -11,6 +11,8 @@ import (
 func NewHTTPS(request *http.Request, conn net.Conn, additions ...Addition) (net.Conn, *C.Metadata) {
 	metadata := parseHTTPAddr(request)
 	metadata.Type = C.HTTPS
+	metadata.RawSrcAddr = conn.LocalAddr()
+	metadata.RawDstAddr = conn.RemoteAddr()
 	ApplyAdditions(metadata, WithSrcAddr(conn.RemoteAddr()), WithInAddr(conn.LocalAddr()))
 	ApplyAdditions(metadata, additions...)
 	return conn, metadata

diff --git a/listener/sing/sing.go b/listener/sing/sing.go
@@ -129,6 +129,12 @@ func (h *ListenerHandler) NewConnection(ctx context.Context, conn net.Conn, meta
 		NetWork: C.TCP,
 		Type:    h.Type,
 	}
+	if metadata.Destination.IsIP() && metadata.Destination.Fqdn == "" {
+		cMetadata.RawSrcAddr = metadata.Destination.Unwrap().TCPAddr()
+	}
+	if metadata.Source.IsIP() && metadata.Source.Fqdn == "" {
+		cMetadata.RawDstAddr = metadata.Source.Unwrap().TCPAddr()
+	}
 	inbound.ApplyAdditions(cMetadata, inbound.WithDstAddr(metadata.Destination), inbound.WithSrcAddr(metadata.Source), inbound.WithInAddr(conn.LocalAddr()))
 	inbound.ApplyAdditions(cMetadata, getAdditions(ctx)...)
 	inbound.ApplyAdditions(cMetadata, h.Additions...)
@@ -185,6 +191,12 @@ func (h *ListenerHandler) NewPacketConnection(ctx context.Context, conn network.
 			NetWork: C.UDP,
 			Type:    h.Type,
 		}
+		if metadata.Source.IsIP() && metadata.Source.Fqdn == "" {
+			cMetadata.RawSrcAddr = metadata.Source.Unwrap().UDPAddr()
+		}
+		if dest.IsIP() && dest.Fqdn == "" {
+			cMetadata.RawDstAddr = dest.Unwrap().UDPAddr()
+		}
 		inbound.ApplyAdditions(cMetadata, inbound.WithDstAddr(dest), inbound.WithSrcAddr(metadata.Source), inbound.WithInAddr(conn.LocalAddr()))
 		inbound.ApplyAdditions(cMetadata, getAdditions(ctx)...)
 		inbound.ApplyAdditions(cMetadata, h.Additions...)