| 12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121 |
- using Ryujinx.Common;
- using Ryujinx.Common.Logging;
- using Ryujinx.HLE.HOS.Services.Sockets.Bsd.Impl;
- using Ryujinx.HLE.HOS.Services.Sockets.Bsd.Types;
- using Ryujinx.Memory;
- using System;
- using System.Collections.Generic;
- using System.Net;
- using System.Net.Sockets;
- using System.Numerics;
- using System.Runtime.CompilerServices;
- using System.Text;
- namespace Ryujinx.HLE.HOS.Services.Sockets.Bsd
- {
- [Service("bsd:s", true)]
- [Service("bsd:u", false)]
- class IClient : IpcService
- {
- private static readonly List<IPollManager> _pollManagers = new List<IPollManager>
- {
- EventFileDescriptorPollManager.Instance,
- ManagedSocketPollManager.Instance
- };
- private BsdContext _context;
- private bool _isPrivileged;
- public IClient(ServiceCtx context, bool isPrivileged) : base(context.Device.System.BsdServer)
- {
- _isPrivileged = isPrivileged;
- }
- private ResultCode WriteBsdResult(ServiceCtx context, int result, LinuxError errorCode = LinuxError.SUCCESS)
- {
- if (errorCode != LinuxError.SUCCESS)
- {
- result = -1;
- }
- context.ResponseData.Write(result);
- context.ResponseData.Write((int)errorCode);
- return ResultCode.Success;
- }
- private static AddressFamily ConvertBsdAddressFamily(BsdAddressFamily family)
- {
- switch (family)
- {
- case BsdAddressFamily.Unspecified:
- return AddressFamily.Unspecified;
- case BsdAddressFamily.InterNetwork:
- return AddressFamily.InterNetwork;
- case BsdAddressFamily.InterNetworkV6:
- return AddressFamily.InterNetworkV6;
- case BsdAddressFamily.Unknown:
- return AddressFamily.Unknown;
- default:
- throw new NotImplementedException(family.ToString());
- }
- }
- private LinuxError SetResultErrno(IFileDescriptor socket, int result)
- {
- return result == 0 && !socket.Blocking ? LinuxError.EWOULDBLOCK : LinuxError.SUCCESS;
- }
- private ResultCode SocketInternal(ServiceCtx context, bool exempt)
- {
- BsdAddressFamily domain = (BsdAddressFamily)context.RequestData.ReadInt32();
- BsdSocketType type = (BsdSocketType)context.RequestData.ReadInt32();
- ProtocolType protocol = (ProtocolType)context.RequestData.ReadInt32();
- BsdSocketCreationFlags creationFlags = (BsdSocketCreationFlags)((int)type >> (int)BsdSocketCreationFlags.FlagsShift);
- type &= BsdSocketType.TypeMask;
- if (domain == BsdAddressFamily.Unknown)
- {
- return WriteBsdResult(context, -1, LinuxError.EPROTONOSUPPORT);
- }
- else if ((type == BsdSocketType.Seqpacket || type == BsdSocketType.Raw) && !_isPrivileged)
- {
- if (domain != BsdAddressFamily.InterNetwork || type != BsdSocketType.Raw || protocol != ProtocolType.Icmp)
- {
- return WriteBsdResult(context, -1, LinuxError.ENOENT);
- }
- }
- AddressFamily netDomain = ConvertBsdAddressFamily(domain);
- if (protocol == ProtocolType.IP)
- {
- if (type == BsdSocketType.Stream)
- {
- protocol = ProtocolType.Tcp;
- }
- else if (type == BsdSocketType.Dgram)
- {
- protocol = ProtocolType.Udp;
- }
- }
- ISocket newBsdSocket = new ManagedSocket(netDomain, (SocketType)type, protocol);
- newBsdSocket.Blocking = !creationFlags.HasFlag(BsdSocketCreationFlags.NonBlocking);
- LinuxError errno = LinuxError.SUCCESS;
- int newSockFd = _context.RegisterFileDescriptor(newBsdSocket);
- if (newSockFd == -1)
- {
- errno = LinuxError.EBADF;
- }
- if (exempt)
- {
- newBsdSocket.Disconnect();
- }
- return WriteBsdResult(context, newSockFd, errno);
- }
- private void WriteSockAddr(ServiceCtx context, ulong bufferPosition, ISocket socket, bool isRemote)
- {
- IPEndPoint endPoint = isRemote ? socket.RemoteEndPoint : socket.LocalEndPoint;
- context.Memory.Write(bufferPosition, BsdSockAddr.FromIPEndPoint(endPoint));
- }
- [CommandHipc(0)]
- // Initialize(nn::socket::BsdBufferConfig config, u64 pid, u64 transferMemorySize, KObject<copy, transfer_memory>, pid) -> u32 bsd_errno
- public ResultCode RegisterClient(ServiceCtx context)
- {
- _context = BsdContext.GetOrRegister(context.Request.HandleDesc.PId);
- /*
- typedef struct {
- u32 version; // Observed 1 on 2.0 LibAppletWeb, 2 on 3.0.
- u32 tcp_tx_buf_size; // Size of the TCP transfer (send) buffer (initial or fixed).
- u32 tcp_rx_buf_size; // Size of the TCP recieve buffer (initial or fixed).
- u32 tcp_tx_buf_max_size; // Maximum size of the TCP transfer (send) buffer. If it is 0, the size of the buffer is fixed to its initial value.
- u32 tcp_rx_buf_max_size; // Maximum size of the TCP receive buffer. If it is 0, the size of the buffer is fixed to its initial value.
- u32 udp_tx_buf_size; // Size of the UDP transfer (send) buffer (typically 0x2400 bytes).
- u32 udp_rx_buf_size; // Size of the UDP receive buffer (typically 0xA500 bytes).
- u32 sb_efficiency; // Number of buffers for each socket (standard values range from 1 to 8).
- } BsdBufferConfig;
- */
- // bsd_error
- context.ResponseData.Write(0);
- Logger.Stub?.PrintStub(LogClass.ServiceBsd);
- // Close transfer memory immediately as we don't use it.
- context.Device.System.KernelContext.Syscall.CloseHandle(context.Request.HandleDesc.ToCopy[0]);
- return ResultCode.Success;
- }
- [CommandHipc(1)]
- // StartMonitoring(u64, pid)
- public ResultCode StartMonitoring(ServiceCtx context)
- {
- ulong unknown0 = context.RequestData.ReadUInt64();
- Logger.Stub?.PrintStub(LogClass.ServiceBsd, new { unknown0 });
- return ResultCode.Success;
- }
- [CommandHipc(2)]
- // Socket(u32 domain, u32 type, u32 protocol) -> (i32 ret, u32 bsd_errno)
- public ResultCode Socket(ServiceCtx context)
- {
- return SocketInternal(context, false);
- }
- [CommandHipc(3)]
- // SocketExempt(u32 domain, u32 type, u32 protocol) -> (i32 ret, u32 bsd_errno)
- public ResultCode SocketExempt(ServiceCtx context)
- {
- return SocketInternal(context, true);
- }
- [CommandHipc(4)]
- // Open(u32 flags, array<unknown, 0x21> path) -> (i32 ret, u32 bsd_errno)
- public ResultCode Open(ServiceCtx context)
- {
- (ulong bufferPosition, ulong bufferSize) = context.Request.GetBufferType0x21();
- int flags = context.RequestData.ReadInt32();
- byte[] rawPath = new byte[bufferSize];
- context.Memory.Read(bufferPosition, rawPath);
- string path = Encoding.ASCII.GetString(rawPath);
- WriteBsdResult(context, -1, LinuxError.EOPNOTSUPP);
- Logger.Stub?.PrintStub(LogClass.ServiceBsd, new { path, flags });
- return ResultCode.Success;
- }
- [CommandHipc(5)]
- // Select(u32 nfds, nn::socket::timeval timeout, buffer<nn::socket::fd_set, 0x21, 0> readfds_in, buffer<nn::socket::fd_set, 0x21, 0> writefds_in, buffer<nn::socket::fd_set, 0x21, 0> errorfds_in)
- // -> (i32 ret, u32 bsd_errno, buffer<nn::socket::fd_set, 0x22, 0> readfds_out, buffer<nn::socket::fd_set, 0x22, 0> writefds_out, buffer<nn::socket::fd_set, 0x22, 0> errorfds_out)
- public ResultCode Select(ServiceCtx context)
- {
- int fdsCount = context.RequestData.ReadInt32();
- int timeout = context.RequestData.ReadInt32();
- (ulong readFdsInBufferPosition, ulong readFdsInBufferSize) = context.Request.GetBufferType0x21(0);
- (ulong writeFdsInBufferPosition, ulong writeFdsInBufferSize) = context.Request.GetBufferType0x21(1);
- (ulong errorFdsInBufferPosition, ulong errorFdsInBufferSize) = context.Request.GetBufferType0x21(2);
- (ulong readFdsOutBufferPosition, ulong readFdsOutBufferSize) = context.Request.GetBufferType0x22(0);
- (ulong writeFdsOutBufferPosition, ulong writeFdsOutBufferSize) = context.Request.GetBufferType0x22(1);
- (ulong errorFdsOutBufferPosition, ulong errorFdsOutBufferSize) = context.Request.GetBufferType0x22(2);
- List<IFileDescriptor> readFds = _context.RetrieveFileDescriptorsFromMask(context.Memory.GetSpan(readFdsInBufferPosition, (int)readFdsInBufferSize));
- List<IFileDescriptor> writeFds = _context.RetrieveFileDescriptorsFromMask(context.Memory.GetSpan(writeFdsInBufferPosition, (int)writeFdsInBufferSize));
- List<IFileDescriptor> errorFds = _context.RetrieveFileDescriptorsFromMask(context.Memory.GetSpan(errorFdsInBufferPosition, (int)errorFdsInBufferSize));
- int actualFdsCount = readFds.Count + writeFds.Count + errorFds.Count;
- if (fdsCount == 0 || actualFdsCount == 0)
- {
- WriteBsdResult(context, 0);
- return ResultCode.Success;
- }
- PollEvent[] events = new PollEvent[actualFdsCount];
- int index = 0;
- foreach (IFileDescriptor fd in readFds)
- {
- events[index] = new PollEvent(new PollEventData { InputEvents = PollEventTypeMask.Input }, fd);
- index++;
- }
- foreach (IFileDescriptor fd in writeFds)
- {
- events[index] = new PollEvent(new PollEventData { InputEvents = PollEventTypeMask.Output }, fd);
- index++;
- }
- foreach (IFileDescriptor fd in errorFds)
- {
- events[index] = new PollEvent(new PollEventData { InputEvents = PollEventTypeMask.Error }, fd);
- index++;
- }
- List<PollEvent>[] eventsByPollManager = new List<PollEvent>[_pollManagers.Count];
- for (int i = 0; i < eventsByPollManager.Length; i++)
- {
- eventsByPollManager[i] = new List<PollEvent>();
- foreach (PollEvent evnt in events)
- {
- if (_pollManagers[i].IsCompatible(evnt))
- {
- eventsByPollManager[i].Add(evnt);
- }
- }
- }
- int updatedCount = 0;
- for (int i = 0; i < _pollManagers.Count; i++)
- {
- if (eventsByPollManager[i].Count > 0)
- {
- _pollManagers[i].Select(eventsByPollManager[i], timeout, out int updatedPollCount);
- updatedCount += updatedPollCount;
- }
- }
- readFds.Clear();
- writeFds.Clear();
- errorFds.Clear();
- foreach (PollEvent pollEvent in events)
- {
- for (int i = 0; i < _pollManagers.Count; i++)
- {
- if (eventsByPollManager[i].Contains(pollEvent))
- {
- if (pollEvent.Data.OutputEvents.HasFlag(PollEventTypeMask.Input))
- {
- readFds.Add(pollEvent.FileDescriptor);
- }
- if (pollEvent.Data.OutputEvents.HasFlag(PollEventTypeMask.Output))
- {
- writeFds.Add(pollEvent.FileDescriptor);
- }
- if (pollEvent.Data.OutputEvents.HasFlag(PollEventTypeMask.Error))
- {
- errorFds.Add(pollEvent.FileDescriptor);
- }
- }
- }
- }
- using var readFdsOut = context.Memory.GetWritableRegion(readFdsOutBufferPosition, (int)readFdsOutBufferSize);
- using var writeFdsOut = context.Memory.GetWritableRegion(writeFdsOutBufferPosition, (int)writeFdsOutBufferSize);
- using var errorFdsOut = context.Memory.GetWritableRegion(errorFdsOutBufferPosition, (int)errorFdsOutBufferSize);
- _context.BuildMask(readFds, readFdsOut.Memory.Span);
- _context.BuildMask(writeFds, writeFdsOut.Memory.Span);
- _context.BuildMask(errorFds, errorFdsOut.Memory.Span);
- WriteBsdResult(context, updatedCount);
- return ResultCode.Success;
- }
- [CommandHipc(6)]
- // Poll(u32 nfds, u32 timeout, buffer<unknown, 0x21, 0> fds) -> (i32 ret, u32 bsd_errno, buffer<unknown, 0x22, 0>)
- public ResultCode Poll(ServiceCtx context)
- {
- int fdsCount = context.RequestData.ReadInt32();
- int timeout = context.RequestData.ReadInt32();
- (ulong inputBufferPosition, ulong inputBufferSize) = context.Request.GetBufferType0x21();
- (ulong outputBufferPosition, ulong outputBufferSize) = context.Request.GetBufferType0x22();
- if (timeout < -1 || fdsCount < 0 || (ulong)(fdsCount * 8) > inputBufferSize)
- {
- return WriteBsdResult(context, -1, LinuxError.EINVAL);
- }
- PollEvent[] events = new PollEvent[fdsCount];
- for (int i = 0; i < fdsCount; i++)
- {
- PollEventData pollEventData = context.Memory.Read<PollEventData>(inputBufferPosition + (ulong)(i * Unsafe.SizeOf<PollEventData>()));
- IFileDescriptor fileDescriptor = _context.RetrieveFileDescriptor(pollEventData.SocketFd);
- if (fileDescriptor == null)
- {
- return WriteBsdResult(context, -1, LinuxError.EBADF);
- }
- events[i] = new PollEvent(pollEventData, fileDescriptor);
- }
- List<PollEvent> discoveredEvents = new List<PollEvent>();
- List<PollEvent>[] eventsByPollManager = new List<PollEvent>[_pollManagers.Count];
- for (int i = 0; i < eventsByPollManager.Length; i++)
- {
- eventsByPollManager[i] = new List<PollEvent>();
- foreach (PollEvent evnt in events)
- {
- if (_pollManagers[i].IsCompatible(evnt))
- {
- eventsByPollManager[i].Add(evnt);
- discoveredEvents.Add(evnt);
- }
- }
- }
- foreach (PollEvent evnt in events)
- {
- if (!discoveredEvents.Contains(evnt))
- {
- Logger.Error?.Print(LogClass.ServiceBsd, $"Poll operation is not supported for {evnt.FileDescriptor.GetType().Name}!");
- return WriteBsdResult(context, -1, LinuxError.EBADF);
- }
- }
- int updateCount = 0;
- LinuxError errno = LinuxError.SUCCESS;
- if (fdsCount != 0)
- {
- bool IsUnexpectedLinuxError(LinuxError error)
- {
- return error != LinuxError.SUCCESS && error != LinuxError.ETIMEDOUT;
- }
- // Hybrid approach
- long budgetLeftMilliseconds;
- if (timeout == -1)
- {
- budgetLeftMilliseconds = PerformanceCounter.ElapsedMilliseconds + uint.MaxValue;
- }
- else
- {
- budgetLeftMilliseconds = PerformanceCounter.ElapsedMilliseconds + timeout;
- }
- do
- {
- for (int i = 0; i < eventsByPollManager.Length; i++)
- {
- if (eventsByPollManager[i].Count == 0)
- {
- continue;
- }
- errno = _pollManagers[i].Poll(eventsByPollManager[i], 0, out updateCount);
- if (IsUnexpectedLinuxError(errno))
- {
- break;
- }
- if (updateCount > 0)
- {
- break;
- }
- }
- if (updateCount > 0)
- {
- break;
- }
- // If we are here, that mean nothing was available, sleep for 50ms
- context.Device.System.KernelContext.Syscall.SleepThread(50 * 1000000);
- }
- while (PerformanceCounter.ElapsedMilliseconds < budgetLeftMilliseconds);
- }
- else if (timeout == -1)
- {
- // FIXME: If we get a timeout of -1 and there is no fds to wait on, this should kill the KProcess. (need to check that with re)
- throw new InvalidOperationException();
- }
- else
- {
- context.Device.System.KernelContext.Syscall.SleepThread(timeout);
- }
- // TODO: Spanify
- for (int i = 0; i < fdsCount; i++)
- {
- context.Memory.Write(outputBufferPosition + (ulong)(i * Unsafe.SizeOf<PollEventData>()), events[i].Data);
- }
- // In case of non blocking call timeout should not be returned.
- if (timeout == 0 && errno == LinuxError.ETIMEDOUT)
- {
- errno = LinuxError.SUCCESS;
- }
- return WriteBsdResult(context, updateCount, errno);
- }
- [CommandHipc(7)]
- // Sysctl(buffer<unknown, 0x21, 0>, buffer<unknown, 0x21, 0>) -> (i32 ret, u32 bsd_errno, u32, buffer<unknown, 0x22, 0>)
- public ResultCode Sysctl(ServiceCtx context)
- {
- WriteBsdResult(context, -1, LinuxError.EOPNOTSUPP);
- Logger.Stub?.PrintStub(LogClass.ServiceBsd);
- return ResultCode.Success;
- }
- [CommandHipc(8)]
- // Recv(u32 socket, u32 flags) -> (i32 ret, u32 bsd_errno, array<i8, 0x22> message)
- public ResultCode Recv(ServiceCtx context)
- {
- int socketFd = context.RequestData.ReadInt32();
- BsdSocketFlags socketFlags = (BsdSocketFlags)context.RequestData.ReadInt32();
- (ulong receivePosition, ulong receiveLength) = context.Request.GetBufferType0x22();
- WritableRegion receiveRegion = context.Memory.GetWritableRegion(receivePosition, (int)receiveLength);
- LinuxError errno = LinuxError.EBADF;
- ISocket socket = _context.RetrieveSocket(socketFd);
- int result = -1;
- if (socket != null)
- {
- errno = socket.Receive(out result, receiveRegion.Memory.Span, socketFlags);
- if (errno == LinuxError.SUCCESS)
- {
- SetResultErrno(socket, result);
- receiveRegion.Dispose();
- }
- }
- return WriteBsdResult(context, result, errno);
- }
- [CommandHipc(9)]
- // RecvFrom(u32 sock, u32 flags) -> (i32 ret, u32 bsd_errno, u32 addrlen, buffer<i8, 0x22, 0> message, buffer<nn::socket::sockaddr_in, 0x22, 0x10>)
- public ResultCode RecvFrom(ServiceCtx context)
- {
- int socketFd = context.RequestData.ReadInt32();
- BsdSocketFlags socketFlags = (BsdSocketFlags)context.RequestData.ReadInt32();
- (ulong receivePosition, ulong receiveLength) = context.Request.GetBufferType0x22(0);
- (ulong sockAddrOutPosition, ulong sockAddrOutSize) = context.Request.GetBufferType0x22(1);
- WritableRegion receiveRegion = context.Memory.GetWritableRegion(receivePosition, (int)receiveLength);
- LinuxError errno = LinuxError.EBADF;
- ISocket socket = _context.RetrieveSocket(socketFd);
- int result = -1;
- if (socket != null)
- {
- errno = socket.ReceiveFrom(out result, receiveRegion.Memory.Span, receiveRegion.Memory.Span.Length, socketFlags, out IPEndPoint endPoint);
- if (errno == LinuxError.SUCCESS)
- {
- SetResultErrno(socket, result);
- receiveRegion.Dispose();
- if (sockAddrOutSize != 0 && sockAddrOutSize >= (ulong) Unsafe.SizeOf<BsdSockAddr>())
- {
- context.Memory.Write(sockAddrOutPosition, BsdSockAddr.FromIPEndPoint(endPoint));
- }
- else
- {
- errno = LinuxError.ENOMEM;
- }
- }
- }
- return WriteBsdResult(context, result, errno);
- }
- [CommandHipc(10)]
- // Send(u32 socket, u32 flags, buffer<i8, 0x21, 0>) -> (i32 ret, u32 bsd_errno)
- public ResultCode Send(ServiceCtx context)
- {
- int socketFd = context.RequestData.ReadInt32();
- BsdSocketFlags socketFlags = (BsdSocketFlags)context.RequestData.ReadInt32();
- (ulong sendPosition, ulong sendSize) = context.Request.GetBufferType0x21();
- ReadOnlySpan<byte> sendBuffer = context.Memory.GetSpan(sendPosition, (int)sendSize);
- LinuxError errno = LinuxError.EBADF;
- ISocket socket = _context.RetrieveSocket(socketFd);
- int result = -1;
- if (socket != null)
- {
- errno = socket.Send(out result, sendBuffer, socketFlags);
- if (errno == LinuxError.SUCCESS)
- {
- SetResultErrno(socket, result);
- }
- }
- return WriteBsdResult(context, result, errno);
- }
- [CommandHipc(11)]
- // SendTo(u32 socket, u32 flags, buffer<i8, 0x21, 0>, buffer<nn::socket::sockaddr_in, 0x21, 0x10>) -> (i32 ret, u32 bsd_errno)
- public ResultCode SendTo(ServiceCtx context)
- {
- int socketFd = context.RequestData.ReadInt32();
- BsdSocketFlags socketFlags = (BsdSocketFlags)context.RequestData.ReadInt32();
- (ulong sendPosition, ulong sendSize) = context.Request.GetBufferType0x21(0);
- (ulong bufferPosition, ulong bufferSize) = context.Request.GetBufferType0x21(1);
- ReadOnlySpan<byte> sendBuffer = context.Memory.GetSpan(sendPosition, (int)sendSize);
- LinuxError errno = LinuxError.EBADF;
- ISocket socket = _context.RetrieveSocket(socketFd);
- int result = -1;
- if (socket != null)
- {
- IPEndPoint endPoint = context.Memory.Read<BsdSockAddr>(bufferPosition).ToIPEndPoint();
- errno = socket.SendTo(out result, sendBuffer, sendBuffer.Length, socketFlags, endPoint);
- if (errno == LinuxError.SUCCESS)
- {
- SetResultErrno(socket, result);
- }
- }
- return WriteBsdResult(context, result, errno);
- }
- [CommandHipc(12)]
- // Accept(u32 socket) -> (i32 ret, u32 bsd_errno, u32 addrlen, buffer<nn::socket::sockaddr_in, 0x22, 0x10> addr)
- public ResultCode Accept(ServiceCtx context)
- {
- int socketFd = context.RequestData.ReadInt32();
- (ulong bufferPos, ulong bufferSize) = context.Request.GetBufferType0x22();
- LinuxError errno = LinuxError.EBADF;
- ISocket socket = _context.RetrieveSocket(socketFd);
- if (socket != null)
- {
- errno = socket.Accept(out ISocket newSocket);
- if (newSocket == null && errno == LinuxError.SUCCESS)
- {
- errno = LinuxError.EWOULDBLOCK;
- }
- else if (errno == LinuxError.SUCCESS)
- {
- int newSockFd = _context.RegisterFileDescriptor(newSocket);
- if (newSockFd == -1)
- {
- errno = LinuxError.EBADF;
- }
- else
- {
- WriteSockAddr(context, bufferPos, newSocket, true);
- }
- WriteBsdResult(context, newSockFd, errno);
- context.ResponseData.Write(0x10);
- return ResultCode.Success;
- }
- }
- return WriteBsdResult(context, -1, errno);
- }
- [CommandHipc(13)]
- // Bind(u32 socket, buffer<nn::socket::sockaddr_in, 0x21, 0x10> addr) -> (i32 ret, u32 bsd_errno)
- public ResultCode Bind(ServiceCtx context)
- {
- int socketFd = context.RequestData.ReadInt32();
- (ulong bufferPosition, ulong bufferSize) = context.Request.GetBufferType0x21();
- LinuxError errno = LinuxError.EBADF;
- ISocket socket = _context.RetrieveSocket(socketFd);
- if (socket != null)
- {
- IPEndPoint endPoint = context.Memory.Read<BsdSockAddr>(bufferPosition).ToIPEndPoint();
- errno = socket.Bind(endPoint);
- }
- return WriteBsdResult(context, 0, errno);
- }
- [CommandHipc(14)]
- // Connect(u32 socket, buffer<nn::socket::sockaddr_in, 0x21, 0x10>) -> (i32 ret, u32 bsd_errno)
- public ResultCode Connect(ServiceCtx context)
- {
- int socketFd = context.RequestData.ReadInt32();
- (ulong bufferPosition, ulong bufferSize) = context.Request.GetBufferType0x21();
- LinuxError errno = LinuxError.EBADF;
- ISocket socket = _context.RetrieveSocket(socketFd);
- if (socket != null)
- {
- IPEndPoint endPoint = context.Memory.Read<BsdSockAddr>(bufferPosition).ToIPEndPoint();
- errno = socket.Connect(endPoint);
- }
- return WriteBsdResult(context, 0, errno);
- }
- [CommandHipc(15)]
- // GetPeerName(u32 socket) -> (i32 ret, u32 bsd_errno, u32 addrlen, buffer<nn::socket::sockaddr_in, 0x22, 0x10> addr)
- public ResultCode GetPeerName(ServiceCtx context)
- {
- int socketFd = context.RequestData.ReadInt32();
- (ulong bufferPosition, ulong bufferSize) = context.Request.GetBufferType0x22();
- LinuxError errno = LinuxError.EBADF;
- ISocket socket = _context.RetrieveSocket(socketFd);
- if (socket != null)
- {
- errno = LinuxError.ENOTCONN;
- if (socket.RemoteEndPoint != null)
- {
- errno = LinuxError.SUCCESS;
- WriteSockAddr(context, bufferPosition, socket, true);
- WriteBsdResult(context, 0, errno);
- context.ResponseData.Write(Unsafe.SizeOf<BsdSockAddr>());
- }
- }
- return WriteBsdResult(context, 0, errno);
- }
- [CommandHipc(16)]
- // GetSockName(u32 socket) -> (i32 ret, u32 bsd_errno, u32 addrlen, buffer<nn::socket::sockaddr_in, 0x22, 0x10> addr)
- public ResultCode GetSockName(ServiceCtx context)
- {
- int socketFd = context.RequestData.ReadInt32();
- (ulong bufferPos, ulong bufferSize) = context.Request.GetBufferType0x22();
- LinuxError errno = LinuxError.EBADF;
- ISocket socket = _context.RetrieveSocket(socketFd);
- if (socket != null)
- {
- errno = LinuxError.SUCCESS;
- WriteSockAddr(context, bufferPos, socket, false);
- WriteBsdResult(context, 0, errno);
- context.ResponseData.Write(Unsafe.SizeOf<BsdSockAddr>());
- }
- return WriteBsdResult(context, 0, errno);
- }
- [CommandHipc(17)]
- // GetSockOpt(u32 socket, u32 level, u32 option_name) -> (i32 ret, u32 bsd_errno, u32, buffer<unknown, 0x22, 0>)
- public ResultCode GetSockOpt(ServiceCtx context)
- {
- int socketFd = context.RequestData.ReadInt32();
- SocketOptionLevel level = (SocketOptionLevel)context.RequestData.ReadInt32();
- BsdSocketOption option = (BsdSocketOption)context.RequestData.ReadInt32();
- (ulong bufferPosition, ulong bufferSize) = context.Request.GetBufferType0x22();
- WritableRegion optionValue = context.Memory.GetWritableRegion(bufferPosition, (int)bufferSize);
- LinuxError errno = LinuxError.EBADF;
- ISocket socket = _context.RetrieveSocket(socketFd);
- if (socket != null)
- {
- errno = socket.GetSocketOption(option, level, optionValue.Memory.Span);
- if (errno == LinuxError.SUCCESS)
- {
- optionValue.Dispose();
- }
- }
- return WriteBsdResult(context, 0, errno);
- }
- [CommandHipc(18)]
- // Listen(u32 socket, u32 backlog) -> (i32 ret, u32 bsd_errno)
- public ResultCode Listen(ServiceCtx context)
- {
- int socketFd = context.RequestData.ReadInt32();
- int backlog = context.RequestData.ReadInt32();
- LinuxError errno = LinuxError.EBADF;
- ISocket socket = _context.RetrieveSocket(socketFd);
- if (socket != null)
- {
- errno = socket.Listen(backlog);
- }
- return WriteBsdResult(context, 0, errno);
- }
- [CommandHipc(19)]
- // Ioctl(u32 fd, u32 request, u32 bufcount, buffer<unknown, 0x21, 0>, buffer<unknown, 0x21, 0>, buffer<unknown, 0x21, 0>, buffer<unknown, 0x21, 0>) -> (i32 ret, u32 bsd_errno, buffer<unknown, 0x22, 0>, buffer<unknown, 0x22, 0>, buffer<unknown, 0x22, 0>, buffer<unknown, 0x22, 0>)
- public ResultCode Ioctl(ServiceCtx context)
- {
- int socketFd = context.RequestData.ReadInt32();
- BsdIoctl cmd = (BsdIoctl)context.RequestData.ReadInt32();
- int bufferCount = context.RequestData.ReadInt32();
- LinuxError errno = LinuxError.EBADF;
- ISocket socket = _context.RetrieveSocket(socketFd);
- if (socket != null)
- {
- switch (cmd)
- {
- case BsdIoctl.AtMark:
- errno = LinuxError.SUCCESS;
- (ulong bufferPosition, ulong bufferSize) = context.Request.GetBufferType0x22();
- // FIXME: OOB not implemented.
- context.Memory.Write(bufferPosition, 0);
- break;
- default:
- errno = LinuxError.EOPNOTSUPP;
- Logger.Warning?.Print(LogClass.ServiceBsd, $"Unsupported Ioctl Cmd: {cmd}");
- break;
- }
- }
- return WriteBsdResult(context, 0, errno);
- }
- [CommandHipc(20)]
- // Fcntl(u32 socket, u32 cmd, u32 arg) -> (i32 ret, u32 bsd_errno)
- public ResultCode Fcntl(ServiceCtx context)
- {
- int socketFd = context.RequestData.ReadInt32();
- int cmd = context.RequestData.ReadInt32();
- int arg = context.RequestData.ReadInt32();
- int result = 0;
- LinuxError errno = LinuxError.EBADF;
- ISocket socket = _context.RetrieveSocket(socketFd);
- if (socket != null)
- {
- errno = LinuxError.SUCCESS;
- if (cmd == 0x3)
- {
- result = !socket.Blocking ? 0x800 : 0;
- }
- else if (cmd == 0x4 && arg == 0x800)
- {
- socket.Blocking = false;
- result = 0;
- }
- else
- {
- errno = LinuxError.EOPNOTSUPP;
- }
- }
- return WriteBsdResult(context, result, errno);
- }
- [CommandHipc(21)]
- // SetSockOpt(u32 socket, u32 level, u32 option_name, buffer<unknown, 0x21, 0> option_value) -> (i32 ret, u32 bsd_errno)
- public ResultCode SetSockOpt(ServiceCtx context)
- {
- int socketFd = context.RequestData.ReadInt32();
- SocketOptionLevel level = (SocketOptionLevel)context.RequestData.ReadInt32();
- BsdSocketOption option = (BsdSocketOption)context.RequestData.ReadInt32();
- (ulong bufferPos, ulong bufferSize) = context.Request.GetBufferType0x21();
- ReadOnlySpan<byte> optionValue = context.Memory.GetSpan(bufferPos, (int)bufferSize);
- LinuxError errno = LinuxError.EBADF;
- ISocket socket = _context.RetrieveSocket(socketFd);
- if (socket != null)
- {
- errno = socket.SetSocketOption(option, level, optionValue);
- }
- return WriteBsdResult(context, 0, errno);
- }
- [CommandHipc(22)]
- // Shutdown(u32 socket, u32 how) -> (i32 ret, u32 bsd_errno)
- public ResultCode Shutdown(ServiceCtx context)
- {
- int socketFd = context.RequestData.ReadInt32();
- int how = context.RequestData.ReadInt32();
- LinuxError errno = LinuxError.EBADF;
- ISocket socket = _context.RetrieveSocket(socketFd);
- if (socket != null)
- {
- errno = LinuxError.EINVAL;
- if (how >= 0 && how <= 2)
- {
- errno = socket.Shutdown((BsdSocketShutdownFlags)how);
- }
- }
- return WriteBsdResult(context, 0, errno);
- }
- [CommandHipc(23)]
- // ShutdownAllSockets(u32 how) -> (i32 ret, u32 bsd_errno)
- public ResultCode ShutdownAllSockets(ServiceCtx context)
- {
- int how = context.RequestData.ReadInt32();
- LinuxError errno = LinuxError.EINVAL;
- if (how >= 0 && how <= 2)
- {
- errno = _context.ShutdownAllSockets((BsdSocketShutdownFlags)how);
- }
- return WriteBsdResult(context, 0, errno);
- }
- [CommandHipc(24)]
- // Write(u32 fd, buffer<i8, 0x21, 0> message) -> (i32 ret, u32 bsd_errno)
- public ResultCode Write(ServiceCtx context)
- {
- int fd = context.RequestData.ReadInt32();
- (ulong sendPosition, ulong sendSize) = context.Request.GetBufferType0x21();
- ReadOnlySpan<byte> sendBuffer = context.Memory.GetSpan(sendPosition, (int)sendSize);
- LinuxError errno = LinuxError.EBADF;
- IFileDescriptor file = _context.RetrieveFileDescriptor(fd);
- int result = -1;
- if (file != null)
- {
- errno = file.Write(out result, sendBuffer);
- if (errno == LinuxError.SUCCESS)
- {
- SetResultErrno(file, result);
- }
- }
- return WriteBsdResult(context, result, errno);
- }
- [CommandHipc(25)]
- // Read(u32 fd) -> (i32 ret, u32 bsd_errno, buffer<i8, 0x22, 0> message)
- public ResultCode Read(ServiceCtx context)
- {
- int fd = context.RequestData.ReadInt32();
- (ulong receivePosition, ulong receiveLength) = context.Request.GetBufferType0x22();
- WritableRegion receiveRegion = context.Memory.GetWritableRegion(receivePosition, (int)receiveLength);
- LinuxError errno = LinuxError.EBADF;
- IFileDescriptor file = _context.RetrieveFileDescriptor(fd);
- int result = -1;
- if (file != null)
- {
- errno = file.Read(out result, receiveRegion.Memory.Span);
- if (errno == LinuxError.SUCCESS)
- {
- SetResultErrno(file, result);
- receiveRegion.Dispose();
- }
- }
- return WriteBsdResult(context, result, errno);
- }
- [CommandHipc(26)]
- // Close(u32 fd) -> (i32 ret, u32 bsd_errno)
- public ResultCode Close(ServiceCtx context)
- {
- int fd = context.RequestData.ReadInt32();
- LinuxError errno = LinuxError.EBADF;
- if (_context.CloseFileDescriptor(fd))
- {
- errno = LinuxError.SUCCESS;
- }
- return WriteBsdResult(context, 0, errno);
- }
- [CommandHipc(27)]
- // DuplicateSocket(u32 fd, u64 reserved) -> (i32 ret, u32 bsd_errno)
- public ResultCode DuplicateSocket(ServiceCtx context)
- {
- int fd = context.RequestData.ReadInt32();
- ulong reserved = context.RequestData.ReadUInt64();
- LinuxError errno = LinuxError.ENOENT;
- int newSockFd = -1;
- if (_isPrivileged)
- {
- errno = LinuxError.SUCCESS;
- newSockFd = _context.DuplicateFileDescriptor(fd);
- if (newSockFd == -1)
- {
- errno = LinuxError.EBADF;
- }
- }
- return WriteBsdResult(context, newSockFd, errno);
- }
- [CommandHipc(29)] // 7.0.0+
- // RecvMMsg(u32 fd, u32 vlen, u32 flags, u32 reserved, nn::socket::TimeVal timeout) -> (i32 ret, u32 bsd_errno, buffer<bytes, 6> message);
- public ResultCode RecvMMsg(ServiceCtx context)
- {
- int socketFd = context.RequestData.ReadInt32();
- int vlen = context.RequestData.ReadInt32();
- BsdSocketFlags socketFlags = (BsdSocketFlags)context.RequestData.ReadInt32();
- uint reserved = context.RequestData.ReadUInt32();
- TimeVal timeout = context.RequestData.ReadStruct<TimeVal>();
- ulong receivePosition = context.Request.ReceiveBuff[0].Position;
- ulong receiveLength = context.Request.ReceiveBuff[0].Size;
- WritableRegion receiveRegion = context.Memory.GetWritableRegion(receivePosition, (int)receiveLength);
- LinuxError errno = LinuxError.EBADF;
- ISocket socket = _context.RetrieveSocket(socketFd);
- int result = -1;
- if (socket != null)
- {
- errno = BsdMMsgHdr.Deserialize(out BsdMMsgHdr message, receiveRegion.Memory.Span, vlen);
- if (errno == LinuxError.SUCCESS)
- {
- errno = socket.RecvMMsg(out result, message, socketFlags, timeout);
- if (errno == LinuxError.SUCCESS)
- {
- errno = BsdMMsgHdr.Serialize(receiveRegion.Memory.Span, message);
- }
- }
- }
- if (errno == LinuxError.SUCCESS)
- {
- SetResultErrno(socket, result);
- receiveRegion.Dispose();
- }
- return WriteBsdResult(context, result, errno);
- }
- [CommandHipc(30)] // 7.0.0+
- // SendMMsg(u32 fd, u32 vlen, u32 flags) -> (i32 ret, u32 bsd_errno, buffer<bytes, 6> message);
- public ResultCode SendMMsg(ServiceCtx context)
- {
- int socketFd = context.RequestData.ReadInt32();
- int vlen = context.RequestData.ReadInt32();
- BsdSocketFlags socketFlags = (BsdSocketFlags)context.RequestData.ReadInt32();
- ulong receivePosition = context.Request.ReceiveBuff[0].Position;
- ulong receiveLength = context.Request.ReceiveBuff[0].Size;
- WritableRegion receiveRegion = context.Memory.GetWritableRegion(receivePosition, (int)receiveLength);
- LinuxError errno = LinuxError.EBADF;
- ISocket socket = _context.RetrieveSocket(socketFd);
- int result = -1;
- if (socket != null)
- {
- errno = BsdMMsgHdr.Deserialize(out BsdMMsgHdr message, receiveRegion.Memory.Span, vlen);
- if (errno == LinuxError.SUCCESS)
- {
- errno = socket.SendMMsg(out result, message, socketFlags);
- if (errno == LinuxError.SUCCESS)
- {
- errno = BsdMMsgHdr.Serialize(receiveRegion.Memory.Span, message);
- }
- }
- }
- if (errno == LinuxError.SUCCESS)
- {
- SetResultErrno(socket, result);
- receiveRegion.Dispose();
- }
- return WriteBsdResult(context, result, errno);
- }
- [CommandHipc(31)] // 7.0.0+
- // EventFd(nn::socket::EventFdFlags flags, u64 initval) -> (i32 ret, u32 bsd_errno)
- public ResultCode EventFd(ServiceCtx context)
- {
- EventFdFlags flags = (EventFdFlags)context.RequestData.ReadUInt32();
- context.RequestData.BaseStream.Position += 4; // Padding
- ulong initialValue = context.RequestData.ReadUInt64();
- EventFileDescriptor newEventFile = new EventFileDescriptor(initialValue, flags);
- LinuxError errno = LinuxError.SUCCESS;
- int newSockFd = _context.RegisterFileDescriptor(newEventFile);
- if (newSockFd == -1)
- {
- errno = LinuxError.EBADF;
- }
- return WriteBsdResult(context, newSockFd, errno);
- }
- }
- }
|