#include <cstdint>

#define k_memset ((void (*)( uint32_t , uint32_t, uint32_t))0xfff09d60)
#define KiReport ((void (*)( const char*, ... ))0xfff0ad0c)
#define IopShell_AsyncCallback (0xfff1b7d8)
#define IopShell_ReadCallback (0xfff1b9a0)
#define k_memcpy ((void (*)(void*,void*, uint32_t))0xfff09e44)
#define IPCKDriver_OpenAsync ((int32_t (*)(uint32_t client_ram_pid, const char *device, uint32_t mode, uint32_t asyncCallback, void * context))0xfff0b9c8)
#define WaitAsyncReply ((uint32_t (*)(void* , uint32_t, uint32_t))0xfff180dc)

#define IPCKDriver_IoctlAsync ((int32_t (*)( \
                                uint32_t clientProcessId,\
                                uint32_t loaderProcessId,\
                                uint32_t handle,\
                                uint32_t request,\
                                void* inBuf,\
                                uint32_t inLen,\
                                void* outBuf,\
                                uint32_t outLen,\
                                uint32_t asyncCallback,\
                                void * context))0xfff0bc5c)


uint32_t IPCKDriver_IOS_Open_Sync(uint32_t clientProcessId, const char *device_name, uint32_t mode) {
    register int r13 asm ("r13");
    k_memcpy((void *) (r13 - 0x1180), (void *) device_name, 0x10);
    void *context = (void *) (r13 + 0x450);
    auto error = IPCKDriver_OpenAsync(clientProcessId, (const char *) (r13 - 0x1180), mode, IopShell_AsyncCallback, context);
    if (error < 0) {
        return error;
    }
    return WaitAsyncReply(context, 20000000, 6);
}

uint32_t IPCKDriver_IOS_Ioctl_Sync(uint32_t clientProcessId, uint32_t handle, uint32_t request, void *inBuf, uint32_t inLen, void *outBuf, uint32_t ouLen) {
    register int r13 asm ("r13");
    void *context = (void *) (r13 + 0x450);

    auto error = IPCKDriver_IoctlAsync(clientProcessId, 0xFFFFFFFF, handle, request, inBuf, inLen, outBuf, ouLen, IopShell_AsyncCallback, context);

    if (error < 0) {
        return error;
    }
    uint32_t res = WaitAsyncReply(context, 20000000, 6);
    return res;
}


void IopShellInitInternal() {
    register int r13 asm ("r13");
    *((uint32_t *) (r13 + 0x450)) = 0;
    *((uint32_t *) (r13 + 0x454)) = 0;
    *((uint32_t *) (r13 + 0x458)) = 0xffffffe3;
    auto handle = IPCKDriver_IOS_Open_Sync(0, "/dev/iopsh", (uint32_t) 5);
    if (static_cast<int>(handle) < 0) {
        KiReport("####################################################################################\n");
        KiReport("IopShellInit failed\n");
        KiReport("####################################################################################\n");
        return;
    }
    *(uint32_t *) (r13 + -0x6d84) = static_cast<uint32_t>(handle);
    *(uint32_t *) (r13 + -0x1178) = 5;
    *(uint32_t *) (r13 + -0x117c) = 0x1040;

    auto result = IPCKDriver_IOS_Ioctl_Sync(0, static_cast<uint32_t>(handle), 3, (void *) (r13 - 0x1180), 0x14, (void *) (r13 - 0x1140), 0x4);
    if (static_cast<int>(result) < 0) {
        KiReport("####################################################################################\n");
        KiReport("IopShellInit failed\n");
        KiReport("####################################################################################\n");
        return;
    }
    *(uint32_t *) (r13 + -0x117c) = 0x636F7300; // "cos"
    result = IPCKDriver_IOS_Ioctl_Sync(0, static_cast<uint32_t>(handle), 1, (void *) (r13 - 0x1180), 0x14, (void *) (r13 - 0x1140), 0x4);
    if (static_cast<int>(result) < 0) {
        KiReport("####################################################################################\n");
        KiReport("IopShellInit failed\n");
        KiReport("####################################################################################\n");
        return;
    }
    *((uint32_t *) (r13 + 0x450)) = 0;
    *((uint32_t *) (r13 + 0x454)) = 0;
    *((uint32_t *) (r13 + 0x458)) = 0xffffffe3;
    k_memset((r13 - 0x1180), 0, 0x1040);
    result = IPCKDriver_IoctlAsync(0, 0xFFFFFFFF, handle, 5, nullptr, 0, (void *) (r13 - 0x1180), 0x1040, IopShell_ReadCallback, (void*) (r13 + 0x450));
    if (static_cast<int>(result) < 0) {
        KiReport("####################################################################################\n");
        KiReport("IopShellInit failed\n");
        KiReport("####################################################################################\n");
        return;
    }
    KiReport("IopShellInit was successful\n");
}