From deb76c519998f280c3353a1269e9b7b12c68bdc4 Mon Sep 17 00:00:00 2001
From: Roman Khimov <roman@nspcc.ru>
Date: Mon, 23 Dec 2019 16:08:58 +0300
Subject: [PATCH] core: add negative non-interop test for all interops

Try to feed some garbage into these functions.
---
 pkg/core/interops_test.go | 88 +++++++++++++++++++++++++++++++++++++++
 1 file changed, 88 insertions(+)
 create mode 100644 pkg/core/interops_test.go

diff --git a/pkg/core/interops_test.go b/pkg/core/interops_test.go
new file mode 100644
index 000000000..b5a3e96b1
--- /dev/null
+++ b/pkg/core/interops_test.go
@@ -0,0 +1,88 @@
+package core
+
+import (
+	"reflect"
+	"runtime"
+	"testing"
+
+	"github.com/CityOfZion/neo-go/pkg/core/storage"
+	"github.com/CityOfZion/neo-go/pkg/smartcontract/trigger"
+	"github.com/CityOfZion/neo-go/pkg/vm"
+	"github.com/stretchr/testify/require"
+)
+
+func testNonInterop(t *testing.T, value interface{}, f func(*interopContext, *vm.VM) error) {
+	v := vm.New()
+	v.Estack().PushVal(value)
+	context := newInteropContext(trigger.Application, newTestChain(t), storage.NewMemoryStore(), nil, nil)
+	require.Error(t, f(context, v))
+}
+
+func TestUnexpectedNonInterops(t *testing.T) {
+	vals := map[string]interface{}{
+		"int":    1,
+		"bool":   false,
+		"string": "smth",
+		"array":  []int{1, 2, 3},
+	}
+
+	// All of these functions expect an interop item on the stack.
+	funcs := []func(*interopContext, *vm.VM) error{
+		(*interopContext).accountGetBalance,
+		(*interopContext).accountGetScriptHash,
+		(*interopContext).accountGetVotes,
+		(*interopContext).assetGetAdmin,
+		(*interopContext).assetGetAmount,
+		(*interopContext).assetGetAssetID,
+		(*interopContext).assetGetAssetType,
+		(*interopContext).assetGetAvailable,
+		(*interopContext).assetGetIssuer,
+		(*interopContext).assetGetOwner,
+		(*interopContext).assetGetPrecision,
+		(*interopContext).assetRenew,
+		(*interopContext).attrGetData,
+		(*interopContext).attrGetUsage,
+		(*interopContext).blockGetTransaction,
+		(*interopContext).blockGetTransactionCount,
+		(*interopContext).blockGetTransactions,
+		(*interopContext).contractGetScript,
+		(*interopContext).contractGetStorageContext,
+		(*interopContext).contractIsPayable,
+		(*interopContext).headerGetConsensusData,
+		(*interopContext).headerGetHash,
+		(*interopContext).headerGetIndex,
+		(*interopContext).headerGetMerkleRoot,
+		(*interopContext).headerGetNextConsensus,
+		(*interopContext).headerGetPrevHash,
+		(*interopContext).headerGetTimestamp,
+		(*interopContext).headerGetVersion,
+		(*interopContext).inputGetHash,
+		(*interopContext).inputGetIndex,
+		(*interopContext).invocationTxGetScript,
+		(*interopContext).outputGetAssetID,
+		(*interopContext).outputGetScriptHash,
+		(*interopContext).outputGetValue,
+		(*interopContext).storageContextAsReadOnly,
+		(*interopContext).storageDelete,
+		(*interopContext).storageGet,
+		(*interopContext).storagePut,
+		(*interopContext).storagePutEx,
+		(*interopContext).txGetAttributes,
+		(*interopContext).txGetHash,
+		(*interopContext).txGetInputs,
+		(*interopContext).txGetOutputs,
+		(*interopContext).txGetReferences,
+		(*interopContext).txGetType,
+		(*interopContext).txGetUnspentCoins,
+		(*interopContext).txGetWitnesses,
+		(*interopContext).witnessGetVerificationScript,
+	}
+	for _, f := range funcs {
+		for k, v := range vals {
+			fname := runtime.FuncForPC(reflect.ValueOf(f).Pointer()).Name()
+			t.Run(k+"/"+fname, func(t *testing.T) {
+				testNonInterop(t, v, f)
+			})
+		}
+	}
+}