Merge pull request #137 from fpreynaud/bugfix

Fix bug in WCC module
Pennyw0rth · Dec 8, 2023 · 76836dc · 76836dc
2 parents 388208d + 5c87c27
commit 76836dc
Showing 1 changed file with 4 additions and 8 deletions.
diff --git a/nxc/modules/wcc.py b/nxc/modules/wcc.py
@@ -588,16 +588,12 @@ def get_value(subkey_handle, dwIndex=0):
         subkey_handle = ans["phkResult"]
 
         if valueName is None:
-            _, _, data = get_value(subkey_handle)
+            return get_value(subkey_handle)[3]
         else:
-            found = False
-            for _, name, _data in subkey_values(subkey_handle):
+            for _, name, data in subkey_values(subkey_handle):
                 if name.upper() == valueName.upper():
-                    found = True
-                    break
-            if not found:
-                return DCERPCSessionError(error_code=ERROR_OBJECT_NOT_FOUND)
-        return data
+                    return data
+            return DCERPCSessionError(error_code=ERROR_OBJECT_NOT_FOUND)
 
     # Methods for getting values from SAMR and SCM #
     ################################################