mirror of
https://gitlab.com/pulsechaincom/erigon-pulse.git
synced 2024-12-25 13:07:17 +00:00
298 lines
6.8 KiB
Go
298 lines
6.8 KiB
Go
|
package les
|
||
|
|
||
|
import (
|
||
|
"math/big"
|
||
|
"testing"
|
||
|
|
||
|
"github.com/ethereum/go-ethereum/common"
|
||
|
"github.com/ethereum/go-ethereum/les/flowcontrol"
|
||
|
"github.com/ethereum/go-ethereum/p2p"
|
||
|
"github.com/ethereum/go-ethereum/p2p/enode"
|
||
|
"github.com/ethereum/go-ethereum/rlp"
|
||
|
)
|
||
|
|
||
|
const (
|
||
|
test_networkid = 10
|
||
|
protocol_version = 2123
|
||
|
)
|
||
|
|
||
|
var (
|
||
|
hash = common.HexToHash("some string")
|
||
|
genesis = common.HexToHash("genesis hash")
|
||
|
headNum = uint64(1234)
|
||
|
td = big.NewInt(123)
|
||
|
)
|
||
|
|
||
|
//ulc connects to trusted peer and send announceType=announceTypeSigned
|
||
|
func TestPeerHandshakeSetAnnounceTypeToAnnounceTypeSignedForTrustedPeer(t *testing.T) {
|
||
|
|
||
|
var id enode.ID = newNodeID(t).ID()
|
||
|
|
||
|
//peer to connect(on ulc side)
|
||
|
p := peer{
|
||
|
Peer: p2p.NewPeer(id, "test peer", []p2p.Cap{}),
|
||
|
version: protocol_version,
|
||
|
isTrusted: true,
|
||
|
rw: &rwStub{
|
||
|
WriteHook: func(recvList keyValueList) {
|
||
|
//checking that ulc sends to peer allowedRequests=onlyAnnounceRequests and announceType = announceTypeSigned
|
||
|
recv := recvList.decode()
|
||
|
var reqType uint64
|
||
|
|
||
|
err := recv.get("announceType", &reqType)
|
||
|
if err != nil {
|
||
|
t.Fatal(err)
|
||
|
}
|
||
|
|
||
|
if reqType != announceTypeSigned {
|
||
|
t.Fatal("Expected announceTypeSigned")
|
||
|
}
|
||
|
},
|
||
|
ReadHook: func(l keyValueList) keyValueList {
|
||
|
l = l.add("serveHeaders", nil)
|
||
|
l = l.add("serveChainSince", uint64(0))
|
||
|
l = l.add("serveStateSince", uint64(0))
|
||
|
l = l.add("txRelay", nil)
|
||
|
l = l.add("flowControl/BL", uint64(0))
|
||
|
l = l.add("flowControl/MRR", uint64(0))
|
||
|
l = l.add("flowControl/MRC", RequestCostList{})
|
||
|
|
||
|
return l
|
||
|
},
|
||
|
},
|
||
|
network: test_networkid,
|
||
|
}
|
||
|
|
||
|
err := p.Handshake(td, hash, headNum, genesis, nil)
|
||
|
if err != nil {
|
||
|
t.Fatalf("Handshake error: %s", err)
|
||
|
}
|
||
|
|
||
|
if p.announceType != announceTypeSigned {
|
||
|
t.Fatal("Incorrect announceType")
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func TestPeerHandshakeAnnounceTypeSignedForTrustedPeersPeerNotInTrusted(t *testing.T) {
|
||
|
var id enode.ID = newNodeID(t).ID()
|
||
|
p := peer{
|
||
|
Peer: p2p.NewPeer(id, "test peer", []p2p.Cap{}),
|
||
|
version: protocol_version,
|
||
|
rw: &rwStub{
|
||
|
WriteHook: func(recvList keyValueList) {
|
||
|
//checking that ulc sends to peer allowedRequests=noRequests and announceType != announceTypeSigned
|
||
|
recv := recvList.decode()
|
||
|
var reqType uint64
|
||
|
|
||
|
err := recv.get("announceType", &reqType)
|
||
|
if err != nil {
|
||
|
t.Fatal(err)
|
||
|
}
|
||
|
|
||
|
if reqType == announceTypeSigned {
|
||
|
t.Fatal("Expected not announceTypeSigned")
|
||
|
}
|
||
|
},
|
||
|
ReadHook: func(l keyValueList) keyValueList {
|
||
|
l = l.add("serveHeaders", nil)
|
||
|
l = l.add("serveChainSince", uint64(0))
|
||
|
l = l.add("serveStateSince", uint64(0))
|
||
|
l = l.add("txRelay", nil)
|
||
|
l = l.add("flowControl/BL", uint64(0))
|
||
|
l = l.add("flowControl/MRR", uint64(0))
|
||
|
l = l.add("flowControl/MRC", RequestCostList{})
|
||
|
|
||
|
return l
|
||
|
},
|
||
|
},
|
||
|
network: test_networkid,
|
||
|
}
|
||
|
|
||
|
err := p.Handshake(td, hash, headNum, genesis, nil)
|
||
|
if err != nil {
|
||
|
t.Fatal(err)
|
||
|
}
|
||
|
if p.announceType == announceTypeSigned {
|
||
|
t.Fatal("Incorrect announceType")
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func TestPeerHandshakeDefaultAllRequests(t *testing.T) {
|
||
|
var id enode.ID = newNodeID(t).ID()
|
||
|
|
||
|
s := generateLesServer()
|
||
|
|
||
|
p := peer{
|
||
|
Peer: p2p.NewPeer(id, "test peer", []p2p.Cap{}),
|
||
|
version: protocol_version,
|
||
|
rw: &rwStub{
|
||
|
ReadHook: func(l keyValueList) keyValueList {
|
||
|
l = l.add("announceType", uint64(announceTypeSigned))
|
||
|
l = l.add("allowedRequests", uint64(0))
|
||
|
|
||
|
return l
|
||
|
},
|
||
|
},
|
||
|
network: test_networkid,
|
||
|
}
|
||
|
|
||
|
err := p.Handshake(td, hash, headNum, genesis, s)
|
||
|
if err != nil {
|
||
|
t.Fatal(err)
|
||
|
}
|
||
|
|
||
|
if p.isOnlyAnnounce {
|
||
|
t.Fatal("Incorrect announceType")
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func TestPeerHandshakeServerSendOnlyAnnounceRequestsHeaders(t *testing.T) {
|
||
|
var id enode.ID = newNodeID(t).ID()
|
||
|
|
||
|
s := generateLesServer()
|
||
|
s.onlyAnnounce = true
|
||
|
|
||
|
p := peer{
|
||
|
Peer: p2p.NewPeer(id, "test peer", []p2p.Cap{}),
|
||
|
version: protocol_version,
|
||
|
rw: &rwStub{
|
||
|
ReadHook: func(l keyValueList) keyValueList {
|
||
|
l = l.add("announceType", uint64(announceTypeSigned))
|
||
|
|
||
|
return l
|
||
|
},
|
||
|
WriteHook: func(l keyValueList) {
|
||
|
for _, v := range l {
|
||
|
if v.Key == "serveHeaders" ||
|
||
|
v.Key == "serveChainSince" ||
|
||
|
v.Key == "serveStateSince" ||
|
||
|
v.Key == "txRelay" {
|
||
|
t.Fatalf("%v exists", v.Key)
|
||
|
}
|
||
|
}
|
||
|
},
|
||
|
},
|
||
|
network: test_networkid,
|
||
|
}
|
||
|
|
||
|
err := p.Handshake(td, hash, headNum, genesis, s)
|
||
|
if err != nil {
|
||
|
t.Fatal(err)
|
||
|
}
|
||
|
}
|
||
|
func TestPeerHandshakeClientReceiveOnlyAnnounceRequestsHeaders(t *testing.T) {
|
||
|
var id enode.ID = newNodeID(t).ID()
|
||
|
|
||
|
p := peer{
|
||
|
Peer: p2p.NewPeer(id, "test peer", []p2p.Cap{}),
|
||
|
version: protocol_version,
|
||
|
rw: &rwStub{
|
||
|
ReadHook: func(l keyValueList) keyValueList {
|
||
|
l = l.add("flowControl/BL", uint64(0))
|
||
|
l = l.add("flowControl/MRR", uint64(0))
|
||
|
l = l.add("flowControl/MRC", RequestCostList{})
|
||
|
|
||
|
l = l.add("announceType", uint64(announceTypeSigned))
|
||
|
|
||
|
return l
|
||
|
},
|
||
|
},
|
||
|
network: test_networkid,
|
||
|
isTrusted: true,
|
||
|
}
|
||
|
|
||
|
err := p.Handshake(td, hash, headNum, genesis, nil)
|
||
|
if err != nil {
|
||
|
t.Fatal(err)
|
||
|
}
|
||
|
|
||
|
if !p.isOnlyAnnounce {
|
||
|
t.Fatal("isOnlyAnnounce must be true")
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func TestPeerHandshakeClientReturnErrorOnUselessPeer(t *testing.T) {
|
||
|
var id enode.ID = newNodeID(t).ID()
|
||
|
|
||
|
p := peer{
|
||
|
Peer: p2p.NewPeer(id, "test peer", []p2p.Cap{}),
|
||
|
version: protocol_version,
|
||
|
rw: &rwStub{
|
||
|
ReadHook: func(l keyValueList) keyValueList {
|
||
|
l = l.add("flowControl/BL", uint64(0))
|
||
|
l = l.add("flowControl/MRR", uint64(0))
|
||
|
l = l.add("flowControl/MRC", RequestCostList{})
|
||
|
|
||
|
l = l.add("announceType", uint64(announceTypeSigned))
|
||
|
|
||
|
return l
|
||
|
},
|
||
|
},
|
||
|
network: test_networkid,
|
||
|
}
|
||
|
|
||
|
err := p.Handshake(td, hash, headNum, genesis, nil)
|
||
|
if err == nil {
|
||
|
t.FailNow()
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func generateLesServer() *LesServer {
|
||
|
s := &LesServer{
|
||
|
defParams: &flowcontrol.ServerParams{
|
||
|
BufLimit: uint64(300000000),
|
||
|
MinRecharge: uint64(50000),
|
||
|
},
|
||
|
fcManager: flowcontrol.NewClientManager(1, 2, 3),
|
||
|
fcCostStats: &requestCostStats{
|
||
|
stats: make(map[uint64]*linReg, len(reqList)),
|
||
|
},
|
||
|
}
|
||
|
for _, code := range reqList {
|
||
|
s.fcCostStats.stats[code] = &linReg{cnt: 100}
|
||
|
}
|
||
|
return s
|
||
|
}
|
||
|
|
||
|
type rwStub struct {
|
||
|
ReadHook func(l keyValueList) keyValueList
|
||
|
WriteHook func(l keyValueList)
|
||
|
}
|
||
|
|
||
|
func (s *rwStub) ReadMsg() (p2p.Msg, error) {
|
||
|
payload := keyValueList{}
|
||
|
payload = payload.add("protocolVersion", uint64(protocol_version))
|
||
|
payload = payload.add("networkId", uint64(test_networkid))
|
||
|
payload = payload.add("headTd", td)
|
||
|
payload = payload.add("headHash", hash)
|
||
|
payload = payload.add("headNum", headNum)
|
||
|
payload = payload.add("genesisHash", genesis)
|
||
|
|
||
|
if s.ReadHook != nil {
|
||
|
payload = s.ReadHook(payload)
|
||
|
}
|
||
|
|
||
|
size, p, err := rlp.EncodeToReader(payload)
|
||
|
if err != nil {
|
||
|
return p2p.Msg{}, err
|
||
|
}
|
||
|
|
||
|
return p2p.Msg{
|
||
|
Size: uint32(size),
|
||
|
Payload: p,
|
||
|
}, nil
|
||
|
}
|
||
|
|
||
|
func (s *rwStub) WriteMsg(m p2p.Msg) error {
|
||
|
recvList := keyValueList{}
|
||
|
if err := m.Decode(&recvList); err != nil {
|
||
|
return err
|
||
|
}
|
||
|
|
||
|
if s.WriteHook != nil {
|
||
|
s.WriteHook(recvList)
|
||
|
}
|
||
|
|
||
|
return nil
|
||
|
}
|