forks/go-ethereum
1
0
Fork 1
mirror of https://github.com/ethereum/go-ethereum.git synced 2026-08-11 14:33:52 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

make findnode public

Browse source 
Signed-off-by: Chen Kai <281165273grape@gmail.com>
This commit is contained in:
Chen Kai 2024-12-10 16:31:22 +08:00
parent fbfb4083e1
commit 8a63d44af2
2 changed files with 6 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
p2p/discover/v5_udp.go
View file

@ -374,7 +374,7 @@ func (t *UDPv5) lookupWorker(destNode *enode.Node, target enode.ID) ([]*enode.No
err error err error
) )
var r []*enode.Node var r []*enode.Node
r, err = t.findnode(destNode, dists) r, err = t.Findnode(destNode, dists)
if errors.Is(err, errClosed) { if errors.Is(err, errClosed) {
return nil, err return nil, err
} }
@ -429,7 +429,7 @@ func (t *UDPv5) PingWithResp(n *enode.Node) (*v5wire.Pong, error) {
// RequestENR requests n's record. // RequestENR requests n's record.
func (t *UDPv5) RequestENR(n *enode.Node) (*enode.Node, error) { func (t *UDPv5) RequestENR(n *enode.Node) (*enode.Node, error) {
nodes, err := t.findnode(n, []uint{0}) nodes, err := t.Findnode(n, []uint{0})
if err != nil { if err != nil {
return nil, err return nil, err
} }
@ -439,8 +439,8 @@ func (t *UDPv5) RequestENR(n *enode.Node) (*enode.Node, error) {
return nodes[0], nil return nodes[0], nil
} }
// findnode calls FINDNODE on a node and waits for responses. // Findnode calls FINDNODE on a node and waits for responses.
func (t *UDPv5) findnode(n *enode.Node, distances []uint) ([]*enode.Node, error) { func (t *UDPv5) Findnode(n *enode.Node, distances []uint) ([]*enode.Node, error) {
resp := t.callToNode(n, v5wire.NodesMsg, &v5wire.Findnode{Distances: distances}) resp := t.callToNode(n, v5wire.NodesMsg, &v5wire.Findnode{Distances: distances})
return t.waitForNodes(resp, distances) return t.waitForNodes(resp, distances)
} }

4
p2p/discover/v5_udp_test.go
View file

@ -288,7 +288,7 @@ func TestUDPv5_findnodeCall(t *testing.T) {
) )
go func() { go func() {
var err error var err error
response, err = test.udp.findnode(remote, distances) response, err = test.udp.Findnode(remote, distances)
done <- err done <- err
}() }()
@ -398,7 +398,7 @@ func TestUDPv5_callTimeoutReset(t *testing.T) {
done = make(chan error, 1) done = make(chan error, 1)
) )
go func() { go func() {
_, err := test.udp.findnode(remote, []uint{distance}) _, err := test.udp.Findnode(remote, []uint{distance})
done <- err done <- err
}() }()