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;
}