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.
97 lines
2.5 KiB
97 lines
2.5 KiB
package statesync
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
|
|
"github.com/gogo/protobuf/proto"
|
|
|
|
ssproto "github.com/tendermint/tendermint/proto/tendermint/statesync"
|
|
)
|
|
|
|
const (
|
|
// snapshotMsgSize is the maximum size of a snapshotResponseMessage
|
|
snapshotMsgSize = int(4e6)
|
|
// chunkMsgSize is the maximum size of a chunkResponseMessage
|
|
chunkMsgSize = int(16e6)
|
|
)
|
|
|
|
// mustEncodeMsg encodes a Protobuf message, panicing on error.
|
|
func mustEncodeMsg(pb proto.Message) []byte {
|
|
msg := ssproto.Message{}
|
|
switch pb := pb.(type) {
|
|
case *ssproto.ChunkRequest:
|
|
msg.Sum = &ssproto.Message_ChunkRequest{ChunkRequest: pb}
|
|
case *ssproto.ChunkResponse:
|
|
msg.Sum = &ssproto.Message_ChunkResponse{ChunkResponse: pb}
|
|
case *ssproto.SnapshotsRequest:
|
|
msg.Sum = &ssproto.Message_SnapshotsRequest{SnapshotsRequest: pb}
|
|
case *ssproto.SnapshotsResponse:
|
|
msg.Sum = &ssproto.Message_SnapshotsResponse{SnapshotsResponse: pb}
|
|
default:
|
|
panic(fmt.Errorf("unknown message type %T", pb))
|
|
}
|
|
bz, err := proto.Marshal(&msg)
|
|
if err != nil {
|
|
panic(fmt.Errorf("unable to marshal %T: %w", pb, err))
|
|
}
|
|
return bz
|
|
}
|
|
|
|
// decodeMsg decodes a Protobuf message.
|
|
func decodeMsg(bz []byte) (proto.Message, error) {
|
|
pb := &ssproto.Message{}
|
|
err := proto.Unmarshal(bz, pb)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
switch msg := pb.Sum.(type) {
|
|
case *ssproto.Message_ChunkRequest:
|
|
return msg.ChunkRequest, nil
|
|
case *ssproto.Message_ChunkResponse:
|
|
return msg.ChunkResponse, nil
|
|
case *ssproto.Message_SnapshotsRequest:
|
|
return msg.SnapshotsRequest, nil
|
|
case *ssproto.Message_SnapshotsResponse:
|
|
return msg.SnapshotsResponse, nil
|
|
default:
|
|
return nil, fmt.Errorf("unknown message type %T", msg)
|
|
}
|
|
}
|
|
|
|
// validateMsg validates a message.
|
|
func validateMsg(pb proto.Message) error {
|
|
if pb == nil {
|
|
return errors.New("message cannot be nil")
|
|
}
|
|
switch msg := pb.(type) {
|
|
case *ssproto.ChunkRequest:
|
|
if msg.Height == 0 {
|
|
return errors.New("height cannot be 0")
|
|
}
|
|
case *ssproto.ChunkResponse:
|
|
if msg.Height == 0 {
|
|
return errors.New("height cannot be 0")
|
|
}
|
|
if msg.Missing && len(msg.Chunk) > 0 {
|
|
return errors.New("missing chunk cannot have contents")
|
|
}
|
|
if !msg.Missing && msg.Chunk == nil {
|
|
return errors.New("chunk cannot be nil")
|
|
}
|
|
case *ssproto.SnapshotsRequest:
|
|
case *ssproto.SnapshotsResponse:
|
|
if msg.Height == 0 {
|
|
return errors.New("height cannot be 0")
|
|
}
|
|
if len(msg.Hash) == 0 {
|
|
return errors.New("snapshot has no hash")
|
|
}
|
|
if msg.Chunks == 0 {
|
|
return errors.New("snapshot has no chunks")
|
|
}
|
|
default:
|
|
return fmt.Errorf("unknown message type %T", msg)
|
|
}
|
|
return nil
|
|
}
|