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/tmc2208.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/tmc2208.py')
| -rw-r--r-- | klippy/extras/tmc2208.py | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/klippy/extras/tmc2208.py b/klippy/extras/tmc2208.py index 4134f026..dcd7b890 100644 --- a/klippy/extras/tmc2208.py +++ b/klippy/extras/tmc2208.py @@ -427,7 +427,9 @@ class TMC2208: '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) self.printer.lookup_object('toolhead').wait_moves() |