/****************************************************************************
 * USB Loader GX Team
 * banner.c
 *
 * Dump opening.bnr thanks to Wiipower
 ***************************************************************************/

#include <gctypes.h>
#include <gccore.h>
#include <ogcsys.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <malloc.h>

#include "libfat/fat.h"
#include "fatmounter.h"
#include "usbloader/wdvd.h"
#include "usbloader/disc.h"
#include "banner.h"
#include "patches/fst.h"
#include "usbloader/fstfile.h"

s32 dump_banner(const u8* discid, const char * dest)
{
    // Mount the disc
    //Disc_SetWBFS(1, (u8*)discid);
    Disc_SetUSB(discid);

    Disc_Open();

    u64 offset;
    s32 ret;

    ret = __Disc_FindPartition(&offset);
    if (ret < 0) return ret;

    ret = WDVD_OpenPartition(offset);

    if (ret < 0)
    {
        //printf("ERROR: OpenPartition(0x%llx) %d\n", offset, ret);
        return ret;
    }

    // Read where to find the fst.bin
    u32 *buffer = memalign(32, 0x20);

    if (buffer == NULL)
    {
        //Out of memory
        return -1;
    }

    ret = WDVD_Read(buffer, 0x20, 0x420);
    if (ret < 0) return ret;

    // Read fst.bin
    void *fstbuffer = memalign(32, buffer[2] * 4);
    FST_ENTRY *fst = (FST_ENTRY *) fstbuffer;

    if (fst == NULL)
    {
        //Out of memory
        free(buffer);
        return -1;
    }

    ret = WDVD_Read(fstbuffer, buffer[2] * 4, buffer[1] * 4);
    if (ret < 0) return ret;

    free(buffer);

    // Search the fst.bin
    u32 count = fst[0].filelen;
    int i;
    u32 index = 0;

    for (i = 1; i < count; i++)
    {
        if (strstr(fstfiles(fst, i), "opening.bnr") != NULL)
        {
            index = i;
        }
    }

    if (index == 0)
    {
        //opening.bnr not found
        free(fstbuffer);
        return -1;
    }

    // Load the .bnr
    u8 *banner = memalign(32, fst[index].filelen);

    if (banner == NULL)
    {
        //Out of memory
        free(fstbuffer);
        return -1;
    }

    ret = WDVD_Read((void *) banner, fst[index].filelen, fst[index].fileoffset * 4);
    if (ret < 0) return ret;

    WDVD_Reset();
    WDVD_ClosePartition();
    //fatInitDefault();
    //SDCard_Init();
    WDVD_SetUSBMode(NULL, 0);
    FILE *fp = fopen(dest, "wb");
    if (fp)
    {
        fwrite(banner, 1, fst[index].filelen, fp);
        fclose(fp);
    }
    free(fstbuffer);
    free(banner);

    return 1;
}