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

#![allow(non_snake_case, non_camel_case_types)]

extern crate qapi_spec as qapi;
extern crate serde;
#[macro_use]
extern crate serde_derive;

include!(concat!(env!("OUT_DIR"), "/qga.rs"));

use std::{io, str, fmt, error};

#[derive(Copy, Clone, Debug, Deserialize, Serialize)]
#[serde(rename_all = "kebab-case")]
pub enum GuestShutdownMode {
    Halt,
    Powerdown,
    Reboot,
}

impl GuestExecStatus {
    pub fn result(self) -> Result<Self, Self> {
        if self.exited {
            if self.exitcode != Some(0) || self.signal.is_some() {
                Err(self)
            } else {
                Ok(self)
            }
        } else {
            Ok(self)
        }
    }

    fn message(&self) -> String {
        let (err0, err1) = if let Some(Ok(data)) = self.err_data.as_ref().map(|s| str::from_utf8(s)) {
            (": ", data)
        } else {
            ("", "")
        };

        let sig = if let Some(signal) = self.signal {
            format!(" (terminated by signal {})", signal)
        } else {
            Default::default()
        };

        if let Some(code) = self.exitcode {
            format!("guest process exited with code {}{}{}{}", code, sig, err0, err1)
        } else if self.exited {
            format!("guest process exited{}{}{}", sig, err0, err1)
        } else {
            format!("guest process is still running")
        }
    }
}

impl fmt::Display for GuestExecStatus {
    fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
        write!(f, "{}", self.message())
    }
}

impl error::Error for GuestExecStatus {
    fn description(&self) -> &str {
        "guest process exit status"
    }
}

impl From<GuestExecStatus> for io::Error {
    fn from(s: GuestExecStatus) -> Self {
        io::Error::new(io::ErrorKind::Other, s.to_string())
    }
}