使用Mono.Cecil添加静态构造函数会导致TypeInitializationException

时间:2016-11-28 16:20:17

标签: c# cil mono.cecil

我正在尝试使用Mono Cecil将静态构造函数添加到以下程序中:

namespace SimpleTarget
{
    class C
    {
        public void M()
        {
            Console.WriteLine("Hello, World!");
        }
    }
}

以下代码添加了静态构造函数:

namespace AddStaticConstructor
{
    class Program
    {
        static void Main(string[] args)
        {
            var assemblyPath = args[0];
            var module = ModuleDefinition.ReadModule(assemblyPath);

            var corlib = ModuleDefinition.ReadModule(typeof(object).Module.FullyQualifiedName);
            var method = corlib.Types.First(t => t.Name.Equals("Console")).Methods.First(m => m.Name.Contains("WriteLine"));

            var methodToCall = module.Import(method);

            foreach (var type in module.Types)
            {
                if (!type.Name.Contains("C")) continue;

                var staticConstructorAttributes =
                    Mono.Cecil.MethodAttributes.Private |
                    Mono.Cecil.MethodAttributes.HideBySig |
                    Mono.Cecil.MethodAttributes.Static |
                    Mono.Cecil.MethodAttributes.SpecialName |
                    Mono.Cecil.MethodAttributes.RTSpecialName;

                MethodDefinition staticConstructor = new MethodDefinition(".cctor", staticConstructorAttributes, module.TypeSystem.Void);
                type.Methods.Add(staticConstructor);

                type.IsBeforeFieldInit = false;

                var il = staticConstructor.Body.GetILProcessor();
                il.Append(Instruction.Create(OpCodes.Ret));

                Instruction ldMethodName = il.Create(OpCodes.Ldstr, type.FullName);
                Instruction callOurMethod = il.Create(OpCodes.Call, methodToCall);

                Instruction firstInstruction = staticConstructor.Body.Instructions[0];
                // Inserts the callOurMethod instruction before the first instruction


                il.InsertBefore(firstInstruction, ldMethodName);
                il.InsertAfter(ldMethodName, callOurMethod);
            }

            module.Write(assemblyPath);
        }
    }
}

查看dotPeek中反编译的二进制文件,看起来好像一切都设置正确。当尝试使用修改后的C类型时,我得到一个TypeInitializationException,内部异常“System.InvalidProgramException:JIT Compiler遇到内部限制”

在使用静态构造函数之前,我还需要正确设置其他内容吗?

谢谢!

1 个答案:

答案 0 :(得分:2)

问题是你在System.WriteLine得到了错误的重载:

var corlib = ModuleDefinition.ReadModule(typeof(object).Module.FullyQualifiedName);
var method = corlib.Types.First(t => t.Name.Equals("Console")).Methods.First(m => m.Name.Contains("WriteLine"));
var methodToCall = module.Import(method);

使用这个简单的代码来获取你想要使用的重载:

var wlMethod = typeof (Console).GetMethod(nameof(Console.WriteLine), new[] {typeof (string)});
var methodToCall = module.ImportReference(wlMethod);