fix(host_agent): every Docker read is bounded, and the agent reports its own footprint (#5764)
CI / lint (push) Successful in 2s
CI / integration (push) Successful in 1m20s
CI / unit (push) Failing after 1m32s
CI / publish (push) Skipped

#5764 "The agent reads a container's whole log backlog into memory, on its
main loop, with no overall deadline". Docker03's 1.13.0 agent went silent
for a day at 2.1 GB and 46 h of CPU, then ignored SIGTERM.

- One shared reader, _docker_get: each answer has an overall deadline and a
  byte cap, not only a per-recv timeout. A trickling or endless body is cut
  off (15 s / 32 MiB for JSON, 5 s / 4 MiB for a log read).
- Container logs: since the cursor but only the newest 2,000 lines, all
  within 10 s. A note says when older lines were skipped, and a container
  whose answer is too big skips ahead to now instead of re-reading its
  backlog every interval.
- The agent reports its own RSS and CPU; Steward records agent_rss_bytes
  and agent_cpu_pct, so a runaway agent shows before it goes silent.

Agent 1.15.0.

Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>
This commit is contained in:
2026-10-14 18:01:45 -04:00
co-authored by Claude Opus 5.5
parent 82a2a3d41f
commit a805e1d539
7 changed files with 292 additions and 53 deletions
@@ -0,0 +1,139 @@
# tests/plugins/host_agent/test_agent_docker_bounds.py
"""#5764: everything the agent reads from Docker is bounded in time and size.
A 1.13.0 agent read a container's whole log backlog into memory on its main
loop, with only a per-read timeout, and went silent for a day at 2.1 GB. The
reader tests here talk to a real Unix socket that trickles or floods, so the
bound is the one the agent actually has, not a mocked one.
"""
import os
import socket
import tempfile
import threading
import time
import pytest
from plugins.host_agent import agent as a
@pytest.fixture
def docker_sock():
"""A Unix socket that answers each connection with `serve(conn)`."""
d = tempfile.mkdtemp(dir="/tmp")
path = os.path.join(d, "d.sock")
srv = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
srv.bind(path)
srv.listen(1)
holder = {}
def run():
conn, _ = srv.accept()
conn.recv(4096)
try:
holder["serve"](conn)
except OSError:
pass # the agent hung up on us: expected
finally:
conn.close()
def start(serve):
holder["serve"] = serve
threading.Thread(target=run, daemon=True).start()
return path
yield start
srv.close()
def test_an_answer_that_keeps_trickling_is_cut_off_at_its_deadline(docker_sock):
def trickle(conn):
conn.sendall(b"HTTP/1.1 200 OK\r\nContent-Type: application/json\r\n\r\n[")
while True: # a byte every 50 ms: never idle 5 s, never done
conn.sendall(b" ")
time.sleep(0.05)
path = docker_sock(trickle)
began = time.monotonic()
with pytest.raises(OSError):
a._docker_get(path, "/containers/json", "application/json",
timeout=5.0, deadline=0.4, max_bytes=1 << 20)
assert time.monotonic() - began < 2.0
def test_an_answer_over_its_size_is_refused_not_held(docker_sock):
def flood(conn):
conn.sendall(b"HTTP/1.1 200 OK\r\n\r\n")
block = b"x" * 65536
for _ in range(64): # 4 MiB on offer
conn.sendall(block)
path = docker_sock(flood)
with pytest.raises(a.DockerAnswerTooLarge):
a._docker_get(path, "/containers/web/logs", "application/octet-stream",
timeout=5.0, deadline=5.0, max_bytes=256 * 1024)
def test_a_whole_answer_still_reads(docker_sock):
def ok(conn):
conn.sendall(b"HTTP/1.1 200 OK\r\nTransfer-Encoding: chunked\r\n\r\n"
b"2\r\n[]\r\n0\r\n\r\n")
assert a._docker_request(docker_sock(ok), "/containers/json") == []
def _frame(stream: int, payload: bytes) -> bytes:
return bytes([stream, 0, 0, 0]) + len(payload).to_bytes(4, "big") + payload
def _log_lines(n: int) -> bytes:
return b"".join(_frame(1, f"2026-10-14T12:00:{i % 60:02d}.{i:09d}Z line {i}\n".encode())
for i in range(n))
def test_a_busy_container_is_read_from_its_newest_lines_and_says_what_it_skipped(monkeypatch):
asked = []
def fake_raw(socket_path, path, timeout=a.DOCKER_API_TIMEOUT):
asked.append(path)
return _log_lines(a.DOCKER_LOG_MAX_LINES)
monkeypatch.setattr(a, "_docker_request_raw", fake_raw)
state = {"docker_log_cursors": {"web": 1791979200.0}} # 2026-10-14T12:00:00Z
out = a.collect_docker_logs("/sock", [{"name": "web", "status": "running"}], state,
max_bytes=1 << 30)
assert "since=" in asked[0] and f"tail={a.DOCKER_LOG_MAX_LINES}" in asked[0]
notes = [r["line"] for r in out if r["container"] == "_steward"]
assert any("older ones skipped" in n for n in notes)
def test_a_container_whose_log_is_too_big_skips_ahead_instead_of_rereading(monkeypatch):
def too_big(socket_path, path, timeout=a.DOCKER_API_TIMEOUT):
raise a.DockerAnswerTooLarge("too big")
monkeypatch.setattr(a, "_docker_request_raw", too_big)
state = {"docker_log_cursors": {"web": 1791979200.0}}
out = a.collect_docker_logs("/sock", [{"name": "web", "status": "running"}], state,
now=1791979500.0)
assert state["docker_log_cursors"]["web"] == 1791979500.0 # next read starts at now
assert "skipped ahead to now" in out[-1]["line"]
def test_log_reading_stops_at_its_deadline(monkeypatch):
asked = []
monkeypatch.setattr(a, "_docker_request_raw",
lambda s, p, timeout=a.DOCKER_API_TIMEOUT: asked.append(p) or b"")
out = a.collect_docker_logs("/sock", [{"name": "web", "status": "running"}], {},
deadline=0.0)
assert asked == []
assert "limit" in out[-1]["line"]
def test_the_agent_reports_its_own_memory_and_cpu():
state: dict = {}
first = a.collect_self(state)
assert first["rss_bytes"] > 1024 * 1024
assert "cpu_pct" not in first # no previous sample to measure from
sum(range(200_000))
second = a.collect_self(state)
assert second["cpu_pct"] >= 0.0
@@ -113,3 +113,12 @@ def test_a_wired_board_input_still_counts():
assert counts_as_hottest("SYSTIN", 38.0) and counts_as_hottest("AUXTIN1", 60.0)
assert not counts_as_hottest("SYSTIN", 114.0) and not counts_as_hottest("PCH_CHIP_TEMP", 0.0)
assert counts_as_hottest("Package id 0", 101.0) # a CPU past 100 is a hot CPU
def test_the_agent_s_own_footprint_is_recorded():
"""#5764: a runaway agent shows on the record before it goes silent."""
sample = {"ts": "2026-10-14T22:00:00+00:00",
"agent": {"rss_bytes": 18_000_000, "cpu_pct": 1.5}}
idx = _index(_expand_sample_to_metrics(sample, "d3", datetime.now(timezone.utc)))
assert idx[("agent_rss_bytes", "d3")] == 18_000_000
assert idx[("agent_cpu_pct", "d3")] == 1.5