yck1509-ConfuserEx/Confuser.DynCipher/Transforms/ExpansionTransform.cs

using System;
using System.Linq;
using Confuser.DynCipher.AST;

namespace Confuser.DynCipher.Transforms {
	internal class ExpansionTransform {
		static bool ProcessStatement(Statement st, StatementBlock block) {
			if (st is AssignmentStatement) {
				var assign = (AssignmentStatement)st;
				if (assign.Value is BinOpExpression) {
					var exp = (BinOpExpression)assign.Value;
					if ((exp.Left is BinOpExpression || exp.Right is BinOpExpression) &&
					    exp.Left != assign.Target) {
						block.Statements.Add(new AssignmentStatement {
							Target = assign.Target,
							Value = exp.Left
						});
						block.Statements.Add(new AssignmentStatement {
							Target = assign.Target,
							Value = new BinOpExpression {
								Left = assign.Target,
								Operation = exp.Operation,
								Right = exp.Right
							}
						});
						return true;
					}
				}
			}
			block.Statements.Add(st);
			return false;
		}

		public static void Run(StatementBlock block) {
			bool workDone;
			do {
				workDone = false;
				Statement[] copy = block.Statements.ToArray();
				block.Statements.Clear();
				foreach (Statement st in copy)
					workDone |= ProcessStatement(st, block);
			} while (workDone);
		}
	}
}