use std::{ptr, str};
use crate::connect::Connect;
use crate::error::Error;
#[derive(Debug)]
pub struct NodeDevice {
ptr: Option<sys::virNodeDevicePtr>,
}
impl Drop for NodeDevice {
fn drop(&mut self) {
if self.ptr.is_some() {
if let Err(e) = self.free() {
panic!("Unable to drop memory for NodeDevice: {}", e)
}
}
}
}
impl NodeDevice {
pub fn new(ptr: sys::virNodeDevicePtr) -> NodeDevice {
NodeDevice { ptr: Some(ptr) }
}
pub fn as_ptr(&self) -> sys::virNodeDevicePtr {
self.ptr.unwrap()
}
pub fn lookup_by_name(conn: &Connect, id: &str) -> Result<NodeDevice, Error> {
unsafe {
let ptr = sys::virNodeDeviceLookupByName(conn.as_ptr(), string_to_c_chars!(id));
if ptr.is_null() {
return Err(Error::last_error());
}
Ok(NodeDevice::new(ptr))
}
}
pub fn lookup_scsi_host_by_www(
conn: &Connect,
wwnn: &str,
wwpn: &str,
flags: u32,
) -> Result<NodeDevice, Error> {
unsafe {
let ptr = sys::virNodeDeviceLookupSCSIHostByWWN(
conn.as_ptr(),
string_to_c_chars!(wwnn),
string_to_c_chars!(wwpn),
flags as libc::c_uint,
);
if ptr.is_null() {
return Err(Error::last_error());
}
Ok(NodeDevice::new(ptr))
}
}
pub fn create_xml(conn: &Connect, xml: &str, flags: u32) -> Result<NodeDevice, Error> {
unsafe {
let ptr = sys::virNodeDeviceCreateXML(
conn.as_ptr(),
string_to_c_chars!(xml),
flags as libc::c_uint,
);
if ptr.is_null() {
return Err(Error::last_error());
}
Ok(NodeDevice::new(ptr))
}
}
pub fn get_name(&self) -> Result<String, Error> {
unsafe {
let n = sys::virNodeDeviceGetName(self.as_ptr());
if n.is_null() {
return Err(Error::last_error());
}
Ok(c_chars_to_string!(n, nofree))
}
}
pub fn get_parent(&self) -> Result<String, Error> {
unsafe {
let n = sys::virNodeDeviceGetParent(self.as_ptr());
if n.is_null() {
return Err(Error::last_error());
}
Ok(c_chars_to_string!(n, nofree))
}
}
pub fn get_xml_desc(&self, flags: u32) -> Result<String, Error> {
unsafe {
let xml = sys::virNodeDeviceGetXMLDesc(self.as_ptr(), flags as libc::c_uint);
if xml.is_null() {
return Err(Error::last_error());
}
Ok(c_chars_to_string!(xml))
}
}
pub fn destroy(&self) -> Result<u32, Error> {
unsafe {
let ret = sys::virNodeDeviceDestroy(self.as_ptr());
if ret == -1 {
return Err(Error::last_error());
}
Ok(ret as u32)
}
}
pub fn detach(&self) -> Result<u32, Error> {
unsafe {
let ret = sys::virNodeDeviceDettach(self.as_ptr());
if ret == -1 {
return Err(Error::last_error());
}
Ok(ret as u32)
}
}
pub fn reset(&self) -> Result<u32, Error> {
unsafe {
let ret = sys::virNodeDeviceReset(self.as_ptr());
if ret == -1 {
return Err(Error::last_error());
}
Ok(ret as u32)
}
}
pub fn reattach(&self) -> Result<u32, Error> {
unsafe {
let ret = sys::virNodeDeviceReAttach(self.as_ptr());
if ret == -1 {
return Err(Error::last_error());
}
Ok(ret as u32)
}
}
pub fn detach_flags(&self, driver: &str, flags: u32) -> Result<u32, Error> {
unsafe {
let ret = sys::virNodeDeviceDetachFlags(
self.as_ptr(),
string_to_c_chars!(driver),
flags as libc::c_uint,
);
if ret == -1 {
return Err(Error::last_error());
}
Ok(ret as u32)
}
}
pub fn free(&mut self) -> Result<(), Error> {
unsafe {
if sys::virNodeDeviceFree(self.as_ptr()) == -1 {
return Err(Error::last_error());
}
self.ptr = None;
Ok(())
}
}
pub fn num_of_devices(conn: &Connect, cap: &str, flags: u32) -> Result<u32, Error> {
unsafe {
let num = sys::virNodeNumOfDevices(
conn.as_ptr(),
string_to_c_chars!(cap),
flags as libc::c_uint,
);
if num == -1 {
return Err(Error::last_error());
}
Ok(num as u32)
}
}
pub fn num_of_caps(&self) -> Result<u32, Error> {
unsafe {
let num = sys::virNodeDeviceNumOfCaps(self.as_ptr());
if num == -1 {
return Err(Error::last_error());
}
Ok(num as u32)
}
}
#[allow(clippy::needless_range_loop)]
pub fn list_caps(&self) -> Result<Vec<String>, Error> {
unsafe {
let mut names: [*mut libc::c_char; 1024] = [ptr::null_mut(); 1024];
let size = sys::virNodeDeviceListCaps(self.as_ptr(), names.as_mut_ptr(), 1024);
if size == -1 {
return Err(Error::last_error());
}
let mut array: Vec<String> = Vec::new();
for x in 0..size as usize {
array.push(c_chars_to_string!(names[x]));
}
Ok(array)
}
}
}