Skip to main content

alien_core/resources/
sandbox.rs

1//! Sandbox resource for running untrusted code in an isolated environment.
2//!
3//! The declaration provisions a durable parent, and the application creates and destroys
4//! individual sandboxes through its binding at runtime.
5//!
6//! The capability set differs per platform and is published rather than assumed. Calling an
7//! unsupported capability is a typed error naming both the platform and the capability, so a
8//! portable application can branch on `SandboxCapabilities` before it calls.
9
10use crate::error::{ErrorData, Result};
11use crate::resource::{ResourceDefinition, ResourceOutputsDefinition, ResourceRef, ResourceType};
12use crate::resources::ToolchainConfig;
13use crate::Platform;
14use alien_error::AlienError;
15use bon::Builder;
16use serde::{Deserialize, Serialize};
17use sha2::{Digest, Sha256};
18use std::any::Any;
19use std::fmt::Debug;
20
21/// Specifies where the sandbox's root filesystem comes from.
22#[derive(Debug, Clone, PartialEq, Eq, Serialize, Deserialize)]
23#[cfg_attr(feature = "openapi", derive(utoipa::ToSchema))]
24#[serde(rename_all = "camelCase", tag = "type")]
25pub enum SandboxCode {
26    /// A prebuilt container image used as the sandbox root filesystem.
27    #[serde(rename_all = "camelCase")]
28    Image {
29        /// Image reference (e.g. `ubuntu:24.04`, `ghcr.io/myorg/sandbox:latest`). AWS wants an
30        /// `s3://` bundle; Azure takes a catalog name such as `ubuntu` or an amd64 registry
31        /// image, told apart by syntax. Each refuses what it cannot take while planning.
32        image: String,
33        /// The container image a publisher built the `s3://` bundle in `image` from. An AWS
34        /// deployment with a bundle Storage of its own renders its bundle from this image; one
35        /// whose setup granted the publisher's store builds from `image`. Set by a publisher,
36        /// never by authoring. Kept inside `code` so a controller that predates it ignores it.
37        #[serde(default, skip_serializing_if = "Option::is_none")]
38        #[cfg_attr(feature = "openapi", schema(ignore))]
39        bundled_from: Option<String>,
40    },
41    /// A Dockerfile `alien build` builds into the sandbox's base image.
42    ///
43    /// AWS, Azure and GCP, and docker only: the base image is a root filesystem, not a binary laid
44    /// on one. `alien release` pushes it; on AWS the bundle layers the sandbox agent on afterwards,
45    /// and on GCP the release layers it on and pushes the result.
46    #[serde(rename_all = "camelCase")]
47    Source {
48        /// The source directory to build from
49        src: String,
50        /// Toolchain configuration with type-safe options
51        toolchain: ToolchainConfig,
52    },
53}
54
55/// Hard ceilings enforced on a sandbox.
56///
57/// These are limits, not scheduling requests. Untrusted code does not respect a hint, so every
58/// field is enforced by the platform and a platform that cannot enforce one is rejected at plan
59/// time rather than silently ignoring it.
60#[derive(Debug, Clone, PartialEq, Eq, Serialize, Deserialize)]
61#[cfg_attr(feature = "openapi", derive(utoipa::ToSchema))]
62#[serde(rename_all = "camelCase", deny_unknown_fields)]
63pub struct SandboxLimits {
64    /// CPU ceiling in cores or millicores (e.g. `"1"`, `"500m"`)
65    pub cpu: String,
66    /// Memory ceiling (e.g. `"2Gi"`, `"512Mi"`)
67    pub memory: String,
68    /// Disk ceiling (e.g. `"20Gi"`)
69    pub disk: String,
70    /// Maximum number of processes, which bounds fork bombs.
71    ///
72    /// Optional because only a container runtime has the primitive: Kubernetes sets a pid ceiling
73    /// per node, not per pod, and neither AWS MicroVMs nor Azure sandboxes expose one. Declaring
74    /// it on a platform that cannot apply it is refused at plan time.
75    #[serde(default, skip_serializing_if = "Option::is_none")]
76    pub max_processes: Option<u32>,
77}
78
79/// One of the five sizes a Lambda MicroVM can be built at.
80///
81/// AWS has no ceiling knob: `minimumMemoryInMiB` sets a *baseline* and a running MicroVM bursts
82/// vertically to four times it with no way to opt out. A declared ceiling is therefore honoured by
83/// picking the tier whose **peak** stays inside it, not the tier whose baseline matches it.
84#[derive(Debug, Clone, Copy, PartialEq, Eq)]
85pub struct MicrovmTier {
86    /// What `minimumMemoryInMiB` is set to.
87    pub baseline_memory_mib: i64,
88    /// The most memory the MicroVM can reach, in MiB.
89    pub peak_memory_mib: i64,
90    /// The most vCPU the MicroVM can reach.
91    pub peak_vcpu: u32,
92    /// The most disk the MicroVM can use, in MiB.
93    pub max_disk_mib: i64,
94}
95
96/// The published sizes, smallest first. Baseline memory to vCPU is 2 GB per vCPU, peak is four
97/// times baseline, and disk is fixed per tier rather than independently selectable.
98/// Longest life AWS will run a MicroVM for, from `RunMicrovm`'s `maximumDurationInSeconds`.
99const AWS_MAX_LIFETIME_SECONDS: u32 = 28_800;
100
101/// Longest `activeDeadlineSeconds` the Kubernetes API server accepts on a pod. The field is an
102/// int64, but pod validation caps it at the largest i32.
103const KUBERNETES_MAX_LIFETIME_SECONDS: u32 = i32::MAX as u32;
104
105/// Azure's sandbox sizing rule, quoted from the data plane's own refusal of an oversized request:
106/// *CPU must be n×250m for n=1..64 (0.25–16 cores); Memory ≤ cores × 2Gi; Disk ≤ cores × 20Gi*.
107const AZURE_CPU_STEP_MILLICORES: i64 = 250;
108const AZURE_MAX_CPU_MILLICORES: i64 = 16_000;
109const AZURE_MEMORY_MIB_PER_CORE: i64 = 2 * 1024;
110const AZURE_DISK_MIB_PER_CORE: i64 = 20 * 1024;
111
112const MICROVM_TIERS: &[MicrovmTier] = &[
113    MicrovmTier {
114        baseline_memory_mib: 512,
115        peak_memory_mib: 2048,
116        peak_vcpu: 1,
117        max_disk_mib: 8192,
118    },
119    MicrovmTier {
120        baseline_memory_mib: 1024,
121        peak_memory_mib: 4096,
122        peak_vcpu: 2,
123        max_disk_mib: 8192,
124    },
125    MicrovmTier {
126        baseline_memory_mib: 2048,
127        peak_memory_mib: 8192,
128        peak_vcpu: 4,
129        max_disk_mib: 8192,
130    },
131    MicrovmTier {
132        baseline_memory_mib: 4096,
133        peak_memory_mib: 16384,
134        peak_vcpu: 8,
135        max_disk_mib: 16384,
136    },
137    MicrovmTier {
138        baseline_memory_mib: 8192,
139        peak_memory_mib: 32768,
140        peak_vcpu: 16,
141        max_disk_mib: 32768,
142    },
143];
144
145pub use super::sandbox_policy::{is_public_ipv4, EgressPolicy};
146
147/// Outbound policy. Only the new Internet value opts into gateway enforcement.
148#[derive(Debug, Clone, PartialEq, Eq)]
149pub enum SandboxEgress {
150    None,
151    Internet,
152    /// Deprecated direct backend access, unchanged from released deployments.
153    Allow,
154    /// Deprecated backend hostname allowlist, unchanged from released deployments.
155    AllowDomains {
156        domains: Vec<String>,
157    },
158}
159
160/// Deprecated object declarations preserve their released backend behavior.
161#[derive(Debug, Clone, Serialize, Deserialize)]
162#[cfg_attr(feature = "openapi", derive(utoipa::ToSchema))]
163#[serde(tag = "mode", rename_all = "camelCase", deny_unknown_fields)]
164pub enum LegacySandboxEgress {
165    /// Deprecated alias of "none".
166    Deny,
167    /// Deprecated direct backend access. Use "internet" to opt into the gateway,
168    /// fixed outbound IP, connection logs and per-sandbox runtime policies.
169    Allow,
170    /// Deprecated backend hostname rules. Use "internet" and runtime policies
171    /// to opt into gateway enforcement instead.
172    AllowDomains { domains: Vec<String> },
173}
174
175#[derive(Serialize, Deserialize)]
176#[serde(rename_all = "camelCase")]
177enum SandboxEgressMode {
178    None,
179    Internet,
180}
181
182#[derive(Serialize, Deserialize)]
183#[serde(untagged)]
184enum SandboxEgressInput {
185    Mode(SandboxEgressMode),
186    Legacy(LegacySandboxEgress),
187}
188
189impl Serialize for SandboxEgress {
190    fn serialize<S: serde::Serializer>(
191        &self,
192        serializer: S,
193    ) -> std::result::Result<S::Ok, S::Error> {
194        match self {
195            Self::None => serializer.serialize_str("none"),
196            Self::Internet => serializer.serialize_str("internet"),
197            Self::Allow => LegacySandboxEgress::Allow.serialize(serializer),
198            Self::AllowDomains { domains } => LegacySandboxEgress::AllowDomains {
199                domains: domains.clone(),
200            }
201            .serialize(serializer),
202        }
203    }
204}
205
206impl<'de> Deserialize<'de> for SandboxEgress {
207    fn deserialize<D: serde::Deserializer<'de>>(
208        deserializer: D,
209    ) -> std::result::Result<Self, D::Error> {
210        Ok(match SandboxEgressInput::deserialize(deserializer)? {
211            SandboxEgressInput::Mode(SandboxEgressMode::None)
212            | SandboxEgressInput::Legacy(LegacySandboxEgress::Deny) => Self::None,
213            SandboxEgressInput::Mode(SandboxEgressMode::Internet) => Self::Internet,
214            SandboxEgressInput::Legacy(LegacySandboxEgress::Allow) => Self::Allow,
215            SandboxEgressInput::Legacy(LegacySandboxEgress::AllowDomains { domains }) => {
216                Self::AllowDomains { domains }
217            }
218        })
219    }
220}
221
222#[cfg(feature = "openapi")]
223impl utoipa::PartialSchema for SandboxEgress {
224    fn schema() -> utoipa::openapi::RefOr<utoipa::openapi::schema::Schema> {
225        use utoipa::openapi::schema::{ObjectBuilder, OneOfBuilder, Type};
226        OneOfBuilder::new()
227            .item(
228                ObjectBuilder::new()
229                    .schema_type(Type::String)
230                    .enum_values(Some(["none", "internet"])),
231            )
232            .item(LegacySandboxEgress::schema())
233            .into()
234    }
235}
236#[cfg(feature = "openapi")]
237impl utoipa::ToSchema for SandboxEgress {}
238
239impl SandboxEgress {
240    /// Deprecated spelling retained for Rust callers of the released API.
241    #[allow(non_upper_case_globals)]
242    #[deprecated(note = "use SandboxEgress::None")]
243    pub const Deny: Self = Self::None;
244
245    /// Backend switch; hostname rules must be handled by a backend matcher.
246    pub fn internet_access_switch(&self) -> Option<bool> {
247        match self {
248            Self::None => Some(false),
249            Self::Internet | Self::Allow => Some(true),
250            Self::AllowDomains { .. } => None,
251        }
252    }
253}
254
255/// Per-session deadline metadata read by the external lifecycle reaper.
256pub const SANDBOX_EXPIRY_LABEL: &str = "alienExpiresAtMs";
257
258/// Azure gateway identities cannot renew without an in-guest agent.
259pub const AZURE_GATEWAY_MAX_LIFETIME_SECONDS: u32 = 24 * 60 * 60;
260
261/// How long a sandbox may live and when it is paused.
262///
263/// Declaration-time ceilings, not per-request values: every sandbox created through this
264/// declaration's binding is held to them, whatever a caller asks for at runtime.
265#[derive(Debug, Clone, PartialEq, Eq, Serialize, Deserialize)]
266#[cfg_attr(feature = "openapi", derive(utoipa::ToSchema))]
267#[serde(rename_all = "camelCase", deny_unknown_fields)]
268pub struct SandboxLifecyclePolicy {
269    /// Wall-clock ceiling on a single sandbox, after which the platform terminates it.
270    ///
271    /// Optional because not every backend has the primitive: Kubernetes has
272    /// `activeDeadlineSeconds` and AWS `maximumDurationInSeconds`, while neither Azure nor Local
273    /// expose one, so declaring a ceiling there is refused at plan time rather than accepted and
274    /// never applied. AWS caps it at 8 hours.
275    #[serde(default, skip_serializing_if = "Option::is_none")]
276    pub max_lifetime_seconds: Option<u32>,
277    /// Idle period after which the sandbox is paused, where the platform supports it.
278    ///
279    /// Stored state written before the rename calls this `idleSuspendSeconds`.
280    #[serde(alias = "idleSuspendSeconds", skip_serializing_if = "Option::is_none")]
281    pub idle_pause_seconds: Option<u32>,
282}
283
284/// What a platform's sandbox backend can actually do.
285///
286/// Published so portable code can branch before calling rather than discovering a gap through
287/// an error. Every field here corresponds to a capability that at least one platform lacks;
288/// create, exec and terminate are the guaranteed floor and are therefore not listed.
289#[derive(Debug, Clone, Copy, PartialEq, Eq, Serialize, Deserialize)]
290#[cfg_attr(feature = "openapi", derive(utoipa::ToSchema))]
291#[serde(rename_all = "camelCase", deny_unknown_fields)]
292pub struct SandboxCapabilities {
293    /// Files can be moved in and out of a sandbox
294    pub files: bool,
295    /// A later call can reach a sandbox created by an earlier one
296    pub reconnect: bool,
297    /// A command can be started, polled and cancelled across separate calls, so it outlives the
298    /// one that started it. False where nothing inside the sandbox owns the process in between.
299    pub jobs: bool,
300    /// A running command takes input on stdin, can be signalled, and streams its output as it is
301    /// written, so a caller can drive a REPL, a language server or a tool protocol over it.
302    pub interactive_commands: bool,
303    /// An authenticated, port-scoped capability to reach a service inside the sandbox
304    pub preview: bool,
305    /// Sandbox state can be paused and resumed
306    pub pause_resume: bool,
307    /// A sandbox's full state can be captured and used to create another
308    pub snapshot: bool,
309    /// Egress can be restricted to a hostname allowlist
310    pub domain_egress_rules: bool,
311    /// Whether a declared `deny` is actually enforced, rather than accepted and dropped.
312    /// It covers routed traffic; the `deny` mode says where DNS still resolves.
313    pub egress_deny: bool,
314    /// The platform enforces the declared cpu, memory and disk ceilings
315    pub enforced_limits: bool,
316    /// The platform can cap how many processes a sandbox runs
317    pub process_limit: bool,
318    /// The platform terminates a sandbox at a declared wall-clock deadline
319    pub sandbox_lifetime: bool,
320    /// A command runs in its own PID namespace and cannot see or signal the agent's processes.
321    ///
322    /// Only where an agent runs as root. Creating the namespace needs `CAP_SYS_ADMIN`, and the
323    /// Kubernetes sandbox pod drops every capability — which is also what denies `ptrace` by
324    /// construction, so granting it there would remove a lock to add one.
325    pub supervisor_pid_namespace: bool,
326    /// The process supervising a command is a different identity from the command.
327    ///
328    /// False where a command runs as the agent's own user: it can then read the supervisor's
329    /// environment and signal it. Separate from `supervisorPidNamespace`, which is about
330    /// visibility rather than identity — a backend can have one without the other.
331    pub supervisor_isolation: bool,
332}
333
334impl SandboxCapabilities {
335    /// Returns what the given platform's sandbox backend supports.
336    ///
337    /// Errors for platforms with no sandbox backend, rather than returning an all-false set —
338    /// "every capability is missing" and "this platform has no sandboxes" are different
339    /// conditions and an application should not have to tell them apart by inspection.
340    pub fn for_platform(platform: Platform) -> Result<Self> {
341        match platform {
342            Platform::Aws => Ok(Self {
343                files: true,
344                reconnect: true,
345                jobs: true,
346                // The in-guest agent owns the process and its pipes, and the MicroVM proxy
347                // streams a response while separate requests carry input and signals.
348                interactive_commands: true,
349                preview: true,
350                pause_resume: true,
351                snapshot: false,
352                domain_egress_rules: false,
353                // Routed egress only: while the connector VPC has DNS support on, a session
354                // resolves names through its resolver, which no security group filters.
355                egress_deny: true,
356                enforced_limits: true,
357                // Nothing in the API bounds process count.
358                process_limit: false,
359                // `maximumDurationInSeconds` on `RunMicrovm`, which Lambda enforces by
360                // terminating the MicroVM. Capped at 8 hours by the service.
361                sandbox_lifetime: true,
362                // Measured, not assumed: the agent inside a Lambda MicroVM runs as uid 0 with
363                // `CapEff: 00000000a80425fb`, the standard container default set, which excludes
364                // `CAP_SYS_ADMIN`. It can drop privilege (`CAP_SETUID`/`CAP_SETGID` are held) and
365                // it cannot create a namespace. No backend offers this today.
366                supervisor_pid_namespace: false,
367                // The agent runs as uid 0 and `setuid`s the command to uid 60000, so the command
368                // runs under a different identity than the process supervising it.
369                supervisor_isolation: true,
370            }),
371            Platform::Azure => Ok(Self::azure()),
372            Platform::Gcp => Ok(Self::gcp_agent_platform()),
373            // Preview needs a gateway that validates a sandbox-and-port capability, which this
374            // backend has none of.
375            Platform::Kubernetes => Ok(Self {
376                files: true,
377                reconnect: true,
378                jobs: true,
379                // The pod runs the same agent, but this client does not drive its interactive
380                // routes yet, so the capability is not claimed until it is proven here.
381                interactive_commands: false,
382                preview: false,
383                pause_resume: false,
384                snapshot: false,
385                domain_egress_rules: false,
386                egress_deny: true,
387                enforced_limits: true,
388                // A pid ceiling is a kubelet setting per node, not a pod field.
389                process_limit: false,
390                // `activeDeadlineSeconds` on the pod, which the kubelet enforces.
391                sandbox_lifetime: true,
392                // The pod drops every capability, including the `CAP_SYS_ADMIN` the agent would
393                // need to unshare. That is also what denies `ptrace`, so this stays false rather
394                // than the pod being weakened to make it true.
395                supervisor_pid_namespace: false,
396                // The pod pins one uid (`run_as_user: 65534` on both pod and container) with
397                // `capabilities.drop: [ALL]` and `allow_privilege_escalation: false`, so no
398                // process can setuid to split the command off from a supervisor. No uid split is
399                // possible, so none exists.
400                supervisor_isolation: false,
401            }),
402            Platform::Local => Ok(Self {
403                files: true,
404                reconnect: true,
405                // Nothing runs inside the sandbox: the manager drives Docker from outside it.
406                jobs: false,
407                // The manager holds the exec's attached stdin for as long as the output stream
408                // is open, so an interactive command needs nothing inside the sandbox.
409                interactive_commands: true,
410                preview: true,
411                pause_resume: false,
412                snapshot: false,
413                domain_egress_rules: false,
414                egress_deny: true,
415                enforced_limits: true,
416                // Docker's `--pids-limit`.
417                process_limit: true,
418                sandbox_lifetime: false,
419                // Local has no in-sandbox agent: the manager drives Docker from outside, so
420                // there is no supervisor inside the sandbox to isolate from.
421                supervisor_pid_namespace: false,
422                // The supervisor is the manager on the host, outside the container entirely, and
423                // `docker exec` runs the command as the workload uid — a different identity by
424                // construction.
425                supervisor_isolation: true,
426            }),
427            Platform::Machines | Platform::Test => {
428                Err(AlienError::new(ErrorData::SandboxPlatformUnsupported {
429                    platform: platform.to_string(),
430                }))
431            }
432        }
433    }
434
435    /// What the Azure sandbox backend supports; the body of the `Platform::Azure` arm.
436    pub fn azure() -> Self {
437        Self {
438            files: true,
439            reconnect: true,
440            // No Alien process runs inside the sandbox to own a command between two calls.
441            jobs: false,
442            // Not claimed: the agent-mode client does not drive the agent's interactive routes
443            // yet, and the platform's own exec takes no stdin.
444            interactive_commands: false,
445            // A sandbox port carries a URL and an auth config, and the auth config offers two
446            // things: anonymous, or Entra ID with an allowlist of human email addresses.
447            // Neither is a credential scoped to a port for a fixed time, which is what a
448            // preview capability is. Returning the anonymous URL would publish the port.
449            preview: false,
450            pause_resume: true,
451            // False for a client reason, not a cloud one: this client has no snapshot call, and
452            // `CreateSandboxRequest` has no field to consume the id it would return. Also
453            // unclaimed: Microsoft does not garbage-collect snapshots, so an id is a bill that grows.
454            snapshot: false,
455            domain_egress_rules: true,
456            egress_deny: true,
457            // Enforced inside the sandbox, not at create: an over-allocation raises `MemoryError`
458            // while the sandbox keeps running. `azure_sandbox_limits` checks the continuous
459            // sizing rule at plan time instead of matching a tier.
460            enforced_limits: true,
461            process_limit: false,
462            // Auto-suspend and auto-delete exist; a wall-clock ceiling does not. Accepting
463            // `maxLifetimeSeconds` here would be the silent no-op the capability set exists
464            // to prevent, so this is a decision rather than a gap.
465            sandbox_lifetime: false,
466            // No Alien process inside an Azure sandbox, so there is no supervisor to isolate.
467            supervisor_pid_namespace: false,
468            // No Alien process runs the command at all — the platform's own data plane does,
469            // so there is no separate supervisor identity to speak of.
470            supervisor_isolation: false,
471        }
472    }
473
474    /// What the GCP Agent Platform sandbox backend supports; the body of the `Platform::Gcp` arm.
475    pub fn gcp_agent_platform() -> Self {
476        Self {
477            // Agent file operations move over the sandbox envelope.
478            files: true,
479            // Reaching a sandbox across processes is safe because `generation` is derived from the
480            // container boot id read through the agent's health op, so a caller detects a container
481            // replaced under a stable sandbox name rather than reconnecting to a blank one.
482            reconnect: true,
483            jobs: true,
484            // `:execute` is one request with one answer, capped at about 30 seconds. Input
485            // cannot reach a command after its call has answered, so there is no stream to drive.
486            interactive_commands: false,
487            // No method mints a port-scoped ingress capability; the only ingress is `:execute`.
488            preview: false,
489            // `:resume` can return a fresh container while reporting success, so a pause does not
490            // keep the sandbox's state.
491            pause_resume: false,
492            // The create path never sends `sandbox_environment_snapshot`, so no sandbox state is
493            // reachable through the trait; declared false until the client carries it.
494            snapshot: false,
495            // Egress is shaped by VPC and DNS peering, which is not a hostname allowlist.
496            domain_egress_rules: false,
497            // A declared `deny` blocks both routed egress and DNS.
498            egress_deny: true,
499            // The declared ceilings are enforced, but by terminating the sandbox on breach rather
500            // than by refusing the allocation — a caller reading `true` should expect the sandbox
501            // to die, not a clean error at the point of the request.
502            enforced_limits: true,
503            // No ceiling on process count is observed.
504            process_limit: false,
505            // `ttl` maps to a sandbox `expireTime` the platform terminates at.
506            sandbox_lifetime: true,
507            // No PID-namespace isolation between the command and anything supervising it.
508            supervisor_pid_namespace: false,
509            // No separate supervisor identity: the command is not run under a different identity
510            // than the process supervising it.
511            supervisor_isolation: false,
512        }
513    }
514
515    /// Returns a typed error if the named capability is absent on this platform.
516    pub fn require(&self, capability: SandboxCapability, platform: Platform) -> Result<()> {
517        let available = match capability {
518            SandboxCapability::Files => self.files,
519            SandboxCapability::Reconnect => self.reconnect,
520            SandboxCapability::Jobs => self.jobs,
521            SandboxCapability::InteractiveCommands => self.interactive_commands,
522            SandboxCapability::Preview => self.preview,
523            SandboxCapability::PauseResume => self.pause_resume,
524            SandboxCapability::Snapshot => self.snapshot,
525            SandboxCapability::DomainEgressRules => self.domain_egress_rules,
526            SandboxCapability::EgressDeny => self.egress_deny,
527            SandboxCapability::EnforcedLimits => self.enforced_limits,
528            SandboxCapability::ProcessLimit => self.process_limit,
529            SandboxCapability::SandboxLifetime => self.sandbox_lifetime,
530            SandboxCapability::SupervisorPidNamespace => self.supervisor_pid_namespace,
531            SandboxCapability::SupervisorIsolation => self.supervisor_isolation,
532        };
533
534        if available {
535            return Ok(());
536        }
537
538        Err(AlienError::new(ErrorData::SandboxCapabilityUnsupported {
539            capability: capability.as_str().to_string(),
540            platform: platform.to_string(),
541        }))
542    }
543}
544
545/// Names a single sandbox capability, so an unsupported call can report which one it needed.
546#[derive(Debug, Clone, Copy, PartialEq, Eq, Serialize, Deserialize)]
547#[cfg_attr(feature = "openapi", derive(utoipa::ToSchema))]
548#[serde(rename_all = "camelCase")]
549pub enum SandboxCapability {
550    /// Moving files in and out of a sandbox
551    Files,
552    /// Reaching a sandbox created by an earlier call
553    Reconnect,
554    /// Starting, polling and cancelling a command across separate calls
555    Jobs,
556    /// Writing to, signalling and streaming a running command
557    InteractiveCommands,
558    /// An authenticated, port-scoped ingress capability
559    Preview,
560    /// Pausing and resuming sandbox state
561    PauseResume,
562    /// Capturing full sandbox state
563    Snapshot,
564    /// Restricting egress to a hostname allowlist
565    DomainEgressRules,
566    /// Refusing outbound access when a sandbox declares none
567    EgressDeny,
568    /// Platform-enforced resource ceilings
569    EnforcedLimits,
570    /// A ceiling on the number of processes a sandbox may run
571    ProcessLimit,
572    /// A wall-clock ceiling on a sandbox, applied by the platform rather than by a caller
573    SandboxLifetime,
574    /// A command runs in its own PID namespace, isolated from the agent supervising it
575    SupervisorPidNamespace,
576    /// A command runs under a different identity than the process supervising it
577    SupervisorIsolation,
578}
579
580impl SandboxCapability {
581    /// Returns the stable identifier used in errors and capability queries.
582    pub fn as_str(&self) -> &'static str {
583        match self {
584            Self::Files => "files",
585            Self::Reconnect => "reconnect",
586            Self::Jobs => "jobs",
587            Self::InteractiveCommands => "interactiveCommands",
588            Self::Preview => "preview",
589            Self::PauseResume => "pauseResume",
590            Self::Snapshot => "snapshot",
591            Self::DomainEgressRules => "domainEgressRules",
592            Self::EgressDeny => "egressDeny",
593            Self::EnforcedLimits => "enforcedLimits",
594            Self::ProcessLimit => "processLimit",
595            Self::SandboxLifetime => "sandboxLifetime",
596            Self::SupervisorPidNamespace => "supervisorPidNamespace",
597            Self::SupervisorIsolation => "supervisorIsolation",
598        }
599    }
600}
601
602/// Opt-in supervisor-owned network enforcement. Caller requests cannot change this identity.
603#[derive(Debug, Clone, PartialEq, Eq, Serialize, Deserialize)]
604#[cfg_attr(feature = "openapi", derive(utoipa::ToSchema))]
605#[serde(rename_all = "camelCase", deny_unknown_fields)]
606pub struct SandboxPrivilegedSupervisor {
607    /// Nonzero numeric uid and primary gid for every command, including the image entrypoint.
608    pub command_uid: u32,
609}
610
611/// An isolated environment for running untrusted code, created at runtime.
612#[derive(Debug, Clone, PartialEq, Eq, Serialize, Deserialize, Builder)]
613#[cfg_attr(feature = "openapi", derive(utoipa::ToSchema))]
614#[serde(rename_all = "camelCase", deny_unknown_fields)]
615#[builder(start_fn = new)]
616pub struct Sandbox {
617    /// Identifier for the sandbox. Must be DNS-compatible: lowercase letters, digits and hyphens,
618    /// starting and ending with a letter or digit, with no consecutive hyphens.
619    #[builder(start_fn)]
620    pub id: String,
621    /// Runtime network links authored by preflight, not parent lifecycle dependencies.
622    /// Sandbox has no public link builder; gateway reach is wired during planning.
623    #[builder(field)]
624    #[serde(default, skip_serializing_if = "Vec::is_empty")]
625    #[cfg_attr(feature = "openapi", schema(ignore))]
626    pub links: Vec<ResourceRef>,
627    /// Where the sandbox's root filesystem comes from
628    pub code: SandboxCode,
629    /// Original image uid resolved from the built OCI archives. Absent on released artifacts
630    /// that retain the legacy boot contract. This is build output, not an authoring option.
631    #[builder(skip)]
632    #[serde(default, skip_serializing_if = "Option::is_none")]
633    #[cfg_attr(feature = "openapi", schema(ignore))]
634    pub image_user: Option<u32>,
635    /// Private ECR base image an AWS build pulls; `code.image` names only the S3 bundle, so this is
636    /// what the cross-account grant opens. Live only: the grant needs the customer account,
637    /// which registration reports. Absent means the base image is pulled anonymously.
638    #[serde(skip_serializing_if = "Option::is_none")]
639    pub private_base_image: Option<String>,
640    /// Enforced resource ceilings.
641    ///
642    /// Optional because not every platform can enforce them, and a declaration that names none
643    /// takes the platform's own defaults. Naming them on a platform that cannot enforce them is
644    /// rejected at plan time rather than silently ignored.
645    #[serde(skip_serializing_if = "Option::is_none")]
646    pub limits: Option<SandboxLimits>,
647    /// Outbound network policy
648    pub egress: SandboxEgress,
649    /// Default runtime policy for creates that omit one.
650    #[serde(default, skip_serializing_if = "Option::is_none")]
651    pub default_policy: Option<EgressPolicy>,
652    /// Have Alien's agent install the declared egress policy before running any image code.
653    /// Unsupported backends refuse this at plan time.
654    // Keep released declarations readable without offering a new authoring knob.
655    #[builder(skip)]
656    #[serde(default, skip_serializing_if = "Option::is_none")]
657    pub privileged_supervisor: Option<SandboxPrivilegedSupervisor>,
658    /// Sandbox lifetime ceiling and idle behaviour.
659    ///
660    /// Stored state written before the rename calls this `session`. A stack state or release
661    /// that old must stay readable, otherwise its deployment can no longer be updated or deleted.
662    #[serde(alias = "session")]
663    pub lifecycle: SandboxLifecyclePolicy,
664    /// Ports eligible for a preview capability. An application reaches its sandbox through the
665    /// provider, so it cannot widen its own ingress at runtime; a holder of a remote binding's
666    /// credentials is bounded by no port condition, which is why a remote sandbox declares none.
667    #[builder(default)]
668    #[serde(default, skip_serializing_if = "Vec::is_empty")]
669    pub preview_ports: Vec<u16>,
670}
671
672/// Whether the artifact being rendered restricts which network modes it accepts.
673///
674/// Cloud setup needs explicit subnets for restricted sandbox connectors. Kubernetes targets do
675/// not emit these cloud backends.
676pub fn restricts_network_mode(stack: &crate::Stack, targets_kubernetes: bool) -> bool {
677    !targets_kubernetes && stack_needs_named_subnets_at_setup(stack)
678}
679
680/// Whether any setup-owned resource forces setup to name subnets.
681///
682/// Restricted sandbox connectors require subnet IDs. Callers rendering an artifact want
683/// [`restricts_network_mode`] instead:
684/// this one answers for the declaration, which on a Kubernetes target is not what gets emitted.
685pub fn stack_needs_named_subnets_at_setup(stack: &crate::Stack) -> bool {
686    stack.resources().any(|(_, entry)| {
687        entry
688            .config
689            .downcast_ref::<Sandbox>()
690            .is_some_and(|sandbox| crate::sandbox_egress::sandbox_needs_connector(stack, sandbox))
691    })
692}
693
694impl Sandbox {
695    pub(crate) fn setup_digest_value(&self) -> serde_json::Value {
696        match &self.egress {
697            SandboxEgress::None => serde_json::json!({"mode":"deny"}),
698            value => serde_json::to_value(value).expect("egress serializes"),
699        }
700    }
701
702    /// The resource type identifier for Sandbox
703    pub const RESOURCE_TYPE: ResourceType = ResourceType::from_static("sandbox");
704
705    /// Returns the sandbox's unique identifier.
706    pub fn id(&self) -> &str {
707        &self.id
708    }
709
710    /// Released supervisor declarations retain their direct cloud route; only
711    /// the new Internet declaration uses the deployment gateway.
712    pub fn cloud_egress(&self) -> &SandboxEgress {
713        if self.privileged_supervisor.is_some() && self.egress != SandboxEgress::Internet {
714            &SandboxEgress::Allow
715        } else {
716            &self.egress
717        }
718    }
719
720    /// Startup contract for the privileged agent; never supplied by an exec caller.
721    pub fn supervisor_environment(&self) -> std::collections::BTreeMap<String, String> {
722        let mut env = std::collections::BTreeMap::new();
723        if let Some(supervisor) = &self.privileged_supervisor {
724            env.insert(
725                "ALIEN_SANDBOX_EXPECTED_EXEC_UID".to_string(),
726                supervisor.command_uid.to_string(),
727            );
728            env.insert(
729                "ALIEN_SANDBOX_EGRESS".to_string(),
730                // A runtime default is mutable gateway policy, never a startup allowlist.
731                if self.egress == SandboxEgress::None {
732                    r#"{"mode":"deny"}"#.to_string()
733                } else {
734                    serde_json::to_string(&self.egress).expect("egress serializes")
735                },
736            );
737        }
738        env
739    }
740
741    /// The declared ceilings, or the defaults a platform applies when none were named.
742    ///
743    /// Backends want a concrete set: a sandbox with no declared ceilings still runs inside
744    /// whatever the platform gives it, and a backend that had to branch on `None` would end up
745    /// inventing its own default anyway.
746    pub fn resolved_limits(&self) -> SandboxLimits {
747        self.limits.clone().unwrap_or_else(default_limits)
748    }
749
750    /// Validates the declaration against what the target platform can enforce.
751    ///
752    /// Runs at plan time so an unenforceable limit or an unsupported egress mode fails before
753    /// anything is provisioned, rather than at the first exec.
754    pub fn validate_for_platform(&self, platform: Platform) -> Result<()> {
755        if let Some(policy) = &self.default_policy {
756            if self.egress != SandboxEgress::Internet {
757                return Err(AlienError::new(ErrorData::SandboxEgressInvalid {
758                    target: "defaultPolicy".into(),
759                    reason: "defaultPolicy requires the new internet declaration".into(),
760                }));
761            }
762            policy.validate()?;
763        }
764        if self.image_user == Some(0) && matches!(platform, Platform::Local | Platform::Kubernetes)
765        {
766            return Err(AlienError::new(ErrorData::SandboxCapabilityUnsupported {
767                capability: "root image USER: this backend does not yet separate bounded root boot from non-root exec; use a non-root image USER".into(),
768                platform: platform.to_string(),
769            }));
770        }
771        let mut capabilities = SandboxCapabilities::for_platform(platform)?;
772        if let Some(supervisor) = &self.privileged_supervisor {
773            if platform != Platform::Aws {
774                return Err(AlienError::new(ErrorData::SandboxCapabilityUnsupported {
775                    capability: "privilegedSupervisor".to_string(),
776                    platform: platform.to_string(),
777                }));
778            }
779            if supervisor.command_uid == 0 || supervisor.command_uid == u32::MAX {
780                return Err(AlienError::new(ErrorData::SandboxLimitInvalid {
781                    resource_id: self.id.clone(),
782                    field: "privilegedSupervisor.commandUid".to_string(),
783                    value: supervisor.command_uid.to_string(),
784                    reason: "must be a non-root Linux uid other than the invalid uid sentinel"
785                        .to_string(),
786                }));
787            }
788            if let SandboxEgress::AllowDomains { domains } = &self.egress {
789                for domain in domains {
790                    let hostname = domain.strip_suffix('.').unwrap_or(domain);
791                    if hostname.len() > 253
792                        || !hostname.split('.').all(|label| {
793                            !label.is_empty()
794                                && label.len() <= 63
795                                && label
796                                    .bytes()
797                                    .all(|c| c.is_ascii_alphanumeric() || c == b'-')
798                                && !label.starts_with('-')
799                                && !label.ends_with('-')
800                        })
801                    {
802                        return Err(AlienError::new(ErrorData::SandboxLimitInvalid {
803                            resource_id: self.id.clone(), field: "egress.domains".to_string(), value: domain.clone(),
804                            reason: "privileged supervisor allowlists require exact DNS hostnames; wildcards are unsupported".to_string(),
805                        }));
806                    }
807                }
808            }
809            capabilities.domain_egress_rules = true;
810        }
811
812        // `alien build` builds an AWS, Azure or GCP sandbox's base image, so source is a
813        // declaration there and the emitters refuse it only if it reaches them unbuilt. Everywhere
814        // else the image is pulled as declared, and an empty image string would schedule a pod
815        // that can never run.
816        if matches!(&self.code, SandboxCode::Source { .. })
817            && !matches!(platform, Platform::Aws | Platform::Azure | Platform::Gcp)
818        {
819            return Err(AlienError::new(ErrorData::SandboxLimitInvalid {
820                resource_id: self.id.clone(),
821                field: "code".to_string(),
822                value: "source".to_string(),
823                reason: format!(
824                    "no sandbox backend builds an image from source on {platform}; give \
825                     code.image a prebuilt reference"
826                ),
827            }));
828        }
829
830        // Elsewhere `code.image` is pulled directly, so a second reference would be a grant
831        // nothing reads.
832        if self.private_base_image.is_some() && platform != Platform::Aws {
833            return Err(AlienError::new(ErrorData::SandboxCapabilityUnsupported {
834                capability: "privateBaseImage".to_string(),
835                platform: platform.to_string(),
836            }));
837        }
838
839        // Read before the limits, because the image is declared whether or not any are. A source
840        // declaration has no image until `alien build` writes one.
841        if platform == Platform::Azure && matches!(&self.code, SandboxCode::Image { .. }) {
842            self.azure_image()?;
843        }
844
845        // Read before the limits, like the image, because a lifetime is declared whether or not
846        // any limits are.
847        if let Some(seconds) = self.lifecycle.max_lifetime_seconds {
848            // First, so a platform with no deadline says so before any complaint about the value.
849            capabilities.require(SandboxCapability::SandboxLifetime, platform)?;
850
851            if seconds == 0 {
852                return Err(AlienError::new(ErrorData::SandboxLimitInvalid {
853                    resource_id: self.id.clone(),
854                    field: "maxLifetimeSeconds".to_string(),
855                    value: "0".to_string(),
856                    reason: "a sandbox must be allowed to run for at least 1 second".to_string(),
857                }));
858            }
859
860            // Each ceiling is the platform's own, and the platform rejects a value above it
861            // rather than clamping, so it would pass planning, render into the package, and fail
862            // at the first sandbox. The ceilings differ, which is why they sit under a platform
863            // match rather than on the type.
864            let ceiling = match platform {
865                Platform::Aws => Some((AWS_MAX_LIFETIME_SECONDS, "AWS runs a MicroVM for")),
866                Platform::Kubernetes => {
867                    Some((KUBERNETES_MAX_LIFETIME_SECONDS, "Kubernetes runs a pod for"))
868                }
869                // The Agent Platform documents `ttl` without a valid range, so a value it refuses
870                // still fails at the first sandbox rather than here.
871                Platform::Gcp => None,
872                // Never reached: the capability check above refuses Azure and Local, and
873                // `for_platform` refuses Machines and Test at the top of this function.
874                Platform::Azure | Platform::Local | Platform::Machines | Platform::Test => None,
875            };
876            if let Some((max, runs_for)) = ceiling {
877                if seconds > max {
878                    return Err(AlienError::new(ErrorData::SandboxLimitInvalid {
879                        resource_id: self.id.clone(),
880                        field: "maxLifetimeSeconds".to_string(),
881                        value: seconds.to_string(),
882                        reason: format!("{runs_for} between 1 and {max} seconds"),
883                    }));
884                }
885            }
886        }
887
888        let Some(limits) = self.limits.as_ref() else {
889            // Nothing declared, so nothing to enforce and nothing to reject.
890            return self.validate_capabilities(&capabilities, platform);
891        };
892
893        validate_quantity(&self.id, "cpu", &limits.cpu)?;
894        validate_quantity(&self.id, "memory", &limits.memory)?;
895        validate_quantity(&self.id, "disk", &limits.disk)?;
896
897        if let Some(max_processes) = limits.max_processes {
898            if max_processes == 0 {
899                return Err(AlienError::new(ErrorData::SandboxLimitInvalid {
900                    resource_id: self.id.clone(),
901                    field: "maxProcesses".to_string(),
902                    value: "0".to_string(),
903                    reason: "a sandbox that may run no processes cannot run code".to_string(),
904                }));
905            }
906            capabilities.require(SandboxCapability::ProcessLimit, platform)?;
907        }
908
909        // Declaring limits a platform ignores is worse than not declaring them: the stack reads
910        // as bounded while the sandbox is not.
911        capabilities.require(SandboxCapability::EnforcedLimits, platform)?;
912
913        if platform == Platform::Azure {
914            self.azure_sandbox_limits()?;
915        }
916
917        if platform == Platform::Aws {
918            // Refused here rather than at emit so a customer sees it while planning, and so both
919            // package formats inherit the same answer.
920            self.microvm_tier()?;
921        }
922
923        self.validate_capabilities(&capabilities, platform)
924    }
925
926    /// What Azure creates this sandbox from: a catalog name or a registry image, told apart by
927    /// [`classify_azure_sandbox_image`]. Refused while planning, because a value the data plane
928    /// rejects would otherwise surface at the first sandbox, long after the apply.
929    pub fn azure_image(&self) -> Result<AzureSandboxImage<'_>> {
930        let SandboxCode::Image { image, .. } = &self.code else {
931            return Err(AlienError::new(ErrorData::SandboxLimitInvalid {
932                resource_id: self.id.clone(),
933                field: "code".to_string(),
934                value: "source".to_string(),
935                reason: "the sandbox still has unbuilt source code. Run 'alien build' first"
936                    .to_string(),
937            }));
938        };
939
940        classify_azure_sandbox_image(image).ok_or_else(|| {
941            let image = image.trim();
942            AlienError::new(ErrorData::SandboxLimitInvalid {
943                resource_id: self.id.clone(),
944                field: "code.image".to_string(),
945                value: image.to_string(),
946                reason: if image.is_empty() {
947                    "a sandbox has to name an image".to_string()
948                } else {
949                    "Azure creates a sandbox from a catalog name such as 'ubuntu' or from a \
950                     registry image such as 'docker.io/library/python:3.14-slim'"
951                        .to_string()
952                },
953            })
954        })
955    }
956
957    /// Checks the declared ceilings against Azure's sizing rule (the `AZURE_*` constants above).
958    /// Refused at plan time, like [`Self::microvm_tier`], so a bad value is a declaration to fix
959    /// rather than a runtime fault at create.
960    pub fn azure_sandbox_limits(&self) -> Result<()> {
961        let Some(limits) = self.limits.as_ref() else {
962            // Nothing declared means the binding substitutes Alien's own default sizing, which is
963            // inside the rule — asserted where those constants live, since this cannot see them.
964            return Ok(());
965        };
966
967        let refused = |field: &str, value: &str, reason: &str| {
968            AlienError::new(ErrorData::SandboxLimitInvalid {
969                resource_id: self.id.clone(),
970                field: field.to_string(),
971                value: value.to_string(),
972                reason: reason.to_string(),
973            })
974        };
975
976        let cpu_millicores = millicores(&limits.cpu)
977            .ok_or_else(|| refused("cpu", &limits.cpu, "expected cores or millicores"))?;
978
979        // The multiple is checked, not just the range: `333m` sits inside 0.25–16 cores and is
980        // still refused on the wire, so a bounds-only check would pass a declaration that fails
981        // at create.
982        if cpu_millicores % AZURE_CPU_STEP_MILLICORES != 0
983            || !(AZURE_CPU_STEP_MILLICORES..=AZURE_MAX_CPU_MILLICORES).contains(&cpu_millicores)
984        {
985            return Err(refused(
986                "cpu",
987                &limits.cpu,
988                "Azure allocates cpu in steps of 250m from 250m to 16000m",
989            ));
990        }
991
992        // Both ceilings are derived from the cpu, so they cannot be checked before it is known.
993        let memory_ceiling_mib = cpu_millicores * AZURE_MEMORY_MIB_PER_CORE / 1000;
994        let disk_ceiling_mib = cpu_millicores * AZURE_DISK_MIB_PER_CORE / 1000;
995
996        let memory_mib = quantity_mib(&limits.memory)
997            .ok_or_else(|| refused("memory", &limits.memory, "Azure sizes memory in whole MiB"))?;
998        if memory_mib > memory_ceiling_mib {
999            return Err(refused(
1000                "memory",
1001                &limits.memory,
1002                &format!(
1003                    "Azure allows at most 2Gi of memory per core, or {memory_ceiling_mib}Mi \
1004                          at the declared cpu"
1005                ),
1006            ));
1007        }
1008
1009        let disk_mib = quantity_mib(&limits.disk)
1010            .ok_or_else(|| refused("disk", &limits.disk, "Azure sizes disk in whole MiB"))?;
1011        if disk_mib > disk_ceiling_mib {
1012            return Err(refused(
1013                "disk",
1014                &limits.disk,
1015                &format!(
1016                    "Azure allows at most 20Gi of disk per core, or {disk_ceiling_mib}Mi at \
1017                          the declared cpu"
1018                ),
1019            ));
1020        }
1021
1022        Ok(())
1023    }
1024
1025    /// The MicroVM size that keeps every declared ceiling, or why none does.
1026    ///
1027    /// AWS sizes are discrete and a running MicroVM bursts to four times its baseline, so the
1028    /// only tier that honours a ceiling is one whose peak fits inside it. A declaration no tier
1029    /// satisfies is refused: shipping the nearest size would give the customer a sandbox that
1030    /// exceeds the bound they wrote down.
1031    pub fn microvm_tier(&self) -> Result<MicrovmTier> {
1032        let Some(limits) = self.limits.as_ref() else {
1033            // Nothing declared: AWS's own default baseline, which is also `default_limits`.
1034            return Ok(MICROVM_TIERS[2]);
1035        };
1036
1037        let memory_mib = quantity_mib(&limits.memory).ok_or_else(|| {
1038            AlienError::new(ErrorData::SandboxLimitInvalid {
1039                resource_id: self.id.clone(),
1040                field: "memory".to_string(),
1041                value: limits.memory.clone(),
1042                reason: "AWS sizes a MicroVM in whole MiB".to_string(),
1043            })
1044        })?;
1045        let disk_mib = quantity_mib(&limits.disk).ok_or_else(|| {
1046            AlienError::new(ErrorData::SandboxLimitInvalid {
1047                resource_id: self.id.clone(),
1048                field: "disk".to_string(),
1049                value: limits.disk.clone(),
1050                reason: "AWS sizes a MicroVM's disk in whole MiB".to_string(),
1051            })
1052        })?;
1053        let cpu_millicores = millicores(&limits.cpu).ok_or_else(|| {
1054            AlienError::new(ErrorData::SandboxLimitInvalid {
1055                resource_id: self.id.clone(),
1056                field: "cpu".to_string(),
1057                value: limits.cpu.clone(),
1058                reason: "expected cores or millicores".to_string(),
1059            })
1060        })?;
1061
1062        // Memory and disk choose the size; cpu is then checked rather than used to choose.
1063        // AWS couples cpu to memory at 2 GB per vCPU, so letting a low cpu ceiling select the
1064        // size too would quietly hand back a machine four times smaller than the memory ceiling
1065        // asked for, with nothing to indicate it.
1066        let sized = |tier: &&MicrovmTier| {
1067            tier.peak_memory_mib <= memory_mib && tier.max_disk_mib <= disk_mib
1068        };
1069
1070        let tier = MICROVM_TIERS
1071            .iter()
1072            .rev()
1073            .find(sized)
1074            .copied()
1075            .ok_or_else(|| {
1076                AlienError::new(ErrorData::SandboxLimitInvalid {
1077                    resource_id: self.id.clone(),
1078                    field: "memory".to_string(),
1079                    value: limits.memory.clone(),
1080                    reason: format!(
1081                        "a Lambda MicroVM bursts to four times its baseline, so the smallest \
1082                         ceiling AWS can hold is 2Gi memory with 8Gi disk; '{}' memory and '{}' \
1083                         disk fit no size",
1084                        limits.memory, limits.disk
1085                    ),
1086                })
1087            })?;
1088
1089        let required_millicores = i64::from(tier.peak_vcpu) * 1000;
1090        if cpu_millicores < required_millicores {
1091            return Err(AlienError::new(ErrorData::SandboxLimitInvalid {
1092                resource_id: self.id.clone(),
1093                field: "cpu".to_string(),
1094                value: limits.cpu.clone(),
1095                reason: format!(
1096                    "AWS allocates one vCPU per 2GB, so a MicroVM sized to a '{}' memory ceiling \
1097                     reaches {} vCPU; declare cpu '{}' or lower the memory ceiling",
1098                    limits.memory, tier.peak_vcpu, tier.peak_vcpu
1099                ),
1100            }));
1101        }
1102
1103        Ok(tier)
1104    }
1105
1106    /// The capability checks that do not depend on declared limits.
1107    fn validate_capabilities(
1108        &self,
1109        capabilities: &SandboxCapabilities,
1110        platform: Platform,
1111    ) -> Result<()> {
1112        if matches!(self.egress, SandboxEgress::AllowDomains { .. }) {
1113            capabilities.require(SandboxCapability::DomainEgressRules, platform)?;
1114        }
1115
1116        // `allow` asks for no restriction, so a backend that ignores it fails loudly on the first
1117        // blocked connection. `deny` asks for one, and a backend that ignores it puts untrusted
1118        // code on the internet with nothing to notice — so only this direction is gated.
1119        // An empty list is not a restriction anyone wrote down: it renders as a deny-all wearing
1120        // an allowlist's label, which reads at a glance as the opposite of what it does.
1121        if let SandboxEgress::AllowDomains { domains } = &self.egress {
1122            if domains.is_empty() {
1123                return Err(AlienError::new(ErrorData::SandboxLimitInvalid {
1124                    resource_id: self.id.clone(),
1125                    field: "egress.domains".to_string(),
1126                    value: "[]".to_string(),
1127                    reason: "an allowlist naming no domain denies everything; declare \
1128                             egress: deny if that is what was meant"
1129                        .to_string(),
1130                }));
1131            }
1132        }
1133
1134        if matches!(self.egress, SandboxEgress::None) {
1135            capabilities.require(SandboxCapability::EgressDeny, platform)?;
1136        }
1137
1138        if !self.preview_ports.is_empty() {
1139            capabilities.require(SandboxCapability::Preview, platform)?;
1140        }
1141
1142        if self.lifecycle.idle_pause_seconds.is_some() {
1143            capabilities.require(SandboxCapability::PauseResume, platform)?;
1144        }
1145
1146        if self.lifecycle.max_lifetime_seconds.is_some() {
1147            capabilities.require(SandboxCapability::SandboxLifetime, platform)?;
1148        }
1149
1150        Ok(())
1151    }
1152}
1153
1154/// Ceilings applied when a declaration names none.
1155///
1156/// Modest on purpose: an undeclared sandbox is one whose author did not think about sizing, and
1157/// the safe reading of that is a small box rather than a generous one.
1158fn default_limits() -> SandboxLimits {
1159    SandboxLimits {
1160        cpu: "1".to_string(),
1161        memory: "2Gi".to_string(),
1162        disk: "8Gi".to_string(),
1163        max_processes: None,
1164    }
1165}
1166
1167/// Validates a Kubernetes-style resource quantity such as `500m`, `2Gi` or `1`.
1168fn validate_quantity(resource_id: &str, field: &str, value: &str) -> Result<()> {
1169    let invalid = |reason: &str| {
1170        AlienError::new(ErrorData::SandboxLimitInvalid {
1171            resource_id: resource_id.to_string(),
1172            field: field.to_string(),
1173            value: value.to_string(),
1174            reason: reason.to_string(),
1175        })
1176    };
1177
1178    let digits_end = value
1179        .find(|c: char| !c.is_ascii_digit() && c != '.')
1180        .unwrap_or(value.len());
1181    let (number, suffix) = value.split_at(digits_end);
1182
1183    let parsed: f64 = number
1184        .parse()
1185        .map_err(|_| invalid("expected a number, optionally followed by a unit suffix"))?;
1186
1187    if parsed <= 0.0 {
1188        return Err(invalid("must be greater than zero"));
1189    }
1190
1191    const SUFFIXES: &[&str] = &["", "m", "k", "M", "G", "T", "Ki", "Mi", "Gi", "Ti"];
1192    if !SUFFIXES.contains(&suffix) {
1193        return Err(invalid(
1194            "unit must be one of m, k, M, G, T, Ki, Mi, Gi, Ti, or absent",
1195        ));
1196    }
1197
1198    Ok(())
1199}
1200
1201/// Splits a quantity into its number and unit suffix.
1202fn split_quantity(value: &str) -> Option<(f64, &str)> {
1203    let trimmed = value.trim();
1204    let digits_end = trimmed
1205        .find(|c: char| !c.is_ascii_digit() && c != '.')
1206        .unwrap_or(trimmed.len());
1207    let (number, suffix) = trimmed.split_at(digits_end);
1208    number.parse().ok().map(|number| (number, suffix))
1209}
1210
1211/// A memory or disk quantity in whole MiB, rounded down.
1212///
1213/// Every suffix `validate_quantity` accepts is handled here. Reading only `Gi` and `Mi` and
1214/// falling back for the rest would turn a declared `4G` into a different size than the customer
1215/// asked for, which for a ceiling means a sandbox larger than its bound.
1216pub fn quantity_mib(value: &str) -> Option<i64> {
1217    let (number, suffix) = split_quantity(value)?;
1218    let bytes = match suffix {
1219        "" => number,
1220        "k" => number * 1e3,
1221        "M" => number * 1e6,
1222        "G" => number * 1e9,
1223        "T" => number * 1e12,
1224        "Ki" => number * 1024.0,
1225        "Mi" => number * 1024.0 * 1024.0,
1226        "Gi" => number * 1024.0 * 1024.0 * 1024.0,
1227        "Ti" => number * 1024.0 * 1024.0 * 1024.0 * 1024.0,
1228        // `m` is a millicore suffix; memory has no use for it.
1229        _ => return None,
1230    };
1231    Some((bytes / (1024.0 * 1024.0)) as i64)
1232}
1233
1234/// A CPU quantity in millicores.
1235pub fn millicores(value: &str) -> Option<i64> {
1236    let (number, suffix) = split_quantity(value)?;
1237    match suffix {
1238        "" => Some((number * 1000.0) as i64),
1239        "m" => Some(number as i64),
1240        _ => None,
1241    }
1242}
1243
1244/// Outputs generated by a successfully provisioned Sandbox parent.
1245#[derive(Debug, Clone, PartialEq, Eq, Serialize, Deserialize)]
1246#[cfg_attr(feature = "openapi", derive(utoipa::ToSchema))]
1247#[serde(rename_all = "camelCase")]
1248pub struct SandboxOutputs {
1249    /// Name of the durable parent that sandboxes are created inside
1250    pub parent_name: String,
1251    /// Platform-specific identifier for the parent (image ARN, sandbox group id, namespace)
1252    #[serde(skip_serializing_if = "Option::is_none")]
1253    pub identifier: Option<String>,
1254    /// Data-plane endpoint sandboxes are created through, where the platform has one
1255    #[serde(skip_serializing_if = "Option::is_none")]
1256    pub endpoint: Option<String>,
1257}
1258
1259impl ResourceOutputsDefinition for SandboxOutputs {
1260    fn get_resource_type(&self) -> ResourceType {
1261        Sandbox::RESOURCE_TYPE
1262    }
1263
1264    fn as_any(&self) -> &dyn Any {
1265        self
1266    }
1267
1268    fn box_clone(&self) -> Box<dyn ResourceOutputsDefinition> {
1269        Box::new(self.clone())
1270    }
1271
1272    fn outputs_eq(&self, other: &dyn ResourceOutputsDefinition) -> bool {
1273        other.as_any().downcast_ref::<SandboxOutputs>() == Some(self)
1274    }
1275
1276    fn to_json_value(&self) -> serde_json::Result<serde_json::Value> {
1277        serde_json::to_value(self)
1278    }
1279}
1280
1281impl ResourceDefinition for Sandbox {
1282    fn get_resource_type(&self) -> ResourceType {
1283        Self::RESOURCE_TYPE
1284    }
1285
1286    fn id(&self) -> &str {
1287        &self.id
1288    }
1289
1290    fn replace_after_failed_create_is_safe(&self) -> bool {
1291        // A create adopts an existing image with the same name, and deleting it deletes that image,
1292        // which this deployment may never have built.
1293        false
1294    }
1295
1296    fn get_dependencies(&self) -> Vec<ResourceRef> {
1297        Vec::new()
1298    }
1299
1300    fn validate_update(&self, new_config: &dyn ResourceDefinition) -> Result<()> {
1301        let new_sandbox = new_config
1302            .as_any()
1303            .downcast_ref::<Sandbox>()
1304            .ok_or_else(|| {
1305                AlienError::new(ErrorData::UnexpectedResourceType {
1306                    resource_id: self.id.clone(),
1307                    expected: Self::RESOURCE_TYPE,
1308                    actual: new_config.get_resource_type(),
1309                })
1310            })?;
1311
1312        if self.id != new_sandbox.id {
1313            return Err(AlienError::new(ErrorData::InvalidResourceUpdate {
1314                resource_id: self.id.clone(),
1315                reason: "the 'id' field is immutable".to_string(),
1316            }));
1317        }
1318
1319        Ok(())
1320    }
1321
1322    fn as_any(&self) -> &dyn Any {
1323        self
1324    }
1325
1326    fn as_any_mut(&mut self) -> &mut dyn Any {
1327        self
1328    }
1329
1330    fn box_clone(&self) -> Box<dyn ResourceDefinition> {
1331        Box::new(self.clone())
1332    }
1333
1334    fn resource_eq(&self, other: &dyn ResourceDefinition) -> bool {
1335        other.as_any().downcast_ref::<Sandbox>() == Some(self)
1336    }
1337
1338    fn to_json_value(&self) -> serde_json::Result<serde_json::Value> {
1339        serde_json::to_value(self)
1340    }
1341}
1342
1343/// What an Azure sandbox starts from.
1344#[derive(Debug, Clone, Copy, PartialEq, Eq)]
1345pub enum AzureSandboxImage<'a> {
1346    /// A public catalog disk image, such as `ubuntu`, which the data plane names directly.
1347    Catalog(&'a str),
1348    /// A registry image, which the controller builds into a disk image in the sandbox's group
1349    /// before any sandbox can start from it.
1350    Registry(&'a str),
1351}
1352
1353impl<'a> AzureSandboxImage<'a> {
1354    /// The declared value, trimmed.
1355    pub fn as_str(&self) -> &'a str {
1356        match self {
1357            Self::Catalog(value) | Self::Registry(value) => value,
1358        }
1359    }
1360}
1361
1362/// Label key the controller writes on every disk image it builds, and the provider finds it by.
1363pub const AZURE_DISK_IMAGE_LABEL: &str = "alienImage";
1364
1365/// Classifies a declared `code.image` for Azure, or `None` when it is neither kind. A bare
1366/// `[A-Za-z0-9._-]+` is checked first and always a catalog name; anything else must carry `/`,
1367/// `:` or `@` and parse as an OCI reference.
1368pub fn classify_azure_sandbox_image(image: &str) -> Option<AzureSandboxImage<'_>> {
1369    let image = image.trim();
1370    if image.is_empty() {
1371        return None;
1372    }
1373    if image
1374        .chars()
1375        .all(|c| c.is_ascii_alphanumeric() || matches!(c, '.' | '_' | '-'))
1376    {
1377        return Some(AzureSandboxImage::Catalog(image));
1378    }
1379    (image.contains(|c| matches!(c, '/' | ':' | '@')) && OCI_REFERENCE.is_match(image))
1380        .then_some(AzureSandboxImage::Registry(image))
1381}
1382
1383/// The label value naming the disk image built from `reference`: a digest, since a label value
1384/// may not carry a reference's `/`, `:` and `@`.
1385pub fn azure_disk_image_label(reference: &str) -> String {
1386    let digest = format!("{:x}", Sha256::digest(reference.trim().as_bytes()));
1387    digest[..32].to_string()
1388}
1389
1390/// The distribution reference grammar: `[host[:port]/]path[:tag][@digest]`.
1391static OCI_REFERENCE: std::sync::LazyLock<regex::Regex> = std::sync::LazyLock::new(|| {
1392    let domain_component = r"(?:[a-zA-Z0-9]|[a-zA-Z0-9][a-zA-Z0-9-]*[a-zA-Z0-9])";
1393    let domain = format!(r"{domain_component}(?:\.{domain_component})*(?::[0-9]+)?");
1394    let path_component = r"[a-z0-9]+(?:(?:[._]|__|-+)[a-z0-9]+)*";
1395    let name = format!(r"(?:{domain}/)?{path_component}(?:/{path_component})*");
1396    let tag = r"[A-Za-z0-9_][A-Za-z0-9_.-]{0,127}";
1397    let digest = r"[A-Za-z][A-Za-z0-9]*(?:[-_+.][A-Za-z][A-Za-z0-9]*)*:[0-9a-fA-F]{32,}";
1398    regex::Regex::new(&format!(r"^{name}(?::{tag})?(?:@{digest})?$"))
1399        .expect("the OCI reference grammar compiles")
1400});
1401
1402/// The one token a sandbox bundle URI may carry, replaced with the deploying region.
1403///
1404/// AWS builds a MicroVM image only from a bucket in the image's own region, so a vendor
1405/// publishing to every supported region needs one stored URI that resolves per region.
1406pub const BUNDLE_REGION_TOKEN: &str = "{region}";
1407
1408/// A bundle URI split around its region token, or carried whole when it has none.
1409#[derive(Debug, Clone, Copy, PartialEq, Eq)]
1410pub enum BundleUri<'a> {
1411    /// No token: emitted exactly as it is today.
1412    Literal(&'a str),
1413    /// The text either side of the token, for an emitter to rejoin around its own region
1414    /// expression.
1415    Regional { before: &'a str, after: &'a str },
1416}
1417
1418/// The prefix a rebuild's new key still sits under: everything above the file name and the
1419/// version segment beneath it. `None` when nothing sits there, meaning no prefix can be granted
1420/// without also granting objects a rebuild never reads. Shared so both emitters agree on it.
1421pub fn stable_bundle_key_prefix(key: &str) -> Option<&str> {
1422    let (above_file, _) = key.rsplit_once('/')?;
1423    let (above_version, _) = above_file.rsplit_once('/')?;
1424    Some(above_version)
1425}
1426
1427/// Reads a sandbox bundle URI, refusing anything an image build would only reject later.
1428///
1429/// The token is accepted in the bucket alone. A key-position token would name an object that does
1430/// not exist, and any other brace is a typo that would otherwise reach S3 verbatim and fail ~160s
1431/// into the build — which is the failure this whole check exists to move to plan time.
1432pub fn parse_bundle_uri(uri: &str) -> std::result::Result<BundleUri<'_>, String> {
1433    let path = uri
1434        .strip_prefix("s3://")
1435        .ok_or_else(|| format!("'{uri}' is not an s3:// URI"))?;
1436    let (bucket, key) = path
1437        .split_once('/')
1438        .ok_or_else(|| format!("'{uri}' names a bucket with no object key"))?;
1439
1440    // Both emitters interpolate this path into the build role's resource ARN, where `*` and `?`
1441    // are IAM wildcards rather than literal characters. S3 accepts them in a key, so a bundle
1442    // published under one would silently widen the grant past the bundle it names.
1443    if path.contains('*') || path.contains('?') {
1444        return Err(format!(
1445            "'{uri}' carries an IAM wildcard; the bundle's path is interpolated into the build \
1446             role's grant, so '*' and '?' would widen it past the bundle"
1447        ));
1448    }
1449
1450    if key.contains('{') || key.contains('}') {
1451        return Err(format!(
1452            "'{uri}' places a token in the object key; {BUNDLE_REGION_TOKEN} is accepted in the \
1453             bucket name alone"
1454        ));
1455    }
1456
1457    let Some((before, after)) = bucket.split_once(BUNDLE_REGION_TOKEN) else {
1458        if bucket.contains('{') || bucket.contains('}') {
1459            return Err(format!(
1460                "'{uri}' carries a token this build does not know; {BUNDLE_REGION_TOKEN} is the \
1461                 only one"
1462            ));
1463        }
1464        return Ok(BundleUri::Literal(uri));
1465    };
1466
1467    if after.contains(BUNDLE_REGION_TOKEN) {
1468        return Err(format!("'{uri}' repeats {BUNDLE_REGION_TOKEN}"));
1469    }
1470    if before.contains('{') || before.contains('}') || after.contains('{') || after.contains('}') {
1471        return Err(format!(
1472            "'{uri}' carries a token this build does not know; {BUNDLE_REGION_TOKEN} is the only one"
1473        ));
1474    }
1475
1476    Ok(BundleUri::Regional {
1477        before: &uri[.."s3://".len() + before.len()],
1478        after: &uri["s3://".len() + before.len() + BUNDLE_REGION_TOKEN.len()..],
1479    })
1480}
1481
1482/// Where a private ECR image's region comes from.
1483#[derive(Debug, Clone, Copy, PartialEq, Eq)]
1484pub enum EcrImageRegion<'a> {
1485    /// A region named in the host.
1486    Literal(&'a str),
1487    /// [`BUNDLE_REGION_TOKEN`] in the host: the region the deployment renders.
1488    Deployment,
1489}
1490
1491/// The ECR repository a private image reference is pulled from.
1492#[derive(Debug, Clone, Copy, PartialEq, Eq)]
1493pub struct EcrImageRepository<'a> {
1494    pub account_id: &'a str,
1495    pub region: EcrImageRegion<'a>,
1496    /// The repository name, which may carry `/`; never a tag or digest.
1497    pub repository: &'a str,
1498}
1499
1500impl EcrImageRepository<'_> {
1501    /// The repository's ARN, with `region` standing in for a region the host leaves to the
1502    /// deployment. The partition is the deployment's: a GovCloud host ends `.amazonaws.com` too.
1503    pub fn arn(&self, partition: &str, region: &str) -> String {
1504        let region = match self.region {
1505            EcrImageRegion::Literal(region) => region,
1506            EcrImageRegion::Deployment => region,
1507        };
1508        format!(
1509            "arn:{partition}:ecr:{region}:{}:repository/{}",
1510            self.account_id, self.repository
1511        )
1512    }
1513}
1514
1515/// Reads `privateBaseImage` as the one repository a build role may pull from.
1516///
1517/// The name is interpolated into an IAM ARN, so it is held to ECR's own repository grammar: that
1518/// refuses `*` and `?`, which would widen the grant, and `$` and braces, which a CloudFormation
1519/// `Sub` or a Terraform template would read as an expression.
1520pub fn parse_ecr_image_repository(
1521    image: &str,
1522) -> std::result::Result<EcrImageRepository<'_>, String> {
1523    let refuse = |reason: &str| format!("privateBaseImage '{image}' {reason}");
1524    let (host, path) = image
1525        .split_once('/')
1526        .ok_or_else(|| refuse("names no repository"))?;
1527    let (account_id, rest) = host.split_once(".dkr.ecr.").ok_or_else(|| {
1528        refuse("is not served by a private ECR registry (<account>.dkr.ecr.<region>.amazonaws.com)")
1529    })?;
1530    // `.com.cn` first: a China host ends with the shorter suffix too.
1531    let region = rest
1532        .strip_suffix(".amazonaws.com.cn")
1533        .or_else(|| rest.strip_suffix(".amazonaws.com"))
1534        .ok_or_else(|| refuse("is not served by a private ECR registry (<account>.dkr.ecr.<region>.amazonaws.com)"))?;
1535    if account_id.len() != 12 || !account_id.bytes().all(|b| b.is_ascii_digit()) {
1536        return Err(refuse("names no 12-digit account in its registry host"));
1537    }
1538    let region = if region == BUNDLE_REGION_TOKEN {
1539        EcrImageRegion::Deployment
1540    } else if !region.is_empty()
1541        && region
1542            .bytes()
1543            .all(|b| b.is_ascii_lowercase() || b.is_ascii_digit() || b == b'-')
1544    {
1545        EcrImageRegion::Literal(region)
1546    } else {
1547        return Err(refuse(&format!(
1548            "names no region in its registry host; give one or {BUNDLE_REGION_TOKEN}"
1549        )));
1550    };
1551
1552    let repository = match path.split_once('@') {
1553        Some((repository, _digest)) => repository,
1554        None => match path.rsplit_once('/') {
1555            Some((parent, last)) => match last.split_once(':') {
1556                Some((name, _tag)) => &path[..parent.len() + 1 + name.len()],
1557                None => path,
1558            },
1559            None => path.split_once(':').map_or(path, |(name, _tag)| name),
1560        },
1561    };
1562    let valid_segment = |segment: &str| {
1563        !segment.is_empty()
1564            && segment.bytes().all(|b| {
1565                b.is_ascii_lowercase() || b.is_ascii_digit() || matches!(b, b'.' | b'_' | b'-')
1566            })
1567    };
1568    if !repository.split('/').all(valid_segment) {
1569        return Err(refuse(
1570            "names a repository outside ECR's grammar (lowercase letters, digits, '.', '_', '-', and '/' between them)",
1571        ));
1572    }
1573    Ok(EcrImageRepository {
1574        account_id,
1575        region,
1576        repository,
1577    })
1578}
1579
1580#[cfg(test)]
1581mod tests {
1582    use super::*;
1583
1584    #[test]
1585    fn private_database_setup_accepts_the_default_network() {
1586        for lifecycle in [
1587            crate::ResourceLifecycle::Frozen,
1588            crate::ResourceLifecycle::Live,
1589        ] {
1590            let stack = crate::Stack::new("database".to_string())
1591                .add(
1592                    crate::Postgres::new("metadata".to_string()).build(),
1593                    lifecycle,
1594                )
1595                .build();
1596            assert!(!restricts_network_mode(&stack, false));
1597            assert!(!restricts_network_mode(&stack, true));
1598        }
1599        assert!(!restricts_network_mode(
1600            &crate::Stack::new("empty".to_string()).build(),
1601            false,
1602        ));
1603    }
1604
1605    /// The vectors' `repositoryKey`: registry host and repository, never the tag or digest, so a
1606    /// new tag keeps the key. A reference the parser refuses stays whole.
1607    fn repository_key(image: &str) -> String {
1608        match parse_ecr_image_repository(image) {
1609            Ok(parsed) => {
1610                let host = image.split_once('/').map_or(image, |(host, _)| host);
1611                format!("{host}/{}", parsed.repository)
1612            }
1613            Err(_) => image.to_string(),
1614        }
1615    }
1616
1617    #[test]
1618    fn a_private_base_image_names_one_repository() {
1619        let vectors: serde_json::Value = serde_json::from_str(include_str!(
1620            "../../tests/fixtures/ecr-image-repository-parity.json"
1621        ))
1622        .expect("the ECR repository vectors must be JSON");
1623        let field = |case: &serde_json::Value, name: &str| {
1624            case[name]
1625                .as_str()
1626                .unwrap_or_else(|| panic!("vector must carry {name}: {case}"))
1627                .to_string()
1628        };
1629
1630        for case in vectors["accepted"].as_array().expect("accepted vectors") {
1631            let image = field(case, "image");
1632            let region = field(case, "region");
1633            let region = if region == BUNDLE_REGION_TOKEN {
1634                EcrImageRegion::Deployment
1635            } else {
1636                EcrImageRegion::Literal(&region)
1637            };
1638            assert_eq!(
1639                parse_ecr_image_repository(&image),
1640                Ok(EcrImageRepository {
1641                    account_id: &field(case, "accountId"),
1642                    region,
1643                    repository: &field(case, "repository"),
1644                }),
1645                "{image}"
1646            );
1647            assert_eq!(
1648                repository_key(&image),
1649                field(case, "repositoryKey"),
1650                "{image}"
1651            );
1652        }
1653
1654        for case in vectors["refused"].as_array().expect("refused vectors") {
1655            let image = field(case, "image");
1656            assert!(
1657                parse_ecr_image_repository(&image).is_err(),
1658                "{image} must be refused"
1659            );
1660            assert_eq!(
1661                repository_key(&image),
1662                field(case, "repositoryKey"),
1663                "{image}"
1664            );
1665        }
1666    }
1667
1668    #[test]
1669    fn a_repository_arn_takes_the_deployment_region_only_where_the_host_leaves_it() {
1670        let regional =
1671            parse_ecr_image_repository("123456789012.dkr.ecr.{region}.amazonaws.com/team/base:1")
1672                .expect("parses");
1673        let pinned =
1674            parse_ecr_image_repository("123456789012.dkr.ecr.eu-west-1.amazonaws.com/team/base:1")
1675                .expect("parses");
1676        assert_eq!(
1677            regional.arn("aws-us-gov", "us-gov-west-1"),
1678            "arn:aws-us-gov:ecr:us-gov-west-1:123456789012:repository/team/base"
1679        );
1680        assert_eq!(
1681            pinned.arn("aws", "us-east-1"),
1682            "arn:aws:ecr:eu-west-1:123456789012:repository/team/base"
1683        );
1684    }
1685
1686    /// A wildcard reaching the grant would widen it past the bundle, and it widens the Frozen
1687    /// object grant as readily as the Live prefix — both interpolate the path into the ARN.
1688    #[test]
1689    fn a_uri_carrying_an_iam_wildcard_is_refused() {
1690        for uri in [
1691            "s3://acme/team-*/v1/bundle.zip",
1692            "s3://acme/sandbox-bundle/f00d/bundle?.zip",
1693            "s3://acme-*/sandbox-bundle/f00d/bundle.zip",
1694        ] {
1695            let error = parse_bundle_uri(uri).expect_err("a wildcard must be refused");
1696            assert!(error.contains("IAM wildcard"), "for {uri}: {error}");
1697        }
1698
1699        parse_bundle_uri("s3://acme/sandbox-bundle/f00d/bundle.zip")
1700            .expect("an ordinary key still parses");
1701    }
1702
1703    /// The rule both package formats grant by, pinned here rather than in either. The
1704    /// near-misses the cases separate: the key's first segment grants objects a rebuild never
1705    /// reads, and the object's own directory pins the version segment that moves.
1706    #[test]
1707    fn a_grantable_prefix_stops_above_the_segment_that_moves() {
1708        assert_eq!(
1709            stable_bundle_key_prefix("sandbox-bundle/f00dcafe/bundle.zip"),
1710            Some("sandbox-bundle")
1711        );
1712        assert_eq!(
1713            stable_bundle_key_prefix("artifacts/team-a/sandbox/f00dcafe/bundle.zip"),
1714            Some("artifacts/team-a/sandbox"),
1715            "a deeper key narrows the prefix, it never widens to the first segment"
1716        );
1717
1718        // Nothing sits above the version segment, so no prefix a moved bundle stays inside
1719        // exists. Emitting the object grant instead installs a role that denies the next rebuild.
1720        assert_eq!(stable_bundle_key_prefix("agents/bundle.zip"), None);
1721        assert_eq!(stable_bundle_key_prefix("bundle.zip"), None);
1722    }
1723
1724    fn sandbox_with(egress: SandboxEgress, preview_ports: Vec<u16>) -> Sandbox {
1725        Sandbox::new("agent-sbx".to_string())
1726            .code(SandboxCode::Image {
1727                image: "ubuntu".to_string(),
1728                bundled_from: None,
1729            })
1730            .limits(SandboxLimits {
1731                cpu: "1".to_string(),
1732                memory: "2Gi".to_string(),
1733                disk: "20Gi".to_string(),
1734                max_processes: None,
1735            })
1736            .egress(egress)
1737            .lifecycle(SandboxLifecyclePolicy {
1738                max_lifetime_seconds: None,
1739                idle_pause_seconds: None,
1740            })
1741            .preview_ports(preview_ports)
1742            .build()
1743    }
1744
1745    /// A URI with no token must come back whole, because every bundle configured today has none
1746    /// and emitting one differently would change every existing customer's template.
1747    #[test]
1748    fn a_uri_without_a_token_is_carried_whole() {
1749        assert_eq!(
1750            parse_bundle_uri("s3://acme-artifacts-us-east-2/agents/bundle.zip"),
1751            Ok(BundleUri::Literal(
1752                "s3://acme-artifacts-us-east-2/agents/bundle.zip"
1753            ))
1754        );
1755    }
1756
1757    /// The split has to rejoin to the original with the region in place, or an emitter builds a
1758    /// URI that is subtly not the one the vendor configured.
1759    #[test]
1760    fn a_regional_uri_splits_either_side_of_the_token() {
1761        let BundleUri::Regional { before, after } =
1762            parse_bundle_uri("s3://acme-artifacts-{region}/agents/bundle.zip")
1763                .expect("the token is accepted in the bucket")
1764        else {
1765            panic!("a bucket-position token must split");
1766        };
1767
1768        assert_eq!(before, "s3://acme-artifacts-");
1769        assert_eq!(after, "/agents/bundle.zip");
1770        assert_eq!(
1771            format!("{before}us-east-2{after}"),
1772            "s3://acme-artifacts-us-east-2/agents/bundle.zip",
1773            "the halves must rejoin to the URI the vendor meant"
1774        );
1775    }
1776
1777    /// Each of these reaches S3 verbatim and dies ~160s into an image build if it is not refused
1778    /// here, which is the whole reason this runs at plan time.
1779    #[test]
1780    fn a_token_this_build_cannot_resolve_is_refused() {
1781        for uri in [
1782            "s3://acme-artifacts-{regio}/bundle.zip",
1783            "s3://acme-artifacts/{region}/bundle.zip",
1784            "s3://acme-artifacts-{region}-{region}/bundle.zip",
1785            "s3://acme-artifacts/bundle-{version}.zip",
1786            "s3://acme}-artifacts-{region}/bundle.zip",
1787            "s3://acme{-artifacts-{region}/bundle.zip",
1788        ] {
1789            assert!(
1790                parse_bundle_uri(uri).is_err(),
1791                "'{uri}' must be refused before it can reach an image build"
1792            );
1793        }
1794    }
1795
1796    #[test]
1797    fn resource_type_is_stable() {
1798        assert_eq!(Sandbox::RESOURCE_TYPE.as_ref(), "sandbox");
1799    }
1800
1801    #[test]
1802    fn capability_sets_are_per_platform() {
1803        let gcp = SandboxCapabilities::for_platform(Platform::Gcp).expect("gcp is supported");
1804        assert!(
1805            gcp.reconnect,
1806            "generation from the container boot id makes a sandbox reachable across processes"
1807        );
1808        assert!(!gcp.preview);
1809        assert!(gcp.enforced_limits);
1810
1811        let azure = SandboxCapabilities::for_platform(Platform::Azure).expect("azure is supported");
1812        assert!(azure.files, "every backend moves files");
1813        assert!(gcp.files);
1814        // Azure is the only backend whose egress policy matches on host pattern, and the only
1815        // one where `deny` and a hostname list are the same object.
1816        assert!(azure.domain_egress_rules);
1817        assert!(azure.egress_deny);
1818        // The data plane honours a continuous cpu/memory/disk surface and refuses anything
1819        // outside `n×250m`, with the rule in the message.
1820        assert!(azure.enforced_limits);
1821        assert!(azure.pause_resume);
1822        // Both stay false for reasons that are not "unbuilt": a snapshot id has nothing to
1823        // consume it on any backend, and an Azure port's auth is anonymous or a human allowlist,
1824        // neither of which is a port-scoped credential.
1825        assert!(!azure.snapshot);
1826        assert!(!azure.preview);
1827
1828        let aws = SandboxCapabilities::for_platform(Platform::Aws).expect("aws is supported");
1829        assert!(!aws.snapshot, "AWS has no user-callable sandbox snapshot");
1830        assert!(aws.pause_resume);
1831
1832        let k8s =
1833            SandboxCapabilities::for_platform(Platform::Kubernetes).expect("k8s is supported");
1834        assert!(
1835            !k8s.preview,
1836            "the sandbox-scoped ingress gateway does not exist yet"
1837        );
1838    }
1839
1840    /// Whether the process supervising a command is a separate identity from the command.
1841    ///
1842    /// Values are measured, not inferred. AWS: the agent runs as uid 0 with
1843    /// `CapEff: 00000000a80425fb` and `setuid`s the command to uid 60000, so the two differ.
1844    /// Kubernetes: the sandbox pod pins `run_as_user: 65534` on both pod and container with
1845    /// `capabilities.drop: [ALL]` and `allow_privilege_escalation: false`, so no uid split is
1846    /// possible (`kubernetes_spec.rs`). Local: `docker exec` runs as the workload uid while the
1847    /// manager supervises from the host. Azure and Agent Platform have no in-sandbox supervisor.
1848    #[test]
1849    fn supervisor_isolation_is_per_platform() {
1850        let value = |platform| {
1851            SandboxCapabilities::for_platform(platform)
1852                .expect("supported")
1853                .supervisor_isolation
1854        };
1855
1856        assert!(
1857            value(Platform::Aws),
1858            "root agent setuids the command to 60000"
1859        );
1860        assert!(
1861            value(Platform::Local),
1862            "the supervisor is on the host, outside the container"
1863        );
1864        assert!(
1865            !value(Platform::Kubernetes),
1866            "a single pinned uid cannot be split"
1867        );
1868        assert!(!value(Platform::Azure), "no Alien process runs the command");
1869        assert!(
1870            !value(Platform::Gcp),
1871            "no separate supervisor identity runs the command"
1872        );
1873    }
1874
1875    /// The point of the field: AWS and GCP report the *same* `supervisor_pid_namespace` (neither
1876    /// has `CAP_SYS_ADMIN`), so that axis alone reads them as equivalent. They are not — AWS
1877    /// separates the command's identity from the supervisor's and Agent Platform does not.
1878    #[test]
1879    fn supervisor_isolation_separates_aws_from_a_subprocess_backend() {
1880        let aws = SandboxCapabilities::for_platform(Platform::Aws).expect("aws is supported");
1881        let gcp = SandboxCapabilities::for_platform(Platform::Gcp).expect("gcp is supported");
1882
1883        assert_eq!(
1884            aws.supervisor_pid_namespace, gcp.supervisor_pid_namespace,
1885            "the older axis cannot tell them apart"
1886        );
1887        assert!(
1888            aws.supervisor_isolation,
1889            "AWS setuids the command off the supervisor"
1890        );
1891        assert!(
1892            !gcp.supervisor_isolation,
1893            "the command runs under no separate supervisor identity"
1894        );
1895    }
1896
1897    /// The Agent Platform row, each value against the behaviour it was measured from. `reconnect`
1898    /// is the tripwire: it is `true` only because `generation` is derived from the container boot
1899    /// id read through the agent's health op, so a caller detects a replaced container instead of
1900    /// reconnecting to a blank one. It is also the body of the `Platform::Gcp` arm, asserted below.
1901    #[test]
1902    fn gcp_agent_platform_row_matches_measured_backend() {
1903        let row = SandboxCapabilities::gcp_agent_platform();
1904
1905        assert!(row.files, "agent file ops move over the sandbox envelope");
1906        assert!(
1907            row.reconnect,
1908            "generation is derived from the container boot id, so a sandbox is reachable across \
1909             processes"
1910        );
1911        assert!(
1912            !row.preview,
1913            "the only ingress is :execute; no port-scoped capability"
1914        );
1915        assert!(
1916            !row.pause_resume,
1917            ":resume can return a fresh container, so a pause keeps no state"
1918        );
1919        assert!(
1920            !row.snapshot,
1921            "the create path never sends a snapshot, so none is reachable through the trait"
1922        );
1923        assert!(
1924            !row.domain_egress_rules,
1925            "VPC and DNS peering is not a hostname allowlist"
1926        );
1927        assert!(
1928            row.egress_deny,
1929            "a declared deny blocks both egress and DNS"
1930        );
1931        assert!(
1932            row.enforced_limits,
1933            "ceilings are enforced, by terminating the sandbox on breach"
1934        );
1935        assert!(!row.process_limit, "no process-count ceiling is observed");
1936        assert!(row.sandbox_lifetime, "ttl maps to a sandbox expireTime");
1937        assert!(!row.supervisor_pid_namespace, "no PID-namespace isolation");
1938        assert!(
1939            !row.supervisor_isolation,
1940            "the command is not run under a separate supervisor identity"
1941        );
1942
1943        // Agent Platform is the registered GCP backend, so the arm returns exactly this row.
1944        let live = SandboxCapabilities::for_platform(Platform::Gcp).expect("gcp is supported");
1945        assert_eq!(
1946            live, row,
1947            "the Platform::Gcp arm is the Agent Platform capability row"
1948        );
1949    }
1950
1951    #[test]
1952    fn platforms_without_a_backend_are_an_error_not_an_empty_set() {
1953        let error = SandboxCapabilities::for_platform(Platform::Machines)
1954            .expect_err("Machines has no sandbox backend");
1955        assert_eq!(error.code, "SANDBOX_PLATFORM_UNSUPPORTED");
1956    }
1957
1958    #[test]
1959    fn unsupported_capability_names_platform_and_capability() {
1960        let capabilities = SandboxCapabilities::for_platform(Platform::Gcp).expect("supported");
1961        let error = capabilities
1962            .require(SandboxCapability::Preview, Platform::Gcp)
1963            .expect_err("GCP has no preview");
1964
1965        assert_eq!(error.code, "SANDBOX_CAPABILITY_UNSUPPORTED");
1966        let rendered = error.to_string();
1967        assert!(
1968            rendered.contains("preview"),
1969            "names the capability: {rendered}"
1970        );
1971        assert!(rendered.contains("gcp"), "names the platform: {rendered}");
1972    }
1973
1974    /// Azure matches on hostname; AWS and Kubernetes match CIDRs, and Local and GCP have a
1975    /// switch rather than a filter. Accepting a hostname list on those four would leave a stack
1976    /// reading as restricted while the sandbox reaches the whole internet.
1977    #[test]
1978    fn a_hostname_allowlist_is_refused_everywhere_it_would_be_approximated() {
1979        let sandbox = sandbox_with(
1980            SandboxEgress::AllowDomains {
1981                domains: vec!["example.com".to_string()],
1982            },
1983            vec![],
1984        );
1985
1986        for platform in [
1987            Platform::Aws,
1988            Platform::Gcp,
1989            Platform::Kubernetes,
1990            Platform::Local,
1991        ] {
1992            let error = sandbox
1993                .validate_for_platform(platform)
1994                .expect_err("only Azure expresses a hostname allowlist");
1995            assert_eq!(
1996                error.code, "SANDBOX_CAPABILITY_UNSUPPORTED",
1997                "on {platform:?}"
1998            );
1999        }
2000
2001        assert!(
2002            SandboxCapabilities::for_platform(Platform::Azure)
2003                .expect("supported")
2004                .domain_egress_rules,
2005            "Azure's egress policy matches on host pattern"
2006        );
2007    }
2008
2009    /// `deny` is the declaration that carries a security promise, so a backend that cannot keep
2010    /// it has to refuse rather than accept it and run the code with open egress.
2011    #[test]
2012    fn a_denied_egress_is_refused_where_it_would_not_be_enforced() {
2013        let sandbox = sandbox_with(SandboxEgress::None, vec![]);
2014
2015        // GCP is asserted at the capability rather than through validation: this sandbox declares
2016        // ceilings GCP cannot enforce, so it is refused for a reason unrelated to egress.
2017        assert!(
2018            SandboxCapabilities::for_platform(Platform::Gcp)
2019                .expect("supported")
2020                .egress_deny
2021        );
2022
2023        for platform in [Platform::Aws, Platform::Kubernetes, Platform::Local] {
2024            sandbox
2025                .validate_for_platform(platform)
2026                .expect("deny is enforced here");
2027        }
2028
2029        // Declares no ceilings, which Azure refuses for its own reason, so this isolates egress.
2030        let egress_only = Sandbox::new("sbx".to_string())
2031            .code(SandboxCode::Image {
2032                image: "alpine".to_string(),
2033                bundled_from: None,
2034            })
2035            .egress(SandboxEgress::None)
2036            .lifecycle(SandboxLifecyclePolicy {
2037                max_lifetime_seconds: None,
2038                idle_pause_seconds: None,
2039            })
2040            .build();
2041
2042        egress_only
2043            .validate_for_platform(Platform::Azure)
2044            .expect("Azure creates the sandbox under a Deny policy with full inspection");
2045    }
2046
2047    /// A sandbox naming no ceilings is valid on every platform and still resolves to a concrete
2048    /// set. The rule Azure applies to ceilings that *are* declared is pinned separately, by
2049    /// `azure_sizes_follow_the_rule_the_data_plane_states`.
2050    #[test]
2051    fn a_sandbox_declaring_no_ceilings_takes_the_platforms_own() {
2052        let undeclared = Sandbox::new("sbx".to_string())
2053            .code(SandboxCode::Image {
2054                image: "alpine".to_string(),
2055                bundled_from: None,
2056            })
2057            .egress(SandboxEgress::None)
2058            .lifecycle(SandboxLifecyclePolicy {
2059                max_lifetime_seconds: None,
2060                idle_pause_seconds: None,
2061            })
2062            .build();
2063
2064        undeclared
2065            .validate_for_platform(Platform::Azure)
2066            .expect("a sandbox naming no ceilings takes the platform's own");
2067
2068        // A backend still gets a concrete set, so nothing downstream has to invent one.
2069        assert_eq!(undeclared.resolved_limits().cpu, "1");
2070    }
2071
2072    /// Pins Azure's own sizing rule: `250m`, `1500m` and `4000m` are valid, `32000m` and `333m`
2073    /// are refused. Checked at plan time so a bad value is a declaration to fix, not a package
2074    /// that renders without error and dies at the first sandbox.
2075    #[test]
2076    fn azure_sizes_follow_the_rule_the_data_plane_states() {
2077        let sized = |cpu: &str, memory: &str, disk: &str| {
2078            let mut sandbox = sandbox_with(SandboxEgress::None, vec![]);
2079            let limits = sandbox
2080                .limits
2081                .as_mut()
2082                .expect("the fixture declares limits");
2083            limits.cpu = cpu.to_string();
2084            limits.memory = memory.to_string();
2085            limits.disk = disk.to_string();
2086            sandbox.validate_for_platform(Platform::Azure)
2087        };
2088
2089        sized("250m", "512Mi", "5120Mi").expect("the smallest step the data plane accepts");
2090        sized("4000m", "8192Mi", "40960Mi").expect("cpu, memory and disk are all honoured");
2091        sized("16000m", "32Gi", "320Gi").expect("the top of the range");
2092
2093        // Same off-step case `azure_sandbox_limits` checks the multiple for, not just the range.
2094        let off_step = sized("333m", "512Mi", "5120Mi").expect_err("333m is not a step of 250m");
2095        assert_eq!(off_step.code, "SANDBOX_LIMIT_INVALID", "{off_step}");
2096        assert!(off_step.to_string().contains("cpu"), "{off_step}");
2097
2098        let too_big = sized("32000m", "64Gi", "640Gi").expect_err("32 cores is over the ceiling");
2099        assert_eq!(too_big.code, "SANDBOX_LIMIT_INVALID", "{too_big}");
2100
2101        // Both ceilings are derived from the cpu, so the same memory passes at one size and fails
2102        // at another - which is what makes them worth checking rather than bounding absolutely.
2103        sized("1000m", "2Gi", "20Gi").expect("2Gi is exactly one core's worth");
2104        let over_memory = sized("250m", "2Gi", "5120Mi").expect_err("2Gi needs a full core");
2105        assert_eq!(over_memory.code, "SANDBOX_LIMIT_INVALID", "{over_memory}");
2106        assert!(over_memory.to_string().contains("memory"), "{over_memory}");
2107
2108        let over_disk = sized("250m", "512Mi", "20Gi").expect_err("20Gi needs a full core");
2109        assert!(over_disk.to_string().contains("disk"), "{over_disk}");
2110    }
2111
2112    #[test]
2113    fn preview_ports_require_the_preview_capability() {
2114        let sandbox = sandbox_with(SandboxEgress::None, vec![8080]);
2115
2116        sandbox
2117            .validate_for_platform(Platform::Aws)
2118            .expect("AWS mints a port-scoped JWE");
2119
2120        let error = sandbox
2121            .validate_for_platform(Platform::Kubernetes)
2122            .expect_err("Kubernetes preview is deferred");
2123        assert_eq!(error.code, "SANDBOX_CAPABILITY_UNSUPPORTED");
2124    }
2125
2126    /// A grant nothing reads is the silent no-op the capability contract exists to prevent, and
2127    /// here it is worse than useless: the reader would take it for a base image that needs
2128    /// authenticating while the platform pulls `code.image` itself.
2129    #[test]
2130    fn a_private_base_image_is_refused_off_aws() {
2131        let mut sandbox = sandbox_with(SandboxEgress::Allow, vec![]);
2132        sandbox.code = SandboxCode::Image {
2133            image: "s3://acme-artifacts/agents/bundle.zip".to_string(),
2134            bundled_from: None,
2135        };
2136        sandbox.private_base_image =
2137            Some("123456789012.dkr.ecr.{region}.amazonaws.com/acme:tag".to_string());
2138
2139        sandbox
2140            .validate_for_platform(Platform::Aws)
2141            .expect("AWS builds its image from a bundle, so a base image sits behind code.image");
2142
2143        for platform in [
2144            Platform::Gcp,
2145            Platform::Azure,
2146            Platform::Kubernetes,
2147            Platform::Local,
2148        ] {
2149            let error = sandbox
2150                .validate_for_platform(platform)
2151                .expect_err("a backend that builds no image must refuse a base image for one");
2152            assert_eq!(error.code, "SANDBOX_CAPABILITY_UNSUPPORTED");
2153            assert!(
2154                error.to_string().contains("privateBaseImage"),
2155                "the refusal must name the field the user declared: {error}"
2156            );
2157        }
2158    }
2159
2160    #[test]
2161    fn gcp_accepts_a_sandbox_declaring_enforced_limits() {
2162        let sandbox = sandbox_with(SandboxEgress::Allow, vec![]);
2163        sandbox
2164            .validate_for_platform(Platform::Gcp)
2165            .expect("Agent Platform enforces declared ceilings, by terminating on breach");
2166    }
2167
2168    #[test]
2169    fn invalid_quantities_are_rejected_with_the_offending_field() {
2170        let mut sandbox = sandbox_with(SandboxEgress::None, vec![]);
2171        sandbox
2172            .limits
2173            .as_mut()
2174            .expect("the fixture declares limits")
2175            .memory = "2Gb".to_string();
2176
2177        let error = sandbox
2178            .validate_for_platform(Platform::Aws)
2179            .expect_err("Gb is not a valid suffix");
2180        assert_eq!(error.code, "SANDBOX_LIMIT_INVALID");
2181        assert!(error.to_string().contains("memory"));
2182
2183        sandbox
2184            .limits
2185            .as_mut()
2186            .expect("the fixture declares limits")
2187            .memory = "2Gi".to_string();
2188        sandbox
2189            .limits
2190            .as_mut()
2191            .expect("the fixture declares limits")
2192            .cpu = "0".to_string();
2193        let error = sandbox
2194            .validate_for_platform(Platform::Aws)
2195            .expect_err("zero cpu is not a ceiling");
2196        assert_eq!(error.code, "SANDBOX_LIMIT_INVALID");
2197    }
2198
2199    #[test]
2200    fn zero_max_processes_is_rejected() {
2201        let mut sandbox = sandbox_with(SandboxEgress::None, vec![]);
2202        sandbox
2203            .limits
2204            .as_mut()
2205            .expect("the fixture declares limits")
2206            .max_processes = Some(0);
2207
2208        let error = sandbox
2209            .validate_for_platform(Platform::Local)
2210            .expect_err("a sandbox must be able to run at least one process");
2211        assert_eq!(error.code, "SANDBOX_LIMIT_INVALID");
2212        assert!(error.to_string().contains("maxProcesses"));
2213    }
2214
2215    /// A process ceiling needs a container runtime. Kubernetes sets one per node rather than per
2216    /// pod, and neither MicroVMs nor Azure sandboxes expose one, so accepting the declaration
2217    /// anywhere else would mean carrying a bound nothing applies.
2218    #[test]
2219    fn a_process_ceiling_is_accepted_only_where_a_runtime_can_apply_it() {
2220        let mut sandbox = sandbox_with(SandboxEgress::None, vec![]);
2221        sandbox
2222            .limits
2223            .as_mut()
2224            .expect("the fixture declares limits")
2225            .max_processes = Some(256);
2226
2227        sandbox
2228            .validate_for_platform(Platform::Local)
2229            .expect("Docker takes a pids limit");
2230
2231        for platform in [Platform::Aws, Platform::Azure, Platform::Kubernetes] {
2232            let error = sandbox
2233                .validate_for_platform(platform)
2234                .expect_err("a process ceiling nothing applies must be refused");
2235            assert_eq!(error.code, "SANDBOX_CAPABILITY_UNSUPPORTED");
2236        }
2237    }
2238
2239    /// Lambda rejects a run above 28,800 seconds rather than clamping it, so a value beyond that
2240    /// would pass planning, render into the package, and fail at the first sandbox. Kubernetes
2241    /// takes the same field with a far higher ceiling, so 28,800 is AWS's alone. The lifetime is a
2242    /// lifecycle field, so a sandbox that declares no limits is held to it too.
2243    #[test]
2244    fn a_lifetime_aws_would_reject_is_refused_while_planning() {
2245        let with_limits = sandbox_with(SandboxEgress::None, vec![]);
2246        let mut without_limits = with_limits.clone();
2247        without_limits.limits = None;
2248
2249        for (declared, mut sandbox) in [("with", with_limits), ("without", without_limits)] {
2250            for seconds in [28_801, 30_000, 100_000] {
2251                sandbox.lifecycle.max_lifetime_seconds = Some(seconds);
2252                let error = sandbox
2253                    .validate_for_platform(Platform::Aws)
2254                    .expect_err("a lifetime above what AWS runs is refused");
2255                assert_eq!(
2256                    error.code, "SANDBOX_LIMIT_INVALID",
2257                    "{seconds}s {declared} limits"
2258                );
2259
2260                // Kubernetes' ceiling is far higher, so the same declaration is fine there.
2261                sandbox
2262                    .validate_for_platform(Platform::Kubernetes)
2263                    .expect("Kubernetes takes an activeDeadlineSeconds above 28,800");
2264            }
2265
2266            sandbox.lifecycle.max_lifetime_seconds = Some(28_800);
2267            sandbox
2268                .validate_for_platform(Platform::Aws)
2269                .expect("the ceiling itself is allowed");
2270        }
2271    }
2272
2273    /// A zero lifetime is a sandbox that may not run at all. The Kubernetes API server refuses an
2274    /// `activeDeadlineSeconds` below 1 when the pod is created, so planning refuses it on every
2275    /// platform, with or without limits declared.
2276    #[test]
2277    fn a_zero_lifetime_is_refused_while_planning() {
2278        let with_limits = sandbox_with(SandboxEgress::None, vec![]);
2279        let mut without_limits = with_limits.clone();
2280        without_limits.limits = None;
2281
2282        for (declared, mut sandbox) in [("with", with_limits), ("without", without_limits)] {
2283            sandbox.lifecycle.max_lifetime_seconds = Some(0);
2284            for platform in [Platform::Aws, Platform::Kubernetes, Platform::Gcp] {
2285                let error = sandbox
2286                    .validate_for_platform(platform)
2287                    .expect_err("a zero lifetime is refused");
2288                assert_eq!(
2289                    error.code, "SANDBOX_LIMIT_INVALID",
2290                    "{platform} {declared} limits"
2291                );
2292                assert!(
2293                    error.to_string().contains("maxLifetimeSeconds"),
2294                    "{platform} {declared} limits"
2295                );
2296            }
2297
2298            // Where the field is unsupported, that is the answer, not the value.
2299            for platform in [Platform::Azure, Platform::Local] {
2300                let error = sandbox
2301                    .validate_for_platform(platform)
2302                    .expect_err("a lifetime is refused where nothing applies it");
2303                assert_eq!(
2304                    error.code, "SANDBOX_CAPABILITY_UNSUPPORTED",
2305                    "{platform} {declared} limits"
2306                );
2307            }
2308
2309            sandbox.lifecycle.max_lifetime_seconds = Some(1);
2310            sandbox
2311                .validate_for_platform(Platform::Kubernetes)
2312                .expect("one second is the API server's floor");
2313        }
2314    }
2315
2316    /// The API server caps `activeDeadlineSeconds` at the largest i32 although the field is an
2317    /// int64, and `maxLifetimeSeconds` is a u32, so without this check the top half of its
2318    /// range would plan fine and be refused when the pod is created.
2319    #[test]
2320    fn a_lifetime_kubernetes_would_reject_is_refused_while_planning() {
2321        let with_limits = sandbox_with(SandboxEgress::None, vec![]);
2322        let mut without_limits = with_limits.clone();
2323        without_limits.limits = None;
2324
2325        for (declared, mut sandbox) in [("with", with_limits), ("without", without_limits)] {
2326            for seconds in [2_147_483_648, u32::MAX] {
2327                sandbox.lifecycle.max_lifetime_seconds = Some(seconds);
2328                let error = sandbox
2329                    .validate_for_platform(Platform::Kubernetes)
2330                    .expect_err("a lifetime above what the API server takes is refused");
2331                assert_eq!(
2332                    error.code, "SANDBOX_LIMIT_INVALID",
2333                    "{seconds}s {declared} limits"
2334                );
2335                assert!(
2336                    error.to_string().contains("maxLifetimeSeconds"),
2337                    "{seconds}s {declared} limits"
2338                );
2339            }
2340
2341            sandbox.lifecycle.max_lifetime_seconds = Some(2_147_483_647);
2342            sandbox
2343                .validate_for_platform(Platform::Kubernetes)
2344                .expect("the ceiling itself is allowed");
2345        }
2346    }
2347
2348    /// An image Azure can take neither as a catalog name nor as a registry image is refused while
2349    /// planning, not at the first sandbox.
2350    #[test]
2351    fn an_image_azure_cannot_pull_is_refused_while_planning() {
2352        let mut sandbox = sandbox_with(SandboxEgress::None, vec![]);
2353        // Azure enforces no declared ceiling, so a sandbox carrying limits is refused before the
2354        // image is ever read.
2355        sandbox.limits = None;
2356
2357        // A digest too short to be one, blanks, a space and a query string.
2358        for image in ["ubuntu@sha256:abc", "", "   ", "ubuntu latest", "ubuntu?x"] {
2359            sandbox.code = SandboxCode::Image {
2360                image: image.to_string(),
2361                bundled_from: None,
2362            };
2363            let error = sandbox
2364                .validate_for_platform(Platform::Azure)
2365                .expect_err("an image Azure has nowhere to put is refused");
2366            assert_eq!(error.code, "SANDBOX_LIMIT_INVALID", "image '{image}'");
2367        }
2368
2369        for image in ["ubuntu", "ubuntu-22.04", "debian_slim"] {
2370            sandbox.code = SandboxCode::Image {
2371                image: image.to_string(),
2372                bundled_from: None,
2373            };
2374            sandbox
2375                .validate_for_platform(Platform::Azure)
2376                .unwrap_or_else(|error| panic!("'{image}' is a catalog name: {error}"));
2377        }
2378
2379        for image in [
2380            "ubuntu:24.04",
2381            "ghcr.io/myorg/sandbox:latest",
2382            "docker.io/library/python:3.14-slim",
2383            "localhost:5000/team/agent@sha256:51dafde81dbdb6ebde285137a295cf18a47ca95234fe388a343719cb97305b3d",
2384        ] {
2385            sandbox.code = SandboxCode::Image {
2386                image: image.to_string(),
2387                bundled_from: None,
2388            };
2389            sandbox
2390                .validate_for_platform(Platform::Azure)
2391                .unwrap_or_else(|error| panic!("'{image}' is a registry image: {error}"));
2392        }
2393
2394        // Surrounding space is trimmed rather than carried into the create body.
2395        sandbox.code = SandboxCode::Image {
2396            image: " ubuntu ".to_string(),
2397            bundled_from: None,
2398        };
2399        assert_eq!(
2400            sandbox
2401                .azure_image()
2402                .expect("a padded name is still a name"),
2403            AzureSandboxImage::Catalog("ubuntu")
2404        );
2405    }
2406
2407    /// Every value the catalog allowlist `[A-Za-z0-9._-]+` accepts stays a catalog name, so a
2408    /// declaration that planned under it keeps its meaning. Exhaustive to three characters, then a
2409    /// fixed-seed sample of longer ones, some with surrounding space.
2410    #[test]
2411    fn every_value_the_catalog_allowlist_accepted_is_still_a_catalog_name() {
2412        const ALPHABET: &[u8] =
2413            b"abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789._-";
2414
2415        let assert_catalog = |value: &str| {
2416            assert_eq!(
2417                classify_azure_sandbox_image(value),
2418                Some(AzureSandboxImage::Catalog(value.trim())),
2419                "'{value}'"
2420            );
2421        };
2422
2423        for a in ALPHABET {
2424            assert_catalog(&String::from_utf8(vec![*a]).unwrap());
2425            for b in ALPHABET {
2426                assert_catalog(&String::from_utf8(vec![*a, *b]).unwrap());
2427                for c in ALPHABET {
2428                    assert_catalog(&String::from_utf8(vec![*a, *b, *c]).unwrap());
2429                }
2430            }
2431        }
2432
2433        let mut state: u64 = 0x9E37_79B9_7F4A_7C15;
2434        let mut next = || {
2435            state ^= state << 13;
2436            state ^= state >> 7;
2437            state ^= state << 17;
2438            state
2439        };
2440        for _ in 0..20_000 {
2441            let len = 4 + (next() % 60) as usize;
2442            let mut value: String = (0..len)
2443                .map(|_| ALPHABET[(next() % ALPHABET.len() as u64) as usize] as char)
2444                .collect();
2445            if next() % 4 == 0 {
2446                value = format!("  {value}\t");
2447            }
2448            assert_catalog(&value);
2449        }
2450    }
2451
2452    /// The label is how the provider finds the image the controller built, so both must derive
2453    /// the same one; it also has to fit a label value, which a raw reference does not.
2454    #[test]
2455    fn the_disk_image_label_is_stable_and_label_safe() {
2456        let label = azure_disk_image_label("docker.io/library/python:3.14-slim");
2457        assert_eq!(
2458            label,
2459            azure_disk_image_label(" docker.io/library/python:3.14-slim ")
2460        );
2461        assert_ne!(
2462            label,
2463            azure_disk_image_label("docker.io/library/python:3.13-slim")
2464        );
2465        assert_eq!(label.len(), 32);
2466        assert!(label.chars().all(|c| c.is_ascii_hexdigit()), "{label}");
2467    }
2468
2469    /// A deadline is accepted only where the platform itself terminates on it — the kubelet's
2470    /// `activeDeadlineSeconds` and Lambda's `maximumDurationInSeconds`. Everywhere else it would
2471    /// need a reaper that does not exist, so it is refused rather than accepted and dropped.
2472    #[test]
2473    fn a_sandbox_deadline_is_accepted_only_where_the_platform_applies_it() {
2474        let mut sandbox = sandbox_with(SandboxEgress::None, vec![]);
2475        sandbox.lifecycle.max_lifetime_seconds = Some(3600);
2476
2477        sandbox
2478            .validate_for_platform(Platform::Kubernetes)
2479            .expect("the kubelet enforces activeDeadlineSeconds");
2480        sandbox
2481            .validate_for_platform(Platform::Aws)
2482            .expect("Lambda terminates the MicroVM at maximumDurationInSeconds");
2483
2484        for platform in [Platform::Azure, Platform::Local] {
2485            let error = sandbox
2486                .validate_for_platform(platform)
2487                .expect_err("a deadline nothing applies must be refused");
2488            assert_eq!(error.code, "SANDBOX_CAPABILITY_UNSUPPORTED");
2489        }
2490    }
2491
2492    /// A MicroVM bursts to four times its baseline with no way to opt out, so a ceiling is kept
2493    /// by choosing the size whose *peak* fits inside it. Sizing by baseline would hand back a
2494    /// sandbox that can reach four times what the customer declared.
2495    #[test]
2496    fn an_aws_size_is_chosen_so_its_peak_stays_inside_the_declared_ceiling() {
2497        let sandbox = sandbox_with(SandboxEgress::None, vec![]);
2498        let tier = sandbox
2499            .microvm_tier()
2500            .expect("2Gi/1cpu/20Gi is satisfiable");
2501
2502        assert_eq!(
2503            tier.peak_memory_mib, 2048,
2504            "the peak is the declared ceiling"
2505        );
2506        assert_eq!(
2507            tier.baseline_memory_mib, 512,
2508            "which is a quarter of it as the baseline"
2509        );
2510        assert!(tier.max_disk_mib <= 20 * 1024);
2511    }
2512
2513    /// AWS allocates one vCPU per 2GB, so a cpu ceiling below what the memory ceiling implies
2514    /// cannot be honoured together with it. Letting cpu choose the size instead would hand back a
2515    /// machine four times smaller than the memory asked for, with nothing to indicate it.
2516    #[test]
2517    fn a_cpu_ceiling_below_what_the_memory_implies_is_refused_not_quietly_downsized() {
2518        let mut sandbox = sandbox_with(SandboxEgress::None, vec![]);
2519        {
2520            let limits = sandbox
2521                .limits
2522                .as_mut()
2523                .expect("the fixture declares limits");
2524            limits.cpu = "1".to_string();
2525            limits.memory = "8Gi".to_string();
2526        }
2527
2528        let error = sandbox
2529            .microvm_tier()
2530            .expect_err("1 cpu and 8Gi cannot both be ceilings on AWS");
2531        assert!(
2532            error.to_string().contains("4 vCPU"),
2533            "the refusal must say what the memory ceiling implies: {error}"
2534        );
2535
2536        sandbox
2537            .limits
2538            .as_mut()
2539            .expect("the fixture declares limits")
2540            .cpu = "4".to_string();
2541        let tier = sandbox.microvm_tier().expect("4 cpu matches 8Gi");
2542        assert_eq!(tier.peak_memory_mib, 8192);
2543    }
2544
2545    /// Below AWS's smallest peak there is no size that holds the ceiling, and rounding up to the
2546    /// nearest one would silently exceed it.
2547    #[test]
2548    fn an_aws_ceiling_smaller_than_any_size_is_refused_rather_than_rounded() {
2549        let mut sandbox = sandbox_with(SandboxEgress::None, vec![]);
2550        sandbox
2551            .limits
2552            .as_mut()
2553            .expect("the fixture declares limits")
2554            .memory = "1Gi".to_string();
2555
2556        let error = sandbox
2557            .validate_for_platform(Platform::Aws)
2558            .expect_err("no MicroVM size peaks at or below 1Gi");
2559        assert_eq!(error.code, "SANDBOX_LIMIT_INVALID");
2560        assert!(
2561            error.to_string().contains("2Gi"),
2562            "the refusal must say what the smallest holdable ceiling is: {error}"
2563        );
2564    }
2565
2566    /// Source is refused at plan time on every platform whose sandbox image `alien build` does
2567    /// not build (the reason is at the check in `validate_for_platform`).
2568    #[test]
2569    fn source_code_is_refused_where_no_backend_builds_it() {
2570        let sandbox = Sandbox::new("agent".to_string())
2571            .code(SandboxCode::Source {
2572                src: "./sandbox".to_string(),
2573                toolchain: ToolchainConfig::Docker {
2574                    dockerfile: None,
2575                    build_args: None,
2576                    target: None,
2577                },
2578            })
2579            .egress(SandboxEgress::None)
2580            .lifecycle(SandboxLifecyclePolicy {
2581                max_lifetime_seconds: None,
2582                idle_pause_seconds: None,
2583            })
2584            .build();
2585
2586        sandbox
2587            .validate_for_platform(Platform::Aws)
2588            .expect("an AWS sandbox base image is built by `alien build`");
2589        sandbox
2590            .validate_for_platform(Platform::Azure)
2591            .expect("an Azure sandbox base image is built by `alien build`");
2592        sandbox
2593            .validate_for_platform(Platform::Gcp)
2594            .expect("a GCP sandbox base image is built by `alien build`");
2595
2596        for platform in [Platform::Kubernetes, Platform::Local] {
2597            let error = sandbox
2598                .validate_for_platform(platform)
2599                .expect_err("no backend builds a sandbox image from source here");
2600            assert_eq!(error.code, "SANDBOX_LIMIT_INVALID");
2601            assert!(
2602                error.to_string().contains("code.image"),
2603                "the refusal must say what to write instead: {error}"
2604            );
2605            assert!(
2606                error.to_string().contains(&platform.to_string()),
2607                "the refusal must name the platform that cannot build it: {error}"
2608            );
2609        }
2610    }
2611
2612    /// `validate_quantity` accepts nine suffixes. Reading only `Gi` and `Mi` would size a
2613    /// declared `4G` as though it were `4Gi`, which for a ceiling means exceeding it.
2614    #[test]
2615    fn every_accepted_unit_converts_rather_than_falling_back() {
2616        assert_eq!(quantity_mib("2Gi"), Some(2048));
2617        assert_eq!(quantity_mib("512Mi"), Some(512));
2618        assert_eq!(quantity_mib("4G"), Some(3814));
2619        assert_eq!(quantity_mib("1Ti"), Some(1024 * 1024));
2620        assert_eq!(millicores("1"), Some(1000));
2621        assert_eq!(millicores("500m"), Some(500));
2622    }
2623
2624    #[test]
2625    fn privileged_supervisor_fixes_identity_and_requires_an_enforceable_backend() {
2626        let mut sandbox = sandbox_with(
2627            SandboxEgress::AllowDomains {
2628                domains: vec!["example.com".to_string()],
2629            },
2630            vec![],
2631        );
2632        sandbox.privileged_supervisor = Some(SandboxPrivilegedSupervisor { command_uid: 60001 });
2633        sandbox
2634            .validate_for_platform(Platform::Aws)
2635            .expect("AWS can enforce the policy in the agent");
2636        assert_eq!(sandbox.cloud_egress(), &SandboxEgress::Allow);
2637        assert_eq!(
2638            sandbox.supervisor_environment()["ALIEN_SANDBOX_EXPECTED_EXEC_UID"],
2639            "60001"
2640        );
2641        for platform in [
2642            Platform::Local,
2643            Platform::Kubernetes,
2644            Platform::Azure,
2645            Platform::Gcp,
2646        ] {
2647            let error = sandbox
2648                .validate_for_platform(platform)
2649                .expect_err("cannot grant a privilege boundary the backend lacks");
2650            assert_eq!(error.code, "SANDBOX_CAPABILITY_UNSUPPORTED");
2651            assert!(error.message.contains("privilegedSupervisor"));
2652        }
2653        for uid in [0, u32::MAX] {
2654            sandbox.privileged_supervisor.as_mut().unwrap().command_uid = uid;
2655            let error = sandbox
2656                .validate_for_platform(Platform::Aws)
2657                .expect_err("invalid uid must fail at plan time");
2658            assert_eq!(error.code, "SANDBOX_LIMIT_INVALID");
2659        }
2660    }
2661
2662    #[test]
2663    fn unknown_fields_are_rejected() {
2664        let json = r#"{
2665            "id": "sbx",
2666            "code": {"type": "image", "image": "ubuntu:24.04"},
2667            "limits": {"cpu": "1", "memory": "2Gi", "disk": "20Gi"},
2668            "egress": {"mode": "deny"},
2669            "lifecycle": {},
2670            "unexpected": true
2671        }"#;
2672
2673        serde_json::from_str::<Sandbox>(json).expect_err("deny_unknown_fields must reject");
2674    }
2675
2676    #[test]
2677    fn serialization_roundtrips() {
2678        let sandbox = sandbox_with(
2679            SandboxEgress::AllowDomains {
2680                domains: vec!["example.com".to_string()],
2681            },
2682            vec![8080, 9090],
2683        );
2684
2685        let json = serde_json::to_string(&sandbox).expect("serializes");
2686        let restored: Sandbox = serde_json::from_str(&json).expect("deserializes");
2687        assert_eq!(sandbox, restored);
2688    }
2689
2690    #[test]
2691    fn id_is_immutable_across_updates() {
2692        let original = sandbox_with(SandboxEgress::None, vec![]);
2693        let renamed = Sandbox::new("other".to_string())
2694            .code(SandboxCode::Image {
2695                image: "ubuntu".to_string(),
2696                bundled_from: None,
2697            })
2698            .limits(
2699                original
2700                    .limits
2701                    .clone()
2702                    .expect("the fixture declares limits"),
2703            )
2704            .egress(SandboxEgress::None)
2705            .lifecycle(SandboxLifecyclePolicy {
2706                max_lifetime_seconds: None,
2707                idle_pause_seconds: None,
2708            })
2709            .build();
2710
2711        original
2712            .validate_update(&original.clone())
2713            .expect("an unchanged config is a valid update");
2714        original
2715            .validate_update(&renamed)
2716            .expect_err("renaming a sandbox is not an update");
2717    }
2718
2719    /// Azure declares an idle-pause policy but not a wall-clock ceiling.
2720    ///
2721    /// The two travel together in `SandboxLifecyclePolicy` and are gated separately on purpose:
2722    /// Azure pauses on idle and has no maximum lifetime, so accepting one and refusing the
2723    /// other is the honest split rather than an inconsistency.
2724    #[test]
2725    fn azure_takes_an_idle_policy_and_still_refuses_a_lifetime_ceiling() {
2726        let with_policy = |lifecycle: SandboxLifecyclePolicy| {
2727            Sandbox::new("sbx".to_string())
2728                .code(SandboxCode::Image {
2729                    image: "ubuntu".to_string(),
2730                    bundled_from: None,
2731                })
2732                .egress(SandboxEgress::Allow)
2733                .lifecycle(lifecycle)
2734                .build()
2735                .validate_for_platform(Platform::Azure)
2736        };
2737
2738        with_policy(SandboxLifecyclePolicy {
2739            max_lifetime_seconds: None,
2740            idle_pause_seconds: Some(900),
2741        })
2742        .expect("Azure pauses a sandbox on idle");
2743
2744        let error = with_policy(SandboxLifecyclePolicy {
2745            max_lifetime_seconds: Some(3600),
2746            idle_pause_seconds: None,
2747        })
2748        .expect_err("Azure has no wall-clock ceiling to enforce one with");
2749        assert_eq!(error.code, "SANDBOX_CAPABILITY_UNSUPPORTED");
2750        assert!(
2751            error.message.contains("sandboxLifetime"),
2752            "names the capability: {}",
2753            error.message
2754        );
2755    }
2756
2757    /// An allowlist naming nothing is a deny-all wearing an allowlist's label.
2758    ///
2759    /// It renders as a `Deny` default with no rules — the shape the Azure provider adds a
2760    /// catch-all to avoid — and a reader scanning the declaration sees "allowDomains" and reads
2761    /// the opposite of what it does.
2762    #[test]
2763    fn an_allowlist_with_no_domains_is_refused() {
2764        let declared = |domains: Vec<String>| {
2765            Sandbox::new("sbx".to_string())
2766                .code(SandboxCode::Image {
2767                    image: "ubuntu".to_string(),
2768                    bundled_from: None,
2769                })
2770                .egress(SandboxEgress::AllowDomains { domains })
2771                .lifecycle(SandboxLifecyclePolicy {
2772                    max_lifetime_seconds: None,
2773                    idle_pause_seconds: None,
2774                })
2775                .build()
2776                .validate_for_platform(Platform::Azure)
2777        };
2778
2779        let error = declared(vec![]).expect_err("an empty allowlist must be refused");
2780        assert_eq!(error.code, "SANDBOX_LIMIT_INVALID");
2781
2782        declared(vec!["api.example.com".to_string()])
2783            .expect("a named domain is what an allowlist is for");
2784    }
2785
2786    /// The two expressible modes map to the boolean; a host list maps to nothing so the caller has
2787    /// to refuse rather than silently pick a side.
2788    #[test]
2789    fn internet_access_switch_maps_only_the_two_expressible_modes() {
2790        assert_eq!(SandboxEgress::Allow.internet_access_switch(), Some(true));
2791        assert_eq!(SandboxEgress::None.internet_access_switch(), Some(false));
2792        assert_eq!(
2793            SandboxEgress::AllowDomains {
2794                domains: vec!["api.example.com".to_string()]
2795            }
2796            .internet_access_switch(),
2797            None,
2798            "a host list has no boolean and must not be approximated"
2799        );
2800    }
2801
2802    /// A sandbox resource as a stack state recorded it before `session` became `lifecycle` and
2803    /// `idleSuspendSeconds` became `idlePauseSeconds`. Such a state is still what a deployment
2804    /// that old holds, and reading it is the first step of updating or deleting that deployment.
2805    #[test]
2806    fn stack_state_written_before_the_lifecycle_rename_still_reads() {
2807        let stored: crate::StackResourceState = serde_json::from_value(serde_json::json!({
2808            "type": "sandbox",
2809            "config": {
2810                "id": "sandbox",
2811                "code": {
2812                    "type": "image",
2813                    "image": "s3://example-bundles/analysis/v1/bundle.zip"
2814                },
2815                "type": "sandbox",
2816                "egress": { "mode": "allow" },
2817                "session": { "maxLifetimeSeconds": 28800, "idleSuspendSeconds": 600 }
2818            },
2819            "status": "running",
2820            "outputs": {
2821                "type": "sandbox",
2822                "identifier": "arn:aws:lambda:us-east-2:123456789012:microvm-image:example-sandbox",
2823                "parentName": "arn:aws:lambda:us-east-2:123456789012:microvm-image:example-sandbox"
2824            },
2825            "lifecycle": "frozen",
2826            "dependencies": [
2827                { "id": "management", "type": "remote-stack-management" },
2828                { "id": "access", "type": "resource-access" }
2829            ],
2830            "controllerPlatform": "aws"
2831        }))
2832        .expect("a stack state written before the rename must still deserialize");
2833
2834        let expected = Sandbox::new("sandbox".to_string())
2835            .code(SandboxCode::Image {
2836                image: "s3://example-bundles/analysis/v1/bundle.zip".to_string(),
2837                bundled_from: None,
2838            })
2839            .egress(SandboxEgress::Allow)
2840            .lifecycle(SandboxLifecyclePolicy {
2841                max_lifetime_seconds: Some(28_800),
2842                idle_pause_seconds: Some(600),
2843            })
2844            .build();
2845        assert_eq!(stored.config.downcast_ref::<Sandbox>(), Some(&expected));
2846
2847        let rewritten = serde_json::to_value(&stored.config).expect("the sandbox serializes");
2848        assert_eq!(
2849            rewritten["lifecycle"],
2850            serde_json::json!({ "maxLifetimeSeconds": 28800, "idlePauseSeconds": 600 }),
2851            "the next write stores the current names"
2852        );
2853        assert!(rewritten.get("session").is_none());
2854    }
2855}