mirror of
https://github.com/Sea-Haven-Industries/open-swe.git
synced 2026-10-07 16:19:09 +00:00
Compare commits
4 commits
d880cdabb3
...
5a0793bd35
| Author | SHA1 | Date | |
|---|---|---|---|
| 5a0793bd35 | |||
| 8552e871f4 | |||
| 969b8e3882 | |||
| ebeb2663fd |
9 changed files with 222 additions and 24 deletions
|
|
@ -45,6 +45,19 @@
|
|||
"suppression_justification": "Test fixture, not a real credential. Same fake Datadog API key \"secret-api-1234\" passed into connect_datadog by test_datadog_roundtrip_and_redaction. Never a live secret; scoped to the unit test only.",
|
||||
"owner": "adam@seahavenind.com",
|
||||
"added": "2026-06-29"
|
||||
},
|
||||
{
|
||||
"id": "gitleaks-generic-api-key-23",
|
||||
"title": "Hardcoded credential flagged in GitHub-token TTL test fixture (CWE-798)",
|
||||
"file": "tests/test_github_token_ttl.py",
|
||||
"line": 23,
|
||||
"rule": "CWE-798",
|
||||
"severity": "high",
|
||||
"status": "false-positive",
|
||||
"justification": "Test fixture, not a real credential. The literal \"ghp_secret\" is a fake GitHub token used by the cached-token TTL/revocation unit tests to exercise cache_github_token_for_thread / get_github_token expiry and invalidation. It is not a valid 40-char GitHub PAT, is never a live secret, and is scoped to the unit test only. Pre-existing test fixture, not introduced by this change.",
|
||||
"suppression_justification": "Test fixture, not a real credential. The literal \"ghp_secret\" is a fake GitHub token used by the cached-token TTL/revocation unit tests to exercise cache_github_token_for_thread / get_github_token expiry and invalidation. It is not a valid 40-char GitHub PAT, is never a live secret, and is scoped to the unit test only. Pre-existing test fixture, not introduced by this change.",
|
||||
"owner": "adam@seahavenind.com",
|
||||
"added": "2026-06-29"
|
||||
}
|
||||
]
|
||||
}
|
||||
|
|
|
|||
|
|
@ -59,6 +59,7 @@ from .server import (
|
|||
DEFAULT_LLM_MAX_TOKENS,
|
||||
DEFAULT_RECURSION_LIMIT,
|
||||
MODEL_CALL_RECURSION_LIMIT,
|
||||
SandboxRepoMismatchError,
|
||||
_general_purpose_subagent,
|
||||
ensure_sandbox_for_thread,
|
||||
graph_loaded_for_execution,
|
||||
|
|
@ -851,12 +852,18 @@ async def get_reviewer_agent(config: RunnableConfig) -> Pregel:
|
|||
if repo_config.get("owner") and repo_config.get("name")
|
||||
else None
|
||||
)
|
||||
sandbox_backend = await ensure_sandbox_for_thread(
|
||||
thread_id,
|
||||
github_proxy_token=github_proxy_token,
|
||||
github_proxy_repositories=[repo_name_for_scope] if repo_name_for_scope else None,
|
||||
repo=repo_for_snapshot,
|
||||
)
|
||||
try:
|
||||
sandbox_backend = await ensure_sandbox_for_thread(
|
||||
thread_id,
|
||||
github_proxy_token=github_proxy_token,
|
||||
github_proxy_repositories=[repo_name_for_scope] if repo_name_for_scope else None,
|
||||
repo=repo_for_snapshot,
|
||||
)
|
||||
except SandboxRepoMismatchError as exc:
|
||||
# Repo-binding refusal at the run boundary: log for alarming and surface the
|
||||
# sanitized terminal error rather than crash-looping the reviewer worker.
|
||||
logger.error("Refusing reviewer run for thread %s: %s", thread_id, exc)
|
||||
raise RuntimeError(str(exc)) from exc
|
||||
|
||||
work_dir = await aresolve_sandbox_work_dir(sandbox_backend)
|
||||
|
||||
|
|
|
|||
|
|
@ -451,14 +451,19 @@ async def ensure_sandbox_for_thread(
|
|||
sandbox_backend = SANDBOX_BACKENDS.get(thread_id)
|
||||
sandbox_id = await get_sandbox_id_from_metadata(thread_id)
|
||||
|
||||
# Repo-binding guard (TID-COLLIDE-01): refuse to reuse a thread's sandbox for
|
||||
# a repo it is not bound to, so a colliding thread_id from a different repo
|
||||
# cannot bind to (or clobber) another thread's sandbox.
|
||||
if sandbox_id == SANDBOX_CREATING and not sandbox_backend:
|
||||
logger.info("Sandbox creation in progress for thread %s, waiting...", thread_id)
|
||||
sandbox_id = await _resolve_creating_sentinel(thread_id)
|
||||
|
||||
# Repo-binding guard (TID-COLLIDE-01): a sandbox is never served to a repo
|
||||
# unless its binding is known and matches.
|
||||
current_repo = repo_cache_key(repo)
|
||||
bound_repo = await get_bound_repo_from_metadata(thread_id)
|
||||
proxy_bound = getattr(sandbox_backend, "bound_repo", None)
|
||||
effective_bound = bound_repo or (proxy_bound if isinstance(proxy_bound, str) else None)
|
||||
if current_repo and effective_bound and effective_bound != current_repo:
|
||||
# Known binding that does not match the current repo: refuse outright so a
|
||||
# colliding thread_id from a different repo cannot reuse/clobber it.
|
||||
logger.error(
|
||||
"Repo mismatch for thread %s: bound=%s current=%s; refusing sandbox reuse",
|
||||
thread_id,
|
||||
|
|
@ -466,10 +471,26 @@ async def ensure_sandbox_for_thread(
|
|||
current_repo,
|
||||
)
|
||||
raise SandboxRepoMismatchError(thread_id, effective_bound, current_repo)
|
||||
|
||||
if sandbox_id == SANDBOX_CREATING and not sandbox_backend:
|
||||
logger.info("Sandbox creation in progress for thread %s, waiting...", thread_id)
|
||||
sandbox_id = await _resolve_creating_sentinel(thread_id)
|
||||
if (
|
||||
current_repo
|
||||
and not effective_bound
|
||||
and sandbox_backend is None
|
||||
and isinstance(sandbox_id, str)
|
||||
and sandbox_id not in (None, SANDBOX_CREATING)
|
||||
):
|
||||
# Fail CLOSED for unbound-legacy threads (migration window): a thread with a
|
||||
# persisted sandbox_id but no in-memory cache and no recorded bound_repo
|
||||
# cannot be confirmed to belong to the current repo, so never
|
||||
# reconnect-and-serve it. Drop the stale id and recreate a fresh sandbox
|
||||
# bound to this repo below.
|
||||
logger.error(
|
||||
"reconnect-with-missing-binding for thread %s: persisted sandbox %s has no "
|
||||
"bound_repo; refusing reuse and recreating for repo %s",
|
||||
thread_id,
|
||||
sandbox_id,
|
||||
current_repo,
|
||||
)
|
||||
sandbox_id = None
|
||||
|
||||
if sandbox_backend:
|
||||
logger.info("Using cached sandbox backend for thread %s", thread_id)
|
||||
|
|
@ -668,11 +689,21 @@ async def get_agent(config: RunnableConfig) -> Pregel:
|
|||
)
|
||||
team_defaults_task = asyncio.create_task(get_team_default_model_pair("agent"))
|
||||
profile_task = asyncio.create_task(load_profile(profile_login)) if profile_login else None
|
||||
triggering_user_identity, sandbox_backend, team_defaults = await asyncio.gather(
|
||||
triggering_user_identity_task,
|
||||
sandbox_task,
|
||||
team_defaults_task,
|
||||
)
|
||||
try:
|
||||
triggering_user_identity, sandbox_backend, team_defaults = await asyncio.gather(
|
||||
triggering_user_identity_task,
|
||||
sandbox_task,
|
||||
team_defaults_task,
|
||||
)
|
||||
except SandboxRepoMismatchError as exc:
|
||||
# Repo-binding refusal at the run boundary: log for alarming and surface the
|
||||
# already-sanitized terminal error (no sandbox/token internals) to the caller,
|
||||
# rather than letting an opaque deep-stack exception crash-loop the worker.
|
||||
logger.error("Refusing agent run for thread %s: %s", thread_id, exc)
|
||||
for pending in (triggering_user_identity_task, team_defaults_task, profile_task):
|
||||
if pending is not None and not pending.done():
|
||||
pending.cancel()
|
||||
raise RuntimeError(str(exc)) from exc
|
||||
profile = await profile_task if profile_task is not None else None
|
||||
del github_token
|
||||
|
||||
|
|
|
|||
|
|
@ -192,7 +192,12 @@ async def get_github_token_for_user(ls_user_id: str, tenant_id: str) -> dict[str
|
|||
return result
|
||||
if auth_url:
|
||||
return {"auth_url": auth_url}
|
||||
return {"error": f"Unexpected auth result: {response_data}"}
|
||||
# Log the full upstream body server-side only; the returned error becomes a
|
||||
# user-facing Slack/Linear comment, so never echo the raw response body.
|
||||
logger.error(
|
||||
"GitHub auth returned an unexpected result (no token/url): %s", response_data
|
||||
)
|
||||
return {"error": "GitHub auth returned an unexpected result"}
|
||||
|
||||
except httpx.HTTPStatusError as e:
|
||||
# Log the full upstream body server-side only; the returned error becomes a
|
||||
|
|
|
|||
|
|
@ -28,15 +28,20 @@ class GitHubAuthError(Exception):
|
|||
|
||||
|
||||
def repo_cache_key(repo: Any) -> str | None:
|
||||
"""Normalize a repo dict/string to ``owner/name`` (None when unknown)."""
|
||||
"""Normalize a repo dict/string to a casefolded ``owner/name`` (None if unknown).
|
||||
|
||||
GitHub owner/name are case-insensitive, so the key is casefolded on both the
|
||||
write (binding) and read (compare) sides to keep ``Org/Repo`` and ``org/repo``
|
||||
a single repo and avoid spurious cross-repo mismatches.
|
||||
"""
|
||||
if isinstance(repo, str):
|
||||
cleaned = repo.strip()
|
||||
return cleaned or None
|
||||
return cleaned.casefold() or None
|
||||
if isinstance(repo, Mapping):
|
||||
owner = repo.get("owner")
|
||||
name = repo.get("name")
|
||||
if isinstance(owner, str) and isinstance(name, str) and owner and name:
|
||||
return f"{owner}/{name}"
|
||||
return f"{owner}/{name}".casefold()
|
||||
return None
|
||||
|
||||
|
||||
|
|
|
|||
|
|
@ -19,6 +19,7 @@ from deepagents.backends.protocol import (
|
|||
)
|
||||
from langgraph.config import get_config
|
||||
|
||||
from .github_token import repo_cache_key
|
||||
from .sandbox import create_sandbox
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
|
@ -167,7 +168,9 @@ async def get_bound_repo_from_metadata(thread_id: str) -> str | None:
|
|||
if not isinstance(metadata, dict):
|
||||
return None
|
||||
bound_repo = metadata.get("bound_repo")
|
||||
return bound_repo if isinstance(bound_repo, str) and bound_repo else None
|
||||
# Casefold on read so legacy metadata written before normalization (e.g.
|
||||
# ``Org/Repo``) still compares equal to the casefolded current repo key.
|
||||
return repo_cache_key(bound_repo) if isinstance(bound_repo, str) and bound_repo else None
|
||||
|
||||
|
||||
def clear_sandbox_backend(thread_id: str) -> None:
|
||||
|
|
|
|||
|
|
@ -43,6 +43,19 @@ class _Client:
|
|||
return self._resp
|
||||
|
||||
|
||||
class _OkResp:
|
||||
"""A 2xx response whose JSON body lacks both a token and an auth url."""
|
||||
|
||||
def __init__(self, payload: Any) -> None:
|
||||
self._payload = payload
|
||||
|
||||
def raise_for_status(self) -> None:
|
||||
return None
|
||||
|
||||
def json(self) -> Any:
|
||||
return self._payload
|
||||
|
||||
|
||||
def test_http_error_returns_generic_message(monkeypatch: pytest.MonkeyPatch) -> None:
|
||||
monkeypatch.setattr(auth, "GITHUB_OAUTH_PROVIDER_ID", "provider-id")
|
||||
monkeypatch.setattr(auth, "X_SERVICE_AUTH_JWT_SECRET", "jwt-secret")
|
||||
|
|
@ -53,3 +66,16 @@ def test_http_error_returns_generic_message(monkeypatch: pytest.MonkeyPatch) ->
|
|||
assert "error" in result
|
||||
assert _SECRET_BODY not in result["error"]
|
||||
assert "500" in result["error"]
|
||||
|
||||
|
||||
def test_unexpected_result_returns_generic_message(monkeypatch: pytest.MonkeyPatch) -> None:
|
||||
"""The 2xx-but-missing-token/url branch must not echo the upstream body."""
|
||||
monkeypatch.setattr(auth, "GITHUB_OAUTH_PROVIDER_ID", "provider-id")
|
||||
monkeypatch.setattr(auth, "X_SERVICE_AUTH_JWT_SECRET", "jwt-secret")
|
||||
body = {"unexpected_field": _SECRET_BODY}
|
||||
monkeypatch.setattr(httpx, "AsyncClient", lambda *a, **k: _Client(_OkResp(body)))
|
||||
|
||||
result = asyncio.run(auth.get_github_token_for_user("ls-user", "tenant"))
|
||||
|
||||
assert result == {"error": "GitHub auth returned an unexpected result"}
|
||||
assert _SECRET_BODY not in result["error"]
|
||||
|
|
|
|||
|
|
@ -309,7 +309,7 @@ class TestRefreshProxyOnSandboxReuse:
|
|||
|
||||
@pytest.mark.asyncio
|
||||
async def test_refreshes_proxy_when_reconnecting_to_existing_langsmith_sandbox(self) -> None:
|
||||
"""Reconnected sandboxes should also get a fresh proxy token."""
|
||||
"""A bound thread reconnecting to its sandbox should get a fresh proxy token."""
|
||||
config = self._execution_config()
|
||||
mock_sandbox = MagicMock(id="sandbox-existing")
|
||||
|
||||
|
|
@ -325,6 +325,13 @@ class TestRefreshProxyOnSandboxReuse:
|
|||
new_callable=AsyncMock,
|
||||
return_value="sandbox-existing",
|
||||
),
|
||||
# Thread is bound to the current repo, so reconnect proceeds (an unbound
|
||||
# legacy thread would instead fail closed and recreate).
|
||||
patch(
|
||||
"agent.server.get_bound_repo_from_metadata",
|
||||
new_callable=AsyncMock,
|
||||
return_value="langchain-ai/open-swe",
|
||||
),
|
||||
patch("agent.server.create_sandbox", return_value=mock_sandbox) as mock_create,
|
||||
patch(
|
||||
"agent.server.get_github_app_installation_token_with_expiry",
|
||||
|
|
|
|||
|
|
@ -53,11 +53,24 @@ def test_unbound_token_served_when_repo_unknown() -> None:
|
|||
assert github_token.get_github_token({"configurable": {"thread_id": "tid"}}) == "ghp_legacy"
|
||||
|
||||
|
||||
def test_cached_token_reused_for_same_repo_case_insensitive() -> None:
|
||||
"""``Org/Repo`` and ``org/repo`` are the same repo: no spurious refusal."""
|
||||
future = (datetime.now(UTC) + timedelta(hours=1)).isoformat()
|
||||
github_token.cache_github_token_for_thread(
|
||||
"tid", "ghp_repoA", expires_at=future, repo={"owner": "Acme", "name": "Alpha"}
|
||||
)
|
||||
cfg = {"configurable": {"thread_id": "tid", "repo": {"owner": "acme", "name": "alpha"}}}
|
||||
assert github_token.get_github_token(cfg) == "ghp_repoA"
|
||||
|
||||
|
||||
def test_repo_cache_key_normalizes() -> None:
|
||||
assert github_token.repo_cache_key({"owner": "o", "name": "r"}) == "o/r"
|
||||
assert github_token.repo_cache_key("o/r") == "o/r"
|
||||
assert github_token.repo_cache_key({"owner": "o"}) is None
|
||||
assert github_token.repo_cache_key(None) is None
|
||||
# Casefolded so different casing of the same repo collapses to one key.
|
||||
assert github_token.repo_cache_key({"owner": "Org", "name": "Repo"}) == "org/repo"
|
||||
assert github_token.repo_cache_key("Org/Repo") == "org/repo"
|
||||
|
||||
|
||||
# --- sandbox repo binding ----------------------------------------------------
|
||||
|
|
@ -122,3 +135,91 @@ async def test_ensure_sandbox_allows_matching_repo(monkeypatch: pytest.MonkeyPat
|
|||
result = await server.ensure_sandbox_for_thread("tid", repo={"owner": "acme", "name": "alpha"})
|
||||
assert result is backend
|
||||
assert calls["git"] == 1
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_ensure_sandbox_reuses_same_repo_case_insensitive(
|
||||
monkeypatch: pytest.MonkeyPatch,
|
||||
) -> None:
|
||||
"""A bound sandbox is reused when the current repo differs only by casing."""
|
||||
|
||||
async def fake_sandbox_id(_tid: str) -> str:
|
||||
return "sb-A"
|
||||
|
||||
async def fake_bound_repo(_tid: str) -> str:
|
||||
return "acme/alpha"
|
||||
|
||||
backend = _FakeBackend("sb-A")
|
||||
backend.bound_repo = "acme/alpha"
|
||||
|
||||
async def fake_check(b: Any, *_a: Any, **_k: Any) -> Any:
|
||||
return b
|
||||
|
||||
async def fake_refresh(b: Any, *_a: Any, **_k: Any) -> Any:
|
||||
return b
|
||||
|
||||
async def fake_git(_b: Any) -> None:
|
||||
return None
|
||||
|
||||
monkeypatch.setattr(server, "get_sandbox_id_from_metadata", fake_sandbox_id)
|
||||
monkeypatch.setattr(server, "get_bound_repo_from_metadata", fake_bound_repo)
|
||||
monkeypatch.setattr(server, "check_or_recreate_sandbox", fake_check)
|
||||
monkeypatch.setattr(server, "_refresh_github_proxy_or_recreate", fake_refresh)
|
||||
monkeypatch.setattr(server, "set_sandbox_backend", lambda _tid, b, **_k: b)
|
||||
monkeypatch.setattr(server, "_configure_git_identity", fake_git)
|
||||
server.SANDBOX_BACKENDS["tid"] = backend # type: ignore[assignment]
|
||||
|
||||
# Different casing of the same repo must not raise and must reuse the sandbox.
|
||||
result = await server.ensure_sandbox_for_thread("tid", repo={"owner": "ACME", "name": "Alpha"})
|
||||
assert result is backend
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_legacy_unbound_sandbox_recreated_not_reused(
|
||||
monkeypatch: pytest.MonkeyPatch,
|
||||
) -> None:
|
||||
"""A legacy thread (sandbox_id present, bound_repo absent) must fail closed.
|
||||
|
||||
With no recorded binding, the existing sandbox cannot be confirmed to belong
|
||||
to the current repo, so it is never reconnected-and-served: a fresh sandbox is
|
||||
created and bound to the requesting repo instead.
|
||||
"""
|
||||
|
||||
async def fake_sandbox_id(_tid: str) -> str:
|
||||
return "sb-legacy"
|
||||
|
||||
async def fake_bound_repo(_tid: str) -> None:
|
||||
return None
|
||||
|
||||
def fake_reconnect(*_a: Any, **_k: Any) -> Any: # pragma: no cover - must not run
|
||||
raise AssertionError("must not reconnect to a legacy unbound sandbox")
|
||||
|
||||
fresh = _FakeBackend("sb-fresh")
|
||||
|
||||
async def fake_create_with_proxy(*_a: Any, **_k: Any) -> Any:
|
||||
return fresh
|
||||
|
||||
async def fake_git(_b: Any) -> None:
|
||||
return None
|
||||
|
||||
class _Threads:
|
||||
async def update(self, **_k: Any) -> None:
|
||||
return None
|
||||
|
||||
async def get(self, *_a: Any, **_k: Any) -> dict[str, Any]:
|
||||
return {}
|
||||
|
||||
class _Client:
|
||||
threads = _Threads()
|
||||
|
||||
monkeypatch.setattr(server, "get_sandbox_id_from_metadata", fake_sandbox_id)
|
||||
monkeypatch.setattr(server, "get_bound_repo_from_metadata", fake_bound_repo)
|
||||
monkeypatch.setattr(server, "create_sandbox", fake_reconnect)
|
||||
monkeypatch.setattr(server, "_create_sandbox_with_proxy", fake_create_with_proxy)
|
||||
monkeypatch.setattr(server, "_configure_git_identity", fake_git)
|
||||
monkeypatch.setattr(server, "client", _Client())
|
||||
|
||||
result = await server.ensure_sandbox_for_thread("tid", repo={"owner": "evil", "name": "beta"})
|
||||
|
||||
assert result.id == "sb-fresh"
|
||||
assert server.SANDBOX_BACKENDS["tid"].bound_repo == "evil/beta"
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue