arduino/esp8266
1
0
Fork 0
mirror of https://github.com/esp8266/Arduino.git synced 2025-06-06 05:21:22 +03:00
Code

Add an option to reboot on failed update

Browse Source
This commit is contained in:
John Doe 2015-07-01 15:38:30 +03:00 committed by Ivan Grokhotkov
parent 03a2b4808b
commit 2e08c5d797
2 changed files with 11 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
cores/esp8266/Esp.cpp
View File

@ -358,10 +358,12 @@ uint32_t EspClass::getFreeSketchSpace() {
return freeSpaceEnd - freeSpaceStart;
}
bool EspClass::updateSketch(Stream& in, uint32_t size) {
bool EspClass::updateSketch(Stream& in, uint32_t size, bool restartOnFail) {
if (size > getFreeSketchSpace())
if (size > getFreeSketchSpace()){
if(restartOnFail) ESP.restart();
return false;
}
uint32_t usedSize = getSketchSize();
uint32_t freeSpaceStart = (usedSize + FLASH_SECTOR_SIZE - 1) & (~(FLASH_SECTOR_SIZE - 1));
@ -374,8 +376,10 @@ bool EspClass::updateSketch(Stream& in, uint32_t size) {
noInterrupts();
int rc = SPIEraseAreaEx(freeSpaceStart, roundedSize);
interrupts();
if (rc)
if (rc){
if(restartOnFail) ESP.restart();
return false;
}
#ifdef DEBUG_SERIAL
DEBUG_SERIAL.println("erase done");
@ -397,12 +401,14 @@ bool EspClass::updateSketch(Stream& in, uint32_t size) {
#ifdef DEBUG_SERIAL
DEBUG_SERIAL.println("stream read failed");
#endif
if(restartOnFail) ESP.restart();
return false;
}
if(addr == freeSpaceStart) {
// check for valid first magic byte
if(*((uint8 *) buffer.get()) != 0xE9) {
if(restartOnFail) ESP.restart();
return false;
}
}
@ -414,6 +420,7 @@ bool EspClass::updateSketch(Stream& in, uint32_t size) {
#ifdef DEBUG_SERIAL
DEBUG_SERIAL.println("write failed");
#endif
if(restartOnFail) ESP.restart();
return false;
}

2
cores/esp8266/Esp.h
View File

@ -116,7 +116,7 @@ class EspClass {
uint32_t getSketchSize();
uint32_t getFreeSketchSpace();
bool updateSketch(Stream& in, uint32_t size);
bool updateSketch(Stream& in, uint32_t size, bool restartOnFail = false);
String getResetInfo();
struct rst_info * getResetInfoPtr();