/*
    gbapatcher.h
    Copyright (C) 2008-2009 somebody
    Copyright (C) 2009 yellow wood goblin

    SPDX-License-Identifier: GPL-3.0-or-later
*/

#pragma once

#include <nds.h>
#include <setjmp.h>

// typedef void (*TFlashWriteFunc)(u32 address,const u8* buffer,u32 size);

class CGbaWriter {
  public:
    virtual ~CGbaWriter();
    virtual void Open(void) = 0;
    virtual void Write(u32 address, const u8* buffer, u32 size) = 0;
    virtual void Close(void) = 0;
    virtual u32 MaxSize(void) = 0;
};

class CGbaPatcher {
  private:
    enum TPatchType {
        // flash
        ESRAMLabel,
        EEEPROMLabel,
        EEEPROMRead,
        EEEPROMWrite,
        EEEPROMWrite124,
        EEEPROMWrite126,
        EFLASH1M10XLabel,
        EFLASH1M102_1,
        EFLASH1M102_2,
        EFLASH1M102_3,
        EFLASH1M102_4,
        EFLASH1M102_5,
        EFLASH1M103_1,
        EFLASH1M103_2,
        EFLASH1M103_3,
        EFLASH1M103_4,
        EFLASHLabel,
        EFLASH_1,
        EFLASH_2,
        EFLASH_3,
        EFLASH_4,
        EFLASHV123_1,
        EFLASHV123_2,
        EFLASHV123_3,
        EFLASHV123_4,
        EFLASHV123_5,
        EFLASH512Label,
        EFLASH512_1,
        EFLASH512_2,
        // last
        ELast
    };
    enum { EMax = 32 };

  private:
    struct SPatchInfo {
        u32 iOffset;
        TPatchType iType;
    };
    struct SPatchInfo2 {
        u32 iOffset;
        u32 iValue;
    };
    struct SPatchInfo3 {
        u32 iLocation;
        u32 iAddress;
        u32 iRa;
        u32 iRb;
    };
    struct SPatchInfo4 {
        u32 iOffset;
        u16 iValue;
    };

  private:
    SPatchInfo iPatchInfo[EMax];
    u32 iCount;
    SPatchInfo2 iPatchInfo2[EMax * 2];
    u32 iCount2;
    SPatchInfo3 iPatchInfo3[EMax];
    u32 iCount3;
    SPatchInfo4 iPatchInfo4[EMax];
    u32 iCount4;
    u32 iSize;
    u32* iData;
    u32 iTrimSize;
    u32 iResultSize;
    jmp_buf iJumpBuf;
    u32 iSaveSize;
    CGbaWriter* iWriter;
    u8 iRamPage;

  private:
    CGbaPatcher();
    void Error(const char* anError);
    inline u32* Data(void) { return iData; };
    inline u16* Data16(void) { return (u16*)iData; };
    inline u8* Data8(void) { return (u8*)iData; };
    void Add(u32 anOffset, TPatchType aType);
    void Add2(u32 anOffset, u32 aValue);
    void Add3(u32 aLocation, u32 anAddress, u32 aRa, u32 aRb);
    void Add4(u32 anOffset, u16 aValue);
    void PatchInternal(void);
    // flash
    bool PatchSRAM(void);
    bool PatchEEPROM(void);
    bool PatchFLASH1M_V10X(void);
    bool PatchFLASH(void);
    void PatchEEPROM111(void);
    // sleep
    bool PatchSleep(void);
    // common sleep
    void CommonSleepSearch(void);
    void CommonSleepPatch(void);
    void SetTrimSize(void);
    // classic nes collection
    bool PatchNes(void);
    // cracks
    bool PatchDragonBallZ(void);
#ifdef __TEST
  private:
    void DumpPatchInfo(void);
#endif
  public:
    CGbaPatcher(u32 aSize, CGbaWriter* aWriter, u32* aData, u8 aRamPage);
    u32 Patch(void);
    inline u32 ResultSize(void) {
        return iResultSize;
    };
};