From 5dd0fe2f5380538733661fb5926c07d4e9f45546 Mon Sep 17 00:00:00 2001
From: Delweng <delweng@gmail.com>
Date: Wed, 29 Oct 2025 17:34:19 +0800
Subject: [PATCH] p2p: cleanup v4 if v5 failed (#33005)

Clean the previous resource (v4) if the latter (v5) failed.
---
 p2p/server.go      |  5 +++++
 p2p/server_test.go | 27 +++++++++++++++++++++++++++
 2 files changed, 32 insertions(+)

diff --git a/p2p/server.go b/p2p/server.go
index ddd4f5d072..10c855f1c4 100644
--- a/p2p/server.go
+++ b/p2p/server.go
@@ -490,6 +490,11 @@ func (srv *Server) setupDiscovery() error {
 		}
 		srv.discv5, err = discover.ListenV5(sconn, srv.localnode, cfg)
 		if err != nil {
+			// Clean up v4 if v5 setup fails.
+			if srv.discv4 != nil {
+				srv.discv4.Close()
+				srv.discv4 = nil
+			}
 			return err
 		}
 	}
diff --git a/p2p/server_test.go b/p2p/server_test.go
index d42926cf4c..7bc7379099 100644
--- a/p2p/server_test.go
+++ b/p2p/server_test.go
@@ -579,6 +579,33 @@ func TestServerInboundThrottle(t *testing.T) {
 	}
 }
 
+func TestServerDiscoveryV5FailureRollsBackV4(t *testing.T) {
+	badBootstrap := enode.NewV4(&newkey().PublicKey, net.ParseIP("127.0.0.1"), 30303, 0) // invalid V5 of a V4 node
+	srv := &Server{
+		Config: Config{
+			PrivateKey:       newkey(),
+			ListenAddr:       "",
+			DiscAddr:         "127.0.0.1:0",
+			MaxPeers:         5,
+			DiscoveryV4:      true,
+			DiscoveryV5:      true,
+			BootstrapNodesV5: []*enode.Node{badBootstrap},
+			Logger:           testlog.Logger(t, log.LvlTrace),
+		},
+	}
+	err := srv.Start()
+	if err == nil {
+		t.Fatal("expected discovery v5 startup failure")
+	}
+	if !strings.Contains(err.Error(), "bad bootstrap node") {
+		t.Fatalf("unexpected error: %v", err)
+	}
+	if srv.DiscoveryV4() != nil {
+		t.Fatal("discovery v4 not cleaned after failure")
+	}
+	srv.Stop()
+}
+
 func listenFakeAddr(network, laddr string, remoteAddr net.Addr) (net.Listener, error) {
 	l, err := net.Listen(network, laddr)
 	if err == nil {