Mirrors/inceptor
6
0
Fork 0
mirror of https://github.com/klezVirus/inceptor.git synced 2024-06-25 16:28:35 +00:00
Code Issues Projects Releases Wiki Activity

Migrated prints to Console methods

Browse Source
This commit is contained in:
klezVirus 2021-09-21 09:26:21 +01:00
parent 7ac791f9f8
commit 06bb7436b0
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
inceptor/utils/ThreatCheck.py
View File

@ -13,6 +13,7 @@ class ThreatCheck:
def __init__(self):
self.engines = ["Defender", "AMSI"]
self.path = str(Config().get_path("DIRECTORIES", "libs").joinpath("ThreatCheck.exe"))
self.debug = Config().get_boolean("DEBUG", "utilities")
def check(self, filename):
Console.auto_line(f"[+] Starting ThreatCheck Scan At {datetime.utcnow()}")
@ -22,8 +23,8 @@ class ThreatCheck:
for engine in self.engines:
try:
cmd = f"\"{self.path}\" -f {filename} -e {engine}"
if Config().get_boolean("DEBUG", "utilities"):
print(cmd)
if self.debug:
Console.auto_line(f" [>] ThreatCheck cmdline: {cmd}")
output = subprocess.check_output(cmd, stderr=subprocess.STDOUT).decode().rstrip()
if output.find("No threat found") >= 0:
Console.auto_line(f" [+] SUCCESS: {engine} Bypassed!")
@ -34,6 +35,6 @@ class ThreatCheck:
except subprocess.CalledProcessError as e:
for line in e.output.decode().split("\n"):
if re.search(r"error", line):
print(f" [-] Error: {line}")
Console.auto_line(f" [-] Error: {line}")
continue
Console.auto_line(f"[+] ThreatCheck Scan Finished At {datetime.utcnow()}")