module: add an allocation function with flags

Add a new mod_alloc_ext() allocation function with support for
allocator flags.

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

Author: lyakh

src/audio/module_adapter/module/generic.c

@@ -218,15 +218,16 @@ void *mod_balloc_align(struct processing_module *mod, size_t size, size_t alignm
 EXPORT_SYMBOL(mod_balloc_align);
 
 /**
- * Allocates aligned memory block for module.
+ * Allocates aligned memory block with flags for module.
  * @param mod		Pointer to the module this memory block is allocatd for.
+ * @param flags		Allocator flags.
  * @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_alloc_align(struct processing_module *mod, size_t size, size_t alignment)
+void *mod_alloc_ext(struct processing_module *mod, uint32_t flags, size_t size, size_t alignment)
 {
 	struct module_resources *res = &mod->priv.resources;
 	struct module_resource *container;
@@ -245,7 +246,7 @@ void *mod_alloc_align(struct processing_module *mod, size_t size, size_t alignme
 	}
 
 	/* Allocate memory for module */
-	ptr = rmalloc_align(SOF_MEM_FLAG_USER, size, alignment);
+	ptr = rmalloc_align(flags, size, alignment);
 
 	if (!ptr) {
 		comp_err(mod->dev, "Failed to alloc %zu bytes %zu alignment for comp %#x.",
@@ -265,7 +266,7 @@ void *mod_alloc_align(struct processing_module *mod, size_t size, size_t alignme
 
 	return ptr;
 }
-EXPORT_SYMBOL(mod_alloc_align);
+EXPORT_SYMBOL(mod_alloc_ext);
 
 /**
  * Creates a blob handler and releases it when the module is unloaded

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

@@ -189,12 +189,27 @@ 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_ext(struct processing_module *mod, uint32_t flags, size_t size, size_t alignment);
+
+/**
+ * Allocates aligned memory block for module.
+ * @param mod		Pointer to the module this memory block is allocated 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.
+ */
+static inline void *mod_alloc_align(struct processing_module *mod, size_t size, size_t alignment)
+{
+	return mod_alloc_ext(mod, 0, size, alignment);
+}
+
 static inline void *mod_balloc(struct processing_module *mod, size_t size)
 {
 	return mod_balloc_align(mod, size, 0);
 }
 
-void *mod_alloc_align(struct processing_module *mod, size_t size, size_t alignment);
 static inline void *mod_alloc(struct processing_module *mod, size_t size)
 {
 	return mod_alloc_align(mod, size, 0);

test/cmocka/src/common_mocks.c

@@ -102,10 +102,12 @@ void WEAK *mod_balloc_align(struct processing_module *mod, size_t size, size_t a
 	return ret;
 }
 
-void WEAK *mod_alloc_align(struct processing_module *mod, size_t size, size_t alignment)
+void WEAK *mod_alloc_ext(struct processing_module *mod, uint32_t flags, size_t size,
+			 size_t alignment)
 {
 	void *ret;
 	(void)mod;
+	(void)flags;
 	(void)alignment;
 
 	ret = malloc(size);