Merge pull request #4651 from Crypt-iQ/zpay32_fuzz

fuzz/zpay32: add Decode/Encode fuzz tests.
This commit is contained in:
Eugene 2020-10-22 10:23:24 -04:00 committed by GitHub
commit 426db548dd
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 71 additions and 0 deletions

22
fuzz/zpay32/decode.go Normal file
View File

@ -0,0 +1,22 @@
// +build gofuzz
package zpay32fuzz
import (
"github.com/btcsuite/btcd/chaincfg"
"github.com/lightningnetwork/lnd/zpay32"
)
// Fuzz_decode is used by go-fuzz.
func Fuzz_decode(data []byte) int {
inv, err := zpay32.Decode(string(data), &chaincfg.TestNet3Params)
if err != nil {
return 1
}
// Call these functions as a sanity check to make sure the invoice
// is well-formed.
_ = inv.MinFinalCLTVExpiry()
_ = inv.Expiry()
return 1
}

49
fuzz/zpay32/encode.go Normal file
View File

@ -0,0 +1,49 @@
// +build gofuzz
package zpay32fuzz
import (
"encoding/hex"
"fmt"
"github.com/btcsuite/btcd/btcec"
"github.com/btcsuite/btcd/chaincfg"
"github.com/lightningnetwork/lnd/zpay32"
)
// Fuzz_encode is used by go-fuzz.
func Fuzz_encode(data []byte) int {
inv, err := zpay32.Decode(string(data), &chaincfg.TestNet3Params)
if err != nil {
return 1
}
// Call these functions as a sanity check to make sure the invoice
// is well-formed.
_ = inv.MinFinalCLTVExpiry()
_ = inv.Expiry()
// Initialize the static key we will be using for this fuzz test.
testPrivKeyBytes, _ := hex.DecodeString("e126f68f7eafcc8b74f54d269fe206be715000f94dac067d1c04a8ca3b2db734")
testPrivKey, _ := btcec.PrivKeyFromBytes(btcec.S256(), testPrivKeyBytes)
// Then, initialize the testMessageSigner so we can encode out
// invoices with this private key.
testMessageSigner := zpay32.MessageSigner{
SignCompact: func(hash []byte) ([]byte, error) {
sig, err := btcec.SignCompact(btcec.S256(),
testPrivKey, hash, true)
if err != nil {
return nil, fmt.Errorf("can't sign the "+
"message: %v", err)
}
return sig, nil
},
}
_, err = inv.Encode(testMessageSigner)
if err != nil {
return 1
}
return 1
}