You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
137 lines
3.3 KiB
137 lines
3.3 KiB
package statesync
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
|
|
"github.com/gogo/protobuf/proto"
|
|
)
|
|
|
|
// Wrap implements the p2p Wrapper interface and wraps a state sync proto message.
|
|
func (m *Message) Wrap(pb proto.Message) error {
|
|
switch msg := pb.(type) {
|
|
case *ChunkRequest:
|
|
m.Sum = &Message_ChunkRequest{ChunkRequest: msg}
|
|
|
|
case *ChunkResponse:
|
|
m.Sum = &Message_ChunkResponse{ChunkResponse: msg}
|
|
|
|
case *SnapshotsRequest:
|
|
m.Sum = &Message_SnapshotsRequest{SnapshotsRequest: msg}
|
|
|
|
case *SnapshotsResponse:
|
|
m.Sum = &Message_SnapshotsResponse{SnapshotsResponse: msg}
|
|
|
|
case *LightBlockRequest:
|
|
m.Sum = &Message_LightBlockRequest{LightBlockRequest: msg}
|
|
|
|
case *LightBlockResponse:
|
|
m.Sum = &Message_LightBlockResponse{LightBlockResponse: msg}
|
|
|
|
case *ParamsRequest:
|
|
m.Sum = &Message_ParamsRequest{ParamsRequest: msg}
|
|
|
|
case *ParamsResponse:
|
|
m.Sum = &Message_ParamsResponse{ParamsResponse: msg}
|
|
|
|
default:
|
|
return fmt.Errorf("unknown message: %T", msg)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// Unwrap implements the p2p Wrapper interface and unwraps a wrapped state sync
|
|
// proto message.
|
|
func (m *Message) Unwrap() (proto.Message, error) {
|
|
switch msg := m.Sum.(type) {
|
|
case *Message_ChunkRequest:
|
|
return m.GetChunkRequest(), nil
|
|
|
|
case *Message_ChunkResponse:
|
|
return m.GetChunkResponse(), nil
|
|
|
|
case *Message_SnapshotsRequest:
|
|
return m.GetSnapshotsRequest(), nil
|
|
|
|
case *Message_SnapshotsResponse:
|
|
return m.GetSnapshotsResponse(), nil
|
|
|
|
case *Message_LightBlockRequest:
|
|
return m.GetLightBlockRequest(), nil
|
|
|
|
case *Message_LightBlockResponse:
|
|
return m.GetLightBlockResponse(), nil
|
|
|
|
case *Message_ParamsRequest:
|
|
return m.GetParamsRequest(), nil
|
|
|
|
case *Message_ParamsResponse:
|
|
return m.GetParamsResponse(), nil
|
|
|
|
default:
|
|
return nil, fmt.Errorf("unknown message: %T", msg)
|
|
}
|
|
}
|
|
|
|
// Validate validates the message returning an error upon failure.
|
|
func (m *Message) Validate() error {
|
|
if m == nil {
|
|
return errors.New("message cannot be nil")
|
|
}
|
|
|
|
switch msg := m.Sum.(type) {
|
|
case *Message_ChunkRequest:
|
|
if m.GetChunkRequest().Height == 0 {
|
|
return errors.New("height cannot be 0")
|
|
}
|
|
|
|
case *Message_ChunkResponse:
|
|
if m.GetChunkResponse().Height == 0 {
|
|
return errors.New("height cannot be 0")
|
|
}
|
|
if m.GetChunkResponse().Missing && len(m.GetChunkResponse().Chunk) > 0 {
|
|
return errors.New("missing chunk cannot have contents")
|
|
}
|
|
if !m.GetChunkResponse().Missing && m.GetChunkResponse().Chunk == nil {
|
|
return errors.New("chunk cannot be nil")
|
|
}
|
|
|
|
case *Message_SnapshotsRequest:
|
|
|
|
case *Message_SnapshotsResponse:
|
|
if m.GetSnapshotsResponse().Height == 0 {
|
|
return errors.New("height cannot be 0")
|
|
}
|
|
if len(m.GetSnapshotsResponse().Hash) == 0 {
|
|
return errors.New("snapshot has no hash")
|
|
}
|
|
if m.GetSnapshotsResponse().Chunks == 0 {
|
|
return errors.New("snapshot has no chunks")
|
|
}
|
|
|
|
case *Message_LightBlockRequest:
|
|
if m.GetLightBlockRequest().Height == 0 {
|
|
return errors.New("height cannot be 0")
|
|
}
|
|
|
|
// light block validation handled by the backfill process
|
|
case *Message_LightBlockResponse:
|
|
|
|
case *Message_ParamsRequest:
|
|
if m.GetParamsRequest().Height == 0 {
|
|
return errors.New("height cannot be 0")
|
|
}
|
|
|
|
case *Message_ParamsResponse:
|
|
resp := m.GetParamsResponse()
|
|
if resp.Height == 0 {
|
|
return errors.New("height cannot be 0")
|
|
}
|
|
|
|
default:
|
|
return fmt.Errorf("unknown message type: %T", msg)
|
|
}
|
|
|
|
return nil
|
|
}
|