Minor updates

Confuser.Core/ConfuserContext.cs

@@ -142,6 +142,11 @@ public ModuleDefMD CurrentModule {
 		/// <value>The output <c>byte[]</c> debug symbol.</value>
 		public byte[] CurrentModuleSymbol { get; internal set; }
 
+		/// <summary>
+		///		Gets the token used to indicate cancellation
+		/// </summary>
+		public CancellationToken CancellationToken { get { return token; } }
+
 		/// <summary>
 		///     Throws a System.OperationCanceledException if protection process has been canceled.
 		/// </summary>

Confuser.Protections/Constants/ReferenceReplacer.cs

@@ -373,25 +373,9 @@ static void ReplaceCFG(MethodDef method, List<Tuple<Instruction, uint, IMethod>>
 			}
 
 			// Update state for blocks not in use
-			var done = new HashSet<ControlFlowBlock>();
-			var q = new Stack<ControlFlowBlock>(graph.Where(b => (b.Type & ControlFlowBlockType.Entry) != 0).Reverse());
-			while (q.Count > 0) {
-				var block = q.Pop();
-				if (done.Contains(block))
-					continue;
-				done.Add(block);
-
-				foreach (var succ in block.Targets)
-					q.Push(succ);
-
-				if (blockReferences.ContainsKey(block.Id))
-					continue;
-				InsertEmptyStateUpdate(cfgCtx, block);
-			}
-
 			for (int i = 0; i < graph.Count; i++) {
 				var block = graph[i];
-				if (done.Contains(block) || blockReferences.ContainsKey(block.Id))
+				if (blockReferences.ContainsKey(block.Id))
 					continue;
 				InsertEmptyStateUpdate(cfgCtx, block);
 			}