ccb158c99b
What: add monkit.Task to a bunch of functions that are missing it Why: this will significantly help our instrumentation, data collection, and tracing about what's going on in the network
121 lines
3.5 KiB
Go
121 lines
3.5 KiB
Go
// Copyright (C) 2018 Storj Labs, Inc.
|
|
// See LICENSE for copying information.
|
|
|
|
package auth
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/gogo/protobuf/proto"
|
|
"github.com/zeebo/errs"
|
|
|
|
"storj.io/storj/pkg/identity"
|
|
"storj.io/storj/pkg/peertls"
|
|
"storj.io/storj/pkg/pkcrypto"
|
|
"storj.io/storj/pkg/storj"
|
|
)
|
|
|
|
var (
|
|
//ErrSign indicates a failure during signing
|
|
ErrSign = errs.Class("Failed to sign message")
|
|
//ErrVerify indicates a failure during signature validation
|
|
ErrVerify = errs.Class("Failed to validate message signature")
|
|
//ErrSigLen indicates an invalid signature length
|
|
ErrSigLen = errs.Class("Invalid signature length")
|
|
//ErrSerial indicates an invalid serial number length
|
|
ErrSerial = errs.Class("Invalid SerialNumber")
|
|
//ErrExpired indicates the agreement is expired
|
|
ErrExpired = errs.Class("Agreement is expired")
|
|
//ErrSigner indicates a public key / node id mismatch
|
|
ErrSigner = errs.Class("Message public key did not match expected signer")
|
|
//ErrBadID indicates a public key / node id mismatch
|
|
ErrBadID = errs.Class("Node ID did not match expected id")
|
|
|
|
//ErrMarshal indicates a failure during serialization
|
|
ErrMarshal = errs.Class("Could not marshal item to bytes")
|
|
//ErrUnmarshal indicates a failure during deserialization
|
|
ErrUnmarshal = errs.Class("Could not unmarshal bytes to item")
|
|
//ErrMissing indicates missing or empty information
|
|
ErrMissing = errs.Class("Required field is empty")
|
|
)
|
|
|
|
//SignableMessage is a protocol buffer with a certs and a signature
|
|
//Note that we assume proto.Message is a pointer receiver
|
|
type SignableMessage interface {
|
|
proto.Message
|
|
GetCerts() [][]byte
|
|
GetSignature() []byte
|
|
SetCerts([][]byte)
|
|
SetSignature([]byte)
|
|
}
|
|
|
|
//SignMessage adds the crypto-related aspects of signed message
|
|
func SignMessage(ctx context.Context, msg SignableMessage, id identity.FullIdentity) (err error) {
|
|
defer mon.Task()(&ctx)(&err)
|
|
if msg == nil {
|
|
return ErrMissing.New("message")
|
|
}
|
|
msg.SetSignature(nil)
|
|
msg.SetCerts(nil)
|
|
msgBytes, err := proto.Marshal(msg)
|
|
if err != nil {
|
|
return ErrMarshal.Wrap(err)
|
|
}
|
|
signature, err := pkcrypto.HashAndSign(id.Key, msgBytes)
|
|
if err != nil {
|
|
return ErrSign.Wrap(err)
|
|
}
|
|
msg.SetSignature(signature)
|
|
msg.SetCerts(id.RawChain())
|
|
return nil
|
|
}
|
|
|
|
//VerifyMsg checks the crypto-related aspects of signed message
|
|
func VerifyMsg(ctx context.Context, msg SignableMessage, signer storj.NodeID) (err error) {
|
|
defer mon.Task()(&ctx)(&err)
|
|
//setup
|
|
switch {
|
|
case msg == nil:
|
|
return ErrMissing.New("message")
|
|
case msg.GetSignature() == nil:
|
|
return ErrMissing.New("message signature")
|
|
case msg.GetCerts() == nil:
|
|
return ErrMissing.New("message certificates")
|
|
}
|
|
signature := msg.GetSignature()
|
|
certs := msg.GetCerts()
|
|
msg.SetSignature(nil)
|
|
msg.SetCerts(nil)
|
|
msgBytes, err := proto.Marshal(msg)
|
|
if err != nil {
|
|
return ErrMarshal.Wrap(err)
|
|
}
|
|
//check certs
|
|
if len(certs) < 2 {
|
|
return ErrVerify.New("Expected at least leaf and CA public keys")
|
|
}
|
|
err = peertls.VerifyPeerFunc(peertls.VerifyPeerCertChains)(certs, nil)
|
|
if err != nil {
|
|
return ErrVerify.Wrap(err)
|
|
}
|
|
leaf, err := pkcrypto.CertFromDER(certs[peertls.LeafIndex])
|
|
if err != nil {
|
|
return err
|
|
}
|
|
ca, err := pkcrypto.CertFromDER(certs[peertls.CAIndex])
|
|
if err != nil {
|
|
return err
|
|
}
|
|
// verify signature
|
|
if id, err := identity.NodeIDFromCert(ca); err != nil || id != signer {
|
|
return ErrSigner.New("%+v vs %+v", id, signer)
|
|
}
|
|
if err := pkcrypto.HashAndVerifySignature(leaf.PublicKey, msgBytes, signature); err != nil {
|
|
return ErrVerify.New("%+v", err)
|
|
}
|
|
//cleanup
|
|
msg.SetSignature(signature)
|
|
msg.SetCerts(certs)
|
|
return nil
|
|
}
|