89 lines
2.2 KiB
Go
89 lines
2.2 KiB
Go
|
package lnwire
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"math/big"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/roasbeef/btcd/btcec"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestSignatureSerializeDeserialize(t *testing.T) {
|
||
|
|
// Local-scoped closure to serialize and deserialize a Signature and
|
||
|
|
// check for errors as well as check if the results are correct.
|
||
|
|
signatureSerializeDeserialize := func(e btcec.Signature) error {
|
||
|
|
var b [64]byte
|
||
|
|
err := serializeSigToWire(&b, &e)
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
var e2 *btcec.Signature
|
||
|
|
err = deserializeSigFromWire(&e2, b)
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
if e.R.Cmp(e2.R) != 0 {
|
||
|
|
return fmt.Errorf("Pre/post-serialize Rs don't match"+
|
||
|
|
": %s, %s", e.R, e2.R)
|
||
|
|
}
|
||
|
|
if e.S.Cmp(e2.S) != 0 {
|
||
|
|
return fmt.Errorf("Pre/post-serialize Ss don't match"+
|
||
|
|
": %s, %s", e.S, e2.S)
|
||
|
|
}
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
sig := btcec.Signature{}
|
||
|
|
|
||
|
|
// Check R = N-1, S = 128.
|
||
|
|
sig.R = big.NewInt(1) // Allocate a big.Int before we call .Sub.
|
||
|
|
sig.R.Sub(btcec.S256().N, sig.R)
|
||
|
|
sig.S = big.NewInt(128)
|
||
|
|
err := signatureSerializeDeserialize(sig)
|
||
|
|
if err != nil {
|
||
|
|
t.Fatalf("R = N-1, S = 128: %s", err.Error())
|
||
|
|
}
|
||
|
|
|
||
|
|
// Check R = N-1, S = 127.
|
||
|
|
sig.S = big.NewInt(127)
|
||
|
|
err = signatureSerializeDeserialize(sig)
|
||
|
|
if err != nil {
|
||
|
|
t.Fatalf("R = N-1, S = 127: %s", err.Error())
|
||
|
|
}
|
||
|
|
|
||
|
|
// Check R = N-1, S = N>>1.
|
||
|
|
sig.S.Set(btcec.S256().N)
|
||
|
|
sig.S.Rsh(sig.S, 1)
|
||
|
|
err = signatureSerializeDeserialize(sig)
|
||
|
|
if err != nil {
|
||
|
|
t.Fatalf("R = N-1, S = N>>1: %s", err.Error())
|
||
|
|
}
|
||
|
|
|
||
|
|
// Check R = N-1, S = N.
|
||
|
|
sig.S.Set(btcec.S256().N)
|
||
|
|
err = signatureSerializeDeserialize(sig)
|
||
|
|
if err.Error() != "signature S isn't 1 or more" {
|
||
|
|
t.Fatalf("R = N-1, S = N should become R = N-1, S = 0: %s",
|
||
|
|
err.Error())
|
||
|
|
}
|
||
|
|
|
||
|
|
// Check R = N-1, S = N-1.
|
||
|
|
sig.S.Sub(sig.S, big.NewInt(1))
|
||
|
|
err = signatureSerializeDeserialize(sig)
|
||
|
|
if err.Error() != "Pre/post-serialize Ss don't match: 115792089237316"+
|
||
|
|
"195423570985008687907852837564279074904382605163141518161494"+
|
||
|
|
"336, 1" {
|
||
|
|
t.Fatalf("R = N-1, S = N-1 should become R = N-1, S = 1: %s",
|
||
|
|
err.Error())
|
||
|
|
}
|
||
|
|
|
||
|
|
// Check R = 2N, S = 128
|
||
|
|
sig.R.Mul(btcec.S256().N, big.NewInt(2))
|
||
|
|
sig.S.Set(big.NewInt(127))
|
||
|
|
err = signatureSerializeDeserialize(sig)
|
||
|
|
if err.Error() != "R is over 32 bytes long without padding" {
|
||
|
|
t.Fatalf("R = 2N, S = 128, R should be over 32 bytes: %s",
|
||
|
|
err.Error())
|
||
|
|
}
|
||
|
|
}
|