Main embedded asm doesn't always have the same asm name as the original asm

This commit is contained in:
de4dot 2012-07-24 19:05:50 +02:00
parent e1f8793302
commit c31e6c2c3d
2 changed files with 9 additions and 1 deletions

View File

@ -242,6 +242,14 @@ namespace de4dot.code.deobfuscators.CodeWall {
return null;
}
public AssemblyInfo findMain() {
foreach (var asmInfo in assemblyInfos) {
if (asmInfo.isEntryPointAssembly)
return asmInfo;
}
return null;
}
public void remove(AssemblyInfo asmInfo) {
assemblyInfos.Remove(asmInfo);
}

View File

@ -186,7 +186,7 @@ namespace de4dot.code.deobfuscators.CodeWall {
var asm = module.Assembly;
if (asm == null || assemblyDecrypter == null)
return null;
var asmInfo = assemblyDecrypter.findMain(asm.FullName);
var asmInfo = assemblyDecrypter.findMain(asm.FullName) ?? assemblyDecrypter.findMain();
if (asmInfo == null)
return null;