mirror of
https://github.com/Ryujinx/Ryujinx.git
synced 2024-10-01 12:30:00 +02:00
da7e702751
* Update BSD service to handle libnx's 'smart IPC buffers' for address info * Use existing "GetBufferType0x21" for certain BSD socket methods * Parse address port as unsigned short * Fix bounds check on reading the IPC buffer * Implement Read, Write methods * rebased and cleaned * addressed nits * remove unused swap method * fixed alignments
215 lines
6.4 KiB
C#
215 lines
6.4 KiB
C#
using System.Collections.Generic;
|
|
using System.IO;
|
|
|
|
namespace Ryujinx.HLE.HOS.Ipc
|
|
{
|
|
class IpcMessage
|
|
{
|
|
public IpcMessageType Type { get; set; }
|
|
|
|
public IpcHandleDesc HandleDesc { get; set; }
|
|
|
|
public List<IpcPtrBuffDesc> PtrBuff { get; private set; }
|
|
public List<IpcBuffDesc> SendBuff { get; private set; }
|
|
public List<IpcBuffDesc> ReceiveBuff { get; private set; }
|
|
public List<IpcBuffDesc> ExchangeBuff { get; private set; }
|
|
public List<IpcRecvListBuffDesc> RecvListBuff { get; private set; }
|
|
|
|
public List<int> ObjectIds { get; private set; }
|
|
|
|
public byte[] RawData { get; set; }
|
|
|
|
public IpcMessage()
|
|
{
|
|
PtrBuff = new List<IpcPtrBuffDesc>();
|
|
SendBuff = new List<IpcBuffDesc>();
|
|
ReceiveBuff = new List<IpcBuffDesc>();
|
|
ExchangeBuff = new List<IpcBuffDesc>();
|
|
RecvListBuff = new List<IpcRecvListBuffDesc>();
|
|
|
|
ObjectIds = new List<int>();
|
|
}
|
|
|
|
public IpcMessage(byte[] Data, long CmdPtr) : this()
|
|
{
|
|
using (MemoryStream MS = new MemoryStream(Data))
|
|
{
|
|
BinaryReader Reader = new BinaryReader(MS);
|
|
|
|
Initialize(Reader, CmdPtr);
|
|
}
|
|
}
|
|
|
|
private void Initialize(BinaryReader Reader, long CmdPtr)
|
|
{
|
|
int Word0 = Reader.ReadInt32();
|
|
int Word1 = Reader.ReadInt32();
|
|
|
|
Type = (IpcMessageType)(Word0 & 0xffff);
|
|
|
|
int PtrBuffCount = (Word0 >> 16) & 0xf;
|
|
int SendBuffCount = (Word0 >> 20) & 0xf;
|
|
int RecvBuffCount = (Word0 >> 24) & 0xf;
|
|
int XchgBuffCount = (Word0 >> 28) & 0xf;
|
|
|
|
int RawDataSize = (Word1 >> 0) & 0x3ff;
|
|
int RecvListFlags = (Word1 >> 10) & 0xf;
|
|
bool HndDescEnable = ((Word1 >> 31) & 0x1) != 0;
|
|
|
|
if (HndDescEnable)
|
|
{
|
|
HandleDesc = new IpcHandleDesc(Reader);
|
|
}
|
|
|
|
for (int Index = 0; Index < PtrBuffCount; Index++)
|
|
{
|
|
PtrBuff.Add(new IpcPtrBuffDesc(Reader));
|
|
}
|
|
|
|
void ReadBuff(List<IpcBuffDesc> Buff, int Count)
|
|
{
|
|
for (int Index = 0; Index < Count; Index++)
|
|
{
|
|
Buff.Add(new IpcBuffDesc(Reader));
|
|
}
|
|
}
|
|
|
|
ReadBuff(SendBuff, SendBuffCount);
|
|
ReadBuff(ReceiveBuff, RecvBuffCount);
|
|
ReadBuff(ExchangeBuff, XchgBuffCount);
|
|
|
|
RawDataSize *= 4;
|
|
|
|
long RecvListPos = Reader.BaseStream.Position + RawDataSize;
|
|
|
|
long Pad0 = GetPadSize16(Reader.BaseStream.Position + CmdPtr);
|
|
|
|
Reader.BaseStream.Seek(Pad0, SeekOrigin.Current);
|
|
|
|
int RecvListCount = RecvListFlags - 2;
|
|
|
|
if (RecvListCount == 0)
|
|
{
|
|
RecvListCount = 1;
|
|
}
|
|
else if (RecvListCount < 0)
|
|
{
|
|
RecvListCount = 0;
|
|
}
|
|
|
|
RawData = Reader.ReadBytes(RawDataSize);
|
|
|
|
Reader.BaseStream.Seek(RecvListPos, SeekOrigin.Begin);
|
|
|
|
for (int Index = 0; Index < RecvListCount; Index++)
|
|
{
|
|
RecvListBuff.Add(new IpcRecvListBuffDesc(Reader));
|
|
}
|
|
}
|
|
|
|
public byte[] GetBytes(long CmdPtr)
|
|
{
|
|
using (MemoryStream MS = new MemoryStream())
|
|
{
|
|
BinaryWriter Writer = new BinaryWriter(MS);
|
|
|
|
int Word0;
|
|
int Word1;
|
|
|
|
Word0 = (int)Type;
|
|
Word0 |= (PtrBuff.Count & 0xf) << 16;
|
|
Word0 |= (SendBuff.Count & 0xf) << 20;
|
|
Word0 |= (ReceiveBuff.Count & 0xf) << 24;
|
|
Word0 |= (ExchangeBuff.Count & 0xf) << 28;
|
|
|
|
byte[] HandleData = new byte[0];
|
|
|
|
if (HandleDesc != null)
|
|
{
|
|
HandleData = HandleDesc.GetBytes();
|
|
}
|
|
|
|
int DataLength = RawData?.Length ?? 0;
|
|
|
|
int Pad0 = (int)GetPadSize16(CmdPtr + 8 + HandleData.Length);
|
|
|
|
//Apparently, padding after Raw Data is 16 bytes, however when there is
|
|
//padding before Raw Data too, we need to subtract the size of this padding.
|
|
//This is the weirdest padding I've seen so far...
|
|
int Pad1 = 0x10 - Pad0;
|
|
|
|
DataLength = (DataLength + Pad0 + Pad1) / 4;
|
|
|
|
Word1 = DataLength & 0x3ff;
|
|
|
|
if (HandleDesc != null)
|
|
{
|
|
Word1 |= 1 << 31;
|
|
}
|
|
|
|
Writer.Write(Word0);
|
|
Writer.Write(Word1);
|
|
Writer.Write(HandleData);
|
|
|
|
MS.Seek(Pad0, SeekOrigin.Current);
|
|
|
|
if (RawData != null)
|
|
{
|
|
Writer.Write(RawData);
|
|
}
|
|
|
|
Writer.Write(new byte[Pad1]);
|
|
|
|
return MS.ToArray();
|
|
}
|
|
}
|
|
|
|
private long GetPadSize16(long Position)
|
|
{
|
|
if ((Position & 0xf) != 0)
|
|
{
|
|
return 0x10 - (Position & 0xf);
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
public (long Position, long Size) GetBufferType0x21(int Index = 0)
|
|
{
|
|
if (PtrBuff.Count > Index &&
|
|
PtrBuff[Index].Position != 0 &&
|
|
PtrBuff[Index].Size != 0)
|
|
{
|
|
return (PtrBuff[Index].Position, PtrBuff[Index].Size);
|
|
}
|
|
|
|
if (SendBuff.Count > Index &&
|
|
SendBuff[Index].Position != 0 &&
|
|
SendBuff[Index].Size != 0)
|
|
{
|
|
return (SendBuff[Index].Position, SendBuff[Index].Size);
|
|
}
|
|
|
|
return (0, 0);
|
|
}
|
|
|
|
public (long Position, long Size) GetBufferType0x22(int Index = 0)
|
|
{
|
|
if (RecvListBuff.Count > Index &&
|
|
RecvListBuff[Index].Position != 0 &&
|
|
RecvListBuff[Index].Size != 0)
|
|
{
|
|
return (RecvListBuff[Index].Position, RecvListBuff[Index].Size);
|
|
}
|
|
|
|
if (ReceiveBuff.Count > Index &&
|
|
ReceiveBuff[Index].Position != 0 &&
|
|
ReceiveBuff[Index].Size != 0)
|
|
{
|
|
return (ReceiveBuff[Index].Position, ReceiveBuff[Index].Size);
|
|
}
|
|
|
|
return (0, 0);
|
|
}
|
|
}
|
|
}
|