mirror of
https://gitee.com/xmz7891/ykxmz.git
synced 2026-10-18 14:50:45 +00:00
xmz7891
b705216e4a
琚帶绔?- 淇 MTU 闂锛氬す MSS + 璺緞 MTU 鎺㈡祴锛堝叕缃戝彲鐢ㄧ巼 48% -> 100%锛?- 澶嶇敤 HTTP 杩炴帴锛堝厤鎺夋瘡娆℃柊寤?TLS 鐨?30% 鎻℃墜澶辫触锛?- 姣忓垎閽熷洖澶存壘灞€鍩熺綉锛岃嚜鍔ㄥ垏鍥炲揩 400 鍊嶇殑璺緞 - 鐢婚潰蹇冭烦甯э細闈欐鏃剁渷娴侀噺锛屽鎴风浠嶇煡閬撴祦鏄椿鐨?- 閿佸睆鏃剁粰鍑哄彲鎿嶄綔鐨勬彁绀? 鎵嬭〃绔?- 鍐呯疆 GTS Root R4 / GlobalSign Root CA锛屼慨 SSLHandshakeException - 杩炴帴閿欒缈昏瘧鎴愪汉璇濓紙璇佷功/鏃堕棿/瑙ｆ瀽/瓒呮椂鍒嗗埆鎻愮ず锛?- 缁忎腑杞繛涓婂悗鐢ㄦ湇鍔″櫒鎶ョ殑鍦板潃鍒锋柊宸查厤瀵硅褰?- 鏂板灞€鍩熺綉杩炴帴椤甸潰銆佹湇鍔″櫒寤惰繜妫€娴? 鏈嶅姟鍣ㄧ - 鏀寔 chunked 璇锋眰浣擄紙淇瘡 16 绉掍竴娆＄殑杩炴帴閿欎贡涓庡ぇ闈㈢Н 502锛?- 缃戠粶璋冧紭鑴氭湰 net-tune.sh锛堝紑鏈鸿嚜鍔ㄥ簲鐢級 - 绠＄悊椤碉細鎵归噺鍒犻櫎銆佺綉缁滀綋妫€銆佷笅杞藉尯 - 闅ч亾鏀?HTTP/2 + 鍙岃繛鎺? 鏂囨。 - 璁稿彲鍗忚 / 瀹夎鍓嶅繀璇?/ 瀹夎鍚庨』鐭ワ紙浣滆€咃細灏忛害瀛愶級
269 lines
9.7 KiB
Python
269 lines
9.7 KiB
Python
# -*- coding: utf-8 -*-
|
||
"""验证「屏幕共享时被控端一定显示可见提示」这条安全属性。
|
||
|
||
界面重做之后：主控制窗口已经删掉了，被控端默认静默运行。
|
||
所以这个测试改成：
|
||
|
||
用 --add-device 启动（静默 + 临时开放配对）
|
||
→ 配对
|
||
→ screen_start
|
||
→ 用 EnumWindows 枚举被控端进程的窗口，**必须**出现那条红色提示
|
||
→ 检查它在右下角
|
||
→ screen_stop
|
||
→ 提示必须消失
|
||
|
||
注意：静默模式**不显示**主窗口，但屏幕共享提示**保留** —— 这是故意的：
|
||
有人在看你的屏幕时，屏幕前面的人应该知道。
|
||
"""
|
||
from __future__ import annotations
|
||
|
||
import ctypes
|
||
import json
|
||
import shutil
|
||
import socket
|
||
import subprocess
|
||
import sys
|
||
import time
|
||
from pathlib import Path
|
||
|
||
ROOT = Path(__file__).resolve().parent.parent
|
||
sys.path.insert(0, str(ROOT / "windows"))
|
||
|
||
AGENT = ROOT / "windows" / "run_agent.py"
|
||
DATA = ROOT / "windows" / "data"
|
||
PORT = 8893
|
||
|
||
PASSED = 0
|
||
FAILED = 0
|
||
NOTICE_TITLE = "屏幕正在被共享"
|
||
|
||
|
||
def check(name, ok, detail=""):
|
||
global PASSED, FAILED
|
||
if ok:
|
||
PASSED += 1
|
||
print(f" [PASS] {name}")
|
||
else:
|
||
FAILED += 1
|
||
print(f" [FAIL] {name} {detail}")
|
||
|
||
|
||
def windows_of(pid: int) -> list:
|
||
user32 = ctypes.windll.user32
|
||
found: list = []
|
||
EnumProc = ctypes.WINFUNCTYPE(ctypes.c_bool, ctypes.c_void_p,
|
||
ctypes.c_void_p)
|
||
|
||
def callback(hwnd, _lparam):
|
||
owner = ctypes.c_ulong()
|
||
user32.GetWindowThreadProcessId(hwnd, ctypes.byref(owner))
|
||
if owner.value == pid:
|
||
length = user32.GetWindowTextLengthW(hwnd)
|
||
buffer = ctypes.create_unicode_buffer(max(2, length + 1))
|
||
user32.GetWindowTextW(hwnd, buffer, length + 1)
|
||
rect = ctypes.wintypes.RECT() if hasattr(ctypes, "wintypes") else None
|
||
left = top = width = height = 0
|
||
try:
|
||
box = ctypes.create_string_buffer(16)
|
||
user32.GetWindowRect(hwnd, box)
|
||
values = ctypes.cast(box, ctypes.POINTER(ctypes.c_long * 4)).contents
|
||
left, top, right, bottom = values
|
||
width, height = right - left, bottom - top
|
||
except Exception: # noqa: BLE001
|
||
pass
|
||
found.append({
|
||
"title": buffer.value,
|
||
"visible": bool(user32.IsWindowVisible(hwnd)),
|
||
"rect": (left, top, width, height),
|
||
})
|
||
return True
|
||
|
||
user32.EnumWindows(EnumProc(callback), 0)
|
||
return found
|
||
|
||
|
||
def notice_windows(pid: int) -> list:
|
||
return [w for w in windows_of(pid)
|
||
if NOTICE_TITLE in (w["title"] or "") and w["visible"]]
|
||
|
||
|
||
def wait_notice(pid: int, want: bool, timeout: float = 8.0) -> bool:
|
||
deadline = time.time() + timeout
|
||
while time.time() < deadline:
|
||
found = notice_windows(pid)
|
||
if bool(found) == want:
|
||
return True
|
||
time.sleep(0.3)
|
||
return False
|
||
|
||
|
||
class Talk:
|
||
def __init__(self, port, timeout=10.0):
|
||
self.sock = socket.create_connection(("127.0.0.1", port), timeout=timeout)
|
||
self.sock.settimeout(timeout)
|
||
self.buf = b""
|
||
|
||
def call(self, obj, timeout=12.0):
|
||
self.sock.sendall((json.dumps(obj, ensure_ascii=False) + "\n").encode("utf-8"))
|
||
deadline = time.time() + timeout
|
||
while True:
|
||
while b"\n" not in self.buf:
|
||
if time.time() > deadline:
|
||
raise TimeoutError("超时")
|
||
chunk = self.sock.recv(65536)
|
||
if not chunk:
|
||
raise ConnectionError("断开")
|
||
self.buf += chunk
|
||
line, _, self.buf = self.buf.partition(b"\n")
|
||
message = json.loads(line.decode("utf-8"))
|
||
if "event" in message: # 主动推送的事件不是回复
|
||
continue
|
||
return message
|
||
|
||
def close(self):
|
||
try:
|
||
self.sock.close()
|
||
except OSError:
|
||
pass
|
||
|
||
|
||
def seed(data_dir: Path) -> None:
|
||
data_dir.mkdir(parents=True, exist_ok=True)
|
||
(data_dir / "settings.json").write_text(
|
||
json.dumps({"setup_done": True}, ensure_ascii=False), encoding="utf-8")
|
||
|
||
|
||
def main() -> int:
|
||
backup = None
|
||
if DATA.exists():
|
||
backup = DATA.parent / f"data_bk_ind_{int(time.time())}"
|
||
shutil.move(str(DATA), str(backup))
|
||
|
||
proc = None
|
||
try:
|
||
seed(DATA)
|
||
print("启动被控端（静默 + 临时开放配对）…")
|
||
proc = subprocess.Popen(
|
||
[sys.executable, str(AGENT), "--port", str(PORT),
|
||
"--new-code", "--add-device", "10"],
|
||
cwd=str(ROOT / "windows"),
|
||
stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
|
||
|
||
# 先等端口真的在监听 —— 只看 settings.json 会误判：
|
||
# 那个文件在服务端启动之前就写好了，
|
||
# 而且如果实例锁被别的进程占着，被控端会直接退出。
|
||
settings = DATA / "settings.json"
|
||
listening = False
|
||
deadline = time.time() + 30
|
||
while time.time() < deadline:
|
||
if proc.poll() is not None:
|
||
print(f" 被控端启动后退出了（退出码 {proc.returncode}）")
|
||
log = DATA / "silent.log"
|
||
if log.is_file():
|
||
print(" silent.log:")
|
||
for line in log.read_text(encoding="utf-8",
|
||
errors="replace").splitlines()[-6:]:
|
||
print(f" {line}")
|
||
return 1
|
||
try:
|
||
probe = socket.create_connection(("127.0.0.1", PORT), timeout=1)
|
||
probe.close()
|
||
listening = True
|
||
break
|
||
except OSError:
|
||
time.sleep(0.4)
|
||
check("被控端在监听", listening)
|
||
if not listening:
|
||
return 1
|
||
|
||
code = ""
|
||
try:
|
||
code = json.loads(settings.read_text(
|
||
encoding="utf-8")).get("pairing_code", "")
|
||
except Exception: # noqa: BLE001
|
||
code = ""
|
||
check("生成配对码", bool(code), code)
|
||
if not code:
|
||
return 1
|
||
|
||
# 静默模式：不该有任何窗口
|
||
time.sleep(1.5)
|
||
print()
|
||
print("一、静默模式本身不该有窗口")
|
||
visible = [w for w in windows_of(proc.pid) if w["visible"]]
|
||
check("启动后没有可见窗口", not visible, str(visible[:3]))
|
||
|
||
print()
|
||
print("二、配对")
|
||
talk = Talk(PORT)
|
||
reply = talk.call({"cmd": "pair", "deviceId": "ind-test",
|
||
"deviceName": "提示条测试", "code": code})
|
||
check("配对成功", reply.get("ok") is True, str(reply)[:120])
|
||
|
||
print()
|
||
print("三、开始共享 → 必须出现提示")
|
||
check("共享前没有提示条", not notice_windows(proc.pid))
|
||
reply = talk.call({"cmd": "screen_start", "maxWidth": 320,
|
||
"quality": 40, "fps": 2}, timeout=15)
|
||
check("开始屏幕共享", reply.get("ok") is True, str(reply)[:120])
|
||
appeared = wait_notice(proc.pid, True)
|
||
check("★ 共享中出现了红色提示条", appeared,
|
||
str([w["title"] for w in windows_of(proc.pid) if w["visible"]]))
|
||
if appeared:
|
||
found = notice_windows(proc.pid)
|
||
left, top, width, height = found[0]["rect"]
|
||
user32 = ctypes.windll.user32
|
||
screen_w = user32.GetSystemMetrics(0)
|
||
screen_h = user32.GetSystemMetrics(1)
|
||
print(f" 提示条位置: ({left},{top}) {width}x{height}，"
|
||
f"屏幕 {screen_w}x{screen_h}")
|
||
check("提示条在右下角",
|
||
left > screen_w * 0.5 and top > screen_h * 0.5,
|
||
f"({left},{top})")
|
||
check("提示条可见尺寸合理", width > 40 and height > 10,
|
||
f"{width}x{height}")
|
||
|
||
print()
|
||
print("四、停止共享 → 提示必须消失")
|
||
reply = talk.call({"cmd": "screen_stop"}, timeout=15)
|
||
check("停止屏幕共享", reply.get("ok") is True, str(reply)[:120])
|
||
check("★ 停止后提示条消失", wait_notice(proc.pid, False))
|
||
|
||
print()
|
||
print("五、再次共享 → 提示再次出现")
|
||
talk.call({"cmd": "screen_start", "maxWidth": 320,
|
||
"quality": 40, "fps": 2}, timeout=15)
|
||
check("★ 提示条再次出现", wait_notice(proc.pid, True))
|
||
talk.call({"cmd": "screen_stop"}, timeout=15)
|
||
check("再次停止后消失", wait_notice(proc.pid, False))
|
||
|
||
talk.close()
|
||
|
||
finally:
|
||
if proc is not None:
|
||
try:
|
||
proc.terminate()
|
||
proc.wait(timeout=8)
|
||
except Exception: # noqa: BLE001
|
||
try:
|
||
proc.kill()
|
||
except Exception: # noqa: BLE001
|
||
pass
|
||
time.sleep(1.0)
|
||
try:
|
||
if DATA.exists():
|
||
shutil.rmtree(DATA, ignore_errors=True)
|
||
if backup is not None and backup.exists():
|
||
shutil.move(str(backup), str(DATA))
|
||
except Exception as exc: # noqa: BLE001
|
||
print(f" !! 还原失败: {exc}")
|
||
|
||
print()
|
||
print("=" * 68)
|
||
print(f"通过 {PASSED} 项，失败 {FAILED} 项")
|
||
return 1 if FAILED else 0
|
||
|
||
|
||
if __name__ == "__main__":
|
||
sys.exit(main())
|