sascha/esphome
1
0
Fork 0
mirror of https://github.com/esphome/esphome.git synced 2025-10-31 15:12:06 +00:00
Code Issues Releases Wiki Activity

web_server_ support for ota backend idf

Browse Source
This commit is contained in:
J. Nick Koston
2025-07-01 10:05:23 -05:00
parent 93dc5765bb
commit 649ad47e62
2 changed files with 11 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
esphome/components/ota_base/ota_backend_esp_idf.cpp
View File

@@ -67,7 +67,10 @@ OTAResponseTypes IDFOTABackend::begin(size_t image_size) {
return OTA_RESPONSE_OK; return OTA_RESPONSE_OK;
} }
void IDFOTABackend::set_update_md5(const char *expected_md5) { memcpy(this->expected_bin_md5_, expected_md5, 32); } void IDFOTABackend::set_update_md5(const char *expected_md5) {
memcpy(this->expected_bin_md5_, expected_md5, 32);
this->md5_set_ = true;
}
OTAResponseTypes IDFOTABackend::write(uint8_t *data, size_t len) { OTAResponseTypes IDFOTABackend::write(uint8_t *data, size_t len) {
esp_err_t err = esp_ota_write(this->update_handle_, data, len); esp_err_t err = esp_ota_write(this->update_handle_, data, len);
@@ -84,10 +87,12 @@ OTAResponseTypes IDFOTABackend::write(uint8_t *data, size_t len) {
} }
OTAResponseTypes IDFOTABackend::end() { OTAResponseTypes IDFOTABackend::end() {
this->md5_.calculate(); if (this->md5_set_) {
if (!this->md5_.equals_hex(this->expected_bin_md5_)) { this->md5_.calculate();
this->abort(); if (!this->md5_.equals_hex(this->expected_bin_md5_)) {
return OTA_RESPONSE_ERROR_MD5_MISMATCH; this->abort();
return OTA_RESPONSE_ERROR_MD5_MISMATCH;
}
} }
esp_err_t err = esp_ota_end(this->update_handle_); esp_err_t err = esp_ota_end(this->update_handle_);
this->update_handle_ = 0; this->update_handle_ = 0;

1
esphome/components/ota_base/ota_backend_esp_idf.h
View File

@@ -24,6 +24,7 @@ class IDFOTABackend : public OTABackend {
const esp_partition_t *partition_; const esp_partition_t *partition_;
md5::MD5Digest md5_{}; md5::MD5Digest md5_{};
char expected_bin_md5_[32]; char expected_bin_md5_[32];
bool md5_set_{false};
}; };
} // namespace ota_base } // namespace ota_base