Ryujinx/Ryujinx.Horizon/Sm/Impl/ServiceManager.cs
gdkchan 08831eecf7
IPC refactor part 3+4: New server HIPC message processor (#4188)
* IPC refactor part 3 + 4: New server HIPC message processor with source generator based serialization

* Make types match on calls to AlignUp/AlignDown

* Formatting

* Address some PR feedback

* Move BitfieldExtensions to Ryujinx.Common.Utilities and consolidate implementations

* Rename Reader/Writer to SpanReader/SpanWriter and move to Ryujinx.Common.Memory

* Implement EventType

* Address more PR feedback

* Log request processing errors since they are not normal

* Rename waitable to multiwait and add missing lock

* PR feedback

* Ac_K PR feedback
2023-01-04 23:15:45 +01:00

197 lines
5.1 KiB
C#

using Ryujinx.Horizon.Common;
using Ryujinx.Horizon.Sdk.OsTypes;
using Ryujinx.Horizon.Sdk.Sf;
using Ryujinx.Horizon.Sdk.Sm;
namespace Ryujinx.Horizon.Sm.Impl
{
class ServiceManager
{
private const int MaxServicesCount = 256;
private readonly ServiceInfo[] _services;
public ServiceManager()
{
_services = new ServiceInfo[MaxServicesCount];
}
public Result GetService(out int handle, ulong processId, ServiceName name)
{
handle = 0;
Result result = ValidateServiceName(name);
if (result.IsFailure)
{
return result;
}
// TODO: Validation with GetProcessInfo etc.
int serviceIndex = GetServiceInfo(name);
if (serviceIndex < 0)
{
return SfResult.RequestDeferredByUser;
}
result = GetServiceImpl(out handle, ref _services[serviceIndex]);
if (result == KernelResult.SessionCountExceeded)
{
return SmResult.OutOfSessions;
}
return result;
}
private Result GetServiceImpl(out int handle, ref ServiceInfo serviceInfo)
{
return HorizonStatic.Syscall.ConnectToPort(out handle, serviceInfo.PortHandle);
}
public Result RegisterService(out int handle, ulong processId, ServiceName name, int maxSessions, bool isLight)
{
handle = 0;
Result result = ValidateServiceName(name);
if (result.IsFailure)
{
return result;
}
// TODO: Validation with GetProcessInfo etc.
if (HasServiceInfo(name))
{
return SmResult.AlreadyRegistered;
}
return RegisterServiceImpl(out handle, processId, name, maxSessions, isLight);
}
public Result RegisterServiceForSelf(out int handle, ServiceName name, int maxSessions)
{
return RegisterServiceImpl(out handle, Os.GetCurrentProcessId(), name, maxSessions, false);
}
private Result RegisterServiceImpl(out int handle, ulong processId, ServiceName name, int maxSessions, bool isLight)
{
handle = 0;
Result result = ValidateServiceName(name);
if (!result.IsSuccess)
{
return result;
}
if (HasServiceInfo(name))
{
return SmResult.AlreadyRegistered;
}
int freeServiceIndex = GetFreeService();
if (freeServiceIndex < 0)
{
return SmResult.OutOfServices;
}
ref ServiceInfo freeService = ref _services[freeServiceIndex];
result = HorizonStatic.Syscall.CreatePort(out handle, out int clientPort, maxSessions, isLight, null);
if (!result.IsSuccess)
{
return result;
}
freeService.PortHandle = clientPort;
freeService.Name = name;
freeService.OwnerProcessId = processId;
return Result.Success;
}
public Result UnregisterService(ulong processId, ServiceName name)
{
Result result = ValidateServiceName(name);
if (result.IsFailure)
{
return result;
}
// TODO: Validation with GetProcessInfo etc.
int serviceIndex = GetServiceInfo(name);
if (serviceIndex < 0)
{
return SmResult.NotRegistered;
}
ref var serviceInfo = ref _services[serviceIndex];
if (serviceInfo.OwnerProcessId != processId)
{
return SmResult.NotAllowed;
}
serviceInfo.Free();
return Result.Success;
}
private static Result ValidateServiceName(ServiceName name)
{
if (name[0] == 0)
{
return SmResult.InvalidServiceName;
}
int nameLength = 1;
for (; nameLength < name.Length; nameLength++)
{
if (name[nameLength] == 0)
{
break;
}
}
while (nameLength < name.Length)
{
if (name[nameLength++] != 0)
{
return SmResult.InvalidServiceName;
}
}
return Result.Success;
}
private bool HasServiceInfo(ServiceName name)
{
return GetServiceInfo(name) != -1;
}
private int GetFreeService()
{
return GetServiceInfo(ServiceName.Invalid);
}
private int GetServiceInfo(ServiceName name)
{
for (int index = 0; index < MaxServicesCount; index++)
{
if (_services[index].Name == name)
{
return index;
}
}
return -1;
}
}
}