From 33e3713fbf579d3612a097f313beb8068ad8c422 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E4=B8=96=E7=95=8C?= <i@sekai.icu>
Date: Thu, 18 Aug 2022 16:16:33 +0800
Subject: [PATCH] Fix socks udp return error

---
 protocol/socks/handshake.go | 1 +
 1 file changed, 1 insertion(+)

diff --git a/protocol/socks/handshake.go b/protocol/socks/handshake.go
index 81869fb..5d98b8d 100644
--- a/protocol/socks/handshake.go
+++ b/protocol/socks/handshake.go
@@ -216,6 +216,7 @@ func HandleConnection0(ctx context.Context, conn net.Conn, version byte, authent
 				close(done)
 			}()
 			err = common.Error(io.Copy(io.Discard, conn))
+			<-done
 			return E.Errors(innerError, err)
 		default:
 			err = socks5.WriteResponse(conn, socks5.Response{