llext: add support for adding LLEXT memory to a memory domain

Add functions for adding LLEXT partitions to a memory domain for
user-space modules.

Signed-off-by: Guennadi Liakhovetski <guennadi.liakhovetski@linux.intel.com>

Author: lyakh

src/include/sof/llext_manager.h

@@ -15,6 +15,7 @@
 struct comp_dev;
 struct comp_driver;
 struct comp_ipc_config;
+struct k_mem_domain;
 
 static inline bool module_is_llext(const struct sof_man_module *mod)
 {
@@ -29,12 +30,15 @@ int llext_manager_free_module(const uint32_t component_id);
 
 int llext_manager_add_library(uint32_t module_id);
 
+int llext_manager_add_domain(const uint32_t component_id, struct k_mem_domain *domain);
+
 bool comp_is_llext(struct comp_dev *comp);
 #else
 #define module_is_llext(mod) false
 #define llext_manager_allocate_module(ipc_config, ipc_specific_config) 0
 #define llext_manager_free_module(component_id) 0
 #define llext_manager_add_library(module_id) 0
+#define llext_manager_add_domain(component_id, domain) 0
 #define comp_is_llext(comp) false
 #endif
 

src/library_manager/llext_manager.c

@@ -25,6 +25,7 @@
 #include <sof/audio/module_adapter/module/modules.h>
 
 #include <zephyr/cache.h>
+#include <zephyr/app_memory/mem_domain.h>
 #include <zephyr/drivers/mm/system_mm.h>
 #include <zephyr/llext/buf_loader.h>
 #include <zephyr/llext/loader.h>
@@ -716,6 +717,66 @@ uintptr_t llext_manager_allocate_module(const struct comp_ipc_config *ipc_config
 	return mod_manifest->module.entry_point;
 }
 
+#ifdef CONFIG_USERSPACE
+static int llext_manager_add_partition(struct k_mem_domain *domain,
+				       uintptr_t addr, size_t size,
+				       k_mem_partition_attr_t attr)
+{
+	size_t pre_pad_size = addr & (PAGE_SZ - 1);
+	struct k_mem_partition part = {
+		.start = addr - pre_pad_size,
+		.size = ALIGN_UP(pre_pad_size + size, PAGE_SZ),
+		.attr = attr,
+	};
+
+	tr_dbg(&lib_manager_tr, "add %#zx @ %lx partition", part.size, part.start);
+	return k_mem_domain_add_partition(domain, &part);
+}
+
+int llext_manager_add_domain(const uint32_t component_id, struct k_mem_domain *domain)
+{
+	const uint32_t module_id = IPC4_MOD_ID(component_id);
+	struct lib_manager_mod_ctx *ctx = lib_manager_get_mod_ctx(module_id);
+	const uint32_t entry_index = LIB_MANAGER_GET_MODULE_INDEX(module_id);
+	const unsigned int mod_idx = llext_manager_mod_find(ctx, entry_index);
+	struct lib_manager_module *mctx = ctx->mod + mod_idx;
+	int ret;
+
+	/* Executable code (.text) */
+	uintptr_t va_base_text = mctx->segment[LIB_MANAGER_TEXT].addr;
+	size_t text_size = mctx->segment[LIB_MANAGER_TEXT].size;
+
+	/* Read-only data (.rodata and others) */
+	uintptr_t va_base_rodata = mctx->segment[LIB_MANAGER_RODATA].addr;
+	size_t rodata_size = mctx->segment[LIB_MANAGER_RODATA].size;
+
+	/* Writable data (.data, .bss and others) */
+	uintptr_t va_base_data = mctx->segment[LIB_MANAGER_DATA].addr;
+	size_t data_size = mctx->segment[LIB_MANAGER_DATA].size;
+
+	ret = llext_manager_add_partition(domain, va_base_text, text_size,
+					  K_MEM_PARTITION_P_RX_U_RX);
+	if (ret < 0)
+		return ret;
+
+	if (rodata_size) {
+		ret = llext_manager_add_partition(domain, va_base_rodata, rodata_size,
+						  K_MEM_PARTITION_P_RO_U_RO);
+		if (ret < 0)
+			return ret;
+	}
+
+	if (data_size) {
+		ret = llext_manager_add_partition(domain, va_base_data, data_size,
+						  K_MEM_PARTITION_P_RW_U_RW);
+		if (ret < 0)
+			return ret;
+	}
+
+	return 0;
+}
+#endif
+
 int llext_manager_free_module(const uint32_t component_id)
 {
 	const uint32_t module_id = IPC4_MOD_ID(component_id);