private static void ResolveAsm(string assemblyPath)
{
ILog log = log4net.LogManager.GetLogger(MethodBase.GetCurrentMethod().DeclaringType);
log.Info(string.Format("load source {0}", assemblyPath));
// 复制路径
var directoryPath = Path.Combine(AppDomain.CurrentDomain.BaseDirectory, "aoptemp");
var targetPath = Path.Combine(directoryPath, Path.GetFileName(assemblyPath));
log.Info(string.Format("copy {0} to {1}", assemblyPath, targetPath));
File.Copy(assemblyPath, targetPath, true);
// copy pdb
string oPath = Path.GetDirectoryName(assemblyPath);
var sourceSymbol = oPath + @"\" + Path.GetFileNameWithoutExtension(assemblyPath) + ".pdb";
var symbolPath = directoryPath + @"\" + Path.GetFileNameWithoutExtension(targetPath) + ".pdb";
try
{
File.Copy(sourceSymbol, symbolPath, true);
}
catch (Exception ex)
{
log.Error(string.Format("copy {0} to {1}", assemblyPath, targetPath));
return;
}
Stream stream = new FileStream(targetPath, FileMode.Open);
var clrAsm = Assembly.Load(stream.ToBytes());
if (clrAsm.GetCustomAttributes(typeof(AopAssembly), false).Length == 0)
return;
AssemblyDefinition asm = AssemblyDefinition.ReadAssembly(stream, new ReaderParameters() { ReadSymbols = true });
MethodReference il2 = asm.MainModule.Import(typeof(Type).GetMethod("GetTypeFromHandle", new Type[1] { typeof(RuntimeTypeHandle) }));
MethodReference il3 = asm.MainModule.Import(typeof(Type).GetMethod("GetMethod", new Type[1] { typeof(String) }));
MethodReference il4 = asm.MainModule.Import(typeof(TypeService).GetMethod("GetAttribute", new Type[2] { typeof(Type), typeof(MemberInfo) }));
MethodReference il5 = asm.MainModule.Import(typeof(AopAttribute).GetMethod("Invoke"));
// 搜索并插入调用的il代码
foreach (TypeDefinition type in asm.MainModule.Types)
{
if (!type.HasCustomAttributes) continue;
foreach (CustomAttribute attribute in type.CustomAttributes)
{
// 查找到实现aop标签的类
if (IsAssignableFromAopAttribute(attribute.Constructor.DeclaringType))
{
var clrType = clrAsm.GetType(type.FullName);
if (clrType == null) continue;
// 查找其实现基于aop标签的方法
foreach (MethodDefinition method in type.Methods)
{
if (!method.HasCustomAttributes) continue;
MethodInfo clrMethod = null;
try
{
clrMethod = clrType.GetMethod(method.Name, method.Parameters.Select(X => Type.GetType(X.ParameterType.GetAssemblyQualifiedName())).ToArray());
}
catch (Exception ex) { }
if (clrMethod == null) continue;
foreach (CustomAttribute methodAttribute in method.CustomAttributes)
{
if (IsAssignableFromAopAttribute(methodAttribute.Constructor.DeclaringType))
{
var clrAttributeType = Type.GetType(methodAttribute.Constructor.DeclaringType.GetAssemblyQualifiedName());
if (clrAttributeType == null) continue;
var clrAttribute = clrMethod.GetCustomAttributes(clrAttributeType, true)[0] as AopAttribute;
var aopTypes = clrAttribute.AopType;
List<Instruction> fixedOffset = new List<Instruction>();
if (aopTypes == AopTypes.Packed)
{
foreach (Instruction il in method.Body.Instructions)
{
if (il.OpCode == OpCodes.Ret || il.Offset == 0)
{
fixedOffset.Add(il);
}
}
}
else if (aopTypes == AopTypes.Prefixed)
{
foreach (Instruction il in method.Body.Instructions)
{
if (il.Offset == 1)
{
fixedOffset.Add(il);
}
}
}
else if (aopTypes == AopTypes.Suffixed)
{
foreach (Instruction il in method.Body.Instructions)
{
if (il.OpCode == OpCodes.Ret)
{
fixedOffset.Add(il);
}
}
}
var worker = method.Body.GetILProcessor();
foreach (var offset in fixedOffset)
{
worker.InsertBefore(offset, worker.Create(OpCodes.Ldtoken, methodAttribute.Constructor.DeclaringType));
worker.InsertBefore(offset, worker.Create(OpCodes.Call, il2));
worker.InsertBefore(offset, worker.Create(OpCodes.Ldtoken, type));
worker.InsertBefore(offset, worker.Create(OpCodes.Call, il2));
worker.InsertBefore(offset, worker.Create(OpCodes.Ldstr, method.Name));
worker.InsertBefore(offset, worker.Create(OpCodes.Call, il3));
worker.InsertBefore(offset, worker.Create(OpCodes.Call, il4));
worker.InsertBefore(offset, worker.Create(OpCodes.Isinst, methodAttribute.Constructor.DeclaringType));
worker.InsertBefore(offset, worker.Create(OpCodes.Callvirt, il5));
worker.InsertBefore(offset, worker.Create(OpCodes.Nop));
}
if (aopTypes != AopTypes.None)
{
log.Info(string.Format("resolve aop method : {0} with {1}", clrMethod.ToString(), clrAttributeType.Name));
}
}
}
}
}
}
}
stream.Close();
log.Info(string.Format("save {0} to {1}", asm.Name, assemblyPath));
// 保存
asm.Write(assemblyPath, new WriterParameters() { WriteSymbols = true });
// 替换原文件
//File.Copy(targetPath, assemblyPath, true);
//File.Copy(symbolPath, sourceSymbol, true);
}
private byte[] GetImageBytesFromFilePath(string fileName, string rootPath)
{
byte[] content = null;
string filePath = ImageUploadHelper.GetUploadRootPath(rootPath) + "Temp\\" + fileName;
if (File.Exists(filePath))
{
using (FileStream fs = new FileStream(filePath, FileMode.Open, FileAccess.Read, FileShare.Read))
{
content = fs.ToBytes();
}
}
return content;
}
private static byte[] GetImageBytesFromFilePath(string filePath, out string fileName)
{
byte[] content = null;
fileName = Path.GetFileName(filePath);
if (File.Exists(filePath))
{
using (FileStream fs = new FileStream(filePath, FileMode.Open, FileAccess.Read, FileShare.Read))
{
content = fs.ToBytes();
}
}
return content;
}
