mirror of
https://github.com/Ryujinx-NX/Ryujinx.git
synced 2024-11-14 21:17:43 -07:00
eafadf10c7
Dispose unicorn when done
327 lines
9.6 KiB
C#
327 lines
9.6 KiB
C#
using Ryujinx.Tests.Unicorn.Native;
|
|
using System;
|
|
|
|
namespace Ryujinx.Tests.Unicorn
|
|
{
|
|
public class UnicornAArch64 : IDisposable
|
|
{
|
|
internal readonly IntPtr uc;
|
|
private bool _isDisposed = false;
|
|
|
|
public IndexedProperty<int, ulong> X
|
|
{
|
|
get
|
|
{
|
|
return new IndexedProperty<int, ulong>(
|
|
(int i) => GetX(i),
|
|
(int i, ulong value) => SetX(i, value));
|
|
}
|
|
}
|
|
|
|
public IndexedProperty<int, SimdValue> Q
|
|
{
|
|
get
|
|
{
|
|
return new IndexedProperty<int, SimdValue>(
|
|
(int i) => GetQ(i),
|
|
(int i, SimdValue value) => SetQ(i, value));
|
|
}
|
|
}
|
|
|
|
public ulong LR
|
|
{
|
|
get => GetRegister(ArmRegister.LR);
|
|
set => SetRegister(ArmRegister.LR, value);
|
|
}
|
|
|
|
public ulong SP
|
|
{
|
|
get => GetRegister(ArmRegister.SP);
|
|
set => SetRegister(ArmRegister.SP, value);
|
|
}
|
|
|
|
public ulong PC
|
|
{
|
|
get => GetRegister(ArmRegister.PC);
|
|
set => SetRegister(ArmRegister.PC, value);
|
|
}
|
|
|
|
public uint Pstate
|
|
{
|
|
get => (uint)GetRegister(ArmRegister.PSTATE);
|
|
set => SetRegister(ArmRegister.PSTATE, (uint)value);
|
|
}
|
|
|
|
public int Fpcr
|
|
{
|
|
get => (int)GetRegister(ArmRegister.FPCR);
|
|
set => SetRegister(ArmRegister.FPCR, (uint)value);
|
|
}
|
|
|
|
public int Fpsr
|
|
{
|
|
get => (int)GetRegister(ArmRegister.FPSR);
|
|
set => SetRegister(ArmRegister.FPSR, (uint)value);
|
|
}
|
|
|
|
public bool OverflowFlag
|
|
{
|
|
get => (Pstate & 0x10000000u) != 0;
|
|
set => Pstate = (Pstate & ~0x10000000u) | (value ? 0x10000000u : 0u);
|
|
}
|
|
|
|
public bool CarryFlag
|
|
{
|
|
get => (Pstate & 0x20000000u) != 0;
|
|
set => Pstate = (Pstate & ~0x20000000u) | (value ? 0x20000000u : 0u);
|
|
}
|
|
|
|
public bool ZeroFlag
|
|
{
|
|
get => (Pstate & 0x40000000u) != 0;
|
|
set => Pstate = (Pstate & ~0x40000000u) | (value ? 0x40000000u : 0u);
|
|
}
|
|
|
|
public bool NegativeFlag
|
|
{
|
|
get => (Pstate & 0x80000000u) != 0;
|
|
set => Pstate = (Pstate & ~0x80000000u) | (value ? 0x80000000u : 0u);
|
|
}
|
|
|
|
public UnicornAArch64()
|
|
{
|
|
Interface.Checked(Interface.uc_open(UnicornArch.UC_ARCH_ARM64, UnicornMode.UC_MODE_LITTLE_ENDIAN, out uc));
|
|
|
|
SetRegister(ArmRegister.CPACR_EL1, 0x00300000);
|
|
}
|
|
|
|
~UnicornAArch64()
|
|
{
|
|
Dispose(false);
|
|
}
|
|
|
|
public void Dispose()
|
|
{
|
|
Dispose(true);
|
|
GC.SuppressFinalize(this);
|
|
}
|
|
|
|
protected virtual void Dispose(bool disposing)
|
|
{
|
|
if (!_isDisposed)
|
|
{
|
|
Interface.Checked(Native.Interface.uc_close(uc));
|
|
_isDisposed = true;
|
|
}
|
|
}
|
|
|
|
public void RunForCount(ulong count)
|
|
{
|
|
Interface.Checked(Native.Interface.uc_emu_start(uc, this.PC, 0xFFFFFFFFFFFFFFFFu, 0, count));
|
|
}
|
|
|
|
public void Step()
|
|
{
|
|
RunForCount(1);
|
|
}
|
|
|
|
private static ArmRegister[] XRegisters = new ArmRegister[31]
|
|
{
|
|
ArmRegister.X0,
|
|
ArmRegister.X1,
|
|
ArmRegister.X2,
|
|
ArmRegister.X3,
|
|
ArmRegister.X4,
|
|
ArmRegister.X5,
|
|
ArmRegister.X6,
|
|
ArmRegister.X7,
|
|
ArmRegister.X8,
|
|
ArmRegister.X9,
|
|
ArmRegister.X10,
|
|
ArmRegister.X11,
|
|
ArmRegister.X12,
|
|
ArmRegister.X13,
|
|
ArmRegister.X14,
|
|
ArmRegister.X15,
|
|
ArmRegister.X16,
|
|
ArmRegister.X17,
|
|
ArmRegister.X18,
|
|
ArmRegister.X19,
|
|
ArmRegister.X20,
|
|
ArmRegister.X21,
|
|
ArmRegister.X22,
|
|
ArmRegister.X23,
|
|
ArmRegister.X24,
|
|
ArmRegister.X25,
|
|
ArmRegister.X26,
|
|
ArmRegister.X27,
|
|
ArmRegister.X28,
|
|
ArmRegister.X29,
|
|
ArmRegister.X30,
|
|
};
|
|
|
|
private static ArmRegister[] QRegisters = new ArmRegister[32]
|
|
{
|
|
ArmRegister.Q0,
|
|
ArmRegister.Q1,
|
|
ArmRegister.Q2,
|
|
ArmRegister.Q3,
|
|
ArmRegister.Q4,
|
|
ArmRegister.Q5,
|
|
ArmRegister.Q6,
|
|
ArmRegister.Q7,
|
|
ArmRegister.Q8,
|
|
ArmRegister.Q9,
|
|
ArmRegister.Q10,
|
|
ArmRegister.Q11,
|
|
ArmRegister.Q12,
|
|
ArmRegister.Q13,
|
|
ArmRegister.Q14,
|
|
ArmRegister.Q15,
|
|
ArmRegister.Q16,
|
|
ArmRegister.Q17,
|
|
ArmRegister.Q18,
|
|
ArmRegister.Q19,
|
|
ArmRegister.Q20,
|
|
ArmRegister.Q21,
|
|
ArmRegister.Q22,
|
|
ArmRegister.Q23,
|
|
ArmRegister.Q24,
|
|
ArmRegister.Q25,
|
|
ArmRegister.Q26,
|
|
ArmRegister.Q27,
|
|
ArmRegister.Q28,
|
|
ArmRegister.Q29,
|
|
ArmRegister.Q30,
|
|
ArmRegister.Q31,
|
|
};
|
|
|
|
public ulong GetX(int index)
|
|
{
|
|
if ((uint)index > 30)
|
|
{
|
|
throw new ArgumentOutOfRangeException(nameof(index));
|
|
}
|
|
|
|
return GetRegister(XRegisters[index]);
|
|
}
|
|
|
|
public void SetX(int index, ulong value)
|
|
{
|
|
if ((uint)index > 30)
|
|
{
|
|
throw new ArgumentOutOfRangeException(nameof(index));
|
|
}
|
|
|
|
SetRegister(XRegisters[index], value);
|
|
}
|
|
|
|
public SimdValue GetQ(int index)
|
|
{
|
|
if ((uint)index > 31)
|
|
{
|
|
throw new ArgumentOutOfRangeException(nameof(index));
|
|
}
|
|
|
|
return GetVector(QRegisters[index]);
|
|
}
|
|
|
|
public void SetQ(int index, SimdValue value)
|
|
{
|
|
if ((uint)index > 31)
|
|
{
|
|
throw new ArgumentOutOfRangeException(nameof(index));
|
|
}
|
|
|
|
SetVector(QRegisters[index], value);
|
|
}
|
|
|
|
private ulong GetRegister(ArmRegister register)
|
|
{
|
|
byte[] data = new byte[8];
|
|
|
|
Interface.Checked(Native.Interface.uc_reg_read(uc, (int)register, data));
|
|
|
|
return (ulong)BitConverter.ToInt64(data, 0);
|
|
}
|
|
|
|
private void SetRegister(ArmRegister register, ulong value)
|
|
{
|
|
byte[] data = BitConverter.GetBytes(value);
|
|
|
|
Interface.Checked(Interface.uc_reg_write(uc, (int)register, data));
|
|
}
|
|
|
|
private SimdValue GetVector(ArmRegister register)
|
|
{
|
|
byte[] data = new byte[16];
|
|
|
|
Interface.Checked(Interface.uc_reg_read(uc, (int)register, data));
|
|
|
|
return new SimdValue(data);
|
|
}
|
|
|
|
private void SetVector(ArmRegister register, SimdValue value)
|
|
{
|
|
byte[] data = value.ToArray();
|
|
|
|
Interface.Checked(Interface.uc_reg_write(uc, (int)register, data));
|
|
}
|
|
|
|
public byte[] MemoryRead(ulong address, ulong size)
|
|
{
|
|
byte[] value = new byte[size];
|
|
|
|
Interface.Checked(Interface.uc_mem_read(uc, address, value, size));
|
|
|
|
return value;
|
|
}
|
|
|
|
public byte MemoryRead8 (ulong address) => MemoryRead(address, 1)[0];
|
|
public UInt16 MemoryRead16(ulong address) => (UInt16)BitConverter.ToInt16(MemoryRead(address, 2), 0);
|
|
public UInt32 MemoryRead32(ulong address) => (UInt32)BitConverter.ToInt32(MemoryRead(address, 4), 0);
|
|
public UInt64 MemoryRead64(ulong address) => (UInt64)BitConverter.ToInt64(MemoryRead(address, 8), 0);
|
|
|
|
public void MemoryWrite(ulong address, byte[] value)
|
|
{
|
|
Interface.Checked(Interface.uc_mem_write(uc, address, value, (ulong)value.Length));
|
|
}
|
|
|
|
public void MemoryWrite8 (ulong address, byte value) => MemoryWrite(address, new byte[]{value});
|
|
public void MemoryWrite16(ulong address, Int16 value) => MemoryWrite(address, BitConverter.GetBytes(value));
|
|
public void MemoryWrite16(ulong address, UInt16 value) => MemoryWrite(address, BitConverter.GetBytes(value));
|
|
public void MemoryWrite32(ulong address, Int32 value) => MemoryWrite(address, BitConverter.GetBytes(value));
|
|
public void MemoryWrite32(ulong address, UInt32 value) => MemoryWrite(address, BitConverter.GetBytes(value));
|
|
public void MemoryWrite64(ulong address, Int64 value) => MemoryWrite(address, BitConverter.GetBytes(value));
|
|
public void MemoryWrite64(ulong address, UInt64 value) => MemoryWrite(address, BitConverter.GetBytes(value));
|
|
|
|
public void MemoryMap(ulong address, ulong size, MemoryPermission permissions)
|
|
{
|
|
Interface.Checked(Interface.uc_mem_map(uc, address, size, (uint)permissions));
|
|
}
|
|
|
|
public void MemoryUnmap(ulong address, ulong size)
|
|
{
|
|
Interface.Checked(Interface.uc_mem_unmap(uc, address, size));
|
|
}
|
|
|
|
public void MemoryProtect(ulong address, ulong size, MemoryPermission permissions)
|
|
{
|
|
Interface.Checked(Interface.uc_mem_protect(uc, address, size, (uint)permissions));
|
|
}
|
|
|
|
public static bool IsAvailable()
|
|
{
|
|
try
|
|
{
|
|
Interface.uc_version(out _, out _);
|
|
|
|
return true;
|
|
}
|
|
catch (DllNotFoundException)
|
|
{
|
|
return false;
|
|
}
|
|
}
|
|
}
|
|
} |