添加 CacheRenamePolicy.cs

backup
walon 2025-04-18 08:50:51 +08:00
parent 2989cdef97
commit 6671685893
3 changed files with 90 additions and 1 deletions

View File

@ -40,7 +40,7 @@ namespace Obfuz
_assemblyCache = new AssemblyCache(new PathAssemblyResolver(options.AssemblySearchDirs.ToArray())); _assemblyCache = new AssemblyCache(new PathAssemblyResolver(options.AssemblySearchDirs.ToArray()));
_obfuscateRuleConfig = new ObfuscateRuleConfig(); _obfuscateRuleConfig = new ObfuscateRuleConfig();
_obfuscateRuleConfig.LoadXmls(options.ObfuscationRuleFiles); _obfuscateRuleConfig.LoadXmls(options.ObfuscationRuleFiles);
_renamePolicy = new CombineRenamePolicy(new SystemRenamePolicy(), new UnityRenamePolicy(), _obfuscateRuleConfig); _renamePolicy = new CacheRenamePolicy(new CombineRenamePolicy(new SystemRenamePolicy(), new UnityRenamePolicy(), _obfuscateRuleConfig));
//_nameMaker = new TestNameMaker(); //_nameMaker = new TestNameMaker();
_nameMaker = NameMakerFactory.CreateNameMakerBaseASCIICharSet(); _nameMaker = NameMakerFactory.CreateNameMakerBaseASCIICharSet();

View File

@ -0,0 +1,88 @@
using dnlib.DotNet;
using System.Collections.Generic;
using System.Linq;
namespace Obfuz.Rename
{
public class CacheRenamePolicy : RenamePolicyBase
{
private readonly IRenamePolicy _underlyingPolicy;
private readonly Dictionary<object, bool> _computeCache = new Dictionary<object, bool>();
public CacheRenamePolicy(IRenamePolicy underlyingPolicy)
{
_underlyingPolicy = underlyingPolicy;
}
public override bool NeedRename(ModuleDefMD mod)
{
if (!_computeCache.TryGetValue(mod, out var value))
{
value = _underlyingPolicy.NeedRename(mod);
_computeCache.Add(mod, value);
}
return value;
}
public override bool NeedRename(TypeDef typeDef)
{
if (!_computeCache.TryGetValue(typeDef, out var value))
{
value = _underlyingPolicy.NeedRename(typeDef);
_computeCache[typeDef] = value;
}
return value;
}
public override bool NeedRename(MethodDef methodDef)
{
if (!_computeCache.TryGetValue(methodDef, out var value))
{
value = _underlyingPolicy.NeedRename(methodDef);
_computeCache[methodDef] = value;
}
return value;
}
public override bool NeedRename(FieldDef fieldDef)
{
if (!_computeCache.TryGetValue(fieldDef, out var value))
{
value = _underlyingPolicy.NeedRename(fieldDef);
_computeCache[fieldDef] = value;
}
return value;
}
public override bool NeedRename(PropertyDef propertyDef)
{
if (!_computeCache.TryGetValue(propertyDef, out var value))
{
value = _underlyingPolicy.NeedRename(propertyDef);
_computeCache[propertyDef] = value;
}
return value;
}
public override bool NeedRename(EventDef eventDef)
{
if (!_computeCache.TryGetValue(eventDef, out var value))
{
value = _underlyingPolicy.NeedRename(eventDef);
_computeCache[eventDef] = value;
}
return value;
}
public override bool NeedRename(ParamDef paramDef)
{
if (!_computeCache.TryGetValue(paramDef, out var value))
{
value = _underlyingPolicy.NeedRename(paramDef);
_computeCache[paramDef] = value;
}
return value;
}
}
}

View File

@ -6,6 +6,7 @@ namespace Obfuz.Rename
public class CombineRenamePolicy : IRenamePolicy public class CombineRenamePolicy : IRenamePolicy
{ {
private readonly IRenamePolicy[] _policies; private readonly IRenamePolicy[] _policies;
public CombineRenamePolicy(params IRenamePolicy[] policies) public CombineRenamePolicy(params IRenamePolicy[] policies)
{ {
_policies = policies; _policies = policies;