From 1a7c32b4e2e73235d99fe674ec82b58d43b68668 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E4=B8=96=E7=95=8C?= <i@sekai.icu>
Date: Wed, 12 Apr 2023 20:31:10 +0800
Subject: [PATCH] Fix ReadFrom result

---
 none.go                     | 6 +++++-
 shadowaead/protocol.go      | 6 +++++-
 shadowaead_2022/protocol.go | 6 +++++-
 shadowstream/protocol.go    | 6 +++++-
 4 files changed, 20 insertions(+), 4 deletions(-)

diff --git a/none.go b/none.go
index c1f4433..50a7de4 100644
--- a/none.go
+++ b/none.go
@@ -155,7 +155,11 @@ func (c *nonePacketConn) ReadFrom(p []byte) (n int, addr net.Addr, err error) {
 	if err != nil {
 		return
 	}
-	addr = destination.UDPAddr()
+	if destination.IsFqdn() {
+		addr = destination
+	} else {
+		addr = destination.UDPAddr()
+	}
 	n = copy(p, buffer.Bytes())
 	return
 }
diff --git a/shadowaead/protocol.go b/shadowaead/protocol.go
index 0e8c712..32f71c0 100644
--- a/shadowaead/protocol.go
+++ b/shadowaead/protocol.go
@@ -296,7 +296,11 @@ func (c *clientPacketConn) ReadFrom(p []byte) (n int, addr net.Addr, err error)
 	if err != nil {
 		return
 	}
-	addr = destination.UDPAddr()
+	if destination.IsFqdn() {
+		addr = destination
+	} else {
+		addr = destination.UDPAddr()
+	}
 	n = copy(p, buffer.Bytes())
 	return
 }
diff --git a/shadowaead_2022/protocol.go b/shadowaead_2022/protocol.go
index ca12904..022f932 100644
--- a/shadowaead_2022/protocol.go
+++ b/shadowaead_2022/protocol.go
@@ -702,7 +702,11 @@ func (c *clientPacketConn) ReadFrom(p []byte) (n int, addr net.Addr, err error)
 	if err != nil {
 		return
 	}
-	addr = destination.UDPAddr()
+	if destination.IsFqdn() {
+		addr = destination
+	} else {
+		addr = destination.UDPAddr()
+	}
 	n = copy(p, buffer.Bytes())
 	return
 }
diff --git a/shadowstream/protocol.go b/shadowstream/protocol.go
index 10e26d7..0238690 100644
--- a/shadowstream/protocol.go
+++ b/shadowstream/protocol.go
@@ -294,7 +294,11 @@ func (c *clientPacketConn) ReadFrom(p []byte) (n int, addr net.Addr, err error)
 	if err != nil {
 		return
 	}
-	addr = destination.UDPAddr()
+	if destination.IsFqdn() {
+		addr = destination
+	} else {
+		addr = destination.UDPAddr()
+	}
 	n = copy(p, buffer.Bytes())
 	return
 }