From e2dfa07cec76f1bb27c1e0e91d918221c7057222 Mon Sep 17 00:00:00 2001
From: Martin Holst Swende <martin@swende.se>
Date: Sat, 17 Jun 2017 16:56:20 +0200
Subject: [PATCH] core/vm: Initial stab at some extended benchmarking

---
 core/vm/instructions_test.go | 182 +++++++++++++++++++++++++++++++++++
 1 file changed, 182 insertions(+)

diff --git a/core/vm/instructions_test.go b/core/vm/instructions_test.go
index ae428aeaba..4f2a7d61db 100644
--- a/core/vm/instructions_test.go
+++ b/core/vm/instructions_test.go
@@ -40,3 +40,185 @@ func TestByteOp(t *testing.T) {
 		}
 	}
 }
+
+func getBenchmarker(op func(pc *uint64, evm *EVM, contract *Contract, memory *Memory, stack *Stack) ([]byte, error), args ...string) func(b *testing.B) {
+	x := func(bench *testing.B) {
+		var (
+			env   = NewEVM(Context{}, nil, params.TestChainConfig, Config{EnableJit: false, ForceJit: false})
+			stack = newstack()
+		)
+		// convert args
+		byteArgs := make([][]byte, len(args))
+		for i, arg := range args {
+			byteArgs[i] = common.Hex2Bytes(arg)
+		}
+		pc := uint64(0)
+		bench.ResetTimer()
+		for i := 0; i < bench.N; i++ {
+			for _, arg := range byteArgs {
+				a := new(big.Int).SetBytes(arg)
+				stack.push(a)
+			}
+			op(&pc, env, nil, nil, stack)
+			stack.pop()
+		}
+	}
+	return x
+}
+
+func BenchmarkOpAdd(b *testing.B) {
+	x := "ABCDEF090807060504030201ffffffffffffffffffffffffffffffffffffffff"
+	y := "ABCDEF090807060504030201ffffffffffffffffffffffffffffffffffffffff"
+
+	f := getBenchmarker(opAdd, x, y)
+	f(b)
+}
+func BenchmarkOpSub(b *testing.B) {
+	x := "ABCDEF090807060504030201ffffffffffffffffffffffffffffffffffffffff"
+	y := "ABCDEF090807060504030201ffffffffffffffffffffffffffffffffffffffff"
+
+	f := getBenchmarker(opSub, x, y)
+	f(b)
+}
+func BenchmarkOpMul(b *testing.B) {
+	x := "ABCDEF090807060504030201ffffffffffffffffffffffffffffffffffffffff"
+	y := "ABCDEF090807060504030201ffffffffffffffffffffffffffffffffffffffff"
+
+	f := getBenchmarker(opMul, x, y)
+
+	f(b)
+}
+func BenchmarkOpDiv(b *testing.B) {
+	x := "ABCDEF090807060504030201ffffffffffffffffffffffffffffffffffffffff"
+	y := "ABCDEF090807060504030201ffffffffffffffffffffffffffffffffffffffff"
+
+	f := getBenchmarker(opDiv, x, y)
+	f(b)
+}
+func BenchmarkOpSdiv(b *testing.B) {
+	x := "ABCDEF090807060504030201ffffffffffffffffffffffffffffffffffffffff"
+	y := "ABCDEF090807060504030201ffffffffffffffffffffffffffffffffffffffff"
+
+	f := getBenchmarker(opSdiv, x, y)
+	f(b)
+}
+func BenchmarkOpMod(b *testing.B) {
+	x := "ABCDEF090807060504030201ffffffffffffffffffffffffffffffffffffffff"
+	y := "ABCDEF090807060504030201ffffffffffffffffffffffffffffffffffffffff"
+
+	f := getBenchmarker(opMod, x, y)
+	f(b)
+}
+func BenchmarkOpSmod(b *testing.B) {
+	x := "ABCDEF090807060504030201ffffffffffffffffffffffffffffffffffffffff"
+	y := "ABCDEF090807060504030201ffffffffffffffffffffffffffffffffffffffff"
+
+	f := getBenchmarker(opSmod, x, y)
+	f(b)
+}
+func BenchmarkOpExp(b *testing.B) {
+	x := "ABCDEF090807060504030201ffffffffffffffffffffffffffffffffffffffff"
+	y := "ABCDEF090807060504030201ffffffffffffffffffffffffffffffffffffffff"
+
+	f := getBenchmarker(opExp, x, y)
+
+	f(b)
+}
+func BenchmarkOpSignExtend(b *testing.B) {
+	x := "ABCDEF090807060504030201ffffffffffffffffffffffffffffffffffffffff"
+	y := "ABCDEF090807060504030201ffffffffffffffffffffffffffffffffffffffff"
+
+	f := getBenchmarker(opSignExtend, x, y)
+
+	f(b)
+}
+func BenchmarkOpLt(b *testing.B) {
+	x := "ABCDEF090807060504030201ffffffffffffffffffffffffffffffffffffffff"
+	y := "ABCDEF090807060504030201ffffffffffffffffffffffffffffffffffffffff"
+
+	f := getBenchmarker(opLt, x, y)
+	f(b)
+}
+func BenchmarkOpGt(b *testing.B) {
+	x := "ABCDEF090807060504030201ffffffffffffffffffffffffffffffffffffffff"
+	y := "ABCDEF090807060504030201ffffffffffffffffffffffffffffffffffffffff"
+
+	f := getBenchmarker(opGt, x, y)
+	f(b)
+}
+func BenchmarkOpSlt(b *testing.B) {
+	x := "ABCDEF090807060504030201ffffffffffffffffffffffffffffffffffffffff"
+	y := "ABCDEF090807060504030201ffffffffffffffffffffffffffffffffffffffff"
+
+	f := getBenchmarker(opSlt, x, y)
+	f(b)
+}
+func BenchmarkOpSgt(b *testing.B) {
+	x := "ABCDEF090807060504030201ffffffffffffffffffffffffffffffffffffffff"
+	y := "ABCDEF090807060504030201ffffffffffffffffffffffffffffffffffffffff"
+
+	f := getBenchmarker(opSgt, x, y)
+
+	f(b)
+}
+func BenchmarkOpEq(b *testing.B) {
+	x := "ABCDEF090807060504030201ffffffffffffffffffffffffffffffffffffffff"
+	y := "ABCDEF090807060504030201ffffffffffffffffffffffffffffffffffffffff"
+
+	f := getBenchmarker(opEq, x, y)
+	f(b)
+}
+func BenchmarkOpAnd(b *testing.B) {
+	x := "ABCDEF090807060504030201ffffffffffffffffffffffffffffffffffffffff"
+	y := "ABCDEF090807060504030201ffffffffffffffffffffffffffffffffffffffff"
+
+	f := getBenchmarker(opAnd, x, y)
+	f(b)
+}
+func BenchmarkOpOr(b *testing.B) {
+	x := "ABCDEF090807060504030201ffffffffffffffffffffffffffffffffffffffff"
+	y := "ABCDEF090807060504030201ffffffffffffffffffffffffffffffffffffffff"
+
+	f := getBenchmarker(opOr, x, y)
+	f(b)
+}
+func BenchmarkOpXor(b *testing.B) {
+	x := "ABCDEF090807060504030201ffffffffffffffffffffffffffffffffffffffff"
+	y := "ABCDEF090807060504030201ffffffffffffffffffffffffffffffffffffffff"
+
+	f := getBenchmarker(opXor, x, y)
+	f(b)
+}
+func BenchmarkOpByte(b *testing.B) {
+	x := "ABCDEF090807060504030201ffffffffffffffffffffffffffffffffffffffff"
+	y := "ABCDEF090807060504030201ffffffffffffffffffffffffffffffffffffffff"
+
+	f := getBenchmarker(opByte, x, y)
+	f(b)
+}
+
+func BenchmarkOpAddmod(b *testing.B) {
+	x := "ABCDEF090807060504030201ffffffffffffffffffffffffffffffffffffffff"
+	y := "ABCDEF090807060504030201ffffffffffffffffffffffffffffffffffffffff"
+	z := "ABCDEF090807060504030201ffffffffffffffffffffffffffffffffffffffff"
+
+	f := getBenchmarker(opAddmod, x, y, z)
+	f(b)
+}
+func BenchmarkOpMulmod(b *testing.B) {
+	x := "ABCDEF090807060504030201ffffffffffffffffffffffffffffffffffffffff"
+	y := "ABCDEF090807060504030201ffffffffffffffffffffffffffffffffffffffff"
+	z := "ABCDEF090807060504030201ffffffffffffffffffffffffffffffffffffffff"
+
+	f := getBenchmarker(opMulmod, x, y, z)
+	f(b)
+}
+
+//func BenchmarkOpSha3(b *testing.B) {
+//	x := "0"
+//	y := "32"
+//
+//	f := getBenchmarker(opSha3, x, y)
+//
+//	f(b)
+//}