From f371a4cb837defee4d9e437cde357b684abc0ac1 Mon Sep 17 00:00:00 2001
From: 2Shirt <2xShirt@gmail.com>
Date: Sat, 4 Feb 2023 17:20:06 -0800
Subject: [PATCH] Limit KVRT scan to Users and ProgramData folders

---
 scripts/wk/repairs/win.py | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/scripts/wk/repairs/win.py b/scripts/wk/repairs/win.py
index 33cd7cbb..a26257a6 100644
--- a/scripts/wk/repairs/win.py
+++ b/scripts/wk/repairs/win.py
@@ -88,6 +88,7 @@ GPUPDATE_SUCCESS_STRINGS = (
   )
 IN_CONEMU = 'ConEmuPID' in os.environ
 MENU_PRESETS = Menu()
+PROGRAMDATA = os.environ.get('{ALLUSERSPROFILE}', r'C:\ProgramData')
 PROGRAMFILES_32 = os.environ.get(
   'PROGRAMFILES(X86)', os.environ.get(
     'PROGRAMFILES', r'C:\Program Files (x86)',
@@ -1031,12 +1032,18 @@ def run_kvrt():
   log_path.parent.mkdir(parents=True, exist_ok=True)
   quarantine_path = set_quarantine_path('KVRT')
   quarantine_path.mkdir(parents=True, exist_ok=True)
+  report_path = quarantine_path.joinpath('Reports')
+  report_path.mkdir(parents=True, exist_ok=True)
+  scan_list = log_path.with_suffix('.lst')
+  scan_list.write_text(
+    f'{PROGRAMDATA}\n{SYSTEMDRIVE}\\Users', encoding='utf-8',
+    )
   cmd_args = (
     '-accepteula',
     '-d', str(quarantine_path),
     '-dontencrypt', '-fixednames',
     '-processlevel', '1',
-    '-custom', SYSTEMDRIVE,
+    '-customlist', str(scan_list),
     '-silent', '-adinsilent',
     )