p2p/discover: only get the peer nodes in netrestrict config

hadv · hadv · commit 91018a442359 · 2022-07-15T21:59:57.000+07:00
diff --git a/p2p/discover/v5_udp.go b/p2p/discover/v5_udp.go
@@ -407,6 +407,9 @@ func (t *UDPv5) verifyResponseNode(c *callV5, r *enr.Record, distances []uint, s
 	if err := netutil.CheckRelayIP(c.node.IP(), node.IP()); err != nil {
 		return nil, err
 	}
+	if t.netrestrict != nil && !t.netrestrict.Contains(node.IP()) {
+		return nil, errors.New("not contained in netrestrict list")
+	}
 	if c.node.UDP() <= 1024 {
 		return nil, errLowPort
 	}

Original file line number	Diff line number	Diff line change
`@@ -407,6 +407,9 @@ func (t UDPv5) verifyResponseNode(c callV5, r *enr.Record, distances []uint, s`
`407`	`407`	`if err := netutil.CheckRelayIP(c.node.IP(), node.IP()); err != nil {`
`408`	`408`	`return nil, err`
`409`	`409`	`}`
	`410`	`+ if t.netrestrict != nil && !t.netrestrict.Contains(node.IP()) {`
	`411`	`+ return nil, errors.New("not contained in netrestrict list")`
	`412`	`+ }`
`410`	`413`	`if c.node.UDP() <= 1024 {`
`411`	`414`	`return nil, errLowPort`
`412`	`415`	`}`