From 7760593a88114f08afb7776ba89ffa059646358c Mon Sep 17 00:00:00 2001
From: Devine Lu Linvega <aliceffekt@gmail.com>
Date: Wed, 27 Mar 2024 10:18:06 -0700
Subject: [PATCH] (uxnasm) Walk macro

---
 src/uxnasm.c | 41 +++++++++++++++++++++++++----------------
 1 file changed, 25 insertions(+), 16 deletions(-)

diff --git a/src/uxnasm.c b/src/uxnasm.c
index b529a89..271f1a9 100644
--- a/src/uxnasm.c
+++ b/src/uxnasm.c
@@ -19,7 +19,7 @@ typedef signed char Sint8;
 typedef unsigned short Uint16;
 
 typedef struct {
-	char *name, items[0x40][0x40];
+	char *name, content[0x80];
 	Uint8 len;
 } Macro;
 
@@ -156,7 +156,7 @@ makemacro(char *name, FILE *f)
 			if(!walkcomment(word, f)) return error_asm("Comment error");
 			continue;
 		}
-		scpy(word, m->items[m->len++], 0x40);
+		scat(scat(m->content, word), " ");
 	}
 	return 1;
 }
@@ -256,18 +256,31 @@ writehex(char *w)
 }
 
 static int
-tokenize(FILE *f)
+walkmacro(Macro *m)
 {
-	char c;
-	char *cptr = token;
+	char c, *contentptr = m->content, *cptr = token;
+	while((c = *contentptr++)) {
+		if(c < 0x21) {
+			*cptr++ = 0x00;
+			if(token[0] && !parse(token, NULL)) return 0;
+			cptr = token;
+		} else
+			*cptr++ = c;
+	}
+	return 1;
+}
+
+static int
+walkfile(FILE *f)
+{
+	char c, *cptr = token;
 	while(fread(&c, 1, 1, f)) {
 		if(c < 0x21) {
 			*cptr++ = 0x00;
 			if(c == 0x0a)
 				p.line++;
-			if(token[0])
-				if(!parse(token, f))
-					return 0;
+			if(token[0] && !parse(token, f))
+				return 0;
 			cptr = token;
 		} else if(cptr - token < 0x3f)
 			*cptr++ = c;
@@ -286,7 +299,7 @@ makeinclude(char *filename)
 		return error_top("Invalid source", filename);
 	scpy(filename, source, 0x40);
 	p.line = 0;
-	res = tokenize(f);
+	res = walkfile(f);
 	fclose(f);
 	return res;
 }
@@ -294,7 +307,6 @@ makeinclude(char *filename)
 static int
 parse(char *w, FILE *f)
 {
-	int i;
 	char c;
 	Macro *m;
 	switch(w[0]) {
@@ -328,12 +340,9 @@ parse(char *w, FILE *f)
 			return writehex(w);
 		else if(isopcode(w))
 			return writebyte(findopcode(w));
-		else if((m = findmacro(w))) {
-			for(i = 0; i < m->len; i++)
-				if(!parse(m->items[i], f))
-					return 0;
-			return 1;
-		} else
+		else if((m = findmacro(w)))
+			return walkmacro(m);
+		else
 			return addref(w, ' ', p.ptr + 1) && writebyte(0x60) && writeshort(0xffff);
 	}
 	return 1;