R/ChocolArm64/ATranslator.cs

198 lines
5.7 KiB
C#
Raw Normal View History

2018-02-04 23:08:20 +00:00
using ChocolArm64.Decoder;
using ChocolArm64.Events;
2018-02-04 23:08:20 +00:00
using ChocolArm64.Instruction;
using ChocolArm64.Memory;
2018-02-04 23:08:20 +00:00
using ChocolArm64.Translation;
using System;
using System.Collections.Concurrent;
2018-02-04 23:08:20 +00:00
using System.Collections.Generic;
using System.Reflection.Emit;
namespace ChocolArm64
{
public class ATranslator
2018-02-04 23:08:20 +00:00
{
private HashSet<long> SubBlocks;
private ConcurrentDictionary<long, ATranslatedSub> CachedSubs;
2018-02-04 23:08:20 +00:00
private ConcurrentDictionary<long, string> SymbolTable;
public event EventHandler<ACpuTraceEventArgs> CpuTrace;
public bool EnableCpuTrace { get; set; }
2018-02-18 04:57:33 +00:00
public ATranslator(IReadOnlyDictionary<long, string> SymbolTable = null)
2018-02-04 23:08:20 +00:00
{
SubBlocks = new HashSet<long>();
CachedSubs = new ConcurrentDictionary<long, ATranslatedSub>();
2018-02-04 23:08:20 +00:00
if (SymbolTable != null)
{
this.SymbolTable = new ConcurrentDictionary<long, string>(SymbolTable);
}
else
{
this.SymbolTable = new ConcurrentDictionary<long, string>();
}
2018-02-04 23:08:20 +00:00
}
internal void ExecuteSubroutine(AThread Thread, long Position)
2018-02-04 23:08:20 +00:00
{
do
{
if (EnableCpuTrace)
2018-02-04 23:08:20 +00:00
{
if (!SymbolTable.TryGetValue(Position, out string SubName))
{
SubName = string.Empty;
}
CpuTrace?.Invoke(this, new ACpuTraceEventArgs(Position, SubName));
2018-02-04 23:08:20 +00:00
}
if (!CachedSubs.TryGetValue(Position, out ATranslatedSub Sub))
{
Sub = TranslateTier0(Thread.Memory, Position);
}
if (Sub.ShouldReJit())
2018-02-04 23:08:20 +00:00
{
TranslateTier1(Thread.Memory, Position);
2018-02-04 23:08:20 +00:00
}
Position = Sub.Execute(Thread.ThreadState, Thread.Memory);
2018-02-04 23:08:20 +00:00
}
while (Position != 0 && Thread.ThreadState.Running);
2018-02-04 23:08:20 +00:00
}
internal bool TryGetCachedSub(AOpCode OpCode, out ATranslatedSub Sub)
2018-02-04 23:08:20 +00:00
{
if (OpCode.Emitter != AInstEmit.Bl)
{
Sub = null;
return false;
}
return TryGetCachedSub(((AOpCodeBImmAl)OpCode).Imm, out Sub);
}
internal bool TryGetCachedSub(long Position, out ATranslatedSub Sub)
2018-02-04 23:08:20 +00:00
{
return CachedSubs.TryGetValue(Position, out Sub);
}
internal bool HasCachedSub(long Position)
2018-02-04 23:08:20 +00:00
{
return CachedSubs.ContainsKey(Position);
}
private ATranslatedSub TranslateTier0(AMemory Memory, long Position)
{
ABlock Block = ADecoder.DecodeBasicBlock(this, Memory, Position);
ABlock[] Graph = new ABlock[] { Block };
string SubName = GetSubName(Position);
AILEmitterCtx Context = new AILEmitterCtx(this, Graph, Block, SubName);
do
{
Context.EmitOpCode();
}
while (Context.AdvanceOpCode());
ATranslatedSub Subroutine = Context.GetSubroutine();
lock (SubBlocks)
{
if (SubBlocks.Contains(Position))
{
SubBlocks.Remove(Position);
Subroutine.SetType(ATranslatedSubType.SubBlock);
}
else
{
Subroutine.SetType(ATranslatedSubType.SubTier0);
}
}
CachedSubs.AddOrUpdate(Position, Subroutine, (Key, OldVal) => Subroutine);
AOpCode LastOp = Block.GetLastOp();
lock (SubBlocks)
{
if (LastOp.Emitter != AInstEmit.Ret &&
LastOp.Emitter != AInstEmit.Br)
{
SubBlocks.Add(LastOp.Position + 4);
}
}
return Subroutine;
}
private void TranslateTier1(AMemory Memory, long Position)
2018-02-04 23:08:20 +00:00
{
(ABlock[] Graph, ABlock Root) Cfg = ADecoder.DecodeSubroutine(this, Memory, Position);
string SubName = GetSubName(Position);
PropagateName(Cfg.Graph, SubName);
2018-02-04 23:08:20 +00:00
AILEmitterCtx Context = new AILEmitterCtx(this, Cfg.Graph, Cfg.Root, SubName);
2018-02-04 23:08:20 +00:00
if (Context.CurrBlock.Position != Position)
{
Context.Emit(OpCodes.Br, Context.GetLabel(Position));
}
do
{
Context.EmitOpCode();
}
while (Context.AdvanceOpCode());
//Mark all methods that calls this method for ReJiting,
//since we can now call it directly which is faster.
foreach (ATranslatedSub TS in CachedSubs.Values)
{
if (TS.HasCallee(Position))
2018-02-04 23:08:20 +00:00
{
TS.MarkForReJit();
}
}
ATranslatedSub Subroutine = Context.GetSubroutine();
Subroutine.SetType(ATranslatedSubType.SubTier1);
CachedSubs.AddOrUpdate(Position, Subroutine, (Key, OldVal) => Subroutine);
}
2018-02-04 23:08:20 +00:00
private string GetSubName(long Position)
{
return SymbolTable.GetOrAdd(Position, $"Sub{Position:x16}");
2018-02-04 23:08:20 +00:00
}
private void PropagateName(ABlock[] Graph, string Name)
{
foreach (ABlock Block in Graph)
{
AOpCode LastOp = Block.GetLastOp();
if (LastOp != null &&
(LastOp.Emitter == AInstEmit.Bl ||
LastOp.Emitter == AInstEmit.Blr))
{
SymbolTable.TryAdd(LastOp.Position + 4, Name);
}
}
}
2018-02-04 23:08:20 +00:00
}
}