userspace: helper: Add helper functions for memory domain manipulation

Add helper functions user_add_memory and user_remove_memory that allows to
add/remove memory regions from the memory domain. The purpose of these
functions is to round addresses appropriately for the memory domain.

Signed-off-by: Jaroslaw Stelter <Jaroslaw.Stelter@intel.com>
Signed-off-by: Adrian Warecki <adrian.warecki@intel.com>

Author: jxstelter

zephyr/include/rtos/userspace_helper.h

@@ -40,6 +40,38 @@ struct userspace_context;
  */
 struct sys_heap *module_driver_heap_init(void);
 
+/**
+ * Add memory region to non-privileged module memory domain.
+ * @param domain - pointer to the modules memory domain.
+ * @param addr   - pointer to memory region start
+ * @param size   - size of the memory region
+ * @param attr   - memory region access attributes
+ *
+ * @return 0 for success, error otherwise.
+ *
+ * @note
+ * Function used only when CONFIG_USERSPACE is set.
+ * Function adds page aligned region to the memory domain.
+ * Caller should take care to not expose other data than these
+ * intended to be shared with the module.
+ */
+int user_add_memory(struct k_mem_domain *domain, uintptr_t addr, size_t size, uint32_t attr);
+
+/**
+ * Remove memory region from non-privileged module memory domain.
+ * @param domain - pointer to the modules memory domain.
+ * @param addr   - pointer to memory region start
+ * @param size   - size of the memory region
+ *
+ * @return 0 for success, error otherwise.
+ *
+ * @note
+ * Function used only when CONFIG_USERSPACE is set.
+ * Function removes previously added page aligned region
+ * from the memory domain.
+ */
+int user_remove_memory(struct k_mem_domain *domain, uintptr_t addr, size_t size);
+
 /**
  * Add DP scheduler created thread to module memory domain.
  * @param thread_id - id of thread to be added to memory domain.

zephyr/lib/userspace_helper.c

@@ -156,6 +156,39 @@ int user_memory_init_shared(k_tid_t thread_id, struct processing_module *mod)
 	return k_mem_domain_add_thread(comp_dom, thread_id);
 }
 
+int user_add_memory(struct k_mem_domain *domain, uintptr_t addr, size_t size, uint32_t attr)
+{
+	struct k_mem_partition part;
+	int ret;
+
+	k_mem_region_align(&part.start, &part.size, addr, size, HOST_PAGE_SIZE);
+	/* Define parameters for partition */
+	part.attr = attr;
+	ret = k_mem_domain_add_partition(domain, &part);
+	/* -EINVAL means that given page is already in the domain */
+	/* Not an error case for us. */
+	if (ret == -EINVAL)
+		return 0;
+
+	return ret;
+}
+
+int user_remove_memory(struct k_mem_domain *domain, uintptr_t addr, size_t size)
+{
+	struct k_mem_partition part;
+	int ret;
+
+	/* Define parameters for user_partition */
+	k_mem_region_align(&part.start, &part.size, addr, size, HOST_PAGE_SIZE);
+	ret = k_mem_domain_remove_partition(domain, &part);
+	/* -ENOENT means that given partition is already removed */
+	/* Not an error case for us. */
+	if (ret == -ENOENT)
+		return 0;
+
+	return ret;
+}
+
 #else /* CONFIG_USERSPACE */
 
 void *user_stack_allocate(size_t stack_size, uint32_t options)