123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122 |
- // Copyright 2018 The go-ethereum Authors
- // This file is part of the go-ethereum library.
- //
- // The go-ethereum library is free software: you can redistribute it and/or modify
- // it under the terms of the GNU Lesser General Public License as published by
- // the Free Software Foundation, either version 3 of the License, or
- // (at your option) any later version.
- //
- // The go-ethereum library is distributed in the hope that it will be useful,
- // but WITHOUT ANY WARRANTY; without even the implied warranty of
- // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
- // GNU Lesser General Public License for more details.
- //
- // You should have received a copy of the GNU Lesser General Public License
- // along with the go-ethereum library. If not, see <http://www.gnu.org/licenses/>.
- package enode
- import (
- "math/rand"
- "net"
- "testing"
- "github.com/ethereum/go-ethereum/crypto"
- "github.com/ethereum/go-ethereum/p2p/enr"
- "github.com/stretchr/testify/assert"
- )
- func newLocalNodeForTesting() (*LocalNode, *DB) {
- db, _ := OpenDB("")
- key, _ := crypto.GenerateKey()
- return NewLocalNode(db, key), db
- }
- func TestLocalNode(t *testing.T) {
- ln, db := newLocalNodeForTesting()
- defer db.Close()
- if ln.Node().ID() != ln.ID() {
- t.Fatal("inconsistent ID")
- }
- ln.Set(enr.WithEntry("x", uint(3)))
- var x uint
- if err := ln.Node().Load(enr.WithEntry("x", &x)); err != nil {
- t.Fatal("can't load entry 'x':", err)
- } else if x != 3 {
- t.Fatal("wrong value for entry 'x':", x)
- }
- }
- func TestLocalNodeSeqPersist(t *testing.T) {
- ln, db := newLocalNodeForTesting()
- defer db.Close()
- if s := ln.Node().Seq(); s != 1 {
- t.Fatalf("wrong initial seq %d, want 1", s)
- }
- ln.Set(enr.WithEntry("x", uint(1)))
- if s := ln.Node().Seq(); s != 2 {
- t.Fatalf("wrong seq %d after set, want 2", s)
- }
- // Create a new instance, it should reload the sequence number.
- // The number increases just after that because a new record is
- // created without the "x" entry.
- ln2 := NewLocalNode(db, ln.key)
- if s := ln2.Node().Seq(); s != 3 {
- t.Fatalf("wrong seq %d on new instance, want 3", s)
- }
- // Create a new instance with a different node key on the same database.
- // This should reset the sequence number.
- key, _ := crypto.GenerateKey()
- ln3 := NewLocalNode(db, key)
- if s := ln3.Node().Seq(); s != 1 {
- t.Fatalf("wrong seq %d on instance with changed key, want 1", s)
- }
- }
- // This test checks behavior of the endpoint predictor.
- func TestLocalNodeEndpoint(t *testing.T) {
- var (
- fallback = &net.UDPAddr{IP: net.IP{127, 0, 0, 1}, Port: 80}
- predicted = &net.UDPAddr{IP: net.IP{127, 0, 1, 2}, Port: 81}
- staticIP = net.IP{127, 0, 1, 2}
- )
- ln, db := newLocalNodeForTesting()
- defer db.Close()
- // Nothing is set initially.
- assert.Equal(t, net.IP(nil), ln.Node().IP())
- assert.Equal(t, 0, ln.Node().UDP())
- assert.Equal(t, uint64(1), ln.Node().Seq())
- // Set up fallback address.
- ln.SetFallbackIP(fallback.IP)
- ln.SetFallbackUDP(fallback.Port)
- assert.Equal(t, fallback.IP, ln.Node().IP())
- assert.Equal(t, fallback.Port, ln.Node().UDP())
- assert.Equal(t, uint64(2), ln.Node().Seq())
- // Add endpoint statements from random hosts.
- for i := 0; i < iptrackMinStatements; i++ {
- assert.Equal(t, fallback.IP, ln.Node().IP())
- assert.Equal(t, fallback.Port, ln.Node().UDP())
- assert.Equal(t, uint64(2), ln.Node().Seq())
- from := &net.UDPAddr{IP: make(net.IP, 4), Port: 90}
- rand.Read(from.IP)
- ln.UDPEndpointStatement(from, predicted)
- }
- assert.Equal(t, predicted.IP, ln.Node().IP())
- assert.Equal(t, predicted.Port, ln.Node().UDP())
- assert.Equal(t, uint64(3), ln.Node().Seq())
- // Static IP overrides prediction.
- ln.SetStaticIP(staticIP)
- assert.Equal(t, staticIP, ln.Node().IP())
- assert.Equal(t, fallback.Port, ln.Node().UDP())
- assert.Equal(t, uint64(4), ln.Node().Seq())
- }
|