Merge pull request #11 from ElrondNetwork/protect-against-missing-sc-…

…method Add verification in RunSmartContractCall() against missing SC methods
multiversx · Oct 29, 2019 · ae44239 · ae44239
2 parents e53417f + 029f60c
commit ae44239
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/arwen/arwen.go b/arwen/arwen.go
@@ -180,7 +180,12 @@ func (host *vmContext) RunSmartContractCall(input *vmcommon.ContractCallInput) (
 		return host.createVMOutputInCaseOfError(vmcommon.UserError), nil
 	}
 
-	function := instance.Exports[host.callFunction]
+	function, ok := instance.Exports[host.callFunction]
+	if !ok {
+		fmt.Println("arwen Error", "Function not found")
+		return host.createVMOutputInCaseOfError(vmcommon.FunctionNotFound), nil
+	}
+
 	result, err := function()
 	if err != nil {
 		fmt.Println("arwen Error", err.Error())