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