/*
 * Copyright (c) Souldbminer, Lightos_ and Horizon OC Contributors
 *
 * This program is free software; you can redistribute it and/or modify it
 * under the terms and conditions of the GNU General Public License,
 * version 2, as published by the Free Software Foundation.
 *
 * This program is distributed in the hope it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License for
 * more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 * 
 */
 
/* --------------------------------------------------------------------------
 * "THE BEER-WARE LICENSE" (Revision 42):
 * <p-sam@d3vs.net>, <natinusala@gmail.com>, <m4x@m4xw.net>
 * wrote this file. As long as you retain this notice you can do whatever you
 * want with this stuff. If you meet any of us some day, and you think this
 * stuff is worth it, you can buy us a beer in return.  - The sys-clk authors
 * --------------------------------------------------------------------------
 */


#include "nxExt/ipc_server.h"
#include <string.h>

Result ipcServerInit(IpcServer* server, const char* name, u32 max_sessions)
{
    if(max_sessions < 1 || max_sessions > (MAX_WAIT_OBJECTS - 1))
    {
        return MAKERESULT(Module_Libnx, LibnxError_BadInput);
    }

    server->srvName = smEncodeName(name);
    server->max = max_sessions + 1;
    server->count = 0;

    Result rc = smRegisterService(&server->handles[0], server->srvName, false, max_sessions);
    if(R_SUCCEEDED(rc))
    {
        server->count = 1;
    }
    return rc;
}

Result ipcServerExit(IpcServer* server)
{
    for(u32 i = 0; i < server->count; i++)
    {
        svcCloseHandle(server->handles[i]);
    }
    server->count = 0;
    return smUnregisterService(server->srvName);
}

static Result _ipcServerAddSession(IpcServer* server, Handle session)
{
    if(server->count >= server->max)
    {
        return MAKERESULT(Module_Libnx, LibnxError_OutOfMemory);
    }

    server->handles[server->count] = session;
    server->count++;
    return 0;
}

static Result _ipcServerDeleteSession(IpcServer* server, u32 index)
{
    if(!index || index >= server->count)
    {
        return MAKERESULT(Module_Libnx, LibnxError_BadInput);
    }

    svcCloseHandle(server->handles[index]);

    for(u32 j = index; j < (server->count - 1); j++)
    {
        server->handles[j] = server->handles[j + 1];
    }
    server->count--;
    return 0;
}

static Result _ipcServerParseRequest(IpcServerRequest* r)
{
    u8* base = armGetTls();

    r->hipc = hipcParseRequest(base);
    r->data.cmdId = 0;
    r->data.size = 0;
    r->data.ptr =  NULL;

    if(r->hipc.meta.type == CmifCommandType_Request)
    {
        IpcServerRawHeader* header = cmifGetAlignedDataStart(r->hipc.data.data_words, base);
        size_t dataSize = r->hipc.meta.num_data_words * 4;

        if(!header || dataSize < sizeof(IpcServerRawHeader) || header->magic != CMIF_IN_HEADER_MAGIC)
        {
            return MAKERESULT(Module_Libnx, LibnxError_BadInput);
        }

        r->data.cmdId = header->cmdId;
        if(dataSize > sizeof(IpcServerRawHeader))
        {
            r->data.size = dataSize - sizeof(IpcServerRawHeader);
            r->data.ptr = ((u8*)header) + sizeof(IpcServerRawHeader);
        }
    }

    return 0;
}

static void _ipcServerPrepareResponse(Result rc, void* data, size_t dataSize)
{
    u8* base = armGetTls();
    HipcRequest hipc = hipcMakeRequestInline(base,
        .type = CmifCommandType_Request,
        .num_data_words = (sizeof(IpcServerRawHeader) + dataSize + 0x10) / 4,
    );

    IpcServerRawHeader* rawHeader = cmifGetAlignedDataStart(hipc.data_words, base);
    rawHeader->magic = CMIF_OUT_HEADER_MAGIC;
    rawHeader->result = rc;

    if(R_SUCCEEDED(rc))
    {
        memcpy(((u8*)rawHeader) + sizeof(IpcServerRawHeader), data, dataSize);
    }
}

static Result _ipcServerProcessNewSession(IpcServer* server)
{
    Handle session;
    Result rc = svcAcceptSession(&session, server->handles[0]);
    if(R_SUCCEEDED(rc) && R_FAILED(rc = _ipcServerAddSession(server, session)))
    {
        svcCloseHandle(session);
    }
    return rc;
}

static Result _ipcServerProcessSession(IpcServer* server, IpcServerRequestHandler handler, void* userdata, u32 handleIndex)
{
    s32 unusedIndex;
    IpcServerRequest r;
    size_t dataSize = 0;
    u8 data[IPC_SERVER_EXT_RESPONSE_MAX_DATA_SIZE];
    bool close = false;

    Result rc = svcReplyAndReceive(&unusedIndex, &server->handles[handleIndex], 1, 0, UINT64_MAX);
    if(R_SUCCEEDED(rc))
    {
        rc = _ipcServerParseRequest(&r);
    }

    if(R_SUCCEEDED(rc))
    {
        switch(r.hipc.meta.type)
        {
            case CmifCommandType_Request:
                _ipcServerPrepareResponse(
                    handler(userdata, &r, data, &dataSize),
                    data,
                    dataSize
                );
                break;
            case CmifCommandType_Close:
                _ipcServerPrepareResponse(0, NULL, 0);
                close = true;
                break;
            default:
                _ipcServerPrepareResponse(MAKERESULT(11, 403), NULL, 0);
                break;
        }

        rc = svcReplyAndReceive(&unusedIndex, &server->handles[handleIndex], 0, server->handles[handleIndex], 0);
        if(rc == KERNELRESULT(TimedOut))
        {
            rc = 0;
        }
    }

    if(R_FAILED(rc) || close)
    {
        _ipcServerDeleteSession(server, handleIndex);
    }

    return rc;
}

Result ipcServerProcess(IpcServer* server, IpcServerRequestHandler handler, void* userdata)
{
    s32 handleIndex = -1;
    Result rc = svcWaitSynchronization(&handleIndex, server->handles, server->count, UINT64_MAX);

    if(R_SUCCEEDED(rc) && (handleIndex < 0 || handleIndex >= server->count))
    {
        rc = MAKERESULT(Module_Libnx, LibnxError_NotFound);
    }

    if(R_SUCCEEDED(rc))
    {
        if(handleIndex)
        {
            rc = _ipcServerProcessSession(server, handler, userdata, handleIndex);
        }
        else
        {
            rc = _ipcServerProcessNewSession(server);
        }
    }

    return rc;
}