使用 Mono.Cecil 在 C# 程序集中注入方法

Question

我正在开发一些 .Net 应用程序，我需要用我自己的代码注入任何新的程序集方法。我正在使用 Mono.Cecil 获取装配体，我找到了一些样本，但它们已经足够旧了。不幸的是，github wiki 上的 migraton 部分没有任何信息。

所以，我有这个代码：

using System; 
using Mono.Cecil; 
using Mono.Cecil.Cil; 

namespace CustomFieldsInjection 
{ 
    public partial class Injector 
    { 
        public static void MethodInjection(string assemblyFilename, string typeName, string methodName) 
        { 
            AssemblyDefinition assembly = AssemblyFactory.GetAssembly(assemblyFilename); 

            TypeReference returnTypeReference = assembly.MainModule.Import(typeof(void)); 

            MethodDefinition methodDefinition = new MethodDefinition(methodName, MethodAttributes.Public | MethodAttributes.Static, returnTypeReference); 

            Instruction instruction1 = methodDefinition.Body.CilWorker.Create(OpCodes.Nop); 

            Instruction instruction2 = methodDefinition.Body.CilWorker.Create(OpCodes.Ldstr, methodName); 

            MethodReference writeline = assembly.MainModule.Import(typeof(Console).GetMethod("WriteLine", new Type[] { typeof(string) })); 

            methodDefinition.Body.CilWorker.Append(instruction1); 
            methodDefinition.Body.CilWorker.Append(instruction2); 

            methodDefinition.Body.CilWorker.InsertAfter(instruction2, methodDefinition.Body.CilWorker.Create (OpCodes.Call, writeline)); 

            methodDefinition.Body.CilWorker.Append (methodDefinition.Body.CilWorker.Create(OpCodes.Ret))

            assembly.MainModule.Inject(methodDefinition, assembly.MainModule.Types[typeName]); 
            MethodReference methodReference = null; 

            foreach (MethodDefinition method in assembly.MainModule.Types[typeName].Methods) 
            { 
                if (method.Name == methodName) 
                { 
                    methodReference = assembly.MainModule.Import(method); 
                    break; 
                } 
            } 

            Instruction callTest = methodDefinition.Body.CilWorker.Create(OpCodes.Call, methodReference); 

            if (assembly.EntryPoint != null) 
            { 
                assembly.EntryPoint.Body.CilWorker.InsertBefore(assembly.EntryPoint.Body.Instructions[0], callTest); 
            } 

            AssemblyFactory.SaveAssembly(assembly, assemblyFilename); 
        } 
    } 
}

这是旧样本。大多数功能都是最新的。我对这个结构很感兴趣：

assembly.MainModule.Inject(methodDefinition, assembly.MainModule.Types[typeName]);

我找不到这种设计的新类似物。有人可以告诉我它可以被替换吗？

Answer 1

我不熟悉你所指的构造，但是向现有类型添加 MethodDefinition 非常容易

using (var assemblyDefinition = AssemblyDefinition.ReadAssembly("assemblyPath")) {
    var module = AssemblyDefinition.MainModule;

    //Select the type you need to open for addition
    var typeDef = module.Types.First(td => td.Name == "footer");

    //Add your MethodDefinition
    typeDef.Methods.Add(your_method_definition);

    //Write the assembly back
    assemblyDefinition.Write();
}

注意：如果您还没有使用 cecil 0.10.0.0，您将使用略有不同的 ReadAssembly() 和 Write() 变体（没有 using，并将 assemblyPath 传递给Write, 主要是...)

使用 Mono.Cecil 在 C# 程序集中注入方法

Inject method in C# assembly with Mono.Cecil

c#

.net

mono

mono.cecil