Allow sharing ReferenceCounter between engines (#340)

neo-project · Jul 5, 2020 · 87d7675 · 87d7675
1 parent baf4989
commit 87d7675
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/src/neo-vm/ExecutionEngine.cs b/src/neo-vm/ExecutionEngine.cs
@@ -38,7 +38,7 @@ public class ExecutionEngine : IDisposable
 
         #endregion
 
-        public ReferenceCounter ReferenceCounter { get; } = new ReferenceCounter();
+        public ReferenceCounter ReferenceCounter { get; }
         public Stack<ExecutionContext> InvocationStack { get; } = new Stack<ExecutionContext>();
         public ExecutionContext CurrentContext { get; private set; }
         public ExecutionContext EntryContext { get; private set; }
@@ -61,9 +61,14 @@ internal protected set
             }
         }
 
-        public ExecutionEngine()
+        public ExecutionEngine() : this(new ReferenceCounter())
         {
-            ResultStack = new EvaluationStack(ReferenceCounter);
+        }
+
+        protected ExecutionEngine(ReferenceCounter referenceCounter)
+        {
+            this.ReferenceCounter = referenceCounter;
+            this.ResultStack = new EvaluationStack(referenceCounter);
         }
 
         #region Limits