From 58eb377ac4cedbb5c0424ce2e1e76bd1cd528f89 Mon Sep 17 00:00:00 2001
From: ginuerzh <ginuerzh@gmail.com>
Date: Wed, 16 Nov 2022 14:54:05 +0800
Subject: [PATCH] fix panic in port forwarding

---
 handler/forward/local/handler.go  | 2 +-
 handler/forward/remote/handler.go | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

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" {