Improved firmware upgrade

lib/AmsConfiguration/include/AmsConfiguration.h

@@ -11,6 +11,7 @@
 
 #define CONFIG_SYSTEM_START 8
 #define CONFIG_METER_START 32
+#define CONFIG_UPGRADE_INFO_START 216
 #define CONFIG_UI_START 248
 #define CONFIG_GPIO_START 266
 #define CONFIG_ENTSOE_START 290
@@ -213,6 +214,13 @@ struct TempSensorConfig {
 	bool common;
 };
 
+struct UpgradeInformation {
+	char fromVersion[8];
+	char toVersion[8];
+	int16_t exitCode;
+	int16_t errorCode;
+}; // 20
+
 class AmsConfiguration {
 public:
 	bool hasConfig();
@@ -297,6 +305,10 @@ public:
 
     bool isSensorAddressEqual(uint8_t a[8], uint8_t b[8]);
 
+	bool getUpgradeInformation(UpgradeInformation&);
+	bool setUpgradeInformation(int16_t exitCode, int16_t errorCode, const char* currentVersion, const char* nextVersion);
+	void clearUpgradeInformation(UpgradeInformation&);
+
 	void clear();
 
 protected:

lib/AmsConfiguration/include/hexutils.h

@@ -7,6 +7,6 @@
 String toHex(uint8_t* in);
 String toHex(uint8_t* in, uint16_t size);
 void fromHex(uint8_t *out, String in, uint16_t size);
-void stripNonAscii(uint8_t* in, uint16_t size, bool extended = false);
+bool stripNonAscii(uint8_t* in, uint16_t size, bool extended = false);
 
 #endif

lib/AmsConfiguration/src/AmsConfiguration.cpp

@@ -679,6 +679,45 @@ void AmsConfiguration::clearUiConfig(UiConfig& config) {
 	config.showTemperaturePlot = 2;
 }
 
+bool AmsConfiguration::setUpgradeInformation(int16_t exitCode, int16_t errorCode, const char* currentVersion, const char* nextVersion) {
+	UpgradeInformation upinfo;
+	upinfo.exitCode = exitCode;
+	upinfo.errorCode = errorCode;
+	strcpy(upinfo.fromVersion, currentVersion);
+	strcpy(upinfo.toVersion, nextVersion);
+
+	stripNonAscii((uint8_t*) upinfo.fromVersion, 8);
+	stripNonAscii((uint8_t*) upinfo.toVersion, 8);
+
+	EEPROM.begin(EEPROM_SIZE);
+	EEPROM.put(CONFIG_UPGRADE_INFO_START, upinfo);
+	bool ret = EEPROM.commit();
+	EEPROM.end();
+	return ret;
+}
+
+bool AmsConfiguration::getUpgradeInformation(UpgradeInformation& upinfo) {
+	if(hasConfig()) {
+		EEPROM.begin(EEPROM_SIZE);
+		EEPROM.get(CONFIG_UPGRADE_INFO_START, upinfo);
+		EEPROM.end();
+		if(stripNonAscii((uint8_t*) upinfo.fromVersion, 8) || stripNonAscii((uint8_t*) upinfo.toVersion, 8)) {
+			clearUpgradeInformation(upinfo);
+		}
+		return true;
+	} else {
+		clearUpgradeInformation(upinfo);
+		return false;
+	}
+}
+
+void AmsConfiguration::clearUpgradeInformation(UpgradeInformation& upinfo) {
+	upinfo.exitCode = -1;
+	upinfo.errorCode = 0;
+	memset(upinfo.fromVersion, 0, 8);
+	memset(upinfo.toVersion, 0, 8);
+}
+
 
 void AmsConfiguration::clear() {
 	EEPROM.begin(EEPROM_SIZE);
@@ -734,6 +773,10 @@ void AmsConfiguration::clear() {
 	clearUiConfig(ui);
 	EEPROM.put(CONFIG_UI_START, ui);
 
+	UpgradeInformation upinfo;
+	clearUpgradeInformation(upinfo);
+	EEPROM.put(CONFIG_UPGRADE_INFO_START, upinfo);
+
 	EEPROM.put(EEPROM_CONFIG_ADDRESS, EEPROM_CLEARED_INDICATOR);
 	EEPROM.commit();
 	EEPROM.end();

lib/AmsConfiguration/src/hexutils.cpp

@@ -22,7 +22,8 @@ void fromHex(uint8_t *out, String in, uint16_t size) {
 	}
 }
 
-void stripNonAscii(uint8_t* in, uint16_t size, bool extended) {
+bool stripNonAscii(uint8_t* in, uint16_t size, bool extended) {
+	bool ret = false;
 	for(uint16_t i = 0; i < size; i++) {
 		if(in[i] == 0) { // Clear the rest with null-terminator
 			memset(in+i, 0, size-i);
@@ -30,9 +31,12 @@ void stripNonAscii(uint8_t* in, uint16_t size, bool extended) {
 		}
 		if(extended && (in[i] < 32 || in[i] == 127 || in[i] == 129 || in[i] == 141 || in[i] == 143 || in[i] == 144 || in[i] == 157)) {
 			memset(in+i, ' ', 1);
+			ret = true;
 		} else if(in[i] < 32 || in[i] > 126) {
 			memset(in+i, ' ', 1);
+			ret = true;
 		}
 	}
 	memset(in+size-1, 0, 1); // Make sure the last character is null-terminator
+	return ret;
 }