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

preen

Browse Source
This commit is contained in:
J. Nick Koston 2025-06-29 21:41:57 -05:00
parent 8c8dd7b4bc
commit 004f4b51d1
No known key found for this signature in database
1 changed files with 14 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
esphome/components/web_server_idf/web_server_idf.cpp
View File

@ -569,6 +569,14 @@ void AsyncEventSourceResponse::deferrable_send_state(void *source, const char *e
#ifdef USE_WEBSERVER_OTA #ifdef USE_WEBSERVER_OTA
esp_err_t AsyncWebServer::handle_multipart_upload_(httpd_req_t *r, const char *content_type) { esp_err_t AsyncWebServer::handle_multipart_upload_(httpd_req_t *r, const char *content_type) {
// Constants for upload handling
static constexpr size_t MULTIPART_CHUNK_SIZE = 1460; // Match Arduino AsyncWebServer buffer size
static constexpr size_t YIELD_INTERVAL_BYTES = 16 * 1024; // Yield every 16KB to prevent watchdog
// Upload indices for handleUpload callbacks
static constexpr size_t UPLOAD_INDEX_BEGIN = 0;
static constexpr size_t UPLOAD_INDEX_WRITE = 1;
static constexpr size_t UPLOAD_INDEX_END = 2;
// Parse boundary from content type // Parse boundary from content type
const char *boundary_start = nullptr; const char *boundary_start = nullptr;
size_t boundary_len = 0; size_t boundary_len = 0;
@ -617,29 +625,28 @@ esp_err_t AsyncWebServer::handle_multipart_upload_(httpd_req_t *r, const char *c
} }
if (!ctx.started) { if (!ctx.started) {
ctx.handler->handleUpload(ctx.req, ctx.filename, 0, nullptr, 0, false); ctx.handler->handleUpload(ctx.req, ctx.filename, UPLOAD_INDEX_BEGIN, nullptr, 0, false);
ctx.started = true; ctx.started = true;
} }
ctx.handler->handleUpload(ctx.req, ctx.filename, 1, const_cast<uint8_t *>(data), len, false); ctx.handler->handleUpload(ctx.req, ctx.filename, UPLOAD_INDEX_WRITE, const_cast<uint8_t *>(data), len, false);
}); });
reader.set_part_complete_callback([&ctx]() { reader.set_part_complete_callback([&ctx]() {
if (ctx.started) { if (ctx.started) {
ctx.handler->handleUpload(ctx.req, ctx.filename, 2, nullptr, 0, true); ctx.handler->handleUpload(ctx.req, ctx.filename, UPLOAD_INDEX_END, nullptr, 0, true);
ctx.filename.clear(); ctx.filename.clear();
ctx.started = false; ctx.started = false;
} }
}); });
// Process chunks // Process chunks
static constexpr size_t CHUNK_SIZE = 1460; std::unique_ptr<char[]> buffer(new char[MULTIPART_CHUNK_SIZE]);
std::unique_ptr<char[]> buffer(new char[CHUNK_SIZE]);
size_t remaining = r->content_len; size_t remaining = r->content_len;
size_t bytes_since_yield = 0; size_t bytes_since_yield = 0;
while (remaining > 0) { while (remaining > 0) {
size_t to_read = std::min(remaining, CHUNK_SIZE); size_t to_read = std::min(remaining, MULTIPART_CHUNK_SIZE);
int recv_len = httpd_req_recv(r, buffer.get(), to_read); int recv_len = httpd_req_recv(r, buffer.get(), to_read);
if (recv_len <= 0) { if (recv_len <= 0) {
@ -659,7 +666,7 @@ esp_err_t AsyncWebServer::handle_multipart_upload_(httpd_req_t *r, const char *c
bytes_since_yield += recv_len; bytes_since_yield += recv_len;
// Yield periodically to let main loop run // Yield periodically to let main loop run
if (bytes_since_yield > 16 * 1024) { if (bytes_since_yield > YIELD_INTERVAL_BYTES) {
vTaskDelay(1); vTaskDelay(1);
bytes_since_yield = 0; bytes_since_yield = 0;
} }