zc0: embed of an array type inside a function

1 files changed, 4 insertions, 2 deletions

diff --git a/src/tools/zc0.c b/src/tools/zc0.c
index f28c655..51ea9f8 100644
--- a/src/tools/zc0.c
+++ b/src/tools/zc0.c
@@ -994,12 +994,13 @@ static void embeddecl(int local)
 		}
 	} else {
 		int count; w = embed_width(ty, star);
-		if (local) die("zc0 embeds an array at the top level, not in a function");
 		if (!w) die("embed declares a bstr, a char*, a number or an array of numbers");
 		if (len % w) die("the embedded file's size is not a whole number of elements");
 		count = (int)(len / w);
 		if (arr >= 0 && count > arr) die("the embedded file is larger than the array it is declared as");
 		if (arr >= 0) count = arr;
+		/* in a function the elements sit in the code, jumped over, and the local holds their address */
+		if (local) { j = here; lit(0); op(O_JMP); }
 		a = here;
 		for (i = 0; i < count; i++) {
 			uint64_t u = 0; int k;
@@ -1009,7 +1010,8 @@ static void embeddecl(int local)
 			emit((int64_t)u);
 		}
 		if (!count) emit(0);
-		add(n, K_GARR, a);
+		if (local) { setlit(j, here); l = newloc(n); lit(a); laddr(l->slot); op(O_ST); }
+		else add(n, K_GARR, a);
 	}
 	free(buf);
 }