de4dot-cex/de4dot.code/deobfuscators/ILProtector/MainType.cs
2012-07-02 21:49:45 +02:00

108 lines
2.9 KiB
C#

/*
Copyright (C) 2011-2012 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.Collections.Generic;
using Mono.Cecil;
using Mono.Cecil.Cil;
using de4dot.blocks;
namespace de4dot.code.deobfuscators.ILProtector {
class MainType {
ModuleDefinition module;
List<MethodDefinition> protectMethods;
TypeDefinition invokerDelegate;
FieldDefinition invokerInstanceField;
public IEnumerable<MethodDefinition> ProtectMethods {
get { return protectMethods; }
}
public TypeDefinition InvokerDelegate {
get { return invokerDelegate; }
}
public FieldDefinition InvokerInstanceField {
get { return invokerInstanceField; }
}
public bool Detected {
get { return protectMethods != null; }
}
public MainType(ModuleDefinition module) {
this.module = module;
}
public void find() {
checkMethod(DotNetUtils.getModuleTypeCctor(module));
}
static string[] ilpLocals = new string[] {
"System.Boolean",
"System.IntPtr",
"System.Object[]",
};
bool checkMethod(MethodDefinition cctor) {
if (cctor == null || cctor.Body == null)
return false;
if (!new LocalTypes(cctor).exactly(ilpLocals))
return false;
var type = cctor.DeclaringType;
var methods = getPinvokeMethods(type, "Protect");
if (methods.Count == 0)
methods = getPinvokeMethods(type, "P0");
if (methods.Count != 2)
return false;
if (type.Fields.Count != 1)
return false;
var theField = type.Fields[0];
var theDelegate = theField.FieldType as TypeDefinition;
if (theDelegate == null || !DotNetUtils.derivesFromDelegate(theDelegate))
return false;
protectMethods = methods;
invokerDelegate = theDelegate;
invokerInstanceField = theField;
return true;
}
static List<MethodDefinition> getPinvokeMethods(TypeDefinition type, string name) {
var list = new List<MethodDefinition>();
foreach (var method in type.Methods) {
if (method.PInvokeInfo != null && method.PInvokeInfo.EntryPoint == name)
list.Add(method);
}
return list;
}
public void cleanUp() {
var cctor = DotNetUtils.getModuleTypeCctor(module);
if (cctor != null) {
cctor.Body.InitLocals = false;
cctor.Body.Variables.Clear();
cctor.Body.Instructions.Clear();
cctor.Body.Instructions.Add(Instruction.Create(OpCodes.Ret));
cctor.Body.ExceptionHandlers.Clear();
}
}
}
}