Fixing bug when key does not exist

fortra · Oct 21, 2024 · a16f4b8 · a16f4b8
1 parent 6c1c618
commit a16f4b8
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/impacket/winregistry.py b/impacket/winregistry.py
@@ -660,8 +660,11 @@ def enumKey(self, key):
     def enumValues(self,key):
         path = self.__keyToNodePath(key)
         node = self.__findNode(path)
-        values = list(node.data.keys())
-        return [s.encode('utf-8') for s in values]
+        if not node:
+            return None
+        else:
+            values = list(node.data.keys())
+            return [s.encode('utf-8') for s in values]
 
     def getValue(self, keyValue, valueName=None):
         """ returns a tuple with (ValueType, ValueData) for the requested keyValue