From 5705e612b7efb51e910845afc29e3e9a405774b5 Mon Sep 17 00:00:00 2001 From: Charles Garwood Date: Tue, 6 Oct 2020 09:23:55 -0400 Subject: [PATCH] Protect add/remove node services from invalid instance ID --- homeassistant/components/ozw/services.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/homeassistant/components/ozw/services.py b/homeassistant/components/ozw/services.py index 9f66c3c33fd8cf..e1f71e636b3efd 100644 --- a/homeassistant/components/ozw/services.py +++ b/homeassistant/components/ozw/services.py @@ -111,6 +111,8 @@ def async_add_node(self, service): instance_id = service.data[const.ATTR_INSTANCE_ID] secure = service.data[const.ATTR_SECURE] instance = self._manager.get_instance(instance_id) + if instance is None: + raise ValueError(f"No OpenZWave Instance with ID {instance_id}") instance.add_node(secure) @callback @@ -118,6 +120,8 @@ def async_remove_node(self, service): """Enter exclusion mode on the controller.""" instance_id = service.data[const.ATTR_INSTANCE_ID] instance = self._manager.get_instance(instance_id) + if instance is None: + raise ValueError(f"No OpenZWave Instance with ID {instance_id}") instance.remove_node() @callback