mirror of
https://github.com/gbdev/rgbds.git
synced 2025-11-20 18:22:07 +00:00
25be5c6561
Improve error messages generated by `errx()`, `warnx()` and similar. Set `progname` to a static string with the name of the program so that the path of the binary isn't included in error messages. Signed-off-by: AntonioND <antonio_nd@outlook.com>
458 lines
9.5 KiB
C
458 lines
9.5 KiB
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: ");
|
|
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);
|
|
}
|
|
|
|
void
|
|
warning(const char *fmt, ...)
|
|
{
|
|
va_list args;
|
|
va_start(args, fmt);
|
|
|
|
fprintf(stderr, "warning: ");
|
|
fstk_Dump();
|
|
fprintf(stderr, ":\n\t");
|
|
vfprintf(stderr, fmt, args);
|
|
fprintf(stderr, "\n");
|
|
|
|
va_end(args);
|
|
}
|
|
|
|
static void
|
|
usage(void)
|
|
{
|
|
printf(
|
|
"Usage: rgbasm [-hvE] [-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 = "rgbasm";
|
|
|
|
/* 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;
|
|
DefaultOptions.exportall = false;
|
|
|
|
opt_SetCurrentOptions(&DefaultOptions);
|
|
|
|
newopt = CurrentOptions;
|
|
|
|
while ((ch = getopt(argc, argv, "b:D:g:hi:o:p:vE")) != -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;
|
|
case 'E':
|
|
newopt.exportall = 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();
|
|
sym_SetExportAll(CurrentOptions.exportall);
|
|
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;
|
|
}
|