From 28569d118e907e9dbd1ffadf939dca2fba7c0503 Mon Sep 17 00:00:00 2001
From: Andrew Alderwick <andrew@alderwick.co.uk>
Date: Fri, 27 Aug 2021 23:47:05 +0100
Subject: [PATCH] Implemented includes in uxnasm (no asma support yet)

---
 src/uxnasm.c | 24 +++++++++++++++++++++++-
 1 file changed, 23 insertions(+), 1 deletion(-)

diff --git a/src/uxnasm.c b/src/uxnasm.c
index a220af2..c59c1d6 100644
--- a/src/uxnasm.c
+++ b/src/uxnasm.c
@@ -157,7 +157,7 @@ makemacro(char *name, FILE *f)
 		return error("Macro name is invalid", name);
 	m = &p.macros[p.mlen++];
 	scpy(name, m->name, 64);
-	while(fscanf(f, "%63s", word)) {
+	while(fscanf(f, "%63s", word) == 1) {
 		if(word[0] == '{') continue;
 		if(word[0] == '}') break;
 		if(m->len > 64)
@@ -284,6 +284,21 @@ parsetoken(char *w)
 	return error("Invalid token", w);
 }
 
+static int
+doinclude(FILE *f, int (*pass)(FILE *))
+{
+	char word[64];
+	FILE *finc;
+	int ret;
+	if(fscanf(f, "%63s", word) != 1)
+		return error("End of input", "include");
+	if(!(finc = fopen(word, "r")))
+		return error("Include failed to open", word);
+	ret = pass(finc);
+	fclose(finc);
+	return ret;
+}
+
 static int
 pass1(FILE *f)
 {
@@ -308,6 +323,9 @@ pass1(FILE *f)
 		} else if(w[0] == '&') {
 			if(!makelabel(sublabel(subw, scope, w + 1), addr))
 				return error("Pass 1 - Invalid sublabel", w);
+		} else if(scmp(w, "include", 8)) {
+			if(!doinclude(f, pass1))
+				return 0;
 		} else if(sihx(w))
 			addr += slen(w) / 2;
 		else
@@ -340,6 +358,10 @@ pass2(FILE *f)
 		} else if(w[0] == '@') {
 			scpy(w + 1, scope, 64);
 			continue;
+		} else if(scmp(w, "include", 8)) {
+			if(!doinclude(f, pass2))
+				return 0;
+			continue;
 		}
 		if(w[1] == '&')
 			scpy(sublabel(subw, scope, w + 2), w + 1, 64);