shithub: rgbds

ref: 6e0aca47d4c95c46d1e77994aaae48b9aa619a84
dir: /src/asm/main.c/

#include <math.h>
#include <stdarg.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <time.h>
#include <unistd.h>

#include "asm/symbol.h"
#include "asm/fstack.h"
#include "asm/output.h"
#include "asm/main.h"
#include "extern/err.h"
#include "extern/reallocarray.h"

int yyparse(void);
void setuplex(void);

int cldefines_index;
int cldefines_size;
char **cldefines;

char *progname;

clock_t nStartClock, nEndClock;
SLONG nLineNo;
ULONG nTotalLines, nPass, nPC, nIFDepth, nErrors;

extern int yydebug;

/*
 * Option stack
 */

struct sOptions DefaultOptions;
struct sOptions CurrentOptions;

struct sOptionStackEntry {
	struct sOptions Options;
	struct sOptionStackEntry *pNext;
};

struct sOptionStackEntry *pOptionStack = NULL;

void 
opt_SetCurrentOptions(struct sOptions * pOpt)
{
	if (nGBGfxID != -1) {
		lex_FloatDeleteRange(nGBGfxID, CurrentOptions.gbgfx[0],
		    CurrentOptions.gbgfx[0]);
		lex_FloatDeleteRange(nGBGfxID, CurrentOptions.gbgfx[1],
		    CurrentOptions.gbgfx[1]);
		lex_FloatDeleteRange(nGBGfxID, CurrentOptions.gbgfx[2],
		    CurrentOptions.gbgfx[2]);
		lex_FloatDeleteRange(nGBGfxID, CurrentOptions.gbgfx[3],
		    CurrentOptions.gbgfx[3]);
		lex_FloatDeleteSecondRange(nGBGfxID, CurrentOptions.gbgfx[0],
		    CurrentOptions.gbgfx[0]);
		lex_FloatDeleteSecondRange(nGBGfxID, CurrentOptions.gbgfx[1],
		    CurrentOptions.gbgfx[1]);
		lex_FloatDeleteSecondRange(nGBGfxID, CurrentOptions.gbgfx[2],
		    CurrentOptions.gbgfx[2]);
		lex_FloatDeleteSecondRange(nGBGfxID, CurrentOptions.gbgfx[3],
		    CurrentOptions.gbgfx[3]);
	}
	if (nBinaryID != -1) {
		lex_FloatDeleteRange(nBinaryID, CurrentOptions.binary[0],
		    CurrentOptions.binary[0]);
		lex_FloatDeleteRange(nBinaryID, CurrentOptions.binary[1],
		    CurrentOptions.binary[1]);
		lex_FloatDeleteSecondRange(nBinaryID, CurrentOptions.binary[0],
		    CurrentOptions.binary[0]);
		lex_FloatDeleteSecondRange(nBinaryID, CurrentOptions.binary[1],
		    CurrentOptions.binary[1]);
	}
	CurrentOptions = *pOpt;

	if (nGBGfxID != -1) {
		lex_FloatAddRange(nGBGfxID, CurrentOptions.gbgfx[0],
		    CurrentOptions.gbgfx[0]);
		lex_FloatAddRange(nGBGfxID, CurrentOptions.gbgfx[1],
		    CurrentOptions.gbgfx[1]);
		lex_FloatAddRange(nGBGfxID, CurrentOptions.gbgfx[2],
		    CurrentOptions.gbgfx[2]);
		lex_FloatAddRange(nGBGfxID, CurrentOptions.gbgfx[3],
		    CurrentOptions.gbgfx[3]);
		lex_FloatAddSecondRange(nGBGfxID, CurrentOptions.gbgfx[0],
		    CurrentOptions.gbgfx[0]);
		lex_FloatAddSecondRange(nGBGfxID, CurrentOptions.gbgfx[1],
		    CurrentOptions.gbgfx[1]);
		lex_FloatAddSecondRange(nGBGfxID, CurrentOptions.gbgfx[2],
		    CurrentOptions.gbgfx[2]);
		lex_FloatAddSecondRange(nGBGfxID, CurrentOptions.gbgfx[3],
		    CurrentOptions.gbgfx[3]);
	}
	if (nBinaryID != -1) {
		lex_FloatAddRange(nBinaryID, CurrentOptions.binary[0],
		    CurrentOptions.binary[0]);
		lex_FloatAddRange(nBinaryID, CurrentOptions.binary[1],
		    CurrentOptions.binary[1]);
		lex_FloatAddSecondRange(nBinaryID, CurrentOptions.binary[0],
		    CurrentOptions.binary[0]);
		lex_FloatAddSecondRange(nBinaryID, CurrentOptions.binary[1],
		    CurrentOptions.binary[1]);
	}
}

void 
opt_Parse(char *s)
{
	struct sOptions newopt;

	newopt = CurrentOptions;

	switch (s[0]) {
	case 'g':
		if (strlen(&s[1]) == 4) {
			newopt.gbgfx[0] = s[1];
			newopt.gbgfx[1] = s[2];
			newopt.gbgfx[2] = s[3];
			newopt.gbgfx[3] = s[4];
		} else {
			errx(1, "Must specify exactly 4 characters for "
			    "option 'g'");
		}
		break;
	case 'b':
		if (strlen(&s[1]) == 2) {
			newopt.binary[0] = s[1];
			newopt.binary[1] = s[2];
		} else {
			errx(1, "Must specify exactly 2 characters for option "
			    "'b'");
		}
		break;
	case 'z':
		if (strlen(&s[1]) <= 2) {
			int result;

			result = sscanf(&s[1], "%lx", &newopt.fillchar);
			if (!((result == EOF) || (result == 1))) {
				errx(1, "Invalid argument for option 'z'");
			}
		} else {
			errx(1, "Invalid argument for option 'z'");
			exit(1);
		}
		break;
	default:
		fatalerror("Unknown option");
		break;
	}

	opt_SetCurrentOptions(&newopt);
}

void 
opt_Push(void)
{
	struct sOptionStackEntry *pOpt;

	if ((pOpt = malloc(sizeof(struct sOptionStackEntry))) != NULL) {
		pOpt->Options = CurrentOptions;
		pOpt->pNext = pOptionStack;
		pOptionStack = pOpt;
	} else
		fatalerror("No memory for option stack");
}

void 
opt_Pop(void)
{
	if (pOptionStack) {
		struct sOptionStackEntry *pOpt;

		pOpt = pOptionStack;
		opt_SetCurrentOptions(&(pOpt->Options));
		pOptionStack = pOpt->pNext;
		free(pOpt);
	} else
		fatalerror("No entries in the option stack");
}

void
opt_AddDefine(char *s)
{
	char *value, *equals;
	if(cldefines_index >= cldefines_size)
	{
		cldefines_size *= 2;
		cldefines = reallocarray(cldefines, cldefines_size,
		                        2 * sizeof(void *));
		if(!cldefines)
		{
			fatalerror("No memory for command line defines");
		}
	}
	equals = strchr(s, '=');
	if(equals)
	{
		*equals = '\0';
		value = equals + 1;
	}
	else
	{
		value = "1";
	}
	cldefines[cldefines_index++] = s;
	cldefines[cldefines_index++] = value;
}

void
opt_ParseDefines()
{
	int i;

	for(i = 0; i < cldefines_index; i += 2)
	{
		sym_AddString(cldefines[i], cldefines[i + 1]);
	}
}

/*
 * Error handling
 */
void
verror(const char *fmt, va_list args)
{
	fprintf(stderr, "ERROR:\t");
	fstk_Dump();
	fprintf(stderr, " :\n\t");
	vfprintf(stderr, fmt, args);
	fprintf(stderr, "\n");
	nErrors += 1;
}

void 
yyerror(const char *fmt, ...)
{
	va_list args;
	va_start(args, fmt);
	verror(fmt, args);
	va_end(args);
}

void 
fatalerror(const char *fmt, ...)
{
	va_list args;
	va_start(args, fmt);
	verror(fmt, args);
	va_end(args);
	exit(5);
}

static void 
usage(void)
{
	printf(
"Usage: rgbasm [-hv] [-b chars] [-Dname[=value]] [-g chars] [-i path]\n"
"              [-o outfile] [-p pad_value] file.asm\n");
	exit(1);
}

int 
main(int argc, char *argv[])
{
	int ch;
	char *ep;

	struct sOptions newopt;

	char *tzMainfile;

	cldefines_size = 32;
	cldefines = reallocarray(cldefines, cldefines_size,
	                        2 * sizeof(void *));
	if(!cldefines)
	{
		fatalerror("No memory for command line defines");
	}

	if (argc == 1)
		usage();

	progname = argv[0];

	/* yydebug=1; */

	DefaultOptions.gbgfx[0] = '0';
	DefaultOptions.gbgfx[1] = '1';
	DefaultOptions.gbgfx[2] = '2';
	DefaultOptions.gbgfx[3] = '3';
	DefaultOptions.binary[0] = '0';
	DefaultOptions.binary[1] = '1';
	DefaultOptions.fillchar = 0;
	DefaultOptions.verbose = false;
	DefaultOptions.haltnop = true;

	opt_SetCurrentOptions(&DefaultOptions);

	newopt = CurrentOptions;

	while ((ch = getopt(argc, argv, "b:D:g:hi:o:p:v")) != -1) {
		switch (ch) {
		case 'b':
			if (strlen(optarg) == 2) {
				newopt.binary[0] = optarg[1];
				newopt.binary[1] = optarg[2];
			} else {
				errx(1, "Must specify exactly 2 characters for "
				    "option 'b'");
			}
			break;
		case 'D':
			opt_AddDefine(optarg);
			break;
		case 'g':
			if (strlen(optarg) == 4) {
				newopt.gbgfx[0] = optarg[1];
				newopt.gbgfx[1] = optarg[2];
				newopt.gbgfx[2] = optarg[3];
				newopt.gbgfx[3] = optarg[4];
			} else {
				errx(1, "Must specify exactly 4 characters for "
				    "option 'g'");
			}
			break;
		case 'h':
			newopt.haltnop = false;
			break;
		case 'i':
			fstk_AddIncludePath(optarg);
			break;
		case 'o':
			out_SetFileName(optarg);
			break;
		case 'p':
			newopt.fillchar = strtoul(optarg, &ep, 0);
			if (optarg[0] == '\0' || *ep != '\0') {
				errx(1, "Invalid argument for option 'p'");
			}
			if (newopt.fillchar < 0 || newopt.fillchar > 0xFF) {
				errx(1, "Argument for option 'p' must be "
				    "between 0 and 0xFF");
			}
			break;
		case 'v':
			newopt.verbose = true;
			break;
		default:
			usage();
		}
	}
	argc -= optind;
	argv += optind;

	opt_SetCurrentOptions(&newopt);

	DefaultOptions = CurrentOptions;

	if (argc == 0)
		usage();

	tzMainfile = argv[argc - 1];

	setuplex();

	if (CurrentOptions.verbose) {
		printf("Assembling %s\n", tzMainfile);
	}

	nStartClock = clock();

	nLineNo = 1;
	nTotalLines = 0;
	nIFDepth = 0;
	nPC = 0;
	nPass = 1;
	nErrors = 0;
	sym_PrepPass1();
	fstk_Init(tzMainfile);
	opt_ParseDefines();

	if (CurrentOptions.verbose) {
		printf("Pass 1...\n");
	}

	yy_set_state(LEX_STATE_NORMAL);
	opt_SetCurrentOptions(&DefaultOptions);

	if (yyparse() != 0 || nErrors != 0) {
		errx(1, "Assembly aborted in pass 1 (%ld errors)!", nErrors);
	}

	if (nIFDepth != 0) {
		errx(1, "Unterminated IF construct (%ld levels)!", nIFDepth);
	}

	nTotalLines = 0;
	nLineNo = 1;
	nIFDepth = 0;
	nPC = 0;
	nPass = 2;
	nErrors = 0;
	sym_PrepPass2();
	out_PrepPass2();
	fstk_Init(tzMainfile);
	yy_set_state(LEX_STATE_NORMAL);
	opt_SetCurrentOptions(&DefaultOptions);
	opt_ParseDefines();

	if (CurrentOptions.verbose) {
		printf("Pass 2...\n");
	}

	if (yyparse() != 0 || nErrors != 0) {
		errx(1, "Assembly aborted in pass 2 (%ld errors)!", nErrors);
	}

	double timespent;

	nEndClock = clock();
	timespent = ((double)(nEndClock - nStartClock))
	    / (double)CLOCKS_PER_SEC;
	if (CurrentOptions.verbose) {
		printf("Success! %ld lines in %d.%02d seconds ", nTotalLines,
		    (int) timespent, ((int) (timespent * 100.0)) % 100);
		if (timespent == 0)
			printf("(INFINITY lines/minute)\n");
		else
			printf("(%d lines/minute)\n",
			    (int) (60 / timespent * nTotalLines));
	}
	out_WriteObject();
	return 0;
}