From 72100bc1f2c2825248f32ee512612a7ac803f4a6 Mon Sep 17 00:00:00 2001 From: Artem Belevich Date: Sun, 2 Jun 2019 08:41:15 -0700 Subject: tmc2130: Check field name validity. (#1697) This avoids crashing Klipper if user entered invalid field name. Signed-off-by: Artem Belevich --- klippy/extras/tmc5160.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) (limited to 'klippy/extras/tmc5160.py') diff --git a/klippy/extras/tmc5160.py b/klippy/extras/tmc5160.py index b7f226b4..70736f2e 100644 --- a/klippy/extras/tmc5160.py +++ b/klippy/extras/tmc5160.py @@ -428,7 +428,9 @@ class TMC5160: 'VALUE' not in params): raise gcode.error("Invalid command format") field = gcode.get_str('FIELD', params) - reg = self.fields.field_to_register[field] + reg = self.fields.field_to_register.get(field) + if reg is None: + raise gcode.error("Unknown field name '%s'" % field) value = gcode.get_int('VALUE', params) self.fields.set_field(field, value) print_time = self.printer.lookup_object('toolhead').get_last_move_time() -- cgit v1.2.3-70-g09d2