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

优化 Rename Property

backup
walon 2025-04-15 12:42:09 +08:00
parent 99034725d1
commit d6dd1a81f7
1 changed files with 64 additions and 37 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

101
Editor/Rename/SymbolRename.cs
View File

@ -428,9 +428,59 @@ namespace Obfuz
Debug.Log("Rename methods end"); Debug.Log("Rename methods end");
} }
class RefPropertyMetas
{
public List<CustomAttribute> customAttributes = new List<CustomAttribute>();
}
private void BuildHierarchyProperties(TypeDef type, List<PropertyDef> properties)
{
while (type != null)
{
properties.AddRange(type.Properties);
type = MetaUtil.GetBaseTypeDef(type);
}
}
private void BuildRefPropertyMetasMap(Dictionary<PropertyDef, RefPropertyMetas> refPropertyMetasMap)
{
foreach (var e in _refTypeRefMetasMap)
{
TypeDef typeDef = e.Key;
var hierarchyProperties = new List<PropertyDef>();
BuildHierarchyProperties(typeDef, hierarchyProperties);
RefTypeDefMetas typeDefMetas = e.Value;
foreach (CustomAttribute ca in typeDefMetas.customAttributes)
{
foreach (var arg in ca.NamedArguments)
{
if (arg.IsField)
{
continue;
}
foreach (PropertyDef field in hierarchyProperties)
{
if (field.Name == arg.Name)
{
if (!refPropertyMetasMap.TryGetValue(field, out var fieldMetas))
{
fieldMetas = new RefPropertyMetas();
refPropertyMetasMap.Add(field, fieldMetas);
}
fieldMetas.customAttributes.Add(ca);
break;
}
}
}
}
}
}
private void RenameProperties() private void RenameProperties()
{ {
Debug.Log("Rename properties begin"); Debug.Log("Rename properties begin");
var refPropertyMetasMap = new Dictionary<PropertyDef, RefPropertyMetas>();
BuildRefPropertyMetasMap(refPropertyMetasMap);
foreach (ObfuzAssemblyInfo ass in _ctx.assemblies) foreach (ObfuzAssemblyInfo ass in _ctx.assemblies)
{ {
foreach (TypeDef type in ass.module.GetTypes()) foreach (TypeDef type in ass.module.GetTypes())
@ -439,7 +489,7 @@ namespace Obfuz
{ {
if (_renamePolicy.NeedRename(property)) if (_renamePolicy.NeedRename(property))
{ {
Rename(property); Rename(property, refPropertyMetasMap.TryGetValue(property, out var refPropertyMeta) ? refPropertyMeta : null);
} }
else else
{ {
@ -533,38 +583,6 @@ namespace Obfuz
Debug.Log($"rename typedef. assembly:{type.Module.Name} oldName:{oldFullName} => newName:{newFullName}"); Debug.Log($"rename typedef. assembly:{type.Module.Name} oldName:{oldFullName} => newName:{newFullName}");
} }
private void RenameFieldNameInCustomAttributes(ModuleDefMD referenceMeMod, ModuleDefMD mod, TypeDef declaringType, string oldFieldOrPropertyName, string newName)
{
List<CustomAttributeInfo> customAttributes = _customAttributeArgumentsWithTypeByMods[referenceMeMod];
foreach (CustomAttributeInfo cai in customAttributes)
{
CustomAttribute oldAttr = cai.customAttributes[cai.index];
if (MetaUtil.GetTypeDefOrGenericTypeBaseThrowException(oldAttr.Constructor.DeclaringType) != declaringType)
{
continue;
}
bool anyChange = false;
if (cai.namedArguments != null)
{
foreach (CANamedArgument arg in cai.namedArguments)
{
if (arg.Name == oldFieldOrPropertyName)
{
anyChange = true;
arg.Name = newName;
}
}
}
if (anyChange)
{
cai.customAttributes[cai.index] = new CustomAttribute(oldAttr.Constructor,
cai.arguments != null ? cai.arguments : oldAttr.ConstructorArguments,
cai.namedArguments != null ? cai.namedArguments : oldAttr.NamedArguments);
}
}
}
private void Rename(FieldDef field, RefFieldMetas fieldMetas) private void Rename(FieldDef field, RefFieldMetas fieldMetas)
{ {
string oldName = field.Name; string oldName = field.Name;
@ -700,14 +718,23 @@ namespace Obfuz
_renameRecordMap.AddRenameRecord(eventDef, oldName, newName); _renameRecordMap.AddRenameRecord(eventDef, oldName, newName);
} }
private void Rename(PropertyDef property) private void Rename(PropertyDef property, RefPropertyMetas refPropertyMetas)
{ {
string oldName = property.Name; string oldName = property.Name;
string newName = _nameMaker.GetNewName(property, oldName); string newName = _nameMaker.GetNewName(property, oldName);
ModuleDefMD mod = (ModuleDefMD)property.DeclaringType.Module;
foreach (ObfuzAssemblyInfo ass in GetReferenceMeAssemblies(mod)) if (refPropertyMetas != null)
{ {
RenameFieldNameInCustomAttributes(ass.module, mod, property.DeclaringType, oldName, newName); foreach (var ca in refPropertyMetas.customAttributes)
{
foreach (var arg in ca.NamedArguments)
{
if (arg.Name == oldName)
{
arg.Name = newName;
}
}
}
} }
property.Name = newName; property.Name = newName;
_renameRecordMap.AddRenameRecord(property, oldName, newName); _renameRecordMap.AddRenameRecord(property, oldName, newName);