fix: Resolve a NRE when a connection is closed early

SierraSoftworks · Aug 13, 2024 · b3dbde6 · b3dbde6
1 parent 9be4117
commit b3dbde6
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/proxy/socket.go b/proxy/socket.go
@@ -77,6 +77,10 @@ func (l *Listener) handleConnection(ctx context.Context, srv *tsnet.Server, conn
 	select {
 	case <-ctx.Done():
 	case <-close:
-		log.Debug().Str("remote", conn.RemoteAddr().String()).Msg("Connection closed")
+		if addr := remote.RemoteAddr(); addr != nil {
+			log.Debug().Str("remote", addr.String()).Msg("Connection closed")
+		} else {
+			log.Debug().Msg("Connection closed")
+		}
 	}
 }