// Copyright 2009 Dolphin Emulator Project
// Copyright 2005 Duddie, wntrmute, Hermes
// Licensed under GPLv2+
// Refer to the license.txt file included.

#include <cstdio>
#include <cstdlib>
#include <fstream>
#include <iostream>
#include <map>
#include <string>

#include "Common/CommonTypes.h"
#include "Common/FileUtil.h"

#include "Core/DSP/DSPAssembler.h"
#include "Core/DSP/DSPDisassembler.h"
#include "Core/DSP/DSPInterpreter.h"
#include "Core/DSP/DSPTables.h"

static const char *err_string[] =
{
	"",
	"Unknown Error",
	"Unknown opcode",
	"Not enough parameters",
	"Too many parameters",
	"Wrong parameter",
	"Expected parameter of type 'string'",
	"Expected parameter of type 'value'",
	"Expected parameter of type 'register'",
	"Expected parameter of type 'memory pointer'",
	"Expected parameter of type 'immediate'",
	"Incorrect binary value",
	"Incorrect hexadecimal value",
	"Incorrect decimal value",
	"Label already exists",
	"Label not defined",
	"No matching brackets",
	"This opcode cannot be extended",
	"Given extending params for non extensible opcode",
	"Wrong parameter: must be accumulator register",
	"Wrong parameter: must be mid accumulator register",
	"Invalid register",
	"Number out of range"
};

DSPAssembler::DSPAssembler(const AssemblerSettings &settings) :
	gdg_buffer(nullptr),
	m_cur_addr(0),
	m_cur_pass(0),
	m_current_param(0),
	settings_(settings)

{
}

DSPAssembler::~DSPAssembler()
{
	if (gdg_buffer)
		free(gdg_buffer);
}

bool DSPAssembler::Assemble(const std::string& text, std::vector<u16> &code, std::vector<int> *line_numbers)
{
	if (line_numbers)
		line_numbers->clear();
	const char *fname = "tmp.asm";
	if (!File::WriteStringToFile(text, fname))
		return false;
	InitPass(1);
	if (!AssembleFile(fname, 1))
		return false;

	// We now have the size of the output buffer
	if (m_totalSize > 0)
	{
		gdg_buffer = (char *)malloc(m_totalSize * sizeof(u16) + 4);
		if (!gdg_buffer)
			return false;

		memset(gdg_buffer, 0, m_totalSize * sizeof(u16));
	} else
		return false;

	InitPass(2);
	if (!AssembleFile(fname, 2))
		return false;

	code.resize(m_totalSize);
	for (int i = 0; i < m_totalSize; i++)
	{
		code[i] = *(u16 *)(gdg_buffer + i * 2);
	}

	if (gdg_buffer)
	{
		free(gdg_buffer);
		gdg_buffer = nullptr;
	}

	last_error_str = "(no errors)";
	last_error = ERR_OK;

	return true;
}

void DSPAssembler::ShowError(err_t err_code, const char *extra_info)
{

	if (!settings_.force)
		failed = true;

	char error_buffer[1024];
	char *buf_ptr = error_buffer;
	buf_ptr += sprintf(buf_ptr, "%i : %s ", code_line, cur_line.c_str());
	if (!extra_info)
		extra_info = "-";

	if (m_current_param == 0)
		buf_ptr += sprintf(buf_ptr, "ERROR: %s Line: %d : %s\n", err_string[err_code], code_line, extra_info);
	else
		buf_ptr += sprintf(buf_ptr, "ERROR: %s Line: %d Param: %d : %s\n",
						   err_string[err_code], code_line, m_current_param, extra_info);
	last_error_str = error_buffer;
	last_error = err_code;
}

static char *skip_spaces(char *ptr)
{
	while (*ptr == ' ')
		ptr++;
	return ptr;
}

// Parse a standalone value - it can be a number in one of several formats or a label.
s32 DSPAssembler::ParseValue(const char *str)
{
	bool negative = false;
	s32 val = 0;
	const char *ptr = str;

	if (ptr[0] == '#')
	{
		ptr++;
		negative = true;  // Wow! Double # (needed one to get in here) negates???
	}
	if (ptr[0] == '-')
	{
		ptr++;
		negative = true;
	}
	if (ptr[0] == '0')
	{
		if (ptr[1] >= '0' && ptr[1] <= '9')
		{
			for (int i = 0; ptr[i] != 0; i++)
			{
				val *= 10;
				if (ptr[i] >= '0' && ptr[i] <= '9')
					val += ptr[i] - '0';
				else
					ShowError(ERR_INCORRECT_DEC, str);
			}
		}
		else
		{
			switch (ptr[1])
			{
			case 'X': // hex
				for (int i = 2 ; ptr[i] != 0 ; i++)
				{
					val <<= 4;
					if (ptr[i] >= 'a' && ptr[i] <= 'f')
						val += (ptr[i]-'a'+10);
					else if (ptr[i] >= 'A' && ptr[i] <= 'F')
						val += (ptr[i]-'A'+10);
					else if (ptr[i] >= '0' && ptr[i] <= '9')
						val += (ptr[i] - '0');
					else
						ShowError(ERR_INCORRECT_HEX, str);
				}
				break;
			case '\'': // binary
				for (int i = 2; ptr[i] != 0; i++)
				{
					val *=2;
					if (ptr[i] >= '0' && ptr[i] <= '1')
						val += ptr[i] - '0';
					else
						ShowError(ERR_INCORRECT_BIN, str);
				}
				break;
			default:
				// value is 0 or error
				val = 0;
				break;
			}
		}
	}
	else
	{
		// Symbol starts with a digit - it's a dec number.
		if (ptr[0] >= '0' && ptr[0] <= '9')
		{
			for (int i = 0; ptr[i] != 0; i++)
			{
				val *= 10;
				if (ptr[i] >= '0' && ptr[i] <= '9')
					val += ptr[i] - '0';
				else
					ShowError(ERR_INCORRECT_DEC, str);
			}
		}
		else  // Everything else is a label.
		{
			// Lookup label
			u16 value;
			if (labels.GetLabelValue(ptr, &value))
				return value;
			if (m_cur_pass == 2)
				ShowError(ERR_UNKNOWN_LABEL, str);
		}
	}
	if (negative)
		return -val;
	return val;
}

// This function splits the given src string into three parts:
// - Text before the first opening ('(') parenthesis
// - Text within the first and last opening ('(') and closing (')') parentheses.
// - If text follows after these parentheses, then this is what is returned from the function.
//
// Note that the first opening parenthesis and the last closing parenthesis are discarded from the string.
// For example: Say "Test (string) 1234" is the string passed in as src.
//
// - src will become "Test "
// - dst will become "string"
// - Returned string from the function will be " 1234"
//
char *DSPAssembler::FindBrackets(char *src, char *dst)
{
	s32 len = (s32) strlen(src);
	s32 first = -1;
	s32 count = 0;
	s32 i, j;
	j = 0;
	for (i = 0 ; i < len ; i++)
	{
		if (src[i] == '(')
		{
			if (first < 0)
			{
				count = 1;
				src[i] = 0x0;
				first = i;
			}
			else
			{
				count++;
				dst[j++] = src[i];
			}
		}
		else if (src[i] == ')')
		{
			if (--count == 0)
			{
				dst[j] = 0;
				return &src[i+1];
			}
			else
			{
				dst[j++] = src[i];
			}
		}
		else
		{
			if (first >= 0)
				dst[j++] = src[i];
		}
	}
	if (count)
		ShowError(ERR_NO_MATCHING_BRACKETS);
	return nullptr;
}

// Bizarre in-place expression evaluator.
u32 DSPAssembler::ParseExpression(const char *ptr)
{
	char *pbuf;
	s32 val = 0;

	char *d_buffer = (char *)malloc(1024);
	char *s_buffer = (char *)malloc(1024);
	strcpy(s_buffer, ptr);

	while ((pbuf = FindBrackets(s_buffer, d_buffer)) != nullptr)
	{
		val = ParseExpression(d_buffer);
		sprintf(d_buffer, "%s%d%s", s_buffer, val, pbuf);
		strcpy(s_buffer, d_buffer);
	}

	int j = 0;
	for (int i = 0; i < ((s32)strlen(s_buffer) + 1) ; i++)
	{
		char c = s_buffer[i];
		if (c != ' ')
			d_buffer[j++] = c;
	}

	for (int i = 0; i < ((s32)strlen(d_buffer) + 1) ; i++)
	{
		char c = d_buffer[i];
		if (c == '-')
		{
			if (i == 0)
				c = '#';
			else
			{
				switch (d_buffer[i - 1])
				{
				case '/':
				case '%':
				case '*':
					c = '#';
				}
			}
		}
		d_buffer[i] = c;
	}

	while ((pbuf = strstr(d_buffer, "+")) != nullptr)
	{
		*pbuf = 0x0;
		val = ParseExpression(d_buffer) + ParseExpression(pbuf+1);
		sprintf(d_buffer, "%d", val);
	}

	while ((pbuf = strstr(d_buffer, "-")) != nullptr)
	{
		*pbuf = 0x0;
		val = ParseExpression(d_buffer) - ParseExpression(pbuf+1);
		if (val < 0)
		{
			val = 0x10000 + (val & 0xffff); // ATTENTION: avoid a terrible bug!!! number cannot write with '-' in sprintf
			fprintf(stderr, "WARNING: Number Underflow at Line: %d \n", code_line);
		}
		sprintf(d_buffer, "%d", val);
	}

	while ((pbuf = strstr(d_buffer, "*")) != nullptr)
	{
		*pbuf = 0x0;
		val = ParseExpression(d_buffer) * ParseExpression(pbuf+1);
		sprintf(d_buffer, "%d", val);
	}

	while ((pbuf = strstr(d_buffer, "/")) != nullptr)
	{
		*pbuf = 0x0;
		val = ParseExpression(d_buffer) / ParseExpression(pbuf+1);
		sprintf(d_buffer, "%d", val);
	}

	while ((pbuf = strstr(d_buffer, "|")) != nullptr)
	{
		*pbuf = 0x0;
		val = ParseExpression(d_buffer) | ParseExpression(pbuf+1);
		sprintf(d_buffer, "%d", val);
	}

	while ((pbuf = strstr(d_buffer, "&")) != nullptr)
	{
		*pbuf = 0x0;
		val = ParseExpression(d_buffer) & ParseExpression(pbuf+1);
		sprintf(d_buffer, "%d", val);
	}

	val = ParseValue(d_buffer);
	free(d_buffer);
	free(s_buffer);
	return val;
}

// Destroys parstr
u32 DSPAssembler::GetParams(char *parstr, param_t *par)
{
	u32 count = 0;
	char *tmpstr = skip_spaces(parstr);
	tmpstr = strtok(tmpstr, ",\x00");
	for (int i = 0; i < 10; i++)
	{
		if (tmpstr == nullptr)
			break;
		tmpstr = skip_spaces(tmpstr);
		if (strlen(tmpstr) == 0)
			break;
		if (tmpstr)
			count++;
		else
			break;

		par[i].type = P_NONE;
		switch (tmpstr[0])
		{
		case '"':
			par[i].str = strtok(tmpstr, "\"");
			par[i].type = P_STR;
			break;
		case '#':
			par[i].val = ParseExpression(tmpstr + 1);
			par[i].type = P_IMM;
			break;
		case '@':
			if (tmpstr[1] == '$')
			{
				par[i].val = ParseExpression(tmpstr + 2);
				par[i].type = P_PRG;
			}
			else
			{
				par[i].val = ParseExpression(tmpstr + 1);
				par[i].type = P_MEM;
			}
			break;
		case '$':
			par[i].val = ParseExpression(tmpstr + 1);
			par[i].type = P_REG;
			break;

		default:
			par[i].val = ParseExpression(tmpstr);
			par[i].type = P_VAL;
			break;
		}
		tmpstr = strtok(nullptr, ",\x00");
	}
	return count;
}

const opc_t *DSPAssembler::FindOpcode(const char *opcode, u32 par_count, const opc_t * const opcod, int opcod_size)
{
	if (opcode[0] == 'C' && opcode[1] == 'W')
		return &cw;

	AliasMap::const_iterator alias_iter = aliases.find(opcode);
	if (alias_iter != aliases.end())
		opcode = alias_iter->second.c_str();
	for (int i = 0; i < opcod_size; i++)
	{
		const opc_t *opc = &opcod[i];
		if (strcmp(opc->name, opcode) == 0)
		{
			if (par_count < opc->param_count)
			{
				ShowError(ERR_NOT_ENOUGH_PARAMETERS);
			}
			if (par_count > opc->param_count)
			{
				ShowError(ERR_TOO_MANY_PARAMETERS);
			}
			return opc;
		}
	}
	ShowError(ERR_UNKNOWN_OPCODE);
	return nullptr;
}

// weird...
static u16 get_mask_shifted_down(u16 mask)
{
	while (!(mask & 1))
		mask >>= 1;
	return mask;
}

bool DSPAssembler::VerifyParams(const opc_t *opc, param_t *par, int count, bool ext)
{
	for (int i = 0; i < count; i++)
	{
		const int current_param = i + 1;  // just for display.
		if (opc->params[i].type != par[i].type || (par[i].type & P_REG))
		{
			if (par[i].type == P_VAL &&
				(opc->params[i].type == P_ADDR_I || opc->params[i].type == P_ADDR_D))
			{
				// Data and instruction addresses are valid as VAL values.
				continue;
			}

			if ((opc->params[i].type & P_REG) && (par[i].type & P_REG))
			{
				// Just a temp. Should be replaced with more purposeful vars.
				int value;

				// modified by Hermes: test the register range
				switch ((unsigned)opc->params[i].type)
				{
				case P_REG18:
				case P_REG19:
				case P_REG1A:
					value = (opc->params[i].type >> 8) & 31;
					if ((int)par[i].val < value ||
						(int)par[i].val > value + get_mask_shifted_down(opc->params[i].mask))
					{
						if (ext)
							fprintf(stderr, "(ext) ");

						fprintf(stderr, "%s   (param %i)", cur_line.c_str(), current_param);
						ShowError(ERR_INVALID_REGISTER);
					}
					break;
				case P_PRG:
					if ((int)par[i].val < 0 || (int)par[i].val > 0x3)
					{
						if (ext)
							fprintf(stderr, "(ext) ");

						fprintf(stderr, "%s   (param %i)", cur_line.c_str(), current_param);
						ShowError(ERR_INVALID_REGISTER);
					}
					break;
				case P_ACC:
					if ((int)par[i].val < 0x20 || (int)par[i].val > 0x21)
					{
						if (ext)
							fprintf(stderr, "(ext) ");

						if (par[i].val >= 0x1e && par[i].val <= 0x1f)
						{
							fprintf(stderr, "%i : %s ", code_line, cur_line.c_str());
							fprintf(stderr, "WARNING: $ACM%d register used instead of $ACC%d register Line: %d Param: %d Ext: %d\n",
								(par[i].val & 1), (par[i].val & 1), code_line, current_param, ext);
						}
						else if (par[i].val >= 0x1c && par[i].val <= 0x1d)
						{
							fprintf(stderr, "WARNING: $ACL%d register used instead of $ACC%d register Line: %d Param: %d\n",
								(par[i].val & 1), (par[i].val & 1), code_line, current_param);
						}
						else
						{
							ShowError(ERR_WRONG_PARAMETER_ACC);
						}
					}
					break;
				case P_ACCM:
					if ((int)par[i].val < 0x1e || (int)par[i].val > 0x1f)
					{
						if (ext)
							fprintf(stderr, "(ext) ");

						if (par[i].val >= 0x1c && par[i].val <= 0x1d)
						{
							fprintf(stderr, "WARNING: $ACL%d register used instead of $ACM%d register Line: %d Param: %d\n",
								(par[i].val & 1), (par[i].val & 1), code_line, current_param);
						}
						else if (par[i].val >= 0x20 && par[i].val <= 0x21)
						{
							fprintf(stderr, "WARNING: $ACC%d register used instead of $ACM%d register Line: %d Param: %d\n",
								(par[i].val & 1), (par[i].val & 1), code_line, current_param);
						}
						else
						{
							ShowError(ERR_WRONG_PARAMETER_ACC);
						}
					}
					break;

				case P_ACCL:
					if ((int)par[i].val < 0x1c || (int)par[i].val > 0x1d)
					{
						if (ext)
							fprintf(stderr, "(ext) ");

						if (par[i].val >= 0x1e && par[i].val <= 0x1f)
						{
							fprintf(stderr, "%s ", cur_line.c_str());
							fprintf(stderr, "WARNING: $ACM%d register used instead of $ACL%d register Line: %d Param: %d\n",
								(par[i].val & 1), (par[i].val & 1), code_line, current_param);
						}
						else if (par[i].val >= 0x20 && par[i].val <= 0x21)
						{
							fprintf(stderr, "%s ", cur_line.c_str());
							fprintf(stderr, "WARNING: $ACC%d register used instead of $ACL%d register Line: %d Param: %d\n",
								(par[i].val & 1), (par[i].val & 1), code_line, current_param);
						}
						else
						{
							ShowError(ERR_WRONG_PARAMETER_ACC);
						}
					}
					break;
/*				case P_ACCM_D: //P_ACC_MID:
					if ((int)par[i].val < 0x1e || (int)par[i].val > 0x1f)
					{
						ShowError(ERR_WRONG_PARAMETER_MID_ACC);
					}
					break;*/
				}
				continue;
			}

			switch (par[i].type & (P_REG | 7))
			{
			case P_REG:
				if (ext) fprintf(stderr, "(ext) ");
				ShowError(ERR_EXPECTED_PARAM_REG);
				break;
			case P_MEM:
				if (ext) fprintf(stderr, "(ext) ");
				ShowError(ERR_EXPECTED_PARAM_MEM);
				break;
			case P_VAL:
				if (ext) fprintf(stderr, "(ext) ");
				ShowError(ERR_EXPECTED_PARAM_VAL);
				break;
			case P_IMM:
				if (ext) fprintf(stderr, "(ext) ");
				ShowError(ERR_EXPECTED_PARAM_IMM);
				break;
			}
			ShowError(ERR_WRONG_PARAMETER);
			break;
		}
		else if ((opc->params[i].type & 3) != 0 && (par[i].type & 3) != 0)
		{
			// modified by Hermes: test NUMBER range
			int value = get_mask_shifted_down(opc->params[i].mask);
			unsigned int valueu = 0xffff & ~(value >> 1);
			if ((int)par[i].val < 0)
			{
				if (value == 7) // value 7 por sbclr/sbset
				{
					fprintf(stderr,"Value must be from 0x0 to 0x%x\n", value);
					ShowError(ERR_OUT_RANGE_NUMBER);
				}
				else if (opc->params[i].type == P_MEM)
				{
					if (value < 256)
						fprintf(stderr, "Address value must be from 0x%x to 0x%x\n",valueu, (value>>1));
					else
						fprintf(stderr, "Address value must be from 0x0 to 0x%x\n", value);

					ShowError(ERR_OUT_RANGE_NUMBER);
				}
				else if ((int)par[i].val < -((value >> 1) + 1))
				{
					if (value < 128)
						fprintf(stderr, "Value must be from -0x%x to 0x%x, is %i\n",
						        (value >> 1) + 1, value >> 1, par[i].val);
					else
						fprintf(stderr, "Value must be from -0x%x to 0x%x or 0x0 to 0x%x, is %i\n",
						        (value >> 1) + 1, value >> 1, value, par[i].val);

					ShowError(ERR_OUT_RANGE_NUMBER);
				}
			}
			else
			{
				if (value == 7) // value 7 por sbclr/sbset
				{
					if (par[i].val > (unsigned)value)
					{
						fprintf(stderr,"Value must be from 0x%x to 0x%x, is %i\n",valueu, value, par[i].val);
						ShowError(ERR_OUT_RANGE_NUMBER);
					}
				}
				else if (opc->params[i].type == P_MEM)
				{
					if (value < 256)
						value >>= 1; // addressing 8 bit with sign
					if (par[i].val > (unsigned)value &&
						(par[i].val < valueu || par[i].val > (unsigned)0xffff))
					{
						if (value < 256)
							fprintf(stderr,"Address value must be from 0x%x to 0x%x, is %04x\n", valueu, value, par[i].val);
						else
							fprintf(stderr,"Address value must be minor of 0x%x\n", value+1);
						ShowError(ERR_OUT_RANGE_NUMBER);
					}
				}
				else
				{
					if (value < 128)
						value >>= 1; // special case ASL/ASR/LSL/LSR
					if (par[i].val > (unsigned)value)
					{
						if (value < 64)
							fprintf(stderr,"Value must be from -0x%x to 0x%x, is %i\n", (value + 1), value, par[i].val);
						else
							fprintf(stderr,"Value must be minor of 0x%x, is %i\n", value + 1, par[i].val);
						ShowError(ERR_OUT_RANGE_NUMBER);
					}
				}
			}
			continue;
		}
	}
	m_current_param = 0;
	return true;
}


// Merge opcode with params.
void DSPAssembler::BuildCode(const opc_t *opc, param_t *par, u32 par_count, u16 *outbuf)
{
	outbuf[m_cur_addr] |= opc->opcode;
	for (u32 i = 0; i < par_count; i++)
	{
		// Ignore the "reverse" parameters since they are implicit.
		if (opc->params[i].type != P_ACC_D && opc->params[i].type != P_ACCM_D)
		{
			u16 t16 = outbuf[m_cur_addr + opc->params[i].loc];
			u16 v16 = par[i].val;
			if (opc->params[i].lshift > 0)
				v16 <<= opc->params[i].lshift;
			else
				v16 >>= -opc->params[i].lshift;
			v16 &= opc->params[i].mask;
			outbuf[m_cur_addr + opc->params[i].loc] = t16 | v16;
		}
	}
}

void DSPAssembler::InitPass(int pass)
{
	failed = false;
	if (pass == 1)
	{
		// Reset label table. Pre-populate with hw addresses and registers.
		labels.Clear();
		labels.RegisterDefaults();
		aliases.clear();
		aliases["S15"] = "SET15";
		aliases["S16"] = "SET16";
		aliases["S40"] = "SET40";
	}
	m_cur_addr = 0;
	m_totalSize = 0;
	cur_segment = SEGMENT_CODE;
	segment_addr[SEGMENT_CODE] = 0;
	segment_addr[SEGMENT_DATA] = 0;
	segment_addr[SEGMENT_OVERLAY] = 0;
}

bool DSPAssembler::AssembleFile(const char *fname, int pass)
{
	int disable_text = 0; // modified by Hermes

	std::ifstream fsrc;
	OpenFStream(fsrc, fname, std::ios_base::in);

	if (fsrc.fail())
	{
		std::cerr << "Cannot open file " << fname << std::endl;
		return false;
	}

	//printf("%s: Pass %d\n", fname, pass);
	code_line = 0;
	m_cur_pass = pass;

#define LINEBUF_SIZE 1024
	char line[LINEBUF_SIZE] = {0};
	while (!failed && !fsrc.fail() && !fsrc.eof())
	{
		int opcode_size = 0;
		fsrc.getline(line, LINEBUF_SIZE);
		if (fsrc.fail())
			break;

		cur_line = line;
		//printf("A: %s\n", line);
		code_line++;

		param_t params[10] = {{0, P_NONE, nullptr}};
		param_t params_ext[10] = {{0, P_NONE, nullptr}};

		bool upper = true;
		for (int i = 0; i < LINEBUF_SIZE; i++)
		{
			char c = line[i];
			// This stuff handles /**/ and // comments.
			// modified by Hermes : added // and /* */ for long commentaries
			if (c == '/')
			{
				if (i < 1023)
				{
					if (line[i+1] == '/')
						c = 0x00;
					else if (line[i+1] == '*')
					{
						// toggle comment mode.
						disable_text = !disable_text;
					}
				}
			}
			else if (c == '*')
			{
				if (i < 1023 && line[i+1] == '/' && disable_text)
				{
					disable_text = 0;
					c = 32;
					line[i + 1] = 32;
				}
			}

			// turn text into spaces if disable_text is on (in a comment).
			if (disable_text && ((unsigned char)c) > 32) c = 32;

			if (c == 0x0a || c == 0x0d || c == ';')
				c = 0x00;
			if (c == 0x09) // tabs to spaces
				c = ' ';
			if (c == '"')
				upper = !upper;
			if (upper && c >= 'a' && c <= 'z') // convert to uppercase
				c = c - 'a' + 'A';
			line[i] = c;
			if (c == 0)
				break; // modified by Hermes
		}
		char *ptr = line;

		std::string label;

		size_t col_pos = std::string(line).find(":");
		if (col_pos != std::string::npos)
		{
			bool valid = true;

			for (int j = 0; j < (int)col_pos; j++)
			{
				if (j == 0)
					if (!((ptr[j] >= 'A' && ptr[j] <= 'Z') || (ptr[j] == '_')))
						valid = false;
				if (!((ptr[j] >= '0' && ptr[j] <= '9') || (ptr[j] >= 'A' && ptr[j] <= 'Z') || (ptr[j] == '_')))
					valid = false;
			}
			if (valid)
			{
				label = std::string(line).substr(0, col_pos);
				ptr += col_pos + 1;
			}
		}

		char* opcode = strtok(ptr, " ");
		char* opcode_ext = nullptr;

		u32 params_count = 0;
		u32 params_count_ext = 0;
		if (opcode)
		{
			if ((opcode_ext = strstr(opcode, "'")) != nullptr)
			{
				opcode_ext[0] = '\0';
				opcode_ext++;
				if (strlen(opcode_ext) == 0)
					opcode_ext = nullptr;
			}
			// now we have opcode and label

			params_count = 0;
			params_count_ext = 0;

			char *paramstr = strtok(nullptr, "\0");
			char *paramstr_ext = nullptr;
			// there is valid opcode so probably we have parameters

			if (paramstr)
			{
				if ((paramstr_ext = strstr(paramstr, ":")) != nullptr)
				{
					paramstr_ext[0] = '\0';
					paramstr_ext++;
				}
			}

			if (paramstr)
				params_count = GetParams(paramstr, params);
			if (paramstr_ext)
				params_count_ext = GetParams(paramstr_ext, params_ext);
		}

		if (!label.empty())
		{
			// there is a valid label so lets store it in labels table
			u32 lval = m_cur_addr;
			if (opcode)
			{
				if (strcmp(opcode, "EQU") == 0)
				{
					lval = params[0].val;
					opcode = nullptr;
				}
			}
			if (pass == 1)
				labels.RegisterLabel(label, lval);
		}

		if (opcode == nullptr)
			continue;

		// check if opcode is reserved compiler word
		if (strcmp("INCLUDE", opcode) == 0)
		{
			if (params[0].type == P_STR)
			{
				char *tmpstr;
				u32 thisCodeline = code_line;

				if (include_dir.size())
				{
					tmpstr = (char *)malloc(include_dir.size() + strlen(params[0].str) + 2);
					sprintf(tmpstr, "%s/%s", include_dir.c_str(), params[0].str);
				}
				else
				{
					tmpstr = (char *)malloc(strlen(params[0].str) + 1);
					strcpy(tmpstr, params[0].str);
				}

				AssembleFile(tmpstr, pass);

				code_line = thisCodeline;

				free(tmpstr);
			}
			else
				ShowError(ERR_EXPECTED_PARAM_STR);
			continue;
		}

		if (strcmp("INCDIR", opcode) == 0)
		{
			if (params[0].type == P_STR)
				include_dir = params[0].str;
			else
				ShowError(ERR_EXPECTED_PARAM_STR);
			continue;
		}

		if (strcmp("ORG", opcode) == 0)
		{
			if (params[0].type == P_VAL)
				m_cur_addr = params[0].val;
			else
				ShowError(ERR_EXPECTED_PARAM_VAL);
			continue;
		}

		if (strcmp("SEGMENT", opcode) == 0)
		{
			if (params[0].type == P_STR)
			{
				segment_addr[cur_segment] = m_cur_addr;
				if (strcmp("DATA", params[0].str) == 0)
					cur_segment = SEGMENT_DATA;
				if (strcmp("CODE", params[0].str) == 0)
					cur_segment = SEGMENT_CODE;
				m_cur_addr = segment_addr[cur_segment];
			}
			else
				ShowError(ERR_EXPECTED_PARAM_STR);
			continue;
		}

		const opc_t *opc = FindOpcode(opcode, params_count, opcodes, opcodes_size);
		if (!opc)
			opc = &cw;

		opcode_size = opc->size;

		VerifyParams(opc, params, params_count);

		const opc_t *opc_ext = nullptr;
		// Check for opcode extensions.
		if (opc->extended)
		{
			if (opcode_ext)
			{
				opc_ext = FindOpcode(opcode_ext, params_count_ext, opcodes_ext, opcodes_ext_size);
				VerifyParams(opc_ext, params_ext, params_count_ext, true);
			}
			else if (params_count_ext)
				ShowError(ERR_EXT_PAR_NOT_EXT);
		}
		else
		{
			if (opcode_ext)
				ShowError(ERR_EXT_CANT_EXTEND_OPCODE);
			if (params_count_ext)
				ShowError(ERR_EXT_PAR_NOT_EXT);
		}

		if (pass == 2)
		{
			// generate binary
			((u16 *)gdg_buffer)[m_cur_addr] = 0x0000;
			BuildCode(opc, params, params_count, (u16 *)gdg_buffer);
			if (opc_ext)
				BuildCode(opc_ext, params_ext, params_count_ext, (u16 *)gdg_buffer);
		}

		m_cur_addr += opcode_size;
		m_totalSize += opcode_size;
	};

	if (!failed)
		fsrc.close();

	return !failed;
}
