fix: improve OTA update handling and logging for better error reporting

This commit is contained in:
Manuel Weiser 2025-02-20 20:04:58 +01:00
parent d41f0f3e67
commit 44dd485e17
2 changed files with 35 additions and 11 deletions

View File

@ -45,7 +45,7 @@ build_flags =
-mtext-section-literals -mtext-section-literals
'-D VERSION="${common.version}"' '-D VERSION="${common.version}"'
-DASYNCWEBSERVER_REGEX -DASYNCWEBSERVER_REGEX
-DCORE_DEBUG_LEVEL=1 -DCORE_DEBUG_LEVEL=3
-DCONFIG_ARDUHAL_LOG_COLORS=1 -DCONFIG_ARDUHAL_LOG_COLORS=1
-DOTA_DEBUG=1 -DOTA_DEBUG=1
-DARDUINO_RUNNING_CORE=1 -DARDUINO_RUNNING_CORE=1
@ -66,6 +66,7 @@ build_flags =
-DCONFIG_PARTITION_TABLE_OFFSET=0x8000 -DCONFIG_PARTITION_TABLE_OFFSET=0x8000
-DCONFIG_PARTITION_TABLE_MD5=y -DCONFIG_PARTITION_TABLE_MD5=y
-DBOOT_APP_PARTITION_OTA_0=1 -DBOOT_APP_PARTITION_OTA_0=1
-DCONFIG_LOG_DEFAULT_LEVEL=3
extra_scripts = extra_scripts =
scripts/extra_script.py scripts/extra_script.py

View File

@ -22,7 +22,7 @@ void stopAllTasks() {
void handleOTAUpload(AsyncWebServerRequest *request, String filename, size_t index, uint8_t *data, size_t len, bool final) { void handleOTAUpload(AsyncWebServerRequest *request, String filename, size_t index, uint8_t *data, size_t len, bool final) {
if (!index) { if (!index) {
bool isFullImage = true; bool isFullImage = filename.endsWith(".bin");
Serial.printf("Update Start: %s (type: %s)\n", filename.c_str(), isFullImage ? "full" : "OTA"); Serial.printf("Update Start: %s (type: %s)\n", filename.c_str(), isFullImage ? "full" : "OTA");
if (request->contentLength() == 0) { if (request->contentLength() == 0) {
@ -30,46 +30,69 @@ void handleOTAUpload(AsyncWebServerRequest *request, String filename, size_t ind
return; return;
} }
// Berechne verfügbaren Speicherplatz
size_t updateSize = request->contentLength();
size_t maxSketchSpace = (ESP.getFreeSketchSpace() - 0x1000) & 0xFFFFF000;
Serial.printf("Update size: %u bytes\n", updateSize);
Serial.printf("Available space: %u bytes\n", maxSketchSpace);
if (updateSize > maxSketchSpace) {
Serial.printf("Error: Not enough space. Need %u bytes but only have %u bytes available\n",
updateSize, maxSketchSpace);
request->send(400, "application/json",
"{\"status\":\"error\",\"message\":\"Not enough space for update\"}");
return;
}
if (!tasksAreStopped && (RfidReaderTask || BambuMqttTask || ScaleTask)) { if (!tasksAreStopped && (RfidReaderTask || BambuMqttTask || ScaleTask)) {
stopAllTasks(); stopAllTasks();
tasksAreStopped = true; tasksAreStopped = true;
} }
// Ensure SPIFFS is ended before update
if (SPIFFS.begin()) {
SPIFFS.end();
}
bool success; bool success;
if (isFullImage) { if (isFullImage) {
// Full image update ohne Magic Byte Check, aber mit U_FLASH success = Update.begin(updateSize, U_FLASH);
success = Update.begin(UPDATE_SIZE_UNKNOWN, U_FLASH);
} else { } else {
// Normales OTA update mit Magic Byte Check
if (data[0] != 0xE9) { if (data[0] != 0xE9) {
Serial.printf("Wrong magic byte: 0x%02X (expected 0xE9)\n", data[0]); Serial.printf("Wrong magic byte: 0x%02X (expected 0xE9)\n", data[0]);
request->send(400, "application/json", "{\"status\":\"error\",\"message\":\"Invalid firmware format\"}"); request->send(400, "application/json",
"{\"status\":\"error\",\"message\":\"Invalid firmware format\"}");
return; return;
} }
success = Update.begin(request->contentLength()); success = Update.begin(updateSize);
} }
if (!success) { if (!success) {
Update.printError(Serial); Update.printError(Serial);
request->send(400, "application/json", "{\"status\":\"error\",\"message\":\"Update start failed\"}"); request->send(400, "application/json",
"{\"status\":\"error\",\"message\":\"Update initialization failed\"}");
return; return;
} }
} }
if (Update.write(data, len) != len) { if (Update.write(data, len) != len) {
Update.printError(Serial); Update.printError(Serial);
request->send(400, "application/json", "{\"status\":\"error\",\"message\":\"Write failed\"}"); request->send(400, "application/json",
"{\"status\":\"error\",\"message\":\"Write failed\"}");
return; return;
} }
if (final) { if (final) {
if (!Update.end(true)) { if (!Update.end(true)) {
Update.printError(Serial); Update.printError(Serial);
request->send(400, "application/json", "{\"status\":\"error\",\"message\":\"Update failed\"}"); request->send(400, "application/json",
"{\"status\":\"error\",\"message\":\"Update failed\"}");
return; return;
} }
Serial.println("Update successful, restarting..."); Serial.println("Update successful, restarting...");
request->send(200, "application/json", "{\"status\":\"success\",\"message\":\"Update successful! Device will restart...\",\"restart\":true}"); request->send(200, "application/json",
"{\"status\":\"success\",\"message\":\"Update successful! Device will restart...\",\"restart\":true}");
delay(500); delay(500);
ESP.restart(); ESP.restart();
} }