firepilot 1.2.0

Interface to work with Firecracker microVMs
Documentation 
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

use crate::builder::{Builder, BuilderError};
use firepilot_models::models::BootSource;

use super::assert_not_none;

#[derive(Debug)]
pub struct KernelBuilder {
    pub boot_args: Option<String>,
    pub initrd_path: Option<String>,
    pub kernel_image_path: Option<String>,
}

impl KernelBuilder {
    pub fn new() -> KernelBuilder {
        KernelBuilder {
            boot_args: None,
            initrd_path: None,
            kernel_image_path: None,
        }
    }

    pub fn with_boot_args(mut self, boot_args: String) -> KernelBuilder {
        self.boot_args = Some(boot_args);
        self
    }

    pub fn with_initrd_path(mut self, initrd_path: String) -> KernelBuilder {
        self.initrd_path = Some(initrd_path);
        self
    }

    pub fn with_kernel_image_path(mut self, kernel_image_path: String) -> KernelBuilder {
        self.kernel_image_path = Some(kernel_image_path);
        self
    }
}

impl Builder<BootSource> for KernelBuilder {
    fn try_build(self) -> Result<BootSource, BuilderError> {
        assert_not_none(stringify!(self.kernel_image_path), &self.kernel_image_path)?;
        Ok(BootSource {
            kernel_image_path: self.kernel_image_path.unwrap(),
            initrd_path: self.initrd_path,
            boot_args: self.boot_args,
        })
    }
}

#[cfg(test)]
mod tests {
    use crate::builder::kernel::KernelBuilder;
    use crate::builder::Builder;

    #[test]
    fn full_kernel() {
        KernelBuilder::new()
            .with_kernel_image_path("path/to/kernel".to_string())
            .with_initrd_path("path/to/initrd".to_string())
            .with_boot_args("console=ttyS0 reboot=k panic=1 pci=off".to_string())
            .try_build()
            .unwrap();
    }

    #[test]
    #[should_panic]
    fn partial_kernel() {
        KernelBuilder::new()
            .with_initrd_path("path/to/initrd".to_string())
            .try_build()
            .unwrap();
    }
}