mirror of
https://github.com/edera-dev/krata.git
synced 2025-08-02 21:00:55 +00:00
hypha: implement basic container init
This commit is contained in:
parent
db1ba23c8c
commit
a412f6e823
@ -44,6 +44,10 @@ version = "0.14.2"
|
|||||||
version = "1.6.1"
|
version = "1.6.1"
|
||||||
features = ["v4"]
|
features = ["v4"]
|
||||||
|
|
||||||
|
[dependencies.sys-mount]
|
||||||
|
default-features = false
|
||||||
|
version = "2.1.1"
|
||||||
|
|
||||||
[lib]
|
[lib]
|
||||||
path = "src/lib.rs"
|
path = "src/lib.rs"
|
||||||
|
|
||||||
@ -52,5 +56,5 @@ name = "hyphactl"
|
|||||||
path = "bin/controller.rs"
|
path = "bin/controller.rs"
|
||||||
|
|
||||||
[[bin]]
|
[[bin]]
|
||||||
name = "hyphad"
|
name = "hyphactr"
|
||||||
path = "bin/container.rs"
|
path = "bin/container.rs"
|
||||||
|
@ -1,7 +1,9 @@
|
|||||||
|
use hypha::container::init::ContainerInit;
|
||||||
use hypha::error::Result;
|
use hypha::error::Result;
|
||||||
|
|
||||||
fn main() -> Result<()> {
|
fn main() -> Result<()> {
|
||||||
env_logger::init();
|
env_logger::init();
|
||||||
|
let mut container = ContainerInit::new();
|
||||||
|
container.init()?;
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
@ -28,6 +28,8 @@ enum Commands {
|
|||||||
cpus: u32,
|
cpus: u32,
|
||||||
#[arg(short, long, default_value_t = 512)]
|
#[arg(short, long, default_value_t = 512)]
|
||||||
mem: u64,
|
mem: u64,
|
||||||
|
#[arg(long)]
|
||||||
|
config_bundle: Option<String>,
|
||||||
},
|
},
|
||||||
Destroy {
|
Destroy {
|
||||||
#[arg(short, long)]
|
#[arg(short, long)]
|
||||||
@ -64,10 +66,18 @@ fn main() -> Result<()> {
|
|||||||
image,
|
image,
|
||||||
cpus,
|
cpus,
|
||||||
mem,
|
mem,
|
||||||
|
config_bundle,
|
||||||
} => {
|
} => {
|
||||||
let kernel = map_kernel_path(&store_path, kernel);
|
let kernel = map_kernel_path(&store_path, kernel);
|
||||||
let initrd = map_initrd_path(&store_path, initrd);
|
let initrd = map_initrd_path(&store_path, initrd);
|
||||||
let domid = controller.launch(&kernel, &initrd, &image, cpus, mem)?;
|
let domid = controller.launch(
|
||||||
|
&kernel,
|
||||||
|
&initrd,
|
||||||
|
config_bundle.as_deref(),
|
||||||
|
&image,
|
||||||
|
cpus,
|
||||||
|
mem,
|
||||||
|
)?;
|
||||||
println!("launched domain: {}", domid);
|
println!("launched domain: {}", domid);
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -92,7 +102,12 @@ fn main() -> Result<()> {
|
|||||||
];
|
];
|
||||||
table.push_row_string(&row)?;
|
table.push_row_string(&row)?;
|
||||||
}
|
}
|
||||||
println!("{}", table.to_string());
|
|
||||||
|
if table.num_records() == 1 {
|
||||||
|
println!("no containers have been launched");
|
||||||
|
} else {
|
||||||
|
println!("{}", table.to_string());
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
Ok(())
|
Ok(())
|
||||||
|
48
hypha/src/container/init.rs
Normal file
48
hypha/src/container/init.rs
Normal file
@ -0,0 +1,48 @@
|
|||||||
|
use crate::error::Result;
|
||||||
|
use std::fs;
|
||||||
|
use std::path::Path;
|
||||||
|
use sys_mount::{FilesystemType, Mount, MountFlags};
|
||||||
|
|
||||||
|
const IMAGE_BLOCK_DEVICE_PATH: &str = "/dev/xvda";
|
||||||
|
const CONFIG_BLOCK_DEVICE_PATH: &str = "/dev/xvdb";
|
||||||
|
|
||||||
|
const IMAGE_MOUNT_PATH: &str = "/image";
|
||||||
|
const CONFIG_MOUNT_PATH: &str = "/config";
|
||||||
|
|
||||||
|
pub struct ContainerInit {}
|
||||||
|
|
||||||
|
impl Default for ContainerInit {
|
||||||
|
fn default() -> Self {
|
||||||
|
Self::new()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl ContainerInit {
|
||||||
|
pub fn new() -> ContainerInit {
|
||||||
|
ContainerInit {}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn init(&mut self) -> Result<()> {
|
||||||
|
self.prepare_mounts()?;
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
fn prepare_mounts(&mut self) -> Result<()> {
|
||||||
|
let image_mount_path = Path::new(IMAGE_MOUNT_PATH);
|
||||||
|
let config_mount_path = Path::new(CONFIG_MOUNT_PATH);
|
||||||
|
self.mount_squashfs(Path::new(IMAGE_BLOCK_DEVICE_PATH), image_mount_path)?;
|
||||||
|
self.mount_squashfs(Path::new(CONFIG_BLOCK_DEVICE_PATH), config_mount_path)?;
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
fn mount_squashfs(&mut self, from: &Path, to: &Path) -> Result<()> {
|
||||||
|
if !to.is_dir() {
|
||||||
|
fs::create_dir(to)?;
|
||||||
|
}
|
||||||
|
Mount::builder()
|
||||||
|
.fstype(FilesystemType::Manual("squashfs"))
|
||||||
|
.flags(MountFlags::RDONLY)
|
||||||
|
.mount(from, to)?;
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
}
|
@ -1 +1 @@
|
|||||||
|
pub mod init;
|
||||||
|
@ -8,12 +8,17 @@ use uuid::Uuid;
|
|||||||
|
|
||||||
pub struct ConfigBlock<'a> {
|
pub struct ConfigBlock<'a> {
|
||||||
pub image_info: &'a ImageInfo,
|
pub image_info: &'a ImageInfo,
|
||||||
|
pub config_bundle: Option<&'a str>,
|
||||||
pub file: PathBuf,
|
pub file: PathBuf,
|
||||||
pub dir: PathBuf,
|
pub dir: PathBuf,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl ConfigBlock<'_> {
|
impl ConfigBlock<'_> {
|
||||||
pub fn new<'a>(uuid: &Uuid, image_info: &'a ImageInfo) -> Result<ConfigBlock<'a>> {
|
pub fn new<'a>(
|
||||||
|
uuid: &Uuid,
|
||||||
|
image_info: &'a ImageInfo,
|
||||||
|
config_bundle: Option<&'a str>,
|
||||||
|
) -> Result<ConfigBlock<'a>> {
|
||||||
let mut dir = std::env::temp_dir().clone();
|
let mut dir = std::env::temp_dir().clone();
|
||||||
dir.push(format!("hypha-cfg-{}", uuid));
|
dir.push(format!("hypha-cfg-{}", uuid));
|
||||||
fs::create_dir_all(&dir)?;
|
fs::create_dir_all(&dir)?;
|
||||||
@ -21,12 +26,17 @@ impl ConfigBlock<'_> {
|
|||||||
file.push("config.squashfs");
|
file.push("config.squashfs");
|
||||||
Ok(ConfigBlock {
|
Ok(ConfigBlock {
|
||||||
image_info,
|
image_info,
|
||||||
|
config_bundle,
|
||||||
file,
|
file,
|
||||||
dir,
|
dir,
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn build(&self) -> Result<()> {
|
pub fn build(&self) -> Result<()> {
|
||||||
|
let config_bundle_content = match self.config_bundle {
|
||||||
|
None => None,
|
||||||
|
Some(path) => Some(fs::read(path)?),
|
||||||
|
};
|
||||||
let manifest = self.image_info.config.to_string()?;
|
let manifest = self.image_info.config.to_string()?;
|
||||||
let mut writer = FilesystemWriter::default();
|
let mut writer = FilesystemWriter::default();
|
||||||
writer.push_dir(
|
writer.push_dir(
|
||||||
@ -48,6 +58,18 @@ impl ConfigBlock<'_> {
|
|||||||
mtime: 0,
|
mtime: 0,
|
||||||
},
|
},
|
||||||
)?;
|
)?;
|
||||||
|
if let Some(config_bundle_content) = config_bundle_content.as_ref() {
|
||||||
|
writer.push_file(
|
||||||
|
config_bundle_content.as_slice(),
|
||||||
|
"/bundle",
|
||||||
|
NodeHeader {
|
||||||
|
permissions: 384,
|
||||||
|
uid: 0,
|
||||||
|
gid: 0,
|
||||||
|
mtime: 0,
|
||||||
|
},
|
||||||
|
)?;
|
||||||
|
}
|
||||||
let mut file = File::create(&self.file)?;
|
let mut file = File::create(&self.file)?;
|
||||||
writer.write(&mut file)?;
|
writer.write(&mut file)?;
|
||||||
Ok(())
|
Ok(())
|
||||||
|
@ -63,6 +63,7 @@ impl Controller {
|
|||||||
&mut self,
|
&mut self,
|
||||||
kernel_path: &str,
|
kernel_path: &str,
|
||||||
initrd_path: &str,
|
initrd_path: &str,
|
||||||
|
config_bundle_path: Option<&str>,
|
||||||
image: &str,
|
image: &str,
|
||||||
vcpus: u32,
|
vcpus: u32,
|
||||||
mem: u64,
|
mem: u64,
|
||||||
@ -70,7 +71,7 @@ impl Controller {
|
|||||||
let uuid = Uuid::new_v4();
|
let uuid = Uuid::new_v4();
|
||||||
let name = format!("hypha-{uuid}");
|
let name = format!("hypha-{uuid}");
|
||||||
let image_info = self.compile(image)?;
|
let image_info = self.compile(image)?;
|
||||||
let cfgblk = ConfigBlock::new(&uuid, &image_info)?;
|
let cfgblk = ConfigBlock::new(&uuid, &image_info, config_bundle_path)?;
|
||||||
cfgblk.build()?;
|
cfgblk.build()?;
|
||||||
|
|
||||||
let image_squashfs_path = image_info
|
let image_squashfs_path = image_info
|
||||||
@ -110,6 +111,7 @@ impl Controller {
|
|||||||
writable: false,
|
writable: false,
|
||||||
},
|
},
|
||||||
],
|
],
|
||||||
|
filesystems: vec![],
|
||||||
extra_keys: vec![
|
extra_keys: vec![
|
||||||
("hypha/uuid".to_string(), uuid.to_string()),
|
("hypha/uuid".to_string(), uuid.to_string()),
|
||||||
(
|
(
|
||||||
|
@ -35,6 +35,15 @@ impl Error for HyphaError {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[macro_export]
|
||||||
|
macro_rules! hypha_err {
|
||||||
|
($($arg:tt)*) => {{
|
||||||
|
use $crate::error::HyphaError;
|
||||||
|
let text = std::fmt::format(format_args!($($arg)*));
|
||||||
|
Err(HyphaError::new(text.as_str()))
|
||||||
|
}}
|
||||||
|
}
|
||||||
|
|
||||||
impl From<std::io::Error> for HyphaError {
|
impl From<std::io::Error> for HyphaError {
|
||||||
fn from(value: std::io::Error) -> Self {
|
fn from(value: std::io::Error) -> Self {
|
||||||
HyphaError::new(value.to_string().as_str())
|
HyphaError::new(value.to_string().as_str())
|
||||||
|
@ -21,6 +21,7 @@ fn main() -> Result<(), XenClientError> {
|
|||||||
initrd_path: initrd_path.as_str(),
|
initrd_path: initrd_path.as_str(),
|
||||||
cmdline: "debug elevator=noop",
|
cmdline: "debug elevator=noop",
|
||||||
disks: vec![],
|
disks: vec![],
|
||||||
|
filesystems: vec![],
|
||||||
extra_keys: vec![],
|
extra_keys: vec![],
|
||||||
};
|
};
|
||||||
let domid = client.create(&config)?;
|
let domid = client.create(&config)?;
|
||||||
|
@ -97,6 +97,11 @@ pub struct DomainDisk<'a> {
|
|||||||
pub writable: bool,
|
pub writable: bool,
|
||||||
}
|
}
|
||||||
|
|
||||||
|
pub struct DomainFilesystem<'a> {
|
||||||
|
pub path: &'a str,
|
||||||
|
pub tag: &'a str,
|
||||||
|
}
|
||||||
|
|
||||||
pub struct DomainConfig<'a> {
|
pub struct DomainConfig<'a> {
|
||||||
pub backend_domid: u32,
|
pub backend_domid: u32,
|
||||||
pub name: &'a str,
|
pub name: &'a str,
|
||||||
@ -106,6 +111,7 @@ pub struct DomainConfig<'a> {
|
|||||||
pub initrd_path: &'a str,
|
pub initrd_path: &'a str,
|
||||||
pub cmdline: &'a str,
|
pub cmdline: &'a str,
|
||||||
pub disks: Vec<DomainDisk<'a>>,
|
pub disks: Vec<DomainDisk<'a>>,
|
||||||
|
pub filesystems: Vec<DomainFilesystem<'a>>,
|
||||||
pub extra_keys: Vec<(String, String)>,
|
pub extra_keys: Vec<(String, String)>,
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -403,6 +409,16 @@ impl XenClient {
|
|||||||
disk,
|
disk,
|
||||||
)?;
|
)?;
|
||||||
}
|
}
|
||||||
|
for (index, filesystem) in config.filesystems.iter().enumerate() {
|
||||||
|
self.fs_9p_device_add(
|
||||||
|
&dom_path,
|
||||||
|
&backend_dom_path,
|
||||||
|
config.backend_domid,
|
||||||
|
domid,
|
||||||
|
index,
|
||||||
|
filesystem,
|
||||||
|
)?;
|
||||||
|
}
|
||||||
self.call.unpause_domain(domid)?;
|
self.call.unpause_domain(domid)?;
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
@ -497,6 +513,43 @@ impl XenClient {
|
|||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
||||||
|
fn fs_9p_device_add(
|
||||||
|
&mut self,
|
||||||
|
dom_path: &str,
|
||||||
|
backend_dom_path: &str,
|
||||||
|
backend_domid: u32,
|
||||||
|
domid: u32,
|
||||||
|
index: usize,
|
||||||
|
filesystem: &DomainFilesystem,
|
||||||
|
) -> Result<(), XenClientError> {
|
||||||
|
let id = 90 + index as u64;
|
||||||
|
let backend_items: Vec<(&str, String)> = vec![
|
||||||
|
("frontend-id", domid.to_string()),
|
||||||
|
("online", "1".to_string()),
|
||||||
|
("state", "1".to_string()),
|
||||||
|
("path", filesystem.path.to_string()),
|
||||||
|
("security-model", "none".to_string()),
|
||||||
|
];
|
||||||
|
|
||||||
|
let frontend_items: Vec<(&str, String)> = vec![
|
||||||
|
("backend-id", backend_domid.to_string()),
|
||||||
|
("state", "1".to_string()),
|
||||||
|
("tag", filesystem.tag.to_string()),
|
||||||
|
];
|
||||||
|
|
||||||
|
self.device_add(
|
||||||
|
"9pfs",
|
||||||
|
id,
|
||||||
|
dom_path,
|
||||||
|
backend_dom_path,
|
||||||
|
backend_domid,
|
||||||
|
domid,
|
||||||
|
frontend_items,
|
||||||
|
backend_items,
|
||||||
|
)?;
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
#[allow(clippy::too_many_arguments)]
|
#[allow(clippy::too_many_arguments)]
|
||||||
fn device_add(
|
fn device_add(
|
||||||
&mut self,
|
&mut self,
|
||||||
|
Loading…
Reference in New Issue
Block a user