From 417d709d605a070d3e59cb7cb04e763eccffa372 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/https.go |  2 ++
 listener/sing/sing.go    | 12 ++++++++++++
 2 files changed, 14 insertions(+)

diff --git a/adapter/inbound/https.go b/adapter/inbound/https.go
index 55f6731a4..24b30804b 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.RemoteAddr()
+	metadata.RawDstAddr = conn.LocalAddr()
 	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 10390e732..0d4bb9265 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.Source.IsIP() && metadata.Source.Fqdn == "" {
+		cMetadata.RawSrcAddr = metadata.Source.Unwrap().TCPAddr()
+	}
+	if metadata.Destination.IsIP() && metadata.Destination.Fqdn == "" {
+		cMetadata.RawDstAddr = metadata.Destination.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...)