PHXH
/
obfuz
mirror of https://github.com/focus-creative-games/obfuz
2
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

新增 ReflectionCompatibilityDetector

1.x
walon 2025-06-13 21:00:58 +08:00
parent 28b841562c
commit 3f00d5ca91
4 changed files with 308 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

285
Editor/ObfusPasses/SymbolObfus/ReflectionCompatibilityDetector.cs Normal file
View File

@ -0,0 +1,285 @@
using dnlib.DotNet;
using dnlib.DotNet.Emit;
using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
using UnityEngine;
namespace Obfuz.ObfusPasses.SymbolObfus
{
public class ReflectionCompatibilityDetector
{
private readonly List<ModuleDef> _obfuscatedAndNotObfuscatedModules;
private readonly IObfuscationPolicy _renamePolicy;
public ReflectionCompatibilityDetector(List<ModuleDef> obfuscatedAndNotObfuscatedModules, IObfuscationPolicy renamePolicy)
{
_obfuscatedAndNotObfuscatedModules = obfuscatedAndNotObfuscatedModules;
_renamePolicy = renamePolicy;
}
public void Analyze()
{
foreach (ModuleDef mod in _obfuscatedAndNotObfuscatedModules)
{
foreach (TypeDef type in mod.GetTypes())
{
foreach (MethodDef method in type.Methods)
{
AnalyzeMethod(method);
}
}
}
}
private MethodDef _curCallingMethod;
private IList<Instruction> _curInstructions;
private int _curInstIndex;
private void AnalyzeMethod(MethodDef method)
{
if (!method.HasBody)
{
return;
}
_curCallingMethod = method;
_curInstructions = method.Body.Instructions;
_curInstIndex = 0;
for (int n = _curInstructions.Count; _curInstIndex < n; _curInstIndex++)
{
var inst = _curInstructions[_curInstIndex];
switch (inst.OpCode.Code)
{
case Code.Call:
{
AnalyzeCall(inst.Operand as IMethod);
break;
}
case Code.Callvirt:
{
ITypeDefOrRef constrainedType = null;
if (_curInstIndex > 0)
{
var prevInst = _curInstructions[_curInstIndex - 1];
if (prevInst.OpCode.Code == Code.Constrained)
{
constrainedType = prevInst.Operand as ITypeDefOrRef;
}
}
AnalyzeCallvir(inst.Operand as IMethod, constrainedType);
break;
}
}
}
}
private ITypeDefOrRef FindLatestTypeOf(int backwardFindInstructionCount)
{
// find sequence ldtoken <type>;
for (int i = 2; i <= backwardFindInstructionCount; i++)
{
int index = _curInstIndex - i;
if (index < 0)
{
return null;
}
Instruction inst1 = _curInstructions[index];
Instruction inst2 = _curInstructions[index + 1];
if (inst1.OpCode.Code == Code.Ldtoken && inst2.OpCode.Code == Code.Call)
{
if (!(inst1.Operand is ITypeDefOrRef typeDefOrRef))
{
continue;
}
IMethod method = inst2.Operand as IMethod;
if (method.Name == "GetTypeFromHandle" && method.DeclaringType.FullName == "System.Type")
{
// Ldtoken <type>; Call System.Type.GetTypeFromHandle(System.RuntimeTypeHandle handle)
return typeDefOrRef;
}
}
}
return null;
}
private void AnalyzeCall(IMethod calledMethod)
{
TypeDef callType = calledMethod.DeclaringType.ResolveTypeDef();
if (callType == null)
{
return;
}
switch (callType.FullName)
{
case "System.Enum":
{
AnalyzeEnum(calledMethod, callType);
break;
}
case "System.Type":
{
AnalyzeGetType(calledMethod, callType);
break;
}
case "System.Reflection.Assembly":
{
if (calledMethod.Name == "GetType")
{
AnalyzeGetType(calledMethod, callType);
}
break;
}
}
}
private void AnalyzeCallvir(IMethod calledMethod, ITypeDefOrRef constrainedType)
{
TypeDef callType = calledMethod.DeclaringType.ResolveTypeDef();
if (callType == null)
{
return;
}
string calledMethodName = calledMethod.Name;
switch (callType.FullName)
{
case "System.Object":
{
if (calledMethodName == "ToString")
{
if (constrainedType != null)
{
TypeDef enumTypeDef = constrainedType.ResolveTypeDef();
if (enumTypeDef != null && enumTypeDef.IsEnum && _renamePolicy.NeedRename(enumTypeDef))
{
Debug.LogError($"[ReflectionCompatibilityDetector] Reflection compatibility issue in {_curCallingMethod}: Enum.ToString() T:{enumTypeDef.FullName} is renamed.");
}
}
}
break;
}
case "System.Type":
{
AnalyzeGetType(calledMethod, callType);
break;
}
}
}
private TypeSig GetMethodGenericParameter(IMethod method)
{
if (method is MethodSpec ms)
{
return ms.GenericInstMethodSig.GenericArguments.FirstOrDefault();
}
else
{
return null;
}
}
private void AnalyzeEnum(IMethod method, TypeDef typeDef)
{
const int extraSearchInstructionCount = 3;
TypeDef parseType = GetMethodGenericParameter(method)?.ToTypeDefOrRef()?.ResolveTypeDef();
switch (method.Name)
{
case "Parse":
{
if (parseType != null)
{
// Enum.Parse<T>(string name) or Enum.Parse<T>(string name, bool caseInsensitive)
if (_renamePolicy.NeedRename(parseType))
{
Debug.LogError($"[ReflectionCompatibilityDetector] Reflection compatibility issue in {_curCallingMethod}: Enum.Parse<T> T:{parseType.FullName} is renamed.");
}
}
else
{
// Enum.Parse(Type type, string name) or Enum.Parse(Type type, string name, bool ignoreCase)
TypeDef enumType = FindLatestTypeOf(method.GetParamCount() + extraSearchInstructionCount)?.ResolveTypeDef();
if (enumType != null && enumType.IsEnum && _renamePolicy.NeedRename(enumType))
{
Debug.LogError($"[ReflectionCompatibilityDetector] Reflection compatibility issue in {_curCallingMethod}: Enum.Parse argument type:{enumType.FullName} is renamed.");
}
else
{
Debug.LogWarning($"[ReflectionCompatibilityDetector] Reflection compatibility issue in {_curCallingMethod}: Enum.Parse argument `type` should not be renamed.");
}
}
break;
}
case "TryParse":
{
if (parseType != null)
{
// Enum.TryParse<T>(string name, out T result) or Enum.TryParse<T>(string name, bool ignoreCase, out T result)
if (_renamePolicy.NeedRename(parseType))
{
Debug.LogError($"[ReflectionCompatibilityDetector] Reflection compatibility issue in {_curCallingMethod}: Enum.TryParse<T> T:{parseType.FullName} is renamed.");
}
}
else
{
throw new Exception("impossible");
}
break;
}
case "GetName":
{
// Enum.GetName(Type type, object value)
TypeDef enumType = FindLatestTypeOf(method.GetParamCount() + extraSearchInstructionCount)?.ResolveTypeDef();
if (enumType != null && enumType.IsEnum && enumType.Fields.Any(f => _renamePolicy.NeedRename(f)))
{
Debug.LogError($"[ReflectionCompatibilityDetector] Reflection compatibility issue in {_curCallingMethod}: Enum.GetName field of type:{enumType.FullName} is renamed.");
}
else
{
Debug.LogWarning($"[ReflectionCompatibilityDetector] Reflection compatibility issue in {_curCallingMethod}: Enum.GetName field of argument `type` should not be renamed.");
}
break;
}
case "GetNames":
{
// Enum.GetNames(Type type)
TypeDef enumType = FindLatestTypeOf(method.GetParamCount() + extraSearchInstructionCount)?.ResolveTypeDef();
if (enumType != null && enumType.IsEnum && enumType.Fields.Any(f => _renamePolicy.NeedRename(f)))
{
Debug.LogError($"[ReflectionCompatibilityDetector] Reflection compatibility issue in {_curCallingMethod}: Enum.GetNams field of type:{enumType.FullName} is renamed.");
}
else
{
Debug.LogWarning($"[ReflectionCompatibilityDetector] Reflection compatibility issue in {_curCallingMethod}: Enum.GetNames field of argument `type` should not be renamed.");
}
break;
}
}
}
private void AnalyzeGetType(IMethod method, TypeDef declaringType)
{
switch (method.Name)
{
case "GetType":
{
Debug.LogWarning($"[ReflectionCompatibilityDetector] Reflection compatibility issue in {_curCallingMethod}: Type.GetType argument `typeName` should not be renamed.");
break;
}
case "GetField":
case "GetFields":
case "GetMethod":
case "GetMethods":
case "GetProperty":
case "GetProperties":
case "GetEvent":
case "GetEvents":
case "GetMembers":
{
Debug.LogWarning($"[ReflectionCompatibilityDetector] Reflection compatibility issue in {_curCallingMethod}: called method:{method} the members of type should not be renamed.");
break;
}
}
}
}
}

11
Editor/ObfusPasses/SymbolObfus/ReflectionCompatibilityDetector.cs.meta Normal file
View File

@ -0,0 +1,11 @@
fileFormatVersion: 2
guid: b98c97a4d1db5d945bce0fdd2bd09202
MonoImporter:
externalObjects: {}
serializedVersion: 2
defaultReferences: []
executionOrder: 0
icon: {instanceID: 0}
userData:
assetBundleName:
assetBundleVariant:

7
Editor/ObfusPasses/SymbolObfus/SymbolRename.cs
View File

@ -15,6 +15,7 @@ namespace Obfuz.ObfusPasses.SymbolObfus
public class SymbolRename
{
private readonly bool _useConsistentNamespaceObfuscation;
private readonly bool _detectReflectionCompatibility;
private readonly List<string> _obfuscationRuleFiles;
private readonly string _mappingXmlPath;
@ -42,6 +43,7 @@ namespace Obfuz.ObfusPasses.SymbolObfus
public SymbolRename(SymbolObfuscationSettingsFacade settings)
{
_useConsistentNamespaceObfuscation = settings.useConsistentNamespaceObfuscation;
_detectReflectionCompatibility = settings.detectReflectionCompatibility;
_mappingXmlPath = settings.symbolMappingFile;
_obfuscationRuleFiles = settings.ruleFiles.ToList();
_renameRecordMap = new RenameRecordMap(settings.symbolMappingFile, settings.debug, settings.keepUnknownSymbolInSymbolMappingFile);
@ -178,6 +180,11 @@ namespace Obfuz.ObfusPasses.SymbolObfus
{
_renameRecordMap.Init(_toObfuscatedModules, _nameMaker);
PrecomputeNeedRename();
if (_detectReflectionCompatibility)
{
var reflectionCompatibilityDetector = new ReflectionCompatibilityDetector(_obfuscatedAndNotObfuscatedModules, _renamePolicy);
reflectionCompatibilityDetector.Analyze();
}
RenameTypes();
RenameFields();
RenameMethods();

5
Editor/Settings/SymbolObfuscationSettings.cs
View File

@ -11,6 +11,7 @@ namespace Obfuz.Settings
public bool debug;
public string obfuscatedNamePrefix;
public bool useConsistentNamespaceObfuscation;
public bool detectReflectionCompatibility;
public bool keepUnknownSymbolInSymbolMappingFile;
public string symbolMappingFile;
public List<string> ruleFiles;
@ -28,6 +29,9 @@ namespace Obfuz.Settings
[Tooltip("obfuscate same namespace to one name")]
public bool useConsistentNamespaceObfuscation = true;
[Tooltip("detect reflection compatibility, if true, will detect if the obfuscated name is compatibility with reflection, such as Type.GetType(), Enum.Parse(), etc.")]
public bool detectReflectionCompatibility = true;
[Tooltip("keep unknown symbol in symbol mapping file, if false, unknown symbol will be removed from mapping file")]
public bool keepUnknownSymbolInSymbolMappingFile = true;
@ -55,6 +59,7 @@ namespace Obfuz.Settings
debug = debug,
obfuscatedNamePrefix = obfuscatedNamePrefix,
useConsistentNamespaceObfuscation = useConsistentNamespaceObfuscation,
detectReflectionCompatibility = detectReflectionCompatibility,
keepUnknownSymbolInSymbolMappingFile = keepUnknownSymbolInSymbolMappingFile,
symbolMappingFile = GetSymbolMappingFile(),
ruleFiles = ruleFiles?.ToList() ?? new List<string>(),