libs5-go/protocol/signed/signed_message.go

169 lines
3.1 KiB
Go
Raw Normal View History

package signed
2024-01-06 11:33:46 +00:00
import (
"crypto/ed25519"
"errors"
"git.lumeweb.com/LumeWeb/libs5-go/encoding"
2024-01-07 08:58:22 +00:00
"git.lumeweb.com/LumeWeb/libs5-go/interfaces"
2024-01-06 11:33:46 +00:00
"git.lumeweb.com/LumeWeb/libs5-go/net"
"git.lumeweb.com/LumeWeb/libs5-go/protocol/base"
2024-01-06 11:33:46 +00:00
"git.lumeweb.com/LumeWeb/libs5-go/types"
"github.com/vmihailenco/msgpack/v5"
2024-01-08 03:57:03 +00:00
"io"
2024-01-06 11:33:46 +00:00
)
var (
_ base.IncomingMessageTyped = (*SignedMessage)(nil)
_ msgpack.CustomDecoder = (*signedMessagePayoad)(nil)
_ msgpack.CustomEncoder = (*SignedMessage)(nil)
2024-01-06 11:33:46 +00:00
)
var (
errInvalidSignature = errors.New("Invalid signature found")
)
type SignedMessage struct {
nodeId *encoding.NodeId
signature []byte
message []byte
base.IncomingMessageTypedImpl
2024-01-06 11:33:46 +00:00
}
func (s *SignedMessage) SetNodeId(nodeId *encoding.NodeId) {
s.nodeId = nodeId
}
func (s *SignedMessage) SetSignature(signature []byte) {
s.signature = signature
}
func (s *SignedMessage) SetMessage(message []byte) {
s.message = message
}
func NewSignedMessageRequest(message []byte) *SignedMessage {
return &SignedMessage{message: message}
}
2024-01-06 11:33:46 +00:00
type signedMessagePayoad struct {
kind int
message msgpack.RawMessage
}
func (s *signedMessagePayoad) DecodeMsgpack(dec *msgpack.Decoder) error {
kind, err := dec.DecodeInt()
if err != nil {
return err
}
s.kind = kind
2024-01-08 03:57:03 +00:00
message, err := io.ReadAll(dec.Buffered())
2024-01-06 11:33:46 +00:00
if err != nil {
return err
}
s.message = message
return nil
}
func NewSignedMessage() *SignedMessage {
return &SignedMessage{}
}
func (s *SignedMessage) HandleMessage(node interfaces.Node, peer net.Peer, verifyId bool) error {
2024-01-06 11:33:46 +00:00
var payload signedMessagePayoad
err := msgpack.Unmarshal(s.message, &payload)
if err != nil {
return err
}
if msgHandler, valid := GetMessageType(types.ProtocolMethod(payload.kind)); valid {
2024-01-06 11:33:46 +00:00
msgHandler.SetIncomingMessage(s)
msgHandler.SetSelf(msgHandler)
2024-01-06 11:33:46 +00:00
err := msgpack.Unmarshal(payload.message, &msgHandler)
if err != nil {
return err
}
err = msgHandler.HandleMessage(node, peer, verifyId)
if err != nil {
return err
}
}
return nil
}
func (s *SignedMessage) DecodeMessage(dec *msgpack.Decoder) error {
nodeId, err := dec.DecodeBytes()
if err != nil {
return err
}
s.nodeId = encoding.NewNodeId(nodeId)
signature, err := dec.DecodeBytes()
if err != nil {
return err
}
s.signature = signature
message, err := dec.DecodeBytes()
if err != nil {
return err
}
s.message = message
if !ed25519.Verify(s.nodeId.Raw()[1:], s.message, s.signature) {
2024-01-06 11:33:46 +00:00
return errInvalidSignature
}
return nil
}
func (s *SignedMessage) EncodeMsgpack(enc *msgpack.Encoder) error {
err := enc.EncodeInt(int64(types.ProtocolMethodSignedMessage))
if err != nil {
return err
}
err = enc.EncodeBytes(s.nodeId.Raw())
if err != nil {
return err
}
err = enc.EncodeBytes(s.signature)
if err != nil {
return err
}
err = enc.EncodeBytes(s.message)
if err != nil {
return err
}
return nil
}
func (s *SignedMessage) Sign(node interfaces.Node) error {
if s.nodeId == nil {
panic("nodeId is nil")
}
2024-01-06 11:33:46 +00:00
if s.message == nil {
panic("message is nil")
}
s.signature = ed25519.Sign(node.Config().KeyPair.ExtractBytes(), s.message)
return nil
2024-01-06 11:33:46 +00:00
}