1use std::sync::Arc;
11
12#[cfg(target_os = "linux")]
21struct PostureKernelVerifier {
22 strict: bool,
23 signing_keys: Vec<String>,
24 allowed_hashes: Vec<String>,
25 daemon: Option<Arc<boatramp_server::DaemonRuntime>>,
26}
27
28#[cfg(target_os = "linux")]
31impl std::fmt::Debug for PostureKernelVerifier {
32 fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
33 f.debug_struct("PostureKernelVerifier")
34 .field("strict", &self.strict)
35 .field("signing_keys", &self.signing_keys.len())
36 .field("allowed_hashes", &self.allowed_hashes.len())
37 .field("has_daemon", &self.daemon.is_some())
38 .finish()
39 }
40}
41
42#[cfg(target_os = "linux")]
43impl boatramp_firecracker::KernelVerifier for PostureKernelVerifier {
44 fn verify(&self, bytes: &[u8], expected_hash: &str) -> std::result::Result<(), String> {
46 let sig = self
50 .daemon
51 .as_ref()
52 .and_then(|d| d.effective().default_kernel.clone())
53 .filter(|dk| dk.sha256 == expected_hash)
54 .and_then(|dk| dk.sig);
55 let kref = boatramp_core::daemon_config::KernelRef {
56 source: expected_hash.to_string(),
57 sha256: expected_hash.to_string(),
58 sig,
59 };
60 boatramp_core::kernel_trust::verify_kernel(
61 bytes,
62 &kref,
63 self.strict,
64 &self.signing_keys,
65 &self.allowed_hashes,
66 )
67 .map_err(|e| e.to_string())
68 }
69}
70
71#[cfg(target_os = "macos")]
76struct VzPostureKernelVerifier {
77 strict: bool,
78 signing_keys: Vec<String>,
79 allowed_hashes: Vec<String>,
80 daemon: Option<Arc<boatramp_server::DaemonRuntime>>,
81}
82
83#[cfg(target_os = "macos")]
84impl std::fmt::Debug for VzPostureKernelVerifier {
85 fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
86 f.debug_struct("VzPostureKernelVerifier")
87 .field("strict", &self.strict)
88 .field("signing_keys", &self.signing_keys.len())
89 .field("allowed_hashes", &self.allowed_hashes.len())
90 .field("has_daemon", &self.daemon.is_some())
91 .finish()
92 }
93}
94
95#[cfg(target_os = "macos")]
96impl boatramp_vz::KernelVerifier for VzPostureKernelVerifier {
97 fn verify(&self, bytes: &[u8], expected_hash: &str) -> std::result::Result<(), String> {
98 let sig = self
99 .daemon
100 .as_ref()
101 .and_then(|d| d.effective().default_kernel.clone())
102 .filter(|dk| dk.sha256 == expected_hash)
103 .and_then(|dk| dk.sig);
104 let kref = boatramp_core::daemon_config::KernelRef {
105 source: expected_hash.to_string(),
106 sha256: expected_hash.to_string(),
107 sig,
108 };
109 boatramp_core::kernel_trust::verify_kernel(
110 bytes,
111 &kref,
112 self.strict,
113 &self.signing_keys,
114 &self.allowed_hashes,
115 )
116 .map_err(|e| e.to_string())
117 }
118}
119
120#[cfg(target_os = "macos")]
127fn macos_supports_vz() -> bool {
128 if cfg!(not(target_arch = "aarch64")) {
130 return false;
131 }
132 let major = sysctl_string("kern.osproductversion")
133 .and_then(|v| v.split('.').next().and_then(|m| m.parse::<u32>().ok()));
134 matches!(major, Some(m) if m >= 15)
135}
136
137#[cfg(target_os = "macos")]
140fn sysctl_string(name: &str) -> Option<String> {
141 let out = std::process::Command::new("sysctl")
142 .args(["-n", name])
143 .output()
144 .ok()?;
145 if !out.status.success() {
146 return None;
147 }
148 Some(String::from_utf8_lossy(&out.stdout).trim().to_string())
149}
150
151pub async fn build_compute(
158 cfg: Option<&crate::config::ComputeConfig>,
159 storage: std::sync::Arc<dyn boatramp_core::Storage>,
160 data_dir: &std::path::Path,
161 node_id: u64,
162 strict: bool,
163 daemon: Option<Arc<boatramp_server::DaemonRuntime>>,
164 worker_exe: Option<&std::path::Path>,
169) -> (
170 boatramp_core::compute::BackendRegistry,
171 boatramp_core::compute::Node,
172) {
173 use boatramp_core::compute::{BackendKind, BackendRegistry, Node};
174 let mut backends: BackendRegistry = std::collections::BTreeMap::new();
175 let empty_node = |id| Node {
176 id,
177 region: None,
178 labels: std::collections::BTreeMap::new(),
179 free_vcpus: 0,
180 free_mem_mib: 0,
181 backends: Vec::new(),
182 };
183 let Some(cfg) = cfg else {
184 return (backends, empty_node(node_id));
185 };
186
187 match boatramp_docker::DockerBackend::connect() {
189 Ok(docker) => {
190 let docker = docker
194 .with_endpoint(cfg.docker_endpoint)
195 .with_volume_mode(cfg.docker_volume_mode)
196 .with_data_dir(data_dir)
197 .with_writable_root_allowed(!strict)
198 .with_cap_add_allowed(!strict);
199 if docker.reachable().await {
200 backends.insert("docker".to_string(), std::sync::Arc::new(docker));
201 } else {
202 tracing::debug!("no reachable docker daemon; skipping docker backend");
203 }
204 }
205 Err(e) => tracing::debug!(%e, "docker backend unavailable"),
206 }
207
208 #[cfg(target_os = "linux")]
214 let bridge_ready = match boatramp_core::ipam::IpPool::new(&cfg.subnet) {
215 Ok(pool) => {
216 match boatramp_container::ensure_bridge(&cfg.bridge, pool.gateway(), pool.prefix_len())
217 .await
218 {
219 Ok(()) => true,
220 Err(e) => {
221 tracing::warn!(%e, bridge = %cfg.bridge, "could not create the compute bridge (need CAP_NET_ADMIN); container + embedded-VMM backends disabled");
222 false
223 }
224 }
225 }
226 Err(e) => {
227 tracing::warn!(%e, subnet = %cfg.subnet, "bad compute subnet; container + embedded-VMM backends disabled");
228 false
229 }
230 };
231
232 #[cfg(target_os = "linux")]
234 if bridge_ready {
235 match worker_exe.map_or_else(std::env::current_exe, |p| Ok(p.to_path_buf())) {
236 Ok(self_exe) => match boatramp_container::ContainerBackend::new(
237 storage.clone(),
238 data_dir.to_path_buf(),
239 cfg.bridge.clone(),
240 &cfg.subnet,
241 self_exe,
242 ) {
243 Ok(c) => {
244 let c = c.with_cap_add_allowed(!strict);
247 backends.insert("container".to_string(), std::sync::Arc::new(c));
248 }
249 Err(e) => tracing::warn!(%e, "container backend unavailable"),
250 },
251 Err(e) => tracing::warn!(%e, "current_exe for container backend"),
252 }
253 }
254 #[cfg(all(target_os = "linux", target_arch = "x86_64"))]
261 if bridge_ready && std::path::Path::new("/dev/kvm").exists() {
262 match (
263 worker_exe.map_or_else(std::env::current_exe, |p| Ok(p.to_path_buf())),
264 boatramp_core::ipam::IpPool::new(&cfg.subnet),
265 ) {
266 (Ok(self_exe), Ok(pool)) => {
267 let gateway = pool.gateway().to_string();
268 let verifier: Arc<dyn boatramp_firecracker::KernelVerifier> =
270 Arc::new(PostureKernelVerifier {
271 strict,
272 signing_keys: cfg.kernel_signing_pubkeys.clone(),
273 allowed_hashes: cfg.kernel_allowed_hashes.clone(),
274 daemon: daemon.clone(),
275 });
276 match boatramp_firecracker::EmbeddedVmmBackend::new(
277 storage.clone(),
278 self_exe, data_dir.to_path_buf(),
280 cfg.bridge.clone(),
281 gateway,
282 &cfg.subnet,
283 verifier,
284 ) {
285 Ok(vmm) => {
286 backends.insert("vmm-embedded".to_string(), std::sync::Arc::new(vmm));
287 }
288 Err(e) => tracing::warn!(%e, "embedded VMM backend unavailable"),
289 }
290 }
291 (Err(e), _) => tracing::warn!(%e, "current_exe for VMM backend"),
292 (_, Err(e)) => tracing::warn!(%e, "bad compute subnet for VMM backend"),
293 }
294 } else {
295 tracing::debug!("no /dev/kvm; skipping embedded VMM backend");
296 }
297
298 #[cfg(target_os = "macos")]
304 if macos_supports_vz() {
305 match (
306 worker_exe.map_or_else(std::env::current_exe, |p| Ok(p.to_path_buf())),
307 boatramp_core::ipam::IpPool::new(&cfg.subnet),
308 ) {
309 (Ok(self_exe), Ok(_pool)) => {
310 let verifier: Arc<dyn boatramp_vz::KernelVerifier> =
311 Arc::new(VzPostureKernelVerifier {
312 strict,
313 signing_keys: cfg.kernel_signing_pubkeys.clone(),
314 allowed_hashes: cfg.kernel_allowed_hashes.clone(),
315 daemon: daemon.clone(),
316 });
317 match boatramp_vz::VzBackend::new(
318 storage.clone(),
319 self_exe, data_dir.to_path_buf(),
321 &cfg.subnet, verifier,
323 ) {
324 Ok(vz) => {
326 let vz = vz.with_writable_root_allowed(!strict);
327 backends.insert("vmm-vz".to_string(), std::sync::Arc::new(vz));
328 }
329 Err(e) => tracing::warn!(%e, "macOS VMM backend unavailable"),
330 }
331 }
332 (Err(e), _) => tracing::warn!(%e, "current_exe for macOS VMM backend"),
333 (_, Err(e)) => tracing::warn!(%e, "bad compute subnet for macOS VMM backend"),
334 }
335 } else {
336 tracing::debug!("not Apple silicon + macOS 15+; skipping macOS VMM backend");
337 }
338
339 let _ = (&storage, data_dir); #[cfg(not(any(all(target_os = "linux", target_arch = "x86_64"), target_os = "macos")))]
344 let _ = (strict, &daemon);
345
346 let free_vcpus = if cfg.vcpus > 0 {
347 cfg.vcpus
348 } else {
349 std::thread::available_parallelism()
350 .map(|n| n.get() as u32)
351 .unwrap_or(1)
352 };
353 let free_mem_mib = if cfg.mem_mib > 0 { cfg.mem_mib } else { 1024 };
354 let advertised: Vec<BackendKind> = backends
355 .iter()
356 .map(|(id, b)| {
357 let caps = b.capabilities();
358 BackendKind {
359 id: id.clone(),
360 isolation: caps.isolation,
361 persistent_volumes: caps.persistent_volumes,
362 scale_to_zero: caps.scale_to_zero,
363 }
364 })
365 .collect();
366 tracing::info!(backends = ?advertised, free_vcpus, free_mem_mib, "compute node inventory");
367 let node = Node {
368 id: node_id,
369 region: cfg.region.clone(),
370 labels: std::collections::BTreeMap::new(),
371 free_vcpus,
372 free_mem_mib,
373 backends: advertised,
374 };
375 (backends, node)
376}
377
378pub struct NodeComputeExec {
386 backends: boatramp_core::compute::BackendRegistry,
387 deploy: boatramp_core::deploy::DeployStore,
388}
389
390impl NodeComputeExec {
391 pub fn new(
395 backends: boatramp_core::compute::BackendRegistry,
396 deploy: boatramp_core::deploy::DeployStore,
397 ) -> Self {
398 Self { backends, deploy }
399 }
400}
401
402#[async_trait::async_trait]
403impl boatramp_core::compute::ComputeExec for NodeComputeExec {
404 async fn exec(
405 &self,
406 project: &str,
407 workload: &str,
408 argv: &[String],
409 stdin: Option<&[u8]>,
410 ) -> Result<boatramp_core::compute::ExecOutput, boatramp_core::compute::ExecError> {
411 use boatramp_core::compute::{BackendError, ExecError, ReplicaPhase};
412 use boatramp_core::project::ProjectRef;
413 let states = self
414 .deploy
415 .list_replica_states(ProjectRef::new(project), workload)
416 .await
417 .map_err(|e| ExecError::Other(e.to_string()))?;
418 let target = states
421 .iter()
422 .find(|s| s.phase == ReplicaPhase::Running && s.healthy)
423 .or_else(|| states.iter().find(|s| s.phase == ReplicaPhase::Running))
424 .ok_or_else(|| ExecError::NoReplica(workload.to_string()))?;
425 let backend = self
426 .backends
427 .get(&target.backend)
428 .ok_or_else(|| ExecError::Unsupported(target.backend.clone()))?;
429 match backend.exec(&target.handle, argv, stdin).await {
430 Ok(out) => Ok(out),
431 Err(BackendError::Unsupported) => Err(ExecError::Unsupported(target.backend.clone())),
432 Err(e) => Err(ExecError::Other(e.to_string())),
433 }
434 }
435}