From cc16dac0b4f817f5c7a86bdaab842aa0309ea6c5 Mon Sep 17 00:00:00 2001
From: Roman Khimov <roman@nspcc.ru>
Date: Mon, 23 Dec 2019 18:02:39 +0300
Subject: [PATCH] vm: optimize script loading a bit

Preseed the scriptHash value when we already know it. Eliminates this time
waste from the pprof graph, but doesn't really change anything in the 1.4M ->
1.5M 100K mainnet blocks import test.
---
 pkg/vm/vm.go | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

diff --git a/pkg/vm/vm.go b/pkg/vm/vm.go
index 204846f49..eae4aaab5 100644
--- a/pkg/vm/vm.go
+++ b/pkg/vm/vm.go
@@ -238,6 +238,14 @@ func (v *VM) LoadScript(b []byte) {
 	v.istack.PushVal(ctx)
 }
 
+// loadScriptWithHash if similar to the LoadScript method, but it also loads
+// given script hash directly into the Context to avoid its recalculations. It's
+// up to user of this function to make sure the script and hash match each other.
+func (v *VM) loadScriptWithHash(b []byte, hash util.Uint160) {
+	v.LoadScript(b)
+	v.istack.Top().Value().(*Context).scriptHash = hash
+}
+
 // Context returns the current executed context. Nil if there is no context,
 // which implies no program is loaded.
 func (v *VM) Context() *Context {
@@ -1120,7 +1128,7 @@ func (v *VM) execute(ctx *Context, op opcode.Opcode, parameter []byte) (err erro
 			_ = v.istack.Pop()
 		}
 
-		v.LoadScript(script)
+		v.loadScriptWithHash(script, hash)
 
 	case opcode.RET:
 		oldCtx := v.istack.Pop().Value().(*Context)
@@ -1336,6 +1344,7 @@ func (v *VM) execute(ctx *Context, op opcode.Opcode, parameter []byte) (err erro
 				panic(fmt.Sprintf("could not find script %s", hash))
 			}
 			newCtx = NewContext(script)
+			newCtx.scriptHash = hash
 		}
 		newCtx.rvcount = rvcount
 		newCtx.estack = NewStack("evaluation")