使用Mono.Cecil编织代码时分支目标错误问题排查

问题根源 #

你手动修改分支目标的偏移地址是核心问题——嵌套结构中，每插入一段日志指令都会改变后续所有指令的偏移量，之前计算的固定偏移会全部失效，导致循环跳转指向非法地址。Mono.Cecil的分支指令本质是通过Instruction对象引用关联目标，而非直接存储偏移，直接操作偏移必然出错。

正确的插桩流程 #

1. 提前收集关键信息

   • 遍历方法的所有序列点，记录每个序列点对应的Instruction（记为originalInst）。
   • 遍历方法的所有指令，收集所有分支指令（BranchInstruction、SwitchInstruction等），记录它们的原始目标Instruction。

2. 插入日志指令块
   对每个originalInst，在其之前插入日志指令：

   // 示例：插入Console.WriteLine("日志内容")的IL指令
   var module = method.Module;
   var writeLineMethod = module.ImportReference(typeof(Console).GetMethod("WriteLine", new[] { typeof(string) }));
   
   // 创建日志指令块
   var logInsts = new List<Instruction>
   {
       Instruction.Create(OpCodes.Ldstr, $"[LOG] 执行序列点: {originalInst.Offset}"),
       Instruction.Create(OpCodes.Call, writeLineMethod)
   };
   
   // 插入到原指令之前
   foreach (var inst in logInsts)
   {
       method.Body.Instructions.Insert(method.Body.Instructions.IndexOf(originalInst), inst);
   }
   // 记录原指令对应的日志块起始指令
   var instMap = new Dictionary<Instruction, Instruction>();
   instMap[originalInst] = logInsts[0];
   

3. 修正分支目标引用
   遍历所有分支指令，将它们的目标从originalInst替换为对应的日志块起始指令：

   foreach (var inst in method.Body.Instructions)
   {
       if (inst is BranchInstruction branchInst)
       {
           if (instMap.TryGetValue(branchInst.Target, out var newTarget))
           {
               branchInst.Target = newTarget;
           }
       }
       else if (inst is SwitchInstruction switchInst)
       {
           for (int i = 0; i < switchInst.Targets.Length; i++)
           {
               if (instMap.TryGetValue(switchInst.Targets[i], out var newTarget))
               {
                   switchInst.Targets[i] = newTarget;
               }
           }
       }
   }
   

4. 收尾处理
   插桩完成后，让Cecil重新计算指令偏移：

   method.Body.CalculateOffsets();
   

嵌套结构的额外注意事项 #

• 循环的回跳目标（比如for循环的条件判断指令）通常也是一个序列点，必须确保将回跳分支指向日志块的起始，而非跳过日志。
• 不要在遍历指令列表的同时修改列表（比如边遍历边插入指令），会导致遍历索引混乱，必须先收集所有需要插桩的位置，再批量插入。

内容的提问来源于stack exchange，提问作者Peter Lenkefi