Added code for using dlp::clnt

Author: Pixel-Pop

libctru/include/3ds.h

@@ -79,6 +79,7 @@ extern "C" {
 #include <3ds/services/loader.h>
 #include <3ds/services/y2r.h>
 #include <3ds/services/mcuhwc.h>
+#include <3ds/services/dlpclnt.h>
 
 #include <3ds/gpu/gx.h>
 #include <3ds/gpu/gpu.h>

libctru/include/3ds/services/dlpclnt.h

@@ -0,0 +1,62 @@
+#include <3ds/types.h>
+#include <3ds/result.h>
+#include <3ds/svc.h>
+#include <3ds/srv.h>
+#include <3ds/synchronization.h>
+#include <3ds/services/cfgu.h>
+#include <3ds/services/ndm.h>
+#include <3ds/ipc.h>
+
+typedef struct {
+   u32 uniqueId;
+   u32 revision;
+   u8 macAddr[6];
+} dlpTitleInfo;
+
+Result dlpClntInit(void);
+
+void dlpClntExit(void);
+
+bool dlpClntWaitForEvent(bool nextEvent, bool wait);
+
+u64 dlpCreateChildTid(u32 uniqueId, u32 revision);
+
+Result DLPCLNT_Initialize(size_t sharedMemSize, u8 maxScanTitles, size_t unk, Handle sharedmemHandle, Handle eventHandle);
+
+Result DLPCLNT_Finalize(void);
+
+//DLPCLNT_GetEventDesc();
+
+Result DLPCLNT_GetChannel(u16* channel);
+
+Result DLPCLNT_StartScan(u16 channel, u8* macAddr);
+
+Result DLPCLNT_StopScan(void);
+/*
+DLPCLNT_GetServerInfo();
+
+DLPCLNT_GetTitleInfo();
+*/
+Result DLPCLNT_GetTitleInfoInOrder(void* buf, size_t size, size_t* actual_size);
+/*
+DLPCLNT_DeleteScanInfo();
+*/
+Result DLPCLNT_PrepareForSystemDownload(u8* macAddr, u32 uniqueId, u32 revision);
+/*
+DLPCLNT_StartSystemDownload();
+*/
+Result DLPCLNT_StartTitleDownload(u8* macAddr, u32 uniqueId, u32 revision);
+
+Result DLPCLNT_GetMyStatus(u32* status);
+/*
+DLPCLNT_GetConnectingNodes();
+
+DLPCLNT_GetNodeInfo();
+*/
+Result DLPCLNT_GetWirelessRebootPassphrase(void* buf);
+
+Result DLPCLNT_StopSession(void);
+/*
+DLPCLNT_GetCupVersion();
+
+DLPCLNT_GetDupAvailability();*/

libctru/include/3ds/services/ns.h

@@ -33,6 +33,8 @@ Result NS_TerminateTitle(void);
  */
 Result NS_LaunchApplicationFIRM(u64 titleid, u32 flags);
 
+Result NS_SetWirelessRebootInfo(u8* macAddr, u8* passphrase);
+
 /**
  * @brief Reboots to a title.
  * @param mediatype Mediatype of the title.

libctru/source/services/dlpclnt.c

@@ -0,0 +1,296 @@
+#include <malloc.h>
+#include <string.h>
+#include <3ds/types.h>
+#include <3ds/result.h>
+#include <3ds/svc.h>
+#include <3ds/srv.h>
+#include <3ds/synchronization.h>
+#include <3ds/services/dlpclnt.h>
+#include <3ds/services/cfgu.h>
+#include <3ds/services/ndm.h>
+#include <3ds/ipc.h>
+
+
+Handle dlpClntHandle;
+Handle dlpClntMemHandle;
+u8* dlpClntMemAddr;
+size_t dlpClntMemSize;
+Handle dlpClntEventHandle;
+
+static int dlpClntRefCount;
+
+u32 ndm_state;
+
+Result dlpClntInit() {
+	Result ret = 0;
+	ndm_state = 0;
+
+	if (AtomicPostIncrement(&dlpClntRefCount)) return 0;
+
+	ret = ndmuInit();
+	if (R_FAILED(ret))goto end;
+
+	ndm_state = 1;
+	ret = NDMU_EnterExclusiveState(NDM_EXCLUSIVE_STATE_LOCAL_COMMUNICATIONS);
+	if (R_FAILED(ret))goto end;
+
+	ndm_state = 2;
+
+	dlpClntMemSize = 0x232000;
+	dlpClntMemAddr = memalign(0x1000, dlpClntMemSize);
+	if (dlpClntMemAddr == NULL) {
+		ret = -1;
+		goto end;
+	}
+
+	ret = svcCreateMemoryBlock(&dlpClntMemHandle, (u32)dlpClntMemAddr, dlpClntMemSize, 0x0, MEMPERM_READ | MEMPERM_WRITE);
+	if (R_FAILED(ret))goto end;
+
+	ret = svcCreateEvent(&dlpClntEventHandle, RESET_ONESHOT);
+	if (R_FAILED(ret))goto end;
+
+	ret = srvGetServiceHandle(&dlpClntHandle, "dlp:CLNT");
+	if (R_FAILED(ret))goto end;
+
+	ret = DLPCLNT_Initialize(dlpClntMemSize, 0x10, 0x200000, dlpClntMemHandle, dlpClntEventHandle);
+	if (R_FAILED(ret)) {
+		svcCloseHandle(dlpClntHandle);
+		dlpClntHandle = 0;
+		goto end;
+	}
+
+	return ret;
+end:
+	dlpClntExit();
+	return ret;
+}
+
+void dlpClntExit() {
+	if (AtomicDecrement(&dlpClntRefCount)) return;
+
+	if (dlpClntHandle)
+	{
+		DLPCLNT_Finalize();
+		svcCloseHandle(dlpClntHandle);
+		dlpClntHandle = 0;
+	}
+
+	if (ndm_state) {
+		if (ndm_state == 2)NDMU_LeaveExclusiveState();
+		ndmuExit();
+		ndm_state = 0;
+	}
+
+	if (dlpClntEventHandle) {
+		svcCloseHandle(dlpClntMemHandle);
+		dlpClntMemHandle = 0;
+	}
+
+	if (dlpClntMemHandle)
+	{
+		svcCloseHandle(dlpClntMemHandle);
+		dlpClntMemHandle = 0;
+	}
+
+	dlpClntMemAddr = NULL;
+	dlpClntMemSize = 0;
+}
+
+bool dlpClntWaitForEvent(bool nextEvent, bool wait) {
+	bool ret = true;
+	u64 delayvalue = U64_MAX;
+
+	if (!wait)delayvalue = 0;
+
+	if (nextEvent)svcClearEvent(dlpClntEventHandle);
+
+	if (svcWaitSynchronization(dlpClntEventHandle, delayvalue) != 0 && !wait)ret = false;
+
+	if (!nextEvent)svcClearEvent(dlpClntEventHandle);
+
+	return ret;
+}
+
+u64 dlpCreateChildTid(u32 uniqueId, u32 revision) {
+	u64 tid = 0;
+	if (uniqueId) {
+		tid = (u64)0x40001 << 32;
+		tid |= revision | ((uniqueId & 0xff0fffff) << 8);
+	}
+	return tid;
+}
+
+Result DLPCLNT_Initialize(size_t sharedMemSize, u8 maxScanTitles, size_t unk, Handle sharedMemHandle, Handle eventHandle) {
+	u32* cmdbuf = getThreadCommandBuffer();
+
+	cmdbuf[0] = IPC_MakeHeader(0x1,3,3); // 0x100C3
+	cmdbuf[1] = sharedMemSize;
+	cmdbuf[2] = maxScanTitles;
+	cmdbuf[3] = unk;
+	cmdbuf[4] = IPC_Desc_SharedHandles(2);
+	cmdbuf[5] = sharedMemHandle;
+	cmdbuf[6] = eventHandle;
+
+	Result ret = 0;
+	if (R_FAILED(ret = svcSendSyncRequest(dlpClntHandle)))return ret;
+
+	return cmdbuf[1];
+}
+
+Result DLPCLNT_Finalize() {
+	u32* cmdbuf = getThreadCommandBuffer();
+
+	cmdbuf[0] = IPC_MakeHeader(0x2,0,0); // 0x20000
+
+	Result ret = 0;
+	if (R_FAILED(ret = svcSendSyncRequest(dlpClntHandle)))return ret;
+
+	return cmdbuf[1];
+}
+
+Result DLPCLNT_GetChannel(u16* channel) {
+	u32* cmdbuf = getThreadCommandBuffer();
+
+	cmdbuf[0] = IPC_MakeHeader(0x4,0,0); // 0x40000
+
+	Result ret = 0;
+	if (R_FAILED(ret = svcSendSyncRequest(dlpClntHandle)))return ret;
+
+	*channel = cmdbuf[2];
+
+	return cmdbuf[1];
+}
+
+Result DLPCLNT_StartScan(u16 channel, u8* macAddr) {
+	u32* cmdbuf = getThreadCommandBuffer();
+
+	cmdbuf[0] = IPC_MakeHeader(0x5,6,0); // 0x50180
+	cmdbuf[1] = channel;
+	cmdbuf[2] = 0; // tidLow filter
+	cmdbuf[3] = 0; // tidHigh filter
+	if (macAddr) {
+		memcpy(cmdbuf + 4, macAddr, 6);
+	}
+	else {
+		cmdbuf[4] = 0; // mac address filter
+		cmdbuf[5] = 0;
+	}
+	cmdbuf[6] = 0; // unknown state filter
+
+	Result ret = 0;
+	if (R_FAILED(ret = svcSendSyncRequest(dlpClntHandle)))return ret;
+
+	return cmdbuf[1];
+}
+
+Result DLPCLNT_StopScan() {
+	u32* cmdbuf = getThreadCommandBuffer();
+
+	cmdbuf[0] = IPC_MakeHeader(0x6,0,0); // 0x60000
+
+	Result ret = 0;
+	if (R_FAILED(ret = svcSendSyncRequest(dlpClntHandle)))return ret;
+
+	return cmdbuf[1];
+}
+
+Result DLPCLNT_GetTitleInfoInOrder(void* buf, size_t size, size_t* actual_size) {
+	u32* cmdbuf = getThreadCommandBuffer();
+	u32 saved_threadstorage[2];
+
+	cmdbuf[0] = IPC_MakeHeader(0x9,1,0); // 0x90040
+	cmdbuf[1] = 0; // 0 = Iterate?, 1 = Don't Iterate?
+
+	u32* staticbufs = getThreadStaticBuffers();
+	saved_threadstorage[0] = staticbufs[0];
+	saved_threadstorage[1] = staticbufs[1];
+
+	staticbufs[0] = IPC_Desc_StaticBuffer(size, 0);
+	staticbufs[1] = (u32)buf;
+
+	Result ret = 0;
+	ret = svcSendSyncRequest(dlpClntHandle);
+
+	staticbufs[0] = saved_threadstorage[0];
+	staticbufs[1] = saved_threadstorage[1];
+
+	if (R_FAILED(ret))return ret;
+
+	ret = cmdbuf[1];
+
+	if (R_SUCCEEDED(ret))
+	{
+		if (actual_size)*actual_size = cmdbuf[2];
+	}
+
+	return ret;
+}
+
+Result DLPCLNT_PrepareForSystemDownload(u8* macAddr, u32 uniqueId, u32 revision) {
+	u32* cmdbuf = getThreadCommandBuffer();
+
+	u64 tid = dlpCreateChildTid(uniqueId, revision);
+
+	cmdbuf[0] = IPC_MakeHeader(0xB,4,0); // 0xB0100
+	memcpy(cmdbuf + 1, macAddr, 6);
+	cmdbuf[3] = tid & 0xFFFFFFFF;
+	cmdbuf[4] = tid >> 32;
+
+	Result ret = 0;
+	if (R_FAILED(ret = svcSendSyncRequest(dlpClntHandle)))return ret;
+
+	return cmdbuf[1];
+}
+
+Result DLPCLNT_StartTitleDownload(u8* macAddr, u32 uniqueId, u32 revision) {
+	u32* cmdbuf = getThreadCommandBuffer();
+
+	u64 tid = dlpCreateChildTid(uniqueId, revision);
+
+	cmdbuf[0] = IPC_MakeHeader(0xD,4,0); // 0xD0100
+	memcpy(cmdbuf + 1, macAddr, 6);
+	cmdbuf[3] = tid & 0xFFFFFFFF;
+	cmdbuf[4] = tid >> 32;
+
+	Result ret = 0;
+	if (R_FAILED(ret = svcSendSyncRequest(dlpClntHandle)))return ret;
+
+	return cmdbuf[1];
+}
+
+Result DLPCLNT_GetMyStatus(u32* status) {
+	u32* cmdbuf = getThreadCommandBuffer();
+
+	cmdbuf[0] = IPC_MakeHeader(0xE,0,0); // 0xE00000
+
+	Result ret = 0;
+	if (R_FAILED(ret = svcSendSyncRequest(dlpClntHandle)))return ret;
+
+	memcpy(status, cmdbuf + 2, sizeof(u32) * 3);
+
+	return cmdbuf[1];
+}
+
+Result DLPCLNT_GetWirelessRebootPassphrase(void* buf) {
+	u32* cmdbuf = getThreadCommandBuffer();
+
+	cmdbuf[0] = IPC_MakeHeader(0x11,0,0); // 0x110000
+
+	Result ret = 0;
+	if (R_FAILED(ret = svcSendSyncRequest(dlpClntHandle)))return ret;
+
+	memcpy(buf, cmdbuf + 2, 9);
+
+	return cmdbuf[1];
+}
+
+Result DLPCLNT_StopSession(void) {
+	u32* cmdbuf = getThreadCommandBuffer();
+
+	cmdbuf[0] = IPC_MakeHeader(0x12,0,0); // 0x120000
+
+	Result ret = 0;
+	if (R_FAILED(ret = svcSendSyncRequest(dlpClntHandle)))return ret;
+
+	return cmdbuf[1];
+}