]> git.proxmox.com Git - proxmox-backup.git/blobdiff - src/tools/sgutils2.rs
typo fixes all over the place
[proxmox-backup.git] / src / tools / sgutils2.rs
index 749b694721d675df88f46f466a7d16fb6239b3e2..52352b15dd811ffb8d79c17442100998e8a27db2 100644 (file)
 //! Incomplete, but we currently do not need more.
 //!
 //! See: `/usr/include/scsi/sg_pt.h`
+//!
+//! The SCSI Commands Reference Manual also contains some useful information.
 
 use std::os::unix::io::AsRawFd;
+use std::ptr::NonNull;
 
-use anyhow::{bail, Error};
+use anyhow::{bail, format_err, Error};
+use endian_trait::Endian;
+use serde::{Deserialize, Serialize};
 use libc::{c_char, c_int};
+use std::ffi::CStr;
+
+use proxmox::tools::io::ReadExt;
+
+#[derive(Debug)]
+pub struct SenseInfo {
+    pub sense_key: u8,
+    pub asc: u8,
+    pub ascq: u8,
+}
+
+impl ToString for SenseInfo {
+
+    fn to_string(&self) -> String {
+
+        let sense_text = SENSE_KEY_DESCRIPTIONS
+            .get(self.sense_key as usize)
+            .map(|s| String::from(*s))
+            .unwrap_or_else(|| format!("Invalid sense {:02X}", self.sense_key));
+
+        if self.asc == 0 && self.ascq == 0 {
+            return sense_text;
+        }
+
+        let additional_sense_text = get_asc_ascq_string(self.asc, self.ascq);
+
+        format!("{}, {}", sense_text, additional_sense_text)
+    }
+}
+
+#[derive(Debug)]
+pub struct ScsiError {
+    pub error: Error,
+    pub sense: Option<SenseInfo>,
+}
+
+impl std::fmt::Display for ScsiError {
+    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
+        write!(f, "{}", self.error)
+    }
+}
+
+impl std::error::Error for ScsiError {
+    fn source(&self) -> Option<&(dyn std::error::Error + 'static)> {
+        self.error.source()
+    }
+}
+
+impl From<anyhow::Error> for ScsiError {
+    fn from(error: anyhow::Error) -> Self {
+        Self { error, sense: None }
+    }
+}
+
+impl From<std::io::Error> for ScsiError {
+    fn from(error: std::io::Error) -> Self {
+        Self { error: error.into(), sense: None }
+    }
+}
 
 // Opaque wrapper for sg_pt_base
 #[repr(C)]
 struct SgPtBase { _private: [u8; 0] }
 
-impl Drop for SgPtBase  {
+#[repr(transparent)]
+struct SgPt {
+    raw: NonNull<SgPtBase>,
+}
+
+impl Drop for SgPt {
     fn drop(&mut self) {
-        unsafe { destruct_scsi_pt_obj(self as *mut SgPtBase) };
+        unsafe { destruct_scsi_pt_obj(self.as_mut_ptr()) };
+    }
+}
+
+impl SgPt {
+    fn new() -> Result<Self, Error> {
+        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_ptr()
     }
+
+    fn as_mut_ptr(&mut self) -> *mut SgPtBase {
+        self.raw.as_ptr()
+    }
+}
+
+/// Peripheral device type text (see `inquiry` command)
+///
+/// see [https://en.wikipedia.org/wiki/SCSI_Peripheral_Device_Type]
+pub const PERIPHERAL_DEVICE_TYPE_TEXT: [&'static str; 32] = [
+    "Disk Drive",
+    "Tape Drive",
+    "Printer",
+    "Processor",
+    "Write-once",
+    "CD-ROM",  // 05h
+    "Scanner",
+    "Optical",
+    "Medium Changer", // 08h
+    "Communications",
+    "ASC IT8",
+    "ASC IT8",
+    "RAID Array",
+    "Enclosure Services",
+    "Simplified direct-access",
+    "Optical card reader/writer",
+    "Bridging Expander",
+    "Object-based Storage",
+    "Automation/Drive Interface",
+    "Security manager",
+    "Reserved",
+    "Reserved",
+    "Reserved",
+    "Reserved",
+    "Reserved",
+    "Reserved",
+    "Reserved",
+    "Reserved",
+    "Reserved",
+    "Reserved",
+    "Reserved",
+    "Unknown",
+];
+
+//  SENSE KEYS
+pub const SENSE_KEY_NO_SENSE: u8        = 0x00;
+pub const SENSE_KEY_RECOVERED_ERROR: u8 = 0x01;
+pub const SENSE_KEY_NOT_READY: u8       = 0x02;
+pub const SENSE_KEY_MEDIUM_ERROR: u8    = 0x03;
+pub const SENSE_KEY_HARDWARE_ERROR: u8  = 0x04;
+pub const SENSE_KEY_ILLEGAL_REQUEST: u8 = 0x05;
+pub const SENSE_KEY_UNIT_ATTENTION: u8  = 0x06;
+pub const SENSE_KEY_DATA_PROTECT: u8    = 0x07;
+pub const SENSE_KEY_BLANK_CHECK: u8     = 0x08;
+pub const SENSE_KEY_COPY_ABORTED: u8    = 0x0a;
+pub const SENSE_KEY_ABORTED_COMMAND: u8 = 0x0b;
+pub const SENSE_KEY_VOLUME_OVERFLOW: u8 = 0x0d;
+pub const SENSE_KEY_MISCOMPARE: u8      = 0x0e;
+
+/// Sense Key Descriptions
+pub const SENSE_KEY_DESCRIPTIONS: [&'static str; 16] = [
+    "No Sense",
+    "Recovered Error",
+    "Not Ready",
+    "Medium Error",
+    "Hardware Error",
+    "Illegal Request",
+    "Unit Attention",
+    "Data Protect",
+    "Blank Check",
+    "Vendor specific",
+    "Copy Aborted",
+    "Aborted Command",
+    "Equal",
+    "Volume Overflow",
+    "Miscompare",
+    "Completed",
+];
+
+#[repr(C, packed)]
+#[derive(Endian)]
+// Standard Inquiry page - 36 bytes
+struct InquiryPage {
+    peripheral_type: u8,
+    rmb: u8,
+    version: u8,
+    flags3: u8,
+    additional_length: u8,
+    flags5: u8,
+    flags6: u8,
+    flags7: u8,
+    vendor: [u8; 8],
+    product: [u8; 16],
+    revision: [u8; 4],
+}
+
+#[repr(C, packed)]
+#[derive(Endian, Debug)]
+struct RequestSenseFixed {
+    response_code: u8,
+    obsolete: u8,
+    flags2: u8,
+    information: [u8;4],
+    additional_sense_len: u8,
+    command_specific_information: [u8;4],
+    additional_sense_code: u8,
+    additional_sense_code_qualifier: u8,
+    field_replacable_unit_code: u8,
+    sense_key_specific: [u8; 3],
 }
 
+#[repr(C, packed)]
+#[derive(Endian, Debug)]
+struct RequestSenseDescriptor{
+    response_code: u8,
+    sense_key: u8,
+    additional_sense_code: u8,
+    additional_sense_code_qualifier: u8,
+    reserved: [u8;4],
+    additional_sense_len: u8,
+}
+
+/// Inquiry result
+#[derive(Serialize, Deserialize, Debug)]
+pub struct InquiryInfo {
+    /// Peripheral device type (0-31)
+    pub peripheral_type: u8,
+    /// Peripheral device type as string
+    pub peripheral_type_text: String,
+    /// Vendor
+    pub vendor: String,
+    /// Product
+    pub product: String,
+    /// Revision
+    pub revision: String,
+}
+
+pub const SCSI_PT_DO_START_OK:c_int = 0;
+pub const SCSI_PT_DO_BAD_PARAMS:c_int = 1;
+pub const SCSI_PT_DO_TIMEOUT:c_int = 2;
+
 pub const SCSI_PT_RESULT_GOOD:c_int = 0;
 pub const SCSI_PT_RESULT_STATUS:c_int = 1;
 pub const SCSI_PT_RESULT_SENSE:c_int = 2;
@@ -26,8 +247,8 @@ pub const SCSI_PT_RESULT_TRANSPORT_ERR:c_int = 3;
 pub const SCSI_PT_RESULT_OS_ERR:c_int = 4;
 
 #[link(name = "sgutils2")]
-extern {
-    
+extern "C" {
+
     #[allow(dead_code)]
     fn scsi_pt_open_device(
         device_name: * const c_char,
@@ -45,7 +266,7 @@ extern {
 
     fn set_scsi_pt_data_in(
         objp: *mut SgPtBase,
-        dxferp: *const u8,
+        dxferp: *mut u8,
         dxfer_ilen: c_int,
     );
 
@@ -63,7 +284,7 @@ extern {
 
     fn set_scsi_pt_sense(
         objp: *mut SgPtBase,
-        sense: *const u8,
+        sense: *mut u8,
         max_sense_len: c_int,
     );
 
@@ -80,23 +301,16 @@ extern {
 
     fn get_scsi_pt_status_response(objp: *const SgPtBase) -> c_int;
 
-    #[allow(dead_code)]
     fn get_scsi_pt_result_category(objp: *const SgPtBase) -> c_int;
-}
 
-/// Creates a `Box<SgPtBase>`
-///
-/// Which get automatically dropped, so you do not need to call
-/// destruct_scsi_pt_obj yourself.
-fn boxed_scsi_pt_obj() -> Result<Box<SgPtBase>, Error> {
-    let objp = unsafe {
-        construct_scsi_pt_obj()
-    };
-    if objp.is_null() {
-        bail!("construct_scsi_pt_ob failed");
-    }
+    fn get_scsi_pt_os_err(objp: *const SgPtBase) -> c_int;
 
-    Ok(unsafe { std::mem::transmute(objp)})
+    fn sg_get_asc_ascq_str(
+        asc: c_int,
+        ascq:c_int,
+        buff_len: c_int,
+        buffer: *mut c_char,
+    ) -> * const c_char;
 }
 
 /// Safe interface to run RAW SCSI commands
@@ -104,6 +318,30 @@ pub struct SgRaw<'a, F> {
     file: &'a mut F,
     buffer: Box<[u8]>,
     sense_buffer: [u8; 32],
+    timeout: i32,
+}
+
+/// Get the string associated with ASC/ASCQ values
+pub fn get_asc_ascq_string(asc: u8, ascq: u8) -> String {
+
+    let mut buffer = [0u8; 1024];
+    let res = unsafe {
+        sg_get_asc_ascq_str(
+            asc as c_int,
+            ascq as c_int,
+            buffer.len() as c_int,
+            buffer.as_mut_ptr() as * mut c_char,
+        )
+    };
+
+    proxmox::try_block!({
+        if res.is_null() { // just to be safe
+            bail!("unexpected NULL ptr");
+        }
+        Ok(unsafe { CStr::from_ptr(res) }.to_str()?.to_owned())
+    }).unwrap_or_else(|_err: Error| {
+        format!("ASC={:02x}x, ASCQ={:02x}x", asc, ascq)
+    })
 }
 
 /// Allocate a page aligned buffer
@@ -138,19 +376,28 @@ impl <'a, F: AsRawFd> SgRaw<'a, F> {
 
         let sense_buffer = [0u8; 32];
 
-        Ok(Self { file, buffer, sense_buffer })
+        Ok(Self { file, buffer, sense_buffer, timeout: 0 })
+    }
+
+    /// Set the command timeout in seconds (0 means default (60 seconds))
+    pub fn set_timeout(&mut self, seconds: usize) {
+        if seconds > (i32::MAX as usize) {
+            self.timeout = i32::MAX; // don't care about larger values
+        } else {
+            self.timeout = seconds as i32;
+        }
     }
 
     // create new object with initialized data_in and sense buffer
-    fn create_boxed_scsi_pt_obj(&mut self) -> Result<Box<SgPtBase>, Error> {
+    fn create_scsi_pt_obj(&mut self) -> Result<SgPt, Error> {
 
-        let mut ptvp = boxed_scsi_pt_obj()?;
+        let mut ptvp = SgPt::new()?;
 
         if self.buffer.len() > 0 {
             unsafe {
                 set_scsi_pt_data_in(
-                    &mut *ptvp,
-                    self.buffer.as_ptr(),
+                    ptvp.as_mut_ptr(),
+                    self.buffer.as_mut_ptr(),
                     self.buffer.len() as c_int,
                 )
             };
@@ -158,8 +405,8 @@ impl <'a, F: AsRawFd> SgRaw<'a, F> {
 
         unsafe {
             set_scsi_pt_sense(
-                &mut *ptvp,
-                self.sense_buffer.as_ptr(),
+                ptvp.as_mut_ptr(),
+                self.sense_buffer.as_mut_ptr(),
                 self.sense_buffer.len() as c_int,
             )
         };
@@ -167,50 +414,118 @@ impl <'a, F: AsRawFd> SgRaw<'a, F> {
         Ok(ptvp)
     }
 
+    fn do_scsi_pt_checked(&mut self, ptvp: &mut SgPt) -> Result<(), ScsiError> {
+
+        let res = unsafe { do_scsi_pt(ptvp.as_mut_ptr(), self.file.as_raw_fd(), self.timeout, 0) };
+        match res {
+            SCSI_PT_DO_START_OK => { /* Ok */ },
+            SCSI_PT_DO_BAD_PARAMS => return Err(format_err!("do_scsi_pt failed - bad pass through setup").into()),
+            SCSI_PT_DO_TIMEOUT => return Err(format_err!("do_scsi_pt failed - timeout").into()),
+            code if code < 0 => {
+                let errno = unsafe { get_scsi_pt_os_err(ptvp.as_ptr()) };
+                let err = nix::Error::from_errno(nix::errno::Errno::from_i32(errno));
+                return Err(format_err!("do_scsi_pt failed with err {}", err).into());
+            }
+            unknown => return Err(format_err!("do_scsi_pt failed: unknown error {}", unknown).into()),
+        }
+
+        if res < 0 {
+            let err = nix::Error::last();
+            return Err(format_err!("do_scsi_pt failed  - {}", err).into());
+        }
+        if res != 0 {
+            return Err(format_err!("do_scsi_pt failed {}", res).into());
+        }
+
+        let sense_len = unsafe { get_scsi_pt_sense_len(ptvp.as_ptr()) };
+
+        let res_cat = unsafe { get_scsi_pt_result_category(ptvp.as_ptr()) };
+        match res_cat {
+            SCSI_PT_RESULT_GOOD => return Ok(()),
+            SCSI_PT_RESULT_STATUS => {
+                let status = unsafe { get_scsi_pt_status_response(ptvp.as_ptr()) };
+                if status != 0 {
+                    return Err(format_err!("unknown scsi error - status response {}", status).into());
+                }
+                return Ok(());
+            }
+            SCSI_PT_RESULT_SENSE => {
+                if sense_len == 0 {
+                    return Err(format_err!("scsi command failed, but got no sense data").into());
+                }
+
+                let code = self.sense_buffer[0] & 0x7f;
+
+                let mut reader = &self.sense_buffer[..(sense_len as usize)];
+
+                let sense = match code {
+                    0x70 => {
+                        let sense: RequestSenseFixed = unsafe { reader.read_be_value()? };
+                        SenseInfo {
+                            sense_key: sense.flags2 & 0xf,
+                            asc: sense.additional_sense_code,
+                            ascq: sense.additional_sense_code_qualifier,
+                        }
+                    }
+                    0x72 => {
+                        let sense: RequestSenseDescriptor = unsafe { reader.read_be_value()? };
+                        SenseInfo {
+                            sense_key: sense.sense_key & 0xf,
+                            asc: sense.additional_sense_code,
+                            ascq: sense.additional_sense_code_qualifier,
+                        }
+                    }
+                    0x71 | 0x73 => {
+                        return Err(format_err!("scsi command failed: received deferred Sense").into());
+                    }
+                    unknown => {
+                        return Err(format_err!("scsi command failed: invalid Sense response code {:x}", unknown).into());
+                    }
+                };
+
+                return Err(ScsiError {
+                    error: format_err!("{}", sense.to_string()),
+                    sense: Some(sense),
+                });
+            }
+            SCSI_PT_RESULT_TRANSPORT_ERR => return Err(format_err!("scsi command failed: transport error").into()),
+            SCSI_PT_RESULT_OS_ERR => {
+                let errno = unsafe { get_scsi_pt_os_err(ptvp.as_ptr()) };
+                let err = nix::Error::from_errno(nix::errno::Errno::from_i32(errno));
+                return Err(format_err!("scsi command failed with err {}", err).into());
+            }
+            unknown => return Err(format_err!("scsi command failed: unknown result category {}", unknown).into()),
+        }
+    }
+
     /// Run the specified RAW SCSI command
-    pub fn do_command(&mut self, cmd: &[u8]) -> Result<&[u8], Error> {
+    pub fn do_command(&mut self, cmd: &[u8]) -> Result<&[u8], ScsiError> {
 
         if !unsafe { sg_is_scsi_cdb(cmd.as_ptr(), cmd.len() as c_int) } {
-            bail!("no valid SCSI command");
+            return Err(format_err!("no valid SCSI command").into());
         }
 
         if self.buffer.len() < 16 {
-            bail!("output buffer too small");
+            return Err(format_err!("output buffer too small").into());
         }
 
-        let mut ptvp = self.create_boxed_scsi_pt_obj()?;
+        let mut ptvp = self.create_scsi_pt_obj()?;
 
         unsafe {
             set_scsi_pt_cdb(
-                &mut *ptvp,
+                ptvp.as_mut_ptr(),
                 cmd.as_ptr(),
                 cmd.len() as c_int,
             )
         };
 
-        let res = unsafe { do_scsi_pt(&mut *ptvp, self.file.as_raw_fd(), 0, 0) };
-        if res < 0 {
-            let err = nix::Error::last();
-            bail!("do_scsi_pt failed  - {}", err);
-        }
-        if res != 0 {
-            bail!("do_scsi_pt failed {}", res);
-        }
-
-        // todo: what about sense data?
-        let _sense_len = unsafe { get_scsi_pt_sense_len(&*ptvp) };
+        self.do_scsi_pt_checked(&mut ptvp)?;
 
-        let status = unsafe { get_scsi_pt_status_response(&*ptvp) };
-        if status != 0 {
-            // toto: improve error reporting
-            bail!("unknown scsi error - status response {}", status);
-        }
-
-        let data_len = self.buffer.len() -
-            (unsafe { get_scsi_pt_resid(&*ptvp) } as usize);
-        if data_len == 0 {
-            bail!("do_scsi_pt failed - no data received");
+        let resid = unsafe { get_scsi_pt_resid(ptvp.as_ptr()) } as usize;
+        if resid > self.buffer.len() {
+            return Err(format_err!("do_scsi_pt failed - got strange resid (value too big)").into());
         }
+        let data_len = self.buffer.len() - resid;
 
         Ok(&self.buffer[..data_len])
     }
@@ -229,40 +544,71 @@ impl <'a, F: AsRawFd> SgRaw<'a, F> {
             bail!("wrong transfer buffer alignment");
         }
 
-        let mut ptvp = self.create_boxed_scsi_pt_obj()?;
+        let mut ptvp = self.create_scsi_pt_obj()?;
 
         unsafe {
             set_scsi_pt_data_out(
-                &mut *ptvp,
+                ptvp.as_mut_ptr(),
                 data.as_ptr(),
                 data.len() as c_int,
             );
 
             set_scsi_pt_cdb(
-                &mut *ptvp,
+                ptvp.as_mut_ptr(),
                 cmd.as_ptr(),
                 cmd.len() as c_int,
             );
-         };
-
-        let res = unsafe { do_scsi_pt(&mut *ptvp, self.file.as_raw_fd(), 0, 0) };
-        if res < 0 {
-            let err = nix::Error::last();
-            bail!("do_scsi_pt failed  - {}", err);
-        }
-        if res != 0 {
-            bail!("do_scsi_pt failed {}", res);
-        }
-
-        // todo: what about sense data?
-        let _sense_len = unsafe { get_scsi_pt_sense_len(&*ptvp) };
+        };
 
-        let status = unsafe { get_scsi_pt_status_response(&*ptvp) };
-        if status != 0 {
-            // toto: improve error reporting
-            bail!("unknown scsi error - status response {}", status);
-        }
+        self.do_scsi_pt_checked(&mut ptvp)?;
 
         Ok(())
     }
 }
+
+// Useful helpers
+
+/// Converts SCSI ASCII text into String, trim zero and spaces
+pub fn scsi_ascii_to_string(data: &[u8]) -> String {
+    String::from_utf8_lossy(data)
+        .trim_matches(char::from(0))
+        .trim()
+        .to_string()
+}
+
+/// Read SCSI Inquiry page
+///
+/// Returns Product/Vendor/Revision and device type.
+pub fn scsi_inquiry<F: AsRawFd>(
+    file: &mut F,
+) -> Result<InquiryInfo, Error> {
+
+    let allocation_len: u8 = std::mem::size_of::<InquiryPage>() as u8;
+
+    let mut sg_raw = SgRaw::new(file, allocation_len as usize)?;
+    sg_raw.set_timeout(30); // use short timeout
+
+    let mut cmd = Vec::new();
+    cmd.extend(&[0x12, 0, 0, 0, allocation_len, 0]); // INQUIRY
+
+    let data = sg_raw.do_command(&cmd)
+        .map_err(|err| format_err!("SCSI inquiry failed - {}", err))?;
+
+    proxmox::try_block!({
+        let mut reader = &data[..];
+
+        let page: InquiryPage  = unsafe { reader.read_be_value()? };
+
+        let peripheral_type = page.peripheral_type & 31;
+
+        let info = InquiryInfo {
+            peripheral_type,
+            peripheral_type_text: PERIPHERAL_DEVICE_TYPE_TEXT[peripheral_type as usize].to_string(),
+            vendor: scsi_ascii_to_string(&page.vendor),
+            product: scsi_ascii_to_string(&page.product),
+            revision: scsi_ascii_to_string(&page.revision),
+        };
+
+        Ok(info)
+    }).map_err(|err: Error| format_err!("decode inquiry page failed - {}", err))
+}