microsandbox-runtime 0.7.0

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

//! Startup workload execution for detached `msb run -- CMD`.

use std::path::Path;

use microsandbox_agent_client::{AgentClient, TypedMessage};
use microsandbox_protocol::{
    exec::{ExecExited, ExecFailed, ExecRequest},
    message::MessageType,
};

use crate::vm::StartupCommand;
use crate::{RuntimeError, RuntimeResult};

//--------------------------------------------------------------------------------------------------
// Types
//--------------------------------------------------------------------------------------------------

/// Terminal status for the startup workload exec session.
pub(crate) enum StartupCommandExit {
    /// The command ran and exited with the contained status code.
    Exited(i32),

    /// agentd could not spawn the command.
    Failed(ExecFailed),
}

//--------------------------------------------------------------------------------------------------
// Functions
//--------------------------------------------------------------------------------------------------

pub(crate) async fn run_startup_command(
    agent_sock_path: &Path,
    command: StartupCommand,
) -> RuntimeResult<StartupCommandExit> {
    let client = AgentClient::connect(agent_sock_path)
        .await
        .map_err(|err| RuntimeError::Custom(format!("startup command connect: {err}")))?;

    let request = ExecRequest {
        cmd: command.cmd,
        args: command.args,
        env: command.env,
        cwd: command.cwd,
        user: command.user,
        tty: false,
        rows: 24,
        cols: 80,
        rlimits: Vec::new(),
    };

    let mut rx = client
        .stream(TypedMessage::new(MessageType::ExecRequest, &request))
        .await
        .map_err(|err| RuntimeError::Custom(format!("startup command dispatch: {err}")))?;

    while let Some(message) = rx
        .recv()
        .await
        .map_err(|err| RuntimeError::Custom(format!("startup command stream: {err}")))?
    {
        match MessageType::from_wire_str(&message.t) {
            Some(MessageType::ExecExited) => {
                let exited = message
                    .payload::<ExecExited>()
                    .map_err(|err| RuntimeError::Custom(format!("startup command exit: {err}")))?;
                return Ok(StartupCommandExit::Exited(exited.code));
            }
            Some(MessageType::ExecFailed) => {
                let failed = message.payload::<ExecFailed>().map_err(|err| {
                    RuntimeError::Custom(format!("startup command failure: {err}"))
                })?;
                return Ok(StartupCommandExit::Failed(failed));
            }
            _ => {}
        }
    }

    Err(RuntimeError::Custom(
        "startup command stream ended before terminal event".into(),
    ))
}