c140163f52
Introduce ProxyWarmupGate to manage MTProxy warmup state and throttle startup fanout; wire it into FileLoadOperation, FileLoader, FileLoaderPriorityQueue and ProxyRuntimeStateStore to classify requests and enforce limits/delays. Refactor ConnectionsManager DNS path: unified DoH loader, DNS resolver chain (cache/system/Google/Cloudflare), DoH query encoding, IPv6 support, renamed Mozilla->Cloudflare task and improved DNS logging. Add log atomicity guard and tighten several Tools checks; update runtime verifier with warmup fanout thresholds and DNS log validation. Miscellaneous logging/message wording fixes and throttling adjustments.
89 lines
3 KiB
Python
89 lines
3 KiB
Python
#!/usr/bin/env python3
|
|
from pathlib import Path
|
|
import re
|
|
import sys
|
|
|
|
|
|
ROOT = Path(__file__).resolve().parents[1]
|
|
BUFFERS_CPP = ROOT / "TMessagesProj/jni/tgnet/BuffersStorage.cpp"
|
|
BUFFERS_H = ROOT / "TMessagesProj/jni/tgnet/BuffersStorage.h"
|
|
MTPROXY_ALL = ROOT / "Tools/check_mtproxy_all.py"
|
|
|
|
|
|
def read(path: Path) -> str:
|
|
return path.read_text(encoding="utf-8", errors="replace")
|
|
|
|
|
|
def require(condition: bool, message: str, failures: list[str]) -> None:
|
|
if not condition:
|
|
failures.append(message)
|
|
|
|
|
|
def main() -> int:
|
|
failures: list[str] = []
|
|
buffers_cpp = read(BUFFERS_CPP)
|
|
buffers_h = read(BUFFERS_H)
|
|
mtproxy_all = read(MTPROXY_ALL)
|
|
|
|
require(
|
|
"too much %d buffers" not in buffers_cpp,
|
|
"buffer pool pressure must not use failure-looking 'too much buffers' wording",
|
|
failures,
|
|
)
|
|
require(
|
|
"buffer_pool_pressure size=%u active=%u cap=%u alloc_fallback=%u" in buffers_cpp,
|
|
"buffer pool pressure must log neutral size/active/cap/alloc_fallback counters",
|
|
failures,
|
|
)
|
|
require(
|
|
"BUFFER_POOL_PRESSURE_LOG_INTERVAL_MS" in buffers_cpp
|
|
and re.search(r"BUFFER_POOL_PRESSURE_LOG_INTERVAL_MS\s*=\s*5000", buffers_cpp),
|
|
"buffer pool pressure log must be throttled to a once-per-five-seconds interval",
|
|
failures,
|
|
)
|
|
require(
|
|
"lastPressureLogByCapacity" in buffers_h
|
|
and "lastPressureLogByCapacity[capacity]" in buffers_cpp,
|
|
"buffer pool pressure throttling must be tracked per capacity",
|
|
failures,
|
|
)
|
|
require(
|
|
"BufferPoolDebugStats" in buffers_h
|
|
and "bufferPoolStatsByCapacity" in buffers_h
|
|
and "allocFallbackCount" in buffers_cpp
|
|
and "reusedCount" in buffers_cpp
|
|
and "peakCachedCount" in buffers_cpp,
|
|
"buffer pool pressure must maintain per-size debug counters for fallback, reuse, and peak cache",
|
|
failures,
|
|
)
|
|
require(
|
|
"BUFFER_POOL_SUMMARY_INTERVAL_MS = 60 * 1000" in buffers_cpp
|
|
and "buffer_pool_summary size=%u peak=%u alloc_fallback=%u reused=%u" in buffers_cpp,
|
|
"buffer pool must emit a throttled summary per size class",
|
|
failures,
|
|
)
|
|
require(
|
|
"bufferPoolMaxCountForCapacity" in buffers_cpp
|
|
and "capacity == 1024 + 200" in buffers_cpp
|
|
and re.search(r"capacity\s*==\s*1024\s*\+\s*200[^;{]*\{[^}]*return\s+80\s*;", buffers_cpp, re.S),
|
|
"1224-byte buffers used by TLS frame assembly must get the hot-buffer cache cap, not the generic cap of 10",
|
|
failures,
|
|
)
|
|
require(
|
|
'"check_buffer_pool_pressure.py"' in mtproxy_all,
|
|
"full guard suite must include buffer-pool pressure guard",
|
|
failures,
|
|
)
|
|
|
|
if failures:
|
|
print("Buffer pool pressure guard failed:", file=sys.stderr)
|
|
for failure in failures:
|
|
print(f" - {failure}", file=sys.stderr)
|
|
return 1
|
|
|
|
print("Buffer pool pressure guard passed.")
|
|
return 0
|
|
|
|
|
|
if __name__ == "__main__":
|
|
raise SystemExit(main())
|