fix assembler crashing when labels aren't found

MESYETI · Nov 17, 2023 · 90db64d · 90db64d
1 parent 7a6541d
commit 90db64d
Showing 1 changed file with 11 additions and 6 deletions.
diff --git a/source/assembler/assembler.d b/source/assembler/assembler.d
@@ -392,12 +392,10 @@ class Assembler {
 					}
 
 					if (!valid) {
-						Error(
-							format(
-								"Parameter %d is invalid for instruction %s", i + 1,
-								inst.name
-							)
-						);
+						Error(format(
+							"Parameter %d is invalid for instruction %s", i + 1,
+							inst.name
+						));
 						exit(1);
 					}
 				}
@@ -448,6 +446,13 @@ class Assembler {
 								case NodeType.Identifier: {
 									auto paramNode = cast(IdentifierNode) param;
 
+									if (paramNode.name !in labels) {
+										Error(format(
+											"No such label: '%s'", paramNode.name
+										));
+										return;
+									}
+
 									addr = labels[paramNode.name];
 
 									string[] bsInstructions = [