From 27fe7b444f01702bd133e2f676f56495332735f7 Mon Sep 17 00:00:00 2001
From: Kelvin Zhang <zhangkelvin@google.com>
Date: Tue, 8 Oct 2024 10:38:02 -0700
Subject: [PATCH] [app][uefi] Enable mspace feature on dlmalloc

This allows us to use an identitiy mapped piece of memory as the heap
for UEFI app. UEFI app's allocation would go from this heap.
---
 external/lib/heap/dlmalloc/rules.mk | 2 ++
 lib/uefi/rules.mk                   | 2 ++
 2 files changed, 4 insertions(+)

diff --git a/external/lib/heap/dlmalloc/rules.mk b/external/lib/heap/dlmalloc/rules.mk
index 3a5ae667b9..b62598861c 100644
--- a/external/lib/heap/dlmalloc/rules.mk
+++ b/external/lib/heap/dlmalloc/rules.mk
@@ -2,6 +2,8 @@ LOCAL_DIR := $(GET_LOCAL_DIR)
 
 MODULE := $(LOCAL_DIR)
 
+MODULE_DEFINES=MSPACES=1
+
 MODULE_SRCS += \
 	$(LOCAL_DIR)/dlmalloc.c
 
diff --git a/lib/uefi/rules.mk b/lib/uefi/rules.mk
index c5a1e59cad..6dabf4ab71 100644
--- a/lib/uefi/rules.mk
+++ b/lib/uefi/rules.mk
@@ -4,6 +4,8 @@ MODULE := $(LOCAL_DIR)
 
 MODULE_INCLUDES += $(LOCAL_DIR)/include
 
+MODULE_DEFINES=MSPACES=1
+
 MODULE_SRCS += \
 	$(LOCAL_DIR)/uefi.cpp \
 	$(LOCAL_DIR)/text_protocol.cpp \