TrueCloudLab/neo-go
8
0
Fork 0
mirror of https://github.com/nspcc-dev/neo-go.git synced 2024-11-25 13:47:19 +00:00
Code Issues Projects Releases Packages Wiki Activity

vm: add fuzz test for ParseMultiSigContract

Browse source 
Signed-off-by: Evgeniy Stratonikov <evgeniy@nspcc.ru>
This commit is contained in:
Evgeniy Stratonikov 2022-05-30 15:46:40 +03:00 committed by Anna Shaleva
parent affe1ecb4f
commit db977ce38d
1 changed files with 31 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

31
pkg/vm/fuzz_test.go
View file

@ -5,6 +5,8 @@ package vm
import (
"testing"
"github.com/nspcc-dev/neo-go/pkg/crypto/keys"
"github.com/nspcc-dev/neo-go/pkg/smartcontract"
"github.com/nspcc-dev/neo-go/pkg/vm/opcode"
"github.com/stretchr/testify/require"
)
@ -35,6 +37,35 @@ func FuzzIsScriptCorrect(f *testing.F) {
})
}
func FuzzParseMultiSigContract(f *testing.F) {
pubs := make(keys.PublicKeys, 10)
for i := range pubs {
p, _ := keys.NewPrivateKey()
pubs[i] = p.PublicKey()
}
s, _ := smartcontract.CreateMultiSigRedeemScript(1, pubs[:1])
f.Add(s)
s, _ = smartcontract.CreateMultiSigRedeemScript(3, pubs[:6])
f.Add(s)
s, _ = smartcontract.CreateMultiSigRedeemScript(1, pubs)
f.Add(s)
f.Fuzz(func(t *testing.T, script []byte) {
var b [][]byte
var ok bool
var n int
require.NotPanics(t, func() {
n, b, ok = ParseMultiSigContract(script)
})
if ok {
require.True(t, n <= len(b))
}
})
}
func FuzzVMDontPanic(f *testing.F) {
for _, s := range fuzzSeedValidScripts {
f.Add(s)