From 32caa6f57015879077adc0fd725d2a5ddb873bf0 Mon Sep 17 00:00:00 2001
From: wwqgtxx <wwqgtxx@gmail.com>
Date: Tue, 10 Sep 2024 20:57:28 +0800
Subject: [PATCH] fix: RawSrcAddr and RawSrcAddr in metadata

---
 adapter/inbound/http.go  |  4 ++--
 adapter/inbound/https.go |  2 ++
 listener/sing/sing.go    | 12 ++++++++++++
 3 files changed, 16 insertions(+), 2 deletions(-)

diff --git a/adapter/inbound/http.go b/adapter/inbound/http.go
index f7d45399fb..148f4e007c 100644
--- 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
index 55f6731a43..f38265c1c4 100644
--- 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
index 10390e7326..6bae70939a 100644
--- 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...)