// ------------------------------
// Decompiled by Deathway
// Date : 2007-05-09
// ------------------------------
#include "stdafx.h"
#include "WZScriptEncode.h"

// GS-N 0.99.60T // 48c100 Status : Completed 2007-4-30
//	GS-N	1.00.18	JPN	0x004A7270	-	Completed

unsigned char XorTable[10]={0xA1, 0xB2, 0xAA, 0x12, 0x23, 0xF1, 0xF3, 0xD3, 0x78, 0x02}; // For some reason this is 16


CWZScriptEncode::CWZScriptEncode()
{
	this->m_pBuffer=0;
	this->m_iBufferSize=0;
	this->m_iBufferCount=0;
	this->m_hFile=INVALID_HANDLE_VALUE;
}

CWZScriptEncode::~CWZScriptEncode()
{
	this->Close();
}


int CWZScriptEncode::Open(char* filename)
{
	int filesize;
	char* pBuffer;
	int iReadFileSize;

	if ( this->m_hFile != INVALID_HANDLE_VALUE )
	{
		CloseHandle( this->m_hFile );
	}

	this->m_hFile = CreateFile(filename, GENERIC_READ, FILE_SHARE_READ, NULL, OPEN_EXISTING, 0x80 , NULL); // 0x80 == NORMAL

	if ( this->m_hFile == INVALID_HANDLE_VALUE )
	{
		return 0;
	}

	filesize = GetFileSize(this->m_hFile , NULL);
	pBuffer = (char*)malloc(filesize);
	ReadFile(this->m_hFile, (void*)pBuffer, filesize, (unsigned long*)&iReadFileSize, 0);

	if ( iReadFileSize == 0 )
	{
		this->Close();
		return 0;
	}

	this->Encode(pBuffer, iReadFileSize);
	this->SetScriptParsingBuffer(pBuffer, iReadFileSize);

	CloseHandle(this->m_hFile );
	return 1;
}

void CWZScriptEncode::Close()
{
	if ( this->m_pBuffer != 0)
	{
		free( this->m_pBuffer );
	}
}

void CWZScriptEncode::Encode(char* buffer, int size)
{
	int i;

	for (i=0;i<size; i++)
	{
		buffer[i]=buffer[i]^XorTable[i%10];	
	}
}

int CWZScriptEncode::GetC()
{
	if ( this->m_iBufferCount >= this->m_iBufferSize )
	{
		return -1;
	}
	else
	{
		return this->m_pBuffer [ this->m_iBufferCount++ ];
	}
}

void CWZScriptEncode::UnGetC(int ch)
{
	if ( this->m_iBufferCount <= 0)
	{
		return;
	}
	else
	{
		this->m_iBufferCount--;

		this->m_pBuffer [ this->m_iBufferCount ] = ch;
	}
}


void CWZScriptEncode::SetScriptParsingBuffer( char* buffer, int size)
{
	this->m_pBuffer = buffer;
	this->m_iBufferSize = size;
	this->m_iBufferCount = 0;
}

int CWZScriptEncode::GetNumber()
{
	return this->TokenNumber;
}

char* CWZScriptEncode::GetString()
{
	return &this->TokenString[0];
}

enum WZSMDToken CWZScriptEncode::GetToken()
{
	char ch;
	char* p;
	char TempString[100];

	this->TokenString[0]=0;	// Set String NULL

	do
	{
		if ( (ch=this->GetC ()) == (-1) )
		{
			return T_END;
		}
		else
		{	
			if (ch=='/')		// this is /
			{
				if((ch=this->GetC ()) == '/')
				{
					while ( ((ch=this->GetC()) != ('\n')) )	// End of Line (EOL)
					{
						continue;
					}
				}
			}
		}
	}
	while (isspace(ch) != 0);

	switch(ch)
	{

	case 0x23:	// #
		return this->CurrentToken=T_COMMAND;
		break;
	
	case 0x3B:	// ;
		return this->CurrentToken=T_SEMICOLON;
		break;
	
	case 0x2C:	// ,

		return this->CurrentToken=T_COMMA;
		break;
	
	case 0x7B:	// {
		return this->CurrentToken = T_LP;
		break;

	case 0x7D:	// }
		return this->CurrentToken = T_RP;
		break;
	// Nice Index Table :)	by Deathway
	case 0x2D:	//-
	case 0x2E:	//.
	case 0x30:	//0
	case 0x31:	//1
	case 0x32:	//2
	case 0x33:	//3
	case 0x34:	//4		
	case 0x35:	//5
	case 0x36:	//6
	case 0x37:	//7
	case 0x38:	//8
	case 0x39:	//9
		this->UnGetC (ch);
		p = TempString;
		
		while (((ch=this->GetC()) != -1) && ((ch == 0x2E) || (isdigit(ch) != 0) || (ch == 0x2D))) 
		{
			*p = ch;
			p++;
		}		
		*p = 0;
		this->TokenNumber = atof( TempString);
		return this->CurrentToken  = T_NUMBER ;
		break;

	case 0x22:	// "	
		p=&this->TokenString[0];
		
		while (((ch=this->GetC() ) != -1 ) && (ch != 0x22))	// nice
		{
			*p = ch;
			p++;
		}
		if (ch != 0x22 )
		{
			this->UnGetC(ch);
		}
		*p = 0;
		return this->CurrentToken = T_NAME;
		break;

	default:	
		if (isalpha(ch))
		{
			p=&this->TokenString[0];
			*p=ch;
			p++;

			while ( ((ch=this->GetC() ) != -1) && ( (ch == 0x2E) || (ch == 0x5F) || (isalnum(ch) != 0) ) )
			{
				*p=ch;
				p++;
				
			}

			this->UnGetC(ch);
			*p=0;
			this->CurrentToken=T_NAME;
			return T_NAME;
		}
		else
		{
			return this->CurrentToken = T_SMD_ERROR;
		}
		break;

	}

}