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.
118 lines
3.0 KiB
118 lines
3.0 KiB
package blockchain
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
|
|
"github.com/gogo/protobuf/proto"
|
|
|
|
bcproto "github.com/tendermint/tendermint/proto/blockchain"
|
|
"github.com/tendermint/tendermint/types"
|
|
)
|
|
|
|
const (
|
|
// NOTE: keep up to date with bcproto.BlockResponse
|
|
BlockResponseMessagePrefixSize = 4
|
|
BlockResponseMessageFieldKeySize = 1
|
|
MaxMsgSize = types.MaxBlockSizeBytes +
|
|
BlockResponseMessagePrefixSize +
|
|
BlockResponseMessageFieldKeySize
|
|
)
|
|
|
|
// EncodeMsg encodes a Protobuf message
|
|
func EncodeMsg(pb proto.Message) ([]byte, error) {
|
|
msg := bcproto.Message{}
|
|
|
|
switch pb := pb.(type) {
|
|
case *bcproto.BlockRequest:
|
|
msg.Sum = &bcproto.Message_BlockRequest{BlockRequest: pb}
|
|
case *bcproto.BlockResponse:
|
|
msg.Sum = &bcproto.Message_BlockResponse{BlockResponse: pb}
|
|
case *bcproto.NoBlockResponse:
|
|
msg.Sum = &bcproto.Message_NoBlockResponse{NoBlockResponse: pb}
|
|
case *bcproto.StatusRequest:
|
|
msg.Sum = &bcproto.Message_StatusRequest{StatusRequest: pb}
|
|
case *bcproto.StatusResponse:
|
|
msg.Sum = &bcproto.Message_StatusResponse{StatusResponse: pb}
|
|
default:
|
|
return nil, fmt.Errorf("unknown message type %T", pb)
|
|
}
|
|
|
|
bz, err := proto.Marshal(&msg)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("unable to marshal %T: %w", pb, err)
|
|
}
|
|
|
|
return bz, nil
|
|
}
|
|
|
|
// DecodeMsg decodes a Protobuf message.
|
|
func DecodeMsg(bz []byte) (proto.Message, error) {
|
|
pb := &bcproto.Message{}
|
|
|
|
err := proto.Unmarshal(bz, pb)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
switch msg := pb.Sum.(type) {
|
|
case *bcproto.Message_BlockRequest:
|
|
return msg.BlockRequest, nil
|
|
case *bcproto.Message_BlockResponse:
|
|
return msg.BlockResponse, nil
|
|
case *bcproto.Message_NoBlockResponse:
|
|
return msg.NoBlockResponse, nil
|
|
case *bcproto.Message_StatusRequest:
|
|
return msg.StatusRequest, nil
|
|
case *bcproto.Message_StatusResponse:
|
|
return msg.StatusResponse, 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 *bcproto.BlockRequest:
|
|
if msg.Height < 0 {
|
|
return errors.New("negative Height")
|
|
}
|
|
case *bcproto.BlockResponse:
|
|
_, err := types.BlockFromProto(msg.Block)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
case *bcproto.NoBlockResponse:
|
|
if msg.Height < 0 {
|
|
return errors.New("negative Height")
|
|
}
|
|
case *bcproto.StatusResponse:
|
|
if msg.Base < 0 {
|
|
return errors.New("negative Base")
|
|
}
|
|
if msg.Height < 0 {
|
|
return errors.New("negative Height")
|
|
}
|
|
if msg.Base > msg.Height {
|
|
return fmt.Errorf("base %v cannot be greater than height %v", msg.Base, msg.Height)
|
|
}
|
|
case *bcproto.StatusRequest:
|
|
if msg.Base < 0 {
|
|
return errors.New("negative Base")
|
|
}
|
|
if msg.Height < 0 {
|
|
return errors.New("negative Height")
|
|
}
|
|
if msg.Base > msg.Height {
|
|
return fmt.Errorf("base %v cannot be greater than height %v", msg.Base, msg.Height)
|
|
}
|
|
default:
|
|
return fmt.Errorf("unknown message type %T", msg)
|
|
}
|
|
return nil
|
|
}
|