Extend ExecutionContext with states

src/neo-vm/ExecutionContext.cs

@@ -1,11 +1,15 @@
-using System.Diagnostics;
+using System;
+using System.Collections.Generic;
+using System.Diagnostics;
 using System.Runtime.CompilerServices;
 
 namespace Neo.VM
 {
     [DebuggerDisplay("RVCount={RVCount}, InstructionPointer={InstructionPointer}")]
     public class ExecutionContext
     {
+        private readonly Dictionary<Type, object> states = new Dictionary<Type, object>();
+
         /// <summary>
         /// Number of items to be returned
         /// </summary>
@@ -82,10 +86,20 @@ internal ExecutionContext Clone()
         [MethodImpl(MethodImplOptions.AggressiveInlining)]
         private Instruction GetInstruction(int ip) => Script.GetInstruction(ip);
 
+        public T GetState<T>()
+        {
+            return (T)states[typeof(T)];
+        }
+
         internal bool MoveNext()
         {
             InstructionPointer += CurrentInstruction.Size;
             return InstructionPointer < Script.Length;
         }
+
+        public void SetState<T>(T state)
+        {
+            states[typeof(T)] = state;
+        }
     }
 }

src/neo-vm/ExecutionEngine.cs

@@ -62,13 +62,6 @@ public class ExecutionEngine : IDisposable
         public ExecutionContext EntryContext => InvocationStack.Count > 0 ? InvocationStack.Peek(InvocationStack.Count - 1) : null;
         public VMState State { get; internal protected set; } = VMState.BREAK;
 
-        #region Events
-
-        public event EventHandler<ExecutionContext> ContextLoaded;
-        public event EventHandler<ExecutionContext> ContextUnloaded;
-
-        #endregion
-
         #region Limits
 
         /// <summary>
@@ -170,6 +163,10 @@ private static int GetItemCount(IEnumerable<StackItem> items)
 
         #endregion
 
+        protected virtual void ContextUnloaded(ExecutionContext context)
+        {
+        }
+
         public virtual void Dispose()
         {
             InvocationStack.Clear();
@@ -297,11 +294,11 @@ private bool ExecuteInstruction()
                                 context_pop.AltStack.CopyTo(CurrentContext.AltStack);
                             }
                             CheckStackSize(false, 0);
-                            ContextUnloaded?.Invoke(this, context_pop);
                             if (InvocationStack.Count == 0)
                             {
                                 State = VMState.HALT;
                             }
+                            ContextUnloaded(context_pop);
                             return true;
                         }
                     case OpCode.SYSCALL:
@@ -1074,12 +1071,11 @@ private bool ExecuteInstruction()
             return true;
         }
 
-        private void LoadContext(ExecutionContext context)
+        protected virtual void LoadContext(ExecutionContext context)
         {
             if (InvocationStack.Count >= MaxInvocationStackSize)
                 throw new InvalidOperationException();
             InvocationStack.Push(context);
-            ContextLoaded?.Invoke(this, context);
         }
 
         public ExecutionContext LoadScript(byte[] script, int rvcount = -1)