309 lines
9.4 KiB
C#
309 lines
9.4 KiB
C#
/*
|
|
Copyright (C) 2011-2013 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;
|
|
using de4dot.blocks;
|
|
|
|
namespace de4dot.code.deobfuscators.DeepSea {
|
|
// DS 4.x can move fields from a class to a struct. This class restores the fields.
|
|
class FieldsRestorer {
|
|
ModuleDefMD module;
|
|
TypeDefDict<List<TypeDef>> structToOwners = new TypeDefDict<List<TypeDef>>();
|
|
FieldDefAndDeclaringTypeDict<bool> structFieldsToFix = new FieldDefAndDeclaringTypeDict<bool>();
|
|
TypeDefDict<FieldDefAndDeclaringTypeDict<FieldDef>> typeToFieldsDict = new TypeDefDict<FieldDefAndDeclaringTypeDict<FieldDef>>();
|
|
|
|
public List<TypeDef> FieldStructs {
|
|
get {
|
|
var list = new List<TypeDef>(structToOwners.Count);
|
|
foreach (var structType in structToOwners.GetKeys()) {
|
|
if (!HasNoMethods(structType))
|
|
continue;
|
|
|
|
list.Add(structType);
|
|
}
|
|
return list;
|
|
}
|
|
}
|
|
|
|
static bool HasNoMethods(TypeDef type) {
|
|
if (type.Methods.Count == 0)
|
|
return true;
|
|
if (type.BaseType == null)
|
|
return false;
|
|
if (type.BaseType.FullName != "System.Object")
|
|
return false;
|
|
if (type.Methods.Count != 1)
|
|
return false;
|
|
var ctor = type.Methods[0];
|
|
if (ctor.Name != ".ctor" || ctor.MethodSig.GetParamCount() != 0)
|
|
return false;
|
|
return true;
|
|
}
|
|
|
|
public FieldsRestorer(ModuleDefMD module) {
|
|
this.module = module;
|
|
}
|
|
|
|
public void Initialize() {
|
|
foreach (var kv in GetMovedTypes()) {
|
|
var structType = kv.Key;
|
|
structToOwners.Add(structType, kv.Value);
|
|
|
|
foreach (var ownerType in kv.Value) {
|
|
foreach (var ownerField in ownerType.Fields) {
|
|
if (DotNetUtils.GetType(module, ownerField.FieldSig.GetFieldType()) != structType)
|
|
continue;
|
|
structFieldsToFix.Add(ownerField, true);
|
|
break;
|
|
}
|
|
|
|
var fieldsDict = new FieldDefAndDeclaringTypeDict<FieldDef>();
|
|
typeToFieldsDict.Add(ownerType, fieldsDict);
|
|
foreach (var structField in structType.Fields) {
|
|
var newField = module.UpdateRowId(new FieldDefUser(structField.Name, structField.FieldSig.Clone(), structField.Attributes));
|
|
ownerType.Fields.Add(newField);
|
|
fieldsDict.Add(structField, newField);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
Dictionary<TypeDef, List<TypeDef>> GetMovedTypes() {
|
|
var candidates = new Dictionary<TypeDef, List<TypeDef>>();
|
|
var typeToStruct = new Dictionary<TypeDef, TypeDef>();
|
|
foreach (var type in module.GetTypes()) {
|
|
foreach (var field in GetPossibleFields(type)) {
|
|
var fieldType = DotNetUtils.GetType(module, field.FieldSig.GetFieldType());
|
|
if (fieldType == null)
|
|
continue;
|
|
if (!CheckBaseType(fieldType))
|
|
continue;
|
|
if ((fieldType.Attributes & ~TypeAttributes.Sealed) != TypeAttributes.NestedAssembly)
|
|
continue;
|
|
if (fieldType.NestedTypes.Count > 0)
|
|
continue;
|
|
if (fieldType.GenericParameters.Count > 0)
|
|
continue;
|
|
if (fieldType.Fields.Count == 0)
|
|
continue;
|
|
if (fieldType.HasEvents || fieldType.HasProperties || fieldType.HasInterfaces)
|
|
continue;
|
|
if (CheckMethods(fieldType))
|
|
continue;
|
|
if (!CheckFields(fieldType))
|
|
continue;
|
|
|
|
List<TypeDef> list;
|
|
if (!candidates.TryGetValue(fieldType, out list))
|
|
candidates[fieldType] = list = new List<TypeDef>();
|
|
list.Add(type);
|
|
typeToStruct[type] = fieldType;
|
|
break;
|
|
}
|
|
}
|
|
|
|
foreach (var type in module.GetTypes()) {
|
|
TypeDef structType;
|
|
typeToStruct.TryGetValue(type, out structType);
|
|
|
|
foreach (var field in type.Fields) {
|
|
if (field.IsStatic || field.FieldSig.GetFieldType().TryGetTypeDef() != structType)
|
|
RemoveType(candidates, field.FieldSig.GetFieldType());
|
|
}
|
|
foreach (var method in type.Methods) {
|
|
RemoveType(candidates, method.MethodSig.GetRetType());
|
|
foreach (var parameter in method.MethodSig.GetParams())
|
|
RemoveType(candidates, parameter);
|
|
if (method.Body != null) {
|
|
foreach (var local in method.Body.Variables)
|
|
RemoveType(candidates, local.Type);
|
|
}
|
|
}
|
|
}
|
|
|
|
return candidates;
|
|
}
|
|
|
|
IEnumerable<FieldDef> GetPossibleFields(TypeDef type) {
|
|
var typeToFields = new TypeDefDict<List<FieldDef>>();
|
|
foreach (var field in type.Fields) {
|
|
if (field.Attributes != FieldAttributes.Private)
|
|
continue;
|
|
var fieldType = DotNetUtils.GetType(module, field.FieldSig.GetFieldType());
|
|
if (fieldType == null)
|
|
continue;
|
|
if (!CheckBaseType(fieldType))
|
|
continue;
|
|
var list = typeToFields.Find(fieldType);
|
|
if (list == null)
|
|
typeToFields.Add(fieldType, list = new List<FieldDef>());
|
|
list.Add(field);
|
|
}
|
|
|
|
foreach (var list in typeToFields.GetValues()) {
|
|
if (list.Count == 1)
|
|
yield return list[0];
|
|
}
|
|
}
|
|
|
|
static bool CheckBaseType(TypeDef type) {
|
|
if (type == null || type.BaseType == null)
|
|
return false;
|
|
var fn = type.BaseType.FullName;
|
|
return fn == "System.ValueType" || fn == "System.Object";
|
|
}
|
|
|
|
void RemoveType(Dictionary<TypeDef, List<TypeDef>> candidates, TypeSig type) {
|
|
var typeDef = DotNetUtils.GetType(module, type);
|
|
if (typeDef == null)
|
|
return;
|
|
candidates.Remove(typeDef);
|
|
}
|
|
|
|
static bool CheckMethods(TypeDef type) {
|
|
foreach (var method in type.Methods) {
|
|
if (method.Name == ".cctor")
|
|
continue;
|
|
if (type.BaseType != null && type.BaseType.FullName == "System.Object" && method.Name == ".ctor" && method.MethodSig.GetParamCount() == 0)
|
|
continue;
|
|
if (!method.IsStatic)
|
|
return true;
|
|
if (method.GenericParameters.Count > 0)
|
|
return true;
|
|
if (method.Body == null)
|
|
return true;
|
|
if (method.ImplMap != null)
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
static bool CheckFields(TypeDef type) {
|
|
if (type.Fields.Count == 0)
|
|
return false;
|
|
foreach (var field in type.Fields) {
|
|
if (field.IsStatic)
|
|
return false;
|
|
if (!field.IsAssembly)
|
|
return false;
|
|
}
|
|
return true;
|
|
}
|
|
|
|
public void Deobfuscate(Blocks blocks) {
|
|
DeobfuscateNormal(blocks);
|
|
FixFieldCtorCalls(blocks);
|
|
}
|
|
|
|
void DeobfuscateNormal(Blocks blocks) {
|
|
var instrsToRemove = new List<int>();
|
|
foreach (var block in blocks.MethodBlocks.GetAllBlocks()) {
|
|
instrsToRemove.Clear();
|
|
var instrs = block.Instructions;
|
|
for (int i = instrs.Count - 1; i >= 0; i--) {
|
|
var instr = instrs[i];
|
|
if (instr.OpCode.Code != Code.Ldflda && instr.OpCode.Code != Code.Ldfld)
|
|
continue;
|
|
var structField = instr.Operand as IField;
|
|
if (structField == null || !structFieldsToFix.Find(structField))
|
|
continue;
|
|
|
|
var ldStFld = instrs[FindLdStFieldIndex(instrs, i + 1)];
|
|
ldStFld.Operand = GetNewField(structField, ldStFld.Operand as IField);
|
|
instrsToRemove.Add(i);
|
|
}
|
|
if (instrsToRemove.Count > 0)
|
|
block.Remove(instrsToRemove);
|
|
}
|
|
}
|
|
|
|
void FixFieldCtorCalls(Blocks blocks) {
|
|
if (blocks.Method.Name != ".ctor")
|
|
return;
|
|
var instrsToRemove = new List<int>();
|
|
foreach (var block in blocks.MethodBlocks.GetAllBlocks()) {
|
|
var instrs = block.Instructions;
|
|
for (int i = 0; i < instrs.Count; i++) {
|
|
var stfld = instrs[i];
|
|
if (stfld.OpCode.Code != Code.Stfld)
|
|
continue;
|
|
var field = stfld.Operand as IField;
|
|
if (field == null)
|
|
continue;
|
|
if (!structFieldsToFix.Find(field))
|
|
continue;
|
|
var instrs2 = ToInstructionList(instrs);
|
|
var instrPushes = DotNetUtils.GetArgPushes(instrs2, i);
|
|
if (instrPushes == null || instrPushes.Count != 2)
|
|
continue;
|
|
block.Remove(i, 1);
|
|
block.Remove(instrs2.IndexOf(instrPushes[1]), 1);
|
|
block.Remove(instrs2.IndexOf(instrPushes[0]), 1);
|
|
i -= 3;
|
|
}
|
|
}
|
|
}
|
|
|
|
static IList<Instruction> ToInstructionList(IEnumerable<Instr> instrs) {
|
|
var newInstrs = new List<Instruction>();
|
|
foreach (var instr in instrs)
|
|
newInstrs.Add(instr.Instruction);
|
|
return newInstrs;
|
|
}
|
|
|
|
FieldDef GetNewField(IField structField, IField oldFieldRef) {
|
|
var fieldsDict = typeToFieldsDict.Find(structField.DeclaringType);
|
|
if (fieldsDict == null)
|
|
throw new ApplicationException("Could not find structField declaringType");
|
|
var newField = fieldsDict.Find(oldFieldRef);
|
|
if (newField == null)
|
|
throw new ApplicationException("Could not find new field");
|
|
return newField;
|
|
}
|
|
|
|
static int FindLdStFieldIndex(IList<Instr> instrs, int index) {
|
|
int stack = 0;
|
|
for (int i = index; i < instrs.Count; i++) {
|
|
var instr = instrs[i];
|
|
|
|
if (stack == 0 && (instr.OpCode.Code == Code.Ldfld || instr.OpCode.Code == Code.Ldflda))
|
|
return i;
|
|
if (stack == 1 && instr.OpCode.Code == Code.Stfld)
|
|
return i;
|
|
|
|
int pushes, pops;
|
|
instr.Instruction.CalculateStackUsage(false, out pushes, out pops);
|
|
stack -= pops;
|
|
if (stack < 0)
|
|
break;
|
|
stack += pushes;
|
|
}
|
|
throw new ApplicationException("Could not find ldfld/stfld");
|
|
}
|
|
|
|
public void CleanUp() {
|
|
foreach (var field in structFieldsToFix.GetKeys())
|
|
field.DeclaringType.Fields.Remove(field);
|
|
}
|
|
}
|
|
}
|