Ryujinx/Ryujinx.Core/OsHle/Process.cs

318 lines
8.7 KiB
C#
Raw Normal View History

2018-02-05 00:08:20 +01:00
using ChocolArm64;
using ChocolArm64.Events;
2018-02-05 00:08:20 +01:00
using ChocolArm64.Memory;
using Ryujinx.Core.Loaders;
using Ryujinx.Core.Loaders.Executables;
using Ryujinx.Core.OsHle.Exceptions;
using Ryujinx.Core.OsHle.Handles;
using Ryujinx.Core.OsHle.Svc;
2018-02-05 00:08:20 +01:00
using System;
using System.Collections.Concurrent;
using System.Collections.Generic;
using System.Threading;
2018-02-05 00:08:20 +01:00
namespace Ryujinx.Core.OsHle
2018-02-05 00:08:20 +01:00
{
public class Process : IDisposable
2018-02-05 00:08:20 +01:00
{
private const int TlsSize = 0x200;
private const int TotalTlsSlots = 32;
private Switch Ns;
public int ProcessId { get; private set; }
private ATranslator Translator;
2018-02-05 00:08:20 +01:00
public AMemory Memory { get; private set; }
public KProcessScheduler Scheduler { get; private set; }
2018-02-05 00:08:20 +01:00
2018-02-14 06:43:21 +01:00
private SvcHandler SvcHandler;
2018-02-05 00:08:20 +01:00
private ConcurrentDictionary<int, AThread> TlsSlots;
private ConcurrentDictionary<long, HThread> ThreadsByTpidr;
2018-02-05 00:08:20 +01:00
private List<Executable> Executables;
private HThread MainThread;
2018-02-05 00:08:20 +01:00
private long ImageBase;
public Process(Switch Ns, int ProcessId)
2018-02-05 00:08:20 +01:00
{
this.Ns = Ns;
this.ProcessId = ProcessId;
Memory = Ns.Memory;
Scheduler = new KProcessScheduler();
SvcHandler = new SvcHandler(Ns, this);
TlsSlots = new ConcurrentDictionary<int, AThread>();
ThreadsByTpidr = new ConcurrentDictionary<long, HThread>();
2018-02-05 00:08:20 +01:00
Executables = new List<Executable>();
ImageBase = MemoryRegions.AddrSpaceStart;
2018-02-05 00:08:20 +01:00
MapRWMemRegion(
MemoryRegions.TlsPagesAddress,
MemoryRegions.TlsPagesSize,
MemoryType.ThreadLocal);
2018-02-05 00:08:20 +01:00
}
public void LoadProgram(IExecutable Program)
2018-02-05 00:08:20 +01:00
{
Logging.Info($"Image base at 0x{ImageBase:x16}.");
2018-02-05 00:08:20 +01:00
Executable Executable = new Executable(Program, Memory, ImageBase);
Executables.Add(Executable);
ImageBase = AMemoryHelper.PageRoundUp(Executable.ImageEnd);
}
public void SetEmptyArgs()
{
ImageBase += AMemoryMgr.PageSize;
}
public bool Run(bool UseHbAbi = false)
2018-02-05 00:08:20 +01:00
{
if (Executables.Count == 0)
{
return false;
}
MapRWMemRegion(
MemoryRegions.MainStackAddress,
MemoryRegions.MainStackSize,
MemoryType.Normal);
long StackTop = MemoryRegions.MainStackAddress + MemoryRegions.MainStackSize;
2018-02-05 00:08:20 +01:00
int Handle = MakeThread(Executables[0].ImageBase, StackTop, 0, 0, 0);
2018-02-05 00:08:20 +01:00
if (Handle == -1)
{
return false;
}
MainThread = Ns.Os.Handles.GetData<HThread>(Handle);
2018-02-05 00:08:20 +01:00
if (UseHbAbi)
{
long HbAbiDataPosition = AMemoryHelper.PageRoundUp(Executables[0].ImageEnd);
Homebrew.WriteHbAbiData(Memory, HbAbiDataPosition, Handle);
MainThread.Thread.ThreadState.X0 = (ulong)HbAbiDataPosition;
MainThread.Thread.ThreadState.X1 = ulong.MaxValue;
}
Scheduler.StartThread(MainThread);
2018-02-05 00:08:20 +01:00
return true;
}
private void MapRWMemRegion(long Position, long Size, MemoryType Type)
{
Memory.Manager.Map(Position, Size, (int)Type, AMemoryPerm.RW);
}
2018-02-05 00:08:20 +01:00
public void StopAllThreads()
{
if (MainThread != null)
{
while (MainThread.Thread.IsAlive)
2018-02-05 00:08:20 +01:00
{
MainThread.Thread.StopExecution();
2018-02-05 00:08:20 +01:00
}
}
foreach (AThread Thread in TlsSlots.Values)
{
while (Thread.IsAlive)
2018-02-05 00:08:20 +01:00
{
Thread.StopExecution();
}
}
}
public int MakeThread(
long EntryPoint,
long StackTop,
long ArgsPtr,
int Priority,
int ProcessorId)
{
ThreadPriority ThreadPrio;
2018-02-05 00:08:20 +01:00
if (Priority < 12)
{
ThreadPrio = ThreadPriority.Highest;
}
else if (Priority < 24)
{
ThreadPrio = ThreadPriority.AboveNormal;
}
else if (Priority < 36)
{
ThreadPrio = ThreadPriority.Normal;
}
else if (Priority < 48)
{
ThreadPrio = ThreadPriority.BelowNormal;
}
else
{
ThreadPrio = ThreadPriority.Lowest;
}
AThread Thread = new AThread(GetTranslator(), Memory, ThreadPrio, EntryPoint);
HThread ThreadHnd = new HThread(Thread, ProcessorId, Priority);
2018-02-05 00:08:20 +01:00
int Handle = Ns.Os.Handles.GenerateId(ThreadHnd);
int TlsSlot = GetFreeTlsSlot(Thread);
2018-02-05 00:08:20 +01:00
if (TlsSlot == -1 || Handle == -1)
{
return -1;
}
long Tpidr = MemoryRegions.TlsPagesAddress + TlsSlot * TlsSize;
2018-02-18 20:28:07 +01:00
Thread.ThreadState.Break += BreakHandler;
Thread.ThreadState.SvcCall += SvcHandler.SvcCall;
Thread.ThreadState.Undefined += UndefinedHandler;
Thread.ThreadState.ProcessId = ProcessId;
Thread.ThreadState.ThreadId = Ns.Os.IdGen.GenerateId();
Thread.ThreadState.Tpidr = Tpidr;
2018-02-18 20:28:07 +01:00
Thread.ThreadState.X0 = (ulong)ArgsPtr;
Thread.ThreadState.X1 = (ulong)Handle;
Thread.ThreadState.X31 = (ulong)StackTop;
2018-02-05 00:08:20 +01:00
Thread.WorkFinished += ThreadFinished;
2018-02-18 20:28:07 +01:00
ThreadsByTpidr.TryAdd(Thread.ThreadState.Tpidr, ThreadHnd);
2018-02-05 00:08:20 +01:00
return Handle;
}
private void BreakHandler(object sender, AInstExceptionEventArgs e)
{
throw new GuestBrokeExecutionException();
}
private void UndefinedHandler(object sender, AInstUndefinedEventArgs e)
{
throw new UndefinedInstructionException(e.Position, e.RawOpCode);
}
private ATranslator GetTranslator()
{
if (Translator == null)
{
Dictionary<long, string> SymbolTable = new Dictionary<long, string>();
foreach (Executable Exe in Executables)
{
foreach (KeyValuePair<long, string> KV in Exe.SymbolTable)
{
SymbolTable.TryAdd(Exe.ImageBase + KV.Key, KV.Value);
}
}
Translator = new ATranslator(SymbolTable);
Translator.CpuTrace += CpuTraceHandler;
}
return Translator;
}
private void CpuTraceHandler(object sender, ACpuTraceEventArgs e)
{
string NsoName = string.Empty;
for (int Index = Executables.Count - 1; Index >= 0; Index--)
{
if (e.Position >= Executables[Index].ImageBase)
{
NsoName = $"{(e.Position - Executables[Index].ImageBase):x16}";
break;
}
}
Logging.Trace($"Executing at 0x{e.Position:x16} {e.SubName} {NsoName}");
}
public void EnableCpuTracing()
{
Translator.EnableCpuTrace = true;
}
public void DisableCpuTracing()
{
Translator.EnableCpuTrace = false;
}
2018-02-05 00:08:20 +01:00
private int GetFreeTlsSlot(AThread Thread)
{
for (int Index = 1; Index < TotalTlsSlots; Index++)
{
if (TlsSlots.TryAdd(Index, Thread))
{
return Index;
}
}
return -1;
}
private void ThreadFinished(object sender, EventArgs e)
{
if (sender is AThread Thread)
{
2018-02-18 20:28:07 +01:00
TlsSlots.TryRemove(GetTlsSlot(Thread.ThreadState.Tpidr), out _);
2018-02-05 00:08:20 +01:00
Ns.Os.IdGen.DeleteId(Thread.ThreadId);
}
}
private int GetTlsSlot(long Position)
{
return (int)((Position - MemoryRegions.TlsPagesAddress) / TlsSize);
2018-02-05 00:08:20 +01:00
}
2018-02-19 20:37:13 +01:00
public HThread GetThread(long Tpidr)
{
2018-02-19 20:37:13 +01:00
if (!ThreadsByTpidr.TryGetValue(Tpidr, out HThread Thread))
{
Logging.Error($"Thread with TPIDR 0x{Tpidr:x16} not found!");
}
return Thread;
}
public void Dispose()
{
Dispose(true);
}
protected virtual void Dispose(bool Disposing)
{
if (Disposing)
{
Scheduler.Dispose();
}
}
2018-02-05 00:08:20 +01:00
}
}