sangge/BackDoorBuster
sangge/BackDoorBuster
1
0
Fork 0
Code Issues Pull Requests Projects 1 Activity

style: 添加扫描动画

Browse Source
This commit is contained in:
dqy 2024-06-02 20:24:03 +08:00
parent b99334ed12
commit 7eb4de8e6c
1 changed files with 27 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

40
detection/__main__.py
View File

@ -9,6 +9,7 @@ from .pyc_detection import disassemble_pyc
from .utils import *
import sys
from colorama import init, Fore, Style
from tqdm import tqdm
PYCDC_FLAG = True
PYCDC_ADDR_FLAG = True
@ -361,27 +362,40 @@ def process_path(
):
results = {"high": [], "medium": [], "low": [], "none": []}
if os.path.isdir(path):
# 获取所有文件
all_files = []
for root, dirs, files in os.walk(path):
for file in files:
file_extension = os.path.splitext(file)[1]
if file_extension in SUPPORTED_EXTENSIONS:
file_path = os.path.join(root, file)
file_results = checkModeAndDetect(
mode, file_path, file_extension, pycdc_addr
)
if file_results is not None:
for key in file_results:
if key != "none": # Exclude 'none' risk level
results[key].extend(
[
(f"{file_path}: Line {line_num}", line)
for line_num, line in file_results[key]
]
)
all_files.append(file_path)
# 扫描动画
for file_path in tqdm(all_files, desc="Scanning files", unit="file"):
file_extension = os.path.splitext(file_path)[1]
file_results = checkModeAndDetect(
mode, file_path, file_extension, pycdc_addr
)
if file_results is not None:
for key in file_results:
if key != "none": # Exclude 'none' risk level
results[key].extend(
[
(f"{file_path}: Line {line_num}", line)
for line_num, line in file_results[key]
]
)
elif os.path.isfile(path):
file_results = ""
file_extension = os.path.splitext(path)[1]
if file_extension in SUPPORTED_EXTENSIONS:
file_results = checkModeAndDetect(mode, path, file_extension, pycdc_addr)
# 扫描动画
with tqdm(total=100, desc="Scanning file", unit="%", ncols=100) as pbar:
for i in range(10):
file_results = checkModeAndDetect(
mode, path, file_extension, pycdc_addr
)
pbar.update(10) # Update the progress bar by 10%
if file_results is not None:
for key in file_results:
if key != "none": # Exclude 'none' risk level