hostap: Add heap allocation tracking for hostap

Create "wifi mem" command that can show how the heap memory
is being used in hostap. The command is available if
CONFIG_WIFI_NM_WPA_SUPPLICANT_DEBUG_ALLOC is enabled.

Signed-off-by: Jukka Rissanen <[email protected]>

Author: jukkar

modules/hostap/Kconfig

@@ -85,6 +85,20 @@ choice LOG_MODE
 endchoice
 endif # WIFI_NM_WPA_SUPPLICANT_LOG_LEVEL_DBG
 
+config WIFI_NM_WPA_SUPPLICANT_DEBUG_ALLOC
+	bool "Debug memory allocation in hostap"
+	help
+	  Enables printing of network packet and buffer allocations and frees for
+	  each allocation. This can produce lot of output so it is disabled by
+	  default.
+
+config WIFI_NM_WPA_SUPPLICANT_DEBUG_ALLOC_LEN
+	int "Max number of allocation tracking entries"
+	default 200
+	depends on WIFI_NM_WPA_SUPPLICANT_DEBUG_ALLOC
+	help
+	  Each entry can track one allocation. Increase this if needed.
+
 # Memory optimizations
 config WIFI_NM_WPA_SUPPLICANT_ADVANCED_FEATURES
 	bool "Advanced features"

subsys/net/l2/wifi/CMakeLists.txt

@@ -2,6 +2,7 @@
 
 zephyr_library()
 zephyr_library_include_directories(${ZEPHYR_BASE}/subsys/net/ip)
+zephyr_library_include_directories(${HOSTAP_BASE})
 zephyr_library_include_directories_ifdef(
   CONFIG_NET_L2_WIFI_SHELL ${ZEPHYR_BASE}/subsys/net/lib/shell
   )

subsys/net/l2/wifi/wifi_shell.c

@@ -2913,6 +2913,63 @@ void parse_mode_args_to_params(const struct shell *sh, int argc,
 	}
 }
 
+#if defined(CONFIG_WIFI_NM_WPA_SUPPLICANT_DEBUG_ALLOC)
+#include <src/utils/os.h>
+
+static void mem_alloc_cb(void *data,
+			 size_t size,
+			 const char *file_alloc,
+			 const char *func_alloc,
+			 int line_alloc,
+			 const char *file_free,
+			 const char *func_free,
+			 int line_free,
+			 bool in_use,
+			 void *user_data)
+{
+	const struct shell *sh = (const struct shell *)user_data;
+
+	if (in_use) {
+		if (func_free != NULL) {
+			PR("Alloc: %zu bytes at %s():%d, free at %s():%d\n",
+			   size, func_alloc, line_alloc, func_free, line_free);
+		} else {
+			PR("Alloc: %zu bytes at %s():%d (%s)\n",
+			   size, func_alloc, line_alloc, file_alloc);
+		}
+	} else {
+		if (size > 0) {
+			PR("Free: %zu bytes at %s():%d (%s)\n",
+			   size, func_free, line_free, file_free);
+		}
+	}
+}
+#endif
+
+static int cmd_wifi_mem(const struct shell *sh, size_t argc, char *argv[])
+{
+	ARG_UNUSED(argc);
+	ARG_UNUSED(argv);
+
+#if defined(CONFIG_WIFI_NM_WPA_SUPPLICANT_DEBUG_ALLOC)
+	int64_t total_usage;
+
+	total_usage = hostap_get_total_allocs();
+
+	PR("Total heap memory usage: %lld bytes\n", total_usage);
+	PR("Max heap memory usage: %zu bytes\n\n", hostap_get_max_allocs());
+
+	hostap_allocs_foreach(mem_alloc_cb, (void *)sh);
+
+	return 0;
+#else
+	PR_INFO("Set %s to enable %s support.\n",
+		"CONFIG_WIFI_NM_WPA_SUPPLICANT_DEBUG_ALLOC", "hostap memory usage");
+
+	return -ENOEXEC;
+#endif
+}
+
 static int cmd_wifi_mode(const struct shell *sh, size_t argc, char *argv[])
 {
 	struct net_if *iface;
@@ -4054,6 +4111,11 @@ SHELL_SUBCMD_ADD((wifi), disconnect, NULL,
 		 cmd_wifi_disconnect,
 		 1, 2);
 
+SHELL_SUBCMD_ADD((wifi), mem, NULL,
+		 "Show memory allocations in hostap.\n",
+		 cmd_wifi_mem,
+		 1, 0);
+
 SHELL_SUBCMD_ADD((wifi), mode, NULL,
 		 "mode operational setting\n"
 		 "This command may be used to set the Wi-Fi device into a specific "