0
0
Fork 0
This repository has been archived on 2024-10-12. You can view files and clone it, but cannot push or open issues or pull requests.
ryujinx-final/Ryujinx.HLE/HOS/Services/Sm/IUserInterface.cs

249 lines
8 KiB
C#
Raw Normal View History

using Ryujinx.Common.Logging;
using Ryujinx.HLE.Exceptions;
using Ryujinx.HLE.HOS.Ipc;
using Ryujinx.HLE.HOS.Kernel;
using Ryujinx.HLE.HOS.Kernel.Common;
using Ryujinx.HLE.HOS.Kernel.Ipc;
using System;
using System.Collections.Concurrent;
using System.Collections.Generic;
using System.IO;
using System.Linq;
using System.Reflection;
2018-02-04 23:08:20 +00:00
namespace Ryujinx.HLE.HOS.Services.Sm
2018-02-04 23:08:20 +00:00
{
class IUserInterface : IpcService
2018-02-04 23:08:20 +00:00
{
private static Dictionary<string, Type> _services;
private static readonly ConcurrentDictionary<string, KPort> _registeredServices;
private readonly ServerBase _commonServer;
private bool _isInitialized;
public IUserInterface(KernelContext context)
{
_commonServer = new ServerBase(context, "CommonServer");
}
static IUserInterface()
{
_registeredServices = new ConcurrentDictionary<string, KPort>();
_services = Assembly.GetExecutingAssembly().GetTypes()
.SelectMany(type => type.GetCustomAttributes(typeof(ServiceAttribute), true)
.Select(service => (((ServiceAttribute)service).Name, type)))
.ToDictionary(service => service.Name, service => service.type);
}
2018-02-04 23:08:20 +00:00
[CommandHipc(0)]
[CommandTipc(0)] // 12.0.0+
// Initialize(pid, u64 reserved)
public ResultCode Initialize(ServiceCtx context)
2018-02-04 23:08:20 +00:00
{
_isInitialized = true;
2018-02-04 23:08:20 +00:00
return ResultCode.Success;
2018-02-04 23:08:20 +00:00
}
[CommandTipc(1)] // 12.0.0+
// GetService(ServiceName name) -> handle<move, session>
public ResultCode GetServiceTipc(ServiceCtx context)
{
context.Response.HandleDesc = IpcHandleDesc.MakeMove(0);
return GetService(context);
}
[CommandHipc(1)]
public ResultCode GetService(ServiceCtx context)
2018-02-04 23:08:20 +00:00
{
if (!_isInitialized)
2018-02-04 23:08:20 +00:00
{
return ResultCode.NotInitialized;
2018-02-04 23:08:20 +00:00
}
string name = ReadName(context);
2018-02-04 23:08:20 +00:00
if (name == string.Empty)
2018-02-04 23:08:20 +00:00
{
return ResultCode.InvalidName;
}
2018-02-04 23:08:20 +00:00
KSession session = new KSession(context.Device.System.KernelContext);
if (_registeredServices.TryGetValue(name, out KPort port))
{
KernelResult result = port.EnqueueIncomingSession(session.ServerSession);
if (result != KernelResult.Success)
2018-02-04 23:08:20 +00:00
{
throw new InvalidOperationException($"Session enqueue on port returned error \"{result}\".");
2018-02-04 23:08:20 +00:00
}
}
else
{
if (_services.TryGetValue(name, out Type type))
{
ServiceAttribute serviceAttribute = (ServiceAttribute)type.GetCustomAttributes(typeof(ServiceAttribute)).First(service => ((ServiceAttribute)service).Name == name);
IpcService service = serviceAttribute.Parameter != null
? (IpcService)Activator.CreateInstance(type, context, serviceAttribute.Parameter)
: (IpcService)Activator.CreateInstance(type, context);
service.TrySetServer(_commonServer);
service.Server.AddSessionObj(session.ServerSession, service);
}
else
{
if (context.Device.Configuration.IgnoreMissingServices)
{
Logger.Warning?.Print(LogClass.Service, $"Missing service {name} ignored");
}
else
{
throw new NotImplementedException(name);
}
}
}
if (context.Process.HandleTable.GenerateHandle(session.ClientSession, out int handle) != KernelResult.Success)
{
throw new InvalidOperationException("Out of handles!");
}
session.ServerSession.DecrementReferenceCount();
session.ClientSession.DecrementReferenceCount();
context.Response.HandleDesc = IpcHandleDesc.MakeMove(handle);
return ResultCode.Success;
}
[CommandHipc(2)]
// RegisterService(ServiceName name, u8 isLight, u32 maxHandles) -> handle<move, port>
public ResultCode RegisterServiceHipc(ServiceCtx context)
{
if (!_isInitialized)
{
return ResultCode.NotInitialized;
}
long namePosition = context.RequestData.BaseStream.Position;
string name = ReadName(context);
context.RequestData.BaseStream.Seek(namePosition + 8, SeekOrigin.Begin);
bool isLight = (context.RequestData.ReadInt32() & 1) != 0;
int maxSessions = context.RequestData.ReadInt32();
2018-02-04 23:08:20 +00:00
return RegisterService(context, name, isLight, maxSessions);
}
[CommandTipc(2)] // 12.0.0+
// RegisterService(ServiceName name, u32 maxHandles, u8 isLight) -> handle<move, port>
public ResultCode RegisterServiceTipc(ServiceCtx context)
{
if (!_isInitialized)
{
context.Response.HandleDesc = IpcHandleDesc.MakeMove(0);
return ResultCode.NotInitialized;
}
long namePosition = context.RequestData.BaseStream.Position;
string name = ReadName(context);
context.RequestData.BaseStream.Seek(namePosition + 8, SeekOrigin.Begin);
int maxSessions = context.RequestData.ReadInt32();
bool isLight = (context.RequestData.ReadInt32() & 1) != 0;
return RegisterService(context, name, isLight, maxSessions);
}
private ResultCode RegisterService(ServiceCtx context, string name, bool isLight, int maxSessions)
{
if (string.IsNullOrEmpty(name))
{
return ResultCode.InvalidName;
}
Logger.Info?.Print(LogClass.ServiceSm, $"Register \"{name}\".");
KPort port = new KPort(context.Device.System.KernelContext, maxSessions, isLight, 0);
2018-02-04 23:08:20 +00:00
if (!_registeredServices.TryAdd(name, port))
{
return ResultCode.AlreadyRegistered;
}
if (context.Process.HandleTable.GenerateHandle(port.ServerPort, out int handle) != KernelResult.Success)
{
throw new InvalidOperationException("Out of handles!");
}
2018-02-04 23:08:20 +00:00
context.Response.HandleDesc = IpcHandleDesc.MakeMove(handle);
2018-02-04 23:08:20 +00:00
return ResultCode.Success;
2018-02-04 23:08:20 +00:00
}
[CommandHipc(3)]
[CommandTipc(3)] // 12.0.0+
// UnregisterService(ServiceName name)
public ResultCode UnregisterService(ServiceCtx context)
{
if (!_isInitialized)
{
return ResultCode.NotInitialized;
}
long namePosition = context.RequestData.BaseStream.Position;
string name = ReadName(context);
context.RequestData.BaseStream.Seek(namePosition + 8, SeekOrigin.Begin);
bool isLight = (context.RequestData.ReadInt32() & 1) != 0;
int maxSessions = context.RequestData.ReadInt32();
if (string.IsNullOrEmpty(name))
{
return ResultCode.InvalidName;
}
if (!_registeredServices.TryRemove(name, out _))
{
return ResultCode.NotRegistered;
}
return ResultCode.Success;
}
private static string ReadName(ServiceCtx context)
{
string name = string.Empty;
for (int index = 0; index < 8 &&
context.RequestData.BaseStream.Position <
context.RequestData.BaseStream.Length; index++)
{
byte chr = context.RequestData.ReadByte();
if (chr >= 0x20 && chr < 0x7f)
{
name += (char)chr;
}
}
return name;
}
2018-02-04 23:08:20 +00:00
}
}