#include <ogcsys.h>
#include <stdio.h>
#include <gccore.h>
#include <string.h>
#include <malloc.h>

#include "fatmounter.h"
#include "apploader.h"
#include "wdvd.h"
#include "fstfile.h"
#include "../patches/dvd_broadway.h"

extern u8 mountMethod;

/** Alternate dolloader made by WiiPower modified by dimok **/
bool Load_Dol(void **buffer, int* dollen, char * filepath)
{
    int ret;
    FILE* file;
    void* dol_buffer;

    char fullpath[200];
    char gameidbuffer6[7];
    memset(gameidbuffer6, 0, 7);
    memcpy(gameidbuffer6, (char*) 0x80000000, 6);
    snprintf(fullpath, 200, "%s%s.dol", filepath, gameidbuffer6);

    //    SDCard_Init();
    //    USBDevice_Init();

    file = fopen(fullpath, "rb");

    if (file == NULL)
    {
        fclose(file);
        //     SDCard_deInit();
        //     USBDevice_deInit();
        return false;
    }

    int filesize;
    fseek(file, 0, SEEK_END);
    filesize = ftell(file);
    fseek(file, 0, SEEK_SET);

    dol_buffer = malloc(filesize);
    if (dol_buffer == NULL)
    {
        fclose(file);
        //      SDCard_deInit();
        //      USBDevice_deInit();
        return false;
    }
    ret = fread(dol_buffer, 1, filesize, file);
    if (ret != filesize)
    {
        free(dol_buffer);
        fclose(file);
        //     SDCard_deInit();
        //     USBDevice_deInit();
        return false;
    }
    fclose(file);

    // SDCard_deInit();
    // USBDevice_deInit();
    *buffer = dol_buffer;
    *dollen = filesize;
    return true;
}

bool Remove_001_Protection(void *Address, int Size)
{
    u8 SearchPattern[16] = { 0x40, 0x82, 0x00, 0x0C, 0x38, 0x60, 0x00, 0x01, 0x48, 0x00, 0x02, 0x44, 0x38, 0x61, 0x00,
            0x18 };
    u8 PatchData[16] =
            { 0x40, 0x82, 0x00, 0x04, 0x38, 0x60, 0x00, 0x01, 0x48, 0x00, 0x02, 0x44, 0x38, 0x61, 0x00, 0x18 };

    void *Addr = Address;
    void *Addr_end = Address + Size;

    while (Addr <= Addr_end - sizeof(SearchPattern))
    {
        if (memcmp(Addr, SearchPattern, sizeof(SearchPattern)) == 0)
        {
            memcpy(Addr, PatchData, sizeof(PatchData));
            return true;
        }
        Addr += 4;
    }
    return false;
}

typedef struct _dolheader
{
    u32 text_pos[7];
    u32 data_pos[11];
    u32 text_start[7];
    u32 data_start[11];
    u32 text_size[7];
    u32 data_size[11];
    u32 bss_start;
    u32 bss_size;
    u32 entry_point;
} dolheader;

static dolheader *dolfile;

u32 load_dol_image(void *dolstart, u8 videoSelected, u8 patchcountrystring, u8 vipatch, u8 cheat)
{

    u32 i;

    if (dolstart)
    {
        dolfile = (dolheader *) dolstart;
        for (i = 0; i < 7; i++)
        {
            if ((!dolfile->text_size[i]) || (dolfile->text_start[i] < 0x100)) continue;

            ICInvalidateRange((void *) dolfile->text_start[i], dolfile->text_size[i]);
            memmove((void *) dolfile->text_start[i], dolstart + dolfile->text_pos[i], dolfile->text_size[i]);
            gamepatches((void *) dolfile->text_start[i], dolfile->text_size[i], videoSelected, patchcountrystring,
                    vipatch, cheat);
            Remove_001_Protection((void *) dolfile->data_start[i], dolfile->data_size[i]);
        }

        for (i = 0; i < 11; i++)
        {
            if ((!dolfile->data_size[i]) || (dolfile->data_start[i] < 0x100)) continue;

            memmove((void *) dolfile->data_start[i], dolstart + dolfile->data_pos[i], dolfile->data_size[i]);
            gamepatches((void *) dolfile->data_start[i], dolfile->data_size[i], videoSelected, patchcountrystring,
                    vipatch, cheat);
            Remove_001_Protection((void *) dolfile->data_start[i], dolfile->data_size[i]);
            DCFlushRangeNoSync((void *) dolfile->data_start[i], dolfile->data_size[i]);
        }
        /*
         memset ((void *) dolfile->bss_start, 0, dolfile->bss_size);
         DCFlushRange((void *) dolfile->bss_start, dolfile->bss_size);
         */
        return dolfile->entry_point;
    }
    return 0;
}

static int i;
static int phase;

u32 load_dol_start(void *dolstart)
{
    if (dolstart)
    {
        dolfile = (dolheader *) dolstart;
        return dolfile->entry_point;
    }
    else
    {
        return 0;
    }

    memset((void *) dolfile->bss_start, 0, dolfile->bss_size);
    DCFlushRange((void *) dolfile->bss_start, dolfile->bss_size);

    phase = 0;
    i = 0;
}

bool load_dol_image_modified(void **offset, u32 *pos, u32 *len)
{
    if (phase == 0)
    {
        if (i == 7)
        {
            phase = 1;
            i = 0;
        }
        else
        {
            if ((!dolfile->text_size[i]) || (dolfile->text_start[i] < 0x100))
            {
                *offset = 0;
                *pos = 0;
                *len = 0;
            }
            else
            {
                *offset = (void *) dolfile->text_start[i];
                *pos = dolfile->text_pos[i];
                *len = dolfile->text_size[i];
            }
            i++;
            return true;
        }
    }

    if (phase == 1)
    {
        if (i == 11)
        {
            phase = 2;
            return false;
        }

        if ((!dolfile->data_size[i]) || (dolfile->data_start[i] < 0x100))
        {
            *offset = 0;
            *pos = 0;
            *len = 0;
        }
        else
        {
            *offset = (void *) dolfile->data_start[i];
            *pos = dolfile->data_pos[i];
            *len = dolfile->data_size[i];
        }
        i++;
        return true;
    }
    return false;
}
static vu32 dvddone = 0;
void __dvd_readidcb(s32 result)
{
    dvddone = result;
}
u32 Load_Dol_from_disc(u32 doloffset, u8 videoSelected, u8 patchcountrystring, u8 vipatch, u8 cheat)
{
    int ret;
    void *dol_header;
    u32 entrypoint;

    dol_header = memalign(32, sizeof(dolheader));
    if (dol_header == NULL)
    {
        return -1;
    }

    if (!mountMethod)
        ret = WDVD_Read(dol_header, sizeof(dolheader), (doloffset << 2));

    else
    {
        dvddone = 0;
        ret = bwDVD_LowRead(dol_header, sizeof(dolheader), doloffset, __dvd_readidcb);
        while (ret >= 0 && dvddone == 0)
            ;
    }

    entrypoint = load_dol_start(dol_header);

    if (entrypoint == 0)
    {
        free(dol_header);
        return -1;
    }

    void *offset;
    u32 pos;
    u32 len;

    while (load_dol_image_modified(&offset, &pos, &len))
    {
        if (len != 0)
        {
            ret = WDVD_Read(offset, len, (doloffset << 2) + pos);

            gamepatches(offset, len, videoSelected, patchcountrystring, vipatch, cheat);

            Remove_001_Protection(offset, len);

            DCFlushRange(offset, len);
        }
    }

    free(dol_header);

    return entrypoint;

}