57 lines
1.3 KiB
Go
57 lines
1.3 KiB
Go
package compiler_test
|
|
|
|
import (
|
|
"fmt"
|
|
"testing"
|
|
|
|
"github.com/nspcc-dev/neo-go/pkg/core/interop/interopnames"
|
|
"github.com/nspcc-dev/neo-go/pkg/crypto/keys"
|
|
"github.com/nspcc-dev/neo-go/pkg/vm"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
// In this test, we only check that needed interop
|
|
// is called with the provided arguments in the right order.
|
|
func TestVerifyGood(t *testing.T) {
|
|
msg := []byte("test message")
|
|
pub, sig := signMessage(t, msg)
|
|
src := getVerifyProg(pub, sig)
|
|
|
|
v, p := vmAndCompileInterop(t, src)
|
|
p.interops[interopnames.ToID([]byte(interopnames.SystemCryptoCheckSig))] = func(v *vm.VM) error {
|
|
assert.Equal(t, pub, v.Estack().Pop().Bytes())
|
|
assert.Equal(t, sig, v.Estack().Pop().Bytes())
|
|
v.Estack().PushVal(true)
|
|
return nil
|
|
}
|
|
|
|
require.NoError(t, v.Run())
|
|
}
|
|
|
|
func signMessage(t *testing.T, msg []byte) ([]byte, []byte) {
|
|
key, err := keys.NewPrivateKey()
|
|
require.NoError(t, err)
|
|
|
|
sig := key.Sign(msg)
|
|
pub := key.PublicKey().Bytes()
|
|
|
|
return pub, sig
|
|
}
|
|
|
|
func getVerifyProg(pub, sig []byte) string {
|
|
pubS := fmt.Sprintf("%#v", pub)
|
|
sigS := fmt.Sprintf("%#v", sig)
|
|
|
|
return `
|
|
package hello
|
|
|
|
import "github.com/nspcc-dev/neo-go/pkg/interop/crypto"
|
|
|
|
func Main() bool {
|
|
pub := ` + pubS + `
|
|
sig := ` + sigS + `
|
|
return crypto.CheckSig(pub, sig)
|
|
}
|
|
`
|
|
}
|