diff options
| author | Artem Belevich <artemb@gmail.com> | 2019-06-02 08:41:15 -0700 |
|---|---|---|
| committer | KevinOConnor <kevin@koconnor.net> | 2019-06-02 11:41:15 -0400 |
| commit | 72100bc1f2c2825248f32ee512612a7ac803f4a6 (patch) | |
| tree | 9f6abbb915f736a971723e33c1c4c64573a9dee3 /klippy/extras/tmc2660.py | |
| parent | 3910a44cc9762ea674f1c43237ade1f0a2e941da (diff) | |
| download | kutter-72100bc1f2c2825248f32ee512612a7ac803f4a6.tar.gz kutter-72100bc1f2c2825248f32ee512612a7ac803f4a6.tar.xz kutter-72100bc1f2c2825248f32ee512612a7ac803f4a6.zip | |
tmc2130: Check field name validity. (#1697)
This avoids crashing Klipper if user entered invalid field name.
Signed-off-by: Artem Belevich <artemb@gmail.com>
Diffstat (limited to 'klippy/extras/tmc2660.py')
| -rw-r--r-- | klippy/extras/tmc2660.py | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/klippy/extras/tmc2660.py b/klippy/extras/tmc2660.py index 721fbe43..7f0c95d1 100644 --- a/klippy/extras/tmc2660.py +++ b/klippy/extras/tmc2660.py @@ -294,7 +294,9 @@ class TMC2660: '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) pt = self.printer.lookup_object('toolhead').get_last_move_time() |