de4dot-cex/de4dot.code/deobfuscators/StringCounts.cs

/*
    Copyright (C) 2011-2015 de4dot@gmail.com

    This file is part of de4dot.

    de4dot is free software: you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation, either version 3 of the License, or
    (at your option) any later version.

    de4dot is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU General Public License for more details.

    You should have received a copy of the GNU General Public License
    along with de4dot.  If not, see <http://www.gnu.org/licenses/>.
*/

using System;
using System.Collections.Generic;
using dnlib.DotNet;
using dnlib.DotNet.Emit;

namespace de4dot.code.deobfuscators {
	public class StringCounts {
		Dictionary<string, int> strings = new Dictionary<string, int>(StringComparer.Ordinal);

		public IEnumerable<string> Strings {
			get { return strings.Keys; }
		}

		public int NumStrings {
			get { return strings.Count; }
		}

		public void Add(string s) {
			int count;
			strings.TryGetValue(s, out count);
			strings[s] = count + 1;
		}

		public bool Exists(string s) {
			if (s == null)
				return false;
			return strings.ContainsKey(s);
		}

		public bool All(IList<string> list) {
			foreach (var s in list) {
				if (!Exists(s))
					return false;
			}
			return true;
		}

		public bool Exactly(IList<string> list) {
			return list.Count == strings.Count && All(list);
		}

		public int Count(string s) {
			int count;
			strings.TryGetValue(s, out count);
			return count;
		}
	}

	public class FieldTypes : StringCounts {
		public FieldTypes(TypeDef type) {
			Initialize(type.Fields);
		}

		public FieldTypes(IEnumerable<FieldDef> fields) {
			Initialize(fields);
		}

		void Initialize(IEnumerable<FieldDef> fields) {
			if (fields == null)
				return;
			foreach (var field in fields) {
				var type = field.FieldSig.GetFieldType();
				if (type != null)
					Add(type.FullName);
			}
		}
	}

	public class LocalTypes : StringCounts {
		public LocalTypes(MethodDef method) {
			if (method != null && method.Body != null)
				Initialize(method.Body.Variables);
		}

		public LocalTypes(IEnumerable<Local> locals) {
			Initialize(locals);
		}

		void Initialize(IEnumerable<Local> locals) {
			if (locals == null)
				return;
			foreach (var local in locals)
				Add(local.Type.FullName);
		}
	}
}