de4dot-cex/de4dot.code/renamer/VariableNameState.cs
2011-09-22 04:55:30 +02:00

94 lines
3.6 KiB
C#

/*
Copyright (C) 2011 de4dot@gmail.com
This file is part of de4dot.
de4dot is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.
de4dot is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with de4dot. If not, see <http://www.gnu.org/licenses/>.
*/
using Mono.Cecil;
using de4dot.deobfuscators;
namespace de4dot.renamer {
// State when renaming type members
class VariableNameState {
protected TypeNames variableNameCreator = new VariableNameCreator(); // For fields and method args
protected TypeNames propertyNameCreator = new PropertyNameCreator();
protected INameCreator eventNameCreator = new NameCreator("Event_");
public INameCreator staticMethodNameCreator = new NameCreator("smethod_");
public INameCreator virtualMethodNameCreator = new NameCreator("vmethod_");
public INameCreator instanceMethodNameCreator = new NameCreator("method_");
protected INameCreator genericPropertyNameCreator = new NameCreator("Prop_");
public PinvokeNameCreator pinvokeNameCreator = new PinvokeNameCreator();
Func<string, bool> isValidName;
public Func<string, bool> IsValidName {
get { return isValidName; }
set { isValidName = value; }
}
public virtual VariableNameState clone() {
var rv = new VariableNameState();
cloneInit(rv);
return rv;
}
protected void cloneInit(VariableNameState variableNameState) {
variableNameState.variableNameCreator = variableNameCreator.clone();
variableNameState.propertyNameCreator = propertyNameCreator.clone();
variableNameState.eventNameCreator = eventNameCreator.clone();
variableNameState.staticMethodNameCreator = staticMethodNameCreator.clone();
variableNameState.virtualMethodNameCreator = virtualMethodNameCreator.clone();
variableNameState.instanceMethodNameCreator = instanceMethodNameCreator.clone();
variableNameState.genericPropertyNameCreator = genericPropertyNameCreator.clone();
variableNameState.pinvokeNameCreator = new PinvokeNameCreator();
variableNameState.isValidName = isValidName;
}
public string getNewPropertyName(PropertyDefinition propertyDefinition) {
var propType = propertyDefinition.PropertyType;
if (propType is GenericParameter)
return genericPropertyNameCreator.newName();
return propertyNameCreator.newName(propType);
}
public string getNewEventName(EventDefinition eventDefinition) {
return eventNameCreator.newName();
}
public string getNewFieldName(FieldDefinition field) {
return variableNameCreator.newName(field.FieldType);
}
public string getNewParamName(ParameterDefinition param) {
return variableNameCreator.newName(param.ParameterType);
}
}
class InterfaceVariableNameState : VariableNameState {
public InterfaceVariableNameState() {
propertyNameCreator = new GlobalInterfacePropertyNameCreator();
eventNameCreator = new GlobalNameCreator(new NameCreator("I_Event_"));
virtualMethodNameCreator = new GlobalNameCreator(new NameCreator("imethod_"));
genericPropertyNameCreator = new GlobalNameCreator(new NameCreator("I_Prop_"));
}
public override VariableNameState clone() {
var rv = new InterfaceVariableNameState();
cloneInit(rv);
return rv;
}
}
}