From 85ebafd3f6c4e39f445ac7eac42bebc6587c1a7c Mon Sep 17 00:00:00 2001 From: Kevin O'Connor Date: Thu, 26 Sep 2024 00:47:10 -0400 Subject: configfile: Don't read the autosave data if multiple autosave headers present Also, verify new autosave looks valid prior to writing it. Signed-off-by: Kevin O'Connor --- klippy/configfile.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) (limited to 'klippy/configfile.py') diff --git a/klippy/configfile.py b/klippy/configfile.py index 5b9896a0..497f35ed 100644 --- a/klippy/configfile.py +++ b/klippy/configfile.py @@ -247,7 +247,7 @@ class ConfigAutoSave: regular_data = data[:pos] autosave_data = data[pos + len(AUTOSAVE_HEADER):].strip() # Check for errors and strip line prefixes - if "\n#*# " in regular_data: + if "\n#*# " in regular_data or autosave_data.find(AUTOSAVE_HEADER) >= 0: logging.warning("Can't read autosave from config file" " - autosave state corrupted") return data, "" @@ -361,6 +361,11 @@ class ConfigAutoSave: regular_data = self._strip_duplicates(regular_data, self.fileconfig) self._disallow_include_conflicts(regular_data, cfgname, gcode) data = regular_data.rstrip() + autosave_data + new_regular_data, new_autosave_data = self._find_autosave_data(data) + if not new_autosave_data: + raise gcode.error( + "Existing config autosave is corrupted." + " Can't complete SAVE_CONFIG") # Determine filenames datestr = time.strftime("-%Y%m%d_%H%M%S") backup_name = cfgname + datestr -- cgit v1.2.3-70-g09d2