update contract return type verification

CityOfZion · Jul 9, 2019 · 003ff11 · 003ff11
1 parent 9bf3c1e
commit 003ff11
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/neo/SmartContract/StateMachine.py b/neo/SmartContract/StateMachine.py
@@ -639,9 +639,7 @@ def Contract_Create(self, engine: ExecutionEngine):
         if len(param_list) > 252:
             return False
         return_type = int(engine.CurrentContext.EvaluationStack.Pop().GetBigInteger())
-        try:
-            ContractParameterType(return_type)
-        except ValueError:
+        if return_type > 0xff:
             raise ValueError("Invalid return type data popped from stack")
 
         contract_properties = int(engine.CurrentContext.EvaluationStack.Pop().GetBigInteger())