aboutsummaryrefslogtreecommitdiff
path: root/ChocolArm64/Instruction/AInstEmitFlow.cs
diff options
context:
space:
mode:
Diffstat (limited to 'ChocolArm64/Instruction/AInstEmitFlow.cs')
-rw-r--r--ChocolArm64/Instruction/AInstEmitFlow.cs31
1 files changed, 31 insertions, 0 deletions
diff --git a/ChocolArm64/Instruction/AInstEmitFlow.cs b/ChocolArm64/Instruction/AInstEmitFlow.cs
index 91262834..89979d05 100644
--- a/ChocolArm64/Instruction/AInstEmitFlow.cs
+++ b/ChocolArm64/Instruction/AInstEmitFlow.cs
@@ -35,6 +35,14 @@ namespace ChocolArm64.Instruction
{
AOpCodeBImmAl Op = (AOpCodeBImmAl)Context.CurrOp;
+ if (AOptimizations.GenerateCallStack)
+ {
+ Context.EmitLdarg(ATranslatedSub.StateArgIdx);
+ Context.EmitLdc_I8(Op.Imm);
+
+ Context.EmitPrivateCall(typeof(AThreadState), nameof(AThreadState.EnterMethod));
+ }
+
Context.EmitLdc_I(Op.Position + 4);
Context.EmitStint(AThreadState.LRIndex);
Context.EmitStoreState();
@@ -72,6 +80,14 @@ namespace ChocolArm64.Instruction
{
AOpCodeBReg Op = (AOpCodeBReg)Context.CurrOp;
+ if (AOptimizations.GenerateCallStack)
+ {
+ Context.EmitLdarg(ATranslatedSub.StateArgIdx);
+ Context.EmitLdintzr(Op.Rn);
+
+ Context.EmitPrivateCall(typeof(AThreadState), nameof(AThreadState.EnterMethod));
+ }
+
Context.EmitLdc_I(Op.Position + 4);
Context.EmitStint(AThreadState.LRIndex);
Context.EmitStoreState();
@@ -84,6 +100,14 @@ namespace ChocolArm64.Instruction
{
AOpCodeBReg Op = (AOpCodeBReg)Context.CurrOp;
+ if (AOptimizations.GenerateCallStack)
+ {
+ Context.EmitLdarg(ATranslatedSub.StateArgIdx);
+ Context.EmitLdintzr(Op.Rn);
+
+ Context.EmitPrivateCall(typeof(AThreadState), nameof(AThreadState.JumpMethod));
+ }
+
Context.EmitStoreState();
Context.EmitLdintzr(Op.Rn);
@@ -105,6 +129,13 @@ namespace ChocolArm64.Instruction
public static void Ret(AILEmitterCtx Context)
{
+ if (AOptimizations.GenerateCallStack)
+ {
+ Context.EmitLdarg(ATranslatedSub.StateArgIdx);
+
+ Context.EmitPrivateCall(typeof(AThreadState), nameof(AThreadState.ExitMethod));
+ }
+
Context.EmitStoreState();
Context.EmitLdint(AThreadState.LRIndex);