diff --git a/handler/forward/local/handler.go b/handler/forward/local/handler.go index 3e4c527..e8d761c 100644 --- a/handler/forward/local/handler.go +++ b/handler/forward/local/handler.go @@ -91,7 +91,7 @@ func (h *forwardHandler) Handle(ctx context.Context, conn net.Conn, opts ...hand network = "udp" } - var rw io.ReadWriter + var rw io.ReadWriter = conn var host string if h.md.sniffing { if network == "tcp" { diff --git a/handler/forward/remote/handler.go b/handler/forward/remote/handler.go index db35c5a..071e1e4 100644 --- a/handler/forward/remote/handler.go +++ b/handler/forward/remote/handler.go @@ -82,7 +82,7 @@ func (h *forwardHandler) Handle(ctx context.Context, conn net.Conn, opts ...hand network = "udp" } - var rw io.ReadWriter + var rw io.ReadWriter = conn var host string if h.md.sniffing { if network == "tcp" {