perf(proxy): test the remote CDM hosts only after the walk finds service code

This commit is contained in:
imSp4rky
2026-10-24 17:57:48 -06:00
parent fdb5369534
commit d0f9307506
2 changed files with 5 additions and 3 deletions
+3 -1
View File
@@ -129,7 +129,9 @@ def test_raw_socket_connect_to_a_host_name_warns(caplog, address, warns):
assert bool(leaks(caplog)) is warns
def test_core_frame_ends_the_walk(caplog):
def test_core_frame_ends_the_walk(caplog, monkeypatch):
"""A core lookup returns before the CDM host set is built, which a server does for every request it sends."""
monkeypatch.setattr(run_proxy, "cdm_hosts", lambda: pytest.fail("cdm_hosts() ran for a core lookup"))
with caplog.at_level(logging.WARNING):
source = "sys.audit('socket.getaddrinfo', 'vault.example', 443, 0, 0, 0)"
run_as(os.path.join(run_proxy.CORE_DIR, "core", "vaults.py"), source)
+2 -2
View File
@@ -190,10 +190,10 @@ def check(host: Any, frame: Optional[FrameType]) -> None:
if isinstance(host, bytes):
host = host.decode(errors="replace")
host = str(host or "").lower().strip("[]")
if not host or is_local(host) or host in proxy_hosts or host in cdm_hosts():
if not host or is_local(host) or host in proxy_hosts:
return
found = find_service(frame, through_core=False)
if not found:
if not found or host in cdm_hosts():
return
service_frame, entry = found
while frame is not None and frame is not service_frame: