#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;
}