p2p: handle ip4/6 quic addresses

ethereum · Aug 14, 2024 · 299eff3 · 299eff3
1 parent f3ab1b3
commit 299eff3
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/p2p/enode/node.go b/p2p/enode/node.go
@@ -202,12 +202,16 @@ 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 if n.ip.Is6() {
+		n.Load((*enr.QUIC6)(&quic))
+	}
 	if !n.ip.IsValid() || n.ip.IsUnspecified() || quic == 0 {
 		return netip.AddrPort{}, false
 	}
-	return netip.AddrPortFrom(n.ip, uint16(quic)), true
+	return netip.AddrPortFrom(n.ip, quic), true
 }
 
 // Pubkey returns the secp256k1 public key of the node, if present.