use std::{env, thread, time};
use virt::connect::Connect;
use virt::domain::Domain;
use virt::error::{clear_error_callback, Error};
use virt::sys;
fn suspend_and_resume(conn: &Connect, name: &str, sec: u64) -> Result<(), Error> {
if let Ok(dom) = Domain::lookup_by_name(conn, name) {
if dom.suspend().is_ok() {
println!("Domain '{:?}' suspended, info: {:?}", name, dom.get_info());
thread::sleep(time::Duration::from_millis(sec * 1000));
if dom.resume().is_ok() {
println!("Domain '{:?}' resumed, info: {:?}", name, dom.get_info());
return Ok(());
}
}
}
Err(Error::last_error())
}
fn fetch_domains(conn: &Connect) -> Result<(), Error> {
let flags = sys::VIR_CONNECT_LIST_DOMAINS_ACTIVE;
if let Ok(doms) = conn.list_all_domains(flags) {
println!("Running domains:");
println!("----------------");
for dom in doms {
println!(
"{}",
dom.get_name().unwrap_or_else(|_| String::from("no-name"))
);
}
return Ok(());
}
Err(Error::last_error())
}
fn main() {
clear_error_callback();
let uri = env::args().nth(1);
let name = env::args().nth(2).unwrap_or_default();
println!("Attempting to connect to hypervisor: '{uri:?}'");
let mut conn = match Connect::open(uri.as_deref()) {
Ok(c) => c,
Err(e) => panic!("No connection to hypervisor: {e}"),
};
if name.is_empty() {
if let Err(e) = fetch_domains(&conn) {
println!("Failed to fetch domains: {e}");
}
} else if let Err(e) = suspend_and_resume(&conn, &name, 1) {
println!("Failed to suspend/resume: {e}");
}
if let Err(e) = conn.close() {
panic!("Failed to disconnect from hypervisor: {e}");
}
}