de4dot-cex/de4dot.code/deobfuscators/Goliath_NET/DecrypterBase.cs
2015-10-29 22:45:26 +01:00

301 lines
8.0 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 System.IO;
using dnlib.DotNet;
using dnlib.DotNet.Emit;
using de4dot.blocks;
namespace de4dot.code.deobfuscators.Goliath_NET {
abstract class DecrypterBase {
protected ModuleDefMD module;
EmbeddedResource encryptedResource;
TypeDef decrypterType;
TypeDef delegateType;
TypeDef delegateInitType;
protected BinaryReader decryptedReader;
MethodDefAndDeclaringTypeDict<Info> decrypterMethods = new MethodDefAndDeclaringTypeDict<Info>();
protected class Info {
public MethodDef method;
public int offset;
public bool referenced = false;
public Info(MethodDef method, int offset) {
this.method = method;
this.offset = offset;
}
}
public bool Detected {
get { return encryptedResource != null; }
}
public Resource EncryptedResource {
get { return encryptedResource; }
}
public TypeDef Type {
get { return decrypterType; }
}
public TypeDef DelegateInitType {
get { return delegateInitType ?? FindDelegateInitType(); }
}
public TypeDef DelegateType {
get { return delegateType; }
}
public IEnumerable<TypeDef> DecrypterTypes {
get {
var types = new TypeDefDict<TypeDef>();
foreach (var info in decrypterMethods.GetValues()) {
if (info.referenced)
types.Add(info.method.DeclaringType, info.method.DeclaringType);
}
return types.GetValues();
}
}
public DecrypterBase(ModuleDefMD module) {
this.module = module;
}
protected Info GetInfo(MethodDef method) {
var info = decrypterMethods.Find(method);
if (info == null)
return null;
info.referenced = true;
return info;
}
public void Find() {
foreach (var tmp in module.Resources) {
var resource = tmp as EmbeddedResource;
if (resource == null)
continue;
if (!resource.Name.String.EndsWith(".resources", StringComparison.Ordinal))
continue;
string ns, name;
SplitTypeName(resource.Name.String.Substring(0, resource.Name.String.Length - 10), out ns, out name);
var type = new TypeRefUser(module, ns, name, module).Resolve();
if (type == null)
continue;
if (!CheckDecrypterType(type))
continue;
encryptedResource = resource;
decrypterType = type;
break;
}
}
protected abstract bool CheckDecrypterType(TypeDef type);
void SplitTypeName(string fullName, out string ns, out string name) {
int index = fullName.LastIndexOf('.');
if (index < 0) {
ns = "";
name = fullName;
}
else {
ns = fullName.Substring(0, index);
name = fullName.Substring(index + 1);
}
}
public void Initialize() {
if (encryptedResource == null)
return;
decryptedReader = new BinaryReader(new MemoryStream(Decrypt(encryptedResource.GetResourceData())));
delegateType = null;
foreach (var type in module.GetTypes()) {
var cctor = type.FindStaticConstructor();
if (cctor == null)
continue;
if (type.Fields.Count != 1)
continue;
var field = type.Fields[0];
var tmpDelegateType = DotNetUtils.GetType(module, field.FieldType);
if (tmpDelegateType == null)
continue;
if (!CheckDelegateType(tmpDelegateType))
continue;
if (delegateType != null && delegateType != tmpDelegateType)
continue;
if (!CheckCctor(cctor))
continue;
delegateType = tmpDelegateType;
foreach (var method in type.Methods) {
if (method.Name == ".cctor")
continue;
if (!method.IsStatic || method.Body == null)
continue;
var sig = method.MethodSig;
if (sig == null || sig.Params.Count != 0)
continue;
if (sig.RetType.GetElementType() == ElementType.Void)
continue;
var info = GetDecrypterInfo(method, field);
if (info == null)
continue;
decrypterMethods.Add(info.method, info);
}
}
}
Info GetDecrypterInfo(MethodDef method, FieldDef delegateField) {
try {
int index = 0;
var instrs = method.Body.Instructions;
if (instrs[index].OpCode.Code != Code.Ldsfld)
return null;
var field = instrs[index++].Operand as FieldDef;
if (field != delegateField)
return null;
if (!instrs[index].IsLdcI4())
return null;
int offset = instrs[index++].GetLdcI4Value();
if (instrs[index].OpCode.Code != Code.Call && instrs[index].OpCode.Code != Code.Callvirt)
return null;
var calledMethod = instrs[index++].Operand as IMethod;
if (calledMethod.Name != "Invoke")
return null;
if (instrs[index].OpCode.Code == Code.Unbox_Any)
index++;
if (instrs[index++].OpCode.Code != Code.Ret)
return null;
return new Info(method, offset);
}
catch (ArgumentOutOfRangeException) {
return null;
}
}
bool CheckCctor(MethodDef cctor) {
var ldtokenType = GetLdtokenType(cctor);
if (!new SigComparer().Equals(ldtokenType, cctor.DeclaringType))
return false;
MethodDef initMethod = null;
foreach (var method in DotNetUtils.GetCalledMethods(module, cctor)) {
if (DotNetUtils.IsMethod(method, "System.Void", "(System.Type)")) {
initMethod = method;
break;
}
}
if (initMethod == null || initMethod.Body == null)
return false;
return true;
}
static ITypeDefOrRef GetLdtokenType(MethodDef method) {
if (method == null || method.Body == null)
return null;
foreach (var instr in method.Body.Instructions) {
if (instr.OpCode.Code != Code.Ldtoken)
continue;
return instr.Operand as ITypeDefOrRef;
}
return null;
}
bool CheckDelegateType(TypeDef type) {
if (!DotNetUtils.DerivesFromDelegate(type))
return false;
var invoke = type.FindMethod("Invoke");
if (invoke == null)
return false;
return CheckDelegateInvokeMethod(invoke);
}
protected abstract bool CheckDelegateInvokeMethod(MethodDef invokeMethod);
byte[] Decrypt(byte[] encryptedData) {
const int KEY_LEN = 0x100;
if (encryptedData.Length < KEY_LEN)
throw new ApplicationException("Invalid encrypted data length");
var decryptedData = new byte[encryptedData.Length - KEY_LEN];
var pkt = PublicKey.GetRawData(PublicKeyBase.ToPublicKeyToken(module.Assembly.PublicKey));
if (pkt == null || pkt.Length == 0)
pkt = new byte[8];
for (int i = 0, j = 0, ki = 0; i < decryptedData.Length; i++) {
ki = (ki + 1) % (KEY_LEN - 1);
j = (j + encryptedData[ki] + pkt[i % 8]) % (KEY_LEN - 1);
var tmp = encryptedData[j];
encryptedData[j] = encryptedData[ki];
encryptedData[ki] = tmp;
decryptedData[i] = (byte)(encryptedData[KEY_LEN + i] ^ encryptedData[(encryptedData[j] + encryptedData[ki]) % (KEY_LEN - 1)]);
}
return decryptedData;
}
TypeDef FindDelegateInitType() {
if (delegateType == null)
return null;
foreach (var type in module.Types) {
if (type.HasProperties || type.HasEvents || type.HasFields)
continue;
foreach (var method in type.Methods) {
if (!method.IsStatic || method.IsPrivate || method.Body == null)
continue;
var ldtokenType = GetLdtokenType(method);
if (ldtokenType == null)
continue;
if (!new SigComparer().Equals(ldtokenType, delegateType))
continue;
delegateInitType = type;
return delegateInitType;
}
}
return null;
}
public IEnumerable<MethodDef> GetMethods() {
var list = new List<MethodDef>(decrypterMethods.Count);
foreach (var info in decrypterMethods.GetValues())
list.Add(info.method);
return list;
}
}
}