module: Add mod_balloc_aling() function

Add mod_balloc_aling() to enable also automatically freed buffer
memory allocations. The difference at the moment is that "buffers" are
allocated from virtual heap, and other things are allocated from Zephyr
heap assigned for user space shared memory.

Signed-off-by: Jyri Sarha <jyri.sarha@linux.intel.com>

Author: Jyri Sarha

src/audio/module_adapter/module/generic.c

@@ -159,6 +159,54 @@ static void container_put(struct processing_module *mod, struct module_resource
 	list_item_append(&container->list, &res->free_cont_list);
 }
 
+/**
+ * Allocates aligned buffer memory block for module.
+ * @param mod		Pointer to the module this memory block is allocatd for.
+ * @param bytes		Size in bytes.
+ * @param alignment	Alignment in bytes.
+ * @return Pointer to the allocated memory or NULL if failed.
+ *
+ * The allocated memory is automatically freed when the module is unloaded.
+ */
+void *mod_balloc_align(struct processing_module *mod, size_t size, size_t alignment)
+{
+	struct module_resource *container = container_get(mod);
+	struct module_resources *res = &mod->priv.resources;
+	void *ptr;
+
+	MEM_API_CHECK_THREAD(res);
+	if (!container)
+		return NULL;
+
+	if (!size) {
+		comp_err(mod->dev, "requested allocation of 0 bytes.");
+		container_put(mod, container);
+		return NULL;
+	}
+
+	/* Allocate buffer memory for module */
+	ptr = rballoc_align(SOF_MEM_FLAG_USER, size, alignment);
+
+	if (!ptr) {
+		comp_err(mod->dev, "Failed to alloc %zu bytes %zu alignment for comp %#x.",
+			 size, alignment, dev_comp_id(mod->dev));
+		container_put(mod, container);
+		return NULL;
+	}
+	/* Store reference to allocated memory */
+	container->ptr = ptr;
+	container->size = size;
+	container->type = MOD_RES_HEAP;
+	list_item_prepend(&container->list, &res->res_list);
+
+	res->heap_usage += size;
+	if (res->heap_usage > res->heap_high_water_mark)
+		res->heap_high_water_mark = res->heap_usage;
+
+	return ptr;
+}
+EXPORT_SYMBOL(mod_balloc_align);
+
 /**
  * Allocates aligned memory block for module.
  * @param mod		Pointer to the module this memory block is allocatd for.

src/include/sof/audio/module_adapter/module/generic.h

@@ -188,6 +188,7 @@ struct module_processing_data {
 /*****************************************************************************/
 int module_load_config(struct comp_dev *dev, const void *cfg, size_t size);
 int module_init(struct processing_module *mod);
+void *mod_balloc_align(struct processing_module *mod, size_t size, size_t alignment);
 void *mod_alloc_align(struct processing_module *mod, size_t size, size_t alignment);
 void *mod_alloc(struct processing_module *mod, size_t size);
 void *mod_zalloc(struct processing_module *mod, size_t size);