mirror of
https://github.com/aljazceru/breez-lnd.git
synced 2025-12-17 06:04:20 +01:00
8b7c88537c
To make it possible to use a remote lnrpc server as a signer for our wallet, we need to change our main interface to sign the message instead of the message's digest. Otherwise we'd need to alter the lnrpc.SignMessage RPC to accept a digest instead of only the message which has security implications.
72 lines
1.6 KiB
Go
72 lines
1.6 KiB
Go
package keychain
|
|
|
|
import (
|
|
"github.com/btcsuite/btcd/btcec"
|
|
"github.com/btcsuite/btcd/chaincfg/chainhash"
|
|
)
|
|
|
|
func NewPubKeyMessageSigner(keyDesc KeyDescriptor,
|
|
signer MessageSignerRing) *PubKeyMessageSigner {
|
|
|
|
return &PubKeyMessageSigner{
|
|
keyDesc: keyDesc,
|
|
digestSigner: signer,
|
|
}
|
|
}
|
|
|
|
type PubKeyMessageSigner struct {
|
|
keyDesc KeyDescriptor
|
|
digestSigner MessageSignerRing
|
|
}
|
|
|
|
func (p *PubKeyMessageSigner) PubKey() *btcec.PublicKey {
|
|
return p.keyDesc.PubKey
|
|
}
|
|
|
|
func (p *PubKeyMessageSigner) SignMessage(message []byte,
|
|
doubleHash bool) (*btcec.Signature, error) {
|
|
|
|
return p.digestSigner.SignMessage(p.keyDesc, message, doubleHash)
|
|
}
|
|
|
|
func (p *PubKeyMessageSigner) SignMessageCompact(msg []byte,
|
|
doubleHash bool) ([]byte, error) {
|
|
|
|
return p.digestSigner.SignMessageCompact(p.keyDesc, msg, doubleHash)
|
|
}
|
|
|
|
type PrivKeyMessageSigner struct {
|
|
PrivKey *btcec.PrivateKey
|
|
}
|
|
|
|
func (p *PrivKeyMessageSigner) PubKey() *btcec.PublicKey {
|
|
return p.PrivKey.PubKey()
|
|
}
|
|
|
|
func (p *PrivKeyMessageSigner) SignMessage(msg []byte,
|
|
doubleHash bool) (*btcec.Signature, error) {
|
|
|
|
var digest []byte
|
|
if doubleHash {
|
|
digest = chainhash.DoubleHashB(msg)
|
|
} else {
|
|
digest = chainhash.HashB(msg)
|
|
}
|
|
return p.PrivKey.Sign(digest)
|
|
}
|
|
|
|
func (p *PrivKeyMessageSigner) SignMessageCompact(msg []byte,
|
|
doubleHash bool) ([]byte, error) {
|
|
|
|
var digest []byte
|
|
if doubleHash {
|
|
digest = chainhash.DoubleHashB(msg)
|
|
} else {
|
|
digest = chainhash.HashB(msg)
|
|
}
|
|
return btcec.SignCompact(btcec.S256(), p.PrivKey, digest, true)
|
|
}
|
|
|
|
var _ SingleKeyMessageSigner = (*PubKeyMessageSigner)(nil)
|
|
var _ SingleKeyMessageSigner = (*PrivKeyMessageSigner)(nil)
|