libnx/nx/source/nvidia/map.c

112 lines
2.5 KiB
C

#include <string.h>
#include "../services/service_guard.h"
#include "arm/cache.h"
#include "kernel/svc.h"
#include "services/nv.h"
#include "nvidia/ioctl.h"
#include "nvidia/map.h"
static u32 g_nvmap_fd = -1;
#define nvMapInitialize nvMapInit
NX_GENERATE_SERVICE_GUARD(nvMap);
Result _nvMapInitialize(void)
{
return nvOpen(&g_nvmap_fd, "/dev/nvmap");
}
void _nvMapCleanup(void)
{
if (g_nvmap_fd != -1) {
nvClose(g_nvmap_fd);
g_nvmap_fd = -1;
}
}
u32 nvMapGetFd(void)
{
return g_nvmap_fd;
}
Result nvMapCreate(NvMap* m, void* cpu_addr, u32 size, u32 align, NvKind kind, bool is_cpu_cacheable)
{
Result rc;
if (align < 0x1000)
align = 0x1000;
if (align & (align-1))
return MAKERESULT(Module_Libnx, LibnxError_BadInput);
if (!size || (size & 0xFFF))
return MAKERESULT(Module_Libnx, LibnxError_BadInput);
if (!cpu_addr || ((uintptr_t)cpu_addr & 0xFFF))
return MAKERESULT(Module_Libnx, LibnxError_BadInput);
m->has_init = true;
m->is_cpu_cacheable = is_cpu_cacheable;
m->size = size;
m->handle = -1;
m->cpu_addr = cpu_addr;
m->kind = kind;
rc = nvioctlNvmap_Create(g_nvmap_fd, size, &m->handle);
if (R_SUCCEEDED(rc))
rc = nvioctlNvmap_Alloc(g_nvmap_fd, m->handle,
0, is_cpu_cacheable ? 1 : 0, align, kind, m->cpu_addr);
if (R_SUCCEEDED(rc) && !is_cpu_cacheable) {
armDCacheFlush(m->cpu_addr, m->size);
svcSetMemoryAttribute(m->cpu_addr, m->size, 8, 8);
}
if (R_SUCCEEDED(rc))
rc = nvioctlNvmap_GetId(g_nvmap_fd, m->handle, &m->id);
if (R_FAILED(rc))
nvMapClose(m);
return rc;
}
Result nvMapLoadRemote(NvMap* m, u32 id)
{
Result rc;
memset(m, 0, sizeof(*m));
rc = nvioctlNvmap_FromId(g_nvmap_fd, id, &m->handle);
if (R_SUCCEEDED(rc)) {
m->has_init = true;
m->id = id;
rc = nvioctlNvmap_Param(g_nvmap_fd, m->handle, NvMapParam_Size, &m->size);
}
if (R_SUCCEEDED(rc))
rc = nvioctlNvmap_Param(g_nvmap_fd, m->handle, NvMapParam_Kind, (u32*)&m->kind);
if (R_FAILED(rc))
nvMapClose(m);
return rc;
}
void nvMapClose(NvMap* m)
{
if (!m->has_init)
return;
if (m->handle != -1) {
nvioctlNvmap_Free(g_nvmap_fd, m->handle);
m->handle = -1;
}
if (m->cpu_addr) {
if (!m->is_cpu_cacheable)
svcSetMemoryAttribute(m->cpu_addr, m->size, 8, 0);
m->cpu_addr = NULL;
}
m->has_init = false;
}