arcbox-core 0.6.8

Core orchestration layer for ArcBox
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78

//! `Runtime`'s implementation of the computer layer's [`SandboxHost`]
//! seam: the sandbox protocols sequence host state through this trait,
//! and the daemon-side ownership of listeners, DNS, and the generation
//! fence stays here.

use arcbox_computer::SandboxHost;
use arcbox_engine::agent_client::AgentClient;

use super::Runtime;

impl SandboxHost for Runtime {
    type StateGuard<'a> = tokio::sync::MutexGuard<'a, u64>;

    async fn lock_host_state(&self) -> Self::StateGuard<'_> {
        self.lock_sandbox_host_state().await
    }

    async fn clear_host_state(&self) {
        self.clear_sandbox_host_state().await;
    }

    async fn remove_ports(&self, sandbox_id: &str) {
        self.remove_sandbox_ports(sandbox_id).await;
    }

    async fn deregister_dns(&self, sandbox_id: &str) {
        self.deregister_sandbox_dns(sandbox_id).await;
    }

    async fn register_dns(&self, sandbox_id: &str, ip: std::net::IpAddr) {
        self.register_sandbox_dns(sandbox_id, ip).await;
    }

    async fn host_state_generation(&self) -> u64 {
        self.sandbox_host_state_generation().await
    }

    async fn expose_port(
        &self,
        machine: &str,
        exposure: &arcbox_computer::ports::SandboxPortExposure,
    ) -> arcbox_engine::Result<()> {
        use crate::error::CoreError;
        self.expose_sandbox_port(machine, exposure)
            .await
            .map_err(|e| match e {
                // Common/Net cover the non-macOS fallback (PortForwarder's
                // direct TCP/UDP connect can fail with either). On P0
                // macOS, `start_port_forwarding_macos` only ever
                // constructs `CoreError::Machine` — Common/Net are
                // unreachable there, and `other` is the sole arm that
                // fires. Kept as a match on the full error type, not a
                // cfg'd one, so a future macOS error path lands here
                // losslessly without a second edit site.
                CoreError::Common(c) => arcbox_engine::EngineError::Common(c),
                CoreError::Net(n) => arcbox_engine::EngineError::Net(n),
                other => arcbox_engine::EngineError::Machine(other.to_string()),
            })
    }

    async fn unexpose_port(&self, sandbox_id: &str, sandbox_port: u16, protocol_key: &str) {
        self.unexpose_sandbox_port(sandbox_id, sandbox_port, protocol_key)
            .await;
    }

    async fn port_mappings_if_unchanged(
        &self,
        sandbox_id: &str,
        expected_generation: u64,
    ) -> Option<Vec<arcbox_computer::ports::SandboxPortMapping>> {
        self.sandbox_port_mappings_if_unchanged(sandbox_id, expected_generation)
            .await
    }

    fn agent(&self, machine: &str) -> arcbox_engine::Result<AgentClient> {
        self.machine_manager().connect_agent(machine)
    }
}