From 7c900edd2db53fc45c2eca9511dc2951e6e611d6 Mon Sep 17 00:00:00 2001
From: Evgenii Stratonikov <evgeniy@nspcc.ru>
Date: Mon, 2 Dec 2019 10:51:45 +0300
Subject: [PATCH 1/3] network: count only handshaked peers as connected

---
 pkg/network/server.go | 19 ++++++++++++++++++-
 1 file changed, 18 insertions(+), 1 deletion(-)

diff --git a/pkg/network/server.go b/pkg/network/server.go
index 28e1c71a9..2b5ec376f 100644
--- a/pkg/network/server.go
+++ b/pkg/network/server.go
@@ -253,7 +253,7 @@ func (s *Server) tryStartConsensus() {
 		return
 	}
 
-	if s.PeerCount() >= s.MinPeers {
+	if s.HandshakedPeersCount() >= s.MinPeers {
 		log.Info("minimum amount of peers were connected to")
 		if s.connected.CAS(false, true) {
 			s.consensus.Start()
@@ -282,6 +282,23 @@ func (s *Server) PeerCount() int {
 	return len(s.peers)
 }
 
+// HandshakedPeersCount returns the number of connected peers
+// which have already performed handshake.
+func (s *Server) HandshakedPeersCount() int {
+	s.lock.RLock()
+	defer s.lock.RUnlock()
+
+	var count int
+
+	for p := range s.peers {
+		if p.Handshaked() {
+			count++
+		}
+	}
+
+	return count
+}
+
 // startProtocol starts a long running background loop that interacts
 // every ProtoTickInterval with the peer.
 func (s *Server) startProtocol(p Peer) {

From b649ffb88e42076d5c2968817cc0e843582799c1 Mon Sep 17 00:00:00 2001
From: Evgenii Stratonikov <evgeniy@nspcc.ru>
Date: Mon, 2 Dec 2019 16:18:54 +0300
Subject: [PATCH 2/3] consensus: set preparation hash in recovery message

---
 pkg/consensus/recovery_message.go | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/pkg/consensus/recovery_message.go b/pkg/consensus/recovery_message.go
index 8e0d62533..9ef2761b8 100644
--- a/pkg/consensus/recovery_message.go
+++ b/pkg/consensus/recovery_message.go
@@ -136,11 +136,18 @@ func (m *recoveryMessage) AddPayload(p payload.ConsensusPayload) {
 	switch p.Type() {
 	case payload.PrepareRequestType:
 		m.prepareRequest = p.GetPrepareRequest().(*prepareRequest)
+		h := p.Hash()
+		m.preparationHash = &h
 	case payload.PrepareResponseType:
 		m.preparationPayloads = append(m.preparationPayloads, &preparationCompact{
 			ValidatorIndex:   p.ValidatorIndex(),
 			InvocationScript: p.(*Payload).Witness.InvocationScript,
 		})
+
+		if m.preparationHash == nil {
+			h := p.GetPrepareResponse().PreparationHash()
+			m.preparationHash = &h
+		}
 	case payload.ChangeViewType:
 		m.changeViewPayloads = append(m.changeViewPayloads, &changeViewCompact{
 			ValidatorIndex:     p.ValidatorIndex(),

From 9ddebfdb8aa4dd835a06daf7e3cab1e45a736a73 Mon Sep 17 00:00:00 2001
From: Evgenii Stratonikov <evgeniy@nspcc.ru>
Date: Mon, 2 Dec 2019 17:00:17 +0300
Subject: [PATCH 3/3] consensus: add tests for recovery message

---
 pkg/consensus/recovery_message_test.go | 80 ++++++++++++++++++++++++++
 1 file changed, 80 insertions(+)
 create mode 100644 pkg/consensus/recovery_message_test.go

diff --git a/pkg/consensus/recovery_message_test.go b/pkg/consensus/recovery_message_test.go
new file mode 100644
index 000000000..438235d3d
--- /dev/null
+++ b/pkg/consensus/recovery_message_test.go
@@ -0,0 +1,80 @@
+package consensus
+
+import (
+	"testing"
+
+	"github.com/CityOfZion/neo-go/pkg/util"
+	"github.com/nspcc-dev/dbft/payload"
+	"github.com/stretchr/testify/require"
+)
+
+func TestRecoveryMessage_Setters(t *testing.T) {
+	r := &recoveryMessage{}
+	p := new(Payload)
+	p.SetType(payload.RecoveryMessageType)
+	p.SetPayload(r)
+
+	req := &prepareRequest{
+		timestamp:         87,
+		nonce:             321,
+		transactionHashes: []util.Uint256{{1}},
+		minerTx:           *newMinerTx(123),
+		nextConsensus:     util.Uint160{1, 2},
+	}
+	p1 := new(Payload)
+	p1.SetType(payload.PrepareRequestType)
+	p1.SetPayload(req)
+
+	t.Run("prepare response is added", func(t *testing.T) {
+		p2 := new(Payload)
+		p2.SetType(payload.PrepareResponseType)
+		p2.SetPayload(&prepareResponse{
+			preparationHash: p1.Hash(),
+		})
+
+		r.AddPayload(p2)
+		require.NotNil(t, r.PreparationHash())
+		require.Equal(t, p1.Hash(), *r.PreparationHash())
+
+		ps := r.GetPrepareResponses(p)
+		require.Len(t, ps, 1)
+		require.Equal(t, p2, ps[0])
+	})
+
+	t.Run("prepare request is added", func(t *testing.T) {
+		pr := r.GetPrepareRequest(p)
+		require.Nil(t, pr)
+
+		r.AddPayload(p1)
+		pr = r.GetPrepareRequest(p)
+		require.NotNil(t, pr)
+		require.Equal(t, p1, pr)
+	})
+
+	t.Run("change view is added", func(t *testing.T) {
+		p3 := new(Payload)
+		p3.SetType(payload.ChangeViewType)
+		p3.SetPayload(&changeView{
+			newViewNumber: 1,
+			timestamp:     12345,
+		})
+
+		r.AddPayload(p3)
+
+		ps := r.GetChangeViews(p)
+		require.Len(t, ps, 1)
+		require.Equal(t, p3, ps[0])
+	})
+
+	t.Run("commit is added", func(t *testing.T) {
+		p4 := new(Payload)
+		p4.SetType(payload.CommitType)
+		p4.SetPayload(randomMessage(t, commitType))
+
+		r.AddPayload(p4)
+
+		ps := r.GetCommits(p)
+		require.Len(t, ps, 1)
+		require.Equal(t, p4, ps[0])
+	})
+}