src/main/wacc/backend/RuntimeError.scala

@@ -117,6 +117,22 @@ object RuntimeError {
     )
   }
 
-  val all: Chain[RuntimeError] =
+  case object OutOfMemoryError extends RuntimeError {
-    Chain(ZeroDivError, BadChrError, NullPtrError, OverflowError, OutOfBoundsError)
+    val strLabel = ".L._errOutOfMemory_str0"
+    val errStr = "fatal error: out of memory"
+    val errLabel = ".L._errOutOfMemory"
+
+    def generateHandler: Chain[AsmLine] = Chain(
+      LabelDef(OutOfMemoryError.errLabel),
+      stackAlign,
+      Load(RDI, IndexAddress(RIP, LabelArg(OutOfMemoryError.strLabel))),
+      assemblyIR.Call(CLibFunc.PrintF),
+      Move(RDI, ImmediateVal(255)),
+      assemblyIR.Call(CLibFunc.Exit)
+    )
+  }
+
+  val all: Chain[RuntimeError] =
+    Chain(ZeroDivError, BadChrError, NullPtrError, OverflowError, OutOfBoundsError,
+      OutOfMemoryError)
 }

src/main/wacc/backend/asmGenerator.scala

@@ -141,8 +141,13 @@ object asmGenerator {
 
     chain ++= wrapBuiltinFunc(
       labelGenerator.getLabel(Builtin.Malloc),
-      Chain(stackAlign, assemblyIR.Call(CLibFunc.Malloc))
+      Chain(
-      // Out of memory check is optional
+        stackAlign,
+        assemblyIR.Call(CLibFunc.Malloc),
+        // Out of memory check
+        Compare(RAX, ImmediateVal(0)),
+        Jump(LabelArg(OutOfMemoryError.errLabel), Cond.Equal)
+      )
     )
 
     chain ++= wrapBuiltinFunc(