forks/go-ethereum
1
0
Fork 1
mirror of https://github.com/ethereum/go-ethereum.git synced 2026-08-19 18:32:23 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

lightnodes should be omitted from the addressbook

Browse source
This commit is contained in:
Mark Vujevits 2018-10-19 11:20:26 +02:00
parent 3fae03a6b7
commit 608ac82f53
2 changed files with 40 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

20
swarm/network/kademlia.go
View file

@ -261,7 +261,7 @@ func (k *Kademlia) On(p *Peer) (uint8, bool) {
// found among live peers, do nothing
return v
})
if ins {
if ins && !p.BzzPeer.LightNode {
a := newEntry(p.BzzAddr)
a.conn = p
// insert new online peer into addrs
@ -329,14 +329,18 @@ func (k *Kademlia) Off(p *Peer) {
k.lock.Lock()
defer k.lock.Unlock()
var del bool
k.addrs, _, _, _ = pot.Swap(k.addrs, p, pof, func(v pot.Val) pot.Val {
// v cannot be nil, must check otherwise we overwrite entry
if v == nil {
panic(fmt.Sprintf("connected peer not found %v", p))
}
if !p.BzzPeer.LightNode {
k.addrs, _, _, _ = pot.Swap(k.addrs, p, pof, func(v pot.Val) pot.Val {
// v cannot be nil, must check otherwise we overwrite entry
if v == nil {
panic(fmt.Sprintf("connected peer not found %v", p))
}
del = true
return newEntry(p.BzzAddr)
})
} else {
del = true
return newEntry(p.BzzAddr)
})
}
if del {
k.conns, _, _, _ = pot.Swap(k.conns, p, pof, func(_ pot.Val) pot.Val {

32
swarm/network/kademlia_test.go
View file

@ -46,19 +46,19 @@ func newTestKademlia(b string) *Kademlia {
return NewKademlia(base, params)
}
func newTestKadPeer(k *Kademlia, s string) *Peer {
return NewPeer(&BzzPeer{BzzAddr: testKadPeerAddr(s)}, k)
func newTestKadPeer(k *Kademlia, s string, lightNode bool) *Peer {
return NewPeer(&BzzPeer{BzzAddr: testKadPeerAddr(s), LightNode: lightNode}, k)
}
func On(k *Kademlia, ons ...string) {
for _, s := range ons {
k.On(newTestKadPeer(k, s))
k.On(newTestKadPeer(k, s, false))
}
}
func Off(k *Kademlia, offs ...string) {
for _, s := range offs {
k.Off(newTestKadPeer(k, s))
k.Off(newTestKadPeer(k, s, false))
}
}
@ -254,6 +254,30 @@ func TestSuggestPeerFindPeers(t *testing.T) {
}
func TestOffEffectingAddressBookNormalNode(t *testing.T) {
k := newTestKademlia("00000000")
k.On(newTestKadPeer(k, "01000000", false))
if k.addrs.Size() != 1 {
t.Fatal("known peer addresses should contain 1 entry")
}
k.Off(newTestKadPeer(k, "01000000", false))
if k.addrs.Size() != 1 {
t.Fatal("known peer addresses should contain 1 entry")
}
}
func TestOffEffectingAddressBookLightNode(t *testing.T) {
k := newTestKademlia("00000000")
k.On(newTestKadPeer(k, "01000000", true))
if k.addrs.Size() != 0 {
t.Fatal("known peer addresses should contain 0 entry")
}
k.Off(newTestKadPeer(k, "01000000", true))
if k.addrs.Size() != 0 {
t.Fatal("known peer addresses should contain 0 entry")
}
}
func TestSuggestPeerRetries(t *testing.T) {
k := newTestKademlia("00000000")
k.RetryInterval = int64(300 * time.Millisecond) // cycle