GetInvocationCounter

neo.UnitTests/UT_Syscalls.cs

@@ -0,0 +1,63 @@
+using Microsoft.VisualStudio.TestTools.UnitTesting;
+using Neo.Ledger;
+using Neo.SmartContract;
+using Neo.VM;
+using System.Linq;
+
+namespace Neo.UnitTests
+{
+    [TestClass]
+    public class UT_Syscalls
+    {
+        [TestMethod]
+        public void System_Runtime_GetInvocationCounter()
+        {
+            var snapshot = TestBlockchain.GetStore().GetSnapshot();
+            var contracts = (TestDataCache<UInt160, ContractState>)snapshot.Contracts;
+
+            // Call System.Runtime.GetInvocationCounter syscall
+
+            var script = new ScriptBuilder();
+            script.EmitSysCall(InteropService.System_Runtime_GetInvocationCounter);
+
+            // Init A,B,C contracts
+            // First two drops is for drop method and arguments
+
+            var contractA = new ContractState() { Script = new byte[] { (byte)OpCode.DROP, (byte)OpCode.DROP }.Concat(script.ToArray()).ToArray() };
+            var contractB = new ContractState() { Script = new byte[] { (byte)OpCode.DROP, (byte)OpCode.DROP, (byte)OpCode.NOP }.Concat(script.ToArray()).ToArray() };
+            var contractC = new ContractState() { Script = new byte[] { (byte)OpCode.DROP, (byte)OpCode.DROP, (byte)OpCode.NOP, (byte)OpCode.NOP }.Concat(script.ToArray()).ToArray() };
+
+            contracts.Add(contractA.ScriptHash, contractA);
+            contracts.Add(contractB.ScriptHash, contractB);
+            contracts.Add(contractC.ScriptHash, contractC);
+
+            // Call A,A,B,C
+
+            script = new ScriptBuilder();
+            script.EmitSysCall(InteropService.System_Contract_Call, contractA.ScriptHash.ToArray(), "dummyMain", 0);
+            script.EmitSysCall(InteropService.System_Contract_Call, contractA.ScriptHash.ToArray(), "dummyMain", 0);
+            script.EmitSysCall(InteropService.System_Contract_Call, contractB.ScriptHash.ToArray(), "dummyMain", 0);
+            script.EmitSysCall(InteropService.System_Contract_Call, contractC.ScriptHash.ToArray(), "dummyMain", 0);
+
+            // Execute
+
+            var engine = new ApplicationEngine(TriggerType.Application, null, snapshot, 0, true);
+            engine.LoadScript(script.ToArray());
+            Assert.AreEqual(engine.Execute(), VMState.HALT);
+
+            // Check the results
+
+            CollectionAssert.AreEqual
+                (
+                engine.ResultStack.Select(u => (int)((VM.Types.Integer)u).GetBigInteger()).ToArray(),
+                new int[]
+                {
+                    1, /* C */
+                    2, /* B */
+                    1, /* A */
+                    1  /* A */
+                }
+                );
+        }
+    }
+}

neo/SmartContract/ApplicationEngine.cs

@@ -19,6 +19,7 @@ public partial class ApplicationEngine : ExecutionEngine
         private readonly List<NotifyEventArgs> notifications = new List<NotifyEventArgs>();
         private readonly List<IDisposable> disposables = new List<IDisposable>();
 
+        public Dictionary<UInt160, int> InvocationCounter { get; } = new Dictionary<UInt160, int>();
         public TriggerType Trigger { get; }
         public IVerifiable ScriptContainer { get; }
         public Snapshot Snapshot { get; }

neo/SmartContract/InteropService.cs

@@ -39,6 +39,7 @@ public static partial class InteropService
         public static readonly uint System_Runtime_GetTime = Register("System.Runtime.GetTime", Runtime_GetTime, 0_00000250);
         public static readonly uint System_Runtime_Serialize = Register("System.Runtime.Serialize", Runtime_Serialize, 0_00100000);
         public static readonly uint System_Runtime_Deserialize = Register("System.Runtime.Deserialize", Runtime_Deserialize, 0_00500000);
+        public static readonly uint System_Runtime_GetInvocationCounter = Register("System.Runtime.GetInvocationCounter", Runtime_GetInvocationCounter, 10000/*TODO Compute Gas*/);
         public static readonly uint System_Crypto_Verify = Register("System.Crypto.Verify", Crypto_Verify, 0_01000000);
         public static readonly uint System_Blockchain_GetHeight = Register("System.Blockchain.GetHeight", Blockchain_GetHeight, 0_00000400);
         public static readonly uint System_Blockchain_GetHeader = Register("System.Blockchain.GetHeader", Blockchain_GetHeader, 0_00007000);
@@ -212,6 +213,17 @@ private static bool Runtime_Serialize(ApplicationEngine engine)
             return true;
         }
 
+        private static bool Runtime_GetInvocationCounter(ApplicationEngine engine)
+        {
+            if (!engine.InvocationCounter.TryGetValue(engine.CurrentScriptHash, out var counter))
+            {
+                counter = 1;
+            }
+
+            engine.CurrentContext.EvaluationStack.Push(counter);
+            return true;
+        }
+
         private static bool Runtime_Deserialize(ApplicationEngine engine)
         {
             StackItem item;
@@ -499,6 +511,15 @@ private static bool Contract_Call(ApplicationEngine engine)
             if (currentManifest != null && !currentManifest.CanCall(contract.Manifest, method.GetString()))
                 return false;
 
+            if (engine.InvocationCounter.TryGetValue(contract.ScriptHash, out var counter))
+            {
+                engine.InvocationCounter[contract.ScriptHash] = counter + 1;
+            }
+            else
+            {
+                engine.InvocationCounter[contract.ScriptHash] = 1;
+            }
+
             ExecutionContext context_new = engine.LoadScript(contract.Script, 1);
             context_new.EvaluationStack.Push(args);
             context_new.EvaluationStack.Push(method);