添加ObfuzIgnoreAttribute
parent
c17397eac0
commit
abecc20a4a
|
@ -1,4 +1,5 @@
|
||||||
using dnlib.DotNet;
|
using dnlib.DotNet;
|
||||||
|
using Obfuz.Utils;
|
||||||
using System;
|
using System;
|
||||||
using System.Collections.Generic;
|
using System.Collections.Generic;
|
||||||
using System.Linq;
|
using System.Linq;
|
||||||
|
@ -9,6 +10,10 @@ namespace Obfuz
|
||||||
{
|
{
|
||||||
public class NotObfuscatedMethodWhiteList
|
public class NotObfuscatedMethodWhiteList
|
||||||
{
|
{
|
||||||
|
private bool ShouldBeIgnoredByCustomAttribute(IHasCustomAttribute obj)
|
||||||
|
{
|
||||||
|
return MetaUtil.HasObfuzIgnoreAttribute(obj) || MetaUtil.HasCompilerGeneratedAttribute(obj);
|
||||||
|
}
|
||||||
|
|
||||||
public bool IsInWhiteList(ModuleDef module)
|
public bool IsInWhiteList(ModuleDef module)
|
||||||
{
|
{
|
||||||
|
@ -17,6 +22,10 @@ namespace Obfuz
|
||||||
{
|
{
|
||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
|
if (ShouldBeIgnoredByCustomAttribute(module))
|
||||||
|
{
|
||||||
|
return true;
|
||||||
|
}
|
||||||
return false;
|
return false;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -26,6 +35,10 @@ namespace Obfuz
|
||||||
{
|
{
|
||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
|
if (ShouldBeIgnoredByCustomAttribute(method))
|
||||||
|
{
|
||||||
|
return true;
|
||||||
|
}
|
||||||
return false;
|
return false;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -39,6 +52,10 @@ namespace Obfuz
|
||||||
{
|
{
|
||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
|
if (ShouldBeIgnoredByCustomAttribute(type))
|
||||||
|
{
|
||||||
|
return true;
|
||||||
|
}
|
||||||
if (type.FullName == "Obfuz.EncryptionVM.GeneratedEncryptionVirtualMachine")
|
if (type.FullName == "Obfuz.EncryptionVM.GeneratedEncryptionVirtualMachine")
|
||||||
{
|
{
|
||||||
return true;
|
return true;
|
||||||
|
|
|
@ -25,7 +25,7 @@ namespace Obfuz.ObfusPasses.FieldEncrypt
|
||||||
_encryptionLevel = settings.encryptionLevel;
|
_encryptionLevel = settings.encryptionLevel;
|
||||||
}
|
}
|
||||||
|
|
||||||
protected override bool NeedProcessNotObfuscatedAssembly => true;
|
protected override bool ForceProcessAllAssembliesAndIgnoreAllPolicy => true;
|
||||||
|
|
||||||
public override void Start()
|
public override void Start()
|
||||||
{
|
{
|
||||||
|
|
|
@ -7,33 +7,33 @@ namespace Obfuz.ObfusPasses
|
||||||
{
|
{
|
||||||
public abstract class InstructionObfuscationPassBase : ObfuscationPassBase
|
public abstract class InstructionObfuscationPassBase : ObfuscationPassBase
|
||||||
{
|
{
|
||||||
protected virtual bool NeedProcessNotObfuscatedAssembly => false;
|
protected virtual bool ForceProcessAllAssembliesAndIgnoreAllPolicy => false;
|
||||||
|
|
||||||
protected abstract bool NeedObfuscateMethod(MethodDef method);
|
protected abstract bool NeedObfuscateMethod(MethodDef method);
|
||||||
|
|
||||||
public override void Process()
|
public override void Process()
|
||||||
{
|
{
|
||||||
var ctx = ObfuscationPassContext.Current;
|
var ctx = ObfuscationPassContext.Current;
|
||||||
var modules = NeedProcessNotObfuscatedAssembly ? ctx.obfuscatedAndNotObfuscatedModules : ctx.toObfuscatedModules;
|
var modules = ForceProcessAllAssembliesAndIgnoreAllPolicy ? ctx.obfuscatedAndNotObfuscatedModules : ctx.toObfuscatedModules;
|
||||||
NotObfuscatedMethodWhiteList whiteList = ctx.whiteList;
|
NotObfuscatedMethodWhiteList whiteList = ctx.whiteList;
|
||||||
ConfigurablePassPolicy passPolicy = ctx.passPolicy;
|
ConfigurablePassPolicy passPolicy = ctx.passPolicy;
|
||||||
foreach (ModuleDef mod in modules)
|
foreach (ModuleDef mod in modules)
|
||||||
{
|
{
|
||||||
if (whiteList.IsInWhiteList(mod) || !Support(passPolicy.GetAssemblyObfuscationPasses(mod)))
|
if (!ForceProcessAllAssembliesAndIgnoreAllPolicy && (whiteList.IsInWhiteList(mod) || !Support(passPolicy.GetAssemblyObfuscationPasses(mod))))
|
||||||
{
|
{
|
||||||
continue;
|
continue;
|
||||||
}
|
}
|
||||||
// ToArray to avoid modify list exception
|
// ToArray to avoid modify list exception
|
||||||
foreach (TypeDef type in mod.GetTypes().ToArray())
|
foreach (TypeDef type in mod.GetTypes().ToArray())
|
||||||
{
|
{
|
||||||
if (whiteList.IsInWhiteList(type) || !Support(passPolicy.GetTypeObfuscationPasses(type)))
|
if (!ForceProcessAllAssembliesAndIgnoreAllPolicy && (whiteList.IsInWhiteList(type) || !Support(passPolicy.GetTypeObfuscationPasses(type))))
|
||||||
{
|
{
|
||||||
continue;
|
continue;
|
||||||
}
|
}
|
||||||
// ToArray to avoid modify list exception
|
// ToArray to avoid modify list exception
|
||||||
foreach (MethodDef method in type.Methods.ToArray())
|
foreach (MethodDef method in type.Methods.ToArray())
|
||||||
{
|
{
|
||||||
if (!method.HasBody || ctx.whiteList.IsInWhiteList(method) || !Support(passPolicy.GetMethodObfuscationPasses(method)) || !NeedObfuscateMethod(method))
|
if (!method.HasBody || (!ForceProcessAllAssembliesAndIgnoreAllPolicy && (ctx.whiteList.IsInWhiteList(method) || !Support(passPolicy.GetMethodObfuscationPasses(method)) || !NeedObfuscateMethod(method))))
|
||||||
{
|
{
|
||||||
continue;
|
continue;
|
||||||
}
|
}
|
||||||
|
|
|
@ -778,5 +778,15 @@ namespace Obfuz.Utils
|
||||||
result.Append(")");
|
result.Append(")");
|
||||||
return result.ToString();
|
return result.ToString();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
public static bool HasObfuzIgnoreAttribute(IHasCustomAttribute obj)
|
||||||
|
{
|
||||||
|
return obj.CustomAttributes.Any(ca => ca.AttributeType.FullName == "Obfuz.ObfuzIgnoreAttribute");
|
||||||
|
}
|
||||||
|
|
||||||
|
public static bool HasCompilerGeneratedAttribute(IHasCustomAttribute obj)
|
||||||
|
{
|
||||||
|
return obj.CustomAttributes.Any(ca => ca.AttributeType.FullName == "System.Runtime.CompilerServices.CompilerGeneratedAttribute");
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
|
@ -0,0 +1,13 @@
|
||||||
|
using System;
|
||||||
|
using System.Collections.Generic;
|
||||||
|
using System.Linq;
|
||||||
|
using System.Text;
|
||||||
|
using System.Threading.Tasks;
|
||||||
|
|
||||||
|
namespace Obfuz
|
||||||
|
{
|
||||||
|
[AttributeUsage(AttributeTargets.All, Inherited = false, AllowMultiple = false)]
|
||||||
|
public class ObfuzIgnoreAttribute : Attribute
|
||||||
|
{
|
||||||
|
}
|
||||||
|
}
|
Loading…
Reference in New Issue