state("pcsx2-qt"){}

state("pcsx2"){}

startup
{
    vars.modBase = IntPtr.Zero;
    vars.EEBase  = IntPtr.Zero;
    vars.loading = (byte)0;
    vars.DEBUG = false;
    vars.lastLoading = (byte)255;
}

init
{
    System.Diagnostics.Process p = game;

    vars.modBase = IntPtr.Zero;
    for (int i = 0; i < modules.Count(); i++)
    {
        string n = modules[i].ModuleName.ToLower();
        if (n.StartsWith("pcsx2"))
        {
            vars.modBase = modules[i].BaseAddress;
            break;
        }
    }

    if ((IntPtr)vars.modBase == IntPtr.Zero)
        return false;

    vars.EEBase = IntPtr.Zero;

    IntPtr baseAddr = (IntPtr)vars.modBase;

    int e_lfanew = p.ReadValue<int>(IntPtr.Add(baseAddr, 0x3C));
    if (e_lfanew > 0)
    {
        IntPtr nt = IntPtr.Add(baseAddr, e_lfanew);
        int sig = p.ReadValue<int>(nt);

        if (sig == 0x00004550)
        {
            short magic = p.ReadValue<short>(IntPtr.Add(nt, 0x18));
            int dataDir = (magic == 0x10B) ? 0x60 : 0x70;

            int exportRva  = p.ReadValue<int>(IntPtr.Add(IntPtr.Add(nt, 0x18), dataDir + 0));
            int exportSize = p.ReadValue<int>(IntPtr.Add(IntPtr.Add(nt, 0x18), dataDir + 4));

            if (exportRva != 0 && exportSize != 0)
            {
                IntPtr exportDir = IntPtr.Add(baseAddr, exportRva);

                int numberOfNames      = p.ReadValue<int>(IntPtr.Add(exportDir, 0x18));
                int addressOfFunctions = p.ReadValue<int>(IntPtr.Add(exportDir, 0x1C));
                int addressOfNames     = p.ReadValue<int>(IntPtr.Add(exportDir, 0x20));
                int addressOfOrdinals  = p.ReadValue<int>(IntPtr.Add(exportDir, 0x24));

                for (int i = 0; i < numberOfNames; i++)
                {
                    int nameRva = p.ReadValue<int>(IntPtr.Add(IntPtr.Add(baseAddr, addressOfNames), i * 4));
                    byte[] b = p.ReadBytes(IntPtr.Add(baseAddr, nameRva), 8);

                    if (b.Length >= 6 &&
                        b[0] == 0x45 && b[1] == 0x45 && b[2] == 0x6D &&
                        b[3] == 0x65 && b[4] == 0x6D && b[5] == 0x00)
                    {
                        short ordinalIndex = p.ReadValue<short>(IntPtr.Add(IntPtr.Add(baseAddr, addressOfOrdinals), i * 2));
                        int funcRva = p.ReadValue<int>(IntPtr.Add(IntPtr.Add(baseAddr, addressOfFunctions), ordinalIndex * 4));
                        IntPtr exportAddr = IntPtr.Add(baseAddr, funcRva);

                        vars.EEBase = p.ReadPointer(exportAddr);
                        break;
                    }
                }
            }
        }
    }

    return true;
}

update
{
    System.Diagnostics.Process p = game;

    IntPtr ee = (IntPtr)vars.EEBase;

    if (ee == IntPtr.Zero)
    {
        return;
    }

    IntPtr hostFlagAddr = IntPtr.Add(ee, 0x43AADC); //Load Timer Pauses on Loads, Model Loads.
    byte v = p.ReadValue<byte>(hostFlagAddr);

    vars.loading = v;

    if ((bool)vars.DEBUG && v != (byte)vars.lastLoading)
    {
        print("EEBase       = 0x" + ee.ToString("X"));
        print("HostFlagAddr = 0x" + hostFlagAddr.ToString("X"));
        print("Loading Value= " + v);
        vars.lastLoading = v;
    }
}

isLoading
{
    return vars.loading != 0;
}