microvms_app/control/mod.rs
1// SPDX-License-Identifier: Apache-2.0
2//! The control-plane client: SigV4-signed rest-json, and every trap closed before the
3//! request leaves the process.
4//!
5//! # The shape of the thing
6//!
7//! [`ControlPlane`] is the whole surface. It holds a [`transport::Transport`] (signed and
8//! real, or a contract recorder in a test), a [`Region`], and a [`Clock`] — and every
9//! operation is a method on it. There is no builder, no config struct, and no partially
10//! constructed state: `ControlPlane::new` (`microvms_core::prelude::ControlPlaneExt`) resolves
11//! credentials and either yields a usable client or says why not.
12//!
13//! # What is closed here, and how strongly
14//!
15//! * **TRAP-1** — [`token`]: image tokens are always fresh. Run tokens are fresh unless
16//! explicitly persisted for recovery of one launch. Never reuse them for a new VM.
17//! * **TRAP-2** — [`image`]: a build stuck in `CREATING` past the stall grace with every
18//! build `PENDING` is rejected naming the replay signature. S2.
19//! * **TRAP-3** — [`CreateImageRequest::repair_guest_identity`] is a `bool`, and the
20//! request injects `["ALL"]` itself. There is no capability list to put
21//! `CAP_SYS_ADMIN` in. S1.
22//! * **TRAP-4** — [`connector::ConnectorIntent`]: connectors are derived ARNs from a
23//! closed intent enum. S1.
24//! * **TRAP-5** — [`RunHookPayload`]: 4096 bytes inclusive, checked before any call. S2,
25//! and the type means an unchecked string cannot reach the field.
26//! * **TRAP-8** — [`ControlPlane::wait_for_running`]: a terminal state before RUNNING is
27//! rejected with the state *and* `stateReason` attached. S2.
28//! * **TRAP-11, revised again** — `mint_shell_auth_token` exists now, and the guard
29//! changed ground with it: the closure is no longer the method's absence but the exec
30//! path's separation from it, which is why the test still counts the calls a full
31//! lifecycle made and asserts zero shell operations among them. The connector half is
32//! unchanged: `SHELL_INGRESS` is a [`connector::ConnectorIntent`] variant, and the
33//! lifecycle test asserts a launch carries exactly the connectors its caller asked
34//! for. See [`connector`].
35//!
36//! # TRAP-11 is the one that needs saying out loud
37//!
38//! `CreateMicrovmShellAuthToken` is implemented — issue #69 builds `microvm shell` on
39//! it, and [`ControlPlane::mint_shell_auth_token`] is its one door. The original reason
40//! for staying away entirely — that the shell gates a console-only debugging flow, not
41//! a programmatic exec path — did not survive measurement: `docs/PLATFORM.md`
42//! (2026-08-15) found a real PTY over a WebSocket, programmatically drivable. What
43//! holds is narrower: **one interactive session is not programmatic exec** — no exec
44//! ids, no idempotency, no separated stdout/stderr, no exit codes — so the exec path
45//! never requests the shell connector and never mints a shell token, and the shell is
46//! its own surface ([`crate::session::shell`]) rather than a method the exec path can
47//! wander into.
48//!
49//! # Nothing in this module reads the service model at runtime
50//!
51//! Every constraint is a constant in [`crate::constants`], checked by the build gate
52//! against the pinned model (TRAP-12). That matters because botocore's
53//! `VALIDATED_METADATA_ATTRS` is `{required, min, document, union}` — `max`, `pattern`,
54//! and `enum` violations go to the wire — so every guard in this module is load-bearing
55//! rather than belt-and-braces, and "the SDK validates the model already" was never true.
56//!
57//! # And "botocore validates `min`" was never true *of this client*
58//!
59//! The sentence above is about `max`, `pattern`, and `enum`, and for a while it left `min`
60//! implicitly exempt: `min` **is** in `VALIDATED_METADATA_ATTRS`, so a `min` violation really is
61//! refused before the wire — by botocore. This client does not use botocore. It signs with
62//! `aws-sigv4` and sends with `reqwest`, and `validate.py` is nowhere in the dependency graph.
63//! That reasoning came from the deleted Python client, where it held, and it did not survive the
64//! port; issue #24 measured the consequence, `maxIdleDurationSeconds: 59` on the wire.
65//!
66//! So the rule is simpler than it was: **every constraint the model states on a member this
67//! client sends is enforced by this module or by nothing.** The guards below are the whole of it:
68//!
69//! * [`require_valid_image_name`] — `ImageName` (1..=64, `[a-zA-Z0-9-_]+`).
70//! * [`require_duration_in_range`] — `maximumDurationInSeconds` (1..=28800).
71//! * [`require_valid_version`] — `Version` (1..=2048, `[^\s]+`).
72//! * [`require_non_blank`] — `NonBlankString`, the model's most-reused shape.
73//! * [`require_valid_identifier`] — `MicrovmIdentifier`/`MicrovmImageIdentifier` (1..=256).
74//! * [`require_valid_role_arn`] — `RoleArn` (20..=2048, plus the twelve-digit account).
75//! * [`require_valid_port`] — `PortNumber`/`HooksPortInteger` (`min: 1`; 0 is not a port).
76//! * [`require_idle_duration`] — `maxIdleDurationSeconds` (`min: 60`).
77//! * [`require_valid_tags`] — `TagKey`/`TagValue`, two ceilings and two minima.
78//!
79//! Three constraints are closed by a **type** rather than by a function, which is the stronger
80//! form: [`RunHookPayload`] cannot hold over 4096 bytes, [`ops::VersionStatus`] and
81//! [`ops::HookState`] cannot spell a value the enum does not have, and
82//! [`connector::ConnectorIntent`] cannot name a connector the platform does not publish. A
83//! constraint the type system enforces cannot be forgotten at a call site.
84
85pub mod artifact;
86pub mod connector;
87pub mod context;
88pub mod ensure;
89pub mod image;
90pub mod microvm;
91pub mod ops;
92pub mod project;
93pub mod services;
94pub mod token;
95pub mod transport;
96
97use std::sync::Arc;
98use std::time::Duration;
99
100pub use artifact::{
101 BaseImage, Ecosystem, ProjectFiles, WrapOptions, artifact_content_hash, build_artifact,
102 default_dockerfile, wrap_dockerfile,
103};
104pub use connector::{
105 ConnectorIntent, EgressPosture, PLATFORM_HONOURS_OMITTED_EGRESS, egress_posture_for,
106};
107pub use context::{BuildContext, ContextEntry};
108pub use ensure::{EnsureImageRequest, EnsuredImage};
109pub use image::{Image, WaitOpts};
110pub use microvm::{Microvm, MicrovmFilter, ProxyToken, RunHookPayload};
111pub use services::BuildServices;
112
113use crate::error::{Error, ErrorKind};
114use crate::hooks::{BuildHookTimeout, RunHookTimeout};
115use crate::region::Region;
116use crate::sizing::SizeClass;
117
118/// The default agent port, matching the daemon's own default.
119pub const DEFAULT_AGENT_PORT: u16 = 9000;
120
121use crate::adapters::Adapters;
122/// The clock every wait here measures its deadline on. It's the crate's one [`Clock`], so the
123/// plane and the sessions it hands out read the same time source.
124pub use crate::clock::Clock;
125use crate::entropy::Entropy;
126
127/// The control-plane client.
128///
129/// The transport, clock, entropy and adapters sit behind `Arc`, so a caller holding one across
130/// tasks does not need a second credential chain.
131pub struct ControlPlane {
132 transport: Arc<dyn transport::Transport>,
133 region: Region,
134 clock: Arc<dyn Clock>,
135 /// Where client-token nonces, agent tokens and identity seeds come from.
136 entropy: Arc<dyn Entropy>,
137 /// What a sandbox over this plane builds its session backend and its STS and S3 client
138 /// from.
139 adapters: Arc<dyn Adapters>,
140 /// The agent port, which the hooks block and the proxy token both need.
141 port: u16,
142}
143
144impl std::fmt::Debug for ControlPlane {
145 fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
146 f.debug_struct("ControlPlane")
147 .field("region", &self.region)
148 .field("port", &self.port)
149 .finish_non_exhaustive()
150 }
151}
152
153impl ControlPlane {
154 /// A client over caller-supplied ports, on the default agent port.
155 ///
156 /// Every port is explicit, including the entropy. A plane over a scripted transport
157 /// still mints client tokens, and a deterministic default would mint repeating ones
158 /// (TRAP-1), so nothing here picks a source for the caller.
159 ///
160 /// `ControlPlane::new(region)` and `ControlPlane::with_transport(transport, region, clock)`
161 /// are in `microvms_core::prelude`: they wire the production transport, clock, entropy and
162 /// adapters, which is I/O this constructor leaves to its caller. A test builds one with
163 /// `testing::control_plane` instead.
164 pub fn from_ports(
165 transport: Arc<dyn transport::Transport>,
166 region: Region,
167 clock: Arc<dyn Clock>,
168 entropy: Arc<dyn Entropy>,
169 adapters: Arc<dyn Adapters>,
170 ) -> Self {
171 Self {
172 transport,
173 region,
174 clock,
175 entropy,
176 adapters,
177 port: DEFAULT_AGENT_PORT,
178 }
179 }
180
181 /// Replaces the source client-token nonces, agent tokens and identity seeds are drawn
182 /// from. The seam a test pins those values through.
183 #[must_use]
184 pub fn with_entropy(mut self, entropy: Arc<dyn Entropy>) -> Self {
185 self.entropy = entropy;
186 self
187 }
188
189 /// Sets the agent port the hooks block and the proxy token name.
190 ///
191 /// # Why this is fallible where every other builder method is not
192 ///
193 /// Because `port` is the one value on this type that the model bounds, and 0 is a value a
194 /// caller reaches for on purpose. `PortNumber` and `HooksPortInteger` are both `min: 1`;
195 /// `u16` closes the ceiling and nothing closed the floor, so before this returned a `Result`
196 /// the port set here landed in two places on the wire — `hooks.port` on every
197 /// `CreateMicrovmImage` built through the plane, and `allowedPorts: [{"port": 0}]` on every
198 /// token it minted (issue #24).
199 ///
200 /// Both of those are guarded at the call itself as well ([`require_valid_port`] in
201 /// [`ControlPlane::create_image`] and in [`ControlPlane::mint_auth_token_for`]), so the
202 /// constraint would be closed without this. It is fallible anyway because the two refusals
203 /// arrive a long way from the mistake: the create one after an artifact upload, and the mint
204 /// one at the moment a session needs a credential. Refusing here means `--port 0` fails
205 /// before a control plane is even usable, which is the difference between "that port is not
206 /// legal" and "your launch failed".
207 ///
208 /// Zero rather than the ceiling is what this actually catches, and it is not a typo case: 0
209 /// is what "let the kernel choose" means to a listener, so it is the value a caller passes
210 /// when they have a socket they have not bound yet.
211 pub fn with_port(mut self, port: u16) -> Result<Self, Error> {
212 require_valid_port("port", port)?;
213 self.port = port;
214 Ok(self)
215 }
216
217 /// The region every ARN in a request is derived for.
218 pub fn region(&self) -> &Region {
219 &self.region
220 }
221
222 /// The agent port.
223 pub fn port(&self) -> u16 {
224 self.port
225 }
226
227 pub(crate) fn transport(&self) -> &dyn transport::Transport {
228 self.transport.as_ref()
229 }
230
231 pub(crate) fn clock(&self) -> &dyn Clock {
232 self.clock.as_ref()
233 }
234
235 /// The source this plane draws client-token nonces, agent tokens and identity seeds from.
236 ///
237 /// Public so the composition root's wiring can be checked from outside this crate: a
238 /// production plane must carry the OS pool, never a deterministic source (TRAP-1).
239 pub fn entropy(&self) -> &dyn Entropy {
240 self.entropy.as_ref()
241 }
242
243 /// The clock itself, for a session that has to read the same one.
244 pub(crate) fn shared_clock(&self) -> Arc<dyn Clock> {
245 Arc::clone(&self.clock)
246 }
247
248 pub(crate) fn adapters(&self) -> &dyn Adapters {
249 self.adapters.as_ref()
250 }
251}
252
253/// Everything `CreateMicrovmImage` needs, with the traps closed in the type.
254///
255/// # What is deliberately not here
256///
257/// A `client_token` field. See [`token`] for the fifteen-hour wedge that absence prevents;
258/// [`CreateImageRequest::token_scope`] is a **label** folded in beside a fresh nonce and
259/// cannot become the token.
260///
261/// A `capabilities` list. See `repair_guest_identity`.
262///
263/// An `architecture` field. The model's enum has exactly one value, so the only thing a
264/// field could express is a rejected request — and the rejection arrives *after* the
265/// artifact upload, reported as a constraint on a field the caller thought was a choice.
266/// `ARM_64` is injected. A caller who built an x86 binary finds out from `doctor`'s ELF
267/// check, not from a build cycle.
268#[derive(Clone, Debug)]
269pub struct CreateImageRequest {
270 /// The image name. Validated against `ImageName` (1..=64, `[a-zA-Z0-9-_]+`) before the
271 /// artifact is uploaded, because AWS's rejection arrives *after* the upload.
272 pub name: String,
273 /// The base image, pairing the `baseImageArn` with the Dockerfile `FROM`.
274 pub base_image: BaseImage,
275 /// `baseImageVersion`, or `None` to take whatever the service currently defaults to.
276 ///
277 /// # Why pinning is worth a field
278 ///
279 /// Without it a build floats. The managed base's version list is not static — `al2023-1`
280 /// carried one version in June and two by July (`"0"` and `"1"`, measured 2026-08-16) — so
281 /// two builds of identical inputs weeks apart can sit on different bases, and neither
282 /// recorded which. That is a reproducibility hole with no local symptom: the build
283 /// succeeds either way, and the difference appears in the guest.
284 ///
285 /// `None` is the default and stays the default, because pinning without knowing the legal
286 /// values is how a caller pins to a version that has been withdrawn.
287 /// [`ControlPlane::managed_base_versions`] is what answers that, and the version strings
288 /// are **bare integers** for a managed base where a custom image's are `"1.0"` — the two
289 /// are not comparable, so a value from anywhere else does not belong here.
290 ///
291 /// Checked against the `Version` shape before the call, because the create happens after
292 /// the artifact upload.
293 pub base_image_version: Option<String>,
294 /// The Dockerfile, or `None` for the derived default.
295 ///
296 /// A caller-supplied one is checked against `base_image`'s `docker_ref`: the two
297 /// disagreeing builds against a base none of the measured platform behaviour applies
298 /// to.
299 pub dockerfile: Option<String>,
300 /// The daemon binary's bytes, zipped into the artifact.
301 pub binary: Vec<u8>,
302 /// The project's dependency files — the lockfile plus its manifest — zipped into the
303 /// artifact beside the Dockerfile, or `None` for an image with no environment layer.
304 ///
305 /// This is what lets the derived Dockerfile's install step (`uv sync --locked`,
306 /// `npm ci`, `cargo fetch` per [`Ecosystem`]) see the files it installs from (#74).
307 /// The entry names are fixed by the ecosystem; there is no way to put an arbitrary
308 /// file into the shared snapshot through this field, which is the TRAP-5 property
309 /// restated as a type.
310 pub project_files: Option<ProjectFiles>,
311 /// Where the artifact is uploaded to. `create_image` does not upload: the caller puts
312 /// the bytes there and passes the URI. [`crate::sandbox::Sandbox::ensure_image`] is the
313 /// path that uploads for itself ([`services`]).
314 pub code_artifact_uri: String,
315 /// The build role, which must grant logs on `/aws/lambda-microvms/*`.
316 pub build_role_arn: String,
317 /// The size class. Selects a documented baseline; it does not size the VM (TRAP-10).
318 pub size: SizeClass,
319 /// Whether to repair guest identity (TRAP-3).
320 ///
321 /// # Why a bool and not a capability list
322 ///
323 /// Measured 2026-08-06: without the capability, a guest running as root still gets
324 /// `EPERM` from `sethostname` and from a bind mount over
325 /// `/proc/sys/kernel/random/boot_id`, because the MicroVM drops `CAP_SYS_ADMIN` by
326 /// default. Writing `/etc/machine-id` needs no capability and succeeds either way,
327 /// which is what makes the gap easy to miss — identity repair looks like it works
328 /// until you check the two steps that need the kernel's permission rather than the
329 /// filesystem's.
330 ///
331 /// `Capability`'s enum is exactly `["ALL"]`, so there is no way to ask for
332 /// `CAP_SYS_ADMIN` alone. A list parameter would let a caller write
333 /// `["CAP_SYS_ADMIN"]` — the request AWS rejects *after* the artifact upload — so the
334 /// intent is a boolean and the request injects `["ALL"]` itself.
335 pub repair_guest_identity: bool,
336 /// Whether the daemon should inherit the image's `WORKDIR`.
337 ///
338 /// Rejected when nothing declares one: most public ARM64 bases leave it empty, so the
339 /// inheritance silently resolves to `/` and the symptom appears in the guest a build
340 /// cycle later.
341 pub inherit_workdir: bool,
342 /// The run-family hook timeout: `run`, `resume`, `suspend`, `terminate`. Caps at 60s,
343 /// and the type is what keeps a build-sized value out of it.
344 pub run_hook_timeout: RunHookTimeout,
345 /// The build-family hook timeout: `ready`, `validate`. Caps at 3600s.
346 pub build_hook_timeout: BuildHookTimeout,
347 /// Tags, or none.
348 pub tags: std::collections::BTreeMap<String, String>,
349 /// `logging.cloudWatch.logGroup`, or `None` for the service default — a
350 /// service-created group under `/aws/lambda-microvms/<image-name>` with random stream
351 /// names.
352 ///
353 /// Checked against `CloudWatchLoggingLogGroupString` (1..=512, `[a-zA-Z0-9_\-/.#]+`)
354 /// before the artifact upload. The build role must be able to write to whatever this
355 /// names: the conformance account's role grants logs only on
356 /// `/aws/lambda-microvms/*`, so a group outside that prefix builds with no logs at
357 /// all — the same failure mode as the wrong-prefix policy (docs/PLATFORM.md).
358 pub log_group: Option<String>,
359 /// `logging.cloudWatch.logStream` — a stream-name **prefix**, never the exact name.
360 ///
361 /// # The client always appends a per-build discriminator
362 ///
363 /// The wire member is an EXACT stream name (prefixes unsupported — measured 2026-08,
364 /// docs/PLATFORM.md 'An image build is three VMs and three log streams'), and one
365 /// build emits three streams. A fixed configured name collapses all three of every
366 /// build into one stream, so [`ControlPlane::create_image`] sends
367 /// `<this>/<16 hex>` with fresh CSPRNG per create attempt — the same mechanism as
368 /// the `clientToken` nonce (TRAP-1) — and the resolved name comes back on
369 /// [`image::Image::log_stream`] so the caller can find their logs.
370 ///
371 /// Capped at [`crate::constants::MAX_USER_LOG_STREAM_LEN`] (495) so the resolved name
372 /// fits the shape's 512, and refused when it carries `:` or `*` (the pattern is
373 /// `[^:*]*`). Requires [`Self::log_group`]: a stream inside a group the service
374 /// chose randomly is a stream nobody can predict the location of.
375 pub log_stream: Option<String>,
376 /// A **label** folded into the create token for CloudTrail readability, defaulting to
377 /// the image name. Not the token, and cannot become it (TRAP-1).
378 pub token_scope: Option<String>,
379}
380
381impl CreateImageRequest {
382 /// A request with the measured defaults: the al2023 base, the default size class, the
383 /// derived Dockerfile, identity repair off, and 30-second hook timeouts.
384 ///
385 /// The hook timeouts are `expect`ed rather than propagated because 30 is legal for both
386 /// families by construction — it is below the run family's 60s ceiling, which is the
387 /// lower of the two — and a `Result` here would put a `?` on every call site for a
388 /// value this function chose itself.
389 pub fn new(
390 name: impl Into<String>,
391 binary: Vec<u8>,
392 code_artifact_uri: impl Into<String>,
393 build_role_arn: impl Into<String>,
394 ) -> Self {
395 Self {
396 name: name.into(),
397 base_image: BaseImage::al2023(),
398 base_image_version: None,
399 dockerfile: None,
400 binary,
401 project_files: None,
402 code_artifact_uri: code_artifact_uri.into(),
403 build_role_arn: build_role_arn.into(),
404 size: SizeClass::DEFAULT,
405 repair_guest_identity: false,
406 inherit_workdir: false,
407 run_hook_timeout: RunHookTimeout::try_new(30).expect("30s is under the 60s ceiling"),
408 build_hook_timeout: BuildHookTimeout::try_new(30)
409 .expect("30s is under the 3600s ceiling"),
410 tags: std::collections::BTreeMap::new(),
411 log_group: None,
412 log_stream: None,
413 token_scope: None,
414 }
415 }
416}
417
418/// Everything `RunMicrovm` needs.
419///
420/// `run_hook_payload` is a [`RunHookPayload`], which cannot hold an over-ceiling value —
421/// so TRAP-5 is closed before this struct exists rather than checked when it is used.
422#[derive(Clone, Debug)]
423pub struct RunMicrovmRequest {
424 /// The image ARN or ID to launch.
425 pub image_identifier: String,
426 /// `imageVersion`, or `None` for the image's own latest active version.
427 ///
428 /// # The launch half of blue/green
429 ///
430 /// `None` is what this client always sent, and it means "whatever
431 /// `latestActiveImageVersion` is now" — which is right for the ordinary case and wrong for
432 /// the two that matter. A canary wants to launch against exactly the version it is
433 /// testing, not against whatever became latest while it was starting. And a rollback wants
434 /// to re-pin to the known-good version, which "latest" cannot express at all once a bad
435 /// version is the latest one.
436 ///
437 /// With [`ControlPlane::set_image_version_status`] the three moves compose: build v2,
438 /// canary-launch pinned to v2, and on failure set v2 INACTIVE and re-pin to v1. A version
439 /// set INACTIVE refuses to launch when pinned here, which is what makes the retire real
440 /// rather than advisory.
441 ///
442 /// Checked against the `Version` shape before the call — see
443 /// [`require_valid_version`] on why a rollback is the worst moment for a rejection about
444 /// the request.
445 pub image_version: Option<String>,
446 /// The execution role. Optional in the model; every real launch needs one.
447 pub execution_role_arn: Option<String>,
448 /// The connectors to request, as intents (TRAP-4). Ingress is required for a session
449 /// to work at all; omitting egress omits the connector from the request (measured
450 /// 2026-09-12: the platform gave the VM outbound network anyway, `docs/PLATFORM.md`).
451 pub connectors: Vec<ConnectorIntent>,
452 /// Customer-managed Lambda VPC egress connector ARNs, created through `lambda-core`.
453 /// Cannot be combined with [`ConnectorIntent::Egress`]. Internet isolation requires
454 /// a VPC without an internet gateway or NAT gateway; an ARN alone proves no isolation.
455 pub egress_network_connectors: Vec<String>,
456 /// The already-validated payload carrying the agent token.
457 pub run_hook_payload: RunHookPayload,
458 /// `maximumDurationInSeconds`, checked against 1..=28800 when the request is sent.
459 pub max_duration_sec: u32,
460 /// `idlePolicy.maxIdleDurationSeconds`, checked against the model's `min: 60` before the call.
461 ///
462 /// This comment used to say the guard was deliberately absent, because `min` is one of the four
463 /// keys botocore's `VALIDATED_METADATA_ATTRS` enforces locally. True of botocore, and this
464 /// client never touches it — see [`require_idle_duration`]. `59` reached the wire.
465 pub max_idle_sec: u32,
466 /// `idlePolicy.suspendedDurationSeconds`.
467 ///
468 /// The model's `min` is **0**, so every `u32` satisfies it and there is no guard — an absence
469 /// with a reason rather than an oversight, and the drift gate pins the 0 so a future floor is
470 /// noticed. Its ceiling is unstated.
471 ///
472 /// The requested value drives the local resume window. The service also reports its
473 /// active policy in [`Microvm::idle_policy`].
474 pub suspended_sec: u32,
475 /// `idlePolicy.autoResumeEnabled`.
476 pub auto_resume: bool,
477 /// A label for the run token, defaulting to the image identifier (TRAP-1).
478 pub token_scope: Option<String>,
479 /// Optional durable launch idempotency key. Reuse only for retries of one launch,
480 /// with identical parameters and run-hook payload. None preserves fresh defaults.
481 /// Never derive this from an image name or reuse it for a different VM.
482 pub client_token: Option<String>,
483 /// Per-VM `logging`: a CloudWatch group (and optional exact stream) or disabled.
484 /// `None` leaves the service's default destination in place (#201).
485 pub logging: Option<ops::Logging>,
486}
487
488impl RunMicrovmRequest {
489 /// A launch with the measured defaults: ingress only, ten-minute idle and suspended
490 /// windows, a one-hour maximum duration, no auto-resume.
491 pub fn new(image_identifier: impl Into<String>, run_hook_payload: RunHookPayload) -> Self {
492 Self {
493 image_identifier: image_identifier.into(),
494 image_version: None,
495 execution_role_arn: None,
496 connectors: vec![ConnectorIntent::AllIngress],
497 egress_network_connectors: Vec::new(),
498 run_hook_payload,
499 max_duration_sec: 3_600,
500 max_idle_sec: 600,
501 suspended_sec: 600,
502 auto_resume: false,
503 token_scope: None,
504 client_token: None,
505 logging: None,
506 }
507 }
508
509 /// Requests the managed internet egress connector. Omission does not block internet access.
510 #[must_use]
511 pub fn with_egress(mut self) -> Self {
512 if !self.connectors.contains(&ConnectorIntent::Egress) {
513 self.connectors.push(ConnectorIntent::Egress);
514 }
515 self
516 }
517
518 /// Routes egress through a customer-managed Lambda network connector.
519 #[must_use]
520 pub fn with_egress_network_connector(mut self, arn: impl Into<String>) -> Self {
521 self.egress_network_connectors.push(arn.into());
522 self
523 }
524
525 /// Requests a shell-capable launch: the ingress set becomes the measured pair,
526 /// `[HTTP_INGRESS, SHELL_INGRESS]`.
527 ///
528 /// **Replaces** `ALL_INGRESS` rather than adding beside it, because the platform
529 /// forbids the combination and says so only at token-mint time — the VM launches,
530 /// reaches RUNNING, and bills before the failure appears (`docs/PLATFORM.md`,
531 /// 2026-08-15). `[HTTP_INGRESS, SHELL_INGRESS]` is the set measured to both launch
532 /// and mint a shell token, and `HTTP_INGRESS` stays in the pair so the daemon
533 /// endpoint keeps working. Egress is untouched: shell access and outbound network
534 /// are separate questions.
535 #[must_use]
536 pub fn with_shell(mut self) -> Self {
537 self.connectors
538 .retain(|intent| *intent != ConnectorIntent::AllIngress);
539 for wanted in [ConnectorIntent::HttpIngress, ConnectorIntent::ShellIngress] {
540 if !self.connectors.contains(&wanted) {
541 self.connectors.push(wanted);
542 }
543 }
544 self
545 }
546
547 /// Pins the launch to one `imageVersion`. See the field for why that matters.
548 #[must_use]
549 pub fn with_image_version(mut self, version: impl Into<String>) -> Self {
550 self.image_version = Some(version.into());
551 self
552 }
553}
554
555/// Rejects a `maximumDurationInSeconds` outside the service range.
556///
557/// 28800 is eight hours and the hard ceiling on any single VM's life. A longer session
558/// needs a second VM, not a larger number — which is what the message says, because
559/// "outside the accepted range" alone invites someone to look for the flag that raises it.
560pub fn require_duration_in_range(seconds: u32) -> Result<u32, Error> {
561 if (1..=crate::constants::MAX_DURATION_SEC).contains(&seconds) {
562 return Ok(seconds);
563 }
564 Err(Error::invalid_arg(format!(
565 "maximumDurationInSeconds={seconds} is outside the accepted range 1..{} (service model \
566 {}) — {} seconds is eight hours, the hard ceiling on any one VM's life. A longer \
567 session needs a second VM, not a larger number.",
568 crate::constants::MAX_DURATION_SEC,
569 crate::constants::MODEL_API_VERSION,
570 crate::constants::MAX_DURATION_SEC,
571 )))
572}
573
574/// Rejects a `Version`/`NonBlankString` value the service would reject.
575///
576/// # Why this one is checked locally when so many are not
577///
578/// The model's `Version` shape is `min: 1, max: 2048, pattern: [^\s]+`, and issue #24 lists
579/// `NonBlankString` as its most-reused unguarded shape. Two of the three constraints are ones
580/// botocore would not catch even if this client used it (`max` and `pattern` are outside
581/// `VALIDATED_METADATA_ATTRS`), and this client uses `aws-sigv4` plus `reqwest` and never
582/// touches botocore's validator at all.
583///
584/// What makes it worth a guard rather than a comment is *where* the two callers sit.
585/// `CreateMicrovmImage.baseImageVersion` is sent **after** the artifact upload, so the
586/// service's rejection costs the caller the upload — the same argument
587/// [`require_valid_image_name`] makes. And `RunMicrovm.imageVersion` is the pinned-launch
588/// half of a rollback: a blank there is a `ValidationException` at the moment someone is
589/// trying to re-pin away from a bad version, which is the worst possible time for the failure
590/// to be about the request rather than about the version.
591///
592/// The pattern is `[^\s]+` — **no whitespace anywhere**, not merely "not blank". A version
593/// copied out of a terminal with a trailing newline satisfies "non-empty" and fails the
594/// pattern, so the message names the character rather than saying the value is invalid.
595pub fn require_valid_version(member: &str, version: &str) -> Result<(), Error> {
596 let model = crate::constants::MODEL_API_VERSION;
597 if version.is_empty() {
598 return Err(Error::invalid_arg(format!(
599 "{member} is empty, but the Version shape requires at least 1 character (service \
600 model {model}). Omit the member entirely to let the service choose — an absent \
601 version and a blank one are different requests, and only the first is legal."
602 )));
603 }
604 if version.len() > crate::constants::MAX_VERSION_LEN {
605 return Err(Error::invalid_arg(format!(
606 "{member} is {} characters, over the Version ceiling of {} (service model {model}).",
607 version.len(),
608 crate::constants::MAX_VERSION_LEN,
609 )));
610 }
611 if let Some(found) = version.chars().find(char::is_ascii_whitespace) {
612 return Err(Error::invalid_arg(format!(
613 "{member} {version:?} contains whitespace ({found:?}), which the Version pattern \
614 {:?} forbids anywhere in the value (service model {model}). A version pasted from \
615 a terminal carries a trailing newline and looks fine; this is that.",
616 crate::constants::VERSION_PATTERN,
617 )));
618 }
619 Ok(())
620}
621
622/// Rejects a `NonBlankString` value the service would reject.
623///
624/// # The model's most-reused shape, and the three members this client sends
625///
626/// `NonBlankString` is `min: 1, max: 2048, pattern: [^\s]+` and 45 members name it. Most are
627/// responses; the three this client puts on the wire are `CodeArtifact.uri`,
628/// `CreateMicrovmImage.baseImageArn`, and `ListMicrovmImages.nameFilter`, and issue #24 named
629/// all three as reachable with no guard.
630///
631/// Each of the three fails in its own expensive way. `codeArtifact.uri` and `baseImageArn` ride
632/// on the create call, which happens **after** the artifact upload — the ordering
633/// [`ControlPlane::create_image`] is arranged around, so a rejection about either costs the
634/// caller the upload. `nameFilter` is worse in a quieter way: it goes in the **query string**,
635/// so a blank one is a `nameFilter=` that either 400s or silently filters differently from what
636/// was meant, and [`ControlPlane::find_image_by_name`] is the resolver a launch depends on.
637///
638/// # `[^\s]+` forbids whitespace *anywhere*
639///
640/// Not merely "not blank". An S3 URI pasted from a console with a trailing newline satisfies
641/// "non-empty" and fails the pattern, which is why the message names the character it found
642/// rather than saying the value is invalid. The same reading [`require_valid_version`] documents
643/// for `Version`, which is the identical constraint triple on a different shape.
644///
645/// A separate function from [`require_valid_version`] even though the two shapes agree today,
646/// for the reason [`crate::constants::MAX_NON_BLANK_LEN`] gives: they are two shapes, AWS can
647/// move one, and one guard serving both would silently stop being about whichever moved.
648pub fn require_non_blank(member: &str, value: &str) -> Result<(), Error> {
649 let model = crate::constants::MODEL_API_VERSION;
650 if value.is_empty() {
651 return Err(Error::invalid_arg(format!(
652 "{member} is empty, but the NonBlankString shape requires at least 1 character \
653 (service model {model}). The shape's own documentation reads 'a string which is not \
654 empty or blank (only whitespace)'."
655 )));
656 }
657 if value.len() > crate::constants::MAX_NON_BLANK_LEN {
658 return Err(Error::invalid_arg(format!(
659 "{member} is {} characters, over the NonBlankString ceiling of {} (service model \
660 {model}).",
661 value.len(),
662 crate::constants::MAX_NON_BLANK_LEN,
663 )));
664 }
665 if let Some(found) = value.chars().find(char::is_ascii_whitespace) {
666 return Err(Error::invalid_arg(format!(
667 "{member} {value:?} contains whitespace ({found:?}), which the NonBlankString \
668 pattern {:?} forbids anywhere in the value (service model {model}). A URI pasted \
669 from a console carries a trailing newline and looks fine; this is that.",
670 crate::constants::NON_BLANK_PATTERN,
671 )));
672 }
673 Ok(())
674}
675
676/// Rejects a `MicrovmIdentifier`/`MicrovmImageIdentifier` the service would reject.
677///
678/// # Twelve members across every implemented operation
679///
680/// Both shapes are `min: 1, max: 256`, and between them they bound the identifier on
681/// `GetMicrovm`, `SuspendMicrovm`, `ResumeMicrovm`, `TerminateMicrovm`,
682/// `CreateMicrovmAuthToken`, `GetMicrovmImage`, `DeleteMicrovmImage`,
683/// `DeleteMicrovmImageVersion`, `GetMicrovmImageVersion`, `UpdateMicrovmImageVersion`,
684/// `GetMicrovmImageBuild`, the three listings, and `RunMicrovm.imageIdentifier`. Issue #24
685/// counted six implemented operations for each shape; every one of them is guarded here.
686///
687/// # An empty identifier is the case that pays for this
688///
689/// Ten of those members are **URI parameters**, and an empty one does not produce a validation
690/// error about a blank field: it collapses the path. `GET /2025-09-09/microvms/` addresses the
691/// *listing*, not a VM, so an empty id sent to `get_microvm` asks a different question and can
692/// get a 200 back for it — and `DELETE` on a collapsed path is worse. That failure mode is
693/// invisible in the service's answer, which is the strongest argument for a local refusal
694/// anywhere in this module.
695///
696/// # The model contradicts itself, and this is the side that can be guarded
697///
698/// `MicrovmImageArn` permits 2048 and is what the service *answers* with, so the model allows a
699/// legal response value that is an illegal request value — see
700/// [`crate::constants::MAX_IDENTIFIER_LEN`] for the whole account. The message says so, because
701/// a caller who hit it by echoing an ARN back is looking at a service inconsistency rather than
702/// at their own mistake, and telling them "shorten it" would be useless advice about a value
703/// they did not choose.
704pub fn require_valid_identifier(member: &str, identifier: &str) -> Result<(), Error> {
705 let model = crate::constants::MODEL_API_VERSION;
706 if identifier.is_empty() {
707 return Err(Error::invalid_arg(format!(
708 "{member} is empty, but the identifier shapes require at least 1 character (service \
709 model {model}). An empty identifier is not refused as a blank field — most of these \
710 are URI parameters, so it collapses the path onto the collection: \
711 `/microvms/` is the listing rather than one VM, and a request that addresses the \
712 wrong resource can succeed."
713 )));
714 }
715 if identifier.len() > crate::constants::MAX_IDENTIFIER_LEN {
716 return Err(Error::invalid_arg(format!(
717 "{member} is {} characters, over the {} the MicrovmIdentifier and \
718 MicrovmImageIdentifier shapes allow (service model {model}). Note the model \
719 disagrees with itself here: MicrovmImageArn permits {}, and that is the shape the \
720 service answers with on GetMicrovm's imageArn — so if this value came back from a \
721 response rather than being chosen here, it is a service-side inconsistency worth \
722 reporting rather than a length to shorten.",
723 identifier.len(),
724 crate::constants::MAX_IDENTIFIER_LEN,
725 crate::constants::MAX_IMAGE_ARN_LEN,
726 )));
727 }
728 Ok(())
729}
730
731/// Rejects a `RoleArn` the service would reject, before the artifact upload.
732///
733/// # Why the build role in particular
734///
735/// `CreateMicrovmImage.buildRoleArn` is sent on the create call, and the create call happens
736/// **after** the caller has uploaded the artifact. So the service's rejection of a malformed
737/// role ARN arrives having already cost the upload, which is the exact ordering
738/// [`ControlPlane::create_image`] exists to protect against and the reason issue #24 singled
739/// this member out. `RunMicrovm.executionRoleArn` is cheaper to get wrong but wrong in a
740/// confusing place: it is optional in the model, so a malformed one is a `ValidationException`
741/// on a member a caller may not know is being filled from their infra config.
742///
743/// # Three constraints, three messages
744///
745/// `min: 20`, `max: 2048`, and the pattern. They get separate messages because the pattern's
746/// wording ("this does not look like an IAM role ARN") is unhelpful for a value that is merely
747/// 3000 characters, and the length wording is unhelpful for a role *name* — which is the most
748/// common mistake and is short.
749///
750/// See [`crate::constants::is_valid_role_arn`] for what the structural check decides. It cannot
751/// tell you the role exists or grants the right log permissions; those are the failures that
752/// actually happen most and none of them is knowable without IAM.
753pub fn require_valid_role_arn(member: &str, arn: &str) -> Result<(), Error> {
754 let model = crate::constants::MODEL_API_VERSION;
755 if arn.len() < crate::constants::MIN_ROLE_ARN_LEN {
756 return Err(Error::invalid_arg(format!(
757 "{member} {arn:?} is {} characters, under the RoleArn minimum of {} (service model \
758 {model}). A value this short is almost always a role *name* where the ARN was \
759 wanted — `arn:aws:iam::<12-digit-account>:role/{arn}` is the shape.",
760 arn.len(),
761 crate::constants::MIN_ROLE_ARN_LEN,
762 )));
763 }
764 if arn.len() > crate::constants::MAX_ROLE_ARN_LEN {
765 return Err(Error::invalid_arg(format!(
766 "{member} is {} characters, over the RoleArn ceiling of {} (service model {model}).",
767 arn.len(),
768 crate::constants::MAX_ROLE_ARN_LEN,
769 )));
770 }
771 if !crate::constants::is_valid_role_arn(arn) {
772 return Err(Error::invalid_arg(format!(
773 "{member} {arn:?} does not match the RoleArn pattern {:?} (service model {model}). \
774 It wants `arn:aws:iam::` then **exactly twelve digits** of account id then `:role/` \
775 and a name — so the three things this catches are a role name passed as an ARN, an \
776 ARN for some other service, and an account id with a digit dropped. Rejected here \
777 rather than by AWS because for buildRoleArn the create call happens *after* the \
778 artifact upload, so the service's answer costs you the upload first. Whether the \
779 role exists and grants logs on {}/* is not checkable locally and is still the \
780 service's answer to give.",
781 crate::constants::ROLE_ARN_PATTERN,
782 image::BUILD_LOG_GROUP_PREFIX,
783 )));
784 }
785 Ok(())
786}
787
788/// Rejects a `logging.cloudWatch.logGroup` the service would reject, before the artifact
789/// upload.
790///
791/// The shape is `CloudWatchLoggingLogGroupString`: 1..=512, pattern `[a-zA-Z0-9_\-/.#]+`.
792/// The character the pattern excludes that a caller writes first is the colon — a group
793/// pasted as an ARN (`arn:aws:logs:...`) fails here with the pattern named, rather than as
794/// a `ValidationException` after the upload.
795pub fn require_valid_log_group(group: &str) -> Result<(), Error> {
796 let model = crate::constants::MODEL_API_VERSION;
797 if group.is_empty() {
798 return Err(Error::invalid_arg(format!(
799 "logging.cloudWatch.logGroup is empty, but the CloudWatchLoggingLogGroupString \
800 shape requires at least 1 character (service model {model}). Omit the setting \
801 entirely to take the service default — a service-created group under \
802 {}/<image-name>.",
803 image::BUILD_LOG_GROUP_PREFIX,
804 )));
805 }
806 if group.len() > crate::constants::MAX_LOG_GROUP_LEN {
807 return Err(Error::invalid_arg(format!(
808 "logging.cloudWatch.logGroup is {} characters, over the \
809 CloudWatchLoggingLogGroupString ceiling of {} (service model {model}).",
810 group.len(),
811 crate::constants::MAX_LOG_GROUP_LEN,
812 )));
813 }
814 if !crate::constants::is_valid_log_group(group) {
815 return Err(Error::invalid_arg(format!(
816 "logging.cloudWatch.logGroup {group:?} does not match the \
817 CloudWatchLoggingLogGroupString pattern {:?} (service model {model}) — letters, \
818 digits, and `_ - / . #` only. A colon usually means an ARN was pasted where the \
819 group *name* was wanted. Rejected here rather than by AWS because the create call \
820 happens after the artifact upload, so the service's answer costs you the upload \
821 first.",
822 crate::constants::LOG_GROUP_PATTERN,
823 )));
824 }
825 Ok(())
826}
827
828/// Rejects a caller-supplied `logging.cloudWatch.logStream` prefix the resolved name could
829/// not legally carry, before the artifact upload.
830///
831/// The shape is `CloudWatchLoggingLogStreamString`: 1..=512, pattern `[^:*]*` — anything
832/// but `:` and `*`. The caller's value is capped at
833/// [`crate::constants::MAX_USER_LOG_STREAM_LEN`] (495) rather than 512, because this
834/// client always appends `/<16 hex>` of per-build discriminator before the value reaches
835/// the wire — see [`CreateImageRequest::log_stream`] for why a verbatim stream name is
836/// never sent — and 495 + 17 is exactly the shape's ceiling.
837pub fn require_valid_log_stream(stream: &str) -> Result<(), Error> {
838 let model = crate::constants::MODEL_API_VERSION;
839 if stream.is_empty() {
840 return Err(Error::invalid_arg(format!(
841 "logging.cloudWatch.logStream is empty, but the CloudWatchLoggingLogStreamString \
842 shape requires at least 1 character (service model {model}). Omit the setting to \
843 let the service name the streams inside the configured group."
844 )));
845 }
846 if stream.len() > crate::constants::MAX_USER_LOG_STREAM_LEN {
847 return Err(Error::invalid_arg(format!(
848 "logging.cloudWatch.logStream is {} characters, over this client's ceiling of {}. \
849 The shape's own ceiling is {} (service model {model}), and the client always \
850 appends a 17-character per-build discriminator (`/` + 16 hex) — one image build \
851 is three VMs writing three streams, and the member is an exact stream name, so a \
852 fixed configured name would collapse every build's streams into one. The \
853 discriminator is what keeps them tellable apart, and it needs its room.",
854 stream.len(),
855 crate::constants::MAX_USER_LOG_STREAM_LEN,
856 crate::constants::MAX_LOG_STREAM_LEN,
857 )));
858 }
859 if let Some(found) = stream.chars().find(|c| matches!(c, ':' | '*')) {
860 return Err(Error::invalid_arg(format!(
861 "logging.cloudWatch.logStream {stream:?} contains {found:?}, which the \
862 CloudWatchLoggingLogStreamString pattern {:?} forbids anywhere in the value \
863 (service model {model}). A `*` usually means a prefix or glob was intended — the \
864 member is an exact stream name, prefixes are unsupported, and this client's own \
865 per-build suffix is how a configured name behaves like one.",
866 crate::constants::LOG_STREAM_PATTERN,
867 )));
868 }
869 Ok(())
870}
871
872/// Rejects a port the `PortNumber` or `HooksPortInteger` shapes would reject. **Minimum 1.**
873///
874/// # Port 0 is representable and means something else
875///
876/// Zero is what "let the kernel pick a port" means to a listener, so it is the value a caller
877/// reaches for when they do not have a port yet — and issue #24 measured that
878/// [`ControlPlane::with_port`]`(0)` was representable and produced `allowedPorts: [{"port": 0}]`
879/// on the wire, plus a `hooks.port` of 0 on every image built through that plane. Neither is a
880/// port the platform can forward to.
881///
882/// # There is no ceiling check here, and that is the type's doing rather than an omission
883///
884/// `PortNumber.max` and `HooksPortInteger.max` are both 65535, which is `u16::MAX` — so a `max`
885/// branch against the constant would be `port > u16::MAX`, a comparison that is false for every
886/// input. Clippy's `absurd_extreme_comparisons` says so, and it is right for the reason
887/// [`ControlPlane::create_image`] gives about the absent `hooks.port` check: a branch no input can
888/// reach reads as protection, appears in a coverage report as protection, and no test can make it
889/// fire.
890///
891/// The ceiling is still *pinned* — [`crate::constants::MAX_PORT`] is in the drift gate, and
892/// `the_port_floor_is_one_and_the_ceiling_is_what_a_u16_holds` asserts the equality with
893/// `u16::MAX` that makes this reasoning valid. If AWS ever lowers the ceiling below 65535 the gate
894/// goes red, that test goes red, and a real branch belongs here.
895pub fn require_valid_port(member: &str, port: u16) -> Result<(), Error> {
896 if port >= crate::constants::MIN_PORT {
897 return Ok(());
898 }
899 Err(Error::invalid_arg(format!(
900 "{member} is {port}, under the PortNumber minimum of {} (service model {}). Port 0 means \
901 'let the kernel choose' to a listener and is not a port the platform can forward to — a \
902 proxy token minted for it authorizes nothing, and an image whose hooks block names it has \
903 no reachable hook endpoint. Name the port the daemon actually listens on \
904 ({DEFAULT_AGENT_PORT} by default).",
905 crate::constants::MIN_PORT,
906 crate::constants::MODEL_API_VERSION,
907 )))
908}
909
910/// Rejects an `idlePolicy.maxIdleDurationSeconds` under the model's `min: 60`.
911///
912/// # The exemption this replaces was inherited from a client that no longer exists
913///
914/// [`RunMicrovmRequest::max_idle_sec`] and `constants.rs` both used to say there was
915/// deliberately no guard here, because `min` is one of the four keys in botocore's
916/// `VALIDATED_METADATA_ATTRS` and botocore therefore refuses it locally with a clear message.
917/// The premise is true and the conclusion does not apply: this client never touches botocore.
918/// `ControlPlane` signs with `aws-sigv4` and sends with `reqwest`, so `validate.py` is not on
919/// the path, and issue #24 measured `max_idle_sec: 59` reaching the wire.
920///
921/// It is the one place in this module where the *reason* for having no guard was wrong rather
922/// than the number, which is worth saying because the same reasoning would exempt every `min` in
923/// the model — `Version`'s, `NonBlankString`'s, the identifiers', `allowedPorts`'s. All of them
924/// are now guarded, and none of them by botocore.
925///
926/// No maximum, because the model states none. The bound that actually ends a VM's life is
927/// `maximumDurationInSeconds` ([`require_duration_in_range`]).
928pub fn require_idle_duration(seconds: u32) -> Result<(), Error> {
929 if seconds >= crate::constants::MIN_IDLE_DURATION_SEC {
930 return Ok(());
931 }
932 Err(Error::invalid_arg(format!(
933 "idlePolicy.maxIdleDurationSeconds={seconds} is under the minimum of {} (service model \
934 {}). This used to be documented as a constraint botocore enforced locally, which was \
935 true of the deleted Python client and never of this one — it signs with aws-sigv4 and \
936 sends with reqwest, so nothing validates a request before the wire except this crate.",
937 crate::constants::MIN_IDLE_DURATION_SEC,
938 crate::constants::MODEL_API_VERSION,
939 )))
940}
941
942/// Rejects a tag map the service would reject, naming the key and the half that failed.
943///
944/// # Sent since tags existed, checked by nothing
945///
946/// [`CreateImageRequest::tags`] goes straight onto `CreateMicrovmImage.tags` and was never
947/// validated (issue #24). Same ordering argument as everything else on that call: the rejection
948/// lands after the artifact upload.
949///
950/// # The key and the value have different rules, so the message says which
951///
952/// `TagKey` is `min: 1, max: 128`; `TagValue` is `min: 0, max: 256`. So an empty **value** is
953/// legal and an empty **key** is not, and the ceilings differ by 2x. They share one pattern.
954/// Every message here names the offending key, because a caller applying a dozen tags from a
955/// config file needs to know which one — a message that only said "a tag is invalid" would send
956/// them to read all twelve.
957///
958/// The key is quoted in the message even when the key itself is the problem, which is the
959/// deliberate choice: a key with a stray newline or a zero-width character in it is
960/// indistinguishable from a good one until it is printed with `{:?}`.
961pub fn require_valid_tags(tags: &std::collections::BTreeMap<String, String>) -> Result<(), Error> {
962 let model = crate::constants::MODEL_API_VERSION;
963 for (key, value) in tags {
964 if key.is_empty() {
965 return Err(Error::invalid_arg(format!(
966 "a tag key is empty, but TagKey requires at least 1 character (service model \
967 {model}). Note TagValue's minimum is 0, so an empty tag *value* is legal and an \
968 empty key is not."
969 )));
970 }
971 if key.len() > crate::constants::MAX_TAG_KEY_LEN {
972 return Err(Error::invalid_arg(format!(
973 "the tag key {key:?} is {} characters, over the TagKey ceiling of {} (service \
974 model {model}). TagValue allows {}, so the two halves are not \
975 interchangeable.",
976 key.len(),
977 crate::constants::MAX_TAG_KEY_LEN,
978 crate::constants::MAX_TAG_VALUE_LEN,
979 )));
980 }
981 if !crate::constants::is_valid_tag_component(key) {
982 return Err(Error::invalid_arg(format!(
983 "the tag key {key:?} has a character outside the TagKey pattern {:?} (service \
984 model {model}). Letters, digits, separators, and `_ . : / = + - @` — spaces and \
985 non-Latin scripts are fine, commas and `#` and `%` are not, and neither is a \
986 newline (it is a control character, not a separator).",
987 crate::constants::TAG_COMPONENT_PATTERN,
988 )));
989 }
990 if value.len() > crate::constants::MAX_TAG_VALUE_LEN {
991 return Err(Error::invalid_arg(format!(
992 "the value of tag {key:?} is {} characters, over the TagValue ceiling of {} \
993 (service model {model}).",
994 value.len(),
995 crate::constants::MAX_TAG_VALUE_LEN,
996 )));
997 }
998 if !crate::constants::is_valid_tag_component(value) {
999 return Err(Error::invalid_arg(format!(
1000 "the value of tag {key:?} has a character outside the TagValue pattern {:?} \
1001 (service model {model}). It is the same character set the key takes; the value \
1002 differs only in allowing an empty string.",
1003 crate::constants::TAG_COMPONENT_PATTERN,
1004 )));
1005 }
1006 }
1007 Ok(())
1008}
1009
1010/// Rejects an image name the service would reject, before the artifact upload.
1011///
1012/// The three cases get their own messages because the pattern message ("no dots, no
1013/// slashes") is actively misleading for a 70-character name that contains neither.
1014pub fn require_valid_image_name(name: &str) -> Result<(), Error> {
1015 let model = crate::constants::MODEL_API_VERSION;
1016 if name.is_empty() {
1017 return Err(Error::invalid_arg(format!(
1018 "the image name is empty, but ImageName requires at least 1 character (service \
1019 model {model})."
1020 )));
1021 }
1022 if name.len() > crate::constants::MAX_IMAGE_NAME_LEN {
1023 return Err(Error::invalid_arg(format!(
1024 "the image name is {} characters, over the ImageName ceiling of {} (service model \
1025 {model}). Rejected here rather than by AWS, because the create call happens \
1026 *after* the artifact upload — so the service's answer costs you the upload first.",
1027 name.len(),
1028 crate::constants::MAX_IMAGE_NAME_LEN,
1029 )));
1030 }
1031 if !crate::constants::is_valid_image_name(name) {
1032 return Err(Error::invalid_arg(format!(
1033 "the image name {name:?} does not match the ImageName pattern {:?} (service model \
1034 {model}). Letters, digits, hyphen, and underscore only — no dots and no slashes, \
1035 which are the two separators a namespaced name reaches for first.",
1036 crate::constants::IMAGE_NAME_PATTERN,
1037 )));
1038 }
1039 Ok(())
1040}
1041
1042/// The `Hooks` block with all six hooks enabled and both timeout families set.
1043///
1044/// All six because `ready` and `validate` are image-*build* hooks: the build calls them to
1045/// decide whether the snapshot it just produced is usable, before any instance exists and
1046/// therefore before any token has been delivered. Gating them on bootstrap state fails the
1047/// *build* rather than the run, which is a confusing place to discover the mistake.
1048///
1049/// The two timeouts are separate types, so the 60x gap cannot be crossed by passing one
1050/// number twice — see [`crate::hooks`].
1051/// `ENABLED` is [`ops::HookState::Enabled`] rather than a `&str` local, which is issue #24's
1052/// point about this function: the literal appeared six times with no constant naming either
1053/// value, so a typo in one of the six was a `ValidationException` on a create call made after
1054/// the artifact upload. It is now a compile error.
1055pub fn hooks_block(port: u16, run: RunHookTimeout, build: BuildHookTimeout) -> ops::Hooks {
1056 const ENABLED: ops::HookState = ops::HookState::Enabled;
1057 ops::Hooks {
1058 port,
1059 microvm_hooks: ops::MicrovmHooks {
1060 run: ENABLED,
1061 run_timeout_in_seconds: run.as_secs(),
1062 resume: ENABLED,
1063 resume_timeout_in_seconds: run.as_secs(),
1064 suspend: ENABLED,
1065 suspend_timeout_in_seconds: run.as_secs(),
1066 terminate: ENABLED,
1067 terminate_timeout_in_seconds: run.as_secs(),
1068 },
1069 microvm_image_hooks: ops::MicrovmImageHooks {
1070 ready: ENABLED,
1071 ready_timeout_in_seconds: build.as_secs(),
1072 validate: ENABLED,
1073 validate_timeout_in_seconds: build.as_secs(),
1074 },
1075 }
1076}
1077
1078/// A deadline elapsed on the client side. The remote resource is untouched.
1079pub(crate) fn timed_out(what: &str, waited: Duration) -> Error {
1080 Error::new(
1081 ErrorKind::Timeout,
1082 format!(
1083 "{what} within {:.0}s. This is a client-side deadline: nothing was cancelled, so \
1084 the resource is in whatever state the service last reported.",
1085 waited.as_secs_f64()
1086 ),
1087 )
1088}
1089
1090#[cfg(any(test, feature = "test-support"))]
1091pub(crate) mod fake;
1092
1093#[cfg(test)]
1094mod posture_fuzz;
1095
1096#[cfg(test)]
1097mod tests {
1098 use super::*;
1099
1100 /// The three image-name rejections each get their own message, because the pattern
1101 /// wording is misleading for a name that is merely long.
1102 #[test]
1103 fn each_image_name_rejection_explains_its_own_case() {
1104 let empty = require_valid_image_name("").expect_err("min is 1");
1105 assert!(
1106 empty.to_string().contains("at least 1 character"),
1107 "{empty}"
1108 );
1109
1110 let long = require_valid_image_name(&"a".repeat(65)).expect_err("max is 64");
1111 let message = long.to_string();
1112 assert!(message.contains("65 characters"), "{message}");
1113 assert!(
1114 message.contains("after* the artifact upload"),
1115 "the reason to check locally is the cost: {message}"
1116 );
1117 assert!(
1118 !message.contains("no dots"),
1119 "the pattern wording is misleading for a merely-long name: {message}"
1120 );
1121
1122 let dotted = require_valid_image_name("my.image").expect_err("dots are not in the pattern");
1123 let message = dotted.to_string();
1124 assert!(message.contains("no dots and no slashes"), "{message}");
1125 assert!(message.contains("[a-zA-Z0-9-_]+"), "{message}");
1126 }
1127
1128 /// The boundaries are inclusive on both sides, and every rejection is an `InvalidArg`
1129 /// so the CLI reports it as refused-locally rather than as a platform failure.
1130 #[test]
1131 fn an_image_name_at_the_ceiling_is_accepted_and_one_past_it_is_not() {
1132 require_valid_image_name(&"a".repeat(64)).expect("64 is the ceiling");
1133 require_valid_image_name("a").expect("1 is the minimum");
1134 for bad in ["", &"a".repeat(65), "my.image", "team/image", "a b"] {
1135 let error = require_valid_image_name(bad).expect_err("rejected");
1136 assert_eq!(error.kind(), ErrorKind::InvalidArg, "{bad:?}");
1137 }
1138 }
1139
1140 /// The duration range, at both boundaries. 28801 is the value the botocore gap was
1141 /// confirmed with, and the message has to say a second VM rather than a bigger number.
1142 #[test]
1143 fn the_duration_range_is_inclusive_and_names_the_eight_hour_ceiling() {
1144 assert_eq!(require_duration_in_range(1).expect("1 fits"), 1);
1145 assert_eq!(
1146 require_duration_in_range(28_800).expect("the ceiling fits"),
1147 28_800
1148 );
1149
1150 for bad in [0, 28_801, u32::MAX] {
1151 let error = require_duration_in_range(bad).expect_err("outside 1..28800");
1152 assert_eq!(error.kind(), ErrorKind::InvalidArg, "{bad}");
1153 let message = error.to_string();
1154 assert!(message.contains("eight hours"), "{message}");
1155 assert!(
1156 message.contains("a second VM, not a larger number"),
1157 "{message}"
1158 );
1159 }
1160 }
1161
1162 /// The hooks block enables all six hooks, and the two families carry their own
1163 /// timeouts — which is the whole reason they are two types.
1164 #[test]
1165 fn the_hooks_block_enables_all_six_with_the_two_families_kept_apart() {
1166 let hooks = hooks_block(
1167 9000,
1168 RunHookTimeout::try_new(45).expect("45s is a legal run timeout"),
1169 BuildHookTimeout::try_new(1800).expect("1800s is a legal build timeout"),
1170 );
1171
1172 assert_eq!(hooks.port, 9000);
1173 for state in [
1174 hooks.microvm_hooks.run,
1175 hooks.microvm_hooks.resume,
1176 hooks.microvm_hooks.suspend,
1177 hooks.microvm_hooks.terminate,
1178 hooks.microvm_image_hooks.ready,
1179 hooks.microvm_image_hooks.validate,
1180 ] {
1181 assert_eq!(state, ops::HookState::Enabled, "all six hooks are served");
1182 // The wire spelling too, against a literal: the variant name and the string the
1183 // service reads are two different things, and `HookState::Enabled` rendering as
1184 // anything but `ENABLED` is a rejected create on six fields at once.
1185 assert_eq!(state.as_str(), "ENABLED");
1186 }
1187
1188 assert_eq!(hooks.microvm_hooks.run_timeout_in_seconds, 45);
1189 assert_eq!(hooks.microvm_hooks.terminate_timeout_in_seconds, 45);
1190 assert_eq!(hooks.microvm_image_hooks.ready_timeout_in_seconds, 1800);
1191 assert_eq!(
1192 hooks.microvm_image_hooks.validate_timeout_in_seconds, 1800,
1193 "1800 is legal for the build family and illegal for the run one, which is why \
1194 one shared number cannot serve both"
1195 );
1196 }
1197
1198 /// The default request carries the measured defaults, and identity repair is **off** —
1199 /// a capability granted by default is a capability nobody asked for.
1200 #[test]
1201 fn the_default_create_request_repairs_no_identity_and_takes_the_default_size() {
1202 let request = CreateImageRequest::new("img", vec![1, 2, 3], "s3://b/k", "arn:role");
1203 assert!(!request.repair_guest_identity);
1204 assert!(!request.inherit_workdir);
1205 assert_eq!(request.size, SizeClass::DEFAULT);
1206 assert_eq!(request.base_image, BaseImage::al2023());
1207 assert_eq!(request.token_scope, None);
1208 assert_eq!(request.dockerfile, None);
1209 }
1210
1211 /// The default launch requests ingress and **not** egress: the right default for a
1212 /// daemon that needs none. What the omission buys is measured in `docs/PLATFORM.md`
1213 /// (2026-09-12: the platform gave the connector-less VM outbound network anyway).
1214 #[test]
1215 fn the_default_launch_requests_ingress_only() {
1216 let payload = RunHookPayload::for_agent_token("token").expect("a token fits");
1217 let request = RunMicrovmRequest::new("arn:image", payload);
1218 assert_eq!(request.connectors, vec![ConnectorIntent::AllIngress]);
1219
1220 let with_egress = request.with_egress();
1221 assert_eq!(
1222 with_egress.connectors,
1223 vec![ConnectorIntent::AllIngress, ConnectorIntent::Egress]
1224 );
1225 }
1226
1227 /// Asking for egress twice does not send it twice. `NetworkConnectorList` caps at 10
1228 /// and a duplicate is a wasted slot at best.
1229 #[test]
1230 fn asking_for_egress_twice_adds_it_once() {
1231 let payload = RunHookPayload::for_agent_token("token").expect("a token fits");
1232 let request = RunMicrovmRequest::new("arn:image", payload)
1233 .with_egress()
1234 .with_egress();
1235 assert_eq!(request.connectors.len(), 2);
1236 }
1237
1238 /// `with_shell` **replaces** `ALL_INGRESS` with the measured pair rather than adding
1239 /// beside it: the platform accepts `ALL_INGRESS` + finer ingress at launch and
1240 /// refuses it only when a shell token is minted, after the VM has run and billed —
1241 /// and `run_microvm` refuses the combination client-side, so a `with_shell` that
1242 /// merely appended would build a request this client's own validation rejects.
1243 ///
1244 /// **Falsification** — watched fail 2026-08-31: dropping the `retain` from
1245 /// `with_shell` fails the first assertion with `ALL_INGRESS` still in the set.
1246 #[test]
1247 fn with_shell_replaces_all_ingress_with_the_measured_pair() {
1248 let payload = RunHookPayload::for_agent_token("token").expect("a token fits");
1249 let request = RunMicrovmRequest::new("arn:image", payload).with_shell();
1250 assert_eq!(
1251 request.connectors,
1252 vec![ConnectorIntent::HttpIngress, ConnectorIntent::ShellIngress]
1253 );
1254
1255 // Egress is a separate question and survives in either order.
1256 let payload = RunHookPayload::for_agent_token("token").expect("a token fits");
1257 let both = RunMicrovmRequest::new("arn:image", payload)
1258 .with_egress()
1259 .with_shell()
1260 .with_shell();
1261 assert_eq!(
1262 both.connectors,
1263 vec![
1264 ConnectorIntent::Egress,
1265 ConnectorIntent::HttpIngress,
1266 ConnectorIntent::ShellIngress
1267 ],
1268 "egress survives, and asking twice adds nothing"
1269 );
1270 }
1271
1272 /// A client-side deadline is an `ErrorKind::Timeout` and says the resource was not
1273 /// touched — because the caller's next question is whether to clean something up.
1274 #[test]
1275 fn a_client_side_deadline_says_the_resource_is_untouched() {
1276 let error = timed_out("the image did not become usable", Duration::from_secs(300));
1277 assert_eq!(error.kind(), ErrorKind::Timeout);
1278 let message = error.to_string();
1279 assert!(message.contains("300s"), "{message}");
1280 assert!(message.contains("nothing was cancelled"), "{message}");
1281 }
1282
1283 /// The default port matches the daemon's, and a caller can change it.
1284 #[test]
1285 fn the_agent_port_defaults_to_nine_thousand_and_is_overridable() {
1286 let plane = crate::testing::control_plane(
1287 Arc::new(fake::FakeControlPlane::new()),
1288 Region::UsEast1,
1289 Arc::new(fake::TestClock::new()),
1290 );
1291 assert_eq!(plane.port(), DEFAULT_AGENT_PORT);
1292 assert_eq!(plane.port(), 9000);
1293 assert_eq!(
1294 plane.with_port(8080).expect("8080 is a legal port").port(),
1295 8080
1296 );
1297 }
1298
1299 /// **TRAP-1, the compile surface.** Image requests cannot carry
1300 /// a caller-supplied token.
1301 ///
1302 /// Asserted by destructuring rather than by grep: a struct pattern must name every
1303 /// field, so an image `client_token` added later fails to compile this test. That is the
1304 /// closest a test can get to asserting an absence.
1305 #[test]
1306 fn image_requests_forbid_tokens_and_run_requests_require_explicit_opt_in() {
1307 let CreateImageRequest {
1308 name: _,
1309 base_image: _,
1310 base_image_version: _,
1311 dockerfile: _,
1312 binary: _,
1313 // Dependency-file *bytes* under ecosystem-fixed entry names — nothing
1314 // token-shaped, and no field a caller could name an arbitrary file into.
1315 project_files: _,
1316 code_artifact_uri: _,
1317 build_role_arn: _,
1318 size: _,
1319 repair_guest_identity: _,
1320 inherit_workdir: _,
1321 run_hook_timeout: _,
1322 build_hook_timeout: _,
1323 tags: _,
1324 // The logging pair. The stream is a *prefix* the create call suffixes with a
1325 // fresh nonce — the caller cannot name the exact wire stream, which is the
1326 // same closure shape as the token's.
1327 log_group: _,
1328 log_stream: _,
1329 // A *label*, folded in beside a fresh nonce. The only token-adjacent field,
1330 // and it cannot become the token.
1331 token_scope: _,
1332 } = CreateImageRequest::new("img", Vec::new(), "s3://b/k", "arn:role");
1333
1334 let RunMicrovmRequest {
1335 image_identifier: _,
1336 image_version: _,
1337 execution_role_arn: _,
1338 connectors: _,
1339 egress_network_connectors: _,
1340 run_hook_payload: _,
1341 max_duration_sec: _,
1342 max_idle_sec: _,
1343 suspended_sec: _,
1344 auto_resume: _,
1345 token_scope: _,
1346 // A per-VM log destination: a group name and an exact stream, never a token.
1347 logging: _,
1348 client_token,
1349 } = RunMicrovmRequest::new(
1350 "arn:image",
1351 RunHookPayload::for_agent_token("t").expect("fits"),
1352 );
1353 assert!(client_token.is_none());
1354 }
1355
1356 /// **TRAP-1's last mile.** The wire types' `client_token` fields are `pub(crate)`, so a
1357 /// caller outside this crate cannot build a request body with a token of their own.
1358 ///
1359 /// Worth its own test because the hole was real and I opened it: `ops` is a public module
1360 /// and the wire structs are public, so `pub client_token` on them meant an external
1361 /// caller could bypass `CreateImageRequest` entirely, hand-build a
1362 /// `CreateMicrovmImageWire` with a content digest, and serialize it — the wedge with one
1363 /// extra step, past a request type that has no such field.
1364 ///
1365 /// This test can only observe the *inside* of the boundary, since a compile failure
1366 /// outside the crate is not something an inline test can assert. What it pins is that the
1367 /// field is reachable from here and that the value in it came from the minter — so a
1368 /// later widening back to `pub` is at least a visible diff against a test that documents
1369 /// why it is narrow. The external half is enforced by the compiler.
1370 #[test]
1371 fn the_wire_types_token_field_is_crate_private_and_minted() {
1372 let entropy = crate::entropy::testing::SequenceEntropy::new();
1373 let wire = ops::RunMicrovmWire {
1374 image_identifier: "arn:image".to_string(),
1375 image_version: None,
1376 execution_role_arn: None,
1377 ingress_network_connectors: Vec::new(),
1378 egress_network_connectors: None,
1379 idle_policy: ops::IdlePolicy {
1380 max_idle_duration_seconds: 600,
1381 suspended_duration_seconds: 600,
1382 auto_resume_enabled: false,
1383 },
1384 maximum_duration_in_seconds: 3_600,
1385 run_hook_payload: String::new(),
1386 logging: None,
1387 // Reachable from inside the crate, and only ever populated this way.
1388 client_token: token::run_token_with("arn:image", &entropy).expect("scripted"),
1389 };
1390 assert!(wire.client_token.starts_with("run-"));
1391 assert!(wire.client_token.len() <= crate::constants::MAX_CLIENT_TOKEN_LEN);
1392 assert_ne!(
1393 wire.client_token,
1394 token::run_token_with("arn:image", &entropy).expect("scripted"),
1395 "two mints of the same scope differ, which is the property the narrowing protects"
1396 );
1397 }
1398
1399 // ── issue #24's guards, at the message level ─────────────────────────────
1400 //
1401 // Each of these asserts the *refusal and its wording*; the zero-call proofs live beside
1402 // the operations they protect, in `image.rs` and `microvm.rs`, because only there can a
1403 // test observe that no control-plane call happened.
1404
1405 /// The three `NonBlankString` rejections each explain their own case, and the pattern one
1406 /// names the character it found.
1407 ///
1408 /// The whitespace message has to name the character, because the whole class of failure this
1409 /// catches is a value that *looks* right: a URI or a filter pasted out of a console carries a
1410 /// trailing newline, satisfies "non-empty", and fails the pattern.
1411 ///
1412 /// **Guard proof.** Delete the `is_empty` branch and the blank case returns `Ok`; delete the
1413 /// whitespace branch and the `"s3://b/k\n"` case does.
1414 #[test]
1415 fn each_non_blank_rejection_explains_its_own_case() {
1416 let blank = require_non_blank("codeArtifact.uri", "").expect_err("min is 1");
1417 assert_eq!(blank.kind(), ErrorKind::InvalidArg);
1418 assert!(
1419 blank.to_string().contains("at least 1 character"),
1420 "{blank}"
1421 );
1422 assert!(
1423 blank.to_string().contains("NonBlankString"),
1424 "the shape has to be named: {blank}"
1425 );
1426
1427 let long = require_non_blank("baseImageArn", &"a".repeat(2049)).expect_err("max is 2048");
1428 let message = long.to_string();
1429 assert!(message.contains("2049 characters"), "{message}");
1430 assert!(message.contains("ceiling of 2048"), "{message}");
1431
1432 let newline = require_non_blank("codeArtifact.uri", "s3://bucket/key.zip\n")
1433 .expect_err("whitespace anywhere is refused");
1434 let message = newline.to_string();
1435 assert!(
1436 message.contains(r"'\n'"),
1437 "the message must name the character it found: {message}"
1438 );
1439 assert!(message.contains("trailing newline"), "{message}");
1440
1441 // Whitespace in the middle, not only at the end: `[^\s]+` forbids it anywhere.
1442 let inner =
1443 require_non_blank("nameFilter", "coding agents").expect_err("an inner space too");
1444 assert!(
1445 inner.to_string().contains("anywhere in the value"),
1446 "{inner}"
1447 );
1448
1449 // And the legal values pass, so the guard is not refusing everything.
1450 require_non_blank("codeArtifact.uri", "s3://bucket/agentd.zip").expect("a real URI");
1451 require_non_blank(
1452 "baseImageArn",
1453 "arn:aws:lambda:us-east-1:aws:microvm-image:al2023-1",
1454 )
1455 .expect("a real base ARN");
1456 require_non_blank("nameFilter", "coding-agents").expect("a real filter");
1457 require_non_blank("imageVersion", &"9".repeat(2048)).expect("2048 is the ceiling");
1458 }
1459
1460 /// The identifier guard's two cases, and the over-long one names the model's own
1461 /// contradiction rather than telling the caller to shorten a value they may not have chosen.
1462 ///
1463 /// The empty message is the one that carries the reason this guard exists at all: ten of the
1464 /// twelve members are URI parameters, so an empty one collapses the path onto the collection
1465 /// and addresses the wrong resource — which the service can answer 200 to.
1466 ///
1467 /// **Guard proof.** Delete the `is_empty` branch and the blank case returns `Ok`; drop the
1468 /// `MicrovmImageArn` clause and the assertion on `2048` goes red.
1469 #[test]
1470 fn the_identifier_guard_names_the_collapsed_path_and_the_models_contradiction() {
1471 let blank = require_valid_identifier("microvmIdentifier", "").expect_err("min is 1");
1472 assert_eq!(blank.kind(), ErrorKind::InvalidArg);
1473 let message = blank.to_string();
1474 assert!(message.contains("at least 1 character"), "{message}");
1475 assert!(
1476 message.contains("collapses the path"),
1477 "the reason an empty URI parameter is dangerous has to be in the message: {message}"
1478 );
1479
1480 let long =
1481 require_valid_identifier("imageIdentifier", &"a".repeat(257)).expect_err("max is 256");
1482 let message = long.to_string();
1483 assert!(message.contains("257 characters"), "{message}");
1484 assert!(message.contains("256"), "{message}");
1485 assert!(
1486 message.contains("MicrovmImageArn permits 2048"),
1487 "the model's own contradiction has to be named, or the advice is 'shorten a value \
1488 the service gave you': {message}"
1489 );
1490
1491 require_valid_identifier(
1492 "imageIdentifier",
1493 "arn:aws:lambda:us-east-1:1:microvm-image:i",
1494 )
1495 .expect("a real ARN");
1496 require_valid_identifier("microvmIdentifier", "mvm-abc123").expect("a real id");
1497 require_valid_identifier("imageIdentifier", &"a".repeat(256)).expect("256 is the ceiling");
1498 }
1499
1500 /// The role-ARN guard's three cases, and the short one guesses the actual mistake.
1501 ///
1502 /// A value under 20 characters is almost always a role *name*, so the message renders the ARN
1503 /// the caller probably meant rather than restating the bound. That is the difference between
1504 /// a diagnostic and a validation error.
1505 ///
1506 /// **Guard proof.** Delete the `is_valid_role_arn` branch and the eleven-digit case returns
1507 /// `Ok` — which is the case nothing else catches.
1508 #[test]
1509 fn the_role_arn_guard_guesses_the_mistake_a_short_value_is() {
1510 let name = require_valid_role_arn("buildRoleArn", "build-role").expect_err("min is 20");
1511 assert_eq!(name.kind(), ErrorKind::InvalidArg);
1512 let message = name.to_string();
1513 assert!(message.contains("role *name*"), "{message}");
1514 assert!(
1515 message.contains("arn:aws:iam::<12-digit-account>:role/build-role"),
1516 "the message renders the ARN they probably meant: {message}"
1517 );
1518
1519 let long = require_valid_role_arn(
1520 "executionRoleArn",
1521 &format!("arn:aws:iam::123456789012:role/{}", "a".repeat(2048)),
1522 )
1523 .expect_err("max is 2048");
1524 assert!(long.to_string().contains("ceiling of 2048"), "{long}");
1525
1526 // Eleven digits, which is the case no eyeball and no other check catches.
1527 let eleven = require_valid_role_arn("buildRoleArn", "arn:aws:iam::12345678901:role/build")
1528 .expect_err("the account id is eleven digits");
1529 let message = eleven.to_string();
1530 assert!(
1531 message.contains("exactly twelve digits"),
1532 "the digit count is the whole value of this check: {message}"
1533 );
1534 assert!(
1535 message.contains("after* the artifact upload"),
1536 "the reason to check locally is the cost: {message}"
1537 );
1538 assert!(
1539 message.contains("/aws/lambda-microvms/*"),
1540 "and the message says what is still the service's answer to give: {message}"
1541 );
1542
1543 require_valid_role_arn(
1544 "buildRoleArn",
1545 "arn:aws:iam::392583147479:role/bonk-sandbox-microvm-build",
1546 )
1547 .expect("the conformance account's real build role");
1548 }
1549
1550 /// The port guard refuses 0 and says what 0 means, because 0 is not a typo.
1551 ///
1552 /// The ceiling branch cannot fire from a `u16` call site — `MAX_PORT` is `u16::MAX` — so what
1553 /// is asserted here is the floor and the two legal boundaries.
1554 ///
1555 /// **Guard proof.** Change the comparison to `port < 0` (or delete the branch) and the zero
1556 /// case returns `Ok`.
1557 #[test]
1558 fn the_port_guard_refuses_zero_and_says_what_zero_means() {
1559 let zero = require_valid_port("allowedPorts[].port", 0).expect_err("min is 1");
1560 assert_eq!(zero.kind(), ErrorKind::InvalidArg);
1561 let message = zero.to_string();
1562 assert!(
1563 message.contains("let the kernel choose"),
1564 "0 is a value a caller passes on purpose, so the message has to say why it is not a \
1565 port: {message}"
1566 );
1567 assert!(
1568 message.contains("authorizes nothing"),
1569 "the proxy-token consequence: {message}"
1570 );
1571 assert!(message.contains("9000"), "the default is named: {message}");
1572
1573 require_valid_port("port", 1).expect("1 is the minimum");
1574 require_valid_port("port", 9000).expect("the agent port");
1575 require_valid_port("port", 65_535).expect("65535 is the ceiling");
1576 }
1577
1578 /// The idle-duration guard refuses 59 and says the botocore reasoning did not transfer.
1579 ///
1580 /// The message names the wrong premise on purpose. The exemption was not a wrong number, it
1581 /// was a correct fact about a client that no longer exists — and a reader who deletes this
1582 /// guard will do it by reasoning that `min` is validated locally, which is exactly the
1583 /// sentence the message answers.
1584 ///
1585 /// **Guard proof.** Delete the call in `run_microvm` and the zero-call test in `microvm.rs`
1586 /// goes red; invert the comparison here and 600 is refused while 59 passes.
1587 #[test]
1588 fn the_idle_guard_refuses_fifty_nine_and_names_the_premise_that_did_not_transfer() {
1589 let under = require_idle_duration(59).expect_err("min is 60");
1590 assert_eq!(under.kind(), ErrorKind::InvalidArg);
1591 let message = under.to_string();
1592 assert!(message.contains("59"), "{message}");
1593 assert!(message.contains("minimum of 60"), "{message}");
1594 assert!(
1595 message.contains("botocore"),
1596 "the message has to name the premise, because the premise is what was wrong: \
1597 {message}"
1598 );
1599 assert!(
1600 message.contains("aws-sigv4"),
1601 "and say what this client actually does instead: {message}"
1602 );
1603
1604 require_idle_duration(60).expect("60 is the minimum");
1605 require_idle_duration(600).expect("the default");
1606 // No maximum in the model, and the client adds none.
1607 require_idle_duration(u32::MAX).expect("the model states no ceiling");
1608 }
1609
1610 /// Every tag rejection names the offending key, and the key and the value keep their
1611 /// different rules.
1612 ///
1613 /// The two asymmetries are the content of this test: an empty value is legal and an empty key
1614 /// is not, and the ceilings are 128 against 256. A guard that used one rule for both would
1615 /// pass a 200-character key or refuse a 200-character value, and neither is what the service
1616 /// does.
1617 ///
1618 /// **Guard proof.** Use `MAX_TAG_KEY_LEN` for the value branch and the 256-character-value
1619 /// case fails; drop the `key` from any message and the assertion naming it goes red.
1620 #[test]
1621 fn every_tag_rejection_names_the_key_and_the_two_halves_keep_their_own_rules() {
1622 use std::collections::BTreeMap;
1623
1624 let one = |key: &str, value: &str| {
1625 let mut tags = BTreeMap::new();
1626 tags.insert(key.to_string(), value.to_string());
1627 tags
1628 };
1629
1630 // An empty **value** is legal. TagValue's min is 0.
1631 require_valid_tags(&one("owner", "")).expect("an empty tag value is legal");
1632 // An empty **key** is not. TagKey's min is 1.
1633 let blank = require_valid_tags(&one("", "conformance")).expect_err("TagKey min is 1");
1634 assert_eq!(blank.kind(), ErrorKind::InvalidArg);
1635 assert!(
1636 blank.to_string().contains("TagValue's minimum is 0"),
1637 "the asymmetry is the thing to say: {blank}"
1638 );
1639
1640 // The two ceilings differ by 2x, and a 256-character *value* is legal.
1641 require_valid_tags(&one("owner", &"v".repeat(256))).expect("TagValue max is 256");
1642 let long_value = require_valid_tags(&one("owner", &"v".repeat(257)))
1643 .expect_err("257 is one past TagValue");
1644 assert!(long_value.to_string().contains("\"owner\""), "{long_value}");
1645 assert!(
1646 long_value.to_string().contains("257 characters"),
1647 "{long_value}"
1648 );
1649
1650 require_valid_tags(&one(&"k".repeat(128), "v")).expect("TagKey max is 128");
1651 let long_key =
1652 require_valid_tags(&one(&"k".repeat(129), "v")).expect_err("129 is one past TagKey");
1653 let message = long_key.to_string();
1654 assert!(message.contains("129 characters"), "{message}");
1655 assert!(
1656 message.contains("TagValue allows 256"),
1657 "the message says the two halves are not interchangeable: {message}"
1658 );
1659
1660 // The pattern, both halves, with the key named either way.
1661 let bad_key = require_valid_tags(&one("cost,centre", "x")).expect_err("a comma");
1662 let message = bad_key.to_string();
1663 assert!(message.contains("\"cost,centre\""), "{message}");
1664 assert!(
1665 message.contains("commas and `#` and `%` are not"),
1666 "{message}"
1667 );
1668
1669 let bad_value = require_valid_tags(&one("owner", "50%")).expect_err("a percent");
1670 let message = bad_value.to_string();
1671 assert!(
1672 message.contains("the value of tag \"owner\""),
1673 "the message says which half as well as which key: {message}"
1674 );
1675
1676 // A newline in a key: legal-looking, and the reason keys are printed with `{:?}`.
1677 let newline = require_valid_tags(&one("owner\n", "x")).expect_err("a control character");
1678 assert!(
1679 newline.to_string().contains(r#""owner\n""#),
1680 "a key with a stray newline is indistinguishable from a good one until it is quoted: \
1681 {newline}"
1682 );
1683
1684 // And a realistic tag set passes, including a space and a non-Latin key.
1685 let mut real = BTreeMap::new();
1686 real.insert("owner".to_string(), "conformance".to_string());
1687 real.insert("cost centre".to_string(), "team/agents".to_string());
1688 real.insert("所有者".to_string(), "らいす".to_string());
1689 real.insert("empty".to_string(), String::new());
1690 require_valid_tags(&real).expect("a realistic tag set");
1691
1692 // An empty map is not a tag problem, and `create_image` omits the member entirely.
1693 require_valid_tags(&BTreeMap::new()).expect("no tags is not an invalid tag");
1694 }
1695
1696 /// **Issue #24's `HookState` half.** The two values are named by a type, so neither can be
1697 /// misspelled, and the block this client sends is six `Enabled`s.
1698 ///
1699 /// The wire spellings are asserted against literals rather than derived from the variant
1700 /// names, because the variant name and the string the service reads are two different things
1701 /// — the same reason `ops::VersionStatus` does it that way.
1702 ///
1703 /// **Guard proof.** Change `HookState::Enabled`'s rename to `"ENABLE"` and this fails on the
1704 /// serialization assertion; add a third variant and the `HOOK_STATES` comparison fails.
1705 #[test]
1706 fn the_hook_state_enum_is_the_models_two_values_and_nothing_else() {
1707 assert_eq!(ops::HookState::Disabled.as_str(), "DISABLED");
1708 assert_eq!(ops::HookState::Enabled.as_str(), "ENABLED");
1709 assert_eq!(ops::HookState::Enabled.to_string(), "ENABLED");
1710
1711 // Through serde, which is what actually reaches the wire — `as_str` could be right while
1712 // the `rename` is wrong.
1713 assert_eq!(
1714 serde_json::to_value(ops::HookState::Enabled).expect("serialises"),
1715 serde_json::json!("ENABLED")
1716 );
1717 assert_eq!(
1718 serde_json::to_value(ops::HookState::Disabled).expect("serialises"),
1719 serde_json::json!("DISABLED")
1720 );
1721
1722 // The typed spelling and the drift gate's array agree, or one of them is checking
1723 // something the other does not send.
1724 let spelled = [ops::HookState::Disabled, ops::HookState::Enabled];
1725 let mut wire: Vec<&str> = spelled.iter().map(|state| state.as_str()).collect();
1726 wire.sort_unstable();
1727 let mut pinned = crate::constants::HOOK_STATES.to_vec();
1728 pinned.sort_unstable();
1729 assert_eq!(wire, pinned);
1730
1731 // And the misspellings a `String` field allowed are now parse failures on the way in,
1732 // which is the closest an inline test gets to asserting they are compile errors on the
1733 // way out.
1734 for typo in ["\"ENABLE\"", "\"Enabled\"", "\"ACTIVE\"", "\"enabled\""] {
1735 serde_json::from_str::<ops::HookState>(typo)
1736 .expect_err(&format!("{typo} is not a HookState"));
1737 }
1738 }
1739}