fix panic

Signed-off-by: He Xian <[email protected]>
hexian000 · Sep 17, 2024 · 17ba4e6 · 17ba4e6
1 parent 0eb60be
commit 17ba4e6
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/v2/handler.go b/v2/handler.go
@@ -115,6 +115,7 @@ func (h *ForwardHandler) Serve(ctx context.Context, accepted net.Conn) {
 
 // TunnelHandler forwards connections over the tunnel
 type TunnelHandler struct {
+	l net.Listener
 	s *Server
 	t *Tunnel
 }
@@ -136,7 +137,7 @@ func (h *TunnelHandler) Serve(ctx context.Context, accepted net.Conn) {
 		ioClose(dialed)
 		return
 	}
-	slog.Debugf("%v -> %q: forward established", h.t.l.Addr(), h.t.name)
+	slog.Debugf("%v -> %q: forward established", h.l.Addr(), h.t.name)
 }
 
 // EmptyHandler rejects all connections

diff --git a/v2/tunnel.go b/v2/tunnel.go
@@ -69,7 +69,7 @@ func (t *Tunnel) Start() error {
 			return err
 		}
 		slog.Noticef("forward listen: %v", l.Addr())
-		h := &TunnelHandler{s: t.s, t: t}
+		h := &TunnelHandler{l: l, s: t.s, t: t}
 		if err := t.s.g.Go(func() {
 			t.s.Serve(l, h)
 		}); err != nil {