diff options
Diffstat (limited to 'src')
| -rw-r--r-- | src/device.rs | 93 | ||||
| -rw-r--r-- | src/lib.rs | 4 | 
2 files changed, 95 insertions, 2 deletions
diff --git a/src/device.rs b/src/device.rs index f247f58..4032db6 100644 --- a/src/device.rs +++ b/src/device.rs @@ -208,6 +208,38 @@ pub struct VolumeStatus {      pub active: bool,  } +/// Information about the SD card in a Storage device. +#[derive(Debug)] +pub struct SdCardData { +    /// The serial number of the SD card. +    pub serial_number: u32, +    /// The size of the SD card in GB. +    pub size: u8, +    /// The year the card was manufactured, e. g. 17 for 2017. +    pub manufacturing_year: u8, +    /// The month the card was manufactured. +    pub manufacturing_month: u8, +    /// The OEM ID. +    pub oem: u16, +    /// The manufacturer ID. +    pub manufacturer: u8, +} + +#[derive(Debug)] +/// Production information for a Storage device. +pub struct StorageProductionInfo { +    /// The major firmware version, e. g. 0 in v0.40. +    pub firmware_version_major: u8, +    /// The minor firmware version, e. g. 40 in v0.40. +    pub firmware_version_minor: u8, +    /// The internal firmware version. +    pub firmware_version_internal: u8, +    /// The serial number of the CPU. +    pub serial_number_cpu: u32, +    /// Information about the SD card. +    pub sd_card: SdCardData, +} +  /// The status of a Nitrokey Storage device.  #[derive(Debug)]  pub struct StorageStatus { @@ -1166,6 +1198,48 @@ impl Storage {          result.and(Ok(StorageStatus::from(raw_status)))      } +    /// Returns the production information for the connected storage device. +    /// +    /// # Example +    /// +    /// ```no_run +    /// # use nitrokey::CommandError; +    /// +    /// fn use_volume() {} +    /// +    /// # fn try_main() -> Result<(), CommandError> { +    /// let device = nitrokey::Storage::connect()?; +    /// match device.get_production_info() { +    ///     Ok(data) => { +    ///         println!("SD card ID:   {:#x}", data.sd_card.serial_number); +    ///         println!("SD card size: {} GB", data.sd_card.size); +    ///     }, +    ///     Err(err) => println!("Could not get Storage production info: {}", err), +    /// }; +    /// #     Ok(()) +    /// # } +    /// ``` +    pub fn get_production_info(&self) -> Result<StorageProductionInfo, CommandError> { +        let mut raw_data = nitrokey_sys::NK_storage_ProductionTest { +            FirmwareVersion_au8: [0, 2], +            FirmwareVersionInternal_u8: 0, +            SD_Card_Size_u8: 0, +            CPU_CardID_u32: 0, +            SmartCardID_u32: 0, +            SD_CardID_u32: 0, +            SC_UserPwRetryCount: 0, +            SC_AdminPwRetryCount: 0, +            SD_Card_ManufacturingYear_u8: 0, +            SD_Card_ManufacturingMonth_u8: 0, +            SD_Card_OEM_u16: 0, +            SD_WriteSpeed_u16: 0, +            SD_Card_Manufacturer_u8: 0, +        }; +        let raw_result = unsafe { nitrokey_sys::NK_get_storage_production_info(&mut raw_data) }; +        let result = get_command_result(raw_result); +        result.and(Ok(StorageProductionInfo::from(raw_data))) +    } +      /// Blinks the red and green LED alternatively and infinitely until the device is reconnected.      pub fn wink(&self) -> Result<(), CommandError> {          get_command_result(unsafe { nitrokey_sys::NK_wink() }) @@ -1209,6 +1283,25 @@ impl Device for Storage {  impl GenerateOtp for Storage {} +impl From<nitrokey_sys::NK_storage_ProductionTest> for StorageProductionInfo { +    fn from(data: nitrokey_sys::NK_storage_ProductionTest) -> Self { +        Self { +            firmware_version_major: data.FirmwareVersion_au8[0], +            firmware_version_minor: data.FirmwareVersion_au8[1], +            firmware_version_internal: data.FirmwareVersionInternal_u8, +            serial_number_cpu: data.CPU_CardID_u32, +            sd_card: SdCardData { +                serial_number: data.SD_CardID_u32, +                size: data.SD_Card_Size_u8, +                manufacturing_year: data.SD_Card_ManufacturingYear_u8, +                manufacturing_month: data.SD_Card_ManufacturingMonth_u8, +                oem: data.SD_Card_OEM_u16, +                manufacturer: data.SD_Card_Manufacturer_u8, +            } +        } +    } +} +  impl From<nitrokey_sys::NK_storage_status> for StorageStatus {      fn from(status: nitrokey_sys::NK_storage_status) -> Self {          StorageStatus { @@ -98,8 +98,8 @@ use nitrokey_sys;  pub use crate::auth::{Admin, Authenticate, User};  pub use crate::config::Config;  pub use crate::device::{ -    connect, connect_model, Device, DeviceWrapper, Model, Pro, Storage, StorageStatus, VolumeMode, -    VolumeStatus, +    connect, connect_model, Device, DeviceWrapper, Model, Pro, SdCardData, Storage, +    StorageProductionInfo, StorageStatus, VolumeMode, VolumeStatus,  };  pub use crate::otp::{ConfigureOtp, GenerateOtp, OtpMode, OtpSlotData};  pub use crate::pws::{GetPasswordSafe, PasswordSafe, SLOT_COUNT};  | 
