component: dp: Allocate task structure from module driver heap

Allocate task structures from the module's driver heap to enable userspace
access. This change allows the DP task to operate in userspace context.
Also update LL task allocation to use the same heap for consistency,
as the memory release logic is shared and does not differentiate between
heaps.

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

Author: softwarecki

src/idc/idc.c

@@ -184,8 +184,8 @@ static int idc_prepare(uint32_t comp_id)
 	/* we're running LL on different core, so allocate our own task */
 	if (!dev->task && dev->ipc_config.proc_domain == COMP_PROCESSING_DOMAIN_LL) {
 		/* allocate task for shared component */
-		dev->task = rzalloc(SOF_MEM_FLAG_USER,
-				    sizeof(*dev->task));
+		dev->task = module_driver_heap_rzalloc(dev->drv->user_heap, SOF_MEM_FLAG_USER,
+						       sizeof(*dev->task));
 		if (!dev->task) {
 			ret = -ENOMEM;
 			goto out;
@@ -197,7 +197,7 @@ static int idc_prepare(uint32_t comp_id)
 					    dev->priority, comp_task, dev,
 					    dev->ipc_config.core, 0);
 		if (ret < 0) {
-			rfree(dev->task);
+			module_driver_heap_free(dev->drv->user_heap, dev->task);
 			goto out;
 		}
 	}

src/include/sof/audio/component_ext.h

@@ -50,7 +50,7 @@ static inline void comp_free(struct comp_dev *dev)
 	if ((dev->is_shared || dev->ipc_config.proc_domain == COMP_PROCESSING_DOMAIN_DP) &&
 	    dev->task) {
 		schedule_task_free(dev->task);
-		rfree(dev->task);
+		module_driver_heap_free(dev->drv->user_heap, dev->task);
 		dev->task = NULL;
 	}
 

src/schedule/zephyr_dp_schedule.c

@@ -477,6 +477,7 @@ int scheduler_dp_task_init(struct task **task,
 			   size_t stack_size)
 {
 	void __sparse_cache *p_stack = NULL;
+	struct sys_heap *const user_heap = mod->dev->drv->user_heap;
 
 	/* memory allocation helper structure */
 	struct {
@@ -496,8 +497,8 @@ int scheduler_dp_task_init(struct task **task,
 	 * As the structure contains zephyr kernel specific data, it must be located in
 	 * shared, non cached memory
 	 */
-	task_memory = rzalloc(SOF_MEM_FLAG_KERNEL | SOF_MEM_FLAG_COHERENT,
-			      sizeof(*task_memory));
+	task_memory = module_driver_heap_rzalloc(user_heap, SOF_MEM_FLAG_USER |
+						 SOF_MEM_FLAG_COHERENT, sizeof(*task_memory));
 	if (!task_memory) {
 		tr_err(&dp_tr, "memory alloc failed");
 		return -ENOMEM;
@@ -542,7 +543,7 @@ int scheduler_dp_task_init(struct task **task,
 err:
 	/* cleanup - free all allocated resources */
 	rfree((__sparse_force void *)p_stack);
-	rfree(task_memory);
+	module_driver_heap_free(user_heap, task_memory);
 	return ret;
 }