Skip to content

Commit b4fedb5

Browse files
committed
fix: upper bound check
1 parent 60d95ef commit b4fedb5

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

pkg/discv5/driver_dialer.go

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -193,15 +193,15 @@ func (d *DialDriver) monitorDatabase() {
193193
}
194194

195195
if comp, err := maddr.ValueForProtocol(ma.P_UDP); err == nil {
196-
if udp, err := strconv.Atoi(comp); err == nil {
196+
if udp, err := strconv.Atoi(comp); err == nil && udp <= math.MaxUint16 {
197197
if ip4 {
198198
r.Set(enr.UDP(uint16(udp)))
199199
} else {
200200
r.Set(enr.UDP6(uint16(udp)))
201201
}
202202
}
203203
} else if comp, err := maddr.ValueForProtocol(ma.P_TCP); err == nil {
204-
if tcp, err := strconv.Atoi(comp); err == nil {
204+
if tcp, err := strconv.Atoi(comp); err == nil && tcp <= math.MaxUint16 {
205205
if ip4 {
206206
r.Set(enr.TCP(uint16(tcp)))
207207
} else {

0 commit comments

Comments
 (0)