zephyr: shell: Add sof module_heap_usage command

Adds new command to sof sub-command set. The new command,
module_heap_usage, prints out all active component ids, their heap
memory usage allocated through module API, and configured maximum
heap size.

NOTE: For the moment there is nothing preventing a module from
allocating more memory than the configured maximum.

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

Author: Jyri Sarha

zephyr/sof_shell.c

@@ -7,6 +7,7 @@
 
 #include <rtos/sof.h> /* sof_get() */
 #include <sof/schedule/ll_schedule_domain.h>
+#include <sof/audio/module_adapter/module/generic.h>
 
 #include <zephyr/kernel.h>
 #include <zephyr/sys/printk.h>
@@ -41,11 +42,39 @@ static int cmd_sof_test_inject_sched_gap(const struct shell *sh,
 	return 0;
 }
 
+static int cmd_sof_module_heap_usage(const struct shell *sh,
+				     size_t argc, char *argv[])
+{
+	struct ipc *ipc = sof_get()->ipc;
+	struct list_item *clist, *_clist;
+	struct ipc_comp_dev *icd;
+
+	if (!ipc) {
+		shell_print(sh, "No IPC");
+		return 0;
+	}
+
+	list_for_item_safe(clist, _clist, &ipc->comp_list) {
+		icd = container_of(clist, struct ipc_comp_dev, list);
+		if (icd->type != COMP_TYPE_COMPONENT)
+			continue;
+
+		shell_print(sh, "comp id 0x%08x\t%8zu bytes\t(%zu max)", icd->id,
+			    module_adapter_heap_usage(comp_mod(icd->cd)),
+			    comp_mod(icd->cd)->priv.cfg.heap_bytes);
+	}
+	return 0;
+}
+
 SHELL_STATIC_SUBCMD_SET_CREATE(sof_commands,
 	SHELL_CMD(test_inject_sched_gap, NULL,
 		  "Inject a gap to audio scheduling\n",
 		  cmd_sof_test_inject_sched_gap),
 
+	SHELL_CMD(module_heap_usage, NULL,
+		  "Print heap memory usage of each module\n",
+		  cmd_sof_module_heap_usage),
+
 	SHELL_SUBCMD_SET_END
 );