86 lines
3.3 KiB
Python
86 lines
3.3 KiB
Python
"""采集 T-104 阶段 A 的一次 Back 后本机证据。"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import argparse
|
|
from math import isfinite
|
|
from pathlib import Path
|
|
import sys
|
|
|
|
|
|
CLIENT_ROOT = Path(__file__).resolve().parents[1]
|
|
sys.path.insert(0, str(CLIENT_ROOT / "src"))
|
|
|
|
from cmbuyer_client.device.adb import AdbClient, DeviceConnectionError, SubprocessAdbRunner
|
|
from cmbuyer_client.device.baseline import NoReconnectUiautomatorConnector
|
|
from cmbuyer_client.pdd.sku_selection import SkuSelectionError
|
|
from cmbuyer_client.pdd.sku_selection_runner import (
|
|
SkuExitSpikeCapturer,
|
|
SkuSelectionRunError,
|
|
)
|
|
|
|
|
|
def parse_arguments(argv: list[str] | None = None) -> argparse.Namespace:
|
|
parser = argparse.ArgumentParser(description="采集 T-104 阶段 A 的一次 Back 后本机证据。")
|
|
parser.add_argument("--serial", required=True, help="ADB device serial；禁止自动选择。")
|
|
parser.add_argument("--output-dir", required=True, type=Path, help="全新本机证据目录；不得覆盖。")
|
|
parser.add_argument("--timeout", type=float, default=10.0, help="ADB 与设备 RPC 超时（秒）。")
|
|
parser.add_argument("--adb", default="adb", help="adb 可执行文件路径。")
|
|
return parser.parse_args(argv)
|
|
|
|
|
|
def validate_arguments(arguments: argparse.Namespace) -> None:
|
|
if type(arguments.serial) is not str or not arguments.serial.strip():
|
|
raise ValueError("必须显式提供非空 --serial。")
|
|
if not isinstance(arguments.output_dir, Path) or not arguments.output_dir.name:
|
|
raise ValueError("--output-dir 必须是明确的全新目录。")
|
|
if (
|
|
not isinstance(arguments.timeout, (int, float))
|
|
or isinstance(arguments.timeout, bool)
|
|
or arguments.timeout <= 0
|
|
or not isfinite(arguments.timeout)
|
|
):
|
|
raise ValueError("--timeout 必须是大于 0 的有限数值。")
|
|
|
|
|
|
def main(argv: list[str] | None = None) -> int:
|
|
arguments = parse_arguments(argv)
|
|
try:
|
|
validate_arguments(arguments)
|
|
except ValueError as error:
|
|
print(f"失败：{error}", file=sys.stderr)
|
|
return 2
|
|
|
|
try:
|
|
import adbutils
|
|
import uiautomator2 as u2
|
|
except ImportError:
|
|
print("失败：缺少 uiautomator2；请在采购工具虚拟环境中运行。", file=sys.stderr)
|
|
return 2
|
|
|
|
capturer = SkuExitSpikeCapturer(
|
|
AdbClient(SubprocessAdbRunner(arguments.adb), timeout_seconds=arguments.timeout),
|
|
NoReconnectUiautomatorConnector(
|
|
adbutils.AdbClient(socket_timeout=arguments.timeout).device_list,
|
|
u2.connect,
|
|
),
|
|
timeout_seconds=arguments.timeout,
|
|
)
|
|
try:
|
|
capturer.capture(arguments.serial, arguments.output_dir)
|
|
except (DeviceConnectionError, SkuSelectionError, SkuSelectionRunError):
|
|
# 不回显第三方异常、serial、页面正文或本机路径。
|
|
print("规格安全退出取证失败：已停止，未发布本地证据目录。", file=sys.stderr)
|
|
return 1
|
|
except OSError:
|
|
print("规格安全退出取证失败：无法发布本地证据目录。", file=sys.stderr)
|
|
return 1
|
|
|
|
print("规格安全退出取证完成。")
|
|
print("人工复核：请在指定目录检查退出后截图、XML、应用摘要和 manifest。")
|
|
return 0
|
|
|
|
|
|
if __name__ == "__main__":
|
|
raise SystemExit(main())
|