krata/xenclient/src/create.rs

use std::collections::HashMap;

pub struct DomainConfig {
    pub max_vcpus: u32,
    pub mem_mb: u64,
    pub kernel_path: String,
    pub initrd_path: String,
    pub cmdline: String,
}

pub struct PvDomainStore {
    kernel: String,
    ramdisk: Option<String>,
    cmdline: Option<String>,
}

pub struct DomainStore {
    vm_entries: HashMap<String, String>,
    domain_entries: HashMap<String, String>,
}

impl DomainStore {
    pub fn new() -> DomainStore {
        DomainStore {
            vm_entries: HashMap::new(),
            domain_entries: HashMap::new(),
        }
    }

    pub fn put_vm(&mut self, key: &str, value: String) {
        self.vm_entries.insert(key.to_string(), value);
    }

    pub fn put_vm_str(&mut self, key: &str, value: &str) {
        self.put_vm(key, value.to_string());
    }

    pub fn put_domain(&mut self, key: &str, value: String) {
        self.vm_entries.insert(key.to_string(), value);
    }

    pub fn put_domain_str(&mut self, key: &str, value: &str) {
        self.put_domain(key, value.to_string());
    }

    pub fn configure_memory(&mut self, maxkb: u32, targetkb: u32, videokb: u32) {
        self.put_domain("memory/static-max", maxkb.to_string());
        self.put_domain("memory/target", targetkb.to_string());
        self.put_domain("memory/videoram", videokb.to_string());
    }

    pub fn configure_cpus(&mut self, _maxvcpus: u32) {}

    pub fn configure_pv(&mut self, pv: PvDomainStore) {
        self.put_vm_str("image/ostype", "linux");
        self.put_vm("image/kernel", pv.kernel);

        match pv.ramdisk {
            None => {}
            Some(ramdisk) => self.put_vm("image/ramdisk", ramdisk),
        }

        match pv.cmdline {
            None => {}
            Some(cmdline) => self.put_vm("image/cmdline", cmdline),
        }
    }

    pub fn clone_vm_entries(&self) -> HashMap<String, String> {
        self.vm_entries.clone()
    }

    pub fn clone_domain_entries(&self) -> HashMap<String, String> {
        self.domain_entries.clone()
    }
}

impl Default for DomainStore {
    fn default() -> Self {
        DomainStore::new()
    }
}

impl PvDomainStore {
    pub fn new(kernel: String, ramdisk: Option<String>, cmdline: Option<String>) -> PvDomainStore {
        PvDomainStore {
            kernel,
            ramdisk,
            cmdline,
        }
    }
}