85 lines
2.8 KiB
C#
85 lines
2.8 KiB
C#
using dnlib.DotNet;
|
|
using dnlib.DotNet.Emit;
|
|
using Obfuz.Utils;
|
|
using Obfuz.Emit;
|
|
using System;
|
|
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using System.Runtime.CompilerServices;
|
|
using System.Text;
|
|
using System.Threading.Tasks;
|
|
using UnityEngine.Assertions;
|
|
|
|
namespace Obfuz.DynamicProxy
|
|
{
|
|
public class ProxyCallPass : MethodBodyObfuscationPassBase
|
|
{
|
|
private readonly IRandom _random;
|
|
private readonly IProxyCallPolicy _dynamicProxyPolicy;
|
|
private readonly IProxyCallObfuscator _dynamicProxyObfuscator;
|
|
|
|
public ProxyCallPass()
|
|
{
|
|
_random = new RandomWithKey(new byte[] { 0x1, 0x2, 0x3, 0x4 }, 0x5);
|
|
_dynamicProxyPolicy = new ConfigProxyCallPolicy();
|
|
_dynamicProxyObfuscator = new DefaultProxyCallObfuscator(_random);
|
|
}
|
|
|
|
public override void Stop(ObfuscationPassContext ctx)
|
|
{
|
|
_dynamicProxyObfuscator.Done();
|
|
}
|
|
|
|
public override void Start(ObfuscationPassContext ctx)
|
|
{
|
|
|
|
}
|
|
|
|
protected override bool NeedObfuscateMethod(MethodDef method)
|
|
{
|
|
return _dynamicProxyPolicy.NeedDynamicProxyCallInMethod(method);
|
|
}
|
|
|
|
protected override bool TryObfuscateInstruction(MethodDef method, Instruction inst, IList<Instruction> instructions, int instructionIndex,
|
|
List<Instruction> outputInstructions, List<Instruction> totalFinalInstructions)
|
|
{
|
|
IMethod calledMethod = inst.Operand as IMethod;
|
|
if (calledMethod == null || !calledMethod.IsMethod)
|
|
{
|
|
return false;
|
|
}
|
|
if (MetaUtil.ContainsContainsGenericParameter(calledMethod))
|
|
{
|
|
return false;
|
|
}
|
|
|
|
switch (inst.OpCode.Code)
|
|
{
|
|
case Code.Call:
|
|
{
|
|
if (!_dynamicProxyPolicy.NeedDynamicProxyCalledMethod(calledMethod, false))
|
|
{
|
|
return false;
|
|
}
|
|
_dynamicProxyObfuscator.Obfuscate(method, calledMethod, false, outputInstructions);
|
|
return true;
|
|
}
|
|
case Code.Callvirt:
|
|
{
|
|
if (instructionIndex > 0 && instructions[instructionIndex - 1].OpCode.Code == Code.Constrained)
|
|
{
|
|
return false;
|
|
}
|
|
if (!_dynamicProxyPolicy.NeedDynamicProxyCalledMethod(calledMethod, true))
|
|
{
|
|
return false;
|
|
}
|
|
_dynamicProxyObfuscator.Obfuscate(method, calledMethod, true, outputInstructions);
|
|
return true;
|
|
}
|
|
default: return false;
|
|
}
|
|
}
|
|
}
|
|
}
|