Spaces:
Running
on
Zero
Running
on
Zero
import sys | |
import subprocess | |
import os | |
def security_check(): | |
print("[START] Security scan") | |
custom_nodes_path = os.path.abspath(os.path.join(os.path.dirname(__file__), '..', '..')) | |
comfyui_path = os.path.abspath(os.path.join(custom_nodes_path, '..')) | |
guide = { | |
"ComfyUI_LLMVISION": """ | |
0.Remove ComfyUI\\custom_nodes\\ComfyUI_LLMVISION. | |
1.Remove pip packages: openai-1.16.3.dist-info, anthropic-0.21.4.dist-info, openai-1.30.2.dist-info, anthropic-0.21.5.dist-info, anthropic-0.26.1.dist-info, %LocalAppData%\\rundll64.exe | |
(For portable versions, it is recommended to reinstall. If you are using a venv, it is advised to recreate the venv.) | |
2.Remove these files in your system: lib/browser/admin.py, Cadmino.py, Fadmino.py, VISION-D.exe, BeamNG.UI.exe | |
3.Check your Windows registry for the key listed above and remove it. | |
(HKEY_CURRENT_USER\\Software\\OpenAICLI) | |
4.Run a malware scanner. | |
5.Change all of your passwords, everywhere. | |
(Reinstall OS is recommended.) | |
\n | |
Detailed information: https://old.reddit.com/r/comfyui/comments/1dbls5n/psa_if_youve_used_the_comfyui_llmvision_node_from/ | |
""", | |
"lolMiner": """ | |
1. Remove pip packages: lolMiner* | |
2. Remove files: lolMiner*, 4G_Ethash_Linux_Readme.txt, mine* in ComfyUI dir. | |
(Reinstall ComfyUI is recommended.) | |
""" | |
} | |
node_blacklist = {"ComfyUI_LLMVISION": "ComfyUI_LLMVISION"} | |
pip_blacklist = {"AppleBotzz": "ComfyUI_LLMVISION"} | |
file_blacklist = { | |
"ComfyUI_LLMVISION": ["%LocalAppData%\\rundll64.exe"], | |
"lolMiner": [os.path.join(comfyui_path, 'lolMiner')] | |
} | |
installed_pips = subprocess.check_output([sys.executable, '-m', "pip", "freeze"], text=True) | |
detected = set() | |
try: | |
anthropic_info = subprocess.check_output([sys.executable, '-m', "pip", "show", "anthropic"], text=True, stderr=subprocess.DEVNULL) | |
anthropic_reqs = [x for x in anthropic_info.split('\n') if x.startswith("Requires")][0].split(': ')[1] | |
if "pycrypto" in anthropic_reqs: | |
location = [x for x in anthropic_info.split('\n') if x.startswith("Location")][0].split(': ')[1] | |
for fi in os.listdir(location): | |
if fi.startswith("anthropic"): | |
guide["ComfyUI_LLMVISION"] = f"\n0.Remove {os.path.join(location, fi)}" + guide["ComfyUI_LLMVISION"] | |
detected.add("ComfyUI_LLMVISION") | |
except subprocess.CalledProcessError: | |
pass | |
for k, v in node_blacklist.items(): | |
if os.path.exists(os.path.join(custom_nodes_path, k)): | |
print(f"[SECURITY ALERT] custom node '{k}' is dangerous.") | |
detected.add(v) | |
for k, v in pip_blacklist.items(): | |
if k in installed_pips: | |
detected.add(v) | |
break | |
for k, v in file_blacklist.items(): | |
for x in v: | |
if os.path.exists(os.path.expandvars(x)): | |
detected.add(k) | |
break | |
if len(detected) > 0: | |
for line in installed_pips.split('\n'): | |
for k, v in pip_blacklist.items(): | |
if k in line: | |
print(f"[SECURITY ALERT] '{line}' is dangerous.") | |
print("\n########################################################################") | |
print(" Malware has been detected, forcibly terminating ComfyUI execution.") | |
print("########################################################################\n") | |
for x in detected: | |
print(f"\n======== TARGET: {x} =========") | |
print(f"\nTODO:") | |
print(guide.get(x)) | |
exit(-1) | |
print("[DONE] Security scan") | |