Add AppEngine.ValidateCallFlags to be callable by subclasses (#1784)

* Update ApplicationEngine.cs separate call flag validation into separate function so it can be called by ApplicationEngine subclasses. This is needed for the debugger, which overrides a few of the standard service implementations * fix whitespace * protected internal addGas Co-authored-by: Harry Pierson <[email protected]>
neo-project · Jul 22, 2020 · 13f303d · 13f303d
1 parent 54a1094
commit 13f303d
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/src/neo/SmartContract/ApplicationEngine.cs b/src/neo/SmartContract/ApplicationEngine.cs
@@ -58,7 +58,7 @@ protected ApplicationEngine(TriggerType trigger, IVerifiable container, StoreVie
             this.testMode = testMode;
         }
 
-        internal void AddGas(long gas)
+        protected internal void AddGas(long gas)
         {
             GasConsumed = checked(GasConsumed + gas);
             if (!testMode && GasConsumed > gas_amount)
@@ -217,12 +217,17 @@ public override void Dispose()
             base.Dispose();
         }
 
-        protected override void OnSysCall(uint method)
+        protected void ValidateCallFlags(InteropDescriptor descriptor)
         {
-            InteropDescriptor descriptor = services[method];
             ExecutionContextState state = CurrentContext.GetState<ExecutionContextState>();
             if (!state.CallFlags.HasFlag(descriptor.RequiredCallFlags))
                 throw new InvalidOperationException($"Cannot call this SYSCALL with the flag {state.CallFlags}.");
+        }
+
+        protected override void OnSysCall(uint method)
+        {
+            InteropDescriptor descriptor = services[method];
+            ValidateCallFlags(descriptor);
             AddGas(descriptor.FixedPrice);
             List<object> parameters = descriptor.Parameters.Count > 0
                 ? new List<object>()