integrate local MP comm within the IPC module

This commit is contained in:
Arisotura 2023-03-11 21:16:37 +01:00
parent d296ce3f5e
commit 6fed0f8d75
4 changed files with 571 additions and 42 deletions

View File

@ -19,6 +19,18 @@
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#ifdef __WIN32__
#include <windows.h>
#else
#include <fcntl.h>
#include <semaphore.h>
#include <time.h>
#ifdef __APPLE__
#include "sem_timedwait.h"
#endif
#endif
#include <QSharedMemory>
#include "IPC.h"
@ -40,9 +52,14 @@ int InstanceID;
struct BufferHeader
{
u16 NumInstances;
u16 NumInstances; // total number of instances present
u16 InstanceBitmask; // bitmask of all instances present
u16 ConnectedBitmask; // bitmask of which instances are ready to send/receive MP packets
u32 CommandWriteOffset;
u32 MPPacketWriteOffset;
u32 MPReplyWriteOffset;
u16 MPHostInstanceID; // instance ID from which the last CMD frame was sent
u16 MPReplyBitmask; // bitmask of which clients replied in time
};
struct CommandHeader
@ -54,17 +71,179 @@ struct CommandHeader
u16 Length;
};
u32 CommandReadOffset;
struct MPPacketHeader
{
u32 Magic;
u32 SenderID;
u32 Type; // 0=regular 1=CMD 2=reply 3=ack
u32 Length;
u64 Timestamp;
};
const u32 kBufferSize = 0x4000;
u32 CommandReadOffset;
u32 MPPacketReadOffset;
u32 MPReplyReadOffset;
const u32 kBufferSize = 0x30000;
const u32 kMaxCommandSize = 0x800;
const u32 kMaxFrameSize = 0x800;
const u32 kCommandStart = sizeof(BufferHeader);
const u32 kCommandEnd = kBufferSize;
const u32 kCommandEnd = (kBufferSize / 3);
const u32 kMPPacketStart = kCommandEnd;
const u32 kMPPacketEnd = (2 * (kBufferSize / 3));
const u32 kMPReplyStart = kMPPacketEnd;
const u32 kMPReplyEnd = kBufferSize;
bool CmdRecvFlags[Cmd_MAX];
int MPRecvTimeout;
int MPLastHostID;
void Init()
// we need to come up with our own abstraction layer for named semaphores
// because QSystemSemaphore doesn't support waiting with a timeout
// and, as such, is unsuitable to our needs
#ifdef __WIN32__
bool SemInited[32];
HANDLE SemPool[32];
void SemPoolInit()
{
for (int i = 0; i < 32; i++)
{
SemPool[i] = INVALID_HANDLE_VALUE;
SemInited[i] = false;
}
}
void SemDeinit(int num);
void SemPoolDeinit()
{
for (int i = 0; i < 32; i++)
SemDeinit(i);
}
bool SemInit(int num)
{
if (SemInited[num])
return true;
char semname[64];
sprintf(semname, "Local\\melonNIFI_Sem%02d", num);
HANDLE sem = CreateSemaphore(nullptr, 0, 64, semname);
SemPool[num] = sem;
SemInited[num] = true;
return sem != INVALID_HANDLE_VALUE;
}
void SemDeinit(int num)
{
if (SemPool[num] != INVALID_HANDLE_VALUE)
{
CloseHandle(SemPool[num]);
SemPool[num] = INVALID_HANDLE_VALUE;
}
SemInited[num] = false;
}
bool SemPost(int num)
{
SemInit(num);
return ReleaseSemaphore(SemPool[num], 1, nullptr) != 0;
}
bool SemWait(int num, int timeout)
{
return WaitForSingleObject(SemPool[num], timeout) == WAIT_OBJECT_0;
}
void SemReset(int num)
{
while (WaitForSingleObject(SemPool[num], 0) == WAIT_OBJECT_0);
}
#else
bool SemInited[32];
sem_t* SemPool[32];
void SemPoolInit()
{
for (int i = 0; i < 32; i++)
{
SemPool[i] = SEM_FAILED;
SemInited[i] = false;
}
}
void SemDeinit(int num);
void SemPoolDeinit()
{
for (int i = 0; i < 32; i++)
SemDeinit(i);
}
bool SemInit(int num)
{
if (SemInited[num])
return true;
char semname[64];
sprintf(semname, "/melonNIFI_Sem%02d", num);
sem_t* sem = sem_open(semname, O_CREAT, 0644, 0);
SemPool[num] = sem;
SemInited[num] = true;
return sem != SEM_FAILED;
}
void SemDeinit(int num)
{
if (SemPool[num] != SEM_FAILED)
{
sem_close(SemPool[num]);
SemPool[num] = SEM_FAILED;
}
SemInited[num] = false;
}
bool SemPost(int num)
{
SemInit(num);
return sem_post(SemPool[num]) == 0;
}
bool SemWait(int num, int timeout)
{
if (!timeout)
return sem_trywait(SemPool[num]) == 0;
struct timespec ts;
clock_gettime(CLOCK_REALTIME, &ts);
ts.tv_nsec += timeout * 1000000;
long sec = ts.tv_nsec / 1000000000;
ts.tv_nsec -= sec * 1000000000;
ts.tv_sec += sec;
return sem_timedwait(SemPool[num], &ts) == 0;
}
void SemReset(int num)
{
while (sem_trywait(SemPool[num]) == 0);
}
#endif
bool Init()
{
InstanceID = 0;
@ -78,7 +257,7 @@ void Init()
printf("IPC sharedmem create failed :(\n");
delete Buffer;
Buffer = nullptr;
return;
return false;
}
Buffer->lock();
@ -92,11 +271,15 @@ void Init()
if (header->NumInstances == 0)
{
// initialize the FIFO
// initialize the FIFOs
header->CommandWriteOffset = kCommandStart;
header->MPPacketWriteOffset = kMPPacketStart;
header->MPReplyWriteOffset = kMPReplyStart;
}
CommandReadOffset = header->CommandWriteOffset;
MPPacketReadOffset = header->MPPacketWriteOffset;
MPReplyReadOffset = header->MPReplyWriteOffset;
u16 mask = header->InstanceBitmask;
for (int i = 0; i < 16; i++)
@ -113,7 +296,22 @@ void Init()
memset(CmdRecvFlags, 0, sizeof(CmdRecvFlags));
printf("IPC: instance ID %d\n", InstanceID);
MPLastHostID = -1;
MPRecvTimeout = 25;
printf("IPC: init OK, instance ID %d\n", InstanceID);
return true;
}
bool InitSema()
{
// prepare semaphores
// semaphores 0-15: regular frames; semaphore I is posted when instance I needs to process a new frame
// semaphores 16-31: MP replies; semaphore I is posted when instance I needs to process a new MP reply
SemPoolInit();
if (!SemInit(InstanceID)) return false;
if (!SemInit(16+InstanceID)) return false;
}
void DeInit()
@ -123,6 +321,7 @@ void DeInit()
Buffer->lock();
u8* data = (u8*)Buffer->data();
BufferHeader* header = (BufferHeader*)&data[0];
header->ConnectedBitmask &= ~(1 << InstanceID);
header->InstanceBitmask &= ~(1<<InstanceID);
header->NumInstances--;
Buffer->unlock();
@ -133,16 +332,63 @@ void DeInit()
Buffer = nullptr;
}
void DeInitSema()
{
SemPoolDeinit();
}
void FIFORead(void* buf, int len)
void SetMPRecvTimeout(int timeout)
{
MPRecvTimeout = timeout;
}
void MPBegin()
{
Buffer->lock();
BufferHeader* header = (BufferHeader*)Buffer->data();
MPPacketReadOffset = header->MPPacketWriteOffset;
MPReplyReadOffset = header->MPReplyWriteOffset;
SemReset(InstanceID);
SemReset(16+InstanceID);
header->ConnectedBitmask |= (1 << InstanceID);
Buffer->unlock();
}
void MPEnd()
{
Buffer->lock();
BufferHeader* header = (BufferHeader*)Buffer->data();
//SemReset(InstanceID);
//SemReset(16+InstanceID);
header->ConnectedBitmask &= ~(1 << InstanceID);
Buffer->unlock();
}
template<int fifo> void FIFORead(void* buf, int len)
{
u8* data = (u8*)Buffer->data();
u32 offset, start, end;
if (fifo == 0)
{
offset = CommandReadOffset;
start = kCommandStart;
end = kCommandEnd;
}
else if (fifo == 1)
{
offset = MPPacketReadOffset;
start = kMPPacketStart;
end = kMPPacketEnd;
}
else if (fifo == 2)
{
offset = MPReplyReadOffset;
start = kMPReplyStart;
end = kMPReplyEnd;
}
if ((offset + len) >= end)
{
@ -157,19 +403,35 @@ void FIFORead(void* buf, int len)
offset += len;
}
CommandReadOffset = offset;
if (fifo == 0) CommandReadOffset = offset;
else if (fifo == 1) MPPacketReadOffset = offset;
else if (fifo == 2) MPReplyReadOffset = offset;
}
void FIFOWrite(void* buf, int len)
template<int fifo> void FIFOWrite(void* buf, int len)
{
u8* data = (u8*)Buffer->data();
BufferHeader* header = (BufferHeader*)&data[0];
u32 offset, start, end;
if (fifo == 0)
{
offset = header->CommandWriteOffset;
start = kCommandStart;
end = kCommandEnd;
}
else if (fifo == 1)
{
offset = header->MPPacketWriteOffset;
start = kMPPacketStart;
end = kMPPacketEnd;
}
else if (fifo == 2)
{
offset = header->MPReplyWriteOffset;
start = kMPReplyStart;
end = kMPReplyEnd;
}
if ((offset + len) >= end)
{
@ -184,10 +446,13 @@ void FIFOWrite(void* buf, int len)
offset += len;
}
header->CommandWriteOffset = offset;
if (fifo == 0) header->CommandWriteOffset = offset;
else if (fifo == 1) header->MPPacketWriteOffset = offset;
else if (fifo == 2) header->MPReplyWriteOffset = offset;
}
void Process()
void ProcessCommands()
{
memset(CmdRecvFlags, 0, sizeof(CmdRecvFlags));
@ -201,7 +466,7 @@ void Process()
CommandHeader cmdheader;
u8 cmddata[kMaxCommandSize];
FIFORead(&cmdheader, sizeof(cmdheader));
FIFORead<0>(&cmdheader, sizeof(cmdheader));
if ((cmdheader.Magic != 0x4D434C4D) || (cmdheader.Length > kMaxCommandSize))
{
@ -212,7 +477,7 @@ void Process()
}
if (cmdheader.Length)
FIFORead(cmddata, cmdheader.Length);
FIFORead<0>(cmddata, cmdheader.Length);
if (!(cmdheader.Recipients & (1<<InstanceID)))
continue;
@ -229,6 +494,7 @@ void Process()
}*/
CmdRecvFlags[cmdheader.Command] = true;
// TODO: store the command data, for future commands that will need it
// TODO: also what if, say, we get multiple pause commands before CommandReceived() is called?
}
Buffer->unlock();
@ -270,9 +536,9 @@ bool SendCommand(u16 recipients, u16 command, u16 len, void* cmddata)
cmdheader.Recipients = recipients;
cmdheader.Command = command;
cmdheader.Length = len;
FIFOWrite(&cmdheader, sizeof(cmdheader));
FIFOWrite<0>(&cmdheader, sizeof(cmdheader));
if (len)
FIFOWrite(cmddata, len);
FIFOWrite<0>(cmddata, len);
Buffer->unlock();
return true;
@ -284,4 +550,235 @@ bool CommandReceived(u16 command)
return CmdRecvFlags[command];
}
int SendMPPacketGeneric(u32 type, u8* packet, int len, u64 timestamp)
{
Buffer->lock();
u8* data = (u8*)Buffer->data();
BufferHeader* header = (BufferHeader*)&data[0];
u16 mask = header->ConnectedBitmask;
// TODO: check if the FIFO is full!
MPPacketHeader pktheader;
pktheader.Magic = 0x4946494E;
pktheader.SenderID = InstanceID;
pktheader.Type = type;
pktheader.Length = len;
pktheader.Timestamp = timestamp;
type &= 0xFFFF;
if (type != 2)
{
FIFOWrite<1>(&pktheader, sizeof(pktheader));
if (len) FIFOWrite<1>(packet, len);
}
if (type == 1)
{
// NOTE: this is not guarded against, say, multiple multiplay games happening on the same machine
// we would need to pass the packet's SenderID through the wifi module for that
header->MPHostInstanceID = InstanceID;
header->MPReplyBitmask = 0;
MPReplyReadOffset = header->MPReplyWriteOffset;
SemReset(16 + InstanceID);
}
else if (type == 2)
{
FIFOWrite<2>(&pktheader, sizeof(pktheader));
if (len) FIFOWrite<2>(packet, len);
header->MPReplyBitmask |= (1 << InstanceID);
}
Buffer->unlock();
if (type == 2)
{
SemPost(16 + header->MPHostInstanceID);
}
else
{
for (int i = 0; i < 16; i++)
{
if (mask & (1<<i))
SemPost(i);
}
}
return len;
}
int RecvMPPacketGeneric(u8* packet, bool block, u64* timestamp)
{
for (;;)
{
if (!SemWait(InstanceID, block ? MPRecvTimeout : 0))
{
return 0;
}
Buffer->lock();
u8* data = (u8*)Buffer->data();
BufferHeader* header = (BufferHeader*)&data[0];
MPPacketHeader pktheader;
FIFORead<1>(&pktheader, sizeof(pktheader));
if (pktheader.Magic != 0x4946494E)
{
printf("PACKET FIFO OVERFLOW\n");
MPPacketReadOffset = header->MPPacketWriteOffset;
SemReset(InstanceID);
Buffer->unlock();
return 0;
}
if (pktheader.SenderID == InstanceID)
{
// skip this packet
MPPacketReadOffset += pktheader.Length;
if (MPPacketReadOffset >= kMPPacketEnd)
MPPacketReadOffset += kMPPacketStart - kMPPacketEnd;
Buffer->unlock();
continue;
}
if (pktheader.Length)
{
FIFORead<1>(packet, pktheader.Length);
if (pktheader.Type == 1)
MPLastHostID = pktheader.SenderID;
}
if (timestamp) *timestamp = pktheader.Timestamp;
Buffer->unlock();
return pktheader.Length;
}
}
int SendMPPacket(u8* packet, int len, u64 timestamp)
{
return SendMPPacketGeneric(0, packet, len, timestamp);
}
int RecvMPPacket(u8* packet, u64* timestamp)
{
return RecvMPPacketGeneric(packet, false, timestamp);
}
int SendMPCmd(u8* packet, int len, u64 timestamp)
{
return SendMPPacketGeneric(1, packet, len, timestamp);
}
int SendMPReply(u8* packet, int len, u64 timestamp, u16 aid)
{
return SendMPPacketGeneric(2 | (aid<<16), packet, len, timestamp);
}
int SendMPAck(u8* packet, int len, u64 timestamp)
{
return SendMPPacketGeneric(3, packet, len, timestamp);
}
int RecvMPHostPacket(u8* packet, u64* timestamp)
{
if (MPLastHostID != -1)
{
// check if the host is still connected
Buffer->lock();
u8* data = (u8*)Buffer->data();
BufferHeader* header = (BufferHeader*)&data[0];
u16 curinstmask = header->ConnectedBitmask;
Buffer->unlock();
if (!(curinstmask & (1 << MPLastHostID)))
return -1;
}
return RecvMPPacketGeneric(packet, true, timestamp);
}
u16 RecvMPReplies(u8* packets, u64 timestamp, u16 aidmask)
{
u16 ret = 0;
u16 myinstmask = (1 << InstanceID);
u16 curinstmask;
{
Buffer->lock();
u8* data = (u8*)Buffer->data();
BufferHeader* header = (BufferHeader*)&data[0];
curinstmask = header->ConnectedBitmask;
Buffer->unlock();
}
// if all clients have left: return early
if ((myinstmask & curinstmask) == curinstmask)
return 0;
for (;;)
{
if (!SemWait(16+InstanceID, MPRecvTimeout))
{
// no more replies available
return ret;
}
Buffer->lock();
u8* data = (u8*)Buffer->data();
BufferHeader* header = (BufferHeader*)&data[0];
MPPacketHeader pktheader;
FIFORead<2>(&pktheader, sizeof(pktheader));
if (pktheader.Magic != 0x4946494E)
{
printf("REPLY FIFO OVERFLOW\n");
MPReplyReadOffset = header->MPReplyWriteOffset;
SemReset(16+InstanceID);
Buffer->unlock();
return 0;
}
if ((pktheader.SenderID == InstanceID) || // packet we sent out (shouldn't happen, but hey)
(pktheader.Timestamp < (timestamp - 32))) // stale packet
{
// skip this packet
MPReplyReadOffset += pktheader.Length;
if (MPReplyReadOffset >= kMPReplyEnd)
MPReplyReadOffset += kMPReplyStart - kMPReplyEnd;
Buffer->unlock();
continue;
}
if (pktheader.Length)
{
u32 aid = (pktheader.Type >> 16);
FIFORead<2>(&packets[(aid-1)*1024], pktheader.Length);
ret |= (1 << aid);
}
myinstmask |= (1 << pktheader.SenderID);
if (((myinstmask & curinstmask) == curinstmask) ||
((ret & aidmask) == aidmask))
{
// all the clients have sent their reply
Buffer->unlock();
return ret;
}
Buffer->unlock();
}
}
}

View File

@ -33,14 +33,27 @@ enum
extern int InstanceID;
void Init();
bool Init();
bool InitSema();
void DeInit();
void DeInitSema();
void Process();
void SetMPRecvTimeout(int timeout);
void MPBegin();
void MPEnd();
void ProcessCommands();
bool SendCommand(u16 recipients, u16 command, u16 len, void* data);
bool CommandReceived(u16 command);
int SendMPPacket(u8* data, int len, u64 timestamp);
int RecvMPPacket(u8* data, u64* timestamp);
int SendMPCmd(u8* data, int len, u64 timestamp);
int SendMPReply(u8* data, int len, u64 timestamp, u16 aid);
int SendMPAck(u8* data, int len, u64 timestamp);
int RecvMPHostPacket(u8* data, u64* timestamp);
u16 RecvMPReplies(u8* data, u64 timestamp, u16 aidmask);
}
#endif // IPC_H

View File

@ -90,12 +90,13 @@ void Init(int argc, char** argv)
EmuDirectory = confdir.toStdString();
#endif
IPC::Init();
//IPC::Init();
//IPC::SetMPRecvTimeout(Config::MPRecvTimeout);
}
void DeInit()
{
IPC::DeInit();
//IPC::DeInit();
}
void SignalStop(StopReason reason)
@ -531,57 +532,67 @@ void WriteGBASave(const u8* savedata, u32 savelen, u32 writeoffset, u32 writelen
bool MP_Init()
{
return LocalMP::Init();
//return LocalMP::Init();
return true;
}
void MP_DeInit()
{
return LocalMP::DeInit();
//return LocalMP::DeInit();
}
void MP_Begin()
{
return LocalMP::Begin();
//return LocalMP::Begin();
return IPC::MPBegin();
}
void MP_End()
{
return LocalMP::End();
//return LocalMP::End();
return IPC::MPEnd();
}
int MP_SendPacket(u8* data, int len, u64 timestamp)
{
return LocalMP::SendPacket(data, len, timestamp);
//return LocalMP::SendPacket(data, len, timestamp);
return IPC::SendMPPacket(data, len, timestamp);
}
int MP_RecvPacket(u8* data, u64* timestamp)
{
return LocalMP::RecvPacket(data, timestamp);
//return LocalMP::RecvPacket(data, timestamp);
return IPC::RecvMPPacket(data, timestamp);
}
int MP_SendCmd(u8* data, int len, u64 timestamp)
{
return LocalMP::SendCmd(data, len, timestamp);
//return LocalMP::SendCmd(data, len, timestamp);
return IPC::SendMPCmd(data, len, timestamp);
}
int MP_SendReply(u8* data, int len, u64 timestamp, u16 aid)
{
return LocalMP::SendReply(data, len, timestamp, aid);
//return LocalMP::SendReply(data, len, timestamp, aid);
return IPC::SendMPReply(data, len, timestamp, aid);
}
int MP_SendAck(u8* data, int len, u64 timestamp)
{
return LocalMP::SendAck(data, len, timestamp);
//return LocalMP::SendAck(data, len, timestamp);
return IPC::SendMPAck(data, len, timestamp);
}
int MP_RecvHostPacket(u8* data, u64* timestamp)
{
return LocalMP::RecvHostPacket(data, timestamp);
//return LocalMP::RecvHostPacket(data, timestamp);
return IPC::RecvMPHostPacket(data, timestamp);
}
u16 MP_RecvReplies(u8* data, u64 timestamp, u16 aidmask)
{
return LocalMP::RecvReplies(data, timestamp, aidmask);
//return LocalMP::RecvReplies(data, timestamp, aidmask);
return IPC::RecvMPReplies(data, timestamp, aidmask);
}
bool LAN_Init()

View File

@ -317,6 +317,9 @@ void EmuThread::run()
{
u32 mainScreenPos[3];
IPC::InitSema();
IPC::SetMPRecvTimeout(Config::MPRecvTimeout);
NDS::Init();
mainScreenPos[0] = 0;
@ -359,7 +362,7 @@ void EmuThread::run()
while (EmuRunning != emuStatus_Exit)
{
IPC::Process();
IPC::ProcessCommands();
if (IPC::CommandReceived(IPC::Cmd_Pause)) emit windowIPCPause();
@ -658,6 +661,7 @@ void EmuThread::run()
GPU::DeInitRenderer();
NDS::DeInit();
IPC::DeInitSema();
//Platform::LAN_DeInit();
}
@ -2965,7 +2969,8 @@ void MainWindow::onOpenMPSettings()
void MainWindow::onMPSettingsFinished(int res)
{
AudioInOut::AudioMute(mainWindow);
LocalMP::SetRecvTimeout(Config::MPRecvTimeout);
//LocalMP::SetRecvTimeout(Config::MPRecvTimeout);
IPC::SetMPRecvTimeout(Config::MPRecvTimeout);
emuThread->emuUnpause();
}
@ -3272,6 +3277,8 @@ int main(int argc, char** argv)
MelonApplication melon(argc, argv);
IPC::Init();
CLI::CommandLineOptions* options = CLI::ManageArgs(melon);
// http://stackoverflow.com/questions/14543333/joystick-wont-work-using-sdl
@ -3386,6 +3393,7 @@ int main(int argc, char** argv)
SDL_Quit();
Platform::DeInit();
IPC::DeInit();
return ret;
}