initial commit
This commit is contained in:
@@ -0,0 +1,48 @@
|
||||
using System.Collections.Generic;
|
||||
using dnlib.DotNet;
|
||||
using dnlib.DotNet.Emit;
|
||||
|
||||
namespace Obfuscator.Obfuscator.CtrlFlow;
|
||||
|
||||
public class BlockParser
|
||||
{
|
||||
public static List<Block> ParseMethod(MethodDef method)
|
||||
{
|
||||
List<Block> list = new List<Block>();
|
||||
Block block = new Block();
|
||||
int num = 0;
|
||||
int num2 = 0;
|
||||
block.Number = num;
|
||||
block.Instructions.Add(Instruction.Create(OpCodes.Nop));
|
||||
list.Add(block);
|
||||
block = new Block();
|
||||
Stack<ExceptionHandler> stack = new Stack<ExceptionHandler>();
|
||||
foreach (Instruction instruction in method.Body.Instructions)
|
||||
{
|
||||
foreach (ExceptionHandler exceptionHandler in method.Body.ExceptionHandlers)
|
||||
{
|
||||
if (exceptionHandler.HandlerStart == instruction || exceptionHandler.TryStart == instruction || exceptionHandler.FilterStart == instruction)
|
||||
{
|
||||
stack.Push(exceptionHandler);
|
||||
}
|
||||
}
|
||||
foreach (ExceptionHandler exceptionHandler2 in method.Body.ExceptionHandlers)
|
||||
{
|
||||
if (exceptionHandler2.HandlerEnd == instruction || exceptionHandler2.TryEnd == instruction)
|
||||
{
|
||||
stack.Pop();
|
||||
}
|
||||
}
|
||||
instruction.CalculateStackUsage(out var pushes, out var pops);
|
||||
block.Instructions.Add(instruction);
|
||||
num2 += pushes - pops;
|
||||
if (pushes == 0 && instruction.OpCode != OpCodes.Nop && (num2 == 0 || instruction.OpCode == OpCodes.Ret) && stack.Count == 0)
|
||||
{
|
||||
num = (block.Number = num + 1);
|
||||
list.Add(block);
|
||||
block = new Block();
|
||||
}
|
||||
}
|
||||
return list;
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user