From b01ed56ce71c56eb289e2bf6cf471a2183c7ac3f Mon Sep 17 00:00:00 2001
From: Igor M <m1el.2027@gmail.com>
Date: Sun, 17 Mar 2024 19:19:25 +0200
Subject: [PATCH] handle malloc fail at start

---
 src/hbas.c | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/src/hbas.c b/src/hbas.c
index 64cee2f..1498521 100644
--- a/src/hbas.c
+++ b/src/hbas.c
@@ -227,6 +227,9 @@ AsmError assemble(InstHt ht, char *input, size_t len, ByteVec *out,
     ByteVec rv = {malloc(MIN_SIZE), MIN_SIZE, 0};
     HoleVec holes = {malloc(MIN_SIZE * sizeof(Hole)), MIN_SIZE, 0};
     LabelVec labels = {malloc(MIN_SIZE * sizeof(Label)), MIN_SIZE, 0};
+    if (rv.buf == NULL || holes.buf == NULL || labels.buf == NULL) {
+        return ErrOutOfMemory;
+    }
     size_t line = 0;
     size_t line_start = 0;
     size_t pos = 0;