public NewTransformer(INewTransformerInfoWrapper infoWrapper, ICodeRewriter replacer, bool factoryMakeProtected) { ArgumentUtility.CheckNotNull("infoWrapper", infoWrapper); ArgumentUtility.CheckNotNull("replacer", replacer); _infoWrapper = infoWrapper; _replacer = replacer; _factoryMakeProtected = factoryMakeProtected; }
public NewTransformer(INewTransformerInfoWrapper infoWrapper, ICodeRewriter replacer, bool factoryMakeProtected) { ArgumentUtility.CheckNotNull ("infoWrapper", infoWrapper); ArgumentUtility.CheckNotNull ("replacer", replacer); _infoWrapper = infoWrapper; _replacer = replacer; _factoryMakeProtected = factoryMakeProtected; }
public bool CreateNewObjectMethod(AssemblyDefinition assembly, MethodDefinition templateMethod, IAssemblyTracker tracker, INewTransformerInfoWrapper infoWrapper) { MethodDefinition factoryMethod = null; if ((factoryMethod = infoWrapper.GetFactoryMethod (templateMethod, assembly, tracker)) != null) { if (factoryMethod.GenericParameters.Count != 1 || factoryMethod.Parameters.Count != 1 || !factoryMethod.IsStatic) throw new ArgumentException ("Factory method to create object does not have correct signature [public static T Create<T> (ParamList)]"); TypeReference returnType = templateMethod.DeclaringType; if (templateMethod.DeclaringType.HasGenericParameters) { returnType = new GenericInstanceType (templateMethod.DeclaringType); foreach (var a in templateMethod.DeclaringType.GenericParameters.ToArray ()) { returnType.GenericParameters.Add (a); ((GenericInstanceType) returnType).GenericArguments.Add (a); } } var importedFactoryMethod = templateMethod.Module.Import (factoryMethod); var genericInstanceMethod = new GenericInstanceMethod (importedFactoryMethod); genericInstanceMethod.GenericArguments.Add (templateMethod.DeclaringType); var paramlistDef = factoryMethod.Parameters[0].ParameterType.Resolve (); var importedParamListCreateMethod = templateMethod.Module.Import (SearchParamListFactoryMethod (paramlistDef, templateMethod)); if (importedParamListCreateMethod == null) throw new ArgumentException ("Factory method: no corresponding 'create' method could have been found. [argument count]"); var newObjectMethod = new MethodDefinition ( infoWrapper.GetWrapperMethodName (templateMethod), MethodAttributes.Public | MethodAttributes.Static, returnType ); var instructions = newObjectMethod.Body.Instructions; foreach (var param in templateMethod.Parameters) { newObjectMethod.Parameters.Add (param); instructions.Add (Instruction.Create (OpCodes.Ldarg, param)); } instructions.Add(Instruction.Create (OpCodes.Call, importedParamListCreateMethod)); instructions.Add(Instruction.Create (OpCodes.Call, genericInstanceMethod)); instructions.Add (Instruction.Create (OpCodes.Ret)); newObjectMethod.Body.OptimizeMacros (); newObjectMethod.IsHideBySig = true; templateMethod.DeclaringType.Methods.Add (newObjectMethod); return true; } return false; }
public bool ReplaceNewStatements(AssemblyDefinition containingAssembly, TypeDefinition containingType, MethodDefinition targetMethod, IAssemblyTracker tracker, INewTransformerInfoWrapper infoWrapper) { var isModified = false; if (targetMethod.HasBody) { var instructions = targetMethod.Body.Instructions; for (int i = 0; i < instructions.Count; i++) { if (instructions[i].OpCode == OpCodes.Newobj) { var constructor = (MethodReference) instructions[i].Operand; MethodDefinition factoryMethod = null; // Expected factory method signature: public static T Create<T> (ParamList) if (!constructor.Parameters.Any (p => p.ParameterType == containingType.Module.TypeSystem.IntPtr) && ((factoryMethod = infoWrapper.GetFactoryMethod (constructor, containingAssembly, tracker)) != null)) { isModified = true; if (factoryMethod.GenericParameters.Count != 1 || factoryMethod.Parameters.Count != 1 || !factoryMethod.IsStatic) throw new ArgumentException ("Factory method to create object does not have correct signature [public static T Create<T> (ParamList)]"); var importedFactoryMethod = containingType.Module.Import (factoryMethod); var genericInstanceMethod = new GenericInstanceMethod (importedFactoryMethod); genericInstanceMethod.GenericArguments.Add (constructor.DeclaringType); var paramlistDef = factoryMethod.Parameters[0].ParameterType.Resolve (); var importedParamListCreateMethod = containingType.Module.Import (SearchParamListFactoryMethod (paramlistDef, constructor)); if (importedParamListCreateMethod == null) throw new ArgumentException ("Factory method: no corresponding 'Create' method could have been found. [argument count]"); #region ILsample // 1: ldstr "Test" // 2: newObj // 3: next // ::---- becomes ->> // 1: ldstr "Test" // 2: nop (original call statement - kept as branch target) // 3: call ParamList.Create<string> (string) // 4: call ObjectFactory // 5: next #endregion targetMethod.Body.SimplifyMacros (); instructions[i].OpCode = OpCodes.Nop; instructions.Insert (i + 1, Instruction.Create (OpCodes.Call, importedParamListCreateMethod)); instructions.Insert (i + 2, Instruction.Create (OpCodes.Call, genericInstanceMethod)); targetMethod.Body.OptimizeMacros (); } } } } return isModified; }
public bool CreateNewObjectMethod(AssemblyDefinition assembly, MethodDefinition templateMethod, IAssemblyTracker tracker, INewTransformerInfoWrapper infoWrapper) { MethodDefinition factoryMethod = null; if ((factoryMethod = infoWrapper.GetFactoryMethod(templateMethod, assembly, tracker)) != null) { if (factoryMethod.GenericParameters.Count != 1 || factoryMethod.Parameters.Count != 1 || !factoryMethod.IsStatic) { throw new ArgumentException("Factory method to create object does not have correct signature [public static T Create<T> (ParamList)]"); } TypeReference returnType = templateMethod.DeclaringType; if (templateMethod.DeclaringType.HasGenericParameters) { returnType = new GenericInstanceType(templateMethod.DeclaringType); foreach (var a in templateMethod.DeclaringType.GenericParameters.ToArray()) { returnType.GenericParameters.Add(a); ((GenericInstanceType)returnType).GenericArguments.Add(a); } } var importedFactoryMethod = templateMethod.Module.Import(factoryMethod); var genericInstanceMethod = new GenericInstanceMethod(importedFactoryMethod); genericInstanceMethod.GenericArguments.Add(templateMethod.DeclaringType); var paramlistDef = factoryMethod.Parameters[0].ParameterType.Resolve(); var importedParamListCreateMethod = templateMethod.Module.Import(SearchParamListFactoryMethod(paramlistDef, templateMethod)); if (importedParamListCreateMethod == null) { throw new ArgumentException("Factory method: no corresponding 'create' method could have been found. [argument count]"); } var newObjectMethod = new MethodDefinition( infoWrapper.GetWrapperMethodName(templateMethod), MethodAttributes.Public | MethodAttributes.Static, returnType ); var instructions = newObjectMethod.Body.Instructions; foreach (var param in templateMethod.Parameters) { newObjectMethod.Parameters.Add(param); instructions.Add(Instruction.Create(OpCodes.Ldarg, param)); } instructions.Add(Instruction.Create(OpCodes.Call, importedParamListCreateMethod)); instructions.Add(Instruction.Create(OpCodes.Call, genericInstanceMethod)); instructions.Add(Instruction.Create(OpCodes.Ret)); newObjectMethod.Body.OptimizeMacros(); newObjectMethod.IsHideBySig = true; templateMethod.DeclaringType.Methods.Add(newObjectMethod); return(true); } return(false); }
public bool ReplaceNewStatements(AssemblyDefinition containingAssembly, TypeDefinition containingType, MethodDefinition targetMethod, IAssemblyTracker tracker, INewTransformerInfoWrapper infoWrapper) { var isModified = false; if (targetMethod.HasBody) { var instructions = targetMethod.Body.Instructions; for (int i = 0; i < instructions.Count; i++) { if (instructions[i].OpCode == OpCodes.Newobj) { var constructor = (MethodReference)instructions[i].Operand; MethodDefinition factoryMethod = null; // Expected factory method signature: public static T Create<T> (ParamList) if (!constructor.Parameters.Any(p => p.ParameterType == containingType.Module.TypeSystem.IntPtr) && ((factoryMethod = infoWrapper.GetFactoryMethod(constructor, containingAssembly, tracker)) != null)) { isModified = true; if (factoryMethod.GenericParameters.Count != 1 || factoryMethod.Parameters.Count != 1 || !factoryMethod.IsStatic) { throw new ArgumentException("Factory method to create object does not have correct signature [public static T Create<T> (ParamList)]"); } var importedFactoryMethod = containingType.Module.Import(factoryMethod); var genericInstanceMethod = new GenericInstanceMethod(importedFactoryMethod); genericInstanceMethod.GenericArguments.Add(constructor.DeclaringType); var paramlistDef = factoryMethod.Parameters[0].ParameterType.Resolve(); var importedParamListCreateMethod = containingType.Module.Import(SearchParamListFactoryMethod(paramlistDef, constructor)); if (importedParamListCreateMethod == null) { throw new ArgumentException("Factory method: no corresponding 'Create' method could have been found. [argument count]"); } #region ILsample // 1: ldstr "Test" // 2: newObj // 3: next // ::---- becomes ->> // 1: ldstr "Test" // 2: nop (original call statement - kept as branch target) // 3: call ParamList.Create<string> (string) // 4: call ObjectFactory // 5: next #endregion targetMethod.Body.SimplifyMacros(); instructions[i].OpCode = OpCodes.Nop; instructions.Insert(i + 1, Instruction.Create(OpCodes.Call, importedParamListCreateMethod)); instructions.Insert(i + 2, Instruction.Create(OpCodes.Call, genericInstanceMethod)); targetMethod.Body.OptimizeMacros(); } } } } return(isModified); }