From 03000662fe6b683b0d1271107bbc47782464278c Mon Sep 17 00:00:00 2001
From: 2Shirt <2xShirt@gmail.com>
Date: Sat, 1 May 2021 17:12:11 -0600
Subject: [PATCH] Add KVRT sections

---
 scripts/auto_repairs.py   |  2 +-
 scripts/wk/repairs/win.py | 26 ++++++++++++++++++++++++++
 2 files changed, 27 insertions(+), 1 deletion(-)

diff --git a/scripts/auto_repairs.py b/scripts/auto_repairs.py
index ad229c3a..d199d918 100644
--- a/scripts/auto_repairs.py
+++ b/scripts/auto_repairs.py
@@ -76,7 +76,7 @@ BASE_MENUS = {
     'Malware Cleanup': (
       MenuEntry('BleachBit',                'auto_bleachbit'),
       MenuEntry('HitmanPro',                'auto_hitmanpro'),
-      MenuEntry('KVRT',                     placeholder_function),
+      MenuEntry('KVRT',                     'auto_kvrt'),
       MenuEntry('Windows Defender',         placeholder_function),
       MenuEntry('Reboot',                   'auto_reboot'),
       ),
diff --git a/scripts/wk/repairs/win.py b/scripts/wk/repairs/win.py
index 1d217430..d77d19d1 100644
--- a/scripts/wk/repairs/win.py
+++ b/scripts/wk/repairs/win.py
@@ -628,6 +628,12 @@ def auto_hitmanpro(group, name):
   save_settings(group, name, result=result)
 
 
+def auto_kvrt(group, name):
+  """Run KVRT scan."""
+  result = TRY_PRINT.run('KVRT...', run_kvrt, msg_good='DONE')
+  save_settings(group, name, result=result)
+
+
 def auto_reboot(group, name):
   """Reboot the system."""
   save_settings(group, name, done=True, failed=False, message='DONE')
@@ -771,6 +777,26 @@ def run_hitmanpro():
     )
 
 
+def run_kvrt():
+  """Run KVRT scan."""
+  log_path = format_log_path(log_name='KVRT', timestamp=True, tool=True)
+  log_path.parent.mkdir(parents=True, exist_ok=True)
+  quarantine_path = set_local_storage_path(
+    'Quarantine', 'KVRT', date=True,
+    )
+  quarantine_path.mkdir(parents=True, exist_ok=True)
+  cmd_args = (
+    '-accepteula',
+    '-d', quarantine_path,
+    '-dontencrypt', '-fixednames',
+    '-processlevel', '1',
+    '-custom', SYSTEMDRIVE,
+    '-silent', '-adinsilent',
+    )
+  proc = run_tool('KVRT', 'KVRT', *cmd_args, download=True)
+  log_path.write_text(proc.stdout)
+
+
 def run_rkill():
   """Run RKill scan."""
   log_path = format_log_path(log_name='RKill', timestamp=True, tool=True)