tyler/proxmox-backup
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

tools::sgutils2: use NonNull

Browse Source 
Signed-off-by: Wolfgang Bumiller <w.bumiller@proxmox.com>
This commit is contained in:
Wolfgang Bumiller 2021-01-25 14:54:33 +01:00
parent fe61280b6b
commit b7f9b25e4d
1 changed files with 10 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
src/tools/sgutils2.rs
View File

@ -5,6 +5,7 @@
//! See: `/usr/include/scsi/sg_pt.h`
use std::os::unix::io::AsRawFd;
use std::ptr::NonNull;
use anyhow::{bail, format_err, Error};
use endian_trait::Endian;
@ -18,32 +19,29 @@ struct SgPtBase { _private: [u8; 0] }
#[repr(transparent)]
struct SgPt {
raw: *mut SgPtBase,
raw: NonNull<SgPtBase>,
}
impl Drop for SgPt {
fn drop(&mut self) {
unsafe { destruct_scsi_pt_obj(self.raw) };
unsafe { destruct_scsi_pt_obj(self.as_mut_ptr()) };
}
}
impl SgPt {
fn new() -> Result<Self, Error> {
let raw = unsafe { construct_scsi_pt_obj() };
if raw.is_null() {
bail!("construct_scsi_pt_ob failed");
}
Ok(Self { raw })
Ok(Self {
raw: NonNull::new(unsafe { construct_scsi_pt_obj() })
.ok_or_else(|| format_err!("construct_scsi_pt_ob failed"))?,
})
}
fn as_ptr(&self) -> *const SgPtBase {
self.raw as *const SgPtBase
self.raw.as_ptr()
}
fn as_mut_ptr(&mut self) -> *mut SgPtBase {
self.raw
self.raw.as_ptr()
}
}