From ac67adae86c34584000fa6f15cf834839a8e6ccf Mon Sep 17 00:00:00 2001 From: TSR Berry <20988865+TSRBerry@users.noreply.github.com> Date: Mon, 29 Jul 2024 01:54:07 +0200 Subject: [PATCH] [WIP] Add ManagedProxySocket implementation --- .../Sockets/Bsd/Impl/ManagedProxySocket.cs | 408 ++++++++++++++++++ .../Sockets/Bsd/Proxy/ProxyManager.cs | 5 +- 2 files changed, 412 insertions(+), 1 deletion(-) create mode 100644 src/Ryujinx.HLE/HOS/Services/Sockets/Bsd/Impl/ManagedProxySocket.cs diff --git a/src/Ryujinx.HLE/HOS/Services/Sockets/Bsd/Impl/ManagedProxySocket.cs b/src/Ryujinx.HLE/HOS/Services/Sockets/Bsd/Impl/ManagedProxySocket.cs new file mode 100644 index 000000000..b2461836a --- /dev/null +++ b/src/Ryujinx.HLE/HOS/Services/Sockets/Bsd/Impl/ManagedProxySocket.cs @@ -0,0 +1,408 @@ +using Ryujinx.Common.Logging; +using Ryujinx.HLE.HOS.Services.Sockets.Bsd.Types; +using RyuSocks; +using RyuSocks.Auth; +using RyuSocks.Commands; +using RyuSocks.Types; +using System; +using System.Collections.Generic; +using System.Net; +using System.Net.Sockets; + +namespace Ryujinx.HLE.HOS.Services.Sockets.Bsd.Impl +{ + class ManagedProxySocket : ISocket + { + private static readonly Dictionary _authMethods = new() + { + { AuthMethod.NoAuth, new NoAuth() }, + }; + + private readonly bool _isUdpSocket; + private readonly bool _acceptedConnection; + + public SocksClient ProxyClient { get; } + + // TODO: Make sure Blocking is used properly + public bool Blocking { get; set; } + public int RefCount { get; set; } + + // TODO: Assign LocalEndPoint and RemoteEndPoint + public IPEndPoint RemoteEndPoint { get; private set; } + public IPEndPoint LocalEndPoint { get; private set; } + + public AddressFamily AddressFamily => ProxyClient.AddressFamily; + public SocketType SocketType => ProxyClient.SocketType; + public ProtocolType ProtocolType => ProxyClient.ProtocolType; + public IntPtr Handle => throw new NotSupportedException("Can't get the handle of a proxy socket."); + + public ManagedProxySocket(AddressFamily addressFamily, SocketType socketType, ProtocolType protocolType, EndPoint proxyEndpoint) + { + if (addressFamily != proxyEndpoint.AddressFamily && addressFamily != AddressFamily.Unspecified) + { + throw new ArgumentException( + $"Invalid {nameof(System.Net.Sockets.AddressFamily)}", nameof(addressFamily)); + } + + if (socketType != SocketType.Stream && socketType != SocketType.Dgram) + { + throw new ArgumentException( + $"Invalid {nameof(System.Net.Sockets.SocketType)}", nameof(socketType)); + } + + if (protocolType != ProtocolType.Tcp && protocolType != ProtocolType.Udp) + { + throw new ArgumentException( + $"Invalid {nameof(System.Net.Sockets.ProtocolType)}", nameof(protocolType)); + } + + _isUdpSocket = socketType == SocketType.Dgram && protocolType == ProtocolType.Udp; + + ProxyClient = proxyEndpoint switch + { + IPEndPoint ipEndPoint => new SocksClient(ipEndPoint) { OfferedAuthMethods = _authMethods }, + DnsEndPoint dnsEndPoint => new SocksClient(dnsEndPoint) { OfferedAuthMethods = _authMethods }, + _ => throw new ArgumentException($"Unsupported {nameof(EndPoint)} type", nameof(proxyEndpoint)) + }; + + ProxyClient.Authenticate(); + + RefCount = 1; + } + + private ManagedProxySocket(ManagedProxySocket oldSocket, SocksClient proxyClient) + { + ProxyClient = proxyClient; + LocalEndPoint = oldSocket.LocalEndPoint; + RemoteEndPoint = oldSocket.RemoteEndPoint; + _acceptedConnection = true; + RefCount = 1; + } + + private static LinuxError ToLinuxError(ReplyField proxyReply) + { + return proxyReply switch + { + ReplyField.Succeeded => LinuxError.SUCCESS, + ReplyField.ServerFailure => LinuxError.ECONNRESET, + ReplyField.ConnectionNotAllowed => LinuxError.ECONNREFUSED, + ReplyField.NetworkUnreachable => LinuxError.ENETUNREACH, + ReplyField.HostUnreachable => LinuxError.EHOSTUNREACH, + ReplyField.ConnectionRefused => LinuxError.ECONNREFUSED, + ReplyField.TTLExpired => LinuxError.EHOSTUNREACH, + ReplyField.CommandNotSupported => LinuxError.EOPNOTSUPP, + ReplyField.AddressTypeNotSupported => LinuxError.EAFNOSUPPORT, + _ => throw new ArgumentOutOfRangeException(nameof(proxyReply)) + }; + } + + public void Dispose() + { + ProxyClient.Dispose(); + } + + public LinuxError Read(out int readSize, Span buffer) + { + return Receive(out readSize, buffer, BsdSocketFlags.None); + } + + public LinuxError Write(out int writeSize, ReadOnlySpan buffer) + { + return Send(out writeSize, buffer, BsdSocketFlags.None); + } + + public LinuxError Receive(out int receiveSize, Span buffer, BsdSocketFlags flags) + { + LinuxError result; + bool shouldBlockAfterOperation = false; + + if (Blocking && flags.HasFlag(BsdSocketFlags.DontWait)) + { + Blocking = false; + shouldBlockAfterOperation = true; + } + + byte[] proxyBuffer = new byte[buffer.Length + ProxyClient.GetRequiredWrapperSpace()]; + + try + { + receiveSize = ProxyClient.Receive( + proxyBuffer, + WinSockHelper.ConvertBsdSocketFlags(flags), + out SocketError errorCode + ); + + proxyBuffer[..receiveSize].CopyTo(buffer); + + result = WinSockHelper.ConvertError((WsaError)errorCode); + } + catch (ProxyException exception) + { + Logger.Error?.Print( + LogClass.ServiceBsd, + $"An error occured while trying to receive data: {exception}" + ); + + receiveSize = -1; + result = ToLinuxError(exception.ReplyCode); + } + catch (SocketException exception) + { + receiveSize = -1; + result = WinSockHelper.ConvertError((WsaError)exception.ErrorCode); + } + + if (shouldBlockAfterOperation) + { + Blocking = true; + } + + return result; + } + + public LinuxError ReceiveFrom(out int receiveSize, Span buffer, int size, BsdSocketFlags flags, out IPEndPoint remoteEndPoint) + { + LinuxError result; + remoteEndPoint = new IPEndPoint(IPAddress.Any, 0); + bool shouldBlockAfterOperation = false; + + byte[] proxyBuffer = new byte[size + ProxyClient.GetRequiredWrapperSpace()]; + + if (Blocking && flags.HasFlag(BsdSocketFlags.DontWait)) + { + Blocking = false; + shouldBlockAfterOperation = true; + } + + try + { + EndPoint temp = new IPEndPoint(IPAddress.Any, 0); + + receiveSize = ProxyClient.ReceiveFrom(proxyBuffer, WinSockHelper.ConvertBsdSocketFlags(flags), ref temp); + + remoteEndPoint = (IPEndPoint)temp; + result = LinuxError.SUCCESS; + } + catch (ProxyException exception) + { + Logger.Error?.Print( + LogClass.ServiceBsd, + $"An error occured while trying to receive data: {exception}" + ); + + receiveSize = -1; + result = ToLinuxError(exception.ReplyCode); + } + catch (SocketException exception) + { + receiveSize = -1; + + result = WinSockHelper.ConvertError((WsaError)exception.ErrorCode); + } + + if (shouldBlockAfterOperation) + { + Blocking = true; + } + + return result; + } + + public LinuxError Send(out int sendSize, ReadOnlySpan buffer, BsdSocketFlags flags) + { + try + { + sendSize = ProxyClient.Send(buffer, WinSockHelper.ConvertBsdSocketFlags(flags)); + + return LinuxError.SUCCESS; + } + catch (ProxyException exception) + { + Logger.Error?.Print( + LogClass.ServiceBsd, + $"An error occured while trying to send data: {exception}" + ); + + sendSize = -1; + + return ToLinuxError(exception.ReplyCode); + } + catch (SocketException exception) + { + sendSize = -1; + + return WinSockHelper.ConvertError((WsaError)exception.ErrorCode); + } + } + + public LinuxError SendTo(out int sendSize, ReadOnlySpan buffer, int size, BsdSocketFlags flags, IPEndPoint remoteEndPoint) + { + try + { + // NOTE: sendSize might be larger than size and/or buffer.Length. + sendSize = ProxyClient.SendTo(buffer[..size], WinSockHelper.ConvertBsdSocketFlags(flags), remoteEndPoint); + + return LinuxError.SUCCESS; + } + catch (ProxyException exception) + { + Logger.Error?.Print( + LogClass.ServiceBsd, + $"An error occured while trying to send data: {exception}" + ); + + sendSize = -1; + + return ToLinuxError(exception.ReplyCode); + } + catch (SocketException exception) + { + sendSize = -1; + + return WinSockHelper.ConvertError((WsaError)exception.ErrorCode); + } + } + + public LinuxError RecvMMsg(out int vlen, BsdMMsgHdr message, BsdSocketFlags flags, TimeVal timeout) + { + throw new NotImplementedException(); + } + + public LinuxError SendMMsg(out int vlen, BsdMMsgHdr message, BsdSocketFlags flags) + { + throw new NotImplementedException(); + } + + public LinuxError GetSocketOption(BsdSocketOption option, SocketOptionLevel level, Span optionValue) + { + // TODO: Call ProxyClient.GetSocketOption() when it's implemented + throw new NotImplementedException(); + } + + public LinuxError SetSocketOption(BsdSocketOption option, SocketOptionLevel level, ReadOnlySpan optionValue) + { + // TODO: Call ProxyClient.SetSocketOption() when it's implemented + throw new NotImplementedException(); + } + + public bool Poll(int microSeconds, SelectMode mode) + { + // TODO: Call ProxyClient.Poll() when it's implemented + throw new NotImplementedException(); + } + + public LinuxError Bind(IPEndPoint localEndPoint) + { + ProxyClient.RequestCommand = _isUdpSocket ? ProxyCommand.UdpAssociate : ProxyCommand.Bind; + + try + { + ProxyClient.Bind(localEndPoint); + } + catch (ProxyException exception) + { + Logger.Error?.Print( + LogClass.ServiceBsd, + $"Request for {ProxyClient.RequestCommand} command failed: {exception}" + ); + + return ToLinuxError(exception.ReplyCode); + } + catch (SocketException exception) + { + return WinSockHelper.ConvertError((WsaError)exception.ErrorCode); + } + + return LinuxError.SUCCESS; + } + + public LinuxError Connect(IPEndPoint remoteEndPoint) + { + ProxyClient.RequestCommand = ProxyCommand.Connect; + + try + { + ProxyClient.Connect(remoteEndPoint.Address, remoteEndPoint.Port); + } + catch (ProxyException exception) + { + Logger.Error?.Print( + LogClass.ServiceBsd, + $"Request for {ProxyClient.RequestCommand} command failed: {exception}" + ); + + return ToLinuxError(exception.ReplyCode); + } + catch (SocketException exception) + { + return WinSockHelper.ConvertError((WsaError)exception.ErrorCode); + } + + return LinuxError.SUCCESS; + } + + public LinuxError Listen(int backlog) + { + // NOTE: Only one client can connect with the default SOCKS5 commands. + if (ProxyClient.RequestCommand != ProxyCommand.Bind) + { + return LinuxError.EOPNOTSUPP; + } + + return LinuxError.SUCCESS; + } + + public LinuxError Accept(out ISocket newSocket) + { + newSocket = null; + + if (ProxyClient.RequestCommand != ProxyCommand.Bind) + { + return LinuxError.EOPNOTSUPP; + } + + // NOTE: Only one client can connect with the default SOCKS5 commands. + if (_acceptedConnection) + { + return LinuxError.EOPNOTSUPP; + } + + try + { + SocksClient newProxyClient = ProxyClient.Accept(); + newSocket = new ManagedProxySocket(this, newProxyClient); + } + catch (ProxyException exception) + { + Logger.Error?.Print( + LogClass.ServiceBsd, + $"Failed to accept client connection: {exception}" + ); + + return ToLinuxError(exception.ReplyCode); + } + catch (SocketException exception) + { + return WinSockHelper.ConvertError((WsaError)exception.ErrorCode); + } + + return LinuxError.SUCCESS; + } + + public void Disconnect() + { + // TODO: Call ProxyClient.Disconnect() when it's implemented + } + + public LinuxError Shutdown(BsdSocketShutdownFlags how) + { + // TODO: Call ProxyClient.Shutdown() when it's implemented + return LinuxError.SUCCESS; + } + + public void Close() + { + ProxyClient.Close(); + } + } +} diff --git a/src/Ryujinx.HLE/HOS/Services/Sockets/Bsd/Proxy/ProxyManager.cs b/src/Ryujinx.HLE/HOS/Services/Sockets/Bsd/Proxy/ProxyManager.cs index a9a219273..c5a0b5ed3 100644 --- a/src/Ryujinx.HLE/HOS/Services/Sockets/Bsd/Proxy/ProxyManager.cs +++ b/src/Ryujinx.HLE/HOS/Services/Sockets/Bsd/Proxy/ProxyManager.cs @@ -18,7 +18,10 @@ namespace Ryujinx.HLE.HOS.Services.Sockets.Bsd.Proxy internal static ISocket GetSocket(AddressFamily addressFamily, SocketType socketType, ProtocolType protocolType) { - // TODO: Return proxy socket if AddressFamily, SocketType and ProtocolType match. + if (_proxyEndpoints.TryGetValue(GetKey(addressFamily, socketType, protocolType), out EndPoint endPoint)) + { + return new ManagedProxySocket(addressFamily, socketType, protocolType, endPoint); + } return new ManagedSocket(addressFamily, socketType, protocolType); }