Files
fips/testing/chaos/sim/docker_exec.py
c1589c02af Cross-compilation and macOS compatibility fixes (#1)
- Ethernet ioctl: cfg-gate the ioctl request parameter type — c_int on
  musl, c_ulong on gnu — so the same code compiles on both
  x86_64-unknown-linux-gnu and x86_64-unknown-linux-musl targets

- Chaos harness macOS support: replace direct host 'ip link' invocations
  with a privileged Docker container helper (--net=host --pid=host) that
  shares the Docker VM's namespaces, making veth pair setup work on both
  Linux and macOS (where host 'ip' is unavailable and container PIDs
  live inside the Docker Desktop VM)

- Python 3.9 compat: add 'from __future__ import annotations' to
  docker_exec.py for X|Y union syntax support

- Add deploy/ to .gitignore

Co-authored-by: origami74 <origami74@gmail.com>
2026-02-26 03:48:51 -08:00

74 lines
2.2 KiB
Python

"""Thin wrapper around subprocess for docker exec calls."""
from __future__ import annotations
import subprocess
import logging
log = logging.getLogger(__name__)
class DockerExecError(Exception):
def __init__(self, container, cmd, returncode, stderr):
self.container = container
self.cmd = cmd
self.returncode = returncode
self.stderr = stderr
super().__init__(
f"docker exec {container}: {cmd!r} returned {returncode}: {stderr}"
)
def docker_exec(container: str, cmd: str, timeout: int = 30) -> str:
"""Execute a command in a running container, return stdout."""
result = subprocess.run(
["docker", "exec", container, "/bin/bash", "-c", cmd],
capture_output=True,
text=True,
timeout=timeout,
)
if result.returncode != 0:
raise DockerExecError(container, cmd, result.returncode, result.stderr)
return result.stdout
def docker_exec_quiet(container: str, cmd: str, timeout: int = 30) -> str | None:
"""Execute a command, return stdout on success or None on failure (logged)."""
try:
return docker_exec(container, cmd, timeout)
except (DockerExecError, subprocess.TimeoutExpired) as e:
log.warning("docker exec failed on %s: %s", container, e)
return None
def docker_compose(
compose_file: str,
args: list[str],
timeout: int = 300,
check: bool = True,
) -> subprocess.CompletedProcess:
"""Run a docker compose command with the given compose file."""
cmd = ["docker", "compose", "-f", compose_file] + args
log.info("Running: %s", " ".join(cmd))
return subprocess.run(
cmd,
capture_output=True,
text=True,
timeout=timeout,
check=check,
)
def is_container_running(container: str) -> bool:
"""Check if a container is running."""
try:
result = subprocess.run(
["docker", "inspect", "-f", "{{.State.Running}}", container],
capture_output=True,
text=True,
timeout=10,
)
return result.returncode == 0 and result.stdout.strip() == "true"
except subprocess.TimeoutExpired:
return False