feat(webui): add project workspaces and access controls (#4007)
* feat(webui): add project workspaces and access controls * feat(webui): add project workspaces and access controls * refactor(tools): centralize workspace access resolution * refactor(webui): remove unused workspace host state * fix(webui): hide estimated file edit label * fix(webui): clarify file edit deletion feedback * fix(webui): label deleted file activity * fix(webui): flatten file edit activity rows * fix(core): remove path-only patch deletion * fix(core): keep apply patch non-destructive * refactor(webui): trim workspace host plumbing * fix(tools): register exec with tools config
This commit is contained in:
@@ -9,6 +9,7 @@ from unittest.mock import patch
|
||||
import pytest
|
||||
|
||||
from nanobot.agent.tools.shell import ExecTool
|
||||
from nanobot.security.workspace_access import bind_workspace_scope, build_workspace_scope, reset_workspace_scope
|
||||
|
||||
|
||||
def _fake_resolve_private(hostname, port, family=0, type_=0):
|
||||
@@ -42,6 +43,70 @@ async def test_exec_blocks_wget_localhost():
|
||||
assert "Error" in result
|
||||
|
||||
|
||||
def test_exec_full_workspace_scope_allows_loopback(tmp_path):
|
||||
tool = ExecTool(working_dir=str(tmp_path))
|
||||
scope = build_workspace_scope(tmp_path, "full", source_channel="websocket")
|
||||
token = bind_workspace_scope(scope)
|
||||
try:
|
||||
with patch("nanobot.security.network.socket.getaddrinfo", _fake_resolve_localhost):
|
||||
error = tool._guard_command("curl http://localhost:8765/", str(tmp_path))
|
||||
finally:
|
||||
reset_workspace_scope(token)
|
||||
assert error is None
|
||||
|
||||
|
||||
def test_exec_core_full_workspace_scope_blocks_loopback(tmp_path):
|
||||
tool = ExecTool(working_dir=str(tmp_path))
|
||||
scope = build_workspace_scope(tmp_path, "full")
|
||||
token = bind_workspace_scope(scope)
|
||||
try:
|
||||
with patch("nanobot.security.network.socket.getaddrinfo", _fake_resolve_localhost):
|
||||
error = tool._guard_command("curl http://localhost:8765/", str(tmp_path))
|
||||
finally:
|
||||
reset_workspace_scope(token)
|
||||
assert error is not None
|
||||
assert "internal/private" in error
|
||||
|
||||
|
||||
def test_exec_full_workspace_scope_blocks_loopback_when_local_service_disabled(tmp_path):
|
||||
tool = ExecTool(working_dir=str(tmp_path), webui_allow_local_service_access=False)
|
||||
scope = build_workspace_scope(tmp_path, "full", source_channel="websocket")
|
||||
token = bind_workspace_scope(scope)
|
||||
try:
|
||||
with patch("nanobot.security.network.socket.getaddrinfo", _fake_resolve_localhost):
|
||||
error = tool._guard_command("curl http://localhost:8765/", str(tmp_path))
|
||||
finally:
|
||||
reset_workspace_scope(token)
|
||||
assert error is not None
|
||||
assert "internal/private" in error
|
||||
|
||||
|
||||
def test_exec_restricted_workspace_scope_blocks_loopback(tmp_path):
|
||||
tool = ExecTool(working_dir=str(tmp_path))
|
||||
scope = build_workspace_scope(tmp_path, "restricted", source_channel="websocket")
|
||||
token = bind_workspace_scope(scope)
|
||||
try:
|
||||
with patch("nanobot.security.network.socket.getaddrinfo", _fake_resolve_localhost):
|
||||
error = tool._guard_command("curl http://localhost:8765/", str(tmp_path))
|
||||
finally:
|
||||
reset_workspace_scope(token)
|
||||
assert error is not None
|
||||
assert "internal/private" in error
|
||||
|
||||
|
||||
def test_exec_full_workspace_scope_still_blocks_metadata(tmp_path):
|
||||
tool = ExecTool(working_dir=str(tmp_path))
|
||||
scope = build_workspace_scope(tmp_path, "full", source_channel="websocket")
|
||||
token = bind_workspace_scope(scope)
|
||||
try:
|
||||
with patch("nanobot.security.network.socket.getaddrinfo", _fake_resolve_private):
|
||||
error = tool._guard_command("curl http://169.254.169.254/latest/meta-data/", str(tmp_path))
|
||||
finally:
|
||||
reset_workspace_scope(token)
|
||||
assert error is not None
|
||||
assert "internal/private" in error
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_exec_allows_normal_commands():
|
||||
tool = ExecTool(timeout=5)
|
||||
|
||||
Reference in New Issue
Block a user