fast-get: allocate on specific heap

Pass a "heap" argument to fast_get() and fast_put() for user-space
DP allocations.

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

Author: lyakh

src/audio/module_adapter/module/generic.c

@@ -326,7 +326,7 @@ const void *mod_fast_get(struct processing_module *mod, const void * const dram_
 	if (!container)
 		return NULL;
 
-	ptr = fast_get(dram_ptr, size);
+	ptr = fast_get(res->heap, dram_ptr, size);
 	if (!ptr) {
 		container_put(mod, container);
 		return NULL;
@@ -358,7 +358,7 @@ static int free_contents(struct processing_module *mod, struct module_resource *
 #endif
 #if CONFIG_FAST_GET
 	case MOD_RES_FAST_GET:
-		fast_put(container->sram_ptr);
+		fast_put(res->heap, container->sram_ptr);
 		return 0;
 #endif
 	default:

src/include/sof/lib/fast-get.h

@@ -10,7 +10,8 @@
 
 #include <stddef.h>
 
-const void *fast_get(const void * const dram_ptr, size_t size);
-void fast_put(const void *sram_ptr);
+struct k_heap;
+const void *fast_get(struct k_heap *heap, const void * const dram_ptr, size_t size);
+void fast_put(struct k_heap *heap, const void *sram_ptr);
 
 #endif /* __SOF_LIB_FAST_GET_H__ */

zephyr/lib/fast-get.c

@@ -80,7 +80,7 @@ static struct sof_fast_get_entry *fast_get_find_entry(struct sof_fast_get_data *
 	return NULL;
 }
 
-const void *fast_get(const void *dram_ptr, size_t size)
+const void *fast_get(struct k_heap *heap, const void *dram_ptr, size_t size)
 {
 	struct sof_fast_get_data *data = &fast_get_data;
 	struct sof_fast_get_entry *entry;
@@ -116,7 +116,7 @@ const void *fast_get(const void *dram_ptr, size_t size)
 		goto out;
 	}
 
-	ret = rmalloc(SOF_MEM_FLAG_USER, size);
+	ret = sof_heap_alloc(heap, SOF_MEM_FLAG_USER, size, 0);
 	if (!ret)
 		goto out;
 	entry->size = size;
@@ -146,7 +146,7 @@ static struct sof_fast_get_entry *fast_put_find_entry(struct sof_fast_get_data *
 	return NULL;
 }
 
-void fast_put(const void *sram_ptr)
+void fast_put(struct k_heap *heap, const void *sram_ptr)
 {
 	struct sof_fast_get_data *data = &fast_get_data;
 	struct sof_fast_get_entry *entry;
@@ -160,7 +160,7 @@ void fast_put(const void *sram_ptr)
 	}
 	entry->refcount--;
 	if (!entry->refcount) {
-		rfree(entry->sram_ptr);
+		sof_heap_free(heap, entry->sram_ptr);
 		memset(entry, 0, sizeof(*entry));
 	}
 out: