loader: Add support for shared multi-heap.

Signed-off-by: iabdalkader <[email protected]>

Author: iabdalkader

loader/fixups.c

@@ -70,4 +70,52 @@ int camera_ext_clock_enable(void)
 
 SYS_INIT(camera_ext_clock_enable, POST_KERNEL, CONFIG_CLOCK_CONTROL_PWM_INIT_PRIORITY);
 
-#endif
+#endif
+
+#if defined(CONFIG_SHARED_MULTI_HEAP)
+#include <zephyr/kernel.h>
+#include <zephyr/devicetree.h>
+#include <zephyr/multi_heap/shared_multi_heap.h>
+
+struct memory_region_t {
+	uintptr_t dt_addr;
+	size_t dt_size;
+	const char *dt_name;
+};
+
+#define _BUILD_MEM_REGION(node_id)      \
+	{ .dt_addr = DT_REG_ADDR(node_id),  \
+      .dt_size = DT_REG_SIZE(node_id),  \
+      .dt_name = DT_PROP(node_id, zephyr_memory_region) \
+    },
+
+int smh_init(void)
+{
+    int ret = 0;
+    ret = shared_multi_heap_pool_init();
+    if (ret != 0) {
+        return ret;
+    }
+
+    const struct memory_region_t regions[] = {
+        DT_FOREACH_STATUS_OKAY(zephyr_memory_region, _BUILD_MEM_REGION)
+    };
+
+    for (size_t i=0; i<ARRAY_SIZE(regions); i++) {
+        if (!strncmp("SDRAM", regions[i].dt_name, 5)) {
+            struct shared_multi_heap_region smh_sdram = {
+                .addr = regions[i].dt_addr,
+                .size = regions[i].dt_size,
+                .attr = SMH_REG_ATTR_EXTERNAL,
+            };
+            ret = shared_multi_heap_add(&smh_sdram, NULL);
+            if (ret != 0) {
+                return ret;
+            }
+        }
+    }
+	return 0;
+}
+
+SYS_INIT(smh_init, POST_KERNEL, CONFIG_CLOCK_CONTROL_PWM_INIT_PRIORITY);
+#endif

loader/llext_exports.c

@@ -103,6 +103,11 @@ FORCE_EXPORT_SYM(video_buffer_alloc);
 FORCE_EXPORT_SYM(video_buffer_release);
 #endif
 
+#if defined(CONFIG_SHARED_MULTI_HEAP)
+FORCE_EXPORT_SYM(shared_multi_heap_aligned_alloc);
+FORCE_EXPORT_SYM(shared_multi_heap_free);
+#endif
+
 #if defined(CONFIG_NET_SOCKETS)
 FORCE_EXPORT_SYM(getaddrinfo);
 FORCE_EXPORT_SYM(socket);
@@ -176,4 +181,4 @@ FORCE_EXPORT_SYM(__aeabi_f2d);
 FORCE_EXPORT_SYM(__aeabi_idivmod);
 FORCE_EXPORT_SYM(__aeabi_ldivmod);
 FORCE_EXPORT_SYM(__aeabi_ul2f);
-FORCE_EXPORT_SYM(__cxa_pure_virtual);
+FORCE_EXPORT_SYM(__cxa_pure_virtual);