jeans/esphome
jeans/esphome
1
0
Fork 0
mirror of https://github.com/esphome/esphome.git synced 2025-08-01 16:07:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

devug

Browse Source
This commit is contained in:
J. Nick Koston 2025-05-08 17:18:43 -05:00
parent abb09b7fee
commit 57425a765b
No known key found for this signature in database
2 changed files with 59 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
esphome/components/api/__init__.py
View File

@ -56,6 +56,8 @@ CONF_ENCRYPTION = "encryption"
CONF_HEAP_TRACING = "heap_tracing"
CONF_HEAP_TRACING_STANDALONE = "standalone" # vs SYSTEM
CONF_HEAP_TRACING_RECORDS = "num_records"
CONF_HEAP_TASK_TRACKING = "task_tracking"
CONF_HEAP_TASK_MAX = "max_tasks"
def validate_encryption_key(value):
@ -106,6 +108,8 @@ HEAP_TRACING_SCHEMA = cv.Schema(
{
cv.Optional(CONF_HEAP_TRACING_STANDALONE, default=True): cv.boolean,
cv.Optional(CONF_HEAP_TRACING_RECORDS, default=100): cv.positive_int,
cv.Optional(CONF_HEAP_TASK_TRACKING, default=True): cv.boolean,
cv.Optional(CONF_HEAP_TASK_MAX, default=10): cv.positive_int,
}
)
@ -212,6 +216,10 @@ async def to_code(config):
else:
add_idf_sdkconfig_option("CONFIG_HEAP_TRACING_SYSTEM", True)
# Enable heap task tracking if requested
if heap_tracing_config[CONF_HEAP_TASK_TRACKING]:
add_idf_sdkconfig_option("CONFIG_HEAP_TASK_TRACKING", True)
# Generate code to implement heap tracing
cg.add_global(cg.RawStatement('#include "esp_heap_trace.h"'))
@ -223,6 +231,12 @@ async def to_code(config):
)
)
# If task tracking is enabled, add the task tracking code
if heap_tracing_config[CONF_HEAP_TASK_TRACKING]:
max_tasks = heap_tracing_config[CONF_HEAP_TASK_MAX]
# Add the global define to update the max tasks value in the implementation
cg.add_define(f"MAX_HEAP_TASKS {max_tasks}")
# Add helper functions for heap tracing with extern "C" to make them globally accessible
cg.add_global(
cg.RawStatement(

45
esphome/components/api/api_server.cpp
View File

@ -16,10 +16,48 @@
#ifdef USE_API_HEAP_TRACE
#include "esp_heap_trace.h"
#include "esp_heap_task_info.h"
// Forward declare heap tracing functions that will be used in the API class
extern "C" void start_heap_trace();
extern "C" void stop_and_dump_heap_trace();
// Maximum number of tasks we expect to track
#define MAX_HEAP_TASKS 10
// Global storage for task heap info
static heap_task_info_t task_info[MAX_HEAP_TASKS];
static size_t tcb_info_size = 0;
static heap_task_totals_t heap_totals[1] = {{.caps = MALLOC_CAP_DEFAULT}};
// Dump task heap information
extern "C" void dump_task_heap_info() {
heap_task_info_params_t heap_info = {
.task_info = task_info,
.size = MAX_HEAP_TASKS,
.totals = heap_totals,
.totals_size = 1,
};
esp_err_t err = heap_caps_get_per_task_info(&heap_info, &tcb_info_size);
if (err != ESP_OK) {
ESP_LOGE("HEAP", "Failed to get per-task heap info: %d", err);
return;
}
ESP_LOGI("HEAP", "Task Heap Information (%d tasks):", tcb_info_size);
ESP_LOGI("HEAP", "-------------------------------------");
ESP_LOGI("HEAP", "%-20s %10s", "Task", "Heap Usage");
ESP_LOGI("HEAP", "-------------------------------------");
for (size_t i = 0; i < tcb_info_size; i++) {
ESP_LOGI("HEAP", "%-20s %10d bytes", task_info[i].task_name, task_info[i].caps[0]);
}
ESP_LOGI("HEAP", "-------------------------------------");
ESP_LOGI("HEAP", "Total heap allocated: %d bytes", heap_totals[0].size);
ESP_LOGI("HEAP", "-------------------------------------");
}
#endif
#include <algorithm>
@ -176,6 +214,9 @@ void APIServer::loop() {
ESP_LOGI(TAG, "Dumping heap trace information");
stop_and_dump_heap_trace();
// Also dump task-specific heap information
dump_task_heap_info();
// Start a new trace for the next period
start_heap_trace();
@ -499,6 +540,10 @@ void APIServer::on_shutdown() {
// Make sure to stop tracing on shutdown to get final results
ESP_LOGI(TAG, "Final heap trace dump on shutdown");
stop_and_dump_heap_trace();
// Dump final task heap information
ESP_LOGI(TAG, "Final task heap information dump on shutdown");
dump_task_heap_info();
#endif
}