de4dot-cex/de4dot.code/deobfuscators/Agile_NET/vm/v1/FieldsInfo.cs
2015-10-29 22:45:26 +01:00

83 lines
2.1 KiB
C#

/*
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;
namespace de4dot.code.deobfuscators.Agile_NET.vm.v1 {
class FieldsInfo {
public static readonly object EnumType = new object();
Dictionary<string, int> fieldTypes = new Dictionary<string, int>(StringComparer.Ordinal);
int numEnums = 0;
public FieldsInfo(TypeDef type)
: this(type.Fields) {
}
public FieldsInfo(IEnumerable<FieldDef> fields) {
foreach (var field in fields) {
var fieldTypeDef = field.FieldSig.GetFieldType().TryGetTypeDef();
if (fieldTypeDef != null && fieldTypeDef.IsEnum)
AddEnum();
else
Add(field.FieldSig.GetFieldType());
}
}
public FieldsInfo(object[] fieldTypes) {
foreach (var o in fieldTypes) {
if (o == EnumType)
AddEnum();
else
Add((string)o);
}
}
void Add(TypeSig type) {
Add(type.GetFullName());
}
void Add(string typeFullName) {
int count;
fieldTypes.TryGetValue(typeFullName, out count);
fieldTypes[typeFullName] = count + 1;
}
void AddEnum() {
numEnums++;
}
public bool IsSame(FieldsInfo other) {
if (numEnums != other.numEnums)
return false;
if (fieldTypes.Count != other.fieldTypes.Count)
return false;
foreach (var kv in fieldTypes) {
int num;
if (!other.fieldTypes.TryGetValue(kv.Key, out num))
return false;
if (kv.Value != num)
return false;
}
return true;
}
}
}