2022-01-12 13:31:08 -05:00
|
|
|
|
using System.Collections.Concurrent;
|
|
|
|
|
using System.Collections.Generic;
|
|
|
|
|
|
|
|
|
|
namespace Ryujinx.HLE.HOS.Services.Sockets.Bsd
|
|
|
|
|
{
|
|
|
|
|
class BsdContext
|
|
|
|
|
{
|
2022-02-09 15:18:07 -05:00
|
|
|
|
private static ConcurrentDictionary<ulong, BsdContext> _registry = new ConcurrentDictionary<ulong, BsdContext>();
|
2022-01-12 13:31:08 -05:00
|
|
|
|
|
|
|
|
|
private readonly object _lock = new object();
|
|
|
|
|
|
|
|
|
|
private List<IFileDescriptor> _fds;
|
|
|
|
|
|
|
|
|
|
private BsdContext()
|
|
|
|
|
{
|
|
|
|
|
_fds = new List<IFileDescriptor>();
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public ISocket RetrieveSocket(int socketFd)
|
|
|
|
|
{
|
|
|
|
|
IFileDescriptor file = RetrieveFileDescriptor(socketFd);
|
|
|
|
|
|
|
|
|
|
if (file is ISocket socket)
|
|
|
|
|
{
|
|
|
|
|
return socket;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return null;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public IFileDescriptor RetrieveFileDescriptor(int fd)
|
|
|
|
|
{
|
|
|
|
|
lock (_lock)
|
|
|
|
|
{
|
|
|
|
|
if (fd >= 0 && _fds.Count > fd)
|
|
|
|
|
{
|
|
|
|
|
return _fds[fd];
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return null;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public int RegisterFileDescriptor(IFileDescriptor file)
|
|
|
|
|
{
|
|
|
|
|
lock (_lock)
|
|
|
|
|
{
|
|
|
|
|
for (int fd = 0; fd < _fds.Count; fd++)
|
|
|
|
|
{
|
|
|
|
|
if (_fds[fd] == null)
|
|
|
|
|
{
|
|
|
|
|
_fds[fd] = file;
|
|
|
|
|
|
|
|
|
|
return fd;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
_fds.Add(file);
|
|
|
|
|
|
|
|
|
|
return _fds.Count - 1;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public int DuplicateFileDescriptor(int fd)
|
|
|
|
|
{
|
|
|
|
|
IFileDescriptor oldFile = RetrieveFileDescriptor(fd);
|
|
|
|
|
|
|
|
|
|
if (oldFile != null)
|
|
|
|
|
{
|
|
|
|
|
lock (_lock)
|
|
|
|
|
{
|
|
|
|
|
oldFile.Refcount++;
|
|
|
|
|
|
|
|
|
|
return RegisterFileDescriptor(oldFile);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return -1;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public bool CloseFileDescriptor(int fd)
|
|
|
|
|
{
|
|
|
|
|
IFileDescriptor file = RetrieveFileDescriptor(fd);
|
|
|
|
|
|
|
|
|
|
if (file != null)
|
|
|
|
|
{
|
|
|
|
|
file.Refcount--;
|
|
|
|
|
|
|
|
|
|
if (file.Refcount <= 0)
|
|
|
|
|
{
|
|
|
|
|
file.Dispose();
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
lock (_lock)
|
|
|
|
|
{
|
|
|
|
|
_fds[fd] = null;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return true;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return false;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public LinuxError ShutdownAllSockets(BsdSocketShutdownFlags how)
|
|
|
|
|
{
|
|
|
|
|
lock (_lock)
|
|
|
|
|
{
|
|
|
|
|
foreach (IFileDescriptor file in _fds)
|
|
|
|
|
{
|
|
|
|
|
if (file is ISocket socket)
|
|
|
|
|
{
|
|
|
|
|
LinuxError errno = socket.Shutdown(how);
|
|
|
|
|
|
|
|
|
|
if (errno != LinuxError.SUCCESS)
|
|
|
|
|
{
|
|
|
|
|
return errno;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return LinuxError.SUCCESS;
|
|
|
|
|
}
|
|
|
|
|
|
2022-02-09 15:18:07 -05:00
|
|
|
|
public static BsdContext GetOrRegister(ulong processId)
|
2022-01-12 13:31:08 -05:00
|
|
|
|
{
|
|
|
|
|
BsdContext context = GetContext(processId);
|
|
|
|
|
|
|
|
|
|
if (context == null)
|
|
|
|
|
{
|
|
|
|
|
context = new BsdContext();
|
|
|
|
|
|
|
|
|
|
_registry.TryAdd(processId, context);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return context;
|
|
|
|
|
}
|
|
|
|
|
|
2022-02-09 15:18:07 -05:00
|
|
|
|
public static BsdContext GetContext(ulong processId)
|
2022-01-12 13:31:08 -05:00
|
|
|
|
{
|
|
|
|
|
if (!_registry.TryGetValue(processId, out BsdContext processContext))
|
|
|
|
|
{
|
|
|
|
|
return null;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return processContext;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|