opnsense-nordvpn-plugin/tests/reconciler.php

235 lines
5.8 KiB
PHP
Raw Normal View History

<?php
// SPDX-License-Identifier: BSD-2-Clause
// SPDX-FileCopyrightText: 2026 ms2k
namespace OPNsense\NordVPN;
class Settings
{
public object $general;
public object $managed;
public function __construct()
{
$this->general = (object)['countries' => ''];
$this->managed = (object)[
'server_country' => 'NL',
'server_hostname' => 'nl.example.invalid',
];
}
}
$root = dirname(__DIR__);
foreach (['NordApi', 'EndpointStore', 'StatusStore', 'SystemRunner', 'TunnelHealth', 'ManagedConfiguration', 'Reconciler'] as $class) {
require_once $root . '/src/opnsense/mvc/app/library/OPNsense/NordVPN/' . $class . '.php';
}
class TestConfiguration extends ManagedConfiguration
{
private Settings $testSettings;
public function __construct()
{
$this->testSettings = new Settings();
}
public function settings(): Settings
{
return $this->testSettings;
}
public function interfaceName(): ?string
{
return 'wg0';
}
public function repairGatewayInterface(): bool
{
return false;
}
}
class AvailableHealth extends TunnelHealth
{
public function __construct()
{
}
public function check(string $interface): array
{
return ['available' => true, 'last_handshake' => 1000];
}
}
class FailOnceRunner extends SystemRunner
{
public int $firewallCalls = 0;
public function applyFirewall(): void
{
$this->firewallCalls++;
if ($this->firewallCalls === 1) {
throw new \RuntimeException('Synthetic firewall failure.');
}
}
}
class RepairConfiguration extends TestConfiguration
{
public function repairGatewayInterface(): bool
{
return true;
}
}
class RepairRunner extends SystemRunner
{
public int $routingCalls = 0;
public int $firewallCalls = 0;
public function applyRouting(): void
{
$this->routingCalls++;
}
public function applyFirewall(): void
{
$this->firewallCalls++;
}
}
class FailoverConfiguration extends TestConfiguration
{
public bool $candidateUpdated = false;
public int $repairCalls = 0;
public function updateCandidate(array $candidate): void
{
$this->candidateUpdated = true;
}
public function repairGatewayInterface(): bool
{
if (!$this->candidateUpdated) {
return false;
}
$this->candidateUpdated = false;
$this->repairCalls++;
return true;
}
}
class RecoveringHealth extends TunnelHealth
{
private int $checks = 0;
public function __construct()
{
}
public function check(string $interface): array
{
$this->checks++;
return ['available' => $this->checks > 1, 'last_handshake' => $this->checks > 1 ? 1000 : 0];
}
}
class FailoverRunner extends SystemRunner
{
public int $wireGuardCalls = 0;
public int $routingCalls = 0;
public int $firewallCalls = 0;
public function applyWireGuard(): void
{
$this->wireGuardCalls++;
}
public function applyRouting(): void
{
$this->routingCalls++;
}
public function applyFirewall(): void
{
$this->firewallCalls++;
}
}
$repairPath = sys_get_temp_dir() . '/nordvpn-repair-' . bin2hex(random_bytes(8)) . '.json';
$repairRunner = new RepairRunner();
(new Reconciler(
new NordApi(static fn() => []),
new RepairConfiguration(),
new AvailableHealth(),
new StatusStore($repairPath),
$repairRunner
))->run();
if ($repairRunner->routingCalls !== 1 || $repairRunner->firewallCalls !== 2) {
throw new \RuntimeException('Gateway interface repair was not applied.');
}
unlink($repairPath);
echo "ok - reconciliation applies gateway interface repair\n";
$failoverPath = sys_get_temp_dir() . '/nordvpn-failover-' . bin2hex(random_bytes(8)) . '.json';
$failoverConfiguration = new FailoverConfiguration();
$failoverRunner = new FailoverRunner();
$publicKey = base64_encode(str_repeat("\0", 32));
(new Reconciler(
new NordApi(static fn() => [[
'hostname' => 'nl2.example.invalid',
'station' => '198.51.100.2',
'locations' => [['country' => ['code' => 'NL']]],
'technologies' => [[
'id' => 35,
'pivot' => ['status' => 'online'],
'metadata' => [['name' => 'public_key', 'value' => $publicKey]],
]],
]]),
$failoverConfiguration,
new RecoveringHealth(),
new StatusStore($failoverPath),
$failoverRunner,
static fn($seconds) => null
))->run();
if ($failoverRunner->wireGuardCalls !== 1 ||
$failoverConfiguration->repairCalls !== 1 ||
$failoverRunner->routingCalls !== 1 ||
$failoverRunner->firewallCalls !== 1) {
throw new \RuntimeException('Server failover did not repair the gateway interface.');
}
unlink($failoverPath);
echo "ok - server failover repairs the gateway interface\n";
$path = sys_get_temp_dir() . '/nordvpn-retry-' . bin2hex(random_bytes(8)) . '.json';
$status = new StatusStore($path);
$runner = new FailOnceRunner();
$reconciler = new Reconciler(
new NordApi(static fn() => []),
new TestConfiguration(),
new AvailableHealth(),
$status,
$runner
);
try {
$reconciler->run();
throw new \RuntimeException('The first firewall failure was not reported.');
} catch (\RuntimeException $exception) {
if ($exception->getMessage() !== 'Synthetic firewall failure.') {
throw $exception;
}
}
if (empty($status->read()['firewall_pending'])) {
throw new \RuntimeException('The failed firewall apply was not retained as pending.');
}
$reconciler->run();
if ($runner->firewallCalls !== 2 || !empty($status->read()['firewall_pending'])) {
throw new \RuntimeException('The pending firewall apply was not completed on retry.');
}
unlink($path);
echo "ok - reconciliation retries a failed firewall apply\n";