3D-Druck/Filaman
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 110 Wiki Activity

refactor: simplify OTA update process by removing unnecessary buffer and adjusting offsets

Browse Source
This commit is contained in:
Manuel Weiser 2025-02-20 14:16:49 +01:00
parent 249e896ea4
commit b0c4af7c4e
2 changed files with 38 additions and 132 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
platformio.ini
View File

@ -66,7 +66,7 @@ extra_scripts =
pre:scripts/pre_spiffs.py ; wird als zweites ausgeführt pre:scripts/pre_spiffs.py ; wird als zweites ausgeführt
pre:scripts/combine_html.py ; wird als drittes ausgeführt pre:scripts/combine_html.py ; wird als drittes ausgeführt
scripts/gzip_files.py scripts/gzip_files.py
scripts/create_full_bin.py ; Neues Skript zur Erstellung der full.bin # scripts/create_full_bin.py # Nicht mehr benötigt, da full.bin im Workflow erstellt wird
; Remove or comment out the targets line ; Remove or comment out the targets line
;targets = buildfs, build ;targets = buildfs, build

168
src/ota.cpp
View File

@ -28,24 +28,14 @@ void handleOTAUpload(AsyncWebServerRequest *request, String filename, size_t ind
static size_t contentLength = 0; static size_t contentLength = 0;
static bool isFullImage = false; static bool isFullImage = false;
static uint32_t currentOffset = 0; static uint32_t currentOffset = 0;
static uint8_t *spiffsBuffer = nullptr;
static size_t spiffsSize = 0;
static const uint32_t SPIFFS_START = 0x3D0000; // Korrigierter SPIFFS start address aus partitions.csv
static const uint32_t SPIFFS_SIZE = 0x30000; // Korrigierte SPIFFS size aus partitions.csv
// Stop all tasks to save resources // Offset-Definitionen aus dem Workflow
if (!tasksAreStopped && (RfidReaderTask || BambuMqttTask || ScaleTask)) { static const uint32_t BOOTLOADER_OFFSET = 0x1000;
stopAllTasks(); static const uint32_t PARTITIONS_OFFSET = 0x8000;
tasksAreStopped = true; static const uint32_t FIRMWARE_OFFSET = 0x10000;
} static const uint32_t SPIFFS_OFFSET = 0x3D0000;
if (!index) { if (!index) {
// Reset static variables
if (spiffsBuffer) {
free(spiffsBuffer);
spiffsBuffer = nullptr;
}
contentLength = request->contentLength(); contentLength = request->contentLength();
Serial.printf("Update size: %u bytes\n", contentLength); Serial.printf("Update size: %u bytes\n", contentLength);
@ -54,136 +44,52 @@ void handleOTAUpload(AsyncWebServerRequest *request, String filename, size_t ind
return; return;
} }
if (data[0] == ESP_MAGIC) { // Stop all tasks to save resources
isFullImage = (contentLength > SPIFFS_START); if (!tasksAreStopped && (RfidReaderTask || BambuMqttTask || ScaleTask)) {
stopAllTasks();
if (isFullImage) { tasksAreStopped = true;
if (!Update.begin(SPIFFS_START, U_FLASH)) { }
Serial.printf("Not enough space for firmware: %u required\n", SPIFFS_START);
request->send(400, "application/json", "{\"status\":\"error\",\"message\":\"Not enough space available\"}"); isFullImage = (contentLength > SPIFFS_OFFSET);
return; if (!isFullImage) {
} // Regular firmware update
if (!Update.begin(contentLength)) {
spiffsSize = SPIFFS_SIZE; Serial.printf("Not enough space: %u required\n", contentLength);
spiffsBuffer = (uint8_t*)malloc(spiffsSize); request->send(400, "application/json", "{\"status\":\"error\",\"message\":\"Not enough space available\"}");
if (!spiffsBuffer) { return;
Serial.println("Failed to allocate SPIFFS buffer");
request->send(400, "application/json", "{\"status\":\"error\",\"message\":\"Memory allocation failed\"}");
return;
}
memset(spiffsBuffer, 0xFF, spiffsSize);
Serial.println("Full image update started");
} else {
if (!Update.begin(contentLength, U_FLASH)) {
Serial.printf("Not enough space: %u required\n", contentLength);
request->send(400, "application/json", "{\"status\":\"error\",\"message\":\"Not enough space available\"}");
return;
}
Serial.println("Firmware update started");
} }
Serial.println("Firmware update started");
} else { } else {
request->send(400, "application/json", "{\"status\":\"error\",\"message\":\"Invalid image format\"}"); // Full image update - start with bootloader
return; if (!Update.begin(contentLength, U_FLASH)) {
Serial.printf("Not enough space for full image: %u required\n", contentLength);
request->send(400, "application/json", "{\"status\":\"error\",\"message\":\"Not enough space available\"}");
return;
}
Serial.println("Full image update started");
} }
currentOffset = 0; currentOffset = 0;
} }
if (isFullImage) { // Write data
// Firmware Teil if (Update.write(data, len) != len) {
if (currentOffset < SPIFFS_START) { String errorMsg = Update.errorString();
size_t writeLen = min(len, (size_t)(SPIFFS_START - currentOffset)); if (errorMsg != "No Error") {
if (Update.write(data, writeLen) != writeLen) { Update.printError(Serial);
String errorMsg = Update.errorString(); request->send(400, "application/json", "{\"status\":\"error\",\"message\":\"Error writing update: " + errorMsg + "\"}");
if (errorMsg != "No Error") { return;
Update.printError(Serial);
if (spiffsBuffer) free(spiffsBuffer);
request->send(400, "application/json", "{\"status\":\"error\",\"message\":\"Error writing firmware: " + errorMsg + "\"}");
return;
}
}
}
// SPIFFS Teil
if (currentOffset + len > SPIFFS_START) {
uint32_t spiffsOffset = (currentOffset >= SPIFFS_START) ?
(currentOffset - SPIFFS_START) : 0;
uint32_t dataOffset = (currentOffset < SPIFFS_START) ?
(SPIFFS_START - currentOffset) : 0;
if (spiffsOffset < spiffsSize) {
size_t copyLen = min(len - dataOffset, spiffsSize - spiffsOffset);
memcpy(spiffsBuffer + spiffsOffset, data + dataOffset, copyLen);
Serial.printf("Collecting SPIFFS data: offset=%u, len=%u\n", spiffsOffset, copyLen);
}
}
} else {
if (Update.write(data, len) != len) {
String errorMsg = Update.errorString();
if (errorMsg != "No Error") {
Update.printError(Serial);
request->send(400, "application/json", "{\"status\":\"error\",\"message\":\"Error writing update: " + errorMsg + "\"}");
return;
}
} }
} }
currentOffset += len; currentOffset += len;
if (final) { if (final) {
if (Update.end(true)) { if (Update.end(true)) {
Serial.println("Firmware update complete"); Serial.println("Update complete");
if (isFullImage && spiffsBuffer) {
Serial.println("Starting SPIFFS update");
if (SPIFFS.begin(true)) {
SPIFFS.end();
}
if (!Update.begin(spiffsSize, U_SPIFFS)) {
Serial.println("Could not begin SPIFFS update");
free(spiffsBuffer);
request->send(400, "application/json", "{\"status\":\"error\",\"message\":\"SPIFFS update initialization failed\"}");
return;
}
Serial.printf("Writing SPIFFS data: %u bytes\n", spiffsSize);
if (Update.write(spiffsBuffer, spiffsSize) != spiffsSize) {
Serial.println("SPIFFS Write Failed");
Update.printError(Serial);
free(spiffsBuffer);
request->send(400, "application/json", "{\"status\":\"error\",\"message\":\"SPIFFS write failed\"}");
return;
}
if (!Update.end(true)) {
Serial.println("SPIFFS End Failed");
Update.printError(Serial);
free(spiffsBuffer);
request->send(400, "application/json", "{\"status\":\"error\",\"message\":\"SPIFFS finish failed\"}");
return;
}
free(spiffsBuffer);
spiffsBuffer = nullptr;
Serial.println("SPIFFS update complete");
if (!SPIFFS.begin(true)) {
Serial.println("SPIFFS mount after update failed");
request->send(400, "application/json", "{\"status\":\"error\",\"message\":\"SPIFFS verification failed\"}");
return;
}
SPIFFS.end();
}
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(1000); delay(1000);
ESP.restart(); ESP.restart();
} else { } else {
if (spiffsBuffer) {
free(spiffsBuffer);
spiffsBuffer = nullptr;
}
String errorMsg = Update.errorString(); String errorMsg = Update.errorString();
if (errorMsg != "No Error") { if (errorMsg != "No Error") {
Update.printError(Serial); Update.printError(Serial);