Ryujinx/Ryujinx.HLE/HOS/Services/Sm/IUserInterface.cs

181 lines
5.4 KiB
C#
Raw Normal View History

using Ryujinx.Common.Logging;
using Ryujinx.HLE.HOS.Ipc;
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;
2018-02-05 00:08:20 +01:00
namespace Ryujinx.HLE.HOS.Services.Sm
2018-02-05 00:08:20 +01:00
{
class IUserInterface : IpcService
2018-02-05 00:08:20 +01:00
{
private Dictionary<int, ServiceProcessRequest> _commands;
public override IReadOnlyDictionary<int, ServiceProcessRequest> Commands => _commands;
private ConcurrentDictionary<string, KPort> _registeredServices;
private bool _isInitialized;
public IUserInterface()
{
_commands = new Dictionary<int, ServiceProcessRequest>
{
{ 0, Initialize },
{ 1, GetService },
{ 2, RegisterService },
{ 3, UnregisterService }
};
_registeredServices = new ConcurrentDictionary<string, KPort>();
}
public static void InitializePort(Horizon system)
{
KPort port = new KPort(system, 256, false, 0);
port.ClientPort.SetName("sm:");
port.ClientPort.Service = new IUserInterface();
}
2018-02-05 00:08:20 +01:00
public long Initialize(ServiceCtx context)
2018-02-05 00:08:20 +01:00
{
_isInitialized = true;
2018-02-05 00:08:20 +01:00
return 0;
}
public long GetService(ServiceCtx context)
2018-02-05 00:08:20 +01:00
{
if (!_isInitialized)
2018-02-05 00:08:20 +01:00
{
return ErrorCode.MakeError(ErrorModule.Sm, SmErr.NotInitialized);
2018-02-05 00:08:20 +01:00
}
string name = ReadName(context);
2018-02-05 00:08:20 +01:00
if (name == string.Empty)
2018-02-05 00:08:20 +01:00
{
return ErrorCode.MakeError(ErrorModule.Sm, SmErr.InvalidName);
}
2018-02-05 00:08:20 +01:00
KSession session = new KSession(context.Device.System);
if (_registeredServices.TryGetValue(name, out KPort port))
{
KernelResult result = port.EnqueueIncomingSession(session.ServerSession);
if (result != KernelResult.Success)
2018-02-05 00:08:20 +01:00
{
throw new InvalidOperationException($"Session enqueue on port returned error \"{result}\".");
2018-02-05 00:08:20 +01:00
}
}
else
{
session.ClientSession.Service = ServiceFactory.MakeService(context.Device.System, name);
}
if (context.Process.HandleTable.GenerateHandle(session.ClientSession, out int handle) != KernelResult.Success)
{
throw new InvalidOperationException("Out of handles!");
}
context.Response.HandleDesc = IpcHandleDesc.MakeMove(handle);
return 0;
}
public long RegisterService(ServiceCtx context)
{
if (!_isInitialized)
{
return ErrorCode.MakeError(ErrorModule.Sm, SmErr.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-05 00:08:20 +01:00
if (name == string.Empty)
{
return ErrorCode.MakeError(ErrorModule.Sm, SmErr.InvalidName);
}
Logger.PrintInfo(LogClass.ServiceSm, $"Register \"{name}\".");
KPort port = new KPort(context.Device.System, maxSessions, isLight, 0);
2018-02-05 00:08:20 +01:00
if (!_registeredServices.TryAdd(name, port))
{
return ErrorCode.MakeError(ErrorModule.Sm, SmErr.AlreadyRegistered);
}
if (context.Process.HandleTable.GenerateHandle(port.ServerPort, out int handle) != KernelResult.Success)
{
throw new InvalidOperationException("Out of handles!");
}
2018-02-05 00:08:20 +01:00
context.Response.HandleDesc = IpcHandleDesc.MakeMove(handle);
2018-02-05 00:08:20 +01:00
return 0;
}
public long UnregisterService(ServiceCtx context)
{
if (!_isInitialized)
{
return ErrorCode.MakeError(ErrorModule.Sm, SmErr.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 (name == string.Empty)
{
return ErrorCode.MakeError(ErrorModule.Sm, SmErr.InvalidName);
}
if (!_registeredServices.TryRemove(name, out _))
{
return ErrorCode.MakeError(ErrorModule.Sm, SmErr.NotRegistered);
}
return 0;
}
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-05 00:08:20 +01:00
}
}