public void Weave(
MethodDefinition method,
CustomAttribute aspect,
AspectMethods overriddenAspectMethods,
ModuleDefinition moduleDefinition,
TypeReference type,
byte[] unweavedAssembly)
{
if (overriddenAspectMethods == AspectMethods.None)
{
return;
}
if (overriddenAspectMethods.HasFlag(AspectMethods.CompileTimeValidate))
{
var asm = AppDomain.CurrentDomain.Load(unweavedAssembly);
string methodName = method.Name;
string[] methodParams = method.Parameters.Select(p => p.ParameterType.FullName).ToArray();
string[] ctorParams = aspect.Constructor.Parameters.Select(p => p.ParameterType.FullName).ToArray();
try
{
var typeInfo = asm.GetTypes().FirstOrDefault(t => t.FullName == type.FullName);
if (typeInfo == null)
{
throw new InvalidOperationException(String.Format("Could not find type '{0}'.", type.FullName));
}
Type[] parameters = methodParams.Select(Type.GetType).ToArray();
var methodInfo = typeInfo.GetMethod(methodName, System.Reflection.BindingFlags.Public | System.Reflection.BindingFlags.NonPublic | System.Reflection.BindingFlags.Static | System.Reflection.BindingFlags.Instance, null, parameters, new ParameterModifier[0]);
Type aspectType = asm.GetTypes().FirstOrDefault(t => t.FullName == aspect.AttributeType.FullName);
var ctorInfo = aspectType.GetConstructor(ctorParams.Select(Type.GetType).ToArray());
if (ctorInfo == null)
{
throw new InvalidOperationException("Could not find constructor for aspect.");
}
object[] ctorArgs = GetRuntimeAttributeArgs(aspect.ConstructorArguments);
var aspectInstance = Activator.CreateInstance(aspectType, ctorArgs) as OnMethodBoundaryAspect;
if (aspectInstance == null)
{
throw new InvalidOperationException("Could not create aspect.");
}
foreach (var fieldSetter in aspect.Fields)
{
var field = aspectType.GetField(fieldSetter.Name);
if (field == null)
{
throw new InvalidOperationException(String.Format("Could not find field named {0}", fieldSetter.Name));
}
field.SetValue(aspectInstance, fieldSetter.Argument.Value);
}
foreach (var propSetter in aspect.Properties)
{
var prop = aspectType.GetProperty(propSetter.Name);
if (prop == null)
{
throw new InvalidOperationException(String.Format("Could not find property named {0}", propSetter.Name));
}
prop.SetValue(aspectInstance, propSetter.Argument.Value);
}
if (!aspectInstance.CompileTimeValidate(methodInfo))
{
return;
}
}
catch (Exception e)
{
throw new InvalidOperationException("Error while trying to compile-time validate.", e);
}
}
var creator = new InstructionBlockChainCreator(method, aspect.AttributeType, moduleDefinition, WeaveCounter);
_methodBodyChanger = new MethodBodyPatcher(method.Name, method);
var saveReturnValue = creator.SaveReturnValue();
var loadReturnValue = creator.LoadValueOnStack(saveReturnValue);
_methodBodyChanger.Unify(saveReturnValue, loadReturnValue);
if (WeaveCounter == 0)
{
_createArgumentsArray = creator.CreateMethodArgumentsArray();
}
var createMethodExecutionArgsInstance = creator.CreateMethodExecutionArgsInstance(_createArgumentsArray, type);
_methodBodyChanger.AddCreateMethodExecutionArgs(createMethodExecutionArgsInstance);
var createAspectInstance = creator.LoadAspectInstance(aspect, type);
if (overriddenAspectMethods.HasFlag(AspectMethods.OnEntry))
{
var callAspectOnEntry = creator.CallAspectOnEntry(createAspectInstance,
createMethodExecutionArgsInstance);
var loadExecuteBodyVar = creator.LoadExecuteBodyVariable(createMethodExecutionArgsInstance, out NamedInstructionBlockChain executeBodyVar);
executeBodyVar = creator.LoadValueOnStack(executeBodyVar);
_methodBodyChanger.AddOnEntryCall(createAspectInstance, callAspectOnEntry, loadExecuteBodyVar, executeBodyVar);
}
if (overriddenAspectMethods.HasFlag(AspectMethods.OnExit))
{
var setMethodExecutionArgsReturnValue =
creator.SetMethodExecutionArgsReturnValue(createMethodExecutionArgsInstance, loadReturnValue);
var callAspectOnExit = creator.CallAspectOnExit(createAspectInstance,
createMethodExecutionArgsInstance);
var readReturnValue = creator.ReadReturnValue(createMethodExecutionArgsInstance, loadReturnValue);
_methodBodyChanger.AddOnExitCall(createAspectInstance, callAspectOnExit,
setMethodExecutionArgsReturnValue, readReturnValue);
}
if (overriddenAspectMethods.HasFlag(AspectMethods.OnException))
{
var setMethodExecutionArgsExceptionFromStack =
creator.SetMethodExecutionArgsExceptionFromStack(createMethodExecutionArgsInstance);
var callAspectOnException = creator.CallAspectOnException(createAspectInstance,
createMethodExecutionArgsInstance);
_methodBodyChanger.AddOnExceptionCall(createAspectInstance, callAspectOnException,
setMethodExecutionArgsExceptionFromStack);
}
if (_methodBodyChanger.HasMultipleReturnAndEndsWithThrow)
{
_methodBodyChanger.ReplaceThrowAtEndOfRealBodyWithReturn();
}
else if (_methodBodyChanger.EndsWithThrow)
{
var saveThrownException = creator.SaveThrownException();
var loadThrownException = creator.LoadValueOnStack(saveThrownException);
var loadThrownException2 = creator.LoadValueOnStack(saveThrownException);
_methodBodyChanger.FixThrowAtEndOfRealBody(
saveThrownException,
loadThrownException,
loadThrownException2);
}
_methodBodyChanger.OptimizeBody();
Catel.Fody.CecilExtensions.UpdateDebugInfo(method);
WeaveCounter++;
}
public void Weave(
MethodDefinition method,
CustomAttribute aspect,
AspectMethods overriddenAspectMethods,
ModuleDefinition moduleDefinition)
{
if (overriddenAspectMethods == AspectMethods.None)
{
return;
}
var creator = new InstructionBlockChainCreator(method, aspect.AttributeType, moduleDefinition, WeaveCounter);
_methodBodyChanger = new MethodBodyPatcher(method.Name, method);
var saveReturnValue = creator.SaveReturnValue();
var loadReturnValue = creator.LoadValueOnStack(saveReturnValue);
_methodBodyChanger.Unify(saveReturnValue, loadReturnValue);
if (WeaveCounter == 0)
{
_createArgumentsArray = creator.CreateMethodArgumentsArray();
}
var createMethodExecutionArgsInstance = creator.CreateMethodExecutionArgsInstance(_createArgumentsArray);
_methodBodyChanger.AddCreateMethodExecutionArgs(createMethodExecutionArgsInstance);
var createAspectInstance = creator.CreateAspectInstance(aspect);
if (overriddenAspectMethods.HasFlag(AspectMethods.OnEntry))
{
var callAspectOnEntry = creator.CallAspectOnEntry(createAspectInstance,
createMethodExecutionArgsInstance);
_methodBodyChanger.AddOnEntryCall(createAspectInstance, callAspectOnEntry);
}
if (overriddenAspectMethods.HasFlag(AspectMethods.OnExit))
{
var setMethodExecutionArgsReturnValue =
creator.SetMethodExecutionArgsReturnValue(createMethodExecutionArgsInstance, loadReturnValue);
var callAspectOnExit = creator.CallAspectOnExit(createAspectInstance,
createMethodExecutionArgsInstance);
_methodBodyChanger.AddOnExitCall(createAspectInstance, callAspectOnExit,
setMethodExecutionArgsReturnValue);
}
if (overriddenAspectMethods.HasFlag(AspectMethods.OnException))
{
var setMethodExecutionArgsExceptionFromStack =
creator.SetMethodExecutionArgsExceptionFromStack(createMethodExecutionArgsInstance);
var callAspectOnException = creator.CallAspectOnException(createAspectInstance,
createMethodExecutionArgsInstance);
_methodBodyChanger.AddOnExceptionCall(createAspectInstance, callAspectOnException,
setMethodExecutionArgsExceptionFromStack);
}
if (_methodBodyChanger.EndsWithThrow)
{
var saveThrownException = creator.SaveThrownException();
var loadThrownException = creator.LoadValueOnStack(saveThrownException);
var loadThrownException2 = creator.LoadValueOnStack(saveThrownException);
_methodBodyChanger.FixThrowAtEndOfRealBody(
saveThrownException,
loadThrownException,
loadThrownException2);
}
_methodBodyChanger.OptimizeBody();
WeaveCounter++;
}
public void Weave(
MethodDefinition method,
CustomAttribute aspect,
AspectMethods overriddenAspectMethods,
ModuleDefinition moduleDefinition)
{
if (overriddenAspectMethods == AspectMethods.None)
return;
var creator = new InstructionBlockChainCreator(method, aspect.AttributeType, moduleDefinition, WeaveCounter);
_methodBodyChanger = new MethodBodyPatcher(method.Name, method);
var saveReturnValue = creator.SaveReturnValue();
var loadReturnValue = creator.LoadValueOnStack(saveReturnValue);
_methodBodyChanger.Unify(saveReturnValue, loadReturnValue);
if (WeaveCounter == 0)
_createArgumentsArray = creator.CreateMethodArgumentsArray();
var createMethodExecutionArgsInstance = creator.CreateMethodExecutionArgsInstance(_createArgumentsArray);
_methodBodyChanger.AddCreateMethodExecutionArgs(createMethodExecutionArgsInstance);
var createAspectInstance = creator.CreateAspectInstance(aspect);
if (overriddenAspectMethods.HasFlag(AspectMethods.OnEntry))
{
var callAspectOnEntry = creator.CallAspectOnEntry(createAspectInstance,
createMethodExecutionArgsInstance);
_methodBodyChanger.AddOnEntryCall(createAspectInstance, callAspectOnEntry);
}
if (overriddenAspectMethods.HasFlag(AspectMethods.OnExit))
{
var setMethodExecutionArgsReturnValue =
creator.SetMethodExecutionArgsReturnValue(createMethodExecutionArgsInstance, loadReturnValue);
var callAspectOnExit = creator.CallAspectOnExit(createAspectInstance,
createMethodExecutionArgsInstance);
_methodBodyChanger.AddOnExitCall(createAspectInstance, callAspectOnExit,
setMethodExecutionArgsReturnValue);
}
if (overriddenAspectMethods.HasFlag(AspectMethods.OnException))
{
var setMethodExecutionArgsExceptionFromStack =
creator.SetMethodExecutionArgsExceptionFromStack(createMethodExecutionArgsInstance);
var callAspectOnException = creator.CallAspectOnException(createAspectInstance,
createMethodExecutionArgsInstance);
_methodBodyChanger.AddOnExceptionCall(createAspectInstance, callAspectOnException,
setMethodExecutionArgsExceptionFromStack);
}
if (_methodBodyChanger.EndsWithThrow)
{
var saveThrownException = creator.SaveThrownException();
var loadThrownException = creator.LoadValueOnStack(saveThrownException);
var loadThrownException2 = creator.LoadValueOnStack(saveThrownException);
_methodBodyChanger.FixThrowAtEndOfRealBody(
saveThrownException,
loadThrownException,
loadThrownException2);
}
_methodBodyChanger.OptimizeBody();
WeaveCounter++;
}
