p2p: handle ip4/6 quic addresses

ethereum · Aug 13, 2024 · 2db0c5c · 2db0c5c
1 parent f3ab1b3
commit 2db0c5c
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/p2p/enode/node.go b/p2p/enode/node.go
@@ -202,8 +202,12 @@ func (n *Node) TCPEndpoint() (netip.AddrPort, bool) {
 
 // QUICEndpoint returns the announced QUIC endpoint.
 func (n *Node) QUICEndpoint() (netip.AddrPort, bool) {
-	var quic enr.QUIC
-	n.Load(&quic)
+	var quic uint16
+	if n.ip.Is4() || n.ip.Is4In6() {
+		n.Load((*enr.QUIC)(&quic))
+	} else {
+		n.Load((*enr.QUIC6)(&quic))
+	}
 	if !n.ip.IsValid() || n.ip.IsUnspecified() || quic == 0 {
 		return netip.AddrPort{}, false
 	}