libctru/libctru/source/services/irrst.c

175 lines
3.5 KiB
C

/*
_irrst.c - C-stick, ZL/ZR
*/
#include <stdlib.h>
#include <string.h>
#include <3ds/types.h>
#include <3ds/result.h>
#include <3ds/svc.h>
#include <3ds/srv.h>
#include <3ds/allocator/mappable.h>
#include <3ds/synchronization.h>
#include <3ds/services/irrst.h>
#include <3ds/ipc.h>
#include <3ds/env.h>
Handle irrstHandle;
Handle irrstMemHandle;
Handle irrstEvent;
vu32* irrstSharedMem;
static u32 kHeld;
static circlePosition csPos;
static int irrstRefCount;
Result irrstInit(void)
{
if (AtomicPostIncrement(&irrstRefCount)) return 0;
Result ret=0;
// Request service.
if(R_FAILED(ret=srvGetServiceHandle(&irrstHandle, "ir:rst"))) goto cleanup0;
// Get sharedmem handle.
if(R_FAILED(ret=IRRST_GetHandles(&irrstMemHandle, &irrstEvent))) goto cleanup1;
// Initialize ir:rst
if(envGetHandle("ir:rst") == 0) ret = IRRST_Initialize(10, 0);
// Map ir:rst shared memory.
irrstSharedMem=(vu32*)mappableAlloc(0x98);
if(!irrstSharedMem)
{
ret = -1;
goto cleanup1;
}
if(R_FAILED(ret = svcMapMemoryBlock(irrstMemHandle, (u32)irrstSharedMem, MEMPERM_READ, MEMPERM_DONTCARE))) goto cleanup2;
return 0;
cleanup2:
svcCloseHandle(irrstMemHandle);
if(irrstSharedMem != NULL)
{
mappableFree((void*) irrstSharedMem);
irrstSharedMem = NULL;
}
cleanup1:
svcCloseHandle(irrstHandle);
cleanup0:
AtomicDecrement(&irrstRefCount);
return ret;
}
void irrstExit(void)
{
if (AtomicDecrement(&irrstRefCount)) return;
// Reset internal state.
kHeld = 0;
svcCloseHandle(irrstEvent);
// Unmap ir:rst sharedmem and close handles.
svcUnmapMemoryBlock(irrstMemHandle, (u32)irrstSharedMem);
if(envGetHandle("ir:rst") == 0) IRRST_Shutdown();
svcCloseHandle(irrstMemHandle);
svcCloseHandle(irrstHandle);
if(irrstSharedMem != NULL)
{
mappableFree((void*) irrstSharedMem);
irrstSharedMem = NULL;
}
}
void irrstWaitForEvent(bool nextEvent)
{
if(nextEvent)svcClearEvent(irrstEvent);
svcWaitSynchronization(irrstEvent, U64_MAX);
if(!nextEvent)svcClearEvent(irrstEvent);
}
u32 irrstCheckSectionUpdateTime(vu32 *sharedmem_section, u32 id)
{
s64 tick0=0, tick1=0;
if(id==0)
{
tick0 = *((u64*)&sharedmem_section[0]);
tick1 = *((u64*)&sharedmem_section[2]);
if(tick0==tick1 || tick0<0 || tick1<0)return 1;
}
return 0;
}
void irrstScanInput(void)
{
if(irrstRefCount==0)return;
u32 Id=0;
kHeld = 0;
memset(&csPos, 0, sizeof(circlePosition));
Id = irrstSharedMem[4]; //PAD / circle-pad
if(Id>7)Id=7;
if(irrstCheckSectionUpdateTime(irrstSharedMem, Id)==0)
{
kHeld = irrstSharedMem[6 + Id*4];
csPos = *(circlePosition*)&irrstSharedMem[6 + Id*4 + 3];
}
}
u32 irrstKeysHeld(void)
{
if(irrstRefCount>0)return kHeld;
return 0;
}
void irrstCstickRead(circlePosition* pos)
{
if (pos) *pos = csPos;
}
Result IRRST_GetHandles(Handle* outMemHandle, Handle* outEventHandle)
{
u32* cmdbuf=getThreadCommandBuffer();
cmdbuf[0]=IPC_MakeHeader(0x1,0,0); // 0x10000
Result ret=0;
if(R_FAILED(ret=svcSendSyncRequest(irrstHandle)))return ret;
if(outMemHandle)*outMemHandle=cmdbuf[3];
if(outEventHandle)*outEventHandle=cmdbuf[4];
return cmdbuf[1];
}
Result IRRST_Initialize(u32 unk1, u8 unk2)
{
u32* cmdbuf=getThreadCommandBuffer();
cmdbuf[0]=IPC_MakeHeader(0x2,2,0); // 0x20080
cmdbuf[1]=unk1;
cmdbuf[2]=unk2;
Result ret=0;
if(R_FAILED(ret=svcSendSyncRequest(irrstHandle)))return ret;
return cmdbuf[1];
}
Result IRRST_Shutdown(void)
{
u32* cmdbuf=getThreadCommandBuffer();
cmdbuf[0]=IPC_MakeHeader(0x3,0,0); // 0x30000
Result ret=0;
if(R_FAILED(ret=svcSendSyncRequest(irrstHandle)))return ret;
return cmdbuf[1];
}