新增基于 WordSet的 NameMaker

backup
walon 2025-04-15 18:30:17 +08:00
parent 1d1b257640
commit dcf7fd3ae5
4 changed files with 91 additions and 26 deletions

View File

@ -34,7 +34,8 @@ namespace Obfuz
_options = options;
_assemblyCache = new AssemblyCache(new PathAssemblyResolver(options.AssemblySearchDirs.ToArray()));
_renamePolicy = new CombineRenamePolicy(new SystemRenamePolicy(), new UnityRenamePolicy(), new XmlConfigRenamePolicy());
_nameMaker = new TestNameMaker();
//_nameMaker = new TestNameMaker();
_nameMaker = NameMakerFactory.CreateNameMakerBaseASCIICharSet();
}
public void DoIt()

View File

@ -0,0 +1,23 @@
using dnlib.DotNet;
namespace Obfuz.Rename
{
public interface INameMaker
{
string GetNewName(ModuleDefMD mod, string originalName);
string GetNewName(TypeDef typeDef, string originalName);
string GetNewNamespace(TypeDef typeDef, string originalNamespace);
string GetNewName(MethodDef methodDef, string originalName);
string GetNewName(ParamDef param, string originalName);
string GetNewName(FieldDef fieldDef, string originalName);
string GetNewName(PropertyDef propertyDef, string originalName);
string GetNewName(EventDef eventDef, string originalName);
}
}

View File

@ -7,72 +7,90 @@ using System.Threading.Tasks;
namespace Obfuz.Rename
{
public interface INameMaker
public class NameScope
{
string GetNewName(ModuleDefMD mod, string originalName);
private readonly List<string> _wordSet;
private int _nextIndex;
string GetNewName(TypeDef typeDef, string originalName);
public NameScope(List<string> wordSet)
{
_wordSet = wordSet;
_nextIndex = 0;
}
string GetNewNamespace(TypeDef typeDef, string originalNamespace);
string GetNewName(MethodDef methodDef, string originalName);
string GetNewName(ParamDef param, string originalName);
string GetNewName(FieldDef fieldDef, string originalName);
string GetNewName(PropertyDef propertyDef, string originalName);
string GetNewName(EventDef eventDef, string originalName);
public string GetNewName(string originalName)
{
if (_nextIndex >= _wordSet.Count)
throw new InvalidOperationException("No more names available in the word set.");
string newName = _wordSet[_nextIndex++];
return newName;
}
}
public class NameMaker : INameMaker
{
private int _nextIndex;
private readonly List<string> _wordSet;
private string GetDefaultNewName(string originName)
private readonly Dictionary<object, NameScope> _nameScopes = new Dictionary<object, NameScope>();
public NameMaker(List<string> wordSet)
{
return $"{originName}>{_nextIndex++}";
_wordSet = wordSet;
}
private NameScope GetNameScope(object key)
{
if (!_nameScopes.TryGetValue(key, out var nameScope))
{
nameScope = new NameScope(_wordSet);
_nameScopes[key] = nameScope;
}
return nameScope;
}
private string GetDefaultNewName(object scope, string originName)
{
return GetNameScope(scope).GetNewName(originName);
}
public string GetNewName(ModuleDefMD mod, string originalName)
{
return GetDefaultNewName(originalName);
return GetDefaultNewName(this, originalName);
}
public string GetNewNamespace(TypeDef typeDef, string originalNamespace)
{
return GetDefaultNewName(originalNamespace);
return GetDefaultNewName(typeDef.Module, originalNamespace);
}
public string GetNewName(TypeDef typeDef, string originalName)
{
return GetDefaultNewName(originalName);
return GetDefaultNewName(typeDef.Module, originalName);
}
public string GetNewName(MethodDef methodDef, string originalName)
{
return GetDefaultNewName(originalName);
return GetDefaultNewName(methodDef.DeclaringType, originalName);
}
public string GetNewName(ParamDef param, string originalName)
{
return GetDefaultNewName(originalName);
return "1";
}
public string GetNewName(FieldDef fieldDef, string originalName)
{
return GetDefaultNewName(originalName);
return GetDefaultNewName(fieldDef.DeclaringType, originalName);
}
public string GetNewName(PropertyDef propertyDef, string originalName)
{
return GetDefaultNewName(originalName);
return GetDefaultNewName(propertyDef.DeclaringType, originalName);
}
public string GetNewName(EventDef eventDef, string originalName)
{
return GetDefaultNewName(originalName);
return GetDefaultNewName(eventDef.DeclaringType, originalName);
}
}
}

View File

@ -0,0 +1,23 @@
using System;
using System.Collections.Generic;
using System.Linq;
using System.Runtime.Remoting.Messaging;
using System.Text;
using System.Threading.Tasks;
namespace Obfuz.Rename
{
public static class NameMakerFactory
{
public static INameMaker CreateNameMakerBaseASCIICharSet()
{
var words = new List<string>();
for (int i = 0; i < 26; i++)
{
words.Add(((char)('a' + i)).ToString());
words.Add(((char)('A' + i)).ToString());
}
return new NameMaker(words);
}
}
}