PR suggestion commit incorporation

Fixes for PR (not to include in release):
    Guard os.cpu_count() None to avoid crash.
    Enumerate subkeys without the 1024 cap; use QueryInfoKey.
    Close the registry handle (resource leak) and tighten the call.

Signed-off-by: Shahm Najeeb <[email protected]>

Author: DefinetlyNotAI

CODE/config.ini

@@ -102,12 +102,11 @@ max_retry_time = 30
 ###################################################
 
 [VulnScan Settings]
-# Following extensions to be ignored by the model
-# In MB, max file size that the model is allowed to scan, if commented out disables the limit, you can also just say None
+# Max characters of text from each file to analyze. Set an integer or None to disable truncation.
 text_char_limit = None
 # Max workers to be used, either integer or use "auto" to make it decide the best value
 max_workers = auto
-# Sensitivity threshold for the model to flag something as a sensitive
+# Sensitivity threshold (0.0–1.0) for the model to flag content as sensitive
 threshold = 0.6
 # Paths for required files
 model = vulnscan/Model_SenseMacro.4n1.pth

CODE/usb_history.py

@@ -25,35 +25,35 @@ def _save_history(self, message: str):
     @staticmethod
     def _get_last_write_time(root_key, sub_key_path):
         """Return the precise last write time of a registry key, or None on failure."""
+        handle = ctypes.wintypes.HANDLE()
         try:
-            handle = ctypes.wintypes.HANDLE()
-            if ctypes.windll.advapi32.RegOpenKeyExW(root_key, sub_key_path, 0, winreg.KEY_READ,
-                                                    ctypes.byref(handle)) != 0:
+            advapi32 = ctypes.windll.advapi32
+            if advapi32.RegOpenKeyExW(root_key, sub_key_path, 0, winreg.KEY_READ, ctypes.byref(handle)) != 0:
                 return None
-
             ft = ctypes.wintypes.FILETIME()
-            if ctypes.windll.advapi32.RegQueryInfoKeyW(handle, None, None, None, None, None, None, None, None, None,
-                                                       None, ctypes.byref(ft)) != 0:
+            if advapi32.RegQueryInfoKeyW(handle, None, None, None, None, None, None, None, None, None, None,
+                                         ctypes.byref(ft)) != 0:
                 return None
-
             t = ((ft.dwHighDateTime << 32) + ft.dwLowDateTime) // 10
             return datetime(1601, 1, 1) + timedelta(microseconds=t)
-        except Exception:
-            return None
+        finally:
+            if handle:
+                ctypes.windll.advapi32.RegCloseKey(handle)
 
     @staticmethod
     def _enum_subkeys(root, path, warn_func):
         """Yield all subkeys of a registry key, logging warnings on errors."""
         try:
             with winreg.OpenKey(root, path) as key:
-                for i in range(1024):  # reasonable upper bound to avoid infinite loops
+                subkey_count, _, _ = winreg.QueryInfoKey(key)
+                for i in range(subkey_count):
                     try:
                         yield winreg.EnumKey(key, i)
                     except OSError as e:
-                        if e.winerror == 259:  # no more data
+                        if getattr(e, "winerror", None) == 259:  # ERROR_NO_MORE_ITEMS
                             break
                         warn_func(f"Error enumerating {path} index {i}: {e}")
-        except Exception as e:
+        except OSError as e:
             warn_func(f"Failed to open registry key {path}: {e}")
 
     @staticmethod

CODE/vulnscan.py

@@ -22,7 +22,7 @@
 MAX_TEXT_LENGTH = int(MAX_TEXT_LENGTH) if MAX_TEXT_LENGTH not in (None, "None", "") else None
 # Threading
 NUM_WORKERS = config.get("VulnScan Settings", "max_workers", fallback="auto")
-NUM_WORKERS = min(32, os.cpu_count() * 2) if NUM_WORKERS == "auto" else int(NUM_WORKERS)
+NUM_WORKERS = min(32, (os.cpu_count() or 1) * 2) if NUM_WORKERS == "auto" else int(NUM_WORKERS)
 # Classification threshold
 SENSITIVE_THRESHOLD = float(
     config.get("VulnScan Settings", "threshold", fallback=0.6))  # Probability cutoff to consider a file sensitive