neo-go/pkg/consensus/recovery_message.go

306 lines
8.6 KiB
Go
Raw Normal View History

2019-11-08 15:40:21 +00:00
package consensus
import (
"errors"
"github.com/nspcc-dev/dbft/crypto"
2019-11-15 10:32:40 +00:00
"github.com/nspcc-dev/dbft/payload"
"github.com/nspcc-dev/neo-go/pkg/io"
"github.com/nspcc-dev/neo-go/pkg/util"
2019-11-08 15:40:21 +00:00
)
type (
// recoveryMessage represents dBFT Recovery message.
recoveryMessage struct {
2019-11-15 10:32:40 +00:00
preparationHash *util.Uint256
preparationPayloads []*preparationCompact
commitPayloads []*commitCompact
changeViewPayloads []*changeViewCompact
stateRootEnabled bool
prepareRequest *message
2019-11-08 15:40:21 +00:00
}
changeViewCompact struct {
ValidatorIndex uint8
2019-11-08 15:40:21 +00:00
OriginalViewNumber byte
Timestamp uint64
2019-11-08 15:40:21 +00:00
InvocationScript []byte
}
commitCompact struct {
ViewNumber byte
ValidatorIndex uint8
2019-11-15 10:32:40 +00:00
Signature [signatureSize]byte
2019-11-08 15:40:21 +00:00
InvocationScript []byte
}
preparationCompact struct {
ValidatorIndex uint8
2019-11-08 15:40:21 +00:00
InvocationScript []byte
}
)
2019-11-15 10:32:40 +00:00
var _ payload.RecoveryMessage = (*recoveryMessage)(nil)
2019-11-08 15:40:21 +00:00
// DecodeBinary implements io.Serializable interface.
func (m *recoveryMessage) DecodeBinary(r *io.BinReader) {
2019-11-15 10:32:40 +00:00
r.ReadArray(&m.changeViewPayloads)
2019-11-08 15:40:21 +00:00
var hasReq = r.ReadBool()
2019-11-08 15:40:21 +00:00
if hasReq {
m.prepareRequest = &message{stateRootEnabled: m.stateRootEnabled}
2019-11-15 10:32:40 +00:00
m.prepareRequest.DecodeBinary(r)
if r.Err == nil && m.prepareRequest.Type != prepareRequestType {
r.Err = errors.New("recovery message PrepareRequest has wrong type")
return
}
2019-11-08 15:40:21 +00:00
} else {
l := r.ReadVarUint()
if l != 0 {
2019-11-15 10:32:40 +00:00
if l == util.Uint256Size {
m.preparationHash = new(util.Uint256)
2019-12-06 15:37:46 +00:00
r.ReadBytes(m.preparationHash[:])
2019-11-08 15:40:21 +00:00
} else {
r.Err = errors.New("invalid data")
}
} else {
2019-11-15 10:32:40 +00:00
m.preparationHash = nil
2019-11-08 15:40:21 +00:00
}
}
2019-11-15 10:32:40 +00:00
r.ReadArray(&m.preparationPayloads)
r.ReadArray(&m.commitPayloads)
2019-11-08 15:40:21 +00:00
}
// EncodeBinary implements io.Serializable interface.
func (m *recoveryMessage) EncodeBinary(w *io.BinWriter) {
2019-11-15 10:32:40 +00:00
w.WriteArray(m.changeViewPayloads)
2019-11-08 15:40:21 +00:00
2019-11-15 10:32:40 +00:00
hasReq := m.prepareRequest != nil
w.WriteBool(hasReq)
2019-11-08 15:40:21 +00:00
if hasReq {
2019-11-15 10:32:40 +00:00
m.prepareRequest.EncodeBinary(w)
2019-11-08 15:40:21 +00:00
} else {
2019-11-15 10:32:40 +00:00
if m.preparationHash == nil {
2019-11-08 15:40:21 +00:00
w.WriteVarUint(0)
} else {
2019-11-15 10:32:40 +00:00
w.WriteVarUint(util.Uint256Size)
w.WriteBytes(m.preparationHash[:])
2019-11-08 15:40:21 +00:00
}
}
2019-11-15 10:32:40 +00:00
w.WriteArray(m.preparationPayloads)
w.WriteArray(m.commitPayloads)
2019-11-08 15:40:21 +00:00
}
// DecodeBinary implements io.Serializable interface.
func (p *changeViewCompact) DecodeBinary(r *io.BinReader) {
p.ValidatorIndex = r.ReadB()
p.OriginalViewNumber = r.ReadB()
p.Timestamp = r.ReadU64LE()
p.InvocationScript = r.ReadVarBytes(1024)
2019-11-08 15:40:21 +00:00
}
// EncodeBinary implements io.Serializable interface.
func (p *changeViewCompact) EncodeBinary(w *io.BinWriter) {
w.WriteB(p.ValidatorIndex)
w.WriteB(p.OriginalViewNumber)
w.WriteU64LE(p.Timestamp)
w.WriteVarBytes(p.InvocationScript)
2019-11-08 15:40:21 +00:00
}
// DecodeBinary implements io.Serializable interface.
func (p *commitCompact) DecodeBinary(r *io.BinReader) {
p.ViewNumber = r.ReadB()
p.ValidatorIndex = r.ReadB()
2019-12-06 15:37:46 +00:00
r.ReadBytes(p.Signature[:])
p.InvocationScript = r.ReadVarBytes(1024)
2019-11-08 15:40:21 +00:00
}
// EncodeBinary implements io.Serializable interface.
func (p *commitCompact) EncodeBinary(w *io.BinWriter) {
w.WriteB(p.ViewNumber)
w.WriteB(p.ValidatorIndex)
w.WriteBytes(p.Signature[:])
w.WriteVarBytes(p.InvocationScript)
2019-11-08 15:40:21 +00:00
}
// DecodeBinary implements io.Serializable interface.
func (p *preparationCompact) DecodeBinary(r *io.BinReader) {
p.ValidatorIndex = r.ReadB()
p.InvocationScript = r.ReadVarBytes(1024)
2019-11-08 15:40:21 +00:00
}
// EncodeBinary implements io.Serializable interface.
func (p *preparationCompact) EncodeBinary(w *io.BinWriter) {
w.WriteB(p.ValidatorIndex)
w.WriteVarBytes(p.InvocationScript)
2019-11-08 15:40:21 +00:00
}
2019-11-15 10:32:40 +00:00
// AddPayload implements payload.RecoveryMessage interface.
func (m *recoveryMessage) AddPayload(p payload.ConsensusPayload) {
validator := uint8(p.ValidatorIndex())
2019-11-15 10:32:40 +00:00
switch p.Type() {
case payload.PrepareRequestType:
m.prepareRequest = &message{
Type: prepareRequestType,
ViewNumber: p.ViewNumber(),
payload: p.GetPrepareRequest().(*prepareRequest),
stateRootEnabled: m.stateRootEnabled,
}
h := p.Hash()
m.preparationHash = &h
m.preparationPayloads = append(m.preparationPayloads, &preparationCompact{
ValidatorIndex: validator,
InvocationScript: p.(*Payload).Witness.InvocationScript,
})
2019-11-15 10:32:40 +00:00
case payload.PrepareResponseType:
m.preparationPayloads = append(m.preparationPayloads, &preparationCompact{
ValidatorIndex: validator,
2019-11-15 10:32:40 +00:00
InvocationScript: p.(*Payload).Witness.InvocationScript,
})
if m.preparationHash == nil {
h := p.GetPrepareResponse().PreparationHash()
m.preparationHash = &h
}
2019-11-15 10:32:40 +00:00
case payload.ChangeViewType:
m.changeViewPayloads = append(m.changeViewPayloads, &changeViewCompact{
ValidatorIndex: validator,
2019-11-15 10:32:40 +00:00
OriginalViewNumber: p.ViewNumber(),
Timestamp: p.GetChangeView().Timestamp() / nsInMs,
2019-11-15 10:32:40 +00:00
InvocationScript: p.(*Payload).Witness.InvocationScript,
})
case payload.CommitType:
m.commitPayloads = append(m.commitPayloads, &commitCompact{
ValidatorIndex: validator,
2019-11-15 10:32:40 +00:00
ViewNumber: p.ViewNumber(),
Signature: p.GetCommit().(*commit).signature,
InvocationScript: p.(*Payload).Witness.InvocationScript,
})
}
}
// GetPrepareRequest implements payload.RecoveryMessage interface.
func (m *recoveryMessage) GetPrepareRequest(p payload.ConsensusPayload, validators []crypto.PublicKey, primary uint16) payload.ConsensusPayload {
2019-11-15 10:32:40 +00:00
if m.prepareRequest == nil {
return nil
}
var compact *preparationCompact
for _, p := range m.preparationPayloads {
if p != nil && p.ValidatorIndex == uint8(primary) {
compact = p
break
}
}
if compact == nil {
return nil
}
req := fromPayload(prepareRequestType, p.(*Payload), m.prepareRequest.payload)
req.SetValidatorIndex(primary)
req.Witness.InvocationScript = compact.InvocationScript
req.Witness.VerificationScript = getVerificationScript(uint8(primary), validators)
return req
2019-11-15 10:32:40 +00:00
}
// GetPrepareResponses implements payload.RecoveryMessage interface.
func (m *recoveryMessage) GetPrepareResponses(p payload.ConsensusPayload, validators []crypto.PublicKey) []payload.ConsensusPayload {
2019-11-15 10:32:40 +00:00
if m.preparationHash == nil {
return nil
}
ps := make([]payload.ConsensusPayload, len(m.preparationPayloads))
for i, resp := range m.preparationPayloads {
r := fromPayload(prepareResponseType, p.(*Payload), &prepareResponse{
2019-11-15 10:32:40 +00:00
preparationHash: *m.preparationHash,
})
r.SetValidatorIndex(uint16(resp.ValidatorIndex))
r.Witness.InvocationScript = resp.InvocationScript
r.Witness.VerificationScript = getVerificationScript(resp.ValidatorIndex, validators)
ps[i] = r
2019-11-15 10:32:40 +00:00
}
return ps
}
// GetChangeViews implements payload.RecoveryMessage interface.
func (m *recoveryMessage) GetChangeViews(p payload.ConsensusPayload, validators []crypto.PublicKey) []payload.ConsensusPayload {
2019-11-15 10:32:40 +00:00
ps := make([]payload.ConsensusPayload, len(m.changeViewPayloads))
for i, cv := range m.changeViewPayloads {
c := fromPayload(changeViewType, p.(*Payload), &changeView{
2019-11-15 10:32:40 +00:00
newViewNumber: cv.OriginalViewNumber + 1,
timestamp: cv.Timestamp,
})
c.message.ViewNumber = cv.OriginalViewNumber
c.SetValidatorIndex(uint16(cv.ValidatorIndex))
c.Witness.InvocationScript = cv.InvocationScript
c.Witness.VerificationScript = getVerificationScript(cv.ValidatorIndex, validators)
ps[i] = c
2019-11-15 10:32:40 +00:00
}
return ps
}
// GetCommits implements payload.RecoveryMessage interface.
func (m *recoveryMessage) GetCommits(p payload.ConsensusPayload, validators []crypto.PublicKey) []payload.ConsensusPayload {
2019-11-15 10:32:40 +00:00
ps := make([]payload.ConsensusPayload, len(m.commitPayloads))
for i, c := range m.commitPayloads {
cc := fromPayload(commitType, p.(*Payload), &commit{signature: c.Signature})
cc.SetValidatorIndex(uint16(c.ValidatorIndex))
cc.Witness.InvocationScript = c.InvocationScript
cc.Witness.VerificationScript = getVerificationScript(c.ValidatorIndex, validators)
ps[i] = cc
2019-11-15 10:32:40 +00:00
}
return ps
}
// PreparationHash implements payload.RecoveryMessage interface.
func (m *recoveryMessage) PreparationHash() *util.Uint256 {
return m.preparationHash
}
// SetPreparationHash implements payload.RecoveryMessage interface.
func (m *recoveryMessage) SetPreparationHash(h *util.Uint256) {
m.preparationHash = h
}
func getVerificationScript(i uint8, validators []crypto.PublicKey) []byte {
if int(i) >= len(validators) {
return nil
}
pub, ok := validators[i].(*publicKey)
if !ok {
return nil
}
return pub.GetVerificationScript()
}
2019-11-15 10:32:40 +00:00
func fromPayload(t messageType, recovery *Payload, p io.Serializable) *Payload {
return &Payload{
network: recovery.network,
message: &message{
Type: t,
ViewNumber: recovery.message.ViewNumber,
payload: p,
stateRootEnabled: recovery.stateRootEnabled,
2019-11-15 10:32:40 +00:00
},
version: recovery.Version(),
prevHash: recovery.PrevHash(),
height: recovery.Height(),
}
}