usbloadergx/source/cheats/gct.cpp

/*
 * gct.h
 * Class to handle Ocarina TXT Cheatfiles
 * nIxx
 */

#include <iostream>
#include <fstream>
#include <stdlib.h>
#include <string.h>
#include "gct.h"

#define ERRORRANGE "Error: CheatNr out of range"

GCTCheats::GCTCheats(void)
{
    iCntCheats = 0;
}

GCTCheats::~GCTCheats(void)
{

    string sGameID = "";
    string sGameTitle = "";
    /*string sCheatName[MAXCHEATS];
     string sCheats[MAXCHEATS];
     string sCheatComment[MAXCHEATS];*/
}

int GCTCheats::getCnt()
{
    return iCntCheats;
}

string GCTCheats::getGameName(void)
{
    return sGameTitle;
}

string GCTCheats::getGameID(void)
{
    return sGameID;
}

string GCTCheats::getCheat(int nr)
{
    if (nr <= (iCntCheats - 1))
    {
        return sCheats[nr];
    }
    else
    {
        return ERRORRANGE;
    }
}

string GCTCheats::getCheatName(int nr)
{
    if (nr <= (iCntCheats - 1))
    {
        return sCheatName[nr];
    }
    else
    {
        return ERRORRANGE;
    }
}

string GCTCheats::getCheatComment(int nr)
{
    if (nr <= (iCntCheats - 1))
    {
        return sCheatComment[nr];
    }
    else
    {
        return ERRORRANGE;
    }
}

int GCTCheats::createGCT(int nr, const char * filename)
{

    if (nr == 0) return 0;

    ofstream filestr;
    filestr.open(filename);

    if (filestr.fail()) return 0;

    //Header and Footer
    char header[] = { 0x00, 0xd0, 0xc0, 0xde, 0x00, 0xd0, 0xc0, 0xde };
    char footer[] = { 0xF0, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00 };

    string buf = getCheat(nr);
    filestr.write(header, sizeof(header));

    int x = 0;
    long int li;
    int len = buf.size();

    while (x < len)
    {
        string temp = buf.substr(x, 2);
        li = strtol(temp.c_str(), NULL, 16);
        temp = li;
        filestr.write(temp.c_str(), 1);
        x += 2;
    }
    filestr.write(footer, sizeof(footer));

    filestr.close();
    return 1;
}

int GCTCheats::createGCT(const char * chtbuffer, const char * filename)
{

    ofstream filestr;
    filestr.open(filename);

    if (filestr.fail()) return 0;

    //Header and Footer
    char header[] = { 0x00, 0xd0, 0xc0, 0xde, 0x00, 0xd0, 0xc0, 0xde };
    char footer[] = { 0xF0, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00 };

    string buf = chtbuffer;
    filestr.write(header, sizeof(header));

    int x = 0;
    long int li;
    int len = buf.size();

    while (x < len)
    {
        string temp = buf.substr(x, 2);
        li = strtol(temp.c_str(), NULL, 16);
        temp = li;
        filestr.write(temp.c_str(), 1);
        x += 2;
    }

    filestr.write(footer, sizeof(footer));

    filestr.close();

    return 1;
}

int GCTCheats::createGCT(int nr[], int cnt, const char * filename)
{

    if (cnt == 0) return 0;

    ofstream filestr;
    filestr.open(filename);

    if (filestr.fail()) return 0;

    //Header and Footer
    char header[] = { 0x00, 0xd0, 0xc0, 0xde, 0x00, 0xd0, 0xc0, 0xde };
    char footer[] = { 0xF0, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00 };

    filestr.write(header, sizeof(header));

    int c = 0;
    while (c != cnt)
    {
        int actnr = nr[c];
        string buf = getCheat(actnr);
        long int li;
        int len = buf.size();
        int x = 0;

        while (x < len)
        {
            string temp = buf.substr(x, 2);
            li = strtol(temp.c_str(), NULL, 16);
            temp = li;
            filestr.write(temp.c_str(), 1);
            x += 2;
        }
        c++;
    }

    filestr.write(footer, sizeof(footer));
    filestr.close();
    return 1;
}

int GCTCheats::openTxtfile(const char * filename)
{
    ifstream filestr;
    int i = 0;
    string str;
    filestr.open(filename);

    if (filestr.fail()) return 0;

    filestr.seekg(0, ios_base::end);
    int size = filestr.tellg();
    if (size <= 0) return -1;
    filestr.seekg(0, ios_base::beg);

    getline(filestr, sGameID);
    if (sGameID[sGameID.length() - 1] == '\r') sGameID.erase(sGameID.length() - 1);

    getline(filestr, sGameTitle);
    if (sGameTitle[sGameTitle.length() - 1] == '\r') sGameTitle.erase(sGameTitle.length() - 1);

    getline(filestr, sCheatName[i]); // skip first line if file uses CRLF
    if (!sGameTitle[sGameTitle.length() - 1] == '\r') filestr.seekg(0, ios_base::beg);

    while (!filestr.eof())
    {
        getline(filestr, sCheatName[i]); // '\n' delimiter by default
        if (sCheatName[i][sCheatName[i].length() - 1] == '\r') sCheatName[i].erase(sCheatName[i].length() - 1);

        string cheatdata;
        bool emptyline = false;

        do
        {
            getline(filestr, str);
            if (str[str.length() - 1] == '\r') str.erase(str.length() - 1);

            if (str == "" || str[0] == '\r' || str[0] == '\n')
            {
                emptyline = true;
                break;
            }

            if (IsCode(str))
            {
                // remove any garbage (comment) after code
                while (str.size() > 17)
                {
                    str.erase(str.length() - 1);
                }
                cheatdata.append(str);
                size_t found = cheatdata.find(' ');
                cheatdata.replace(found, 1, "");
            }
            else
            {
                //printf("%i",str.size());
                sCheatComment[i] = str;
            }
            if (filestr.eof()) break;

        } while (!emptyline);

        sCheats[i] = cheatdata;
        i++;
        if (i == MAXCHEATS) break;
    }
    iCntCheats = i;
    filestr.close();
    return 1;
}

bool GCTCheats::IsCode(const std::string& str)
{
    if (str[8] == ' ' && str.size() >= 17)
    {
        // accept strings longer than 17 in case there is a comment on the same line as the code
        char part1[9];
        char part2[9];
        snprintf(part1, sizeof(part1), "%c%c%c%c%c%c%c%c", str[0], str[1], str[2], str[3], str[4], str[5], str[6],
                str[7]);
        snprintf(part2, sizeof(part2), "%c%c%c%c%c%c%c%c", str[9], str[10], str[11], str[12], str[13], str[14],
                str[15], str[16]);
        if ((strtok(part1, "0123456789ABCDEFabcdef") == NULL) && (strtok(part2, "0123456789ABCDEFabcdef") == NULL))
        {
            return true;
        }
    }
    return false;
}