From 87fa534a7af257405cd114e4139ad1d34fcc7ca7 Mon Sep 17 00:00:00 2001
From: Aryadev Chavali <aryadev@aryadevchavali.com>
Date: Tue, 3 Dec 2024 04:22:59 +0000
Subject: asm_write_{init, exit} when asm_translate_nodes

---
 assembler.c | 2 ++
 1 file changed, 2 insertions(+)

(limited to 'assembler.c')

diff --git a/assembler.c b/assembler.c
index df92f1e..8270f01 100644
--- a/assembler.c
+++ b/assembler.c
@@ -119,6 +119,7 @@ void ast_ref_to_asm_label(u64, struct Label *, u64, i64 *, i64 *);
 void asm_translate_nodes(buffer_t **asm_buffer, struct PResult nodes,
                          const char *src_name)
 {
+  asm_write_init(asm_buffer);
   // First pass: Setup the ASM label array
   struct Label labels[nodes.labels ? nodes.labels * 2 : 1];
   if (nodes.labels)
@@ -184,6 +185,7 @@ void asm_translate_nodes(buffer_t **asm_buffer, struct PResult nodes,
                           table[node.type].len);
     }
   }
+  asm_write_exit(asm_buffer);
 }
 
 /* Implementations for throwaway functions */
-- 
cgit v1.2.3-13-gbd6f