Compare commits
1 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
| 639fb66e5b |
+75
-23
@@ -8,10 +8,12 @@ Each connection is stored in data/cell_links.json and manifests as:
|
||||
- An iptables FORWARD rule set (service-level access control)
|
||||
"""
|
||||
|
||||
import ipaddress
|
||||
import json
|
||||
import logging
|
||||
import os
|
||||
import random
|
||||
import re
|
||||
import subprocess
|
||||
from datetime import datetime, timezone, timedelta
|
||||
from typing import Any, Dict, List, Optional
|
||||
@@ -30,15 +32,56 @@ _BACKOFF_BASE_S = 60
|
||||
_BACKOFF_MAX_S = 3600
|
||||
|
||||
|
||||
def _remote_api_url(dns_ip: Optional[str]) -> Optional[str]:
|
||||
# Strict formats for fields imported from a remote cell's invite. The domain and
|
||||
# dns_ip flow into a `curl --resolve <domain>:443:<dns_ip>` argv (peer-sync push);
|
||||
# anchoring them — domain must start alphanumeric, dns_ip must be an IP — prevents
|
||||
# a malicious invite injecting a leading-dash value that curl reads as a flag.
|
||||
_INVITE_HOSTNAME_RE = re.compile(r'^[A-Za-z0-9]([A-Za-z0-9.-]{0,253}[A-Za-z0-9])?$')
|
||||
_INVITE_CELL_NAME_RE = re.compile(r'^[A-Za-z0-9][A-Za-z0-9 _.-]{0,63}$')
|
||||
_INVITE_ENDPOINT_RE = re.compile(r'^[A-Za-z0-9][A-Za-z0-9._-]*:\d{1,5}$')
|
||||
|
||||
|
||||
def _validate_invite_fields(invite: Dict[str, Any]) -> None:
|
||||
"""Reject a remote cell's invite whose fields aren't strictly well-formed.
|
||||
|
||||
Defence-in-depth: these values come from another cell and reach iptables,
|
||||
DNS config, and a curl argv (the peer-sync push --resolves <domain>:443:
|
||||
<dns_ip>). Anchoring domain/dns_ip/endpoint to start alphanumeric blocks a
|
||||
malicious leading-dash value that curl would read as a flag. The public_key
|
||||
is validated downstream by WireGuardManager.add_cell_peer. Raise ValueError
|
||||
on anything malformed.
|
||||
"""
|
||||
name = invite.get('cell_name', '')
|
||||
if not isinstance(name, str) or not _INVITE_CELL_NAME_RE.match(name):
|
||||
raise ValueError(f'invalid cell_name {name!r}')
|
||||
domain = invite.get('domain', '')
|
||||
if not isinstance(domain, str) or not _INVITE_HOSTNAME_RE.match(domain):
|
||||
raise ValueError(f'invalid domain {domain!r}: must be a hostname')
|
||||
try:
|
||||
ipaddress.ip_address(str(invite.get('dns_ip', '')))
|
||||
except ValueError:
|
||||
raise ValueError(f"invalid dns_ip {invite.get('dns_ip')!r}")
|
||||
try:
|
||||
ipaddress.ip_network(str(invite.get('vpn_subnet', '')), strict=False)
|
||||
except ValueError:
|
||||
raise ValueError(f"invalid vpn_subnet {invite.get('vpn_subnet')!r}")
|
||||
endpoint = invite.get('endpoint')
|
||||
if endpoint and not _INVITE_ENDPOINT_RE.match(str(endpoint)):
|
||||
raise ValueError(f'invalid endpoint {endpoint!r}')
|
||||
|
||||
|
||||
def _remote_api_url(domain: Optional[str]) -> Optional[str]:
|
||||
"""Base URL for a linked cell's API, reached over the WG tunnel.
|
||||
|
||||
Cross-cell peer-sync goes to the remote's Caddy on 443 (the WireGuard server
|
||||
DNATs VPN-IP:443 → Caddy → API). The API's own :3000 is bound to 127.0.0.1
|
||||
and is NOT reachable from another cell, so we must target HTTPS/443, not
|
||||
http://<ip>:3000.
|
||||
DNATs VPN-IP:443 → Caddy → API; the API's own :3000 binds 127.0.0.1 and is
|
||||
unreachable from another cell). The URL uses the remote cell's DOMAIN — not
|
||||
its VPN IP — because Caddy only holds a certificate for the domain (ACME) or
|
||||
the .cell name (internal CA); a request by bare IP has no matching SNI and the
|
||||
TLS handshake fails. The push connects to the VPN IP over the tunnel via
|
||||
`curl --resolve <domain>:443:<dns_ip>` (see _push_permissions_to_remote).
|
||||
"""
|
||||
return f"https://{dns_ip}" if dns_ip else None
|
||||
return f"https://{domain}" if domain else None
|
||||
|
||||
|
||||
def _compute_next_retry(attempts: int) -> str:
|
||||
@@ -76,13 +119,12 @@ class CellLinkManager:
|
||||
link['permissions'] = _default_perms()
|
||||
changed = True
|
||||
# Phase 1 migration: permission-sync tracking fields
|
||||
if 'remote_api_url' not in link:
|
||||
link['remote_api_url'] = _remote_api_url(link.get('dns_ip'))
|
||||
changed = True
|
||||
# Migrate legacy http://<ip>:3000 URLs (unreachable across
|
||||
# cells) to the HTTPS/Caddy form.
|
||||
elif str(link.get('remote_api_url', '')).startswith('http://'):
|
||||
link['remote_api_url'] = _remote_api_url(link.get('dns_ip'))
|
||||
# Domain-based HTTPS URL. Rebuild if missing, or if it's a
|
||||
# legacy form: http://<ip>:3000 (unreachable) or https://<ip>
|
||||
# (no matching Caddy cert by bare IP).
|
||||
_want_url = _remote_api_url(link.get('domain'))
|
||||
if link.get('remote_api_url') != _want_url and _want_url:
|
||||
link['remote_api_url'] = _want_url
|
||||
changed = True
|
||||
if 'last_push_status' not in link:
|
||||
link['last_push_status'] = 'never'
|
||||
@@ -197,19 +239,26 @@ class CellLinkManager:
|
||||
payload = json.dumps(body)
|
||||
endpoint = url.rstrip('/') + '/api/cells/peer-sync/permissions'
|
||||
|
||||
# Determine local WG IP so the remote can authenticate us by source subnet.
|
||||
# MASQUERADE rewrites source to cell-wireguard's eth0 IP (172.20.x.x), which
|
||||
# is NOT in the cell's vpn_subnet. Passing the true WG IP in X-Forwarded-For
|
||||
# lets _authenticate_peer_cell() find the matching cell link.
|
||||
# Determine local WG IP for X-Forwarded-For (belt-and-suspenders for the
|
||||
# remote's source-subnet auth). With the peer-sync masquerade exclusion
|
||||
# the remote's Caddy already sees our real VPN source and appends it, but
|
||||
# passing it explicitly is harmless.
|
||||
local_wg_ip = self._local_wg_ip()
|
||||
xff_header = f'X-Forwarded-For: {local_wg_ip}' if local_wg_ip else None
|
||||
|
||||
# Reach the remote over the WG tunnel by its VPN IP, but present the
|
||||
# cell's DOMAIN as SNI/Host so Caddy serves its certificate — a request
|
||||
# to a bare IP has no matching cert and the TLS handshake fails. -k still
|
||||
# covers LAN mode (internal-CA cert curl won't chain to).
|
||||
domain = link.get('domain')
|
||||
dns_ip = link.get('dns_ip')
|
||||
cmd = [
|
||||
'docker', 'exec', 'cell-wireguard',
|
||||
# -k: the request reaches Caddy by the remote's VPN IP over the
|
||||
# encrypted WG tunnel, so the TLS cert (issued for the cell's domain)
|
||||
# won't match the IP — the tunnel already authenticates the peer.
|
||||
'curl', '-s', '-k', '-o', '/dev/null', '-w', '%{http_code}',
|
||||
]
|
||||
if domain and dns_ip:
|
||||
cmd += ['--resolve', f'{domain}:443:{dns_ip}']
|
||||
cmd += [
|
||||
'-X', 'POST',
|
||||
'-H', 'Content-Type: application/json',
|
||||
]
|
||||
@@ -537,6 +586,7 @@ class CellLinkManager:
|
||||
for field in ('cell_name', 'public_key', 'vpn_subnet', 'dns_ip', 'domain'):
|
||||
if field not in invite:
|
||||
raise ValueError(f"Invite missing field: {field!r}")
|
||||
_validate_invite_fields(invite)
|
||||
|
||||
links = self._load()
|
||||
name = invite['cell_name']
|
||||
@@ -567,7 +617,7 @@ class CellLinkManager:
|
||||
old_domain = existing.get('domain', '')
|
||||
existing['dns_ip'] = invite['dns_ip']
|
||||
existing['vpn_subnet'] = invite['vpn_subnet']
|
||||
existing['remote_api_url'] = _remote_api_url(invite['dns_ip'])
|
||||
existing['remote_api_url'] = _remote_api_url(invite['domain'])
|
||||
if invite.get('endpoint'):
|
||||
existing['endpoint'] = invite['endpoint']
|
||||
if domain_changed:
|
||||
@@ -629,7 +679,7 @@ class CellLinkManager:
|
||||
'domain': invite['domain'],
|
||||
'connected_at': datetime.utcnow().isoformat(),
|
||||
'permissions': _default_perms(),
|
||||
'remote_api_url': _remote_api_url(invite['dns_ip']),
|
||||
'remote_api_url': _remote_api_url(invite['domain']),
|
||||
'last_push_status': 'never',
|
||||
'last_push_at': None,
|
||||
'last_push_error': None,
|
||||
@@ -651,6 +701,7 @@ class CellLinkManager:
|
||||
def add_connection(self, invite: Dict[str, Any],
|
||||
inbound_services: Optional[List[str]] = None) -> Dict[str, Any]:
|
||||
"""Import a remote cell's invite and establish the connection."""
|
||||
_validate_invite_fields(invite)
|
||||
links = self._load()
|
||||
name = invite['cell_name']
|
||||
if any(l['cell_name'] == name for l in links):
|
||||
@@ -689,7 +740,7 @@ class CellLinkManager:
|
||||
'domain': invite['domain'],
|
||||
'connected_at': datetime.utcnow().isoformat(),
|
||||
'permissions': perms,
|
||||
'remote_api_url': _remote_api_url(invite['dns_ip']),
|
||||
'remote_api_url': _remote_api_url(invite['domain']),
|
||||
'last_push_status': 'never',
|
||||
'last_push_at': None,
|
||||
'last_push_error': None,
|
||||
@@ -747,8 +798,9 @@ class CellLinkManager:
|
||||
try:
|
||||
import firewall_manager as _fm
|
||||
_fm.clear_cell_rules(cell_name)
|
||||
_fm.remove_cell_subnet_route(link.get('vpn_subnet', ''))
|
||||
except Exception as e:
|
||||
logger.warning(f"clear_cell_rules for {cell_name} failed (non-fatal): {e}")
|
||||
logger.warning(f"firewall teardown for {cell_name} failed (non-fatal): {e}")
|
||||
|
||||
self.wireguard_manager.remove_peer(link['public_key'])
|
||||
self.network_manager.remove_cell_dns_forward(link['domain'])
|
||||
|
||||
@@ -438,11 +438,26 @@ def apply_cell_rules(cell_name: str, vpn_subnet: str, inbound_services: List[str
|
||||
_iptables(['-I', 'FORWARD', '-s', vpn_subnet, '-d', caddy_ip,
|
||||
'-p', 'tcp', '--dport', '443',
|
||||
'-m', 'comment', '--comment', tag, '-j', 'ACCEPT'])
|
||||
# Preserve the linked cell's real VPN source on peer-sync traffic:
|
||||
# the blanket `-o eth0 MASQUERADE` would rewrite it to cell-wireguard's
|
||||
# bridge IP, and the remote side authenticates the push by matching the
|
||||
# source (via X-Forwarded-For) to the cell's VPN subnet. RETURN before
|
||||
# the MASQUERADE (inserted at the top of nat POSTROUTING). Caddy's reply
|
||||
# to the real VPN IP routes back via the cell-subnet host route
|
||||
# (ensure_cell_subnet_routes). The :80 service path keeps masquerade.
|
||||
_iptables(['-t', 'nat', '-I', 'POSTROUTING', '-s', vpn_subnet,
|
||||
'-d', caddy_ip, '-p', 'tcp', '--dport', '443',
|
||||
'-m', 'comment', '--comment', tag, '-j', 'RETURN'])
|
||||
|
||||
# Ensure reply traffic (e.g. ICMP, TCP ACKs) for connections initiated
|
||||
# by local peers to this cell is not dropped by the cell's catch-all DROP.
|
||||
ensure_forward_stateful()
|
||||
|
||||
# Host route so Caddy's peer-sync reply (to the linked cell's un-masqueraded
|
||||
# VPN IP) leaves via cell-wireguard rather than the default gateway. Added at
|
||||
# startup for all links; ensure it on runtime link-add too. Idempotent.
|
||||
ensure_cell_subnet_routes([{'vpn_subnet': vpn_subnet}])
|
||||
|
||||
logger.info(
|
||||
f"Applied cell rules for {cell_name} ({vpn_subnet}): "
|
||||
f"inbound={inbound_services} exit_relay={exit_relay}"
|
||||
@@ -689,6 +704,25 @@ def ensure_cell_subnet_routes(cell_links: List[Dict[str, Any]]) -> None:
|
||||
logger.warning(f'ensure_cell_subnet_routes: {subnet}: {e}')
|
||||
|
||||
|
||||
def remove_cell_subnet_route(vpn_subnet: str) -> None:
|
||||
"""Remove the host route for a disconnected cell's VPN subnet (idempotent).
|
||||
|
||||
Counterpart to ensure_cell_subnet_routes. Without it the route lingers after a
|
||||
cell is unlinked — blackholing that subnet via cell-wireguard, and (on a host
|
||||
that runs the API/tests directly, e.g. a dev box) making is_local_request /
|
||||
_local_subnets treat the stale subnet as locally attached.
|
||||
"""
|
||||
if not vpn_subnet:
|
||||
return
|
||||
WG_BRIDGE_IP = '172.20.0.9'
|
||||
try:
|
||||
_run(['docker', 'run', '--rm', '--network', 'host', '--cap-add', 'NET_ADMIN',
|
||||
'alpine', 'ip', 'route', 'del', vpn_subnet, 'via', WG_BRIDGE_IP],
|
||||
check=False)
|
||||
except Exception as e:
|
||||
logger.warning(f'remove_cell_subnet_route: {vpn_subnet}: {e}')
|
||||
|
||||
|
||||
# ---------------------------------------------------------------------------
|
||||
# DNS ACL (CoreDNS Corefile generation)
|
||||
# ---------------------------------------------------------------------------
|
||||
|
||||
@@ -14,9 +14,39 @@ import json
|
||||
import shutil
|
||||
from unittest.mock import MagicMock, patch
|
||||
|
||||
import cell_link_manager
|
||||
from cell_link_manager import CellLinkManager
|
||||
|
||||
|
||||
_fw_patch = None
|
||||
|
||||
|
||||
def setUpModule():
|
||||
"""Stop cell-link unit tests from running real firewall side-effects.
|
||||
|
||||
add_connection/remove_connection call into firewall_manager, which shells out
|
||||
to `docker exec cell-wireguard iptables` and `docker run` host-route changes.
|
||||
On the dev/CI host those mutate live routes — a stale cell-subnet route once
|
||||
made is_local_request treat a VPN subnet as local and broke the full suite.
|
||||
Tests that assert specific firewall calls use their own local patch, which
|
||||
takes precedence within its context.
|
||||
"""
|
||||
global _fw_patch
|
||||
_fw_patch = patch.multiple(
|
||||
'firewall_manager',
|
||||
apply_cell_rules=MagicMock(return_value=True),
|
||||
clear_cell_rules=MagicMock(),
|
||||
ensure_cell_subnet_routes=MagicMock(),
|
||||
remove_cell_subnet_route=MagicMock(),
|
||||
)
|
||||
_fw_patch.start()
|
||||
|
||||
|
||||
def tearDownModule():
|
||||
if _fw_patch is not None:
|
||||
_fw_patch.stop()
|
||||
|
||||
|
||||
def _make_wg_mock():
|
||||
wg = MagicMock()
|
||||
wg.get_keys.return_value = {'public_key': 'serverpubkey=', 'private_key': 'serverprivkey='}
|
||||
@@ -50,6 +80,37 @@ SAMPLE_INVITE = {
|
||||
}
|
||||
|
||||
|
||||
class TestInviteFieldValidation(unittest.TestCase):
|
||||
"""_validate_invite_fields rejects malformed remote-invite fields.
|
||||
|
||||
The domain/dns_ip flow into a `curl --resolve` argv on peer-sync push, so a
|
||||
leading-dash domain (argument injection) and non-IP dns_ip must be rejected.
|
||||
"""
|
||||
|
||||
def test_valid_invite_passes(self):
|
||||
cell_link_manager._validate_invite_fields(SAMPLE_INVITE) # no raise
|
||||
|
||||
def test_rejects_leading_dash_domain(self):
|
||||
bad = {**SAMPLE_INVITE, 'domain': '-oProxyCommand=evil'}
|
||||
with self.assertRaises(ValueError):
|
||||
cell_link_manager._validate_invite_fields(bad)
|
||||
|
||||
def test_rejects_non_ip_dns_ip(self):
|
||||
bad = {**SAMPLE_INVITE, 'dns_ip': '-x'}
|
||||
with self.assertRaises(ValueError):
|
||||
cell_link_manager._validate_invite_fields(bad)
|
||||
|
||||
def test_rejects_bad_subnet(self):
|
||||
bad = {**SAMPLE_INVITE, 'vpn_subnet': 'not-a-cidr'}
|
||||
with self.assertRaises(ValueError):
|
||||
cell_link_manager._validate_invite_fields(bad)
|
||||
|
||||
def test_rejects_bad_endpoint(self):
|
||||
bad = {**SAMPLE_INVITE, 'endpoint': '-evil:51820'}
|
||||
with self.assertRaises(ValueError):
|
||||
cell_link_manager._validate_invite_fields(bad)
|
||||
|
||||
|
||||
class TestCellLinkManagerInvite(unittest.TestCase):
|
||||
|
||||
def setUp(self):
|
||||
@@ -146,6 +207,15 @@ class TestCellLinkManagerConnections(unittest.TestCase):
|
||||
self.mgr.remove_connection('office')
|
||||
self.nm.remove_cell_dns_forward.assert_called_once_with('office.cell')
|
||||
|
||||
def test_remove_connection_removes_host_subnet_route(self):
|
||||
"""Unlinking a cell removes its host route so the subnet isn't left
|
||||
blackholed / treated as locally attached."""
|
||||
import firewall_manager as _fm
|
||||
self.mgr.add_connection(SAMPLE_INVITE)
|
||||
_fm.remove_cell_subnet_route.reset_mock()
|
||||
self.mgr.remove_connection('office')
|
||||
_fm.remove_cell_subnet_route.assert_called_once_with('10.1.0.0/24')
|
||||
|
||||
def test_remove_connection_deletes_from_list(self):
|
||||
self.mgr.add_connection(SAMPLE_INVITE)
|
||||
self.mgr.remove_connection('office')
|
||||
@@ -190,7 +260,9 @@ class TestCellLinkManagerConnections(unittest.TestCase):
|
||||
result = self.mgr.accept_invite(updated_invite)
|
||||
|
||||
self.assertEqual(result['dns_ip'], '10.1.0.2')
|
||||
self.assertEqual(result['remote_api_url'], 'https://10.1.0.2')
|
||||
# remote_api_url is domain-based (the push --resolves it to the VPN IP),
|
||||
# so a dns_ip change does not alter it.
|
||||
self.assertEqual(result['remote_api_url'], 'https://office.cell')
|
||||
self.nm.remove_cell_dns_forward.assert_called()
|
||||
self.nm.add_cell_dns_forward.assert_called_with(
|
||||
domain='office.cell', dns_ip='10.1.0.2')
|
||||
@@ -615,7 +687,7 @@ class TestAcceptInviteNew(unittest.TestCase):
|
||||
with patch('firewall_manager.apply_cell_rules'):
|
||||
result = self.mgr.accept_invite(updated)
|
||||
self.assertEqual(result['dns_ip'], '10.1.0.5')
|
||||
self.assertEqual(result['remote_api_url'], 'https://10.1.0.5')
|
||||
self.assertEqual(result['remote_api_url'], 'https://office.cell')
|
||||
self.nm.remove_cell_dns_forward.assert_called()
|
||||
self.nm.add_cell_dns_forward.assert_called_with(
|
||||
domain='office.cell', dns_ip='10.1.0.5')
|
||||
@@ -1055,10 +1127,11 @@ class TestPermissionSync(unittest.TestCase):
|
||||
self.assertIn('last_push_at', link)
|
||||
self.assertIn('last_remote_update_at', link)
|
||||
|
||||
def test_add_connection_sets_remote_api_url_from_dns_ip(self):
|
||||
def test_add_connection_sets_remote_api_url_from_domain(self):
|
||||
link = self._add_office()
|
||||
# Cross-cell API is reached over the tunnel via Caddy/443, not :3000.
|
||||
self.assertEqual(link['remote_api_url'], 'https://10.1.0.1')
|
||||
# Cross-cell API is reached via the remote's domain over Caddy/443 (the
|
||||
# push --resolves the domain to the VPN IP over the tunnel).
|
||||
self.assertEqual(link['remote_api_url'], 'https://office.cell')
|
||||
|
||||
def test_add_connection_triggers_push(self):
|
||||
push_mock = MagicMock(return_value={'ok': True, 'error': None})
|
||||
@@ -1332,7 +1405,7 @@ class TestPermissionSync(unittest.TestCase):
|
||||
self.assertIn('last_push_status', link)
|
||||
self.assertIn('last_push_at', link)
|
||||
self.assertIn('last_remote_update_at', link)
|
||||
self.assertEqual(link['remote_api_url'], 'https://10.1.0.1')
|
||||
self.assertEqual(link['remote_api_url'], 'https://office.cell')
|
||||
self.assertTrue(link['pending_push']) # pre-existing → marked pending
|
||||
self.assertEqual(link['last_push_status'], 'never')
|
||||
|
||||
@@ -1341,9 +1414,11 @@ class TestPermissionSync(unittest.TestCase):
|
||||
raw = json.load(f)
|
||||
self.assertIn('pending_push', raw[0])
|
||||
|
||||
def test_load_migrates_legacy_http_3000_url_to_https(self):
|
||||
"""An existing link with the old http://<ip>:3000 URL (unreachable across
|
||||
cells) is rewritten to the HTTPS/Caddy form on load."""
|
||||
def test_load_migrates_legacy_url_forms_to_https_domain(self):
|
||||
"""Legacy remote_api_url forms — http://<ip>:3000 (unreachable) and
|
||||
https://<ip> (no matching Caddy cert by bare IP) — are rewritten on load to
|
||||
the domain-based HTTPS form."""
|
||||
for legacy_url in ('http://10.1.0.9:3000', 'https://10.1.0.9'):
|
||||
legacy = [{
|
||||
'cell_name': 'office',
|
||||
'public_key': 'officepubkey=',
|
||||
@@ -1351,13 +1426,14 @@ class TestPermissionSync(unittest.TestCase):
|
||||
'dns_ip': '10.1.0.9',
|
||||
'domain': 'office.cell',
|
||||
'permissions': {'inbound': {}, 'outbound': {}},
|
||||
'remote_api_url': 'http://10.1.0.9:3000',
|
||||
'remote_api_url': legacy_url,
|
||||
}]
|
||||
links_file = os.path.join(self.test_dir, 'cell_links.json')
|
||||
with open(links_file, 'w') as f:
|
||||
json.dump(legacy, f)
|
||||
link = self.mgr.list_connections()[0]
|
||||
self.assertEqual(link['remote_api_url'], 'https://10.1.0.9')
|
||||
self.assertEqual(link['remote_api_url'], 'https://office.cell',
|
||||
f'failed to migrate {legacy_url!r}')
|
||||
|
||||
|
||||
class TestExitOffer(unittest.TestCase):
|
||||
|
||||
@@ -652,6 +652,7 @@ class TestCellRules(unittest.TestCase):
|
||||
patch.object(firewall_manager, '_get_cell_api_ip', return_value=self._FAKE_API_IP), \
|
||||
patch.object(firewall_manager, '_get_caddy_container_ip', return_value=self._FAKE_CADDY_IP), \
|
||||
patch.object(firewall_manager, '_get_dns_container_ip', return_value=self._FAKE_DNS_IP), \
|
||||
patch.object(firewall_manager, 'ensure_cell_subnet_routes', return_value=None), \
|
||||
patch.object(firewall_manager, 'ensure_forward_stateful', return_value=True):
|
||||
firewall_manager.apply_cell_rules(cell_name, vpn_subnet, inbound_services)
|
||||
|
||||
@@ -742,6 +743,20 @@ class TestCellRules(unittest.TestCase):
|
||||
]
|
||||
self.assertFalse(api_3000, 'Peer-sync must not target cell-api:3000')
|
||||
|
||||
def test_apply_cell_rules_excludes_peer_sync_from_masquerade(self):
|
||||
"""Peer-sync to Caddy:443 must RETURN in nat POSTROUTING (skip the blanket
|
||||
MASQUERADE) so the remote sees the linked cell's real VPN source for auth."""
|
||||
calls = self._capture_apply('office', '10.0.1.0/24', [])
|
||||
returns = [
|
||||
c for c in calls
|
||||
if '-t' in c and 'nat' in c and 'POSTROUTING' in c
|
||||
and '-s' in c and '10.0.1.0/24' in c
|
||||
and '-d' in c and self._FAKE_CADDY_IP in c
|
||||
and '--dport' in c and '443' in c
|
||||
and '-j' in c and c[c.index('-j') + 1] == 'RETURN'
|
||||
]
|
||||
self.assertTrue(returns, 'Expected nat POSTROUTING RETURN to preserve peer-sync source')
|
||||
|
||||
def test_apply_cell_rules_api_sync_accept_before_catchall_drop(self):
|
||||
"""The API-sync ACCEPT must be inserted after service rules so it ends up above DROP."""
|
||||
insertion_order = []
|
||||
@@ -756,6 +771,7 @@ class TestCellRules(unittest.TestCase):
|
||||
patch.object(firewall_manager, '_get_cell_api_ip', return_value='172.20.0.10'), \
|
||||
patch.object(firewall_manager, '_get_caddy_container_ip', return_value='172.20.0.2'), \
|
||||
patch.object(firewall_manager, '_get_dns_container_ip', return_value='172.20.0.3'), \
|
||||
patch.object(firewall_manager, 'ensure_cell_subnet_routes', return_value=None), \
|
||||
patch.object(firewall_manager, 'ensure_forward_stateful', return_value=True):
|
||||
firewall_manager.apply_cell_rules('office', '10.0.1.0/24', [])
|
||||
|
||||
@@ -852,6 +868,31 @@ class TestCellRules(unittest.TestCase):
|
||||
# peer rule for a different entity must survive
|
||||
self.assertIn('pic-peer-10-0-0-2/32', content)
|
||||
|
||||
# ── remove_cell_subnet_route ──────────────────────────────────────────────
|
||||
|
||||
def test_remove_cell_subnet_route_issues_ip_route_del(self):
|
||||
"""remove_cell_subnet_route deletes the host route for the cell's subnet."""
|
||||
captured = {}
|
||||
|
||||
def fake_run(cmd, **kw):
|
||||
captured['cmd'] = cmd
|
||||
return MagicMock(returncode=0, stdout='', stderr='')
|
||||
|
||||
with patch.object(firewall_manager, '_run', side_effect=fake_run):
|
||||
firewall_manager.remove_cell_subnet_route('10.1.0.0/24')
|
||||
cmd = captured.get('cmd', [])
|
||||
self.assertIn('ip', cmd)
|
||||
self.assertIn('route', cmd)
|
||||
self.assertIn('del', cmd)
|
||||
self.assertIn('10.1.0.0/24', cmd)
|
||||
self.assertIn('172.20.0.9', cmd)
|
||||
|
||||
def test_remove_cell_subnet_route_noop_on_empty(self):
|
||||
"""An empty subnet is a no-op (no docker call)."""
|
||||
with patch.object(firewall_manager, '_run') as run:
|
||||
firewall_manager.remove_cell_subnet_route('')
|
||||
run.assert_not_called()
|
||||
|
||||
# ── apply_all_cell_rules ──────────────────────────────────────────────────
|
||||
|
||||
def test_apply_all_cell_rules_calls_apply_for_each(self):
|
||||
@@ -1130,6 +1171,7 @@ class TestEnsureForwardStateful(unittest.TestCase):
|
||||
patch.object(firewall_manager, '_get_caddy_container_ip', return_value='172.20.0.2'), \
|
||||
patch.object(firewall_manager, '_get_dns_container_ip', return_value='172.20.0.3'), \
|
||||
patch.object(firewall_manager, '_get_cell_api_ip', return_value='172.20.0.10'), \
|
||||
patch.object(firewall_manager, 'ensure_cell_subnet_routes', return_value=None), \
|
||||
patch.object(firewall_manager, 'ensure_forward_stateful') as mock_stateful:
|
||||
firewall_manager.apply_cell_rules('testcell', '10.0.0.0/24', [])
|
||||
mock_stateful.assert_called_once()
|
||||
|
||||
Reference in New Issue
Block a user