mirror of
https://github.com/yyhuni/xingrin.git
synced 2026-02-03 21:23:15 +08:00
47 lines
1.2 KiB
Python
47 lines
1.2 KiB
Python
"""
|
|
Vuln Scan 共享工具函数
|
|
"""
|
|
|
|
import logging
|
|
import subprocess
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
def calculate_timeout_by_line_count(
|
|
tool_config: dict,
|
|
file_path: str,
|
|
base_per_time: int = 1,
|
|
) -> int:
|
|
"""
|
|
根据文件行数自动计算超时时间
|
|
|
|
Args:
|
|
tool_config: 工具配置（保留参数，未来可能用于更复杂的计算）
|
|
file_path: 输入文件路径
|
|
base_per_time: 每行的基础时间（秒）
|
|
|
|
Returns:
|
|
int: 计算出的超时时间（秒）
|
|
"""
|
|
try:
|
|
result = subprocess.run(
|
|
["wc", "-l", file_path],
|
|
capture_output=True,
|
|
text=True,
|
|
check=True,
|
|
)
|
|
line_count = int(result.stdout.strip().split()[0])
|
|
timeout = line_count * base_per_time
|
|
logger.info(
|
|
"timeout 自动计算: 文件=%s, 行数=%d, 每行时间=%d秒, timeout=%d秒",
|
|
file_path,
|
|
line_count,
|
|
base_per_time,
|
|
timeout,
|
|
)
|
|
return timeout
|
|
except Exception as e:
|
|
logger.error("wc -l 计算行数失败: %s", e)
|
|
raise RuntimeError(f"自动计算超时时间失败: {e}") from e
|